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
|
||||
|
||||
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)
|
||||
|
||||
message = message.split(b"\n")[0] # get first line only
|
||||
message = Message.from_bytes(message)
|
||||
|
||||
mac_id = message.payload.get("mac", None)
|
||||
|
||||
Reference in New Issue
Block a user