1#!/bin/bash 2# SPDX-License-Identifier: GPL-2.0 3 4. "$(dirname "${0}")/mptcp_lib.sh" 5 6rndh=$(printf %x $sec)-$(mktemp -u XXXXXX) 7ns="ns1-$rndh" 8ksft_skip=4 9test_cnt=1 10timeout_poll=100 11timeout_test=$((timeout_poll * 2 + 1)) 12ret=0 13 14flush_pids() 15{ 16 # mptcp_connect in join mode will sleep a bit before completing, 17 # give it some time 18 sleep 1.1 19 20 ip netns pids "${ns}" | xargs --no-run-if-empty kill -SIGUSR1 &>/dev/null 21} 22 23cleanup() 24{ 25 ip netns pids "${ns}" | xargs --no-run-if-empty kill -SIGKILL &>/dev/null 26 27 ip netns del $ns 28} 29 30mptcp_lib_check_mptcp 31 32ip -Version > /dev/null 2>&1 33if [ $? -ne 0 ];then 34 echo "SKIP: Could not run test without ip tool" 35 exit $ksft_skip 36fi 37ss -h | grep -q MPTCP 38if [ $? -ne 0 ];then 39 echo "SKIP: ss tool does not support MPTCP" 40 exit $ksft_skip 41fi 42 43__chk_nr() 44{ 45 local command="$1" 46 local expected=$2 47 local msg="$3" 48 local skip="${4:-SKIP}" 49 local nr 50 51 nr=$(eval $command) 52 53 printf "%-50s" "$msg" 54 if [ $nr != $expected ]; then 55 if [ $nr = "$skip" ] && ! mptcp_lib_expect_all_features; then 56 echo "[ skip ] Feature probably not supported" 57 else 58 echo "[ fail ] expected $expected found $nr" 59 ret=$test_cnt 60 fi 61 else 62 echo "[ ok ]" 63 fi 64 test_cnt=$((test_cnt+1)) 65} 66 67__chk_msk_nr() 68{ 69 local condition=$1 70 shift 1 71 72 __chk_nr "ss -inmHMN $ns | $condition" "$@" 73} 74 75chk_msk_nr() 76{ 77 __chk_msk_nr "grep -c token:" "$@" 78} 79 80wait_msk_nr() 81{ 82 local condition="grep -c token:" 83 local expected=$1 84 local timeout=20 85 local msg nr 86 local max=0 87 local i=0 88 89 shift 1 90 msg=$* 91 92 while [ $i -lt $timeout ]; do 93 nr=$(ss -inmHMN $ns | $condition) 94 [ $nr == $expected ] && break; 95 [ $nr -gt $max ] && max=$nr 96 i=$((i + 1)) 97 sleep 1 98 done 99 100 printf "%-50s" "$msg" 101 if [ $i -ge $timeout ]; then 102 echo "[ fail ] timeout while expecting $expected max $max last $nr" 103 ret=$test_cnt 104 elif [ $nr != $expected ]; then 105 echo "[ fail ] expected $expected found $nr" 106 ret=$test_cnt 107 else 108 echo "[ ok ]" 109 fi 110 test_cnt=$((test_cnt+1)) 111} 112 113chk_msk_fallback_nr() 114{ 115 __chk_msk_nr "grep -c fallback" "$@" 116} 117 118chk_msk_remote_key_nr() 119{ 120 __chk_msk_nr "grep -c remote_key" "$@" 121} 122 123__chk_listen() 124{ 125 local filter="$1" 126 local expected=$2 127 local msg="$3" 128 129 __chk_nr "ss -N $ns -Ml '$filter' | grep -c LISTEN" "$expected" "$msg" 0 130} 131 132chk_msk_listen() 133{ 134 lport=$1 135 136 # destination port search should always return empty list 137 __chk_listen "dport $lport" 0 "listen match for dport $lport" 138 139 # should return 'our' mptcp listen socket 140 __chk_listen "sport $lport" 1 "listen match for sport $lport" 141 142 __chk_listen "src inet:0.0.0.0:$lport" 1 "listen match for saddr and sport" 143 144 __chk_listen "" 1 "all listen sockets" 145 146 nr=$(ss -Ml $filter | wc -l) 147} 148 149# $1: ns, $2: port 150wait_local_port_listen() 151{ 152 local listener_ns="${1}" 153 local port="${2}" 154 155 local port_hex i 156 157 port_hex="$(printf "%04X" "${port}")" 158 for i in $(seq 10); do 159 ip netns exec "${listener_ns}" cat /proc/net/tcp | \ 160 awk "BEGIN {rc=1} {if (\$2 ~ /:${port_hex}\$/ && \$4 ~ /0A/) {rc=0; exit}} END {exit rc}" && 161 break 162 sleep 0.1 163 done 164} 165 166wait_connected() 167{ 168 local listener_ns="${1}" 169 local port="${2}" 170 171 local port_hex i 172 173 port_hex="$(printf "%04X" "${port}")" 174 for i in $(seq 10); do 175 ip netns exec ${listener_ns} grep -q " 0100007F:${port_hex} " /proc/net/tcp && break 176 sleep 0.1 177 done 178} 179 180trap cleanup EXIT 181ip netns add $ns 182ip -n $ns link set dev lo up 183 184echo "a" | \ 185 timeout ${timeout_test} \ 186 ip netns exec $ns \ 187 ./mptcp_connect -p 10000 -l -t ${timeout_poll} -w 20 \ 188 0.0.0.0 >/dev/null & 189wait_local_port_listen $ns 10000 190chk_msk_nr 0 "no msk on netns creation" 191chk_msk_listen 10000 192 193echo "b" | \ 194 timeout ${timeout_test} \ 195 ip netns exec $ns \ 196 ./mptcp_connect -p 10000 -r 0 -t ${timeout_poll} -w 20 \ 197 127.0.0.1 >/dev/null & 198wait_connected $ns 10000 199chk_msk_nr 2 "after MPC handshake " 200chk_msk_remote_key_nr 2 "....chk remote_key" 201chk_msk_fallback_nr 0 "....chk no fallback" 202flush_pids 203 204 205echo "a" | \ 206 timeout ${timeout_test} \ 207 ip netns exec $ns \ 208 ./mptcp_connect -p 10001 -l -s TCP -t ${timeout_poll} -w 20 \ 209 0.0.0.0 >/dev/null & 210wait_local_port_listen $ns 10001 211echo "b" | \ 212 timeout ${timeout_test} \ 213 ip netns exec $ns \ 214 ./mptcp_connect -p 10001 -r 0 -t ${timeout_poll} -w 20 \ 215 127.0.0.1 >/dev/null & 216wait_connected $ns 10001 217chk_msk_fallback_nr 1 "check fallback" 218flush_pids 219 220NR_CLIENTS=100 221for I in `seq 1 $NR_CLIENTS`; do 222 echo "a" | \ 223 timeout ${timeout_test} \ 224 ip netns exec $ns \ 225 ./mptcp_connect -p $((I+10001)) -l -w 20 \ 226 -t ${timeout_poll} 0.0.0.0 >/dev/null & 227done 228wait_local_port_listen $ns $((NR_CLIENTS + 10001)) 229 230for I in `seq 1 $NR_CLIENTS`; do 231 echo "b" | \ 232 timeout ${timeout_test} \ 233 ip netns exec $ns \ 234 ./mptcp_connect -p $((I+10001)) -w 20 \ 235 -t ${timeout_poll} 127.0.0.1 >/dev/null & 236done 237 238wait_msk_nr $((NR_CLIENTS*2)) "many msk socket present" 239flush_pids 240 241exit $ret 242