X-Git-Url: https://git.rrq.au/?a=blobdiff_plain;f=functions;h=bc65448082cb4c697df37d2d6f8e82f2bc5a5a4f;hb=999f45597a2b2461d43d05b3a4726c29d2eed0ac;hp=45a634a1e575395e86c8457b342e3b2a39d5cc98;hpb=f86ce1827c0f74b3ac15598946b41477dd5a4066;p=rrq%2Foverlay-boot.git

diff --git a/functions b/functions
index 45a634a..bc65448 100644
--- a/functions
+++ b/functions
@@ -3,6 +3,8 @@
 # Rerun with sudo if needed
 [ $(id -u) = 0 ] || exec sudo $0 $@
 
+export ACTION="$(basename $0)"
+
 # Function to write a message and exit with error code
 die() {
     echo "$*" >&2
@@ -164,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
+}