tjh.dev
Linux kernel mirror (for testing)
git.kernel.org/pub/scm/linux/kernel/git/torvalds/linux.git
kernel
os
linux
1#!/bin/bash
2# SPDX-License-Identifier: GPL-2.0-only */
3#
4# Wrapper script which performs setup and cleanup for nx_huge_pages_test.
5# Makes use of root privileges to set up huge pages and KVM module parameters.
6#
7# Copyright (C) 2022, Google LLC.
8
9set -e
10
11NX_HUGE_PAGES=$(cat /sys/module/kvm/parameters/nx_huge_pages)
12NX_HUGE_PAGES_RECOVERY_RATIO=$(cat /sys/module/kvm/parameters/nx_huge_pages_recovery_ratio)
13NX_HUGE_PAGES_RECOVERY_PERIOD=$(cat /sys/module/kvm/parameters/nx_huge_pages_recovery_period_ms)
14HUGE_PAGES=$(cat /sys/kernel/mm/hugepages/hugepages-2048kB/nr_hugepages)
15
16# If we're already root, the host might not have sudo.
17if [ $(whoami) == "root" ]; then
18 function do_sudo () {
19 "$@"
20 }
21else
22 function do_sudo () {
23 sudo "$@"
24 }
25fi
26
27set +e
28
29function sudo_echo () {
30 echo "$1" | do_sudo tee -a "$2" > /dev/null
31}
32
33NXECUTABLE="$(dirname $0)/nx_huge_pages_test"
34
35sudo_echo test /dev/null || exit 4 # KSFT_SKIP=4
36
37(
38 set -e
39
40 sudo_echo 1 /sys/module/kvm/parameters/nx_huge_pages
41 sudo_echo 1 /sys/module/kvm/parameters/nx_huge_pages_recovery_ratio
42 sudo_echo 100 /sys/module/kvm/parameters/nx_huge_pages_recovery_period_ms
43 sudo_echo "$(( $HUGE_PAGES + 3 ))" /sys/kernel/mm/hugepages/hugepages-2048kB/nr_hugepages
44
45 # Test with reboot permissions
46 if [ $(whoami) == "root" ] || sudo setcap cap_sys_boot+ep $NXECUTABLE 2> /dev/null; then
47 echo Running test with CAP_SYS_BOOT enabled
48 $NXECUTABLE -t 887563923 -p 100 -r
49 test $(whoami) == "root" || sudo setcap cap_sys_boot-ep $NXECUTABLE
50 else
51 echo setcap failed, skipping nx_huge_pages_test with CAP_SYS_BOOT enabled
52 fi
53
54 # Test without reboot permissions
55 if [ $(whoami) != "root" ] ; then
56 echo Running test with CAP_SYS_BOOT disabled
57 $NXECUTABLE -t 887563923 -p 100
58 else
59 echo Running as root, skipping nx_huge_pages_test with CAP_SYS_BOOT disabled
60 fi
61)
62RET=$?
63
64sudo_echo "$NX_HUGE_PAGES" /sys/module/kvm/parameters/nx_huge_pages
65sudo_echo "$NX_HUGE_PAGES_RECOVERY_RATIO" /sys/module/kvm/parameters/nx_huge_pages_recovery_ratio
66sudo_echo "$NX_HUGE_PAGES_RECOVERY_PERIOD" /sys/module/kvm/parameters/nx_huge_pages_recovery_period_ms
67sudo_echo "$HUGE_PAGES" /sys/kernel/mm/hugepages/hugepages-2048kB/nr_hugepages
68
69exit $RET