VM: Explicitly load function and context before calling a closure.

runtime/vm/flow_graph_builder.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/flow_graph_builder.h"
 
 #include "lib/invocation_mirror.h"
 #include "vm/ast_printer.h"
 #include "vm/bit_vector.h"
 #include "vm/class_finalizer.h"
@@ skipping 2358 matching lines @@
     const intptr_t result_cid = GetResultCidOfNativeFactory(node->function());
     if (result_cid != kDynamicCid) {
       call->set_result_cid(result_cid);
       call->set_is_native_list_factory(true);
     }
   }
   ReturnDefinition(call);
 }
 
 
-ClosureCallInstr* EffectGraphVisitor::BuildClosureCall(
-    ClosureCallNode* node) {
+void EffectGraphVisitor::BuildClosureCall(
+    ClosureCallNode* node, bool result_needed) {
   ValueGraphVisitor for_closure(owner());
   node->closure()->Visit(&for_closure);
   Append(for_closure);
-  PushArgumentInstr* push_closure = PushArgument(for_closure.value());
+
+  LocalVariable* tmp_var = EnterTempLocalScope(for_closure.value());
 
   ZoneGrowableArray<PushArgumentInstr*>* arguments =
       new ZoneGrowableArray<PushArgumentInstr*>(node->arguments()->length());
+  Value* closure_val = Bind(new LoadLocalInstr(*tmp_var));
+  PushArgumentInstr* push_closure = PushArgument(closure_val);
   arguments->Add(push_closure);
   BuildPushArguments(*node->arguments(), arguments);
 
   // Save context around the call.
   ASSERT(owner()->parsed_function()->saved_current_context_var() != NULL);
   BuildSaveContext(*owner()->parsed_function()->saved_current_context_var());
-  return new ClosureCallInstr(node, arguments);
+  closure_val = Bind(new LoadLocalInstr(*tmp_var));
+  Value* context_val = Bind(new LoadFieldInstr(closure_val,
+                                               Closure::context_offset(),
+                                               AbstractType::ZoneHandle(),
+                                               true));  // Immutable.
+  AddInstruction(new StoreContextInstr(context_val));
+  closure_val = Bind(new LoadLocalInstr(*tmp_var));
+  Value* function_val = Bind(new LoadFieldInstr(closure_val,
+                                                Closure::function_offset(),
+                                                AbstractType::ZoneHandle(),
+                                                true));  // Immutable.
+  Definition* closure_call =
+      new ClosureCallInstr(function_val, node, arguments);
+  if (result_needed) {
+    Value* result = Bind(closure_call);
+    Do(new StoreLocalInstr(*tmp_var, result));
+    // Restore context from temp.
+    BuildRestoreContext(
+        *owner()->parsed_function()->saved_current_context_var());
+    ReturnDefinition(ExitTempLocalScope(tmp_var));
+  } else {
+    Do(closure_call);
+    // Restore context from saved location.
+    BuildRestoreContext(
+        *owner()->parsed_function()->saved_current_context_var());
+    Do(ExitTempLocalScope(tmp_var));
+  }
 }
 
 
 void EffectGraphVisitor::VisitClosureCallNode(ClosureCallNode* node) {
-  Do(BuildClosureCall(node));
-  // Restore context from saved location.
-  ASSERT(owner()->parsed_function()->saved_current_context_var() != NULL);
-  BuildRestoreContext(*owner()->parsed_function()->saved_current_context_var());
+  BuildClosureCall(node, false);
 }
 
 
 void ValueGraphVisitor::VisitClosureCallNode(ClosureCallNode* node) {
-  Value* result = Bind(BuildClosureCall(node));
-  // Restore context from temp.
-  ASSERT(owner()->parsed_function()->saved_current_context_var() != NULL);
-  BuildRestoreContext(*owner()->parsed_function()->saved_current_context_var());
-  ReturnValue(result);
+  BuildClosureCall(node, true);
 }
 
 
 void EffectGraphVisitor::VisitCloneContextNode(CloneContextNode* node) {
   Value* context = Bind(new CurrentContextInstr());
   Value* clone = Bind(new CloneContextInstr(node->token_pos(), context));
   AddInstruction(new StoreContextInstr(clone));
 }
 
 
@@ skipping 1436 matching lines @@
                                   function.token_pos(),
                                   LanguageError::kError,
                                   Heap::kNew,
                                   "FlowGraphBuilder Bailout: %s %s",
                                   String::Handle(function.name()).ToCString(),
                                   reason));
   Isolate::Current()->long_jump_base()->Jump(1, error);
 }
 
 }  // namespace dart