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
 80cleanup() {
 81	ip netns del "${NS2}"
 82	ip netns del "${NS1}"
 83}
 84
 85trap cleanup EXIT
 86
 87# Create virtual ethernet pair between network namespaces
 88ip netns add "${NS1}"
 89ip netns add "${NS2}"
 90
 91# Configure system settings
 92ip netns exec "${NS1}" sysctl -w -q "${path_sysctl_mem}=1000000"
 93ip netns exec "${NS2}" sysctl -w -q "${path_sysctl_mem}=1000000"
 94
 95ip link add "${DEV}" mtu "${DEV_MTU}" netns "${NS1}" type veth \
 96  peer name "${DEV}" mtu "${DEV_MTU}" netns "${NS2}"
 97
 98# Bring the devices up
 99ip -netns "${NS1}" link set "${DEV}" up
100ip -netns "${NS2}" link set "${DEV}" up
101
102# Set fixed MAC addresses on the devices
103ip -netns "${NS1}" link set dev "${DEV}" address 02:02:02:02:02:02
104ip -netns "${NS2}" link set dev "${DEV}" address 06:06:06:06:06:06
105
106# Add fixed IP addresses to the devices
107ip -netns "${NS1}" addr add 192.168.1.1/24 dev "${DEV}"
108ip -netns "${NS2}" addr add 192.168.1.2/24 dev "${DEV}"
109ip -netns "${NS1}" addr add       fd::1/64 dev "${DEV}" nodad
110ip -netns "${NS2}" addr add       fd::2/64 dev "${DEV}" nodad
111
112# Optionally disable sg or csum offload to test edge cases
113# ip netns exec "${NS1}" ethtool -K "${DEV}" sg off
114
115do_test() {
116	local readonly ARGS="$1"
117
118	echo "ipv${IP} ${TXMODE} ${ARGS}"
119	ip netns exec "${NS2}" "${BIN_RX}" "-${IP}" -t 2 -C 2 -S "${SADDR}" -D "${DADDR}" -r "${RXMODE}" &
120	sleep 0.2
121	ip netns exec "${NS1}" "${BIN_TX}" "-${IP}" -t 1 -D "${DADDR}" ${ARGS} "${TXMODE}"
122	wait
123}
124
125do_test "${EXTRA_ARGS}"
126echo ok