221 lines · bash
1#!/bin/bash2# SPDX-License-Identifier: GPL-2.03#4# Run a series of udpgro functional tests.5 6source net_helper.sh7 8readonly PEER_NS="ns-peer-$(mktemp -u XXXXXX)"9 10# set global exit status, but never reset nonzero one.11check_err()12{13 if [ $ret -eq 0 ]; then14 ret=$115 fi16}17 18cleanup() {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/null23 [ -n "$ns" ] && ip netns del $ns 2>/dev/null24}25trap cleanup EXIT26 27cfg_veth() {28 ip netns add "${PEER_NS}"29 ip -netns "${PEER_NS}" link set lo up30 ip link add type veth31 ip link set dev veth0 up32 ip addr add dev veth0 192.168.1.2/2433 ip addr add dev veth0 2001:db8::2/64 nodad34 35 ip link set dev veth1 netns "${PEER_NS}"36 ip -netns "${PEER_NS}" addr add dev veth1 192.168.1.1/2437 ip -netns "${PEER_NS}" addr add dev veth1 2001:db8::1/64 nodad38 ip -netns "${PEER_NS}" link set dev veth1 up39 ip netns exec "${PEER_NS}" ethtool -K veth1 gro on40}41 42run_one() {43 # use 'rx' as separator between sender args and receiver args44 local -r all="$@"45 local -r tx_args=${all%rx*}46 local -r rx_args=${all#*rx}47 local ret=048 49 cfg_veth50 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 udp55 ./udpgso_bench_tx ${tx_args}56 check_err $?57 wait ${PID1}58 check_err $?59 [ "$ret" -eq 0 ] && echo "ok" || echo "failed"60 return $ret61}62 63run_test() {64 local -r args=$@65 66 printf " %-40s" "$1"67 ./in_netns.sh $0 __subprocess $2 rx -G -r $368}69 70run_one_nat() {71 # use 'rx' as separator between sender args and receiver args72 local addr1 addr2 pid family="" ipt_cmd=ip6tables73 local -r all="$@"74 local -r tx_args=${all%rx*}75 local -r rx_args=${all#*rx}76 local ret=077 78 if [[ ${tx_args} = *-4* ]]; then79 ipt_cmd=iptables80 family=-481 addr1=192.168.1.182 addr2=192.168.1.3/2483 else84 addr1=2001:db8::185 addr2="2001:db8::3/64 nodad"86 fi87 88 cfg_veth89 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 packets95 # will land on the 'plain' one96 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 udp102 ./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 $ret109}110 111run_one_2sock() {112 # use 'rx' as separator between sender args and receiver args113 local -r all="$@"114 local -r tx_args=${all%rx*}115 local -r rx_args=${all#*rx}116 local ret=0117 118 cfg_veth119 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 udp126 ./udpgso_bench_tx ${tx_args} -p 12345127 check_err $?128 wait_local_port_listen "${PEER_NS}" 8000 udp129 ./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 $ret137}138 139run_nat_test() {140 local -r args=$@141 142 printf " %-40s" "$1"143 ./in_netns.sh $0 __subprocess_nat $2 rx -r $3144}145 146run_2sock_test() {147 local -r args=$@148 149 printf " %-40s" "$1"150 ./in_netns.sh $0 __subprocess_2sock $2 rx -G -r $3151}152 153run_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=0158 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 place165 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 xmit170 # * segmentation happens in BH context, veth napi poll is delayed after171 # the transmission of the last segment172 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 $ret205}206 207if [[ $# -eq 0 ]]; then208 run_all209elif [[ $1 == "__subprocess" ]]; then210 shift211 run_one $@212elif [[ $1 == "__subprocess_nat" ]]; then213 shift214 run_one_nat $@215elif [[ $1 == "__subprocess_2sock" ]]; then216 shift217 run_one_2sock $@218fi219 220exit $?221