// SPDX-License-Identifier: GPL-2.0 AND MIT
/*
 * Copyright © 2023 Intel Corporation
 */

#include <kunit/test.h>

#include "xe_device.h"
#include "xe_kunit_helpers.h"

static int guc_dbm_test_init(struct kunit *test)
{
 struct xe_guc_db_mgr *dbm;

 xe_kunit_helper_xe_device_test_init(test);
 dbm = &xe_device_get_gt(test->priv, 0)->uc.guc.dbm;

 mutex_init(dbm_mutex(dbm));
 test->priv = dbm;
 return 0;
}

static void test_empty(struct kunit *test)
{
 struct xe_guc_db_mgr *dbm = test->priv;

 KUNIT_ASSERT_EQ(test, xe_guc_db_mgr_init(dbm, 0), 0);
 KUNIT_ASSERT_EQ(test, dbm->count, 0);

 mutex_lock(dbm_mutex(dbm));
 KUNIT_EXPECT_LT(test, xe_guc_db_mgr_reserve_id_locked(dbm), 0);
 mutex_unlock(dbm_mutex(dbm));

 KUNIT_EXPECT_LT(test, xe_guc_db_mgr_reserve_range(dbm, 1, 0), 0);
}

static void test_default(struct kunit *test)
{
 struct xe_guc_db_mgr *dbm = test->priv;

 KUNIT_ASSERT_EQ(test, xe_guc_db_mgr_init(dbm, ~0), 0);
 KUNIT_ASSERT_EQ(test, dbm->count, GUC_NUM_DOORBELLS);
}

static const unsigned int guc_dbm_params[] = {
 GUC_NUM_DOORBELLS / 64,
 GUC_NUM_DOORBELLS / 32,
 GUC_NUM_DOORBELLS / 8,
 GUC_NUM_DOORBELLS,
};

static void uint_param_get_desc(const unsigned int *p, char *desc)
{
 snprintf(desc, KUNIT_PARAM_DESC_SIZE, "%u", *p);
}

KUNIT_ARRAY_PARAM(guc_dbm, guc_dbm_params, uint_param_get_desc);

static void test_size(struct kunit *test)
{
 const unsigned int *p = test->param_value;
 struct xe_guc_db_mgr *dbm = test->priv;
 unsigned int n;
 int id;

 KUNIT_ASSERT_EQ(test, xe_guc_db_mgr_init(dbm, *p), 0);
 KUNIT_ASSERT_EQ(test, dbm->count, *p);

 mutex_lock(dbm_mutex(dbm));
 for (n = 0; n < *p; n++) {
  KUNIT_EXPECT_GE(test, id = xe_guc_db_mgr_reserve_id_locked(dbm), 0);
  KUNIT_EXPECT_LT(test, id, dbm->count);
 }
 KUNIT_EXPECT_LT(test, xe_guc_db_mgr_reserve_id_locked(dbm), 0);
 mutex_unlock(dbm_mutex(dbm));

 mutex_lock(dbm_mutex(dbm));
 for (n = 0; n < *p; n++)
  xe_guc_db_mgr_release_id_locked(dbm, n);
 mutex_unlock(dbm_mutex(dbm));
}

static void test_reuse(struct kunit *test)
{
 const unsigned int *p = test->param_value;
 struct xe_guc_db_mgr *dbm = test->priv;
 unsigned int n;

 KUNIT_ASSERT_EQ(test, xe_guc_db_mgr_init(dbm, *p), 0);

 mutex_lock(dbm_mutex(dbm));
 for (n = 0; n < *p; n++)
  KUNIT_EXPECT_GE(test, xe_guc_db_mgr_reserve_id_locked(dbm), 0);
 KUNIT_EXPECT_LT(test, xe_guc_db_mgr_reserve_id_locked(dbm), 0);
 mutex_unlock(dbm_mutex(dbm));

 mutex_lock(dbm_mutex(dbm));
 for (n = 0; n < *p; n++) {
  xe_guc_db_mgr_release_id_locked(dbm, n);
  KUNIT_EXPECT_EQ(test, xe_guc_db_mgr_reserve_id_locked(dbm), n);
 }
 KUNIT_EXPECT_LT(test, xe_guc_db_mgr_reserve_id_locked(dbm), 0);
 mutex_unlock(dbm_mutex(dbm));

 mutex_lock(dbm_mutex(dbm));
 for (n = 0; n < *p; n++)
  xe_guc_db_mgr_release_id_locked(dbm, n);
 mutex_unlock(dbm_mutex(dbm));
}

static void test_range_overlap(struct kunit *test)
{
 const unsigned int *p = test->param_value;
 struct xe_guc_db_mgr *dbm = test->priv;
 int id1, id2, id3;
 unsigned int n;

 KUNIT_ASSERT_EQ(test, xe_guc_db_mgr_init(dbm, ~0), 0);
 KUNIT_ASSERT_LE(test, *p, dbm->count);

 KUNIT_ASSERT_GE(test, id1 = xe_guc_db_mgr_reserve_range(dbm, *p, 0), 0);
 for (n = 0; n < dbm->count - *p; n++) {
  KUNIT_ASSERT_GE(test, id2 = xe_guc_db_mgr_reserve_range(dbm, 1, 0), 0);
  KUNIT_ASSERT_NE(test, id2, id1);
  KUNIT_ASSERT_NE_MSG(test, id2 < id1, id2 > id1 + *p - 1,
        "id1=%d id2=%d", id1, id2);
 }
 KUNIT_ASSERT_LT(test, xe_guc_db_mgr_reserve_range(dbm, 1, 0), 0);
 xe_guc_db_mgr_release_range(dbm, 0, dbm->count);

 if (*p >= 1) {
  KUNIT_ASSERT_GE(test, id1 = xe_guc_db_mgr_reserve_range(dbm, 1, 0), 0);
  KUNIT_ASSERT_GE(test, id2 = xe_guc_db_mgr_reserve_range(dbm, *p - 1, 0), 0);
  KUNIT_ASSERT_NE(test, id2, id1);
  KUNIT_ASSERT_NE_MSG(test, id1 < id2, id1 > id2 + *p - 2,
        "id1=%d id2=%d", id1, id2);
  for (n = 0; n < dbm->count - *p; n++) {
   KUNIT_ASSERT_GE(test, id3 = xe_guc_db_mgr_reserve_range(dbm, 1, 0), 0);
   KUNIT_ASSERT_NE(test, id3, id1);
   KUNIT_ASSERT_NE(test, id3, id2);
   KUNIT_ASSERT_NE_MSG(test, id3 < id2, id3 > id2 + *p - 2,
         "id3=%d id2=%d", id3, id2);
  }
  KUNIT_ASSERT_LT(test, xe_guc_db_mgr_reserve_range(dbm, 1, 0), 0);
  xe_guc_db_mgr_release_range(dbm, 0, dbm->count);
 }
}

static void test_range_compact(struct kunit *test)
{
 const unsigned int *p = test->param_value;
 struct xe_guc_db_mgr *dbm = test->priv;
 unsigned int n;

 KUNIT_ASSERT_EQ(test, xe_guc_db_mgr_init(dbm, ~0), 0);
 KUNIT_ASSERT_NE(test, *p, 0);
 KUNIT_ASSERT_LE(test, *p, dbm->count);
 if (dbm->count % *p)
  kunit_skip(test, "must be divisible");

 KUNIT_ASSERT_GE(test, xe_guc_db_mgr_reserve_range(dbm, *p, 0), 0);
 for (n = 1; n < dbm->count / *p; n++)
  KUNIT_ASSERT_GE(test, xe_guc_db_mgr_reserve_range(dbm, *p, 0), 0);
 KUNIT_ASSERT_LT(test, xe_guc_db_mgr_reserve_range(dbm, 1, 0), 0);
 xe_guc_db_mgr_release_range(dbm, 0, dbm->count);
}

static void test_range_spare(struct kunit *test)
{
 const unsigned int *p = test->param_value;
 struct xe_guc_db_mgr *dbm = test->priv;
 int id;

 KUNIT_ASSERT_EQ(test, xe_guc_db_mgr_init(dbm, ~0), 0);
 KUNIT_ASSERT_LE(test, *p, dbm->count);

 KUNIT_ASSERT_LT(test, xe_guc_db_mgr_reserve_range(dbm, *p, dbm->count), 0);
 KUNIT_ASSERT_LT(test, xe_guc_db_mgr_reserve_range(dbm, *p, dbm->count - *p + 1), 0);
 KUNIT_ASSERT_EQ(test, id = xe_guc_db_mgr_reserve_range(dbm, *p, dbm->count - *p), 0);
 KUNIT_ASSERT_LT(test, xe_guc_db_mgr_reserve_range(dbm, 1, dbm->count - *p), 0);
 xe_guc_db_mgr_release_range(dbm, id, *p);
}

static struct kunit_case guc_dbm_test_cases[] = {
 KUNIT_CASE(test_empty),
 KUNIT_CASE(test_default),
 KUNIT_CASE_PARAM(test_size, guc_dbm_gen_params),
 KUNIT_CASE_PARAM(test_reuse, guc_dbm_gen_params),
 KUNIT_CASE_PARAM(test_range_overlap, guc_dbm_gen_params),
 KUNIT_CASE_PARAM(test_range_compact, guc_dbm_gen_params),
 KUNIT_CASE_PARAM(test_range_spare, guc_dbm_gen_params),
 {}
};

static struct kunit_suite guc_dbm_suite = {
 .name = "guc_dbm",
 .test_cases = guc_dbm_test_cases,
 .init = guc_dbm_test_init,
};

kunit_test_suites(&guc_dbm_suite);