Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2024-12-29 10:35:28 +00:00
Code Issues Projects Releases Packages Wiki Activity

[RDBSS] More capture

Browse source
This commit is contained in:
Pierre Schweitzer 2017-11-01 15:12:08 +01:00
parent 2511ba8f8f
commit 5b960479bf
No known key found for this signature in database
GPG key ID: 7545556C3D585B0B
1 changed files with 36 additions and 34 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

70
sdk/lib/drivers/rdbsslib/rdbss.c
View file

@ -772,6 +772,7 @@ RxAddToWorkque(
ULONG Queued; ULONG Queued;
KIRQL OldIrql; KIRQL OldIrql;
WORK_QUEUE_TYPE Queue; WORK_QUEUE_TYPE Queue;
RxCaptureParamBlock; RxCaptureParamBlock;
RxContext->PostRequest = FALSE; RxContext->PostRequest = FALSE;
@ -1240,6 +1241,9 @@ RxCanonicalizeNameAndObtainNetRoot(
NET_ROOT_TYPE NetRootType; NET_ROOT_TYPE NetRootType;
UNICODE_STRING CanonicalName; UNICODE_STRING CanonicalName;
RxCaptureParamBlock;
RxCaptureFileObject;
PAGED_CODE(); PAGED_CODE();
NetRootType = NET_ROOT_WILD; NetRootType = NET_ROOT_WILD;
@ -1248,7 +1252,7 @@ RxCanonicalizeNameAndObtainNetRoot(
RtlInitEmptyUnicodeString(&CanonicalName, NULL, 0); RtlInitEmptyUnicodeString(&CanonicalName, NULL, 0);
/* if not relative opening, just handle the passed name */ /* if not relative opening, just handle the passed name */
if (RxContext->CurrentIrpSp->FileObject->RelatedFileObject == NULL) if (capFileObject->RelatedFileObject == NULL)
{ {
Status = RxFirstCanonicalize(RxContext, FileName, &CanonicalName, &NetRootType); Status = RxFirstCanonicalize(RxContext, FileName, &CanonicalName, &NetRootType);
if (!NT_SUCCESS(Status)) if (!NT_SUCCESS(Status))
@ -1261,9 +1265,8 @@ RxCanonicalizeNameAndObtainNetRoot(
PFCB Fcb; PFCB Fcb;
/* Make sure we have a valid FCB and a FOBX */ /* Make sure we have a valid FCB and a FOBX */
Fcb = RxContext->CurrentIrpSp->FileObject->RelatedFileObject->FsContext; Fcb = capFileObject->RelatedFileObject->FsContext;
if (Fcb == NULL || if (Fcb == NULL || capFileObject->RelatedFileObject->FsContext2 == NULL)
RxContext->CurrentIrpSp->FileObject->RelatedFileObject->FsContext2 == NULL)
{ {
return STATUS_INVALID_PARAMETER; return STATUS_INVALID_PARAMETER;
} }
@ -1612,26 +1615,25 @@ NTSTATUS
RxCollapseOrCreateSrvOpen( RxCollapseOrCreateSrvOpen(
PRX_CONTEXT RxContext) PRX_CONTEXT RxContext)
{ {
PFCB Fcb;
NTSTATUS Status; NTSTATUS Status;
ULONG Disposition; ULONG Disposition;
PSRV_OPEN SrvOpen; PSRV_OPEN SrvOpen;
USHORT ShareAccess; USHORT ShareAccess;
PIO_STACK_LOCATION Stack;
ACCESS_MASK DesiredAccess; ACCESS_MASK DesiredAccess;
RX_BLOCK_CONDITION FcbCondition; RX_BLOCK_CONDITION FcbCondition;
RxCaptureFcb;
RxCaptureParamBlock;
PAGED_CODE(); PAGED_CODE();
DPRINT("RxCollapseOrCreateSrvOpen(%p)\n", RxContext); DPRINT("RxCollapseOrCreateSrvOpen(%p)\n", RxContext);
Fcb = (PFCB)RxContext->pFcb; ASSERT(RxIsFcbAcquiredExclusive(capFcb));
ASSERT(RxIsFcbAcquiredExclusive(Fcb)); ++capFcb->UncleanCount;
++Fcb->UncleanCount;
Stack = RxContext->CurrentIrpSp; DesiredAccess = capPARAMS->Parameters.Create.SecurityContext->DesiredAccess & FILE_ALL_ACCESS;
DesiredAccess = Stack->Parameters.Create.SecurityContext->DesiredAccess & FILE_ALL_ACCESS; ShareAccess = capPARAMS->Parameters.Create.ShareAccess & FILE_SHARE_VALID_FLAGS;
ShareAccess = Stack->Parameters.Create.ShareAccess & FILE_SHARE_VALID_FLAGS;
Disposition = RxContext->Create.NtCreateParameters.Disposition; Disposition = RxContext->Create.NtCreateParameters.Disposition;
@ -1640,7 +1642,7 @@ RxCollapseOrCreateSrvOpen(
if (Status == STATUS_NOT_FOUND) if (Status == STATUS_NOT_FOUND)
{ {
/* If none found, create one */ /* If none found, create one */
SrvOpen = RxCreateSrvOpen((PV_NET_ROOT)RxContext->Create.pVNetRoot, Fcb); SrvOpen = RxCreateSrvOpen((PV_NET_ROOT)RxContext->Create.pVNetRoot, capFcb);
if (SrvOpen == NULL) if (SrvOpen == NULL)
{ {
Status = STATUS_INSUFFICIENT_RESOURCES; Status = STATUS_INSUFFICIENT_RESOURCES;
@ -1665,7 +1667,7 @@ RxCollapseOrCreateSrvOpen(
/* Cookie to check the mini-rdr doesn't mess with RX_CONTEXT */ /* Cookie to check the mini-rdr doesn't mess with RX_CONTEXT */
RxContext->CurrentIrp->IoStatus.Information = 0xABCDEF; RxContext->CurrentIrp->IoStatus.Information = 0xABCDEF;
/* Inform the mini-rdr we're handling a create */ /* Inform the mini-rdr we're handling a create */
MINIRDR_CALL(Status, RxContext, Fcb->MRxDispatch, MRxCreate, (RxContext)); MINIRDR_CALL(Status, RxContext, capFcb->MRxDispatch, MRxCreate, (RxContext));
ASSERT(RxContext->CurrentIrp->IoStatus.Information == 0xABCDEF); ASSERT(RxContext->CurrentIrp->IoStatus.Information == 0xABCDEF);
DPRINT("MRxCreate returned: %x\n", Status); DPRINT("MRxCreate returned: %x\n", Status);
@ -1674,23 +1676,23 @@ RxCollapseOrCreateSrvOpen(
/* In case of overwrite, reset file size */ /* In case of overwrite, reset file size */
if (Disposition == FILE_OVERWRITE || Disposition == FILE_OVERWRITE_IF) if (Disposition == FILE_OVERWRITE || Disposition == FILE_OVERWRITE_IF)
{ {
RxAcquirePagingIoResource(RxContext, Fcb); RxAcquirePagingIoResource(RxContext, capFcb);
Fcb->Header.AllocationSize.QuadPart = 0LL; capFcb->Header.AllocationSize.QuadPart = 0LL;
Fcb->Header.FileSize.QuadPart = 0LL; capFcb->Header.FileSize.QuadPart = 0LL;
Fcb->Header.ValidDataLength.QuadPart = 0LL; capFcb->Header.ValidDataLength.QuadPart = 0LL;
RxContext->CurrentIrpSp->FileObject->SectionObjectPointer = &Fcb->NonPaged->SectionObjectPointers; RxContext->CurrentIrpSp->FileObject->SectionObjectPointer = &capFcb->NonPaged->SectionObjectPointers;
CcSetFileSizes(RxContext->CurrentIrpSp->FileObject, (PCC_FILE_SIZES)&Fcb->Header.AllocationSize); CcSetFileSizes(RxContext->CurrentIrpSp->FileObject, (PCC_FILE_SIZES)&capFcb->Header.AllocationSize);
RxReleasePagingIoResource(RxContext, Fcb); RxReleasePagingIoResource(RxContext, capFcb);
} }
else else
{ {
/* Otherwise, adjust sizes */ /* Otherwise, adjust sizes */
RxContext->CurrentIrpSp->FileObject->SectionObjectPointer = &Fcb->NonPaged->SectionObjectPointers; RxContext->CurrentIrpSp->FileObject->SectionObjectPointer = &capFcb->NonPaged->SectionObjectPointers;
if (CcIsFileCached(RxContext->CurrentIrpSp->FileObject)) if (CcIsFileCached(RxContext->CurrentIrpSp->FileObject))
{ {
RxAdjustAllocationSizeforCC(Fcb); RxAdjustAllocationSizeforCC(capFcb);
} }
CcSetFileSizes(RxContext->CurrentIrpSp->FileObject, (PCC_FILE_SIZES)&Fcb->Header.AllocationSize); CcSetFileSizes(RxContext->CurrentIrpSp->FileObject, (PCC_FILE_SIZES)&capFcb->Header.AllocationSize);
} }
} }
@ -1701,15 +1703,15 @@ RxCollapseOrCreateSrvOpen(
/* Set SRV_OPEN state - good or bad - depending on whether create succeed */ /* Set SRV_OPEN state - good or bad - depending on whether create succeed */
RxTransitionSrvOpen(SrvOpen, (Status == STATUS_SUCCESS ? Condition_Good : Condition_Bad)); RxTransitionSrvOpen(SrvOpen, (Status == STATUS_SUCCESS ? Condition_Good : Condition_Bad));
ASSERT(RxIsFcbAcquiredExclusive(Fcb)); ASSERT(RxIsFcbAcquiredExclusive(capFcb));
RxCompleteSrvOpenKeyAssociation(SrvOpen); RxCompleteSrvOpenKeyAssociation(SrvOpen);
if (Status == STATUS_SUCCESS) if (Status == STATUS_SUCCESS)
{ {
if (BooleanFlagOn(Stack->Parameters.Create.Options, FILE_DELETE_ON_CLOSE)) if (BooleanFlagOn(capPARAMS->Parameters.Create.Options, FILE_DELETE_ON_CLOSE))
{ {
ClearFlag(Fcb->FcbState, FCB_STATE_COLLAPSING_ENABLED); ClearFlag(capFcb->FcbState, FCB_STATE_COLLAPSING_ENABLED);
} }
SrvOpen->CreateOptions = RxContext->Create.NtCreateParameters.CreateOptions; SrvOpen->CreateOptions = RxContext->Create.NtCreateParameters.CreateOptions;
FcbCondition = Condition_Good; FcbCondition = Condition_Good;
@ -1729,8 +1731,8 @@ RxCollapseOrCreateSrvOpen(
} }
/* Set FCB state - good or bad - depending on whether create succeed */ /* Set FCB state - good or bad - depending on whether create succeed */
DPRINT("Transitioning FCB %p to condition %lx\n", Fcb, Fcb->Condition); DPRINT("Transitioning FCB %p to condition %lx\n", capFcb, capFcb->Condition);
RxTransitionNetFcb(Fcb, FcbCondition); RxTransitionNetFcb(capFcb, FcbCondition);
} }
else if (Status == STATUS_SUCCESS) else if (Status == STATUS_SUCCESS)
{ {
@ -1750,12 +1752,12 @@ RxCollapseOrCreateSrvOpen(
++SrvOpen->OpenCount; ++SrvOpen->OpenCount;
ExtraOpen = TRUE; ExtraOpen = TRUE;
RxReleaseFcb(RxContext, Fcb); RxReleaseFcb(RxContext, capFcb);
RxContext->Create.FcbAcquired = FALSE; RxContext->Create.FcbAcquired = FALSE;
RxWaitForStableSrvOpen(SrvOpen, RxContext); RxWaitForStableSrvOpen(SrvOpen, RxContext);
if (NT_SUCCESS(RxAcquireExclusiveFcb(RxContext, Fcb))) if (NT_SUCCESS(RxAcquireExclusiveFcb(RxContext, capFcb)))
{ {
RxContext->Create.FcbAcquired = TRUE; RxContext->Create.FcbAcquired = TRUE;
} }
@ -1766,9 +1768,9 @@ RxCollapseOrCreateSrvOpen(
/* Inform the mini-rdr we do an opening with a reused SRV_OPEN */ /* Inform the mini-rdr we do an opening with a reused SRV_OPEN */
if (IsGood) if (IsGood)
{ {
MINIRDR_CALL(Status, RxContext, Fcb->MRxDispatch, MRxCollapseOpen, (RxContext)); MINIRDR_CALL(Status, RxContext, capFcb->MRxDispatch, MRxCollapseOpen, (RxContext));
ASSERT(RxIsFcbAcquiredExclusive(Fcb)); ASSERT(RxIsFcbAcquiredExclusive(capFcb));
} }
else else
{ {
@ -1782,7 +1784,7 @@ RxCollapseOrCreateSrvOpen(
} }
} }
--Fcb->UncleanCount; --capFcb->UncleanCount;
DPRINT("Status: %x\n", Status); DPRINT("Status: %x\n", Status);
return Status; return Status;