Check received message length to prevent infinite loop in _recv_request method

1 files changed, 4 insertions, 1 deletions

diff --git a/main.py b/main.py
index 46740e9..18772f3 100644
--- a/main.py
+++ b/main.py
@@ -135,7 +135,10 @@ class HTTPServer:
 
         buffer = bytearray()
         while not self.stop_event.is_set():
-            buffer += client.recv(self.buf_len)
+            msg = client.recv(self.buf_len)
+            if len(msg) == 0:
+                break
+            buffer += msg
             if buffer[-4:] == b'\r\n\r\n':
                 return Request.create(buffer)
         return None