Chris@16: //
Chris@16: // impl/handler_alloc_hook.ipp
Chris@16: // ~~~~~~~~~~~~~~~~~~~~~~~~~~~
Chris@16: //
Chris@101: // Copyright (c) 2003-2015 Christopher M. Kohlhoff (chris at kohlhoff dot com)
Chris@16: //
Chris@16: // Distributed under the Boost Software License, Version 1.0. (See accompanying
Chris@16: // file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
Chris@16: //
Chris@16: 
Chris@16: #ifndef BOOST_ASIO_IMPL_HANDLER_ALLOC_HOOK_IPP
Chris@16: #define BOOST_ASIO_IMPL_HANDLER_ALLOC_HOOK_IPP
Chris@16: 
Chris@16: #if defined(_MSC_VER) && (_MSC_VER >= 1200)
Chris@16: # pragma once
Chris@16: #endif // defined(_MSC_VER) && (_MSC_VER >= 1200)
Chris@16: 
Chris@16: #include <boost/asio/detail/config.hpp>
Chris@16: #include <boost/asio/detail/call_stack.hpp>
Chris@16: #include <boost/asio/handler_alloc_hook.hpp>
Chris@16: 
Chris@16: #if !defined(BOOST_ASIO_DISABLE_SMALL_BLOCK_RECYCLING)
Chris@16: # if defined(BOOST_ASIO_HAS_IOCP)
Chris@16: #  include <boost/asio/detail/win_iocp_thread_info.hpp>
Chris@16: # else // defined(BOOST_ASIO_HAS_IOCP)
Chris@16: #  include <boost/asio/detail/task_io_service_thread_info.hpp>
Chris@16: # endif // defined(BOOST_ASIO_HAS_IOCP)
Chris@16: #endif // !defined(BOOST_ASIO_DISABLE_SMALL_BLOCK_RECYCLING)
Chris@16: 
Chris@16: #include <boost/asio/detail/push_options.hpp>
Chris@16: 
Chris@16: namespace boost {
Chris@16: namespace asio {
Chris@16: 
Chris@16: #if defined(BOOST_ASIO_HAS_IOCP)
Chris@16: namespace detail { class win_iocp_io_service; }
Chris@16: #endif // defined(BOOST_ASIO_HAS_IOCP)
Chris@16: 
Chris@16: void* asio_handler_allocate(std::size_t size, ...)
Chris@16: {
Chris@16: #if !defined(BOOST_ASIO_DISABLE_SMALL_BLOCK_RECYCLING)
Chris@16: # if defined(BOOST_ASIO_HAS_IOCP)
Chris@16:   typedef detail::win_iocp_io_service io_service_impl;
Chris@16:   typedef detail::win_iocp_thread_info thread_info;
Chris@16: # else // defined(BOOST_ASIO_HAS_IOCP)
Chris@16:   typedef detail::task_io_service io_service_impl;
Chris@16:   typedef detail::task_io_service_thread_info thread_info;
Chris@16: # endif // defined(BOOST_ASIO_HAS_IOCP)
Chris@16:   typedef detail::call_stack<io_service_impl, thread_info> call_stack;
Chris@16:   return thread_info::allocate(call_stack::top(), size);
Chris@16: #else // !defined(BOOST_ASIO_DISABLE_SMALL_BLOCK_RECYCLING)
Chris@16:   return ::operator new(size);
Chris@16: #endif // !defined(BOOST_ASIO_DISABLE_SMALL_BLOCK_RECYCLING)
Chris@16: }
Chris@16: 
Chris@16: void asio_handler_deallocate(void* pointer, std::size_t size, ...)
Chris@16: {
Chris@16: #if !defined(BOOST_ASIO_DISABLE_SMALL_BLOCK_RECYCLING)
Chris@16: # if defined(BOOST_ASIO_HAS_IOCP)
Chris@16:   typedef detail::win_iocp_io_service io_service_impl;
Chris@16:   typedef detail::win_iocp_thread_info thread_info;
Chris@16: # else // defined(BOOST_ASIO_HAS_IOCP)
Chris@16:   typedef detail::task_io_service io_service_impl;
Chris@16:   typedef detail::task_io_service_thread_info thread_info;
Chris@16: # endif // defined(BOOST_ASIO_HAS_IOCP)
Chris@16:   typedef detail::call_stack<io_service_impl, thread_info> call_stack;
Chris@16:   thread_info::deallocate(call_stack::top(), pointer, size);
Chris@16: #else // !defined(BOOST_ASIO_DISABLE_SMALL_BLOCK_RECYCLING)
Chris@16:   (void)size;
Chris@16:   ::operator delete(pointer);
Chris@16: #endif // !defined(BOOST_ASIO_DISABLE_SMALL_BLOCK_RECYCLING)
Chris@16: }
Chris@16: 
Chris@16: } // namespace asio
Chris@16: } // namespace boost
Chris@16: 
Chris@16: #include <boost/asio/detail/pop_options.hpp>
Chris@16: 
Chris@16: #endif // BOOST_ASIO_IMPL_HANDLER_ALLOC_HOOK_IPP