127 lines · bash
1#!/bin/bash2#3# Send data between two processes across namespaces4# Run twice: once without and once with zerocopy5 6set -e7 8readonly DEV="veth0"9readonly DEV_MTU=6553510readonly 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 test26if [[ "$#" -eq "0" ]]; then27 IPs=( "4" "6" )28 29 for IP in "${IPs[@]}"; do30 for mode in $(seq 1 3); do31 $0 "$IP" udp -m "$mode" -t 1 -n 3232 $0 "$IP" tcp -m "$mode" -t 1 -n 133 done34 done35 36 echo "OK. All tests passed"37 exit 038fi39 40# Argument parsing41if [[ "$#" -lt "2" ]]; then42 echo "Usage: $0 [4|6] [tcp|udp|raw|raw_hdrincl|packet|packet_dgram] <args>"43 exit 144fi45 46readonly IP="$1"47shift48readonly TXMODE="$1"49shift50readonly EXTRA_ARGS="$@"51 52# Argument parsing: configure addresses53if [[ "${IP}" == "4" ]]; then54 readonly SADDR="${SADDR4}"55 readonly DADDR="${DADDR4}"56elif [[ "${IP}" == "6" ]]; then57 readonly SADDR="${SADDR6}"58 readonly DADDR="${DADDR6}"59else60 echo "Invalid IP version ${IP}"61 exit 162fi63 64# Argument parsing: select receive mode65#66# This differs from send mode for67# - packet: use raw recv, because packet receives skb clones68# - raw_hdrinc: use raw recv, because hdrincl is a tx-only option69case "${TXMODE}" in70'packet' | 'packet_dgram' | 'raw_hdrincl')71 RXMODE='raw'72 ;;73*)74 RXMODE="${TXMODE}"75 ;;76esac77 78# Start of state changes: install cleanup handler79 80cleanup() {81 ip netns del "${NS2}"82 ip netns del "${NS1}"83}84 85trap cleanup EXIT86 87# Create virtual ethernet pair between network namespaces88ip netns add "${NS1}"89ip netns add "${NS2}"90 91# Configure system settings92ip 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 up99ip -netns "${NS1}" link set "${DEV}" up100ip -netns "${NS2}" link set "${DEV}" up101 102# Set fixed MAC addresses on the devices103ip -netns "${NS1}" link set dev "${DEV}" address 02:02:02:02:02:02104ip -netns "${NS2}" link set dev "${DEV}" address 06:06:06:06:06:06105 106# Add fixed IP addresses to the devices107ip -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}" nodad110ip -netns "${NS2}" addr add fd::2/64 dev "${DEV}" nodad111 112# Optionally disable sg or csum offload to test edge cases113# ip netns exec "${NS1}" ethtool -K "${DEV}" sg off114 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.2121 ip netns exec "${NS1}" "${BIN_TX}" "-${IP}" -t 1 -D "${DADDR}" ${ARGS} "${TXMODE}"122 wait123}124 125do_test "${EXTRA_ARGS}"126echo ok127