winiosocketex.cpp

五行MMORPG引擎系统V1.0

// SocketModel.cpp.
//
//////////////////////////////////////////////////////////////////////

#include <winsock2.h>
#include "server/net/IOSocketEx.h"
#include "platform/platformAssert.h"

#pragma comment(lib, "ws2_32.lib") 

namespace RPGServer
{

IOSocketEx::IOSocketEx()
{
	//::InitializeCriticalSection(&m_csErrorNo);
}

IOSocketEx::~IOSocketEx()
{
	//::DeleteCriticalSection(&m_csErrorNo);
}



//释放Winsock2动态连接库
void IOSocketEx::CleanupLibrary(void)
{
    WSACleanup();
}


//注册WINSOCK2.2DLL
int IOSocketEx::InitLibrary(void)
{
	WSADATA         WSD;
	WORD wVersionRequired = MAKEWORD( _SOCKET_MAJOR_VERSION, _SOCKET_MINOR_VERSION );
	dMemset(&WSD,sizeof(WSADATA),0);

	int nErrorNo = WSAStartup(wVersionRequired, &WSD);
	if ( SOCKET_SUCCESS != nErrorNo )
	{
		//SetLastError( nErrorNo );
		return ( SOCKET_ERROR );
	}

	if ( LOBYTE( WSD.wVersion ) != _SOCKET_MINOR_VERSION ||
		 HIBYTE( WSD.wVersion ) != _SOCKET_MAJOR_VERSION ) 
	{
		WSACleanup( );
		//SetLastError( WSAVERNOTSUPPORTED );
		return (SOCKET_ERROR); 
	}
	//成功初始化
	return (SOCKET_SUCCESS);
}



// 设置套接字是否为阻塞的
//入口:套接字,是否需要阻塞的
//出口:如果正确那么返回0，错误返回-1
int IOSocketEx::BlockSocket(SLSocket hSocket, bool bBlock/*false*/)
{
	u_long IoctlLong = (bBlock) ? 0 : 1;
	if (ioctlsocket(hSocket, FIONBIO, &IoctlLong) == SOCKET_ERROR)
	{
		SetLastError( WSAGetLastError() );
		return (SOCKET_ERROR);
    }
	return (SOCKET_SUCCESS);
}

//设置套接字属性
//入口:套接字
//出口:如果正确那么返回0，错误返回-1
int IOSocketEx::SetSocketOption(SLSocket hSocket)
{
    int nActivate = 1;
	//允许地址重用
    if (setsockopt(hSocket, SOL_SOCKET, SO_REUSEADDR, (const char *) &nActivate,
		sizeof(nActivate)) == SOCKET_ERROR )
    {
        SetLastError( WSAGetLastError() );
        return (SOCKET_ERROR);//return (-1)
    }
	return (SOCKET_SUCCESS);
}


//接收所有数据，注意在这个函数调用之前必须确认是否有接收消息到来
//入口:套接字,数据缓冲区,缓冲区大小
//出口:如果正确那么返回接收的字节数量，错误返回错误代码
int IOSocketEx::RecvLL(SLSocket hSocket, char *pszBuffer, int nBufferSize)
{
	DWORD		dwRtxBytes = 0,
				dwRtxFlags = 0;
	WSABUF	WSABuff;
   
	//清空缓冲
	dMemset(&WSABuff,sizeof(WSABUF), 0);

	WSABuff.len = nBufferSize;
	WSABuff.buf = pszBuffer;

	int nRet = WSARecv(hSocket, &WSABuff, 1, &dwRtxBytes, &dwRtxFlags,NULL, NULL) ;
	//如果正确就返回本次接收的字节个数，如果错误返回错误号码(负数)
	return (nRet == SOCKET_SUCCESS) ? (int) dwRtxBytes : -WSAGetLastError();
}



int IOSocketEx::RecvData_Block(SLSocket hSocket, char *pszBuffer, int nBufferSize, U32 dwTimeout)
{
	AssertWarn(hSocket,"Socket句柄为NULL")
	if(hSocket==NULL)
		return ( SOCKET_ERROR );
	FD_SET fd = {1, hSocket};
	TIMEVAL tv = {dwTimeout, 0};
	int nBytesReceived=0;
	if(select(0, &fd, NULL, NULL, &tv) == 0) 
		goto CLEAR;
	if((nBytesReceived = recv(hSocket, pszBuffer, nBufferSize, 0)) == SOCKET_ERROR)
		goto CLEAR;
	return nBytesReceived;

CLEAR:
	SetLastError(WSAGetLastError());//超时
	return(SOCKET_ERROR);
}

// 接收数据(阻塞直至收到数据为止)
int IOSocketEx::RecvData_Event(SLSocket hSocket, char *pszBuffer, int nBufferSize, U32 dwTimeout)
{
	HANDLE hReadEvent = CreateEvent(NULL, false, false, NULL);
	if (hReadEvent == NULL)
	{
		SetLastError( (int)GetLastError() );
		return ( SOCKET_ERROR );
	}

	int		nRecvBytes = 0;
	U32	dwWaitResult ;
	for (;;)
	{
		// 注册FD_READ | FD_CLOSE 事件 
		// (因为可能在等待FD_READ事件中，对方关闭套接字，所以要关注FD_CLOSE)
		if( WSAEventSelect(hSocket, (WSAEVENT) hReadEvent, FD_READ | FD_CLOSE) 	== SOCKET_ERROR)
		{
			CloseHandle(hReadEvent);
			SetLastError( WSAGetLastError() );
			return (SOCKET_ERROR);
		}
	
		// 等等FD_READ | FD_CLOSE事件的发生
		dwWaitResult = WSAWaitForMultipleEvents(1, &hReadEvent, TRUE,dwTimeout, TRUE);

		if (dwWaitResult != WSA_WAIT_EVENT_0)
		{
			// 清除事件
			WSAEventSelect(hSocket, (WSAEVENT) hReadEvent, 0);
			CloseHandle(hReadEvent);
			SetLastError( WSAGetLastError() );
			return (SOCKET_ERROR);
		}
		
		////////////////////////////////////////////////////////////// 
		///	注意：即使 dwWaitResult == WSA_WAIT_EVENT0 ，也应该 
		///			进一步检查网络是否发生错误
		///////////////////////////////////////////////////////////////
		WSANETWORKEVENTS NetEvent;
		if(WSAEnumNetworkEvents(hSocket,(WSAEVENT)hReadEvent,&NetEvent) == SOCKET_ERROR)
		{
			// 清除事件
			WSAEventSelect(hSocket, (WSAEVENT) hReadEvent, 0);
			CloseHandle(hReadEvent);
			SetLastError( WSAGetLastError() );
			return (SOCKET_ERROR);
		}
		//判断发生了什么事件 FD_READ 或 FD_CLOSE	
		if( ( NetEvent.lNetworkEvents == FD_CLOSE ) ||
			 ( NetEvent.lNetworkEvents	== FD_READ && 
			   NetEvent.iErrorCode[FD_READ_BIT] !=0 ) )	// 发生错误
		{
			// 清除事件
			WSAEventSelect(hSocket, (WSAEVENT) hReadEvent, 0);
			CloseHandle(hReadEvent);
			SetLastError(WSAGetLastError() );
			return (SOCKET_ERROR);
		}
		////////////////////////////////////////////////////////////////
		// 清除事件
		WSAEventSelect(hSocket, (WSAEVENT) hReadEvent, 0);
	
		// 接收数据
		if ((nRecvBytes = RecvLL(hSocket, pszBuffer, nBufferSize)) >= 0)
			break;	// 跳出循环

		//Recv返回的是错误代码的负数，所以需要调转过来
		int nErrorCode = -nRecvBytes;

		if ( nErrorCode != WSAEWOULDBLOCK )	//太多的未完成重叠操作
		{
			CloseHandle(hReadEvent);
			SetLastError( nErrorCode );
			return (SOCKET_ERROR);
		}
		//阻塞住了
		Sleep(_BLOCKED_SNDRCV_SLEEP);
    }
    CloseHandle(hReadEvent);
    return (nRecvBytes);
}


//发送数据，阻塞
//入口:套接字，发送的字串,字串长度，超时值
int IOSocketEx::Send_Block(SLSocket hSocket,char const * pszBuffer, int nBufferSize, U32 dwTimeout)
{
	AssertWarn(hSocket,"Socket句柄为NULL")

	if(hSocket==NULL||pszBuffer==NULL)
		return (SOCKET_ERROR);
	FD_SET fd = {1, hSocket};
	TIMEVAL tv = {dwTimeout, 0};
	int nBytesSent=0;
	if(select(0, NULL, &fd, NULL, &tv) == 0)
		goto CLEAR;//选择发送超时
	if((nBytesSent = send(hSocket, pszBuffer, nBufferSize, 0)) == SOCKET_ERROR) 
		goto CLEAR;//发送出错误
	return nBytesSent;

CLEAR:
	SetLastError(WSAGetLastError());//超时
	return(SOCKET_ERROR);
}


//发送全部缓冲区中数据，阻塞
//入口:套接字，发送的字串,字串长度，超时值
//出口:正确返回发送的字节数量，错误返回SOCKET_ERROR
int IOSocketEx::SendData_Block(SLSocket hSocket,char const * pszBuffer, 
		int nBufferSize, U32 dwTimeout)

{
	if(hSocket==NULL)
		return(SOCKET_ERROR);
	int nBytesSent = 0;
	int nBytesThisTime;
	const char* pszTemp = pszBuffer;
	do {
		nBytesThisTime = Send_Block(hSocket,pszTemp, nBufferSize-nBytesSent, dwTimeout);
		if(nBytesThisTime<0)
			return(SOCKET_ERROR);
		//如果一次没有发送成功
		nBytesSent += nBytesThisTime;
		//改变当前字符指针
		pszTemp += nBytesThisTime;
	} while(nBytesSent < nBufferSize);
	return nBytesSent;
}


//一次发送数据，但不一定全部都发送
//入口:套接字，接收方地址信息,地址结构，结构长度，缓冲区，缓冲区长度，超时
//出口:正确返回发送字节数量，错误返回SOCKET_ERROR
int IOSocketEx::SendTo_Block(SLSocket hSocket, sockaddr * pTo,
		int nAddrLen,char const * pszBuffer, int nBufferSize, U32 dwTimeout)
{
	if(hSocket==NULL||pszBuffer==NULL)
		return SOCKET_ERROR;

	FD_SET fd = {1, hSocket};
	TIMEVAL tv = {dwTimeout, 0};
	int nBytesSent=0;
	if(select(0, NULL, &fd, NULL, &tv) == 0) 
		goto CLEAR;
	nBytesSent = sendto(hSocket, pszBuffer, nBufferSize, 0, pTo, nAddrLen);
	if(nBytesSent == SOCKET_ERROR)
		goto CLEAR;
	return nBytesSent;

CLEAR:
	SetLastError(WSAGetLastError());//超时
	return(SOCKET_ERROR);
}