Add test case for dirty tracking on a domain attached to PASID, also
confirm attachment to PASID fail if device doesn't support dirty tracking.

Suggested-by: Lu Baolu <[email protected]>
Signed-off-by: Zhenzhong Duan <[email protected]>
---
 tools/testing/selftests/iommu/iommufd.c | 27 +++++++++++++++++++++++++
 1 file changed, 27 insertions(+)

diff --git a/tools/testing/selftests/iommu/iommufd.c 
b/tools/testing/selftests/iommu/iommufd.c
index dadad277f4eb..d1fe5dbc2813 100644
--- a/tools/testing/selftests/iommu/iommufd.c
+++ b/tools/testing/selftests/iommu/iommufd.c
@@ -2275,6 +2275,33 @@ TEST_F(iommufd_dirty_tracking, set_dirty_tracking)
        test_ioctl_destroy(hwpt_id);
 }
 
+TEST_F(iommufd_dirty_tracking, pasid_set_dirty_tracking)
+{
+       uint32_t stddev_id, ioas_id, hwpt_id, pasid = 100;
+       uint32_t dev_flags = MOCK_FLAGS_DEVICE_PASID;
+
+       /* Regular case */
+       test_cmd_hwpt_alloc(self->idev_id, self->ioas_id,
+                           IOMMU_HWPT_ALLOC_PASID | 
IOMMU_HWPT_ALLOC_DIRTY_TRACKING,
+                           &hwpt_id);
+       test_cmd_mock_domain_flags(hwpt_id, dev_flags, &stddev_id, NULL, NULL);
+       ASSERT_EQ(0, _test_cmd_pasid_attach(self->fd, stddev_id, pasid, 
hwpt_id));
+       test_cmd_set_dirty_tracking(hwpt_id, true);
+       test_cmd_set_dirty_tracking(hwpt_id, false);
+       ASSERT_EQ(0, _test_cmd_pasid_detach(self->fd, stddev_id, pasid));
+
+       test_ioctl_destroy(stddev_id);
+
+       /* IOMMU device does not support dirty tracking */
+       dev_flags |= MOCK_FLAGS_DEVICE_NO_DIRTY;
+       test_ioctl_ioas_alloc(&ioas_id);
+       test_cmd_mock_domain_flags(ioas_id, dev_flags, &stddev_id, NULL, NULL);
+       EXPECT_ERRNO(EINVAL, _test_cmd_pasid_attach(self->fd, stddev_id, pasid, 
hwpt_id));
+
+       test_ioctl_destroy(stddev_id);
+       test_ioctl_destroy(hwpt_id);
+}
+
 TEST_F(iommufd_dirty_tracking, device_dirty_capability)
 {
        uint32_t caps = 0;
-- 
2.47.3


Reply via email to