lib/ObjectYAML/CodeViewYAMLSymbols.cpp

   1 //===- CodeViewYAMLSymbols.cpp - CodeView YAMLIO Symbol implementation ----===//
   2 //
   3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
   4 // See https://llvm.org/LICENSE.txt for license information.
   5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
   6 //
   7 //===----------------------------------------------------------------------===//
   8 //
   9 // This file defines classes for handling the YAML representation of CodeView
  10 // Debug Info.
  11 //
  12 //===----------------------------------------------------------------------===//
  13
  14 #include "llvm/ObjectYAML/CodeViewYAMLSymbols.h"
  15 #include "llvm/ADT/ArrayRef.h"
  16 #include "llvm/ADT/StringRef.h"
  17 #include "llvm/DebugInfo/CodeView/CodeView.h"
  18 #include "llvm/DebugInfo/CodeView/CodeViewError.h"
  19 #include "llvm/DebugInfo/CodeView/EnumTables.h"
  20 #include "llvm/DebugInfo/CodeView/RecordSerialization.h"
  21 #include "llvm/DebugInfo/CodeView/SymbolDeserializer.h"
  22 #include "llvm/DebugInfo/CodeView/SymbolRecord.h"
  23 #include "llvm/DebugInfo/CodeView/SymbolSerializer.h"
  24 #include "llvm/DebugInfo/CodeView/TypeIndex.h"
  25 #include "llvm/ObjectYAML/YAML.h"
  26 #include "llvm/Support/Allocator.h"
  27 #include "llvm/Support/Error.h"
  28 #include "llvm/Support/YAMLTraits.h"
  29 #include <algorithm>
  30 #include <cstdint>
  31 #include <cstring>
  32 #include <string>
  33 #include <vector>
  34
  35 using namespace llvm;
  36 using namespace llvm::codeview;
  37 using namespace llvm::CodeViewYAML;
  38 using namespace llvm::CodeViewYAML::detail;
  39 using namespace llvm::yaml;
  40
  41 LLVM_YAML_IS_FLOW_SEQUENCE_VECTOR(TypeIndex)
  42 LLVM_YAML_IS_SEQUENCE_VECTOR(LocalVariableAddrGap)
  43
  44 // We only need to declare these, the definitions are in CodeViewYAMLTypes.cpp
  45 LLVM_YAML_DECLARE_SCALAR_TRAITS(APSInt, QuotingType::None)
  46 LLVM_YAML_DECLARE_SCALAR_TRAITS(TypeIndex, QuotingType::None)
  47
  48 LLVM_YAML_DECLARE_ENUM_TRAITS(SymbolKind)
  49 LLVM_YAML_DECLARE_ENUM_TRAITS(FrameCookieKind)
  50
  51 LLVM_YAML_DECLARE_BITSET_TRAITS(CompileSym2Flags)
  52 LLVM_YAML_DECLARE_BITSET_TRAITS(CompileSym3Flags)
  53 LLVM_YAML_DECLARE_BITSET_TRAITS(ExportFlags)
  54 LLVM_YAML_DECLARE_BITSET_TRAITS(PublicSymFlags)
  55 LLVM_YAML_DECLARE_BITSET_TRAITS(LocalSymFlags)
  56 LLVM_YAML_DECLARE_BITSET_TRAITS(ProcSymFlags)
  57 LLVM_YAML_DECLARE_BITSET_TRAITS(FrameProcedureOptions)
  58 LLVM_YAML_DECLARE_ENUM_TRAITS(CPUType)
  59 LLVM_YAML_DECLARE_ENUM_TRAITS(RegisterId)
  60 LLVM_YAML_DECLARE_ENUM_TRAITS(TrampolineType)
  61 LLVM_YAML_DECLARE_ENUM_TRAITS(ThunkOrdinal)
  62
  63 LLVM_YAML_STRONG_TYPEDEF(StringRef, TypeName)
  64
  65 LLVM_YAML_DECLARE_SCALAR_TRAITS(TypeName, QuotingType::Single)
  66
  67 StringRef ScalarTraits<TypeName>::input(StringRef S, void *V, TypeName &T) {
  68   return ScalarTraits<StringRef>::input(S, V, T.value);
  69 }
  70
  71 void ScalarTraits<TypeName>::output(const TypeName &T, void *V,
  72                                     raw_ostream &R) {
  73   ScalarTraits<StringRef>::output(T.value, V, R);
  74 }
  75
  76 void ScalarEnumerationTraits<SymbolKind>::enumeration(IO &io,
  77                                                       SymbolKind &Value) {
  78   auto SymbolNames = getSymbolTypeNames();
  79   for (const auto &E : SymbolNames)
  80     io.enumCase(Value, E.Name.str().c_str(), E.Value);
  81 }
  82
  83 void ScalarBitSetTraits<CompileSym2Flags>::bitset(IO &io,
  84                                                   CompileSym2Flags &Flags) {
  85   auto FlagNames = getCompileSym2FlagNames();
  86   for (const auto &E : FlagNames) {
  87     io.bitSetCase(Flags, E.Name.str().c_str(),
  88                   static_cast<CompileSym2Flags>(E.Value));
  89   }
  90 }
  91
  92 void ScalarBitSetTraits<CompileSym3Flags>::bitset(IO &io,
  93                                                   CompileSym3Flags &Flags) {
  94   auto FlagNames = getCompileSym3FlagNames();
  95   for (const auto &E : FlagNames) {
  96     io.bitSetCase(Flags, E.Name.str().c_str(),
  97                   static_cast<CompileSym3Flags>(E.Value));
  98   }
  99 }
 100
 101 void ScalarBitSetTraits<ExportFlags>::bitset(IO &io, ExportFlags &Flags) {
 102   auto FlagNames = getExportSymFlagNames();
 103   for (const auto &E : FlagNames) {
 104     io.bitSetCase(Flags, E.Name.str().c_str(),
 105                   static_cast<ExportFlags>(E.Value));
 106   }
 107 }
 108
 109 void ScalarBitSetTraits<PublicSymFlags>::bitset(IO &io, PublicSymFlags &Flags) {
 110   auto FlagNames = getPublicSymFlagNames();
 111   for (const auto &E : FlagNames) {
 112     io.bitSetCase(Flags, E.Name.str().c_str(),
 113                   static_cast<PublicSymFlags>(E.Value));
 114   }
 115 }
 116
 117 void ScalarBitSetTraits<LocalSymFlags>::bitset(IO &io, LocalSymFlags &Flags) {
 118   auto FlagNames = getLocalFlagNames();
 119   for (const auto &E : FlagNames) {
 120     io.bitSetCase(Flags, E.Name.str().c_str(),
 121                   static_cast<LocalSymFlags>(E.Value));
 122   }
 123 }
 124
 125 void ScalarBitSetTraits<ProcSymFlags>::bitset(IO &io, ProcSymFlags &Flags) {
 126   auto FlagNames = getProcSymFlagNames();
 127   for (const auto &E : FlagNames) {
 128     io.bitSetCase(Flags, E.Name.str().c_str(),
 129                   static_cast<ProcSymFlags>(E.Value));
 130   }
 131 }
 132
 133 void ScalarBitSetTraits<FrameProcedureOptions>::bitset(
 134     IO &io, FrameProcedureOptions &Flags) {
 135   auto FlagNames = getFrameProcSymFlagNames();
 136   for (const auto &E : FlagNames) {
 137     io.bitSetCase(Flags, E.Name.str().c_str(),
 138                   static_cast<FrameProcedureOptions>(E.Value));
 139   }
 140 }
 141
 142 void ScalarEnumerationTraits<CPUType>::enumeration(IO &io, CPUType &Cpu) {
 143   auto CpuNames = getCPUTypeNames();
 144   for (const auto &E : CpuNames) {
 145     io.enumCase(Cpu, E.Name.str().c_str(), static_cast<CPUType>(E.Value));
 146   }
 147 }
 148
 149 void ScalarEnumerationTraits<RegisterId>::enumeration(IO &io, RegisterId &Reg) {
 150   auto RegNames = getRegisterNames(CPUType::X64);
 151   for (const auto &E : RegNames) {
 152     io.enumCase(Reg, E.Name.str().c_str(), static_cast<RegisterId>(E.Value));
 153   }
 154   io.enumFallback<Hex16>(Reg);
 155 }
 156
 157 void ScalarEnumerationTraits<TrampolineType>::enumeration(
 158     IO &io, TrampolineType &Tramp) {
 159   auto TrampNames = getTrampolineNames();
 160   for (const auto &E : TrampNames) {
 161     io.enumCase(Tramp, E.Name.str().c_str(),
 162                 static_cast<TrampolineType>(E.Value));
 163   }
 164 }
 165
 166 void ScalarEnumerationTraits<ThunkOrdinal>::enumeration(IO &io,
 167                                                         ThunkOrdinal &Ord) {
 168   auto ThunkNames = getThunkOrdinalNames();
 169   for (const auto &E : ThunkNames) {
 170     io.enumCase(Ord, E.Name.str().c_str(), static_cast<ThunkOrdinal>(E.Value));
 171   }
 172 }
 173
 174 void ScalarEnumerationTraits<FrameCookieKind>::enumeration(
 175     IO &io, FrameCookieKind &FC) {
 176   auto ThunkNames = getFrameCookieKindNames();
 177   for (const auto &E : ThunkNames) {
 178     io.enumCase(FC, E.Name.str().c_str(),
 179                 static_cast<FrameCookieKind>(E.Value));
 180   }
 181 }
 182
 183 namespace llvm {
 184 namespace yaml {
 185 template <> struct MappingTraits<LocalVariableAddrRange> {
 186   static void mapping(IO &io, LocalVariableAddrRange &Range) {
 187     io.mapRequired("OffsetStart", Range.OffsetStart);
 188     io.mapRequired("ISectStart", Range.ISectStart);
 189     io.mapRequired("Range", Range.Range);
 190   }
 191 };
 192 template <> struct MappingTraits<LocalVariableAddrGap> {
 193   static void mapping(IO &io, LocalVariableAddrGap &Gap) {
 194     io.mapRequired("GapStartOffset", Gap.GapStartOffset);
 195     io.mapRequired("Range", Gap.Range);
 196   }
 197 };
 198 } // namespace yaml
 199 } // namespace llvm
 200
 201 namespace llvm {
 202 namespace CodeViewYAML {
 203 namespace detail {
 204
 205 struct SymbolRecordBase {
 206   codeview::SymbolKind Kind;
 207
 208   explicit SymbolRecordBase(codeview::SymbolKind K) : Kind(K) {}
 209   virtual ~SymbolRecordBase() = default;
 210
 211   virtual void map(yaml::IO &io) = 0;
 212   virtual codeview::CVSymbol
 213   toCodeViewSymbol(BumpPtrAllocator &Allocator,
 214                    CodeViewContainer Container) const = 0;
 215   virtual Error fromCodeViewSymbol(codeview::CVSymbol Type) = 0;
 216 };
 217
 218 template <typename T> struct SymbolRecordImpl : public SymbolRecordBase {
 219   explicit SymbolRecordImpl(codeview::SymbolKind K)
 220       : SymbolRecordBase(K), Symbol(static_cast<SymbolRecordKind>(K)) {}
 221
 222   void map(yaml::IO &io) override;
 223
 224   codeview::CVSymbol
 225   toCodeViewSymbol(BumpPtrAllocator &Allocator,
 226                    CodeViewContainer Container) const override {
 227     return SymbolSerializer::writeOneSymbol(Symbol, Allocator, Container);
 228   }
 229
 230   Error fromCodeViewSymbol(codeview::CVSymbol CVS) override {
 231     return SymbolDeserializer::deserializeAs<T>(CVS, Symbol);
 232   }
 233
 234   mutable T Symbol;
 235 };
 236
 237 struct UnknownSymbolRecord : public SymbolRecordBase {
 238   explicit UnknownSymbolRecord(codeview::SymbolKind K) : SymbolRecordBase(K) {}
 239
 240   void map(yaml::IO &io) override;
 241
 242   CVSymbol toCodeViewSymbol(BumpPtrAllocator &Allocator,
 243                             CodeViewContainer Container) const override {
 244     RecordPrefix Prefix;
 245     uint32_t TotalLen = sizeof(RecordPrefix) + Data.size();
 246     Prefix.RecordKind = Kind;
 247     Prefix.RecordLen = TotalLen - 2;
 248     uint8_t *Buffer = Allocator.Allocate<uint8_t>(TotalLen);
 249     ::memcpy(Buffer, &Prefix, sizeof(RecordPrefix));
 250     ::memcpy(Buffer + sizeof(RecordPrefix), Data.data(), Data.size());
 251     return CVSymbol(ArrayRef<uint8_t>(Buffer, TotalLen));
 252   }
 253
 254   Error fromCodeViewSymbol(CVSymbol CVS) override {
 255     this->Kind = CVS.kind();
 256     Data = CVS.RecordData.drop_front(sizeof(RecordPrefix));
 257     return Error::success();
 258   }
 259
 260   std::vector<uint8_t> Data;
 261 };
 262
 263 template <> void SymbolRecordImpl<ScopeEndSym>::map(IO &IO) {}
 264
 265 void UnknownSymbolRecord::map(yaml::IO &io) {
 266   yaml::BinaryRef Binary;
 267   if (io.outputting())
 268     Binary = yaml::BinaryRef(Data);
 269   io.mapRequired("Data", Binary);
 270   if (!io.outputting()) {
 271     std::string Str;
 272     raw_string_ostream OS(Str);
 273     Binary.writeAsBinary(OS);
 274     OS.flush();
 275     Data.assign(Str.begin(), Str.end());
 276   }
 277 }
 278
 279 template <> void SymbolRecordImpl<Thunk32Sym>::map(IO &IO) {
 280   IO.mapRequired("Parent", Symbol.Parent);
 281   IO.mapRequired("End", Symbol.End);
 282   IO.mapRequired("Next", Symbol.Next);
 283   IO.mapRequired("Off", Symbol.Offset);
 284   IO.mapRequired("Seg", Symbol.Segment);
 285   IO.mapRequired("Len", Symbol.Length);
 286   IO.mapRequired("Ordinal", Symbol.Thunk);
 287 }
 288
 289 template <> void SymbolRecordImpl<TrampolineSym>::map(IO &IO) {
 290   IO.mapRequired("Type", Symbol.Type);
 291   IO.mapRequired("Size", Symbol.Size);
 292   IO.mapRequired("ThunkOff", Symbol.ThunkOffset);
 293   IO.mapRequired("TargetOff", Symbol.TargetOffset);
 294   IO.mapRequired("ThunkSection", Symbol.ThunkSection);
 295   IO.mapRequired("TargetSection", Symbol.TargetSection);
 296 }
 297
 298 template <> void SymbolRecordImpl<SectionSym>::map(IO &IO) {
 299   IO.mapRequired("SectionNumber", Symbol.SectionNumber);
 300   IO.mapRequired("Alignment", Symbol.Alignment);
 301   IO.mapRequired("Rva", Symbol.Rva);
 302   IO.mapRequired("Length", Symbol.Length);
 303   IO.mapRequired("Characteristics", Symbol.Characteristics);
 304   IO.mapRequired("Name", Symbol.Name);
 305 }
 306
 307 template <> void SymbolRecordImpl<CoffGroupSym>::map(IO &IO) {
 308   IO.mapRequired("Size", Symbol.Size);
 309   IO.mapRequired("Characteristics", Symbol.Characteristics);
 310   IO.mapRequired("Offset", Symbol.Offset);
 311   IO.mapRequired("Segment", Symbol.Segment);
 312   IO.mapRequired("Name", Symbol.Name);
 313 }
 314
 315 template <> void SymbolRecordImpl<ExportSym>::map(IO &IO) {
 316   IO.mapRequired("Ordinal", Symbol.Ordinal);
 317   IO.mapRequired("Flags", Symbol.Flags);
 318   IO.mapRequired("Name", Symbol.Name);
 319 }
 320
 321 template <> void SymbolRecordImpl<ProcSym>::map(IO &IO) {
 322   IO.mapOptional("PtrParent", Symbol.Parent, 0U);
 323   IO.mapOptional("PtrEnd", Symbol.End, 0U);
 324   IO.mapOptional("PtrNext", Symbol.Next, 0U);
 325   IO.mapRequired("CodeSize", Symbol.CodeSize);
 326   IO.mapRequired("DbgStart", Symbol.DbgStart);
 327   IO.mapRequired("DbgEnd", Symbol.DbgEnd);
 328   IO.mapRequired("FunctionType", Symbol.FunctionType);
 329   IO.mapOptional("Offset", Symbol.CodeOffset, 0U);
 330   IO.mapOptional("Segment", Symbol.Segment, uint16_t(0));
 331   IO.mapRequired("Flags", Symbol.Flags);
 332   IO.mapRequired("DisplayName", Symbol.Name);
 333 }
 334
 335 template <> void SymbolRecordImpl<RegisterSym>::map(IO &IO) {
 336   IO.mapRequired("Type", Symbol.Index);
 337   IO.mapRequired("Seg", Symbol.Register);
 338   IO.mapRequired("Name", Symbol.Name);
 339 }
 340
 341 template <> void SymbolRecordImpl<PublicSym32>::map(IO &IO) {
 342   IO.mapRequired("Flags", Symbol.Flags);
 343   IO.mapOptional("Offset", Symbol.Offset, 0U);
 344   IO.mapOptional("Segment", Symbol.Segment, uint16_t(0));
 345   IO.mapRequired("Name", Symbol.Name);
 346 }
 347
 348 template <> void SymbolRecordImpl<ProcRefSym>::map(IO &IO) {
 349   IO.mapRequired("SumName", Symbol.SumName);
 350   IO.mapRequired("SymOffset", Symbol.SymOffset);
 351   IO.mapRequired("Mod", Symbol.Module);
 352   IO.mapRequired("Name", Symbol.Name);
 353 }
 354
 355 template <> void SymbolRecordImpl<EnvBlockSym>::map(IO &IO) {
 356   IO.mapRequired("Entries", Symbol.Fields);
 357 }
 358
 359 template <> void SymbolRecordImpl<InlineSiteSym>::map(IO &IO) {
 360   IO.mapOptional("PtrParent", Symbol.Parent, 0U);
 361   IO.mapOptional("PtrEnd", Symbol.End, 0U);
 362   IO.mapRequired("Inlinee", Symbol.Inlinee);
 363   // TODO: The binary annotations
 364 }
 365
 366 template <> void SymbolRecordImpl<LocalSym>::map(IO &IO) {
 367   IO.mapRequired("Type", Symbol.Type);
 368   IO.mapRequired("Flags", Symbol.Flags);
 369
 370   IO.mapRequired("VarName", Symbol.Name);
 371 }
 372
 373 template <> void SymbolRecordImpl<DefRangeSym>::map(IO &IO) {
 374   IO.mapRequired("Program", Symbol.Program);
 375   IO.mapRequired("Range", Symbol.Range);
 376   IO.mapRequired("Gaps", Symbol.Gaps);
 377 }
 378
 379 template <> void SymbolRecordImpl<DefRangeSubfieldSym>::map(IO &IO) {
 380   IO.mapRequired("Program", Symbol.Program);
 381   IO.mapRequired("OffsetInParent", Symbol.OffsetInParent);
 382   IO.mapRequired("Range", Symbol.Range);
 383   IO.mapRequired("Gaps", Symbol.Gaps);
 384 }
 385
 386 template <> void SymbolRecordImpl<DefRangeRegisterSym>::map(IO &IO) {
 387   IO.mapRequired("Register", Symbol.Hdr.Register);
 388   IO.mapRequired("MayHaveNoName", Symbol.Hdr.MayHaveNoName);
 389   IO.mapRequired("Range", Symbol.Range);
 390   IO.mapRequired("Gaps", Symbol.Gaps);
 391 }
 392
 393 template <> void SymbolRecordImpl<DefRangeFramePointerRelSym>::map(IO &IO) {
 394   IO.mapRequired("Offset", Symbol.Offset);
 395   IO.mapRequired("Range", Symbol.Range);
 396   IO.mapRequired("Gaps", Symbol.Gaps);
 397 }
 398
 399 template <> void SymbolRecordImpl<DefRangeSubfieldRegisterSym>::map(IO &IO) {
 400   IO.mapRequired("Register", Symbol.Hdr.Register);
 401   IO.mapRequired("MayHaveNoName", Symbol.Hdr.MayHaveNoName);
 402   IO.mapRequired("OffsetInParent", Symbol.Hdr.OffsetInParent);
 403   IO.mapRequired("Range", Symbol.Range);
 404   IO.mapRequired("Gaps", Symbol.Gaps);
 405 }
 406
 407 template <>
 408 void SymbolRecordImpl<DefRangeFramePointerRelFullScopeSym>::map(IO &IO) {
 409   IO.mapRequired("Register", Symbol.Offset);
 410 }
 411
 412 template <> void SymbolRecordImpl<DefRangeRegisterRelSym>::map(IO &IO) {
 413   IO.mapRequired("Register", Symbol.Hdr.Register);
 414   IO.mapRequired("Flags", Symbol.Hdr.Flags);
 415   IO.mapRequired("BasePointerOffset", Symbol.Hdr.BasePointerOffset);
 416   IO.mapRequired("Range", Symbol.Range);
 417   IO.mapRequired("Gaps", Symbol.Gaps);
 418 }
 419
 420 template <> void SymbolRecordImpl<BlockSym>::map(IO &IO) {
 421   IO.mapOptional("PtrParent", Symbol.Parent, 0U);
 422   IO.mapOptional("PtrEnd", Symbol.End, 0U);
 423   IO.mapRequired("CodeSize", Symbol.CodeSize);
 424   IO.mapOptional("Offset", Symbol.CodeOffset, 0U);
 425   IO.mapOptional("Segment", Symbol.Segment, uint16_t(0));
 426   IO.mapRequired("BlockName", Symbol.Name);
 427 }
 428
 429 template <> void SymbolRecordImpl<LabelSym>::map(IO &IO) {
 430   IO.mapOptional("Offset", Symbol.CodeOffset, 0U);
 431   IO.mapOptional("Segment", Symbol.Segment, uint16_t(0));
 432   IO.mapRequired("Flags", Symbol.Flags);
 433   IO.mapRequired("Flags", Symbol.Flags);
 434   IO.mapRequired("DisplayName", Symbol.Name);
 435 }
 436
 437 template <> void SymbolRecordImpl<ObjNameSym>::map(IO &IO) {
 438   IO.mapRequired("Signature", Symbol.Signature);
 439   IO.mapRequired("ObjectName", Symbol.Name);
 440 }
 441
 442 template <> void SymbolRecordImpl<Compile2Sym>::map(IO &IO) {
 443   IO.mapRequired("Flags", Symbol.Flags);
 444   IO.mapRequired("Machine", Symbol.Machine);
 445   IO.mapRequired("FrontendMajor", Symbol.VersionFrontendMajor);
 446   IO.mapRequired("FrontendMinor", Symbol.VersionFrontendMinor);
 447   IO.mapRequired("FrontendBuild", Symbol.VersionFrontendBuild);
 448   IO.mapRequired("BackendMajor", Symbol.VersionBackendMajor);
 449   IO.mapRequired("BackendMinor", Symbol.VersionBackendMinor);
 450   IO.mapRequired("BackendBuild", Symbol.VersionBackendBuild);
 451   IO.mapRequired("Version", Symbol.Version);
 452 }
 453
 454 template <> void SymbolRecordImpl<Compile3Sym>::map(IO &IO) {
 455   IO.mapRequired("Flags", Symbol.Flags);
 456   IO.mapRequired("Machine", Symbol.Machine);
 457   IO.mapRequired("FrontendMajor", Symbol.VersionFrontendMajor);
 458   IO.mapRequired("FrontendMinor", Symbol.VersionFrontendMinor);
 459   IO.mapRequired("FrontendBuild", Symbol.VersionFrontendBuild);
 460   IO.mapRequired("FrontendQFE", Symbol.VersionFrontendQFE);
 461   IO.mapRequired("BackendMajor", Symbol.VersionBackendMajor);
 462   IO.mapRequired("BackendMinor", Symbol.VersionBackendMinor);
 463   IO.mapRequired("BackendBuild", Symbol.VersionBackendBuild);
 464   IO.mapRequired("BackendQFE", Symbol.VersionBackendQFE);
 465   IO.mapRequired("Version", Symbol.Version);
 466 }
 467
 468 template <> void SymbolRecordImpl<FrameProcSym>::map(IO &IO) {
 469   IO.mapRequired("TotalFrameBytes", Symbol.TotalFrameBytes);
 470   IO.mapRequired("PaddingFrameBytes", Symbol.PaddingFrameBytes);
 471   IO.mapRequired("OffsetToPadding", Symbol.OffsetToPadding);
 472   IO.mapRequired("BytesOfCalleeSavedRegisters",
 473                  Symbol.BytesOfCalleeSavedRegisters);
 474   IO.mapRequired("OffsetOfExceptionHandler", Symbol.OffsetOfExceptionHandler);
 475   IO.mapRequired("SectionIdOfExceptionHandler",
 476                  Symbol.SectionIdOfExceptionHandler);
 477   IO.mapRequired("Flags", Symbol.Flags);
 478 }
 479
 480 template <> void SymbolRecordImpl<CallSiteInfoSym>::map(IO &IO) {
 481   IO.mapOptional("Offset", Symbol.CodeOffset, 0U);
 482   IO.mapOptional("Segment", Symbol.Segment, uint16_t(0));
 483   IO.mapRequired("Type", Symbol.Type);
 484 }
 485
 486 template <> void SymbolRecordImpl<FileStaticSym>::map(IO &IO) {
 487   IO.mapRequired("Index", Symbol.Index);
 488   IO.mapRequired("ModFilenameOffset", Symbol.ModFilenameOffset);
 489   IO.mapRequired("Flags", Symbol.Flags);
 490   IO.mapRequired("Name", Symbol.Name);
 491 }
 492
 493 template <> void SymbolRecordImpl<HeapAllocationSiteSym>::map(IO &IO) {
 494   IO.mapOptional("Offset", Symbol.CodeOffset, 0U);
 495   IO.mapOptional("Segment", Symbol.Segment, uint16_t(0));
 496   IO.mapRequired("CallInstructionSize", Symbol.CallInstructionSize);
 497   IO.mapRequired("Type", Symbol.Type);
 498 }
 499
 500 template <> void SymbolRecordImpl<FrameCookieSym>::map(IO &IO) {
 501   IO.mapRequired("Register", Symbol.Register);
 502   IO.mapRequired("CookieKind", Symbol.CookieKind);
 503   IO.mapRequired("Flags", Symbol.Flags);
 504 }
 505
 506 template <> void SymbolRecordImpl<CallerSym>::map(IO &IO) {
 507   IO.mapRequired("FuncID", Symbol.Indices);
 508 }
 509
 510 template <> void SymbolRecordImpl<UDTSym>::map(IO &IO) {
 511   IO.mapRequired("Type", Symbol.Type);
 512   IO.mapRequired("UDTName", Symbol.Name);
 513 }
 514
 515 template <> void SymbolRecordImpl<BuildInfoSym>::map(IO &IO) {
 516   IO.mapRequired("BuildId", Symbol.BuildId);
 517 }
 518
 519 template <> void SymbolRecordImpl<BPRelativeSym>::map(IO &IO) {
 520   IO.mapRequired("Offset", Symbol.Offset);
 521   IO.mapRequired("Type", Symbol.Type);
 522   IO.mapRequired("VarName", Symbol.Name);
 523 }
 524
 525 template <> void SymbolRecordImpl<RegRelativeSym>::map(IO &IO) {
 526   IO.mapRequired("Offset", Symbol.Offset);
 527   IO.mapRequired("Type", Symbol.Type);
 528   IO.mapRequired("Register", Symbol.Register);
 529   IO.mapRequired("VarName", Symbol.Name);
 530 }
 531
 532 template <> void SymbolRecordImpl<ConstantSym>::map(IO &IO) {
 533   IO.mapRequired("Type", Symbol.Type);
 534   IO.mapRequired("Value", Symbol.Value);
 535   IO.mapRequired("Name", Symbol.Name);
 536 }
 537
 538 template <> void SymbolRecordImpl<DataSym>::map(IO &IO) {
 539   IO.mapRequired("Type", Symbol.Type);
 540   IO.mapOptional("Offset", Symbol.DataOffset, 0U);
 541   IO.mapOptional("Segment", Symbol.Segment, uint16_t(0));
 542   IO.mapRequired("DisplayName", Symbol.Name);
 543 }
 544
 545 template <> void SymbolRecordImpl<ThreadLocalDataSym>::map(IO &IO) {
 546   IO.mapRequired("Type", Symbol.Type);
 547   IO.mapOptional("Offset", Symbol.DataOffset, 0U);
 548   IO.mapOptional("Segment", Symbol.Segment, uint16_t(0));
 549   IO.mapRequired("DisplayName", Symbol.Name);
 550 }
 551
 552 template <> void SymbolRecordImpl<UsingNamespaceSym>::map(IO &IO) {
 553   IO.mapRequired("Namespace", Symbol.Name);
 554 }
 555
 556 template <> void SymbolRecordImpl<AnnotationSym>::map(IO &IO) {
 557   IO.mapOptional("Offset", Symbol.CodeOffset, 0U);
 558   IO.mapOptional("Segment", Symbol.Segment, uint16_t(0));
 559   IO.mapRequired("Strings", Symbol.Strings);
 560 }
 561
 562 } // end namespace detail
 563 } // end namespace CodeViewYAML
 564 } // end namespace llvm
 565
 566 CVSymbol CodeViewYAML::SymbolRecord::toCodeViewSymbol(
 567     BumpPtrAllocator &Allocator, CodeViewContainer Container) const {
 568   return Symbol->toCodeViewSymbol(Allocator, Container);
 569 }
 570
 571 namespace llvm {
 572 namespace yaml {
 573
 574 template <> struct MappingTraits<SymbolRecordBase> {
 575   static void mapping(IO &io, SymbolRecordBase &Record) { Record.map(io); }
 576 };
 577
 578 } // end namespace yaml
 579 } // end namespace llvm
 580
 581 template <typename SymbolType>
 582 static inline Expected<CodeViewYAML::SymbolRecord>
 583 fromCodeViewSymbolImpl(CVSymbol Symbol) {
 584   CodeViewYAML::SymbolRecord Result;
 585
 586   auto Impl = std::make_shared<SymbolType>(Symbol.kind());
 587   if (auto EC = Impl->fromCodeViewSymbol(Symbol))
 588     return std::move(EC);
 589   Result.Symbol = Impl;
 590   return Result;
 591 }
 592
 593 Expected<CodeViewYAML::SymbolRecord>
 594 CodeViewYAML::SymbolRecord::fromCodeViewSymbol(CVSymbol Symbol) {
 595 #define SYMBOL_RECORD(EnumName, EnumVal, ClassName)                            \
 596   case EnumName:                                                               \
 597     return fromCodeViewSymbolImpl<SymbolRecordImpl<ClassName>>(Symbol);
 598 #define SYMBOL_RECORD_ALIAS(EnumName, EnumVal, AliasName, ClassName)           \
 599   SYMBOL_RECORD(EnumName, EnumVal, ClassName)
 600   switch (Symbol.kind()) {
 601 #include "llvm/DebugInfo/CodeView/CodeViewSymbols.def"
 602   default:
 603     return fromCodeViewSymbolImpl<UnknownSymbolRecord>(Symbol);
 604   }
 605   return make_error<CodeViewError>(cv_error_code::corrupt_record);
 606 }
 607
 608 template <typename ConcreteType>
 609 static void mapSymbolRecordImpl(IO &IO, const char *Class, SymbolKind Kind,
 610                                 CodeViewYAML::SymbolRecord &Obj) {
 611   if (!IO.outputting())
 612     Obj.Symbol = std::make_shared<ConcreteType>(Kind);
 613
 614   IO.mapRequired(Class, *Obj.Symbol);
 615 }
 616
 617 void MappingTraits<CodeViewYAML::SymbolRecord>::mapping(
 618     IO &IO, CodeViewYAML::SymbolRecord &Obj) {
 619   SymbolKind Kind;
 620   if (IO.outputting())
 621     Kind = Obj.Symbol->Kind;
 622   IO.mapRequired("Kind", Kind);
 623
 624 #define SYMBOL_RECORD(EnumName, EnumVal, ClassName)                            \
 625   case EnumName:                                                               \
 626     mapSymbolRecordImpl<SymbolRecordImpl<ClassName>>(IO, #ClassName, Kind,     \
 627                                                      Obj);                     \
 628     break;
 629 #define SYMBOL_RECORD_ALIAS(EnumName, EnumVal, AliasName, ClassName)           \
 630   SYMBOL_RECORD(EnumName, EnumVal, ClassName)
 631   switch (Kind) {
 632 #include "llvm/DebugInfo/CodeView/CodeViewSymbols.def"
 633   default:
 634     mapSymbolRecordImpl<UnknownSymbolRecord>(IO, "UnknownSym", Kind, Obj);
 635   }
 636 }