Infrastructure for reading V8's initial snapshot from external files

gin/isolate_holder.cc

 // Copyright 2013 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 "gin/public/isolate_holder.h"
 
 #include <stdlib.h>
 #include <string.h>
 
 #include "base/logging.h"
 #include "base/message_loop/message_loop.h"
 #include "base/rand_util.h"
 #include "base/sys_info.h"
 #include "gin/array_buffer.h"
 #include "gin/debug_impl.h"
 #include "gin/function_template.h"
 #include "gin/per_isolate_data.h"
 #include "gin/public/v8_platform.h"
 #include "gin/run_microtasks_observer.h"
 
+#ifdef V8_USE_EXTERNAL_STARTUP_DATA
+#include "base/files/memory_mapped_file.h"
+#include "base/path_service.h"
+#ifdef OS_MACOSX
+#include "base/mac/bundle_locations.h"
+#endif  // OS_MACOSX
+#endif  // V8_USE_EXTERNAL_STARTUP_DATA
+
 namespace gin {
 
 namespace {
 
 v8::ArrayBuffer::Allocator* g_array_buffer_allocator = NULL;
 
 bool GenerateEntropy(unsigned char* buffer, size_t amount) {
   base::RandBytes(buffer, amount);
   return true;
 }
 
+#ifdef V8_USE_EXTERNAL_STARTUP_DATA
+base::MemoryMappedFile* g_mapped_natives = NULL;
+base::MemoryMappedFile* g_mapped_snapshot = NULL;
+#endif
+
 }  // namespace
 
+
+#ifdef V8_USE_EXTERNAL_STARTUP_DATA
+//static
+bool IsolateHolder::MapV8Files(
+  base::FilePath* natives_path, base::FilePath* snapshot_path,
+  int natives_fd, int snapshot_fd) {
+
+  int flags = base::File::FLAG_OPEN | base::File::FLAG_READ;
+
+  g_mapped_natives = new base::MemoryMappedFile;
+  if (!g_mapped_natives->IsValid()) {
+#ifdef OS_ANDROID
+    if (natives_fd == -1
+        ? !g_mapped_natives->Initialize(base::File(*natives_path, flags))
+        : !g_mapped_natives->Initialize(base::File(natives_fd))
+#else
+    if (!g_mapped_natives->Initialize(base::File(*natives_path, flags))
+#endif
+      ) {
+      delete g_mapped_natives;
+      g_mapped_natives = NULL;
+      LOG(ERROR) << "Couldn't mmap v8 natives data file";
+      return false;
+    }
+  }
+
+  g_mapped_snapshot = new base::MemoryMappedFile;
+  if (!g_mapped_snapshot->IsValid()) {
+#ifdef OS_ANDROID
+    if (snapshot_fd == -1
+        ? !g_mapped_snapshot->Initialize(base::File(*snapshot_path, flags))
+        : !g_mapped_snapshot->Initialize(base::File(snapshot_fd))
+#else
+    if (!g_mapped_snapshot->Initialize(base::File(*snapshot_path, flags))
+#endif
+      ) {
+      delete g_mapped_snapshot;
+      g_mapped_snapshot = NULL;
+      LOG(ERROR) << "Couldn't mmap v8 snapshot data file";
+      return false;
+    }
+  }
+
+  return true;
+}
+
+// static
+bool IsolateHolder::LoadV8Snapshot() {
+  if (g_mapped_natives && g_mapped_snapshot)
+    return true;
+
+#if !defined(OS_MACOSX)
+  base::FilePath data_path;
+  PathService::Get(
+#if defined(OS_ANDROID)
+    base::DIR_ANDROID_APP_DATA,
+#elif defined(OS_POSIX) || defined(OS_WIN)
+    base::DIR_EXE,
+#endif
+    &data_path);
+#else
+  base::FilePath data_path = base::mac::FrameworkBundlePath();
+#endif
+  DCHECK(!data_path.empty());
+
+  base::FilePath natives_path = data_path.AppendASCII("natives_blob.bin");
+  base::FilePath snapshot_path = data_path.AppendASCII("snapshot_blob.bin");
+
+  return IsolateHolder::MapV8Files(&natives_path, &snapshot_path);
+}
+#endif  // V8_USE_EXTERNAL_STARTUP_DATA
+
+#if defined(V8_USE_EXTERNAL_STARTUP_DATA) && defined(OS_ANDROID)
+//static
+bool IsolateHolder::LoadV8SnapshotFD(int natives_fd, int snapshot_fd) {
+  if (g_mapped_natives && g_mapped_snapshot)
+    return true;
+
+  return IsolateHolder::MapV8Files(NULL, NULL, natives_fd, snapshot_fd);
+}
+#endif  // V8_USE_EXTERNAL_STARTUP_DATA && OS_ANDROID
+
 IsolateHolder::IsolateHolder() {
   CHECK(g_array_buffer_allocator)
       << "You need to invoke gin::IsolateHolder::Initialize first";
   v8::Isolate::CreateParams params;
   params.entry_hook = DebugImpl::GetFunctionEntryHook();
   params.code_event_handler = DebugImpl::GetJitCodeEventHandler();
   params.constraints.ConfigureDefaults(base::SysInfo::AmountOfPhysicalMemory(),
                                        base::SysInfo::AmountOfVirtualMemory(),
                                        base::SysInfo::NumberOfProcessors());
   isolate_ = v8::Isolate::New(params);
@@ skipping 37 matching lines @@
   if (v8_is_initialized)
     return;
   v8::V8::InitializePlatform(V8Platform::Get());
   v8::V8::SetArrayBufferAllocator(allocator);
   g_array_buffer_allocator = allocator;
   if (mode == gin::IsolateHolder::kStrictMode) {
     static const char v8_flags[] = "--use_strict";
     v8::V8::SetFlagsFromString(v8_flags, sizeof(v8_flags) - 1);
   }
   v8::V8::SetEntropySource(&GenerateEntropy);
+#ifdef V8_USE_EXTERNAL_STARTUP_DATA
+  v8::StartupData natives;
+  natives.data = reinterpret_cast<const char*>(g_mapped_natives->data());
+  natives.raw_size = g_mapped_natives->length();
+  natives.compressed_size = g_mapped_natives->length();
+  v8::V8::SetNativesDataBlob(&natives);
+
+  v8::StartupData snapshot;
+  snapshot.data = reinterpret_cast<const char*>(g_mapped_snapshot->data());
+  snapshot.raw_size = g_mapped_snapshot->length();
+  snapshot.compressed_size = g_mapped_snapshot->length();
+  v8::V8::SetSnapshotDataBlob(&snapshot);
+#endif  // V8_USE_EXTERNAL_STARTUP_DATA
   v8::V8::Initialize();
   v8_is_initialized = true;
 }
 
 void IsolateHolder::AddRunMicrotasksObserver() {
   DCHECK(!task_observer_.get());
   task_observer_.reset(new RunMicrotasksObserver(isolate_));;
   base::MessageLoop::current()->AddTaskObserver(task_observer_.get());
 }
 
 void IsolateHolder::RemoveRunMicrotasksObserver() {
   DCHECK(task_observer_.get());
   base::MessageLoop::current()->RemoveTaskObserver(task_observer_.get());
   task_observer_.reset();
 }
 
 }  // namespace gin