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
|
// SPDX-License-Identifier: GPL-2.0
#include <test_progs.h>
#include <network_helpers.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <string.h>
#include "test_tc_qevent.skel.h"
#define NS_TX "tc_qevent_tx"
#define NS_RX "tc_qevent_rx"
#define IP_TX "10.255.0.1"
#define IP_RX "10.255.0.2"
#define PIN_PATH "/sys/fs/bpf/tc_qevent_redirect"
static void blast_udp(void)
{
struct sockaddr_in dst = {};
char buf[1400] = {};
int fd, i;
fd = socket(AF_INET, SOCK_DGRAM, 0);
if (!ASSERT_GE(fd, 0, "udp socket"))
return;
dst.sin_family = AF_INET;
dst.sin_port = htons(12345);
inet_pton(AF_INET, IP_RX, &dst.sin_addr);
/*
* Push far more than the RED queue can hold. Once qavg crosses qth_min
* every further packet hits the congestion_drop / early_drop qevent.
*/
for (i = 0; i < 50000; i++)
sendto(fd, buf, sizeof(buf), MSG_DONTWAIT,
(struct sockaddr *)&dst, sizeof(dst));
close(fd);
}
static void run_qevent_redirect(struct bpf_program *prog, __u64 *counter)
{
struct nstoken *tok = NULL;
int err;
SYS_NOFAIL("ip netns del %s", NS_TX);
SYS_NOFAIL("ip netns del %s", NS_RX);
unlink(PIN_PATH);
err = bpf_program__pin(prog, PIN_PATH);
if (!ASSERT_OK(err, "pin prog"))
return;
SYS(unpin, "ip netns add %s", NS_TX);
SYS(del_tx, "ip netns add %s", NS_RX);
SYS(del_rx, "ip -n %s link add veth0 type veth peer name veth1 netns %s", NS_TX, NS_RX);
SYS(del_rx, "ip -n %s addr add %s/24 dev veth0", NS_TX, IP_TX);
SYS(del_rx, "ip -n %s link set veth0 up", NS_TX);
SYS(del_rx, "ip -n %s addr add %s/24 dev veth1", NS_RX, IP_RX);
SYS(del_rx, "ip -n %s link set veth1 up", NS_RX);
tok = open_netns(NS_TX);
if (!ASSERT_OK_PTR(tok, "open_netns"))
goto del_rx;
SYS(close_ns, "tc qdisc add dev veth0 root handle 1: htb default 1");
SYS(close_ns, "tc class add dev veth0 parent 1: classid 1:1 htb rate 1mbit ceil 1mbit");
if (system("tc qdisc add dev veth0 parent 1:1 handle 11: red "
"limit 500000 avpkt 1000 probability 1 min 5000 max 6000 "
"burst 6 qevent early_drop block 10 2>/dev/null")) {
test__skip();
goto close_ns;
}
if (system("tc filter add block 10 bpf da object-pinned "
PIN_PATH " 2>/dev/null")) {
test__skip();
goto close_ns;
}
blast_udp();
ASSERT_GT(*counter, 0, "qevent classifier ran");
close_ns:
close_netns(tok);
del_rx:
SYS_NOFAIL("ip netns del %s", NS_RX);
del_tx:
SYS_NOFAIL("ip netns del %s", NS_TX);
unpin:
bpf_program__unpin(prog, PIN_PATH);
}
void test_tc_qevent(void)
{
struct test_tc_qevent *skel;
skel = test_tc_qevent__open_and_load();
if (!ASSERT_OK_PTR(skel, "open_and_load"))
return;
if (test__start_subtest("redirect_verdict"))
run_qevent_redirect(skel->progs.qevent_redirect_verdict,
&skel->bss->verdict_calls);
if (test__start_subtest("redirect_helper"))
run_qevent_redirect(skel->progs.qevent_redirect_helper,
&skel->bss->helper_calls);
test_tc_qevent__destroy(skel);
}
|
