Fixed not check data length read from socket (#110)

eandersson · Oct 22, 2021 · b849b43 · b849b43
1 parent 3abf0ee
commit b849b43
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/amqpstorm/io.py b/amqpstorm/io.py
@@ -270,6 +270,8 @@ def _receive(self):
         data_in = EMPTY_BUFFER
         try:
             data_in = self._read_from_socket()
+            if len(data_in) == 0:
+                raise socket.error("connection closed by server")
         except socket.timeout:
             pass
         except compatibility.SSLWantReadError: