ANSC_STATUS HttpSpoPrvwoNotify ( ANSC_HANDLE hThisObject, ANSC_HANDLE hSocket, ULONG ulEvent, ANSC_HANDLE hReserved ) { ANSC_STATUS returnStatus = ANSC_STATUS_SUCCESS; PHTTP_SIMPLE_PROXY_OBJECT pMyObject = (PHTTP_SIMPLE_PROXY_OBJECT )hThisObject; PHTTP_SIMPLE_PROXY_PROPERTY pProperty = (PHTTP_SIMPLE_PROXY_PROPERTY )&pMyObject->Property; PANSC_SIMPLE_PROXY_TCP_OBJECT pSimpleProxy = (PANSC_SIMPLE_PROXY_TCP_OBJECT )pMyObject->hSimpleProxy; PANSC_DAEMON_SOCKET_TCP_OBJECT pClientSocket = (PANSC_DAEMON_SOCKET_TCP_OBJECT)hSocket; PANSC_BROKER_SOCKET_TCP_OBJECT pServerSocket = (PANSC_BROKER_SOCKET_TCP_OBJECT)NULL; PANSC_BUFFER_DESCRIPTOR pBufferDesp = (PANSC_BUFFER_DESCRIPTOR )pClientSocket->GetBufferContext((ANSC_HANDLE)pClientSocket); PHTTP_PROXY_SESSION_OBJECT pSession = NULL; AnscTrace ( "Event = %d is indicated on client connection of %d.%d.%d.%d / TCP %d\n", ulEvent, pClientSocket->PeerAddress.Dot[0], pClientSocket->PeerAddress.Dot[1], pClientSocket->PeerAddress.Dot[2], pClientSocket->PeerAddress.Dot[3], pClientSocket->PeerPort ); switch ( ulEvent ) { case ANSC_SPTOWO_EVENT_SOCKET_ERROR : pClientSocket->ToClean((ANSC_HANDLE)pClientSocket, TRUE, HTTP_SPO_SOCKET_TTC); break; case ANSC_SPTOWO_EVENT_SOCKET_CLOSED : AnscAcquireLock(&pMyObject->SyncLock); pSession = (PHTTP_PROXY_SESSION_OBJECT)pClientSocket->GetClientContext((ANSC_HANDLE)pClientSocket); if ( !pSession ) { AnscReleaseLock(&pMyObject->SyncLock); pClientSocket->Finish((ANSC_HANDLE)pClientSocket); break; } else { pSession->AcquireAccess((ANSC_HANDLE)pSession); } AnscReleaseLock(&pMyObject->SyncLock); pBufferDesp->BlockSize += pClientSocket->RecvPacketSize; returnStatus = pClientSocket->SetBufferContext ( (ANSC_HANDLE)pClientSocket, NULL, 0, (ANSC_HANDLE)NULL ); returnStatus = pSession->FinishedByClient ( (ANSC_HANDLE)pSession, AnscBdoGetBlock (pBufferDesp), AnscBdoGetBlockSize(pBufferDesp), (ANSC_HANDLE)pBufferDesp ); pSession->ReleaseAccess((ANSC_HANDLE)pSession); pClientSocket->Finish((ANSC_HANDLE)pClientSocket); break; case ANSC_SPTOWO_EVENT_RESOURCES : break; case ANSC_SPTOWO_EVENT_TIME_OUT : break; default : break; } return ANSC_STATUS_SUCCESS; }
ULONG HttpSpoPrvwoQuery ( ANSC_HANDLE hThisObject, ANSC_HANDLE hSocket, PVOID buffer, ULONG ulSize, PANSC_HANDLE phQueryContext ) { ANSC_STATUS returnStatus = ANSC_STATUS_SUCCESS; PHTTP_SIMPLE_PROXY_OBJECT pMyObject = (PHTTP_SIMPLE_PROXY_OBJECT )hThisObject; PHTTP_SIMPLE_PROXY_PROPERTY pProperty = (PHTTP_SIMPLE_PROXY_PROPERTY )&pMyObject->Property; PANSC_SIMPLE_PROXY_TCP_OBJECT pSimpleProxy = (PANSC_SIMPLE_PROXY_TCP_OBJECT )pMyObject->hSimpleProxy; PANSC_DAEMON_SOCKET_TCP_OBJECT pClientSocket = (PANSC_DAEMON_SOCKET_TCP_OBJECT)hSocket; PANSC_BROKER_SOCKET_TCP_OBJECT pServerSocket = (PANSC_BROKER_SOCKET_TCP_OBJECT)NULL; PANSC_BUFFER_DESCRIPTOR pBufferDesp = (PANSC_BUFFER_DESCRIPTOR )pClientSocket->GetBufferContext((ANSC_HANDLE)pClientSocket); PHTTP_PROXY_SESSION_OBJECT pSession = NULL; ULONG ulSptoPmode = ANSC_SPTOWO_PMODE_PROCESS_SYNC; ULONG ulPsoQmode = HTTP_PSO_QMODE_PROCESS; AnscAcquireLock(&pMyObject->SyncLock); pSession = (PHTTP_PROXY_SESSION_OBJECT)pClientSocket->GetClientContext((ANSC_HANDLE)pClientSocket); if ( !pSession ) { AnscReleaseLock(&pMyObject->SyncLock); return ANSC_SPTOWO_PMODE_FINISH; } else { pSession->AcquireAccess((ANSC_HANDLE)pSession); } AnscReleaseLock(&pMyObject->SyncLock); pBufferDesp->BlockSize += ulSize; ulPsoQmode = pSession->QueryForClient ( (ANSC_HANDLE)pSession, buffer, ulSize, (ANSC_HANDLE)pBufferDesp ); switch ( ulPsoQmode ) { case HTTP_PSO_QMODE_COLLECT : pBufferDesp->BlockSize -= ulSize; ulSptoPmode = ANSC_SPTOWO_PMODE_COLLECT; break; case HTTP_PSO_QMODE_FORWARD : case HTTP_PSO_QMODE_PROCESS : ulSptoPmode = ANSC_SPTOWO_PMODE_PROCESS_SYNC; break; default : ulSptoPmode = ANSC_SPTOWO_PMODE_FINISH; break; } *phQueryContext = (ANSC_HANDLE)ulPsoQmode; pSession->ReleaseAccess((ANSC_HANDLE)pSession); return ulSptoPmode; }
ANSC_STATUS HttpSpoPrvwoProcessSync ( ANSC_HANDLE hThisObject, ANSC_HANDLE hSocket, PVOID buffer, ULONG ulSize, ANSC_HANDLE hQueryContext ) { ANSC_STATUS returnStatus = ANSC_STATUS_SUCCESS; PHTTP_SIMPLE_PROXY_OBJECT pMyObject = (PHTTP_SIMPLE_PROXY_OBJECT )hThisObject; PHTTP_SIMPLE_PROXY_PROPERTY pProperty = (PHTTP_SIMPLE_PROXY_PROPERTY )&pMyObject->Property; PANSC_SIMPLE_PROXY_TCP_OBJECT pSimpleProxy = (PANSC_SIMPLE_PROXY_TCP_OBJECT )pMyObject->hSimpleProxy; PHTTP_SBC_INTERFACE pSbcIf = (PHTTP_SBC_INTERFACE )pMyObject->hSbcIf; PANSC_DAEMON_SOCKET_TCP_OBJECT pClientSocket = (PANSC_DAEMON_SOCKET_TCP_OBJECT)hSocket; PANSC_BROKER_SOCKET_TCP_OBJECT pServerSocket = (PANSC_BROKER_SOCKET_TCP_OBJECT)NULL; PANSC_BUFFER_DESCRIPTOR pBufferDesp = (PANSC_BUFFER_DESCRIPTOR )pClientSocket->GetBufferContext((ANSC_HANDLE)pClientSocket); PHTTP_PROXY_SESSION_OBJECT pSession = NULL; ULONG ulPsoQmode = (ULONG)hQueryContext; BOOL bFinishSocket = FALSE; AnscAcquireLock(&pMyObject->SyncLock); pSession = (PHTTP_PROXY_SESSION_OBJECT)pClientSocket->GetClientContext((ANSC_HANDLE)pClientSocket); if ( !pSession ) { AnscReleaseLock(&pMyObject->SyncLock); pClientSocket->Finish((ANSC_HANDLE)pClientSocket); return ANSC_STATUS_SUCCESS; } else { pSession->AcquireAccess((ANSC_HANDLE)pSession); } AnscReleaseLock(&pMyObject->SyncLock); switch ( ulPsoQmode ) { case HTTP_PSO_QMODE_FORWARD : pServerSocket = (PANSC_BROKER_SOCKET_TCP_OBJECT)pSession->GetServerSocket((ANSC_HANDLE)pSession); if ( !pServerSocket) { bFinishSocket = TRUE; returnStatus = ANSC_STATUS_INTERNAL_ERROR; break; } else { returnStatus = pServerSocket->Send ( (ANSC_HANDLE)pServerSocket, buffer, ulSize, (ANSC_HANDLE)NULL ); } pBufferDesp->BlockSize = 0; break; case HTTP_PSO_QMODE_PROCESS : pClientSocket->SetBufferContext ( (ANSC_HANDLE)pClientSocket, NULL, 0, (ANSC_HANDLE)NULL ); returnStatus = pSession->RecvFromClient ( (ANSC_HANDLE)pSession, buffer, ulSize, (ANSC_HANDLE)pBufferDesp ); if ( pSession->GetSessionState((ANSC_HANDLE)pSession) == HTTP_PSO_STATE_FINISHED ) { bFinishSocket = TRUE; break; } break; default : returnStatus = ANSC_STATUS_UNAPPLICABLE; break; } pSession->ReleaseAccess((ANSC_HANDLE)pSession); if ( bFinishSocket ) { pClientSocket->Finish((ANSC_HANDLE)pClientSocket); } return returnStatus; }
ANSC_STATUS HttpSpoPrvwoRemove ( ANSC_HANDLE hThisObject, ANSC_HANDLE hSocket ) { ANSC_STATUS returnStatus = ANSC_STATUS_SUCCESS; PHTTP_SIMPLE_PROXY_OBJECT pMyObject = (PHTTP_SIMPLE_PROXY_OBJECT )hThisObject; PHTTP_SIMPLE_PROXY_PROPERTY pProperty = (PHTTP_SIMPLE_PROXY_PROPERTY )&pMyObject->Property; PANSC_SIMPLE_PROXY_TCP_OBJECT pSimpleProxy = (PANSC_SIMPLE_PROXY_TCP_OBJECT )pMyObject->hSimpleProxy; PHTTP_WAM_INTERFACE pWamIf = (PHTTP_WAM_INTERFACE )pMyObject->hWamIf; PANSC_DAEMON_SOCKET_TCP_OBJECT pClientSocket = (PANSC_DAEMON_SOCKET_TCP_OBJECT)hSocket; PANSC_BROKER_SOCKET_TCP_OBJECT pServerSocket = (PANSC_BROKER_SOCKET_TCP_OBJECT)NULL; PANSC_BUFFER_DESCRIPTOR pBufferDesp = (PANSC_BUFFER_DESCRIPTOR )pClientSocket->GetBufferContext((ANSC_HANDLE)pClientSocket); PHTTP_PROXY_SESSION_OBJECT pSession = NULL; AnscTrace ( "PrvwoRemove removes the client connection of %d.%d.%d.%d / TCP %d\n", pClientSocket->PeerAddress.Dot[0], pClientSocket->PeerAddress.Dot[1], pClientSocket->PeerAddress.Dot[2], pClientSocket->PeerAddress.Dot[3], pClientSocket->PeerPort ); AnscAcquireLock(&pMyObject->SyncLock); pSession = (PHTTP_PROXY_SESSION_OBJECT)pClientSocket->GetClientContext((ANSC_HANDLE)pClientSocket); if ( pSession ) { pSession->AcquireAccess((ANSC_HANDLE)pSession); pServerSocket = (PANSC_BROKER_SOCKET_TCP_OBJECT)pSession->GetServerSocket((ANSC_HANDLE)pSession); } else { pServerSocket = NULL; } if ( pServerSocket ) { pServerSocket->SetClientContext((ANSC_HANDLE)pServerSocket, (ANSC_HANDLE)NULL ); pServerSocket->ToClean ((ANSC_HANDLE)pServerSocket, TRUE, HTTP_SPO_SOCKET_TTC); } AnscReleaseLock(&pMyObject->SyncLock); /* * Just a couple regular clean-up tasks to do: * * (1) dis-associate the proxy session object from the client socket * (2) clean the foreign buffer associated with the client socket * (3) dis-associate the proxy session object from the server socket * * Note that we CANNOT clean the foreign buffer associated with the server socket because the * access to that buffer is NOT synchornized between the prv_recv() and pub_recv routines. It * SHOULD be done in the pub_remove() function. */ pClientSocket->SetClientContext((ANSC_HANDLE)pClientSocket, (ANSC_HANDLE)NULL); pClientSocket->SetBufferContext((ANSC_HANDLE)pClientSocket, NULL, 0, (ANSC_HANDLE)NULL); if ( pBufferDesp ) { AnscFreeBdo((ANSC_HANDLE)pBufferDesp); } if ( !pSession ) { return ANSC_STATUS_SUCCESS; } pSession->ReleaseAccess((ANSC_HANDLE)pSession); pSession->Close ((ANSC_HANDLE)pSession); pSession->Return ((ANSC_HANDLE)pSession); return ANSC_STATUS_SUCCESS; }