// SPDX-License-Identifier: GPL-2.0
#include <linux/err.h>
#include <event-parse.h>
#include "evsel.h"
#include "tests.h"
#include "debug.h"

static int evsel__test_field(struct evsel *evsel, const char *name, int size, bool should_be_signed)
{
        struct tep_format_field *field = evsel__field(evsel, name);
        int is_signed;
        int ret = 0;

        if (field == NULL) {
                pr_debug("%s: \"%s\" field not found!\n", evsel->name, name);
                return -1;
        }

        is_signed = !!(field->flags & TEP_FIELD_IS_SIGNED);
        if (should_be_signed && !is_signed) {
                pr_debug("%s: \"%s\" signedness(%d) is wrong, should be %d\n",
                         evsel->name, name, is_signed, should_be_signed);
                ret = -1;
        }

        if (field->size != size) {
                pr_debug("%s: \"%s\" size (%d) should be %d!\n",
                         evsel->name, name, field->size, size);
                ret = -1;
        }

        return ret;
}

static int test__perf_evsel__tp_sched_test(struct test_suite *test __maybe_unused,
                                           int subtest __maybe_unused)
{
        struct evsel *evsel = evsel__newtp("sched", "sched_switch");
        int ret = TEST_OK;

        if (IS_ERR(evsel)) {
                pr_debug("evsel__newtp failed with %ld\n", PTR_ERR(evsel));
                return PTR_ERR(evsel) == -EACCES ? TEST_SKIP : TEST_FAIL;
        }

        if (evsel__test_field(evsel, "prev_comm", 16, false))
                ret = TEST_FAIL;

        if (evsel__test_field(evsel, "prev_pid", 4, true))
                ret = TEST_FAIL;

        if (evsel__test_field(evsel, "prev_prio", 4, true))
                ret = TEST_FAIL;

        if (evsel__test_field(evsel, "prev_state", sizeof(long), true))
                ret = TEST_FAIL;

        if (evsel__test_field(evsel, "next_comm", 16, false))
                ret = TEST_FAIL;

        if (evsel__test_field(evsel, "next_pid", 4, true))
                ret = TEST_FAIL;

        if (evsel__test_field(evsel, "next_prio", 4, true))
                ret = TEST_FAIL;

        evsel__delete(evsel);

        evsel = evsel__newtp("sched", "sched_wakeup");

        if (IS_ERR(evsel)) {
                pr_debug("evsel__newtp failed with %ld\n", PTR_ERR(evsel));
                return TEST_FAIL;
        }

        if (evsel__test_field(evsel, "comm", 16, false))
                ret = TEST_FAIL;

        if (evsel__test_field(evsel, "pid", 4, true))
                ret = TEST_FAIL;

        if (evsel__test_field(evsel, "prio", 4, true))
                ret = TEST_FAIL;

        if (evsel__test_field(evsel, "target_cpu", 4, true))
                ret = TEST_FAIL;

        evsel__delete(evsel);
        return ret;
}

static struct test_case tests__perf_evsel__tp_sched_test[] = {
        TEST_CASE_REASON("Parse sched tracepoints fields",
                         perf_evsel__tp_sched_test,
                         "permissions"),
        {       .name = NULL, }
};

struct test_suite suite__perf_evsel__tp_sched_test = {
        .desc = "Parse sched tracepoints fields",
        .test_cases = tests__perf_evsel__tp_sched_test,
};