// -*- C++ -*- // Copyright (C) 2005 Free Software Foundation, Inc. // // This file is part of the GNU ISO C++ Library. This library is free // software; you can redistribute it and/or modify it under the // terms of the GNU General Public License as published by the // Free Software Foundation; either version 2, or (at your option) // any later version. // This library is distributed in the hope that it will be useful, // but WITHOUT ANY WARRANTY; without even the implied warranty of // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the // GNU General Public License for more details. // You should have received a copy of the GNU General Public License along // with this library; see the file COPYING. If not, write to the Free // Software Foundation, 59 Temple Place - Suite 330, Boston, MA 02111-1307, // USA. // As a special exception, you may use this file as part of a free software // library without restriction. Specifically, if other files instantiate // templates or use macros or inline functions from this file, or you compile // this file and link it with other files to produce an executable, this // file does not by itself cause the resulting executable to be covered by // the GNU General Public License. This exception does not however // invalidate any other reasons why the executable file might be covered by // the GNU General Public License. // boost/checked_delete.hpp // Copyright (c) 2002, 2003 Peter Dimov // Copyright (c) 2003 Daniel Frey // Copyright (c) 2003 Howard Hinnant // boost/shared_count.hpp // Copyright (c) 2001, 2002, 2003 Peter Dimov and Multi Media Ltd. // shared_ptr.hpp // (C) Copyright Greg Colvin and Beman Dawes 1998, 1999. // Copyright (c) 2001, 2002, 2003 Peter Dimov // weak_ptr.hpp // Copyright (c) 2001, 2002, 2003 Peter Dimov // enable_shared_from_this.hpp // Copyright (c) 2002 Peter Dimov #ifndef _TR1_MEMORY #define _TR1_MEMORY 1 #include "../memory" #include // std::less #include // std::exception #include // std::bad_alloc #include // std::type_info in get_deleter #include // std::size_t #include // for std::swap #include // for std::basic_ostream #include // for std::abort #include #include #include #include // namespace std::tr1 namespace std { namespace tr1 { class bad_weak_ptr : public std::exception { public: virtual char const* what() const throw() { return "tr1::bad_weak_ptr"; } }; // Helper for exception objects in // TODO this should be defined in a different file. void __throw_bad_weak_ptr() { #if __EXCEPTIONS throw bad_weak_ptr(); #else std::abort(); #endif } // verify that types are complete for increased safety template inline void checked_delete(_Tp * __x) { // intentionally complex - simplification causes regressions typedef char type_must_be_complete[ sizeof(_Tp)? 1: -1 ]; (void) sizeof(type_must_be_complete); delete __x; } template struct checked_deleter { typedef void result_type; typedef _Tp* argument_type; void operator()(_Tp * x) const { std::tr1::checked_delete(x); } }; class _Sp_counted_base { public: _Sp_counted_base() : _M_use_count(1), _M_weak_count(1) { } virtual ~_Sp_counted_base() // nothrow { } // dispose() is called when _M_use_count drops to zero, to release // the resources managed by *this. virtual void dispose() = 0; // nothrow // destroy() is called when _M_weak_count drops to zero. virtual void destroy() // nothrow { delete this; } virtual void* get_deleter(const std::type_info&) = 0; void add_ref_copy() { __gnu_cxx::__atomic_add(&_M_use_count, 1); } void add_ref_lock() { if (_M_use_count <= 0) // TODO not yet MT safe XXX { __throw_bad_weak_ptr(); } __gnu_cxx::__atomic_add(&_M_use_count, 1); } void release() // nothrow { if (__gnu_cxx::__exchange_and_add(&_M_use_count, -1) <= 1) { dispose(); weak_release(); } } void weak_add_ref() // nothrow { __gnu_cxx::__atomic_add(&_M_weak_count, 1); } void weak_release() // nothrow { if (__gnu_cxx::__exchange_and_add(&_M_weak_count, -1) <= 1) { destroy(); } } long use_count() const // nothrow { return _M_use_count; // XXX is this MT safe? } private: _Sp_counted_base(_Sp_counted_base const&); _Sp_counted_base& operator= (_Sp_counted_base const&); _Atomic_word _M_use_count; // #shared _Atomic_word _M_weak_count; // #weak + (#shared != 0) }; template class _Sp_counted_base_impl : public _Sp_counted_base { public: /** * @brief * @pre d(p) must not throw. */ _Sp_counted_base_impl(_Ptr __p, _Deleter __d) : _M_ptr(__p), _M_del(__d) { } virtual void dispose() // nothrow { _M_del(_M_ptr); } virtual void* get_deleter(const std::type_info& __ti) { return __ti == typeid(_Deleter) ? &_M_del : 0; } private: _Sp_counted_base_impl(const _Sp_counted_base_impl&); _Sp_counted_base_impl& operator=(const _Sp_counted_base_impl&); _Ptr _M_ptr; // copy constructor must not throw _Deleter _M_del; // copy constructor must not throw }; class weak_count; class shared_count { private: _Sp_counted_base* _M_pi; friend class weak_count; public: shared_count() : _M_pi(0) // nothrow { } template shared_count(_Ptr __p, _Deleter __d) : _M_pi(0) { try { _M_pi = new _Sp_counted_base_impl<_Ptr, _Deleter>(__p, __d); } catch(...) { __d(__p); // delete __p __throw_exception_again; } } // auto_ptr<_Tp> is special cased to provide the strong guarantee template explicit shared_count(std::auto_ptr<_Tp>& __r) : _M_pi(new _Sp_counted_base_impl<_Tp*,checked_deleter<_Tp> >( __r.get(), checked_deleter<_Tp>() )) { __r.release(); } // throws bad_weak_ptr when __r.use_count() == 0 explicit shared_count(const weak_count& __r); ~shared_count() // nothrow { if (_M_pi != 0) _M_pi->release(); } shared_count(const shared_count& __r) : _M_pi(__r._M_pi) // nothrow { if (_M_pi != 0) _M_pi->add_ref_copy(); } shared_count& operator=(const shared_count& __r) // nothrow { _Sp_counted_base* __tmp = __r._M_pi; if(__tmp != _M_pi) { if(__tmp != 0) __tmp->add_ref_copy(); if(_M_pi != 0) _M_pi->release(); _M_pi = __tmp; } return *this; } void swap(shared_count& __r) // nothrow { _Sp_counted_base* __tmp = __r._M_pi; __r._M_pi = _M_pi; _M_pi = __tmp; } long use_count() const // nothrow { return _M_pi != 0 ? _M_pi->use_count() : 0; } bool unique() const // nothrow { return this->use_count() == 1; } friend inline bool operator==(const shared_count& __a, const shared_count& __b) { return __a._M_pi == __b._M_pi; } friend inline bool operator<(const shared_count& __a, const shared_count& __b) { return std::less<_Sp_counted_base*>()(__a._M_pi, __b._M_pi); } void* get_deleter(const std::type_info& __ti) const { return _M_pi ? _M_pi->get_deleter(__ti) : 0; } }; class weak_count { private: _Sp_counted_base * _M_pi; friend class shared_count; public: weak_count() : _M_pi(0) // nothrow { } weak_count(const shared_count& __r) : _M_pi(__r._M_pi) // nothrow { if (_M_pi != 0) _M_pi->weak_add_ref(); } weak_count(const weak_count& __r) : _M_pi(__r._M_pi) // nothrow { if (_M_pi != 0) _M_pi->weak_add_ref(); } ~weak_count() // nothrow { if (_M_pi != 0) _M_pi->weak_release(); } weak_count& operator=(const shared_count& __r) // nothrow { _Sp_counted_base* __tmp = __r._M_pi; if (__tmp != 0) __tmp->weak_add_ref(); if (_M_pi != 0) _M_pi->weak_release(); _M_pi = __tmp; return *this; } weak_count& operator=(const weak_count& __r) // nothrow { _Sp_counted_base * __tmp = __r._M_pi; if (__tmp != 0) __tmp->weak_add_ref(); if (_M_pi != 0) _M_pi->weak_release(); _M_pi = __tmp; return *this; } void swap(weak_count& __r) // nothrow { _Sp_counted_base * __tmp = __r._M_pi; __r._M_pi = _M_pi; _M_pi = __tmp; } long use_count() const // nothrow { return _M_pi != 0 ? _M_pi->use_count() : 0; } friend inline bool operator==(const weak_count& __a, const weak_count& __b) { return __a._M_pi == __b._M_pi; } friend inline bool operator<(const weak_count& __a, const weak_count& __b) { return std::less<_Sp_counted_base*>()(__a._M_pi, __b._M_pi); } }; inline shared_count::shared_count(const weak_count& __r) : _M_pi(__r._M_pi) { if (_M_pi != 0) { _M_pi->add_ref_lock(); } else { __throw_bad_weak_ptr(); } } // fwd decls template class weak_ptr; template class enable_shared_from_this; struct __static_cast_tag {}; struct __const_cast_tag {}; struct __dynamic_cast_tag {}; struct __polymorphic_cast_tag {}; template struct shared_ptr_traits { typedef _Tp & reference; }; template<> struct shared_ptr_traits { typedef void reference; }; template<> struct shared_ptr_traits { typedef void reference; }; template<> struct shared_ptr_traits { typedef void reference; }; template<> struct shared_ptr_traits { typedef void reference; }; // enable_shared_from_this support // friend of enable_shared_from_this template void __enable_shared_from_this( const shared_count& __pn, const enable_shared_from_this<_Tp1>* __pe, const _Tp2* __px ); inline void __enable_shared_from_this(const shared_count&, ...) { } /** * @class shared_ptr * * A smart pointer with reference-counted copy semantics. * The object pointed to is deleted when the last shared_ptr pointing to it * is destroyed or reset. */ template class shared_ptr { typedef typename shared_ptr_traits<_Tp>::reference _Reference; public: typedef _Tp element_type; /** @brief Construct an empty %shared_ptr. * @post use_count()==0 && get()==0 */ shared_ptr() : _M_ptr(0), _M_refcount() // never throws { } /** @brief Construct a %shared_ptr that owns the pointer @a p. * @param p A pointer that is convertible to element_type*. * @post use_count()==1 && get()==p * @throw std::bad_alloc, in which case @c delete @a p is called. */ template explicit shared_ptr(_Tp1* __p) : _M_ptr(__p), _M_refcount(__p, checked_deleter<_Tp1>()) { __glibcxx_function_requires(_ConvertibleConcept<_Tp1*, _Tp*>) // __glibcxx_function_requires(_CompleteConcept<_Tp1*>) __enable_shared_from_this( _M_refcount, __p, __p ); } // // Requirements: D's copy constructor and destructor must not throw // // shared_ptr will release p by calling d(p) // /** @brief Construct a %shared_ptr that owns the pointer @a p * and the deleter @a d. * @param p A pointer. * @param d A deleter. * @post use_count()==1 && get()==p * @throw std::bad_alloc, in which case @a d(p) is called. */ template shared_ptr(_Tp1* __p, _Deleter __d) : _M_ptr(__p), _M_refcount(__p, __d) { __glibcxx_function_requires(_ConvertibleConcept<_Tp1*, _Tp*>) // TODO requires D is CopyConstructible and d(p) well-formed __enable_shared_from_this( _M_refcount, __p, __p ); } // generated copy constructor, assignment, destructor are fine. /** @brief If @a r is empty, constructs an empty %shared_ptr; otherwise * construct a %shared_ptr that shares ownership with @a r. * @param r A %shared_ptr. * @post get()==r.get() && use_count()==r.use_count() * @throw std::bad_alloc, in which case */ template shared_ptr(const shared_ptr<_Tp1>& __r) : _M_ptr(__r._M_ptr), _M_refcount(__r._M_refcount) // never throws { __glibcxx_function_requires(_ConvertibleConcept<_Tp1*, _Tp*>) } /** @brief Constructs a %shared_ptr that shares ownership with @a r * and stores a copy of the pointer stored in @a r. * @param r A weak_ptr. * @post use_count()==r.use_count() * @throw bad_weak_ptr when r.expired(), * in which case the constructor has no effect. */ template explicit shared_ptr(const weak_ptr<_Tp1>& __r) : _M_refcount(__r._M_refcount) // may throw { __glibcxx_function_requires(_ConvertibleConcept<_Tp1*, _Tp*>) // it is now safe to copy r__._M_ptr, as _M_refcount(__r._M_refcount) // did not throw _M_ptr = __r._M_ptr; } /** * @post use_count()==1 and r.get()==0 */ template explicit shared_ptr(std::auto_ptr<_Tp1>& __r) : _M_ptr(__r.get()), _M_refcount() { // TODO requires r.release() convertible to _Tp*, Tp1 is complete, // delete r.release() well-formed _Tp1 * __tmp = __r.get(); _M_refcount = shared_count(__r); __enable_shared_from_this( _M_refcount, __tmp, __tmp ); } template shared_ptr(const shared_ptr<_Tp1>& __r, __static_cast_tag) : _M_ptr(static_cast(__r._M_ptr)) , _M_refcount(__r._M_refcount) { } template shared_ptr(const shared_ptr<_Tp1>& __r, __const_cast_tag) : _M_ptr(const_cast(__r._M_ptr)) , _M_refcount(__r._M_refcount) { } template shared_ptr(const shared_ptr<_Tp1>& __r, __dynamic_cast_tag) : _M_ptr(dynamic_cast(__r._M_ptr)) , _M_refcount(__r._M_refcount) { if (_M_ptr == 0) // need to allocate new counter -- the cast failed { _M_refcount = shared_count(); } } template shared_ptr& operator=(const shared_ptr<_Tp1>& __r) // never throws { _M_ptr = __r._M_ptr; _M_refcount = __r._M_refcount; // shared_count::op= doesn't throw return *this; } template shared_ptr& operator=(std::auto_ptr<_Tp1>& __r) { shared_ptr(__r).swap(*this); return *this; } void reset() // never throws { shared_ptr().swap(*this); } template void reset(_Tp1* __p) // _Tp1 must be complete { _GLIBCXX_DEBUG_ASSERT(__p == 0 || __p != _M_ptr); // catch self-reset errors shared_ptr(__p).swap(*this); } template void reset(_Tp1 * __p, _Deleter __d) { shared_ptr(__p, __d).swap(*this); } // error to instantiate if _Tp is [cv-qual] void _Reference operator*() const // never throws { _GLIBCXX_DEBUG_ASSERT(_M_ptr != 0); return *_M_ptr; } _Tp* operator->() const // never throws { _GLIBCXX_DEBUG_ASSERT(_M_ptr != 0); return _M_ptr; } _Tp* get() const // never throws { return _M_ptr; } // implicit conversion to "bool" private: typedef _Tp* shared_ptr::*__unspecified_bool_type; public: operator __unspecified_bool_type() const // never throws { return _M_ptr == 0 ? 0 : &shared_ptr::_M_ptr; } bool unique() const // never throws { return _M_refcount.unique(); } long use_count() const // never throws { return _M_refcount.use_count(); } void swap(shared_ptr<_Tp>& __other) // never throws { std::swap(_M_ptr, __other._M_ptr); _M_refcount.swap(__other._M_refcount); } private: template bool _M_less(const shared_ptr<_Tp1>& __rhs) const { return _M_refcount < __rhs._M_refcount; } void* _M_get_deleter(const std::type_info& __ti) const { return _M_refcount.get_deleter(__ti); } template friend class shared_ptr; template friend class weak_ptr; // friends injected into enclosing namespace and found by ADL: // get_deleter (experimental) template friend inline _Del* get_deleter(const shared_ptr& __p) { return static_cast<_Del*>(__p._M_get_deleter(typeid(_Del))); } template friend inline bool operator==(const shared_ptr& __a, const shared_ptr<_Tp1>& __b) { return __a.get() == __b.get(); } template friend inline bool operator!=(const shared_ptr& __a, const shared_ptr<_Tp1>& __b) { return __a.get() != __b.get(); } template friend inline bool operator<(const shared_ptr& __a, const shared_ptr<_Tp1>& __b) { return __a._M_less(__b); } _Tp* _M_ptr; // contained pointer shared_count _M_refcount; // reference counter }; // shared_ptr // 2.2.3.9 shared_ptr casts /** @warning The seemingly equivalent * shared_ptr(static_cast(r.get())) * will eventually result in undefined behaviour, * attempting to delete the same object twice. */ template shared_ptr<_Tp> static_pointer_cast(const shared_ptr<_Tp1>& __r) { return shared_ptr<_Tp>(__r, __static_cast_tag()); } /** @warning The seemingly equivalent * shared_ptr(const_cast(r.get())) * will eventually result in undefined behaviour, * attempting to delete the same object twice. */ template shared_ptr<_Tp> const_pointer_cast(const shared_ptr<_Tp1>& __r) { return shared_ptr<_Tp>(__r, __const_cast_tag()); } /** @warning The seemingly equivalent * shared_ptr(dynamic_cast(r.get())) * will eventually result in undefined behaviour, * attempting to delete the same object twice. */ template shared_ptr<_Tp> dynamic_pointer_cast(const shared_ptr<_Tp1>& __r) { return shared_ptr<_Tp>(__r, __dynamic_cast_tag()); } // operator<< template std::basic_ostream<_Ch,_Tr>& operator<<(std::basic_ostream<_Ch,_Tr>& __os, const shared_ptr<_Tp>& __p) { __os << __p.get(); return __os; } template class weak_ptr { public: typedef _Tp element_type; weak_ptr() : _M_ptr(0), _M_refcount() // never throws { } // generated copy constructor, assignment, destructor are fine // // The "obvious" converting constructor implementation: // // template // weak_ptr(weak_ptr const & r): _M_ptr(r._M_ptr), _M_refcount(r._M_refcount) // never throws // { // } // // has a serious problem. // // r._M_ptr may already have been invalidated. The _M_ptr(r._M_ptr) // conversion may require access to *r._M_ptr (virtual inheritance). // // It is not possible to avoid spurious access violations since // in multithreaded programs r._M_ptr may be invalidated at any point. // template weak_ptr(const weak_ptr<_Tp1>& r) : _M_refcount(r._M_refcount) // never throws { __glibcxx_function_requires(_ConvertibleConcept<_Tp1*, _Tp*>) _M_ptr = r.lock().get(); } template weak_ptr(const shared_ptr<_Tp1>& r) : _M_ptr(r._M_ptr), _M_refcount(r._M_refcount) // never throws { __glibcxx_function_requires(_ConvertibleConcept<_Tp1*, _Tp*>) } template weak_ptr& operator=(const weak_ptr<_Tp1>& r) // never throws { _M_ptr = r.lock().get(); _M_refcount = r._M_refcount; return *this; } template weak_ptr& operator=(const shared_ptr<_Tp1>& r) // never throws { _M_ptr = r._M_ptr; _M_refcount = r._M_refcount; return *this; } shared_ptr<_Tp> lock() const // never throws { #ifdef __GTHREADS // optimization: avoid throw overhead if (expired()) { return shared_ptr(); } try { return shared_ptr(*this); } catch (const bad_weak_ptr&) { // Q: how can we get here? // A: another thread may have invalidated r after the use_count test above. return shared_ptr(); } #else // optimization: avoid try/catch overhead when single threaded return expired() ? shared_ptr() : shared_ptr(*this); #endif } // XXX MT long use_count() const // never throws { return _M_refcount.use_count(); } bool expired() const // never throws { return _M_refcount.use_count() == 0; } void reset() // never throws { weak_ptr().swap(*this); } void swap(weak_ptr& __s) // never throws { std::swap(_M_ptr, __s._M_ptr); _M_refcount.swap(__s._M_refcount); } private: template bool _M_less(const weak_ptr<_Tp1>& __rhs) const { return _M_refcount < __rhs._M_refcount; } // used by __enable_shared_from_this void _M_assign(_Tp* __ptr, const shared_count& __refcount) { _M_ptr = __ptr; _M_refcount = __refcount; } // friend injected into namespace and found by ADL template friend inline bool operator<(const weak_ptr& __lhs, const weak_ptr<_Tp1>& __rhs) { return __lhs._M_less(__rhs); } template friend class weak_ptr; template friend class shared_ptr; friend class enable_shared_from_this<_Tp>; _Tp* _M_ptr; // contained pointer weak_count _M_refcount; // reference counter }; // weak_ptr template class enable_shared_from_this { protected: enable_shared_from_this() { } enable_shared_from_this(const enable_shared_from_this&) { } enable_shared_from_this& operator=(const enable_shared_from_this&) { return *this; } ~enable_shared_from_this() { } public: shared_ptr<_Tp> shared_from_this() { shared_ptr<_Tp> p(this->_M_weak_this); return p; } shared_ptr shared_from_this() const { shared_ptr p(this->_M_weak_this); return p; } private: template void _M_weak_assign(_Tp1* __p, const shared_count& __n) const { _M_weak_this._M_assign(__p, __n); } template friend void __enable_shared_from_this( const shared_count& __pn, const enable_shared_from_this* __pe, const _Tp1* __px) { if(__pe != 0) __pe->_M_weak_assign(const_cast<_Tp1*>(__px), __pn); } mutable weak_ptr<_Tp> _M_weak_this; }; } // namespace tr1 /** * @brief std::swap() specialisation for shared_ptr. * @relates shared_ptr. */ template inline void swap(tr1::shared_ptr<_Tp>& __a, tr1::shared_ptr<_Tp>& __b) { __a.swap(__b); } /** * @brief std::swap() specialisation for weak_ptr. * @relates weak_ptr. */ template void swap(tr1::weak_ptr<_Tp>& __a, tr1::weak_ptr<_Tp>& __b) { __a.swap(__b); } } // namespace std #endif