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), Age(0), 29 NamedStreams(NamedStreams) { 30 ::memset(&Guid, 0, sizeof(Guid)); 31 } 32 33 void InfoStreamBuilder::setVersion(PdbRaw_ImplVer V) { Ver = V; } 34 35 void InfoStreamBuilder::setAge(uint32_t A) { Age = A; } 36 37 void InfoStreamBuilder::setSignature(uint32_t S) { Signature = S; } 38 39 void InfoStreamBuilder::setGuid(GUID G) { Guid = G; } 40 41 void InfoStreamBuilder::addFeature(PdbRaw_FeatureSig Sig) { 42 Features.push_back(Sig); 43 } 44 45 Error InfoStreamBuilder::finalizeMsfLayout() { 46 uint32_t Length = sizeof(InfoStreamHeader) + 47 NamedStreams.calculateSerializedLength() + 48 (Features.size() + 1) * sizeof(uint32_t); 49 if (auto EC = Msf.setStreamSize(StreamPDB, Length)) 50 return EC; 51 return Error::success(); 52 } 53 54 Error InfoStreamBuilder::commit(const msf::MSFLayout &Layout, 55 WritableBinaryStreamRef Buffer) const { 56 auto InfoS = WritableMappedBlockStream::createIndexedStream( 57 Layout, Buffer, StreamPDB, Msf.getAllocator()); 58 BinaryStreamWriter Writer(*InfoS); 59 60 InfoStreamHeader H; 61 // Leave the build id fields 0 so they can be set as the last step before 62 // committing the file to disk. 63 ::memset(&H, 0, sizeof(H)); 64 H.Version = Ver; 65 if (auto EC = Writer.writeObject(H)) 66 return EC; 67 68 if (auto EC = NamedStreams.commit(Writer)) 69 return EC; 70 if (auto EC = Writer.writeInteger(0)) 71 return EC; 72 for (auto E : Features) { 73 if (auto EC = Writer.writeEnum(E)) 74 return EC; 75 } 76 assert(Writer.bytesRemaining() == 0); 77 return Error::success(); 78 } 79