diff options
Diffstat (limited to 'rust/src/binaryview.rs')
| -rw-r--r-- | rust/src/binaryview.rs | 10 |
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)) } } |
