NTSTATUS
SrvProcessFlush_SMB_V2(
PSRV_EXEC_CONTEXT pExecContext
)
{
NTSTATUS ntStatus = STATUS_SUCCESS;
PLWIO_SRV_CONNECTION pConnection = pExecContext->pConnection;
PSRV_PROTOCOL_EXEC_CONTEXT pCtxProtocol = pExecContext->pProtocolContext;
PSRV_EXEC_CONTEXT_SMB_V2 pCtxSmb2 = pCtxProtocol->pSmb2Context;
PSRV_FLUSH_STATE_SMB_V2 pFlushState = NULL;
PLWIO_SRV_SESSION_2 pSession = NULL;
PLWIO_SRV_TREE_2 pTree = NULL;
PLWIO_SRV_FILE_2 pFile = NULL;
BOOLEAN bInLock = FALSE;
pFlushState = (PSRV_FLUSH_STATE_SMB_V2)pCtxSmb2->hState;
if (pFlushState)
{
InterlockedIncrement(&pFlushState->refCount);
}
else
{
ULONG iMsg = pCtxSmb2->iMsg;
PSRV_MESSAGE_SMB_V2 pSmbRequest = &pCtxSmb2->pRequests[iMsg];
PSMB2_FID pFid = NULL; // Do not free
ntStatus = SrvConnection2FindSession_SMB_V2(
pCtxSmb2,
pConnection,
pSmbRequest->pHeader->ullSessionId,
&pSession);
BAIL_ON_NT_STATUS(ntStatus);
ntStatus = SrvSetStatSession2Info(pExecContext, pSession);
BAIL_ON_NT_STATUS(ntStatus);
ntStatus = SrvSession2FindTree_SMB_V2(
pCtxSmb2,
pSession,
pSmbRequest->pHeader->ulTid,
&pTree);
BAIL_ON_NT_STATUS(ntStatus);
ntStatus = SMB2UnmarshalFlushRequest(pSmbRequest, &pFid);
BAIL_ON_NT_STATUS(ntStatus);
ntStatus = SrvTree2FindFile_SMB_V2(
pCtxSmb2,
pTree,
pFid,
LwIsSetFlag(
pSmbRequest->pHeader->ulFlags,
SMB2_FLAGS_RELATED_OPERATION),
&pFile);
BAIL_ON_NT_STATUS(ntStatus);
ntStatus = SrvBuildFlushState_SMB_V2(
pFid,
pFile,
&pFlushState);
BAIL_ON_NT_STATUS(ntStatus);
pCtxSmb2->hState = pFlushState;
InterlockedIncrement(&pFlushState->refCount);
pCtxSmb2->pfnStateRelease = &SrvReleaseFlushStateHandle_SMB_V2;
}
LWIO_LOCK_MUTEX(bInLock, &pFlushState->mutex);
switch (pFlushState->stage)
{
case SRV_FLUSH_STAGE_SMB_V2_INITIAL:
pFlushState->stage = SRV_FLUSH_STAGE_SMB_V2_FLUSH_COMPLETED;
SrvPrepareFlushStateAsync_SMB_V2(pFlushState, pExecContext);
ntStatus = IoFlushBuffersFile(
pFlushState->pFile->hFile,
pFlushState->pAcb,
&pFlushState->ioStatusBlock);
BAIL_ON_NT_STATUS(ntStatus);
SrvReleaseFlushStateAsync_SMB_V2(pFlushState); // completed synchronously
// intentional fall through
case SRV_FLUSH_STAGE_SMB_V2_FLUSH_COMPLETED:
ntStatus = pFlushState->ioStatusBlock.Status;
BAIL_ON_NT_STATUS(ntStatus);
pFlushState->stage = SRV_FLUSH_STAGE_SMB_V2_BUILD_RESPONSE;
// intentional fall through
case SRV_FLUSH_STAGE_SMB_V2_BUILD_RESPONSE:
ntStatus = SrvBuildFlushResponse_SMB_V2(pExecContext);
BAIL_ON_NT_STATUS(ntStatus);
pFlushState->stage = SRV_FLUSH_STAGE_SMB_V2_DONE;
// intentional fall through
case SRV_FLUSH_STAGE_SMB_V2_DONE:
break;
}
cleanup:
if (pFile)
{
SrvFile2Release(pFile);
}
if (pTree)
{
SrvTree2Release(pTree);
}
if (pSession)
{
SrvSession2Release(pSession);
}
if (pFlushState)
{
LWIO_UNLOCK_MUTEX(bInLock, &pFlushState->mutex);
SrvReleaseFlushState_SMB_V2(pFlushState);
}
return ntStatus;
error:
switch (ntStatus)
{
case STATUS_PENDING:
// TODO: Add an indicator to the file object to trigger a
// cleanup if the connection gets closed and all the
// files involved have to be closed
break;
default:
if (pFlushState)
{
SrvReleaseFlushStateAsync_SMB_V2(pFlushState);
}
break;
}
goto cleanup;
}
static
NTSTATUS
SrvBuildFlushResponse_SMB_V2(
PSRV_EXEC_CONTEXT pExecContext
)
{
NTSTATUS ntStatus = 0;
PSRV_PROTOCOL_EXEC_CONTEXT pCtxProtocol = pExecContext->pProtocolContext;
PSRV_EXEC_CONTEXT_SMB_V2 pCtxSmb2 = pCtxProtocol->pSmb2Context;
ULONG iMsg = pCtxSmb2->iMsg;
PSRV_MESSAGE_SMB_V2 pSmbRequest = &pCtxSmb2->pRequests[iMsg];
PSRV_MESSAGE_SMB_V2 pSmbResponse = &pCtxSmb2->pResponses[iMsg];
PBYTE pOutBuffer = pSmbResponse->pBuffer;
ULONG ulBytesAvailable = pSmbResponse->ulBytesAvailable;
ULONG ulOffset = 0;
ULONG ulBytesUsed = 0;
ULONG ulTotalBytesUsed = 0;
ntStatus = SMB2MarshalHeader(
pOutBuffer,
ulOffset,
ulBytesAvailable,
COM2_FLUSH,
pSmbRequest->pHeader->usEpoch,
pSmbRequest->pHeader->usCredits,
pSmbRequest->pHeader->ulPid,
pSmbRequest->pHeader->ullCommandSequence,
pCtxSmb2->pTree->ulTid,
pCtxSmb2->pSession->ullUid,
0LL, /* Async Id */
STATUS_SUCCESS,
TRUE,
LwIsSetFlag(
pSmbRequest->pHeader->ulFlags,
SMB2_FLAGS_RELATED_OPERATION),
&pSmbResponse->pHeader,
&pSmbResponse->ulHeaderSize);
BAIL_ON_NT_STATUS(ntStatus);
pOutBuffer += pSmbResponse->ulHeaderSize;
ulOffset += pSmbResponse->ulHeaderSize;
ulBytesAvailable -= pSmbResponse->ulHeaderSize;
ulTotalBytesUsed += pSmbResponse->ulHeaderSize;
ntStatus = SMB2MarshalFlushResponse(
pOutBuffer,
ulOffset,
ulBytesAvailable,
&ulBytesUsed);
BAIL_ON_NT_STATUS(ntStatus);
// pOutBuffer += ulBytesUsed;
// ulOffset += ulBytesUsed;
// ulBytesAvailable -= ulBytesUsed;
ulTotalBytesUsed += ulBytesUsed;
pSmbResponse->ulMessageSize = ulTotalBytesUsed;
cleanup:
return ntStatus;
error:
if (ulTotalBytesUsed)
{
pSmbResponse->pHeader = NULL;
pSmbResponse->ulMessageSize = 0;
memset(pSmbResponse->pBuffer, 0, ulTotalBytesUsed);
}
pSmbResponse->ulMessageSize = 0;
goto cleanup;
}
static
NTSTATUS
SrvBuildNotifyResponse_SMB_V2(
PSRV_EXEC_CONTEXT pExecContext,
PSRV_NOTIFY_STATE_SMB_V2 pNotifyState
)
{
NTSTATUS ntStatus = STATUS_SUCCESS;
PSRV_PROTOCOL_EXEC_CONTEXT pCtxProtocol = pExecContext->pProtocolContext;
PSRV_EXEC_CONTEXT_SMB_V2 pCtxSmb2 = pCtxProtocol->pSmb2Context;
ULONG iMsg = pCtxSmb2->iMsg;
PSRV_MESSAGE_SMB_V2 pSmbRequest = &pCtxSmb2->pRequests[iMsg];
PSRV_MESSAGE_SMB_V2 pSmbResponse = &pCtxSmb2->pResponses[iMsg];
PSMB2_NOTIFY_RESPONSE_HEADER pNotifyResponseHeader = NULL; // do not free
PBYTE pData = NULL;
ULONG ulDataLength = 0;
PBYTE pOutBuffer = pSmbResponse->pBuffer;
ULONG ulBytesAvailable = pSmbResponse->ulBytesAvailable;
ULONG ulDataOffset = 0;
ULONG ulOffset = 0;
ULONG ulBytesUsed = 0;
ULONG ulTotalBytesUsed = 0;
ntStatus = SrvCreditorAdjustCredits(
pExecContext->pConnection->pCreditor,
pSmbRequest->pHeader->ullCommandSequence,
pNotifyState->ullAsyncId,
pSmbRequest->pHeader->usCredits,
&pExecContext->usCreditsGranted);
BAIL_ON_NT_STATUS(ntStatus);
ntStatus = SMB2MarshalHeader(
pOutBuffer,
ulOffset,
ulBytesAvailable,
COM2_NOTIFY,
pSmbRequest->pHeader->usEpoch,
pExecContext->usCreditsGranted,
pSmbRequest->pHeader->ulPid,
pSmbRequest->pHeader->ullCommandSequence,
pCtxSmb2->pTree->ulTid,
pCtxSmb2->pSession->ullUid,
pNotifyState->ullAsyncId,
pSmbRequest->pHeader->error,
TRUE,
LwIsSetFlag(
pSmbRequest->pHeader->ulFlags,
SMB2_FLAGS_RELATED_OPERATION),
&pSmbResponse->pHeader,
&pSmbResponse->ulHeaderSize);
BAIL_ON_NT_STATUS(ntStatus);
pOutBuffer += pSmbResponse->ulHeaderSize;
ulOffset += pSmbResponse->ulHeaderSize;
ulBytesAvailable -= pSmbResponse->ulHeaderSize;
ulTotalBytesUsed += pSmbResponse->ulHeaderSize;
if ((pNotifyState->ioStatusBlock.Status == STATUS_SUCCESS) &&
pNotifyState->ulBytesUsed > 0)
{
ntStatus = SrvMarshalNotifyResponse_SMB_V2(
pNotifyState->pBuffer,
pNotifyState->ulBytesUsed,
&pData,
&ulDataLength);
BAIL_ON_NT_STATUS(ntStatus);
}
ntStatus = SMB2MarshalNotifyResponse(
pOutBuffer,
ulOffset,
ulBytesAvailable,
pData,
ulDataLength,
&ulDataOffset,
&pNotifyResponseHeader,
&ulBytesUsed);
BAIL_ON_NT_STATUS(ntStatus);
// pOutBuffer += ulBytesUsed;
// ulOffset += ulBytesUsed;
// ulBytesAvailable -= ulBytesUsed;
ulTotalBytesUsed += ulBytesUsed;
pSmbResponse->ulMessageSize = ulTotalBytesUsed;
cleanup:
if (pData)
{
SrvFreeMemory(pData);
}
return ntStatus;
error:
if (ulTotalBytesUsed)
{
pSmbResponse->pHeader = NULL;
pSmbResponse->ulHeaderSize = 0;
memset(pSmbResponse->pBuffer, 0, ulTotalBytesUsed);
}
pSmbResponse->ulMessageSize = 0;
goto cleanup;
}
