1*9880d681SAndroid Build Coastguard Worker //===- DbiStreamBuilder.cpp - PDB Dbi Stream Creation -----------*- C++ -*-===//
2*9880d681SAndroid Build Coastguard Worker //
3*9880d681SAndroid Build Coastguard Worker // The LLVM Compiler Infrastructure
4*9880d681SAndroid Build Coastguard Worker //
5*9880d681SAndroid Build Coastguard Worker // This file is distributed under the University of Illinois Open Source
6*9880d681SAndroid Build Coastguard Worker // License. See LICENSE.TXT for details.
7*9880d681SAndroid Build Coastguard Worker //
8*9880d681SAndroid Build Coastguard Worker //===----------------------------------------------------------------------===//
9*9880d681SAndroid Build Coastguard Worker
10*9880d681SAndroid Build Coastguard Worker #include "llvm/DebugInfo/PDB/Raw/DbiStreamBuilder.h"
11*9880d681SAndroid Build Coastguard Worker
12*9880d681SAndroid Build Coastguard Worker #include "llvm/DebugInfo/CodeView/StreamWriter.h"
13*9880d681SAndroid Build Coastguard Worker #include "llvm/DebugInfo/PDB/Raw/DbiStream.h"
14*9880d681SAndroid Build Coastguard Worker #include "llvm/DebugInfo/PDB/Raw/MappedBlockStream.h"
15*9880d681SAndroid Build Coastguard Worker #include "llvm/DebugInfo/PDB/Raw/RawError.h"
16*9880d681SAndroid Build Coastguard Worker
17*9880d681SAndroid Build Coastguard Worker using namespace llvm;
18*9880d681SAndroid Build Coastguard Worker using namespace llvm::codeview;
19*9880d681SAndroid Build Coastguard Worker using namespace llvm::pdb;
20*9880d681SAndroid Build Coastguard Worker
DbiStreamBuilder(PDBFile & File)21*9880d681SAndroid Build Coastguard Worker DbiStreamBuilder::DbiStreamBuilder(PDBFile &File)
22*9880d681SAndroid Build Coastguard Worker : File(File), Age(1), BuildNumber(0), PdbDllVersion(0), PdbDllRbld(0),
23*9880d681SAndroid Build Coastguard Worker Flags(0), MachineType(PDB_Machine::x86) {}
24*9880d681SAndroid Build Coastguard Worker
setVersionHeader(PdbRaw_DbiVer V)25*9880d681SAndroid Build Coastguard Worker void DbiStreamBuilder::setVersionHeader(PdbRaw_DbiVer V) { VerHeader = V; }
26*9880d681SAndroid Build Coastguard Worker
setAge(uint32_t A)27*9880d681SAndroid Build Coastguard Worker void DbiStreamBuilder::setAge(uint32_t A) { Age = A; }
28*9880d681SAndroid Build Coastguard Worker
setBuildNumber(uint16_t B)29*9880d681SAndroid Build Coastguard Worker void DbiStreamBuilder::setBuildNumber(uint16_t B) { BuildNumber = B; }
30*9880d681SAndroid Build Coastguard Worker
setPdbDllVersion(uint16_t V)31*9880d681SAndroid Build Coastguard Worker void DbiStreamBuilder::setPdbDllVersion(uint16_t V) { PdbDllVersion = V; }
32*9880d681SAndroid Build Coastguard Worker
setPdbDllRbld(uint16_t R)33*9880d681SAndroid Build Coastguard Worker void DbiStreamBuilder::setPdbDllRbld(uint16_t R) { PdbDllRbld = R; }
34*9880d681SAndroid Build Coastguard Worker
setFlags(uint16_t F)35*9880d681SAndroid Build Coastguard Worker void DbiStreamBuilder::setFlags(uint16_t F) { Flags = F; }
36*9880d681SAndroid Build Coastguard Worker
setMachineType(PDB_Machine M)37*9880d681SAndroid Build Coastguard Worker void DbiStreamBuilder::setMachineType(PDB_Machine M) { MachineType = M; }
38*9880d681SAndroid Build Coastguard Worker
build()39*9880d681SAndroid Build Coastguard Worker Expected<std::unique_ptr<DbiStream>> DbiStreamBuilder::build() {
40*9880d681SAndroid Build Coastguard Worker if (!VerHeader.hasValue())
41*9880d681SAndroid Build Coastguard Worker return make_error<RawError>(raw_error_code::unspecified,
42*9880d681SAndroid Build Coastguard Worker "Missing DBI Stream Version");
43*9880d681SAndroid Build Coastguard Worker
44*9880d681SAndroid Build Coastguard Worker auto DbiS = MappedBlockStream::createIndexedStream(StreamDBI, File);
45*9880d681SAndroid Build Coastguard Worker if (!DbiS)
46*9880d681SAndroid Build Coastguard Worker return DbiS.takeError();
47*9880d681SAndroid Build Coastguard Worker auto DS = std::move(*DbiS);
48*9880d681SAndroid Build Coastguard Worker DbiStream::HeaderInfo *H =
49*9880d681SAndroid Build Coastguard Worker static_cast<DbiStream::HeaderInfo *>(DS->getAllocator().Allocate(
50*9880d681SAndroid Build Coastguard Worker sizeof(DbiStream::HeaderInfo),
51*9880d681SAndroid Build Coastguard Worker llvm::AlignOf<DbiStream::HeaderInfo>::Alignment));
52*9880d681SAndroid Build Coastguard Worker H->VersionHeader = *VerHeader;
53*9880d681SAndroid Build Coastguard Worker H->VersionSignature = -1;
54*9880d681SAndroid Build Coastguard Worker H->Age = Age;
55*9880d681SAndroid Build Coastguard Worker H->BuildNumber = BuildNumber;
56*9880d681SAndroid Build Coastguard Worker H->Flags = Flags;
57*9880d681SAndroid Build Coastguard Worker H->PdbDllRbld = PdbDllRbld;
58*9880d681SAndroid Build Coastguard Worker H->PdbDllVersion = PdbDllVersion;
59*9880d681SAndroid Build Coastguard Worker H->MachineType = static_cast<uint16_t>(MachineType);
60*9880d681SAndroid Build Coastguard Worker
61*9880d681SAndroid Build Coastguard Worker H->ECSubstreamSize = 0;
62*9880d681SAndroid Build Coastguard Worker H->FileInfoSize = 0;
63*9880d681SAndroid Build Coastguard Worker H->ModiSubstreamSize = 0;
64*9880d681SAndroid Build Coastguard Worker H->OptionalDbgHdrSize = 0;
65*9880d681SAndroid Build Coastguard Worker H->SecContrSubstreamSize = 0;
66*9880d681SAndroid Build Coastguard Worker H->SectionMapSize = 0;
67*9880d681SAndroid Build Coastguard Worker H->TypeServerSize = 0;
68*9880d681SAndroid Build Coastguard Worker H->SymRecordStreamIndex = DbiStream::InvalidStreamIndex;
69*9880d681SAndroid Build Coastguard Worker H->PublicSymbolStreamIndex = DbiStream::InvalidStreamIndex;
70*9880d681SAndroid Build Coastguard Worker H->MFCTypeServerIndex = DbiStream::InvalidStreamIndex;
71*9880d681SAndroid Build Coastguard Worker H->GlobalSymbolStreamIndex = DbiStream::InvalidStreamIndex;
72*9880d681SAndroid Build Coastguard Worker
73*9880d681SAndroid Build Coastguard Worker auto Dbi = llvm::make_unique<DbiStream>(File, std::move(DS));
74*9880d681SAndroid Build Coastguard Worker Dbi->Header = H;
75*9880d681SAndroid Build Coastguard Worker return std::move(Dbi);
76*9880d681SAndroid Build Coastguard Worker }
77