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

fixes for win32

Browse Source
This commit is contained in:
Andrea Fioraldi 2021-03-18 16:32:16 +01:00
parent 6f1058f8d1
commit cc0da780ef
2 changed files with 6 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
libafl/build.rs
View File

@ -1,7 +1,7 @@
fn main() {
#[cfg(target_os = "windows")]
windows::build!(
windows::win32::system_services::{HANDLE},
windows::win32::system_services::{HANDLE, BOOL},
windows::win32::windows_programming::CloseHandle,
// API needed for the shared memory
windows::win32::system_services::{CreateFileMappingA, OpenFileMappingA, MapViewOfFile, UnmapViewOfFile},

12
libafl/src/bolts/shmem.rs
View File

@ -446,7 +446,7 @@ pub mod shmem {
use super::ShMem;
use crate::{
bolts::bindings::{
windows::win32::system_services::HANDLE,
windows::win32::system_services::{HANDLE, BOOL},
windows::win32::system_services::{
CreateFileMappingA, MapViewOfFile, OpenFileMappingA, UnmapViewOfFile,
},
@ -509,7 +509,7 @@ pub mod shmem {
impl Win32ShMem {
pub fn from_str(map_str_bytes: &[u8; 20], map_size: usize) -> Result<Self, Error> {
unsafe {
let handle = OpenFileMappingA(FILE_MAP_ALL_ACCESS, 0, map_str_bytes as *const u8 as *const i8);
let handle = OpenFileMappingA(FILE_MAP_ALL_ACCESS, BOOL(0), map_str_bytes as *const u8 as *const i8);
if handle == HANDLE(0) {
return Err(Error::Unknown(format!(
"Cannot open shared memory {}",
@ -562,14 +562,12 @@ pub mod shmem {
String::from_utf8_lossy(map_str_bytes)
)));
}
let mut ret = Self {
shm_str: [0; 20],
Ok(Self {
shm_str: map_str_bytes[0..20],
handle: handle,
map: map,
map_size: map_size,
};
ret.shm_str.clone_from_slice(map_str_bytes[0..20]);
Ok(ret)
})
}
}
}