llvm/tools/llvm-demangle-fuzzer/llvm-demangle-fuzzer.cpp

//===--- 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) {
  if (Size == 0)
    return 0;

  bool UseItanium = Data[0] < 128;
  std::string NullTerminatedString((const char *)Data + 1, Size - 1);

  if (UseItanium) {
    free(llvm::itaniumDemangle(NullTerminatedString.c_str(), nullptr, nullptr,
                               nullptr));
  } else {
    free(llvm::microsoftDemangle(NullTerminatedString.c_str(), nullptr, nullptr,
                                 nullptr));
  }

  return 0;
}