Skip to content

Instantly share code, notes, and snippets.

@massyah

massyah/websocket_server.js

Last active April 5, 2017 12:21
Download ZIP

Revisions

  1. massyah renamed this gist Apr 5, 2017. 1 changed file with 0 additions and 0 deletions.
    0 websocket_client.js → websocket_server.js
    View file
    File renamed without changes.
  2. massyah created this gist Apr 5, 2017.
    52 changes: 52 additions & 0 deletions websocket_client.js
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,52 @@
    #!/usr/bin/env node
    var WebSocketServer = require('websocket').server;
    var http = require('http');

    var server = http.createServer(function(request, response) {
    console.log((new Date()) + ' Received request for ' + request.url);
    response.writeHead(404);
    response.end();
    });
    server.listen(8080, function() {
    console.log((new Date()) + ' Server is listening on port 8080');
    });

    wsServer = new WebSocketServer({
    httpServer: server,
    // You should not use autoAcceptConnections for production
    // applications, as it defeats all standard cross-origin protection
    // facilities built into the protocol and the browser. You should
    // *always* verify the connection's origin and decide whether or not
    // to accept it.
    autoAcceptConnections: false
    });

    function originIsAllowed(origin) {
    // put logic here to detect whether the specified origin is allowed.
    return true;
    }

    wsServer.on('request', function(request) {
    if (!originIsAllowed(request.origin)) {
    // Make sure we only accept requests from an allowed origin
    request.reject();
    console.log((new Date()) + ' Connection from origin ' + request.origin + ' rejected.');
    return;
    }

    var connection = request.accept('echo-protocol', request.origin);
    console.log((new Date()) + ' Connection accepted.');
    connection.on('message', function(message) {
    if (message.type === 'utf8') {
    console.log('Received Message: ' + message.utf8Data);
    connection.sendUTF(message.utf8Data);
    }
    else if (message.type === 'binary') {
    console.log('Received Binary Message of ' + message.binaryData.length + ' bytes');
    connection.sendBytes(message.binaryData);
    }
    });
    connection.on('close', function(reasonCode, description) {
    console.log((new Date()) + ' Peer ' + connection.remoteAddress + ' disconnected.');
    });
    });