[headless] Embed pak file into binary.

ui/base/resource/data_pack.cc

 // Copyright (c) 2012 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 "ui/base/resource/data_pack.h"
 
 #include <errno.h>
 #include <utility>
 
 #include "base/files/file_util.h"
@@ skipping 62 matching lines @@
 
 DataPack::~DataPack() {
 }
 
 bool DataPack::LoadFromPath(const base::FilePath& path) {
   mmap_.reset(new base::MemoryMappedFile);
   if (!mmap_->Initialize(path)) {
     DLOG(ERROR) << "Failed to mmap datapack";
     UMA_HISTOGRAM_ENUMERATION("DataPack.Load", INIT_FAILED,
                               LOAD_ERRORS_COUNT);
-    mmap_.reset();
+    Clear();
     return false;
   }
+  length_ = mmap_->length();
+  data_ = mmap_->data();
   return LoadImpl();
 }
 
 bool DataPack::LoadFromFile(base::File file) {
   return LoadFromFileRegion(std::move(file),
                             base::MemoryMappedFile::Region::kWholeFile);
 }
 
 bool DataPack::LoadFromFileRegion(
     base::File file,
     const base::MemoryMappedFile::Region& region) {
   mmap_.reset(new base::MemoryMappedFile);
   if (!mmap_->Initialize(std::move(file), region)) {
     DLOG(ERROR) << "Failed to mmap datapack";
     UMA_HISTOGRAM_ENUMERATION("DataPack.Load", INIT_FAILED_FROM_FILE,
                               LOAD_ERRORS_COUNT);
-    mmap_.reset();
+    Clear();
     return false;
   }
+  length_ = mmap_->length();
+  data_ = mmap_->data();
+  return LoadImpl();
+}
+
+bool DataPack::LoadFromBuffer(base::StringPiece buffer) {
+  length_ = buffer.length();
+  data_ = reinterpret_cast<const uint8_t*>(buffer.data());
   return LoadImpl();
 }
 
 bool DataPack::LoadImpl() {
   // Sanity check the header of the file.
-  if (kHeaderLength > mmap_->length()) {
+  if (kHeaderLength > length_) {
     DLOG(ERROR) << "Data pack file corruption: incomplete file header.";
     UMA_HISTOGRAM_ENUMERATION("DataPack.Load", HEADER_TRUNCATED,
                               LOAD_ERRORS_COUNT);
-    mmap_.reset();
+    // Clear memory mapped file even when we're loading data from a buffer,
+    // in order to simplify code.
+    Clear();
     return false;
   }
 
   // Parse the header of the file.
   // First uint32_t: version; second: resource count;
-  const uint32_t* ptr = reinterpret_cast<const uint32_t*>(mmap_->data());
+  const uint32_t* ptr = reinterpret_cast<const uint32_t*>(data_);
   uint32_t version = ptr[0];
   if (version != kFileFormatVersion) {
     LOG(ERROR) << "Bad data pack version: got " << version << ", expected "
                << kFileFormatVersion;
     UMA_HISTOGRAM_ENUMERATION("DataPack.Load", BAD_VERSION,
                               LOAD_ERRORS_COUNT);
-    mmap_.reset();
+    Clear();
     return false;
   }
   resource_count_ = ptr[1];
 
   // third: text encoding.
   const uint8_t* ptr_encoding = reinterpret_cast<const uint8_t*>(ptr + 2);
   text_encoding_type_ = static_cast<TextEncodingType>(*ptr_encoding);
   if (text_encoding_type_ != UTF8 && text_encoding_type_ != UTF16 &&
       text_encoding_type_ != BINARY) {
     LOG(ERROR) << "Bad data pack text encoding: got " << text_encoding_type_
                << ", expected between " << BINARY << " and " << UTF16;
     UMA_HISTOGRAM_ENUMERATION("DataPack.Load", WRONG_ENCODING,
                               LOAD_ERRORS_COUNT);
-    mmap_.reset();
+    Clear();
     return false;
   }
 
   // Sanity check the file.
   // 1) Check we have enough entries. There's an extra entry after the last item
   // which gives the length of the last item.
   if (kHeaderLength + (resource_count_ + 1) * sizeof(DataPackEntry) >
-      mmap_->length()) {
+      length_) {
     LOG(ERROR) << "Data pack file corruption: too short for number of "
                   "entries specified.";
     UMA_HISTOGRAM_ENUMERATION("DataPack.Load", INDEX_TRUNCATED,
                               LOAD_ERRORS_COUNT);
-    mmap_.reset();
+    Clear();
     return false;
   }
   // 2) Verify the entries are within the appropriate bounds. There's an extra
   // entry after the last item which gives us the length of the last item.
   for (size_t i = 0; i < resource_count_ + 1; ++i) {
     const DataPackEntry* entry = reinterpret_cast<const DataPackEntry*>(
-        mmap_->data() + kHeaderLength + (i * sizeof(DataPackEntry)));
-    if (entry->file_offset > mmap_->length()) {
+        data_ + kHeaderLength + (i * sizeof(DataPackEntry)));
+    if (entry->file_offset > length_) {
       LOG(ERROR) << "Entry #" << i << " in data pack points off end of file. "
                  << "Was the file corrupted?";
       UMA_HISTOGRAM_ENUMERATION("DataPack.Load", ENTRY_NOT_FOUND,
                                 LOAD_ERRORS_COUNT);
-      mmap_.reset();
+      Clear();
       return false;
     }
   }
 
   return true;
 }
 
+void DataPack::Clear() {
+  mmap_.reset();
+  length_ = 0;
+  data_ = nullptr;
+}
+
 bool DataPack::HasResource(uint16_t resource_id) const {
-  return !!bsearch(&resource_id, mmap_->data() + kHeaderLength, resource_count_,
+  return !!bsearch(&resource_id, data_ + kHeaderLength, resource_count_,
                    sizeof(DataPackEntry), DataPackEntry::CompareById);
 }
 
 bool DataPack::GetStringPiece(uint16_t resource_id,
                               base::StringPiece* data) const {
   // It won't be hard to make this endian-agnostic, but it's not worth
   // bothering to do right now.
 #if defined(__BYTE_ORDER)
   // Linux check
   static_assert(__BYTE_ORDER == __LITTLE_ENDIAN,
                 "datapack assumes little endian");
 #elif defined(__BIG_ENDIAN__)
   // Mac check
   #error DataPack assumes little endian
 #endif
 
   const DataPackEntry* target = reinterpret_cast<const DataPackEntry*>(
-      bsearch(&resource_id, mmap_->data() + kHeaderLength, resource_count_,
+      bsearch(&resource_id, data_ + kHeaderLength, resource_count_,
               sizeof(DataPackEntry), DataPackEntry::CompareById));
   if (!target) {
     return false;
   }
 
   const DataPackEntry* next_entry = target + 1;
   // If the next entry points beyond the end of the file this data pack's entry
   // table is corrupt. Log an error and return false. See
   // http://crbug.com/371301.
-  if (next_entry->file_offset > mmap_->length()) {
+  if (next_entry->file_offset > length_) {
     size_t entry_index = target -
-        reinterpret_cast<const DataPackEntry*>(mmap_->data() + kHeaderLength);
+        reinterpret_cast<const DataPackEntry*>(data_ + kHeaderLength);
     LOG(ERROR) << "Entry #" << entry_index << " in data pack points off end "
                << "of file. This should have been caught when loading. Was the "
                << "file modified?";
     return false;
   }
 
   size_t length = next_entry->file_offset - target->file_offset;
-  data->set(reinterpret_cast<const char*>(mmap_->data() + target->file_offset),
+  data->set(reinterpret_cast<const char*>(data_ + target->file_offset),
             length);
   return true;
 }
 
 base::RefCountedStaticMemory* DataPack::GetStaticMemory(
     uint16_t resource_id) const {
   base::StringPiece piece;
   if (!GetStringPiece(resource_id, &piece))
     return NULL;
 
@@ skipping 10 matching lines @@
 
 bool DataPack::HasOnlyMaterialDesignAssets() const {
   return has_only_material_design_assets_;
 }
 
 #if DCHECK_IS_ON()
 void DataPack::CheckForDuplicateResources(
     const ScopedVector<ResourceHandle>& packs) {
   for (size_t i = 0; i < resource_count_ + 1; ++i) {
     const DataPackEntry* entry = reinterpret_cast<const DataPackEntry*>(
-        mmap_->data() + kHeaderLength + (i * sizeof(DataPackEntry)));
+        data_ + kHeaderLength + (i * sizeof(DataPackEntry)));
     const uint16_t resource_id = entry->resource_id;
     const float resource_scale = GetScaleForScaleFactor(scale_factor_);
     for (const ResourceHandle* handle : packs) {
       if (HasOnlyMaterialDesignAssets() !=
           handle->HasOnlyMaterialDesignAssets()) {
         continue;
       }
       if (GetScaleForScaleFactor(handle->GetScaleFactor()) != resource_scale)
         continue;
       DCHECK(!handle->HasResource(resource_id)) << "Duplicate resource "
@@ skipping 89 matching lines @@
       return false;
     }
   }
 
   base::CloseFile(file);
 
   return true;
 }
 
 }  // namespace ui