122 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="./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 test25if [[ "$#" -eq "0" ]]; then26 $0 4 tcp -t 127 $0 6 tcp -t 128 $0 4 udp -t 129 $0 6 udp -t 130 echo "OK. All tests passed"31 exit 032fi33 34# Argument parsing35if [[ "$#" -lt "2" ]]; then36 echo "Usage: $0 [4|6] [tcp|udp|raw|raw_hdrincl|packet|packet_dgram] <args>"37 exit 138fi39 40readonly IP="$1"41shift42readonly TXMODE="$1"43shift44readonly EXTRA_ARGS="$@"45 46# Argument parsing: configure addresses47if [[ "${IP}" == "4" ]]; then48 readonly SADDR="${SADDR4}"49 readonly DADDR="${DADDR4}"50elif [[ "${IP}" == "6" ]]; then51 readonly SADDR="${SADDR6}"52 readonly DADDR="${DADDR6}"53else54 echo "Invalid IP version ${IP}"55 exit 156fi57 58# Argument parsing: select receive mode59#60# This differs from send mode for61# - packet: use raw recv, because packet receives skb clones62# - raw_hdrinc: use raw recv, because hdrincl is a tx-only option63case "${TXMODE}" in64'packet' | 'packet_dgram' | 'raw_hdrincl')65 RXMODE='raw'66 ;;67*)68 RXMODE="${TXMODE}"69 ;;70esac71 72# Start of state changes: install cleanup handler73 74cleanup() {75 ip netns del "${NS2}"76 ip netns del "${NS1}"77}78 79trap cleanup EXIT80 81# Create virtual ethernet pair between network namespaces82ip netns add "${NS1}"83ip netns add "${NS2}"84 85# Configure system settings86ip netns exec "${NS1}" sysctl -w -q "${path_sysctl_mem}=1000000"87ip netns exec "${NS2}" sysctl -w -q "${path_sysctl_mem}=1000000"88 89ip link add "${DEV}" mtu "${DEV_MTU}" netns "${NS1}" type veth \90 peer name "${DEV}" mtu "${DEV_MTU}" netns "${NS2}"91 92# Bring the devices up93ip -netns "${NS1}" link set "${DEV}" up94ip -netns "${NS2}" link set "${DEV}" up95 96# Set fixed MAC addresses on the devices97ip -netns "${NS1}" link set dev "${DEV}" address 02:02:02:02:02:0298ip -netns "${NS2}" link set dev "${DEV}" address 06:06:06:06:06:0699 100# Add fixed IP addresses to the devices101ip -netns "${NS1}" addr add 192.168.1.1/24 dev "${DEV}"102ip -netns "${NS2}" addr add 192.168.1.2/24 dev "${DEV}"103ip -netns "${NS1}" addr add fd::1/64 dev "${DEV}" nodad104ip -netns "${NS2}" addr add fd::2/64 dev "${DEV}" nodad105 106# Optionally disable sg or csum offload to test edge cases107# ip netns exec "${NS1}" ethtool -K "${DEV}" sg off108 109do_test() {110 local readonly ARGS="$1"111 112 echo "ipv${IP} ${TXMODE} ${ARGS}"113 ip netns exec "${NS2}" "${BIN}" "-${IP}" -i "${DEV}" -t 2 -C 2 -S "${SADDR}" -D "${DADDR}" ${ARGS} -r "${RXMODE}" &114 sleep 0.2115 ip netns exec "${NS1}" "${BIN}" "-${IP}" -i "${DEV}" -t 1 -C 3 -S "${SADDR}" -D "${DADDR}" ${ARGS} "${TXMODE}"116 wait117}118 119do_test "${EXTRA_ARGS}"120do_test "-z ${EXTRA_ARGS}"121echo ok122