[linux-2.6-block.git] / drivers / vfio / iommufd.c

// SPDX-License-Identifier: GPL-2.0-only
/*
 * Copyright (c) 2021-2022, NVIDIA CORPORATION & AFFILIATES
 */
#include <linux/vfio.h>
#include <linux/iommufd.h>

#include "vfio.h"

MODULE_IMPORT_NS(IOMMUFD);
MODULE_IMPORT_NS(IOMMUFD_VFIO);

int vfio_iommufd_bind(struct vfio_device *vdev, struct iommufd_ctx *ictx)
{
	u32 ioas_id;
	u32 device_id;
	int ret;

	lockdep_assert_held(&vdev->dev_set->lock);

	/*
	 * If the driver doesn't provide this op then it means the device does
	 * not do DMA at all. So nothing to do.
	 */
	if (!vdev->ops->bind_iommufd)
		return 0;

	ret = vdev->ops->bind_iommufd(vdev, ictx, &device_id);
	if (ret)
		return ret;

	ret = iommufd_vfio_compat_ioas_id(ictx, &ioas_id);
	if (ret)
		goto err_unbind;
	ret = vdev->ops->attach_ioas(vdev, &ioas_id);
	if (ret)
		goto err_unbind;

	/*
	 * The legacy path has no way to return the device id or the selected
	 * pt_id
	 */
	return 0;

err_unbind:
	if (vdev->ops->unbind_iommufd)
		vdev->ops->unbind_iommufd(vdev);
	return ret;
}

void vfio_iommufd_unbind(struct vfio_device *vdev)
{
	lockdep_assert_held(&vdev->dev_set->lock);

	if (vdev->ops->unbind_iommufd)
		vdev->ops->unbind_iommufd(vdev);
}

/*
 * The physical standard ops mean that the iommufd_device is bound to the
 * physical device vdev->dev that was provided to vfio_init_group_dev(). Drivers
 * using this ops set should call vfio_register_group_dev()
 */
int vfio_iommufd_physical_bind(struct vfio_device *vdev,
			       struct iommufd_ctx *ictx, u32 *out_device_id)
{
	struct iommufd_device *idev;

	idev = iommufd_device_bind(ictx, vdev->dev, out_device_id);
	if (IS_ERR(idev))
		return PTR_ERR(idev);
	vdev->iommufd_device = idev;
	return 0;
}
EXPORT_SYMBOL_GPL(vfio_iommufd_physical_bind);

void vfio_iommufd_physical_unbind(struct vfio_device *vdev)
{
	lockdep_assert_held(&vdev->dev_set->lock);

	if (vdev->iommufd_attached) {
		iommufd_device_detach(vdev->iommufd_device);
		vdev->iommufd_attached = false;
	}
	iommufd_device_unbind(vdev->iommufd_device);
	vdev->iommufd_device = NULL;
}
EXPORT_SYMBOL_GPL(vfio_iommufd_physical_unbind);

int vfio_iommufd_physical_attach_ioas(struct vfio_device *vdev, u32 *pt_id)
{
	int rc;

	rc = iommufd_device_attach(vdev->iommufd_device, pt_id);
	if (rc)
		return rc;
	vdev->iommufd_attached = true;
	return 0;
}
EXPORT_SYMBOL_GPL(vfio_iommufd_physical_attach_ioas);

/*
 * The emulated standard ops mean that vfio_device is going to use the
 * "mdev path" and will call vfio_pin_pages()/vfio_dma_rw(). Drivers using this
 * ops set should call vfio_register_emulated_iommu_dev().
 */

static void vfio_emulated_unmap(void *data, unsigned long iova,
				unsigned long length)
{
	struct vfio_device *vdev = data;

	vdev->ops->dma_unmap(vdev, iova, length);
}

static const struct iommufd_access_ops vfio_user_ops = {
	.needs_pin_pages = 1,
	.unmap = vfio_emulated_unmap,
};

int vfio_iommufd_emulated_bind(struct vfio_device *vdev,
			       struct iommufd_ctx *ictx, u32 *out_device_id)
{
	lockdep_assert_held(&vdev->dev_set->lock);

	vdev->iommufd_ictx = ictx;
	iommufd_ctx_get(ictx);
	return 0;
}
EXPORT_SYMBOL_GPL(vfio_iommufd_emulated_bind);

void vfio_iommufd_emulated_unbind(struct vfio_device *vdev)
{
	lockdep_assert_held(&vdev->dev_set->lock);

	if (vdev->iommufd_access) {
		iommufd_access_destroy(vdev->iommufd_access);
		vdev->iommufd_access = NULL;
	}
	iommufd_ctx_put(vdev->iommufd_ictx);
	vdev->iommufd_ictx = NULL;
}
EXPORT_SYMBOL_GPL(vfio_iommufd_emulated_unbind);

int vfio_iommufd_emulated_attach_ioas(struct vfio_device *vdev, u32 *pt_id)
{
	struct iommufd_access *user;

	lockdep_assert_held(&vdev->dev_set->lock);

	user = iommufd_access_create(vdev->iommufd_ictx, *pt_id, &vfio_user_ops,
				     vdev);
	if (IS_ERR(user))
		return PTR_ERR(user);
	vdev->iommufd_access = user;
	return 0;
}
EXPORT_SYMBOL_GPL(vfio_iommufd_emulated_attach_ioas);
Commit	Line	Data
a4d1f91d JG	1	// SPDX-License-Identifier: GPL-2.0-only
	2	/*
	3	* Copyright (c) 2021-2022, NVIDIA CORPORATION & AFFILIATES
	4	*/
	5	#include <linux/vfio.h>
	6	#include <linux/iommufd.h>
	7
	8	#include "vfio.h"
	9
	10	MODULE_IMPORT_NS(IOMMUFD);
	11	MODULE_IMPORT_NS(IOMMUFD_VFIO);
	12
	13	int vfio_iommufd_bind(struct vfio_device vdev, struct iommufd_ctx ictx)
	14	{
	15	u32 ioas_id;
	16	u32 device_id;
	17	int ret;
	18
	19	lockdep_assert_held(&vdev->dev_set->lock);
	20
	21	/*
	22	* If the driver doesn't provide this op then it means the device does
	23	* not do DMA at all. So nothing to do.
	24	*/
	25	if (!vdev->ops->bind_iommufd)
	26	return 0;
	27
	28	ret = vdev->ops->bind_iommufd(vdev, ictx, &device_id);
	29	if (ret)
	30	return ret;
	31
	32	ret = iommufd_vfio_compat_ioas_id(ictx, &ioas_id);
	33	if (ret)
	34	goto err_unbind;
	35	ret = vdev->ops->attach_ioas(vdev, &ioas_id);
	36	if (ret)
	37	goto err_unbind;
	38
	39	/*
	40	* The legacy path has no way to return the device id or the selected
	41	* pt_id
	42	*/
	43	return 0;
	44
	45	err_unbind:
	46	if (vdev->ops->unbind_iommufd)
	47	vdev->ops->unbind_iommufd(vdev);
	48	return ret;
	49	}
	50
	51	void vfio_iommufd_unbind(struct vfio_device *vdev)
	52	{
	53	lockdep_assert_held(&vdev->dev_set->lock);
	54
	55	if (vdev->ops->unbind_iommufd)
	56	vdev->ops->unbind_iommufd(vdev);
	57	}
	58
	59	/*
	60	* The physical standard ops mean that the iommufd_device is bound to the
	61	* physical device vdev->dev that was provided to vfio_init_group_dev(). Drivers
	62	* using this ops set should call vfio_register_group_dev()
	63	*/
	64	int vfio_iommufd_physical_bind(struct vfio_device *vdev,
65	struct iommufd_ctx ictx, u32 out_device_id)
66	{
67	struct iommufd_device *idev;
68
69	idev = iommufd_device_bind(ictx, vdev->dev, out_device_id);
70	if (IS_ERR(idev))
71	return PTR_ERR(idev);
72	vdev->iommufd_device = idev;
73	return 0;
74	}
75	EXPORT_SYMBOL_GPL(vfio_iommufd_physical_bind);
76
77	void vfio_iommufd_physical_unbind(struct vfio_device *vdev)
78	{
79	lockdep_assert_held(&vdev->dev_set->lock);
80
81	if (vdev->iommufd_attached) {
82	iommufd_device_detach(vdev->iommufd_device);
83	vdev->iommufd_attached = false;
84	}
85	iommufd_device_unbind(vdev->iommufd_device);
86	vdev->iommufd_device = NULL;
87	}
88	EXPORT_SYMBOL_GPL(vfio_iommufd_physical_unbind);
89
90	int vfio_iommufd_physical_attach_ioas(struct vfio_device vdev, u32 pt_id)
91	{
92	int rc;
93
94	rc = iommufd_device_attach(vdev->iommufd_device, pt_id);
95	if (rc)
96	return rc;
97	vdev->iommufd_attached = true;
98	return 0;
99	}
100	EXPORT_SYMBOL_GPL(vfio_iommufd_physical_attach_ioas);
4741f2e9 JG	101
	102	/*
	103	* The emulated standard ops mean that vfio_device is going to use the
	104	* "mdev path" and will call vfio_pin_pages()/vfio_dma_rw(). Drivers using this
	105	* ops set should call vfio_register_emulated_iommu_dev().
	106	*/
	107
	108	static void vfio_emulated_unmap(void *data, unsigned long iova,
	109	unsigned long length)
	110	{
	111	struct vfio_device *vdev = data;
	112
	113	vdev->ops->dma_unmap(vdev, iova, length);
	114	}
	115
	116	static const struct iommufd_access_ops vfio_user_ops = {
	117	.needs_pin_pages = 1,
	118	.unmap = vfio_emulated_unmap,
	119	};
	120
	121	int vfio_iommufd_emulated_bind(struct vfio_device *vdev,
	122	struct iommufd_ctx ictx, u32 out_device_id)
	123	{
	124	lockdep_assert_held(&vdev->dev_set->lock);
	125
	126	vdev->iommufd_ictx = ictx;
	127	iommufd_ctx_get(ictx);
	128	return 0;
	129	}
	130	EXPORT_SYMBOL_GPL(vfio_iommufd_emulated_bind);
	131
	132	void vfio_iommufd_emulated_unbind(struct vfio_device *vdev)
	133	{
	134	lockdep_assert_held(&vdev->dev_set->lock);
	135
	136	if (vdev->iommufd_access) {
	137	iommufd_access_destroy(vdev->iommufd_access);
	138	vdev->iommufd_access = NULL;
	139	}
	140	iommufd_ctx_put(vdev->iommufd_ictx);
	141	vdev->iommufd_ictx = NULL;
	142	}
	143	EXPORT_SYMBOL_GPL(vfio_iommufd_emulated_unbind);
	144
	145	int vfio_iommufd_emulated_attach_ioas(struct vfio_device vdev, u32 pt_id)
	146	{
	147	struct iommufd_access *user;
	148
	149	lockdep_assert_held(&vdev->dev_set->lock);
	150
	151	user = iommufd_access_create(vdev->iommufd_ictx, *pt_id, &vfio_user_ops,
	152	vdev);
	153	if (IS_ERR(user))
	154	return PTR_ERR(user);
	155	vdev->iommufd_access = user;
	156	return 0;
	157	}
	158	EXPORT_SYMBOL_GPL(vfio_iommufd_emulated_attach_ioas);