Home

nevrax.com

Nevrax.org News Mailing-list Documentation CVS Bugs License

Documentation                       Search   sock.cpp Go to the documentation of this file. /* Copyright, 2000 Nevrax Ltd. * * This file is part of NEVRAX NEL. * NEVRAX NEL is free software; you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation; either version 2, or (at your option) * any later version. * NEVRAX NEL is distributed in the hope that it will be useful, but * WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * General Public License for more details. * You should have received a copy of the GNU General Public License * along with NEVRAX NEL; see the file COPYING. If not, write to the * Free Software Foundation, Inc., 59 Temple Place - Suite 330, Boston, * MA 02111-1307, USA. */ #include "stdnet.h" #include "nel/net/sock.h" #include "nel/misc/time_nl.h" #ifdef NL_OS_WINDOWS # include <winsock2.h> //# include <windows.h> # define socklen_t int # define ERROR_NUM WSAGetLastError() # define ERROR_WOULDBLOCK WSAEWOULDBLOCK #elif defined NL_OS_UNIX # include <unistd.h> # include <sys/types.h> # include <sys/time.h> # include <sys/socket.h> # include <netinet/in.h> # include <netinet/tcp.h> # include <arpa/inet.h> # include <netdb.h> # include <fcntl.h> # include <errno.h> # define SOCKET_ERROR -1 # define INVALID_SOCKET -1 # define ERROR_NUM errno # define ERROR_WOULDBLOCK EWOULDBLOCK # define ERROR_MSG strerror(errno) typedef int SOCKET; #endif using namespace std; using namespace NLMISC; namespace NLNET { bool CSock::_Initialized = false; long CSock::_TimeoutS = 0; long CSock::_TimeoutMs = 0; /* * ESocket constructor */ ESocket::ESocket( const char *reason, bool systemerror, CInetAddress *addr ) { /*it doesnt work on linux, should do something more cool std::stringstream ss; ss << "Socket error: " << reason; if ( systemerror ) { ss << " (" << ERROR_NUM; #ifdef NL_OS_UNIX ss << ": " << ERROR_MSG; #endif ss << ") " << std::endl; } _Reason = ss.str(); */ _Reason = "Socket error: "; uint errornum = CSock::getLastError(); char str[256]; if ( addr != NULL ) { // Version with address smprintf( str, 256, reason, addr->asString().c_str() ); // reason *must* contain "%s" _Reason += str; } else { // Version without address _Reason += reason; } if ( systemerror ) { _Reason += " ("; smprintf( str, 256, "%d", errornum ); _Reason += str; if ( errornum != 0 ) { _Reason += ": "; _Reason += CSock::errorString( errornum ); } _Reason += ")"; } nlwarning( "Exception will be launched: %s", _Reason.c_str() ); } /* * Initializes the network engine if it is not already done (under Windows, calls WSAStartup()). */ void CSock::initNetwork() { if ( ! CSock::_Initialized ) { #ifdef NL_OS_WINDOWS WORD winsock_version = MAKEWORD( 2, 0 ); WSADATA wsaData; if ( WSAStartup( winsock_version, &wsaData ) != 0 ) { throw ESocket( "Winsock initialization failed" ); } #endif CSock::_Initialized = true; } } /* * Releases the network engine */ void CSock::releaseNetwork() { #ifdef NL_OS_WINDOWS WSACleanup(); #endif CSock::_Initialized = false; } /* Returns the code of the last error that has occured. * Note: This code is platform-dependant. On Unix, it is errno; on Windows it is the Winsock error code. * See also errorString() */ uint CSock::getLastError() { return (uint)ERROR_NUM; } /* * Returns a string explaining the network error (see getLastError()) */ std::string CSock::errorString( uint errorcode ) { #ifdef NL_OS_WINDOWS switch( errorcode ) { case WSAEINTR /*10004*/: return "Blocking operation interrupted"; case WSAEINVAL /*10022*/: return "Invalid socket (maybe not bound) or argument"; case WSAEMFILE /*10024*/: return "Too many open sockets"; case WSAENOTSOCK /*10038*/: return "Socket operation on nonsocket (maybe invalid select descriptor)"; case WSAEMSGSIZE /*10040*/: return "Message too long"; case WSAEADDRINUSE /*10048*/: return "Address already in use (is this service already running in this computer?)"; case WSAEADDRNOTAVAIL/*10049*/: return "Address not available"; case WSAENETDOWN /*10050*/: return "Network is down"; case WSAENETUNREACH /*10051*/: return "Network is unreachable"; case WSAECONNRESET /*10054*/: return "Connection reset by peer"; case WSAENOBUFS /*10055*/: return "No buffer space available; please close applications or reboot"; case WSAESHUTDOWN /*10058*/: return "Cannot send/receive after socket shutdown"; case WSAETIMEDOUT /*10060*/: return "Connection timed-out"; case WSAECONNREFUSED /*10061*/: return "Connection refused, the server may be offline"; case WSAEHOSTUNREACH /*10065*/: return "Remote host is unreachable"; case WSANOTINITIALISED /*093*/: return "'Windows Sockets' not initialized"; default: return ""; } #elif defined NL_OS_UNIX return std::string( strerror( errorcode ) ); #endif } /* * Constructor */ CSock::CSock( bool logging ) : _Sock( INVALID_SOCKET ), _Logging( logging ), _BytesReceived( 0 ), _BytesSent( 0 ), _NonBlocking( false ), _MaxReceiveTime( 0 ), _MaxSendTime( 0 ) { nlassert( CSock::_Initialized ); /*{ CSynchronized<bool>::CAccessor sync( &_SyncConnected ); sync.value() = false; }*/ _Connected = false; } /* * Construct a CSock object using an existing connected socket descriptor and its associated remote address */ CSock::CSock( SOCKET sock, const CInetAddress& remoteaddr ) : _Sock( sock ), _Logging( true ), _BytesReceived( 0 ), _BytesSent( 0 ), _RemoteAddr( remoteaddr ), _NonBlocking( false ), _MaxReceiveTime( 0 ), _MaxSendTime( 0 ) { nlassert( CSock::_Initialized ); /*{ CSynchronized<bool>::CAccessor sync( &_SyncConnected ); sync.value() = true; }*/ _Connected = true; // Check remote address if ( ! _RemoteAddr.isValid() ) { throw ESocket( "Could not init a socket object with an invalid address", false ); } // Get local socket name setLocalAddress(); } /* * Creates the socket and get a valid descriptor */ void CSock::createSocket( int type, int protocol ) { nlassert( _Sock == INVALID_SOCKET ); _Sock = socket( AF_INET, type, protocol ); // or IPPROTO_IP (=0) ? if ( _Sock == INVALID_SOCKET ) { throw ESocket( "Socket creation failed" ); } if ( _Logging ) { // nldebug( "LNETL0: Socket %d open (TCP)", _Sock ); } } /* * Closes the listening socket */ void CSock::close() { if ( _Logging ) { nldebug( "LNETL0: Socket %d closing for %s at %s", _Sock, _RemoteAddr.asString().c_str(), _LocalAddr.asString().c_str() ); } #ifdef NL_OS_WINDOWS closesocket( _Sock ); #elif defined NL_OS_UNIX ::close( _Sock ); #endif _Sock = INVALID_SOCKET; } /* * Destructor */ CSock::~CSock() { nlinfo( "Report for %s socket %s: Max send time: %u Max recv time: %u", _NonBlocking?"non-blocking":"blocking", remoteAddr().asString().c_str(), _MaxSendTime, _MaxReceiveTime ); nlinfo( "Max send time: %u", _MaxSendTime); if ( _Sock != INVALID_SOCKET ) { if ( _Logging ) { nldebug( "LNETL0: Socket %d closing for %s at %s", _Sock, _RemoteAddr.asString().c_str(), _LocalAddr.asString().c_str() ); } if ( connected() ) { #ifdef NL_OS_WINDOWS shutdown( _Sock, SD_BOTH ); } closesocket( _Sock ); #elif defined NL_OS_UNIX shutdown( _Sock, SHUT_RDWR ); } ::close( _Sock ); #endif _Sock = INVALID_SOCKET; } } /* * Connection */ void CSock::connect( const CInetAddress& addr ) { nldebug( "LNETL0: Socket %d connecting to %s...", _Sock, addr.asString().c_str() ); // Check address if ( ! addr.isValid() ) { throw ESocket( "Unable to connect to invalid address", false ); } #ifndef NL_OS_WINDOWS // Set Reuse Address On (does not work on Win98 and is useless on Win2000) int value = true; if ( setsockopt( _Sock, SOL_SOCKET, SO_REUSEADDR, &value, sizeof(value) ) == SOCKET_ERROR ) { throw ESocket( "ReuseAddr failed" ); } #endif // Connection (when _Sock is a datagram socket, connect establishes a default destination address) if ( ::connect( _Sock, (const sockaddr *)(addr.sockAddr()), sizeof(sockaddr_in) ) != 0 ) { /* if ( _Logging ) { #ifdef NL_OS_WINDOWS nldebug( "Impossible to connect socket %d to %s %s (%d)", _Sock, addr.hostName().c_str(), addr.asIPString().c_str(), ERROR_NUM ); #elif defined NL_OS_UNIX nldebug( "Impossible to connect socket %d to %s %s (%d:%s)", _Sock, addr.hostName().c_str(), addr.asIPString().c_str(), ERROR_NUM, strerror(ERROR_NUM) ); #endif } */ throw ESocketConnectionFailed( addr ); } setLocalAddress(); if ( _Logging ) { nldebug( "LNETL0: Socket %d connected to %s (local %s)", _Sock, addr.asString().c_str(), _LocalAddr.asString().c_str() ); } _RemoteAddr = addr; _BytesReceived = 0; _BytesSent = 0; /*CSynchronized<bool>::CAccessor sync( &_SyncConnected ); sync.value() = true;*/ _Connected = true; } /* * Checks if there is some data to receive */ bool CSock::dataAvailable() { fd_set fdset; FD_ZERO( &fdset ); FD_SET( _Sock, &fdset ); timeval tv; tv.tv_sec = CSock::_TimeoutS; tv.tv_usec = CSock::_TimeoutMs; // Test for message received. int res = select( _Sock+1, &fdset, NULL, NULL, &tv ); switch ( res ) { case 0 : return false; case -1 : throw ESocket( "CSock::dataAvailable(): select failed" ); return false; default : return true; } } /* * Sets the local address */ void CSock::setLocalAddress() { sockaddr saddr; socklen_t saddrlen = sizeof(saddr); if ( getsockname( _Sock, &saddr, &saddrlen ) != 0 ) { throw ESocket( "Unable to find local address" ); } _LocalAddr.setSockAddr( (const sockaddr_in *)&saddr ); } /* * Sends data, or returns false if it would block */ CSock::TSockResult CSock::send( const uint8 *buffer, uint32& len, bool throw_exception ) { uint32 realLen = len; TTicks before = CTime::getPerformanceTime(); len = ::send( _Sock, (const char*)buffer, len, 0 ); _MaxSendTime = max( (uint32)(CTime::ticksToSecond(CTime::getPerformanceTime()-before)*1000.0f), _MaxSendTime ); if ( _Logging ) { // nldebug ("LNETL0: CSock::send(): Sent %d bytes to %d res: %d (%d)", realLen, _Sock, len, ERROR_NUM); } if ( len == SOCKET_ERROR ) { if ( ERROR_NUM == ERROR_WOULDBLOCK ) { len = 0; return Ok; } if ( throw_exception ) { throw ESocket( "Unable to send data" ); } return Error; } _BytesSent += len; return Ok; } /* * Receives data */ CSock::TSockResult CSock::receive( uint8 *buffer, uint32& len, bool throw_exception ) { if ( _NonBlocking ) { // Receive incoming message (only the received part) uint32 realLen = len; TTicks before = CTime::getPerformanceTime(); len = ::recv( _Sock, (char*)buffer, len, 0 ); if ( _Logging ) { // nldebug ("LNETL0: CSock::receive(): NBM Received %d bytes to %d res: %d (%d)", realLen, _Sock, len, ERROR_NUM); } _MaxReceiveTime = max( (uint32)(CTime::ticksToSecond(CTime::getPerformanceTime()-before)*1000.0f), _MaxReceiveTime ); switch ( len ) { // Graceful disconnection case 0 : { /*{ CSynchronized<bool>::CAccessor sync( &_SyncConnected ); sync.value() = false; }*/ _Connected = false; if ( throw_exception ) { throw ESocketConnectionClosed(); } return CSock::ConnectionClosed; } // Socket error or call would block case SOCKET_ERROR : { len = 0; if ( ERROR_NUM == ERROR_WOULDBLOCK ) { // Call would block return CSock::WouldBlock; } else { // Socket error if ( throw_exception ) { throw ESocket( "Unable to receive data" ); } return CSock::Error; } } } } else // Blocking Mode { // Receive incoming message, waiting until a complete message has arrived uint total = 0; uint brecvd; uint32 realLen = len; while ( total < len ) { TTicks before = CTime::getPerformanceTime(); brecvd = ::recv( _Sock, (char*)(buffer+total), len-total, 0 ); // nldebug ("LNETL0: CSock::receive(): BM Received %d bytes to %d res: %d (%d)", realLen, _Sock, len, ERROR_NUM); _MaxReceiveTime = max( (uint32)(CTime::ticksToSecond(CTime::getPerformanceTime()-before)*1000.0f), _MaxReceiveTime ); switch ( brecvd ) { // Graceful disconnection case 0 : { /*{ CSynchronized<bool>::CAccessor sync( &_SyncConnected ); sync.value() = false; }*/ _Connected = false; if ( throw_exception ) { throw ESocketConnectionClosed(); } return CSock::ConnectionClosed; } // Socket error case SOCKET_ERROR : { if ( throw_exception ) { throw ESocket( "Unable to receive data" ); } return CSock::Error; } } total += brecvd; } } /*if ( _Logging ) { nldebug( "LNETL0: Socket %d received %d bytes", _Sock, len ); }*/ _BytesReceived += len; return CSock::Ok; } /* * Returns if the socket is connected */ bool CSock::connected() { /*bool b; { //nldebug( "LNETL0: CSock::connected-BEGIN (socket %u)", descriptor() ); CSynchronized<bool>::CAccessor sync( &_SyncConnected ); b = sync.value(); } //nldebug( "LNETL0: CSock::connected-END" ); return b;*/ return _Connected; } /* * Sets the socket in nonblocking mode */ void CSock::setNonBlockingMode ( bool bm ) { if ( _NonBlocking != bm ) { #ifdef NL_OS_WINDOWS u_long b = bm; if ( ioctlsocket( _Sock, FIONBIO, &b ) != 0 ) #else if ( fcntl( _Sock, F_SETFL, FNDELAY | fcntl( _Sock, F_GETFL, 0 ) ) == -1 ) #endif { throw ESocket( "Cannot set nonblocking mode" ); } _NonBlocking = bm; } } } // NLNET