Howard Hinnant | c52f43e | 2010-08-22 00:59:46 +0000 | [diff] [blame] | 1 | //===----------------------------------------------------------------------===// |
| 2 | // |
| 3 | // The LLVM Compiler Infrastructure |
| 4 | // |
Howard Hinnant | b64f8b0 | 2010-11-16 22:09:02 +0000 | [diff] [blame] | 5 | // This file is dual licensed under the MIT and the University of Illinois Open |
| 6 | // Source Licenses. See LICENSE.TXT for details. |
Howard Hinnant | c52f43e | 2010-08-22 00:59:46 +0000 | [diff] [blame] | 7 | // |
| 8 | //===----------------------------------------------------------------------===// |
| 9 | |
| 10 | // <memory> |
| 11 | |
| 12 | // allocator: |
| 13 | // pointer allocate(size_type n, allocator<void>::const_pointer hint=0); |
| 14 | |
| 15 | #include <memory> |
Howard Hinnant | c52f43e | 2010-08-22 00:59:46 +0000 | [diff] [blame] | 16 | #include <cassert> |
| 17 | |
Eric Fiselier | a831287 | 2018-03-22 04:42:56 +0000 | [diff] [blame^] | 18 | #include "test_macros.h" |
Eric Fiselier | 4eb5b6d | 2014-12-22 22:38:59 +0000 | [diff] [blame] | 19 | #include "count_new.hpp" |
Howard Hinnant | c52f43e | 2010-08-22 00:59:46 +0000 | [diff] [blame] | 20 | |
Howard Hinnant | c52f43e | 2010-08-22 00:59:46 +0000 | [diff] [blame] | 21 | |
Eric Fiselier | a831287 | 2018-03-22 04:42:56 +0000 | [diff] [blame^] | 22 | #ifdef TEST_HAS_NO_ALIGNED_ALLOCATION |
| 23 | static const bool UsingAlignedNew = false; |
| 24 | #else |
| 25 | static const bool UsingAlignedNew = true; |
| 26 | #endif |
| 27 | |
| 28 | #ifdef __STDCPP_DEFAULT_NEW_ALIGNMENT__ |
| 29 | static const size_t MaxAligned = __STDCPP_DEFAULT_NEW_ALIGNMENT__; |
| 30 | #else |
| 31 | static const size_t MaxAligned = std::alignment_of<std::max_align_t>::value; |
| 32 | #endif |
| 33 | |
| 34 | static const size_t OverAligned = MaxAligned * 2; |
| 35 | |
| 36 | |
| 37 | template <size_t Align> |
| 38 | struct TEST_ALIGNAS(Align) AlignedType { |
| 39 | char data; |
| 40 | static int constructed; |
| 41 | AlignedType() { ++constructed; } |
| 42 | AlignedType(AlignedType const&) { ++constructed; } |
| 43 | ~AlignedType() { --constructed; } |
Howard Hinnant | c52f43e | 2010-08-22 00:59:46 +0000 | [diff] [blame] | 44 | }; |
Eric Fiselier | a831287 | 2018-03-22 04:42:56 +0000 | [diff] [blame^] | 45 | template <size_t Align> |
| 46 | int AlignedType<Align>::constructed = 0; |
Howard Hinnant | c52f43e | 2010-08-22 00:59:46 +0000 | [diff] [blame] | 47 | |
Eric Fiselier | a831287 | 2018-03-22 04:42:56 +0000 | [diff] [blame^] | 48 | |
| 49 | template <size_t Align> |
| 50 | void test_aligned() { |
| 51 | typedef AlignedType<Align> T; |
| 52 | T::constructed = 0; |
| 53 | globalMemCounter.reset(); |
| 54 | std::allocator<T> a; |
| 55 | const bool IsOverAlignedType = Align > MaxAligned; |
| 56 | const bool ExpectAligned = IsOverAlignedType && UsingAlignedNew; |
| 57 | { |
Eric Fiselier | 4eb5b6d | 2014-12-22 22:38:59 +0000 | [diff] [blame] | 58 | assert(globalMemCounter.checkOutstandingNewEq(0)); |
Eric Fiselier | a831287 | 2018-03-22 04:42:56 +0000 | [diff] [blame^] | 59 | assert(T::constructed == 0); |
Eric Fiselier | 4eb5b6d | 2014-12-22 22:38:59 +0000 | [diff] [blame] | 60 | globalMemCounter.last_new_size = 0; |
Eric Fiselier | a831287 | 2018-03-22 04:42:56 +0000 | [diff] [blame^] | 61 | globalMemCounter.last_new_align = 0; |
| 62 | T* volatile ap = a.allocate(3); |
Eric Fiselier | 4eb5b6d | 2014-12-22 22:38:59 +0000 | [diff] [blame] | 63 | assert(globalMemCounter.checkOutstandingNewEq(1)); |
Eric Fiselier | a831287 | 2018-03-22 04:42:56 +0000 | [diff] [blame^] | 64 | assert(globalMemCounter.checkNewCalledEq(1)); |
| 65 | assert(globalMemCounter.checkAlignedNewCalledEq(ExpectAligned)); |
| 66 | assert(globalMemCounter.checkLastNewSizeEq(3 * sizeof(T))); |
| 67 | assert(globalMemCounter.checkLastNewAlignEq(ExpectAligned ? Align : 0)); |
| 68 | assert(T::constructed == 0); |
| 69 | globalMemCounter.last_delete_align = 0; |
Howard Hinnant | c52f43e | 2010-08-22 00:59:46 +0000 | [diff] [blame] | 70 | a.deallocate(ap, 3); |
Eric Fiselier | 4eb5b6d | 2014-12-22 22:38:59 +0000 | [diff] [blame] | 71 | assert(globalMemCounter.checkOutstandingNewEq(0)); |
Eric Fiselier | a831287 | 2018-03-22 04:42:56 +0000 | [diff] [blame^] | 72 | assert(globalMemCounter.checkDeleteCalledEq(1)); |
| 73 | assert(globalMemCounter.checkAlignedDeleteCalledEq(ExpectAligned)); |
| 74 | assert(globalMemCounter.checkLastDeleteAlignEq(ExpectAligned ? Align : 0)); |
| 75 | assert(T::constructed == 0); |
| 76 | } |
| 77 | globalMemCounter.reset(); |
| 78 | { |
Eric Fiselier | 4eb5b6d | 2014-12-22 22:38:59 +0000 | [diff] [blame] | 79 | globalMemCounter.last_new_size = 0; |
Eric Fiselier | a831287 | 2018-03-22 04:42:56 +0000 | [diff] [blame^] | 80 | globalMemCounter.last_new_align = 0; |
| 81 | T* volatile ap2 = a.allocate(11, (const void*)5); |
Eric Fiselier | 4eb5b6d | 2014-12-22 22:38:59 +0000 | [diff] [blame] | 82 | assert(globalMemCounter.checkOutstandingNewEq(1)); |
Eric Fiselier | a831287 | 2018-03-22 04:42:56 +0000 | [diff] [blame^] | 83 | assert(globalMemCounter.checkNewCalledEq(1)); |
| 84 | assert(globalMemCounter.checkAlignedNewCalledEq(ExpectAligned)); |
| 85 | assert(globalMemCounter.checkLastNewSizeEq(11 * sizeof(T))); |
| 86 | assert(globalMemCounter.checkLastNewAlignEq(ExpectAligned ? Align : 0)); |
| 87 | assert(T::constructed == 0); |
| 88 | globalMemCounter.last_delete_align = 0; |
| 89 | a.deallocate(ap2, 11); |
Eric Fiselier | 4eb5b6d | 2014-12-22 22:38:59 +0000 | [diff] [blame] | 90 | assert(globalMemCounter.checkOutstandingNewEq(0)); |
Eric Fiselier | a831287 | 2018-03-22 04:42:56 +0000 | [diff] [blame^] | 91 | assert(globalMemCounter.checkDeleteCalledEq(1)); |
| 92 | assert(globalMemCounter.checkAlignedDeleteCalledEq(ExpectAligned)); |
| 93 | assert(globalMemCounter.checkLastDeleteAlignEq(ExpectAligned ? Align : 0)); |
| 94 | assert(T::constructed == 0); |
| 95 | } |
| 96 | } |
| 97 | |
| 98 | int main() { |
| 99 | test_aligned<1>(); |
| 100 | test_aligned<2>(); |
| 101 | test_aligned<4>(); |
| 102 | test_aligned<8>(); |
| 103 | test_aligned<16>(); |
| 104 | test_aligned<MaxAligned>(); |
| 105 | test_aligned<OverAligned>(); |
| 106 | test_aligned<OverAligned * 2>(); |
Howard Hinnant | c52f43e | 2010-08-22 00:59:46 +0000 | [diff] [blame] | 107 | } |