hitcounts observer in libpng example

This commit is contained in:
Andrea Fioraldi 2021-03-01 10:34:59 +01:00
parent 405bfc5294
commit bd800b63f5

View File

@ -16,7 +16,7 @@ use libafl::{
inputs::Input, inputs::Input,
mutators::scheduled::HavocBytesMutator, mutators::scheduled::HavocBytesMutator,
mutators::token_mutations::TokensMetadata, mutators::token_mutations::TokensMetadata,
observers::StdMapObserver, observers::{HitcountsMapObserver, StdMapObserver},
stages::mutational::StdMutationalStage, stages::mutational::StdMutationalStage,
state::{HasCorpus, HasMetadata, State}, state::{HasCorpus, HasMetadata, State},
stats::SimpleStats, stats::SimpleStats,
@ -79,10 +79,11 @@ fn fuzz(corpus_dirs: Vec<PathBuf>, objective_dir: PathBuf, broker_port: u16) ->
.expect("Failed to setup the restarter".into()); .expect("Failed to setup the restarter".into());
// Create an observation channel using the coverage map // Create an observation channel using the coverage map
let edges_observer = let edges_observer = HitcountsMapObserver::new(StdMapObserver::new_from_ptr(
StdMapObserver::new_from_ptr("edges", unsafe { __lafl_edges_map }, unsafe { "edges",
__lafl_max_edges_size as usize unsafe { __lafl_edges_map },
}); unsafe { __lafl_max_edges_size as usize },
));
// If not restarting, create a State from scratch // If not restarting, create a State from scratch
let mut state = state.unwrap_or_else(|| { let mut state = state.unwrap_or_else(|| {