CbC/CbC_llvm: lib/ObjectYAML/COFFYAML.cpp comparison

comparison lib/ObjectYAML/COFFYAML.cpp @ 121:803732b1fca8

LLVM 5.0

author	kono
date	Fri, 27 Oct 2017 17:07:41 +0900
parents	1172e4bd9c6f
children	3a76565eade5

comparison

equal deleted inserted replaced

-:1172e4bd9c6f
+:803732b1fca8
 // This file defines classes for handling the YAML representation of COFF.
 //
 //===----------------------------------------------------------------------===//
 #include "llvm/ObjectYAML/COFFYAML.h"
+#include "llvm/ADT/StringRef.h"
+#include "llvm/Support/YAMLTraits.h"
+#include <cstdint>
+#include <cstring>
 #define ECase(X) IO.enumCase(Value, #X, COFF::X);
 namespace llvm {
 namespace COFFYAML {
 Section::Section() { memset(&Header, 0, sizeof(COFF::section)); }
 Symbol::Symbol() { memset(&Header, 0, sizeof(COFF::symbol)); }
 Object::Object() { memset(&Header, 0, sizeof(COFF::header)); }
-}
+} // end namespace COFFYAML
 namespace yaml {
 void ScalarEnumerationTraits<COFFYAML::COMDATType>::enumeration(
 IO &IO, COFFYAML::COMDATType &Value) {
 IO.enumCase(Value, "0", 0);
 ECase(IMAGE_COMDAT_SELECT_NODUPLICATES);
 ECase(IMAGE_COMDAT_SELECT_ANY);
 ECase(IMAGE_REL_AMD64_SSPAN32);
 }
 void ScalarEnumerationTraits<COFF::WindowsSubsystem>::enumeration(
 IO &IO, COFF::WindowsSubsystem &Value) {
 ECase(IMAGE_SUBSYSTEM_UNKNOWN);
 ECase(IMAGE_SUBSYSTEM_NATIVE);
 ECase(IMAGE_SUBSYSTEM_WINDOWS_GUI);
 ECase(IMAGE_SUBSYSTEM_WINDOWS_CUI);
 ECase(IMAGE_SUBSYSTEM_OS2_CUI);
 ECase(IMAGE_SUBSYSTEM_POSIX_CUI);
 ECase(IMAGE_SUBSYSTEM_NATIVE_WINDOWS);
 ECase(IMAGE_SUBSYSTEM_WINDOWS_CE_GUI);
 ECase(IMAGE_SUBSYSTEM_EFI_APPLICATION);
 ECase(IMAGE_SUBSYSTEM_EFI_BOOT_SERVICE_DRIVER);
 ECase(IMAGE_SUBSYSTEM_EFI_RUNTIME_DRIVER);
 ECase(IMAGE_SUBSYSTEM_EFI_ROM);
 ECase(IMAGE_SUBSYSTEM_XBOX);
 ECase(IMAGE_SUBSYSTEM_WINDOWS_BOOT_APPLICATION);
 }
 #undef ECase
 #define BCase(X) IO.bitSetCase(Value, #X, COFF::X);
 void ScalarBitSetTraits<COFF::Characteristics>::bitset(
 BCase(IMAGE_DLL_CHARACTERISTICS_TERMINAL_SERVER_AWARE);
 }
 #undef BCase
 namespace {
 struct NSectionSelectionType {
 NSectionSelectionType(IO &)
 : SelectionType(COFFYAML::COMDATType(0)) {}
 NSectionSelectionType(IO &, uint8_t C)
 : SelectionType(COFFYAML::COMDATType(C)) {}
 uint8_t denormalize(IO &) { return SelectionType; }
 COFFYAML::COMDATType SelectionType;
 };
 struct NWeakExternalCharacteristics {
 NWeakExternalCharacteristics(IO &)
 : Characteristics(COFFYAML::WeakExternalCharacteristics(0)) {}
 NWeakExternalCharacteristics(IO &, uint32_t C)
 : Characteristics(COFFYAML::WeakExternalCharacteristics(C)) {}
 uint32_t denormalize(IO &) { return Characteristics; }
 COFFYAML::WeakExternalCharacteristics Characteristics;
 };
 struct NSectionCharacteristics {
 NSectionCharacteristics(IO &)
 : Characteristics(COFF::SectionCharacteristics(0)) {}
 NSectionCharacteristics(IO &, uint32_t C)
 : Characteristics(COFF::SectionCharacteristics(C)) {}
 uint32_t denormalize(IO &) { return Characteristics; }
 COFF::SectionCharacteristics Characteristics;
 };
 struct NAuxTokenType {
 NAuxTokenType(IO &)
 : AuxType(COFFYAML::AuxSymbolType(0)) {}
 NAuxTokenType(IO &, uint8_t C)
 : AuxType(COFFYAML::AuxSymbolType(C)) {}
 uint32_t denormalize(IO &) { return AuxType; }
 COFFYAML::AuxSymbolType AuxType;
 };
 struct NStorageClass {
 NStorageClass(IO &) : StorageClass(COFF::SymbolStorageClass(0)) {}
 NStorageClass(IO &, uint8_t S) : StorageClass(COFF::SymbolStorageClass(S)) {}
 uint8_t denormalize(IO &) { return StorageClass; }
 COFF::SymbolStorageClass StorageClass;
 };
 struct NMachine {
 NMachine(IO &) : Machine(COFF::MachineTypes(0)) {}
 NMachine(IO &, uint16_t M) : Machine(COFF::MachineTypes(M)) {}
 uint16_t denormalize(IO &) { return Machine; }
 COFF::MachineTypes Machine;
 };
 struct NHeaderCharacteristics {
 NHeaderCharacteristics(IO &) : Characteristics(COFF::Characteristics(0)) {}
 NHeaderCharacteristics(IO &, uint16_t C)
 : Characteristics(COFF::Characteristics(C)) {}
 uint16_t denormalize(IO &) { return Characteristics; }
 COFF::Characteristics Characteristics;
 };
 template <typename RelocType>
 struct NType {
 NType(IO &) : Type(RelocType(0)) {}
 NType(IO &, uint16_t T) : Type(RelocType(T)) {}
 uint16_t denormalize(IO &) { return Type; }
 RelocType Type;
 };
 struct NWindowsSubsystem {
 NWindowsSubsystem(IO &) : Subsystem(COFF::WindowsSubsystem(0)) {}
 NWindowsSubsystem(IO &, uint16_t C) : Subsystem(COFF::WindowsSubsystem(C)) {}
 uint16_t denormalize(IO &) { return Subsystem; }
 COFF::WindowsSubsystem Subsystem;
 };
 struct NDLLCharacteristics {
 NDLLCharacteristics(IO &) : Characteristics(COFF::DLLCharacteristics(0)) {}
 NDLLCharacteristics(IO &, uint16_t C)
 : Characteristics(COFF::DLLCharacteristics(C)) {}
 uint16_t denormalize(IO &) { return Characteristics; }
 COFF::DLLCharacteristics Characteristics;
 };
-}
+} // end anonymous namespace
 void MappingTraits<COFFYAML::Relocation>::mapping(IO &IO,
 COFFYAML::Relocation &Rel) {
 IO.mapRequired("VirtualAddress", Rel.VirtualAddress);
 IO.mapRequired("SymbolName", Rel.SymbolName);
 IO.mapRequired("Name", Sec.Name);
 IO.mapRequired("Characteristics", NC->Characteristics);
 IO.mapOptional("VirtualAddress", Sec.Header.VirtualAddress, 0U);
 IO.mapOptional("VirtualSize", Sec.Header.VirtualSize, 0U);
 IO.mapOptional("Alignment", Sec.Alignment, 0U);
-IO.mapRequired("SectionData", Sec.SectionData);
+// If this is a .debug$S or .debug$T section parse the semantic representation
+// of the symbols/types.  If it is any other kind of section, just deal in raw
+// bytes.
+IO.mapOptional("SectionData", Sec.SectionData);
+if (Sec.Name == ".debug$S")
+IO.mapOptional("Subsections", Sec.DebugS);
+else if (Sec.Name == ".debug$T")
+IO.mapOptional("Types", Sec.DebugT);
 IO.mapOptional("Relocations", Sec.Relocations);
 }
 void MappingTraits<COFFYAML::Object>::mapping(IO &IO, COFFYAML::Object &Obj) {
 IO.mapTag("!COFF", true);
 IO.mapRequired("header", Obj.Header);
 IO.mapRequired("sections", Obj.Sections);
 IO.mapRequired("symbols", Obj.Symbols);
 }
-}
+} // end namespace yaml
-}
+} // end namespace llvm

Mercurial > hg > CbC > CbC_llvm

comparison lib/ObjectYAML/COFFYAML.cpp @ 121:803732b1fca8