[llvm] r350534 - [llvm-demangle-fuzzer] Also fuzz microsoftDemangle().

[Matt Morehouse via llvm-commits]

Author: morehouse
Date: Mon Jan  7 08:14:00 2019
New Revision: 350534

URL: http://llvm.org/viewvc/llvm-project?rev=350534&view=rev
Log:
[llvm-demangle-fuzzer] Also fuzz microsoftDemangle().

Summary:
Use first byte of input to determine whether to call itaniumDemangle()
or microsoftDemangle().

Addresses https://bugs.llvm.org/show_bug.cgi?id=39582.

Reviewers: kcc, thakis

Reviewed By: kcc, thakis

Subscribers: mgorny, thakis, erik.pilkington, llvm-commits

Differential Revision: https://reviews.llvm.org/D54780

Added:
    llvm/trunk/tools/llvm-itanium-demangle-fuzzer/
    llvm/trunk/tools/llvm-itanium-demangle-fuzzer/CMakeLists.txt
    llvm/trunk/tools/llvm-itanium-demangle-fuzzer/DummyDemanglerFuzzer.cpp
    llvm/trunk/tools/llvm-itanium-demangle-fuzzer/llvm-itanium-demangle-fuzzer.cpp
    llvm/trunk/tools/llvm-microsoft-demangle-fuzzer/
    llvm/trunk/tools/llvm-microsoft-demangle-fuzzer/CMakeLists.txt
    llvm/trunk/tools/llvm-microsoft-demangle-fuzzer/DummyDemanglerFuzzer.cpp
    llvm/trunk/tools/llvm-microsoft-demangle-fuzzer/llvm-microsoft-demangle-fuzzer.cpp
Removed:
    llvm/trunk/tools/llvm-demangle-fuzzer/

Added: llvm/trunk/tools/llvm-itanium-demangle-fuzzer/CMakeLists.txt
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/tools/llvm-itanium-demangle-fuzzer/CMakeLists.txt?rev=350534&view=auto
==============================================================================
--- llvm/trunk/tools/llvm-itanium-demangle-fuzzer/CMakeLists.txt (added)
+++ llvm/trunk/tools/llvm-itanium-demangle-fuzzer/CMakeLists.txt Mon Jan  7 08:14:00 2019
@@ -0,0 +1,10 @@
+set(LLVM_LINK_COMPONENTS
+  Demangle
+  FuzzMutate
+  Support
+)
+
+add_llvm_fuzzer(llvm-itanium-demangle-fuzzer
+  llvm-itanium-demangle-fuzzer.cpp
+  DUMMY_MAIN DummyDemanglerFuzzer.cpp
+  )

Added: llvm/trunk/tools/llvm-itanium-demangle-fuzzer/DummyDemanglerFuzzer.cpp
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/tools/llvm-itanium-demangle-fuzzer/DummyDemanglerFuzzer.cpp?rev=350534&view=auto
==============================================================================
--- llvm/trunk/tools/llvm-itanium-demangle-fuzzer/DummyDemanglerFuzzer.cpp (added)
+++ llvm/trunk/tools/llvm-itanium-demangle-fuzzer/DummyDemanglerFuzzer.cpp Mon Jan  7 08:14:00 2019
@@ -0,0 +1,19 @@
+//===--- DummyDemanglerMain.cpp - Entry point to sanity check the fuzzer --===//
+//
+//                     The LLVM Compiler Infrastructure
+//
+// This file is distributed under the University of Illinois Open Source
+// License. See LICENSE.TXT for details.
+//
+//===----------------------------------------------------------------------===//
+//
+// Implementation of main so we can build and test without linking libFuzzer.
+//
+//===----------------------------------------------------------------------===//
+
+#include "llvm/FuzzMutate/FuzzerCLI.h"
+
+extern "C" int LLVMFuzzerTestOneInput(const uint8_t *Data, size_t Size);
+int main(int argc, char *argv[]) {
+  return llvm::runFuzzerOnInputs(argc, argv, LLVMFuzzerTestOneInput);
+}

Added: llvm/trunk/tools/llvm-itanium-demangle-fuzzer/llvm-itanium-demangle-fuzzer.cpp
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/tools/llvm-itanium-demangle-fuzzer/llvm-itanium-demangle-fuzzer.cpp?rev=350534&view=auto
==============================================================================
--- llvm/trunk/tools/llvm-itanium-demangle-fuzzer/llvm-itanium-demangle-fuzzer.cpp (added)
+++ llvm/trunk/tools/llvm-itanium-demangle-fuzzer/llvm-itanium-demangle-fuzzer.cpp Mon Jan  7 08:14:00 2019
@@ -0,0 +1,24 @@
+//===--- llvm-demangle-fuzzer.cpp - Fuzzer for the Itanium Demangler ------===//
+//
+//                     The LLVM Compiler Infrastructure
+//
+// This file is distributed under the University of Illinois Open Source
+// License. See LICENSE.TXT for details.
+//
+//===----------------------------------------------------------------------===//
+
+#include "llvm/Demangle/Demangle.h"
+
+#include <cstdint>
+#include <cstdlib>
+#include <string>
+
+extern "C" int LLVMFuzzerTestOneInput(const uint8_t *Data, size_t Size) {
+  std::string NullTerminatedString((const char *)Data, Size);
+  int status = 0;
+  if (char *demangle = llvm::itaniumDemangle(NullTerminatedString.c_str(), nullptr,
+                                         nullptr, &status))
+    free(demangle);
+
+  return 0;
+}

Added: llvm/trunk/tools/llvm-microsoft-demangle-fuzzer/CMakeLists.txt
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/tools/llvm-microsoft-demangle-fuzzer/CMakeLists.txt?rev=350534&view=auto
==============================================================================
--- llvm/trunk/tools/llvm-microsoft-demangle-fuzzer/CMakeLists.txt (added)
+++ llvm/trunk/tools/llvm-microsoft-demangle-fuzzer/CMakeLists.txt Mon Jan  7 08:14:00 2019
@@ -0,0 +1,10 @@
+set(LLVM_LINK_COMPONENTS
+  Demangle
+  FuzzMutate
+  Support
+)
+
+add_llvm_fuzzer(llvm-microsoft-demangle-fuzzer
+  llvm-microsoft-demangle-fuzzer.cpp
+  DUMMY_MAIN DummyDemanglerFuzzer.cpp
+  )

Added: llvm/trunk/tools/llvm-microsoft-demangle-fuzzer/DummyDemanglerFuzzer.cpp
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/tools/llvm-microsoft-demangle-fuzzer/DummyDemanglerFuzzer.cpp?rev=350534&view=auto
==============================================================================
--- llvm/trunk/tools/llvm-microsoft-demangle-fuzzer/DummyDemanglerFuzzer.cpp (added)
+++ llvm/trunk/tools/llvm-microsoft-demangle-fuzzer/DummyDemanglerFuzzer.cpp Mon Jan  7 08:14:00 2019
@@ -0,0 +1,19 @@
+//===--- DummyDemanglerMain.cpp - Entry point to sanity check the fuzzer --===//
+//
+//                     The LLVM Compiler Infrastructure
+//
+// This file is distributed under the University of Illinois Open Source
+// License. See LICENSE.TXT for details.
+//
+//===----------------------------------------------------------------------===//
+//
+// Implementation of main so we can build and test without linking libFuzzer.
+//
+//===----------------------------------------------------------------------===//
+
+#include "llvm/FuzzMutate/FuzzerCLI.h"
+
+extern "C" int LLVMFuzzerTestOneInput(const uint8_t *Data, size_t Size);
+int main(int argc, char *argv[]) {
+  return llvm::runFuzzerOnInputs(argc, argv, LLVMFuzzerTestOneInput);
+}

Added: llvm/trunk/tools/llvm-microsoft-demangle-fuzzer/llvm-microsoft-demangle-fuzzer.cpp
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/tools/llvm-microsoft-demangle-fuzzer/llvm-microsoft-demangle-fuzzer.cpp?rev=350534&view=auto
==============================================================================
--- llvm/trunk/tools/llvm-microsoft-demangle-fuzzer/llvm-microsoft-demangle-fuzzer.cpp (added)
+++ llvm/trunk/tools/llvm-microsoft-demangle-fuzzer/llvm-microsoft-demangle-fuzzer.cpp Mon Jan  7 08:14:00 2019
@@ -0,0 +1,21 @@
+//===--- llvm-demangle-fuzzer.cpp - Fuzzer for the Itanium Demangler ------===//
+//
+//                     The LLVM Compiler Infrastructure
+//
+// This file is distributed under the University of Illinois Open Source
+// License. See LICENSE.TXT for details.
+//
+//===----------------------------------------------------------------------===//
+
+#include "llvm/Demangle/Demangle.h"
+
+#include <cstdint>
+#include <cstdlib>
+#include <string>
+
+extern "C" int LLVMFuzzerTestOneInput(const uint8_t *Data, size_t Size) {
+  std::string NullTerminatedString((const char *)Data, Size);
+  free(llvm::microsoftDemangle(NullTerminatedString.c_str(), nullptr, nullptr,
+                               nullptr));
+  return 0;
+}