greybus: vibrator: add missing protocol-register error handling

[linux-2.6-microblaze.git] / drivers / staging / greybus / vibrator.c

diff --git a/drivers/staging/greybus/vibrator.c b/drivers/staging/greybus/vibrator.c
index fd40cda..2b80326 100644 (file)
--- a/drivers/staging/greybus/vibrator.c
+++ b/drivers/staging/greybus/vibrator.c
@@ -171,7 +171,16 @@ static __init int protocol_init(void)
        if (retval)
                return retval;
 
-       return gb_protocol_register(&vibrator_protocol);
+       retval = gb_protocol_register(&vibrator_protocol);
+       if (retval)
+               goto err_class_unregister;
+
+       return 0;
+
+err_class_unregister:
+       class_unregister(&vibrator_class);
+
+       return retval;
 }
 module_init(protocol_init);