[PATCH v2 6/9] test: Add test for mapping IOMMUs for PCI devices

[Mark Kettenis]

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 at openbsd.org>
---
 arch/sandbox/dts/test.dts |  2 ++
 test/dm/iommu.c           | 30 ++++++++++++++++++++++++++++++
 2 files changed, 32 insertions(+)

diff --git a/arch/sandbox/dts/test.dts b/arch/sandbox/dts/test.dts
index dffe10adbf..b576b61a8c 100644
--- a/arch/sandbox/dts/test.dts
+++ b/arch/sandbox/dts/test.dts
@@ -1025,6 +1025,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 at 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);
-- 
2.39.0