tools/testing/selftests/net/mptcp/diag.sh

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