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    return NativeSession::createFromPdbPath(Path, Session);
28
29#if LLVM_ENABLE_DIA_SDK
30  return DIASession::createFromPdb(Path, Session);
31#else
32  return make_error<PDBError>(pdb_error_code::dia_sdk_not_present);
33#endif
34}
35
36Error llvm::pdb::loadDataForEXE(PDB_ReaderType Type, StringRef Path,
37                                std::unique_ptr<IPDBSession> &Session) {
38  // Create the correct concrete instance type based on the value of Type.
39  if (Type == PDB_ReaderType::Native) {
40    Expected<std::string> PdbPath = NativeSession::searchForPdb({Path});
41    if (!PdbPath)
42      return PdbPath.takeError();
43    return NativeSession::createFromPdbPath(PdbPath.get(), Session);
44  }
45
46#if LLVM_ENABLE_DIA_SDK
47  return DIASession::createFromExe(Path, Session);
48#else
49  return make_error<PDBError>(pdb_error_code::dia_sdk_not_present);
50#endif
51}
52