//===- MachOYAML.h - Mach-O YAMLIO implementation ---------------*- C++ -*-===//
 
//
 
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
 
// See https://llvm.org/LICENSE.txt for license information.
 
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
 
//
 
//===----------------------------------------------------------------------===//
 
///
 
/// \file
 
/// This file declares classes for handling the YAML representation
 
/// of Mach-O.
 
///
 
//===----------------------------------------------------------------------===//
 
 
 
#ifndef LLVM_OBJECTYAML_MACHOYAML_H
 
#define LLVM_OBJECTYAML_MACHOYAML_H
 
 
 
#include "llvm/ADT/StringRef.h"
 
#include "llvm/BinaryFormat/MachO.h"
 
#include "llvm/ObjectYAML/DWARFYAML.h"
 
#include "llvm/ObjectYAML/YAML.h"
 
#include "llvm/Support/YAMLTraits.h"
 
#include <cstdint>
 
#include <optional>
 
#include <string>
 
#include <vector>
 
 
 
namespace llvm {
 
namespace MachOYAML {
 
 
 
struct Relocation {
 
  // Offset in the section to what is being relocated.
 
  llvm::yaml::Hex32 address;
 
  // Symbol index if r_extern == 1 else section index.
 
  uint32_t symbolnum;
 
  bool is_pcrel;
 
  // Real length = 2 ^ length.
 
  uint8_t length;
 
  bool is_extern;
 
  uint8_t type;
 
  bool is_scattered;
 
  int32_t value;
 
};
 
 
 
struct Section {
 
  char sectname[16];
 
  char segname[16];
 
  llvm::yaml::Hex64 addr;
 
  uint64_t size;
 
  llvm::yaml::Hex32 offset;
 
  uint32_t align;
 
  llvm::yaml::Hex32 reloff;
 
  uint32_t nreloc;
 
  llvm::yaml::Hex32 flags;
 
  llvm::yaml::Hex32 reserved1;
 
  llvm::yaml::Hex32 reserved2;
 
  llvm::yaml::Hex32 reserved3;
 
  std::optional<llvm::yaml::BinaryRef> content;
 
  std::vector<Relocation> relocations;
 
};
 
 
 
struct FileHeader {
 
  llvm::yaml::Hex32 magic;
 
  llvm::yaml::Hex32 cputype;
 
  llvm::yaml::Hex32 cpusubtype;
 
  llvm::yaml::Hex32 filetype;
 
  uint32_t ncmds;
 
  uint32_t sizeofcmds;
 
  llvm::yaml::Hex32 flags;
 
  llvm::yaml::Hex32 reserved;
 
};
 
 
 
struct LoadCommand {
 
  virtual ~LoadCommand();
 
 
 
  llvm::MachO::macho_load_command Data;
 
  std::vector<Section> Sections;
 
  std::vector<MachO::build_tool_version> Tools;
 
  std::vector<llvm::yaml::Hex8> PayloadBytes;
 
  std::string Content;
 
  uint64_t ZeroPadBytes;
 
};
 
 
 
struct NListEntry {
 
  uint32_t n_strx;
 
  llvm::yaml::Hex8 n_type;
 
  uint8_t n_sect;
 
  uint16_t n_desc;
 
  uint64_t n_value;
 
};
 
 
 
struct RebaseOpcode {
 
  MachO::RebaseOpcode Opcode;
 
  uint8_t Imm;
 
  std::vector<yaml::Hex64> ExtraData;
 
};
 
 
 
struct BindOpcode {
 
  MachO::BindOpcode Opcode;
 
  uint8_t Imm;
 
  std::vector<yaml::Hex64> ULEBExtraData;
 
  std::vector<int64_t> SLEBExtraData;
 
  StringRef Symbol;
 
};
 
 
 
struct ExportEntry {
 
  uint64_t TerminalSize = 0;
 
  uint64_t NodeOffset = 0;
 
  std::string Name;
 
  llvm::yaml::Hex64 Flags = 0;
 
  llvm::yaml::Hex64 Address = 0;
 
  llvm::yaml::Hex64 Other = 0;
 
  std::string ImportName;
 
  std::vector<MachOYAML::ExportEntry> Children;
 
};
 
 
 
struct DataInCodeEntry {
 
  llvm::yaml::Hex32 Offset;
 
  uint16_t Length;
 
  llvm::yaml::Hex16 Kind;
 
};
 
 
 
struct LinkEditData {
 
  std::vector<MachOYAML::RebaseOpcode> RebaseOpcodes;
 
  std::vector<MachOYAML::BindOpcode> BindOpcodes;
 
  std::vector<MachOYAML::BindOpcode> WeakBindOpcodes;
 
  std::vector<MachOYAML::BindOpcode> LazyBindOpcodes;
 
  MachOYAML::ExportEntry ExportTrie;
 
  std::vector<NListEntry> NameList;
 
  std::vector<StringRef> StringTable;
 
  std::vector<yaml::Hex32> IndirectSymbols;
 
  std::vector<yaml::Hex64> FunctionStarts;
 
  std::vector<DataInCodeEntry> DataInCode;
 
  std::vector<yaml::Hex8> ChainedFixups;
 
 
 
  bool isEmpty() const;
 
};
 
 
 
struct Object {
 
  bool IsLittleEndian;
 
  FileHeader Header;
 
  std::vector<LoadCommand> LoadCommands;
 
  std::vector<Section> Sections;
 
  LinkEditData LinkEdit;
 
  std::optional<llvm::yaml::BinaryRef> RawLinkEditSegment;
 
  DWARFYAML::Data DWARF;
 
};
 
 
 
struct FatHeader {
 
  llvm::yaml::Hex32 magic;
 
  uint32_t nfat_arch;
 
};
 
 
 
struct FatArch {
 
  llvm::yaml::Hex32 cputype;
 
  llvm::yaml::Hex32 cpusubtype;
 
  llvm::yaml::Hex64 offset;
 
  uint64_t size;
 
  uint32_t align;
 
  llvm::yaml::Hex32 reserved;
 
};
 
 
 
struct UniversalBinary {
 
  FatHeader Header;
 
  std::vector<FatArch> FatArchs;
 
  std::vector<Object> Slices;
 
};
 
 
 
} // end namespace MachOYAML
 
} // end namespace llvm
 
 
 
LLVM_YAML_IS_SEQUENCE_VECTOR(llvm::MachOYAML::LoadCommand)
 
LLVM_YAML_IS_SEQUENCE_VECTOR(llvm::MachOYAML::Relocation)
 
LLVM_YAML_IS_SEQUENCE_VECTOR(llvm::MachOYAML::Section)
 
LLVM_YAML_IS_SEQUENCE_VECTOR(llvm::MachOYAML::RebaseOpcode)
 
LLVM_YAML_IS_SEQUENCE_VECTOR(llvm::MachOYAML::BindOpcode)
 
LLVM_YAML_IS_SEQUENCE_VECTOR(llvm::MachOYAML::ExportEntry)
 
LLVM_YAML_IS_SEQUENCE_VECTOR(llvm::MachOYAML::NListEntry)
 
LLVM_YAML_IS_SEQUENCE_VECTOR(llvm::MachOYAML::Object)
 
LLVM_YAML_IS_SEQUENCE_VECTOR(llvm::MachOYAML::FatArch)
 
LLVM_YAML_IS_SEQUENCE_VECTOR(llvm::MachOYAML::DataInCodeEntry)
 
LLVM_YAML_IS_SEQUENCE_VECTOR(llvm::MachO::build_tool_version)
 
 
 
namespace llvm {
 
 
 
class raw_ostream;
 
 
 
namespace yaml {
 
 
 
template <> struct MappingTraits<MachOYAML::FileHeader> {
 
  static void mapping(IO &IO, MachOYAML::FileHeader &FileHeader);
 
};
 
 
 
template <> struct MappingTraits<MachOYAML::Object> {
 
  static void mapping(IO &IO, MachOYAML::Object &Object);
 
};
 
 
 
template <> struct MappingTraits<MachOYAML::FatHeader> {
 
  static void mapping(IO &IO, MachOYAML::FatHeader &FatHeader);
 
};
 
 
 
template <> struct MappingTraits<MachOYAML::FatArch> {
 
  static void mapping(IO &IO, MachOYAML::FatArch &FatArch);
 
};
 
 
 
template <> struct MappingTraits<MachOYAML::UniversalBinary> {
 
  static void mapping(IO &IO, MachOYAML::UniversalBinary &UniversalBinary);
 
};
 
 
 
template <> struct MappingTraits<MachOYAML::LoadCommand> {
 
  static void mapping(IO &IO, MachOYAML::LoadCommand &LoadCommand);
 
};
 
 
 
template <> struct MappingTraits<MachOYAML::LinkEditData> {
 
  static void mapping(IO &IO, MachOYAML::LinkEditData &LinkEditData);
 
};
 
 
 
template <> struct MappingTraits<MachOYAML::RebaseOpcode> {
 
  static void mapping(IO &IO, MachOYAML::RebaseOpcode &RebaseOpcode);
 
};
 
 
 
template <> struct MappingTraits<MachOYAML::BindOpcode> {
 
  static void mapping(IO &IO, MachOYAML::BindOpcode &BindOpcode);
 
};
 
 
 
template <> struct MappingTraits<MachOYAML::ExportEntry> {
 
  static void mapping(IO &IO, MachOYAML::ExportEntry &ExportEntry);
 
};
 
 
 
template <> struct MappingTraits<MachOYAML::Relocation> {
 
  static void mapping(IO &IO, MachOYAML::Relocation &R);
 
};
 
 
 
template <> struct MappingTraits<MachOYAML::Section> {
 
  static void mapping(IO &IO, MachOYAML::Section &Section);
 
  static std::string validate(IO &io, MachOYAML::Section &Section);
 
};
 
 
 
template <> struct MappingTraits<MachOYAML::NListEntry> {
 
  static void mapping(IO &IO, MachOYAML::NListEntry &NListEntry);
 
};
 
 
 
template <> struct MappingTraits<MachO::build_tool_version> {
 
  static void mapping(IO &IO, MachO::build_tool_version &tool);
 
};
 
 
 
template <> struct MappingTraits<MachOYAML::DataInCodeEntry> {
 
  static void mapping(IO &IO, MachOYAML::DataInCodeEntry &DataInCodeEntry);
 
};
 
 
 
#define HANDLE_LOAD_COMMAND(LCName, LCValue, LCStruct)                         \
 
  io.enumCase(value, #LCName, MachO::LCName);
 
 
 
template <> struct ScalarEnumerationTraits<MachO::LoadCommandType> {
 
  static void enumeration(IO &io, MachO::LoadCommandType &value) {
 
#include "llvm/BinaryFormat/MachO.def"
 
    io.enumFallback<Hex32>(value);
 
  }
 
};
 
 
 
#define ENUM_CASE(Enum) io.enumCase(value, #Enum, MachO::Enum);
 
 
 
template <> struct ScalarEnumerationTraits<MachO::RebaseOpcode> {
 
  static void enumeration(IO &io, MachO::RebaseOpcode &value) {
 
    ENUM_CASE(REBASE_OPCODE_DONE)
 
    ENUM_CASE(REBASE_OPCODE_SET_TYPE_IMM)
 
    ENUM_CASE(REBASE_OPCODE_SET_SEGMENT_AND_OFFSET_ULEB)
 
    ENUM_CASE(REBASE_OPCODE_ADD_ADDR_ULEB)
 
    ENUM_CASE(REBASE_OPCODE_ADD_ADDR_IMM_SCALED)
 
    ENUM_CASE(REBASE_OPCODE_DO_REBASE_IMM_TIMES)
 
    ENUM_CASE(REBASE_OPCODE_DO_REBASE_ULEB_TIMES)
 
    ENUM_CASE(REBASE_OPCODE_DO_REBASE_ADD_ADDR_ULEB)
 
    ENUM_CASE(REBASE_OPCODE_DO_REBASE_ULEB_TIMES_SKIPPING_ULEB)
 
    io.enumFallback<Hex8>(value);
 
  }
 
};
 
 
 
template <> struct ScalarEnumerationTraits<MachO::BindOpcode> {
 
  static void enumeration(IO &io, MachO::BindOpcode &value) {
 
    ENUM_CASE(BIND_OPCODE_DONE)
 
    ENUM_CASE(BIND_OPCODE_SET_DYLIB_ORDINAL_IMM)
 
    ENUM_CASE(BIND_OPCODE_SET_DYLIB_ORDINAL_ULEB)
 
    ENUM_CASE(BIND_OPCODE_SET_DYLIB_SPECIAL_IMM)
 
    ENUM_CASE(BIND_OPCODE_SET_SYMBOL_TRAILING_FLAGS_IMM)
 
    ENUM_CASE(BIND_OPCODE_SET_TYPE_IMM)
 
    ENUM_CASE(BIND_OPCODE_SET_ADDEND_SLEB)
 
    ENUM_CASE(BIND_OPCODE_SET_SEGMENT_AND_OFFSET_ULEB)
 
    ENUM_CASE(BIND_OPCODE_ADD_ADDR_ULEB)
 
    ENUM_CASE(BIND_OPCODE_DO_BIND)
 
    ENUM_CASE(BIND_OPCODE_DO_BIND_ADD_ADDR_ULEB)
 
    ENUM_CASE(BIND_OPCODE_DO_BIND_ADD_ADDR_IMM_SCALED)
 
    ENUM_CASE(BIND_OPCODE_DO_BIND_ULEB_TIMES_SKIPPING_ULEB)
 
    io.enumFallback<Hex8>(value);
 
  }
 
};
 
 
 
// This trait is used for 16-byte chars in Mach structures used for strings
 
using char_16 = char[16];
 
 
 
template <> struct ScalarTraits<char_16> {
 
  static void output(const char_16 &Val, void *, raw_ostream &Out);
 
  static StringRef input(StringRef Scalar, void *, char_16 &Val);
 
  static QuotingType mustQuote(StringRef S);
 
};
 
 
 
// This trait is used for UUIDs. It reads and writes them matching otool's
 
// formatting style.
 
using uuid_t = raw_ostream::uuid_t;
 
 
 
template <> struct ScalarTraits<uuid_t> {
 
  static void output(const uuid_t &Val, void *, raw_ostream &Out);
 
  static StringRef input(StringRef Scalar, void *, uuid_t &Val);
 
  static QuotingType mustQuote(StringRef S);
 
};
 
 
 
// Load Command struct mapping traits
 
 
 
#define LOAD_COMMAND_STRUCT(LCStruct)                                          \
 
  template <> struct MappingTraits<MachO::LCStruct> {                          \
 
    static void mapping(IO &IO, MachO::LCStruct &LoadCommand);                 \
 
  };
 
 
 
#include "llvm/BinaryFormat/MachO.def"
 
 
 
// Extra structures used by load commands
 
template <> struct MappingTraits<MachO::dylib> {
 
  static void mapping(IO &IO, MachO::dylib &LoadCommand);
 
};
 
 
 
template <> struct MappingTraits<MachO::fvmlib> {
 
  static void mapping(IO &IO, MachO::fvmlib &LoadCommand);
 
};
 
 
 
template <> struct MappingTraits<MachO::section> {
 
  static void mapping(IO &IO, MachO::section &LoadCommand);
 
};
 
 
 
template <> struct MappingTraits<MachO::section_64> {
 
  static void mapping(IO &IO, MachO::section_64 &LoadCommand);
 
};
 
 
 
} // end namespace yaml
 
 
 
} // end namespace llvm
 
 
 
#endif // LLVM_OBJECTYAML_MACHOYAML_H