tools/testing/selftests/kvm/x86_64/nx_huge_pages_test.sh

   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
   9 set -e
  10
  11 NX_HUGE_PAGES=$(cat /sys/module/kvm/parameters/nx_huge_pages)
  12 NX_HUGE_PAGES_RECOVERY_RATIO=$(cat /sys/module/kvm/parameters/nx_huge_pages_recovery_ratio)
  13 NX_HUGE_PAGES_RECOVERY_PERIOD=$(cat /sys/module/kvm/parameters/nx_huge_pages_recovery_period_ms)
  14 HUGE_PAGES=$(cat /sys/kernel/mm/hugepages/hugepages-2048kB/nr_hugepages)
  15
  16 # If we're already root, the host might not have sudo.
  17 if [ $(whoami) == "root" ]; then
  18         function do_sudo () {
  19                 "$@"
  20         }
  21 else
  22         function do_sudo () {
  23                 sudo "$@"
  24         }
  25 fi
  26
  27 set +e
  28
  29 function sudo_echo () {
  30         echo "$1" | do_sudo tee -a "$2" > /dev/null
  31 }
  32
  33 NXECUTABLE="$(dirname $0)/nx_huge_pages_test"
  34
  35 sudo_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 )
  62 RET=$?
  63
  64 sudo_echo "$NX_HUGE_PAGES" /sys/module/kvm/parameters/nx_huge_pages
  65 sudo_echo "$NX_HUGE_PAGES_RECOVERY_RATIO" /sys/module/kvm/parameters/nx_huge_pages_recovery_ratio
  66 sudo_echo "$NX_HUGE_PAGES_RECOVERY_PERIOD" /sys/module/kvm/parameters/nx_huge_pages_recovery_period_ms
  67 sudo_echo "$HUGE_PAGES" /sys/kernel/mm/hugepages/hugepages-2048kB/nr_hugepages
  68
  69 exit $RET