tools/perf/builtin-inject.c

   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
  26 struct 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
  42 struct event_entry {
  43         struct list_head node;
  44         u32              tid;
  45         union perf_event event[0];
  46 };
  47
  48 static 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
  60 static 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
  69 static 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
  77 static 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
  85 static 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
  93 static 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
 113 static 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
 132 static 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
 172 static s64
 173 perf_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
 183 static 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
 191 static 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
 199 static 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
 212 typedef 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
 218 static 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
 234 static 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
 248 static 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
 272 static 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
 286 static 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
 310 static 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
 323 static 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
 336 static int perf_event__repipe_namespaces(struct perf_tool *tool,
 337                                          union perf_event *event,
 338                                          struct perf_sample *sample,
 339                                          struct machine *machine)
 340 {
 341         int err = perf_event__process_namespaces(tool, event, sample, machine);
 342
 343         perf_event__repipe(tool, event, sample, machine);
 344
 345         return err;
 346 }
 347
 348 static int perf_event__repipe_exit(struct perf_tool *tool,
 349                                    union perf_event *event,
 350                                    struct perf_sample *sample,
 351                                    struct machine *machine)
 352 {
 353         int err;
 354
 355         err = perf_event__process_exit(tool, event, sample, machine);
 356         perf_event__repipe(tool, event, sample, machine);
 357
 358         return err;
 359 }
 360
 361 static int perf_event__repipe_tracing_data(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_tracing_data(tool, event, session);
 369
 370         return err;
 371 }
 372
 373 static int perf_event__repipe_id_index(struct perf_tool *tool,
 374                                        union perf_event *event,
 375                                        struct perf_session *session)
 376 {
 377         int err;
 378
 379         perf_event__repipe_synth(tool, event);
 380         err = perf_event__process_id_index(tool, event, session);
 381
 382         return err;
 383 }
 384
 385 static int dso__read_build_id(struct dso *dso)
 386 {
 387         if (dso->has_build_id)
 388                 return 0;
 389
 390         if (filename__read_build_id(dso->long_name, dso->build_id,
 391                                     sizeof(dso->build_id)) > 0) {
 392                 dso->has_build_id = true;
 393                 return 0;
 394         }
 395
 396         return -1;
 397 }
 398
 399 static int dso__inject_build_id(struct dso *dso, struct perf_tool *tool,
 400                                 struct machine *machine)
 401 {
 402         u16 misc = PERF_RECORD_MISC_USER;
 403         int err;
 404
 405         if (dso__read_build_id(dso) < 0) {
 406                 pr_debug("no build_id found for %s\n", dso->long_name);
 407                 return -1;
 408         }
 409
 410         if (dso->kernel)
 411                 misc = PERF_RECORD_MISC_KERNEL;
 412
 413         err = perf_event__synthesize_build_id(tool, dso, misc, perf_event__repipe,
 414                                               machine);
 415         if (err) {
 416                 pr_err("Can't synthesize build_id event for %s\n", dso->long_name);
 417                 return -1;
 418         }
 419
 420         return 0;
 421 }
 422
 423 static int perf_event__inject_buildid(struct perf_tool *tool,
 424                                       union perf_event *event,
 425                                       struct perf_sample *sample,
 426                                       struct perf_evsel *evsel __maybe_unused,
 427                                       struct machine *machine)
 428 {
 429         struct addr_location al;
 430         struct thread *thread;
 431
 432         thread = machine__findnew_thread(machine, sample->pid, sample->tid);
 433         if (thread == NULL) {
 434                 pr_err("problem processing %d event, skipping it.\n",
 435                        event->header.type);
 436                 goto repipe;
 437         }
 438
 439         thread__find_addr_map(thread, sample->cpumode, MAP__FUNCTION, sample->ip, &al);
 440
 441         if (al.map != NULL) {
 442                 if (!al.map->dso->hit) {
 443                         al.map->dso->hit = 1;
 444                         if (map__load(al.map) >= 0) {
 445                                 dso__inject_build_id(al.map->dso, tool, machine);
 446                                 /*
 447                                  * If this fails, too bad, let the other side
 448                                  * account this as unresolved.
 449                                  */
 450                         } else {
 451 #ifdef HAVE_LIBELF_SUPPORT
 452                                 pr_warning("no symbols found in %s, maybe "
 453                                            "install a debug package?\n",
 454                                            al.map->dso->long_name);
 455 #endif
 456                         }
 457                 }
 458         }
 459
 460         thread__put(thread);
 461 repipe:
 462         perf_event__repipe(tool, event, sample, machine);
 463         return 0;
 464 }
 465
 466 static int perf_inject__sched_process_exit(struct perf_tool *tool,
 467                                            union perf_event *event __maybe_unused,
 468                                            struct perf_sample *sample,
 469                                            struct perf_evsel *evsel __maybe_unused,
 470                                            struct machine *machine __maybe_unused)
 471 {
 472         struct perf_inject *inject = container_of(tool, struct perf_inject, tool);
 473         struct event_entry *ent;
 474
 475         list_for_each_entry(ent, &inject->samples, node) {
 476                 if (sample->tid == ent->tid) {
 477                         list_del_init(&ent->node);
 478                         free(ent);
 479                         break;
 480                 }
 481         }
 482
 483         return 0;
 484 }
 485
 486 static int perf_inject__sched_switch(struct perf_tool *tool,
 487                                      union perf_event *event,
 488                                      struct perf_sample *sample,
 489                                      struct perf_evsel *evsel,
 490                                      struct machine *machine)
 491 {
 492         struct perf_inject *inject = container_of(tool, struct perf_inject, tool);
 493         struct event_entry *ent;
 494
 495         perf_inject__sched_process_exit(tool, event, sample, evsel, machine);
 496
 497         ent = malloc(event->header.size + sizeof(struct event_entry));
 498         if (ent == NULL) {
 499                 color_fprintf(stderr, PERF_COLOR_RED,
 500                              "Not enough memory to process sched switch event!");
 501                 return -1;
 502         }
 503
 504         ent->tid = sample->tid;
 505         memcpy(&ent->event, event, event->header.size);
 506         list_add(&ent->node, &inject->samples);
 507         return 0;
 508 }
 509
 510 static int perf_inject__sched_stat(struct perf_tool *tool,
 511                                    union perf_event *event __maybe_unused,
 512                                    struct perf_sample *sample,
 513                                    struct perf_evsel *evsel,
 514                                    struct machine *machine)
 515 {
 516         struct event_entry *ent;
 517         union perf_event *event_sw;
 518         struct perf_sample sample_sw;
 519         struct perf_inject *inject = container_of(tool, struct perf_inject, tool);
 520         u32 pid = perf_evsel__intval(evsel, sample, "pid");
 521
 522         list_for_each_entry(ent, &inject->samples, node) {
 523                 if (pid == ent->tid)
 524                         goto found;
 525         }
 526
 527         return 0;
 528 found:
 529         event_sw = &ent->event[0];
 530         perf_evsel__parse_sample(evsel, event_sw, &sample_sw);
 531
 532         sample_sw.period = sample->period;
 533         sample_sw.time   = sample->time;
 534         perf_event__synthesize_sample(event_sw, evsel->attr.sample_type,
 535                                       evsel->attr.read_format, &sample_sw,
 536                                       false);
 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
 541 static void sig_handler(int sig __maybe_unused)
 542 {
 543         session_done = 1;
 544 }
 545
 546 static int perf_evsel__check_stype(struct perf_evsel *evsel,
 547                                    u64 sample_type, const char *sample_msg)
 548 {
 549         struct perf_event_attr *attr = &evsel->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
 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 perf_evsel *evsel __maybe_unused,
 565                        struct machine *machine __maybe_unused)
 566 {
 567         return 0;
 568 }
 569
 570 static void strip_init(struct perf_inject *inject)
 571 {
 572         struct perf_evlist *evlist = inject->session->evlist;
 573         struct perf_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
 581 static bool has_tracking(struct perf_evsel *evsel)
 582 {
 583         return evsel->attr.mmap || evsel->attr.mmap2 || evsel->attr.comm ||
 584                evsel->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  */
 595 static bool ok_to_remove(struct perf_evlist *evlist,
 596                          struct perf_evsel *evsel_to_remove)
 597 {
 598         struct perf_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->attr.sample_type & COMPAT_MASK) ==
 609                             (evsel_to_remove->attr.sample_type & COMPAT_MASK))
 610                                 ok = true;
 611                 }
 612         }
 613
 614         return ok && cnt == 1;
 615 }
 616
 617 static void strip_fini(struct perf_inject *inject)
 618 {
 619         struct perf_evlist *evlist = inject->session->evlist;
 620         struct perf_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                         perf_evlist__remove(evlist, evsel);
 628                         perf_evsel__delete(evsel);
 629                 }
 630         }
 631 }
 632
 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_file *file_out = &inject->output;
 638         int fd = perf_data_file__fd(file_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 perf_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 (!file_out->is_pipe)
 694                 lseek(fd, output_data_offset, SEEK_SET);
 695
 696         ret = perf_session__process_events(session);
 697
 698         if (!file_out->is_pipe) {
 699                 if (inject->build_ids)
 700                         perf_header__set_feat(&session->header,
 701                                               HEADER_BUILD_ID);
 702                 /*
 703                  * Keep all buildids when there is unprocessed AUX data because
 704                  * it is not known which ones the AUX trace hits.
 705                  */
 706                 if (perf_header__has_feat(&session->header, HEADER_BUILD_ID) &&
 707                     inject->have_auxtrace && !inject->itrace_synth_opts.set)
 708                         dsos__hit_all(session);
 709                 /*
 710                  * The AUX areas have been removed and replaced with
 711                  * synthesized hardware events, so clear the feature flag and
 712                  * remove the evsel.
 713                  */
 714                 if (inject->itrace_synth_opts.set) {
 715                         struct perf_evsel *evsel;
 716
 717                         perf_header__clear_feat(&session->header,
 718                                                 HEADER_AUXTRACE);
 719                         if (inject->itrace_synth_opts.last_branch)
 720                                 perf_header__set_feat(&session->header,
 721                                                       HEADER_BRANCH_STACK);
 722                         evsel = perf_evlist__id2evsel_strict(session->evlist,
 723                                                              inject->aux_id);
 724                         if (evsel) {
 725                                 pr_debug("Deleting %s\n",
 726                                          perf_evsel__name(evsel));
 727                                 perf_evlist__remove(session->evlist, evsel);
 728                                 perf_evsel__delete(evsel);
 729                         }
 730                         if (inject->strip)
 731                                 strip_fini(inject);
 732                 }
 733                 session->header.data_offset = output_data_offset;
 734                 session->header.data_size = inject->bytes_written;
 735                 perf_session__write_header(session, session->evlist, fd, true);
 736         }
 737
 738         return ret;
 739 }
 740
 741 int cmd_inject(int argc, const char **argv, const char *prefix __maybe_unused)
 742 {
 743         struct perf_inject inject = {
 744                 .tool = {
 745                         .sample         = perf_event__repipe_sample,
 746                         .mmap           = perf_event__repipe,
 747                         .mmap2          = perf_event__repipe,
 748                         .comm           = perf_event__repipe,
 749                         .fork           = perf_event__repipe,
 750                         .exit           = perf_event__repipe,
 751                         .lost           = perf_event__repipe,
 752                         .lost_samples   = perf_event__repipe,
 753                         .aux            = perf_event__repipe,
 754                         .itrace_start   = perf_event__repipe,
 755                         .context_switch = perf_event__repipe,
 756                         .read           = perf_event__repipe_sample,
 757                         .throttle       = perf_event__repipe,
 758                         .unthrottle     = perf_event__repipe,
 759                         .attr           = perf_event__repipe_attr,
 760                         .tracing_data   = perf_event__repipe_op2_synth,
 761                         .auxtrace_info  = perf_event__repipe_op2_synth,
 762                         .auxtrace       = perf_event__repipe_auxtrace,
 763                         .auxtrace_error = perf_event__repipe_op2_synth,
 764                         .time_conv      = perf_event__repipe_op2_synth,
 765                         .finished_round = perf_event__repipe_oe_synth,
 766                         .build_id       = perf_event__repipe_op2_synth,
 767                         .id_index       = perf_event__repipe_op2_synth,
 768                 },
 769                 .input_name  = "-",
 770                 .samples = LIST_HEAD_INIT(inject.samples),
 771                 .output = {
 772                         .path = "-",
 773                         .mode = PERF_DATA_MODE_WRITE,
 774                 },
 775         };
 776         struct perf_data_file file = {
 777                 .mode = PERF_DATA_MODE_READ,
 778         };
 779         int ret;
 780
 781         struct option options[] = {
 782                 OPT_BOOLEAN('b', "build-ids", &inject.build_ids,
 783                             "Inject build-ids into the output stream"),
 784                 OPT_STRING('i', "input", &inject.input_name, "file",
 785                            "input file name"),
 786                 OPT_STRING('o', "output", &inject.output.path, "file",
 787                            "output file name"),
 788                 OPT_BOOLEAN('s', "sched-stat", &inject.sched_stat,
 789                             "Merge sched-stat and sched-switch for getting events "
 790                             "where and how long tasks slept"),
 791 #ifdef HAVE_JITDUMP
 792                 OPT_BOOLEAN('j', "jit", &inject.jit_mode, "merge jitdump files into perf.data file"),
 793 #endif
 794                 OPT_INCR('v', "verbose", &verbose,
 795                          "be more verbose (show build ids, etc)"),
 796                 OPT_STRING(0, "kallsyms", &symbol_conf.kallsyms_name, "file",
 797                            "kallsyms pathname"),
 798                 OPT_BOOLEAN('f', "force", &file.force, "don't complain, do it"),
 799                 OPT_CALLBACK_OPTARG(0, "itrace", &inject.itrace_synth_opts,
 800                                     NULL, "opts", "Instruction Tracing options",
 801                                     itrace_parse_synth_opts),
 802                 OPT_BOOLEAN(0, "strip", &inject.strip,
 803                             "strip non-synthesized events (use with --itrace)"),
 804                 OPT_END()
 805         };
 806         const char * const inject_usage[] = {
 807                 "perf inject [<options>]",
 808                 NULL
 809         };
 810 #ifndef HAVE_JITDUMP
 811         set_option_nobuild(options, 'j', "jit", "NO_LIBELF=1", true);
 812 #endif
 813         argc = parse_options(argc, argv, options, inject_usage, 0);
 814
 815         /*
 816          * Any (unrecognized) arguments left?
 817          */
 818         if (argc)
 819                 usage_with_options(inject_usage, options);
 820
 821         if (inject.strip && !inject.itrace_synth_opts.set) {
 822                 pr_err("--strip option requires --itrace option\n");
 823                 return -1;
 824         }
 825
 826         if (perf_data_file__open(&inject.output)) {
 827                 perror("failed to create output file");
 828                 return -1;
 829         }
 830
 831         inject.tool.ordered_events = inject.sched_stat;
 832
 833         file.path = inject.input_name;
 834         inject.session = perf_session__new(&file, true, &inject.tool);
 835         if (inject.session == NULL)
 836                 return -1;
 837
 838         if (inject.build_ids) {
 839                 /*
 840                  * to make sure the mmap records are ordered correctly
 841                  * and so that the correct especially due to jitted code
 842                  * mmaps. We cannot generate the buildid hit list and
 843                  * inject the jit mmaps at the same time for now.
 844                  */
 845                 inject.tool.ordered_events = true;
 846                 inject.tool.ordering_requires_timestamps = true;
 847         }
 848 #ifdef HAVE_JITDUMP
 849         if (inject.jit_mode) {
 850                 inject.tool.mmap2          = perf_event__jit_repipe_mmap2;
 851                 inject.tool.mmap           = perf_event__jit_repipe_mmap;
 852                 inject.tool.ordered_events = true;
 853                 inject.tool.ordering_requires_timestamps = true;
 854                 /*
 855                  * JIT MMAP injection injects all MMAP events in one go, so it
 856                  * does not obey finished_round semantics.
 857                  */
 858                 inject.tool.finished_round = perf_event__drop_oe;
 859         }
 860 #endif
 861         ret = symbol__init(&inject.session->header.env);
 862         if (ret < 0)
 863                 goto out_delete;
 864
 865         ret = __cmd_inject(&inject);
 866
 867 out_delete:
 868         perf_session__delete(inject.session);
 869         return ret;
 870 }