Simplify pushing function argument for context allocation.

src/full-codegen.cc

Index: src/full-codegen.cc
diff --git a/src/full-codegen.cc b/src/full-codegen.cc
index 4101b29f507719905fa648fb50f58b2144e08b82..8aa763405aba574ef234f31999e8ae694f295382 100644
--- a/src/full-codegen.cc
+++ b/src/full-codegen.cc
@@ -819,7 +819,7 @@ void FullCodeGenerator::VisitBlock(Block* stmt) {
     { Comment cmnt(masm_, "[ Extend block context");
       scope_ = stmt->block_scope();
       __ Push(scope_->GetSerializedScopeInfo());
-      PushFunctionArgumentForContextAllocation();
+      PushFunctionArgumentForContextAllocation(result_register());
       __ CallRuntime(Runtime::kPushBlockContext, 2);
       StoreToFrameField(StandardFrameConstants::kContextOffset,
                         context_register());
@@ -965,7 +965,7 @@ void FullCodeGenerator::VisitWithStatement(WithStatement* stmt) {
   SetStatementPosition(stmt);
 
   VisitForStackValue(stmt->expression());
-  PushFunctionArgumentForContextAllocation();
+  PushFunctionArgumentForContextAllocation(result_register());
   __ CallRuntime(Runtime::kPushWithContext, 2);
   decrement_stack_height();
   StoreToFrameField(StandardFrameConstants::kContextOffset, context_register());
@@ -1116,7 +1116,7 @@ void FullCodeGenerator::VisitTryCatchStatement(TryCatchStatement* stmt) {
   { Comment cmnt(masm_, "[ Extend catch context");
     __ Push(stmt->variable()->name());
     __ push(result_register());
-    PushFunctionArgumentForContextAllocation();
+    PushFunctionArgumentForContextAllocation(result_register());
     __ CallRuntime(Runtime::kPushCatchContext, 3);
     StoreToFrameField(StandardFrameConstants::kContextOffset,
                       context_register());