// SPDX-License-Identifier: GPL-2.0
#include <stdbool.h>
#include <stdlib.h>
#include <string.h>
#include "tests.h"
#include "dso.h"
#include "debug.h"
#include "event.h"

static int test(const char *path, bool alloc_name, bool kmod,
                int comp, const char *name)
{
        struct kmod_path m;

        memset(&m, 0x0, sizeof(m));

        TEST_ASSERT_VAL("kmod_path__parse",
                        !__kmod_path__parse(&m, path, alloc_name));

        pr_debug("%s - alloc name %d, kmod %d, comp %d, name '%s'\n",
                 path, alloc_name, m.kmod, m.comp, m.name);

        TEST_ASSERT_VAL("wrong kmod", m.kmod == kmod);
        TEST_ASSERT_VAL("wrong comp", m.comp == comp);

        if (name)
                TEST_ASSERT_VAL("wrong name", m.name && !strcmp(name, m.name));
        else
                TEST_ASSERT_VAL("wrong name", !m.name);

        free(m.name);
        return 0;
}

static int test_is_kernel_module(const char *path, int cpumode, bool expect)
{
        TEST_ASSERT_VAL("is_kernel_module",
                        (!!is_kernel_module(path, cpumode)) == (!!expect));
        pr_debug("%s (cpumode: %d) - is_kernel_module: %s\n",
                        path, cpumode, expect ? "true" : "false");
        return 0;
}

#define T(path, an, k, c, n) \
        TEST_ASSERT_VAL("failed", !test(path, an, k, c, n))

#define M(path, c, e) \
        TEST_ASSERT_VAL("failed", !test_is_kernel_module(path, c, e))

static int test__kmod_path__parse(struct test_suite *t __maybe_unused, int subtest __maybe_unused)
{
        /* path                alloc_name  kmod  comp   name   */
        T("/xxxx/xxxx/x-x.ko", true      , true, 0    , "[x_x]");
        T("/xxxx/xxxx/x-x.ko", false     , true, 0    , NULL   );
        T("/xxxx/xxxx/x-x.ko", true      , true, 0    , "[x_x]");
        T("/xxxx/xxxx/x-x.ko", false     , true, 0    , NULL   );
        M("/xxxx/xxxx/x-x.ko", PERF_RECORD_MISC_CPUMODE_UNKNOWN, true);
        M("/xxxx/xxxx/x-x.ko", PERF_RECORD_MISC_KERNEL, true);
        M("/xxxx/xxxx/x-x.ko", PERF_RECORD_MISC_USER, false);

#ifdef HAVE_ZLIB_SUPPORT
        /* path                alloc_name   kmod  comp  name  */
        T("/xxxx/xxxx/x.ko.gz", true     , true, 1   , "[x]");
        T("/xxxx/xxxx/x.ko.gz", false    , true, 1   , NULL );
        T("/xxxx/xxxx/x.ko.gz", true     , true, 1   , "[x]");
        T("/xxxx/xxxx/x.ko.gz", false    , true, 1   , NULL );
        M("/xxxx/xxxx/x.ko.gz", PERF_RECORD_MISC_CPUMODE_UNKNOWN, true);
        M("/xxxx/xxxx/x.ko.gz", PERF_RECORD_MISC_KERNEL, true);
        M("/xxxx/xxxx/x.ko.gz", PERF_RECORD_MISC_USER, false);

        /* path              alloc_name  kmod   comp  name  */
        T("/xxxx/xxxx/x.gz", true      , false, 1   , "x.gz");
        T("/xxxx/xxxx/x.gz", false     , false, 1   , NULL  );
        T("/xxxx/xxxx/x.gz", true      , false, 1   , "x.gz");
        T("/xxxx/xxxx/x.gz", false     , false, 1   , NULL  );
        M("/xxxx/xxxx/x.gz", PERF_RECORD_MISC_CPUMODE_UNKNOWN, false);
        M("/xxxx/xxxx/x.gz", PERF_RECORD_MISC_KERNEL, false);
        M("/xxxx/xxxx/x.gz", PERF_RECORD_MISC_USER, false);

        /* path   alloc_name  kmod   comp  name   */
        T("x.gz", true      , false, 1   , "x.gz");
        T("x.gz", false     , false, 1   , NULL  );
        T("x.gz", true      , false, 1   , "x.gz");
        T("x.gz", false     , false, 1   , NULL  );
        M("x.gz", PERF_RECORD_MISC_CPUMODE_UNKNOWN, false);
        M("x.gz", PERF_RECORD_MISC_KERNEL, false);
        M("x.gz", PERF_RECORD_MISC_USER, false);

        /* path      alloc_name  kmod  comp  name  */
        T("x.ko.gz", true      , true, 1   , "[x]");
        T("x.ko.gz", false     , true, 1   , NULL );
        T("x.ko.gz", true      , true, 1   , "[x]");
        T("x.ko.gz", false     , true, 1   , NULL );
        M("x.ko.gz", PERF_RECORD_MISC_CPUMODE_UNKNOWN, true);
        M("x.ko.gz", PERF_RECORD_MISC_KERNEL, true);
        M("x.ko.gz", PERF_RECORD_MISC_USER, false);
#endif

        /* path            alloc_name  kmod  comp   name           */
        T("[test_module]", true      , true, false, "[test_module]");
        T("[test_module]", false     , true, false, NULL           );
        T("[test_module]", true      , true, false, "[test_module]");
        T("[test_module]", false     , true, false, NULL           );
        M("[test_module]", PERF_RECORD_MISC_CPUMODE_UNKNOWN, true);
        M("[test_module]", PERF_RECORD_MISC_KERNEL, true);
        M("[test_module]", PERF_RECORD_MISC_USER, false);

        /* path            alloc_name  kmod  comp   name           */
        T("[test.module]", true      , true, false, "[test.module]");
        T("[test.module]", false     , true, false, NULL           );
        T("[test.module]", true      , true, false, "[test.module]");
        T("[test.module]", false     , true, false, NULL           );
        M("[test.module]", PERF_RECORD_MISC_CPUMODE_UNKNOWN, true);
        M("[test.module]", PERF_RECORD_MISC_KERNEL, true);
        M("[test.module]", PERF_RECORD_MISC_USER, false);

        /* path     alloc_name  kmod   comp   name    */
        T("[vdso]", true      , false, false, "[vdso]");
        T("[vdso]", false     , false, false, NULL    );
        T("[vdso]", true      , false, false, "[vdso]");
        T("[vdso]", false     , false, false, NULL    );
        M("[vdso]", PERF_RECORD_MISC_CPUMODE_UNKNOWN, false);
        M("[vdso]", PERF_RECORD_MISC_KERNEL, false);
        M("[vdso]", PERF_RECORD_MISC_USER, false);

        T("[vdso32]", true      , false, false, "[vdso32]");
        T("[vdso32]", false     , false, false, NULL      );
        T("[vdso32]", true      , false, false, "[vdso32]");
        T("[vdso32]", false     , false, false, NULL      );
        M("[vdso32]", PERF_RECORD_MISC_CPUMODE_UNKNOWN, false);
        M("[vdso32]", PERF_RECORD_MISC_KERNEL, false);
        M("[vdso32]", PERF_RECORD_MISC_USER, false);

        T("[vdsox32]", true      , false, false, "[vdsox32]");
        T("[vdsox32]", false     , false, false, NULL       );
        T("[vdsox32]", true      , false, false, "[vdsox32]");
        T("[vdsox32]", false     , false, false, NULL       );
        M("[vdsox32]", PERF_RECORD_MISC_CPUMODE_UNKNOWN, false);
        M("[vdsox32]", PERF_RECORD_MISC_KERNEL, false);
        M("[vdsox32]", PERF_RECORD_MISC_USER, false);

        /* path         alloc_name  kmod   comp   name        */
        T("[vsyscall]", true      , false, false, "[vsyscall]");
        T("[vsyscall]", false     , false, false, NULL        );
        T("[vsyscall]", true      , false, false, "[vsyscall]");
        T("[vsyscall]", false     , false, false, NULL        );
        M("[vsyscall]", PERF_RECORD_MISC_CPUMODE_UNKNOWN, false);
        M("[vsyscall]", PERF_RECORD_MISC_KERNEL, false);
        M("[vsyscall]", PERF_RECORD_MISC_USER, false);

        /* path                alloc_name  kmod   comp   name      */
        T("[kernel.kallsyms]", true      , false, false, "[kernel.kallsyms]");
        T("[kernel.kallsyms]", false     , false, false, NULL               );
        T("[kernel.kallsyms]", true      , false, false, "[kernel.kallsyms]");
        T("[kernel.kallsyms]", false     , false, false, NULL               );
        M("[kernel.kallsyms]", PERF_RECORD_MISC_CPUMODE_UNKNOWN, false);
        M("[kernel.kallsyms]", PERF_RECORD_MISC_KERNEL, false);
        M("[kernel.kallsyms]", PERF_RECORD_MISC_USER, false);

        return 0;
}

DEFINE_SUITE("kmod_path__parse", kmod_path__parse);