1 //===- InfoStreamBuilder.cpp - PDB Info Stream Creation ---------*- 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 #include "llvm/DebugInfo/PDB/Native/InfoStreamBuilder.h" 11 12 #include "llvm/DebugInfo/MSF/MSFBuilder.h" 13 #include "llvm/DebugInfo/MSF/MappedBlockStream.h" 14 #include "llvm/DebugInfo/PDB/Native/InfoStream.h" 15 #include "llvm/DebugInfo/PDB/Native/NamedStreamMap.h" 16 #include "llvm/DebugInfo/PDB/Native/PDBFileBuilder.h" 17 #include "llvm/DebugInfo/PDB/Native/RawError.h" 18 #include "llvm/DebugInfo/PDB/Native/RawTypes.h" 19 #include "llvm/Support/BinaryStreamWriter.h" 20 21 using namespace llvm; 22 using namespace llvm::codeview; 23 using namespace llvm::msf; 24 using namespace llvm::pdb; 25 26 InfoStreamBuilder::InfoStreamBuilder(msf::MSFBuilder &Msf, 27 NamedStreamMap &NamedStreams) 28 : Msf(Msf), Ver(PdbRaw_ImplVer::PdbImplVC70), Sig(-1), Age(0), 29 NamedStreams(NamedStreams) {} 30 31 void InfoStreamBuilder::setVersion(PdbRaw_ImplVer V) { Ver = V; } 32 33 void InfoStreamBuilder::setSignature(uint32_t S) { Sig = S; } 34 35 void InfoStreamBuilder::setAge(uint32_t A) { Age = A; } 36 37 void InfoStreamBuilder::setGuid(GUID G) { Guid = G; } 38 39 void InfoStreamBuilder::addFeature(PdbRaw_FeatureSig Sig) { 40 Features.push_back(Sig); 41 } 42 43 Error InfoStreamBuilder::finalizeMsfLayout() { 44 uint32_t Length = sizeof(InfoStreamHeader) + NamedStreams.finalize() + 45 (Features.size() + 1) * sizeof(uint32_t); 46 if (auto EC = Msf.setStreamSize(StreamPDB, Length)) 47 return EC; 48 return Error::success(); 49 } 50 51 Error InfoStreamBuilder::commit(const msf::MSFLayout &Layout, 52 WritableBinaryStreamRef Buffer) const { 53 auto InfoS = WritableMappedBlockStream::createIndexedStream( 54 Layout, Buffer, StreamPDB, Msf.getAllocator()); 55 BinaryStreamWriter Writer(*InfoS); 56 57 InfoStreamHeader H; 58 H.Age = Age; 59 H.Signature = Sig; 60 H.Version = Ver; 61 H.Guid = Guid; 62 if (auto EC = Writer.writeObject(H)) 63 return EC; 64 65 if (auto EC = NamedStreams.commit(Writer)) 66 return EC; 67 if (auto EC = Writer.writeInteger(0)) 68 return EC; 69 for (auto E : Features) { 70 if (auto EC = Writer.writeEnum(E)) 71 return EC; 72 } 73 return Error::success(); 74 } 75