562 lines
20 KiB
C++
562 lines
20 KiB
C++
//
|
|
// windows/basic_stream_handle.hpp
|
|
// ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
|
//
|
|
// Copyright (c) 2003-2023 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_WINDOWS_BASIC_STREAM_HANDLE_HPP
|
|
#define ASIO_WINDOWS_BASIC_STREAM_HANDLE_HPP
|
|
|
|
#if defined(_MSC_VER) && (_MSC_VER >= 1200)
|
|
# pragma once
|
|
#endif // defined(_MSC_VER) && (_MSC_VER >= 1200)
|
|
|
|
#include "asio/detail/config.hpp"
|
|
#include "asio/windows/basic_overlapped_handle.hpp"
|
|
|
|
#if defined(ASIO_HAS_WINDOWS_STREAM_HANDLE) \
|
|
|| defined(GENERATING_DOCUMENTATION)
|
|
|
|
#include "asio/detail/push_options.hpp"
|
|
|
|
namespace asio {
|
|
namespace windows {
|
|
|
|
/// Provides stream-oriented handle functionality.
|
|
/**
|
|
* The windows::basic_stream_handle class provides asynchronous and blocking
|
|
* stream-oriented handle functionality.
|
|
*
|
|
* @par Thread Safety
|
|
* @e Distinct @e objects: Safe.@n
|
|
* @e Shared @e objects: Unsafe.
|
|
*
|
|
* @par Concepts:
|
|
* AsyncReadStream, AsyncWriteStream, Stream, SyncReadStream, SyncWriteStream.
|
|
*/
|
|
template <typename Executor = any_io_executor>
|
|
class basic_stream_handle
|
|
: public basic_overlapped_handle<Executor>
|
|
{
|
|
private:
|
|
class initiate_async_write_some;
|
|
class initiate_async_read_some;
|
|
|
|
public:
|
|
/// The type of the executor associated with the object.
|
|
typedef Executor executor_type;
|
|
|
|
/// Rebinds the handle type to another executor.
|
|
template <typename Executor1>
|
|
struct rebind_executor
|
|
{
|
|
/// The handle type when rebound to the specified executor.
|
|
typedef basic_stream_handle<Executor1> other;
|
|
};
|
|
|
|
/// The native representation of a handle.
|
|
#if defined(GENERATING_DOCUMENTATION)
|
|
typedef implementation_defined native_handle_type;
|
|
#else
|
|
typedef asio::detail::win_iocp_handle_service::native_handle_type
|
|
native_handle_type;
|
|
#endif
|
|
|
|
/// Construct a stream handle without opening it.
|
|
/**
|
|
* This constructor creates a stream handle without opening it.
|
|
*
|
|
* @param ex The I/O executor that the stream handle will use, by default, to
|
|
* dispatch handlers for any asynchronous operations performed on the stream
|
|
* handle.
|
|
*/
|
|
explicit basic_stream_handle(const executor_type& ex)
|
|
: basic_overlapped_handle<Executor>(ex)
|
|
{
|
|
}
|
|
|
|
/// Construct a stream handle without opening it.
|
|
/**
|
|
* This constructor creates a stream handle without opening it. The handle
|
|
* needs to be opened or assigned before data can be written to or read from
|
|
* it.
|
|
*
|
|
* @param context An execution context which provides the I/O executor that
|
|
* the stream handle will use, by default, to dispatch handlers for any
|
|
* asynchronous operations performed on the stream handle.
|
|
*/
|
|
template <typename ExecutionContext>
|
|
explicit basic_stream_handle(ExecutionContext& context,
|
|
typename constraint<
|
|
is_convertible<ExecutionContext&, execution_context&>::value,
|
|
defaulted_constraint
|
|
>::type = defaulted_constraint())
|
|
: basic_overlapped_handle<Executor>(context)
|
|
{
|
|
}
|
|
|
|
/// Construct a stream handle on an existing native handle.
|
|
/**
|
|
* This constructor creates a stream handle object to hold an existing native
|
|
* handle.
|
|
*
|
|
* @param ex The I/O executor that the stream handle will use, by default, to
|
|
* dispatch handlers for any asynchronous operations performed on the stream
|
|
* handle.
|
|
*
|
|
* @param handle The new underlying handle implementation.
|
|
*
|
|
* @throws asio::system_error Thrown on failure.
|
|
*/
|
|
basic_stream_handle(const executor_type& ex, const native_handle_type& handle)
|
|
: basic_overlapped_handle<Executor>(ex, handle)
|
|
{
|
|
}
|
|
|
|
/// Construct a stream handle on an existing native handle.
|
|
/**
|
|
* This constructor creates a stream handle object to hold an existing native
|
|
* handle.
|
|
*
|
|
* @param context An execution context which provides the I/O executor that
|
|
* the stream handle will use, by default, to dispatch handlers for any
|
|
* asynchronous operations performed on the stream handle.
|
|
*
|
|
* @param handle The new underlying handle implementation.
|
|
*
|
|
* @throws asio::system_error Thrown on failure.
|
|
*/
|
|
template <typename ExecutionContext>
|
|
basic_stream_handle(ExecutionContext& context,
|
|
const native_handle_type& handle,
|
|
typename constraint<
|
|
is_convertible<ExecutionContext&, execution_context&>::value
|
|
>::type = 0)
|
|
: basic_overlapped_handle<Executor>(context, handle)
|
|
{
|
|
}
|
|
|
|
#if defined(ASIO_HAS_MOVE) || defined(GENERATING_DOCUMENTATION)
|
|
/// Move-construct a stream handle from another.
|
|
/**
|
|
* This constructor moves a stream handle from one object to another.
|
|
*
|
|
* @param other The other stream handle object from which the move
|
|
* will occur.
|
|
*
|
|
* @note Following the move, the moved-from object is in the same state as if
|
|
* constructed using the @c basic_stream_handle(const executor_type&)
|
|
* constructor.
|
|
*/
|
|
basic_stream_handle(basic_stream_handle&& other)
|
|
: basic_overlapped_handle<Executor>(std::move(other))
|
|
{
|
|
}
|
|
|
|
/// Move-assign a stream handle from another.
|
|
/**
|
|
* This assignment operator moves a stream handle from one object to
|
|
* another.
|
|
*
|
|
* @param other The other stream handle object from which the move will occur.
|
|
*
|
|
* @note Following the move, the moved-from object is in the same state as if
|
|
* constructed using the @c basic_stream_handle(const executor_type&)
|
|
* constructor.
|
|
*/
|
|
basic_stream_handle& operator=(basic_stream_handle&& other)
|
|
{
|
|
basic_overlapped_handle<Executor>::operator=(std::move(other));
|
|
return *this;
|
|
}
|
|
|
|
/// Move-construct a stream handle from a handle of another executor type.
|
|
/**
|
|
* This constructor moves a stream handle from one object to another.
|
|
*
|
|
* @param other The other stream handle object from which the move
|
|
* will occur.
|
|
*
|
|
* @note Following the move, the moved-from object is in the same state as if
|
|
* constructed using the @c basic_stream_handle(const executor_type&)
|
|
* constructor.
|
|
*/
|
|
template<typename Executor1>
|
|
basic_stream_handle(basic_stream_handle<Executor1>&& other,
|
|
typename constraint<
|
|
is_convertible<Executor1, Executor>::value,
|
|
defaulted_constraint
|
|
>::type = defaulted_constraint())
|
|
: basic_overlapped_handle<Executor>(std::move(other))
|
|
{
|
|
}
|
|
|
|
/// Move-assign a stream handle from a handle of another executor type.
|
|
/**
|
|
* This assignment operator moves a stream handle from one object to
|
|
* another.
|
|
*
|
|
* @param other The other stream handle object from which the move will occur.
|
|
*
|
|
* @note Following the move, the moved-from object is in the same state as if
|
|
* constructed using the @c basic_stream_handle(const executor_type&)
|
|
* constructor.
|
|
*/
|
|
template<typename Executor1>
|
|
typename constraint<
|
|
is_convertible<Executor1, Executor>::value,
|
|
basic_stream_handle&
|
|
>::type operator=(basic_stream_handle<Executor1>&& other)
|
|
{
|
|
basic_overlapped_handle<Executor>::operator=(std::move(other));
|
|
return *this;
|
|
}
|
|
#endif // defined(ASIO_HAS_MOVE) || defined(GENERATING_DOCUMENTATION)
|
|
|
|
/// Write some data to the handle.
|
|
/**
|
|
* This function is used to write data to the stream handle. The function call
|
|
* will block until one or more bytes of the data has been written
|
|
* successfully, or until an error occurs.
|
|
*
|
|
* @param buffers One or more data buffers to be written to the handle.
|
|
*
|
|
* @returns The number of bytes written.
|
|
*
|
|
* @throws asio::system_error Thrown on failure. An error code of
|
|
* asio::error::eof indicates that the connection was closed by the
|
|
* peer.
|
|
*
|
|
* @note The write_some operation may not transmit all of the data to the
|
|
* peer. Consider using the @ref write function if you need to ensure that
|
|
* all data is written before the blocking operation completes.
|
|
*
|
|
* @par Example
|
|
* To write a single data buffer use the @ref buffer function as follows:
|
|
* @code
|
|
* handle.write_some(asio::buffer(data, size));
|
|
* @endcode
|
|
* See the @ref buffer documentation for information on writing multiple
|
|
* buffers in one go, and how to use it with arrays, boost::array or
|
|
* std::vector.
|
|
*/
|
|
template <typename ConstBufferSequence>
|
|
std::size_t write_some(const ConstBufferSequence& buffers)
|
|
{
|
|
asio::error_code ec;
|
|
std::size_t s = this->impl_.get_service().write_some(
|
|
this->impl_.get_implementation(), buffers, ec);
|
|
asio::detail::throw_error(ec, "write_some");
|
|
return s;
|
|
}
|
|
|
|
/// Write some data to the handle.
|
|
/**
|
|
* This function is used to write data to the stream handle. The function call
|
|
* will block until one or more bytes of the data has been written
|
|
* successfully, or until an error occurs.
|
|
*
|
|
* @param buffers One or more data buffers to be written to the handle.
|
|
*
|
|
* @param ec Set to indicate what error occurred, if any.
|
|
*
|
|
* @returns The number of bytes written. Returns 0 if an error occurred.
|
|
*
|
|
* @note The write_some operation may not transmit all of the data to the
|
|
* peer. Consider using the @ref write function if you need to ensure that
|
|
* all data is written before the blocking operation completes.
|
|
*/
|
|
template <typename ConstBufferSequence>
|
|
std::size_t write_some(const ConstBufferSequence& buffers,
|
|
asio::error_code& ec)
|
|
{
|
|
return this->impl_.get_service().write_some(
|
|
this->impl_.get_implementation(), buffers, ec);
|
|
}
|
|
|
|
/// Start an asynchronous write.
|
|
/**
|
|
* This function is used to asynchronously write data to the stream handle.
|
|
* It is an initiating function for an @ref asynchronous_operation, and always
|
|
* returns immediately.
|
|
*
|
|
* @param buffers One or more data buffers to be written to the handle.
|
|
* Although the buffers object may be copied as necessary, ownership of the
|
|
* underlying memory blocks is retained by the caller, which must guarantee
|
|
* that they remain valid until the completion handler is called.
|
|
*
|
|
* @param token The @ref completion_token that will be used to produce a
|
|
* completion handler, which will be called when the write completes.
|
|
* Potential completion tokens include @ref use_future, @ref use_awaitable,
|
|
* @ref yield_context, or a function object with the correct completion
|
|
* signature. The function signature of the completion handler must be:
|
|
* @code void handler(
|
|
* const asio::error_code& error, // Result of operation.
|
|
* std::size_t bytes_transferred // Number of bytes written.
|
|
* ); @endcode
|
|
* Regardless of whether the asynchronous operation completes immediately or
|
|
* not, the completion handler will not be invoked from within this function.
|
|
* On immediate completion, invocation of the handler will be performed in a
|
|
* manner equivalent to using asio::post().
|
|
*
|
|
* @par Completion Signature
|
|
* @code void(asio::error_code, std::size_t) @endcode
|
|
*
|
|
* @note The write operation may not transmit all of the data to the peer.
|
|
* Consider using the @ref async_write function if you need to ensure that all
|
|
* data is written before the asynchronous operation completes.
|
|
*
|
|
* @par Example
|
|
* To write a single data buffer use the @ref buffer function as follows:
|
|
* @code
|
|
* handle.async_write_some(asio::buffer(data, size), handler);
|
|
* @endcode
|
|
* See the @ref buffer documentation for information on writing multiple
|
|
* buffers in one go, and how to use it with arrays, boost::array or
|
|
* std::vector.
|
|
*
|
|
* @par Per-Operation Cancellation
|
|
* This asynchronous operation supports cancellation for the following
|
|
* asio::cancellation_type values:
|
|
*
|
|
* @li @c cancellation_type::terminal
|
|
*
|
|
* @li @c cancellation_type::partial
|
|
*
|
|
* @li @c cancellation_type::total
|
|
*/
|
|
template <typename ConstBufferSequence,
|
|
ASIO_COMPLETION_TOKEN_FOR(void (asio::error_code,
|
|
std::size_t)) WriteToken
|
|
ASIO_DEFAULT_COMPLETION_TOKEN_TYPE(executor_type)>
|
|
ASIO_INITFN_AUTO_RESULT_TYPE_PREFIX(WriteToken,
|
|
void (asio::error_code, std::size_t))
|
|
async_write_some(const ConstBufferSequence& buffers,
|
|
ASIO_MOVE_ARG(WriteToken) token
|
|
ASIO_DEFAULT_COMPLETION_TOKEN(executor_type))
|
|
ASIO_INITFN_AUTO_RESULT_TYPE_SUFFIX((
|
|
async_initiate<WriteToken,
|
|
void (asio::error_code, std::size_t)>(
|
|
declval<initiate_async_write_some>(), token, buffers)))
|
|
{
|
|
return async_initiate<WriteToken,
|
|
void (asio::error_code, std::size_t)>(
|
|
initiate_async_write_some(this), token, buffers);
|
|
}
|
|
|
|
/// Read some data from the handle.
|
|
/**
|
|
* This function is used to read data from the stream handle. The function
|
|
* call will block until one or more bytes of data has been read successfully,
|
|
* or until an error occurs.
|
|
*
|
|
* @param buffers One or more buffers into which the data will be read.
|
|
*
|
|
* @returns The number of bytes read.
|
|
*
|
|
* @throws asio::system_error Thrown on failure. An error code of
|
|
* asio::error::eof indicates that the connection was closed by the
|
|
* peer.
|
|
*
|
|
* @note The read_some operation may not read all of the requested number of
|
|
* bytes. Consider using the @ref read function if you need to ensure that
|
|
* the requested amount of data is read before the blocking operation
|
|
* completes.
|
|
*
|
|
* @par Example
|
|
* To read into a single data buffer use the @ref buffer function as follows:
|
|
* @code
|
|
* handle.read_some(asio::buffer(data, size));
|
|
* @endcode
|
|
* See the @ref buffer documentation for information on reading into multiple
|
|
* buffers in one go, and how to use it with arrays, boost::array or
|
|
* std::vector.
|
|
*/
|
|
template <typename MutableBufferSequence>
|
|
std::size_t read_some(const MutableBufferSequence& buffers)
|
|
{
|
|
asio::error_code ec;
|
|
std::size_t s = this->impl_.get_service().read_some(
|
|
this->impl_.get_implementation(), buffers, ec);
|
|
asio::detail::throw_error(ec, "read_some");
|
|
return s;
|
|
}
|
|
|
|
/// Read some data from the handle.
|
|
/**
|
|
* This function is used to read data from the stream handle. The function
|
|
* call will block until one or more bytes of data has been read successfully,
|
|
* or until an error occurs.
|
|
*
|
|
* @param buffers One or more buffers into which the data will be read.
|
|
*
|
|
* @param ec Set to indicate what error occurred, if any.
|
|
*
|
|
* @returns The number of bytes read. Returns 0 if an error occurred.
|
|
*
|
|
* @note The read_some operation may not read all of the requested number of
|
|
* bytes. Consider using the @ref read function if you need to ensure that
|
|
* the requested amount of data is read before the blocking operation
|
|
* completes.
|
|
*/
|
|
template <typename MutableBufferSequence>
|
|
std::size_t read_some(const MutableBufferSequence& buffers,
|
|
asio::error_code& ec)
|
|
{
|
|
return this->impl_.get_service().read_some(
|
|
this->impl_.get_implementation(), buffers, ec);
|
|
}
|
|
|
|
/// Start an asynchronous read.
|
|
/**
|
|
* This function is used to asynchronously read data from the stream handle.
|
|
* It is an initiating function for an @ref asynchronous_operation, and always
|
|
* returns immediately.
|
|
*
|
|
* @param buffers One or more buffers into which the data will be read.
|
|
* Although the buffers object may be copied as necessary, ownership of the
|
|
* underlying memory blocks is retained by the caller, which must guarantee
|
|
* that they remain valid until the completion handler is called.
|
|
*
|
|
* @param token The @ref completion_token that will be used to produce a
|
|
* completion handler, which will be called when the read completes.
|
|
* Potential completion tokens include @ref use_future, @ref use_awaitable,
|
|
* @ref yield_context, or a function object with the correct completion
|
|
* signature. The function signature of the completion handler must be:
|
|
* @code void handler(
|
|
* const asio::error_code& error, // Result of operation.
|
|
* std::size_t bytes_transferred // Number of bytes read.
|
|
* ); @endcode
|
|
* Regardless of whether the asynchronous operation completes immediately or
|
|
* not, the completion handler will not be invoked from within this function.
|
|
* On immediate completion, invocation of the handler will be performed in a
|
|
* manner equivalent to using asio::post().
|
|
*
|
|
* @par Completion Signature
|
|
* @code void(asio::error_code, std::size_t) @endcode
|
|
*
|
|
* @note The read operation may not read all of the requested number of bytes.
|
|
* Consider using the @ref async_read function if you need to ensure that the
|
|
* requested amount of data is read before the asynchronous operation
|
|
* completes.
|
|
*
|
|
* @par Example
|
|
* To read into a single data buffer use the @ref buffer function as follows:
|
|
* @code
|
|
* handle.async_read_some(asio::buffer(data, size), handler);
|
|
* @endcode
|
|
* See the @ref buffer documentation for information on reading into multiple
|
|
* buffers in one go, and how to use it with arrays, boost::array or
|
|
* std::vector.
|
|
*
|
|
* @par Per-Operation Cancellation
|
|
* This asynchronous operation supports cancellation for the following
|
|
* asio::cancellation_type values:
|
|
*
|
|
* @li @c cancellation_type::terminal
|
|
*
|
|
* @li @c cancellation_type::partial
|
|
*
|
|
* @li @c cancellation_type::total
|
|
*/
|
|
template <typename MutableBufferSequence,
|
|
ASIO_COMPLETION_TOKEN_FOR(void (asio::error_code,
|
|
std::size_t)) ReadToken
|
|
ASIO_DEFAULT_COMPLETION_TOKEN_TYPE(executor_type)>
|
|
ASIO_INITFN_AUTO_RESULT_TYPE_PREFIX(ReadToken,
|
|
void (asio::error_code, std::size_t))
|
|
async_read_some(const MutableBufferSequence& buffers,
|
|
ASIO_MOVE_ARG(ReadToken) token
|
|
ASIO_DEFAULT_COMPLETION_TOKEN(executor_type))
|
|
ASIO_INITFN_AUTO_RESULT_TYPE_SUFFIX((
|
|
async_initiate<ReadToken,
|
|
void (asio::error_code, std::size_t)>(
|
|
declval<initiate_async_read_some>(), token, buffers)))
|
|
{
|
|
return async_initiate<ReadToken,
|
|
void (asio::error_code, std::size_t)>(
|
|
initiate_async_read_some(this), token, buffers);
|
|
}
|
|
|
|
private:
|
|
class initiate_async_write_some
|
|
{
|
|
public:
|
|
typedef Executor executor_type;
|
|
|
|
explicit initiate_async_write_some(basic_stream_handle* self)
|
|
: self_(self)
|
|
{
|
|
}
|
|
|
|
const executor_type& get_executor() const ASIO_NOEXCEPT
|
|
{
|
|
return self_->get_executor();
|
|
}
|
|
|
|
template <typename WriteHandler, typename ConstBufferSequence>
|
|
void operator()(ASIO_MOVE_ARG(WriteHandler) handler,
|
|
const ConstBufferSequence& buffers) const
|
|
{
|
|
// If you get an error on the following line it means that your handler
|
|
// does not meet the documented type requirements for a WriteHandler.
|
|
ASIO_WRITE_HANDLER_CHECK(WriteHandler, handler) type_check;
|
|
|
|
detail::non_const_lvalue<WriteHandler> handler2(handler);
|
|
self_->impl_.get_service().async_write_some(
|
|
self_->impl_.get_implementation(), buffers,
|
|
handler2.value, self_->impl_.get_executor());
|
|
}
|
|
|
|
private:
|
|
basic_stream_handle* self_;
|
|
};
|
|
|
|
class initiate_async_read_some
|
|
{
|
|
public:
|
|
typedef Executor executor_type;
|
|
|
|
explicit initiate_async_read_some(basic_stream_handle* self)
|
|
: self_(self)
|
|
{
|
|
}
|
|
|
|
const executor_type& get_executor() const ASIO_NOEXCEPT
|
|
{
|
|
return self_->get_executor();
|
|
}
|
|
|
|
template <typename ReadHandler, typename MutableBufferSequence>
|
|
void operator()(ASIO_MOVE_ARG(ReadHandler) handler,
|
|
const MutableBufferSequence& buffers) const
|
|
{
|
|
// If you get an error on the following line it means that your handler
|
|
// does not meet the documented type requirements for a ReadHandler.
|
|
ASIO_READ_HANDLER_CHECK(ReadHandler, handler) type_check;
|
|
|
|
detail::non_const_lvalue<ReadHandler> handler2(handler);
|
|
self_->impl_.get_service().async_read_some(
|
|
self_->impl_.get_implementation(), buffers,
|
|
handler2.value, self_->impl_.get_executor());
|
|
}
|
|
|
|
private:
|
|
basic_stream_handle* self_;
|
|
};
|
|
};
|
|
|
|
} // namespace windows
|
|
} // namespace asio
|
|
|
|
#include "asio/detail/pop_options.hpp"
|
|
|
|
#endif // defined(ASIO_HAS_WINDOWS_STREAM_HANDLE)
|
|
// || defined(GENERATING_DOCUMENTATION)
|
|
|
|
#endif // ASIO_WINDOWS_BASIC_STREAM_HANDLE_HPP
|