]> CyberLeo.Net >> Repos - FreeBSD/FreeBSD.git/blob - contrib/llvm/lib/DebugInfo/CodeView/SimpleTypeSerializer.cpp
Fix a memory leak in if_delgroups() introduced in r334118.
[FreeBSD/FreeBSD.git] / contrib / llvm / lib / DebugInfo / CodeView / SimpleTypeSerializer.cpp
1 #include "llvm/DebugInfo/CodeView/SimpleTypeSerializer.h"
2
3 using namespace llvm;
4 using namespace llvm::codeview;
5
6 static void addPadding(BinaryStreamWriter &Writer) {
7   uint32_t Align = Writer.getOffset() % 4;
8   if (Align == 0)
9     return;
10
11   int PaddingBytes = 4 - Align;
12   while (PaddingBytes > 0) {
13     uint8_t Pad = static_cast<uint8_t>(LF_PAD0 + PaddingBytes);
14     cantFail(Writer.writeInteger(Pad));
15     --PaddingBytes;
16   }
17 }
18
19 SimpleTypeSerializer::SimpleTypeSerializer() : ScratchBuffer(MaxRecordLength) {}
20
21 SimpleTypeSerializer::~SimpleTypeSerializer() {}
22
23 template <typename T>
24 ArrayRef<uint8_t> SimpleTypeSerializer::serialize(T &Record) {
25   BinaryStreamWriter Writer(ScratchBuffer, support::little);
26   TypeRecordMapping Mapping(Writer);
27
28   // Write the record prefix first with a dummy length but real kind.
29   RecordPrefix DummyPrefix(uint16_t(Record.getKind()));
30   cantFail(Writer.writeObject(DummyPrefix));
31
32   RecordPrefix *Prefix = reinterpret_cast<RecordPrefix *>(ScratchBuffer.data());
33   CVType CVT(Prefix, sizeof(RecordPrefix));
34
35   cantFail(Mapping.visitTypeBegin(CVT));
36   cantFail(Mapping.visitKnownRecord(CVT, Record));
37   cantFail(Mapping.visitTypeEnd(CVT));
38
39   addPadding(Writer);
40
41   // Update the size and kind after serialization.
42   Prefix->RecordKind = CVT.kind();
43   Prefix->RecordLen = Writer.getOffset() - sizeof(uint16_t);
44
45   return {ScratchBuffer.data(), Writer.getOffset()};
46 }
47
48 // Explicitly instantiate the member function for each known type so that we can
49 // implement this in the cpp file.
50 #define TYPE_RECORD(EnumName, EnumVal, Name)                                   \
51   template ArrayRef<uint8_t> llvm::codeview::SimpleTypeSerializer::serialize(  \
52       Name##Record &Record);
53 #define TYPE_RECORD_ALIAS(EnumName, EnumVal, Name, AliasName)
54 #define MEMBER_RECORD(EnumName, EnumVal, Name)
55 #define MEMBER_RECORD_ALIAS(EnumName, EnumVal, Name, AliasName)
56 #include "llvm/DebugInfo/CodeView/CodeViewTypes.def"