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

From: Wander Lairson Costa
Date: Thu Apr 18 2024 - 12:01:29 EST


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

Signed-off-by: Wander Lairson Costa <wander@xxxxxxxxxx>
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