contrib/llvm/include/llvm/ObjectYAML/MachOYAML.h

   1 //===- MachOYAML.h - Mach-O YAMLIO implementation ---------------*- C++ -*-===//
   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 /// \file
  11 /// \brief This file declares classes for handling the YAML representation
  12 /// of Mach-O.
  13 ///
  14 //===----------------------------------------------------------------------===//
  15
  16 #ifndef LLVM_OBJECTYAML_MACHOYAML_H
  17 #define LLVM_OBJECTYAML_MACHOYAML_H
  18
  19 #include "llvm/ObjectYAML/YAML.h"
  20 #include "llvm/ObjectYAML/DWARFYAML.h"
  21 #include "llvm/Support/MachO.h"
  22
  23 namespace llvm {
  24 namespace MachOYAML {
  25
  26 struct Section {
  27   char sectname[16];
  28   char segname[16];
  29   llvm::yaml::Hex64 addr;
  30   uint64_t size;
  31   llvm::yaml::Hex32 offset;
  32   uint32_t align;
  33   llvm::yaml::Hex32 reloff;
  34   uint32_t nreloc;
  35   llvm::yaml::Hex32 flags;
  36   llvm::yaml::Hex32 reserved1;
  37   llvm::yaml::Hex32 reserved2;
  38   llvm::yaml::Hex32 reserved3;
  39 };
  40
  41 struct FileHeader {
  42   llvm::yaml::Hex32 magic;
  43   llvm::yaml::Hex32 cputype;
  44   llvm::yaml::Hex32 cpusubtype;
  45   llvm::yaml::Hex32 filetype;
  46   uint32_t ncmds;
  47   uint32_t sizeofcmds;
  48   llvm::yaml::Hex32 flags;
  49   llvm::yaml::Hex32 reserved;
  50 };
  51
  52 struct LoadCommand {
  53   virtual ~LoadCommand();
  54   llvm::MachO::macho_load_command Data;
  55   std::vector<Section> Sections;
  56   std::vector<llvm::yaml::Hex8> PayloadBytes;
  57   std::string PayloadString;
  58   uint64_t ZeroPadBytes;
  59 };
  60
  61 struct NListEntry {
  62   uint32_t n_strx;
  63   llvm::yaml::Hex8 n_type;
  64   uint8_t n_sect;
  65   uint16_t n_desc;
  66   uint64_t n_value;
  67 };
  68 struct RebaseOpcode {
  69   MachO::RebaseOpcode Opcode;
  70   uint8_t Imm;
  71   std::vector<yaml::Hex64> ExtraData;
  72 };
  73
  74 struct BindOpcode {
  75   MachO::BindOpcode Opcode;
  76   uint8_t Imm;
  77   std::vector<yaml::Hex64> ULEBExtraData;
  78   std::vector<int64_t> SLEBExtraData;
  79   StringRef Symbol;
  80 };
  81
  82 struct ExportEntry {
  83   ExportEntry()
  84       : TerminalSize(0), NodeOffset(0), Name(), Flags(0), Address(0), Other(0),
  85         ImportName(), Children() {}
  86   uint64_t TerminalSize;
  87   uint64_t NodeOffset;
  88   std::string Name;
  89   llvm::yaml::Hex64 Flags;
  90   llvm::yaml::Hex64 Address;
  91   llvm::yaml::Hex64 Other;
  92   std::string ImportName;
  93   std::vector<MachOYAML::ExportEntry> Children;
  94 };
  95
  96 struct LinkEditData {
  97   std::vector<MachOYAML::RebaseOpcode> RebaseOpcodes;
  98   std::vector<MachOYAML::BindOpcode> BindOpcodes;
  99   std::vector<MachOYAML::BindOpcode> WeakBindOpcodes;
 100   std::vector<MachOYAML::BindOpcode> LazyBindOpcodes;
 101   MachOYAML::ExportEntry ExportTrie;
 102   std::vector<NListEntry> NameList;
 103   std::vector<StringRef> StringTable;
 104
 105   bool isEmpty() const;
 106 };
 107
 108 struct Object {
 109   bool IsLittleEndian;
 110   FileHeader Header;
 111   std::vector<LoadCommand> LoadCommands;
 112   std::vector<Section> Sections;
 113   LinkEditData LinkEdit;
 114   DWARFYAML::Data DWARF;
 115 };
 116
 117 struct FatHeader {
 118   llvm::yaml::Hex32 magic;
 119   uint32_t nfat_arch;
 120 };
 121
 122 struct FatArch {
 123   llvm::yaml::Hex32 cputype;
 124   llvm::yaml::Hex32 cpusubtype;
 125   llvm::yaml::Hex64 offset;
 126   uint64_t size;
 127   uint32_t align;
 128   llvm::yaml::Hex32 reserved;
 129 };
 130
 131 struct UniversalBinary {
 132   FatHeader Header;
 133   std::vector<FatArch> FatArchs;
 134   std::vector<Object> Slices;
 135 };
 136
 137 } // namespace llvm::MachOYAML
 138 } // namespace llvm
 139
 140 LLVM_YAML_IS_SEQUENCE_VECTOR(llvm::MachOYAML::LoadCommand)
 141 LLVM_YAML_IS_SEQUENCE_VECTOR(llvm::MachOYAML::Section)
 142 LLVM_YAML_IS_SEQUENCE_VECTOR(llvm::yaml::Hex64)
 143 LLVM_YAML_IS_SEQUENCE_VECTOR(int64_t)
 144 LLVM_YAML_IS_SEQUENCE_VECTOR(llvm::MachOYAML::RebaseOpcode)
 145 LLVM_YAML_IS_SEQUENCE_VECTOR(llvm::MachOYAML::BindOpcode)
 146 LLVM_YAML_IS_SEQUENCE_VECTOR(llvm::MachOYAML::ExportEntry)
 147 LLVM_YAML_IS_SEQUENCE_VECTOR(llvm::MachOYAML::NListEntry)
 148 LLVM_YAML_IS_SEQUENCE_VECTOR(llvm::MachOYAML::Object)
 149 LLVM_YAML_IS_SEQUENCE_VECTOR(llvm::MachOYAML::FatArch)
 150
 151 namespace llvm {
 152 namespace yaml {
 153
 154 template <> struct MappingTraits<MachOYAML::FileHeader> {
 155   static void mapping(IO &IO, MachOYAML::FileHeader &FileHeader);
 156 };
 157
 158 template <> struct MappingTraits<MachOYAML::Object> {
 159   static void mapping(IO &IO, MachOYAML::Object &Object);
 160 };
 161
 162 template <> struct MappingTraits<MachOYAML::FatHeader> {
 163   static void mapping(IO &IO, MachOYAML::FatHeader &FatHeader);
 164 };
 165
 166 template <> struct MappingTraits<MachOYAML::FatArch> {
 167   static void mapping(IO &IO, MachOYAML::FatArch &FatArch);
 168 };
 169
 170 template <> struct MappingTraits<MachOYAML::UniversalBinary> {
 171   static void mapping(IO &IO, MachOYAML::UniversalBinary &UniversalBinary);
 172 };
 173
 174 template <> struct MappingTraits<MachOYAML::LoadCommand> {
 175   static void mapping(IO &IO, MachOYAML::LoadCommand &LoadCommand);
 176 };
 177
 178 template <> struct MappingTraits<MachOYAML::LinkEditData> {
 179   static void mapping(IO &IO, MachOYAML::LinkEditData &LinkEditData);
 180 };
 181
 182 template <> struct MappingTraits<MachOYAML::RebaseOpcode> {
 183   static void mapping(IO &IO, MachOYAML::RebaseOpcode &RebaseOpcode);
 184 };
 185
 186 template <> struct MappingTraits<MachOYAML::BindOpcode> {
 187   static void mapping(IO &IO, MachOYAML::BindOpcode &BindOpcode);
 188 };
 189
 190 template <> struct MappingTraits<MachOYAML::ExportEntry> {
 191   static void mapping(IO &IO, MachOYAML::ExportEntry &ExportEntry);
 192 };
 193
 194 template <> struct MappingTraits<MachOYAML::Section> {
 195   static void mapping(IO &IO, MachOYAML::Section &Section);
 196 };
 197
 198 template <> struct MappingTraits<MachOYAML::NListEntry> {
 199   static void mapping(IO &IO, MachOYAML::NListEntry &NListEntry);
 200 };
 201
 202 #define HANDLE_LOAD_COMMAND(LCName, LCValue, LCStruct)                         \
 203   io.enumCase(value, #LCName, MachO::LCName);
 204
 205 template <> struct ScalarEnumerationTraits<MachO::LoadCommandType> {
 206   static void enumeration(IO &io, MachO::LoadCommandType &value) {
 207 #include "llvm/Support/MachO.def"
 208     io.enumFallback<Hex32>(value);
 209   }
 210 };
 211
 212 #define ENUM_CASE(Enum) io.enumCase(value, #Enum, MachO::Enum);
 213
 214 template <> struct ScalarEnumerationTraits<MachO::RebaseOpcode> {
 215   static void enumeration(IO &io, MachO::RebaseOpcode &value) {
 216     ENUM_CASE(REBASE_OPCODE_DONE)
 217     ENUM_CASE(REBASE_OPCODE_SET_TYPE_IMM)
 218     ENUM_CASE(REBASE_OPCODE_SET_SEGMENT_AND_OFFSET_ULEB)
 219     ENUM_CASE(REBASE_OPCODE_ADD_ADDR_ULEB)
 220     ENUM_CASE(REBASE_OPCODE_ADD_ADDR_IMM_SCALED)
 221     ENUM_CASE(REBASE_OPCODE_DO_REBASE_IMM_TIMES)
 222     ENUM_CASE(REBASE_OPCODE_DO_REBASE_ULEB_TIMES)
 223     ENUM_CASE(REBASE_OPCODE_DO_REBASE_ADD_ADDR_ULEB)
 224     ENUM_CASE(REBASE_OPCODE_DO_REBASE_ULEB_TIMES_SKIPPING_ULEB)
 225     io.enumFallback<Hex8>(value);
 226   }
 227 };
 228
 229 template <> struct ScalarEnumerationTraits<MachO::BindOpcode> {
 230   static void enumeration(IO &io, MachO::BindOpcode &value) {
 231     ENUM_CASE(BIND_OPCODE_DONE)
 232     ENUM_CASE(BIND_OPCODE_SET_DYLIB_ORDINAL_IMM)
 233     ENUM_CASE(BIND_OPCODE_SET_DYLIB_ORDINAL_ULEB)
 234     ENUM_CASE(BIND_OPCODE_SET_DYLIB_SPECIAL_IMM)
 235     ENUM_CASE(BIND_OPCODE_SET_SYMBOL_TRAILING_FLAGS_IMM)
 236     ENUM_CASE(BIND_OPCODE_SET_TYPE_IMM)
 237     ENUM_CASE(BIND_OPCODE_SET_ADDEND_SLEB)
 238     ENUM_CASE(BIND_OPCODE_SET_SEGMENT_AND_OFFSET_ULEB)
 239     ENUM_CASE(BIND_OPCODE_ADD_ADDR_ULEB)
 240     ENUM_CASE(BIND_OPCODE_DO_BIND)
 241     ENUM_CASE(BIND_OPCODE_DO_BIND_ADD_ADDR_ULEB)
 242     ENUM_CASE(BIND_OPCODE_DO_BIND_ADD_ADDR_IMM_SCALED)
 243     ENUM_CASE(BIND_OPCODE_DO_BIND_ULEB_TIMES_SKIPPING_ULEB)
 244     io.enumFallback<Hex8>(value);
 245   }
 246 };
 247
 248 // This trait is used for 16-byte chars in Mach structures used for strings
 249 typedef char char_16[16];
 250
 251 template <> struct ScalarTraits<char_16> {
 252   static void output(const char_16 &Val, void *, llvm::raw_ostream &Out);
 253
 254   static StringRef input(StringRef Scalar, void *, char_16 &Val);
 255   static bool mustQuote(StringRef S);
 256 };
 257
 258 // This trait is used for UUIDs. It reads and writes them matching otool's
 259 // formatting style.
 260 typedef uint8_t uuid_t[16];
 261
 262 template <> struct ScalarTraits<uuid_t> {
 263   static void output(const uuid_t &Val, void *, llvm::raw_ostream &Out);
 264
 265   static StringRef input(StringRef Scalar, void *, uuid_t &Val);
 266   static bool mustQuote(StringRef S);
 267 };
 268
 269 // Load Command struct mapping traits
 270
 271 #define LOAD_COMMAND_STRUCT(LCStruct)                                          \
 272   template <> struct MappingTraits<MachO::LCStruct> {                          \
 273     static void mapping(IO &IO, MachO::LCStruct &LoadCommand);                 \
 274   };
 275
 276 #include "llvm/Support/MachO.def"
 277
 278 // Extra structures used by load commands
 279 template <> struct MappingTraits<MachO::dylib> {
 280   static void mapping(IO &IO, MachO::dylib &LoadCommand);
 281 };
 282
 283 template <> struct MappingTraits<MachO::fvmlib> {
 284   static void mapping(IO &IO, MachO::fvmlib &LoadCommand);
 285 };
 286
 287 template <> struct MappingTraits<MachO::section> {
 288   static void mapping(IO &IO, MachO::section &LoadCommand);
 289 };
 290
 291 template <> struct MappingTraits<MachO::section_64> {
 292   static void mapping(IO &IO, MachO::section_64 &LoadCommand);
 293 };
 294
 295 } // namespace llvm::yaml
 296
 297 } // namespace llvm
 298
 299 #endif