From 9204669a7f0c21ec330ba49f19f9e7887a330d18 Mon Sep 17 00:00:00 2001 From: n9k Date: Tue, 22 Feb 2022 11:43:36 +0000 Subject: [PATCH] Catch exception when inbound websocket data is not JSON --- anonstream/websocket.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/anonstream/websocket.py b/anonstream/websocket.py index b575aca..26b1340 100644 --- a/anonstream/websocket.py +++ b/anonstream/websocket.py @@ -1,4 +1,5 @@ import asyncio +import json from quart import current_app, websocket @@ -33,8 +34,12 @@ async def websocket_outbound(queue, user): async def websocket_inbound(queue, user): while True: - receipt = await websocket.receive_json() - see(user) + try: + receipt = await websocket.receive_json() + except json.JSONDecodeError: + receipt = None + finally: + see(user) try: nonce, comment, digest, answer = parse_websocket_data(receipt) except Malformed as e: