147 lines
4.7 KiB
C++
147 lines
4.7 KiB
C++
//
|
|
// basic_socket_iostream.hpp
|
|
// ~~~~~~~~~~~~~~~~~~~~~~~~~
|
|
//
|
|
// Copyright (c) 2003-2007 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) \
|
|
{ \
|
|
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)
|
|
{
|
|
}
|
|
|
|
#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
|