55 lines
1.7 KiB
C
55 lines
1.7 KiB
C
|
// SPDX-License-Identifier: GPL-2.0
|
||
|
|
#include <test_progs.h>
|
||
|
|
#include <network_helpers.h>
|
||
|
|
|
||
|
|
void test_xdp(void)
|
||
|
|
{
|
||
|
|
struct vip key4 = {.protocol = 6, .family = AF_INET};
|
||
|
|
struct vip key6 = {.protocol = 6, .family = AF_INET6};
|
||
|
|
struct iptnl_info value4 = {.family = AF_INET};
|
||
|
|
struct iptnl_info value6 = {.family = AF_INET6};
|
||
|
|
const char *file = "./test_xdp.bpf.o";
|
||
|
|
struct bpf_object *obj;
|
||
|
|
char buf[128];
|
||
|
|
struct ipv6hdr iph6;
|
||
|
|
struct iphdr iph;
|
||
|
|
int err, prog_fd, map_fd;
|
||
|
|
LIBBPF_OPTS(bpf_test_run_opts, topts,
|
||
|
|
.data_in = &pkt_v4,
|
||
|
|
.data_size_in = sizeof(pkt_v4),
|
||
|
|
.data_out = buf,
|
||
|
|
.data_size_out = sizeof(buf),
|
||
|
|
.repeat = 1,
|
||
|
|
);
|
||
|
|
|
||
|
|
err = bpf_prog_test_load(file, BPF_PROG_TYPE_XDP, &obj, &prog_fd);
|
||
|
|
if (CHECK_FAIL(err))
|
||
|
|
return;
|
||
|
|
|
||
|
|
map_fd = bpf_find_map(__func__, obj, "vip2tnl");
|
||
|
|
if (map_fd < 0)
|
||
|
|
goto out;
|
||
|
|
bpf_map_update_elem(map_fd, &key4, &value4, 0);
|
||
|
|
bpf_map_update_elem(map_fd, &key6, &value6, 0);
|
||
|
|
|
||
|
|
err = bpf_prog_test_run_opts(prog_fd, &topts);
|
||
|
|
memcpy(&iph, buf + sizeof(struct ethhdr), sizeof(iph));
|
||
|
|
ASSERT_OK(err, "test_run");
|
||
|
|
ASSERT_EQ(topts.retval, XDP_TX, "ipv4 test_run retval");
|
||
|
|
ASSERT_EQ(topts.data_size_out, 74, "ipv4 test_run data_size_out");
|
||
|
|
ASSERT_EQ(iph.protocol, IPPROTO_IPIP, "ipv4 test_run iph.protocol");
|
||
|
|
|
||
|
|
topts.data_in = &pkt_v6;
|
||
|
|
topts.data_size_in = sizeof(pkt_v6);
|
||
|
|
topts.data_size_out = sizeof(buf);
|
||
|
|
|
||
|
|
err = bpf_prog_test_run_opts(prog_fd, &topts);
|
||
|
|
memcpy(&iph6, buf + sizeof(struct ethhdr), sizeof(iph6));
|
||
|
|
ASSERT_OK(err, "test_run");
|
||
|
|
ASSERT_EQ(topts.retval, XDP_TX, "ipv6 test_run retval");
|
||
|
|
ASSERT_EQ(topts.data_size_out, 114, "ipv6 test_run data_size_out");
|
||
|
|
ASSERT_EQ(iph6.nexthdr, IPPROTO_IPV6, "ipv6 test_run iph6.nexthdr");
|
||
|
|
out:
|
||
|
|
bpf_object__close(obj);
|
||
|
|
}
|