tests: add LD_PRELOAD free alternative for nproc-quota.sh

* tests/nproc/nproc-quota-systemd.sh: Add a new root test.
* tests/local.mk: Reference the new test.

diff --git a/tests/local.mk b/tests/local.mk
index 92be1c420723d05531f6908b327e7964b2ce5d42..33527d37954a9871360a18cce83a01072b30ca18 100644 (file)
--- a/tests/local.mk
+++ b/tests/local.mk
@@ -127,6 +127,7 @@ all_root_tests =                            \
   tests/ls/no-cap.sh                           \
   tests/ls/nameless-uid.sh                     \
   tests/nproc/nproc-quota.sh                   \
+  tests/nproc/nproc-quota-systemd.sh                   \
   tests/chcon/chcon.sh                         \
   tests/chroot/chroot-credentials.sh           \
   tests/misc/selinux.sh                                \

diff --git a/tests/nproc/nproc-quota-systemd.sh b/tests/nproc/nproc-quota-systemd.sh
new file mode 100755 (executable)
index 0000000..07e6de9
--- /dev/null
+++ b/tests/nproc/nproc-quota-systemd.sh
@@ -0,0 +1,55 @@
+#!/bin/sh
+# Ensure that "nproc" honors cgroup quotas and kernel scheduler from systemd
+
+# Copyright (C) 2025 Free Software Foundation, Inc.
+
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+
+# You should have received a copy of the GNU General Public License
+# along with this program.  If not, see <https://www.gnu.org/licenses/>.
+
+. "${srcdir=.}/tests/init.sh"; path_prepend_ ./src
+print_ver_ nproc
+require_root_
+
+systemd-run --version || skip_ 'systemd-run not available'
+
+all=$(nproc --all)
+test ${all} != 1 || skip_ 'Single CPU detected'
+
+check_sched_quota()
+{
+  cpu=$1; shift
+  pct=$1; shift
+
+  if systemd-run --scope -q -p CPUQuota=$pct chrt $* true; then
+    test $(systemd-run --scope -q -p CPUQuota=$pct chrt $* nproc) = $cpu ||
+     return 1
+  fi
+  return 0
+}
+
+check_sched_quota 1 100% --other 0 || fail=1
+check_sched_quota 1 149% --other 0 || fail=1
+check_sched_quota 2 150% --other 0 || fail=1
+check_sched_quota 2 249% --other 0 || fail=1
+
+check_sched_quota 1 100% --idle 0 || fail=1
+check_sched_quota 1 100% --batch 0 || fail=1
+
+# some schedulers should use all threads
+check_sched_quota $all 100% --fifo 1 || fail=1
+check_sched_quota $all 100% --rr 1 || fail=1
+check_sched_quota $all 100% --deadline \
+ --sched-runtime 100000000 --sched-deadline 1000000000 \
+ --sched-period 1000000000 0 || fail=1
+
+Exit $fail