Loading...
1// SPDX-License-Identifier: GPL-2.0
2#include <sys/sysmacros.h>
3#include <sys/types.h>
4#include <errno.h>
5#include <libgen.h>
6#include <stdio.h>
7#include <stdlib.h>
8#include <string.h>
9#include <fcntl.h>
10#include <unistd.h>
11#include <inttypes.h>
12#include <byteswap.h>
13#include <sys/stat.h>
14#include <sys/mman.h>
15#include <linux/stringify.h>
16
17#include "build-id.h"
18#include "event.h"
19#include "debug.h"
20#include "evlist.h"
21#include "namespaces.h"
22#include "symbol.h"
23#include <elf.h>
24
25#include "tsc.h"
26#include "session.h"
27#include "jit.h"
28#include "jitdump.h"
29#include "genelf.h"
30#include "thread.h"
31
32#include <linux/ctype.h>
33#include <linux/zalloc.h>
34
35struct jit_buf_desc {
36 struct perf_data *output;
37 struct perf_session *session;
38 struct machine *machine;
39 struct nsinfo *nsi;
40 union jr_entry *entry;
41 void *buf;
42 uint64_t sample_type;
43 size_t bufsize;
44 FILE *in;
45 bool needs_bswap; /* handles cross-endianness */
46 bool use_arch_timestamp;
47 void *debug_data;
48 void *unwinding_data;
49 uint64_t unwinding_size;
50 uint64_t unwinding_mapped_size;
51 uint64_t eh_frame_hdr_size;
52 size_t nr_debug_entries;
53 uint32_t code_load_count;
54 u64 bytes_written;
55 struct rb_root code_root;
56 char dir[PATH_MAX];
57};
58
59struct jit_tool {
60 struct perf_tool tool;
61 struct perf_data output;
62 struct perf_data input;
63 u64 bytes_written;
64};
65
66#define hmax(a, b) ((a) > (b) ? (a) : (b))
67#define get_jit_tool(t) (container_of(tool, struct jit_tool, tool))
68
69static int
70jit_emit_elf(struct jit_buf_desc *jd,
71 char *filename,
72 const char *sym,
73 uint64_t code_addr,
74 const void *code,
75 int csize,
76 void *debug,
77 int nr_debug_entries,
78 void *unwinding,
79 uint32_t unwinding_header_size,
80 uint32_t unwinding_size)
81{
82 int ret, fd, saved_errno;
83 struct nscookie nsc;
84
85 if (verbose > 0)
86 fprintf(stderr, "write ELF image %s\n", filename);
87
88 nsinfo__mountns_enter(jd->nsi, &nsc);
89 fd = open(filename, O_CREAT|O_TRUNC|O_WRONLY, 0644);
90 saved_errno = errno;
91 nsinfo__mountns_exit(&nsc);
92 if (fd == -1) {
93 pr_warning("cannot create jit ELF %s: %s\n", filename, strerror(saved_errno));
94 return -1;
95 }
96
97 ret = jit_write_elf(fd, code_addr, sym, (const void *)code, csize, debug, nr_debug_entries,
98 unwinding, unwinding_header_size, unwinding_size);
99
100 close(fd);
101
102 if (ret) {
103 nsinfo__mountns_enter(jd->nsi, &nsc);
104 unlink(filename);
105 nsinfo__mountns_exit(&nsc);
106 }
107
108 return ret;
109}
110
111static void
112jit_close(struct jit_buf_desc *jd)
113{
114 if (!(jd && jd->in))
115 return;
116 funlockfile(jd->in);
117 fclose(jd->in);
118 jd->in = NULL;
119}
120
121static int
122jit_validate_events(struct perf_session *session)
123{
124 struct evsel *evsel;
125
126 /*
127 * check that all events use CLOCK_MONOTONIC
128 */
129 evlist__for_each_entry(session->evlist, evsel) {
130 if (evsel->core.attr.use_clockid == 0 || evsel->core.attr.clockid != CLOCK_MONOTONIC)
131 return -1;
132 }
133 return 0;
134}
135
136static int
137jit_open(struct jit_buf_desc *jd, const char *name)
138{
139 struct jitheader header;
140 struct nscookie nsc;
141 struct jr_prefix *prefix;
142 ssize_t bs, bsz = 0;
143 void *n, *buf = NULL;
144 int ret, retval = -1;
145
146 nsinfo__mountns_enter(jd->nsi, &nsc);
147 jd->in = fopen(name, "r");
148 nsinfo__mountns_exit(&nsc);
149 if (!jd->in)
150 return -1;
151
152 bsz = hmax(sizeof(header), sizeof(*prefix));
153
154 buf = malloc(bsz);
155 if (!buf)
156 goto error;
157
158 /*
159 * protect from writer modifying the file while we are reading it
160 */
161 flockfile(jd->in);
162
163 ret = fread(buf, sizeof(header), 1, jd->in);
164 if (ret != 1)
165 goto error;
166
167 memcpy(&header, buf, sizeof(header));
168
169 if (header.magic != JITHEADER_MAGIC) {
170 if (header.magic != JITHEADER_MAGIC_SW)
171 goto error;
172 jd->needs_bswap = true;
173 }
174
175 if (jd->needs_bswap) {
176 header.version = bswap_32(header.version);
177 header.total_size = bswap_32(header.total_size);
178 header.pid = bswap_32(header.pid);
179 header.elf_mach = bswap_32(header.elf_mach);
180 header.timestamp = bswap_64(header.timestamp);
181 header.flags = bswap_64(header.flags);
182 }
183
184 jd->use_arch_timestamp = header.flags & JITDUMP_FLAGS_ARCH_TIMESTAMP;
185
186 if (verbose > 2)
187 pr_debug("version=%u\nhdr.size=%u\nts=0x%llx\npid=%d\nelf_mach=%d\nuse_arch_timestamp=%d\n",
188 header.version,
189 header.total_size,
190 (unsigned long long)header.timestamp,
191 header.pid,
192 header.elf_mach,
193 jd->use_arch_timestamp);
194
195 if (header.version > JITHEADER_VERSION) {
196 pr_err("wrong jitdump version %u, expected " __stringify(JITHEADER_VERSION),
197 header.version);
198 goto error;
199 }
200
201 if (header.flags & JITDUMP_FLAGS_RESERVED) {
202 pr_err("jitdump file contains invalid or unsupported flags 0x%llx\n",
203 (unsigned long long)header.flags & JITDUMP_FLAGS_RESERVED);
204 goto error;
205 }
206
207 if (jd->use_arch_timestamp && !jd->session->time_conv.time_mult) {
208 pr_err("jitdump file uses arch timestamps but there is no timestamp conversion\n");
209 goto error;
210 }
211
212 /*
213 * validate event is using the correct clockid
214 */
215 if (!jd->use_arch_timestamp && jit_validate_events(jd->session)) {
216 pr_err("error, jitted code must be sampled with perf record -k 1\n");
217 goto error;
218 }
219
220 bs = header.total_size - sizeof(header);
221
222 if (bs > bsz) {
223 n = realloc(buf, bs);
224 if (!n)
225 goto error;
226 bsz = bs;
227 buf = n;
228 /* read extra we do not know about */
229 ret = fread(buf, bs - bsz, 1, jd->in);
230 if (ret != 1)
231 goto error;
232 }
233 /*
234 * keep dirname for generating files and mmap records
235 */
236 strcpy(jd->dir, name);
237 dirname(jd->dir);
238 free(buf);
239
240 return 0;
241error:
242 free(buf);
243 funlockfile(jd->in);
244 fclose(jd->in);
245 return retval;
246}
247
248static union jr_entry *
249jit_get_next_entry(struct jit_buf_desc *jd)
250{
251 struct jr_prefix *prefix;
252 union jr_entry *jr;
253 void *addr;
254 size_t bs, size;
255 int id, ret;
256
257 if (!(jd && jd->in))
258 return NULL;
259
260 if (jd->buf == NULL) {
261 size_t sz = getpagesize();
262 if (sz < sizeof(*prefix))
263 sz = sizeof(*prefix);
264
265 jd->buf = malloc(sz);
266 if (jd->buf == NULL)
267 return NULL;
268
269 jd->bufsize = sz;
270 }
271
272 prefix = jd->buf;
273
274 /*
275 * file is still locked at this point
276 */
277 ret = fread(prefix, sizeof(*prefix), 1, jd->in);
278 if (ret != 1)
279 return NULL;
280
281 if (jd->needs_bswap) {
282 prefix->id = bswap_32(prefix->id);
283 prefix->total_size = bswap_32(prefix->total_size);
284 prefix->timestamp = bswap_64(prefix->timestamp);
285 }
286 id = prefix->id;
287 size = prefix->total_size;
288
289 bs = (size_t)size;
290 if (bs < sizeof(*prefix))
291 return NULL;
292
293 if (id >= JIT_CODE_MAX) {
294 pr_warning("next_entry: unknown record type %d, skipping\n", id);
295 }
296 if (bs > jd->bufsize) {
297 void *n;
298 n = realloc(jd->buf, bs);
299 if (!n)
300 return NULL;
301 jd->buf = n;
302 jd->bufsize = bs;
303 }
304
305 addr = ((void *)jd->buf) + sizeof(*prefix);
306
307 ret = fread(addr, bs - sizeof(*prefix), 1, jd->in);
308 if (ret != 1)
309 return NULL;
310
311 jr = (union jr_entry *)jd->buf;
312
313 switch(id) {
314 case JIT_CODE_DEBUG_INFO:
315 if (jd->needs_bswap) {
316 uint64_t n;
317 jr->info.code_addr = bswap_64(jr->info.code_addr);
318 jr->info.nr_entry = bswap_64(jr->info.nr_entry);
319 for (n = 0 ; n < jr->info.nr_entry; n++) {
320 jr->info.entries[n].addr = bswap_64(jr->info.entries[n].addr);
321 jr->info.entries[n].lineno = bswap_32(jr->info.entries[n].lineno);
322 jr->info.entries[n].discrim = bswap_32(jr->info.entries[n].discrim);
323 }
324 }
325 break;
326 case JIT_CODE_UNWINDING_INFO:
327 if (jd->needs_bswap) {
328 jr->unwinding.unwinding_size = bswap_64(jr->unwinding.unwinding_size);
329 jr->unwinding.eh_frame_hdr_size = bswap_64(jr->unwinding.eh_frame_hdr_size);
330 jr->unwinding.mapped_size = bswap_64(jr->unwinding.mapped_size);
331 }
332 break;
333 case JIT_CODE_CLOSE:
334 break;
335 case JIT_CODE_LOAD:
336 if (jd->needs_bswap) {
337 jr->load.pid = bswap_32(jr->load.pid);
338 jr->load.tid = bswap_32(jr->load.tid);
339 jr->load.vma = bswap_64(jr->load.vma);
340 jr->load.code_addr = bswap_64(jr->load.code_addr);
341 jr->load.code_size = bswap_64(jr->load.code_size);
342 jr->load.code_index= bswap_64(jr->load.code_index);
343 }
344 jd->code_load_count++;
345 break;
346 case JIT_CODE_MOVE:
347 if (jd->needs_bswap) {
348 jr->move.pid = bswap_32(jr->move.pid);
349 jr->move.tid = bswap_32(jr->move.tid);
350 jr->move.vma = bswap_64(jr->move.vma);
351 jr->move.old_code_addr = bswap_64(jr->move.old_code_addr);
352 jr->move.new_code_addr = bswap_64(jr->move.new_code_addr);
353 jr->move.code_size = bswap_64(jr->move.code_size);
354 jr->move.code_index = bswap_64(jr->move.code_index);
355 }
356 break;
357 case JIT_CODE_MAX:
358 default:
359 /* skip unknown record (we have read them) */
360 break;
361 }
362 return jr;
363}
364
365static int
366jit_inject_event(struct jit_buf_desc *jd, union perf_event *event)
367{
368 ssize_t size;
369
370 size = perf_data__write(jd->output, event, event->header.size);
371 if (size < 0)
372 return -1;
373
374 jd->bytes_written += size;
375 return 0;
376}
377
378static pid_t jr_entry_pid(struct jit_buf_desc *jd, union jr_entry *jr)
379{
380 if (jd->nsi && nsinfo__in_pidns(jd->nsi))
381 return nsinfo__tgid(jd->nsi);
382 return jr->load.pid;
383}
384
385static pid_t jr_entry_tid(struct jit_buf_desc *jd, union jr_entry *jr)
386{
387 if (jd->nsi && nsinfo__in_pidns(jd->nsi))
388 return nsinfo__pid(jd->nsi);
389 return jr->load.tid;
390}
391
392static uint64_t convert_timestamp(struct jit_buf_desc *jd, uint64_t timestamp)
393{
394 struct perf_tsc_conversion tc = { .time_shift = 0, };
395 struct perf_record_time_conv *time_conv = &jd->session->time_conv;
396
397 if (!jd->use_arch_timestamp)
398 return timestamp;
399
400 tc.time_shift = time_conv->time_shift;
401 tc.time_mult = time_conv->time_mult;
402 tc.time_zero = time_conv->time_zero;
403
404 /*
405 * The event TIME_CONV was extended for the fields from "time_cycles"
406 * when supported cap_user_time_short, for backward compatibility,
407 * checks the event size and assigns these extended fields if these
408 * fields are contained in the event.
409 */
410 if (event_contains(*time_conv, time_cycles)) {
411 tc.time_cycles = time_conv->time_cycles;
412 tc.time_mask = time_conv->time_mask;
413 tc.cap_user_time_zero = time_conv->cap_user_time_zero;
414 tc.cap_user_time_short = time_conv->cap_user_time_short;
415
416 if (!tc.cap_user_time_zero)
417 return 0;
418 }
419
420 return tsc_to_perf_time(timestamp, &tc);
421}
422
423static int jit_repipe_code_load(struct jit_buf_desc *jd, union jr_entry *jr)
424{
425 struct perf_sample sample;
426 union perf_event *event;
427 struct perf_tool *tool = jd->session->tool;
428 uint64_t code, addr;
429 uintptr_t uaddr;
430 char *filename;
431 struct stat st;
432 size_t size;
433 u16 idr_size;
434 const char *sym;
435 uint64_t count;
436 int ret, csize, usize;
437 pid_t nspid, pid, tid;
438 struct {
439 u32 pid, tid;
440 u64 time;
441 } *id;
442
443 nspid = jr->load.pid;
444 pid = jr_entry_pid(jd, jr);
445 tid = jr_entry_tid(jd, jr);
446 csize = jr->load.code_size;
447 usize = jd->unwinding_mapped_size;
448 addr = jr->load.code_addr;
449 sym = (void *)((unsigned long)jr + sizeof(jr->load));
450 code = (unsigned long)jr + jr->load.p.total_size - csize;
451 count = jr->load.code_index;
452 idr_size = jd->machine->id_hdr_size;
453
454 event = calloc(1, sizeof(*event) + idr_size);
455 if (!event)
456 return -1;
457
458 filename = event->mmap2.filename;
459 size = snprintf(filename, PATH_MAX, "%s/jitted-%d-%" PRIu64 ".so",
460 jd->dir,
461 nspid,
462 count);
463
464 size++; /* for \0 */
465
466 size = PERF_ALIGN(size, sizeof(u64));
467 uaddr = (uintptr_t)code;
468 ret = jit_emit_elf(jd, filename, sym, addr, (const void *)uaddr, csize, jd->debug_data, jd->nr_debug_entries,
469 jd->unwinding_data, jd->eh_frame_hdr_size, jd->unwinding_size);
470
471 if (jd->debug_data && jd->nr_debug_entries) {
472 zfree(&jd->debug_data);
473 jd->nr_debug_entries = 0;
474 }
475
476 if (jd->unwinding_data && jd->eh_frame_hdr_size) {
477 zfree(&jd->unwinding_data);
478 jd->eh_frame_hdr_size = 0;
479 jd->unwinding_mapped_size = 0;
480 jd->unwinding_size = 0;
481 }
482
483 if (ret) {
484 free(event);
485 return -1;
486 }
487 if (nsinfo__stat(filename, &st, jd->nsi))
488 memset(&st, 0, sizeof(st));
489
490 event->mmap2.header.type = PERF_RECORD_MMAP2;
491 event->mmap2.header.misc = PERF_RECORD_MISC_USER;
492 event->mmap2.header.size = (sizeof(event->mmap2) -
493 (sizeof(event->mmap2.filename) - size) + idr_size);
494
495 event->mmap2.pgoff = GEN_ELF_TEXT_OFFSET;
496 event->mmap2.start = addr;
497 event->mmap2.len = usize ? ALIGN_8(csize) + usize : csize;
498 event->mmap2.pid = pid;
499 event->mmap2.tid = tid;
500 event->mmap2.ino = st.st_ino;
501 event->mmap2.maj = major(st.st_dev);
502 event->mmap2.min = minor(st.st_dev);
503 event->mmap2.prot = st.st_mode;
504 event->mmap2.flags = MAP_SHARED;
505 event->mmap2.ino_generation = 1;
506
507 id = (void *)((unsigned long)event + event->mmap.header.size - idr_size);
508 if (jd->sample_type & PERF_SAMPLE_TID) {
509 id->pid = pid;
510 id->tid = tid;
511 }
512 if (jd->sample_type & PERF_SAMPLE_TIME)
513 id->time = convert_timestamp(jd, jr->load.p.timestamp);
514
515 /*
516 * create pseudo sample to induce dso hit increment
517 * use first address as sample address
518 */
519 memset(&sample, 0, sizeof(sample));
520 sample.cpumode = PERF_RECORD_MISC_USER;
521 sample.pid = pid;
522 sample.tid = tid;
523 sample.time = id->time;
524 sample.ip = addr;
525
526 ret = perf_event__process_mmap2(tool, event, &sample, jd->machine);
527 if (ret)
528 goto out;
529
530 ret = jit_inject_event(jd, event);
531 /*
532 * mark dso as use to generate buildid in the header
533 */
534 if (!ret)
535 build_id__mark_dso_hit(tool, event, &sample, NULL, jd->machine);
536
537out:
538 free(event);
539 return ret;
540}
541
542static int jit_repipe_code_move(struct jit_buf_desc *jd, union jr_entry *jr)
543{
544 struct perf_sample sample;
545 union perf_event *event;
546 struct perf_tool *tool = jd->session->tool;
547 char *filename;
548 size_t size;
549 struct stat st;
550 int usize;
551 u16 idr_size;
552 int ret;
553 pid_t nspid, pid, tid;
554 struct {
555 u32 pid, tid;
556 u64 time;
557 } *id;
558
559 nspid = jr->load.pid;
560 pid = jr_entry_pid(jd, jr);
561 tid = jr_entry_tid(jd, jr);
562 usize = jd->unwinding_mapped_size;
563 idr_size = jd->machine->id_hdr_size;
564
565 /*
566 * +16 to account for sample_id_all (hack)
567 */
568 event = calloc(1, sizeof(*event) + 16);
569 if (!event)
570 return -1;
571
572 filename = event->mmap2.filename;
573 size = snprintf(filename, PATH_MAX, "%s/jitted-%d-%" PRIu64 ".so",
574 jd->dir,
575 nspid,
576 jr->move.code_index);
577
578 size++; /* for \0 */
579
580 if (nsinfo__stat(filename, &st, jd->nsi))
581 memset(&st, 0, sizeof(st));
582
583 size = PERF_ALIGN(size, sizeof(u64));
584
585 event->mmap2.header.type = PERF_RECORD_MMAP2;
586 event->mmap2.header.misc = PERF_RECORD_MISC_USER;
587 event->mmap2.header.size = (sizeof(event->mmap2) -
588 (sizeof(event->mmap2.filename) - size) + idr_size);
589 event->mmap2.pgoff = GEN_ELF_TEXT_OFFSET;
590 event->mmap2.start = jr->move.new_code_addr;
591 event->mmap2.len = usize ? ALIGN_8(jr->move.code_size) + usize
592 : jr->move.code_size;
593 event->mmap2.pid = pid;
594 event->mmap2.tid = tid;
595 event->mmap2.ino = st.st_ino;
596 event->mmap2.maj = major(st.st_dev);
597 event->mmap2.min = minor(st.st_dev);
598 event->mmap2.prot = st.st_mode;
599 event->mmap2.flags = MAP_SHARED;
600 event->mmap2.ino_generation = 1;
601
602 id = (void *)((unsigned long)event + event->mmap.header.size - idr_size);
603 if (jd->sample_type & PERF_SAMPLE_TID) {
604 id->pid = pid;
605 id->tid = tid;
606 }
607 if (jd->sample_type & PERF_SAMPLE_TIME)
608 id->time = convert_timestamp(jd, jr->load.p.timestamp);
609
610 /*
611 * create pseudo sample to induce dso hit increment
612 * use first address as sample address
613 */
614 memset(&sample, 0, sizeof(sample));
615 sample.cpumode = PERF_RECORD_MISC_USER;
616 sample.pid = pid;
617 sample.tid = tid;
618 sample.time = id->time;
619 sample.ip = jr->move.new_code_addr;
620
621 ret = perf_event__process_mmap2(tool, event, &sample, jd->machine);
622 if (ret)
623 return ret;
624
625 ret = jit_inject_event(jd, event);
626 if (!ret)
627 build_id__mark_dso_hit(tool, event, &sample, NULL, jd->machine);
628
629 return ret;
630}
631
632static int jit_repipe_debug_info(struct jit_buf_desc *jd, union jr_entry *jr)
633{
634 void *data;
635 size_t sz;
636
637 if (!(jd && jr))
638 return -1;
639
640 sz = jr->prefix.total_size - sizeof(jr->info);
641 data = malloc(sz);
642 if (!data)
643 return -1;
644
645 memcpy(data, &jr->info.entries, sz);
646
647 jd->debug_data = data;
648
649 /*
650 * we must use nr_entry instead of size here because
651 * we cannot distinguish actual entry from padding otherwise
652 */
653 jd->nr_debug_entries = jr->info.nr_entry;
654
655 return 0;
656}
657
658static int
659jit_repipe_unwinding_info(struct jit_buf_desc *jd, union jr_entry *jr)
660{
661 void *unwinding_data;
662 uint32_t unwinding_data_size;
663
664 if (!(jd && jr))
665 return -1;
666
667 unwinding_data_size = jr->prefix.total_size - sizeof(jr->unwinding);
668 unwinding_data = malloc(unwinding_data_size);
669 if (!unwinding_data)
670 return -1;
671
672 memcpy(unwinding_data, &jr->unwinding.unwinding_data,
673 unwinding_data_size);
674
675 jd->eh_frame_hdr_size = jr->unwinding.eh_frame_hdr_size;
676 jd->unwinding_size = jr->unwinding.unwinding_size;
677 jd->unwinding_mapped_size = jr->unwinding.mapped_size;
678 free(jd->unwinding_data);
679 jd->unwinding_data = unwinding_data;
680
681 return 0;
682}
683
684static int
685jit_process_dump(struct jit_buf_desc *jd)
686{
687 union jr_entry *jr;
688 int ret = 0;
689
690 while ((jr = jit_get_next_entry(jd))) {
691 switch(jr->prefix.id) {
692 case JIT_CODE_LOAD:
693 ret = jit_repipe_code_load(jd, jr);
694 break;
695 case JIT_CODE_MOVE:
696 ret = jit_repipe_code_move(jd, jr);
697 break;
698 case JIT_CODE_DEBUG_INFO:
699 ret = jit_repipe_debug_info(jd, jr);
700 break;
701 case JIT_CODE_UNWINDING_INFO:
702 ret = jit_repipe_unwinding_info(jd, jr);
703 break;
704 default:
705 ret = 0;
706 continue;
707 }
708 }
709 return ret;
710}
711
712static int
713jit_inject(struct jit_buf_desc *jd, char *path)
714{
715 int ret;
716
717 if (verbose > 0)
718 fprintf(stderr, "injecting: %s\n", path);
719
720 ret = jit_open(jd, path);
721 if (ret)
722 return -1;
723
724 ret = jit_process_dump(jd);
725
726 jit_close(jd);
727
728 if (verbose > 0)
729 fprintf(stderr, "injected: %s (%d)\n", path, ret);
730
731 return 0;
732}
733
734/*
735 * File must be with pattern .../jit-XXXX.dump
736 * where XXXX is the PID of the process which did the mmap()
737 * as captured in the RECORD_MMAP record
738 */
739static int
740jit_detect(char *mmap_name, pid_t pid, struct nsinfo *nsi)
741 {
742 char *p;
743 char *end = NULL;
744 pid_t pid2;
745
746 if (verbose > 2)
747 fprintf(stderr, "jit marker trying : %s\n", mmap_name);
748 /*
749 * get file name
750 */
751 p = strrchr(mmap_name, '/');
752 if (!p)
753 return -1;
754
755 /*
756 * match prefix
757 */
758 if (strncmp(p, "/jit-", 5))
759 return -1;
760
761 /*
762 * skip prefix
763 */
764 p += 5;
765
766 /*
767 * must be followed by a pid
768 */
769 if (!isdigit(*p))
770 return -1;
771
772 pid2 = (int)strtol(p, &end, 10);
773 if (!end)
774 return -1;
775
776 /*
777 * pid does not match mmap pid
778 * pid==0 in system-wide mode (synthesized)
779 */
780 if (pid && pid2 != nsinfo__nstgid(nsi))
781 return -1;
782 /*
783 * validate suffix
784 */
785 if (strcmp(end, ".dump"))
786 return -1;
787
788 if (verbose > 0)
789 fprintf(stderr, "jit marker found: %s\n", mmap_name);
790
791 return 0;
792}
793
794static void jit_add_pid(struct machine *machine, pid_t pid)
795{
796 struct thread *thread = machine__findnew_thread(machine, pid, pid);
797
798 if (!thread) {
799 pr_err("%s: thread %d not found or created\n", __func__, pid);
800 return;
801 }
802
803 thread__set_priv(thread, (void *)true);
804 thread__put(thread);
805}
806
807static bool jit_has_pid(struct machine *machine, pid_t pid)
808{
809 struct thread *thread = machine__find_thread(machine, pid, pid);
810 void *priv;
811
812 if (!thread)
813 return false;
814
815 priv = thread__priv(thread);
816 thread__put(thread);
817 return (bool)priv;
818}
819
820int
821jit_process(struct perf_session *session,
822 struct perf_data *output,
823 struct machine *machine,
824 char *filename,
825 pid_t pid,
826 pid_t tid,
827 u64 *nbytes)
828{
829 struct thread *thread;
830 struct nsinfo *nsi;
831 struct evsel *first;
832 struct jit_buf_desc jd;
833 int ret;
834
835 thread = machine__findnew_thread(machine, pid, tid);
836 if (thread == NULL) {
837 pr_err("problem processing JIT mmap event, skipping it.\n");
838 return 0;
839 }
840
841 nsi = nsinfo__get(thread__nsinfo(thread));
842 thread__put(thread);
843
844 /*
845 * first, detect marker mmap (i.e., the jitdump mmap)
846 */
847 if (jit_detect(filename, pid, nsi)) {
848 nsinfo__put(nsi);
849
850 /*
851 * Strip //anon*, [anon:* and /memfd:* mmaps if we processed a jitdump for this pid
852 */
853 if (jit_has_pid(machine, pid) &&
854 ((strncmp(filename, "//anon", 6) == 0) ||
855 (strncmp(filename, "[anon:", 6) == 0) ||
856 (strncmp(filename, "/memfd:", 7) == 0)))
857 return 1;
858
859 return 0;
860 }
861
862 memset(&jd, 0, sizeof(jd));
863
864 jd.session = session;
865 jd.output = output;
866 jd.machine = machine;
867 jd.nsi = nsi;
868
869 /*
870 * track sample_type to compute id_all layout
871 * perf sets the same sample type to all events as of now
872 */
873 first = evlist__first(session->evlist);
874 jd.sample_type = first->core.attr.sample_type;
875
876 *nbytes = 0;
877
878 ret = jit_inject(&jd, filename);
879 if (!ret) {
880 jit_add_pid(machine, pid);
881 *nbytes = jd.bytes_written;
882 ret = 1;
883 }
884
885 nsinfo__put(jd.nsi);
886 free(jd.buf);
887
888 return ret;
889}
1// SPDX-License-Identifier: GPL-2.0
2#include <sys/sysmacros.h>
3#include <sys/types.h>
4#include <errno.h>
5#include <libgen.h>
6#include <stdio.h>
7#include <stdlib.h>
8#include <string.h>
9#include <fcntl.h>
10#include <unistd.h>
11#include <inttypes.h>
12#include <byteswap.h>
13#include <sys/stat.h>
14#include <sys/mman.h>
15#include <linux/stringify.h>
16
17#include "build-id.h"
18#include "event.h"
19#include "debug.h"
20#include "evlist.h"
21#include "namespaces.h"
22#include "symbol.h"
23#include <elf.h>
24
25#include "tsc.h"
26#include "session.h"
27#include "jit.h"
28#include "jitdump.h"
29#include "genelf.h"
30#include "thread.h"
31
32#include <linux/ctype.h>
33#include <linux/zalloc.h>
34
35struct jit_buf_desc {
36 struct perf_data *output;
37 struct perf_session *session;
38 struct machine *machine;
39 struct nsinfo *nsi;
40 union jr_entry *entry;
41 void *buf;
42 uint64_t sample_type;
43 size_t bufsize;
44 FILE *in;
45 bool needs_bswap; /* handles cross-endianness */
46 bool use_arch_timestamp;
47 void *debug_data;
48 void *unwinding_data;
49 uint64_t unwinding_size;
50 uint64_t unwinding_mapped_size;
51 uint64_t eh_frame_hdr_size;
52 size_t nr_debug_entries;
53 uint32_t code_load_count;
54 u64 bytes_written;
55 struct rb_root code_root;
56 char dir[PATH_MAX];
57};
58
59struct debug_line_info {
60 unsigned long vma;
61 unsigned int lineno;
62 /* The filename format is unspecified, absolute path, relative etc. */
63 char const filename[];
64};
65
66struct jit_tool {
67 struct perf_tool tool;
68 struct perf_data output;
69 struct perf_data input;
70 u64 bytes_written;
71};
72
73#define hmax(a, b) ((a) > (b) ? (a) : (b))
74#define get_jit_tool(t) (container_of(tool, struct jit_tool, tool))
75
76static int
77jit_emit_elf(struct jit_buf_desc *jd,
78 char *filename,
79 const char *sym,
80 uint64_t code_addr,
81 const void *code,
82 int csize,
83 void *debug,
84 int nr_debug_entries,
85 void *unwinding,
86 uint32_t unwinding_header_size,
87 uint32_t unwinding_size)
88{
89 int ret, fd, saved_errno;
90 struct nscookie nsc;
91
92 if (verbose > 0)
93 fprintf(stderr, "write ELF image %s\n", filename);
94
95 nsinfo__mountns_enter(jd->nsi, &nsc);
96 fd = open(filename, O_CREAT|O_TRUNC|O_WRONLY, 0644);
97 saved_errno = errno;
98 nsinfo__mountns_exit(&nsc);
99 if (fd == -1) {
100 pr_warning("cannot create jit ELF %s: %s\n", filename, strerror(saved_errno));
101 return -1;
102 }
103
104 ret = jit_write_elf(fd, code_addr, sym, (const void *)code, csize, debug, nr_debug_entries,
105 unwinding, unwinding_header_size, unwinding_size);
106
107 close(fd);
108
109 if (ret) {
110 nsinfo__mountns_enter(jd->nsi, &nsc);
111 unlink(filename);
112 nsinfo__mountns_exit(&nsc);
113 }
114
115 return ret;
116}
117
118static void
119jit_close(struct jit_buf_desc *jd)
120{
121 if (!(jd && jd->in))
122 return;
123 funlockfile(jd->in);
124 fclose(jd->in);
125 jd->in = NULL;
126}
127
128static int
129jit_validate_events(struct perf_session *session)
130{
131 struct evsel *evsel;
132
133 /*
134 * check that all events use CLOCK_MONOTONIC
135 */
136 evlist__for_each_entry(session->evlist, evsel) {
137 if (evsel->core.attr.use_clockid == 0 || evsel->core.attr.clockid != CLOCK_MONOTONIC)
138 return -1;
139 }
140 return 0;
141}
142
143static int
144jit_open(struct jit_buf_desc *jd, const char *name)
145{
146 struct jitheader header;
147 struct nscookie nsc;
148 struct jr_prefix *prefix;
149 ssize_t bs, bsz = 0;
150 void *n, *buf = NULL;
151 int ret, retval = -1;
152
153 nsinfo__mountns_enter(jd->nsi, &nsc);
154 jd->in = fopen(name, "r");
155 nsinfo__mountns_exit(&nsc);
156 if (!jd->in)
157 return -1;
158
159 bsz = hmax(sizeof(header), sizeof(*prefix));
160
161 buf = malloc(bsz);
162 if (!buf)
163 goto error;
164
165 /*
166 * protect from writer modifying the file while we are reading it
167 */
168 flockfile(jd->in);
169
170 ret = fread(buf, sizeof(header), 1, jd->in);
171 if (ret != 1)
172 goto error;
173
174 memcpy(&header, buf, sizeof(header));
175
176 if (header.magic != JITHEADER_MAGIC) {
177 if (header.magic != JITHEADER_MAGIC_SW)
178 goto error;
179 jd->needs_bswap = true;
180 }
181
182 if (jd->needs_bswap) {
183 header.version = bswap_32(header.version);
184 header.total_size = bswap_32(header.total_size);
185 header.pid = bswap_32(header.pid);
186 header.elf_mach = bswap_32(header.elf_mach);
187 header.timestamp = bswap_64(header.timestamp);
188 header.flags = bswap_64(header.flags);
189 }
190
191 jd->use_arch_timestamp = header.flags & JITDUMP_FLAGS_ARCH_TIMESTAMP;
192
193 if (verbose > 2)
194 pr_debug("version=%u\nhdr.size=%u\nts=0x%llx\npid=%d\nelf_mach=%d\nuse_arch_timestamp=%d\n",
195 header.version,
196 header.total_size,
197 (unsigned long long)header.timestamp,
198 header.pid,
199 header.elf_mach,
200 jd->use_arch_timestamp);
201
202 if (header.version > JITHEADER_VERSION) {
203 pr_err("wrong jitdump version %u, expected " __stringify(JITHEADER_VERSION),
204 header.version);
205 goto error;
206 }
207
208 if (header.flags & JITDUMP_FLAGS_RESERVED) {
209 pr_err("jitdump file contains invalid or unsupported flags 0x%llx\n",
210 (unsigned long long)header.flags & JITDUMP_FLAGS_RESERVED);
211 goto error;
212 }
213
214 if (jd->use_arch_timestamp && !jd->session->time_conv.time_mult) {
215 pr_err("jitdump file uses arch timestamps but there is no timestamp conversion\n");
216 goto error;
217 }
218
219 /*
220 * validate event is using the correct clockid
221 */
222 if (!jd->use_arch_timestamp && jit_validate_events(jd->session)) {
223 pr_err("error, jitted code must be sampled with perf record -k 1\n");
224 goto error;
225 }
226
227 bs = header.total_size - sizeof(header);
228
229 if (bs > bsz) {
230 n = realloc(buf, bs);
231 if (!n)
232 goto error;
233 bsz = bs;
234 buf = n;
235 /* read extra we do not know about */
236 ret = fread(buf, bs - bsz, 1, jd->in);
237 if (ret != 1)
238 goto error;
239 }
240 /*
241 * keep dirname for generating files and mmap records
242 */
243 strcpy(jd->dir, name);
244 dirname(jd->dir);
245
246 return 0;
247error:
248 funlockfile(jd->in);
249 fclose(jd->in);
250 return retval;
251}
252
253static union jr_entry *
254jit_get_next_entry(struct jit_buf_desc *jd)
255{
256 struct jr_prefix *prefix;
257 union jr_entry *jr;
258 void *addr;
259 size_t bs, size;
260 int id, ret;
261
262 if (!(jd && jd->in))
263 return NULL;
264
265 if (jd->buf == NULL) {
266 size_t sz = getpagesize();
267 if (sz < sizeof(*prefix))
268 sz = sizeof(*prefix);
269
270 jd->buf = malloc(sz);
271 if (jd->buf == NULL)
272 return NULL;
273
274 jd->bufsize = sz;
275 }
276
277 prefix = jd->buf;
278
279 /*
280 * file is still locked at this point
281 */
282 ret = fread(prefix, sizeof(*prefix), 1, jd->in);
283 if (ret != 1)
284 return NULL;
285
286 if (jd->needs_bswap) {
287 prefix->id = bswap_32(prefix->id);
288 prefix->total_size = bswap_32(prefix->total_size);
289 prefix->timestamp = bswap_64(prefix->timestamp);
290 }
291 id = prefix->id;
292 size = prefix->total_size;
293
294 bs = (size_t)size;
295 if (bs < sizeof(*prefix))
296 return NULL;
297
298 if (id >= JIT_CODE_MAX) {
299 pr_warning("next_entry: unknown record type %d, skipping\n", id);
300 }
301 if (bs > jd->bufsize) {
302 void *n;
303 n = realloc(jd->buf, bs);
304 if (!n)
305 return NULL;
306 jd->buf = n;
307 jd->bufsize = bs;
308 }
309
310 addr = ((void *)jd->buf) + sizeof(*prefix);
311
312 ret = fread(addr, bs - sizeof(*prefix), 1, jd->in);
313 if (ret != 1)
314 return NULL;
315
316 jr = (union jr_entry *)jd->buf;
317
318 switch(id) {
319 case JIT_CODE_DEBUG_INFO:
320 if (jd->needs_bswap) {
321 uint64_t n;
322 jr->info.code_addr = bswap_64(jr->info.code_addr);
323 jr->info.nr_entry = bswap_64(jr->info.nr_entry);
324 for (n = 0 ; n < jr->info.nr_entry; n++) {
325 jr->info.entries[n].addr = bswap_64(jr->info.entries[n].addr);
326 jr->info.entries[n].lineno = bswap_32(jr->info.entries[n].lineno);
327 jr->info.entries[n].discrim = bswap_32(jr->info.entries[n].discrim);
328 }
329 }
330 break;
331 case JIT_CODE_UNWINDING_INFO:
332 if (jd->needs_bswap) {
333 jr->unwinding.unwinding_size = bswap_64(jr->unwinding.unwinding_size);
334 jr->unwinding.eh_frame_hdr_size = bswap_64(jr->unwinding.eh_frame_hdr_size);
335 jr->unwinding.mapped_size = bswap_64(jr->unwinding.mapped_size);
336 }
337 break;
338 case JIT_CODE_CLOSE:
339 break;
340 case JIT_CODE_LOAD:
341 if (jd->needs_bswap) {
342 jr->load.pid = bswap_32(jr->load.pid);
343 jr->load.tid = bswap_32(jr->load.tid);
344 jr->load.vma = bswap_64(jr->load.vma);
345 jr->load.code_addr = bswap_64(jr->load.code_addr);
346 jr->load.code_size = bswap_64(jr->load.code_size);
347 jr->load.code_index= bswap_64(jr->load.code_index);
348 }
349 jd->code_load_count++;
350 break;
351 case JIT_CODE_MOVE:
352 if (jd->needs_bswap) {
353 jr->move.pid = bswap_32(jr->move.pid);
354 jr->move.tid = bswap_32(jr->move.tid);
355 jr->move.vma = bswap_64(jr->move.vma);
356 jr->move.old_code_addr = bswap_64(jr->move.old_code_addr);
357 jr->move.new_code_addr = bswap_64(jr->move.new_code_addr);
358 jr->move.code_size = bswap_64(jr->move.code_size);
359 jr->move.code_index = bswap_64(jr->move.code_index);
360 }
361 break;
362 case JIT_CODE_MAX:
363 default:
364 /* skip unknown record (we have read them) */
365 break;
366 }
367 return jr;
368}
369
370static int
371jit_inject_event(struct jit_buf_desc *jd, union perf_event *event)
372{
373 ssize_t size;
374
375 size = perf_data__write(jd->output, event, event->header.size);
376 if (size < 0)
377 return -1;
378
379 jd->bytes_written += size;
380 return 0;
381}
382
383static pid_t jr_entry_pid(struct jit_buf_desc *jd, union jr_entry *jr)
384{
385 if (jd->nsi && jd->nsi->in_pidns)
386 return jd->nsi->tgid;
387 return jr->load.pid;
388}
389
390static pid_t jr_entry_tid(struct jit_buf_desc *jd, union jr_entry *jr)
391{
392 if (jd->nsi && jd->nsi->in_pidns)
393 return jd->nsi->pid;
394 return jr->load.tid;
395}
396
397static uint64_t convert_timestamp(struct jit_buf_desc *jd, uint64_t timestamp)
398{
399 struct perf_tsc_conversion tc = { .time_shift = 0, };
400 struct perf_record_time_conv *time_conv = &jd->session->time_conv;
401
402 if (!jd->use_arch_timestamp)
403 return timestamp;
404
405 tc.time_shift = time_conv->time_shift;
406 tc.time_mult = time_conv->time_mult;
407 tc.time_zero = time_conv->time_zero;
408
409 /*
410 * The event TIME_CONV was extended for the fields from "time_cycles"
411 * when supported cap_user_time_short, for backward compatibility,
412 * checks the event size and assigns these extended fields if these
413 * fields are contained in the event.
414 */
415 if (event_contains(*time_conv, time_cycles)) {
416 tc.time_cycles = time_conv->time_cycles;
417 tc.time_mask = time_conv->time_mask;
418 tc.cap_user_time_zero = time_conv->cap_user_time_zero;
419 tc.cap_user_time_short = time_conv->cap_user_time_short;
420
421 if (!tc.cap_user_time_zero)
422 return 0;
423 }
424
425 return tsc_to_perf_time(timestamp, &tc);
426}
427
428static int jit_repipe_code_load(struct jit_buf_desc *jd, union jr_entry *jr)
429{
430 struct perf_sample sample;
431 union perf_event *event;
432 struct perf_tool *tool = jd->session->tool;
433 uint64_t code, addr;
434 uintptr_t uaddr;
435 char *filename;
436 struct stat st;
437 size_t size;
438 u16 idr_size;
439 const char *sym;
440 uint64_t count;
441 int ret, csize, usize;
442 pid_t nspid, pid, tid;
443 struct {
444 u32 pid, tid;
445 u64 time;
446 } *id;
447
448 nspid = jr->load.pid;
449 pid = jr_entry_pid(jd, jr);
450 tid = jr_entry_tid(jd, jr);
451 csize = jr->load.code_size;
452 usize = jd->unwinding_mapped_size;
453 addr = jr->load.code_addr;
454 sym = (void *)((unsigned long)jr + sizeof(jr->load));
455 code = (unsigned long)jr + jr->load.p.total_size - csize;
456 count = jr->load.code_index;
457 idr_size = jd->machine->id_hdr_size;
458
459 event = calloc(1, sizeof(*event) + idr_size);
460 if (!event)
461 return -1;
462
463 filename = event->mmap2.filename;
464 size = snprintf(filename, PATH_MAX, "%s/jitted-%d-%" PRIu64 ".so",
465 jd->dir,
466 nspid,
467 count);
468
469 size++; /* for \0 */
470
471 size = PERF_ALIGN(size, sizeof(u64));
472 uaddr = (uintptr_t)code;
473 ret = jit_emit_elf(jd, filename, sym, addr, (const void *)uaddr, csize, jd->debug_data, jd->nr_debug_entries,
474 jd->unwinding_data, jd->eh_frame_hdr_size, jd->unwinding_size);
475
476 if (jd->debug_data && jd->nr_debug_entries) {
477 zfree(&jd->debug_data);
478 jd->nr_debug_entries = 0;
479 }
480
481 if (jd->unwinding_data && jd->eh_frame_hdr_size) {
482 zfree(&jd->unwinding_data);
483 jd->eh_frame_hdr_size = 0;
484 jd->unwinding_mapped_size = 0;
485 jd->unwinding_size = 0;
486 }
487
488 if (ret) {
489 free(event);
490 return -1;
491 }
492 if (nsinfo__stat(filename, &st, jd->nsi))
493 memset(&st, 0, sizeof(st));
494
495 event->mmap2.header.type = PERF_RECORD_MMAP2;
496 event->mmap2.header.misc = PERF_RECORD_MISC_USER;
497 event->mmap2.header.size = (sizeof(event->mmap2) -
498 (sizeof(event->mmap2.filename) - size) + idr_size);
499
500 event->mmap2.pgoff = GEN_ELF_TEXT_OFFSET;
501 event->mmap2.start = addr;
502 event->mmap2.len = usize ? ALIGN_8(csize) + usize : csize;
503 event->mmap2.pid = pid;
504 event->mmap2.tid = tid;
505 event->mmap2.ino = st.st_ino;
506 event->mmap2.maj = major(st.st_dev);
507 event->mmap2.min = minor(st.st_dev);
508 event->mmap2.prot = st.st_mode;
509 event->mmap2.flags = MAP_SHARED;
510 event->mmap2.ino_generation = 1;
511
512 id = (void *)((unsigned long)event + event->mmap.header.size - idr_size);
513 if (jd->sample_type & PERF_SAMPLE_TID) {
514 id->pid = pid;
515 id->tid = tid;
516 }
517 if (jd->sample_type & PERF_SAMPLE_TIME)
518 id->time = convert_timestamp(jd, jr->load.p.timestamp);
519
520 /*
521 * create pseudo sample to induce dso hit increment
522 * use first address as sample address
523 */
524 memset(&sample, 0, sizeof(sample));
525 sample.cpumode = PERF_RECORD_MISC_USER;
526 sample.pid = pid;
527 sample.tid = tid;
528 sample.time = id->time;
529 sample.ip = addr;
530
531 ret = perf_event__process_mmap2(tool, event, &sample, jd->machine);
532 if (ret)
533 return ret;
534
535 ret = jit_inject_event(jd, event);
536 /*
537 * mark dso as use to generate buildid in the header
538 */
539 if (!ret)
540 build_id__mark_dso_hit(tool, event, &sample, NULL, jd->machine);
541
542 return ret;
543}
544
545static int jit_repipe_code_move(struct jit_buf_desc *jd, union jr_entry *jr)
546{
547 struct perf_sample sample;
548 union perf_event *event;
549 struct perf_tool *tool = jd->session->tool;
550 char *filename;
551 size_t size;
552 struct stat st;
553 int usize;
554 u16 idr_size;
555 int ret;
556 pid_t nspid, pid, tid;
557 struct {
558 u32 pid, tid;
559 u64 time;
560 } *id;
561
562 nspid = jr->load.pid;
563 pid = jr_entry_pid(jd, jr);
564 tid = jr_entry_tid(jd, jr);
565 usize = jd->unwinding_mapped_size;
566 idr_size = jd->machine->id_hdr_size;
567
568 /*
569 * +16 to account for sample_id_all (hack)
570 */
571 event = calloc(1, sizeof(*event) + 16);
572 if (!event)
573 return -1;
574
575 filename = event->mmap2.filename;
576 size = snprintf(filename, PATH_MAX, "%s/jitted-%d-%" PRIu64 ".so",
577 jd->dir,
578 nspid,
579 jr->move.code_index);
580
581 size++; /* for \0 */
582
583 if (nsinfo__stat(filename, &st, jd->nsi))
584 memset(&st, 0, sizeof(st));
585
586 size = PERF_ALIGN(size, sizeof(u64));
587
588 event->mmap2.header.type = PERF_RECORD_MMAP2;
589 event->mmap2.header.misc = PERF_RECORD_MISC_USER;
590 event->mmap2.header.size = (sizeof(event->mmap2) -
591 (sizeof(event->mmap2.filename) - size) + idr_size);
592 event->mmap2.pgoff = GEN_ELF_TEXT_OFFSET;
593 event->mmap2.start = jr->move.new_code_addr;
594 event->mmap2.len = usize ? ALIGN_8(jr->move.code_size) + usize
595 : jr->move.code_size;
596 event->mmap2.pid = pid;
597 event->mmap2.tid = tid;
598 event->mmap2.ino = st.st_ino;
599 event->mmap2.maj = major(st.st_dev);
600 event->mmap2.min = minor(st.st_dev);
601 event->mmap2.prot = st.st_mode;
602 event->mmap2.flags = MAP_SHARED;
603 event->mmap2.ino_generation = 1;
604
605 id = (void *)((unsigned long)event + event->mmap.header.size - idr_size);
606 if (jd->sample_type & PERF_SAMPLE_TID) {
607 id->pid = pid;
608 id->tid = tid;
609 }
610 if (jd->sample_type & PERF_SAMPLE_TIME)
611 id->time = convert_timestamp(jd, jr->load.p.timestamp);
612
613 /*
614 * create pseudo sample to induce dso hit increment
615 * use first address as sample address
616 */
617 memset(&sample, 0, sizeof(sample));
618 sample.cpumode = PERF_RECORD_MISC_USER;
619 sample.pid = pid;
620 sample.tid = tid;
621 sample.time = id->time;
622 sample.ip = jr->move.new_code_addr;
623
624 ret = perf_event__process_mmap2(tool, event, &sample, jd->machine);
625 if (ret)
626 return ret;
627
628 ret = jit_inject_event(jd, event);
629 if (!ret)
630 build_id__mark_dso_hit(tool, event, &sample, NULL, jd->machine);
631
632 return ret;
633}
634
635static int jit_repipe_debug_info(struct jit_buf_desc *jd, union jr_entry *jr)
636{
637 void *data;
638 size_t sz;
639
640 if (!(jd && jr))
641 return -1;
642
643 sz = jr->prefix.total_size - sizeof(jr->info);
644 data = malloc(sz);
645 if (!data)
646 return -1;
647
648 memcpy(data, &jr->info.entries, sz);
649
650 jd->debug_data = data;
651
652 /*
653 * we must use nr_entry instead of size here because
654 * we cannot distinguish actual entry from padding otherwise
655 */
656 jd->nr_debug_entries = jr->info.nr_entry;
657
658 return 0;
659}
660
661static int
662jit_repipe_unwinding_info(struct jit_buf_desc *jd, union jr_entry *jr)
663{
664 void *unwinding_data;
665 uint32_t unwinding_data_size;
666
667 if (!(jd && jr))
668 return -1;
669
670 unwinding_data_size = jr->prefix.total_size - sizeof(jr->unwinding);
671 unwinding_data = malloc(unwinding_data_size);
672 if (!unwinding_data)
673 return -1;
674
675 memcpy(unwinding_data, &jr->unwinding.unwinding_data,
676 unwinding_data_size);
677
678 jd->eh_frame_hdr_size = jr->unwinding.eh_frame_hdr_size;
679 jd->unwinding_size = jr->unwinding.unwinding_size;
680 jd->unwinding_mapped_size = jr->unwinding.mapped_size;
681 jd->unwinding_data = unwinding_data;
682
683 return 0;
684}
685
686static int
687jit_process_dump(struct jit_buf_desc *jd)
688{
689 union jr_entry *jr;
690 int ret = 0;
691
692 while ((jr = jit_get_next_entry(jd))) {
693 switch(jr->prefix.id) {
694 case JIT_CODE_LOAD:
695 ret = jit_repipe_code_load(jd, jr);
696 break;
697 case JIT_CODE_MOVE:
698 ret = jit_repipe_code_move(jd, jr);
699 break;
700 case JIT_CODE_DEBUG_INFO:
701 ret = jit_repipe_debug_info(jd, jr);
702 break;
703 case JIT_CODE_UNWINDING_INFO:
704 ret = jit_repipe_unwinding_info(jd, jr);
705 break;
706 default:
707 ret = 0;
708 continue;
709 }
710 }
711 return ret;
712}
713
714static int
715jit_inject(struct jit_buf_desc *jd, char *path)
716{
717 int ret;
718
719 if (verbose > 0)
720 fprintf(stderr, "injecting: %s\n", path);
721
722 ret = jit_open(jd, path);
723 if (ret)
724 return -1;
725
726 ret = jit_process_dump(jd);
727
728 jit_close(jd);
729
730 if (verbose > 0)
731 fprintf(stderr, "injected: %s (%d)\n", path, ret);
732
733 return 0;
734}
735
736/*
737 * File must be with pattern .../jit-XXXX.dump
738 * where XXXX is the PID of the process which did the mmap()
739 * as captured in the RECORD_MMAP record
740 */
741static int
742jit_detect(char *mmap_name, pid_t pid, struct nsinfo *nsi)
743 {
744 char *p;
745 char *end = NULL;
746 pid_t pid2;
747
748 if (verbose > 2)
749 fprintf(stderr, "jit marker trying : %s\n", mmap_name);
750 /*
751 * get file name
752 */
753 p = strrchr(mmap_name, '/');
754 if (!p)
755 return -1;
756
757 /*
758 * match prefix
759 */
760 if (strncmp(p, "/jit-", 5))
761 return -1;
762
763 /*
764 * skip prefix
765 */
766 p += 5;
767
768 /*
769 * must be followed by a pid
770 */
771 if (!isdigit(*p))
772 return -1;
773
774 pid2 = (int)strtol(p, &end, 10);
775 if (!end)
776 return -1;
777
778 /*
779 * pid does not match mmap pid
780 * pid==0 in system-wide mode (synthesized)
781 */
782 if (pid && pid2 != nsi->nstgid)
783 return -1;
784 /*
785 * validate suffix
786 */
787 if (strcmp(end, ".dump"))
788 return -1;
789
790 if (verbose > 0)
791 fprintf(stderr, "jit marker found: %s\n", mmap_name);
792
793 return 0;
794}
795
796static void jit_add_pid(struct machine *machine, pid_t pid)
797{
798 struct thread *thread = machine__findnew_thread(machine, pid, pid);
799
800 if (!thread) {
801 pr_err("%s: thread %d not found or created\n", __func__, pid);
802 return;
803 }
804
805 thread->priv = (void *)1;
806}
807
808static bool jit_has_pid(struct machine *machine, pid_t pid)
809{
810 struct thread *thread = machine__find_thread(machine, pid, pid);
811
812 if (!thread)
813 return 0;
814
815 return (bool)thread->priv;
816}
817
818int
819jit_process(struct perf_session *session,
820 struct perf_data *output,
821 struct machine *machine,
822 char *filename,
823 pid_t pid,
824 pid_t tid,
825 u64 *nbytes)
826{
827 struct thread *thread;
828 struct nsinfo *nsi;
829 struct evsel *first;
830 struct jit_buf_desc jd;
831 int ret;
832
833 thread = machine__findnew_thread(machine, pid, tid);
834 if (thread == NULL) {
835 pr_err("problem processing JIT mmap event, skipping it.\n");
836 return 0;
837 }
838
839 nsi = nsinfo__get(thread->nsinfo);
840 thread__put(thread);
841
842 /*
843 * first, detect marker mmap (i.e., the jitdump mmap)
844 */
845 if (jit_detect(filename, pid, nsi)) {
846 nsinfo__put(nsi);
847
848 // Strip //anon* mmaps if we processed a jitdump for this pid
849 if (jit_has_pid(machine, pid) && (strncmp(filename, "//anon", 6) == 0))
850 return 1;
851
852 return 0;
853 }
854
855 memset(&jd, 0, sizeof(jd));
856
857 jd.session = session;
858 jd.output = output;
859 jd.machine = machine;
860 jd.nsi = nsi;
861
862 /*
863 * track sample_type to compute id_all layout
864 * perf sets the same sample type to all events as of now
865 */
866 first = evlist__first(session->evlist);
867 jd.sample_type = first->core.attr.sample_type;
868
869 *nbytes = 0;
870
871 ret = jit_inject(&jd, filename);
872 if (!ret) {
873 jit_add_pid(machine, pid);
874 *nbytes = jd.bytes_written;
875 ret = 1;
876 }
877
878 nsinfo__put(jd.nsi);
879
880 return ret;
881}