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