tools/testing/selftests/net/msg_zerocopy.sh

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