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

   1 #!/bin/bash
   2 # SPDX-License-Identifier: GPL-2.0
   3
   4 # Double quotes to prevent globbing and word splitting is recommended in new
   5 # code but we accept it, especially because there were too many before having
   6 # address all other issues detected by shellcheck.
   7 #shellcheck disable=SC2086
   8
   9 . "$(dirname "${0}")/mptcp_lib.sh"
  10
  11 ns=""
  12 timeout_poll=30
  13 timeout_test=$((timeout_poll * 2 + 1))
  14 ret=0
  15
  16 flush_pids()
  17 {
  18         # mptcp_connect in join mode will sleep a bit before completing,
  19         # give it some time
  20         sleep 1.1
  21
  22         ip netns pids "${ns}" | xargs --no-run-if-empty kill -SIGUSR1 &>/dev/null
  23
  24         for _ in $(seq $((timeout_poll * 10))); do
  25                 [ -z "$(ip netns pids "${ns}")" ] && break
  26                 sleep 0.1
  27         done
  28 }
  29
  30 # This function is used in the cleanup trap
  31 #shellcheck disable=SC2317
  32 cleanup()
  33 {
  34         ip netns pids "${ns}" | xargs --no-run-if-empty kill -SIGKILL &>/dev/null
  35
  36         mptcp_lib_ns_exit "${ns}"
  37 }
  38
  39 mptcp_lib_check_mptcp
  40 mptcp_lib_check_tools ip ss
  41
  42 get_msk_inuse()
  43 {
  44         ip netns exec $ns cat /proc/net/protocols | awk '$1~/^MPTCP$/{print $3}'
  45 }
  46
  47 __chk_nr()
  48 {
  49         local command="$1"
  50         local expected=$2
  51         local msg="$3"
  52         local skip="${4-SKIP}"
  53         local nr
  54
  55         nr=$(eval $command)
  56
  57         mptcp_lib_print_title "$msg"
  58         if [ "$nr" != "$expected" ]; then
  59                 if [ "$nr" = "$skip" ] && ! mptcp_lib_expect_all_features; then
  60                         mptcp_lib_pr_skip "Feature probably not supported"
  61                         mptcp_lib_result_skip "${msg}"
  62                 else
  63                         mptcp_lib_pr_fail "expected $expected found $nr"
  64                         mptcp_lib_result_fail "${msg}"
  65                         ret=${KSFT_FAIL}
  66                 fi
  67         else
  68                 mptcp_lib_pr_ok
  69                 mptcp_lib_result_pass "${msg}"
  70         fi
  71 }
  72
  73 __chk_msk_nr()
  74 {
  75         local condition=$1
  76         shift 1
  77
  78         __chk_nr "ss -inmHMN $ns | $condition" "$@"
  79 }
  80
  81 chk_msk_nr()
  82 {
  83         __chk_msk_nr "grep -c token:" "$@"
  84 }
  85
  86 chk_listener_nr()
  87 {
  88         local expected=$1
  89         local msg="$2"
  90
  91         __chk_nr "ss -nlHMON $ns | wc -l" "$expected" "$msg - mptcp" 0
  92         __chk_nr "ss -nlHtON $ns | wc -l" "$expected" "$msg - subflows"
  93 }
  94
  95 wait_msk_nr()
  96 {
  97         local condition="grep -c token:"
  98         local expected=$1
  99         local timeout=20
 100         local msg nr
 101         local max=0
 102         local i=0
 103
 104         shift 1
 105         msg=$*
 106
 107         while [ $i -lt $timeout ]; do
 108                 nr=$(ss -inmHMN $ns | $condition)
 109                 [ $nr == $expected ] && break;
 110                 [ $nr -gt $max ] && max=$nr
 111                 i=$((i + 1))
 112                 sleep 1
 113         done
 114
 115         mptcp_lib_print_title "$msg"
 116         if [ $i -ge $timeout ]; then
 117                 mptcp_lib_pr_fail "timeout while expecting $expected max $max last $nr"
 118                 mptcp_lib_result_fail "${msg} # timeout"
 119                 ret=${KSFT_FAIL}
 120         elif [ $nr != $expected ]; then
 121                 mptcp_lib_pr_fail "expected $expected found $nr"
 122                 mptcp_lib_result_fail "${msg} # unexpected result"
 123                 ret=${KSFT_FAIL}
 124         else
 125                 mptcp_lib_pr_ok
 126                 mptcp_lib_result_pass "${msg}"
 127         fi
 128 }
 129
 130 chk_msk_fallback_nr()
 131 {
 132         __chk_msk_nr "grep -c fallback" "$@"
 133 }
 134
 135 chk_msk_remote_key_nr()
 136 {
 137         __chk_msk_nr "grep -c remote_key" "$@"
 138 }
 139
 140 __chk_listen()
 141 {
 142         local filter="$1"
 143         local expected=$2
 144         local msg="$3"
 145
 146         __chk_nr "ss -N $ns -Ml '$filter' | grep -c LISTEN" "$expected" "$msg" 0
 147 }
 148
 149 chk_msk_listen()
 150 {
 151         lport=$1
 152
 153         # destination port search should always return empty list
 154         __chk_listen "dport $lport" 0 "listen match for dport $lport"
 155
 156         # should return 'our' mptcp listen socket
 157         __chk_listen "sport $lport" 1 "listen match for sport $lport"
 158
 159         __chk_listen "src inet:0.0.0.0:$lport" 1 "listen match for saddr and sport"
 160
 161         __chk_listen "" 1 "all listen sockets"
 162
 163         nr=$(ss -Ml $filter | wc -l)
 164 }
 165
 166 chk_msk_inuse()
 167 {
 168         local expected=$1
 169         local msg="....chk ${2:-${expected}} msk in use"
 170         local listen_nr
 171
 172         if [ "${expected}" -eq 0 ]; then
 173                 msg+=" after flush"
 174         fi
 175
 176         listen_nr=$(ss -N "${ns}" -Ml | grep -c LISTEN)
 177         expected=$((expected + listen_nr))
 178
 179         for _ in $(seq 10); do
 180                 if [ "$(get_msk_inuse)" -eq $expected ]; then
 181                         break
 182                 fi
 183                 sleep 0.1
 184         done
 185
 186         __chk_nr get_msk_inuse $expected "${msg}" 0
 187 }
 188
 189 # $1: cestab nr
 190 chk_msk_cestab()
 191 {
 192         local expected=$1
 193         local msg="....chk ${2:-${expected}} cestab"
 194
 195         if [ "${expected}" -eq 0 ]; then
 196                 msg+=" after flush"
 197         fi
 198
 199         __chk_nr "mptcp_lib_get_counter ${ns} MPTcpExtMPCurrEstab" \
 200                  "${expected}" "${msg}" ""
 201 }
 202
 203 msk_info_get_value()
 204 {
 205         local port="${1}"
 206         local info="${2}"
 207
 208         ss -N "${ns}" -inHM dport "${port}" | \
 209                 mptcp_lib_get_info_value "${info}" "${info}"
 210 }
 211
 212 chk_msk_info()
 213 {
 214         local port="${1}"
 215         local info="${2}"
 216         local cnt="${3}"
 217         local msg="....chk ${info}"
 218         local delta_ms=250  # half what we waited before, just to be sure
 219         local now
 220
 221         now=$(msk_info_get_value "${port}" "${info}")
 222
 223         mptcp_lib_print_title "${msg}"
 224         if { [ -z "${cnt}" ] || [ -z "${now}" ]; } &&
 225            ! mptcp_lib_expect_all_features; then
 226                 mptcp_lib_pr_skip "Feature probably not supported"
 227                 mptcp_lib_result_skip "${msg}"
 228         elif [ "$((cnt + delta_ms))" -lt "${now}" ]; then
 229                 mptcp_lib_pr_ok
 230                 mptcp_lib_result_pass "${msg}"
 231         else
 232                 mptcp_lib_pr_fail "value of ${info} changed by $((now - cnt))ms," \
 233                                   "expected at least ${delta_ms}ms"
 234                 mptcp_lib_result_fail "${msg}"
 235                 ret=${KSFT_FAIL}
 236         fi
 237 }
 238
 239 chk_last_time_info()
 240 {
 241         local port="${1}"
 242         local data_sent data_recv ack_recv
 243
 244         data_sent=$(msk_info_get_value "${port}" "last_data_sent")
 245         data_recv=$(msk_info_get_value "${port}" "last_data_recv")
 246         ack_recv=$(msk_info_get_value "${port}" "last_ack_recv")
 247
 248         sleep 0.5  # wait to check after if the timestamps difference
 249
 250         chk_msk_info "${port}" "last_data_sent" "${data_sent}"
 251         chk_msk_info "${port}" "last_data_recv" "${data_recv}"
 252         chk_msk_info "${port}" "last_ack_recv" "${ack_recv}"
 253 }
 254
 255 wait_connected()
 256 {
 257         local listener_ns="${1}"
 258         local port="${2}"
 259
 260         local port_hex i
 261
 262         port_hex="$(printf "%04X" "${port}")"
 263         for i in $(seq 10); do
 264                 ip netns exec ${listener_ns} grep -q " 0100007F:${port_hex} " /proc/net/tcp && break
 265                 sleep 0.1
 266         done
 267 }
 268
 269 trap cleanup EXIT
 270 mptcp_lib_ns_init ns
 271
 272 echo "a" | \
 273         timeout ${timeout_test} \
 274                 ip netns exec $ns \
 275                         ./mptcp_connect -p 10000 -l -t ${timeout_poll} -w 20 \
 276                                 0.0.0.0 >/dev/null &
 277 mptcp_lib_wait_local_port_listen $ns 10000
 278 chk_msk_nr 0 "no msk on netns creation"
 279 chk_msk_listen 10000
 280
 281 echo "b" | \
 282         timeout ${timeout_test} \
 283                 ip netns exec $ns \
 284                         ./mptcp_connect -p 10000 -r 0 -t ${timeout_poll} -w 20 \
 285                                 127.0.0.1 >/dev/null &
 286 wait_connected $ns 10000
 287 chk_msk_nr 2 "after MPC handshake"
 288 chk_last_time_info 10000
 289 chk_msk_remote_key_nr 2 "....chk remote_key"
 290 chk_msk_fallback_nr 0 "....chk no fallback"
 291 chk_msk_inuse 2
 292 chk_msk_cestab 2
 293 flush_pids
 294
 295 chk_msk_inuse 0 "2->0"
 296 chk_msk_cestab 0 "2->0"
 297
 298 echo "a" | \
 299         timeout ${timeout_test} \
 300                 ip netns exec $ns \
 301                         ./mptcp_connect -p 10001 -l -s TCP -t ${timeout_poll} -w 20 \
 302                                 0.0.0.0 >/dev/null &
 303 mptcp_lib_wait_local_port_listen $ns 10001
 304 echo "b" | \
 305         timeout ${timeout_test} \
 306                 ip netns exec $ns \
 307                         ./mptcp_connect -p 10001 -r 0 -t ${timeout_poll} -w 20 \
 308                                 127.0.0.1 >/dev/null &
 309 wait_connected $ns 10001
 310 chk_msk_fallback_nr 1 "check fallback"
 311 chk_msk_inuse 1
 312 chk_msk_cestab 1
 313 flush_pids
 314
 315 chk_msk_inuse 0 "1->0"
 316 chk_msk_cestab 0 "1->0"
 317
 318 NR_CLIENTS=100
 319 for I in $(seq 1 $NR_CLIENTS); do
 320         echo "a" | \
 321                 timeout ${timeout_test} \
 322                         ip netns exec $ns \
 323                                 ./mptcp_connect -p $((I+10001)) -l -w 20 \
 324                                         -t ${timeout_poll} 0.0.0.0 >/dev/null &
 325 done
 326 mptcp_lib_wait_local_port_listen $ns $((NR_CLIENTS + 10001))
 327
 328 for I in $(seq 1 $NR_CLIENTS); do
 329         echo "b" | \
 330                 timeout ${timeout_test} \
 331                         ip netns exec $ns \
 332                                 ./mptcp_connect -p $((I+10001)) -w 20 \
 333                                         -t ${timeout_poll} 127.0.0.1 >/dev/null &
 334 done
 335
 336 wait_msk_nr $((NR_CLIENTS*2)) "many msk socket present"
 337 chk_msk_inuse $((NR_CLIENTS*2)) "many"
 338 chk_msk_cestab $((NR_CLIENTS*2)) "many"
 339 flush_pids
 340
 341 chk_msk_inuse 0 "many->0"
 342 chk_msk_cestab 0 "many->0"
 343
 344 chk_listener_nr 0 "no listener sockets"
 345 NR_SERVERS=100
 346 for I in $(seq 1 $NR_SERVERS); do
 347         ip netns exec $ns ./mptcp_connect -p $((I + 20001)) \
 348                 -t ${timeout_poll} -l 0.0.0.0 >/dev/null 2>&1 &
 349 done
 350 mptcp_lib_wait_local_port_listen $ns $((NR_SERVERS + 20001))
 351
 352 chk_listener_nr $NR_SERVERS "many listener sockets"
 353
 354 # graceful termination
 355 for I in $(seq 1 $NR_SERVERS); do
 356         echo a | ip netns exec $ns ./mptcp_connect -p $((I + 20001)) 127.0.0.1 >/dev/null 2>&1 &
 357 done
 358 flush_pids
 359
 360 mptcp_lib_result_print_all_tap
 361 exit $ret