Linux Audio

Check our new training course

Loading...
v6.13.7
 1// SPDX-License-Identifier: GPL-2.0
 2#include <test_progs.h>
 3#include <network_helpers.h>
 4
 5void test_xdp(void)
 6{
 7	struct vip key4 = {.protocol = 6, .family = AF_INET};
 8	struct vip key6 = {.protocol = 6, .family = AF_INET6};
 9	struct iptnl_info value4 = {.family = AF_INET};
10	struct iptnl_info value6 = {.family = AF_INET6};
11	const char *file = "./test_xdp.bpf.o";
12	struct bpf_object *obj;
13	char buf[128];
14	struct ipv6hdr iph6;
15	struct iphdr iph;
16	int err, prog_fd, map_fd;
17	LIBBPF_OPTS(bpf_test_run_opts, topts,
18		.data_in = &pkt_v4,
19		.data_size_in = sizeof(pkt_v4),
20		.data_out = buf,
21		.data_size_out = sizeof(buf),
22		.repeat = 1,
23	);
24
25	err = bpf_prog_test_load(file, BPF_PROG_TYPE_XDP, &obj, &prog_fd);
26	if (CHECK_FAIL(err))
27		return;
28
29	map_fd = bpf_find_map(__func__, obj, "vip2tnl");
30	if (map_fd < 0)
31		goto out;
32	bpf_map_update_elem(map_fd, &key4, &value4, 0);
33	bpf_map_update_elem(map_fd, &key6, &value6, 0);
34
35	err = bpf_prog_test_run_opts(prog_fd, &topts);
36	memcpy(&iph, buf + sizeof(struct ethhdr), sizeof(iph));
37	ASSERT_OK(err, "test_run");
38	ASSERT_EQ(topts.retval, XDP_TX, "ipv4 test_run retval");
39	ASSERT_EQ(topts.data_size_out, 74, "ipv4 test_run data_size_out");
40	ASSERT_EQ(iph.protocol, IPPROTO_IPIP, "ipv4 test_run iph.protocol");
41
42	topts.data_in = &pkt_v6;
43	topts.data_size_in = sizeof(pkt_v6);
44	topts.data_size_out = sizeof(buf);
45
46	err = bpf_prog_test_run_opts(prog_fd, &topts);
47	memcpy(&iph6, buf + sizeof(struct ethhdr), sizeof(iph6));
48	ASSERT_OK(err, "test_run");
49	ASSERT_EQ(topts.retval, XDP_TX, "ipv6 test_run retval");
50	ASSERT_EQ(topts.data_size_out, 114, "ipv6 test_run data_size_out");
51	ASSERT_EQ(iph6.nexthdr, IPPROTO_IPV6, "ipv6 test_run iph6.nexthdr");
52out:
53	bpf_object__close(obj);
54}
v6.2
 1// SPDX-License-Identifier: GPL-2.0
 2#include <test_progs.h>
 3#include <network_helpers.h>
 4
 5void test_xdp(void)
 6{
 7	struct vip key4 = {.protocol = 6, .family = AF_INET};
 8	struct vip key6 = {.protocol = 6, .family = AF_INET6};
 9	struct iptnl_info value4 = {.family = AF_INET};
10	struct iptnl_info value6 = {.family = AF_INET6};
11	const char *file = "./test_xdp.bpf.o";
12	struct bpf_object *obj;
13	char buf[128];
14	struct ipv6hdr iph6;
15	struct iphdr iph;
16	int err, prog_fd, map_fd;
17	LIBBPF_OPTS(bpf_test_run_opts, topts,
18		.data_in = &pkt_v4,
19		.data_size_in = sizeof(pkt_v4),
20		.data_out = buf,
21		.data_size_out = sizeof(buf),
22		.repeat = 1,
23	);
24
25	err = bpf_prog_test_load(file, BPF_PROG_TYPE_XDP, &obj, &prog_fd);
26	if (CHECK_FAIL(err))
27		return;
28
29	map_fd = bpf_find_map(__func__, obj, "vip2tnl");
30	if (map_fd < 0)
31		goto out;
32	bpf_map_update_elem(map_fd, &key4, &value4, 0);
33	bpf_map_update_elem(map_fd, &key6, &value6, 0);
34
35	err = bpf_prog_test_run_opts(prog_fd, &topts);
36	memcpy(&iph, buf + sizeof(struct ethhdr), sizeof(iph));
37	ASSERT_OK(err, "test_run");
38	ASSERT_EQ(topts.retval, XDP_TX, "ipv4 test_run retval");
39	ASSERT_EQ(topts.data_size_out, 74, "ipv4 test_run data_size_out");
40	ASSERT_EQ(iph.protocol, IPPROTO_IPIP, "ipv4 test_run iph.protocol");
41
42	topts.data_in = &pkt_v6;
43	topts.data_size_in = sizeof(pkt_v6);
44	topts.data_size_out = sizeof(buf);
45
46	err = bpf_prog_test_run_opts(prog_fd, &topts);
47	memcpy(&iph6, buf + sizeof(struct ethhdr), sizeof(iph6));
48	ASSERT_OK(err, "test_run");
49	ASSERT_EQ(topts.retval, XDP_TX, "ipv6 test_run retval");
50	ASSERT_EQ(topts.data_size_out, 114, "ipv6 test_run data_size_out");
51	ASSERT_EQ(iph6.nexthdr, IPPROTO_IPV6, "ipv6 test_run iph6.nexthdr");
52out:
53	bpf_object__close(obj);
54}