fix(asan): pointer used after free in autoref_ptr_test (#361)

include/dsn/utility/autoref_ptr.h

@@ -37,6 +37,8 @@
 
 #include <atomic>
 #include <cassert>
+#include <type_traits>
+#include <utility>
 
 namespace dsn {
 class ref_counter
@@ -106,21 +108,15 @@ class ref_ptr
             _obj->add_ref();
     }
 
-    template <typename U>
-    ref_ptr(U *obj) : _obj(obj)
-    {
-        if (nullptr != _obj)
-            _obj->add_ref();
-    }
-
     ref_ptr(const ref_ptr<T> &r)
     {
         _obj = r.get();
         if (nullptr != _obj)
             _obj->add_ref();
     }
 
-    template <typename U>
+    template <typename U,
+              typename = typename std::enable_if<std::is_convertible<U *, T *>::value>::type>
     ref_ptr(const ref_ptr<U> &r)
     {
         _obj = r.get();
@@ -130,8 +126,9 @@ class ref_ptr
 
     ref_ptr(ref_ptr<T> &&r) : _obj(r._obj) { r._obj = nullptr; }
 
-    template <typename U>
-    ref_ptr(ref_ptr<U> &&r) : _obj(r._obj)
+    template <typename U,
+              typename = typename std::enable_if<std::is_convertible<U *, T *>::value>::type>
+    ref_ptr(ref_ptr<U> &&r) noexcept : _obj(r._obj)
     {
         r._obj = nullptr;
     }
@@ -143,72 +140,24 @@ class ref_ptr
         }
     }
 
-    ref_ptr<T> &operator=(T *obj)
-    {
-        if (_obj == obj)
-            return *this;
-
-        if (nullptr != _obj) {
-            _obj->release_ref();
-        }
-
-        _obj = obj;
-
-        if (obj != nullptr) {
-            _obj->add_ref();
-        }
+    ref_ptr<T> &operator=(T *obj) { return *this = ref_ptr(obj); }
 
-        return *this;
-    }
-
-    template <typename U>
-    ref_ptr<T> &operator=(U *obj)
+    ref_ptr<T> &operator=(ref_ptr<T> r) noexcept
     {
-        if (_obj == obj)
-            return *this;
-        if (nullptr != _obj) {
-            _obj->release_ref();
-        }
-        _obj = obj;
-        if (_obj != nullptr) {
-            _obj->add_ref();
-        }
+        swap(r);
         return *this;
     }
 
-    ref_ptr<T> &operator=(const ref_ptr<T> &obj) { return operator=(obj._obj); }
-
-    template <typename U>
-    ref_ptr<T> &operator=(const ref_ptr<U> &obj)
+    template <typename U,
+              typename = typename std::enable_if<std::is_convertible<U *, T *>::value>::type>
+    ref_ptr<T> &operator=(ref_ptr<U> r) noexcept
     {
-        return operator=(obj._obj);
-    }
-
-    ref_ptr<T> &operator=(ref_ptr<T> &&obj)
-    {
-        if (this == &obj) {
-            return *this;
-        }
-
-        if (nullptr != _obj) {
-            _obj->release_ref();
-        }
-
-        _obj = obj._obj;
-        obj._obj = nullptr;
+        ref_ptr<T> p(r);
+        swap(p);
         return *this;
     }
 
-    template <typename U>
-    ref_ptr<T> &operator=(ref_ptr<U> &&obj)
-    {
-        if (nullptr != _obj) {
-            _obj->release_ref();
-        }
-        _obj = obj._obj;
-        obj._obj = nullptr;
-        return *this;
-    }
+    void swap(ref_ptr<T> &r) noexcept { std::swap(_obj, r._obj); }
 
     T *get() const { return _obj; }