Added checking in copy_from_user and copy_to_user.

1 files changed, 6 insertions, 1 deletions

diff --git a/ncr.c b/ncr.c
index 352e0edf0a0..abdad6d0a7a 100644
--- a/ncr.c
+++ b/ncr.c
@@ -78,13 +78,18 @@ void ncr_master_key_reset(void)
 static int ncr_master_key_set(void* __user arg)
 {
 struct ncr_master_key_st st;
+int ret;
 
 	if (current_euid() != 0 && !capable(CAP_SYS_ADMIN)) {
 		err();
 		return -EPERM;
 	}
 
-	copy_from_user(&st, arg, sizeof(st));
+	ret = copy_from_user(&st, arg, sizeof(st));
+	if (unlikely(ret)) {
+		err();
+		return ret;
+	}
 
 	if (st.key_size > sizeof(master_key.key.secret.data)) {
 		err();