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
11ns=""
12timeout_poll=30
13timeout_test=$((timeout_poll * 2 + 1))
14ret=0
15
16flush_pids()
17{
18	# mptcp_connect in join mode will sleep a bit before completing,
19	# give it some time
20	sleep 1.1
21
22	ip netns pids "${ns}" | xargs --no-run-if-empty kill -SIGUSR1 &>/dev/null
23
24	for _ in $(seq $((timeout_poll * 10))); do
25		[ -z "$(ip netns pids "${ns}")" ] && break
26		sleep 0.1
27	done
28}
29
30# This function is used in the cleanup trap
31#shellcheck disable=SC2317
32cleanup()
33{
34	ip netns pids "${ns}" | xargs --no-run-if-empty kill -SIGKILL &>/dev/null
35
36	mptcp_lib_ns_exit "${ns}"
37}
38
39mptcp_lib_check_mptcp
40mptcp_lib_check_tools ip ss
41
42get_msk_inuse()
43{
44	ip netns exec $ns cat /proc/net/protocols | awk '$1~/^MPTCP$/{print $3}'
45}
46
47__chk_nr()
48{
49	local command="$1"
50	local expected=$2
51	local msg="$3"
52	local skip="${4-SKIP}"
53	local nr
54
55	nr=$(eval $command)
56
57	mptcp_lib_print_title "$msg"
58	if [ "$nr" != "$expected" ]; then
59		if [ "$nr" = "$skip" ] && ! mptcp_lib_expect_all_features; then
60			mptcp_lib_pr_skip "Feature probably not supported"
61			mptcp_lib_result_skip "${msg}"
62		else
63			mptcp_lib_pr_fail "expected $expected found $nr"
64			mptcp_lib_result_fail "${msg}"
65			ret=${KSFT_FAIL}
66		fi
67	else
68		mptcp_lib_pr_ok
69		mptcp_lib_result_pass "${msg}"
70	fi
71}
72
73__chk_msk_nr()
74{
75	local condition=$1
76	shift 1
77
78	__chk_nr "ss -inmHMN $ns | $condition" "$@"
79}
80
81chk_msk_nr()
82{
83	__chk_msk_nr "grep -c token:" "$@"
84}
85
86chk_listener_nr()
87{
88	local expected=$1
89	local msg="$2"
90
91	__chk_nr "ss -nlHMON $ns | wc -l" "$expected" "$msg - mptcp" 0
92	__chk_nr "ss -nlHtON $ns | wc -l" "$expected" "$msg - subflows"
93}
94
95wait_msk_nr()
96{
97	local condition="grep -c token:"
98	local expected=$1
99	local timeout=20
100	local msg nr
101	local max=0
102	local i=0
103
104	shift 1
105	msg=$*
106
107	while [ $i -lt $timeout ]; do
108		nr=$(ss -inmHMN $ns | $condition)
109		[ $nr == $expected ] && break;
110		[ $nr -gt $max ] && max=$nr
111		i=$((i + 1))
112		sleep 1
113	done
114
115	mptcp_lib_print_title "$msg"
116	if [ $i -ge $timeout ]; then
117		mptcp_lib_pr_fail "timeout while expecting $expected max $max last $nr"
118		mptcp_lib_result_fail "${msg} # timeout"
119		ret=${KSFT_FAIL}
120	elif [ $nr != $expected ]; then
121		mptcp_lib_pr_fail "expected $expected found $nr"
122		mptcp_lib_result_fail "${msg} # unexpected result"
123		ret=${KSFT_FAIL}
124	else
125		mptcp_lib_pr_ok
126		mptcp_lib_result_pass "${msg}"
127	fi
128}
129
130chk_msk_fallback_nr()
131{
132	__chk_msk_nr "grep -c fallback" "$@"
133}
134
135chk_msk_remote_key_nr()
136{
137	__chk_msk_nr "grep -c remote_key" "$@"
138}
139
140__chk_listen()
141{
142	local filter="$1"
143	local expected=$2
144	local msg="$3"
145
146	__chk_nr "ss -N $ns -Ml '$filter' | grep -c LISTEN" "$expected" "$msg" 0
147}
148
149chk_msk_listen()
150{
151	lport=$1
152
153	# destination port search should always return empty list
154	__chk_listen "dport $lport" 0 "listen match for dport $lport"
155
156	# should return 'our' mptcp listen socket
157	__chk_listen "sport $lport" 1 "listen match for sport $lport"
158
159	__chk_listen "src inet:0.0.0.0:$lport" 1 "listen match for saddr and sport"
160
161	__chk_listen "" 1 "all listen sockets"
162
163	nr=$(ss -Ml $filter | wc -l)
164}
165
166chk_msk_inuse()
167{
168	local expected=$1
169	local msg="....chk ${2:-${expected}} msk in use"
170	local listen_nr
171
172	if [ "${expected}" -eq 0 ]; then
173		msg+=" after flush"
174	fi
175
176	listen_nr=$(ss -N "${ns}" -Ml | grep -c LISTEN)
177	expected=$((expected + listen_nr))
178
179	for _ in $(seq 10); do
180		if [ "$(get_msk_inuse)" -eq $expected ]; then
181			break
182		fi
183		sleep 0.1
184	done
185
186	__chk_nr get_msk_inuse $expected "${msg}" 0
187}
188
189# $1: cestab nr
190chk_msk_cestab()
191{
192	local expected=$1
193	local msg="....chk ${2:-${expected}} cestab"
194
195	if [ "${expected}" -eq 0 ]; then
196		msg+=" after flush"
197	fi
198
199	__chk_nr "mptcp_lib_get_counter ${ns} MPTcpExtMPCurrEstab" \
200		 "${expected}" "${msg}" ""
201}
202
203chk_dump_one()
204{
205	local ss_token
206	local token
207	local msg
208
209	ss_token="$(ss -inmHMN $ns |
210		    mptcp_lib_get_info_value "token" "token")"
211
212	token="$(ip netns exec $ns ./mptcp_diag -t $ss_token |\
213		 awk -F':[ \t]+' '/^token/ {print $2}')"
214
215	msg="....chk dump_one"
216
217	mptcp_lib_print_title "$msg"
218	if [ -n "$ss_token" ] && [ "$ss_token" = "$token" ]; then
219		mptcp_lib_pr_ok
220		mptcp_lib_result_pass "${msg}"
221	else
222		mptcp_lib_pr_fail "expected $ss_token found $token"
223		mptcp_lib_result_fail "${msg}"
224		ret=${KSFT_FAIL}
225	fi
226}
227
228msk_info_get_value()
229{
230	local port="${1}"
231	local info="${2}"
232
233	ss -N "${ns}" -inHM dport "${port}" | \
234		mptcp_lib_get_info_value "${info}" "${info}"
235}
236
237chk_msk_info()
238{
239	local port="${1}"
240	local info="${2}"
241	local cnt="${3}"
242	local msg="....chk ${info}"
243	local delta_ms=250  # half what we waited before, just to be sure
244	local now
245
246	now=$(msk_info_get_value "${port}" "${info}")
247
248	mptcp_lib_print_title "${msg}"
249	if { [ -z "${cnt}" ] || [ -z "${now}" ]; } &&
250	   ! mptcp_lib_expect_all_features; then
251		mptcp_lib_pr_skip "Feature probably not supported"
252		mptcp_lib_result_skip "${msg}"
253	elif [ "$((cnt + delta_ms))" -lt "${now}" ]; then
254		mptcp_lib_pr_ok
255		mptcp_lib_result_pass "${msg}"
256	else
257		mptcp_lib_pr_fail "value of ${info} changed by $((now - cnt))ms," \
258				  "expected at least ${delta_ms}ms"
259		mptcp_lib_result_fail "${msg}"
260		ret=${KSFT_FAIL}
261	fi
262}
263
264chk_last_time_info()
265{
266	local port="${1}"
267	local data_sent data_recv ack_recv
268
269	data_sent=$(msk_info_get_value "${port}" "last_data_sent")
270	data_recv=$(msk_info_get_value "${port}" "last_data_recv")
271	ack_recv=$(msk_info_get_value "${port}" "last_ack_recv")
272
273	sleep 0.5  # wait to check after if the timestamps difference
274
275	chk_msk_info "${port}" "last_data_sent" "${data_sent}"
276	chk_msk_info "${port}" "last_data_recv" "${data_recv}"
277	chk_msk_info "${port}" "last_ack_recv" "${ack_recv}"
278}
279
280wait_connected()
281{
282	local listener_ns="${1}"
283	local port="${2}"
284
285	local port_hex i
286
287	port_hex="$(printf "%04X" "${port}")"
288	for i in $(seq 10); do
289		ip netns exec ${listener_ns} grep -q " 0100007F:${port_hex} " /proc/net/tcp && break
290		sleep 0.1
291	done
292}
293
294trap cleanup EXIT
295mptcp_lib_ns_init ns
296
297echo "a" | \
298	timeout ${timeout_test} \
299		ip netns exec $ns \
300			./mptcp_connect -p 10000 -l -t ${timeout_poll} -w 20 \
301				0.0.0.0 >/dev/null &
302mptcp_lib_wait_local_port_listen $ns 10000
303chk_msk_nr 0 "no msk on netns creation"
304chk_msk_listen 10000
305
306echo "b" | \
307	timeout ${timeout_test} \
308		ip netns exec $ns \
309			./mptcp_connect -p 10000 -r 0 -t ${timeout_poll} -w 20 \
310				127.0.0.1 >/dev/null &
311wait_connected $ns 10000
312chk_msk_nr 2 "after MPC handshake"
313chk_last_time_info 10000
314chk_msk_remote_key_nr 2 "....chk remote_key"
315chk_msk_fallback_nr 0 "....chk no fallback"
316chk_msk_inuse 2
317chk_msk_cestab 2
318chk_dump_one
319flush_pids
320
321chk_msk_inuse 0 "2->0"
322chk_msk_cestab 0 "2->0"
323
324echo "a" | \
325	timeout ${timeout_test} \
326		ip netns exec $ns \
327			./mptcp_connect -p 10001 -l -s TCP -t ${timeout_poll} -w 20 \
328				0.0.0.0 >/dev/null &
329mptcp_lib_wait_local_port_listen $ns 10001
330echo "b" | \
331	timeout ${timeout_test} \
332		ip netns exec $ns \
333			./mptcp_connect -p 10001 -r 0 -t ${timeout_poll} -w 20 \
334				127.0.0.1 >/dev/null &
335wait_connected $ns 10001
336chk_msk_fallback_nr 1 "check fallback"
337chk_msk_inuse 1
338chk_msk_cestab 1
339flush_pids
340
341chk_msk_inuse 0 "1->0"
342chk_msk_cestab 0 "1->0"
343
344NR_CLIENTS=100
345for I in $(seq 1 $NR_CLIENTS); do
346	echo "a" | \
347		timeout ${timeout_test} \
348			ip netns exec $ns \
349				./mptcp_connect -p $((I+10001)) -l -w 20 \
350					-t ${timeout_poll} 0.0.0.0 >/dev/null &
351done
352mptcp_lib_wait_local_port_listen $ns $((NR_CLIENTS + 10001))
353
354for I in $(seq 1 $NR_CLIENTS); do
355	echo "b" | \
356		timeout ${timeout_test} \
357			ip netns exec $ns \
358				./mptcp_connect -p $((I+10001)) -w 20 \
359					-t ${timeout_poll} 127.0.0.1 >/dev/null &
360done
361
362wait_msk_nr $((NR_CLIENTS*2)) "many msk socket present"
363chk_msk_inuse $((NR_CLIENTS*2)) "many"
364chk_msk_cestab $((NR_CLIENTS*2)) "many"
365flush_pids
366
367chk_msk_inuse 0 "many->0"
368chk_msk_cestab 0 "many->0"
369
370chk_listener_nr 0 "no listener sockets"
371NR_SERVERS=100
372for I in $(seq 1 $NR_SERVERS); do
373	ip netns exec $ns ./mptcp_connect -p $((I + 20001)) \
374		-t ${timeout_poll} -l 0.0.0.0 >/dev/null 2>&1 &
375done
376mptcp_lib_wait_local_port_listen $ns $((NR_SERVERS + 20001))
377
378chk_listener_nr $NR_SERVERS "many listener sockets"
379
380# graceful termination
381for I in $(seq 1 $NR_SERVERS); do
382	echo a | ip netns exec $ns ./mptcp_connect -p $((I + 20001)) 127.0.0.1 >/dev/null 2>&1 &
383done
384flush_pids
385
386mptcp_lib_result_print_all_tap
387exit $ret
388