summaryrefslogtreecommitdiff
path: root/rust/src/types.rs
diff options
context:
space:
mode:
authorRyan Snyder <ryan@vector35.com>2021-01-21 18:27:48 +0000
committerKyleMiles <krm504@nyu.edu>2021-01-21 19:06:55 +0000
commitd3140edec185f47235b9e4642bdd56d6c585a341 (patch)
treea61859c29e4e3539daea2b761bb1439d942beaf4 /rust/src/types.rs
parentc0ddbf0c76d3f1bb7a2b2024f749afc8b9482575 (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.rs109
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)
+ }
+}
+