Public/winfsp
1
0
Fork 0
mirror of https://github.com/winfsp/winfsp.git synced 2025-04-23 08:53:01 -05:00
Code Issues Packages Projects Releases Wiki Activity

sys: FspFsvolCreateClose, FspFsvolCleanup, FspFsvolClose

Browse Source
This commit is contained in:
Bill Zissimopoulos 2015-12-08 00:06:24 -08:00
parent fa2ab4f8f7
commit 25154ca7c1
3 changed files with 39 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
src/sys/cleanup.c
View File

@ -60,6 +60,8 @@ static NTSTATUS FspFsvolCleanup(
FspFsvrtDeviceExtension(FsvrtDeviceObject); FspFsvrtDeviceExtension(FsvrtDeviceObject);
PFILE_OBJECT FileObject = IrpSp->FileObject; PFILE_OBJECT FileObject = IrpSp->FileObject;
FSP_FILE_CONTEXT *FsContext = FileObject->FsContext; FSP_FILE_CONTEXT *FsContext = FileObject->FsContext;
UINT64 UserContext = FsContext->UserContext;
UINT64 UserContext2 = (UINT_PTR)FileObject->FsContext2;
BOOLEAN FileNameRequired = 0 != FsvrtDeviceExtension->VolumeParams.FileNameRequired; BOOLEAN FileNameRequired = 0 != FsvrtDeviceExtension->VolumeParams.FileNameRequired;
BOOLEAN DeletePending; BOOLEAN DeletePending;
LONG OpenCount; LONG OpenCount;
@ -101,11 +103,7 @@ static NTSTATUS FspFsvolCleanup(
/* create the user-mode file system request */ /* create the user-mode file system request */
Result = FspIopCreateRequest(Irp, FileNameRequired ? &FsContext->FileName : 0, 0, &Request); Result = FspIopCreateRequest(Irp, FileNameRequired ? &FsContext->FileName : 0, 0, &Request);
if (!NT_SUCCESS(Result)) if (!NT_SUCCESS(Result))
{ goto leak_exit;
/* IRP_MJ_CLEANUP cannot really fail :-\ */
Result = STATUS_SUCCESS;
goto exit;
}
/* /*
* The new request is associated with our IRP and will be deleted during its completion. * The new request is associated with our IRP and will be deleted during its completion.
@ -113,12 +111,24 @@ static NTSTATUS FspFsvolCleanup(
/* populate the Cleanup request */ /* populate the Cleanup request */
Request->Kind = FspFsctlTransactCleanupKind; Request->Kind = FspFsctlTransactCleanupKind;
Request->Req.Cleanup.UserContext = FsContext->UserContext; Request->Req.Cleanup.UserContext = UserContext;
Request->Req.Cleanup.UserContext2 = (UINT_PTR)FileObject->FsContext2; Request->Req.Cleanup.UserContext2 = UserContext2;
Request->Req.Cleanup.Delete = DeletePending && 0 == OpenCount; Request->Req.Cleanup.Delete = DeletePending && 0 == OpenCount;
Result = STATUS_PENDING; Result = STATUS_PENDING;
goto exit;
leak_exit:;
#if DBG
DEBUGLOG("FileObject=%p[%p:\"%wZ\"], UserContext=%llx, UserContext2=%p: "
"error: the user-mode file system handle will be leaked!",
IrpSp->FileObject, IrpSp->FileObject->RelatedFileObject, IrpSp->FileObject->FileName,
UserContext, UserContext2);
#endif
/* IRP_MJ_CLEANUP cannot really fail :-\ */
Result = STATUS_SUCCESS;
exit:; exit:;
} }
finally finally

29
src/sys/close.c
View File

@ -60,6 +60,8 @@ static NTSTATUS FspFsvolClose(
FspFsvrtDeviceExtension(FsvrtDeviceObject); FspFsvrtDeviceExtension(FsvrtDeviceObject);
PFILE_OBJECT FileObject = IrpSp->FileObject; PFILE_OBJECT FileObject = IrpSp->FileObject;
FSP_FILE_CONTEXT *FsContext = FileObject->FsContext; FSP_FILE_CONTEXT *FsContext = FileObject->FsContext;
UINT64 UserContext = FsContext->UserContext;
UINT64 UserContext2 = (UINT_PTR)FileObject->FsContext2;
BOOLEAN FileNameRequired = 0 != FsvrtDeviceExtension->VolumeParams.FileNameRequired; BOOLEAN FileNameRequired = 0 != FsvrtDeviceExtension->VolumeParams.FileNameRequired;
FSP_FSCTL_TRANSACT_REQ *Request; FSP_FSCTL_TRANSACT_REQ *Request;
@ -69,11 +71,7 @@ static NTSTATUS FspFsvolClose(
/* create the user-mode file system request */ /* create the user-mode file system request */
Result = FspIopCreateRequest(0, FileNameRequired ? &FsContext->FileName : 0, 0, &Request); Result = FspIopCreateRequest(0, FileNameRequired ? &FsContext->FileName : 0, 0, &Request);
if (!NT_SUCCESS(Result)) if (!NT_SUCCESS(Result))
{ goto leak_exit;
/* IRP_MJ_CLOSE cannot really fail :-\ */
Result = STATUS_SUCCESS;
goto exit;
}
/* /*
* The new request is associated with our IRP and will be deleted during its completion. * The new request is associated with our IRP and will be deleted during its completion.
@ -81,19 +79,26 @@ static NTSTATUS FspFsvolClose(
/* populate the Close request */ /* populate the Close request */
Request->Kind = FspFsctlTransactCloseKind; Request->Kind = FspFsctlTransactCloseKind;
Request->Req.Close.UserContext = FsContext->UserContext; Request->Req.Close.UserContext = UserContext;
Request->Req.Close.UserContext2 = (UINT_PTR)FileObject->FsContext2; Request->Req.Close.UserContext2 = UserContext2;
/* post as a work request; this allows us to complete our own IRP and return immediately! */ /* post as a work request; this allows us to complete our own IRP and return immediately! */
if (!FspIopPostWorkRequest(DeviceObject, Request)) if (!FspIopPostWorkRequest(DeviceObject, Request))
{
/* no need to delete the request here as FspIopPostWorkRequest() will do so in all cases */ /* no need to delete the request here as FspIopPostWorkRequest() will do so in all cases */
goto leak_exit;
/* IRP_MJ_CLOSE cannot really fail :-\ */ Result = STATUS_SUCCESS;
Result = STATUS_SUCCESS;
goto exit;
}
goto exit;
leak_exit:;
#if DBG
DEBUGLOG("FileObject=%p[%p:\"%wZ\"], UserContext=%llx, UserContext2=%p: "
"error: the user-mode file system handle will be leaked!",
IrpSp->FileObject, IrpSp->FileObject->RelatedFileObject, IrpSp->FileObject->FileName,
UserContext, UserContext2);
#endif
/* IRP_MJ_CLOSE cannot really fail :-\ */
Result = STATUS_SUCCESS; Result = STATUS_SUCCESS;
exit:; exit:;

8
src/sys/create.c
View File

@ -680,6 +680,8 @@ static VOID FspFsvolCreateClose(
FspFsvrtDeviceExtension(FsvolDeviceExtension->FsvrtDeviceObject); FspFsvrtDeviceExtension(FsvolDeviceExtension->FsvrtDeviceObject);
PFILE_OBJECT FileObject = IrpSp->FileObject; PFILE_OBJECT FileObject = IrpSp->FileObject;
FSP_FILE_CONTEXT *FsContext = FileObject->FsContext; FSP_FILE_CONTEXT *FsContext = FileObject->FsContext;
UINT64 UserContext = FsContext->UserContext;
UINT64 UserContext2 = (UINT_PTR)FileObject->FsContext2;
BOOLEAN FileNameRequired = 0 != FsvrtDeviceExtension->VolumeParams.FileNameRequired; BOOLEAN FileNameRequired = 0 != FsvrtDeviceExtension->VolumeParams.FileNameRequired;
FSP_FSCTL_TRANSACT_REQ *Request; FSP_FSCTL_TRANSACT_REQ *Request;
@ -690,8 +692,8 @@ static VOID FspFsvolCreateClose(
/* populate the CreateCleanupClose request */ /* populate the CreateCleanupClose request */
Request->Kind = FspFsctlTransactCreateCleanupCloseKind; Request->Kind = FspFsctlTransactCreateCleanupCloseKind;
Request->Req.Cleanup.UserContext = FsContext->UserContext; Request->Req.Cleanup.UserContext = UserContext;
Request->Req.Cleanup.UserContext2 = (UINT_PTR)FileObject->FsContext2; Request->Req.Cleanup.UserContext2 = UserContext2;
Request->Req.Cleanup.Delete = FILE_CREATED == Response->IoStatus.Information; Request->Req.Cleanup.Delete = FILE_CREATED == Response->IoStatus.Information;
/* post as a work request */ /* post as a work request */
@ -706,7 +708,7 @@ leak_exit:;
DEBUGLOG("FileObject=%p[%p:\"%wZ\"], UserContext=%llx, UserContext2=%p: " DEBUGLOG("FileObject=%p[%p:\"%wZ\"], UserContext=%llx, UserContext2=%p: "
"error: the user-mode file system handle will be leaked!", "error: the user-mode file system handle will be leaked!",
IrpSp->FileObject, IrpSp->FileObject->RelatedFileObject, IrpSp->FileObject->FileName, IrpSp->FileObject, IrpSp->FileObject->RelatedFileObject, IrpSp->FileObject->FileName,
FsContext->UserContext, FileObject->FsContext2); UserContext, UserContext2);
#endif #endif
exit: exit: