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.
70 lines
2.2 KiB
70 lines
2.2 KiB
#include "map_driver.hpp"
|
|
#include "mapper_ctx/mapper_ctx.hpp"
|
|
#include "vdm_ctx/vdm_ctx.hpp"
|
|
#include "vdm/vdm.hpp"
|
|
#include "set_mgr/set_mgr.hpp"
|
|
|
|
namespace mapper
|
|
{
|
|
auto map_driver(std::uint32_t pid, std::uint8_t* drv_image, std::size_t image_size, void** entry_data) -> std::pair<mapper_error, void*>
|
|
{
|
|
std::vector<std::uint8_t> drv_buffer(drv_image, image_size + drv_image);
|
|
if (!drv_buffer.size())
|
|
return { mapper_error::image_invalid, nullptr };
|
|
|
|
const auto [drv_handle, drv_key] = vdm::load_drv();
|
|
if (drv_handle == INVALID_HANDLE_VALUE || drv_key.empty())
|
|
return { mapper_error::load_error, nullptr };
|
|
|
|
const auto context_pid = util::create_context();
|
|
|
|
if (!context_pid)
|
|
return { mapper_error::failed_to_create_proc, nullptr };
|
|
|
|
vdm::read_phys_t _read_phys =
|
|
[&](void* addr, void* buffer, std::size_t size) -> bool
|
|
{
|
|
return vdm::read_phys(addr, buffer, size);
|
|
};
|
|
|
|
vdm::write_phys_t _write_phys =
|
|
[&](void* addr, void* buffer, std::size_t size) -> bool
|
|
{
|
|
return vdm::write_phys(addr, buffer, size);
|
|
};
|
|
|
|
vdm::vdm_ctx v_ctx(_read_phys, _write_phys);
|
|
ptm::ptm_ctx desired_ctx(&v_ctx, GetCurrentProcessId());
|
|
ptm::ptm_ctx zombie_ctx(&v_ctx, context_pid);
|
|
nasa::mapper_ctx mapper(&desired_ctx, &zombie_ctx);
|
|
|
|
// disable the working set manager thread
|
|
// this thread loops forever and tries to
|
|
// page stuff to disk that is not accessed alot...
|
|
const auto result =
|
|
set_mgr::stop_setmgr(v_ctx,
|
|
set_mgr::get_setmgr_pethread(v_ctx));
|
|
|
|
if (result != STATUS_SUCCESS)
|
|
return { mapper_error::set_mgr_failure, nullptr };
|
|
|
|
// increment a process terminate counter
|
|
// then terminate the process... this makes it
|
|
// so the process can never actually fully close...
|
|
const auto [inc_ref_result, terminated] =
|
|
v_ctx.zombie_process(context_pid);
|
|
|
|
if (inc_ref_result != STATUS_SUCCESS || !terminated)
|
|
return { mapper_error::zombie_process_failed, nullptr };
|
|
|
|
const auto [drv_base, drv_entry] = mapper.map(drv_buffer);
|
|
if (!drv_base || !drv_entry)
|
|
return { mapper_error::init_failed, nullptr };
|
|
|
|
mapper.call_entry(drv_entry, entry_data);
|
|
if (!vdm::unload_drv(drv_handle, drv_key))
|
|
return { mapper_error::unload_error, nullptr };
|
|
|
|
return { mapper_error::error_success, drv_base };
|
|
}
|
|
} |