udpgso_bench.sh

  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