Implement a heap profiler for the Dart managed heap.

runtime/vm/heap.cc

 // Copyright (c) 2012, 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 "vm/heap.h"
 
 #include "platform/assert.h"
 #include "platform/utils.h"
 #include "vm/compiler_stats.h"
 #include "vm/flags.h"
+#include "vm/heap_profiler.h"
 #include "vm/isolate.h"
 #include "vm/object.h"
 #include "vm/os.h"
 #include "vm/pages.h"
 #include "vm/scavenger.h"
+#include "vm/stack_frame.h"
 #include "vm/verifier.h"
 #include "vm/virtual_memory.h"
 
 namespace dart {
 
 DEFINE_FLAG(bool, verbose_gc, false, "Enables verbose GC.");
 DEFINE_FLAG(bool, verify_before_gc, false,
             "Enables heap verification before GC.");
 DEFINE_FLAG(bool, verify_after_gc, false,
             "Enables heap verification after GC.");
@@ skipping 180 matching lines @@
 void Heap::PrintSizes() const {
   OS::PrintErr("New space (%dk of %dk) "
                "Old space (%dk of %dk) "
                "Code space (%dk of %dk)\n",
                (new_space_->in_use() / KB), (new_space_->capacity() / KB),
                (old_space_->in_use() / KB), (old_space_->capacity() / KB),
                (code_space_->in_use() / KB), (code_space_->capacity() / KB));
 }
 
 
+void Heap::Profile(Dart_HeapProfileWriteCallback callback, void* stream) const {
+  HeapProfiler profiler(callback, stream);
+
+  // Dump the root set.
+  HeapProfilerRootVisitor root_visitor(&profiler);
+  Isolate* isolate = Isolate::Current();
+  Isolate* vm_isolate = Dart::vm_isolate();
+  isolate->VisitObjectPointers(&root_visitor, false,
+                               StackFrameIterator::kDontValidateFrames);
+  HeapProfilerWeakRootVisitor weak_root_visitor(&root_visitor);
+  isolate->VisitWeakPersistentHandles(&weak_root_visitor, true);
+
+  // Dump the current and VM isolate heaps.
+  HeapProfilerObjectVisitor object_visitor(&profiler);
+  isolate->heap()->new_space_->VisitObjects(&object_visitor);
+  isolate->heap()->old_space_->VisitObjects(&object_visitor);
+  isolate->heap()->code_space_->VisitObjects(&object_visitor);
+  vm_isolate->heap()->new_space_->VisitObjects(&object_visitor);
+  vm_isolate->heap()->old_space_->VisitObjects(&object_visitor);
+}
+
+
 #if defined(DEBUG)
 NoGCScope::NoGCScope() : StackResource(Isolate::Current()) {
   isolate()->IncrementNoGCScopeDepth();
 }
 
 
 NoGCScope::~NoGCScope() {
   isolate()->DecrementNoGCScopeDepth();
 }
 #endif  // defined(DEBUG)
 
 }  // namespace dart