1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
|
//===----------------------------------------------------------------------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
// UNSUPPORTED: c++03, c++11, c++14
// TODO: Change to XFAIL once https://github.com/llvm/llvm-project/issues/40340 is fixed
// UNSUPPORTED: availability-pmr-missing
// test_memory_resource requires RTTI for dynamic_cast
// UNSUPPORTED: no-rtti
// <memory_resource>
// template <class T> class polymorphic_allocator
// T* polymorphic_allocator<T>::allocate(size_t n)
#include <memory_resource>
#include <cassert>
#include <exception>
#include <limits>
#include <memory>
#include <type_traits>
#include "test_macros.h"
#include "test_std_memory_resource.h"
template <std::size_t S, size_t Align>
void testForSizeAndAlign() {
struct T {
alignas(Align) std::byte buf[S];
};
TestResource R;
std::pmr::polymorphic_allocator<T> a(&R);
for (int N = 1; N <= 5; ++N) {
auto ret = a.allocate(N);
assert(R.checkAlloc(ret, N * sizeof(T), alignof(T)));
a.deallocate(ret, N);
R.reset();
}
}
#ifndef TEST_HAS_NO_EXCEPTIONS
template <std::size_t S>
void testAllocForSizeThrows() {
struct T {
std::byte buf[S];
};
using Alloc = std::pmr::polymorphic_allocator<T>;
using Traits = std::allocator_traits<Alloc>;
NullResource R;
Alloc a(&R);
// Test that allocating exactly the max size does not throw.
std::size_t maxSize = Traits::max_size(a);
std::size_t sizeTypeMax = std::numeric_limits<std::size_t>::max();
if (maxSize != sizeTypeMax) {
// Test that allocating size_t(~0) throws bad alloc.
try {
(void)a.allocate(sizeTypeMax);
assert(false);
} catch (const std::exception&) {
}
// Test that allocating even one more than the max size does throw.
std::size_t overSize = maxSize + 1;
try {
(void)a.allocate(overSize);
assert(false);
} catch (const std::exception&) {
}
}
}
#endif // TEST_HAS_NO_EXCEPTIONS
int main(int, char**) {
{
std::pmr::polymorphic_allocator<int> a;
ASSERT_SAME_TYPE(decltype(a.allocate(0)), int*);
static_assert(!noexcept(a.allocate(0)), "");
}
{
constexpr std::size_t MA = alignof(std::max_align_t);
testForSizeAndAlign<1, 1>();
testForSizeAndAlign<1, 2>();
testForSizeAndAlign<1, MA>();
testForSizeAndAlign<2, 2>();
testForSizeAndAlign<73, alignof(void*)>();
testForSizeAndAlign<73, MA>();
testForSizeAndAlign<13, MA>();
}
#ifndef TEST_HAS_NO_EXCEPTIONS
{
testAllocForSizeThrows<1>();
testAllocForSizeThrows<2>();
testAllocForSizeThrows<4>();
testAllocForSizeThrows<8>();
testAllocForSizeThrows<16>();
testAllocForSizeThrows<73>();
testAllocForSizeThrows<13>();
}
#endif
return 0;
}
|
