tcp.h

Go to the documentation of this file.

/*  Glass - a distributed computing library
    Copyright (C) 2003-2009 Bruno Barberi Gnecco
    Copyright (C) 2009-2010 Corollarium Technologies


    This library is free software; you can redistribute it and/or
    modify it under the terms of the GNU Lesser General Public
    License as published by the Free Software Foundation; either
    version 2 of the License, or (at your option) any later version.

    This library 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
    Lesser General Public License for more details.

    You should have received a copy of the GNU Lesser General Public
    License along with this library; if not, write to the Free Software
    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 */

#ifndef _GLASS_PROTOCOL_TCP_HH_
#define _GLASS_PROTOCOL_TCP_HH_

#include <list>

#if defined(_MSC_VER)
# include <winsock2.h>
# include <time.h>
#else
# include <sys/time.h>
# include <sys/select.h>
# include <unistd.h>
#endif
#include <sys/types.h>
#include "protocol.h"
#include "packet.h"

namespace libglass {

class PacketLog;


#if defined(_MSC_VER)
// Instantiate std::list<T> template containers used by class Protocol
LIBGLASS_IMP_TEMPLATE_STL_LIST(Packet*);
#endif // _MSC_VER

class LIBGLASS_API TCP : public Protocol {
private:
        int readPacket(Packet &p, nodeFD sock);
        unsigned int writePacket(nodeFD fd, chain &m, unsigned int length);

        std::list<Packet *>             queue;

        boost::mutex    mutex;          
        const struct in_addr gethostbyname_r(const char *host) const throw(Exception);

protected:
#if defined(_MSC_VER) && (_MSC_VER < 1300)  // earlier than .NET compiler (VC 7.0)
        // Microsoft Knowledge Base Article - 241569
        // Workaround for BUG: C2258 and C2252 on in Place Initialization of Static Const Members
        enum{ default_port = 9191 };
#else
        static const unsigned int               default_port = 9191;
#endif

        union {
                struct {
                        int             sockfd;         
                        int             fdmax;          
                        fd_set          clients;        
                        struct timeval  default_timeout;
                        bool            no_timeout;
                } server;
                struct {
                        int             sockfd;         
                } client;
        } data;

        PacketLog                       *plog;

        int _listen(unsigned int port);
        nodeFD _accept(void);
        nodeFD _connect(const char *host, unsigned int port);
        void _disconnect(nodeFD node);
        bool _send(Packet &p);
        int _recv(Packet &p);
        int _safely_close_socket(nodeFD fd);

public:
        TCP();
        ~TCP();

        nodeId getUniqueId(void);
};

} //namespace

#endif // _GLASS_PROTOCOL_TCP_HH_

---