mirror of
https://github.com/winfsp/winfsp.git
synced 2025-06-08 13:02:10 -05:00
sys: ioq: FspIoqRetryCompleteIrp(), FspIoqNextCompleteIrp()
This commit is contained in:
parent
0de4ba4f1b
commit
ffbe13e6d9
@ -366,8 +366,8 @@ typedef struct
|
|||||||
KSPIN_LOCK SpinLock;
|
KSPIN_LOCK SpinLock;
|
||||||
BOOLEAN Stopped;
|
BOOLEAN Stopped;
|
||||||
KEVENT PendingIrpEvent;
|
KEVENT PendingIrpEvent;
|
||||||
LIST_ENTRY PendingIrpList, ProcessIrpList, RetryIrpList;
|
LIST_ENTRY PendingIrpList, ProcessIrpList, RetriedIrpList;
|
||||||
IO_CSQ PendingIoCsq, ProcessIoCsq, RetryIoCsq;
|
IO_CSQ PendingIoCsq, ProcessIoCsq, RetriedIoCsq;
|
||||||
ULONG IrpTimeout;
|
ULONG IrpTimeout;
|
||||||
ULONG PendingIrpCapacity, PendingIrpCount;
|
ULONG PendingIrpCapacity, PendingIrpCount;
|
||||||
VOID (*CompleteCanceledIrp)(PIRP Irp);
|
VOID (*CompleteCanceledIrp)(PIRP Irp);
|
||||||
@ -386,7 +386,7 @@ PIRP FspIoqNextPendingIrp(FSP_IOQ *Ioq, PIRP BoundaryIrp, PLARGE_INTEGER Timeout
|
|||||||
BOOLEAN FspIoqStartProcessingIrp(FSP_IOQ *Ioq, PIRP Irp);
|
BOOLEAN FspIoqStartProcessingIrp(FSP_IOQ *Ioq, PIRP Irp);
|
||||||
PIRP FspIoqEndProcessingIrp(FSP_IOQ *Ioq, UINT_PTR IrpHint);
|
PIRP FspIoqEndProcessingIrp(FSP_IOQ *Ioq, UINT_PTR IrpHint);
|
||||||
BOOLEAN FspIoqRetryCompleteIrp(FSP_IOQ *Ioq, PIRP Irp);
|
BOOLEAN FspIoqRetryCompleteIrp(FSP_IOQ *Ioq, PIRP Irp);
|
||||||
PIRP FspIoqNextCompletedIrp(FSP_IOQ *Ioq, PIRP BoundaryIrp);
|
PIRP FspIoqNextCompleteIrp(FSP_IOQ *Ioq, PIRP BoundaryIrp);
|
||||||
|
|
||||||
/* I/O processing */
|
/* I/O processing */
|
||||||
#define FSP_FSCTL_WORK \
|
#define FSP_FSCTL_WORK \
|
||||||
|
@ -288,26 +288,26 @@ static VOID FspIoqProcessCompleteCanceledIrp(PIO_CSQ IoCsq, PIRP Irp)
|
|||||||
Ioq->CompleteCanceledIrp(Irp);
|
Ioq->CompleteCanceledIrp(Irp);
|
||||||
}
|
}
|
||||||
|
|
||||||
static NTSTATUS FspIoqRetryInsertIrpEx(PIO_CSQ IoCsq, PIRP Irp, PVOID InsertContext)
|
static NTSTATUS FspIoqRetriedInsertIrpEx(PIO_CSQ IoCsq, PIRP Irp, PVOID InsertContext)
|
||||||
{
|
{
|
||||||
FSP_IOQ *Ioq = CONTAINING_RECORD(IoCsq, FSP_IOQ, RetryIoCsq);
|
FSP_IOQ *Ioq = CONTAINING_RECORD(IoCsq, FSP_IOQ, RetriedIoCsq);
|
||||||
if (Ioq->Stopped)
|
if (Ioq->Stopped)
|
||||||
return STATUS_CANCELLED;
|
return STATUS_CANCELLED;
|
||||||
InsertTailList(&Ioq->RetryIrpList, &Irp->Tail.Overlay.ListEntry);
|
InsertTailList(&Ioq->RetriedIrpList, &Irp->Tail.Overlay.ListEntry);
|
||||||
return STATUS_SUCCESS;
|
return STATUS_SUCCESS;
|
||||||
}
|
}
|
||||||
|
|
||||||
static VOID FspIoqRetryRemoveIrp(PIO_CSQ IoCsq, PIRP Irp)
|
static VOID FspIoqRetriedRemoveIrp(PIO_CSQ IoCsq, PIRP Irp)
|
||||||
{
|
{
|
||||||
RemoveEntryList(&Irp->Tail.Overlay.ListEntry);
|
RemoveEntryList(&Irp->Tail.Overlay.ListEntry);
|
||||||
}
|
}
|
||||||
|
|
||||||
static PIRP FspIoqRetryPeekNextIrp(PIO_CSQ IoCsq, PIRP Irp, PVOID PeekContext)
|
static PIRP FspIoqRetriedPeekNextIrp(PIO_CSQ IoCsq, PIRP Irp, PVOID PeekContext)
|
||||||
{
|
{
|
||||||
FSP_IOQ *Ioq = CONTAINING_RECORD(IoCsq, FSP_IOQ, RetryIoCsq);
|
FSP_IOQ *Ioq = CONTAINING_RECORD(IoCsq, FSP_IOQ, RetriedIoCsq);
|
||||||
if (PeekContext && Ioq->Stopped)
|
if (PeekContext && Ioq->Stopped)
|
||||||
return 0;
|
return 0;
|
||||||
PLIST_ENTRY Head = &Ioq->RetryIrpList;
|
PLIST_ENTRY Head = &Ioq->RetriedIrpList;
|
||||||
PLIST_ENTRY Entry = 0 == Irp ? Head->Flink : Irp->Tail.Overlay.ListEntry.Flink;
|
PLIST_ENTRY Entry = 0 == Irp ? Head->Flink : Irp->Tail.Overlay.ListEntry.Flink;
|
||||||
if (Head == Entry)
|
if (Head == Entry)
|
||||||
return 0;
|
return 0;
|
||||||
@ -337,22 +337,22 @@ static PIRP FspIoqRetryPeekNextIrp(PIO_CSQ IoCsq, PIRP Irp, PVOID PeekContext)
|
|||||||
}
|
}
|
||||||
|
|
||||||
_IRQL_raises_(DISPATCH_LEVEL)
|
_IRQL_raises_(DISPATCH_LEVEL)
|
||||||
static VOID FspIoqRetryAcquireLock(PIO_CSQ IoCsq, _At_(*PIrql, _IRQL_saves_) PKIRQL PIrql)
|
static VOID FspIoqRetriedAcquireLock(PIO_CSQ IoCsq, _At_(*PIrql, _IRQL_saves_) PKIRQL PIrql)
|
||||||
{
|
{
|
||||||
FSP_IOQ *Ioq = CONTAINING_RECORD(IoCsq, FSP_IOQ, RetryIoCsq);
|
FSP_IOQ *Ioq = CONTAINING_RECORD(IoCsq, FSP_IOQ, RetriedIoCsq);
|
||||||
KeAcquireSpinLock(&Ioq->SpinLock, PIrql);
|
KeAcquireSpinLock(&Ioq->SpinLock, PIrql);
|
||||||
}
|
}
|
||||||
|
|
||||||
_IRQL_requires_(DISPATCH_LEVEL)
|
_IRQL_requires_(DISPATCH_LEVEL)
|
||||||
static VOID FspIoqRetryReleaseLock(PIO_CSQ IoCsq, _IRQL_restores_ KIRQL Irql)
|
static VOID FspIoqRetriedReleaseLock(PIO_CSQ IoCsq, _IRQL_restores_ KIRQL Irql)
|
||||||
{
|
{
|
||||||
FSP_IOQ *Ioq = CONTAINING_RECORD(IoCsq, FSP_IOQ, RetryIoCsq);
|
FSP_IOQ *Ioq = CONTAINING_RECORD(IoCsq, FSP_IOQ, RetriedIoCsq);
|
||||||
KeReleaseSpinLock(&Ioq->SpinLock, Irql);
|
KeReleaseSpinLock(&Ioq->SpinLock, Irql);
|
||||||
}
|
}
|
||||||
|
|
||||||
static VOID FspIoqRetryCompleteCanceledIrp(PIO_CSQ IoCsq, PIRP Irp)
|
static VOID FspIoqRetriedCompleteCanceledIrp(PIO_CSQ IoCsq, PIRP Irp)
|
||||||
{
|
{
|
||||||
FSP_IOQ *Ioq = CONTAINING_RECORD(IoCsq, FSP_IOQ, RetryIoCsq);
|
FSP_IOQ *Ioq = CONTAINING_RECORD(IoCsq, FSP_IOQ, RetriedIoCsq);
|
||||||
Ioq->CompleteCanceledIrp(Irp);
|
Ioq->CompleteCanceledIrp(Irp);
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -375,7 +375,7 @@ NTSTATUS FspIoqCreate(
|
|||||||
KeInitializeEvent(&Ioq->PendingIrpEvent, NotificationEvent, FALSE);
|
KeInitializeEvent(&Ioq->PendingIrpEvent, NotificationEvent, FALSE);
|
||||||
InitializeListHead(&Ioq->PendingIrpList);
|
InitializeListHead(&Ioq->PendingIrpList);
|
||||||
InitializeListHead(&Ioq->ProcessIrpList);
|
InitializeListHead(&Ioq->ProcessIrpList);
|
||||||
InitializeListHead(&Ioq->RetryIrpList);
|
InitializeListHead(&Ioq->RetriedIrpList);
|
||||||
IoCsqInitializeEx(&Ioq->PendingIoCsq,
|
IoCsqInitializeEx(&Ioq->PendingIoCsq,
|
||||||
FspIoqPendingInsertIrpEx,
|
FspIoqPendingInsertIrpEx,
|
||||||
FspIoqPendingRemoveIrp,
|
FspIoqPendingRemoveIrp,
|
||||||
@ -390,13 +390,13 @@ NTSTATUS FspIoqCreate(
|
|||||||
FspIoqProcessAcquireLock,
|
FspIoqProcessAcquireLock,
|
||||||
FspIoqProcessReleaseLock,
|
FspIoqProcessReleaseLock,
|
||||||
FspIoqProcessCompleteCanceledIrp);
|
FspIoqProcessCompleteCanceledIrp);
|
||||||
IoCsqInitializeEx(&Ioq->RetryIoCsq,
|
IoCsqInitializeEx(&Ioq->RetriedIoCsq,
|
||||||
FspIoqRetryInsertIrpEx,
|
FspIoqRetriedInsertIrpEx,
|
||||||
FspIoqRetryRemoveIrp,
|
FspIoqRetriedRemoveIrp,
|
||||||
FspIoqRetryPeekNextIrp,
|
FspIoqRetriedPeekNextIrp,
|
||||||
FspIoqRetryAcquireLock,
|
FspIoqRetriedAcquireLock,
|
||||||
FspIoqRetryReleaseLock,
|
FspIoqRetriedReleaseLock,
|
||||||
FspIoqRetryCompleteCanceledIrp);
|
FspIoqRetriedCompleteCanceledIrp);
|
||||||
Ioq->IrpTimeout = ConvertInterruptTimeToSec(IrpTimeout->QuadPart + InterruptTimeToSecFactor - 1);
|
Ioq->IrpTimeout = ConvertInterruptTimeToSec(IrpTimeout->QuadPart + InterruptTimeToSecFactor - 1);
|
||||||
/* convert to seconds (and round up) */
|
/* convert to seconds (and round up) */
|
||||||
Ioq->PendingIrpCapacity = IrpCapacity;
|
Ioq->PendingIrpCapacity = IrpCapacity;
|
||||||
@ -427,7 +427,7 @@ VOID FspIoqStop(FSP_IOQ *Ioq)
|
|||||||
Ioq->CompleteCanceledIrp(Irp);
|
Ioq->CompleteCanceledIrp(Irp);
|
||||||
while (0 != (Irp = FspCsqRemoveNextIrp(&Ioq->ProcessIoCsq, 0)))
|
while (0 != (Irp = FspCsqRemoveNextIrp(&Ioq->ProcessIoCsq, 0)))
|
||||||
Ioq->CompleteCanceledIrp(Irp);
|
Ioq->CompleteCanceledIrp(Irp);
|
||||||
while (0 != (Irp = FspCsqRemoveNextIrp(&Ioq->RetryIoCsq, 0)))
|
while (0 != (Irp = FspCsqRemoveNextIrp(&Ioq->RetriedIoCsq, 0)))
|
||||||
Ioq->CompleteCanceledIrp(Irp);
|
Ioq->CompleteCanceledIrp(Irp);
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -527,14 +527,14 @@ BOOLEAN FspIoqRetryCompleteIrp(FSP_IOQ *Ioq, PIRP Irp)
|
|||||||
if (FspIrpTimestampInfinity != FspIrpTimestamp(Irp))
|
if (FspIrpTimestampInfinity != FspIrpTimestamp(Irp))
|
||||||
FspIrpTimestamp(Irp) = QueryInterruptTimeInSec() + Ioq->IrpTimeout;
|
FspIrpTimestamp(Irp) = QueryInterruptTimeInSec() + Ioq->IrpTimeout;
|
||||||
#endif
|
#endif
|
||||||
Result = FspCsqInsertIrpEx(&Ioq->RetryIoCsq, Irp, 0, 0);
|
Result = FspCsqInsertIrpEx(&Ioq->RetriedIoCsq, Irp, 0, 0);
|
||||||
return NT_SUCCESS(Result);
|
return NT_SUCCESS(Result);
|
||||||
}
|
}
|
||||||
|
|
||||||
PIRP FspIoqNextCompletedIrp(FSP_IOQ *Ioq, PIRP BoundaryIrp)
|
PIRP FspIoqNextCompleteIrp(FSP_IOQ *Ioq, PIRP BoundaryIrp)
|
||||||
{
|
{
|
||||||
FSP_IOQ_PEEK_CONTEXT PeekContext;
|
FSP_IOQ_PEEK_CONTEXT PeekContext;
|
||||||
PeekContext.IrpHint = 0 != BoundaryIrp ? BoundaryIrp : (PVOID)1;
|
PeekContext.IrpHint = 0 != BoundaryIrp ? BoundaryIrp : (PVOID)1;
|
||||||
PeekContext.ExpirationTime = 0;
|
PeekContext.ExpirationTime = 0;
|
||||||
return IoCsqRemoveNextIrp(&Ioq->RetryIoCsq, &PeekContext);
|
return IoCsqRemoveNextIrp(&Ioq->RetriedIoCsq, &PeekContext);
|
||||||
}
|
}
|
||||||
|
Loading…
x
Reference in New Issue
Block a user