MachOUniversalWriter.h

Go to the documentation of this file.

//===- MachOUniversalWriter.h - MachO universal binary writer----*- C++ -*-===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
//
// Declares the Slice class and writeUniversalBinary function for writing a
// MachO universal binary file.
//
//===----------------------------------------------------------------------===//
 
#ifndef LLVM_OBJECT_MACHOUNIVERSALWRITER_H
#define LLVM_OBJECT_MACHOUNIVERSALWRITER_H
 
#include "llvm/ADT/ArrayRef.h"
#include "llvm/ADT/StringRef.h"
#include "llvm/ADT/Twine.h"
#include "llvm/BinaryFormat/MachO.h"
#include "llvm/Support/Error.h"
#include <cstdint>
#include <string>
 
namespace llvm {
class LLVMContext;
 
namespace object {
class Archive;
class Binary;
class IRObjectFile;
class MachOObjectFile;
 
class Slice {
  const Binary *B;
  uint32_t CPUType;
  uint32_t CPUSubType;
  std::string ArchName;
 
  // P2Alignment field stores slice alignment values from universal
  // binaries. This is also needed to order the slices so the total
  // file size can be calculated before creating the output buffer.
  uint32_t P2Alignment;
 
  Slice(const IRObjectFile &IRO, uint32_t CPUType, uint32_t CPUSubType,
        std::string ArchName, uint32_t Align);
 
public:
  explicit Slice(const MachOObjectFile &O);
 
  Slice(const MachOObjectFile &O, uint32_t Align);
 
  /// This constructor takes pre-specified \param CPUType , \param CPUSubType ,
  /// \param ArchName , \param Align instead of inferring them from the archive
  /// members.
  Slice(const Archive &A, uint32_t CPUType, uint32_t CPUSubType,
        std::string ArchName, uint32_t Align);
 
  static Expected<Slice> create(const Archive &A,
                                LLVMContext *LLVMCtx = nullptr);
 
  static Expected<Slice> create(const IRObjectFile &IRO, uint32_t Align);
 
  void setP2Alignment(uint32_t Align) { P2Alignment = Align; }
 
  const Binary *getBinary() const { return B; }
 
  uint32_t getCPUType() const { return CPUType; }
 
  uint32_t getCPUSubType() const { return CPUSubType; }
 
  uint32_t getP2Alignment() const { return P2Alignment; }
 
  uint64_t getCPUID() const {
    return static_cast<uint64_t>(CPUType) << 32 | CPUSubType;
  }
 
  std::string getArchString() const {
    if (!ArchName.empty())
      return ArchName;
    return ("unknown(" + Twine(CPUType) + "," +
            Twine(CPUSubType & ~MachO::CPU_SUBTYPE_MASK) + ")")
        .str();
  }
 
  friend bool operator<(const Slice &Lhs, const Slice &Rhs) {
    if (Lhs.CPUType == Rhs.CPUType)
      return Lhs.CPUSubType < Rhs.CPUSubType;
    // force arm64-family to follow after all other slices for
    // compatibility with cctools lipo
    if (Lhs.CPUType == MachO::CPU_TYPE_ARM64)
      return false;
    if (Rhs.CPUType == MachO::CPU_TYPE_ARM64)
      return true;
    // Sort by alignment to minimize file size
    return Lhs.P2Alignment < Rhs.P2Alignment;
  }
};
 
enum class FatHeaderType { FatHeader, Fat64Header };
 
Error writeUniversalBinary(ArrayRef<Slice> Slices, StringRef OutputFileName,
                           FatHeaderType FatHeader = FatHeaderType::FatHeader);
 
Error writeUniversalBinaryToStream(
    ArrayRef<Slice> Slices, raw_ostream &Out,
    FatHeaderType FatHeader = FatHeaderType::FatHeader);
 
} // end namespace object
 
} // end namespace llvm
 
#endif // LLVM_OBJECT_MACHOUNIVERSALWRITER_H