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