pstore/ram: Add check for kstrdup

Add check for the return value of kstrdup() and return the error
if it fails in order to avoid NULL pointer dereference.

Fixes: e163fdb3f7f8 ("pstore/ram: Regularize prz label allocation lifetime")
Signed-off-by: Jiasheng Jiang <[email protected]>
Signed-off-by: Kees Cook <[email protected]>
Link: https://lore.kernel.org/r/[email protected]

diff --git a/fs/pstore/ram_core.c b/fs/pstore/ram_core.c
index 966191d3a5ba20fb9f5dadd6a07cccad8b1942eb..85aaf0fc6d7d1847631ccf4ef0f62ec2ee16f083 100644 (file)
--- a/fs/pstore/ram_core.c
+++ b/fs/pstore/ram_core.c
@@ -599,6 +599,8 @@ struct persistent_ram_zone *persistent_ram_new(phys_addr_t start, size_t size,
        raw_spin_lock_init(&prz->buffer_lock);
        prz->flags = flags;
        prz->label = kstrdup(label, GFP_KERNEL);
+       if (!prz->label)
+               goto err;
 
        ret = persistent_ram_buffer_map(start, size, prz, memtype);
        if (ret)