[PATCH v1 08/10] iommufd/selftest: Add IOMMU_TEST_OP_TRIGGER_VIRQ for VIRQ coverage

[Nicolin Chen]

The handler will get vdev_id structure from the given mdev and convert it
to its per-viommu virtual device ID to mimic a real IOMMU driver.

Signed-off-by: Nicolin Chen <nicol...@nvidia.com>
---
 drivers/iommu/iommufd/iommufd_test.h | 10 ++++++++++
 drivers/iommu/iommufd/selftest.c     | 30 ++++++++++++++++++++++++++++
 2 files changed, 40 insertions(+)

diff --git a/drivers/iommu/iommufd/iommufd_test.h 
b/drivers/iommu/iommufd/iommufd_test.h
index 56bade6146ff..736ae5f8152e 100644
--- a/drivers/iommu/iommufd/iommufd_test.h
+++ b/drivers/iommu/iommufd/iommufd_test.h
@@ -24,6 +24,7 @@ enum {
        IOMMU_TEST_OP_MD_CHECK_IOTLB,
        IOMMU_TEST_OP_TRIGGER_IOPF,
        IOMMU_TEST_OP_DEV_CHECK_CACHE,
+       IOMMU_TEST_OP_TRIGGER_VIRQ,
 };
 
 enum {
@@ -145,6 +146,9 @@ struct iommu_test_cmd {
                        __u32 id;
                        __u32 cache;
                } check_dev_cache;
+               struct {
+                       __u32 dev_id;
+               } trigger_virq;
        };
        __u32 last;
 };
@@ -210,4 +214,10 @@ struct iommu_viommu_invalidate_selftest {
        __u32 cache_id;
 };
 
+#define IOMMU_VIRQ_TYPE_SELFTEST 0xbeefbeef
+
+struct iommu_viommu_irq_selftest {
+       __u32 vdev_id;
+};
+
 #endif
diff --git a/drivers/iommu/iommufd/selftest.c b/drivers/iommu/iommufd/selftest.c
index ea2861d34b4a..75fccd466018 100644
--- a/drivers/iommu/iommufd/selftest.c
+++ b/drivers/iommu/iommufd/selftest.c
@@ -1560,6 +1560,34 @@ static int iommufd_test_trigger_iopf(struct iommufd_ucmd 
*ucmd,
        return 0;
 }
 
+static int iommufd_test_trigger_virq(struct iommufd_ucmd *ucmd,
+                                    struct iommu_test_cmd *cmd)
+{
+       struct iommufd_device *idev;
+       struct mock_dev *mdev;
+
+       idev = iommufd_get_device(ucmd, cmd->trigger_virq.dev_id);
+       if (IS_ERR(idev))
+               return PTR_ERR(idev);
+       mdev = container_of(idev->dev, struct mock_dev, dev);
+
+       mutex_lock(&mdev->lock);
+       if (mdev->vdev_id) {
+               struct iommu_viommu_irq_selftest test = {
+                       .vdev_id = mdev->vdev_id->id,
+               };
+
+               iommufd_viommu_report_irq(mdev->vdev_id->viommu,
+                                         IOMMU_VIRQ_TYPE_SELFTEST,
+                                         &test, sizeof(test));
+       }
+       mutex_unlock(&mdev->lock);
+
+       iommufd_put_object(ucmd->ictx, &idev->obj);
+
+       return 0;
+}
+
 void iommufd_selftest_destroy(struct iommufd_object *obj)
 {
        struct selftest_obj *sobj = container_of(obj, struct selftest_obj, obj);
@@ -1641,6 +1669,8 @@ int iommufd_test(struct iommufd_ucmd *ucmd)
                                          cmd->dirty.flags);
        case IOMMU_TEST_OP_TRIGGER_IOPF:
                return iommufd_test_trigger_iopf(ucmd, cmd);
+       case IOMMU_TEST_OP_TRIGGER_VIRQ:
+               return iommufd_test_trigger_virq(ucmd, cmd);
        default:
                return -EOPNOTSUPP;
        }
-- 
2.43.0