Fix memory leak in functions calling BNCreateUserFunction

1 files changed, 8 insertions, 2 deletions

diff --git a/rust/src/binaryview.rs b/rust/src/binaryview.rs
index 31a12105..d6916e2c 100644
--- a/rust/src/binaryview.rs
+++ b/rust/src/binaryview.rs
@@ -853,9 +853,15 @@ pub trait BinaryViewExt: BinaryViewBase {
         }
     }
 
-    fn create_user_function(&self, plat: &Platform, addr: u64) {
+    fn create_user_function(&self, plat: &Platform, addr: u64) -> Result<Ref<Function>> {
         unsafe {
-            BNCreateUserFunction(self.as_ref().handle, plat.handle, addr);
+            let func = BNCreateUserFunction(self.as_ref().handle, plat.handle, addr);
+
+            if func.is_null() {
+                return Err(());
+            }
+
+            Ok(Function::from_raw(func))
         }
     }