selftests/net/msg_zerocopy.sh

0001 #!/bin/bash
0002 #
0003 # Send data between two processes across namespaces
0004 # Run twice: once without and once with zerocopy
0005
0006 set -e
0007
0008 readonly DEV="veth0"
0009 readonly DEV_MTU=65535
0010 readonly BIN="./msg_zerocopy"
0011
0012 readonly RAND="$(mktemp -u XXXXXX)"
0013 readonly NSPREFIX="ns-${RAND}"
0014 readonly NS1="${NSPREFIX}1"
0015 readonly NS2="${NSPREFIX}2"
0016
0017 readonly SADDR4='192.168.1.1'
0018 readonly DADDR4='192.168.1.2'
0019 readonly SADDR6='fd::1'
0020 readonly DADDR6='fd::2'
0021
0022 readonly path_sysctl_mem="net.core.optmem_max"
0023
0024 # No arguments: automated test
0025 if [[ "$#" -eq "0" ]]; then
0026         $0 4 tcp -t 1
0027         $0 6 tcp -t 1
0028         $0 4 udp -t 1
0029         $0 6 udp -t 1
0030         echo "OK. All tests passed"
0031         exit 0
0032 fi
0033
0034 # Argument parsing
0035 if [[ "$#" -lt "2" ]]; then
0036         echo "Usage: $0 [4|6] [tcp|udp|raw|raw_hdrincl|packet|packet_dgram] <args>"
0037         exit 1
0038 fi
0039
0040 readonly IP="$1"
0041 shift
0042 readonly TXMODE="$1"
0043 shift
0044 readonly EXTRA_ARGS="$@"
0045
0046 # Argument parsing: configure addresses
0047 if [[ "${IP}" == "4" ]]; then
0048         readonly SADDR="${SADDR4}"
0049         readonly DADDR="${DADDR4}"
0050 elif [[ "${IP}" == "6" ]]; then
0051         readonly SADDR="${SADDR6}"
0052         readonly DADDR="${DADDR6}"
0053 else
0054         echo "Invalid IP version ${IP}"
0055         exit 1
0056 fi
0057
0058 # Argument parsing: select receive mode
0059 #
0060 # This differs from send mode for
0061 # - packet:     use raw recv, because packet receives skb clones
0062 # - raw_hdrinc: use raw recv, because hdrincl is a tx-only option
0063 case "${TXMODE}" in
0064 'packet' | 'packet_dgram' | 'raw_hdrincl')
0065         RXMODE='raw'
0066         ;;
0067 *)
0068         RXMODE="${TXMODE}"
0069         ;;
0070 esac
0071
0072 # Start of state changes: install cleanup handler
0073 save_sysctl_mem="$(sysctl -n ${path_sysctl_mem})"
0074
0075 cleanup() {
0076         ip netns del "${NS2}"
0077         ip netns del "${NS1}"
0078         sysctl -w -q "${path_sysctl_mem}=${save_sysctl_mem}"
0079 }
0080
0081 trap cleanup EXIT
0082
0083 # Configure system settings
0084 sysctl -w -q "${path_sysctl_mem}=1000000"
0085
0086 # Create virtual ethernet pair between network namespaces
0087 ip netns add "${NS1}"
0088 ip netns add "${NS2}"
0089
0090 ip link add "${DEV}" mtu "${DEV_MTU}" netns "${NS1}" type veth \
0091   peer name "${DEV}" mtu "${DEV_MTU}" netns "${NS2}"
0092
0093 # Bring the devices up
0094 ip -netns "${NS1}" link set "${DEV}" up
0095 ip -netns "${NS2}" link set "${DEV}" up
0096
0097 # Set fixed MAC addresses on the devices
0098 ip -netns "${NS1}" link set dev "${DEV}" address 02:02:02:02:02:02
0099 ip -netns "${NS2}" link set dev "${DEV}" address 06:06:06:06:06:06
0100
0101 # Add fixed IP addresses to the devices
0102 ip -netns "${NS1}" addr add 192.168.1.1/24 dev "${DEV}"
0103 ip -netns "${NS2}" addr add 192.168.1.2/24 dev "${DEV}"
0104 ip -netns "${NS1}" addr add       fd::1/64 dev "${DEV}" nodad
0105 ip -netns "${NS2}" addr add       fd::2/64 dev "${DEV}" nodad
0106
0107 # Optionally disable sg or csum offload to test edge cases
0108 # ip netns exec "${NS1}" ethtool -K "${DEV}" sg off
0109
0110 do_test() {
0111         local readonly ARGS="$1"
0112
0113         echo "ipv${IP} ${TXMODE} ${ARGS}"
0114         ip netns exec "${NS2}" "${BIN}" "-${IP}" -i "${DEV}" -t 2 -C 2 -S "${SADDR}" -D "${DADDR}" ${ARGS} -r "${RXMODE}" &
0115         sleep 0.2
0116         ip netns exec "${NS1}" "${BIN}" "-${IP}" -i "${DEV}" -t 1 -C 3 -S "${SADDR}" -D "${DADDR}" ${ARGS} "${TXMODE}"
0117         wait
0118 }
0119
0120 do_test "${EXTRA_ARGS}"
0121 do_test "-z ${EXTRA_ARGS}"
0122 echo ok