From: Si-Wei Liu <si-wei....@oracle.com>

When calculating the physical address range based on the iotlb and mr
[start,end) ranges, the offset of mr->start relative to map->start
is not taken into account. This leads to some incorrect and duplicate
mappings.

For the case when mr->start < map->start the code is already correct:
the range in [mr->start, map->start) was handled by a different
iteration.

Fixes: 94abbccdf291 ("vdpa/mlx5: Add shared memory registration code")
Cc: sta...@vger.kernel.org
Signed-off-by: Si-Wei Liu <si-wei....@oracle.com>
Signed-off-by: Dragos Tatulea <dtatu...@nvidia.com>
---
 drivers/vdpa/mlx5/core/mr.c | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/drivers/vdpa/mlx5/core/mr.c b/drivers/vdpa/mlx5/core/mr.c
index 2dd21e0b399e..7d0c83b5b071 100644
--- a/drivers/vdpa/mlx5/core/mr.c
+++ b/drivers/vdpa/mlx5/core/mr.c
@@ -373,7 +373,7 @@ static int map_direct_mr(struct mlx5_vdpa_dev *mvdev, 
struct mlx5_vdpa_direct_mr
        struct page *pg;
        unsigned int nsg;
        int sglen;
-       u64 pa;
+       u64 pa, offset;
        u64 paend;
        struct scatterlist *sg;
        struct device *dma = mvdev->vdev.dma_dev;
@@ -396,8 +396,10 @@ static int map_direct_mr(struct mlx5_vdpa_dev *mvdev, 
struct mlx5_vdpa_direct_mr
        sg = mr->sg_head.sgl;
        for (map = vhost_iotlb_itree_first(iotlb, mr->start, mr->end - 1);
             map; map = vhost_iotlb_itree_next(map, mr->start, mr->end - 1)) {
-               paend = map->addr + maplen(map, mr);
-               for (pa = map->addr; pa < paend; pa += sglen) {
+               offset = mr->start > map->start ? mr->start - map->start : 0;
+               pa = map->addr + offset;
+               paend = map->addr + offset + maplen(map, mr);
+               for (; pa < paend; pa += sglen) {
                        pg = pfn_to_page(__phys_to_pfn(pa));
                        if (!sg) {
                                mlx5_vdpa_warn(mvdev, "sg null. start 0x%llx, 
end 0x%llx\n",
-- 
2.46.1


Reply via email to