rtc: nvmem: fix possible use after free

In cas of probe failure, devres may free the memory allocated for
rtc->nvram before devm_rtc_release_device() is called. This leads to
rtc_nvram_unregister using it after being freed which may lead to a crash.

This has been shown to happen after commit 461e557b9727 ("rtc: nvmem: use
devm_nvmem_register()")

Reported-by: kernel test robot <lkp@intel.com>
Signed-off-by: Alexandre Belloni <alexandre.belloni@bootlin.com>

diff --git a/drivers/rtc/nvmem.c b/drivers/rtc/nvmem.c
index ebdfe8e..dce518d 100644 (file)
--- a/drivers/rtc/nvmem.c
+++ b/drivers/rtc/nvmem.c
@@ -12,6 +12,7 @@
 #include <linux/types.h>
 #include <linux/nvmem-consumer.h>
 #include <linux/rtc.h>
+#include <linux/slab.h>
 #include <linux/sysfs.h>
 
 /*
@@ -45,9 +46,7 @@ static int rtc_nvram_register(struct rtc_device *rtc,
 {
        int err;
 
-       rtc->nvram = devm_kzalloc(rtc->dev.parent,
-                               sizeof(struct bin_attribute),
-                               GFP_KERNEL);
+       rtc->nvram = kzalloc(sizeof(struct bin_attribute), GFP_KERNEL);
        if (!rtc->nvram)
                return -ENOMEM;
 
@@ -64,7 +63,7 @@ static int rtc_nvram_register(struct rtc_device *rtc,
        err = sysfs_create_bin_file(&rtc->dev.parent->kobj,
                                    rtc->nvram);
        if (err) {
-               devm_kfree(rtc->dev.parent, rtc->nvram);
+               kfree(rtc->nvram);
                rtc->nvram = NULL;
        }
 
@@ -74,6 +73,8 @@ static int rtc_nvram_register(struct rtc_device *rtc,
 static void rtc_nvram_unregister(struct rtc_device *rtc)
 {
        sysfs_remove_bin_file(&rtc->dev.parent->kobj, rtc->nvram);
+       kfree(rtc->nvram);
+       rtc->nvram = NULL;
 }
 
 /*