Cleanups needed for async step-out

runtime/vm/object_store.cc

 // Copyright (c) 2013, 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/object_store.h"
 
 #include "vm/exceptions.h"
 #include "vm/dart_entry.h"
 #include "vm/isolate.h"
 #include "vm/object.h"
@@ skipping 79 matching lines @@
       preallocated_stack_trace_(StackTrace::null()),
       lookup_port_handler_(Function::null()),
       empty_uint32_array_(TypedData::null()),
       handle_message_function_(Function::null()),
       simple_instance_of_function_(Function::null()),
       simple_instance_of_true_function_(Function::null()),
       simple_instance_of_false_function_(Function::null()),
       async_clear_thread_stack_trace_(Function::null()),
       async_set_thread_stack_trace_(Function::null()),
       async_star_move_next_helper_(Function::null()),
+      complete_on_async_return_(Function::null()),
+      async_star_stream_controller_(Class::null()),
       library_load_error_table_(Array::null()),
       unique_dynamic_targets_(Array::null()),
       token_objects_(GrowableObjectArray::null()),
       token_objects_map_(Array::null()),
       megamorphic_cache_table_(GrowableObjectArray::null()),
       megamorphic_miss_code_(Code::null()),
       megamorphic_miss_function_(Function::null()) {
   for (RawObject** current = from(); current <= to(); current++) {
     ASSERT(*current == Object::null());
   }
@@ skipping 139 matching lines @@
                                       function_name, 1, Object::null_array());
   set_async_set_thread_stack_trace(function);
 
   function_name ^= async_lib.PrivateName(Symbols::ClearAsyncThreadStackTrace());
   ASSERT(!function_name.IsNull());
   function ^= Resolver::ResolveStatic(async_lib, Object::null_string(),
                                       function_name, 0, Object::null_array());
   ASSERT(!function.IsNull());
   set_async_clear_thread_stack_trace(function);
 
+
   function_name ^= async_lib.PrivateName(Symbols::AsyncStarMoveNextHelper());
   ASSERT(!function_name.IsNull());
   function ^= Resolver::ResolveStatic(async_lib, Object::null_string(),
                                       function_name, 1, Object::null_array());
   ASSERT(!function.IsNull());
   set_async_star_move_next_helper(function);
 
+  function_name ^= async_lib.PrivateName(Symbols::_CompleteOnAsyncReturn());
+  ASSERT(!function_name.IsNull());
+  function ^= Resolver::ResolveStatic(async_lib, Object::null_string(),
+                                      function_name, 2, Object::null_array());
+  ASSERT(!function.IsNull());
+  set_complete_on_async_return(function);
+  if (FLAG_async_debugger_stepping) {
+    // Disable debugging and inlining the _CompleteOnAsyncReturn function.
+    function.set_is_debuggable(false);
+    function.set_is_inlinable(false);
+  }
+
+  cls =
+      async_lib.LookupClassAllowPrivate(Symbols::_AsyncStarStreamController());
+  ASSERT(!cls.IsNull());
+  set_async_star_stream_controller(cls);
+
+  if (FLAG_async_debugger_stepping) {
+    // Disable debugging and inlining of all functions on the
+    // _AsyncStarStreamController class.
+    const Array& functions = Array::Handle(cls.functions());
+    for (intptr_t i = 0; i < functions.Length(); i++) {
+      function ^= functions.At(i);
+      if (function.IsNull()) {
+        break;
+      }
+      function.set_is_debuggable(false);
+      function.set_is_inlinable(false);
+    }
+  }
+
   const Library& internal_lib = Library::Handle(_internal_library());
   cls = internal_lib.LookupClass(Symbols::Symbol());
   set_symbol_class(cls);
 
   const Library& core_lib = Library::Handle(core_library());
   cls = core_lib.LookupClassAllowPrivate(Symbols::_CompileTimeError());
   ASSERT(!cls.IsNull());
   set_compiletime_error_class(cls);
 
   // Cache the core private functions used for fast instance of checks.
   simple_instance_of_function_ =
       PrivateObjectLookup(Symbols::_simpleInstanceOf());
   simple_instance_of_true_function_ =
       PrivateObjectLookup(Symbols::_simpleInstanceOfTrue());
   simple_instance_of_false_function_ =
       PrivateObjectLookup(Symbols::_simpleInstanceOfFalse());
 #endif
 }
 
 }  // namespace dart