// SPDX-License-Identifier: GPL-2.0
// Copyright (C) 2025 Arm Ltd.
/* This file is intended to be included into mpam_devices.c */

#include <kunit/test.h>

/*
 * This test catches fields that aren't being sanitised - but can't tell you
 * which one...
 */
static void test__props_mismatch(struct kunit *test)
{
	struct mpam_props parent = { 0 };
	struct mpam_props child;

	memset(&child, 0xff, sizeof(child));
	__props_mismatch(&parent, &child, false);

	memset(&child, 0, sizeof(child));
	KUNIT_EXPECT_EQ(test, memcmp(&parent, &child, sizeof(child)), 0);

	memset(&child, 0xff, sizeof(child));
	__props_mismatch(&parent, &child, true);

	KUNIT_EXPECT_EQ(test, memcmp(&parent, &child, sizeof(child)), 0);
}

static struct list_head fake_classes_list;
static struct mpam_class fake_class = { 0 };
static struct mpam_component fake_comp1 = { 0 };
static struct mpam_component fake_comp2 = { 0 };
static struct mpam_vmsc fake_vmsc1 = { 0 };
static struct mpam_vmsc fake_vmsc2 = { 0 };
static struct mpam_msc fake_msc1 = { 0 };
static struct mpam_msc fake_msc2 = { 0 };
static struct mpam_msc_ris fake_ris1 = { 0 };
static struct mpam_msc_ris fake_ris2 = { 0 };
static struct platform_device fake_pdev = { 0 };

static inline void reset_fake_hierarchy(void)
{
	INIT_LIST_HEAD(&fake_classes_list);

	memset(&fake_class, 0, sizeof(fake_class));
	fake_class.level = 3;
	fake_class.type = MPAM_CLASS_CACHE;
	INIT_LIST_HEAD_RCU(&fake_class.components);
	INIT_LIST_HEAD(&fake_class.classes_list);

	memset(&fake_comp1, 0, sizeof(fake_comp1));
	memset(&fake_comp2, 0, sizeof(fake_comp2));
	fake_comp1.comp_id = 1;
	fake_comp2.comp_id = 2;
	INIT_LIST_HEAD(&fake_comp1.vmsc);
	INIT_LIST_HEAD(&fake_comp1.class_list);
	INIT_LIST_HEAD(&fake_comp2.vmsc);
	INIT_LIST_HEAD(&fake_comp2.class_list);

	memset(&fake_vmsc1, 0, sizeof(fake_vmsc1));
	memset(&fake_vmsc2, 0, sizeof(fake_vmsc2));
	INIT_LIST_HEAD(&fake_vmsc1.ris);
	INIT_LIST_HEAD(&fake_vmsc1.comp_list);
	fake_vmsc1.msc = &fake_msc1;
	INIT_LIST_HEAD(&fake_vmsc2.ris);
	INIT_LIST_HEAD(&fake_vmsc2.comp_list);
	fake_vmsc2.msc = &fake_msc2;

	memset(&fake_ris1, 0, sizeof(fake_ris1));
	memset(&fake_ris2, 0, sizeof(fake_ris2));
	fake_ris1.ris_idx = 1;
	INIT_LIST_HEAD(&fake_ris1.msc_list);
	fake_ris2.ris_idx = 2;
	INIT_LIST_HEAD(&fake_ris2.msc_list);

	fake_msc1.pdev = &fake_pdev;
	fake_msc2.pdev = &fake_pdev;

	list_add(&fake_class.classes_list, &fake_classes_list);
}

static void test_mpam_enable_merge_features(struct kunit *test)
{
	reset_fake_hierarchy();

	mutex_lock(&mpam_list_lock);

	/* One Class+Comp, two RIS in one vMSC with common features */
	fake_comp1.class = &fake_class;
	list_add(&fake_comp1.class_list, &fake_class.components);
	fake_comp2.class = NULL;
	fake_vmsc1.comp = &fake_comp1;
	list_add(&fake_vmsc1.comp_list, &fake_comp1.vmsc);
	fake_vmsc2.comp = NULL;
	fake_ris1.vmsc = &fake_vmsc1;
	list_add(&fake_ris1.vmsc_list, &fake_vmsc1.ris);
	fake_ris2.vmsc = &fake_vmsc1;
	list_add(&fake_ris2.vmsc_list, &fake_vmsc1.ris);

	mpam_set_feature(mpam_feat_cpor_part, &fake_ris1.props);
	mpam_set_feature(mpam_feat_cpor_part, &fake_ris2.props);
	fake_ris1.props.cpbm_wd = 4;
	fake_ris2.props.cpbm_wd = 4;

	mpam_enable_merge_features(&fake_classes_list);

	KUNIT_EXPECT_TRUE(test, mpam_has_feature(mpam_feat_cpor_part, &fake_class.props));
	KUNIT_EXPECT_EQ(test, fake_class.props.cpbm_wd, 4);

	reset_fake_hierarchy();

	/* One Class+Comp, two RIS in one vMSC with non-overlapping features */
	fake_comp1.class = &fake_class;
	list_add(&fake_comp1.class_list, &fake_class.components);
	fake_comp2.class = NULL;
	fake_vmsc1.comp = &fake_comp1;
	list_add(&fake_vmsc1.comp_list, &fake_comp1.vmsc);
	fake_vmsc2.comp = NULL;
	fake_ris1.vmsc = &fake_vmsc1;
	list_add(&fake_ris1.vmsc_list, &fake_vmsc1.ris);
	fake_ris2.vmsc = &fake_vmsc1;
	list_add(&fake_ris2.vmsc_list, &fake_vmsc1.ris);

	mpam_set_feature(mpam_feat_cpor_part, &fake_ris1.props);
	mpam_set_feature(mpam_feat_cmax_cmin, &fake_ris2.props);
	fake_ris1.props.cpbm_wd = 4;
	fake_ris2.props.cmax_wd = 4;

	mpam_enable_merge_features(&fake_classes_list);

	/* Multiple RIS within one MSC controlling the same resource can be mismatched */
	KUNIT_EXPECT_TRUE(test, mpam_has_feature(mpam_feat_cpor_part, &fake_class.props));
	KUNIT_EXPECT_TRUE(test, mpam_has_feature(mpam_feat_cmax_cmin, &fake_class.props));
	KUNIT_EXPECT_TRUE(test, mpam_has_feature(mpam_feat_cmax_cmin, &fake_vmsc1.props));
	KUNIT_EXPECT_EQ(test, fake_class.props.cpbm_wd, 4);
	KUNIT_EXPECT_EQ(test, fake_vmsc1.props.cmax_wd, 4);
	KUNIT_EXPECT_EQ(test, fake_class.props.cmax_wd, 4);

	reset_fake_hierarchy();

	/* One Class+Comp, two MSC with overlapping features */
	fake_comp1.class = &fake_class;
	list_add(&fake_comp1.class_list, &fake_class.components);
	fake_comp2.class = NULL;
	fake_vmsc1.comp = &fake_comp1;
	list_add(&fake_vmsc1.comp_list, &fake_comp1.vmsc);
	fake_vmsc2.comp = &fake_comp1;
	list_add(&fake_vmsc2.comp_list, &fake_comp1.vmsc);
	fake_ris1.vmsc = &fake_vmsc1;
	list_add(&fake_ris1.vmsc_list, &fake_vmsc1.ris);
	fake_ris2.vmsc = &fake_vmsc2;
	list_add(&fake_ris2.vmsc_list, &fake_vmsc2.ris);

	mpam_set_feature(mpam_feat_cpor_part, &fake_ris1.props);
	mpam_set_feature(mpam_feat_cpor_part, &fake_ris2.props);
	fake_ris1.props.cpbm_wd = 4;
	fake_ris2.props.cpbm_wd = 4;

	mpam_enable_merge_features(&fake_classes_list);

	KUNIT_EXPECT_TRUE(test, mpam_has_feature(mpam_feat_cpor_part, &fake_class.props));
	KUNIT_EXPECT_EQ(test, fake_class.props.cpbm_wd, 4);

	reset_fake_hierarchy();

	/* One Class+Comp, two MSC with non-overlapping features */
	fake_comp1.class = &fake_class;
	list_add(&fake_comp1.class_list, &fake_class.components);
	fake_comp2.class = NULL;
	fake_vmsc1.comp = &fake_comp1;
	list_add(&fake_vmsc1.comp_list, &fake_comp1.vmsc);
	fake_vmsc2.comp = &fake_comp1;
	list_add(&fake_vmsc2.comp_list, &fake_comp1.vmsc);
	fake_ris1.vmsc = &fake_vmsc1;
	list_add(&fake_ris1.vmsc_list, &fake_vmsc1.ris);
	fake_ris2.vmsc = &fake_vmsc2;
	list_add(&fake_ris2.vmsc_list, &fake_vmsc2.ris);

	mpam_set_feature(mpam_feat_cpor_part, &fake_ris1.props);
	mpam_set_feature(mpam_feat_cmax_cmin, &fake_ris2.props);
	fake_ris1.props.cpbm_wd = 4;
	fake_ris2.props.cmax_wd = 4;

	mpam_enable_merge_features(&fake_classes_list);

	/*
	 * Multiple RIS in different MSC can't control the same resource,
	 * mismatched features can not be supported.
	 */
	KUNIT_EXPECT_FALSE(test, mpam_has_feature(mpam_feat_cpor_part, &fake_class.props));
	KUNIT_EXPECT_FALSE(test, mpam_has_feature(mpam_feat_cmax_cmin, &fake_class.props));
	KUNIT_EXPECT_EQ(test, fake_class.props.cpbm_wd, 0);
	KUNIT_EXPECT_EQ(test, fake_class.props.cmax_wd, 0);

	reset_fake_hierarchy();

	/* One Class+Comp, two MSC with incompatible overlapping features */
	fake_comp1.class = &fake_class;
	list_add(&fake_comp1.class_list, &fake_class.components);
	fake_comp2.class = NULL;
	fake_vmsc1.comp = &fake_comp1;
	list_add(&fake_vmsc1.comp_list, &fake_comp1.vmsc);
	fake_vmsc2.comp = &fake_comp1;
	list_add(&fake_vmsc2.comp_list, &fake_comp1.vmsc);
	fake_ris1.vmsc = &fake_vmsc1;
	list_add(&fake_ris1.vmsc_list, &fake_vmsc1.ris);
	fake_ris2.vmsc = &fake_vmsc2;
	list_add(&fake_ris2.vmsc_list, &fake_vmsc2.ris);

	mpam_set_feature(mpam_feat_cpor_part, &fake_ris1.props);
	mpam_set_feature(mpam_feat_cpor_part, &fake_ris2.props);
	mpam_set_feature(mpam_feat_mbw_part, &fake_ris1.props);
	mpam_set_feature(mpam_feat_mbw_part, &fake_ris2.props);
	fake_ris1.props.cpbm_wd = 5;
	fake_ris2.props.cpbm_wd = 3;
	fake_ris1.props.mbw_pbm_bits = 5;
	fake_ris2.props.mbw_pbm_bits = 3;

	mpam_enable_merge_features(&fake_classes_list);

	/*
	 * Multiple RIS in different MSC can't control the same resource,
	 * mismatched features can not be supported.
	 */
	KUNIT_EXPECT_FALSE(test, mpam_has_feature(mpam_feat_cpor_part, &fake_class.props));
	KUNIT_EXPECT_FALSE(test, mpam_has_feature(mpam_feat_mbw_part, &fake_class.props));
	KUNIT_EXPECT_EQ(test, fake_class.props.cpbm_wd, 0);
	KUNIT_EXPECT_EQ(test, fake_class.props.mbw_pbm_bits, 0);

	reset_fake_hierarchy();

	/* One Class+Comp, two MSC with overlapping features that need tweaking */
	fake_comp1.class = &fake_class;
	list_add(&fake_comp1.class_list, &fake_class.components);
	fake_comp2.class = NULL;
	fake_vmsc1.comp = &fake_comp1;
	list_add(&fake_vmsc1.comp_list, &fake_comp1.vmsc);
	fake_vmsc2.comp = &fake_comp1;
	list_add(&fake_vmsc2.comp_list, &fake_comp1.vmsc);
	fake_ris1.vmsc = &fake_vmsc1;
	list_add(&fake_ris1.vmsc_list, &fake_vmsc1.ris);
	fake_ris2.vmsc = &fake_vmsc2;
	list_add(&fake_ris2.vmsc_list, &fake_vmsc2.ris);

	mpam_set_feature(mpam_feat_mbw_min, &fake_ris1.props);
	mpam_set_feature(mpam_feat_mbw_min, &fake_ris2.props);
	mpam_set_feature(mpam_feat_cmax_cmax, &fake_ris1.props);
	mpam_set_feature(mpam_feat_cmax_cmax, &fake_ris2.props);
	fake_ris1.props.bwa_wd = 5;
	fake_ris2.props.bwa_wd = 3;
	fake_ris1.props.cmax_wd = 5;
	fake_ris2.props.cmax_wd = 3;

	mpam_enable_merge_features(&fake_classes_list);

	/*
	 * RIS with different control properties need to be sanitised so the
	 * class has the common set of properties.
	 */
	KUNIT_EXPECT_TRUE(test, mpam_has_feature(mpam_feat_mbw_min, &fake_class.props));
	KUNIT_EXPECT_TRUE(test, mpam_has_feature(mpam_feat_cmax_cmax, &fake_class.props));
	KUNIT_EXPECT_EQ(test, fake_class.props.bwa_wd, 3);
	KUNIT_EXPECT_EQ(test, fake_class.props.cmax_wd, 3);

	reset_fake_hierarchy();

	/* One Class Two Comp with overlapping features */
	fake_comp1.class = &fake_class;
	list_add(&fake_comp1.class_list, &fake_class.components);
	fake_comp2.class = &fake_class;
	list_add(&fake_comp2.class_list, &fake_class.components);
	fake_vmsc1.comp = &fake_comp1;
	list_add(&fake_vmsc1.comp_list, &fake_comp1.vmsc);
	fake_vmsc2.comp = &fake_comp2;
	list_add(&fake_vmsc2.comp_list, &fake_comp2.vmsc);
	fake_ris1.vmsc = &fake_vmsc1;
	list_add(&fake_ris1.vmsc_list, &fake_vmsc1.ris);
	fake_ris2.vmsc = &fake_vmsc2;
	list_add(&fake_ris2.vmsc_list, &fake_vmsc2.ris);

	mpam_set_feature(mpam_feat_cpor_part, &fake_ris1.props);
	mpam_set_feature(mpam_feat_cpor_part, &fake_ris2.props);
	fake_ris1.props.cpbm_wd = 4;
	fake_ris2.props.cpbm_wd = 4;

	mpam_enable_merge_features(&fake_classes_list);

	KUNIT_EXPECT_TRUE(test, mpam_has_feature(mpam_feat_cpor_part, &fake_class.props));
	KUNIT_EXPECT_EQ(test, fake_class.props.cpbm_wd, 4);

	reset_fake_hierarchy();

	/* One Class Two Comp with non-overlapping features */
	fake_comp1.class = &fake_class;
	list_add(&fake_comp1.class_list, &fake_class.components);
	fake_comp2.class = &fake_class;
	list_add(&fake_comp2.class_list, &fake_class.components);
	fake_vmsc1.comp = &fake_comp1;
	list_add(&fake_vmsc1.comp_list, &fake_comp1.vmsc);
	fake_vmsc2.comp = &fake_comp2;
	list_add(&fake_vmsc2.comp_list, &fake_comp2.vmsc);
	fake_ris1.vmsc = &fake_vmsc1;
	list_add(&fake_ris1.vmsc_list, &fake_vmsc1.ris);
	fake_ris2.vmsc = &fake_vmsc2;
	list_add(&fake_ris2.vmsc_list, &fake_vmsc2.ris);

	mpam_set_feature(mpam_feat_cpor_part, &fake_ris1.props);
	mpam_set_feature(mpam_feat_cmax_cmin, &fake_ris2.props);
	fake_ris1.props.cpbm_wd = 4;
	fake_ris2.props.cmax_wd = 4;

	mpam_enable_merge_features(&fake_classes_list);

	/*
	 * Multiple components can't control the same resource, mismatched features can
	 * not be supported.
	 */
	KUNIT_EXPECT_FALSE(test, mpam_has_feature(mpam_feat_cpor_part, &fake_class.props));
	KUNIT_EXPECT_FALSE(test, mpam_has_feature(mpam_feat_cmax_cmin, &fake_class.props));
	KUNIT_EXPECT_EQ(test, fake_class.props.cpbm_wd, 0);
	KUNIT_EXPECT_EQ(test, fake_class.props.cmax_wd, 0);

	mutex_unlock(&mpam_list_lock);
}

static void __test_mpam_reset_msc_bitmap(struct mpam_msc *msc, u16 reg, u16 wd)
{
	/* Avoid warnings when running with CONFIG_DEBUG_PREEMPT */
	guard(preempt)();

	mpam_reset_msc_bitmap(msc, reg, wd);
}

static void test_mpam_reset_msc_bitmap(struct kunit *test)
{
	char __iomem *buf = (__force char __iomem *)kunit_kzalloc(test, SZ_16K, GFP_KERNEL);
	struct mpam_msc fake_msc = {};
	u32 *test_result;

	if (!buf)
		return;

	fake_msc.mapped_hwpage = buf;
	fake_msc.mapped_hwpage_sz = SZ_16K;
	cpumask_copy(&fake_msc.accessibility, cpu_possible_mask);

	/* Satisfy lockdep checks */
	mutex_init(&fake_msc.part_sel_lock);
	mutex_lock(&fake_msc.part_sel_lock);

	test_result = (__force u32 *)(buf + MPAMCFG_CPBM);

	__test_mpam_reset_msc_bitmap(&fake_msc, MPAMCFG_CPBM, 0);
	KUNIT_EXPECT_EQ(test, test_result[0], 0);
	KUNIT_EXPECT_EQ(test, test_result[1], 0);
	test_result[0] = 0;
	test_result[1] = 0;

	__test_mpam_reset_msc_bitmap(&fake_msc, MPAMCFG_CPBM, 1);
	KUNIT_EXPECT_EQ(test, test_result[0], 1);
	KUNIT_EXPECT_EQ(test, test_result[1], 0);
	test_result[0] = 0;
	test_result[1] = 0;

	__test_mpam_reset_msc_bitmap(&fake_msc, MPAMCFG_CPBM, 16);
	KUNIT_EXPECT_EQ(test, test_result[0], 0xffff);
	KUNIT_EXPECT_EQ(test, test_result[1], 0);
	test_result[0] = 0;
	test_result[1] = 0;

	__test_mpam_reset_msc_bitmap(&fake_msc, MPAMCFG_CPBM, 32);
	KUNIT_EXPECT_EQ(test, test_result[0], 0xffffffff);
	KUNIT_EXPECT_EQ(test, test_result[1], 0);
	test_result[0] = 0;
	test_result[1] = 0;

	__test_mpam_reset_msc_bitmap(&fake_msc, MPAMCFG_CPBM, 33);
	KUNIT_EXPECT_EQ(test, test_result[0], 0xffffffff);
	KUNIT_EXPECT_EQ(test, test_result[1], 1);
	test_result[0] = 0;
	test_result[1] = 0;

	mutex_unlock(&fake_msc.part_sel_lock);
}

static struct kunit_case mpam_devices_test_cases[] = {
	KUNIT_CASE(test_mpam_reset_msc_bitmap),
	KUNIT_CASE(test_mpam_enable_merge_features),
	KUNIT_CASE(test__props_mismatch),
	{}
};

static struct kunit_suite mpam_devices_test_suite = {
	.name = "mpam_devices_test_suite",
	.test_cases = mpam_devices_test_cases,
};

kunit_test_suites(&mpam_devices_test_suite);
