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