xref: /linux/tools/perf/util/jitdump.c (revision 5e3992fe72748ed3892be876f09d4d990548b7af)
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 
35 struct 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 
59 struct 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 
69 static int
70 jit_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 
111 static void
112 jit_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 
121 static int
122 jit_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 
136 static int
137 jit_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;
241 error:
242 	free(buf);
243 	funlockfile(jd->in);
244 	fclose(jd->in);
245 	return retval;
246 }
247 
248 static union jr_entry *
249 jit_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 
365 static int
366 jit_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 
378 static 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 
385 static 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 
392 static 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 
423 static 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 
537 out:
538 	free(event);
539 	return ret;
540 }
541 
542 static 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 
632 static 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 
658 static int
659 jit_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 	jd->unwinding_data = unwinding_data;
679 
680 	return 0;
681 }
682 
683 static int
684 jit_process_dump(struct jit_buf_desc *jd)
685 {
686 	union jr_entry *jr;
687 	int ret = 0;
688 
689 	while ((jr = jit_get_next_entry(jd))) {
690 		switch(jr->prefix.id) {
691 		case JIT_CODE_LOAD:
692 			ret = jit_repipe_code_load(jd, jr);
693 			break;
694 		case JIT_CODE_MOVE:
695 			ret = jit_repipe_code_move(jd, jr);
696 			break;
697 		case JIT_CODE_DEBUG_INFO:
698 			ret = jit_repipe_debug_info(jd, jr);
699 			break;
700 		case JIT_CODE_UNWINDING_INFO:
701 			ret = jit_repipe_unwinding_info(jd, jr);
702 			break;
703 		default:
704 			ret = 0;
705 			continue;
706 		}
707 	}
708 	return ret;
709 }
710 
711 static int
712 jit_inject(struct jit_buf_desc *jd, char *path)
713 {
714 	int ret;
715 
716 	if (verbose > 0)
717 		fprintf(stderr, "injecting: %s\n", path);
718 
719 	ret = jit_open(jd, path);
720 	if (ret)
721 		return -1;
722 
723 	ret = jit_process_dump(jd);
724 
725 	jit_close(jd);
726 
727 	if (verbose > 0)
728 		fprintf(stderr, "injected: %s (%d)\n", path, ret);
729 
730 	return 0;
731 }
732 
733 /*
734  * File must be with pattern .../jit-XXXX.dump
735  * where XXXX is the PID of the process which did the mmap()
736  * as captured in the RECORD_MMAP record
737  */
738 static int
739 jit_detect(char *mmap_name, pid_t pid, struct nsinfo *nsi)
740  {
741 	char *p;
742 	char *end = NULL;
743 	pid_t pid2;
744 
745 	if (verbose > 2)
746 		fprintf(stderr, "jit marker trying : %s\n", mmap_name);
747 	/*
748 	 * get file name
749 	 */
750 	p = strrchr(mmap_name, '/');
751 	if (!p)
752 		return -1;
753 
754 	/*
755 	 * match prefix
756 	 */
757 	if (strncmp(p, "/jit-", 5))
758 		return -1;
759 
760 	/*
761 	 * skip prefix
762 	 */
763 	p += 5;
764 
765 	/*
766 	 * must be followed by a pid
767 	 */
768 	if (!isdigit(*p))
769 		return -1;
770 
771 	pid2 = (int)strtol(p, &end, 10);
772 	if (!end)
773 		return -1;
774 
775 	/*
776 	 * pid does not match mmap pid
777 	 * pid==0 in system-wide mode (synthesized)
778 	 */
779 	if (pid && pid2 != nsinfo__nstgid(nsi))
780 		return -1;
781 	/*
782 	 * validate suffix
783 	 */
784 	if (strcmp(end, ".dump"))
785 		return -1;
786 
787 	if (verbose > 0)
788 		fprintf(stderr, "jit marker found: %s\n", mmap_name);
789 
790 	return 0;
791 }
792 
793 static void jit_add_pid(struct machine *machine, pid_t pid)
794 {
795 	struct thread *thread = machine__findnew_thread(machine, pid, pid);
796 
797 	if (!thread) {
798 		pr_err("%s: thread %d not found or created\n", __func__, pid);
799 		return;
800 	}
801 
802 	thread->priv = (void *)1;
803 }
804 
805 static bool jit_has_pid(struct machine *machine, pid_t pid)
806 {
807 	struct thread *thread = machine__find_thread(machine, pid, pid);
808 
809 	if (!thread)
810 		return 0;
811 
812 	return (bool)thread->priv;
813 }
814 
815 int
816 jit_process(struct perf_session *session,
817 	    struct perf_data *output,
818 	    struct machine *machine,
819 	    char *filename,
820 	    pid_t pid,
821 	    pid_t tid,
822 	    u64 *nbytes)
823 {
824 	struct thread *thread;
825 	struct nsinfo *nsi;
826 	struct evsel *first;
827 	struct jit_buf_desc jd;
828 	int ret;
829 
830 	thread = machine__findnew_thread(machine, pid, tid);
831 	if (thread == NULL) {
832 		pr_err("problem processing JIT mmap event, skipping it.\n");
833 		return 0;
834 	}
835 
836 	nsi = nsinfo__get(thread->nsinfo);
837 	thread__put(thread);
838 
839 	/*
840 	 * first, detect marker mmap (i.e., the jitdump mmap)
841 	 */
842 	if (jit_detect(filename, pid, nsi)) {
843 		nsinfo__put(nsi);
844 
845 		/*
846 		 * Strip //anon*, [anon:* and /memfd:* mmaps if we processed a jitdump for this pid
847 		 */
848 		if (jit_has_pid(machine, pid) &&
849 			((strncmp(filename, "//anon", 6) == 0) ||
850 			 (strncmp(filename, "[anon:", 6) == 0) ||
851 			 (strncmp(filename, "/memfd:", 7) == 0)))
852 			return 1;
853 
854 		return 0;
855 	}
856 
857 	memset(&jd, 0, sizeof(jd));
858 
859 	jd.session = session;
860 	jd.output  = output;
861 	jd.machine = machine;
862 	jd.nsi = nsi;
863 
864 	/*
865 	 * track sample_type to compute id_all layout
866 	 * perf sets the same sample type to all events as of now
867 	 */
868 	first = evlist__first(session->evlist);
869 	jd.sample_type = first->core.attr.sample_type;
870 
871 	*nbytes = 0;
872 
873 	ret = jit_inject(&jd, filename);
874 	if (!ret) {
875 		jit_add_pid(machine, pid);
876 		*nbytes = jd.bytes_written;
877 		ret = 1;
878 	}
879 
880 	nsinfo__put(jd.nsi);
881 	free(jd.buf);
882 
883 	return ret;
884 }
885