mirrors/fort
1
0
Fork 0
mirror of https://github.com/tnodir/fort synced 2024-11-15 04:27:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

Driver: Expand kernel stack for driver entry

Browse Source
This commit is contained in:
Nodir Temirkhodjaev 2023-04-19 10:18:48 +03:00
parent 7a805d77c9
commit 5846e535da
1 changed files with 20 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
src/driver/fortdrv.c
View File

@ -82,6 +82,23 @@ static void fort_driver_unload(PDRIVER_OBJECT driver)
fort_device_set(NULL); fort_device_set(NULL);
} }
static void NTAPI fort_driver_load_device_expand(PVOID param)
{
NTSTATUS *status = param;
*status = fort_device_load();
}
inline static void fort_driver_load_device(NTSTATUS *status)
{
const NTSTATUS status_expand = KeExpandKernelStackAndCallout(
&fort_driver_load_device_expand, status, KERNEL_STACK_SIZE);
if (!NT_SUCCESS(status_expand)) {
*status = status_expand;
}
}
static NTSTATUS fort_driver_load(PDRIVER_OBJECT driver, PUNICODE_STRING reg_path) static NTSTATUS fort_driver_load(PDRIVER_OBJECT driver, PUNICODE_STRING reg_path)
{ {
NTSTATUS status; NTSTATUS status;
@ -106,7 +123,9 @@ static NTSTATUS fort_driver_load(PDRIVER_OBJECT driver, PUNICODE_STRING reg_path
driver->MajorFunction[IRP_MJ_DEVICE_CONTROL] = &fort_device_control; driver->MajorFunction[IRP_MJ_DEVICE_CONTROL] = &fort_device_control;
driver->MajorFunction[IRP_MJ_SHUTDOWN] = &fort_device_shutdown; driver->MajorFunction[IRP_MJ_SHUTDOWN] = &fort_device_shutdown;
return fort_device_load(); fort_driver_load_device(&status);
return status;
} }
NTSTATUS DriverCallbacksSetup(PFORT_PROXYCB_INFO cb_info) NTSTATUS DriverCallbacksSetup(PFORT_PROXYCB_INFO cb_info)