Merge some files and rewrite the update history

kernel/ksud.c

@@ -20,6 +20,7 @@
 #include "kernel_compat.h"
 #include "selinux/selinux.h"
 
+
 static const char KERNEL_SU_RC[] =
 	"\n"
 
@@ -47,12 +48,21 @@ static void stop_vfs_read_hook();
 static void stop_execve_hook();
 static void stop_input_hook();
 
+#ifdef CONFIG_KPROBES
 static struct work_struct stop_vfs_read_work;
 static struct work_struct stop_execve_hook_work;
 static struct work_struct stop_input_hook_work;
+#else
+bool ksu_vfs_read_hook __read_mostly = true;
+bool ksu_execveat_hook __read_mostly = true;
+bool ksu_input_hook __read_mostly = true;
+#endif
 
 u32 ksu_devpts_sid;
 
+// Detect whether it is on or not
+static bool is_boot_phase = true;
+
 void on_post_fs_data(void)
 {
 	static bool done = false;
@@ -68,6 +78,9 @@ void on_post_fs_data(void)
 
 	ksu_devpts_sid = ksu_get_devpts_sid();
 	pr_info("devpts sid: %d\n", ksu_devpts_sid);
+
+	// End of boot state
+    is_boot_phase = false;
 }
 
 #define MAX_ARG_STRINGS 0x7FFFFFFF
@@ -144,6 +157,11 @@ int ksu_handle_execveat_ksud(int *fd, struct filename **filename_ptr,
 			     struct user_arg_ptr *argv,
 			     struct user_arg_ptr *envp, int *flags)
 {
+#ifndef CONFIG_KPROBES
+ 	if (!ksu_execveat_hook) {
+ 		return 0;
+ 	}
+ #endif
 	struct filename *filename;
 
 	static const char app_process[] = "/system/bin/app_process";
@@ -295,6 +313,11 @@ static ssize_t read_iter_proxy(struct kiocb *iocb, struct iov_iter *to)
 int ksu_handle_vfs_read(struct file **file_ptr, char __user **buf_ptr,
 			size_t *count_ptr, loff_t **pos)
 {
+#ifndef CONFIG_KPROBES
+ 	if (!ksu_vfs_read_hook) {
+ 		return 0;
+ 	}
+#endif
 	struct file *file;
 	char __user *buf;
 	size_t count;
@@ -403,10 +426,15 @@ static bool is_volumedown_enough(unsigned int count)
 int ksu_handle_input_handle_event(unsigned int *type, unsigned int *code,
 				  int *value)
 {
+#ifndef CONFIG_KPROBES
+ 	if (!ksu_input_hook) {
+ 		return 0;
+ 	}
+#endif
 	if (*type == EV_KEY && *code == KEY_VOLUMEDOWN) {
 		int val = *value;
 		pr_info("KEY_VOLUMEDOWN val: %d\n", val);
-		if (val) {
+		if (val && is_boot_phase) {
 			// key pressed, count it
 			volumedown_pressed_count += 1;
 			if (is_volumedown_enough(volumedown_pressed_count)) {
@@ -440,6 +468,7 @@ bool ksu_is_safe_mode()
 	return false;
 }
 
+#ifdef CONFIG_KPROBES
 static int sys_execve_handler_pre(struct kprobe *p, struct pt_regs *regs)
 {
 	struct pt_regs *real_regs = PT_REAL_REGS(regs);
@@ -511,17 +540,28 @@ static void do_stop_input_hook(struct work_struct *work)
 {
 	unregister_kprobe(&input_event_kp);
 }
+#endif
 
 static void stop_vfs_read_hook()
 {
+#ifdef CONFIG_KPROBES
 	bool ret = schedule_work(&stop_vfs_read_work);
 	pr_info("unregister vfs_read kprobe: %d!\n", ret);
+#else
+ 	ksu_vfs_read_hook = false;
+ 	pr_info("stop vfs_read_hook\n");
+#endif
 }
 
 static void stop_execve_hook()
 {
+#ifdef CONFIG_KPROBES
 	bool ret = schedule_work(&stop_execve_hook_work);
 	pr_info("unregister execve kprobe: %d!\n", ret);
+#else
+ 	ksu_execveat_hook = false;
+ 	pr_info("stop execve_hook\n");
+#endif
 }
 
 static void stop_input_hook()
@@ -531,13 +571,19 @@ static void stop_input_hook()
 		return;
 	}
 	input_hook_stopped = true;
+#ifdef CONFIG_KPROBES
 	bool ret = schedule_work(&stop_input_hook_work);
 	pr_info("unregister input kprobe: %d!\n", ret);
+#else
+ 	ksu_input_hook = false;
+ 	pr_info("stop input_hook\n");
+#endif
 }
 
 // ksud: module support
 void ksu_ksud_init()
 {
+#ifdef CONFIG_KPROBES
 	int ret;
 
 	ret = register_kprobe(&execve_kp);
@@ -552,12 +598,17 @@ void ksu_ksud_init()
 	INIT_WORK(&stop_vfs_read_work, do_stop_vfs_read_hook);
 	INIT_WORK(&stop_execve_hook_work, do_stop_execve_hook);
 	INIT_WORK(&stop_input_hook_work, do_stop_input_hook);
+#endif
 }
 
 void ksu_ksud_exit()
 {
+#ifdef CONFIG_KPROBES
 	unregister_kprobe(&execve_kp);
 	// this should be done before unregister vfs_read_kp
 	// unregister_kprobe(&vfs_read_kp);
 	unregister_kprobe(&input_event_kp);
+
+	is_boot_phase = false;
+#endif
 }