COFF/Binary.hpp

Go to the documentation of this file.

/* Copyright 2017 - 2026 R. Thomas
 * Copyright 2017 - 2026 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_COFF_BINARY_H
#define LIEF_COFF_BINARY_H
#include "LIEF/iterators.hpp"
#include "LIEF/span.hpp"
#include "LIEF/visibility.h"
 
#include "LIEF/COFF/String.hpp"
 
#include "LIEF/asm/Instruction.hpp"
 
#include <string_view>
#include <unordered_map>
#include <memory>
#include <string>
#include <vector>
 
namespace LIEF {
 
namespace assembly {
class Engine;
}
 
namespace COFF {
class Header;
class Parser;
class Section;
class Relocation;
class Symbol;

class LIEF_API Binary {
  public:
  friend class Parser;

  using sections_t = std::vector<std::unique_ptr<Section>>;

  using it_sections = ref_iterator<sections_t&, Section*>;

  using it_const_sections = const_ref_iterator<const sections_t&, const Section*>;

  using relocations_t = std::vector<std::unique_ptr<Relocation>>;

  using it_relocations = ref_iterator<relocations_t&, Relocation*>;

  using it_const_relocations =
      const_ref_iterator<const relocations_t&, const Relocation*>;

  using strings_table_t = std::vector<String>;

  using it_strings_table = ref_iterator<strings_table_t&>;

  using it_const_strings_table = const_ref_iterator<const strings_table_t&>;

  using symbols_t = std::vector<std::unique_ptr<Symbol>>;

  using it_symbols = ref_iterator<symbols_t&, Symbol*>;

  using it_const_symbols = const_ref_iterator<const symbols_t&, const Symbol*>;

  using instructions_it = iterator_range<assembly::Instruction::Iterator>;

  using it_functions = filter_iterator<symbols_t&, Symbol*>;

  using it_const_function = const_filter_iterator<const symbols_t&, const Symbol*>;

  const Header& header() const {
    return *header_;
  }
 
  Header& header() {
    return *header_;
  }

  it_sections sections() LIEF_LIFETIMEBOUND {
    return sections_;
  }
 
  it_const_sections sections() const LIEF_LIFETIMEBOUND {
    return sections_;
  }

  it_relocations relocations() LIEF_LIFETIMEBOUND {
    return relocations_;
  }
 
  it_const_relocations relocations() const LIEF_LIFETIMEBOUND {
    return relocations_;
  }

  it_symbols symbols() LIEF_LIFETIMEBOUND {
    return symbols_;
  }
 
  it_const_symbols symbols() const LIEF_LIFETIMEBOUND {
    return symbols_;
  }

  it_const_strings_table string_table() const LIEF_LIFETIMEBOUND {
    return strings_table_;
  }
 
  it_strings_table string_table() LIEF_LIFETIMEBOUND {
    return strings_table_;
  }

  String* find_string(uint32_t offset) LIEF_LIFETIMEBOUND {
    auto it = std::find_if(strings_table_.begin(), strings_table_.end(),
                           [offset](const String& item) {
                             return offset == item.offset();
                           });
    return it == strings_table_.end() ? nullptr : &*it;
  }
 
  const String* find_string(uint32_t offset) const LIEF_LIFETIMEBOUND {
    return const_cast<Binary*>(this)->find_string(offset);
  }

  it_const_function functions() const LIEF_LIFETIMEBOUND;
 
  it_functions functions() LIEF_LIFETIMEBOUND;

  const Symbol* find_function(const std::string& name) const LIEF_LIFETIMEBOUND;
 
  Symbol* find_function(const std::string& name) LIEF_LIFETIMEBOUND {
    return const_cast<Symbol*>(
        static_cast<const Binary*>(this)->find_function(name)
    );
  }

  const Symbol*
      find_demangled_function(const std::string& name) const LIEF_LIFETIMEBOUND;
 
  Symbol* find_demangled_function(const std::string& name) LIEF_LIFETIMEBOUND {
    return const_cast<Symbol*>(
        static_cast<const Binary*>(this)->find_demangled_function(name)
    );
  }

  instructions_it disassemble(const Symbol& symbol) const;

  instructions_it disassemble(const std::string& symbol) const;

  instructions_it disassemble(const uint8_t* buffer, size_t size,
                              uint64_t address = 0) const;
 

  instructions_it disassemble(const std::vector<uint8_t>& buffer,
                              uint64_t address = 0) const {
    return disassemble(buffer.data(), buffer.size(), address);
  }
 
  instructions_it disassemble(LIEF::span<const uint8_t> buffer,
                              uint64_t address = 0) const {
    return disassemble(buffer.data(), buffer.size(), address);
  }
 
  instructions_it disassemble(LIEF::span<uint8_t> buffer,
                              uint64_t address = 0) const {
    return disassemble(buffer.data(), buffer.size(), address);
  }

  Section* get_section(std::string_view name) LIEF_LIFETIMEBOUND {
    return const_cast<Section*>(
        static_cast<const Binary*>(this)->get_section(name)
    );
  }
 
  const Section* get_section(std::string_view name) const LIEF_LIFETIMEBOUND;
 
  std::string to_string() const;
 
  LIEF_API friend std::ostream& operator<<(std::ostream& os, const Binary& bin) {
    os << bin.to_string();
    return os;
  }
 
  ~Binary();
 
  private:
  Binary();
  std::unique_ptr<Header> header_;
  sections_t sections_;
  relocations_t relocations_;
  strings_table_t strings_table_;
  symbols_t symbols_;
 
  mutable std::unordered_map<uint32_t, std::unique_ptr<assembly::Engine>> engines_;
 
  assembly::Engine* get_engine(uint64_t address) const;
 
  template<uint32_t Key, class F>
  LIEF_LOCAL assembly::Engine* get_cache_engine(uint64_t address, F&& f) const;
};
 
}
}
#endif