[gentoo-commits] proj/openrc:master commit in: sh/, etc/

[William Hubbs]

commit:     20df56f0b2a01bd2f8a70b8fb3fbd051c1aab8d8
Author:     Patrick Lauer <patrick <AT> gentoo <DOT> org>
AuthorDate: Sat Nov 12 21:19:55 2011 +0000
Commit:     William Hubbs <williamh <AT> gentoo <DOT> org>
CommitDate: Thu Nov 17 22:22:59 2011 +0000
URL:        http://git.overlays.gentoo.org/gitweb/?p=proj/openrc.git;a=commit;h=20df56f0

Add CGroup support

X-Gentoo-Bug: 390297
X-Gentoo-Bug: http://bugs.gentoo.org/show_bug.cgi?id=390297

---
 etc/rc.conf.Linux  |    7 +++++++
 sh/runscript.sh.in |   22 ++++++++++++++++++++++
 2 files changed, 29 insertions(+), 0 deletions(-)

diff --git a/etc/rc.conf.Linux b/etc/rc.conf.Linux
index 9b5859a..585da3c 100644
--- a/etc/rc.conf.Linux
+++ b/etc/rc.conf.Linux
@@ -1,3 +1,10 @@
+# Set the control group for this service.
+# If you do not set this, the default setting is the value of
+# RC_SVCNAME.
+# This setting is ignored if you do not have CONFIG_CGROUPS active in
+# your kernel.
+# RC_CGROUP="foo"
+
 ##############################################################################
 # LINUX SPECIFIC OPTIONS
 

diff --git a/sh/runscript.sh.in b/sh/runscript.sh.in
index f86a510..73b4a59 100644
--- a/sh/runscript.sh.in
+++ b/sh/runscript.sh.in
@@ -189,6 +189,16 @@ fi
 # Load any system overrides
 sourcex -e "@SYSCONFDIR@/rc.conf"
 
+# Attach to CGroup - dir existing is enough for us
+if [ -d /sys/fs/cgroup/ ]; then
+	# use RC_SVCNAME unless overridden in conf.d
+	SVC_CGROUP=${RC_CGROUP:-$RC_SVCNAME}
+	mkdir -p /sys/fs/cgroup/${SVC_CGROUP}
+	# now attach self to cgroup - any children of this process will inherit this
+	echo $$ > /sys/fs/cgroup/${SVC_CGROUP}/tasks
+	# TODO: set res limits from conf.d
+fi
+
 # Apply any ulimit defined
 [ -n "${rc_ulimit:-$RC_ULIMIT}" ] && ulimit ${rc_ulimit:-$RC_ULIMIT}
 
@@ -291,3 +301,15 @@ while [ -n "$1" ]; do
 	eerror "$RC_SVCNAME: unknown function \`$1'"
 	exit 1
 done
+
+# CGroup cleanup
+if [ -d /sys/fs/cgroup/ ]; then
+	# use RC_SVCNAME unless overridden in conf.d
+	SVC_CGROUP=${RC_CGROUP:-$RC_SVCNAME}
+	# reattach to root cgroup
+	echo $$ > /sys/fs/cgroup/tasks
+	# remove cgroup if empty, will fail if any task attached
+	rmdir  /sys/fs/cgroup/${SVC_CGROUP} 2>/dev/null
+fi
+
+exit 0