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
10
11readonly KSFT_PASS=0
12readonly KSFT_FAIL=1
13readonly KSFT_SKIP=4
14
15num_pass=0
16num_err=0
17num_skip=0
18
19kselftest_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
31kselftest_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
48wake_children() {
49	local -r jobs="$(jobs -p)"
50
51	if [[ "${jobs}" != "" ]]; then
52		kill -1 ${jobs} 2>/dev/null
53	fi
54}
55trap wake_children EXIT
56
57run_one() {
58	local -r args=$@
59
60	./udpgso_bench_rx &
61	./udpgso_bench_rx -t &
62
63	./udpgso_bench_tx ${args}
64}
65
66run_in_netns() {
67	local -r args=$@
68
69	./in_netns.sh $0 __subprocess ${args}
70	kselftest_test_exitcode $?
71}
72
73run_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
98run_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
113run_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
127if [[ $# -eq 0 ]]; then
128	run_all
129	kselftest_exit
130elif [[ $1 == "__subprocess" ]]; then
131	shift
132	run_one $@
133else
134	run_in_netns $@
135fi
136