[PATCH v2 2/2] kunit: avoid memory leak on device register error

[Wander Lairson Costa]

If the device register fails, free the allocated memory before
returning.

Signed-off-by: Wander Lairson Costa <wan...@redhat.com>
Fixes: d03c720e03bd ("kunit: Add APIs for managing devices")
---
 lib/kunit/device.c | 1 +
 1 file changed, 1 insertion(+)

diff --git a/lib/kunit/device.c b/lib/kunit/device.c
index 25c81ed465fb..d8c09dcb3e79 100644
--- a/lib/kunit/device.c
+++ b/lib/kunit/device.c
@@ -131,6 +131,7 @@ static struct kunit_device 
*kunit_device_register_internal(struct kunit *test,
        err = device_register(&kunit_dev->dev);
        if (err) {
                put_device(&kunit_dev->dev);
+               kfree(kunit_dev);
                return ERR_PTR(err);
        }
 
-- 
2.44.0