Index: ui/base/resource/data_pack.cc |
diff --git a/ui/base/resource/data_pack.cc b/ui/base/resource/data_pack.cc |
index 18fee46d0699aeda0910a51328cd336d87855155..1fcdfe1dff58a8629bf7d0858b2f512680562872 100644 |
--- a/ui/base/resource/data_pack.cc |
+++ b/ui/base/resource/data_pack.cc |
@@ -17,7 +17,7 @@ |
namespace { |
-static const uint32 kFileFormatVersion = 2; |
+static const uint32 kFileFormatVersion = 3; |
// Length of file header: version and entry count. |
static const size_t kHeaderLength = 2 * sizeof(uint32); |
@@ -25,7 +25,6 @@ static const size_t kHeaderLength = 2 * sizeof(uint32); |
struct DataPackEntry { |
uint16 resource_id; |
uint32 file_offset; |
- uint32 length; |
static int CompareById(const void* void_key, const void* void_entry) { |
uint16 key = *reinterpret_cast<const uint16*>(void_key); |
@@ -42,7 +41,7 @@ struct DataPackEntry { |
}; |
#pragma pack(pop) |
-COMPILE_ASSERT(sizeof(DataPackEntry) == 10, size_of_header_must_be_ten); |
+COMPILE_ASSERT(sizeof(DataPackEntry) == 6, size_of_entry_must_be_six); |
// We're crashing when trying to load a pak file on Windows. Add some error |
// codes for logging. |
@@ -101,11 +100,12 @@ bool DataPack::Load(const FilePath& path) { |
mmap_.reset(); |
return false; |
} |
- // 2) Verify the entries are within the appropriate bounds. |
- for (size_t i = 0; i < resource_count_; ++i) { |
+ // 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 + entry->length > mmap_->length()) { |
+ if (entry->file_offset > mmap_->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, |
@@ -131,14 +131,17 @@ bool DataPack::GetStringPiece(uint16 resource_id, |
#error DataPack assumes little endian |
#endif |
- DataPackEntry* target = reinterpret_cast<DataPackEntry*>( |
+ const DataPackEntry* target = reinterpret_cast<const DataPackEntry*>( |
bsearch(&resource_id, mmap_->data() + kHeaderLength, resource_count_, |
sizeof(DataPackEntry), DataPackEntry::CompareById)); |
if (!target) { |
return false; |
} |
- data->set(mmap_->data() + target->file_offset, target->length); |
+ const DataPackEntry* next_entry = target + 1; |
+ size_t length = next_entry->file_offset - target->file_offset; |
+ |
+ data->set(mmap_->data() + target->file_offset, length); |
return true; |
} |
@@ -173,8 +176,9 @@ bool DataPack::WritePack(const FilePath& path, |
return false; |
} |
- // Each entry is 1 uint16 + 2 uint32s. |
- uint32 index_length = entry_count * sizeof(DataPackEntry); |
+ // Each entry is a uint16 + a uint32. We have an extra entry after the last |
+ // item so we can compute the size of the list item. |
+ uint32 index_length = (entry_count + 1) * sizeof(DataPackEntry); |
uint32 data_offset = kHeaderLength + index_length; |
for (std::map<uint16, base::StringPiece>::const_iterator it = |
resources.begin(); |
@@ -192,14 +196,22 @@ bool DataPack::WritePack(const FilePath& path, |
return false; |
} |
- uint32 len = it->second.length(); |
- if (fwrite(&len, sizeof(len), 1, file) != 1) { |
- LOG(ERROR) << "Failed to write length for " << resource_id; |
- file_util::CloseFile(file); |
- return false; |
- } |
+ data_offset += it->second.length(); |
+ } |
- data_offset += len; |
+ // We place an extra entry after the last item that allows us to read the |
+ // size of the last item. |
+ uint16 resource_id = 0; |
+ if (fwrite(&resource_id, sizeof(resource_id), 1, file) != 1) { |
+ LOG(ERROR) << "Failed to write extra resource id."; |
+ file_util::CloseFile(file); |
+ return false; |
+ } |
+ |
+ if (fwrite(&data_offset, sizeof(data_offset), 1, file) != 1) { |
+ LOG(ERROR) << "Failed to write extra offset."; |
+ file_util::CloseFile(file); |
+ return false; |
} |
for (std::map<uint16, base::StringPiece>::const_iterator it = |