Messages in this thread Patch in this message | | | From | Xiaoke Wang <> | Subject | [PATCH] init/initramfs.c: check the return value of kstrdup() | Date | Mon, 13 Dec 2021 16:58:16 +0800 |
| |
kstrdup() is also a memory allocation function and it is similar with kmalloc() in some way. Once some internal memory errors happen, it will return NULL. It is better to check the return value of it so to catch the memory error in time.
Signed-off-by: Xiaoke Wang <xkernel.wang@foxmail.com> --- init/initramfs.c | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-)
diff --git a/init/initramfs.c b/init/initramfs.c index a842c05..49deffb 100644 --- a/init/initramfs.c +++ b/init/initramfs.c @@ -139,8 +139,12 @@ static void __init dir_add(const char *name, time64_t mtime) struct dir_entry *de = kmalloc(sizeof(struct dir_entry), GFP_KERNEL); if (!de) panic_show_mem("can't allocate dir_entry buffer"); - INIT_LIST_HEAD(&de->list); de->name = kstrdup(name, GFP_KERNEL); + if (!de->name) { + kfree(de); + panic_show_mem("can't duplicate dir name"); + } + INIT_LIST_HEAD(&de->list); de->mtime = mtime; list_add(&de->list, &dir_list); } --
| |