btf_module.c (780B)
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}