xref: /linux/tools/testing/selftests/bpf/trace_helpers.h (revision 69050f8d6d075dc01af7a5f2f550a8067510366f)
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 struct ksyms;
27 
28 typedef int (*ksym_cmp_t)(const void *p1, const void *p2);
29 typedef int (*ksym_search_cmp_t)(const void *p1, const struct ksym *p2);
30 
31 int load_kallsyms(void);
32 struct ksym *ksym_search(long key);
33 long ksym_get_addr(const char *name);
34 
35 struct ksyms *load_kallsyms_local(void);
36 struct ksym *ksym_search_local(struct ksyms *ksyms, long key);
37 long ksym_get_addr_local(struct ksyms *ksyms, const char *name);
38 void free_kallsyms_local(struct ksyms *ksyms);
39 
40 struct ksyms *load_kallsyms_custom_local(ksym_cmp_t cmp_cb);
41 struct ksym *search_kallsyms_custom_local(struct ksyms *ksyms, const void *p1,
42 					  ksym_search_cmp_t cmp_cb);
43 
44 /* open kallsyms and find addresses on the fly, faster than load + search. */
45 int kallsyms_find(const char *sym, unsigned long long *addr);
46 
47 void read_trace_pipe(void);
48 int read_trace_pipe_iter(void (*cb)(const char *str, void *data),
49 			 void *data, int iter);
50 
51 ssize_t get_uprobe_offset(const void *addr);
52 ssize_t get_rel_offset(uintptr_t addr);
53 
54 int read_build_id(const char *path, char *build_id, size_t size);
55 
56 int bpf_get_ksyms(char ***symsp, size_t *cntp, bool kernel);
57 int bpf_get_addrs(unsigned long **addrsp, size_t *cntp, bool kernel);
58 
59 #endif
60