showmap: delete tmpfile on exit

This commit is contained in:
van Hauser 2020-02-18 10:35:27 +01:00
parent 3e20edf10e
commit ac08a38db9

View File

@ -162,7 +162,7 @@ static void classify_counts(u8* mem, const u8* map) {
static void at_exit_handler(void) { static void at_exit_handler(void) {
if (out_file) unlink(out_file); /* Ignore errors */ if (stdin_file) unlink(stdin_file);
} }
@ -949,7 +949,7 @@ int main(int argc, char** argv, char** envp) {
} }
stdin_file = alloc_printf("%s/.afl-tmin-temp-%u", use_dir, getpid()); stdin_file = alloc_printf("%s/.afl-showmap-temp-%u", use_dir, getpid());
unlink(stdin_file); unlink(stdin_file);
atexit(at_exit_handler); atexit(at_exit_handler);
out_fd = open(stdin_file, O_RDWR | O_CREAT | O_EXCL, 0600); out_fd = open(stdin_file, O_RDWR | O_CREAT | O_EXCL, 0600);
@ -1014,6 +1014,13 @@ int main(int argc, char** argv, char** envp) {
tcnt, highest, total, out_file); tcnt, highest, total, out_file);
} }
if (stdin_file) {
unlink(stdin_file);
stdin_file = NULL;
}
exit(child_crashed * 2 + child_timed_out); exit(child_crashed * 2 + child_timed_out);