From: oech3 <79379754+oech3@users.noreply.github.com>
Date: Tue, 30 Dec 2025 06:53:31 +0000 (+0900)
Subject: tests: add LD_PRELOAD free alternative for nproc-quota.sh
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=3f0ab90497153a3db0b254104a98c85e5666600b;p=thirdparty%2Fcoreutils.git

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 92be1c4207..33527d3795 100644
--- 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
index 0000000000..07e6de93e7
--- /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