@@ -131,11 +131,22 @@ struct mdev_driver vfio_mdev_driver = {
static int __init vfio_mdev_init(void)
{
- return mdev_register_driver(&vfio_mdev_driver, THIS_MODULE);
+ int ret;
+
+ ret = mdev_register_driver(&vfio_mdev_driver, THIS_MODULE);
+ if (ret)
+ return ret;
+
+ ret = vfio_register_bus_driver(&vfio_mdev_driver.driver);
+ if (ret)
+ mdev_unregister_driver(&vfio_mdev_driver);
+
+ return ret;
}
static void __exit vfio_mdev_exit(void)
{
+ vfio_unregister_bus_driver(&vfio_mdev_driver.driver);
mdev_unregister_driver(&vfio_mdev_driver);
}
@@ -1397,6 +1397,7 @@ static void vfio_pci_try_bus_reset(struct vfio_pci_device *vdev)
static void __exit vfio_pci_cleanup(void)
{
+ vfio_unregister_bus_driver(&vfio_pci_driver.driver);
pci_unregister_driver(&vfio_pci_driver);
vfio_pci_uninit_perm_bits();
}
@@ -1456,6 +1457,12 @@ static int __init vfio_pci_init(void)
if (ret)
goto out_driver;
+ ret = vfio_register_bus_driver(&vfio_pci_driver.driver);
+ if (ret) {
+ pci_unregister_driver(&vfio_pci_driver);
+ goto out_driver;
+ }
+
vfio_pci_fill_ids();
return 0;
@@ -109,7 +109,29 @@ static int vfio_amba_remove(struct amba_device *adev)
},
};
-module_amba_driver(vfio_amba_driver);
+static void __exit vfio_amba_exit(void)
+{
+ vfio_unregister_bus_driver(&vfio_amba_driver.drv);
+ amba_driver_unregister(&vfio_amba_driver);
+}
+
+static int __init vfio_amba_init(void)
+{
+ int ret;
+
+ ret = amba_driver_register(&vfio_amba_driver);
+ if (ret)
+ return ret;
+
+ ret = vfio_register_bus_driver(&vfio_amba_driver.drv);
+ if (ret)
+ amba_driver_unregister(&vfio_amba_driver);
+
+ return ret;
+}
+
+module_init(vfio_amba_init);
+module_exit(vfio_amba_exit);
MODULE_VERSION(DRIVER_VERSION);
MODULE_LICENSE("GPL v2");
@@ -100,7 +100,29 @@ static int vfio_platform_remove(struct platform_device *pdev)
},
};
-module_platform_driver(vfio_platform_driver);
+static void __exit vfio_platform_exit(void)
+{
+ vfio_unregister_bus_driver(&vfio_platform_driver.driver);
+ platform_driver_unregister(&vfio_platform_driver);
+}
+
+static int __init vfio_platform_init(void)
+{
+ int ret;
+
+ ret = platform_driver_register(&vfio_platform_driver);
+ if (ret)
+ return ret;
+
+ ret = vfio_register_bus_driver(&vfio_platform_driver.driver);
+ if (ret)
+ platform_driver_unregister(&vfio_platform_driver);
+
+ return ret;
+}
+
+module_init(vfio_platform_init);
+module_exit(vfio_platform_exit);
MODULE_VERSION(DRIVER_VERSION);
MODULE_LICENSE("GPL v2");