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

Fix edge module generators (#2702)

Browse Source 
* fix generators
This commit is contained in:
Romain Malmain 2024-11-18 19:48:31 +01:00 committed by GitHub
parent b324e88631
commit b3d3c38b29
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 12 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
libafl_qemu/src/modules/edges/helpers.rs
View File

@ -191,20 +191,18 @@ mod generators {
} }
} }
let mask: usize = get_mask::<IS_CONST_MAP, MAP_SIZE>(); let mask = get_mask::<IS_CONST_MAP, MAP_SIZE>() as u64;
let id = hash_me(src as u64) ^ hash_me(dest as u64); #[allow(clippy::unnecessary_cast)]
let id = (hash_me(src as u64) ^ hash_me(dest as u64)) & mask;
unsafe { if !IS_CONST_MAP {
let nxt = (id as usize + 1) & mask; unsafe {
*LIBAFL_QEMU_EDGES_MAP_SIZE_PTR =
if !IS_CONST_MAP { max(*LIBAFL_QEMU_EDGES_MAP_SIZE_PTR, id as usize);
*LIBAFL_QEMU_EDGES_MAP_SIZE_PTR = nxt;
} }
} }
// GuestAddress is u32 for 32 bit guests
#[allow(clippy::unnecessary_cast)]
Some(id) Some(id)
} else { } else {
None None
@ -247,15 +245,13 @@ mod generators {
} }
} }
let mask: usize = get_mask::<IS_CONST_MAP, MAP_SIZE>(); let mask = get_mask::<IS_CONST_MAP, MAP_SIZE>() as u64;
let id = hash_me(pc as u64); let id = hash_me(pc as u64) & mask;
unsafe { if !IS_CONST_MAP {
let nxt = (id as usize + 1) & mask; unsafe {
*LIBAFL_QEMU_EDGES_MAP_SIZE_PTR = max(*LIBAFL_QEMU_EDGES_MAP_SIZE_PTR, id as usize);
if !IS_CONST_MAP {
*LIBAFL_QEMU_EDGES_MAP_SIZE_PTR = nxt;
} }
} }