contrib/llvm/lib/DebugInfo/PDB/PDB.cpp

   1 //===- PDB.cpp - base header file for creating a PDB reader ---------------===//
   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 #include "llvm/DebugInfo/PDB/PDB.h"
  11 #include "llvm/ADT/StringRef.h"
  12 #include "llvm/Config/config.h"
  13 #include "llvm/DebugInfo/PDB/GenericError.h"
  14 #if LLVM_ENABLE_DIA_SDK
  15 #include "llvm/DebugInfo/PDB/DIA/DIASession.h"
  16 #endif
  17 #include "llvm/DebugInfo/PDB/Native/NativeSession.h"
  18 #include "llvm/Support/Error.h"
  19 #include "llvm/Support/MemoryBuffer.h"
  20
  21 using namespace llvm;
  22 using namespace llvm::pdb;
  23
  24 Error llvm::pdb::loadDataForPDB(PDB_ReaderType Type, StringRef Path,
  25                                 std::unique_ptr<IPDBSession> &Session) {
  26   // Create the correct concrete instance type based on the value of Type.
  27   if (Type == PDB_ReaderType::Native) {
  28     ErrorOr<std::unique_ptr<MemoryBuffer>> ErrorOrBuffer =
  29         MemoryBuffer::getFileOrSTDIN(Path, /*FileSize=*/-1,
  30                                      /*RequiresNullTerminator=*/false);
  31     if (!ErrorOrBuffer)
  32       return errorCodeToError(ErrorOrBuffer.getError());
  33
  34     return NativeSession::createFromPdb(std::move(*ErrorOrBuffer), Session);
  35   }
  36
  37 #if LLVM_ENABLE_DIA_SDK
  38   return DIASession::createFromPdb(Path, Session);
  39 #else
  40   return make_error<PDBError>(pdb_error_code::dia_sdk_not_present);
  41 #endif
  42 }
  43
  44 Error llvm::pdb::loadDataForEXE(PDB_ReaderType Type, StringRef Path,
  45                                 std::unique_ptr<IPDBSession> &Session) {
  46   // Create the correct concrete instance type based on the value of Type.
  47   if (Type == PDB_ReaderType::Native)
  48     return NativeSession::createFromExe(Path, Session);
  49
  50 #if LLVM_ENABLE_DIA_SDK
  51   return DIASession::createFromExe(Path, Session);
  52 #else
  53   return make_error<PDBError>(pdb_error_code::dia_sdk_not_present);
  54 #endif
  55 }