04fcb5f9a1
bpf_prog_test_run is being deprecated in favor of the OPTS-based bpf_prog_test_run_opts. We end up unable to use CHECK in most cases, so replace usages with ASSERT_* calls. Signed-off-by: Delyan Kratunov <delyank@fb.com> Signed-off-by: Andrii Nakryiko <andrii@kernel.org> Link: https://lore.kernel.org/bpf/20220202235423.1097270-2-delyank@fb.com
111 lines
3.0 KiB
C
111 lines
3.0 KiB
C
// SPDX-License-Identifier: GPL-2.0
|
|
#include <test_progs.h>
|
|
#include <network_helpers.h>
|
|
|
|
void test_xdp_update_frags(void)
|
|
{
|
|
const char *file = "./test_xdp_update_frags.o";
|
|
struct bpf_program *prog;
|
|
struct bpf_object *obj;
|
|
int err, prog_fd;
|
|
__u32 *offset;
|
|
__u8 *buf;
|
|
LIBBPF_OPTS(bpf_test_run_opts, topts);
|
|
|
|
obj = bpf_object__open(file);
|
|
if (libbpf_get_error(obj))
|
|
return;
|
|
|
|
prog = bpf_object__next_program(obj, NULL);
|
|
if (bpf_object__load(obj))
|
|
return;
|
|
|
|
prog_fd = bpf_program__fd(prog);
|
|
|
|
buf = malloc(128);
|
|
if (!ASSERT_OK_PTR(buf, "alloc buf 128b"))
|
|
goto out;
|
|
|
|
memset(buf, 0, 128);
|
|
offset = (__u32 *)buf;
|
|
*offset = 16;
|
|
buf[*offset] = 0xaa; /* marker at offset 16 (head) */
|
|
buf[*offset + 15] = 0xaa; /* marker at offset 31 (head) */
|
|
|
|
topts.data_in = buf;
|
|
topts.data_out = buf;
|
|
topts.data_size_in = 128;
|
|
topts.data_size_out = 128;
|
|
|
|
err = bpf_prog_test_run_opts(prog_fd, &topts);
|
|
|
|
/* test_xdp_update_frags: buf[16,31]: 0xaa -> 0xbb */
|
|
ASSERT_OK(err, "xdp_update_frag");
|
|
ASSERT_EQ(topts.retval, XDP_PASS, "xdp_update_frag retval");
|
|
ASSERT_EQ(buf[16], 0xbb, "xdp_update_frag buf[16]");
|
|
ASSERT_EQ(buf[31], 0xbb, "xdp_update_frag buf[31]");
|
|
|
|
free(buf);
|
|
|
|
buf = malloc(9000);
|
|
if (!ASSERT_OK_PTR(buf, "alloc buf 9Kb"))
|
|
goto out;
|
|
|
|
memset(buf, 0, 9000);
|
|
offset = (__u32 *)buf;
|
|
*offset = 5000;
|
|
buf[*offset] = 0xaa; /* marker at offset 5000 (frag0) */
|
|
buf[*offset + 15] = 0xaa; /* marker at offset 5015 (frag0) */
|
|
|
|
topts.data_in = buf;
|
|
topts.data_out = buf;
|
|
topts.data_size_in = 9000;
|
|
topts.data_size_out = 9000;
|
|
|
|
err = bpf_prog_test_run_opts(prog_fd, &topts);
|
|
|
|
/* test_xdp_update_frags: buf[5000,5015]: 0xaa -> 0xbb */
|
|
ASSERT_OK(err, "xdp_update_frag");
|
|
ASSERT_EQ(topts.retval, XDP_PASS, "xdp_update_frag retval");
|
|
ASSERT_EQ(buf[5000], 0xbb, "xdp_update_frag buf[5000]");
|
|
ASSERT_EQ(buf[5015], 0xbb, "xdp_update_frag buf[5015]");
|
|
|
|
memset(buf, 0, 9000);
|
|
offset = (__u32 *)buf;
|
|
*offset = 3510;
|
|
buf[*offset] = 0xaa; /* marker at offset 3510 (head) */
|
|
buf[*offset + 15] = 0xaa; /* marker at offset 3525 (frag0) */
|
|
|
|
err = bpf_prog_test_run_opts(prog_fd, &topts);
|
|
|
|
/* test_xdp_update_frags: buf[3510,3525]: 0xaa -> 0xbb */
|
|
ASSERT_OK(err, "xdp_update_frag");
|
|
ASSERT_EQ(topts.retval, XDP_PASS, "xdp_update_frag retval");
|
|
ASSERT_EQ(buf[3510], 0xbb, "xdp_update_frag buf[3510]");
|
|
ASSERT_EQ(buf[3525], 0xbb, "xdp_update_frag buf[3525]");
|
|
|
|
memset(buf, 0, 9000);
|
|
offset = (__u32 *)buf;
|
|
*offset = 7606;
|
|
buf[*offset] = 0xaa; /* marker at offset 7606 (frag0) */
|
|
buf[*offset + 15] = 0xaa; /* marker at offset 7621 (frag1) */
|
|
|
|
err = bpf_prog_test_run_opts(prog_fd, &topts);
|
|
|
|
/* test_xdp_update_frags: buf[7606,7621]: 0xaa -> 0xbb */
|
|
ASSERT_OK(err, "xdp_update_frag");
|
|
ASSERT_EQ(topts.retval, XDP_PASS, "xdp_update_frag retval");
|
|
ASSERT_EQ(buf[7606], 0xbb, "xdp_update_frag buf[7606]");
|
|
ASSERT_EQ(buf[7621], 0xbb, "xdp_update_frag buf[7621]");
|
|
|
|
free(buf);
|
|
out:
|
|
bpf_object__close(obj);
|
|
}
|
|
|
|
void test_xdp_adjust_frags(void)
|
|
{
|
|
if (test__start_subtest("xdp_adjust_frags"))
|
|
test_xdp_update_frags();
|
|
}
|