AsyncFramedTcpConnection.h

Go to the documentation of this file.

 
#ifndef ASYNC_FRAMED_TCP_CONNECTION_INCLUDED
#define ASYNC_FRAMED_TCP_CONNECTION_INCLUDED
 
 
/****************************************************************************
 *
 * System Includes
 *
 ****************************************************************************/
 
#include <stdint.h>
#include <vector>
#include <deque>
#include <cstring>
 
 
/****************************************************************************
 *
 * Project Includes
 *
 ****************************************************************************/
 
#include <AsyncTcpConnection.h>
 
 
/****************************************************************************
 *
 * Local Includes
 *
 ****************************************************************************/
 
 
 
/****************************************************************************
 *
 * Forward declarations
 *
 ****************************************************************************/
 
 
 
/****************************************************************************
 *
 * Namespace
 *
 ****************************************************************************/
 
namespace Async
{
 
 
/****************************************************************************
 *
 * Forward declarations of classes inside of the declared namespace
 *
 ****************************************************************************/
 
 
 
/****************************************************************************
 *
 * Defines & typedefs
 *
 ****************************************************************************/
 
 
 
/****************************************************************************
 *
 * Exported Global Variables
 *
 ****************************************************************************/
 
 
 
/****************************************************************************
 *
 * Class definitions
 *
 ****************************************************************************/
 
class FramedTcpConnection : public TcpConnection
{
  public:
    explicit FramedTcpConnection(size_t recv_buf_len = DEFAULT_RECV_BUF_LEN);
 
    FramedTcpConnection(int sock, const IpAddress& remote_addr,
                        uint16_t remote_port,
                        size_t recv_buf_len = DEFAULT_RECV_BUF_LEN);
 
    virtual ~FramedTcpConnection(void);
 
    virtual TcpConnection& operator=(TcpConnection&& other_base) override;
 
    void setMaxRxFrameSize(uint32_t frame_size)
    {
      m_max_rx_frame_size = frame_size;
    }
 
    void setMaxTxFrameSize(uint32_t frame_size)
    {
      m_max_tx_frame_size = frame_size;
    }
 
    void setMaxFrameSize(uint32_t frame_size)
    {
      setMaxRxFrameSize(frame_size);
      setMaxTxFrameSize(frame_size);
    }
 
    virtual int write(const void *buf, int count) override;
 
    sigc::signal<void, FramedTcpConnection *, DisconnectReason> disconnected;
 
    sigc::signal<void, FramedTcpConnection *,
                 std::vector<uint8_t>&> frameReceived;
 
  protected:
    sigc::signal<int, TcpConnection*, void*, int> dataReceived;
    sigc::signal<void, bool> sendBufferFull;
 
    FramedTcpConnection& operator=(const FramedTcpConnection&) = delete;
 
    virtual void closeConnection(void) override;
 
    virtual void onDisconnected(DisconnectReason reason) override;
 
    virtual int onDataReceived(void *buf, int count) override;
 
    virtual void emitDisconnected(DisconnectReason reason) override
    {
      disconnected(this, reason);
      TcpConnection::emitDisconnected(reason);
    }
 
  private:
    static const uint32_t DEFAULT_MAX_FRAME_SIZE = 1024 * 1024; // 1MB
 
    struct QueueItem
    {
      char* m_buf;
      int   m_size;
      int   m_pos;
 
      QueueItem(const void* buf, int count)
        : m_buf(0), m_size(4+count), m_pos(0)
      {
        m_buf = new char[4+count];
        char *ptr = m_buf;
        *ptr++ = static_cast<uint32_t>(count) >> 24;
        *ptr++ = (static_cast<uint32_t>(count) >> 16) & 0xff;
        *ptr++ = (static_cast<uint32_t>(count) >> 8) & 0xff;
        *ptr++ = (static_cast<uint32_t>(count)) & 0xff;
        std::memcpy(ptr, buf, count);
      }
      ~QueueItem(void) { delete [] m_buf; }
    };
    typedef std::deque<QueueItem*> TxQueue;
 
    uint32_t              m_max_rx_frame_size;
    uint32_t              m_max_tx_frame_size;
    bool                  m_size_received;
    uint32_t              m_frame_size;
    std::vector<uint8_t>  m_frame;
    TxQueue               m_txq;
 
    FramedTcpConnection(const FramedTcpConnection&) = delete;
    //void onSendBufferFull(bool is_full);
    void disconnectCleanup(void);
 
};  /* class FramedTcpConnection */
 
 
} /* namespace */
 
#endif /* ASYNC_FRAMED_TCP_CONNECTION_INCLUDED */
 
 
 
/*
 * This file has not been truncated
 */