diff options
| author | Ryan Snyder <ryan@vector35.com> | 2021-01-21 18:27:48 +0000 |
|---|---|---|
| committer | KyleMiles <krm504@nyu.edu> | 2021-01-21 19:06:55 +0000 |
| commit | d3140edec185f47235b9e4642bdd56d6c585a341 (patch) | |
| tree | a61859c29e4e3539daea2b761bb1439d942beaf4 /rust/src/types.rs | |
| parent | c0ddbf0c76d3f1bb7a2b2024f749afc8b9482575 (diff) | |
This is a combination of 23 commits, the work of Ryan Snyder:
Initial fresh repo
Add support for recent calling convention API updates and folds the binaryninjacore-sys crate directly into this one.
Add support for auto function analysis suppression
Finish moving binaryninjacore-sys back into this crate
Update for Symbol/Segment core API changes
Update for Symbol API cleanup
api: advance submodule reference, support Token changes
arch/lifting: support for flags in custom architectures
arch/lifting: support default flag write behaviors, handle more ops
build: enable headless binary support on MacOS via evil hack
bv: add BinaryView wrapper support, remove wrong comment
api: update to latest binja dev branch support
deps: bump dep versions
rust: bump to 2018 edition
api: bump to avoid cargo submodule brokenness
build: improve binaryninja path detection; enable linux linkhack
bv: stub for bv load settings
arch: fix flag related crash, minor llil update
api: update for recent changes
macos: disable linkhack briefly
Diffstat (limited to 'rust/src/types.rs')
| -rw-r--r-- | rust/src/types.rs | 109 |
1 files changed, 109 insertions, 0 deletions
diff --git a/rust/src/types.rs b/rust/src/types.rs new file mode 100644 index 00000000..80046f87 --- /dev/null +++ b/rust/src/types.rs @@ -0,0 +1,109 @@ +use std::mem; +use std::slice; +use binaryninjacore_sys::*; + +use crate::rc::*; + +#[derive(PartialEq, Eq, Hash)] +pub struct Type { + pub(crate) handle: *mut BNType, +} + +unsafe impl Send for Type {} +unsafe impl Sync for Type {} + +impl Type { + pub(crate) unsafe fn from_raw(handle: *mut BNType) -> Self { + debug_assert!(!handle.is_null()); + + Self { handle } + } +} + +impl ToOwned for Type { + type Owned = Ref<Self>; + + fn to_owned(&self) -> Self::Owned { + unsafe { RefCountable::inc_ref(self) } + } +} + +unsafe impl RefCountable for Type { + unsafe fn inc_ref(handle: &Self) -> Ref<Self> { + Ref::new(Self { + handle: BNNewTypeReference(handle.handle), + }) + } + + unsafe fn dec_ref(handle: &Self) { + BNFreeType(handle.handle); + } +} + +#[repr(C)] +pub struct QualifiedName { + object: BNQualifiedName, +} + +impl QualifiedName { + pub fn string(&self) -> String { + use std::ffi::CStr; + + unsafe { + slice::from_raw_parts(self.object.name, self.object.nameCount) + .iter() + .map(|c| CStr::from_ptr(*c).to_string_lossy()) + .collect::<Vec<_>>() + .join("::") + } + } +} + +impl Drop for QualifiedName { + fn drop(&mut self) { + unsafe { BNFreeQualifiedName(&mut self.object); } + } +} + +#[repr(C)] +pub struct QualifiedNameAndType { + object: BNQualifiedNameAndType, +} + +impl QualifiedNameAndType { + pub fn name(&self) -> &QualifiedName { + unsafe { + mem::transmute(&self.object.name) + } + } + + pub fn type_object(&self) -> Guard<Type> { + unsafe { + Guard::new(Type::from_raw(self.object.type_), self) + } + } +} + +impl Drop for QualifiedNameAndType { + fn drop(&mut self) { + unsafe { BNFreeQualifiedNameAndType(&mut self.object); } + } +} + +unsafe impl CoreOwnedArrayProvider for QualifiedNameAndType { + type Raw = BNQualifiedNameAndType; + type Context = (); + + unsafe fn free(raw: *mut Self::Raw, count: usize, _context: &Self::Context) { + BNFreeTypeList(raw, count); + } +} + +unsafe impl<'a> CoreOwnedArrayWrapper<'a> for QualifiedNameAndType { + type Wrapped = &'a QualifiedNameAndType; + + unsafe fn wrap_raw(raw: &'a Self::Raw, _context: &'a Self::Context) -> Self::Wrapped { + mem::transmute(raw) + } +} + |
