bpf/prog_tests/ksyms_module.c

0001 // SPDX-License-Identifier: GPL-2.0
0002 /* Copyright (c) 2021 Facebook */
0003
0004 #include <test_progs.h>
0005 #include <network_helpers.h>
0006 #include "test_ksyms_module.lskel.h"
0007 #include "test_ksyms_module.skel.h"
0008
0009 static void test_ksyms_module_lskel(void)
0010 {
0011     struct test_ksyms_module_lskel *skel;
0012     int err;
0013     LIBBPF_OPTS(bpf_test_run_opts, topts,
0014         .data_in = &pkt_v4,
0015         .data_size_in = sizeof(pkt_v4),
0016         .repeat = 1,
0017     );
0018
0019     if (!env.has_testmod) {
0020         test__skip();
0021         return;
0022     }
0023
0024     skel = test_ksyms_module_lskel__open_and_load();
0025     if (!ASSERT_OK_PTR(skel, "test_ksyms_module_lskel__open_and_load"))
0026         return;
0027     err = bpf_prog_test_run_opts(skel->progs.load.prog_fd, &topts);
0028     if (!ASSERT_OK(err, "bpf_prog_test_run"))
0029         goto cleanup;
0030     ASSERT_EQ(topts.retval, 0, "retval");
0031     ASSERT_EQ(skel->bss->out_bpf_testmod_ksym, 42, "bpf_testmod_ksym");
0032 cleanup:
0033     test_ksyms_module_lskel__destroy(skel);
0034 }
0035
0036 static void test_ksyms_module_libbpf(void)
0037 {
0038     struct test_ksyms_module *skel;
0039     int err;
0040     LIBBPF_OPTS(bpf_test_run_opts, topts,
0041         .data_in = &pkt_v4,
0042         .data_size_in = sizeof(pkt_v4),
0043         .repeat = 1,
0044     );
0045
0046     if (!env.has_testmod) {
0047         test__skip();
0048         return;
0049     }
0050
0051     skel = test_ksyms_module__open_and_load();
0052     if (!ASSERT_OK_PTR(skel, "test_ksyms_module__open"))
0053         return;
0054     err = bpf_prog_test_run_opts(bpf_program__fd(skel->progs.load), &topts);
0055     if (!ASSERT_OK(err, "bpf_prog_test_run"))
0056         goto cleanup;
0057     ASSERT_EQ(topts.retval, 0, "retval");
0058     ASSERT_EQ(skel->bss->out_bpf_testmod_ksym, 42, "bpf_testmod_ksym");
0059 cleanup:
0060     test_ksyms_module__destroy(skel);
0061 }
0062
0063 void test_ksyms_module(void)
0064 {
0065     if (test__start_subtest("lskel"))
0066         test_ksyms_module_lskel();
0067     if (test__start_subtest("libbpf"))
0068         test_ksyms_module_libbpf();
0069 }