networking.h

Go to the documentation of this file.

// -*- Mode: C++; tab-width: 2; -*-
// vi: set ts=2:
//

#ifndef BALL_SYSTEM_NETWORKING_H
#define BALL_SYSTEM_NETWORKING_H

#ifndef BALL_COMMON_GLOBAL_HH
# include<BALL/COMMON/global.h>
#endif

#ifndef BALL_DATATYPE_STRING_H
# include <BALL/DATATYPE/string.h>
#endif

#ifdef BALL_HAS_BOOST_ASIO
# include <boost/asio.hpp>
#else
#define BOOST_DATE_TIME_NO_LIB
#define BOOST_REGEX_NO_LIB
# include <asio.hpp>
#endif

#include <QtCore/QThread>

namespace BALL
{
  class BALL_EXPORT TCPIOStream
    : public BALL_ASIO_NAMESPACE::ip::tcp::iostream
  {
    public:
      TCPIOStream() 
        : BALL_ASIO_NAMESPACE::ip::tcp::iostream()
      {
      }

      TCPIOStream(const String& hostname, const String& protocol)
        : BALL_ASIO_NAMESPACE::ip::tcp::iostream(hostname, protocol)
      {
      }

      TCPIOStream(const String& hostname, Position port)
        : BALL_ASIO_NAMESPACE::ip::tcp::iostream(hostname, String(port))
      {
      }
  };

  class BALL_EXPORT TCPServer
  {
    public:
      TCPServer(Size port, bool restart = true)
        : port_(port), 
          restart_(restart),
          connected_stream_(),
          io_service_(),
          acceptor_(io_service_)
      {};

      virtual ~TCPServer();

      virtual void activate();
      virtual void deactivate();

      virtual void startAccepting();
      virtual void handleConnection();
      virtual void connectionRequested();

      void setPort(Size port);
      Size getPort() const;

    protected:
      Size port_; 
      bool restart_;

      TCPIOStream connected_stream_;

      BALL_ASIO_NAMESPACE::io_service io_service_;

      BALL_ASIO_NAMESPACE::ip::tcp::acceptor acceptor_;
  };

  class BALL_EXPORT TCPServerThread
    : public TCPServer,
      public virtual QThread
  {
    public:
      TCPServerThread(Size port, bool asynchronous = true, bool restart = true);
      
      virtual void run();
      virtual void deactivate();
      virtual void activate_async();
      virtual void handleAsyncConnection();
      virtual void handleClose();

      bool isRunning();

    protected:
      bool use_async_io_;
      bool is_running_;
  };
} // namespace BALL

#endif  // BALL_SYSTEM_NETWORKING_H