1 //===- DbiStreamBuilder.h - PDB Dbi 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_DBISTREAMBUILDER_H 10 #define LLVM_DEBUGINFO_PDB_NATIVE_DBISTREAMBUILDER_H 11 12 #include "llvm/ADT/StringMap.h" 13 #include "llvm/ADT/StringRef.h" 14 #include "llvm/BinaryFormat/COFF.h" 15 #include "llvm/Object/COFF.h" 16 #include "llvm/Support/Allocator.h" 17 #include "llvm/Support/Error.h" 18 19 #include "llvm/DebugInfo/CodeView/DebugFrameDataSubsection.h" 20 #include "llvm/DebugInfo/PDB/Native/PDBStringTableBuilder.h" 21 #include "llvm/DebugInfo/PDB/Native/RawConstants.h" 22 #include "llvm/DebugInfo/PDB/Native/RawTypes.h" 23 #include "llvm/DebugInfo/PDB/PDBTypes.h" 24 #include "llvm/Support/BinaryByteStream.h" 25 #include "llvm/Support/BinaryStreamRef.h" 26 27 namespace llvm { 28 29 class BinaryStreamWriter; 30 namespace codeview { 31 struct FrameData; 32 } 33 namespace msf { 34 class MSFBuilder; 35 struct MSFLayout; 36 } 37 namespace pdb { 38 class DbiModuleDescriptorBuilder; 39 40 class DbiStreamBuilder { 41 public: 42 DbiStreamBuilder(msf::MSFBuilder &Msf); 43 ~DbiStreamBuilder(); 44 45 DbiStreamBuilder(const DbiStreamBuilder &) = delete; 46 DbiStreamBuilder &operator=(const DbiStreamBuilder &) = delete; 47 48 void setVersionHeader(PdbRaw_DbiVer V); 49 void setAge(uint32_t A); 50 void setBuildNumber(uint16_t B); 51 void setBuildNumber(uint8_t Major, uint8_t Minor); 52 void setPdbDllVersion(uint16_t V); 53 void setPdbDllRbld(uint16_t R); 54 void setFlags(uint16_t F); 55 void setMachineType(PDB_Machine M); 56 void setMachineType(COFF::MachineTypes M); 57 58 // Add given bytes as a new stream. 59 Error addDbgStream(pdb::DbgHeaderType Type, ArrayRef<uint8_t> Data); 60 61 uint32_t addECName(StringRef Name); 62 63 uint32_t calculateSerializedLength() const; 64 65 void setGlobalsStreamIndex(uint32_t Index); 66 void setPublicsStreamIndex(uint32_t Index); 67 void setSymbolRecordStreamIndex(uint32_t Index); 68 void addNewFpoData(const codeview::FrameData &FD); 69 void addOldFpoData(const object::FpoData &Fpo); 70 71 Expected<DbiModuleDescriptorBuilder &> addModuleInfo(StringRef ModuleName); 72 Error addModuleSourceFile(DbiModuleDescriptorBuilder &Module, StringRef File); 73 Expected<uint32_t> getSourceFileNameIndex(StringRef FileName); 74 75 Error finalizeMsfLayout(); 76 77 Error commit(const msf::MSFLayout &Layout, WritableBinaryStreamRef MsfBuffer); 78 addSectionContrib(const SectionContrib & SC)79 void addSectionContrib(const SectionContrib &SC) { 80 SectionContribs.emplace_back(SC); 81 } 82 83 // Populate the Section Map from COFF section headers. 84 void createSectionMap(ArrayRef<llvm::object::coff_section> SecHdrs); 85 86 private: 87 struct DebugStream { 88 std::function<Error(BinaryStreamWriter &)> WriteFn; 89 uint32_t Size = 0; 90 uint16_t StreamNumber = kInvalidStreamIndex; 91 }; 92 93 Error finalize(); 94 uint32_t calculateModiSubstreamSize() const; 95 uint32_t calculateNamesOffset() const; 96 uint32_t calculateSectionContribsStreamSize() const; 97 uint32_t calculateSectionMapStreamSize() const; 98 uint32_t calculateFileInfoSubstreamSize() const; 99 uint32_t calculateNamesBufferSize() const; 100 uint32_t calculateDbgStreamsSize() const; 101 102 Error generateFileInfoSubstream(); 103 104 msf::MSFBuilder &Msf; 105 BumpPtrAllocator &Allocator; 106 107 std::optional<PdbRaw_DbiVer> VerHeader; 108 uint32_t Age; 109 uint16_t BuildNumber; 110 uint16_t PdbDllVersion; 111 uint16_t PdbDllRbld; 112 uint16_t Flags; 113 PDB_Machine MachineType; 114 uint32_t GlobalsStreamIndex = kInvalidStreamIndex; 115 uint32_t PublicsStreamIndex = kInvalidStreamIndex; 116 uint32_t SymRecordStreamIndex = kInvalidStreamIndex; 117 118 const DbiStreamHeader *Header; 119 120 std::vector<std::unique_ptr<DbiModuleDescriptorBuilder>> ModiList; 121 122 std::optional<codeview::DebugFrameDataSubsection> NewFpoData; 123 std::vector<object::FpoData> OldFpoData; 124 125 StringMap<uint32_t> SourceFileNames; 126 127 PDBStringTableBuilder ECNamesBuilder; 128 WritableBinaryStreamRef NamesBuffer; 129 MutableBinaryByteStream FileInfoBuffer; 130 std::vector<SectionContrib> SectionContribs; 131 std::vector<SecMapEntry> SectionMap; 132 std::array<std::optional<DebugStream>, (int)DbgHeaderType::Max> DbgStreams; 133 }; 134 } // namespace pdb 135 } 136 137 #endif 138