xref: /openbmc/linux/tools/perf/builtin-inject.c (revision 31e67366)
1 // SPDX-License-Identifier: GPL-2.0
2 /*
3  * builtin-inject.c
4  *
5  * Builtin inject command: Examine the live mode (stdin) event stream
6  * and repipe it to stdout while optionally injecting additional
7  * events into it.
8  */
9 #include "builtin.h"
10 
11 #include "util/color.h"
12 #include "util/dso.h"
13 #include "util/vdso.h"
14 #include "util/evlist.h"
15 #include "util/evsel.h"
16 #include "util/map.h"
17 #include "util/session.h"
18 #include "util/tool.h"
19 #include "util/debug.h"
20 #include "util/build-id.h"
21 #include "util/data.h"
22 #include "util/auxtrace.h"
23 #include "util/jit.h"
24 #include "util/symbol.h"
25 #include "util/synthetic-events.h"
26 #include "util/thread.h"
27 #include "util/namespaces.h"
28 
29 #include <linux/err.h>
30 #include <subcmd/parse-options.h>
31 #include <uapi/linux/mman.h> /* To get things like MAP_HUGETLB even on older libc headers */
32 
33 #include <linux/list.h>
34 #include <errno.h>
35 #include <signal.h>
36 
37 struct perf_inject {
38 	struct perf_tool	tool;
39 	struct perf_session	*session;
40 	bool			build_ids;
41 	bool			build_id_all;
42 	bool			sched_stat;
43 	bool			have_auxtrace;
44 	bool			strip;
45 	bool			jit_mode;
46 	const char		*input_name;
47 	struct perf_data	output;
48 	u64			bytes_written;
49 	u64			aux_id;
50 	struct list_head	samples;
51 	struct itrace_synth_opts itrace_synth_opts;
52 	char			event_copy[PERF_SAMPLE_MAX_SIZE];
53 };
54 
55 struct event_entry {
56 	struct list_head node;
57 	u32		 tid;
58 	union perf_event event[];
59 };
60 
61 static int dso__inject_build_id(struct dso *dso, struct perf_tool *tool,
62 				struct machine *machine, u8 cpumode, u32 flags);
63 
64 static int output_bytes(struct perf_inject *inject, void *buf, size_t sz)
65 {
66 	ssize_t size;
67 
68 	size = perf_data__write(&inject->output, buf, sz);
69 	if (size < 0)
70 		return -errno;
71 
72 	inject->bytes_written += size;
73 	return 0;
74 }
75 
76 static int perf_event__repipe_synth(struct perf_tool *tool,
77 				    union perf_event *event)
78 {
79 	struct perf_inject *inject = container_of(tool, struct perf_inject,
80 						  tool);
81 
82 	return output_bytes(inject, event, event->header.size);
83 }
84 
85 static int perf_event__repipe_oe_synth(struct perf_tool *tool,
86 				       union perf_event *event,
87 				       struct ordered_events *oe __maybe_unused)
88 {
89 	return perf_event__repipe_synth(tool, event);
90 }
91 
92 #ifdef HAVE_JITDUMP
93 static int perf_event__drop_oe(struct perf_tool *tool __maybe_unused,
94 			       union perf_event *event __maybe_unused,
95 			       struct ordered_events *oe __maybe_unused)
96 {
97 	return 0;
98 }
99 #endif
100 
101 static int perf_event__repipe_op2_synth(struct perf_session *session,
102 					union perf_event *event)
103 {
104 	return perf_event__repipe_synth(session->tool, event);
105 }
106 
107 static int perf_event__repipe_op4_synth(struct perf_session *session,
108 					union perf_event *event,
109 					u64 data __maybe_unused)
110 {
111 	return perf_event__repipe_synth(session->tool, event);
112 }
113 
114 static int perf_event__repipe_attr(struct perf_tool *tool,
115 				   union perf_event *event,
116 				   struct evlist **pevlist)
117 {
118 	struct perf_inject *inject = container_of(tool, struct perf_inject,
119 						  tool);
120 	int ret;
121 
122 	ret = perf_event__process_attr(tool, event, pevlist);
123 	if (ret)
124 		return ret;
125 
126 	if (!inject->output.is_pipe)
127 		return 0;
128 
129 	return perf_event__repipe_synth(tool, event);
130 }
131 
132 static int perf_event__repipe_event_update(struct perf_tool *tool,
133 					   union perf_event *event,
134 					   struct evlist **pevlist __maybe_unused)
135 {
136 	return perf_event__repipe_synth(tool, event);
137 }
138 
139 #ifdef HAVE_AUXTRACE_SUPPORT
140 
141 static int copy_bytes(struct perf_inject *inject, int fd, off_t size)
142 {
143 	char buf[4096];
144 	ssize_t ssz;
145 	int ret;
146 
147 	while (size > 0) {
148 		ssz = read(fd, buf, min(size, (off_t)sizeof(buf)));
149 		if (ssz < 0)
150 			return -errno;
151 		ret = output_bytes(inject, buf, ssz);
152 		if (ret)
153 			return ret;
154 		size -= ssz;
155 	}
156 
157 	return 0;
158 }
159 
160 static s64 perf_event__repipe_auxtrace(struct perf_session *session,
161 				       union perf_event *event)
162 {
163 	struct perf_tool *tool = session->tool;
164 	struct perf_inject *inject = container_of(tool, struct perf_inject,
165 						  tool);
166 	int ret;
167 
168 	inject->have_auxtrace = true;
169 
170 	if (!inject->output.is_pipe) {
171 		off_t offset;
172 
173 		offset = lseek(inject->output.file.fd, 0, SEEK_CUR);
174 		if (offset == -1)
175 			return -errno;
176 		ret = auxtrace_index__auxtrace_event(&session->auxtrace_index,
177 						     event, offset);
178 		if (ret < 0)
179 			return ret;
180 	}
181 
182 	if (perf_data__is_pipe(session->data) || !session->one_mmap) {
183 		ret = output_bytes(inject, event, event->header.size);
184 		if (ret < 0)
185 			return ret;
186 		ret = copy_bytes(inject, perf_data__fd(session->data),
187 				 event->auxtrace.size);
188 	} else {
189 		ret = output_bytes(inject, event,
190 				   event->header.size + event->auxtrace.size);
191 	}
192 	if (ret < 0)
193 		return ret;
194 
195 	return event->auxtrace.size;
196 }
197 
198 #else
199 
200 static s64
201 perf_event__repipe_auxtrace(struct perf_session *session __maybe_unused,
202 			    union perf_event *event __maybe_unused)
203 {
204 	pr_err("AUX area tracing not supported\n");
205 	return -EINVAL;
206 }
207 
208 #endif
209 
210 static int perf_event__repipe(struct perf_tool *tool,
211 			      union perf_event *event,
212 			      struct perf_sample *sample __maybe_unused,
213 			      struct machine *machine __maybe_unused)
214 {
215 	return perf_event__repipe_synth(tool, event);
216 }
217 
218 static int perf_event__drop(struct perf_tool *tool __maybe_unused,
219 			    union perf_event *event __maybe_unused,
220 			    struct perf_sample *sample __maybe_unused,
221 			    struct machine *machine __maybe_unused)
222 {
223 	return 0;
224 }
225 
226 static int perf_event__drop_aux(struct perf_tool *tool,
227 				union perf_event *event __maybe_unused,
228 				struct perf_sample *sample,
229 				struct machine *machine __maybe_unused)
230 {
231 	struct perf_inject *inject = container_of(tool, struct perf_inject, tool);
232 
233 	if (!inject->aux_id)
234 		inject->aux_id = sample->id;
235 
236 	return 0;
237 }
238 
239 static union perf_event *
240 perf_inject__cut_auxtrace_sample(struct perf_inject *inject,
241 				 union perf_event *event,
242 				 struct perf_sample *sample)
243 {
244 	size_t sz1 = sample->aux_sample.data - (void *)event;
245 	size_t sz2 = event->header.size - sample->aux_sample.size - sz1;
246 	union perf_event *ev = (union perf_event *)inject->event_copy;
247 
248 	if (sz1 > event->header.size || sz2 > event->header.size ||
249 	    sz1 + sz2 > event->header.size ||
250 	    sz1 < sizeof(struct perf_event_header) + sizeof(u64))
251 		return event;
252 
253 	memcpy(ev, event, sz1);
254 	memcpy((void *)ev + sz1, (void *)event + event->header.size - sz2, sz2);
255 	ev->header.size = sz1 + sz2;
256 	((u64 *)((void *)ev + sz1))[-1] = 0;
257 
258 	return ev;
259 }
260 
261 typedef int (*inject_handler)(struct perf_tool *tool,
262 			      union perf_event *event,
263 			      struct perf_sample *sample,
264 			      struct evsel *evsel,
265 			      struct machine *machine);
266 
267 static int perf_event__repipe_sample(struct perf_tool *tool,
268 				     union perf_event *event,
269 				     struct perf_sample *sample,
270 				     struct evsel *evsel,
271 				     struct machine *machine)
272 {
273 	struct perf_inject *inject = container_of(tool, struct perf_inject,
274 						  tool);
275 
276 	if (evsel && evsel->handler) {
277 		inject_handler f = evsel->handler;
278 		return f(tool, event, sample, evsel, machine);
279 	}
280 
281 	build_id__mark_dso_hit(tool, event, sample, evsel, machine);
282 
283 	if (inject->itrace_synth_opts.set && sample->aux_sample.size)
284 		event = perf_inject__cut_auxtrace_sample(inject, event, sample);
285 
286 	return perf_event__repipe_synth(tool, event);
287 }
288 
289 static int perf_event__repipe_mmap(struct perf_tool *tool,
290 				   union perf_event *event,
291 				   struct perf_sample *sample,
292 				   struct machine *machine)
293 {
294 	int err;
295 
296 	err = perf_event__process_mmap(tool, event, sample, machine);
297 	perf_event__repipe(tool, event, sample, machine);
298 
299 	return err;
300 }
301 
302 #ifdef HAVE_JITDUMP
303 static int perf_event__jit_repipe_mmap(struct perf_tool *tool,
304 				       union perf_event *event,
305 				       struct perf_sample *sample,
306 				       struct machine *machine)
307 {
308 	struct perf_inject *inject = container_of(tool, struct perf_inject, tool);
309 	u64 n = 0;
310 	int ret;
311 
312 	/*
313 	 * if jit marker, then inject jit mmaps and generate ELF images
314 	 */
315 	ret = jit_process(inject->session, &inject->output, machine,
316 			  event->mmap.filename, event->mmap.pid, event->mmap.tid, &n);
317 	if (ret < 0)
318 		return ret;
319 	if (ret) {
320 		inject->bytes_written += n;
321 		return 0;
322 	}
323 	return perf_event__repipe_mmap(tool, event, sample, machine);
324 }
325 #endif
326 
327 static struct dso *findnew_dso(int pid, int tid, const char *filename,
328 			       struct dso_id *id, struct machine *machine)
329 {
330 	struct thread *thread;
331 	struct nsinfo *nsi = NULL;
332 	struct nsinfo *nnsi;
333 	struct dso *dso;
334 	bool vdso;
335 
336 	thread = machine__findnew_thread(machine, pid, tid);
337 	if (thread == NULL) {
338 		pr_err("cannot find or create a task %d/%d.\n", tid, pid);
339 		return NULL;
340 	}
341 
342 	vdso = is_vdso_map(filename);
343 	nsi = nsinfo__get(thread->nsinfo);
344 
345 	if (vdso) {
346 		/* The vdso maps are always on the host and not the
347 		 * container.  Ensure that we don't use setns to look
348 		 * them up.
349 		 */
350 		nnsi = nsinfo__copy(nsi);
351 		if (nnsi) {
352 			nsinfo__put(nsi);
353 			nnsi->need_setns = false;
354 			nsi = nnsi;
355 		}
356 		dso = machine__findnew_vdso(machine, thread);
357 	} else {
358 		dso = machine__findnew_dso_id(machine, filename, id);
359 	}
360 
361 	if (dso)
362 		dso->nsinfo = nsi;
363 	else
364 		nsinfo__put(nsi);
365 
366 	thread__put(thread);
367 	return dso;
368 }
369 
370 static int perf_event__repipe_buildid_mmap(struct perf_tool *tool,
371 					   union perf_event *event,
372 					   struct perf_sample *sample,
373 					   struct machine *machine)
374 {
375 	struct dso *dso;
376 
377 	dso = findnew_dso(event->mmap.pid, event->mmap.tid,
378 			  event->mmap.filename, NULL, machine);
379 
380 	if (dso && !dso->hit) {
381 		dso->hit = 1;
382 		dso__inject_build_id(dso, tool, machine, sample->cpumode, 0);
383 		dso__put(dso);
384 	}
385 
386 	return perf_event__repipe(tool, event, sample, machine);
387 }
388 
389 static int perf_event__repipe_mmap2(struct perf_tool *tool,
390 				   union perf_event *event,
391 				   struct perf_sample *sample,
392 				   struct machine *machine)
393 {
394 	int err;
395 
396 	err = perf_event__process_mmap2(tool, event, sample, machine);
397 	perf_event__repipe(tool, event, sample, machine);
398 
399 	return err;
400 }
401 
402 #ifdef HAVE_JITDUMP
403 static int perf_event__jit_repipe_mmap2(struct perf_tool *tool,
404 					union perf_event *event,
405 					struct perf_sample *sample,
406 					struct machine *machine)
407 {
408 	struct perf_inject *inject = container_of(tool, struct perf_inject, tool);
409 	u64 n = 0;
410 	int ret;
411 
412 	/*
413 	 * if jit marker, then inject jit mmaps and generate ELF images
414 	 */
415 	ret = jit_process(inject->session, &inject->output, machine,
416 			  event->mmap2.filename, event->mmap2.pid, event->mmap2.tid, &n);
417 	if (ret < 0)
418 		return ret;
419 	if (ret) {
420 		inject->bytes_written += n;
421 		return 0;
422 	}
423 	return perf_event__repipe_mmap2(tool, event, sample, machine);
424 }
425 #endif
426 
427 static int perf_event__repipe_buildid_mmap2(struct perf_tool *tool,
428 					    union perf_event *event,
429 					    struct perf_sample *sample,
430 					    struct machine *machine)
431 {
432 	struct dso_id dso_id = {
433 		.maj = event->mmap2.maj,
434 		.min = event->mmap2.min,
435 		.ino = event->mmap2.ino,
436 		.ino_generation = event->mmap2.ino_generation,
437 	};
438 	struct dso *dso;
439 
440 	dso = findnew_dso(event->mmap2.pid, event->mmap2.tid,
441 			  event->mmap2.filename, &dso_id, machine);
442 
443 	if (dso && !dso->hit) {
444 		dso->hit = 1;
445 		dso__inject_build_id(dso, tool, machine, sample->cpumode,
446 				     event->mmap2.flags);
447 		dso__put(dso);
448 	}
449 
450 	perf_event__repipe(tool, event, sample, machine);
451 
452 	return 0;
453 }
454 
455 static int perf_event__repipe_fork(struct perf_tool *tool,
456 				   union perf_event *event,
457 				   struct perf_sample *sample,
458 				   struct machine *machine)
459 {
460 	int err;
461 
462 	err = perf_event__process_fork(tool, event, sample, machine);
463 	perf_event__repipe(tool, event, sample, machine);
464 
465 	return err;
466 }
467 
468 static int perf_event__repipe_comm(struct perf_tool *tool,
469 				   union perf_event *event,
470 				   struct perf_sample *sample,
471 				   struct machine *machine)
472 {
473 	int err;
474 
475 	err = perf_event__process_comm(tool, event, sample, machine);
476 	perf_event__repipe(tool, event, sample, machine);
477 
478 	return err;
479 }
480 
481 static int perf_event__repipe_namespaces(struct perf_tool *tool,
482 					 union perf_event *event,
483 					 struct perf_sample *sample,
484 					 struct machine *machine)
485 {
486 	int err = perf_event__process_namespaces(tool, event, sample, machine);
487 
488 	perf_event__repipe(tool, event, sample, machine);
489 
490 	return err;
491 }
492 
493 static int perf_event__repipe_exit(struct perf_tool *tool,
494 				   union perf_event *event,
495 				   struct perf_sample *sample,
496 				   struct machine *machine)
497 {
498 	int err;
499 
500 	err = perf_event__process_exit(tool, event, sample, machine);
501 	perf_event__repipe(tool, event, sample, machine);
502 
503 	return err;
504 }
505 
506 static int perf_event__repipe_tracing_data(struct perf_session *session,
507 					   union perf_event *event)
508 {
509 	int err;
510 
511 	perf_event__repipe_synth(session->tool, event);
512 	err = perf_event__process_tracing_data(session, event);
513 
514 	return err;
515 }
516 
517 static int dso__read_build_id(struct dso *dso)
518 {
519 	struct nscookie nsc;
520 
521 	if (dso->has_build_id)
522 		return 0;
523 
524 	nsinfo__mountns_enter(dso->nsinfo, &nsc);
525 	if (filename__read_build_id(dso->long_name, &dso->bid) > 0)
526 		dso->has_build_id = true;
527 	nsinfo__mountns_exit(&nsc);
528 
529 	return dso->has_build_id ? 0 : -1;
530 }
531 
532 static int dso__inject_build_id(struct dso *dso, struct perf_tool *tool,
533 				struct machine *machine, u8 cpumode, u32 flags)
534 {
535 	int err;
536 
537 	if (is_anon_memory(dso->long_name) || flags & MAP_HUGETLB)
538 		return 0;
539 	if (is_no_dso_memory(dso->long_name))
540 		return 0;
541 
542 	if (dso__read_build_id(dso) < 0) {
543 		pr_debug("no build_id found for %s\n", dso->long_name);
544 		return -1;
545 	}
546 
547 	err = perf_event__synthesize_build_id(tool, dso, cpumode,
548 					      perf_event__repipe, machine);
549 	if (err) {
550 		pr_err("Can't synthesize build_id event for %s\n", dso->long_name);
551 		return -1;
552 	}
553 
554 	return 0;
555 }
556 
557 int perf_event__inject_buildid(struct perf_tool *tool, union perf_event *event,
558 			       struct perf_sample *sample,
559 			       struct evsel *evsel __maybe_unused,
560 			       struct machine *machine)
561 {
562 	struct addr_location al;
563 	struct thread *thread;
564 
565 	thread = machine__findnew_thread(machine, sample->pid, sample->tid);
566 	if (thread == NULL) {
567 		pr_err("problem processing %d event, skipping it.\n",
568 		       event->header.type);
569 		goto repipe;
570 	}
571 
572 	if (thread__find_map(thread, sample->cpumode, sample->ip, &al)) {
573 		if (!al.map->dso->hit) {
574 			al.map->dso->hit = 1;
575 			dso__inject_build_id(al.map->dso, tool, machine,
576 					     sample->cpumode, al.map->flags);
577 		}
578 	}
579 
580 	thread__put(thread);
581 repipe:
582 	perf_event__repipe(tool, event, sample, machine);
583 	return 0;
584 }
585 
586 static int perf_inject__sched_process_exit(struct perf_tool *tool,
587 					   union perf_event *event __maybe_unused,
588 					   struct perf_sample *sample,
589 					   struct evsel *evsel __maybe_unused,
590 					   struct machine *machine __maybe_unused)
591 {
592 	struct perf_inject *inject = container_of(tool, struct perf_inject, tool);
593 	struct event_entry *ent;
594 
595 	list_for_each_entry(ent, &inject->samples, node) {
596 		if (sample->tid == ent->tid) {
597 			list_del_init(&ent->node);
598 			free(ent);
599 			break;
600 		}
601 	}
602 
603 	return 0;
604 }
605 
606 static int perf_inject__sched_switch(struct perf_tool *tool,
607 				     union perf_event *event,
608 				     struct perf_sample *sample,
609 				     struct evsel *evsel,
610 				     struct machine *machine)
611 {
612 	struct perf_inject *inject = container_of(tool, struct perf_inject, tool);
613 	struct event_entry *ent;
614 
615 	perf_inject__sched_process_exit(tool, event, sample, evsel, machine);
616 
617 	ent = malloc(event->header.size + sizeof(struct event_entry));
618 	if (ent == NULL) {
619 		color_fprintf(stderr, PERF_COLOR_RED,
620 			     "Not enough memory to process sched switch event!");
621 		return -1;
622 	}
623 
624 	ent->tid = sample->tid;
625 	memcpy(&ent->event, event, event->header.size);
626 	list_add(&ent->node, &inject->samples);
627 	return 0;
628 }
629 
630 static int perf_inject__sched_stat(struct perf_tool *tool,
631 				   union perf_event *event __maybe_unused,
632 				   struct perf_sample *sample,
633 				   struct evsel *evsel,
634 				   struct machine *machine)
635 {
636 	struct event_entry *ent;
637 	union perf_event *event_sw;
638 	struct perf_sample sample_sw;
639 	struct perf_inject *inject = container_of(tool, struct perf_inject, tool);
640 	u32 pid = evsel__intval(evsel, sample, "pid");
641 
642 	list_for_each_entry(ent, &inject->samples, node) {
643 		if (pid == ent->tid)
644 			goto found;
645 	}
646 
647 	return 0;
648 found:
649 	event_sw = &ent->event[0];
650 	evsel__parse_sample(evsel, event_sw, &sample_sw);
651 
652 	sample_sw.period = sample->period;
653 	sample_sw.time	 = sample->time;
654 	perf_event__synthesize_sample(event_sw, evsel->core.attr.sample_type,
655 				      evsel->core.attr.read_format, &sample_sw);
656 	build_id__mark_dso_hit(tool, event_sw, &sample_sw, evsel, machine);
657 	return perf_event__repipe(tool, event_sw, &sample_sw, machine);
658 }
659 
660 static void sig_handler(int sig __maybe_unused)
661 {
662 	session_done = 1;
663 }
664 
665 static int evsel__check_stype(struct evsel *evsel, u64 sample_type, const char *sample_msg)
666 {
667 	struct perf_event_attr *attr = &evsel->core.attr;
668 	const char *name = evsel__name(evsel);
669 
670 	if (!(attr->sample_type & sample_type)) {
671 		pr_err("Samples for %s event do not have %s attribute set.",
672 			name, sample_msg);
673 		return -EINVAL;
674 	}
675 
676 	return 0;
677 }
678 
679 static int drop_sample(struct perf_tool *tool __maybe_unused,
680 		       union perf_event *event __maybe_unused,
681 		       struct perf_sample *sample __maybe_unused,
682 		       struct evsel *evsel __maybe_unused,
683 		       struct machine *machine __maybe_unused)
684 {
685 	return 0;
686 }
687 
688 static void strip_init(struct perf_inject *inject)
689 {
690 	struct evlist *evlist = inject->session->evlist;
691 	struct evsel *evsel;
692 
693 	inject->tool.context_switch = perf_event__drop;
694 
695 	evlist__for_each_entry(evlist, evsel)
696 		evsel->handler = drop_sample;
697 }
698 
699 static int __cmd_inject(struct perf_inject *inject)
700 {
701 	int ret = -EINVAL;
702 	struct perf_session *session = inject->session;
703 	struct perf_data *data_out = &inject->output;
704 	int fd = perf_data__fd(data_out);
705 	u64 output_data_offset;
706 
707 	signal(SIGINT, sig_handler);
708 
709 	if (inject->build_ids || inject->sched_stat ||
710 	    inject->itrace_synth_opts.set || inject->build_id_all) {
711 		inject->tool.mmap	  = perf_event__repipe_mmap;
712 		inject->tool.mmap2	  = perf_event__repipe_mmap2;
713 		inject->tool.fork	  = perf_event__repipe_fork;
714 		inject->tool.tracing_data = perf_event__repipe_tracing_data;
715 	}
716 
717 	output_data_offset = session->header.data_offset;
718 
719 	if (inject->build_id_all) {
720 		inject->tool.mmap	  = perf_event__repipe_buildid_mmap;
721 		inject->tool.mmap2	  = perf_event__repipe_buildid_mmap2;
722 	} else if (inject->build_ids) {
723 		inject->tool.sample = perf_event__inject_buildid;
724 	} else if (inject->sched_stat) {
725 		struct evsel *evsel;
726 
727 		evlist__for_each_entry(session->evlist, evsel) {
728 			const char *name = evsel__name(evsel);
729 
730 			if (!strcmp(name, "sched:sched_switch")) {
731 				if (evsel__check_stype(evsel, PERF_SAMPLE_TID, "TID"))
732 					return -EINVAL;
733 
734 				evsel->handler = perf_inject__sched_switch;
735 			} else if (!strcmp(name, "sched:sched_process_exit"))
736 				evsel->handler = perf_inject__sched_process_exit;
737 			else if (!strncmp(name, "sched:sched_stat_", 17))
738 				evsel->handler = perf_inject__sched_stat;
739 		}
740 	} else if (inject->itrace_synth_opts.set) {
741 		session->itrace_synth_opts = &inject->itrace_synth_opts;
742 		inject->itrace_synth_opts.inject = true;
743 		inject->tool.comm	    = perf_event__repipe_comm;
744 		inject->tool.namespaces	    = perf_event__repipe_namespaces;
745 		inject->tool.exit	    = perf_event__repipe_exit;
746 		inject->tool.id_index	    = perf_event__process_id_index;
747 		inject->tool.auxtrace_info  = perf_event__process_auxtrace_info;
748 		inject->tool.auxtrace	    = perf_event__process_auxtrace;
749 		inject->tool.aux	    = perf_event__drop_aux;
750 		inject->tool.itrace_start   = perf_event__drop_aux,
751 		inject->tool.ordered_events = true;
752 		inject->tool.ordering_requires_timestamps = true;
753 		/* Allow space in the header for new attributes */
754 		output_data_offset = 4096;
755 		if (inject->strip)
756 			strip_init(inject);
757 	}
758 
759 	if (!inject->itrace_synth_opts.set)
760 		auxtrace_index__free(&session->auxtrace_index);
761 
762 	if (!data_out->is_pipe)
763 		lseek(fd, output_data_offset, SEEK_SET);
764 
765 	ret = perf_session__process_events(session);
766 	if (ret)
767 		return ret;
768 
769 	if (!data_out->is_pipe) {
770 		if (inject->build_ids)
771 			perf_header__set_feat(&session->header,
772 					      HEADER_BUILD_ID);
773 		/*
774 		 * Keep all buildids when there is unprocessed AUX data because
775 		 * it is not known which ones the AUX trace hits.
776 		 */
777 		if (perf_header__has_feat(&session->header, HEADER_BUILD_ID) &&
778 		    inject->have_auxtrace && !inject->itrace_synth_opts.set)
779 			dsos__hit_all(session);
780 		/*
781 		 * The AUX areas have been removed and replaced with
782 		 * synthesized hardware events, so clear the feature flag.
783 		 */
784 		if (inject->itrace_synth_opts.set) {
785 			perf_header__clear_feat(&session->header,
786 						HEADER_AUXTRACE);
787 			if (inject->itrace_synth_opts.last_branch ||
788 			    inject->itrace_synth_opts.add_last_branch)
789 				perf_header__set_feat(&session->header,
790 						      HEADER_BRANCH_STACK);
791 		}
792 		session->header.data_offset = output_data_offset;
793 		session->header.data_size = inject->bytes_written;
794 		perf_session__write_header(session, session->evlist, fd, true);
795 	}
796 
797 	return ret;
798 }
799 
800 int cmd_inject(int argc, const char **argv)
801 {
802 	struct perf_inject inject = {
803 		.tool = {
804 			.sample		= perf_event__repipe_sample,
805 			.read		= perf_event__repipe_sample,
806 			.mmap		= perf_event__repipe,
807 			.mmap2		= perf_event__repipe,
808 			.comm		= perf_event__repipe,
809 			.namespaces	= perf_event__repipe,
810 			.cgroup		= perf_event__repipe,
811 			.fork		= perf_event__repipe,
812 			.exit		= perf_event__repipe,
813 			.lost		= perf_event__repipe,
814 			.lost_samples	= perf_event__repipe,
815 			.aux		= perf_event__repipe,
816 			.itrace_start	= perf_event__repipe,
817 			.context_switch	= perf_event__repipe,
818 			.throttle	= perf_event__repipe,
819 			.unthrottle	= perf_event__repipe,
820 			.ksymbol	= perf_event__repipe,
821 			.bpf		= perf_event__repipe,
822 			.text_poke	= perf_event__repipe,
823 			.attr		= perf_event__repipe_attr,
824 			.event_update	= perf_event__repipe_event_update,
825 			.tracing_data	= perf_event__repipe_op2_synth,
826 			.finished_round	= perf_event__repipe_oe_synth,
827 			.build_id	= perf_event__repipe_op2_synth,
828 			.id_index	= perf_event__repipe_op2_synth,
829 			.auxtrace_info	= perf_event__repipe_op2_synth,
830 			.auxtrace_error	= perf_event__repipe_op2_synth,
831 			.time_conv	= perf_event__repipe_op2_synth,
832 			.thread_map	= perf_event__repipe_op2_synth,
833 			.cpu_map	= perf_event__repipe_op2_synth,
834 			.stat_config	= perf_event__repipe_op2_synth,
835 			.stat		= perf_event__repipe_op2_synth,
836 			.stat_round	= perf_event__repipe_op2_synth,
837 			.feature	= perf_event__repipe_op2_synth,
838 			.compressed	= perf_event__repipe_op4_synth,
839 			.auxtrace	= perf_event__repipe_auxtrace,
840 		},
841 		.input_name  = "-",
842 		.samples = LIST_HEAD_INIT(inject.samples),
843 		.output = {
844 			.path = "-",
845 			.mode = PERF_DATA_MODE_WRITE,
846 			.use_stdio = true,
847 		},
848 	};
849 	struct perf_data data = {
850 		.mode = PERF_DATA_MODE_READ,
851 		.use_stdio = true,
852 	};
853 	int ret;
854 
855 	struct option options[] = {
856 		OPT_BOOLEAN('b', "build-ids", &inject.build_ids,
857 			    "Inject build-ids into the output stream"),
858 		OPT_BOOLEAN(0, "buildid-all", &inject.build_id_all,
859 			    "Inject build-ids of all DSOs into the output stream"),
860 		OPT_STRING('i', "input", &inject.input_name, "file",
861 			   "input file name"),
862 		OPT_STRING('o', "output", &inject.output.path, "file",
863 			   "output file name"),
864 		OPT_BOOLEAN('s', "sched-stat", &inject.sched_stat,
865 			    "Merge sched-stat and sched-switch for getting events "
866 			    "where and how long tasks slept"),
867 #ifdef HAVE_JITDUMP
868 		OPT_BOOLEAN('j', "jit", &inject.jit_mode, "merge jitdump files into perf.data file"),
869 #endif
870 		OPT_INCR('v', "verbose", &verbose,
871 			 "be more verbose (show build ids, etc)"),
872 		OPT_STRING(0, "kallsyms", &symbol_conf.kallsyms_name, "file",
873 			   "kallsyms pathname"),
874 		OPT_BOOLEAN('f', "force", &data.force, "don't complain, do it"),
875 		OPT_CALLBACK_OPTARG(0, "itrace", &inject.itrace_synth_opts,
876 				    NULL, "opts", "Instruction Tracing options\n"
877 				    ITRACE_HELP,
878 				    itrace_parse_synth_opts),
879 		OPT_BOOLEAN(0, "strip", &inject.strip,
880 			    "strip non-synthesized events (use with --itrace)"),
881 		OPT_END()
882 	};
883 	const char * const inject_usage[] = {
884 		"perf inject [<options>]",
885 		NULL
886 	};
887 #ifndef HAVE_JITDUMP
888 	set_option_nobuild(options, 'j', "jit", "NO_LIBELF=1", true);
889 #endif
890 	argc = parse_options(argc, argv, options, inject_usage, 0);
891 
892 	/*
893 	 * Any (unrecognized) arguments left?
894 	 */
895 	if (argc)
896 		usage_with_options(inject_usage, options);
897 
898 	if (inject.strip && !inject.itrace_synth_opts.set) {
899 		pr_err("--strip option requires --itrace option\n");
900 		return -1;
901 	}
902 
903 	if (perf_data__open(&inject.output)) {
904 		perror("failed to create output file");
905 		return -1;
906 	}
907 
908 	data.path = inject.input_name;
909 	inject.session = perf_session__new(&data, true, &inject.tool);
910 	if (IS_ERR(inject.session))
911 		return PTR_ERR(inject.session);
912 
913 	if (zstd_init(&(inject.session->zstd_data), 0) < 0)
914 		pr_warning("Decompression initialization failed.\n");
915 
916 	if (inject.build_ids && !inject.build_id_all) {
917 		/*
918 		 * to make sure the mmap records are ordered correctly
919 		 * and so that the correct especially due to jitted code
920 		 * mmaps. We cannot generate the buildid hit list and
921 		 * inject the jit mmaps at the same time for now.
922 		 */
923 		inject.tool.ordered_events = true;
924 		inject.tool.ordering_requires_timestamps = true;
925 	}
926 
927 	if (inject.sched_stat) {
928 		inject.tool.ordered_events = true;
929 	}
930 
931 #ifdef HAVE_JITDUMP
932 	if (inject.jit_mode) {
933 		inject.tool.mmap2	   = perf_event__jit_repipe_mmap2;
934 		inject.tool.mmap	   = perf_event__jit_repipe_mmap;
935 		inject.tool.ordered_events = true;
936 		inject.tool.ordering_requires_timestamps = true;
937 		/*
938 		 * JIT MMAP injection injects all MMAP events in one go, so it
939 		 * does not obey finished_round semantics.
940 		 */
941 		inject.tool.finished_round = perf_event__drop_oe;
942 	}
943 #endif
944 	ret = symbol__init(&inject.session->header.env);
945 	if (ret < 0)
946 		goto out_delete;
947 
948 	ret = __cmd_inject(&inject);
949 
950 out_delete:
951 	zstd_fini(&(inject.session->zstd_data));
952 	perf_session__delete(inject.session);
953 	return ret;
954 }
955