Mascali
/
piston
mirror of https://github.com/engineer-man/piston.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

api: signaling process via ws

This commit is contained in:
Thomas Hobson 2021-07-17 00:22:55 +12:00
parent f58927d79a
commit 3436648add
No known key found for this signature in database
GPG Key ID: 9F1FD9D87950DB6F
2 changed files with 44 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
api/src/api/v2.js
View File

@ -9,6 +9,8 @@ const { Job } = require('../job');
const package = require('../package');
const logger = require('logplease').create('api/v2');
const SIGNALS = ["SIGABRT","SIGALRM","SIGBUS","SIGCHLD","SIGCLD","SIGCONT","SIGEMT","SIGFPE","SIGHUP","SIGILL","SIGINFO","SIGINT","SIGIO","SIGIOT","SIGKILL","SIGLOST","SIGPIPE","SIGPOLL","SIGPROF","SIGPWR","SIGQUIT","SIGSEGV","SIGSTKFLT","SIGSTOP","SIGTSTP","SIGSYS","SIGTERM","SIGTRAP","SIGTTIN","SIGTTOU","SIGUNUSED","SIGURG","SIGUSR1","SIGUSR2","SIGVTALRM","SIGXCPU","SIGXFSZ","SIGWINCH"]
// ref: https://man7.org/linux/man-pages/man7/signal.7.html
function get_job(body){
const {
@ -148,9 +150,10 @@ router.ws('/connect', async (ws, req) => {
try{
const msg = JSON.parse(data);
if(msg.type === "init"){
switch(msg.type){
case "init":
if(job === null){
const job = await get_job(msg);
job = await get_job(msg);
await job.prime();
@ -167,8 +170,8 @@ router.ws('/connect', async (ws, req) => {
}else{
ws.close(4000, "Already Initialized");
}
}else if(msg.type === "data"){
break;
case "data":
if(job !== null){
if(msg.stream === "stdin"){
eventBus.emit("stdin", msg.data)
@ -178,7 +181,20 @@ router.ws('/connect', async (ws, req) => {
}else{
ws.close(4003, "Not yet initialized")
}
break;
case "signal":
if(job !== null){
if(SIGNALS.includes(msg.signal)){
eventBus.emit("signal", msg.signal)
}else{
ws.close(4005, "Invalid signal")
}
}else{
ws.close(4003, "Not yet initialized")
}
break;
}
}catch(error){
ws.send(JSON.stringify({type: "error", message: error.message}))
ws.close(4002, "Notified Error")

16
api/src/job.js
View File

@ -110,28 +110,36 @@ class Job {
eventBus.on("stdin", (data) => {
proc.stdin.write(data);
})
eventBus.on("kill", (signal) => {
proc.kill(signal)
})
}
const kill_timeout = set_timeout(
_ => proc.kill('SIGKILL'),
timeout
);
proc.stderr.on('data', data => {
if (stderr.length > config.output_max_size) {
if(eventBus !== null) {
eventBus.emit("stderr", data);
} else if (stderr.length > config.output_max_size) {
proc.kill('SIGKILL');
} else {
if(eventBus !== null) eventBus.emit("stderr", data);
stderr += data;
output += data;
}
});
proc.stdout.on('data', data => {
if (stdout.length > config.output_max_size) {
if(eventBus !== null){
eventBus.emit("stdout", data);
} else if (stdout.length > config.output_max_size) {
proc.kill('SIGKILL');
} else {
if(eventBus !== null) eventBus.emit("stdout", data);
stdout += data;
output += data;
}