ANSC_STATUS
AnscDktoTsaRecvAppMessage
(
ANSC_HANDLE hThisObject,
ANSC_HANDLE hMessageBdo
)
{
ANSC_STATUS returnStatus = ANSC_STATUS_SUCCESS;
PANSC_DAEMON_SOCKET_TCP_OBJECT pMyObject = (PANSC_DAEMON_SOCKET_TCP_OBJECT)hThisObject;
PANSC_DAEMON_SERVER_TCP_OBJECT pServer = (PANSC_DAEMON_SERVER_TCP_OBJECT)pMyObject->hDaemonServer;
PANSC_DAEMON_ENGINE_TCP_OBJECT pEngine = (PANSC_DAEMON_ENGINE_TCP_OBJECT)pMyObject->hDaemonEngine;
PANSC_DSTO_WORKER_OBJECT pWorker = (PANSC_DSTO_WORKER_OBJECT )pServer->hWorker;
PANSC_BUFFER_DESCRIPTOR pPayloadBdo = (PANSC_BUFFER_DESCRIPTOR )hMessageBdo;
void* pRecvBuffer = (void* )NULL;
ANSC_HANDLE hRecvHandle = (ANSC_HANDLE )NULL;
ULONG ulRecvSize = (ULONG )0;
ULONG ulLeftSize = (ULONG )AnscBdoGetBlockSize(pPayloadBdo);
ULONG ulCopySize = (ULONG )0;
ULONG ulServingT1 = 0;
ULONG ulServingT2 = 0;
ULONG ulWaitCount = 0;
/*
* Since the TLS handshake is done in the asynchronous task instead of the socket_recv task,
* it's possible that we start receiving TLS client data traffic (i.e. HTTP) before handshake
* task considers the handshaking is done. To accommodate such race condition, we need to sleep
* for a while if there's no buffer available.
*/
while ( !pMyObject->GetRecvBuffer
(
(ANSC_HANDLE)pMyObject,
&ulRecvSize
) )
{
AnscSleep(10);
ulWaitCount++;
if ( ulWaitCount > 30 ) /* sleep for 300ms */
{
returnStatus = ANSC_STATUS_INTERNAL_ERROR;
goto EXIT1;
}
}
while ( ulLeftSize > 0 )
{
pRecvBuffer =
pMyObject->GetRecvBuffer
(
(ANSC_HANDLE)pMyObject,
&ulRecvSize
);
if ( !pRecvBuffer )
{
returnStatus =
pWorker->Notify
(
pWorker->hWorkerContext,
(ANSC_HANDLE)pMyObject,
ANSC_DSTOWO_EVENT_RESOURCES,
(ANSC_HANDLE)NULL
);
break;
}
else if ( ulRecvSize == 0 )
{
pMyObject->ToClean((ANSC_HANDLE)pMyObject, TRUE, 2);
break;
}
if ( ulRecvSize >= ulLeftSize )
{
ulCopySize = ulLeftSize;
}
else
{
ulCopySize = ulRecvSize;
}
if ( TRUE )
{
AnscCopyMemory
(
pRecvBuffer,
AnscBdoGetBlock(pPayloadBdo),
ulCopySize
);
AnscBdoShrinkRight(pPayloadBdo, ulCopySize);
ulLeftSize -= ulCopySize;
}
pMyObject->RecvBytesCount += ulCopySize;
pMyObject->LastRecvAt = AnscGetTickInSecondsAbs();
/*
* We have successfully transferred the received data into the buffer supplied by the
* socket owener though may not use up the while buffer. Now is time to notify our loyal
* socket owner about this exciting event.
*/
ulServingT1 = AnscGetTickInMilliSeconds();
returnStatus =
pMyObject->Recv
(
(ANSC_HANDLE)pMyObject,
pRecvBuffer,
ulCopySize
);
ulServingT2 = AnscGetTickInMilliSeconds();
pEngine->AvgServingTime =
(pEngine->AvgServingTime == 0)? (ulServingT2 - ulServingT1) : ((ulServingT2 - ulServingT1) + pEngine->AvgServingTime * 7) / 8;
pEngine->TscCount++;
}
goto EXIT1;
/******************************************************************
GRACEFUL ROLLBACK PROCEDURES AND EXIT DOORS
******************************************************************/
EXIT1:
if ( pPayloadBdo )
{
AnscFreeBdo((ANSC_HANDLE)pPayloadBdo);
}
return returnStatus;
}
ANSC_STATUS
AnscSctoTsaRecvAppMessage
(
ANSC_HANDLE hThisObject,
ANSC_HANDLE hMessageBdo
)
{
ANSC_STATUS returnStatus = ANSC_STATUS_SUCCESS;
PANSC_SIMPLE_CLIENT_TCP_OBJECT pMyObject = (PANSC_SIMPLE_CLIENT_TCP_OBJECT)hThisObject;
PANSC_SCTO_WORKER_OBJECT pWorker = (PANSC_SCTO_WORKER_OBJECT )pMyObject->hWorker;
PANSC_BUFFER_DESCRIPTOR pPayloadBdo = (PANSC_BUFFER_DESCRIPTOR )hMessageBdo;
void* pRecvBuffer = (void* )NULL;
ANSC_HANDLE hRecvHandle = (ANSC_HANDLE )NULL;
ULONG ulRecvSize = (ULONG )0;
ULONG ulLeftSize = (ULONG )AnscBdoGetBlockSize(pPayloadBdo);
ULONG ulCopySize = (ULONG )0;
while ( ulLeftSize > 0 )
{
pRecvBuffer =
pMyObject->GetRecvBuffer
(
(ANSC_HANDLE)pMyObject,
&ulRecvSize
);
if ( !pRecvBuffer )
{
returnStatus =
pWorker->Notify
(
pWorker->hWorkerContext,
ANSC_SCTOWO_EVENT_RESOURCES,
(ANSC_HANDLE)NULL
);
break;
}
if ( ulRecvSize >= ulLeftSize )
{
ulCopySize = ulLeftSize;
}
else
{
ulCopySize = ulRecvSize;
}
if ( TRUE )
{
AnscCopyMemory
(
pRecvBuffer,
AnscBdoGetBlock(pPayloadBdo),
ulCopySize
);
AnscBdoShrinkRight(pPayloadBdo, ulCopySize);
ulLeftSize -= ulCopySize;
}
pMyObject->RecvBytesCount += ulCopySize;
pMyObject->LastRecvAt = AnscGetTickInSeconds();
returnStatus =
pMyObject->Recv
(
(ANSC_HANDLE)pMyObject,
pRecvBuffer,
ulCopySize
);
}
goto EXIT1;
/******************************************************************
GRACEFUL ROLLBACK PROCEDURES AND EXIT DOORS
******************************************************************/
EXIT1:
if ( pPayloadBdo )
{
AnscFreeBdo((ANSC_HANDLE)pPayloadBdo);
}
return returnStatus;
}
