|

TCPServer.hpp

Go to the documentation of this file.

#ifndef IONFLUX_TOOLS_TCPSERVER
#define IONFLUX_TOOLS_TCPSERVER
/* ==========================================================================
 * Ionflux Tools
 * Copyright (c) 2004 Joern P. Meier
 * mail@ionflux.org
 * --------------------------------------------------------------------------
 * TCPServer.hpp                    Generic TCP server.
 * ==========================================================================
 * 
 * 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/tools.hpp"
#include "ionflux/Reporter.hpp"
#include "ionflux/TCPSocket.hpp"
#include "ionflux/TCPRemotePeer.hpp"
#include "ionflux/IOHandler.hpp"
#include "ionflux/SelectMultiplexer.hpp"

namespace Ionflux
{

namespace Tools
{

class TCPServer
: public IOHandler
{
    protected:
        Ionflux::Tools::Reporter log;
        TCPSocket serverSocket;
        std::vector<TCPRemotePeer *> clients;
        std::vector<TCPRemotePeer *> trash;
        unsigned int maxClients;
        int currentClientID;
        IOMultiplexer *iomp;
        bool manageIomp;
        IOEvent serverSocketEvent;
        
        virtual void addClient(TCPRemotePeer *client);
        
        virtual void removeClient(TCPRemotePeer *client);
        
        virtual void cleanupClients();
        
        virtual void onConnect(TCPRemotePeer &client);
        
        virtual void onReject(TCPRemotePeer &client, int reason);
        
        virtual void onReceive(TCPRemotePeer &client);
        
        virtual void onDisconnect(TCPRemotePeer &client);
        
        virtual void disconnect(TCPRemotePeer *peer);
        
    public:
        static const int DEFAULT_MAX_CLIENTS;
        static const int REJECTED_REASON_MAX_CLIENTS;
        
        TCPServer();
        
        TCPServer(IOMultiplexer *initIomp);
        
        virtual ~TCPServer();
        
        virtual bool init();
        
        virtual void run();
        
        virtual void cleanup();
        
        virtual void broadcast(const std::string &bytes);
        
        virtual void onIO(const IOEvent &event);
        
        virtual void setMaxClients(unsigned int newMaxClients);
        
        virtual void setPort(int newPort);
        
        virtual unsigned int getMaxClients();
        
        virtual int getPort();
        
        static void shutdownHandler(int signum);
        
        virtual Reporter &getLog();
};

}

}

#endif

---