// 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");