[PATCH 3/4] tests/intel/xe_vm: Add DRM_IOCTL_XE_VM_GET_FAULTS validation tests

[Jonathan Cavitt <jonathan.cavitt@intel.com>]

Add tests to xe_vm that exercise the new DRM_IOCTL_XE_VM_GET_FAULTS
ioctl.  Specifically, add input validation tests that exercise the
return values for improperly formatted ioctl structures.

Signed-off-by: Jonathan Cavitt <jonathan.cavitt@intel.com>
---
 tests/intel/xe_vm.c | 79 +++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 79 insertions(+)

diff --git a/tests/intel/xe_vm.c b/tests/intel/xe_vm.c
index fdf74c1140..40928b441c 100644
--- a/tests/intel/xe_vm.c
+++ b/tests/intel/xe_vm.c
@@ -2343,6 +2343,69 @@ static void invalid_vm_id(int fd)
 	do_ioctl_err(fd, DRM_IOCTL_XE_VM_DESTROY, &destroy, ENOENT);
 }
 
+/**
+ * SUBTEST: vm-get-faults-invalid-reserved
+ * Functionality: ioctl_input_validation
+ * Description: Check query with invalid reserved returns expected error code
+ *
+ * SUBTEST: vm-get-faults-invalid-vm-id
+ * Functionality: ioctl_input_validation
+ * Description: Check query with invalid vm_id returns expected error code
+ *
+ * SUBTEST: vm-get-faults-invalid-size
+ * Functionality: ioctl_input_validation
+ * Description: Check query with invalid size returns expected error code
+ *
+ * SUBTEST: vm-get-faults-invalid-fault-count
+ * Functionality: ioctl_input_validation
+ * Description: Check query with invalid fault_count returns expected error code
+ */
+static void get_faults_invalid_reserved(int fd, uint32_t vm)
+{
+	struct drm_xe_vm_get_faults query = {
+		.reserved[0] = 0xdeadbeef,
+	};
+
+	do_ioctl_err(fd, DRM_IOCTL_XE_VM_GET_FAULTS, &query, EINVAL);
+}
+
+static void get_faults_invalid_vm_id(int fd, uint32_t vm)
+{
+	struct drm_xe_vm_get_faults query = {
+		.vm_id = 0xdeadbeef,
+	};
+
+	do_ioctl_err(fd, DRM_IOCTL_XE_VM_GET_FAULTS, &query, ENOENT);
+}
+
+static void get_faults_invalid_size(int fd, uint32_t vm)
+{
+	struct drm_xe_vm_get_faults query = {
+		.size = 1,
+	};
+
+	do_ioctl_err(fd, DRM_IOCTL_XE_VM_GET_FAULTS, &query, EINVAL);
+}
+
+static void get_faults_invalid_fault_count(int fd, uint32_t vm)
+{
+	struct drm_xe_vm_get_faults query = {
+		.fault_count = 1,
+	};
+
+	do_ioctl_err(fd, DRM_IOCTL_XE_VM_GET_FAULTS, &query, EINVAL);
+}
+
+static void test_get_faults(int fd, void (*func)(int fd, uint32_t vm))
+{
+	uint32_t vm;
+
+	vm = xe_vm_create(fd, DRM_XE_VM_CREATE_FLAG_LR_MODE ||
+			  DRM_XE_VM_CREATE_FLAG_FAULT_MODE, 0);
+	func(fd, vm);
+	xe_vm_destroy(fd, vm);
+}
+
 igt_main
 {
 	struct drm_xe_engine_class_instance *hwe, *hwe_non_copy = NULL;
@@ -2455,6 +2518,17 @@ igt_main
                 { }
         };
 
+	const struct vm_get_faults {
+		const char *name;
+		void (*test)(int fd, uint32_t vm);
+	} xe_vm_get_faults_tests[] = {
+		{ "invalid-reserved", get_faults_invalid_reserved },
+		{ "invalid-vm-id", get_faults_invalid_vm_id },
+		{ "invalid-size", get_faults_invalid_size },
+		{ "invalid-fault-count", get_faults_invalid_fault_count },
+		{ }
+	};
+
 	igt_fixture {
 		fd = drm_open_driver(DRIVER_XE);
 
@@ -2734,6 +2808,11 @@ igt_main
 	igt_subtest("invalid-vm-id")
 		invalid_vm_id(fd);
 
+	for (const struct vm_get_faults *f = xe_vm_get_faults_tests; f->name; f++) {
+		igt_subtest_f("vm-get-faults-%s", f->name)
+			test_get_faults(fd, f->test);
+	}
+
 	igt_fixture
 		drm_close_driver(fd);
 }
-- 
2.43.0