aboutsummaryrefslogtreecommitdiffstats
path: root/contrib/libs/llvm12/lib/DebugInfo/PDB/Native/InfoStreamBuilder.cpp
blob: 02844c4fa57871a401b190e4b65659044bd99f04 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
//===- InfoStreamBuilder.cpp - PDB Info Stream Creation ---------*- C++ -*-===// 
// 
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 
// See https://llvm.org/LICENSE.txt for license information. 
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 
// 
//===----------------------------------------------------------------------===// 
 
#include "llvm/DebugInfo/PDB/Native/InfoStreamBuilder.h" 
 
#include "llvm/DebugInfo/MSF/MSFBuilder.h" 
#include "llvm/DebugInfo/MSF/MappedBlockStream.h" 
#include "llvm/DebugInfo/PDB/Native/InfoStream.h" 
#include "llvm/DebugInfo/PDB/Native/NamedStreamMap.h" 
#include "llvm/DebugInfo/PDB/Native/PDBFileBuilder.h" 
#include "llvm/DebugInfo/PDB/Native/RawError.h" 
#include "llvm/DebugInfo/PDB/Native/RawTypes.h" 
#include "llvm/Support/BinaryStreamWriter.h" 
 
using namespace llvm; 
using namespace llvm::codeview; 
using namespace llvm::msf; 
using namespace llvm::pdb; 
 
InfoStreamBuilder::InfoStreamBuilder(msf::MSFBuilder &Msf, 
                                     NamedStreamMap &NamedStreams) 
    : Msf(Msf), Ver(PdbRaw_ImplVer::PdbImplVC70), Age(0), 
      NamedStreams(NamedStreams) { 
  ::memset(&Guid, 0, sizeof(Guid)); 
} 
 
void InfoStreamBuilder::setVersion(PdbRaw_ImplVer V) { Ver = V; } 
 
void InfoStreamBuilder::addFeature(PdbRaw_FeatureSig Sig) { 
  Features.push_back(Sig); 
} 
 
void InfoStreamBuilder::setHashPDBContentsToGUID(bool B) { 
  HashPDBContentsToGUID = B; 
} 
 
void InfoStreamBuilder::setAge(uint32_t A) { Age = A; } 
 
void InfoStreamBuilder::setSignature(uint32_t S) { Signature = S; } 
 
void InfoStreamBuilder::setGuid(GUID G) { Guid = G; } 
 
 
Error InfoStreamBuilder::finalizeMsfLayout() { 
  uint32_t Length = sizeof(InfoStreamHeader) + 
                    NamedStreams.calculateSerializedLength() + 
                    (Features.size() + 1) * sizeof(uint32_t); 
  if (auto EC = Msf.setStreamSize(StreamPDB, Length)) 
    return EC; 
  return Error::success(); 
} 
 
Error InfoStreamBuilder::commit(const msf::MSFLayout &Layout, 
                                WritableBinaryStreamRef Buffer) const { 
  auto InfoS = WritableMappedBlockStream::createIndexedStream( 
      Layout, Buffer, StreamPDB, Msf.getAllocator()); 
  BinaryStreamWriter Writer(*InfoS); 
 
  InfoStreamHeader H; 
  // Leave the build id fields 0 so they can be set as the last step before 
  // committing the file to disk. 
  ::memset(&H, 0, sizeof(H)); 
  H.Version = Ver; 
  if (auto EC = Writer.writeObject(H)) 
    return EC; 
 
  if (auto EC = NamedStreams.commit(Writer)) 
    return EC; 
  if (auto EC = Writer.writeInteger(0)) 
    return EC; 
  for (auto E : Features) { 
    if (auto EC = Writer.writeEnum(E)) 
      return EC; 
  } 
  assert(Writer.bytesRemaining() == 0); 
  return Error::success(); 
}