|

TCPClient.hpp

Go to the documentation of this file.

#ifndef IONFLUX_TOOLS_TCPCLIENT
#define IONFLUX_TOOLS_TCPCLIENT
/* ==========================================================================
 * Ionflux Tools
 * Copyright (c) 2004 Joern P. Meier
 * mail@ionflux.org
 * --------------------------------------------------------------------------
 * TCPClient.hpp                    Generic TCP client.
 * ==========================================================================
 * 
 * This file is part of Ionflux Tools.
 * 
 * Ionflux Tools 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 of the License, or (at  your option)
 * any later version.
 * 
 * Ionflux Tools 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 Ionflux Tools; if not, write to the Free Software Foundation, 
 * Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
 * 
 * ========================================================================== */

#include <string>
#include <iostream>
#include <vector>
#include <signal.h>
#include "ionflux/Reporter.hpp"
#include "ionflux/TCPRemotePeer.hpp"
#include "ionflux/IOHandler.hpp"
#include "ionflux/SelectMultiplexer.hpp"

namespace Ionflux
{

namespace Tools
{

class TCPClient
: public IOHandler
{
    protected:
        Ionflux::Tools::Reporter log;
        bool interactive;
        std::vector<TCPRemotePeer *> peers;
        std::vector<TCPRemotePeer *> trash;
        int currentPeerID;
        IOMultiplexer *iomp;
        bool manageIomp;
        IOEvent stdinEvent;
        
        virtual void addPeer(TCPRemotePeer *peer);
        
        virtual void removePeer(TCPRemotePeer *peer);
        
        virtual void cleanupPeers();
        
        virtual void onConnect(TCPRemotePeer &peer);
        
        virtual void onReceive(TCPRemotePeer &peer);
        
        virtual void onDisconnect(TCPRemotePeer &peer);
        
        virtual void disconnect(TCPRemotePeer *peer);
        
    public:
        TCPClient();
        
        TCPClient(bool initInteractive);
        
        TCPClient(bool initInteractive, IOMultiplexer *initIomp);
        
        virtual ~TCPClient();
        
        virtual void onInput(const std::string &line);
        
        virtual void run();
        
        virtual void cleanup();
        
        virtual bool addConnection(const std::string &host, int port);
        
        virtual void broadcast(const std::string &bytes);
        
        virtual void onIO(const IOEvent &event);
        
        static void shutdownHandler(int signum);
        
        virtual Reporter &getLog();

};

}

}

#endif

---