diff options
| author | van Hauser <vh@thc.org> | 2020-08-14 13:23:14 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2020-08-14 13:23:14 +0200 |
| commit | a7537b5511ad767d2240cf2dc6d3e261daa676f9 (patch) | |
| tree | 2cd8418efff844e634425e050ef716301adfd84b /src/afl-forkserver.c | |
| parent | 15e799f7ae666418e75c6a79db833c5316b21f97 (diff) | |
| parent | 83281503b3f1d4109c6f7a6af62a2133d55f1150 (diff) | |
| download | afl++-a7537b5511ad767d2240cf2dc6d3e261daa676f9.tar.gz | |
Merge branch 'debug' into dev
Diffstat (limited to 'src/afl-forkserver.c')
| -rw-r--r-- | src/afl-forkserver.c | 17 |
1 files changed, 14 insertions, 3 deletions
diff --git a/src/afl-forkserver.c b/src/afl-forkserver.c index 25983f26..173cc70f 100644 --- a/src/afl-forkserver.c +++ b/src/afl-forkserver.c @@ -498,11 +498,21 @@ void afl_fsrv_start(afl_forkserver_t *fsrv, char **argv, char pid_buf[16]; sprintf(pid_buf, "%d", fsrv->fsrv_pid); - if (fsrv->cmplog_binary) + + if (fsrv->taint_mode) { + + setenv("__AFL_TARGET_PID3", pid_buf, 1); + + } else if (fsrv->cmplog_binary) { + setenv("__AFL_TARGET_PID2", pid_buf, 1); - else + + } else { + setenv("__AFL_TARGET_PID1", pid_buf, 1); + } + /* Close the unneeded endpoints. */ close(ctl_pipe[0]); @@ -937,7 +947,7 @@ void afl_fsrv_write_to_testcase(afl_forkserver_t *fsrv, u8 *buf, size_t len) { } else { - s32 fd = fsrv->out_fd; + s32 fd; if (fsrv->out_file) { @@ -956,6 +966,7 @@ void afl_fsrv_write_to_testcase(afl_forkserver_t *fsrv, u8 *buf, size_t len) { } else { + fd = fsrv->out_fd; lseek(fd, 0, SEEK_SET); } |