mailbox/omap: unregister mbox classmailbox-linux-4.14.y

[ Upstream commit 1f90a2162fb3cdfd9c44380bf16209af00f7acbe ]

platform_driver_register() can fail here and we must unregister mbox
class.

Signed-off-by: Arvind Yadav <arvind.yadav.cs@gmail.com>
Acked-by: Suman Anna <s-anna@ti.com>
Signed-off-by: Jassi Brar <jaswinder.singh@linaro.org>
[s-anna@ti.com: cherry-pick commit '1f90a2162fb3' from v4.15]
Signed-off-by: Suman Anna <s-anna@ti.com>

1 files changed, 5 insertions, 1 deletions

diff --git a/drivers/mailbox/omap-mailbox.c b/drivers/mailbox/omap-mailbox.c
index c5e8b9cb170d..2517038a8452 100644
--- a/drivers/mailbox/omap-mailbox.c
+++ b/drivers/mailbox/omap-mailbox.c
@@ -906,7 +906,11 @@ static int __init omap_mbox_init(void)
         mbox_kfifo_size = max_t(unsigned int, mbox_kfifo_size,
                                                         sizeof(mbox_msg_t));
 
-        return platform_driver_register(&omap_mbox_driver);
+        err = platform_driver_register(&omap_mbox_driver);
+        if (err)
+                class_unregister(&omap_mbox_class);
+
+        return err;
 }
 subsys_initcall(omap_mbox_init);