88 lines
2.7 KiB
TypeScript
88 lines
2.7 KiB
TypeScript
|
// Cgroup management code
|
||
|
|
// this sucks, ill mess with it later
|
||
|
|
|
||
|
|
import { appendFileSync, existsSync, mkdirSync, readFileSync, writeFileSync } from 'node:fs';
|
||
|
|
import path from 'node:path';
|
||
|
|
|
||
|
|
export class CGroupController {
|
||
|
|
private controller;
|
||
|
|
private cg: CGroup;
|
||
|
|
|
||
|
|
constructor(controller: string, cg: CGroup) {
|
||
|
|
this.controller = controller;
|
||
|
|
this.cg = cg;
|
||
|
|
}
|
||
|
|
|
||
|
|
WriteValue(key: string, value: string) {
|
||
|
|
writeFileSync(path.join(this.cg.Path(), `${this.controller}.${key}`), value);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
export class CGroup {
|
||
|
|
private path;
|
||
|
|
|
||
|
|
constructor(path: string) {
|
||
|
|
this.path = path;
|
||
|
|
}
|
||
|
|
|
||
|
|
private InitControllers() {
|
||
|
|
// Configure this "root" cgroup to provide cpu and cpuset controllers to the leaf
|
||
|
|
// QEMU cgroups. A bit iffy but whatever.
|
||
|
|
writeFileSync(path.join(this.path, 'cgroup.subtree_control'), '+cpu +cpuset');
|
||
|
|
}
|
||
|
|
|
||
|
|
GetController(controller: string) {
|
||
|
|
return new CGroupController(controller, this);
|
||
|
|
}
|
||
|
|
|
||
|
|
Path(): string {
|
||
|
|
return this.path;
|
||
|
|
}
|
||
|
|
|
||
|
|
HasSubgroup(name: string): boolean {
|
||
|
|
let subgroup_root = path.join(this.path, name);
|
||
|
|
if (existsSync(subgroup_root)) return true;
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
// Gets a CGroup inside of this cgroup.
|
||
|
|
GetSubgroup(name: string): CGroup {
|
||
|
|
// make the subgroup if it doesn't already exist
|
||
|
|
let subgroup_root = path.join(this.path, name);
|
||
|
|
if (!this.HasSubgroup(name)) {
|
||
|
|
mkdirSync(subgroup_root);
|
||
|
|
// We need to make the subgroup threaded before we can attach a process to it.
|
||
|
|
// It's a bit weird, but oh well. Blame linux people, not me.
|
||
|
|
writeFileSync(path.join(subgroup_root, 'cgroup.type'), 'threaded');
|
||
|
|
}
|
||
|
|
return new CGroup(subgroup_root);
|
||
|
|
}
|
||
|
|
|
||
|
|
// Attaches a process to this cgroup.
|
||
|
|
AttachProcess(pid: number) {
|
||
|
|
appendFileSync(path.join(this.path, 'cgroup.procs'), pid.toString());
|
||
|
|
}
|
||
|
|
|
||
|
|
// Returns a CGroup instance for the process' current cgroup, prepared for subgroup usage.
|
||
|
|
// This will only fail if you are not using systemd or elogind,
|
||
|
|
// since even logind user sessions are run inside of a user@[UID] slice.
|
||
|
|
// NOTE: This only supports cgroups2-only systems. Systemd practically enforces that so /shrug
|
||
|
|
static Self(): CGroup {
|
||
|
|
const kCgroupSelfPath = '/proc/self/cgroup';
|
||
|
|
if (!existsSync(kCgroupSelfPath)) throw new Error('This process is not in a CGroup.');
|
||
|
|
let res = readFileSync(kCgroupSelfPath, { encoding: 'utf-8' });
|
||
|
|
|
||
|
|
// Make sure the first/only line is a cgroups2 0::/path/to/cgroup entry.
|
||
|
|
// Legacy cgroups1 is not supported.
|
||
|
|
if (res[0] != '0') throw new Error('CGroup.Self() does not work with cgroups 1 systems. Please do not the cgroups 1.');
|
||
|
|
let cg_path = res.substring(3, res.indexOf('\n'));
|
||
|
|
|
||
|
|
let cg = new CGroup(path.join('/sys/fs/cgroup', cg_path));
|
||
|
|
|
||
|
|
// Do root level group initalization
|
||
|
|
cg.InitControllers();
|
||
|
|
|
||
|
|
return cg;
|
||
|
|
}
|
||
|
|
}
|