123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286 |
- //
- // detail/handler_alloc_helpers.hpp
- // ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
- //
- // Copyright (c) 2003-2021 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_HANDLER_ALLOC_HELPERS_HPP
- #define BOOST_ASIO_DETAIL_HANDLER_ALLOC_HELPERS_HPP
- #if defined(_MSC_VER) && (_MSC_VER >= 1200)
- # pragma once
- #endif // defined(_MSC_VER) && (_MSC_VER >= 1200)
- #include <boost/asio/detail/config.hpp>
- #include <boost/asio/detail/memory.hpp>
- #include <boost/asio/detail/noncopyable.hpp>
- #include <boost/asio/detail/recycling_allocator.hpp>
- #include <boost/asio/detail/thread_info_base.hpp>
- #include <boost/asio/associated_allocator.hpp>
- #include <boost/asio/handler_alloc_hook.hpp>
- #include <boost/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
- // boost_asio_handler_alloc_helpers namespace is defined here for that purpose.
- namespace boost_asio_handler_alloc_helpers {
- #if defined(BOOST_ASIO_NO_DEPRECATED)
- template <typename Handler>
- inline void error_if_hooks_are_defined(Handler& h)
- {
- using boost::asio::asio_handler_allocate;
- // If you get an error here it is because some of your handlers still
- // overload asio_handler_allocate, but this hook is no longer used.
- (void)static_cast<boost::asio::asio_handler_allocate_is_no_longer_used>(
- asio_handler_allocate(static_cast<std::size_t>(0),
- boost::asio::detail::addressof(h)));
- using boost::asio::asio_handler_deallocate;
- // If you get an error here it is because some of your handlers still
- // overload asio_handler_deallocate, but this hook is no longer used.
- (void)static_cast<boost::asio::asio_handler_deallocate_is_no_longer_used>(
- asio_handler_deallocate(static_cast<void*>(0),
- static_cast<std::size_t>(0), boost::asio::detail::addressof(h)));
- }
- #endif // defined(BOOST_ASIO_NO_DEPRECATED)
- template <typename Handler>
- inline void* allocate(std::size_t s, Handler& h)
- {
- #if !defined(BOOST_ASIO_HAS_HANDLER_HOOKS)
- return ::operator new(s);
- #elif defined(BOOST_ASIO_NO_DEPRECATED)
- // The asio_handler_allocate hook is no longer used to obtain memory.
- (void)&error_if_hooks_are_defined<Handler>;
- (void)h;
- #if !defined(BOOST_ASIO_DISABLE_SMALL_BLOCK_RECYCLING)
- return boost::asio::detail::thread_info_base::allocate(
- boost::asio::detail::thread_context::top_of_thread_call_stack(), s);
- #else // !defined(BOOST_ASIO_DISABLE_SMALL_BLOCK_RECYCLING)
- return ::operator new(size);
- #endif // !defined(BOOST_ASIO_DISABLE_SMALL_BLOCK_RECYCLING)
- #else
- using boost::asio::asio_handler_allocate;
- return asio_handler_allocate(s, boost::asio::detail::addressof(h));
- #endif
- }
- template <typename Handler>
- inline void deallocate(void* p, std::size_t s, Handler& h)
- {
- #if !defined(BOOST_ASIO_HAS_HANDLER_HOOKS)
- ::operator delete(p);
- #elif defined(BOOST_ASIO_NO_DEPRECATED)
- // The asio_handler_allocate hook is no longer used to obtain memory.
- (void)&error_if_hooks_are_defined<Handler>;
- (void)h;
- #if !defined(BOOST_ASIO_DISABLE_SMALL_BLOCK_RECYCLING)
- boost::asio::detail::thread_info_base::deallocate(
- boost::asio::detail::thread_context::top_of_thread_call_stack(), p, s);
- #else // !defined(BOOST_ASIO_DISABLE_SMALL_BLOCK_RECYCLING)
- (void)s;
- ::operator delete(p);
- #endif // !defined(BOOST_ASIO_DISABLE_SMALL_BLOCK_RECYCLING)
- #else
- using boost::asio::asio_handler_deallocate;
- asio_handler_deallocate(p, s, boost::asio::detail::addressof(h));
- #endif
- }
- } // namespace boost_asio_handler_alloc_helpers
- namespace boost {
- namespace asio {
- namespace detail {
- template <typename Handler, typename T>
- class hook_allocator
- {
- public:
- typedef T value_type;
- 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*>(
- boost_asio_handler_alloc_helpers::allocate(sizeof(T) * n, handler_));
- }
- void deallocate(T* p, std::size_t n)
- {
- boost_asio_handler_alloc_helpers::deallocate(p, sizeof(T) * n, handler_);
- }
- //private:
- Handler& handler_;
- };
- template <typename Handler>
- class hook_allocator<Handler, void>
- {
- public:
- typedef void value_type;
- 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_;
- };
- template <typename Handler, typename Allocator>
- struct get_hook_allocator
- {
- typedef Allocator type;
- static type get(Handler&, const Allocator& a)
- {
- return a;
- }
- };
- template <typename Handler, typename T>
- struct get_hook_allocator<Handler, std::allocator<T> >
- {
- typedef hook_allocator<Handler, T> type;
- static type get(Handler& handler, const std::allocator<T>&)
- {
- return type(handler);
- }
- };
- } // namespace detail
- } // namespace asio
- } // namespace boost
- #define BOOST_ASIO_DEFINE_HANDLER_PTR(op) \
- struct ptr \
- { \
- Handler* h; \
- op* v; \
- op* p; \
- ~ptr() \
- { \
- reset(); \
- } \
- static op* allocate(Handler& handler) \
- { \
- typedef typename ::boost::asio::associated_allocator< \
- Handler>::type associated_allocator_type; \
- typedef typename ::boost::asio::detail::get_hook_allocator< \
- Handler, associated_allocator_type>::type hook_allocator_type; \
- BOOST_ASIO_REBIND_ALLOC(hook_allocator_type, op) a( \
- ::boost::asio::detail::get_hook_allocator< \
- Handler, associated_allocator_type>::get( \
- handler, ::boost::asio::get_associated_allocator(handler))); \
- return a.allocate(1); \
- } \
- void reset() \
- { \
- if (p) \
- { \
- p->~op(); \
- p = 0; \
- } \
- if (v) \
- { \
- typedef typename ::boost::asio::associated_allocator< \
- Handler>::type associated_allocator_type; \
- typedef typename ::boost::asio::detail::get_hook_allocator< \
- Handler, associated_allocator_type>::type hook_allocator_type; \
- BOOST_ASIO_REBIND_ALLOC(hook_allocator_type, op) a( \
- ::boost::asio::detail::get_hook_allocator< \
- Handler, associated_allocator_type>::get( \
- *h, ::boost::asio::get_associated_allocator(*h))); \
- a.deallocate(static_cast<op*>(v), 1); \
- v = 0; \
- } \
- } \
- } \
- /**/
- #define BOOST_ASIO_DEFINE_TAGGED_HANDLER_ALLOCATOR_PTR(purpose, op) \
- struct ptr \
- { \
- const Alloc* a; \
- void* v; \
- op* p; \
- ~ptr() \
- { \
- reset(); \
- } \
- static op* allocate(const Alloc& a) \
- { \
- typedef typename ::boost::asio::detail::get_recycling_allocator< \
- Alloc, purpose>::type recycling_allocator_type; \
- BOOST_ASIO_REBIND_ALLOC(recycling_allocator_type, op) a1( \
- ::boost::asio::detail::get_recycling_allocator< \
- Alloc, purpose>::get(a)); \
- return a1.allocate(1); \
- } \
- void reset() \
- { \
- if (p) \
- { \
- p->~op(); \
- p = 0; \
- } \
- if (v) \
- { \
- typedef typename ::boost::asio::detail::get_recycling_allocator< \
- Alloc, purpose>::type recycling_allocator_type; \
- BOOST_ASIO_REBIND_ALLOC(recycling_allocator_type, op) a1( \
- ::boost::asio::detail::get_recycling_allocator< \
- Alloc, purpose>::get(*a)); \
- a1.deallocate(static_cast<op*>(v), 1); \
- v = 0; \
- } \
- } \
- } \
- /**/
- #define BOOST_ASIO_DEFINE_HANDLER_ALLOCATOR_PTR(op) \
- BOOST_ASIO_DEFINE_TAGGED_HANDLER_ALLOCATOR_PTR( \
- ::boost::asio::detail::thread_info_base::default_tag, op ) \
- /**/
- #include <boost/asio/detail/pop_options.hpp>
- #endif // BOOST_ASIO_DETAIL_HANDLER_ALLOC_HELPERS_HPP
|