Index: third_party/afl/src/experimental/persistent_demo/persistent_demo.c |
diff --git a/third_party/afl/src/experimental/persistent_demo/persistent_demo.c b/third_party/afl/src/experimental/persistent_demo/persistent_demo.c |
new file mode 100644 |
index 0000000000000000000000000000000000000000..0b1e29897b27f42091ea44e4f08af35643963bc5 |
--- /dev/null |
+++ b/third_party/afl/src/experimental/persistent_demo/persistent_demo.c |
@@ -0,0 +1,89 @@ |
+/* |
+ american fuzzy lop - persistent mode example |
+ -------------------------------------------- |
+ |
+ Written and maintained by Michal Zalewski <lcamtuf@google.com> |
+ |
+ 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) { |
+ |
+ 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. */ |
+ |
+ 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. */ |
+ |
+ 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!'. */ |
+ |
+ 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"); |
+ 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; |
+ |
+} |