Implement iova_to_phys_length for Spreadtrum IOMMU driver,
returning the actual PTE mapping size.

Signed-off-by: Guanghui Feng <[email protected]>
---
 drivers/iommu/sprd-iommu.c | 20 +++++++++++++++-----
 1 file changed, 15 insertions(+), 5 deletions(-)

diff --git a/drivers/iommu/sprd-iommu.c b/drivers/iommu/sprd-iommu.c
index c1a34445d244..f8dbe24c332a 100644
--- a/drivers/iommu/sprd-iommu.c
+++ b/drivers/iommu/sprd-iommu.c
@@ -366,8 +366,9 @@ static void sprd_iommu_sync(struct iommu_domain *domain,
        sprd_iommu_sync_map(domain, 0, 0);
 }
 
-static phys_addr_t sprd_iommu_iova_to_phys(struct iommu_domain *domain,
-                                          dma_addr_t iova)
+static phys_addr_t sprd_iommu_iova_to_phys_length(struct iommu_domain *domain,
+                                                 dma_addr_t iova,
+                                                 size_t *mapped_length)
 {
        struct sprd_iommu_domain *dom = to_sprd_domain(domain);
        unsigned long flags;
@@ -375,14 +376,23 @@ static phys_addr_t sprd_iommu_iova_to_phys(struct 
iommu_domain *domain,
        unsigned long start = domain->geometry.aperture_start;
        unsigned long end = domain->geometry.aperture_end;
 
+       if (mapped_length)
+               *mapped_length = 0;
+
        if (WARN_ON(iova < start || iova > end))
-               return 0;
+               return PHYS_ADDR_MAX;
 
        spin_lock_irqsave(&dom->pgtlock, flags);
        pa = *(dom->pgt_va + ((iova - start) >> SPRD_IOMMU_PAGE_SHIFT));
-       pa = (pa << SPRD_IOMMU_PAGE_SHIFT) + ((iova - start) & 
(SPRD_IOMMU_PAGE_SIZE - 1));
        spin_unlock_irqrestore(&dom->pgtlock, flags);
 
+       if (!pa)
+               return PHYS_ADDR_MAX;
+
+       pa = (pa << SPRD_IOMMU_PAGE_SHIFT) + ((iova - start) & 
(SPRD_IOMMU_PAGE_SIZE - 1));
+       if (mapped_length)
+               *mapped_length = SPRD_IOMMU_PAGE_SIZE;
+
        return pa;
 }
 
@@ -420,7 +430,7 @@ static const struct iommu_ops sprd_iommu_ops = {
                .unmap_pages    = sprd_iommu_unmap,
                .iotlb_sync_map = sprd_iommu_sync_map,
                .iotlb_sync     = sprd_iommu_sync,
-               .iova_to_phys   = sprd_iommu_iova_to_phys,
+               .iova_to_phys_length    = sprd_iommu_iova_to_phys_length,
                .free           = sprd_iommu_domain_free,
        }
 };
-- 
2.43.7

Reply via email to