lists.openwall.net   lists  /  announce  owl-users  owl-dev  john-users  john-dev  passwdqc-users  yescrypt  popa3d-users  /  oss-security  kernel-hardening  musl  sabotage  tlsify  passwords  /  crypt-dev  xvendor  /  Bugtraq  Full-Disclosure  linux-kernel  linux-netdev  linux-ext4  linux-hardening  linux-cve-announce  PHC 
Open Source and information security mailing list archives
 
Hash Suite: Windows password security audit tool. GUI, reports in PDF.
[<prev] [next>] [<thread-prev] [thread-next>] [day] [month] [year] [list]
Message-ID: <463a089c0d29b97be77f33fac6ed19273e23ab1d.1745646960.git.nicolinc@nvidia.com>
Date: Fri, 25 Apr 2025 22:58:07 -0700
From: Nicolin Chen <nicolinc@...dia.com>
To: <jgg@...dia.com>, <kevin.tian@...el.com>, <corbet@....net>,
	<will@...nel.org>
CC: <bagasdotme@...il.com>, <robin.murphy@....com>, <joro@...tes.org>,
	<thierry.reding@...il.com>, <vdumpa@...dia.com>, <jonathanh@...dia.com>,
	<shuah@...nel.org>, <jsnitsel@...hat.com>, <nathan@...nel.org>,
	<peterz@...radead.org>, <yi.l.liu@...el.com>, <mshavit@...gle.com>,
	<praan@...gle.com>, <zhangzekun11@...wei.com>, <iommu@...ts.linux.dev>,
	<linux-doc@...r.kernel.org>, <linux-kernel@...r.kernel.org>,
	<linux-arm-kernel@...ts.infradead.org>, <linux-tegra@...r.kernel.org>,
	<linux-kselftest@...r.kernel.org>, <patches@...ts.linux.dev>,
	<mochs@...dia.com>, <alok.a.tiwari@...cle.com>, <vasant.hegde@....com>
Subject: [PATCH v2 12/22] iommufd/selftest: Add coverage for IOMMUFD_CMD_VCMDQ_ALLOC

Some simple tests for IOMMUFD_CMD_VCMDQ_ALLOC infrastructure covering the
new iommufd_vcmdq_depend/undepend() helpers.

Signed-off-by: Nicolin Chen <nicolinc@...dia.com>
---
 drivers/iommu/iommufd/iommufd_test.h          |  3 +
 tools/testing/selftests/iommu/iommufd_utils.h | 30 +++++++++
 drivers/iommu/iommufd/selftest.c              | 67 +++++++++++++++++++
 tools/testing/selftests/iommu/iommufd.c       | 59 ++++++++++++++++
 .../selftests/iommu/iommufd_fail_nth.c        |  6 ++
 5 files changed, 165 insertions(+)

diff --git a/drivers/iommu/iommufd/iommufd_test.h b/drivers/iommu/iommufd/iommufd_test.h
index fbf9ecb35a13..a0831d78fef1 100644
--- a/drivers/iommu/iommufd/iommufd_test.h
+++ b/drivers/iommu/iommufd/iommufd_test.h
@@ -265,4 +265,7 @@ struct iommu_viommu_event_selftest {
 	__u32 virt_id;
 };
 
+#define IOMMU_VCMDQ_TYPE_SELFTEST 0xdeadbeef
+#define IOMMU_TEST_VCMDQ_MAX 2
+
 #endif
diff --git a/tools/testing/selftests/iommu/iommufd_utils.h b/tools/testing/selftests/iommu/iommufd_utils.h
index a5d4cbd089ba..d6d8fedf2226 100644
--- a/tools/testing/selftests/iommu/iommufd_utils.h
+++ b/tools/testing/selftests/iommu/iommufd_utils.h
@@ -956,6 +956,36 @@ static int _test_cmd_vdevice_alloc(int fd, __u32 viommu_id, __u32 idev_id,
 		     _test_cmd_vdevice_alloc(self->fd, viommu_id, idev_id,   \
 					     virt_id, vdev_id))
 
+static int _test_cmd_vcmdq_alloc(int fd, __u32 viommu_id, __u32 type, __u32 idx,
+				 __u64 addr, __u64 length, __u32 *vcmdq_id)
+{
+	struct iommu_vcmdq_alloc cmd = {
+		.size = sizeof(cmd),
+		.viommu_id = viommu_id,
+		.type = type,
+		.index = idx,
+		.addr = addr,
+		.length = length,
+	};
+	int ret;
+
+	ret = ioctl(fd, IOMMU_VCMDQ_ALLOC, &cmd);
+	if (ret)
+		return ret;
+	if (vcmdq_id)
+		*vcmdq_id = cmd.out_vcmdq_id;
+	return 0;
+}
+
+#define test_cmd_vcmdq_alloc(viommu_id, type, idx, addr, len, vcmdq_id)    \
+	ASSERT_EQ(0, _test_cmd_vcmdq_alloc(self->fd, viommu_id, type, idx, \
+					   addr, len, vcmdq_id))
+#define test_err_vcmdq_alloc(_errno, viommu_id, type, idx, addr, len,      \
+			     vcmdq_id)                                     \
+	EXPECT_ERRNO(_errno,                                               \
+		     _test_cmd_vcmdq_alloc(self->fd, viommu_id, type, idx, \
+					   addr, len, vcmdq_id))
+
 static int _test_cmd_veventq_alloc(int fd, __u32 viommu_id, __u32 type,
 				   __u32 *veventq_id, __u32 *veventq_fd)
 {
diff --git a/drivers/iommu/iommufd/selftest.c b/drivers/iommu/iommufd/selftest.c
index b04bd2fbc53d..d6cc5b78821b 100644
--- a/drivers/iommu/iommufd/selftest.c
+++ b/drivers/iommu/iommufd/selftest.c
@@ -148,6 +148,7 @@ to_mock_nested(struct iommu_domain *domain)
 struct mock_viommu {
 	struct iommufd_viommu core;
 	struct mock_iommu_domain *s2_parent;
+	struct mock_vcmdq *mock_vcmdq[IOMMU_TEST_VCMDQ_MAX];
 };
 
 static inline struct mock_viommu *to_mock_viommu(struct iommufd_viommu *viommu)
@@ -155,6 +156,18 @@ static inline struct mock_viommu *to_mock_viommu(struct iommufd_viommu *viommu)
 	return container_of(viommu, struct mock_viommu, core);
 }
 
+struct mock_vcmdq {
+	struct iommufd_vcmdq core;
+	struct mock_viommu *mock_viommu;
+	struct mock_vcmdq *prev;
+	u16 index;
+};
+
+static inline struct mock_vcmdq *to_mock_vcmdq(struct iommufd_vcmdq *vcmdq)
+{
+	return container_of(vcmdq, struct mock_vcmdq, core);
+}
+
 enum selftest_obj_type {
 	TYPE_IDEV,
 };
@@ -727,10 +740,64 @@ static int mock_viommu_cache_invalidate(struct iommufd_viommu *viommu,
 	return rc;
 }
 
+/* Test iommufd_vcmdq_depend/_undepend() */
+static struct iommufd_vcmdq *
+mock_vcmdq_alloc(struct iommufd_viommu *viommu, unsigned int type, u32 index,
+		 dma_addr_t addr, size_t length)
+{
+	struct mock_viommu *mock_viommu = to_mock_viommu(viommu);
+	struct mock_vcmdq *mock_vcmdq, *prev = 0;
+	int rc;
+
+	if (type != IOMMU_VCMDQ_TYPE_SELFTEST)
+		return ERR_PTR(-EOPNOTSUPP);
+	if (index >= IOMMU_TEST_VCMDQ_MAX)
+		return ERR_PTR(-EINVAL);
+	if (mock_viommu->mock_vcmdq[index])
+		return ERR_PTR(-EEXIST);
+	if (index) {
+		prev = mock_viommu->mock_vcmdq[index - 1];
+		if (!prev)
+			return ERR_PTR(-EIO);
+	}
+
+	mock_vcmdq = iommufd_vcmdq_alloc(viommu, struct mock_vcmdq, core);
+	if (IS_ERR(mock_vcmdq))
+		return ERR_CAST(mock_vcmdq);
+
+	if (prev) {
+		rc = iommufd_vcmdq_depend(mock_vcmdq, prev, core);
+		if (rc)
+			goto free_vcmdq;
+	}
+	mock_vcmdq->prev = prev;
+	mock_vcmdq->mock_viommu = mock_viommu;
+	mock_viommu->mock_vcmdq[index] = mock_vcmdq;
+
+	return &mock_vcmdq->core;
+free_vcmdq:
+	iommufd_struct_destroy(viommu->ictx, mock_vcmdq, core);
+	return ERR_PTR(rc);
+}
+
+static void mock_vcmdq_destroy(struct iommufd_vcmdq *vcmdq)
+{
+	struct mock_vcmdq *mock_vcmdq = to_mock_vcmdq(vcmdq);
+	struct mock_viommu *mock_viommu = mock_vcmdq->mock_viommu;
+
+	mock_viommu->mock_vcmdq[mock_vcmdq->index] = NULL;
+	if (mock_vcmdq->prev)
+		iommufd_vcmdq_undepend(mock_vcmdq, mock_vcmdq->prev, core);
+
+	/* iommufd core frees mock_vcmdq and vcmdq */
+}
+
 static struct iommufd_viommu_ops mock_viommu_ops = {
 	.destroy = mock_viommu_destroy,
 	.alloc_domain_nested = mock_viommu_alloc_domain_nested,
 	.cache_invalidate = mock_viommu_cache_invalidate,
+	.vcmdq_alloc = mock_vcmdq_alloc,
+	.vcmdq_destroy = mock_vcmdq_destroy,
 };
 
 static struct iommufd_viommu *
diff --git a/tools/testing/selftests/iommu/iommufd.c b/tools/testing/selftests/iommu/iommufd.c
index 8ebbb7fda02d..7c464f6eb37b 100644
--- a/tools/testing/selftests/iommu/iommufd.c
+++ b/tools/testing/selftests/iommu/iommufd.c
@@ -3031,6 +3031,65 @@ TEST_F(iommufd_viommu, vdevice_cache)
 	}
 }
 
+TEST_F(iommufd_viommu, vcmdq)
+{
+	uint32_t viommu_id = self->viommu_id;
+	__u64 iova = MOCK_APERTURE_START;
+	uint32_t vcmdq_id[2];
+
+	if (viommu_id) {
+		/* Fail IOMMU_VCMDQ_TYPE_DEFAULT */
+		test_err_vcmdq_alloc(EOPNOTSUPP, viommu_id,
+				     IOMMU_VCMDQ_TYPE_DEFAULT, 0, iova,
+				     PAGE_SIZE, &vcmdq_id[0]);
+		/* Fail queue addr and length */
+		test_err_vcmdq_alloc(EINVAL, viommu_id,
+				     IOMMU_VCMDQ_TYPE_SELFTEST, 0, 0, PAGE_SIZE,
+				     &vcmdq_id[0]);
+		test_err_vcmdq_alloc(EINVAL, viommu_id,
+				     IOMMU_VCMDQ_TYPE_SELFTEST, 0, iova, 0,
+				     &vcmdq_id[0]);
+		test_err_vcmdq_alloc(EOVERFLOW, viommu_id,
+				     IOMMU_VCMDQ_TYPE_SELFTEST, 0, ~(uint64_t)0,
+				     PAGE_SIZE, &vcmdq_id[0]);
+		/* Fail missing iova */
+		test_err_vcmdq_alloc(ENXIO, viommu_id,
+				     IOMMU_VCMDQ_TYPE_SELFTEST, 0, iova,
+				     PAGE_SIZE, &vcmdq_id[0]);
+
+		/* Map iova */
+		test_ioctl_ioas_map(buffer, PAGE_SIZE, &iova);
+
+		/* Fail index=1 and =MAX; must start from index=0 */
+		test_err_vcmdq_alloc(EIO, viommu_id,
+				     IOMMU_VCMDQ_TYPE_SELFTEST, 1, iova,
+				     PAGE_SIZE, &vcmdq_id[0]);
+		test_err_vcmdq_alloc(EINVAL, viommu_id,
+				     IOMMU_VCMDQ_TYPE_SELFTEST,
+				     IOMMU_TEST_VCMDQ_MAX, iova, PAGE_SIZE,
+				     &vcmdq_id[0]);
+
+		/* Allocate index=0 */
+		test_cmd_vcmdq_alloc(viommu_id, IOMMU_VCMDQ_TYPE_SELFTEST, 0,
+				     iova, PAGE_SIZE, &vcmdq_id[0]);
+		/* Fail duplicate */
+		test_err_vcmdq_alloc(EEXIST, viommu_id,
+				     IOMMU_VCMDQ_TYPE_SELFTEST, 0,
+				     iova, PAGE_SIZE, &vcmdq_id[0]);
+
+		/* Allocate index=1 */
+		test_cmd_vcmdq_alloc(viommu_id, IOMMU_VCMDQ_TYPE_SELFTEST, 1,
+				     iova, PAGE_SIZE, &vcmdq_id[1]);
+		/* Fail to destroy, due to dependency */
+		EXPECT_ERRNO(EBUSY,
+			     _test_ioctl_destroy(self->fd, vcmdq_id[0]));
+
+		/* Destroy in descending order */
+		test_ioctl_destroy(vcmdq_id[1]);
+		test_ioctl_destroy(vcmdq_id[0]);
+	}
+}
+
 FIXTURE(iommufd_device_pasid)
 {
 	int fd;
diff --git a/tools/testing/selftests/iommu/iommufd_fail_nth.c b/tools/testing/selftests/iommu/iommufd_fail_nth.c
index f7ccf1822108..ffad3f2875bd 100644
--- a/tools/testing/selftests/iommu/iommufd_fail_nth.c
+++ b/tools/testing/selftests/iommu/iommufd_fail_nth.c
@@ -634,6 +634,7 @@ TEST_FAIL_NTH(basic_fail_nth, device)
 	uint32_t idev_id;
 	uint32_t hwpt_id;
 	uint32_t viommu_id;
+	uint32_t vcmdq_id;
 	uint32_t vdev_id;
 	__u64 iova;
 
@@ -696,6 +697,11 @@ TEST_FAIL_NTH(basic_fail_nth, device)
 	if (_test_cmd_vdevice_alloc(self->fd, viommu_id, idev_id, 0, &vdev_id))
 		return -1;
 
+	if (_test_cmd_vcmdq_alloc(self->fd, viommu_id,
+				  IOMMU_VCMDQ_TYPE_SELFTEST, 0, iova, PAGE_SIZE,
+				  &vcmdq_id))
+		return -1;
+
 	if (_test_ioctl_fault_alloc(self->fd, &fault_id, &fault_fd))
 		return -1;
 	close(fault_fd);
-- 
2.43.0


Powered by blists - more mailing lists

Powered by Openwall GNU/*/Linux Powered by OpenVZ