nevrax.org : docs

00001 
00007 /* Copyright, 2000 Nevrax Ltd.
00008  *
00009  * This file is part of NEVRAX NEL.
00010  * NEVRAX NEL is free software; you can redistribute it and/or modify
00011  * it under the terms of the GNU General Public License as published by
00012  * the Free Software Foundation; either version 2, or (at your option)
00013  * any later version.
00014 
00015  * NEVRAX NEL is distributed in the hope that it will be useful, but
00016  * WITHOUT ANY WARRANTY; without even the implied warranty of
00017  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
00018  * General Public License for more details.
00019 
00020  * You should have received a copy of the GNU General Public License
00021  * along with NEVRAX NEL; see the file COPYING. If not, write to the
00022  * Free Software Foundation, Inc., 59 Temple Place - Suite 330, Boston,
00023  * MA 02111-1307, USA.
00024  */
00025 
00026 #include "stdnet.h"
00027 
00028 #include "nel/net/tcp_sock.h"
00029 
00030 #ifdef NL_OS_WINDOWS
00031 #include <winsock2.h>
00032 #define socklen_t int
00033 #define ERROR_NUM WSAGetLastError()
00034 
00035 #elif defined NL_OS_UNIX
00036 #include <unistd.h>
00037 #include <sys/types.h>
00038 #include <sys/time.h>
00039 #include <sys/socket.h>
00040 #include <netinet/in.h>
00041 #include <netinet/tcp.h>
00042 #include <arpa/inet.h>
00043 #include <netdb.h>
00044 #include <errno.h>
00045 #define SOCKET_ERROR -1
00046 #define INVALID_SOCKET -1
00047 #define ERROR_NUM errno
00048 #define ERROR_MSG strerror(errno)
00049 typedef int SOCKET;
00050 
00051 #endif
00052 
00053 using namespace NLMISC;
00054 
00055 namespace NLNET {
00056 
00057 
00058 /*
00059  * Constructor
00060  */
00061 CTcpSock::CTcpSock( bool logging ) :
00062         CSock( logging )
00063 {}
00064 
00065 
00066 /*
00067  * Construct a CTcpSocket object using an already connected socket 
00068  */
00069 CTcpSock::CTcpSock( SOCKET sock, const CInetAddress& remoteaddr ) :
00070         CSock( sock, remoteaddr )
00071 {}
00072 
00073 
00074 /* Connection. You can reconnect a socket after being disconnected.
00075  * This method does not return a boolean, otherwise a programmer could ignore the result and no
00076  * exception would be thrown if connection fails :
00077  * - If addr is not valid, an exception ESocket is thrown
00078  * - If connect() fails for another reason, an exception ESocketConnectionFailed is thrown
00079  */
00080 void CTcpSock::connect( const CInetAddress& addr )
00081 {
00082         // Create a new socket
00083         if ( _Sock != INVALID_SOCKET )
00084         {
00085           if ( _Logging )
00086             {
00087 //              nldebug( "LNETL0: Closing socket %d before reconnecting", _Sock );
00088             }
00089           close();
00090         }       
00091         createSocket( SOCK_STREAM, IPPROTO_TCP );
00092 
00093         // Connection
00094         CSock::connect( addr );
00095 }
00096 
00097 
00098 /*
00099  * Active disconnection. After disconnecting, you can't connect back with the same socket.
00100  */
00101 void CTcpSock::disconnect()
00102 {
00103         nldebug( "LNETL0: Socket %d disconnecting from %s...", _Sock, _RemoteAddr.asString().c_str() );
00104 
00105         // This shutdown resets the connection immediatly (not a graceful closure)
00106 #ifdef NL_OS_WINDOWS
00107         ::shutdown( _Sock, SD_BOTH );
00108 #elif defined NL_OS_UNIX
00109         ::shutdown( _Sock, SHUT_RDWR );
00110 #endif
00111         /*CSynchronized<bool>::CAccessor sync( &_SyncConnected );
00112         sync.value() = false;*/
00113         _Connected = false;
00114 }
00115 
00116 
00117 /*
00118  * Active disconnection for download way only
00119  */
00120 void CTcpSock::shutdownReceiving()
00121 {
00122 #ifdef NL_OS_WINDOWS
00123         ::shutdown( _Sock, SD_RECEIVE );
00124 #elif defined NL_OS_UNIX
00125         ::shutdown( _Sock, SHUT_RD );
00126 #endif
00127 }
00128 
00129 
00130 /*
00131  * Active disconnection for upload way only
00132  */
00133 void CTcpSock::shutdownSending()
00134 {
00135 #ifdef NL_OS_WINDOWS
00136         ::shutdown( _Sock, SD_SEND );
00137 #elif defined NL_OS_UNIX
00138         ::shutdown( _Sock, SHUT_WR );
00139 #endif
00140 }
00141 
00142 
00143 /*
00144  * Sets/unsets TCP_NODELAY
00145  */
00146 void CTcpSock::setNoDelay( bool value )
00147 {
00148         int b = value?1:0;
00149         if ( setsockopt( _Sock, IPPROTO_TCP, TCP_NODELAY, (char*)&b, sizeof(b) ) != 0 )
00150         {
00151                 throw ESocket( "setNoDelay failed" );
00152         }
00153 }
00154 
00155 
00156 /* Sets a custom TCP Window size (SO_RCVBUF and SO_SNDBUF).
00157  * You must close the socket is necessary, before calling this method.
00158  *
00159  * See http://www.ncsa.uiuc.edu/People/vwelch/net_perf/tcp_windows.html
00160  */
00161 void CTcpSock::connectWithCustomWindowSize( const CInetAddress& addr, int windowsize )
00162 {
00163         // Create socket
00164         if ( _Sock != INVALID_SOCKET )
00165         {
00166                 nlerror( "Cannot connect with custom window size when already connected" );
00167         }
00168         createSocket( SOCK_STREAM, IPPROTO_TCP );
00169 
00170         // Change window size
00171         if ( setsockopt( _Sock, SOL_SOCKET, SO_SNDBUF, (char*)&windowsize, sizeof(windowsize) ) != 0
00172           || setsockopt( _Sock, SOL_SOCKET, SO_RCVBUF, (char*)&windowsize, sizeof(windowsize) ) != 0 )
00173         {
00174                 throw ESocket( "setWindowSize failed" );
00175         }
00176         
00177         // Connection
00178         CSock::connect( addr );
00179 }
00180 
00181 
00182 /*
00183  * Returns the TCP Window Size for the current socket
00184  */
00185 uint32 CTcpSock::getWindowSize()
00186 {
00187         int windowsize = 0;
00188         socklen_t len = sizeof( windowsize );
00189 
00190         /* send buffer -- query for buffer size */
00191         if ( getsockopt( _Sock, SOL_SOCKET, SO_SNDBUF, (char*) &windowsize, &len ) == 0 )
00192         {
00193                 return windowsize;
00194         }
00195         else
00196         {
00197                 return 0;
00198         }
00199 }
00200 
00201 
00202 } // NLNET
tcp_sock.cpp
