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