diff options
Diffstat (limited to '3rdparty/asio/include/asio/detail/thread_info_base.hpp')
-rw-r--r-- | 3rdparty/asio/include/asio/detail/thread_info_base.hpp | 256 |
1 files changed, 256 insertions, 0 deletions
diff --git a/3rdparty/asio/include/asio/detail/thread_info_base.hpp b/3rdparty/asio/include/asio/detail/thread_info_base.hpp new file mode 100644 index 00000000000..2223e8b3faf --- /dev/null +++ b/3rdparty/asio/include/asio/detail/thread_info_base.hpp @@ -0,0 +1,256 @@ +// +// detail/thread_info_base.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 ASIO_DETAIL_THREAD_INFO_BASE_HPP +#define ASIO_DETAIL_THREAD_INFO_BASE_HPP + +#if defined(_MSC_VER) && (_MSC_VER >= 1200) +# pragma once +#endif // defined(_MSC_VER) && (_MSC_VER >= 1200) + +#include "asio/detail/config.hpp" +#include <climits> +#include <cstddef> +#include "asio/detail/memory.hpp" +#include "asio/detail/noncopyable.hpp" + +#if defined(ASIO_HAS_STD_EXCEPTION_PTR) \ + && !defined(ASIO_NO_EXCEPTIONS) +# include <exception> +# include "asio/multiple_exceptions.hpp" +#endif // defined(ASIO_HAS_STD_EXCEPTION_PTR) + // && !defined(ASIO_NO_EXCEPTIONS) + +#include "asio/detail/push_options.hpp" + +namespace asio { +namespace detail { + +#ifndef ASIO_RECYCLING_ALLOCATOR_CACHE_SIZE +# define ASIO_RECYCLING_ALLOCATOR_CACHE_SIZE 2 +#endif // ASIO_RECYCLING_ALLOCATOR_CACHE_SIZE + +class thread_info_base + : private noncopyable +{ +public: + struct default_tag + { + enum + { + cache_size = ASIO_RECYCLING_ALLOCATOR_CACHE_SIZE, + begin_mem_index = 0, + end_mem_index = cache_size + }; + }; + + struct awaitable_frame_tag + { + enum + { + cache_size = ASIO_RECYCLING_ALLOCATOR_CACHE_SIZE, + begin_mem_index = default_tag::end_mem_index, + end_mem_index = begin_mem_index + cache_size + }; + }; + + struct executor_function_tag + { + enum + { + cache_size = ASIO_RECYCLING_ALLOCATOR_CACHE_SIZE, + begin_mem_index = awaitable_frame_tag::end_mem_index, + end_mem_index = begin_mem_index + cache_size + }; + }; + + struct cancellation_signal_tag + { + enum + { + cache_size = ASIO_RECYCLING_ALLOCATOR_CACHE_SIZE, + begin_mem_index = executor_function_tag::end_mem_index, + end_mem_index = begin_mem_index + cache_size + }; + }; + + struct parallel_group_tag + { + enum + { + cache_size = ASIO_RECYCLING_ALLOCATOR_CACHE_SIZE, + begin_mem_index = cancellation_signal_tag::end_mem_index, + end_mem_index = begin_mem_index + cache_size + }; + }; + + enum { max_mem_index = parallel_group_tag::end_mem_index }; + + thread_info_base() +#if defined(ASIO_HAS_STD_EXCEPTION_PTR) \ + && !defined(ASIO_NO_EXCEPTIONS) + : has_pending_exception_(0) +#endif // defined(ASIO_HAS_STD_EXCEPTION_PTR) + // && !defined(ASIO_NO_EXCEPTIONS) + { + for (int i = 0; i < max_mem_index; ++i) + reusable_memory_[i] = 0; + } + + ~thread_info_base() + { + for (int i = 0; i < max_mem_index; ++i) + { + // The following test for non-null pointers is technically redundant, but + // it is significantly faster when using a tight io_context::poll() loop + // in latency sensitive applications. + if (reusable_memory_[i]) + aligned_delete(reusable_memory_[i]); + } + } + + static void* allocate(thread_info_base* this_thread, + std::size_t size, std::size_t align = ASIO_DEFAULT_ALIGN) + { + return allocate(default_tag(), this_thread, size, align); + } + + static void deallocate(thread_info_base* this_thread, + void* pointer, std::size_t size) + { + deallocate(default_tag(), this_thread, pointer, size); + } + + template <typename Purpose> + static void* allocate(Purpose, thread_info_base* this_thread, + std::size_t size, std::size_t align = ASIO_DEFAULT_ALIGN) + { + std::size_t chunks = (size + chunk_size - 1) / chunk_size; + + if (this_thread) + { + for (int mem_index = Purpose::begin_mem_index; + mem_index < Purpose::end_mem_index; ++mem_index) + { + if (this_thread->reusable_memory_[mem_index]) + { + void* const pointer = this_thread->reusable_memory_[mem_index]; + unsigned char* const mem = static_cast<unsigned char*>(pointer); + if (static_cast<std::size_t>(mem[0]) >= chunks + && reinterpret_cast<std::size_t>(pointer) % align == 0) + { + this_thread->reusable_memory_[mem_index] = 0; + mem[size] = mem[0]; + return pointer; + } + } + } + + for (int mem_index = Purpose::begin_mem_index; + mem_index < Purpose::end_mem_index; ++mem_index) + { + if (this_thread->reusable_memory_[mem_index]) + { + void* const pointer = this_thread->reusable_memory_[mem_index]; + this_thread->reusable_memory_[mem_index] = 0; + aligned_delete(pointer); + break; + } + } + } + + void* const pointer = aligned_new(align, chunks * chunk_size + 1); + unsigned char* const mem = static_cast<unsigned char*>(pointer); + mem[size] = (chunks <= UCHAR_MAX) ? static_cast<unsigned char>(chunks) : 0; + return pointer; + } + + template <typename Purpose> + static void deallocate(Purpose, thread_info_base* this_thread, + void* pointer, std::size_t size) + { + if (size <= chunk_size * UCHAR_MAX) + { + if (this_thread) + { + for (int mem_index = Purpose::begin_mem_index; + mem_index < Purpose::end_mem_index; ++mem_index) + { + if (this_thread->reusable_memory_[mem_index] == 0) + { + unsigned char* const mem = static_cast<unsigned char*>(pointer); + mem[0] = mem[size]; + this_thread->reusable_memory_[mem_index] = pointer; + return; + } + } + } + } + + aligned_delete(pointer); + } + + void capture_current_exception() + { +#if defined(ASIO_HAS_STD_EXCEPTION_PTR) \ + && !defined(ASIO_NO_EXCEPTIONS) + switch (has_pending_exception_) + { + case 0: + has_pending_exception_ = 1; + pending_exception_ = std::current_exception(); + break; + case 1: + has_pending_exception_ = 2; + pending_exception_ = + std::make_exception_ptr<multiple_exceptions>( + multiple_exceptions(pending_exception_)); + break; + default: + break; + } +#endif // defined(ASIO_HAS_STD_EXCEPTION_PTR) + // && !defined(ASIO_NO_EXCEPTIONS) + } + + void rethrow_pending_exception() + { +#if defined(ASIO_HAS_STD_EXCEPTION_PTR) \ + && !defined(ASIO_NO_EXCEPTIONS) + if (has_pending_exception_ > 0) + { + has_pending_exception_ = 0; + std::exception_ptr ex( + ASIO_MOVE_CAST(std::exception_ptr)( + pending_exception_)); + std::rethrow_exception(ex); + } +#endif // defined(ASIO_HAS_STD_EXCEPTION_PTR) + // && !defined(ASIO_NO_EXCEPTIONS) + } + +private: + enum { chunk_size = 4 }; + void* reusable_memory_[max_mem_index]; + +#if defined(ASIO_HAS_STD_EXCEPTION_PTR) \ + && !defined(ASIO_NO_EXCEPTIONS) + int has_pending_exception_; + std::exception_ptr pending_exception_; +#endif // defined(ASIO_HAS_STD_EXCEPTION_PTR) + // && !defined(ASIO_NO_EXCEPTIONS) +}; + +} // namespace detail +} // namespace asio + +#include "asio/detail/pop_options.hpp" + +#endif // ASIO_DETAIL_THREAD_INFO_BASE_HPP |