AsyncTcpServerBase.h

Go to the documentation of this file.

 
#ifndef ASYNC_TCP_SERVER_BASE_INCLUDED
#define ASYNC_TCP_SERVER_BASE_INCLUDED
 
 
/****************************************************************************
 *
 * System Includes
 *
 ****************************************************************************/
 
#include <string>
#include <vector>
#include <map>
#include <sigc++/sigc++.h>
 
 
/****************************************************************************
 *
 * Project Includes
 *
 ****************************************************************************/
 
#include <AsyncTimer.h>
#include <AsyncTcpConnection.h>
#include <AsyncSslContext.h>
 
 
/****************************************************************************
 *
 * Local Includes
 *
 ****************************************************************************/
 
 
 
/****************************************************************************
 *
 * Forward declarations
 *
 ****************************************************************************/
 
 
 
/****************************************************************************
 *
 * Namespace
 *
 ****************************************************************************/
 
namespace Async
{
 
/****************************************************************************
 *
 * Forward declarations of classes inside of the declared namespace
 *
 ****************************************************************************/
 
class FdWatch;
 
 
/****************************************************************************
 *
 * Defines & typedefs
 *
 ****************************************************************************/
 
 
 
/****************************************************************************
 *
 * Exported Global Variables
 *
 ****************************************************************************/
 
 
 
/****************************************************************************
 *
 * Class definitions
 *
 ****************************************************************************/
 
class TcpServerBase : public sigc::trackable
{
  public:
    TcpServerBase(const std::string& port_str,
                  const Async::IpAddress &bind_ip);
 
    virtual ~TcpServerBase(void);
 
    int numberOfClients(void);
 
    TcpConnection *getClient(unsigned int index);
 
    int writeAll(const void *buf, int count);
 
    int writeOnly(TcpConnection *con, const void *buf, int count);
 
    int writeExcept(TcpConnection *con, const void *buf, int count);
 
    void setSslContext(SslContext& ctx);
 
    void setConnectionThrottling(unsigned bucket_max, float bucket_inc,
                                 int inc_interval_ms);
 
  protected:
    virtual void createConnection(int sock, const IpAddress& remote_addr,
                                  uint16_t remote_port) = 0;
    void addConnection(TcpConnection *con);
    void removeConnection(TcpConnection *con);
    virtual void emitClientConnected(TcpConnection *con) = 0;
 
  private:
    using TcpConnectionList = std::vector<TcpConnection*>;
    struct ConThrotItem
    {
      ConThrotItem(float bucket_max) : m_bucket(bucket_max) {}
      float             m_bucket;
      TcpConnectionList m_pending_connections;
    };
    using ConThrotMap = std::map<IpAddress, ConThrotItem>;
 
    int               m_sock;
    FdWatch*          m_rd_watch;
    TcpConnectionList m_tcpConnectionList;
    SslContext*       m_ssl_ctx               = nullptr;
 
    ConThrotMap       m_con_throt_map;
    Timer             m_con_throt_timer;
    unsigned          m_con_throt_bucket_max  = 0;
    unsigned          m_con_throt_bucket_inc  = 0;
 
    void cleanup(void);
    void onConnection(FdWatch *watch);
    void updateConnThrotMap(Timer*);
 
};  /* class TcpServerBase */
 
 
} /* namespace */
 
#endif /* ASYNC_TCP_SERVER_BASE_INCLUDED */
 
 
 
/*
 * This file has not been truncated
 */