xref: /linux/tools/testing/selftests/bpf/prog_tests/btf_module.c (revision c532de5a67a70f8533d495f8f2aaa9a0491c3ad0)
1 /* SPDX-License-Identifier: GPL-2.0 */
2 /* Copyright (c) 2021 Hengqi Chen */
3 
4 #include <test_progs.h>
5 #include <bpf/btf.h>
6 
7 static const char *module_name = "bpf_testmod";
8 static const char *symbol_name = "bpf_testmod_test_read";
9 
10 void test_btf_module()
11 {
12 	struct btf *vmlinux_btf, *module_btf;
13 	__s32 type_id;
14 
15 	if (!env.has_testmod) {
16 		test__skip();
17 		return;
18 	}
19 
20 	vmlinux_btf = btf__load_vmlinux_btf();
21 	if (!ASSERT_OK_PTR(vmlinux_btf, "could not load vmlinux BTF"))
22 		return;
23 
24 	module_btf = btf__load_module_btf(module_name, vmlinux_btf);
25 	if (!ASSERT_OK_PTR(module_btf, "could not load module BTF"))
26 		goto cleanup;
27 
28 	type_id = btf__find_by_name(module_btf, symbol_name);
29 	ASSERT_GT(type_id, 0, "func not found");
30 
31 cleanup:
32 	btf__free(module_btf);
33 	btf__free(vmlinux_btf);
34 }
35