From 0f04f751e1f99827d556391a38239d917cae5bb1 Mon Sep 17 00:00:00 2001
From: Mononaut <github@monospace.live>
Date: Mon, 1 May 2023 17:59:48 -0600
Subject: [PATCH] Disconnect websocket clients on error

---
 backend/src/api/websocket-handler.ts | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/backend/src/api/websocket-handler.ts b/backend/src/api/websocket-handler.ts
index 3a444701f..cdbf87719 100644
--- a/backend/src/api/websocket-handler.ts
+++ b/backend/src/api/websocket-handler.ts
@@ -96,7 +96,10 @@ class WebsocketHandler {
 
     this.wss.on('connection', (client: WebSocket) => {
       this.numConnected++;
-      client.on('error', logger.info);
+      client.on('error', (e) => {
+        logger.info('websocket client error: ' + (e instanceof Error ? e.message : e));
+        client.close();
+      });
       client.on('close', () => {
         this.numDisconnected++;
       });
@@ -283,6 +286,7 @@ class WebsocketHandler {
           }
         } catch (e) {
           logger.debug('Error parsing websocket message: ' + (e instanceof Error ? e.message : e));
+          client.close();
         }
       });
     });