1// 2// impl/handler_alloc_hook.ipp 3// ~~~~~~~~~~~~~~~~~~~~~~~~~~~ 4// 5// Copyright (c) 2003-2015 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 11#ifndef BOOST_ASIO_IMPL_HANDLER_ALLOC_HOOK_IPP 12#define BOOST_ASIO_IMPL_HANDLER_ALLOC_HOOK_IPP 13 14#if defined(_MSC_VER) && (_MSC_VER >= 1200) 15# pragma once 16#endif // defined(_MSC_VER) && (_MSC_VER >= 1200) 17 18#include <boost/asio/detail/config.hpp> 19#include <boost/asio/detail/call_stack.hpp> 20#include <boost/asio/handler_alloc_hook.hpp> 21 22#if !defined(BOOST_ASIO_DISABLE_SMALL_BLOCK_RECYCLING) 23# if defined(BOOST_ASIO_HAS_IOCP) 24# include <boost/asio/detail/win_iocp_thread_info.hpp> 25# else // defined(BOOST_ASIO_HAS_IOCP) 26# include <boost/asio/detail/task_io_service_thread_info.hpp> 27# endif // defined(BOOST_ASIO_HAS_IOCP) 28#endif // !defined(BOOST_ASIO_DISABLE_SMALL_BLOCK_RECYCLING) 29 30#include <boost/asio/detail/push_options.hpp> 31 32namespace boost { 33namespace asio { 34 35#if defined(BOOST_ASIO_HAS_IOCP) 36namespace detail { class win_iocp_io_service; } 37#endif // defined(BOOST_ASIO_HAS_IOCP) 38 39void* asio_handler_allocate(std::size_t size, ...) 40{ 41#if !defined(BOOST_ASIO_DISABLE_SMALL_BLOCK_RECYCLING) 42# if defined(BOOST_ASIO_HAS_IOCP) 43 typedef detail::win_iocp_io_service io_service_impl; 44 typedef detail::win_iocp_thread_info thread_info; 45# else // defined(BOOST_ASIO_HAS_IOCP) 46 typedef detail::task_io_service io_service_impl; 47 typedef detail::task_io_service_thread_info thread_info; 48# endif // defined(BOOST_ASIO_HAS_IOCP) 49 typedef detail::call_stack<io_service_impl, thread_info> call_stack; 50 return thread_info::allocate(call_stack::top(), size); 51#else // !defined(BOOST_ASIO_DISABLE_SMALL_BLOCK_RECYCLING) 52 return ::operator new(size); 53#endif // !defined(BOOST_ASIO_DISABLE_SMALL_BLOCK_RECYCLING) 54} 55 56void asio_handler_deallocate(void* pointer, std::size_t size, ...) 57{ 58#if !defined(BOOST_ASIO_DISABLE_SMALL_BLOCK_RECYCLING) 59# if defined(BOOST_ASIO_HAS_IOCP) 60 typedef detail::win_iocp_io_service io_service_impl; 61 typedef detail::win_iocp_thread_info thread_info; 62# else // defined(BOOST_ASIO_HAS_IOCP) 63 typedef detail::task_io_service io_service_impl; 64 typedef detail::task_io_service_thread_info thread_info; 65# endif // defined(BOOST_ASIO_HAS_IOCP) 66 typedef detail::call_stack<io_service_impl, thread_info> call_stack; 67 thread_info::deallocate(call_stack::top(), pointer, size); 68#else // !defined(BOOST_ASIO_DISABLE_SMALL_BLOCK_RECYCLING) 69 (void)size; 70 ::operator delete(pointer); 71#endif // !defined(BOOST_ASIO_DISABLE_SMALL_BLOCK_RECYCLING) 72} 73 74} // namespace asio 75} // namespace boost 76 77#include <boost/asio/detail/pop_options.hpp> 78 79#endif // BOOST_ASIO_IMPL_HANDLER_ALLOC_HOOK_IPP 80