Server.h

Go to the documentation of this file.

#ifndef GCP_UTIL_SERVER_H
#define GCP_UTIL_SERVER_H

#include <list>
#include <string>
#include <stdio.h>

#include "gcp/util/common/FdSet.h"
#include "gcp/util/common/NetDat.h"
#include "gcp/util/common/NetHandler.h"
#include "gcp/util/common/Runnable.h"
#include "gcp/util/common/SignalTaskMsg.h"
#include "gcp/util/common/TimeOut.h"

namespace gcp {
  namespace util {
    
    class Port;
    class SignalTask;
    class TcpListener;
    
    class Server : public Runnable {
      
    public:
      
      struct ServerConnection {
    NetHandler handler_;
    Server* parent_;

    ServerConnection(int fd, 
             unsigned readBufSize, unsigned sendBufSize, 
             Server* parent);

    ~ServerConnection();
      };

      Server(bool spawnThread, int listenPort, unsigned readBufSize=0, 
         unsigned sendBufSize=0);
      
      virtual ~Server();
      
      virtual void run();

    protected:

      unsigned readBufSize_;

      unsigned sendBufSize_;

      TimeOut timeOut_;

      // Method called when we time out in our select loop

      virtual void timeOutAction() {};

      // Method called when data have been completely read from a client

      virtual void readClientData(NetHandler& handler) {};

      // Method called immediately after a client has connected

      virtual void acceptClientAction() {};

      // Method to check any other fds that may be registered

      virtual void checkOtherFds() {};

      void sendClientData(NetDat& dat);

      void setReadBufSize(unsigned size);
      void setSendBufSize(unsigned size);

      void setTimeOutSeconds(unsigned int seconds);

    protected:

      FdSet fdSet_;
      
    private:
      
      unsigned int timeOutSeconds_;

      SignalTask* signalTask_;
      
      bool stop_;
      
      TcpListener* listener_;
      
      std::list<ServerConnection*> clients_;
      
      void initMembers(int listenPort, unsigned readBufSize, unsigned sendBufSize);
      
      void listen(unsigned port, unsigned nClients = 5);
      
      void sendShutdownMsg();
      
      void acceptConnection();
      
    private:

      void checkClients();
      
      void readFromClient(NetHandler* client);
      
      static SIGNALTASK_HANDLER_FN(shutDown);
      
      static RUN_FN(runFn);
      
      // Check clients for data to be read

      void checkClientsForReadableData();

      // Check clients for writability

      void checkClientsForWritability();

      static NET_READ_HANDLER(readHandler);
      static NET_SEND_HANDLER(sendHandler);
      static NET_ERROR_HANDLER(errHandler);

    }; // End class Server
    
  } // End namespace util
} // End namespace gcp



#endif // End #ifndef GCP_UTIL_SERVER_H