1 //===- DIASession.h - DIA 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_DIA_DIASESSION_H
11 #define LLVM_DEBUGINFO_PDB_DIA_DIASESSION_H
12 
13 #include "DIASupport.h"
14 #include "llvm/DebugInfo/PDB/IPDBSession.h"
15 #include "llvm/Support/Error.h"
16 
17 #include <system_error>
18 
19 namespace llvm {
20 class StringRef;
21 
22 namespace pdb {
23 class DIASession : public IPDBSession {
24 public:
25   explicit DIASession(CComPtr<IDiaSession> DiaSession);
26 
27   static Error createFromPdb(StringRef Path,
28                              std::unique_ptr<IPDBSession> &Session);
29   static Error createFromExe(StringRef Path,
30                              std::unique_ptr<IPDBSession> &Session);
31 
32   uint64_t getLoadAddress() const override;
33   bool setLoadAddress(uint64_t Address) override;
34   std::unique_ptr<PDBSymbolExe> getGlobalScope() override;
35   std::unique_ptr<PDBSymbol> getSymbolById(SymIndexId SymbolId) const override;
36 
37   bool addressForVA(uint64_t VA, uint32_t &Section,
38                     uint32_t &Offset) const override;
39   bool addressForRVA(uint32_t RVA, uint32_t &Section,
40                      uint32_t &Offset) const override;
41 
42   std::unique_ptr<PDBSymbol>
43   findSymbolByAddress(uint64_t Address, PDB_SymType Type) const override;
44   std::unique_ptr<PDBSymbol> findSymbolByRVA(uint32_t RVA,
45                                              PDB_SymType Type) const override;
46   std::unique_ptr<PDBSymbol>
47   findSymbolBySectOffset(uint32_t Section, uint32_t Offset,
48                          PDB_SymType Type) const override;
49 
50   std::unique_ptr<IPDBEnumLineNumbers>
51   findLineNumbers(const PDBSymbolCompiland &Compiland,
52                   const IPDBSourceFile &File) const override;
53   std::unique_ptr<IPDBEnumLineNumbers>
54   findLineNumbersByAddress(uint64_t Address, uint32_t Length) const override;
55   std::unique_ptr<IPDBEnumLineNumbers>
56   findLineNumbersByRVA(uint32_t RVA, uint32_t Length) const override;
57   std::unique_ptr<IPDBEnumLineNumbers>
58   findLineNumbersBySectOffset(uint32_t Section, uint32_t Offset,
59                               uint32_t Length) const override;
60 
61   std::unique_ptr<IPDBEnumSourceFiles>
62   findSourceFiles(const PDBSymbolCompiland *Compiland, llvm::StringRef Pattern,
63                   PDB_NameSearchFlags Flags) const override;
64   std::unique_ptr<IPDBSourceFile>
65   findOneSourceFile(const PDBSymbolCompiland *Compiland,
66                     llvm::StringRef Pattern,
67                     PDB_NameSearchFlags Flags) const override;
68   std::unique_ptr<IPDBEnumChildren<PDBSymbolCompiland>>
69   findCompilandsForSourceFile(llvm::StringRef Pattern,
70                               PDB_NameSearchFlags Flags) const override;
71   std::unique_ptr<PDBSymbolCompiland>
72   findOneCompilandForSourceFile(llvm::StringRef Pattern,
73                                 PDB_NameSearchFlags Flags) const override;
74   std::unique_ptr<IPDBEnumSourceFiles> getAllSourceFiles() const override;
75   std::unique_ptr<IPDBEnumSourceFiles> getSourceFilesForCompiland(
76       const PDBSymbolCompiland &Compiland) const override;
77   std::unique_ptr<IPDBSourceFile>
78   getSourceFileById(uint32_t FileId) const override;
79 
80   std::unique_ptr<IPDBEnumDataStreams> getDebugStreams() const override;
81 
82   std::unique_ptr<IPDBEnumTables> getEnumTables() const override;
83 
84   std::unique_ptr<IPDBEnumInjectedSources> getInjectedSources() const override;
85 
86   std::unique_ptr<IPDBEnumSectionContribs> getSectionContribs() const override;
87 
88   std::unique_ptr<IPDBEnumFrameData> getFrameData() const override;
89 private:
90   CComPtr<IDiaSession> Session;
91 };
92 } // namespace pdb
93 } // namespace llvm
94 #endif
95