Re: [PATCH] tty: vt: Check for NULL pointer after calling kzalloc

From: Jiri Slaby
Date: Fri Jan 21 2022 - 02:01:33 EST


On 21. 01. 22, 7:53, Jiasheng Jiang wrote:
As the potential failure of the allocation, the kzalloc() will return
NULL pointer.
Therefore, it should be better to check it in order to avoid the
dereference of the NULL pointer.
When it fails, we should free all the allocated memory and return error
number.
To make the code more clear, I use the 'err_free', like how
vc_allocate() deals with the allocation failure.

https://lore.kernel.org/all/X+CRTqDQwVYSZQBi@xxxxxxxxx/

Fixes: a5f4f52e8211 ("vt: use kzalloc() instead of the bootmem allocator")
Signed-off-by: Jiasheng Jiang <jiasheng@xxxxxxxxxxx>
---
drivers/tty/vt/vt.c | 19 +++++++++++++++++++
1 file changed, 19 insertions(+)

diff --git a/drivers/tty/vt/vt.c b/drivers/tty/vt/vt.c
index f8c87c4d7399..343fa6fffc18 100644
--- a/drivers/tty/vt/vt.c
+++ b/drivers/tty/vt/vt.c
@@ -3519,11 +3519,17 @@ static int __init con_init(void)
for (currcons = 0; currcons < MIN_NR_CONSOLES; currcons++) {
vc_cons[currcons].d = vc = kzalloc(sizeof(struct vc_data), GFP_NOWAIT);
+ if (!vc)
+ goto err_free;
+
INIT_WORK(&vc_cons[currcons].SAK_work, vc_SAK);
tty_port_init(&vc->port);
visual_init(vc, currcons, 1);
/* Assuming vc->vc_{cols,rows,screenbuf_size} are sane here. */
vc->vc_screenbuf = kzalloc(vc->vc_screenbuf_size, GFP_NOWAIT);
+ if (!vc->vc_screenbuf)
+ goto err_free_vc;
+
vc_init(vc, vc->vc_rows, vc->vc_cols,
currcons || !vc->vc_sw->con_save_screen);
}
@@ -3545,6 +3551,19 @@ static int __init con_init(void)
register_console(&vt_console_driver);
#endif
return 0;
+err_free_vc:
+ visual_deinit(vc);
+ kfree(vc);
+ vc_cons[currcons].d = NULL;
+err_free:
+ for (i = 0; i < currcons; i++) {
+ kfree(vc_cons[currcons].d->vc_screenbuf);
+ visual_deinit(vc_cons[currcons].d);
+ kfree(vc_cons[currcons].d);
+ vc_cons[currcons].d = NULL;
+ }
+ console_unlock();
+ return -ENOMEM;
}
console_initcall(con_init);


--
js