api: signaling process via ws

api/src/api/v2.js

@@ -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");
                     }
-
+                    break;
-            }else if(msg.type === "data"){
+            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")

api/src/job.js

@@ -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;
                 }