// SPDX-License-Identifier: GPL-2.0
#include <linux/compiler.h>
#include <perf/cpumap.h>
#include <string.h>
#include "cpumap.h"
#include "evlist.h"
#include "evsel.h"
#include "header.h"
#include "machine.h"
#include "util/synthetic-events.h"
#include "tool.h"
#include "tests.h"
#include "debug.h"

static int process_event_unit(const struct perf_tool *tool __maybe_unused,
                              union perf_event *event,
                              struct perf_sample *sample __maybe_unused,
                              struct machine *machine __maybe_unused)
{
        struct perf_record_event_update *ev = (struct perf_record_event_update *)event;

        TEST_ASSERT_VAL("wrong id", ev->id == 123);
        TEST_ASSERT_VAL("wrong id", ev->type == PERF_EVENT_UPDATE__UNIT);
        TEST_ASSERT_VAL("wrong unit", !strcmp(ev->unit, "KRAVA"));
        return 0;
}

static int process_event_scale(const struct perf_tool *tool __maybe_unused,
                               union perf_event *event,
                               struct perf_sample *sample __maybe_unused,
                               struct machine *machine __maybe_unused)
{
        struct perf_record_event_update *ev = (struct perf_record_event_update *)event;

        TEST_ASSERT_VAL("wrong id", ev->id == 123);
        TEST_ASSERT_VAL("wrong id", ev->type == PERF_EVENT_UPDATE__SCALE);
        TEST_ASSERT_VAL("wrong scale", ev->scale.scale == 0.123);
        return 0;
}

struct event_name {
        struct perf_tool tool;
        const char *name;
};

static int process_event_name(const struct perf_tool *tool,
                              union perf_event *event,
                              struct perf_sample *sample __maybe_unused,
                              struct machine *machine __maybe_unused)
{
        struct event_name *tmp = container_of(tool, struct event_name, tool);
        struct perf_record_event_update *ev = (struct perf_record_event_update *)event;

        TEST_ASSERT_VAL("wrong id", ev->id == 123);
        TEST_ASSERT_VAL("wrong id", ev->type == PERF_EVENT_UPDATE__NAME);
        TEST_ASSERT_VAL("wrong name", !strcmp(ev->name, tmp->name));
        return 0;
}

static int process_event_cpus(const struct perf_tool *tool __maybe_unused,
                              union perf_event *event,
                              struct perf_sample *sample __maybe_unused,
                              struct machine *machine __maybe_unused)
{
        struct perf_record_event_update *ev = (struct perf_record_event_update *)event;
        struct perf_cpu_map *map;

        map = cpu_map__new_data(&ev->cpus.cpus);

        TEST_ASSERT_VAL("wrong id", ev->id == 123);
        TEST_ASSERT_VAL("wrong type", ev->type == PERF_EVENT_UPDATE__CPUS);
        TEST_ASSERT_VAL("wrong cpus", perf_cpu_map__nr(map) == 3);
        TEST_ASSERT_VAL("wrong cpus", perf_cpu_map__cpu(map, 0).cpu == 1);
        TEST_ASSERT_VAL("wrong cpus", perf_cpu_map__cpu(map, 1).cpu == 2);
        TEST_ASSERT_VAL("wrong cpus", perf_cpu_map__cpu(map, 2).cpu == 3);
        perf_cpu_map__put(map);
        return 0;
}

static int test__event_update(struct test_suite *test __maybe_unused, int subtest __maybe_unused)
{
        struct evsel *evsel;
        struct event_name tmp;
        struct evlist *evlist = evlist__new_default();

        TEST_ASSERT_VAL("failed to get evlist", evlist);

        evsel = evlist__first(evlist);

        TEST_ASSERT_VAL("failed to allocate ids",
                        !perf_evsel__alloc_id(&evsel->core, 1, 1));

        perf_evlist__id_add(&evlist->core, &evsel->core, 0, 0, 123);

        free((char *)evsel->unit);
        evsel->unit = strdup("KRAVA");

        TEST_ASSERT_VAL("failed to synthesize attr update unit",
                        !perf_event__synthesize_event_update_unit(NULL, evsel, process_event_unit));

        evsel->scale = 0.123;

        TEST_ASSERT_VAL("failed to synthesize attr update scale",
                        !perf_event__synthesize_event_update_scale(NULL, evsel, process_event_scale));

        perf_tool__init(&tmp.tool, /*ordered_events=*/false);
        tmp.name = evsel__name(evsel);

        TEST_ASSERT_VAL("failed to synthesize attr update name",
                        !perf_event__synthesize_event_update_name(&tmp.tool, evsel, process_event_name));

        perf_cpu_map__put(evsel->core.pmu_cpus);
        evsel->core.pmu_cpus = perf_cpu_map__new("1,2,3");

        TEST_ASSERT_VAL("failed to synthesize attr update cpus",
                        !perf_event__synthesize_event_update_cpus(&tmp.tool, evsel, process_event_cpus));

        evlist__delete(evlist);
        return 0;
}

DEFINE_SUITE("Synthesize attr update", event_update);