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 
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 make_error<GenericError>(generic_error_code::invalid_path, Path);
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<GenericError>("DIA is not installed on the system");
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<GenericError>("DIA is not installed on the system");
53 #endif
54 }
55