bpf/prog_tests/xdp.c

0001 // SPDX-License-Identifier: GPL-2.0
0002 #include <test_progs.h>
0003 #include <network_helpers.h>
0004
0005 void test_xdp(void)
0006 {
0007     struct vip key4 = {.protocol = 6, .family = AF_INET};
0008     struct vip key6 = {.protocol = 6, .family = AF_INET6};
0009     struct iptnl_info value4 = {.family = AF_INET};
0010     struct iptnl_info value6 = {.family = AF_INET6};
0011     const char *file = "./test_xdp.o";
0012     struct bpf_object *obj;
0013     char buf[128];
0014     struct ipv6hdr iph6;
0015     struct iphdr iph;
0016     int err, prog_fd, map_fd;
0017     LIBBPF_OPTS(bpf_test_run_opts, topts,
0018         .data_in = &pkt_v4,
0019         .data_size_in = sizeof(pkt_v4),
0020         .data_out = buf,
0021         .data_size_out = sizeof(buf),
0022         .repeat = 1,
0023     );
0024
0025     err = bpf_prog_test_load(file, BPF_PROG_TYPE_XDP, &obj, &prog_fd);
0026     if (CHECK_FAIL(err))
0027         return;
0028
0029     map_fd = bpf_find_map(__func__, obj, "vip2tnl");
0030     if (map_fd < 0)
0031         goto out;
0032     bpf_map_update_elem(map_fd, &key4, &value4, 0);
0033     bpf_map_update_elem(map_fd, &key6, &value6, 0);
0034
0035     err = bpf_prog_test_run_opts(prog_fd, &topts);
0036     memcpy(&iph, buf + sizeof(struct ethhdr), sizeof(iph));
0037     ASSERT_OK(err, "test_run");
0038     ASSERT_EQ(topts.retval, XDP_TX, "ipv4 test_run retval");
0039     ASSERT_EQ(topts.data_size_out, 74, "ipv4 test_run data_size_out");
0040     ASSERT_EQ(iph.protocol, IPPROTO_IPIP, "ipv4 test_run iph.protocol");
0041
0042     topts.data_in = &pkt_v6;
0043     topts.data_size_in = sizeof(pkt_v6);
0044     topts.data_size_out = sizeof(buf);
0045
0046     err = bpf_prog_test_run_opts(prog_fd, &topts);
0047     memcpy(&iph6, buf + sizeof(struct ethhdr), sizeof(iph6));
0048     ASSERT_OK(err, "test_run");
0049     ASSERT_EQ(topts.retval, XDP_TX, "ipv6 test_run retval");
0050     ASSERT_EQ(topts.data_size_out, 114, "ipv6 test_run data_size_out");
0051     ASSERT_EQ(iph6.nexthdr, IPPROTO_IPV6, "ipv6 test_run iph6.nexthdr");
0052 out:
0053     bpf_object__close(obj);
0054 }