You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
30 lines
866 B
30 lines
866 B
4 years ago
|
#pragma once
|
||
|
|
#include "util/utils.hpp"
|
||
|
|
#include <type_traits>
|
||
|
|
#include <dbghelp.h>
|
||
|
|
|
||
|
|
#pragma comment(lib, "Dbghelp.lib")
|
||
|
|
namespace drv
|
||
|
|
{
|
||
|
|
using kalloc_t = std::function<decltype(malloc)>;
|
||
|
|
using kmemcpy_t = std::function<decltype(memcpy)>;
|
||
|
|
using kmemset_t = std::function<decltype(memset)>;
|
||
|
|
|
||
|
|
using image_base_t = std::uintptr_t;
|
||
|
|
using image_entry_t = std::uintptr_t;
|
||
|
|
using mapper_routines_t = std::pair<kalloc_t, kmemcpy_t>;
|
||
|
|
using drv_buffer_t = std::vector<std::uint8_t>;
|
||
|
|
|
||
|
|
class hmdm_ctx
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
explicit hmdm_ctx(const mapper_routines_t& routines);
|
||
|
|
auto map_module(drv_buffer_t& drv_buffer, bool zero_headers = true) -> std::pair<image_base_t, image_entry_t>;
|
||
|
|
|
||
|
|
const kalloc_t kalloc;
|
||
|
|
const kmemcpy_t kmemcpy;
|
||
|
|
private:
|
||
|
|
auto resolve_imports(drv_buffer_t& drv_buffer) const -> void;
|
||
|
|
auto fix_relocs(drv_buffer_t& drv_buffer) const -> void;
|
||
|
|
};
|
||
|
|
}
|