|
|
|
@ -4,12 +4,79 @@ namespace mm
|
|
|
|
|
{
|
|
|
|
|
auto translate(virt_addr_t virt_addr) -> u64
|
|
|
|
|
{
|
|
|
|
|
return {};
|
|
|
|
|
virt_addr_t cursor{ vmxroot_pml4 };
|
|
|
|
|
if (!reinterpret_cast<ppml4e>(cursor.value)[virt_addr.pml4_index].present)
|
|
|
|
|
return {};
|
|
|
|
|
|
|
|
|
|
cursor.pt_index = virt_addr.pml4_index;
|
|
|
|
|
if (!reinterpret_cast<ppdpte>(cursor.value)[virt_addr.pdpt_index].present)
|
|
|
|
|
return {};
|
|
|
|
|
|
|
|
|
|
// handle 1gb large page...
|
|
|
|
|
if (reinterpret_cast<ppdpte>(cursor.value)[virt_addr.pdpt_index].page_size)
|
|
|
|
|
return (reinterpret_cast<ppdpte>(cursor.value)
|
|
|
|
|
[virt_addr.pdpt_index].pfn << 12) + virt_addr.offset_1gb;
|
|
|
|
|
|
|
|
|
|
cursor.pd_index = virt_addr.pml4_index;
|
|
|
|
|
cursor.pt_index = virt_addr.pdpt_index;
|
|
|
|
|
if (!reinterpret_cast<ppde>(cursor.value)[virt_addr.pd_index].present)
|
|
|
|
|
return {};
|
|
|
|
|
|
|
|
|
|
// handle 2mb large page...
|
|
|
|
|
if (reinterpret_cast<ppde>(cursor.value)[virt_addr.pd_index].page_size)
|
|
|
|
|
return (reinterpret_cast<ppde>(cursor.value)
|
|
|
|
|
[virt_addr.pd_index].pfn << 12) + virt_addr.offset_2mb;
|
|
|
|
|
|
|
|
|
|
cursor.pdpt_index = virt_addr.pml4_index;
|
|
|
|
|
cursor.pd_index = virt_addr.pdpt_index;
|
|
|
|
|
cursor.pt_index = virt_addr.pd_index;
|
|
|
|
|
if (!reinterpret_cast<ppte>(cursor.value)[virt_addr.pt_index].present)
|
|
|
|
|
return {};
|
|
|
|
|
|
|
|
|
|
return (reinterpret_cast<ppte>(cursor.value)
|
|
|
|
|
[virt_addr.pt_index].pfn << 12) + virt_addr.offset_4kb;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
auto translate(virt_addr_t virt_addr, u64 pml4_phys, map_type type) -> u64
|
|
|
|
|
{
|
|
|
|
|
return {};
|
|
|
|
|
const auto pml4 =
|
|
|
|
|
reinterpret_cast<ppml4e>(
|
|
|
|
|
map_page(pml4_phys, type));
|
|
|
|
|
|
|
|
|
|
if (!pml4[virt_addr.pml4_index].present)
|
|
|
|
|
return {};
|
|
|
|
|
|
|
|
|
|
const auto pdpt =
|
|
|
|
|
reinterpret_cast<ppdpte>(
|
|
|
|
|
map_page(pml4[virt_addr
|
|
|
|
|
.pml4_index].pfn << 12, type));
|
|
|
|
|
|
|
|
|
|
if (!pdpt[virt_addr.pdpt_index].present)
|
|
|
|
|
return {};
|
|
|
|
|
|
|
|
|
|
if (pdpt[virt_addr.pdpt_index].page_size)
|
|
|
|
|
return (pdpt[virt_addr.pdpt_index].pfn << 12) + virt_addr.offset_1gb;
|
|
|
|
|
|
|
|
|
|
const auto pd =
|
|
|
|
|
reinterpret_cast<ppde>(
|
|
|
|
|
map_page(pdpt[virt_addr
|
|
|
|
|
.pdpt_index].pfn << 12, type));
|
|
|
|
|
|
|
|
|
|
if (!pd[virt_addr.pd_index].present)
|
|
|
|
|
return {};
|
|
|
|
|
|
|
|
|
|
if (pd[virt_addr.pd_index].page_size)
|
|
|
|
|
return (pd[virt_addr.pd_index].pfn << 12) + virt_addr.offset_2mb;
|
|
|
|
|
|
|
|
|
|
const auto pt =
|
|
|
|
|
reinterpret_cast<ppte>(
|
|
|
|
|
map_page(pd[virt_addr
|
|
|
|
|
.pd_index].pfn << 12, type));
|
|
|
|
|
|
|
|
|
|
if (!pt[virt_addr.pt_index].present)
|
|
|
|
|
return {};
|
|
|
|
|
|
|
|
|
|
return (pt[virt_addr.pt_index].pfn << 12) + virt_addr.offset_4kb;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
auto map_page(u64 phys_addr, map_type type) -> u64
|
|
|
|
@ -20,14 +87,63 @@ namespace mm
|
|
|
|
|
|
|
|
|
|
result.pt_index = (cpuid_value
|
|
|
|
|
.cpuid_additional_information
|
|
|
|
|
.initial_apic_id * 2)
|
|
|
|
|
+ (unsigned)type;
|
|
|
|
|
.initial_apic_id * 2)
|
|
|
|
|
+ (unsigned)type;
|
|
|
|
|
|
|
|
|
|
reinterpret_cast<ppte>(vmxroot_pml4)
|
|
|
|
|
[result.pt_index].pfn = phys_addr >> 12;
|
|
|
|
|
|
|
|
|
|
__invlpg(result.value);
|
|
|
|
|
result.offset = virt_addr_t{ (void*)phys_addr }.offset;
|
|
|
|
|
result.offset_4kb = virt_addr_t{ (void*)phys_addr }.offset_4kb;
|
|
|
|
|
return reinterpret_cast<u64>(result.value);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
auto map_virt(u64 dirbase, u64 virt_addr, map_type map_type) -> u64
|
|
|
|
|
{
|
|
|
|
|
const auto phys_addr =
|
|
|
|
|
translate(virt_addr_t{ (void*)
|
|
|
|
|
virt_addr }, dirbase, map_type);
|
|
|
|
|
|
|
|
|
|
if (!phys_addr)
|
|
|
|
|
return {};
|
|
|
|
|
|
|
|
|
|
return map_page(phys_addr, map_type);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
auto copy_virt(u64 dirbase_src, u64 virt_src, u64 dirbase_dest, u64 virt_dest, u64 size) -> bool
|
|
|
|
|
{
|
|
|
|
|
while (size)
|
|
|
|
|
{
|
|
|
|
|
auto dest_size = PAGE_SIZE - virt_addr_t{ (void*)virt_dest }.offset_4kb;
|
|
|
|
|
if (size < dest_size)
|
|
|
|
|
dest_size = size;
|
|
|
|
|
|
|
|
|
|
auto src_size = PAGE_SIZE - virt_addr_t{ (void*)virt_src }.offset_4kb;
|
|
|
|
|
if (size < src_size)
|
|
|
|
|
src_size = size;
|
|
|
|
|
|
|
|
|
|
const auto mapped_src =
|
|
|
|
|
reinterpret_cast<void*>(
|
|
|
|
|
map_virt(dirbase_src, virt_src, map_type::src));
|
|
|
|
|
|
|
|
|
|
if (!mapped_src)
|
|
|
|
|
return false;
|
|
|
|
|
|
|
|
|
|
const auto mapped_dest =
|
|
|
|
|
reinterpret_cast<void*>(
|
|
|
|
|
map_virt(dirbase_dest, virt_dest, map_type::dest));
|
|
|
|
|
|
|
|
|
|
if (!mapped_dest)
|
|
|
|
|
return false;
|
|
|
|
|
|
|
|
|
|
// copy directly between the two pages...
|
|
|
|
|
auto current_size = min(dest_size, src_size);
|
|
|
|
|
memcpy(mapped_dest, mapped_src, current_size);
|
|
|
|
|
|
|
|
|
|
virt_src += current_size;
|
|
|
|
|
virt_dest += current_size;
|
|
|
|
|
size -= current_size;
|
|
|
|
|
}
|
|
|
|
|
return true;
|
|
|
|
|
}
|
|
|
|
|
}
|