Ben Murdoch | 4a90d5f | 2016-03-22 12:00:34 +0000 | [diff] [blame] | 1 | // Copyright 2015 the V8 project authors. All rights reserved. |
| 2 | // Use of this source code is governed by a BSD-style license that can be |
| 3 | // found in the LICENSE file. |
| 4 | |
| 5 | #include "src/startup-data-util.h" |
| 6 | |
| 7 | #include <stdlib.h> |
| 8 | #include <string.h> |
| 9 | |
| 10 | #include "src/base/logging.h" |
| 11 | #include "src/base/platform/platform.h" |
| 12 | #include "src/utils.h" |
| 13 | |
| 14 | |
| 15 | namespace v8 { |
| 16 | namespace internal { |
| 17 | |
| 18 | #ifdef V8_USE_EXTERNAL_STARTUP_DATA |
| 19 | |
| 20 | namespace { |
| 21 | |
| 22 | v8::StartupData g_natives; |
| 23 | v8::StartupData g_snapshot; |
| 24 | |
| 25 | |
| 26 | void ClearStartupData(v8::StartupData* data) { |
| 27 | data->data = nullptr; |
| 28 | data->raw_size = 0; |
| 29 | } |
| 30 | |
| 31 | |
| 32 | void DeleteStartupData(v8::StartupData* data) { |
| 33 | delete[] data->data; |
| 34 | ClearStartupData(data); |
| 35 | } |
| 36 | |
| 37 | |
| 38 | void FreeStartupData() { |
| 39 | DeleteStartupData(&g_natives); |
| 40 | DeleteStartupData(&g_snapshot); |
| 41 | } |
| 42 | |
| 43 | |
| 44 | void Load(const char* blob_file, v8::StartupData* startup_data, |
| 45 | void (*setter_fn)(v8::StartupData*)) { |
| 46 | ClearStartupData(startup_data); |
| 47 | |
| 48 | CHECK(blob_file); |
| 49 | |
| 50 | FILE* file = fopen(blob_file, "rb"); |
| 51 | if (!file) { |
| 52 | PrintF(stderr, "Failed to open startup resource '%s'.\n", blob_file); |
| 53 | return; |
| 54 | } |
| 55 | |
| 56 | fseek(file, 0, SEEK_END); |
| 57 | startup_data->raw_size = static_cast<int>(ftell(file)); |
| 58 | rewind(file); |
| 59 | |
| 60 | startup_data->data = new char[startup_data->raw_size]; |
| 61 | int read_size = static_cast<int>(fread(const_cast<char*>(startup_data->data), |
| 62 | 1, startup_data->raw_size, file)); |
| 63 | fclose(file); |
| 64 | |
| 65 | if (startup_data->raw_size == read_size) { |
| 66 | (*setter_fn)(startup_data); |
| 67 | } else { |
| 68 | PrintF(stderr, "Corrupted startup resource '%s'.\n", blob_file); |
| 69 | } |
| 70 | } |
| 71 | |
| 72 | |
| 73 | void LoadFromFiles(const char* natives_blob, const char* snapshot_blob) { |
| 74 | Load(natives_blob, &g_natives, v8::V8::SetNativesDataBlob); |
| 75 | Load(snapshot_blob, &g_snapshot, v8::V8::SetSnapshotDataBlob); |
| 76 | |
| 77 | atexit(&FreeStartupData); |
| 78 | } |
| 79 | |
| 80 | |
| 81 | char* RelativePath(char** buffer, const char* exec_path, const char* name) { |
| 82 | DCHECK(exec_path); |
| 83 | int path_separator = static_cast<int>(strlen(exec_path)) - 1; |
| 84 | while (path_separator >= 0 && |
| 85 | !base::OS::isDirectorySeparator(exec_path[path_separator])) { |
| 86 | path_separator--; |
| 87 | } |
| 88 | if (path_separator >= 0) { |
| 89 | int name_length = static_cast<int>(strlen(name)); |
| 90 | *buffer = |
| 91 | reinterpret_cast<char*>(calloc(path_separator + name_length + 2, 1)); |
| 92 | *buffer[0] = '\0'; |
| 93 | strncat(*buffer, exec_path, path_separator + 1); |
| 94 | strncat(*buffer, name, name_length); |
| 95 | } else { |
| 96 | *buffer = strdup(name); |
| 97 | } |
| 98 | return *buffer; |
| 99 | } |
| 100 | |
| 101 | } // namespace |
| 102 | #endif // V8_USE_EXTERNAL_STARTUP_DATA |
| 103 | |
| 104 | |
| 105 | void InitializeExternalStartupData(const char* directory_path) { |
| 106 | #ifdef V8_USE_EXTERNAL_STARTUP_DATA |
| 107 | char* natives; |
| 108 | char* snapshot; |
| 109 | LoadFromFiles(RelativePath(&natives, directory_path, "natives_blob.bin"), |
Ben Murdoch | c561043 | 2016-08-08 18:44:38 +0100 | [diff] [blame^] | 110 | RelativePath(&snapshot, directory_path, "snapshot_blob.bin")); |
Ben Murdoch | 4a90d5f | 2016-03-22 12:00:34 +0000 | [diff] [blame] | 111 | free(natives); |
| 112 | free(snapshot); |
| 113 | #endif // V8_USE_EXTERNAL_STARTUP_DATA |
| 114 | } |
| 115 | |
| 116 | |
| 117 | void InitializeExternalStartupData(const char* natives_blob, |
| 118 | const char* snapshot_blob) { |
| 119 | #ifdef V8_USE_EXTERNAL_STARTUP_DATA |
| 120 | LoadFromFiles(natives_blob, snapshot_blob); |
| 121 | #endif // V8_USE_EXTERNAL_STARTUP_DATA |
| 122 | } |
| 123 | |
| 124 | } // namespace internal |
| 125 | } // namespace v8 |