tools/testing/selftests/net/udpgro_bench.sh

   1 #!/bin/bash
   2 # SPDX-License-Identifier: GPL-2.0
   3 #
   4 # Run a series of udpgro benchmarks
   5
   6 readonly PEER_NS="ns-peer-$(mktemp -u XXXXXX)"
   7
   8 cleanup() {
   9         local -r jobs="$(jobs -p)"
  10         local -r ns="$(ip netns list|grep $PEER_NS)"
  11
  12         [ -n "${jobs}" ] && kill -INT ${jobs} 2>/dev/null
  13         [ -n "$ns" ] && ip netns del $ns 2>/dev/null
  14 }
  15 trap cleanup EXIT
  16
  17 run_one() {
  18         # use 'rx' as separator between sender args and receiver args
  19         local -r all="$@"
  20         local -r tx_args=${all%rx*}
  21         local rx_args=${all#*rx}
  22
  23         [[ "${tx_args}" == *"-4"* ]] && rx_args="${rx_args} -4"
  24
  25         ip netns add "${PEER_NS}"
  26         ip -netns "${PEER_NS}" link set lo up
  27         ip link add type veth
  28         ip link set dev veth0 up
  29         ip addr add dev veth0 192.168.1.2/24
  30         ip addr add dev veth0 2001:db8::2/64 nodad
  31
  32         ip link set dev veth1 netns "${PEER_NS}"
  33         ip -netns "${PEER_NS}" addr add dev veth1 192.168.1.1/24
  34         ip -netns "${PEER_NS}" addr add dev veth1 2001:db8::1/64 nodad
  35         ip -netns "${PEER_NS}" link set dev veth1 up
  36
  37         ip -n "${PEER_NS}" link set veth1 xdp object ../bpf/xdp_dummy.o section xdp_dummy
  38         ip netns exec "${PEER_NS}" ./udpgso_bench_rx ${rx_args} -r &
  39         ip netns exec "${PEER_NS}" ./udpgso_bench_rx -t ${rx_args} -r &
  40
  41         # Hack: let bg programs complete the startup
  42         sleep 0.1
  43         ./udpgso_bench_tx ${tx_args}
  44 }
  45
  46 run_in_netns() {
  47         local -r args=$@
  48
  49         ./in_netns.sh $0 __subprocess ${args}
  50 }
  51
  52 run_udp() {
  53         local -r args=$@
  54
  55         echo "udp gso - over veth touching data"
  56         run_in_netns ${args} -S 0 rx
  57
  58         echo "udp gso and gro - over veth touching data"
  59         run_in_netns ${args} -S 0 rx -G
  60 }
  61
  62 run_tcp() {
  63         local -r args=$@
  64
  65         echo "tcp - over veth touching data"
  66         run_in_netns ${args} -t rx
  67 }
  68
  69 run_all() {
  70         local -r core_args="-l 4"
  71         local -r ipv4_args="${core_args} -4 -D 192.168.1.1"
  72         local -r ipv6_args="${core_args} -6 -D 2001:db8::1"
  73
  74         echo "ipv4"
  75         run_tcp "${ipv4_args}"
  76         run_udp "${ipv4_args}"
  77
  78         echo "ipv6"
  79         run_tcp "${ipv4_args}"
  80         run_udp "${ipv6_args}"
  81 }
  82
  83 if [ ! -f ../bpf/xdp_dummy.o ]; then
  84         echo "Missing xdp_dummy helper. Build bpf selftest first"
  85         exit -1
  86 fi
  87
  88 if [[ $# -eq 0 ]]; then
  89         run_all
  90 elif [[ $1 == "__subprocess" ]]; then
  91         shift
  92         run_one $@
  93 else
  94         run_in_netns $@
  95 fi