自学教程：C++ AddFreeBufferObj函数代码示例

void CTcpServer::HandleReceive(CONNID dwConnID, TSocketObj* pSocketObj, TBufferObj* pBufferObj){	if(m_bMarkSilence) pSocketObj->activeTime = ::TimeGetTime();	EnHandleResult hr = TriggerFireReceive(pSocketObj, pBufferObj);	if(hr == HR_OK || hr == HR_IGNORE)	{		if(ContinueReceive(pSocketObj, pBufferObj, hr))		{			{				CSpinLock locallock(pSocketObj->sgPause);				pSocketObj->recving = FALSE;			}			DoReceive(pSocketObj, pBufferObj);		}	}	if(hr == HR_CLOSED)	{		AddFreeBufferObj(pBufferObj);	}	else if(hr == HR_ERROR)	{		TRACE("<S-CNNID: %Iu> OnReceive() event return 'HR_ERROR', connection will be closed !/n", dwConnID);		AddFreeSocketObj(pSocketObj, SCF_ERROR, SO_RECEIVE, ENSURE_ERROR_CANCELLED);		AddFreeBufferObj(pBufferObj);	}}

void CTcpServer::HandleAccept(SOCKET soListen, TBufferObj* pBufferObj){	VERIFY(::PostIocpAccept(m_hCompletePort));	int iLocalSockaddrLen;	int iRemoteSockaddrLen;	HP_PSOCKADDR pLocalSockAddr;	HP_PSOCKADDR pRemoteSockAddr;	int iAddrLen = HP_SOCKADDR::AddrSize(m_usFamily) + 16;	m_pfnGetAcceptExSockaddrs							(								pBufferObj->buff.buf,								0,								iAddrLen,								iAddrLen,								(SOCKADDR**)&pLocalSockAddr,								&iLocalSockaddrLen,								(SOCKADDR**)&pRemoteSockAddr,								&iRemoteSockaddrLen							);	CONNID dwConnID = 0;	SOCKET socket	= pBufferObj->client;	if(!HasStarted() || !m_bfActiveSockets.AcquireLock(dwConnID))	{		::ManualCloseSocket(socket, SD_BOTH);		AddFreeBufferObj(pBufferObj);		return;	}	TSocketObj* pSocketObj = GetFreeSocketObj(dwConnID, socket);	pRemoteSockAddr->Copy(pSocketObj->remoteAddr);	AddClientSocketObj(dwConnID, pSocketObj);	::SSO_UpdateAcceptContext(socket, soListen);	::CreateIoCompletionPort((HANDLE)socket, m_hCompletePort, (ULONG_PTR)pSocketObj, 0);	if(TriggerFireAccept(pSocketObj) != HR_ERROR)		DoReceive(pSocketObj, pBufferObj);	else	{		AddFreeSocketObj(pSocketObj, SCF_NONE);		AddFreeBufferObj(pBufferObj);	}}

int CTcpAgent::SendDirect(TSocketObj* pSocketObj, const BYTE* pBuffer, int iLength){	int result	= NO_ERROR;	int iRemain	= iLength;	while(iRemain > 0)	{		int iBufferSize = min(iRemain, (int)m_dwSocketBufferSize);		TBufferObj* pBufferObj = GetFreeBufferObj(iBufferSize);		memcpy(pBufferObj->buff.buf, pBuffer, iBufferSize);		result = ::PostSend(pSocketObj, pBufferObj);		if(result != NO_ERROR)		{			AddFreeBufferObj(pBufferObj);			break;		}		iRemain -= iBufferSize;		pBuffer += iBufferSize;	}	if(result != NO_ERROR)		CheckError(pSocketObj->connID, SO_SEND, result);	return result;}

int CTcpAgent::SendItem(TSocketObj* pSocketObj){	int result = NO_ERROR;	while(pSocketObj->sndBuff.Size() > 0)	{		::InterlockedIncrement(&pSocketObj->sndCount);		TBufferObj* pBufferObj	= pSocketObj->sndBuff.PopFront();		int iBufferSize			= pBufferObj->buff.len;		ASSERT(iBufferSize > 0 && iBufferSize <= (int)m_dwSocketBufferSize);		pSocketObj->pending	   -= iBufferSize;		result = ::PostSendNotCheck(pSocketObj, pBufferObj);		if(result != NO_ERROR)		{			if(result != WSA_IO_PENDING)				AddFreeBufferObj(pBufferObj);			break;		}	}	return result;}

void CTcpAgent::HandleIo(CONNID dwConnID, TSocketObj* pSocketObj, TBufferObj* pBufferObj, DWORD dwBytes, DWORD dwErrorCode){	ASSERT(pBufferObj != nullptr);	ASSERT(pSocketObj != nullptr);	if(dwErrorCode != NO_ERROR)	{		HandleError(dwConnID, pBufferObj, dwErrorCode);		return;	}	if(dwBytes == 0 && pBufferObj->operation != SO_CONNECT)	{		AddFreeSocketObj(dwConnID, SCF_CLOSE);		AddFreeBufferObj(pBufferObj);		return;	}	pBufferObj->buff.len = dwBytes;	switch(pBufferObj->operation)	{	case SO_CONNECT:		HandleConnect(dwConnID, pSocketObj, pBufferObj);		break;	case SO_SEND:		HandleSend(dwConnID, pSocketObj, pBufferObj);		break;	case SO_RECEIVE:		HandleReceive(dwConnID, pSocketObj, pBufferObj);		break;	default:		ASSERT(FALSE);	}}

int CTcpServer::DoSend(CONNID dwConnID, TSocketObj* pSocketObj, const BYTE* pBuffer, int iLen){	int result	= NO_ERROR;	int iRemain	= iLen;	CCriSecLock locallock(pSocketObj->crisec);	while(iRemain > 0)	{		int iBufferSize = min(iRemain, (int)m_dwSocketBufferSize);		TBufferObj* pBufferObj = GetFreeBufferObj(iBufferSize);		memcpy(pBufferObj->buff.buf, pBuffer, iBufferSize);		result = ::PostSend(pSocketObj, pBufferObj);		if(result != NO_ERROR)		{			AddFreeBufferObj(pBufferObj);			break;		}		iRemain -= iBufferSize;		pBuffer += iBufferSize;	}	return result;}

EnHandleResult CTcpServer::TriggerFireSend(TSocketObj* pSocketObj, TBufferObj* pBufferObj){	EnHandleResult rs = (EnHandleResult)HR_CLOSED;	if(m_enOnSendSyncPolicy == OSSP_NONE)		rs = TRIGGER(FireSend(pSocketObj, (BYTE*)pBufferObj->buff.buf, pBufferObj->buff.len));	else	{		ASSERT(m_enOnSendSyncPolicy >= OSSP_CLOSE && m_enOnSendSyncPolicy <= OSSP_RECEIVE);		if(TSocketObj::IsValid(pSocketObj))		{			CCriSecLock locallock(m_enOnSendSyncPolicy == OSSP_CLOSE ? pSocketObj->csSend : pSocketObj->csRecv);			if(TSocketObj::IsValid(pSocketObj))			{				rs = TRIGGER(FireSend(pSocketObj, (BYTE*)pBufferObj->buff.buf, pBufferObj->buff.len));			}		}	}	if(rs == HR_ERROR)	{		TRACE("<S-CNNID: %Iu> OnSend() event should not return 'HR_ERROR' !!/n", pSocketObj->connID);		ASSERT(FALSE);	}	if(pBufferObj->ReleaseSendCounter() == 0)		AddFreeBufferObj(pBufferObj);	return rs;}

BOOL CTcpServer::ContinueReceive(TSocketObj* pSocketObj, TBufferObj* pBufferObj, EnHandleResult& hr){	int rs = NO_ERROR;	for(int i = 0; i < MAX_IOCP_CONTINUE_RECEIVE || MAX_IOCP_CONTINUE_RECEIVE < 0; i++)	{		if(hr != HR_OK && hr != HR_IGNORE)			break;		if(pBufferObj->buff.len != m_dwSocketBufferSize)			break;		pBufferObj->buff.len = m_dwSocketBufferSize;		rs =::NoBlockReceiveNotCheck(pBufferObj);		if(rs != NO_ERROR)			break;		hr = TriggerFireReceive(pSocketObj, pBufferObj);	}	if(rs != NO_ERROR && rs != WSAEWOULDBLOCK)	{		if(rs == WSAEDISCON)			AddFreeSocketObj(pSocketObj, SCF_CLOSE);		else			CheckError(pSocketObj, SO_RECEIVE, rs);				AddFreeBufferObj(pBufferObj);		return FALSE;	}	return TRUE;}

int CTcpServer::SendItem(TSocketObj* pSocketObj){	int result = NO_ERROR;	while(pSocketObj->sndBuff.Size() > 0)	{		TBufferObj* pBufferObj	= pSocketObj->sndBuff.PopFront();		int iBufferSize			= pBufferObj->buff.len;		ASSERT(iBufferSize > 0 && iBufferSize <= (int)m_dwSocketBufferSize);		pSocketObj->pending	   -= iBufferSize;		::InterlockedExchangeAdd(&pSocketObj->sndCount, iBufferSize);		result			= ::PostSendNotCheck(pSocketObj, pBufferObj);		LONG sndCounter	= pBufferObj->ReleaseSendCounter();		if(sndCounter == 0 || !IOCP_SUCCESS(result))			AddFreeBufferObj(pBufferObj);		if(result != NO_ERROR)			break;	}	return result;}

int CTcpServer::DoReceive(TSocketObj* pSocketObj, TBufferObj* pBufferObj){	int result		= NO_ERROR;	BOOL bNeedFree	= FALSE;	{		CSpinLock locallock(pSocketObj->sgPause);		if(pSocketObj->paused || pSocketObj->recving)			bNeedFree = TRUE;		else		{			pSocketObj->recving	 = TRUE;			pBufferObj->buff.len = m_dwSocketBufferSize;			result = ::PostReceive(pSocketObj, pBufferObj);		}	}	if(result != NO_ERROR)	{		CheckError(pSocketObj, SO_RECEIVE, result);		bNeedFree = TRUE;	}	if(bNeedFree) AddFreeBufferObj(pBufferObj);	return result;}

BOOL CTcpServer::DoAccept(){	BOOL isOK = FALSE;	if(HasStarted())	{		SOCKET		soClient	= socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);		TBufferObj*	pBufferObj	= GetFreeBufferObj();		ASSERT(soClient != INVALID_SOCKET);		isOK = (::PostAccept(m_pfnAcceptEx, m_soListen, soClient, pBufferObj) == NO_ERROR);		if(!isOK)		{			VERIFY(!HasStarted());			::ManualCloseSocket(soClient);			AddFreeBufferObj(pBufferObj);		}	}	if(!isOK)	{		::InterlockedDecrement(&m_iRemainAcceptSockets);		ASSERT(m_iRemainAcceptSockets >= 0);	}	return isOK;}

int CUdpServer::SendItem(TUdpSocketObj* pSocketObj){	int result = NO_ERROR;	while(pSocketObj->sndBuff.Size() > 0)	{		::InterlockedIncrement(&pSocketObj->sndCount);		TItemPtr itPtr(m_itPool, pSocketObj->sndBuff.PopFront());		int iBufferSize = itPtr->Size();		ASSERT(iBufferSize > 0 && iBufferSize <= (int)m_dwMaxDatagramSize);		pSocketObj->pending		 -= iBufferSize;		TUdpBufferObj* pBufferObj = GetFreeBufferObj(iBufferSize);		memcpy(&pBufferObj->remoteAddr, &pSocketObj->remoteAddr, sizeof(SOCKADDR_IN));		memcpy(pBufferObj->buff.buf, itPtr->Ptr(), iBufferSize);		int result = ::PostSendToNotCheck(m_soListen, pBufferObj);		if(result != NO_ERROR)		{			if(result != WSA_IO_PENDING)				AddFreeBufferObj(pBufferObj);;			break;		}	}	return result;}

void CIocpServer::HandleIo(TSocketObj* pSocketObj, TBufferObj* pBufferObj, DWORD dwBytes, DWORD dwErrorCode){	ASSERT(pBufferObj != NULL);	ASSERT(pSocketObj != NULL);	//判断是否已经标记失败	if(dwErrorCode != NO_ERROR)	{		if(pBufferObj->operation != SO_ACCEPT)		{			FireError(pSocketObj->connID, pBufferObj->operation, dwErrorCode);			AddFreeSocketObj(pSocketObj->connID);		}		else		{			DeleteAcceptSocket(pBufferObj->client, TRUE);		}		AddFreeBufferObj(pBufferObj);		return;	}	if(dwBytes == 0 && pBufferObj->operation != SO_ACCEPT)	{		FireClose(pSocketObj->connID);		AddFreeSocketObj(pSocketObj->connID);		AddFreeBufferObj(pBufferObj);		return;	}	pBufferObj->buff.len = dwBytes;	switch(pBufferObj->operation)	{	case SO_ACCEPT:		HandleAccept((SOCKET)pSocketObj, pBufferObj);		break;	case SO_SEND:		HandleSend(pSocketObj, pBufferObj);		break;	case SO_RECEIVE:		HandleReceive(pSocketObj, pBufferObj);		break;	default:		ASSERT(FALSE);	}}

void CUdpServer::HandleSend(CONNID dwConnID, TUdpBufferObj* pBufferObj){	BOOL bNeedFree = FALSE;	switch(m_enSendPolicy)	{	case SP_PACK:		{			TUdpSocketObj* pSocketObj = FindSocketObj(dwConnID);			if(TUdpSocketObj::IsValid(pSocketObj))			{				long sndCount = ::InterlockedDecrement(&pSocketObj->sndCount);				TriggerFireSend(dwConnID, pBufferObj);				if(sndCount == 0) DoSendPack(pSocketObj);			}			else				bNeedFree = TRUE;		}		break;	case SP_SAFE:		{			long sndCount				= -1;			TUdpSocketObj* pSocketObj	= FindSocketObj(dwConnID);			if(TUdpSocketObj::IsValid(pSocketObj))			{				sndCount = ::InterlockedDecrement(&pSocketObj->sndCount);				if(sndCount == 0 && !pSocketObj->smooth)				{					CCriSecLock locallock(pSocketObj->crisec);					if((sndCount = pSocketObj->sndCount) == 0)						pSocketObj->smooth = TRUE;				}				TriggerFireSend(dwConnID, pBufferObj);				if(sndCount == 0) DoSendSafe(pSocketObj);			}			else				bNeedFree = TRUE;		}		break;	case SP_DIRECT:		{			TriggerFireSend(dwConnID, pBufferObj);		}		break;	default:		ASSERT(FALSE);	}	if(bNeedFree) AddFreeBufferObj(pBufferObj);}

void CTcpServer::HandleReceive(CONNID dwConnID, TSocketObj* pSocketObj, TBufferObj* pBufferObj){	if(m_bMarkSilence) pSocketObj->activeTime = ::TimeGetTime();	EnHandleResult hr = TriggerFireReceive(pSocketObj, pBufferObj);	if(hr == HR_OK || hr == HR_IGNORE)		DoReceive(dwConnID, pSocketObj, pBufferObj);	else if(hr == HR_CLOSED)	{		AddFreeBufferObj(pBufferObj);	}	else	{		AddFreeSocketObj(pSocketObj, SCF_ERROR, SO_RECEIVE, ERROR_CANCELLED);		AddFreeBufferObj(pBufferObj);	}}

DWORD CTcpAgent::ConnectToServer(CONNID dwConnID, SOCKET& soClient, LPCTSTR pszRemoteAddress, USHORT usPort){	TCHAR szAddress[40];	int iAddressLen = sizeof(szAddress) / sizeof(TCHAR);	if(!::GetIPAddress(pszRemoteAddress, szAddress, iAddressLen))		return ERROR_INVALID_ADDRESS;	SOCKADDR_IN addr;	if(!::sockaddr_A_2_IN(AF_INET, szAddress, usPort, addr))		return ERROR_INVALID_ADDRESS;	TBufferObj* pBufferObj = GetFreeBufferObj();	TSocketObj* pSocketObj = GetFreeSocketObj(dwConnID, soClient);	memcpy(&pSocketObj->remoteAddr, &addr, sizeof(SOCKADDR_IN));	AddClientSocketObj(dwConnID, pSocketObj);	DWORD result = NO_ERROR;	if(m_bAsyncConnect)	{		if(::CreateIoCompletionPort((HANDLE)soClient, m_hCompletePort, (ULONG_PTR)pSocketObj, 0))			result = ::PostConnect(m_pfnConnectEx, soClient, addr, pBufferObj);		else			result = ::GetLastError();	}	else	{		if(::connect(soClient, (SOCKADDR*)&addr, sizeof(SOCKADDR_IN)) != SOCKET_ERROR)		{			if(::CreateIoCompletionPort((HANDLE)soClient, m_hCompletePort, (ULONG_PTR)pSocketObj, 0))			{				BOOL bOnOff	= (m_dwKeepAliveTime > 0 && m_dwKeepAliveInterval > 0);				VERIFY(::SSO_KeepAliveVals(pSocketObj->socket, bOnOff, m_dwKeepAliveTime, m_dwKeepAliveInterval) == NO_ERROR);				if(FireConnect(dwConnID) != HR_ERROR)					result = DoReceive(dwConnID, pSocketObj, pBufferObj);				else					result = ERROR_FUNCTION_FAILED;			}			else				result = ::GetLastError();		}		else			result = ::WSAGetLastError();	}	if(result != NO_ERROR)	{		AddFreeSocketObj(dwConnID, SCF_NONE);		AddFreeBufferObj(pBufferObj);		soClient = INVALID_SOCKET;	}	return result;}

void CTcpServer::HandleReceive(CONNID dwConnID, TSocketObj* pSocketObj, TBufferObj* pBufferObj){	EnHandleResult hr = FireReceive(pSocketObj, (BYTE*)pBufferObj->buff.buf, pBufferObj->buff.len);	if(hr == HR_OK || hr == HR_IGNORE)		DoReceive(dwConnID, pSocketObj, pBufferObj);	else if(hr == HR_CLOSED)	{		ASSERT(m_enRecvPolicy == RP_SERIAL);		AddFreeBufferObj(pBufferObj);	}	else	{		TRACE("<S-CNNID: %Iu> OnReceive() event return 'HR_ERROR', connection will be closed !/n", dwConnID);		AddFreeSocketObj(pSocketObj, SCF_ERROR, SO_RECEIVE, ERROR_CANCELLED);		AddFreeBufferObj(pBufferObj);	}}

void CTcpAgent::HandleReceive(CONNID dwConnID, TSocketObj* pSocketObj, TBufferObj* pBufferObj){	if(m_bMarkSilence) pSocketObj->activeTime = ::TimeGetTime();	EnHandleResult hr = TriggerFireReceive(pSocketObj, pBufferObj);	if(hr == HR_OK || hr == HR_IGNORE)		DoReceive(dwConnID, pSocketObj, pBufferObj);	else if(hr == HR_CLOSED)	{		AddFreeBufferObj(pBufferObj);	}	else	{		TRACE("<A-CNNID: %Iu> OnReceive() event return 'HR_ERROR', connection will be closed !/n", dwConnID);		AddFreeSocketObj(pSocketObj, SCF_ERROR, SO_RECEIVE, ERROR_CANCELLED);		AddFreeBufferObj(pBufferObj);	}}

void CTcpAgent::TriggerFireSend(CONNID dwConnID, TBufferObj* pBufferObj){	if(FireSend(dwConnID, (BYTE*)pBufferObj->buff.buf, pBufferObj->buff.len) == HR_ERROR)	{		TRACE("<A-CNNID: %Iu> OnSend() event should not return 'HR_ERROR' !!/n", dwConnID);		ASSERT(FALSE);	}	AddFreeBufferObj(pBufferObj);}

void CUdpServer::HandleError(CONNID dwConnID, TUdpBufferObj* pBufferObj, DWORD dwErrorCode){	if(dwConnID != 0)		AddFreeSocketObj(dwConnID, SCF_ERROR, pBufferObj->operation, dwErrorCode);		if(pBufferObj->operation == SO_RECEIVE)		DoReceive(pBufferObj);	else		AddFreeBufferObj(pBufferObj);}

void CTcpServer::HandleSend(CONNID dwConnID, TSocketObj* pSocketObj, TBufferObj* pBufferObj){	if(FireSend(dwConnID, (BYTE*)pBufferObj->buff.buf, pBufferObj->buff.len) == ISocketListener::HR_ERROR)	{		TRACE("<S-CNNID: %Iu> OnSend() event should not return 'HR_ERROR' !!/n", dwConnID);		ASSERT(FALSE);	}	AddFreeBufferObj(pBufferObj);}

void CTcpAgent::HandleReceive(CONNID dwConnID, TSocketObj* pSocketObj, TBufferObj* pBufferObj){	if(FireReceive(dwConnID, (BYTE*)pBufferObj->buff.buf, pBufferObj->buff.len) != HR_ERROR)		DoReceive(dwConnID, pSocketObj, pBufferObj);	else	{		TRACE("<A-CNNID: %Iu> OnReceive() event return 'HR_ERROR', connection will be closed !/n", dwConnID);		AddFreeSocketObj(dwConnID, SCF_ERROR, SO_RECEIVE, ERROR_FUNCTION_FAILED);		AddFreeBufferObj(pBufferObj);	}}

int CTcpAgent::DoConnect(CONNID dwConnID, TSocketObj* pSocketObj, TBufferObj* pBufferObj){	int result = ::PostConnect(m_pfnConnectEx, pSocketObj->socket, pSocketObj->remoteAddr, pBufferObj);	if(result != NO_ERROR)	{		CheckError(pSocketObj, SO_CONNECT, result);		AddFreeBufferObj(pBufferObj);	}	return result;}

void CTcpAgent::HandleConnect(CONNID dwConnID, TSocketObj* pSocketObj, TBufferObj* pBufferObj){	::SSO_UpdateConnectContext(pBufferObj->client, 0);	if(TriggerFireConnect(pSocketObj) != HR_ERROR)		DoReceive(dwConnID, pSocketObj, pBufferObj);	else	{		AddFreeSocketObj(pSocketObj, SCF_NONE);		AddFreeBufferObj(pBufferObj);	}}

void CTcpServer::HandleError(CONNID dwConnID, TSocketObj* pSocketObj, TBufferObj* pBufferObj, DWORD dwErrorCode){	if(pBufferObj->operation != SO_ACCEPT)		CheckError(pSocketObj, pBufferObj->operation, dwErrorCode);	else	{		::ManualCloseSocket(pBufferObj->client);		VERIFY(::PostIocpAccept(m_hCompletePort));	}	AddFreeBufferObj(pBufferObj);}

int CIocpServer::DoSend(TSocketObj* pSocketObj, TBufferObj* pBufferObj){	int result = ::PostSend(pSocketObj, pBufferObj);	if(result != NO_ERROR)	{		FireError(pSocketObj->connID, SO_SEND, result);		AddFreeSocketObj(pSocketObj->connID);		AddFreeBufferObj(pBufferObj);	}	return result;}

void CTcpServer::HandleError(CONNID dwConnID, TBufferObj* pBufferObj, DWORD dwErrorCode){	if(pBufferObj->operation != SO_ACCEPT)		CheckError(dwConnID, pBufferObj->operation, dwErrorCode);	else	{		::ManualCloseSocket(pBufferObj->client);		::PostQueuedCompletionStatus(m_hCompletePort, IOCP_SI_ACCEPT, 0, nullptr);	}	AddFreeBufferObj(pBufferObj);}

int CTcpAgent::DoReceive(CONNID dwConnID, TSocketObj* pSocketObj, TBufferObj* pBufferObj){	pBufferObj->buff.len = m_dwSocketBufferSize;	int result			 = ::PostReceive(pSocketObj, pBufferObj);	if(result != NO_ERROR)	{		CheckError(dwConnID, SO_RECEIVE, result);		AddFreeBufferObj(pBufferObj);	}	return result;}