Assert sequence validity on non-thread-safe RefCount manipulations (2)

base/memory/ref_counted.cc

 // Copyright (c) 2011 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "base/memory/ref_counted.h"
+
 #include "base/threading/thread_collision_warner.h"
 
 namespace base {
+namespace {
+
+#if DCHECK_IS_ON()
+AtomicRefCount g_cross_thread_ref_count_access_allow_count = 0;
+#endif
+
+}  // namespace
 
 namespace subtle {
 
 bool RefCountedThreadSafeBase::HasOneRef() const {
   return AtomicRefCountIsOne(
       &const_cast<RefCountedThreadSafeBase*>(this)->ref_count_);
 }
 
 RefCountedThreadSafeBase::RefCountedThreadSafeBase() : ref_count_(0) {
 #if DCHECK_IS_ON()
@@ skipping 22 matching lines @@
 #endif
   if (!AtomicRefCountDec(&ref_count_)) {
 #if DCHECK_IS_ON()
     in_dtor_ = true;
 #endif
     return true;
   }
   return false;
 }
 
+RefCountedBase::RefCountedBase() : ref_count_(0) {
+  DetachFromSequence();
+}
+
+RefCountedBase::~RefCountedBase() {
+#if DCHECK_IS_ON()
+  DCHECK(in_dtor_) << "RefCounted object deleted without calling Release()";
+#endif
+}

[gab, 2017/02/16 21:38:46]

Keep these inline (they're no-ops in non-dcheck builds)

[tzik, 2017/02/17 11:36:08]

Done.

+
+void RefCountedBase::DetachFromSequence() const {
+#if DCHECK_IS_ON()
+  sequence_checker_.DetachFromSequence();
+#endif
+}
+
+bool RefCountedBase::CalledOnValidSequence() const {
+#if DCHECK_IS_ON()
+  if (!AtomicRefCountIsZero(&g_cross_thread_ref_count_access_allow_count))
+    return true;
+  return sequence_checker_.CalledOnValidSequence();

[gab, 2017/02/16 21:38:46]

Merge if into this call:

return CalledOnValidSequenced() || !AtomicRefCountIsZero();

[tzik, 2017/02/17 11:36:08]

Done.

+#else
+  return true;
+#endif
+}
+
 }  // namespace subtle
 
+#if DCHECK_IS_ON()
+ScopedAllowCrossThreadRefCountAccess::ScopedAllowCrossThreadRefCountAccess() {
+  AtomicRefCountInc(&g_cross_thread_ref_count_access_allow_count);
+}
+
+ScopedAllowCrossThreadRefCountAccess::~ScopedAllowCrossThreadRefCountAccess() {
+  AtomicRefCountDec(&g_cross_thread_ref_count_access_allow_count);
+}
+#endif
+
 }  // namespace base