tools/testing/selftests/bpf/test_bpftool_metadata.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 TESTNAME=bpftool_metadata
   8 BPF_FS=$(awk '$3 == "bpf" {print $2; exit}' /proc/mounts)
   9 BPF_DIR=$BPF_FS/test_$TESTNAME
  10
  11 _cleanup()
  12 {
  13         set +e
  14         rm -rf $BPF_DIR 2> /dev/null
  15 }
  16
  17 cleanup_skip()
  18 {
  19         echo "selftests: $TESTNAME [SKIP]"
  20         _cleanup
  21
  22         exit $ksft_skip
  23 }
  24
  25 cleanup()
  26 {
  27         if [ "$?" = 0 ]; then
  28                 echo "selftests: $TESTNAME [PASS]"
  29         else
  30                 echo "selftests: $TESTNAME [FAILED]"
  31         fi
  32         _cleanup
  33 }
  34
  35 if [ $(id -u) -ne 0 ]; then
  36         echo "selftests: $TESTNAME [SKIP] Need root privileges"
  37         exit $ksft_skip
  38 fi
  39
  40 if [ -z "$BPF_FS" ]; then
  41         echo "selftests: $TESTNAME [SKIP] Could not run test without bpffs mounted"
  42         exit $ksft_skip
  43 fi
  44
  45 if ! bpftool version > /dev/null 2>&1; then
  46         echo "selftests: $TESTNAME [SKIP] Could not run test without bpftool"
  47         exit $ksft_skip
  48 fi
  49
  50 set -e
  51
  52 trap cleanup_skip EXIT
  53
  54 mkdir $BPF_DIR
  55
  56 trap cleanup EXIT
  57
  58 bpftool prog load metadata_unused.o $BPF_DIR/unused
  59
  60 METADATA_PLAIN="$(bpftool prog)"
  61 echo "$METADATA_PLAIN" | grep 'a = "foo"' > /dev/null
  62 echo "$METADATA_PLAIN" | grep 'b = 1' > /dev/null
  63
  64 bpftool prog --json | grep '"metadata":{"a":"foo","b":1}' > /dev/null
  65
  66 bpftool map | grep 'metadata.rodata' > /dev/null
  67
  68 rm $BPF_DIR/unused
  69
  70 bpftool prog load metadata_used.o $BPF_DIR/used
  71
  72 METADATA_PLAIN="$(bpftool prog)"
  73 echo "$METADATA_PLAIN" | grep 'a = "bar"' > /dev/null
  74 echo "$METADATA_PLAIN" | grep 'b = 2' > /dev/null
  75
  76 bpftool prog --json | grep '"metadata":{"a":"bar","b":2}' > /dev/null
  77
  78 bpftool map | grep 'metadata.rodata' > /dev/null
  79
  80 rm $BPF_DIR/used
  81
  82 exit 0