You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and dots ('.'), can be up to 35 characters long. Letters must be lowercase.
221 lines
7.7 KiB
221 lines
7.7 KiB
// |
|
// detail/select_reactor.hpp |
|
// ~~~~~~~~~~~~~~~~~~~~~~~~~ |
|
// |
|
// Copyright (c) 2003-2011 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 BOOST_ASIO_DETAIL_SELECT_REACTOR_HPP |
|
#define BOOST_ASIO_DETAIL_SELECT_REACTOR_HPP |
|
|
|
#if defined(_MSC_VER) && (_MSC_VER >= 1200) |
|
# pragma once |
|
#endif // defined(_MSC_VER) && (_MSC_VER >= 1200) |
|
|
|
#include <boost/asio/detail/config.hpp> |
|
|
|
#if defined(BOOST_ASIO_HAS_IOCP) \ |
|
|| (!defined(BOOST_ASIO_HAS_DEV_POLL) \ |
|
&& !defined(BOOST_ASIO_HAS_EPOLL) \ |
|
&& !defined(BOOST_ASIO_HAS_KQUEUE)) |
|
|
|
#include <boost/limits.hpp> |
|
#include <cstddef> |
|
#include <boost/asio/detail/fd_set_adapter.hpp> |
|
#include <boost/asio/detail/mutex.hpp> |
|
#include <boost/asio/detail/op_queue.hpp> |
|
#include <boost/asio/detail/reactor_op.hpp> |
|
#include <boost/asio/detail/reactor_op_queue.hpp> |
|
#include <boost/asio/detail/select_interrupter.hpp> |
|
#include <boost/asio/detail/select_reactor_fwd.hpp> |
|
#include <boost/asio/detail/socket_types.hpp> |
|
#include <boost/asio/detail/timer_op.hpp> |
|
#include <boost/asio/detail/timer_queue_base.hpp> |
|
#include <boost/asio/detail/timer_queue_fwd.hpp> |
|
#include <boost/asio/detail/timer_queue_set.hpp> |
|
#include <boost/asio/io_service.hpp> |
|
|
|
#if defined(BOOST_ASIO_HAS_IOCP) |
|
# include <boost/asio/detail/thread.hpp> |
|
#endif // defined(BOOST_ASIO_HAS_IOCP) |
|
|
|
#include <boost/asio/detail/push_options.hpp> |
|
|
|
namespace boost { |
|
namespace asio { |
|
namespace detail { |
|
|
|
class select_reactor |
|
: public boost::asio::detail::service_base<select_reactor> |
|
{ |
|
public: |
|
#if defined(BOOST_WINDOWS) || defined(__CYGWIN__) |
|
enum op_types { read_op = 0, write_op = 1, except_op = 2, |
|
max_select_ops = 3, connect_op = 3, max_ops = 4 }; |
|
#else // defined(BOOST_WINDOWS) || defined(__CYGWIN__) |
|
enum op_types { read_op = 0, write_op = 1, except_op = 2, |
|
max_select_ops = 3, connect_op = 1, max_ops = 3 }; |
|
#endif // defined(BOOST_WINDOWS) || defined(__CYGWIN__) |
|
|
|
// Per-descriptor data. |
|
struct per_descriptor_data |
|
{ |
|
}; |
|
|
|
// Constructor. |
|
BOOST_ASIO_DECL select_reactor(boost::asio::io_service& io_service); |
|
|
|
// Destructor. |
|
BOOST_ASIO_DECL ~select_reactor(); |
|
|
|
// Destroy all user-defined handler objects owned by the service. |
|
BOOST_ASIO_DECL void shutdown_service(); |
|
|
|
// Recreate internal descriptors following a fork. |
|
BOOST_ASIO_DECL void fork_service( |
|
boost::asio::io_service::fork_event fork_ev); |
|
|
|
// Initialise the task, but only if the reactor is not in its own thread. |
|
BOOST_ASIO_DECL void init_task(); |
|
|
|
// Register a socket with the reactor. Returns 0 on success, system error |
|
// code on failure. |
|
BOOST_ASIO_DECL int register_descriptor(socket_type, per_descriptor_data&); |
|
|
|
// Register a descriptor with an associated single operation. Returns 0 on |
|
// success, system error code on failure. |
|
BOOST_ASIO_DECL int register_internal_descriptor( |
|
int op_type, socket_type descriptor, |
|
per_descriptor_data& descriptor_data, reactor_op* op); |
|
|
|
// Post a reactor operation for immediate completion. |
|
void post_immediate_completion(reactor_op* op) |
|
{ |
|
io_service_.post_immediate_completion(op); |
|
} |
|
|
|
// Start a new operation. The reactor operation will be performed when the |
|
// given descriptor is flagged as ready, or an error has occurred. |
|
BOOST_ASIO_DECL void start_op(int op_type, socket_type descriptor, |
|
per_descriptor_data&, reactor_op* op, bool); |
|
|
|
// Cancel all operations associated with the given descriptor. The |
|
// handlers associated with the descriptor will be invoked with the |
|
// operation_aborted error. |
|
BOOST_ASIO_DECL void cancel_ops(socket_type descriptor, per_descriptor_data&); |
|
|
|
// Cancel any operations that are running against the descriptor and remove |
|
// its registration from the reactor. |
|
BOOST_ASIO_DECL void deregister_descriptor(socket_type descriptor, |
|
per_descriptor_data&, bool closing); |
|
|
|
// Remote the descriptor's registration from the reactor. |
|
BOOST_ASIO_DECL void deregister_internal_descriptor( |
|
socket_type descriptor, per_descriptor_data& descriptor_data); |
|
|
|
// Move descriptor registration from one descriptor_data object to another. |
|
BOOST_ASIO_DECL void move_descriptor(socket_type descriptor, |
|
per_descriptor_data& target_descriptor_data, |
|
per_descriptor_data& source_descriptor_data); |
|
|
|
// Add a new timer queue to the reactor. |
|
template <typename Time_Traits> |
|
void add_timer_queue(timer_queue<Time_Traits>& queue); |
|
|
|
// Remove a timer queue from the reactor. |
|
template <typename Time_Traits> |
|
void remove_timer_queue(timer_queue<Time_Traits>& queue); |
|
|
|
// Schedule a new operation in the given timer queue to expire at the |
|
// specified absolute time. |
|
template <typename Time_Traits> |
|
void schedule_timer(timer_queue<Time_Traits>& queue, |
|
const typename Time_Traits::time_type& time, |
|
typename timer_queue<Time_Traits>::per_timer_data& timer, timer_op* op); |
|
|
|
// Cancel the timer operations associated with the given token. Returns the |
|
// number of operations that have been posted or dispatched. |
|
template <typename Time_Traits> |
|
std::size_t cancel_timer(timer_queue<Time_Traits>& queue, |
|
typename timer_queue<Time_Traits>::per_timer_data& timer, |
|
std::size_t max_cancelled = (std::numeric_limits<std::size_t>::max)()); |
|
|
|
// Run select once until interrupted or events are ready to be dispatched. |
|
BOOST_ASIO_DECL void run(bool block, op_queue<operation>& ops); |
|
|
|
// Interrupt the select loop. |
|
BOOST_ASIO_DECL void interrupt(); |
|
|
|
private: |
|
#if defined(BOOST_ASIO_HAS_IOCP) |
|
// Run the select loop in the thread. |
|
BOOST_ASIO_DECL void run_thread(); |
|
|
|
// Entry point for the select loop thread. |
|
BOOST_ASIO_DECL static void call_run_thread(select_reactor* reactor); |
|
#endif // defined(BOOST_ASIO_HAS_IOCP) |
|
|
|
// Helper function to add a new timer queue. |
|
BOOST_ASIO_DECL void do_add_timer_queue(timer_queue_base& queue); |
|
|
|
// Helper function to remove a timer queue. |
|
BOOST_ASIO_DECL void do_remove_timer_queue(timer_queue_base& queue); |
|
|
|
// Get the timeout value for the select call. |
|
BOOST_ASIO_DECL timeval* get_timeout(timeval& tv); |
|
|
|
// Cancel all operations associated with the given descriptor. This function |
|
// does not acquire the select_reactor's mutex. |
|
BOOST_ASIO_DECL void cancel_ops_unlocked(socket_type descriptor, |
|
const boost::system::error_code& ec); |
|
|
|
// The io_service implementation used to post completions. |
|
io_service_impl& io_service_; |
|
|
|
// Mutex to protect access to internal data. |
|
boost::asio::detail::mutex mutex_; |
|
|
|
// The interrupter is used to break a blocking select call. |
|
select_interrupter interrupter_; |
|
|
|
// The queues of read, write and except operations. |
|
reactor_op_queue<socket_type> op_queue_[max_ops]; |
|
|
|
// The file descriptor sets to be passed to the select system call. |
|
fd_set_adapter fd_sets_[max_select_ops]; |
|
|
|
// The timer queues. |
|
timer_queue_set timer_queues_; |
|
|
|
#if defined(BOOST_ASIO_HAS_IOCP) |
|
// Does the reactor loop thread need to stop. |
|
bool stop_thread_; |
|
|
|
// The thread that is running the reactor loop. |
|
boost::asio::detail::thread* thread_; |
|
#endif // defined(BOOST_ASIO_HAS_IOCP) |
|
|
|
// Whether the service has been shut down. |
|
bool shutdown_; |
|
}; |
|
|
|
} // namespace detail |
|
} // namespace asio |
|
} // namespace boost |
|
|
|
#include <boost/asio/detail/pop_options.hpp> |
|
|
|
#include <boost/asio/detail/impl/select_reactor.hpp> |
|
#if defined(BOOST_ASIO_HEADER_ONLY) |
|
# include <boost/asio/detail/impl/select_reactor.ipp> |
|
#endif // defined(BOOST_ASIO_HEADER_ONLY) |
|
|
|
#endif // defined(BOOST_ASIO_HAS_IOCP) |
|
// || (!defined(BOOST_ASIO_HAS_DEV_POLL) |
|
// && !defined(BOOST_ASIO_HAS_EPOLL) |
|
// && !defined(BOOST_ASIO_HAS_KQUEUE)) |
|
|
|
#endif // BOOST_ASIO_DETAIL_SELECT_REACTOR_HPP
|
|
|