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

Driver: fortdrv: Call IoUnregisterShutdownNotification() if before it was registered

Browse Source
This commit is contained in:
Nodir Temirkhodjaev 2023-04-13 11:46:02 +03:00
parent 4e5e933bb9
commit f5b66d7713
1 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
src/driver/fortdrv.c
View File

@ -11,18 +11,21 @@
static void fort_driver_delete_device(PDRIVER_OBJECT driver)
{
if (driver->DeviceObject == NULL)
PDEVICE_OBJECT device_obj = driver->DeviceObject;
if (device_obj == NULL)
return;
/* Unregister Device Callbacks */
IoUnregisterShutdownNotification(driver->DeviceObject);
if ((device_obj->Flags & DO_BUFFERED_IO) != 0) {
IoUnregisterShutdownNotification(driver->DeviceObject);
}
/* Delete Device Link */
UNICODE_STRING device_link;
RtlInitUnicodeString(&device_link, FORT_DOS_DEVICE_NAME);
IoDeleteSymbolicLink(&device_link);
IoDeleteDevice(driver->DeviceObject);
IoDeleteDevice(device_obj);
driver->DeviceObject = NULL;
}
@ -42,14 +45,14 @@ static NTSTATUS fort_driver_create_device(PDRIVER_OBJECT driver, PUNICODE_STRING
return status;
}
device_obj->Flags |= DO_BUFFERED_IO;
status = IoRegisterShutdownNotification(device_obj);
if (!NT_SUCCESS(status)) {
LOG("Register Shutdown: Error: %x\n", status);
return status;
}
device_obj->Flags |= DO_BUFFERED_IO; /* after IoRegisterShutdownNotification()! */
UNICODE_STRING device_link;
RtlInitUnicodeString(&device_link, FORT_DOS_DEVICE_NAME);