ubi: reset mtd_devs when ubi part fail

[u-boot] / drivers / mtd / ubi / build.c

diff --git a/drivers/mtd/ubi/build.c b/drivers/mtd/ubi/build.c
index 584cf5f22b38f4830ced26e6e011df1bd3115ac5..290d524b1be2577d8a9da710a8c3518cc2d1d632 100644 (file)
--- a/drivers/mtd/ubi/build.c
+++ b/drivers/mtd/ubi/build.c
@@ -1358,6 +1358,10 @@ out_version:
 out_class:
        class_destroy(ubi_class);
 out:
+#ifdef __UBOOT__
+       /* Reset any globals that the driver depends on being zeroed */
+       mtd_devs = 0;
+#endif
        ubi_err("cannot initialize UBI, error %d", err);
        return err;
 }
@@ -1384,6 +1388,10 @@ void ubi_exit(void)
        misc_deregister(&ubi_ctrl_cdev);
        class_remove_file(ubi_class, &ubi_version);
        class_destroy(ubi_class);
+#ifdef __UBOOT__
+       /* Reset any globals that the driver depends on being zeroed */
+       mtd_devs = 0;
+#endif
 }
 module_exit(ubi_exit);