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

Make unique_std_input_file more unique (#3247)

Browse Source
This commit is contained in:
lazymio 2025-05-19 05:17:29 +08:00 committed by GitHub
parent 542b7f75b7
commit 73cc2f5b9c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
libafl_bolts/src/fs.rs
View File

@ -1,7 +1,10 @@
//! `LibAFL` functionality for filesystem interaction
use core::sync::atomic::AtomicU64;
use core::sync::atomic::Ordering;
#[cfg(unix)]
use std::os::unix::prelude::{AsRawFd, RawFd};
use std::sync::OnceLock;
use alloc::string::String;
use alloc::sync::Arc;
@ -23,7 +26,11 @@ pub const INPUTFILE_STD: &str = ".cur_input";
/// Derives a filename from [`INPUTFILE_STD`] that may be used to deliver testcases to the target.
/// It ensures the filename is unique to the fuzzer process.
pub fn get_unique_std_input_file() -> String {
format!("{}_{}", INPUTFILE_STD, std::process::id())
static STD_COUNT: OnceLock<AtomicU64> = OnceLock::new();
let next = STD_COUNT
.get_or_init(|| AtomicU64::new(0))
.fetch_add(1, Ordering::SeqCst);
format!("{}_{}_{}", INPUTFILE_STD, std::process::id(), next)
}
/// Write a file atomically