Support ws3

Janix520 · May 22, 2017 · e9bf335 · e9bf335
1 parent 67b6430
commit e9bf335
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/websocket-relay.js b/websocket-relay.js
@@ -24,12 +24,12 @@ var STREAM_SECRET = process.argv[2],
 // Websocket Server
 var socketServer = new WebSocket.Server({port: WEBSOCKET_PORT, perMessageDeflate: false});
 socketServer.connectionCount = 0;
-socketServer.on('connection', function(socket) {
+socketServer.on('connection', function(socket, upgradeReq) {
 	socketServer.connectionCount++;
 	console.log(
 		'New WebSocket Connection: ', 
-		socket.upgradeReq.socket.remoteAddress,
-		socket.upgradeReq.headers['user-agent'],
+		(upgradeReq || socket.upgradeReq).socket.remoteAddress,
+		(upgradeReq || socket.upgradeReq).headers['user-agent'],
 		'('+socketServer.connectionCount+' total)'
 	);
 	socket.on('close', function(code, message){