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
 
#include <boost/asio.hpp>
 
#include <QtCore/QThread>
 
namespace BALL
{
  class BALL_EXPORT TCPIOStream
    : public boost::asio::ip::tcp::iostream
  {
    public:
      TCPIOStream() 
        : boost::asio::ip::tcp::iostream()
      {
      }
 
      TCPIOStream(const String& hostname, const String& protocol)
        : boost::asio::ip::tcp::iostream(static_cast<const std::string&>(hostname),
                                          static_cast<const std::string&>(protocol))
      {
      }
 
      TCPIOStream(const String& hostname, Position port)
        : boost::asio::ip::tcp::iostream(static_cast<const std::string&>(hostname),
                                          std::string(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_;
 
      boost::asio::io_context io_service_;
 
      boost::asio::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