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-03-26 20:01:13 +01:00
parent e3da821ee9
commit 49d259f3ae
3 changed files with 39 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
src/driver/driver_main.cpp
View File

@ -55,26 +55,31 @@ private:
global_driver* global_driver_instance{nullptr};
extern "C" void __cdecl __std_terminate()
{
KeBugCheckEx(DRIVER_VIOLATION, 14, 0, 0, 0);
}
_Function_class_(DRIVER_UNLOAD) void unload(const PDRIVER_OBJECT driver_object)
{
if (global_driver_instance)
try
{
global_driver_instance->pre_destroy(driver_object);
delete global_driver_instance;
if (global_driver_instance)
{
global_driver_instance->pre_destroy(driver_object);
delete global_driver_instance;
}
}
catch (std::exception& e)
{
debug_log("Destruction error occured: %s\n", e.what());
}
catch (...)
{
debug_log("Unknown destruction error occured. This should not happen!");
}
}
extern "C" NTSTATUS DriverEntry(const PDRIVER_OBJECT driver_object, PUNICODE_STRING /*registry_path*/)
{
driver_object->DriverUnload = unload;
try
{
driver_object->DriverUnload = unload;
global_driver_instance = new global_driver(driver_object);
}
catch (std::exception& e)

4
src/driver/irp.cpp
View File

@ -55,10 +55,10 @@ namespace
switch (ioctr_code)
{
case HELLO_DRV_IOCTL:
debug_log("[< HelloDriver >] Hello from the Driver!\n");
debug_log("Hello from the Driver!\n");
break;
default:
debug_log("[-] Invalid IOCTL Code: 0x%X\n", ioctr_code);
debug_log("Invalid IOCTL Code: 0x%X\n", ioctr_code);
irp->IoStatus.Status = STATUS_INVALID_DEVICE_REQUEST;
break;
}

24
src/driver/new.cpp
View File

@ -1,14 +1,29 @@
#include "std_include.hpp"
#include "new.hpp"
#include "exception.hpp"
namespace
{
void* perform_allocation(const size_t size, const POOL_TYPE pool, const unsigned long tag)
{
auto* memory = ExAllocatePoolWithTag(pool, size, tag);
if (!memory)
{
throw std::runtime_error("Memory allocation failed");
}
return memory;
}
}
void* __cdecl operator new(const size_t size, const POOL_TYPE pool, const unsigned long tag)
{
return ExAllocatePoolWithTag(pool, size, tag);
return perform_allocation(size, pool, tag);
}
void* __cdecl operator new[](const size_t size, const POOL_TYPE pool, const unsigned long tag)
{
return ExAllocatePoolWithTag(pool, size, tag);
return perform_allocation(size, pool, tag);
}
void* __cdecl operator new(const size_t size)
@ -46,3 +61,8 @@ void __cdecl operator delete[](void* ptr)
{
ExFreePool(ptr);
}
extern "C" void __cdecl __std_terminate()
{
KeBugCheckEx(DRIVER_VIOLATION, 14, 0, 0, 0);
}