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

Launcher: use libc::open instead of fs::File

Browse Source
This commit is contained in:
Andrea Fioraldi 2021-10-14 15:57:32 +02:00
parent 2e55d24f5a
commit ab7672c8d1
1 changed files with 15 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
libafl/src/bolts/launcher.rs
View File

@ -18,7 +18,7 @@ use crate::bolts::os::startable_self;
use crate::bolts::os::{dup2, fork, ForkResult}; use crate::bolts::os::{dup2, fork, ForkResult};
#[cfg(all(unix, feature = "std"))] #[cfg(all(unix, feature = "std"))]
use std::{fs::File, os::unix::io::AsRawFd}; use std::ffi::CString;
#[cfg(feature = "std")] #[cfg(feature = "std")]
use std::net::SocketAddr; use std::net::SocketAddr;
@ -98,9 +98,16 @@ where
let mut handles = vec![]; let mut handles = vec![];
println!("spawning on cores: {:?}", self.cores); println!("spawning on cores: {:?}", self.cores);
let file = self let file = self.stdout_file.map(|filename| {
.stdout_file let fname = CString::new(filename).unwrap();
.map(|filename| File::create(filename).unwrap()); unsafe {
libc::open(
fname.as_ptr(),
libc::O_WRONLY | libc::O_CREAT | libc::O_TRUNC,
0666,
)
}
});
// Spawn clients // Spawn clients
for (index, (id, bind_to)) in core_ids.iter().enumerate().take(num_cores).enumerate() { for (index, (id, bind_to)) in core_ids.iter().enumerate().take(num_cores).enumerate() {
@ -121,9 +128,10 @@ where
std::thread::sleep(std::time::Duration::from_secs((index + 1) as u64)); std::thread::sleep(std::time::Duration::from_secs((index + 1) as u64));
#[cfg(feature = "std")] #[cfg(feature = "std")]
if file.is_some() { if let Some(fd) = file {
dup2(file.as_ref().unwrap().as_raw_fd(), libc::STDOUT_FILENO)?; dup2(fd, libc::STDOUT_FILENO)?;
dup2(file.as_ref().unwrap().as_raw_fd(), libc::STDERR_FILENO)?; dup2(fd, libc::STDERR_FILENO)?;
unsafe { libc::close(fd) };
} }
// Fuzzer client. keeps retrying the connection to broker till the broker starts // Fuzzer client. keeps retrying the connection to broker till the broker starts
let (state, mgr) = RestartingMgr::<I, OT, S, SP, ST>::builder() let (state, mgr) = RestartingMgr::<I, OT, S, SP, ST>::builder()