[llvm] r271300 - [obj2yaml][yaml2obj] Support for reading and dumping the MachO export trie

Chris Bieneman via llvm-commits llvm-commits at lists.llvm.org
Tue May 31 14:07:03 PDT 2016


Should be fixed with r271334.

-Chris

> On May 31, 2016, at 2:04 PM, Chris Bieneman via llvm-commits <llvm-commits at lists.llvm.org> wrote:
> 
> My bad. Committing shortly.
> 
> -Chris
> 
>> On May 31, 2016, at 1:27 PM, Reid Kleckner <rnk at google.com <mailto:rnk at google.com>> wrote:
>> 
>> MSan doesn't like your change:
>> http://lab.llvm.org:8011/builders/sanitizer-x86_64-linux-fast/builds/13302/steps/check-llvm%20msan/logs/stdio <http://lab.llvm.org:8011/builders/sanitizer-x86_64-linux-fast/builds/13302/steps/check-llvm%20msan/logs/stdio>
>> 
>> Try adding a default member initializer for these fields:
>> +struct ExportEntry {
>> +  uint64_t TerminalSize;
>> +  uint64_t NodeOffset;
>> 
>> On Tue, May 31, 2016 at 10:26 AM, Chris Bieneman via llvm-commits <llvm-commits at lists.llvm.org <mailto:llvm-commits at lists.llvm.org>> wrote:
>> Author: cbieneman
>> Date: Tue May 31 12:26:36 2016
>> New Revision: 271300
>> 
>> URL: http://llvm.org/viewvc/llvm-project?rev=271300&view=rev <http://llvm.org/viewvc/llvm-project?rev=271300&view=rev>
>> Log:
>> [obj2yaml][yaml2obj] Support for reading and dumping the MachO export trie
>> 
>> The MachO export trie is a serially encoded trie keyed by symbol name. This code parses the trie and preserves the structure so that it can be dumped again.
>> 
>> Added:
>>     llvm/trunk/test/ObjectYAML/MachO/export_trie.yaml
>> Modified:
>>     llvm/trunk/include/llvm/ObjectYAML/MachOYAML.h
>>     llvm/trunk/lib/ObjectYAML/MachOYAML.cpp
>>     llvm/trunk/tools/obj2yaml/macho2yaml.cpp
>>     llvm/trunk/tools/yaml2obj/yaml2macho.cpp
>> 
>> Modified: llvm/trunk/include/llvm/ObjectYAML/MachOYAML.h
>> URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/include/llvm/ObjectYAML/MachOYAML.h?rev=271300&r1=271299&r2=271300&view=diff <http://llvm.org/viewvc/llvm-project/llvm/trunk/include/llvm/ObjectYAML/MachOYAML.h?rev=271300&r1=271299&r2=271300&view=diff>
>> ==============================================================================
>> --- llvm/trunk/include/llvm/ObjectYAML/MachOYAML.h (original)
>> +++ llvm/trunk/include/llvm/ObjectYAML/MachOYAML.h Tue May 31 12:26:36 2016
>> @@ -71,11 +71,23 @@ struct BindOpcode {
>>    StringRef Symbol;
>>  };
>> 
>> +struct ExportEntry {
>> +  uint64_t TerminalSize;
>> +  uint64_t NodeOffset;
>> +  std::string Name;
>> +  llvm::yaml::Hex64 Flags;
>> +  llvm::yaml::Hex64 Address;
>> +  llvm::yaml::Hex64 Other;
>> +  std::string ImportName;
>> +  std::vector<MachOYAML::ExportEntry> Children;
>> +};
>> +
>>  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;
>>  };
>> 
>>  struct Object {
>> @@ -95,6 +107,7 @@ LLVM_YAML_IS_SEQUENCE_VECTOR(llvm::yaml:
>>  LLVM_YAML_IS_SEQUENCE_VECTOR(int64_t)
>>  LLVM_YAML_IS_SEQUENCE_VECTOR(llvm::MachOYAML::RebaseOpcode)
>>  LLVM_YAML_IS_SEQUENCE_VECTOR(llvm::MachOYAML::BindOpcode)
>> +LLVM_YAML_IS_SEQUENCE_VECTOR(llvm::MachOYAML::ExportEntry)
>> 
>>  namespace llvm {
>>  namespace yaml {
>> @@ -123,6 +136,10 @@ template <> struct MappingTraits<MachOYA
>>    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::Section> {
>>    static void mapping(IO &IO, MachOYAML::Section &Section);
>>  };
>> 
>> Modified: llvm/trunk/lib/ObjectYAML/MachOYAML.cpp
>> URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/lib/ObjectYAML/MachOYAML.cpp?rev=271300&r1=271299&r2=271300&view=diff <http://llvm.org/viewvc/llvm-project/llvm/trunk/lib/ObjectYAML/MachOYAML.cpp?rev=271300&r1=271299&r2=271300&view=diff>
>> ==============================================================================
>> --- llvm/trunk/lib/ObjectYAML/MachOYAML.cpp (original)
>> +++ llvm/trunk/lib/ObjectYAML/MachOYAML.cpp Tue May 31 12:26:36 2016
>> @@ -103,6 +103,7 @@ void MappingTraits<MachOYAML::LinkEditDa
>>    IO.mapOptional("BindOpcodes", LinkEditData.BindOpcodes);
>>    IO.mapOptional("WeakBindOpcodes", LinkEditData.WeakBindOpcodes);
>>    IO.mapOptional("LazyBindOpcodes", LinkEditData.LazyBindOpcodes);
>> +  IO.mapOptional("ExportTrie", LinkEditData.ExportTrie);
>>  }
>> 
>>  void MappingTraits<MachOYAML::RebaseOpcode>::mapping(
>> @@ -121,6 +122,18 @@ void MappingTraits<MachOYAML::BindOpcode
>>    IO.mapOptional("Symbol", BindOpcode.Symbol);
>>  }
>> 
>> +void MappingTraits<MachOYAML::ExportEntry>::mapping(
>> +    IO &IO, MachOYAML::ExportEntry &ExportEntry) {
>> +  IO.mapRequired("TerminalSize", ExportEntry.TerminalSize);
>> +  IO.mapOptional("NodeOffset", ExportEntry.NodeOffset);
>> +  IO.mapOptional("Name", ExportEntry.Name);
>> +  IO.mapOptional("Flags", ExportEntry.Flags);
>> +  IO.mapOptional("Address", ExportEntry.Address);
>> +  IO.mapOptional("Other", ExportEntry.Other);
>> +  IO.mapOptional("ImportName", ExportEntry.ImportName);
>> +  IO.mapOptional("Children", ExportEntry.Children);
>> +}
>> +
>>  template <typename StructType>
>>  void mapLoadCommandData(IO &IO, MachOYAML::LoadCommand &LoadCommand) {}
>> 
>> @@ -141,6 +154,12 @@ void mapLoadCommandData<MachO::dylib_com
>>      IO &IO, MachOYAML::LoadCommand &LoadCommand) {
>>    IO.mapOptional("PayloadString", LoadCommand.PayloadString);
>>  }
>> +
>> +template <>
>> +void mapLoadCommandData<MachO::rpath_command>(
>> +    IO &IO, MachOYAML::LoadCommand &LoadCommand) {
>> +  IO.mapOptional("PayloadString", LoadCommand.PayloadString);
>> +}
>> 
>>  template <>
>>  void mapLoadCommandData<MachO::dylinker_command>(
>> 
>> Added: llvm/trunk/test/ObjectYAML/MachO/export_trie.yaml
>> URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/test/ObjectYAML/MachO/export_trie.yaml?rev=271300&view=auto <http://llvm.org/viewvc/llvm-project/llvm/trunk/test/ObjectYAML/MachO/export_trie.yaml?rev=271300&view=auto>
>> ==============================================================================
>> --- llvm/trunk/test/ObjectYAML/MachO/export_trie.yaml (added)
>> +++ llvm/trunk/test/ObjectYAML/MachO/export_trie.yaml Tue May 31 12:26:36 2016
>> @@ -0,0 +1,191 @@
>> +# RUN: yaml2obj -format=macho %s | obj2yaml | FileCheck %s
>> +
>> +--- !mach-o
>> +FileHeader:
>> +  magic:           0xFEEDFACF
>> +  cputype:         0x01000007
>> +  cpusubtype:      0x80000003
>> +  filetype:        0x00000002
>> +  ncmds:           16
>> +  sizeofcmds:      1408
>> +  flags:           0x00218085
>> +  reserved:        0x00000000
>> +LoadCommands:
>> +  - cmd:             LC_SEGMENT_64
>> +    cmdsize:         72
>> +    segname:         __PAGEZERO
>> +    vmaddr:          0
>> +    vmsize:          4294967296
>> +    fileoff:         0
>> +    filesize:        0
>> +    maxprot:         0
>> +    initprot:        0
>> +    nsects:          0
>> +    flags:           0
>> +  - cmd:             LC_SEGMENT_64
>> +    cmdsize:         552
>> +    segname:         __TEXT
>> +    vmaddr:          4294967296
>> +    vmsize:          8192
>> +    fileoff:         0
>> +    filesize:        8192
>> +    maxprot:         7
>> +    initprot:        5
>> +    nsects:          6
>> +    flags:           0
>> +  - cmd:             LC_SEGMENT_64
>> +    cmdsize:         312
>> +    segname:         __DATA
>> +    vmaddr:          4294975488
>> +    vmsize:          4096
>> +    fileoff:         8192
>> +    filesize:        4096
>> +    maxprot:         7
>> +    initprot:        3
>> +    nsects:          3
>> +    flags:           0
>> +  - cmd:             LC_SEGMENT_64
>> +    cmdsize:         72
>> +    segname:         __LINKEDIT
>> +    vmaddr:          4294979584
>> +    vmsize:          4096
>> +    fileoff:         12288
>> +    filesize:        2508
>> +    maxprot:         7
>> +    initprot:        1
>> +    nsects:          0
>> +    flags:           0
>> +  - cmd:             LC_DYLD_INFO_ONLY
>> +    cmdsize:         48
>> +    rebase_off:      12288
>> +    rebase_size:     8
>> +    bind_off:        12296
>> +    bind_size:       96
>> +    weak_bind_off:   0
>> +    weak_bind_size:  0
>> +    lazy_bind_off:   12392
>> +    lazy_bind_size:  624
>> +    export_off:      13016
>> +    export_size:     48
>> +  - cmd:             LC_SYMTAB
>> +    cmdsize:         24
>> +    symoff:          13080
>> +    nsyms:           30
>> +    stroff:          13700
>> +    strsize:         1096
>> +  - cmd:             LC_DYSYMTAB
>> +    cmdsize:         80
>> +    ilocalsym:       0
>> +    nlocalsym:       9
>> +    iextdefsym:      9
>> +    nextdefsym:      2
>> +    iundefsym:       11
>> +    nundefsym:       19
>> +    tocoff:          0
>> +    ntoc:            0
>> +    modtaboff:       0
>> +    nmodtab:         0
>> +    extrefsymoff:    0
>> +    nextrefsyms:     0
>> +    indirectsymoff:  13560
>> +    nindirectsyms:   35
>> +    extreloff:       0
>> +    nextrel:         0
>> +    locreloff:       0
>> +    nlocrel:         0
>> +  - cmd:             LC_LOAD_DYLINKER
>> +    cmdsize:         32
>> +    name:            12
>> +    PayloadString:   /usr/lib/dyld
>> +    ZeroPadBytes:    7
>> +  - cmd:             LC_UUID
>> +    cmdsize:         24
>> +    uuid:            461A1B28-822F-3F38-B670-645419E636F5
>> +  - cmd:             LC_VERSION_MIN_MACOSX
>> +    cmdsize:         16
>> +    version:         658176
>> +    sdk:             658176
>> +  - cmd:             LC_SOURCE_VERSION
>> +    cmdsize:         16
>> +    version:         0
>> +  - cmd:             LC_MAIN
>> +    cmdsize:         24
>> +    entryoff:        4448
>> +    stacksize:       0
>> +  - cmd:             LC_LOAD_DYLIB
>> +    cmdsize:         48
>> +    dylib:
>> +      name:            24
>> +      timestamp:       2
>> +      current_version: 7864576
>> +      compatibility_version: 65536
>> +    PayloadString:   '/usr/lib/libc++.1.dylib'
>> +    ZeroPadBytes:    1
>> +  - cmd:             LC_LOAD_DYLIB
>> +    cmdsize:         56
>> +    dylib:
>> +      name:            24
>> +      timestamp:       2
>> +      current_version: 80349697
>> +      compatibility_version: 65536
>> +    PayloadString:   /usr/lib/libSystem.B.dylib
>> +    ZeroPadBytes:    6
>> +  - cmd:             LC_FUNCTION_STARTS
>> +    cmdsize:         16
>> +    dataoff:         13064
>> +    datasize:        16
>> +  - cmd:             LC_DATA_IN_CODE
>> +    cmdsize:         16
>> +    dataoff:         13080
>> +    datasize:        0
>> +LinkEditData:
>> +  ExportTrie:
>> +    TerminalSize:    0
>> +    NodeOffset:      0
>> +    Name:            ''
>> +    Flags:           0x0000000000000000
>> +    Address:         0x0000000000000000
>> +    Other:           0x0000000000000000
>> +    ImportName:      ''
>> +    Children:
>> +      - TerminalSize:    0
>> +        NodeOffset:      5
>> +        Name:            _
>> +        Flags:           0x0000000000000000
>> +        Address:         0x0000000000000000
>> +        Other:           0x0000000000000000
>> +        ImportName:      ''
>> +        Children:
>> +          - TerminalSize:    2
>> +            NodeOffset:      33
>> +            Name:            _mh_execute_header
>> +            Flags:           0x0000000000000000
>> +            Address:         0x0000000000000000
>> +            Other:           0x0000000000000000
>> +            ImportName:      ''
>> +          - TerminalSize:    3
>> +            NodeOffset:      37
>> +            Name:            main
>> +            Flags:           0x0000000000000000
>> +            Address:         0x0000000000001160
>> +            Other:           0x0000000000000000
>> +            ImportName:      ''
>> +...
>> +
>> +#CHECK: ExportTrie:
>> +#CHECK:     TerminalSize:    0
>> +#CHECK:     NodeOffset:      0
>> +#CHECK:     Name:            ''
>> +#CHECK:     Children:
>> +#CHECK:       - TerminalSize:    0
>> +#CHECK:         NodeOffset:      5
>> +#CHECK:         Name:            _
>> +#CHECK:         Children:
>> +#CHECK:           - TerminalSize:    2
>> +#CHECK:             NodeOffset:      33
>> +#CHECK:             Name:            _mh_execute_header
>> +#CHECK:             Address:         0x0000000000000000
>> +#CHECK:           - TerminalSize:    3
>> +#CHECK:             NodeOffset:      37
>> +#CHECK:             Name:            main
>> +#CHECK:             Address:         0x0000000000001160
>> 
>> Modified: llvm/trunk/tools/obj2yaml/macho2yaml.cpp
>> URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/tools/obj2yaml/macho2yaml.cpp?rev=271300&r1=271299&r2=271300&view=diff <http://llvm.org/viewvc/llvm-project/llvm/trunk/tools/obj2yaml/macho2yaml.cpp?rev=271300&r1=271299&r2=271300&view=diff>
>> ==============================================================================
>> --- llvm/trunk/tools/obj2yaml/macho2yaml.cpp (original)
>> +++ llvm/trunk/tools/obj2yaml/macho2yaml.cpp Tue May 31 12:26:36 2016
>> @@ -32,6 +32,7 @@ class MachODumper {
>>    void dumpRebaseOpcodes(std::unique_ptr<MachOYAML::Object> &Y);
>>    void dumpBindOpcodes(std::vector<MachOYAML::BindOpcode> &BindOpcodes,
>>                         ArrayRef<uint8_t> OpcodeBuffer, bool Lazy = false);
>> +  void dumpExportTrie(std::unique_ptr<MachOYAML::Object> &Y);
>> 
>>  public:
>>    MachODumper(const object::MachOObjectFile &O) : Obj(O) {}
>> @@ -149,6 +150,13 @@ const char *MachODumper::processLoadComm
>>    return readString<MachO::dylinker_command>(LC, LoadCmd);
>>  }
>> 
>> +template <>
>> +const char *MachODumper::processLoadCommandData<MachO::rpath_command>(
>> +    MachOYAML::LoadCommand &LC,
>> +    const llvm::object::MachOObjectFile::LoadCommandInfo &LoadCmd) {
>> +  return readString<MachO::rpath_command>(LC, LoadCmd);
>> +}
>> +
>>  Expected<std::unique_ptr<MachOYAML::Object>> MachODumper::dump() {
>>    auto Y = make_unique<MachOYAML::Object>();
>>    dumpHeader(Y);
>> @@ -199,8 +207,9 @@ void MachODumper::dumpLinkEdit(std::uniq
>>    dumpBindOpcodes(Y->LinkEdit.BindOpcodes, Obj.getDyldInfoBindOpcodes());
>>    dumpBindOpcodes(Y->LinkEdit.WeakBindOpcodes,
>>                    Obj.getDyldInfoWeakBindOpcodes());
>> -  dumpBindOpcodes(Y->LinkEdit.LazyBindOpcodes,
>> -                  Obj.getDyldInfoLazyBindOpcodes(), true);
>> +  dumpBindOpcodes(Y->LinkEdit.LazyBindOpcodes, Obj.getDyldInfoLazyBindOpcodes(),
>> +                  true);
>> +  dumpExportTrie(Y);
>>  }
>> 
>>  void MachODumper::dumpRebaseOpcodes(std::unique_ptr<MachOYAML::Object> &Y) {
>> @@ -244,6 +253,13 @@ void MachODumper::dumpRebaseOpcodes(std:
>>    }
>>  }
>> 
>> +StringRef ReadStringRef(const uint8_t *Start) {
>> +  const uint8_t *Itr = Start;
>> +  for (; *Itr; ++Itr)
>> +    ;
>> +  return StringRef(reinterpret_cast<const char *>(Start), Itr - Start);
>> +}
>> +
>>  void MachODumper::dumpBindOpcodes(
>>      std::vector<MachOYAML::BindOpcode> &BindOpcodes,
>>      ArrayRef<uint8_t> OpcodeBuffer, bool Lazy) {
>> @@ -257,7 +273,6 @@ void MachODumper::dumpBindOpcodes(
>>      unsigned Count;
>>      uint64_t ULEB = 0;
>>      int64_t SLEB = 0;
>> -    const uint8_t *SymStart;
>> 
>>      switch (BindOp.Opcode) {
>>      case MachO::BIND_OPCODE_DO_BIND_ULEB_TIMES_SKIPPING_ULEB:
>> @@ -282,12 +297,8 @@ void MachODumper::dumpBindOpcodes(
>>        break;
>> 
>>      case MachO::BIND_OPCODE_SET_SYMBOL_TRAILING_FLAGS_IMM:
>> -      SymStart = ++OpCode;
>> -      while (*OpCode) {
>> -        ++OpCode;
>> -      }
>> -      BindOp.Symbol = StringRef(reinterpret_cast<const char *>(SymStart),
>> -                                OpCode - SymStart);
>> +      BindOp.Symbol = ReadStringRef(OpCode + 1);
>> +      OpCode += BindOp.Symbol.size() + 1;
>>        break;
>>      default:
>>        break;
>> @@ -302,6 +313,117 @@ void MachODumper::dumpBindOpcodes(
>>    }
>>  }
>> 
>> +/*!
>> + * /brief processes a node from the export trie, and its children.
>> + *
>> + * To my knowledge there is no documentation of the encoded format of this data
>> + * other than in the heads of the Apple linker engineers. To that end hopefully
>> + * this comment and the implementation below can serve to light the way for
>> + * anyone crazy enough to come down this path in the future.
>> + *
>> + * This function reads and preserves the trie structure of the export trie. To
>> + * my knowledge there is no code anywhere else that reads the data and preserves
>> + * the Trie. LD64 (sources available at opensource.apple.com <http://opensource.apple.com/>) has a similar
>> + * implementation that parses the export trie into a vector. That code as well
>> + * as LLVM's libObject MachO implementation were the basis for this.
>> + *
>> + * The export trie is an encoded trie. The node serialization is a bit awkward.
>> + * The below pseudo-code is the best description I've come up with for it.
>> + *
>> + * struct SerializedNode {
>> + *   ULEB128 TerminalSize;
>> + *   struct TerminalData { <-- This is only present if TerminalSize > 0
>> + *     ULEB128 Flags;
>> + *     ULEB128 Address; <-- Present if (! Flags & REEXPORT )
>> + *     ULEB128 Other; <-- Present if ( Flags & REEXPORT ||
>> + *                                     Flags & STUB_AND_RESOLVER )
>> + *     char[] ImportName; <-- Present if ( Flags & REEXPORT )
>> + *   }
>> + *   uint8_t ChildrenCount;
>> + *   Pair<char[], ULEB128> ChildNameOffsetPair[ChildrenCount];
>> + *   SerializedNode Children[ChildrenCount]
>> + * }
>> + *
>> + * Terminal nodes are nodes that represent actual exports. They can appear
>> + * anywhere in the tree other than at the root; they do not need to be leaf
>> + * nodes. When reading the data out of the trie this routine reads it in-order,
>> + * but it puts the child names and offsets directly into the child nodes. This
>> + * results in looping over the children twice during serialization and
>> + * de-serialization, but it makes the YAML representation more human readable.
>> + *
>> + * Below is an example of the graph from a "Hello World" executable:
>> + *
>> + * -------
>> + * | ''  |
>> + * -------
>> + *    |
>> + * -------
>> + * | '_' |
>> + * -------
>> + *    |
>> + *    |----------------------------------------|
>> + *    |                                        |
>> + *  ------------------------      ---------------------
>> + *  | '_mh_execute_header' |      | 'main'            |
>> + *  | Flags: 0x00000000    |      | Flags: 0x00000000 |
>> + *  | Addr:  0x00000000    |      | Addr:  0x00001160 |
>> + *  ------------------------      ---------------------
>> + *
>> + * This graph represents the trie for the exports "__mh_execute_header" and
>> + * "_main". In the graph only the "_main" and "__mh_execute_header" nodes are
>> + * terminal.
>> +*/
>> +
>> +const uint8_t *processExportNode(const uint8_t *CurrPtr,
>> +                                 const uint8_t *const End,
>> +                                 MachOYAML::ExportEntry &Entry) {
>> +  if (CurrPtr >= End)
>> +    return CurrPtr;
>> +  unsigned Count = 0;
>> +  Entry.TerminalSize = decodeULEB128(CurrPtr, &Count);
>> +  CurrPtr += Count;
>> +  if (Entry.TerminalSize != 0) {
>> +    Entry.Flags = decodeULEB128(CurrPtr, &Count);
>> +    CurrPtr += Count;
>> +    if (Entry.Flags & MachO::EXPORT_SYMBOL_FLAGS_REEXPORT) {
>> +      Entry.Address = 0;
>> +      Entry.Other = decodeULEB128(CurrPtr, &Count);
>> +      CurrPtr += Count;
>> +      Entry.ImportName = std::string(reinterpret_cast<const char *>(CurrPtr));
>> +    } else {
>> +      Entry.Address = decodeULEB128(CurrPtr, &Count);
>> +      CurrPtr += Count;
>> +      if (Entry.Flags & MachO::EXPORT_SYMBOL_FLAGS_STUB_AND_RESOLVER) {
>> +        Entry.Other = decodeULEB128(CurrPtr, &Count);
>> +        CurrPtr += Count;
>> +      } else
>> +        Entry.Other = 0;
>> +    }
>> +  }
>> +  uint8_t childrenCount = *CurrPtr++;
>> +  if (childrenCount == 0)
>> +    return CurrPtr;
>> +
>> +  Entry.Children.insert(Entry.Children.begin(), (size_t)childrenCount,
>> +                        MachOYAML::ExportEntry());
>> +  for (auto &Child : Entry.Children) {
>> +    Child.Name = std::string(reinterpret_cast<const char *>(CurrPtr));
>> +    CurrPtr += Child.Name.length() + 1;
>> +    Child.NodeOffset = decodeULEB128(CurrPtr, &Count);
>> +    CurrPtr += Count;
>> +  }
>> +  for (auto &Child : Entry.Children) {
>> +    CurrPtr = processExportNode(CurrPtr, End, Child);
>> +  }
>> +  return CurrPtr;
>> +}
>> +
>> +void MachODumper::dumpExportTrie(std::unique_ptr<MachOYAML::Object> &Y) {
>> +  MachOYAML::LinkEditData &LEData = Y->LinkEdit;
>> +  auto ExportsTrie = Obj.getDyldInfoExportsTrie();
>> +  processExportNode(ExportsTrie.begin(), ExportsTrie.end(), LEData.ExportTrie);
>> +}
>> +
>>  Error macho2yaml(raw_ostream &Out, const object::MachOObjectFile &Obj) {
>>    MachODumper Dumper(Obj);
>>    Expected<std::unique_ptr<MachOYAML::Object>> YAML = Dumper.dump();
>> 
>> Modified: llvm/trunk/tools/yaml2obj/yaml2macho.cpp
>> URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/tools/yaml2obj/yaml2macho.cpp?rev=271300&r1=271299&r2=271300&view=diff <http://llvm.org/viewvc/llvm-project/llvm/trunk/tools/yaml2obj/yaml2macho.cpp?rev=271300&r1=271299&r2=271300&view=diff>
>> ==============================================================================
>> --- llvm/trunk/tools/yaml2obj/yaml2macho.cpp (original)
>> +++ llvm/trunk/tools/yaml2obj/yaml2macho.cpp Tue May 31 12:26:36 2016
>> @@ -48,7 +48,9 @@ private:
>>    Error writeLinkEditData(raw_ostream &OS);
>>    void writeBindOpcodes(raw_ostream &OS, uint64_t offset,
>>                          std::vector<MachOYAML::BindOpcode> &BindOpcodes);
>> +  Error writeExportTrie(raw_ostream &OS);
>> 
>> +  void dumpExportEntry(raw_ostream &OS, MachOYAML::ExportEntry &Entry);
>>    void ZeroToOffset(raw_ostream &OS, size_t offset);
>> 
>>    MachOYAML::Object &Obj;
>> @@ -161,6 +163,12 @@ size_t writeLoadCommandData<MachO::dylin
>>    return writePayloadString(LC, OS);
>>  }
>> 
>> +template <>
>> +size_t writeLoadCommandData<MachO::rpath_command>(MachOYAML::LoadCommand &LC,
>> +                                                     raw_ostream &OS) {
>> +  return writePayloadString(LC, OS);
>> +}
>> +
>>  void ZeroFillBytes(raw_ostream &OS, size_t Size) {
>>    std::vector<uint8_t> FillData;
>>    FillData.insert(FillData.begin(), Size, 0);
>> @@ -282,9 +290,39 @@ void MachOWriter::writeBindOpcodes(
>>      }
>>      if (!Opcode.Symbol.empty()) {
>>        OS.write(Opcode.Symbol.data(), Opcode.Symbol.size());
>> -      OS.write("\0", 1);
>> +      OS.write('\0');
>> +    }
>> +  }
>> +}
>> +
>> +void MachOWriter::dumpExportEntry(raw_ostream &OS, MachOYAML::ExportEntry &Entry) {
>> +  encodeSLEB128(Entry.TerminalSize, OS);
>> +  if (Entry.TerminalSize > 0) {
>> +    encodeSLEB128(Entry.Flags, OS);
>> +    if ( Entry.Flags & MachO::EXPORT_SYMBOL_FLAGS_REEXPORT ) {
>> +      encodeSLEB128(Entry.Other, OS);
>> +      OS << Entry.ImportName;
>> +      OS.write('\0');
>> +    }
>> +    else {
>> +      encodeSLEB128(Entry.Address, OS);
>> +      if (Entry.Flags & MachO::EXPORT_SYMBOL_FLAGS_STUB_AND_RESOLVER)
>> +        encodeSLEB128(Entry.Other, OS);
>>      }
>>    }
>> +  OS.write(static_cast<uint8_t>(Entry.Children.size()));
>> +  for (auto EE : Entry.Children){
>> +    OS << EE.Name;
>> +    OS.write('\0');
>> +    encodeSLEB128(EE.NodeOffset, OS);
>> +  }
>> +  for (auto EE : Entry.Children)
>> +    dumpExportEntry(OS, EE);
>> +}
>> +
>> +Error MachOWriter::writeExportTrie(raw_ostream &OS) {
>> +  dumpExportEntry(OS, Obj.LinkEdit.ExportTrie);
>> +  return Error::success();
>>  }
>> 
>>  Error MachOWriter::writeLinkEditData(raw_ostream &OS) {
>> @@ -318,6 +356,10 @@ Error MachOWriter::writeLinkEditData(raw
>>    writeBindOpcodes(OS, DyldInfoOnlyCmd->lazy_bind_off,
>>                     LinkEdit.LazyBindOpcodes);
>> 
>> +  ZeroToOffset(OS, DyldInfoOnlyCmd->export_off);
>> +  if(auto Err = writeExportTrie(OS))
>> +    return Err;
>> +
>>    // Fill to the end of the string table
>>    ZeroToOffset(OS, SymtabCmd->stroff + SymtabCmd->strsize);
>> 
>> 
>> 
>> _______________________________________________
>> llvm-commits mailing list
>> llvm-commits at lists.llvm.org <mailto:llvm-commits at lists.llvm.org>
>> http://lists.llvm.org/cgi-bin/mailman/listinfo/llvm-commits <http://lists.llvm.org/cgi-bin/mailman/listinfo/llvm-commits>
>> 
> 
> _______________________________________________
> llvm-commits mailing list
> llvm-commits at lists.llvm.org
> http://lists.llvm.org/cgi-bin/mailman/listinfo/llvm-commits

-------------- next part --------------
An HTML attachment was scrubbed...
URL: <http://lists.llvm.org/pipermail/llvm-commits/attachments/20160531/dbee3ae6/attachment-0001.html>


More information about the llvm-commits mailing list