win_iocp_null_buffers_op.hpp 3.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112
  1. //
  2. // detail/win_iocp_null_buffers_op.hpp
  3. // ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
  4. //
  5. // Copyright (c) 2003-2011 Christopher M. Kohlhoff (chris at kohlhoff dot com)
  6. //
  7. // Distributed under the Boost Software License, Version 1.0. (See accompanying
  8. // file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
  9. //
  10. #ifndef ASIO_DETAIL_WIN_IOCP_NULL_BUFFERS_OP_HPP
  11. #define ASIO_DETAIL_WIN_IOCP_NULL_BUFFERS_OP_HPP
  12. #if defined(_MSC_VER) && (_MSC_VER >= 1200)
  13. # pragma once
  14. #endif // defined(_MSC_VER) && (_MSC_VER >= 1200)
  15. #include "asio/detail/config.hpp"
  16. #if defined(ASIO_HAS_IOCP)
  17. #include <boost/utility/addressof.hpp>
  18. #include "asio/detail/bind_handler.hpp"
  19. #include "asio/detail/buffer_sequence_adapter.hpp"
  20. #include "asio/detail/fenced_block.hpp"
  21. #include "asio/detail/handler_alloc_helpers.hpp"
  22. #include "asio/detail/handler_invoke_helpers.hpp"
  23. #include "asio/detail/reactor_op.hpp"
  24. #include "asio/detail/socket_ops.hpp"
  25. #include "asio/error.hpp"
  26. #include "asio/detail/push_options.hpp"
  27. namespace asio {
  28. namespace detail {
  29. template <typename Handler>
  30. class win_iocp_null_buffers_op : public reactor_op
  31. {
  32. public:
  33. ASIO_DEFINE_HANDLER_PTR(win_iocp_null_buffers_op);
  34. win_iocp_null_buffers_op(socket_ops::weak_cancel_token_type cancel_token,
  35. Handler handler)
  36. : reactor_op(&win_iocp_null_buffers_op::do_perform,
  37. &win_iocp_null_buffers_op::do_complete),
  38. cancel_token_(cancel_token),
  39. handler_(handler)
  40. {
  41. }
  42. static bool do_perform(reactor_op*)
  43. {
  44. return true;
  45. }
  46. static void do_complete(io_service_impl* owner, operation* base,
  47. asio::error_code ec, std::size_t bytes_transferred)
  48. {
  49. // Take ownership of the operation object.
  50. win_iocp_null_buffers_op* o(static_cast<win_iocp_null_buffers_op*>(base));
  51. ptr p = { boost::addressof(o->handler_), o, o };
  52. // The reactor may have stored a result in the operation object.
  53. if (o->ec_)
  54. ec = o->ec_;
  55. // Map non-portable errors to their portable counterparts.
  56. if (ec.value() == ERROR_NETNAME_DELETED)
  57. {
  58. if (o->cancel_token_.expired())
  59. ec = asio::error::operation_aborted;
  60. else
  61. ec = asio::error::connection_reset;
  62. }
  63. else if (ec.value() == ERROR_PORT_UNREACHABLE)
  64. {
  65. ec = asio::error::connection_refused;
  66. }
  67. // Make a copy of the handler so that the memory can be deallocated before
  68. // the upcall is made. Even if we're not about to make an upcall, a
  69. // sub-object of the handler may be the true owner of the memory associated
  70. // with the handler. Consequently, a local copy of the handler is required
  71. // to ensure that any owning sub-object remains valid until after we have
  72. // deallocated the memory here.
  73. detail::binder2<Handler, asio::error_code, std::size_t>
  74. handler(o->handler_, ec, bytes_transferred);
  75. p.h = boost::addressof(handler.handler_);
  76. p.reset();
  77. // Make the upcall if required.
  78. if (owner)
  79. {
  80. asio::detail::fenced_block b;
  81. asio_handler_invoke_helpers::invoke(handler, handler.handler_);
  82. }
  83. }
  84. private:
  85. socket_ops::weak_cancel_token_type cancel_token_;
  86. Handler handler_;
  87. };
  88. } // namespace detail
  89. } // namespace asio
  90. #include "asio/detail/pop_options.hpp"
  91. #endif // defined(ASIO_HAS_IOCP)
  92. #endif // ASIO_DETAIL_WIN_IOCP_NULL_BUFFERS_OP_HPP