lib/DebugInfo/CodeView/SymbolSerializer.cpp

   1 //===- SymbolSerializer.cpp -----------------------------------------------===//
   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/CodeView/SymbolSerializer.h"
  11 #include "llvm/ADT/ArrayRef.h"
  12 #include "llvm/DebugInfo/CodeView/SymbolRecord.h"
  13 #include "llvm/Support/Endian.h"
  14 #include "llvm/Support/Error.h"
  15 #include <cassert>
  16 #include <cstdint>
  17 #include <cstring>
  18
  19 using namespace llvm;
  20 using namespace llvm::codeview;
  21
  22 SymbolSerializer::SymbolSerializer(BumpPtrAllocator &Allocator,
  23                                    CodeViewContainer Container)
  24     : Storage(Allocator), Stream(RecordBuffer, support::little), Writer(Stream),
  25       Mapping(Writer, Container) {}
  26
  27 Error SymbolSerializer::visitSymbolBegin(CVSymbol &Record) {
  28   assert(!CurrentSymbol.hasValue() && "Already in a symbol mapping!");
  29
  30   Writer.setOffset(0);
  31
  32   if (auto EC = writeRecordPrefix(Record.kind()))
  33     return EC;
  34
  35   CurrentSymbol = Record.kind();
  36   if (auto EC = Mapping.visitSymbolBegin(Record))
  37     return EC;
  38
  39   return Error::success();
  40 }
  41
  42 Error SymbolSerializer::visitSymbolEnd(CVSymbol &Record) {
  43   assert(CurrentSymbol.hasValue() && "Not in a symbol mapping!");
  44
  45   if (auto EC = Mapping.visitSymbolEnd(Record))
  46     return EC;
  47
  48   uint32_t RecordEnd = Writer.getOffset();
  49   uint16_t Length = RecordEnd - 2;
  50   Writer.setOffset(0);
  51   if (auto EC = Writer.writeInteger(Length))
  52     return EC;
  53
  54   uint8_t *StableStorage = Storage.Allocate<uint8_t>(RecordEnd);
  55   ::memcpy(StableStorage, &RecordBuffer[0], RecordEnd);
  56   Record.RecordData = ArrayRef<uint8_t>(StableStorage, RecordEnd);
  57   CurrentSymbol.reset();
  58
  59   return Error::success();
  60 }