From: Mark Kettenis <kettenis@openbsd.org>
Date: Sat, 21 Jan 2023 19:27:57 +0000 (+0100)
Subject: test: Add test for mapping IOMMUs for PCI devices
X-Git-Url: http://git.dujemihanovic.xyz/img/%7B%7B%20%24image.RelPermalink%20%7D%7D?a=commitdiff_plain;h=08386da0c65c5669a447f763ab4cd91d9efc7cd1;p=u-boot.git

test: Add test for mapping IOMMUs for PCI devices

Test that we correctly probe an IOMMU that is mapped by an
"iommu-map" device tree property of a PCIe controller node.

Signed-off-by: Mark Kettenis <kettenis@openbsd.org>
Reviewed-by: Simon Glass <sjg@chromium.org>
---

diff --git a/arch/sandbox/dts/test.dts b/arch/sandbox/dts/test.dts
index 9d96e479ca..8c05927670 100644
--- a/arch/sandbox/dts/test.dts
+++ b/arch/sandbox/dts/test.dts
@@ -1036,6 +1036,8 @@
 		#size-cells = <2>;
 		ranges = <0x02000000 0 0x10000000 0x10000000 0 0x2000000
 				0x01000000 0 0x20000000 0x20000000 0 0x2000>;
+		iommu-map = <0x0010 &iommu 0 1>;
+		iommu-map-mask = <0xfffffff8>;
 		pci@0,0 {
 			compatible = "pci-generic";
 			reg = <0x0000 0 0 0 0>;
diff --git a/test/dm/iommu.c b/test/dm/iommu.c
index 6f932ef233..62d38f1214 100644
--- a/test/dm/iommu.c
+++ b/test/dm/iommu.c
@@ -68,3 +68,33 @@ static int dm_test_iommu_noiommu(struct unit_test_state *uts)
 	return 0;
 }
 DM_TEST(dm_test_iommu_noiommu, UT_TESTF_SCAN_FDT);
+
+static int dm_test_iommu_pci(struct unit_test_state *uts)
+{
+	struct udevice *dev;
+
+	ut_assertok(uclass_find_device(UCLASS_IOMMU, 0, &dev));
+	ut_assert(!(dev_get_flags(dev) & DM_FLAG_ACTIVATED));
+
+	/* Probing P2SB probes the IOMMU through the "iommu-map" property */
+	ut_assertok(uclass_probe_all(UCLASS_P2SB));
+	ut_assert(dev_get_flags(dev) & DM_FLAG_ACTIVATED);
+
+	return 0;
+}
+DM_TEST(dm_test_iommu_pci, UT_TESTF_SCAN_FDT);
+
+static int dm_test_iommu_pci_noiommu(struct unit_test_state *uts)
+{
+	struct udevice *dev;
+
+	ut_assertok(uclass_find_device(UCLASS_IOMMU, 0, &dev));
+	ut_assert(!(dev_get_flags(dev) & DM_FLAG_ACTIVATED));
+
+	/* Probing PMC should not probe the IOMMU */
+	ut_assertok(uclass_probe_all(UCLASS_ACPI_PMC));
+	ut_assert(!(dev_get_flags(dev) & DM_FLAG_ACTIVATED));
+
+	return 0;
+}
+DM_TEST(dm_test_iommu_pci_noiommu, UT_TESTF_SCAN_FDT);