[vm, gc] Require a safepoint for heap iteration.

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/flags.h"
 #include "vm/isolate.h"
 #include "vm/lockers.h"
 #include "vm/object.h"
 #include "vm/object_set.h"
 #include "vm/os.h"
 #include "vm/pages.h"
 #include "vm/raw_object.h"
+#include "vm/safepoint.h"
 #include "vm/scavenger.h"
 #include "vm/service.h"
 #include "vm/service_event.h"
 #include "vm/service_isolate.h"
 #include "vm/stack_frame.h"
 #include "vm/tags.h"
 #include "vm/timeline.h"
 #include "vm/verifier.h"
 #include "vm/virtual_memory.h"
 #include "vm/weak_table.h"
@@ skipping 156 matching lines @@
 
 void Heap::VisitObjectsNoImagePages(ObjectVisitor* visitor) const {
   new_space_.VisitObjects(visitor);
   old_space_.VisitObjectsNoImagePages(visitor);
 }
 
 void Heap::VisitObjectsImagePages(ObjectVisitor* visitor) const {
   old_space_.VisitObjectsImagePages(visitor);
 }
 
-HeapIterationScope::HeapIterationScope(bool writable)
-    : StackResource(Thread::Current()),
-      old_space_(isolate()->heap()->old_space()),
+HeapIterationScope::HeapIterationScope(Thread* thread, bool writable)
+    : StackResource(thread),
+      heap_(isolate()->heap()),
+      old_space_(heap_->old_space()),
       writable_(writable) {
   {
     // It's not yet safe to iterate over a paged space while it's concurrently
     // sweeping, so wait for any such task to complete first.
     MonitorLocker ml(old_space_->tasks_lock());
 #if defined(DEBUG)
     // We currently don't support nesting of HeapIterationScopes.
-    ASSERT(old_space_->iterating_thread_ != thread());
+    ASSERT(old_space_->iterating_thread_ != thread);
 #endif
     while (old_space_->tasks() > 0) {
-      ml.WaitWithSafepointCheck(thread());
+      ml.WaitWithSafepointCheck(thread);
     }
 #if defined(DEBUG)
     ASSERT(old_space_->iterating_thread_ == NULL);
-    old_space_->iterating_thread_ = thread();
+    old_space_->iterating_thread_ = thread;
 #endif
     old_space_->set_tasks(1);
   }
 
+  isolate()->safepoint_handler()->SafepointThreads(thread);
+
   if (writable_) {
-    thread()->heap()->WriteProtectCode(false);
+    heap_->WriteProtectCode(false);
   }
 }
 
 HeapIterationScope::~HeapIterationScope() {
   if (writable_) {
-    thread()->heap()->WriteProtectCode(true);
+    heap_->WriteProtectCode(true);
   }
 
+  isolate()->safepoint_handler()->ResumeThreads(thread());
+
   MonitorLocker ml(old_space_->tasks_lock());
 #if defined(DEBUG)
   ASSERT(old_space_->iterating_thread_ == thread());
   old_space_->iterating_thread_ = NULL;
 #endif
   ASSERT(old_space_->tasks() == 1);
   old_space_->set_tasks(0);
   ml.NotifyAll();
 }
 
-void Heap::IterateObjects(ObjectVisitor* visitor) const {
-  // The visitor must not allocate from the heap.
-  NoSafepointScope no_safepoint_scope_;
-  new_space_.VisitObjects(visitor);
-  IterateOldObjects(visitor);
+void HeapIterationScope::IterateObjects(ObjectVisitor* visitor) const {
+  heap_->VisitObjects(visitor);
 }
 
-void Heap::IterateOldObjects(ObjectVisitor* visitor) const {
-  HeapIterationScope heap_iteration_scope;
-  old_space_.VisitObjects(visitor);
+void HeapIterationScope::IterateObjectsNoImagePages(
+    ObjectVisitor* visitor) const {
+  heap_->new_space()->VisitObjects(visitor);
+  heap_->old_space()->VisitObjectsNoImagePages(visitor);
 }
 
-void Heap::IterateOldObjectsNoImagePages(ObjectVisitor* visitor) const {
-  HeapIterationScope heap_iteration_scope;
-  old_space_.VisitObjectsNoImagePages(visitor);
+void HeapIterationScope::IterateOldObjects(ObjectVisitor* visitor) const {
+  old_space_->VisitObjects(visitor);
+}
+
+void HeapIterationScope::IterateOldObjectsNoImagePages(
+    ObjectVisitor* visitor) const {
+  old_space_->VisitObjectsNoImagePages(visitor);
+}
+
+void HeapIterationScope::IterateVMIsolateObjects(ObjectVisitor* visitor) const {
+  Dart::vm_isolate()->heap()->VisitObjects(visitor);
+}
+
+void HeapIterationScope::IterateObjectPointers(ObjectPointerVisitor* visitor,
+                                               bool validate_frames) {
+  isolate()->VisitObjectPointers(visitor, validate_frames);
+}
+
+void HeapIterationScope::IterateStackPointers(ObjectPointerVisitor* visitor,
+                                              bool validate_frames) {
+  isolate()->VisitStackPointers(visitor, validate_frames);
 }
 
 void Heap::VisitObjectPointers(ObjectPointerVisitor* visitor) const {
   new_space_.VisitObjectPointers(visitor);
   old_space_.VisitObjectPointers(visitor);
 }
 
 RawInstructions* Heap::FindObjectInCodeSpace(FindObjectVisitor* visitor) const {
   // Only executable pages can have RawInstructions objects.
   RawObject* raw_obj = old_space_.FindObject(visitor, HeapPage::kExecutable);
   ASSERT((raw_obj == Object::null()) ||
          (raw_obj->GetClassId() == kInstructionsCid));
   return reinterpret_cast<RawInstructions*>(raw_obj);
 }
 
 RawObject* Heap::FindOldObject(FindObjectVisitor* visitor) const {
-  HeapIterationScope heap_iteration_scope;
   return old_space_.FindObject(visitor, HeapPage::kData);
 }
 
 RawObject* Heap::FindNewObject(FindObjectVisitor* visitor) const {
   return new_space_.FindObject(visitor);
 }
 
 RawObject* Heap::FindObject(FindObjectVisitor* visitor) const {
   // The visitor must not allocate from the heap.
   NoSafepointScope no_safepoint_scope;
@@ skipping 263 matching lines @@
     // VM isolate heap is premarked.
     VerifyObjectVisitor vm_object_visitor(isolate(), allocated_set,
                                           kRequireMarked);
     vm_isolate->heap()->VisitObjects(&vm_object_visitor);
   }
 
   return allocated_set;
 }
 
 bool Heap::Verify(MarkExpectation mark_expectation) const {
-  HeapIterationScope heap_iteration_scope;
+  HeapIterationScope heap_iteration_scope(Thread::Current());
   return VerifyGC(mark_expectation);
 }
 
 bool Heap::VerifyGC(MarkExpectation mark_expectation) const {
   StackZone stack_zone(Thread::Current());
 
   // Change the new space's top_ with the more up-to-date thread's view of top_
   new_space_.FlushTLS();
 
   ObjectSet* allocated_set =
@@ skipping 268 matching lines @@
     : StackResource(thread) {
   Dart::vm_isolate()->heap()->WriteProtect(false);
 }
 
 WritableVMIsolateScope::~WritableVMIsolateScope() {
   ASSERT(Dart::vm_isolate()->heap()->UsedInWords(Heap::kNew) == 0);
   Dart::vm_isolate()->heap()->WriteProtect(true);
 }
 
 }  // namespace dart