From 6de9d4035a3becaa7d92812ca3ededfaf9a0cb29 Mon Sep 17 00:00:00 2001
From: Niels Lohmann <niels.lohmann@gmail.com>
Date: Sun, 25 Dec 2016 16:06:56 +0100
Subject: [PATCH] :bug: fixed AFL driver to also read binary data
---
test/src/fuzzer-driver_afl.cpp | 19 ++++++++++++-------
1 file changed, 12 insertions(+), 7 deletions(-)
diff --git a/test/src/fuzzer-driver_afl.cpp b/test/src/fuzzer-driver_afl.cpp
index e386033a..eddeae41 100644
--- a/test/src/fuzzer-driver_afl.cpp
+++ b/test/src/fuzzer-driver_afl.cpp
@@ -11,9 +11,9 @@ passed byte array.
Licensed under the MIT License <http://opensource.org/licenses/MIT>.
*/
-#include <sstream>
-#include <cstdint>
-#include <iostream>
+#include <vector> // for vector
+#include <cstdint> // for uint8_t
+#include <iostream> // for cin
extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size);
@@ -23,10 +23,15 @@ int main()
while (__AFL_LOOP(1000))
{
#endif
- // copy stdin to stringstream to pass it to fuzzer as byte array
- std::stringstream ss;
- ss << std::cin.rdbuf();
- LLVMFuzzerTestOneInput(reinterpret_cast<const uint8_t*>(ss.str().c_str()), ss.str().size());
+ // copy stdin to byte vector
+ std::vector<uint8_t> vec;
+ char c;
+ while (std::cin.get(c))
+ {
+ vec.push_back(static_cast<uint8_t>(c));
+ }
+
+ LLVMFuzzerTestOneInput(vec.data(), vec.size());
#ifdef __AFL_HAVE_MANUAL_CONTROL
}
#endif