MeshCentral/meshrelay.js

/**
* @description Meshcentral MeshRelay
* @author Ylian Saint-Hilaire
* @version v0.0.1
*/

// Construct a MeshRelay object, called upon connection
module.exports.CreateMeshRelayKey = function (parent, func) {
    parent.crypto.randomBytes(16, function (err, buf) {
        var key = buf.toString('hex').toUpperCase() + ':' + Date.now();
        key += ':' + parent.crypto.createHmac('SHA256', parent.relayRandom).update(key).digest('hex');
        func(key);
    });
}

module.exports.CreateMeshRelay = function (parent, ws, req) {
    var obj = {};
    obj.ws = ws;
    obj.peer = null;
    obj.id = req.query['id'];

    //console.log('Got relay connection for: ' + obj.id);
    if (obj.id == undefined) { obj.ws.close(); obj.id = null; return null; } // Attempt to connect without id, drop this.

    // Validate that the id is valid, we only need to do this on non-authenticated sessions.
    // TODO: Figure out when this needs to be done.
    /*
    if (!parent.args.notls) {
        // Check the identifier, if running without TLS, skip this.
        var ids = obj.id.split(':');
        if (ids.length != 3) { obj.ws.close(); obj.id = null; return null; } // Invalid ID, drop this.
        if (parent.crypto.createHmac('SHA256', parent.relayRandom).update(ids[0] + ':' + ids[1]).digest('hex') != ids[2]) { obj.ws.close(); obj.id = null; return null; } // Invalid HMAC, drop this.
        if ((Date.now() - parseInt(ids[1])) > 120000) { obj.ws.close(); obj.id = null; return null; } // Expired time, drop this.
        obj.id = ids[0];
    }
    */

    // Check the peer connection status
    {
        var relayinfo = parent.wsrelays[obj.id];
        if (relayinfo) {
            if (relayinfo.state == 1) {
                // Connect to peer
                obj.peer = relayinfo.peer1;
                obj.peer.peer = obj;
                relayinfo.peer2 = obj;
                relayinfo.state = 2;
                obj.ws.send('c'); // Send connect to both peers
                relayinfo.peer1.ws.send('c');

                relayinfo.peer1.ws.peer = relayinfo.peer2.ws;
                relayinfo.peer2.ws.peer = relayinfo.peer1.ws;

            } else {
                // Connected already, drop (TODO: maybe we should re-connect?)
                obj.id = null;
                obj.ws.close();
                return null;
            }
        } else {
            // Setup the connection, wait for peer
            parent.wsrelays[obj.id] = { peer1 : obj, state : 1 };
        }
    }
    
    ws.flushSink = function () {
        try { ws.resume(); } catch (e) { }
    };

    // When data is received from the mesh relay web socket
    ws.on('message', function (data) 
        {
            if (this.peer != null) { try { this.pause(); this.peer.send(data, ws.flushSink); } catch (e) { } }
        });

    // If error, do nothing
    ws.on('error', function (err) { console.log(err); });

    // If the mesh relay web socket is closed
    ws.on('close', function (req) {
        //console.log('Got relay disconnection for: ' + obj.id);
        if (obj.id != null) {
            var relayinfo = parent.wsrelays[obj.id];
            if (relayinfo.state == 2) {
                // Disconnect the peer
                var peer = (relayinfo.peer1 == obj)?relayinfo.peer2:relayinfo.peer1;
                peer.id = null;
                peer.ws._socket.end();
            }
            delete parent.wsrelays[obj.id];
            obj.peer = null;
            obj.id = null;
        }
    });
    
    return obj;
}