1 files changed, 87 insertions, 0 deletions
diff --git a/src/extra_tests/fuzzers/jigs/driver.h b/src/extra_tests/fuzzers/jigs/driver.h
new file mode 100644
index 000000000..530cc80b7
--- /dev/null
+++ b/src/extra_tests/fuzzers/jigs/driver.h
@@ -0,0 +1,87 @@
+/*
+* (C) 2015,2016 Jack Lloyd
+*
+* Botan is released under the Simplified BSD License (see license.txt)
+*/
+
+#ifndef FUZZER_DRIVER_H_
+#define FUZZER_DRIVER_H_
+
+#include <stdint.h>
+#include <iostream>
+#include <vector>
+#include <stdlib.h> // for setenv
+#include <botan/exceptn.h>
+
+using namespace Botan;
+
+void fuzz(const uint8_t in[], size_t len);
+
+void fuzzer_init()
+   {
+   /*
+   * This disables the mlock pool, as overwrites within the pool are
+   * opaque to ASan or other instrumentation.
+   */
+   ::setenv("BOTAN_MLOCK_POOL_SIZE", "0", 1);
+   }
+
+#if defined(USE_LLVM_FUZZER)
+
+// Called by main() in libFuzzer
+extern "C" int LLVMFuzzerTestOneInput(const uint8_t in[], size_t len)
+   {
+   fuzz(in, len);
+   return 0;
+   }
+
+int LLVMFuzzerInitialize(int *argc, char ***argv) {
+  fuzzer_init();
+  return 0;
+}
+
+#else
+
+// Read stdin for AFL
+
+int main(int argc, char* argv[])
+   {
+   const size_t max_read = 4096;
+
+   fuzzer_init();
+
+#if defined(__AFL_LOOP)
+   while(__AFL_LOOP(1000))
+#endif
+      {
+      std::vector<uint8_t> buf(max_read);
+      std::cin.read((char*)buf.data(), buf.size());
+      size_t got = std::cin.gcount();
+
+      buf.resize(got);
+      buf.shrink_to_fit();
+
+      fuzz(buf.data(), got);
+      }
+   }
+
+#endif
+
+#endif
+
+// Some helpers for the fuzzer jigs
+
+#define FUZZER_ASSERT_EQUAL(x, y) do {                                  \
+   if(x != y) {                                                         \
+      std::cerr << #x << " = " << x << " !=\n" << #y << " = " << y         \
+                << " at " << __LINE__ << ":" << __FILE__ << std::endl;     \
+      abort();                                                          \
+} } while(0)
+
+#define FUZZER_ASSERT_TRUE(e)                                           \
+   do {                                                                 \
+   if(!(e)) {                                                           \
+   std::cerr << "Expression " << #e << " was false at "                 \
+             << __LINE__ << ":" << __FILE__ << std::endl;               \
+   abort();                                                             \
+   } } while(0)