tcp_http.zip_C Tcp_http库_tcp

#include <win_tcpsocket.h> #ifdef _DEBUG #include <stdio.h> #endif DWORD CWinTcpSocket::m_dwRefCount = 0; CWinTcpSocket::CWinTcpSocket( void ) { WSADATA wd; if( ++m_dwRefCount == 1 ) { ::WSAStartup( 0x0101, &wd ); } } CWinTcpSocket::~CWinTcpSocket( void ) { Close(); if( --m_dwRefCount == 0 ) { ::WSACleanup(); } } bool CWinTcpSocket::Create( int af ) { m_hSocket = ::socket( af, SOCK_STREAM, IPPROTO_TCP ); if( m_hSocket == INVALID_SOCKET ) { #ifdef _DEBUG ::MessageBox( NULL, "Socket creation failed - why?!", NULL, MB_OK | MB_ICONERROR ); #endif return false; } else { return true; } } bool CWinTcpSocket::Connect( const char *pszHost, int nPort ) { unsigned long ulAddr = 0; hostent *pEnt = ::gethostbyname( pszHost ); SOCKADDR_IN addr; if( pEnt == 0 ) { ulAddr = ::inet_addr( pszHost ); if( ulAddr == INADDR_NONE ) { #ifdef _DEBUG ::MessageBox( NULL, "Invalid address!", NULL, MB_OK | MB_ICONERROR ); #endif return false; } else { addr.sin_family = AF_INET; } } else { memcpy( &ulAddr, pEnt->h_addr_list[0], sizeof( long ) ); addr.sin_family = pEnt->h_addrtype; } addr.sin_addr.s_addr = ulAddr; addr.sin_port = ::htons( nPort ); memset( addr.sin_zero, 0, sizeof( addr.sin_zero ) ); if( ::connect( m_hSocket, (const sockaddr *)&addr, sizeof( SOCKADDR_IN ) ) == SOCKET_ERROR ) { #ifdef _DEBUG char szBuf[64]; sprintf( szBuf, "Unable to connect: %X / %d", ::WSAGetLastError(), ::WSAGetLastError() ); ::MessageBox( NULL, szBuf, NULL, MB_OK | MB_ICONERROR ); #endif return false; } else { return true; } } bool CWinTcpSocket::Bind( int nLocalPort ) { SOCKADDR_IN addr; addr.sin_addr.s_addr = INADDR_ANY; addr.sin_family = AF_INET; addr.sin_port = ::htons( nLocalPort ); memset( addr.sin_zero, 0, sizeof( addr.sin_zero ) ); if( ::bind( m_hSocket, (const sockaddr *)&addr, sizeof( SOCKADDR_IN ) ) == SOCKET_ERROR ) { return false; } else { return true; } } bool CWinTcpSocket::Accept( CWinTcpSocket *pSocket ) { if( pSocket == 0 ) { return false; } SOCKADDR_IN addr; int len = sizeof( SOCKADDR_IN ); memset( &addr, 0, sizeof( SOCKADDR_IN ) ); pSocket->m_hSocket = ::accept( m_hSocket, (sockaddr*)&addr, &len ); if( pSocket->m_hSocket == INVALID_SOCKET ) { return false; } else { return true; } } bool CWinTcpSocket::Listen( int nBackLog ) { if( ::listen( m_hSocket, nBackLog ) == SOCKET_ERROR ) { return false; } else { return true; } } int CWinTcpSocket::Send( const void *pData, int nDataLen, int nFlags ) { return ::send( m_hSocket, (const char *)pData, nDataLen, nFlags ); } int CWinTcpSocket::SendText( const char *pszText ) { return Send( pszText, strlen( pszText ) ); } int CWinTcpSocket::Recv( void *pData, int nDataLen, int nFlags ) { return ::recv( m_hSocket, (char *)pData, nDataLen, nFlags ); } int CWinTcpSocket::RecvLine( char *pszBuf, int nLen, bool bEcho ) { int nCount = 0; int nRdLen; char ch = 0; while( ch != '\n' && nCount < nLen ) { nRdLen = Recv( &ch, 1 ); if( nRdLen == 0 || nRdLen == SOCKET_ERROR ) { nCount = 0; break; } if( ch != '\n' && ch != '\r' ) { pszBuf[nCount] = ch; nCount++; } if( bEcho ) { Send( &ch, 1 ); } } if( nCount != 0 ) { pszBuf[nCount] = 0; } return nCount ? nCount : nRdLen; } bool CWinTcpSocket::Shutdown( int nHow ) { return ::shutdown( m_hSocket, nHow ) == SOCKET_ERROR ? false : true; } bool CWinTcpSocket::Close( void ) { return ::closesocket( m_hSocket ) == SOCKET_ERROR ? false : true; }