[standalone] Automatically decompress gzip'd resources, including sources and script snapshots.

runtime/bin/dartutils.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 "bin/dartutils.h"
 
 #include "bin/crypto.h"
 #include "bin/directory.h"
 #include "bin/extensions.h"
 #include "bin/file.h"
@@ skipping 30 matching lines @@
 const char* const DartUtils::kCoreLibURL = "dart:core";
 const char* const DartUtils::kInternalLibURL = "dart:_internal";
 const char* const DartUtils::kIsolateLibURL = "dart:isolate";
 const char* const DartUtils::kIOLibURL = "dart:io";
 const char* const DartUtils::kIOLibPatchURL = "dart:io-patch";
 const char* const DartUtils::kUriLibURL = "dart:uri";
 const char* const DartUtils::kHttpScheme = "http:";
 const char* const DartUtils::kVMServiceLibURL = "dart:vmservice";
 
 struct MagicNumberData {
-  static const intptr_t kLength = 4;
+  static const intptr_t kMaxLength = 4;
 
-  const uint8_t bytes[kLength];
-  bool should_skip;
+  intptr_t length;
+  const uint8_t bytes[kMaxLength];
 };
 
-MagicNumberData snapshot_magic_number = {{0xf5, 0xf5, 0xdc, 0xdc}, true};
-MagicNumberData kernel_magic_number = {{0x90, 0xab, 0xcd, 0xef}, false};
+MagicNumberData snapshot_magic_number = {4, {0xf5, 0xf5, 0xdc, 0xdc}};
+MagicNumberData kernel_magic_number = {4, {0x90, 0xab, 0xcd, 0xef}};
+MagicNumberData gzip_magic_number = {2, {0x1f, 0x8b, 0, 0}};
 
 static bool IsWindowsHost() {
 #if defined(HOST_OS_WINDOWS)
   return true;
 #else   // defined(HOST_OS_WINDOWS)
   return false;
 #endif  // defined(HOST_OS_WINDOWS)
 }
 
 const char* DartUtils::MapLibraryUrl(const char* url_string) {
@@ skipping 263 matching lines @@
 }
 
 Dart_Handle DartUtils::ResolveScript(Dart_Handle url) {
   const int kNumArgs = 1;
   Dart_Handle dart_args[kNumArgs];
   dart_args[0] = url;
   return Dart_Invoke(DartUtils::BuiltinLib(), NewString("_resolveScriptUri"),
                      kNumArgs, dart_args);
 }
 
-static bool CheckMagicNumber(const uint8_t** buf,
-                             intptr_t* len,
+static bool CheckMagicNumber(const uint8_t* buffer,
+                             intptr_t buffer_length,
                              const MagicNumberData& magic_number) {
-  if ((*len >= MagicNumberData::kLength) &&
-      (memcmp(*buf, magic_number.bytes, MagicNumberData::kLength) == 0)) {
-    if (magic_number.should_skip) {
-      *buf += MagicNumberData::kLength;
-      *len -= MagicNumberData::kLength;
-    }
-    return true;
+  if ((buffer_length >= magic_number.length)) {
+    return memcmp(buffer, magic_number.bytes, magic_number.length) == 0;
   }
   return false;
 }
 
-DartUtils::MagicNumber DartUtils::SniffForMagicNumber(const uint8_t** buf,
-                                                      intptr_t* len) {
-  if (CheckMagicNumber(buf, len, snapshot_magic_number)) {
+DartUtils::MagicNumber DartUtils::SniffForMagicNumber(const uint8_t* buffer,
+                                                      intptr_t buffer_length) {
+  if (CheckMagicNumber(buffer, buffer_length, snapshot_magic_number)) {
     return kSnapshotMagicNumber;
   }
 
-  if (CheckMagicNumber(buf, len, kernel_magic_number)) {
+  if (CheckMagicNumber(buffer, buffer_length, kernel_magic_number)) {
     return kKernelMagicNumber;
   }
 
+  if (CheckMagicNumber(buffer, buffer_length, gzip_magic_number)) {
+    return kGzipMagicNumber;
+  }
+
   return kUnknownMagicNumber;
 }
 
-void DartUtils::WriteMagicNumber(File* file) {
+void DartUtils::WriteSnapshotMagicNumber(File* file) {
   // Write a magic number and version information into the snapshot file.
-  bool bytes_written =
-      file->WriteFully(snapshot_magic_number.bytes, MagicNumberData::kLength);
+  bool bytes_written = file->WriteFully(snapshot_magic_number.bytes,
+                                        snapshot_magic_number.length);
   ASSERT(bytes_written);
 }
 
+void DartUtils::SkipSnapshotMagicNumber(const uint8_t** buffer,
+                                        intptr_t* buffer_length) {
+  *buffer += snapshot_magic_number.length;
+  *buffer_length -= snapshot_magic_number.length;
+}
+
 void FUNCTION_NAME(Builtin_GetCurrentDirectory)(Dart_NativeArguments args) {
   const char* current = Directory::Current();
   if (current != NULL) {
     Dart_SetReturnValue(args, DartUtils::NewString(current));
   } else {
     Dart_Handle err = DartUtils::NewError("Failed to get current directory.");
     Dart_PropagateError(err);
   }
 }
 
@@ skipping 568 matching lines @@
       new CObjectString(CObject::NewString(os_error->message()));
   CObjectArray* result = new CObjectArray(CObject::NewArray(3));
   result->SetAt(0, new CObjectInt32(CObject::NewInt32(kOSError)));
   result->SetAt(1, new CObjectInt32(CObject::NewInt32(os_error->code())));
   result->SetAt(2, error_message);
   return result;
 }
 
 }  // namespace bin
 }  // namespace dart