impl Ref for ScriptingInstance

Author: rbran

rust/src/scriptingprovider.rs

@@ -6,13 +6,13 @@ use binaryninjacore_sys::*;
 use crate::basic_block::BasicBlock;
 use crate::binary_view::BinaryView;
 use crate::function::{Function, NativeBlock};
-use crate::rc::{Array, CoreArrayProvider, CoreArrayProviderInner};
+use crate::rc::{Array, CoreArrayProvider, CoreArrayProviderInner, Ref, RefCountable};
 use crate::string::{BnStrCompatible, BnString};
 
 pub type ScriptingProviderExecuteResult = BNScriptingProviderExecuteResult;
 pub type ScriptingProviderInputReadyState = BNScriptingProviderInputReadyState;
 
-#[derive(Clone, Copy)]
+#[derive(Clone, Copy, Eq, PartialEq, Hash)]
 #[repr(transparent)]
 pub struct ScriptingProvider {
     handle: ptr::NonNull<BNScriptingProvider>,
@@ -143,16 +143,22 @@ pub struct ScriptingInstance {
     handle: ptr::NonNull<BNScriptingInstance>,
 }
 
-impl Clone for ScriptingInstance {
-    fn clone(&self) -> Self {
-        let result = unsafe { BNNewScriptingInstanceReference(self.as_raw()) };
-        unsafe { Self::from_raw(ptr::NonNull::new(result).unwrap()) }
+impl ToOwned for ScriptingInstance {
+    type Owned = Ref<Self>;
+
+    fn to_owned(&self) -> Self::Owned {
+        unsafe { Self::inc_ref(self) }
     }
 }
 
-impl Drop for ScriptingInstance {
-    fn drop(&mut self) {
-        unsafe { BNFreeScriptingInstance(self.as_raw()) }
+unsafe impl RefCountable for ScriptingInstance {
+    unsafe fn inc_ref(handle: &Self) -> Ref<Self> {
+        let result = unsafe { BNNewScriptingInstanceReference(handle.as_raw()) };
+        unsafe { Self::ref_from_raw(ptr::NonNull::new(result).unwrap()) }
+    }
+
+    unsafe fn dec_ref(handle: &Self) {
+        unsafe { BNFreeScriptingInstance(handle.as_raw()) }
     }
 }
 
@@ -161,6 +167,10 @@ impl ScriptingInstance {
         Self { handle }
     }
 
+    pub(crate) unsafe fn ref_from_raw(handle: ptr::NonNull<BNScriptingInstance>) -> Ref<Self> {
+        Ref::new(Self { handle })
+    }
+
     pub(crate) unsafe fn into_raw(self) -> *mut BNScriptingInstance {
         mem::ManuallyDrop::new(self).handle.as_ptr()
     }