118 lines
2.7 KiB
JavaScript
118 lines
2.7 KiB
JavaScript
/**
|
|
* Copyright 2013-2017 Zaid Abdulla
|
|
*
|
|
* This file is part of GenieACS.
|
|
*
|
|
* GenieACS is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU Affero General Public License as
|
|
* published by the Free Software Foundation, either version 3 of the
|
|
* License, or (at your option) any later version.
|
|
*
|
|
* GenieACS is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU Affero General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Affero General Public License
|
|
* along with GenieACS. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
"use strict";
|
|
|
|
const childProcess = require("child_process");
|
|
const crypto = require ("crypto");
|
|
|
|
const config = require("./config");
|
|
|
|
const TIMEOUT = config.get('EXT_TIMEOUT');
|
|
|
|
const processes = {};
|
|
const jobs = {};
|
|
|
|
function messageHandler(message) {
|
|
let func = jobs[message[0]];
|
|
if (func) {
|
|
delete jobs[message[0]];
|
|
func(null, message[1], message[2]);
|
|
}
|
|
}
|
|
|
|
|
|
function run(args, callback) {
|
|
let scriptName = args[0];
|
|
|
|
if (!processes[scriptName]) {
|
|
let p = childProcess.fork('lib/extension-wrapper', [scriptName]);
|
|
processes[scriptName] = p;
|
|
|
|
p.on('error', function() {
|
|
if (process[scriptName] === p) {
|
|
kill(processes[scriptName]);
|
|
delete processes[scriptName];
|
|
}
|
|
});
|
|
|
|
p.on('disconnect', function() {
|
|
if (process[scriptName] === p) {
|
|
kill(processes[scriptName]);
|
|
delete processes[scriptName];
|
|
}
|
|
});
|
|
|
|
p.on('message', messageHandler);
|
|
}
|
|
|
|
let id = crypto.randomBytes(8).toString('hex');
|
|
jobs[id] = callback;
|
|
setTimeout(function() {
|
|
if (id in jobs) {
|
|
delete jobs[id];
|
|
return callback(null, {code: "timeout", message: "Extension timed out"});
|
|
}
|
|
}, TIMEOUT);
|
|
|
|
return processes[scriptName].send([id, args.slice(1)]);
|
|
};
|
|
|
|
|
|
function kill(process, callback) {
|
|
if (process.signalCode !== null || process.exitCode !== null)
|
|
return;
|
|
|
|
let timeToKill = Date.now() + 5000;
|
|
|
|
process.kill();
|
|
|
|
let t = setInterval(function() {
|
|
if (process.signalCode !== null || process.exitCode !== null) {
|
|
clearInterval(t);
|
|
if (callback)
|
|
callback();
|
|
}
|
|
else if (Date.now() > timeToKill) {
|
|
process.kill('SIGKILL');
|
|
clearInterval(t);
|
|
if (callback)
|
|
callback();
|
|
}
|
|
}, 100);
|
|
}
|
|
|
|
|
|
function killAll(callback) {
|
|
let counter = 1;
|
|
for (let n in processes) {
|
|
++counter;
|
|
kill(processes[n], function() {
|
|
if (--counter == 0 && callback)
|
|
callback();
|
|
});
|
|
delete processes[n];
|
|
}
|
|
if (--counter == 0 && callback)
|
|
callback();
|
|
}
|
|
|
|
|
|
exports.run = run;
|
|
exports.killAll = killAll;
|