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