iommu/exynos: Add devices attached to the System MMU to an IOMMU group

Patch written by Antonios Motakis <a.motakis@virtualopensystems.com>:

IOMMU groups are expected by certain users of the IOMMU API,
e.g. VFIO. Since each device is behind its own System MMU, we
can allocate a new IOMMU group for each device.

Reviewed-by: Cho KyongHo <pullip.cho@samsung.com>
Signed-off-by: Antonios Motakis <a.motakis@virtualopensystems.com>
Signed-off-by: Shaik Ameeer Basha <shaik.ameer@samsung.com>
Signed-off-by: Joerg Roedel <jroedel@suse.de>

1 files changed, 28 insertions, 0 deletions

diff --git a/drivers/iommu/exynos-iommu.c b/drivers/iommu/exynos-iommu.c
index 3291619c9db0..d18dc379f536 100644
--- a/drivers/iommu/exynos-iommu.c
+++ b/drivers/iommu/exynos-iommu.c
@@ -964,6 +964,32 @@ static phys_addr_t exynos_iommu_iova_to_phys(struct iommu_domain *domain,
         return phys;
 }
 
+static int exynos_iommu_add_device(struct device *dev)
+{
+        struct iommu_group *group;
+        int ret;
+
+        group = iommu_group_get(dev);
+
+        if (!group) {
+                group = iommu_group_alloc();
+                if (IS_ERR(group)) {
+                        dev_err(dev, "Failed to allocate IOMMU group\n");
+                        return PTR_ERR(group);
+                }
+        }
+
+        ret = iommu_group_add_device(group, dev);
+        iommu_group_put(group);
+
+        return ret;
+}
+
+static void exynos_iommu_remove_device(struct device *dev)
+{
+        iommu_group_remove_device(dev);
+}
+
 static struct iommu_ops exynos_iommu_ops = {
         .domain_init = &exynos_iommu_domain_init,
         .domain_destroy = &exynos_iommu_domain_destroy,
@@ -972,6 +998,8 @@ static struct iommu_ops exynos_iommu_ops = {
         .map = &exynos_iommu_map,
         .unmap = &exynos_iommu_unmap,
         .iova_to_phys = &exynos_iommu_iova_to_phys,
+        .add_device = &exynos_iommu_add_device,
+        .remove_device = &exynos_iommu_remove_device,
         .pgsize_bitmap = SECT_SIZE | LPAGE_SIZE | SPAGE_SIZE,
 };