// SPDX-License-Identifier: GPL-2.0
/*
 * Intel Platform Monitory Technology Discovery KUNIT tests
 *
 * Copyright (c) 2025, Intel Corporation.
 * All Rights Reserved.
 */

#include <kunit/test.h>
#include <linux/err.h>
#include <linux/intel_pmt_features.h>
#include <linux/intel_vsec.h>
#include <linux/module.h>
#include <linux/slab.h>

#define PMT_FEATURE_COUNT (FEATURE_MAX + 1)

static void
validate_pmt_regions(struct kunit *test, struct pmt_feature_group *feature_group, int feature_id)
{
        int i;

        kunit_info(test, "Feature ID %d [%s] has %d regions.\n", feature_id,
                   pmt_feature_names[feature_id], feature_group->count);

        for (i = 0; i < feature_group->count; i++) {
                struct telemetry_region *region = &feature_group->regions[i];

                kunit_info(test, "  - Region %d: cdie_mask=%u, package_id=%u, partition=%u, segment=%u,",
                           i, region->plat_info.cdie_mask, region->plat_info.package_id,
                           region->plat_info.partition, region->plat_info.segment);
                kunit_info(test, "\t\tbus=%u, device=%u, function=%u, guid=0x%x,",
                           region->plat_info.bus_number, region->plat_info.device_number,
                           region->plat_info.function_number, region->guid);
                kunit_info(test, "\t\taddr=%p, size=%zu, num_rmids=%u", region->addr, region->size,
                           region->num_rmids);


                KUNIT_ASSERT_GE(test, region->plat_info.cdie_mask, 0);
                KUNIT_ASSERT_GE(test, region->plat_info.package_id, 0);
                KUNIT_ASSERT_GE(test, region->plat_info.partition, 0);
                KUNIT_ASSERT_GE(test, region->plat_info.segment, 0);
                KUNIT_ASSERT_GE(test, region->plat_info.bus_number, 0);
                KUNIT_ASSERT_GE(test, region->plat_info.device_number, 0);
                KUNIT_ASSERT_GE(test, region->plat_info.function_number, 0);

                KUNIT_ASSERT_NE(test, region->guid, 0);

                KUNIT_ASSERT_NOT_ERR_OR_NULL(test, (__force const void *)region->addr);
        }
}

static void linebreak(struct kunit *test)
{
        kunit_info(test, "*****************************************************************************\n");
}

static void test_intel_pmt_get_regions_by_feature(struct kunit *test)
{
        struct pmt_feature_group *feature_group;
        int num_available = 0;
        int feature_id;

        /* Iterate through all possible feature IDs */
        for (feature_id = 1; feature_id < PMT_FEATURE_COUNT; feature_id++, linebreak(test)) {
                const char *name;

                if (!pmt_feature_id_is_valid(feature_id))
                        continue;

                name = pmt_feature_names[feature_id];

                feature_group = intel_pmt_get_regions_by_feature(feature_id);
                if (IS_ERR(feature_group)) {
                        if (PTR_ERR(feature_group) == -ENOENT)
                                kunit_warn(test, "intel_pmt_get_regions_by_feature() reporting feature %d [%s] is not present.\n",
                                           feature_id, name);
                        else
                                kunit_warn(test, "intel_pmt_get_regions_by_feature() returned error %ld while attempt to lookup %d [%s].\n",
                                           PTR_ERR(feature_group), feature_id, name);

                        continue;
                }

                if (!feature_group) {
                        kunit_warn(test, "Feature ID %d: %s is not available.\n", feature_id, name);
                        continue;
                }

                num_available++;

                validate_pmt_regions(test, feature_group, feature_id);

                intel_pmt_put_feature_group(feature_group);
        }

        if (num_available == 0)
                kunit_warn(test, "No PMT region groups were available for any feature ID (0-10).\n");
}

static struct kunit_case intel_pmt_discovery_test_cases[] = {
        KUNIT_CASE(test_intel_pmt_get_regions_by_feature),
        {}
};

static struct kunit_suite intel_pmt_discovery_test_suite = {
        .name = "pmt_discovery_test",
        .test_cases = intel_pmt_discovery_test_cases,
};

kunit_test_suite(intel_pmt_discovery_test_suite);

MODULE_IMPORT_NS("INTEL_PMT_DISCOVERY");
MODULE_AUTHOR("David E. Box <david.e.box@linux.intel.com>");
MODULE_DESCRIPTION("Intel PMT Discovery KUNIT test driver");
MODULE_LICENSE("GPL");