98 lines
3.4 KiB
JavaScript
98 lines
3.4 KiB
JavaScript
/**
|
|
* @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;
|
|
}
|