feat(backend_server.py): add error handling for HELLO message handler in _accept_connection()
This commit is contained in:
@@ -73,9 +73,19 @@ class BackendServer:
|
|||||||
# wait for hello message to get mac_id
|
# wait for hello message to get mac_id
|
||||||
|
|
||||||
conn.settimeout(5)
|
conn.settimeout(5)
|
||||||
message = conn.recv(1024)
|
try:
|
||||||
|
message = conn.recv(1024)
|
||||||
|
if not message:
|
||||||
|
self.logger.error(f"[-] No data received from {addr}")
|
||||||
|
conn.close()
|
||||||
|
return
|
||||||
|
except socket.timeout:
|
||||||
|
self.logger.error(f"[-] Timeout waiting for hello from {addr}")
|
||||||
|
conn.close()
|
||||||
|
return
|
||||||
conn.settimeout(None)
|
conn.settimeout(None)
|
||||||
|
|
||||||
|
message = message.split(b"\n")[0] # get first line only
|
||||||
message = Message.from_bytes(message)
|
message = Message.from_bytes(message)
|
||||||
|
|
||||||
mac_id = message.payload.get("mac", None)
|
mac_id = message.payload.get("mac", None)
|
||||||
|
|||||||
Reference in New Issue
Block a user