From 7126838eb01311c3342bb27cfd25f3fdbae9152d Mon Sep 17 00:00:00 2001
From: Jonathan Lamothe <jonathan@jlamothe.net>
Date: Tue, 23 Nov 2021 11:10:34 -0500
Subject: [PATCH] fixed a bug in readLoop that would cause the server hang
 while reading the request

---
 src/Network/Gemini/Capsule/Internal.hs | 24 ++++++++++++++----------
 1 file changed, 14 insertions(+), 10 deletions(-)

diff --git a/src/Network/Gemini/Capsule/Internal.hs b/src/Network/Gemini/Capsule/Internal.hs
index e637d1a..0bf5902 100644
--- a/src/Network/Gemini/Capsule/Internal.hs
+++ b/src/Network/Gemini/Capsule/Internal.hs
@@ -97,8 +97,9 @@ strFromConn maxLen conn = do
       Right s -> Just s
     stripCRLF $ T.unpack txt
 
--- | Reads from a connection up to a maximum number of bytes,
--- returning 'Nothing' if the limit is exceeded
+-- | Reads from a connection up to a maximum number of bytes or a
+-- newline character is encountered, returning 'Nothing' if the limit
+-- is exceeded
 readMax
   :: Int
   -- ^ the maximum number of bytes
@@ -117,13 +118,16 @@ stripCRLF :: String -> Maybe String
 stripCRLF = undefined
 
 readLoop :: Int -> S.InputStream BS.ByteString -> MaybeT IO Builder
-readLoop maxLen src =
-  lift (S.read src) >>= \case
-    Nothing -> return mempty
-    Just bs -> do
-      let len = BS.length bs
-      when (len > maxLen) $
-        fail "maximum length exceeded"
-      (byteString bs <>) <$> readLoop (maxLen - len) src
+readLoop maxLen src = lift (S.read src) >>= \case
+  Nothing -> return mempty
+  Just bs -> do
+    let
+      len = BS.length bs
+      b   = byteString bs
+    when (len > maxLen) $
+      fail "maximum length exceeded"
+    if BS.any (== 0xa) bs
+      then return b
+      else (b <>) <$> readLoop (maxLen - len) src
 
 --jl