| // Copyright (c) 2017, 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 "bin/snapshot_utils.h" |
| |
| #include "bin/dartutils.h" |
| #include "bin/error_exit.h" |
| #include "bin/extensions.h" |
| #include "bin/file.h" |
| #include "bin/platform.h" |
| #include "include/dart_api.h" |
| #include "platform/utils.h" |
| |
| namespace dart { |
| namespace bin { |
| |
| extern const char* kVmSnapshotDataSymbolName; |
| extern const char* kVmSnapshotInstructionsSymbolName; |
| extern const char* kIsolateSnapshotDataSymbolName; |
| extern const char* kIsolateSnapshotInstructionsSymbolName; |
| |
| static const int64_t kAppSnapshotHeaderSize = 5 * kInt64Size; |
| static const int64_t kAppSnapshotPageSize = 4 * KB; |
| |
| class MappedAppSnapshot : public AppSnapshot { |
| public: |
| MappedAppSnapshot(MappedMemory* vm_snapshot_data, |
| MappedMemory* vm_snapshot_instructions, |
| MappedMemory* isolate_snapshot_data, |
| MappedMemory* isolate_snapshot_instructions) |
| : vm_data_mapping_(vm_snapshot_data), |
| vm_instructions_mapping_(vm_snapshot_instructions), |
| isolate_data_mapping_(isolate_snapshot_data), |
| isolate_instructions_mapping_(isolate_snapshot_instructions) {} |
| |
| ~MappedAppSnapshot() { |
| delete vm_data_mapping_; |
| delete vm_instructions_mapping_; |
| delete isolate_data_mapping_; |
| delete isolate_instructions_mapping_; |
| } |
| |
| void SetBuffers(const uint8_t** vm_data_buffer, |
| const uint8_t** vm_instructions_buffer, |
| const uint8_t** isolate_data_buffer, |
| const uint8_t** isolate_instructions_buffer) { |
| if (vm_data_mapping_ != NULL) { |
| *vm_data_buffer = |
| reinterpret_cast<const uint8_t*>(vm_data_mapping_->address()); |
| } |
| if (vm_instructions_mapping_ != NULL) { |
| *vm_instructions_buffer = |
| reinterpret_cast<const uint8_t*>(vm_instructions_mapping_->address()); |
| } |
| if (isolate_data_mapping_ != NULL) { |
| *isolate_data_buffer = |
| reinterpret_cast<const uint8_t*>(isolate_data_mapping_->address()); |
| } |
| if (isolate_instructions_mapping_ != NULL) { |
| *isolate_instructions_buffer = reinterpret_cast<const uint8_t*>( |
| isolate_instructions_mapping_->address()); |
| } |
| } |
| |
| private: |
| MappedMemory* vm_data_mapping_; |
| MappedMemory* vm_instructions_mapping_; |
| MappedMemory* isolate_data_mapping_; |
| MappedMemory* isolate_instructions_mapping_; |
| }; |
| |
| static AppSnapshot* TryReadAppSnapshotBlobs(const char* script_name) { |
| File* file = File::Open(NULL, script_name, File::kRead); |
| if (file == NULL) { |
| return NULL; |
| } |
| if (file->Length() < kAppSnapshotHeaderSize) { |
| file->Release(); |
| return NULL; |
| } |
| int64_t header[5]; |
| ASSERT(sizeof(header) == kAppSnapshotHeaderSize); |
| if (!file->ReadFully(&header, kAppSnapshotHeaderSize)) { |
| file->Release(); |
| return NULL; |
| } |
| ASSERT(sizeof(header[0]) == appjit_magic_number.length); |
| if (memcmp(&header[0], appjit_magic_number.bytes, |
| appjit_magic_number.length) != 0) { |
| file->Release(); |
| return NULL; |
| } |
| |
| int64_t vm_data_size = header[1]; |
| int64_t vm_data_position = |
| Utils::RoundUp(file->Position(), kAppSnapshotPageSize); |
| int64_t vm_instructions_size = header[2]; |
| int64_t vm_instructions_position = vm_data_position + vm_data_size; |
| if (vm_instructions_size != 0) { |
| vm_instructions_position = |
| Utils::RoundUp(vm_instructions_position, kAppSnapshotPageSize); |
| } |
| int64_t isolate_data_size = header[3]; |
| int64_t isolate_data_position = Utils::RoundUp( |
| vm_instructions_position + vm_instructions_size, kAppSnapshotPageSize); |
| int64_t isolate_instructions_size = header[4]; |
| int64_t isolate_instructions_position = |
| isolate_data_position + isolate_data_size; |
| if (isolate_instructions_size != 0) { |
| isolate_instructions_position = |
| Utils::RoundUp(isolate_instructions_position, kAppSnapshotPageSize); |
| } |
| |
| MappedMemory* vm_data_mapping = NULL; |
| if (vm_data_size != 0) { |
| vm_data_mapping = |
| file->Map(File::kReadOnly, vm_data_position, vm_data_size); |
| if (vm_data_mapping == NULL) { |
| FATAL1("Failed to memory map snapshot: %s\n", script_name); |
| } |
| } |
| |
| MappedMemory* vm_instr_mapping = NULL; |
| if (vm_instructions_size != 0) { |
| vm_instr_mapping = file->Map(File::kReadExecute, vm_instructions_position, |
| vm_instructions_size); |
| if (vm_instr_mapping == NULL) { |
| FATAL1("Failed to memory map snapshot: %s\n", script_name); |
| } |
| } |
| |
| MappedMemory* isolate_data_mapping = NULL; |
| if (isolate_data_size != 0) { |
| isolate_data_mapping = |
| file->Map(File::kReadOnly, isolate_data_position, isolate_data_size); |
| if (isolate_data_mapping == NULL) { |
| FATAL1("Failed to memory map snapshot: %s\n", script_name); |
| } |
| } |
| |
| MappedMemory* isolate_instr_mapping = NULL; |
| if (isolate_instructions_size != 0) { |
| isolate_instr_mapping = |
| file->Map(File::kReadExecute, isolate_instructions_position, |
| isolate_instructions_size); |
| if (isolate_instr_mapping == NULL) { |
| FATAL1("Failed to memory map snapshot: %s\n", script_name); |
| } |
| } |
| |
| file->Release(); |
| return new MappedAppSnapshot(vm_data_mapping, vm_instr_mapping, |
| isolate_data_mapping, isolate_instr_mapping); |
| } |
| |
| #if defined(DART_PRECOMPILED_RUNTIME) |
| class DylibAppSnapshot : public AppSnapshot { |
| public: |
| DylibAppSnapshot(void* library, |
| const uint8_t* vm_snapshot_data, |
| const uint8_t* vm_snapshot_instructions, |
| const uint8_t* isolate_snapshot_data, |
| const uint8_t* isolate_snapshot_instructions) |
| : library_(library), |
| vm_snapshot_data_(vm_snapshot_data), |
| vm_snapshot_instructions_(vm_snapshot_instructions), |
| isolate_snapshot_data_(isolate_snapshot_data), |
| isolate_snapshot_instructions_(isolate_snapshot_instructions) {} |
| |
| ~DylibAppSnapshot() { Extensions::UnloadLibrary(library_); } |
| |
| void SetBuffers(const uint8_t** vm_data_buffer, |
| const uint8_t** vm_instructions_buffer, |
| const uint8_t** isolate_data_buffer, |
| const uint8_t** isolate_instructions_buffer) { |
| *vm_data_buffer = vm_snapshot_data_; |
| *vm_instructions_buffer = vm_snapshot_instructions_; |
| *isolate_data_buffer = isolate_snapshot_data_; |
| *isolate_instructions_buffer = isolate_snapshot_instructions_; |
| } |
| |
| private: |
| void* library_; |
| const uint8_t* vm_snapshot_data_; |
| const uint8_t* vm_snapshot_instructions_; |
| const uint8_t* isolate_snapshot_data_; |
| const uint8_t* isolate_snapshot_instructions_; |
| }; |
| |
| static AppSnapshot* TryReadAppSnapshotDynamicLibrary(const char* script_name) { |
| void* library = Extensions::LoadExtensionLibrary(script_name); |
| if (library == NULL) { |
| return NULL; |
| } |
| |
| const uint8_t* vm_data_buffer = reinterpret_cast<const uint8_t*>( |
| Extensions::ResolveSymbol(library, kVmSnapshotDataSymbolName)); |
| if (vm_data_buffer == NULL) { |
| FATAL1("Failed to resolve symbol '%s'\n", kVmSnapshotDataSymbolName); |
| } |
| |
| const uint8_t* vm_instructions_buffer = reinterpret_cast<const uint8_t*>( |
| Extensions::ResolveSymbol(library, kVmSnapshotInstructionsSymbolName)); |
| if (vm_instructions_buffer == NULL) { |
| FATAL1("Failed to resolve symbol '%s'\n", |
| kVmSnapshotInstructionsSymbolName); |
| } |
| |
| const uint8_t* isolate_data_buffer = reinterpret_cast<const uint8_t*>( |
| Extensions::ResolveSymbol(library, kIsolateSnapshotDataSymbolName)); |
| if (isolate_data_buffer == NULL) { |
| FATAL1("Failed to resolve symbol '%s'\n", kIsolateSnapshotDataSymbolName); |
| } |
| |
| const uint8_t* isolate_instructions_buffer = |
| reinterpret_cast<const uint8_t*>(Extensions::ResolveSymbol( |
| library, kIsolateSnapshotInstructionsSymbolName)); |
| if (isolate_instructions_buffer == NULL) { |
| FATAL1("Failed to resolve symbol '%s'\n", |
| kIsolateSnapshotInstructionsSymbolName); |
| } |
| |
| return new DylibAppSnapshot(library, vm_data_buffer, vm_instructions_buffer, |
| isolate_data_buffer, isolate_instructions_buffer); |
| } |
| #endif // defined(DART_PRECOMPILED_RUNTIME) |
| |
| AppSnapshot* Snapshot::TryReadAppSnapshot(const char* script_name) { |
| if (File::GetType(NULL, script_name, true) != File::kIsFile) { |
| // If 'script_name' refers to a pipe, don't read to check for an app |
| // snapshot since we cannot rewind if it isn't (and couldn't mmap it in |
| // anyway if it was). |
| return NULL; |
| } |
| AppSnapshot* snapshot = TryReadAppSnapshotBlobs(script_name); |
| if (snapshot != NULL) { |
| return snapshot; |
| } |
| #if defined(DART_PRECOMPILED_RUNTIME) |
| // For testing AOT with the standalone embedder, we also support loading |
| // from a dynamic library to simulate what happens on iOS. |
| snapshot = TryReadAppSnapshotDynamicLibrary(script_name); |
| if (snapshot != NULL) { |
| return snapshot; |
| } |
| #endif // defined(DART_PRECOMPILED_RUNTIME) |
| return NULL; |
| } |
| |
| static void WriteSnapshotFile(const char* filename, |
| bool write_magic_number, |
| const uint8_t* buffer, |
| const intptr_t size) { |
| File* file = File::Open(NULL, filename, File::kWriteTruncate); |
| if (file == NULL) { |
| ErrorExit(kErrorExitCode, "Unable to open file %s for writing snapshot\n", |
| filename); |
| } |
| |
| if (write_magic_number) { |
| // Write the magic number to indicate file is a script snapshot. |
| DartUtils::WriteSnapshotMagicNumber(file); |
| } |
| |
| if (!file->WriteFully(buffer, size)) { |
| ErrorExit(kErrorExitCode, "Unable to write file %s for writing snapshot\n", |
| filename); |
| } |
| file->Release(); |
| } |
| |
| static bool WriteInt64(File* file, int64_t size) { |
| return file->WriteFully(&size, sizeof(size)); |
| } |
| |
| static void WriteAppSnapshot(const char* filename, |
| uint8_t* vm_data_buffer, |
| intptr_t vm_data_size, |
| uint8_t* vm_instructions_buffer, |
| intptr_t vm_instructions_size, |
| uint8_t* isolate_data_buffer, |
| intptr_t isolate_data_size, |
| uint8_t* isolate_instructions_buffer, |
| intptr_t isolate_instructions_size) { |
| File* file = File::Open(NULL, filename, File::kWriteTruncate); |
| if (file == NULL) { |
| ErrorExit(kErrorExitCode, "Unable to write snapshot file '%s'\n", filename); |
| } |
| |
| file->WriteFully(appjit_magic_number.bytes, appjit_magic_number.length); |
| WriteInt64(file, vm_data_size); |
| WriteInt64(file, vm_instructions_size); |
| WriteInt64(file, isolate_data_size); |
| WriteInt64(file, isolate_instructions_size); |
| ASSERT(file->Position() == kAppSnapshotHeaderSize); |
| |
| file->SetPosition(Utils::RoundUp(file->Position(), kAppSnapshotPageSize)); |
| if (!file->WriteFully(vm_data_buffer, vm_data_size)) { |
| ErrorExit(kErrorExitCode, "Unable to write snapshot file '%s'\n", filename); |
| } |
| |
| if (vm_instructions_size != 0) { |
| file->SetPosition(Utils::RoundUp(file->Position(), kAppSnapshotPageSize)); |
| if (!file->WriteFully(vm_instructions_buffer, vm_instructions_size)) { |
| ErrorExit(kErrorExitCode, "Unable to write snapshot file '%s'\n", |
| filename); |
| } |
| } |
| |
| file->SetPosition(Utils::RoundUp(file->Position(), kAppSnapshotPageSize)); |
| if (!file->WriteFully(isolate_data_buffer, isolate_data_size)) { |
| ErrorExit(kErrorExitCode, "Unable to write snapshot file '%s'\n", filename); |
| } |
| |
| if (isolate_instructions_size != 0) { |
| file->SetPosition(Utils::RoundUp(file->Position(), kAppSnapshotPageSize)); |
| if (!file->WriteFully(isolate_instructions_buffer, |
| isolate_instructions_size)) { |
| ErrorExit(kErrorExitCode, "Unable to write snapshot file '%s'\n", |
| filename); |
| } |
| } |
| |
| file->Flush(); |
| file->Release(); |
| } |
| |
| void Snapshot::GenerateScript(const char* snapshot_filename) { |
| // First create a snapshot. |
| uint8_t* buffer = NULL; |
| intptr_t size = 0; |
| Dart_Handle result = Dart_CreateScriptSnapshot(&buffer, &size); |
| if (Dart_IsError(result)) { |
| ErrorExit(kErrorExitCode, "%s\n", Dart_GetError(result)); |
| } |
| |
| WriteSnapshotFile(snapshot_filename, true, buffer, size); |
| } |
| |
| void Snapshot::GenerateAppJIT(const char* snapshot_filename) { |
| #if defined(TARGET_ARCH_IA32) || defined(TARGET_ARCH_DBC) |
| // Snapshots with code are not supported on IA32 or DBC. |
| uint8_t* isolate_buffer = NULL; |
| intptr_t isolate_size = 0; |
| |
| Dart_Handle result = |
| Dart_CreateSnapshot(NULL, NULL, &isolate_buffer, &isolate_size); |
| if (Dart_IsError(result)) { |
| ErrorExit(kErrorExitCode, "%s\n", Dart_GetError(result)); |
| } |
| |
| WriteAppSnapshot(snapshot_filename, NULL, 0, NULL, 0, isolate_buffer, |
| isolate_size, NULL, 0); |
| #else |
| uint8_t* isolate_data_buffer = NULL; |
| intptr_t isolate_data_size = 0; |
| uint8_t* isolate_instructions_buffer = NULL; |
| intptr_t isolate_instructions_size = 0; |
| Dart_Handle result = Dart_CreateAppJITSnapshotAsBlobs( |
| &isolate_data_buffer, &isolate_data_size, &isolate_instructions_buffer, |
| &isolate_instructions_size); |
| if (Dart_IsError(result)) { |
| ErrorExit(kErrorExitCode, "%s\n", Dart_GetError(result)); |
| } |
| WriteAppSnapshot(snapshot_filename, NULL, 0, NULL, 0, isolate_data_buffer, |
| isolate_data_size, isolate_instructions_buffer, |
| isolate_instructions_size); |
| #endif |
| } |
| |
| void Snapshot::GenerateAppAOTAsBlobs(const char* snapshot_filename) { |
| uint8_t* vm_data_buffer = NULL; |
| intptr_t vm_data_size = 0; |
| uint8_t* vm_instructions_buffer = NULL; |
| intptr_t vm_instructions_size = 0; |
| uint8_t* isolate_data_buffer = NULL; |
| intptr_t isolate_data_size = 0; |
| uint8_t* isolate_instructions_buffer = NULL; |
| intptr_t isolate_instructions_size = 0; |
| Dart_Handle result = Dart_CreateAppAOTSnapshotAsBlobs( |
| &vm_data_buffer, &vm_data_size, &vm_instructions_buffer, |
| &vm_instructions_size, &isolate_data_buffer, &isolate_data_size, |
| &isolate_instructions_buffer, &isolate_instructions_size); |
| if (Dart_IsError(result)) { |
| ErrorExit(kErrorExitCode, "%s\n", Dart_GetError(result)); |
| } |
| WriteAppSnapshot(snapshot_filename, vm_data_buffer, vm_data_size, |
| vm_instructions_buffer, vm_instructions_size, |
| isolate_data_buffer, isolate_data_size, |
| isolate_instructions_buffer, isolate_instructions_size); |
| } |
| |
| void Snapshot::GenerateAppAOTAsAssembly(const char* snapshot_filename) { |
| uint8_t* assembly_buffer = NULL; |
| intptr_t assembly_size = 0; |
| Dart_Handle result = |
| Dart_CreateAppAOTSnapshotAsAssembly(&assembly_buffer, &assembly_size); |
| if (Dart_IsError(result)) { |
| ErrorExit(kErrorExitCode, "%s\n", Dart_GetError(result)); |
| } |
| WriteSnapshotFile(snapshot_filename, false, assembly_buffer, assembly_size); |
| } |
| |
| } // namespace bin |
| } // namespace dart |
