vsnet_socketbase.cpp

Go to the documentation of this file.

#include <config.h>

#include "vsnet_headers.h"
#if !defined (_WIN32) || defined (__CYGWIN__)
#include <sys/ioctl.h>
#endif

#include "vsnet_oss.h"
#include "vsnet_socketbase.h"
#include "vsnet_socketset.h"

/***********************************************************************
* VsnetSocket - definition
***********************************************************************/

VsnetSocket::VsnetSocket( int sock, const AddressIP &remote_ip, const char *socktype, SocketSet &sets ) :
    VsnetSocketBase( sock, socktype, sets )
    , _remote_ip( remote_ip )
{}

VsnetSocket::~VsnetSocket()
{}

//VsnetSocket& VsnetSocket::operator=( const VsnetSocket& orig )
//{
//VsnetSocketBase::operator=( orig );
//_remote_ip = orig._remote_ip;
//return *this;
//}

bool VsnetSocket::eq( const VsnetSocket &r ) const
{
    return ( isTcp() == r.isTcp() ) && ( get_fd() == r.get_fd() ) && (_remote_ip == r._remote_ip);
}

bool VsnetSocket::lt( const VsnetSocket &r ) const
{
    if ( !isTcp() && r.isTcp() ) return true;
    if ( isTcp() == r.isTcp() ) {
        if ( get_fd() < r.get_fd() ) return true;
        if ( get_fd() == r.get_fd() )
            if (_remote_ip < r._remote_ip) return true;
    }
    return false;
}

bool VsnetSocket::sameAddress( const VsnetSocket &r ) const
{
    const VsnetSocket *r2 = (const VsnetSocket*) &r;
    return ( isTcp() == r2->isTcp() ) && (_remote_ip == r2->_remote_ip);
}

/***********************************************************************
* VsnetSocketBase - definition
***********************************************************************/

VsnetSocketBase::VsnetSocketBase( int fd, const char *socktype, SocketSet &sets ) :
    _fd( fd )
    , _set( &sets )
    , _noblock( -1 )
{
    MAKE_VALID

    char buf[100];

    sprintf( buf, "%d", fd );
    unsigned int len = strlen( socktype )+strlen( buf )+4;
    _socktype = new char[len];

    sprintf( _socktype, "%s (%s)", buf, socktype );

    assert( strlen( _socktype ) == len-1 );
    if (fd >= 0)
        set_block();
    sets.set( this );
}
bool VsnetSocketBase::isReadyToSend( fd_set *write_set_select )
{
    return get_write_fd() >= 0 && FD_ISSET( get_write_fd(), write_set_select );
}
VsnetSocketBase::~VsnetSocketBase()
{
    CHECK_VALID
    if (_set) _set->unset( this );
    delete[] _socktype;
    MAKE_INVALID
}

void VsnetSocketBase::setSet( SocketSet *set )
{
    if (this->_set) this->_set->unset( this );
    this->_set = set;
    if (this->_set) this->_set->set( this );
}

const char* VsnetSocketBase::get_socktype() const
{
    return _socktype;
}

bool VsnetSocketBase::valid() const
{
    CHECK_VALID
    return _fd >= 0;
}

int VsnetSocketBase::get_fd() const
{
    CHECK_VALID
    return _fd;
}

int VsnetSocketBase::close_fd()
{
    CHECK_VALID
    int ret = 0;
    if (_fd >= 0) {
        ret = VsnetOSS::close_socket( _fd );
        _fd = -1;
    }
    return ret;
}

bool VsnetSocketBase::set_nonblock()
{
    CHECK_VALID
    if ( !valid() ) {
        COUT<<"Failed to set blocking socket "<<_socktype<<std::endl;
        return false;
    }
    bool ret = VsnetOSS::set_blocking( _fd, false );
    if (ret)
        _noblock = 1;
    return ret;
}

bool VsnetSocketBase::set_block()
{
    CHECK_VALID
    if ( !valid() ) {
        COUT<<"Failed to set blocking socket "<<_socktype<<std::endl;
        return false;
    }
    bool ret = VsnetOSS::set_blocking( _fd, true );
    if (ret)
        _noblock = 0;
    return ret;
}

bool VsnetSocketBase::get_nonblock() const
{
    CHECK_VALID
    return _noblock == 1;
}

void VsnetSocketBase::disconnect( const char *s )
{
    CHECK_VALID
    if (_fd > 0)
        if (_set) _set->unset( this );
    child_disconnect( s );
}