1 files changed, 3 insertions, 4 deletions

diff --git a/drivers/vfio/pci/vfio_pci.c b/drivers/vfio/pci/vfio_pci.c
index 6d369fe9d30b..6c119944bbb6 100644
--- a/drivers/vfio/pci/vfio_pci.c
+++ b/drivers/vfio/pci/vfio_pci.c
@@ -408,7 +408,7 @@ static int vfio_pci_mmap(void *device_data, struct vm_area_struct *vma)
         struct vfio_pci_device *vdev = device_data;
         struct pci_dev *pdev = vdev->pdev;
         unsigned int index;
-        u64 phys_len, req_len, pgoff, req_start, phys;
+        u64 phys_len, req_len, pgoff, req_start;
         int ret;
 
         index = vma->vm_pgoff >> (VFIO_PCI_OFFSET_SHIFT - PAGE_SHIFT);
@@ -463,10 +463,9 @@ static int vfio_pci_mmap(void *device_data, struct vm_area_struct *vma)
         vma->vm_private_data = vdev;
         vma->vm_flags |= VM_IO | VM_DONTEXPAND | VM_DONTDUMP;
         vma->vm_page_prot = pgprot_noncached(vma->vm_page_prot);
+        vma->vm_pgoff = (pci_resource_start(pdev, index) >> PAGE_SHIFT) + pgoff;
 
-        phys = (pci_resource_start(pdev, index) >> PAGE_SHIFT) + pgoff;
-
-        return remap_pfn_range(vma, vma->vm_start, phys,
+        return remap_pfn_range(vma, vma->vm_start, vma->vm_pgoff,
                                req_len, vma->vm_page_prot);
 }