1 /* SPDX-License-Identifier: GPL-2.0 */ 2 #ifndef __TRACE_HELPER_H 3 #define __TRACE_HELPER_H 4 5 #include <bpf/libbpf.h> 6 7 #ifdef __x86_64__ 8 #define SYS_PREFIX "__x64_" 9 #elif defined(__s390x__) 10 #define SYS_PREFIX "__s390x_" 11 #elif defined(__aarch64__) 12 #define SYS_PREFIX "__arm64_" 13 #elif defined(__riscv) 14 #define SYS_PREFIX "__riscv_" 15 #else 16 #define SYS_PREFIX "" 17 #endif 18 19 #define __ALIGN_MASK(x, mask) (((x)+(mask))&~(mask)) 20 #define ALIGN(x, a) __ALIGN_MASK(x, (typeof(x))(a)-1) 21 22 struct ksym { 23 long addr; 24 char *name; 25 }; 26 27 struct ksyms { 28 struct ksym *syms; 29 size_t sym_cap; 30 size_t sym_cnt; 31 char **filtered_syms; 32 size_t filtered_cnt; 33 }; 34 35 typedef int (*ksym_cmp_t)(const void *p1, const void *p2); 36 typedef int (*ksym_search_cmp_t)(const void *p1, const struct ksym *p2); 37 38 int load_kallsyms(void); 39 struct ksym *ksym_search(long key); 40 long ksym_get_addr(const char *name); 41 unsigned long long module_get_base_addr(const char *module); 42 unsigned long long module_get_base_addr_sec(const char *module, const char *sec); 43 44 struct ksyms *load_kallsyms_local(void); 45 struct ksym *ksym_search_local(struct ksyms *ksyms, long key); 46 long ksym_get_addr_local(struct ksyms *ksyms, const char *name); 47 void free_kallsyms_local(struct ksyms *ksyms); 48 49 struct ksyms *load_kallsyms_custom_local(ksym_cmp_t cmp_cb); 50 struct ksym *search_kallsyms_custom_local(struct ksyms *ksyms, const void *p1, 51 ksym_search_cmp_t cmp_cb); 52 53 /* open kallsyms and find addresses on the fly, faster than load + search. */ 54 int kallsyms_find(const char *sym, unsigned long long *addr); 55 56 void read_trace_pipe(void); 57 int read_trace_pipe_iter(void (*cb)(const char *str, void *data), 58 void *data, int iter); 59 60 ssize_t get_uprobe_offset(const void *addr); 61 ssize_t get_rel_offset(uintptr_t addr); 62 63 int read_build_id(const char *path, char *build_id, size_t size); 64 65 int bpf_get_ksyms(struct ksyms **ksymsp, bool kernel); 66 int bpf_get_addrs(unsigned long **addrsp, size_t *cntp, bool kernel); 67 68 bool is_unsafe_function(const char *name); 69 #endif 70