vfio/fsl-mc: Scan DPRC objects on vfio-fsl-mc driver bind (704f5082) · Commits · 戴 / test

drivers/vfio/fsl-mc/vfio_fsl_mc.c

+91 −0

Original line number	Diff line number	Diff line
		@@ -15,6 +15,8 @@

		#include "vfio_fsl_mc_private.h"

		static struct fsl_mc_driver vfio_fsl_mc_driver;

		static int vfio_fsl_mc_open(void *device_data)
		{
		if (!try_module_get(THIS_MODULE))
		@@ -84,6 +86,80 @@ static const struct vfio_device_ops vfio_fsl_mc_ops = {
		.mmap = vfio_fsl_mc_mmap,
		};

		static int vfio_fsl_mc_bus_notifier(struct notifier_block *nb,
		unsigned long action, void *data)
		{
		struct vfio_fsl_mc_device *vdev = container_of(nb,
		struct vfio_fsl_mc_device, nb);
		struct device *dev = data;
		struct fsl_mc_device *mc_dev = to_fsl_mc_device(dev);
		struct fsl_mc_device *mc_cont = to_fsl_mc_device(mc_dev->dev.parent);

		if (action == BUS_NOTIFY_ADD_DEVICE &&
		vdev->mc_dev == mc_cont) {
		mc_dev->driver_override = kasprintf(GFP_KERNEL, "%s",
		vfio_fsl_mc_ops.name);
		if (!mc_dev->driver_override)
		dev_warn(dev, "VFIO_FSL_MC: Setting driver override for device in dprc %s failed\n",
		dev_name(&mc_cont->dev));
		else
		dev_info(dev, "VFIO_FSL_MC: Setting driver override for device in dprc %s\n",
		dev_name(&mc_cont->dev));
		} else if (action == BUS_NOTIFY_BOUND_DRIVER &&
		vdev->mc_dev == mc_cont) {
		struct fsl_mc_driver *mc_drv = to_fsl_mc_driver(dev->driver);

		if (mc_drv && mc_drv != &vfio_fsl_mc_driver)
		dev_warn(dev, "VFIO_FSL_MC: Object %s bound to driver %s while DPRC bound to vfio-fsl-mc\n",
		dev_name(dev), mc_drv->driver.name);
		}

		return 0;
		}

		static int vfio_fsl_mc_init_device(struct vfio_fsl_mc_device *vdev)
		{
		struct fsl_mc_device *mc_dev = vdev->mc_dev;
		int ret;

		/* Non-dprc devices share mc_io from parent */
		if (!is_fsl_mc_bus_dprc(mc_dev)) {
		struct fsl_mc_device *mc_cont = to_fsl_mc_device(mc_dev->dev.parent);

		mc_dev->mc_io = mc_cont->mc_io;
		return 0;
		}

		vdev->nb.notifier_call = vfio_fsl_mc_bus_notifier;
		ret = bus_register_notifier(&fsl_mc_bus_type, &vdev->nb);
		if (ret)
		return ret;

		/* open DPRC, allocate a MC portal */
		ret = dprc_setup(mc_dev);
		if (ret) {
		dev_err(&mc_dev->dev, "VFIO_FSL_MC: Failed to setup DPRC (%d)\n", ret);
		goto out_nc_unreg;
		}

		ret = dprc_scan_container(mc_dev, false);
		if (ret) {
		dev_err(&mc_dev->dev, "VFIO_FSL_MC: Container scanning failed (%d)\n", ret);
		goto out_dprc_cleanup;
		}

		return 0;

		out_dprc_cleanup:
		dprc_remove_devices(mc_dev, NULL, 0);
		dprc_cleanup(mc_dev);
		out_nc_unreg:
		bus_unregister_notifier(&fsl_mc_bus_type, &vdev->nb);
		vdev->nb.notifier_call = NULL;

		return ret;
		}

		static int vfio_fsl_mc_probe(struct fsl_mc_device *mc_dev)
		{
		struct iommu_group *group;
		@@ -110,8 +186,15 @@ static int vfio_fsl_mc_probe(struct fsl_mc_device *mc_dev)
		dev_err(dev, "VFIO_FSL_MC: Failed to add to vfio group\n");
		goto out_group_put;
		}

		ret = vfio_fsl_mc_init_device(vdev);
		if (ret)
		goto out_group_dev;

		return 0;

		out_group_dev:
		vfio_del_group_dev(dev);
		out_group_put:
		vfio_iommu_group_put(group, dev);
		return ret;
		@@ -126,6 +209,14 @@ static int vfio_fsl_mc_remove(struct fsl_mc_device *mc_dev)
		if (!vdev)
		return -EINVAL;

		if (is_fsl_mc_bus_dprc(mc_dev)) {
		dprc_remove_devices(mc_dev, NULL, 0);
		dprc_cleanup(mc_dev);
		}

		if (vdev->nb.notifier_call)
		bus_unregister_notifier(&fsl_mc_bus_type, &vdev->nb);

		vfio_iommu_group_put(mc_dev->dev.iommu_group, dev);

		return 0;

drivers/vfio/fsl-mc/vfio_fsl_mc_private.h

+1 −0

Original line number	Diff line number	Diff line
		@@ -9,6 +9,7 @@

		struct vfio_fsl_mc_device {
		struct fsl_mc_device *mc_dev;
		struct notifier_block nb;
		};

		#endif /* VFIO_FSL_MC_PRIVATE_H */

Admin message