1 //===- PDB.cpp - base header file for creating a PDB reader ---------------===// 2 // 3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4 // See https://llvm.org/LICENSE.txt for license information. 5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6 // 7 //===----------------------------------------------------------------------===// 8 9 #include "llvm/DebugInfo/PDB/PDB.h" 10 #include "llvm/ADT/StringRef.h" 11 #include "llvm/Config/config.h" 12 #include "llvm/DebugInfo/PDB/GenericError.h" 13 #if LLVM_ENABLE_DIA_SDK 14 #include "llvm/DebugInfo/PDB/DIA/DIASession.h" 15 #endif 16 #include "llvm/DebugInfo/PDB/Native/NativeSession.h" 17 #include "llvm/Support/Error.h" 18 #include "llvm/Support/MemoryBuffer.h" 19 20 using namespace llvm; 21 using namespace llvm::pdb; 22 23 Error llvm::pdb::loadDataForPDB(PDB_ReaderType Type, StringRef Path, 24 std::unique_ptr<IPDBSession> &Session) { 25 // Create the correct concrete instance type based on the value of Type. 26 if (Type == PDB_ReaderType::Native) { 27 ErrorOr<std::unique_ptr<MemoryBuffer>> ErrorOrBuffer = 28 MemoryBuffer::getFileOrSTDIN(Path, /*FileSize=*/-1, 29 /*RequiresNullTerminator=*/false); 30 if (!ErrorOrBuffer) 31 return errorCodeToError(ErrorOrBuffer.getError()); 32 33 return NativeSession::createFromPdb(std::move(*ErrorOrBuffer), Session); 34 } 35 36 #if LLVM_ENABLE_DIA_SDK 37 return DIASession::createFromPdb(Path, Session); 38 #else 39 return make_error<PDBError>(pdb_error_code::dia_sdk_not_present); 40 #endif 41 } 42 43 Error llvm::pdb::loadDataForEXE(PDB_ReaderType Type, StringRef Path, 44 std::unique_ptr<IPDBSession> &Session) { 45 // Create the correct concrete instance type based on the value of Type. 46 if (Type == PDB_ReaderType::Native) 47 return NativeSession::createFromExe(Path, Session); 48 49 #if LLVM_ENABLE_DIA_SDK 50 return DIASession::createFromExe(Path, Session); 51 #else 52 return make_error<PDBError>(pdb_error_code::dia_sdk_not_present); 53 #endif 54 } 55