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
loadDataForPDB(PDB_ReaderType Type,StringRef Path,std::unique_ptr<IPDBSession> & Session)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
loadDataForEXE(PDB_ReaderType Type,StringRef Path,std::unique_ptr<IPDBSession> & Session)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