| // |
| // detail/impl/win_iocp_io_service.hpp |
| // ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ |
| // |
| // Copyright (c) 2003-2010 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_IMPL_WIN_IOCP_IO_SERVICE_HPP |
| #define BOOST_ASIO_DETAIL_IMPL_WIN_IOCP_IO_SERVICE_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) |
| |
| #include <boost/asio/detail/call_stack.hpp> |
| #include <boost/asio/detail/completion_handler.hpp> |
| #include <boost/asio/detail/fenced_block.hpp> |
| #include <boost/asio/detail/handler_alloc_helpers.hpp> |
| #include <boost/asio/detail/handler_invoke_helpers.hpp> |
| |
| #include <boost/asio/detail/push_options.hpp> |
| |
| namespace boost { |
| namespace asio { |
| namespace detail { |
| |
| template <typename Handler> |
| void win_iocp_io_service::dispatch(Handler handler) |
| { |
| if (call_stack<win_iocp_io_service>::contains(this)) |
| { |
| boost::asio::detail::fenced_block b; |
| boost_asio_handler_invoke_helpers::invoke(handler, handler); |
| } |
| else |
| post(handler); |
| } |
| |
| template <typename Handler> |
| void win_iocp_io_service::post(Handler handler) |
| { |
| // Allocate and construct an operation to wrap the handler. |
| typedef completion_handler<Handler> op; |
| typename op::ptr p = { boost::addressof(handler), |
| boost_asio_handler_alloc_helpers::allocate( |
| sizeof(op), handler), 0 }; |
| p.p = new (p.v) op(handler); |
| |
| post_immediate_completion(p.p); |
| p.v = p.p = 0; |
| } |
| |
| template <typename Time_Traits> |
| void win_iocp_io_service::add_timer_queue( |
| timer_queue<Time_Traits>& queue) |
| { |
| do_add_timer_queue(queue); |
| } |
| |
| template <typename Time_Traits> |
| void win_iocp_io_service::remove_timer_queue( |
| timer_queue<Time_Traits>& queue) |
| { |
| do_remove_timer_queue(queue); |
| } |
| |
| template <typename Time_Traits> |
| void win_iocp_io_service::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) |
| { |
| // If the service has been shut down we silently discard the timer. |
| if (::InterlockedExchangeAdd(&shutdown_, 0) != 0) |
| { |
| post_immediate_completion(op); |
| return; |
| } |
| |
| mutex::scoped_lock lock(dispatch_mutex_); |
| |
| bool earliest = queue.enqueue_timer(time, timer, op); |
| work_started(); |
| if (earliest) |
| update_timeout(); |
| } |
| |
| template <typename Time_Traits> |
| std::size_t win_iocp_io_service::cancel_timer(timer_queue<Time_Traits>& queue, |
| typename timer_queue<Time_Traits>::per_timer_data& timer) |
| { |
| // If the service has been shut down we silently ignore the cancellation. |
| if (::InterlockedExchangeAdd(&shutdown_, 0) != 0) |
| return 0; |
| |
| mutex::scoped_lock lock(dispatch_mutex_); |
| op_queue<win_iocp_operation> ops; |
| std::size_t n = queue.cancel_timer(timer, ops); |
| post_deferred_completions(ops); |
| return n; |
| } |
| |
| } // namespace detail |
| } // namespace asio |
| } // namespace boost |
| |
| #include <boost/asio/detail/pop_options.hpp> |
| |
| #endif // defined(BOOST_ASIO_HAS_IOCP) |
| |
| #endif // BOOST_ASIO_DETAIL_IMPL_WIN_IOCP_IO_SERVICE_HPP |