Back to home page

OSCL-LXR
 
 

     

0001 #!/bin/bash
0002 # SPDX-License-Identifier: GPL-2.0
0003 
0004 time_start=$(date +%s)
0005 
0006 optstring="S:R:d:e:l:r:h4cm:f:tC"
0007 ret=0
0008 sin=""
0009 sout=""
0010 cin_disconnect=""
0011 cin=""
0012 cout=""
0013 ksft_skip=4
0014 capture=false
0015 timeout_poll=30
0016 timeout_test=$((timeout_poll * 2 + 1))
0017 ipv6=true
0018 ethtool_random_on=true
0019 tc_delay="$((RANDOM%50))"
0020 tc_loss=$((RANDOM%101))
0021 testmode=""
0022 sndbuf=0
0023 rcvbuf=0
0024 options_log=true
0025 do_tcp=0
0026 checksum=false
0027 filesize=0
0028 connect_per_transfer=1
0029 
0030 if [ $tc_loss -eq 100 ];then
0031         tc_loss=1%
0032 elif [ $tc_loss -ge 10 ]; then
0033         tc_loss=0.$tc_loss%
0034 elif [ $tc_loss -ge 1 ]; then
0035         tc_loss=0.0$tc_loss%
0036 else
0037         tc_loss=""
0038 fi
0039 
0040 usage() {
0041         echo "Usage: $0 [ -a ]"
0042         echo -e "\t-d: tc/netem delay in milliseconds, e.g. \"-d 10\" (default random)"
0043         echo -e "\t-l: tc/netem loss percentage, e.g. \"-l 0.02\" (default random)"
0044         echo -e "\t-r: tc/netem reorder mode, e.g. \"-r 25% 50% gap 5\", use "-r 0" to disable reordering (default random)"
0045         echo -e "\t-e: ethtool features to disable, e.g.: \"-e tso -e gso\" (default: randomly disable any of tso/gso/gro)"
0046         echo -e "\t-4: IPv4 only: disable IPv6 tests (default: test both IPv4 and IPv6)"
0047         echo -e "\t-c: capture packets for each test using tcpdump (default: no capture)"
0048         echo -e "\t-f: size of file to transfer in bytes (default random)"
0049         echo -e "\t-S: set sndbuf value (default: use kernel default)"
0050         echo -e "\t-R: set rcvbuf value (default: use kernel default)"
0051         echo -e "\t-m: test mode (poll, sendfile; default: poll)"
0052         echo -e "\t-t: also run tests with TCP (use twice to non-fallback tcp)"
0053         echo -e "\t-C: enable the MPTCP data checksum"
0054 }
0055 
0056 while getopts "$optstring" option;do
0057         case "$option" in
0058         "h")
0059                 usage $0
0060                 exit 0
0061                 ;;
0062         "d")
0063                 if [ $OPTARG -ge 0 ];then
0064                         tc_delay="$OPTARG"
0065                 else
0066                         echo "-d requires numeric argument, got \"$OPTARG\"" 1>&2
0067                         exit 1
0068                 fi
0069                 ;;
0070         "e")
0071                 ethtool_args="$ethtool_args $OPTARG off"
0072                 ethtool_random_on=false
0073                 ;;
0074         "l")
0075                 tc_loss="$OPTARG"
0076                 ;;
0077         "r")
0078                 tc_reorder="$OPTARG"
0079                 ;;
0080         "4")
0081                 ipv6=false
0082                 ;;
0083         "c")
0084                 capture=true
0085                 ;;
0086         "S")
0087                 if [ $OPTARG -ge 0 ];then
0088                         sndbuf="$OPTARG"
0089                 else
0090                         echo "-S requires numeric argument, got \"$OPTARG\"" 1>&2
0091                         exit 1
0092                 fi
0093                 ;;
0094         "R")
0095                 if [ $OPTARG -ge 0 ];then
0096                         rcvbuf="$OPTARG"
0097                 else
0098                         echo "-R requires numeric argument, got \"$OPTARG\"" 1>&2
0099                         exit 1
0100                 fi
0101                 ;;
0102         "m")
0103                 testmode="$OPTARG"
0104                 ;;
0105         "f")
0106                 filesize="$OPTARG"
0107                 ;;
0108         "t")
0109                 do_tcp=$((do_tcp+1))
0110                 ;;
0111         "C")
0112                 checksum=true
0113                 ;;
0114         "?")
0115                 usage $0
0116                 exit 1
0117                 ;;
0118         esac
0119 done
0120 
0121 sec=$(date +%s)
0122 rndh=$(printf %x $sec)-$(mktemp -u XXXXXX)
0123 ns1="ns1-$rndh"
0124 ns2="ns2-$rndh"
0125 ns3="ns3-$rndh"
0126 ns4="ns4-$rndh"
0127 
0128 TEST_COUNT=0
0129 
0130 cleanup()
0131 {
0132         rm -f "$cin_disconnect" "$cout_disconnect"
0133         rm -f "$cin" "$cout"
0134         rm -f "$sin" "$sout"
0135         rm -f "$capout"
0136 
0137         local netns
0138         for netns in "$ns1" "$ns2" "$ns3" "$ns4";do
0139                 ip netns del $netns
0140                 rm -f /tmp/$netns.{nstat,out}
0141         done
0142 }
0143 
0144 ip -Version > /dev/null 2>&1
0145 if [ $? -ne 0 ];then
0146         echo "SKIP: Could not run test without ip tool"
0147         exit $ksft_skip
0148 fi
0149 
0150 sin=$(mktemp)
0151 sout=$(mktemp)
0152 cin=$(mktemp)
0153 cout=$(mktemp)
0154 capout=$(mktemp)
0155 cin_disconnect="$cin".disconnect
0156 cout_disconnect="$cout".disconnect
0157 trap cleanup EXIT
0158 
0159 for i in "$ns1" "$ns2" "$ns3" "$ns4";do
0160         ip netns add $i || exit $ksft_skip
0161         ip -net $i link set lo up
0162 done
0163 
0164 #  "$ns1"              ns2                    ns3                     ns4
0165 # ns1eth2    ns2eth1   ns2eth3      ns3eth2   ns3eth4       ns4eth3
0166 #                           - drop 1% ->            reorder 25%
0167 #                           <- TSO off -
0168 
0169 ip link add ns1eth2 netns "$ns1" type veth peer name ns2eth1 netns "$ns2"
0170 ip link add ns2eth3 netns "$ns2" type veth peer name ns3eth2 netns "$ns3"
0171 ip link add ns3eth4 netns "$ns3" type veth peer name ns4eth3 netns "$ns4"
0172 
0173 ip -net "$ns1" addr add 10.0.1.1/24 dev ns1eth2
0174 ip -net "$ns1" addr add dead:beef:1::1/64 dev ns1eth2 nodad
0175 
0176 ip -net "$ns1" link set ns1eth2 up
0177 ip -net "$ns1" route add default via 10.0.1.2
0178 ip -net "$ns1" route add default via dead:beef:1::2
0179 
0180 ip -net "$ns2" addr add 10.0.1.2/24 dev ns2eth1
0181 ip -net "$ns2" addr add dead:beef:1::2/64 dev ns2eth1 nodad
0182 ip -net "$ns2" link set ns2eth1 up
0183 
0184 ip -net "$ns2" addr add 10.0.2.1/24 dev ns2eth3
0185 ip -net "$ns2" addr add dead:beef:2::1/64 dev ns2eth3 nodad
0186 ip -net "$ns2" link set ns2eth3 up
0187 ip -net "$ns2" route add default via 10.0.2.2
0188 ip -net "$ns2" route add default via dead:beef:2::2
0189 ip netns exec "$ns2" sysctl -q net.ipv4.ip_forward=1
0190 ip netns exec "$ns2" sysctl -q net.ipv6.conf.all.forwarding=1
0191 
0192 ip -net "$ns3" addr add 10.0.2.2/24 dev ns3eth2
0193 ip -net "$ns3" addr add dead:beef:2::2/64 dev ns3eth2 nodad
0194 ip -net "$ns3" link set ns3eth2 up
0195 
0196 ip -net "$ns3" addr add 10.0.3.2/24 dev ns3eth4
0197 ip -net "$ns3" addr add dead:beef:3::2/64 dev ns3eth4 nodad
0198 ip -net "$ns3" link set ns3eth4 up
0199 ip -net "$ns3" route add default via 10.0.2.1
0200 ip -net "$ns3" route add default via dead:beef:2::1
0201 ip netns exec "$ns3" sysctl -q net.ipv4.ip_forward=1
0202 ip netns exec "$ns3" sysctl -q net.ipv6.conf.all.forwarding=1
0203 
0204 ip -net "$ns4" addr add 10.0.3.1/24 dev ns4eth3
0205 ip -net "$ns4" addr add dead:beef:3::1/64 dev ns4eth3 nodad
0206 ip -net "$ns4" link set ns4eth3 up
0207 ip -net "$ns4" route add default via 10.0.3.2
0208 ip -net "$ns4" route add default via dead:beef:3::2
0209 
0210 if $checksum; then
0211         for i in "$ns1" "$ns2" "$ns3" "$ns4";do
0212                 ip netns exec $i sysctl -q net.mptcp.checksum_enabled=1
0213         done
0214 fi
0215 
0216 set_ethtool_flags() {
0217         local ns="$1"
0218         local dev="$2"
0219         local flags="$3"
0220 
0221         ip netns exec $ns ethtool -K $dev $flags 2>/dev/null
0222         [ $? -eq 0 ] && echo "INFO: set $ns dev $dev: ethtool -K $flags"
0223 }
0224 
0225 set_random_ethtool_flags() {
0226         local flags=""
0227         local r=$RANDOM
0228 
0229         local pick1=$((r & 1))
0230         local pick2=$((r & 2))
0231         local pick3=$((r & 4))
0232 
0233         [ $pick1 -ne 0 ] && flags="tso off"
0234         [ $pick2 -ne 0 ] && flags="$flags gso off"
0235         [ $pick3 -ne 0 ] && flags="$flags gro off"
0236 
0237         [ -z "$flags" ] && return
0238 
0239         set_ethtool_flags "$1" "$2" "$flags"
0240 }
0241 
0242 if $ethtool_random_on;then
0243         set_random_ethtool_flags "$ns3" ns3eth2
0244         set_random_ethtool_flags "$ns4" ns4eth3
0245 else
0246         set_ethtool_flags "$ns3" ns3eth2 "$ethtool_args"
0247         set_ethtool_flags "$ns4" ns4eth3 "$ethtool_args"
0248 fi
0249 
0250 print_file_err()
0251 {
0252         ls -l "$1" 1>&2
0253         echo "Trailing bytes are: "
0254         tail -c 27 "$1"
0255 }
0256 
0257 check_transfer()
0258 {
0259         local in=$1
0260         local out=$2
0261         local what=$3
0262 
0263         cmp "$in" "$out" > /dev/null 2>&1
0264         if [ $? -ne 0 ] ;then
0265                 echo "[ FAIL ] $what does not match (in, out):"
0266                 print_file_err "$in"
0267                 print_file_err "$out"
0268 
0269                 return 1
0270         fi
0271 
0272         return 0
0273 }
0274 
0275 check_mptcp_disabled()
0276 {
0277         local disabled_ns
0278         disabled_ns="ns_disabled-$sech-$(mktemp -u XXXXXX)"
0279         ip netns add ${disabled_ns} || exit $ksft_skip
0280 
0281         # net.mptcp.enabled should be enabled by default
0282         if [ "$(ip netns exec ${disabled_ns} sysctl net.mptcp.enabled | awk '{ print $3 }')" -ne 1 ]; then
0283                 echo -e "net.mptcp.enabled sysctl is not 1 by default\t\t[ FAIL ]"
0284                 ret=1
0285                 return 1
0286         fi
0287         ip netns exec ${disabled_ns} sysctl -q net.mptcp.enabled=0
0288 
0289         local err=0
0290         LC_ALL=C ip netns exec ${disabled_ns} ./mptcp_connect -p 10000 -s MPTCP 127.0.0.1 < "$cin" 2>&1 | \
0291                 grep -q "^socket: Protocol not available$" && err=1
0292         ip netns delete ${disabled_ns}
0293 
0294         if [ ${err} -eq 0 ]; then
0295                 echo -e "New MPTCP socket cannot be blocked via sysctl\t\t[ FAIL ]"
0296                 ret=1
0297                 return 1
0298         fi
0299 
0300         echo -e "New MPTCP socket can be blocked via sysctl\t\t[ OK ]"
0301         return 0
0302 }
0303 
0304 # $1: IP address
0305 is_v6()
0306 {
0307         [ -z "${1##*:*}" ]
0308 }
0309 
0310 do_ping()
0311 {
0312         local listener_ns="$1"
0313         local connector_ns="$2"
0314         local connect_addr="$3"
0315         local ping_args="-q -c 1"
0316 
0317         if is_v6 "${connect_addr}"; then
0318                 $ipv6 || return 0
0319                 ping_args="${ping_args} -6"
0320         fi
0321 
0322         ip netns exec ${connector_ns} ping ${ping_args} $connect_addr >/dev/null
0323         if [ $? -ne 0 ] ; then
0324                 echo "$listener_ns -> $connect_addr connectivity [ FAIL ]" 1>&2
0325                 ret=1
0326 
0327                 return 1
0328         fi
0329 
0330         return 0
0331 }
0332 
0333 # $1: ns, $2: MIB counter
0334 get_mib_counter()
0335 {
0336         local listener_ns="${1}"
0337         local mib="${2}"
0338 
0339         # strip the header
0340         ip netns exec "${listener_ns}" \
0341                 nstat -z -a "${mib}" | \
0342                         tail -n+2 | \
0343                         while read a count c rest; do
0344                                 echo $count
0345                         done
0346 }
0347 
0348 # $1: ns, $2: port
0349 wait_local_port_listen()
0350 {
0351         local listener_ns="${1}"
0352         local port="${2}"
0353 
0354         local port_hex i
0355 
0356         port_hex="$(printf "%04X" "${port}")"
0357         for i in $(seq 10); do
0358                 ip netns exec "${listener_ns}" cat /proc/net/tcp* | \
0359                         awk "BEGIN {rc=1} {if (\$2 ~ /:${port_hex}\$/ && \$4 ~ /0A/) {rc=0; exit}} END {exit rc}" &&
0360                         break
0361                 sleep 0.1
0362         done
0363 }
0364 
0365 do_transfer()
0366 {
0367         local listener_ns="$1"
0368         local connector_ns="$2"
0369         local cl_proto="$3"
0370         local srv_proto="$4"
0371         local connect_addr="$5"
0372         local local_addr="$6"
0373         local extra_args="$7"
0374 
0375         local port
0376         port=$((10000+$TEST_COUNT))
0377         TEST_COUNT=$((TEST_COUNT+1))
0378 
0379         if [ "$rcvbuf" -gt 0 ]; then
0380                 extra_args="$extra_args -R $rcvbuf"
0381         fi
0382 
0383         if [ "$sndbuf" -gt 0 ]; then
0384                 extra_args="$extra_args -S $sndbuf"
0385         fi
0386 
0387         if [ -n "$testmode" ]; then
0388                 extra_args="$extra_args -m $testmode"
0389         fi
0390 
0391         if [ -n "$extra_args" ] && $options_log; then
0392                 echo "INFO: extra options: $extra_args"
0393         fi
0394         options_log=false
0395 
0396         :> "$cout"
0397         :> "$sout"
0398         :> "$capout"
0399 
0400         local addr_port
0401         addr_port=$(printf "%s:%d" ${connect_addr} ${port})
0402         printf "%.3s %-5s -> %.3s (%-20s) %-5s\t" ${connector_ns} ${cl_proto} ${listener_ns} ${addr_port} ${srv_proto}
0403 
0404         if $capture; then
0405                 local capuser
0406                 if [ -z $SUDO_USER ] ; then
0407                         capuser=""
0408                 else
0409                         capuser="-Z $SUDO_USER"
0410                 fi
0411 
0412                 local capfile="${rndh}-${connector_ns:0:3}-${listener_ns:0:3}-${cl_proto}-${srv_proto}-${connect_addr}-${port}"
0413                 local capopt="-i any -s 65535 -B 32768 ${capuser}"
0414 
0415                 ip netns exec ${listener_ns}  tcpdump ${capopt} -w "${capfile}-listener.pcap"  >> "${capout}" 2>&1 &
0416                 local cappid_listener=$!
0417 
0418                 ip netns exec ${connector_ns} tcpdump ${capopt} -w "${capfile}-connector.pcap" >> "${capout}" 2>&1 &
0419                 local cappid_connector=$!
0420 
0421                 sleep 1
0422         fi
0423 
0424         NSTAT_HISTORY=/tmp/${listener_ns}.nstat ip netns exec ${listener_ns} \
0425                 nstat -n
0426         if [ ${listener_ns} != ${connector_ns} ]; then
0427                 NSTAT_HISTORY=/tmp/${connector_ns}.nstat ip netns exec ${connector_ns} \
0428                         nstat -n
0429         fi
0430 
0431         local stat_synrx_last_l=$(get_mib_counter "${listener_ns}" "MPTcpExtMPCapableSYNRX")
0432         local stat_ackrx_last_l=$(get_mib_counter "${listener_ns}" "MPTcpExtMPCapableACKRX")
0433         local stat_cookietx_last=$(get_mib_counter "${listener_ns}" "TcpExtSyncookiesSent")
0434         local stat_cookierx_last=$(get_mib_counter "${listener_ns}" "TcpExtSyncookiesRecv")
0435         local stat_csum_err_s=$(get_mib_counter "${listener_ns}" "MPTcpExtDataCsumErr")
0436         local stat_csum_err_c=$(get_mib_counter "${connector_ns}" "MPTcpExtDataCsumErr")
0437 
0438         timeout ${timeout_test} \
0439                 ip netns exec ${listener_ns} \
0440                         ./mptcp_connect -t ${timeout_poll} -l -p $port -s ${srv_proto} \
0441                                 $extra_args $local_addr < "$sin" > "$sout" &
0442         local spid=$!
0443 
0444         wait_local_port_listen "${listener_ns}" "${port}"
0445 
0446         local start
0447         start=$(date +%s%3N)
0448         timeout ${timeout_test} \
0449                 ip netns exec ${connector_ns} \
0450                         ./mptcp_connect -t ${timeout_poll} -p $port -s ${cl_proto} \
0451                                 $extra_args $connect_addr < "$cin" > "$cout" &
0452         local cpid=$!
0453 
0454         wait $cpid
0455         local retc=$?
0456         wait $spid
0457         local rets=$?
0458 
0459         local stop
0460         stop=$(date +%s%3N)
0461 
0462         if $capture; then
0463                 sleep 1
0464                 kill ${cappid_listener}
0465                 kill ${cappid_connector}
0466         fi
0467 
0468         NSTAT_HISTORY=/tmp/${listener_ns}.nstat ip netns exec ${listener_ns} \
0469                 nstat | grep Tcp > /tmp/${listener_ns}.out
0470         if [ ${listener_ns} != ${connector_ns} ]; then
0471                 NSTAT_HISTORY=/tmp/${connector_ns}.nstat ip netns exec ${connector_ns} \
0472                         nstat | grep Tcp > /tmp/${connector_ns}.out
0473         fi
0474 
0475         local duration
0476         duration=$((stop-start))
0477         printf "(duration %05sms) " "${duration}"
0478         if [ ${rets} -ne 0 ] || [ ${retc} -ne 0 ]; then
0479                 echo "[ FAIL ] client exit code $retc, server $rets" 1>&2
0480                 echo -e "\nnetns ${listener_ns} socket stat for ${port}:" 1>&2
0481                 ip netns exec ${listener_ns} ss -Menita 1>&2 -o "sport = :$port"
0482                 cat /tmp/${listener_ns}.out
0483                 echo -e "\nnetns ${connector_ns} socket stat for ${port}:" 1>&2
0484                 ip netns exec ${connector_ns} ss -Menita 1>&2 -o "dport = :$port"
0485                 [ ${listener_ns} != ${connector_ns} ] && cat /tmp/${connector_ns}.out
0486 
0487                 echo
0488                 cat "$capout"
0489                 return 1
0490         fi
0491 
0492         check_transfer $sin $cout "file received by client"
0493         retc=$?
0494         check_transfer $cin $sout "file received by server"
0495         rets=$?
0496 
0497         local stat_synrx_now_l=$(get_mib_counter "${listener_ns}" "MPTcpExtMPCapableSYNRX")
0498         local stat_ackrx_now_l=$(get_mib_counter "${listener_ns}" "MPTcpExtMPCapableACKRX")
0499         local stat_cookietx_now=$(get_mib_counter "${listener_ns}" "TcpExtSyncookiesSent")
0500         local stat_cookierx_now=$(get_mib_counter "${listener_ns}" "TcpExtSyncookiesRecv")
0501         local stat_ooo_now=$(get_mib_counter "${listener_ns}" "TcpExtTCPOFOQueue")
0502 
0503         expect_synrx=$((stat_synrx_last_l))
0504         expect_ackrx=$((stat_ackrx_last_l))
0505 
0506         cookies=$(ip netns exec ${listener_ns} sysctl net.ipv4.tcp_syncookies)
0507         cookies=${cookies##*=}
0508 
0509         if [ ${cl_proto} = "MPTCP" ] && [ ${srv_proto} = "MPTCP" ]; then
0510                 expect_synrx=$((stat_synrx_last_l+$connect_per_transfer))
0511                 expect_ackrx=$((stat_ackrx_last_l+$connect_per_transfer))
0512         fi
0513 
0514         if [ ${stat_synrx_now_l} -lt ${expect_synrx} ]; then
0515                 printf "[ FAIL ] lower MPC SYN rx (%d) than expected (%d)\n" \
0516                         "${stat_synrx_now_l}" "${expect_synrx}" 1>&2
0517                 retc=1
0518         fi
0519         if [ ${stat_ackrx_now_l} -lt ${expect_ackrx} -a ${stat_ooo_now} -eq 0 ]; then
0520                 if [ ${stat_ooo_now} -eq 0 ]; then
0521                         printf "[ FAIL ] lower MPC ACK rx (%d) than expected (%d)\n" \
0522                                 "${stat_ackrx_now_l}" "${expect_ackrx}" 1>&2
0523                         rets=1
0524                 else
0525                         printf "[ Note ] fallback due to TCP OoO"
0526                 fi
0527         fi
0528 
0529         if $checksum; then
0530                 local csum_err_s=$(get_mib_counter "${listener_ns}" "MPTcpExtDataCsumErr")
0531                 local csum_err_c=$(get_mib_counter "${connector_ns}" "MPTcpExtDataCsumErr")
0532 
0533                 local csum_err_s_nr=$((csum_err_s - stat_csum_err_s))
0534                 if [ $csum_err_s_nr -gt 0 ]; then
0535                         printf "[ FAIL ]\nserver got $csum_err_s_nr data checksum error[s]"
0536                         rets=1
0537                 fi
0538 
0539                 local csum_err_c_nr=$((csum_err_c - stat_csum_err_c))
0540                 if [ $csum_err_c_nr -gt 0 ]; then
0541                         printf "[ FAIL ]\nclient got $csum_err_c_nr data checksum error[s]"
0542                         retc=1
0543                 fi
0544         fi
0545 
0546         if [ $retc -eq 0 ] && [ $rets -eq 0 ]; then
0547                 printf "[ OK ]"
0548         fi
0549 
0550         if [ $cookies -eq 2 ];then
0551                 if [ $stat_cookietx_last -ge $stat_cookietx_now ] ;then
0552                         printf " WARN: CookieSent: did not advance"
0553                 fi
0554                 if [ $stat_cookierx_last -ge $stat_cookierx_now ] ;then
0555                         printf " WARN: CookieRecv: did not advance"
0556                 fi
0557         else
0558                 if [ $stat_cookietx_last -ne $stat_cookietx_now ] ;then
0559                         printf " WARN: CookieSent: changed"
0560                 fi
0561                 if [ $stat_cookierx_last -ne $stat_cookierx_now ] ;then
0562                         printf " WARN: CookieRecv: changed"
0563                 fi
0564         fi
0565 
0566         if [ ${stat_synrx_now_l} -gt ${expect_synrx} ]; then
0567                 printf " WARN: SYNRX: expect %d, got %d (probably retransmissions)" \
0568                         "${expect_synrx}" "${stat_synrx_now_l}"
0569         fi
0570         if [ ${stat_ackrx_now_l} -gt ${expect_ackrx} ]; then
0571                 printf " WARN: ACKRX: expect %d, got %d (probably retransmissions)" \
0572                         "${expect_ackrx}" "${stat_ackrx_now_l}"
0573         fi
0574 
0575         echo
0576         cat "$capout"
0577         [ $retc -eq 0 ] && [ $rets -eq 0 ]
0578 }
0579 
0580 make_file()
0581 {
0582         local name=$1
0583         local who=$2
0584         local SIZE=$filesize
0585         local ksize
0586         local rem
0587 
0588         if [ $SIZE -eq 0 ]; then
0589                 local MAXSIZE=$((1024 * 1024 * 8))
0590                 local MINSIZE=$((1024 * 256))
0591 
0592                 SIZE=$(((RANDOM * RANDOM + MINSIZE) % MAXSIZE))
0593         fi
0594 
0595         ksize=$((SIZE / 1024))
0596         rem=$((SIZE - (ksize * 1024)))
0597 
0598         dd if=/dev/urandom of="$name" bs=1024 count=$ksize 2> /dev/null
0599         dd if=/dev/urandom conv=notrunc of="$name" bs=1 count=$rem 2> /dev/null
0600         echo -e "\nMPTCP_TEST_FILE_END_MARKER" >> "$name"
0601 
0602         echo "Created $name (size $(du -b "$name")) containing data sent by $who"
0603 }
0604 
0605 run_tests_lo()
0606 {
0607         local listener_ns="$1"
0608         local connector_ns="$2"
0609         local connect_addr="$3"
0610         local loopback="$4"
0611         local extra_args="$5"
0612         local lret=0
0613 
0614         # skip if test programs are running inside same netns for subsequent runs.
0615         if [ $loopback -eq 0 ] && [ ${listener_ns} = ${connector_ns} ]; then
0616                 return 0
0617         fi
0618 
0619         # skip if we don't want v6
0620         if ! $ipv6 && is_v6 "${connect_addr}"; then
0621                 return 0
0622         fi
0623 
0624         local local_addr
0625         if is_v6 "${connect_addr}"; then
0626                 local_addr="::"
0627         else
0628                 local_addr="0.0.0.0"
0629         fi
0630 
0631         do_transfer ${listener_ns} ${connector_ns} MPTCP MPTCP \
0632                     ${connect_addr} ${local_addr} "${extra_args}"
0633         lret=$?
0634         if [ $lret -ne 0 ]; then
0635                 ret=$lret
0636                 return 1
0637         fi
0638 
0639         if [ $do_tcp -eq 0 ]; then
0640                 # don't bother testing fallback tcp except for loopback case.
0641                 if [ ${listener_ns} != ${connector_ns} ]; then
0642                         return 0
0643                 fi
0644         fi
0645 
0646         do_transfer ${listener_ns} ${connector_ns} MPTCP TCP \
0647                     ${connect_addr} ${local_addr} "${extra_args}"
0648         lret=$?
0649         if [ $lret -ne 0 ]; then
0650                 ret=$lret
0651                 return 1
0652         fi
0653 
0654         do_transfer ${listener_ns} ${connector_ns} TCP MPTCP \
0655                     ${connect_addr} ${local_addr} "${extra_args}"
0656         lret=$?
0657         if [ $lret -ne 0 ]; then
0658                 ret=$lret
0659                 return 1
0660         fi
0661 
0662         if [ $do_tcp -gt 1 ] ;then
0663                 do_transfer ${listener_ns} ${connector_ns} TCP TCP \
0664                             ${connect_addr} ${local_addr} "${extra_args}"
0665                 lret=$?
0666                 if [ $lret -ne 0 ]; then
0667                         ret=$lret
0668                         return 1
0669                 fi
0670         fi
0671 
0672         return 0
0673 }
0674 
0675 run_tests()
0676 {
0677         run_tests_lo $1 $2 $3 0
0678 }
0679 
0680 run_test_transparent()
0681 {
0682         local connect_addr="$1"
0683         local msg="$2"
0684 
0685         local connector_ns="$ns1"
0686         local listener_ns="$ns2"
0687         local lret=0
0688         local r6flag=""
0689 
0690         # skip if we don't want v6
0691         if ! $ipv6 && is_v6 "${connect_addr}"; then
0692                 return 0
0693         fi
0694 
0695 ip netns exec "$listener_ns" nft -f /dev/stdin <<"EOF"
0696 flush ruleset
0697 table inet mangle {
0698         chain divert {
0699                 type filter hook prerouting priority -150;
0700 
0701                 meta l4proto tcp socket transparent 1 meta mark set 1 accept
0702                 tcp dport 20000 tproxy to :20000 meta mark set 1 accept
0703         }
0704 }
0705 EOF
0706         if [ $? -ne 0 ]; then
0707                 echo "SKIP: $msg, could not load nft ruleset"
0708                 return
0709         fi
0710 
0711         local local_addr
0712         if is_v6 "${connect_addr}"; then
0713                 local_addr="::"
0714                 r6flag="-6"
0715         else
0716                 local_addr="0.0.0.0"
0717         fi
0718 
0719         ip -net "$listener_ns" $r6flag rule add fwmark 1 lookup 100
0720         if [ $? -ne 0 ]; then
0721                 ip netns exec "$listener_ns" nft flush ruleset
0722                 echo "SKIP: $msg, ip $r6flag rule failed"
0723                 return
0724         fi
0725 
0726         ip -net "$listener_ns" route add local $local_addr/0 dev lo table 100
0727         if [ $? -ne 0 ]; then
0728                 ip netns exec "$listener_ns" nft flush ruleset
0729                 ip -net "$listener_ns" $r6flag rule del fwmark 1 lookup 100
0730                 echo "SKIP: $msg, ip route add local $local_addr failed"
0731                 return
0732         fi
0733 
0734         echo "INFO: test $msg"
0735 
0736         TEST_COUNT=10000
0737         local extra_args="-o TRANSPARENT"
0738         do_transfer ${listener_ns} ${connector_ns} MPTCP MPTCP \
0739                     ${connect_addr} ${local_addr} "${extra_args}"
0740         lret=$?
0741 
0742         ip netns exec "$listener_ns" nft flush ruleset
0743         ip -net "$listener_ns" $r6flag rule del fwmark 1 lookup 100
0744         ip -net "$listener_ns" route del local $local_addr/0 dev lo table 100
0745 
0746         if [ $lret -ne 0 ]; then
0747                 echo "FAIL: $msg, mptcp connection error" 1>&2
0748                 ret=$lret
0749                 return 1
0750         fi
0751 
0752         echo "PASS: $msg"
0753         return 0
0754 }
0755 
0756 run_tests_peekmode()
0757 {
0758         local peekmode="$1"
0759 
0760         echo "INFO: with peek mode: ${peekmode}"
0761         run_tests_lo "$ns1" "$ns1" 10.0.1.1 1 "-P ${peekmode}"
0762         run_tests_lo "$ns1" "$ns1" dead:beef:1::1 1 "-P ${peekmode}"
0763 }
0764 
0765 run_tests_disconnect()
0766 {
0767         local peekmode="$1"
0768         local old_cin=$cin
0769         local old_sin=$sin
0770 
0771         cat $cin $cin $cin > "$cin".disconnect
0772 
0773         # force do_transfer to cope with the multiple tranmissions
0774         sin="$cin.disconnect"
0775         sin_disconnect=$old_sin
0776         cin="$cin.disconnect"
0777         cin_disconnect="$old_cin"
0778         connect_per_transfer=3
0779 
0780         echo "INFO: disconnect"
0781         run_tests_lo "$ns1" "$ns1" 10.0.1.1 1 "-I 3 -i $old_cin"
0782         run_tests_lo "$ns1" "$ns1" dead:beef:1::1 1 "-I 3 -i $old_cin"
0783 
0784         # restore previous status
0785         sin=$old_sin
0786         sin_disconnect="$cout".disconnect
0787         cin=$old_cin
0788         cin_disconnect="$cin".disconnect
0789         connect_per_transfer=1
0790 }
0791 
0792 display_time()
0793 {
0794         time_end=$(date +%s)
0795         time_run=$((time_end-time_start))
0796 
0797         echo "Time: ${time_run} seconds"
0798 }
0799 
0800 stop_if_error()
0801 {
0802         local msg="$1"
0803 
0804         if [ ${ret} -ne 0 ]; then
0805                 echo "FAIL: ${msg}" 1>&2
0806                 display_time
0807                 exit ${ret}
0808         fi
0809 }
0810 
0811 make_file "$cin" "client"
0812 make_file "$sin" "server"
0813 
0814 check_mptcp_disabled
0815 
0816 stop_if_error "The kernel configuration is not valid for MPTCP"
0817 
0818 echo "INFO: validating network environment with pings"
0819 for sender in "$ns1" "$ns2" "$ns3" "$ns4";do
0820         do_ping "$ns1" $sender 10.0.1.1
0821         do_ping "$ns1" $sender dead:beef:1::1
0822 
0823         do_ping "$ns2" $sender 10.0.1.2
0824         do_ping "$ns2" $sender dead:beef:1::2
0825         do_ping "$ns2" $sender 10.0.2.1
0826         do_ping "$ns2" $sender dead:beef:2::1
0827 
0828         do_ping "$ns3" $sender 10.0.2.2
0829         do_ping "$ns3" $sender dead:beef:2::2
0830         do_ping "$ns3" $sender 10.0.3.2
0831         do_ping "$ns3" $sender dead:beef:3::2
0832 
0833         do_ping "$ns4" $sender 10.0.3.1
0834         do_ping "$ns4" $sender dead:beef:3::1
0835 done
0836 
0837 stop_if_error "Could not even run ping tests"
0838 
0839 [ -n "$tc_loss" ] && tc -net "$ns2" qdisc add dev ns2eth3 root netem loss random $tc_loss delay ${tc_delay}ms
0840 echo -n "INFO: Using loss of $tc_loss "
0841 test "$tc_delay" -gt 0 && echo -n "delay $tc_delay ms "
0842 
0843 reorder_delay=$(($tc_delay / 4))
0844 
0845 if [ -z "${tc_reorder}" ]; then
0846         reorder1=$((RANDOM%10))
0847         reorder1=$((100 - reorder1))
0848         reorder2=$((RANDOM%100))
0849 
0850         if [ $reorder_delay -gt 0 ] && [ $reorder1 -lt 100 ] && [ $reorder2 -gt 0 ]; then
0851                 tc_reorder="reorder ${reorder1}% ${reorder2}%"
0852                 echo -n "$tc_reorder with delay ${reorder_delay}ms "
0853         fi
0854 elif [ "$tc_reorder" = "0" ];then
0855         tc_reorder=""
0856 elif [ "$reorder_delay" -gt 0 ];then
0857         # reordering requires some delay
0858         tc_reorder="reorder $tc_reorder"
0859         echo -n "$tc_reorder with delay ${reorder_delay}ms "
0860 fi
0861 
0862 echo "on ns3eth4"
0863 
0864 tc -net "$ns3" qdisc add dev ns3eth4 root netem delay ${reorder_delay}ms $tc_reorder
0865 
0866 run_tests_lo "$ns1" "$ns1" 10.0.1.1 1
0867 stop_if_error "Could not even run loopback test"
0868 
0869 run_tests_lo "$ns1" "$ns1" dead:beef:1::1 1
0870 stop_if_error "Could not even run loopback v6 test"
0871 
0872 for sender in $ns1 $ns2 $ns3 $ns4;do
0873         # ns1<->ns2 is not subject to reordering/tc delays. Use it to test
0874         # mptcp syncookie support.
0875         if [ $sender = $ns1 ]; then
0876                 ip netns exec "$ns2" sysctl -q net.ipv4.tcp_syncookies=2
0877         else
0878                 ip netns exec "$ns2" sysctl -q net.ipv4.tcp_syncookies=1
0879         fi
0880 
0881         run_tests "$ns1" $sender 10.0.1.1
0882         run_tests "$ns1" $sender dead:beef:1::1
0883 
0884         run_tests "$ns2" $sender 10.0.1.2
0885         run_tests "$ns2" $sender dead:beef:1::2
0886         run_tests "$ns2" $sender 10.0.2.1
0887         run_tests "$ns2" $sender dead:beef:2::1
0888 
0889         run_tests "$ns3" $sender 10.0.2.2
0890         run_tests "$ns3" $sender dead:beef:2::2
0891         run_tests "$ns3" $sender 10.0.3.2
0892         run_tests "$ns3" $sender dead:beef:3::2
0893 
0894         run_tests "$ns4" $sender 10.0.3.1
0895         run_tests "$ns4" $sender dead:beef:3::1
0896 
0897         stop_if_error "Tests with $sender as a sender have failed"
0898 done
0899 
0900 run_tests_peekmode "saveWithPeek"
0901 run_tests_peekmode "saveAfterPeek"
0902 stop_if_error "Tests with peek mode have failed"
0903 
0904 # connect to ns4 ip address, ns2 should intercept/proxy
0905 run_test_transparent 10.0.3.1 "tproxy ipv4"
0906 run_test_transparent dead:beef:3::1 "tproxy ipv6"
0907 stop_if_error "Tests with tproxy have failed"
0908 
0909 run_tests_disconnect
0910 
0911 display_time
0912 exit $ret
[ Source navigation ] [ Diff markup ] [ Identifier search ] [ general search ]

This page was automatically generated by the 2.1.0 LXR engine.

The LXR team
Valid CSS 2.1! Valid HTML 4.01!