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