123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826 |
- //
- // detail/impl/epoll_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_EPOLL_REACTOR_IPP
- #define ASIO_DETAIL_IMPL_EPOLL_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_EPOLL)
- #include <cstddef>
- #include <sys/epoll.h>
- #include "asio/detail/epoll_reactor.hpp"
- #include "asio/detail/scheduler.hpp"
- #include "asio/detail/throw_error.hpp"
- #include "asio/error.hpp"
- #if defined(ASIO_HAS_TIMERFD)
- # include <sys/timerfd.h>
- #endif // defined(ASIO_HAS_TIMERFD)
- #include "asio/detail/push_options.hpp"
- namespace asio {
- namespace detail {
- epoll_reactor::epoll_reactor(asio::execution_context& ctx)
- : execution_context_service_base<epoll_reactor>(ctx),
- scheduler_(use_service<scheduler>(ctx)),
- mutex_(ASIO_CONCURRENCY_HINT_IS_LOCKING(
- REACTOR_REGISTRATION, scheduler_.concurrency_hint())),
- interrupter_(),
- epoll_fd_(do_epoll_create()),
- timer_fd_(do_timerfd_create()),
- shutdown_(false),
- registered_descriptors_mutex_(mutex_.enabled())
- {
- // Add the interrupter's descriptor to epoll.
- epoll_event ev = { 0, { 0 } };
- ev.events = EPOLLIN | EPOLLERR | EPOLLET;
- ev.data.ptr = &interrupter_;
- epoll_ctl(epoll_fd_, EPOLL_CTL_ADD, interrupter_.read_descriptor(), &ev);
- interrupter_.interrupt();
- // Add the timer descriptor to epoll.
- if (timer_fd_ != -1)
- {
- ev.events = EPOLLIN | EPOLLERR;
- ev.data.ptr = &timer_fd_;
- epoll_ctl(epoll_fd_, EPOLL_CTL_ADD, timer_fd_, &ev);
- }
- }
- epoll_reactor::~epoll_reactor()
- {
- if (epoll_fd_ != -1)
- close(epoll_fd_);
- if (timer_fd_ != -1)
- close(timer_fd_);
- }
- void epoll_reactor::shutdown()
- {
- mutex::scoped_lock lock(mutex_);
- shutdown_ = true;
- lock.unlock();
- op_queue<operation> ops;
- while (descriptor_state* state = registered_descriptors_.first())
- {
- for (int i = 0; i < max_ops; ++i)
- ops.push(state->op_queue_[i]);
- state->shutdown_ = true;
- registered_descriptors_.free(state);
- }
- timer_queues_.get_all_timers(ops);
- scheduler_.abandon_operations(ops);
- }
- void epoll_reactor::notify_fork(
- asio::execution_context::fork_event fork_ev)
- {
- if (fork_ev == asio::execution_context::fork_child)
- {
- if (epoll_fd_ != -1)
- ::close(epoll_fd_);
- epoll_fd_ = -1;
- epoll_fd_ = do_epoll_create();
- if (timer_fd_ != -1)
- ::close(timer_fd_);
- timer_fd_ = -1;
- timer_fd_ = do_timerfd_create();
- interrupter_.recreate();
- // Add the interrupter's descriptor to epoll.
- epoll_event ev = { 0, { 0 } };
- ev.events = EPOLLIN | EPOLLERR | EPOLLET;
- ev.data.ptr = &interrupter_;
- epoll_ctl(epoll_fd_, EPOLL_CTL_ADD, interrupter_.read_descriptor(), &ev);
- interrupter_.interrupt();
- // Add the timer descriptor to epoll.
- if (timer_fd_ != -1)
- {
- ev.events = EPOLLIN | EPOLLERR;
- ev.data.ptr = &timer_fd_;
- epoll_ctl(epoll_fd_, EPOLL_CTL_ADD, timer_fd_, &ev);
- }
- update_timeout();
- // Re-register all descriptors with epoll.
- mutex::scoped_lock descriptors_lock(registered_descriptors_mutex_);
- for (descriptor_state* state = registered_descriptors_.first();
- state != 0; state = state->next_)
- {
- ev.events = state->registered_events_;
- ev.data.ptr = state;
- int result = epoll_ctl(epoll_fd_, EPOLL_CTL_ADD, state->descriptor_, &ev);
- if (result != 0)
- {
- asio::error_code ec(errno,
- asio::error::get_system_category());
- asio::detail::throw_error(ec, "epoll re-registration");
- }
- }
- }
- }
- void epoll_reactor::init_task()
- {
- scheduler_.init_task();
- }
- int epoll_reactor::register_descriptor(socket_type descriptor,
- epoll_reactor::per_descriptor_data& descriptor_data)
- {
- descriptor_data = allocate_descriptor_state();
- ASIO_HANDLER_REACTOR_REGISTRATION((
- context(), static_cast<uintmax_t>(descriptor),
- reinterpret_cast<uintmax_t>(descriptor_data)));
- {
- mutex::scoped_lock descriptor_lock(descriptor_data->mutex_);
- descriptor_data->reactor_ = this;
- descriptor_data->descriptor_ = descriptor;
- descriptor_data->shutdown_ = false;
- for (int i = 0; i < max_ops; ++i)
- descriptor_data->try_speculative_[i] = true;
- }
- epoll_event ev = { 0, { 0 } };
- ev.events = EPOLLIN | EPOLLERR | EPOLLHUP | EPOLLPRI | EPOLLET;
- descriptor_data->registered_events_ = ev.events;
- ev.data.ptr = descriptor_data;
- int result = epoll_ctl(epoll_fd_, EPOLL_CTL_ADD, descriptor, &ev);
- if (result != 0)
- {
- if (errno == EPERM)
- {
- // This file descriptor type is not supported by epoll. However, if it is
- // a regular file then operations on it will not block. We will allow
- // this descriptor to be used and fail later if an operation on it would
- // otherwise require a trip through the reactor.
- descriptor_data->registered_events_ = 0;
- return 0;
- }
- return errno;
- }
- return 0;
- }
- int epoll_reactor::register_internal_descriptor(
- int op_type, socket_type descriptor,
- epoll_reactor::per_descriptor_data& descriptor_data, reactor_op* op)
- {
- descriptor_data = allocate_descriptor_state();
- ASIO_HANDLER_REACTOR_REGISTRATION((
- context(), static_cast<uintmax_t>(descriptor),
- reinterpret_cast<uintmax_t>(descriptor_data)));
- {
- mutex::scoped_lock descriptor_lock(descriptor_data->mutex_);
- descriptor_data->reactor_ = this;
- descriptor_data->descriptor_ = descriptor;
- descriptor_data->shutdown_ = false;
- descriptor_data->op_queue_[op_type].push(op);
- for (int i = 0; i < max_ops; ++i)
- descriptor_data->try_speculative_[i] = true;
- }
- epoll_event ev = { 0, { 0 } };
- ev.events = EPOLLIN | EPOLLERR | EPOLLHUP | EPOLLPRI | EPOLLET;
- descriptor_data->registered_events_ = ev.events;
- ev.data.ptr = descriptor_data;
- int result = epoll_ctl(epoll_fd_, EPOLL_CTL_ADD, descriptor, &ev);
- if (result != 0)
- return errno;
- return 0;
- }
- void epoll_reactor::move_descriptor(socket_type,
- epoll_reactor::per_descriptor_data& target_descriptor_data,
- epoll_reactor::per_descriptor_data& source_descriptor_data)
- {
- target_descriptor_data = source_descriptor_data;
- source_descriptor_data = 0;
- }
- void epoll_reactor::call_post_immediate_completion(
- operation* op, bool is_continuation, const void* self)
- {
- static_cast<const epoll_reactor*>(self)->post_immediate_completion(
- op, is_continuation);
- }
- void epoll_reactor::start_op(int op_type, socket_type descriptor,
- epoll_reactor::per_descriptor_data& descriptor_data, reactor_op* op,
- bool is_continuation, bool allow_speculative,
- void (*on_immediate)(operation*, bool, const void*),
- const void* immediate_arg)
- {
- if (!descriptor_data)
- {
- op->ec_ = asio::error::bad_descriptor;
- on_immediate(op, is_continuation, immediate_arg);
- return;
- }
- mutex::scoped_lock descriptor_lock(descriptor_data->mutex_);
- if (descriptor_data->shutdown_)
- {
- on_immediate(op, is_continuation, immediate_arg);
- return;
- }
- if (descriptor_data->op_queue_[op_type].empty())
- {
- if (allow_speculative
- && (op_type != read_op
- || descriptor_data->op_queue_[except_op].empty()))
- {
- if (descriptor_data->try_speculative_[op_type])
- {
- if (reactor_op::status status = op->perform())
- {
- if (status == reactor_op::done_and_exhausted)
- if (descriptor_data->registered_events_ != 0)
- descriptor_data->try_speculative_[op_type] = false;
- descriptor_lock.unlock();
- on_immediate(op, is_continuation, immediate_arg);
- return;
- }
- }
- if (descriptor_data->registered_events_ == 0)
- {
- op->ec_ = asio::error::operation_not_supported;
- on_immediate(op, is_continuation, immediate_arg);
- return;
- }
- if (op_type == write_op)
- {
- if ((descriptor_data->registered_events_ & EPOLLOUT) == 0)
- {
- epoll_event ev = { 0, { 0 } };
- ev.events = descriptor_data->registered_events_ | EPOLLOUT;
- ev.data.ptr = descriptor_data;
- if (epoll_ctl(epoll_fd_, EPOLL_CTL_MOD, descriptor, &ev) == 0)
- {
- descriptor_data->registered_events_ |= ev.events;
- }
- else
- {
- op->ec_ = asio::error_code(errno,
- asio::error::get_system_category());
- on_immediate(op, is_continuation, immediate_arg);
- return;
- }
- }
- }
- }
- else if (descriptor_data->registered_events_ == 0)
- {
- op->ec_ = asio::error::operation_not_supported;
- on_immediate(op, is_continuation, immediate_arg);
- return;
- }
- else
- {
- if (op_type == write_op)
- {
- descriptor_data->registered_events_ |= EPOLLOUT;
- }
- epoll_event ev = { 0, { 0 } };
- ev.events = descriptor_data->registered_events_;
- ev.data.ptr = descriptor_data;
- epoll_ctl(epoll_fd_, EPOLL_CTL_MOD, descriptor, &ev);
- }
- }
- descriptor_data->op_queue_[op_type].push(op);
- scheduler_.work_started();
- }
- void epoll_reactor::cancel_ops(socket_type,
- epoll_reactor::per_descriptor_data& descriptor_data)
- {
- if (!descriptor_data)
- return;
- mutex::scoped_lock descriptor_lock(descriptor_data->mutex_);
- op_queue<operation> ops;
- for (int i = 0; i < max_ops; ++i)
- {
- while (reactor_op* op = descriptor_data->op_queue_[i].front())
- {
- op->ec_ = asio::error::operation_aborted;
- descriptor_data->op_queue_[i].pop();
- ops.push(op);
- }
- }
- descriptor_lock.unlock();
- scheduler_.post_deferred_completions(ops);
- }
- void epoll_reactor::cancel_ops_by_key(socket_type,
- epoll_reactor::per_descriptor_data& descriptor_data,
- int op_type, void* cancellation_key)
- {
- if (!descriptor_data)
- return;
- mutex::scoped_lock descriptor_lock(descriptor_data->mutex_);
- op_queue<operation> ops;
- op_queue<reactor_op> other_ops;
- while (reactor_op* op = descriptor_data->op_queue_[op_type].front())
- {
- descriptor_data->op_queue_[op_type].pop();
- if (op->cancellation_key_ == cancellation_key)
- {
- op->ec_ = asio::error::operation_aborted;
- ops.push(op);
- }
- else
- other_ops.push(op);
- }
- descriptor_data->op_queue_[op_type].push(other_ops);
- descriptor_lock.unlock();
- scheduler_.post_deferred_completions(ops);
- }
- void epoll_reactor::deregister_descriptor(socket_type descriptor,
- epoll_reactor::per_descriptor_data& descriptor_data, bool closing)
- {
- if (!descriptor_data)
- return;
- mutex::scoped_lock descriptor_lock(descriptor_data->mutex_);
- if (!descriptor_data->shutdown_)
- {
- if (closing)
- {
- // The descriptor will be automatically removed from the epoll set when
- // it is closed.
- }
- else if (descriptor_data->registered_events_ != 0)
- {
- epoll_event ev = { 0, { 0 } };
- epoll_ctl(epoll_fd_, EPOLL_CTL_DEL, descriptor, &ev);
- }
- op_queue<operation> ops;
- for (int i = 0; i < max_ops; ++i)
- {
- while (reactor_op* op = descriptor_data->op_queue_[i].front())
- {
- op->ec_ = asio::error::operation_aborted;
- descriptor_data->op_queue_[i].pop();
- ops.push(op);
- }
- }
- descriptor_data->descriptor_ = -1;
- descriptor_data->shutdown_ = true;
- descriptor_lock.unlock();
- ASIO_HANDLER_REACTOR_DEREGISTRATION((
- context(), static_cast<uintmax_t>(descriptor),
- reinterpret_cast<uintmax_t>(descriptor_data)));
- scheduler_.post_deferred_completions(ops);
- // Leave descriptor_data set so that it will be freed by the subsequent
- // call to cleanup_descriptor_data.
- }
- else
- {
- // We are shutting down, so prevent cleanup_descriptor_data from freeing
- // the descriptor_data object and let the destructor free it instead.
- descriptor_data = 0;
- }
- }
- void epoll_reactor::deregister_internal_descriptor(socket_type descriptor,
- epoll_reactor::per_descriptor_data& descriptor_data)
- {
- if (!descriptor_data)
- return;
- mutex::scoped_lock descriptor_lock(descriptor_data->mutex_);
- if (!descriptor_data->shutdown_)
- {
- epoll_event ev = { 0, { 0 } };
- epoll_ctl(epoll_fd_, EPOLL_CTL_DEL, descriptor, &ev);
- op_queue<operation> ops;
- for (int i = 0; i < max_ops; ++i)
- ops.push(descriptor_data->op_queue_[i]);
- descriptor_data->descriptor_ = -1;
- descriptor_data->shutdown_ = true;
- descriptor_lock.unlock();
- ASIO_HANDLER_REACTOR_DEREGISTRATION((
- context(), static_cast<uintmax_t>(descriptor),
- reinterpret_cast<uintmax_t>(descriptor_data)));
- // Leave descriptor_data set so that it will be freed by the subsequent
- // call to cleanup_descriptor_data.
- }
- else
- {
- // We are shutting down, so prevent cleanup_descriptor_data from freeing
- // the descriptor_data object and let the destructor free it instead.
- descriptor_data = 0;
- }
- }
- void epoll_reactor::cleanup_descriptor_data(
- per_descriptor_data& descriptor_data)
- {
- if (descriptor_data)
- {
- free_descriptor_state(descriptor_data);
- descriptor_data = 0;
- }
- }
- void epoll_reactor::run(long usec, op_queue<operation>& ops)
- {
- // This code relies on the fact that the scheduler queues the reactor task
- // behind all descriptor operations generated by this function. This means,
- // that by the time we reach this point, any previously returned descriptor
- // operations have already been dequeued. Therefore it is now safe for us to
- // reuse and return them for the scheduler to queue again.
- // Calculate timeout. Check the timer queues only if timerfd is not in use.
- int timeout;
- if (usec == 0)
- timeout = 0;
- else
- {
- timeout = (usec < 0) ? -1 : ((usec - 1) / 1000 + 1);
- if (timer_fd_ == -1)
- {
- mutex::scoped_lock lock(mutex_);
- timeout = get_timeout(timeout);
- }
- }
- // Block on the epoll descriptor.
- epoll_event events[128];
- int num_events = epoll_wait(epoll_fd_, events, 128, timeout);
- #if defined(ASIO_ENABLE_HANDLER_TRACKING)
- // Trace the waiting events.
- for (int i = 0; i < num_events; ++i)
- {
- void* ptr = events[i].data.ptr;
- if (ptr == &interrupter_)
- {
- // Ignore.
- }
- # if defined(ASIO_HAS_TIMERFD)
- else if (ptr == &timer_fd_)
- {
- // Ignore.
- }
- # endif // defined(ASIO_HAS_TIMERFD)
- else
- {
- unsigned event_mask = 0;
- if ((events[i].events & EPOLLIN) != 0)
- event_mask |= ASIO_HANDLER_REACTOR_READ_EVENT;
- if ((events[i].events & EPOLLOUT))
- event_mask |= ASIO_HANDLER_REACTOR_WRITE_EVENT;
- if ((events[i].events & (EPOLLERR | EPOLLHUP)) != 0)
- event_mask |= ASIO_HANDLER_REACTOR_ERROR_EVENT;
- ASIO_HANDLER_REACTOR_EVENTS((context(),
- reinterpret_cast<uintmax_t>(ptr), event_mask));
- }
- }
- #endif // defined(ASIO_ENABLE_HANDLER_TRACKING)
- #if defined(ASIO_HAS_TIMERFD)
- bool check_timers = (timer_fd_ == -1);
- #else // defined(ASIO_HAS_TIMERFD)
- bool check_timers = true;
- #endif // defined(ASIO_HAS_TIMERFD)
- // Dispatch the waiting events.
- for (int i = 0; i < num_events; ++i)
- {
- void* ptr = events[i].data.ptr;
- if (ptr == &interrupter_)
- {
- // No need to reset the interrupter since we're leaving the descriptor
- // in a ready-to-read state and relying on edge-triggered notifications
- // to make it so that we only get woken up when the descriptor's epoll
- // registration is updated.
- #if defined(ASIO_HAS_TIMERFD)
- if (timer_fd_ == -1)
- check_timers = true;
- #else // defined(ASIO_HAS_TIMERFD)
- check_timers = true;
- #endif // defined(ASIO_HAS_TIMERFD)
- }
- #if defined(ASIO_HAS_TIMERFD)
- else if (ptr == &timer_fd_)
- {
- check_timers = true;
- }
- #endif // defined(ASIO_HAS_TIMERFD)
- else
- {
- // The descriptor operation doesn't count as work in and of itself, so we
- // don't call work_started() here. This still allows the scheduler to
- // stop if the only remaining operations are descriptor operations.
- descriptor_state* descriptor_data = static_cast<descriptor_state*>(ptr);
- if (!ops.is_enqueued(descriptor_data))
- {
- descriptor_data->set_ready_events(events[i].events);
- ops.push(descriptor_data);
- }
- else
- {
- descriptor_data->add_ready_events(events[i].events);
- }
- }
- }
- if (check_timers)
- {
- mutex::scoped_lock common_lock(mutex_);
- timer_queues_.get_ready_timers(ops);
- #if defined(ASIO_HAS_TIMERFD)
- if (timer_fd_ != -1)
- {
- itimerspec new_timeout;
- itimerspec old_timeout;
- int flags = get_timeout(new_timeout);
- timerfd_settime(timer_fd_, flags, &new_timeout, &old_timeout);
- }
- #endif // defined(ASIO_HAS_TIMERFD)
- }
- }
- void epoll_reactor::interrupt()
- {
- epoll_event ev = { 0, { 0 } };
- ev.events = EPOLLIN | EPOLLERR | EPOLLET;
- ev.data.ptr = &interrupter_;
- epoll_ctl(epoll_fd_, EPOLL_CTL_MOD, interrupter_.read_descriptor(), &ev);
- }
- int epoll_reactor::do_epoll_create()
- {
- #if defined(EPOLL_CLOEXEC)
- int fd = epoll_create1(EPOLL_CLOEXEC);
- #else // defined(EPOLL_CLOEXEC)
- int fd = -1;
- errno = EINVAL;
- #endif // defined(EPOLL_CLOEXEC)
- if (fd == -1 && (errno == EINVAL || errno == ENOSYS))
- {
- fd = epoll_create(epoll_size);
- if (fd != -1)
- ::fcntl(fd, F_SETFD, FD_CLOEXEC);
- }
- if (fd == -1)
- {
- asio::error_code ec(errno,
- asio::error::get_system_category());
- asio::detail::throw_error(ec, "epoll");
- }
- return fd;
- }
- int epoll_reactor::do_timerfd_create()
- {
- #if defined(ASIO_HAS_TIMERFD)
- # if defined(TFD_CLOEXEC)
- int fd = timerfd_create(CLOCK_MONOTONIC, TFD_CLOEXEC);
- # else // defined(TFD_CLOEXEC)
- int fd = -1;
- errno = EINVAL;
- # endif // defined(TFD_CLOEXEC)
- if (fd == -1 && errno == EINVAL)
- {
- fd = timerfd_create(CLOCK_MONOTONIC, 0);
- if (fd != -1)
- ::fcntl(fd, F_SETFD, FD_CLOEXEC);
- }
- return fd;
- #else // defined(ASIO_HAS_TIMERFD)
- return -1;
- #endif // defined(ASIO_HAS_TIMERFD)
- }
- epoll_reactor::descriptor_state* epoll_reactor::allocate_descriptor_state()
- {
- mutex::scoped_lock descriptors_lock(registered_descriptors_mutex_);
- return registered_descriptors_.alloc(ASIO_CONCURRENCY_HINT_IS_LOCKING(
- REACTOR_IO, scheduler_.concurrency_hint()));
- }
- void epoll_reactor::free_descriptor_state(epoll_reactor::descriptor_state* s)
- {
- mutex::scoped_lock descriptors_lock(registered_descriptors_mutex_);
- registered_descriptors_.free(s);
- }
- void epoll_reactor::do_add_timer_queue(timer_queue_base& queue)
- {
- mutex::scoped_lock lock(mutex_);
- timer_queues_.insert(&queue);
- }
- void epoll_reactor::do_remove_timer_queue(timer_queue_base& queue)
- {
- mutex::scoped_lock lock(mutex_);
- timer_queues_.erase(&queue);
- }
- void epoll_reactor::update_timeout()
- {
- #if defined(ASIO_HAS_TIMERFD)
- if (timer_fd_ != -1)
- {
- itimerspec new_timeout;
- itimerspec old_timeout;
- int flags = get_timeout(new_timeout);
- timerfd_settime(timer_fd_, flags, &new_timeout, &old_timeout);
- return;
- }
- #endif // defined(ASIO_HAS_TIMERFD)
- interrupt();
- }
- int epoll_reactor::get_timeout(int msec)
- {
- // 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 int max_msec = 5 * 60 * 1000;
- return timer_queues_.wait_duration_msec(
- (msec < 0 || max_msec < msec) ? max_msec : msec);
- }
- #if defined(ASIO_HAS_TIMERFD)
- int epoll_reactor::get_timeout(itimerspec& ts)
- {
- ts.it_interval.tv_sec = 0;
- ts.it_interval.tv_nsec = 0;
- long usec = timer_queues_.wait_duration_usec(5 * 60 * 1000 * 1000);
- ts.it_value.tv_sec = usec / 1000000;
- ts.it_value.tv_nsec = usec ? (usec % 1000000) * 1000 : 1;
- return usec ? 0 : TFD_TIMER_ABSTIME;
- }
- #endif // defined(ASIO_HAS_TIMERFD)
- struct epoll_reactor::perform_io_cleanup_on_block_exit
- {
- explicit perform_io_cleanup_on_block_exit(epoll_reactor* r)
- : reactor_(r), first_op_(0)
- {
- }
- ~perform_io_cleanup_on_block_exit()
- {
- if (first_op_)
- {
- // Post the remaining completed operations for invocation.
- if (!ops_.empty())
- reactor_->scheduler_.post_deferred_completions(ops_);
- // A user-initiated operation has completed, but there's no need to
- // explicitly call work_finished() here. Instead, we'll take advantage of
- // the fact that the scheduler will call work_finished() once we return.
- }
- else
- {
- // No user-initiated operations have completed, so we need to compensate
- // for the work_finished() call that the scheduler will make once this
- // operation returns.
- reactor_->scheduler_.compensating_work_started();
- }
- }
- epoll_reactor* reactor_;
- op_queue<operation> ops_;
- operation* first_op_;
- };
- epoll_reactor::descriptor_state::descriptor_state(bool locking)
- : operation(&epoll_reactor::descriptor_state::do_complete),
- mutex_(locking)
- {
- }
- operation* epoll_reactor::descriptor_state::perform_io(uint32_t events)
- {
- mutex_.lock();
- perform_io_cleanup_on_block_exit io_cleanup(reactor_);
- mutex::scoped_lock descriptor_lock(mutex_, mutex::scoped_lock::adopt_lock);
- // Exception operations must be processed first to ensure that any
- // out-of-band data is read before normal data.
- static const int flag[max_ops] = { EPOLLIN, EPOLLOUT, EPOLLPRI };
- for (int j = max_ops - 1; j >= 0; --j)
- {
- if (events & (flag[j] | EPOLLERR | EPOLLHUP))
- {
- try_speculative_[j] = true;
- while (reactor_op* op = op_queue_[j].front())
- {
- if (reactor_op::status status = op->perform())
- {
- op_queue_[j].pop();
- io_cleanup.ops_.push(op);
- if (status == reactor_op::done_and_exhausted)
- {
- try_speculative_[j] = false;
- break;
- }
- }
- else
- break;
- }
- }
- }
- // The first operation will be returned for completion now. The others will
- // be posted for later by the io_cleanup object's destructor.
- io_cleanup.first_op_ = io_cleanup.ops_.front();
- io_cleanup.ops_.pop();
- return io_cleanup.first_op_;
- }
- void epoll_reactor::descriptor_state::do_complete(
- void* owner, operation* base,
- const asio::error_code& ec, std::size_t bytes_transferred)
- {
- if (owner)
- {
- descriptor_state* descriptor_data = static_cast<descriptor_state*>(base);
- uint32_t events = static_cast<uint32_t>(bytes_transferred);
- if (operation* op = descriptor_data->perform_io(events))
- {
- op->complete(owner, ec, 0);
- }
- }
- }
- } // namespace detail
- } // namespace asio
- #include "asio/detail/pop_options.hpp"
- #endif // defined(ASIO_HAS_EPOLL)
- #endif // ASIO_DETAIL_IMPL_EPOLL_REACTOR_IPP
|