fix hook support for Jackalope

googleprojectzero · Dec 21, 2024 · 6b7383c · 6b7383c
1 parent c0aeb2b
commit 6b7383c
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/arch/arm64/arm64_litecov.cpp b/arch/arm64/arm64_litecov.cpp
@@ -90,6 +90,10 @@ InstructionResult LiteCov::InstrumentInstruction(ModuleInfo *module,
                                                  Instruction &inst,
                                                  size_t bb_address,
                                                  size_t instruction_address) {
+
+  InstructionResult tinyinst_ret = TinyInst::InstrumentInstruction(module, inst, bb_address, instruction_address);
+  if(tinyinst_ret != INST_NOTHANDLED) return tinyinst_ret;
+
   if (!compare_coverage) {
     return INST_NOTHANDLED;
   }

diff --git a/arch/x86/x86_litecov.cpp b/arch/x86/x86_litecov.cpp
@@ -154,6 +154,10 @@ InstructionResult LiteCov::InstrumentInstruction(ModuleInfo *module,
                                                  Instruction &inst,
                                                  size_t bb_address,
                                                  size_t instruction_address) {
+
+  InstructionResult tinyinst_ret = TinyInst::InstrumentInstruction(module, inst, bb_address, instruction_address);
+  if(tinyinst_ret != INST_NOTHANDLED) return tinyinst_ret;
+
   if (!compare_coverage) {
     return INST_NOTHANDLED;
   }