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

Add post library API as custom mutator and rename pre_save

Browse Source
This commit is contained in:
rish9101
2020-05-13 18:59:12 +05:30
parent f8b3d34225
commit 9627458ecc
12 changed files with 75 additions and 93 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
src/afl-fuzz-init.c
View File

@ -297,12 +297,16 @@ void setup_post(afl_state_t *afl) {
dh = dlopen(fn, RTLD_NOW);
if (!dh) { FATAL("%s", dlerror()); }
afl->post_handler = dlsym(dh, "afl_postprocess");
if (!afl->post_handler) { FATAL("Symbol 'afl_postprocess' not found."); }
afl->post_init = dlsym(dh, "afl_postprocess_init");
if (!afl->post_init) { FATAL("Symbol 'afl_postprocess_init' not found."); }
afl->post_deinit = dlsym(dh, "afl_postprocess_deinit");
if (!afl->post_deinit) {
struct custom_mutator * mutator;
mutator = ck_alloc(sizeof(struct custom_mutator));
memset(mutator, 0, sizeof(struct custom_mutator));
mutator->afl_custom_post_process = dlsym(dh, "afl_postprocess");
if (!mutator->afl_custom_post_process) { FATAL("Symbol 'afl_postprocess' not found."); }
mutator->afl_custom_init = dlsym(dh, "afl_postprocess_init");
if (!mutator->afl_custom_init) { FATAL("Symbol 'afl_postprocess_init' not found."); }
mutator->afl_custom_deinit = dlsym(dh, "afl_postprocess_deinit");
if (!mutator->afl_custom_post_process) {
FATAL("Symbol 'afl_postprocess_deinit' not found.");
@ -311,16 +315,18 @@ void setup_post(afl_state_t *afl) {
/* Do a quick test. It's better to segfault now than later =) */
u8 *post_buf = NULL;
afl->post_data = afl->post_init(afl);
if (!afl->post_data) { FATAL("Could not initialize post handler."); }
mutator->data = mutator->afl_custom_init(afl, rand_below(afl, 0xFFFFFFFF));
if (!mutator->data) { FATAL("Could not initialize post handler."); }
size_t post_len = afl->post_handler(afl->post_data, tbuf, tlen, &post_buf);
size_t post_len = mutator->afl_custom_post_process(mutator->data, tbuf, tlen, &post_buf);
if (!post_len || !post_buf) {
SAYF("Empty return in test post handler for buf=\"hello\\0\".");
}
afl->post_library_mutator = mutator;
OKF("Postprocessor installed successfully.");
}