Public/winfsp
1
0
Fork 0
mirror of https://github.com/winfsp/winfsp.git synced 2025-07-03 09:22:57 -05:00
Code Issues Packages Projects Releases Wiki Activity

sys: FSP_FILE_NODE: add HandleCount field and refactor cleanup/close/rename

Browse Source
This commit is contained in:
Bill Zissimopoulos
2016-03-15 15:09:47 -07:00
parent 66d20d3d35
commit bd413b9c9b
7 changed files with 151 additions and 148 deletions
Download Patch File Download Diff File Expand all files Collapse all files

100
src/sys/cleanup.c
View File

@ -14,7 +14,6 @@ static NTSTATUS FspFsvolCleanup(
PDEVICE_OBJECT DeviceObject, PIRP Irp, PIO_STACK_LOCATION IrpSp);
FSP_IOCMPL_DISPATCH FspFsvolCleanupComplete;
static FSP_IOP_REQUEST_FINI FspFsvolCleanupRequestFini;
static VOID FspFsvolCleanupUninitialize(PVOID Context);
FSP_DRIVER_DISPATCH FspCleanup;
#ifdef ALLOC_PRAGMA
@ -23,7 +22,6 @@ FSP_DRIVER_DISPATCH FspCleanup;
#pragma alloc_text(PAGE, FspFsvolCleanup)
#pragma alloc_text(PAGE, FspFsvolCleanupComplete)
#pragma alloc_text(PAGE, FspFsvolCleanupRequestFini)
#pragma alloc_text(PAGE, FspFsvolCleanupUninitialize)
#pragma alloc_text(PAGE, FspCleanup)
#endif
@ -33,13 +31,6 @@ enum
RequestIrp = 0,
};
typedef struct
{
PFILE_OBJECT FileObject;
LARGE_INTEGER TruncateSize, *PTruncateSize;
WORK_QUEUE_ITEM WorkQueueItem;
} FSP_FSVOL_CLEANUP_UNINITIALIZE_WORK_ITEM;
static NTSTATUS FspFsctlCleanup(
PDEVICE_OBJECT DeviceObject, PIRP Irp, PIO_STACK_LOCATION IrpSp)
{
@ -78,11 +69,9 @@ static NTSTATUS FspFsvolCleanup(
ASSERT(FileNode == FileDesc->FileNode);
FspFileNodeAcquireExclusive(FileNode, Main);
FspFileNodeCleanup(FileNode, FileObject, &DeletePending);
if (DeletePending)
FspFileNodeAcquireExclusive(FileNode, Full);
else
FspFileNodeAcquireShared(FileNode, Full);
/* create the user-mode file system request; MustSucceed because IRP_MJ_CLEANUP cannot fail */
FspIopCreateRequestMustSucceedEx(Irp, DeletePending ? &FileNode->FileName : 0, 0,
@ -92,6 +81,8 @@ static NTSTATUS FspFsvolCleanup(
Request->Req.Cleanup.UserContext2 = FileDesc->UserContext2;
Request->Req.Cleanup.Delete = DeletePending;
FspFileNodeAcquireExclusive(FileNode, Pgio);
FspFileNodeSetOwner(FileNode, Full, Request);
FspIopRequestContext(Request, RequestIrp) = Irp;
@ -129,93 +120,14 @@ static VOID FspFsvolCleanupRequestFini(FSP_FSCTL_TRANSACT_REQ *Request, PVOID Co
PIO_STACK_LOCATION IrpSp = IoGetCurrentIrpStackLocation(Irp);
PFILE_OBJECT FileObject = IrpSp->FileObject;
FSP_FILE_NODE *FileNode = FileObject->FsContext;
FSP_FSCTL_FILE_INFO FileInfo;
LARGE_INTEGER TruncateSize = { 0 }, *PTruncateSize = 0;
BOOLEAN DeletePending = 0 != Request->Req.Cleanup.Delete;
BOOLEAN DeletedFromContextTable;
BOOLEAN Success;
FspFileNodeClose(FileNode, FileObject, &DeletedFromContextTable);
if (DeletedFromContextTable)
{
if (DeletePending)
PTruncateSize = &TruncateSize;
else if (FileNode->TruncateOnClose)
{
/*
* Even when FileInfoTimeout != Infinity,
* this is the last size that the cache manager knows.
*/
FspFileNodeGetFileInfo(FileNode, &FileInfo);
FspFileNodeReleaseOwner(FileNode, Pgio, Request);
TruncateSize.QuadPart = FileInfo.FileSize;
PTruncateSize = &TruncateSize;
}
}
FspFileNodeCleanupComplete(FileNode, FileObject);
if (DeletePending)
{
/* FileNode is Exclusive Full; release Pgio */
FspFileNodeReleaseOwner(FileNode, Pgio, Request);
/* FileNode is now Exclusive Main; owner is Request */
}
else
{
/* FileNode is Shared Full; reacquire as Exclusive Main for CcUninitializeCacheMap */
FspFileNodeReleaseOwner(FileNode, Full, Request);
Success = DEBUGTEST(90, TRUE) && FspFileNodeTryAcquireExclusive(FileNode, Main);
if (!Success)
{
/* oh, maaan! we now have to do delayed uninitialize! */
FSP_FSVOL_CLEANUP_UNINITIALIZE_WORK_ITEM *WorkItem;
WorkItem = FspAllocatePoolMustSucceed(
NonPagedPool, sizeof *WorkItem, FSP_ALLOC_INTERNAL_TAG);
WorkItem->FileObject = FileObject;
WorkItem->TruncateSize = TruncateSize;
WorkItem->PTruncateSize = 0 != PTruncateSize ? &WorkItem->TruncateSize : 0;
ExInitializeWorkItem(&WorkItem->WorkQueueItem, FspFsvolCleanupUninitialize, WorkItem);
/* make sure that the file object (and corresponding device object) stay around! */
ObReferenceObject(FileObject);
ExQueueWorkItem(&WorkItem->WorkQueueItem, CriticalWorkQueue);
return;
}
/* FileNode is now Exclusive Main; owner is current thread */
}
CcUninitializeCacheMap(FileObject, PTruncateSize, 0);
/* this works correctly even if owner is current thread */
FspFileNodeReleaseOwner(FileNode, Main, Request);
}
static VOID FspFsvolCleanupUninitialize(PVOID Context)
{
PAGED_CODE();
FSP_FSVOL_CLEANUP_UNINITIALIZE_WORK_ITEM *WorkItem = Context;
PFILE_OBJECT FileObject = WorkItem->FileObject;
FSP_FILE_NODE *FileNode = FileObject->FsContext;
LARGE_INTEGER *PTruncateSize = WorkItem->PTruncateSize;
FspFileNodeAcquireExclusive(FileNode, Main);
CcUninitializeCacheMap(FileObject, PTruncateSize, 0);
FspFileNodeRelease(FileNode, Main);
ObDereferenceObject(FileObject);
FspFree(WorkItem);
}
NTSTATUS FspCleanup(
PDEVICE_OBJECT DeviceObject, PIRP Irp)
{