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

   1 #!/bin/bash
   2 # SPDX-License-Identifier: GPL-2.0
   3
   4 . "$(dirname "${0}")/mptcp_lib.sh"
   5
   6 ret=0
   7 sin=""
   8 sout=""
   9 cin=""
  10 cout=""
  11 ksft_skip=4
  12 timeout_poll=30
  13 timeout_test=$((timeout_poll * 2 + 1))
  14 mptcp_connect=""
  15
  16 sec=$(date +%s)
  17 rndh=$(printf %x $sec)-$(mktemp -u XXXXXX)
  18 ns1="ns1-$rndh"
  19 ns2="ns2-$rndh"
  20 ns_sbox="ns_sbox-$rndh"
  21
  22 add_mark_rules()
  23 {
  24         local ns=$1
  25         local m=$2
  26
  27         local t
  28         for t in iptables ip6tables; do
  29                 # just to debug: check we have multiple subflows connection requests
  30                 ip netns exec $ns $t -A OUTPUT -p tcp --syn -m mark --mark $m -j ACCEPT
  31
  32                 # RST packets might be handled by a internal dummy socket
  33                 ip netns exec $ns $t -A OUTPUT -p tcp --tcp-flags RST RST -m mark --mark 0 -j ACCEPT
  34
  35                 ip netns exec $ns $t -A OUTPUT -p tcp -m mark --mark $m -j ACCEPT
  36                 ip netns exec $ns $t -A OUTPUT -p tcp -m mark --mark 0 -j DROP
  37         done
  38 }
  39
  40 init()
  41 {
  42         local netns
  43         for netns in "$ns1" "$ns2" "$ns_sbox";do
  44                 ip netns add $netns || exit $ksft_skip
  45                 ip -net $netns link set lo up
  46                 ip netns exec $netns sysctl -q net.mptcp.enabled=1
  47                 ip netns exec $netns sysctl -q net.ipv4.conf.all.rp_filter=0
  48                 ip netns exec $netns sysctl -q net.ipv4.conf.default.rp_filter=0
  49         done
  50
  51         local i
  52         for i in `seq 1 4`; do
  53                 ip link add ns1eth$i netns "$ns1" type veth peer name ns2eth$i netns "$ns2"
  54                 ip -net "$ns1" addr add 10.0.$i.1/24 dev ns1eth$i
  55                 ip -net "$ns1" addr add dead:beef:$i::1/64 dev ns1eth$i nodad
  56                 ip -net "$ns1" link set ns1eth$i up
  57
  58                 ip -net "$ns2" addr add 10.0.$i.2/24 dev ns2eth$i
  59                 ip -net "$ns2" addr add dead:beef:$i::2/64 dev ns2eth$i nodad
  60                 ip -net "$ns2" link set ns2eth$i up
  61
  62                 # let $ns2 reach any $ns1 address from any interface
  63                 ip -net "$ns2" route add default via 10.0.$i.1 dev ns2eth$i metric 10$i
  64
  65                 ip netns exec $ns1 ./pm_nl_ctl add 10.0.$i.1 flags signal
  66                 ip netns exec $ns1 ./pm_nl_ctl add dead:beef:$i::1 flags signal
  67
  68                 ip netns exec $ns2 ./pm_nl_ctl add 10.0.$i.2 flags signal
  69                 ip netns exec $ns2 ./pm_nl_ctl add dead:beef:$i::2 flags signal
  70         done
  71
  72         ip netns exec $ns1 ./pm_nl_ctl limits 8 8
  73         ip netns exec $ns2 ./pm_nl_ctl limits 8 8
  74
  75         add_mark_rules $ns1 1
  76         add_mark_rules $ns2 2
  77 }
  78
  79 cleanup()
  80 {
  81         local netns
  82         for netns in "$ns1" "$ns2" "$ns_sbox"; do
  83                 ip netns del $netns
  84         done
  85         rm -f "$cin" "$cout"
  86         rm -f "$sin" "$sout"
  87 }
  88
  89 mptcp_lib_check_mptcp
  90
  91 ip -Version > /dev/null 2>&1
  92 if [ $? -ne 0 ];then
  93         echo "SKIP: Could not run test without ip tool"
  94         exit $ksft_skip
  95 fi
  96
  97 iptables -V > /dev/null 2>&1
  98 if [ $? -ne 0 ];then
  99         echo "SKIP: Could not run all tests without iptables tool"
 100         exit $ksft_skip
 101 fi
 102
 103 ip6tables -V > /dev/null 2>&1
 104 if [ $? -ne 0 ];then
 105         echo "SKIP: Could not run all tests without ip6tables tool"
 106         exit $ksft_skip
 107 fi
 108
 109 check_mark()
 110 {
 111         local ns=$1
 112         local af=$2
 113
 114         local tables=iptables
 115
 116         if [ $af -eq 6 ];then
 117                 tables=ip6tables
 118         fi
 119
 120         local counters values
 121         counters=$(ip netns exec $ns $tables -v -L OUTPUT | grep DROP)
 122         values=${counters%DROP*}
 123
 124         local v
 125         for v in $values; do
 126                 if [ $v -ne 0 ]; then
 127                         echo "FAIL: got $tables $values in ns $ns , not 0 - not all expected packets marked" 1>&2
 128                         return 1
 129                 fi
 130         done
 131
 132         return 0
 133 }
 134
 135 print_file_err()
 136 {
 137         ls -l "$1" 1>&2
 138         echo "Trailing bytes are: "
 139         tail -c 27 "$1"
 140 }
 141
 142 check_transfer()
 143 {
 144         local in=$1
 145         local out=$2
 146         local what=$3
 147
 148         cmp "$in" "$out" > /dev/null 2>&1
 149         if [ $? -ne 0 ] ;then
 150                 echo "[ FAIL ] $what does not match (in, out):"
 151                 print_file_err "$in"
 152                 print_file_err "$out"
 153                 ret=1
 154
 155                 return 1
 156         fi
 157
 158         return 0
 159 }
 160
 161 # $1: IP address
 162 is_v6()
 163 {
 164         [ -z "${1##*:*}" ]
 165 }
 166
 167 do_transfer()
 168 {
 169         local listener_ns="$1"
 170         local connector_ns="$2"
 171         local cl_proto="$3"
 172         local srv_proto="$4"
 173         local connect_addr="$5"
 174
 175         local port=12001
 176
 177         :> "$cout"
 178         :> "$sout"
 179
 180         local mptcp_connect="./mptcp_connect -r 20"
 181
 182         local local_addr
 183         if is_v6 "${connect_addr}"; then
 184                 local_addr="::"
 185         else
 186                 local_addr="0.0.0.0"
 187         fi
 188
 189         timeout ${timeout_test} \
 190                 ip netns exec ${listener_ns} \
 191                         $mptcp_connect -t ${timeout_poll} -l -M 1 -p $port -s ${srv_proto} -c TIMESTAMPNS,TCPINQ \
 192                                 ${local_addr} < "$sin" > "$sout" &
 193         local spid=$!
 194
 195         sleep 1
 196
 197         timeout ${timeout_test} \
 198                 ip netns exec ${connector_ns} \
 199                         $mptcp_connect -t ${timeout_poll} -M 2 -p $port -s ${cl_proto} -c TIMESTAMPNS,TCPINQ \
 200                                 $connect_addr < "$cin" > "$cout" &
 201
 202         local cpid=$!
 203
 204         wait $cpid
 205         local retc=$?
 206         wait $spid
 207         local rets=$?
 208
 209         if [ ${rets} -ne 0 ] || [ ${retc} -ne 0 ]; then
 210                 echo " client exit code $retc, server $rets" 1>&2
 211                 echo -e "\nnetns ${listener_ns} socket stat for ${port}:" 1>&2
 212                 ip netns exec ${listener_ns} ss -Menita 1>&2 -o "sport = :$port"
 213
 214                 echo -e "\nnetns ${connector_ns} socket stat for ${port}:" 1>&2
 215                 ip netns exec ${connector_ns} ss -Menita 1>&2 -o "dport = :$port"
 216
 217                 ret=1
 218                 return 1
 219         fi
 220
 221         if [ $local_addr = "::" ];then
 222                 check_mark $listener_ns 6
 223                 check_mark $connector_ns 6
 224         else
 225                 check_mark $listener_ns 4
 226                 check_mark $connector_ns 4
 227         fi
 228
 229         check_transfer $cin $sout "file received by server"
 230
 231         rets=$?
 232
 233         if [ $retc -eq 0 ] && [ $rets -eq 0 ];then
 234                 return 0
 235         fi
 236
 237         return 1
 238 }
 239
 240 make_file()
 241 {
 242         local name=$1
 243         local who=$2
 244         local size=$3
 245
 246         dd if=/dev/urandom of="$name" bs=1024 count=$size 2> /dev/null
 247         echo -e "\nMPTCP_TEST_FILE_END_MARKER" >> "$name"
 248
 249         echo "Created $name (size $size KB) containing data sent by $who"
 250 }
 251
 252 do_mptcp_sockopt_tests()
 253 {
 254         local lret=0
 255
 256         ip netns exec "$ns_sbox" ./mptcp_sockopt
 257         lret=$?
 258
 259         if [ $lret -ne 0 ]; then
 260                 echo "FAIL: SOL_MPTCP getsockopt" 1>&2
 261                 ret=$lret
 262                 return
 263         fi
 264
 265         ip netns exec "$ns_sbox" ./mptcp_sockopt -6
 266         lret=$?
 267
 268         if [ $lret -ne 0 ]; then
 269                 echo "FAIL: SOL_MPTCP getsockopt (ipv6)" 1>&2
 270                 ret=$lret
 271                 return
 272         fi
 273 }
 274
 275 run_tests()
 276 {
 277         local listener_ns="$1"
 278         local connector_ns="$2"
 279         local connect_addr="$3"
 280         local lret=0
 281
 282         do_transfer ${listener_ns} ${connector_ns} MPTCP MPTCP ${connect_addr}
 283
 284         lret=$?
 285
 286         if [ $lret -ne 0 ]; then
 287                 ret=$lret
 288                 return
 289         fi
 290 }
 291
 292 do_tcpinq_test()
 293 {
 294         ip netns exec "$ns_sbox" ./mptcp_inq "$@"
 295         local lret=$?
 296         if [ $lret -ne 0 ];then
 297                 ret=$lret
 298                 echo "FAIL: mptcp_inq $@" 1>&2
 299                 return $lret
 300         fi
 301
 302         echo "PASS: TCP_INQ cmsg/ioctl $@"
 303         return $lret
 304 }
 305
 306 do_tcpinq_tests()
 307 {
 308         local lret=0
 309
 310         local args
 311         for args in "-t tcp" "-r tcp"; do
 312                 do_tcpinq_test $args
 313                 lret=$?
 314                 if [ $lret -ne 0 ] ; then
 315                         return $lret
 316                 fi
 317                 do_tcpinq_test -6 $args
 318                 lret=$?
 319                 if [ $lret -ne 0 ] ; then
 320                         return $lret
 321                 fi
 322         done
 323
 324         do_tcpinq_test -r tcp -t tcp
 325
 326         return $?
 327 }
 328
 329 sin=$(mktemp)
 330 sout=$(mktemp)
 331 cin=$(mktemp)
 332 cout=$(mktemp)
 333 init
 334 make_file "$cin" "client" 1
 335 make_file "$sin" "server" 1
 336 trap cleanup EXIT
 337
 338 run_tests $ns1 $ns2 10.0.1.1
 339 run_tests $ns1 $ns2 dead:beef:1::1
 340
 341 if [ $ret -eq 0 ];then
 342         echo "PASS: all packets had packet mark set"
 343 fi
 344
 345 do_mptcp_sockopt_tests
 346 if [ $ret -eq 0 ];then
 347         echo "PASS: SOL_MPTCP getsockopt has expected information"
 348 fi
 349
 350 do_tcpinq_tests
 351 exit $ret