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