From 834d4f0a184151bde1ab29bafa1170277247f8f9 Mon Sep 17 00:00:00 2001
From: =?utf8?q?Paul=20H=C3=A4nsch?= <paul@plutz.net>
Date: Wed, 4 Jul 2018 23:39:41 +0200
Subject: [PATCH] Handle arbitrary status response codes in builtin web server

---
 cgilite.sh | 18 +++++++++++++++---
 1 file changed, 15 insertions(+), 3 deletions(-)

diff --git a/cgilite.sh b/cgilite.sh
index 1f2ebb3..7998288 100755
--- a/cgilite.sh
+++ b/cgilite.sh
@@ -81,11 +81,23 @@ if [ -z "$REQUEST_METHOD" -a -z "$SERVER_PROTOCOL" ]; then
     # Without Content-Length header, connection will terminate after
     # script.
 
+    cgilite_status='200 OK'; cgilite_response=''; cgilite_cl='';
     . "$0" |sed '1{s;^Status: ;HTTP/1.1 ;; t; s;^;HTTP/1.1 200 OK\r\n;;}' \
     | while read -r l; do case $l in
-      Content-Length:*) printf '%s\n' "$l"; cat;;
-      $CR) printf 'Connection: close\r\n\r\n'; cat; exit 1;;
-      *) printf '%s\n' "$l";;
+      Status:*) cgilite_status="${l#Status: }";;
+      Content-Length:*) cgilite_response="$cgilite_response${BR}$l"; cgilite_cl='x';;
+      $CR) if [ "$cgilite_cl" ]; then
+             printf '%s %s\r\n%s\n\r\n' \
+               'HTTP/1.1' "${cgilite_status%${CR}}" \
+               "$cgilite_response"
+             cat; exit 0
+           else
+             printf '%s %s\r\n%s\n%s\r\n\r\n' \
+               'HTTP/1.1' "${cgilite_status%${CR}}" \
+               "$cgilite_response" "Connection: close"
+             cat; exit 1
+           fi;;
+      *) cgilite_response="$cgilite_response${BR}$l";;
     esac; done || exit 0;
     (sleep $cgilite_timeout && kill $$) & cgilite_watchdog=$!
   done
-- 
2.39.2