309 lines
7.8 KiB
Bash
309 lines
7.8 KiB
Bash
|
#!/bin/bash
|
||
|
# SPDX-License-Identifier: GPL-2.0
|
||
|
|
||
|
. "$(dirname "${0}")/mptcp_lib.sh"
|
||
|
|
||
|
sec=$(date +%s)
|
||
|
rndh=$(printf %x $sec)-$(mktemp -u XXXXXX)
|
||
|
ns1="ns1-$rndh"
|
||
|
ns2="ns2-$rndh"
|
||
|
ns3="ns3-$rndh"
|
||
|
capture=false
|
||
|
ksft_skip=4
|
||
|
timeout_poll=30
|
||
|
timeout_test=$((timeout_poll * 2 + 1))
|
||
|
test_cnt=1
|
||
|
ret=0
|
||
|
bail=0
|
||
|
slack=50
|
||
|
|
||
|
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=$((2 * 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
|
||
|
ip netns exec $i sysctl -q net.ipv4.conf.all.rp_filter=0
|
||
|
ip netns exec $i sysctl -q net.ipv4.conf.default.rp_filter=0
|
||
|
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 -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
|
||
|
|
||
|
# debug build can slow down measurably the test program
|
||
|
# we use quite tight time limit on the run-time, to ensure
|
||
|
# maximum B/W usage.
|
||
|
# Use kmemleak/lockdep/kasan/prove_locking presence as a rough
|
||
|
# estimate for this being a debug kernel and increase the
|
||
|
# maximum run-time accordingly. Observed run times for CI builds
|
||
|
# running selftests, including kbuild, were used to determine the
|
||
|
# amount of time to add.
|
||
|
grep -q ' kmemleak_init$\| lockdep_init$\| kasan_init$\| prove_locking$' /proc/kallsyms && slack=$((slack+550))
|
||
|
}
|
||
|
|
||
|
# $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"
|
||
|
|
||
|
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
|
||
|
|
||
|
timeout ${timeout_test} \
|
||
|
ip netns exec ${ns3} \
|
||
|
./mptcp_connect -jt ${timeout_poll} -l -p $port -T $max_time \
|
||
|
0.0.0.0 < "$sin" > "$sout" &
|
||
|
local spid=$!
|
||
|
|
||
|
wait_local_port_listen "${ns3}" "${port}"
|
||
|
|
||
|
timeout ${timeout_test} \
|
||
|
ip netns exec ${ns1} \
|
||
|
./mptcp_connect -jt ${timeout_poll} -p $port -T $max_time \
|
||
|
10.0.3.3 < "$cin" > "$cout" &
|
||
|
local cpid=$!
|
||
|
|
||
|
wait $cpid
|
||
|
local retc=$?
|
||
|
wait $spid
|
||
|
local rets=$?
|
||
|
|
||
|
if $capture; then
|
||
|
sleep 1
|
||
|
kill ${cappid_listener}
|
||
|
kill ${cappid_connector}
|
||
|
fi
|
||
|
|
||
|
cmp $sin $cout > /dev/null 2>&1
|
||
|
local cmps=$?
|
||
|
cmp $cin $sout > /dev/null 2>&1
|
||
|
local cmpc=$?
|
||
|
|
||
|
printf "%-16s" " max $max_time "
|
||
|
if [ $retc -eq 0 ] && [ $rets -eq 0 ] && \
|
||
|
[ $cmpc -eq 0 ] && [ $cmps -eq 0 ]; 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 measured in ms, account for transfer size, aggregated link speed
|
||
|
# and header overhead (10%)
|
||
|
# ms byte -> bit 10% mbit -> kbit -> bit 10%
|
||
|
local time=$((1000 * size * 8 * 10 / ((rate1 + rate2) * 1000 * 1000 * 9) ))
|
||
|
|
||
|
# mptcp_connect will do some sleeps to allow the mp_join handshake
|
||
|
# completion (see mptcp_connect): 200ms on each side, add some slack
|
||
|
time=$((time + 400 + slack))
|
||
|
|
||
|
printf "%-60s" "$msg"
|
||
|
do_transfer $small $large $time
|
||
|
lret=$?
|
||
|
if [ $lret -ne 0 ]; then
|
||
|
ret=$lret
|
||
|
[ $bail -eq 0 ] || exit $ret
|
||
|
fi
|
||
|
|
||
|
printf "%-60s" "$msg - reverse direction"
|
||
|
do_transfer $large $small $time
|
||
|
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
|