tools/perf/builtin-inject.c

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