MachO/Relocation.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_RELOCATION_COMMAND_H
#define LIEF_MACHO_RELOCATION_COMMAND_H
#include <ostream>
#include <memory>
 
#include "LIEF/Abstract/Relocation.hpp"
 
#include "LIEF/MachO/Header.hpp"
#include "LIEF/visibility.h"
#include "LIEF/Object.hpp"
 
namespace LIEF {
namespace MachO {
class BinaryParser;
class Section;
class SegmentCommand;
class Symbol;
 
class LIEF_API Relocation : public LIEF::Relocation {
 
  friend class BinaryParser;
 
  public:
  using LIEF::Relocation::address;
  using LIEF::Relocation::size;
 
  enum class ORIGIN {
    UNKNOWN        = 0,
    DYLDINFO       = 1,
    RELOC_TABLE    = 2,
    CHAINED_FIXUPS = 3,
  };
 
  static constexpr auto R_SCATTERED = uint32_t(0x80000000);
  static constexpr auto R_ABS = uint32_t(0);
 
  Relocation() = default;
  Relocation(uint64_t address, uint8_t type);
 
  Relocation& operator=(const Relocation& other);
  Relocation(const Relocation& other);
  void swap(Relocation& other) noexcept;
 
  ~Relocation() override = default;
 
  virtual std::unique_ptr<Relocation> clone() const = 0;
 
  virtual bool is_pc_relative() const = 0;
 
  virtual uint8_t type() const {
    return type_;
  }
 
  Header::CPU_TYPE architecture() const {
    return architecture_;
  }
 
  virtual ORIGIN origin() const = 0;
 
  bool has_symbol() const {
    return symbol() != nullptr;
  }
 
  Symbol* symbol() {
    return symbol_;
  }
  const Symbol* symbol() const {
    return symbol_;
  }
 
  bool has_section() const {
    return section() != nullptr;
  }
 
  Section* section() {
    return section_;
  }
  const Section* section() const {
    return section_;
  }
 
  bool has_segment() const {
    return segment() != nullptr;
  }
 
  SegmentCommand* segment() {
    return segment_;
  }
  const SegmentCommand* segment() const {
    return segment_;
  }
 
  virtual void pc_relative(bool val) = 0;
  virtual void type(uint8_t type);
 
  void accept(Visitor& visitor) const override;
 
  virtual std::ostream& print(std::ostream& os) const;
 
  LIEF_API friend std::ostream& operator<<(std::ostream& os, const Relocation& relocation);
 
  protected:
  Symbol*         symbol_ = nullptr;
  uint8_t         type_ = 0;
  Header::CPU_TYPE architecture_ = Header::CPU_TYPE::ANY;
  Section*        section_ = nullptr;
  SegmentCommand* segment_ = nullptr;
};
 
 
LIEF_API const char* to_string(Relocation::ORIGIN e);
 
}
}
#endif