peerplays-fc/include/fc/network/tcp_socket.hpp

#pragma once
#include <fc/utility.hpp>
#include <fc/fwd.hpp>
#include <fc/io/iostream.hpp>
#include <fc/time.hpp>

namespace fc {
  namespace ip { class endpoint; }

  class tcp_socket_io_hooks;

  class tcp_socket : public virtual iostream
  {
    public:
      tcp_socket();
      ~tcp_socket();

      void     connect_to( const fc::ip::endpoint& remote_endpoint );
      void     bind( const fc::ip::endpoint& local_endpoint );
      void     enable_keep_alives(const fc::microseconds& interval);
      void set_io_hooks(tcp_socket_io_hooks* new_hooks);
      void set_reuse_address(bool enable = true); // set SO_REUSEADDR
      fc::ip::endpoint remote_endpoint() const;
      fc::ip::endpoint local_endpoint() const;

      using istream::get;
      void get( char& c )
      {
          read( &c, 1 );
      }


      /// istream interface
      /// @{
      virtual size_t   readsome( char* buffer, size_t max );
      virtual size_t   readsome(const std::shared_ptr<char>& buffer, size_t max, size_t offset);
      virtual bool     eof()const;
      /// @}

      /// ostream interface
      /// @{
      virtual size_t   writesome( const char* buffer, size_t len );
      virtual size_t   writesome(const std::shared_ptr<const char>& buffer, size_t len, size_t offset);
      virtual void     flush();
      virtual void     close();
      /// @}

      void open();
      bool   is_open()const;

    private:
      friend class tcp_server;
      class impl;
      #ifdef _WIN64
      fc::fwd<impl,0x81> my;
      #else
      fc::fwd<impl,0x54> my;
      #endif
  };
  typedef std::shared_ptr<tcp_socket> tcp_socket_ptr;


  class tcp_server
  {
    public:
      tcp_server();
      ~tcp_server();

      void     close();
      void     accept( tcp_socket& s );
      void     set_reuse_address(bool enable = true); // set SO_REUSEADDR, call before listen
      void     listen( uint16_t port );
      void     listen( const fc::ip::endpoint& ep );
      fc::ip::endpoint get_local_endpoint() const;
      uint16_t get_port()const;
    private:
      // non copyable
      tcp_server( const tcp_server& );
      tcp_server& operator=(const tcp_server& s );

      class impl;
      impl* my;
  };

} // namesapce fc