| #!/bin/sh |
| # SPDX-License-Identifier: GPL-2.0-or-later |
| # Copyright (c) 2015 SUSE |
| # Author: Cedric Hnyda <[email protected]> |
| # Usage |
| # ./pids.sh caseno max |
| |
| TST_CLEANUP=cleanup |
| TST_SETUP=setup |
| TST_TESTFUNC=do_test |
| TST_POS_ARGS=3 |
| TST_USAGE=usage |
| TST_NEEDS_ROOT=1 |
| TST_NEEDS_CMDS="killall" |
| |
| . tst_test.sh |
| |
| caseno=$1 |
| max=$2 |
| subcgroup_num=$3 |
| mounted=1 |
| |
| usage() |
| { |
| cat << EOF |
| usage: $0 caseno max_processes |
| |
| caseno - testcase number from interval 1-9 |
| max_processes - maximal number of processes to attach |
| (not applicable to testcase 5) |
| subcgroup_num - number of subgroups created in group |
| (only applicable to testcase 7) |
| OPTIONS |
| EOF |
| } |
| |
| cleanup() |
| { |
| killall -9 pids_task2 >/dev/null 2>&1 |
| |
| tst_res TINFO "removing created directories" |
| rmdir $testpath |
| if [ "$mounted" -ne "1" ]; then |
| tst_res TINFO "Umounting pids" |
| umount $mount_point |
| rmdir $mount_point |
| fi |
| } |
| |
| setup() |
| { |
| exist=`grep -w pids /proc/cgroups | cut -f1`; |
| if [ "$exist" = "" ]; then |
| tst_brk TCONF NULL "pids not supported" |
| fi |
| |
| mount_point=`grep -w pids /proc/mounts | cut -f 2 | cut -d " " -f2` |
| |
| if [ "$mount_point" = "" ]; then |
| mounted=0 |
| mount_point=/dev/cgroup |
| fi |
| |
| testpath=$mount_point/ltp_pids_$caseno |
| |
| if [ "$mounted" -eq "0" ]; then |
| ROD mkdir -p $mount_point |
| ROD mount -t cgroup -o pids none $mount_point |
| fi |
| ROD mkdir -p $testpath |
| } |
| |
| start_pids_tasks2() |
| { |
| start_pids_tasks2_path $testpath $1 |
| } |
| |
| start_pids_tasks2_path() |
| { |
| path=$1 |
| nb=$2 |
| for i in `seq 1 $nb`; do |
| pids_task2 & |
| echo $! > $path/tasks |
| done |
| |
| if [ $(cat "$path/tasks" | wc -l) -ne $nb ]; then |
| tst_brk TBROK "failed to attach process" |
| fi |
| } |
| |
| stop_pids_tasks() |
| { |
| stop_pids_tasks_path $testpath |
| } |
| |
| stop_pids_tasks_path() |
| { |
| local i |
| path=$1 |
| |
| for i in `cat $path/tasks`; do |
| ROD kill -9 $i |
| wait $i |
| done |
| } |
| |
| case1() |
| { |
| start_pids_tasks2 $max |
| |
| # should return 0 because there is no limit |
| pids_task1 "$testpath/tasks" |
| ret=$? |
| |
| if [ "$ret" -eq "2" ]; then |
| tst_res TFAIL "fork failed unexpectedly" |
| elif [ "$ret" -eq "0" ]; then |
| tst_res TPASS "fork didn't fail" |
| else |
| tst_res TBROK "pids_task1 failed" |
| fi |
| |
| stop_pids_tasks |
| } |
| |
| case2() |
| { |
| tmp=$((max - 1)) |
| tst_res TINFO "limit the number of pid to $max" |
| ROD echo $max \> $testpath/pids.max |
| |
| start_pids_tasks2 $tmp |
| |
| # should return 2 because the limit of pids is reached |
| pids_task1 "$testpath/tasks" |
| ret=$? |
| |
| if [ "$ret" -eq "2" ]; then |
| tst_res TPASS "fork failed as expected" |
| elif [ "$ret" -eq "0" ]; then |
| tst_res TFAIL "fork didn't fail despite the limit" |
| else |
| tst_res TBROK "pids_task1 failed" |
| fi |
| |
| stop_pids_tasks |
| } |
| |
| case3() |
| { |
| lim=$((max + 2)) |
| tst_res TINFO "limit the number of avalaible pid to $lim" |
| ROD echo $lim \> $testpath/pids.max |
| |
| start_pids_tasks2 $max |
| |
| pids_task1 "$testpath/tasks" |
| ret=$? |
| |
| if [ "$ret" -eq "2" ]; then |
| tst_res TFAIL "fork failed unexpectedly" |
| elif [ "$ret" -eq "0" ]; then |
| tst_res TPASS "fork worked as expected" |
| else |
| tst_res TBROK "pids_task1 failed" |
| fi |
| |
| stop_pids_tasks |
| } |
| |
| case4() |
| { |
| tst_res TINFO "limit the number of avalaible pid to 0" |
| ROD echo 0 \> $testpath/pids.max |
| |
| start_pids_tasks2 $max |
| |
| tst_res TPASS "all process were attached" |
| |
| stop_pids_tasks |
| } |
| |
| case5() |
| { |
| tst_res TINFO "try to limit the number of avalaible pid to -1" |
| echo -1 > $testpath/pids.max |
| |
| if [ "$?" -eq "0" ]; then |
| tst_res TFAIL "managed to set the limit to -1" |
| else |
| tst_res TPASS "didn't manage to set the limit to -1" |
| fi |
| } |
| |
| case6() |
| { |
| tst_res TINFO "set a limit that is smaller than current number of pids" |
| start_pids_tasks2 $max |
| |
| lim=$((max - 1)) |
| ROD echo $lim \> $testpath/pids.max |
| |
| pids_task1 "$testpath/tasks" |
| ret=$? |
| |
| if [ "$ret" -eq "2" ]; then |
| tst_res TPASS "fork failed as expected" |
| elif [ "$ret" -eq "0" ]; then |
| tst_res TFAIL "fork didn't fail despite the limit" |
| else |
| tst_res TBROK "pids_task1 failed" |
| fi |
| |
| stop_pids_tasks |
| } |
| |
| case7() |
| { |
| tst_res TINFO "the number of all child cgroup tasks larger than its parent limit" |
| |
| lim=$((max / subcgroup_num)) |
| if [ "$((lim * subcgroup_num))" -ne "$max" ]; then |
| tst_res TWARN "input max value must be a multiplier of $subcgroup_num" |
| return |
| fi |
| |
| ROD echo $max \> $testpath/pids.max |
| |
| for i in `seq 1 $subcgroup_num`; do |
| mkdir $testpath/child$i |
| start_pids_tasks2_path $testpath/child$i $lim |
| done |
| |
| pids_task1 "$testpath/tasks" |
| ret=$? |
| |
| if [ "$ret" -eq "2" ]; then |
| tst_res TPASS "parent cgroup fork failed as expected" |
| elif [ "$ret" -eq "0" ]; then |
| tst_res TFAIL "parent cgroup fork didn't fail despite the limit" |
| else |
| tst_res TBROK "parent cgroup pids_task1 failed" |
| fi |
| |
| for i in `seq 1 $subcgroup_num`; do |
| pids_task1 "$testpath/child$i/tasks" |
| ret=$? |
| |
| if [ "$ret" -eq "2" ]; then |
| tst_res TPASS "child$i cgroup fork failed as expected" |
| elif [ "$ret" -eq "0" ]; then |
| tst_res TFAIL "child$i cgroup fork didn't fail despite the limit" |
| else |
| tst_res TBROK "child$i cgroup pids_task1 failed" |
| fi |
| done |
| |
| for i in `seq 1 $subcgroup_num`; do |
| stop_pids_tasks_path $testpath/child$i |
| rmdir $testpath/child$i |
| done |
| |
| stop_pids_tasks |
| } |
| |
| case8() |
| { |
| tst_res TINFO "set child cgroup limit smaller than its parent limit" |
| ROD echo $max \> $testpath/pids.max |
| mkdir $testpath/child |
| |
| lim=$((max - 1)) |
| ROD echo $lim \> $testpath/child/pids.max |
| tmp=$((max - 2)) |
| start_pids_tasks2_path $testpath/child $tmp |
| |
| pids_task1 "$testpath/child/tasks" |
| ret=$? |
| |
| if [ "$ret" -eq "2" ]; then |
| tst_res TPASS "fork failed as expected" |
| elif [ "$ret" -eq "0" ]; then |
| tst_res TFAIL "fork didn't fail despite the limit" |
| else |
| tst_res TBROK "pids_task1 failed" |
| fi |
| |
| stop_pids_tasks_path $testpath/child |
| rmdir $testpath/child |
| } |
| |
| case9() |
| { |
| tst_res TINFO "migrate cgroup" |
| lim=$((max - 1)) |
| |
| for i in 1 2; do |
| mkdir $testpath/child$i |
| ROD echo $max \> $testpath/child$i/pids.max |
| start_pids_tasks2_path $testpath/child$i $lim |
| done |
| |
| pid=`head -n 1 $testpath/child1/tasks`; |
| ROD echo $pid \> $testpath/child2/tasks |
| |
| if grep -q "$pid" "$testpath/child2/tasks"; then |
| tst_res TPASS "migrate pid $pid from cgroup1 to cgroup2 as expected" |
| else |
| tst_res TPASS "migrate pid $pid from cgroup1 to cgroup2 failed" |
| fi |
| |
| if [ $(cat "$testpath/child1/pids.current") -eq $((lim - 1)) ]; then |
| tst_res TPASS "migrate child1 cgroup as expected" |
| else |
| tst_res TFAIL "migrate child1 cgroup failed" |
| fi |
| |
| if [ $(cat "$testpath/child2/pids.current") -eq $((lim + 1)) ]; then |
| tst_res TPASS "migrate child2 cgroup as expected" |
| else |
| tst_res TFAIL "migrate child2 cgroup failed" |
| fi |
| |
| pids_task1 "$testpath/child1/tasks" |
| ret=$? |
| |
| if [ "$ret" -eq "2" ]; then |
| tst_res TFAIL "child1 fork failed unexpectedly" |
| elif [ "$ret" -eq "0" ]; then |
| tst_res TPASS "child1 fork worked as expected" |
| else |
| tst_res TBROK "child1 pids_task1 failed" |
| fi |
| |
| pids_task1 "$testpath/child2/tasks" |
| ret=$? |
| |
| if [ "$ret" -eq "2" ]; then |
| tst_res TPASS "child2 fork failed as expected" |
| elif [ "$ret" -eq "0" ]; then |
| tst_res TFAIL "child2 fork didn't fail despite the limit" |
| else |
| tst_res TBROK "child2 pids_task1 failed" |
| fi |
| |
| for i in 1 2; do |
| stop_pids_tasks_path $testpath/child$i |
| rmdir $testpath/child$i |
| done |
| stop_pids_tasks |
| } |
| |
| do_test() |
| { |
| tst_res TINFO "Running testcase $caseno with $max processes" |
| case$caseno |
| } |
| |
| tst_run |
