nicokaiser / websocket-protocol-fallback.js

Created September 15, 2011


    
    View websocket-protocol-fallback.js 
  
  
      websocket-protocol-fallback.js
    
    
        Raw
      
    
    File suppressed. Click to show.
  
  
            1
            2
            3
            4
            5
            6
            7
            8
            9
            10
            11
            12
            13
            14
            15
            16
            17
            18
            19
            20
            21
            22
            23
            24
            25
            26
            27
            28
            29
            30
            31
            32
            33
            34
            35
            36
            37
            38
            39
            40
            41
            42
            43
            44
            45
            46
            47
            48
            49
            50
            51
            52
            53
            54
            55
            56
            57
            58
            59
            60
            61
            62
            63
            64
            65
            66
            67
            68
            69
            70
            71
            72
            73
            74
            75
            76
            77
            78
            79
            80
            81
            82
            83
            84
            85
            86
            87
            88
            89
            90
            91
            92
            93
            94
            95
            96
            97
            98
            99
            100
            101
            102
            103
            104
            105
            106
            107
            108
            109
            110
            111
            112
          
          
            #!/usr/bin/env node
 
// Example of how to fallback to alternative websocket library for old protocol clients
// see https://gist.github.com/1148686
 
var http = require('http'),
    WebSocketRequest = require('websocket').request,
    ws = require('websocket-server');
 
 
var httpServer = http.createServer(function(request, response) {
    console.log((new Date()) + " Received request for " + request.url);
    response.writeHead(404);
    response.end();
});
httpServer.listen(10080, function() {
    console.log((new Date()) + " Server is listening on port 10080");
});
 
 
// node-websocket-server
 
var miksagoConnection = require('./node_modules/websocket-server/lib/ws/connection');
 
var miksagoServer = ws.createServer();
miksagoServer.server = httpServer;
 
miksagoServer.addListener('connection', function(connection) {
    // Add remoteAddress property
    connection.remoteAddress = connection._socket.remoteAddress;
    
    // We want to use "sendUTF" regardless of the server implementation
    connection.sendUTF = connection.send;
    handleConnection(connection);
});
 
 
// WebSocket-Node config
 
var wsServerConfig =  {
    // All options *except* 'httpServer' are required when bypassing
    // WebSocketServer.
    maxReceivedFrameSize: 0x10000,
    maxReceivedMessageSize: 0x100000,
    fragmentOutgoingMessages: true,
    fragmentationThreshold: 0x4000,
    keepalive: true,
    keepaliveInterval: 20000,
    assembleFragments: true,
    // autoAcceptConnections is not applicable when bypassing WebSocketServer
    // autoAcceptConnections: false,
    disableNagleAlgorithm: true,
    closeTimeout: 5000
};
 
 
// Handle the upgrade event ourselves instead of using WebSocketServer
 
httpServer.on('upgrade', function(req, socket, head) {
 
    if (typeof req.headers['sec-websocket-version'] !== 'undefined') {
 
        // WebSocket hybi-08/-09/-10 connection (WebSocket-Node)
        var wsRequest = new WebSocketRequest(socket, req, wsServerConfig);
        try {
            wsRequest.readHandshake();
            var wsConnection = wsRequest.accept(wsRequest.requestedProtocols[0], wsRequest.origin);
            handleConnection(wsConnection);
        }
        catch(e) {
            console.log("WebSocket Request unsupported by WebSocket-Node: " + e.toString());
            return;
        }
 
    } else {
 
        // WebSocket hixie-75/-76/hybi-00 connection (node-websocket-server)
        if (req.method === 'GET' &&
            (req.headers.upgrade && req.headers.connection) &&
            req.headers.upgrade.toLowerCase() === 'websocket' &&
            req.headers.connection.toLowerCase() === 'upgrade') {
            new miksagoConnection(miksagoServer.manager, miksagoServer.options, req, socket, head);
        }
 
    }
 
});
 
 
// A common connection handler
 
function handleConnection(connection) {
    console.log((new Date()) + " Connection accepted.");
    connection.addListener('message', function(wsMessage) {
        var message = wsMessage;
 
        // WebSocket-Node adds a "type", node-websocket-server does not
        if (typeof wsMessage.type !== 'undefined') {
            if (wsMessage.type !== 'utf8') {
                return;
            }
            message = wsMessage.utf8Data;
        }
        console.log("Received Message: " + message);
        
        connection.sendUTF(message);
    });
    
    connection.addListener('close', function() {
        console.log((new Date()) + " Peer " + connection.remoteAddress + " disconnected.");
    });
}