---

communSocket.cpp

/*
 *                         Sun Public License Notice
 *
 * The contents of this file are subject to the Sun Public License
 * Version 1.0 (the "License"); you may not use this file except
 * in compliance with the License. A copy of the License is available
 * at http://www.sun.com/
 *
 * The Original Code is the Java Profiler module.  The Initial Developers
 * of the Original Code are Jan Stola, Pavel Vacha, Michal Pise, Petr Luner,
 * Lukas Petru and Marek Przeczek.
 *
 * Portions created by Jan Stola are Copyright (C) 2000-2001.
 * All Rights Reserved.
 *
 * Portions created by Pavel Vacha are Copyright (C) 2000-2001.
 * All Rights Reserved.
 *
 * Portions created by Michal Pise are Copyright (C) 2000-2001.
 * All Rights Reserved.
 *
 * Portions created by Petr Luner are Copyright (C) 2000-2001.
 * All Rights Reserved.
 *
 * Portions created by Lukas Petru are Copyright (C) 2000-2001.
 * All Rights Reserved.
 *
 * Portions created by Marek Przeczek are Copyright (C) 2000-2001.
 * All Rights Reserved.
 *
 * Contributors: Jan Stola, Pavel Vacha, Michal Pise, Petr Luner,
 *               Lukas Petru and Marek Przeczek.
 */

#include "../commun/communSocket.h"

CommunSocket::CommunSocket( int connectMode, const String& hostname, unsigned short port) :

        _connectMode( connectMode),
        _hostname( hostname),
        _port( port) {

        FD_ZERO( &_fds);

#ifdef WIN32
        _csock = INVALID_SOCKET;

        WSADATA wsaData;
        if( _failed = (WSAStartup( MAKEWORD( 1, 1), &wsaData) != 0)) return;

        if( _failed = ((_sock = socket( AF_INET, SOCK_STREAM, 0)) == INVALID_SOCKET)) return;
#else
        _csock = -1;

        if( _failed = ((_sock = socket( PF_INET, SOCK_STREAM, 0)) < 0)) return;
#endif

        if( _connectMode) { // client

                _csock = _sock;
                return;
        }

        sockaddr_in sin;
        sin.sin_family = AF_INET;
        sin.sin_port   = htons( port);
        sin.sin_addr.s_addr = htonl( INADDR_ANY);

#ifdef WIN32
        if( _failed = (bind( _sock, (sockaddr*)&sin, sizeof( sin)) == SOCKET_ERROR)) return;
        if( _failed = (listen( _sock, 1) == SOCKET_ERROR)) return;
#else
        if( _failed = (bind( _sock, (sockaddr*)&sin, sizeof( sin)) < 0)) return;
        if( _failed = (listen( _sock, 1) < 0)) return;
#endif
}

CommunSocket::~CommunSocket() {

#ifdef WIN32
        if( !_connectMode) { // server
        
                shutdown( _sock, SD_BOTH);
                closesocket( _sock);
        }

        shutdown( _csock, SD_BOTH);
        closesocket( _csock);

        WSACleanup();
#else
        if( !_connectMode) { // server
        
                shutdown( _sock, SHUT_RDWR);
                close( _sock);
        }

        shutdown( _csock, SHUT_RDWR);
        close( _csock);
#endif
}

int CommunSocket::initialize() {

        sockaddr_in cin;

#ifdef WIN32
        int csize = sizeof( cin);
#else
        socklen_t csize = sizeof( cin);
#endif

        if( !_connectMode) { // server
#ifdef WIN32
                if( _csock != INVALID_SOCKET) {

                        shutdown( _csock, SD_BOTH);
                        closesocket( _csock);
                }

                _failed = ((_csock = accept( _sock, (sockaddr*)&cin, &csize)) == INVALID_SOCKET);
#else
                if( _csock >= 0) {

                        shutdown( _csock, SHUT_RDWR);
                        close( _csock);
                }

                _failed = ((_csock = accept( _sock, (sockaddr*)&cin, &csize)) < 0);
#endif
        }
        else { // client

                cin.sin_family = AF_INET;
                cin.sin_port   = htons( _port);

                hostent* h = gethostbyname( _hostname);
                if( _failed = (h == NULL)) return 0;

                cin.sin_addr.s_addr = *(unsigned long*)h->h_addr_list[0];

#ifdef WIN32
                _failed = (connect( _csock, (sockaddr*)&cin, csize) == SOCKET_ERROR);
#else
                _failed = (connect( _csock, (sockaddr*)&cin, csize) < 0);
#endif
        }

        if( !_failed) {

                FD_ZERO( &_fds);
                FD_SET( _csock, &_fds);
        }
        else if( _connectMode) { // failed and is client

#ifdef WIN32
                shutdown( _csock, SD_BOTH);
                closesocket( _csock);

                _csock = socket( AF_INET, SOCK_STREAM, 0);
#else
                shutdown( _csock, SHUT_RDWR);
                close( _csock);

                _csock = socket( PF_INET, SOCK_STREAM, 0);
#endif
        }

        return !_failed;
}

Commun& CommunSocket::operator<<( const Buffer& b) {

        // send in network-byte-order
        jint size = b.getSize();
        jint nsize = htonl( size);

#ifdef WIN32
        if( _failed = (send( _csock, (char*)&nsize, sizeof( nsize), 0) == SOCKET_ERROR)) return *this;

        _failed = (size && send( _csock, b.getBuffer(), size, 0) == SOCKET_ERROR);
#else
        if( _failed = (send( _csock, (char*)&nsize, sizeof( nsize), 0) < 0)) return *this;

        _failed = (size && send( _csock, b.getBuffer(), size, 0) < 0);
#endif

        return *this;
}

Commun& CommunSocket::operator>>( Buffer& b) {

        jint nsize;
        int rc = recv( _csock, (char*)&nsize, sizeof( nsize), 0);
        
#ifdef WIN32
        if( _failed = (rc == SOCKET_ERROR || rc < sizeof( nsize))) return *this;
#else
        if( _failed = (rc < 0 || rc < sizeof( nsize))) return *this;
#endif

        // convert back to host-byte-order
        jint size = ntohl( nsize);
        
        if( size) {

                char* buf = new char[size];

                rc = recv( _csock, buf, size, 0);

#ifdef WIN32
                _failed = (rc == SOCKET_ERROR || rc < size);
#else
                _failed = (rc < 0 || rc < size);
#endif

                if( !_failed) b = Buffer( buf, size);

                delete[] buf;
        }
        else b.clear();

        return *this;
}

int CommunSocket::isReady() {

        int rc = select( _csock+1, &_fds, NULL, NULL, NULL);

#ifdef WIN32
        if( rc == SOCKET_ERROR) return !(_failed = 1);
#else
        if( rc < 0) return !(_failed = 1);
#endif

        return rc;
}

---