From 1c96390f3cade804b2667bb69bec4419703a6d56 Mon Sep 17 00:00:00 2001 From: Artur Borecki Date: Fri, 29 Aug 2025 00:23:48 +0200 Subject: [PATCH] feat(backend_server.py): add error handling for HELLO message handler in `_accept_connection()` --- src/judas_server/backend/backend_server.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/src/judas_server/backend/backend_server.py b/src/judas_server/backend/backend_server.py index 1c20b85..b2b1396 100644 --- a/src/judas_server/backend/backend_server.py +++ b/src/judas_server/backend/backend_server.py @@ -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)