// 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 "net/http/mock_http_cache.h" #include #include #include #include "base/bind.h" #include "base/location.h" #include "base/single_thread_task_runner.h" #include "base/threading/thread_task_runner_handle.h" #include "net/base/completion_callback.h" #include "net/base/net_errors.h" #include "testing/gtest/include/gtest/gtest.h" namespace net { namespace { // During testing, we are going to limit the size of a cache entry to this many // bytes using DCHECKs in order to prevent a test from causing unbounded memory // growth. In practice cache entry shouldn't come anywhere near this limit for // tests that use the mock cache. If they do, that's likely a problem with the // test. If a test requires using massive cache entries, they should use a real // cache backend instead. const int kMaxMockCacheEntrySize = 100 * 1000 * 1000; // We can override the test mode for a given operation by setting this global // variable. int g_test_mode = 0; int GetTestModeForEntry(const std::string& key) { // 'key' is prefixed with an identifier if it corresponds to a cached POST. // Skip past that to locate the actual URL. // // TODO(darin): It breaks the abstraction a bit that we assume 'key' is an // URL corresponding to a registered MockTransaction. It would be good to // have another way to access the test_mode. GURL url; if (isdigit(key[0])) { size_t slash = key.find('/'); DCHECK(slash != std::string::npos); url = GURL(key.substr(slash + 1)); } else { url = GURL(key); } const MockTransaction* t = FindMockTransaction(url); DCHECK(t); return t->test_mode; } void CallbackForwader(const CompletionCallback& callback, int result) { callback.Run(result); } } // namespace //----------------------------------------------------------------------------- struct MockDiskEntry::CallbackInfo { scoped_refptr entry; CompletionCallback callback; int result; }; MockDiskEntry::MockDiskEntry(const std::string& key) : key_(key), in_memory_data_(0), doomed_(false), sparse_(false), fail_requests_(false), fail_sparse_requests_(false), busy_(false), delayed_(false), cancel_(false), defer_op_(DEFER_NONE), resume_return_code_(0) { test_mode_ = GetTestModeForEntry(key); } void MockDiskEntry::Doom() { doomed_ = true; } void MockDiskEntry::Close() { Release(); } std::string MockDiskEntry::GetKey() const { return key_; } base::Time MockDiskEntry::GetLastUsed() const { return base::Time::Now(); } base::Time MockDiskEntry::GetLastModified() const { return base::Time::Now(); } int32_t MockDiskEntry::GetDataSize(int index) const { DCHECK(index >= 0 && index < kNumCacheEntryDataIndices); return static_cast(data_[index].size()); } int MockDiskEntry::ReadData(int index, int offset, IOBuffer* buf, int buf_len, const CompletionCallback& callback) { DCHECK(index >= 0 && index < kNumCacheEntryDataIndices); DCHECK(!callback.is_null()); if (fail_requests_) return ERR_CACHE_READ_FAILURE; if (offset < 0 || offset > static_cast(data_[index].size())) return ERR_FAILED; if (static_cast(offset) == data_[index].size()) return 0; int num = std::min(buf_len, static_cast(data_[index].size()) - offset); memcpy(buf->data(), &data_[index][offset], num); if (MockHttpCache::GetTestMode(test_mode_) & TEST_MODE_SYNC_CACHE_READ) return num; // Pause and resume. if (defer_op_ == DEFER_READ) { defer_op_ = DEFER_NONE; resume_callback_ = callback; resume_return_code_ = num; return ERR_IO_PENDING; } CallbackLater(callback, num); return ERR_IO_PENDING; } void MockDiskEntry::ResumeDiskEntryOperation() { DCHECK(!resume_callback_.is_null()); CallbackLater(resume_callback_, resume_return_code_); resume_callback_.Reset(); resume_return_code_ = 0; } int MockDiskEntry::WriteData(int index, int offset, IOBuffer* buf, int buf_len, const CompletionCallback& callback, bool truncate) { DCHECK(index >= 0 && index < kNumCacheEntryDataIndices); DCHECK(!callback.is_null()); DCHECK(truncate); if (fail_requests_) { CallbackLater(callback, ERR_CACHE_READ_FAILURE); return ERR_IO_PENDING; } if (offset < 0 || offset > static_cast(data_[index].size())) return ERR_FAILED; DCHECK_LT(offset + buf_len, kMaxMockCacheEntrySize); data_[index].resize(offset + buf_len); if (buf_len) memcpy(&data_[index][offset], buf->data(), buf_len); if (MockHttpCache::GetTestMode(test_mode_) & TEST_MODE_SYNC_CACHE_WRITE) return buf_len; CallbackLater(callback, buf_len); return ERR_IO_PENDING; } int MockDiskEntry::ReadSparseData(int64_t offset, IOBuffer* buf, int buf_len, const CompletionCallback& callback) { DCHECK(!callback.is_null()); if (fail_sparse_requests_) return ERR_NOT_IMPLEMENTED; if (!sparse_ || busy_ || cancel_) return ERR_CACHE_OPERATION_NOT_SUPPORTED; if (offset < 0) return ERR_FAILED; if (fail_requests_) return ERR_CACHE_READ_FAILURE; DCHECK(offset < std::numeric_limits::max()); int real_offset = static_cast(offset); if (!buf_len) return 0; int num = std::min(static_cast(data_[1].size()) - real_offset, buf_len); memcpy(buf->data(), &data_[1][real_offset], num); if (MockHttpCache::GetTestMode(test_mode_) & TEST_MODE_SYNC_CACHE_READ) return num; CallbackLater(callback, num); busy_ = true; delayed_ = false; return ERR_IO_PENDING; } int MockDiskEntry::WriteSparseData(int64_t offset, IOBuffer* buf, int buf_len, const CompletionCallback& callback) { DCHECK(!callback.is_null()); if (fail_sparse_requests_) return ERR_NOT_IMPLEMENTED; if (busy_ || cancel_) return ERR_CACHE_OPERATION_NOT_SUPPORTED; if (!sparse_) { if (data_[1].size()) return ERR_CACHE_OPERATION_NOT_SUPPORTED; sparse_ = true; } if (offset < 0) return ERR_FAILED; if (!buf_len) return 0; if (fail_requests_) return ERR_CACHE_READ_FAILURE; DCHECK(offset < std::numeric_limits::max()); int real_offset = static_cast(offset); if (static_cast(data_[1].size()) < real_offset + buf_len) { DCHECK_LT(real_offset + buf_len, kMaxMockCacheEntrySize); data_[1].resize(real_offset + buf_len); } memcpy(&data_[1][real_offset], buf->data(), buf_len); if (MockHttpCache::GetTestMode(test_mode_) & TEST_MODE_SYNC_CACHE_WRITE) return buf_len; CallbackLater(callback, buf_len); return ERR_IO_PENDING; } int MockDiskEntry::GetAvailableRange(int64_t offset, int len, int64_t* start, const CompletionCallback& callback) { DCHECK(!callback.is_null()); if (!sparse_ || busy_ || cancel_) return ERR_CACHE_OPERATION_NOT_SUPPORTED; if (offset < 0) return ERR_FAILED; if (fail_requests_) return ERR_CACHE_READ_FAILURE; *start = offset; DCHECK(offset < std::numeric_limits::max()); int real_offset = static_cast(offset); if (static_cast(data_[1].size()) < real_offset) return 0; int num = std::min(static_cast(data_[1].size()) - real_offset, len); int count = 0; for (; num > 0; num--, real_offset++) { if (!count) { if (data_[1][real_offset]) { count++; *start = real_offset; } } else { if (!data_[1][real_offset]) break; count++; } } if (MockHttpCache::GetTestMode(test_mode_) & TEST_MODE_SYNC_CACHE_WRITE) return count; CallbackLater(callback, count); return ERR_IO_PENDING; } bool MockDiskEntry::CouldBeSparse() const { if (fail_sparse_requests_) return false; return sparse_; } void MockDiskEntry::CancelSparseIO() { cancel_ = true; } int MockDiskEntry::ReadyForSparseIO(const CompletionCallback& callback) { if (fail_sparse_requests_) return ERR_NOT_IMPLEMENTED; if (!cancel_) return OK; cancel_ = false; DCHECK(!callback.is_null()); if (MockHttpCache::GetTestMode(test_mode_) & TEST_MODE_SYNC_CACHE_READ) return OK; // The pending operation is already in the message loop (and hopefully // already in the second pass). Just notify the caller that it finished. CallbackLater(callback, 0); return ERR_IO_PENDING; } // If |value| is true, don't deliver any completion callbacks until called // again with |value| set to false. Caution: remember to enable callbacks // again or all subsequent tests will fail. // Static. void MockDiskEntry::IgnoreCallbacks(bool value) { if (ignore_callbacks_ == value) return; ignore_callbacks_ = value; if (!value) StoreAndDeliverCallbacks(false, NULL, CompletionCallback(), 0); } MockDiskEntry::~MockDiskEntry() { } // Unlike the callbacks for MockHttpTransaction, we want this one to run even // if the consumer called Close on the MockDiskEntry. We achieve that by // leveraging the fact that this class is reference counted. void MockDiskEntry::CallbackLater(const CompletionCallback& callback, int result) { if (ignore_callbacks_) return StoreAndDeliverCallbacks(true, this, callback, result); base::ThreadTaskRunnerHandle::Get()->PostTask( FROM_HERE, base::Bind(&MockDiskEntry::RunCallback, this, callback, result)); } void MockDiskEntry::RunCallback(const CompletionCallback& callback, int result) { if (busy_) { // This is kind of hacky, but controlling the behavior of just this entry // from a test is sort of complicated. What we really want to do is // delay the delivery of a sparse IO operation a little more so that the // request start operation (async) will finish without seeing the end of // this operation (already posted to the message loop)... and without // just delaying for n mS (which may cause trouble with slow bots). So // we re-post this operation (all async sparse IO operations will take two // trips through the message loop instead of one). if (!delayed_) { delayed_ = true; return CallbackLater(callback, result); } } busy_ = false; callback.Run(result); } // When |store| is true, stores the callback to be delivered later; otherwise // delivers any callback previously stored. // Static. void MockDiskEntry::StoreAndDeliverCallbacks(bool store, MockDiskEntry* entry, const CompletionCallback& callback, int result) { static std::vector callback_list; if (store) { CallbackInfo c = {entry, callback, result}; callback_list.push_back(c); } else { for (size_t i = 0; i < callback_list.size(); i++) { CallbackInfo& c = callback_list[i]; c.entry->CallbackLater(c.callback, c.result); } callback_list.clear(); } } // Statics. bool MockDiskEntry::ignore_callbacks_ = false; //----------------------------------------------------------------------------- MockDiskCache::MockDiskCache() : open_count_(0), create_count_(0), doomed_count_(0), fail_requests_(false), soft_failures_(false), double_create_check_(true), fail_sparse_requests_(false), support_in_memory_entry_data_(true), defer_op_(MockDiskEntry::DEFER_NONE), resume_return_code_(0) {} MockDiskCache::~MockDiskCache() { ReleaseAll(); } CacheType MockDiskCache::GetCacheType() const { return DISK_CACHE; } int32_t MockDiskCache::GetEntryCount() const { return static_cast(entries_.size()); } int MockDiskCache::OpenEntry(const std::string& key, disk_cache::Entry** entry, const CompletionCallback& callback) { DCHECK(!callback.is_null()); if (fail_requests_) return ERR_CACHE_OPEN_FAILURE; EntryMap::iterator it = entries_.find(key); if (it == entries_.end()) return ERR_CACHE_OPEN_FAILURE; if (it->second->is_doomed()) { it->second->Release(); entries_.erase(it); return ERR_CACHE_OPEN_FAILURE; } open_count_++; it->second->AddRef(); *entry = it->second; if (soft_failures_) it->second->set_fail_requests(); if (GetTestModeForEntry(key) & TEST_MODE_SYNC_CACHE_START) return OK; CallbackLater(callback, OK); return ERR_IO_PENDING; } int MockDiskCache::CreateEntry(const std::string& key, disk_cache::Entry** entry, const CompletionCallback& callback) { DCHECK(!callback.is_null()); if (fail_requests_) return ERR_CACHE_CREATE_FAILURE; EntryMap::iterator it = entries_.find(key); if (it != entries_.end()) { if (!it->second->is_doomed()) { if (double_create_check_) NOTREACHED(); else return ERR_CACHE_CREATE_FAILURE; } it->second->Release(); entries_.erase(it); } create_count_++; MockDiskEntry* new_entry = new MockDiskEntry(key); new_entry->AddRef(); entries_[key] = new_entry; new_entry->AddRef(); *entry = new_entry; if (soft_failures_) new_entry->set_fail_requests(); if (fail_sparse_requests_) new_entry->set_fail_sparse_requests(); if (GetTestModeForEntry(key) & TEST_MODE_SYNC_CACHE_START) return OK; // Pause and resume. if (defer_op_ == MockDiskEntry::DEFER_CREATE) { defer_op_ = MockDiskEntry::DEFER_NONE; resume_callback_ = callback; resume_return_code_ = OK; return ERR_IO_PENDING; } CallbackLater(callback, OK); return ERR_IO_PENDING; } int MockDiskCache::DoomEntry(const std::string& key, const CompletionCallback& callback) { DCHECK(!callback.is_null()); EntryMap::iterator it = entries_.find(key); if (it != entries_.end()) { it->second->Release(); entries_.erase(it); doomed_count_++; } if (GetTestModeForEntry(key) & TEST_MODE_SYNC_CACHE_START) return OK; CallbackLater(callback, OK); return ERR_IO_PENDING; } int MockDiskCache::DoomAllEntries(const CompletionCallback& callback) { return ERR_NOT_IMPLEMENTED; } int MockDiskCache::DoomEntriesBetween(const base::Time initial_time, const base::Time end_time, const CompletionCallback& callback) { return ERR_NOT_IMPLEMENTED; } int MockDiskCache::DoomEntriesSince(const base::Time initial_time, const CompletionCallback& callback) { return ERR_NOT_IMPLEMENTED; } int MockDiskCache::CalculateSizeOfAllEntries( const CompletionCallback& callback) { return ERR_NOT_IMPLEMENTED; } class MockDiskCache::NotImplementedIterator : public Iterator { public: int OpenNextEntry(disk_cache::Entry** next_entry, const CompletionCallback& callback) override { return ERR_NOT_IMPLEMENTED; } }; std::unique_ptr MockDiskCache::CreateIterator() { return std::unique_ptr(new NotImplementedIterator()); } void MockDiskCache::GetStats(base::StringPairs* stats) { } void MockDiskCache::OnExternalCacheHit(const std::string& key) { } size_t MockDiskCache::DumpMemoryStats( base::trace_event::ProcessMemoryDump* pmd, const std::string& parent_absolute_name) const { return 0u; } uint8_t MockDiskCache::GetEntryInMemoryData(const std::string& key) { if (!support_in_memory_entry_data_) return 0; EntryMap::iterator it = entries_.find(key); if (it != entries_.end()) return it->second->in_memory_data(); return 0; } void MockDiskCache::SetEntryInMemoryData(const std::string& key, uint8_t data) { EntryMap::iterator it = entries_.find(key); if (it != entries_.end()) it->second->set_in_memory_data(data); } void MockDiskCache::ReleaseAll() { for (auto entry : entries_) entry.second->Release(); entries_.clear(); } void MockDiskCache::CallbackLater(const CompletionCallback& callback, int result) { base::ThreadTaskRunnerHandle::Get()->PostTask( FROM_HERE, base::Bind(&CallbackForwader, callback, result)); } bool MockDiskCache::IsDiskEntryDoomed(const std::string& key) { auto it = entries_.find(key); if (it != entries_.end()) return it->second->is_doomed(); return false; } void MockDiskCache::ResumeCacheOperation() { DCHECK(!resume_callback_.is_null()); CallbackLater(resume_callback_, resume_return_code_); resume_callback_.Reset(); resume_return_code_ = 0; } scoped_refptr MockDiskCache::GetDiskEntryRef( const std::string& key) { auto it = entries_.find(key); if (it == entries_.end()) return nullptr; return it->second; } //----------------------------------------------------------------------------- int MockBackendFactory::CreateBackend( NetLog* net_log, std::unique_ptr* backend, const CompletionCallback& callback) { backend->reset(new MockDiskCache()); return OK; } //----------------------------------------------------------------------------- MockHttpCache::MockHttpCache() : MockHttpCache(false) {} MockHttpCache::MockHttpCache( std::unique_ptr disk_cache_factory) : MockHttpCache(std::move(disk_cache_factory), false) {} MockHttpCache::MockHttpCache(bool is_main_cache) : MockHttpCache(std::make_unique(), is_main_cache) {} MockHttpCache::MockHttpCache( std::unique_ptr disk_cache_factory, bool is_main_cache) : http_cache_(std::make_unique(), std::move(disk_cache_factory), is_main_cache) {} disk_cache::Backend* MockHttpCache::backend() { TestCompletionCallback cb; disk_cache::Backend* backend; int rv = http_cache_.GetBackend(&backend, cb.callback()); rv = cb.GetResult(rv); return (rv == OK) ? backend : NULL; } MockDiskCache* MockHttpCache::disk_cache() { return static_cast(backend()); } int MockHttpCache::CreateTransaction(std::unique_ptr* trans) { return http_cache_.CreateTransaction(DEFAULT_PRIORITY, trans); } void MockHttpCache::SimulateCacheLockTimeout() { http_cache_.SimulateCacheLockTimeoutForTesting(); } void MockHttpCache::SimulateCacheLockTimeoutAfterHeaders() { http_cache_.SimulateCacheLockTimeoutAfterHeadersForTesting(); } void MockHttpCache::FailConditionalizations() { http_cache_.FailConditionalizationForTest(); } bool MockHttpCache::ReadResponseInfo(disk_cache::Entry* disk_entry, HttpResponseInfo* response_info, bool* response_truncated) { int size = disk_entry->GetDataSize(0); TestCompletionCallback cb; scoped_refptr buffer(new IOBuffer(size)); int rv = disk_entry->ReadData(0, 0, buffer.get(), size, cb.callback()); rv = cb.GetResult(rv); EXPECT_EQ(size, rv); return HttpCache::ParseResponseInfo(buffer->data(), size, response_info, response_truncated); } bool MockHttpCache::WriteResponseInfo(disk_cache::Entry* disk_entry, const HttpResponseInfo* response_info, bool skip_transient_headers, bool response_truncated) { base::Pickle pickle; response_info->Persist( &pickle, skip_transient_headers, response_truncated); TestCompletionCallback cb; scoped_refptr data( new WrappedIOBuffer(reinterpret_cast(pickle.data()))); int len = static_cast(pickle.size()); int rv = disk_entry->WriteData(0, 0, data.get(), len, cb.callback(), true); rv = cb.GetResult(rv); return (rv == len); } bool MockHttpCache::OpenBackendEntry(const std::string& key, disk_cache::Entry** entry) { TestCompletionCallback cb; int rv = backend()->OpenEntry(key, entry, cb.callback()); return (cb.GetResult(rv) == OK); } bool MockHttpCache::CreateBackendEntry(const std::string& key, disk_cache::Entry** entry, NetLog* net_log) { TestCompletionCallback cb; int rv = backend()->CreateEntry(key, entry, cb.callback()); return (cb.GetResult(rv) == OK); } // Static. int MockHttpCache::GetTestMode(int test_mode) { if (!g_test_mode) return test_mode; return g_test_mode; } // Static. void MockHttpCache::SetTestMode(int test_mode) { g_test_mode = test_mode; } bool MockHttpCache::IsWriterPresent(const std::string& key) { HttpCache::ActiveEntry* entry = http_cache_.FindActiveEntry(key); if (entry) return entry->writer; return false; } bool MockHttpCache::IsHeadersTransactionPresent(const std::string& key) { HttpCache::ActiveEntry* entry = http_cache_.FindActiveEntry(key); if (entry) return entry->headers_transaction; return false; } int MockHttpCache::GetCountReaders(const std::string& key) { HttpCache::ActiveEntry* entry = http_cache_.FindActiveEntry(key); if (entry) return entry->readers.size(); return false; } int MockHttpCache::GetCountAddToEntryQueue(const std::string& key) { HttpCache::ActiveEntry* entry = http_cache_.FindActiveEntry(key); if (entry) return entry->add_to_entry_queue.size(); return false; } int MockHttpCache::GetCountDoneHeadersQueue(const std::string& key) { HttpCache::ActiveEntry* entry = http_cache_.FindActiveEntry(key); if (entry) return entry->done_headers_queue.size(); return false; } //----------------------------------------------------------------------------- int MockDiskCacheNoCB::CreateEntry(const std::string& key, disk_cache::Entry** entry, const CompletionCallback& callback) { return ERR_IO_PENDING; } //----------------------------------------------------------------------------- int MockBackendNoCbFactory::CreateBackend( NetLog* net_log, std::unique_ptr* backend, const CompletionCallback& callback) { backend->reset(new MockDiskCacheNoCB()); return OK; } //----------------------------------------------------------------------------- MockBlockingBackendFactory::MockBlockingBackendFactory() : backend_(NULL), block_(true), fail_(false) { } MockBlockingBackendFactory::~MockBlockingBackendFactory() { } int MockBlockingBackendFactory::CreateBackend( NetLog* net_log, std::unique_ptr* backend, const CompletionCallback& callback) { if (!block_) { if (!fail_) backend->reset(new MockDiskCache()); return Result(); } backend_ = backend; callback_ = callback; return ERR_IO_PENDING; } void MockBlockingBackendFactory::FinishCreation() { block_ = false; if (!callback_.is_null()) { if (!fail_) backend_->reset(new MockDiskCache()); CompletionCallback cb = callback_; callback_.Reset(); cb.Run(Result()); // This object can be deleted here. } } } // namespace net