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