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
128
|
/*
Copyright 2017 Glen Joseph Fernandes
(glenjofe@gmail.com)
Distributed under the Boost Software License, Version 1.0.
(http://www.boost.org/LICENSE_1_0.txt)
*/
#include <boost/core/lightweight_test.hpp>
#include <boost/make_shared.hpp>
#if !defined(BOOST_NO_CXX11_ALLOCATOR)
struct counters {
unsigned allocate;
unsigned construct;
};
template<class T = void>
class creator {
public:
typedef T value_type;
creator(counters* state)
: state_(state) { }
template<class U>
creator(const creator<U>& other)
: state_(other.state()) { }
T* allocate(std::size_t size) {
void* ptr = ::operator new(sizeof(T) * size);
++state_->allocate;
return static_cast<T*>(ptr);
}
void deallocate(T* ptr, std::size_t) {
::operator delete(ptr);
--state_->allocate;
}
template<class... Args>
void construct(T* ptr, Args&&... args) {
::new(static_cast<void*>(ptr)) T(std::forward<Args>(args)...);
++state_->construct;
}
void destroy(T* ptr) {
ptr->~T();
--state_->construct;
}
counters* state() const {
return state_;
}
private:
counters* state_;
};
template<class T, class U>
inline bool
operator==(const creator<T>& lhs, const creator<U>& rhs)
{
return lhs.state() == rhs.state();
}
template<class T, class U>
inline bool
operator!=(const creator<T>& lhs, const creator<U>& rhs)
{
return !(lhs == rhs);
}
struct deleter {
template<class U>
void operator()(U ptr) const {
delete ptr;
}
};
int main()
{
{
counters state = { };
boost::shared_ptr<int> pointer(new int(), deleter(),
creator<int>(&state));
BOOST_TEST(state.allocate == 1);
BOOST_TEST(state.construct == 0);
pointer.reset();
BOOST_TEST(state.allocate == 0);
}
{
counters state = { };
boost::shared_ptr<int> pointer =
boost::allocate_shared<int>(creator<int>(&state));
BOOST_TEST(state.allocate == 1);
BOOST_TEST(state.construct == 1);
pointer.reset();
BOOST_TEST(state.allocate == 0);
BOOST_TEST(state.construct == 0);
}
{
counters state = { };
boost::shared_ptr<int[]> pointer =
boost::allocate_shared<int[]>(creator<>(&state), 5);
BOOST_TEST(state.allocate == 1);
BOOST_TEST(state.construct == 5);
pointer.reset();
BOOST_TEST(state.allocate == 0);
BOOST_TEST(state.construct == 0);
}
{
counters state = { };
boost::shared_ptr<int[5]> pointer =
boost::allocate_shared<int[5]>(creator<>(&state));
BOOST_TEST(state.allocate == 1);
BOOST_TEST(state.construct == 5);
pointer.reset();
BOOST_TEST(state.allocate == 0);
BOOST_TEST(state.construct == 0);
}
return boost::report_errors();
}
#else
int main()
{
return 0;
}
#endif
|
