ExternalVendorCode/AFLplusplus
1
0
Fork 0
mirror of https://github.com/AFLplusplus/AFLplusplus.git synced 2025-06-21 13:51:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

weight power

Browse Source
This commit is contained in:
vanhauser-thc
2024-07-19 13:52:22 +02:00
parent bbcb3dd53e
commit ae0d8b8d90
3 changed files with 25 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
include/afl-fuzz.h
View File

@ -338,15 +338,16 @@ enum {
enum { enum {
/* 00 */ EXPLORE, /* AFL default, Exploration-based constant schedule */ /* 00 */ EXPLORE, /* AFL default, exploration-based constant schedule */
/* 01 */ MMOPT, /* Modified MOPT schedule */ /* 01 */ EXPLOIT, /* AFL's exploitation-based const. */
/* 02 */ EXPLOIT, /* AFL's exploitation-based const. */ /* 02 */ WEIGHT, /* Based on seed weighting algorithm */
/* 03 */ FAST, /* Exponential schedule */ /* 03 */ MMOPT, /* Modified MOPT schedule */
/* 04 */ COE, /* Cut-Off Exponential schedule */ /* 04 */ FAST, /* Exponential schedule */
/* 05 */ LIN, /* Linear schedule */ /* 05 */ COE, /* Cut-Off Exponential schedule */
/* 06 */ QUAD, /* Quadratic schedule */ /* 06 */ LIN, /* Linear schedule */
/* 07 */ RARE, /* Rare edges */ /* 07 */ QUAD, /* Quadratic schedule */
/* 08 */ SEEK, /* EXPLORE that ignores timings */ /* 08 */ RARE, /* Rare edges */
/* 09 */ SEEK, /* EXPLORE that ignores timings */
POWER_SCHEDULES_NUM POWER_SCHEDULES_NUM

11
src/afl-fuzz-queue.c
View File

@ -1020,6 +1020,17 @@ void cull_queue(afl_state_t *afl) {
u32 calculate_score(afl_state_t *afl, struct queue_entry *q) { u32 calculate_score(afl_state_t *afl, struct queue_entry *q) {
if (likely(afl->schedule == WEIGHT)) {
u32 val = 100;
if (unlikely(q->favored)) { val = val << 1; }
if (unlikely(!q->was_fuzzed)) { val = val << 1; }
return val * q->weight;
}
u32 cal_cycles = afl->total_cal_cycles; u32 cal_cycles = afl->total_cal_cycles;
u32 bitmap_entries = afl->total_bitmap_entries; u32 bitmap_entries = afl->total_bitmap_entries;

4
src/afl-fuzz.c
View File

@ -745,6 +745,10 @@ int main(int argc, char **argv_orig, char **envp) {
afl->schedule = EXPLOIT; afl->schedule = EXPLOIT;
} else if (!stricmp(optarg, "weight")) {
afl->schedule = WEIGHT;
} else if (!stricmp(optarg, "lin")) { } else if (!stricmp(optarg, "lin")) {
afl->schedule = LIN; afl->schedule = LIN;