Added option --script-snapshot in gen_snapshot to generate the snapshot of a scipt into the file.

runtime/bin/main.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 <stdlib.h>
 #include <string.h>
 #include <stdio.h>
 
 #include "include/dart_api.h"
 #include "include/dart_debugger_api.h"
 
 #include "bin/builtin.h"
 #include "bin/dartutils.h"
 #include "bin/dbg_connection.h"
 #include "bin/directory.h"
 #include "bin/eventhandler.h"
 #include "bin/extensions.h"
 #include "bin/file.h"
 #include "bin/isolate_data.h"
 #include "bin/platform.h"
 #include "bin/process.h"
 #include "platform/globals.h"
 
 // snapshot_buffer points to a snapshot if we link in a snapshot otherwise
 // it is initialized to NULL.
 extern const uint8_t* snapshot_buffer;
 
 
-// Global state that stores the original working directory..
-static const char* original_working_directory = NULL;
-
-
 // Global state that stores the import URL map specified on the
 // command line.
 static CommandLineOptions* import_map_options = NULL;
 
 
 // Global state that indicates whether perf_events symbol information
 // is to be generated or not.
 static File* perf_events_symbols_file = NULL;
 
 
 // Global state that indicates whether pprof symbol information is
 // to be generated or not.
 static const char* generate_pprof_symbols_filename = NULL;
 
 
+// Global state that stores a pointer to the application script snapshot.
+static bool use_script_snapshot = false;
+static File* snapshot_file = NULL;
+
+
 // Global state that stores a file name for flow graph debugging output.
 // NULL if no output is generated.
 static File* flow_graph_file = NULL;
 
 // Global state that indicates whether there is a debug breakpoint.
 // This pointer points into an argv buffer and does not need to be
 // free'd.
 static const char* breakpoint_at = NULL;
 
 
@@ skipping 10 matching lines @@
 // (This pointer points into an argv buffer and does not need to be
 // free'd.)
 static const char* package_root = NULL;
 
 
 // Global flag that is used to indicate that we want to compile all the
 // dart functions and not run anything.
 static bool has_compile_all = false;
 
 
-static bool IsWindowsHost() {
-#if defined(TARGET_OS_WINDOWS)
-  return true;
-#else  // defined(TARGET_OS_WINDOWS)
-  return false;
-#endif  // defined(TARGET_OS_WINDOWS)
-}
-
-
 static bool IsValidFlag(const char* name,
                         const char* prefix,
                         intptr_t prefix_length) {
   intptr_t name_length = strlen(name);
   return ((name_length > prefix_length) &&
           (strncmp(name, prefix, prefix_length) == 0));
 }
 
 
 static void ProcessBreakpointOption(const char* funcname) {
@@ skipping 54 matching lines @@
   }
 }
 
 
 static void ProcessPprofOption(const char* filename) {
   ASSERT(filename != NULL);
   generate_pprof_symbols_filename = filename;
 }
 
 
+static void ProcessScriptSnapshotOption(const char* filename) {
+  if (filename != NULL && strlen(filename) != 0) {
+    use_script_snapshot = true;
+    snapshot_file = File::Open(filename, File::kRead);
+  }
+}
+
+
 static void ProcessFlowGraphOption(const char* flowgraph_option) {
   ASSERT(flowgraph_option != NULL);
   flow_graph_file = File::Open("flowgraph.cfg", File::kWriteTruncate);
   ASSERT(flow_graph_file != NULL);
 }
 
 
 static void ProcessImportMapOption(const char* map) {
   ASSERT(map != NULL);
   import_map_options->AddArgument(map);
 }
 
 
 static struct {
   const char* option_name;
   void (*process)(const char* option);
 } main_options[] = {
   { "--break_at=", ProcessBreakpointOption },
   { "--compile_all", ProcessCompileAllOption },
   { "--debug", ProcessDebugOption },
+  { "--generate_flow_graph", ProcessFlowGraphOption },
   { "--generate_perf_events_symbols", ProcessPerfEventsOption },
   { "--generate_pprof_symbols=", ProcessPprofOption },
   { "--import_map=", ProcessImportMapOption },
   { "--package-root=", ProcessPackageRootOption },
-  { "--generate_flow_graph", ProcessFlowGraphOption },
+  { "--use_script_snapshot=", ProcessScriptSnapshotOption },
   { NULL, NULL }
 };
 
 
 static bool ProcessMainOptions(const char* option) {
   int i = 0;
   const char* name = main_options[0].option_name;
   while (name != NULL) {
     int length = strlen(name);
     if (strncmp(option, name, length) == 0) {
@@ skipping 167 matching lines @@
     if (buffer_size > 0) {
       ASSERT(buffer != NULL);
       pprof_file->WriteFully(buffer, buffer_size);
     }
     delete pprof_file;  // Closes the file.
     Dart_ExitScope();
   }
 }
 
 
-
-static Dart_Handle ResolveScriptUri(Dart_Handle script_uri,
-                                    Dart_Handle builtin_lib) {
-  const int kNumArgs = 3;
-  Dart_Handle dart_args[kNumArgs];
-  dart_args[0] = Dart_NewString(original_working_directory);
-  dart_args[1] = script_uri;
-  dart_args[2] = (IsWindowsHost() ? Dart_True() : Dart_False());
-  return Dart_Invoke(
-      builtin_lib, Dart_NewString("_resolveScriptUri"), kNumArgs, dart_args);
-}
-
-
-static Dart_Handle FilePathFromUri(Dart_Handle script_uri,
-                                   Dart_Handle builtin_lib) {
-  const int kNumArgs = 2;
-  Dart_Handle dart_args[kNumArgs];
-  dart_args[0] = script_uri;
-  dart_args[1] = (IsWindowsHost() ? Dart_True() : Dart_False());
-  Dart_Handle script_path = Dart_Invoke(
-      builtin_lib, Dart_NewString("_filePathFromUri"), kNumArgs, dart_args);
-  return script_path;
-}
-
-
-static Dart_Handle LibraryTagHandler(Dart_LibraryTag tag,
-                                     Dart_Handle library,
-                                     Dart_Handle url) {
-  if (!Dart_IsLibrary(library)) {
-    return Dart_Error("not a library");
-  }
-  if (!Dart_IsString8(url)) {
-    return Dart_Error("url is not a string");
-  }
-  const char* url_string = NULL;
-  Dart_Handle result = Dart_StringToCString(url, &url_string);
-  if (Dart_IsError(result)) {
-    return result;
-  }
-  bool is_dart_scheme_url = DartUtils::IsDartSchemeURL(url_string);
-  bool is_dart_extension_url = DartUtils::IsDartExtensionSchemeURL(url_string);
-  if (tag == kCanonicalizeUrl) {
-    // If this is a Dart Scheme URL then it is not modified as it will be
-    // handled by the VM internally.
-    if (is_dart_scheme_url) {
-      return url;
-    }
-    // Resolve the url within the context of the library's URL.
-    Dart_Handle builtin_lib = Builtin::LoadLibrary(Builtin::kBuiltinLibrary);
-    Dart_Handle library_url = Dart_LibraryUrl(library);
-    if (Dart_IsError(library_url)) {
-      return library_url;
-    }
-    const int kNumArgs = 2;
-    Dart_Handle dart_args[kNumArgs];
-    dart_args[0] = library_url;
-    dart_args[1] = url;
-    return Dart_Invoke(
-        builtin_lib, Dart_NewString("_resolveUri"), kNumArgs, dart_args);
-  }
-  if (is_dart_scheme_url) {
-    ASSERT(tag == kImportTag);
-    // Handle imports of other built-in libraries present in the SDK.
-    Builtin::BuiltinLibraryId id;
-    if (DartUtils::IsDartCryptoLibURL(url_string)) {
-      id = Builtin::kCryptoLibrary;
-    } else if (DartUtils::IsDartIOLibURL(url_string)) {
-      id = Builtin::kIOLibrary;
-    } else if (DartUtils::IsDartJsonLibURL(url_string)) {
-      id = Builtin::kJsonLibrary;
-    } else if (DartUtils::IsDartUriLibURL(url_string)) {
-      id = Builtin::kUriLibrary;
-    } else if (DartUtils::IsDartUtfLibURL(url_string)) {
-      id = Builtin::kUtfLibrary;
-    } else if (DartUtils::IsDartWebLibURL(url_string)) {
-      id = Builtin::kWebLibrary;
-    } else {
-      return Dart_Error("Do not know how to load '%s'", url_string);
-    }
-    return Builtin::LoadLibrary(id);
-  } else {
-    // Get the file path out of the url.
-    Dart_Handle builtin_lib = Builtin::LoadLibrary(Builtin::kBuiltinLibrary);
-    Dart_Handle file_path = FilePathFromUri(url, builtin_lib);
-    if (Dart_IsError(file_path)) {
-      return file_path;
-    }
-    Dart_StringToCString(file_path, &url_string);
-  }
-  if (is_dart_extension_url) {
-    if (tag != kImportTag) {
-      return Dart_Error("Dart extensions must use import: '%s'", url_string);
-    }
-    return Extensions::LoadExtension(url_string, library);
-  }
-  return DartUtils::LoadSource(NULL,
-                               library,
-                               url,
-                               tag,
-                               url_string);
-}
-
-
-static Dart_Handle ReadSource(Dart_Handle script_uri,
-                              Dart_Handle builtin_lib) {
-  Dart_Handle script_path = FilePathFromUri(script_uri, builtin_lib);
-  if (Dart_IsError(script_path)) {
-    return script_path;
-  }
-  const char* script_path_cstr;
-  Dart_StringToCString(script_path, &script_path_cstr);
-  Dart_Handle source = DartUtils::ReadStringFromFile(script_path_cstr);
-  return source;
-}
-
-
-static Dart_Handle LoadScript(const char* script_uri,
-                              bool resolve_script,
-                              Dart_Handle builtin_lib) {
-  Dart_Handle resolved_script_uri;
-  if (resolve_script) {
-    resolved_script_uri = ResolveScriptUri(Dart_NewString(script_uri),
-                                           builtin_lib);
-    if (Dart_IsError(resolved_script_uri)) {
-      return resolved_script_uri;
-    }
-  } else {
-    resolved_script_uri = Dart_NewString(script_uri);
-  }
-  Dart_Handle source = ReadSource(resolved_script_uri, builtin_lib);
-  if (Dart_IsError(source)) {
-    return source;
-  }
-  return Dart_LoadScript(resolved_script_uri, source);
-}
-
-
 static Dart_Handle CreateImportMap(CommandLineOptions* map) {
   intptr_t length =  (map == NULL) ? 0 : map->count();
   Dart_Handle import_map = Dart_NewList(length * 2);
   for (intptr_t i = 0; i < length; i++) {
     ASSERT(map->GetArgument(i) != NULL);
     char* name = strdup(map->GetArgument(i));
     ASSERT(name != NULL);
     char* map_name = strchr(name, ',');
     intptr_t index = i * 2;
     if (map_name != NULL) {
       *map_name = '\0';
       map_name += 1;
       Dart_ListSetAt(import_map, index, Dart_NewString(name));
       Dart_ListSetAt(import_map, (index + 1), Dart_NewString(map_name));
     } else {
       Dart_ListSetAt(import_map, index, Dart_NewString(name));
       Dart_ListSetAt(import_map, (index + 1), Dart_NewString(""));
     }
     free(name);
   }
   return import_map;
 }
 
 
+#define CHECK_RESULT(result)                                                   \
+  if (Dart_IsError(result)) {                                                  \
+    *error = strdup(Dart_GetError(result));                                    \
+    Dart_ExitScope();                                                          \
+    Dart_ShutdownIsolate();                                                    \
+    return false;                                                              \
+  }                                                                            \
+
+
 // Returns true on success, false on failure.
 static bool CreateIsolateAndSetupHelper(const char* script_uri,
                                         const char* main,
                                         bool resolve_script,
                                         void* data,
                                         char** error) {
   Dart_Isolate isolate =
       Dart_CreateIsolate(script_uri, main, snapshot_buffer, data, error);
   if (isolate == NULL) {
     return false;
   }
 
   Dart_EnterScope();
 
   if (snapshot_buffer != NULL) {
     // Setup the native resolver as the snapshot does not carry it.
     Builtin::SetupLibrary(Builtin::LoadLibrary(Builtin::kBuiltinLibrary),
                           Builtin::kBuiltinLibrary);
     Builtin::SetupLibrary(Builtin::LoadLibrary(Builtin::kIOLibrary),
                           Builtin::kIOLibrary);
   }
 
   // Set up the library tag handler for this isolate.
-  Dart_Handle result = Dart_SetLibraryTagHandler(LibraryTagHandler);
-  if (Dart_IsError(result)) {
-    *error = strdup(Dart_GetError(result));
-    Dart_ExitScope();
-    Dart_ShutdownIsolate();
-    return false;
-  }
+  Dart_Handle result = Dart_SetLibraryTagHandler(DartUtils::LibraryTagHandler);
+  CHECK_RESULT(result);
 
   // Set up the import map for this isolate.
   result = Dart_SetImportMap(CreateImportMap(import_map_options));
-  if (Dart_IsError(result)) {
-    *error = strdup(Dart_GetError(result));
-    Dart_ExitScope();
-    Dart_ShutdownIsolate();
-    return false;
-  }
+  CHECK_RESULT(result);
 
-  // Prepare builtin and its dependent libraries for use to resolve URIs.
-  Dart_Handle uri_lib = Builtin::LoadLibrary(Builtin::kUriLibrary);
-  if (Dart_IsError(uri_lib)) {
-    *error = strdup(Dart_GetError(uri_lib));
-    Dart_ExitScope();
-    Dart_ShutdownIsolate();
-    return false;
-  }
-  Dart_Handle builtin_lib = Builtin::LoadLibrary(Builtin::kBuiltinLibrary);
-  if (Dart_IsError(builtin_lib)) {
-    *error = strdup(Dart_GetError(builtin_lib));
-    Dart_ExitScope();
-    Dart_ShutdownIsolate();
-    return false;
-  }
-
-  // Setup the corelib 'print' function.
-  Dart_Handle print =
-      Dart_Invoke(builtin_lib, Dart_NewString("_getPrintClosure"), 0, 0);
-  Dart_Handle coreimpl = Dart_LookupLibrary(Dart_NewString("dart:coreimpl"));
-  Dart_Handle print_impl =
-      Dart_GetClass(coreimpl, Dart_NewString("PrintImplementation"));
-  Dart_SetField(print_impl, Dart_NewString("_printClosure"), print);
-
-  // Setup the IO library.
-  Dart_Handle io_lib = Builtin::LoadLibrary(Builtin::kIOLibrary);
-  Builtin::SetupIOLibrary(io_lib);
-
-  if (package_root != NULL) {
-    const int kNumArgs = 1;
-    Dart_Handle dart_args[kNumArgs];
-
-    Dart_Handle handle = Dart_NewString(package_root);
-    if (Dart_IsError(handle)) {
-      *error = strdup(Dart_GetError(handle));
+  // Load the specified application script into the newly created isolate.
+  Dart_Handle library;
+  if (use_script_snapshot) {
+    if (snapshot_file == NULL) {
+      *error = strdup("Invalid script snapshot file name specified");

[Ivan Posva, 2012/08/31 23:20:51]

You will end up here if you ever spawn another isolate from a script snapshotted
isolate.

[siva, 2012/09/01 00:25:44]

I have set use_script_snapshot to false after the initial load this would
prevent any further uses of the script snapshot.

SpawnUri should definitely not be using the snapshot but we need to figure out
how SpawnFunction should work. Maybe it should not call back and instead clone
the script object from the spawning isolate. I have filed a bug (4880) for this.


On 2012/08/31 23:20:51, Ivan Posva wrote:

       Dart_ExitScope();
       Dart_ShutdownIsolate();
       return false;
     }
-    dart_args[0] = handle;
-
-    Dart_Handle result = Dart_Invoke(builtin_lib,
-        Dart_NewString("_setPackageRoot"), kNumArgs, dart_args);
-    if (Dart_IsError(result)) {
-      *error = strdup(Dart_GetError(result));
+    size_t len = snapshot_file->Length();
+    uint8_t* buffer = reinterpret_cast<uint8_t*>(malloc(len));
+    if (buffer == NULL) {
+      delete snapshot_file;
+      snapshot_file = NULL;
+      *error = strdup("Unable to read contents of script snapshot file");
       Dart_ExitScope();
       Dart_ShutdownIsolate();
       return false;
     }
+    snapshot_file->ReadFully(buffer, len);
+    library = Dart_LoadScriptFromSnapshot(buffer);
+    free(buffer);
+    delete snapshot_file;
+    snapshot_file = NULL;
+  } else {
+    // Prepare builtin and its dependent libraries for use to resolve URIs.
+    Dart_Handle uri_lib = Builtin::LoadLibrary(Builtin::kUriLibrary);
+    CHECK_RESULT(uri_lib);
+    Dart_Handle builtin_lib = Builtin::LoadLibrary(Builtin::kBuiltinLibrary);
+    CHECK_RESULT(builtin_lib);
+
+    // Setup the corelib 'print' function.
+    Dart_Handle print =
+        Dart_Invoke(builtin_lib, Dart_NewString("_getPrintClosure"), 0, 0);

[Ivan Posva, 2012/08/31 23:20:51]

This also needs to happen when setting up the isolate for the script snapshot.

[siva, 2012/09/01 00:25:44]

Good point. Move this code to dartutils and call here and in gensnapshot

On 2012/08/31 23:20:51, Ivan Posva wrote:

+    Dart_Handle coreimpl = Dart_LookupLibrary(Dart_NewString("dart:coreimpl"));
+    Dart_Handle print_impl =
+        Dart_GetClass(coreimpl, Dart_NewString("PrintImplementation"));
+    Dart_SetField(print_impl, Dart_NewString("_printClosure"), print);
+
+    // Setup the IO library.
+    Dart_Handle io_lib = Builtin::LoadLibrary(Builtin::kIOLibrary);
+    Builtin::SetupIOLibrary(io_lib);
+
+    if (package_root != NULL) {
+      const int kNumArgs = 1;
+      Dart_Handle dart_args[kNumArgs];
+
+      Dart_Handle handle = Dart_NewString(package_root);
+      CHECK_RESULT(handle);
+      dart_args[0] = handle;
+      Dart_Handle result = Dart_Invoke(
+          builtin_lib, Dart_NewString("_setPackageRoot"), kNumArgs, dart_args);
+      CHECK_RESULT(result);
+    }
+    library = DartUtils::LoadScript(script_uri, resolve_script, builtin_lib);
   }
-
-  // Load the specified application script into the newly created isolate.
-  Dart_Handle library = LoadScript(script_uri, resolve_script, builtin_lib);
-  if (Dart_IsError(library)) {
-    *error = strdup(Dart_GetError(library));
-    Dart_ExitScope();
-    Dart_ShutdownIsolate();
-    return false;
-  }
+  CHECK_RESULT(library);
   if (!Dart_IsLibrary(library)) {
     char errbuf[256];
     snprintf(errbuf, sizeof(errbuf),
              "Expected a library when loading script: %s",
              script_uri);
     *error = strdup(errbuf);
     Dart_ExitScope();
     Dart_ShutdownIsolate();
     return false;
   }
@@ skipping 75 matching lines @@
 
 
 static int ErrorExit(const char* format, ...) {
   va_list arguments;
   va_start(arguments, format);
   vfprintf(stderr, format, arguments);
   va_end(arguments);
 
   Dart_ExitScope();
   Dart_ShutdownIsolate();
-  free(const_cast<char*>(original_working_directory));
 
   return kErrorExitCode;
 }
 
 
 static void ShutdownIsolate(void* callback_data) {
   IsolateData* isolate_data = reinterpret_cast<IsolateData*>(callback_data);
   EventHandler* handler = isolate_data->event_handler;
   if (handler != NULL) handler->Shutdown();
   delete isolate_data;
@@ skipping 33 matching lines @@
     }
   }
 
   Dart_SetVMFlags(vm_options.count(), vm_options.arguments());
 
   // Initialize the Dart VM.
   Dart_Initialize(CreateIsolateAndSetup,
                   NULL,
                   ShutdownIsolate);
 
-  original_working_directory = Directory::Current();
+  DartUtils::SetOriginalWorkingDirectory();
 
   // Call CreateIsolateAndSetup which creates an isolate and loads up
   // the specified application script.
   char* error = NULL;
   char* isolate_name = BuildIsolateName(script_name, "main");
   if (!CreateIsolateAndSetupHelper(script_name,
                                    "main",
                                    true,  // Canonicalize the script name.
                                    new IsolateData(),
                                    &error)) {
     fprintf(stderr, "%s\n", error);
-    free(const_cast<char*>(original_working_directory));
     free(error);
     delete [] isolate_name;
     return kErrorExitCode;  // Indicates we encountered an error.
   }
   delete [] isolate_name;
 
   Dart_Isolate isolate = Dart_CurrentIsolate();
   ASSERT(isolate != NULL);
   Dart_Handle result;
 
@@ skipping 46 matching lines @@
   }
 
   Dart_ExitScope();
   // Dump symbol information for the profiler.
   DumpPprofSymbolInfo();
   // Shutdown the isolate.
   Dart_ShutdownIsolate();
   // Terminate process exit-code handler.
   Process::TerminateExitCodeHandler();
 
-  free(const_cast<char*>(original_working_directory));
-
   return 0;
 }