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

Driver: Force reauth on conf changes.

Browse Source
This commit is contained in:
Nodir Temirkhodjaev 2017-09-07 09:05:15 +05:00
parent 3e00b080da
commit 9655bb564a
1 changed files with 22 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
src/driver/fortdrv.c
View File

@ -304,6 +304,26 @@ fort_callout_remove (void)
g_device->active = FALSE; g_device->active = FALSE;
} }
static NTSTATUS
fort_callout_force_reauth (PDEVICE_OBJECT device)
{
NTSTATUS status;
// Unregister
fort_callout_remove();
fort_prov_unregister();
// Register
status = fort_prov_register(!g_device->prov_temporary,
g_device->prov_boot, NULL, NULL);
if (status == STATUS_SUCCESS) {
status = fort_callout_install(device);
}
return status;
}
static NTSTATUS static NTSTATUS
fort_device_create (PDEVICE_OBJECT device, PIRP irp) fort_device_create (PDEVICE_OBJECT device, PIRP irp)
{ {
@ -375,7 +395,7 @@ fort_device_control (PDEVICE_OBJECT device, PIRP irp)
status = STATUS_INSUFFICIENT_RESOURCES; status = STATUS_INSUFFICIENT_RESOURCES;
} else { } else {
fort_conf_ref_set(conf_ref); fort_conf_ref_set(conf_ref);
status = STATUS_SUCCESS; status = fort_callout_force_reauth(device);
} }
} }
break; break;
@ -386,7 +406,7 @@ fort_device_control (PDEVICE_OBJECT device, PIRP irp)
if (len == sizeof(FORT_CONF_FLAGS)) { if (len == sizeof(FORT_CONF_FLAGS)) {
fort_conf_ref_flags_set(conf_flags); fort_conf_ref_flags_set(conf_flags);
status = STATUS_SUCCESS; status = fort_callout_force_reauth(device);
} }
break; break;
} }