Details | Last modification | View Log | RSS feed
Rev | Author | Line No. | Line |
---|---|---|---|
14 | pmbaty | 1 | //===- GSIStreamBuilder.h - PDB Publics/Globals Stream Creation -*- 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_GSISTREAMBUILDER_H |
||
10 | #define LLVM_DEBUGINFO_PDB_NATIVE_GSISTREAMBUILDER_H |
||
11 | |||
12 | #include "llvm/ADT/DenseSet.h" |
||
13 | #include "llvm/DebugInfo/CodeView/CVRecord.h" |
||
14 | #include "llvm/DebugInfo/CodeView/CodeView.h" |
||
15 | #include "llvm/DebugInfo/PDB/Native/GlobalsStream.h" |
||
16 | #include "llvm/DebugInfo/PDB/Native/RawConstants.h" |
||
17 | #include "llvm/Support/BinaryStreamRef.h" |
||
18 | #include "llvm/Support/Error.h" |
||
19 | |||
20 | namespace llvm { |
||
21 | namespace codeview { |
||
22 | class ConstantSym; |
||
23 | class DataSym; |
||
24 | class ProcRefSym; |
||
25 | } // namespace codeview |
||
26 | template <typename T> struct BinaryItemTraits; |
||
27 | |||
28 | template <> struct BinaryItemTraits<codeview::CVSymbol> { |
||
29 | static size_t length(const codeview::CVSymbol &Item) { |
||
30 | return Item.RecordData.size(); |
||
31 | } |
||
32 | static ArrayRef<uint8_t> bytes(const codeview::CVSymbol &Item) { |
||
33 | return Item.RecordData; |
||
34 | } |
||
35 | }; |
||
36 | |||
37 | namespace msf { |
||
38 | class MSFBuilder; |
||
39 | struct MSFLayout; |
||
40 | } // namespace msf |
||
41 | namespace pdb { |
||
42 | struct GSIHashStreamBuilder; |
||
43 | struct BulkPublic; |
||
44 | struct SymbolDenseMapInfo; |
||
45 | |||
46 | class GSIStreamBuilder { |
||
47 | |||
48 | public: |
||
49 | explicit GSIStreamBuilder(msf::MSFBuilder &Msf); |
||
50 | ~GSIStreamBuilder(); |
||
51 | |||
52 | GSIStreamBuilder(const GSIStreamBuilder &) = delete; |
||
53 | GSIStreamBuilder &operator=(const GSIStreamBuilder &) = delete; |
||
54 | |||
55 | Error finalizeMsfLayout(); |
||
56 | |||
57 | Error commit(const msf::MSFLayout &Layout, WritableBinaryStreamRef Buffer); |
||
58 | |||
59 | uint32_t getPublicsStreamIndex() const { return PublicsStreamIndex; } |
||
60 | uint32_t getGlobalsStreamIndex() const { return GlobalsStreamIndex; } |
||
61 | uint32_t getRecordStreamIndex() const { return RecordStreamIndex; } |
||
62 | |||
63 | // Add public symbols in bulk. |
||
64 | void addPublicSymbols(std::vector<BulkPublic> &&PublicsIn); |
||
65 | |||
66 | void addGlobalSymbol(const codeview::ProcRefSym &Sym); |
||
67 | void addGlobalSymbol(const codeview::DataSym &Sym); |
||
68 | void addGlobalSymbol(const codeview::ConstantSym &Sym); |
||
69 | |||
70 | // Add a pre-serialized global symbol record. The caller must ensure that the |
||
71 | // symbol data remains alive until the global stream is committed to disk. |
||
72 | void addGlobalSymbol(const codeview::CVSymbol &Sym); |
||
73 | |||
74 | private: |
||
75 | void finalizePublicBuckets(); |
||
76 | void finalizeGlobalBuckets(uint32_t RecordZeroOffset); |
||
77 | |||
78 | template <typename T> void serializeAndAddGlobal(const T &Symbol); |
||
79 | |||
80 | uint32_t calculatePublicsHashStreamSize() const; |
||
81 | uint32_t calculateGlobalsHashStreamSize() const; |
||
82 | Error commitSymbolRecordStream(WritableBinaryStreamRef Stream); |
||
83 | Error commitPublicsHashStream(WritableBinaryStreamRef Stream); |
||
84 | Error commitGlobalsHashStream(WritableBinaryStreamRef Stream); |
||
85 | |||
86 | uint32_t PublicsStreamIndex = kInvalidStreamIndex; |
||
87 | uint32_t GlobalsStreamIndex = kInvalidStreamIndex; |
||
88 | uint32_t RecordStreamIndex = kInvalidStreamIndex; |
||
89 | msf::MSFBuilder &Msf; |
||
90 | std::unique_ptr<GSIHashStreamBuilder> PSH; |
||
91 | std::unique_ptr<GSIHashStreamBuilder> GSH; |
||
92 | |||
93 | // List of all of the public records. These are stored unserialized so that we |
||
94 | // can defer copying the names until we are ready to commit the PDB. |
||
95 | std::vector<BulkPublic> Publics; |
||
96 | |||
97 | // List of all of the global records. |
||
98 | std::vector<codeview::CVSymbol> Globals; |
||
99 | |||
100 | // Hash table for deduplicating global typedef and constant records. Only used |
||
101 | // for globals. |
||
102 | llvm::DenseSet<codeview::CVSymbol, SymbolDenseMapInfo> GlobalsSeen; |
||
103 | }; |
||
104 | |||
105 | /// This struct is equivalent to codeview::PublicSym32, but it has been |
||
106 | /// optimized for size to speed up bulk serialization and sorting operations |
||
107 | /// during PDB writing. |
||
108 | struct BulkPublic { |
||
109 | BulkPublic() : Flags(0), BucketIdx(0) {} |
||
110 | |||
111 | const char *Name = nullptr; |
||
112 | uint32_t NameLen = 0; |
||
113 | |||
114 | // Offset of the symbol record in the publics stream. |
||
115 | uint32_t SymOffset = 0; |
||
116 | |||
117 | // Section offset of the symbol in the image. |
||
118 | uint32_t Offset = 0; |
||
119 | |||
120 | // Section index of the section containing the symbol. |
||
121 | uint16_t Segment = 0; |
||
122 | |||
123 | // PublicSymFlags. |
||
124 | uint16_t Flags : 4; |
||
125 | |||
126 | // GSI hash table bucket index. The maximum value is IPHR_HASH. |
||
127 | uint16_t BucketIdx : 12; |
||
128 | static_assert(IPHR_HASH <= 1 << 12, "bitfield too small"); |
||
129 | |||
130 | void setFlags(codeview::PublicSymFlags F) { |
||
131 | Flags = uint32_t(F); |
||
132 | assert(Flags == uint32_t(F) && "truncated"); |
||
133 | } |
||
134 | |||
135 | void setBucketIdx(uint16_t B) { |
||
136 | assert(B < IPHR_HASH); |
||
137 | BucketIdx = B; |
||
138 | } |
||
139 | |||
140 | StringRef getName() const { return StringRef(Name, NameLen); } |
||
141 | }; |
||
142 | |||
143 | static_assert(sizeof(BulkPublic) <= 24, "unexpected size increase"); |
||
144 | static_assert(std::is_trivially_copyable<BulkPublic>::value, |
||
145 | "should be trivial"); |
||
146 | |||
147 | } // namespace pdb |
||
148 | } // namespace llvm |
||
149 | |||
150 | #endif |