#!/bin/bash # SPDX-License-Identifier: GPL-2.0 . "$(dirname "${0}")/mptcp_lib.sh" sec=$(date +%s) rndh=$(printf %x $sec)-$(mktemp -u XXXXXX) ns="ns1-$rndh" ksft_skip=4 test_cnt=1 timeout_poll=100 timeout_test=$((timeout_poll * 2 + 1)) ret=0 flush_pids() { # mptcp_connect in join mode will sleep a bit before completing, # give it some time sleep 1.1 ip netns pids "${ns}" | xargs --no-run-if-empty kill -SIGUSR1 &>/dev/null for _ in $(seq $((timeout_poll * 10))); do [ -z "$(ip netns pids "${ns}")" ] && break sleep 0.1 done } cleanup() { ip netns pids "${ns}" | xargs --no-run-if-empty kill -SIGKILL &>/dev/null ip netns del $ns } 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 ss -h | grep -q MPTCP if [ $? -ne 0 ];then echo "SKIP: ss tool does not support MPTCP" exit $ksft_skip fi get_msk_inuse() { ip netns exec $ns cat /proc/net/protocols | awk '$1~/^MPTCP$/{print $3}' } __chk_nr() { local command="$1" local expected=$2 local msg="$3" local skip="${4-SKIP}" local nr nr=$(eval $command) printf "%-50s" "$msg" if [ "$nr" != "$expected" ]; then if [ "$nr" = "$skip" ] && ! mptcp_lib_expect_all_features; then echo "[ skip ] Feature probably not supported" mptcp_lib_result_skip "${msg}" else echo "[ fail ] expected $expected found $nr" mptcp_lib_result_fail "${msg}" ret=$test_cnt fi else echo "[ ok ]" mptcp_lib_result_pass "${msg}" fi test_cnt=$((test_cnt+1)) } __chk_msk_nr() { local condition=$1 shift 1 __chk_nr "ss -inmHMN $ns | $condition" "$@" } chk_msk_nr() { __chk_msk_nr "grep -c token:" "$@" } chk_listener_nr() { local expected=$1 local msg="$2" __chk_nr "ss -inmlHMON $ns | wc -l" "$expected" "$msg - mptcp" 0 __chk_nr "ss -inmlHtON $ns | wc -l" "$expected" "$msg - subflows" } wait_msk_nr() { local condition="grep -c token:" local expected=$1 local timeout=20 local msg nr local max=0 local i=0 shift 1 msg=$* while [ $i -lt $timeout ]; do nr=$(ss -inmHMN $ns | $condition) [ $nr == $expected ] && break; [ $nr -gt $max ] && max=$nr i=$((i + 1)) sleep 1 done printf "%-50s" "$msg" if [ $i -ge $timeout ]; then echo "[ fail ] timeout while expecting $expected max $max last $nr" mptcp_lib_result_fail "${msg} # timeout" ret=$test_cnt elif [ $nr != $expected ]; then echo "[ fail ] expected $expected found $nr" mptcp_lib_result_fail "${msg} # unexpected result" ret=$test_cnt else echo "[ ok ]" mptcp_lib_result_pass "${msg}" fi test_cnt=$((test_cnt+1)) } chk_msk_fallback_nr() { __chk_msk_nr "grep -c fallback" "$@" } chk_msk_remote_key_nr() { __chk_msk_nr "grep -c remote_key" "$@" } __chk_listen() { local filter="$1" local expected=$2 local msg="$3" __chk_nr "ss -N $ns -Ml '$filter' | grep -c LISTEN" "$expected" "$msg" 0 } chk_msk_listen() { lport=$1 # destination port search should always return empty list __chk_listen "dport $lport" 0 "listen match for dport $lport" # should return 'our' mptcp listen socket __chk_listen "sport $lport" 1 "listen match for sport $lport" __chk_listen "src inet:0.0.0.0:$lport" 1 "listen match for saddr and sport" __chk_listen "" 1 "all listen sockets" nr=$(ss -Ml $filter | wc -l) } chk_msk_inuse() { local expected=$1 local msg="....chk ${2:-${expected}} msk in use" local listen_nr if [ "${expected}" -eq 0 ]; then msg+=" after flush" fi listen_nr=$(ss -N "${ns}" -Ml | grep -c LISTEN) expected=$((expected + listen_nr)) for _ in $(seq 10); do if [ $(get_msk_inuse) -eq $expected ];then break fi sleep 0.1 done __chk_nr get_msk_inuse $expected "${msg}" 0 } # $1: cestab nr chk_msk_cestab() { local expected=$1 local msg="....chk ${2:-${expected}} cestab" if [ "${expected}" -eq 0 ]; then msg+=" after flush" fi __chk_nr "mptcp_lib_get_counter ${ns} MPTcpExtMPCurrEstab" \ "${expected}" "${msg}" "" } wait_connected() { 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} grep -q " 0100007F:${port_hex} " /proc/net/tcp && break sleep 0.1 done } trap cleanup EXIT ip netns add $ns ip -n $ns link set dev lo up echo "a" | \ timeout ${timeout_test} \ ip netns exec $ns \ ./mptcp_connect -p 10000 -l -t ${timeout_poll} -w 20 \ 0.0.0.0 >/dev/null & mptcp_lib_wait_local_port_listen $ns 10000 chk_msk_nr 0 "no msk on netns creation" chk_msk_listen 10000 echo "b" | \ timeout ${timeout_test} \ ip netns exec $ns \ ./mptcp_connect -p 10000 -r 0 -t ${timeout_poll} -w 20 \ 127.0.0.1 >/dev/null & wait_connected $ns 10000 chk_msk_nr 2 "after MPC handshake " chk_msk_remote_key_nr 2 "....chk remote_key" chk_msk_fallback_nr 0 "....chk no fallback" chk_msk_inuse 2 chk_msk_cestab 2 flush_pids chk_msk_inuse 0 "2->0" chk_msk_cestab 0 "2->0" echo "a" | \ timeout ${timeout_test} \ ip netns exec $ns \ ./mptcp_connect -p 10001 -l -s TCP -t ${timeout_poll} -w 20 \ 0.0.0.0 >/dev/null & mptcp_lib_wait_local_port_listen $ns 10001 echo "b" | \ timeout ${timeout_test} \ ip netns exec $ns \ ./mptcp_connect -p 10001 -r 0 -t ${timeout_poll} -w 20 \ 127.0.0.1 >/dev/null & wait_connected $ns 10001 chk_msk_fallback_nr 1 "check fallback" chk_msk_inuse 1 chk_msk_cestab 1 flush_pids chk_msk_inuse 0 "1->0" chk_msk_cestab 0 "1->0" NR_CLIENTS=100 for I in `seq 1 $NR_CLIENTS`; do echo "a" | \ timeout ${timeout_test} \ ip netns exec $ns \ ./mptcp_connect -p $((I+10001)) -l -w 20 \ -t ${timeout_poll} 0.0.0.0 >/dev/null & done mptcp_lib_wait_local_port_listen $ns $((NR_CLIENTS + 10001)) for I in `seq 1 $NR_CLIENTS`; do echo "b" | \ timeout ${timeout_test} \ ip netns exec $ns \ ./mptcp_connect -p $((I+10001)) -w 20 \ -t ${timeout_poll} 127.0.0.1 >/dev/null & done wait_msk_nr $((NR_CLIENTS*2)) "many msk socket present" chk_msk_inuse $((NR_CLIENTS*2)) "many" chk_msk_cestab $((NR_CLIENTS*2)) "many" flush_pids chk_msk_inuse 0 "many->0" chk_msk_cestab 0 "many->0" chk_listener_nr 0 "no listener sockets" NR_SERVERS=100 for I in $(seq 1 $NR_SERVERS); do ip netns exec $ns ./mptcp_connect -p $((I + 20001)) \ -t ${timeout_poll} -l 0.0.0.0 >/dev/null 2>&1 & done for I in $(seq 1 $NR_SERVERS); do mptcp_lib_wait_local_port_listen $ns $((I + 20001)) done chk_listener_nr $NR_SERVERS "many listener sockets" # graceful termination for I in $(seq 1 $NR_SERVERS); do echo a | ip netns exec $ns ./mptcp_connect -p $((I + 20001)) 127.0.0.1 >/dev/null 2>&1 & done flush_pids mptcp_lib_result_print_all_tap exit $ret