agatha/pyc2
1
0
Fork 0
Code Issues 1 Pull Requests Packages Projects Releases Wiki Activity

fix: handle uncaught UnicodeDecodeError

Browse Source 
UnicodeDecodeError occurred when trying to connect to the server with HTTPS. The error
was not caught by the broad Exception clause in `handle_agent`, causing a traceback
to be dumped.
This commit is contained in:
agatha 2024-09-07 17:09:32 -04:00
parent f71d86fce7
commit 6c0eb7f941
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
server/src/main.py
View File

@ -87,6 +87,8 @@ class Server:
conn.send(json.dumps({"command": command}).encode()) conn.send(json.dumps({"command": command}).encode())
except json.decoder.JSONDecodeError: except json.decoder.JSONDecodeError:
logger.error(f"Invalid JSON received from {addr}") logger.error(f"Invalid JSON received from {addr}")
except UnicodeDecodeError:
logger.error(f"Invalid UTF-8 received from {addr}")
except Exception as e: except Exception as e:
logger.error(f"Error handling agent {agent_id}: {e}") logger.error(f"Error handling agent {agent_id}: {e}")
finally: finally: