From 4b7e21c94c446fd1995dc2f4ad715640d58cc79e Mon Sep 17 00:00:00 2001 From: Chanwoo Choi Date: Thu, 19 Sep 2024 10:39:37 +0900 Subject: [PATCH] util: kernel: Check for returned value is NULL in kernel_get_process_taskstats() Change-Id: I0dba6c77957722aed371b5dd1551454f144d2ca1 Signed-off-by: Chanwoo Choi --- src/util/kernel.c | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/src/util/kernel.c b/src/util/kernel.c index 6259968..e3dedda 100644 --- a/src/util/kernel.c +++ b/src/util/kernel.c @@ -343,7 +343,14 @@ int kernel_get_process_taskstats(struct taskstats *stats, int cmd_type, pid_t pi if (ret < 0) goto err_genl_close; - cb = nl_cb_get(nl_cb_alloc(NL_CB_CUSTOM)); + cb = nl_cb_alloc(NL_CB_CUSTOM); + if (!cb) + return -ENOMEM; + + cb = nl_cb_get(cb); + if (!cb) + return -EINVAL; + nl_cb_set(cb, NL_CB_VALID, NL_CB_CUSTOM, &parse_task_stats, stats); nl_cb_err(cb, NL_CB_CUSTOM, &print_receive_error, NULL); -- 2.34.1