Implement automatic loading of dart:core_native_fields library

bin/gen_snapshot.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.
 
 // Generate a snapshot file after loading all the scripts specified on the
 // command line.
 
 #include <stdlib.h>
 #include <string.h>
 #include <stdio.h>
 
 #include "include/dart_api.h"
 
 #include "bin/builtin.h"
+#include "bin/dartutils.h"
 #include "bin/file.h"
 #include "bin/globals.h"
 #include "bin/process_script.h"
 
 // Global state that indicates whether a snapshot is to be created and
 // if so which file to write the snapshot into.
 static const char* snapshot_filename = NULL;
 
 // Global state that captures the URL mappings specified on the command line.
 static CommandLineOptions* url_mapping = NULL;
@@ skipping 188 matching lines @@
   Dart_Handle source = ReadStringFromFile(script_name);
   if (Dart_IsError(source)) {
     return source;  // source contains the error string.
   }
   Dart_Handle url = Dart_NewString(script_name);
 
   return Dart_LoadScript(url, source, CreateSnapshotLibraryTagHandler);
 }
 
 
+static Dart_Handle BuiltinSnapshotLibraryTagHandler(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_chars = NULL;
+  Dart_Handle result = Dart_StringToCString(url, &url_chars);
+  if (Dart_IsError(result)) {
+    return Dart_Error("accessing url characters failed");
+  }
+  // We only support canonicalization of "dart:".
+  static const char* kDartScheme = "dart:";
+  static const intptr_t kDartSchemeLen = strlen(kDartScheme);
+  if (strncmp(url_chars, kDartScheme, kDartSchemeLen) == 0) {
+    if (tag == kCanonicalizeUrl) {
+      return url;
+    }
+    return Dart_Error("unexpected tag encountered %d", tag);
+  }
+  return Dart_Error("unsupported url encountered %s", url_chars);
+}
+
+
+static Dart_Handle LoadGenericSnapshotCreationScript() {
+  Dart_Handle source = Builtin_Source();
+  if (Dart_IsError(source)) {
+    return source;  // source contains the error string.
+  }
+  Dart_Handle url = Dart_NewString(DartUtils::kBuiltinLibURL);
+  Dart_Handle lib = Dart_LoadScript(url,
+                                    source,
+                                    BuiltinSnapshotLibraryTagHandler);
+  if (!Dart_IsError(lib)) {
+    Builtin_SetupLibrary(lib);
+  }
+  return lib;
+}
+
+
 static void* SnapshotCreateCallback(void* data) {
   const char* script_name = reinterpret_cast<const char*>(data);
   Dart_Handle result;
+  Dart_Handle library;
   Dart_EnterScope();
 
   ASSERT(snapshot_filename != NULL);
 
-  // If a file is specified on the command line, load it up before a snapshot
-  // is created.
+  // Load up the script before a snapshot is created.
   if (script_name != NULL) {
     // Load the specified script.
-    Dart_Handle library = LoadSnapshotCreationScript(script_name);
-    if (Dart_IsError(library)) {
-      const char* err_msg = Dart_GetError(library);
-      fprintf(stderr, "Errors encountered while loading script: %s\n", err_msg);
-      Dart_ExitScope();
-      exit(255);
-    }
-
-    if (!Dart_IsLibrary(library)) {
-      fprintf(stderr,
-              "Expected a library when loading script: %s",
-              script_name);
-      Dart_ExitScope();
-      exit(255);
-    }
+    library = LoadSnapshotCreationScript(script_name);
   } else {
-    // Implicitly load builtin library.
-    Builtin_LoadLibrary();
-    // Setup the native resolver for built in library functions.
-    Builtin_SetNativeResolver();
+    // This is a generic dart snapshot which needs builtin library setup.
+    library = LoadGenericSnapshotCreationScript();
   }
-
+  if (Dart_IsError(library)) {
+    const char* err_msg = Dart_GetError(library);
+    fprintf(stderr, "Errors encountered while loading script: %s\n", err_msg);
+    Dart_ExitScope();
+    exit(255);
+  }
+  ASSERT(Dart_IsLibrary(library));
   uint8_t* buffer = NULL;
   intptr_t size = 0;
   // First create the snapshot.
   result = Dart_CreateSnapshot(&buffer, &size);
   if (Dart_IsError(result)) {
     const char* err_msg = Dart_GetError(result);
     fprintf(stderr, "Error while creating snapshot: %s\n", err_msg);
     Dart_ExitScope();
     exit(255);
   }
@@ skipping 44 matching lines @@
   // and writes out a snapshot.
   Dart_Isolate isolate = Dart_CreateIsolate(NULL, script_name);
   if (isolate == NULL) {
     return 255;
   }
 
   // Shutdown the isolate.
   Dart_ShutdownIsolate();
   return 0;
 }