xref: /linux/tools/testing/selftests/net/io_uring_zerocopy_tx.sh (revision 9d463f7863f2cc3722da9c670c88158409cb7c42) 
1#!/bin/bash
2#
3# Send data between two processes across namespaces
4# Run twice: once without and once with zerocopy
5
6set -e
7
8readonly DEV="veth0"
9readonly DEV_MTU=65535
10readonly BIN_TX="./io_uring_zerocopy_tx"
11readonly BIN_RX="./msg_zerocopy"
12
13readonly RAND="$(mktemp -u XXXXXX)"
14readonly NSPREFIX="ns-${RAND}"
15readonly NS1="${NSPREFIX}1"
16readonly NS2="${NSPREFIX}2"
17
18readonly SADDR4='192.168.1.1'
19readonly DADDR4='192.168.1.2'
20readonly SADDR6='fd::1'
21readonly DADDR6='fd::2'
22
23readonly path_sysctl_mem="net.core.optmem_max"
24
25# No arguments: automated test
26if [[ "$#" -eq "0" ]]; then
27	IPs=( "4" "6" )
28
29	for IP in "${IPs[@]}"; do
30		for mode in $(seq 1 3); do
31			$0 "$IP" udp -m "$mode" -t 1 -n 32
32			$0 "$IP" tcp -m "$mode" -t 1 -n 1
33		done
34	done
35
36	echo "OK. All tests passed"
37	exit 0
38fi
39
40# Argument parsing
41if [[ "$#" -lt "2" ]]; then
42	echo "Usage: $0 [4|6] [tcp|udp|raw|raw_hdrincl|packet|packet_dgram] <args>"
43	exit 1
44fi
45
46readonly IP="$1"
47shift
48readonly TXMODE="$1"
49shift
50readonly EXTRA_ARGS="$@"
51
52# Argument parsing: configure addresses
53if [[ "${IP}" == "4" ]]; then
54	readonly SADDR="${SADDR4}"
55	readonly DADDR="${DADDR4}"
56elif [[ "${IP}" == "6" ]]; then
57	readonly SADDR="${SADDR6}"
58	readonly DADDR="${DADDR6}"
59else
60	echo "Invalid IP version ${IP}"
61	exit 1
62fi
63
64# Argument parsing: select receive mode
65#
66# This differs from send mode for
67# - packet:	use raw recv, because packet receives skb clones
68# - raw_hdrinc: use raw recv, because hdrincl is a tx-only option
69case "${TXMODE}" in
70'packet' | 'packet_dgram' | 'raw_hdrincl')
71	RXMODE='raw'
72	;;
73*)
74	RXMODE="${TXMODE}"
75	;;
76esac
77
78# Start of state changes: install cleanup handler
79
80old_io_uring_disabled=0
81cleanup() {
82	ip netns del "${NS2}"
83	ip netns del "${NS1}"
84	if [ "$old_io_uring_disabled" -ne 0 ]; then
85		sysctl -w -q kernel.io_uring_disabled="$old_io_uring_disabled" 2>/dev/null || true
86	fi
87}
88
89trap cleanup EXIT
90
91# Create virtual ethernet pair between network namespaces
92ip netns add "${NS1}"
93ip netns add "${NS2}"
94
95# Configure system settings
96ip netns exec "${NS1}" sysctl -w -q "${path_sysctl_mem}=1000000"
97ip netns exec "${NS2}" sysctl -w -q "${path_sysctl_mem}=1000000"
98
99ip link add "${DEV}" mtu "${DEV_MTU}" netns "${NS1}" type veth \
100  peer name "${DEV}" mtu "${DEV_MTU}" netns "${NS2}"
101
102# Bring the devices up
103ip -netns "${NS1}" link set "${DEV}" up
104ip -netns "${NS2}" link set "${DEV}" up
105
106# Set fixed MAC addresses on the devices
107ip -netns "${NS1}" link set dev "${DEV}" address 02:02:02:02:02:02
108ip -netns "${NS2}" link set dev "${DEV}" address 06:06:06:06:06:06
109
110# Add fixed IP addresses to the devices
111ip -netns "${NS1}" addr add 192.168.1.1/24 dev "${DEV}"
112ip -netns "${NS2}" addr add 192.168.1.2/24 dev "${DEV}"
113ip -netns "${NS1}" addr add       fd::1/64 dev "${DEV}" nodad
114ip -netns "${NS2}" addr add       fd::2/64 dev "${DEV}" nodad
115
116# Optionally disable sg or csum offload to test edge cases
117# ip netns exec "${NS1}" ethtool -K "${DEV}" sg off
118
119do_test() {
120	local readonly ARGS="$1"
121
122	echo "ipv${IP} ${TXMODE} ${ARGS}"
123	ip netns exec "${NS2}" "${BIN_RX}" "-${IP}" -t 2 -C 2 -S "${SADDR}" -D "${DADDR}" -r "${RXMODE}" &
124	sleep 0.2
125	ip netns exec "${NS1}" "${BIN_TX}" "-${IP}" -t 1 -D "${DADDR}" ${ARGS} "${TXMODE}"
126	wait
127}
128
129old_io_uring_disabled=$(sysctl -n kernel.io_uring_disabled 2>/dev/null || echo "0")
130if [ "$old_io_uring_disabled" -ne 0 ]; then
131	sysctl -w -q kernel.io_uring_disabled=0
132fi
133
134do_test "${EXTRA_ARGS}"
135echo ok
136