1#!/bin/bash
  2# SPDX-License-Identifier: GPL-2.0
  3
  4# This test is for checking drop monitor functionality.
  5source lib.sh
  6ret=0
  7
  8# all tests in this script. Can be overridden with -t option
  9TESTS="
 10	sw_drops
 11	hw_drops
 12"
 13
 14NETDEVSIM_PATH=/sys/bus/netdevsim/
 15DEV_ADDR=1337
 16DEV=netdevsim${DEV_ADDR}
 17DEVLINK_DEV=netdevsim/${DEV}
 18
 19log_test()
 20{
 21	local rc=$1
 22	local expected=$2
 23	local msg="$3"
 24
 25	if [ ${rc} -eq ${expected} ]; then
 26		printf "    TEST: %-60s  [ OK ]\n" "${msg}"
 27		nsuccess=$((nsuccess+1))
 28	else
 29		ret=1
 30		nfail=$((nfail+1))
 31		printf "    TEST: %-60s  [FAIL]\n" "${msg}"
 32	fi
 33}
 34
 35setup()
 36{
 37	modprobe netdevsim &> /dev/null
 38
 39	set -e
 40	setup_ns NS1
 41	$IP link add dummy10 up type dummy
 42
 43	$NS_EXEC echo "$DEV_ADDR 1" > ${NETDEVSIM_PATH}/new_device
 44	udevadm settle
 45	local netdev=$($NS_EXEC ls ${NETDEVSIM_PATH}/devices/${DEV}/net/)
 46	$IP link set dev $netdev up
 47
 48	set +e
 49}
 50
 51cleanup()
 52{
 53	$NS_EXEC echo "$DEV_ADDR" > ${NETDEVSIM_PATH}/del_device
 54	cleanup_ns ${NS1}
 55}
 56
 57sw_drops_test()
 58{
 59	echo
 60	echo "Software drops test"
 61
 62	setup
 63
 64	local dir=$(mktemp -d)
 65
 66	$TC qdisc add dev dummy10 clsact
 67	$TC filter add dev dummy10 egress pref 1 handle 101 proto ip \
 68		flower dst_ip 192.0.2.10 action drop
 69
 70	$NS_EXEC mausezahn dummy10 -a 00:11:22:33:44:55 -b 00:aa:bb:cc:dd:ee \
 71		-A 192.0.2.1 -B 192.0.2.10 -t udp sp=12345,dp=54321 -c 0 -q \
 72		-d 100msec &
 73	timeout 5 dwdump -o sw -w ${dir}/packets.pcap
 74	(( $(tshark -r ${dir}/packets.pcap \
 75		-Y 'ip.dst == 192.0.2.10' 2> /dev/null | wc -l) != 0))
 76	log_test $? 0 "Capturing active software drops"
 77
 78	rm ${dir}/packets.pcap
 79
 80	kill_process %%
 81	timeout 5 dwdump -o sw -w ${dir}/packets.pcap
 82	(( $(tshark -r ${dir}/packets.pcap \
 83		-Y 'ip.dst == 192.0.2.10' 2> /dev/null | wc -l) == 0))
 84	log_test $? 0 "Capturing inactive software drops"
 85
 86	rm -r $dir
 87
 88	cleanup
 89}
 90
 91hw_drops_test()
 92{
 93	echo
 94	echo "Hardware drops test"
 95
 96	setup
 97
 98	local dir=$(mktemp -d)
 99
100	$DEVLINK trap set $DEVLINK_DEV trap blackhole_route action trap
101	timeout 5 dwdump -o hw -w ${dir}/packets.pcap
102	(( $(tshark -r ${dir}/packets.pcap \
103		-Y 'net_dm.hw_trap_name== blackhole_route' 2> /dev/null \
104		| wc -l) != 0))
105	log_test $? 0 "Capturing active hardware drops"
106
107	rm ${dir}/packets.pcap
108
109	$DEVLINK trap set $DEVLINK_DEV trap blackhole_route action drop
110	timeout 5 dwdump -o hw -w ${dir}/packets.pcap
111	(( $(tshark -r ${dir}/packets.pcap \
112		-Y 'net_dm.hw_trap_name== blackhole_route' 2> /dev/null \
113		| wc -l) == 0))
114	log_test $? 0 "Capturing inactive hardware drops"
115
116	rm -r $dir
117
118	cleanup
119}
120
121################################################################################
122# usage
123
124usage()
125{
126	cat <<EOF
127usage: ${0##*/} OPTS
128
129        -t <test>   Test(s) to run (default: all)
130                    (options: $TESTS)
131EOF
132}
133
134################################################################################
135# main
136
137while getopts ":t:h" opt; do
138	case $opt in
139		t) TESTS=$OPTARG;;
140		h) usage; exit 0;;
141		*) usage; exit 1;;
142	esac
143done
144
145if [ "$(id -u)" -ne 0 ];then
146	echo "SKIP: Need root privileges"
147	exit $ksft_skip;
148fi
149
150if [ ! -x "$(command -v ip)" ]; then
151	echo "SKIP: Could not run test without ip tool"
152	exit $ksft_skip
153fi
154
155if [ ! -x "$(command -v devlink)" ]; then
156	echo "SKIP: Could not run test without devlink tool"
157	exit $ksft_skip
158fi
159
160if [ ! -x "$(command -v tshark)" ]; then
161	echo "SKIP: Could not run test without tshark tool"
162	exit $ksft_skip
163fi
164
165if [ ! -x "$(command -v dwdump)" ]; then
166	echo "SKIP: Could not run test without dwdump tool"
167	exit $ksft_skip
168fi
169
170if [ ! -x "$(command -v udevadm)" ]; then
171	echo "SKIP: Could not run test without udevadm tool"
172	exit $ksft_skip
173fi
174
175if [ ! -x "$(command -v timeout)" ]; then
176	echo "SKIP: Could not run test without timeout tool"
177	exit $ksft_skip
178fi
179
180if [ ! -x "$(command -v mausezahn)" ]; then
181	echo "SKIP: Could not run test without mausezahn tool"
182	exit $ksft_skip
183fi
184
185tshark -G fields 2> /dev/null | grep -q net_dm
186if [ $? -ne 0 ]; then
187	echo "SKIP: tshark too old, missing net_dm dissector"
188	exit $ksft_skip
189fi
190
191# create netns first so we can get the namespace name
192setup_ns NS1
193cleanup &> /dev/null
194trap cleanup EXIT
195
196IP="ip -netns ${NS1}"
197TC="tc -netns ${NS1}"
198DEVLINK="devlink -N ${NS1}"
199NS_EXEC="ip netns exec ${NS1}"
200
201for t in $TESTS
202do
203	case $t in
204	sw_drops|sw)			sw_drops_test;;
205	hw_drops|hw)			hw_drops_test;;
206
207	help) echo "Test names: $TESTS"; exit 0;;
208	esac
209done
210
211if [ "$TESTS" != "none" ]; then
212	printf "\nTests passed: %3d\n" ${nsuccess}
213	printf "Tests failed: %3d\n"   ${nfail}
214fi
215
216exit $ret