1 //===----------------------------------------------------------------------===//
2 //
3 // The LLVM Compiler Infrastructure
4 //
5 // This file is dual licensed under the MIT and the University of Illinois Open
6 // Source Licenses. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9
10 // <memory>
11
12 // template<class Y, class D, class A> shared_ptr(Y* p, D d, A a);
13
14 #include <memory>
15 #include <cassert>
16 #include "../test_deleter.h"
17 #include "test_allocator.h"
18 #include "min_allocator.h"
19
20 struct A
21 {
22 static int count;
23
AA24 A() {++count;}
AA25 A(const A&) {++count;}
~AA26 ~A() {--count;}
27 };
28
29 int A::count = 0;
30
31
main()32 int main()
33 {
34 {
35 A* ptr = new A;
36 std::shared_ptr<A> p(ptr, test_deleter<A>(3), test_allocator<A>(5));
37 assert(A::count == 1);
38 assert(p.use_count() == 1);
39 assert(p.get() == ptr);
40 test_deleter<A>* d = std::get_deleter<test_deleter<A> >(p);
41 assert(test_deleter<A>::count == 1);
42 assert(test_deleter<A>::dealloc_count == 0);
43 assert(d);
44 assert(d->state() == 3);
45 assert(test_allocator<A>::count == 1);
46 assert(test_allocator<A>::alloc_count == 1);
47 }
48 assert(A::count == 0);
49 assert(test_deleter<A>::count == 0);
50 assert(test_deleter<A>::dealloc_count == 1);
51 assert(test_allocator<A>::count == 0);
52 assert(test_allocator<A>::alloc_count == 0);
53 test_deleter<A>::dealloc_count = 0;
54 // Test an allocator with a minimal interface
55 {
56 A* ptr = new A;
57 std::shared_ptr<A> p(ptr, test_deleter<A>(3), bare_allocator<void>());
58 assert(A::count == 1);
59 assert(p.use_count() == 1);
60 assert(p.get() == ptr);
61 test_deleter<A>* d = std::get_deleter<test_deleter<A> >(p);
62 assert(test_deleter<A>::count == 1);
63 assert(test_deleter<A>::dealloc_count == 0);
64 assert(d);
65 assert(d->state() == 3);
66 }
67 assert(A::count == 0);
68 assert(test_deleter<A>::count == 0);
69 assert(test_deleter<A>::dealloc_count == 1);
70 test_deleter<A>::dealloc_count = 0;
71 #if __cplusplus >= 201103L
72 // Test an allocator that returns class-type pointers
73 {
74 A* ptr = new A;
75 std::shared_ptr<A> p(ptr, test_deleter<A>(3), min_allocator<void>());
76 assert(A::count == 1);
77 assert(p.use_count() == 1);
78 assert(p.get() == ptr);
79 test_deleter<A>* d = std::get_deleter<test_deleter<A> >(p);
80 assert(test_deleter<A>::count == 1);
81 assert(test_deleter<A>::dealloc_count == 0);
82 assert(d);
83 assert(d->state() == 3);
84 }
85 assert(A::count == 0);
86 assert(test_deleter<A>::count == 0);
87 assert(test_deleter<A>::dealloc_count == 1);
88 #endif
89 }
90