The following APIs are provided for translating user pfn to host pfn in a VFIO
 driver::
 
-       extern int vfio_pin_pages(struct device *dev, unsigned long *user_pfn,
+       int vfio_pin_pages(struct vfio_device *device, unsigned long *user_pfn,
                                  int npage, int prot, unsigned long *phys_pfn);
 
-       extern int vfio_unpin_pages(struct device *dev, unsigned long *user_pfn,
+       int vfio_unpin_pages(struct vfio_device *device, unsigned long *user_pfn,
                                    int npage);
 
 These functions call back into the back-end IOMMU module by using the pin_pages
 
 {
        int ret = 0;
 
-       ret = vfio_pin_pages(vdev->dev, pa->pa_iova_pfn, pa->pa_nr,
+       ret = vfio_pin_pages(vdev, pa->pa_iova_pfn, pa->pa_nr,
                             IOMMU_READ | IOMMU_WRITE, pa->pa_pfn);
 
        if (ret < 0) {
                goto err_out;
        } else if (ret > 0 && ret != pa->pa_nr) {
-               vfio_unpin_pages(vdev->dev, pa->pa_iova_pfn, ret);
+               vfio_unpin_pages(vdev, pa->pa_iova_pfn, ret);
                ret = -EINVAL;
                goto err_out;
        }
 {
        /* Only unpin if any pages were pinned to begin with */
        if (pa->pa_nr)
-               vfio_unpin_pages(vdev->dev, pa->pa_iova_pfn, pa->pa_nr);
+               vfio_unpin_pages(vdev, pa->pa_iova_pfn, pa->pa_nr);
        pa->pa_nr = 0;
        kfree(pa->pa_iova_pfn);
 }
 
                q->saved_isc = VFIO_AP_ISC_INVALID;
        }
        if (q->saved_pfn && !WARN_ON(!q->matrix_mdev)) {
-               vfio_unpin_pages(mdev_dev(q->matrix_mdev->mdev),
-                                &q->saved_pfn, 1);
+               vfio_unpin_pages(&q->matrix_mdev->vdev, &q->saved_pfn, 1);
                q->saved_pfn = 0;
        }
 }
                return status;
        }
 
-       ret = vfio_pin_pages(mdev_dev(q->matrix_mdev->mdev), &g_pfn, 1,
+       ret = vfio_pin_pages(&q->matrix_mdev->vdev, &g_pfn, 1,
                             IOMMU_READ | IOMMU_WRITE, &h_pfn);
        switch (ret) {
        case 1:
                break;
        case AP_RESPONSE_OTHERWISE_CHANGED:
                /* We could not modify IRQ setings: clear new configuration */
-               vfio_unpin_pages(mdev_dev(q->matrix_mdev->mdev), &g_pfn, 1);
+               vfio_unpin_pages(&q->matrix_mdev->vdev, &g_pfn, 1);
                kvm_s390_gisc_unregister(kvm, isc);
                break;
        default:
                struct vfio_iommu_type1_dma_unmap *unmap = data;
                unsigned long g_pfn = unmap->iova >> PAGE_SHIFT;
 
-               vfio_unpin_pages(mdev_dev(matrix_mdev->mdev), &g_pfn, 1);
+               vfio_unpin_pages(&matrix_mdev->vdev, &g_pfn, 1);
                return NOTIFY_OK;
        }
 
 
 /*
  * Pin a set of guest PFNs and return their associated host PFNs for local
  * domain only.
- * @dev [in]     : device
+ * @device [in]  : device
  * @user_pfn [in]: array of user/guest PFNs to be pinned.
  * @npage [in]   : count of elements in user_pfn array.  This count should not
  *                be greater VFIO_PIN_PAGES_MAX_ENTRIES.
  * @phys_pfn[out]: array of host PFNs
  * Return error or number of pages pinned.
  */
-int vfio_pin_pages(struct device *dev, unsigned long *user_pfn, int npage,
-                  int prot, unsigned long *phys_pfn)
+int vfio_pin_pages(struct vfio_device *device, unsigned long *user_pfn,
+                  int npage, int prot, unsigned long *phys_pfn)
 {
        struct vfio_container *container;
-       struct vfio_group *group;
+       struct vfio_group *group = device->group;
        struct vfio_iommu_driver *driver;
        int ret;
 
-       if (!dev || !user_pfn || !phys_pfn || !npage)
+       if (!user_pfn || !phys_pfn || !npage)
                return -EINVAL;
 
        if (npage > VFIO_PIN_PAGES_MAX_ENTRIES)
                return -E2BIG;
 
-       group = vfio_group_get_from_dev(dev);
-       if (!group)
-               return -ENODEV;
-
-       if (group->dev_counter > 1) {
-               ret = -EINVAL;
-               goto err_pin_pages;
-       }
+       if (group->dev_counter > 1)
+               return -EINVAL;
 
        ret = vfio_group_add_container_user(group);
        if (ret)
-               goto err_pin_pages;
+               return ret;
 
        container = group->container;
        driver = container->iommu_driver;
 
        vfio_group_try_dissolve_container(group);
 
-err_pin_pages:
-       vfio_group_put(group);
        return ret;
 }
 EXPORT_SYMBOL(vfio_pin_pages);
 
 /*
  * Unpin set of host PFNs for local domain only.
- * @dev [in]     : device
+ * @device [in]  : device
  * @user_pfn [in]: array of user/guest PFNs to be unpinned. Number of user/guest
  *                PFNs should not be greater than VFIO_PIN_PAGES_MAX_ENTRIES.
  * @npage [in]   : count of elements in user_pfn array.  This count should not
  *                 be greater than VFIO_PIN_PAGES_MAX_ENTRIES.
  * Return error or number of pages unpinned.
  */
-int vfio_unpin_pages(struct device *dev, unsigned long *user_pfn, int npage)
+int vfio_unpin_pages(struct vfio_device *device, unsigned long *user_pfn,
+                    int npage)
 {
        struct vfio_container *container;
-       struct vfio_group *group;
        struct vfio_iommu_driver *driver;
        int ret;
 
-       if (!dev || !user_pfn || !npage)
+       if (!user_pfn || !npage)
                return -EINVAL;
 
        if (npage > VFIO_PIN_PAGES_MAX_ENTRIES)
                return -E2BIG;
 
-       group = vfio_group_get_from_dev(dev);
-       if (!group)
-               return -ENODEV;
-
-       ret = vfio_group_add_container_user(group);
+       ret = vfio_group_add_container_user(device->group);
        if (ret)
-               goto err_unpin_pages;
+               return ret;
 
-       container = group->container;
+       container = device->group->container;
        driver = container->iommu_driver;
        if (likely(driver && driver->ops->unpin_pages))
                ret = driver->ops->unpin_pages(container->iommu_data, user_pfn,
        else
                ret = -ENOTTY;
 
-       vfio_group_try_dissolve_container(group);
+       vfio_group_try_dissolve_container(device->group);
 
-err_unpin_pages:
-       vfio_group_put(group);
        return ret;
 }
 EXPORT_SYMBOL(vfio_unpin_pages);
 
 
 #define VFIO_PIN_PAGES_MAX_ENTRIES     (PAGE_SIZE/sizeof(unsigned long))
 
-extern int vfio_pin_pages(struct device *dev, unsigned long *user_pfn,
+extern int vfio_pin_pages(struct vfio_device *device, unsigned long *user_pfn,
                          int npage, int prot, unsigned long *phys_pfn);
-extern int vfio_unpin_pages(struct device *dev, unsigned long *user_pfn,
+extern int vfio_unpin_pages(struct vfio_device *device, unsigned long *user_pfn,
                            int npage);
 
 extern int vfio_group_pin_pages(struct vfio_group *group,