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.
103 lines
2.6 KiB
103 lines
2.6 KiB
#include "injector_ctx.hpp"
|
|
|
|
namespace nasa
|
|
{
|
|
injector_ctx::injector_ctx(nasa::mem_ctx* map_into, nasa::mem_ctx* map_from)
|
|
:
|
|
map_into(map_into),
|
|
map_from(map_from),
|
|
pml4_index_map{}
|
|
{}
|
|
|
|
injector_ctx::~injector_ctx()
|
|
{
|
|
const auto pml4 = reinterpret_cast<ppml4e>(
|
|
map_into->set_page(
|
|
map_into->get_dirbase()));
|
|
|
|
// zero inserted pml4e's...
|
|
for (const auto [real_idx, inserted_idx] : pml4_index_map)
|
|
pml4[inserted_idx] = pml4e{ NULL };
|
|
|
|
while (!SwitchToThread());
|
|
}
|
|
|
|
bool injector_ctx::init() const
|
|
{
|
|
const auto source_pml4 =
|
|
reinterpret_cast<ppml4e>(
|
|
map_from->set_page(
|
|
map_from->get_dirbase()));
|
|
|
|
const auto target_pml4 =
|
|
reinterpret_cast<ppml4e>(
|
|
map_into->set_page(
|
|
map_into->get_dirbase()));
|
|
|
|
std::vector<std::pair<std::uint8_t, pml4e>> present_pml4es;
|
|
std::vector<std::uint8_t> empty_pml4es;
|
|
|
|
// find present pml4e's in map_from, and also find empty pml4e's in map_into...
|
|
for (auto idx = 100u; idx < 256; ++idx)
|
|
{
|
|
if (source_pml4[idx].present)
|
|
present_pml4es.push_back({ idx, source_pml4[idx] });
|
|
|
|
if (!target_pml4[idx].value)
|
|
empty_pml4es.push_back(idx);
|
|
}
|
|
|
|
// pretty much never going to happen but just in case :^)
|
|
if (present_pml4es.size() > empty_pml4es.size())
|
|
return false;
|
|
|
|
// setup translation table and insert pml4e's...
|
|
for (auto idx = 0u; idx < present_pml4es.size(); ++idx)
|
|
{
|
|
pml4_index_map.insert({ present_pml4es[idx].first, empty_pml4es[idx] });
|
|
target_pml4[empty_pml4es[idx]] = present_pml4es[idx].second;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
// translate the pml4 index to the correct pml4 index...
|
|
std::uintptr_t injector_ctx::translate(std::uintptr_t translate) const
|
|
{
|
|
virt_addr_t virt_addr{ reinterpret_cast<void*>(translate) };
|
|
try
|
|
{
|
|
virt_addr.pml4_index = pml4_index_map.at(virt_addr.pml4_index);
|
|
}
|
|
catch (const std::out_of_range& e)
|
|
{
|
|
// the pml4e is not in the map so we need to go get it
|
|
// and put it inside of the map. also put it inside of
|
|
// map_into's pml4...
|
|
|
|
const auto map_into_pml4 =
|
|
reinterpret_cast<ppml4e>(
|
|
map_into->set_page(
|
|
map_into->get_dirbase()));
|
|
|
|
const auto map_from_pml4 =
|
|
reinterpret_cast<ppml4e>(
|
|
map_from->set_page(
|
|
map_from->get_dirbase()));
|
|
|
|
const auto new_pml4e =
|
|
map_from_pml4[virt_addr.pml4_index];
|
|
|
|
for (auto idx = 100u; idx < 256; ++idx)
|
|
{
|
|
if (!map_into_pml4[idx].value)
|
|
{
|
|
map_into_pml4[idx] = new_pml4e;
|
|
pml4_index_map[virt_addr.pml4_index] = idx;
|
|
virt_addr.pml4_index = idx;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
return reinterpret_cast<std::uintptr_t>(virt_addr.value);
|
|
}
|
|
} |