We always use the smallest supported page size of vfio_iommu as
pgsize. Drop parameter "pgsize" of update_user_bitmap.

Signed-off-by: Keqian Zhu <[email protected]>
---
 drivers/vfio/vfio_iommu_type1.c | 9 ++++-----
 1 file changed, 4 insertions(+), 5 deletions(-)

diff --git a/drivers/vfio/vfio_iommu_type1.c b/drivers/vfio/vfio_iommu_type1.c
index 2d7a5cd9b916..edb0a6468e8d 100644
--- a/drivers/vfio/vfio_iommu_type1.c
+++ b/drivers/vfio/vfio_iommu_type1.c
@@ -989,10 +989,9 @@ static void vfio_update_pgsize_bitmap(struct vfio_iommu 
*iommu)
 }
 
 static int update_user_bitmap(u64 __user *bitmap, struct vfio_iommu *iommu,
-                             struct vfio_dma *dma, dma_addr_t base_iova,
-                             size_t pgsize)
+                             struct vfio_dma *dma, dma_addr_t base_iova)
 {
-       unsigned long pgshift = __ffs(pgsize);
+       unsigned long pgshift = __ffs(iommu->pgsize_bitmap);
        unsigned long nbits = dma->size >> pgshift;
        unsigned long bit_offset = (dma->iova - base_iova) >> pgshift;
        unsigned long copy_offset = bit_offset / BITS_PER_LONG;
@@ -1057,7 +1056,7 @@ static int vfio_iova_dirty_bitmap(u64 __user *bitmap, 
struct vfio_iommu *iommu,
                if (dma->iova > iova + size - 1)
                        break;
 
-               ret = update_user_bitmap(bitmap, iommu, dma, iova, pgsize);
+               ret = update_user_bitmap(bitmap, iommu, dma, iova);
                if (ret)
                        return ret;
 
@@ -1203,7 +1202,7 @@ static int vfio_dma_do_unmap(struct vfio_iommu *iommu,
 
                if (unmap->flags & VFIO_DMA_UNMAP_FLAG_GET_DIRTY_BITMAP) {
                        ret = update_user_bitmap(bitmap->data, iommu, dma,
-                                                unmap->iova, pgsize);
+                                                unmap->iova);
                        if (ret)
                                break;
                }
-- 
2.23.0

Reply via email to