From d3140edec185f47235b9e4642bdd56d6c585a341 Mon Sep 17 00:00:00 2001 From: Ryan Snyder Date: Thu, 21 Jan 2021 18:27:48 +0000 Subject: 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 --- rust/src/fileaccessor.rs | 78 ++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 78 insertions(+) create mode 100644 rust/src/fileaccessor.rs (limited to 'rust/src/fileaccessor.rs') 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: &'a mut F) -> Self + where + F: 'a + Read + Write + Seek + Sized + { + use std::os::raw::c_void; + + extern "C" fn cb_get_length(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(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(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::), + read: Some(cb_read::), + write: Some(cb_write::), + }, + _ref: PhantomData, + } + } +} + -- cgit v1.3.1