tools/testing/selftests/bpf/test_kmod.sh

   1 #!/bin/sh
   2 # SPDX-License-Identifier: GPL-2.0
   3
   4 # Kselftest framework requirement - SKIP code is 4.
   5 ksft_skip=4
   6
   7 msg="skip all tests:"
   8 if [ "$(id -u)" != "0" ]; then
   9         echo $msg please run this as root >&2
  10         exit $ksft_skip
  11 fi
  12
  13 SRC_TREE=../../../../
  14
  15 test_run()
  16 {
  17         sysctl -w net.core.bpf_jit_enable=$1 2>&1 > /dev/null
  18         sysctl -w net.core.bpf_jit_harden=$2 2>&1 > /dev/null
  19
  20         echo "[ JIT enabled:$1 hardened:$2 ]"
  21         dmesg -C
  22         insmod $SRC_TREE/lib/test_bpf.ko 2> /dev/null
  23         if [ $? -ne 0 ]; then
  24                 rc=1
  25         fi
  26         rmmod  test_bpf 2> /dev/null
  27         dmesg | grep FAIL
  28 }
  29
  30 test_save()
  31 {
  32         JE=`sysctl -n net.core.bpf_jit_enable`
  33         JH=`sysctl -n net.core.bpf_jit_harden`
  34 }
  35
  36 test_restore()
  37 {
  38         sysctl -w net.core.bpf_jit_enable=$JE 2>&1 > /dev/null
  39         sysctl -w net.core.bpf_jit_harden=$JH 2>&1 > /dev/null
  40 }
  41
  42 rc=0
  43 test_save
  44 test_run 0 0
  45 test_run 1 0
  46 test_run 1 1
  47 test_run 1 2
  48 test_restore
  49 exit $rc