• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
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
3 // found in the LICENSE file.
4 
5 #include "net/disk_cache/blockfile/rankings.h"
6 
7 #include "base/metrics/histogram.h"
8 #include "net/disk_cache/blockfile/backend_impl.h"
9 #include "net/disk_cache/blockfile/disk_format.h"
10 #include "net/disk_cache/blockfile/entry_impl.h"
11 #include "net/disk_cache/blockfile/errors.h"
12 #include "net/disk_cache/blockfile/histogram_macros.h"
13 #include "net/disk_cache/blockfile/stress_support.h"
14 
15 // Provide a BackendImpl object to macros from histogram_macros.h.
16 #define CACHE_UMA_BACKEND_IMPL_OBJ backend_
17 
18 using base::Time;
19 using base::TimeTicks;
20 
21 namespace disk_cache {
22 // This is used by crash_cache.exe to generate unit test files.
23 NET_EXPORT_PRIVATE RankCrashes g_rankings_crash = NO_CRASH;
24 }
25 
26 namespace {
27 
28 enum Operation {
29   INSERT = 1,
30   REMOVE
31 };
32 
33 // This class provides a simple lock for the LRU list of rankings. Whenever an
34 // entry is to be inserted or removed from the list, a transaction object should
35 // be created to keep track of the operation. If the process crashes before
36 // finishing the operation, the transaction record (stored as part of the user
37 // data on the file header) can be used to finish the operation.
38 class Transaction {
39  public:
40   // addr is the cache addres of the node being inserted or removed. We want to
41   // avoid having the compiler doing optimizations on when to read or write
42   // from user_data because it is the basis of the crash detection. Maybe
43   // volatile is not enough for that, but it should be a good hint.
44   Transaction(volatile disk_cache::LruData* data, disk_cache::Addr addr,
45               Operation op, int list);
46   ~Transaction();
47  private:
48   volatile disk_cache::LruData* data_;
49   DISALLOW_COPY_AND_ASSIGN(Transaction);
50 };
51 
Transaction(volatile disk_cache::LruData * data,disk_cache::Addr addr,Operation op,int list)52 Transaction::Transaction(volatile disk_cache::LruData* data,
53                          disk_cache::Addr addr, Operation op, int list)
54     : data_(data) {
55   DCHECK(!data_->transaction);
56   DCHECK(addr.is_initialized());
57   data_->operation = op;
58   data_->operation_list = list;
59   data_->transaction = addr.value();
60 }
61 
~Transaction()62 Transaction::~Transaction() {
63   DCHECK(data_->transaction);
64   data_->transaction = 0;
65   data_->operation = 0;
66   data_->operation_list = 0;
67 }
68 
69 // Code locations that can generate crashes.
70 enum CrashLocation {
71   ON_INSERT_1, ON_INSERT_2, ON_INSERT_3, ON_INSERT_4, ON_REMOVE_1, ON_REMOVE_2,
72   ON_REMOVE_3, ON_REMOVE_4, ON_REMOVE_5, ON_REMOVE_6, ON_REMOVE_7, ON_REMOVE_8
73 };
74 
75 #ifndef NDEBUG
TerminateSelf()76 void TerminateSelf() {
77 #if defined(OS_WIN)
78   // Windows does more work on _exit() than we would like, so we force exit.
79   TerminateProcess(GetCurrentProcess(), 0);
80 #elif defined(OS_POSIX)
81   // On POSIX, _exit() will terminate the process with minimal cleanup,
82   // and it is cleaner than killing.
83   _exit(0);
84 #endif
85 }
86 #endif  // NDEBUG
87 
88 // Generates a crash on debug builds, acording to the value of g_rankings_crash.
89 // This used by crash_cache.exe to generate unit-test files.
GenerateCrash(CrashLocation location)90 void GenerateCrash(CrashLocation location) {
91 #ifndef NDEBUG
92   if (disk_cache::NO_CRASH == disk_cache::g_rankings_crash)
93     return;
94   switch (location) {
95     case ON_INSERT_1:
96       switch (disk_cache::g_rankings_crash) {
97         case disk_cache::INSERT_ONE_1:
98         case disk_cache::INSERT_LOAD_1:
99           TerminateSelf();
100         default:
101           break;
102       }
103       break;
104     case ON_INSERT_2:
105       if (disk_cache::INSERT_EMPTY_1 == disk_cache::g_rankings_crash)
106         TerminateSelf();
107       break;
108     case ON_INSERT_3:
109       switch (disk_cache::g_rankings_crash) {
110         case disk_cache::INSERT_EMPTY_2:
111         case disk_cache::INSERT_ONE_2:
112         case disk_cache::INSERT_LOAD_2:
113           TerminateSelf();
114         default:
115           break;
116       }
117       break;
118     case ON_INSERT_4:
119       switch (disk_cache::g_rankings_crash) {
120         case disk_cache::INSERT_EMPTY_3:
121         case disk_cache::INSERT_ONE_3:
122           TerminateSelf();
123         default:
124           break;
125       }
126       break;
127     case ON_REMOVE_1:
128       switch (disk_cache::g_rankings_crash) {
129         case disk_cache::REMOVE_ONE_1:
130         case disk_cache::REMOVE_HEAD_1:
131         case disk_cache::REMOVE_TAIL_1:
132         case disk_cache::REMOVE_LOAD_1:
133           TerminateSelf();
134         default:
135           break;
136       }
137       break;
138     case ON_REMOVE_2:
139       if (disk_cache::REMOVE_ONE_2 == disk_cache::g_rankings_crash)
140         TerminateSelf();
141       break;
142     case ON_REMOVE_3:
143       if (disk_cache::REMOVE_ONE_3 == disk_cache::g_rankings_crash)
144         TerminateSelf();
145       break;
146     case ON_REMOVE_4:
147       if (disk_cache::REMOVE_HEAD_2 == disk_cache::g_rankings_crash)
148         TerminateSelf();
149       break;
150     case ON_REMOVE_5:
151       if (disk_cache::REMOVE_TAIL_2 == disk_cache::g_rankings_crash)
152         TerminateSelf();
153       break;
154     case ON_REMOVE_6:
155       if (disk_cache::REMOVE_TAIL_3 == disk_cache::g_rankings_crash)
156         TerminateSelf();
157       break;
158     case ON_REMOVE_7:
159       switch (disk_cache::g_rankings_crash) {
160         case disk_cache::REMOVE_ONE_4:
161         case disk_cache::REMOVE_LOAD_2:
162         case disk_cache::REMOVE_HEAD_3:
163           TerminateSelf();
164         default:
165           break;
166       }
167       break;
168     case ON_REMOVE_8:
169       switch (disk_cache::g_rankings_crash) {
170         case disk_cache::REMOVE_HEAD_4:
171         case disk_cache::REMOVE_LOAD_3:
172           TerminateSelf();
173         default:
174           break;
175       }
176       break;
177     default:
178       NOTREACHED();
179       return;
180   }
181 #endif  // NDEBUG
182 }
183 
184 // Update the timestamp fields of |node|.
UpdateTimes(disk_cache::CacheRankingsBlock * node,bool modified)185 void UpdateTimes(disk_cache::CacheRankingsBlock* node, bool modified) {
186   base::Time now = base::Time::Now();
187   node->Data()->last_used = now.ToInternalValue();
188   if (modified)
189     node->Data()->last_modified = now.ToInternalValue();
190 }
191 
192 }  // namespace
193 
194 namespace disk_cache {
195 
ScopedRankingsBlock()196 Rankings::ScopedRankingsBlock::ScopedRankingsBlock() : rankings_(NULL) {}
197 
ScopedRankingsBlock(Rankings * rankings)198 Rankings::ScopedRankingsBlock::ScopedRankingsBlock(Rankings* rankings)
199     : rankings_(rankings) {}
200 
ScopedRankingsBlock(Rankings * rankings,CacheRankingsBlock * node)201 Rankings::ScopedRankingsBlock::ScopedRankingsBlock(
202     Rankings* rankings, CacheRankingsBlock* node)
203     : scoped_ptr<CacheRankingsBlock>(node), rankings_(rankings) {}
204 
Iterator()205 Rankings::Iterator::Iterator() {
206   memset(this, 0, sizeof(Iterator));
207 }
208 
Reset()209 void Rankings::Iterator::Reset() {
210   if (my_rankings) {
211     for (int i = 0; i < 3; i++)
212       ScopedRankingsBlock(my_rankings, nodes[i]);
213   }
214   memset(this, 0, sizeof(Iterator));
215 }
216 
Rankings()217 Rankings::Rankings() : init_(false) {}
218 
~Rankings()219 Rankings::~Rankings() {}
220 
Init(BackendImpl * backend,bool count_lists)221 bool Rankings::Init(BackendImpl* backend, bool count_lists) {
222   DCHECK(!init_);
223   if (init_)
224     return false;
225 
226   backend_ = backend;
227   control_data_ = backend_->GetLruData();
228   count_lists_ = count_lists;
229 
230   ReadHeads();
231   ReadTails();
232 
233   if (control_data_->transaction)
234     CompleteTransaction();
235 
236   init_ = true;
237   return true;
238 }
239 
Reset()240 void Rankings::Reset() {
241   init_ = false;
242   for (int i = 0; i < LAST_ELEMENT; i++) {
243     heads_[i].set_value(0);
244     tails_[i].set_value(0);
245   }
246   control_data_ = NULL;
247 }
248 
Insert(CacheRankingsBlock * node,bool modified,List list)249 void Rankings::Insert(CacheRankingsBlock* node, bool modified, List list) {
250   Trace("Insert 0x%x l %d", node->address().value(), list);
251   DCHECK(node->HasData());
252   Addr& my_head = heads_[list];
253   Addr& my_tail = tails_[list];
254   Transaction lock(control_data_, node->address(), INSERT, list);
255   CacheRankingsBlock head(backend_->File(my_head), my_head);
256   if (my_head.is_initialized()) {
257     if (!GetRanking(&head))
258       return;
259 
260     if (head.Data()->prev != my_head.value() &&  // Normal path.
261         head.Data()->prev != node->address().value()) {  // FinishInsert().
262       backend_->CriticalError(ERR_INVALID_LINKS);
263       return;
264     }
265 
266     head.Data()->prev = node->address().value();
267     head.Store();
268     GenerateCrash(ON_INSERT_1);
269     UpdateIterators(&head);
270   }
271 
272   node->Data()->next = my_head.value();
273   node->Data()->prev = node->address().value();
274   my_head.set_value(node->address().value());
275 
276   if (!my_tail.is_initialized() || my_tail.value() == node->address().value()) {
277     my_tail.set_value(node->address().value());
278     node->Data()->next = my_tail.value();
279     WriteTail(list);
280     GenerateCrash(ON_INSERT_2);
281   }
282 
283   UpdateTimes(node, modified);
284   node->Store();
285   GenerateCrash(ON_INSERT_3);
286 
287   // The last thing to do is move our head to point to a node already stored.
288   WriteHead(list);
289   IncrementCounter(list);
290   GenerateCrash(ON_INSERT_4);
291   backend_->FlushIndex();
292 }
293 
294 // If a, b and r are elements on the list, and we want to remove r, the possible
295 // states for the objects if a crash happens are (where y(x, z) means for object
296 // y, prev is x and next is z):
297 // A. One element:
298 //    1. r(r, r), head(r), tail(r)                    initial state
299 //    2. r(r, r), head(0), tail(r)                    WriteHead()
300 //    3. r(r, r), head(0), tail(0)                    WriteTail()
301 //    4. r(0, 0), head(0), tail(0)                    next.Store()
302 //
303 // B. Remove a random element:
304 //    1. a(x, r), r(a, b), b(r, y), head(x), tail(y)  initial state
305 //    2. a(x, r), r(a, b), b(a, y), head(x), tail(y)  next.Store()
306 //    3. a(x, b), r(a, b), b(a, y), head(x), tail(y)  prev.Store()
307 //    4. a(x, b), r(0, 0), b(a, y), head(x), tail(y)  node.Store()
308 //
309 // C. Remove head:
310 //    1. r(r, b), b(r, y), head(r), tail(y)           initial state
311 //    2. r(r, b), b(r, y), head(b), tail(y)           WriteHead()
312 //    3. r(r, b), b(b, y), head(b), tail(y)           next.Store()
313 //    4. r(0, 0), b(b, y), head(b), tail(y)           prev.Store()
314 //
315 // D. Remove tail:
316 //    1. a(x, r), r(a, r), head(x), tail(r)           initial state
317 //    2. a(x, r), r(a, r), head(x), tail(a)           WriteTail()
318 //    3. a(x, a), r(a, r), head(x), tail(a)           prev.Store()
319 //    4. a(x, a), r(0, 0), head(x), tail(a)           next.Store()
Remove(CacheRankingsBlock * node,List list,bool strict)320 void Rankings::Remove(CacheRankingsBlock* node, List list, bool strict) {
321   Trace("Remove 0x%x (0x%x 0x%x) l %d", node->address().value(),
322         node->Data()->next, node->Data()->prev, list);
323   DCHECK(node->HasData());
324   if (strict)
325     InvalidateIterators(node);
326 
327   Addr next_addr(node->Data()->next);
328   Addr prev_addr(node->Data()->prev);
329   if (!next_addr.is_initialized() || next_addr.is_separate_file() ||
330       !prev_addr.is_initialized() || prev_addr.is_separate_file()) {
331     if (next_addr.is_initialized() || prev_addr.is_initialized()) {
332       LOG(ERROR) << "Invalid rankings info.";
333       STRESS_NOTREACHED();
334     }
335     return;
336   }
337 
338   CacheRankingsBlock next(backend_->File(next_addr), next_addr);
339   CacheRankingsBlock prev(backend_->File(prev_addr), prev_addr);
340   if (!GetRanking(&next) || !GetRanking(&prev)) {
341     STRESS_NOTREACHED();
342     return;
343   }
344 
345   if (!CheckLinks(node, &prev, &next, &list))
346     return;
347 
348   Transaction lock(control_data_, node->address(), REMOVE, list);
349   prev.Data()->next = next.address().value();
350   next.Data()->prev = prev.address().value();
351   GenerateCrash(ON_REMOVE_1);
352 
353   CacheAddr node_value = node->address().value();
354   Addr& my_head = heads_[list];
355   Addr& my_tail = tails_[list];
356   if (node_value == my_head.value() || node_value == my_tail.value()) {
357     if (my_head.value() == my_tail.value()) {
358       my_head.set_value(0);
359       my_tail.set_value(0);
360 
361       WriteHead(list);
362       GenerateCrash(ON_REMOVE_2);
363       WriteTail(list);
364       GenerateCrash(ON_REMOVE_3);
365     } else if (node_value == my_head.value()) {
366       my_head.set_value(next.address().value());
367       next.Data()->prev = next.address().value();
368 
369       WriteHead(list);
370       GenerateCrash(ON_REMOVE_4);
371     } else if (node_value == my_tail.value()) {
372       my_tail.set_value(prev.address().value());
373       prev.Data()->next = prev.address().value();
374 
375       WriteTail(list);
376       GenerateCrash(ON_REMOVE_5);
377 
378       // Store the new tail to make sure we can undo the operation if we crash.
379       prev.Store();
380       GenerateCrash(ON_REMOVE_6);
381     }
382   }
383 
384   // Nodes out of the list can be identified by invalid pointers.
385   node->Data()->next = 0;
386   node->Data()->prev = 0;
387 
388   // The last thing to get to disk is the node itself, so before that there is
389   // enough info to recover.
390   next.Store();
391   GenerateCrash(ON_REMOVE_7);
392   prev.Store();
393   GenerateCrash(ON_REMOVE_8);
394   node->Store();
395   DecrementCounter(list);
396   UpdateIterators(&next);
397   UpdateIterators(&prev);
398   backend_->FlushIndex();
399 }
400 
401 // A crash in between Remove and Insert will lead to a dirty entry not on the
402 // list. We want to avoid that case as much as we can (as while waiting for IO),
403 // but the net effect is just an assert on debug when attempting to remove the
404 // entry. Otherwise we'll need reentrant transactions, which is an overkill.
UpdateRank(CacheRankingsBlock * node,bool modified,List list)405 void Rankings::UpdateRank(CacheRankingsBlock* node, bool modified, List list) {
406   Addr& my_head = heads_[list];
407   if (my_head.value() == node->address().value()) {
408     UpdateTimes(node, modified);
409     node->set_modified();
410     return;
411   }
412 
413   TimeTicks start = TimeTicks::Now();
414   Remove(node, list, true);
415   Insert(node, modified, list);
416   CACHE_UMA(AGE_MS, "UpdateRank", 0, start);
417 }
418 
GetNext(CacheRankingsBlock * node,List list)419 CacheRankingsBlock* Rankings::GetNext(CacheRankingsBlock* node, List list) {
420   ScopedRankingsBlock next(this);
421   if (!node) {
422     Addr& my_head = heads_[list];
423     if (!my_head.is_initialized())
424       return NULL;
425     next.reset(new CacheRankingsBlock(backend_->File(my_head), my_head));
426   } else {
427     if (!node->HasData())
428       node->Load();
429     Addr& my_tail = tails_[list];
430     if (!my_tail.is_initialized())
431       return NULL;
432     if (my_tail.value() == node->address().value())
433       return NULL;
434     Addr address(node->Data()->next);
435     if (address.value() == node->address().value())
436       return NULL;  // Another tail? fail it.
437     next.reset(new CacheRankingsBlock(backend_->File(address), address));
438   }
439 
440   TrackRankingsBlock(next.get(), true);
441 
442   if (!GetRanking(next.get()))
443     return NULL;
444 
445   ConvertToLongLived(next.get());
446   if (node && !CheckSingleLink(node, next.get()))
447     return NULL;
448 
449   return next.release();
450 }
451 
GetPrev(CacheRankingsBlock * node,List list)452 CacheRankingsBlock* Rankings::GetPrev(CacheRankingsBlock* node, List list) {
453   ScopedRankingsBlock prev(this);
454   if (!node) {
455     Addr& my_tail = tails_[list];
456     if (!my_tail.is_initialized())
457       return NULL;
458     prev.reset(new CacheRankingsBlock(backend_->File(my_tail), my_tail));
459   } else {
460     if (!node->HasData())
461       node->Load();
462     Addr& my_head = heads_[list];
463     if (!my_head.is_initialized())
464       return NULL;
465     if (my_head.value() == node->address().value())
466       return NULL;
467     Addr address(node->Data()->prev);
468     if (address.value() == node->address().value())
469       return NULL;  // Another head? fail it.
470     prev.reset(new CacheRankingsBlock(backend_->File(address), address));
471   }
472 
473   TrackRankingsBlock(prev.get(), true);
474 
475   if (!GetRanking(prev.get()))
476     return NULL;
477 
478   ConvertToLongLived(prev.get());
479   if (node && !CheckSingleLink(prev.get(), node))
480     return NULL;
481 
482   return prev.release();
483 }
484 
FreeRankingsBlock(CacheRankingsBlock * node)485 void Rankings::FreeRankingsBlock(CacheRankingsBlock* node) {
486   TrackRankingsBlock(node, false);
487 }
488 
TrackRankingsBlock(CacheRankingsBlock * node,bool start_tracking)489 void Rankings::TrackRankingsBlock(CacheRankingsBlock* node,
490                                   bool start_tracking) {
491   if (!node)
492     return;
493 
494   IteratorPair current(node->address().value(), node);
495 
496   if (start_tracking)
497     iterators_.push_back(current);
498   else
499     iterators_.remove(current);
500 }
501 
SelfCheck()502 int Rankings::SelfCheck() {
503   int total = 0;
504   int error = 0;
505   for (int i = 0; i < LAST_ELEMENT; i++) {
506     int partial = CheckList(static_cast<List>(i));
507     if (partial < 0 && !error)
508       error = partial;
509     else if (partial > 0)
510       total += partial;
511   }
512 
513   return error ? error : total;
514 }
515 
SanityCheck(CacheRankingsBlock * node,bool from_list) const516 bool Rankings::SanityCheck(CacheRankingsBlock* node, bool from_list) const {
517   if (!node->VerifyHash())
518     return false;
519 
520   const RankingsNode* data = node->Data();
521 
522   if ((!data->next && data->prev) || (data->next && !data->prev))
523     return false;
524 
525   // Both pointers on zero is a node out of the list.
526   if (!data->next && !data->prev && from_list)
527     return false;
528 
529   List list = NO_USE;  // Initialize it to something.
530   if ((node->address().value() == data->prev) && !IsHead(data->prev, &list))
531     return false;
532 
533   if ((node->address().value() == data->next) && !IsTail(data->next, &list))
534     return false;
535 
536   if (!data->next && !data->prev)
537     return true;
538 
539   Addr next_addr(data->next);
540   Addr prev_addr(data->prev);
541   if (!next_addr.SanityCheckV2() || next_addr.file_type() != RANKINGS ||
542       !prev_addr.SanityCheckV2() || prev_addr.file_type() != RANKINGS)
543     return false;
544 
545   return true;
546 }
547 
DataSanityCheck(CacheRankingsBlock * node,bool from_list) const548 bool Rankings::DataSanityCheck(CacheRankingsBlock* node, bool from_list) const {
549   const RankingsNode* data = node->Data();
550   if (!data->contents)
551     return false;
552 
553   // It may have never been inserted.
554   if (from_list && (!data->last_used || !data->last_modified))
555     return false;
556 
557   return true;
558 }
559 
SetContents(CacheRankingsBlock * node,CacheAddr address)560 void Rankings::SetContents(CacheRankingsBlock* node, CacheAddr address) {
561   node->Data()->contents = address;
562   node->Store();
563 }
564 
ReadHeads()565 void Rankings::ReadHeads() {
566   for (int i = 0; i < LAST_ELEMENT; i++)
567     heads_[i] = Addr(control_data_->heads[i]);
568 }
569 
ReadTails()570 void Rankings::ReadTails() {
571   for (int i = 0; i < LAST_ELEMENT; i++)
572     tails_[i] = Addr(control_data_->tails[i]);
573 }
574 
WriteHead(List list)575 void Rankings::WriteHead(List list) {
576   control_data_->heads[list] = heads_[list].value();
577 }
578 
WriteTail(List list)579 void Rankings::WriteTail(List list) {
580   control_data_->tails[list] = tails_[list].value();
581 }
582 
GetRanking(CacheRankingsBlock * rankings)583 bool Rankings::GetRanking(CacheRankingsBlock* rankings) {
584   if (!rankings->address().is_initialized())
585     return false;
586 
587   TimeTicks start = TimeTicks::Now();
588   if (!rankings->Load())
589     return false;
590 
591   if (!SanityCheck(rankings, true)) {
592     backend_->CriticalError(ERR_INVALID_LINKS);
593     return false;
594   }
595 
596   backend_->OnEvent(Stats::OPEN_RANKINGS);
597 
598   // Note that if the cache is in read_only mode, open entries are not marked
599   // as dirty, except when an entry is doomed. We have to look for open entries.
600   if (!backend_->read_only() && !rankings->Data()->dirty)
601     return true;
602 
603   EntryImpl* entry = backend_->GetOpenEntry(rankings);
604   if (!entry) {
605     if (backend_->read_only())
606       return true;
607 
608     // We cannot trust this entry, but we cannot initiate a cleanup from this
609     // point (we may be in the middle of a cleanup already). The entry will be
610     // deleted when detected from a regular open/create path.
611     rankings->Data()->dirty = backend_->GetCurrentEntryId() - 1;
612     if (!rankings->Data()->dirty)
613       rankings->Data()->dirty--;
614     return true;
615   }
616 
617   // Note that we should not leave this module without deleting rankings first.
618   rankings->SetData(entry->rankings()->Data());
619 
620   CACHE_UMA(AGE_MS, "GetRankings", 0, start);
621   return true;
622 }
623 
ConvertToLongLived(CacheRankingsBlock * rankings)624 void Rankings::ConvertToLongLived(CacheRankingsBlock* rankings) {
625   if (rankings->own_data())
626     return;
627 
628   // We cannot return a shared node because we are not keeping a reference
629   // to the entry that owns the buffer. Make this node a copy of the one that
630   // we have, and let the iterator logic update it when the entry changes.
631   CacheRankingsBlock temp(NULL, Addr(0));
632   *temp.Data() = *rankings->Data();
633   rankings->StopSharingData();
634   *rankings->Data() = *temp.Data();
635 }
636 
CompleteTransaction()637 void Rankings::CompleteTransaction() {
638   Addr node_addr(static_cast<CacheAddr>(control_data_->transaction));
639   if (!node_addr.is_initialized() || node_addr.is_separate_file()) {
640     NOTREACHED();
641     LOG(ERROR) << "Invalid rankings info.";
642     return;
643   }
644 
645   Trace("CompleteTransaction 0x%x", node_addr.value());
646 
647   CacheRankingsBlock node(backend_->File(node_addr), node_addr);
648   if (!node.Load())
649     return;
650 
651   node.Store();
652 
653   Addr& my_head = heads_[control_data_->operation_list];
654   Addr& my_tail = tails_[control_data_->operation_list];
655 
656   // We want to leave the node inside the list. The entry must me marked as
657   // dirty, and will be removed later. Otherwise, we'll get assertions when
658   // attempting to remove the dirty entry.
659   if (INSERT == control_data_->operation) {
660     Trace("FinishInsert h:0x%x t:0x%x", my_head.value(), my_tail.value());
661     FinishInsert(&node);
662   } else if (REMOVE == control_data_->operation) {
663     Trace("RevertRemove h:0x%x t:0x%x", my_head.value(), my_tail.value());
664     RevertRemove(&node);
665   } else {
666     NOTREACHED();
667     LOG(ERROR) << "Invalid operation to recover.";
668   }
669 }
670 
FinishInsert(CacheRankingsBlock * node)671 void Rankings::FinishInsert(CacheRankingsBlock* node) {
672   control_data_->transaction = 0;
673   control_data_->operation = 0;
674   Addr& my_head = heads_[control_data_->operation_list];
675   Addr& my_tail = tails_[control_data_->operation_list];
676   if (my_head.value() != node->address().value()) {
677     if (my_tail.value() == node->address().value()) {
678       // This part will be skipped by the logic of Insert.
679       node->Data()->next = my_tail.value();
680     }
681 
682     Insert(node, true, static_cast<List>(control_data_->operation_list));
683   }
684 
685   // Tell the backend about this entry.
686   backend_->RecoveredEntry(node);
687 }
688 
RevertRemove(CacheRankingsBlock * node)689 void Rankings::RevertRemove(CacheRankingsBlock* node) {
690   Addr next_addr(node->Data()->next);
691   Addr prev_addr(node->Data()->prev);
692   if (!next_addr.is_initialized() || !prev_addr.is_initialized()) {
693     // The operation actually finished. Nothing to do.
694     control_data_->transaction = 0;
695     return;
696   }
697   if (next_addr.is_separate_file() || prev_addr.is_separate_file()) {
698     NOTREACHED();
699     LOG(WARNING) << "Invalid rankings info.";
700     control_data_->transaction = 0;
701     return;
702   }
703 
704   CacheRankingsBlock next(backend_->File(next_addr), next_addr);
705   CacheRankingsBlock prev(backend_->File(prev_addr), prev_addr);
706   if (!next.Load() || !prev.Load())
707     return;
708 
709   CacheAddr node_value = node->address().value();
710   DCHECK(prev.Data()->next == node_value ||
711          prev.Data()->next == prev_addr.value() ||
712          prev.Data()->next == next.address().value());
713   DCHECK(next.Data()->prev == node_value ||
714          next.Data()->prev == next_addr.value() ||
715          next.Data()->prev == prev.address().value());
716 
717   if (node_value != prev_addr.value())
718     prev.Data()->next = node_value;
719   if (node_value != next_addr.value())
720     next.Data()->prev = node_value;
721 
722   List my_list = static_cast<List>(control_data_->operation_list);
723   Addr& my_head = heads_[my_list];
724   Addr& my_tail = tails_[my_list];
725   if (!my_head.is_initialized() || !my_tail.is_initialized()) {
726     my_head.set_value(node_value);
727     my_tail.set_value(node_value);
728     WriteHead(my_list);
729     WriteTail(my_list);
730   } else if (my_head.value() == next.address().value()) {
731     my_head.set_value(node_value);
732     prev.Data()->next = next.address().value();
733     WriteHead(my_list);
734   } else if (my_tail.value() == prev.address().value()) {
735     my_tail.set_value(node_value);
736     next.Data()->prev = prev.address().value();
737     WriteTail(my_list);
738   }
739 
740   next.Store();
741   prev.Store();
742   control_data_->transaction = 0;
743   control_data_->operation = 0;
744   backend_->FlushIndex();
745 }
746 
CheckLinks(CacheRankingsBlock * node,CacheRankingsBlock * prev,CacheRankingsBlock * next,List * list)747 bool Rankings::CheckLinks(CacheRankingsBlock* node, CacheRankingsBlock* prev,
748                           CacheRankingsBlock* next, List* list) {
749   CacheAddr node_addr = node->address().value();
750   if (prev->Data()->next == node_addr &&
751       next->Data()->prev == node_addr) {
752     // A regular linked node.
753     return true;
754   }
755 
756   Trace("CheckLinks 0x%x (0x%x 0x%x)", node_addr,
757         prev->Data()->next, next->Data()->prev);
758 
759   if (node_addr != prev->address().value() &&
760       node_addr != next->address().value() &&
761       prev->Data()->next == next->address().value() &&
762       next->Data()->prev == prev->address().value()) {
763     // The list is actually ok, node is wrong.
764     Trace("node 0x%x out of list %d", node_addr, list);
765     node->Data()->next = 0;
766     node->Data()->prev = 0;
767     node->Store();
768     return false;
769   }
770 
771   if (prev->Data()->next == node_addr ||
772       next->Data()->prev == node_addr) {
773     // Only one link is weird, lets double check.
774     if (prev->Data()->next != node_addr && IsHead(node_addr, list))
775       return true;
776 
777     if (next->Data()->prev != node_addr && IsTail(node_addr, list))
778       return true;
779   }
780 
781   LOG(ERROR) << "Inconsistent LRU.";
782   STRESS_NOTREACHED();
783 
784   backend_->CriticalError(ERR_INVALID_LINKS);
785   return false;
786 }
787 
CheckSingleLink(CacheRankingsBlock * prev,CacheRankingsBlock * next)788 bool Rankings::CheckSingleLink(CacheRankingsBlock* prev,
789                                CacheRankingsBlock* next) {
790   if (prev->Data()->next != next->address().value() ||
791       next->Data()->prev != prev->address().value()) {
792     LOG(ERROR) << "Inconsistent LRU.";
793 
794     backend_->CriticalError(ERR_INVALID_LINKS);
795     return false;
796   }
797 
798   return true;
799 }
800 
CheckList(List list)801 int Rankings::CheckList(List list) {
802   Addr last1, last2;
803   int head_items;
804   int rv = CheckListSection(list, last1, last2, true,  // Head to tail.
805                             &last1, &last2, &head_items);
806   if (rv == ERR_NO_ERROR)
807     return head_items;
808 
809   return rv;
810 }
811 
812 // Note that the returned error codes assume a forward walk (from head to tail)
813 // so they have to be adjusted accordingly by the caller. We use two stop values
814 // to be able to detect a corrupt node at the end that is not linked going back.
CheckListSection(List list,Addr end1,Addr end2,bool forward,Addr * last,Addr * second_last,int * num_items)815 int Rankings::CheckListSection(List list, Addr end1, Addr end2, bool forward,
816                                Addr* last, Addr* second_last, int* num_items) {
817   Addr current = forward ? heads_[list] : tails_[list];
818   *last = *second_last = current;
819   *num_items = 0;
820   if (!current.is_initialized())
821     return ERR_NO_ERROR;
822 
823   if (!current.SanityCheckForRankings())
824     return ERR_INVALID_HEAD;
825 
826   scoped_ptr<CacheRankingsBlock> node;
827   Addr prev_addr(current);
828   do {
829     node.reset(new CacheRankingsBlock(backend_->File(current), current));
830     node->Load();
831     if (!SanityCheck(node.get(), true))
832       return ERR_INVALID_ENTRY;
833 
834     CacheAddr next = forward ? node->Data()->next : node->Data()->prev;
835     CacheAddr prev = forward ? node->Data()->prev : node->Data()->next;
836 
837     if (prev != prev_addr.value())
838       return ERR_INVALID_PREV;
839 
840     Addr next_addr(next);
841     if (!next_addr.SanityCheckForRankings())
842       return ERR_INVALID_NEXT;
843 
844     prev_addr = current;
845     current = next_addr;
846     *second_last = *last;
847     *last = current;
848     (*num_items)++;
849 
850     if (next_addr == prev_addr) {
851       Addr last = forward ? tails_[list] : heads_[list];
852       if (next_addr == last)
853         return ERR_NO_ERROR;
854       return ERR_INVALID_TAIL;
855     }
856   } while (current != end1 && current != end2);
857   return ERR_NO_ERROR;
858 }
859 
IsHead(CacheAddr addr,List * list) const860 bool Rankings::IsHead(CacheAddr addr, List* list) const {
861   for (int i = 0; i < LAST_ELEMENT; i++) {
862     if (addr == heads_[i].value()) {
863       if (*list != i)
864         Trace("Changing list %d to %d", *list, i);
865       *list = static_cast<List>(i);
866       return true;
867     }
868   }
869   return false;
870 }
871 
IsTail(CacheAddr addr,List * list) const872 bool Rankings::IsTail(CacheAddr addr, List* list) const {
873   for (int i = 0; i < LAST_ELEMENT; i++) {
874     if (addr == tails_[i].value()) {
875       if (*list != i)
876         Trace("Changing list %d to %d", *list, i);
877       *list = static_cast<List>(i);
878       return true;
879     }
880   }
881   return false;
882 }
883 
884 // We expect to have just a few iterators at any given time, maybe two or three,
885 // But we could have more than one pointing at the same mode. We walk the list
886 // of cache iterators and update all that are pointing to the given node.
UpdateIterators(CacheRankingsBlock * node)887 void Rankings::UpdateIterators(CacheRankingsBlock* node) {
888   CacheAddr address = node->address().value();
889   for (IteratorList::iterator it = iterators_.begin(); it != iterators_.end();
890        ++it) {
891     if (it->first == address && it->second->HasData()) {
892       CacheRankingsBlock* other = it->second;
893       *other->Data() = *node->Data();
894     }
895   }
896 }
897 
InvalidateIterators(CacheRankingsBlock * node)898 void Rankings::InvalidateIterators(CacheRankingsBlock* node) {
899   CacheAddr address = node->address().value();
900   for (IteratorList::iterator it = iterators_.begin(); it != iterators_.end();
901        ++it) {
902     if (it->first == address)
903       it->second->Discard();
904   }
905 }
906 
IncrementCounter(List list)907 void Rankings::IncrementCounter(List list) {
908   if (!count_lists_)
909     return;
910 
911   DCHECK(control_data_->sizes[list] < kint32max);
912   if (control_data_->sizes[list] < kint32max)
913     control_data_->sizes[list]++;
914 }
915 
DecrementCounter(List list)916 void Rankings::DecrementCounter(List list) {
917   if (!count_lists_)
918     return;
919 
920   DCHECK(control_data_->sizes[list] > 0);
921   if (control_data_->sizes[list] > 0)
922     control_data_->sizes[list]--;
923 }
924 
925 }  // namespace disk_cache
926