OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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 #include <utility> | 8 #include <utility> |
9 | 9 |
10 #include "base/bind.h" | 10 #include "base/bind.h" |
(...skipping 78 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
89 true, | 89 true, |
90 thread_, | 90 thread_, |
91 net_log, | 91 net_log, |
92 backend, | 92 backend, |
93 callback); | 93 callback); |
94 } | 94 } |
95 | 95 |
96 //----------------------------------------------------------------------------- | 96 //----------------------------------------------------------------------------- |
97 | 97 |
98 HttpCache::ActiveEntry::ActiveEntry(disk_cache::Entry* entry) | 98 HttpCache::ActiveEntry::ActiveEntry(disk_cache::Entry* entry) |
99 : disk_entry(entry), | 99 : disk_entry(entry) {} |
100 writer(NULL), | |
101 will_process_pending_queue(false), | |
102 doomed(false) { | |
103 } | |
104 | 100 |
105 HttpCache::ActiveEntry::~ActiveEntry() { | 101 HttpCache::ActiveEntry::~ActiveEntry() { |
106 if (disk_entry) { | 102 if (disk_entry) { |
107 disk_entry->Close(); | 103 disk_entry->Close(); |
108 disk_entry = NULL; | 104 disk_entry = nullptr; |
109 } | 105 } |
110 } | 106 } |
111 | 107 |
112 size_t HttpCache::ActiveEntry::EstimateMemoryUsage() const { | 108 size_t HttpCache::ActiveEntry::EstimateMemoryUsage() const { |
113 // Skip |disk_entry| which is tracked in simple_backend_impl; Skip |readers| | 109 // Skip |disk_entry| which is tracked in simple_backend_impl; Skip |readers| |
114 // and |pending_queue| because the Transactions are owned by their respective | 110 // and |add_to_entry_queue| because the Transactions are owned by their |
115 // URLRequestHttpJobs. | 111 // respective URLRequestHttpJobs. |
116 return 0; | 112 return 0; |
117 } | 113 } |
118 | 114 |
119 bool HttpCache::ActiveEntry::HasNoTransactions() { | 115 bool HttpCache::ActiveEntry::HasNoTransactions() { |
120 return !writer && readers.empty() && pending_queue.empty(); | 116 return !writer && readers.empty() && add_to_entry_queue.empty() && |
| 117 validated_queue.empty() && !validating_transaction; |
| 118 } |
| 119 |
| 120 bool HttpCache::ActiveEntry::HasNoActiveTransactions() { |
| 121 return !writer && readers.empty() && !validating_transaction; |
| 122 } |
| 123 |
| 124 bool HttpCache::ActiveEntry::IsReader(Transaction* transaction) { |
| 125 return readers.count(transaction); |
| 126 } |
| 127 |
| 128 int HttpCache::ActiveEntry::AdvanceState(Transaction* transaction) { |
| 129 switch (state) { |
| 130 case State::NONE: |
| 131 return DoAdvanceStateNone(transaction); |
| 132 case State::VALIDATING: |
| 133 return DoAdvanceStateValidating(transaction); |
| 134 case State::VALIDATING_WRITE: |
| 135 return DoAdvanceStateValidatingWrite(transaction); |
| 136 case State::READ: |
| 137 return DoAdvanceStateRead(transaction); |
| 138 case State::VALIDATING_READ: |
| 139 return DoAdvanceStateValidatingRead(transaction); |
| 140 } |
| 141 } |
| 142 |
| 143 int HttpCache::ActiveEntry::RegressState(Transaction* transaction) { |
| 144 switch (state) { |
| 145 case State::NONE: |
| 146 NOTREACHED() << "bad state"; |
| 147 return ERR_FAILURE; |
| 148 case State::VALIDATING: |
| 149 DoRegressStateValidating(transaction); |
| 150 break; |
| 151 case State::VALIDATING_WRITE: |
| 152 DoRegressStateValidatingWrite(transaction); |
| 153 break; |
| 154 case State::WRITE: |
| 155 |
| 156 case State::READ: |
| 157 DoRegressStateRead(transaction); |
| 158 break; |
| 159 case State::VALIDATING_READ: |
| 160 DoRegressStateValidatingRead(transaction); |
| 161 break; |
| 162 } |
| 163 } |
| 164 |
| 165 int HttpCache::ActiveEntry::DoAdvanceStateNone(Transaction* transaction) { |
| 166 if (transaction->mode() & Transaction::WRITE) { |
| 167 validating_transaction = transaction; |
| 168 // entry->writer = transaction; |
| 169 state = State::VALIDATING; |
| 170 } else { |
| 171 entry->readers.insert(transaction); |
| 172 state = State::READ; |
| 173 } |
| 174 return OK; |
| 175 } |
| 176 |
| 177 int HttpCache::ActiveEntry::DoAdvanceStateValidating(Transaction* transaction) { |
| 178 if (transaction == validating_transaction) { |
| 179 writer = transaction; |
| 180 state = STATE::WRITE; |
| 181 return OK; |
| 182 } |
| 183 // Note the entry may be new or may already have a response body written to |
| 184 // it. In both cases, for a |transaction| with write mode set needs to wait |
| 185 // because only one transaction is allowed to be validating at a time. |
| 186 // If |transaction| is read-only, then it needs to wait because its possible |
| 187 // that the validating_transaction dooms the entry if its not a match or it |
| 188 // needs to wait anyways if the entry is not completely written yet. |
| 189 entry->add_to_entry_queue.push_back(transaction); |
| 190 return ERR_IO_PENDING; |
| 191 } |
| 192 |
| 193 int HttpCache::ActiveEntry::DoAdvanceStateWrite(Transaction* transaction) { |
| 194 if (transaction->mode() & Transaction::WRITE) { |
| 195 DCHECK_NE(transaction, validating_transaction); |
| 196 validating_transaction = transaction; |
| 197 state = State::VALIDATING_WRITE; |
| 198 return OK; |
| 199 } |
| 200 |
| 201 entry->add_to_entry_queue.push_back(transaction); |
| 202 return ERR_IO_PENDING; |
| 203 } |
| 204 |
| 205 int HttpCache::ActiveEntry::DoAdvanceStateValidatingWrite( |
| 206 Transaction* transaction) { |
| 207 if (transaction == validating_transaction) { |
| 208 validated_queue.push_back(transaction); |
| 209 validating_transaction = nullptr; |
| 210 state = State::WRITE; |
| 211 return OK; |
| 212 } |
| 213 |
| 214 entry->add_to_entry_queue.push_back(transaction); |
| 215 return ERR_IO_PENDING; |
| 216 } |
| 217 |
| 218 int HttpCache::ActiveEntry::DoAdvanceStateRead(Transaction* transaction) { |
| 219 if (transaction->mode() & Transaction::WRITE) { |
| 220 DCHECK_EQ(validating_transaction, nullptr); |
| 221 validating_transaction = transaction; |
| 222 state = State::VALIDATING_READ; |
| 223 } else { |
| 224 entry->readers.insert(transaction); |
| 225 } |
| 226 return OK; |
| 227 } |
| 228 |
| 229 int HttpCache::ActiveEntry::DoAdvanceStateValidatingRead( |
| 230 Transaction* transaction) { |
| 231 if ((transaction->mode() & Transaction::WRITE) && |
| 232 (transaction != validating_transaction)) { |
| 233 entry->add_to_entry_queue.push_back(transaction); |
| 234 return ERR_IO_PENDING; |
| 235 } else { |
| 236 entry->readers.insert(transaction); |
| 237 if (transaction == validating_transaction) { |
| 238 validating_transaction = nullptr; |
| 239 state = State::READ; |
| 240 } |
| 241 } |
| 242 return OK; |
| 243 } |
| 244 |
| 245 int HttpCache::ActiveEntry::DoRegressStateValidating(Transaction* transaction) { |
| 246 DCHECK_EQ(validating_transaction, transaction); |
| 247 validating_transaction = nullptr; |
| 248 state = State::NONE; |
| 249 } |
| 250 |
| 251 int HttpCache::ActiveEntry::DoRegressStateValidatingWrite( |
| 252 Transaction* transaction) { |
| 253 DCHECK(transaction == validating_transaction || transaction == writer); |
| 254 if (transaction == validating_transaction) { |
| 255 validating_transaction = nullptr; |
| 256 state = State::WRITE; |
| 257 } |
| 258 writer = nullptr; |
| 259 state = State::VALIDATING; |
| 260 } |
| 261 |
| 262 int HttpCache::ActiveEntry::DoRegressStateWrite(Transaction* transaction) { |
| 263 DCHECK_EQ(transaction, writer); |
| 264 writer = nullptr; |
| 265 state = State::NONE; |
| 266 } |
| 267 |
| 268 int HttpCache::ActiveEntry::DoRegressStateRead(Transaction* transaction) { |
| 269 auto it = entry->readers.find(trans); |
| 270 DCHECK(it != entry->readers.end()); |
| 271 |
| 272 entry->readers.erase(it); |
| 273 state = State::NONE; |
| 274 } |
| 275 |
| 276 int HttpCache::ActiveEntry::DoRegressStateValidatingRead( |
| 277 Transaction* transaction) { |
| 278 if (transaction == validating_transaction) { |
| 279 validating_transaction = nullptr; |
| 280 state = State::READ; |
| 281 return; |
| 282 } |
| 283 auto it = entry->readers.find(trans); |
| 284 DCHECK(it != entry->readers.end()); |
| 285 |
| 286 entry->readers.erase(it); |
| 287 state = State |
121 } | 288 } |
122 | 289 |
123 //----------------------------------------------------------------------------- | 290 //----------------------------------------------------------------------------- |
124 | 291 |
125 // This structure keeps track of work items that are attempting to create or | 292 // This structure keeps track of work items that are attempting to create or |
126 // open cache entries or the backend itself. | 293 // open cache entries or the backend itself. |
127 struct HttpCache::PendingOp { | 294 struct HttpCache::PendingOp { |
128 PendingOp() : disk_entry(NULL) {} | 295 PendingOp() : disk_entry(NULL) {} |
129 ~PendingOp() {} | 296 ~PendingOp() {} |
130 | 297 |
(...skipping 222 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
353 } | 520 } |
354 } | 521 } |
355 } | 522 } |
356 | 523 |
357 HttpCache::~HttpCache() { | 524 HttpCache::~HttpCache() { |
358 // Transactions should see an invalid cache after this point; otherwise they | 525 // Transactions should see an invalid cache after this point; otherwise they |
359 // could see an inconsistent object (half destroyed). | 526 // could see an inconsistent object (half destroyed). |
360 weak_factory_.InvalidateWeakPtrs(); | 527 weak_factory_.InvalidateWeakPtrs(); |
361 | 528 |
362 // If we have any active entries remaining, then we need to deactivate them. | 529 // If we have any active entries remaining, then we need to deactivate them. |
363 // We may have some pending calls to OnProcessPendingQueue, but since those | 530 // We may have some pending tasks to process pending queue for reading or |
364 // won't run (due to our destruction), we can simply ignore the corresponding | 531 // parallel validation, but since those won't run (due to our destruction), |
365 // will_process_pending_queue flag. | 532 // we can simply ignore the corresponding flags. |
366 while (!active_entries_.empty()) { | 533 while (!active_entries_.empty()) { |
367 ActiveEntry* entry = active_entries_.begin()->second.get(); | 534 ActiveEntry* entry = active_entries_.begin()->second.get(); |
368 entry->will_process_pending_queue = false; | 535 entry->will_process_waiting_transactions = false; |
369 entry->pending_queue.clear(); | 536 entry->will_process_parallel_validation = false; |
| 537 entry->add_to_entry_queue.clear(); |
370 entry->readers.clear(); | 538 entry->readers.clear(); |
371 entry->writer = NULL; | 539 entry->validated_queue.clear(); |
| 540 entry->validating_transaction = nullptr; |
| 541 entry->writer = nullptr; |
372 DeactivateEntry(entry); | 542 DeactivateEntry(entry); |
373 } | 543 } |
374 | 544 |
375 doomed_entries_.clear(); | 545 doomed_entries_.clear(); |
376 | 546 |
377 // Before deleting pending_ops_, we have to make sure that the disk cache is | 547 // Before deleting pending_ops_, we have to make sure that the disk cache is |
378 // done with said operations, or it will attempt to use deleted data. | 548 // done with said operations, or it will attempt to use deleted data. |
379 disk_cache_.reset(); | 549 disk_cache_.reset(); |
380 | 550 |
381 for (auto pending_it = pending_ops_.begin(); pending_it != pending_ops_.end(); | 551 for (auto pending_it = pending_ops_.begin(); pending_it != pending_ops_.end(); |
(...skipping 239 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
621 // We keep track of doomed entries so that we can ensure that they are | 791 // We keep track of doomed entries so that we can ensure that they are |
622 // cleaned up properly when the cache is destroyed. | 792 // cleaned up properly when the cache is destroyed. |
623 ActiveEntry* entry_ptr = entry.get(); | 793 ActiveEntry* entry_ptr = entry.get(); |
624 DCHECK_EQ(0u, doomed_entries_.count(entry_ptr)); | 794 DCHECK_EQ(0u, doomed_entries_.count(entry_ptr)); |
625 doomed_entries_[entry_ptr] = std::move(entry); | 795 doomed_entries_[entry_ptr] = std::move(entry); |
626 | 796 |
627 entry_ptr->disk_entry->Doom(); | 797 entry_ptr->disk_entry->Doom(); |
628 entry_ptr->doomed = true; | 798 entry_ptr->doomed = true; |
629 | 799 |
630 DCHECK(entry_ptr->writer || !entry_ptr->readers.empty() || | 800 DCHECK(entry_ptr->writer || !entry_ptr->readers.empty() || |
631 entry_ptr->will_process_pending_queue); | 801 !entry_ptr->validated_queue.empty() || |
| 802 entry_ptr->validating_transaction || |
| 803 entry_ptr->will_process_waiting_transactions || |
| 804 entry_ptr->will_process_parallel_validation); |
632 return OK; | 805 return OK; |
633 } | 806 } |
634 | 807 |
635 int HttpCache::AsyncDoomEntry(const std::string& key, Transaction* trans) { | 808 int HttpCache::AsyncDoomEntry(const std::string& key, Transaction* trans) { |
636 std::unique_ptr<WorkItem> item = | 809 std::unique_ptr<WorkItem> item = |
637 base::MakeUnique<WorkItem>(WI_DOOM_ENTRY, trans, nullptr); | 810 base::MakeUnique<WorkItem>(WI_DOOM_ENTRY, trans, nullptr); |
638 PendingOp* pending_op = GetPendingOp(key); | 811 PendingOp* pending_op = GetPendingOp(key); |
639 if (pending_op->writer) { | 812 if (pending_op->writer) { |
640 pending_op->pending_queue.push_back(std::move(item)); | 813 pending_op->pending_queue.push_back(std::move(item)); |
641 return ERR_IO_PENDING; | 814 return ERR_IO_PENDING; |
(...skipping 35 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
677 DCHECK(entry->doomed); | 850 DCHECK(entry->doomed); |
678 DCHECK(entry->HasNoTransactions()); | 851 DCHECK(entry->HasNoTransactions()); |
679 | 852 |
680 auto it = doomed_entries_.find(entry); | 853 auto it = doomed_entries_.find(entry); |
681 DCHECK(it != doomed_entries_.end()); | 854 DCHECK(it != doomed_entries_.end()); |
682 doomed_entries_.erase(it); | 855 doomed_entries_.erase(it); |
683 } | 856 } |
684 | 857 |
685 HttpCache::ActiveEntry* HttpCache::FindActiveEntry(const std::string& key) { | 858 HttpCache::ActiveEntry* HttpCache::FindActiveEntry(const std::string& key) { |
686 auto it = active_entries_.find(key); | 859 auto it = active_entries_.find(key); |
687 return it != active_entries_.end() ? it->second.get() : NULL; | 860 return it != active_entries_.end() ? it->second.get() : nullptr; |
| 861 } |
| 862 |
| 863 bool HttpCache::IsEntryAlive(const std::string& key, ActiveEntry* entry) { |
| 864 ActiveEntry* active_entry = FindActiveEntry(key); |
| 865 if (active_entry && active_entry == entry) |
| 866 return true; |
| 867 return doomed_entries_.count(entry); |
688 } | 868 } |
689 | 869 |
690 HttpCache::ActiveEntry* HttpCache::ActivateEntry( | 870 HttpCache::ActiveEntry* HttpCache::ActivateEntry( |
691 disk_cache::Entry* disk_entry) { | 871 disk_cache::Entry* disk_entry) { |
692 DCHECK(!FindActiveEntry(disk_entry->GetKey())); | 872 DCHECK(!FindActiveEntry(disk_entry->GetKey())); |
693 ActiveEntry* entry = new ActiveEntry(disk_entry); | 873 ActiveEntry* entry = new ActiveEntry(disk_entry); |
694 active_entries_[disk_entry->GetKey()] = base::WrapUnique(entry); | 874 active_entries_[disk_entry->GetKey()] = base::WrapUnique(entry); |
695 return entry; | 875 return entry; |
696 } | 876 } |
697 | 877 |
698 void HttpCache::DeactivateEntry(ActiveEntry* entry) { | 878 void HttpCache::DeactivateEntry(ActiveEntry* entry) { |
699 DCHECK(!entry->will_process_pending_queue); | |
700 DCHECK(!entry->doomed); | 879 DCHECK(!entry->doomed); |
701 DCHECK(entry->disk_entry); | 880 DCHECK(entry->disk_entry); |
702 DCHECK(entry->HasNoTransactions()); | 881 DCHECK(entry->HasNoTransactions()); |
703 | 882 |
704 std::string key = entry->disk_entry->GetKey(); | 883 std::string key = entry->disk_entry->GetKey(); |
705 if (key.empty()) | 884 if (key.empty()) |
706 return SlowDeactivateEntry(entry); | 885 return SlowDeactivateEntry(entry); |
707 | 886 |
708 auto it = active_entries_.find(key); | 887 auto it = active_entries_.find(key); |
709 DCHECK(it != active_entries_.end()); | 888 DCHECK(it != active_entries_.end()); |
(...skipping 109 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
819 } | 998 } |
820 | 999 |
821 void HttpCache::DestroyEntry(ActiveEntry* entry) { | 1000 void HttpCache::DestroyEntry(ActiveEntry* entry) { |
822 if (entry->doomed) { | 1001 if (entry->doomed) { |
823 FinalizeDoomedEntry(entry); | 1002 FinalizeDoomedEntry(entry); |
824 } else { | 1003 } else { |
825 DeactivateEntry(entry); | 1004 DeactivateEntry(entry); |
826 } | 1005 } |
827 } | 1006 } |
828 | 1007 |
829 int HttpCache::AddTransactionToEntry(ActiveEntry* entry, Transaction* trans) { | 1008 int HttpCache::AddTransactionToEntry(ActiveEntry* entry, |
| 1009 Transaction* transaction) { |
830 DCHECK(entry); | 1010 DCHECK(entry); |
831 DCHECK(entry->disk_entry); | 1011 DCHECK(entry->disk_entry); |
832 | 1012 |
833 // We implement a basic reader/writer lock for the disk cache entry. If | 1013 // If a pending queue processing task is posted, then this transaction |
834 // there is already a writer, then everyone has to wait for the writer to | 1014 // should be added after the queue task is processed, to maintain FIFO |
835 // finish before they can access the cache entry. There can be multiple | 1015 // ordering. |
836 // readers. | 1016 if (entry->will_process_waiting_transactions || |
837 // | 1017 entry->will_process_parallel_validation) { |
838 // NOTE: If the transaction can only write, then the entry should not be in | 1018 entry->add_to_entry_queue.push_back(transaction); |
839 // use (since any existing entry should have already been doomed). | |
840 | |
841 if (entry->writer || entry->will_process_pending_queue) { | |
842 entry->pending_queue.push_back(trans); | |
843 return ERR_IO_PENDING; | 1019 return ERR_IO_PENDING; |
844 } | 1020 } |
845 | 1021 |
846 if (trans->mode() & Transaction::WRITE) { | 1022 int rv = AddTransactionToEntryImpl(entry, transaction); |
847 // transaction needs exclusive access to the entry | 1023 if (rv == ERR_IO_PENDING) |
848 if (entry->readers.empty()) { | 1024 entry->add_to_entry_queue.push_back(transaction); |
849 entry->writer = trans; | 1025 return rv; |
850 } else { | 1026 } |
851 entry->pending_queue.push_back(trans); | 1027 |
852 return ERR_IO_PENDING; | 1028 // We implement a basic reader/writer lock for the disk cache entry. If there |
853 } | 1029 // is a writer, then all read-only transactions must wait. Non read-only |
854 } else { | 1030 // transactions can proceed to their validation phase. Validation is allowed |
855 // transaction needs read access to the entry | 1031 // for one transaction at a time so that we do not end up with wasted network |
856 entry->readers.insert(trans); | 1032 // requests. |
| 1033 int HttpCache::AddTransactionToEntryImpl(ActiveEntry* entry, |
| 1034 Transaction* transaction) { |
| 1035 if (transaction->mode() & Transaction::WRITE) |
| 1036 return AddTransactionToEntryForWrite(entry, transaction); |
| 1037 return AddTransactionToEntryForRead(entry, transaction); |
| 1038 } |
| 1039 |
| 1040 int HttpCache::AddTransactionToEntryForRead(ActiveEntry* entry, |
| 1041 Transaction* transaction) { |
| 1042 if (entry->writer) { |
| 1043 return ERR_IO_PENDING; |
857 } | 1044 } |
858 | 1045 entry->readers.insert(transaction); |
859 // We do this before calling EntryAvailable to force any further calls to | 1046 if (!entry->add_to_entry_queue.empty()) |
860 // AddTransactionToEntry to add their transaction to the pending queue, which | 1047 ProcessWaitingTransactions(entry); |
861 // ensures FIFO ordering. | |
862 if (!entry->writer && !entry->pending_queue.empty()) | |
863 ProcessPendingQueue(entry); | |
864 | |
865 return OK; | 1048 return OK; |
866 } | 1049 } |
867 | 1050 |
868 void HttpCache::DoneWithEntry(ActiveEntry* entry, Transaction* trans, | 1051 int HttpCache::AddTransactionToEntryForWrite(ActiveEntry* entry, |
| 1052 Transaction* transaction) { |
| 1053 // A transaction that has the write mode can become the next validating |
| 1054 // transaction. It can also become the writer if there is no active writer at |
| 1055 // this time. |
| 1056 if (entry->validating_transaction) { |
| 1057 return ERR_IO_PENDING; |
| 1058 } |
| 1059 entry->validating_transaction = transaction; |
| 1060 if (!entry->writer && entry->readers.empty()) { |
| 1061 entry->writer = transaction; |
| 1062 } |
| 1063 |
| 1064 // If there is no writer, we can add more readers. |
| 1065 if (!entry->writer && !entry->add_to_entry_queue.empty()) |
| 1066 ProcessWaitingTransactions(entry); |
| 1067 return OK; |
| 1068 } |
| 1069 |
| 1070 int HttpCache::DoneValidationWithEntry(ActiveEntry* entry, |
| 1071 Transaction* transaction, |
| 1072 bool is_match) { |
| 1073 DCHECK(entry->writer == transaction || |
| 1074 entry->validating_transaction == transaction); |
| 1075 entry->validating_transaction = nullptr; |
| 1076 |
| 1077 if (!is_match) { |
| 1078 if (entry->writer == transaction) |
| 1079 entry->writer = nullptr; |
| 1080 if (entry->HasNoActiveTransactions()) |
| 1081 DestroyEntryRestartPendingTransactions(entry); |
| 1082 else |
| 1083 DoomEntryRestartPendingTransactions(entry); |
| 1084 return OK; |
| 1085 } |
| 1086 |
| 1087 int rv = OK; |
| 1088 // If another transaction is writing the response, let this transaction wait |
| 1089 // till the response is complete. |
| 1090 if (entry->writer != transaction) { |
| 1091 entry->validated_queue.push_back(transaction); |
| 1092 rv = ERR_IO_PENDING; |
| 1093 } |
| 1094 // Else the transaction should either be the writer if its responsible for |
| 1095 // writing the response or should have been converted to a reader. |
| 1096 |
| 1097 // In all cases another transaction can start its validation phase. |
| 1098 ProcessParallelValidation(entry); |
| 1099 return rv; |
| 1100 } |
| 1101 |
| 1102 void HttpCache::DoneValidationWriteEntry(ActiveEntry* entry, |
| 1103 Transaction* transaction) { |
| 1104 entry->validating_transaction = nullptr; |
| 1105 ProcessParallelValidation(entry); |
| 1106 } |
| 1107 |
| 1108 void HttpCache::DoneWithEntry(ActiveEntry* entry, |
| 1109 Transaction* transaction, |
869 bool cancel) { | 1110 bool cancel) { |
870 // If we already posted a task to move on to the next transaction and this was | 1111 // If we already posted a task to move on to the next transaction and this was |
871 // the writer, there is nothing to cancel. | 1112 // the writer, there is nothing to cancel. |
872 if (entry->will_process_pending_queue && entry->readers.empty()) | 1113 if (entry->will_process_waiting_transactions && entry->writer == transaction) |
873 return; | 1114 return; |
874 | 1115 |
875 if (entry->writer) { | 1116 // Transaction is waiting in the validated_queue. |
876 DCHECK(trans == entry->writer); | 1117 auto i = std::find(entry->validated_queue.begin(), |
| 1118 entry->validated_queue.end(), transaction); |
| 1119 if (i != entry->validated_queue.end()) { |
| 1120 entry->validated_queue.erase(i); |
| 1121 return; |
| 1122 } |
877 | 1123 |
878 // Assume there was a failure. | 1124 // Assume there was a failure. |
879 bool success = false; | 1125 bool success = false; |
880 if (cancel) { | 1126 if (transaction == entry->writer && cancel) { |
881 DCHECK(entry->disk_entry); | 1127 DCHECK(entry->disk_entry); |
882 // This is a successful operation in the sense that we want to keep the | 1128 // This is a successful operation in the sense that we want to keep the |
883 // entry. | 1129 // entry. |
884 success = trans->AddTruncatedFlag(); | 1130 success = transaction->AddTruncatedFlag(); |
885 // The previous operation may have deleted the entry. | 1131 // The previous operation may have deleted the entry. |
886 if (!trans->entry()) | 1132 if (!transaction->entry()) |
887 return; | 1133 return; |
888 } | |
889 DoneWritingToEntry(entry, success); | |
890 } else { | |
891 DoneReadingFromEntry(entry, trans); | |
892 } | 1134 } |
| 1135 |
| 1136 // Transaction is writing to the entry, either in the validation phase or |
| 1137 // response-body phase. |
| 1138 if (transaction == entry->writer || |
| 1139 transaction == entry->validating_transaction) { |
| 1140 DoneWritingToEntry(entry, success, transaction); |
| 1141 return; |
| 1142 } |
| 1143 |
| 1144 // Transaction is reading from the entry. |
| 1145 DoneReadingFromEntry(entry, transaction); |
893 } | 1146 } |
894 | 1147 |
895 void HttpCache::DoneWritingToEntry(ActiveEntry* entry, bool success) { | 1148 void HttpCache::DoneWritingToEntry(ActiveEntry* entry, |
896 DCHECK(entry->readers.empty()); | 1149 bool success, |
| 1150 Transaction* transaction) { |
| 1151 DCHECK(transaction == entry->writer || |
| 1152 transaction == entry->validating_transaction); |
897 | 1153 |
898 entry->writer = NULL; | 1154 if (transaction == entry->writer) |
| 1155 entry->writer = nullptr; |
899 | 1156 |
| 1157 // Transaction is done writing to entry in the validation phase. |
| 1158 if (transaction == entry->validating_transaction) { |
| 1159 entry->validating_transaction = nullptr; |
| 1160 |
| 1161 if (!entry->writer) |
| 1162 DoneWritingToEntryProcessOtherTransactions(entry, success); |
| 1163 return; |
| 1164 } |
| 1165 |
| 1166 // Transaction is done writing to entry in the response body phase. |
| 1167 // Whether its a failure or success, validating transaction can be the new |
| 1168 // writer. |
| 1169 if (entry->validating_transaction) { |
| 1170 entry->writer = entry->validating_transaction; |
| 1171 return; |
| 1172 } |
| 1173 |
| 1174 DoneWritingToEntryProcessOtherTransactions(entry, success); |
| 1175 } |
| 1176 |
| 1177 void HttpCache::DoneWritingToEntryProcessOtherTransactions(ActiveEntry* entry, |
| 1178 bool success) { |
900 if (success) { | 1179 if (success) { |
901 ProcessPendingQueue(entry); | 1180 DCHECK(entry->readers.empty()); |
902 } else { | 1181 ProcessWaitingTransactions(entry); |
903 DCHECK(!entry->will_process_pending_queue); | 1182 return; |
| 1183 } |
904 | 1184 |
905 // We failed to create this entry. | 1185 DestroyEntryRestartPendingTransactions(entry); |
906 TransactionList pending_queue; | |
907 pending_queue.swap(entry->pending_queue); | |
908 | |
909 entry->disk_entry->Doom(); | |
910 DestroyEntry(entry); | |
911 | |
912 // We need to do something about these pending entries, which now need to | |
913 // be added to a new entry. | |
914 while (!pending_queue.empty()) { | |
915 // ERR_CACHE_RACE causes the transaction to restart the whole process. | |
916 pending_queue.front()->io_callback().Run(ERR_CACHE_RACE); | |
917 pending_queue.pop_front(); | |
918 } | |
919 } | |
920 } | 1186 } |
921 | 1187 |
922 void HttpCache::DoneReadingFromEntry(ActiveEntry* entry, Transaction* trans) { | 1188 void HttpCache::DoneReadingFromEntry(ActiveEntry* entry, Transaction* trans) { |
923 DCHECK(!entry->writer); | 1189 DCHECK(!entry->writer); |
924 | 1190 |
925 auto it = entry->readers.find(trans); | 1191 auto it = entry->readers.find(trans); |
926 DCHECK(it != entry->readers.end()); | 1192 DCHECK(it != entry->readers.end()); |
927 | 1193 |
928 entry->readers.erase(it); | 1194 entry->readers.erase(it); |
929 | 1195 |
930 ProcessPendingQueue(entry); | 1196 ProcessWaitingTransactions(entry); |
931 } | 1197 } |
932 | 1198 |
933 void HttpCache::ConvertWriterToReader(ActiveEntry* entry) { | 1199 bool HttpCache::ConvertWriterToReader(ActiveEntry* entry, |
934 DCHECK(entry->writer); | 1200 Transaction* transaction) { |
935 DCHECK(entry->writer->mode() == Transaction::READ_WRITE); | 1201 DCHECK_EQ(entry->validating_transaction, transaction); |
936 DCHECK(entry->readers.empty()); | 1202 DCHECK_EQ(transaction->mode(), Transaction::READ_WRITE); |
937 | 1203 |
938 Transaction* trans = entry->writer; | 1204 // We need to differentiate between the scenarios when the entry is already |
| 1205 // written to the cache vs. when the entry is not completely written and this |
| 1206 // is a validating transaction. In the latter case, it cannot become the |
| 1207 // reader and will be added to the validated_queue when |
| 1208 // DoneValidationWithEntry will be called for this transaction. |
939 | 1209 |
940 entry->writer = NULL; | 1210 // Entry is not completely written yet. |
941 entry->readers.insert(trans); | 1211 if (entry->writer && entry->writer != transaction) |
| 1212 return false; |
942 | 1213 |
943 ProcessPendingQueue(entry); | 1214 entry->validating_transaction = nullptr; |
| 1215 entry->writer = nullptr; |
| 1216 entry->readers.insert(transaction); |
| 1217 ProcessWaitingTransactions(entry); |
| 1218 return true; |
944 } | 1219 } |
945 | 1220 |
946 LoadState HttpCache::GetLoadStateForPendingTransaction( | 1221 LoadState HttpCache::GetLoadStateForPendingTransaction( |
947 const Transaction* trans) { | 1222 const Transaction* trans) { |
948 auto i = active_entries_.find(trans->key()); | 1223 auto i = active_entries_.find(trans->key()); |
949 if (i == active_entries_.end()) { | 1224 if (i == active_entries_.end()) { |
950 // If this is really a pending transaction, and it is not part of | 1225 // If this is really a pending transaction, and it is not part of |
951 // active_entries_, we should be creating the backend or the entry. | 1226 // active_entries_, we should be creating the backend or the entry. |
952 return LOAD_STATE_WAITING_FOR_CACHE; | 1227 return LOAD_STATE_WAITING_FOR_CACHE; |
953 } | 1228 } |
(...skipping 29 matching lines...) Expand all Loading... |
983 | 1258 |
984 for (auto k = doomed_entries_.begin(); k != doomed_entries_.end() && !found; | 1259 for (auto k = doomed_entries_.begin(); k != doomed_entries_.end() && !found; |
985 ++k) { | 1260 ++k) { |
986 found = RemovePendingTransactionFromEntry(k->first, trans); | 1261 found = RemovePendingTransactionFromEntry(k->first, trans); |
987 } | 1262 } |
988 | 1263 |
989 DCHECK(found) << "Pending transaction not found"; | 1264 DCHECK(found) << "Pending transaction not found"; |
990 } | 1265 } |
991 | 1266 |
992 bool HttpCache::RemovePendingTransactionFromEntry(ActiveEntry* entry, | 1267 bool HttpCache::RemovePendingTransactionFromEntry(ActiveEntry* entry, |
993 Transaction* trans) { | 1268 Transaction* transaction) { |
994 TransactionList& pending_queue = entry->pending_queue; | 1269 TransactionList& add_to_entry_queue = entry->add_to_entry_queue; |
995 | 1270 |
996 auto j = find(pending_queue.begin(), pending_queue.end(), trans); | 1271 auto j = |
997 if (j == pending_queue.end()) | 1272 find(add_to_entry_queue.begin(), add_to_entry_queue.end(), transaction); |
| 1273 if (j == add_to_entry_queue.end()) |
998 return false; | 1274 return false; |
999 | 1275 |
1000 pending_queue.erase(j); | 1276 add_to_entry_queue.erase(j); |
1001 return true; | 1277 return true; |
1002 } | 1278 } |
1003 | 1279 |
1004 bool HttpCache::RemovePendingTransactionFromPendingOp(PendingOp* pending_op, | 1280 bool HttpCache::RemovePendingTransactionFromPendingOp(PendingOp* pending_op, |
1005 Transaction* trans) { | 1281 Transaction* trans) { |
1006 if (pending_op->writer->Matches(trans)) { | 1282 if (pending_op->writer->Matches(trans)) { |
1007 pending_op->writer->ClearTransaction(); | 1283 pending_op->writer->ClearTransaction(); |
1008 pending_op->writer->ClearEntry(); | 1284 pending_op->writer->ClearEntry(); |
1009 return true; | 1285 return true; |
1010 } | 1286 } |
1011 WorkItemList& pending_queue = pending_op->pending_queue; | 1287 WorkItemList& pending_queue = pending_op->pending_queue; |
1012 | 1288 |
1013 for (auto it = pending_queue.begin(); it != pending_queue.end(); ++it) { | 1289 for (auto it = pending_queue.begin(); it != pending_queue.end(); ++it) { |
1014 if ((*it)->Matches(trans)) { | 1290 if ((*it)->Matches(trans)) { |
1015 pending_queue.erase(it); | 1291 pending_queue.erase(it); |
1016 return true; | 1292 return true; |
1017 } | 1293 } |
1018 } | 1294 } |
1019 return false; | 1295 return false; |
1020 } | 1296 } |
1021 | 1297 |
1022 void HttpCache::ProcessPendingQueue(ActiveEntry* entry) { | 1298 void HttpCache::ProcessWaitingTransactions(ActiveEntry* entry) { |
1023 // Multiple readers may finish with an entry at once, so we want to batch up | 1299 // Multiple readers may finish with an entry at once, so we want to batch up |
1024 // calls to OnProcessPendingQueue. This flag also tells us that we should | 1300 // calls to OnProcessWaitingTransactions. This flag also tells us that we |
1025 // not delete the entry before OnProcessPendingQueue runs. | 1301 // should not delete the entry before OnProcessWaitingTransactions runs. |
1026 if (entry->will_process_pending_queue) | 1302 if (entry->will_process_waiting_transactions) |
1027 return; | 1303 return; |
1028 entry->will_process_pending_queue = true; | |
1029 | 1304 |
| 1305 entry->will_process_waiting_transactions = true; |
| 1306 |
| 1307 // Post a task instead of invoking the io callback of another transaction here |
| 1308 // to avoid re-entrancy. |
1030 base::ThreadTaskRunnerHandle::Get()->PostTask( | 1309 base::ThreadTaskRunnerHandle::Get()->PostTask( |
1031 FROM_HERE, | 1310 FROM_HERE, base::Bind(&HttpCache::OnProcessWaitingTransactions, |
1032 base::Bind(&HttpCache::OnProcessPendingQueue, GetWeakPtr(), entry)); | 1311 GetWeakPtr(), entry->disk_entry->GetKey(), entry)); |
1033 } | 1312 } |
1034 | 1313 |
1035 void HttpCache::OnProcessPendingQueue(ActiveEntry* entry) { | 1314 void HttpCache::OnProcessWaitingTransactions(const std::string& key, |
1036 entry->will_process_pending_queue = false; | 1315 ActiveEntry* entry) { |
1037 DCHECK(!entry->writer); | 1316 // Check if the entry is still alive either as an active entry or doomed |
| 1317 // entry. |
| 1318 if (!IsEntryAlive(key, entry)) |
| 1319 return; |
| 1320 |
| 1321 entry->will_process_waiting_transactions = false; |
1038 | 1322 |
1039 // If no one is interested in this entry, then we can deactivate it. | 1323 // If no one is interested in this entry, then we can deactivate it. |
1040 if (entry->HasNoTransactions()) { | 1324 if (entry->HasNoTransactions()) { |
| 1325 DestroyEntry(entry); |
| 1326 return; |
| 1327 } |
| 1328 |
| 1329 if (entry->validated_queue.empty() && entry->add_to_entry_queue.empty()) |
| 1330 return; |
| 1331 |
| 1332 Transaction* next = nullptr; |
| 1333 TransactionList* list = nullptr; |
| 1334 int rv = OK; |
| 1335 |
| 1336 // To maintain FIFO order of transactions, validated_queue should be processed |
| 1337 // first and then add_to_entry_queue. |
| 1338 if (!entry->validated_queue.empty()) { |
| 1339 next = entry->validated_queue.front(); |
| 1340 list = &entry->validated_queue; |
| 1341 rv = AddTransactionToEntryForRead(entry, next); |
| 1342 } else { |
| 1343 // Promote next transaction from the pending queue. |
| 1344 next = entry->add_to_entry_queue.front(); |
| 1345 list = &entry->add_to_entry_queue; |
| 1346 rv = AddTransactionToEntryImpl(entry, next); |
| 1347 } |
| 1348 |
| 1349 if (rv != ERR_IO_PENDING) { |
| 1350 list->erase(list->begin()); |
| 1351 next->io_callback().Run(rv); |
| 1352 } |
| 1353 } |
| 1354 |
| 1355 void HttpCache::ProcessParallelValidation(ActiveEntry* entry) { |
| 1356 if (entry->will_process_parallel_validation) |
| 1357 return; |
| 1358 |
| 1359 entry->will_process_parallel_validation = true; |
| 1360 |
| 1361 // Post a task instead of invoking the io callback of another transaction here |
| 1362 // to avoid re-entrancy. |
| 1363 base::ThreadTaskRunnerHandle::Get()->PostTask( |
| 1364 FROM_HERE, base::Bind(&HttpCache::OnProcessParallelValidation, |
| 1365 GetWeakPtr(), entry->disk_entry->GetKey(), entry)); |
| 1366 } |
| 1367 |
| 1368 void HttpCache::OnProcessParallelValidation(const std::string& key, |
| 1369 ActiveEntry* entry) { |
| 1370 // Check if the entry is still alive either as an active entry or doomed |
| 1371 // entry. |
| 1372 if (!IsEntryAlive(key, entry)) |
| 1373 return; |
| 1374 |
| 1375 entry->will_process_parallel_validation = false; |
| 1376 |
| 1377 // A ProcessWaitingTransactions task before this may add a |
| 1378 // validating_transaction. In that case, do nothing. |
| 1379 if (entry->validating_transaction) |
| 1380 return; |
| 1381 |
| 1382 // If no one is interested in this entry, then we can deactivate it. |
| 1383 if (entry->HasNoTransactions()) { |
1041 DestroyEntry(entry); | 1384 DestroyEntry(entry); |
1042 return; | 1385 return; |
1043 } | 1386 } |
1044 | 1387 |
1045 if (entry->pending_queue.empty()) | 1388 if (entry->add_to_entry_queue.empty()) |
1046 return; | 1389 return; |
1047 | 1390 |
1048 // Promote next transaction from the pending queue. | 1391 Transaction* next = entry->add_to_entry_queue.front(); |
1049 Transaction* next = entry->pending_queue.front(); | |
1050 if ((next->mode() & Transaction::WRITE) && !entry->readers.empty()) | |
1051 return; // Have to wait. | |
1052 | 1392 |
1053 entry->pending_queue.erase(entry->pending_queue.begin()); | 1393 int rv = AddTransactionToEntryImpl(entry, next); |
1054 | 1394 |
1055 int rv = AddTransactionToEntry(entry, next); | 1395 // Read only transaction cannot validate the entry. |
| 1396 if (!(next->mode() & Transaction::WRITE)) |
| 1397 return; |
| 1398 |
1056 if (rv != ERR_IO_PENDING) { | 1399 if (rv != ERR_IO_PENDING) { |
| 1400 entry->add_to_entry_queue.erase(entry->add_to_entry_queue.begin()); |
1057 next->io_callback().Run(rv); | 1401 next->io_callback().Run(rv); |
1058 } | 1402 } |
1059 } | 1403 } |
1060 | 1404 |
| 1405 void HttpCache::DoomEntryRestartPendingTransactions(ActiveEntry* entry) { |
| 1406 DoomActiveEntry(entry->disk_entry->GetKey()); |
| 1407 TransactionList list = GetAllPendingTransactions(entry); |
| 1408 RestartPendingTransactions(list); |
| 1409 } |
| 1410 |
| 1411 void HttpCache::DestroyEntryRestartPendingTransactions(ActiveEntry* entry) { |
| 1412 entry->disk_entry->Doom(); |
| 1413 TransactionList list = GetAllPendingTransactions(entry); |
| 1414 DestroyEntry(entry); |
| 1415 RestartPendingTransactions(list); |
| 1416 } |
| 1417 |
| 1418 HttpCache::TransactionList HttpCache::GetAllPendingTransactions( |
| 1419 ActiveEntry* entry) { |
| 1420 TransactionList list; |
| 1421 for (auto* transaction : entry->validated_queue) |
| 1422 list.push_back(transaction); |
| 1423 entry->validated_queue.clear(); |
| 1424 |
| 1425 for (auto* transaction : entry->add_to_entry_queue) |
| 1426 list.push_back(transaction); |
| 1427 entry->add_to_entry_queue.clear(); |
| 1428 |
| 1429 return list; |
| 1430 } |
| 1431 |
| 1432 void HttpCache::RestartPendingTransactions(const TransactionList& list) { |
| 1433 // ERR_CACHE_RACE causes the transaction to restart the whole process. |
| 1434 for (auto* transaction : list) |
| 1435 transaction->io_callback().Run(ERR_CACHE_RACE); |
| 1436 } |
| 1437 |
1061 void HttpCache::OnIOComplete(int result, PendingOp* pending_op) { | 1438 void HttpCache::OnIOComplete(int result, PendingOp* pending_op) { |
1062 WorkItemOperation op = pending_op->writer->operation(); | 1439 WorkItemOperation op = pending_op->writer->operation(); |
1063 | 1440 |
1064 // Completing the creation of the backend is simpler than the other cases. | 1441 // Completing the creation of the backend is simpler than the other cases. |
1065 if (op == WI_CREATE_BACKEND) | 1442 if (op == WI_CREATE_BACKEND) |
1066 return OnBackendCreated(result, pending_op); | 1443 return OnBackendCreated(result, pending_op); |
1067 | 1444 |
1068 std::unique_ptr<WorkItem> item = std::move(pending_op->writer); | 1445 std::unique_ptr<WorkItem> item = std::move(pending_op->writer); |
1069 bool fail_requests = false; | 1446 bool fail_requests = false; |
1070 | 1447 |
(...skipping 124 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
1195 building_backend_ = false; | 1572 building_backend_ = false; |
1196 DeletePendingOp(pending_op); | 1573 DeletePendingOp(pending_op); |
1197 } | 1574 } |
1198 | 1575 |
1199 // The cache may be gone when we return from the callback. | 1576 // The cache may be gone when we return from the callback. |
1200 if (!item->DoCallback(result, disk_cache_.get())) | 1577 if (!item->DoCallback(result, disk_cache_.get())) |
1201 item->NotifyTransaction(result, NULL); | 1578 item->NotifyTransaction(result, NULL); |
1202 } | 1579 } |
1203 | 1580 |
1204 } // namespace net | 1581 } // namespace net |
OLD | NEW |