OLD | NEW |
1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "net/http/http_cache.h" | 5 #include "net/http/http_cache.h" |
6 | 6 |
7 #include <algorithm> | 7 #include <algorithm> |
8 | 8 |
9 #include "base/compiler_specific.h" | 9 #include "base/compiler_specific.h" |
10 | 10 |
11 #if defined(OS_POSIX) | 11 #if defined(OS_POSIX) |
12 #include <unistd.h> | 12 #include <unistd.h> |
13 #endif | 13 #endif |
14 | 14 |
15 #include "base/callback.h" | 15 #include "base/callback.h" |
16 #include "base/format_macros.h" | 16 #include "base/format_macros.h" |
17 #include "base/message_loop.h" | 17 #include "base/message_loop.h" |
18 #include "base/pickle.h" | 18 #include "base/pickle.h" |
19 #include "base/ref_counted.h" | 19 #include "base/ref_counted.h" |
| 20 #include "base/stl_util-inl.h" |
20 #include "base/string_util.h" | 21 #include "base/string_util.h" |
21 #include "net/base/io_buffer.h" | 22 #include "net/base/io_buffer.h" |
22 #include "net/base/load_flags.h" | 23 #include "net/base/load_flags.h" |
23 #include "net/base/net_errors.h" | 24 #include "net/base/net_errors.h" |
24 #include "net/disk_cache/disk_cache.h" | 25 #include "net/disk_cache/disk_cache.h" |
25 #include "net/http/http_cache_transaction.h" | 26 #include "net/http/http_cache_transaction.h" |
26 #include "net/http/http_network_layer.h" | 27 #include "net/http/http_network_layer.h" |
27 #include "net/http/http_network_session.h" | 28 #include "net/http/http_network_session.h" |
28 #include "net/http/http_request_info.h" | 29 #include "net/http/http_request_info.h" |
29 #include "net/http/http_response_headers.h" | 30 #include "net/http/http_response_headers.h" |
(...skipping 12 matching lines...) Expand all Loading... |
42 } | 43 } |
43 | 44 |
44 HttpCache::ActiveEntry::~ActiveEntry() { | 45 HttpCache::ActiveEntry::~ActiveEntry() { |
45 if (disk_entry) | 46 if (disk_entry) |
46 disk_entry->Close(); | 47 disk_entry->Close(); |
47 } | 48 } |
48 | 49 |
49 //----------------------------------------------------------------------------- | 50 //----------------------------------------------------------------------------- |
50 | 51 |
51 // This structure keeps track of work items that are attempting to create or | 52 // This structure keeps track of work items that are attempting to create or |
52 // open cache entries. | 53 // open cache entries or the backend itself. |
53 struct HttpCache::NewEntry { | 54 struct HttpCache::PendingOp { |
54 NewEntry() : disk_entry(NULL), writer(NULL) {} | 55 PendingOp() : disk_entry(NULL), writer(NULL), callback(NULL) {} |
55 ~NewEntry() {} | 56 ~PendingOp() {} |
56 | 57 |
57 disk_cache::Entry* disk_entry; | 58 disk_cache::Entry* disk_entry; |
58 WorkItem* writer; | 59 WorkItem* writer; |
| 60 CompletionCallback* callback; // BackendCallback. |
59 WorkItemList pending_queue; | 61 WorkItemList pending_queue; |
60 }; | 62 }; |
61 | 63 |
62 //----------------------------------------------------------------------------- | 64 //----------------------------------------------------------------------------- |
63 | 65 |
64 // The type of operation represented by a work item. | 66 // The type of operation represented by a work item. |
65 enum WorkItemOperation { | 67 enum WorkItemOperation { |
| 68 WI_CREATE_BACKEND, |
66 WI_OPEN_ENTRY, | 69 WI_OPEN_ENTRY, |
67 WI_CREATE_ENTRY, | 70 WI_CREATE_ENTRY, |
68 WI_DOOM_ENTRY | 71 WI_DOOM_ENTRY |
69 }; | 72 }; |
70 | 73 |
71 // A work item encapsulates a single request for cache entry with all the | 74 // A work item encapsulates a single request to the backend with all the |
72 // information needed to complete that request. | 75 // information needed to complete that request. |
73 class HttpCache::WorkItem { | 76 class HttpCache::WorkItem { |
74 public: | 77 public: |
75 WorkItem(ActiveEntry** entry, Transaction* trans, WorkItemOperation operation) | 78 WorkItem(WorkItemOperation operation, Transaction* trans, ActiveEntry** entry) |
76 : entry_(entry), trans_(trans), operation_(operation) {} | 79 : operation_(operation), trans_(trans), entry_(entry), callback_(NULL), |
| 80 backend_(NULL) {} |
| 81 WorkItem(WorkItemOperation operation, Transaction* trans, |
| 82 CompletionCallback* cb, disk_cache::Backend** backend) |
| 83 : operation_(operation), trans_(trans), entry_(NULL), callback_(cb), |
| 84 backend_(backend) {} |
77 ~WorkItem() {} | 85 ~WorkItem() {} |
78 WorkItemOperation operation() { return operation_; } | |
79 | 86 |
80 // Calls back the transaction with the result of the operation. | 87 // Calls back the transaction with the result of the operation. |
81 void NotifyTransaction(int result, ActiveEntry* entry) { | 88 void NotifyTransaction(int result, ActiveEntry* entry) { |
82 if (entry_) | 89 if (entry_) |
83 *entry_ = entry; | 90 *entry_ = entry; |
84 if (trans_) | 91 if (trans_) |
85 trans_->io_callback()->Run(result); | 92 trans_->io_callback()->Run(result); |
86 } | 93 } |
87 | 94 |
| 95 // Notifies the caller about the operation completion. |
| 96 void DoCallback(int result, disk_cache::Backend* backend) { |
| 97 if (backend_) |
| 98 *backend_ = backend; |
| 99 if (callback_) |
| 100 callback_->Run(result); |
| 101 } |
| 102 |
| 103 WorkItemOperation operation() { return operation_; } |
88 void ClearTransaction() { trans_ = NULL; } | 104 void ClearTransaction() { trans_ = NULL; } |
89 void ClearEntry() { entry_ = NULL; } | 105 void ClearEntry() { entry_ = NULL; } |
| 106 void ClearCallback() { callback_ = NULL; } |
90 bool Matches(Transaction* trans) const { return trans == trans_; } | 107 bool Matches(Transaction* trans) const { return trans == trans_; } |
91 bool IsValid() const { return trans_ || entry_; } | 108 bool IsValid() const { return trans_ || entry_ || callback_; } |
92 | 109 |
93 private: | 110 private: |
| 111 WorkItemOperation operation_; |
| 112 Transaction* trans_; |
94 ActiveEntry** entry_; | 113 ActiveEntry** entry_; |
95 Transaction* trans_; | 114 CompletionCallback* callback_; // User callback. |
96 WorkItemOperation operation_; | 115 disk_cache::Backend** backend_; |
97 }; | 116 }; |
98 | 117 |
99 //----------------------------------------------------------------------------- | 118 //----------------------------------------------------------------------------- |
100 | 119 |
101 // This class is a specialized type of CompletionCallback that allows us to | 120 // This class is a specialized type of CompletionCallback that allows us to |
102 // pass multiple arguments to the completion routine. | 121 // pass multiple arguments to the completion routine. |
103 class HttpCache::BackendCallback : public CallbackRunner<Tuple1<int> > { | 122 class HttpCache::BackendCallback : public CallbackRunner<Tuple1<int> > { |
104 public: | 123 public: |
105 BackendCallback(HttpCache* cache, NewEntry* entry) | 124 BackendCallback(HttpCache* cache, PendingOp* pending_op) |
106 : cache_(cache), entry_(entry) {} | 125 : cache_(cache), pending_op_(pending_op) {} |
107 ~BackendCallback() {} | 126 ~BackendCallback() {} |
108 | 127 |
109 virtual void RunWithParams(const Tuple1<int>& params) { | 128 virtual void RunWithParams(const Tuple1<int>& params) { |
110 cache_->OnIOComplete(params.a, entry_); | 129 cache_->OnIOComplete(params.a, pending_op_); |
111 delete this; | 130 delete this; |
112 } | 131 } |
113 | 132 |
114 private: | 133 private: |
115 HttpCache* cache_; | 134 HttpCache* cache_; |
116 NewEntry* entry_; | 135 PendingOp* pending_op_; |
117 DISALLOW_COPY_AND_ASSIGN(BackendCallback); | 136 DISALLOW_COPY_AND_ASSIGN(BackendCallback); |
118 }; | 137 }; |
119 | 138 |
120 //----------------------------------------------------------------------------- | 139 //----------------------------------------------------------------------------- |
121 | 140 |
122 // This class encapsulates a transaction whose only purpose is to write metadata | 141 // This class encapsulates a transaction whose only purpose is to write metadata |
123 // to a given entry. | 142 // to a given entry. |
124 class HttpCache::MetadataWriter { | 143 class HttpCache::MetadataWriter { |
125 public: | 144 public: |
126 explicit MetadataWriter(HttpCache::Transaction* trans) | 145 explicit MetadataWriter(HttpCache::Transaction* trans) |
(...skipping 71 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
198 HttpCache::HttpCache(NetworkChangeNotifier* network_change_notifier, | 217 HttpCache::HttpCache(NetworkChangeNotifier* network_change_notifier, |
199 HostResolver* host_resolver, | 218 HostResolver* host_resolver, |
200 ProxyService* proxy_service, | 219 ProxyService* proxy_service, |
201 SSLConfigService* ssl_config_service, | 220 SSLConfigService* ssl_config_service, |
202 HttpAuthHandlerFactory* http_auth_handler_factory, | 221 HttpAuthHandlerFactory* http_auth_handler_factory, |
203 const FilePath& cache_dir, | 222 const FilePath& cache_dir, |
204 MessageLoop* cache_thread, | 223 MessageLoop* cache_thread, |
205 int cache_size) | 224 int cache_size) |
206 : disk_cache_dir_(cache_dir), | 225 : disk_cache_dir_(cache_dir), |
207 cache_thread_(cache_thread), | 226 cache_thread_(cache_thread), |
| 227 temp_backend_(NULL), |
| 228 building_backend_(false), |
208 mode_(NORMAL), | 229 mode_(NORMAL), |
209 type_(DISK_CACHE), | 230 type_(DISK_CACHE), |
210 network_layer_(HttpNetworkLayer::CreateFactory( | 231 network_layer_(HttpNetworkLayer::CreateFactory( |
211 network_change_notifier, host_resolver, proxy_service, | 232 network_change_notifier, host_resolver, proxy_service, |
212 ssl_config_service, http_auth_handler_factory)), | 233 ssl_config_service, http_auth_handler_factory)), |
213 ALLOW_THIS_IN_INITIALIZER_LIST(task_factory_(this)), | 234 ALLOW_THIS_IN_INITIALIZER_LIST(task_factory_(this)), |
214 enable_range_support_(true), | 235 enable_range_support_(true), |
215 cache_size_(cache_size) { | 236 cache_size_(cache_size), |
| 237 create_backend_fn_(NULL) { |
216 } | 238 } |
217 | 239 |
218 HttpCache::HttpCache(HttpNetworkSession* session, | 240 HttpCache::HttpCache(HttpNetworkSession* session, |
219 const FilePath& cache_dir, | 241 const FilePath& cache_dir, |
220 MessageLoop* cache_thread, | 242 MessageLoop* cache_thread, |
221 int cache_size) | 243 int cache_size) |
222 : disk_cache_dir_(cache_dir), | 244 : disk_cache_dir_(cache_dir), |
223 cache_thread_(cache_thread), | 245 cache_thread_(cache_thread), |
| 246 temp_backend_(NULL), |
| 247 building_backend_(false), |
224 mode_(NORMAL), | 248 mode_(NORMAL), |
225 type_(DISK_CACHE), | 249 type_(DISK_CACHE), |
226 network_layer_(HttpNetworkLayer::CreateFactory(session)), | 250 network_layer_(HttpNetworkLayer::CreateFactory(session)), |
227 ALLOW_THIS_IN_INITIALIZER_LIST(task_factory_(this)), | 251 ALLOW_THIS_IN_INITIALIZER_LIST(task_factory_(this)), |
228 enable_range_support_(true), | 252 enable_range_support_(true), |
229 cache_size_(cache_size) { | 253 cache_size_(cache_size), |
| 254 create_backend_fn_(NULL) { |
230 } | 255 } |
231 | 256 |
232 HttpCache::HttpCache(NetworkChangeNotifier* network_change_notifier, | 257 HttpCache::HttpCache(NetworkChangeNotifier* network_change_notifier, |
233 HostResolver* host_resolver, | 258 HostResolver* host_resolver, |
234 ProxyService* proxy_service, | 259 ProxyService* proxy_service, |
235 SSLConfigService* ssl_config_service, | 260 SSLConfigService* ssl_config_service, |
236 HttpAuthHandlerFactory* http_auth_handler_factory, | 261 HttpAuthHandlerFactory* http_auth_handler_factory, |
237 int cache_size) | 262 int cache_size) |
238 : mode_(NORMAL), | 263 : cache_thread_(NULL), |
| 264 temp_backend_(NULL), |
| 265 building_backend_(false), |
| 266 mode_(NORMAL), |
239 type_(MEMORY_CACHE), | 267 type_(MEMORY_CACHE), |
240 network_layer_(HttpNetworkLayer::CreateFactory( | 268 network_layer_(HttpNetworkLayer::CreateFactory( |
241 network_change_notifier, host_resolver, proxy_service, | 269 network_change_notifier, host_resolver, proxy_service, |
242 ssl_config_service, http_auth_handler_factory)), | 270 ssl_config_service, http_auth_handler_factory)), |
243 ALLOW_THIS_IN_INITIALIZER_LIST(task_factory_(this)), | 271 ALLOW_THIS_IN_INITIALIZER_LIST(task_factory_(this)), |
244 enable_range_support_(true), | 272 enable_range_support_(true), |
245 cache_size_(cache_size) { | 273 cache_size_(cache_size), |
| 274 create_backend_fn_(NULL) { |
246 } | 275 } |
247 | 276 |
248 HttpCache::HttpCache(HttpTransactionFactory* network_layer, | 277 HttpCache::HttpCache(HttpTransactionFactory* network_layer, |
249 disk_cache::Backend* disk_cache) | 278 disk_cache::Backend* disk_cache) |
250 : mode_(NORMAL), | 279 : cache_thread_(NULL), |
| 280 temp_backend_(NULL), |
| 281 building_backend_(false), |
| 282 mode_(NORMAL), |
251 type_(DISK_CACHE), | 283 type_(DISK_CACHE), |
252 network_layer_(network_layer), | 284 network_layer_(network_layer), |
253 disk_cache_(disk_cache), | 285 disk_cache_(disk_cache), |
254 ALLOW_THIS_IN_INITIALIZER_LIST(task_factory_(this)), | 286 ALLOW_THIS_IN_INITIALIZER_LIST(task_factory_(this)), |
255 enable_range_support_(true), | 287 enable_range_support_(true), |
256 cache_size_(0) { | 288 cache_size_(0), |
| 289 create_backend_fn_(NULL) { |
257 } | 290 } |
258 | 291 |
259 HttpCache::~HttpCache() { | 292 HttpCache::~HttpCache() { |
260 // If we have any active entries remaining, then we need to deactivate them. | 293 // If we have any active entries remaining, then we need to deactivate them. |
261 // We may have some pending calls to OnProcessPendingQueue, but since those | 294 // We may have some pending calls to OnProcessPendingQueue, but since those |
262 // won't run (due to our destruction), we can simply ignore the corresponding | 295 // won't run (due to our destruction), we can simply ignore the corresponding |
263 // will_process_pending_queue flag. | 296 // will_process_pending_queue flag. |
264 while (!active_entries_.empty()) { | 297 while (!active_entries_.empty()) { |
265 ActiveEntry* entry = active_entries_.begin()->second; | 298 ActiveEntry* entry = active_entries_.begin()->second; |
266 entry->will_process_pending_queue = false; | 299 entry->will_process_pending_queue = false; |
267 entry->pending_queue.clear(); | 300 entry->pending_queue.clear(); |
268 entry->readers.clear(); | 301 entry->readers.clear(); |
269 entry->writer = NULL; | 302 entry->writer = NULL; |
270 DeactivateEntry(entry); | 303 DeactivateEntry(entry); |
271 } | 304 } |
272 | 305 |
273 ActiveEntriesSet::iterator it = doomed_entries_.begin(); | 306 STLDeleteElements(&doomed_entries_); |
274 for (; it != doomed_entries_.end(); ++it) | 307 |
275 delete *it; | 308 PendingOpsMap::iterator pending_it = pending_ops_.begin(); |
| 309 for (; pending_it != pending_ops_.end(); ++pending_it) { |
| 310 // We are not notifying the transactions about the cache going away, even |
| 311 // though they are waiting for a callback that will never fire. |
| 312 PendingOp* pending_op = pending_it->second; |
| 313 delete pending_op->writer; |
| 314 delete pending_op->callback; |
| 315 |
| 316 STLDeleteElements(&pending_op->pending_queue); |
| 317 delete pending_op; |
| 318 } |
276 } | 319 } |
277 | 320 |
278 disk_cache::Backend* HttpCache::GetBackend() { | 321 disk_cache::Backend* HttpCache::GetBackend() { |
279 if (disk_cache_.get()) | 322 if (disk_cache_.get()) |
280 return disk_cache_.get(); | 323 return disk_cache_.get(); |
281 | 324 |
282 DCHECK_GE(cache_size_, 0); | 325 DCHECK_GE(cache_size_, 0); |
283 if (type_ == MEMORY_CACHE) { | 326 if (type_ == MEMORY_CACHE) { |
284 // We may end up with no folder name and no cache if the initialization | 327 // We may end up with no folder name and no cache if the initialization |
285 // of the disk cache fails. We want to be sure that what we wanted to have | 328 // of the disk cache fails. We want to be sure that what we wanted to have |
286 // was an in-memory cache. | 329 // was an in-memory cache. |
287 disk_cache_.reset(disk_cache::CreateInMemoryCacheBackend(cache_size_)); | 330 disk_cache_.reset(disk_cache::CreateInMemoryCacheBackend(cache_size_)); |
288 } else if (!disk_cache_dir_.empty()) { | 331 } else if (!disk_cache_dir_.empty()) { |
289 disk_cache_.reset(disk_cache::CreateCacheBackend(disk_cache_dir_, true, | 332 disk_cache_.reset(disk_cache::CreateCacheBackend(disk_cache_dir_, true, |
290 cache_size_, type_)); | 333 cache_size_, type_)); |
291 disk_cache_dir_ = FilePath(); // Reclaim memory. | 334 disk_cache_dir_ = FilePath(); // Reclaim memory. |
292 } | 335 } |
293 return disk_cache_.get(); | 336 return disk_cache_.get(); |
294 } | 337 } |
295 | 338 |
296 int HttpCache::GetBackend(disk_cache::Backend** backend, | 339 int HttpCache::GetBackend(disk_cache::Backend** backend, |
297 CompletionCallback* callback) { | 340 CompletionCallback* callback) { |
298 DCHECK(callback != NULL); | 341 DCHECK(callback != NULL); |
299 *backend = GetBackend(); | 342 |
300 return OK; | 343 if (disk_cache_.get()) { |
| 344 *backend = disk_cache_.get(); |
| 345 return OK; |
| 346 } |
| 347 |
| 348 DCHECK_GE(cache_size_, 0); |
| 349 return CreateBackend(backend, callback); |
301 } | 350 } |
302 | 351 |
303 int HttpCache::CreateTransaction(scoped_ptr<HttpTransaction>* trans) { | 352 int HttpCache::CreateTransaction(scoped_ptr<HttpTransaction>* trans) { |
304 // Do lazy initialization of disk cache if needed. | 353 // Do lazy initialization of disk cache if needed. |
305 GetBackend(); | 354 if (!disk_cache_.get()) |
| 355 CreateBackend(NULL, NULL); // We don't care about the result. |
| 356 |
306 trans->reset(new HttpCache::Transaction(this, enable_range_support_)); | 357 trans->reset(new HttpCache::Transaction(this, enable_range_support_)); |
307 return OK; | 358 return OK; |
308 } | 359 } |
309 | 360 |
310 HttpCache* HttpCache::GetCache() { | 361 HttpCache* HttpCache::GetCache() { |
311 return this; | 362 return this; |
312 } | 363 } |
313 | 364 |
314 HttpNetworkSession* HttpCache::GetSession() { | 365 HttpNetworkSession* HttpCache::GetSession() { |
315 net::HttpNetworkLayer* network = | 366 net::HttpNetworkLayer* network = |
(...skipping 69 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
385 if (session) { | 436 if (session) { |
386 session->tcp_socket_pool()->CloseIdleSockets(); | 437 session->tcp_socket_pool()->CloseIdleSockets(); |
387 if (session->spdy_session_pool()) | 438 if (session->spdy_session_pool()) |
388 session->spdy_session_pool()->CloseAllSessions(); | 439 session->spdy_session_pool()->CloseAllSessions(); |
389 session->ReplaceTCPSocketPool(); | 440 session->ReplaceTCPSocketPool(); |
390 } | 441 } |
391 } | 442 } |
392 | 443 |
393 //----------------------------------------------------------------------------- | 444 //----------------------------------------------------------------------------- |
394 | 445 |
| 446 int HttpCache::CreateBackend(disk_cache::Backend** backend, |
| 447 CompletionCallback* callback) { |
| 448 // We may end up with no folder name and no cache if the initialization |
| 449 // of the disk cache fails. |
| 450 if (type_ != MEMORY_CACHE && disk_cache_dir_.empty()) |
| 451 return ERR_FAILED; |
| 452 |
| 453 DCHECK_GE(cache_size_, 0); |
| 454 building_backend_ = true; |
| 455 |
| 456 scoped_ptr<WorkItem> item(new WorkItem(WI_CREATE_BACKEND, NULL, callback, |
| 457 backend)); |
| 458 |
| 459 // This is the only operation that we can do that is not related to any given |
| 460 // entry, so we use an empty key for it. |
| 461 PendingOp* pending_op = GetPendingOp(""); |
| 462 if (pending_op->writer) { |
| 463 if (callback) |
| 464 pending_op->pending_queue.push_back(item.release()); |
| 465 return ERR_IO_PENDING; |
| 466 } |
| 467 |
| 468 DCHECK(pending_op->pending_queue.empty()); |
| 469 |
| 470 pending_op->writer = item.release(); |
| 471 BackendCallback* my_callback = new BackendCallback(this, pending_op); |
| 472 pending_op->callback = my_callback; |
| 473 |
| 474 // See if this is a unit test. TODO(rvargas): Cleanup this after removing the |
| 475 // overloaded method. |
| 476 int rv; |
| 477 if (create_backend_fn_) { |
| 478 rv = create_backend_fn_(type_, disk_cache_dir_, cache_size_, true, |
| 479 cache_thread_, &temp_backend_, my_callback); |
| 480 } else { |
| 481 rv = disk_cache::CreateCacheBackend(type_, disk_cache_dir_, cache_size_, |
| 482 true, cache_thread_, &temp_backend_, |
| 483 my_callback); |
| 484 } |
| 485 |
| 486 if (rv != ERR_IO_PENDING) { |
| 487 pending_op->writer->ClearCallback(); |
| 488 my_callback->Run(rv); |
| 489 } |
| 490 |
| 491 return rv; |
| 492 } |
| 493 |
| 494 int HttpCache::GetBackendForTransaction(Transaction* trans) { |
| 495 if (disk_cache_.get()) |
| 496 return OK; |
| 497 |
| 498 if (!building_backend_) |
| 499 return ERR_FAILED; |
| 500 |
| 501 WorkItem* item = new WorkItem(WI_CREATE_BACKEND, trans, NULL, NULL); |
| 502 PendingOp* pending_op = GetPendingOp(""); |
| 503 DCHECK(pending_op->writer); |
| 504 pending_op->pending_queue.push_back(item); |
| 505 return ERR_IO_PENDING; |
| 506 } |
| 507 |
395 // Generate a key that can be used inside the cache. | 508 // Generate a key that can be used inside the cache. |
396 std::string HttpCache::GenerateCacheKey(const HttpRequestInfo* request) { | 509 std::string HttpCache::GenerateCacheKey(const HttpRequestInfo* request) { |
397 // Strip out the reference, username, and password sections of the URL. | 510 // Strip out the reference, username, and password sections of the URL. |
398 std::string url = HttpUtil::SpecForRequest(request->url); | 511 std::string url = HttpUtil::SpecForRequest(request->url); |
399 | 512 |
400 DCHECK(mode_ != DISABLE); | 513 DCHECK(mode_ != DISABLE); |
401 if (mode_ == NORMAL) { | 514 if (mode_ == NORMAL) { |
402 // No valid URL can begin with numerals, so we should not have to worry | 515 // No valid URL can begin with numerals, so we should not have to worry |
403 // about collisions with normal URLs. | 516 // about collisions with normal URLs. |
404 if (request->upload_data && request->upload_data->identifier()) { | 517 if (request->upload_data && request->upload_data->identifier()) { |
(...skipping 45 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
450 | 563 |
451 entry->disk_entry->Doom(); | 564 entry->disk_entry->Doom(); |
452 entry->doomed = true; | 565 entry->doomed = true; |
453 | 566 |
454 DCHECK(entry->writer || !entry->readers.empty()); | 567 DCHECK(entry->writer || !entry->readers.empty()); |
455 return OK; | 568 return OK; |
456 } | 569 } |
457 | 570 |
458 int HttpCache::AsyncDoomEntry(const std::string& key, Transaction* trans) { | 571 int HttpCache::AsyncDoomEntry(const std::string& key, Transaction* trans) { |
459 DCHECK(trans); | 572 DCHECK(trans); |
460 WorkItem* item = new WorkItem(NULL, trans, WI_DOOM_ENTRY); | 573 WorkItem* item = new WorkItem(WI_DOOM_ENTRY, trans, NULL); |
461 NewEntry* new_entry = GetNewEntry(key); | 574 PendingOp* pending_op = GetPendingOp(key); |
462 if (new_entry->writer) { | 575 if (pending_op->writer) { |
463 new_entry->pending_queue.push_back(item); | 576 pending_op->pending_queue.push_back(item); |
464 return ERR_IO_PENDING; | 577 return ERR_IO_PENDING; |
465 } | 578 } |
466 | 579 |
467 DCHECK(new_entry->pending_queue.empty()); | 580 DCHECK(pending_op->pending_queue.empty()); |
468 | 581 |
469 new_entry->writer = item; | 582 pending_op->writer = item; |
470 BackendCallback* my_callback = new BackendCallback(this, new_entry); | 583 BackendCallback* my_callback = new BackendCallback(this, pending_op); |
| 584 pending_op->callback = my_callback; |
471 | 585 |
472 int rv = disk_cache_->DoomEntry(key, my_callback); | 586 int rv = disk_cache_->DoomEntry(key, my_callback); |
473 if (rv != ERR_IO_PENDING) { | 587 if (rv != ERR_IO_PENDING) { |
474 item->ClearTransaction(); | 588 item->ClearTransaction(); |
475 my_callback->Run(rv); | 589 my_callback->Run(rv); |
476 } | 590 } |
477 | 591 |
478 return rv; | 592 return rv; |
479 } | 593 } |
480 | 594 |
(...skipping 48 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
529 for (ActiveEntriesMap::iterator it = active_entries_.begin(); | 643 for (ActiveEntriesMap::iterator it = active_entries_.begin(); |
530 it != active_entries_.end(); ++it) { | 644 it != active_entries_.end(); ++it) { |
531 if (it->second == entry) { | 645 if (it->second == entry) { |
532 active_entries_.erase(it); | 646 active_entries_.erase(it); |
533 delete entry; | 647 delete entry; |
534 break; | 648 break; |
535 } | 649 } |
536 } | 650 } |
537 } | 651 } |
538 | 652 |
539 HttpCache::NewEntry* HttpCache::GetNewEntry(const std::string& key) { | 653 HttpCache::PendingOp* HttpCache::GetPendingOp(const std::string& key) { |
540 DCHECK(!FindActiveEntry(key)); | 654 DCHECK(!FindActiveEntry(key)); |
541 | 655 |
542 NewEntriesMap::const_iterator it = new_entries_.find(key); | 656 PendingOpsMap::const_iterator it = pending_ops_.find(key); |
543 if (it != new_entries_.end()) | 657 if (it != pending_ops_.end()) |
544 return it->second; | 658 return it->second; |
545 | 659 |
546 NewEntry* entry = new NewEntry(); | 660 PendingOp* operation = new PendingOp(); |
547 new_entries_[key] = entry; | 661 pending_ops_[key] = operation; |
548 return entry; | 662 return operation; |
549 } | 663 } |
550 | 664 |
551 void HttpCache::DeleteNewEntry(NewEntry* entry) { | 665 void HttpCache::DeletePendingOp(PendingOp* pending_op) { |
552 std::string key; | 666 std::string key; |
553 if (entry->disk_entry) | 667 if (pending_op->disk_entry) |
554 key = entry->disk_entry->GetKey(); | 668 key = pending_op->disk_entry->GetKey(); |
555 | 669 |
556 if (!key.empty()) { | 670 if (!key.empty()) { |
557 NewEntriesMap::iterator it = new_entries_.find(key); | 671 PendingOpsMap::iterator it = pending_ops_.find(key); |
558 DCHECK(it != new_entries_.end()); | 672 DCHECK(it != pending_ops_.end()); |
559 new_entries_.erase(it); | 673 pending_ops_.erase(it); |
560 } else { | 674 } else { |
561 for (NewEntriesMap::iterator it = new_entries_.begin(); | 675 for (PendingOpsMap::iterator it = pending_ops_.begin(); |
562 it != new_entries_.end(); ++it) { | 676 it != pending_ops_.end(); ++it) { |
563 if (it->second == entry) { | 677 if (it->second == pending_op) { |
564 new_entries_.erase(it); | 678 pending_ops_.erase(it); |
565 break; | 679 break; |
566 } | 680 } |
567 } | 681 } |
568 } | 682 } |
| 683 DCHECK(pending_op->pending_queue.empty()); |
569 | 684 |
570 delete entry; | 685 delete pending_op; |
571 } | 686 } |
572 | 687 |
573 int HttpCache::OpenEntry(const std::string& key, ActiveEntry** entry, | 688 int HttpCache::OpenEntry(const std::string& key, ActiveEntry** entry, |
574 Transaction* trans) { | 689 Transaction* trans) { |
575 ActiveEntry* active_entry = FindActiveEntry(key); | 690 ActiveEntry* active_entry = FindActiveEntry(key); |
576 if (active_entry) { | 691 if (active_entry) { |
577 *entry = active_entry; | 692 *entry = active_entry; |
578 return OK; | 693 return OK; |
579 } | 694 } |
580 | 695 |
581 WorkItem* item = new WorkItem(entry, trans, WI_OPEN_ENTRY); | 696 WorkItem* item = new WorkItem(WI_OPEN_ENTRY, trans, entry); |
582 NewEntry* new_entry = GetNewEntry(key); | 697 PendingOp* pending_op = GetPendingOp(key); |
583 if (new_entry->writer) { | 698 if (pending_op->writer) { |
584 new_entry->pending_queue.push_back(item); | 699 pending_op->pending_queue.push_back(item); |
585 return ERR_IO_PENDING; | 700 return ERR_IO_PENDING; |
586 } | 701 } |
587 | 702 |
588 DCHECK(new_entry->pending_queue.empty()); | 703 DCHECK(pending_op->pending_queue.empty()); |
589 | 704 |
590 new_entry->writer = item; | 705 pending_op->writer = item; |
591 BackendCallback* my_callback = new BackendCallback(this, new_entry); | 706 BackendCallback* my_callback = new BackendCallback(this, pending_op); |
| 707 pending_op->callback = my_callback; |
592 | 708 |
593 int rv = disk_cache_->OpenEntry(key, &(new_entry->disk_entry), my_callback); | 709 int rv = disk_cache_->OpenEntry(key, &(pending_op->disk_entry), my_callback); |
594 if (rv != ERR_IO_PENDING) { | 710 if (rv != ERR_IO_PENDING) { |
595 item->ClearTransaction(); | 711 item->ClearTransaction(); |
596 my_callback->Run(rv); | 712 my_callback->Run(rv); |
597 } | 713 } |
598 | 714 |
599 return rv; | 715 return rv; |
600 } | 716 } |
601 | 717 |
602 int HttpCache::CreateEntry(const std::string& key, ActiveEntry** entry, | 718 int HttpCache::CreateEntry(const std::string& key, ActiveEntry** entry, |
603 Transaction* trans) { | 719 Transaction* trans) { |
604 DCHECK(!FindActiveEntry(key)); | 720 DCHECK(!FindActiveEntry(key)); |
605 | 721 |
606 WorkItem* item = new WorkItem(entry, trans, WI_CREATE_ENTRY); | 722 WorkItem* item = new WorkItem(WI_CREATE_ENTRY, trans, entry); |
607 NewEntry* new_entry = GetNewEntry(key); | 723 PendingOp* pending_op = GetPendingOp(key); |
608 if (new_entry->writer) { | 724 if (pending_op->writer) { |
609 new_entry->pending_queue.push_back(item); | 725 pending_op->pending_queue.push_back(item); |
610 return ERR_IO_PENDING; | 726 return ERR_IO_PENDING; |
611 } | 727 } |
612 | 728 |
613 DCHECK(new_entry->pending_queue.empty()); | 729 DCHECK(pending_op->pending_queue.empty()); |
614 | 730 |
615 new_entry->writer = item; | 731 pending_op->writer = item; |
616 BackendCallback* my_callback = new BackendCallback(this, new_entry); | 732 BackendCallback* my_callback = new BackendCallback(this, pending_op); |
| 733 pending_op->callback = my_callback; |
617 | 734 |
618 int rv = disk_cache_->CreateEntry(key, &(new_entry->disk_entry), my_callback); | 735 int rv = disk_cache_->CreateEntry(key, &(pending_op->disk_entry), |
| 736 my_callback); |
619 if (rv != ERR_IO_PENDING) { | 737 if (rv != ERR_IO_PENDING) { |
620 item->ClearTransaction(); | 738 item->ClearTransaction(); |
621 my_callback->Run(rv); | 739 my_callback->Run(rv); |
622 } | 740 } |
623 | 741 |
624 return rv; | 742 return rv; |
625 } | 743 } |
626 | 744 |
627 void HttpCache::DestroyEntry(ActiveEntry* entry) { | 745 void HttpCache::DestroyEntry(ActiveEntry* entry) { |
628 if (entry->doomed) { | 746 if (entry->doomed) { |
(...skipping 119 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
748 | 866 |
749 void HttpCache::RemovePendingTransaction(Transaction* trans) { | 867 void HttpCache::RemovePendingTransaction(Transaction* trans) { |
750 ActiveEntriesMap::const_iterator i = active_entries_.find(trans->key()); | 868 ActiveEntriesMap::const_iterator i = active_entries_.find(trans->key()); |
751 bool found = false; | 869 bool found = false; |
752 if (i != active_entries_.end()) | 870 if (i != active_entries_.end()) |
753 found = RemovePendingTransactionFromEntry(i->second, trans); | 871 found = RemovePendingTransactionFromEntry(i->second, trans); |
754 | 872 |
755 if (found) | 873 if (found) |
756 return; | 874 return; |
757 | 875 |
758 NewEntriesMap::const_iterator j = new_entries_.find(trans->key()); | 876 if (building_backend_) { |
759 if (j != new_entries_.end()) | 877 PendingOpsMap::const_iterator j = pending_ops_.find(""); |
760 found = RemovePendingTransactionFromNewEntry(j->second, trans); | 878 if (j != pending_ops_.end()) |
| 879 found = RemovePendingTransactionFromPendingOp(j->second, trans); |
| 880 |
| 881 if (found) |
| 882 return; |
| 883 } |
| 884 |
| 885 PendingOpsMap::const_iterator j = pending_ops_.find(trans->key()); |
| 886 if (j != pending_ops_.end()) |
| 887 found = RemovePendingTransactionFromPendingOp(j->second, trans); |
| 888 |
| 889 if (found) |
| 890 return; |
761 | 891 |
762 ActiveEntriesSet::iterator k = doomed_entries_.begin(); | 892 ActiveEntriesSet::iterator k = doomed_entries_.begin(); |
763 for (; k != doomed_entries_.end() && !found; ++k) | 893 for (; k != doomed_entries_.end() && !found; ++k) |
764 found = RemovePendingTransactionFromEntry(*k, trans); | 894 found = RemovePendingTransactionFromEntry(*k, trans); |
765 | 895 |
766 DCHECK(found) << "Pending transaction not found"; | 896 DCHECK(found) << "Pending transaction not found"; |
767 } | 897 } |
768 | 898 |
769 bool HttpCache::RemovePendingTransactionFromEntry(ActiveEntry* entry, | 899 bool HttpCache::RemovePendingTransactionFromEntry(ActiveEntry* entry, |
770 Transaction* trans) { | 900 Transaction* trans) { |
771 TransactionList& pending_queue = entry->pending_queue; | 901 TransactionList& pending_queue = entry->pending_queue; |
772 | 902 |
773 TransactionList::iterator j = | 903 TransactionList::iterator j = |
774 find(pending_queue.begin(), pending_queue.end(), trans); | 904 find(pending_queue.begin(), pending_queue.end(), trans); |
775 if (j == pending_queue.end()) | 905 if (j == pending_queue.end()) |
776 return false; | 906 return false; |
777 | 907 |
778 pending_queue.erase(j); | 908 pending_queue.erase(j); |
779 return true; | 909 return true; |
780 } | 910 } |
781 | 911 |
782 bool HttpCache::RemovePendingTransactionFromNewEntry(NewEntry* entry, | 912 bool HttpCache::RemovePendingTransactionFromPendingOp(PendingOp* pending_op, |
783 Transaction* trans) { | 913 Transaction* trans) { |
784 if (entry->writer->Matches(trans)) { | 914 if (pending_op->writer->Matches(trans)) { |
785 entry->writer->ClearTransaction(); | 915 pending_op->writer->ClearTransaction(); |
786 entry->writer->ClearEntry(); | 916 pending_op->writer->ClearEntry(); |
787 return true; | 917 return true; |
788 } | 918 } |
789 WorkItemList& pending_queue = entry->pending_queue; | 919 WorkItemList& pending_queue = pending_op->pending_queue; |
790 | 920 |
791 WorkItemList::iterator it = pending_queue.begin(); | 921 WorkItemList::iterator it = pending_queue.begin(); |
792 for (; it != pending_queue.end(); ++it) { | 922 for (; it != pending_queue.end(); ++it) { |
793 if ((*it)->Matches(trans)) { | 923 if ((*it)->Matches(trans)) { |
794 delete *it; | 924 delete *it; |
795 pending_queue.erase(it); | 925 pending_queue.erase(it); |
796 return true; | 926 return true; |
797 } | 927 } |
798 } | 928 } |
799 return false; | 929 return false; |
(...skipping 30 matching lines...) Expand all Loading... |
830 return; // Have to wait. | 960 return; // Have to wait. |
831 | 961 |
832 entry->pending_queue.erase(entry->pending_queue.begin()); | 962 entry->pending_queue.erase(entry->pending_queue.begin()); |
833 | 963 |
834 int rv = AddTransactionToEntry(entry, next); | 964 int rv = AddTransactionToEntry(entry, next); |
835 if (rv != ERR_IO_PENDING) { | 965 if (rv != ERR_IO_PENDING) { |
836 next->io_callback()->Run(rv); | 966 next->io_callback()->Run(rv); |
837 } | 967 } |
838 } | 968 } |
839 | 969 |
840 void HttpCache::OnIOComplete(int result, NewEntry* new_entry) { | 970 void HttpCache::OnIOComplete(int result, PendingOp* pending_op) { |
841 scoped_ptr<WorkItem> item(new_entry->writer); | 971 WorkItemOperation op = pending_op->writer->operation(); |
842 WorkItemOperation op = item->operation(); | 972 |
| 973 // Completing the creation of the backend is simpler than the other cases. |
| 974 if (op == WI_CREATE_BACKEND) |
| 975 return OnBackendCreated(result, pending_op); |
| 976 |
| 977 scoped_ptr<WorkItem> item(pending_op->writer); |
843 bool fail_requests = false; | 978 bool fail_requests = false; |
844 | 979 |
845 ActiveEntry* entry = NULL; | 980 ActiveEntry* entry = NULL; |
846 std::string key; | 981 std::string key; |
847 if (result == OK) { | 982 if (result == OK) { |
848 if (op == WI_DOOM_ENTRY) { | 983 if (op == WI_DOOM_ENTRY) { |
849 // Anything after a Doom has to be restarted. | 984 // Anything after a Doom has to be restarted. |
850 fail_requests = true; | 985 fail_requests = true; |
851 } else if (item->IsValid()) { | 986 } else if (item->IsValid()) { |
852 key = new_entry->disk_entry->GetKey(); | 987 key = pending_op->disk_entry->GetKey(); |
853 entry = ActivateEntry(key, new_entry->disk_entry); | 988 entry = ActivateEntry(key, pending_op->disk_entry); |
854 } else { | 989 } else { |
855 // The writer transaction is gone. | 990 // The writer transaction is gone. |
856 if (op == WI_CREATE_ENTRY) | 991 if (op == WI_CREATE_ENTRY) |
857 new_entry->disk_entry->Doom(); | 992 pending_op->disk_entry->Doom(); |
858 new_entry->disk_entry->Close(); | 993 pending_op->disk_entry->Close(); |
859 fail_requests = true; | 994 fail_requests = true; |
860 } | 995 } |
861 } | 996 } |
862 | 997 |
863 // We are about to notify a bunch of transactions, and they may decide to | 998 // We are about to notify a bunch of transactions, and they may decide to |
864 // re-issue a request (or send a different one). If we don't delete new_entry, | 999 // re-issue a request (or send a different one). If we don't delete |
865 // the new request will be appended to the end of the list, and we'll see it | 1000 // pending_op, the new request will be appended to the end of the list, and |
866 // again from this point before it has a chance to complete (and we'll be | 1001 // we'll see it again from this point before it has a chance to complete (and |
867 // messing out the request order). The down side is that if for some reason | 1002 // we'll be messing out the request order). The down side is that if for some |
868 // notifying request A ends up cancelling request B (for the same key), we | 1003 // reason notifying request A ends up cancelling request B (for the same key), |
869 // won't find request B anywhere (because it would be in a local variable | 1004 // we won't find request B anywhere (because it would be in a local variable |
870 // here) and that's bad. If there is a chance for that to happen, we'll have | 1005 // here) and that's bad. If there is a chance for that to happen, we'll have |
871 // to move the callback used to be a CancelableCallback. By the way, for this | 1006 // to move the callback used to be a CancelableCallback. By the way, for this |
872 // to happen the action (to cancel B) has to be synchronous to the | 1007 // to happen the action (to cancel B) has to be synchronous to the |
873 // notification for request A. | 1008 // notification for request A. |
874 WorkItemList pending_items; | 1009 WorkItemList pending_items; |
875 pending_items.swap(new_entry->pending_queue); | 1010 pending_items.swap(pending_op->pending_queue); |
876 DeleteNewEntry(new_entry); | 1011 DeletePendingOp(pending_op); |
877 | 1012 |
878 item->NotifyTransaction(result, entry); | 1013 item->NotifyTransaction(result, entry); |
879 | 1014 |
880 while (!pending_items.empty()) { | 1015 while (!pending_items.empty()) { |
881 item.reset(pending_items.front()); | 1016 item.reset(pending_items.front()); |
882 pending_items.pop_front(); | 1017 pending_items.pop_front(); |
883 | 1018 |
884 if (item->operation() == WI_DOOM_ENTRY) { | 1019 if (item->operation() == WI_DOOM_ENTRY) { |
885 // A queued doom request is always a race. | 1020 // A queued doom request is always a race. |
886 fail_requests = true; | 1021 fail_requests = true; |
(...skipping 26 matching lines...) Expand all Loading... |
913 // Failed Create followed by an Open. | 1048 // Failed Create followed by an Open. |
914 item->NotifyTransaction(ERR_CACHE_RACE, NULL); | 1049 item->NotifyTransaction(ERR_CACHE_RACE, NULL); |
915 fail_requests = true; | 1050 fail_requests = true; |
916 } else { | 1051 } else { |
917 item->NotifyTransaction(result, entry); | 1052 item->NotifyTransaction(result, entry); |
918 } | 1053 } |
919 } | 1054 } |
920 } | 1055 } |
921 } | 1056 } |
922 | 1057 |
| 1058 void HttpCache::OnBackendCreated(int result, PendingOp* pending_op) { |
| 1059 scoped_ptr<WorkItem> item(pending_op->writer); |
| 1060 WorkItemOperation op = item->operation(); |
| 1061 DCHECK_EQ(WI_CREATE_BACKEND, op); |
| 1062 |
| 1063 if (type_ != MEMORY_CACHE) |
| 1064 disk_cache_dir_ = FilePath(); // Reclaim memory. |
| 1065 |
| 1066 if (result == OK) |
| 1067 disk_cache_.reset(temp_backend_); |
| 1068 |
| 1069 item->DoCallback(result, temp_backend_); |
| 1070 |
| 1071 // Notify and all callers and delete all pending work items. |
| 1072 while (!pending_op->pending_queue.empty()) { |
| 1073 scoped_ptr<WorkItem> pending_item(pending_op->pending_queue.front()); |
| 1074 pending_op->pending_queue.pop_front(); |
| 1075 DCHECK_EQ(WI_CREATE_BACKEND, pending_item->operation()); |
| 1076 |
| 1077 // This could be an external caller or a transaction waiting on Start(). |
| 1078 pending_item->DoCallback(result, temp_backend_); |
| 1079 pending_item->NotifyTransaction(result, NULL); |
| 1080 } |
| 1081 |
| 1082 DeletePendingOp(pending_op); |
| 1083 building_backend_ = false; |
| 1084 } |
| 1085 |
923 } // namespace net | 1086 } // namespace net |
OLD | NEW |