From: Chanwoo Choi <cw00.choi@samsung.com>
Date: Thu, 19 Sep 2024 01:39:37 +0000 (+0900)
Subject: util: kernel: Check for returned value is NULL in kernel_get_process_taskstats()
X-Git-Tag: accepted/tizen/unified/20240919.163750^0
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=refs%2Fheads%2Faccepted%2Ftizen_unified_dev;p=platform%2Fcore%2Fsystem%2Fpass.git

util: kernel: Check for returned value is NULL in kernel_get_process_taskstats()

Change-Id: I0dba6c77957722aed371b5dd1551454f144d2ca1
Signed-off-by: Chanwoo Choi <cw00.choi@samsung.com>
---

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);