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> explicit shared_ptr(auto_ptr<Y>&& r);
13 
14 #include <memory>
15 #include <new>
16 #include <cstdlib>
17 #include <cassert>
18 
19 bool throw_next = false;
20 
21 void* operator new(std::size_t s) throw(std::bad_alloc)
22 {
23     if (throw_next)
24         throw std::bad_alloc();
25     return std::malloc(s);
26 }
27 
28 void  operator delete(void* p) throw()
29 {
30     std::free(p);
31 }
32 
33 struct B
34 {
35     static int count;
36 
37     B() {++count;}
38     B(const B&) {++count;}
39     virtual ~B() {--count;}
40 };
41 
42 int B::count = 0;
43 
44 struct A
45     : public B
46 {
47     static int count;
48 
49     A() {++count;}
50     A(const A&) {++count;}
51     ~A() {--count;}
52 };
53 
54 int A::count = 0;
55 
56 int main()
57 {
58     {
59     std::auto_ptr<A> ptr(new A);
60     A* raw_ptr = ptr.get();
61 #ifndef _LIBCPP_HAS_NO_RVALUE_REFERENCES
62     std::shared_ptr<B> p(std::move(ptr));
63 #else
64     std::shared_ptr<B> p(ptr);
65 #endif
66     assert(A::count == 1);
67     assert(B::count == 1);
68     assert(p.use_count() == 1);
69     assert(p.get() == raw_ptr);
70     assert(ptr.get() == 0);
71     }
72     assert(A::count == 0);
73     {
74     std::auto_ptr<A> ptr(new A);
75     A* raw_ptr = ptr.get();
76     throw_next = true;
77     try
78     {
79 #ifndef _LIBCPP_HAS_NO_RVALUE_REFERENCES
80         std::shared_ptr<B> p(std::move(ptr));
81 #else
82         std::shared_ptr<B> p(ptr);
83 #endif
84         assert(false);
85     }
86     catch (...)
87     {
88 #ifndef _LIBCPP_HAS_NO_RVALUE_REFERENCES
89         assert(A::count == 1);
90         assert(B::count == 1);
91         assert(ptr.get() == raw_ptr);
92 #else
93         // Without rvalue references, ptr got copied into
94         // the shared_ptr destructor and the copy was
95         // destroyed during unwinding.
96         assert(A::count == 0);
97         assert(B::count == 0);
98 #endif
99     }
100     }
101     assert(A::count == 0);
102 }
103