tjh.dev
Linux kernel mirror (for testing)
git.kernel.org/pub/scm/linux/kernel/git/torvalds/linux.git
kernel
os
linux
1#!/bin/bash
2# SPDX-License-Identifier: GPL-2.0
3#
4# Run a series of udpgso benchmarks
5
6readonly GREEN='\033[0;92m'
7readonly YELLOW='\033[0;33m'
8readonly RED='\033[0;31m'
9readonly NC='\033[0m' # No Color
10readonly TESTPORT=8000
11
12readonly KSFT_PASS=0
13readonly KSFT_FAIL=1
14readonly KSFT_SKIP=4
15
16num_pass=0
17num_err=0
18num_skip=0
19
20kselftest_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
32kselftest_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
49wake_children() {
50 local -r jobs="$(jobs -p)"
51
52 if [[ "${jobs}" != "" ]]; then
53 kill -1 ${jobs} 2>/dev/null
54 fi
55}
56trap wake_children EXIT
57
58run_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
82run_in_netns() {
83 local -r args=$@
84
85 ./in_netns.sh $0 __subprocess ${args}
86 kselftest_test_exitcode $?
87}
88
89run_udp() {
90 local -r args=$@
91
92 echo "udp"
93 run_in_netns ${args}
94
95 echo "udp sendmmsg"
96 run_in_netns ${args} -m
97
98 echo "udp gso"
99 run_in_netns ${args} -S 0
100
101 echo "udp gso zerocopy"
102 run_in_netns ${args} -S 0 -z
103
104 echo "udp gso timestamp"
105 run_in_netns ${args} -S 0 -T
106
107 echo "udp gso zerocopy audit"
108 run_in_netns ${args} -S 0 -z -a
109
110 echo "udp gso timestamp audit"
111 run_in_netns ${args} -S 0 -T -a
112
113 echo "udp gso zerocopy timestamp audit"
114 run_in_netns ${args} -S 0 -T -z -a
115}
116
117run_tcp() {
118 local -r args=$@
119
120 echo "tcp"
121 run_in_netns ${args} -t
122
123 echo "tcp zerocopy"
124 run_in_netns ${args} -t -z
125
126 # excluding for now because test fails intermittently
127 # add -P option to include poll() to reduce possibility of lost messages
128 #echo "tcp zerocopy audit"
129 #run_in_netns ${args} -t -z -P -a
130}
131
132run_all() {
133 local -r core_args="-l 3"
134 local -r ipv4_args="${core_args} -4 -D 127.0.0.1"
135 local -r ipv6_args="${core_args} -6 -D ::1"
136
137 echo "ipv4"
138 run_tcp "${ipv4_args}"
139 run_udp "${ipv4_args}"
140
141 echo "ipv6"
142 run_tcp "${ipv6_args}"
143 run_udp "${ipv6_args}"
144}
145
146if [[ $# -eq 0 ]]; then
147 run_all
148 kselftest_exit
149elif [[ $1 == "__subprocess" ]]; then
150 shift
151 run_one $@
152else
153 run_in_netns $@
154fi