401 lines
11 KiB
C++
401 lines
11 KiB
C++
//
|
|
// detail/impl/select_reactor.ipp
|
|
// ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
|
//
|
|
// 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_DETAIL_IMPL_SELECT_REACTOR_IPP
|
|
#define ASIO_DETAIL_IMPL_SELECT_REACTOR_IPP
|
|
|
|
#if defined(_MSC_VER) && (_MSC_VER >= 1200)
|
|
# pragma once
|
|
#endif // defined(_MSC_VER) && (_MSC_VER >= 1200)
|
|
|
|
#include "asio/detail/config.hpp"
|
|
|
|
#if defined(ASIO_HAS_IOCP) \
|
|
|| (!defined(ASIO_HAS_DEV_POLL) \
|
|
&& !defined(ASIO_HAS_EPOLL) \
|
|
&& !defined(ASIO_HAS_KQUEUE) \
|
|
&& !defined(ASIO_WINDOWS_RUNTIME))
|
|
|
|
#include "asio/detail/fd_set_adapter.hpp"
|
|
#include "asio/detail/select_reactor.hpp"
|
|
#include "asio/detail/signal_blocker.hpp"
|
|
#include "asio/detail/socket_ops.hpp"
|
|
|
|
#if defined(ASIO_HAS_IOCP)
|
|
# include "asio/detail/win_iocp_io_context.hpp"
|
|
#else // defined(ASIO_HAS_IOCP)
|
|
# include "asio/detail/scheduler.hpp"
|
|
#endif // defined(ASIO_HAS_IOCP)
|
|
|
|
#include "asio/detail/push_options.hpp"
|
|
|
|
namespace asio {
|
|
namespace detail {
|
|
|
|
#if defined(ASIO_HAS_IOCP)
|
|
class select_reactor::thread_function
|
|
{
|
|
public:
|
|
explicit thread_function(select_reactor* r)
|
|
: this_(r)
|
|
{
|
|
}
|
|
|
|
void operator()()
|
|
{
|
|
this_->run_thread();
|
|
}
|
|
|
|
private:
|
|
select_reactor* this_;
|
|
};
|
|
#endif // defined(ASIO_HAS_IOCP)
|
|
|
|
select_reactor::select_reactor(asio::execution_context& ctx)
|
|
: execution_context_service_base<select_reactor>(ctx),
|
|
scheduler_(use_service<scheduler_type>(ctx)),
|
|
mutex_(),
|
|
interrupter_(),
|
|
#if defined(ASIO_HAS_IOCP)
|
|
stop_thread_(false),
|
|
thread_(0),
|
|
restart_reactor_(this),
|
|
#endif // defined(ASIO_HAS_IOCP)
|
|
shutdown_(false)
|
|
{
|
|
#if defined(ASIO_HAS_IOCP)
|
|
asio::detail::signal_blocker sb;
|
|
thread_ = new asio::detail::thread(thread_function(this));
|
|
#endif // defined(ASIO_HAS_IOCP)
|
|
}
|
|
|
|
select_reactor::~select_reactor()
|
|
{
|
|
shutdown();
|
|
}
|
|
|
|
void select_reactor::shutdown()
|
|
{
|
|
asio::detail::mutex::scoped_lock lock(mutex_);
|
|
shutdown_ = true;
|
|
#if defined(ASIO_HAS_IOCP)
|
|
stop_thread_ = true;
|
|
if (thread_)
|
|
interrupter_.interrupt();
|
|
#endif // defined(ASIO_HAS_IOCP)
|
|
lock.unlock();
|
|
|
|
#if defined(ASIO_HAS_IOCP)
|
|
if (thread_)
|
|
{
|
|
thread_->join();
|
|
delete thread_;
|
|
thread_ = 0;
|
|
}
|
|
#endif // defined(ASIO_HAS_IOCP)
|
|
|
|
op_queue<operation> ops;
|
|
|
|
for (int i = 0; i < max_ops; ++i)
|
|
op_queue_[i].get_all_operations(ops);
|
|
|
|
timer_queues_.get_all_timers(ops);
|
|
|
|
scheduler_.abandon_operations(ops);
|
|
}
|
|
|
|
void select_reactor::notify_fork(
|
|
asio::execution_context::fork_event fork_ev)
|
|
{
|
|
#if defined(ASIO_HAS_IOCP)
|
|
(void)fork_ev;
|
|
#else // defined(ASIO_HAS_IOCP)
|
|
if (fork_ev == asio::execution_context::fork_child)
|
|
interrupter_.recreate();
|
|
#endif // defined(ASIO_HAS_IOCP)
|
|
}
|
|
|
|
void select_reactor::init_task()
|
|
{
|
|
scheduler_.init_task();
|
|
}
|
|
|
|
int select_reactor::register_descriptor(socket_type,
|
|
select_reactor::per_descriptor_data&)
|
|
{
|
|
return 0;
|
|
}
|
|
|
|
int select_reactor::register_internal_descriptor(
|
|
int op_type, socket_type descriptor,
|
|
select_reactor::per_descriptor_data&, reactor_op* op)
|
|
{
|
|
asio::detail::mutex::scoped_lock lock(mutex_);
|
|
|
|
op_queue_[op_type].enqueue_operation(descriptor, op);
|
|
interrupter_.interrupt();
|
|
|
|
return 0;
|
|
}
|
|
|
|
void select_reactor::move_descriptor(socket_type,
|
|
select_reactor::per_descriptor_data&,
|
|
select_reactor::per_descriptor_data&)
|
|
{
|
|
}
|
|
|
|
void select_reactor::call_post_immediate_completion(
|
|
operation* op, bool is_continuation, const void* self)
|
|
{
|
|
static_cast<const select_reactor*>(self)->post_immediate_completion(
|
|
op, is_continuation);
|
|
}
|
|
|
|
void select_reactor::start_op(int op_type, socket_type descriptor,
|
|
select_reactor::per_descriptor_data&, reactor_op* op, bool is_continuation,
|
|
bool, void (*on_immediate)(operation*, bool, const void*),
|
|
const void* immediate_arg)
|
|
{
|
|
asio::detail::mutex::scoped_lock lock(mutex_);
|
|
|
|
if (shutdown_)
|
|
{
|
|
on_immediate(op, is_continuation, immediate_arg);
|
|
return;
|
|
}
|
|
|
|
bool first = op_queue_[op_type].enqueue_operation(descriptor, op);
|
|
scheduler_.work_started();
|
|
if (first)
|
|
interrupter_.interrupt();
|
|
}
|
|
|
|
void select_reactor::cancel_ops(socket_type descriptor,
|
|
select_reactor::per_descriptor_data&)
|
|
{
|
|
asio::detail::mutex::scoped_lock lock(mutex_);
|
|
cancel_ops_unlocked(descriptor, asio::error::operation_aborted);
|
|
}
|
|
|
|
void select_reactor::cancel_ops_by_key(socket_type descriptor,
|
|
select_reactor::per_descriptor_data&,
|
|
int op_type, void* cancellation_key)
|
|
{
|
|
asio::detail::mutex::scoped_lock lock(mutex_);
|
|
op_queue<operation> ops;
|
|
bool need_interrupt = op_queue_[op_type].cancel_operations_by_key(
|
|
descriptor, ops, cancellation_key, asio::error::operation_aborted);
|
|
scheduler_.post_deferred_completions(ops);
|
|
if (need_interrupt)
|
|
interrupter_.interrupt();
|
|
}
|
|
|
|
void select_reactor::deregister_descriptor(socket_type descriptor,
|
|
select_reactor::per_descriptor_data&, bool)
|
|
{
|
|
asio::detail::mutex::scoped_lock lock(mutex_);
|
|
cancel_ops_unlocked(descriptor, asio::error::operation_aborted);
|
|
}
|
|
|
|
void select_reactor::deregister_internal_descriptor(
|
|
socket_type descriptor, select_reactor::per_descriptor_data&)
|
|
{
|
|
asio::detail::mutex::scoped_lock lock(mutex_);
|
|
op_queue<operation> ops;
|
|
for (int i = 0; i < max_ops; ++i)
|
|
op_queue_[i].cancel_operations(descriptor, ops);
|
|
}
|
|
|
|
void select_reactor::cleanup_descriptor_data(
|
|
select_reactor::per_descriptor_data&)
|
|
{
|
|
}
|
|
|
|
void select_reactor::run(long usec, op_queue<operation>& ops)
|
|
{
|
|
asio::detail::mutex::scoped_lock lock(mutex_);
|
|
|
|
#if defined(ASIO_HAS_IOCP)
|
|
// Check if the thread is supposed to stop.
|
|
if (stop_thread_)
|
|
return;
|
|
#endif // defined(ASIO_HAS_IOCP)
|
|
|
|
// Set up the descriptor sets.
|
|
for (int i = 0; i < max_select_ops; ++i)
|
|
fd_sets_[i].reset();
|
|
fd_sets_[read_op].set(interrupter_.read_descriptor());
|
|
socket_type max_fd = 0;
|
|
bool have_work_to_do = !timer_queues_.all_empty();
|
|
for (int i = 0; i < max_select_ops; ++i)
|
|
{
|
|
have_work_to_do = have_work_to_do || !op_queue_[i].empty();
|
|
fd_sets_[i].set(op_queue_[i], ops);
|
|
if (fd_sets_[i].max_descriptor() > max_fd)
|
|
max_fd = fd_sets_[i].max_descriptor();
|
|
}
|
|
|
|
#if defined(ASIO_WINDOWS) || defined(__CYGWIN__)
|
|
// Connection operations on Windows use both except and write fd_sets.
|
|
have_work_to_do = have_work_to_do || !op_queue_[connect_op].empty();
|
|
fd_sets_[write_op].set(op_queue_[connect_op], ops);
|
|
if (fd_sets_[write_op].max_descriptor() > max_fd)
|
|
max_fd = fd_sets_[write_op].max_descriptor();
|
|
fd_sets_[except_op].set(op_queue_[connect_op], ops);
|
|
if (fd_sets_[except_op].max_descriptor() > max_fd)
|
|
max_fd = fd_sets_[except_op].max_descriptor();
|
|
#endif // defined(ASIO_WINDOWS) || defined(__CYGWIN__)
|
|
|
|
// We can return immediately if there's no work to do and the reactor is
|
|
// not supposed to block.
|
|
if (!usec && !have_work_to_do)
|
|
return;
|
|
|
|
// Determine how long to block while waiting for events.
|
|
timeval tv_buf = { 0, 0 };
|
|
timeval* tv = usec ? get_timeout(usec, tv_buf) : &tv_buf;
|
|
|
|
lock.unlock();
|
|
|
|
// Block on the select call until descriptors become ready.
|
|
asio::error_code ec;
|
|
int retval = socket_ops::select(static_cast<int>(max_fd + 1),
|
|
fd_sets_[read_op], fd_sets_[write_op], fd_sets_[except_op], tv, ec);
|
|
|
|
// Reset the interrupter.
|
|
if (retval > 0 && fd_sets_[read_op].is_set(interrupter_.read_descriptor()))
|
|
{
|
|
if (!interrupter_.reset())
|
|
{
|
|
lock.lock();
|
|
#if defined(ASIO_HAS_IOCP)
|
|
stop_thread_ = true;
|
|
scheduler_.post_immediate_completion(&restart_reactor_, false);
|
|
#else // defined(ASIO_HAS_IOCP)
|
|
interrupter_.recreate();
|
|
#endif // defined(ASIO_HAS_IOCP)
|
|
}
|
|
--retval;
|
|
}
|
|
|
|
lock.lock();
|
|
|
|
// Dispatch all ready operations.
|
|
if (retval > 0)
|
|
{
|
|
#if defined(ASIO_WINDOWS) || defined(__CYGWIN__)
|
|
// Connection operations on Windows use both except and write fd_sets.
|
|
fd_sets_[except_op].perform(op_queue_[connect_op], ops);
|
|
fd_sets_[write_op].perform(op_queue_[connect_op], ops);
|
|
#endif // defined(ASIO_WINDOWS) || defined(__CYGWIN__)
|
|
|
|
// Exception operations must be processed first to ensure that any
|
|
// out-of-band data is read before normal data.
|
|
for (int i = max_select_ops - 1; i >= 0; --i)
|
|
fd_sets_[i].perform(op_queue_[i], ops);
|
|
}
|
|
timer_queues_.get_ready_timers(ops);
|
|
}
|
|
|
|
void select_reactor::interrupt()
|
|
{
|
|
interrupter_.interrupt();
|
|
}
|
|
|
|
#if defined(ASIO_HAS_IOCP)
|
|
void select_reactor::run_thread()
|
|
{
|
|
asio::detail::mutex::scoped_lock lock(mutex_);
|
|
while (!stop_thread_)
|
|
{
|
|
lock.unlock();
|
|
op_queue<operation> ops;
|
|
run(-1, ops);
|
|
scheduler_.post_deferred_completions(ops);
|
|
lock.lock();
|
|
}
|
|
}
|
|
|
|
void select_reactor::restart_reactor::do_complete(void* owner, operation* base,
|
|
const asio::error_code& /*ec*/, std::size_t /*bytes_transferred*/)
|
|
{
|
|
if (owner)
|
|
{
|
|
select_reactor* reactor = static_cast<restart_reactor*>(base)->reactor_;
|
|
|
|
if (reactor->thread_)
|
|
{
|
|
reactor->thread_->join();
|
|
delete reactor->thread_;
|
|
reactor->thread_ = 0;
|
|
}
|
|
|
|
asio::detail::mutex::scoped_lock lock(reactor->mutex_);
|
|
reactor->interrupter_.recreate();
|
|
reactor->stop_thread_ = false;
|
|
lock.unlock();
|
|
|
|
asio::detail::signal_blocker sb;
|
|
reactor->thread_ =
|
|
new asio::detail::thread(thread_function(reactor));
|
|
}
|
|
}
|
|
#endif // defined(ASIO_HAS_IOCP)
|
|
|
|
void select_reactor::do_add_timer_queue(timer_queue_base& queue)
|
|
{
|
|
mutex::scoped_lock lock(mutex_);
|
|
timer_queues_.insert(&queue);
|
|
}
|
|
|
|
void select_reactor::do_remove_timer_queue(timer_queue_base& queue)
|
|
{
|
|
mutex::scoped_lock lock(mutex_);
|
|
timer_queues_.erase(&queue);
|
|
}
|
|
|
|
timeval* select_reactor::get_timeout(long usec, timeval& tv)
|
|
{
|
|
// By default we will wait no longer than 5 minutes. This will ensure that
|
|
// any changes to the system clock are detected after no longer than this.
|
|
const long max_usec = 5 * 60 * 1000 * 1000;
|
|
usec = timer_queues_.wait_duration_usec(
|
|
(usec < 0 || max_usec < usec) ? max_usec : usec);
|
|
tv.tv_sec = usec / 1000000;
|
|
tv.tv_usec = usec % 1000000;
|
|
return &tv;
|
|
}
|
|
|
|
void select_reactor::cancel_ops_unlocked(socket_type descriptor,
|
|
const asio::error_code& ec)
|
|
{
|
|
bool need_interrupt = false;
|
|
op_queue<operation> ops;
|
|
for (int i = 0; i < max_ops; ++i)
|
|
need_interrupt = op_queue_[i].cancel_operations(
|
|
descriptor, ops, ec) || need_interrupt;
|
|
scheduler_.post_deferred_completions(ops);
|
|
if (need_interrupt)
|
|
interrupter_.interrupt();
|
|
}
|
|
|
|
} // namespace detail
|
|
} // namespace asio
|
|
|
|
#include "asio/detail/pop_options.hpp"
|
|
|
|
#endif // defined(ASIO_HAS_IOCP)
|
|
// || (!defined(ASIO_HAS_DEV_POLL)
|
|
// && !defined(ASIO_HAS_EPOLL)
|
|
// && !defined(ASIO_HAS_KQUEUE))
|
|
// && !defined(ASIO_WINDOWS_RUNTIME))
|
|
|
|
#endif // ASIO_DETAIL_IMPL_SELECT_REACTOR_IPP
|