2017-08-28 12:27:45 -04:00
/ * *
* @ description Meshcentral Multi - Server Support
* @ author Ylian Saint - Hilaire
* @ version v0 . 0.1
* /
// Construct a Mesh Multi-Server object. This is used for MeshCentral-to-MeshCentral communication.
module . exports . CreateMultiServer = function ( parent , args ) {
var obj = { } ;
2017-10-15 02:22:19 -04:00
const WebSocket = require ( 'ws' ) ;
2017-08-28 12:27:45 -04:00
obj . parent = parent ;
obj . crypto = require ( 'crypto' ) ;
2017-09-13 14:25:57 -04:00
obj . peerConfig = parent . config . peers ;
obj . forge = require ( 'node-forge' ) ;
obj . outPeerServers = { } ; // Outgoing peer servers
obj . peerServers = { } ; // All connected servers (in & out). Only present in this list if the connection is setup
2017-09-15 14:45:06 -04:00
obj . serverid = null ;
2017-09-13 14:25:57 -04:00
// Create a mesh server module that will connect to other servers
obj . CreatePeerOutServer = function ( parent , serverid , url ) {
var obj = { } ;
obj . parent = parent ;
obj . serverid = serverid ;
obj . url = url ;
obj . ws = null ;
obj . certificates = parent . parent . certificates ;
obj . common = require ( './common.js' ) ;
obj . forge = require ( 'node-forge' ) ;
obj . crypto = require ( 'crypto' ) ;
obj . pki = obj . forge . pki ;
obj . connectionState = 0 ;
obj . retryTimer = null ;
obj . retryBackoff = 0 ;
obj . connectHandler = null ;
2017-10-18 19:28:05 -04:00
obj . webCertificateHash = obj . parent . parent . webserver . webCertificateHash ;
obj . agentCertificateHashBase64 = obj . parent . parent . webserver . agentCertificateHashBase64 ;
obj . agentCertificateAsn1 = obj . parent . parent . webserver . agentCertificateAsn1 ;
2017-09-13 14:25:57 -04:00
obj . peerServerId = null ;
obj . authenticated = 0 ;
2017-09-17 20:22:18 -04:00
obj . serverCertHash = null ;
2017-09-13 14:25:57 -04:00
// Disconnect from the server and/or stop trying
obj . stop = function ( ) {
obj . connectionState = 0 ;
disconnect ( ) ;
}
// Make one attempt at connecting to the server
function connect ( ) {
obj . retryTimer = null ;
obj . connectionState = 1 ;
// Get the web socket setup
2017-10-15 02:22:19 -04:00
obj . ws = new WebSocket ( obj . url + 'meshserver.ashx' , { rejectUnauthorized : false , cert : obj . certificates . agent . cert , key : obj . certificates . agent . key } ) ;
2017-09-13 14:25:57 -04:00
obj . parent . parent . debug ( 1 , 'OutPeer ' + obj . serverid + ': Connecting to: ' + url + 'meshserver.ashx' ) ;
// Register the connection failed event
2017-10-15 02:22:19 -04:00
obj . ws . on ( 'error' , function ( error ) { obj . parent . parent . debug ( 1 , 'OutPeer ' + obj . serverid + ': Error: ' + error ) ; disconnect ( ) ; } ) ;
obj . ws . on ( 'close' , function ( ) { obj . parent . parent . debug ( 1 , 'OutPeer ' + obj . serverid + ': Disconnected' ) ; disconnect ( ) ; } ) ;
2017-09-13 14:25:57 -04:00
// Register the connection event
2017-10-15 02:22:19 -04:00
obj . ws . on ( 'open' , function ( ) {
2017-09-13 14:25:57 -04:00
obj . parent . parent . debug ( 1 , 'OutPeer ' + obj . serverid + ': Connected' ) ;
obj . connectionState |= 2 ;
2017-10-15 02:22:19 -04:00
obj . nonce = obj . forge . random . getBytesSync ( 48 ) ;
2017-09-13 14:25:57 -04:00
// Get the peer server's certificate and compute the server public key hash
2017-10-15 02:22:19 -04:00
if ( obj . ws . _socket == null ) return ;
var serverCert = obj . forge . pki . certificateFromAsn1 ( obj . forge . asn1 . fromDer ( obj . ws . _socket . getPeerCertificate ( ) . raw . toString ( 'binary' ) ) ) ;
obj . serverCertHash = obj . forge . pki . getPublicKeyFingerprint ( serverCert . publicKey , { encoding : 'binary' , md : obj . forge . md . sha384 . create ( ) } ) ;
// Start authenticate the peer server by sending a auth nonce & server TLS cert hash.
// Send 384 bits SHA384 hash of TLS cert public key + 384 bits nonce
obj . ws . send ( obj . common . ShortToStr ( 1 ) + obj . serverCertHash + obj . nonce ) ; // Command 1, hash + nonce
} ) ;
// If a message is received
obj . ws . on ( 'message' , function ( msg ) {
if ( typeof msg != 'string' ) { msg = msg . toString ( 'binary' ) ; }
if ( msg . length < 2 ) return ;
if ( msg . charCodeAt ( 0 ) == 123 ) {
if ( obj . connectionState == 15 ) { processServerData ( msg ) ; }
} else {
var cmd = obj . common . ReadShort ( msg , 0 ) ;
switch ( cmd ) {
case 1 : {
// Server authentication request
if ( msg . length != 98 ) { obj . parent . parent . debug ( 1 , 'OutPeer: BAD MESSAGE(A1)' ) ; return ; }
// Check that the server hash matches the TLS server certificate public key hash
if ( obj . serverCertHash != msg . substring ( 2 , 50 ) ) { obj . parent . parent . debug ( 1 , 'OutPeer: Server hash mismatch.' ) ; disconnect ( ) ; return ; }
obj . servernonce = msg . substring ( 50 ) ;
// Use our agent certificate root private key to sign the ServerHash + ServerNonce + PeerNonce
var privateKey = obj . forge . pki . privateKeyFromPem ( obj . certificates . agent . key ) ;
var md = obj . forge . md . sha384 . create ( ) ;
md . update ( msg . substring ( 2 ) , 'binary' ) ;
md . update ( obj . nonce , 'binary' ) ;
// Send back our certificate + signature
2017-10-18 19:28:05 -04:00
agentRootCertificateAsn1 = obj . forge . asn1 . toDer ( obj . forge . pki . certificateToAsn1 ( obj . forge . pki . certificateFromPem ( obj . certificates . agent . cert ) ) ) . getBytes ( ) ;
obj . ws . send ( obj . common . ShortToStr ( 2 ) + obj . common . ShortToStr ( agentRootCertificateAsn1 . length ) + agentRootCertificatAsn1 + privateKey . sign ( md ) ) ; // Command 3, signature
2017-10-15 02:22:19 -04:00
break ;
}
case 2 : {
// Server certificate
var certlen = obj . common . ReadShort ( msg , 2 ) , serverCert = null ;
try { serverCert = obj . forge . pki . certificateFromAsn1 ( obj . forge . asn1 . fromDer ( msg . substring ( 4 , 4 + certlen ) ) ) ; } catch ( e ) { }
if ( serverCert == null ) { obj . parent . parent . debug ( 1 , 'OutPeer: Invalid server certificate.' ) ; disconnect ( ) ; return ; }
2017-10-15 20:36:06 -04:00
var serverid = new Buffer ( obj . forge . pki . getPublicKeyFingerprint ( serverCert . publicKey , { encoding : 'binary' , md : obj . forge . md . sha384 . create ( ) } ) , 'binary' ) . toString ( 'base64' ) . replace ( /\+/g , '@' ) . replace ( /\//g , '$' ) ;
2017-10-18 19:28:05 -04:00
if ( serverid !== obj . agentCertificateHashBase64 ) { obj . parent . parent . debug ( 1 , 'OutPeer: Server hash mismatch.' ) ; disconnect ( ) ; return ; }
2017-10-15 02:22:19 -04:00
// Server signature, verify it
var md = obj . forge . md . sha384 . create ( ) ;
md . update ( obj . serverCertHash , 'binary' ) ;
md . update ( obj . nonce , 'binary' ) ;
md . update ( obj . servernonce , 'binary' ) ;
if ( serverCert . publicKey . verify ( md . digest ( ) . bytes ( ) , msg . substring ( 4 + certlen ) ) == false ) { obj . parent . parent . debug ( 1 , 'OutPeer: Server sign check failed.' ) ; disconnect ( ) ; return ; }
// Connection is a success, clean up
delete obj . nonce ;
delete obj . servernonce ;
2017-10-15 20:36:06 -04:00
obj . serverCertHash = new Buffer ( obj . serverCertHash , 'binary' ) . toString ( 'base64' ) . replace ( /\+/g , '@' ) . replace ( /\//g , '$' ) ; // Change this value to base64
2017-10-15 02:22:19 -04:00
obj . connectionState |= 4 ;
obj . retryBackoff = 0 ; // Set backoff connection timer back to fast.
obj . parent . parent . debug ( 1 , 'OutPeer ' + obj . serverid + ': Verified peer connection to ' + obj . url ) ;
// Send information about our server to the peer
2017-10-23 17:09:58 -04:00
if ( obj . connectionState == 15 ) { obj . ws . send ( JSON . stringify ( { action : 'info' , serverid : obj . parent . serverid , dbid : obj . parent . parent . db . identifier , key : obj . parent . parent . webserver . serverKey . toString ( 'hex' ) , serverCertHash : obj . parent . parent . webserver . webCertificateHashBase64 } ) ) ; }
2017-10-15 02:22:19 -04:00
//if ((obj.connectionState == 15) && (obj.connectHandler != null)) { obj.connectHandler(1); }
break ;
}
case 4 : {
// Server confirmed authentication, we are allowed to send commands to the server
obj . connectionState |= 8 ;
2017-10-23 17:09:58 -04:00
if ( obj . connectionState == 15 ) { obj . ws . send ( JSON . stringify ( { action : 'info' , serverid : obj . parent . serverid , dbid : obj . parent . parent . db . identifier , key : obj . parent . parent . webserver . serverKey . toString ( 'hex' ) , serverCertHash : obj . parent . parent . webserver . webCertificateHashBase64 } ) ) ; }
2017-10-15 02:22:19 -04:00
//if ((obj.connectionState == 15) && (obj.connectHandler != null)) { obj.connectHandler(1); }
break ;
}
default : {
obj . parent . parent . debug ( 1 , 'OutPeer ' + obj . serverid + ': Un-handled command: ' + cmd ) ;
break ;
2017-09-13 14:25:57 -04:00
}
}
2017-10-15 02:22:19 -04:00
}
2017-09-13 14:25:57 -04:00
} ) ;
}
// Disconnect from the server, if we need to, try again with a delay.
function disconnect ( ) {
if ( obj . authenticated == 3 ) { obj . parent . ClearPeerServer ( obj , obj . peerServerId ) ; obj . authenticated = 0 ; }
if ( ( obj . connectionState == 15 ) && ( obj . connectHandler != null ) ) { obj . connectHandler ( 0 ) ; }
2017-10-15 02:22:19 -04:00
if ( obj . ws != null ) { obj . ws . close ( ) ; obj . ws = null ; }
2017-09-13 14:25:57 -04:00
if ( obj . retryTimer != null ) { clearTimeout ( obj . retryTimer ) ; obj . retryTimer = null ; }
// Re-try connection
if ( obj . connectionState >= 1 ) { obj . connectionState = 1 ; if ( obj . retryTimer == null ) { obj . retryTimer = setTimeout ( connect , getConnectRetryTime ( ) ) ; } }
}
// Get the next retry time in milliseconds
function getConnectRetryTime ( ) {
if ( obj . retryBackoff < 30000 ) { obj . retryBackoff += Math . floor ( ( Math . random ( ) * 3000 ) + 1000 ) ; }
return obj . retryBackoff ;
}
// Send a JSON message to the peer server
obj . send = function ( msg ) {
try {
2017-10-15 02:22:19 -04:00
if ( obj . ws == null || obj . connectionState != 15 ) { return ; }
if ( typeof msg == 'object' ) { obj . ws . send ( JSON . stringify ( msg ) ) ; return ; }
if ( typeof msg == 'string' ) { obj . ws . send ( msg ) ; return ; }
2017-09-13 14:25:57 -04:00
} catch ( e ) { }
}
// Process incoming peer server JSON data
function processServerData ( msg ) {
var str = msg . toString ( 'utf8' ) ;
if ( str [ 0 ] == '{' ) {
try { command = JSON . parse ( str ) } catch ( e ) { obj . parent . parent . debug ( 1 , 'Unable to parse JSON (' + obj . remoteaddr + ').' ) ; return ; } // If the command can't be parsed, ignore it.
if ( command . action == 'info' ) {
if ( obj . authenticated != 3 ) {
// We get the peer's serverid and database identifier.
if ( ( command . serverid != null ) && ( command . dbid != null ) ) {
2017-09-15 14:45:06 -04:00
if ( command . serverid == obj . parent . serverid ) { console . log ( 'ERROR: Same server ID, trying to peer with self. (' + obj . url + ', ' + command . serverid + ').' ) ; return ; }
2017-09-13 14:25:57 -04:00
if ( command . dbid != obj . parent . parent . db . identifier ) { console . log ( 'ERROR: Database ID mismatch. Trying to peer to a server with the wrong database. (' + obj . url + ', ' + command . serverid + ').' ) ; return ; }
2017-10-15 20:36:06 -04:00
if ( obj . serverCertHash != command . serverCertHash ) { console . log ( 'ERROR: Outer certificate hash mismatch (2). (' + obj . url + ', ' + command . serverid + ').' ) ; return ; }
2017-09-13 14:25:57 -04:00
obj . peerServerId = command . serverid ;
2017-10-15 02:22:19 -04:00
obj . peerServerKey = new Buffer ( command . key , 'hex' ) ;
2017-09-13 14:25:57 -04:00
obj . authenticated = 3 ;
obj . parent . SetupPeerServer ( obj , obj . peerServerId ) ;
}
}
} else if ( obj . authenticated == 3 ) {
// Pass the message to the parent object for processing.
obj . parent . ProcessPeerServerMessage ( obj , obj . peerServerId , command ) ;
}
}
}
connect ( ) ;
return obj ;
}
// Create a mesh server module that received a connection to another server
obj . CreatePeerInServer = function ( parent , ws , req ) {
var obj = { } ;
obj . ws = ws ;
obj . parent = parent ;
obj . common = require ( './common.js' ) ;
obj . forge = require ( 'node-forge' ) ;
obj . crypto = require ( 'crypto' ) ;
obj . authenticated = 0 ;
obj . remoteaddr = obj . ws . _socket . remoteAddress ;
obj . receivedCommands = 0 ;
2017-10-18 19:28:05 -04:00
obj . webCertificateHash = obj . parent . parent . webserver . webCertificateHash ;
obj . agentCertificateHashBase64 = obj . parent . parent . webserver . agentCertificateHashBase64 ;
obj . agentCertificateAsn1 = obj . parent . parent . webserver . agentCertificateAsn1 ;
2017-09-13 14:25:57 -04:00
obj . infoSent = 0 ;
obj . peerServerId = null ;
2017-09-17 20:22:18 -04:00
obj . serverCertHash = null ;
2017-09-13 14:25:57 -04:00
if ( obj . remoteaddr . startsWith ( '::ffff:' ) ) { obj . remoteaddr = obj . remoteaddr . substring ( 7 ) ; }
2017-10-15 02:22:19 -04:00
obj . parent . parent . debug ( 1 , 'InPeer: Connected (' + obj . remoteaddr + ')' ) ;
2017-09-13 14:25:57 -04:00
// Send a message to the peer server
obj . send = function ( data ) {
try {
if ( typeof data == 'string' ) { obj . ws . send ( new Buffer ( data , 'binary' ) ) ; return ; }
if ( typeof data == 'object' ) { obj . ws . send ( JSON . stringify ( data ) ) ; return ; }
obj . ws . send ( data ) ;
} catch ( e ) { }
}
// Disconnect this server
obj . close = function ( arg ) {
if ( ( arg == 1 ) || ( arg == null ) ) { try { obj . ws . close ( ) ; obj . parent . parent . debug ( 1 , 'InPeer: Soft disconnect ' + obj . peerServerId + ' (' + obj . remoteaddr + ')' ) ; } catch ( e ) { console . log ( e ) ; } } // Soft close, close the websocket
if ( arg == 2 ) { try { obj . ws . _socket . _parent . end ( ) ; obj . parent . parent . debug ( 1 , 'InPeer: Hard disconnect ' + obj . peerServerId + ' (' + obj . remoteaddr + ')' ) ; } catch ( e ) { console . log ( e ) ; } } // Hard close, close the TCP socket
if ( obj . authenticated == 3 ) { obj . parent . ClearPeerServer ( obj , obj . peerServerId ) ; obj . authenticated = 0 ; }
}
2017-10-15 02:22:19 -04:00
// When data is received from the peer server web socket
2017-09-13 14:25:57 -04:00
ws . on ( 'message' , function ( msg ) {
2017-10-15 02:22:19 -04:00
if ( typeof msg != 'string' ) { msg = msg . toString ( 'binary' ) ; }
2017-09-13 14:25:57 -04:00
if ( msg . length < 2 ) return ;
if ( obj . authenticated >= 2 ) { // We are authenticated
if ( msg . charCodeAt ( 0 ) == 123 ) { processServerData ( msg ) ; }
if ( msg . length < 2 ) return ;
var cmdid = obj . common . ReadShort ( msg , 0 ) ;
// Process binary commands (if any). None right now.
}
else if ( obj . authenticated < 2 ) { // We are not authenticated
var cmd = obj . common . ReadShort ( msg , 0 ) ;
if ( cmd == 1 ) {
2017-10-15 02:22:19 -04:00
// Peer server authentication request
if ( ( msg . length != 98 ) || ( ( obj . receivedCommands & 1 ) != 0 ) ) return ;
obj . receivedCommands += 1 ; // Peer server can't send the same command twice on the same connection ever. Block DOS attack path.
2017-09-13 14:25:57 -04:00
// Check that the server hash matches out own web certificate hash
2017-10-18 19:28:05 -04:00
if ( obj . webCertificateHash != msg . substring ( 2 , 50 ) ) { obj . close ( ) ; return ; }
2017-09-13 14:25:57 -04:00
2017-10-15 02:22:19 -04:00
// Use our server private key to sign the ServerHash + PeerNonce + ServerNonce
2017-09-13 14:25:57 -04:00
var privateKey = obj . forge . pki . privateKeyFromPem ( obj . parent . parent . certificates . agent . key ) ;
2017-10-15 02:22:19 -04:00
var md = obj . forge . md . sha384 . create ( ) ;
2017-09-13 14:25:57 -04:00
md . update ( msg . substring ( 2 ) , 'binary' ) ;
md . update ( obj . nonce , 'binary' ) ;
2017-10-15 02:22:19 -04:00
obj . peernonce = msg . substring ( 50 ) ;
2017-09-13 14:25:57 -04:00
// Send back our certificate + signature
2017-10-18 19:28:05 -04:00
obj . send ( obj . common . ShortToStr ( 2 ) + obj . common . ShortToStr ( obj . agentCertificateAsn1 . length ) + obj . agentCertificateAsn1 + privateKey . sign ( md ) ) ; // Command 2, certificate + signature
2017-09-13 14:25:57 -04:00
2017-10-15 02:22:19 -04:00
// Check the peer server signature if we can
2017-09-15 14:45:06 -04:00
if ( obj . unauthsign != null ) {
2017-10-15 02:22:19 -04:00
if ( processPeerSignature ( obj . unauthsign ) == false ) { disconnect ( ) ; return ; } else { completePeerServerConnection ( ) ; }
2017-09-13 14:25:57 -04:00
}
}
else if ( cmd == 2 ) {
2017-10-15 02:22:19 -04:00
// Peer server certificate
2017-09-13 14:25:57 -04:00
if ( ( msg . length < 4 ) || ( ( obj . receivedCommands & 2 ) != 0 ) ) return ;
2017-10-15 02:22:19 -04:00
obj . receivedCommands += 2 ; // Peer server can't send the same command twice on the same connection ever. Block DOS attack path.
2017-09-13 14:25:57 -04:00
// Decode the certificate
var certlen = obj . common . ReadShort ( msg , 2 ) ;
obj . unauth = { } ;
obj . unauth . nodeCert = null ;
try { obj . unauth . nodeCert = obj . forge . pki . certificateFromAsn1 ( obj . forge . asn1 . fromDer ( msg . substring ( 4 , 4 + certlen ) ) ) ; } catch ( e ) { return ; }
2017-10-15 20:36:06 -04:00
obj . unauth . nodeid = new Buffer ( obj . forge . pki . getPublicKeyFingerprint ( obj . unauth . nodeCert . publicKey , { encoding : 'binary' , md : obj . forge . md . sha384 . create ( ) } ) , 'binary' ) . toString ( 'base64' ) . replace ( /\+/g , '@' ) . replace ( /\//g , '$' ) ;
2017-09-13 14:25:57 -04:00
2017-10-15 02:22:19 -04:00
// Check the peer server signature if we can
2017-10-15 20:36:06 -04:00
if ( obj . peernonce == null ) { obj . unauthsign = msg . substring ( 4 + certlen ) ; } else { if ( processPeerSignature ( msg . substring ( 4 + certlen ) ) == false ) { obj . close ( ) ; return ; } }
2017-09-13 14:25:57 -04:00
completePeerServerConnection ( ) ;
}
else if ( cmd == 3 ) {
if ( ( msg . length < 56 ) || ( ( obj . receivedCommands & 4 ) != 0 ) ) return ;
2017-10-15 02:22:19 -04:00
obj . receivedCommands += 4 ; // Peer server can't send the same command twice on the same connection ever. Block DOS attack path.
2017-09-13 14:25:57 -04:00
completePeerServerConnection ( ) ;
}
}
} ) ;
// If error, do nothing
ws . on ( 'error' , function ( err ) { obj . parent . parent . debug ( 1 , 'InPeer: Connection Error: ' + err ) ; } ) ;
2017-10-15 02:22:19 -04:00
// If the peer server web socket is closed, clean up.
2017-09-13 14:25:57 -04:00
ws . on ( 'close' , function ( req ) { obj . parent . parent . debug ( 1 , 'InPeer disconnect ' + obj . nodeid + ' (' + obj . remoteaddr + ')' ) ; obj . close ( 0 ) ; } ) ;
2017-10-15 02:22:19 -04:00
// obj.ws._socket._parent.on('close', function (req) { obj.parent.parent.debug(1, 'Peer server TCP disconnect ' + obj.nodeid + ' (' + obj.remoteaddr + ')'); });
2017-09-13 14:25:57 -04:00
2017-10-15 02:22:19 -04:00
// Start authenticate the peer server by sending a auth nonce & server TLS cert hash.
// Send 384 bits SHA382 hash of TLS cert public key + 384 bits nonce
obj . nonce = obj . forge . random . getBytesSync ( 48 ) ;
2017-10-18 19:28:05 -04:00
obj . send ( obj . common . ShortToStr ( 1 ) + obj . webCertificateHash + obj . nonce ) ; // Command 1, hash + nonce
2017-09-13 14:25:57 -04:00
2017-10-15 02:22:19 -04:00
// Once we get all the information about an peer server, run this to hook everything up to the server
2017-09-13 14:25:57 -04:00
function completePeerServerConnection ( ) {
if ( obj . authenticated != 1 ) return ;
obj . send ( obj . common . ShortToStr ( 4 ) ) ;
2017-10-23 17:09:58 -04:00
obj . send ( JSON . stringify ( { action : 'info' , serverid : obj . parent . serverid , dbid : obj . parent . parent . db . identifier , key : obj . parent . parent . webserver . serverKey . toString ( 'hex' ) , serverCertHash : obj . parent . parent . webserver . webCertificateHashBase64 } ) ) ;
2017-09-13 14:25:57 -04:00
obj . authenticated = 2 ;
}
2017-10-15 02:22:19 -04:00
// Verify the peer server signature
function processPeerSignature ( msg ) {
var md = obj . forge . md . sha384 . create ( ) ; // TODO: Switch this to SHA384 on node instead of forge.
2017-10-18 19:28:05 -04:00
md . update ( obj . parent . parent . webserver . webCertificateHash , 'binary' ) ;
2017-09-13 14:25:57 -04:00
md . update ( obj . nonce , 'binary' ) ;
2017-10-15 02:22:19 -04:00
md . update ( obj . peernonce , 'binary' ) ;
2017-09-13 14:25:57 -04:00
if ( obj . unauth . nodeCert . publicKey . verify ( md . digest ( ) . bytes ( ) , msg ) == false ) { return false ; }
2017-10-18 19:28:05 -04:00
if ( obj . unauth . nodeid !== obj . agentCertificateHashBase64 ) { return false ; }
2017-09-13 14:25:57 -04:00
// Connection is a success, clean up
2017-10-15 20:36:06 -04:00
obj . nodeid = obj . unauth . nodeid ;
2017-09-13 14:25:57 -04:00
delete obj . nonce ;
2017-10-15 02:22:19 -04:00
delete obj . peernonce ;
2017-09-13 14:25:57 -04:00
delete obj . unauth ;
if ( obj . unauthsign ) delete obj . unauthsign ;
obj . authenticated = 1 ;
return true ;
}
// Process incoming peer server JSON data
function processServerData ( msg ) {
var str = msg . toString ( 'utf8' ) ;
if ( str [ 0 ] == '{' ) {
try { command = JSON . parse ( str ) } catch ( e ) { obj . parent . parent . debug ( 1 , 'Unable to parse JSON (' + obj . remoteaddr + ').' ) ; return ; } // If the command can't be parsed, ignore it.
if ( command . action == 'info' ) {
if ( obj . authenticated != 3 ) {
// We get the peer's serverid and database identifier.
if ( ( command . serverid != null ) && ( command . dbid != null ) ) {
2017-09-15 14:45:06 -04:00
if ( command . serverid == obj . parent . serverid ) { console . log ( 'ERROR: Same server ID, trying to peer with self. (' + obj . remoteaddr + ', ' + command . serverid + ').' ) ; return ; }
2017-09-13 14:25:57 -04:00
if ( command . dbid != obj . parent . parent . db . identifier ) { console . log ( 'ERROR: Database ID mismatch. Trying to peer to a server with the wrong database. (' + obj . remoteaddr + ', ' + command . serverid + ').' ) ; return ; }
if ( obj . parent . peerConfig . servers [ command . serverid ] == null ) { console . log ( 'ERROR: Unknown peer serverid: ' + command . serverid + ' (' + obj . remoteaddr + ').' ) ; return ; }
obj . peerServerId = command . serverid ;
2017-10-15 02:22:19 -04:00
obj . peerServerKey = new Buffer ( command . key , 'hex' ) ;
2017-09-17 20:22:18 -04:00
obj . serverCertHash = command . serverCertHash ;
2017-09-13 14:25:57 -04:00
obj . authenticated = 3 ;
obj . parent . SetupPeerServer ( obj , obj . peerServerId ) ;
}
}
} else if ( obj . authenticated == 3 ) {
// Pass the message to the parent object for processing.
obj . parent . ProcessPeerServerMessage ( obj , obj . peerServerId , command ) ;
}
}
}
return obj ;
}
// If we have no peering configuration, don't setup this object
if ( obj . peerConfig == null ) { return null ; }
2017-09-15 14:45:06 -04:00
obj . serverid = obj . parent . config . peers . serverId ;
if ( obj . serverid == null ) { obj . serverid = require ( "os" ) . hostname ( ) ; }
if ( obj . parent . config . peers . servers [ obj . serverid ] == null ) { console . log ( "Error: Unable to peer with other servers, \"" + obj . serverid + "\" not present in peer servers list." ) ; return null ; }
2017-08-28 12:27:45 -04:00
2017-09-15 14:45:06 -04:00
// Return the private key of a peer server
obj . getServerCookieKey = function ( serverid ) {
var server = obj . peerServers [ serverid ] ;
if ( server && server . peerServerKey ) return server . peerServerKey ;
return null ;
}
// Dispatch an event to all other MeshCentral2 peer servers
2017-08-28 12:27:45 -04:00
obj . DispatchEvent = function ( ids , source , event ) {
2017-09-13 14:25:57 -04:00
var busmsg = JSON . stringify ( { action : 'bus' , ids : ids , event : event } ) ;
for ( var serverid in obj . peerServers ) { obj . peerServers [ serverid ] . send ( busmsg ) ; }
2017-08-28 12:27:45 -04:00
}
2017-09-13 14:25:57 -04:00
// Dispatch a message to other MeshCentral2 peer servers
obj . DispatchMessage = function ( msg ) {
for ( var serverid in obj . peerServers ) { obj . peerServers [ serverid ] . send ( msg ) ; }
}
2017-08-28 12:27:45 -04:00
2017-09-15 14:45:06 -04:00
// Dispatch a message to other MeshCentral2 peer servers
obj . DispatchMessageSingleServer = function ( msg , serverid ) {
var server = obj . peerServers [ serverid ] ;
if ( server != null ) { server . send ( msg ) ; }
}
2017-09-13 14:25:57 -04:00
// Attempt to connect to all peers
obj . ConnectToPeers = function ( ) {
for ( serverId in obj . peerConfig . servers ) {
// We will only connect to names that are larger then ours. This way, eveyone has one connection to everyone else (no cross-connections).
2017-09-15 14:45:06 -04:00
if ( ( serverId > obj . serverid ) && ( obj . peerConfig . servers [ serverId ] . url != null ) && ( obj . outPeerServers [ serverId ] == null ) ) {
2017-09-13 14:25:57 -04:00
obj . outPeerServers [ serverId ] = obj . CreatePeerOutServer ( obj , serverId , obj . peerConfig . servers [ serverId ] . url ) ;
}
}
}
2017-08-28 12:27:45 -04:00
2017-09-13 14:25:57 -04:00
// We connected to a peer server, setup everything
obj . SetupPeerServer = function ( server , peerServerId ) {
console . log ( 'Connected to peer server ' + peerServerId + '.' ) ;
obj . peerServers [ peerServerId ] = server ;
2017-09-20 17:44:22 -04:00
// Send the list of connections to the peer
2017-09-13 14:25:57 -04:00
server . send ( JSON . stringify ( { action : 'connectivityTable' , connectivityTable : obj . parent . peerConnectivityByNode [ obj . parent . serverId ] } ) ) ;
2017-09-20 17:44:22 -04:00
// Send a list of user sessions to the peer
server . send ( JSON . stringify ( { action : 'sessionsTable' , sessionsTable : Object . keys ( obj . parent . webserver . wssessions2 ) } ) ) ;
2017-09-13 14:25:57 -04:00
}
2017-08-28 12:27:45 -04:00
2017-09-13 14:25:57 -04:00
// We disconnected to a peer server, clean up everything
obj . ClearPeerServer = function ( server , peerServerId ) {
console . log ( 'Disconnected from peer server ' + peerServerId + '.' ) ;
2017-09-20 17:44:22 -04:00
// Clean up the connectivity state
2017-09-13 14:25:57 -04:00
delete obj . peerServers [ peerServerId ] ;
var oldList = obj . parent . peerConnectivityByNode [ peerServerId ] ;
obj . parent . peerConnectivityByNode [ peerServerId ] = { } ;
obj . parent . UpdateConnectivityState ( oldList ) ;
2017-09-20 17:44:22 -04:00
// Clean up the sessions list
for ( var i in obj . parent . webserver . wsPeerSessions [ peerServerId ] ) { delete obj . parent . webserver . wsPeerSessions2 [ obj . parent . webserver . wsPeerSessions [ peerServerId ] [ i ] ] ; }
delete obj . parent . webserver . wsPeerSessions [ peerServerId ] ;
delete obj . parent . webserver . wsPeerSessions3 [ peerServerId ] ;
obj . parent . webserver . recountSessions ( ) ; // Recount all sessions
2017-09-13 14:25:57 -04:00
}
// Process a message coming from a peer server
obj . ProcessPeerServerMessage = function ( server , peerServerId , msg ) {
2017-09-17 20:22:18 -04:00
//console.log('ProcessPeerServerMessage', peerServerId, msg);
2017-09-13 14:25:57 -04:00
switch ( msg . action ) {
case 'bus' : {
obj . parent . DispatchEvent ( msg . ids , null , msg . event , true ) ; // Dispatch the peer event
break ;
}
case 'connectivityTable' : {
obj . parent . peerConnectivityByNode [ peerServerId ] = msg . connectivityTable ;
obj . parent . UpdateConnectivityState ( msg . connectivityTable ) ;
break ;
}
2017-09-20 17:44:22 -04:00
case 'sessionsTable' : {
obj . parent . webserver . wsPeerSessions [ peerServerId ] = msg . sessionsTable ;
var userToSession = { } ;
for ( var i in msg . sessionsTable ) {
var sessionid = msg . sessionsTable [ i ] ;
obj . parent . webserver . wsPeerSessions2 [ sessionid ] = peerServerId ;
var userid = sessionid . split ( '/' ) . slice ( 0 , 3 ) . join ( '/' ) ; // Take the sessionid and keep only the userid partion
if ( userToSession [ userid ] == null ) { userToSession [ userid ] = [ sessionid ] ; } else { userToSession [ userid ] . push ( sessionid ) ; } // UserId -> [ SessionId ]
}
obj . parent . webserver . wsPeerSessions3 [ peerServerId ] = userToSession ; // ServerId --> UserId --> SessionId
obj . parent . webserver . recountSessions ( ) ; // Recount all sessions
break ;
}
case 'sessionStart' : {
obj . parent . webserver . wsPeerSessions [ peerServerId ] . push ( msg . sessionid ) ;
obj . parent . webserver . wsPeerSessions2 [ msg . sessionid ] = peerServerId ;
var userid = msg . sessionid . split ( '/' ) . slice ( 0 , 3 ) . join ( '/' ) ;
if ( obj . parent . webserver . wsPeerSessions3 [ peerServerId ] == null ) { obj . parent . webserver . wsPeerSessions3 [ peerServerId ] = { } ; }
if ( obj . parent . webserver . wsPeerSessions3 [ peerServerId ] [ userid ] == null ) { obj . parent . webserver . wsPeerSessions3 [ peerServerId ] [ userid ] = [ msg . sessionid ] ; } else { obj . parent . webserver . wsPeerSessions3 [ peerServerId ] [ userid ] . push ( msg . sessionid ) ; }
obj . parent . webserver . recountSessions ( msg . sessionid ) ; // Recount a specific user
break ;
}
case 'sessionEnd' : {
var i = obj . parent . webserver . wsPeerSessions [ peerServerId ] . indexOf ( msg . sessionid ) ;
if ( i >= 0 ) { obj . parent . webserver . wsPeerSessions [ peerServerId ] . splice ( i , 1 ) ; }
delete obj . parent . webserver . wsPeerSessions2 [ msg . sessionid ] ;
var userid = msg . sessionid . split ( '/' ) . slice ( 0 , 3 ) . join ( '/' ) ;
if ( obj . parent . webserver . wsPeerSessions3 [ peerServerId ] [ userid ] != null ) {
i = obj . parent . webserver . wsPeerSessions3 [ peerServerId ] [ userid ] . indexOf ( msg . sessionid ) ;
if ( i >= 0 ) {
obj . parent . webserver . wsPeerSessions3 [ peerServerId ] [ userid ] . splice ( i , 1 ) ;
if ( obj . parent . webserver . wsPeerSessions3 [ peerServerId ] [ userid ] . length == 0 ) { delete obj . parent . webserver . wsPeerSessions3 [ peerServerId ] [ userid ] ; }
}
}
obj . parent . webserver . recountSessions ( msg . sessionid ) ; // Recount a specific user
break ;
}
2017-09-13 14:25:57 -04:00
case 'SetConnectivityState' : {
obj . parent . SetConnectivityState ( msg . meshid , msg . nodeid , msg . connectTime , msg . connectType , msg . powerState , peerServerId ) ;
break ;
}
case 'ClearConnectivityState' : {
obj . parent . ClearConnectivityState ( msg . meshid , msg . nodeid , msg . connectType , peerServerId ) ;
break ;
}
2017-09-17 20:22:18 -04:00
case 'relay' : {
// Check if there is a waiting session
var rsession = obj . parent . webserver . wsrelays [ msg . id ] ;
if ( rsession != null ) {
// Yes, there is a waiting session, see if we must initiate.
if ( peerServerId > obj . parent . serverId ) {
// We must initiate the connection to the peer
var userid = null ;
if ( rsession . peer1 . req . session != null ) { userid = rsession . peer1 . req . session . userid ; }
obj . createPeerRelay ( rsession . peer1 . ws , rsession . peer1 . req , peerServerId , userid ) ;
delete obj . parent . webserver . wsrelays [ msg . id ] ;
}
} else {
// Add this relay session to the peer relay list
obj . parent . webserver . wsPeerRelays [ msg . id ] = { serverId : peerServerId , time : Date . now ( ) } ;
// Clear all relay sessions that are more than 1 minute
var oneMinuteAgo = Date . now ( ) - 60000 ;
for ( var id in obj . parent . webserver . wsPeerRelays ) { if ( obj . parent . webserver . wsPeerRelays [ id ] . time < oneMinuteAgo ) { delete obj . parent . webserver . wsPeerRelays [ id ] ; } }
}
break ;
}
2017-09-15 14:45:06 -04:00
case 'msg' : {
if ( msg . sessionid != null ) {
// Route this message to a connected user session
if ( command . fromNodeid != null ) { command . nodeid = command . fromNodeid ; delete command . fromNodeid ; }
var ws = obj . parent . webserver . wssessions2 [ command . sessionid ] ;
if ( ws != null ) { ws . send ( JSON . stringify ( command ) ) ; }
} else if ( msg . nodeid != null ) {
// Route this message to a connected agent
if ( command . fromSessionid != null ) { command . sessionid = command . fromSessionid ; delete command . fromSessionid ; }
var agent = obj . parent . webserver . wsagents [ msg . nodeid ] ;
if ( agent != null ) { delete msg . nodeid ; agent . send ( JSON . stringify ( msg ) ) ; } // Remove the nodeid since it's implyed and send the message to the agent
} else if ( msg . meshid != null ) {
// Route this message to all users of this mesh
if ( command . fromNodeid != null ) { command . nodeid = command . fromNodeid ; delete command . fromNodeid ; }
var cmdstr = JSON . stringify ( command ) ;
for ( var userid in obj . parent . webserver . wssessions ) { // Find all connected users for this mesh and send the message
var user = obj . parent . webserver . users [ userid ] ;
if ( user ) {
var rights = user . links [ msg . meshid ] ;
if ( rights != null ) { // TODO: Look at what rights are needed for message routing
var sessions = obj . parent . webserver . wssessions [ userid ] ;
// Send the message to all users on this server
for ( var i in sessions ) { sessions [ i ] . send ( cmdstr ) ; }
}
}
}
}
break ;
}
}
}
// Create a tunnel connection to a peer server
obj . createPeerRelay = function ( ws , req , serverid , user ) {
var server = obj . peerServers [ serverid ] ;
if ( ( server == null ) || ( server . peerServerKey == null ) ) { return null ; }
var cookieKey = server . peerServerKey ;
2017-09-17 20:22:18 -04:00
// Parse the user if needed
if ( typeof user == 'string' ) { user = { _id : user , domain : user . split ( '/' ) [ 1 ] } ; }
2017-09-15 14:45:06 -04:00
// Build the connection URL
var path = req . path ;
if ( path [ 0 ] == '/' ) path = path . substring ( 1 ) ;
if ( path . substring ( path . length - 11 ) == '/.websocket' ) { path = path . substring ( 0 , path . length - 11 ) ; }
var queryStr = ''
for ( var i in req . query ) { queryStr += ( ( queryStr == '' ) ? '?' : '&' ) + i + '=' + req . query [ i ] ; }
2017-09-17 20:22:18 -04:00
if ( user != null ) { queryStr += ( ( queryStr == '' ) ? '?' : '&' ) + 'auth=' + obj . encodeCookie ( { userid : user . _id , domainid : user . domain } , cookieKey ) ; }
2017-09-15 14:45:06 -04:00
var url = obj . peerConfig . servers [ serverid ] . url + path + queryStr ;
// Setup an connect the web socket
var tunnel = obj . createPeerRelayEx ( ws , url , serverid ) ;
tunnel . connect ( ) ;
}
// Create a tunnel connection to a peer server
// We assume that "ws" is paused already.
obj . createPeerRelayEx = function ( ws , url , serverid ) {
var peerTunnel = { parent : obj , ws1 : ws , ws2 : null , url : url , serverid : serverid } ;
peerTunnel . connect = function ( ) {
// Get the web socket setup
2017-10-15 02:22:19 -04:00
peerTunnel . parent . parent . debug ( 1 , 'FTunnel ' + peerTunnel . serverid + ': Start connect to ' + peerTunnel . url ) ;
peerTunnel . ws2 = new WebSocket ( peerTunnel . url , { rejectUnauthorized : false } ) ;
2017-09-15 14:45:06 -04:00
// Register the connection failed event
2017-10-15 02:22:19 -04:00
peerTunnel . ws2 . on ( 'error' , function ( error ) { peerTunnel . parent . parent . debug ( 1 , 'FTunnel ' + obj . serverid + ': Connection error' ) ; peerTunnel . close ( ) ; } ) ;
2017-09-17 20:22:18 -04:00
2017-10-15 02:22:19 -04:00
// If the peer server web socket is closed, clean up.
2017-10-15 20:36:06 -04:00
peerTunnel . ws2 . on ( 'close' , function ( req ) { peerTunnel . parent . parent . debug ( 1 , 'FTunnel disconnect ' + peerTunnel . serverid ) ; peerTunnel . close ( ) ; } ) ;
2017-09-17 20:22:18 -04:00
2017-10-15 02:22:19 -04:00
// If a message is received from the peer, Peer ---> Browser (TODO: Pipe this?)
peerTunnel . ws2 . on ( 'message' , function ( msg ) { try { peerTunnel . ws2 . pause ( ) ; peerTunnel . ws1 . send ( msg , function ( ) { peerTunnel . ws2 . resume ( ) ; } ) ; } catch ( e ) { } } ) ;
2017-09-15 14:45:06 -04:00
2017-10-15 02:22:19 -04:00
// Register the connection event
peerTunnel . ws2 . on ( 'open' , function ( ) {
peerTunnel . parent . parent . debug ( 1 , 'FTunnel ' + peerTunnel . serverid + ': Connected' ) ;
2017-09-15 14:45:06 -04:00
2017-10-15 02:22:19 -04:00
// Get the peer server's certificate and compute the server public key hash
var serverCert = obj . forge . pki . certificateFromAsn1 ( obj . forge . asn1 . fromDer ( peerTunnel . ws2 . _socket . getPeerCertificate ( ) . raw . toString ( 'binary' ) ) ) ;
2017-10-15 20:36:06 -04:00
var serverCertHashHex = new Buffer ( obj . forge . pki . getPublicKeyFingerprint ( serverCert . publicKey , { encoding : 'binary' , md : obj . forge . md . sha384 . create ( ) } ) , 'binary' ) . toString ( 'base64' ) . replace ( /\+/g , '@' ) . replace ( /\//g , '$' ) ;
2017-09-15 14:45:06 -04:00
2017-10-15 02:22:19 -04:00
// Check if the peer certificate is the expected one for this serverid
2017-10-15 20:36:06 -04:00
if ( obj . peerServers [ serverid ] == null || obj . peerServers [ serverid ] . serverCertHash != serverCertHashHex ) { console . log ( 'ERROR: Outer certificate hash mismatch (1). (' + peerTunnel . url + ', ' + peerTunnel . serverid + ').' ) ; peerTunnel . close ( ) ; return ; }
2017-09-15 14:45:06 -04:00
2017-10-15 02:22:19 -04:00
// Connection accepted, resume the web socket to start the data flow
2017-09-15 14:45:06 -04:00
peerTunnel . ws1 . resume ( ) ;
} ) ;
// If a message is received from the browser, Browser ---> Peer
2017-09-17 20:22:18 -04:00
peerTunnel . ws1 . on ( 'message' , function ( msg ) { try { peerTunnel . ws1 . pause ( ) ; peerTunnel . ws2 . send ( msg , function ( ) { peerTunnel . ws1 . resume ( ) ; } ) ; } catch ( e ) { } } ) ;
2017-09-15 14:45:06 -04:00
// If error, do nothing
2017-10-15 02:22:19 -04:00
peerTunnel . ws1 . on ( 'error' , function ( err ) { peerTunnel . close ( ) ; } ) ;
2017-09-15 14:45:06 -04:00
// If the web socket is closed, close the associated TCP connection.
2017-10-15 20:36:06 -04:00
peerTunnel . ws1 . on ( 'close' , function ( req ) { peerTunnel . parent . parent . debug ( 1 , 'FTunnel disconnect ' + peerTunnel . serverid ) ; peerTunnel . close ( ) ; } ) ;
2017-09-15 14:45:06 -04:00
}
// Disconnect both sides of the tunnel
peerTunnel . close = function ( arg ) {
if ( arg == 2 ) {
// Hard close, close the TCP socket
2017-09-17 20:22:18 -04:00
if ( peerTunnel . ws1 != null ) { try { peerTunnel . ws1 . _socket . _parent . end ( ) ; peerTunnel . parent . parent . debug ( 1 , 'FTunnel1: Hard disconnect' ) ; } catch ( e ) { console . log ( e ) ; } }
if ( peerTunnel . ws2 != null ) { try { peerTunnel . ws2 . _socket . _parent . end ( ) ; peerTunnel . parent . parent . debug ( 1 , 'FTunnel2: Hard disconnect' ) ; } catch ( e ) { console . log ( e ) ; } }
2017-09-15 14:45:06 -04:00
} else {
// Soft close, close the websocket
2017-09-17 20:22:18 -04:00
if ( peerTunnel . ws1 != null ) { try { peerTunnel . ws1 . close ( ) ; peerTunnel . parent . parent . debug ( 1 , 'FTunnel1: Soft disconnect ' ) ; } catch ( e ) { console . log ( e ) ; } }
if ( peerTunnel . ws2 != null ) { try { peerTunnel . ws2 . close ( ) ; peerTunnel . parent . parent . debug ( 1 , 'FTunnel2: Soft disconnect ' ) ; } catch ( e ) { console . log ( e ) ; } }
2017-09-15 14:45:06 -04:00
}
2017-09-13 14:25:57 -04:00
}
2017-09-15 14:45:06 -04:00
return peerTunnel ;
2017-08-28 12:27:45 -04:00
}
2017-09-13 14:25:57 -04:00
setTimeout ( function ( ) { obj . ConnectToPeers ( ) ; } , 1000 ) ; // Delay this a little to make sure we are ready on our side.
2017-08-28 12:27:45 -04:00
return obj ;
}