bpf/prog_tests/skb_load_bytes.c

0001 // SPDX-License-Identifier: GPL-2.0
0002 #include <test_progs.h>
0003 #include <network_helpers.h>
0004 #include "skb_load_bytes.skel.h"
0005
0006 void test_skb_load_bytes(void)
0007 {
0008     struct skb_load_bytes *skel;
0009     int err, prog_fd, test_result;
0010     struct __sk_buff skb = { 0 };
0011
0012     LIBBPF_OPTS(bpf_test_run_opts, tattr,
0013         .data_in = &pkt_v4,
0014         .data_size_in = sizeof(pkt_v4),
0015         .ctx_in = &skb,
0016         .ctx_size_in = sizeof(skb),
0017     );
0018
0019     skel = skb_load_bytes__open_and_load();
0020     if (!ASSERT_OK_PTR(skel, "skel_open_and_load"))
0021         return;
0022
0023     prog_fd = bpf_program__fd(skel->progs.skb_process);
0024     if (!ASSERT_GE(prog_fd, 0, "prog_fd"))
0025         goto out;
0026
0027     skel->bss->load_offset = (uint32_t)(-1);
0028     err = bpf_prog_test_run_opts(prog_fd, &tattr);
0029     if (!ASSERT_OK(err, "bpf_prog_test_run_opts"))
0030         goto out;
0031     test_result = skel->bss->test_result;
0032     if (!ASSERT_EQ(test_result, -EFAULT, "offset -1"))
0033         goto out;
0034
0035     skel->bss->load_offset = (uint32_t)10;
0036     err = bpf_prog_test_run_opts(prog_fd, &tattr);
0037     if (!ASSERT_OK(err, "bpf_prog_test_run_opts"))
0038         goto out;
0039     test_result = skel->bss->test_result;
0040     if (!ASSERT_EQ(test_result, 0, "offset 10"))
0041         goto out;
0042
0043 out:
0044     skb_load_bytes__destroy(skel);
0045 }