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 readonly TESTPORT=8000
  11
  12 readonly KSFT_PASS=0
  13 readonly KSFT_FAIL=1
  14 readonly KSFT_SKIP=4
  15
  16 num_pass=0
  17 num_err=0
  18 num_skip=0
  19
  20 kselftest_test_exitcode() {
  21         local -r exitcode=$1
  22
  23         if [[ ${exitcode} -eq ${KSFT_PASS} ]]; then
  24                 num_pass=$(( $num_pass + 1 ))
  25         elif [[ ${exitcode} -eq ${KSFT_SKIP} ]]; then
  26                 num_skip=$(( $num_skip + 1 ))
  27         else
  28                 num_err=$(( $num_err + 1 ))
  29         fi
  30 }
  31
  32 kselftest_exit() {
  33         echo -e "$(basename $0): PASS=${num_pass} SKIP=${num_skip} FAIL=${num_err}"
  34
  35         if [[ $num_err -ne 0 ]]; then
  36                 echo -e "$(basename $0): ${RED}FAIL${NC}"
  37                 exit ${KSFT_FAIL}
  38         fi
  39
  40         if [[ $num_skip -ne 0 ]]; then
  41                 echo -e "$(basename $0): ${YELLOW}SKIP${NC}"
  42                 exit ${KSFT_SKIP}
  43         fi
  44
  45         echo -e "$(basename $0): ${GREEN}PASS${NC}"
  46         exit ${KSFT_PASS}
  47 }
  48
  49 wake_children() {
  50         local -r jobs="$(jobs -p)"
  51
  52         if [[ "${jobs}" != "" ]]; then
  53                 kill -1 ${jobs} 2>/dev/null
  54         fi
  55 }
  56 trap wake_children EXIT
  57
  58 run_one() {
  59         local -r args=$@
  60         local nr_socks=0
  61         local i=0
  62         local -r timeout=10
  63
  64         ./udpgso_bench_rx -p "$TESTPORT" &
  65         ./udpgso_bench_rx -p "$TESTPORT" -t &
  66
  67         # Wait for the above test program to get ready to receive connections.
  68         while [ "$i" -lt "$timeout" ]; do
  69                 nr_socks="$(ss -lnHi | grep -c "\*:${TESTPORT}")"
  70                 [ "$nr_socks" -eq 2 ] && break
  71                 i=$((i + 1))
  72                 sleep 1
  73         done
  74         if [ "$nr_socks" -ne 2 ]; then
  75                 echo "timed out while waiting for udpgso_bench_rx"
  76                 exit 1
  77         fi
  78
  79         ./udpgso_bench_tx -p "$TESTPORT" ${args}
  80 }
  81
  82 run_in_netns() {
  83         local -r args=$@
  84
  85         ./in_netns.sh $0 __subprocess ${args}
  86         kselftest_test_exitcode $?
  87 }
  88
  89 run_udp() {
  90         local -r args=$@
  91
  92         echo "udp"
  93         run_in_netns ${args}
  94
  95         echo "udp gso"
  96         run_in_netns ${args} -S 0
  97
  98         echo "udp gso zerocopy"
  99         run_in_netns ${args} -S 0 -z
 100
 101         echo "udp gso timestamp"
 102         run_in_netns ${args} -S 0 -T
 103
 104         echo "udp gso zerocopy audit"
 105         run_in_netns ${args} -S 0 -z -a
 106
 107         echo "udp gso timestamp audit"
 108         run_in_netns ${args} -S 0 -T -a
 109
 110         echo "udp gso zerocopy timestamp audit"
 111         run_in_netns ${args} -S 0 -T -z -a
 112 }
 113
 114 run_tcp() {
 115         local -r args=$@
 116
 117         echo "tcp"
 118         run_in_netns ${args} -t
 119
 120         echo "tcp zerocopy"
 121         run_in_netns ${args} -t -z
 122
 123         # excluding for now because test fails intermittently
 124         # add -P option to include poll() to reduce possibility of lost messages
 125         #echo "tcp zerocopy audit"
 126         #run_in_netns ${args} -t -z -P -a
 127 }
 128
 129 run_all() {
 130         local -r core_args="-l 3"
 131         local -r ipv4_args="${core_args} -4 -D 127.0.0.1"
 132         local -r ipv6_args="${core_args} -6 -D ::1"
 133
 134         echo "ipv4"
 135         run_tcp "${ipv4_args}"
 136         run_udp "${ipv4_args}"
 137
 138         echo "ipv6"
 139         run_tcp "${ipv6_args}"
 140         run_udp "${ipv6_args}"
 141 }
 142
 143 if [[ $# -eq 0 ]]; then
 144         run_all
 145         kselftest_exit
 146 elif [[ $1 == "__subprocess" ]]; then
 147         shift
 148         run_one $@
 149 else
 150         run_in_netns $@
 151 fi