// SPDX-License-Identifier: GPL-2.0
/*
 * builtin-inject.c
 *
 * Builtin inject command: Examine the live mode (stdin) event stream
 * and repipe it to stdout while optionally injecting additional
 * events into it.
 */
#include "builtin.h"

#include "util/color.h"
#include "util/dso.h"
#include "util/vdso.h"
#include "util/evlist.h"
#include "util/evsel.h"
#include "util/map.h"
#include "util/session.h"
#include "util/tool.h"
#include "util/debug.h"
#include "util/build-id.h"
#include "util/data.h"
#include "util/auxtrace.h"
#include "util/jit.h"
#include "util/symbol.h"
#include "util/synthetic-events.h"
#include "util/thread.h"
#include "util/namespaces.h"

#include <linux/err.h>
#include <subcmd/parse-options.h>
#include <uapi/linux/mman.h> /* To get things like MAP_HUGETLB even on older libc headers */

#include <linux/list.h>
#include <linux/string.h>
#include <errno.h>
#include <signal.h>

struct perf_inject {
        struct perf_tool        tool;
        struct perf_session     *session;
        bool                    build_ids;
        bool                    build_id_all;
        bool                    sched_stat;
        bool                    have_auxtrace;
        bool                    strip;
        bool                    jit_mode;
        bool                    in_place_update;
        bool                    in_place_update_dry_run;
        const char              *input_name;
        struct perf_data        output;
        u64                     bytes_written;
        u64                     aux_id;
        struct list_head        samples;
        struct itrace_synth_opts itrace_synth_opts;
        char                    event_copy[PERF_SAMPLE_MAX_SIZE];
};

struct event_entry {
        struct list_head node;
        u32              tid;
        union perf_event event[];
};

static int dso__inject_build_id(struct dso *dso, struct perf_tool *tool,
                                struct machine *machine, u8 cpumode, u32 flags);

static int output_bytes(struct perf_inject *inject, void *buf, size_t sz)
{
        ssize_t size;

        size = perf_data__write(&inject->output, buf, sz);
        if (size < 0)
                return -errno;

        inject->bytes_written += size;
        return 0;
}

static int perf_event__repipe_synth(struct perf_tool *tool,
                                    union perf_event *event)
{
        struct perf_inject *inject = container_of(tool, struct perf_inject,
                                                  tool);

        return output_bytes(inject, event, event->header.size);
}

static int perf_event__repipe_oe_synth(struct perf_tool *tool,
                                       union perf_event *event,
                                       struct ordered_events *oe __maybe_unused)
{
        return perf_event__repipe_synth(tool, event);
}

#ifdef HAVE_JITDUMP
static int perf_event__drop_oe(struct perf_tool *tool __maybe_unused,
                               union perf_event *event __maybe_unused,
                               struct ordered_events *oe __maybe_unused)
{
        return 0;
}
#endif

static int perf_event__repipe_op2_synth(struct perf_session *session,
                                        union perf_event *event)
{
        return perf_event__repipe_synth(session->tool, event);
}

static int perf_event__repipe_op4_synth(struct perf_session *session,
                                        union perf_event *event,
                                        u64 data __maybe_unused)
{
        return perf_event__repipe_synth(session->tool, event);
}

static int perf_event__repipe_attr(struct perf_tool *tool,
                                   union perf_event *event,
                                   struct evlist **pevlist)
{
        struct perf_inject *inject = container_of(tool, struct perf_inject,
                                                  tool);
        int ret;

        ret = perf_event__process_attr(tool, event, pevlist);
        if (ret)
                return ret;

        if (!inject->output.is_pipe)
                return 0;

        return perf_event__repipe_synth(tool, event);
}

static int perf_event__repipe_event_update(struct perf_tool *tool,
                                           union perf_event *event,
                                           struct evlist **pevlist __maybe_unused)
{
        return perf_event__repipe_synth(tool, event);
}

#ifdef HAVE_AUXTRACE_SUPPORT

static int copy_bytes(struct perf_inject *inject, int fd, off_t size)
{
        char buf[4096];
        ssize_t ssz;
        int ret;

        while (size > 0) {
                ssz = read(fd, buf, min(size, (off_t)sizeof(buf)));
                if (ssz < 0)
                        return -errno;
                ret = output_bytes(inject, buf, ssz);
                if (ret)
                        return ret;
                size -= ssz;
        }

        return 0;
}

static s64 perf_event__repipe_auxtrace(struct perf_session *session,
                                       union perf_event *event)
{
        struct perf_tool *tool = session->tool;
        struct perf_inject *inject = container_of(tool, struct perf_inject,
                                                  tool);
        int ret;

        inject->have_auxtrace = true;

        if (!inject->output.is_pipe) {
                off_t offset;

                offset = lseek(inject->output.file.fd, 0, SEEK_CUR);
                if (offset == -1)
                        return -errno;
                ret = auxtrace_index__auxtrace_event(&session->auxtrace_index,
                                                     event, offset);
                if (ret < 0)
                        return ret;
        }

        if (perf_data__is_pipe(session->data) || !session->one_mmap) {
                ret = output_bytes(inject, event, event->header.size);
                if (ret < 0)
                        return ret;
                ret = copy_bytes(inject, perf_data__fd(session->data),
                                 event->auxtrace.size);
        } else {
                ret = output_bytes(inject, event,
                                   event->header.size + event->auxtrace.size);
        }
        if (ret < 0)
                return ret;

        return event->auxtrace.size;
}

#else

static s64
perf_event__repipe_auxtrace(struct perf_session *session __maybe_unused,
                            union perf_event *event __maybe_unused)
{
        pr_err("AUX area tracing not supported\n");
        return -EINVAL;
}

#endif

static int perf_event__repipe(struct perf_tool *tool,
                              union perf_event *event,
                              struct perf_sample *sample __maybe_unused,
                              struct machine *machine __maybe_unused)
{
        return perf_event__repipe_synth(tool, event);
}

static int perf_event__drop(struct perf_tool *tool __maybe_unused,
                            union perf_event *event __maybe_unused,
                            struct perf_sample *sample __maybe_unused,
                            struct machine *machine __maybe_unused)
{
        return 0;
}

static int perf_event__drop_aux(struct perf_tool *tool,
                                union perf_event *event __maybe_unused,
                                struct perf_sample *sample,
                                struct machine *machine __maybe_unused)
{
        struct perf_inject *inject = container_of(tool, struct perf_inject, tool);

        if (!inject->aux_id)
                inject->aux_id = sample->id;

        return 0;
}

static union perf_event *
perf_inject__cut_auxtrace_sample(struct perf_inject *inject,
                                 union perf_event *event,
                                 struct perf_sample *sample)
{
        size_t sz1 = sample->aux_sample.data - (void *)event;
        size_t sz2 = event->header.size - sample->aux_sample.size - sz1;
        union perf_event *ev = (union perf_event *)inject->event_copy;

        if (sz1 > event->header.size || sz2 > event->header.size ||
            sz1 + sz2 > event->header.size ||
            sz1 < sizeof(struct perf_event_header) + sizeof(u64))
                return event;

        memcpy(ev, event, sz1);
        memcpy((void *)ev + sz1, (void *)event + event->header.size - sz2, sz2);
        ev->header.size = sz1 + sz2;
        ((u64 *)((void *)ev + sz1))[-1] = 0;

        return ev;
}

typedef int (*inject_handler)(struct perf_tool *tool,
                              union perf_event *event,
                              struct perf_sample *sample,
                              struct evsel *evsel,
                              struct machine *machine);

static int perf_event__repipe_sample(struct perf_tool *tool,
                                     union perf_event *event,
                                     struct perf_sample *sample,
                                     struct evsel *evsel,
                                     struct machine *machine)
{
        struct perf_inject *inject = container_of(tool, struct perf_inject,
                                                  tool);

        if (evsel && evsel->handler) {
                inject_handler f = evsel->handler;
                return f(tool, event, sample, evsel, machine);
        }

        build_id__mark_dso_hit(tool, event, sample, evsel, machine);

        if (inject->itrace_synth_opts.set && sample->aux_sample.size)
                event = perf_inject__cut_auxtrace_sample(inject, event, sample);

        return perf_event__repipe_synth(tool, event);
}

static int perf_event__repipe_mmap(struct perf_tool *tool,
                                   union perf_event *event,
                                   struct perf_sample *sample,
                                   struct machine *machine)
{
        int err;

        err = perf_event__process_mmap(tool, event, sample, machine);
        perf_event__repipe(tool, event, sample, machine);

        return err;
}

#ifdef HAVE_JITDUMP
static int perf_event__jit_repipe_mmap(struct perf_tool *tool,
                                       union perf_event *event,
                                       struct perf_sample *sample,
                                       struct machine *machine)
{
        struct perf_inject *inject = container_of(tool, struct perf_inject, tool);
        u64 n = 0;
        int ret;

        /*
         * if jit marker, then inject jit mmaps and generate ELF images
         */
        ret = jit_process(inject->session, &inject->output, machine,
                          event->mmap.filename, event->mmap.pid, event->mmap.tid, &n);
        if (ret < 0)
                return ret;
        if (ret) {
                inject->bytes_written += n;
                return 0;
        }
        return perf_event__repipe_mmap(tool, event, sample, machine);
}
#endif

static struct dso *findnew_dso(int pid, int tid, const char *filename,
                               struct dso_id *id, struct machine *machine)
{
        struct thread *thread;
        struct nsinfo *nsi = NULL;
        struct nsinfo *nnsi;
        struct dso *dso;
        bool vdso;

        thread = machine__findnew_thread(machine, pid, tid);
        if (thread == NULL) {
                pr_err("cannot find or create a task %d/%d.\n", tid, pid);
                return NULL;
        }

        vdso = is_vdso_map(filename);
        nsi = nsinfo__get(thread->nsinfo);

        if (vdso) {
                /* The vdso maps are always on the host and not the
                 * container.  Ensure that we don't use setns to look
                 * them up.
                 */
                nnsi = nsinfo__copy(nsi);
                if (nnsi) {
                        nsinfo__put(nsi);
                        nnsi->need_setns = false;
                        nsi = nnsi;
                }
                dso = machine__findnew_vdso(machine, thread);
        } else {
                dso = machine__findnew_dso_id(machine, filename, id);
        }

        if (dso) {
                nsinfo__put(dso->nsinfo);
                dso->nsinfo = nsi;
        } else
                nsinfo__put(nsi);

        thread__put(thread);
        return dso;
}

static int perf_event__repipe_buildid_mmap(struct perf_tool *tool,
                                           union perf_event *event,
                                           struct perf_sample *sample,
                                           struct machine *machine)
{
        struct dso *dso;

        dso = findnew_dso(event->mmap.pid, event->mmap.tid,
                          event->mmap.filename, NULL, machine);

        if (dso && !dso->hit) {
                dso->hit = 1;
                dso__inject_build_id(dso, tool, machine, sample->cpumode, 0);
        }
        dso__put(dso);

        return perf_event__repipe(tool, event, sample, machine);
}

static int perf_event__repipe_mmap2(struct perf_tool *tool,
                                   union perf_event *event,
                                   struct perf_sample *sample,
                                   struct machine *machine)
{
        int err;

        err = perf_event__process_mmap2(tool, event, sample, machine);
        perf_event__repipe(tool, event, sample, machine);

        if (event->header.misc & PERF_RECORD_MISC_MMAP_BUILD_ID) {
                struct dso *dso;

                dso = findnew_dso(event->mmap2.pid, event->mmap2.tid,
                                  event->mmap2.filename, NULL, machine);
                if (dso) {
                        /* mark it not to inject build-id */
                        dso->hit = 1;
                }
                dso__put(dso);
        }

        return err;
}

#ifdef HAVE_JITDUMP
static int perf_event__jit_repipe_mmap2(struct perf_tool *tool,
                                        union perf_event *event,
                                        struct perf_sample *sample,
                                        struct machine *machine)
{
        struct perf_inject *inject = container_of(tool, struct perf_inject, tool);
        u64 n = 0;
        int ret;

        /*
         * if jit marker, then inject jit mmaps and generate ELF images
         */
        ret = jit_process(inject->session, &inject->output, machine,
                          event->mmap2.filename, event->mmap2.pid, event->mmap2.tid, &n);
        if (ret < 0)
                return ret;
        if (ret) {
                inject->bytes_written += n;
                return 0;
        }
        return perf_event__repipe_mmap2(tool, event, sample, machine);
}
#endif

static int perf_event__repipe_buildid_mmap2(struct perf_tool *tool,
                                            union perf_event *event,
                                            struct perf_sample *sample,
                                            struct machine *machine)
{
        struct dso_id dso_id = {
                .maj = event->mmap2.maj,
                .min = event->mmap2.min,
                .ino = event->mmap2.ino,
                .ino_generation = event->mmap2.ino_generation,
        };
        struct dso *dso;

        if (event->header.misc & PERF_RECORD_MISC_MMAP_BUILD_ID) {
                /* cannot use dso_id since it'd have invalid info */
                dso = findnew_dso(event->mmap2.pid, event->mmap2.tid,
                                  event->mmap2.filename, NULL, machine);
                if (dso) {
                        /* mark it not to inject build-id */
                        dso->hit = 1;
                }
                dso__put(dso);
                return 0;
        }

        dso = findnew_dso(event->mmap2.pid, event->mmap2.tid,
                          event->mmap2.filename, &dso_id, machine);

        if (dso && !dso->hit) {
                dso->hit = 1;
                dso__inject_build_id(dso, tool, machine, sample->cpumode,
                                     event->mmap2.flags);
        }
        dso__put(dso);

        perf_event__repipe(tool, event, sample, machine);

        return 0;
}

static int perf_event__repipe_fork(struct perf_tool *tool,
                                   union perf_event *event,
                                   struct perf_sample *sample,
                                   struct machine *machine)
{
        int err;

        err = perf_event__process_fork(tool, event, sample, machine);
        perf_event__repipe(tool, event, sample, machine);

        return err;
}

static int perf_event__repipe_comm(struct perf_tool *tool,
                                   union perf_event *event,
                                   struct perf_sample *sample,
                                   struct machine *machine)
{
        int err;

        err = perf_event__process_comm(tool, event, sample, machine);
        perf_event__repipe(tool, event, sample, machine);

        return err;
}

static int perf_event__repipe_namespaces(struct perf_tool *tool,
                                         union perf_event *event,
                                         struct perf_sample *sample,
                                         struct machine *machine)
{
        int err = perf_event__process_namespaces(tool, event, sample, machine);

        perf_event__repipe(tool, event, sample, machine);

        return err;
}

static int perf_event__repipe_exit(struct perf_tool *tool,
                                   union perf_event *event,
                                   struct perf_sample *sample,
                                   struct machine *machine)
{
        int err;

        err = perf_event__process_exit(tool, event, sample, machine);
        perf_event__repipe(tool, event, sample, machine);

        return err;
}

static int perf_event__repipe_tracing_data(struct perf_session *session,
                                           union perf_event *event)
{
        int err;

        perf_event__repipe_synth(session->tool, event);
        err = perf_event__process_tracing_data(session, event);

        return err;
}

static int dso__read_build_id(struct dso *dso)
{
        struct nscookie nsc;

        if (dso->has_build_id)
                return 0;

        nsinfo__mountns_enter(dso->nsinfo, &nsc);
        if (filename__read_build_id(dso->long_name, &dso->bid) > 0)
                dso->has_build_id = true;
        nsinfo__mountns_exit(&nsc);

        return dso->has_build_id ? 0 : -1;
}

static int dso__inject_build_id(struct dso *dso, struct perf_tool *tool,
                                struct machine *machine, u8 cpumode, u32 flags)
{
        int err;

        if (is_anon_memory(dso->long_name) || flags & MAP_HUGETLB)
                return 0;
        if (is_no_dso_memory(dso->long_name))
                return 0;

        if (dso__read_build_id(dso) < 0) {
                pr_debug("no build_id found for %s\n", dso->long_name);
                return -1;
        }

        err = perf_event__synthesize_build_id(tool, dso, cpumode,
                                              perf_event__repipe, machine);
        if (err) {
                pr_err("Can't synthesize build_id event for %s\n", dso->long_name);
                return -1;
        }

        return 0;
}

int perf_event__inject_buildid(struct perf_tool *tool, union perf_event *event,
                               struct perf_sample *sample,
                               struct evsel *evsel __maybe_unused,
                               struct machine *machine)
{
        struct addr_location al;
        struct thread *thread;

        thread = machine__findnew_thread(machine, sample->pid, sample->tid);
        if (thread == NULL) {
                pr_err("problem processing %d event, skipping it.\n",
                       event->header.type);
                goto repipe;
        }

        if (thread__find_map(thread, sample->cpumode, sample->ip, &al)) {
                if (!al.map->dso->hit) {
                        al.map->dso->hit = 1;
                        dso__inject_build_id(al.map->dso, tool, machine,
                                             sample->cpumode, al.map->flags);
                }
        }

        thread__put(thread);
repipe:
        perf_event__repipe(tool, event, sample, machine);
        return 0;
}

static int perf_inject__sched_process_exit(struct perf_tool *tool,
                                           union perf_event *event __maybe_unused,
                                           struct perf_sample *sample,
                                           struct evsel *evsel __maybe_unused,
                                           struct machine *machine __maybe_unused)
{
        struct perf_inject *inject = container_of(tool, struct perf_inject, tool);
        struct event_entry *ent;

        list_for_each_entry(ent, &inject->samples, node) {
                if (sample->tid == ent->tid) {
                        list_del_init(&ent->node);
                        free(ent);
                        break;
                }
        }

        return 0;
}

static int perf_inject__sched_switch(struct perf_tool *tool,
                                     union perf_event *event,
                                     struct perf_sample *sample,
                                     struct evsel *evsel,
                                     struct machine *machine)
{
        struct perf_inject *inject = container_of(tool, struct perf_inject, tool);
        struct event_entry *ent;

        perf_inject__sched_process_exit(tool, event, sample, evsel, machine);

        ent = malloc(event->header.size + sizeof(struct event_entry));
        if (ent == NULL) {
                color_fprintf(stderr, PERF_COLOR_RED,
                             "Not enough memory to process sched switch event!");
                return -1;
        }

        ent->tid = sample->tid;
        memcpy(&ent->event, event, event->header.size);
        list_add(&ent->node, &inject->samples);
        return 0;
}

static int perf_inject__sched_stat(struct perf_tool *tool,
                                   union perf_event *event __maybe_unused,
                                   struct perf_sample *sample,
                                   struct evsel *evsel,
                                   struct machine *machine)
{
        struct event_entry *ent;
        union perf_event *event_sw;
        struct perf_sample sample_sw;
        struct perf_inject *inject = container_of(tool, struct perf_inject, tool);
        u32 pid = evsel__intval(evsel, sample, "pid");

        list_for_each_entry(ent, &inject->samples, node) {
                if (pid == ent->tid)
                        goto found;
        }

        return 0;
found:
        event_sw = &ent->event[0];
        evsel__parse_sample(evsel, event_sw, &sample_sw);

        sample_sw.period = sample->period;
        sample_sw.time   = sample->time;
        perf_event__synthesize_sample(event_sw, evsel->core.attr.sample_type,
                                      evsel->core.attr.read_format, &sample_sw);
        build_id__mark_dso_hit(tool, event_sw, &sample_sw, evsel, machine);
        return perf_event__repipe(tool, event_sw, &sample_sw, machine);
}

static void sig_handler(int sig __maybe_unused)
{
        session_done = 1;
}

static int evsel__check_stype(struct evsel *evsel, u64 sample_type, const char *sample_msg)
{
        struct perf_event_attr *attr = &evsel->core.attr;
        const char *name = evsel__name(evsel);

        if (!(attr->sample_type & sample_type)) {
                pr_err("Samples for %s event do not have %s attribute set.",
                        name, sample_msg);
                return -EINVAL;
        }

        return 0;
}

static int drop_sample(struct perf_tool *tool __maybe_unused,
                       union perf_event *event __maybe_unused,
                       struct perf_sample *sample __maybe_unused,
                       struct evsel *evsel __maybe_unused,
                       struct machine *machine __maybe_unused)
{
        return 0;
}

static void strip_init(struct perf_inject *inject)
{
        struct evlist *evlist = inject->session->evlist;
        struct evsel *evsel;

        inject->tool.context_switch = perf_event__drop;

        evlist__for_each_entry(evlist, evsel)
                evsel->handler = drop_sample;
}

static int parse_vm_time_correlation(const struct option *opt, const char *str, int unset)
{
        struct perf_inject *inject = opt->value;
        const char *args;
        char *dry_run;

        if (unset)
                return 0;

        inject->itrace_synth_opts.set = true;
        inject->itrace_synth_opts.vm_time_correlation = true;
        inject->in_place_update = true;

        if (!str)
                return 0;

        dry_run = skip_spaces(str);
        if (!strncmp(dry_run, "dry-run", strlen("dry-run"))) {
                inject->itrace_synth_opts.vm_tm_corr_dry_run = true;
                inject->in_place_update_dry_run = true;
                args = dry_run + strlen("dry-run");
        } else {
                args = str;
        }

        inject->itrace_synth_opts.vm_tm_corr_args = strdup(args);

        return inject->itrace_synth_opts.vm_tm_corr_args ? 0 : -ENOMEM;
}

static int __cmd_inject(struct perf_inject *inject)
{
        int ret = -EINVAL;
        struct perf_session *session = inject->session;
        struct perf_data *data_out = &inject->output;
        int fd = inject->in_place_update ? -1 : perf_data__fd(data_out);
        u64 output_data_offset;

        signal(SIGINT, sig_handler);

        if (inject->build_ids || inject->sched_stat ||
            inject->itrace_synth_opts.set || inject->build_id_all) {
                inject->tool.mmap         = perf_event__repipe_mmap;
                inject->tool.mmap2        = perf_event__repipe_mmap2;
                inject->tool.fork         = perf_event__repipe_fork;
                inject->tool.tracing_data = perf_event__repipe_tracing_data;
        }

        output_data_offset = session->header.data_offset;

        if (inject->build_id_all) {
                inject->tool.mmap         = perf_event__repipe_buildid_mmap;
                inject->tool.mmap2        = perf_event__repipe_buildid_mmap2;
        } else if (inject->build_ids) {
                inject->tool.sample = perf_event__inject_buildid;
        } else if (inject->sched_stat) {
                struct evsel *evsel;

                evlist__for_each_entry(session->evlist, evsel) {
                        const char *name = evsel__name(evsel);

                        if (!strcmp(name, "sched:sched_switch")) {
                                if (evsel__check_stype(evsel, PERF_SAMPLE_TID, "TID"))
                                        return -EINVAL;

                                evsel->handler = perf_inject__sched_switch;
                        } else if (!strcmp(name, "sched:sched_process_exit"))
                                evsel->handler = perf_inject__sched_process_exit;
                        else if (!strncmp(name, "sched:sched_stat_", 17))
                                evsel->handler = perf_inject__sched_stat;
                }
        } else if (inject->itrace_synth_opts.vm_time_correlation) {
                session->itrace_synth_opts = &inject->itrace_synth_opts;
                memset(&inject->tool, 0, sizeof(inject->tool));
                inject->tool.id_index       = perf_event__process_id_index;
                inject->tool.auxtrace_info  = perf_event__process_auxtrace_info;
                inject->tool.auxtrace       = perf_event__process_auxtrace;
                inject->tool.auxtrace_error = perf_event__process_auxtrace_error;
                inject->tool.ordered_events = true;
                inject->tool.ordering_requires_timestamps = true;
        } else if (inject->itrace_synth_opts.set) {
                session->itrace_synth_opts = &inject->itrace_synth_opts;
                inject->itrace_synth_opts.inject = true;
                inject->tool.comm           = perf_event__repipe_comm;
                inject->tool.namespaces     = perf_event__repipe_namespaces;
                inject->tool.exit           = perf_event__repipe_exit;
                inject->tool.id_index       = perf_event__process_id_index;
                inject->tool.auxtrace_info  = perf_event__process_auxtrace_info;
                inject->tool.auxtrace       = perf_event__process_auxtrace;
                inject->tool.aux            = perf_event__drop_aux;
                inject->tool.itrace_start   = perf_event__drop_aux,
                inject->tool.ordered_events = true;
                inject->tool.ordering_requires_timestamps = true;
                /* Allow space in the header for new attributes */
                output_data_offset = 4096;
                if (inject->strip)
                        strip_init(inject);
        }

        if (!inject->itrace_synth_opts.set)
                auxtrace_index__free(&session->auxtrace_index);

        if (!data_out->is_pipe && !inject->in_place_update)
                lseek(fd, output_data_offset, SEEK_SET);

        ret = perf_session__process_events(session);
        if (ret)
                return ret;

        if (!data_out->is_pipe && !inject->in_place_update) {
                if (inject->build_ids)
                        perf_header__set_feat(&session->header,
                                              HEADER_BUILD_ID);
                /*
                 * Keep all buildids when there is unprocessed AUX data because
                 * it is not known which ones the AUX trace hits.
                 */
                if (perf_header__has_feat(&session->header, HEADER_BUILD_ID) &&
                    inject->have_auxtrace && !inject->itrace_synth_opts.set)
                        dsos__hit_all(session);
                /*
                 * The AUX areas have been removed and replaced with
                 * synthesized hardware events, so clear the feature flag.
                 */
                if (inject->itrace_synth_opts.set) {
                        perf_header__clear_feat(&session->header,
                                                HEADER_AUXTRACE);
                        if (inject->itrace_synth_opts.last_branch ||
                            inject->itrace_synth_opts.add_last_branch)
                                perf_header__set_feat(&session->header,
                                                      HEADER_BRANCH_STACK);
                }
                session->header.data_offset = output_data_offset;
                session->header.data_size = inject->bytes_written;
                perf_session__write_header(session, session->evlist, fd, true);
        }

        return ret;
}

int cmd_inject(int argc, const char **argv)
{
        struct perf_inject inject = {
                .tool = {
                        .sample         = perf_event__repipe_sample,
                        .read           = perf_event__repipe_sample,
                        .mmap           = perf_event__repipe,
                        .mmap2          = perf_event__repipe,
                        .comm           = perf_event__repipe,
                        .namespaces     = perf_event__repipe,
                        .cgroup         = perf_event__repipe,
                        .fork           = perf_event__repipe,
                        .exit           = perf_event__repipe,
                        .lost           = perf_event__repipe,
                        .lost_samples   = perf_event__repipe,
                        .aux            = perf_event__repipe,
                        .itrace_start   = perf_event__repipe,
                        .context_switch = perf_event__repipe,
                        .throttle       = perf_event__repipe,
                        .unthrottle     = perf_event__repipe,
                        .ksymbol        = perf_event__repipe,
                        .bpf            = perf_event__repipe,
                        .text_poke      = perf_event__repipe,
                        .attr           = perf_event__repipe_attr,
                        .event_update   = perf_event__repipe_event_update,
                        .tracing_data   = perf_event__repipe_op2_synth,
                        .finished_round = perf_event__repipe_oe_synth,
                        .build_id       = perf_event__repipe_op2_synth,
                        .id_index       = perf_event__repipe_op2_synth,
                        .auxtrace_info  = perf_event__repipe_op2_synth,
                        .auxtrace_error = perf_event__repipe_op2_synth,
                        .time_conv      = perf_event__repipe_op2_synth,
                        .thread_map     = perf_event__repipe_op2_synth,
                        .cpu_map        = perf_event__repipe_op2_synth,
                        .stat_config    = perf_event__repipe_op2_synth,
                        .stat           = perf_event__repipe_op2_synth,
                        .stat_round     = perf_event__repipe_op2_synth,
                        .feature        = perf_event__repipe_op2_synth,
                        .compressed     = perf_event__repipe_op4_synth,
                        .auxtrace       = perf_event__repipe_auxtrace,
                },
                .input_name  = "-",
                .samples = LIST_HEAD_INIT(inject.samples),
                .output = {
                        .path = "-",
                        .mode = PERF_DATA_MODE_WRITE,
                        .use_stdio = true,
                },
        };
        struct perf_data data = {
                .mode = PERF_DATA_MODE_READ,
                .use_stdio = true,
        };
        int ret;

        struct option options[] = {
                OPT_BOOLEAN('b', "build-ids", &inject.build_ids,
                            "Inject build-ids into the output stream"),
                OPT_BOOLEAN(0, "buildid-all", &inject.build_id_all,
                            "Inject build-ids of all DSOs into the output stream"),
                OPT_STRING('i', "input", &inject.input_name, "file",
                           "input file name"),
                OPT_STRING('o', "output", &inject.output.path, "file",
                           "output file name"),
                OPT_BOOLEAN('s', "sched-stat", &inject.sched_stat,
                            "Merge sched-stat and sched-switch for getting events "
                            "where and how long tasks slept"),
#ifdef HAVE_JITDUMP
                OPT_BOOLEAN('j', "jit", &inject.jit_mode, "merge jitdump files into perf.data file"),
#endif
                OPT_INCR('v', "verbose", &verbose,
                         "be more verbose (show build ids, etc)"),
                OPT_STRING(0, "kallsyms", &symbol_conf.kallsyms_name, "file",
                           "kallsyms pathname"),
                OPT_BOOLEAN('f', "force", &data.force, "don't complain, do it"),
                OPT_CALLBACK_OPTARG(0, "itrace", &inject.itrace_synth_opts,
                                    NULL, "opts", "Instruction Tracing options\n"
                                    ITRACE_HELP,
                                    itrace_parse_synth_opts),
                OPT_BOOLEAN(0, "strip", &inject.strip,
                            "strip non-synthesized events (use with --itrace)"),
                OPT_CALLBACK_OPTARG(0, "vm-time-correlation", &inject, NULL, "opts",
                                    "correlate time between VM guests and the host",
                                    parse_vm_time_correlation),
                OPT_END()
        };
        const char * const inject_usage[] = {
                "perf inject [<options>]",
                NULL
        };
#ifndef HAVE_JITDUMP
        set_option_nobuild(options, 'j', "jit", "NO_LIBELF=1", true);
#endif
        argc = parse_options(argc, argv, options, inject_usage, 0);

        /*
         * Any (unrecognized) arguments left?
         */
        if (argc)
                usage_with_options(inject_usage, options);

        if (inject.strip && !inject.itrace_synth_opts.set) {
                pr_err("--strip option requires --itrace option\n");
                return -1;
        }

        if (inject.in_place_update) {
                if (!strcmp(inject.input_name, "-")) {
                        pr_err("Input file name required for in-place updating\n");
                        return -1;
                }
                if (strcmp(inject.output.path, "-")) {
                        pr_err("Output file name must not be specified for in-place updating\n");
                        return -1;
                }
                if (!data.force && !inject.in_place_update_dry_run) {
                        pr_err("The input file would be updated in place, "
                                "the --force option is required.\n");
                        return -1;
                }
                if (!inject.in_place_update_dry_run)
                        data.in_place_update = true;
        } else if (perf_data__open(&inject.output)) {
                perror("failed to create output file");
                return -1;
        }

        data.path = inject.input_name;
        inject.session = perf_session__new(&data, inject.output.is_pipe, &inject.tool);
        if (IS_ERR(inject.session)) {
                ret = PTR_ERR(inject.session);
                goto out_close_output;
        }

        if (zstd_init(&(inject.session->zstd_data), 0) < 0)
                pr_warning("Decompression initialization failed.\n");

        if (inject.build_ids && !inject.build_id_all) {
                /*
                 * to make sure the mmap records are ordered correctly
                 * and so that the correct especially due to jitted code
                 * mmaps. We cannot generate the buildid hit list and
                 * inject the jit mmaps at the same time for now.
                 */
                inject.tool.ordered_events = true;
                inject.tool.ordering_requires_timestamps = true;
        }

        if (inject.sched_stat) {
                inject.tool.ordered_events = true;
        }

#ifdef HAVE_JITDUMP
        if (inject.jit_mode) {
                inject.tool.mmap2          = perf_event__jit_repipe_mmap2;
                inject.tool.mmap           = perf_event__jit_repipe_mmap;
                inject.tool.ordered_events = true;
                inject.tool.ordering_requires_timestamps = true;
                /*
                 * JIT MMAP injection injects all MMAP events in one go, so it
                 * does not obey finished_round semantics.
                 */
                inject.tool.finished_round = perf_event__drop_oe;
        }
#endif
        ret = symbol__init(&inject.session->header.env);
        if (ret < 0)
                goto out_delete;

        ret = __cmd_inject(&inject);

out_delete:
        zstd_fini(&(inject.session->zstd_data));
        perf_session__delete(inject.session);
out_close_output:
        perf_data__close(&inject.output);
        free(inject.itrace_synth_opts.vm_tm_corr_args);
        return ret;
}