Merge pull request #1928 from mlgiraud/feat/add_start_end_to_hook

Feature: Bindings: add start end to block hook
unicorn-engine · Mar 11, 2024 · 227ff1d · 227ff1d
2 parents e03109d + d4f5bf1
commit 227ff1d
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/bindings/rust/src/lib.rs b/bindings/rust/src/lib.rs
@@ -620,6 +620,8 @@ impl<'a, D> Unicorn<'a, D> {
     /// Add a block hook.
     pub fn add_block_hook<F: 'a>(
         &mut self,
+        begin: u64,
+        end: u64,
         callback: F,
     ) -> Result<UcHookId, uc_error>
     where
@@ -638,8 +640,8 @@ impl<'a, D> Unicorn<'a, D> {
                 HookType::BLOCK,
                 ffi::block_hook_proxy::<D, F> as _,
                 user_data.as_mut() as *mut _ as _,
-                1,
-                0,
+                begin,
+                end,
             )
         }
         .and_then(|| {

diff --git a/tests/rust-tests/main.rs b/tests/rust-tests/main.rs
@@ -763,7 +763,7 @@ fn x86_block_callback() {
     assert_eq!(emu.mem_write(0x1000, &x86_code32), Ok(()));
 
     let hook = emu
-        .add_block_hook(callback)
+        .add_block_hook(1, 0, callback)
         .expect("failed to add block hook");
     assert_eq!(
         emu.emu_start(0x1000, 0x1002, 10 * SECOND_SCALE, 1000),