[libFuzzer] implement memcmp hook for data-flow-guided fuzzing (w/o dfsan), extend...

[oota-llvm.git] / lib / Fuzzer / test / MemcmpTest.cpp

diff --git a/lib/Fuzzer/test/MemcmpTest.cpp b/lib/Fuzzer/test/MemcmpTest.cpp
index 510a24398005e94a39939531eb5bfceaa87f63d2..cabdff8f0753d5c6689d8126001d8829b86bf7b8 100644 (file)
--- a/lib/Fuzzer/test/MemcmpTest.cpp
+++ b/lib/Fuzzer/test/MemcmpTest.cpp
@@ -5,8 +5,13 @@
 #include <cstdlib>
 
 extern "C" void LLVMFuzzerTestOneInput(const uint8_t *Data, size_t Size) {
+  // TODO: check other sizes.
   if (Size >= 8 && memcmp(Data, "01234567", 8) == 0) {
-    fprintf(stderr, "BINGO\n");
-    exit(1);
+    if (Size >= 12 && memcmp(Data + 8, "ABCD", 4) == 0) {
+      if (Size >= 14 && memcmp(Data + 12, "XY", 2) == 0) {
+        fprintf(stderr, "BINGO\n");
+        exit(1);
+      }
+    }
   }
 }