70 lines · bash
1#!/bin/bash2# 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 -e10 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" ]; then18 function do_sudo () {19 "$@"20 }21else22 function do_sudo () {23 sudo "$@"24 }25fi26 27set +e28 29function sudo_echo () {30 echo "$1" | do_sudo tee -a "$2" > /dev/null31}32 33NXECUTABLE="$(dirname $0)/nx_huge_pages_test"34 35sudo_echo test /dev/null || exit 4 # KSFT_SKIP=436 37(38 set -e39 40 sudo_echo 1 /sys/module/kvm/parameters/nx_huge_pages41 sudo_echo 1 /sys/module/kvm/parameters/nx_huge_pages_recovery_ratio42 sudo_echo 100 /sys/module/kvm/parameters/nx_huge_pages_recovery_period_ms43 sudo_echo "$(( $HUGE_PAGES + 3 ))" /sys/kernel/mm/hugepages/hugepages-2048kB/nr_hugepages44 45 # Test with reboot permissions46 if [ $(whoami) == "root" ] || sudo setcap cap_sys_boot+ep $NXECUTABLE 2> /dev/null; then47 echo Running test with CAP_SYS_BOOT enabled48 $NXECUTABLE -t 887563923 -p 100 -r49 test $(whoami) == "root" || sudo setcap cap_sys_boot-ep $NXECUTABLE50 else51 echo setcap failed, skipping nx_huge_pages_test with CAP_SYS_BOOT enabled52 fi53 54 # Test without reboot permissions55 if [ $(whoami) != "root" ] ; then56 echo Running test with CAP_SYS_BOOT disabled57 $NXECUTABLE -t 887563923 -p 10058 else59 echo Running as root, skipping nx_huge_pages_test with CAP_SYS_BOOT disabled60 fi61)62RET=$?63 64sudo_echo "$NX_HUGE_PAGES" /sys/module/kvm/parameters/nx_huge_pages65sudo_echo "$NX_HUGE_PAGES_RECOVERY_RATIO" /sys/module/kvm/parameters/nx_huge_pages_recovery_ratio66sudo_echo "$NX_HUGE_PAGES_RECOVERY_PERIOD" /sys/module/kvm/parameters/nx_huge_pages_recovery_period_ms67sudo_echo "$HUGE_PAGES" /sys/kernel/mm/hugepages/hugepages-2048kB/nr_hugepages68 69exit $RET70