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 42 struct ksyms *load_kallsyms_local(void); 43 struct ksym *ksym_search_local(struct ksyms *ksyms, long key); 44 long ksym_get_addr_local(struct ksyms *ksyms, const char *name); 45 void free_kallsyms_local(struct ksyms *ksyms); 46 47 struct ksyms *load_kallsyms_custom_local(ksym_cmp_t cmp_cb); 48 struct ksym *search_kallsyms_custom_local(struct ksyms *ksyms, const void *p1, 49 ksym_search_cmp_t cmp_cb); 50 51 /* open kallsyms and find addresses on the fly, faster than load + search. */ 52 int kallsyms_find(const char *sym, unsigned long long *addr); 53 54 void read_trace_pipe(void); 55 int read_trace_pipe_iter(void (*cb)(const char *str, void *data), 56 void *data, int iter); 57 58 ssize_t get_uprobe_offset(const void *addr); 59 ssize_t get_rel_offset(uintptr_t addr); 60 61 int read_build_id(const char *path, char *build_id, size_t size); 62 63 int bpf_get_ksyms(struct ksyms **ksymsp, bool kernel); 64 int bpf_get_addrs(unsigned long **addrsp, size_t *cntp, bool kernel); 65 66 #endif 67