1 //===----------------------------------------------------------------------===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 
9 // <memory>
10 
11 // shared_ptr
12 
13 // template<class Y, class D> void reset(Y* p, D d);
14 
15 #include <memory>
16 #include <cassert>
17 #include "test_macros.h"
18 #include "deleter_types.h"
19 
20 struct B
21 {
22     static int count;
23 
BB24     B() {++count;}
BB25     B(const B&) {++count;}
~BB26     virtual ~B() {--count;}
27 };
28 
29 int B::count = 0;
30 
31 struct A
32     : public B
33 {
34     static int count;
35 
AA36     A() {++count;}
AA37     A(const A& other) : B(other) {++count;}
~AA38     ~A() {--count;}
39 };
40 
41 int A::count = 0;
42 
main(int,char **)43 int main(int, char**)
44 {
45     {
46         std::shared_ptr<B> p(new B);
47         A* ptr = new A;
48         p.reset(ptr, test_deleter<A>(3));
49         assert(A::count == 1);
50         assert(B::count == 1);
51         assert(p.use_count() == 1);
52         assert(p.get() == ptr);
53         assert(test_deleter<A>::count == 1);
54         assert(test_deleter<A>::dealloc_count == 0);
55 #ifndef TEST_HAS_NO_RTTI
56         test_deleter<A>* d = std::get_deleter<test_deleter<A> >(p);
57         assert(d);
58         assert(d->state() == 3);
59 #endif
60     }
61     assert(A::count == 0);
62     assert(test_deleter<A>::count == 0);
63     assert(test_deleter<A>::dealloc_count == 1);
64     {
65         std::shared_ptr<B> p;
66         A* ptr = new A;
67         p.reset(ptr, test_deleter<A>(3));
68         assert(A::count == 1);
69         assert(B::count == 1);
70         assert(p.use_count() == 1);
71         assert(p.get() == ptr);
72         assert(test_deleter<A>::count == 1);
73         assert(test_deleter<A>::dealloc_count == 1);
74 #ifndef TEST_HAS_NO_RTTI
75         test_deleter<A>* d = std::get_deleter<test_deleter<A> >(p);
76         assert(d);
77         assert(d->state() == 3);
78 #endif
79     }
80     assert(A::count == 0);
81     assert(test_deleter<A>::count == 0);
82     assert(test_deleter<A>::dealloc_count == 2);
83 
84 #if TEST_STD_VER > 14
85     {
86         std::shared_ptr<const A[]> p;
87         A* ptr = new A[8];
88         p.reset(ptr, CDeleter<A[]>());
89         assert(A::count == 8);
90         assert(p.use_count() == 1);
91         assert(p.get() == ptr);
92     }
93     assert(A::count == 0);
94 #endif
95 
96   return 0;
97 }
98