[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.

1 files changed, 2 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)