[Interpreter] Add support for local context loads and stores.

test/cctest/interpreter/test-bytecode-generator.cc

 // Copyright 2015 the V8 project authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "src/v8.h"
 
 #include "src/compiler.h"
 #include "src/interpreter/bytecode-array-iterator.h"
 #include "src/interpreter/bytecode-generator.h"
 #include "src/interpreter/interpreter.h"
@@ skipping 116 matching lines @@
 }
 
 
 template <typename T>
 static void CheckBytecodeArrayEqual(struct ExpectedSnippet<T> expected,
                                     Handle<BytecodeArray> actual,
                                     bool has_unknown = false) {
   CHECK_EQ(actual->frame_size(), expected.frame_size);
   CHECK_EQ(actual->parameter_count(), expected.parameter_count);
   CHECK_EQ(actual->length(), expected.bytecode_length);
-  if (expected.constant_count != -1) {
-    if (expected.constant_count == 0) {
-      CHECK_EQ(actual->constant_pool(), CcTest::heap()->empty_fixed_array());
-    } else {
-      CHECK_EQ(actual->constant_pool()->length(), expected.constant_count);
-      for (int i = 0; i < expected.constant_count; i++) {
-        CheckConstant(expected.constants[i], actual->constant_pool()->get(i));
-      }
+  if (expected.constant_count == 0) {
+    CHECK_EQ(actual->constant_pool(), CcTest::heap()->empty_fixed_array());
+  } else {
+    CHECK_EQ(actual->constant_pool()->length(), expected.constant_count);
+    for (int i = 0; i < expected.constant_count; i++) {
+      CheckConstant(expected.constants[i], actual->constant_pool()->get(i));
     }
   }
 
   BytecodeArrayIterator iterator(actual);
   int i = 0;
   while (!iterator.done()) {
     int bytecode_index = i++;
     Bytecode bytecode = iterator.current_bytecode();
     if (Bytecodes::ToByte(bytecode) != expected.bytecode[bytecode_index]) {
       std::ostringstream stream;
@@ skipping 1330 matching lines @@
         helper.MakeBytecode(snippets[i].code_snippet, helper.kFunctionName);
     CheckBytecodeArrayEqual(snippets[i], bytecode_array);
   }
 }
 
 
 TEST(DeclareGlobals) {
   InitializedHandleScope handle_scope;
   BytecodeGeneratorHelper helper;
 
-  ExpectedSnippet<int> snippets[] = {
+  ExpectedSnippet<InstanceType> snippets[] = {
       {"var a = 1;",
        5 * kPointerSize,
        1,
        45,
        {
            B(Ldar), R(Register::function_closure().index()),              //
            B(Star), R(2),                                                 //
            B(LdaConstant), U8(0),                                         //
            B(Star), R(3),                                                 //
            B(CallRuntime), U16(Runtime::kNewScriptContext), R(2), U8(2),  //
            B(PushContext), R(1),                                          //
            B(LdaConstant), U8(1),                                         //
            B(Star), R(2),                                                 //
            B(LdaZero),                                                    //
            B(Star), R(3),                                                 //
            B(CallRuntime), U16(Runtime::kDeclareGlobals), R(2), U8(2),    //
            B(LdaConstant), U8(2),                                         //
            B(Star), R(2),                                                 //
            B(LdaZero),                                                    //
            B(Star), R(3),                                                 //
            B(LdaSmi8), U8(1),                                             //
            B(Star), R(4),                                                 //
            B(CallRuntime), U16(Runtime::kInitializeVarGlobal), R(2),      //
                            U8(3),                                         //
            B(LdaUndefined),                                               //
            B(Return)                                                      //
        },
-       -1},
+       3,
+       {InstanceType::FIXED_ARRAY_TYPE,
+        InstanceType::FIXED_ARRAY_TYPE,
+        InstanceType::ONE_BYTE_INTERNALIZED_STRING_TYPE}},
       {"function f() {}",
        3 * kPointerSize,
        1,
        29,
        {
            B(Ldar), R(Register::function_closure().index()),              //
            B(Star), R(1),                                                 //
            B(LdaConstant), U8(0),                                         //
            B(Star), R(2),                                                 //
            B(CallRuntime), U16(Runtime::kNewScriptContext), R(1), U8(2),  //
            B(PushContext), R(0),                                          //
            B(LdaConstant), U8(1),                                         //
            B(Star), R(1),                                                 //
            B(LdaZero),                                                    //
            B(Star), R(2),                                                 //
            B(CallRuntime), U16(Runtime::kDeclareGlobals), R(1), U8(2),    //
            B(LdaUndefined),                                               //
            B(Return)                                                      //
        },
-       -1},
+       2,
+       {InstanceType::FIXED_ARRAY_TYPE, InstanceType::FIXED_ARRAY_TYPE}},
       {"var a = 1;\na=2;",
        5 * kPointerSize,
        1,
        52,
        {
            B(Ldar), R(Register::function_closure().index()),              //
            B(Star), R(2),                                                 //
            B(LdaConstant), U8(0),                                         //
            B(Star), R(3),                                                 //
            B(CallRuntime), U16(Runtime::kNewScriptContext), R(2), U8(2),  //
@@ skipping 10 matching lines @@
            B(LdaSmi8), U8(1),                                             //
            B(Star), R(4),                                                 //
            B(CallRuntime), U16(Runtime::kInitializeVarGlobal), R(2),      //
                            U8(3),                                         //
            B(LdaSmi8), U8(2),                                             //
            B(StaGlobal), _,                                               //
            B(Star), R(0),                                                 //
            B(Ldar), R(0),                                                 //
            B(Return)                                                      //
        },
-       -1},
+       3,
+       {InstanceType::FIXED_ARRAY_TYPE,
+        InstanceType::FIXED_ARRAY_TYPE,
+        InstanceType::ONE_BYTE_INTERNALIZED_STRING_TYPE}},
       {"function f() {}\nf();",
        4 * kPointerSize,
        1,
        43,
        {
            B(Ldar), R(Register::function_closure().index()),              //
            B(Star), R(2),                                                 //
            B(LdaConstant), U8(0),                                         //
            B(Star), R(3),                                                 //
            B(CallRuntime), U16(Runtime::kNewScriptContext), R(2), U8(2),  //
            B(PushContext), R(1),                                          //
            B(LdaConstant), U8(1),                                         //
            B(Star), R(2),                                                 //
            B(LdaZero),                                                    //
            B(Star), R(3),                                                 //
            B(CallRuntime), U16(Runtime::kDeclareGlobals), R(2), U8(2),    //
            B(LdaUndefined),                                               //
            B(Star), R(3),                                                 //
            B(LdaGlobal), _,                                               //
            B(Star), R(2),                                                 //
            B(Call), R(2), R(3), U8(0),                                    //
            B(Star), R(0),                                                 //
            B(Ldar), R(0),                                                 //
            B(Return)                                                      //
        },
-       -1},
+       2,
+       {InstanceType::FIXED_ARRAY_TYPE, InstanceType::FIXED_ARRAY_TYPE}},
   };
 
   for (size_t i = 0; i < arraysize(snippets); i++) {
     Handle<BytecodeArray> bytecode_array =
         helper.MakeTopLevelBytecode(snippets[i].code_snippet);
     CheckBytecodeArrayEqual(snippets[i], bytecode_array, true);
   }
 }
 
 
@@ skipping 148 matching lines @@
        "  x = x + 1;"
        "}",
        2 * kPointerSize,
        1,
        29,
        {
            B(LdaZero),             //
            B(Star), R(0),          //
            B(Ldar), R(0),          //
            B(Star), R(1),          //
-           B(LdaSmi8),             //
-           U8(1),                  //
+           B(LdaSmi8), U8(1),      //
            B(TestEqual), R(1),     //
            B(JumpIfFalse), U8(4),  //
            B(Jump), U8(14),        //
            B(Ldar), R(0),          //
            B(Star), R(1),          //
            B(LdaSmi8), U8(1),      //
            B(Add), R(1),           //
            B(Star), R(0),          //
            B(Jump), U8(-22),       //
            B(LdaUndefined),        //
@@ skipping 817 matching lines @@
 }
 
 
 TEST(TryCatch) {
   InitializedHandleScope handle_scope;
   BytecodeGeneratorHelper helper;
 
   // TODO(rmcilroy): modify tests when we have real try catch support.
   ExpectedSnippet<int> snippets[] = {
       {"try { return 1; } catch(e) { return 2; }",
-       0,
+       1 * kPointerSize,
        1,
        5,
        {
            B(LdaSmi8), U8(1),  //
            B(Return),          //
            B(LdaUndefined),    //
            B(Return),          //
        },
        0},
   };
@@ skipping 21 matching lines @@
            B(Star), R(0),      //
            B(LdaSmi8), U8(2),  //
            B(Star), R(0),      //
            B(LdaSmi8), U8(3),  //
            B(Star), R(0),      //
            B(LdaUndefined),    //
            B(Return),          //
        },
        0},
       {"var a = 1; try { a = 2; } catch(e) { a = 20 } finally { a = 3; }",
-       1 * kPointerSize,
+       2 * kPointerSize,
        1,
        14,
        {
            B(LdaSmi8), U8(1),  //
            B(Star), R(0),      //
            B(LdaSmi8), U8(2),  //
            B(Star), R(0),      //
            B(LdaSmi8), U8(3),  //
            B(Star), R(0),      //
            B(LdaUndefined),    //
            B(Return),          //
        },
        0},
   };
 
   for (size_t i = 0; i < arraysize(snippets); i++) {
     Handle<BytecodeArray> bytecode_array =
         helper.MakeBytecodeForFunctionBody(snippets[i].code_snippet);
     CheckBytecodeArrayEqual(snippets[i], bytecode_array);
   }
 }
 
+
+TEST(ContextVariables) {
+  InitializedHandleScope handle_scope;
+  BytecodeGeneratorHelper helper;
+
+  int closure = Register::function_closure().index();
+  int first_context_slot = Context::MIN_CONTEXT_SLOTS;
+  ExpectedSnippet<InstanceType> snippets[] = {
+      {"var a; return function() { a = 1; };",
+       1 * kPointerSize,
+       1,
+       12,
+       {
+           B(CallRuntime), U16(Runtime::kNewFunctionContext),  //
+                           R(closure), U8(1),                  //
+           B(PushContext), R(0),                               //
+           B(LdaConstant), U8(0),                              //
+           B(CreateClosure), U8(0),                            //
+           B(Return),                                          //
+       },
+       1,
+       {InstanceType::SHARED_FUNCTION_INFO_TYPE}},
+      {"var a = 1; return function() { a = 2; };",
+       1 * kPointerSize,
+       1,
+       17,
+       {
+           B(CallRuntime), U16(Runtime::kNewFunctionContext),  //
+                           R(closure), U8(1),                  //
+           B(PushContext), R(0),                               //
+           B(LdaSmi8), U8(1),                                  //
+           B(StaContextSlot), R(0), U8(first_context_slot),    //
+           B(LdaConstant), U8(0),                              //
+           B(CreateClosure), U8(0),                            //
+           B(Return),                                          //
+       },
+       1,
+       {InstanceType::SHARED_FUNCTION_INFO_TYPE}},
+      {"var a = 1; var b = 2; return function() { a = 2; b = 3 };",
+       1 * kPointerSize,
+       1,
+       22,
+       {
+           B(CallRuntime), U16(Runtime::kNewFunctionContext),    //
+                           R(closure), U8(1),                    //
+           B(PushContext), R(0),                                 //
+           B(LdaSmi8), U8(1),                                    //
+           B(StaContextSlot), R(0), U8(first_context_slot),      //
+           B(LdaSmi8), U8(2),                                    //
+           B(StaContextSlot), R(0), U8(first_context_slot + 1),  //
+           B(LdaConstant), U8(0),                                //
+           B(CreateClosure), U8(0),                              //
+           B(Return),                                            //
+       },
+       1,
+       {InstanceType::SHARED_FUNCTION_INFO_TYPE}},
+      {"var a; (function() { a = 2; })(); return a;",
+       3 * kPointerSize,
+       1,
+       24,
+       {
+           B(CallRuntime), U16(Runtime::kNewFunctionContext),  //
+                           R(closure), U8(1),                  //
+           B(PushContext), R(0),                               //
+           B(LdaUndefined),                                    //
+           B(Star), R(2),                                      //
+           B(LdaConstant), U8(0),                              //
+           B(CreateClosure), U8(0),                            //
+           B(Star), R(1),                                      //
+           B(Call), R(1), R(2), U8(0),                         //
+           B(LdaContextSlot), R(0), U8(first_context_slot),    //
+           B(Return),                                          //
+       },
+       1,
+       {InstanceType::SHARED_FUNCTION_INFO_TYPE}},
+      {"'use strict'; let a = 1; { let b = 2; return function() { a + b; }; }",
+       4 * kPointerSize,
+       1,
+       51,
+       {
+           B(CallRuntime), U16(Runtime::kNewFunctionContext),             //
+                           R(closure), U8(1),                             //
+           B(PushContext), R(0),                                          //
+           B(LdaTheHole),                                                 //
+           B(StaContextSlot), R(0), U8(first_context_slot),               //
+           B(LdaConstant), U8(0),                                         //
+           B(LdaSmi8), U8(1),                                             //
+           B(StaContextSlot), R(0), U8(first_context_slot),               //
+           B(LdaConstant), U8(1),                                         //
+           B(Star), R(2),                                                 //
+           B(Ldar), R(closure),                                           //
+           B(Star), R(3),                                                 //
+           B(CallRuntime), U16(Runtime::kPushBlockContext), R(2), U8(2),  //
+           B(PushContext), R(1),                                          //
+           B(LdaTheHole),                                                 //
+           B(StaContextSlot), R(1), U8(first_context_slot),               //
+           B(LdaSmi8), U8(2),                                             //
+           B(StaContextSlot), R(1), U8(first_context_slot),               //
+           B(LdaConstant), U8(2),                                         //
+           B(CreateClosure), U8(0),                                       //
+           B(Return),                                                     //
+           // TODO(rmcilroy): Dead code after this point due to return in nested
+           // block - investigate eliminating this.
+           B(PopContext), R(0),
+           B(LdaUndefined),  //
+           B(Return),        //
+       },
+       3,
+       {InstanceType::ONE_BYTE_INTERNALIZED_STRING_TYPE,
+        InstanceType::FIXED_ARRAY_TYPE,
+        InstanceType::SHARED_FUNCTION_INFO_TYPE}},
+  };
+
+  for (size_t i = 0; i < arraysize(snippets); i++) {
+    Handle<BytecodeArray> bytecode_array =
+        helper.MakeBytecodeForFunctionBody(snippets[i].code_snippet);
+    CheckBytecodeArrayEqual(snippets[i], bytecode_array);
+  }
+}
+
+
+TEST(ContextParameters) {
+  InitializedHandleScope handle_scope;
+  BytecodeGeneratorHelper helper;
+
+  int closure = Register::function_closure().index();
+  int first_context_slot = Context::MIN_CONTEXT_SLOTS;
+  ExpectedSnippet<InstanceType> snippets[] = {
+      {"function f(arg1) { return function() { arg1 = 2; }; }",
+       1 * kPointerSize,
+       2,
+       17,
+       {
+           B(CallRuntime), U16(Runtime::kNewFunctionContext),  //
+                           R(closure), U8(1),                  //
+           B(PushContext), R(0),                               //
+           B(Ldar), R(helper.kLastParamIndex),                 //
+           B(StaContextSlot), R(0), U8(first_context_slot),    //
+           B(LdaConstant), U8(0),                              //
+           B(CreateClosure), U8(0),                            //
+           B(Return),                                          //
+       },
+       1,
+       {InstanceType::SHARED_FUNCTION_INFO_TYPE}},
+      {"function f(arg1) { var a = function() { arg1 = 2; }; return arg1; }",
+       2 * kPointerSize,
+       2,
+       22,
+       {
+           B(CallRuntime), U16(Runtime::kNewFunctionContext),  //
+                           R(closure), U8(1),                  //
+           B(PushContext), R(1),                               //
+           B(Ldar), R(helper.kLastParamIndex),                 //
+           B(StaContextSlot), R(1), U8(first_context_slot),    //
+           B(LdaConstant), U8(0),                              //
+           B(CreateClosure), U8(0),                            //
+           B(Star), R(0),                                      //
+           B(LdaContextSlot), R(1), U8(first_context_slot),    //
+           B(Return),                                          //
+       },
+       1,
+       {InstanceType::SHARED_FUNCTION_INFO_TYPE}},
+      {"function f(a1, a2, a3, a4) { return function() { a1 = a3; }; }",
+       1 * kPointerSize,
+       5,
+       22,
+       {
+           B(CallRuntime), U16(Runtime::kNewFunctionContext),    //
+                           R(closure), U8(1),                    //
+           B(PushContext), R(0),                                 //
+           B(Ldar), R(helper.kLastParamIndex - 3),               //
+           B(StaContextSlot), R(0), U8(first_context_slot + 1),  //
+           B(Ldar), R(helper.kLastParamIndex -1),                //
+           B(StaContextSlot), R(0), U8(first_context_slot),      //
+           B(LdaConstant), U8(0),                                //
+           B(CreateClosure), U8(0),                              //
+           B(Return),                                            //
+       },
+       1,
+       {InstanceType::SHARED_FUNCTION_INFO_TYPE}},
+      {"function f() { var self = this; return function() { self = 2; }; }",
+       1 * kPointerSize,
+       1,
+       17,
+       {
+           B(CallRuntime), U16(Runtime::kNewFunctionContext),  //
+                           R(closure), U8(1),                  //
+           B(PushContext), R(0),                               //
+           B(Ldar), R(helper.kLastParamIndex),                 //
+           B(StaContextSlot), R(0), U8(first_context_slot),    //
+           B(LdaConstant), U8(0),                              //
+           B(CreateClosure), U8(0),                            //
+           B(Return),                                          //
+       },
+       1,
+       {InstanceType::SHARED_FUNCTION_INFO_TYPE}},
+  };
+
+  for (size_t i = 0; i < arraysize(snippets); i++) {
+    Handle<BytecodeArray> bytecode_array =
+        helper.MakeBytecodeForFunction(snippets[i].code_snippet);
+    CheckBytecodeArrayEqual(snippets[i], bytecode_array);
+  }
+}
+
 }  // namespace interpreter
 }  // namespace internal
 }  // namespace v8