ReactOS  0.4.14-dev-368-gfa26425
notify.c File Reference
#include "mntmgr.h"
#include <ioevent.h>
#include <debug.h>
Include dependency graph for notify.c:

Go to the source code of this file.

Macros

#define NDEBUG
 

Functions

VOID SendOnlineNotification (IN PUNICODE_STRING SymbolicName)
 
VOID NTAPI SendOnlineNotificationWorker (IN PVOID Parameter)
 
VOID PostOnlineNotification (IN PDEVICE_EXTENSION DeviceExtension, IN PUNICODE_STRING SymbolicName)
 
VOID WaitForOnlinesToComplete (IN PDEVICE_EXTENSION DeviceExtension)
 
NTSTATUS NTAPI MountMgrTargetDeviceNotification (IN PVOID NotificationStructure, IN PVOID Context)
 
VOID RegisterForTargetDeviceNotification (IN PDEVICE_EXTENSION DeviceExtension, IN PDEVICE_INFORMATION DeviceInformation)
 
VOID MountMgrNotify (IN PDEVICE_EXTENSION DeviceExtension)
 
VOID MountMgrNotifyNameChange (IN PDEVICE_EXTENSION DeviceExtension, IN PUNICODE_STRING DeviceName, IN BOOLEAN ValidateVolume)
 
VOID RemoveWorkItem (IN PUNIQUE_ID_WORK_ITEM WorkItem)
 
VOID NTAPI UniqueIdChangeNotifyWorker (IN PDEVICE_OBJECT DeviceObject, IN PVOID Context)
 
NTSTATUS NTAPI UniqueIdChangeNotifyCompletion (IN PDEVICE_OBJECT DeviceObject, IN PIRP Irp, IN PVOID Context)
 
VOID IssueUniqueIdChangeNotifyWorker (IN PUNIQUE_ID_WORK_ITEM WorkItem, IN PMOUNTDEV_UNIQUE_ID UniqueId)
 
VOID IssueUniqueIdChangeNotify (IN PDEVICE_EXTENSION DeviceExtension, IN PUNICODE_STRING DeviceName, IN PMOUNTDEV_UNIQUE_ID UniqueId)
 

Macro Definition Documentation

◆ NDEBUG

#define NDEBUG

Definition at line 31 of file notify.c.

Function Documentation

◆ IssueUniqueIdChangeNotify()

VOID IssueUniqueIdChangeNotify ( IN PDEVICE_EXTENSION  DeviceExtension,
IN PUNICODE_STRING  DeviceName,
IN PMOUNTDEV_UNIQUE_ID  UniqueId 
)

Definition at line 661 of file notify.c.

664 {
666  PVOID IrpBuffer = NULL;
669  PUNIQUE_ID_WORK_ITEM WorkItem = NULL;
670 
671  /* Get the associated device object */
674  &FileObject,
675  &DeviceObject);
676  if (!NT_SUCCESS(Status))
677  {
678  return;
679  }
680 
681  /* And then, get attached device */
683 
685 
686  /* Allocate a work item */
687  WorkItem = AllocatePool(sizeof(UNIQUE_ID_WORK_ITEM));
688  if (!WorkItem)
689  {
691  return;
692  }
693 
694  WorkItem->Event = NULL;
695  WorkItem->WorkItem = IoAllocateWorkItem(DeviceExtension->DeviceObject);
696  if (!WorkItem->WorkItem)
697  {
699  goto Cleanup;
700  }
701 
702  WorkItem->DeviceExtension = DeviceExtension;
703  WorkItem->StackSize = DeviceObject->StackSize;
704  /* Already provide the IRP */
706 
708 
709  if (!WorkItem->Irp)
710  {
711  goto Cleanup;
712  }
713 
714  /* Ensure it has enough space */
715  IrpBuffer = AllocatePool(sizeof(MOUNTDEV_UNIQUE_ID_CHANGE_NOTIFY_OUTPUT) + 1024);
716  if (!IrpBuffer)
717  {
718  goto Cleanup;
719  }
720 
721  WorkItem->DeviceName.Length = DeviceName->Length;
722  WorkItem->DeviceName.MaximumLength = DeviceName->Length + sizeof(WCHAR);
723  WorkItem->DeviceName.Buffer = AllocatePool(WorkItem->DeviceName.MaximumLength);
724  if (!WorkItem->DeviceName.Buffer)
725  {
726  goto Cleanup;
727  }
728 
729  RtlCopyMemory(WorkItem->DeviceName.Buffer, DeviceName->Buffer, DeviceName->Length);
730  WorkItem->DeviceName.Buffer[DeviceName->Length / sizeof(WCHAR)] = UNICODE_NULL;
731 
732  WorkItem->IrpBuffer = IrpBuffer;
733  WorkItem->IrpBufferLength = sizeof(MOUNTDEV_UNIQUE_ID_CHANGE_NOTIFY_OUTPUT) + 1024;
734 
735  /* Add the worker in the list */
736  KeWaitForSingleObject(&(DeviceExtension->DeviceLock), Executive, KernelMode, FALSE, NULL);
737  InsertHeadList(&(DeviceExtension->UniqueIdWorkerItemListHead), &(WorkItem->UniqueIdWorkerItemListEntry));
738  KeReleaseSemaphore(&(DeviceExtension->DeviceLock), IO_NO_INCREMENT, 1, FALSE);
739 
740  /* And call the worker */
741  IssueUniqueIdChangeNotifyWorker(WorkItem, UniqueId);
742 
743  return;
744 
745 Cleanup:
746  if (IrpBuffer)
747  {
748  FreePool(IrpBuffer);
749  }
750 
751  if (WorkItem->Irp)
752  {
753  IoFreeIrp(WorkItem->Irp);
754  }
755 
756  if (WorkItem->WorkItem)
757  {
758  IoFreeWorkItem(WorkItem->WorkItem);
759  }
760 
761  if (WorkItem)
762  {
763  FreePool(WorkItem);
764  }
765 }
UNICODE_STRING DeviceName
Definition: mntmgr.h:147
NTSYSAPI VOID NTAPI RtlCopyMemory(VOID UNALIGNED *Destination, CONST VOID UNALIGNED *Source, ULONG Length)
USHORT MaximumLength
Definition: env_spec_w32.h:370
FORCEINLINE VOID InsertHeadList(_Inout_ PLIST_ENTRY ListHead, _Inout_ __drv_aliasesMem PLIST_ENTRY Entry)
Definition: rtlfuncs.h:201
LONG NTSTATUS
Definition: precomp.h:26
NTSTATUS NTAPI IoGetDeviceObjectPointer(IN PUNICODE_STRING ObjectName, IN ACCESS_MASK DesiredAccess, OUT PFILE_OBJECT *FileObject, OUT PDEVICE_OBJECT *DeviceObject)
Definition: device.c:1435
VOID NTAPI ObDereferenceObject(IN PVOID Object)
Definition: obref.c:375
WCHAR DeviceName[]
Definition: adapter.cpp:21
PDEVICE_EXTENSION DeviceExtension
Definition: mntmgr.h:143
PIO_WORKITEM NTAPI IoAllocateWorkItem(IN PDEVICE_OBJECT DeviceObject)
Definition: iowork.c:75
#define FreePool(P)
Definition: mntmgr.h:154
NTSTATUS NTAPI KeWaitForSingleObject(IN PVOID Object, IN KWAIT_REASON WaitReason, IN KPROCESSOR_MODE WaitMode, IN BOOLEAN Alertable, IN PLARGE_INTEGER Timeout OPTIONAL)
Definition: wait.c:416
VOID NTAPI IoFreeWorkItem(IN PIO_WORKITEM IoWorkItem)
Definition: iowork.c:64
PIO_WORKITEM WorkItem
Definition: mntmgr.h:142
VOID IssueUniqueIdChangeNotifyWorker(IN PUNIQUE_ID_WORK_ITEM WorkItem, IN PMOUNTDEV_UNIQUE_ID UniqueId)
Definition: notify.c:591
#define UNICODE_NULL
smooth NULL
Definition: ftsmooth.c:416
ULONG IrpBufferLength
Definition: mntmgr.h:148
_Inout_ PFILE_OBJECT FileObject
Definition: cdprocs.h:593
PDEVICE_OBJECT NTAPI IoGetAttachedDeviceReference(PDEVICE_OBJECT DeviceObject)
Definition: device.c:1406
struct _MOUNTDEV_UNIQUE_ID_CHANGE_NOTIFY_OUTPUT MOUNTDEV_UNIQUE_ID_CHANGE_NOTIFY_OUTPUT
__wchar_t WCHAR
Definition: xmlstorage.h:180
#define NT_SUCCESS(StatCode)
Definition: apphelp.c:32
* PFILE_OBJECT
Definition: iotypes.h:1955
#define FILE_READ_ATTRIBUTES
Definition: nt_native.h:647
LIST_ENTRY UniqueIdWorkerItemListEntry
Definition: mntmgr.h:141
static const WCHAR Cleanup[]
Definition: register.c:80
#define AllocatePool(Size)
Definition: mntmgr.h:153
Status
Definition: gdiplustypes.h:24
IN PDEVICE_OBJECT DeviceObject
Definition: fatprocs.h:1560
LONG NTAPI KeReleaseSemaphore(IN PKSEMAPHORE Semaphore, IN KPRIORITY Increment, IN LONG Adjustment, IN BOOLEAN Wait)
Definition: semphobj.c:54
VOID NTAPI IoFreeIrp(IN PIRP Irp)
Definition: irp.c:1666
#define IO_NO_INCREMENT
Definition: iotypes.h:566
PIRP NTAPI IoAllocateIrp(IN CCHAR StackSize, IN BOOLEAN ChargeQuota)
Definition: irp.c:615

Referenced by MountMgrMountedDeviceArrival().

◆ IssueUniqueIdChangeNotifyWorker()

VOID IssueUniqueIdChangeNotifyWorker ( IN PUNIQUE_ID_WORK_ITEM  WorkItem,
IN PMOUNTDEV_UNIQUE_ID  UniqueId 
)

Definition at line 591 of file notify.c.

593 {
594  PIRP Irp;
597  PIO_STACK_LOCATION Stack;
599 
600  /* Get the device object */
601  Status = IoGetDeviceObjectPointer(&(WorkItem->DeviceName),
603  &FileObject,
604  &DeviceObject);
605  if (!NT_SUCCESS(Status))
606  {
607  RemoveWorkItem(WorkItem);
608  return;
609  }
610 
611  /* And then, the attached device */
613 
614  /* Initialize the IRP */
615  Irp = WorkItem->Irp;
616  IoInitializeIrp(Irp, IoSizeOfIrp(WorkItem->StackSize), (CCHAR)WorkItem->StackSize);
617 
618  if (InterlockedExchange((PLONG)&(WorkItem->Event), 0) != 0)
619  {
622  RemoveWorkItem(WorkItem);
623  return;
624  }
625 
626  Irp->AssociatedIrp.SystemBuffer = WorkItem->IrpBuffer;
627  Irp->Tail.Overlay.Thread = PsGetCurrentThread();
628  RtlCopyMemory(Irp->AssociatedIrp.SystemBuffer, UniqueId, UniqueId->UniqueIdLength + sizeof(USHORT));
629 
631 
632  Stack->Parameters.DeviceIoControl.InputBufferLength = UniqueId->UniqueIdLength + sizeof(USHORT);
633  Stack->Parameters.DeviceIoControl.OutputBufferLength = WorkItem->IrpBufferLength;
634  Stack->Parameters.DeviceIoControl.Type3InputBuffer = 0;
635  Stack->Parameters.DeviceIoControl.IoControlCode = IOCTL_MOUNTDEV_UNIQUE_ID_CHANGE_NOTIFY;
637 
638  Status = IoSetCompletionRoutineEx(WorkItem->DeviceExtension->DeviceObject,
639  Irp,
641  WorkItem,
642  TRUE, TRUE, TRUE);
643  if (!NT_SUCCESS(Status))
644  {
647  RemoveWorkItem(WorkItem);
648  return;
649  }
650 
651  /* Call the driver */
655 }
#define TRUE
Definition: types.h:120
NTSYSAPI VOID NTAPI RtlCopyMemory(VOID UNALIGNED *Destination, CONST VOID UNALIGNED *Source, ULONG Length)
#define IOCTL_MOUNTDEV_UNIQUE_ID_CHANGE_NOTIFY
Definition: imports.h:86
#define PsGetCurrentThread()
Definition: env_spec_w32.h:81
_In_ PIRP Irp
Definition: csq.h:116
LONG NTSTATUS
Definition: precomp.h:26
NTSTATUS NTAPI IoGetDeviceObjectPointer(IN PUNICODE_STRING ObjectName, IN ACCESS_MASK DesiredAccess, OUT PFILE_OBJECT *FileObject, OUT PDEVICE_OBJECT *DeviceObject)
Definition: device.c:1435
VOID NTAPI ObDereferenceObject(IN PVOID Object)
Definition: obref.c:375
VOID NTAPI IoInitializeIrp(IN PIRP Irp, IN USHORT PacketSize, IN CCHAR StackSize)
Definition: irp.c:1854
_Inout_ PFILE_OBJECT FileObject
Definition: cdprocs.h:593
PDEVICE_OBJECT NTAPI IoGetAttachedDeviceReference(PDEVICE_OBJECT DeviceObject)
Definition: device.c:1406
#define NT_SUCCESS(StatCode)
Definition: apphelp.c:32
char CCHAR
Definition: typedefs.h:50
NTSTATUS NTAPI IoSetCompletionRoutineEx(IN PDEVICE_OBJECT DeviceObject, IN PIRP Irp, IN PIO_COMPLETION_ROUTINE CompletionRoutine, IN PVOID Context, IN BOOLEAN InvokeOnSuccess, IN BOOLEAN InvokeOnError, IN BOOLEAN InvokeOnCancel)
Definition: iocomp.c:220
* PFILE_OBJECT
Definition: iotypes.h:1955
#define FILE_READ_ATTRIBUTES
Definition: nt_native.h:647
__drv_aliasesMem FORCEINLINE PIO_STACK_LOCATION IoGetNextIrpStackLocation(_In_ PIRP Irp)
Definition: iofuncs.h:2647
#define InterlockedExchange
Definition: armddk.h:54
NTSTATUS NTAPI UniqueIdChangeNotifyCompletion(IN PDEVICE_OBJECT DeviceObject, IN PIRP Irp, IN PVOID Context)
Definition: notify.c:569
Status
Definition: gdiplustypes.h:24
IN PDEVICE_OBJECT DeviceObject
Definition: fatprocs.h:1560
unsigned short USHORT
Definition: pedump.c:61
#define IoSizeOfIrp(_StackSize)
NTSTATUS NTAPI IoCallDriver(IN PDEVICE_OBJECT DeviceObject, IN PIRP Irp)
Definition: irp.c:1218
VOID RemoveWorkItem(IN PUNIQUE_ID_WORK_ITEM WorkItem)
Definition: notify.c:481
struct _NAMED_PIPE_CREATE_PARAMETERS * Parameters
Definition: iotypes.h:2772
signed int * PLONG
Definition: retypes.h:5
#define IRP_MJ_DEVICE_CONTROL
Definition: rdpdr.c:52

Referenced by IssueUniqueIdChangeNotify(), and UniqueIdChangeNotifyWorker().

◆ MountMgrNotify()

VOID MountMgrNotify ( IN PDEVICE_EXTENSION  DeviceExtension)

Definition at line 313 of file notify.c.

314 {
315  PIRP Irp;
316  KIRQL OldIrql;
317  LIST_ENTRY CopyList;
318  PLIST_ENTRY NextEntry;
319 
320  /* Increase the epic number */
321  DeviceExtension->EpicNumber++;
322 
323  InitializeListHead(&CopyList);
324 
325  /* Copy all the pending IRPs for notification */
327  while (!IsListEmpty(&(DeviceExtension->IrpListHead)))
328  {
329  NextEntry = RemoveHeadList(&(DeviceExtension->IrpListHead));
330  Irp = CONTAINING_RECORD(NextEntry, IRP, Tail.Overlay.ListEntry);
332  InsertTailList(&CopyList, &(Irp->Tail.Overlay.ListEntry));
333  }
335 
336  /* Then, notify them one by one */
337  while (!IsListEmpty(&CopyList))
338  {
339  NextEntry = RemoveHeadList(&CopyList);
340  Irp = CONTAINING_RECORD(NextEntry, IRP, Tail.Overlay.ListEntry);
341 
342  *((PULONG)Irp->AssociatedIrp.SystemBuffer) = DeviceExtension->EpicNumber;
343  Irp->IoStatus.Information = sizeof(DeviceExtension->EpicNumber);
344 
346  }
347 }
_In_ PIRP Irp
Definition: csq.h:116
VOID NTAPI IoAcquireCancelSpinLock(OUT PKIRQL Irql)
Definition: util.c:56
IRP
Definition: iotypes.h:2463
#define InsertTailList(ListHead, Entry)
IoSetCancelRoutine(Irp, CancelRoutine)
_Must_inspect_result_ FORCEINLINE BOOLEAN IsListEmpty(_In_ const LIST_ENTRY *ListHead)
Definition: rtlfuncs.h:57
UCHAR KIRQL
Definition: env_spec_w32.h:591
smooth NULL
Definition: ftsmooth.c:416
#define IoCompleteRequest
Definition: irp.c:1240
FORCEINLINE PLIST_ENTRY RemoveHeadList(_Inout_ PLIST_ENTRY ListHead)
Definition: rtlfuncs.h:128
PFLT_MESSAGE_WAITER_QUEUE CONTAINING_RECORD(Csq, DEVICE_EXTENSION, IrpQueue)) -> WaiterQ.mLock) _IRQL_raises_(DISPATCH_LEVEL) VOID NTAPI FltpAcquireMessageWaiterLock(_In_ PIO_CSQ Csq, _Out_ PKIRQL Irql)
Definition: Messaging.c:560
_Requires_lock_held_ Interrupt _Releases_lock_ Interrupt _In_ _IRQL_restores_ KIRQL OldIrql
Definition: kefuncs.h:803
Definition: typedefs.h:117
VOID NTAPI IoReleaseCancelSpinLock(IN KIRQL Irql)
Definition: util.c:150
#define InitializeListHead(ListHead)
Definition: env_spec_w32.h:944
unsigned int * PULONG
Definition: retypes.h:1
#define IO_NO_INCREMENT
Definition: iotypes.h:566

Referenced by MountMgrCreatePointWorker(), MountMgrDeletePoints(), and MountMgrVolumeMountPointChanged().

◆ MountMgrNotifyNameChange()

VOID MountMgrNotifyNameChange ( IN PDEVICE_EXTENSION  DeviceExtension,
IN PUNICODE_STRING  DeviceName,
IN BOOLEAN  ValidateVolume 
)

Definition at line 353 of file notify.c.

356 {
357  PIRP Irp;
358  KEVENT Event;
360  PLIST_ENTRY NextEntry;
362  PIO_STACK_LOCATION Stack;
365  PDEVICE_RELATIONS DeviceRelations;
366  PDEVICE_INFORMATION DeviceInformation;
367  TARGET_DEVICE_CUSTOM_NOTIFICATION DeviceNotification;
368 
369  /* If we have to validate volume */
370  if (ValidateVolume)
371  {
372  /* Then, ensure we can find the device */
373  NextEntry = DeviceExtension->DeviceListHead.Flink;
374  while (NextEntry != &(DeviceExtension->DeviceListHead))
375  {
376  DeviceInformation = CONTAINING_RECORD(NextEntry, DEVICE_INFORMATION, DeviceListEntry);
377  if (RtlCompareUnicodeString(DeviceName, &(DeviceInformation->DeviceName), TRUE) == 0)
378  {
379  break;
380  }
381  }
382 
383  /* No need to notify for a PnP device or if we didn't find the device */
384  if (NextEntry == &(DeviceExtension->DeviceListHead) ||
385  !DeviceInformation->ManuallyRegistered)
386  {
387  return;
388  }
389  }
390 
391  /* Then, get device object */
394  &FileObject,
395  &DeviceObject);
396  if (!NT_SUCCESS(Status))
397  {
398  return;
399  }
400 
402 
404 
405  /* Set up empty IRP (yes, yes!) */
407  DeviceObject,
408  NULL,
409  0,
410  NULL,
411  0,
412  FALSE,
413  &Event,
414  &IoStatusBlock);
415  if (!Irp)
416  {
419  return;
420  }
421 
423 
424  Irp->IoStatus.Status = STATUS_NOT_SUPPORTED;
425  Irp->IoStatus.Information = 0;
426 
427  /* Properly set it, we want to query device relations */
428  Stack->MajorFunction = IRP_MJ_PNP;
430  Stack->Parameters.QueryDeviceRelations.Type = TargetDeviceRelation;
431  Stack->FileObject = FileObject;
432 
433  /* And call driver */
435  if (Status == STATUS_PENDING)
436  {
439  }
440 
443 
444  if (!NT_SUCCESS(Status))
445  {
446  return;
447  }
448 
449  /* Validate device return */
450  DeviceRelations = (PDEVICE_RELATIONS)IoStatusBlock.Information;
451  if (DeviceRelations->Count < 1)
452  {
453  ExFreePool(DeviceRelations);
454  return;
455  }
456 
457  DeviceObject = DeviceRelations->Objects[0];
458  ExFreePool(DeviceRelations);
459 
460  /* Set up real notification */
461  DeviceNotification.Version = 1;
462  DeviceNotification.Size = sizeof(TARGET_DEVICE_CUSTOM_NOTIFICATION);
463  DeviceNotification.Event = GUID_IO_VOLUME_NAME_CHANGE;
464  DeviceNotification.FileObject = NULL;
465  DeviceNotification.NameBufferOffset = -1;
466 
467  /* And report */
469  &DeviceNotification,
470  NULL, NULL);
471 
473 
474  return;
475 }
#define TRUE
Definition: types.h:120
struct _TARGET_DEVICE_CUSTOM_NOTIFICATION TARGET_DEVICE_CUSTOM_NOTIFICATION
_In_ PIRP Irp
Definition: csq.h:116
PDEVICE_OBJECT Objects[1]
Definition: iotypes.h:2055
#define IRP_MJ_PNP
Definition: cdrw_usr.h:52
LONG NTSTATUS
Definition: precomp.h:26
NTSTATUS NTAPI IoGetDeviceObjectPointer(IN PUNICODE_STRING ObjectName, IN ACCESS_MASK DesiredAccess, OUT PFILE_OBJECT *FileObject, OUT PDEVICE_OBJECT *DeviceObject)
Definition: device.c:1435
VOID NTAPI ObDereferenceObject(IN PVOID Object)
Definition: obref.c:375
WCHAR DeviceName[]
Definition: adapter.cpp:21
NTSTATUS NTAPI KeWaitForSingleObject(IN PVOID Object, IN KWAIT_REASON WaitReason, IN KPROCESSOR_MODE WaitMode, IN BOOLEAN Alertable, IN PLARGE_INTEGER Timeout OPTIONAL)
Definition: wait.c:416
_In_ PVOID _In_ ULONG Event
Definition: iotypes.h:435
struct _DEVICE_RELATIONS * PDEVICE_RELATIONS
smooth NULL
Definition: ftsmooth.c:416
_Inout_ PFILE_OBJECT FileObject
Definition: cdprocs.h:593
PFLT_MESSAGE_WAITER_QUEUE CONTAINING_RECORD(Csq, DEVICE_EXTENSION, IrpQueue)) -> WaiterQ.mLock) _IRQL_raises_(DISPATCH_LEVEL) VOID NTAPI FltpAcquireMessageWaiterLock(_In_ PIO_CSQ Csq, _Out_ PKIRQL Irql)
Definition: Messaging.c:560
PDEVICE_OBJECT NTAPI IoGetAttachedDeviceReference(PDEVICE_OBJECT DeviceObject)
Definition: device.c:1406
struct _LIST_ENTRY * Flink
Definition: typedefs.h:119
if(!(yy_init))
Definition: macro.lex.yy.c:714
#define NT_SUCCESS(StatCode)
Definition: apphelp.c:32
#define STATUS_PENDING
Definition: ntstatus.h:82
* PFILE_OBJECT
Definition: iotypes.h:1955
#define FILE_READ_ATTRIBUTES
Definition: nt_native.h:647
ULONG RtlCompareUnicodeString(PUNICODE_STRING s1, PUNICODE_STRING s2, BOOLEAN UpCase)
Definition: string_lib.cpp:31
Definition: typedefs.h:117
__drv_aliasesMem FORCEINLINE PIO_STACK_LOCATION IoGetNextIrpStackLocation(_In_ PIRP Irp)
Definition: iofuncs.h:2647
BOOLEAN ManuallyRegistered
Definition: mntmgr.h:53
Status
Definition: gdiplustypes.h:24
IN PDEVICE_OBJECT DeviceObject
Definition: fatprocs.h:1560
PFILE_OBJECT FileObject
Definition: iotypes.h:2813
#define KeInitializeEvent(pEvt, foo, foo2)
Definition: env_spec_w32.h:477
static OUT PIO_STATUS_BLOCK IoStatusBlock
Definition: pipe.c:75
NTSTATUS NTAPI IoCallDriver(IN PDEVICE_OBJECT DeviceObject, IN PIRP Irp)
Definition: irp.c:1218
#define IRP_MN_QUERY_DEVICE_RELATIONS
PIRP NTAPI IoBuildDeviceIoControlRequest(IN ULONG IoControlCode, IN PDEVICE_OBJECT DeviceObject, IN PVOID InputBuffer, IN ULONG InputBufferLength, IN PVOID OutputBuffer, IN ULONG OutputBufferLength, IN BOOLEAN InternalDeviceIoControl, IN PKEVENT Event, IN PIO_STATUS_BLOCK IoStatusBlock)
Definition: irp.c:881
UNICODE_STRING DeviceName
Definition: mntmgr.h:50
#define STATUS_NOT_SUPPORTED
Definition: ntstatus.h:409
struct _FILE_OBJECT * FileObject
Definition: iotypes.h:975
struct _NAMED_PIPE_CREATE_PARAMETERS * Parameters
Definition: iotypes.h:2772
NTSTATUS NTAPI IoReportTargetDeviceChangeAsynchronous(IN PDEVICE_OBJECT PhysicalDeviceObject, IN PVOID NotificationStructure, IN PDEVICE_CHANGE_COMPLETE_CALLBACK Callback OPTIONAL, IN PVOID Context OPTIONAL)
Definition: pnpreport.c:519
#define ExFreePool(addr)
Definition: env_spec_w32.h:352

Referenced by MountMgrCreatePointWorker(), MountMgrDeletePoints(), MountMgrQueryDosVolumePaths(), and MountMgrVolumeMountPointChanged().

◆ MountMgrTargetDeviceNotification()

NTSTATUS NTAPI MountMgrTargetDeviceNotification ( IN PVOID  NotificationStructure,
IN PVOID  Context 
)

Definition at line 229 of file notify.c.

231 {
232  PDEVICE_EXTENSION DeviceExtension;
233  PDEVICE_INFORMATION DeviceInformation;
235 
236  DeviceInformation = Context;
237  DeviceExtension = DeviceInformation->DeviceExtension;
238  Notification = NotificationStructure;
239 
240  /* If it's to signal that removal is complete, then, execute the function */
241  if (IsEqualGUID(&(Notification->Event), &GUID_TARGET_DEVICE_REMOVE_COMPLETE))
242  {
243  MountMgrMountedDeviceRemoval(DeviceExtension, Notification->SymbolicLinkName);
244  }
245  /* It it's to signal that a volume has been mounted
246  * Verify if a database sync is required and execute it
247  */
248  else if (IsEqualGUID(&(Notification->Event), &GUID_IO_VOLUME_MOUNT))
249  {
250  /* If we were already mounted, then mark us unmounted */
251  if (InterlockedCompareExchange(&(DeviceInformation->MountState),
252  FALSE,
253  FALSE) == TRUE)
254  {
255  InterlockedDecrement(&(DeviceInformation->MountState));
256  }
257  /* Otherwise, start mounting the device and first, reconcile its DB if required */
258  else
259  {
260  if (DeviceInformation->NeedsReconcile)
261  {
262  DeviceInformation->NeedsReconcile = FALSE;
263  ReconcileThisDatabaseWithMaster(DeviceExtension, DeviceInformation);
264  }
265  }
266  }
267 
268  return STATUS_SUCCESS;
269 }
_In_ PVOID NotificationStructure
Definition: iofuncs.h:1203
#define TRUE
Definition: types.h:120
#define InterlockedCompareExchange
Definition: interlocked.h:104
VOID MountMgrMountedDeviceRemoval(IN PDEVICE_EXTENSION Extension, IN PUNICODE_STRING DeviceName)
Definition: mountmgr.c:1451
VOID ReconcileThisDatabaseWithMaster(IN PDEVICE_EXTENSION DeviceExtension, IN PDEVICE_INFORMATION DeviceInformation)
Definition: database.c:1613
BOOLEAN NeedsReconcile
Definition: mntmgr.h:56
PDEVICE_EXTENSION DeviceExtension
Definition: mntmgr.h:62
#define InterlockedDecrement
Definition: armddk.h:52
BOOL WINAPI IsEqualGUID(REFGUID rguid1, REFGUID rguid2)
Definition: compobj.c:4112
struct tagContext Context
Definition: acpixf.h:1030
return STATUS_SUCCESS
Definition: btrfs.c:2938

Referenced by RegisterForTargetDeviceNotification().

◆ PostOnlineNotification()

VOID PostOnlineNotification ( IN PDEVICE_EXTENSION  DeviceExtension,
IN PUNICODE_STRING  SymbolicName 
)

Definition at line 145 of file notify.c.

147 {
148  KIRQL OldIrql;
150 
151  /* Allocate a notification work item */
152  WorkItem = AllocatePool(sizeof(ONLINE_NOTIFICATION_WORK_ITEM));
153  if (!WorkItem)
154  {
155  return;
156  }
157 
159  WorkItem->DeviceExtension = DeviceExtension;
160  WorkItem->SymbolicName.Length = SymbolicName->Length;
161  WorkItem->SymbolicName.MaximumLength = SymbolicName->Length + sizeof(WCHAR);
163  if (!WorkItem->SymbolicName.Buffer)
164  {
165  FreePool(WorkItem);
166  return;
167  }
168 
170  WorkItem->SymbolicName.Buffer[SymbolicName->Length / sizeof(WCHAR)] = UNICODE_NULL;
171 
172  KeAcquireSpinLock(&(DeviceExtension->WorkerLock), &OldIrql);
173  DeviceExtension->OnlineNotificationCount++;
174 
175  /* If no worker are active */
176  if (DeviceExtension->OnlineNotificationWorkerActive == 0)
177  {
178  /* Queue that one for execution */
179  DeviceExtension->OnlineNotificationWorkerActive = 1;
181  }
182  else
183  {
184  /* Otherwise, just put it in the queue list */
185  InsertTailList(&(DeviceExtension->OnlineNotificationListHead), &(WorkItem->WorkItem.List));
186  }
187 
188  KeReleaseSpinLock(&(DeviceExtension->WorkerLock), OldIrql);
189 
190  return;
191 }
NTSYSAPI VOID NTAPI RtlCopyMemory(VOID UNALIGNED *Destination, CONST VOID UNALIGNED *Source, ULONG Length)
VOID NTAPI ExQueueWorkItem(IN PWORK_QUEUE_ITEM WorkItem, IN WORK_QUEUE_TYPE QueueType)
Definition: work.c:717
LIST_ENTRY List
Definition: extypes.h:203
USHORT MaximumLength
Definition: env_spec_w32.h:370
_In_ PNDIS_STRING _In_ PNDIS_STRING SymbolicName
Definition: ndis.h:4676
#define InsertTailList(ListHead, Entry)
#define FreePool(P)
Definition: mntmgr.h:154
UCHAR KIRQL
Definition: env_spec_w32.h:591
WORK_QUEUE_ITEM WorkItem
Definition: mntmgr.h:110
#define UNICODE_NULL
VOID NTAPI SendOnlineNotificationWorker(IN PVOID Parameter)
Definition: notify.c:96
#define ExInitializeWorkItem(Item, Routine, Context)
Definition: exfuncs.h:265
__wchar_t WCHAR
Definition: xmlstorage.h:180
PDEVICE_EXTENSION DeviceExtension
Definition: mntmgr.h:111
#define KeAcquireSpinLock(sl, irql)
Definition: env_spec_w32.h:609
_Requires_lock_held_ Interrupt _Releases_lock_ Interrupt _In_ _IRQL_restores_ KIRQL OldIrql
Definition: kefuncs.h:803
#define AllocatePool(Size)
Definition: mntmgr.h:153
#define KeReleaseSpinLock(sl, irql)
Definition: env_spec_w32.h:627
UNICODE_STRING SymbolicName
Definition: mntmgr.h:112

Referenced by MountMgrCreatePointWorker(), MountMgrVolumeMountPointCreated(), OnlineMountedVolumes(), and ReconcileThisDatabaseWithMasterWorker().

◆ RegisterForTargetDeviceNotification()

VOID RegisterForTargetDeviceNotification ( IN PDEVICE_EXTENSION  DeviceExtension,
IN PDEVICE_INFORMATION  DeviceInformation 
)

Definition at line 275 of file notify.c.

277 {
281 
282  /* Get device object */
283  Status = IoGetDeviceObjectPointer(&(DeviceInformation->DeviceName),
285  &FileObject,
286  &DeviceObject);
287  if (!NT_SUCCESS(Status))
288  {
289  return;
290  }
291 
292  /* And simply register for notifications */
294  0, FileObject,
295  DeviceExtension->DriverObject,
297  DeviceInformation,
298  &(DeviceInformation->TargetDeviceNotificationEntry));
299  if (!NT_SUCCESS(Status))
300  {
301  DeviceInformation->TargetDeviceNotificationEntry = NULL;
302  }
303 
305 
306  return;
307 }
LONG NTSTATUS
Definition: precomp.h:26
NTSTATUS NTAPI MountMgrTargetDeviceNotification(IN PVOID NotificationStructure, IN PVOID Context)
Definition: notify.c:229
NTSTATUS NTAPI IoGetDeviceObjectPointer(IN PUNICODE_STRING ObjectName, IN ACCESS_MASK DesiredAccess, OUT PFILE_OBJECT *FileObject, OUT PDEVICE_OBJECT *DeviceObject)
Definition: device.c:1435
VOID NTAPI ObDereferenceObject(IN PVOID Object)
Definition: obref.c:375
smooth NULL
Definition: ftsmooth.c:416
_Inout_ PFILE_OBJECT FileObject
Definition: cdprocs.h:593
#define NT_SUCCESS(StatCode)
Definition: apphelp.c:32
* PFILE_OBJECT
Definition: iotypes.h:1955
#define FILE_READ_ATTRIBUTES
Definition: nt_native.h:647
NTSTATUS NTAPI IoRegisterPlugPlayNotification(IN IO_NOTIFICATION_EVENT_CATEGORY EventCategory, IN ULONG EventCategoryFlags, IN PVOID EventCategoryData OPTIONAL, IN PDRIVER_OBJECT DriverObject, IN PDRIVER_NOTIFICATION_CALLBACK_ROUTINE CallbackRoutine, IN PVOID Context, OUT PVOID *NotificationEntry)
Definition: pnpnotify.c:249
Status
Definition: gdiplustypes.h:24
IN PDEVICE_OBJECT DeviceObject
Definition: fatprocs.h:1560

Referenced by MountMgrMountedDeviceArrival().

◆ RemoveWorkItem()

VOID RemoveWorkItem ( IN PUNIQUE_ID_WORK_ITEM  WorkItem)

Definition at line 481 of file notify.c.

482 {
483  PDEVICE_EXTENSION DeviceExtension = WorkItem->DeviceExtension;
484 
485  KeWaitForSingleObject(&(DeviceExtension->DeviceLock), Executive, KernelMode, FALSE, NULL);
486 
487  /* If even if being worked, it's too late */
488  if (WorkItem->Event)
489  {
490  KeReleaseSemaphore(&(DeviceExtension->DeviceLock), IO_NO_INCREMENT, 1, FALSE);
491  KeSetEvent(WorkItem->Event, 0, FALSE);
492  }
493  else
494  {
495  /* Otherwise, remove it from the list, and delete it */
496  RemoveEntryList(&(WorkItem->UniqueIdWorkerItemListEntry));
497  KeReleaseSemaphore(&(DeviceExtension->DeviceLock), IO_NO_INCREMENT, 1, FALSE);
498  IoFreeIrp(WorkItem->Irp);
499  FreePool(WorkItem->DeviceName.Buffer);
500  FreePool(WorkItem->IrpBuffer);
501  FreePool(WorkItem);
502  }
503 }
LONG NTAPI KeSetEvent(IN PKEVENT Event, IN KPRIORITY Increment, IN BOOLEAN Wait)
Definition: eventobj.c:159
#define FreePool(P)
Definition: mntmgr.h:154
NTSTATUS NTAPI KeWaitForSingleObject(IN PVOID Object, IN KWAIT_REASON WaitReason, IN KPROCESSOR_MODE WaitMode, IN BOOLEAN Alertable, IN PLARGE_INTEGER Timeout OPTIONAL)
Definition: wait.c:416
FORCEINLINE BOOLEAN RemoveEntryList(_In_ PLIST_ENTRY Entry)
Definition: rtlfuncs.h:105
smooth NULL
Definition: ftsmooth.c:416
LONG NTAPI KeReleaseSemaphore(IN PKSEMAPHORE Semaphore, IN KPRIORITY Increment, IN LONG Adjustment, IN BOOLEAN Wait)
Definition: semphobj.c:54
VOID NTAPI IoFreeIrp(IN PIRP Irp)
Definition: irp.c:1666
#define IO_NO_INCREMENT
Definition: iotypes.h:566

Referenced by IssueUniqueIdChangeNotifyWorker(), and UniqueIdChangeNotifyWorker().

◆ SendOnlineNotification()

VOID SendOnlineNotification ( IN PUNICODE_STRING  SymbolicName)

Definition at line 38 of file notify.c.

39 {
40  PIRP Irp;
41  KEVENT Event;
44  PIO_STACK_LOCATION Stack;
47 
48  /* Get device object */
51  &FileObject,
52  &DeviceObject);
53  if (!NT_SUCCESS(Status))
54  {
55  return;
56  }
57 
58  /* And attached device object */
60 
61  /* And send VOLUME_ONLINE */
65  NULL, 0,
66  NULL, 0,
67  FALSE,
68  &Event,
69  &IoStatusBlock);
70  if (!Irp)
71  {
72  goto Cleanup;
73  }
74 
76  Stack->FileObject = FileObject;
77 
79  if (Status == STATUS_PENDING)
80  {
82  }
83 
84 Cleanup:
87 
88  return;
89 }
_In_ PIRP Irp
Definition: csq.h:116
_In_ PNDIS_STRING _In_ PNDIS_STRING SymbolicName
Definition: ndis.h:4676
LONG NTSTATUS
Definition: precomp.h:26
NTSTATUS NTAPI IoGetDeviceObjectPointer(IN PUNICODE_STRING ObjectName, IN ACCESS_MASK DesiredAccess, OUT PFILE_OBJECT *FileObject, OUT PDEVICE_OBJECT *DeviceObject)
Definition: device.c:1435
VOID NTAPI ObDereferenceObject(IN PVOID Object)
Definition: obref.c:375
NTSTATUS NTAPI KeWaitForSingleObject(IN PVOID Object, IN KWAIT_REASON WaitReason, IN KPROCESSOR_MODE WaitMode, IN BOOLEAN Alertable, IN PLARGE_INTEGER Timeout OPTIONAL)
Definition: wait.c:416
_In_ PVOID _In_ ULONG Event
Definition: iotypes.h:435
smooth NULL
Definition: ftsmooth.c:416
#define IOCTL_VOLUME_ONLINE
Definition: ntddvol.h:62
_Inout_ PFILE_OBJECT FileObject
Definition: cdprocs.h:593
PDEVICE_OBJECT NTAPI IoGetAttachedDeviceReference(PDEVICE_OBJECT DeviceObject)
Definition: device.c:1406
#define NT_SUCCESS(StatCode)
Definition: apphelp.c:32
#define STATUS_PENDING
Definition: ntstatus.h:82
* PFILE_OBJECT
Definition: iotypes.h:1955
#define FILE_READ_ATTRIBUTES
Definition: nt_native.h:647
static const WCHAR Cleanup[]
Definition: register.c:80
__drv_aliasesMem FORCEINLINE PIO_STACK_LOCATION IoGetNextIrpStackLocation(_In_ PIRP Irp)
Definition: iofuncs.h:2647
Status
Definition: gdiplustypes.h:24
IN PDEVICE_OBJECT DeviceObject
Definition: fatprocs.h:1560
PFILE_OBJECT FileObject
Definition: iotypes.h:2813
#define KeInitializeEvent(pEvt, foo, foo2)
Definition: env_spec_w32.h:477
static OUT PIO_STATUS_BLOCK IoStatusBlock
Definition: pipe.c:75
NTSTATUS NTAPI IoCallDriver(IN PDEVICE_OBJECT DeviceObject, IN PIRP Irp)
Definition: irp.c:1218
PIRP NTAPI IoBuildDeviceIoControlRequest(IN ULONG IoControlCode, IN PDEVICE_OBJECT DeviceObject, IN PVOID InputBuffer, IN ULONG InputBufferLength, IN PVOID OutputBuffer, IN ULONG OutputBufferLength, IN BOOLEAN InternalDeviceIoControl, IN PKEVENT Event, IN PIO_STATUS_BLOCK IoStatusBlock)
Definition: irp.c:881

Referenced by MountMgrMountedDeviceArrival(), and SendOnlineNotificationWorker().

◆ SendOnlineNotificationWorker()

VOID NTAPI SendOnlineNotificationWorker ( IN PVOID  Parameter)

Definition at line 96 of file notify.c.

97 {
98  KIRQL OldIrql;
99  PLIST_ENTRY Head;
100  PDEVICE_EXTENSION DeviceExtension;
102  PONLINE_NOTIFICATION_WORK_ITEM NewWorkItem;
103 
105  DeviceExtension = WorkItem->DeviceExtension;
106 
107  /* First, send the notification */
108  SendOnlineNotification(&(WorkItem->SymbolicName));
109 
110  KeAcquireSpinLock(&(DeviceExtension->WorkerLock), &OldIrql);
111  /* If there are no notifications running any longer, reset event */
112  if (--DeviceExtension->OnlineNotificationCount == 0)
113  {
114  KeSetEvent(&(DeviceExtension->OnlineNotificationEvent), 0, FALSE);
115  }
116 
117  /* If there are still notifications in queue */
118  if (!IsListEmpty(&(DeviceExtension->OnlineNotificationListHead)))
119  {
120  /* Queue a new one for execution */
121  Head = RemoveHeadList(&(DeviceExtension->OnlineNotificationListHead));
122  NewWorkItem = CONTAINING_RECORD(Head, ONLINE_NOTIFICATION_WORK_ITEM, WorkItem.List);
123  KeReleaseSpinLock(&(DeviceExtension->WorkerLock), OldIrql);
124  NewWorkItem->WorkItem.List.Blink = NULL;
125  NewWorkItem->WorkItem.List.Flink = NULL;
126  ExQueueWorkItem(&NewWorkItem->WorkItem, DelayedWorkQueue);
127  }
128  else
129  {
130  /* Mark it's over */
131  DeviceExtension->OnlineNotificationWorkerActive = 0;
132  KeReleaseSpinLock(&(DeviceExtension->WorkerLock), OldIrql);
133  }
134 
135  FreePool(WorkItem->SymbolicName.Buffer);
136  FreePool(WorkItem);
137 
138  return;
139 }
VOID NTAPI ExQueueWorkItem(IN PWORK_QUEUE_ITEM WorkItem, IN WORK_QUEUE_TYPE QueueType)
Definition: work.c:717
LIST_ENTRY List
Definition: extypes.h:203
VOID SendOnlineNotification(IN PUNICODE_STRING SymbolicName)
Definition: notify.c:38
struct _LIST_ENTRY * Blink
Definition: typedefs.h:120
_In_ PVOID Parameter
Definition: ldrtypes.h:241
LONG NTAPI KeSetEvent(IN PKEVENT Event, IN KPRIORITY Increment, IN BOOLEAN Wait)
Definition: eventobj.c:159
#define FreePool(P)
Definition: mntmgr.h:154
_Must_inspect_result_ FORCEINLINE BOOLEAN IsListEmpty(_In_ const LIST_ENTRY *ListHead)
Definition: rtlfuncs.h:57
UCHAR KIRQL
Definition: env_spec_w32.h:591
WORK_QUEUE_ITEM WorkItem
Definition: mntmgr.h:110
smooth NULL
Definition: ftsmooth.c:416
FORCEINLINE PLIST_ENTRY RemoveHeadList(_Inout_ PLIST_ENTRY ListHead)
Definition: rtlfuncs.h:128
PFLT_MESSAGE_WAITER_QUEUE CONTAINING_RECORD(Csq, DEVICE_EXTENSION, IrpQueue)) -> WaiterQ.mLock) _IRQL_raises_(DISPATCH_LEVEL) VOID NTAPI FltpAcquireMessageWaiterLock(_In_ PIO_CSQ Csq, _Out_ PKIRQL Irql)
Definition: Messaging.c:560
struct _LIST_ENTRY * Flink
Definition: typedefs.h:119
PDEVICE_EXTENSION DeviceExtension
Definition: mntmgr.h:111
#define KeAcquireSpinLock(sl, irql)
Definition: env_spec_w32.h:609
_Requires_lock_held_ Interrupt _Releases_lock_ Interrupt _In_ _IRQL_restores_ KIRQL OldIrql
Definition: kefuncs.h:803
Definition: typedefs.h:117
struct _ONLINE_NOTIFICATION_WORK_ITEM * PONLINE_NOTIFICATION_WORK_ITEM
#define KeReleaseSpinLock(sl, irql)
Definition: env_spec_w32.h:627
UNICODE_STRING SymbolicName
Definition: mntmgr.h:112

Referenced by PostOnlineNotification().

◆ UniqueIdChangeNotifyCompletion()

NTSTATUS NTAPI UniqueIdChangeNotifyCompletion ( IN PDEVICE_OBJECT  DeviceObject,
IN PIRP  Irp,
IN PVOID  Context 
)

Definition at line 569 of file notify.c.

572 {
573  PUNIQUE_ID_WORK_ITEM WorkItem = Context;
574 
577 
578  /* Simply queue the work item */
579  IoQueueWorkItem(WorkItem->WorkItem,
582  WorkItem);
583 
585 }
#define STATUS_MORE_PROCESSING_REQUIRED
Definition: shellext.h:68
_In_ PIRP Irp
Definition: csq.h:116
#define UNREFERENCED_PARAMETER(P)
Definition: ntbasedef.h:323
VOID NTAPI UniqueIdChangeNotifyWorker(IN PDEVICE_OBJECT DeviceObject, IN PVOID Context)
Definition: notify.c:510
PIO_WORKITEM WorkItem
Definition: mntmgr.h:142
VOID NTAPI IoQueueWorkItem(IN PIO_WORKITEM IoWorkItem, IN PIO_WORKITEM_ROUTINE WorkerRoutine, IN WORK_QUEUE_TYPE QueueType, IN PVOID Context)
Definition: iowork.c:40
IN PDEVICE_OBJECT DeviceObject
Definition: fatprocs.h:1560
struct tagContext Context
Definition: acpixf.h:1030

Referenced by IssueUniqueIdChangeNotifyWorker().

◆ UniqueIdChangeNotifyWorker()

VOID NTAPI UniqueIdChangeNotifyWorker ( IN PDEVICE_OBJECT  DeviceObject,
IN PVOID  Context 
)

Definition at line 510 of file notify.c.

512 {
513  PUNIQUE_ID_WORK_ITEM WorkItem = Context;
514  PMOUNTDEV_UNIQUE_ID OldUniqueId, NewUniqueId;
516 
518 
519  /* Validate worker */
520  if (!NT_SUCCESS(WorkItem->Irp->IoStatus.Status))
521  {
522  RemoveWorkItem(WorkItem);
523  return;
524  }
525 
526  UniqueIdChange = WorkItem->Irp->AssociatedIrp.SystemBuffer;
527  /* Get the old unique ID */
528  OldUniqueId = AllocatePool(UniqueIdChange->OldUniqueIdLength + sizeof(MOUNTDEV_UNIQUE_ID));
529  if (!OldUniqueId)
530  {
531  RemoveWorkItem(WorkItem);
532  return;
533  }
534 
535  OldUniqueId->UniqueIdLength = UniqueIdChange->OldUniqueIdLength;
536  RtlCopyMemory(OldUniqueId->UniqueId,
537  (PVOID)((ULONG_PTR)UniqueIdChange + UniqueIdChange->OldUniqueIdOffset),
538  UniqueIdChange->OldUniqueIdLength);
539 
540  /* Get the new unique ID */
541  NewUniqueId = AllocatePool(UniqueIdChange->NewUniqueIdLength + sizeof(MOUNTDEV_UNIQUE_ID));
542  if (!NewUniqueId)
543  {
544  FreePool(OldUniqueId);
545  RemoveWorkItem(WorkItem);
546  return;
547  }
548 
549  NewUniqueId->UniqueIdLength = UniqueIdChange->NewUniqueIdLength;
550  RtlCopyMemory(NewUniqueId->UniqueId,
551  (PVOID)((ULONG_PTR)UniqueIdChange + UniqueIdChange->NewUniqueIdOffset),
552  UniqueIdChange->NewUniqueIdLength);
553 
554  /* Call the real worker */
555  MountMgrUniqueIdChangeRoutine(WorkItem->DeviceExtension, OldUniqueId, NewUniqueId);
556  IssueUniqueIdChangeNotifyWorker(WorkItem, NewUniqueId);
557 
558  FreePool(NewUniqueId);
559  FreePool(OldUniqueId);
560 
561  return;
562 }
NTSYSAPI VOID NTAPI RtlCopyMemory(VOID UNALIGNED *Destination, CONST VOID UNALIGNED *Source, ULONG Length)
#define UNREFERENCED_PARAMETER(P)
Definition: ntbasedef.h:323
USHORT UniqueIdLength
Definition: imports.h:138
PDEVICE_EXTENSION DeviceExtension
Definition: mntmgr.h:143
#define FreePool(P)
Definition: mntmgr.h:154
uint32_t ULONG_PTR
Definition: typedefs.h:63
VOID IssueUniqueIdChangeNotifyWorker(IN PUNIQUE_ID_WORK_ITEM WorkItem, IN PMOUNTDEV_UNIQUE_ID UniqueId)
Definition: notify.c:591
#define NT_SUCCESS(StatCode)
Definition: apphelp.c:32
#define AllocatePool(Size)
Definition: mntmgr.h:153
IN PDEVICE_OBJECT DeviceObject
Definition: fatprocs.h:1560
VOID RemoveWorkItem(IN PUNIQUE_ID_WORK_ITEM WorkItem)
Definition: notify.c:481
struct tagContext Context
Definition: acpixf.h:1030
VOID MountMgrUniqueIdChangeRoutine(IN PDEVICE_EXTENSION DeviceExtension, IN PMOUNTDEV_UNIQUE_ID OldUniqueId, IN PMOUNTDEV_UNIQUE_ID NewUniqueId)
Definition: uniqueid.c:71
UCHAR UniqueId[1]
Definition: imports.h:139

Referenced by UniqueIdChangeNotifyCompletion().

◆ WaitForOnlinesToComplete()

VOID WaitForOnlinesToComplete ( IN PDEVICE_EXTENSION  DeviceExtension)

Definition at line 197 of file notify.c.

198 {
199  KIRQL OldIrql;
200 
201  KeInitializeEvent(&(DeviceExtension->OnlineNotificationEvent), NotificationEvent, FALSE);
202 
203  KeAcquireSpinLock(&(DeviceExtension->WorkerLock), &OldIrql);
204 
205  /* Just wait all the worker are done */
206  if (DeviceExtension->OnlineNotificationCount != 1)
207  {
208  DeviceExtension->OnlineNotificationCount--;
209  KeReleaseSpinLock(&(DeviceExtension->WorkerLock), OldIrql);
210 
211  KeWaitForSingleObject(&(DeviceExtension->OnlineNotificationEvent),
212  Executive,
213  KernelMode,
214  FALSE,
215  NULL);
216 
217  KeAcquireSpinLock(&(DeviceExtension->WorkerLock), &OldIrql);
218  DeviceExtension->OnlineNotificationCount++;
219  }
220 
221  KeReleaseSpinLock(&(DeviceExtension->WorkerLock), OldIrql);
222 }
NTSTATUS NTAPI KeWaitForSingleObject(IN PVOID Object, IN KWAIT_REASON WaitReason, IN KPROCESSOR_MODE WaitMode, IN BOOLEAN Alertable, IN PLARGE_INTEGER Timeout OPTIONAL)
Definition: wait.c:416
UCHAR KIRQL
Definition: env_spec_w32.h:591
smooth NULL
Definition: ftsmooth.c:416
#define KeAcquireSpinLock(sl, irql)
Definition: env_spec_w32.h:609
_Requires_lock_held_ Interrupt _Releases_lock_ Interrupt _In_ _IRQL_restores_ KIRQL OldIrql
Definition: kefuncs.h:803
#define KeInitializeEvent(pEvt, foo, foo2)
Definition: env_spec_w32.h:477
#define KeReleaseSpinLock(sl, irql)
Definition: env_spec_w32.h:627

Referenced by MountMgrDeviceControl().