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