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: wq: FspWq* routiens

Browse Source 
sys: write: initial implementation
This commit is contained in:
Bill Zissimopoulos
2016-02-23 20:07:33 -08:00
parent c30fec9daf
commit 892e8eb025
6 changed files with 275 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

74
src/sys/wq.c Normal file
View File

@ -0,0 +1,74 @@
/**
* @file sys/wq.c
*
* @copyright 2015 Bill Zissimopoulos
*/
#include <sys/driver.h>
static VOID FspWqWorkRoutine(PVOID Context);
NTSTATUS FspWqCreateAndPostIrpWorkItem(PIRP Irp,
FSP_WQ_REQUEST_WORK *WorkRoutine, FSP_IOP_REQUEST_FINI *RequestFini,
BOOLEAN CreateAndPost)
{
FSP_FSCTL_TRANSACT_REQ *RequestWorkItem = FspIrpRequest(Irp);
if (0 == RequestWorkItem)
{
NTSTATUS Result = FspIopCreateRequestWorkItem(Irp, sizeof(WORK_QUEUE_ITEM),
RequestFini, &RequestWorkItem);
if (!NT_SUCCESS(Result))
return Result;
ASSERT(sizeof(FSP_WQ_REQUEST_WORK *) == sizeof(PVOID));
PIRP TopLevelIrp = IoGetTopLevelIrp();
FspIopRequestContext(RequestWorkItem, FspWqRequestIrpAndFlags) =
(PVOID)((UINT_PTR)Irp | (0 == TopLevelIrp || Irp == TopLevelIrp));
FspIopRequestContext(RequestWorkItem, FspWqRequestWorkRoutine) =
(PVOID)(UINT_PTR)WorkRoutine;
ExInitializeWorkItem((PWORK_QUEUE_ITEM)&RequestWorkItem->Buffer, FspWqWorkRoutine, Irp);
}
if (!CreateAndPost)
return STATUS_SUCCESS;
FspWqPostIrpWorkItem(Irp);
return STATUS_PENDING;
}
VOID FspWqPostIrpWorkItem(PIRP Irp)
{
FSP_FSCTL_TRANSACT_REQ *RequestWorkItem = FspIrpRequest(Irp);
ASSERT(RequestWorkItem->Kind == FspFsctlTransactReservedKind);
ASSERT(RequestWorkItem->Size == sizeof *RequestWorkItem + sizeof(WORK_QUEUE_ITEM));
ASSERT(RequestWorkItem->Hint == (UINT_PTR)Irp);
IoMarkIrpPending(Irp);
ExQueueWorkItem((PWORK_QUEUE_ITEM)&RequestWorkItem->Buffer, CriticalWorkQueue);
}
static VOID FspWqWorkRoutine(PVOID Context)
{
PIRP Irp = Context;
PIO_STACK_LOCATION IrpSp = IoGetCurrentIrpStackLocation(Irp);
FSP_FSCTL_TRANSACT_REQ *RequestWorkItem = FspIrpRequest(Irp);
BOOLEAN TopLevel = (BOOLEAN)
((UINT_PTR)FspIopRequestContext(RequestWorkItem, FspWqRequestIrpAndFlags) & 1);
FSP_WQ_REQUEST_WORK *WorkRoutine = (FSP_WQ_REQUEST_WORK *)(UINT_PTR)
FspIopRequestContext(RequestWorkItem, FspWqRequestWorkRoutine);
NTSTATUS Result;
IoSetTopLevelIrp(TopLevel ? Irp : (PIRP)FSRTL_FSP_TOP_LEVEL_IRP);
Result = WorkRoutine(IrpSp->DeviceObject, Irp, IrpSp, FALSE);
if (STATUS_PENDING != Result)
{
DEBUGLOGIRP(Irp, Result);
FspIopCompleteIrp(Irp, Result);
}
IoSetTopLevelIrp(0);
}