sliceweasel: WIP on cpu slicing

1 files changed, 19 insertions, 3 deletions

diff --git a/dot/local/bin/sliceweasel b/dot/local/bin/sliceweasel
index 1ec45db..c6f07d0 100644..100755
--- a/dot/local/bin/sliceweasel
+++ b/dot/local/bin/sliceweasel
@@ -18,12 +18,17 @@ memory_limit_in_bytes=$(bc -q <<< "$total_memory * $memory_ratio / 1" ) || die "
 
 init_control_group()
 {
-    control_group_dir=/sys/fs/cgroup/memory/"$control_group"
+    control_group_dir=/sys/fs/cgroup/"$1"/"$control_group"
     [ -d "$control_group_dir" ] && return
     $as_root mkdir "$control_group_dir" || die "mkdir failed"
     set_memory_limit $memory_limit_in_bytes || die "set_memory_limit failed"
 }
 
+get_cpu_limit()
+{
+    false
+}
+
 get_memory_limit()
 {
     cat "$control_group_dir"/memory.limit_in_bytes
@@ -60,7 +65,7 @@ get_cgroup()
 
 launch_command()
 {
-    init_control_group                      || die "init_control_group failed"
+    init_control_group memory               || die "init_control_group failed"
     cg=$(get_cgroup memory)                 || die "failed to ascertain current control group"
     [ "$cg" = / ]                           || die "current control group should be '/', but is instead '$cg'"
     join_control_group                      || die "join_control_group failed"
@@ -75,9 +80,19 @@ pretty_show_memory_limit()
            "$((memory_limit * 100 / total_memory))"
 }
 
+recpu_control_group()
+{
+    init_control_group cpu
+    case "$1" in
+        [0-9]*%) set_cpu_limit $((total_cpu * "${1%\%}" / 100)) ;;
+        [0-9]*) set_cpu_limit "$1" ;;
+        [-+*/][0-9]*) set_cpu_limit $(( $(get_cpu_limit) "$1" )) ;;
+    esac
+}
+
 resize_control_group()
 {
-    init_control_group
+    init_control_group memory
     case "$1" in
         '') pretty_show_memory_limit >&2 ;;
         [0-9]*%) set_memory_limit $((total_memory * "${1%\%}" / 100)) ;;
@@ -98,6 +113,7 @@ case "$1" in
     --) shift; launch_command "$@" ;;
     --resize) resize_control_group "$2" ;;
     --resize=*) resize_control_group "${1#--resize=}" ;;
+    --cpu=*) recpu_control_group "${1#--cpu=}"
     --*) usage; [ "$1" = '--help' -o "$1" = '--usage' ] ;;
     *) launch_command "$@" ;;
 esac