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