Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame^] | 1 | //===- MachO.h - MachO object file 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 | // This file declares the MachOObjectFile class, which implement the ObjectFile |
| 11 | // interface for MachO files. |
| 12 | // |
| 13 | //===----------------------------------------------------------------------===// |
| 14 | |
| 15 | #ifndef LLVM_OBJECT_MACHO_H |
| 16 | #define LLVM_OBJECT_MACHO_H |
| 17 | |
| 18 | #include "llvm/ADT/ArrayRef.h" |
| 19 | #include "llvm/ADT/SmallString.h" |
| 20 | #include "llvm/ADT/SmallVector.h" |
| 21 | #include "llvm/ADT/StringExtras.h" |
| 22 | #include "llvm/ADT/StringRef.h" |
| 23 | #include "llvm/ADT/Triple.h" |
| 24 | #include "llvm/ADT/iterator_range.h" |
| 25 | #include "llvm/BinaryFormat/MachO.h" |
| 26 | #include "llvm/MC/SubtargetFeature.h" |
| 27 | #include "llvm/Object/Binary.h" |
| 28 | #include "llvm/Object/ObjectFile.h" |
| 29 | #include "llvm/Object/SymbolicFile.h" |
| 30 | #include "llvm/Support/Error.h" |
| 31 | #include "llvm/Support/Format.h" |
| 32 | #include "llvm/Support/MemoryBuffer.h" |
| 33 | #include "llvm/Support/raw_ostream.h" |
| 34 | #include <cstdint> |
| 35 | #include <memory> |
| 36 | #include <string> |
| 37 | #include <system_error> |
| 38 | |
| 39 | namespace llvm { |
| 40 | namespace object { |
| 41 | |
| 42 | /// DiceRef - This is a value type class that represents a single |
| 43 | /// data in code entry in the table in a Mach-O object file. |
| 44 | class DiceRef { |
| 45 | DataRefImpl DicePimpl; |
| 46 | const ObjectFile *OwningObject = nullptr; |
| 47 | |
| 48 | public: |
| 49 | DiceRef() = default; |
| 50 | DiceRef(DataRefImpl DiceP, const ObjectFile *Owner); |
| 51 | |
| 52 | bool operator==(const DiceRef &Other) const; |
| 53 | bool operator<(const DiceRef &Other) const; |
| 54 | |
| 55 | void moveNext(); |
| 56 | |
| 57 | std::error_code getOffset(uint32_t &Result) const; |
| 58 | std::error_code getLength(uint16_t &Result) const; |
| 59 | std::error_code getKind(uint16_t &Result) const; |
| 60 | |
| 61 | DataRefImpl getRawDataRefImpl() const; |
| 62 | const ObjectFile *getObjectFile() const; |
| 63 | }; |
| 64 | using dice_iterator = content_iterator<DiceRef>; |
| 65 | |
| 66 | /// ExportEntry encapsulates the current-state-of-the-walk used when doing a |
| 67 | /// non-recursive walk of the trie data structure. This allows you to iterate |
| 68 | /// across all exported symbols using: |
| 69 | /// Error Err; |
| 70 | /// for (const llvm::object::ExportEntry &AnExport : Obj->exports(&Err)) { |
| 71 | /// } |
| 72 | /// if (Err) { report error ... |
| 73 | class ExportEntry { |
| 74 | public: |
| 75 | ExportEntry(Error *Err, const MachOObjectFile *O, ArrayRef<uint8_t> Trie); |
| 76 | |
| 77 | StringRef name() const; |
| 78 | uint64_t flags() const; |
| 79 | uint64_t address() const; |
| 80 | uint64_t other() const; |
| 81 | StringRef otherName() const; |
| 82 | uint32_t nodeOffset() const; |
| 83 | |
| 84 | bool operator==(const ExportEntry &) const; |
| 85 | |
| 86 | void moveNext(); |
| 87 | |
| 88 | private: |
| 89 | friend class MachOObjectFile; |
| 90 | |
| 91 | void moveToFirst(); |
| 92 | void moveToEnd(); |
| 93 | uint64_t readULEB128(const uint8_t *&p, const char **error); |
| 94 | void pushDownUntilBottom(); |
| 95 | void pushNode(uint64_t Offset); |
| 96 | |
| 97 | // Represents a node in the mach-o exports trie. |
| 98 | struct NodeState { |
| 99 | NodeState(const uint8_t *Ptr); |
| 100 | |
| 101 | const uint8_t *Start; |
| 102 | const uint8_t *Current; |
| 103 | uint64_t Flags = 0; |
| 104 | uint64_t Address = 0; |
| 105 | uint64_t Other = 0; |
| 106 | const char *ImportName = nullptr; |
| 107 | unsigned ChildCount = 0; |
| 108 | unsigned NextChildIndex = 0; |
| 109 | unsigned ParentStringLength = 0; |
| 110 | bool IsExportNode = false; |
| 111 | }; |
| 112 | using NodeList = SmallVector<NodeState, 16>; |
| 113 | using node_iterator = NodeList::const_iterator; |
| 114 | |
| 115 | Error *E; |
| 116 | const MachOObjectFile *O; |
| 117 | ArrayRef<uint8_t> Trie; |
| 118 | SmallString<256> CumulativeString; |
| 119 | NodeList Stack; |
| 120 | bool Done = false; |
| 121 | |
| 122 | iterator_range<node_iterator> nodes() const { |
| 123 | return make_range(Stack.begin(), Stack.end()); |
| 124 | } |
| 125 | }; |
| 126 | using export_iterator = content_iterator<ExportEntry>; |
| 127 | |
| 128 | // Segment info so SegIndex/SegOffset pairs in a Mach-O Bind or Rebase entry |
| 129 | // can be checked and translated. Only the SegIndex/SegOffset pairs from |
| 130 | // checked entries are to be used with the segmentName(), sectionName() and |
| 131 | // address() methods below. |
| 132 | class BindRebaseSegInfo { |
| 133 | public: |
| 134 | BindRebaseSegInfo(const MachOObjectFile *Obj); |
| 135 | |
| 136 | // Used to check a Mach-O Bind or Rebase entry for errors when iterating. |
| 137 | const char *checkSegAndOffset(int32_t SegIndex, uint64_t SegOffset, |
| 138 | bool endInvalid); |
| 139 | const char *checkCountAndSkip(uint32_t Count, uint32_t Skip, |
| 140 | uint8_t PointerSize, int32_t SegIndex, |
| 141 | uint64_t SegOffset); |
| 142 | // Used with valid SegIndex/SegOffset values from checked entries. |
| 143 | StringRef segmentName(int32_t SegIndex); |
| 144 | StringRef sectionName(int32_t SegIndex, uint64_t SegOffset); |
| 145 | uint64_t address(uint32_t SegIndex, uint64_t SegOffset); |
| 146 | |
| 147 | private: |
| 148 | struct SectionInfo { |
| 149 | uint64_t Address; |
| 150 | uint64_t Size; |
| 151 | StringRef SectionName; |
| 152 | StringRef SegmentName; |
| 153 | uint64_t OffsetInSegment; |
| 154 | uint64_t SegmentStartAddress; |
| 155 | int32_t SegmentIndex; |
| 156 | }; |
| 157 | const SectionInfo &findSection(int32_t SegIndex, uint64_t SegOffset); |
| 158 | |
| 159 | SmallVector<SectionInfo, 32> Sections; |
| 160 | int32_t MaxSegIndex; |
| 161 | }; |
| 162 | |
| 163 | /// MachORebaseEntry encapsulates the current state in the decompression of |
| 164 | /// rebasing opcodes. This allows you to iterate through the compressed table of |
| 165 | /// rebasing using: |
| 166 | /// Error Err; |
| 167 | /// for (const llvm::object::MachORebaseEntry &Entry : Obj->rebaseTable(&Err)) { |
| 168 | /// } |
| 169 | /// if (Err) { report error ... |
| 170 | class MachORebaseEntry { |
| 171 | public: |
| 172 | MachORebaseEntry(Error *Err, const MachOObjectFile *O, |
| 173 | ArrayRef<uint8_t> opcodes, bool is64Bit); |
| 174 | |
| 175 | int32_t segmentIndex() const; |
| 176 | uint64_t segmentOffset() const; |
| 177 | StringRef typeName() const; |
| 178 | StringRef segmentName() const; |
| 179 | StringRef sectionName() const; |
| 180 | uint64_t address() const; |
| 181 | |
| 182 | bool operator==(const MachORebaseEntry &) const; |
| 183 | |
| 184 | void moveNext(); |
| 185 | |
| 186 | private: |
| 187 | friend class MachOObjectFile; |
| 188 | |
| 189 | void moveToFirst(); |
| 190 | void moveToEnd(); |
| 191 | uint64_t readULEB128(const char **error); |
| 192 | |
| 193 | Error *E; |
| 194 | const MachOObjectFile *O; |
| 195 | ArrayRef<uint8_t> Opcodes; |
| 196 | const uint8_t *Ptr; |
| 197 | uint64_t SegmentOffset = 0; |
| 198 | int32_t SegmentIndex = -1; |
| 199 | uint64_t RemainingLoopCount = 0; |
| 200 | uint64_t AdvanceAmount = 0; |
| 201 | uint8_t RebaseType = 0; |
| 202 | uint8_t PointerSize; |
| 203 | bool Done = false; |
| 204 | }; |
| 205 | using rebase_iterator = content_iterator<MachORebaseEntry>; |
| 206 | |
| 207 | /// MachOBindEntry encapsulates the current state in the decompression of |
| 208 | /// binding opcodes. This allows you to iterate through the compressed table of |
| 209 | /// bindings using: |
| 210 | /// Error Err; |
| 211 | /// for (const llvm::object::MachOBindEntry &Entry : Obj->bindTable(&Err)) { |
| 212 | /// } |
| 213 | /// if (Err) { report error ... |
| 214 | class MachOBindEntry { |
| 215 | public: |
| 216 | enum class Kind { Regular, Lazy, Weak }; |
| 217 | |
| 218 | MachOBindEntry(Error *Err, const MachOObjectFile *O, |
| 219 | ArrayRef<uint8_t> Opcodes, bool is64Bit, MachOBindEntry::Kind); |
| 220 | |
| 221 | int32_t segmentIndex() const; |
| 222 | uint64_t segmentOffset() const; |
| 223 | StringRef typeName() const; |
| 224 | StringRef symbolName() const; |
| 225 | uint32_t flags() const; |
| 226 | int64_t addend() const; |
| 227 | int ordinal() const; |
| 228 | |
| 229 | StringRef segmentName() const; |
| 230 | StringRef sectionName() const; |
| 231 | uint64_t address() const; |
| 232 | |
| 233 | bool operator==(const MachOBindEntry &) const; |
| 234 | |
| 235 | void moveNext(); |
| 236 | |
| 237 | private: |
| 238 | friend class MachOObjectFile; |
| 239 | |
| 240 | void moveToFirst(); |
| 241 | void moveToEnd(); |
| 242 | uint64_t readULEB128(const char **error); |
| 243 | int64_t readSLEB128(const char **error); |
| 244 | |
| 245 | Error *E; |
| 246 | const MachOObjectFile *O; |
| 247 | ArrayRef<uint8_t> Opcodes; |
| 248 | const uint8_t *Ptr; |
| 249 | uint64_t SegmentOffset = 0; |
| 250 | int32_t SegmentIndex = -1; |
| 251 | StringRef SymbolName; |
| 252 | bool LibraryOrdinalSet = false; |
| 253 | int Ordinal = 0; |
| 254 | uint32_t Flags = 0; |
| 255 | int64_t Addend = 0; |
| 256 | uint64_t RemainingLoopCount = 0; |
| 257 | uint64_t AdvanceAmount = 0; |
| 258 | uint8_t BindType = 0; |
| 259 | uint8_t PointerSize; |
| 260 | Kind TableKind; |
| 261 | bool Done = false; |
| 262 | }; |
| 263 | using bind_iterator = content_iterator<MachOBindEntry>; |
| 264 | |
| 265 | class MachOObjectFile : public ObjectFile { |
| 266 | public: |
| 267 | struct LoadCommandInfo { |
| 268 | const char *Ptr; // Where in memory the load command is. |
| 269 | MachO::load_command C; // The command itself. |
| 270 | }; |
| 271 | using LoadCommandList = SmallVector<LoadCommandInfo, 4>; |
| 272 | using load_command_iterator = LoadCommandList::const_iterator; |
| 273 | |
| 274 | static Expected<std::unique_ptr<MachOObjectFile>> |
| 275 | create(MemoryBufferRef Object, bool IsLittleEndian, bool Is64Bits, |
| 276 | uint32_t UniversalCputype = 0, uint32_t UniversalIndex = 0); |
| 277 | |
| 278 | void moveSymbolNext(DataRefImpl &Symb) const override; |
| 279 | |
| 280 | uint64_t getNValue(DataRefImpl Sym) const; |
| 281 | Expected<StringRef> getSymbolName(DataRefImpl Symb) const override; |
| 282 | |
| 283 | // MachO specific. |
| 284 | Error checkSymbolTable() const; |
| 285 | |
| 286 | std::error_code getIndirectName(DataRefImpl Symb, StringRef &Res) const; |
| 287 | unsigned getSectionType(SectionRef Sec) const; |
| 288 | |
| 289 | Expected<uint64_t> getSymbolAddress(DataRefImpl Symb) const override; |
| 290 | uint32_t getSymbolAlignment(DataRefImpl Symb) const override; |
| 291 | uint64_t getCommonSymbolSizeImpl(DataRefImpl Symb) const override; |
| 292 | Expected<SymbolRef::Type> getSymbolType(DataRefImpl Symb) const override; |
| 293 | uint32_t getSymbolFlags(DataRefImpl Symb) const override; |
| 294 | Expected<section_iterator> getSymbolSection(DataRefImpl Symb) const override; |
| 295 | unsigned getSymbolSectionID(SymbolRef Symb) const; |
| 296 | unsigned getSectionID(SectionRef Sec) const; |
| 297 | |
| 298 | void moveSectionNext(DataRefImpl &Sec) const override; |
| 299 | std::error_code getSectionName(DataRefImpl Sec, |
| 300 | StringRef &Res) const override; |
| 301 | uint64_t getSectionAddress(DataRefImpl Sec) const override; |
| 302 | uint64_t getSectionIndex(DataRefImpl Sec) const override; |
| 303 | uint64_t getSectionSize(DataRefImpl Sec) const override; |
| 304 | std::error_code getSectionContents(DataRefImpl Sec, |
| 305 | StringRef &Res) const override; |
| 306 | uint64_t getSectionAlignment(DataRefImpl Sec) const override; |
| 307 | bool isSectionCompressed(DataRefImpl Sec) const override; |
| 308 | bool isSectionText(DataRefImpl Sec) const override; |
| 309 | bool isSectionData(DataRefImpl Sec) const override; |
| 310 | bool isSectionBSS(DataRefImpl Sec) const override; |
| 311 | bool isSectionVirtual(DataRefImpl Sec) const override; |
| 312 | bool isSectionBitcode(DataRefImpl Sec) const override; |
| 313 | |
| 314 | /// When dsymutil generates the companion file, it strips all unnecessary |
| 315 | /// sections (e.g. everything in the _TEXT segment) by omitting their body |
| 316 | /// and setting the offset in their corresponding load command to zero. |
| 317 | /// |
| 318 | /// While the load command itself is valid, reading the section corresponds |
| 319 | /// to reading the number of bytes specified in the load command, starting |
| 320 | /// from offset 0 (i.e. the Mach-O header at the beginning of the file). |
| 321 | bool isSectionStripped(DataRefImpl Sec) const override; |
| 322 | |
| 323 | relocation_iterator section_rel_begin(DataRefImpl Sec) const override; |
| 324 | relocation_iterator section_rel_end(DataRefImpl Sec) const override; |
| 325 | |
| 326 | relocation_iterator extrel_begin() const; |
| 327 | relocation_iterator extrel_end() const; |
| 328 | iterator_range<relocation_iterator> external_relocations() const { |
| 329 | return make_range(extrel_begin(), extrel_end()); |
| 330 | } |
| 331 | |
| 332 | relocation_iterator locrel_begin() const; |
| 333 | relocation_iterator locrel_end() const; |
| 334 | |
| 335 | void moveRelocationNext(DataRefImpl &Rel) const override; |
| 336 | uint64_t getRelocationOffset(DataRefImpl Rel) const override; |
| 337 | symbol_iterator getRelocationSymbol(DataRefImpl Rel) const override; |
| 338 | section_iterator getRelocationSection(DataRefImpl Rel) const; |
| 339 | uint64_t getRelocationType(DataRefImpl Rel) const override; |
| 340 | void getRelocationTypeName(DataRefImpl Rel, |
| 341 | SmallVectorImpl<char> &Result) const override; |
| 342 | uint8_t getRelocationLength(DataRefImpl Rel) const; |
| 343 | |
| 344 | // MachO specific. |
| 345 | std::error_code getLibraryShortNameByIndex(unsigned Index, StringRef &) const; |
| 346 | uint32_t getLibraryCount() const; |
| 347 | |
| 348 | section_iterator getRelocationRelocatedSection(relocation_iterator Rel) const; |
| 349 | |
| 350 | // TODO: Would be useful to have an iterator based version |
| 351 | // of the load command interface too. |
| 352 | |
| 353 | basic_symbol_iterator symbol_begin() const override; |
| 354 | basic_symbol_iterator symbol_end() const override; |
| 355 | |
| 356 | // MachO specific. |
| 357 | basic_symbol_iterator getSymbolByIndex(unsigned Index) const; |
| 358 | uint64_t getSymbolIndex(DataRefImpl Symb) const; |
| 359 | |
| 360 | section_iterator section_begin() const override; |
| 361 | section_iterator section_end() const override; |
| 362 | |
| 363 | uint8_t getBytesInAddress() const override; |
| 364 | |
| 365 | StringRef getFileFormatName() const override; |
| 366 | Triple::ArchType getArch() const override; |
| 367 | SubtargetFeatures getFeatures() const override { return SubtargetFeatures(); } |
| 368 | Triple getArchTriple(const char **McpuDefault = nullptr) const; |
| 369 | |
| 370 | relocation_iterator section_rel_begin(unsigned Index) const; |
| 371 | relocation_iterator section_rel_end(unsigned Index) const; |
| 372 | |
| 373 | dice_iterator begin_dices() const; |
| 374 | dice_iterator end_dices() const; |
| 375 | |
| 376 | load_command_iterator begin_load_commands() const; |
| 377 | load_command_iterator end_load_commands() const; |
| 378 | iterator_range<load_command_iterator> load_commands() const; |
| 379 | |
| 380 | /// For use iterating over all exported symbols. |
| 381 | iterator_range<export_iterator> exports(Error &Err) const; |
| 382 | |
| 383 | /// For use examining a trie not in a MachOObjectFile. |
| 384 | static iterator_range<export_iterator> exports(Error &Err, |
| 385 | ArrayRef<uint8_t> Trie, |
| 386 | const MachOObjectFile *O = |
| 387 | nullptr); |
| 388 | |
| 389 | /// For use iterating over all rebase table entries. |
| 390 | iterator_range<rebase_iterator> rebaseTable(Error &Err); |
| 391 | |
| 392 | /// For use examining rebase opcodes in a MachOObjectFile. |
| 393 | static iterator_range<rebase_iterator> rebaseTable(Error &Err, |
| 394 | MachOObjectFile *O, |
| 395 | ArrayRef<uint8_t> Opcodes, |
| 396 | bool is64); |
| 397 | |
| 398 | /// For use iterating over all bind table entries. |
| 399 | iterator_range<bind_iterator> bindTable(Error &Err); |
| 400 | |
| 401 | /// For use iterating over all lazy bind table entries. |
| 402 | iterator_range<bind_iterator> lazyBindTable(Error &Err); |
| 403 | |
| 404 | /// For use iterating over all weak bind table entries. |
| 405 | iterator_range<bind_iterator> weakBindTable(Error &Err); |
| 406 | |
| 407 | /// For use examining bind opcodes in a MachOObjectFile. |
| 408 | static iterator_range<bind_iterator> bindTable(Error &Err, |
| 409 | MachOObjectFile *O, |
| 410 | ArrayRef<uint8_t> Opcodes, |
| 411 | bool is64, |
| 412 | MachOBindEntry::Kind); |
| 413 | |
| 414 | /// For use with a SegIndex,SegOffset pair in MachOBindEntry::moveNext() to |
| 415 | /// validate a MachOBindEntry. |
| 416 | const char *BindEntryCheckSegAndOffset(int32_t SegIndex, uint64_t SegOffset, |
| 417 | bool endInvalid) const { |
| 418 | return BindRebaseSectionTable->checkSegAndOffset(SegIndex, SegOffset, |
| 419 | endInvalid); |
| 420 | } |
| 421 | /// For use in MachOBindEntry::moveNext() to validate a MachOBindEntry for |
| 422 | /// the BIND_OPCODE_DO_BIND_ULEB_TIMES_SKIPPING_ULEB opcode. |
| 423 | const char *BindEntryCheckCountAndSkip(uint32_t Count, uint32_t Skip, |
| 424 | uint8_t PointerSize, int32_t SegIndex, |
| 425 | uint64_t SegOffset) const { |
| 426 | return BindRebaseSectionTable->checkCountAndSkip(Count, Skip, PointerSize, |
| 427 | SegIndex, SegOffset); |
| 428 | } |
| 429 | |
| 430 | /// For use with a SegIndex,SegOffset pair in MachORebaseEntry::moveNext() to |
| 431 | /// validate a MachORebaseEntry. |
| 432 | const char *RebaseEntryCheckSegAndOffset(int32_t SegIndex, uint64_t SegOffset, |
| 433 | bool endInvalid) const { |
| 434 | return BindRebaseSectionTable->checkSegAndOffset(SegIndex, SegOffset, |
| 435 | endInvalid); |
| 436 | } |
| 437 | /// For use in MachORebaseEntry::moveNext() to validate a MachORebaseEntry for |
| 438 | /// the REBASE_OPCODE_DO_*_TIMES* opcodes. |
| 439 | const char *RebaseEntryCheckCountAndSkip(uint32_t Count, uint32_t Skip, |
| 440 | uint8_t PointerSize, int32_t SegIndex, |
| 441 | uint64_t SegOffset) const { |
| 442 | return BindRebaseSectionTable->checkCountAndSkip(Count, Skip, PointerSize, |
| 443 | SegIndex, SegOffset); |
| 444 | } |
| 445 | |
| 446 | /// For use with the SegIndex of a checked Mach-O Bind or Rebase entry to |
| 447 | /// get the segment name. |
| 448 | StringRef BindRebaseSegmentName(int32_t SegIndex) const { |
| 449 | return BindRebaseSectionTable->segmentName(SegIndex); |
| 450 | } |
| 451 | |
| 452 | /// For use with a SegIndex,SegOffset pair from a checked Mach-O Bind or |
| 453 | /// Rebase entry to get the section name. |
| 454 | StringRef BindRebaseSectionName(uint32_t SegIndex, uint64_t SegOffset) const { |
| 455 | return BindRebaseSectionTable->sectionName(SegIndex, SegOffset); |
| 456 | } |
| 457 | |
| 458 | /// For use with a SegIndex,SegOffset pair from a checked Mach-O Bind or |
| 459 | /// Rebase entry to get the address. |
| 460 | uint64_t BindRebaseAddress(uint32_t SegIndex, uint64_t SegOffset) const { |
| 461 | return BindRebaseSectionTable->address(SegIndex, SegOffset); |
| 462 | } |
| 463 | |
| 464 | // In a MachO file, sections have a segment name. This is used in the .o |
| 465 | // files. They have a single segment, but this field specifies which segment |
| 466 | // a section should be put in the final object. |
| 467 | StringRef getSectionFinalSegmentName(DataRefImpl Sec) const; |
| 468 | |
| 469 | // Names are stored as 16 bytes. These returns the raw 16 bytes without |
| 470 | // interpreting them as a C string. |
| 471 | ArrayRef<char> getSectionRawName(DataRefImpl Sec) const; |
| 472 | ArrayRef<char> getSectionRawFinalSegmentName(DataRefImpl Sec) const; |
| 473 | |
| 474 | // MachO specific Info about relocations. |
| 475 | bool isRelocationScattered(const MachO::any_relocation_info &RE) const; |
| 476 | unsigned getPlainRelocationSymbolNum( |
| 477 | const MachO::any_relocation_info &RE) const; |
| 478 | bool getPlainRelocationExternal(const MachO::any_relocation_info &RE) const; |
| 479 | bool getScatteredRelocationScattered( |
| 480 | const MachO::any_relocation_info &RE) const; |
| 481 | uint32_t getScatteredRelocationValue( |
| 482 | const MachO::any_relocation_info &RE) const; |
| 483 | uint32_t getScatteredRelocationType( |
| 484 | const MachO::any_relocation_info &RE) const; |
| 485 | unsigned getAnyRelocationAddress(const MachO::any_relocation_info &RE) const; |
| 486 | unsigned getAnyRelocationPCRel(const MachO::any_relocation_info &RE) const; |
| 487 | unsigned getAnyRelocationLength(const MachO::any_relocation_info &RE) const; |
| 488 | unsigned getAnyRelocationType(const MachO::any_relocation_info &RE) const; |
| 489 | SectionRef getAnyRelocationSection(const MachO::any_relocation_info &RE) const; |
| 490 | |
| 491 | // MachO specific structures. |
| 492 | MachO::section getSection(DataRefImpl DRI) const; |
| 493 | MachO::section_64 getSection64(DataRefImpl DRI) const; |
| 494 | MachO::section getSection(const LoadCommandInfo &L, unsigned Index) const; |
| 495 | MachO::section_64 getSection64(const LoadCommandInfo &L,unsigned Index) const; |
| 496 | MachO::nlist getSymbolTableEntry(DataRefImpl DRI) const; |
| 497 | MachO::nlist_64 getSymbol64TableEntry(DataRefImpl DRI) const; |
| 498 | |
| 499 | MachO::linkedit_data_command |
| 500 | getLinkeditDataLoadCommand(const LoadCommandInfo &L) const; |
| 501 | MachO::segment_command |
| 502 | getSegmentLoadCommand(const LoadCommandInfo &L) const; |
| 503 | MachO::segment_command_64 |
| 504 | getSegment64LoadCommand(const LoadCommandInfo &L) const; |
| 505 | MachO::linker_option_command |
| 506 | getLinkerOptionLoadCommand(const LoadCommandInfo &L) const; |
| 507 | MachO::version_min_command |
| 508 | getVersionMinLoadCommand(const LoadCommandInfo &L) const; |
| 509 | MachO::note_command |
| 510 | getNoteLoadCommand(const LoadCommandInfo &L) const; |
| 511 | MachO::build_version_command |
| 512 | getBuildVersionLoadCommand(const LoadCommandInfo &L) const; |
| 513 | MachO::build_tool_version |
| 514 | getBuildToolVersion(unsigned index) const; |
| 515 | MachO::dylib_command |
| 516 | getDylibIDLoadCommand(const LoadCommandInfo &L) const; |
| 517 | MachO::dyld_info_command |
| 518 | getDyldInfoLoadCommand(const LoadCommandInfo &L) const; |
| 519 | MachO::dylinker_command |
| 520 | getDylinkerCommand(const LoadCommandInfo &L) const; |
| 521 | MachO::uuid_command |
| 522 | getUuidCommand(const LoadCommandInfo &L) const; |
| 523 | MachO::rpath_command |
| 524 | getRpathCommand(const LoadCommandInfo &L) const; |
| 525 | MachO::source_version_command |
| 526 | getSourceVersionCommand(const LoadCommandInfo &L) const; |
| 527 | MachO::entry_point_command |
| 528 | getEntryPointCommand(const LoadCommandInfo &L) const; |
| 529 | MachO::encryption_info_command |
| 530 | getEncryptionInfoCommand(const LoadCommandInfo &L) const; |
| 531 | MachO::encryption_info_command_64 |
| 532 | getEncryptionInfoCommand64(const LoadCommandInfo &L) const; |
| 533 | MachO::sub_framework_command |
| 534 | getSubFrameworkCommand(const LoadCommandInfo &L) const; |
| 535 | MachO::sub_umbrella_command |
| 536 | getSubUmbrellaCommand(const LoadCommandInfo &L) const; |
| 537 | MachO::sub_library_command |
| 538 | getSubLibraryCommand(const LoadCommandInfo &L) const; |
| 539 | MachO::sub_client_command |
| 540 | getSubClientCommand(const LoadCommandInfo &L) const; |
| 541 | MachO::routines_command |
| 542 | getRoutinesCommand(const LoadCommandInfo &L) const; |
| 543 | MachO::routines_command_64 |
| 544 | getRoutinesCommand64(const LoadCommandInfo &L) const; |
| 545 | MachO::thread_command |
| 546 | getThreadCommand(const LoadCommandInfo &L) const; |
| 547 | |
| 548 | MachO::any_relocation_info getRelocation(DataRefImpl Rel) const; |
| 549 | MachO::data_in_code_entry getDice(DataRefImpl Rel) const; |
| 550 | const MachO::mach_header &getHeader() const; |
| 551 | const MachO::mach_header_64 &getHeader64() const; |
| 552 | uint32_t |
| 553 | getIndirectSymbolTableEntry(const MachO::dysymtab_command &DLC, |
| 554 | unsigned Index) const; |
| 555 | MachO::data_in_code_entry getDataInCodeTableEntry(uint32_t DataOffset, |
| 556 | unsigned Index) const; |
| 557 | MachO::symtab_command getSymtabLoadCommand() const; |
| 558 | MachO::dysymtab_command getDysymtabLoadCommand() const; |
| 559 | MachO::linkedit_data_command getDataInCodeLoadCommand() const; |
| 560 | MachO::linkedit_data_command getLinkOptHintsLoadCommand() const; |
| 561 | ArrayRef<uint8_t> getDyldInfoRebaseOpcodes() const; |
| 562 | ArrayRef<uint8_t> getDyldInfoBindOpcodes() const; |
| 563 | ArrayRef<uint8_t> getDyldInfoWeakBindOpcodes() const; |
| 564 | ArrayRef<uint8_t> getDyldInfoLazyBindOpcodes() const; |
| 565 | ArrayRef<uint8_t> getDyldInfoExportsTrie() const; |
| 566 | ArrayRef<uint8_t> getUuid() const; |
| 567 | |
| 568 | StringRef getStringTableData() const; |
| 569 | bool is64Bit() const; |
| 570 | void ReadULEB128s(uint64_t Index, SmallVectorImpl<uint64_t> &Out) const; |
| 571 | |
| 572 | static StringRef guessLibraryShortName(StringRef Name, bool &isFramework, |
| 573 | StringRef &Suffix); |
| 574 | |
| 575 | static Triple::ArchType getArch(uint32_t CPUType); |
| 576 | static Triple getArchTriple(uint32_t CPUType, uint32_t CPUSubType, |
| 577 | const char **McpuDefault = nullptr, |
| 578 | const char **ArchFlag = nullptr); |
| 579 | static bool isValidArch(StringRef ArchFlag); |
| 580 | static Triple getHostArch(); |
| 581 | |
| 582 | bool isRelocatableObject() const override; |
| 583 | |
| 584 | StringRef mapDebugSectionName(StringRef Name) const override; |
| 585 | |
| 586 | bool hasPageZeroSegment() const { return HasPageZeroSegment; } |
| 587 | |
| 588 | static bool classof(const Binary *v) { |
| 589 | return v->isMachO(); |
| 590 | } |
| 591 | |
| 592 | static uint32_t |
| 593 | getVersionMinMajor(MachO::version_min_command &C, bool SDK) { |
| 594 | uint32_t VersionOrSDK = (SDK) ? C.sdk : C.version; |
| 595 | return (VersionOrSDK >> 16) & 0xffff; |
| 596 | } |
| 597 | |
| 598 | static uint32_t |
| 599 | getVersionMinMinor(MachO::version_min_command &C, bool SDK) { |
| 600 | uint32_t VersionOrSDK = (SDK) ? C.sdk : C.version; |
| 601 | return (VersionOrSDK >> 8) & 0xff; |
| 602 | } |
| 603 | |
| 604 | static uint32_t |
| 605 | getVersionMinUpdate(MachO::version_min_command &C, bool SDK) { |
| 606 | uint32_t VersionOrSDK = (SDK) ? C.sdk : C.version; |
| 607 | return VersionOrSDK & 0xff; |
| 608 | } |
| 609 | |
| 610 | static std::string getBuildPlatform(uint32_t platform) { |
| 611 | switch (platform) { |
| 612 | case MachO::PLATFORM_MACOS: return "macos"; |
| 613 | case MachO::PLATFORM_IOS: return "ios"; |
| 614 | case MachO::PLATFORM_TVOS: return "tvos"; |
| 615 | case MachO::PLATFORM_WATCHOS: return "watchos"; |
| 616 | case MachO::PLATFORM_BRIDGEOS: return "bridgeos"; |
| 617 | default: |
| 618 | std::string ret; |
| 619 | raw_string_ostream ss(ret); |
| 620 | ss << format_hex(platform, 8, true); |
| 621 | return ss.str(); |
| 622 | } |
| 623 | } |
| 624 | |
| 625 | static std::string getBuildTool(uint32_t tools) { |
| 626 | switch (tools) { |
| 627 | case MachO::TOOL_CLANG: return "clang"; |
| 628 | case MachO::TOOL_SWIFT: return "swift"; |
| 629 | case MachO::TOOL_LD: return "ld"; |
| 630 | default: |
| 631 | std::string ret; |
| 632 | raw_string_ostream ss(ret); |
| 633 | ss << format_hex(tools, 8, true); |
| 634 | return ss.str(); |
| 635 | } |
| 636 | } |
| 637 | |
| 638 | static std::string getVersionString(uint32_t version) { |
| 639 | uint32_t major = (version >> 16) & 0xffff; |
| 640 | uint32_t minor = (version >> 8) & 0xff; |
| 641 | uint32_t update = version & 0xff; |
| 642 | |
| 643 | SmallString<32> Version; |
| 644 | Version = utostr(major) + "." + utostr(minor); |
| 645 | if (update != 0) |
| 646 | Version += "." + utostr(update); |
| 647 | return Version.str(); |
| 648 | } |
| 649 | |
| 650 | private: |
| 651 | MachOObjectFile(MemoryBufferRef Object, bool IsLittleEndian, bool Is64Bits, |
| 652 | Error &Err, uint32_t UniversalCputype = 0, |
| 653 | uint32_t UniversalIndex = 0); |
| 654 | |
| 655 | uint64_t getSymbolValueImpl(DataRefImpl Symb) const override; |
| 656 | |
| 657 | union { |
| 658 | MachO::mach_header_64 Header64; |
| 659 | MachO::mach_header Header; |
| 660 | }; |
| 661 | using SectionList = SmallVector<const char*, 1>; |
| 662 | SectionList Sections; |
| 663 | using LibraryList = SmallVector<const char*, 1>; |
| 664 | LibraryList Libraries; |
| 665 | LoadCommandList LoadCommands; |
| 666 | using LibraryShortName = SmallVector<StringRef, 1>; |
| 667 | using BuildToolList = SmallVector<const char*, 1>; |
| 668 | BuildToolList BuildTools; |
| 669 | mutable LibraryShortName LibrariesShortNames; |
| 670 | std::unique_ptr<BindRebaseSegInfo> BindRebaseSectionTable; |
| 671 | const char *SymtabLoadCmd = nullptr; |
| 672 | const char *DysymtabLoadCmd = nullptr; |
| 673 | const char *DataInCodeLoadCmd = nullptr; |
| 674 | const char *LinkOptHintsLoadCmd = nullptr; |
| 675 | const char *DyldInfoLoadCmd = nullptr; |
| 676 | const char *UuidLoadCmd = nullptr; |
| 677 | bool HasPageZeroSegment = false; |
| 678 | }; |
| 679 | |
| 680 | /// DiceRef |
| 681 | inline DiceRef::DiceRef(DataRefImpl DiceP, const ObjectFile *Owner) |
| 682 | : DicePimpl(DiceP) , OwningObject(Owner) {} |
| 683 | |
| 684 | inline bool DiceRef::operator==(const DiceRef &Other) const { |
| 685 | return DicePimpl == Other.DicePimpl; |
| 686 | } |
| 687 | |
| 688 | inline bool DiceRef::operator<(const DiceRef &Other) const { |
| 689 | return DicePimpl < Other.DicePimpl; |
| 690 | } |
| 691 | |
| 692 | inline void DiceRef::moveNext() { |
| 693 | const MachO::data_in_code_entry *P = |
| 694 | reinterpret_cast<const MachO::data_in_code_entry *>(DicePimpl.p); |
| 695 | DicePimpl.p = reinterpret_cast<uintptr_t>(P + 1); |
| 696 | } |
| 697 | |
| 698 | // Since a Mach-O data in code reference, a DiceRef, can only be created when |
| 699 | // the OwningObject ObjectFile is a MachOObjectFile a static_cast<> is used for |
| 700 | // the methods that get the values of the fields of the reference. |
| 701 | |
| 702 | inline std::error_code DiceRef::getOffset(uint32_t &Result) const { |
| 703 | const MachOObjectFile *MachOOF = |
| 704 | static_cast<const MachOObjectFile *>(OwningObject); |
| 705 | MachO::data_in_code_entry Dice = MachOOF->getDice(DicePimpl); |
| 706 | Result = Dice.offset; |
| 707 | return std::error_code(); |
| 708 | } |
| 709 | |
| 710 | inline std::error_code DiceRef::getLength(uint16_t &Result) const { |
| 711 | const MachOObjectFile *MachOOF = |
| 712 | static_cast<const MachOObjectFile *>(OwningObject); |
| 713 | MachO::data_in_code_entry Dice = MachOOF->getDice(DicePimpl); |
| 714 | Result = Dice.length; |
| 715 | return std::error_code(); |
| 716 | } |
| 717 | |
| 718 | inline std::error_code DiceRef::getKind(uint16_t &Result) const { |
| 719 | const MachOObjectFile *MachOOF = |
| 720 | static_cast<const MachOObjectFile *>(OwningObject); |
| 721 | MachO::data_in_code_entry Dice = MachOOF->getDice(DicePimpl); |
| 722 | Result = Dice.kind; |
| 723 | return std::error_code(); |
| 724 | } |
| 725 | |
| 726 | inline DataRefImpl DiceRef::getRawDataRefImpl() const { |
| 727 | return DicePimpl; |
| 728 | } |
| 729 | |
| 730 | inline const ObjectFile *DiceRef::getObjectFile() const { |
| 731 | return OwningObject; |
| 732 | } |
| 733 | |
| 734 | } // end namespace object |
| 735 | } // end namespace llvm |
| 736 | |
| 737 | #endif // LLVM_OBJECT_MACHO_H |