修改最大HOOk数量

Author: AndroidReverser-Test

kernel_trace.c

@@ -12,7 +12,7 @@
 #include "kernel_trace.h"
 
 KPM_NAME("kernel_trace");
-KPM_VERSION("4.0.0");
+KPM_VERSION("4.5.0");
 KPM_LICENSE("GPL v2");
 KPM_AUTHOR("Test");
 KPM_DESCRIPTION("use uprobe trace some fun in kpm");
@@ -55,6 +55,11 @@ void before_copy_insn(hook_fargs5_t *args, void *udata){
     if(ins_info){
        memcpy((void *)args->arg2,ins_info->value,INS_LEN);
 //       logkd("+Test-Log+ offset:%lx,fix ins:%x %x %x %x\n",offset,ins_info->value[0],ins_info->value[1],ins_info->value[2],ins_info->value[3]);
+
+       rb_erase(&ins_info->node,&fix_ins_tree);
+       kfree(ins_info->value);
+       kfree(ins_info);
+
        args->ret = 0;
        args->skip_origin = 1;
     }
@@ -210,7 +215,7 @@ static int trace_handler(struct uprobe_consumer *self, struct mpt_regs *regs){
 
 target_out:
 //    logkd("+Test-Log+ fun_name:%s,fun_offset:0x%llx calling\n",tfun->value,fun_offset);
-    int trace_printk_ret = trace_printk(_THIS_IP_,"+Test-Log+ fun_name:%s,fun_offset:0x%llx calling\n",tfun->value,fun_offset);
+    int trace_printk_ret = trace_printk(0,"+Test-Log+ fun_name:%s,fun_offset:0x%llx calling\n",tfun->value,fun_offset);
     if(unlikely(trace_printk_ret<0)){
         logke("+Test-Log+ trace_printk error\n");
     }

kernel_trace.h

@@ -3,7 +3,7 @@
 #include "uprobe_trace.h"
 #include "mrbtree.h"
 
-#define _THIS_IP_  ({ __label__ __here; __here: (unsigned long)&&__here; })
+//#define _THIS_IP_  ({ __label__ __here; __here: (unsigned long)&&__here; })
 #define MAX_PATH_LEN 300
 #define MAX_FUN_NAME 150
 #define INS_LEN 4

uprobe_trace.h

@@ -1,5 +1,5 @@
 #define TRACE_FLAG 511
-#define MAX_HOOK_NUM 1000
+#define MAX_HOOK_NUM 10000
 #define SET_TRACE_SUCCESS 1000
 #define SET_TRACE_ERROR 1001
 

user/uprobe_trace_user.h

@@ -3,7 +3,7 @@
 
 
 #define TRACE_FLAG 511
-#define MAX_HOOK_NUM 1000
+#define MAX_HOOK_NUM 10000
 #define SET_TRACE_SUCCESS 1000
 #define SET_TRACE_ERROR 1001
 
@@ -52,4 +52,9 @@ int set_fun_info(unsigned long uprobe_offset,unsigned long fun_offset,char *fun_
 int clear_all_uprobes(){
     int ret = syscall(__NR_mincore,0,TRACE_FLAG+CLEAR_UPROBE,"");
     return ret;
+}
+
+int set_fun_ins(unsigned long uprobe_offset,char *fix_insn){
+    int fix_insn_ret = syscall(__NR_mincore,uprobe_offset,TRACE_FLAG+FIX_ORI_INS,fix_insn);
+    return fix_insn_ret;
 }