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 6259968b231ca30aeb270aaf53b4e3766cc4cf0c..e3dedda80afd536ca96eb14fa95c75303167eab4 100644 (file)
--- 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);