xref: /linux/tools/testing/selftests/net/mptcp/diag.sh (revision 85502b2214d50ba0ddf2a5fb454e4d28a160d175) 
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
228chk_dump_subflow()
229{
230	local inet_diag_token
231	local subflow_line
232	local ss_output
233	local ss_token
234	local msg
235
236	ss_output=$(ss -tniN $ns)
237
238	subflow_line=$(echo "$ss_output" | \
239		       grep -m1 -Eo '[0-9.]+:[0-9].+ +[0-9.]+:[0-9.]+')
240
241	ss_token=$(echo "$ss_output" | grep -m1 -Eo 'token:[^ ]+')
242
243	inet_diag_token=$(ip netns exec $ns ./mptcp_diag -s "$subflow_line" | \
244			  grep -Eo 'token:[^ ]+')
245
246	msg="....chk dump_subflow"
247
248	mptcp_lib_print_title "$msg"
249	if [ -n "$ss_token" ] && [ "$ss_token" = "$inet_diag_token" ]; then
250		mptcp_lib_pr_ok
251		mptcp_lib_result_pass "${msg}"
252	else
253		mptcp_lib_pr_fail "expected $ss_token found $inet_diag_token"
254		mptcp_lib_result_fail "${msg}"
255		ret=${KSFT_FAIL}
256	fi
257}
258
259msk_info_get_value()
260{
261	local port="${1}"
262	local info="${2}"
263
264	ss -N "${ns}" -inHM dport "${port}" | \
265		mptcp_lib_get_info_value "${info}" "${info}"
266}
267
268chk_msk_info()
269{
270	local port="${1}"
271	local info="${2}"
272	local cnt="${3}"
273	local msg="....chk ${info}"
274	local delta_ms=250  # half what we waited before, just to be sure
275	local now
276
277	now=$(msk_info_get_value "${port}" "${info}")
278
279	mptcp_lib_print_title "${msg}"
280	if { [ -z "${cnt}" ] || [ -z "${now}" ]; } &&
281	   ! mptcp_lib_expect_all_features; then
282		mptcp_lib_pr_skip "Feature probably not supported"
283		mptcp_lib_result_skip "${msg}"
284	elif [ "$((cnt + delta_ms))" -lt "${now}" ]; then
285		mptcp_lib_pr_ok
286		mptcp_lib_result_pass "${msg}"
287	else
288		mptcp_lib_pr_fail "value of ${info} changed by $((now - cnt))ms," \
289				  "expected at least ${delta_ms}ms"
290		mptcp_lib_result_fail "${msg}"
291		ret=${KSFT_FAIL}
292	fi
293}
294
295chk_last_time_info()
296{
297	local port="${1}"
298	local data_sent data_recv ack_recv
299
300	data_sent=$(msk_info_get_value "${port}" "last_data_sent")
301	data_recv=$(msk_info_get_value "${port}" "last_data_recv")
302	ack_recv=$(msk_info_get_value "${port}" "last_ack_recv")
303
304	sleep 0.5  # wait to check after if the timestamps difference
305
306	chk_msk_info "${port}" "last_data_sent" "${data_sent}"
307	chk_msk_info "${port}" "last_data_recv" "${data_recv}"
308	chk_msk_info "${port}" "last_ack_recv" "${ack_recv}"
309}
310
311wait_connected()
312{
313	local listener_ns="${1}"
314	local port="${2}"
315
316	local port_hex i
317
318	port_hex="$(printf "%04X" "${port}")"
319	for i in $(seq 10); do
320		ip netns exec ${listener_ns} grep -q " 0100007F:${port_hex} " /proc/net/tcp && break
321		sleep 0.1
322	done
323}
324
325trap cleanup EXIT
326mptcp_lib_ns_init ns
327
328echo "a" | \
329	timeout ${timeout_test} \
330		ip netns exec $ns \
331			./mptcp_connect -p 10000 -l -t ${timeout_poll} -w 20 \
332				0.0.0.0 >/dev/null &
333mptcp_lib_wait_local_port_listen $ns 10000
334chk_msk_nr 0 "no msk on netns creation"
335chk_msk_listen 10000
336
337echo "b" | \
338	timeout ${timeout_test} \
339		ip netns exec $ns \
340			./mptcp_connect -p 10000 -r 0 -t ${timeout_poll} -w 20 \
341				127.0.0.1 >/dev/null &
342wait_connected $ns 10000
343chk_msk_nr 2 "after MPC handshake"
344chk_last_time_info 10000
345chk_msk_remote_key_nr 2 "....chk remote_key"
346chk_msk_fallback_nr 0 "....chk no fallback"
347chk_msk_inuse 2
348chk_msk_cestab 2
349chk_dump_one
350chk_dump_subflow
351flush_pids
352
353chk_msk_inuse 0 "2->0"
354chk_msk_cestab 0 "2->0"
355
356echo "a" | \
357	timeout ${timeout_test} \
358		ip netns exec $ns \
359			./mptcp_connect -p 10001 -l -s TCP -t ${timeout_poll} -w 20 \
360				0.0.0.0 >/dev/null &
361mptcp_lib_wait_local_port_listen $ns 10001
362echo "b" | \
363	timeout ${timeout_test} \
364		ip netns exec $ns \
365			./mptcp_connect -p 10001 -r 0 -t ${timeout_poll} -w 20 \
366				127.0.0.1 >/dev/null &
367wait_connected $ns 10001
368chk_msk_fallback_nr 1 "check fallback"
369chk_msk_inuse 1
370chk_msk_cestab 1
371flush_pids
372
373chk_msk_inuse 0 "1->0"
374chk_msk_cestab 0 "1->0"
375
376NR_CLIENTS=100
377for I in $(seq 1 $NR_CLIENTS); do
378	echo "a" | \
379		timeout ${timeout_test} \
380			ip netns exec $ns \
381				./mptcp_connect -p $((I+10001)) -l -w 20 \
382					-t ${timeout_poll} 0.0.0.0 >/dev/null &
383done
384mptcp_lib_wait_local_port_listen $ns $((NR_CLIENTS + 10001))
385
386for I in $(seq 1 $NR_CLIENTS); do
387	echo "b" | \
388		timeout ${timeout_test} \
389			ip netns exec $ns \
390				./mptcp_connect -p $((I+10001)) -w 20 \
391					-t ${timeout_poll} 127.0.0.1 >/dev/null &
392done
393
394wait_msk_nr $((NR_CLIENTS*2)) "many msk socket present"
395chk_msk_inuse $((NR_CLIENTS*2)) "many"
396chk_msk_cestab $((NR_CLIENTS*2)) "many"
397flush_pids
398
399chk_msk_inuse 0 "many->0"
400chk_msk_cestab 0 "many->0"
401
402chk_listener_nr 0 "no listener sockets"
403NR_SERVERS=100
404for I in $(seq 1 $NR_SERVERS); do
405	ip netns exec $ns ./mptcp_connect -p $((I + 20001)) \
406		-t ${timeout_poll} -l 0.0.0.0 >/dev/null 2>&1 &
407done
408mptcp_lib_wait_local_port_listen $ns $((NR_SERVERS + 20001))
409
410chk_listener_nr $NR_SERVERS "many listener sockets"
411
412# graceful termination
413for I in $(seq 1 $NR_SERVERS); do
414	echo a | ip netns exec $ns ./mptcp_connect -p $((I + 20001)) 127.0.0.1 >/dev/null 2>&1 &
415done
416flush_pids
417
418mptcp_lib_result_print_all_tap
419exit $ret
420