tools/llvm-dwarfdump/fuzzer/llvm-dwarfdump-fuzzer.cpp

   1 //===-- llvm-dwarfdump-fuzzer.cpp - Fuzz the llvm-dwarfdump tool ----------===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file is distributed under the University of Illinois Open Source
   6 // License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9 ///
  10 /// \file
  11 /// This file implements a function that runs llvm-dwarfdump
  12 ///  on a single input. This function is then linked into the Fuzzer library.
  13 ///
  14 //===----------------------------------------------------------------------===//
  15 #include "llvm/DebugInfo/DIContext.h"
  16 #include "llvm/DebugInfo/DWARF/DWARFContext.h"
  17 #include "llvm/Object/ObjectFile.h"
  18 #include "llvm/Support/MemoryBuffer.h"
  19
  20 using namespace llvm;
  21 using namespace object;
  22
  23 extern "C" int LLVMFuzzerTestOneInput(uint8_t *data, size_t size) {
  24   std::unique_ptr<MemoryBuffer> Buff = MemoryBuffer::getMemBuffer(
  25       StringRef((const char *)data, size), "", false);
  26
  27   Expected<std::unique_ptr<ObjectFile>> ObjOrErr =
  28       ObjectFile::createObjectFile(Buff->getMemBufferRef());
  29   if (auto E = ObjOrErr.takeError()) {
  30     consumeError(std::move(E));
  31     return 0;
  32   }
  33   ObjectFile &Obj = *ObjOrErr.get();
  34   std::unique_ptr<DIContext> DICtx = DWARFContext::create(Obj);
  35
  36
  37   DIDumpOptions opts;
  38   opts.DumpType = DIDT_All;
  39   DICtx->dump(nulls(), opts);
  40   return 0;
  41 }