From 999f45597a2b2461d43d05b3a4726c29d2eed0ac Mon Sep 17 00:00:00 2001
From: Ralph Ronnquist <ralph.ronnquist@gmail.com>
Date: Sat, 9 Apr 2022 09:47:21 +1000
Subject: [PATCH] added optional cpuset accounting

---
 functions    | 11 +++++++++++
 overlay-boot |  1 +
 2 files changed, 12 insertions(+)

diff --git a/functions b/functions
index 2cc9639..bc65448 100644
--- a/functions
+++ b/functions
@@ -166,3 +166,14 @@ is_live() {
 list_running() {
     pgrep -a overlay-boot | awk '{print $4}'
 }
+
+# Start cgroup v2 cpuset accounting if enabled.
+# Needs manual enabling, with:
+#    mount -t cgroup2 cgroup2 /sys/fs/cgroup
+setup_cgroup2_accounting() {
+    local NAME="$1" ME="$2"
+    local ACCDIR="$(awk '$3 == "cgroup2" {print $2; exit}' /proc/mounts)"
+    [ -z "$ACCDIR" ] && return 0
+    mkdir -p "$ACCDIR/$NAME"
+    echo "$ME" > $ACCDIR/$NAME/cgroup.procs
+}
diff --git a/overlay-boot b/overlay-boot
index 44bc76f..fc4b8b4 100755
--- a/overlay-boot
+++ b/overlay-boot
@@ -23,6 +23,7 @@ fi
 subhost_config
 setup_veth_cables $NAME $CABLES
 setup_overlay "$NAME" "$LIVE" "$LOWER" "$UPPER" "$WORK"
+setup_cgroup2_accounting "$NAME" "$$"
 
 exithandler() {
     ip netns del $NAME
-- 
2.39.2