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