ws,server: fix server crash

Fix server crash when `metarhia-jstp` is not in the list of the requested protocols. PR-URL: #380 Reviewed-By: Alexey Orlenko <eaglexrlnk@gmail.com>
metarhia · Sep 13, 2018 · 9c7aa23 · 9c7aa23
1 parent b49f2ce
commit 9c7aa23
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/lib/ws-internal.js b/lib/ws-internal.js
@@ -102,7 +102,10 @@ const initServer = function(options, httpServer) {
   httpServer.wsServer = new WebSocketServer(options);
 
   httpServer.wsServer.on('request', request => {
-    if (!httpServer.isOriginAllowed(request.origin)) {
+    if (
+      !httpServer.isOriginAllowed(request.origin) ||
+      !request.requestedProtocols.includes(constants.WEBSOCKET_PROTOCOL_NAME)
+    ) {
       request.reject();
       return;
     }