| // |
| // basic_socket_iostream.hpp |
| // ~~~~~~~~~~~~~~~~~~~~~~~~~ |
| // |
| // Copyright (c) 2003-2008 Christopher M. Kohlhoff (chris at kohlhoff dot com) |
| // |
| // Distributed under the Boost Software License, Version 1.0. (See accompanying |
| // file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt) |
| // |
| |
| #ifndef ASIO_BASIC_SOCKET_IOSTREAM_HPP |
| #define ASIO_BASIC_SOCKET_IOSTREAM_HPP |
| |
| #if defined(_MSC_VER) && (_MSC_VER >= 1200) |
| # pragma once |
| #endif // defined(_MSC_VER) && (_MSC_VER >= 1200) |
| |
| #include "asio/detail/push_options.hpp" |
| |
| #include "asio/detail/push_options.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 "asio/detail/pop_options.hpp" |
| |
| #include "asio/basic_socket_streambuf.hpp" |
| #include "asio/stream_socket_service.hpp" |
| |
| #if !defined(ASIO_SOCKET_IOSTREAM_MAX_ARITY) |
| #define ASIO_SOCKET_IOSTREAM_MAX_ARITY 5 |
| #endif // !defined(ASIO_SOCKET_IOSTREAM_MAX_ARITY) |
| |
| // A macro that should expand to: |
| // template <typename T1, ..., typename Tn> |
| // explicit basic_socket_iostream(T1 x1, ..., Tn xn) |
| // : basic_iostream<char>(&this->boost::base_from_member< |
| // basic_socket_streambuf<Protocol, StreamSocketService> >::member) |
| // { |
| // if (rdbuf()->connect(x1, ..., xn) == 0) |
| // this->setstate(std::ios_base::failbit); |
| // } |
| // This macro should only persist within this file. |
| |
| #define ASIO_PRIVATE_CTR_DEF(z, n, data) \ |
| template <BOOST_PP_ENUM_PARAMS(n, typename T)> \ |
| explicit basic_socket_iostream(BOOST_PP_ENUM_BINARY_PARAMS(n, T, x)) \ |
| : std::basic_iostream<char>(&this->boost::base_from_member< \ |
| basic_socket_streambuf<Protocol, StreamSocketService> >::member) \ |
| { \ |
| tie(this); \ |
| if (rdbuf()->connect(BOOST_PP_ENUM_PARAMS(n, x)) == 0) \ |
| this->setstate(std::ios_base::failbit); \ |
| } \ |
| /**/ |
| |
| // A macro that should expand to: |
| // template <typename T1, ..., typename Tn> |
| // void connect(T1 x1, ..., Tn xn) |
| // { |
| // if (rdbuf()->connect(x1, ..., xn) == 0) |
| // this->setstate(std::ios_base::failbit); |
| // } |
| // This macro should only persist within this file. |
| |
| #define ASIO_PRIVATE_CONNECT_DEF(z, n, data) \ |
| template <BOOST_PP_ENUM_PARAMS(n, typename T)> \ |
| void connect(BOOST_PP_ENUM_BINARY_PARAMS(n, T, x)) \ |
| { \ |
| if (rdbuf()->connect(BOOST_PP_ENUM_PARAMS(n, x)) == 0) \ |
| this->setstate(std::ios_base::failbit); \ |
| } \ |
| /**/ |
| |
| namespace asio { |
| |
| /// Iostream interface for a socket. |
| template <typename Protocol, |
| typename StreamSocketService = stream_socket_service<Protocol> > |
| class basic_socket_iostream |
| : public boost::base_from_member< |
| basic_socket_streambuf<Protocol, StreamSocketService> >, |
| public std::basic_iostream<char> |
| { |
| public: |
| /// Construct a basic_socket_iostream without establishing a connection. |
| basic_socket_iostream() |
| : std::basic_iostream<char>(&this->boost::base_from_member< |
| basic_socket_streambuf<Protocol, StreamSocketService> >::member) |
| { |
| tie(this); |
| } |
| |
| #if defined(GENERATING_DOCUMENTATION) |
| /// Establish a connection to an endpoint corresponding to a resolver query. |
| /** |
| * This constructor automatically establishes a connection based on the |
| * supplied resolver query parameters. The arguments are used to construct |
| * a resolver query object. |
| */ |
| template <typename T1, ..., typename TN> |
| explicit basic_socket_iostream(T1 t1, ..., TN tn); |
| #else |
| BOOST_PP_REPEAT_FROM_TO( |
| 1, BOOST_PP_INC(ASIO_SOCKET_IOSTREAM_MAX_ARITY), |
| ASIO_PRIVATE_CTR_DEF, _ ) |
| #endif |
| |
| #if defined(GENERATING_DOCUMENTATION) |
| /// Establish a connection to an endpoint corresponding to a resolver query. |
| /** |
| * This function automatically establishes a connection based on the supplied |
| * resolver query parameters. The arguments are used to construct a resolver |
| * query object. |
| */ |
| template <typename T1, ..., typename TN> |
| void connect(T1 t1, ..., TN tn); |
| #else |
| BOOST_PP_REPEAT_FROM_TO( |
| 1, BOOST_PP_INC(ASIO_SOCKET_IOSTREAM_MAX_ARITY), |
| ASIO_PRIVATE_CONNECT_DEF, _ ) |
| #endif |
| |
| /// Close the connection. |
| void close() |
| { |
| if (rdbuf()->close() == 0) |
| this->setstate(std::ios_base::failbit); |
| } |
| |
| /// Return a pointer to the underlying streambuf. |
| basic_socket_streambuf<Protocol, StreamSocketService>* rdbuf() const |
| { |
| return const_cast<basic_socket_streambuf<Protocol, StreamSocketService>*>( |
| &this->boost::base_from_member< |
| basic_socket_streambuf<Protocol, StreamSocketService> >::member); |
| } |
| }; |
| |
| } // namespace asio |
| |
| #undef ASIO_PRIVATE_CTR_DEF |
| #undef ASIO_PRIVATE_CONNECT_DEF |
| |
| #include "asio/detail/pop_options.hpp" |
| |
| #endif // ASIO_BASIC_SOCKET_IOSTREAM_HPP |