xref: /linux/tools/testing/selftests/bpf/prog_tests/ksyms_module.c (revision 69bfec7548f4c1595bac0e3ddfc0458a5af31f4c)
1 // SPDX-License-Identifier: GPL-2.0
2 /* Copyright (c) 2021 Facebook */
3 
4 #include <test_progs.h>
5 #include <network_helpers.h>
6 #include "test_ksyms_module.lskel.h"
7 #include "test_ksyms_module.skel.h"
8 
9 static void test_ksyms_module_lskel(void)
10 {
11 	struct test_ksyms_module_lskel *skel;
12 	int err;
13 	LIBBPF_OPTS(bpf_test_run_opts, topts,
14 		.data_in = &pkt_v4,
15 		.data_size_in = sizeof(pkt_v4),
16 		.repeat = 1,
17 	);
18 
19 	if (!env.has_testmod) {
20 		test__skip();
21 		return;
22 	}
23 
24 	skel = test_ksyms_module_lskel__open_and_load();
25 	if (!ASSERT_OK_PTR(skel, "test_ksyms_module_lskel__open_and_load"))
26 		return;
27 	err = bpf_prog_test_run_opts(skel->progs.load.prog_fd, &topts);
28 	if (!ASSERT_OK(err, "bpf_prog_test_run"))
29 		goto cleanup;
30 	ASSERT_EQ(topts.retval, 0, "retval");
31 	ASSERT_EQ(skel->bss->out_bpf_testmod_ksym, 42, "bpf_testmod_ksym");
32 cleanup:
33 	test_ksyms_module_lskel__destroy(skel);
34 }
35 
36 static void test_ksyms_module_libbpf(void)
37 {
38 	struct test_ksyms_module *skel;
39 	int err;
40 	LIBBPF_OPTS(bpf_test_run_opts, topts,
41 		.data_in = &pkt_v4,
42 		.data_size_in = sizeof(pkt_v4),
43 		.repeat = 1,
44 	);
45 
46 	if (!env.has_testmod) {
47 		test__skip();
48 		return;
49 	}
50 
51 	skel = test_ksyms_module__open_and_load();
52 	if (!ASSERT_OK_PTR(skel, "test_ksyms_module__open"))
53 		return;
54 	err = bpf_prog_test_run_opts(bpf_program__fd(skel->progs.load), &topts);
55 	if (!ASSERT_OK(err, "bpf_prog_test_run"))
56 		goto cleanup;
57 	ASSERT_EQ(topts.retval, 0, "retval");
58 	ASSERT_EQ(skel->bss->out_bpf_testmod_ksym, 42, "bpf_testmod_ksym");
59 cleanup:
60 	test_ksyms_module__destroy(skel);
61 }
62 
63 void test_ksyms_module(void)
64 {
65 	if (test__start_subtest("lskel"))
66 		test_ksyms_module_lskel();
67 	if (test__start_subtest("libbpf"))
68 		test_ksyms_module_libbpf();
69 }
70