90 lines
2.7 KiB
JavaScript
90 lines
2.7 KiB
JavaScript
|
'use strict';
|
||
|
|
|
||
|
|
/* Privileged executor. If Mailtrain is started as root, this process keeps the root privilege to be able to spawn workers
|
||
|
|
that can chroot.
|
||
|
|
*/
|
||
|
|
|
||
|
|
const fileHelpers = require('../lib/file-helpers');
|
||
|
|
const fork = require('child_process').fork;
|
||
|
|
const path = require('path');
|
||
|
|
const log = require('npmlog');
|
||
|
|
const fs = require('fs');
|
||
|
|
const privilegeHelpers = require('../lib/privilege-helpers');
|
||
|
|
|
||
|
|
let processes = {};
|
||
|
|
|
||
|
|
function spawnProcess(tid, executable, args, outputFile, cwd) {
|
||
|
|
|
||
|
|
fs.open(outputFile, 'w', (err, outFd) => {
|
||
|
|
if (err) {
|
||
|
|
log.error('Executor', err);
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
privilegeHelpers.ensureMailtrainOwner(outputFile, (err) => {
|
||
|
|
if (err) {
|
||
|
|
log.info('Executor', 'Cannot change owner of output file of process tid:%s.', tid)
|
||
|
|
}
|
||
|
|
|
||
|
|
const options = {
|
||
|
|
stdio: ['ignore', outFd, outFd, 'ipc'],
|
||
|
|
cwd: cwd,
|
||
|
|
env: {NODE_ENV: process.env.NODE_ENV}
|
||
|
|
};
|
||
|
|
|
||
|
|
const child = fork(executable, args, options);
|
||
|
|
const pid = child.pid;
|
||
|
|
processes[tid] = child;
|
||
|
|
|
||
|
|
log.info('Executor', 'Process started with tid:%s pid:%s.', tid, pid);
|
||
|
|
process.send({
|
||
|
|
type: 'process-started',
|
||
|
|
tid
|
||
|
|
});
|
||
|
|
|
||
|
|
child.on('close', (code, signal) => {
|
||
|
|
|
||
|
|
delete processes[tid];
|
||
|
|
log.info('Executor', 'Process tid:%s pid:%s exited with code %s signal %s.', tid, pid, code, signal);
|
||
|
|
|
||
|
|
fs.close(outFd, (err) => {
|
||
|
|
if (err) {
|
||
|
|
log.error('Executor', err);
|
||
|
|
}
|
||
|
|
|
||
|
|
process.send({
|
||
|
|
type: 'process-finished',
|
||
|
|
tid,
|
||
|
|
code,
|
||
|
|
signal
|
||
|
|
});
|
||
|
|
});
|
||
|
|
});
|
||
|
|
});
|
||
|
|
});
|
||
|
|
}
|
||
|
|
|
||
|
|
process.on('message', msg => {
|
||
|
|
if (msg) {
|
||
|
|
const type = msg.type;
|
||
|
|
|
||
|
|
if (type === 'start-report-processor-worker') {
|
||
|
|
spawnProcess(msg.tid, path.join(__dirname, 'report-processor.js'), [msg.data.id], fileHelpers.getReportOutputFile(msg.data), path.join(__dirname, '..'));
|
||
|
|
|
||
|
|
} else if (type === 'stop-process') {
|
||
|
|
const child = processes[msg.tid];
|
||
|
|
|
||
|
|
if (child) {
|
||
|
|
log.info('Executor', 'Killing process tid:%s pid:%s', msg.tid, child.pid);
|
||
|
|
child.kill();
|
||
|
|
} else {
|
||
|
|
log.info('Executor', 'No running process found with tid:%s pid:%s', msg.tid, child.pid);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
});
|
||
|
|
|
||
|
|
process.send({
|
||
|
|
type: 'executor-started'
|
||
|
|
});
|