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