mirror.MikroTik/tomaae.homeassistant-mikrotik_router
1
0
Fork 0
mirror of https://github.com/tomaae/homeassistant-mikrotik_router.git synced 2025-07-05 15:04:30 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

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

Browse source
This commit is contained in:
tomaae 2020-03-12 13:21:03 +01:00
parent d9dd5330ee
commit d68ffbdc46
1 changed files with 7 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

8
custom_components/mikrotik_router/librouteros_custom/connections.py
View file

@ -22,7 +22,13 @@ class SocketTransport:
""" """
data = bytearray() data = bytearray()
while len(data) != length: 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: if not data:
raise ConnectionClosed('Connection unexpectedly closed.') raise ConnectionClosed('Connection unexpectedly closed.')
return data return data