kp: kp_set_mode(): Handle invalid mode arguments

Signed-off-by: Dakkshesh <dakkshesh5@gmail.com>

Author: beakthoven

main.c

@@ -57,6 +57,12 @@ void kp_set_mode_rollback(unsigned int level, unsigned int duration_ms)
 		return;
 #endif
 
+	if (unlikely(level > 3)) {
+		pr_err("%s: Invalid mode requested, Skipping mode change",
+		       __func__);
+		return;
+	}
+
 	mutex_lock(&kplock);
 	if (level && duration_ms && auto_kprofiles) {
 		kp_override_mode = level;
@@ -82,6 +88,12 @@ void kp_set_mode(unsigned int level)
 		return;
 #endif
 
+	if (unlikely(level > 3)) {
+		pr_err("%s: Invalid mode requested, Skipping mode change",
+		       __func__);
+		return;
+	}
+
 	if (level && auto_kprofiles)
 		kp_mode = level;
 }