#!/bin/bash # SPDX-License-Identifier: GPL-2.0 . "$(dirname "${0}")/mptcp_lib.sh" rndh=$(printf %x $sec)-$(mktemp -u XXXXXX) ns1="ns1-$rndh" ns2="ns2-$rndh" ns3="ns3-$rndh" capture=false ksft_skip=4 timeout=30 test_cnt=1 ret=0 bail=0 usage() { echo "Usage: $0 [ -b ] [ -c ] [ -d ]" echo -e "\t-b: bail out after first error, otherwise runs al testcases" echo -e "\t-c: capture packets for each test using tcpdump (default: no capture)" echo -e "\t-d: debug this script" } cleanup() { rm -f "$cout" "$sout" rm -f "$large" "$small" rm -f "$capout" local netns for netns in "$ns1" "$ns2" "$ns3";do ip netns del $netns done } mptcp_lib_check_mptcp ip -Version > /dev/null 2>&1 if [ $? -ne 0 ];then echo "SKIP: Could not run test without ip tool" exit $ksft_skip fi # "$ns1" ns2 ns3 # ns1eth1 ns2eth1 ns2eth3 ns3eth1 # netem # ns1eth2 ns2eth2 # netem setup() { large=$(mktemp) small=$(mktemp) sout=$(mktemp) cout=$(mktemp) capout=$(mktemp) size=$((2048 * 4096)) dd if=/dev/zero of=$small bs=4096 count=20 >/dev/null 2>&1 dd if=/dev/zero of=$large bs=4096 count=$((size / 4096)) >/dev/null 2>&1 trap cleanup EXIT for i in "$ns1" "$ns2" "$ns3";do ip netns add $i || exit $ksft_skip ip -net $i link set lo up done ip link add ns1eth1 netns "$ns1" type veth peer name ns2eth1 netns "$ns2" ip link add ns1eth2 netns "$ns1" type veth peer name ns2eth2 netns "$ns2" ip link add ns2eth3 netns "$ns2" type veth peer name ns3eth1 netns "$ns3" ip -net "$ns1" addr add 10.0.1.1/24 dev ns1eth1 ip -net "$ns1" addr add dead:beef:1::1/64 dev ns1eth1 nodad ip -net "$ns1" link set ns1eth1 up mtu 1500 ip -net "$ns1" route add default via 10.0.1.2 ip -net "$ns1" route add default via dead:beef:1::2 ip -net "$ns1" addr add 10.0.2.1/24 dev ns1eth2 ip -net "$ns1" addr add dead:beef:2::1/64 dev ns1eth2 nodad ip -net "$ns1" link set ns1eth2 up mtu 1500 ip -net "$ns1" route add default via 10.0.2.2 metric 101 ip -net "$ns1" route add default via dead:beef:2::2 metric 101 ip netns exec "$ns1" ./pm_nl_ctl limits 1 1 ip netns exec "$ns1" ./pm_nl_ctl add 10.0.2.1 dev ns1eth2 flags subflow ip netns exec "$ns1" sysctl -q net.ipv4.conf.all.rp_filter=0 ip -net "$ns2" addr add 10.0.1.2/24 dev ns2eth1 ip -net "$ns2" addr add dead:beef:1::2/64 dev ns2eth1 nodad ip -net "$ns2" link set ns2eth1 up mtu 1500 ip -net "$ns2" addr add 10.0.2.2/24 dev ns2eth2 ip -net "$ns2" addr add dead:beef:2::2/64 dev ns2eth2 nodad ip -net "$ns2" link set ns2eth2 up mtu 1500 ip -net "$ns2" addr add 10.0.3.2/24 dev ns2eth3 ip -net "$ns2" addr add dead:beef:3::2/64 dev ns2eth3 nodad ip -net "$ns2" link set ns2eth3 up mtu 1500 ip netns exec "$ns2" sysctl -q net.ipv4.ip_forward=1 ip netns exec "$ns2" sysctl -q net.ipv6.conf.all.forwarding=1 ip -net "$ns3" addr add 10.0.3.3/24 dev ns3eth1 ip -net "$ns3" addr add dead:beef:3::3/64 dev ns3eth1 nodad ip -net "$ns3" link set ns3eth1 up mtu 1500 ip -net "$ns3" route add default via 10.0.3.2 ip -net "$ns3" route add default via dead:beef:3::2 ip netns exec "$ns3" ./pm_nl_ctl limits 1 1 } # $1: ns, $2: port wait_local_port_listen() { local listener_ns="${1}" local port="${2}" local port_hex i port_hex="$(printf "%04X" "${port}")" for i in $(seq 10); do ip netns exec "${listener_ns}" cat /proc/net/tcp* | \ awk "BEGIN {rc=1} {if (\$2 ~ /:${port_hex}\$/ && \$4 ~ /0A/) {rc=0; exit}} END {exit rc}" && break sleep 0.1 done } do_transfer() { local cin=$1 local sin=$2 local max_time=$3 local port port=$((10000+$test_cnt)) test_cnt=$((test_cnt+1)) :> "$cout" :> "$sout" :> "$capout" local addr_port addr_port=$(printf "%s:%d" ${connect_addr} ${port}) if $capture; then local capuser if [ -z $SUDO_USER ] ; then capuser="" else capuser="-Z $SUDO_USER" fi local capfile="${rndh}-${port}" local capopt="-i any -s 65535 -B 32768 ${capuser}" ip netns exec ${ns3} tcpdump ${capopt} -w "${capfile}-listener.pcap" >> "${capout}" 2>&1 & local cappid_listener=$! ip netns exec ${ns1} tcpdump ${capopt} -w "${capfile}-connector.pcap" >> "${capout}" 2>&1 & local cappid_connector=$! sleep 1 fi ip netns exec ${ns3} ./mptcp_connect -jt $timeout -l -p $port 0.0.0.0 < "$sin" > "$sout" & local spid=$! wait_local_port_listen "${ns3}" "${port}" local start start=$(date +%s%3N) ip netns exec ${ns1} ./mptcp_connect -jt $timeout -p $port 10.0.3.3 < "$cin" > "$cout" & local cpid=$! wait $cpid local retc=$? wait $spid local rets=$? local stop stop=$(date +%s%3N) if $capture; then sleep 1 kill ${cappid_listener} kill ${cappid_connector} fi local duration duration=$((stop-start)) cmp $sin $cout > /dev/null 2>&1 local cmps=$? cmp $cin $sout > /dev/null 2>&1 local cmpc=$? printf "%16s" "$duration max $max_time " if [ $retc -eq 0 ] && [ $rets -eq 0 ] && \ [ $cmpc -eq 0 ] && [ $cmps -eq 0 ] && \ [ $duration -lt $max_time ]; then echo "[ OK ]" cat "$capout" return 0 fi echo " [ fail ]" echo "client exit code $retc, server $rets" 1>&2 echo -e "\nnetns ${ns3} socket stat for $port:" 1>&2 ip netns exec ${ns3} ss -nita 1>&2 -o "sport = :$port" echo -e "\nnetns ${ns1} socket stat for $port:" 1>&2 ip netns exec ${ns1} ss -nita 1>&2 -o "dport = :$port" ls -l $sin $cout ls -l $cin $sout cat "$capout" return 1 } run_test() { local rate1=$1 local rate2=$2 local delay1=$3 local delay2=$4 local lret local dev shift 4 local msg=$* [ $delay1 -gt 0 ] && delay1="delay $delay1" || delay1="" [ $delay2 -gt 0 ] && delay2="delay $delay2" || delay2="" for dev in ns1eth1 ns1eth2; do tc -n $ns1 qdisc del dev $dev root >/dev/null 2>&1 done for dev in ns2eth1 ns2eth2; do tc -n $ns2 qdisc del dev $dev root >/dev/null 2>&1 done tc -n $ns1 qdisc add dev ns1eth1 root netem rate ${rate1}mbit $delay1 tc -n $ns1 qdisc add dev ns1eth2 root netem rate ${rate2}mbit $delay2 tc -n $ns2 qdisc add dev ns2eth1 root netem rate ${rate1}mbit $delay1 tc -n $ns2 qdisc add dev ns2eth2 root netem rate ${rate2}mbit $delay2 # time is measure in ms local time=$((size * 8 * 1000 / (( $rate1 + $rate2) * 1024 *1024) )) # mptcp_connect will do some sleeps to allow the mp_join handshake # completion time=$((time + 1350)) printf "%-50s" "$msg" do_transfer $small $large $((time * 11 / 10)) lret=$? if [ $lret -ne 0 ]; then ret=$lret [ $bail -eq 0 ] || exit $ret fi printf "%-50s" "$msg - reverse direction" do_transfer $large $small $((time * 11 / 10)) lret=$? if [ $lret -ne 0 ]; then ret=$lret [ $bail -eq 0 ] || exit $ret fi } while getopts "bcdh" option;do case "$option" in "h") usage $0 exit 0 ;; "b") bail=1 ;; "c") capture=true ;; "d") set -x ;; "?") usage $0 exit 1 ;; esac done setup run_test 10 10 0 0 "balanced bwidth" run_test 10 10 1 50 "balanced bwidth with unbalanced delay" # we still need some additional infrastructure to pass the following test-cases # run_test 30 10 0 0 "unbalanced bwidth" # run_test 30 10 1 50 "unbalanced bwidth with unbalanced delay" # run_test 30 10 50 1 "unbalanced bwidth with opposed, unbalanced delay" exit $ret