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

Align addresses before calling qemu.unmap in Snapshot module (#2978)

Browse Source 
* alignment

* a
This commit is contained in:
Dongjia "toka" Zhang 2025-02-13 14:17:48 +01:00 committed by GitHub
parent cb471a9282
commit b067435862
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 13 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
libafl_qemu/src/modules/usermode/snapshot.rs
View File

@ -415,12 +415,20 @@ impl SnapshotModule {
.unwrap();
} else if new_brk > self.brk {
// The heap has grown. so we want to drop those
let drop_sz = (new_brk - self.brk) as usize;
// we want to align the addresses before calling unmap
// although it is very unlikely that the brk has an unaligned value
let new_page_boundary = (new_brk + ((SNAPSHOT_PAGE_MASK - 1) as GuestAddr))
& (!(SNAPSHOT_PAGE_SIZE - 1) as GuestAddr);
let old_page_boundary = (self.brk + ((SNAPSHOT_PAGE_MASK - 1) as GuestAddr))
& (!(SNAPSHOT_PAGE_SIZE - 1) as GuestAddr);
if new_page_boundary != old_page_boundary {
let unmap_sz = (new_page_boundary - old_page_boundary) as usize;
// if self.brk is not aligned this call will return an error
// and it will page align this drop_sz too
// and it will page align this unmap_sz too (but it is already aligned for us)
// look at target_munmap in qemu-libafl-bridge
qemu.unmap(self.brk, drop_sz).unwrap();
qemu.unmap(self.brk, unmap_sz).unwrap();
}
}
for acc in &mut self.accesses {