iommu-api: Add ->{un}map callbacks to iommu_ops

This patch adds new callbacks for mapping and unmapping
pages to the iommu_ops structure. These callbacks are aware
of page sizes which makes them different to the
->{un}map_range callbacks.

Signed-off-by: Joerg Roedel <joerg.roedel@amd.com>

2 files changed, 10 insertions, 0 deletions

diff --git a/drivers/base/iommu.c b/drivers/base/iommu.c
index cf7cbec116e..55d37e4609e 100644
--- a/drivers/base/iommu.c
+++ b/drivers/base/iommu.c
@@ -119,6 +119,9 @@ int iommu_map(struct iommu_domain *domain, unsigned long iova,
 
         BUG_ON((iova | paddr) & invalid_mask);
 
+        if (iommu_ops->map)
+                return iommu_ops->map(domain, iova, paddr, gfp_order, prot);
+
         return iommu_ops->map_range(domain, iova, paddr, size, prot);
 }
 EXPORT_SYMBOL_GPL(iommu_map);
@@ -133,6 +136,9 @@ int iommu_unmap(struct iommu_domain *domain, unsigned long iova, int gfp_order)
 
         BUG_ON(iova & invalid_mask);
 
+        if (iommu_ops->unmap)
+                return iommu_ops->unmap(domain, iova, gfp_order);
+
         iommu_ops->unmap_range(domain, iova, size);
 
         return gfp_order;
diff --git a/include/linux/iommu.h b/include/linux/iommu.h
index 6d0035bb1a0..5a7a3d888da 100644
--- a/include/linux/iommu.h
+++ b/include/linux/iommu.h
@@ -36,6 +36,10 @@ struct iommu_ops {
         void (*domain_destroy)(struct iommu_domain *domain);
         int (*attach_dev)(struct iommu_domain *domain, struct device *dev);
         void (*detach_dev)(struct iommu_domain *domain, struct device *dev);
+        int (*map)(struct iommu_domain *domain, unsigned long iova,
+                   phys_addr_t paddr, int gfp_order, int prot);
+        int (*unmap)(struct iommu_domain *domain, unsigned long iova,
+                     int gfp_order);
         int (*map_range)(struct iommu_domain *domain, unsigned long iova,
                          phys_addr_t paddr, size_t size, int prot);
         void (*unmap_range)(struct iommu_domain *domain, unsigned long iova,