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
114
115
116
117
118
119
120
121
122
123
124
125
126
127
|
//////////////////////////////////////////////////////////////////////////////
//
// (C) Copyright Ion Gaztanaga 2015-2015. Distributed under the Boost
// Software License, Version 1.0. (See accompanying file
// LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
//
// See http://www.boost.org/libs/container for documentation.
//
//////////////////////////////////////////////////////////////////////////////
#include <boost/container/pmr/global_resource.hpp>
#include <boost/container/pmr/memory_resource.hpp>
#include <boost/core/lightweight_test.hpp>
#include <boost/core/no_exceptions_support.hpp>
#include "derived_from_memory_resource.hpp"
#include <cstdlib>
#include <new>
using namespace boost::container;
using namespace boost::container::pmr;
#ifdef BOOST_MSVC
#pragma warning (push)
#pragma warning (disable : 4290)
#endif
#if __cplusplus >= 201103L
#define BOOST_CONTAINER_NEW_EXCEPTION_SPECIFIER
#define BOOST_CONTAINER_DELETE_EXCEPTION_SPECIFIER noexcept
#else
#define BOOST_CONTAINER_NEW_EXCEPTION_SPECIFIER throw(std::bad_alloc)
#define BOOST_CONTAINER_DELETE_EXCEPTION_SPECIFIER throw()
#endif
#if defined(BOOST_GCC) && (BOOST_GCC >= 50000)
#pragma GCC diagnostic ignored "-Wsized-deallocation"
#endif
//ASAN does not support operator new overloading
#ifndef BOOST_CONTAINER_ASAN
std::size_t allocation_count = 0;
void* operator new[](std::size_t count) BOOST_CONTAINER_NEW_EXCEPTION_SPECIFIER
{
++allocation_count;
return std::malloc(count);
}
void operator delete[](void *p) BOOST_CONTAINER_DELETE_EXCEPTION_SPECIFIER
{
--allocation_count;
return std::free(p);
}
#endif //BOOST_CONTAINER_ASAN
#ifdef BOOST_MSVC
#pragma warning (pop)
#endif
#ifndef BOOST_CONTAINER_ASAN
void test_new_delete_resource()
{
//Make sure new_delete_resource calls new[]/delete[]
std::size_t memcount = allocation_count;
memory_resource *mr = new_delete_resource();
//each time should return the same pointer
BOOST_TEST(mr == new_delete_resource());
#if !defined(BOOST_CONTAINER_DYNAMIC_LINKING) //No new delete replacement possible new_delete is a DLL
BOOST_TEST(memcount == allocation_count);
#endif
void *addr = mr->allocate(16, 1);
#if !defined(BOOST_CONTAINER_DYNAMIC_LINKING) //No new delete replacement possible new_delete is a DLL
BOOST_TEST((allocation_count - memcount) == 1);
#endif
mr->deallocate(addr, 16, 1);
BOOST_TEST(memcount == allocation_count);
}
#endif //BOOST_CONTAINER_ASAN
void test_null_memory_resource()
{
//Make sure it throw or returns null
memory_resource *mr = null_memory_resource();
#if !defined(BOOST_NO_EXCEPTIONS)
bool bad_allocexception_thrown = false;
try{
mr->allocate(1, 1);
}
catch(std::bad_alloc&) {
bad_allocexception_thrown = true;
}
catch(...) {
}
BOOST_TEST(bad_allocexception_thrown == true);
#else
BOOST_TEST(0 == mr->allocate(1, 1));
#endif
}
void test_default_resource()
{
//Default resource must be new/delete before set_default_resource
BOOST_TEST(get_default_resource() == new_delete_resource());
//Set default resource and obtain previous
derived_from_memory_resource d;
memory_resource *prev_default = set_default_resource(&d);
BOOST_TEST(get_default_resource() == &d);
//Set default resource with null, which should be new/delete
prev_default = set_default_resource(0);
BOOST_TEST(prev_default == &d);
BOOST_TEST(get_default_resource() == new_delete_resource());
}
int main()
{
#ifndef BOOST_CONTAINER_ASAN
test_new_delete_resource();
#endif
test_null_memory_resource();
test_default_resource();
return ::boost::report_errors();
}
|
