commit 6b08be92791be8cf97026e472e38bbef59b5bdff
author Kostya Serebryany <kcc@google.com> Tue Jul 19 18:29:06 2016 +0000
committer Kostya Serebryany <kcc@google.com> Tue Jul 19 18:29:06 2016 +0000
tree bbecfa6ee8984414fd72a7db3240032655eb8f01
parent 8b5fa7a2f20a94aae76433addffc7c3d271f6f77

[libFuzzer] properly intercept memmem

llvm-svn: 276006

llvm/lib/Fuzzer/test/StrstrTest.cpp

diff --git a/llvm/lib/Fuzzer/test/StrstrTest.cpp b/llvm/lib/Fuzzer/test/StrstrTest.cpp
index 90d539b6..dd83953 100644
--- a/llvm/lib/Fuzzer/test/StrstrTest.cpp
+++ b/llvm/lib/Fuzzer/test/StrstrTest.cpp
@@ -9,8 +9,12 @@
 #include <cstdlib>
 
 extern "C" int LLVMFuzzerTestOneInput(const uint8_t *Data, size_t Size) {
+  if (Size < 4) return 0;
   std::string s(reinterpret_cast<const char*>(Data), Size);
-  if (strstr(s.c_str(), "FUZZ") && strcasestr(s.c_str(), "aBcD")) {
+  if (strstr(s.c_str(), "FUZZ") &&
+      strcasestr(s.c_str(), "aBcD") &&
+      memmem(s.data(), s.size(), "kuku", 4)
+      ) {
     fprintf(stderr, "BINGO\n");
     exit(1);
   }