From 8c0b2ab6cea612bc1d65da4d8ac0ecf3fe8e6855 Mon Sep 17 00:00:00 2001 From: tinhnt86 Date: Thu, 8 Oct 2020 16:41:24 +0700 Subject: [PATCH] mcctrl_perf_num: check "os" argument Change-Id: I13c8b0c337cac9bbb240667808e871defce34aab --- executer/kernel/mcctrl/control.c | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/executer/kernel/mcctrl/control.c b/executer/kernel/mcctrl/control.c index 0b930910..f36d00ca 100644 --- a/executer/kernel/mcctrl/control.c +++ b/executer/kernel/mcctrl/control.c @@ -2172,7 +2172,13 @@ static DECLARE_WAIT_QUEUE_HEAD(perfctrlq); long mcctrl_perf_num(ihk_os_t os, unsigned long arg) { - struct mcctrl_usrdata *usrdata = ihk_host_os_get_usrdata(os); + struct mcctrl_usrdata *usrdata; + + if (!os || ihk_host_validate_os(os)) { + return -EINVAL; + } + + usrdata = ihk_host_os_get_usrdata(os); if (!usrdata) { pr_err("%s: error: mcctrl_usrdata not found\n", __func__);