Also add fail_nth coverage too.

Signed-off-by: Nicolin Chen <nicol...@nvidia.com>
---
 tools/testing/selftests/iommu/iommufd.c       | 97 +++++++++++++++++++
 .../selftests/iommu/iommufd_fail_nth.c        | 21 ++++
 2 files changed, 118 insertions(+)

diff --git a/tools/testing/selftests/iommu/iommufd.c 
b/tools/testing/selftests/iommu/iommufd.c
index a1b2b657999d..db7dfc5ae56a 100644
--- a/tools/testing/selftests/iommu/iommufd.c
+++ b/tools/testing/selftests/iommu/iommufd.c
@@ -334,6 +334,103 @@ TEST_F(change_process, basic)
        ASSERT_EQ(child, waitpid(child, NULL, 0));
 }
 
+FIXTURE(iommufd_sw_msi)
+{
+       int fd;
+       uint32_t ioas_id;
+       uint32_t idev_id[2];
+};
+
+FIXTURE_SETUP(iommufd_sw_msi)
+{
+       self->fd = open("/dev/iommu", O_RDWR);
+       ASSERT_NE(-1, self->fd);
+
+       test_ioctl_ioas_alloc(&self->ioas_id);
+       test_cmd_mock_domain(self->ioas_id, NULL, NULL, &self->idev_id[0]);
+       test_cmd_mock_domain_flags(self->ioas_id, MOCK_FLAGS_DEVICE_NO_ATTACH,
+                                  NULL, NULL, &self->idev_id[1]);
+}
+
+FIXTURE_TEARDOWN(iommufd_sw_msi)
+{
+       teardown_iommufd(self->fd, _metadata);
+}
+
+TEST_F(iommufd_sw_msi, basic)
+{
+       struct iommu_option cmd = {
+               .size = sizeof(cmd),
+               .op = IOMMU_OPTION_OP_SET,
+       };
+
+       /* Negative case: object_id must be a device id */
+       cmd.object_id = self->ioas_id;
+       cmd.option_id = IOMMU_OPTION_SW_MSI_START;
+       cmd.val64 = 0x70000000;
+       EXPECT_ERRNO(ENOENT, ioctl(self->fd, IOMMU_OPTION, &cmd));
+       cmd.object_id = 0;
+       cmd.option_id = IOMMU_OPTION_SW_MSI_SIZE;
+       cmd.val64 = 2;
+       EXPECT_ERRNO(ENOENT, ioctl(self->fd, IOMMU_OPTION, &cmd));
+
+       /* Negative case: device must not be attached already */
+       if (self->idev_id[0]) {
+               cmd.object_id = self->idev_id[0];
+               cmd.option_id = IOMMU_OPTION_SW_MSI_START;
+               cmd.val64 = 0x70000000;
+               EXPECT_ERRNO(EBUSY, ioctl(self->fd, IOMMU_OPTION, &cmd));
+       }
+
+       /* Device attached to nothing */
+       if (self->idev_id[1]) {
+               /* Negative case: alignment failures */
+               cmd.object_id = self->idev_id[1];
+               cmd.option_id = IOMMU_OPTION_SW_MSI_START;
+               cmd.val64 = 0x7fffffff;
+               EXPECT_ERRNO(EINVAL, ioctl(self->fd, IOMMU_OPTION, &cmd));
+               cmd.val64 = 0x7fffff00;
+               EXPECT_ERRNO(EINVAL, ioctl(self->fd, IOMMU_OPTION, &cmd));
+               cmd.val64 = 0x7fff0000;
+               EXPECT_ERRNO(EINVAL, ioctl(self->fd, IOMMU_OPTION, &cmd));
+
+               /* Negative case: overlap against [0x80000000, 0x80ffffff] */
+               cmd.option_id = IOMMU_OPTION_SW_MSI_START;
+               cmd.val64 = 0x80000000;
+               EXPECT_ERRNO(EADDRINUSE, ioctl(self->fd, IOMMU_OPTION, &cmd));
+               cmd.val64 = 0x80400000;
+               EXPECT_ERRNO(EADDRINUSE, ioctl(self->fd, IOMMU_OPTION, &cmd));
+               cmd.val64 = 0x80800000;
+               EXPECT_ERRNO(EADDRINUSE, ioctl(self->fd, IOMMU_OPTION, &cmd));
+               cmd.val64 = 0x80c00000;
+               EXPECT_ERRNO(EADDRINUSE, ioctl(self->fd, IOMMU_OPTION, &cmd));
+               /* Though an address that starts 1MB below will be okay ... */
+               cmd.val64 = 0x7ff00000;
+               ASSERT_EQ(0, ioctl(self->fd, IOMMU_OPTION, &cmd));
+               /* ... but not with a 2MB size */
+               cmd.option_id = IOMMU_OPTION_SW_MSI_SIZE;
+               cmd.val64 = 2;
+               EXPECT_ERRNO(EADDRINUSE, ioctl(self->fd, IOMMU_OPTION, &cmd));
+
+               /* Set a safe 2MB window */
+               cmd.option_id = IOMMU_OPTION_SW_MSI_START;
+               cmd.val64 = 0x70000000;
+               ASSERT_EQ(0, ioctl(self->fd, IOMMU_OPTION, &cmd));
+               cmd.option_id = IOMMU_OPTION_SW_MSI_SIZE;
+               cmd.val64 = 2;
+               ASSERT_EQ(0, ioctl(self->fd, IOMMU_OPTION, &cmd));
+
+               /* Read them back to verify */
+               cmd.op = IOMMU_OPTION_OP_GET;
+               cmd.option_id = IOMMU_OPTION_SW_MSI_START;
+               ASSERT_EQ(0, ioctl(self->fd, IOMMU_OPTION, &cmd));
+               ASSERT_EQ(cmd.val64, 0x70000000);
+               cmd.option_id = IOMMU_OPTION_SW_MSI_SIZE;
+               ASSERT_EQ(0, ioctl(self->fd, IOMMU_OPTION, &cmd));
+               ASSERT_EQ(cmd.val64, 2);
+       }
+}
+
 FIXTURE(iommufd_ioas)
 {
        int fd;
diff --git a/tools/testing/selftests/iommu/iommufd_fail_nth.c 
b/tools/testing/selftests/iommu/iommufd_fail_nth.c
index 64b1f8e1b0cf..be0d7735dfeb 100644
--- a/tools/testing/selftests/iommu/iommufd_fail_nth.c
+++ b/tools/testing/selftests/iommu/iommufd_fail_nth.c
@@ -615,6 +615,10 @@ TEST_FAIL_NTH(basic_fail_nth, access_pin_domain)
 /* device.c */
 TEST_FAIL_NTH(basic_fail_nth, device)
 {
+       struct iommu_option cmd = {
+               .size = sizeof(cmd),
+               .op = IOMMU_OPTION_OP_SET,
+       };
        struct iommu_hwpt_selftest data = {
                .iotlb = IOMMU_TEST_IOTLB_DEFAULT,
        };
@@ -624,6 +628,7 @@ TEST_FAIL_NTH(basic_fail_nth, device)
        uint32_t ioas_id;
        uint32_t ioas_id2;
        uint32_t stdev_id;
+       uint32_t idev_id2;
        uint32_t idev_id;
        uint32_t hwpt_id;
        uint32_t viommu_id;
@@ -692,6 +697,22 @@ TEST_FAIL_NTH(basic_fail_nth, device)
                                 IOMMU_HWPT_DATA_SELFTEST, &data, sizeof(data)))
                return -1;
 
+       if (_test_cmd_mock_domain_flags(self->fd, ioas_id,
+                                       MOCK_FLAGS_DEVICE_NO_ATTACH, NULL, NULL,
+                                       &idev_id2))
+               return -1;
+
+       cmd.object_id = idev_id2;
+       cmd.option_id = IOMMU_OPTION_SW_MSI_START;
+       cmd.val64 = 0x70000000;
+       if (ioctl(self->fd, IOMMU_OPTION, &cmd))
+               return -1;
+
+       cmd.option_id = IOMMU_OPTION_SW_MSI_SIZE;
+       cmd.val64 = 2;
+       if (ioctl(self->fd, IOMMU_OPTION, &cmd))
+               return -1;
+
        return 0;
 }
 
-- 
2.43.0


Reply via email to