From 246022805272f0a2392564c8bb4cb8e51a053355 Mon Sep 17 00:00:00 2001
From: Balazs Gerofi <bgerofi@riken.jp>
Date: Tue, 12 Jan 2021 20:37:54 +0900
Subject: [PATCH] mcctrl: abort on invalid addr in mcexec_transfer_image()

Change-Id: Ic064b6ffc30368ff1d3dfb14403e524cbb837ce5
---
 executer/kernel/mcctrl/control.c | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/executer/kernel/mcctrl/control.c b/executer/kernel/mcctrl/control.c
index c6d220ba..c2ea5095 100644
--- a/executer/kernel/mcctrl/control.c
+++ b/executer/kernel/mcctrl/control.c
@@ -281,7 +281,12 @@ int mcexec_transfer_image(ihk_os_t os, struct remote_transfer *__user upt)
 #else
 	rpm = ihk_device_map_virtual(ihk_os_to_dev(os), phys, PAGE_SIZE, NULL, 0);
 #endif
-	
+
+	if (!rpm) {
+		pr_err("%s(): error: invalid remote address\n", __func__);
+		return -EFAULT;
+	}
+
 	if (pt.direction == MCEXEC_UP_TRANSFER_TO_REMOTE) {
 		if (copy_from_user(rpm, pt.userp, pt.size)) {
 			ret = -EFAULT;