From 952d16ca58429d126224662f4f5f496a0535c7ac Mon Sep 17 00:00:00 2001
From: Mason Reed <mason@vector35.com>
Date: Sat, 26 Oct 2024 20:33:45 -0400
Subject: Fix all low level il function getters leaking in rust

---
 rust/src/function.rs | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

(limited to 'rust/src/function.rs')

diff --git a/rust/src/function.rs b/rust/src/function.rs
index 0508ca4d..0df9fcf5 100644
--- a/rust/src/function.rs
+++ b/rust/src/function.rs
@@ -427,7 +427,7 @@ impl Function {
                 return Err(());
             }
 
-            Ok(llil::RegularFunction::from_raw(self.arch(), llil))
+            Ok(llil::RegularFunction::ref_from_raw(self.arch(), llil))
         }
     }
 
@@ -435,7 +435,7 @@ impl Function {
         &self,
     ) -> Option<Ref<llil::RegularFunction<CoreArchitecture>>> {
         let llil = unsafe { BNGetFunctionLowLevelILIfAvailable(self.handle) };
-        (!llil.is_null()).then(|| unsafe { llil::RegularFunction::from_raw(self.arch(), llil) })
+        (!llil.is_null()).then(|| unsafe { llil::RegularFunction::ref_from_raw(self.arch(), llil) })
     }
 
     pub fn lifted_il(&self) -> Result<Ref<llil::LiftedFunction<CoreArchitecture>>, ()> {
@@ -446,13 +446,13 @@ impl Function {
                 return Err(());
             }
 
-            Ok(llil::LiftedFunction::from_raw(self.arch(), llil))
+            Ok(llil::LiftedFunction::ref_from_raw(self.arch(), llil))
         }
     }
 
     pub fn lifted_il_if_available(&self) -> Option<Ref<llil::LiftedFunction<CoreArchitecture>>> {
         let llil = unsafe { BNGetFunctionLiftedILIfAvailable(self.handle) };
-        (!llil.is_null()).then(|| unsafe { llil::LiftedFunction::from_raw(self.arch(), llil) })
+        (!llil.is_null()).then(|| unsafe { llil::LiftedFunction::ref_from_raw(self.arch(), llil) })
     }
 
     pub fn return_type(&self) -> Conf<Ref<Type>> {
-- 
cgit v1.3.1