mirror of
https://github.com/reactos/reactos.git
synced 2024-09-03 17:19:20 +00:00
![George Bișoc](/assets/img/avatar_default.png)
This fixes an issue where ReactOS would assert on QuotaUsage == 0 as the process was still taking up quotas during a quota block de-reference with root cause of ThisBufferLength member field being 0 which made process quota charging/returning flow unbalanced. In addition to that, on FsRtlCancelNotify routine API all we must ensure that if PsChargePoolQuota or ExAllocatePoolWithTag function fails we have to handle the raised exceptions accordingly and return the charged quota back (if we actually charged quotas that is). With said, wrap that part of code with SEH. === DOCUMENTATION REMARKS === The cause of the assert is due to the fact ThisBufferLength was being handled wrongly ever since, until this commit. When FsRtl of the Executive has to filter reported changes (with logic algorithm implemented in FsRtlNotifyFilterReportChange function), the said function will charge the quota of a given process with an amount that is represented as the buffer length whose size is expressed in bytes. This length buffer is preserved in a local variable called NumberOfBytes, which is initialized from BufferLength member field of notification structure or from the length from stack parameters pointed from an IRP. As it currently stands, the code is implemented in such a way that FsRtlNotifyFilterReportChange will charge quotas to a process but it doesn't assign the buffer length to ThisBufferLength. On the first glimpse ThisBufferLength and BufferLength are very similar members that serve exact same purpose but in reality there's a subtle distinction between the two. BufferLength is a member whose length size is given by FSDs (filesystem drivers) during a notification dispatching. Whenever FsRtl receives the notification structure packed with data from the filesystem, the length pointed by BufferLength gets passed to ThisBufferLength and from now on the kernel has to use this member for the whole time of its task to accomplish whatever request it's been given by the filesystem. In other words, BufferLength is strictly used only to pass length size data to the kernel by initializing ThisBufferLength based on that length and unequivocally the kernel uses this member field. What we're doing is that ThisBufferLength never receives the length from BufferLength therefore whenever FsRtl component has to return quotas back it'll return an amount of 0 (which means no amount to return) and that's a bug in the kernel.
1677 lines
58 KiB
C
1677 lines
58 KiB
C
/*
|
|
* PROJECT: ReactOS Kernel
|
|
* LICENSE: GPL - See COPYING in the top level directory
|
|
* FILE: ntoskrnl/fsrtl/notify.c
|
|
* PURPOSE: Change Notifications and Sync for File System Drivers
|
|
* PROGRAMMERS: Pierre Schweitzer
|
|
*/
|
|
|
|
/* INCLUDES ******************************************************************/
|
|
|
|
#include <ntoskrnl.h>
|
|
#define NDEBUG
|
|
#include <debug.h>
|
|
|
|
/* INLINED FUNCTIONS *********************************************************/
|
|
|
|
/*
|
|
* @implemented
|
|
*/
|
|
FORCEINLINE
|
|
VOID
|
|
FsRtlNotifyAcquireFastMutex(IN PREAL_NOTIFY_SYNC RealNotifySync)
|
|
{
|
|
ULONG_PTR CurrentThread = (ULONG_PTR)KeGetCurrentThread();
|
|
|
|
/* Only acquire fast mutex if it's not already acquired by the current thread */
|
|
if (RealNotifySync->OwningThread != CurrentThread)
|
|
{
|
|
ExAcquireFastMutexUnsafe(&(RealNotifySync->FastMutex));
|
|
RealNotifySync->OwningThread = CurrentThread;
|
|
}
|
|
/* Whatever the case, keep trace of the attempt to acquire fast mutex */
|
|
RealNotifySync->OwnerCount++;
|
|
}
|
|
|
|
/*
|
|
* @implemented
|
|
*/
|
|
FORCEINLINE
|
|
VOID
|
|
FsRtlNotifyReleaseFastMutex(IN PREAL_NOTIFY_SYNC RealNotifySync)
|
|
{
|
|
RealNotifySync->OwnerCount--;
|
|
/* Release the fast mutex only if no other instance needs it */
|
|
if (!RealNotifySync->OwnerCount)
|
|
{
|
|
ExReleaseFastMutexUnsafe(&(RealNotifySync->FastMutex));
|
|
RealNotifySync->OwningThread = (ULONG_PTR)0;
|
|
}
|
|
}
|
|
|
|
#define FsRtlNotifyGetLastPartOffset(FullLen, TargLen, Type, Chr) \
|
|
for (FullPosition = 0; FullPosition < FullLen; ++FullPosition) \
|
|
if (((Type)NotifyChange->FullDirectoryName->Buffer)[FullPosition] == Chr) \
|
|
++FullNumberOfParts; \
|
|
for (LastPartOffset = 0; LastPartOffset < TargLen; ++LastPartOffset) { \
|
|
if ( ((Type)TargetDirectory.Buffer)[LastPartOffset] == Chr) { \
|
|
++TargetNumberOfParts; \
|
|
if (TargetNumberOfParts == FullNumberOfParts) \
|
|
break; \
|
|
} \
|
|
}
|
|
|
|
/* PRIVATE FUNCTIONS *********************************************************/
|
|
|
|
VOID
|
|
FsRtlNotifyCompleteIrpList(IN PNOTIFY_CHANGE NotifyChange,
|
|
IN NTSTATUS Status);
|
|
|
|
BOOLEAN
|
|
FsRtlNotifySetCancelRoutine(IN PIRP Irp,
|
|
IN PNOTIFY_CHANGE NotifyChange OPTIONAL);
|
|
|
|
/*
|
|
* @implemented
|
|
*/
|
|
VOID
|
|
NTAPI
|
|
FsRtlCancelNotify(IN PDEVICE_OBJECT DeviceObject,
|
|
IN PIRP Irp)
|
|
{
|
|
PVOID Buffer;
|
|
PIRP NotifyIrp;
|
|
ULONG BufferLength;
|
|
PIO_STACK_LOCATION Stack;
|
|
PNOTIFY_CHANGE NotifyChange;
|
|
PREAL_NOTIFY_SYNC RealNotifySync;
|
|
BOOLEAN PoolQuotaCharged;
|
|
PSECURITY_SUBJECT_CONTEXT _SEH2_VOLATILE SubjectContext = NULL;
|
|
|
|
/* Get the NOTIFY_CHANGE struct and reset it */
|
|
NotifyChange = (PNOTIFY_CHANGE)Irp->IoStatus.Information;
|
|
Irp->IoStatus.Information = 0;
|
|
/* Reset the cancel routine */
|
|
IoSetCancelRoutine(Irp, NULL);
|
|
/* And release lock */
|
|
IoReleaseCancelSpinLock(Irp->CancelIrql);
|
|
/* Get REAL_NOTIFY_SYNC struct */
|
|
RealNotifySync = NotifyChange->NotifySync;
|
|
|
|
FsRtlNotifyAcquireFastMutex(RealNotifySync);
|
|
|
|
_SEH2_TRY
|
|
{
|
|
/* Remove the IRP from the notifications list and mark it pending */
|
|
RemoveEntryList(&(Irp->Tail.Overlay.ListEntry));
|
|
IoMarkIrpPending(Irp);
|
|
|
|
/* Now, the tricky part - let's find a buffer big enough to hold the return data */
|
|
if (NotifyChange->Buffer && NotifyChange->AllocatedBuffer == NULL &&
|
|
((Irp->MdlAddress && MmGetSystemAddressForMdl(Irp->MdlAddress) == NotifyChange->Buffer) ||
|
|
NotifyChange->Buffer == Irp->AssociatedIrp.SystemBuffer))
|
|
{
|
|
/* Assume we didn't find any */
|
|
Buffer = NULL;
|
|
BufferLength = 0;
|
|
|
|
/* If we don't have IRPs, check if current buffer is big enough */
|
|
if (IsListEmpty(&NotifyChange->NotifyIrps))
|
|
{
|
|
if (NotifyChange->BufferLength >= NotifyChange->DataLength)
|
|
{
|
|
BufferLength = NotifyChange->BufferLength;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
/* Otherwise, try to look at next IRP available */
|
|
NotifyIrp = CONTAINING_RECORD(NotifyChange->NotifyIrps.Flink, IRP, Tail.Overlay.ListEntry);
|
|
Stack = IoGetCurrentIrpStackLocation(NotifyIrp);
|
|
|
|
/* If its buffer is big enough, get it */
|
|
if (Stack->Parameters.NotifyDirectory.Length >= NotifyChange->BufferLength)
|
|
{
|
|
/* Is it MDL? */
|
|
if (NotifyIrp->AssociatedIrp.SystemBuffer == NULL)
|
|
{
|
|
if (NotifyIrp->MdlAddress != NULL)
|
|
{
|
|
Buffer = MmGetSystemAddressForMdl(NotifyIrp->MdlAddress);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
Buffer = NotifyIrp->AssociatedIrp.MasterIrp;
|
|
}
|
|
|
|
/* Backup our accepted buffer length */
|
|
BufferLength = Stack->Parameters.NotifyDirectory.Length;
|
|
if (BufferLength > NotifyChange->BufferLength)
|
|
{
|
|
BufferLength = NotifyChange->BufferLength;
|
|
}
|
|
}
|
|
}
|
|
|
|
/* At that point, we *may* have a buffer */
|
|
|
|
/* If it has null length, then note that we won't use it */
|
|
if (BufferLength == 0)
|
|
{
|
|
NotifyChange->Flags |= NOTIFY_IMMEDIATELY;
|
|
}
|
|
else
|
|
{
|
|
/* If we have a buffer length, but no buffer then allocate one */
|
|
if (Buffer == NULL)
|
|
{
|
|
/* Assume we haven't charged quotas */
|
|
PoolQuotaCharged = FALSE;
|
|
|
|
_SEH2_TRY
|
|
{
|
|
/* Charge quotas */
|
|
PsChargePoolQuota(NotifyChange->OwningProcess, PagedPool, BufferLength);
|
|
PoolQuotaCharged = TRUE;
|
|
|
|
/* Allocate buffer */
|
|
Buffer = ExAllocatePoolWithTag(PagedPool | POOL_RAISE_IF_ALLOCATION_FAILURE, BufferLength, TAG_FS_NOTIFICATIONS);
|
|
NotifyChange->AllocatedBuffer = Buffer;
|
|
|
|
/* Copy data in that buffer */
|
|
RtlCopyMemory(Buffer, NotifyChange->Buffer, NotifyChange->DataLength);
|
|
NotifyChange->ThisBufferLength = BufferLength;
|
|
NotifyChange->Buffer = Buffer;
|
|
}
|
|
_SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
|
|
{
|
|
/* Something went wrong, have we charged quotas? */
|
|
if (PoolQuotaCharged)
|
|
{
|
|
/* We did, return quotas */
|
|
PsReturnProcessPagedPoolQuota(NotifyChange->OwningProcess, BufferLength);
|
|
}
|
|
|
|
/* And notify immediately */
|
|
NotifyChange->Flags |= NOTIFY_IMMEDIATELY;
|
|
}
|
|
_SEH2_END;
|
|
}
|
|
}
|
|
|
|
/* If we have to notify immediately, ensure that any buffer is 0-ed out */
|
|
if (NotifyChange->Flags & NOTIFY_IMMEDIATELY)
|
|
{
|
|
NotifyChange->Buffer = 0;
|
|
NotifyChange->AllocatedBuffer = 0;
|
|
NotifyChange->LastEntry = 0;
|
|
NotifyChange->DataLength = 0;
|
|
NotifyChange->ThisBufferLength = 0;
|
|
}
|
|
}
|
|
|
|
/* It's now time to complete - data are ready */
|
|
|
|
/* Set appropriate status and complete */
|
|
Irp->IoStatus.Status = STATUS_CANCELLED;
|
|
IoCompleteRequest(Irp, IO_DISK_INCREMENT);
|
|
|
|
/* If that notification isn't referenced any longer, drop it */
|
|
if (!InterlockedDecrement((PLONG)&(NotifyChange->ReferenceCount)))
|
|
{
|
|
/* If it had an allocated buffer, delete */
|
|
if (NotifyChange->AllocatedBuffer)
|
|
{
|
|
PsReturnProcessPagedPoolQuota(NotifyChange->OwningProcess, NotifyChange->ThisBufferLength);
|
|
ExFreePoolWithTag(NotifyChange->AllocatedBuffer, TAG_FS_NOTIFICATIONS);
|
|
}
|
|
|
|
/* In case of full name, remember subject context for later deletion */
|
|
if (NotifyChange->FullDirectoryName)
|
|
{
|
|
SubjectContext = NotifyChange->SubjectContext;
|
|
}
|
|
|
|
/* We mustn't have ANY change left anymore */
|
|
ASSERT(NotifyChange->NotifyList.Flink == NULL);
|
|
ExFreePoolWithTag(NotifyChange, 0);
|
|
}
|
|
}
|
|
_SEH2_FINALLY
|
|
{
|
|
FsRtlNotifyReleaseFastMutex(RealNotifySync);
|
|
|
|
/* If the subject security context was captured, release and free it */
|
|
if (SubjectContext)
|
|
{
|
|
SeReleaseSubjectContext(SubjectContext);
|
|
ExFreePool(SubjectContext);
|
|
}
|
|
}
|
|
_SEH2_END;
|
|
}
|
|
|
|
/*
|
|
* @implemented
|
|
*/
|
|
VOID
|
|
FsRtlCheckNotifyForDelete(IN PLIST_ENTRY NotifyList,
|
|
IN PVOID FsContext)
|
|
{
|
|
PLIST_ENTRY NextEntry;
|
|
PNOTIFY_CHANGE NotifyChange;
|
|
|
|
if (!IsListEmpty(NotifyList))
|
|
{
|
|
/* Browse the notifications list to find the matching entry */
|
|
for (NextEntry = NotifyList->Flink;
|
|
NextEntry != NotifyList;
|
|
NextEntry = NextEntry->Flink)
|
|
{
|
|
NotifyChange = CONTAINING_RECORD(NextEntry, NOTIFY_CHANGE, NotifyList);
|
|
/* If the current record matches with the given context, it's the good one */
|
|
if (NotifyChange->FsContext == FsContext && !IsListEmpty(&(NotifyChange->NotifyIrps)))
|
|
{
|
|
FsRtlNotifyCompleteIrpList(NotifyChange, STATUS_DELETE_PENDING);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
/*
|
|
*@implemented
|
|
*/
|
|
PNOTIFY_CHANGE
|
|
FsRtlIsNotifyOnList(IN PLIST_ENTRY NotifyList,
|
|
IN PVOID FsContext)
|
|
{
|
|
PLIST_ENTRY NextEntry;
|
|
PNOTIFY_CHANGE NotifyChange;
|
|
|
|
if (!IsListEmpty(NotifyList))
|
|
{
|
|
/* Browse the notifications list to find the matching entry */
|
|
for (NextEntry = NotifyList->Flink;
|
|
NextEntry != NotifyList;
|
|
NextEntry = NextEntry->Flink)
|
|
{
|
|
NotifyChange = CONTAINING_RECORD(NextEntry, NOTIFY_CHANGE, NotifyList);
|
|
/* If the current record matches with the given context, it's the good one */
|
|
if (NotifyChange->FsContext == FsContext)
|
|
{
|
|
return NotifyChange;
|
|
}
|
|
}
|
|
}
|
|
return NULL;
|
|
}
|
|
|
|
/*
|
|
* @implemented
|
|
*/
|
|
VOID
|
|
FsRtlNotifyCompleteIrp(IN PIRP Irp,
|
|
IN PNOTIFY_CHANGE NotifyChange,
|
|
IN ULONG DataLength,
|
|
IN NTSTATUS Status,
|
|
IN BOOLEAN SkipCompletion)
|
|
{
|
|
PVOID Buffer;
|
|
PIO_STACK_LOCATION Stack;
|
|
|
|
PAGED_CODE();
|
|
|
|
/* Check if we need to complete */
|
|
if (!FsRtlNotifySetCancelRoutine(Irp, NotifyChange) && SkipCompletion)
|
|
{
|
|
return;
|
|
}
|
|
|
|
/* No succes => no data to return just complete */
|
|
if (Status != STATUS_SUCCESS)
|
|
{
|
|
goto Completion;
|
|
}
|
|
|
|
/* Ensure there's something to return */
|
|
Stack = IoGetCurrentIrpStackLocation(Irp);
|
|
if (!DataLength || Stack->Parameters.NotifyDirectory.Length < DataLength)
|
|
{
|
|
Status = STATUS_NOTIFY_ENUM_DIR;
|
|
goto Completion;
|
|
}
|
|
|
|
/* Ensture there's a buffer where to find data */
|
|
if (!NotifyChange->AllocatedBuffer)
|
|
{
|
|
Irp->IoStatus.Information = DataLength;
|
|
NotifyChange->Buffer = NULL;
|
|
goto Completion;
|
|
}
|
|
|
|
/* Now, browse all the way to return data
|
|
* and find the one that will work. We will
|
|
* return data whatever happens
|
|
*/
|
|
if (Irp->AssociatedIrp.SystemBuffer)
|
|
{
|
|
Buffer = Irp->AssociatedIrp.SystemBuffer;
|
|
goto CopyAndComplete;
|
|
}
|
|
|
|
if (Irp->MdlAddress)
|
|
{
|
|
Buffer = MmGetSystemAddressForMdl(Irp->MdlAddress);
|
|
goto CopyAndComplete;
|
|
}
|
|
|
|
if (!(Stack->Control & SL_PENDING_RETURNED))
|
|
{
|
|
Buffer = Irp->UserBuffer;
|
|
goto CopyAndComplete;
|
|
}
|
|
|
|
Irp->Flags |= (IRP_BUFFERED_IO | IRP_DEALLOCATE_BUFFER | IRP_SYNCHRONOUS_PAGING_IO);
|
|
Irp->AssociatedIrp.SystemBuffer = NotifyChange->AllocatedBuffer;
|
|
/* Nothing to copy */
|
|
goto ReleaseAndComplete;
|
|
|
|
CopyAndComplete:
|
|
_SEH2_TRY
|
|
{
|
|
RtlCopyMemory(Buffer, NotifyChange->AllocatedBuffer, DataLength);
|
|
}
|
|
_SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
|
|
{
|
|
/* Do nothing */
|
|
}
|
|
_SEH2_END;
|
|
|
|
ReleaseAndComplete:
|
|
PsReturnProcessPagedPoolQuota(NotifyChange->OwningProcess, NotifyChange->ThisBufferLength);
|
|
|
|
/* Release buffer UNLESS it's used */
|
|
if (NotifyChange->AllocatedBuffer != Irp->AssociatedIrp.SystemBuffer &&
|
|
NotifyChange->AllocatedBuffer)
|
|
{
|
|
ExFreePoolWithTag(NotifyChange->AllocatedBuffer, 0);
|
|
}
|
|
|
|
/* Prepare for return */
|
|
NotifyChange->AllocatedBuffer = 0;
|
|
NotifyChange->ThisBufferLength = 0;
|
|
Irp->IoStatus.Information = DataLength;
|
|
NotifyChange->Buffer = NULL;
|
|
|
|
/* Finally complete */
|
|
Completion:
|
|
IoMarkIrpPending(Irp);
|
|
Irp->IoStatus.Status = Status;
|
|
IoCompleteRequest(Irp, IO_DISK_INCREMENT);
|
|
}
|
|
|
|
/*
|
|
* @implemented
|
|
*/
|
|
VOID
|
|
FsRtlNotifyCompleteIrpList(IN PNOTIFY_CHANGE NotifyChange,
|
|
IN NTSTATUS Status)
|
|
{
|
|
PIRP Irp;
|
|
ULONG DataLength;
|
|
PLIST_ENTRY NextEntry;
|
|
|
|
DataLength = NotifyChange->DataLength;
|
|
|
|
NotifyChange->Flags &= (NOTIFY_IMMEDIATELY | WATCH_TREE);
|
|
NotifyChange->DataLength = 0;
|
|
NotifyChange->LastEntry = 0;
|
|
|
|
while (!IsListEmpty(&(NotifyChange->NotifyIrps)))
|
|
{
|
|
/* We take the first entry */
|
|
NextEntry = RemoveHeadList(&(NotifyChange->NotifyIrps));
|
|
Irp = CONTAINING_RECORD(NextEntry, IRP, Tail.Overlay.ListEntry);
|
|
/* We complete it */
|
|
FsRtlNotifyCompleteIrp(Irp, NotifyChange, DataLength, Status, TRUE);
|
|
/* If we're notifying success, just notify first one */
|
|
if (Status == STATUS_SUCCESS)
|
|
break;
|
|
}
|
|
}
|
|
|
|
/*
|
|
* @implemented
|
|
*/
|
|
BOOLEAN
|
|
FsRtlNotifySetCancelRoutine(IN PIRP Irp,
|
|
IN PNOTIFY_CHANGE NotifyChange OPTIONAL)
|
|
{
|
|
PDRIVER_CANCEL CancelRoutine;
|
|
|
|
/* Acquire cancel lock */
|
|
IoAcquireCancelSpinLock(&Irp->CancelIrql);
|
|
|
|
/* If NotifyChange was given */
|
|
if (NotifyChange)
|
|
{
|
|
/* First get cancel routine */
|
|
CancelRoutine = IoSetCancelRoutine(Irp, NULL);
|
|
Irp->IoStatus.Information = 0;
|
|
/* Release cancel lock */
|
|
IoReleaseCancelSpinLock(Irp->CancelIrql);
|
|
/* If there was a cancel routine */
|
|
if (CancelRoutine)
|
|
{
|
|
/* Decrease reference count */
|
|
InterlockedDecrement((PLONG)&NotifyChange->ReferenceCount);
|
|
/* Notify that we removed cancel routine */
|
|
return TRUE;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
/* If IRP is cancel, call FsRtl cancel routine */
|
|
if (Irp->Cancel)
|
|
{
|
|
FsRtlCancelNotify(NULL, Irp);
|
|
}
|
|
else
|
|
{
|
|
/* Otherwise, define FsRtl cancel routine as IRP cancel routine */
|
|
IoSetCancelRoutine(Irp, FsRtlCancelNotify);
|
|
/* Release lock */
|
|
IoReleaseCancelSpinLock(Irp->CancelIrql);
|
|
}
|
|
}
|
|
|
|
/* Return that we didn't removed cancel routine */
|
|
return FALSE;
|
|
}
|
|
|
|
/*
|
|
* @implemented
|
|
*/
|
|
BOOLEAN
|
|
FsRtlNotifyUpdateBuffer(OUT PFILE_NOTIFY_INFORMATION OutputBuffer,
|
|
IN ULONG Action,
|
|
IN PSTRING ParentName,
|
|
IN PSTRING TargetName,
|
|
IN PSTRING StreamName,
|
|
IN BOOLEAN IsUnicode,
|
|
IN ULONG DataLength)
|
|
{
|
|
/* Unless there's an issue with buffers, there's no reason to fail */
|
|
BOOLEAN Succeed = TRUE;
|
|
ULONG AlreadyWritten = 0, ResultSize;
|
|
|
|
PAGED_CODE();
|
|
|
|
/* Update user buffer with the change that occured
|
|
* First copy parent name if any
|
|
* Then copy target name, there's always one
|
|
* And finally, copy stream name if any
|
|
* If these names aren't unicode, then convert first
|
|
*/
|
|
_SEH2_TRY
|
|
{
|
|
OutputBuffer->NextEntryOffset = 0;
|
|
OutputBuffer->Action = Action;
|
|
OutputBuffer->FileNameLength = DataLength - FIELD_OFFSET(FILE_NOTIFY_INFORMATION, FileName);
|
|
if (IsUnicode)
|
|
{
|
|
if (ParentName->Length)
|
|
{
|
|
RtlCopyMemory(OutputBuffer->FileName, ParentName->Buffer, ParentName->Length);
|
|
OutputBuffer->FileName[ParentName->Length / sizeof(WCHAR)] = L'\\';
|
|
AlreadyWritten = ParentName->Length + sizeof(WCHAR);
|
|
}
|
|
RtlCopyMemory((PVOID)((ULONG_PTR)OutputBuffer->FileName + AlreadyWritten),
|
|
TargetName->Buffer, TargetName->Length);
|
|
if (StreamName)
|
|
{
|
|
AlreadyWritten += TargetName->Length;
|
|
OutputBuffer->FileName[AlreadyWritten / sizeof(WCHAR)] = L':';
|
|
RtlCopyMemory((PVOID)((ULONG_PTR)OutputBuffer->FileName + AlreadyWritten + sizeof(WCHAR)),
|
|
StreamName->Buffer, StreamName->Length);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
if (!ParentName->Length)
|
|
{
|
|
ASSERT(StreamName);
|
|
RtlCopyMemory(OutputBuffer->FileName, StreamName->Buffer, StreamName->Length);
|
|
}
|
|
else
|
|
{
|
|
RtlOemToUnicodeN(OutputBuffer->FileName, OutputBuffer->FileNameLength,
|
|
&ResultSize, ParentName->Buffer,
|
|
ParentName->Length);
|
|
OutputBuffer->FileName[ResultSize / sizeof(WCHAR)] = L'\\';
|
|
AlreadyWritten = ResultSize + sizeof(WCHAR);
|
|
|
|
RtlOemToUnicodeN((PVOID)((ULONG_PTR)OutputBuffer->FileName + AlreadyWritten),
|
|
OutputBuffer->FileNameLength, &ResultSize,
|
|
TargetName->Buffer, TargetName->Length);
|
|
|
|
if (StreamName)
|
|
{
|
|
AlreadyWritten += ResultSize;
|
|
OutputBuffer->FileName[AlreadyWritten / sizeof(WCHAR)] = L':';
|
|
RtlOemToUnicodeN((PVOID)((ULONG_PTR)OutputBuffer->FileName + AlreadyWritten + sizeof(WCHAR)),
|
|
OutputBuffer->FileNameLength, &ResultSize,
|
|
StreamName->Buffer, StreamName->Length);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
_SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
|
|
{
|
|
Succeed = FALSE;
|
|
}
|
|
_SEH2_END;
|
|
|
|
return Succeed;
|
|
}
|
|
|
|
/* PUBLIC FUNCTIONS **********************************************************/
|
|
|
|
/*++
|
|
* @name FsRtlNotifyChangeDirectory
|
|
* @implemented
|
|
*
|
|
* Lets FSD know if changes occures in the specified directory.
|
|
* Directory will be reenumerated.
|
|
*
|
|
* @param NotifySync
|
|
* Synchronization object pointer
|
|
*
|
|
* @param FsContext
|
|
* Used to identify the notify structure
|
|
*
|
|
* @param FullDirectoryName
|
|
* String (A or W) containing the full directory name
|
|
*
|
|
* @param NotifyList
|
|
* Notify list pointer (to head)
|
|
*
|
|
* @param WatchTree
|
|
* True to notify changes in subdirectories too
|
|
*
|
|
* @param CompletionFilter
|
|
* Used to define types of changes to notify
|
|
*
|
|
* @param NotifyIrp
|
|
* IRP pointer to complete notify operation. It can be null
|
|
*
|
|
* @return None
|
|
*
|
|
* @remarks This function only redirects to FsRtlNotifyFilterChangeDirectory.
|
|
*
|
|
*--*/
|
|
VOID
|
|
NTAPI
|
|
FsRtlNotifyChangeDirectory(IN PNOTIFY_SYNC NotifySync,
|
|
IN PVOID FsContext,
|
|
IN PSTRING FullDirectoryName,
|
|
IN PLIST_ENTRY NotifyList,
|
|
IN BOOLEAN WatchTree,
|
|
IN ULONG CompletionFilter,
|
|
IN PIRP NotifyIrp)
|
|
{
|
|
FsRtlNotifyFilterChangeDirectory(NotifySync,
|
|
NotifyList,
|
|
FsContext,
|
|
FullDirectoryName,
|
|
WatchTree,
|
|
TRUE,
|
|
CompletionFilter,
|
|
NotifyIrp,
|
|
NULL,
|
|
NULL,
|
|
NULL);
|
|
}
|
|
|
|
/*++
|
|
* @name FsRtlNotifyCleanup
|
|
* @implemented
|
|
*
|
|
* Called by FSD when all handles to FileObject (identified by FsContext) are closed
|
|
*
|
|
* @param NotifySync
|
|
* Synchronization object pointer
|
|
*
|
|
* @param NotifyList
|
|
* Notify list pointer (to head)
|
|
*
|
|
* @param FsContext
|
|
* Used to identify the notify structure
|
|
*
|
|
* @return None
|
|
*
|
|
* @remarks None
|
|
*
|
|
*--*/
|
|
VOID
|
|
NTAPI
|
|
FsRtlNotifyCleanup(IN PNOTIFY_SYNC NotifySync,
|
|
IN PLIST_ENTRY NotifyList,
|
|
IN PVOID FsContext)
|
|
{
|
|
PNOTIFY_CHANGE NotifyChange;
|
|
PREAL_NOTIFY_SYNC RealNotifySync;
|
|
PSECURITY_SUBJECT_CONTEXT _SEH2_VOLATILE SubjectContext = NULL;
|
|
|
|
/* Get real structure hidden behind the opaque pointer */
|
|
RealNotifySync = (PREAL_NOTIFY_SYNC)NotifySync;
|
|
|
|
/* Acquire the fast mutex */
|
|
FsRtlNotifyAcquireFastMutex(RealNotifySync);
|
|
|
|
_SEH2_TRY
|
|
{
|
|
/* Find if there's a matching notification with the FsContext */
|
|
NotifyChange = FsRtlIsNotifyOnList(NotifyList, FsContext);
|
|
if (NotifyChange)
|
|
{
|
|
/* Mark it as to know that cleanup is in process */
|
|
NotifyChange->Flags |= CLEANUP_IN_PROCESS;
|
|
|
|
/* If there are pending IRPs, complete them using the STATUS_NOTIFY_CLEANUP status */
|
|
if (!IsListEmpty(&NotifyChange->NotifyIrps))
|
|
{
|
|
FsRtlNotifyCompleteIrpList(NotifyChange, STATUS_NOTIFY_CLEANUP);
|
|
}
|
|
|
|
/* Decrease reference number and if 0 is reached, it's time to do complete cleanup */
|
|
if (!InterlockedDecrement((PLONG)&(NotifyChange->ReferenceCount)))
|
|
{
|
|
/* Remove it from the notifications list */
|
|
RemoveEntryList(&NotifyChange->NotifyList);
|
|
|
|
/* In case there was an allocated buffer, free it */
|
|
if (NotifyChange->AllocatedBuffer)
|
|
{
|
|
PsReturnProcessPagedPoolQuota(NotifyChange->OwningProcess, NotifyChange->ThisBufferLength);
|
|
ExFreePool(NotifyChange->AllocatedBuffer);
|
|
}
|
|
|
|
/* In case there the string was set, get the captured subject security context */
|
|
if (NotifyChange->FullDirectoryName)
|
|
{
|
|
SubjectContext = NotifyChange->SubjectContext;
|
|
}
|
|
|
|
/* Finally, free the notification, as it's not needed anymore */
|
|
ExFreePoolWithTag(NotifyChange, 'NrSF');
|
|
}
|
|
}
|
|
}
|
|
_SEH2_FINALLY
|
|
{
|
|
/* Release fast mutex */
|
|
FsRtlNotifyReleaseFastMutex(RealNotifySync);
|
|
|
|
/* If the subject security context was captured, release and free it */
|
|
if (SubjectContext)
|
|
{
|
|
SeReleaseSubjectContext(SubjectContext);
|
|
ExFreePool(SubjectContext);
|
|
}
|
|
}
|
|
_SEH2_END;
|
|
}
|
|
|
|
/*++
|
|
* @name FsRtlNotifyFilterChangeDirectory
|
|
* @implemented
|
|
*
|
|
* FILLME
|
|
*
|
|
* @param NotifySync
|
|
* FILLME
|
|
*
|
|
* @param NotifyList
|
|
* FILLME
|
|
*
|
|
* @param FsContext
|
|
* FILLME
|
|
*
|
|
* @param FullDirectoryName
|
|
* FILLME
|
|
*
|
|
* @param WatchTree
|
|
* FILLME
|
|
*
|
|
* @param IgnoreBuffer
|
|
* FILLME
|
|
*
|
|
* @param CompletionFilter
|
|
* FILLME
|
|
*
|
|
* @param NotifyIrp
|
|
* FILLME
|
|
*
|
|
* @param TraverseCallback
|
|
* FILLME
|
|
*
|
|
* @param SubjectContext
|
|
* FILLME
|
|
*
|
|
* @param FilterCallback
|
|
* FILLME
|
|
*
|
|
* @return None
|
|
*
|
|
* @remarks None
|
|
*
|
|
*--*/
|
|
VOID
|
|
NTAPI
|
|
FsRtlNotifyFilterChangeDirectory(IN PNOTIFY_SYNC NotifySync,
|
|
IN PLIST_ENTRY NotifyList,
|
|
IN PVOID FsContext,
|
|
IN PSTRING FullDirectoryName,
|
|
IN BOOLEAN WatchTree,
|
|
IN BOOLEAN IgnoreBuffer,
|
|
IN ULONG CompletionFilter,
|
|
IN PIRP NotifyIrp,
|
|
IN PCHECK_FOR_TRAVERSE_ACCESS TraverseCallback OPTIONAL,
|
|
IN PSECURITY_SUBJECT_CONTEXT SubjectContext OPTIONAL,
|
|
IN PFILTER_REPORT_CHANGE FilterCallback OPTIONAL)
|
|
{
|
|
ULONG SavedLength;
|
|
PIO_STACK_LOCATION Stack;
|
|
PNOTIFY_CHANGE NotifyChange = NULL;
|
|
PREAL_NOTIFY_SYNC RealNotifySync;
|
|
|
|
PAGED_CODE();
|
|
|
|
DPRINT("FsRtlNotifyFilterChangeDirectory(): %p, %p, %p, %wZ, %u, %u, %u, %p, %p, %p, %p\n",
|
|
NotifySync, NotifyList, FsContext, FullDirectoryName, WatchTree, IgnoreBuffer, CompletionFilter, NotifyIrp,
|
|
TraverseCallback, SubjectContext, FilterCallback);
|
|
|
|
/* Get real structure hidden behind the opaque pointer */
|
|
RealNotifySync = (PREAL_NOTIFY_SYNC)NotifySync;
|
|
|
|
/* Acquire the fast mutex */
|
|
FsRtlNotifyAcquireFastMutex(RealNotifySync);
|
|
|
|
_SEH2_TRY
|
|
{
|
|
/* If we have no IRP, FSD is performing a cleanup */
|
|
if (!NotifyIrp)
|
|
{
|
|
/* So, we delete */
|
|
FsRtlCheckNotifyForDelete(NotifyList, FsContext);
|
|
_SEH2_LEAVE;
|
|
}
|
|
|
|
NotifyIrp->IoStatus.Status = STATUS_SUCCESS;
|
|
NotifyIrp->IoStatus.Information = (ULONG_PTR)NULL;
|
|
|
|
Stack = IoGetCurrentIrpStackLocation(NotifyIrp);
|
|
/* If FileObject's been cleaned up, just return */
|
|
if (Stack->FileObject->Flags & FO_CLEANUP_COMPLETE)
|
|
{
|
|
IoMarkIrpPending(NotifyIrp);
|
|
NotifyIrp->IoStatus.Status = STATUS_NOTIFY_CLEANUP;
|
|
IoCompleteRequest(NotifyIrp, IO_DISK_INCREMENT);
|
|
_SEH2_LEAVE;
|
|
}
|
|
|
|
/* Try to find a matching notification has been already registered */
|
|
NotifyChange = FsRtlIsNotifyOnList(NotifyList, FsContext);
|
|
if (NotifyChange)
|
|
{
|
|
/* If it's been found, and is cleaned up, immediatly complete */
|
|
if (NotifyChange->Flags & CLEANUP_IN_PROCESS)
|
|
{
|
|
IoMarkIrpPending(NotifyIrp);
|
|
NotifyIrp->IoStatus.Status = STATUS_NOTIFY_CLEANUP;
|
|
IoCompleteRequest(NotifyIrp, IO_DISK_INCREMENT);
|
|
}
|
|
/* Or if it's about to be deleted, complete */
|
|
else if (NotifyChange->Flags & DELETE_IN_PROCESS)
|
|
{
|
|
IoMarkIrpPending(NotifyIrp);
|
|
NotifyIrp->IoStatus.Status = STATUS_DELETE_PENDING;
|
|
IoCompleteRequest(NotifyIrp, IO_DISK_INCREMENT);
|
|
}
|
|
/* Complete now if asked to (and not asked to notify later on) */
|
|
if ((NotifyChange->Flags & NOTIFY_IMMEDIATELY) && !(NotifyChange->Flags & NOTIFY_LATER))
|
|
{
|
|
NotifyChange->Flags &= ~NOTIFY_IMMEDIATELY;
|
|
IoMarkIrpPending(NotifyIrp);
|
|
NotifyIrp->IoStatus.Status = STATUS_NOTIFY_ENUM_DIR;
|
|
IoCompleteRequest(NotifyIrp, IO_DISK_INCREMENT);
|
|
}
|
|
/* If no data yet, or asked to notify later on, handle */
|
|
else if (NotifyChange->DataLength == 0 || (NotifyChange->Flags & NOTIFY_LATER))
|
|
{
|
|
goto HandleIRP;
|
|
}
|
|
/* Else, just complete with we have */
|
|
else
|
|
{
|
|
SavedLength = NotifyChange->DataLength;
|
|
NotifyChange->DataLength = 0;
|
|
FsRtlNotifyCompleteIrp(NotifyIrp, NotifyChange, SavedLength, STATUS_SUCCESS, FALSE);
|
|
}
|
|
|
|
_SEH2_LEAVE;
|
|
}
|
|
|
|
/* Allocate new notification */
|
|
NotifyChange = ExAllocatePoolWithTag(PagedPool | POOL_RAISE_IF_ALLOCATION_FAILURE,
|
|
sizeof(NOTIFY_CHANGE), 'NrSF');
|
|
RtlZeroMemory(NotifyChange, sizeof(NOTIFY_CHANGE));
|
|
|
|
/* Set basic information */
|
|
NotifyChange->NotifySync = NotifySync;
|
|
NotifyChange->FsContext = FsContext;
|
|
NotifyChange->StreamID = Stack->FileObject->FsContext;
|
|
NotifyChange->TraverseCallback = TraverseCallback;
|
|
NotifyChange->SubjectContext = SubjectContext;
|
|
NotifyChange->FullDirectoryName = FullDirectoryName;
|
|
NotifyChange->FilterCallback = FilterCallback;
|
|
InitializeListHead(&(NotifyChange->NotifyIrps));
|
|
|
|
/* Keep trace of WatchTree */
|
|
if (WatchTree)
|
|
{
|
|
NotifyChange->Flags |= WATCH_TREE;
|
|
}
|
|
|
|
/* If string is empty, faulty to ANSI */
|
|
if (FullDirectoryName->Length == 0)
|
|
{
|
|
NotifyChange->CharacterSize = sizeof(CHAR);
|
|
}
|
|
else
|
|
{
|
|
/* If it can't contain WCHAR, it's ANSI */
|
|
if (FullDirectoryName->Length < sizeof(WCHAR) || ((CHAR*)FullDirectoryName->Buffer)[1] != 0)
|
|
{
|
|
NotifyChange->CharacterSize = sizeof(CHAR);
|
|
}
|
|
else
|
|
{
|
|
NotifyChange->CharacterSize = sizeof(WCHAR);
|
|
}
|
|
|
|
/* Now, check is user is willing to watch root */
|
|
if (FullDirectoryName->Length == NotifyChange->CharacterSize)
|
|
{
|
|
NotifyChange->Flags |= WATCH_ROOT;
|
|
}
|
|
}
|
|
|
|
NotifyChange->CompletionFilter = CompletionFilter;
|
|
|
|
/* In case we have not to ignore buffer , keep its length */
|
|
if (!IgnoreBuffer)
|
|
{
|
|
NotifyChange->BufferLength = Stack->Parameters.NotifyDirectory.Length;
|
|
}
|
|
|
|
NotifyChange->OwningProcess = NotifyIrp->Tail.Overlay.Thread->ThreadsProcess;
|
|
|
|
/* Insert the notification into the notification list */
|
|
InsertTailList(NotifyList, &(NotifyChange->NotifyList));
|
|
|
|
NotifyChange->ReferenceCount = 1;
|
|
|
|
HandleIRP:
|
|
/* Associate the notification to the IRP */
|
|
NotifyIrp->IoStatus.Information = (ULONG_PTR)NotifyChange;
|
|
/* The IRP is pending */
|
|
IoMarkIrpPending(NotifyIrp);
|
|
/* Insert the IRP in the IRP list */
|
|
InsertTailList(&(NotifyChange->NotifyIrps), &(NotifyIrp->Tail.Overlay.ListEntry));
|
|
/* Increment reference count */
|
|
InterlockedIncrement((PLONG)&(NotifyChange->ReferenceCount));
|
|
/* Set cancel routine to FsRtl one */
|
|
FsRtlNotifySetCancelRoutine(NotifyIrp, NULL);
|
|
}
|
|
_SEH2_FINALLY
|
|
{
|
|
/* Release fast mutex */
|
|
FsRtlNotifyReleaseFastMutex(RealNotifySync);
|
|
|
|
/* If the subject security context was captured and there's no notify */
|
|
if (SubjectContext && (!NotifyChange || NotifyChange->FullDirectoryName))
|
|
{
|
|
SeReleaseSubjectContext(SubjectContext);
|
|
ExFreePool(SubjectContext);
|
|
}
|
|
}
|
|
_SEH2_END;
|
|
}
|
|
|
|
/*++
|
|
* @name FsRtlNotifyFilterReportChange
|
|
* @implemented
|
|
*
|
|
* FILLME
|
|
*
|
|
* @param NotifySync
|
|
* FILLME
|
|
*
|
|
* @param NotifyList
|
|
* FILLME
|
|
*
|
|
* @param FullTargetName
|
|
* FILLME
|
|
*
|
|
* @param TargetNameOffset
|
|
* FILLME
|
|
*
|
|
* @param StreamName
|
|
* FILLME
|
|
*
|
|
* @param NormalizedParentName
|
|
* FILLME
|
|
*
|
|
* @param FilterMatch
|
|
* FILLME
|
|
*
|
|
* @param Action
|
|
* FILLME
|
|
*
|
|
* @param TargetContext
|
|
* FILLME
|
|
*
|
|
* @param FilterContext
|
|
* FILLME
|
|
*
|
|
* @return None
|
|
*
|
|
* @remarks None
|
|
*
|
|
*--*/
|
|
VOID
|
|
NTAPI
|
|
FsRtlNotifyFilterReportChange(IN PNOTIFY_SYNC NotifySync,
|
|
IN PLIST_ENTRY NotifyList,
|
|
IN PSTRING FullTargetName,
|
|
IN USHORT TargetNameOffset,
|
|
IN PSTRING StreamName OPTIONAL,
|
|
IN PSTRING NormalizedParentName OPTIONAL,
|
|
IN ULONG FilterMatch,
|
|
IN ULONG Action,
|
|
IN PVOID TargetContext,
|
|
IN PVOID FilterContext)
|
|
{
|
|
PIRP Irp;
|
|
PVOID OutputBuffer;
|
|
USHORT FullPosition;
|
|
PLIST_ENTRY NextEntry;
|
|
PIO_STACK_LOCATION Stack;
|
|
PNOTIFY_CHANGE NotifyChange;
|
|
PREAL_NOTIFY_SYNC RealNotifySync;
|
|
PFILE_NOTIFY_INFORMATION FileNotifyInfo;
|
|
BOOLEAN IsStream, IsParent, PoolQuotaCharged;
|
|
STRING TargetDirectory, TargetName, ParentName, IntNormalizedParentName;
|
|
ULONG NumberOfBytes, TargetNumberOfParts, FullNumberOfParts, LastPartOffset, ParentNameOffset, ParentNameLength;
|
|
ULONG DataLength, AlignedDataLength;
|
|
|
|
TargetDirectory.Length = 0;
|
|
TargetDirectory.MaximumLength = 0;
|
|
TargetDirectory.Buffer = NULL;
|
|
TargetName.Length = 0;
|
|
TargetName.MaximumLength = 0;
|
|
TargetName.Buffer = NULL;
|
|
ParentName.Length = 0;
|
|
ParentName.MaximumLength = 0;
|
|
ParentName.Buffer = NULL;
|
|
IsStream = FALSE;
|
|
|
|
PAGED_CODE();
|
|
|
|
DPRINT("FsRtlNotifyFilterReportChange(%p, %p, %p, %u, %p, %p, %p, %x, %x, %p, %p)\n",
|
|
NotifySync, NotifyList, FullTargetName, TargetNameOffset, StreamName, NormalizedParentName,
|
|
FilterMatch, Action, TargetContext, FilterContext);
|
|
|
|
/* We need offset in name */
|
|
if (!TargetNameOffset && FullTargetName)
|
|
{
|
|
return;
|
|
}
|
|
|
|
/* Get real structure hidden behind the opaque pointer */
|
|
RealNotifySync = (PREAL_NOTIFY_SYNC)NotifySync;
|
|
/* Acquire lock - will be released in finally block */
|
|
FsRtlNotifyAcquireFastMutex(RealNotifySync);
|
|
_SEH2_TRY
|
|
{
|
|
/* Browse all the registered notifications we have */
|
|
for (NextEntry = NotifyList->Flink; NextEntry != NotifyList;
|
|
NextEntry = NextEntry->Flink)
|
|
{
|
|
/* Try to find an entry matching our change */
|
|
NotifyChange = CONTAINING_RECORD(NextEntry, NOTIFY_CHANGE, NotifyList);
|
|
if (FullTargetName != NULL)
|
|
{
|
|
ASSERT(NotifyChange->FullDirectoryName != NULL);
|
|
if (!NotifyChange->FullDirectoryName->Length)
|
|
{
|
|
continue;
|
|
}
|
|
|
|
if (!(FilterMatch & NotifyChange->CompletionFilter))
|
|
{
|
|
continue;
|
|
}
|
|
|
|
/* If no normalized name provided, construct it from full target name */
|
|
if (NormalizedParentName == NULL)
|
|
{
|
|
IntNormalizedParentName.Buffer = FullTargetName->Buffer;
|
|
if (TargetNameOffset != NotifyChange->CharacterSize)
|
|
{
|
|
IntNormalizedParentName.MaximumLength =
|
|
IntNormalizedParentName.Length = TargetNameOffset - NotifyChange->CharacterSize;
|
|
}
|
|
else
|
|
{
|
|
IntNormalizedParentName.MaximumLength =
|
|
IntNormalizedParentName.Length = TargetNameOffset;
|
|
}
|
|
NormalizedParentName = &IntNormalizedParentName;
|
|
}
|
|
|
|
/* heh? Watched directory bigger than changed file? */
|
|
if (NormalizedParentName->Length < NotifyChange->FullDirectoryName->Length)
|
|
{
|
|
continue;
|
|
}
|
|
|
|
/* Same len => parent */
|
|
if (NormalizedParentName->Length == NotifyChange->FullDirectoryName->Length)
|
|
{
|
|
IsParent = TRUE;
|
|
}
|
|
/* If not, then, we have to be watching the tree, otherwise we don't have to report such changes */
|
|
else if (!(NotifyChange->Flags & WATCH_TREE))
|
|
{
|
|
continue;
|
|
}
|
|
/* And finally, we've to check we're properly \-terminated */
|
|
else
|
|
{
|
|
if (!(NotifyChange->Flags & WATCH_ROOT))
|
|
{
|
|
if (NotifyChange->CharacterSize == sizeof(CHAR))
|
|
{
|
|
if (((PSTR)NormalizedParentName->Buffer)[NotifyChange->FullDirectoryName->Length] != '\\')
|
|
{
|
|
continue;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
if (((PWSTR)NormalizedParentName->Buffer)[NotifyChange->FullDirectoryName->Length / sizeof (WCHAR)] != L'\\')
|
|
{
|
|
continue;
|
|
}
|
|
}
|
|
}
|
|
|
|
IsParent = FALSE;
|
|
}
|
|
|
|
/* If len matches, then check that both name are equal */
|
|
if (!RtlEqualMemory(NormalizedParentName->Buffer, NotifyChange->FullDirectoryName->Buffer,
|
|
NotifyChange->FullDirectoryName->Length))
|
|
{
|
|
continue;
|
|
}
|
|
|
|
/* Call traverse callback (only if we have to traverse ;-)) */
|
|
if (!IsParent
|
|
&& NotifyChange->TraverseCallback != NULL
|
|
&& !NotifyChange->TraverseCallback(NotifyChange->FsContext,
|
|
TargetContext,
|
|
NotifyChange->SubjectContext))
|
|
{
|
|
continue;
|
|
}
|
|
|
|
/* And then, filter callback if provided */
|
|
if (NotifyChange->FilterCallback != NULL
|
|
&& FilterContext != NULL
|
|
&& !NotifyChange->FilterCallback(NotifyChange->FsContext, FilterContext))
|
|
{
|
|
continue;
|
|
}
|
|
}
|
|
/* We have a stream! */
|
|
else
|
|
{
|
|
ASSERT(NotifyChange->FullDirectoryName == NULL);
|
|
if (TargetContext != NotifyChange->SubjectContext)
|
|
{
|
|
continue;
|
|
}
|
|
|
|
ParentName.Buffer = NULL;
|
|
ParentName.Length = 0;
|
|
IsStream = TRUE;
|
|
IsParent = FALSE;
|
|
}
|
|
|
|
/* If we don't have to notify immediately, prepare for output */
|
|
if (!(NotifyChange->Flags & NOTIFY_IMMEDIATELY))
|
|
{
|
|
/* If we have something to output... */
|
|
if (NotifyChange->BufferLength)
|
|
{
|
|
/* Get size of the output */
|
|
NumberOfBytes = 0;
|
|
Irp = NULL;
|
|
if (!NotifyChange->ThisBufferLength)
|
|
{
|
|
if (IsListEmpty(&NotifyChange->NotifyIrps))
|
|
{
|
|
NumberOfBytes = NotifyChange->BufferLength;
|
|
}
|
|
else
|
|
{
|
|
Irp = CONTAINING_RECORD(NotifyChange->NotifyIrps.Flink, IRP, Tail.Overlay.ListEntry);
|
|
Stack = IoGetCurrentIrpStackLocation(Irp);
|
|
NumberOfBytes = Stack->Parameters.NotifyDirectory.Length;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
NumberOfBytes = NotifyChange->ThisBufferLength;
|
|
}
|
|
|
|
/* If we're matching parent, we don't care about parent (redundant) */
|
|
if (IsParent)
|
|
{
|
|
ParentName.Length = 0;
|
|
}
|
|
else
|
|
{
|
|
/* If we don't deal with streams, some more work is required */
|
|
if (!IsStream)
|
|
{
|
|
if (NotifyChange->Flags & WATCH_ROOT ||
|
|
(NormalizedParentName->Buffer != FullTargetName->Buffer))
|
|
{
|
|
/* Construct TargetDirectory if we don't have it yet */
|
|
if (TargetDirectory.Buffer == NULL)
|
|
{
|
|
TargetDirectory.Buffer = FullTargetName->Buffer;
|
|
TargetDirectory.Length = TargetNameOffset;
|
|
if (TargetNameOffset != NotifyChange->CharacterSize)
|
|
{
|
|
TargetDirectory.Length = TargetNameOffset - NotifyChange->CharacterSize;
|
|
}
|
|
TargetDirectory.MaximumLength = TargetDirectory.Length;
|
|
}
|
|
/* Now, we start looking for matching parts (unless we watch root) */
|
|
TargetNumberOfParts = 0;
|
|
if (!(NotifyChange->Flags & WATCH_ROOT))
|
|
{
|
|
FullNumberOfParts = 1;
|
|
if (NotifyChange->CharacterSize == sizeof(CHAR))
|
|
{
|
|
FsRtlNotifyGetLastPartOffset(NotifyChange->FullDirectoryName->Length,
|
|
TargetDirectory.Length, PSTR, '\\');
|
|
}
|
|
else
|
|
{
|
|
FsRtlNotifyGetLastPartOffset(NotifyChange->FullDirectoryName->Length / sizeof(WCHAR),
|
|
TargetDirectory.Length / sizeof(WCHAR), PWSTR, L'\\');
|
|
LastPartOffset *= NotifyChange->CharacterSize;
|
|
}
|
|
}
|
|
|
|
/* Then, we can construct proper parent name */
|
|
ParentNameOffset = NotifyChange->CharacterSize + LastPartOffset;
|
|
ParentName.Buffer = &TargetDirectory.Buffer[ParentNameOffset];
|
|
ParentNameLength = TargetDirectory.Length;
|
|
}
|
|
else
|
|
{
|
|
/* Construct parent name even for streams */
|
|
ParentName.Buffer = &NormalizedParentName->Buffer[NotifyChange->FullDirectoryName->Length] + NotifyChange->CharacterSize;
|
|
ParentNameLength = NormalizedParentName->Length - NotifyChange->FullDirectoryName->Length;
|
|
ParentNameOffset = NotifyChange->CharacterSize;
|
|
}
|
|
ParentNameLength -= ParentNameOffset;
|
|
ParentName.Length = ParentNameLength;
|
|
ParentName.MaximumLength = ParentNameLength;
|
|
}
|
|
}
|
|
|
|
/* Start to count amount of data to write, we've first the structure itself */
|
|
DataLength = FIELD_OFFSET(FILE_NOTIFY_INFORMATION, FileName);
|
|
|
|
/* If stream, we'll just append stream name */
|
|
if (IsStream)
|
|
{
|
|
ASSERT(StreamName != NULL);
|
|
DataLength += StreamName->Length;
|
|
}
|
|
else
|
|
{
|
|
/* If not parent, we've to append parent name */
|
|
if (!IsParent)
|
|
{
|
|
if (NotifyChange->CharacterSize == sizeof(CHAR))
|
|
{
|
|
DataLength += RtlOemStringToCountedUnicodeSize(&ParentName);
|
|
}
|
|
else
|
|
{
|
|
DataLength += ParentName.Length;
|
|
}
|
|
DataLength += sizeof(WCHAR);
|
|
}
|
|
|
|
/* Look for target name & construct it, if required */
|
|
if (TargetName.Buffer == NULL)
|
|
{
|
|
TargetName.Buffer = &FullTargetName->Buffer[TargetNameOffset];
|
|
TargetName.Length =
|
|
TargetName.MaximumLength = FullTargetName->Length - TargetNameOffset;
|
|
}
|
|
|
|
/* Then, we will append it as well */
|
|
if (NotifyChange->CharacterSize == sizeof(CHAR))
|
|
{
|
|
DataLength += RtlOemStringToCountedUnicodeSize(&TargetName);
|
|
}
|
|
else
|
|
{
|
|
DataLength += TargetName.Length;
|
|
}
|
|
|
|
/* If we also had a stream name, then we can append it as well */
|
|
if (StreamName != NULL)
|
|
{
|
|
if (NotifyChange->CharacterSize == sizeof(WCHAR))
|
|
{
|
|
DataLength += StreamName->Length + sizeof(WCHAR);
|
|
}
|
|
else
|
|
{
|
|
DataLength = DataLength + RtlOemStringToCountedUnicodeSize(&TargetName) + sizeof(CHAR);
|
|
}
|
|
}
|
|
}
|
|
|
|
/* Get the position where we can put our data (aligned!) */
|
|
AlignedDataLength = ROUND_UP(NotifyChange->DataLength, sizeof(ULONG));
|
|
/* If it's higher than buffer length, then, bail out without outputing */
|
|
if (DataLength > NumberOfBytes || AlignedDataLength + DataLength > NumberOfBytes)
|
|
{
|
|
NotifyChange->Flags |= NOTIFY_IMMEDIATELY;
|
|
}
|
|
else
|
|
{
|
|
OutputBuffer = NULL;
|
|
FileNotifyInfo = NULL;
|
|
/* If we already had a buffer, update last entry position */
|
|
if (NotifyChange->Buffer != NULL)
|
|
{
|
|
FileNotifyInfo = (PVOID)((ULONG_PTR)NotifyChange->Buffer + NotifyChange->LastEntry);
|
|
FileNotifyInfo->NextEntryOffset = AlignedDataLength - NotifyChange->LastEntry;
|
|
NotifyChange->LastEntry = AlignedDataLength;
|
|
/* And get our output buffer */
|
|
OutputBuffer = (PVOID)((ULONG_PTR)NotifyChange->Buffer + AlignedDataLength);
|
|
}
|
|
/* If we hadn't buffer, try to find one */
|
|
else if (Irp != NULL)
|
|
{
|
|
if (Irp->AssociatedIrp.SystemBuffer != NULL)
|
|
{
|
|
OutputBuffer = Irp->AssociatedIrp.SystemBuffer;
|
|
}
|
|
else if (Irp->MdlAddress != NULL)
|
|
{
|
|
OutputBuffer = MmGetSystemAddressForMdl(Irp->MdlAddress);
|
|
}
|
|
|
|
NotifyChange->Buffer = OutputBuffer;
|
|
NotifyChange->ThisBufferLength = NumberOfBytes;
|
|
}
|
|
|
|
/* If we couldn't find one, then allocate one */
|
|
if (NotifyChange->Buffer == NULL)
|
|
{
|
|
/* Assign the length buffer */
|
|
NotifyChange->ThisBufferLength = NumberOfBytes;
|
|
|
|
/* Assume we have not charged quotas */
|
|
PoolQuotaCharged = FALSE;
|
|
_SEH2_TRY
|
|
{
|
|
/* And charge quotas */
|
|
PsChargePoolQuota(NotifyChange->OwningProcess, PagedPool, NotifyChange->ThisBufferLength);
|
|
PoolQuotaCharged = TRUE;
|
|
OutputBuffer = ExAllocatePoolWithTag(PagedPool | POOL_RAISE_IF_ALLOCATION_FAILURE,
|
|
NumberOfBytes, TAG_FS_NOTIFICATIONS);
|
|
NotifyChange->Buffer = OutputBuffer;
|
|
NotifyChange->AllocatedBuffer = OutputBuffer;
|
|
}
|
|
/* If something went wrong during allocation, notify immediately instead of outputing */
|
|
_SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
|
|
{
|
|
if (PoolQuotaCharged)
|
|
{
|
|
PsReturnProcessPagedPoolQuota(NotifyChange->OwningProcess, NotifyChange->ThisBufferLength);
|
|
}
|
|
NotifyChange->Flags |= NOTIFY_IMMEDIATELY;
|
|
}
|
|
_SEH2_END;
|
|
}
|
|
|
|
/* Finally, if we have a buffer, fill it in! */
|
|
if (OutputBuffer != NULL)
|
|
{
|
|
if (FsRtlNotifyUpdateBuffer((FILE_NOTIFY_INFORMATION *)OutputBuffer,
|
|
Action, &ParentName, &TargetName,
|
|
StreamName, NotifyChange->CharacterSize == sizeof(WCHAR),
|
|
DataLength))
|
|
{
|
|
NotifyChange->DataLength = DataLength + AlignedDataLength;
|
|
}
|
|
/* If it failed, notify immediately */
|
|
else
|
|
{
|
|
NotifyChange->Flags |= NOTIFY_IMMEDIATELY;
|
|
}
|
|
}
|
|
}
|
|
|
|
/* If we have to notify right now (something went wrong?) */
|
|
if (NotifyChange->Flags & NOTIFY_IMMEDIATELY)
|
|
{
|
|
/* Ensure that all our buffers are NULL */
|
|
if (NotifyChange->Buffer != NULL)
|
|
{
|
|
if (NotifyChange->AllocatedBuffer != NULL)
|
|
{
|
|
PsReturnProcessPagedPoolQuota(NotifyChange->OwningProcess, NotifyChange->ThisBufferLength);
|
|
ExFreePoolWithTag(NotifyChange->AllocatedBuffer, TAG_FS_NOTIFICATIONS);
|
|
}
|
|
|
|
NotifyChange->Buffer = NULL;
|
|
NotifyChange->AllocatedBuffer = NULL;
|
|
NotifyChange->LastEntry = 0;
|
|
NotifyChange->DataLength = 0;
|
|
NotifyChange->ThisBufferLength = 0;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
/* If asking for old name in case of a rename, notify later on,
|
|
* so that we can wait for new name.
|
|
* http://msdn.microsoft.com/en-us/library/dn392331.aspx
|
|
*/
|
|
if (Action == FILE_ACTION_RENAMED_OLD_NAME)
|
|
{
|
|
NotifyChange->Flags |= NOTIFY_LATER;
|
|
}
|
|
else
|
|
{
|
|
NotifyChange->Flags &= ~NOTIFY_LATER;
|
|
if (!IsListEmpty(&NotifyChange->NotifyIrps))
|
|
{
|
|
FsRtlNotifyCompleteIrpList(NotifyChange, STATUS_SUCCESS);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
_SEH2_FINALLY
|
|
{
|
|
FsRtlNotifyReleaseFastMutex(RealNotifySync);
|
|
}
|
|
_SEH2_END;
|
|
}
|
|
|
|
/*++
|
|
* @name FsRtlNotifyFullChangeDirectory
|
|
* @implemented
|
|
*
|
|
* Lets FSD know if changes occures in the specified directory.
|
|
*
|
|
* @param NotifySync
|
|
* Synchronization object pointer
|
|
*
|
|
* @param NotifyList
|
|
* Notify list pointer (to head)
|
|
*
|
|
* @param FsContext
|
|
* Used to identify the notify structure
|
|
*
|
|
* @param FullDirectoryName
|
|
* String (A or W) containing the full directory name
|
|
*
|
|
* @param WatchTree
|
|
* True to notify changes in subdirectories too
|
|
*
|
|
* @param IgnoreBuffer
|
|
* True to reenumerate directory. It's ignored it NotifyIrp is null
|
|
*
|
|
* @param CompletionFilter
|
|
* Used to define types of changes to notify
|
|
*
|
|
* @param NotifyIrp
|
|
* IRP pointer to complete notify operation. It can be null
|
|
*
|
|
* @param TraverseCallback
|
|
* Pointer to a callback function. It's called each time a change is
|
|
* done in a subdirectory of the main directory. It's ignored it NotifyIrp
|
|
* is null
|
|
*
|
|
* @param SubjectContext
|
|
* Pointer to pass to SubjectContext member of TraverseCallback.
|
|
* It's freed after use. It's ignored it NotifyIrp is null
|
|
*
|
|
* @return None
|
|
*
|
|
* @remarks This function only redirects to FsRtlNotifyFilterChangeDirectory.
|
|
*
|
|
*--*/
|
|
VOID
|
|
NTAPI
|
|
FsRtlNotifyFullChangeDirectory(IN PNOTIFY_SYNC NotifySync,
|
|
IN PLIST_ENTRY NotifyList,
|
|
IN PVOID FsContext,
|
|
IN PSTRING FullDirectoryName,
|
|
IN BOOLEAN WatchTree,
|
|
IN BOOLEAN IgnoreBuffer,
|
|
IN ULONG CompletionFilter,
|
|
IN PIRP NotifyIrp,
|
|
IN PCHECK_FOR_TRAVERSE_ACCESS TraverseCallback OPTIONAL,
|
|
IN PSECURITY_SUBJECT_CONTEXT SubjectContext OPTIONAL)
|
|
{
|
|
FsRtlNotifyFilterChangeDirectory(NotifySync,
|
|
NotifyList,
|
|
FsContext,
|
|
FullDirectoryName,
|
|
WatchTree,
|
|
IgnoreBuffer,
|
|
CompletionFilter,
|
|
NotifyIrp,
|
|
TraverseCallback,
|
|
SubjectContext,
|
|
NULL);
|
|
}
|
|
|
|
/*++
|
|
* @name FsRtlNotifyFullReportChange
|
|
* @implemented
|
|
*
|
|
* Complets the pending notify IRPs.
|
|
*
|
|
* @param NotifySync
|
|
* Synchronization object pointer
|
|
*
|
|
* @param NotifyList
|
|
* Notify list pointer (to head)
|
|
*
|
|
* @param FullTargetName
|
|
* String (A or W) containing the full directory name that changed
|
|
*
|
|
* @param TargetNameOffset
|
|
* Offset, in FullTargetName, of the final component that is in the changed directory
|
|
*
|
|
* @param StreamName
|
|
* String (A or W) containing a stream name
|
|
*
|
|
* @param NormalizedParentName
|
|
* String (A or W) containing the full directory name that changed with long names
|
|
*
|
|
* @param FilterMatch
|
|
* Flags that will be compared to the completion filter
|
|
*
|
|
* @param Action
|
|
* Action code to store in user's buffer
|
|
*
|
|
* @param TargetContext
|
|
* Pointer to a callback function. It's called each time a change is
|
|
* done in a subdirectory of the main directory.
|
|
*
|
|
* @return None
|
|
*
|
|
* @remarks This function only redirects to FsRtlNotifyFilterReportChange.
|
|
*
|
|
*--*/
|
|
VOID
|
|
NTAPI
|
|
FsRtlNotifyFullReportChange(IN PNOTIFY_SYNC NotifySync,
|
|
IN PLIST_ENTRY NotifyList,
|
|
IN PSTRING FullTargetName,
|
|
IN USHORT TargetNameOffset,
|
|
IN PSTRING StreamName OPTIONAL,
|
|
IN PSTRING NormalizedParentName OPTIONAL,
|
|
IN ULONG FilterMatch,
|
|
IN ULONG Action,
|
|
IN PVOID TargetContext)
|
|
{
|
|
FsRtlNotifyFilterReportChange(NotifySync,
|
|
NotifyList,
|
|
FullTargetName,
|
|
TargetNameOffset,
|
|
StreamName,
|
|
NormalizedParentName,
|
|
FilterMatch,
|
|
Action,
|
|
TargetContext,
|
|
NULL);
|
|
}
|
|
|
|
/*++
|
|
* @name FsRtlNotifyInitializeSync
|
|
* @implemented
|
|
*
|
|
* Allocates the internal structure associated with notifications.
|
|
*
|
|
* @param NotifySync
|
|
* Opaque pointer. It will receive the address of the allocated internal structure.
|
|
*
|
|
* @return None
|
|
*
|
|
* @remarks This function raise an exception in case of a failure.
|
|
*
|
|
*--*/
|
|
VOID
|
|
NTAPI
|
|
FsRtlNotifyInitializeSync(IN PNOTIFY_SYNC *NotifySync)
|
|
{
|
|
PREAL_NOTIFY_SYNC RealNotifySync;
|
|
|
|
*NotifySync = NULL;
|
|
|
|
RealNotifySync = ExAllocatePoolWithTag(NonPagedPool | POOL_RAISE_IF_ALLOCATION_FAILURE,
|
|
sizeof(REAL_NOTIFY_SYNC), 'SNSF');
|
|
ExInitializeFastMutex(&(RealNotifySync->FastMutex));
|
|
RealNotifySync->OwningThread = 0;
|
|
RealNotifySync->OwnerCount = 0;
|
|
|
|
*NotifySync = RealNotifySync;
|
|
}
|
|
|
|
/*++
|
|
* @name FsRtlNotifyReportChange
|
|
* @implemented
|
|
*
|
|
* Complets the pending notify IRPs.
|
|
*
|
|
* @param NotifySync
|
|
* Synchronization object pointer
|
|
*
|
|
* @param NotifyList
|
|
* Notify list pointer (to head)
|
|
*
|
|
* @param FullTargetName
|
|
* String (A or W) containing the full directory name that changed
|
|
*
|
|
* @param FileNamePartLength
|
|
* Length of the final component that is in the changed directory
|
|
*
|
|
* @param FilterMatch
|
|
* Flags that will be compared to the completion filter
|
|
*
|
|
* @return None
|
|
*
|
|
* @remarks This function only redirects to FsRtlNotifyFilterReportChange.
|
|
*
|
|
*--*/
|
|
VOID
|
|
NTAPI
|
|
FsRtlNotifyReportChange(IN PNOTIFY_SYNC NotifySync,
|
|
IN PLIST_ENTRY NotifyList,
|
|
IN PSTRING FullTargetName,
|
|
IN PUSHORT FileNamePartLength,
|
|
IN ULONG FilterMatch)
|
|
{
|
|
FsRtlNotifyFilterReportChange(NotifySync,
|
|
NotifyList,
|
|
FullTargetName,
|
|
FullTargetName->Length - *FileNamePartLength,
|
|
NULL,
|
|
NULL,
|
|
FilterMatch,
|
|
0,
|
|
NULL,
|
|
NULL);
|
|
}
|
|
|
|
/*++
|
|
* @name FsRtlNotifyUninitializeSync
|
|
* @implemented
|
|
*
|
|
* Uninitialize a NOTIFY_SYNC object
|
|
*
|
|
* @param NotifySync
|
|
* Address of a pointer to a PNOTIFY_SYNC object previously
|
|
* initialized by FsRtlNotifyInitializeSync()
|
|
*
|
|
* @return None
|
|
*
|
|
* @remarks None
|
|
*
|
|
*--*/
|
|
VOID
|
|
NTAPI
|
|
FsRtlNotifyUninitializeSync(IN PNOTIFY_SYNC *NotifySync)
|
|
{
|
|
if (*NotifySync)
|
|
{
|
|
ExFreePoolWithTag(*NotifySync, 'SNSF');
|
|
*NotifySync = NULL;
|
|
}
|
|
}
|
|
|