• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 //===- NativeSession.h - Native implementation of IPDBSession ---*- C++ -*-===//
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 #ifndef LLVM_DEBUGINFO_PDB_NATIVE_NATIVESESSION_H
11 #define LLVM_DEBUGINFO_PDB_NATIVE_NATIVESESSION_H
12 
13 #include "llvm/ADT/DenseMap.h"
14 #include "llvm/ADT/StringRef.h"
15 #include "llvm/DebugInfo/CodeView/TypeIndex.h"
16 #include "llvm/DebugInfo/PDB/IPDBRawSymbol.h"
17 #include "llvm/DebugInfo/PDB/IPDBSession.h"
18 #include "llvm/DebugInfo/PDB/Native/DbiModuleDescriptor.h"
19 #include "llvm/DebugInfo/PDB/Native/NativeBuiltinSymbol.h"
20 #include "llvm/DebugInfo/PDB/Native/NativeRawSymbol.h"
21 #include "llvm/Support/Allocator.h"
22 #include "llvm/Support/Error.h"
23 
24 namespace llvm {
25 class MemoryBuffer;
26 namespace pdb {
27 class PDBFile;
28 
29 class NativeSession : public IPDBSession {
30 public:
31   NativeSession(std::unique_ptr<PDBFile> PdbFile,
32                 std::unique_ptr<BumpPtrAllocator> Allocator);
33   ~NativeSession() override;
34 
35   static Error createFromPdb(std::unique_ptr<MemoryBuffer> MB,
36                              std::unique_ptr<IPDBSession> &Session);
37   static Error createFromExe(StringRef Path,
38                              std::unique_ptr<IPDBSession> &Session);
39 
40   std::unique_ptr<PDBSymbolCompiland>
41   createCompilandSymbol(DbiModuleDescriptor MI);
42 
43   std::unique_ptr<PDBSymbolTypeEnum>
44   createEnumSymbol(codeview::TypeIndex Index);
45 
46   std::unique_ptr<IPDBEnumSymbols>
47   createTypeEnumerator(codeview::TypeLeafKind Kind);
48 
49   SymIndexId findSymbolByTypeIndex(codeview::TypeIndex TI);
50 
51   uint64_t getLoadAddress() const override;
52   bool setLoadAddress(uint64_t Address) override;
53   std::unique_ptr<PDBSymbolExe> getGlobalScope() override;
54   std::unique_ptr<PDBSymbol> getSymbolById(uint32_t SymbolId) const override;
55 
56   bool addressForVA(uint64_t VA, uint32_t &Section,
57                     uint32_t &Offset) const override;
58   bool addressForRVA(uint32_t RVA, uint32_t &Section,
59                      uint32_t &Offset) const override;
60 
61   std::unique_ptr<PDBSymbol>
62   findSymbolByAddress(uint64_t Address, PDB_SymType Type) const override;
63   std::unique_ptr<PDBSymbol> findSymbolByRVA(uint32_t RVA,
64                                              PDB_SymType Type) const override;
65   std::unique_ptr<PDBSymbol>
66   findSymbolBySectOffset(uint32_t Sect, uint32_t Offset,
67                          PDB_SymType Type) const override;
68 
69   std::unique_ptr<IPDBEnumLineNumbers>
70   findLineNumbers(const PDBSymbolCompiland &Compiland,
71                   const IPDBSourceFile &File) const override;
72   std::unique_ptr<IPDBEnumLineNumbers>
73   findLineNumbersByAddress(uint64_t Address, uint32_t Length) const override;
74   std::unique_ptr<IPDBEnumLineNumbers>
75   findLineNumbersByRVA(uint32_t RVA, uint32_t Length) const override;
76   std::unique_ptr<IPDBEnumLineNumbers>
77   findLineNumbersBySectOffset(uint32_t Section, uint32_t Offset,
78                               uint32_t Length) const override;
79 
80   std::unique_ptr<IPDBEnumSourceFiles>
81   findSourceFiles(const PDBSymbolCompiland *Compiland, llvm::StringRef Pattern,
82                   PDB_NameSearchFlags Flags) const override;
83   std::unique_ptr<IPDBSourceFile>
84   findOneSourceFile(const PDBSymbolCompiland *Compiland,
85                     llvm::StringRef Pattern,
86                     PDB_NameSearchFlags Flags) const override;
87   std::unique_ptr<IPDBEnumChildren<PDBSymbolCompiland>>
88   findCompilandsForSourceFile(llvm::StringRef Pattern,
89                               PDB_NameSearchFlags Flags) const override;
90   std::unique_ptr<PDBSymbolCompiland>
91   findOneCompilandForSourceFile(llvm::StringRef Pattern,
92                                 PDB_NameSearchFlags Flags) const override;
93   std::unique_ptr<IPDBEnumSourceFiles> getAllSourceFiles() const override;
94   std::unique_ptr<IPDBEnumSourceFiles> getSourceFilesForCompiland(
95       const PDBSymbolCompiland &Compiland) const override;
96   std::unique_ptr<IPDBSourceFile>
97   getSourceFileById(uint32_t FileId) const override;
98 
99   std::unique_ptr<IPDBEnumDataStreams> getDebugStreams() const override;
100 
101   std::unique_ptr<IPDBEnumTables> getEnumTables() const override;
102 
103   std::unique_ptr<IPDBEnumInjectedSources> getInjectedSources() const override;
104 
105   std::unique_ptr<IPDBEnumSectionContribs> getSectionContribs() const override;
106 
getPDBFile()107   PDBFile &getPDBFile() { return *Pdb; }
getPDBFile()108   const PDBFile &getPDBFile() const { return *Pdb; }
109 
110 private:
111   std::unique_ptr<PDBFile> Pdb;
112   std::unique_ptr<BumpPtrAllocator> Allocator;
113   std::vector<std::unique_ptr<NativeRawSymbol>> SymbolCache;
114   DenseMap<codeview::TypeIndex, SymIndexId> TypeIndexToSymbolId;
115 };
116 } // namespace pdb
117 } // namespace llvm
118 
119 #endif
120