Module.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_RUNTIME_MODULE_H
#define LIEF_RUNTIME_MODULE_H
#include <string_view>
#include <cstdint>
#include <cstring>
#include <fstream>
#include <ostream>
 
#include "LIEF/compiler_attributes.hpp"
#include "LIEF/iterators.hpp"
#include "LIEF/visibility.h"
 
#include <memory>
#include <string>
#include <vector>
 
namespace LIEF {
class Binary;
namespace runtime {
 
namespace details {
class ModuleIt;
class Module;
}

class LIEF_API Module {
  public:
  class Iterator final
    : public iterator_facade_base<Iterator, std::forward_iterator_tag, Module,
                                  std::ptrdiff_t, const Module*, const Module&> {
    public:
    using implementation = details::ModuleIt;
    using iterator_facade_base::operator++;
 
    LIEF_API Iterator();
 
    LIEF_API Iterator(const Iterator&);
    LIEF_API Iterator& operator=(const Iterator&);
 
    LIEF_API Iterator(Iterator&&) noexcept;
    LIEF_API Iterator& operator=(Iterator&&) noexcept;
 
    LIEF_API Iterator(std::unique_ptr<details::ModuleIt> impl);
    LIEF_API ~Iterator();
 
    friend LIEF_API bool operator==(const Iterator& LHS, const Iterator& RHS);
    friend LIEF_API bool operator!=(const Iterator& LHS, const Iterator& RHS) {
      return !(LHS == RHS);
    }
 
    // NOLINTNEXTLINE(bugprone-derived-method-shadowing-base-method)
    LIEF_API Iterator& operator++();
 
    LIEF_API const Module& operator*() const LIEF_LIFETIMEBOUND;
 
    // NOLINTNEXTLINE(bugprone-derived-method-shadowing-base-method)
    LIEF_API const Module* operator->() const LIEF_LIFETIMEBOUND;

    LIEF_API std::unique_ptr<Module> yield();
 
    private:
    void load() const;
 
    std::unique_ptr<details::ModuleIt> impl_;
    mutable std::unique_ptr<Module> cached_;
  };
  Module() = delete;
  Module(std::unique_ptr<details::Module> impl);
 
  Module(const Module&) = delete;
  Module& operator=(const Module&) = delete;
 
  Module(Module&&) noexcept;
  Module& operator=(Module&&) noexcept;
 
  virtual std::unique_ptr<Module> clone() const;

  uint64_t imagebase() const;

  uint64_t size() const;

  uint64_t end() const {
    return imagebase() + size();
  }

  std::string name() const;

  std::string path() const;

  bool contains(uintptr_t addr) const {
    const uint64_t base = imagebase();
    return base <= addr && addr < (base + size());
  }

  std::vector<uint8_t> dump() const {
    const uint64_t base = imagebase();
    const uint64_t nbytes = size();
    if (base == 0 || nbytes == 0) {
      return {};
    }
    std::vector<uint8_t> out(nbytes);
    std::memcpy(out.data(), reinterpret_cast<const void*>(base), nbytes);
    return out;
  }

  std::vector<uint8_t> dump(const std::string& filepath) const {
    std::vector<uint8_t> out = dump();
    std::ofstream ofs(filepath, std::ios::out | std::ios::binary);
    if (ofs) {
      ofs.write(reinterpret_cast<const char*>(out.data()),
                // NOLINTNEXTLINE(clang-taidy-avoid-static-integer-cast)
                static_cast<std::streamsize>(out.size()));
    }
    return out;
  }

  std::vector<uint8_t> dump(std::ostream& os) const {
    std::vector<uint8_t> out = dump();
    os.write(reinterpret_cast<const char*>(out.data()),
             // NOLINTNEXTLINE(clang-taidy-avoid-static-integer-cast)
             static_cast<std::streamsize>(out.size()));
    return out;
  }
 
  std::string to_string() const;

  template<class T>
  const T* as() const LIEF_LIFETIMEBOUND {
    static_assert(std::is_base_of_v<Module, T>, "Require Module inheritance");
    if (T::classof(this)) {
      return static_cast<const T*>(this);
    }
    return nullptr;
  }
 
  template<class T>
  T* as() LIEF_LIFETIMEBOUND {
    return const_cast<T*>(static_cast<const Module*>(this)->as<T>());
  }
 
  LIEF_API friend std::ostream& operator<<(std::ostream& os, const Module& M) {
    os << M.to_string();
    return os;
  }
 
  virtual ~Module();
 
  protected:
  std::unique_ptr<details::Module> impl_;
};
 
using modules_t = iterator_range<Module::Iterator>;

LIEF_API modules_t modules();

LIEF_API std::unique_ptr<Module> current_module();

inline std::unique_ptr<Module> module_from_name(std::string_view name) {
  auto range = modules();
  for (auto it = range.begin(); it != range.end(); ++it) {
    if (it->name() == name) {
      return it.yield();
    }
  }
  return nullptr;
}

inline std::unique_ptr<Module> module_from_path(std::string_view path) {
  auto range = modules();
  for (auto it = range.begin(); it != range.end(); ++it) {
    if (it->path() == path) {
      return it.yield();
    }
  }
  return nullptr;
}

inline std::unique_ptr<Module> module_from_addr(uintptr_t addr) {
  auto range = modules();
  for (auto it = range.begin(); it != range.end(); ++it) {
    if (it->contains(addr)) {
      return it.yield();
    }
  }
  return nullptr;
}
 
}
}
#endif