ExternalVendorCode/AFLplusplus
1
0
Fork 0
mirror of https://github.com/AFLplusplus/AFLplusplus.git synced 2025-06-14 19:08:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

Added out_file value when using stdio (#524)

Browse Source
This commit is contained in:
Sergio Paganoni
2020-08-24 21:18:51 +02:00
committed by GitHub
parent 19eddbb0c7
commit 1efc6e59b7
2 changed files with 7 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/afl-forkserver.c
View File

@ -941,7 +941,7 @@ void afl_fsrv_write_to_testcase(afl_forkserver_t *fsrv, u8 *buf, size_t len) {
s32 fd = fsrv->out_fd; s32 fd = fsrv->out_fd;
if (fsrv->out_file) { if (!fsrv->use_stdin) {
if (fsrv->no_unlink) { if (fsrv->no_unlink) {
@ -964,7 +964,7 @@ void afl_fsrv_write_to_testcase(afl_forkserver_t *fsrv, u8 *buf, size_t len) {
ck_write(fd, buf, len, fsrv->out_file); ck_write(fd, buf, len, fsrv->out_file);
if (!fsrv->out_file) { if (fsrv->use_stdin) {
if (ftruncate(fd, len)) { PFATAL("ftruncate() failed"); } if (ftruncate(fd, len)) { PFATAL("ftruncate() failed"); }
lseek(fd, 0, SEEK_SET); lseek(fd, 0, SEEK_SET);

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

@ -1841,24 +1841,21 @@ void setup_cmdline_file(afl_state_t *afl, char **argv) {
void setup_stdio_file(afl_state_t *afl) { void setup_stdio_file(afl_state_t *afl) {
u8 *fn;
if (afl->file_extension) { if (afl->file_extension) {
fn = alloc_printf("%s/.cur_input.%s", afl->tmp_dir, afl->file_extension); afl->fsrv.out_file = alloc_printf("%s/.cur_input.%s", afl->tmp_dir, afl->file_extension);
} else { } else {
fn = alloc_printf("%s/.cur_input", afl->tmp_dir); afl->fsrv.out_file = alloc_printf("%s/.cur_input", afl->tmp_dir);
} }
unlink(fn); /* Ignore errors */ unlink(afl->fsrv.out_file); /* Ignore errors */
afl->fsrv.out_fd = open(fn, O_RDWR | O_CREAT | O_EXCL, 0600); afl->fsrv.out_fd = open(afl->fsrv.out_file, O_RDWR | O_CREAT | O_EXCL, 0600);
if (afl->fsrv.out_fd < 0) { PFATAL("Unable to create '%s'", fn); } if (afl->fsrv.out_fd < 0) { PFATAL("Unable to create '%s'", afl->fsrv.out_file); }
ck_free(fn);
} }