2 files changed, 14 insertions, 2 deletions

diff --git a/net/9p/trans_virtio.c b/net/9p/trans_virtio.c
index 549938af02e7..a3cd90a74012 100644
--- a/net/9p/trans_virtio.c
+++ b/net/9p/trans_virtio.c
@@ -767,10 +767,16 @@ static struct p9_trans_module p9_virtio_trans = {
 /* The standard init function */
 static int __init p9_virtio_init(void)
 {
+        int rc;
+
         INIT_LIST_HEAD(&virtio_chan_list);
 
         v9fs_register_trans(&p9_virtio_trans);
-        return register_virtio_driver(&p9_virtio_drv);
+        rc = register_virtio_driver(&p9_virtio_drv);
+        if (rc)
+                v9fs_unregister_trans(&p9_virtio_trans);
+
+        return rc;
 }
 
 static void __exit p9_virtio_cleanup(void)
diff --git a/net/9p/trans_xen.c b/net/9p/trans_xen.c
index 29420ebb8f07..3963eb11c3fb 100644
--- a/net/9p/trans_xen.c
+++ b/net/9p/trans_xen.c
@@ -530,13 +530,19 @@ static struct xenbus_driver xen_9pfs_front_driver = {
 
 static int p9_trans_xen_init(void)
 {
+        int rc;
+
         if (!xen_domain())
                 return -ENODEV;
 
         pr_info("Initialising Xen transport for 9pfs\n");
 
         v9fs_register_trans(&p9_xen_trans);
-        return xenbus_register_frontend(&xen_9pfs_front_driver);
+        rc = xenbus_register_frontend(&xen_9pfs_front_driver);
+        if (rc)
+                v9fs_unregister_trans(&p9_xen_trans);
+
+        return rc;
 }
 module_init(p9_trans_xen_init);