mojo/dart/dart_snapshotter/main.cc

// Copyright 2015 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include <iostream>

#include "base/at_exit.h"
#include "base/basictypes.h"
#include "base/command_line.h"
#include "base/files/file_path.h"
#include "base/files/file_util.h"
#include "base/logging.h"
#include "base/process/memory.h"
#include "dart/runtime/include/dart_api.h"
#include "mojo/dart/dart_snapshotter/vm.h"
#include "mojo/edk/embedder/embedder.h"
#include "mojo/edk/embedder/simple_platform_support.h"
#include "tonic/dart_error.h"
#include "tonic/dart_isolate_scope.h"
#include "tonic/dart_library_loader.h"
#include "tonic/dart_library_provider_files.h"
#include "tonic/dart_script_loader_sync.h"
#include "tonic/dart_state.h"

const char kHelp[] = "help";
const char kPackageRoot[] = "package-root";
const char kSnapshot[] = "snapshot";

const uint8_t magic_number[] = { 0xf5, 0xf5, 0xdc, 0xdc };

void Usage() {
  std::cerr << "Usage: dart_snapshotter"
            << " --" << kPackageRoot << " --" << kSnapshot
            << " <dart-app>" << std::endl;
}

void WriteSnapshot(base::FilePath path) {
  uint8_t* buffer;
  intptr_t size;
  CHECK(!tonic::LogIfError(Dart_CreateScriptSnapshot(&buffer, &size)));

  intptr_t magic_number_len = sizeof(magic_number);
  CHECK_EQ(base::WriteFile(
      path, reinterpret_cast<const char*>(magic_number), sizeof(magic_number)),
      magic_number_len);
  CHECK(base::AppendToFile(
      path, reinterpret_cast<const char*>(buffer), size));
}

int main(int argc, const char* argv[]) {
  base::AtExitManager exit_manager;
  base::EnableTerminationOnHeapCorruption();
  base::CommandLine::Init(argc, argv);

  const base::CommandLine& command_line =
      *base::CommandLine::ForCurrentProcess();

  if (command_line.HasSwitch(kHelp) || command_line.GetArgs().empty()) {
    Usage();
    return 0;
  }

  // Initialize mojo.
  mojo::embedder::Init(
      make_scoped_ptr(new mojo::embedder::SimplePlatformSupport()));

  InitDartVM();

  CHECK(command_line.HasSwitch(kPackageRoot)) << "Need --package-root";
  CHECK(command_line.HasSwitch(kSnapshot)) << "Need --snapshot";
  auto args = command_line.GetArgs();
  CHECK(args.size() == 1);

  Dart_Isolate isolate = CreateDartIsolate();
  CHECK(isolate);

  tonic::DartIsolateScope scope(isolate);
  tonic::DartApiScope api_scope;

  auto isolate_data = SnapshotterDartState::Current();
  CHECK(isolate_data != nullptr);

  // Use tonic's library tag handler.
  CHECK(!tonic::LogIfError(Dart_SetLibraryTagHandler(
            tonic::DartLibraryLoader::HandleLibraryTag)));

  // Use tonic's file system library provider.
  isolate_data->set_library_provider(
      new tonic::DartLibraryProviderFiles(
          command_line.GetSwitchValuePath(kPackageRoot)));

  // Load script.
  tonic::DartScriptLoaderSync::LoadScript(args[0],
                                          isolate_data->library_provider());

  // Write snapshot.
  WriteSnapshot(command_line.GetSwitchValuePath(kSnapshot));

  return 0;
}