[Rust] Avoid leaking a reference within {Medium,High}LevelILFunction::ssa_form

Return a Ref<_> so that the underlying core object will have its
reference count decremented when the caller drops the object.

1 files changed, 2 insertions, 5 deletions

diff --git a/rust/src/high_level_il/function.rs b/rust/src/high_level_il/function.rs
index aad16e14..0813ad04 100644
--- a/rust/src/high_level_il/function.rs
+++ b/rust/src/high_level_il/function.rs
@@ -69,13 +69,10 @@ impl HighLevelILFunction {
         unsafe { BNGetHighLevelILExprCount(self.handle) }
     }
 
-    pub fn ssa_form(&self) -> HighLevelILFunction {
+    pub fn ssa_form(&self) -> Ref<HighLevelILFunction> {
         let ssa = unsafe { BNGetHighLevelILSSAForm(self.handle) };
         assert!(!ssa.is_null());
-        HighLevelILFunction {
-            handle: ssa,
-            full_ast: self.full_ast,
-        }
+        unsafe { HighLevelILFunction::ref_from_raw(ssa, self.full_ast) }
     }
 
     pub fn function(&self) -> Ref<Function> {