300 lines · bash
1#!/bin/bash2# SPDX-License-Identifier: GPL-2.03 4# Double quotes to prevent globbing and word splitting is recommended in new5# code but we accept it, especially because there were too many before having6# address all other issues detected by shellcheck.7#shellcheck disable=SC20868 9. "$(dirname "${0}")/mptcp_lib.sh"10 11ns1=""12ns2=""13ns3=""14capture=false15timeout_poll=3016timeout_test=$((timeout_poll * 2 + 1))17# a bit more space: because we have more to display18MPTCP_LIB_TEST_FORMAT="%02u %-60s"19ret=020bail=021slack=5022large=""23small=""24sout=""25cout=""26capout=""27size=028 29usage() {30 echo "Usage: $0 [ -b ] [ -c ] [ -d ] [ -i]"31 echo -e "\t-b: bail out after first error, otherwise runs al testcases"32 echo -e "\t-c: capture packets for each test using tcpdump (default: no capture)"33 echo -e "\t-d: debug this script"34 echo -e "\t-i: use 'ip mptcp' instead of 'pm_nl_ctl'"35}36 37# This function is used in the cleanup trap38#shellcheck disable=SC231739cleanup()40{41 rm -f "$cout" "$sout"42 rm -f "$large" "$small"43 rm -f "$capout"44 45 mptcp_lib_ns_exit "${ns1}" "${ns2}" "${ns3}"46}47 48mptcp_lib_check_mptcp49mptcp_lib_check_tools ip tc50 51# "$ns1" ns2 ns352# ns1eth1 ns2eth1 ns2eth3 ns3eth153# netem54# ns1eth2 ns2eth255# netem56 57setup()58{59 large=$(mktemp)60 small=$(mktemp)61 sout=$(mktemp)62 cout=$(mktemp)63 capout=$(mktemp)64 size=$((2 * 2048 * 4096))65 66 dd if=/dev/zero of=$small bs=4096 count=20 >/dev/null 2>&167 dd if=/dev/zero of=$large bs=4096 count=$((size / 4096)) >/dev/null 2>&168 69 trap cleanup EXIT70 71 mptcp_lib_ns_init ns1 ns2 ns372 73 ip link add ns1eth1 netns "$ns1" type veth peer name ns2eth1 netns "$ns2"74 ip link add ns1eth2 netns "$ns1" type veth peer name ns2eth2 netns "$ns2"75 ip link add ns2eth3 netns "$ns2" type veth peer name ns3eth1 netns "$ns3"76 77 ip -net "$ns1" addr add 10.0.1.1/24 dev ns1eth178 ip -net "$ns1" addr add dead:beef:1::1/64 dev ns1eth1 nodad79 ip -net "$ns1" link set ns1eth1 up mtu 150080 ip -net "$ns1" route add default via 10.0.1.281 ip -net "$ns1" route add default via dead:beef:1::282 83 ip -net "$ns1" addr add 10.0.2.1/24 dev ns1eth284 ip -net "$ns1" addr add dead:beef:2::1/64 dev ns1eth2 nodad85 ip -net "$ns1" link set ns1eth2 up mtu 150086 ip -net "$ns1" route add default via 10.0.2.2 metric 10187 ip -net "$ns1" route add default via dead:beef:2::2 metric 10188 89 mptcp_lib_pm_nl_set_limits "${ns1}" 1 190 mptcp_lib_pm_nl_add_endpoint "${ns1}" 10.0.2.1 dev ns1eth2 flags subflow91 92 ip -net "$ns2" addr add 10.0.1.2/24 dev ns2eth193 ip -net "$ns2" addr add dead:beef:1::2/64 dev ns2eth1 nodad94 ip -net "$ns2" link set ns2eth1 up mtu 150095 96 ip -net "$ns2" addr add 10.0.2.2/24 dev ns2eth297 ip -net "$ns2" addr add dead:beef:2::2/64 dev ns2eth2 nodad98 ip -net "$ns2" link set ns2eth2 up mtu 150099 100 ip -net "$ns2" addr add 10.0.3.2/24 dev ns2eth3101 ip -net "$ns2" addr add dead:beef:3::2/64 dev ns2eth3 nodad102 ip -net "$ns2" link set ns2eth3 up mtu 1500103 ip netns exec "$ns2" sysctl -q net.ipv4.ip_forward=1104 ip netns exec "$ns2" sysctl -q net.ipv6.conf.all.forwarding=1105 106 ip -net "$ns3" addr add 10.0.3.3/24 dev ns3eth1107 ip -net "$ns3" addr add dead:beef:3::3/64 dev ns3eth1 nodad108 ip -net "$ns3" link set ns3eth1 up mtu 1500109 ip -net "$ns3" route add default via 10.0.3.2110 ip -net "$ns3" route add default via dead:beef:3::2111 112 mptcp_lib_pm_nl_set_limits "${ns3}" 1 1113 114 # debug build can slow down measurably the test program115 # we use quite tight time limit on the run-time, to ensure116 # maximum B/W usage.117 # Use kmemleak/lockdep/kasan/prove_locking presence as a rough118 # estimate for this being a debug kernel and increase the119 # maximum run-time accordingly. Observed run times for CI builds120 # running selftests, including kbuild, were used to determine the121 # amount of time to add.122 grep -q ' kmemleak_init$\| lockdep_init$\| kasan_init$\| prove_locking$' /proc/kallsyms && slack=$((slack+550))123}124 125do_transfer()126{127 local cin=$1128 local sin=$2129 local max_time=$3130 local port131 port=$((10000+MPTCP_LIB_TEST_COUNTER))132 133 :> "$cout"134 :> "$sout"135 :> "$capout"136 137 if $capture; then138 local capuser139 local rndh="${ns1:4}"140 if [ -z $SUDO_USER ] ; then141 capuser=""142 else143 capuser="-Z $SUDO_USER"144 fi145 146 local capfile="${rndh}-${port}"147 local capopt="-i any -s 65535 -B 32768 ${capuser}"148 149 ip netns exec ${ns3} tcpdump ${capopt} -w "${capfile}-listener.pcap" >> "${capout}" 2>&1 &150 local cappid_listener=$!151 152 ip netns exec ${ns1} tcpdump ${capopt} -w "${capfile}-connector.pcap" >> "${capout}" 2>&1 &153 local cappid_connector=$!154 155 sleep 1156 fi157 158 timeout ${timeout_test} \159 ip netns exec ${ns3} \160 ./mptcp_connect -jt ${timeout_poll} -l -p $port -T $max_time \161 0.0.0.0 < "$sin" > "$sout" &162 local spid=$!163 164 mptcp_lib_wait_local_port_listen "${ns3}" "${port}"165 166 timeout ${timeout_test} \167 ip netns exec ${ns1} \168 ./mptcp_connect -jt ${timeout_poll} -p $port -T $max_time \169 10.0.3.3 < "$cin" > "$cout" &170 local cpid=$!171 172 wait $cpid173 local retc=$?174 wait $spid175 local rets=$?176 177 if $capture; then178 sleep 1179 kill ${cappid_listener}180 kill ${cappid_connector}181 fi182 183 cmp $sin $cout > /dev/null 2>&1184 local cmps=$?185 cmp $cin $sout > /dev/null 2>&1186 local cmpc=$?187 188 printf "%-16s" " max $max_time "189 if [ $retc -eq 0 ] && [ $rets -eq 0 ] && \190 [ $cmpc -eq 0 ] && [ $cmps -eq 0 ]; then191 mptcp_lib_pr_ok192 cat "$capout"193 return 0194 fi195 196 mptcp_lib_pr_fail197 echo "client exit code $retc, server $rets" 1>&2198 echo -e "\nnetns ${ns3} socket stat for $port:" 1>&2199 ip netns exec ${ns3} ss -nita 1>&2 -o "sport = :$port"200 echo -e "\nnetns ${ns1} socket stat for $port:" 1>&2201 ip netns exec ${ns1} ss -nita 1>&2 -o "dport = :$port"202 ls -l $sin $cout203 ls -l $cin $sout204 205 cat "$capout"206 return 1207}208 209run_test()210{211 local rate1=$1212 local rate2=$2213 local delay1=$3214 local delay2=$4215 local lret216 local dev217 shift 4218 local msg=$*219 220 [ $delay1 -gt 0 ] && delay1="delay ${delay1}ms" || delay1=""221 [ $delay2 -gt 0 ] && delay2="delay ${delay2}ms" || delay2=""222 223 for dev in ns1eth1 ns1eth2; do224 tc -n $ns1 qdisc del dev $dev root >/dev/null 2>&1225 done226 for dev in ns2eth1 ns2eth2; do227 tc -n $ns2 qdisc del dev $dev root >/dev/null 2>&1228 done229 tc -n $ns1 qdisc add dev ns1eth1 root netem rate ${rate1}mbit $delay1230 tc -n $ns1 qdisc add dev ns1eth2 root netem rate ${rate2}mbit $delay2231 tc -n $ns2 qdisc add dev ns2eth1 root netem rate ${rate1}mbit $delay1232 tc -n $ns2 qdisc add dev ns2eth2 root netem rate ${rate2}mbit $delay2233 234 # time is measured in ms, account for transfer size, aggregated link speed235 # and header overhead (10%)236 # ms byte -> bit 10% mbit -> kbit -> bit 10%237 local time=$((1000 * size * 8 * 10 / ((rate1 + rate2) * 1000 * 1000 * 9) ))238 239 # mptcp_connect will do some sleeps to allow the mp_join handshake240 # completion (see mptcp_connect): 200ms on each side, add some slack241 time=$((time + 400 + slack))242 243 mptcp_lib_print_title "$msg"244 do_transfer $small $large $time245 lret=$?246 mptcp_lib_result_code "${lret}" "${msg}"247 if [ $lret -ne 0 ] && ! mptcp_lib_subtest_is_flaky; then248 ret=$lret249 [ $bail -eq 0 ] || exit $ret250 fi251 252 msg+=" - reverse direction"253 mptcp_lib_print_title "${msg}"254 do_transfer $large $small $time255 lret=$?256 mptcp_lib_result_code "${lret}" "${msg}"257 if [ $lret -ne 0 ] && ! mptcp_lib_subtest_is_flaky; then258 ret=$lret259 [ $bail -eq 0 ] || exit $ret260 fi261}262 263while getopts "bcdhi" option;do264 case "$option" in265 "h")266 usage $0267 exit ${KSFT_PASS}268 ;;269 "b")270 bail=1271 ;;272 "c")273 capture=true274 ;;275 "d")276 set -x277 ;;278 "i")279 mptcp_lib_set_ip_mptcp280 ;;281 "?")282 usage $0283 exit ${KSFT_FAIL}284 ;;285 esac286done287 288setup289mptcp_lib_subtests_last_ts_reset290run_test 10 10 0 0 "balanced bwidth"291run_test 10 10 1 25 "balanced bwidth with unbalanced delay"292 293# we still need some additional infrastructure to pass the following test-cases294MPTCP_LIB_SUBTEST_FLAKY=1 run_test 10 3 0 0 "unbalanced bwidth"295run_test 10 3 1 25 "unbalanced bwidth with unbalanced delay"296run_test 10 3 25 1 "unbalanced bwidth with opposed, unbalanced delay"297 298mptcp_lib_result_print_all_tap299exit $ret300