bpf_qdisc.c

  1// SPDX-License-Identifier: GPL-2.0
  2
  3#include <linux/pkt_sched.h>
  4#include <linux/rtnetlink.h>
  5#include <test_progs.h>
  6
  7#include "network_helpers.h"
  8#include "bpf_qdisc_fifo.skel.h"
  9#include "bpf_qdisc_fq.skel.h"
 10#include "bpf_qdisc_fail__incompl_ops.skel.h"
 11
 12#define LO_IFINDEX 1
 13
 14static const unsigned int total_bytes = 10 * 1024 * 1024;
 15
 16static void do_test(char *qdisc)
 17{
 18	DECLARE_LIBBPF_OPTS(bpf_tc_hook, hook, .ifindex = LO_IFINDEX,
 19			    .attach_point = BPF_TC_QDISC,
 20			    .parent = TC_H_ROOT,
 21			    .handle = 0x8000000,
 22			    .qdisc = qdisc);
 23	int srv_fd = -1, cli_fd = -1;
 24	int err;
 25
 26	err = bpf_tc_hook_create(&hook);
 27	if (!ASSERT_OK(err, "attach qdisc"))
 28		return;
 29
 30	srv_fd = start_server(AF_INET6, SOCK_STREAM, NULL, 0, 0);
 31	if (!ASSERT_OK_FD(srv_fd, "start server"))
 32		goto done;
 33
 34	cli_fd = connect_to_fd(srv_fd, 0);
 35	if (!ASSERT_OK_FD(cli_fd, "connect to client"))
 36		goto done;
 37
 38	err = send_recv_data(srv_fd, cli_fd, total_bytes);
 39	ASSERT_OK(err, "send_recv_data");
 40
 41done:
 42	if (srv_fd != -1)
 43		close(srv_fd);
 44	if (cli_fd != -1)
 45		close(cli_fd);
 46
 47	bpf_tc_hook_destroy(&hook);
 48}
 49
 50static void test_fifo(void)
 51{
 52	struct bpf_qdisc_fifo *fifo_skel;
 53
 54	fifo_skel = bpf_qdisc_fifo__open_and_load();
 55	if (!ASSERT_OK_PTR(fifo_skel, "bpf_qdisc_fifo__open_and_load"))
 56		return;
 57
 58	if (!ASSERT_OK(bpf_qdisc_fifo__attach(fifo_skel), "bpf_qdisc_fifo__attach"))
 59		goto out;
 60
 61	do_test("bpf_fifo");
 62out:
 63	bpf_qdisc_fifo__destroy(fifo_skel);
 64}
 65
 66static void test_fq(void)
 67{
 68	struct bpf_qdisc_fq *fq_skel;
 69
 70	fq_skel = bpf_qdisc_fq__open_and_load();
 71	if (!ASSERT_OK_PTR(fq_skel, "bpf_qdisc_fq__open_and_load"))
 72		return;
 73
 74	if (!ASSERT_OK(bpf_qdisc_fq__attach(fq_skel), "bpf_qdisc_fq__attach"))
 75		goto out;
 76
 77	do_test("bpf_fq");
 78out:
 79	bpf_qdisc_fq__destroy(fq_skel);
 80}
 81
 82static void test_qdisc_attach_to_mq(void)
 83{
 84	DECLARE_LIBBPF_OPTS(bpf_tc_hook, hook,
 85			    .attach_point = BPF_TC_QDISC,
 86			    .parent = TC_H_MAKE(1 << 16, 1),
 87			    .handle = 0x11 << 16,
 88			    .qdisc = "bpf_fifo");
 89	struct bpf_qdisc_fifo *fifo_skel;
 90	int err;
 91
 92	fifo_skel = bpf_qdisc_fifo__open_and_load();
 93	if (!ASSERT_OK_PTR(fifo_skel, "bpf_qdisc_fifo__open_and_load"))
 94		return;
 95
 96	if (!ASSERT_OK(bpf_qdisc_fifo__attach(fifo_skel), "bpf_qdisc_fifo__attach"))
 97		goto out;
 98
 99	SYS(out, "ip link add veth0 type veth peer veth1");
100	hook.ifindex = if_nametoindex("veth0");
101	SYS(out, "tc qdisc add dev veth0 root handle 1: mq");
102
103	err = bpf_tc_hook_create(&hook);
104	ASSERT_OK(err, "attach qdisc");
105
106	bpf_tc_hook_destroy(&hook);
107
108	SYS(out, "tc qdisc delete dev veth0 root mq");
109out:
110	bpf_qdisc_fifo__destroy(fifo_skel);
111}
112
113static void test_qdisc_attach_to_non_root(void)
114{
115	DECLARE_LIBBPF_OPTS(bpf_tc_hook, hook, .ifindex = LO_IFINDEX,
116			    .attach_point = BPF_TC_QDISC,
117			    .parent = TC_H_MAKE(1 << 16, 1),
118			    .handle = 0x11 << 16,
119			    .qdisc = "bpf_fifo");
120	struct bpf_qdisc_fifo *fifo_skel;
121	int err;
122
123	fifo_skel = bpf_qdisc_fifo__open_and_load();
124	if (!ASSERT_OK_PTR(fifo_skel, "bpf_qdisc_fifo__open_and_load"))
125		return;
126
127	if (!ASSERT_OK(bpf_qdisc_fifo__attach(fifo_skel), "bpf_qdisc_fifo__attach"))
128		goto out;
129
130	SYS(out, "tc qdisc add dev lo root handle 1: htb");
131	SYS(out_del_htb, "tc class add dev lo parent 1: classid 1:1 htb rate 75Kbit");
132
133	err = bpf_tc_hook_create(&hook);
134	if (!ASSERT_ERR(err, "attach qdisc"))
135		bpf_tc_hook_destroy(&hook);
136
137out_del_htb:
138	SYS(out, "tc qdisc delete dev lo root htb");
139out:
140	bpf_qdisc_fifo__destroy(fifo_skel);
141}
142
143static void test_incompl_ops(void)
144{
145	struct bpf_qdisc_fail__incompl_ops *skel;
146	struct bpf_link *link;
147
148	skel = bpf_qdisc_fail__incompl_ops__open_and_load();
149	if (!ASSERT_OK_PTR(skel, "bpf_qdisc_fifo__open_and_load"))
150		return;
151
152	link = bpf_map__attach_struct_ops(skel->maps.test);
153	if (!ASSERT_ERR_PTR(link, "bpf_map__attach_struct_ops"))
154		bpf_link__destroy(link);
155
156	bpf_qdisc_fail__incompl_ops__destroy(skel);
157}
158
159static int get_default_qdisc(char *qdisc_name)
160{
161	FILE *f;
162	int num;
163
164	f = fopen("/proc/sys/net/core/default_qdisc", "r");
165	if (!f)
166		return -errno;
167
168	num = fscanf(f, "%s", qdisc_name);
169	fclose(f);
170
171	return num == 1 ? 0 : -EFAULT;
172}
173
174static void test_default_qdisc_attach_to_mq(void)
175{
176	char default_qdisc[IFNAMSIZ] = {};
177	struct bpf_qdisc_fifo *fifo_skel;
178	struct netns_obj *netns = NULL;
179	int err;
180
181	fifo_skel = bpf_qdisc_fifo__open_and_load();
182	if (!ASSERT_OK_PTR(fifo_skel, "bpf_qdisc_fifo__open_and_load"))
183		return;
184
185	if (!ASSERT_OK(bpf_qdisc_fifo__attach(fifo_skel), "bpf_qdisc_fifo__attach"))
186		goto out;
187
188	err = get_default_qdisc(default_qdisc);
189	if (!ASSERT_OK(err, "read sysctl net.core.default_qdisc"))
190		goto out;
191
192	err = write_sysctl("/proc/sys/net/core/default_qdisc", "bpf_fifo");
193	if (!ASSERT_OK(err, "write sysctl net.core.default_qdisc"))
194		goto out;
195
196	netns = netns_new("bpf_qdisc_ns", true);
197	if (!ASSERT_OK_PTR(netns, "netns_new"))
198		goto out;
199
200	SYS(out, "ip link add veth0 type veth peer veth1");
201	SYS(out, "tc qdisc add dev veth0 root handle 1: mq");
202
203	ASSERT_EQ(fifo_skel->bss->init_called, true, "init_called");
204
205	SYS(out, "tc qdisc delete dev veth0 root mq");
206out:
207	netns_free(netns);
208	if (default_qdisc[0])
209		write_sysctl("/proc/sys/net/core/default_qdisc", default_qdisc);
210
211	bpf_qdisc_fifo__destroy(fifo_skel);
212}
213
214void test_ns_bpf_qdisc(void)
215{
216	if (test__start_subtest("fifo"))
217		test_fifo();
218	if (test__start_subtest("fq"))
219		test_fq();
220	if (test__start_subtest("attach to mq"))
221		test_qdisc_attach_to_mq();
222	if (test__start_subtest("attach to non root"))
223		test_qdisc_attach_to_non_root();
224	if (test__start_subtest("incompl_ops"))
225		test_incompl_ops();
226}
227
228void serial_test_bpf_qdisc_default(void)
229{
230	test_default_qdisc_attach_to_mq();
231}