lib/Object/ELFYAML.cpp

   1 //===- ELFYAML.cpp - ELF YAMLIO implementation ----------------------------===//
   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 // This file defines classes for handling the YAML representation of ELF.
  11 //
  12 //===----------------------------------------------------------------------===//
  13
  14 #include "llvm/Object/ELFYAML.h"
  15 #include "llvm/Support/Casting.h"
  16
  17 namespace llvm {
  18
  19 ELFYAML::Section::~Section() {}
  20
  21 namespace yaml {
  22
  23 void
  24 ScalarEnumerationTraits<ELFYAML::ELF_ET>::enumeration(IO &IO,
  25                                                       ELFYAML::ELF_ET &Value) {
  26 #define ECase(X) IO.enumCase(Value, #X, ELF::X);
  27   ECase(ET_NONE)
  28   ECase(ET_REL)
  29   ECase(ET_EXEC)
  30   ECase(ET_DYN)
  31   ECase(ET_CORE)
  32 #undef ECase
  33   IO.enumFallback<Hex16>(Value);
  34 }
  35
  36 void
  37 ScalarEnumerationTraits<ELFYAML::ELF_EM>::enumeration(IO &IO,
  38                                                       ELFYAML::ELF_EM &Value) {
  39 #define ECase(X) IO.enumCase(Value, #X, ELF::X);
  40   ECase(EM_NONE)
  41   ECase(EM_M32)
  42   ECase(EM_SPARC)
  43   ECase(EM_386)
  44   ECase(EM_68K)
  45   ECase(EM_88K)
  46   ECase(EM_486)
  47   ECase(EM_860)
  48   ECase(EM_MIPS)
  49   ECase(EM_S370)
  50   ECase(EM_MIPS_RS3_LE)
  51   ECase(EM_PARISC)
  52   ECase(EM_VPP500)
  53   ECase(EM_SPARC32PLUS)
  54   ECase(EM_960)
  55   ECase(EM_PPC)
  56   ECase(EM_PPC64)
  57   ECase(EM_S390)
  58   ECase(EM_SPU)
  59   ECase(EM_V800)
  60   ECase(EM_FR20)
  61   ECase(EM_RH32)
  62   ECase(EM_RCE)
  63   ECase(EM_ARM)
  64   ECase(EM_ALPHA)
  65   ECase(EM_SH)
  66   ECase(EM_SPARCV9)
  67   ECase(EM_TRICORE)
  68   ECase(EM_ARC)
  69   ECase(EM_H8_300)
  70   ECase(EM_H8_300H)
  71   ECase(EM_H8S)
  72   ECase(EM_H8_500)
  73   ECase(EM_IA_64)
  74   ECase(EM_MIPS_X)
  75   ECase(EM_COLDFIRE)
  76   ECase(EM_68HC12)
  77   ECase(EM_MMA)
  78   ECase(EM_PCP)
  79   ECase(EM_NCPU)
  80   ECase(EM_NDR1)
  81   ECase(EM_STARCORE)
  82   ECase(EM_ME16)
  83   ECase(EM_ST100)
  84   ECase(EM_TINYJ)
  85   ECase(EM_X86_64)
  86   ECase(EM_PDSP)
  87   ECase(EM_PDP10)
  88   ECase(EM_PDP11)
  89   ECase(EM_FX66)
  90   ECase(EM_ST9PLUS)
  91   ECase(EM_ST7)
  92   ECase(EM_68HC16)
  93   ECase(EM_68HC11)
  94   ECase(EM_68HC08)
  95   ECase(EM_68HC05)
  96   ECase(EM_SVX)
  97   ECase(EM_ST19)
  98   ECase(EM_VAX)
  99   ECase(EM_CRIS)
 100   ECase(EM_JAVELIN)
 101   ECase(EM_FIREPATH)
 102   ECase(EM_ZSP)
 103   ECase(EM_MMIX)
 104   ECase(EM_HUANY)
 105   ECase(EM_PRISM)
 106   ECase(EM_AVR)
 107   ECase(EM_FR30)
 108   ECase(EM_D10V)
 109   ECase(EM_D30V)
 110   ECase(EM_V850)
 111   ECase(EM_M32R)
 112   ECase(EM_MN10300)
 113   ECase(EM_MN10200)
 114   ECase(EM_PJ)
 115   ECase(EM_OPENRISC)
 116   ECase(EM_ARC_COMPACT)
 117   ECase(EM_XTENSA)
 118   ECase(EM_VIDEOCORE)
 119   ECase(EM_TMM_GPP)
 120   ECase(EM_NS32K)
 121   ECase(EM_TPC)
 122   ECase(EM_SNP1K)
 123   ECase(EM_ST200)
 124   ECase(EM_IP2K)
 125   ECase(EM_MAX)
 126   ECase(EM_CR)
 127   ECase(EM_F2MC16)
 128   ECase(EM_MSP430)
 129   ECase(EM_BLACKFIN)
 130   ECase(EM_SE_C33)
 131   ECase(EM_SEP)
 132   ECase(EM_ARCA)
 133   ECase(EM_UNICORE)
 134   ECase(EM_EXCESS)
 135   ECase(EM_DXP)
 136   ECase(EM_ALTERA_NIOS2)
 137   ECase(EM_CRX)
 138   ECase(EM_XGATE)
 139   ECase(EM_C166)
 140   ECase(EM_M16C)
 141   ECase(EM_DSPIC30F)
 142   ECase(EM_CE)
 143   ECase(EM_M32C)
 144   ECase(EM_TSK3000)
 145   ECase(EM_RS08)
 146   ECase(EM_SHARC)
 147   ECase(EM_ECOG2)
 148   ECase(EM_SCORE7)
 149   ECase(EM_DSP24)
 150   ECase(EM_VIDEOCORE3)
 151   ECase(EM_LATTICEMICO32)
 152   ECase(EM_SE_C17)
 153   ECase(EM_TI_C6000)
 154   ECase(EM_TI_C2000)
 155   ECase(EM_TI_C5500)
 156   ECase(EM_MMDSP_PLUS)
 157   ECase(EM_CYPRESS_M8C)
 158   ECase(EM_R32C)
 159   ECase(EM_TRIMEDIA)
 160   ECase(EM_HEXAGON)
 161   ECase(EM_8051)
 162   ECase(EM_STXP7X)
 163   ECase(EM_NDS32)
 164   ECase(EM_ECOG1)
 165   ECase(EM_ECOG1X)
 166   ECase(EM_MAXQ30)
 167   ECase(EM_XIMO16)
 168   ECase(EM_MANIK)
 169   ECase(EM_CRAYNV2)
 170   ECase(EM_RX)
 171   ECase(EM_METAG)
 172   ECase(EM_MCST_ELBRUS)
 173   ECase(EM_ECOG16)
 174   ECase(EM_CR16)
 175   ECase(EM_ETPU)
 176   ECase(EM_SLE9X)
 177   ECase(EM_L10M)
 178   ECase(EM_K10M)
 179   ECase(EM_AARCH64)
 180   ECase(EM_AVR32)
 181   ECase(EM_STM8)
 182   ECase(EM_TILE64)
 183   ECase(EM_TILEPRO)
 184   ECase(EM_CUDA)
 185   ECase(EM_TILEGX)
 186   ECase(EM_CLOUDSHIELD)
 187   ECase(EM_COREA_1ST)
 188   ECase(EM_COREA_2ND)
 189   ECase(EM_ARC_COMPACT2)
 190   ECase(EM_OPEN8)
 191   ECase(EM_RL78)
 192   ECase(EM_VIDEOCORE5)
 193   ECase(EM_78KOR)
 194   ECase(EM_56800EX)
 195 #undef ECase
 196 }
 197
 198 void ScalarEnumerationTraits<ELFYAML::ELF_ELFCLASS>::enumeration(
 199     IO &IO, ELFYAML::ELF_ELFCLASS &Value) {
 200 #define ECase(X) IO.enumCase(Value, #X, ELF::X);
 201   // Since the semantics of ELFCLASSNONE is "invalid", just don't accept it
 202   // here.
 203   ECase(ELFCLASS32)
 204   ECase(ELFCLASS64)
 205 #undef ECase
 206 }
 207
 208 void ScalarEnumerationTraits<ELFYAML::ELF_ELFDATA>::enumeration(
 209     IO &IO, ELFYAML::ELF_ELFDATA &Value) {
 210 #define ECase(X) IO.enumCase(Value, #X, ELF::X);
 211   // Since the semantics of ELFDATANONE is "invalid", just don't accept it
 212   // here.
 213   ECase(ELFDATA2LSB)
 214   ECase(ELFDATA2MSB)
 215 #undef ECase
 216 }
 217
 218 void ScalarEnumerationTraits<ELFYAML::ELF_ELFOSABI>::enumeration(
 219     IO &IO, ELFYAML::ELF_ELFOSABI &Value) {
 220 #define ECase(X) IO.enumCase(Value, #X, ELF::X);
 221   ECase(ELFOSABI_NONE)
 222   ECase(ELFOSABI_HPUX)
 223   ECase(ELFOSABI_NETBSD)
 224   ECase(ELFOSABI_GNU)
 225   ECase(ELFOSABI_GNU)
 226   ECase(ELFOSABI_HURD)
 227   ECase(ELFOSABI_SOLARIS)
 228   ECase(ELFOSABI_AIX)
 229   ECase(ELFOSABI_IRIX)
 230   ECase(ELFOSABI_FREEBSD)
 231   ECase(ELFOSABI_TRU64)
 232   ECase(ELFOSABI_MODESTO)
 233   ECase(ELFOSABI_OPENBSD)
 234   ECase(ELFOSABI_OPENVMS)
 235   ECase(ELFOSABI_NSK)
 236   ECase(ELFOSABI_AROS)
 237   ECase(ELFOSABI_FENIXOS)
 238   ECase(ELFOSABI_CLOUDABI)
 239   ECase(ELFOSABI_C6000_ELFABI)
 240   ECase(ELFOSABI_C6000_LINUX)
 241   ECase(ELFOSABI_ARM)
 242   ECase(ELFOSABI_STANDALONE)
 243 #undef ECase
 244 }
 245
 246 void ScalarBitSetTraits<ELFYAML::ELF_EF>::bitset(IO &IO,
 247                                                  ELFYAML::ELF_EF &Value) {
 248   const auto *Object = static_cast<ELFYAML::Object *>(IO.getContext());
 249   assert(Object && "The IO context is not initialized");
 250 #define BCase(X) IO.bitSetCase(Value, #X, ELF::X);
 251 #define BCaseMask(X, M) IO.maskedBitSetCase(Value, #X, ELF::X, ELF::M);
 252   switch (Object->Header.Machine) {
 253   case ELF::EM_ARM:
 254     BCase(EF_ARM_SOFT_FLOAT)
 255     BCase(EF_ARM_VFP_FLOAT)
 256     BCaseMask(EF_ARM_EABI_UNKNOWN, EF_ARM_EABIMASK)
 257     BCaseMask(EF_ARM_EABI_VER1, EF_ARM_EABIMASK)
 258     BCaseMask(EF_ARM_EABI_VER2, EF_ARM_EABIMASK)
 259     BCaseMask(EF_ARM_EABI_VER3, EF_ARM_EABIMASK)
 260     BCaseMask(EF_ARM_EABI_VER4, EF_ARM_EABIMASK)
 261     BCaseMask(EF_ARM_EABI_VER5, EF_ARM_EABIMASK)
 262     break;
 263   case ELF::EM_MIPS:
 264     BCase(EF_MIPS_NOREORDER)
 265     BCase(EF_MIPS_PIC)
 266     BCase(EF_MIPS_CPIC)
 267     BCase(EF_MIPS_ABI2)
 268     BCase(EF_MIPS_32BITMODE)
 269     BCase(EF_MIPS_FP64)
 270     BCase(EF_MIPS_NAN2008)
 271     BCase(EF_MIPS_MICROMIPS)
 272     BCase(EF_MIPS_ARCH_ASE_M16)
 273     BCase(EF_MIPS_ARCH_ASE_MDMX)
 274     BCaseMask(EF_MIPS_ABI_O32, EF_MIPS_ABI)
 275     BCaseMask(EF_MIPS_ABI_O64, EF_MIPS_ABI)
 276     BCaseMask(EF_MIPS_ABI_EABI32, EF_MIPS_ABI)
 277     BCaseMask(EF_MIPS_ABI_EABI64, EF_MIPS_ABI)
 278     BCaseMask(EF_MIPS_ARCH_1, EF_MIPS_ARCH)
 279     BCaseMask(EF_MIPS_ARCH_2, EF_MIPS_ARCH)
 280     BCaseMask(EF_MIPS_ARCH_3, EF_MIPS_ARCH)
 281     BCaseMask(EF_MIPS_ARCH_4, EF_MIPS_ARCH)
 282     BCaseMask(EF_MIPS_ARCH_5, EF_MIPS_ARCH)
 283     BCaseMask(EF_MIPS_ARCH_32, EF_MIPS_ARCH)
 284     BCaseMask(EF_MIPS_ARCH_64, EF_MIPS_ARCH)
 285     BCaseMask(EF_MIPS_ARCH_32R2, EF_MIPS_ARCH)
 286     BCaseMask(EF_MIPS_ARCH_64R2, EF_MIPS_ARCH)
 287     BCaseMask(EF_MIPS_ARCH_32R6, EF_MIPS_ARCH)
 288     BCaseMask(EF_MIPS_ARCH_64R6, EF_MIPS_ARCH)
 289     break;
 290   case ELF::EM_HEXAGON:
 291     BCase(EF_HEXAGON_MACH_V2)
 292     BCase(EF_HEXAGON_MACH_V3)
 293     BCase(EF_HEXAGON_MACH_V4)
 294     BCase(EF_HEXAGON_MACH_V5)
 295     BCase(EF_HEXAGON_ISA_V2)
 296     BCase(EF_HEXAGON_ISA_V3)
 297     BCase(EF_HEXAGON_ISA_V4)
 298     BCase(EF_HEXAGON_ISA_V5)
 299     break;
 300   default:
 301     llvm_unreachable("Unsupported architecture");
 302   }
 303 #undef BCase
 304 #undef BCaseMask
 305 }
 306
 307 void ScalarEnumerationTraits<ELFYAML::ELF_SHT>::enumeration(
 308     IO &IO, ELFYAML::ELF_SHT &Value) {
 309   const auto *Object = static_cast<ELFYAML::Object *>(IO.getContext());
 310   assert(Object && "The IO context is not initialized");
 311 #define ECase(X) IO.enumCase(Value, #X, ELF::X);
 312   ECase(SHT_NULL)
 313   ECase(SHT_PROGBITS)
 314   // No SHT_SYMTAB. Use the top-level `Symbols` key instead.
 315   // FIXME: Issue a diagnostic with this information.
 316   ECase(SHT_STRTAB)
 317   ECase(SHT_RELA)
 318   ECase(SHT_HASH)
 319   ECase(SHT_DYNAMIC)
 320   ECase(SHT_NOTE)
 321   ECase(SHT_NOBITS)
 322   ECase(SHT_REL)
 323   ECase(SHT_SHLIB)
 324   ECase(SHT_DYNSYM)
 325   ECase(SHT_INIT_ARRAY)
 326   ECase(SHT_FINI_ARRAY)
 327   ECase(SHT_PREINIT_ARRAY)
 328   ECase(SHT_GROUP)
 329   ECase(SHT_SYMTAB_SHNDX)
 330   ECase(SHT_LOOS)
 331   ECase(SHT_GNU_ATTRIBUTES)
 332   ECase(SHT_GNU_HASH)
 333   ECase(SHT_GNU_verdef)
 334   ECase(SHT_GNU_verneed)
 335   ECase(SHT_GNU_versym)
 336   ECase(SHT_HIOS)
 337   ECase(SHT_LOPROC)
 338   switch (Object->Header.Machine) {
 339   case ELF::EM_ARM:
 340     ECase(SHT_ARM_EXIDX)
 341     ECase(SHT_ARM_PREEMPTMAP)
 342     ECase(SHT_ARM_ATTRIBUTES)
 343     ECase(SHT_ARM_DEBUGOVERLAY)
 344     ECase(SHT_ARM_OVERLAYSECTION)
 345     break;
 346   case ELF::EM_HEXAGON:
 347     ECase(SHT_HEX_ORDERED)
 348     break;
 349   case ELF::EM_X86_64:
 350     ECase(SHT_X86_64_UNWIND)
 351     break;
 352   case ELF::EM_MIPS:
 353     ECase(SHT_MIPS_REGINFO)
 354     ECase(SHT_MIPS_OPTIONS)
 355     ECase(SHT_MIPS_ABIFLAGS)
 356     break;
 357   default:
 358     // Nothing to do.
 359     break;
 360   }
 361 #undef ECase
 362 }
 363
 364 void ScalarBitSetTraits<ELFYAML::ELF_SHF>::bitset(IO &IO,
 365                                                   ELFYAML::ELF_SHF &Value) {
 366 #define BCase(X) IO.bitSetCase(Value, #X, ELF::X);
 367   BCase(SHF_WRITE)
 368   BCase(SHF_ALLOC)
 369   BCase(SHF_EXCLUDE)
 370   BCase(SHF_EXECINSTR)
 371   BCase(SHF_MERGE)
 372   BCase(SHF_STRINGS)
 373   BCase(SHF_INFO_LINK)
 374   BCase(SHF_LINK_ORDER)
 375   BCase(SHF_OS_NONCONFORMING)
 376   BCase(SHF_GROUP)
 377   BCase(SHF_TLS)
 378 #undef BCase
 379 }
 380
 381 void ScalarEnumerationTraits<ELFYAML::ELF_STT>::enumeration(
 382     IO &IO, ELFYAML::ELF_STT &Value) {
 383 #define ECase(X) IO.enumCase(Value, #X, ELF::X);
 384   ECase(STT_NOTYPE)
 385   ECase(STT_OBJECT)
 386   ECase(STT_FUNC)
 387   ECase(STT_SECTION)
 388   ECase(STT_FILE)
 389   ECase(STT_COMMON)
 390   ECase(STT_TLS)
 391   ECase(STT_GNU_IFUNC)
 392 #undef ECase
 393 }
 394
 395 void ScalarEnumerationTraits<ELFYAML::ELF_STV>::enumeration(
 396     IO &IO, ELFYAML::ELF_STV &Value) {
 397 #define ECase(X) IO.enumCase(Value, #X, ELF::X);
 398   ECase(STV_DEFAULT)
 399   ECase(STV_INTERNAL)
 400   ECase(STV_HIDDEN)
 401   ECase(STV_PROTECTED)
 402 #undef ECase
 403 }
 404
 405 void ScalarBitSetTraits<ELFYAML::ELF_STO>::bitset(IO &IO,
 406                                                   ELFYAML::ELF_STO &Value) {
 407   const auto *Object = static_cast<ELFYAML::Object *>(IO.getContext());
 408   assert(Object && "The IO context is not initialized");
 409 #define BCase(X) IO.bitSetCase(Value, #X, ELF::X);
 410   switch (Object->Header.Machine) {
 411   case ELF::EM_MIPS:
 412     BCase(STO_MIPS_OPTIONAL)
 413     BCase(STO_MIPS_PLT)
 414     BCase(STO_MIPS_PIC)
 415     BCase(STO_MIPS_MICROMIPS)
 416     break;
 417   default:
 418     break; // Nothing to do
 419   }
 420 #undef BCase
 421 #undef BCaseMask
 422 }
 423
 424 void ScalarEnumerationTraits<ELFYAML::ELF_RSS>::enumeration(
 425     IO &IO, ELFYAML::ELF_RSS &Value) {
 426 #define ECase(X) IO.enumCase(Value, #X, ELF::X);
 427   ECase(RSS_UNDEF)
 428   ECase(RSS_GP)
 429   ECase(RSS_GP0)
 430   ECase(RSS_LOC)
 431 #undef ECase
 432 }
 433
 434 void ScalarEnumerationTraits<ELFYAML::ELF_REL>::enumeration(
 435     IO &IO, ELFYAML::ELF_REL &Value) {
 436   const auto *Object = static_cast<ELFYAML::Object *>(IO.getContext());
 437   assert(Object && "The IO context is not initialized");
 438 #define ELF_RELOC(X, Y) IO.enumCase(Value, #X, ELF::X);
 439   switch (Object->Header.Machine) {
 440   case ELF::EM_X86_64:
 441 #include "llvm/Support/ELFRelocs/x86_64.def"
 442     break;
 443   case ELF::EM_MIPS:
 444 #include "llvm/Support/ELFRelocs/Mips.def"
 445     break;
 446   case ELF::EM_HEXAGON:
 447 #include "llvm/Support/ELFRelocs/Hexagon.def"
 448     break;
 449   case ELF::EM_386:
 450 #include "llvm/Support/ELFRelocs/i386.def"
 451     break;
 452   case ELF::EM_AARCH64:
 453 #include "llvm/Support/ELFRelocs/AArch64.def"
 454     break;
 455   case ELF::EM_ARM:
 456 #include "llvm/Support/ELFRelocs/ARM.def"
 457     break;
 458   default:
 459     llvm_unreachable("Unsupported architecture");
 460   }
 461 #undef ELF_RELOC
 462 }
 463
 464 void MappingTraits<ELFYAML::FileHeader>::mapping(IO &IO,
 465                                                  ELFYAML::FileHeader &FileHdr) {
 466   IO.mapRequired("Class", FileHdr.Class);
 467   IO.mapRequired("Data", FileHdr.Data);
 468   IO.mapOptional("OSABI", FileHdr.OSABI, ELFYAML::ELF_ELFOSABI(0));
 469   IO.mapRequired("Type", FileHdr.Type);
 470   IO.mapRequired("Machine", FileHdr.Machine);
 471   IO.mapOptional("Flags", FileHdr.Flags, ELFYAML::ELF_EF(0));
 472   IO.mapOptional("Entry", FileHdr.Entry, Hex64(0));
 473 }
 474
 475 namespace {
 476 struct NormalizedOther {
 477   NormalizedOther(IO &)
 478       : Visibility(ELFYAML::ELF_STV(0)), Other(ELFYAML::ELF_STO(0)) {}
 479   NormalizedOther(IO &, uint8_t Original)
 480       : Visibility(Original & 0x3), Other(Original & ~0x3) {}
 481
 482   uint8_t denormalize(IO &) { return Visibility | Other; }
 483
 484   ELFYAML::ELF_STV Visibility;
 485   ELFYAML::ELF_STO Other;
 486 };
 487 }
 488
 489 void MappingTraits<ELFYAML::Symbol>::mapping(IO &IO, ELFYAML::Symbol &Symbol) {
 490   IO.mapOptional("Name", Symbol.Name, StringRef());
 491   IO.mapOptional("Type", Symbol.Type, ELFYAML::ELF_STT(0));
 492   IO.mapOptional("Section", Symbol.Section, StringRef());
 493   IO.mapOptional("Value", Symbol.Value, Hex64(0));
 494   IO.mapOptional("Size", Symbol.Size, Hex64(0));
 495
 496   MappingNormalization<NormalizedOther, uint8_t> Keys(IO, Symbol.Other);
 497   IO.mapOptional("Visibility", Keys->Visibility, ELFYAML::ELF_STV(0));
 498   IO.mapOptional("Other", Keys->Other, ELFYAML::ELF_STO(0));
 499 }
 500
 501 void MappingTraits<ELFYAML::LocalGlobalWeakSymbols>::mapping(
 502     IO &IO, ELFYAML::LocalGlobalWeakSymbols &Symbols) {
 503   IO.mapOptional("Local", Symbols.Local);
 504   IO.mapOptional("Global", Symbols.Global);
 505   IO.mapOptional("Weak", Symbols.Weak);
 506 }
 507
 508 static void commonSectionMapping(IO &IO, ELFYAML::Section &Section) {
 509   IO.mapOptional("Name", Section.Name, StringRef());
 510   IO.mapRequired("Type", Section.Type);
 511   IO.mapOptional("Flags", Section.Flags, ELFYAML::ELF_SHF(0));
 512   IO.mapOptional("Address", Section.Address, Hex64(0));
 513   IO.mapOptional("Link", Section.Link, StringRef());
 514   IO.mapOptional("AddressAlign", Section.AddressAlign, Hex64(0));
 515   IO.mapOptional("Info", Section.Info, StringRef());
 516 }
 517
 518 static void sectionMapping(IO &IO, ELFYAML::RawContentSection &Section) {
 519   commonSectionMapping(IO, Section);
 520   IO.mapOptional("Content", Section.Content);
 521   IO.mapOptional("Size", Section.Size, Hex64(Section.Content.binary_size()));
 522 }
 523
 524 static void sectionMapping(IO &IO, ELFYAML::RelocationSection &Section) {
 525   commonSectionMapping(IO, Section);
 526   IO.mapOptional("Relocations", Section.Relocations);
 527 }
 528
 529 static void groupSectionMapping(IO &IO, ELFYAML::Group &group) {
 530   commonSectionMapping(IO, group);
 531   IO.mapRequired("Members", group.Members);
 532 }
 533
 534 void MappingTraits<ELFYAML::SectionOrType>::mapping(
 535     IO &IO, ELFYAML::SectionOrType &sectionOrType) {
 536   IO.mapRequired("SectionOrType", sectionOrType.sectionNameOrType);
 537 }
 538
 539 void MappingTraits<std::unique_ptr<ELFYAML::Section>>::mapping(
 540     IO &IO, std::unique_ptr<ELFYAML::Section> &Section) {
 541   ELFYAML::ELF_SHT sectionType;
 542   if (IO.outputting())
 543     sectionType = Section->Type;
 544   else
 545     IO.mapRequired("Type", sectionType);
 546
 547   switch (sectionType) {
 548   case ELF::SHT_REL:
 549   case ELF::SHT_RELA:
 550     if (!IO.outputting())
 551       Section.reset(new ELFYAML::RelocationSection());
 552     sectionMapping(IO, *cast<ELFYAML::RelocationSection>(Section.get()));
 553     break;
 554   case ELF::SHT_GROUP:
 555     if (!IO.outputting())
 556       Section.reset(new ELFYAML::Group());
 557     groupSectionMapping(IO, *cast<ELFYAML::Group>(Section.get()));
 558     break;
 559   default:
 560     if (!IO.outputting())
 561       Section.reset(new ELFYAML::RawContentSection());
 562     sectionMapping(IO, *cast<ELFYAML::RawContentSection>(Section.get()));
 563   }
 564 }
 565
 566 StringRef MappingTraits<std::unique_ptr<ELFYAML::Section>>::validate(
 567     IO &io, std::unique_ptr<ELFYAML::Section> &Section) {
 568   const auto *RawSection = dyn_cast<ELFYAML::RawContentSection>(Section.get());
 569   if (!RawSection || RawSection->Size >= RawSection->Content.binary_size())
 570     return StringRef();
 571   return "Section size must be greater or equal to the content size";
 572 }
 573
 574 namespace {
 575 struct NormalizedMips64RelType {
 576   NormalizedMips64RelType(IO &)
 577       : Type(ELFYAML::ELF_REL(ELF::R_MIPS_NONE)),
 578         Type2(ELFYAML::ELF_REL(ELF::R_MIPS_NONE)),
 579         Type3(ELFYAML::ELF_REL(ELF::R_MIPS_NONE)),
 580         SpecSym(ELFYAML::ELF_REL(ELF::RSS_UNDEF)) {}
 581   NormalizedMips64RelType(IO &, ELFYAML::ELF_REL Original)
 582       : Type(Original & 0xFF), Type2(Original >> 8 & 0xFF),
 583         Type3(Original >> 16 & 0xFF), SpecSym(Original >> 24 & 0xFF) {}
 584
 585   ELFYAML::ELF_REL denormalize(IO &) {
 586     ELFYAML::ELF_REL Res = Type | Type2 << 8 | Type3 << 16 | SpecSym << 24;
 587     return Res;
 588   }
 589
 590   ELFYAML::ELF_REL Type;
 591   ELFYAML::ELF_REL Type2;
 592   ELFYAML::ELF_REL Type3;
 593   ELFYAML::ELF_RSS SpecSym;
 594 };
 595 }
 596
 597 void MappingTraits<ELFYAML::Relocation>::mapping(IO &IO,
 598                                                  ELFYAML::Relocation &Rel) {
 599   const auto *Object = static_cast<ELFYAML::Object *>(IO.getContext());
 600   assert(Object && "The IO context is not initialized");
 601
 602   IO.mapRequired("Offset", Rel.Offset);
 603   IO.mapRequired("Symbol", Rel.Symbol);
 604
 605   if (Object->Header.Machine == ELFYAML::ELF_EM(ELF::EM_MIPS) &&
 606       Object->Header.Class == ELFYAML::ELF_ELFCLASS(ELF::ELFCLASS64)) {
 607     MappingNormalization<NormalizedMips64RelType, ELFYAML::ELF_REL> Key(
 608         IO, Rel.Type);
 609     IO.mapRequired("Type", Key->Type);
 610     IO.mapOptional("Type2", Key->Type2, ELFYAML::ELF_REL(ELF::R_MIPS_NONE));
 611     IO.mapOptional("Type3", Key->Type3, ELFYAML::ELF_REL(ELF::R_MIPS_NONE));
 612     IO.mapOptional("SpecSym", Key->SpecSym, ELFYAML::ELF_RSS(ELF::RSS_UNDEF));
 613   } else
 614     IO.mapRequired("Type", Rel.Type);
 615
 616   IO.mapOptional("Addend", Rel.Addend, (int64_t)0);
 617 }
 618
 619 void MappingTraits<ELFYAML::Object>::mapping(IO &IO, ELFYAML::Object &Object) {
 620   assert(!IO.getContext() && "The IO context is initialized already");
 621   IO.setContext(&Object);
 622   IO.mapRequired("FileHeader", Object.Header);
 623   IO.mapOptional("Sections", Object.Sections);
 624   IO.mapOptional("Symbols", Object.Symbols);
 625   IO.setContext(nullptr);
 626 }
 627
 628 } // end namespace yaml
 629 } // end namespace llvm