toolroll 0.9.92 → 0.9.94

package/dist/bounded-spawn.js

@@ -0,0 +1,220 @@
+/**
+ * One bounded child process tree, for work on a repository's untrusted bytes (docs/plans/code-map.md §1). What is
+ * guaranteed depends on the platform, and the result says which:
+ *
+ *   Linux with a delegated cgroup, Windows   the OS object (containment.ts, policy `preferred`) holds the memory limit,
+ *                                            and dies with the worker. `enforced: "system"`.
+ *   macOS, Linux without a cgroup            no hard memory or descendant bound exists. Memory and disk are sampled every
+ *                                            250 ms; a burst between samples can pass the limit before the kill, and a
+ *                                            helper that left the process group can survive it. `enforced: "checked"`.
+ *
+ * `RLIMIT_FSIZE` applies to every POSIX spawn. The whole group is killed with SIGKILL on any limit, and again when the
+ * root exits, so nothing the root started outlives it.
+ */
+import { execFile, execFileSync, spawn } from "node:child_process";
+import { lstatSync, readdirSync, rmSync } from "node:fs";
+import { join, relative, isAbsolute } from "node:path";
+import { createContainer, effectiveContainment, probeContainmentCapability } from "./containment.js";
+const SAMPLE_MS = 250;
+const PIDS = 64;
+/** Applies the file-size limit (bash counts it in 1024-byte blocks, POSIX shells in 512), then becomes the target with
+ * argv untouched. */
+const PRELUDE = 'if [ "$1" != none ]; then if [ -n "$BASH_VERSION" ]; then b=$(($1/1024)); else b=$(($1/512)); fi; ulimit -f "$b" 2>/dev/null || exit 126; fi\nshift\nexec "$@"';
+let containment = null;
+const preferred = () => containment ??= effectiveContainment("preferred", probeContainmentCapability());
+/** Tests: forget the probed machine. */
+export function resetBoundedContainmentForTests() { containment = null; }
+/** A process's own start time, as `ps` reports it: what tells a live process from a reused pid. */
+export function processStartedAt(pid) {
+    if (process.platform === "win32")
+        return null;
+    try {
+        return execFileSync("ps", ["-o", "lstart=", "-p", String(pid)], { encoding: "utf8", timeout: 2_000, stdio: ["ignore", "pipe", "ignore"] }).trim() || null;
+    }
+    catch {
+        return null;
+    }
+}
+function processGroupOf(pid) {
+    if (process.platform === "win32")
+        return null;
+    try {
+        const group = Number(execFileSync("ps", ["-o", "pgid=", "-p", String(pid)], { encoding: "utf8", timeout: 2_000, stdio: ["ignore", "pipe", "ignore"] }).trim());
+        return Number.isInteger(group) && group > 1 ? group : null;
+    }
+    catch {
+        return null;
+    }
+}
+function groupRss(pgid) {
+    return new Promise(done => {
+        execFile("ps", ["-A", "-o", "pgid=,rss="], { encoding: "utf8", timeout: 2_000, maxBuffer: 4 * 1024 * 1024 }, (error, stdout) => {
+            if (error !== null) {
+                done(null);
+                return;
+            }
+            let kilobytes = 0;
+            for (const line of stdout.split("\n")) {
+                const [group, rss] = line.trim().split(/\s+/);
+                if (Number(group) === pgid)
+                    kilobytes += Number(rss) || 0;
+            }
+            done(kilobytes * 1024);
+        });
+    });
+}
+/** The folder's total size, without following a link. */
+export function folderBytes(folder) {
+    let total = 0;
+    const walk = (dir) => {
+        let names;
+        try {
+            names = readdirSync(dir);
+        }
+        catch {
+            return;
+        }
+        for (const name of names) {
+            const path = join(dir, name);
+            try {
+                const held = lstatSync(path);
+                if (held.isDirectory())
+                    walk(path);
+                else
+                    total += held.size;
+            }
+            catch { /* a temporary pack gone between the list and the look */ }
+        }
+    };
+    walk(folder);
+    return total;
+}
+const killGroup = (pgid) => { try {
+    process.kill(-pgid, "SIGKILL");
+}
+catch { /* gone */ } };
+export const boundedRun = async (file, args, options) => {
+    const windows = process.platform === "win32";
+    const effective = preferred();
+    const made = createContainer(effective, options.label);
+    const container = "container" in made ? made.container : null;
+    let enforced = "checked";
+    if (container !== null && options.memoryBytes !== undefined && container.limit?.({ memoryBytes: options.memoryBytes, pids: PIDS }) === true)
+        enforced = "system";
+    const target = windows ? { file, args: [...args] } : { file: "/bin/sh", args: ["-c", PRELUDE, "so-bounded", options.fileBytes === undefined ? "none" : String(options.fileBytes), file, ...args] };
+    const launch = container?.launch(target.file, target.args) ?? null;
+    const child = spawn(launch?.file ?? target.file, launch?.args ?? target.args, {
+        cwd: options.cwd, env: options.env, shell: false, windowsHide: true, detached: !windows,
+        stdio: ["pipe", "pipe", "pipe", ...(launch?.extraStdio ?? [])],
+    });
+    const stdout = [];
+    let stdoutBytes = 0, stderr = "", killed = null, peakBytes = 0, sampling = false;
+    const pid = child.pid;
+    const kill = (why) => {
+        killed ??= why;
+        if (pid !== undefined && !windows)
+            killGroup(pid);
+        if (container !== null)
+            void container.kill();
+        else if (windows) {
+            try {
+                child.kill("SIGKILL");
+            }
+            catch { /* gone */ }
+        }
+    };
+    if (pid !== undefined) {
+        try {
+            options.onStart?.({ pid, pgid: pid, startedAt: processStartedAt(pid) });
+        }
+        catch (error) {
+            kill("time");
+            throw error;
+        }
+        if (launch !== null) {
+            const attached = await launch.attach(child);
+            if (!attached.ok)
+                kill("time");
+        }
+    }
+    child.stdout?.on("data", (chunk) => {
+        stdoutBytes += chunk.length;
+        if (stdoutBytes > options.maxStdout) {
+            kill("output");
+            return;
+        }
+        stdout.push(chunk);
+    });
+    child.stderr?.on("data", (chunk) => { if (stderr.length < options.maxStderr)
+        stderr += chunk.toString("utf8").slice(0, options.maxStderr - stderr.length); });
+    child.stdin?.on("error", () => { });
+    child.stdin?.end(options.stdin);
+    const timer = setTimeout(() => kill("time"), options.timeoutMs);
+    const sampler = setInterval(() => {
+        if (sampling || pid === undefined)
+            return;
+        sampling = true;
+        void (async () => {
+            try {
+                if (options.memoryBytes !== undefined && enforced === "checked" && !windows) {
+                    const rss = await groupRss(pid);
+                    if (rss !== null) {
+                        peakBytes = Math.max(peakBytes, rss);
+                        if (rss > options.memoryBytes)
+                            kill("memory");
+                    }
+                }
+                if (options.diskBytes !== undefined && options.folder !== undefined && folderBytes(options.folder) > options.diskBytes)
+                    kill("disk");
+            }
+            finally {
+                sampling = false;
+            }
+        })();
+    }, options.sampleMs ?? SAMPLE_MS);
+    const finished = await new Promise(done => {
+        let settled = false;
+        const settle = (code, signal) => { if (!settled) {
+            settled = true;
+            done({ code, signal });
+        } };
+        child.once("error", () => settle(127, null));
+        child.once("close", (code, signal) => settle(code, signal));
+        // A helper that left the group can hold the pipes open past the root's exit: the root's exit is the answer.
+        child.once("exit", (code, signal) => {
+            if (pid !== undefined && !windows)
+                killGroup(pid);
+            setTimeout(() => { child.stdout?.destroy(); child.stderr?.destroy(); settle(code, signal); }, 2_000).unref?.();
+        });
+    });
+    clearTimeout(timer);
+    clearInterval(sampler);
+    if (container !== null) {
+        await container.kill(2_000);
+        container.release();
+    }
+    return { code: finished.code, signal: finished.signal, stdout: Buffer.concat(stdout), stderr, killed, enforced, peakBytes };
+};
+/**
+ * What a crashed build left: its group is killed only if the leader is still the process recorded (the pid and its start
+ * time, so a reused pid isn't killed), and its folder goes if it sits under `root`. On Linux the cgroup janitor has
+ * done this already.
+ */
+export function cleanupLeftover(record, root) {
+    let killed = false, removed = false;
+    if (record.pid !== null && record.pgid !== null && record.startedAt !== null && processStartedAt(record.pid) === record.startedAt && processGroupOf(record.pid) === record.pgid) {
+        killGroup(record.pgid);
+        killed = true;
+    }
+    if (record.folder !== null) {
+        const inner = relative(root, record.folder);
+        if (inner !== "" && !inner.startsWith("..") && !isAbsolute(inner)) {
+            try {
+                rmSync(record.folder, { recursive: true, force: true });
+                removed = true;
+            }
+            catch { /* the next takeover tries again */ }
+        }
+    }
+    return { killed, removed };
+}