Trigger a vIRQ giving an idev ID, to test the loopback whether receiving
or not the vdev_id that was set to the idev by the line above.

Signed-off-by: Nicolin Chen <nicol...@nvidia.com>
---
 tools/testing/selftests/iommu/iommufd_utils.h | 63 +++++++++++++++++++
 tools/testing/selftests/iommu/iommufd.c       | 22 +++++++
 .../selftests/iommu/iommufd_fail_nth.c        |  6 ++
 3 files changed, 91 insertions(+)

diff --git a/tools/testing/selftests/iommu/iommufd_utils.h 
b/tools/testing/selftests/iommu/iommufd_utils.h
index d979f5b0efe8..9f03955cb198 100644
--- a/tools/testing/selftests/iommu/iommufd_utils.h
+++ b/tools/testing/selftests/iommu/iommufd_utils.h
@@ -9,6 +9,7 @@
 #include <sys/ioctl.h>
 #include <stdint.h>
 #include <assert.h>
+#include <poll.h>
 
 #include "../kselftest_harness.h"
 #include "../../../../drivers/iommu/iommufd/iommufd_test.h"
@@ -936,3 +937,65 @@ static int _test_cmd_vdevice_alloc(int fd, __u32 
viommu_id, __u32 idev_id,
        EXPECT_ERRNO(_errno,                                                 \
                     _test_cmd_vdevice_alloc(self->fd, viommu_id, idev_id,   \
                                             virt_id, vdev_id))
+
+static int _test_cmd_virq_alloc(int fd, __u32 viommu_id, __u32 type,
+                               __u32 *virq_id, __u32 *virq_fd)
+{
+       struct iommu_virq_alloc cmd = {
+               .size = sizeof(cmd),
+               .type = type,
+               .viommu_id = viommu_id,
+       };
+       int ret;
+
+       ret = ioctl(fd, IOMMU_VIRQ_ALLOC, &cmd);
+       if (ret)
+               return ret;
+       if (virq_id)
+               *virq_id = cmd.out_virq_id;
+       if (virq_fd)
+               *virq_fd = cmd.out_virq_fd;
+       return 0;
+}
+
+#define test_cmd_virq_alloc(viommu_id, type, virq_id, virq_fd)                \
+       ASSERT_EQ(0, _test_cmd_virq_alloc(self->fd, viommu_id, type, virq_id, \
+                                         virq_fd))
+#define test_err_virq_alloc(_errno, viommu_id, type, virq_id, virq_fd)       \
+       EXPECT_ERRNO(_errno, _test_cmd_virq_alloc(self->fd, viommu_id, type, \
+                                                 virq_id, virq_fd))
+
+static int _test_cmd_trigger_virq(int fd, __u32 dev_id, __u32 event_fd,
+                                 __u32 virt_id)
+{
+       struct iommu_test_cmd trigger_virq_cmd = {
+               .size = sizeof(trigger_virq_cmd),
+               .op = IOMMU_TEST_OP_TRIGGER_VIRQ,
+               .trigger_virq = {
+                       .dev_id = dev_id,
+               },
+       };
+       struct pollfd pollfd = { .fd = event_fd, .events = POLLIN };
+       struct iommu_viommu_irq_selftest irq;
+       ssize_t bytes;
+       int ret;
+
+       ret = ioctl(fd, _IOMMU_TEST_CMD(IOMMU_TEST_OP_TRIGGER_VIRQ),
+                   &trigger_virq_cmd);
+       if (ret)
+               return ret;
+
+       ret = poll(&pollfd, 1, 1000);
+       if (ret < 0)
+               return ret;
+
+       bytes = read(event_fd, &irq, sizeof(irq));
+       if (bytes <= 0)
+               return -EIO;
+
+       return irq.virt_id == virt_id ? 0 : -EINVAL;
+}
+
+#define test_cmd_trigger_virq(dev_id, event_fd, vdev_id) \
+       ASSERT_EQ(0,                                     \
+                 _test_cmd_trigger_virq(self->fd, dev_id, event_fd, vdev_id))
diff --git a/tools/testing/selftests/iommu/iommufd.c 
b/tools/testing/selftests/iommu/iommufd.c
index 212e5d62e13d..b15ebc963e56 100644
--- a/tools/testing/selftests/iommu/iommufd.c
+++ b/tools/testing/selftests/iommu/iommufd.c
@@ -2774,15 +2774,37 @@ TEST_F(iommufd_viommu, vdevice_alloc)
        uint32_t viommu_id = self->viommu_id;
        uint32_t dev_id = self->device_id;
        uint32_t vdev_id = 0;
+       uint32_t virq_id;
+       uint32_t virq_fd;
 
        if (dev_id) {
+               /* Must allocate vdevice before attaching to a nested hwpt */
+               test_err_mock_domain_replace(ENOENT, self->stdev_id,
+                                            self->nested_hwpt_id);
+
+               test_cmd_virq_alloc(viommu_id, IOMMU_VIRQ_TYPE_SELFTEST,
+                                   &virq_id, &virq_fd);
+               test_err_virq_alloc(EEXIST, viommu_id, IOMMU_VIRQ_TYPE_SELFTEST,
+                                   NULL, NULL);
                /* Set vdev_id to 0x99, unset it, and set to 0x88 */
                test_cmd_vdevice_alloc(viommu_id, dev_id, 0x99, &vdev_id);
+               test_cmd_mock_domain_replace(self->stdev_id,
+                                            self->nested_hwpt_id);
+               test_cmd_trigger_virq(dev_id, virq_fd, 0x99);
                test_err_vdevice_alloc(EEXIST, viommu_id, dev_id, 0x99,
                                       &vdev_id);
+               test_cmd_mock_domain_replace(self->stdev_id, self->ioas_id);
                test_ioctl_destroy(vdev_id);
+
+               /* Try again with 0x88 */
                test_cmd_vdevice_alloc(viommu_id, dev_id, 0x88, &vdev_id);
+               test_cmd_mock_domain_replace(self->stdev_id,
+                                            self->nested_hwpt_id);
+               test_cmd_trigger_virq(dev_id, virq_fd, 0x88);
+               close(virq_fd);
+               test_cmd_mock_domain_replace(self->stdev_id, self->ioas_id);
                test_ioctl_destroy(vdev_id);
+               test_ioctl_destroy(virq_id);
        } else {
                test_err_vdevice_alloc(ENOENT, viommu_id, dev_id, 0x99, NULL);
        }
diff --git a/tools/testing/selftests/iommu/iommufd_fail_nth.c 
b/tools/testing/selftests/iommu/iommufd_fail_nth.c
index 64b1f8e1b0cf..442442de3a75 100644
--- a/tools/testing/selftests/iommu/iommufd_fail_nth.c
+++ b/tools/testing/selftests/iommu/iommufd_fail_nth.c
@@ -620,6 +620,7 @@ TEST_FAIL_NTH(basic_fail_nth, device)
        };
        struct iommu_test_hw_info info;
        uint32_t fault_id, fault_fd;
+       uint32_t virq_id, virq_fd;
        uint32_t fault_hwpt_id;
        uint32_t ioas_id;
        uint32_t ioas_id2;
@@ -692,6 +693,11 @@ TEST_FAIL_NTH(basic_fail_nth, device)
                                 IOMMU_HWPT_DATA_SELFTEST, &data, sizeof(data)))
                return -1;
 
+       if (_test_cmd_virq_alloc(self->fd, viommu_id, IOMMU_VIRQ_TYPE_SELFTEST,
+                                &virq_id, &virq_fd))
+               return -1;
+       close(virq_fd);
+
        return 0;
 }
 
-- 
2.43.0


Reply via email to