From c9e3de35ed8dd5e47853e14d827c0fac747d6432 Mon Sep 17 00:00:00 2001
From: Kostya Serebryany <kcc@google.com>
Date: Thu, 22 Sep 2016 00:57:29 +0000
Subject: [PATCH] [libFuzzer] one more test

llvm-svn: 282127
---
 llvm/lib/Fuzzer/test/CMakeLists.txt          |  1 +
 llvm/lib/Fuzzer/test/MinimizeCorpusTest.cpp  | 28 ++++++++++++++++++++
 llvm/lib/Fuzzer/test/trace-pc/CMakeLists.txt |  1 +
 3 files changed, 30 insertions(+)
 create mode 100644 llvm/lib/Fuzzer/test/MinimizeCorpusTest.cpp

diff --git a/llvm/lib/Fuzzer/test/CMakeLists.txt b/llvm/lib/Fuzzer/test/CMakeLists.txt
index a1d6e024d552..1406379b097c 100644
--- a/llvm/lib/Fuzzer/test/CMakeLists.txt
+++ b/llvm/lib/Fuzzer/test/CMakeLists.txt
@@ -80,6 +80,7 @@ set(Tests
   LeakTest
   LeakTimeoutTest
   LoadTest
+  MinimizeCorpusTest
   NullDerefTest
   NullDerefOnEmptyTest
   NthRunCrashTest
diff --git a/llvm/lib/Fuzzer/test/MinimizeCorpusTest.cpp b/llvm/lib/Fuzzer/test/MinimizeCorpusTest.cpp
new file mode 100644
index 000000000000..02ef99fb26ee
--- /dev/null
+++ b/llvm/lib/Fuzzer/test/MinimizeCorpusTest.cpp
@@ -0,0 +1,28 @@
+// This file is distributed under the University of Illinois Open Source
+// License. See LICENSE.TXT for details.
+
+// Test that we can find the minimal item in the corpus (3 bytes: "FUZ").
+#include <cstdint>
+#include <cstdlib>
+#include <cstddef>
+#include <cstring>
+#include <cstdio>
+
+static volatile int Sink;
+
+extern "C" int LLVMFuzzerTestOneInput(const uint8_t *Data, size_t Size) {
+  int8_t Ids[256];
+  memset(Ids, -1, sizeof(Ids));
+  for (size_t i = 0; i < Size; i++)
+    if (Ids[Data[i]] == -1)
+      Ids[Data[i]] = i;
+  int F = Ids[(unsigned char)'F'];
+  int U = Ids[(unsigned char)'U'];
+  int Z = Ids[(unsigned char)'Z'];
+  if (F >= 0 && U > F && Z > U) {
+    Sink++;
+    // printf("IDS: %d %d %d\n", F, U, Z);
+  }
+  return 0;
+}
+
diff --git a/llvm/lib/Fuzzer/test/trace-pc/CMakeLists.txt b/llvm/lib/Fuzzer/test/trace-pc/CMakeLists.txt
index 5b6f87acc6fb..ea160ccc502c 100644
--- a/llvm/lib/Fuzzer/test/trace-pc/CMakeLists.txt
+++ b/llvm/lib/Fuzzer/test/trace-pc/CMakeLists.txt
@@ -8,6 +8,7 @@ set(TracePCTests
   CounterTest
   CallerCalleeTest
   NullDerefTest
+  MinimizeCorpusTest
   )
 
 foreach(Test ${TracePCTests})