Catch connection reset in websocket handshake

eclipse · Feb 10, 2024 · 8153db6 · 8153db6
1 parent 4c0d9aa
commit 8153db6
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/paho/mqtt/client.py b/src/paho/mqtt/client.py
@@ -4746,7 +4746,10 @@ def _do_handshake(self, extra_headers: WebSocketHeaders | None) -> None:
 
         while True:
             # read HTTP response header as lines
-            byte = self._socket.recv(1)
+            try:
+                byte = self._socket.recv(1)
+            except ConnectionResetError:
+                byte = b""
 
             self._readbuffer.extend(byte)