1 //===- InjectedSourceStream.h - PDB Headerblock Stream Access ---*- C++ -*-===//
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 #ifndef LLVM_DEBUGINFO_PDB_NATIVE_INJECTEDSOURCESTREAM_H
10 #define LLVM_DEBUGINFO_PDB_NATIVE_INJECTEDSOURCESTREAM_H
11 
12 #include "llvm/DebugInfo/MSF/MappedBlockStream.h"
13 #include "llvm/DebugInfo/PDB/Native/HashTable.h"
14 #include "llvm/Support/Error.h"
15 
16 namespace llvm {
17 namespace pdb {
18 struct SrcHeaderBlockEntry;
19 struct SrcHeaderBlockHeader;
20 class PDBStringTable;
21 
22 class InjectedSourceStream {
23 public:
24   InjectedSourceStream(std::unique_ptr<msf::MappedBlockStream> Stream);
25   Error reload(const PDBStringTable &Strings);
26 
27   using const_iterator = HashTable<SrcHeaderBlockEntry>::const_iterator;
28   const_iterator begin() const { return InjectedSourceTable.begin(); }
29   const_iterator end() const { return InjectedSourceTable.end(); }
30 
31   uint32_t size() const { return InjectedSourceTable.size(); }
32 
33 private:
34   std::unique_ptr<msf::MappedBlockStream> Stream;
35 
36   const SrcHeaderBlockHeader* Header;
37   HashTable<SrcHeaderBlockEntry> InjectedSourceTable;
38 };
39 }
40 } // namespace llvm
41 
42 #endif
43