tools/testing/selftests/net/udpgro.sh

   1 #!/bin/bash
   2 # SPDX-License-Identifier: GPL-2.0
   3 #
   4 # Run a series of udpgro functional tests.
   5
   6 source net_helper.sh
   7
   8 readonly PEER_NS="ns-peer-$(mktemp -u XXXXXX)"
   9
  10 # set global exit status, but never reset nonzero one.
  11 check_err()
  12 {
  13         if [ $ret -eq 0 ]; then
  14                 ret=$1
  15         fi
  16 }
  17
  18 cleanup() {
  19         local -r jobs="$(jobs -p)"
  20         local -r ns="$(ip netns list|grep $PEER_NS)"
  21
  22         [ -n "${jobs}" ] && kill -1 ${jobs} 2>/dev/null
  23         [ -n "$ns" ] && ip netns del $ns 2>/dev/null
  24 }
  25 trap cleanup EXIT
  26
  27 cfg_veth() {
  28         ip netns add "${PEER_NS}"
  29         ip -netns "${PEER_NS}" link set lo up
  30         ip link add type veth
  31         ip link set dev veth0 up
  32         ip addr add dev veth0 192.168.1.2/24
  33         ip addr add dev veth0 2001:db8::2/64 nodad
  34
  35         ip link set dev veth1 netns "${PEER_NS}"
  36         ip -netns "${PEER_NS}" addr add dev veth1 192.168.1.1/24
  37         ip -netns "${PEER_NS}" addr add dev veth1 2001:db8::1/64 nodad
  38         ip -netns "${PEER_NS}" link set dev veth1 up
  39         ip netns exec "${PEER_NS}" ethtool -K veth1 gro on
  40 }
  41
  42 run_one() {
  43         # use 'rx' as separator between sender args and receiver args
  44         local -r all="$@"
  45         local -r tx_args=${all%rx*}
  46         local -r rx_args=${all#*rx}
  47         local ret=0
  48
  49         cfg_veth
  50
  51         ip netns exec "${PEER_NS}" ./udpgso_bench_rx -C 1000 -R 10 ${rx_args} &
  52         local PID1=$!
  53
  54         wait_local_port_listen ${PEER_NS} 8000 udp
  55         ./udpgso_bench_tx ${tx_args}
  56         check_err $?
  57         wait ${PID1}
  58         check_err $?
  59         [ "$ret" -eq 0 ] && echo "ok" || echo "failed"
  60         return $ret
  61 }
  62
  63 run_test() {
  64         local -r args=$@
  65
  66         printf " %-40s" "$1"
  67         ./in_netns.sh $0 __subprocess $2 rx -G -r $3
  68 }
  69
  70 run_one_nat() {
  71         # use 'rx' as separator between sender args and receiver args
  72         local addr1 addr2 pid family="" ipt_cmd=ip6tables
  73         local -r all="$@"
  74         local -r tx_args=${all%rx*}
  75         local -r rx_args=${all#*rx}
  76         local ret=0
  77
  78         if [[ ${tx_args} = *-4* ]]; then
  79                 ipt_cmd=iptables
  80                 family=-4
  81                 addr1=192.168.1.1
  82                 addr2=192.168.1.3/24
  83         else
  84                 addr1=2001:db8::1
  85                 addr2="2001:db8::3/64 nodad"
  86         fi
  87
  88         cfg_veth
  89         ip -netns "${PEER_NS}" addr add dev veth1 ${addr2}
  90
  91         # fool the GRO engine changing the destination address ...
  92         ip netns exec "${PEER_NS}" $ipt_cmd -t nat -I PREROUTING -d ${addr1} -j DNAT --to-destination ${addr2%/*}
  93
  94         # ... so that GRO will match the UDP_GRO enabled socket, but packets
  95         # will land on the 'plain' one
  96         ip netns exec "${PEER_NS}" ./udpgso_bench_rx -G ${family} -b ${addr1} -n 0 &
  97         local PID1=$!
  98         ip netns exec "${PEER_NS}" ./udpgso_bench_rx -C 1000 -R 10 ${family} -b ${addr2%/*} ${rx_args} &
  99         local PID2=$!
 100
 101         wait_local_port_listen "${PEER_NS}" 8000 udp
 102         ./udpgso_bench_tx ${tx_args}
 103         check_err $?
 104         kill -INT ${PID1}
 105         wait ${PID2}
 106         check_err $?
 107         [ "$ret" -eq 0 ] && echo "ok" || echo "failed"
 108         return $ret
 109 }
 110
 111 run_one_2sock() {
 112         # use 'rx' as separator between sender args and receiver args
 113         local -r all="$@"
 114         local -r tx_args=${all%rx*}
 115         local -r rx_args=${all#*rx}
 116         local ret=0
 117
 118         cfg_veth
 119
 120         ip netns exec "${PEER_NS}" ./udpgso_bench_rx -C 1000 -R 10 ${rx_args} -p 12345 &
 121         local PID1=$!
 122         ip netns exec "${PEER_NS}" ./udpgso_bench_rx -C 2000 -R 10 ${rx_args} &
 123         local PID2=$!
 124
 125         wait_local_port_listen "${PEER_NS}" 12345 udp
 126         ./udpgso_bench_tx ${tx_args} -p 12345
 127         check_err $?
 128         wait_local_port_listen "${PEER_NS}" 8000 udp
 129         ./udpgso_bench_tx ${tx_args}
 130         check_err $?
 131         wait ${PID1}
 132         check_err $?
 133         wait ${PID2}
 134         check_err $?
 135         [ "$ret" -eq 0 ] && echo "ok" || echo "failed"
 136         return $ret
 137 }
 138
 139 run_nat_test() {
 140         local -r args=$@
 141
 142         printf " %-40s" "$1"
 143         ./in_netns.sh $0 __subprocess_nat $2 rx -r $3
 144 }
 145
 146 run_2sock_test() {
 147         local -r args=$@
 148
 149         printf " %-40s" "$1"
 150         ./in_netns.sh $0 __subprocess_2sock $2 rx -G -r $3
 151 }
 152
 153 run_all() {
 154         local -r core_args="-l 4"
 155         local -r ipv4_args="${core_args} -4 -D 192.168.1.1"
 156         local -r ipv6_args="${core_args} -6 -D 2001:db8::1"
 157         ret=0
 158
 159         echo "ipv4"
 160         run_test "no GRO" "${ipv4_args} -M 10 -s 1400" "-4 -n 10 -l 1400"
 161         check_err $?
 162
 163         # explicitly check we are not receiving UDP_SEGMENT cmsg (-S -1)
 164         # when GRO does not take place
 165         run_test "no GRO chk cmsg" "${ipv4_args} -M 10 -s 1400" "-4 -n 10 -l 1400 -S -1"
 166         check_err $?
 167
 168         # the GSO packets are aggregated because:
 169         # * veth schedule napi after each xmit
 170         # * segmentation happens in BH context, veth napi poll is delayed after
 171         #   the transmission of the last segment
 172         run_test "GRO" "${ipv4_args} -M 1 -s 14720 -S 0 " "-4 -n 1 -l 14720"
 173         check_err $?
 174         run_test "GRO chk cmsg" "${ipv4_args} -M 1 -s 14720 -S 0 " "-4 -n 1 -l 14720 -S 1472"
 175         check_err $?
 176         run_test "GRO with custom segment size" "${ipv4_args} -M 1 -s 14720 -S 500 " "-4 -n 1 -l 14720"
 177         check_err $?
 178         run_test "GRO with custom segment size cmsg" "${ipv4_args} -M 1 -s 14720 -S 500 " "-4 -n 1 -l 14720 -S 500"
 179         check_err $?
 180
 181         run_nat_test "bad GRO lookup" "${ipv4_args} -M 1 -s 14720 -S 0" "-n 10 -l 1472"
 182         check_err $?
 183         run_2sock_test "multiple GRO socks" "${ipv4_args} -M 1 -s 14720 -S 0 " "-4 -n 1 -l 14720 -S 1472"
 184         check_err $?
 185
 186         echo "ipv6"
 187         run_test "no GRO" "${ipv6_args} -M 10 -s 1400" "-n 10 -l 1400"
 188         check_err $?
 189         run_test "no GRO chk cmsg" "${ipv6_args} -M 10 -s 1400" "-n 10 -l 1400 -S -1"
 190         check_err $?
 191         run_test "GRO" "${ipv6_args} -M 1 -s 14520 -S 0" "-n 1 -l 14520"
 192         check_err $?
 193         run_test "GRO chk cmsg" "${ipv6_args} -M 1 -s 14520 -S 0" "-n 1 -l 14520 -S 1452"
 194         check_err $?
 195         run_test "GRO with custom segment size" "${ipv6_args} -M 1 -s 14520 -S 500" "-n 1 -l 14520"
 196         check_err $?
 197         run_test "GRO with custom segment size cmsg" "${ipv6_args} -M 1 -s 14520 -S 500" "-n 1 -l 14520 -S 500"
 198         check_err $?
 199
 200         run_nat_test "bad GRO lookup" "${ipv6_args} -M 1 -s 14520 -S 0" "-n 10 -l 1452"
 201         check_err $?
 202         run_2sock_test "multiple GRO socks" "${ipv6_args} -M 1 -s 14520 -S 0 " "-n 1 -l 14520 -S 1452"
 203         check_err $?
 204         return $ret
 205 }
 206
 207 if [[ $# -eq 0 ]]; then
 208         run_all
 209 elif [[ $1 == "__subprocess" ]]; then
 210         shift
 211         run_one $@
 212 elif [[ $1 == "__subprocess_nat" ]]; then
 213         shift
 214         run_one_nat $@
 215 elif [[ $1 == "__subprocess_2sock" ]]; then
 216         shift
 217         run_one_2sock $@
 218 fi
 219
 220 exit $?