123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478 |
- /**
- * @fileoverview Script Compiler / Decompiler / Runner
- * @author Ylian Saint-Hilaire
- * @copyright Intel Corporation 2018-2022
- * @license Apache-2.0
- * @version v0.1.0e
- */
- /*jslint node: true */
- /*jshint node: true */
- /*jshint strict:false */
- /*jshint -W097 */
- /*jshint esversion: 6 */
- "use strict";
- module.exports.CreateAmtScriptEngine = function () {
- var o = {};
- // Core functions
- const script_functionTable1 = ['nop', 'jump', 'set', 'print', 'dialog', 'getitem', 'substr', 'indexof', 'split', 'join', 'length', 'jsonparse', 'jsonstr', 'add', 'substract', 'parseint', 'wsbatchenum', 'wsput', 'wscreate', 'wsdelete', 'wsexec', 'scriptspeed', 'wssubscribe', 'wsunsubscribe', 'readchar', 'signwithdummyca'];
- // functions of type ARG1 = func(ARG2, ARG3, ARG4, ARG5, ARG6)
- const script_functionTable2 = ['encodeuri', 'decodeuri', 'passwordcheck', 'atob', 'btoa', 'hex2str', 'str2hex', 'random', 'md5', 'maketoarray', 'readshort', 'readshortx', 'readint', 'readsint', 'readintx', 'shorttostr', 'shorttostrx', 'inttostr', 'inttostrx'];
- // functions of type ARG1 = func(ARG2, ARG3, ARG4, ARG5, ARG6)
- //script_functionTableX2 = [encodeURI, decodeURI, passwordcheck, window.atob.bind(window), window.btoa.bind(window), hex2rstr, rstr2hex, random, rstr_md5, MakeToArray, ReadShort, ReadShortX, ReadInt, ReadSInt, ReadIntX, ShortToStr, ShortToStrX, IntToStr, IntToStrX];
- // Optional functions of type ARG1 = func(ARG2, ARG3, ARG4, ARG5, ARG6)
- const script_functionTable3 = ['pullsystemstatus', 'pulleventlog', 'pullauditlog', 'pullcertificates', 'pullwatchdog', 'pullsystemdefense', 'pullhardware', 'pulluserinfo', 'pullremoteaccess', 'highlightblock', 'disconnect', 'getsidstring', 'getsidbytearray'];
- /*
- // Optional functions of type ARG1 = func(ARG2, ARG3, ARG4, ARG5, ARG6)
- script_functionTableX3 = [
- PullSystemStatus
- ,
- // ###BEGIN###{EventLog}
- PullEventLog
- // ###END###{EventLog}
- ,
- // ###BEGIN###{AuditLog}
- PullAuditLog
- // ###END###{AuditLog}
- ,
- // ###BEGIN###{Certificates}
- PullCertificates
- // ###END###{Certificates}
- ,
- // ###BEGIN###{AgentPresence}
- PullWatchdog
- // ###END###{AgentPresence}
- ,
- // ###BEGIN###{SystemDefense}
- PullSystemDefense
- // ###END###{SystemDefense}
- ,
- // ###BEGIN###{HardwareInfo}
- PullHardware
- // ###END###{HardwareInfo}
- ,
- PullUserInfo
- ,
- // ###BEGIN###{RemoteAccess}
- PullRemoteAccess
- // ###END###{RemoteAccess}
- ,
- // ###BEGIN###{Scripting-Editor}
- script_HighlightBlock
- // ###END###{Scripting-Editor}
- ,
- // ###BEGIN###{ComputerSelector}
- disconnect
- // ###END###{ComputerSelector}
- ,
- function (runner, x) { return GetSidString(x); }
- ,
- function (runner, x) { return GetSidByteArray(x); }
- ];
- // Setup the script state
- o.script_setup = function(binary, startvars) {
- var obj = { startvars: startvars };
- if (binary.length < 6) { console.error('Invalid script length'); return null; } // Script must have at least 6 byte header
- if (ReadInt(binary, 0) != 0x247D2945) { console.error('Invalid binary script'); return null; } // Check the script magic header
- if (ReadShort(binary, 4) > 1) { console.error('Unsupported script version'); return null; } // Check the script version
- obj.script = binary.substring(6);
- // obj.onStep;
- // obj.onConsole;
- // Reset the script to the start
- obj.reset = function (stepspeed) {
- obj.stop();
- obj.ip = 0;
- obj.variables = startvars;
- obj.state = 1;
- }
- // Start the script
- obj.start = function (stepspeed) {
- obj.stop();
- obj.stepspeed = stepspeed;
- if (stepspeed > 0) { obj.timer = setInterval(function () { obj.step() }, stepspeed); }
- }
- // Stop the script
- obj.stop = function () {
- if (obj.timer != null) { clearInterval(obj.timer); }
- obj.timer = null;
- obj.stepspeed = 0;
- }
- // function used to load and store variable values
- obj.getVar = function (name) { if (name == undefined) return undefined; return obj.getVarEx(name.split('.'), obj.variables); }
- obj.getVarEx = function (name, val) { try { if (name == undefined) return undefined; if (name.length == 0) return val; return obj.getVarEx(name.slice(1), val[name[0]]); } catch (e) { return null; } }
- obj.setVar = function (name, val) { obj.setVarEx(name.split('.'), obj.variables, val); }
- obj.setVarEx = function (name, vars, val) { if (name.length == 1) { vars[name[0]] = val; } else { obj.setVarEx(name.slice(1), vars[name[0]], val); } }
- // Run the script one step forward
- obj.step = function () {
- if (obj.state != 1) return;
- if (obj.ip < obj.script.length) {
- var cmdid = ReadShort(obj.script, obj.ip);
- var cmdlen = ReadShort(obj.script, obj.ip + 2);
- var argcount = ReadShort(obj.script, obj.ip + 4);
- var argptr = obj.ip + 6;
- var args = [];
- // Clear all temp variables (This is optional)
- for (var i in obj.variables) { if (i.startsWith('__')) { delete obj.variables[i]; } }
- // Loop on each argument, moving forward by the argument length each time
- for (var i = 0; i < argcount; i++) {
- var arglen = ReadShort(obj.script, argptr);
- var argval = obj.script.substring(argptr + 2, argptr + 2 + arglen);
- var argtyp = argval.charCodeAt(0);
- argval = argval.substring(1);
- if (argtyp < 2) {
- // Get the value and replace all {var} with variable values
- 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);
- }
- if (argtyp == 2 || argtyp == 3) {
- obj.variables['__' + i] = ReadSInt(argval, 0);
- args.push('__' + i);
- }
- argptr += (2 + arglen);
- }
- // Move instruction pointer forward by command size
- obj.ip += cmdlen;
- // Get all variable values
- var argsval = [];
- for (var i = 0; i < 10; i++) { argsval.push(obj.getVar(args[i])); }
- var storeInArg0;
- try {
- if (cmdid < 10000) {
- // Lets run the actual command
- switch (cmdid) {
- case 0: // nop
- break;
- case 1: // jump(label) or jump(label, a, compare, b)
- if (argsval[2]) {
- if (
- (argsval[2] == '<' && argsval[1] < argsval[3]) ||
- (argsval[2] == '<=' && argsval[1] <= argsval[3]) ||
- (argsval[2] == '!=' && argsval[1] != argsval[3]) ||
- (argsval[2] == '=' && argsval[1] == argsval[3]) ||
- (argsval[2] == '>=' && argsval[1] >= argsval[3]) ||
- (argsval[2] == '>' && argsval[1] > argsval[3])
- ) { obj.ip = argsval[0]; }
- } else {
- obj.ip = argsval[0]; // Set the instruction pointer to the new location in the script
- }
- break;
- case 2: // set(variable, value)
- if (args[1] == undefined) delete obj.variables[args[0]]; else obj.setVar(args[0], argsval[1]);
- break;
- case 3: // print(message)
- if (obj.onConsole) { obj.onConsole(obj.toString(argsval[0]), obj); } else { console.log(obj.toString(argsval[0])); }
- // Q(obj.consoleid).value += () + '\n'); Q(obj.console).scrollTop = Q(obj.console).scrollHeight;
- break;
- case 4: // dialog(title, content, buttons)
- obj.state = 2;
- obj.dialog = true;
- setDialogMode(11, argsval[0], argsval[2], obj.xxStepDialogOk, argsval[1], obj);
- break;
- case 5: // getitem(a, b, c)
- for (var i in argsval[1]) { if (argsval[1][i][argsval[2]] == argsval[3]) { storeInArg0 = i; } };
- break;
- case 6: // substr(variable_dest, variable_src, index, len)
- storeInArg0 = argsval[1].substr(argsval[2], argsval[3]);
- break;
- case 7: // indexOf(variable_dest, variable_src, index, len)
- storeInArg0 = argsval[1].indexOf(argsval[2]);
- break;
- case 8: // split(variable_dest, variable_src, separator)
- storeInArg0 = argsval[1].split(argsval[2]);
- break;
- case 9: // join(variable_dest, variable_src, separator)
- storeInArg0 = argsval[1].join(argsval[2]);
- break;
- case 10: // length(variable_dest, variable_src)
- storeInArg0 = argsval[1].length;
- break;
- case 11: // jsonparse(variable_dest, json)
- storeInArg0 = JSON.parse(argsval[1]);
- break;
- case 12: // jsonstr(variable_dest, variable_src)
- storeInArg0 = JSON.stringify(argsval[1]);
- break;
- case 13: // add(variable_dest, variable_src, value)
- storeInArg0 = (argsval[1] + argsval[2]);
- break;
- case 14: // substract(variable_dest, variable_src, value)
- storeInArg0 = (argsval[1] - argsval[2]);
- break;
- case 15: // parseInt(variable_dest, variable_src)
- storeInArg0 = parseInt(argsval[1]);
- break;
- case 16: // wsbatchenum(name, objectList)
- obj.state = 2;
- obj.amtstack.BatchEnum(argsval[0], argsval[1], obj.xxWsmanReturn, obj);
- break;
- case 17: // wsput(name, args)
- obj.state = 2;
- obj.amtstack.Put(argsval[0], argsval[1], obj.xxWsmanReturn, obj);
- break;
- case 18: // wscreate(name, args)
- obj.state = 2;
- obj.amtstack.Create(argsval[0], argsval[1], obj.xxWsmanReturn, obj);
- break;
- case 19: // wsdelete(name, args)
- obj.state = 2;
- obj.amtstack.Delete(argsval[0], argsval[1], obj.xxWsmanReturn, obj);
- break;
- case 20: // wsexec(name, method, args, selectors)
- obj.state = 2;
- obj.amtstack.Exec(argsval[0], argsval[1], argsval[2], obj.xxWsmanReturn, obj, 0, argsval[3]);
- break;
- case 21: // Script Speed
- obj.stepspeed = argsval[0];
- if (obj.timer != null) { clearInterval(obj.timer); obj.timer = setInterval(function () { obj.step() }, obj.stepspeed); }
- break;
- case 22: // wssubscribe(name, delivery, url, selectors, opaque, user, pass)
- obj.state = 2;
- obj.amtstack.Subscribe(argsval[0], argsval[1], argsval[2], obj.xxWsmanReturn, obj, 0, argsval[3], argsval[4], argsval[5], argsval[6]);
- break;
- case 23: // wsunsubscribe(name, selectors)
- obj.state = 2;
- obj.amtstack.UnSubscribe(argsval[0], obj.xxWsmanReturn, obj, 0, argsval[1]);
- break;
- case 24: // readchar(str, pos)
- console.log(argsval[1], argsval[2], argsval[1].charCodeAt(argsval[2]));
- storeInArg0 = argsval[1].charCodeAt(argsval[2]);
- break;
- case 25: // signWithDummyCa
- // ###BEGIN###{Certificates}
- obj.state = 2;
- // DERKey, xxCaPrivateKey, certattributes, issuerattributes
- amtcert_signWithCaKey(argsval[0], null, argsval[1], { 'CN': 'Untrusted Root Certificate' }, obj.xxSignWithDummyCaReturn);
- // ###END###{Certificates}
- break;
- default: {
- obj.state = 9;
- console.error("Script Error, unknown command: " + cmdid);
- }
- }
- } else {
- if (cmdid < 20000) {
- // functions of type ARG1 = func(ARG2, ARG3, ARG4, ARG5, ARG6)
- storeInArg0 = script_functionTableX2[cmdid - 10000](argsval[1], argsval[2], argsval[3], argsval[4], argsval[5], argsval[6]);
- } else {
- // Optional functions of type ARG1 = func(ARG2, ARG3, ARG4, ARG5, ARG6)
- if (script_functionTableX3 && script_functionTableX3[cmdid - 20000]) {
- storeInArg0 = script_functionTableX3[cmdid - 20000](obj, argsval[1], argsval[2], argsval[3], argsval[4], argsval[5], argsval[6]); // Note that optional calls start with "obj" as first argument.
- }
- }
- }
- if (storeInArg0 != undefined) obj.setVar(args[0], storeInArg0);
- } catch (e) {
- if (typeof e == 'object') { e = e.message; }
- obj.setVar('_exception', e);
- }
- }
- if (obj.state == 1 && obj.ip >= obj.script.length) { obj.state = 0; obj.stop(); }
- if (obj.onStep) obj.onStep(obj);
- return obj;
- };
- obj.xxStepDialogOk = function (button) {
- obj.variables['DialogSelect'] = button;
- obj.state = 1;
- obj.dialog = false;
- if (obj.onStep) obj.onStep(obj);
- };
- // ###BEGIN###{**ClosureAdvancedMode}
- obj.xxWsmanReturnFix = function (x) {
- if (!x || x == null) return;
- if (x.Header) { x['Header'] = x.Header; delete x.Header; }
- if (x.Body) { x['Body'] = x.Body; delete x.Body; }
- if (x.Responses) { x['Responses'] = x.Responses; delete x.Responses; }
- if (x.Response) { x['Response'] = x.Response; delete x.Response; }
- if (x.ReturnValueStr) { x['ReturnValueStr'] = x.ReturnValueStr; delete x.ReturnValueStr; }
- };
- // ###END###{**ClosureAdvancedMode}
- obj.xxWsmanReturn = function (stack, name, responses, status) {
- // ###BEGIN###{**ClosureAdvancedMode}
- // This is required when Google Closure is used
- if (responses) {
- obj.xxWsmanReturnFix(responses);
- for (var i in responses) {
- obj.xxWsmanReturnFix(responses[i]);
- for (var j in responses[i]) { obj.xxWsmanReturnFix(responses[i][j]); }
- }
- }
- // ###END###{**ClosureAdvancedMode}
- obj.setVar(name, responses);
- obj.setVar('wsman_result', status);
- obj.setVar('wsman_result_str', ((httpErrorTable[status]) ? (httpErrorTable[status]) : ('Error #' + status)));
- obj.state = 1;
- if (obj.onStep) obj.onStep(obj);
- };
- // ###BEGIN###{Certificates}
- obj.xxSignWithDummyCaReturn = function (cert) {
- obj.setVar('signed_cert', btoa(_arrayBufferToString(cert)));
- obj.state = 1;
- if (obj.onStep) obj.onStep(obj);
- };
- // ###END###{Certificates}
- obj.toString = function (x) { if (typeof x == 'object') return JSON.stringify(x); return x; };
- obj.reset();
- return obj;
- }
- */
- var ReadShort = function (v, p) { return (v.charCodeAt(p) << 8) + v.charCodeAt(p + 1); };
- var ReadShortX = function (v, p) { return (v.charCodeAt(p + 1) << 8) + v.charCodeAt(p); };
- var ReadInt = function (v, p) { return (v.charCodeAt(p) * 0x1000000) + (v.charCodeAt(p + 1) << 16) + (v.charCodeAt(p + 2) << 8) + v.charCodeAt(p + 3); }; // We use "*0x1000000" instead of "<<24" because the shift converts the number to signed int32.
- var ReadIntX = function (v, p) { return (v.charCodeAt(p + 3) * 0x1000000) + (v.charCodeAt(p + 2) << 16) + (v.charCodeAt(p + 1) << 8) + v.charCodeAt(p); };
- var ShortToStr = function (v) { return String.fromCharCode((v >> 8) & 0xFF, v & 0xFF); };
- var ShortToStrX = function (v) { return String.fromCharCode(v & 0xFF, (v >> 8) & 0xFF); };
- var IntToStr = function (v) { return String.fromCharCode((v >> 24) & 0xFF, (v >> 16) & 0xFF, (v >> 8) & 0xFF, v & 0xFF); };
- var IntToStrX = function (v) { return String.fromCharCode(v & 0xFF, (v >> 8) & 0xFF, (v >> 16) & 0xFF, (v >> 24) & 0xFF); };
- // Argument types: 0 = Variable, 1 = String, 2 = Integer, 3 = Label
- o.script_compile = function (script, onmsg) {
- var r = '', scriptlines = script.split('\n'), labels = {}, labelswap = [], swaps = [];
- // Go thru each script line and encode it
- for (var i in scriptlines) {
- var scriptline = scriptlines[i];
- if (scriptline.startsWith('##SWAP ')) { var x = scriptline.split(' '); if (x.length == 3) { swaps[x[1]] = x[2]; } } // Add a swap instance
- if (scriptline[0] == '#' || scriptline.length == 0) continue; // Skip comments & blank lines
- for (var x in swaps) { scriptline = scriptline.split(x).join(swaps[x]); } // Apply all swaps
- var keywords = scriptline.match(/"[^"]*"|[^\s"]+/g);
- if ((keywords == null) || (keywords.length == 0)) continue; // Skip blank lines
- if (scriptline[0] == ':') { labels[keywords[0].toUpperCase()] = r.length; continue; } // Mark a label position
- var funcIndex = script_functionTable1.indexOf(keywords[0].toLowerCase());
- if (funcIndex == -1) { funcIndex = script_functionTable2.indexOf(keywords[0].toLowerCase()); if (funcIndex >= 0) funcIndex += 10000; }
- if (funcIndex == -1) { funcIndex = script_functionTable3.indexOf(keywords[0].toLowerCase()); if (funcIndex >= 0) funcIndex += 20000; } // Optional methods
- if (funcIndex == -1) { if (onmsg) { onmsg("Unabled to compile, unknown command: " + keywords[0]); } return ''; }
- // Encode CommandId, CmdSize, ArgCount, Arg1Len, Arg1, Arg2Len, Arg2...
- var cmd = ShortToStr(keywords.length - 1);
- for (var j in keywords) {
- if (j == 0) continue;
- if (keywords[j][0] == ':') {
- labelswap.push([keywords[j], r.length + cmd.length + 7]); // Add a label swap
- cmd += ShortToStr(5) + String.fromCharCode(3) + IntToStr(0xFFFFFFFF); // Put an empty label
- } else {
- var argint = parseInt(keywords[j]);
- if (argint == keywords[j]) {
- cmd += ShortToStr(5) + String.fromCharCode(2) + IntToStr(argint);
- } else {
- if (keywords[j][0] == '"' && keywords[j][keywords[j].length - 1] == '"') {
- cmd += ShortToStr(keywords[j].length - 1) + String.fromCharCode(1) + keywords[j].substring(1, keywords[j].length - 1);
- } else {
- cmd += ShortToStr(keywords[j].length + 1) + String.fromCharCode(0) + keywords[j];
- }
- }
- }
- }
- cmd = ShortToStr(funcIndex) + ShortToStr(cmd.length + 4) + cmd;
- r += cmd;
- }
- // Perform all the needed label swaps
- for (i in labelswap) {
- var label = labelswap[i][0].toUpperCase(), position = labelswap[i][1], target = labels[label];
- if (target == undefined) { if (onmsg) { onmsg("Unabled to compile, unknown label: " + label); } return ''; }
- r = r.substr(0, position) + IntToStr(target) + r.substr(position + 4);
- }
- return IntToStr(0x247D2945) + ShortToStr(1) + r;
- };
- // Decompile the script, intended for debugging only
- o.script_decompile = function (binary, onecmd) {
- var r = '', ptr = 6, labels = {};
- if (onecmd >= 0) {
- ptr = onecmd; // If we are decompiling just one command, set the ptr to that command.
- } else {
- if (binary.length < 6) { return '# Invalid script length'; }
- var magic = ReadInt(binary, 0);
- var version = ReadShort(binary, 4);
- if (magic != 0x247D2945) { return '# Invalid binary script: ' + magic; }
- if (version != 1) { return '# Invalid script version'; }
- }
- // Loop on each command, moving forward by the command length each time.
- while (ptr < binary.length) {
- var cmdid = ReadShort(binary, ptr);
- var cmdlen = ReadShort(binary, ptr + 2);
- var argcount = ReadShort(binary, ptr + 4);
- var argptr = ptr + 6;
- var argstr = '';
- if (!(onecmd >= 0)) { r += ":label" + (ptr - 6) + "\n"; }
- // Loop on each argument, moving forward by the argument length each time
- for (var i = 0; i < argcount; i++) {
- var arglen = ReadShort(binary, argptr);
- var argval = binary.substring(argptr + 2, argptr + 2 + arglen);
- var argtyp = argval.charCodeAt(0);
- if (argtyp == 0) { argstr += ' ' + argval.substring(1); } // Variable
- else if (argtyp == 1) { argstr += ' \"' + argval.substring(1) + '\"'; } // String
- else if (argtyp == 2) { argstr += ' ' + ReadInt(argval, 1); } // Integer
- else if (argtyp == 3) { // Label
- var target = ReadInt(argval, 1);
- var label = labels[target];
- if (!label) { label = ":label" + target; labels[label] = target; }
- argstr += ' ' + label;
- }
- argptr += (2 + arglen);
- }
- // Go in the script function table to decode the function
- if (cmdid < 10000) {
- r += script_functionTable1[cmdid] + argstr + "\n";
- } else {
- if (cmdid >= 20000) {
- r += script_functionTable3[cmdid - 20000] + argstr + "\n"; // Optional methods
- } else {
- r += script_functionTable2[cmdid - 10000] + argstr + "\n";
- }
- }
- ptr += cmdlen;
- if (onecmd >= 0) return r; // If we are decompiling just one command, exit now
- }
- // Remove all unused labels
- var scriptlines = r.split('\n');
- r = '';
- for (var i in scriptlines) {
- var line = scriptlines[i];
- if (line[0] != ':') { r += line + '\n'; } else { if (labels[line]) { r += line + '\n'; } }
- }
- return r;
- };
- // Convert the list of blocks into a script that can be compiled
- o.script_blocksToScript = function (script_BuildingBlocks, script_BlockScript) {
- var script = '';
- if (script_BuildingBlocks) {
- if (script_BuildingBlocks['_start']) { script += '##### Starting Block #####\r\n' + script_BuildingBlocks['_start']['code'] + '\r\n\r\n'; }
- for (var i in script_BlockScript) {
- var code = script_BlockScript[i]['code'];
- code = code.split("%%%~%%%").join(i);
- for (var j in script_BlockScript[i]['vars']) { code = code.split("%%%" + j + "%%%").join(script_BlockScript[i]['vars'][j]['value']); }
- script += '##### Block: ' + script_BlockScript[i]['name'] + ' #####\r\nHighlightBlock __t ' + i + '\r\n' + code + '\r\n\r\n';
- }
- if (script_BuildingBlocks['_end']) { script += '##### Ending Block #####\r\n' + script_BuildingBlocks['_end']['code'] + '\r\nHighlightBlock\r\n'; }
- }
- return script;
- };
- return o;
- };
|