35 lines · c
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 7static const char *module_name = "bpf_testmod";8static const char *symbol_name = "bpf_testmod_test_read";9 10void 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 31cleanup:32 btf__free(module_btf);33 btf__free(vmlinux_btf);34}35