catch an error on recv in librouteros and raise an error #8

tomaae · Mar 12, 2020 · d68ffbd · d68ffbd
1 parent d9dd533
commit d68ffbd
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/custom_components/mikrotik_router/librouteros_custom/connections.py b/custom_components/mikrotik_router/librouteros_custom/connections.py
@@ -22,7 +22,13 @@ def read(self, length):
         """
         data = bytearray()
         while len(data) != length:
-            data += self.sock.recv((length - len(data)))
+            try:
+                tmp = self.sock.recv((length - len(data)))
+            except:
+                raise ConnectionClosed('Socket recv failed.')
+            finally:
+                data += tmp
+
             if not data:
                 raise ConnectionClosed('Connection unexpectedly closed.')
         return data