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

Fix WeightedScheduler reporting incorrect queue cylces (#2281)

Browse Source 
* fix WeightedScheduler reporting incorrect queue cylces

* check if runs_in_cycle equals or greater instead of just equals

---------

Co-authored-by: Dongjia "toka" Zhang <tokazerkje@outlook.com>
This commit is contained in:
Aarnav 2024-06-06 19:50:29 +02:00 committed by GitHub
parent af12b4ba24
commit a455ad27a9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
libafl/src/schedulers/weighted.rs
View File

@ -324,13 +324,13 @@ where
let wsmeta = state.metadata_mut::<WeightedScheduleMetadata>()?;
let current_cycles = wsmeta.runs_in_current_cycle();
let runs_in_current_cycle = wsmeta.runs_in_current_cycle();
// TODO deal with corpus_counts decreasing due to removals
if current_cycles >= corpus_counts {
if runs_in_current_cycle >= corpus_counts {
wsmeta.set_runs_current_cycle(0);
} else {
wsmeta.set_runs_current_cycle(current_cycles + 1);
wsmeta.set_runs_current_cycle(runs_in_current_cycle + 1);
}
let idx = if probability < *wsmeta.alias_probability().get(&s).unwrap() {
@ -340,7 +340,7 @@ where
};
// Update depth
if current_cycles > corpus_counts {
if runs_in_current_cycle >= corpus_counts {
let psmeta = state.metadata_mut::<SchedulerMetadata>()?;
psmeta.set_queue_cycles(psmeta.queue_cycles() + 1);
}