[PATCH RFCv2 10/13] iommufd/selftes: Add coverage for IOMMU_OPTION_SW_MSI_START/SIZE

[Nicolin Chen]

Also add fail_nth coverage too.

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

diff --git a/tools/testing/selftests/iommu/iommufd.c 
b/tools/testing/selftests/iommu/iommufd.c
index a1b2b657999d..87b5b5ac927a 100644
--- a/tools/testing/selftests/iommu/iommufd.c
+++ b/tools/testing/selftests/iommu/iommufd.c
@@ -334,6 +334,59 @@ TEST_F(change_process, basic)
        ASSERT_EQ(child, waitpid(child, NULL, 0));
 }
 
+FIXTURE(iommufd_sw_msi)
+{
+       int fd;
+       uint32_t ioas_id;
+};
+
+FIXTURE_SETUP(iommufd_sw_msi)
+{
+       self->fd = open("/dev/iommu", O_RDWR);
+       ASSERT_NE(-1, self->fd);
+
+       test_ioctl_ioas_alloc(&self->ioas_id);
+}
+
+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: assign an object_id to this global option */
+       cmd.object_id = self->ioas_id;
+       cmd.option_id = IOMMU_OPTION_SW_MSI_START;
+       cmd.val64 = 0xffffffff;
+       EXPECT_ERRNO(EOPNOTSUPP, ioctl(self->fd, IOMMU_OPTION, &cmd));
+       cmd.option_id = IOMMU_OPTION_SW_MSI_SIZE;
+       cmd.val64 = 2;
+       EXPECT_ERRNO(EOPNOTSUPP, ioctl(self->fd, IOMMU_OPTION, &cmd));
+
+       cmd.object_id = 0;
+       cmd.option_id = IOMMU_OPTION_SW_MSI_START;
+       cmd.val64 = 0xffffffff;
+       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.object_id = 0;
+       cmd.option_id = IOMMU_OPTION_SW_MSI_START;
+       ASSERT_EQ(0, ioctl(self->fd, IOMMU_OPTION, &cmd));
+       ASSERT_EQ(cmd.val64, 0xffffffff);
+       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..6d7b1417ce45 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,
        };
@@ -634,6 +638,16 @@ TEST_FAIL_NTH(basic_fail_nth, device)
        if (self->fd == -1)
                return -1;
 
+       cmd.option_id = IOMMU_OPTION_SW_MSI_START;
+       cmd.val64 = 0x8000000;
+       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;
+
        if (_test_ioctl_ioas_alloc(self->fd, &ioas_id))
                return -1;
 
-- 
2.43.0