1 //===- MachOUniversal.h - Mach-O universal binaries -------------*- 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 Mach-O fat/universal binaries. 11 // 12 //===----------------------------------------------------------------------===// 13 14 #ifndef LLVM_OBJECT_MACHOUNIVERSAL_H 15 #define LLVM_OBJECT_MACHOUNIVERSAL_H 16 17 #include "llvm/ADT/StringRef.h" 18 #include "llvm/ADT/Triple.h" 19 #include "llvm/Object/Archive.h" 20 #include "llvm/Object/Binary.h" 21 #include "llvm/Object/MachO.h" 22 #include "llvm/Support/ErrorOr.h" 23 #include "llvm/Support/MachO.h" 24 25 namespace llvm { 26 namespace object { 27 28 class MachOUniversalBinary : public Binary { 29 virtual void anchor(); 30 31 uint32_t NumberOfObjects; 32 public: 33 class ObjectForArch { 34 const MachOUniversalBinary *Parent; 35 /// \brief Index of object in the universal binary. 36 uint32_t Index; 37 /// \brief Descriptor of the object. 38 MachO::fat_arch Header; 39 40 public: 41 ObjectForArch(const MachOUniversalBinary *Parent, uint32_t Index); 42 clear()43 void clear() { 44 Parent = nullptr; 45 Index = 0; 46 } 47 48 bool operator==(const ObjectForArch &Other) const { 49 return (Parent == Other.Parent) && (Index == Other.Index); 50 } 51 getNext()52 ObjectForArch getNext() const { return ObjectForArch(Parent, Index + 1); } getCPUType()53 uint32_t getCPUType() const { return Header.cputype; } getCPUSubType()54 uint32_t getCPUSubType() const { return Header.cpusubtype; } getOffset()55 uint32_t getOffset() const { return Header.offset; } getSize()56 uint32_t getSize() const { return Header.size; } getAlign()57 uint32_t getAlign() const { return Header.align; } getArchTypeName()58 std::string getArchTypeName() const { 59 Triple T = MachOObjectFile::getArch(Header.cputype, Header.cpusubtype); 60 return T.getArchName(); 61 } 62 63 ErrorOr<std::unique_ptr<MachOObjectFile>> getAsObjectFile() const; 64 65 ErrorOr<std::unique_ptr<Archive>> getAsArchive() const; 66 }; 67 68 class object_iterator { 69 ObjectForArch Obj; 70 public: object_iterator(const ObjectForArch & Obj)71 object_iterator(const ObjectForArch &Obj) : Obj(Obj) {} 72 const ObjectForArch* operator->() const { 73 return &Obj; 74 } 75 76 bool operator==(const object_iterator &Other) const { 77 return Obj == Other.Obj; 78 } 79 bool operator!=(const object_iterator &Other) const { 80 return !(*this == Other); 81 } 82 83 object_iterator& operator++() { // Preincrement 84 Obj = Obj.getNext(); 85 return *this; 86 } 87 }; 88 89 MachOUniversalBinary(MemoryBufferRef Souce, std::error_code &EC); 90 static ErrorOr<std::unique_ptr<MachOUniversalBinary>> 91 create(MemoryBufferRef Source); 92 begin_objects()93 object_iterator begin_objects() const { 94 return ObjectForArch(this, 0); 95 } end_objects()96 object_iterator end_objects() const { 97 return ObjectForArch(nullptr, 0); 98 } 99 getNumberOfObjects()100 uint32_t getNumberOfObjects() const { return NumberOfObjects; } 101 102 // Cast methods. classof(Binary const * V)103 static inline bool classof(Binary const *V) { 104 return V->isMachOUniversalBinary(); 105 } 106 107 ErrorOr<std::unique_ptr<MachOObjectFile>> 108 getObjectForArch(Triple::ArchType Arch) const; 109 }; 110 111 } 112 } 113 114 #endif 115