1 // SPDX-License-Identifier: GPL-2.0
9 #include <linux/perf_event.h>
11 #include "trace_helpers.h"
13 #define MAX_SYMS 300000
14 static struct ksym syms[MAX_SYMS];
17 static int ksym_cmp(const void *p1, const void *p2)
19 return ((struct ksym *)p1)->addr - ((struct ksym *)p2)->addr;
22 int load_kallsyms(void)
24 FILE *f = fopen("/proc/kallsyms", "r");
25 char func[256], buf[256];
34 if (!fgets(buf, sizeof(buf), f))
36 if (sscanf(buf, "%p %c %s", &addr, &symbol, func) != 3)
40 syms[i].addr = (long) addr;
41 syms[i].name = strdup(func);
46 qsort(syms, sym_cnt, sizeof(struct ksym), ksym_cmp);
50 struct ksym *ksym_search(long key)
52 int start = 0, end = sym_cnt;
56 size_t mid = start + (end - start) / 2;
58 result = key - syms[mid].addr;
67 if (start >= 1 && syms[start - 1].addr < key &&
68 key < syms[start].addr)
70 return &syms[start - 1];
72 /* out of range. return _stext */
76 long ksym_get_addr(const char *name)
80 for (i = 0; i < sym_cnt; i++) {
81 if (strcmp(syms[i].name, name) == 0)
89 static int page_cnt = 8;
90 static struct perf_event_mmap_page *header;
92 int perf_event_mmap_header(int fd, struct perf_event_mmap_page **header)
97 page_size = getpagesize();
98 mmap_size = page_size * (page_cnt + 1);
100 base = mmap(NULL, mmap_size, PROT_READ | PROT_WRITE, MAP_SHARED, fd, 0);
101 if (base == MAP_FAILED) {
102 printf("mmap err\n");
110 int perf_event_mmap(int fd)
112 return perf_event_mmap_header(fd, &header);
115 static int perf_event_poll(int fd)
117 struct pollfd pfd = { .fd = fd, .events = POLLIN };
119 return poll(&pfd, 1, 1000);
122 struct perf_event_sample {
123 struct perf_event_header header;
128 static enum bpf_perf_event_ret
129 bpf_perf_event_print(struct perf_event_header *hdr, void *private_data)
131 struct perf_event_sample *e = (struct perf_event_sample *)hdr;
132 perf_event_print_fn fn = private_data;
135 if (e->header.type == PERF_RECORD_SAMPLE) {
136 ret = fn(e->data, e->size);
137 if (ret != LIBBPF_PERF_EVENT_CONT)
139 } else if (e->header.type == PERF_RECORD_LOST) {
141 struct perf_event_header header;
144 } *lost = (void *) e;
145 printf("lost %lld events\n", lost->lost);
147 printf("unknown event type=%d size=%d\n",
148 e->header.type, e->header.size);
151 return LIBBPF_PERF_EVENT_CONT;
154 int perf_event_poller(int fd, perf_event_print_fn output_fn)
156 enum bpf_perf_event_ret ret;
162 ret = bpf_perf_event_read_simple(header, page_cnt * page_size,
163 page_size, &buf, &len,
164 bpf_perf_event_print,
166 if (ret != LIBBPF_PERF_EVENT_CONT)
174 int perf_event_poller_multi(int *fds, struct perf_event_mmap_page **headers,
175 int num_fds, perf_event_print_fn output_fn)
177 enum bpf_perf_event_ret ret;
183 pfds = calloc(num_fds, sizeof(*pfds));
185 return LIBBPF_PERF_EVENT_ERROR;
187 for (i = 0; i < num_fds; i++) {
189 pfds[i].events = POLLIN;
193 poll(pfds, num_fds, 1000);
194 for (i = 0; i < num_fds; i++) {
195 if (!pfds[i].revents)
198 ret = bpf_perf_event_read_simple(headers[i],
199 page_cnt * page_size,
200 page_size, &buf, &len,
201 bpf_perf_event_print,
203 if (ret != LIBBPF_PERF_EVENT_CONT)