From: Stéphane Graber <stgraber@ubuntu.com>
Date: Thu, 13 Mar 2014 03:15:33 +0000 (-0400)
Subject: tests: Support cgmanager
X-Git-Tag: lxc-1.1.0.alpha1~209
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=3f458ed00479be6001b4e137b63b76c27fe7d8fd;p=thirdparty%2Flxc.git

tests: Support cgmanager

When cgmanager is around, use dbus-send to setup the cgroups, this
allows the tests to work in a container without cgroupfs access.

Signed-off-by: Stéphane Graber <stgraber@ubuntu.com>
Acked-by: Serge E. Hallyn <serge.hallyn@ubuntu.com>
---

diff --git a/src/tests/lxc-test-unpriv b/src/tests/lxc-test-unpriv
index cc0a8609b..0108b2d3c 100755
--- a/src/tests/lxc-test-unpriv
+++ b/src/tests/lxc-test-unpriv
@@ -93,11 +93,27 @@ chown -R $TUSER /run/user/$(id -u $TUSER)
 
 cd $HDIR
 
-for d in /sys/fs/cgroup/*; do
-	[ ! -d $d/lxctest ] && mkdir $d/lxctest
-	chown -R $TUSER $d/lxctest
-	echo $$ > $d/lxctest/tasks
-done
+if [ -e /sys/fs/cgroup/cgmanager/sock ]; then
+	for d in hugetlb cpuset cpu cpuacct memory devices freezer blkio perf_event; do
+		dbus-send --print-reply --address=unix:path=/sys/fs/cgroup/cgmanager/sock \
+			--type=method_call /org/linuxcontainers/cgmanager org.linuxcontainers.cgmanager0_0.Create \
+			string:$d string:$TUSER >/dev/null
+
+		dbus-send --print-reply --address=unix:path=/sys/fs/cgroup/cgmanager/sock \
+			--type=method_call /org/linuxcontainers/cgmanager org.linuxcontainers.cgmanager0_0.Chown \
+			string:$d string:$TUSER int32:$(id -u $TUSER) int32:$(id -g $TUSER) >/dev/null
+
+		dbus-send --print-reply --address=unix:path=/sys/fs/cgroup/cgmanager/sock \
+			--type=method_call /org/linuxcontainers/cgmanager org.linuxcontainers.cgmanager0_0.MovePid \
+			string:$d string:$TUSER int32:$$ >/dev/null
+	done
+else
+	for d in /sys/fs/cgroup/*; do
+		[ ! -d $d/lxctest ] && mkdir $d/lxctest
+		chown -R $TUSER $d/lxctest
+		echo $$ > $d/lxctest/tasks
+	done
+fi
 
 run_cmd lxc-create -t download -n c1 -- -d ubuntu -r trusty -a $ARCH
 run_cmd lxc-start -n c1 -d
diff --git a/src/tests/lxc-test-usernic.in b/src/tests/lxc-test-usernic.in
index 0a514ee47..3618beecd 100755
--- a/src/tests/lxc-test-usernic.in
+++ b/src/tests/lxc-test-usernic.in
@@ -86,11 +86,27 @@ lxc.id_map = u 0 910000 10000
 lxc.id_map = g 0 910000 10000
 EOF
 
-for d in /sys/fs/cgroup/*; do
-	[ ! -d $d/lxctest ] && mkdir $d/lxctest
-	chown -R usernic-user $d/lxctest
-	echo $$ > $d/lxctest/tasks
-done
+if [ -e /sys/fs/cgroup/cgmanager/sock ]; then
+	for d in hugetlb cpuset cpu cpuacct memory devices freezer blkio perf_event; do
+		dbus-send --print-reply --address=unix:path=/sys/fs/cgroup/cgmanager/sock \
+			--type=method_call /org/linuxcontainers/cgmanager org.linuxcontainers.cgmanager0_0.Create \
+			string:$d string:usernic-user >/dev/null
+
+		dbus-send --print-reply --address=unix:path=/sys/fs/cgroup/cgmanager/sock \
+			--type=method_call /org/linuxcontainers/cgmanager org.linuxcontainers.cgmanager0_0.Chown \
+			string:$d string:usernic-user int32:$(id -u usernic-user) int32:$(id -g usernic-user) >/dev/null
+
+		dbus-send --print-reply --address=unix:path=/sys/fs/cgroup/cgmanager/sock \
+			--type=method_call /org/linuxcontainers/cgmanager org.linuxcontainers.cgmanager0_0.MovePid \
+			string:$d string:usernic-user int32:$$ >/dev/null
+	done
+else
+	for d in /sys/fs/cgroup/*; do
+		[ ! -d $d/lxctest ] && mkdir $d/lxctest
+		chown -R usernic-user $d/lxctest
+		echo $$ > $d/lxctest/tasks
+	done
+fi
 
 mkdir -p /run/user/$(id -u usernic-user)
 chown -R usernic-user /run/user/$(id -u usernic-user) /home/usernic-user