Removed MallocHookScope and replaced it with MallocLocker, which behaves like MutexLocker but also …

runtime/vm/malloc_hooks.cc

 // Copyright (c) 2017, the Dart project authors.  Please see the AUTHORS file
 // for details. All rights reserved. Use of this source code is governed by a
 // BSD-style license that can be found in the LICENSE file.
 
 #include "platform/globals.h"
 
 #if defined(DART_USE_TCMALLOC) && !defined(PRODUCT)
 
 #include "vm/malloc_hooks.h"
 
 #include "gperftools/malloc_hook.h"
 
 #include "platform/assert.h"
 #include "vm/hash_map.h"
 #include "vm/json_stream.h"
-#include "vm/lockers.h"
+#include "vm/os_thread.h"
 
 namespace dart {
 
-// A locker-type class to automatically grab and release the
-// in_malloc_hook_flag_.
-class MallocHookScope {
+// A locker-type class similar to MutexLocker which tracks which thread
+// currently holds the lock. We use this instead of MutexLocker and
+// mutex->IsOwnedByCurrentThread() since IsOwnedByCurrentThread() is only
+// enabled for debug mode.
+class MallocLocker : public ValueObject {
  public:
-  static void InitMallocHookFlag() {
-    MutexLocker ml(malloc_hook_scope_mutex_);
-    ASSERT(in_malloc_hook_flag_ == kUnsetThreadLocalKey);
-    in_malloc_hook_flag_ = OSThread::CreateThreadLocal();
-    OSThread::SetThreadLocal(in_malloc_hook_flag_, 0);
+  explicit MallocLocker(Mutex* mutex, ThreadId* owner)
+      : mutex_(mutex), owner_(owner) {
+    ASSERT(owner != NULL);
+    mutex_->Lock();
+    ASSERT(*owner_ == OSThread::kInvalidThreadId);
+    *owner_ = OSThread::GetCurrentThreadId();
   }
 
-  static void DestroyMallocHookFlag() {
-    MutexLocker ml(malloc_hook_scope_mutex_);
-    ASSERT(in_malloc_hook_flag_ != kUnsetThreadLocalKey);
-    OSThread::DeleteThreadLocal(in_malloc_hook_flag_);
-    in_malloc_hook_flag_ = kUnsetThreadLocalKey;
-  }
-
-  MallocHookScope() {
-    MutexLocker ml(malloc_hook_scope_mutex_);
-    ASSERT(in_malloc_hook_flag_ != kUnsetThreadLocalKey);
-    OSThread::SetThreadLocal(in_malloc_hook_flag_, 1);
-  }
-
-  ~MallocHookScope() {
-    MutexLocker ml(malloc_hook_scope_mutex_);
-    ASSERT(in_malloc_hook_flag_ != kUnsetThreadLocalKey);
-    OSThread::SetThreadLocal(in_malloc_hook_flag_, 0);
-  }
-
-  static bool IsInHook() {
-    MutexLocker ml(malloc_hook_scope_mutex_);
-    if (in_malloc_hook_flag_ == kUnsetThreadLocalKey) {
-      // Bail out if the malloc hook flag is invalid. This means that
-      // MallocHookState::TearDown() has been called and MallocHookScope is no
-      // longer intitialized. Don't worry if MallocHookState::TearDown() is
-      // called before the hooks grab the mutex, since
-      // MallocHooksState::Active() is checked after the lock is taken before
-      // proceeding to act on the allocation/free.
-      return false;
-    }
-    return OSThread::GetThreadLocal(in_malloc_hook_flag_);
+  virtual ~MallocLocker() {
+    ASSERT(*owner_ == OSThread::GetCurrentThreadId());
+    *owner_ = OSThread::kInvalidThreadId;
+    mutex_->Unlock();
   }
 
  private:
-  static Mutex* malloc_hook_scope_mutex_;
-  static ThreadLocalKey in_malloc_hook_flag_;
-
-  DISALLOW_ALLOCATION();
-  DISALLOW_COPY_AND_ASSIGN(MallocHookScope);
+  Mutex* mutex_;
+  ThreadId* owner_;
 };
 
 
 // Custom key/value trait specifically for address/size pairs. Unlike
 // RawPointerKeyValueTrait, the default value is -1 as 0 can be a valid entry.
 class AddressKeyValueTrait {
  public:
   typedef const void* Key;
   typedef intptr_t Value;
 
@@ skipping 48 matching lines @@
     active_ = true;
     original_pid_ = OS::ProcessId();
   }
 
   static bool IsOriginalProcess() {
     ASSERT(original_pid_ != kInvalidPid);
     return original_pid_ == OS::ProcessId();
   }
 
   static Mutex* malloc_hook_mutex() { return malloc_hook_mutex_; }
+  static ThreadId* malloc_hook_mutex_owner() {
+    return &malloc_hook_mutex_owner_;
+  }
+  static bool IsLockHeldByCurrentThread() {
+    return (malloc_hook_mutex_owner_ == OSThread::GetCurrentThreadId());
+  }
 
   static intptr_t allocation_count() { return allocation_count_; }
 
   static intptr_t heap_allocated_memory_in_bytes() {
     return heap_allocated_memory_in_bytes_;
   }
 
   static void IncrementHeapAllocatedMemoryInBytes(intptr_t size) {
     ASSERT(malloc_hook_mutex()->IsOwnedByCurrentThread());
     ASSERT(size >= 0);
@@ skipping 24 matching lines @@
     active_ = false;
     original_pid_ = kInvalidPid;
     ResetStats();
     delete address_map_;
   }
 
  private:
   static bool active_;
   static intptr_t original_pid_;
   static Mutex* malloc_hook_mutex_;
+  static ThreadId malloc_hook_mutex_owner_;
   static intptr_t allocation_count_;
   static intptr_t heap_allocated_memory_in_bytes_;
   static AddressMap* address_map_;
 
   static const intptr_t kInvalidPid = -1;
 };
 
 
-// MallocHookScope state.
-Mutex* MallocHookScope::malloc_hook_scope_mutex_ = new Mutex();
-ThreadLocalKey MallocHookScope::in_malloc_hook_flag_ = kUnsetThreadLocalKey;
-
 // MallocHooks state / locks.
 bool MallocHooksState::active_ = false;
 intptr_t MallocHooksState::original_pid_ = MallocHooksState::kInvalidPid;
 Mutex* MallocHooksState::malloc_hook_mutex_ = new Mutex();
+ThreadId MallocHooksState::malloc_hook_mutex_owner_ =
+    OSThread::kInvalidThreadId;
 
 // Memory allocation state information.
 intptr_t MallocHooksState::allocation_count_ = 0;
 intptr_t MallocHooksState::heap_allocated_memory_in_bytes_ = 0;
 AddressMap* MallocHooksState::address_map_ = NULL;
 
 
 void MallocHooks::InitOnce() {
   if (!FLAG_enable_malloc_hooks) {
     return;
   }
-  MutexLocker ml(MallocHooksState::malloc_hook_mutex());
+  MallocLocker ml(MallocHooksState::malloc_hook_mutex(),
+                  MallocHooksState::malloc_hook_mutex_owner());
   ASSERT(!MallocHooksState::Active());
 
-  MallocHookScope::InitMallocHookFlag();
   MallocHooksState::Init();
 
   // Register malloc hooks.
   bool success = false;
   success = MallocHook::AddNewHook(&MallocHooksState::RecordAllocHook);
   ASSERT(success);
   success = MallocHook::AddDeleteHook(&MallocHooksState::RecordFreeHook);
   ASSERT(success);
 }
 
 
 void MallocHooks::TearDown() {
   if (!FLAG_enable_malloc_hooks) {
     return;
   }
-  MutexLocker ml(MallocHooksState::malloc_hook_mutex());
+  MallocLocker ml(MallocHooksState::malloc_hook_mutex(),
+                  MallocHooksState::malloc_hook_mutex_owner());
   ASSERT(MallocHooksState::Active());
 
   // Remove malloc hooks.
   bool success = false;
   success = MallocHook::RemoveNewHook(&MallocHooksState::RecordAllocHook);
   ASSERT(success);
   success = MallocHook::RemoveDeleteHook(&MallocHooksState::RecordFreeHook);
   ASSERT(success);
 
   MallocHooksState::TearDown();
-  MallocHookScope::DestroyMallocHookFlag();
 }
 
 
 void MallocHooks::ResetStats() {
   if (!FLAG_enable_malloc_hooks) {
     return;
   }
-  MutexLocker ml(MallocHooksState::malloc_hook_mutex());
+  MallocLocker ml(MallocHooksState::malloc_hook_mutex(),
+                  MallocHooksState::malloc_hook_mutex_owner());
   if (MallocHooksState::Active()) {
     MallocHooksState::ResetStats();
   }
 }
 
 
 bool MallocHooks::Active() {
   if (!FLAG_enable_malloc_hooks) {
     return false;
   }
   ASSERT(MallocHooksState::malloc_hook_mutex()->IsOwnedByCurrentThread());
   return MallocHooksState::Active();
 }
 
 
 void MallocHooks::PrintToJSONObject(JSONObject* jsobj) {
   if (!FLAG_enable_malloc_hooks) {
     return;
   }
   intptr_t allocated_memory = 0;
   intptr_t allocation_count = 0;
   bool add_usage = false;
   // AddProperty may call malloc which would result in an attempt
   // to acquire the lock recursively so we extract the values first
   // and then add the JSON properties.
   {
-    MutexLocker ml(MallocHooksState::malloc_hook_mutex());
+    MallocLocker ml(MallocHooksState::malloc_hook_mutex(),
+                    MallocHooksState::malloc_hook_mutex_owner());
     if (Active()) {
       allocated_memory = MallocHooksState::heap_allocated_memory_in_bytes();
       allocation_count = MallocHooksState::allocation_count();
       add_usage = true;
     }
   }
   if (add_usage) {
     jsobj->AddProperty("_heapAllocatedMemoryUsage", allocated_memory);
     jsobj->AddProperty("_heapAllocationCount", allocation_count);
   }
 }
 
 
 intptr_t MallocHooks::allocation_count() {
   if (!FLAG_enable_malloc_hooks) {
     return 0;
   }
-  MutexLocker ml(MallocHooksState::malloc_hook_mutex());
+  MallocLocker ml(MallocHooksState::malloc_hook_mutex(),
+                  MallocHooksState::malloc_hook_mutex_owner());
   return MallocHooksState::allocation_count();
 }
 
 
 intptr_t MallocHooks::heap_allocated_memory_in_bytes() {
   if (!FLAG_enable_malloc_hooks) {
     return 0;
   }
-  MutexLocker ml(MallocHooksState::malloc_hook_mutex());
+  MallocLocker ml(MallocHooksState::malloc_hook_mutex(),
+                  MallocHooksState::malloc_hook_mutex_owner());
   return MallocHooksState::heap_allocated_memory_in_bytes();
 }
 
 
 void MallocHooksState::RecordAllocHook(const void* ptr, size_t size) {
-  if (MallocHookScope::IsInHook() || !MallocHooksState::IsOriginalProcess()) {
+  if (MallocHooksState::IsLockHeldByCurrentThread() ||
+      !MallocHooksState::IsOriginalProcess()) {
     return;
   }
 
-  MutexLocker ml(MallocHooksState::malloc_hook_mutex());
+  MallocLocker ml(MallocHooksState::malloc_hook_mutex(),
+                  MallocHooksState::malloc_hook_mutex_owner());
   // Now that we hold the lock, check to make sure everything is still active.
   if ((ptr != NULL) && MallocHooksState::Active()) {
-    // Set the malloc hook flag to avoid calling hooks again if memory is
-    // allocated/freed below.
-    MallocHookScope mhs;
     MallocHooksState::IncrementHeapAllocatedMemoryInBytes(size);
     MallocHooksState::address_map()->Insert(ptr, size);
   }
 }
 
 
 void MallocHooksState::RecordFreeHook(const void* ptr) {
-  if (MallocHookScope::IsInHook() || !MallocHooksState::IsOriginalProcess()) {
+  if (MallocHooksState::IsLockHeldByCurrentThread() ||
+      !MallocHooksState::IsOriginalProcess()) {
     return;
   }
 
-  MutexLocker ml(MallocHooksState::malloc_hook_mutex());
+  MallocLocker ml(MallocHooksState::malloc_hook_mutex(),
+                  MallocHooksState::malloc_hook_mutex_owner());
   // Now that we hold the lock, check to make sure everything is still active.
   if ((ptr != NULL) && MallocHooksState::Active()) {
-    // Set the malloc hook flag to avoid calling hooks again if memory is
-    // allocated/freed below.
-    MallocHookScope mhs;
     intptr_t size = 0;
     if (MallocHooksState::address_map()->Lookup(ptr, &size)) {
       MallocHooksState::DecrementHeapAllocatedMemoryInBytes(size);
       MallocHooksState::address_map()->Remove(ptr);
     }
   }
 }
 
 }  // namespace dart
 
 #endif  // defined(DART_USE_TCMALLOC) && !defined(PRODUCT)