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