EAL.zip_VC扩展库

// WinSocket.cpp: implementation of the CWinSocket class. // ////////////////////////////////////////////////////////////////////// #include "stdafx.h" #include "WinSocket.h" #ifdef _DEBUG #undef THIS_FILE static char THIS_FILE[]=__FILE__; #endif ////////////////////////////////////////////////////////////////////// // Construction/Destruction ////////////////////////////////////////////////////////////////////// #pragma comment(lib, "ws2_32.lib") // BOOL InitSocket(WSADATA* lpwsaData /* = NULL */ ) { // initialize Winsock library WSADATA wsaData; if (lpwsaData == NULL) lpwsaData = &wsaData; WORD wVersionRequested = MAKEWORD(1, 1); int nResult = WSAStartup(wVersionRequested, lpwsaData); if (nResult != 0) return FALSE; if (LOBYTE(lpwsaData->wVersion) != 1 || HIBYTE(lpwsaData->wVersion) != 1) { WSACleanup(); return FALSE; } return TRUE ; } void UnInitSocket() { // 清除事件侦听池。 CASYNCSocket::s_SockEventListenerPool.CleanPool() ; // 反初始化。 WSACleanup() ; } BOOL GetSOCKADDR(SOCKADDR_IN & addr, int port, const char * pHostAddr/* = NULL */ ) { memset(&addr,0,sizeof(SOCKADDR_IN)); addr.sin_family = AF_INET; addr.sin_port = htons((unsigned short)port); // 转换IP地址， 如果addr指向主机名，则进而得到其IP 。 if (pHostAddr == NULL){ addr.sin_addr.s_addr = htonl(INADDR_ANY); } else{ DWORD lResult = inet_addr(pHostAddr); if (lResult == INADDR_NONE){ LPHOSTENT lphost; lphost = gethostbyname(pHostAddr); if (lphost != NULL){ lResult = ((LPIN_ADDR)lphost->h_addr)->s_addr; } else{ WSASetLastError(WSAEINVAL); return FALSE; } } addr.sin_addr.s_addr = lResult; } // return TRUE ; } BOOL GetIPFromAddr(const char * pHostAddr, char * pHostIP) { if (pHostAddr == NULL || *pHostAddr == '\0' || inet_addr(pHostAddr) == INADDR_NONE){ struct hostent * pHost; pHost = gethostbyname(pHostAddr); if(pHost!= NULL && pHost->h_addr_list[0]!= NULL){ ::memcpy(pHostIP, inet_ntoa (*(struct in_addr *)pHost->h_addr_list[0]), 16) ; return TRUE ; } else{ return FALSE ; } } else{ ::strcpy(pHostIP, pHostAddr) ; return TRUE ; } } // CWinSocket::CWinSocket() { m_hSocket = INVALID_SOCKET ; } CWinSocket::CWinSocket(SOCKET hSock) { Attach(hSock) ; } CWinSocket::~CWinSocket() { Close() ; } BOOL CWinSocket::Attach(SOCKET hSocket) { // 如果已连接了SOCKET，则关闭。 if(IS_SOCKET(m_hSocket)) Close() ; // 设置新的SOCKET。 m_hSocket = hSocket ; return TRUE ; } SOCKET CWinSocket::Detach() { SOCKET s = m_hSocket ; m_hSocket = INVALID_SOCKET ; return s ; } BOOL CWinSocket::Create(int nSocketType , int port/* = 0 */ , const char * ip/* = NULL */) { Close() ; m_hSocket = socket(AF_INET,nSocketType,0); if (m_hSocket != INVALID_SOCKET) { if (Bind(port, ip)) return TRUE; int nResult = GetLastError(); Close(); WSASetLastError(nResult); } return FALSE; } BOOL CWinSocket::GetRemoteAddr(char * ip , int * port ) { SOCKADDR_IN sockAddr; memset(&sockAddr, 0, sizeof(sockAddr)); // int nSockAddrLen = sizeof(sockAddr); int iRet = getpeername(m_hSocket, (SOCKADDR*)&sockAddr, &nSockAddrLen); if (SOCKET_ERROR != iRet) { *port = ntohs(sockAddr.sin_port) ; ::strcpy(ip,inet_ntoa(sockAddr.sin_addr)) ; } return iRet != SOCKET_ERROR; } BOOL CWinSocket::GetLocalAddr (char * ip , int * port ) { SOCKADDR_IN sockAddr; memset(&sockAddr, 0, sizeof(sockAddr)); // int nSockAddrLen = sizeof(sockAddr); int iRet = getsockname(m_hSocket, (SOCKADDR*)&sockAddr, &nSockAddrLen); if (SOCKET_ERROR != iRet) { *port = ntohs(sockAddr.sin_port) ; ::strcpy(ip,inet_ntoa(sockAddr.sin_addr)) ; } return iRet != SOCKET_ERROR ; } /*************************************************************************** 名称：TestRecv 描述：处理连接请求 输入：dwTimeoutms 超时时间，单位（毫秒） 输出：无 返回：1、当SOCKET还没有创建时，返回SOCKET_ERROR 2、当SOCKET还没有建立连接时，返回SOCKET_TIMEOUT 3、当SOCKET已创建并成功建立了连接时， 如果有数据到达，则返回数据长度（此时不管中途对方是否已断开连接）， 否则返回SOCKET_TIMEOUT 4、如果在测试期间SOCKET被关闭，则返回SOCKET_ERROR 5、如果在正常连接的情况下，对方断开了连接， 在SOCKET被关闭之前，如果此时缓冲区中的数据没有被读完，则返回数据长度，否则返回SOCKET_ERROR ； 在SOCKET被关闭之后返回SOCKET_ERROR ***************************************************************************/ int CWinSocket::TestRecv(DWORD dwTimeoutms/* = -1 */ ) { DWORD s = dwTimeoutms/1000 ; DWORD ms = dwTimeoutms%1000 ; struct timeval tvRecv = {s, ms*1000}, tvExp = {0,0}; // int iRet ; fd_set fdRecv; FD_ZERO(&fdRecv); FD_SET(m_hSocket,&fdRecv); iRet = select(1,&fdRecv,NULL,NULL,&tvRecv); switch(iRet) { case 0 : return SOCKET_TIMEOUT ; break; case 1:{ int dataLen = GetDataLength() ; if(dataLen <= 0) return SOCKET_ERROR ; else return dataLen ; } break; } // return SOCKET_ERROR ; } /*************************************************************************** 名称：TestSend 描述：处理连接请求 输入：dwTimeoutms 超时时间，单位（毫秒） 输出：无 返回：1、当SOCKET还没有创建时，返回SOCKET_ERROR 2、当SOCKET还没有建立连接时，返回SOCKET_TIMEOUT 3、当SOCKET已创建并成功建立了连接时， 如果可以发送数据，则返回SOCKET_SUCCESS，否则返回SOCKET_TIMEOUT 4、如果在测试期间SOCKET被关闭，则返回SOCKET_ERROR 5、如果在正常连接的情况下，对方断开了连接， 在SOCKET没有被关闭之前返回SOCKET_SUCCESS（但此时调用Send()函数将返回-1，即发送失败）， 在SOCKET被关闭之后返回SOCKET_ERROR ***************************************************************************/ int CWinSocket::TestSend(DWORD dwTimeoutms/* = -1 */ ) { DWORD s = dwTimeoutms/1000 ; DWORD ms = dwTimeoutms%1000 ; struct timeval tvSend = {s,ms*1000}, tvExp = {0,0}; // int iRet ; fd_set fdSend; FD_ZERO(&fdSend); FD_SET(m_hSocket,&fdSend); iRet = select(1,NULL,&fdSend,NULL,&tvSend); switch(iRet) { case 0 : return SOCKET_TIMEOUT ; break; case 1: return SOCKET_SUCCESS ; break; } return SOCKET_ERROR ; } int CWinSocket::RecvFrom(char * buf, int len, char * ip/* = NULL */ , int * port/* = NULL */, int flags/* = 0 */ ) { SOCKADDR_IN sockAddr; memset(&sockAddr, 0, sizeof(sockAddr)); // int nSockAddrLen = sizeof(sockAddr); int nResult = recvfrom(m_hSocket, buf, len, flags ,(SOCKADDR*)&sockAddr, &nSockAddrLen); if(nResult != SOCKET_ERROR) { if(port != NULL) *port = ntohs(sockAddr.sin_port) ; if(ip != NULL) ::strcpy(ip,inet_ntoa(sockAddr.sin_addr)) ; } return nResult; } int CWinSocket::SendTo (const char * buf, int len, const char * ip, int port, int flags/* = 0 */ ) { SOCKADDR_IN sockAddr; if(!GetSOCKADDR(sockAddr, port, ip)) return SOCKET_ERROR ; // return sendto(m_hSocket, buf, len, flags , (SOCKADDR*)&sockAddr, sizeof(sockAddr)); } // int CWinSocket::RecvWantEX(char * buf, int &len, DWORD dwTimeoutms /*= -1*/, int flags /*= 0*/) { int curPos = 0 ; int iRet ; DWORD dwTime = 0 ; while(curPos < len){ iRet = TestRecv(10) ; if(iRet > 0){ // 有数据到达。 int recvLen = Recv(buf+curPos, len-curPos) ; if(recvLen < 0){ len = curPos ; return SOCKET_ERROR ; } curPos += recvLen ; } else if(iRet == SOCKET_TIMEOUT){ dwTime += 10 ; if(dwTime >= dwTimeoutms){ len = curPos ; return SOCKET_TIMEOUT ; } } else { len = curPos ; return SOCKET_ERROR ; } } len = curPos ; return SOCKET_SUCCESS ; } int CWinSocket::SendWantEX(const char * buf, int &len, DWORD dwTimeoutms /*= -1 */, int flags /*= 0*/) { int curPos = 0 ; int iRet ; DWORD dwTime = 0 ; while(curPos < len){ iRet = TestSend(10) ; if(iRet == SOCKET_SUCCESS){ int sendLen = len-curPos > 1400 ? 1400 :