[1166] | 1 | // Allocator that wraps "C" malloc -*- C++ -*-
|
---|
| 2 |
|
---|
| 3 | // Copyright (C) 2001-2021 Free Software Foundation, Inc.
|
---|
| 4 | //
|
---|
| 5 | // This file is part of the GNU ISO C++ Library. This library is free
|
---|
| 6 | // software; you can redistribute it and/or modify it under the
|
---|
| 7 | // terms of the GNU General Public License as published by the
|
---|
| 8 | // Free Software Foundation; either version 3, or (at your option)
|
---|
| 9 | // any later version.
|
---|
| 10 |
|
---|
| 11 | // This library is distributed in the hope that it will be useful,
|
---|
| 12 | // but WITHOUT ANY WARRANTY; without even the implied warranty of
|
---|
| 13 | // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
---|
| 14 | // GNU General Public License for more details.
|
---|
| 15 |
|
---|
| 16 | // Under Section 7 of GPL version 3, you are granted additional
|
---|
| 17 | // permissions described in the GCC Runtime Library Exception, version
|
---|
| 18 | // 3.1, as published by the Free Software Foundation.
|
---|
| 19 |
|
---|
| 20 | // You should have received a copy of the GNU General Public License and
|
---|
| 21 | // a copy of the GCC Runtime Library Exception along with this program;
|
---|
| 22 | // see the files COPYING3 and COPYING.RUNTIME respectively. If not, see
|
---|
| 23 | // <http://www.gnu.org/licenses/>.
|
---|
| 24 |
|
---|
| 25 | /** @file ext/malloc_allocator.h
|
---|
| 26 | * This file is a GNU extension to the Standard C++ Library.
|
---|
| 27 | */
|
---|
| 28 |
|
---|
| 29 | #ifndef _MALLOC_ALLOCATOR_H
|
---|
| 30 | #define _MALLOC_ALLOCATOR_H 1
|
---|
| 31 |
|
---|
| 32 | #include <cstdlib>
|
---|
| 33 | #include <cstddef>
|
---|
| 34 | #include <new>
|
---|
| 35 | #include <bits/functexcept.h>
|
---|
| 36 | #include <bits/move.h>
|
---|
| 37 | #if __cplusplus >= 201103L
|
---|
| 38 | #include <type_traits>
|
---|
| 39 | #endif
|
---|
| 40 |
|
---|
| 41 | namespace __gnu_cxx _GLIBCXX_VISIBILITY(default)
|
---|
| 42 | {
|
---|
| 43 | _GLIBCXX_BEGIN_NAMESPACE_VERSION
|
---|
| 44 |
|
---|
| 45 | /**
|
---|
| 46 | * @brief An allocator that uses malloc.
|
---|
| 47 | * @ingroup allocators
|
---|
| 48 | *
|
---|
| 49 | * This is precisely the allocator defined in the C++ Standard.
|
---|
| 50 | * - all allocation calls malloc
|
---|
| 51 | * - all deallocation calls free
|
---|
| 52 | */
|
---|
| 53 | template<typename _Tp>
|
---|
| 54 | class malloc_allocator
|
---|
| 55 | {
|
---|
| 56 | public:
|
---|
| 57 | typedef _Tp value_type;
|
---|
| 58 | typedef std::size_t size_type;
|
---|
| 59 | typedef std::ptrdiff_t difference_type;
|
---|
| 60 | #if __cplusplus <= 201703L
|
---|
| 61 | typedef _Tp* pointer;
|
---|
| 62 | typedef const _Tp* const_pointer;
|
---|
| 63 | typedef _Tp& reference;
|
---|
| 64 | typedef const _Tp& const_reference;
|
---|
| 65 |
|
---|
| 66 | template<typename _Tp1>
|
---|
| 67 | struct rebind
|
---|
| 68 | { typedef malloc_allocator<_Tp1> other; };
|
---|
| 69 | #endif
|
---|
| 70 |
|
---|
| 71 | #if __cplusplus >= 201103L
|
---|
| 72 | // _GLIBCXX_RESOLVE_LIB_DEFECTS
|
---|
| 73 | // 2103. propagate_on_container_move_assignment
|
---|
| 74 | typedef std::true_type propagate_on_container_move_assignment;
|
---|
| 75 | #endif
|
---|
| 76 |
|
---|
| 77 | _GLIBCXX20_CONSTEXPR
|
---|
| 78 | malloc_allocator() _GLIBCXX_USE_NOEXCEPT { }
|
---|
| 79 |
|
---|
| 80 | _GLIBCXX20_CONSTEXPR
|
---|
| 81 | malloc_allocator(const malloc_allocator&) _GLIBCXX_USE_NOEXCEPT { }
|
---|
| 82 |
|
---|
| 83 | template<typename _Tp1>
|
---|
| 84 | _GLIBCXX20_CONSTEXPR
|
---|
| 85 | malloc_allocator(const malloc_allocator<_Tp1>&)
|
---|
| 86 | _GLIBCXX_USE_NOEXCEPT { }
|
---|
| 87 |
|
---|
| 88 | #if __cplusplus <= 201703L
|
---|
| 89 | ~malloc_allocator() _GLIBCXX_USE_NOEXCEPT { }
|
---|
| 90 |
|
---|
| 91 | pointer
|
---|
| 92 | address(reference __x) const _GLIBCXX_NOEXCEPT
|
---|
| 93 | { return std::__addressof(__x); }
|
---|
| 94 |
|
---|
| 95 | const_pointer
|
---|
| 96 | address(const_reference __x) const _GLIBCXX_NOEXCEPT
|
---|
| 97 | { return std::__addressof(__x); }
|
---|
| 98 | #endif
|
---|
| 99 |
|
---|
| 100 | // NB: __n is permitted to be 0. The C++ standard says nothing
|
---|
| 101 | // about what the return value is when __n == 0.
|
---|
| 102 | _GLIBCXX_NODISCARD _Tp*
|
---|
| 103 | allocate(size_type __n, const void* = 0)
|
---|
| 104 | {
|
---|
| 105 | #if __cplusplus >= 201103L
|
---|
| 106 | // _GLIBCXX_RESOLVE_LIB_DEFECTS
|
---|
| 107 | // 3308. std::allocator<void>().allocate(n)
|
---|
| 108 | static_assert(sizeof(_Tp) != 0, "cannot allocate incomplete types");
|
---|
| 109 | #endif
|
---|
| 110 |
|
---|
| 111 | if (__builtin_expect(__n > this->_M_max_size(), false))
|
---|
| 112 | {
|
---|
| 113 | // _GLIBCXX_RESOLVE_LIB_DEFECTS
|
---|
| 114 | // 3190. allocator::allocate sometimes returns too little storage
|
---|
| 115 | if (__n > (std::size_t(-1) / sizeof(_Tp)))
|
---|
| 116 | std::__throw_bad_array_new_length();
|
---|
| 117 | std::__throw_bad_alloc();
|
---|
| 118 | }
|
---|
| 119 |
|
---|
| 120 | _Tp* __ret = 0;
|
---|
| 121 | #if __cpp_aligned_new
|
---|
| 122 | #if __cplusplus > 201402L && _GLIBCXX_HAVE_ALIGNED_ALLOC
|
---|
| 123 | if (alignof(_Tp) > alignof(std::max_align_t))
|
---|
| 124 | {
|
---|
| 125 | __ret = static_cast<_Tp*>(::aligned_alloc(alignof(_Tp),
|
---|
| 126 | __n * sizeof(_Tp)));
|
---|
| 127 | }
|
---|
| 128 | #else
|
---|
| 129 | # define _GLIBCXX_CHECK_MALLOC_RESULT
|
---|
| 130 | #endif
|
---|
| 131 | #endif
|
---|
| 132 | if (!__ret)
|
---|
| 133 | __ret = static_cast<_Tp*>(std::malloc(__n * sizeof(_Tp)));
|
---|
| 134 | if (!__ret)
|
---|
| 135 | std::__throw_bad_alloc();
|
---|
| 136 | #ifdef _GLIBCXX_CHECK_MALLOC_RESULT
|
---|
| 137 | #undef _GLIBCXX_CHECK_MALLOC_RESULT
|
---|
| 138 | if (reinterpret_cast<std::size_t>(__ret) % alignof(_Tp))
|
---|
| 139 | {
|
---|
| 140 | // Memory returned by malloc is not suitably aligned for _Tp.
|
---|
| 141 | deallocate(__ret, __n);
|
---|
| 142 | std::__throw_bad_alloc();
|
---|
| 143 | }
|
---|
| 144 | #endif
|
---|
| 145 | return __ret;
|
---|
| 146 | }
|
---|
| 147 |
|
---|
| 148 | // __p is not permitted to be a null pointer.
|
---|
| 149 | void
|
---|
| 150 | deallocate(_Tp* __p, size_type)
|
---|
| 151 | { std::free(static_cast<void*>(__p)); }
|
---|
| 152 |
|
---|
| 153 | #if __cplusplus <= 201703L
|
---|
| 154 | size_type
|
---|
| 155 | max_size() const _GLIBCXX_USE_NOEXCEPT
|
---|
| 156 | { return _M_max_size(); }
|
---|
| 157 |
|
---|
| 158 | #if __cplusplus >= 201103L
|
---|
| 159 | template<typename _Up, typename... _Args>
|
---|
| 160 | void
|
---|
| 161 | construct(_Up* __p, _Args&&... __args)
|
---|
| 162 | noexcept(std::is_nothrow_constructible<_Up, _Args...>::value)
|
---|
| 163 | { ::new((void *)__p) _Up(std::forward<_Args>(__args)...); }
|
---|
| 164 |
|
---|
| 165 | template<typename _Up>
|
---|
| 166 | void
|
---|
| 167 | destroy(_Up* __p)
|
---|
| 168 | noexcept(std::is_nothrow_destructible<_Up>::value)
|
---|
| 169 | { __p->~_Up(); }
|
---|
| 170 | #else
|
---|
| 171 | // _GLIBCXX_RESOLVE_LIB_DEFECTS
|
---|
| 172 | // 402. wrong new expression in [some_] allocator::construct
|
---|
| 173 | void
|
---|
| 174 | construct(pointer __p, const _Tp& __val)
|
---|
| 175 | { ::new((void *)__p) value_type(__val); }
|
---|
| 176 |
|
---|
| 177 | void
|
---|
| 178 | destroy(pointer __p) { __p->~_Tp(); }
|
---|
| 179 | #endif
|
---|
| 180 | #endif // ! C++20
|
---|
| 181 |
|
---|
| 182 | template<typename _Up>
|
---|
| 183 | friend _GLIBCXX20_CONSTEXPR bool
|
---|
| 184 | operator==(const malloc_allocator&, const malloc_allocator<_Up>&)
|
---|
| 185 | _GLIBCXX_NOTHROW
|
---|
| 186 | { return true; }
|
---|
| 187 |
|
---|
| 188 | #if __cpp_impl_three_way_comparison < 201907L
|
---|
| 189 | template<typename _Up>
|
---|
| 190 | friend _GLIBCXX20_CONSTEXPR bool
|
---|
| 191 | operator!=(const malloc_allocator&, const malloc_allocator<_Up>&)
|
---|
| 192 | _GLIBCXX_NOTHROW
|
---|
| 193 | { return false; }
|
---|
| 194 | #endif
|
---|
| 195 |
|
---|
| 196 | private:
|
---|
| 197 | _GLIBCXX_CONSTEXPR size_type
|
---|
| 198 | _M_max_size() const _GLIBCXX_USE_NOEXCEPT
|
---|
| 199 | {
|
---|
| 200 | #if __PTRDIFF_MAX__ < __SIZE_MAX__
|
---|
| 201 | return std::size_t(__PTRDIFF_MAX__) / sizeof(_Tp);
|
---|
| 202 | #else
|
---|
| 203 | return std::size_t(-1) / sizeof(_Tp);
|
---|
| 204 | #endif
|
---|
| 205 | }
|
---|
| 206 | };
|
---|
| 207 |
|
---|
| 208 | _GLIBCXX_END_NAMESPACE_VERSION
|
---|
| 209 | } // namespace
|
---|
| 210 |
|
---|
| 211 | #endif
|
---|