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
|
//===----------------------------------------------------------------------===//
//
// 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
//
//===----------------------------------------------------------------------===//
// <string>
// Make sure the size we allocate and deallocate match. See https://github.com/llvm/llvm-project/pull/90292.
#include <string>
#include <cassert>
#include <cstdint>
#include <type_traits>
#include "test_macros.h"
static int allocated_;
template <class T, class Sz>
struct test_alloc {
typedef Sz size_type;
typedef typename std::make_signed<Sz>::type difference_type;
typedef T value_type;
typedef value_type* pointer;
typedef const value_type* const_pointer;
typedef typename std::add_lvalue_reference<value_type>::type reference;
typedef typename std::add_lvalue_reference<const value_type>::type const_reference;
template <class U>
struct rebind {
typedef test_alloc<U, Sz> other;
};
TEST_CONSTEXPR test_alloc() TEST_NOEXCEPT {}
template <class U>
TEST_CONSTEXPR test_alloc(const test_alloc<U, Sz>&) TEST_NOEXCEPT {}
pointer allocate(size_type n, const void* = nullptr) {
allocated_ += n;
return std::allocator<value_type>().allocate(n);
}
void deallocate(pointer p, size_type s) {
allocated_ -= s;
std::allocator<value_type>().deallocate(p, s);
}
template <class U>
friend TEST_CONSTEXPR bool operator==(const test_alloc&, const test_alloc<U, Sz>&) TEST_NOEXCEPT {
return true;
}
#if TEST_STD_VER < 20
template <class U>
friend TEST_CONSTEXPR bool operator!=(const test_alloc&, const test_alloc<U, Sz>&) TEST_NOEXCEPT {
return false;
}
#endif
};
template <class Sz>
void test() {
for (int i = 1; i < 1000; ++i) {
using Str = std::basic_string<char, std::char_traits<char>, test_alloc<char, Sz> >;
{
Str s(i, 't');
assert(allocated_ == 0 || allocated_ >= i);
}
}
assert(allocated_ == 0);
}
int main(int, char**) {
test<uint32_t>();
test<uint64_t>();
test<size_t>();
return 0;
}
|
