再做了一些优化

Author: AndroidReverser-Test

README.md

@@ -13,6 +13,15 @@
 
 **set_fun_info**函数用于在so文件的指定偏移处设置uprobe挂载点，并可传递指定函数名。
 
+**set_fun_info2**函数用于在set_fun_info函数设置成功但失效的情况下，与set_fun_info函数作用一致
+
 **clear_all_uprobes**函数用于清除所有的uprobe挂载点。
 
-上述函数的返回结果有SET_TRACE_SUCCESS、SET_TRACE_ERROR两种，分别表示设置成功和失败。
+上述函数的返回结果有SET_TRACE_SUCCESS、SET_TRACE_ERROR两种，分别表示设置成功和失败。
+
+# 一些疑惑
+set_fun_info2函数其实就是将传入的函数偏移-0x1000再传递到内核，为什么要这样做？其实是因为
+在内核使用uprobe_register函数注册uprobe挂载点的时候在一些情况下会出现实际注册的函数偏移比
+传入的偏移多上0x1000，而如果我们传入的偏移-0x1000就正好抵消了这个影响。至于为什么会这样，
+我翻阅linux内核源码发现问题出现于[内存地址计算错误](https://elixir.bootlin.com/linux/v5.15.74/source/kernel/events/uprobes.c#L1004),从这里开始内存地址就多加上了0x1000，我推测是内存页少算了
+一页，但是具体原因不清楚。

kernel_trace.c

@@ -139,8 +139,7 @@ static int trace_handler(struct uprobe_consumer *self, struct mpt_regs *regs){
         if(tfun){
             goto target_out;
         }else{
-            fun_offset = fun_offset - 0x1000;
-            tfun = search_key_value(&fun_info_tree,fun_offset);
+            tfun = search_key_value(&fun_info_tree,fun_offset- 0x1000);
             if(likely(tfun)){
                 goto target_out;
             }
@@ -150,7 +149,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);
+//    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);
     if(unlikely(trace_printk_ret<0)){
         logke("+Test-Log+ trace_printk error\n");

user/uprobe_trace_user.h

@@ -37,6 +37,13 @@ int set_fun_info(unsigned long fun_offset,char *fun_name){
     return ret;
 }
 
+//set_fun_info函数设置成功但失效的情况下再用这个函数
+//，最好在一个程序内不要与set_fun_info函数同时使用
+int set_fun_info2(unsigned long fun_offset,char *fun_name){
+    int ret = syscall(__NR_mincore,fun_offset-0x1000,TRACE_FLAG+SET_FUN_INFO,fun_name);
+    return ret;
+}
+
 int clear_all_uprobes(){
     int ret = syscall(__NR_mincore,0,TRACE_FLAG+CLEAR_UPROBE,"");
     return ret;