Allow invalidation and recompilation of instance allocation stubs. Requested in order to implement…

runtime/vm/compiler_test.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 "platform/assert.h"
 #include "vm/class_finalizer.h"
+#include "vm/code_patcher.h"
 #include "vm/compiler.h"
 #include "vm/dart_api_impl.h"
 #include "vm/object.h"
 #include "vm/symbols.h"
 #include "vm/unit_test.h"
 
 namespace dart {
 
 TEST_CASE(CompileScript) {
   const char* kScriptChars =
@@ skipping 44 matching lines @@
   EXPECT(!function_moo.IsNull());
 
   EXPECT(CompilerTest::TestCompileFunction(function_moo));
   EXPECT(function_moo.HasCode());
   function_source = function_moo.GetSource();
   EXPECT_STREQ("static moo() {\n    // A.foo();\n  }\n",
                function_source.ToCString());
 }
 
 
+TEST_CASE(RegenerateAllocStubs) {
+  const char* kScriptChars =
+            "class A {\n"
+            "}\n"
+            "unOpt() => new A(); \n"
+            "optIt() => new A(); \n"
+            "main() {\n"
+            "  return unOpt();\n"
+            "}\n";
+
+  Dart_Handle lib = TestCase::LoadTestScript(kScriptChars, NULL);
+  Dart_Handle result = Dart_Invoke(lib, NewString("main"), 0, NULL);
+  EXPECT_VALID(result);
+  RawLibrary* raw_library = Library::RawCast(Api::UnwrapHandle(lib));
+  Library& lib_handle = Library::ZoneHandle(raw_library);
+  Class& cls = Class::Handle(
+      lib_handle.LookupClass(String::Handle(Symbols::New("A"))));
+  EXPECT(!cls.IsNull());
+
+  Isolate* isolate = Isolate::Current();
+  StubCode* stub_code = isolate->stub_code();
+  const Code& stub = Code::Handle(isolate,
+                                  stub_code->GetAllocationStubForClass(cls));
+  Class& owner = Class::Handle();
+  owner ^= stub.owner();
+  owner.DisableAllocationStub();
+  result = Dart_Invoke(lib, NewString("main"), 0, NULL);
+  EXPECT_VALID(result);
+}
+
+
 TEST_CASE(EvalExpression) {
   const char* kScriptChars =
       "int ten = 2 * 5;              \n"
       "get dot => '.';               \n"
       "class A {                     \n"
       "  var apa = 'Herr Nilsson';   \n"
       "  calc(x) => '${x*ten}';      \n"
       "}                             \n"
       "makeObj() => new A();         \n";
 
@@ skipping 46 matching lines @@
     val = lib.Evaluate(expression, Array::empty_array(), Array::empty_array());
   }
 
   EXPECT(!val.IsNull());
   EXPECT(!val.IsError());
   EXPECT(val.IsInteger());
   EXPECT_EQ(7, Integer::Cast(val).AsInt64Value());
 }
 
 }  // namespace dart