stalker-arm64: fix missing ret_reg

frida · Feb 13, 2024 · 268368c · 268368c
1 parent 1ffe22f
commit 268368c
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/gum/backend-arm64/gumstalker-arm64.c b/gum/backend-arm64/gumstalker-arm64.c
@@ -692,7 +692,7 @@ static void gum_exec_block_write_jmp_transfer_code (GumExecBlock * block,
 static void gum_exec_block_write_ret_transfer_code (GumExecBlock * block,
     GumGeneratorContext * gc, arm64_reg ret_reg);
 static void gum_exec_block_write_chaining_return_code (GumExecBlock * block,
-    GumGeneratorContext * gc);
+    GumGeneratorContext * gc, arm64_reg ret_reg);
 static void gum_exec_block_write_slab_transfer_code (GumArm64Writer * from,
     GumArm64Writer * to);
 static void gum_exec_block_backpatch_slab (GumExecBlock * block,
@@ -2960,7 +2960,7 @@ gum_stalker_iterator_put_chaining_return (GumStalkerIterator * self)
 
   gum_exec_block_write_adjust_depth (block, gc->code_writer, -1);
 
-  gum_exec_block_write_chaining_return_code (block, gc);
+  gum_exec_block_write_chaining_return_code (block, gc, ARM64_REG_LR);
 }
 
 GumMemoryAccess
@@ -5038,12 +5038,13 @@ gum_exec_block_write_ret_transfer_code (GumExecBlock * block,
                                         GumGeneratorContext * gc,
                                         arm64_reg ret_reg)
 {
-  gum_exec_block_write_chaining_return_code (block, gc);
+  gum_exec_block_write_chaining_return_code (block, gc, ret_reg);
 }
 
 static void
 gum_exec_block_write_chaining_return_code (GumExecBlock * block, 
-                                           GumGeneratorContext  * gc)
+                                           GumGeneratorContext  * gc,
+                                           arm64_reg ret_reg)
 {
   GumArm64Writer * cw = gc->code_writer;
   GumArm64Writer * cws = gc->slow_writer;