Add Dart_Save/LoadCompilationTrace.

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_tools_api.h"
@@ skipping 66 matching lines @@
   kScript,
   kAppAOT,
   kAppJIT,
 };
 static SnapshotKind gen_snapshot_kind = kNone;
 static const char* snapshot_deps_filename = NULL;
 #if !defined(DART_PRECOMPILED_RUNTIME)
 DFE dfe;
 #endif
 
+// Value of the --save-compilation-trace flag.
+// (This pointer points into an argv buffer and does not need to be
+// free'd.)
+static const char* save_compilation_trace_filename = NULL;
+
+// Value of the --load-compilation-trace flag.
+// (This pointer points into an argv buffer and does not need to be
+// free'd.)
+static const char* load_compilation_trace_filename = NULL;
+
 // Value of the --save-feedback flag.
 // (This pointer points into an argv buffer and does not need to be
 // free'd.)
 static const char* save_feedback_filename = NULL;
 
 // Value of the --load-feedback flag.
 // (This pointer points into an argv buffer and does not need to be
 // free'd.)
 static const char* load_feedback_filename = NULL;
 
@@ skipping 103 matching lines @@
                                   CommandLineOptions* vm_options) {
   ASSERT(arg != NULL);
   if (*arg == '-') {
     return false;
   }
   commandline_packages_file = arg;
   return true;
 }
 
 
+static bool ProcessSaveCompilationTraceOption(const char* arg,
+                                              CommandLineOptions* vm_options) {
+  ASSERT(arg != NULL);
+  if (*arg == '-') {
+    return false;
+  }
+  save_compilation_trace_filename = arg;
+  return true;
+}
+
+
+static bool ProcessLoadCompilationTraceOption(const char* arg,
+                                              CommandLineOptions* vm_options) {
+  ASSERT(arg != NULL);
+  if (*arg == '-') {
+    return false;
+  }
+  load_compilation_trace_filename = arg;
+  return true;
+}
+
+
 static bool ProcessSaveFeedbackOption(const char* arg,
                                       CommandLineOptions* vm_options) {
   ASSERT(arg != NULL);
   if (*arg == '-') {
     return false;
   }
   save_feedback_filename = arg;
   return true;
 }
 
@@ skipping 368 matching lines @@
     {"--dfe=", ProcessFrontendOption},
     {"--platform=", ProcessPlatformOption},
 #endif
     {"--enable-vm-service", ProcessEnableVmServiceOption},
     {"--disable-service-origin-check", ProcessDisableServiceOriginCheckOption},
     {"--observe", ProcessObserveOption},
     {"--snapshot=", ProcessSnapshotFilenameOption},
     {"--snapshot-kind=", ProcessSnapshotKindOption},
     {"--snapshot-depfile=", ProcessSnapshotDepsFilenameOption},
     {"--use-blobs", ProcessUseBlobsOption},
+    {"--save-compilation-trace=", ProcessSaveCompilationTraceOption},
+    {"--load-compilation-trace=", ProcessLoadCompilationTraceOption},
     {"--save-feedback=", ProcessSaveFeedbackOption},
     {"--load-feedback=", ProcessLoadFeedbackOption},
     {"--trace-loading", ProcessTraceLoadingOption},
     {"--hot-reload-test-mode", ProcessHotReloadTestModeOption},
     {"--hot-reload-rollback-test-mode", ProcessHotReloadRollbackTestModeOption},
     {"--short_socket_read", ProcessShortSocketReadOption},
     {"--short_socket_write", ProcessShortSocketWriteOption},
 #if !defined(HOST_OS_MACOS)
     {"--root-certs-file=", ProcessRootCertsFileOption},
     {"--root-certs-cache=", ProcessRootCertsCacheOption},
@@ skipping 789 matching lines @@
   if (file == NULL) {
     ErrorExit(kErrorExitCode, "Unable to open file %s\n", filename);
   }
   if (!file->WriteFully(buffer, size)) {
     ErrorExit(kErrorExitCode, "Unable to write file %s\n", filename);
   }
   file->Release();
 }
 
 
+static void ReadFile(const char* filename, uint8_t** buffer, intptr_t* size) {
+  File* file = File::Open(filename, File::kRead);
+  if (file == NULL) {
+    ErrorExit(kErrorExitCode, "Unable to open file %s\n", filename);
+  }
+  *size = file->Length();
+  *buffer = reinterpret_cast<uint8_t*>(malloc(*size));
+  if (!file->ReadFully(*buffer, *size)) {
+    ErrorExit(kErrorExitCode, "Unable to read file %s\n", filename);
+  }
+  file->Release();
+}
+
+
 bool RunMainIsolate(const char* script_name, CommandLineOptions* dart_options) {
   // Call CreateIsolateAndSetup which creates an isolate and loads up
   // the specified application script.
   char* error = NULL;
   bool is_main_isolate = true;
   int exit_code = 0;
   char* isolate_name = BuildIsolateName(script_name, "main");
   Dart_Isolate isolate = CreateIsolateAndSetupHelper(
       is_main_isolate, script_name, "main", commandline_package_root,
       commandline_packages_file, NULL, &error, &exit_code);
@@ skipping 123 matching lines @@
     }
 
     if (gen_snapshot_kind == kAppAOT) {
       GenerateAppAOTSnapshot();
     } else {
       if (Dart_IsNull(root_lib)) {
         ErrorExit(kErrorExitCode, "Unable to find root library for '%s'\n",
                   script_name);
       }
 
-      if (gen_snapshot_kind == kAppJIT) Dart_SortClasses();
+      if (gen_snapshot_kind == kAppJIT) {
+        result = Dart_SortClasses();
+        CHECK_RESULT(result);
+      }
+
+      if (load_compilation_trace_filename != NULL) {
+        uint8_t* buffer = NULL;
+        intptr_t size = 0;
+        ReadFile(load_compilation_trace_filename, &buffer, &size);
+        result = Dart_LoadCompilationTrace(buffer, size);
+        CHECK_RESULT(result);
+      }
 
       // The helper function _getMainClosure creates a closure for the main
       // entry point which is either explicitly or implictly exported from the
       // root library.
       Dart_Handle main_closure =
           Dart_Invoke(isolate_data->builtin_lib(),
                       Dart_NewStringFromCString("_getMainClosure"), 0, NULL);
       CHECK_RESULT(main_closure);
 
       // Call _startIsolate in the isolate library to enable dispatching the
@@ skipping 17 matching lines @@
         if (!Dart_IsCompilationError(result)) {
           Snapshot::GenerateAppJIT(snapshot_filename);
         }
       }
       CHECK_RESULT(result);
 
       if (save_feedback_filename != NULL) {
         uint8_t* buffer = NULL;
         intptr_t size = 0;
         result = Dart_SaveJITFeedback(&buffer, &size);
-        if (Dart_IsError(result)) {
-          ErrorExit(kErrorExitCode, "%s\n", Dart_GetError(result));
-        }
+        CHECK_RESULT(result);
         WriteFile(save_feedback_filename, buffer, size);
       }
+
+      if (save_compilation_trace_filename != NULL) {
+        uint8_t* buffer = NULL;
+        intptr_t size = 0;
+        result = Dart_SaveCompilationTrace(&buffer, &size);
+        CHECK_RESULT(result);
+        WriteFile(save_compilation_trace_filename, buffer, size);
+      }
     }
   }
 
   if (snapshot_deps_filename != NULL) {
     Loader::ResolveDependenciesAsFilePaths();
     IsolateData* isolate_data =
         reinterpret_cast<IsolateData*>(Dart_IsolateData(isolate));
     ASSERT(isolate_data != NULL);
     MallocGrowableArray<char*>* dependencies = isolate_data->dependencies();
     ASSERT(dependencies != NULL);
@@ skipping 272 matching lines @@
   Platform::Exit(Process::GlobalExitCode());
 }
 
 }  // namespace bin
 }  // namespace dart
 
 int main(int argc, char** argv) {
   dart::bin::main(argc, argv);
   UNREACHABLE();
 }