[PATCH v2 06/16] iommu/dma: extend iommu_dma_*map_phys API to handle MMIO memory

[Leon Romanovsky <leon@xxxxxxxxxx>]

From: Leon Romanovsky <leonro@xxxxxxxxxx>

Combine iommu_dma_*map_phys with iommu_dma_*map_resource interfaces in
order to allow single phys_addr_t flow.

In the following patches, the iommu_dma_map_resource() will be removed
in favour of iommu_dma_map_phys(..., attrs | DMA_ATTR_MMIO) flow.

Signed-off-by: Leon Romanovsky <leonro@xxxxxxxxxx>
---
 drivers/iommu/dma-iommu.c | 15 +++++++++++----
 1 file changed, 11 insertions(+), 4 deletions(-)

diff --git a/drivers/iommu/dma-iommu.c b/drivers/iommu/dma-iommu.c
index aea119f32f96..6804aaf034a1 100644
--- a/drivers/iommu/dma-iommu.c
+++ b/drivers/iommu/dma-iommu.c
@@ -1211,16 +1211,19 @@ dma_addr_t iommu_dma_map_phys(struct device *dev, 
phys_addr_t phys, size_t size,
         */
        if (dev_use_swiotlb(dev, size, dir) &&
            iova_unaligned(iovad, phys, size)) {
+               if (attrs & DMA_ATTR_MMIO)
+                       return DMA_MAPPING_ERROR;
+
                phys = iommu_dma_map_swiotlb(dev, phys, size, dir, attrs);
                if (phys == (phys_addr_t)DMA_MAPPING_ERROR)
                        return DMA_MAPPING_ERROR;
        }
 
-       if (!coherent && !(attrs & DMA_ATTR_SKIP_CPU_SYNC))
+       if (!coherent && !(attrs & (DMA_ATTR_SKIP_CPU_SYNC | DMA_ATTR_MMIO)))
                arch_sync_dma_for_device(phys, size, dir);
 
        iova = __iommu_dma_map(dev, phys, size, prot, dma_mask);
-       if (iova == DMA_MAPPING_ERROR)
+       if (iova == DMA_MAPPING_ERROR && !(attrs & DMA_ATTR_MMIO))
                swiotlb_tbl_unmap_single(dev, phys, size, dir, attrs);
        return iova;
 }
@@ -1228,10 +1231,14 @@ dma_addr_t iommu_dma_map_phys(struct device *dev, 
phys_addr_t phys, size_t size,
 void iommu_dma_unmap_phys(struct device *dev, dma_addr_t dma_handle,
                size_t size, enum dma_data_direction dir, unsigned long attrs)
 {
-       struct iommu_domain *domain = iommu_get_dma_domain(dev);
        phys_addr_t phys;
 
-       phys = iommu_iova_to_phys(domain, dma_handle);
+       if (attrs & DMA_ATTR_MMIO) {
+               __iommu_dma_unmap(dev, dma_handle, size);
+               return;
+       }
+
+       phys = iommu_iova_to_phys(iommu_get_dma_domain(dev), dma_handle);
        if (WARN_ON(!phys))
                return;
 
-- 
2.50.1