// | |
// detail/impl/win_iocp_io_service.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_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 |