First step towards generation of application script snapshots

vm/snapshot_test.cc

 // Copyright (c) 2011, 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/snapshot.h"
 
 #include "vm/assert.h"
 #include "vm/bigint_operations.h"
 #include "vm/class_finalizer.h"
 #include "vm/unit_test.h"
@@ skipping 26 matching lines @@
 
 
 static uint8_t* allocator(uint8_t* ptr, intptr_t old_size, intptr_t new_size) {
   return reinterpret_cast<uint8_t*>(realloc(ptr, new_size));
 }
 
 
 TEST_CASE(SerializeNull) {
   // Write snapshot with object content.
   uint8_t* buffer;
-  SnapshotWriter writer(false, &buffer, &allocator);
+  SnapshotWriter writer(Snapshot::kMessage, &buffer, &allocator);
   const Object& null_object = Object::Handle();
   writer.WriteObject(null_object.raw());
   writer.FinalizeBuffer();
 
   // Create a snapshot object using the buffer.
   const Snapshot* snapshot = Snapshot::SetupFromBuffer(buffer);
 
   // Read object back from the snapshot.
   Isolate* isolate= Isolate::Current();
   SnapshotReader reader(snapshot, isolate->heap(), isolate->object_store());
   const Object& serialized_object = Object::Handle(reader.ReadObject());
   EXPECT(Equals(null_object, serialized_object));
 }
 
 
 TEST_CASE(SerializeSmi1) {
   // Write snapshot with object content.
   uint8_t* buffer;
-  SnapshotWriter writer(false, &buffer, &allocator);
+  SnapshotWriter writer(Snapshot::kMessage, &buffer, &allocator);
   const Smi& smi = Smi::Handle(Smi::New(124));
   writer.WriteObject(smi.raw());
   writer.FinalizeBuffer();
 
   // Create a snapshot object using the buffer.
   const Snapshot* snapshot = Snapshot::SetupFromBuffer(buffer);
 
   // Read object back from the snapshot.
   Isolate* isolate= Isolate::Current();
   SnapshotReader reader(snapshot, isolate->heap(), isolate->object_store());
   const Object& serialized_object = Object::Handle(reader.ReadObject());
   EXPECT(Equals(smi, serialized_object));
 }
 
 
 TEST_CASE(SerializeSmi2) {
   // Write snapshot with object content.
   uint8_t* buffer;
-  SnapshotWriter writer(false, &buffer, &allocator);
+  SnapshotWriter writer(Snapshot::kMessage, &buffer, &allocator);
   const Smi& smi = Smi::Handle(Smi::New(-1));
   writer.WriteObject(smi.raw());
   writer.FinalizeBuffer();
 
   // Create a snapshot object using the buffer.
   const Snapshot* snapshot = Snapshot::SetupFromBuffer(buffer);
 
   // Read object back from the snapshot.
   Isolate* isolate= Isolate::Current();
   SnapshotReader reader(snapshot, isolate->heap(), isolate->object_store());
   const Object& serialized_object = Object::Handle(reader.ReadObject());
   EXPECT(Equals(smi, serialized_object));
 }
 
 
 TEST_CASE(SerializeDouble) {
   // Write snapshot with object content.
   uint8_t* buffer;
-  SnapshotWriter writer(false, &buffer, &allocator);
+  SnapshotWriter writer(Snapshot::kMessage, &buffer, &allocator);
   const Double& dbl = Double::Handle(Double::New(101.29));
   writer.WriteObject(dbl.raw());
   writer.FinalizeBuffer();
 
   // Create a snapshot object using the buffer.
   const Snapshot* snapshot = Snapshot::SetupFromBuffer(buffer);
 
   // Read object back from the snapshot.
   Isolate* isolate= Isolate::Current();
   SnapshotReader reader(snapshot, isolate->heap(), isolate->object_store());
   const Object& serialized_object = Object::Handle(reader.ReadObject());
   EXPECT(Equals(dbl, serialized_object));
 }
 
 
 TEST_CASE(SerializeBool) {
   // Write snapshot with object content.
   uint8_t* buffer;
-  SnapshotWriter writer(false, &buffer, &allocator);
+  SnapshotWriter writer(Snapshot::kMessage, &buffer, &allocator);
   const Bool& bool1 = Bool::Handle(Bool::True());
   const Bool& bool2 = Bool::Handle(Bool::False());
   writer.WriteObject(bool1.raw());
   writer.WriteObject(bool2.raw());
   writer.FinalizeBuffer();
 
   // Create a snapshot object using the buffer.
   const Snapshot* snapshot = Snapshot::SetupFromBuffer(buffer);
 
   // Read object back from the snapshot.
   Isolate* isolate= Isolate::Current();
   SnapshotReader reader(snapshot, isolate->heap(), isolate->object_store());
   EXPECT(Bool::True() == reader.ReadObject());
   EXPECT(Bool::False() == reader.ReadObject());
 }
 
 
 TEST_CASE(SerializeBigint) {
   // Write snapshot with object content.
   uint8_t* buffer;
-  SnapshotWriter writer(false, &buffer, &allocator);
+  SnapshotWriter writer(Snapshot::kMessage, &buffer, &allocator);
   const Bigint& bigint = Bigint::Handle(Bigint::New(0xfffffffffLL));
   writer.WriteObject(bigint.raw());
   writer.FinalizeBuffer();
 
   // Create a snapshot object using the buffer.
   const Snapshot* snapshot = Snapshot::SetupFromBuffer(buffer);
 
   // Read object back from the snapshot.
   Isolate* isolate= Isolate::Current();
   SnapshotReader reader(snapshot, isolate->heap(), isolate->object_store());
   Bigint& obj = Bigint::Handle();
   obj ^= reader.ReadObject();
   OS::Print("%lld", BigintOperations::ToInt64(obj));
   EXPECT_EQ(BigintOperations::ToInt64(bigint), BigintOperations::ToInt64(obj));
 }
 
 
 TEST_CASE(SerializeSingletons) {
   // Write snapshot with object content.
   uint8_t* buffer;
-  SnapshotWriter writer(false, &buffer, &allocator);
+  SnapshotWriter writer(Snapshot::kMessage, &buffer, &allocator);
   writer.WriteObject(Object::class_class());
   writer.WriteObject(Object::null_class());
   writer.WriteObject(Object::type_class());
   writer.WriteObject(Object::type_parameter_class());
   writer.WriteObject(Object::instantiated_type_class());
   writer.WriteObject(Object::abstract_type_arguments_class());
   writer.WriteObject(Object::type_arguments_class());
   writer.WriteObject(Object::instantiated_type_arguments_class());
   writer.WriteObject(Object::function_class());
   writer.WriteObject(Object::field_class());
@@ skipping 32 matching lines @@
   EXPECT(Object::pc_descriptors_class() == reader.ReadObject());
   EXPECT(Object::exception_handlers_class() == reader.ReadObject());
   EXPECT(Object::context_class() == reader.ReadObject());
   EXPECT(Object::context_scope_class() == reader.ReadObject());
 }
 
 
 TEST_CASE(SerializeString) {
   // Write snapshot with object content.
   uint8_t* buffer;
-  SnapshotWriter writer(false, &buffer, &allocator);
+  SnapshotWriter writer(Snapshot::kMessage, &buffer, &allocator);
   String& str = String::Handle(String::New("This string shall be serialized"));
   writer.WriteObject(str.raw());
   writer.FinalizeBuffer();
 
   // Create a snapshot object using the buffer.
   const Snapshot* snapshot = Snapshot::SetupFromBuffer(buffer);
 
   // Read object back from the snapshot.
   Isolate* isolate= Isolate::Current();
   SnapshotReader reader(snapshot, isolate->heap(), isolate->object_store());
   String& serialized_str = String::Handle();
   serialized_str ^= reader.ReadObject();
   EXPECT(str.Equals(serialized_str));
 }
 
 
 TEST_CASE(SerializeArray) {
   // Write snapshot with object content.
   uint8_t* buffer;
-  SnapshotWriter writer(false, &buffer, &allocator);
+  SnapshotWriter writer(Snapshot::kMessage, &buffer, &allocator);
   const int kArrayLength = 10;
   Array& array = Array::Handle(Array::New(kArrayLength));
   Smi& smi = Smi::Handle();
   for (int i = 0; i < kArrayLength; i++) {
     smi ^= Smi::New(i);
     array.SetAt(i, smi);
   }
   writer.WriteObject(array.raw());
   writer.FinalizeBuffer();
 
@@ skipping 19 matching lines @@
   String& url = String::Handle(String::New("dart-test:SerializeScript"));
   String& source = String::Handle(String::New(kScriptChars));
   Script& script = Script::Handle(Script::New(url, source, RawScript::kSource));
   const String& lib_url = String::Handle(String::NewSymbol("TestLib"));
   Library& lib = Library::Handle(Library::New(lib_url));
   lib.Register();
   EXPECT(CompilerTest::TestCompileScript(lib, script));
 
   // Write snapshot with object content.
   uint8_t* buffer;
-  SnapshotWriter writer(false, &buffer, &allocator);
+  SnapshotWriter writer(Snapshot::kMessage, &buffer, &allocator);
   writer.WriteObject(script.raw());
   writer.FinalizeBuffer();
 
   // Create a snapshot object using the buffer.
   const Snapshot* snapshot = Snapshot::SetupFromBuffer(buffer);
 
   // Read object back from the snapshot.
   Isolate* isolate= Isolate::Current();
   SnapshotReader reader(snapshot, isolate->heap(), isolate->object_store());
   Script& serialized_script = Script::Handle();
@@ skipping 48 matching lines @@
 
     // Create a test library and Load up a test script in it.
     TestCase::LoadTestScript(kScriptChars, NULL);
     timer1.Stop();
     OS::PrintErr("Without Snapshot: %dus\n", timer1.TotalElapsedTime());
 
     // Write snapshot with object content.
     Isolate* isolate = Isolate::Current();
     Zone zone(isolate);
     HandleScope scope(isolate);
-    SnapshotWriter writer(true, &buffer, &allocator);
+    SnapshotWriter writer(Snapshot::kFull, &buffer, &allocator);
     writer.WriteFullSnapshot();
   }
 
   // Now Create another isolate using the snapshot and execute a method
   // from the script.
   Timer timer2(true, "Snapshot_test");
   timer2.Start();
   TestCase::CreateTestIsolateFromSnapshot(buffer);
   {
     Dart_EnterScope();  // Start a Dart API scope for invoking API functions.
@@ skipping 35 matching lines @@
     Zone zone(isolate);
     HandleScope scope(isolate);
 
     // Create a test library and Load up a test script in it.
     Dart_Handle lib = TestCase::LoadTestScript(kScriptChars, NULL);
     ClassFinalizer::FinalizePendingClasses();
     timer1.Stop();
     OS::PrintErr("Without Snapshot: %dus\n", timer1.TotalElapsedTime());
 
     // Write snapshot with object content.
-    SnapshotWriter writer(true, &buffer, &allocator);
+    SnapshotWriter writer(Snapshot::kFull, &buffer, &allocator);
     writer.WriteFullSnapshot();
 
     // Invoke a function which returns an object.
     Dart_Handle result = Dart_InvokeStatic(lib,
                                Dart_NewString("FieldsTest"),
                                Dart_NewString("testMain"),
                                0,
                                NULL);
     EXPECT_VALID(result);
   }
@@ skipping 17 matching lines @@
     if (Dart_IsError(result)) {
       // Print the error.  It is probably an unhandled exception.
       fprintf(stderr, "%s\n", Dart_GetError(result));
     }
     EXPECT_VALID(result);
     Dart_ExitScope();
   }
   Dart_ShutdownIsolate();
   free(buffer);
 }
+
+
+UNIT_TEST_CASE(ScriptSnapshot) {
+  const char* kScriptChars =
+      "class Fields  {\n"
+      "  Fields(int i, int j) : fld1 = i, fld2 = j {}\n"
+      "  int fld1;\n"
+      "  final int fld2;\n"
+      "  static int fld3;\n"
+      "  static final int fld4 = 10;\n"
+      "}\n"
+      "class FieldsTest {\n"
+      "  static Fields testMain() {\n"
+      "    Fields obj = new Fields(10, 20);\n"
+      "    return obj;\n"
+      "  }\n"
+      "}\n";
+  Dart_Handle result;
+
+  uint8_t* buffer;
+  intptr_t size;
+
+  // Start an Isolate, load a script and create a script snapshot.
+  {
+    TestIsolateScope __test_isolate__;
+    Dart_EnterScope();  // Start a Dart API scope for invoking API functions.
+
+    // Create a test library and Load up a test script in it.
+    TestCase::LoadTestScript(kScriptChars, NULL);
+
+    // Write out the script snapshot.
+    result = Dart_CreateScriptSnapshot(TestCase::lib(), &buffer, &size);
+    EXPECT_VALID(result);
+    Dart_ExitScope();
+  }
+
+  // Now Create another isolate and load the application snapshot and
+  // execute it.
+  {
+    TestIsolateScope __test_isolate__;
+    Dart_EnterScope();  // Start a Dart API scope for invoking API functions.
+
+    // Load the test library from the snapshot.
+    result = Dart_LoadScriptFromSnapshot(buffer);
+    EXPECT_VALID(result);
+
+    // Invoke a function which returns an object.
+    result = Dart_InvokeStatic(result,
+                               Dart_NewString("FieldsTest"),
+                               Dart_NewString("testMain"),
+                               0,
+                               NULL);
+    EXPECT_VALID(result);
+    Dart_ExitScope();
+  }
+  Dart_ShutdownIsolate();
+  free(buffer);
+}
+
 #endif  // TARGET_ARCH_IA32.
 
 }  // namespace dart