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

OnDiskMetadata

Browse Source
This commit is contained in:
Andrea Fioraldi 2021-11-29 14:28:55 +01:00
parent 6dd107c4ef
commit cf5b4dfb18
2 changed files with 20 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
libafl/src/corpus/ondisk.rs
View File

@ -1,14 +1,17 @@
//! The ondisk corpus stores unused testcases to disk. //! The ondisk corpus stores unused testcases to disk.
use alloc::vec::Vec; use alloc::vec::Vec;
use core::cell::RefCell; use core::{cell::RefCell, time::Duration};
use serde::{Deserialize, Serialize}; use serde::{Deserialize, Serialize};
use std::{fs::OpenOptions, path::PathBuf}; use std::{fs::OpenOptions, path::PathBuf};
#[cfg(feature = "std")] #[cfg(feature = "std")]
use std::{fs, fs::File, io::Write}; use std::{fs, fs::File, io::Write};
use crate::{corpus::Corpus, corpus::Testcase, inputs::Input, state::HasMetadata, Error}; use crate::{
bolts::serdeany::SerdeAnyMap, corpus::Corpus, corpus::Testcase, inputs::Input,
state::HasMetadata, Error,
};
/// Options for the the format of the on-disk metadata /// Options for the the format of the on-disk metadata
#[cfg(feature = "std")] #[cfg(feature = "std")]
@ -109,9 +112,9 @@ where
let mut tmpfile = File::create(&tmpfile_name)?; let mut tmpfile = File::create(&tmpfile_name)?;
let serialized = match self.meta_format.as_ref().unwrap() { let serialized = match self.meta_format.as_ref().unwrap() {
OnDiskMetadataFormat::Postcard => postcard::to_allocvec(testcase.metadata())?, OnDiskMetadataFormat::Postcard => postcard::to_allocvec(&ondisk_meta)?,
OnDiskMetadataFormat::Json => serde_json::to_vec(testcase.metadata())?, OnDiskMetadataFormat::Json => serde_json::to_vec(&ondisk_meta)?,
OnDiskMetadataFormat::JsonPretty => serde_json::to_vec_pretty(testcase.metadata())?, OnDiskMetadataFormat::JsonPretty => serde_json::to_vec_pretty(&ondisk_meta)?,
}; };
tmpfile.write_all(&serialized)?; tmpfile.write_all(&serialized)?;
fs::rename(&tmpfile_name, &filename)?; fs::rename(&tmpfile_name, &filename)?;

12
libafl/src/corpus/testcase.rs
View File

@ -138,6 +138,18 @@ where
self.exec_time = Some(time); self.exec_time = Some(time);
} }
/// Get the executions
#[inline]
pub fn executions(&self) -> &usize {
&self.executions
}
/// Get the executions (mut)
#[inline]
pub fn executions_mut(&mut self) -> &mut usize {
&mut self.executions
}
/// Create a new Testcase instace given an input /// Create a new Testcase instace given an input
#[inline] #[inline]
pub fn new<T>(input: T) -> Self pub fn new<T>(input: T) -> Self