Files
rea/src/process/LinuxResourceLimits.ts
T
Ysopwnable f611d8ce72 feat(captures): inspect recorded packets with bounded TShark delivery (#2318)
Inspect selected fields in existing captures through shared CLI/MCP Evidence.
Page matching frames without changing dissection or two-pass read coverage;
retain failed descriptor/process cleanup ownership and reject excess admission.

Verified with real TShark 4.0.17 and 4.2.2, public Wireshark recordings,
owned HTTP application traffic, real CLI/MCP workflows and schema checks.

Co-authored-by: Olorinm <59321404+Olorinm@users.noreply.github.com>
Assisted-by: Kogasa (gensokyo-skills:kogasa-surprise-testing)
2026-10-12 19:35:35 +00:00

83 lines
2.9 KiB
TypeScript

import { open } from "node:fs/promises";
import { ArtifactReaderFailure } from "../artifacts/ArtifactReader.js";
import type { ArtifactResourceScope } from "../artifacts/ArtifactResourceScope.js";
import { OwnedFileHandle } from "../filesystem/OwnedFileHandle.js";
/** Per-process ceilings inherited by an owned Linux producer. */
export interface LinuxResourceLimits {
readonly address_space_bytes: number;
readonly file_bytes: number;
readonly cpu_seconds: number;
}
const rows = {
address_space_bytes: "Max address space",
file_bytes: "Max file size",
cpu_seconds: "Max cpu time",
} as const;
/** Never raise either inherited soft policy or hard ceiling. */
export const clampLinuxResourceLimits = (
configured: LinuxResourceLimits,
report: string,
): LinuxResourceLimits => {
const effective = { ...configured };
for (const key of Object.keys(rows) as (keyof LinuxResourceLimits)[]) {
const matches = report
.split("\n")
.filter((line) => line.startsWith(`${rows[key]} `));
const values = matches[0]?.slice(rows[key].length).trim().split(/\s+/u);
if (matches.length !== 1 || values === undefined || values.length !== 3)
throw new Error(`Cannot establish inherited Linux limit: ${rows[key]}`);
let ceiling = BigInt(configured[key]);
for (const token of values.slice(0, 2)) {
if (token === "unlimited") continue;
if (!/^[0-9]+$/u.test(token))
throw new Error(`Invalid inherited Linux limit: ${rows[key]}`);
const value = BigInt(token);
if (value < ceiling) ceiling = value;
}
effective[key] = Number(ceiling);
}
return effective;
};
/** Read the trusted kernel report with a fixed allocation before spawning. */
export const readLinuxResourceLimits = async (
configured: LinuxResourceLimits,
resourceScope?: ArtifactResourceScope,
): Promise<LinuxResourceLimits> => {
const file = await open("/proc/self/limits", "r");
const owner = new OwnedFileHandle(file);
let failure: unknown;
try {
const buffer = Buffer.alloc(16 * 1024);
const { bytesRead } = await file.read(buffer, 0, buffer.length, 0);
if (bytesRead === buffer.length)
throw new Error("Inherited Linux limits exceed the report budget.");
return clampLinuxResourceLimits(
configured,
buffer.subarray(0, bytesRead).toString("utf8"),
);
} catch (cause: unknown) {
failure = cause;
throw cause;
} finally {
if (resourceScope === undefined) await file.close();
else {
const cleaned = await resourceScope.release({
kind: "file-handle",
resource: "/proc/self/limits",
handle: owner,
});
if (cleaned.kind === "failed")
throw ArtifactReaderFailure.withCleanup(
failure ?? cleaned.cause,
ArtifactReaderFailure.cleanupObservation(
cleaned.cause,
"/proc/self/limits",
),
);
}
}
};