[PATCH v2 09/13] iommufd/selftest: Add IOMMU_TEST_OP_TRIGGER_VIRQ for vIRQ coverage

[Nicolin Chen]

The handler will get vDEVICE object from the given mdev and convert it to
its per-vIOMMU virtual 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 a6b7a163f636..3037904f2e52 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;
 };
@@ -212,4 +216,10 @@ struct iommu_viommu_invalidate_selftest {
        __u32 cache_id;
 };
 
+#define IOMMU_VIRQ_TYPE_SELFTEST 0xbeefbeef
+
+struct iommu_viommu_irq_selftest {
+       __u32 virt_id;
+};
+
 #endif
diff --git a/drivers/iommu/iommufd/selftest.c b/drivers/iommu/iommufd/selftest.c
index d1438d81e664..0785c9447102 100644
--- a/drivers/iommu/iommufd/selftest.c
+++ b/drivers/iommu/iommufd/selftest.c
@@ -1631,6 +1631,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 iommu_viommu_irq_selftest test = {};
+       struct iommufd_device *idev;
+       struct mock_dev *mdev;
+       int rc = -ENOENT;
+
+       idev = iommufd_get_device(ucmd, cmd->trigger_virq.dev_id);
+       if (IS_ERR(idev))
+               return PTR_ERR(idev);
+       mdev = to_mock_dev(idev->dev);
+
+       down_read(&mdev->viommu_rwsem);
+       if (!mdev->viommu || !mdev->vdev_id)
+               goto out_unlock;
+
+       test.virt_id = mdev->vdev_id;
+       rc = iommufd_viommu_report_irq(&mdev->viommu->core,
+                                      IOMMU_VIRQ_TYPE_SELFTEST, &test,
+                                      sizeof(test));
+out_unlock:
+       up_read(&mdev->viommu_rwsem);
+       iommufd_put_object(ucmd->ictx, &idev->obj);
+
+       return rc;
+}
+
 void iommufd_selftest_destroy(struct iommufd_object *obj)
 {
        struct selftest_obj *sobj = to_selftest_obj(obj);
@@ -1712,6 +1740,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