xref: /llvm-project/llvm/lib/DebugInfo/PDB/PDB.cpp (revision bd159d32c43732a65a943333d04f46e5d30fb9c5)
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 make_error<GenericError>(generic_error_code::invalid_path, Path);
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<GenericError>("DIA is not installed on the system");
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<GenericError>("DIA is not installed on the system");
54 #endif
55 }
56