1#!/bin/bash
2# SPDX-License-Identifier: GPL-2.0
3#
4# Run a series of udpgro functional tests.
5
6source net_helper.sh
7
8readonly PEER_NS="ns-peer-$(mktemp -u XXXXXX)"
9
10# set global exit status, but never reset nonzero one.
11check_err()
12{
13	if [ $ret -eq 0 ]; then
14		ret=$1
15	fi
16}
17
18cleanup() {
19	local -r jobs="$(jobs -p)"
20	local -r ns="$(ip netns list|grep $PEER_NS)"
21
22	[ -n "${jobs}" ] && kill -1 ${jobs} 2>/dev/null
23	[ -n "$ns" ] && ip netns del $ns 2>/dev/null
24}
25trap cleanup EXIT
26
27cfg_veth() {
28	ip netns add "${PEER_NS}"
29	ip -netns "${PEER_NS}" link set lo up
30	ip link add type veth
31	ip link set dev veth0 up
32	ip addr add dev veth0 192.168.1.2/24
33	ip addr add dev veth0 2001:db8::2/64 nodad
34
35	ip link set dev veth1 netns "${PEER_NS}"
36	ip -netns "${PEER_NS}" addr add dev veth1 192.168.1.1/24
37	ip -netns "${PEER_NS}" addr add dev veth1 2001:db8::1/64 nodad
38	ip -netns "${PEER_NS}" link set dev veth1 up
39	ip netns exec "${PEER_NS}" ethtool -K veth1 gro on
40}
41
42run_one() {
43	# use 'rx' as separator between sender args and receiver args
44	local -r all="$@"
45	local -r tx_args=${all%rx*}
46	local -r rx_args=${all#*rx}
47	local ret=0
48
49	cfg_veth
50
51	ip netns exec "${PEER_NS}" ./udpgso_bench_rx -C 1000 -R 10 ${rx_args} &
52	local PID1=$!
53
54	wait_local_port_listen ${PEER_NS} 8000 udp
55	./udpgso_bench_tx ${tx_args}
56	check_err $?
57	wait ${PID1}
58	check_err $?
59	[ "$ret" -eq 0 ] && echo "ok" || echo "failed"
60	return $ret
61}
62
63run_test() {
64	local -r args=$@
65
66	printf " %-40s" "$1"
67	./in_netns.sh $0 __subprocess $2 rx -G -r $3
68}
69
70run_one_nat() {
71	# use 'rx' as separator between sender args and receiver args
72	local addr1 addr2 pid family="" ipt_cmd=ip6tables
73	local -r all="$@"
74	local -r tx_args=${all%rx*}
75	local -r rx_args=${all#*rx}
76	local ret=0
77
78	if [[ ${tx_args} = *-4* ]]; then
79		ipt_cmd=iptables
80		family=-4
81		addr1=192.168.1.1
82		addr2=192.168.1.3/24
83	else
84		addr1=2001:db8::1
85		addr2="2001:db8::3/64 nodad"
86	fi
87
88	cfg_veth
89	ip -netns "${PEER_NS}" addr add dev veth1 ${addr2}
90
91	# fool the GRO engine changing the destination address ...
92	ip netns exec "${PEER_NS}" $ipt_cmd -t nat -I PREROUTING -d ${addr1} -j DNAT --to-destination ${addr2%/*}
93
94	# ... so that GRO will match the UDP_GRO enabled socket, but packets
95	# will land on the 'plain' one
96	ip netns exec "${PEER_NS}" ./udpgso_bench_rx -G ${family} -b ${addr1} -n 0 &
97	local PID1=$!
98	ip netns exec "${PEER_NS}" ./udpgso_bench_rx -C 1000 -R 10 ${family} -b ${addr2%/*} ${rx_args} &
99	local PID2=$!
100
101	wait_local_port_listen "${PEER_NS}" 8000 udp
102	./udpgso_bench_tx ${tx_args}
103	check_err $?
104	kill -INT ${PID1}
105	wait ${PID2}
106	check_err $?
107	[ "$ret" -eq 0 ] && echo "ok" || echo "failed"
108	return $ret
109}
110
111run_one_2sock() {
112	# use 'rx' as separator between sender args and receiver args
113	local -r all="$@"
114	local -r tx_args=${all%rx*}
115	local -r rx_args=${all#*rx}
116	local ret=0
117
118	cfg_veth
119
120	ip netns exec "${PEER_NS}" ./udpgso_bench_rx -C 1000 -R 10 ${rx_args} -p 12345 &
121	local PID1=$!
122	ip netns exec "${PEER_NS}" ./udpgso_bench_rx -C 2000 -R 10 ${rx_args} &
123	local PID2=$!
124
125	wait_local_port_listen "${PEER_NS}" 12345 udp
126	./udpgso_bench_tx ${tx_args} -p 12345
127	check_err $?
128	wait_local_port_listen "${PEER_NS}" 8000 udp
129	./udpgso_bench_tx ${tx_args}
130	check_err $?
131	wait ${PID1}
132	check_err $?
133	wait ${PID2}
134	check_err $?
135	[ "$ret" -eq 0 ] && echo "ok" || echo "failed"
136	return $ret
137}
138
139run_nat_test() {
140	local -r args=$@
141
142	printf " %-40s" "$1"
143	./in_netns.sh $0 __subprocess_nat $2 rx -r $3
144}
145
146run_2sock_test() {
147	local -r args=$@
148
149	printf " %-40s" "$1"
150	./in_netns.sh $0 __subprocess_2sock $2 rx -G -r $3
151}
152
153run_all() {
154	local -r core_args="-l 4"
155	local -r ipv4_args="${core_args} -4 -D 192.168.1.1"
156	local -r ipv6_args="${core_args} -6 -D 2001:db8::1"
157	ret=0
158
159	echo "ipv4"
160	run_test "no GRO" "${ipv4_args} -M 10 -s 1400" "-4 -n 10 -l 1400"
161	check_err $?
162
163	# explicitly check we are not receiving UDP_SEGMENT cmsg (-S -1)
164	# when GRO does not take place
165	run_test "no GRO chk cmsg" "${ipv4_args} -M 10 -s 1400" "-4 -n 10 -l 1400 -S -1"
166	check_err $?
167
168	# the GSO packets are aggregated because:
169	# * veth schedule napi after each xmit
170	# * segmentation happens in BH context, veth napi poll is delayed after
171	#   the transmission of the last segment
172	run_test "GRO" "${ipv4_args} -M 1 -s 14720 -S 0 " "-4 -n 1 -l 14720"
173	check_err $?
174	run_test "GRO chk cmsg" "${ipv4_args} -M 1 -s 14720 -S 0 " "-4 -n 1 -l 14720 -S 1472"
175	check_err $?
176	run_test "GRO with custom segment size" "${ipv4_args} -M 1 -s 14720 -S 500 " "-4 -n 1 -l 14720"
177	check_err $?
178	run_test "GRO with custom segment size cmsg" "${ipv4_args} -M 1 -s 14720 -S 500 " "-4 -n 1 -l 14720 -S 500"
179	check_err $?
180
181	run_nat_test "bad GRO lookup" "${ipv4_args} -M 1 -s 14720 -S 0" "-n 10 -l 1472"
182	check_err $?
183	run_2sock_test "multiple GRO socks" "${ipv4_args} -M 1 -s 14720 -S 0 " "-4 -n 1 -l 14720 -S 1472"
184	check_err $?
185
186	echo "ipv6"
187	run_test "no GRO" "${ipv6_args} -M 10 -s 1400" "-n 10 -l 1400"
188	check_err $?
189	run_test "no GRO chk cmsg" "${ipv6_args} -M 10 -s 1400" "-n 10 -l 1400 -S -1"
190	check_err $?
191	run_test "GRO" "${ipv6_args} -M 1 -s 14520 -S 0" "-n 1 -l 14520"
192	check_err $?
193	run_test "GRO chk cmsg" "${ipv6_args} -M 1 -s 14520 -S 0" "-n 1 -l 14520 -S 1452"
194	check_err $?
195	run_test "GRO with custom segment size" "${ipv6_args} -M 1 -s 14520 -S 500" "-n 1 -l 14520"
196	check_err $?
197	run_test "GRO with custom segment size cmsg" "${ipv6_args} -M 1 -s 14520 -S 500" "-n 1 -l 14520 -S 500"
198	check_err $?
199
200	run_nat_test "bad GRO lookup" "${ipv6_args} -M 1 -s 14520 -S 0" "-n 10 -l 1452"
201	check_err $?
202	run_2sock_test "multiple GRO socks" "${ipv6_args} -M 1 -s 14520 -S 0 " "-n 1 -l 14520 -S 1452"
203	check_err $?
204	return $ret
205}
206
207if [[ $# -eq 0 ]]; then
208	run_all
209elif [[ $1 == "__subprocess" ]]; then
210	shift
211	run_one $@
212elif [[ $1 == "__subprocess_nat" ]]; then
213	shift
214	run_one_nat $@
215elif [[ $1 == "__subprocess_2sock" ]]; then
216	shift
217	run_one_2sock $@
218fi
219
220exit $?
221