1 //===- MachOObject.h - Mach-O Object File Wrapper ---------------*- C++ -*-===//
3 // The LLVM Compiler Infrastructure
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
8 //===----------------------------------------------------------------------===//
10 #ifndef LLVM_OBJECT_MACHOOBJECT_H
11 #define LLVM_OBJECT_MACHOOBJECT_H
14 #include "llvm/ADT/InMemoryStruct.h"
15 #include "llvm/ADT/OwningPtr.h"
16 #include "llvm/ADT/StringRef.h"
17 #include "llvm/Object/MachOFormat.h"
25 /// \brief Wrapper object for manipulating Mach-O object files.
27 /// This class is designed to implement a full-featured, efficient, portable,
28 /// and robust Mach-O interface to Mach-O object files. It does not attempt to
29 /// smooth over rough edges in the Mach-O format or generalize access to object
30 /// independent features.
32 /// The class is designed around accessing the Mach-O object which is expected
33 /// to be fully loaded into memory.
35 /// This class is *not* suitable for concurrent use. For efficient operation,
36 /// the class uses APIs which rely on the ability to cache the results of
37 /// certain calls in internal objects which are not safe for concurrent
38 /// access. This allows the API to be zero-copy on the common paths.
40 // FIXME: It would be cool if we supported a "paged" MemoryBuffer
41 // implementation. This would allow us to implement a more sensible version of
42 // MemoryObject which can work like a MemoryBuffer, but be more efficient for
43 // objects which are in the current address space.
46 struct LoadCommandInfo {
47 /// The load command information.
48 macho::LoadCommand Command;
50 /// The offset to the start of the load command in memory.
55 OwningPtr<MemoryBuffer> Buffer;
57 /// Whether the object is little endian.
59 /// Whether the object is 64-bit.
61 /// Whether the object is swapped endianness from the host.
63 /// Whether the string table has been registered.
66 /// The cached information on the load commands.
67 LoadCommandInfo *LoadCommands;
68 mutable unsigned NumLoadedCommands;
70 /// The cached copy of the header.
72 macho::Header64Ext Header64Ext;
74 /// Cache string table information.
75 StringRef StringTable;
78 MachOObject(MemoryBuffer *Buffer, bool IsLittleEndian, bool Is64Bit);
83 /// \brief Load a Mach-O object from a MemoryBuffer object.
85 /// \param Buffer - The buffer to load the object from. This routine takes
86 /// exclusive ownership of the buffer (which is passed to the returned object
88 /// \param ErrorStr [out] - If given, will be set to a user readable error
89 /// message on failure.
90 /// \returns The loaded object, or null on error.
91 static MachOObject *LoadFromBuffer(MemoryBuffer *Buffer,
92 std::string *ErrorStr = 0);
94 /// @name File Information
97 bool isLittleEndian() const { return IsLittleEndian; }
98 bool isSwappedEndian() const { return IsSwappedEndian; }
99 bool is64Bit() const { return Is64Bit; }
101 unsigned getHeaderSize() const {
102 return Is64Bit ? macho::Header64Size : macho::Header32Size;
106 /// @name String Table Data
109 StringRef getStringTableData() const {
110 assert(HasStringTable && "String table has not been registered!");
114 void RegisterStringTable(macho::SymtabLoadCommand &SLC);
117 /// @name Object Header Access
120 const macho::Header &getHeader() const { return Header; }
121 const macho::Header64Ext &getHeader64Ext() const {
122 assert(is64Bit() && "Invalid access!");
127 /// @name Object Structure Access
130 /// \brief Retrieve the information for the given load command.
131 const LoadCommandInfo &getLoadCommandInfo(unsigned Index) const;
133 void ReadSegmentLoadCommand(
134 const LoadCommandInfo &LCI,
135 InMemoryStruct<macho::SegmentLoadCommand> &Res) const;
136 void ReadSegment64LoadCommand(
137 const LoadCommandInfo &LCI,
138 InMemoryStruct<macho::Segment64LoadCommand> &Res) const;
139 void ReadSymtabLoadCommand(
140 const LoadCommandInfo &LCI,
141 InMemoryStruct<macho::SymtabLoadCommand> &Res) const;
142 void ReadDysymtabLoadCommand(
143 const LoadCommandInfo &LCI,
144 InMemoryStruct<macho::DysymtabLoadCommand> &Res) const;
145 void ReadIndirectSymbolTableEntry(
146 const macho::DysymtabLoadCommand &DLC,
148 InMemoryStruct<macho::IndirectSymbolTableEntry> &Res) const;
150 const LoadCommandInfo &LCI,
152 InMemoryStruct<macho::Section> &Res) const;
154 const LoadCommandInfo &LCI,
156 InMemoryStruct<macho::Section64> &Res) const;
157 void ReadRelocationEntry(
158 uint64_t RelocationTableOffset, unsigned Index,
159 InMemoryStruct<macho::RelocationEntry> &Res) const;
164 } // end namespace object
165 } // end namespace llvm