123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287 |
- #ifndef BOOST_ASIO_BASIC_SOCKET_STREAMBUF_HPP
- #define BOOST_ASIO_BASIC_SOCKET_STREAMBUF_HPP
- #if defined(_MSC_VER) && (_MSC_VER >= 1200)
- # pragma once
- #endif
- #include <boost/asio/detail/push_options.hpp>
- #include <boost/asio/detail/push_options.hpp>
- #include <streambuf>
- #include <boost/array.hpp>
- #include <boost/preprocessor/arithmetic/inc.hpp>
- #include <boost/preprocessor/repetition/enum_binary_params.hpp>
- #include <boost/preprocessor/repetition/enum_params.hpp>
- #include <boost/preprocessor/repetition/repeat_from_to.hpp>
- #include <boost/utility/base_from_member.hpp>
- #include <boost/asio/detail/pop_options.hpp>
- #include <boost/asio/basic_socket.hpp>
- #include <boost/asio/io_service.hpp>
- #include <boost/asio/stream_socket_service.hpp>
- #include <boost/asio/detail/throw_error.hpp>
- #if !defined(BOOST_ASIO_SOCKET_STREAMBUF_MAX_ARITY)
- #define BOOST_ASIO_SOCKET_STREAMBUF_MAX_ARITY 5
- #endif
- #define BOOST_ASIO_PRIVATE_CONNECT_DEF( z, n, data ) \
- template <BOOST_PP_ENUM_PARAMS(n, typename T)> \
- basic_socket_streambuf<Protocol, StreamSocketService>* connect( \
- BOOST_PP_ENUM_BINARY_PARAMS(n, T, x)) \
- { \
- init_buffers(); \
- boost::system::error_code ec; \
- this->basic_socket<Protocol, StreamSocketService>::close(ec); \
- typedef typename Protocol::resolver_query resolver_query; \
- resolver_query query(BOOST_PP_ENUM_PARAMS(n, x)); \
- resolve_and_connect(query, ec); \
- return !ec ? this : 0; \
- } \
-
- namespace boost {
- namespace asio {
- template <typename Protocol,
- typename StreamSocketService = stream_socket_service<Protocol> >
- class basic_socket_streambuf
- : public std::streambuf,
- private boost::base_from_member<io_service>,
- public basic_socket<Protocol, StreamSocketService>
- {
- public:
-
- typedef typename Protocol::endpoint endpoint_type;
-
- basic_socket_streambuf()
- : basic_socket<Protocol, StreamSocketService>(
- boost::base_from_member<boost::asio::io_service>::member),
- unbuffered_(false)
- {
- init_buffers();
- }
-
- virtual ~basic_socket_streambuf()
- {
- if (pptr() != pbase())
- overflow(traits_type::eof());
- }
-
-
- basic_socket_streambuf<Protocol, StreamSocketService>* connect(
- const endpoint_type& endpoint)
- {
- init_buffers();
- boost::system::error_code ec;
- this->basic_socket<Protocol, StreamSocketService>::close(ec);
- this->basic_socket<Protocol, StreamSocketService>::connect(endpoint, ec);
- return !ec ? this : 0;
- }
- #if defined(GENERATING_DOCUMENTATION)
-
-
- template <typename T1, ..., typename TN>
- basic_socket_streambuf<Protocol, StreamSocketService>* connect(
- T1 t1, ..., TN tn);
- #else
- BOOST_PP_REPEAT_FROM_TO(
- 1, BOOST_PP_INC(BOOST_ASIO_SOCKET_STREAMBUF_MAX_ARITY),
- BOOST_ASIO_PRIVATE_CONNECT_DEF, _ )
- #endif
-
-
- basic_socket_streambuf<Protocol, StreamSocketService>* close()
- {
- boost::system::error_code ec;
- sync();
- this->basic_socket<Protocol, StreamSocketService>::close(ec);
- if (!ec)
- init_buffers();
- return !ec ? this : 0;
- }
- protected:
- int_type underflow()
- {
- if (gptr() == egptr())
- {
- boost::system::error_code ec;
- std::size_t bytes_transferred = this->service.receive(
- this->implementation,
- boost::asio::buffer(boost::asio::buffer(get_buffer_) + putback_max),
- 0, ec);
- if (ec)
- return traits_type::eof();
- setg(get_buffer_.begin(), get_buffer_.begin() + putback_max,
- get_buffer_.begin() + putback_max + bytes_transferred);
- return traits_type::to_int_type(*gptr());
- }
- else
- {
- return traits_type::eof();
- }
- }
- int_type overflow(int_type c)
- {
- if (unbuffered_)
- {
- if (traits_type::eq_int_type(c, traits_type::eof()))
- {
-
- return traits_type::not_eof(c);
- }
- else
- {
-
- boost::system::error_code ec;
- char_type ch = traits_type::to_char_type(c);
- this->service.send(this->implementation,
- boost::asio::buffer(&ch, sizeof(char_type)), 0, ec);
- if (ec)
- return traits_type::eof();
- return c;
- }
- }
- else
- {
-
- boost::asio::const_buffer buffer =
- boost::asio::buffer(pbase(), pptr() - pbase());
- while (boost::asio::buffer_size(buffer) > 0)
- {
- boost::system::error_code ec;
- std::size_t bytes_transferred = this->service.send(
- this->implementation, boost::asio::buffer(buffer),
- 0, ec);
- if (ec)
- return traits_type::eof();
- buffer = buffer + bytes_transferred;
- }
- setp(put_buffer_.begin(), put_buffer_.end());
-
- if (traits_type::eq_int_type(c, traits_type::eof()))
- return traits_type::not_eof(c);
-
- *pptr() = traits_type::to_char_type(c);
- pbump(1);
- return c;
- }
- }
- int sync()
- {
- return overflow(traits_type::eof());
- }
- std::streambuf* setbuf(char_type* s, std::streamsize n)
- {
- if (pptr() == pbase() && s == 0 && n == 0)
- {
- unbuffered_ = true;
- setp(0, 0);
- return this;
- }
- return 0;
- }
- private:
- void init_buffers()
- {
- setg(get_buffer_.begin(),
- get_buffer_.begin() + putback_max,
- get_buffer_.begin() + putback_max);
- if (unbuffered_)
- setp(0, 0);
- else
- setp(put_buffer_.begin(), put_buffer_.end());
- }
- template <typename ResolverQuery>
- void resolve_and_connect(const ResolverQuery& query,
- boost::system::error_code& ec)
- {
- typedef typename Protocol::resolver resolver_type;
- typedef typename Protocol::resolver_iterator iterator_type;
- resolver_type resolver(
- boost::base_from_member<boost::asio::io_service>::member);
- iterator_type i = resolver.resolve(query, ec);
- if (!ec)
- {
- iterator_type end;
- ec = boost::asio::error::host_not_found;
- while (ec && i != end)
- {
- this->basic_socket<Protocol, StreamSocketService>::close();
- this->basic_socket<Protocol, StreamSocketService>::connect(*i, ec);
- ++i;
- }
- }
- }
- enum { putback_max = 8 };
- enum { buffer_size = 512 };
- boost::array<char, buffer_size> get_buffer_;
- boost::array<char, buffer_size> put_buffer_;
- bool unbuffered_;
- };
- }
- }
- #undef BOOST_ASIO_PRIVATE_CONNECT_DEF
- #include <boost/asio/detail/pop_options.hpp>
- #endif
|