add OnBasicBlcokTranslated callback

googleprojectzero · Dec 22, 2023 · 9cdc11e · 9cdc11e
1 parent 5a45ad4
commit 9cdc11e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/tinyinst.cpp b/tinyinst.cpp
@@ -603,6 +603,8 @@ void TinyInst::TranslateBasicBlock(char *address,
   unwind_generator->OnBasicBlockEnd(module,
     (size_t)address + offset,
     GetCurrentInstrumentedAddress(module));
+
+  OnBasicBlcokTranslated(module, original_offset, original_offset + last_offset);
 }
 
 // starting from address, starts instrumenting code in the module

diff --git a/tinyinst.h b/tinyinst.h
@@ -152,6 +152,7 @@ class TinyInst : public Debugger {
 
   virtual void OnModuleInstrumented(ModuleInfo* module);
   virtual void OnModuleUninstrumented(ModuleInfo* module);
+  virtual void OnBasicBlcokTranslated(ModuleInfo *module, size_t start_offset, size_t end_offset) { }
 
   int32_t sp_offset;
   Assembler* assembler_;