mirror of
https://github.com/AFLplusplus/AFLplusplus.git
synced 2025-06-18 12:48:06 +00:00
renamed examples/ to utils/
This commit is contained in:
10
utils/persistent_mode/Makefile
Normal file
10
utils/persistent_mode/Makefile
Normal file
@ -0,0 +1,10 @@
|
||||
all:
|
||||
afl-clang-fast -o persistent_demo persistent_demo.c
|
||||
afl-clang-fast -o persistent_demo_new persistent_demo_new.c
|
||||
AFL_DONT_OPTIMIZE=1 afl-clang-fast -o test-instr test-instr.c
|
||||
|
||||
document:
|
||||
AFL_DONT_OPTIMIZE=1 afl-clang-fast -D_AFL_DOCUMENT_MUTATIONS -o test-instr test-instr.c
|
||||
|
||||
clean:
|
||||
rm -f persistent_demo persistent_demo_new test-instr
|
112
utils/persistent_mode/persistent_demo.c
Normal file
112
utils/persistent_mode/persistent_demo.c
Normal file
@ -0,0 +1,112 @@
|
||||
/*
|
||||
american fuzzy lop++ - persistent mode example
|
||||
--------------------------------------------
|
||||
|
||||
Originally written by Michal Zalewski
|
||||
|
||||
Copyright 2015 Google Inc. All rights reserved.
|
||||
|
||||
Licensed under the Apache License, Version 2.0 (the "License");
|
||||
you may not use this file except in compliance with the License.
|
||||
You may obtain a copy of the License at:
|
||||
|
||||
http://www.apache.org/licenses/LICENSE-2.0
|
||||
|
||||
This file demonstrates the high-performance "persistent mode" that may be
|
||||
suitable for fuzzing certain fast and well-behaved libraries, provided that
|
||||
they are stateless or that their internal state can be easily reset
|
||||
across runs.
|
||||
|
||||
To make this work, the library and this shim need to be compiled in LLVM
|
||||
mode using afl-clang-fast (other compiler wrappers will *not* work).
|
||||
|
||||
*/
|
||||
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <unistd.h>
|
||||
#include <signal.h>
|
||||
#include <string.h>
|
||||
|
||||
/* Main entry point. */
|
||||
|
||||
int main(int argc, char **argv) {
|
||||
|
||||
ssize_t len; /* how much input did we read? */
|
||||
char buf[100]; /* Example-only buffer, you'd replace it with other global or
|
||||
local variables appropriate for your use case. */
|
||||
|
||||
/* The number passed to __AFL_LOOP() controls the maximum number of
|
||||
iterations before the loop exits and the program is allowed to
|
||||
terminate normally. This limits the impact of accidental memory leaks
|
||||
and similar hiccups. */
|
||||
|
||||
__AFL_INIT();
|
||||
while (__AFL_LOOP(1000)) {
|
||||
|
||||
/*** PLACEHOLDER CODE ***/
|
||||
|
||||
/* STEP 1: Fully re-initialize all critical variables. In our example, this
|
||||
involves zeroing buf[], our input buffer. */
|
||||
|
||||
memset(buf, 0, 100);
|
||||
|
||||
/* STEP 2: Read input data. When reading from stdin, no special preparation
|
||||
is required. When reading from a named file, you need to close
|
||||
the old descriptor and reopen the file first!
|
||||
|
||||
Beware of reading from buffered FILE* objects such as stdin. Use
|
||||
raw file descriptors or call fopen() / fdopen() in every pass. */
|
||||
|
||||
len = read(0, buf, 100);
|
||||
|
||||
/* STEP 3: This is where we'd call the tested library on the read data.
|
||||
We just have some trivial inline code that faults on 'foo!'. */
|
||||
|
||||
/* do we have enough data? */
|
||||
if (len < 8) continue;
|
||||
|
||||
if (buf[0] == 'f') {
|
||||
|
||||
printf("one\n");
|
||||
if (buf[1] == 'o') {
|
||||
|
||||
printf("two\n");
|
||||
if (buf[2] == 'o') {
|
||||
|
||||
printf("three\n");
|
||||
if (buf[3] == '!') {
|
||||
|
||||
printf("four\n");
|
||||
if (buf[4] == '!') {
|
||||
|
||||
printf("five\n");
|
||||
if (buf[5] == '!') {
|
||||
|
||||
printf("six\n");
|
||||
abort();
|
||||
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
/*** END PLACEHOLDER CODE ***/
|
||||
|
||||
}
|
||||
|
||||
/* Once the loop is exited, terminate normally - AFL will restart the process
|
||||
when this happens, with a clean slate when it comes to allocated memory,
|
||||
leftover file descriptors, etc. */
|
||||
|
||||
return 0;
|
||||
|
||||
}
|
||||
|
117
utils/persistent_mode/persistent_demo_new.c
Normal file
117
utils/persistent_mode/persistent_demo_new.c
Normal file
@ -0,0 +1,117 @@
|
||||
/*
|
||||
american fuzzy lop++ - persistent mode example
|
||||
--------------------------------------------
|
||||
|
||||
Originally written by Michal Zalewski
|
||||
|
||||
Copyright 2015 Google Inc. All rights reserved.
|
||||
|
||||
Licensed under the Apache License, Version 2.0 (the "License");
|
||||
you may not use this file except in compliance with the License.
|
||||
You may obtain a copy of the License at:
|
||||
|
||||
http://www.apache.org/licenses/LICENSE-2.0
|
||||
|
||||
This file demonstrates the high-performance "persistent mode" that may be
|
||||
suitable for fuzzing certain fast and well-behaved libraries, provided that
|
||||
they are stateless or that their internal state can be easily reset
|
||||
across runs.
|
||||
|
||||
To make this work, the library and this shim need to be compiled in LLVM
|
||||
mode using afl-clang-fast (other compiler wrappers will *not* work).
|
||||
|
||||
*/
|
||||
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <unistd.h>
|
||||
#include <signal.h>
|
||||
#include <string.h>
|
||||
|
||||
/* this lets the source compile without afl-clang-fast/lto */
|
||||
#ifndef __AFL_FUZZ_TESTCASE_LEN
|
||||
|
||||
ssize_t fuzz_len;
|
||||
unsigned char fuzz_buf[1024000];
|
||||
|
||||
#define __AFL_FUZZ_TESTCASE_LEN fuzz_len
|
||||
#define __AFL_FUZZ_TESTCASE_BUF fuzz_buf
|
||||
#define __AFL_FUZZ_INIT() void sync(void);
|
||||
#define __AFL_LOOP(x) \
|
||||
((fuzz_len = read(0, fuzz_buf, sizeof(fuzz_buf))) > 0 ? 1 : 0)
|
||||
#define __AFL_INIT() sync()
|
||||
|
||||
#endif
|
||||
|
||||
__AFL_FUZZ_INIT();
|
||||
|
||||
/* Main entry point. */
|
||||
|
||||
int main(int argc, char **argv) {
|
||||
|
||||
ssize_t len; /* how much input did we read? */
|
||||
unsigned char *buf; /* test case buffer pointer */
|
||||
|
||||
/* The number passed to __AFL_LOOP() controls the maximum number of
|
||||
iterations before the loop exits and the program is allowed to
|
||||
terminate normally. This limits the impact of accidental memory leaks
|
||||
and similar hiccups. */
|
||||
|
||||
__AFL_INIT();
|
||||
buf = __AFL_FUZZ_TESTCASE_BUF; // this must be assigned before __AFL_LOOP!
|
||||
|
||||
while (__AFL_LOOP(1000)) { // increase if you have good stability
|
||||
|
||||
len = __AFL_FUZZ_TESTCASE_LEN; // do not use the macro directly in a call!
|
||||
|
||||
fprintf(stderr, "input: %zd \"%s\"\n", len, buf);
|
||||
|
||||
/* do we have enough data? */
|
||||
if (len < 8) continue;
|
||||
|
||||
if (strcmp((char *)buf, "thisisateststring") == 0) printf("teststring\n");
|
||||
|
||||
if (buf[0] == 'f') {
|
||||
|
||||
printf("one\n");
|
||||
if (buf[1] == 'o') {
|
||||
|
||||
printf("two\n");
|
||||
if (buf[2] == 'o') {
|
||||
|
||||
printf("three\n");
|
||||
if (buf[3] == '!') {
|
||||
|
||||
printf("four\n");
|
||||
if (buf[4] == '!') {
|
||||
|
||||
printf("five\n");
|
||||
if (buf[6] == '!') {
|
||||
|
||||
printf("six\n");
|
||||
abort();
|
||||
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
/*** END PLACEHOLDER CODE ***/
|
||||
|
||||
}
|
||||
|
||||
/* Once the loop is exited, terminate normally - AFL will restart the process
|
||||
when this happens, with a clean slate when it comes to allocated memory,
|
||||
leftover file descriptors, etc. */
|
||||
|
||||
return 0;
|
||||
|
||||
}
|
||||
|
69
utils/persistent_mode/test-instr.c
Normal file
69
utils/persistent_mode/test-instr.c
Normal file
@ -0,0 +1,69 @@
|
||||
/*
|
||||
american fuzzy lop++ - a trivial program to test the build
|
||||
--------------------------------------------------------
|
||||
Originally written by Michal Zalewski
|
||||
Copyright 2014 Google Inc. All rights reserved.
|
||||
Copyright 2019-2020 AFLplusplus Project. All rights reserved.
|
||||
Licensed under the Apache License, Version 2.0 (the "License");
|
||||
you may not use this file except in compliance with the License.
|
||||
You may obtain a copy of the License at:
|
||||
http://www.apache.org/licenses/LICENSE-2.0
|
||||
*/
|
||||
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <unistd.h>
|
||||
#include <string.h>
|
||||
#include <sys/types.h>
|
||||
#include <sys/stat.h>
|
||||
#include <fcntl.h>
|
||||
|
||||
__AFL_FUZZ_INIT();
|
||||
|
||||
int main(int argc, char **argv) {
|
||||
|
||||
__AFL_INIT();
|
||||
unsigned char *buf = __AFL_FUZZ_TESTCASE_BUF;
|
||||
|
||||
while (__AFL_LOOP(2147483647)) { // MAX_INT if you have 100% stability
|
||||
|
||||
unsigned int len = __AFL_FUZZ_TESTCASE_LEN;
|
||||
|
||||
#ifdef _AFL_DOCUMENT_MUTATIONS
|
||||
static unsigned int counter = 0;
|
||||
char fn[32];
|
||||
sprintf(fn, "%09u:test-instr", counter);
|
||||
int fd_doc = open(fn, O_WRONLY | O_CREAT | O_TRUNC, 0600);
|
||||
if (fd_doc >= 0) {
|
||||
|
||||
if (write(fd_doc, buf, len) != __afl_fuzz_len) {
|
||||
|
||||
fprintf(stderr, "write of mutation file failed: %s\n", fn);
|
||||
unlink(fn);
|
||||
|
||||
}
|
||||
|
||||
close(fd_doc);
|
||||
|
||||
}
|
||||
|
||||
counter++;
|
||||
#endif
|
||||
|
||||
// fprintf(stderr, "len: %u\n", len);
|
||||
|
||||
if (!len) continue;
|
||||
|
||||
if (buf[0] == '0')
|
||||
printf("Looks like a zero to me!\n");
|
||||
else if (buf[0] == '1')
|
||||
printf("Pretty sure that is a one!\n");
|
||||
else
|
||||
printf("Neither one or zero? How quaint!\n");
|
||||
|
||||
}
|
||||
|
||||
return 0;
|
||||
|
||||
}
|
||||
|
Reference in New Issue
Block a user