// SPDX-License-Identifier: GPL-2.0
#include <linux/types.h>
#include <linux/string.h>
#include <linux/zalloc.h>
#include <stdlib.h>

#include "../../../util/event.h"
#include "../../../util/synthetic-events.h"
#include "../../../util/machine.h"
#include "../../../util/tool.h"
#include "../../../util/map.h"
#include "../../../util/debug.h"
#include "util/sample.h"

#if defined(__x86_64__)

struct perf_event__synthesize_extra_kmaps_cb_args {
        const struct perf_tool *tool;
        perf_event__handler_t process;
        struct machine *machine;
        union perf_event *event;
};

static int perf_event__synthesize_extra_kmaps_cb(struct map *map, void *data)
{
        struct perf_event__synthesize_extra_kmaps_cb_args *args = data;
        union perf_event *event = args->event;
        struct kmap *kmap;
        size_t size;

        if (!__map__is_extra_kernel_map(map))
                return 0;

        kmap = map__kmap(map);

        size = sizeof(event->mmap) - sizeof(event->mmap.filename) +
                      PERF_ALIGN(strlen(kmap->name) + 1, sizeof(u64)) +
                      args->machine->id_hdr_size;

        memset(event, 0, size);

        event->mmap.header.type = PERF_RECORD_MMAP;

        /*
         * kernel uses 0 for user space maps, see kernel/perf_event.c
         * __perf_event_mmap
         */
        if (machine__is_host(args->machine))
                event->header.misc = PERF_RECORD_MISC_KERNEL;
        else
                event->header.misc = PERF_RECORD_MISC_GUEST_KERNEL;

        event->mmap.header.size = size;

        event->mmap.start = map__start(map);
        event->mmap.len   = map__size(map);
        event->mmap.pgoff = map__pgoff(map);
        event->mmap.pid   = args->machine->pid;

        strlcpy(event->mmap.filename, kmap->name, PATH_MAX);

        if (perf_tool__process_synth_event(args->tool, event, args->machine, args->process) != 0)
                return -1;

        return 0;
}

int perf_event__synthesize_extra_kmaps(const struct perf_tool *tool,
                                       perf_event__handler_t process,
                                       struct machine *machine)
{
        int rc;
        struct maps *kmaps = machine__kernel_maps(machine);
        struct perf_event__synthesize_extra_kmaps_cb_args args = {
                .tool = tool,
                .process = process,
                .machine = machine,
                .event = zalloc(sizeof(args.event->mmap) + machine->id_hdr_size),
        };

        if (!args.event) {
                pr_debug("Not enough memory synthesizing mmap event "
                         "for extra kernel maps\n");
                return -1;
        }

        rc = maps__for_each_map(kmaps, perf_event__synthesize_extra_kmaps_cb, &args);

        free(args.event);
        return rc;
}

#endif