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/fileaccessor.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/fileaccessor.rs')
| -rw-r--r-- | rust/src/fileaccessor.rs | 78 |
1 files changed, 78 insertions, 0 deletions
diff --git a/rust/src/fileaccessor.rs b/rust/src/fileaccessor.rs new file mode 100644 index 00000000..e3320be2 --- /dev/null +++ b/rust/src/fileaccessor.rs @@ -0,0 +1,78 @@ +use binaryninjacore_sys::BNFileAccessor; +use std::io::{Read, Write, Seek, SeekFrom}; +use std::marker::PhantomData; +use std::slice; + +pub struct FileAccessor<'a> +{ + pub(crate) api_object: BNFileAccessor, + _ref: PhantomData<&'a mut ()>, +} + +impl<'a> FileAccessor<'a> +{ + pub fn new<F>(f: &'a mut F) -> Self + where + F: 'a + Read + Write + Seek + Sized + { + use std::os::raw::c_void; + + extern "C" fn cb_get_length<F>(ctxt: *mut c_void) -> u64 + where + F: Read + Write + Seek + Sized + { + let f = unsafe { &mut *(ctxt as *mut F) }; + + match f.seek(SeekFrom::End(0)) { + Ok(len) => len, + Err(_) => 0, + } + } + + extern "C" fn cb_read<F>(ctxt: *mut c_void, dest: *mut c_void, offset: u64, len: usize) -> usize + where + F: Read + Write + Seek + Sized + { + let f = unsafe { &mut *(ctxt as *mut F) }; + let dest = unsafe { slice::from_raw_parts_mut(dest as *mut u8, len) }; + + if !f.seek(SeekFrom::Start(offset)).is_ok() { + debug!("Failed to seek to offset {:x}", offset); + return 0; + } + + match f.read(dest) { + Ok(len) => len, + Err(_) => 0, + } + } + + extern "C" fn cb_write<F>(ctxt: *mut c_void, offset: u64, src: *const c_void, len: usize) -> usize + where + F: Read + Write + Seek + Sized + { + let f = unsafe { &mut *(ctxt as *mut F) }; + let src = unsafe { slice::from_raw_parts(src as *const u8, len) }; + + if !f.seek(SeekFrom::Start(offset)).is_ok() { + return 0; + } + + match f.write(src) { + Ok(len) => len, + Err(_) => 0, + } + } + + Self { + api_object: BNFileAccessor { + context: f as *mut F as *mut _, + getLength: Some(cb_get_length::<F>), + read: Some(cb_read::<F>), + write: Some(cb_write::<F>), + }, + _ref: PhantomData, + } + } +} + |
