MachO/Header.hpp

/* Copyright 2017 - 2024 R. Thomas
 * Copyright 2017 - 2024 Quarkslab
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
#ifndef LIEF_MACHO_HEADER_H
#define LIEF_MACHO_HEADER_H
 
#include <ostream>
#include <set>
#include <vector>
 
#include "LIEF/Object.hpp"
#include "LIEF/visibility.h"
#include "LIEF/enums.hpp"
 
#include "LIEF/Abstract/enums.hpp"
 
#include "LIEF/MachO/enums.hpp"
 
namespace LIEF {
namespace MachO {
class BinaryParser;
 
namespace details {
struct mach_header_64;
struct mach_header;
}
 
class LIEF_API Header : public Object {
  friend class BinaryParser;
  public:
  Header() = default;
 
  Header& operator=(const Header& copy) = default;
  Header(const Header& copy) = default;
 
  ~Header() override = default;
 
  enum class FILE_TYPE {
    UNKNOWN     = 0,
    OBJECT      = 0x1u,
    EXECUTE     = 0x2u,
    FVMLIB      = 0x3u,
    CORE        = 0x4u,
    PRELOAD     = 0x5u,
    DYLIB       = 0x6u,
    DYLINKER    = 0x7u,
    BUNDLE      = 0x8u,
    DYLIB_STUB  = 0x9u,
    DSYM        = 0xAu,
    KEXT_BUNDLE = 0xBu
  };
 
  enum class FLAGS : uint32_t {
    NOUNDEFS                = 0x00000001u,
    INCRLINK                = 0x00000002u,
    DYLDLINK                = 0x00000004u,
    BINDATLOAD              = 0x00000008u,
    PREBOUND                = 0x00000010u,
    SPLIT_SEGS              = 0x00000020u,
    LAZY_INIT               = 0x00000040u,
    TWOLEVEL                = 0x00000080u,
    FORCE_FLAT              = 0x00000100u,
    NOMULTIDEFS             = 0x00000200u,
    NOFIXPREBINDING         = 0x00000400u,
    PREBINDABLE             = 0x00000800u,
    ALLMODSBOUND            = 0x00001000u,
    SUBSECTIONS_VIA_SYMBOLS = 0x00002000u,
    CANONICAL               = 0x00004000u,
    WEAK_DEFINES            = 0x00008000u,
    BINDS_TO_WEAK           = 0x00010000u,
    ALLOW_STACK_EXECUTION   = 0x00020000u,
    ROOT_SAFE               = 0x00040000u,
    SETUID_SAFE             = 0x00080000u,
    NO_REEXPORTED_DYLIBS    = 0x00100000u,
    PIE                     = 0x00200000u,
    DEAD_STRIPPABLE_DYLIB   = 0x00400000u,
    HAS_TLV_DESCRIPTORS     = 0x00800000u,
    NO_HEAP_EXECUTION       = 0x01000000u,
    APP_EXTENSION_SAFE      = 0x02000000u
  };
 
  static constexpr int ABI64 = 0x01000000;
 
  enum class CPU_TYPE: int {
    ANY       = -1,
    X86       = 7,
    X86_64    = 7 | ABI64,
    MIPS      = 8,
    MC98000   = 10,
    ARM       = 12,
    ARM64     = 12 | ABI64,
    SPARC     = 14,
    POWERPC   = 18,
    POWERPC64 = 18 | ABI64,
  };
 
  MACHO_TYPES magic() const {
    return magic_;
  }
 
  CPU_TYPE cpu_type() const {
    return cputype_;
  }
 
  uint32_t cpu_subtype() const {
    return cpusubtype_;
  }
 
  FILE_TYPE file_type() const {
    return filetype_;
  }
 
  std::vector<FLAGS> flags_list() const;
 
  bool has(FLAGS flag) const;
 
  uint32_t nb_cmds() const {
    return ncmds_;
  }
 
  uint32_t sizeof_cmds() const {
    return sizeofcmds_;
  }
 
  uint32_t flags() const {
    return flags_;
  }
 
  uint32_t reserved() const {
    return reserved_;
  }
 
  void add(FLAGS flag);
 
  OBJECT_TYPES abstract_object_type() const;
 
  std::pair<ARCHITECTURES, std::set<MODES>> abstract_architecture() const;
 
  ENDIANNESS abstract_endianness() const;
 
  void magic(MACHO_TYPES magic) {
    magic_ = magic;
  }
  void cpu_type(CPU_TYPE type) {
    cputype_ = type;
  }
 
  void cpu_subtype(uint32_t cpusubtype) {
    cpusubtype_ = cpusubtype;
  }
 
  void file_type(FILE_TYPE filetype) {
    filetype_ = filetype;
  }
 
  void nb_cmds(uint32_t ncmds) {
    ncmds_ = ncmds;
  }
 
  void sizeof_cmds(uint32_t sizeofcmds) {
    sizeofcmds_ = sizeofcmds;
  }
 
  void flags(uint32_t flags) {
    flags_ = flags;
  }
 
  void remove(FLAGS flag);
 
  void reserved(uint32_t reserved) {
    reserved_ = reserved;
  }
 
  Header& operator+=(FLAGS c) {
    add(c);
    return *this;
  }
  Header& operator-=(FLAGS c) {
    remove(c);
    return *this;
  }
 
  void accept(Visitor& visitor) const override;
 
  LIEF_API friend std::ostream& operator<<(std::ostream& os, const Header& hdr);
 
  private:
  template<class T>
  LIEF_LOCAL Header(const T& header);
 
  MACHO_TYPES magic_ = MACHO_TYPES::UNKNOWN;
  CPU_TYPE   cputype_ = CPU_TYPE::ANY;
  uint32_t   cpusubtype_;
  FILE_TYPE  filetype_ = FILE_TYPE::UNKNOWN;
  uint32_t   ncmds_ = 0;
  uint32_t   sizeofcmds_ = 0;
  uint32_t   flags_ = 0;
  uint32_t   reserved_ = 0;
};
 
LIEF_API const char* to_string(Header::FILE_TYPE e);
LIEF_API const char* to_string(Header::CPU_TYPE e);
LIEF_API const char* to_string(Header::FLAGS e);
 
}
}
 
ENABLE_BITMASK_OPERATORS(LIEF::MachO::Header::FLAGS)
 
#endif