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
20using namespace llvm;
21using namespace llvm::pdb;
22
23Error 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
43Error 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