make AFL_CUSTOM_INFO overridable

This commit is contained in:
vanhauser-thc 2023-05-21 13:04:17 +02:00
parent 53a869b757
commit 9a6c0ec0c0
2 changed files with 7 additions and 3 deletions

View File

@ -8,7 +8,7 @@ all: symqemu-mutator.so
CFLAGS += -O3 -funroll-loops
symqemu-mutator.so: symqemu.c
$(CC) $(CFLAGS) $(CPPFLAGS) -g -I../../include -shared -fPIC -o symqemu-mutator.so symqemu.c
$(CC) -g $(CFLAGS) $(CPPFLAGS) -g -I../../include -shared -fPIC -o symqemu-mutator.so symqemu.c
clean:
rm -f symqemu-mutator.so *.o *~ core

View File

@ -1833,7 +1833,7 @@ int main(int argc, char **argv_orig, char **envp) {
}
{
if (!getenv("AFL_CUSTOM_INFO_PROGRAM_ARGV")) {
u8 envbuf[8096] = "", tmpbuf[8096] = "";
for (s32 i = optind + 1; i < argc; ++i) {
@ -1864,7 +1864,11 @@ int main(int argc, char **argv_orig, char **envp) {
}
setenv("AFL_CUSTOM_INFO_OUT", afl->out_dir, 1); // same as __AFL_OUT_DIR
if (!getenv("AFL_CUSTOM_INFO_OUT") {
setenv("AFL_CUSTOM_INFO_OUT", afl->out_dir, 1); // same as __AFL_OUT_DIR
}
setup_custom_mutators(afl);