Future/hypervisor
1
0
Fork 0
mirror of https://github.com/momo5502/hypervisor.git synced 2025-08-31 05:57:27 +00:00
Code Activity

More stuff

Browse Source
This commit is contained in:
momo5502
2022-04-17 11:51:31 +02:00
parent 4b06fcbd70
commit 378c52eb47
4 changed files with 33 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
src/driver/irp.cpp
View File

@@ -41,7 +41,7 @@ namespace
void apply_hook(const hook_request* request)
{
auto* buffer = new uint8_t[request->source_data_size];
if(!buffer)
if (!buffer)
{
throw std::runtime_error("Failed to copy buffer");
}
@@ -80,17 +80,27 @@ namespace
t.join();
if(!translation_hints)
if (!translation_hints)
{
debug_log("Failed to generate tranlsation hints");
return;
}
hypervisor::get_instance()->install_ept_hook(request->target_address, buffer, request->source_data_size, translation_hints);
hypervisor::get_instance()->install_ept_hook(request->target_address, buffer, request->source_data_size,
translation_hints);
debug_log("Done1\n");
}
void unhook()
{
const auto instance = hypervisor::get_instance();
if(instance)
{
instance->disable_all_ept_hooks();
}
}
_Function_class_(DRIVER_DISPATCH) NTSTATUS io_ctl_handler(
PDEVICE_OBJECT /*device_object*/, const PIRP irp)
{
@@ -113,6 +123,9 @@ namespace
case HOOK_DRV_IOCTL:
apply_hook(static_cast<hook_request*>(irp_sp->Parameters.DeviceIoControl.Type3InputBuffer));
break;
case UNHOOK_DRV_IOCTL:
unhook();
break;
default:
debug_log("Invalid IOCTL Code: 0x%X\n", ioctr_code);
irp->IoStatus.Status = STATUS_INVALID_DEVICE_REQUEST;