fix: the server could crash if a client sends invalid frames

peers · Nov 17, 2022 · 29394de · 29394de
1 parent 3165d77
commit 29394de
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/services/webSocketServer/index.ts b/src/services/webSocketServer/index.ts
@@ -47,6 +47,9 @@ export class WebSocketServer extends EventEmitter implements IWebSocketServer {
   }
 
   private _onSocketConnection(socket: WebSocket, req: IncomingMessage): void {
+    // An unhandled socket error might crash the server. Handle it first.
+    socket.on("error", error => this._onSocketError(error))
+
     const { query = {} } = url.parse(req.url ?? '', true);
 
     const { id, token, key }: IAuthParams = query;