From a3bf9690cd964e1b0953d78566c6cabfe95f7263 Mon Sep 17 00:00:00 2001
From: elijahr2411 <elijah@elijahr.dev>
Date: Tue, 7 Feb 2023 12:52:03 -0500
Subject: [PATCH] Add a response for non-websocket connections

---
 src/WSServer.ts | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/src/WSServer.ts b/src/WSServer.ts
index 67fc48b..ec0c2b4 100644
--- a/src/WSServer.ts
+++ b/src/WSServer.ts
@@ -54,6 +54,11 @@ export default class WSServer {
         this.server = http.createServer();
         this.socket = new WebSocketServer({noServer: true});
         this.server.on('upgrade', (req : http.IncomingMessage, socket : internal.Duplex, head : Buffer) => this.httpOnUpgrade(req, socket, head));
+        this.server.on('request', (req, res) => {
+            res.writeHead(426);
+            res.write("This server only accepts WebSocket connections.");
+            res.end();
+        });
         this.socket.on('connection', (ws : WebSocket, req : http.IncomingMessage) => this.onConnection(ws, req));
         var initSize = vm.getSize();
         this.framebuffer = new Framebuffer();