xref: /linux/tools/testing/selftests/net/lib.sh (revision 61f96e684edd28ca40555ec49ea1555df31ba619) 
1#!/bin/bash
2# SPDX-License-Identifier: GPL-2.0
3
4net_dir=$(dirname "$(readlink -e "${BASH_SOURCE[0]}")")
5source "$net_dir/lib/sh/defer.sh"
6
7##############################################################################
8# Defines
9
10: "${WAIT_TIMEOUT:=20}"
11
12# Whether to pause on after a failure.
13: "${PAUSE_ON_FAIL:=no}"
14
15BUSYWAIT_TIMEOUT=$((WAIT_TIMEOUT * 1000)) # ms
16
17# Kselftest framework constants.
18ksft_pass=0
19ksft_fail=1
20ksft_xfail=2
21ksft_skip=4
22
23# namespace list created by setup_ns
24NS_LIST=()
25
26# Exit status to return at the end. Set in case one of the tests fails.
27EXIT_STATUS=0
28# Per-test return value. Clear at the beginning of each test.
29RET=0
30
31##############################################################################
32# Helpers
33
34__ksft_status_merge()
35{
36	local a=$1; shift
37	local b=$1; shift
38	local -A weights
39	local weight=0
40
41	local i
42	for i in "$@"; do
43		weights[$i]=$((weight++))
44	done
45
46	if [[ ${weights[$a]} > ${weights[$b]} ]]; then
47		echo "$a"
48		return 0
49	else
50		echo "$b"
51		return 1
52	fi
53}
54
55ksft_status_merge()
56{
57	local a=$1; shift
58	local b=$1; shift
59
60	__ksft_status_merge "$a" "$b" \
61		$ksft_pass $ksft_xfail $ksft_skip $ksft_fail
62}
63
64ksft_exit_status_merge()
65{
66	local a=$1; shift
67	local b=$1; shift
68
69	__ksft_status_merge "$a" "$b" \
70		$ksft_xfail $ksft_pass $ksft_skip $ksft_fail
71}
72
73loopy_wait()
74{
75	local sleep_cmd=$1; shift
76	local timeout_ms=$1; shift
77
78	local start_time="$(date -u +%s%3N)"
79	while true
80	do
81		local out
82		if out=$("$@"); then
83			echo -n "$out"
84			return 0
85		fi
86
87		local current_time="$(date -u +%s%3N)"
88		if ((current_time - start_time > timeout_ms)); then
89			echo -n "$out"
90			return 1
91		fi
92
93		$sleep_cmd
94	done
95}
96
97busywait()
98{
99	local timeout_ms=$1; shift
100
101	loopy_wait : "$timeout_ms" "$@"
102}
103
104# timeout in seconds
105slowwait()
106{
107	local timeout_sec=$1; shift
108
109	loopy_wait "sleep 0.1" "$((timeout_sec * 1000))" "$@"
110}
111
112until_counter_is()
113{
114	local expr=$1; shift
115	local current=$("$@")
116
117	echo $((current))
118	((current $expr))
119}
120
121busywait_for_counter()
122{
123	local timeout=$1; shift
124	local delta=$1; shift
125
126	local base=$("$@")
127	busywait "$timeout" until_counter_is ">= $((base + delta))" "$@"
128}
129
130slowwait_for_counter()
131{
132	local timeout=$1; shift
133	local delta=$1; shift
134
135	local base=$("$@")
136	slowwait "$timeout" until_counter_is ">= $((base + delta))" "$@"
137}
138
139# Check for existence of tools which are built as part of selftests
140# but may also already exist in $PATH
141check_gen_prog()
142{
143	local prog_name=$1; shift
144
145	if ! which $prog_name >/dev/null 2>/dev/null; then
146		PATH=$PWD:$PATH
147		if ! which $prog_name >/dev/null; then
148			echo "'$prog_name' command not found; skipping tests"
149			exit $ksft_skip
150		fi
151	fi
152}
153
154remove_ns_list()
155{
156	local item=$1
157	local ns
158	local ns_list=("${NS_LIST[@]}")
159	NS_LIST=()
160
161	for ns in "${ns_list[@]}"; do
162		if [ "${ns}" != "${item}" ]; then
163			NS_LIST+=("${ns}")
164		fi
165	done
166}
167
168cleanup_ns()
169{
170	local ns=""
171	local ret=0
172
173	for ns in "$@"; do
174		[ -z "${ns}" ] && continue
175		ip netns pids "${ns}" 2> /dev/null | xargs -r kill || true
176		ip netns delete "${ns}" &> /dev/null || true
177		if ! busywait $BUSYWAIT_TIMEOUT ip netns list \| grep -vq "^$ns$" &> /dev/null; then
178			echo "Warn: Failed to remove namespace $ns"
179			ret=1
180		else
181			remove_ns_list "${ns}"
182		fi
183	done
184
185	return $ret
186}
187
188cleanup_all_ns()
189{
190	cleanup_ns "${NS_LIST[@]}"
191}
192
193# setup netns with given names as prefix. e.g
194# setup_ns local remote
195setup_ns()
196{
197	local ns_name=""
198	local ns_list=()
199	for ns_name in "$@"; do
200		# avoid conflicts with local var: internal error
201		if [ "${ns_name}" = "ns_name" ]; then
202			echo "Failed to setup namespace '${ns_name}': invalid name"
203			cleanup_ns "${ns_list[@]}"
204			exit $ksft_fail
205		fi
206
207		# Some test may setup/remove same netns multi times
208		if [ -z "${!ns_name}" ]; then
209			eval "${ns_name}=${ns_name,,}-$(mktemp -u XXXXXX)"
210		else
211			cleanup_ns "${!ns_name}"
212		fi
213
214		if ! ip netns add "${!ns_name}"; then
215			echo "Failed to create namespace $ns_name"
216			cleanup_ns "${ns_list[@]}"
217			return $ksft_skip
218		fi
219		ip -n "${!ns_name}" link set lo up
220		ns_list+=("${!ns_name}")
221	done
222	NS_LIST+=("${ns_list[@]}")
223}
224
225# Create netdevsim with given id and net namespace.
226create_netdevsim() {
227    local id="$1"
228    local ns="$2"
229
230    modprobe netdevsim &> /dev/null
231    udevadm settle
232
233    echo "$id 1" | ip netns exec $ns tee /sys/bus/netdevsim/new_device >/dev/null
234    local dev=$(ip netns exec $ns ls /sys/bus/netdevsim/devices/netdevsim$id/net)
235    ip -netns $ns link set dev $dev name nsim$id
236    ip -netns $ns link set dev nsim$id up
237
238    echo nsim$id
239}
240
241# Remove netdevsim with given id.
242cleanup_netdevsim() {
243    local id="$1"
244
245    if [ -d "/sys/bus/netdevsim/devices/netdevsim$id/net" ]; then
246        echo "$id" > /sys/bus/netdevsim/del_device
247    fi
248}
249
250tc_rule_stats_get()
251{
252	local dev=$1; shift
253	local pref=$1; shift
254	local dir=${1:-ingress}; shift
255	local selector=${1:-.packets}; shift
256
257	tc -j -s filter show dev $dev $dir pref $pref \
258	    | jq ".[1].options.actions[].stats$selector"
259}
260
261tc_rule_handle_stats_get()
262{
263	local id=$1; shift
264	local handle=$1; shift
265	local selector=${1:-.packets}; shift
266	local netns=${1:-""}; shift
267
268	tc $netns -j -s filter show $id \
269	    | jq ".[] | select(.options.handle == $handle) | \
270		  .options.actions[0].stats$selector"
271}
272
273ret_set_ksft_status()
274{
275	local ksft_status=$1; shift
276	local msg=$1; shift
277
278	RET=$(ksft_status_merge $RET $ksft_status)
279	if (( $? )); then
280		retmsg=$msg
281	fi
282}
283
284log_test_result()
285{
286	local test_name=$1; shift
287	local opt_str=$1; shift
288	local result=$1; shift
289	local retmsg=$1; shift
290
291	printf "TEST: %-60s  [%s]\n" "$test_name $opt_str" "$result"
292	if [[ $retmsg ]]; then
293		printf "\t%s\n" "$retmsg"
294	fi
295}
296
297pause_on_fail()
298{
299	if [[ $PAUSE_ON_FAIL == yes ]]; then
300		echo "Hit enter to continue, 'q' to quit"
301		read a
302		[[ $a == q ]] && exit 1
303	fi
304}
305
306handle_test_result_pass()
307{
308	local test_name=$1; shift
309	local opt_str=$1; shift
310
311	log_test_result "$test_name" "$opt_str" " OK "
312}
313
314handle_test_result_fail()
315{
316	local test_name=$1; shift
317	local opt_str=$1; shift
318
319	log_test_result "$test_name" "$opt_str" FAIL "$retmsg"
320	pause_on_fail
321}
322
323handle_test_result_xfail()
324{
325	local test_name=$1; shift
326	local opt_str=$1; shift
327
328	log_test_result "$test_name" "$opt_str" XFAIL "$retmsg"
329	pause_on_fail
330}
331
332handle_test_result_skip()
333{
334	local test_name=$1; shift
335	local opt_str=$1; shift
336
337	log_test_result "$test_name" "$opt_str" SKIP "$retmsg"
338}
339
340log_test()
341{
342	local test_name=$1
343	local opt_str=$2
344
345	if [[ $# -eq 2 ]]; then
346		opt_str="($opt_str)"
347	fi
348
349	if ((RET == ksft_pass)); then
350		handle_test_result_pass "$test_name" "$opt_str"
351	elif ((RET == ksft_xfail)); then
352		handle_test_result_xfail "$test_name" "$opt_str"
353	elif ((RET == ksft_skip)); then
354		handle_test_result_skip "$test_name" "$opt_str"
355	else
356		handle_test_result_fail "$test_name" "$opt_str"
357	fi
358
359	EXIT_STATUS=$(ksft_exit_status_merge $EXIT_STATUS $RET)
360	return $RET
361}
362
363log_test_skip()
364{
365	RET=$ksft_skip retmsg= log_test "$@"
366}
367
368log_test_xfail()
369{
370	RET=$ksft_xfail retmsg= log_test "$@"
371}
372
373log_info()
374{
375	local msg=$1
376
377	echo "INFO: $msg"
378}
379
380tests_run()
381{
382	local current_test
383
384	for current_test in ${TESTS:-$ALL_TESTS}; do
385		in_defer_scope \
386			$current_test
387	done
388}
389
390# Whether FAILs should be interpreted as XFAILs. Internal.
391FAIL_TO_XFAIL=
392
393check_err()
394{
395	local err=$1
396	local msg=$2
397
398	if ((err)); then
399		if [[ $FAIL_TO_XFAIL = yes ]]; then
400			ret_set_ksft_status $ksft_xfail "$msg"
401		else
402			ret_set_ksft_status $ksft_fail "$msg"
403		fi
404	fi
405}
406
407check_fail()
408{
409	local err=$1
410	local msg=$2
411
412	check_err $((!err)) "$msg"
413}
414
415check_err_fail()
416{
417	local should_fail=$1; shift
418	local err=$1; shift
419	local what=$1; shift
420
421	if ((should_fail)); then
422		check_fail $err "$what succeeded, but should have failed"
423	else
424		check_err $err "$what failed"
425	fi
426}
427
428xfail()
429{
430	FAIL_TO_XFAIL=yes "$@"
431}
432
433xfail_on_slow()
434{
435	if [[ $KSFT_MACHINE_SLOW = yes ]]; then
436		FAIL_TO_XFAIL=yes "$@"
437	else
438		"$@"
439	fi
440}
441
442omit_on_slow()
443{
444	if [[ $KSFT_MACHINE_SLOW != yes ]]; then
445		"$@"
446	fi
447}
448
449xfail_on_veth()
450{
451	local dev=$1; shift
452	local kind
453
454	kind=$(ip -j -d link show dev $dev |
455			jq -r '.[].linkinfo.info_kind')
456	if [[ $kind = veth ]]; then
457		FAIL_TO_XFAIL=yes "$@"
458	else
459		"$@"
460	fi
461}
462
463mac_get()
464{
465	local if_name=$1
466
467	ip -j link show dev $if_name | jq -r '.[]["address"]'
468}
469
470kill_process()
471{
472	local pid=$1; shift
473
474	# Suppress noise from killing the process.
475	{ kill $pid && wait $pid; } 2>/dev/null
476}
477
478check_command()
479{
480	local cmd=$1; shift
481
482	if [[ ! -x "$(command -v "$cmd")" ]]; then
483		log_test_skip "$cmd not installed"
484		return $EXIT_STATUS
485	fi
486}
487
488require_command()
489{
490	local cmd=$1; shift
491
492	if ! check_command "$cmd"; then
493		exit $EXIT_STATUS
494	fi
495}
496
497ip_link_add()
498{
499	local name=$1; shift
500
501	ip link add name "$name" "$@"
502	defer ip link del dev "$name"
503}
504
505ip_link_set_master()
506{
507	local member=$1; shift
508	local master=$1; shift
509
510	ip link set dev "$member" master "$master"
511	defer ip link set dev "$member" nomaster
512}
513
514ip_link_set_addr()
515{
516	local name=$1; shift
517	local addr=$1; shift
518
519	local old_addr=$(mac_get "$name")
520	ip link set dev "$name" address "$addr"
521	defer ip link set dev "$name" address "$old_addr"
522}
523
524ip_link_is_up()
525{
526	local name=$1; shift
527
528	local state=$(ip -j link show "$name" |
529		      jq -r '(.[].flags[] | select(. == "UP")) // "DOWN"')
530	[[ $state == "UP" ]]
531}
532
533ip_link_set_up()
534{
535	local name=$1; shift
536
537	if ! ip_link_is_up "$name"; then
538		ip link set dev "$name" up
539		defer ip link set dev "$name" down
540	fi
541}
542
543ip_link_set_down()
544{
545	local name=$1; shift
546
547	if ip_link_is_up "$name"; then
548		ip link set dev "$name" down
549		defer ip link set dev "$name" up
550	fi
551}
552
553ip_addr_add()
554{
555	local name=$1; shift
556
557	ip addr add dev "$name" "$@"
558	defer ip addr del dev "$name" "$@"
559}
560
561ip_route_add()
562{
563	ip route add "$@"
564	defer ip route del "$@"
565}
566
567bridge_vlan_add()
568{
569	bridge vlan add "$@"
570	defer bridge vlan del "$@"
571}
572