code format

This commit is contained in:
van Hauser 2020-06-01 12:30:55 +02:00
parent bca7ce8043
commit 0de25f08ba
4 changed files with 43 additions and 23 deletions

View File

@ -35,12 +35,18 @@ int main(int argc, char **argv) {
sprintf(fn, "%09u:test-instr", counter); sprintf(fn, "%09u:test-instr", counter);
int fd_doc = open(fn, O_WRONLY | O_CREAT | O_TRUNC, 0600); int fd_doc = open(fn, O_WRONLY | O_CREAT | O_TRUNC, 0600);
if (fd_doc >= 0) { if (fd_doc >= 0) {
if (write(fd_doc, __afl_fuzz_ptr, __afl_fuzz_len) != __afl_fuzz_len) { if (write(fd_doc, __afl_fuzz_ptr, __afl_fuzz_len) != __afl_fuzz_len) {
fprintf(stderr, "write of mutation file failed: %s\n", fn); fprintf(stderr, "write of mutation file failed: %s\n", fn);
unlink(fn); unlink(fn);
} }
close(fd_doc); close(fd_doc);
} }
counter++; counter++;
#endif #endif
@ -52,6 +58,7 @@ int main(int argc, char **argv) {
printf("Pretty sure that is a one!\n"); printf("Pretty sure that is a one!\n");
else else
printf("Neither one or zero? How quaint!\n"); printf("Neither one or zero? How quaint!\n");
} }
return 0; return 0;

View File

@ -166,7 +166,8 @@ static void __afl_map_shm_fuzz() {
} }
__afl_fuzz_len_shmem = (u32*) mmap(NULL, sizeof(int), PROT_READ | PROT_WRITE, MAP_SHARED | MAP_ANONYMOUS, -1, 0); __afl_fuzz_len_shmem = (u32 *)mmap(NULL, sizeof(int), PROT_READ | PROT_WRITE,
MAP_SHARED | MAP_ANONYMOUS, -1, 0);
} }
@ -452,19 +453,28 @@ static void __afl_start_snapshots(void) {
#ifdef _AFL_DOCUMENT_MUTATIONS #ifdef _AFL_DOCUMENT_MUTATIONS
if (__afl_fuzz_ptr) { if (__afl_fuzz_ptr) {
static uint32_t counter = 0; static uint32_t counter = 0;
char fn[32]; char fn[32];
sprintf(fn, "%09u:forkserver", counter); sprintf(fn, "%09u:forkserver", counter);
s32 fd_doc = open(fn, O_WRONLY | O_CREAT | O_TRUNC, 0600); s32 fd_doc = open(fn, O_WRONLY | O_CREAT | O_TRUNC, 0600);
if (fd_doc >= 0) { if (fd_doc >= 0) {
if (write(fd_doc, __afl_fuzz_ptr, __afl_fuzz_len) != __afl_fuzz_len) { if (write(fd_doc, __afl_fuzz_ptr, __afl_fuzz_len) != __afl_fuzz_len) {
fprintf(stderr, "write of mutation file failed: %s\n", fn); fprintf(stderr, "write of mutation file failed: %s\n", fn);
unlink(fn); unlink(fn);
} }
close(fd_doc); close(fd_doc);
} }
counter++; counter++;
} }
#endif #endif
/* If we stopped the child in persistent mode, but there was a race /* If we stopped the child in persistent mode, but there was a race
@ -646,19 +656,28 @@ static void __afl_start_forkserver(void) {
#ifdef _AFL_DOCUMENT_MUTATIONS #ifdef _AFL_DOCUMENT_MUTATIONS
if (__afl_fuzz_ptr) { if (__afl_fuzz_ptr) {
static uint32_t counter = 0; static uint32_t counter = 0;
char fn[32]; char fn[32];
sprintf(fn, "%09u:forkserver", counter); sprintf(fn, "%09u:forkserver", counter);
s32 fd_doc = open(fn, O_WRONLY | O_CREAT | O_TRUNC, 0600); s32 fd_doc = open(fn, O_WRONLY | O_CREAT | O_TRUNC, 0600);
if (fd_doc >= 0) { if (fd_doc >= 0) {
if (write(fd_doc, __afl_fuzz_ptr, __afl_fuzz_len) != __afl_fuzz_len) { if (write(fd_doc, __afl_fuzz_ptr, __afl_fuzz_len) != __afl_fuzz_len) {
fprintf(stderr, "write of mutation file failed: %s\n", fn); fprintf(stderr, "write of mutation file failed: %s\n", fn);
unlink(fn); unlink(fn);
} }
close(fd_doc); close(fd_doc);
} }
counter++; counter++;
} }
#endif #endif
/* If we stopped the child in persistent mode, but there was a race /* If we stopped the child in persistent mode, but there was a race

View File

@ -1949,7 +1949,6 @@ static void handle_skipreq(int sig) {
} }
/* Setup shared map for fuzzing with input via sharedmem */ /* Setup shared map for fuzzing with input via sharedmem */
void setup_testcase_shmem(afl_state_t *afl) { void setup_testcase_shmem(afl_state_t *afl) {
@ -1978,7 +1977,6 @@ void setup_testcase_shmem(afl_state_t *afl) {
} }
/* Do a PATH search and find target binary to see that it exists and /* Do a PATH search and find target binary to see that it exists and
isn't a shell script - a common and painful mistake. We also check for isn't a shell script - a common and painful mistake. We also check for
a valid ELF header and for evidence of AFL instrumentation. */ a valid ELF header and for evidence of AFL instrumentation. */

View File

@ -1179,11 +1179,7 @@ int main(int argc, char **argv_orig, char **envp) {
check_binary(afl, argv[optind]); check_binary(afl, argv[optind]);
if (afl->shmem_testcase_mode) { if (afl->shmem_testcase_mode) { setup_testcase_shmem(afl); }
setup_testcase_shmem(afl);
}
afl->start_time = get_cur_time(); afl->start_time = get_cur_time();