| // |
| // detail/handler_alloc_helpers.hpp |
| // ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ |
| // |
| // Copyright (c) 2003-2015 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 ASIO_DETAIL_HANDLER_ALLOC_HELPERS_HPP |
| #define ASIO_DETAIL_HANDLER_ALLOC_HELPERS_HPP |
| |
| #if defined(_MSC_VER) && (_MSC_VER >= 1200) |
| # pragma once |
| #endif // defined(_MSC_VER) && (_MSC_VER >= 1200) |
| |
| #include "asio/detail/config.hpp" |
| #include "asio/detail/memory.hpp" |
| #include "asio/detail/noncopyable.hpp" |
| #include "asio/associated_allocator.hpp" |
| #include "asio/handler_alloc_hook.hpp" |
| |
| #include "asio/detail/push_options.hpp" |
| |
| // Calls to asio_handler_allocate and asio_handler_deallocate must be made from |
| // a namespace that does not contain any overloads of these functions. The |
| // asio_handler_alloc_helpers namespace is defined here for that purpose. |
| namespace asio_handler_alloc_helpers { |
| |
| template <typename Handler> |
| inline void* allocate(std::size_t s, Handler& h) |
| { |
| #if !defined(ASIO_HAS_HANDLER_HOOKS) |
| return ::operator new(s); |
| #else |
| using asio::asio_handler_allocate; |
| return asio_handler_allocate(s, asio::detail::addressof(h)); |
| #endif |
| } |
| |
| template <typename Handler> |
| inline void deallocate(void* p, std::size_t s, Handler& h) |
| { |
| #if !defined(ASIO_HAS_HANDLER_HOOKS) |
| ::operator delete(p); |
| #else |
| using asio::asio_handler_deallocate; |
| asio_handler_deallocate(p, s, asio::detail::addressof(h)); |
| #endif |
| } |
| |
| } // namespace asio_handler_alloc_helpers |
| |
| namespace asio { |
| namespace detail { |
| |
| template <typename Handler, typename T> |
| class hook_allocator |
| { |
| public: |
| template <typename U> |
| struct rebind |
| { |
| typedef hook_allocator<Handler, U> other; |
| }; |
| |
| explicit hook_allocator(Handler& h) |
| : handler_(h) |
| { |
| } |
| |
| template <typename U> |
| hook_allocator(const hook_allocator<Handler, U>& a) |
| : handler_(a.handler_) |
| { |
| } |
| |
| T* allocate(std::size_t n) |
| { |
| return static_cast<T*>( |
| asio_handler_alloc_helpers::allocate(sizeof(T) * n, handler_)); |
| } |
| |
| void deallocate(T* p, std::size_t n) |
| { |
| asio_handler_alloc_helpers::deallocate(p, sizeof(T) * n, handler_); |
| } |
| |
| //private: |
| Handler& handler_; |
| }; |
| |
| template <typename Handler> |
| class hook_allocator<Handler, void> |
| { |
| public: |
| template <typename U> |
| struct rebind |
| { |
| typedef hook_allocator<Handler, U> other; |
| }; |
| |
| explicit hook_allocator(Handler& h) |
| : handler_(h) |
| { |
| } |
| |
| template <typename U> |
| hook_allocator(const hook_allocator<Handler, U>& a) |
| : handler_(a.handler_) |
| { |
| } |
| |
| //private: |
| Handler& handler_; |
| }; |
| |
| } // namespace detail |
| } // namespace asio |
| |
| #define ASIO_DEFINE_HANDLER_PTR(op) \ |
| struct ptr \ |
| { \ |
| typedef typename ::asio::associated_allocator<Handler, \ |
| ::asio::detail::hook_allocator<Handler, \ |
| void> >::type::template rebind<op>::other allocator_type; \ |
| Handler* h; \ |
| op* v; \ |
| op* p; \ |
| ~ptr() \ |
| { \ |
| reset(); \ |
| } \ |
| static op* allocate(Handler& handler) \ |
| { \ |
| allocator_type a(::asio::associated_allocator<Handler, \ |
| ::asio::detail::hook_allocator<Handler, void> >::get(handler, \ |
| ::asio::detail::hook_allocator<Handler, void>(handler))); \ |
| return a.allocate(1); \ |
| } \ |
| void reset() \ |
| { \ |
| allocator_type a(::asio::associated_allocator<Handler, \ |
| ::asio::detail::hook_allocator<Handler, void> >::get(*h, \ |
| ::asio::detail::hook_allocator<Handler, void>(*h))); \ |
| if (p) \ |
| { \ |
| p->~op(); \ |
| p = 0; \ |
| } \ |
| if (v) \ |
| { \ |
| a.deallocate(static_cast<op*>(v), 1); \ |
| v = 0; \ |
| } \ |
| } \ |
| } \ |
| /**/ |
| |
| #define ASIO_DEFINE_HANDLER_ALLOCATOR_PTR(op, alloc) \ |
| struct ptr \ |
| { \ |
| typename alloc::template rebind<op>::other a; \ |
| void* v; \ |
| op* p; \ |
| ~ptr() \ |
| { \ |
| reset(); \ |
| } \ |
| void reset() \ |
| { \ |
| if (p) \ |
| { \ |
| p->~op(); \ |
| p = 0; \ |
| } \ |
| if (v) \ |
| { \ |
| a.deallocate(static_cast<op*>(v), 1); \ |
| v = 0; \ |
| } \ |
| } \ |
| } \ |
| /**/ |
| |
| #include "asio/detail/pop_options.hpp" |
| |
| #endif // ASIO_DETAIL_HANDLER_ALLOC_HELPERS_HPP |