selftests/net/udpgso_bench.sh

0001 #!/bin/bash
0002 # SPDX-License-Identifier: GPL-2.0
0003 #
0004 # Run a series of udpgso benchmarks
0005
0006 readonly GREEN='\033[0;92m'
0007 readonly YELLOW='\033[0;33m'
0008 readonly RED='\033[0;31m'
0009 readonly NC='\033[0m' # No Color
0010
0011 readonly KSFT_PASS=0
0012 readonly KSFT_FAIL=1
0013 readonly KSFT_SKIP=4
0014
0015 num_pass=0
0016 num_err=0
0017 num_skip=0
0018
0019 kselftest_test_exitcode() {
0020         local -r exitcode=$1
0021
0022         if [[ ${exitcode} -eq ${KSFT_PASS} ]]; then
0023                 num_pass=$(( $num_pass + 1 ))
0024         elif [[ ${exitcode} -eq ${KSFT_SKIP} ]]; then
0025                 num_skip=$(( $num_skip + 1 ))
0026         else
0027                 num_err=$(( $num_err + 1 ))
0028         fi
0029 }
0030
0031 kselftest_exit() {
0032         echo -e "$(basename $0): PASS=${num_pass} SKIP=${num_skip} FAIL=${num_err}"
0033
0034         if [[ $num_err -ne 0 ]]; then
0035                 echo -e "$(basename $0): ${RED}FAIL${NC}"
0036                 exit ${KSFT_FAIL}
0037         fi
0038
0039         if [[ $num_skip -ne 0 ]]; then
0040                 echo -e "$(basename $0): ${YELLOW}SKIP${NC}"
0041                 exit ${KSFT_SKIP}
0042         fi
0043
0044         echo -e "$(basename $0): ${GREEN}PASS${NC}"
0045         exit ${KSFT_PASS}
0046 }
0047
0048 wake_children() {
0049         local -r jobs="$(jobs -p)"
0050
0051         if [[ "${jobs}" != "" ]]; then
0052                 kill -1 ${jobs} 2>/dev/null
0053         fi
0054 }
0055 trap wake_children EXIT
0056
0057 run_one() {
0058         local -r args=$@
0059
0060         ./udpgso_bench_rx &
0061         ./udpgso_bench_rx -t &
0062
0063         ./udpgso_bench_tx ${args}
0064 }
0065
0066 run_in_netns() {
0067         local -r args=$@
0068
0069         ./in_netns.sh $0 __subprocess ${args}
0070         kselftest_test_exitcode $?
0071 }
0072
0073 run_udp() {
0074         local -r args=$@
0075
0076         echo "udp"
0077         run_in_netns ${args}
0078
0079         echo "udp gso"
0080         run_in_netns ${args} -S 0
0081
0082         echo "udp gso zerocopy"
0083         run_in_netns ${args} -S 0 -z
0084
0085         echo "udp gso timestamp"
0086         run_in_netns ${args} -S 0 -T
0087
0088         echo "udp gso zerocopy audit"
0089         run_in_netns ${args} -S 0 -z -a
0090
0091         echo "udp gso timestamp audit"
0092         run_in_netns ${args} -S 0 -T -a
0093
0094         echo "udp gso zerocopy timestamp audit"
0095         run_in_netns ${args} -S 0 -T -z -a
0096 }
0097
0098 run_tcp() {
0099         local -r args=$@
0100
0101         echo "tcp"
0102         run_in_netns ${args} -t
0103
0104         echo "tcp zerocopy"
0105         run_in_netns ${args} -t -z
0106
0107         # excluding for now because test fails intermittently
0108         # add -P option to include poll() to reduce possibility of lost messages
0109         #echo "tcp zerocopy audit"
0110         #run_in_netns ${args} -t -z -P -a
0111 }
0112
0113 run_all() {
0114         local -r core_args="-l 3"
0115         local -r ipv4_args="${core_args} -4 -D 127.0.0.1"
0116         local -r ipv6_args="${core_args} -6 -D ::1"
0117
0118         echo "ipv4"
0119         run_tcp "${ipv4_args}"
0120         run_udp "${ipv4_args}"
0121
0122         echo "ipv6"
0123         run_tcp "${ipv6_args}"
0124         run_udp "${ipv6_args}"
0125 }
0126
0127 if [[ $# -eq 0 ]]; then
0128         run_all
0129         kselftest_exit
0130 elif [[ $1 == "__subprocess" ]]; then
0131         shift
0132         run_one $@
0133 else
0134         run_in_netns $@
0135 fi