libs/cmt/include/mace/cmt/asio/tcp/socket.hpp

#ifndef _MACE_CMT_ASIO_TCP_SOCKET_HPP
#define _MACE_CMT_ASIO_TCP_SOCKET_HPP
#include <boost/asio.hpp>
#include <mace/cmt/future.hpp>

namespace mace { namespace cmt { namespace asio { namespace tcp {
  namespace detail { struct socket; }

  class socket  : public boost::asio::ip::tcp::socket {
    public:
      typedef boost::shared_ptr<socket> ptr;

      socket();
      ~socket();

      mace::cmt::future<boost::system::error_code> connect( const boost::asio::ip::tcp::endpoint& ep );

      struct iterator : public std::iterator<std::input_iterator_tag,char,void> {
        iterator( mace::cmt::asio::tcp::socket* _s = NULL)
        :s(_s){ if(_s){++*this;}  }

        inline const char& operator*()const  { return value;  }
        inline const char* operator->()const { return &value; }
        inline char& operator*() { return value;  }
        inline char* operator->(){ return &value; }

        iterator& operator++();
        iterator operator++(int);

        bool operator == ( const iterator& i )const { return s == i.s; }
        bool operator != ( const iterator& i )const { return s != i.s; }

        private:
          char               value;
          mace::cmt::asio::tcp::socket* s;
      };

      size_t read_some( char* buffer, size_t size );
      size_t read( char* buffer, size_t size );
      size_t write( const char* buffer, size_t size );

      void flush();
    private:
      detail::socket* my;
  };

} } } } // namespace mace::cmt::asio::tcp

#endif