[PATCH v4 08/14] iommufd/viommu: Add vdev_to_dev helper

[Nicolin Chen]

This avoids a bigger trouble of moving the struct iommufd_device to the
public header.

Reviewed-by: Jason Gunthorpe <j...@nvidia.com>
Signed-off-by: Nicolin Chen <nicol...@nvidia.com>
---
 include/linux/iommufd.h        | 6 ++++++
 drivers/iommu/iommufd/driver.c | 7 +++++++
 2 files changed, 13 insertions(+)

diff --git a/include/linux/iommufd.h b/include/linux/iommufd.h
index 559f274a26ea..2f4ec6e6df21 100644
--- a/include/linux/iommufd.h
+++ b/include/linux/iommufd.h
@@ -204,6 +204,7 @@ static inline int iommufd_vfio_compat_set_no_iommu(struct 
iommufd_ctx *ictx)
 struct iommufd_object *_iommufd_object_alloc(struct iommufd_ctx *ictx,
                                             size_t size,
                                             enum iommufd_object_type type);
+struct device *vdev_to_dev(struct iommufd_vdevice *vdev);
 #else /* !CONFIG_IOMMUFD_DRIVER */
 static inline struct iommufd_object *
 _iommufd_object_alloc(struct iommufd_ctx *ictx, size_t size,
@@ -211,6 +212,11 @@ _iommufd_object_alloc(struct iommufd_ctx *ictx, size_t 
size,
 {
        return ERR_PTR(-EOPNOTSUPP);
 }
+
+static inline struct device *vdev_to_dev(struct iommufd_vdevice *vdev)
+{
+       return NULL;
+}
 #endif /* CONFIG_IOMMUFD_DRIVER */
 
 /*
diff --git a/drivers/iommu/iommufd/driver.c b/drivers/iommu/iommufd/driver.c
index c0876d3f91c7..a5d750d2cfaa 100644
--- a/drivers/iommu/iommufd/driver.c
+++ b/drivers/iommu/iommufd/driver.c
@@ -36,3 +36,10 @@ struct iommufd_object *_iommufd_object_alloc(struct 
iommufd_ctx *ictx,
        return ERR_PTR(rc);
 }
 EXPORT_SYMBOL_NS_GPL(_iommufd_object_alloc, IOMMUFD);
+
+/* Caller should xa_lock(&viommu->vdevs) to protect the return value */
+struct device *vdev_to_dev(struct iommufd_vdevice *vdev)
+{
+       return vdev ? vdev->idev->dev : NULL;
+}
+EXPORT_SYMBOL_NS_GPL(vdev_to_dev, IOMMUFD);
-- 
2.43.0