From: Guopeng Zhang <[email protected]>

Return the last housekeeping CPU from a root child to an isolated parent.
Check that the parent becomes invalid and releases its CPUs instead of
adding the returned CPU to the isolated mask.

Repeat the test with two isolated ancestors and check that the outermost
isolated partition is invalidated.

Add the nested test cgroup to the global cleanup list so a failed
assertion does not leave the test hierarchy behind.

Signed-off-by: Guopeng Zhang <[email protected]>
---
 .../selftests/cgroup/test_cpuset_prs.sh       | 149 +++++++++++++++++-
 1 file changed, 148 insertions(+), 1 deletion(-)

diff --git a/tools/testing/selftests/cgroup/test_cpuset_prs.sh 
b/tools/testing/selftests/cgroup/test_cpuset_prs.sh
index a3df6b094e0f..99cba4e5277d 100755
--- a/tools/testing/selftests/cgroup/test_cpuset_prs.sh
+++ b/tools/testing/selftests/cgroup/test_cpuset_prs.sh
@@ -122,7 +122,8 @@ cleanup()
 {
        online_cpus
        cd $CGROUP2
-       rmdir A1/A2/A3 A1/A2 A1 B1 test/A1 test/B1 test > /dev/null 2>&1
+       rmdir A1/A2/A3 A1/A2 A1 B1 \
+             test/A1/A2 test/A1 test/B1 test > /dev/null 2>&1
        rmdir rtest/p1/c11 rtest/p1/c12 rtest/p2/c21 \
              rtest/p2/c22 rtest/p1 rtest/p2 rtest > /dev/null 2>&1
        [[ -n "$SCHED_DEBUG" ]] &&
@@ -1222,6 +1223,30 @@ test_boot_isolated()
        echo "$TEST_NAME test PASSED."
 }
 
+# Return success if CPU $2 is present in cpulist $1.
+cpulist_contains()
+{
+       local CPULIST_ARG=$1
+       local CPU_ARG=$2
+       local ITEM FIRST LAST
+       local OLD_IFS=$IFS
+
+       IFS=,
+       for ITEM in $CPULIST_ARG
+       do
+               FIRST=${ITEM%-*}
+               LAST=${ITEM#*-}
+               [[ $ITEM != *-* ]] && LAST=$FIRST
+               if [[ $CPU_ARG -ge $FIRST && $CPU_ARG -le $LAST ]]
+               then
+                       IFS=$OLD_IFS
+                       return 0
+               fi
+       done
+       IFS=$OLD_IFS
+       return 1
+}
+
 #
 # A parent's type and CPU-mask changes must check only CPUs owned directly by
 # the parent, not a boot-isolated CPU owned by a valid child partition.
@@ -1390,6 +1415,127 @@ test_trial_child_invalidation()
        echo "$TEST_NAME test PASSED."
 }
 
+#
+# Returning the last housekeeping CPU to an isolated parent must invalidate
+# the isolated partition instead of adding that CPU to the isolated mask.
+#
+test_housekeeping_cpu_return()
+{
+       TEST_NAME="Housekeeping CPU return"
+       NOHZ_FILE=/sys/devices/system/cpu/nohz_full
+       [[ -r $NOHZ_FILE ]] || {
+               echo "$TEST_NAME test SKIPPED: no nohz_full state"
+               return 0
+       }
+       NOHZ_CPUS=$(cat $NOHZ_FILE)
+       [[ -n "$NOHZ_CPUS" && "$NOHZ_CPUS" != "(null)" ]] || {
+               echo "$TEST_NAME test SKIPPED: no nohz_full CPUs"
+               return 0
+       }
+
+       HK_CPU=
+       HK_COUNT=0
+       TYPE_CPUS=()
+       for ((CPU=0; CPU < NR_CPUS; CPU++))
+       do
+               CPU_ONLINE=/sys/devices/system/cpu/cpu${CPU}/online
+               [[ ! -e $CPU_ONLINE || $(cat $CPU_ONLINE) -eq 1 ]] || continue
+               cpulist_contains "$BOOT_CPUS" $CPU && continue
+               if cpulist_contains "$NOHZ_CPUS" $CPU
+               then
+                       [[ ${#TYPE_CPUS[@]} -lt 3 ]] && TYPE_CPUS+=("$CPU")
+               else
+                       HK_CPU=$CPU
+                       ((HK_COUNT++))
+               fi
+       done
+
+       [[ $HK_COUNT -eq 1 && ${#TYPE_CPUS[@]} -ge 2 ]] || {
+               echo "$TEST_NAME test SKIPPED: requires one full housekeeping 
CPU"
+               return 0
+       }
+       echo "Running $TEST_NAME test ..."
+
+       cd $CGROUP2/test
+       echo member > cpuset.cpus.partition
+       echo +cpuset > cgroup.subtree_control
+       echo $HK_CPU,${TYPE_CPUS[0]},${TYPE_CPUS[1]} > cpuset.cpus
+       test_partition root
+       mkdir A1
+       cd A1
+       echo $HK_CPU > cpuset.cpus
+       test_partition root
+       cd ..
+       test_partition isolated
+       cd A1
+       test_partition member
+       cd ..
+       grep -q '^isolated invalid (partition config conflicts with 
housekeeping setup)$' \
+               cpuset.cpus.partition || {
+               echo "Isolated parent remained valid after housekeeping CPU 
return"
+               exit 1
+       }
+       [[ $(cat $CGROUP2/cpuset.cpus.effective) = "$CPULIST" ]] || {
+               echo "Housekeeping CPU return did not release the partition 
CPUs"
+               exit 1
+       }
+       check_isolcpus "." || {
+               echo "Housekeeping CPU was added to the isolated mask"
+               exit 1
+       }
+       rmdir A1
+       echo member > cpuset.cpus.partition
+       echo "" > cpuset.cpus
+
+       if [[ ${#TYPE_CPUS[@]} -lt 3 ]]
+       then
+               echo "Nested $TEST_NAME test SKIPPED: requires three nohz_full 
CPUs"
+               cd $CGROUP2
+               echo "$TEST_NAME test PASSED."
+               return 0
+       fi
+
+       # Repeat the check with two isolated ancestors.
+       echo $HK_CPU,${TYPE_CPUS[0]},${TYPE_CPUS[1]},${TYPE_CPUS[2]} > 
cpuset.cpus
+       test_partition root
+       mkdir A1
+       cd A1
+       echo $HK_CPU,${TYPE_CPUS[1]},${TYPE_CPUS[2]} > cpuset.cpus
+       test_partition root
+       echo +cpuset > cgroup.subtree_control
+       mkdir A2
+       cd A2
+       echo $HK_CPU > cpuset.cpus
+       test_partition root
+       cd ..
+       test_partition isolated
+       cd ..
+       test_partition isolated
+       cd A1/A2
+       test_partition member
+       cd ../..
+       grep -q '^isolated invalid (partition config conflicts with 
housekeeping setup)$' \
+               cpuset.cpus.partition || {
+               echo "Outermost isolated partition remained valid after 
housekeeping CPU return"
+               exit 1
+       }
+       [[ $(cat $CGROUP2/cpuset.cpus.effective) = "$CPULIST" ]] || {
+               echo "Nested housekeeping CPU return did not release the 
partition CPUs"
+               exit 1
+       }
+       check_isolcpus "." || {
+               echo "Nested housekeeping CPU return added the CPU to the 
isolated mask"
+               exit 1
+       }
+       rmdir A1/A2
+       echo member > A1/cpuset.cpus.partition
+       rmdir A1
+       echo member > cpuset.cpus.partition
+       echo "" > cpuset.cpus
+       cd $CGROUP2
+       echo "$TEST_NAME test PASSED."
+}
+
 #
 # Wait for inotify event for the given file and read it
 # $1: cgroup file to wait for
@@ -1464,5 +1610,6 @@ test_isolated
 test_boot_isolated
 test_child_owned_cpus
 test_trial_child_invalidation
+test_housekeeping_cpu_return
 test_inotify
 echo "All tests PASSED."
-- 
2.43.0


Reply via email to