2019-05-16 08:05:36 +02:00
|
|
|
/**
|
2019-05-17 08:44:59 +02:00
|
|
|
* Stops an actively-running script (represented by a WorkerScript object)
|
2019-05-16 08:05:36 +02:00
|
|
|
* and removes it from the global pool of active scripts.
|
|
|
|
*/
|
|
|
|
import { WorkerScript } from "./WorkerScript";
|
|
|
|
import { workerScripts } from "./WorkerScripts";
|
2019-05-17 22:41:16 +02:00
|
|
|
import { WorkerScriptStartStopEventEmitter } from "./WorkerScriptStartStopEventEmitter";
|
2019-05-16 08:05:36 +02:00
|
|
|
|
|
|
|
import { RunningScript } from "../Script/RunningScript";
|
|
|
|
import { AllServers } from "../Server/AllServers";
|
|
|
|
|
|
|
|
import { compareArrays } from "../../utils/helpers/compareArrays";
|
|
|
|
import { roundToTwo } from "../../utils/helpers/roundToTwo";
|
|
|
|
|
|
|
|
export function killWorkerScript(runningScriptObj: RunningScript, serverIp: string): boolean;
|
|
|
|
export function killWorkerScript(workerScript: WorkerScript): boolean;
|
2019-06-19 10:51:25 +02:00
|
|
|
export function killWorkerScript(pid: number): boolean;
|
|
|
|
export function killWorkerScript(script: RunningScript | WorkerScript | number, serverIp?: string): boolean {
|
2019-05-16 08:05:36 +02:00
|
|
|
if (script instanceof WorkerScript) {
|
2019-06-19 10:51:25 +02:00
|
|
|
stopAndCleanUpWorkerScript(script);
|
2019-05-16 08:05:36 +02:00
|
|
|
|
|
|
|
return true;
|
|
|
|
} else if (script instanceof RunningScript && typeof serverIp === "string") {
|
2019-06-19 10:51:25 +02:00
|
|
|
// Try to kill by PID
|
|
|
|
const res = killWorkerScriptByPid(script.pid);
|
|
|
|
if (res) { return res; }
|
|
|
|
|
|
|
|
// If for some reason that doesn't work, we'll try the old way
|
|
|
|
for (const ws of workerScripts.values()) {
|
|
|
|
if (ws.name == script.filename && ws.serverIp == serverIp &&
|
|
|
|
compareArrays(ws.args, script.args)) {
|
|
|
|
|
|
|
|
stopAndCleanUpWorkerScript(ws);
|
2019-05-16 08:05:36 +02:00
|
|
|
|
|
|
|
return true;
|
|
|
|
}
|
2019-06-19 10:51:25 +02:00
|
|
|
}
|
2019-05-16 08:05:36 +02:00
|
|
|
|
|
|
|
return false;
|
2019-06-19 10:51:25 +02:00
|
|
|
} else if (typeof script === "number") {
|
|
|
|
return killWorkerScriptByPid(script);
|
2019-05-16 08:05:36 +02:00
|
|
|
} else {
|
2019-05-17 08:44:59 +02:00
|
|
|
console.error(`killWorkerScript() called with invalid argument:`);
|
|
|
|
console.error(script);
|
2019-05-16 08:05:36 +02:00
|
|
|
return false;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-06-19 10:51:25 +02:00
|
|
|
function stopAndCleanUpWorkerScript(workerScript: WorkerScript): void {
|
|
|
|
workerScript.env.stopFlag = true;
|
|
|
|
killNetscriptDelay(workerScript);
|
|
|
|
removeWorkerScript(workerScript);
|
|
|
|
}
|
|
|
|
|
|
|
|
function killWorkerScriptByPid(pid: number): boolean {
|
|
|
|
const ws = workerScripts.get(pid);
|
|
|
|
if (ws instanceof WorkerScript) {
|
|
|
|
stopAndCleanUpWorkerScript(ws);
|
|
|
|
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
2019-05-16 08:05:36 +02:00
|
|
|
/**
|
|
|
|
* Helper function that removes the script being killed from the global pool.
|
|
|
|
* Also handles other cleanup-time operations
|
|
|
|
*
|
|
|
|
* @param {WorkerScript | number} - Identifier for WorkerScript. Either the object itself, or
|
|
|
|
* its index in the global workerScripts array
|
|
|
|
*/
|
2019-06-19 10:51:25 +02:00
|
|
|
function removeWorkerScript(workerScript: WorkerScript): void {
|
|
|
|
if (workerScript instanceof WorkerScript) {
|
|
|
|
const ip = workerScript.serverIp;
|
|
|
|
const name = workerScript.name;
|
|
|
|
|
|
|
|
// Get the server on which the script runs
|
|
|
|
const server = AllServers[ip];
|
|
|
|
if (server == null) {
|
|
|
|
console.error(`Could not find server on which this script is running: ${ip}`);
|
2019-05-16 08:05:36 +02:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2019-06-19 10:51:25 +02:00
|
|
|
// Recalculate ram used on that server
|
|
|
|
server.ramUsed = roundToTwo(server.ramUsed - workerScript.ramUsage);
|
|
|
|
if (server.ramUsed < 0) {
|
|
|
|
console.warn(`Server RAM usage went negative (if it's due to floating pt imprecision, it's okay): ${server.ramUsed}`);
|
|
|
|
server.ramUsed = 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
// Delete the RunningScript object from that server
|
|
|
|
for (let i = 0; i < server.runningScripts.length; ++i) {
|
|
|
|
const runningScript = server.runningScripts[i];
|
|
|
|
if (runningScript.filename === name && compareArrays(runningScript.args, workerScript.args)) {
|
|
|
|
server.runningScripts.splice(i, 1);
|
2019-05-16 08:05:36 +02:00
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-06-19 10:51:25 +02:00
|
|
|
// Delete script from global pool (workerScripts)
|
|
|
|
const res = workerScripts.delete(workerScript.pid);
|
|
|
|
if (!res) {
|
|
|
|
console.warn(`removeWorkerScript() called with WorkerScript that wasn't in the global map:`);
|
|
|
|
console.warn(workerScript);
|
2019-05-16 08:05:36 +02:00
|
|
|
}
|
|
|
|
|
2019-06-19 10:51:25 +02:00
|
|
|
WorkerScriptStartStopEventEmitter.emitEvent();
|
|
|
|
} else {
|
|
|
|
console.error(`Invalid argument passed into removeWorkerScript():`);
|
|
|
|
console.error(workerScript);
|
2019-05-16 08:05:36 +02:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Helper function that interrupts a script's delay if it is in the middle of a
|
|
|
|
* timed, blocked operation (like hack(), sleep(), etc.). This allows scripts to
|
|
|
|
* be killed immediately even if they're in the middle of one of those long operations
|
|
|
|
*/
|
|
|
|
function killNetscriptDelay(workerScript: WorkerScript) {
|
|
|
|
if (workerScript instanceof WorkerScript) {
|
|
|
|
if (workerScript.delay) {
|
|
|
|
clearTimeout(workerScript.delay);
|
|
|
|
if (workerScript.delayResolve) {
|
|
|
|
workerScript.delayResolve();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|