[Rust] More session scoped logs

Assign the `session_id` span field before calling into the relocation handler so logs get scoped to the specific view.

2 files changed, 4 insertions, 0 deletions

diff --git a/rust/src/function_recognizer.rs b/rust/src/function_recognizer.rs
index 0d85d827..f46c741d 100644
--- a/rust/src/function_recognizer.rs
+++ b/rust/src/function_recognizer.rs
@@ -50,6 +50,7 @@ where
         let bv = unsafe { BinaryView::from_raw(bv).to_owned() };
         let func = unsafe { Function::from_raw(func).to_owned() };
         let llil = unsafe { LowLevelILFunction::from_raw(llil).to_owned() };
+        let _span = ffi_span!("FunctionRecognizer::recognize_low_level_il", bv);
         context.recognizer.recognize_low_level_il(&bv, &func, &llil)
     }
 
@@ -66,6 +67,7 @@ where
         let bv = unsafe { BinaryView::from_raw(bv).to_owned() };
         let func = unsafe { Function::from_raw(func).to_owned() };
         let mlil = unsafe { MediumLevelILFunction::from_raw(mlil).to_owned() };
+        let _span = ffi_span!("FunctionRecognizer::recognize_medium_level_il", bv);
         context
             .recognizer
             .recognize_medium_level_il(&bv, &func, &mlil)
diff --git a/rust/src/relocation.rs b/rust/src/relocation.rs
index a4f9667e..a2c38221 100644
--- a/rust/src/relocation.rs
+++ b/rust/src/relocation.rs
@@ -442,6 +442,7 @@ where
             .iter()
             .map(RelocationInfo::from_raw)
             .collect::<Vec<_>>();
+        let _span = ffi_span!("RelocationHandler::get_relocation_info", bv);
         let ok =
             custom_handler.get_relocation_info(bv.as_ref(), arch.as_ref(), info.as_mut_slice());
         for (result, info) in result.iter_mut().zip(info.iter()) {
@@ -466,6 +467,7 @@ where
         let arch = unsafe { CoreArchitecture::from_raw(arch) };
         let reloc = unsafe { Relocation::from_raw(reloc) };
         let dest = unsafe { core::slice::from_raw_parts_mut(dest, len) };
+        let _span = ffi_span!("RelocationHandler::apply_relocation", bv);
         custom_handler.apply_relocation(bv.as_ref(), arch.as_ref(), &reloc, dest)
     }