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