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.

114 lines
2.7 KiB

4 years ago
#include "vmxon.hpp"
namespace vmxon
{
auto create_vmxon_region(hv::pvcpu_ctx vcpu_ctx) -> void
{
PHYSICAL_ADDRESS mem_range;
mem_range.QuadPart = ~0ull;
hv::vmx_basic_msr_t vmx_basic;
vmx_basic.control = __readmsr(IA32_VMX_BASIC);
vcpu_ctx->vmxon =
reinterpret_cast<hv::pvmxon_region_ctx>(
MmAllocateContiguousMemory(PAGE_SIZE, mem_range));
vcpu_ctx->vmxon_phys =
MmGetPhysicalAddress(vcpu_ctx->vmxon).QuadPart;
RtlSecureZeroMemory(
vcpu_ctx->vmxon, PAGE_SIZE);
vcpu_ctx->vmxon->header
.bits
.revision_identifier =
vmx_basic.bits
.vmcs_revision_identifier;
}
auto create_vmcs(hv::pvcpu_ctx vcpu_ctx) -> void
{
PHYSICAL_ADDRESS mem_range;
mem_range.QuadPart = ~0ull;
hv::vmx_basic_msr_t vmx_basic;
vmx_basic.control = __readmsr(IA32_VMX_BASIC);
vcpu_ctx->vmcs =
reinterpret_cast<hv::pvmcs_ctx>(
MmAllocateContiguousMemory(PAGE_SIZE, mem_range));
vcpu_ctx->vmcs_phys =
MmGetPhysicalAddress(vcpu_ctx->vmcs).QuadPart;
RtlSecureZeroMemory(
vcpu_ctx->vmcs, PAGE_SIZE);
vcpu_ctx->vmcs->header
.bits
.revision_identifier =
vmx_basic.bits
.vmcs_revision_identifier;
}
auto create_vcpus(hv::pvmx_ctx vmx_ctx) -> void
{
vmx_ctx->vcpu_count =
4 years ago
KeQueryActiveProcessorCountEx(
ALL_PROCESSOR_GROUPS);
vmx_ctx->vcpus =
reinterpret_cast<hv::pvcpu_ctx*>(
ExAllocatePool(NonPagedPool,
sizeof(hv::pvcpu_ctx) * vmx_ctx->vcpu_count));
4 years ago
for (auto idx = 0u; idx < g_vmx_ctx->vcpu_count; ++idx)
4 years ago
{
vmx_ctx->vcpus[idx] =
reinterpret_cast<hv::pvcpu_ctx>(
ExAllocatePool(NonPagedPool, sizeof hv::vcpu_ctx));
vmx_ctx->vcpus[idx]->host_stack =
reinterpret_cast<u64>(
ExAllocatePool(NonPagedPool,
PAGE_SIZE * HOST_STACK_PAGES));
vmx_ctx->vcpus[idx]->gdt =
reinterpret_cast<segment_descriptor_64*>(
ExAllocatePool(NonPagedPool, PAGE_SIZE));
RtlZeroMemory(reinterpret_cast<void*>(
vmx_ctx->vcpus[idx]->host_stack), PAGE_SIZE * HOST_STACK_PAGES);
4 years ago
create_vmxon_region(vmx_ctx->vcpus[idx]);
create_vmcs(vmx_ctx->vcpus[idx]);
}
}
auto init_vmxon() -> void
{
hv::cr_fixed_t cr_fixed;
hv::cr0_t cr0 = { 0 };
hv::cr4_t cr4 = { 0 };
cr_fixed.all = __readmsr(IA32_VMX_CR0_FIXED0);
cr0.flags = __readcr0();
cr0.flags |= cr_fixed.split.low;
cr_fixed.all = __readmsr(IA32_VMX_CR0_FIXED1);
cr0.flags &= cr_fixed.split.low;
__writecr0(cr0.flags);
cr_fixed.all = __readmsr(IA32_VMX_CR4_FIXED0);
cr4.flags = __readcr4();
cr4.flags |= cr_fixed.split.low;
cr_fixed.all = __readmsr(IA32_VMX_CR4_FIXED1);
cr4.flags &= cr_fixed.split.low;
__writecr4(cr4.flags);
const auto vmxon_result =
4 years ago
__vmx_on((unsigned long long*)
&vmxon::g_vmx_ctx->vcpus[
KeGetCurrentProcessorNumber()]->vmxon_phys);
4 years ago
}
}