1#!/bin/bash
2# SPDX-License-Identifier: GPL-2.0
3#
4# Run a series of udpgro benchmarks
5
6readonly PEER_NS="ns-peer-$(mktemp -u XXXXXX)"
7
8BPF_FILE="../bpf/xdp_dummy.bpf.o"
9
10cleanup() {
11	local -r jobs="$(jobs -p)"
12	local -r ns="$(ip netns list|grep $PEER_NS)"
13
14	[ -n "${jobs}" ] && kill -INT ${jobs} 2>/dev/null
15	[ -n "$ns" ] && ip netns del $ns 2>/dev/null
16}
17trap cleanup EXIT
18
19run_one() {
20	# use 'rx' as separator between sender args and receiver args
21	local -r all="$@"
22	local -r tx_args=${all%rx*}
23	local rx_args=${all#*rx}
24
25	[[ "${tx_args}" == *"-4"* ]] && rx_args="${rx_args} -4"
26
27	ip netns add "${PEER_NS}"
28	ip -netns "${PEER_NS}" link set lo up
29	ip link add type veth
30	ip link set dev veth0 up
31	ip addr add dev veth0 192.168.1.2/24
32	ip addr add dev veth0 2001:db8::2/64 nodad
33
34	ip link set dev veth1 netns "${PEER_NS}"
35	ip -netns "${PEER_NS}" addr add dev veth1 192.168.1.1/24
36	ip -netns "${PEER_NS}" addr add dev veth1 2001:db8::1/64 nodad
37	ip -netns "${PEER_NS}" link set dev veth1 up
38
39	ip -n "${PEER_NS}" link set veth1 xdp object ${BPF_FILE} section xdp
40	ip netns exec "${PEER_NS}" ./udpgso_bench_rx ${rx_args} -r &
41	ip netns exec "${PEER_NS}" ./udpgso_bench_rx -t ${rx_args} -r &
42
43	# Hack: let bg programs complete the startup
44	sleep 0.2
45	./udpgso_bench_tx ${tx_args}
46}
47
48run_in_netns() {
49	local -r args=$@
50
51	./in_netns.sh $0 __subprocess ${args}
52}
53
54run_udp() {
55	local -r args=$@
56
57	echo "udp gso - over veth touching data"
58	run_in_netns ${args} -S 0 rx
59
60	echo "udp gso and gro - over veth touching data"
61	run_in_netns ${args} -S 0 rx -G
62}
63
64run_tcp() {
65	local -r args=$@
66
67	echo "tcp - over veth touching data"
68	run_in_netns ${args} -t rx
69}
70
71run_all() {
72	local -r core_args="-l 4"
73	local -r ipv4_args="${core_args} -4 -D 192.168.1.1"
74	local -r ipv6_args="${core_args} -6 -D 2001:db8::1"
75
76	echo "ipv4"
77	run_tcp "${ipv4_args}"
78	run_udp "${ipv4_args}"
79
80	echo "ipv6"
81	run_tcp "${ipv4_args}"
82	run_udp "${ipv6_args}"
83}
84
85if [ ! -f ${BPF_FILE} ]; then
86	echo "Missing ${BPF_FILE}. Build bpf selftest first"
87	exit -1
88fi
89
90if [[ $# -eq 0 ]]; then
91	run_all
92elif [[ $1 == "__subprocess" ]]; then
93	shift
94	run_one $@
95else
96	run_in_netns $@
97fi
98