Index: net/disk_cache/very_simple/very_simple_synchronous_entry.cc |
diff --git a/net/disk_cache/very_simple/very_simple_synchronous_entry.cc b/net/disk_cache/very_simple/very_simple_synchronous_entry.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..8a3ac4372b3fcb9c862d30bbc9588db3b7b7c432 |
--- /dev/null |
+++ b/net/disk_cache/very_simple/very_simple_synchronous_entry.cc |
@@ -0,0 +1,324 @@ |
+// Copyright (c) 2013 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 "net/disk_cache/very_simple/very_simple_synchronous_entry.h" |
+ |
+#include <algorithm> |
+#include <cstring> |
+ |
+#include "base/basictypes.h" |
+#include "base/file_util.h" |
+#include "base/location.h" |
+#include "base/message_loop_proxy.h" |
+#include "base/sha1.h" |
+#include "base/stringprintf.h" |
+#include "base/task_runner.h" |
+#include "net/base/io_buffer.h" |
+#include "net/base/net_errors.h" |
+#include "net/disk_cache/very_simple/very_simple_disk_format.h" |
+#include "third_party/smhasher/src/MurmurHash3.h" |
+ |
+using base::Bind; |
+using base::ClosePlatformFile; |
+using base::GetPlatformFileInfo; |
+using base::PlatformFileError; |
+using base::PlatformFileInfo; |
+using base::PLATFORM_FILE_CREATE_ALWAYS; |
+using base::PLATFORM_FILE_OK; |
+using base::PLATFORM_FILE_OPEN; |
+using base::PLATFORM_FILE_READ; |
+using base::PLATFORM_FILE_WRITE; |
+using base::ReadPlatformFile; |
+using base::TaskRunner; |
+using base::Time; |
+using base::TruncatePlatformFile; |
+using base::WritePlatformFile; |
+ |
+namespace { |
+ |
+std::string GetFilenameForKeyAndIndex(const std::string& key, int index) { |
+ const std::string sha_hash = base::SHA1HashString(key); |
+ return StringPrintf("%02x%02x%02x%02x%02x_%1d", |
+ implicit_cast<unsigned char>(sha_hash[0]), |
+ implicit_cast<unsigned char>(sha_hash[1]), |
+ implicit_cast<unsigned char>(sha_hash[2]), |
+ implicit_cast<unsigned char>(sha_hash[3]), |
+ implicit_cast<unsigned char>(sha_hash[4]), index); |
+} |
+ |
+int32 DataSizeFromKeyAndFileSize(size_t key_size, int64 file_size) { |
+ return file_size - key_size - sizeof(disk_cache::VerySimpleFileHeader); |
+} |
+ |
+int64 FileOffsetFromDataOffset(size_t key_size, int offset) { |
+ const int64 headers_size = sizeof(disk_cache::VerySimpleFileHeader) + |
+ key_size; |
+ return headers_size + offset; |
+} |
+ |
+} // namespace |
+ |
+namespace disk_cache { |
+ |
+// static |
+void VerySimpleSynchronousEntry::OpenEntry( |
+ const FilePath& path, |
+ const std::string& key, |
+ const scoped_refptr<TaskRunner>& callback_runner, |
+ const SynchronousEntryCallback& callback) { |
+ VerySimpleSynchronousEntry* sync_entry = |
+ new VerySimpleSynchronousEntry(callback_runner, path, key); |
+ |
+ if (!sync_entry->InitializeForOpen()) { |
+ delete sync_entry; |
+ callback_runner->PostTask(FROM_HERE, |
+ Bind(callback, |
+ static_cast<VerySimpleSynchronousEntry*>( |
+ NULL), |
+ net::ERR_FAILED)); |
+ return; |
+ } |
+ callback_runner->PostTask(FROM_HERE, Bind(callback, sync_entry, net::OK)); |
rvargas (doing something else)
2013/02/06 03:28:40
nit: merge the two PostTasks?
gavinp
2013/02/08 23:17:51
Done.
|
+} |
+ |
+// static |
+void VerySimpleSynchronousEntry::CreateEntry( |
+ const FilePath& path, |
+ const std::string& key, |
+ const scoped_refptr<TaskRunner>& callback_runner, |
+ const SynchronousEntryCallback& callback) { |
+ VerySimpleSynchronousEntry* sync_entry = |
+ new VerySimpleSynchronousEntry(callback_runner, path, key); |
+ |
+ if (!sync_entry->InitializeForCreate()) { |
+ delete sync_entry; |
+ callback_runner->PostTask(FROM_HERE, |
+ Bind(callback, |
+ static_cast<VerySimpleSynchronousEntry*>( |
+ NULL), |
+ net::ERR_FAILED)); |
+ return; |
+ } |
+ callback_runner->PostTask(FROM_HERE, Bind(callback, sync_entry, net::OK)); |
+} |
+ |
+// static |
+void VerySimpleSynchronousEntry::DoomEntry( |
+ const FilePath& path, |
+ const std::string& key, |
+ scoped_refptr<TaskRunner> callback_runner, |
+ const net::CompletionCallback& callback) { |
+ for (int i = 0; i < kIndexCount; ++i) { |
+ bool delete_result = |
+ file_util::Delete(path.AppendASCII(GetFilenameForKeyAndIndex(key, i)), |
rvargas (doing something else)
2013/02/06 03:28:40
How is Read and Write expected to work after this?
gavinp
2013/02/08 23:17:51
They are not.
pasko-google - do not use
2013/02/11 13:59:25
I think Ricardo refers to when we have pending Rea
gavinp
2013/02/11 17:55:50
Aha, yes. And we should support that, and we don't
|
+ false); |
+ DCHECK(delete_result); |
+ } |
+ if (!callback.is_null()) |
+ callback_runner->PostTask(FROM_HERE, Bind(callback, net::OK)); |
+} |
+ |
+void VerySimpleSynchronousEntry::DoomAndClose() { |
+ scoped_refptr<TaskRunner> callback_runner = callback_runner_; |
+ FilePath path = path_; |
+ std::string key = key_; |
+ |
+ Close(); |
+ // |this| is now deleted. |
+ |
+ DoomEntry(path, key, callback_runner, net::CompletionCallback()); |
+} |
+ |
+void VerySimpleSynchronousEntry::Close() { |
+ for (int i = 0; i < kIndexCount; ++i) { |
+ bool result = ClosePlatformFile(files_[i]); |
+ DCHECK(result); |
+ } |
+ delete this; |
+} |
+ |
+void VerySimpleSynchronousEntry::ReadData( |
+ int index, |
+ int offset, |
+ net::IOBuffer* buf, |
+ int buf_len, |
+ const SynchronousEntryCallback& callback) { |
+ DCHECK(initialized_); |
+ if (status_[index].mode != EntryStatus::ENTRY_READER) |
+ status_[index].data_offset = 0; |
+ DCHECK_EQ(status_[index].data_offset, offset); |
+ status_[index].mode = EntryStatus::ENTRY_READER; |
+ |
+ int64 file_offset = FileOffsetFromDataOffset(key_.size(), offset); |
+ int bytes_read = ReadPlatformFile(files_[index], file_offset, |
+ buf->data(), buf_len); |
+ if (bytes_read > 0) { |
+ last_used_ = Time::Now(); |
+ status_[index].data_offset += bytes_read; |
+ } |
+ int result = (bytes_read >= 0) ? bytes_read : net::ERR_FAILED; |
+ callback_runner_->PostTask(FROM_HERE, Bind(callback, this, result)); |
+} |
+ |
+void VerySimpleSynchronousEntry::WriteData( |
+ int index, |
+ int offset, |
+ net::IOBuffer* buf, |
+ int buf_len, |
+ const SynchronousEntryCallback& callback, |
+ bool truncate) { |
+ DCHECK(initialized_); |
+ if (status_[index].mode != EntryStatus::ENTRY_WRITER) |
+ status_[index].data_offset = 0; |
+ DCHECK_EQ(status_[index].data_offset, offset); |
+ status_[index].mode = EntryStatus::ENTRY_WRITER; |
+ |
+ int64 file_offset = FileOffsetFromDataOffset(key_.size(), offset); |
+ if (buf_len > 0) { |
+ if (WritePlatformFile(files_[index], file_offset, buf->data(), buf_len) != |
+ buf_len) { |
+ callback_runner_->PostTask(FROM_HERE, |
+ Bind(callback, this, net::ERR_FAILED)); |
+ return; |
+ } |
+ data_size_[index] = std::max(data_size_[index], offset + buf_len); |
+ status_[index].data_offset += buf_len; |
+ } |
+ if (truncate) { |
+ data_size_[index] = offset + buf_len; |
+ if (!TruncatePlatformFile(files_[index], file_offset + buf_len)) { |
+ callback_runner_->PostTask(FROM_HERE, |
+ Bind(callback, this, net::ERR_FAILED)); |
+ return; |
+ } |
+ } |
+ last_modified_ = Time::Now(); |
+ callback_runner_->PostTask(FROM_HERE, Bind(callback, this, buf_len)); |
+} |
+ |
+VerySimpleSynchronousEntry::EntryStatus::EntryStatus() |
+ : mode(ENTRY_UNINITIALIZED), |
+ data_offset(0) { |
+} |
+ |
+VerySimpleSynchronousEntry::VerySimpleSynchronousEntry( |
+ const scoped_refptr<TaskRunner>& callback_runner, |
+ const FilePath& path, |
+ const std::string& key) : callback_runner_(callback_runner), |
+ path_(path), |
+ key_(key), |
+ initialized_(false) { |
+} |
+ |
+VerySimpleSynchronousEntry::~VerySimpleSynchronousEntry() { |
+} |
+ |
+bool VerySimpleSynchronousEntry::OpenOrCreateFiles(bool create) { |
+ for (int i = 0; i < kIndexCount; ++i) { |
+ FilePath filename = path_.AppendASCII(GetFilenameForKeyAndIndex(key_, i)); |
+ int flags = PLATFORM_FILE_READ | PLATFORM_FILE_WRITE; |
+ if (create) |
+ flags = flags | PLATFORM_FILE_CREATE_ALWAYS; |
+ else |
+ flags = flags | PLATFORM_FILE_OPEN; |
+ PlatformFileError error; |
+ files_[i] = CreatePlatformFile(filename, flags, NULL, &error); |
+ if (error != PLATFORM_FILE_OK) { |
+ while (--i >= 0) { |
+ bool did_close = ClosePlatformFile(files_[i]); |
+ DCHECK(did_close); |
+ } |
+ return false; |
+ } |
+ } |
+ |
+ for (int i = 0; i < kIndexCount; ++i) { |
+ PlatformFileInfo file_info; |
+ if (!GetPlatformFileInfo(files_[i], &file_info)) |
+ NOTREACHED(); |
+ last_used_ = std::max(last_used_, file_info.last_accessed); |
+ last_modified_ = std::max(last_modified_, file_info.last_modified); |
+ data_size_[i] = DataSizeFromKeyAndFileSize(key_.size(), file_info.size); |
+ } |
+ |
+ return true; |
+} |
+ |
+bool VerySimpleSynchronousEntry::InitializeForOpen() { |
+ DCHECK(!initialized_); |
+ if (!OpenOrCreateFiles(false)) |
+ return false; |
+ |
+ VLOG(8) << "VerySimpleSynchronousEntry::InitializeForOpen"; |
rvargas (doing something else)
2013/02/06 03:28:40
nit: general tracing is bad
gavinp
2013/02/08 23:17:51
Done.
|
+ for (int i = 0; i < kIndexCount; ++i) { |
+ VerySimpleFileHeader header; |
+ if (ReadPlatformFile(files_[i], 0, reinterpret_cast<char*>(&header), |
+ sizeof(header)) != sizeof(header)) { |
+ VLOG(9) << " no header"; |
rvargas (doing something else)
2013/02/06 03:28:40
... and logging on errors is good.
gavinp
2013/02/08 23:17:51
Done.
|
+ return false; |
+ } |
+ if (header.initial_magic_number != kVerySimpleInitialMagicNumber || |
+ header.version != kVerySimpleVersion) { |
+ VLOG(9) << " no magic number"; |
+ return false; |
+ } |
+ |
+ char key[1024]; |
+ DCHECK_LE(header.key_length, sizeof(key)); |
+ if (ReadPlatformFile(files_[i], sizeof(header), key, header.key_length) != |
+ implicit_cast<int>(header.key_length)) { |
+ VLOG(9) << " no key"; |
+ return false; |
+ } |
+ if (std::memcmp(static_cast<const void*>(key_.data()), |
+ static_cast<const void*>(key), |
+ key_.size()) != 0) { |
+ VLOG(9) << " bad key"; |
+ return false; |
+ } |
+ |
+ uint32 check_murmur3; |
+ MurmurHash3_x86_32(static_cast<void*>(key), header.key_length, 0, |
+ &check_murmur3); |
+ if (check_murmur3 != header.key_murmur3) { |
+ VLOG(9) << " bad key hash"; |
+ return false; |
+ } |
+ } |
+ |
+ VLOG(9) << "success we openned it!"; |
+ initialized_ = true; |
+ return true; |
+} |
+ |
+bool VerySimpleSynchronousEntry::InitializeForCreate() { |
+ DCHECK(!initialized_); |
+ if (!OpenOrCreateFiles(true)) |
+ return false; |
+ |
+ for (int i = 0; i < kIndexCount; ++i) { |
+ VerySimpleFileHeader header; |
+ header.initial_magic_number = kVerySimpleInitialMagicNumber; |
+ header.version = kVerySimpleVersion; |
+ |
+ header.key_length = key_.size(); |
+ MurmurHash3_x86_32(static_cast<const void*>(key_.data()), key_.size(), 0, |
+ static_cast<void*>(&header.key_murmur3)); |
+ |
+ if (WritePlatformFile(files_[i], 0, reinterpret_cast<char*>(&header), |
+ sizeof(header)) != sizeof(header)) { |
+ return false; |
+ } |
+ |
+ if (WritePlatformFile(files_[i], sizeof(header), key_.data(), |
+ key_.size()) != implicit_cast<int>(key_.size())) { |
+ return false; |
+ } |
+ } |
+ |
+ initialized_ = true; |
+ return true; |
+} |
+ |
+} // namespace disk_cache |