[libcxx] Allow shared_ptr's unique_ptr converting constructor to support array types.
Refs: https://bugs.llvm.org/show_bug.cgi?id=32147
Differential Revision: https://reviews.llvm.org/D80882
GitOrigin-RevId: 097d77d611d1e1b3972be661fdc3caaa4d1824b4
diff --git a/include/memory b/include/memory
index 484dafe..d326476 100644
--- a/include/memory
+++ b/include/memory
@@ -2702,7 +2702,6 @@
typename enable_if
<
!is_lvalue_reference<_Dp>::value &&
- !is_array<_Yp>::value &&
is_convertible<typename unique_ptr<_Yp, _Dp>::pointer, element_type*>::value,
__nat
>::type = __nat());
@@ -2711,7 +2710,6 @@
typename enable_if
<
is_lvalue_reference<_Dp>::value &&
- !is_array<_Yp>::value &&
is_convertible<typename unique_ptr<_Yp, _Dp>::pointer, element_type*>::value,
__nat
>::type = __nat());
@@ -2752,7 +2750,6 @@
template <class _Yp, class _Dp>
typename enable_if
<
- !is_array<_Yp>::value &&
is_convertible<typename unique_ptr<_Yp, _Dp>::pointer, element_type*>::value,
shared_ptr&
>::type
@@ -3132,7 +3129,6 @@
typename enable_if
<
!is_lvalue_reference<_Dp>::value &&
- !is_array<_Yp>::value &&
is_convertible<typename unique_ptr<_Yp, _Dp>::pointer, element_type*>::value,
__nat
>::type)
@@ -3145,7 +3141,7 @@
#endif
{
typedef typename __shared_ptr_default_allocator<_Yp>::type _AllocT;
- typedef __shared_ptr_pointer<_Yp*, _Dp, _AllocT > _CntrlBlk;
+ typedef __shared_ptr_pointer<typename unique_ptr<_Yp, _Dp>::pointer, _Dp, _AllocT > _CntrlBlk;
__cntrl_ = new _CntrlBlk(__r.get(), __r.get_deleter(), _AllocT());
__enable_weak_this(__r.get(), __r.get());
}
@@ -3158,7 +3154,6 @@
typename enable_if
<
is_lvalue_reference<_Dp>::value &&
- !is_array<_Yp>::value &&
is_convertible<typename unique_ptr<_Yp, _Dp>::pointer, element_type*>::value,
__nat
>::type)
@@ -3171,7 +3166,7 @@
#endif
{
typedef typename __shared_ptr_default_allocator<_Yp>::type _AllocT;
- typedef __shared_ptr_pointer<_Yp*,
+ typedef __shared_ptr_pointer<typename unique_ptr<_Yp, _Dp>::pointer,
reference_wrapper<typename remove_reference<_Dp>::type>,
_AllocT > _CntrlBlk;
__cntrl_ = new _CntrlBlk(__r.get(), _VSTD::ref(__r.get_deleter()), _AllocT());
@@ -3255,7 +3250,6 @@
inline
typename enable_if
<
- !is_array<_Yp>::value &&
is_convertible<typename unique_ptr<_Yp, _Dp>::pointer,
typename shared_ptr<_Tp>::element_type*>::value,
shared_ptr<_Tp>&
diff --git a/test/std/utilities/memory/util.smartptr/util.smartptr.shared/util.smartptr.shared.assign/unique_ptr_Y.pass.cpp b/test/std/utilities/memory/util.smartptr/util.smartptr.shared/util.smartptr.shared.assign/unique_ptr_Y.pass.cpp
index 0096897..f9e1798 100644
--- a/test/std/utilities/memory/util.smartptr/util.smartptr.shared/util.smartptr.shared.assign/unique_ptr_Y.pass.cpp
+++ b/test/std/utilities/memory/util.smartptr/util.smartptr.shared/util.smartptr.shared.assign/unique_ptr_Y.pass.cpp
@@ -41,6 +41,19 @@
int A::count = 0;
+template <class T>
+struct StatefulArrayDeleter {
+ int state = 0;
+
+ StatefulArrayDeleter(int val = 0) : state(val) {}
+ StatefulArrayDeleter(StatefulArrayDeleter const&) { assert(false); }
+
+ void operator()(T* ptr) {
+ assert(state == 42);
+ delete []ptr;
+ }
+};
+
int main(int, char**)
{
{
@@ -112,5 +125,82 @@
assert(B::count == 0);
assert(A::count == 0);
+ {
+ std::unique_ptr<A[]> ptr(new A[8]);
+ A* raw_ptr = ptr.get();
+ std::shared_ptr<B> p;
+ p = std::move(ptr);
+ assert(A::count == 8);
+ assert(B::count == 8);
+ assert(p.use_count() == 1);
+ assert(p.get() == raw_ptr);
+ assert(ptr.get() == 0);
+ }
+ assert(A::count == 0);
+ assert(B::count == 0);
+
+ {
+ std::unique_ptr<A[]> ptr(new A[8]);
+ A* raw_ptr = ptr.get();
+ std::shared_ptr<A> p;
+ p = std::move(ptr);
+ assert(A::count == 8);
+ assert(p.use_count() == 1);
+ assert(p.get() == raw_ptr);
+ assert(ptr.get() == 0);
+ }
+ assert(A::count == 0);
+
+ {
+ std::unique_ptr<int[]> ptr(new int[8]);
+ std::shared_ptr<int> p;
+ p = std::move(ptr);
+ }
+
+#if TEST_STD_VER > 14
+ {
+ StatefulArrayDeleter<A> d;
+ std::unique_ptr<A[], StatefulArrayDeleter<A>&> u(new A[4], d);
+ std::shared_ptr<A[]> p;
+ p = std::move(u);
+ d.state = 42;
+ assert(A::count == 4);
+ }
+ assert(A::count == 0);
+ assert(B::count == 0);
+
+ {
+ std::unique_ptr<A[]> ptr(new A[8]);
+ A* raw_ptr = ptr.get();
+ std::shared_ptr<B[]> p;
+ p = std::move(ptr);
+ assert(A::count == 8);
+ assert(B::count == 8);
+ assert(p.use_count() == 1);
+ assert(p.get() == raw_ptr);
+ assert(ptr.get() == 0);
+ }
+ assert(A::count == 0);
+ assert(B::count == 0);
+
+ {
+ std::unique_ptr<A[]> ptr(new A[8]);
+ A* raw_ptr = ptr.get();
+ std::shared_ptr<A[]> p;
+ p = std::move(ptr);
+ assert(A::count == 8);
+ assert(p.use_count() == 1);
+ assert(p.get() == raw_ptr);
+ assert(ptr.get() == 0);
+ }
+ assert(A::count == 0);
+
+ {
+ std::unique_ptr<int[]> ptr(new int[8]);
+ std::shared_ptr<int[]> p;
+ p = std::move(ptr);
+ }
+#endif // TEST_STD_VER >= 14
+
return 0;
}
diff --git a/test/std/utilities/memory/util.smartptr/util.smartptr.shared/util.smartptr.shared.const/unique_ptr.pass.cpp b/test/std/utilities/memory/util.smartptr/util.smartptr.shared/util.smartptr.shared.const/unique_ptr.pass.cpp
index 398c64e..ad88a3e 100644
--- a/test/std/utilities/memory/util.smartptr/util.smartptr.shared/util.smartptr.shared.const/unique_ptr.pass.cpp
+++ b/test/std/utilities/memory/util.smartptr/util.smartptr.shared/util.smartptr.shared.const/unique_ptr.pass.cpp
@@ -10,7 +10,7 @@
// <memory>
-// template <class Y, class D> explicit shared_ptr(unique_ptr<Y, D>&&r);
+// template <class Y, class D> shared_ptr(unique_ptr<Y, D>&&r);
#include <memory>
#include <new>
@@ -69,6 +69,19 @@
}
};
+template <class T>
+struct StatefulArrayDeleter {
+ int state = 0;
+
+ StatefulArrayDeleter(int val = 0) : state(val) {}
+ StatefulArrayDeleter(StatefulArrayDeleter const&) { assert(false); }
+
+ void operator()(T* ptr) {
+ assert(state == 42);
+ delete []ptr;
+ }
+};
+
int main(int, char**)
{
{
@@ -135,5 +148,76 @@
std::shared_ptr<int> s = std::move(u);
}
- return 0;
+ assert(A::count == 0);
+ {
+ std::unique_ptr<A[]> ptr(new A[8]);
+ A* raw_ptr = ptr.get();
+ std::shared_ptr<B> p(std::move(ptr));
+ assert(A::count == 8);
+ assert(B::count == 8);
+ assert(p.use_count() == 1);
+ assert(p.get() == raw_ptr);
+ assert(ptr.get() == 0);
+ }
+ assert(A::count == 0);
+ assert(B::count == 0);
+
+ {
+ std::unique_ptr<A[]> ptr(new A[8]);
+ A* raw_ptr = ptr.get();
+ std::shared_ptr<A> p(std::move(ptr));
+ assert(A::count == 8);
+ assert(p.use_count() == 1);
+ assert(p.get() == raw_ptr);
+ assert(ptr.get() == 0);
+ }
+ assert(A::count == 0);
+
+ {
+ std::unique_ptr<int[]> ptr(new int[8]);
+ std::shared_ptr<int> p(std::move(ptr));
+ }
+
+#if TEST_STD_VER > 14
+ {
+ StatefulArrayDeleter<A> d;
+ std::unique_ptr<A[], StatefulArrayDeleter<A>&> u(new A[4], d);
+ std::shared_ptr<A[]> p(std::move(u));
+ d.state = 42;
+ assert(A::count == 4);
+ }
+ assert(A::count == 0);
+ assert(B::count == 0);
+
+ {
+ std::unique_ptr<A[]> ptr(new A[8]);
+ A* raw_ptr = ptr.get();
+ std::shared_ptr<B[]> p(std::move(ptr));
+ assert(A::count == 8);
+ assert(B::count == 8);
+ assert(p.use_count() == 1);
+ assert(p.get() == raw_ptr);
+ assert(ptr.get() == 0);
+ }
+ assert(A::count == 0);
+ assert(B::count == 0);
+
+ {
+ std::unique_ptr<A[]> ptr(new A[8]);
+ A* raw_ptr = ptr.get();
+ std::shared_ptr<A[]> p(std::move(ptr));
+ assert(A::count == 8);
+ assert(p.use_count() == 1);
+ assert(p.get() == raw_ptr);
+ assert(ptr.get() == 0);
+ }
+ assert(A::count == 0);
+
+ {
+ std::unique_ptr<int[]> ptr(new int[8]);
+ std::shared_ptr<int[]> p(std::move(ptr));
+ }
+#endif // TEST_STD_VER >= 14
+
+ return 0;
}
