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; }