mirror of
https://github.com/Ylianst/MeshAgent
synced 2025-12-06 00:13:33 +00:00
6f28cdb484
2. Fixed dispatcher on Windows to put username in quotes 3. Added additional logging 4. Added TLS/ALPN support
529 lines
26 KiB
JavaScript
529 lines
26 KiB
JavaScript
/*
|
|
Copyright 2019 Intel Corporation
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
var promise = require('promise');
|
|
var servicemanager = require('service-manager');
|
|
var mgr = new servicemanager();
|
|
|
|
//attachDebugger({ webport: 9995, wait: 1 }).then(console.log);
|
|
|
|
function task()
|
|
{
|
|
this._ObjectID = 'task-scheduler';
|
|
|
|
if (process.platform == 'win32')
|
|
{
|
|
this.getTaskXml = function getTaskXml(name)
|
|
{
|
|
var child = require('child_process').execFile(process.env['windir'] + '\\system32\\schtasks.exe', ['schtasks', '/QUERY', '/TN "' + name+'"', '/XML']);
|
|
child.stdout.str = ''; child.stdout.on('data', function (c) { this.str += c.toString(); });
|
|
child.stderr.str = ''; child.stderr.on('data', function (c) { this.str += c.toString(); });
|
|
child.waitExit();
|
|
if (child.stderr.str.trim() != '') { throw ('Unable to fetch task: ' + name); }
|
|
return (child.stdout.str.trim());
|
|
}
|
|
this.getActionCommand = function getActionCommand(name, xml)
|
|
{
|
|
if (!xml)
|
|
{
|
|
var child = require('child_process').execFile(process.env['windir'] + '\\system32\\schtasks.exe', ['schtasks', '/QUERY', '/TN "' + name + '"', '/XML']);
|
|
child.stdout.str = ''; child.stdout.on('data', function (c) { this.str += c.toString(); });
|
|
child.stderr.str = ''; child.stderr.on('data', function (c) { this.str += c.toString(); });
|
|
child.waitExit();
|
|
if (child.stderr.str.trim() != '') { throw ('Unable to fetch task: ' + name); }
|
|
xml = child.stdout.str;
|
|
}
|
|
var xElement = xml.split('</Exec>')[0].split('<Exec>')[1];
|
|
var command = xElement.split('</Command>')[0].split('<Command>')[1];
|
|
return (command);
|
|
};
|
|
this.editActionCommand = function editActionCommand(name, action, argString, xml)
|
|
{
|
|
if (!xml)
|
|
{
|
|
var child = require('child_process').execFile(process.env['windir'] + '\\system32\\schtasks.exe', ['schtasks', '/QUERY', '/TN "' + name + '"', '/XML']);
|
|
child.stdout.str = ''; child.stdout.on('data', function (c) { this.str += c.toString(); });
|
|
child.stderr.str = ''; child.stderr.on('data', function (c) { this.str += c.toString(); });
|
|
child.waitExit();
|
|
if (child.stderr.str.trim() != '') { throw ('Unable to fetch task: ' + name); }
|
|
xml = child.stdout.str;
|
|
}
|
|
|
|
var pt1 = xml.split('</Exec>'); // xml = pt1.join('</Exec>');
|
|
var pt2 = pt1[0].split('<Exec>'); // pt1[0] = pt2.join('<Exec>');
|
|
var xElement = pt2[1]; // pt2[1] = xElement;
|
|
|
|
var pt3 = xElement.split('</Command>'); // xElement = pt3.join('</Command>');
|
|
var pt4 = pt3[0].split('<Command>'); // pt3[0] = pt4.join('<Command>');
|
|
var command = pt4[1]; // pt4[1] = command;
|
|
|
|
pt4[1] = action;
|
|
pt3[0] = pt4.join('<Command>');
|
|
xElement = pt3.join('</Command>');
|
|
|
|
var pt5 = xElement.split('</Arguments>'); // xElement = pt5.join('</Arguments>');
|
|
var pt6 = pt5[0].split('<Arguments>'); // pt5[0] = pt6.join('<Arguments>');
|
|
var arg = pt6[1]; // pt6[1] = arg;
|
|
|
|
arg = argString;
|
|
pt6[1] = arg;
|
|
pt5[0] = pt6.join('<Arguments>');
|
|
xElement = pt5.join('</Arguments>');
|
|
|
|
pt2[1] = xElement;
|
|
pt1[0] = pt2.join('<Exec>');
|
|
xml = pt1.join('</Exec>');
|
|
|
|
var s = require('fs').createWriteStream(require('os').tmpdir() + name + '.xml', { flags: 'wb' });
|
|
var b = Buffer.alloc(2);
|
|
b[0] = 0xFF;
|
|
b[1] = 0xFE;
|
|
|
|
s.write(b);
|
|
s.write(Buffer.from(xml).toString('utf16'));
|
|
s.end();
|
|
|
|
var child = require('child_process').execFile(process.env['windir'] + '\\system32\\cmd.exe', ['cmd']);
|
|
child.stdout.str = ''; child.stdout.on('data', function (c) { this.str += c.toString(); });
|
|
child.stderr.str = ''; child.stderr.on('data', function (c) { this.str += c.toString(); });
|
|
child.stdin.write('SCHTASKS /DELETE /TN ' + name + ' /F \n');
|
|
child.stdin.write('SCHTASKS /CREATE /TN ' + name + ' /XML ' + require('os').tmpdir() + name + '.xml\n');
|
|
child.stdin.write('erase ' + require('os').tmpdir() + name + '.xml\nexit\n');
|
|
child.waitExit();
|
|
|
|
//console.log(child.stdout.str.trim());
|
|
//console.log(child.stderr.str.trim());
|
|
};
|
|
|
|
this.advancedEditActionCommand = function advancedEditActionCommand(name, action, argString)
|
|
{
|
|
var child = require('child_process').execFile(process.env['windir'] + '\\System32\\WindowsPowerShell\\v1.0\\powershell.exe', ['powershell.exe']);
|
|
child.stdout.str = ''; child.stdout.on('data', function (c) { this.str += c.toString(); });
|
|
child.stderr.str = ''; child.stderr.on('data', function (c) { this.str += c.toString(); });
|
|
child.stdin.write('$Act1 = New-ScheduledTaskAction -Execute "' + action + '" -Argument "' + argString + '"\n');
|
|
child.stdin.write('Set-ScheduledTask "' + name + '" -Action $Act1\nexit\n');
|
|
child.waitExit();
|
|
console.log(child.stdout.str.trim());
|
|
};
|
|
Object.defineProperty(this, "advancedSupport", {
|
|
value: (function ()
|
|
{
|
|
var child = require('child_process').execFile(process.env['windir'] + '\\System32\\WindowsPowerShell\\v1.0\\powershell.exe', ['/C "Get-Module -ListAvailable -Name ScheduledTasks"']);
|
|
child.stdout.str = ''; child.stdout.on('data', function (c) { this.str += c.toString(); });
|
|
child.stderr.str = ''; child.stderr.on('data', function (c) { this.str += c.toString(); });
|
|
child.waitExit();
|
|
return (child.stdout.str.trim() != '');
|
|
})()
|
|
});
|
|
}
|
|
|
|
|
|
this.create = function create(options)
|
|
{
|
|
var ret = new promise(function (res, rej) { this._res = res; this._rej = rej; });
|
|
if(options.name && options.service)
|
|
{
|
|
switch(process.platform)
|
|
{
|
|
case 'win32':
|
|
var parms = ['schtasks', '/Create', '/RU SYSTEM'];
|
|
for (var ftype in options)
|
|
{
|
|
switch(ftype.toUpperCase())
|
|
{
|
|
case 'MINUTE':
|
|
case 'HOURLY':
|
|
case 'DAILY':
|
|
case 'WEEKLY':
|
|
case 'MONTHLY':
|
|
parms.push('/SC ' + ftype.toUpperCase());
|
|
parms.push('/MO ' + options[ftype]);
|
|
break;
|
|
case 'DAY':
|
|
parms.push('/D ' + options[ftype]);
|
|
break;
|
|
case 'MONTH':
|
|
parms.push('/M ' + options[ftype]);
|
|
break;
|
|
case 'TIME':
|
|
parms.push('/ST ' + options[ftype]);
|
|
break;
|
|
case 'NAME':
|
|
parms.push('/TN "' + options[ftype].split('/').join('\\') + '"');
|
|
break;
|
|
case 'SERVICE':
|
|
parms.push('/TR "net start ' + options[ftype] + '"');
|
|
break;
|
|
}
|
|
}
|
|
console.log(parms.join(' '));
|
|
ret.child = require('child_process').execFile(process.env['windir'] + '\\system32\\schtasks.exe', parms);
|
|
ret.child.stdout.str = '';
|
|
ret.child.stdout.on('data', function (chunk) { this.str += chunk.toString(); });
|
|
ret.child.stderr.on('data', function (chunk) { });
|
|
ret.child.promise = ret;
|
|
ret.child.on('exit', function (code) { if (code == 0) { this.promise._res(); } else { this.promise._rej(code); }});
|
|
break;
|
|
case 'linux':
|
|
if (require('fs').existsSync('/etc/cron.d/' + options.name.split('/').join('_').split('.').join('')))
|
|
{
|
|
ret._rej('Task [' + options.name + '] Already exists');
|
|
return (ret);
|
|
}
|
|
var minute = '*';
|
|
var hour = '*';
|
|
var day = '*';
|
|
var month = '*';
|
|
var weekday = '*';
|
|
for (var ftype in options)
|
|
{
|
|
switch(ftype.toUpperCase())
|
|
{
|
|
case 'MINUTE':
|
|
if (!options.TIME && !options.time)
|
|
{
|
|
minute = '*/' + options[ftype];
|
|
}
|
|
break;
|
|
case 'HOURLY':
|
|
if (!options.TIME && !options.time)
|
|
{
|
|
hour = '*/' + options[ftype];
|
|
}
|
|
break;
|
|
case 'DAILY':
|
|
day = '*/' + options[ftype];
|
|
break;
|
|
case 'WEEKLY':
|
|
if (options[ftype] == 1)
|
|
{
|
|
if(!options.DAY && !options.day)
|
|
{
|
|
weekday = 0;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
ret._rej('Only Once/Weekly supported on Linux');
|
|
return (ret);
|
|
}
|
|
break;
|
|
case 'DAY':
|
|
if (options.weekly || options.WEEKLY)
|
|
{
|
|
weekday = options[ftype];
|
|
}
|
|
else
|
|
{
|
|
day = options[ftype];
|
|
}
|
|
break;
|
|
case 'TIME':
|
|
hour = options[ftype].split(':')[0];
|
|
minute = options[ftype].split(':')[1];
|
|
break;
|
|
case 'MONTHLY':
|
|
month = '*/' + options[ftype];
|
|
break;
|
|
}
|
|
}
|
|
|
|
var action = 'SHELL=/bin/sh\nPATH=/usr/local/sbin:/usr/local/bin:/sbin:/bin:/usr/sbin:/usr/bin\n\n';
|
|
action += (minute + ' ' + hour + ' ' + day + ' ' + month + ' ' + weekday + ' root ');
|
|
switch(require('service-manager').manager.getServiceType())
|
|
{
|
|
case 'init':
|
|
var child = require('child_process').execFile('/bin/sh', ['sh']);
|
|
child.stdout.str = '';
|
|
child.stdout.on('data', function (chunk) { this.str += chunk.toString(); });
|
|
child.stderr.on('data', function (chunk) { });
|
|
child.stdin.write("whereis service | awk '{print $2}'\n\exit\n");
|
|
child.waitExit();
|
|
child.stdout.str = child.stdout.str.trim();
|
|
action += (child.stdout.str + ' ' + options.service + ' start >/dev/null 2>&1 \n');
|
|
break;
|
|
case 'upstart':
|
|
var child = require('child_process').execFile('/bin/sh', ['sh']);
|
|
child.stdout.str = '';
|
|
child.stdout.on('data', function (chunk) { this.str += chunk.toString(); });
|
|
child.stderr.on('data', function (chunk) { });
|
|
child.stdin.write("whereis initctl | awk '{print $2}'\n\exit\n");
|
|
child.waitExit();
|
|
child.stdout.str = child.stdout.str.trim();
|
|
action += (child.stdout.str + ' start ' + options.service + ' >/dev/null 2>&1 \n');
|
|
break;
|
|
case 'systemd':
|
|
var child = require('child_process').execFile('/bin/sh', ['sh']);
|
|
child.stdout.str = '';
|
|
child.stdout.on('data', function (chunk) { this.str += chunk.toString(); });
|
|
child.stderr.on('data', function (chunk) { });
|
|
child.stdin.write("whereis systemctl | awk '{print $2}'\n\exit\n");
|
|
child.waitExit();
|
|
child.stdout.str = child.stdout.str.trim();
|
|
action += (child.stdout.str + ' start ' + options.service + ' >/dev/null 2>&1 \n');
|
|
break;
|
|
default:
|
|
ret._rej('Unknown Service Platform: ' + require('service-manager').manager.getServiceType());
|
|
return (ret);
|
|
}
|
|
try
|
|
{
|
|
var m = require('fs').CHMOD_MODES.S_IRUSR | require('fs').CHMOD_MODES.S_IWUSR | require('fs').CHMOD_MODES.S_IROTH;
|
|
require('fs').writeFileSync('/etc/cron.d/' + options.name.split('/').join('_').split('.').join(''), action, { flags: 'wb', mode: m });
|
|
}
|
|
catch(e)
|
|
{
|
|
ret._rej(e);
|
|
return (ret);
|
|
}
|
|
ret._res();
|
|
break;
|
|
case 'darwin':
|
|
var taskname = options.name.split('/').join('_').split('.').join('');
|
|
var plist = '<?xml version="1.0" encoding="UTF-8"?>\n';
|
|
plist += '<!DOCTYPE plist PUBLIC "-//Apple Computer//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">\n';
|
|
plist += '<plist version="1.0">\n';
|
|
plist += ' <dict>\n';
|
|
plist += ' <key>Label</key>\n';
|
|
plist += (' <string>' + taskname + '</string>\n');
|
|
plist += ' <key>ProgramArguments</key>\n';
|
|
plist += ' <array>\n';
|
|
plist += ' <string>/bin/launchctl</string>\n';
|
|
plist += ' <string>start</string>\n';
|
|
plist += (' <string>' + options.service + '</string>\n');
|
|
plist += ' </array>\n';
|
|
plist += ' <key>RunAtLoad</key>\n';
|
|
plist += ' <false/>\n';
|
|
plist += '{{{INTERVAL}}}';
|
|
plist += ' </dict>\n';
|
|
plist += '</plist>';
|
|
|
|
try
|
|
{
|
|
var svc = require('service-manager').manager.getService(options.service);
|
|
if (!svc.isLoaded()) { svc.load(); }
|
|
svc = null;
|
|
}
|
|
catch(se)
|
|
{
|
|
ret._rej(se); return (ret);
|
|
}
|
|
|
|
var interval = null;
|
|
var periodic = [];
|
|
|
|
for (var ftype in options)
|
|
{
|
|
switch (ftype.toUpperCase())
|
|
{
|
|
case 'DAILY':
|
|
var dailyVal = parseInt(options[ftype]);
|
|
if (dailyVal < 1 || dailyVal > 31)
|
|
{
|
|
ret._rej('Invalid Options'); return (ret);
|
|
}
|
|
if (dailyVal > 1)
|
|
{
|
|
var currentDay = (new Date()).getDate(); // 0 - 31
|
|
var actualDay = currentDay;
|
|
do
|
|
{
|
|
currentDay += dailyVal;
|
|
if (currentDay > 31) currentDay = currentDay % 31;
|
|
periodic.push((' <key>Day</key>\n <integer>' + currentDay + '</integer>\n'));
|
|
} while (!(currentDay < actualDay && (currentDay + dailyVal) > actualDay));
|
|
}
|
|
else
|
|
{
|
|
periodic.push('');
|
|
}
|
|
break;
|
|
case 'WEEKLY':
|
|
if (parseInt(options[ftype]) != 1) { ret._rej('Only once weekly is supported'); return (ret); }
|
|
if (options.DAY < 0 || options.DAY > 6 || options.day < 0 || options.day > 6) { ret._rej('DAY out of range'); return (ret); }
|
|
if (options.DAY == null && options.day == null)
|
|
{
|
|
periodic.push((' <key>Day</key>\n <integer>' + (new Date()).getDay() + '</integer>\n'));
|
|
}
|
|
else
|
|
{
|
|
periodic.push('');
|
|
}
|
|
break;
|
|
case 'MONTHLY':
|
|
if (options.month == null && options.MONTH == null)
|
|
{
|
|
var monthlyVal = parseInt(options[ftype]);
|
|
var currentMonth = (new Date()).getMonth();
|
|
var actualMonth= currentMonth;
|
|
do
|
|
{
|
|
currentMonth += monthlyVal;
|
|
if (currentMonth > 12) currentMonth = currentMonth % 12;
|
|
periodic.push((' <key>Month</key>\n <integer>' + currentMonth + '</integer>\n'));
|
|
} while (!(currentMonth < actualMonth && (currentMonth + monthlyVal) > actualMonth));
|
|
}
|
|
else
|
|
{
|
|
periodic.push('');
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
|
|
for (var ftype in options)
|
|
{
|
|
switch (ftype.toUpperCase())
|
|
{
|
|
case 'MINUTE':
|
|
if (interval != null || periodic.length > 0) { ret._rej('Invalid Options'); return (ret); }
|
|
interval = ' <integer>' + (parseInt(options[ftype]) * 60) + '</integer>\n';
|
|
break;
|
|
case 'HOURLY':
|
|
if (interval != null || periodic.length > 0) { ret._rej('Invalid Options'); return (ret); }
|
|
interval = ' <integer>' + (parseInt(options[ftype]) * 60 * 60) + '</integer>\n';
|
|
break;
|
|
case 'DAY':
|
|
for (var d in periodic)
|
|
{
|
|
periodic[d] += (' <key>Day</key>\n <integer>' + options[ftype] + '</integer>\n');
|
|
}
|
|
break;
|
|
case 'MONTH':
|
|
for (var m in periodic)
|
|
{
|
|
periodic[m] += (' <key>Month</key>\n <integer>' + options[ftype] + '</integer>\n');
|
|
}
|
|
break;
|
|
case 'TIME':
|
|
if (interval != null) { ret._rej('Invalid Options'); return (ret); }
|
|
for (var t in periodic)
|
|
{
|
|
periodic[t] += (' <key>Hour</key>\n <integer>' + options[ftype].split(':')[0] + '</integer>\n' + ' <key>Minute</key>\n <integer>' + options[ftype].split(':')[1] + '</integer>\n');
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
if (interval)
|
|
{
|
|
plist = plist.replace('{{{INTERVAL}}}', ' <key>StartInterval</key>\n' + interval);
|
|
}
|
|
|
|
if (periodic.length > 0)
|
|
{
|
|
plist = plist.replace('{{{INTERVAL}}}', ' <key>StartCalendarInterval</key>\n <array><dict>\n' + periodic.join(' </dict>\n <dict>\n') + ' </dict></array>\n');
|
|
}
|
|
require('fs').writeFileSync('/Library/LaunchDaemons/' + taskname + '.plist', plist);
|
|
|
|
var child = require('child_process').execFile('/bin/sh', ['sh']);
|
|
child.stdout.on('data', function (chunk) { });
|
|
child.stdin.write('launchctl load /Library/LaunchDaemons/' + taskname + '.plist\nexit\n');
|
|
child.waitExit();
|
|
|
|
|
|
|
|
ret._res();
|
|
break;
|
|
default:
|
|
ret._rej('Not implemented on ' + process.platform);
|
|
break;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
ret._rej('Invalid Parameters, must at least specify name and service');
|
|
}
|
|
return (ret);
|
|
};
|
|
this.info = function info(name)
|
|
{
|
|
var ret = new promise(function (res, rej) { this._res = res; this._rej = rej; });
|
|
switch (process.platform)
|
|
{
|
|
default:
|
|
ret._rej('Not implemented on ' + process.platform);
|
|
break;
|
|
}
|
|
return (ret);
|
|
};
|
|
this.delete = function _delete(name)
|
|
{
|
|
var ret = new promise(function (res, rej) { this._res = res; this._rej = rej; });
|
|
switch (process.platform)
|
|
{
|
|
case 'win32':
|
|
ret.child = require('child_process').execFile(process.env['windir'] + '\\system32\\schtasks.exe', ['schtasks', '/Delete', '/TN "' + name.split('/').join('\\') + '"', '/F']);
|
|
ret.child.stdout.str = '';
|
|
ret.child.stdout.on('data', function (chunk) { this.str += chunk.toString(); });
|
|
ret.child.stderr.on('data', function (chunk) { });
|
|
ret.child.promise = ret;
|
|
ret.child.on('exit', function (code) { if (code == 0) { this.promise._res(); } else { this.promise._rej(code); } });
|
|
break;
|
|
case 'linux':
|
|
if (require('fs').existsSync('/etc/cron.d/' + name.split('/').join('_').split('.').join('')))
|
|
{
|
|
try
|
|
{
|
|
require('fs').unlinkSync('/etc/cron.d/' + name.split('/').join('_').split('.').join(''));
|
|
}
|
|
catch(e)
|
|
{
|
|
ret._rej(e);
|
|
return (ret);
|
|
}
|
|
ret._res();
|
|
}
|
|
else
|
|
{
|
|
ret._rej('Task [' + name + '] does not exist');
|
|
}
|
|
break;
|
|
case 'darwin':
|
|
var taskname = name.split('/').join('_').split('.').join('');
|
|
if (require('fs').existsSync('/Library/LaunchDaemons/' + taskname + '.plist'))
|
|
{
|
|
var child = require('child_process').execFile('/bin/sh', ['sh']);
|
|
child.stdout.on('data', function (chunk) { });
|
|
child.stdin.write('launchctl unload /Library/LaunchDaemons/' + taskname + '.plist\nexit\n');
|
|
child.waitExit();
|
|
try
|
|
{
|
|
require('fs').unlinkSync('/Library/LaunchDaemons/' + taskname + '.plist');
|
|
}
|
|
catch (e)
|
|
{
|
|
ret._rej(e);
|
|
return (ret);
|
|
}
|
|
ret._res();
|
|
}
|
|
else
|
|
{
|
|
ret._rej('Task [' + name + '] does not exist');
|
|
}
|
|
break;
|
|
default:
|
|
ret._rej('Not implemented on ' + process.platform);
|
|
break;
|
|
}
|
|
return (ret);
|
|
};
|
|
}
|
|
|
|
|
|
module.exports = new task();
|
|
|