Fix FlowGraph::low_level_il crash in Rust APi with no LLIL function o…

…bject
Vector35 · Feb 11, 2025 · f7b92a5 · f7b92a5
1 parent 60927d1
commit f7b92a5
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/rust/src/flowgraph.rs b/rust/src/flowgraph.rs
@@ -58,12 +58,14 @@ impl FlowGraph {
     pub fn low_level_il(&self) -> Result<Ref<RegularLowLevelILFunction<CoreArchitecture>>, ()> {
         unsafe {
             let llil_ptr = BNGetFlowGraphLowLevelILFunction(self.handle);
-            let func_ptr = BNGetLowLevelILOwnerFunction(llil_ptr);
-            let arch_ptr = BNGetFunctionArchitecture(func_ptr);
-            let arch = CoreArchitecture::from_raw(arch_ptr);
-            BNFreeFunction(func_ptr);
             match llil_ptr.is_null() {
-                false => Ok(RegularLowLevelILFunction::ref_from_raw(arch, llil_ptr)),
+                false => {
+                    let func_ptr = BNGetLowLevelILOwnerFunction(llil_ptr);
+                    let arch_ptr = BNGetFunctionArchitecture(func_ptr);
+                    let arch = CoreArchitecture::from_raw(arch_ptr);
+                    BNFreeFunction(func_ptr);
+                    Ok(RegularLowLevelILFunction::ref_from_raw(arch, llil_ptr))
+                },
                 true => Err(()),
             }
         }