vfio: Split the register_device ops call into functions

This is a container item.

A following patch will move the vfio_container functions to their own .c
file.

Reviewed-by: Kevin Tian <kevin.tian@intel.com>
Signed-off-by: Jason Gunthorpe <jgg@nvidia.com>
Link: https://lore.kernel.org/r/7-v3-297af71838d2+b9-vfio_container_split_jgg@nvidia.com
Signed-off-by: Alex Williamson <alex.williamson@redhat.com>

diff --git a/drivers/vfio/vfio_main.c b/drivers/vfio/vfio_main.c
index 33e55e4..1ac7160 100644 (file)
--- a/drivers/vfio/vfio_main.c
+++ b/drivers/vfio/vfio_main.c
@@ -1226,9 +1226,28 @@ static void vfio_device_unassign_container(struct vfio_device *device)
        up_write(&device->group->group_rwsem);
 }
 
+static void vfio_device_container_register(struct vfio_device *device)
+{
+       struct vfio_iommu_driver *iommu_driver =
+               device->group->container->iommu_driver;
+
+       if (iommu_driver && iommu_driver->ops->register_device)
+               iommu_driver->ops->register_device(
+                       device->group->container->iommu_data, device);
+}
+
+static void vfio_device_container_unregister(struct vfio_device *device)
+{
+       struct vfio_iommu_driver *iommu_driver =
+               device->group->container->iommu_driver;
+
+       if (iommu_driver && iommu_driver->ops->unregister_device)
+               iommu_driver->ops->unregister_device(
+                       device->group->container->iommu_data, device);
+}
+
 static struct file *vfio_device_open(struct vfio_device *device)
 {
-       struct vfio_iommu_driver *iommu_driver;
        struct file *filep;
        int ret;
 
@@ -1259,12 +1278,7 @@ static struct file *vfio_device_open(struct vfio_device *device)
                        if (ret)
                                goto err_undo_count;
                }
-
-               iommu_driver = device->group->container->iommu_driver;
-               if (iommu_driver && iommu_driver->ops->register_device)
-                       iommu_driver->ops->register_device(
-                               device->group->container->iommu_data, device);
-
+               vfio_device_container_register(device);
                up_read(&device->group->group_rwsem);
        }
        mutex_unlock(&device->dev_set->lock);
@@ -1302,10 +1316,7 @@ err_close_device:
        if (device->open_count == 1 && device->ops->close_device) {
                device->ops->close_device(device);
 
-               iommu_driver = device->group->container->iommu_driver;
-               if (iommu_driver && iommu_driver->ops->unregister_device)
-                       iommu_driver->ops->unregister_device(
-                               device->group->container->iommu_data, device);
+               vfio_device_container_unregister(device);
        }
 err_undo_count:
        up_read(&device->group->group_rwsem);
@@ -1513,7 +1524,6 @@ static inline void vfio_device_pm_runtime_put(struct vfio_device *device)
 static int vfio_device_fops_release(struct inode *inode, struct file *filep)
 {
        struct vfio_device *device = filep->private_data;
-       struct vfio_iommu_driver *iommu_driver;
 
        mutex_lock(&device->dev_set->lock);
        vfio_assert_device_open(device);
@@ -1521,10 +1531,7 @@ static int vfio_device_fops_release(struct inode *inode, struct file *filep)
        if (device->open_count == 1 && device->ops->close_device)
                device->ops->close_device(device);
 
-       iommu_driver = device->group->container->iommu_driver;
-       if (iommu_driver && iommu_driver->ops->unregister_device)
-               iommu_driver->ops->unregister_device(
-                       device->group->container->iommu_data, device);
+       vfio_device_container_unregister(device);
        up_read(&device->group->group_rwsem);
        device->open_count--;
        if (device->open_count == 0)