Fixed MeshCMD AMT Script Exception

agents/modules_meshcmd/amt-script.js

@@ -122,7 +122,7 @@ module.exports.setup = function(binary, startvars) {
                 if (argtyp < 2) {
                     // Get the value and replace all {var} with variable values
                     argval = argval.toString();
-                    while (argval.split("{").length > 1) { var t = argval.split("{").pop().split("}").shift(); argval = argval.replace('{' + t + '}', obj.getVar(t)); }
+                    if (argval != null) { while (argval.split("{").length > 1) { var t = argval.split("{").pop().split("}").shift(); argval = argval.replace('{' + t + '}', obj.getVar(t)); } }
                     if (argtyp == 1) { obj.variables['__' + i] = decodeURI(argval); argval = '__' + i; } // If argtyp is 1, this is a literal. Store in temp variable.
                     args.push(argval);
                 }

meshagent.js

@@ -46,6 +46,8 @@ module.exports.CreateMeshAgent = function (parent, db, ws, req, args, domain) {
         if (arg == 2) { try { ws._socket._parent.end(); if (obj.nodeid != null) { parent.parent.debug(1, 'Hard disconnect ' + obj.nodeid + ' (' + obj.remoteaddrport + ')'); } } catch (e) { console.log(e); } } // Hard close, close the TCP socket
         // If arg == 3, don't communicate with this agent anymore, but don't disconnect (Duplicate agent).
 
+        // If this is a recovery agent, don't bother with all this clean up.
+        if ((obj.agentInfo.capabilities & 0x40) == 0) {
             // Remove this agent from the webserver list
             if (parent.wsagents[obj.dbNodeKey] == obj) {
                 delete parent.wsagents[obj.dbNodeKey];
@@ -56,7 +58,7 @@ module.exports.CreateMeshAgent = function (parent, db, ws, req, args, domain) {
             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))) {
+            if (((obj.agentInfo) && (obj.agentInfo.capabilities) && (obj.agentInfo.capabilities & 0x20)) || ((mesh) && (mesh.flags) && (mesh.flags & 1))) {
                 // Delete this node including network interface information and events
                 db.Remove(obj.dbNodeKey);                       // Remove node with that id
                 db.Remove('if' + obj.dbNodeKey);                // Remove interface information
@@ -79,6 +81,7 @@ module.exports.CreateMeshAgent = function (parent, db, ws, req, args, domain) {
                 // Update the last connect time
                 if (obj.authenticated == 2) { db.Set({ _id: 'lc' + obj.dbNodeKey, type: 'lastconnect', domain: domain.id, time: obj.connectTime, addr: obj.remoteaddrport }); }
             }
+        }
 
         // If we where updating the agent, clean that up.
         if (obj.agentUpdate != null) {
@@ -522,6 +525,12 @@ module.exports.CreateMeshAgent = function (parent, db, ws, req, args, domain) {
         if ((obj.authenticated != 1) || (obj.meshid == null) || obj.pendingCompleteAgentConnection || (obj.agentInfo == null)) { return; }
         obj.pendingCompleteAgentConnection = true;
 
+        if (obj.agentInfo.capabilities & 0x40) {
+            // This is a recovery agent
+            obj.send(common.ShortToStr(11) + common.ShortToStr(0)); // Command 11, ask for mesh core hash.
+            return;
+        }
+
         // Check if we have too many agent sessions
         if (typeof domain.limits.maxagentsessions == 'number') {
             // Count the number of agent sessions for this domain
@@ -700,10 +709,6 @@ module.exports.CreateMeshAgent = function (parent, db, ws, req, args, domain) {
             // 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.
 
-            if ((obj.agentInfo.capabilities & 64) != 0) {
-                // This is a recovery agent
-                obj.send(common.ShortToStr(11) + common.ShortToStr(0)); // Command 11, ask for mesh core hash.
-            } else {
             // Check if we need to make an native update check
             obj.agentExeInfo = parent.parent.meshAgentBinaries[obj.agentInfo.agentId];
             const corename = parent.parent.meshAgentsArchitectureNumbers[obj.agentInfo.agentId].core;
@@ -720,20 +725,9 @@ module.exports.CreateMeshAgent = function (parent, db, ws, req, args, domain) {
                     agentCoreIsStable(); // No updates needed, agent is ready to go.
                 }
             }
-            }
         });
     }
 
-    function recoveryAgentCoreIsStable() {
-        // Recovery agent is doing ok, lets perform main agent checking.
-
-        // TODO
-        console.log('recoveryAgentCoreIsStable()');
-
-        // Close the recovery agent connection when done.
-        //obj.close(1);
-    }
-
     // Take a basic Intel AMT policy and add all server information to it, making it ready to send to this agent.
     function completeIntelAmtPolicy(amtPolicy) {
         if (amtPolicy == null) return null;
@@ -764,6 +758,21 @@ module.exports.CreateMeshAgent = function (parent, db, ws, req, args, domain) {
         }
     }
 
+    function recoveryAgentCoreIsStable(mesh) {
+        // Recovery agent is doing ok, lets perform main agent checking.
+        //console.log('recoveryAgentCoreIsStable()');
+
+        // Fetch the the real agent nodeid
+        db.Get('da' + obj.dbNodeKey, function (err, nodes) {
+            if (nodes.length == 1) {
+                obj.realNodeKey = nodes[0].raid;
+                obj.send(JSON.stringify({ action: 'diagnostic', value: { command: 'query', value: obj.realNodeKey } }));
+            } else {
+                obj.send(JSON.stringify({ action: 'diagnostic', value: { command: 'query', value: null } }));
+            }
+        });
+    }
+
     function agentCoreIsStable() {
         // Check that the mesh exists
         const mesh = parent.meshes[obj.dbMeshKey];
@@ -772,6 +781,17 @@ module.exports.CreateMeshAgent = function (parent, db, ws, req, args, domain) {
             return; // Probably not worth doing anything else. Hold this agent.
         }
 
+        // Fetch the the real agent nodeid
+        db.Get('ra' + obj.dbNodeKey, function (err, nodes) {
+            if (nodes.length == 1) {
+                obj.diagnosticNodeKey = nodes[0].daid;
+                obj.send(JSON.stringify({ action: 'diagnostic', value: { command: 'query', value: obj.diagnosticNodeKey } }));
+            }
+        });
+
+        // Check if this is a recovery agent
+        if (obj.agentInfo.capabilities & 0x40) { recoveryAgentCoreIsStable(mesh); return; }
+
         // Send Intel AMT policy
         if (obj.agentExeInfo && (obj.agentExeInfo.amt == true) && (mesh.amt != null)) {  // Only send Intel AMT policy to agents what could have AMT.
             try { obj.send(JSON.stringify({ action: 'amtPolicy', amtPolicy: completeIntelAmtPolicy(common.Clone(mesh.amt)) })); } catch (ex) { }
@@ -1066,6 +1086,48 @@ module.exports.CreateMeshAgent = function (parent, db, ws, req, args, domain) {
                         }
                         break;
                     }
+                case 'diagnostic':
+                    {
+                        if (typeof command.value == 'object') {
+                            switch (command.value.command) {
+                                case 'register': {
+                                    // Only main agent can do this
+                                    if (((obj.agentInfo.capabilities & 0x40) == 0) && (typeof command.value.command.value == 'string') && (command.value.command.value.length == 64)) {
+                                        // Store links to diagnostic agent id
+                                        var daNodeKey = 'node/' + domain.id + '/' + command.value.command.value;
+                                        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;
+                                                obj.send(JSON.stringify({ action: 'diagnostic', value: { command: 'query', value: obj.realNodeKey } }));
+                                            } else {
+                                                obj.send(JSON.stringify({ action: 'diagnostic', value: { command: 'query', value: null } }));
+                                            }
+                                        });
+                                    }
+                                    break;
+                                }
+                                case 'log': {
+                                    // Only the diagnostic agent can do
+                                    if (((obj.agentInfo.capabilities & 0x40) != 0) && (typeof command.value.command.value == 'string') && (command.value.command.value.length < 256)) {
+                                        // Log a value in the event log of the main again
+                                        var event = { etype: 'node', action: 'diagnostic', nodeid: obj.dbNodeKey, domain: domain.id, msg: command.value.command.value };
+                                        parent.parent.DispatchEvent(['*', obj.dbMeshKey], obj, event);
+                                    }
+                                    break;
+                                }
+                            }
+                        }
+                        break;
+                    }
                 default: {
                     console.log('Unknown agent action (' + obj.remoteaddrport + '): ' + command.action + '.');
                     break;
@@ -1076,6 +1138,7 @@ module.exports.CreateMeshAgent = function (parent, db, ws, req, args, domain) {
 
     // Change the current core information string and event it
     function ChangeAgentCoreInfo(command) {
+        if (obj.agentInfo.capabilities & 0x40) return;
         if ((command == null) || (command == null)) return; // Safety, should never happen.
 
         // Check that the mesh exists
@@ -1137,6 +1200,7 @@ module.exports.CreateMeshAgent = function (parent, db, ws, req, args, domain) {
 
     // Change the current core information string and event it
     function ChangeAgentLocationInfo(command) {
+        if (obj.agentInfo.capabilities & 0x40) return;
         if ((command == null) || (command == null)) { return; } // Safety, should never happen.
 
         // Check that the mesh exists
@@ -1172,6 +1236,7 @@ module.exports.CreateMeshAgent = function (parent, db, ws, req, args, domain) {
 
     // Update the mesh agent tab in the database
     function ChangeAgentTag(tag) {
+        if (obj.agentInfo.capabilities & 0x40) return;
         if (tag.length == 0) { tag = null; }
         // Get the node and change it if needed
         db.Get(obj.dbNodeKey, function (err, nodes) {

meshuser.js

@@ -1396,6 +1396,10 @@ module.exports.CreateMeshUser = function (parent, db, ws, req, args, domain, use
                                 db.RemoveSMBIOS(node._id);                      // Remove SMBios data
                                 db.RemoveAllNodeEvents(node._id);               // Remove all events for this node
                                 db.removeAllPowerEventsForNode(node._id);       // Remove all power events for this node
+                                db.Get('ra' + obj.dbNodeKey, function (err, nodes) {
+                                    if (nodes.length == 1) { db.Remove('da' + nodes[0].daid); }     // Remove diagnostic agent to real agent link
+                                    db.Remove('ra' + node._id);                                     // Remove real agent to diagnostic agent link
+                                });
 
                                 // Event node deletion
                                 parent.parent.DispatchEvent(['*', node.meshid], obj, { etype: 'node', username: user.name, action: 'removenode', nodeid: node._id, msg: 'Removed device ' + node.name + ' from group ' + mesh.name, domain: domain.id });

public/styles/style.css

@@ -92,7 +92,7 @@
 }
 
 #page_content {
-    max-height: calc(100vh - 108px);
+    /*max-height: calc(100vh - 108px);*/
 }
     .fullscreen #page_content {
         position: absolute;

views/default.handlebars

@@ -1043,7 +1043,6 @@
             if (toggle === 1) { webPageFullScreen = !webPageFullScreen; putstore('webPageFullScreen', webPageFullScreen); }
             var hide = 0;
             if (args.hide) { hide = parseInt(args.hide); }
-            console.log("-----------hide:", hide)
             if (webPageFullScreen == false) {
                 QC('body').remove("fullscreen");
                 QC('body').remove("arg_hide");

views/login.handlebars

@@ -22,7 +22,7 @@
             <div id=topbar class="noselect style3">
                 <div id=toggle title="Toggle full width" onclick="toggleFullScreen(1)">&harr;</div>
             </div>
-            <div id=column_l>
+            <div id=column_l style="height:calc(100vh - 111px)">
                 <h1>Welcome</h1>
                 <div id="welcomeText" style="display:none">Connect to your home or office devices from anywhere in the world using <a href="http://www.meshcommander.com/meshcentral2">MeshCentral</a>, the real time, open source remote monitoring and management web site. You will need to download and install a management agent on your computers. Once installed, computers will show up in the &quot;My Devices&quot; section of this web site and you will be able to monitor them and take control of them.</div>
                 <table id="centralTable" style="">