Warning: This file is not a C or C++ file. It does not have highlighting.
| 1 | // -*- C++ -*- |
|---|---|
| 2 | //===----------------------------------------------------------------------===// |
| 3 | // |
| 4 | // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. |
| 5 | // See https://llvm.org/LICENSE.txt for license information. |
| 6 | // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
| 7 | // |
| 8 | //===----------------------------------------------------------------------===// |
| 9 | |
| 10 | #ifndef _LIBCPP___CXX03___MEMORY_ALLOCATOR_H |
| 11 | #define _LIBCPP___CXX03___MEMORY_ALLOCATOR_H |
| 12 | |
| 13 | #include <__cxx03/__config> |
| 14 | #include <__cxx03/__memory/addressof.h> |
| 15 | #include <__cxx03/__memory/allocate_at_least.h> |
| 16 | #include <__cxx03/__memory/allocator_traits.h> |
| 17 | #include <__cxx03/__type_traits/is_const.h> |
| 18 | #include <__cxx03/__type_traits/is_constant_evaluated.h> |
| 19 | #include <__cxx03/__type_traits/is_same.h> |
| 20 | #include <__cxx03/__type_traits/is_void.h> |
| 21 | #include <__cxx03/__type_traits/is_volatile.h> |
| 22 | #include <__cxx03/__utility/forward.h> |
| 23 | #include <__cxx03/cstddef> |
| 24 | #include <__cxx03/new> |
| 25 | |
| 26 | #if !defined(_LIBCPP_HAS_NO_PRAGMA_SYSTEM_HEADER) |
| 27 | # pragma GCC system_header |
| 28 | #endif |
| 29 | |
| 30 | _LIBCPP_BEGIN_NAMESPACE_STD |
| 31 | |
| 32 | template <class _Tp> |
| 33 | class allocator; |
| 34 | |
| 35 | template <> |
| 36 | class _LIBCPP_TEMPLATE_VIS allocator<void> { |
| 37 | public: |
| 38 | typedef void* pointer; |
| 39 | typedef const void* const_pointer; |
| 40 | typedef void value_type; |
| 41 | |
| 42 | template <class _Up> |
| 43 | struct rebind { |
| 44 | typedef allocator<_Up> other; |
| 45 | }; |
| 46 | }; |
| 47 | |
| 48 | // TODO(LLVM 20): Remove the escape hatch |
| 49 | #ifdef _LIBCPP_ENABLE_REMOVED_ALLOCATOR_CONST |
| 50 | template <> |
| 51 | class _LIBCPP_TEMPLATE_VIS allocator<const void> { |
| 52 | public: |
| 53 | typedef const void* pointer; |
| 54 | typedef const void* const_pointer; |
| 55 | typedef const void value_type; |
| 56 | |
| 57 | template <class _Up> |
| 58 | struct rebind { |
| 59 | typedef allocator<_Up> other; |
| 60 | }; |
| 61 | }; |
| 62 | #endif // _LIBCPP_ENABLE_REMOVED_ALLOCATOR_CONST |
| 63 | |
| 64 | // This class provides a non-trivial default constructor to the class that derives from it |
| 65 | // if the condition is satisfied. |
| 66 | // |
| 67 | // The second template parameter exists to allow giving a unique type to __non_trivial_if, |
| 68 | // which makes it possible to avoid breaking the ABI when making this a base class of an |
| 69 | // existing class. Without that, imagine we have classes D1 and D2, both of which used to |
| 70 | // have no base classes, but which now derive from __non_trivial_if. The layout of a class |
| 71 | // that inherits from both D1 and D2 will change because the two __non_trivial_if base |
| 72 | // classes are not allowed to share the same address. |
| 73 | // |
| 74 | // By making those __non_trivial_if base classes unique, we work around this problem and |
| 75 | // it is safe to start deriving from __non_trivial_if in existing classes. |
| 76 | template <bool _Cond, class _Unique> |
| 77 | struct __non_trivial_if {}; |
| 78 | |
| 79 | template <class _Unique> |
| 80 | struct __non_trivial_if<true, _Unique> { |
| 81 | _LIBCPP_HIDE_FROM_ABI __non_trivial_if() _NOEXCEPT {} |
| 82 | }; |
| 83 | |
| 84 | // allocator |
| 85 | // |
| 86 | // Note: For ABI compatibility between C++20 and previous standards, we make |
| 87 | // allocator<void> trivial in C++20. |
| 88 | |
| 89 | template <class _Tp> |
| 90 | class _LIBCPP_TEMPLATE_VIS allocator : private __non_trivial_if<!is_void<_Tp>::value, allocator<_Tp> > { |
| 91 | static_assert(!is_const<_Tp>::value, "std::allocator does not support const types"); |
| 92 | static_assert(!is_volatile<_Tp>::value, "std::allocator does not support volatile types"); |
| 93 | |
| 94 | public: |
| 95 | typedef size_t size_type; |
| 96 | typedef ptrdiff_t difference_type; |
| 97 | typedef _Tp value_type; |
| 98 | typedef true_type propagate_on_container_move_assignment; |
| 99 | typedef true_type is_always_equal; |
| 100 | |
| 101 | _LIBCPP_HIDE_FROM_ABI allocator() _NOEXCEPT = default; |
| 102 | |
| 103 | template <class _Up> |
| 104 | _LIBCPP_HIDE_FROM_ABI allocator(const allocator<_Up>&) _NOEXCEPT {} |
| 105 | |
| 106 | _LIBCPP_NODISCARD _LIBCPP_HIDE_FROM_ABI _Tp* allocate(size_t __n) { |
| 107 | if (__n > allocator_traits<allocator>::max_size(*this)) |
| 108 | __throw_bad_array_new_length(); |
| 109 | if (__libcpp_is_constant_evaluated()) { |
| 110 | return static_cast<_Tp*>(::operator new(__n * sizeof(_Tp))); |
| 111 | } else { |
| 112 | return static_cast<_Tp*>(std::__libcpp_allocate(__n * sizeof(_Tp), _LIBCPP_ALIGNOF(_Tp))); |
| 113 | } |
| 114 | } |
| 115 | |
| 116 | _LIBCPP_HIDE_FROM_ABI void deallocate(_Tp* __p, size_t __n) _NOEXCEPT { |
| 117 | if (__libcpp_is_constant_evaluated()) { |
| 118 | ::operator delete(__p); |
| 119 | } else { |
| 120 | std::__libcpp_deallocate((void*)__p, __n * sizeof(_Tp), _LIBCPP_ALIGNOF(_Tp)); |
| 121 | } |
| 122 | } |
| 123 | |
| 124 | // C++20 Removed members |
| 125 | typedef _Tp* pointer; |
| 126 | typedef const _Tp* const_pointer; |
| 127 | typedef _Tp& reference; |
| 128 | typedef const _Tp& const_reference; |
| 129 | |
| 130 | template <class _Up> |
| 131 | struct rebind { |
| 132 | typedef allocator<_Up> other; |
| 133 | }; |
| 134 | |
| 135 | _LIBCPP_HIDE_FROM_ABI pointer address(reference __x) const _NOEXCEPT { return std::addressof(__x); } |
| 136 | _LIBCPP_HIDE_FROM_ABI const_pointer address(const_reference __x) const _NOEXCEPT { return std::addressof(__x); } |
| 137 | |
| 138 | _LIBCPP_NODISCARD _LIBCPP_HIDE_FROM_ABI _Tp* allocate(size_t __n, const void*) { return allocate(__n); } |
| 139 | |
| 140 | _LIBCPP_HIDE_FROM_ABI size_type max_size() const _NOEXCEPT { return size_type(~0) / sizeof(_Tp); } |
| 141 | |
| 142 | template <class _Up, class... _Args> |
| 143 | _LIBCPP_HIDE_FROM_ABI void construct(_Up* __p, _Args&&... __args) { |
| 144 | ::new ((void*)__p) _Up(std::forward<_Args>(__args)...); |
| 145 | } |
| 146 | |
| 147 | _LIBCPP_HIDE_FROM_ABI void destroy(pointer __p) { __p->~_Tp(); } |
| 148 | }; |
| 149 | |
| 150 | template <class _Tp, class _Up> |
| 151 | inline _LIBCPP_HIDE_FROM_ABI bool operator==(const allocator<_Tp>&, const allocator<_Up>&) _NOEXCEPT { |
| 152 | return true; |
| 153 | } |
| 154 | |
| 155 | template <class _Tp, class _Up> |
| 156 | inline _LIBCPP_HIDE_FROM_ABI bool operator!=(const allocator<_Tp>&, const allocator<_Up>&) _NOEXCEPT { |
| 157 | return false; |
| 158 | } |
| 159 | |
| 160 | _LIBCPP_END_NAMESPACE_STD |
| 161 | |
| 162 | #endif // _LIBCPP___CXX03___MEMORY_ALLOCATOR_H |
| 163 |
Warning: This file is not a C or C++ file. It does not have highlighting.
