Use sane variables instead of full paths

Signed-off-by: Dhaval Giani <dhaval.giani@gmail.com>

Author: giani

setup-cgroups.sh

@@ -3,20 +3,22 @@
 
 user=$1
 uid=`id -u`
+cgroupdir=/sys/fs/cgroup/memory
+testdir=test
 
 if [ $uid != 0 ]
 then
 	echo "Need to be root"
 	exit
 fi
 
-if [ ! -d /sys/fs/cgroup/memory ]
+if [ ! -d $cgroupdir ]
 then
 	echo "Non-standard cgroup configuration"
 	exit
 fi
 
-mkdir /sys/fs/cgroup/memory/test
-echo 1M > /sys/fs/cgroup/memory/test/memory.limit_in_bytes
+mkdir $cgroupdir/$testdir
+echo 1M > $cgroupdir/$testdir/memory.limit_in_bytes
 
-chown -R $user /sys/fs/cgroup/memory/test 
+chown -R $user $cgroupdir/$testdir 

setup.sh

@@ -1,13 +1,15 @@
 #!/bin/bash
 
+cgroupdir=/sys/fs/cgroup/memory
+testdir=test
 curr_user=`id | cut -d "(" -f 2 | cut -d ")" -f 1`
 
 echo $curr_user
 
 echo "Now we need root"
 sudo ./setup-cgroups.sh $curr_user
 
-echo $$ > /sys/fs/cgroup/memory/test/tasks
+echo $$ > $cgroupdir/$testdir/tasks
 
 echo "Now we are trapped in the correct cgroup"
 cat /proc/self/cgroup | grep memory

teardown-cgroup.sh

@@ -1,28 +1,30 @@
 #!/bin/bash
 
 uid=`id -u`
+cgroupdir=/sys/fs/cgroup/memory
+testdir=test
 
 if [ $uid != 0 ]
 then
 	echo "Need to be root"
 	exit
 fi
 
-if [ ! -d /sys/fs/cgroup/memory ]
+if [ ! -d $cgroupdir ]
 then
 	echo "Non-standard cgroup configuration"
 	exit
 fi
 
-if [ ! -d /sys/fs/cgroup/memory/test ]
+if [ ! -d $cgroupdir/$testdir ]
 then
 	echo "Cgroups not setup"
 	exit
 fi
 
-for pid in `cat /sys/fs/cgroup/memory/test/tasks`
+for pid in `cat $cgroupdir/$testdir/tasks`
 do
-	echo $pid > /sys/fs/cgroup/memory/tasks
+	echo $pid > $cgroupdir/tasks
 done
 
-rmdir /sys/fs/cgroup/memory/test/
+rmdir $cgroupdir/$testdir