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

Call post_exec_all() in calibrate.rs (#603)

Browse Source 
* fix

* fix

* fix

* post_exec

* fix
This commit is contained in:
Dongjia Zhang 2022-05-12 18:14:21 +09:00 committed by GitHub
parent 9e382c4177
commit 62484b12f4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 13 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
libafl/src/stages/calibrate.rs
View File

@ -74,9 +74,11 @@ where
// Run once to get the initial calibration map // Run once to get the initial calibration map
executor.observers_mut().pre_exec_all(state, &input)?; executor.observers_mut().pre_exec_all(state, &input)?;
let mut start = current_time(); let mut start = current_time();
let mut total_time = if executor.run_target(fuzzer, state, mgr, &input)? == ExitKind::Ok { let exit_kind = executor.run_target(fuzzer, state, mgr, &input)?;
let mut total_time = if exit_kind == ExitKind::Ok {
current_time() - start current_time() - start
} else { } else {
mgr.log( mgr.log(
@ -88,6 +90,10 @@ where
Duration::from_secs(1) Duration::from_secs(1)
}; };
executor
.observers_mut()
.post_exec_all(state, &input, &exit_kind)?;
let map_first = &executor let map_first = &executor
.observers() .observers()
.match_name::<O>(&self.map_observer_name) .match_name::<O>(&self.map_observer_name)
@ -111,7 +117,8 @@ where
executor.observers_mut().pre_exec_all(state, &input)?; executor.observers_mut().pre_exec_all(state, &input)?;
start = current_time(); start = current_time();
if executor.run_target(fuzzer, state, mgr, &input)? != ExitKind::Ok { let exit_kind = executor.run_target(fuzzer, state, mgr, &input)?;
if exit_kind != ExitKind::Ok {
if !has_errors { if !has_errors {
mgr.log( mgr.log(
state, state,
@ -129,6 +136,10 @@ where
total_time += current_time() - start; total_time += current_time() - start;
executor
.observers_mut()
.post_exec_all(state, &input, &exit_kind)?;
let map = &executor let map = &executor
.observers() .observers()
.match_name::<O>(&self.map_observer_name) .match_name::<O>(&self.map_observer_name)