2017-08-28 12:27:45 -04:00
/ * *
2018-01-04 15:15:21 -05:00
* @ description MeshCentral MeshAgent communication module
2017-08-28 12:27:45 -04:00
* @ author Ylian Saint - Hilaire & Bryan Roe
2020-01-02 21:30:12 -05:00
* @ copyright Intel Corporation 2018 - 2020
2018-01-04 15:15:21 -05:00
* @ license Apache - 2.0
2017-08-28 12:27:45 -04:00
* @ version v0 . 0.1
* /
2018-08-29 20:40:30 -04:00
/*xjslint node: true */
/*xjslint plusplus: true */
/*xjslint maxlen: 256 */
/*jshint node: true */
/*jshint strict: false */
/*jshint esversion: 6 */
"use strict" ;
2018-08-27 15:24:15 -04:00
2017-08-28 12:27:45 -04:00
// Construct a MeshAgent object, called upon connection
module . exports . CreateMeshAgent = function ( parent , db , ws , req , args , domain ) {
2019-03-08 01:47:27 -05:00
const forge = parent . parent . certificateOperations . forge ;
const common = parent . parent . common ;
2019-05-01 18:02:03 -04:00
parent . agentStats . createMeshAgentCount ++ ;
2020-05-22 00:25:11 -04:00
parent . parent . debug ( 'agent' , 'New agent at ' + req . clientIp + ':' + ws . _socket . remotePort ) ;
2019-03-08 01:47:27 -05:00
2017-08-28 12:27:45 -04:00
var obj = { } ;
obj . domain = domain ;
2019-03-08 01:47:27 -05:00
obj . authenticated = 0 ;
2017-08-28 12:27:45 -04:00
obj . receivedCommands = 0 ;
obj . agentCoreCheck = 0 ;
2020-05-22 00:25:11 -04:00
obj . remoteaddr = req . clientIp ;
2019-03-08 01:47:27 -05:00
obj . remoteaddrport = obj . remoteaddr + ':' + ws . _socket . remotePort ;
obj . nonce = parent . crypto . randomBytes ( 48 ) . toString ( 'binary' ) ;
2019-06-17 18:20:47 -04:00
//ws._socket.setKeepAlive(true, 240000); // Set TCP keep alive, 4 minutes
if ( args . agentidletimeout != 0 ) { ws . _socket . setTimeout ( args . agentidletimeout , function ( ) { obj . close ( 1 ) ; } ) ; } // Inactivity timeout of 2:30 minutes, by default agent will WebSocket ping every 2 minutes and server will pong back.
2019-03-08 01:47:27 -05:00
//obj.nodeid = null;
//obj.meshid = null;
//obj.dbNodeKey = null;
//obj.dbMeshKey = null;
//obj.connectTime = null;
//obj.agentInfo = null;
2017-08-28 12:27:45 -04:00
// Send a message to the mesh agent
2019-08-09 18:13:53 -04:00
obj . send = function ( data , func ) { try { if ( typeof data == 'string' ) { ws . send ( Buffer . from ( data ) , func ) ; } else { ws . send ( data , func ) ; } } catch ( e ) { } } ;
obj . sendBinary = function ( data , func ) { try { if ( typeof data == 'string' ) { ws . send ( Buffer . from ( data , 'binary' ) , func ) ; } else { ws . send ( data , func ) ; } } catch ( e ) { } } ;
2017-08-28 12:27:45 -04:00
// Disconnect this agent
obj . close = function ( arg ) {
2019-08-22 18:31:39 -04:00
if ( ( arg == 1 ) || ( arg == null ) ) { try { ws . close ( ) ; if ( obj . nodeid != null ) { parent . parent . debug ( 'agent' , 'Soft disconnect ' + obj . nodeid + ' (' + obj . remoteaddrport + ')' ) ; } } catch ( e ) { console . log ( e ) ; } } // Soft close, close the websocket
if ( arg == 2 ) { try { ws . _socket . _parent . end ( ) ; if ( obj . nodeid != null ) { parent . parent . debug ( 'agent' , 'Hard disconnect ' + obj . nodeid + ' (' + obj . remoteaddrport + ')' ) ; } } catch ( e ) { console . log ( e ) ; } } // Hard close, close the TCP socket
2019-02-25 20:30:19 -05:00
// If arg == 3, don't communicate with this agent anymore, but don't disconnect (Duplicate agent).
2019-03-09 17:28:08 -05:00
2019-04-10 13:41:10 -04:00
// Remove this agent from the webserver list
if ( parent . wsagents [ obj . dbNodeKey ] == obj ) {
delete parent . wsagents [ obj . dbNodeKey ] ;
parent . parent . ClearConnectivityState ( obj . dbMeshKey , obj . dbNodeKey , 1 ) ;
}
2018-09-26 17:58:55 -04:00
2019-10-29 19:17:29 -04:00
// Remove this agent from the list of agents with bad web certificates
if ( obj . badWebCert ) { delete parent . wsagentsWithBadWebCerts [ obj . badWebCert ] ; }
2019-04-10 13:41:10 -04:00
// Get the current mesh
const mesh = parent . meshes [ obj . dbMeshKey ] ;
// If this is a temporary or recovery agent, or all devices in this group are temporary, remove the agent (0x20 = Temporary, 0x40 = Recovery)
if ( ( ( obj . agentInfo ) && ( obj . agentInfo . capabilities ) && ( ( obj . agentInfo . capabilities & 0x20 ) || ( obj . agentInfo . capabilities & 0x40 ) ) ) || ( ( mesh ) && ( mesh . flags ) && ( mesh . flags & 1 ) ) ) {
// Delete this node including network interface information and events
2020-09-09 17:26:03 -04:00
db . Remove ( obj . dbNodeKey ) ; // Remove node with that id
db . Remove ( 'if' + obj . dbNodeKey ) ; // Remove interface information
db . Remove ( 'nt' + obj . dbNodeKey ) ; // Remove notes
db . Remove ( 'lc' + obj . dbNodeKey ) ; // Remove last connect time
db . Remove ( 'si' + obj . dbNodeKey ) ; // Remove system information
if ( db . RemoveSMBIOS ) { db . RemoveSMBIOS ( obj . dbNodeKey ) ; } // Remove SMBios data
db . RemoveAllNodeEvents ( obj . dbNodeKey ) ; // Remove all events for this node
db . removeAllPowerEventsForNode ( obj . dbNodeKey ) ; // Remove all power events for this node
2019-04-10 13:41:10 -04:00
// Event node deletion
2020-03-17 20:17:04 -04:00
parent . parent . DispatchEvent ( parent . CreateMeshDispatchTargets ( obj . dbMeshKey , [ obj . dbNodeKey ] ) , obj , { etype : 'node' , action : 'removenode' , nodeid : obj . dbNodeKey , domain : domain . id , nolog : 1 } ) ;
2019-04-10 13:41:10 -04:00
// Disconnect all connections if needed
const state = parent . parent . GetConnectivityState ( obj . dbNodeKey ) ;
if ( ( state != null ) && ( state . connectivity != null ) ) {
if ( ( state . connectivity & 1 ) != 0 ) { parent . wsagents [ obj . dbNodeKey ] . close ( ) ; } // Disconnect mesh agent
2020-10-09 18:44:09 -04:00
if ( ( state . connectivity & 2 ) != 0 ) { parent . parent . mpsserver . closeAllForNode ( obj . dbNodeKey ) ; } // Disconnect CIRA connection
2018-04-12 14:15:01 -04:00
}
2019-04-10 13:41:10 -04:00
} else {
// Update the last connect time
2020-05-02 01:30:06 -04:00
if ( obj . authenticated == 2 ) { db . Set ( { _id : 'lc' + obj . dbNodeKey , type : 'lastconnect' , domain : domain . id , time : Date . now ( ) , addr : obj . remoteaddrport , cause : 1 } ) ; }
2018-04-12 14:15:01 -04:00
}
2019-03-09 17:28:08 -05:00
2019-10-06 00:48:17 -04:00
// Set this agent as no longer authenticated
obj . authenticated = - 1 ;
2019-03-09 17:28:08 -05:00
// If we where updating the agent, clean that up.
if ( obj . agentUpdate != null ) {
if ( obj . agentUpdate . fd ) { try { parent . fs . close ( obj . agentUpdate . fd ) ; } catch ( ex ) { } }
parent . parent . taskLimiter . completed ( obj . agentUpdate . taskid ) ; // Indicate this task complete
2019-03-10 14:00:14 -04:00
delete obj . agentUpdate . buf ;
2019-03-09 17:28:08 -05:00
delete obj . agentUpdate ;
}
2019-05-28 13:58:46 -04:00
// Perform timer cleanup
if ( obj . pingtimer ) { clearInterval ( obj . pingtimer ) ; delete obj . pingtimer ; }
if ( obj . pongtimer ) { clearInterval ( obj . pongtimer ) ; delete obj . pongtimer ; }
2019-03-09 17:28:08 -05:00
// Perform aggressive cleanup
if ( obj . nonce ) { delete obj . nonce ; }
if ( obj . nodeid ) { delete obj . nodeid ; }
if ( obj . unauth ) { delete obj . unauth ; }
if ( obj . remoteaddr ) { delete obj . remoteaddr ; }
if ( obj . remoteaddrport ) { delete obj . remoteaddrport ; }
if ( obj . meshid ) { delete obj . meshid ; }
if ( obj . dbNodeKey ) { delete obj . dbNodeKey ; }
if ( obj . dbMeshKey ) { delete obj . dbMeshKey ; }
if ( obj . connectTime ) { delete obj . connectTime ; }
if ( obj . agentInfo ) { delete obj . agentInfo ; }
2019-03-09 18:04:43 -05:00
if ( obj . agentExeInfo ) { delete obj . agentExeInfo ; }
2019-10-28 19:52:12 -04:00
ws . removeAllListeners ( [ 'message' , 'close' , 'error' ] ) ;
2018-08-29 20:40:30 -04:00
} ;
2017-08-28 12:27:45 -04:00
// When data is received from the mesh agent web socket
ws . on ( 'message' , function ( msg ) {
if ( msg . length < 2 ) return ;
2017-10-15 02:22:19 -04:00
if ( typeof msg == 'object' ) { msg = msg . toString ( 'binary' ) ; } // TODO: Could change this entire method to use Buffer instead of binary string
2017-08-28 12:27:45 -04:00
if ( obj . authenticated == 2 ) { // We are authenticated
2018-07-02 17:34:10 -04:00
if ( ( obj . agentUpdate == null ) && ( msg . charCodeAt ( 0 ) == 123 ) ) { processAgentData ( msg ) ; } // Only process JSON messages if meshagent update is not in progress
2017-08-28 12:27:45 -04:00
if ( msg . length < 2 ) return ;
2019-03-08 01:47:27 -05:00
const cmdid = common . ReadShort ( msg , 0 ) ;
2017-08-28 12:27:45 -04:00
if ( cmdid == 11 ) { // MeshCommand_CoreModuleHash
2019-10-28 19:52:12 -04:00
if ( msg . length == 4 ) { ChangeAgentCoreInfo ( { 'caps' : 0 } ) ; } // If the agent indicated that no core is running, clear the core information string.
2017-08-28 12:27:45 -04:00
// Mesh core hash, sent by agent with the hash of the current mesh core.
2019-02-18 20:50:10 -05:00
2019-10-28 13:35:58 -04:00
// If we are performing an agent update, don't update the core.
if ( obj . agentUpdate != null ) { return ; }
2019-02-18 20:50:10 -05:00
// If we are using a custom core, don't try to update it.
if ( obj . agentCoreCheck == 1000 ) {
2019-08-09 18:13:53 -04:00
obj . sendBinary ( common . ShortToStr ( 16 ) + common . ShortToStr ( 0 ) ) ; // MeshCommand_CoreOk. Indicates to the agent that the core is ok. Start it if it's not already started.
2019-02-18 20:50:10 -05:00
agentCoreIsStable ( ) ;
return ;
}
2018-12-29 18:24:33 -05:00
// Get the current meshcore hash
const agentMeshCoreHash = ( msg . length == 52 ) ? msg . substring ( 4 , 52 ) : null ;
2019-03-08 01:47:27 -05:00
// If the agent indicates this is a custom core, we are done.
2019-03-09 17:28:08 -05:00
if ( ( agentMeshCoreHash != null ) && ( agentMeshCoreHash == '\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0' ) ) {
2019-02-18 20:50:10 -05:00
obj . agentCoreCheck = 0 ;
2019-08-09 18:13:53 -04:00
obj . sendBinary ( common . ShortToStr ( 16 ) + common . ShortToStr ( 0 ) ) ; // MeshCommand_CoreOk. Indicates to the agent that the core is ok. Start it if it's not already started.
2019-02-18 20:50:10 -05:00
agentCoreIsStable ( ) ;
return ;
}
2019-03-08 01:47:27 -05:00
// We need to check if the core is current. Figure out what core we need.
2019-06-17 18:37:13 -04:00
var corename = null ;
if ( parent . parent . meshAgentsArchitectureNumbers [ obj . agentInfo . agentId ] != null ) {
if ( obj . agentCoreCheck == 1001 ) {
// If the user asked, use the recovery core.
corename = parent . parent . meshAgentsArchitectureNumbers [ obj . agentInfo . agentId ] . rcore ;
} else if ( obj . agentInfo . capabilities & 0x40 ) {
// If this is a recovery agent, use the agent recovery core.
corename = parent . parent . meshAgentsArchitectureNumbers [ obj . agentInfo . agentId ] . arcore ;
} else {
// This is the normal core for this agent type.
corename = parent . parent . meshAgentsArchitectureNumbers [ obj . agentInfo . agentId ] . core ;
}
2019-03-08 01:47:27 -05:00
}
2019-02-25 17:35:08 -05:00
// If we have a core, use it.
2018-12-29 18:24:33 -05:00
if ( corename != null ) {
2019-03-08 01:47:27 -05:00
const meshcorehash = parent . parent . defaultMeshCoresHash [ corename ] ;
2018-12-29 18:24:33 -05:00
if ( agentMeshCoreHash != meshcorehash ) {
2019-01-21 17:05:50 -05:00
if ( ( obj . agentCoreCheck < 5 ) || ( obj . agentCoreCheck == 1001 ) ) {
2018-12-29 18:24:33 -05:00
if ( meshcorehash == null ) {
2019-01-17 18:07:34 -05:00
// Clear the core
2019-08-09 18:13:53 -04:00
obj . sendBinary ( common . ShortToStr ( 10 ) + common . ShortToStr ( 0 ) ) ; // MeshCommand_CoreModule, ask mesh agent to clear the core
2019-05-01 18:02:03 -04:00
parent . agentStats . clearingCoreCount ++ ;
2019-10-28 19:52:12 -04:00
parent . parent . debug ( 'agent' , "Clearing core" ) ;
2018-12-29 18:24:33 -05:00
} else {
2020-07-17 14:14:24 -04:00
// Setup task limiter options, this system limits how many tasks can run at the same time to spread the server load.
var taskLimiterOptions = { hash : meshcorehash , core : parent . parent . defaultMeshCores [ corename ] , name : corename } ;
// If the agent supports compression, sent the core compressed.
2020-07-28 21:45:17 -04:00
if ( ( obj . agentInfo . capabilities & 0x100 ) && ( parent . parent . defaultMeshCoresDeflate [ corename ] ) ) {
2020-07-21 19:20:17 -04:00
args . core = parent . parent . defaultMeshCoresDeflate [ corename ] ;
}
2019-01-28 18:47:54 -05:00
2019-02-08 17:17:35 -05:00
// Update new core with task limiting so not to flood the server. This is a high priority task.
2019-02-25 19:07:27 -05:00
obj . agentCoreUpdatePending = true ;
2019-03-08 01:47:27 -05:00
parent . parent . taskLimiter . launch ( function ( argument , taskid , taskLimiterQueue ) {
2019-02-25 20:30:19 -05:00
if ( obj . authenticated == 2 ) {
// Send the updated code.
2019-03-08 01:47:27 -05:00
delete obj . agentCoreUpdatePending ;
2019-08-09 18:13:53 -04:00
obj . sendBinary ( common . ShortToStr ( 10 ) + common . ShortToStr ( 0 ) + argument . hash + argument . core , function ( ) { parent . parent . taskLimiter . completed ( taskid ) ; } ) ; // MeshCommand_CoreModule, start core update
2019-05-01 18:02:03 -04:00
parent . agentStats . updatingCoreCount ++ ;
2019-10-28 19:52:12 -04:00
parent . parent . debug ( 'agent' , "Updating core " + argument . name ) ;
2019-02-25 20:30:19 -05:00
} else {
// This agent is probably disconnected, nothing to do.
2019-03-08 01:47:27 -05:00
parent . parent . taskLimiter . completed ( taskid ) ;
2019-02-25 20:30:19 -05:00
}
2020-07-17 14:14:24 -04:00
} , taskLimiterOptions , 0 ) ;
2018-12-29 18:24:33 -05:00
}
obj . agentCoreCheck ++ ;
}
} else {
obj . agentCoreCheck = 0 ;
2019-08-09 18:13:53 -04:00
obj . sendBinary ( common . ShortToStr ( 16 ) + common . ShortToStr ( 0 ) ) ; // MeshCommand_CoreOk. Indicates to the agent that the core is ok. Start it if it's not already started.
2019-02-07 18:00:10 -05:00
agentCoreIsStable ( ) ; // No updates needed, agent is ready to go.
2018-12-29 18:24:33 -05:00
}
}
/ *
2017-08-28 12:27:45 -04:00
// TODO: Check if we have a mesh specific core. If so, use that.
var agentMeshCoreHash = null ;
2017-10-15 02:22:19 -04:00
if ( msg . length == 52 ) { agentMeshCoreHash = msg . substring ( 4 , 52 ) ; }
2019-03-08 01:47:27 -05:00
if ( ( agentMeshCoreHash != parent . parent . defaultMeshCoreHash ) && ( agentMeshCoreHash != parent . parent . defaultMeshCoreNoMeiHash ) ) {
2017-08-28 12:27:45 -04:00
if ( obj . agentCoreCheck < 5 ) { // This check is in place to avoid a looping core update.
2019-03-08 01:47:27 -05:00
if ( parent . parent . defaultMeshCoreHash == null ) {
2017-08-28 12:27:45 -04:00
// Update no core
2019-08-09 18:13:53 -04:00
obj . sendBinary ( common . ShortToStr ( 10 ) + common . ShortToStr ( 0 ) ) ; // Command 10, ask mesh agent to clear the core
2017-08-28 12:27:45 -04:00
} else {
// Update new core
2020-02-17 16:01:13 -05:00
if ( ( parent . parent . meshAgentsArchitectureNumbers [ obj . agentInfo . agentId ] != null ) && ( parent . parent . meshAgentsArchitectureNumbers [ obj . agentInfo . agentId ] . amt == true ) ) {
2019-08-09 18:13:53 -04:00
obj . sendBinary ( common . ShortToStr ( 10 ) + common . ShortToStr ( 0 ) + parent . parent . defaultMeshCoreHash + parent . parent . defaultMeshCore ) ; // Command 10, ask mesh agent to set the core (with MEI support)
2017-12-12 19:04:54 -05:00
} else {
2019-08-09 18:13:53 -04:00
obj . sendBinary ( common . ShortToStr ( 10 ) + common . ShortToStr ( 0 ) + parent . parent . defaultMeshCoreNoMeiHash + parent . parent . defaultMeshCoreNoMei ) ; // Command 10, ask mesh agent to set the core (No MEI)
2017-12-12 19:04:54 -05:00
}
2017-08-28 12:27:45 -04:00
}
obj . agentCoreCheck ++ ;
}
} else {
obj . agentCoreCheck = 0 ;
}
2018-12-29 18:24:33 -05:00
* /
2017-08-28 12:27:45 -04:00
}
else if ( cmdid == 12 ) { // MeshCommand_AgentHash
2017-10-15 02:22:19 -04:00
if ( ( msg . length == 52 ) && ( obj . agentExeInfo != null ) && ( obj . agentExeInfo . update == true ) ) {
2019-03-08 01:47:27 -05:00
const agenthash = msg . substring ( 4 ) ;
2020-12-08 01:55:29 -05:00
if ( compareAgentBinaryHash ( obj . agentExeInfo , agenthash ) ) {
2019-02-08 17:17:35 -05:00
// Mesh agent update required, do it using task limiter so not to flood the network. Medium priority task.
2019-03-08 01:47:27 -05:00
parent . parent . taskLimiter . launch ( function ( argument , taskid , taskLimiterQueue ) {
if ( obj . authenticated != 2 ) { parent . parent . taskLimiter . completed ( taskid ) ; return ; } // If agent disconnection, complete and exit now.
2019-10-28 19:52:12 -04:00
if ( obj . nodeid != null ) { parent . parent . debug ( 'agent' , "Agent update required, NodeID=0x" + obj . nodeid . substring ( 0 , 16 ) + ', ' + obj . agentExeInfo . desc ) ; }
2019-05-24 18:11:00 -04:00
parent . agentStats . agentBinaryUpdate ++ ;
2020-07-28 21:45:17 -04:00
if ( ( obj . agentExeInfo . data == null ) && ( ( ( obj . agentInfo . capabilities & 0x100 ) == 0 ) || ( obj . agentExeInfo . zdata == null ) ) ) {
2019-02-15 01:16:15 -05:00
// Read the agent from disk
2019-03-08 01:47:27 -05:00
parent . fs . open ( obj . agentExeInfo . path , 'r' , function ( err , fd ) {
2019-03-09 18:04:43 -05:00
if ( obj . agentExeInfo == null ) return ; // Agent disconnected during this call.
2019-10-28 13:35:58 -04:00
if ( err ) { parent . parent . debug ( 'agentupdate' , "ERROR: " + err ) ; return console . error ( err ) ; }
2019-10-28 19:52:12 -04:00
obj . agentUpdate = { ptr : 0 , buf : Buffer . alloc ( parent . parent . agentUpdateBlockSize + 4 ) , fd : fd , taskid : taskid } ;
2019-02-15 01:16:15 -05:00
// MeshCommand_CoreModule, ask mesh agent to clear the core.
// The new core will only be sent after the agent updates.
2019-08-09 18:13:53 -04:00
obj . sendBinary ( common . ShortToStr ( 10 ) + common . ShortToStr ( 0 ) ) ;
2019-02-15 01:16:15 -05:00
2020-07-17 17:29:02 -04:00
// We got the agent file open on the server side, tell the agent we are sending an update ending with the SHA384 hash of the result
2019-02-15 01:16:15 -05:00
//console.log("Agent update file open.");
2019-08-09 18:13:53 -04:00
obj . sendBinary ( common . ShortToStr ( 13 ) + common . ShortToStr ( 0 ) ) ; // Command 13, start mesh agent download
2019-02-15 01:16:15 -05:00
// Send the first mesh agent update data block
obj . agentUpdate . buf [ 0 ] = 0 ;
obj . agentUpdate . buf [ 1 ] = 14 ;
obj . agentUpdate . buf [ 2 ] = 0 ;
obj . agentUpdate . buf [ 3 ] = 1 ;
2019-10-28 19:52:12 -04:00
parent . fs . read ( obj . agentUpdate . fd , obj . agentUpdate . buf , 4 , parent . parent . agentUpdateBlockSize , obj . agentUpdate . ptr , function ( err , bytesRead , buffer ) {
2020-12-07 01:24:51 -05:00
if ( obj . agentUpdate == null ) return ;
2019-03-08 01:47:27 -05:00
if ( ( err != null ) || ( bytesRead == 0 ) ) {
// Error reading the agent file, stop here.
try { parent . fs . close ( obj . agentUpdate . fd ) ; } catch ( ex ) { }
parent . parent . taskLimiter . completed ( obj . agentUpdate . taskid ) ; // Indicate this task complete
2019-10-28 13:35:58 -04:00
parent . parent . debug ( 'agentupdate' , "ERROR: Unable to read first block of agent binary from disk." ) ;
2019-03-10 14:00:14 -04:00
delete obj . agentUpdate . buf ;
2019-03-08 01:47:27 -05:00
delete obj . agentUpdate ;
} else {
// Send the first block to the agent
obj . agentUpdate . ptr += bytesRead ;
2019-10-28 13:35:58 -04:00
parent . parent . debug ( 'agentupdate' , "Sent first block of " + bytesRead + " bytes from disk." ) ;
2019-08-09 18:13:53 -04:00
obj . sendBinary ( obj . agentUpdate . buf ) ; // Command 14, mesh agent first data block
2019-03-08 01:47:27 -05:00
}
} ) ;
2019-02-15 01:16:15 -05:00
} ) ;
} else {
// Send the agent from RAM
2019-10-28 19:52:12 -04:00
obj . agentUpdate = { ptr : 0 , buf : Buffer . alloc ( parent . parent . agentUpdateBlockSize + 4 ) , taskid : taskid } ;
2019-01-28 19:53:11 -05:00
// MeshCommand_CoreModule, ask mesh agent to clear the core.
// The new core will only be sent after the agent updates.
2019-08-09 18:13:53 -04:00
obj . sendBinary ( common . ShortToStr ( 10 ) + common . ShortToStr ( 0 ) ) ;
2019-01-28 19:53:11 -05:00
2020-07-17 17:29:02 -04:00
// We got the agent file open on the server side, tell the agent we are sending an update ending with the SHA384 hash of the result
2019-08-09 18:13:53 -04:00
obj . sendBinary ( common . ShortToStr ( 13 ) + common . ShortToStr ( 0 ) ) ; // Command 13, start mesh agent download
2019-01-28 19:53:11 -05:00
// Send the first mesh agent update data block
obj . agentUpdate . buf [ 0 ] = 0 ;
obj . agentUpdate . buf [ 1 ] = 14 ;
obj . agentUpdate . buf [ 2 ] = 0 ;
obj . agentUpdate . buf [ 3 ] = 1 ;
2019-02-15 01:16:15 -05:00
2020-07-17 17:29:02 -04:00
// If agent supports compression, send the compressed agent if possible.
2020-07-28 21:45:17 -04:00
if ( ( obj . agentInfo . capabilities & 0x100 ) && ( obj . agentExeInfo . zdata != null ) ) {
2020-07-17 17:29:02 -04:00
// Send compressed data
obj . agentUpdate . agentUpdateData = obj . agentExeInfo . zdata ;
obj . agentUpdate . agentUpdateHash = obj . agentExeInfo . zhash ;
} else {
// Send uncompressed data
obj . agentUpdate . agentUpdateData = obj . agentExeInfo . data ;
obj . agentUpdate . agentUpdateHash = obj . agentExeInfo . hash ;
}
const len = Math . min ( parent . parent . agentUpdateBlockSize , obj . agentUpdate . agentUpdateData . length - obj . agentUpdate . ptr ) ;
2019-02-15 01:16:15 -05:00
if ( len > 0 ) {
// Send the first block
2020-07-17 17:29:02 -04:00
obj . agentUpdate . agentUpdateData . copy ( obj . agentUpdate . buf , 4 , obj . agentUpdate . ptr , obj . agentUpdate . ptr + len ) ;
2019-01-28 19:53:11 -05:00
obj . agentUpdate . ptr += len ;
2019-08-09 18:13:53 -04:00
obj . sendBinary ( obj . agentUpdate . buf ) ; // Command 14, mesh agent first data block
2019-10-28 13:35:58 -04:00
parent . parent . debug ( 'agentupdate' , "Sent first block of " + len + " bytes from RAM." ) ;
2019-02-15 01:16:15 -05:00
} else {
// Error
2019-10-28 13:35:58 -04:00
parent . parent . debug ( 'agentupdate' , "ERROR: Len of " + len + " is invalid." ) ;
2019-03-08 01:47:27 -05:00
parent . parent . taskLimiter . completed ( obj . agentUpdate . taskid ) ; // Indicate this task complete
2019-03-10 14:00:14 -04:00
delete obj . agentUpdate . buf ;
2019-03-08 01:47:27 -05:00
delete obj . agentUpdate ;
2019-01-28 19:53:11 -05:00
}
2019-02-15 01:16:15 -05:00
}
2019-02-08 17:17:35 -05:00
} , null , 1 ) ;
2019-01-28 19:53:11 -05:00
2019-01-21 17:05:50 -05:00
} else {
// Check the mesh core, if the agent is capable of running one
2019-03-08 01:47:27 -05:00
if ( ( ( obj . agentInfo . capabilities & 16 ) != 0 ) && ( parent . parent . meshAgentsArchitectureNumbers [ obj . agentInfo . agentId ] . core != null ) ) {
2019-08-09 18:13:53 -04:00
obj . sendBinary ( common . ShortToStr ( 11 ) + common . ShortToStr ( 0 ) ) ; // Command 11, ask for mesh core hash.
2019-01-22 18:40:08 -05:00
}
2017-08-28 12:27:45 -04:00
}
}
}
else if ( cmdid == 14 ) { // MeshCommand_AgentBinaryBlock
if ( ( msg . length == 4 ) && ( obj . agentUpdate != null ) ) {
2019-03-08 01:47:27 -05:00
const status = common . ReadShort ( msg , 2 ) ;
2017-08-28 12:27:45 -04:00
if ( status == 1 ) {
2019-02-15 01:16:15 -05:00
if ( obj . agentExeInfo . data == null ) {
// Read the agent from disk
2019-10-28 19:52:12 -04:00
parent . fs . read ( obj . agentUpdate . fd , obj . agentUpdate . buf , 4 , parent . parent . agentUpdateBlockSize , obj . agentUpdate . ptr , function ( err , bytesRead , buffer ) {
2019-03-09 18:04:43 -05:00
if ( ( obj . agentExeInfo == null ) || ( obj . agentUpdate == null ) ) return ; // Agent disconnected during this async call.
2019-03-08 01:47:27 -05:00
if ( ( err != null ) || ( bytesRead < 0 ) ) {
// Error reading the agent file, stop here.
2019-10-28 13:35:58 -04:00
parent . parent . debug ( 'agentupdate' , "ERROR: Unable to read agent #" + obj . agentExeInfo . id + " binary from disk." ) ;
2019-03-08 01:47:27 -05:00
try { parent . fs . close ( obj . agentUpdate . fd ) ; } catch ( ex ) { }
parent . parent . taskLimiter . completed ( obj . agentUpdate . taskid ) ; // Indicate this task complete
2019-03-10 14:00:14 -04:00
delete obj . agentUpdate . buf ;
2019-03-08 01:47:27 -05:00
delete obj . agentUpdate ;
} else {
// Send the next block to the agent
2019-10-28 13:35:58 -04:00
parent . parent . debug ( 'agentupdate' , "Sending disk agent #" + obj . agentExeInfo . id + " block, ptr=" + obj . agentUpdate . ptr + ", len=" + bytesRead + "." ) ;
2019-03-08 01:47:27 -05:00
obj . agentUpdate . ptr += bytesRead ;
2019-10-28 19:52:12 -04:00
if ( bytesRead == parent . parent . agentUpdateBlockSize ) { obj . sendBinary ( obj . agentUpdate . buf ) ; } else { obj . sendBinary ( obj . agentUpdate . buf . slice ( 0 , bytesRead + 4 ) ) ; } // Command 14, mesh agent next data block
if ( ( bytesRead < parent . parent . agentUpdateBlockSize ) || ( obj . agentUpdate . ptr == obj . agentExeInfo . size ) ) {
2019-10-28 13:35:58 -04:00
parent . parent . debug ( 'agentupdate' , "Completed agent #" + obj . agentExeInfo . id + " update from disk, ptr=" + obj . agentUpdate . ptr + "." ) ;
2019-08-09 18:13:53 -04:00
obj . sendBinary ( common . ShortToStr ( 13 ) + common . ShortToStr ( 0 ) + obj . agentExeInfo . hash ) ; // Command 13, end mesh agent download, send agent SHA384 hash
2019-03-08 01:47:27 -05:00
try { parent . fs . close ( obj . agentUpdate . fd ) ; } catch ( ex ) { }
parent . parent . taskLimiter . completed ( obj . agentUpdate . taskid ) ; // Indicate this task complete
2019-03-10 14:00:14 -04:00
delete obj . agentUpdate . buf ;
2019-03-08 01:47:27 -05:00
delete obj . agentUpdate ;
}
2019-02-15 01:16:15 -05:00
}
2019-03-08 01:47:27 -05:00
} ) ;
2017-08-28 12:27:45 -04:00
} else {
2019-02-15 01:16:15 -05:00
// Send the agent from RAM
2020-07-17 17:29:02 -04:00
const len = Math . min ( parent . parent . agentUpdateBlockSize , obj . agentUpdate . agentUpdateData . length - obj . agentUpdate . ptr ) ;
2019-02-15 01:16:15 -05:00
if ( len > 0 ) {
2020-07-17 17:29:02 -04:00
obj . agentUpdate . agentUpdateData . copy ( obj . agentUpdate . buf , 4 , obj . agentUpdate . ptr , obj . agentUpdate . ptr + len ) ;
2019-10-28 19:52:12 -04:00
if ( len == parent . parent . agentUpdateBlockSize ) { obj . sendBinary ( obj . agentUpdate . buf ) ; } else { obj . sendBinary ( obj . agentUpdate . buf . slice ( 0 , len + 4 ) ) ; } // Command 14, mesh agent next data block
2019-10-28 13:35:58 -04:00
parent . parent . debug ( 'agentupdate' , "Sending RAM agent #" + obj . agentExeInfo . id + " block, ptr=" + obj . agentUpdate . ptr + ", len=" + len + "." ) ;
2019-02-15 01:16:15 -05:00
obj . agentUpdate . ptr += len ;
}
2017-08-28 12:27:45 -04:00
2020-07-17 17:29:02 -04:00
if ( obj . agentUpdate . ptr == obj . agentUpdate . agentUpdateData . length ) {
2019-10-28 13:35:58 -04:00
parent . parent . debug ( 'agentupdate' , "Completed agent #" + obj . agentExeInfo . id + " update from RAM, ptr=" + obj . agentUpdate . ptr + "." ) ;
2020-07-17 17:29:02 -04:00
obj . sendBinary ( common . ShortToStr ( 13 ) + common . ShortToStr ( 0 ) + obj . agentUpdate . agentUpdateHash ) ; // Command 13, end mesh agent download, send agent SHA384 hash
2019-03-08 01:47:27 -05:00
parent . parent . taskLimiter . completed ( obj . agentUpdate . taskid ) ; // Indicate this task complete
2019-03-10 14:00:14 -04:00
delete obj . agentUpdate . buf ;
2019-03-08 01:47:27 -05:00
delete obj . agentUpdate ;
2017-08-28 12:27:45 -04:00
}
}
}
}
}
else if ( cmdid == 15 ) { // MeshCommand_AgentTag
2018-01-23 17:15:59 -05:00
var tag = msg . substring ( 2 ) ;
while ( tag . charCodeAt ( tag . length - 1 ) == 0 ) { tag = tag . substring ( 0 , tag . length - 1 ) ; } // Remove end-of-line zeros.
ChangeAgentTag ( tag ) ;
2017-08-28 12:27:45 -04:00
}
2018-01-23 17:15:59 -05:00
} else if ( obj . authenticated < 2 ) { // We are not authenticated
2020-01-29 17:38:41 -05:00
// Check if this is a un-authenticated JSON
if ( msg . charCodeAt ( 0 ) == 123 ) {
var str = msg . toString ( 'utf8' ) , command = null ;
if ( str [ 0 ] == '{' ) {
try { command = JSON . parse ( str ) ; } catch ( ex ) { } // If the command can't be parsed, ignore it.
if ( ( command != null ) && ( command . action === 'agentName' ) && ( typeof command . value == 'string' ) && ( command . value . length > 0 ) && ( command . value . length < 256 ) ) { obj . agentName = command . value ; }
}
return ;
}
2019-03-08 01:47:27 -05:00
const cmd = common . ReadShort ( msg , 0 ) ;
2017-08-28 12:27:45 -04:00
if ( cmd == 1 ) {
// Agent authentication request
2017-10-15 02:22:19 -04:00
if ( ( msg . length != 98 ) || ( ( obj . receivedCommands & 1 ) != 0 ) ) return ;
2017-08-28 12:27:45 -04:00
obj . receivedCommands += 1 ; // Agent can't send the same command twice on the same connection ever. Block DOS attack path.
2020-12-30 19:54:02 -05:00
if ( isIgnoreHashCheck ( ) ) {
2018-12-15 15:34:55 -05:00
// Send the agent web hash back to the agent
2020-05-22 15:54:22 -04:00
// Send 384 bits SHA384 hash of TLS cert + 384 bits nonce
2019-08-09 18:13:53 -04:00
obj . sendBinary ( common . ShortToStr ( 1 ) + msg . substring ( 2 , 50 ) + obj . nonce ) ; // Command 1, hash + nonce. Use the web hash given by the agent.
2018-12-15 15:34:55 -05:00
} else {
// Check that the server hash matches our own web certificate hash (SHA384)
2020-05-22 15:54:22 -04:00
const agentSeenCerthash = msg . substring ( 2 , 50 ) ;
if ( ( getWebCertHash ( domain ) != agentSeenCerthash ) && ( getWebCertFullHash ( domain ) != agentSeenCerthash ) && ( parent . defaultWebCertificateHash != agentSeenCerthash ) && ( parent . defaultWebCertificateFullHash != agentSeenCerthash ) ) {
2019-10-29 19:17:29 -04:00
if ( parent . parent . supportsProxyCertificatesRequest !== false ) {
obj . badWebCert = Buffer . from ( parent . crypto . randomBytes ( 16 ) , 'binary' ) . toString ( 'base64' ) ;
parent . wsagentsWithBadWebCerts [ obj . badWebCert ] = obj ; // Add this agent to the list of of agents with bad web certificates.
2019-10-29 19:31:03 -04:00
parent . parent . updateProxyCertificates ( false ) ;
2019-10-29 19:17:29 -04:00
}
2019-05-01 18:02:03 -04:00
parent . agentStats . agentBadWebCertHashCount ++ ;
2020-04-24 16:17:06 -04:00
parent . parent . debug ( 'agent' , 'Agent bad web cert hash (Agent:' + ( Buffer . from ( msg . substring ( 2 , 50 ) , 'binary' ) . toString ( 'hex' ) . substring ( 0 , 10 ) ) + ' != Server:' + ( Buffer . from ( getWebCertHash ( domain ) , 'binary' ) . toString ( 'hex' ) . substring ( 0 , 10 ) ) + ' or ' + ( Buffer . from ( getWebCertFullHash ( domain ) , 'binary' ) . toString ( 'hex' ) . substring ( 0 , 10 ) ) + '), holding connection (' + obj . remoteaddrport + ').' ) ;
parent . parent . debug ( 'agent' , 'Agent reported web cert hash:' + ( Buffer . from ( msg . substring ( 2 , 50 ) , 'binary' ) . toString ( 'hex' ) ) + '.' ) ;
2020-03-19 16:42:37 -04:00
console . log ( 'Agent bad web cert hash (Agent:' + ( Buffer . from ( msg . substring ( 2 , 50 ) , 'binary' ) . toString ( 'hex' ) . substring ( 0 , 10 ) ) + ' != Server:' + ( Buffer . from ( getWebCertHash ( domain ) , 'binary' ) . toString ( 'hex' ) . substring ( 0 , 10 ) ) + ' or ' + ( Buffer . from ( getWebCertFullHash ( domain ) , 'binary' ) . toString ( 'hex' ) . substring ( 0 , 10 ) ) + '), holding connection (' + obj . remoteaddrport + ').' ) ;
2019-02-14 17:44:38 -05:00
console . log ( 'Agent reported web cert hash:' + ( Buffer . from ( msg . substring ( 2 , 50 ) , 'binary' ) . toString ( 'hex' ) ) + '.' ) ;
return ;
2020-05-22 15:54:22 -04:00
} else {
// The hash matched one of the acceptable values, send the agent web hash back to the agent
// Send 384 bits SHA384 hash of TLS cert + 384 bits nonce
// Command 1, hash + nonce. Use the web hash given by the agent.
obj . sendBinary ( common . ShortToStr ( 1 ) + agentSeenCerthash + obj . nonce ) ;
2019-02-14 17:44:38 -05:00
}
2018-12-15 15:34:55 -05:00
}
2017-08-28 12:27:45 -04:00
// Use our server private key to sign the ServerHash + AgentNonce + ServerNonce
2018-10-31 19:03:09 -04:00
obj . agentnonce = msg . substring ( 50 , 98 ) ;
2018-08-21 18:08:15 -04:00
// Check if we got the agent auth confirmation
if ( ( obj . receivedCommands & 8 ) == 0 ) {
// If we did not get an indication that the agent already validated this server, send the server signature.
if ( obj . useSwarmCert == true ) {
// Perform the hash signature using older swarm server certificate
2019-08-12 17:58:06 -04:00
parent . parent . certificateOperations . acceleratorPerformSignature ( 1 , msg . substring ( 2 ) + obj . nonce , null , function ( tag , signature ) {
2018-08-21 18:08:15 -04:00
// Send back our certificate + signature
2019-08-13 16:55:25 -04:00
obj . sendBinary ( common . ShortToStr ( 2 ) + common . ShortToStr ( parent . swarmCertificateAsn1 . length ) + parent . swarmCertificateAsn1 + signature ) ; // Command 2, certificate + signature
2018-08-21 18:08:15 -04:00
} ) ;
} else {
// Perform the hash signature using the server agent certificate
2019-08-12 17:58:06 -04:00
parent . parent . certificateOperations . acceleratorPerformSignature ( 0 , msg . substring ( 2 ) + obj . nonce , null , function ( tag , signature ) {
2018-08-21 18:08:15 -04:00
// Send back our certificate + signature
2019-08-13 16:55:25 -04:00
obj . sendBinary ( common . ShortToStr ( 2 ) + common . ShortToStr ( parent . agentCertificateAsn1 . length ) + parent . agentCertificateAsn1 + signature ) ; // Command 2, certificate + signature
2018-08-21 18:08:15 -04:00
} ) ;
}
2017-11-07 20:05:22 -05:00
}
2017-08-28 12:27:45 -04:00
// Check the agent signature if we can
2017-09-15 14:45:06 -04:00
if ( obj . unauthsign != null ) {
2019-05-01 18:02:03 -04:00
if ( processAgentSignature ( obj . unauthsign ) == false ) {
parent . agentStats . agentBadSignature1Count ++ ;
2020-04-24 16:17:06 -04:00
parent . parent . debug ( 'agent' , 'Agent connected with bad signature, holding connection (' + obj . remoteaddrport + ').' ) ;
2019-05-01 18:02:03 -04:00
console . log ( 'Agent connected with bad signature, holding connection (' + obj . remoteaddrport + ').' ) ; return ;
} else { completeAgentConnection ( ) ; }
2017-08-28 12:27:45 -04:00
}
}
else if ( cmd == 2 ) {
// Agent certificate
if ( ( msg . length < 4 ) || ( ( obj . receivedCommands & 2 ) != 0 ) ) return ;
obj . receivedCommands += 2 ; // Agent can't send the same command twice on the same connection ever. Block DOS attack path.
// Decode the certificate
2019-03-08 01:47:27 -05:00
const certlen = common . ReadShort ( msg , 2 ) ;
2017-08-28 12:27:45 -04:00
obj . unauth = { } ;
2020-04-24 16:17:06 -04:00
try { obj . unauth . nodeid = Buffer . from ( forge . pki . getPublicKeyFingerprint ( forge . pki . certificateFromAsn1 ( forge . asn1 . fromDer ( msg . substring ( 4 , 4 + certlen ) ) ) . publicKey , { md : forge . md . sha384 . create ( ) } ) . data , 'binary' ) . toString ( 'base64' ) . replace ( /\+/g , '@' ) . replace ( /\//g , '$' ) ; } catch ( ex ) { console . log ( ex ) ; parent . parent . debug ( 'agent' , ex ) ; return ; }
2019-01-02 21:03:34 -05:00
obj . unauth . nodeCertPem = '-----BEGIN CERTIFICATE-----\r\n' + Buffer . from ( msg . substring ( 4 , 4 + certlen ) , 'binary' ) . toString ( 'base64' ) + '\r\n-----END CERTIFICATE-----' ;
2017-08-28 12:27:45 -04:00
// Check the agent signature if we can
2019-05-01 18:02:03 -04:00
if ( obj . agentnonce == null ) { obj . unauthsign = msg . substring ( 4 + certlen ) ; } else {
if ( processAgentSignature ( msg . substring ( 4 + certlen ) ) == false ) {
parent . agentStats . agentBadSignature2Count ++ ;
2020-04-24 16:17:06 -04:00
parent . parent . debug ( 'agent' , 'Agent connected with bad signature, holding connection (' + obj . remoteaddrport + ').' ) ;
2019-05-01 18:02:03 -04:00
console . log ( 'Agent connected with bad signature, holding connection (' + obj . remoteaddrport + ').' ) ; return ;
}
}
2017-08-28 12:27:45 -04:00
completeAgentConnection ( ) ;
}
else if ( cmd == 3 ) {
// Agent meshid
2020-01-29 17:38:41 -05:00
if ( ( msg . length < 70 ) || ( ( obj . receivedCommands & 4 ) != 0 ) ) return ;
2017-08-28 12:27:45 -04:00
obj . receivedCommands += 4 ; // Agent can't send the same command twice on the same connection ever. Block DOS attack path.
// Set the meshid
obj . agentInfo = { } ;
2019-03-08 01:47:27 -05:00
obj . agentInfo . infoVersion = common . ReadInt ( msg , 2 ) ;
obj . agentInfo . agentId = common . ReadInt ( msg , 6 ) ;
obj . agentInfo . agentVersion = common . ReadInt ( msg , 10 ) ;
obj . agentInfo . platformType = common . ReadInt ( msg , 14 ) ;
2020-04-17 14:27:34 -04:00
if ( obj . agentInfo . platformType > 8 || obj . agentInfo . platformType < 1 ) { obj . agentInfo . platformType = 1 ; }
2017-11-07 20:05:22 -05:00
if ( msg . substring ( 50 , 66 ) == '\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0' ) {
2019-01-02 21:03:34 -05:00
obj . meshid = Buffer . from ( msg . substring ( 18 , 50 ) , 'binary' ) . toString ( 'hex' ) ; // Older HEX MeshID
2017-11-07 20:05:22 -05:00
} else {
2019-01-02 21:03:34 -05:00
obj . meshid = Buffer . from ( msg . substring ( 18 , 66 ) , 'binary' ) . toString ( 'base64' ) . replace ( /\+/g , '@' ) . replace ( /\//g , '$' ) ; // New Base64 MeshID
2017-11-07 20:05:22 -05:00
}
//console.log('MeshID', obj.meshid);
2019-03-08 01:47:27 -05:00
obj . agentInfo . capabilities = common . ReadInt ( msg , 66 ) ;
2020-01-29 17:38:41 -05:00
if ( msg . length > 70 ) {
const computerNameLen = common . ReadShort ( msg , 70 ) ;
obj . agentInfo . computerName = Buffer . from ( msg . substring ( 72 , 72 + computerNameLen ) , 'binary' ) . toString ( 'utf8' ) ;
//console.log('computerName', msg.length, computerNameLen, obj.agentInfo.computerName);
} else {
obj . agentInfo . computerName = '' ;
//console.log('computerName-none');
}
2019-03-08 01:47:27 -05:00
obj . dbMeshKey = 'mesh/' + domain . id + '/' + obj . meshid ;
2017-08-28 12:27:45 -04:00
completeAgentConnection ( ) ;
2018-08-21 18:08:15 -04:00
} else if ( cmd == 4 ) {
if ( ( msg . length < 2 ) || ( ( obj . receivedCommands & 8 ) != 0 ) ) return ;
obj . receivedCommands += 8 ; // Agent can't send the same command twice on the same connection ever. Block DOS attack path.
// Agent already authenticated the server, wants to skip the server signature - which is great for server performance.
2017-11-07 20:05:22 -05:00
} else if ( cmd == 5 ) {
// ServerID. Agent is telling us what serverid it expects. Useful if we have many server certificates.
2019-03-08 01:47:27 -05:00
if ( ( msg . substring ( 2 , 34 ) == parent . swarmCertificateHash256 ) || ( msg . substring ( 2 , 50 ) == parent . swarmCertificateHash384 ) ) { obj . useSwarmCert = true ; }
2017-08-28 12:27:45 -04:00
}
}
} ) ;
// If error, do nothing
2020-04-24 16:17:06 -04:00
ws . on ( 'error' , function ( err ) { parent . parent . debug ( 'agent' , 'AGENT WSERR: ' + err ) ; console . log ( 'AGENT WSERR: ' + err ) ; obj . close ( 0 ) ; } ) ;
2017-08-28 12:27:45 -04:00
// If the mesh agent web socket is closed, clean up.
2019-01-22 20:47:51 -05:00
ws . on ( 'close' , function ( req ) {
2019-05-24 18:11:00 -04:00
parent . agentStats . agentClose ++ ;
2019-01-22 20:47:51 -05:00
if ( obj . nodeid != null ) {
2019-03-08 01:47:27 -05:00
const agentId = ( obj . agentInfo && obj . agentInfo . agentId ) ? obj . agentInfo . agentId : 'Unknown' ;
2019-02-12 16:37:01 -05:00
//console.log('Agent disconnect ' + obj.nodeid + ' (' + obj.remoteaddrport + ') id=' + agentId);
2019-08-22 18:31:39 -04:00
parent . parent . debug ( 'agent' , 'Agent disconnect ' + obj . nodeid + ' (' + obj . remoteaddrport + ') id=' + agentId ) ;
2019-01-22 22:34:55 -05:00
// Log the agent disconnection
2019-03-08 01:47:27 -05:00
if ( parent . wsagentsDisconnections [ obj . nodeid ] == null ) {
parent . wsagentsDisconnections [ obj . nodeid ] = 1 ;
2019-01-22 22:34:55 -05:00
} else {
2019-03-08 01:47:27 -05:00
parent . wsagentsDisconnections [ obj . nodeid ] = ++ parent . wsagentsDisconnections [ obj . nodeid ] ;
2019-01-22 22:34:55 -05:00
}
2019-01-22 20:47:51 -05:00
}
obj . close ( 0 ) ;
} ) ;
2017-08-28 12:27:45 -04:00
2019-02-14 17:44:38 -05:00
// Return the mesh for this device, in some cases, we may auto-create the mesh.
function getMeshAutoCreate ( ) {
2019-03-17 22:34:52 -04:00
var mesh = parent . meshes [ obj . dbMeshKey ] ;
2019-10-04 14:38:47 -04:00
// If the mesh was not found and we are in LAN mode, check of the domain can be corrected
if ( ( args . lanonly == true ) && ( mesh == null ) ) {
var smesh = obj . dbMeshKey . split ( '/' ) ;
for ( var i in parent . parent . config . domains ) {
mesh = parent . meshes [ 'mesh/' + i + '/' + smesh [ 2 ] ] ;
if ( mesh != null ) {
obj . domain = domain = parent . parent . config . domains [ i ] ;
obj . meshid = smesh [ 2 ] ;
obj . dbMeshKey = 'mesh/' + i + '/' + smesh [ 2 ] ;
obj . dbNodeKey = 'node/' + domain . id + '/' + obj . nodeid ;
break ;
}
}
}
2019-03-08 01:47:27 -05:00
if ( ( mesh == null ) && ( typeof domain . orphanagentuser == 'string' ) ) {
const adminUser = parent . users [ 'user/' + domain . id + '/' + domain . orphanagentuser . toLowerCase ( ) ] ;
2019-02-14 17:44:38 -05:00
if ( ( adminUser != null ) && ( adminUser . siteadmin == 0xFFFFFFFF ) ) {
// Mesh name is hex instead of base64
2019-03-08 01:47:27 -05:00
const meshname = obj . meshid . substring ( 0 , 18 ) ;
2019-02-14 17:44:38 -05:00
// Create a new mesh for this device
2019-03-08 01:47:27 -05:00
const links = { } ;
2019-02-14 17:44:38 -05:00
links [ adminUser . _id ] = { name : adminUser . name , rights : 0xFFFFFFFF } ;
mesh = { type : 'mesh' , _id : obj . dbMeshKey , name : meshname , mtype : 2 , desc : '' , domain : domain . id , links : links } ;
2020-03-26 22:33:13 -04:00
db . Set ( mesh ) ;
2019-03-08 01:47:27 -05:00
parent . meshes [ obj . dbMeshKey ] = mesh ;
2019-02-14 17:44:38 -05:00
2019-02-14 18:53:22 -05:00
if ( adminUser . links == null ) adminUser . links = { } ;
2019-02-14 17:44:38 -05:00
adminUser . links [ obj . dbMeshKey ] = { rights : 0xFFFFFFFF } ;
2019-03-08 01:47:27 -05:00
db . SetUser ( adminUser ) ;
2020-09-08 21:59:58 -04:00
parent . parent . DispatchEvent ( parent . CreateMeshDispatchTargets ( obj . dbMeshKey , [ adminUser . _id , obj . dbNodeKey ] ) , obj , { etype : 'mesh' , username : adminUser . name , meshid : obj . dbMeshKey , name : meshname , mtype : 2 , desc : '' , action : 'createmesh' , links : links , msgid : 55 , msgArgs : [ obj . meshid ] , msg : "Created device group: " + obj . meshid , domain : domain . id } ) ;
2019-02-14 17:44:38 -05:00
}
2019-02-18 17:32:55 -05:00
} else {
2019-02-20 12:39:41 -05:00
if ( ( mesh != null ) && ( mesh . deleted != null ) && ( mesh . links ) ) {
2019-02-18 17:32:55 -05:00
// Must un-delete this mesh
2020-03-17 20:17:04 -04:00
var ids = parent . CreateMeshDispatchTargets ( mesh . _id , [ obj . dbNodeKey ] ) ;
2019-02-18 17:32:55 -05:00
// See if users still exists, if so, add links to the mesh
for ( var userid in mesh . links ) {
2019-03-08 01:47:27 -05:00
const user = parent . users [ userid ] ;
2019-02-18 17:32:55 -05:00
if ( user ) {
if ( user . links == null ) { user . links = { } ; }
if ( user . links [ mesh . _id ] == null ) {
user . links [ mesh . _id ] = { rights : mesh . links [ userid ] . rights } ;
ids . push ( user . _id ) ;
2019-03-08 01:47:27 -05:00
db . SetUser ( user ) ;
2019-02-18 17:32:55 -05:00
}
}
}
// Send out an event indicating this mesh was "created"
2020-09-08 21:59:58 -04:00
parent . parent . DispatchEvent ( ids , obj , { etype : 'mesh' , meshid : mesh . _id , name : mesh . name , mtype : mesh . mtype , desc : mesh . desc , action : 'createmesh' , links : mesh . links , msgid : 56 , msgArgs : [ mesh . _id ] , msg : "Device group undeleted: " + mesh . _id , domain : domain . id } ) ;
2019-02-18 17:32:55 -05:00
// Mark the mesh as active
delete mesh . deleted ;
2020-03-26 22:33:13 -04:00
db . Set ( mesh ) ;
2019-02-18 17:32:55 -05:00
}
2019-02-14 17:44:38 -05:00
}
return mesh ;
}
2019-05-28 13:58:46 -04:00
// Send a PING/PONG message
function sendPing ( ) { obj . send ( '{"action":"ping"}' ) ; }
function sendPong ( ) { obj . send ( '{"action":"pong"}' ) ; }
2017-08-28 12:27:45 -04:00
// Once we get all the information about an agent, run this to hook everything up to the server
function completeAgentConnection ( ) {
2019-03-10 14:47:03 -04:00
if ( ( obj . authenticated != 1 ) || ( obj . meshid == null ) || obj . pendingCompleteAgentConnection || ( obj . agentInfo == null ) ) { return ; }
2018-08-21 14:02:35 -04:00
obj . pendingCompleteAgentConnection = true ;
2019-05-28 13:58:46 -04:00
// Setup the agent PING/PONG timers
if ( ( typeof args . agentping == 'number' ) && ( obj . pingtimer == null ) ) { obj . pingtimer = setInterval ( sendPing , args . agentping * 1000 ) ; }
else if ( ( typeof args . agentpong == 'number' ) && ( obj . pongtimer == null ) ) { obj . pongtimer = setInterval ( sendPong , args . agentpong * 1000 ) ; }
2019-04-10 13:41:10 -04:00
// If this is a recovery agent
2019-04-09 18:51:05 -04:00
if ( obj . agentInfo . capabilities & 0x40 ) {
2019-04-10 13:41:10 -04:00
// Inform mesh agent that it's authenticated.
delete obj . pendingCompleteAgentConnection ;
obj . authenticated = 2 ;
2019-08-09 18:13:53 -04:00
obj . sendBinary ( common . ShortToStr ( 4 ) ) ;
2019-04-10 13:41:10 -04:00
// Ask for mesh core hash.
2019-08-09 18:13:53 -04:00
obj . sendBinary ( common . ShortToStr ( 11 ) + common . ShortToStr ( 0 ) ) ;
2019-04-09 18:51:05 -04:00
return ;
}
2019-02-11 17:41:15 -05:00
// Check if we have too many agent sessions
if ( typeof domain . limits . maxagentsessions == 'number' ) {
// Count the number of agent sessions for this domain
var domainAgentSessionCount = 0 ;
2019-03-08 01:47:27 -05:00
for ( var i in parent . wsagents ) { if ( parent . wsagents [ i ] . domain . id == domain . id ) { domainAgentSessionCount ++ ; } }
2019-02-11 17:41:15 -05:00
// Check if we have too many user sessions
2019-05-01 18:02:03 -04:00
if ( domainAgentSessionCount >= domain . limits . maxagentsessions ) {
// Too many, hold the connection.
parent . agentStats . agentMaxSessionHoldCount ++ ;
return ;
}
2019-02-11 17:41:15 -05:00
}
2019-02-14 17:44:38 -05:00
/ *
2017-08-28 12:27:45 -04:00
// Check that the mesh exists
2019-03-08 01:47:27 -05:00
var mesh = parent . meshes [ obj . dbMeshKey ] ;
2019-02-14 17:44:38 -05:00
if ( mesh == null ) {
var holdConnection = true ;
2019-03-08 01:47:27 -05:00
if ( typeof domain . orphanagentuser == 'string' ) {
var adminUser = parent . users [ 'user/' + domain . id + '/' + args . orphanagentuser ] ;
2019-02-14 17:44:38 -05:00
if ( ( adminUser != null ) && ( adminUser . siteadmin == 0xFFFFFFFF ) ) {
// Create a new mesh for this device
holdConnection = false ;
var links = { } ;
links [ user . _id ] = { name : adminUser . name , rights : 0xFFFFFFFF } ;
mesh = { type : 'mesh' , _id : obj . dbMeshKey , name : obj . meshid , mtype : 2 , desc : '' , domain : domain . id , links : links } ;
2020-03-26 22:33:13 -04:00
db . Set ( mesh ) ;
2019-03-08 01:47:27 -05:00
parent . meshes [ obj . meshid ] = mesh ;
2020-03-17 20:17:04 -04:00
parent . parent . AddEventDispatch ( parent . CreateMeshDispatchTargets ( obj . meshid , [ obj . dbNodeKey ] ) , ws ) ;
2019-02-14 17:44:38 -05:00
if ( adminUser . links == null ) user . links = { } ;
adminUser . links [ obj . meshid ] = { rights : 0xFFFFFFFF } ;
2019-03-08 01:47:27 -05:00
//adminUser.subscriptions = parent.subscribe(adminUser._id, ws);
db . SetUser ( user ) ;
2020-03-17 20:17:04 -04:00
parent . parent . DispatchEvent ( parent . CreateMeshDispatchTargets ( meshid , [ user . _id , obj . dbNodeKey ] ) , obj , { etype : 'mesh' , username : user . name , meshid : obj . meshid , name : obj . meshid , mtype : 2 , desc : '' , action : 'createmesh' , links : links , msg : 'Mesh created: ' + obj . meshid , domain : domain . id } ) ;
2019-02-14 17:44:38 -05:00
}
}
if ( holdConnection == true ) {
// If we disconnect, the agent will just reconnect. We need to log this or tell agent to connect in a few hours.
2020-04-24 16:17:06 -04:00
parent . parent . debug ( 'agent' , 'Agent connected with invalid domain/mesh, holding connection (' + obj . remoteaddrport + ', ' + obj . dbMeshKey + ').' ) ;
2019-02-14 17:44:38 -05:00
console . log ( 'Agent connected with invalid domain/mesh, holding connection (' + obj . remoteaddrport + ', ' + obj . dbMeshKey + ').' ) ;
return ;
}
}
2020-04-24 16:17:06 -04:00
if ( mesh . mtype != 2 ) { // If we disconnect, the agnet will just reconnect. We need to log this or tell agent to connect in a few hours.
parent . parent . debug ( 'agent' , 'Agent connected with invalid mesh type, holding connection (' + obj . remoteaddrport + ').' ) ;
console . log ( 'Agent connected with invalid mesh type, holding connection (' + obj . remoteaddrport + ').' ) ;
return ;
}
2019-02-14 17:44:38 -05:00
* /
2018-08-21 18:08:15 -04:00
// Check that the node exists
2019-03-08 01:47:27 -05:00
db . Get ( obj . dbNodeKey , function ( err , nodes ) {
2019-03-10 14:47:03 -04:00
if ( obj . agentInfo == null ) { return ; }
2019-05-21 17:19:32 -04:00
var device , mesh ;
2018-08-21 18:08:15 -04:00
2018-09-24 20:47:03 -04:00
// See if this node exists in the database
2019-05-20 21:03:14 -04:00
if ( ( nodes == null ) || ( nodes . length == 0 ) ) {
2019-02-14 17:44:38 -05:00
// This device does not exist, use the meshid given by the device
2019-06-05 18:24:07 -04:00
// Check if we already have too many devices for this domain
if ( domain . limits && ( typeof domain . limits . maxdevices == 'number' ) ) {
db . isMaxType ( domain . limits . maxdevices , 'node' , domain . id , function ( ismax , count ) {
if ( ismax == true ) {
// Too many devices in this domain.
parent . agentStats . maxDomainDevicesReached ++ ;
} else {
// We are under the limit, create the new device.
completeAgentConnection2 ( ) ;
}
} ) ;
2017-08-28 12:27:45 -04:00
} else {
2019-06-05 18:24:07 -04:00
completeAgentConnection2 ( ) ;
2017-08-28 12:27:45 -04:00
}
2019-06-05 18:24:07 -04:00
return ;
2018-08-21 18:08:15 -04:00
} else {
device = nodes [ 0 ] ;
2019-02-14 17:44:38 -05:00
// This device exists, meshid given by the device must be ignored, use the server side one.
2020-04-02 14:26:54 -04:00
if ( ( device . meshid != null ) && ( device . meshid != obj . dbMeshKey ) ) {
2019-02-14 17:44:38 -05:00
obj . dbMeshKey = device . meshid ;
obj . meshid = device . meshid . split ( '/' ) [ 2 ] ;
}
// See if this mesh exists, if it does not we may want to create it.
2019-05-21 17:19:32 -04:00
mesh = getMeshAutoCreate ( ) ;
2019-02-14 17:44:38 -05:00
// Check if the mesh exists
if ( mesh == null ) {
// If we disconnect, the agent will just reconnect. We need to log this or tell agent to connect in a few hours.
2019-05-01 18:02:03 -04:00
parent . agentStats . invalidDomainMesh2Count ++ ;
2020-04-24 16:17:06 -04:00
parent . parent . debug ( 'agent' , 'Agent connected with invalid domain/mesh, holding connection (' + obj . remoteaddrport + ', ' + obj . dbMeshKey + ').' ) ;
2019-02-14 17:44:38 -05:00
console . log ( 'Agent connected with invalid domain/mesh, holding connection (' + obj . remoteaddrport + ', ' + obj . dbMeshKey + ').' ) ;
return ;
}
// Check if the mesh is the right type
if ( mesh . mtype != 2 ) {
// If we disconnect, the agent will just reconnect. We need to log this or tell agent to connect in a few hours.
2019-05-01 18:02:03 -04:00
parent . agentStats . invalidMeshType2Count ++ ;
2020-04-24 16:17:06 -04:00
parent . parent . debug ( 'agent' , 'Agent connected with invalid mesh type, holding connection (' + obj . remoteaddrport + ').' ) ;
2019-02-14 17:44:38 -05:00
console . log ( 'Agent connected with invalid mesh type, holding connection (' + obj . remoteaddrport + ').' ) ;
return ;
2020-04-07 00:18:50 -04:00
}
2019-02-14 17:44:38 -05:00
// Mark when this device connected
obj . connectTime = Date . now ( ) ;
2019-10-06 00:48:17 -04:00
db . Set ( { _id : 'lc' + obj . dbNodeKey , type : 'lastconnect' , domain : domain . id , time : obj . connectTime , addr : obj . remoteaddrport , cause : 1 } ) ;
2019-02-14 17:44:38 -05:00
2019-04-06 00:21:28 -04:00
// Device already exists, look if changes have occured
2018-08-29 20:40:30 -04:00
var changes = [ ] , change = 0 , log = 0 ;
if ( device . agent == null ) { device . agent = { ver : obj . agentInfo . agentVersion , id : obj . agentInfo . agentId , caps : obj . agentInfo . capabilities } ; change = 1 ; }
if ( device . rname != obj . agentInfo . computerName ) { device . rname = obj . agentInfo . computerName ; change = 1 ; changes . push ( 'computer name' ) ; }
if ( device . agent . ver != obj . agentInfo . agentVersion ) { device . agent . ver = obj . agentInfo . agentVersion ; change = 1 ; changes . push ( 'agent version' ) ; }
if ( device . agent . id != obj . agentInfo . agentId ) { device . agent . id = obj . agentInfo . agentId ; change = 1 ; changes . push ( 'agent type' ) ; }
if ( ( device . agent . caps & 24 ) != ( obj . agentInfo . capabilities & 24 ) ) { device . agent . caps = obj . agentInfo . capabilities ; change = 1 ; changes . push ( 'agent capabilities' ) ; } // If agent console or javascript support changes, update capabilities
2019-04-06 00:21:28 -04:00
if ( mesh . flags && ( mesh . flags & 2 ) && ( device . name != obj . agentInfo . computerName ) ) { device . name = obj . agentInfo . computerName ; change = 1 ; } // We want the server name to be sync'ed to the hostname
2018-08-29 20:40:30 -04:00
if ( change == 1 ) {
2019-04-10 13:41:10 -04:00
// Do some clean up if needed, these values should not be in the database.
if ( device . conn != null ) { delete device . conn ; }
if ( device . pwr != null ) { delete device . pwr ; }
if ( device . agct != null ) { delete device . agct ; }
if ( device . cict != null ) { delete device . cict ; }
// Save the updated device in the database
2019-03-08 01:47:27 -05:00
db . Set ( device ) ;
2018-08-29 20:40:30 -04:00
// If this is a temporary device, don't log changes
if ( obj . agentInfo . capabilities & 0x20 ) { log = 0 ; }
// Event the node change
2019-10-11 14:16:36 -04:00
var event = { etype : 'node' , action : 'changenode' , nodeid : obj . dbNodeKey , domain : domain . id , node : parent . CloneSafeNode ( device ) } ;
2018-09-24 20:47:03 -04:00
if ( log == 0 ) { event . nolog = 1 ; } else { event . msg = 'Changed device ' + device . name + ' from group ' + mesh . name + ': ' + changes . join ( ', ' ) ; }
2019-05-28 20:25:23 -04:00
if ( db . changeStream ) { event . noact = 1 ; } // If DB change stream is active, don't use this event to change the node. Another event will come.
2020-03-17 20:17:04 -04:00
parent . parent . DispatchEvent ( parent . CreateMeshDispatchTargets ( device . meshid , [ obj . dbNodeKey ] ) , obj , event ) ;
2018-08-21 18:08:15 -04:00
}
}
2017-08-28 12:27:45 -04:00
2019-06-07 12:44:39 -04:00
completeAgentConnection3 ( device , mesh ) ;
2019-06-05 18:24:07 -04:00
} ) ;
}
2019-06-07 12:44:39 -04:00
function completeAgentConnection2 ( ) {
2019-06-05 18:24:07 -04:00
// See if this mesh exists, if it does not we may want to create it.
var mesh = getMeshAutoCreate ( ) ;
// Check if the mesh exists
if ( mesh == null ) {
// If we disconnect, the agent will just reconnect. We need to log this or tell agent to connect in a few hours.
parent . agentStats . invalidDomainMeshCount ++ ;
2020-04-24 16:17:06 -04:00
parent . parent . debug ( 'agent' , 'Agent connected with invalid domain/mesh, holding connection (' + obj . remoteaddrport + ', ' + obj . dbMeshKey + ').' ) ;
2019-06-05 18:24:07 -04:00
console . log ( 'Agent connected with invalid domain/mesh, holding connection (' + obj . remoteaddrport + ', ' + obj . dbMeshKey + ').' ) ;
return ;
}
// Check if the mesh is the right type
if ( mesh . mtype != 2 ) {
// If we disconnect, the agent will just reconnect. We need to log this or tell agent to connect in a few hours.
parent . agentStats . invalidMeshTypeCount ++ ;
2020-04-24 16:17:06 -04:00
parent . parent . debug ( 'agent' , 'Agent connected with invalid mesh type, holding connection (' + obj . remoteaddrport + ').' ) ;
2019-06-05 18:24:07 -04:00
console . log ( 'Agent connected with invalid mesh type, holding connection (' + obj . remoteaddrport + ').' ) ;
return ;
}
// Mark when this device connected
obj . connectTime = Date . now ( ) ;
2019-10-06 00:48:17 -04:00
db . Set ( { _id : 'lc' + obj . dbNodeKey , type : 'lastconnect' , domain : domain . id , time : obj . connectTime , addr : obj . remoteaddrport , cause : 1 } ) ;
2019-06-05 18:24:07 -04:00
// This node does not exist, create it.
2020-01-29 17:38:41 -05:00
var agentName = obj . agentName ? obj . agentName : obj . agentInfo . computerName ;
var device = { type : 'node' , mtype : mesh . mtype , _id : obj . dbNodeKey , icon : obj . agentInfo . platformType , meshid : obj . dbMeshKey , name : agentName , rname : obj . agentInfo . computerName , domain : domain . id , agent : { ver : obj . agentInfo . agentVersion , id : obj . agentInfo . agentId , caps : obj . agentInfo . capabilities } , host : null } ;
2019-06-05 18:24:07 -04:00
db . Set ( device ) ;
// Event the new node
if ( obj . agentInfo . capabilities & 0x20 ) {
// This is a temporary agent, don't log.
2020-03-17 20:17:04 -04:00
parent . parent . DispatchEvent ( parent . CreateMeshDispatchTargets ( obj . dbMeshKey , [ obj . dbNodeKey ] ) , obj , { etype : 'node' , action : 'addnode' , node : device , domain : domain . id , nolog : 1 } ) ;
2019-06-05 18:24:07 -04:00
} else {
2020-09-08 21:59:58 -04:00
parent . parent . DispatchEvent ( parent . CreateMeshDispatchTargets ( obj . dbMeshKey , [ obj . dbNodeKey ] ) , obj , { etype : 'node' , action : 'addnode' , node : device , msgid : 57 , msgArgs : [ obj . agentInfo . computerName , mesh . name ] , msg : ( 'Added device ' + obj . agentInfo . computerName + ' to device group ' + mesh . name ) , domain : domain . id } ) ;
2019-06-05 18:24:07 -04:00
}
2019-06-07 12:44:39 -04:00
completeAgentConnection3 ( device , mesh ) ;
2019-06-05 18:24:07 -04:00
}
2019-06-07 12:44:39 -04:00
function completeAgentConnection3 ( device , mesh ) {
2019-06-05 18:24:07 -04:00
// Check if this agent is already connected
const dupAgent = parent . wsagents [ obj . dbNodeKey ] ;
parent . wsagents [ obj . dbNodeKey ] = obj ;
if ( dupAgent ) {
// Record duplicate agents
if ( parent . duplicateAgentsLog [ obj . dbNodeKey ] == null ) {
if ( dupAgent . remoteaddr == obj . remoteaddr ) {
parent . duplicateAgentsLog [ obj . dbNodeKey ] = { name : device . name , group : mesh . name , ip : [ obj . remoteaddr ] , count : 1 } ;
2019-05-21 17:19:32 -04:00
} else {
2019-06-05 18:24:07 -04:00
parent . duplicateAgentsLog [ obj . dbNodeKey ] = { name : device . name , group : mesh . name , ip : [ obj . remoteaddr , dupAgent . remoteaddr ] , count : 1 } ;
2019-05-21 17:19:32 -04:00
}
2018-08-21 18:08:15 -04:00
} else {
2019-06-05 18:24:07 -04:00
parent . duplicateAgentsLog [ obj . dbNodeKey ] . name = device . name ;
parent . duplicateAgentsLog [ obj . dbNodeKey ] . group = mesh . name ;
parent . duplicateAgentsLog [ obj . dbNodeKey ] . count ++ ;
if ( parent . duplicateAgentsLog [ obj . dbNodeKey ] . ip . indexOf ( obj . remoteaddr ) == - 1 ) { parent . duplicateAgentsLog [ obj . dbNodeKey ] . ip . push ( obj . remoteaddr ) ; }
2018-08-21 18:08:15 -04:00
}
2017-09-05 20:19:28 -04:00
2019-06-05 18:24:07 -04:00
// Close the duplicate agent
parent . agentStats . duplicateAgentCount ++ ;
2019-08-22 18:31:39 -04:00
if ( obj . nodeid != null ) { parent . parent . debug ( 'agent' , 'Duplicate agent ' + obj . nodeid + ' (' + obj . remoteaddrport + ')' ) ; }
2019-06-05 18:24:07 -04:00
dupAgent . close ( 3 ) ;
} else {
// Indicate the agent is connected
parent . parent . SetConnectivityState ( obj . dbMeshKey , obj . dbNodeKey , obj . connectTime , 1 , 1 ) ;
}
2018-08-21 18:08:15 -04:00
2019-06-05 18:24:07 -04:00
// We are done, ready to communicate with this agent
delete obj . pendingCompleteAgentConnection ;
obj . authenticated = 2 ;
2019-01-22 22:34:55 -05:00
2019-06-05 18:24:07 -04:00
// Check how many times this agent disconnected in the last few minutes.
const disconnectCount = parent . wsagentsDisconnections [ obj . nodeid ] ;
if ( disconnectCount > 6 ) {
2020-04-24 16:17:06 -04:00
parent . parent . debug ( 'agent' , 'Agent in big trouble: NodeId=' + obj . nodeid + ', IP=' + obj . remoteaddrport + ', Agent=' + obj . agentInfo . agentId + '.' ) ;
2019-06-05 18:24:07 -04:00
console . log ( 'Agent in big trouble: NodeId=' + obj . nodeid + ', IP=' + obj . remoteaddrport + ', Agent=' + obj . agentInfo . agentId + '.' ) ;
// TODO: Log or do something to recover?
return ;
}
2018-08-21 18:08:15 -04:00
2019-06-05 18:24:07 -04:00
// Command 4, inform mesh agent that it's authenticated.
2019-08-09 18:13:53 -04:00
obj . sendBinary ( common . ShortToStr ( 4 ) ) ;
2019-01-22 22:34:55 -05:00
2019-06-05 18:24:07 -04:00
if ( disconnectCount > 4 ) {
// Too many disconnections, this agent has issues. Just clear the core.
2019-08-09 18:13:53 -04:00
obj . sendBinary ( common . ShortToStr ( 10 ) + common . ShortToStr ( 0 ) ) ;
2020-04-24 16:17:06 -04:00
parent . parent . debug ( 'agent' , 'Agent in trouble: NodeId=' + obj . nodeid + ', IP=' + obj . remoteaddrport + ', Agent=' + obj . agentInfo . agentId + '.' ) ;
2019-06-05 18:24:07 -04:00
//console.log('Agent in trouble: NodeId=' + obj.nodeid + ', IP=' + obj.remoteaddrport + ', Agent=' + obj.agentInfo.agentId + '.');
// TODO: Log or do something to recover?
return ;
}
2019-03-11 14:18:50 -04:00
2019-06-05 18:24:07 -04:00
// Not sure why, but in rare cases, obj.agentInfo is undefined here.
if ( ( obj . agentInfo == null ) || ( typeof obj . agentInfo . capabilities != 'number' ) ) { return ; } // This is an odd case.
2019-04-09 18:51:05 -04:00
2019-06-05 18:24:07 -04:00
// Check if we need to make an native update check
obj . agentExeInfo = parent . parent . meshAgentBinaries [ obj . agentInfo . agentId ] ;
2019-06-17 18:20:47 -04:00
var corename = null ;
if ( parent . parent . meshAgentsArchitectureNumbers [ obj . agentInfo . agentId ] != null ) {
corename = parent . parent . meshAgentsArchitectureNumbers [ obj . agentInfo . agentId ] . core ;
} else {
// MeshCommand_CoreModule, ask mesh agent to clear the core
2019-08-09 18:13:53 -04:00
obj . sendBinary ( common . ShortToStr ( 10 ) + common . ShortToStr ( 0 ) ) ;
2019-06-17 18:20:47 -04:00
}
2019-06-05 18:24:07 -04:00
if ( ( obj . agentExeInfo != null ) && ( obj . agentExeInfo . update == true ) ) {
// Ask the agent for it's executable binary hash
2019-08-09 18:13:53 -04:00
obj . sendBinary ( common . ShortToStr ( 12 ) + common . ShortToStr ( 0 ) ) ;
2019-06-05 18:24:07 -04:00
} else {
// Check the mesh core, if the agent is capable of running one
if ( ( ( obj . agentInfo . capabilities & 16 ) != 0 ) && ( corename != null ) ) {
2019-08-09 18:13:53 -04:00
obj . sendBinary ( common . ShortToStr ( 11 ) + common . ShortToStr ( 0 ) ) ; // Command 11, ask for mesh core hash.
2019-01-21 17:05:50 -05:00
} else {
2019-06-05 18:24:07 -04:00
agentCoreIsStable ( ) ; // No updates needed, agent is ready to go.
2019-01-21 17:05:50 -05:00
}
2019-06-05 18:24:07 -04:00
}
2019-02-07 18:00:10 -05:00
}
2019-01-21 17:05:50 -05:00
2020-10-22 18:10:39 -04:00
// Indicate to the agent that we want to check Intel AMT configuration
// This may trigger a CIRA-LMS tunnel from the agent so the server can inspect the device.
2019-03-11 00:40:25 -04:00
obj . sendUpdatedIntelAmtPolicy = function ( policy ) {
if ( obj . agentExeInfo && ( obj . agentExeInfo . amt == true ) ) { // Only send Intel AMT policy to agents what could have AMT.
2020-10-21 18:36:07 -04:00
if ( policy == null ) { var mesh = parent . meshes [ obj . dbMeshKey ] ; if ( mesh == null ) return ; policy = mesh . amt ; }
2020-10-22 18:10:39 -04:00
if ( ( policy != null ) && ( policy . type != 0 ) ) {
const cookie = parent . parent . encodeCookie ( { a : 'apf' , n : obj . dbNodeKey , m : obj . dbMeshKey } , parent . parent . loginCookieEncryptionKey ) ;
try { obj . send ( JSON . stringify ( { action : 'amtconfig' , user : '**MeshAgentApfTunnel**' , pass : cookie } ) ) ; } catch ( ex ) { }
}
2019-03-07 22:56:24 -05:00
}
}
2019-04-09 18:51:05 -04:00
function recoveryAgentCoreIsStable ( mesh ) {
2019-05-01 18:02:03 -04:00
parent . agentStats . recoveryCoreIsStableCount ++ ;
2019-04-09 18:51:05 -04:00
// Recovery agent is doing ok, lets perform main agent checking.
//console.log('recoveryAgentCoreIsStable()');
// Fetch the the real agent nodeid
2020-04-07 00:18:50 -04:00
db . Get ( 'da' + obj . dbNodeKey , function ( err , nodes , self ) {
if ( ( nodes != null ) && ( nodes . length == 1 ) ) {
2019-04-12 14:28:57 -04:00
self . realNodeKey = nodes [ 0 ] . raid ;
2019-04-12 19:12:31 -04:00
// Get agent connection state
var agentConnected = false ;
var state = parent . parent . GetConnectivityState ( self . realNodeKey ) ;
if ( state ) { agentConnected = ( ( state . connectivity & 1 ) != 0 ) }
self . send ( JSON . stringify ( { action : 'diagnostic' , value : { command : 'query' , value : self . realNodeKey , agent : agentConnected } } ) ) ;
2020-04-07 00:18:50 -04:00
} else {
2019-04-12 14:28:57 -04:00
self . send ( JSON . stringify ( { action : 'diagnostic' , value : { command : 'query' , value : null } } ) ) ;
2019-04-09 18:51:05 -04:00
}
2019-04-12 14:28:57 -04:00
} , obj ) ;
2019-04-09 18:51:05 -04:00
}
2019-02-07 18:00:10 -05:00
function agentCoreIsStable ( ) {
2019-05-01 18:02:03 -04:00
parent . agentStats . coreIsStableCount ++ ;
2019-02-07 18:00:10 -05:00
// Check that the mesh exists
2019-03-08 01:47:27 -05:00
const mesh = parent . meshes [ obj . dbMeshKey ] ;
2019-02-07 18:00:10 -05:00
if ( mesh == null ) {
2019-05-01 18:02:03 -04:00
parent . agentStats . meshDoesNotExistCount ++ ;
2019-02-07 18:00:10 -05:00
// TODO: Mark this agent as part of a mesh that does not exists.
return ; // Probably not worth doing anything else. Hold this agent.
}
2019-04-10 13:41:10 -04:00
// Check if this is a recovery agent
if ( obj . agentInfo . capabilities & 0x40 ) {
recoveryAgentCoreIsStable ( mesh ) ;
return ;
}
2019-05-01 18:02:03 -04:00
// Fetch the the diagnostic agent nodeid
2019-04-09 18:51:05 -04:00
db . Get ( 'ra' + obj . dbNodeKey , function ( err , nodes ) {
2019-05-20 21:03:14 -04:00
if ( ( nodes != null ) && ( nodes . length == 1 ) ) {
2019-04-09 18:51:05 -04:00
obj . diagnosticNodeKey = nodes [ 0 ] . daid ;
obj . send ( JSON . stringify ( { action : 'diagnostic' , value : { command : 'query' , value : obj . diagnosticNodeKey } } ) ) ;
}
} ) ;
2020-10-21 18:36:07 -04:00
// Indicate that we want to check the Intel AMT configuration
2020-10-22 18:10:39 -04:00
// This may trigger a CIRA-LMS tunnel to the server for further processing
obj . sendUpdatedIntelAmtPolicy ( ) ;
2019-02-07 18:00:10 -05:00
2019-08-07 19:37:45 -04:00
// Fetch system information
2019-08-06 20:58:29 -04:00
db . GetHash ( 'si' + obj . dbNodeKey , function ( err , results ) {
if ( ( results != null ) && ( results . length == 1 ) ) { obj . send ( JSON . stringify ( { action : 'sysinfo' , hash : results [ 0 ] . hash } ) ) ; } else { obj . send ( JSON . stringify ( { action : 'sysinfo' } ) ) ; }
} ) ;
2020-07-06 19:01:44 -04:00
// Set agent core dump
2020-07-06 19:02:11 -04:00
if ( ( parent . parent . config . settings != null ) && ( ( parent . parent . config . settings . agentcoredump === true ) || ( parent . parent . config . settings . agentcoredump === false ) ) ) {
2020-07-06 19:01:44 -04:00
obj . send ( JSON . stringify ( { action : 'coredump' , value : parent . parent . config . settings . agentcoredump } ) ) ;
2020-07-08 14:59:20 -04:00
if ( parent . parent . config . settings . agentcoredump === true ) {
// Check if we requested a core dump file in the last minute, if not, ask if one is present.
if ( ( parent . lastCoreDumpRequest == null ) || ( ( Date . now ( ) - parent . lastCoreDumpRequest ) >= 60000 ) ) { obj . send ( JSON . stringify ( { action : 'getcoredump' } ) ) ; }
}
2020-07-06 19:01:44 -04:00
}
2019-02-07 18:00:10 -05:00
// Do this if IP location is enabled on this domain TODO: Set IP location per device group?
if ( domain . iplocation == true ) {
// Check if we already have IP location information for this node
2019-03-08 01:47:27 -05:00
db . Get ( 'iploc_' + obj . remoteaddr , function ( err , iplocs ) {
2019-05-20 21:03:14 -04:00
if ( ( iplocs != null ) && ( iplocs . length == 1 ) ) {
2019-02-07 18:00:10 -05:00
// We have a location in the database for this remote IP
2019-03-08 01:47:27 -05:00
const iploc = nodes [ 0 ] , x = { } ;
2019-02-07 18:00:10 -05:00
if ( ( iploc != null ) && ( iploc . ip != null ) && ( iploc . loc != null ) ) {
x . publicip = iploc . ip ;
x . iploc = iploc . loc + ',' + ( Math . floor ( ( new Date ( iploc . date ) ) / 1000 ) ) ;
ChangeAgentLocationInfo ( x ) ;
}
} else {
// Check if we need to ask for the IP location
var doIpLocation = 0 ;
if ( device . iploc == null ) {
doIpLocation = 1 ;
2017-09-05 20:19:28 -04:00
} else {
2019-03-08 01:47:27 -05:00
const loc = device . iploc . split ( ',' ) ;
2019-02-07 18:00:10 -05:00
if ( loc . length < 3 ) {
doIpLocation = 2 ;
2017-09-06 13:08:01 -04:00
} else {
2019-02-07 18:00:10 -05:00
var t = new Date ( ( parseFloat ( loc [ 2 ] ) * 1000 ) ) , now = Date . now ( ) ;
t . setDate ( t . getDate ( ) + 20 ) ;
if ( t < now ) { doIpLocation = 3 ; }
2017-09-06 13:08:01 -04:00
}
2019-02-07 18:00:10 -05:00
}
2017-09-05 20:19:28 -04:00
2019-02-07 18:00:10 -05:00
// If we need to ask for IP location, see if we have the quota to do it.
if ( doIpLocation > 0 ) {
2019-03-08 01:47:27 -05:00
db . getValueOfTheDay ( 'ipLocationRequestLimitor' , 10 , function ( ipLocationLimitor ) {
2019-05-20 21:03:14 -04:00
if ( ( ipLocationLimitor != null ) && ( ipLocationLimitor . value > 0 ) ) {
2019-02-07 18:00:10 -05:00
ipLocationLimitor . value -- ;
2019-03-08 01:47:27 -05:00
db . Set ( ipLocationLimitor ) ;
2019-02-07 18:00:10 -05:00
obj . send ( JSON . stringify ( { action : 'iplocation' } ) ) ;
}
} ) ;
2017-09-06 13:08:01 -04:00
}
2019-02-07 18:00:10 -05:00
}
} ) ;
}
2019-10-28 17:27:59 -04:00
if ( parent . parent . pluginHandler != null ) {
parent . parent . pluginHandler . callHook ( 'hook_agentCoreIsStable' , obj , parent ) ;
}
2017-08-28 12:27:45 -04:00
}
2018-01-02 19:52:49 -05:00
2018-11-30 19:42:58 -05:00
// Get the web certificate private key hash for the specified domain
2018-01-02 19:52:49 -05:00
function getWebCertHash ( domain ) {
2019-03-08 01:47:27 -05:00
const hash = parent . webCertificateHashs [ domain . id ] ;
2018-01-12 14:41:26 -05:00
if ( hash != null ) return hash ;
2019-03-08 01:47:27 -05:00
return parent . webCertificateHash ;
2018-01-02 19:52:49 -05:00
}
2018-11-30 19:42:58 -05:00
// Get the web certificate hash for the specified domain
function getWebCertFullHash ( domain ) {
2019-03-08 01:47:27 -05:00
const hash = parent . webCertificateFullHashs [ domain . id ] ;
2018-11-30 19:42:58 -05:00
if ( hash != null ) return hash ;
2019-03-08 01:47:27 -05:00
return parent . webCertificateFullHash ;
2018-11-30 19:42:58 -05:00
}
2017-08-28 12:27:45 -04:00
// Verify the agent signature
function processAgentSignature ( msg ) {
2020-12-31 14:16:47 -05:00
if ( isIgnoreHashCheck ( ) == false ) {
2019-01-11 13:02:54 -05:00
var verified = false ;
if ( msg . length != 384 ) {
// Verify a PKCS7 signature.
var msgDer = null ;
2019-03-08 01:47:27 -05:00
try { msgDer = forge . asn1 . fromDer ( forge . util . createBuffer ( msg , 'binary' ) ) ; } catch ( ex ) { }
2019-01-11 13:02:54 -05:00
if ( msgDer != null ) {
try {
2020-05-22 15:54:22 -04:00
const p7 = forge . pkcs7 . messageFromAsn1 ( msgDer ) ;
const sig = p7 . rawCapture . signature ;
2019-01-11 13:02:54 -05:00
// Verify with key hash
2019-03-08 01:47:27 -05:00
var buf = Buffer . from ( getWebCertHash ( domain ) + obj . nonce + obj . agentnonce , 'binary' ) ;
var verifier = parent . crypto . createVerify ( 'RSA-SHA384' ) ;
2019-01-11 13:02:54 -05:00
verifier . update ( buf ) ;
verified = verifier . verify ( obj . unauth . nodeCertPem , sig , 'binary' ) ;
2020-05-22 15:54:22 -04:00
if ( verified !== true ) {
2019-01-11 13:02:54 -05:00
// Verify with full hash
2019-03-08 01:47:27 -05:00
buf = Buffer . from ( getWebCertFullHash ( domain ) + obj . nonce + obj . agentnonce , 'binary' ) ;
verifier = parent . crypto . createVerify ( 'RSA-SHA384' ) ;
2019-01-11 13:02:54 -05:00
verifier . update ( buf ) ;
verified = verifier . verify ( obj . unauth . nodeCertPem , sig , 'binary' ) ;
}
2020-05-22 15:54:22 -04:00
if ( verified !== true ) {
// Verify with default key hash
buf = Buffer . from ( parent . defaultWebCertificateHash + obj . nonce + obj . agentnonce , 'binary' ) ;
verifier = parent . crypto . createVerify ( 'RSA-SHA384' ) ;
verifier . update ( buf ) ;
verified = verifier . verify ( obj . unauth . nodeCertPem , sig , 'binary' ) ;
}
if ( verified !== true ) {
// Verify with default full hash
buf = Buffer . from ( parent . defaultWebCertificateFullHash + obj . nonce + obj . agentnonce , 'binary' ) ;
verifier = parent . crypto . createVerify ( 'RSA-SHA384' ) ;
verifier . update ( buf ) ;
verified = verifier . verify ( obj . unauth . nodeCertPem , sig , 'binary' ) ;
}
if ( verified !== true ) {
2019-05-01 18:02:03 -04:00
// Not a valid signature
parent . agentStats . invalidPkcsSignatureCount ++ ;
return false ;
2020-04-07 00:18:50 -04:00
}
2019-01-11 13:02:54 -05:00
} catch ( ex ) { } ;
}
}
if ( verified == false ) {
// Verify the RSA signature. This is the fast way, without using forge.
2019-03-08 01:47:27 -05:00
const verify = parent . crypto . createVerify ( 'SHA384' ) ;
verify . end ( Buffer . from ( getWebCertHash ( domain ) + obj . nonce + obj . agentnonce , 'binary' ) ) ; // Test using the private key hash
2019-01-11 13:02:54 -05:00
if ( verify . verify ( obj . unauth . nodeCertPem , Buffer . from ( msg , 'binary' ) ) !== true ) {
2019-03-08 01:47:27 -05:00
const verify2 = parent . crypto . createVerify ( 'SHA384' ) ;
verify2 . end ( Buffer . from ( getWebCertFullHash ( domain ) + obj . nonce + obj . agentnonce , 'binary' ) ) ; // Test using the full cert hash
2019-05-01 18:02:03 -04:00
if ( verify2 . verify ( obj . unauth . nodeCertPem , Buffer . from ( msg , 'binary' ) ) !== true ) {
parent . agentStats . invalidRsaSignatureCount ++ ;
return false ;
}
2019-01-11 13:02:54 -05:00
}
2018-12-15 15:34:55 -05:00
}
2018-11-30 19:42:58 -05:00
}
2017-08-28 12:27:45 -04:00
// Connection is a success, clean up
2017-10-15 20:36:06 -04:00
obj . nodeid = obj . unauth . nodeid ;
2017-08-28 12:27:45 -04:00
obj . dbNodeKey = 'node/' + domain . id + '/' + obj . nodeid ;
delete obj . nonce ;
delete obj . agentnonce ;
delete obj . unauth ;
2019-03-08 01:47:27 -05:00
delete obj . receivedCommands ;
2017-08-28 12:27:45 -04:00
if ( obj . unauthsign ) delete obj . unauthsign ;
2019-05-01 18:02:03 -04:00
parent . agentStats . verifiedAgentConnectionCount ++ ;
2019-08-22 18:31:39 -04:00
parent . parent . debug ( 'agent' , 'Verified agent connection to ' + obj . nodeid + ' (' + obj . remoteaddrport + ').' ) ;
2017-08-28 12:27:45 -04:00
obj . authenticated = 1 ;
return true ;
}
// Process incoming agent JSON data
function processAgentData ( msg ) {
2019-03-08 01:47:27 -05:00
var i , str = msg . toString ( 'utf8' ) , command = null ;
2017-08-28 12:27:45 -04:00
if ( str [ 0 ] == '{' ) {
2020-04-24 16:17:06 -04:00
try { command = JSON . parse ( str ) ; } catch ( ex ) {
// If the command can't be parsed, ignore it.
parent . agentStats . invalidJsonCount ++ ;
parent . parent . debug ( 'agent' , 'Unable to parse agent JSON (' + obj . remoteaddrport + ')' ) ;
console . log ( 'Unable to parse agent JSON (' + obj . remoteaddrport + '): ' + str , ex ) ;
return ;
}
2018-08-27 15:24:15 -04:00
if ( typeof command != 'object' ) { return ; }
2017-08-28 12:27:45 -04:00
switch ( command . action ) {
case 'msg' :
{
2019-10-09 18:56:27 -04:00
// Route a message
parent . routeAgentCommand ( command , obj . domain . id , obj . dbNodeKey , obj . dbMeshKey ) ;
2017-08-28 12:27:45 -04:00
break ;
}
case 'coreinfo' :
{
// Sent by the agent to update agent information
ChangeAgentCoreInfo ( command ) ;
2018-09-27 19:17:05 -04:00
break ;
}
case 'smbios' :
{
2020-09-08 14:58:23 -04:00
// SMBIOS information must never be saved when NeDB is in use. NeDB will currupt that database.
if ( db . SetSMBIOS == null ) break ;
2020-09-07 23:37:58 -04:00
2020-04-24 16:37:37 -04:00
// See if we need to save SMBIOS information
2020-09-07 23:35:13 -04:00
if ( domain . smbios === true ) {
2020-04-24 16:37:37 -04:00
// Store the RAW SMBios table of this computer
// Perform sanity checks before storing
try {
for ( var i in command . value ) { var k = parseInt ( i ) ; if ( ( k != i ) || ( i > 255 ) || ( typeof command . value [ i ] != 'object' ) || ( command . value [ i ] . length == null ) || ( command . value [ i ] . length > 1024 ) || ( command . value [ i ] . length < 0 ) ) { delete command . value [ i ] ; } }
db . SetSMBIOS ( { _id : obj . dbNodeKey , domain : domain . id , time : new Date ( ) , value : command . value } ) ;
} catch ( ex ) { }
}
2018-09-27 19:17:05 -04:00
// Event the node interface information change (This is a lot of traffic, probably don't need this).
2020-03-17 20:17:04 -04:00
//parent.parent.DispatchEvent(parent.CreateMeshDispatchTargets(obj.meshid, [obj.dbNodeKey]), obj, { action: 'smBiosChange', nodeid: obj.dbNodeKey, domain: domain.id, smbios: command.value, nolog: 1 });
2018-09-27 19:17:05 -04:00
2017-08-28 12:27:45 -04:00
break ;
}
case 'netinfo' :
{
2020-06-26 17:43:58 -04:00
// Check if network information is present
if ( ( command . netif2 == null ) && ( command . netif == null ) ) return ;
2020-12-29 02:14:15 -05:00
// Escape any field names that have special characters
if ( command . netif2 != null ) {
for ( var i in command . netif2 ) {
var esc = common . escapeFieldName ( i ) ;
if ( esc !== i ) { command . netif2 [ esc ] = command . netif2 [ i ] ; delete command . netif2 [ i ] ; }
}
}
2017-08-28 12:27:45 -04:00
// Sent by the agent to update agent network interface information
delete command . action ;
command . updateTime = Date . now ( ) ;
command . _id = 'if' + obj . dbNodeKey ;
command . type = 'ifinfo' ;
2019-03-08 01:47:27 -05:00
db . Set ( command ) ;
2017-08-28 12:27:45 -04:00
// Event the node interface information change
2020-03-17 20:17:04 -04:00
parent . parent . DispatchEvent ( parent . CreateMeshDispatchTargets ( obj . meshid , [ obj . dbNodeKey ] ) , obj , { action : 'ifchange' , nodeid : obj . dbNodeKey , domain : domain . id , nolog : 1 } ) ;
2017-08-28 12:27:45 -04:00
2017-08-28 15:48:53 -04:00
break ;
}
2017-09-05 20:19:28 -04:00
case 'iplocation' :
2017-08-28 15:48:53 -04:00
{
// Sent by the agent to update location information
2017-08-30 16:52:56 -04:00
if ( ( command . type == 'publicip' ) && ( command . value != null ) && ( typeof command . value == 'object' ) && ( command . value . ip ) && ( command . value . loc ) ) {
2017-08-28 18:06:29 -04:00
var x = { } ;
x . publicip = command . value . ip ;
2018-08-29 20:40:30 -04:00
x . iploc = command . value . loc + ',' + ( Math . floor ( Date . now ( ) / 1000 ) ) ;
2017-08-28 18:06:29 -04:00
ChangeAgentLocationInfo ( x ) ;
2017-09-06 13:08:01 -04:00
command . value . _id = 'iploc_' + command . value . ip ;
command . value . type = 'iploc' ;
command . value . date = Date . now ( ) ;
2019-03-08 01:47:27 -05:00
db . Set ( command . value ) ; // Store the IP to location data in the database
2017-09-06 13:08:01 -04:00
// Sample Value: { ip: '192.55.64.246', city: 'Hillsboro', region: 'Oregon', country: 'US', loc: '45.4443,-122.9663', org: 'AS4983 Intel Corporation', postal: '97123' }
2017-08-28 18:06:29 -04:00
}
2017-08-28 12:27:45 -04:00
break ;
}
2017-11-07 20:05:22 -05:00
case 'mc1migration' :
{
if ( command . oldnodeid . length != 64 ) break ;
2019-03-08 01:47:27 -05:00
const oldNodeKey = 'node//' + command . oldnodeid . toLowerCase ( ) ;
db . Get ( oldNodeKey , function ( err , nodes ) {
2019-05-20 21:03:14 -04:00
if ( ( nodes != null ) && ( nodes . length != 1 ) ) return ;
2019-03-08 01:47:27 -05:00
const node = nodes [ 0 ] ;
2017-11-07 20:05:22 -05:00
if ( node . meshid == obj . dbMeshKey ) {
// Update the device name & host
2019-03-08 01:47:27 -05:00
const newNode = { "name" : node . name } ;
2017-11-09 20:18:30 -05:00
if ( node . intelamt != null ) { newNode . intelamt = node . intelamt ; }
ChangeAgentCoreInfo ( newNode ) ;
2017-11-07 20:05:22 -05:00
// Delete this node including network interface information and events
2019-03-08 01:47:27 -05:00
db . Remove ( node . _id ) ;
db . Remove ( 'if' + node . _id ) ;
2017-11-07 20:05:22 -05:00
// Event node deletion
2019-03-08 01:47:27 -05:00
const change = 'Migrated device ' + node . name ;
2020-03-17 20:17:04 -04:00
parent . parent . DispatchEvent ( parent . CreateMeshDispatchTargets ( node . meshid , [ obj . dbNodeKey ] ) , obj , { etype : 'node' , action : 'removenode' , nodeid : node . _id , msg : change , domain : node . domain } ) ;
2017-11-07 20:05:22 -05:00
}
} ) ;
break ;
}
2019-01-24 15:08:48 -05:00
case 'openUrl' :
{
// Sent by the agent to return the status of a open URL action.
// Nothing is done right now.
break ;
}
2019-07-29 19:35:48 -04:00
case 'log' :
{
// Log a value in the event log
if ( ( typeof command . msg == 'string' ) && ( command . msg . length < 4096 ) ) {
var event = { etype : 'node' , action : 'agentlog' , nodeid : obj . dbNodeKey , domain : domain . id , msg : command . msg } ;
2020-09-08 21:59:58 -04:00
if ( typeof command . msgid == 'number' ) { event . msgid = command . msgid ; }
if ( Array . isArray ( command . msgArgs ) ) { event . msgArgs = command . msgArgs ; }
if ( typeof command . remoteaddr == 'string' ) { event . remoteaddr = command . remoteaddr ; }
2020-03-17 20:17:04 -04:00
var targets = parent . CreateMeshDispatchTargets ( obj . dbMeshKey , [ obj . dbNodeKey ] ) ;
2019-07-29 19:35:48 -04:00
if ( typeof command . userid == 'string' ) {
var loguser = parent . users [ command . userid ] ;
2019-09-18 15:05:33 -04:00
if ( loguser ) { event . userid = command . userid ; event . username = loguser . name ; targets . push ( command . userid ) ; }
2019-07-29 19:35:48 -04:00
}
if ( ( typeof command . sessionid == 'string' ) && ( command . sessionid . length < 500 ) ) { event . sessionid = command . sessionid ; }
2019-09-18 15:05:33 -04:00
parent . parent . DispatchEvent ( targets , obj , event ) ;
2019-07-29 19:35:48 -04:00
}
break ;
}
2019-05-28 13:58:46 -04:00
case 'ping' : { sendPong ( ) ; break ; }
case 'pong' : { break ; }
2019-02-07 23:06:01 -05:00
case 'getScript' :
{
// Used by the agent to get configuration scripts.
if ( command . type == 1 ) {
2019-03-08 01:47:27 -05:00
parent . getCiraConfigurationScript ( obj . dbMeshKey , function ( script ) {
2019-02-07 23:06:01 -05:00
obj . send ( JSON . stringify ( { action : 'getScript' , type : 1 , script : script . toString ( ) } ) ) ;
} ) ;
} else if ( command . type == 2 ) {
2019-03-08 01:47:27 -05:00
parent . getCiraCleanupScript ( function ( script ) {
2019-02-07 23:06:01 -05:00
obj . send ( JSON . stringify ( { action : 'getScript' , type : 2 , script : script . toString ( ) } ) ) ;
} ) ;
}
break ;
}
2019-04-09 18:51:05 -04:00
case 'diagnostic' :
{
if ( typeof command . value == 'object' ) {
switch ( command . value . command ) {
case 'register' : {
// Only main agent can do this
2020-04-07 00:18:50 -04:00
if ( ( ( obj . agentInfo . capabilities & 0x40 ) == 0 ) && ( typeof command . value . value == 'string' ) && ( command . value . value . length == 64 ) ) {
2019-04-09 18:51:05 -04:00
// Store links to diagnostic agent id
2019-04-12 14:28:57 -04:00
var daNodeKey = 'node/' + domain . id + '/' + db . escapeBase64 ( command . value . value ) ;
2019-04-09 18:51:05 -04:00
db . Set ( { _id : 'da' + daNodeKey , domain : domain . id , time : obj . connectTime , raid : obj . dbNodeKey } ) ; // DiagnosticAgent --> Agent
db . Set ( { _id : 'ra' + obj . dbNodeKey , domain : domain . id , time : obj . connectTime , daid : daNodeKey } ) ; // Agent --> DiagnosticAgent
}
break ;
}
case 'query' : {
// Only the diagnostic agent can do
if ( ( obj . agentInfo . capabilities & 0x40 ) != 0 ) {
// Return nodeid of main agent + connection status
db . Get ( 'da' + obj . dbNodeKey , function ( err , nodes ) {
if ( nodes . length == 1 ) {
obj . realNodeKey = nodes [ 0 ] . raid ;
2019-04-12 19:12:31 -04:00
// Get agent connection state
var agentConnected = false ;
var state = parent . parent . GetConnectivityState ( obj . realNodeKey ) ;
if ( state ) { agentConnected = ( ( state . connectivity & 1 ) != 0 ) }
obj . send ( JSON . stringify ( { action : 'diagnostic' , value : { command : 'query' , value : obj . realNodeKey , agent : agentConnected } } ) ) ;
2019-04-09 18:51:05 -04:00
} else {
obj . send ( JSON . stringify ( { action : 'diagnostic' , value : { command : 'query' , value : null } } ) ) ;
}
} ) ;
}
break ;
}
case 'log' : {
2019-07-29 19:35:48 -04:00
if ( ( ( obj . agentInfo . capabilities & 0x40 ) != 0 ) && ( typeof command . value . value == 'string' ) && ( command . value . value . length < 256 ) ) {
// If this is a diagnostic agent, log the event in the log of the main agent
2020-04-07 00:18:50 -04:00
var event = { etype : 'node' , action : 'diagnostic' , nodeid : obj . realNodeKey , snodeid : obj . dbNodeKey , domain : domain . id , msg : command . value . value } ;
2020-03-17 20:17:04 -04:00
parent . parent . DispatchEvent ( parent . CreateMeshDispatchTargets ( obj . dbMeshKey , [ obj . dbNodeKey ] ) , obj , event ) ;
2019-04-09 18:51:05 -04:00
}
break ;
}
}
}
break ;
}
2019-08-06 20:58:29 -04:00
case 'sysinfo' : {
2019-09-24 13:34:33 -04:00
if ( ( typeof command . data == 'object' ) && ( typeof command . data . hash == 'string' ) ) {
command . data . _id = 'si' + obj . dbNodeKey ;
command . data . type = 'sysinfo' ;
command . data . domain = domain . id ;
command . data . time = Date . now ( ) ;
db . Set ( command . data ) ; // Update system information in the database.
// Event the new sysinfo hash, this will notify everyone that the sysinfo document was changed
var event = { etype : 'node' , action : 'sysinfohash' , nodeid : obj . dbNodeKey , domain : domain . id , hash : command . data . hash , nolog : 1 } ;
2020-03-17 20:17:04 -04:00
parent . parent . DispatchEvent ( parent . CreateMeshDispatchTargets ( obj . dbMeshKey , [ obj . dbNodeKey ] ) , obj , event ) ;
2019-09-24 13:34:33 -04:00
}
2019-08-06 20:58:29 -04:00
break ;
}
case 'sysinfocheck' : {
// Check system information update
db . GetHash ( 'si' + obj . dbNodeKey , function ( err , results ) {
if ( ( results != null ) && ( results . length == 1 ) ) { obj . send ( JSON . stringify ( { action : 'sysinfo' , hash : results [ 0 ] . hash } ) ) ; } else { obj . send ( JSON . stringify ( { action : 'sysinfo' } ) ) ; }
} ) ;
break ;
}
2020-05-11 21:44:46 -04:00
case 'sessions' : {
// This is a list of sessions provided by the agent
if ( obj . sessions == null ) { obj . sessions = { } ; }
2020-10-01 13:30:26 -04:00
if ( typeof command . value != null ) {
if ( command . type == 'kvm' ) { obj . sessions . kvm = command . value ; }
else if ( command . type == 'terminal' ) { obj . sessions . terminal = command . value ; }
else if ( command . type == 'files' ) { obj . sessions . files = command . value ; }
2020-10-01 18:45:44 -04:00
else if ( command . type == 'help' ) { obj . sessions . help = command . value ; }
2020-10-01 13:30:26 -04:00
else if ( command . type == 'tcp' ) { obj . sessions . tcp = command . value ; }
else if ( command . type == 'udp' ) { obj . sessions . udp = command . value ; }
else if ( command . type == 'msg' ) { obj . sessions . msg = command . value ; }
else if ( command . type == 'app' ) { obj . sessions . app = command . value ; }
}
2020-11-28 00:40:30 -05:00
// Any "help" session must have an associated app, if not, remove it.
if ( obj . sessions . help != null ) {
for ( var i in obj . sessions . help ) { if ( obj . sessions . app [ i ] == null ) { delete obj . sessions . help [ i ] ; } }
if ( Object . keys ( obj . sessions . help ) . length == 0 ) { delete obj . sessions . help ; }
}
// Inform everyone of updated sessions
2020-05-11 21:44:46 -04:00
obj . updateSessions ( ) ;
break ;
}
2020-05-30 20:21:45 -04:00
case 'battery' : {
// Device battery and power state
if ( obj . sessions == null ) { obj . sessions = { } ; }
if ( obj . sessions . battery == null ) { obj . sessions . battery = { } ; }
if ( ( command . state == 'ac' ) || ( command . state == 'dc' ) ) { obj . sessions . battery . state = command . state ; } else { delete obj . sessions . battery . state ; }
if ( ( typeof command . level == 'number' ) && ( command . level >= 0 ) && ( command . level <= 100 ) ) { obj . sessions . battery . level = command . level ; } else { delete obj . sessions . battery . level ; }
obj . updateSessions ( ) ;
break ;
}
2020-07-06 20:16:54 -04:00
case 'getcoredump' : {
2020-07-08 14:59:20 -04:00
// Check if we requested a core dump file in the last minute, if so, ignore this.
if ( ( parent . lastCoreDumpRequest != null ) && ( ( Date . now ( ) - parent . lastCoreDumpRequest ) < 60000 ) ) break ;
2020-07-06 20:16:54 -04:00
// Indicates if the agent has a coredump available
2020-07-08 14:59:20 -04:00
if ( ( command . exists === true ) && ( typeof command . agenthashhex == 'string' ) && ( command . agenthashhex . length == 96 ) ) {
// Check if we already have this exact dump file
2020-08-21 14:47:34 -04:00
const coreDumpFile = parent . path . join ( parent . parent . datapath , '..' , 'meshcentral-coredumps' , obj . agentInfo . agentId + '-' + command . agenthashhex + '-' + obj . nodeid + '.dmp' ) ;
2020-07-08 14:59:20 -04:00
parent . fs . stat ( coreDumpFile , function ( err , stats ) {
if ( stats != null ) return ;
obj . coreDumpPresent = true ;
// Check how many files are in the coredumps folder
2020-08-21 14:47:34 -04:00
const coreDumpPath = parent . path . join ( parent . parent . datapath , '..' , 'meshcentral-coredumps' ) ;
2020-07-08 14:59:20 -04:00
parent . fs . readdir ( coreDumpPath , function ( err , files ) {
if ( ( files != null ) && ( files . length >= 20 ) ) return ; // Don't get more than 20 core dump files.
// Get the core dump uploaded to the server.
parent . lastCoreDumpRequest = Date . now ( ) ;
2020-10-27 19:40:12 -04:00
obj . RequestCoreDump ( command . agenthashhex , command . corehashhex ) ;
2020-07-08 14:59:20 -04:00
} ) ;
} ) ;
2020-07-06 20:16:54 -04:00
}
break ;
}
2020-07-17 22:55:49 -04:00
case 'tunnelCloseStats' : {
// TODO: This this extra stats from the tunnel, you can merge this into the tunnel event in the database.
//console.log(command);
2020-07-21 19:20:17 -04:00
// Event the session closed compression data.
2020-09-08 21:59:58 -04:00
var event = { etype : 'node' , action : 'sessioncompression' , nodeid : obj . dbNodeKey , domain : domain . id , sent : command . sent , sentActual : command . sentActual , msgid : 54 , msgArgs : [ command . sentRatio , command . sent , command . sentActual ] , msg : 'Agent closed session with ' + command . sentRatio + '% agent to server compression. Sent: ' + command . sent + ', Compressed: ' + command . sentActual + '.' } ;
2020-07-21 19:20:17 -04:00
parent . parent . DispatchEvent ( parent . CreateMeshDispatchTargets ( obj . dbMeshKey , [ obj . dbNodeKey ] ) , obj , event ) ;
2020-07-17 22:55:49 -04:00
break ;
}
2020-08-12 16:35:36 -04:00
case 'lmsinfo' : {
// Agents send the LMS port bindings
// Example: {"action":"lmsinfo","value":{"ports":["623","16992"]}}
break ;
}
2019-10-08 04:18:40 -04:00
case 'plugin' : {
2019-10-28 12:21:12 -04:00
if ( ( parent . parent . pluginHandler == null ) || ( typeof command . plugin != 'string' ) ) break ;
2020-04-07 00:18:50 -04:00
try {
2019-11-01 16:49:18 -04:00
parent . parent . pluginHandler . plugins [ command . plugin ] . serveraction ( command , obj , parent ) ;
2019-10-10 14:13:25 -04:00
} catch ( e ) {
2020-04-24 16:17:06 -04:00
parent . parent . debug ( 'agent' , 'Error loading plugin handler (' + e + ')' ) ;
2019-10-10 14:13:25 -04:00
console . log ( 'Error loading plugin handler (' + e + ')' ) ;
2019-10-08 04:18:40 -04:00
}
break ;
}
2020-11-03 21:44:43 -05:00
case 'meshToolInfo' : {
if ( typeof command . name != 'string' ) break ;
var info = parent . parent . meshToolsBinaries [ command . name ] ;
if ( ( command . hash != null ) && ( info . hash == command . hash ) ) return ;
const responseCmd = { action : 'meshToolInfo' , name : command . name , hash : info . hash , size : info . size , url : info . url } ;
if ( command . cookie === true ) { responseCmd . url += ( '&auth=' + parent . parent . encodeCookie ( { download : info . dlname } , parent . parent . loginCookieEncryptionKey ) ) ; }
if ( command . pipe === true ) { responseCmd . pipe = true ; }
2020-11-04 00:32:12 -05:00
if ( parent . webCertificateHashs [ domain . id ] != null ) { responseCmd . serverhash = Buffer . from ( parent . webCertificateHashs [ domain . id ] , 'binary' ) . toString ( 'hex' ) ; }
2020-11-03 21:44:43 -05:00
try { ws . send ( JSON . stringify ( responseCmd ) ) ; } catch ( ex ) { }
break ;
}
2019-01-04 20:59:13 -05:00
default : {
2020-04-07 00:18:50 -04:00
parent . agentStats . unknownAgentActionCount ++ ;
2020-08-12 16:35:36 -04:00
parent . parent . debug ( 'agent' , 'Unknown agent action (' + obj . remoteaddrport + '): ' + JSON . stringify ( command ) + '.' ) ;
console . log ( 'Unknown agent action (' + obj . remoteaddrport + '): ' + JSON . stringify ( command ) + '.' ) ;
2019-01-04 20:59:13 -05:00
break ;
}
2017-08-28 12:27:45 -04:00
}
2019-10-28 17:27:59 -04:00
if ( parent . parent . pluginHandler != null ) {
parent . parent . pluginHandler . callHook ( 'hook_processAgentData' , command , obj , parent ) ;
}
2017-08-28 12:27:45 -04:00
}
}
2020-05-11 21:44:46 -04:00
// Notify update of sessions
obj . updateSessions = function ( ) {
// Perform some clean up
for ( var i in obj . sessions ) { if ( Object . keys ( obj . sessions [ i ] ) . length == 0 ) { delete obj . sessions [ i ] ; } }
if ( Object . keys ( obj . sessions ) . length == 0 ) { delete obj . sessions ; }
// Event the new sessions, this will notify everyone that agent sessions have changed
var event = { etype : 'node' , action : 'devicesessions' , nodeid : obj . dbNodeKey , domain : domain . id , sessions : obj . sessions , nolog : 1 } ;
parent . parent . DispatchEvent ( parent . CreateMeshDispatchTargets ( obj . dbMeshKey , [ obj . dbNodeKey ] ) , obj , event ) ;
}
2017-08-28 12:27:45 -04:00
// Change the current core information string and event it
function ChangeAgentCoreInfo ( command ) {
2020-10-08 22:47:24 -04:00
if ( ( obj . agentInfo == null ) || ( obj . agentInfo . capabilities & 0x40 ) ) return ;
2017-09-15 14:45:06 -04:00
if ( ( command == null ) || ( command == null ) ) return ; // Safety, should never happen.
2017-08-28 12:27:45 -04:00
2020-09-23 18:34:46 -04:00
// If the device is pending a change, hold.
if ( obj . deviceChanging === true ) { setTimeout ( function ( ) { ChangeAgentCoreInfo ( command ) ; } , 100 ) ; return ; }
obj . deviceChanging = true ;
2017-08-28 12:27:45 -04:00
// Check that the mesh exists
2019-03-08 01:47:27 -05:00
const mesh = parent . meshes [ obj . dbMeshKey ] ;
2020-09-23 18:34:46 -04:00
if ( mesh == null ) { delete obj . deviceChanging ; return ; }
2018-08-21 18:08:15 -04:00
// Get the node and change it if needed
2019-03-08 01:47:27 -05:00
db . Get ( obj . dbNodeKey , function ( err , nodes ) { // TODO: THIS IS A BIG RACE CONDITION HERE, WE NEED TO FIX THAT. If this call is made twice at the same time on the same device, data will be missed.
2020-09-23 18:34:46 -04:00
if ( ( nodes == null ) || ( nodes . length != 1 ) ) { delete obj . deviceChanging ; return ; }
2019-03-08 01:47:27 -05:00
const device = nodes [ 0 ] ;
2018-08-21 18:08:15 -04:00
if ( device . agent ) {
2019-01-30 16:43:42 -05:00
var changes = [ ] , change = 0 , log = 0 ;
2018-08-21 18:08:15 -04:00
// Check if anything changes
2019-10-01 17:54:38 -04:00
if ( command . name && ( typeof command . name == 'string' ) && ( command . name != device . name ) ) { change = 1 ; log = 1 ; device . name = command . name ; changes . push ( 'name' ) ; }
2019-01-30 16:43:42 -05:00
if ( ( command . caps != null ) && ( device . agent . core != command . value ) ) { if ( ( command . value == null ) && ( device . agent . core != null ) ) { delete device . agent . core ; } else { device . agent . core = command . value ; } change = 1 ; } // Don't save this as an event to the db.
if ( ( command . caps != null ) && ( ( device . agent . caps & 0xFFFFFFE7 ) != ( command . caps & 0xFFFFFFE7 ) ) ) { device . agent . caps = ( ( device . agent . caps & 24 ) + ( command . caps & 0xFFFFFFE7 ) ) ; change = 1 ; } // Allow Javascript on the agent to change all capabilities except console and javascript support, Don't save this as an event to the db.
2019-10-01 17:54:38 -04:00
if ( ( command . osdesc != null ) && ( typeof command . osdesc == 'string' ) && ( device . osdesc != command . osdesc ) ) { device . osdesc = command . osdesc ; change = 1 ; changes . push ( 'os desc' ) ; } // Don't save this as an event to the db.
2020-10-16 17:46:41 -04:00
if ( ( typeof command . root == 'boolean' ) && ( command . root !== device . agent . root ) ) { change = 1 ; device . agent . root = command . root ; }
2019-03-15 15:30:04 -04:00
if ( device . ip != obj . remoteaddr ) { device . ip = obj . remoteaddr ; change = 1 ; }
2018-08-21 18:08:15 -04:00
if ( command . intelamt ) {
if ( ! device . intelamt ) { device . intelamt = { } ; }
2020-11-02 03:44:07 -05:00
if ( ( command . intelamt . Versions != null ) && ( typeof command . intelamt . Versions == 'object' ) ) {
if ( ( command . intelamt . Versions . AMT != null ) && ( typeof command . intelamt . Versions . AMT == 'string' ) && ( command . intelamt . Versions . AMT . length < 12 ) && ( device . intelamt . ver != command . intelamt . Versions . AMT ) ) { changes . push ( 'AMT version' ) ; device . intelamt . ver = command . intelamt . Versions . AMT ; change = 1 ; log = 1 ; }
if ( ( command . intelamt . Versions . Sku != null ) && ( typeof command . intelamt . Versions . Sku == 'string' ) ) { var sku = parseInt ( command . intelamt . Versions . Sku ) ; if ( device . intelamt . sku !== command . intelamt . sku ) { device . intelamt . sku = sku ; change = 1 ; log = 1 ; } }
}
if ( ( command . intelamt . ProvisioningState != null ) && ( typeof command . intelamt . ProvisioningState == 'number' ) && ( device . intelamt . state != command . intelamt . ProvisioningState ) ) { changes . push ( 'AMT state' ) ; device . intelamt . state = command . intelamt . ProvisioningState ; change = 1 ; log = 1 ; }
if ( ( command . intelamt . Flags != null ) && ( typeof command . intelamt . Flags == 'number' ) && ( device . intelamt . flags != command . intelamt . Flags ) ) {
if ( device . intelamt . flags ) { changes . push ( 'AMT flags (' + device . intelamt . flags + ' --> ' + command . intelamt . Flags + ')' ) ; } else { changes . push ( 'AMT flags (' + command . intelamt . Flags + ')' ) ; }
device . intelamt . flags = command . intelamt . Flags ; change = 1 ; log = 1 ;
2018-09-21 19:34:35 -04:00
}
2020-11-02 03:44:07 -05:00
if ( ( command . intelamt . UUID != null ) && ( typeof command . intelamt . UUID == 'string' ) && ( device . intelamt . uuid != command . intelamt . UUID ) ) { changes . push ( 'AMT uuid' ) ; device . intelamt . uuid = command . intelamt . UUID ; change = 1 ; log = 1 ; }
2018-08-21 18:08:15 -04:00
}
2019-08-02 14:35:59 -04:00
if ( command . av ) {
if ( ! device . av ) { device . av = [ ] ; }
2019-10-25 17:51:22 -04:00
if ( ( command . av != null ) && ( JSON . stringify ( device . av ) != JSON . stringify ( command . av ) ) ) { /*changes.push('AV status');*/ device . av = command . av ; change = 1 ; log = 1 ; }
2019-08-02 14:35:59 -04:00
}
2020-11-02 12:47:02 -05:00
if ( ( command . users != null ) && ( Array . isArray ( command . users ) ) && ( device . users != command . users ) ) { device . users = command . users ; change = 1 ; } // Don't save this to the db.
2019-03-08 01:47:27 -05:00
if ( ( mesh . mtype == 2 ) && ( ! args . wanonly ) ) {
2019-02-20 18:26:27 -05:00
// In WAN mode, the hostname of a computer is not important. Don't log hostname changes.
if ( device . host != obj . remoteaddr ) { device . host = obj . remoteaddr ; change = 1 ; changes . push ( 'host' ) ; }
2018-08-21 18:08:15 -04:00
// TODO: Check that the agent has an interface that is the same as the one we got this websocket connection on. Only set if we have a match.
}
2017-08-28 12:27:45 -04:00
2018-09-21 19:34:35 -04:00
// If there are changes, event the new device
2018-08-21 18:08:15 -04:00
if ( change == 1 ) {
2019-04-10 13:41:10 -04:00
// Do some clean up if needed, these values should not be in the database.
if ( device . conn != null ) { delete device . conn ; }
if ( device . pwr != null ) { delete device . pwr ; }
if ( device . agct != null ) { delete device . agct ; }
if ( device . cict != null ) { delete device . cict ; }
2018-09-21 19:34:35 -04:00
// Save to the database
2019-03-08 01:47:27 -05:00
db . Set ( device ) ;
2017-08-28 12:27:45 -04:00
2018-08-21 18:08:15 -04:00
// Event the node change
2019-10-11 14:16:36 -04:00
var event = { etype : 'node' , action : 'changenode' , nodeid : obj . dbNodeKey , domain : domain . id , node : parent . CloneSafeNode ( device ) } ;
2018-09-24 20:47:03 -04:00
if ( changes . length > 0 ) { event . msg = 'Changed device ' + device . name + ' from group ' + mesh . name + ': ' + changes . join ( ', ' ) ; }
2019-07-15 12:48:40 -04:00
if ( ( log == 0 ) || ( ( obj . agentInfo ) && ( obj . agentInfo . capabilities ) && ( obj . agentInfo . capabilities & 0x20 ) ) || ( changes . length == 0 ) ) { event . nolog = 1 ; } // If this is a temporary device, don't log changes
2019-05-28 20:25:23 -04:00
if ( db . changeStream ) { event . noact = 1 ; } // If DB change stream is active, don't use this event to change the node. Another event will come.
2020-03-17 20:17:04 -04:00
parent . parent . DispatchEvent ( parent . CreateMeshDispatchTargets ( device . meshid , [ obj . dbNodeKey ] ) , obj , event ) ;
2017-08-28 12:27:45 -04:00
}
2020-09-23 18:34:46 -04:00
// Device change is done.
delete obj . deviceChanging ;
2018-08-21 18:08:15 -04:00
}
2017-08-28 12:27:45 -04:00
} ) ;
}
2017-08-28 18:06:29 -04:00
// Change the current core information string and event it
function ChangeAgentLocationInfo ( command ) {
2019-04-09 18:51:05 -04:00
if ( obj . agentInfo . capabilities & 0x40 ) return ;
2018-08-21 18:08:15 -04:00
if ( ( command == null ) || ( command == null ) ) { return ; } // Safety, should never happen.
2017-08-28 18:06:29 -04:00
// Check that the mesh exists
2019-03-08 01:47:27 -05:00
const mesh = parent . meshes [ obj . dbMeshKey ] ;
2018-08-21 18:08:15 -04:00
if ( mesh == null ) return ;
2020-09-23 18:34:46 -04:00
// If the device is pending a change, hold.
if ( obj . deviceChanging === true ) { setTimeout ( function ( ) { ChangeAgentLocationInfo ( command ) ; } , 100 ) ; return ; }
obj . deviceChanging = true ;
2018-08-21 18:08:15 -04:00
// Get the node and change it if needed
2019-03-08 01:47:27 -05:00
db . Get ( obj . dbNodeKey , function ( err , nodes ) {
2020-09-23 18:34:46 -04:00
if ( ( nodes == null ) || ( nodes . length != 1 ) ) { delete obj . deviceChanging ; return ; }
2019-03-08 01:47:27 -05:00
const device = nodes [ 0 ] ;
2018-08-21 18:08:15 -04:00
if ( device . agent ) {
var changes = [ ] , change = 0 ;
// Check if anything changes
if ( ( command . publicip ) && ( device . publicip != command . publicip ) ) { device . publicip = command . publicip ; change = 1 ; changes . push ( 'public ip' ) ; }
if ( ( command . iploc ) && ( device . iploc != command . iploc ) ) { device . iploc = command . iploc ; change = 1 ; changes . push ( 'ip location' ) ; }
// If there are changes, save and event
if ( change == 1 ) {
2019-04-10 13:41:10 -04:00
// Do some clean up if needed, these values should not be in the database.
if ( device . conn != null ) { delete device . conn ; }
if ( device . pwr != null ) { delete device . pwr ; }
if ( device . agct != null ) { delete device . agct ; }
if ( device . cict != null ) { delete device . cict ; }
// Save the device
2019-03-08 01:47:27 -05:00
db . Set ( device ) ;
2017-08-28 18:06:29 -04:00
2018-08-21 18:08:15 -04:00
// Event the node change
2020-09-08 21:59:58 -04:00
var event = { etype : 'node' , action : 'changenode' , nodeid : obj . dbNodeKey , domain : domain . id , node : parent . CloneSafeNode ( device ) , msgid : 59 , msgArgs : [ device . name , mesh . name , changes . join ( ', ' ) ] , msg : 'Changed device ' + device . name + ' from group ' + mesh . name + ': ' + changes . join ( ', ' ) } ;
2018-08-21 18:08:15 -04:00
if ( obj . agentInfo . capabilities & 0x20 ) { event . nolog = 1 ; } // If this is a temporary device, don't log changes
2019-05-28 20:25:23 -04:00
if ( db . changeStream ) { event . noact = 1 ; } // If DB change stream is active, don't use this event to change the node. Another event will come.
2020-03-17 20:17:04 -04:00
parent . parent . DispatchEvent ( parent . CreateMeshDispatchTargets ( device . meshid , [ obj . dbNodeKey ] ) , obj , event ) ;
2017-08-28 18:06:29 -04:00
}
2018-08-21 18:08:15 -04:00
}
2020-09-23 18:34:46 -04:00
// Done changing the device
delete obj . deviceChanging ;
2017-08-28 18:06:29 -04:00
} ) ;
}
2017-08-28 12:27:45 -04:00
// Update the mesh agent tab in the database
function ChangeAgentTag ( tag ) {
2019-04-09 18:51:05 -04:00
if ( obj . agentInfo . capabilities & 0x40 ) return ;
2020-09-23 18:34:46 -04:00
if ( ( tag != null ) && ( tag . length == 0 ) ) { tag = null ; }
// If the device is pending a change, hold.
if ( obj . deviceChanging === true ) { setTimeout ( function ( ) { ChangeAgentCoreInfo ( command ) ; } , 100 ) ; return ; }
obj . deviceChanging = true ;
2017-08-28 12:27:45 -04:00
// Get the node and change it if needed
2019-03-08 01:47:27 -05:00
db . Get ( obj . dbNodeKey , function ( err , nodes ) {
2020-09-23 18:34:46 -04:00
if ( ( nodes == null ) || ( nodes . length != 1 ) ) { delete obj . deviceChanging ; return ; }
2019-03-08 01:47:27 -05:00
const device = nodes [ 0 ] ;
2017-08-28 12:27:45 -04:00
if ( device . agent ) {
2020-09-23 18:34:46 -04:00
// Parse the agent tag
var agentTag = null , serverName = null , serverDesc = null , serverTags = null ;
if ( tag != null ) {
var taglines = tag . split ( '\r\n' ) . join ( '\n' ) . split ( '\r' ) . join ( '\n' ) . split ( '\n' ) ;
for ( var i in taglines ) {
var tagline = taglines [ i ] . trim ( ) ;
if ( tagline . length > 0 ) {
if ( tagline . startsWith ( '~' ) ) {
if ( tagline . startsWith ( '~ServerName:' ) && ( tagline . length > 12 ) && ( serverName == null ) ) { serverName = tagline . substring ( 12 ) . trim ( ) ; }
if ( tagline . startsWith ( '~ServerDesc:' ) && ( tagline . length > 12 ) && ( serverDesc == null ) ) { serverDesc = tagline . substring ( 12 ) . trim ( ) ; }
if ( tagline . startsWith ( '~ServerTags:' ) && ( tagline . length > 12 ) && ( serverTags == null ) ) { serverTags = tagline . substring ( 12 ) . split ( ',' ) ; for ( var j in serverTags ) { serverTags [ j ] = serverTags [ j ] . trim ( ) ; } }
} else { if ( agentTag == null ) { agentTag = tagline ; } }
}
}
}
// Set the agent tag
var changes = false ;
if ( device . agent . tag != agentTag ) { device . agent . tag = agentTag ; if ( ( device . agent . tag == null ) || ( device . agent . tag == '' ) ) { delete device . agent . tag ; } changes = true ; }
if ( domain . agenttag != null ) {
// Set the device's server name
if ( ( serverName != null ) && ( domain . agenttag . servername === 1 ) && ( device . name != serverName ) ) { device . name = serverName ; changes = true ; }
// Set the device's server description
if ( ( serverDesc != null ) && ( domain . agenttag . serverdesc === 1 ) && ( device . desc != serverDesc ) ) { device . desc = serverDesc ; changes = true ; }
// Set the device's server description if there is no description
if ( ( serverDesc != null ) && ( domain . agenttag . serverdesc === 2 ) && ( device . desc != serverDesc ) && ( ( device . desc == null ) || ( device . desc == '' ) ) ) { device . desc = serverDesc ; changes = true ; }
if ( ( serverTags != null ) && ( domain . agenttag . servertags != null ) && ( domain . agenttag . servertags != 0 ) ) {
// Sort the tags
serverTags . sort ( ) ;
// Stringify the tags
var st2 = '' , st1 = serverTags . join ( ',' ) ;
if ( device . tags != null ) { st2 = device . tags . join ( ',' ) ; }
// Set the device's server tags
if ( ( domain . agenttag . servertags === 1 ) && ( st1 != st2 ) ) { device . tags = serverTags ; changes = true ; }
// Set the device's server tags if there are not tags
if ( ( domain . agenttag . servertags === 2 ) && ( st2 == '' ) ) { device . tags = serverTags ; changes = true ; }
// Append to device's server tags
if ( ( domain . agenttag . servertags === 3 ) && ( st1 != st2 ) ) {
if ( device . tags == null ) { device . tags = [ ] ; }
for ( var i in serverTags ) { if ( device . tags . indexOf ( serverTags [ i ] ) == - 1 ) { device . tags . push ( serverTags [ i ] ) ; } }
device . tags . sort ( ) ;
changes = true ;
}
}
}
if ( changes == true ) {
2019-04-10 13:41:10 -04:00
// Do some clean up if needed, these values should not be in the database.
if ( device . conn != null ) { delete device . conn ; }
2020-04-07 00:18:50 -04:00
if ( device . pwr != null ) { delete device . pwr ; }
2019-04-10 13:41:10 -04:00
if ( device . agct != null ) { delete device . agct ; }
if ( device . cict != null ) { delete device . cict ; }
2020-09-23 18:34:46 -04:00
// Update the device
2019-03-08 01:47:27 -05:00
db . Set ( device ) ;
2017-08-28 12:27:45 -04:00
// Event the node change
2019-10-11 14:16:36 -04:00
var event = { etype : 'node' , action : 'changenode' , nodeid : obj . dbNodeKey , domain : domain . id , node : parent . CloneSafeNode ( device ) , nolog : 1 } ;
2019-05-28 20:25:23 -04:00
if ( db . changeStream ) { event . noact = 1 ; } // If DB change stream is active, don't use this event to change the node. Another event will come.
2020-03-17 20:17:04 -04:00
parent . parent . DispatchEvent ( parent . CreateMeshDispatchTargets ( device . meshid , [ obj . dbNodeKey ] ) , obj , event ) ;
2017-08-28 12:27:45 -04:00
}
}
2020-09-23 18:34:46 -04:00
// Done changing the device
delete obj . deviceChanging ;
2017-08-28 12:27:45 -04:00
} ) ;
}
2020-12-08 01:55:29 -05:00
// Check if we need to update this agent, return true if agent binary update required.
function compareAgentBinaryHash ( agentExeInfo , agentHash ) {
// If the hash matches or is null, no update required.
if ( ( agentExeInfo . hash == agentHash ) || ( agentHash == '\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0' ) ) return false ;
// If this is a macOS x86 or ARM agent type and it matched the universal binary, no update required.
if ( ( ( agentExeInfo . id == 16 ) || ( agentExeInfo . id == 29 ) ) && ( parent . parent . meshAgentBinaries [ 10005 ] . hash == agentHash ) ) return false ;
// No match, update the agent.
return true ;
}
2020-07-08 02:56:08 -04:00
// Request that the core dump file on this agent be uploaded to the server
2020-10-27 19:40:12 -04:00
obj . RequestCoreDump = function ( agenthashhex , corehashhex ) {
2020-07-08 02:56:08 -04:00
if ( agenthashhex . length > 16 ) { agenthashhex = agenthashhex . substring ( 0 , 16 ) ; }
const cookie = parent . parent . encodeCookie ( { a : 'aft' , b : 'coredump' , c : obj . agentInfo . agentId + '-' + agenthashhex + '-' + obj . nodeid + '.dmp' } , parent . parent . loginCookieEncryptionKey ) ;
obj . send ( '{"action":"msg","type":"tunnel","value":"*/' + ( ( ( domain . dns == null ) && ( domain . id != '' ) ) ? ( domain . id + '/' ) : '' ) + 'agenttransfer.ashx?c=' + cookie + '","rights":"4294967295"}' ) ;
}
2020-12-30 19:54:02 -05:00
// Return true if we need to ignore the agent hash check
function isIgnoreHashCheck ( ) {
if ( ( args . ignoreagenthashcheck === true ) || ( domain . ignoreagenthashcheck === true ) ) return true ;
// Check site wide exceptions
if ( Array . isArray ( args . ignoreagenthashcheck ) ) {
for ( var i = 0 ; i < args . ignoreagenthashcheck . length ; i ++ ) {
if ( require ( 'ipcheck' ) . match ( obj . remoteaddr , args . ignoreagenthashcheck [ i ] ) ) return true ;
}
}
// Check domain wide exceptions
if ( Array . isArray ( domain . ignoreagenthashcheck ) ) {
for ( var i = 0 ; i < domain . ignoreagenthashcheck . length ; i ++ ) {
if ( require ( 'ipcheck' ) . match ( obj . remoteaddr , domain . ignoreagenthashcheck [ i ] ) ) return true ;
}
}
return false ;
}
2019-06-20 17:27:57 -04:00
// Generate a random Intel AMT password
function checkAmtPassword ( p ) { return ( p . length > 7 ) && ( /\d/ . test ( p ) ) && ( /[a-z]/ . test ( p ) ) && ( /[A-Z]/ . test ( p ) ) && ( /\W/ . test ( p ) ) ; }
function getRandomAmtPassword ( ) { var p ; do { p = Buffer . from ( parent . crypto . randomBytes ( 9 ) , 'binary' ) . toString ( 'base64' ) . split ( '/' ) . join ( '@' ) ; } while ( checkAmtPassword ( p ) == false ) ; return p ; }
2017-08-28 12:27:45 -04:00
return obj ;
2018-08-29 20:40:30 -04:00
} ;