1#!/bin/bash 2# SPDX-License-Identifier: GPL-2.0 3# 4# Run a series of udpgro functional tests. 5 6source net_helper.sh 7 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 ]; then 14 ret=$1 15 fi 16} 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/null 23 [ -n "$ns" ] && ip netns del $ns 2>/dev/null 24} 25trap cleanup EXIT 26 27cfg_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 42run_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 63run_test() { 64 local -r args=$@ 65 66 printf " %-40s" "$1" 67 ./in_netns.sh $0 __subprocess $2 rx -G -r $3 68} 69 70run_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 111run_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 139run_nat_test() { 140 local -r args=$@ 141 142 printf " %-40s" "$1" 143 ./in_netns.sh $0 __subprocess_nat $2 rx -r $3 144} 145 146run_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 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=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 207if [[ $# -eq 0 ]]; then 208 run_all 209elif [[ $1 == "__subprocess" ]]; then 210 shift 211 run_one $@ 212elif [[ $1 == "__subprocess_nat" ]]; then 213 shift 214 run_one_nat $@ 215elif [[ $1 == "__subprocess_2sock" ]]; then 216 shift 217 run_one_2sock $@ 218fi 219 220exit $? 221