Future/hypervisor
1
0
Fork 0
mirror of https://github.com/momo5502/hypervisor.git synced 2025-04-19 13:42:55 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

More cleanup

Browse Source
This commit is contained in:
momo5502 2022-04-24 11:21:04 +02:00
parent 1b14d5f657
commit 40b6f5efae
3 changed files with 9 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/driver/irp.cpp
View File

@ -138,9 +138,6 @@ namespace
switch (ioctr_code) switch (ioctr_code)
{ {
case HELLO_DRV_IOCTL:
debug_log("Hello from the Driver!\n");
break;
case HOOK_DRV_IOCTL: case HOOK_DRV_IOCTL:
try_apply_hook(irp_sp); try_apply_hook(irp_sp);
break; break;

15
src/runner/main.cpp
View File

@ -46,17 +46,17 @@ void insert_nop(const driver_device& driver_device, const uint32_t pid, const ui
patch_data(driver_device, pid, addr, buffer.data(), buffer.size()); patch_data(driver_device, pid, addr, buffer.data(), buffer.size());
} }
void remove_hooks(const driver_device& driver_device)
{
(void)driver_device.send(UNHOOK_DRV_IOCTL, driver_device::data{});
}
void unsafe_main(const int /*argc*/, char* /*argv*/[]) void unsafe_main(const int /*argc*/, char* /*argv*/[])
{ {
printf("Pid: %lu\n", GetCurrentProcessId()); printf("Pid: %lu\n", GetCurrentProcessId());
driver driver{get_current_path() / "driver.sys", "MomoLul"}; driver driver{get_current_path() / "driver.sys", "MomoLul"};
driver_device driver_device{"\\\\.\\HelloDev"}; const driver_device driver_device{R"(\\.\HelloDev)"};
driver_device::data input{};
input.resize(4);
(void)driver_device.send(HELLO_DRV_IOCTL, input);
std::string pid; std::string pid;
std::cout << "Please, enter the pid: "; std::cout << "Please, enter the pid: ";
@ -84,8 +84,7 @@ void unsafe_main(const int /*argc*/, char* /*argv*/[])
printf("Press any key to disable all hooks!\n"); printf("Press any key to disable all hooks!\n");
_getch(); _getch();
input.resize(0); remove_hooks(driver_device);
(void)driver_device.send(UNHOOK_DRV_IOCTL, input);
printf("Press any key to exit!\n"); printf("Press any key to exit!\n");
_getch(); _getch();

5
src/shared/irp_data.hpp
View File

@ -1,8 +1,7 @@
#pragma once #pragma once
#define HELLO_DRV_IOCTL CTL_CODE(FILE_DEVICE_UNKNOWN, 0x800, METHOD_NEITHER, FILE_ANY_ACCESS) #define HOOK_DRV_IOCTL CTL_CODE(FILE_DEVICE_UNKNOWN, 0x800, METHOD_NEITHER, FILE_ANY_ACCESS)
#define HOOK_DRV_IOCTL CTL_CODE(FILE_DEVICE_UNKNOWN, 0x801, METHOD_NEITHER, FILE_ANY_ACCESS) #define UNHOOK_DRV_IOCTL CTL_CODE(FILE_DEVICE_UNKNOWN, 0x801, METHOD_NEITHER, FILE_ANY_ACCESS)
#define UNHOOK_DRV_IOCTL CTL_CODE(FILE_DEVICE_UNKNOWN, 0x802, METHOD_NEITHER, FILE_ANY_ACCESS)
static_assert(sizeof(void*) == 8); static_assert(sizeof(void*) == 8);