SocketMuxer.h++

Go to the documentation of this file.

/* ---------------------------------------------------------------------------
   commonc++ - A C++ Common Class Library
   Copyright (C) 2005-2012  Mark A Lindner

   This file is part of commonc++.

   This library is free software; you can redistribute it and/or
   modify it under the terms of the GNU Library General Public
   License as published by the Free Software Foundation; either
   version 2 of the License, or (at your option) any later version.

   This library 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
   Library General Public License for more details.

   You should have received a copy of the GNU Library General Public
   License along with this library; if not, write to the Free
   Software Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
   ---------------------------------------------------------------------------
*/

#ifndef __ccxx_SocketMuxer_hxx
#define __ccxx_SocketMuxer_hxx

#include <commonc++/Common.h++>
#include <commonc++/CircularBuffer.h++>
#include <commonc++/CriticalSection.h++>
#include <commonc++/Iterator.h++>
#include <commonc++/StaticObjectPool.h++>
#include <commonc++/ServerSocket.h++>
#include <commonc++/StreamSocket.h++>
#include <commonc++/Thread.h++>
#include <commonc++/Mutex.h++>

#ifdef CCXX_OS_POSIX
#include <sys/select.h>
#endif

namespace ccxx {

class COMMONCPP_API Connection
{
  friend class SocketMuxer;

  public:

  static const size_t DEFAULT_BUFFER_SIZE;

  virtual ~Connection() throw();

  bool writeData(ByteBuffer& buffer);

  bool writeData(const byte_t *buf, size_t count);

  bool writeLine(const String &text);

  size_t readData(ByteBuffer& buffer, bool fully = true);

  size_t readData(byte_t *buf, size_t count, bool fully = true);

  size_t readLine(String &text, size_t maxLen);

  inline StreamSocket* getSocket() throw()
  { return(_socket); }

  void setReadLowWaterMark(size_t count) throw();

  void setReadHighWaterMark(size_t count) throw();

  inline size_t getReadLowWaterMark() const throw()
  { return(_readLoMark); }

  bool isReadLow() const throw();

  bool isReadHigh() const throw();

  void setWriteLowWaterMark(size_t count) throw();

  void setWriteHighWaterMark(size_t count) throw();

  inline size_t getWriteLowWaterMark() const throw()
  { return(_writeLoMark); }

  inline size_t getWriteHighWaterMark() const throw()
  { return(_writeHiMark); }

  bool isWriteLow() const throw();

  bool isWriteHigh() const throw();

  void close(bool immediate = false) throw();

  inline byte_t getOOBData() throw()
  { _oobFlag = false; return(_oobData); }

  inline bool getOOBFlag() const throw()
  { return(_oobFlag); }

  inline time_ms_t getTimestamp() const throw()
  { return(_lastRecv); }

  bool isClosePending() const throw();

  protected:

  Connection(size_t bufferSize = DEFAULT_BUFFER_SIZE);

  private:

  void read() throw(IOException);
  void readOOB() throw(IOException);
  void write() throw(IOException);

  void attach(StreamSocket *socket) throw();

  inline void setOOBFlag(bool flag) throw()
  {  _oobFlag = flag; }

  inline void setTimestamp(time_ms_t stamp) throw()
  { _lastRecv = stamp; }

  StreamSocket *_socket;

  protected:

  CircularByteBuffer readBuffer;
  CircularByteBuffer writeBuffer;

  private:

  size_t _readLoMark;
  size_t _readHiMark;
  size_t _writeLoMark;
  size_t _writeHiMark;
  bool _oobFlag;
  bool _closePending;
  byte_t _oobData;
  time_ms_t _lastRecv;
  mutable CriticalSection _readLock;
  mutable CriticalSection _writeLock;

  CCXX_COPY_DECLS(Connection);
};

class COMMONCPP_API SocketMuxer : public Thread
{
  public:

  SocketMuxer(uint_t maxConnections = 64, uint_t defaultIdleLimit = 0,
              uint_t sleepInterval = 100);

  virtual ~SocketMuxer() throw();

  void run();
  void cleanup();

  size_t getConnectionCount() const;

  virtual bool init(ServerSocket* socket);

  uint_t writeAll(const byte_t *buf, size_t count);

  protected:

  virtual Connection *connectionReady(const SocketAddress &address) = 0;

  virtual void dataReceived(Connection *connection) = 0;

  virtual void dataSent(Connection *connection);

  virtual void dataReceivedOOB(Connection *connection);

  virtual void connectionClosed(Connection *connection) = 0;

  virtual void connectionTimedOut(Connection *connection) = 0;

  virtual void exceptionOccurred(Connection *connection,
                                 const IOException& ex);

  class ConnectionList; // fwd decl

  ConnectionList *_connections;

  private:

  void _connectionTimedOut(Connection *connection);
  void _connectionClosed(Connection *connection);

  Mutex _mutex;
  StaticObjectPool<StreamSocket> _pool;
  uint_t _idleLimit;
  uint_t _sleepIntervalSec;
  uint_t _sleepIntervalMicroSec;
  ServerSocket* _ssock;

  CCXX_COPY_DECLS(SocketMuxer);
};

};

#endif // __ccxx_SocketMuxer_hxx

/* end of header file */