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 70