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

Don't do generalization on larger inptus (#2603)

Browse Source
This commit is contained in:
Dongjia "toka" Zhang 2024-10-10 16:13:08 +02:00 committed by GitHub
parent a69cd98432
commit 19ccb0807c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 12 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
libafl/src/stages/generalization.rs
View File

@ -111,6 +111,11 @@ where
let input = entry.input_mut().as_mut().unwrap();
let payload: Vec<_> = input.bytes().iter().map(|&x| Some(x)).collect();
if payload.len() > MAX_GENERALIZED_LEN {
return Ok(());
}
let original = input.clone();
let meta = entry.metadata_map().get::<MapNoveltiesMetadata>().ok_or_else(|| {
Error::key_not_found(format!(
@ -311,17 +316,15 @@ where
b'"',
)?;
if payload.len() <= MAX_GENERALIZED_LEN {
// Save the modified input in the corpus
{
let meta = GeneralizedInputMetadata::generalized_from_options(&payload);
// Save the modified input in the corpus
{
let meta = GeneralizedInputMetadata::generalized_from_options(&payload);
assert!(meta.generalized().first() == Some(&GeneralizedItem::Gap));
assert!(meta.generalized().last() == Some(&GeneralizedItem::Gap));
assert!(meta.generalized().first() == Some(&GeneralizedItem::Gap));
assert!(meta.generalized().last() == Some(&GeneralizedItem::Gap));
let mut entry = state.corpus().get(corpus_id)?.borrow_mut();
entry.metadata_map_mut().insert(meta);
}
let mut entry = state.corpus().get(corpus_id)?.borrow_mut();
entry.metadata_map_mut().insert(meta);
}
Ok(())