include/llvm/Object/MachOUniversal.h

   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
  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
  52     ObjectForArch getNext() const { return ObjectForArch(Parent, Index + 1); }
  53     uint32_t getCPUType() const { return Header.cputype; }
  54     uint32_t getCPUSubType() const { return Header.cpusubtype; }
  55     uint32_t getOffset() const { return Header.offset; }
  56     uint32_t getSize() const { return Header.size; }
  57     uint32_t getAlign() const { return Header.align; }
  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:
  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
  93   object_iterator begin_objects() const {
  94     return ObjectForArch(this, 0);
  95   }
  96   object_iterator end_objects() const {
  97     return ObjectForArch(nullptr, 0);
  98   }
  99
 100   uint32_t getNumberOfObjects() const { return NumberOfObjects; }
 101
 102   // Cast methods.
 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