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

afl-showmap -i with stdin

Browse Source
This commit is contained in:
van Hauser
2020-02-04 20:14:36 +01:00
parent 5fa62e40b1
commit 599f78a4bd
1 changed files with 35 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

41
src/afl-showmap.c
View File

@ -67,7 +67,7 @@ s32 forksrv_pid, /* PID of the fork server */
s32 fsrv_ctl_fd, /* Fork server control pipe (write) */
fsrv_st_fd; /* Fork server status pipe (read) */
s32 out_fd; /* Persistent fd for out_file */
s32 out_fd; /* Persistent fd for stdin_file */
s32 dev_null_fd = -1; /* FD to /dev/null */
s32 out_fd = -1, out_dir_fd = -1, dev_urandom_fd = -1;
@ -77,6 +77,7 @@ u8 uses_asan;
u8* trace_bits; /* SHM with instrumentation bitmap */
u8 *out_file, /* Trace output file */
*stdin_file, /* stdin file */
*in_dir, /* input folder */
*doc_path, /* Path to docs */
*at_file; /* Substitution string for @@ */
@ -158,6 +159,14 @@ static void classify_counts(u8* mem, const u8* map) {
}
/* Get rid of temp files (atexit handler). */
static void at_exit_handler(void) {
if (out_file) unlink(out_file); /* Ignore errors */
}
/* Write results. */
static u32 write_results_to_file(u8 *out_file) {
@ -265,12 +274,12 @@ static void write_to_testcase(void* mem, u32 len) {
if (use_stdin) {
lseek(0, 0, SEEK_SET);
lseek(out_fd, 0, SEEK_SET);
ck_write(0, mem, len, out_file);
ck_write(out_fd, mem, len, out_file);
if (ftruncate(0, len)) PFATAL("ftruncate() failed");
lseek(0, 0, SEEK_SET);
if (ftruncate(out_fd, len)) PFATAL("ftruncate() failed");
lseek(out_fd, 0, SEEK_SET);
}
@ -887,7 +896,7 @@ int main(int argc, char** argv) {
if (!quiet_mode) {
show_banner();
ACTF("Executing '%s'...\n", target_path);
ACTF("Executing '%s'...", target_path);
}
@ -932,6 +941,24 @@ int main(int argc, char** argv) {
PFATAL("cannot create output directory %s", out_file);
if (arg_offset) argv[arg_offset] = infile;
else {
u8* use_dir = ".";
if (access(use_dir, R_OK | W_OK | X_OK)) {
use_dir = getenv("TMPDIR");
if (!use_dir) use_dir = "/tmp";
}
stdin_file = alloc_printf("%s/.afl-tmin-temp-%u", use_dir, getpid());
unlink(stdin_file);
atexit(at_exit_handler);
out_fd = open(stdin_file, O_RDWR | O_CREAT | O_EXCL, 0600);
if (out_fd < 0) PFATAL("Unable to create '%s'", out_file);
}
init_forkserver(use_argv);
@ -950,6 +977,8 @@ int main(int argc, char** argv) {
}
}
if (!quiet_mode) OKF("Processed %u input files.", total_execs);
} else {