tools/testing/selftests/net/udpgso_bench.sh

   1 #!/bin/bash
   2 # SPDX-License-Identifier: GPL-2.0
   3 #
   4 # Run a series of udpgso benchmarks
   5
   6 readonly GREEN='\033[0;92m'
   7 readonly YELLOW='\033[0;33m'
   8 readonly RED='\033[0;31m'
   9 readonly NC='\033[0m' # No Color
  10
  11 readonly KSFT_PASS=0
  12 readonly KSFT_FAIL=1
  13 readonly KSFT_SKIP=4
  14
  15 num_pass=0
  16 num_err=0
  17 num_skip=0
  18
  19 kselftest_test_exitcode() {
  20         local -r exitcode=$1
  21
  22         if [[ ${exitcode} -eq ${KSFT_PASS} ]]; then
  23                 num_pass=$(( $num_pass + 1 ))
  24         elif [[ ${exitcode} -eq ${KSFT_SKIP} ]]; then
  25                 num_skip=$(( $num_skip + 1 ))
  26         else
  27                 num_err=$(( $num_err + 1 ))
  28         fi
  29 }
  30
  31 kselftest_exit() {
  32         echo -e "$(basename $0): PASS=${num_pass} SKIP=${num_skip} FAIL=${num_err}"
  33
  34         if [[ $num_err -ne 0 ]]; then
  35                 echo -e "$(basename $0): ${RED}FAIL${NC}"
  36                 exit ${KSFT_FAIL}
  37         fi
  38
  39         if [[ $num_skip -ne 0 ]]; then
  40                 echo -e "$(basename $0): ${YELLOW}SKIP${NC}"
  41                 exit ${KSFT_SKIP}
  42         fi
  43
  44         echo -e "$(basename $0): ${GREEN}PASS${NC}"
  45         exit ${KSFT_PASS}
  46 }
  47
  48 wake_children() {
  49         local -r jobs="$(jobs -p)"
  50
  51         if [[ "${jobs}" != "" ]]; then
  52                 kill -1 ${jobs} 2>/dev/null
  53         fi
  54 }
  55 trap wake_children EXIT
  56
  57 run_one() {
  58         local -r args=$@
  59
  60         ./udpgso_bench_rx &
  61         ./udpgso_bench_rx -t &
  62
  63         ./udpgso_bench_tx ${args}
  64 }
  65
  66 run_in_netns() {
  67         local -r args=$@
  68
  69         ./in_netns.sh $0 __subprocess ${args}
  70         kselftest_test_exitcode $?
  71 }
  72
  73 run_udp() {
  74         local -r args=$@
  75
  76         echo "udp"
  77         run_in_netns ${args}
  78
  79         echo "udp gso"
  80         run_in_netns ${args} -S 0
  81
  82         echo "udp gso zerocopy"
  83         run_in_netns ${args} -S 0 -z
  84
  85         echo "udp gso timestamp"
  86         run_in_netns ${args} -S 0 -T
  87
  88         echo "udp gso zerocopy audit"
  89         run_in_netns ${args} -S 0 -z -a
  90
  91         echo "udp gso timestamp audit"
  92         run_in_netns ${args} -S 0 -T -a
  93
  94         echo "udp gso zerocopy timestamp audit"
  95         run_in_netns ${args} -S 0 -T -z -a
  96 }
  97
  98 run_tcp() {
  99         local -r args=$@
 100
 101         echo "tcp"
 102         run_in_netns ${args} -t
 103
 104         echo "tcp zerocopy"
 105         run_in_netns ${args} -t -z
 106
 107         # excluding for now because test fails intermittently
 108         # add -P option to include poll() to reduce possibility of lost messages
 109         #echo "tcp zerocopy audit"
 110         #run_in_netns ${args} -t -z -P -a
 111 }
 112
 113 run_all() {
 114         local -r core_args="-l 3"
 115         local -r ipv4_args="${core_args} -4 -D 127.0.0.1"
 116         local -r ipv6_args="${core_args} -6 -D ::1"
 117
 118         echo "ipv4"
 119         run_tcp "${ipv4_args}"
 120         run_udp "${ipv4_args}"
 121
 122         echo "ipv6"
 123         run_tcp "${ipv4_args}"
 124         run_udp "${ipv6_args}"
 125 }
 126
 127 if [[ $# -eq 0 ]]; then
 128         run_all
 129         kselftest_exit
 130 elif [[ $1 == "__subprocess" ]]; then
 131         shift
 132         run_one $@
 133 else
 134         run_in_netns $@
 135 fi