#!/bin/sh
#
# Create mountpoint for Control Group v2
#

CGROUP2_CTRL_MEM="+memory "
CGROUP2_CTRL_CPU="+cpu "
CGROUP2_CTRL_IO="+io "
CGROUP2_CTRL_PID="+pids "

start() {
	printf "Mounting cgroupv2: "

	mount -t tmpfs cgroup /sys/fs/cgroup
	mount -t cgroup2 cgroup2 /sys/fs/cgroup
	echo "${CGROUP2_CTRL_MEM}${CGROUP2_CTRL_CPU}${CGROUP2_CTRL_IO}${CGROUP2_CTRL_PID}" \
		> /sys/fs/cgroup/cgroup.subtree_control
	[ $? = 0 ] && echo "OK" || echo "FAIL"
}

stop() {
	printf "Unmounting cgroupv2: "

	umount cgroup2 cgroup
	[ $? = 0 ] && echo "OK" || echo "FAIL"
}

case "$1" in
  start)
	start
	;;
  stop)
	stop
	;;
  restart|reload)
	stop
	start
	;;
  *)
	echo "Usage: $0 {start|stop|restart|reload}"
	exit 1
esac

exit $?