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/platform.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/platform.rs')
| -rw-r--r-- | rust/src/platform.rs | 227 |
1 files changed, 227 insertions, 0 deletions
diff --git a/rust/src/platform.rs b/rust/src/platform.rs new file mode 100644 index 00000000..3343bece --- /dev/null +++ b/rust/src/platform.rs @@ -0,0 +1,227 @@ +use std::borrow::Borrow; + +use binaryninjacore_sys::*; + +use crate::architecture::{Architecture, CoreArchitecture}; +use crate::callingconvention::CallingConvention; +use crate::types::QualifiedNameAndType; +use crate::rc::*; +use crate::string::*; + +#[derive(PartialEq, Eq, Hash)] +pub struct Platform { + pub(crate) handle: *mut BNPlatform, +} + +unsafe impl Send for Platform {} +unsafe impl Sync for Platform {} + +macro_rules! cc_func { + ($get_name:ident, $get_api:ident, $set_name:ident, $set_api:ident) => { + pub fn $get_name(&self) -> Option<Ref<CallingConvention<CoreArchitecture>>> { + let arch = self.arch(); + + unsafe { + let cc = $get_api(self.handle); + + if cc.is_null() { + None + } else { + Some(Ref::new(CallingConvention::from_raw(cc, arch))) + } + } + } + + pub fn $set_name<A: Architecture>(&self, cc: &CallingConvention<A>) { + let arch = self.arch(); + + assert!(cc.arch_handle.borrow().as_ref().0 == arch.0, "use of calling convention with non-matching Platform architecture!"); + + unsafe { + $set_api(self.handle, cc.handle); + } + } + } +} + +impl Platform { + pub(crate) unsafe fn from_raw(handle: *mut BNPlatform) -> Self { + debug_assert!(!handle.is_null()); + + Self { handle } + } + + pub fn by_name<S: BnStrCompatible>(name: S) -> Option<Ref<Self>> { + let raw_name = name.as_bytes_with_nul(); + unsafe { + let res = BNGetPlatformByName(raw_name.as_ref().as_ptr() as *mut _); + + if res.is_null() { + None + } else { + Some(Ref::new(Self { handle: res })) + } + } + } + + pub fn list_all() -> Array<Platform> { + unsafe { + let mut count = 0; + let handles = BNGetPlatformList(&mut count); + + Array::new(handles, count, ()) + } + } + + pub fn list_by_arch(arch: &CoreArchitecture) -> Array<Platform> { + unsafe { + let mut count = 0; + let handles = BNGetPlatformListByArchitecture(arch.0, &mut count); + + Array::new(handles, count, ()) + } + } + + pub fn list_by_os<S: BnStrCompatible>(name: S) -> Array<Platform> { + let raw_name = name.as_bytes_with_nul(); + + unsafe { + let mut count = 0; + let handles = BNGetPlatformListByOS(raw_name.as_ref().as_ptr() as *mut _, &mut count); + + Array::new(handles, count, ()) + } + } + + pub fn list_by_os_and_arch<S: BnStrCompatible>(name: S, arch: &CoreArchitecture) -> Array<Platform> { + let raw_name = name.as_bytes_with_nul(); + + unsafe { + let mut count = 0; + let handles = BNGetPlatformListByOSAndArchitecture(raw_name.as_ref().as_ptr() as *mut _, + arch.0, &mut count); + + Array::new(handles, count, ()) + } + } + + pub fn list_available_os() -> Array<BnString> { + unsafe { + let mut count = 0; + let list = BNGetPlatformOSList(&mut count); + + Array::new(list, count, ()) + } + } + + pub fn new<A: Architecture, S: BnStrCompatible>(arch: &A, name: S) -> Ref<Self> { + let name = name.as_bytes_with_nul(); + unsafe { + let handle = BNCreatePlatform(arch.as_ref().0, name.as_ref().as_ptr() as *mut _); + + assert!(!handle.is_null()); + + Ref::new(Self { handle }) + } + } + + pub fn name(&self) -> BnString { + unsafe { + let raw_name = BNGetPlatformName(self.handle); + BnString::from_raw(raw_name) + } + } + + pub fn arch(&self) -> CoreArchitecture { + unsafe { + CoreArchitecture::from_raw(BNGetPlatformArchitecture(self.handle)) + } + } + + pub fn register_os<S: BnStrCompatible>(&self, os: S) { + let os = os.as_bytes_with_nul(); + + unsafe { + BNRegisterPlatform(os.as_ref().as_ptr() as *mut _, self.handle); + } + } + + cc_func!(get_default_calling_convention, BNGetPlatformDefaultCallingConvention, + set_default_calling_convention, BNRegisterPlatformDefaultCallingConvention); + + cc_func!(get_cdecl_calling_convention, BNGetPlatformCdeclCallingConvention, + set_cdecl_calling_convention, BNRegisterPlatformCdeclCallingConvention); + + cc_func!(get_stdcall_calling_convention, BNGetPlatformStdcallCallingConvention, + set_stdcall_calling_convention, BNRegisterPlatformStdcallCallingConvention); + + cc_func!(get_fastcall_calling_convention, BNGetPlatformFastcallCallingConvention, + set_fastcall_calling_convention, BNRegisterPlatformFastcallCallingConvention); + + cc_func!(get_syscall_convention, BNGetPlatformSystemCallConvention, + set_syscall_convention, BNSetPlatformSystemCallConvention); + + pub fn types(&self) -> Array<QualifiedNameAndType> { + unsafe { + let mut count = 0; + let handles = BNGetPlatformTypes(self.handle, &mut count); + + Array::new(handles, count, ()) + } + } + + pub fn variables(&self) -> Array<QualifiedNameAndType> { + unsafe { + let mut count = 0; + let handles = BNGetPlatformVariables(self.handle, &mut count); + + Array::new(handles, count, ()) + } + } + + pub fn functions(&self) -> Array<QualifiedNameAndType> { + unsafe { + let mut count = 0; + let handles = BNGetPlatformFunctions(self.handle, &mut count); + + Array::new(handles, count, ()) + } + } +} + +impl ToOwned for Platform { + type Owned = Ref<Self>; + + fn to_owned(&self) -> Self::Owned { + unsafe { RefCountable::inc_ref(self) } + } +} + +unsafe impl RefCountable for Platform { + unsafe fn inc_ref(handle: &Self) -> Ref<Self> { + Ref::new(Self { + handle: BNNewPlatformReference(handle.handle), + }) + } + + unsafe fn dec_ref(handle: &Self) { + BNFreePlatform(handle.handle); + } +} + +unsafe impl CoreOwnedArrayProvider for Platform { + type Raw = *mut BNPlatform; + type Context = (); + + unsafe fn free(raw: *mut *mut BNPlatform, count: usize, _context: &()) { + BNFreePlatformList(raw, count); + } +} + +unsafe impl<'a> CoreOwnedArrayWrapper<'a> for Platform { + type Wrapped = Guard<'a, Platform>; + + unsafe fn wrap_raw(raw: &'a *mut BNPlatform, context: &'a ()) -> Guard<'a, Platform> { + Guard::new(Platform::from_raw(*raw), context) + } +} |
