From fe5d8fc71f87d338b5a8a1198dbb0ba02b95b370 Mon Sep 17 00:00:00 2001
From: tinhnt86 <tinhnt86@gmail.com>
Date: Thu, 8 Oct 2020 16:46:43 +0700
Subject: [PATCH] mcctrl_getrusage: validate os input

Change-Id: I97908069f8bc4703b99f9ffca94f3dd33eb64cc4
---
 executer/kernel/mcctrl/control.c | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/executer/kernel/mcctrl/control.c b/executer/kernel/mcctrl/control.c
index 782854fe..42fe9aea 100644
--- a/executer/kernel/mcctrl/control.c
+++ b/executer/kernel/mcctrl/control.c
@@ -2481,6 +2481,10 @@ long mcctrl_getrusage(ihk_os_t ihk_os, struct mcctrl_ioctl_getrusage_desc *__use
 	unsigned long ut;
 	unsigned long st;
 
+	if (!ihk_os || ihk_host_validate_os(ihk_os)) {
+		return -EINVAL;
+	}
+
 	ret = copy_from_user(&desc, _desc, sizeof(struct mcctrl_ioctl_getrusage_desc));
 	if (ret != 0) {
 		printk("%s: copy_from_user failed\n", __FUNCTION__);