SYS-OSS/FRET-LibAFL
4
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

added mutable map

Browse Source
This commit is contained in:
Dominik Maier 2020-10-29 11:38:22 +01:00
parent a42b787eb5
commit 9a0352d820
1 changed files with 11 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
src/observers/mod.rs
View File

@ -1,5 +1,5 @@
use crate::AflError; use crate::AflError;
use std::slice::from_raw_parts; use std::slice::from_raw_parts_mut;
pub trait Observer { pub trait Observer {
fn flush(&mut self) -> Result<(), AflError> { fn flush(&mut self) -> Result<(), AflError> {
@ -19,11 +19,13 @@ pub trait MapObserver<MapT>: Observer {
fn get_map(&self) -> &[MapT]; fn get_map(&self) -> &[MapT];
//fn get_map_mut(&mut self) -> &mut Vec<MapT>; //fn get_map_mut(&mut self) -> &mut Vec<MapT>;
fn get_map_mut(&mut self) -> &mut [u8];
} }
pub struct U8MapObserver { pub struct U8MapObserver {
map: &'static [u8], map: &'static mut [u8],
} }
@ -38,18 +40,20 @@ impl Observer for U8MapObserver {
impl MapObserver<u8> for U8MapObserver { impl MapObserver<u8> for U8MapObserver {
// TODO: Rust
fn get_map(&self) -> &[u8] { fn get_map(&self) -> &[u8] {
return self.map; self.map
}
fn get_map_mut(&mut self) -> &mut [u8] {
self.map
} }
//fn get_map_mut(&mut self) -> &mut Vec<MapT>;
} }
impl U8MapObserver { impl U8MapObserver {
pub fn new(map_ptr: *const u8, len: usize) -> Self { pub fn new(map_ptr: *mut u8, len: usize) -> Self {
unsafe { unsafe {
U8MapObserver{map: from_raw_parts(map_ptr, len)} U8MapObserver{map: from_raw_parts_mut(map_ptr, len)}
} }
} }
} }