summaryrefslogtreecommitdiff
path: root/tools/testing/selftests/drivers/net/netdevsim/psample.sh
blob: ee10b1a8933c44c9deabdf5008ad415e86419d0e (plain)
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
#!/bin/bash
# SPDX-License-Identifier: GPL-2.0
#
# This test is for checking the psample module. It makes use of netdevsim
# which periodically generates "sampled" packets.

lib_dir=$(dirname $0)/../../../net/forwarding

ALL_TESTS="
	psample_enable_test
	psample_group_num_test
	psample_md_test
"
NETDEVSIM_PATH=/sys/bus/netdevsim/
DEV_ADDR=1337
DEV=netdevsim${DEV_ADDR}
DEVLINK_DEV=netdevsim/${DEV}
SYSFS_NET_DIR=/sys/bus/netdevsim/devices/$DEV/net/
PSAMPLE_DIR=/sys/kernel/debug/netdevsim/$DEV/psample/
CAPTURE_FILE=$(mktemp)
NUM_NETIFS=0
source $lib_dir/lib.sh
source $lib_dir/devlink_lib.sh

# Available at https://github.com/Mellanox/libpsample
require_command psample

psample_capture()
{
	rm -f $CAPTURE_FILE

	timeout 2 ip netns exec testns1 psample &> $CAPTURE_FILE
}

psample_enable_test()
{
	RET=0

	echo 1 > $PSAMPLE_DIR/enable
	check_err $? "Failed to enable sampling when should not"

	echo 1 > $PSAMPLE_DIR/enable 2>/dev/null
	check_fail $? "Sampling enablement succeeded when should fail"

	psample_capture
	if [ $(cat $CAPTURE_FILE | wc -l) -eq 0 ]; then
		check_err 1 "Failed to capture sampled packets"
	fi

	echo 0 > $PSAMPLE_DIR/enable
	check_err $? "Failed to disable sampling when should not"

	echo 0 > $PSAMPLE_DIR/enable 2>/dev/null
	check_fail $? "Sampling disablement succeeded when should fail"

	psample_capture
	if [ $(cat $CAPTURE_FILE | wc -l) -ne 0 ]; then
		check_err 1 "Captured sampled packets when should not"
	fi

	log_test "psample enable / disable"
}

psample_group_num_test()
{
	RET=0

	echo 1234 > $PSAMPLE_DIR/group_num
	echo 1 > $PSAMPLE_DIR/enable

	psample_capture
	grep -q -e "group 1234" $CAPTURE_FILE
	check_err $? "Sampled packets reported with wrong group number"

	# New group number should only be used after disable / enable.
	echo 4321 > $PSAMPLE_DIR/group_num

	psample_capture
	grep -q -e "group 4321" $CAPTURE_FILE
	check_fail $? "Group number changed while sampling is active"

	echo 0 > $PSAMPLE_DIR/enable && echo 1 > $PSAMPLE_DIR/enable

	psample_capture
	grep -q -e "group 4321" $CAPTURE_FILE
	check_err $? "Group number did not change after restarting sampling"

	log_test "psample group number"

	echo 0 > $PSAMPLE_DIR/enable
}

psample_md_test()
{
	RET=0

	echo 1 > $PSAMPLE_DIR/enable

	echo 1234 > $PSAMPLE_DIR/in_ifindex
	echo 4321 > $PSAMPLE_DIR/out_ifindex
	psample_capture

	grep -q -e "in-ifindex 1234" $CAPTURE_FILE
	check_err $? "Sampled packets reported with wrong in-ifindex"

	grep -q -e "out-ifindex 4321" $CAPTURE_FILE
	check_err $? "Sampled packets reported with wrong out-ifindex"

	echo 5 > $PSAMPLE_DIR/out_tc
	psample_capture

	grep -q -e "out-tc 5" $CAPTURE_FILE
	check_err $? "Sampled packets reported with wrong out-tc"

	echo $((2**16 - 1)) > $PSAMPLE_DIR/out_tc
	psample_capture

	grep -q -e "out-tc " $CAPTURE_FILE
	check_fail $? "Sampled packets reported with out-tc when should not"

	echo 1 > $PSAMPLE_DIR/out_tc
	echo 10000 > $PSAMPLE_DIR/out_tc_occ_max
	psample_capture

	grep -q -e "out-tc-occ " $CAPTURE_FILE
	check_err $? "Sampled packets not reported with out-tc-occ when should"

	echo 0 > $PSAMPLE_DIR/out_tc_occ_max
	psample_capture

	grep -q -e "out-tc-occ " $CAPTURE_FILE
	check_fail $? "Sampled packets reported with out-tc-occ when should not"

	echo 10000 > $PSAMPLE_DIR/latency_max
	psample_capture

	grep -q -e "latency " $CAPTURE_FILE
	check_err $? "Sampled packets not reported with latency when should"

	echo 0 > $PSAMPLE_DIR/latency_max
	psample_capture

	grep -q -e "latency " $CAPTURE_FILE
	check_fail $? "Sampled packets reported with latency when should not"

	log_test "psample metadata"

	echo 0 > $PSAMPLE_DIR/enable
}

setup_prepare()
{
	modprobe netdevsim &> /dev/null

	echo "$DEV_ADDR 1" > ${NETDEVSIM_PATH}/new_device
	while [ ! -d $SYSFS_NET_DIR ] ; do :; done

	set -e

	ip netns add testns1
	devlink dev reload $DEVLINK_DEV netns testns1

	set +e
}

cleanup()
{
	pre_cleanup
	rm -f $CAPTURE_FILE
	ip netns del testns1
	echo "$DEV_ADDR" > ${NETDEVSIM_PATH}/del_device
	modprobe -r netdevsim &> /dev/null
}

trap cleanup EXIT

setup_prepare

tests_run

exit $EXIT_STATUS