Add a Restore() method to ValueStore and make StorageAPI use it

chrome/browser/extensions/api/storage/storage_api.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "chrome/browser/extensions/api/storage/storage_api.h"
 
 #include <string>
 #include <vector>
 
 #include "base/bind.h"
@@ skipping 55 matching lines @@
 }
 
 void SettingsFunction::AsyncRunWithStorage(ValueStore* storage) {
   bool success = RunWithStorage(storage);
   BrowserThread::PostTask(
       BrowserThread::UI,
       FROM_HERE,
       base::Bind(&SettingsFunction::SendResponse, this, success));
 }
 
-bool SettingsFunction::UseReadResult(ValueStore::ReadResult read_result) {
-  if (read_result->HasError()) {
-    error_ = read_result->error().message;
-    return false;
-  }
+bool SettingsFunction::UseReadResult(ValueStore::ReadResult result,
+                                     ValueStore* storage) {
+  if (result->HasError())
+    return HandleError(result->error(), storage);
 
-  base::DictionaryValue* result = new base::DictionaryValue();
-  result->Swap(&read_result->settings());
-  SetResult(result);
+  base::DictionaryValue* dict = new base::DictionaryValue();
+  dict->Swap(&result->settings());
+  SetResult(dict);
   return true;
 }
 
-bool SettingsFunction::UseWriteResult(ValueStore::WriteResult result) {
-  if (result->HasError()) {
-    error_ = result->error().message;
-    return false;
-  }
+bool SettingsFunction::UseWriteResult(ValueStore::WriteResult result,
+                                      ValueStore* storage) {
+  if (result->HasError())
+    return HandleError(result->error(), storage);
 
   if (!result->changes().empty()) {
     observers_->Notify(
         &SettingsObserver::OnSettingsChanged,
         extension_id(),
         settings_namespace_,
         ValueStoreChange::ToJson(result->changes()));
   }
 
   return true;
 }
 
+bool SettingsFunction::HandleError(const ValueStore::Error& error,
+                                   ValueStore* storage) {
+  // If the method failed due to corruption, and we haven't tried to fix it, we
+  // can try to restore the storage and re-run it. Otherwise, the method has
+  // failed.
+  if (error.code == ValueStore::CORRUPTION && !tried_restoring_storage_) {
+    tried_restoring_storage_ = true;
+
+    // If the corruption is on a particular key, try to restore that key and
+    // re-run.
+    if (error.key.get() && storage->RestoreKey(*error.key))
+      return RunWithStorage(storage);
+
+    // If the full database is corrupted, try to restore the whole thing and
+    // re-run.
+    if (storage->Restore())
+      return RunWithStorage(storage);
+  }
+
+  error_ = error.message;
+  return false;
+}
+
 // Concrete settings functions
 
 namespace {
 
 // Adds all StringValues from a ListValue to a vector of strings.
 void AddAllStringValues(const base::ListValue& from,
                         std::vector<std::string>* to) {
   DCHECK(to->empty());
   std::string as_string;
   for (base::ListValue::const_iterator it = from.begin();
@@ skipping 39 matching lines @@
 };
 
 }  // namespace
 
 bool StorageStorageAreaGetFunction::RunWithStorage(ValueStore* storage) {
   base::Value* input = NULL;
   EXTENSION_FUNCTION_VALIDATE(args_->Get(0, &input));
 
   switch (input->GetType()) {
     case base::Value::TYPE_NULL:
-      return UseReadResult(storage->Get());
+      return UseReadResult(storage->Get(), storage);
 
     case base::Value::TYPE_STRING: {
       std::string as_string;
       input->GetAsString(&as_string);
-      return UseReadResult(storage->Get(as_string));
+      return UseReadResult(storage->Get(as_string), storage);
     }
 
     case base::Value::TYPE_LIST: {
       std::vector<std::string> as_string_list;
       AddAllStringValues(*static_cast<base::ListValue*>(input),
                          &as_string_list);
-      return UseReadResult(storage->Get(as_string_list));
+      return UseReadResult(storage->Get(as_string_list), storage);
     }
 
     case base::Value::TYPE_DICTIONARY: {
-      base::DictionaryValue* as_dict = static_cast<base::DictionaryValue*>(input);
+      base::DictionaryValue* as_dict =
+          static_cast<base::DictionaryValue*>(input);
       ValueStore::ReadResult result = storage->Get(GetKeys(*as_dict));
       if (result->HasError()) {
-        return UseReadResult(result.Pass());
+        return UseReadResult(result.Pass(), storage);
       }
 
       base::DictionaryValue* with_default_values = as_dict->DeepCopy();
       with_default_values->MergeDictionary(&result->settings());
       return UseReadResult(
-          ValueStore::MakeReadResult(make_scoped_ptr(with_default_values)));
+          ValueStore::MakeReadResult(make_scoped_ptr(with_default_values)),
+          storage);
     }
 
     default:
       EXTENSION_FUNCTION_VALIDATE(false);
       return false;
   }
 }
 
 bool StorageStorageAreaGetBytesInUseFunction::RunWithStorage(
     ValueStore* storage) {
@@ skipping 27 matching lines @@
       return false;
   }
 
   SetResult(new base::FundamentalValue(static_cast<int>(bytes_in_use)));
   return true;
 }
 
 bool StorageStorageAreaSetFunction::RunWithStorage(ValueStore* storage) {
   base::DictionaryValue* input = NULL;
   EXTENSION_FUNCTION_VALIDATE(args_->GetDictionary(0, &input));
-  return UseWriteResult(storage->Set(ValueStore::DEFAULTS, *input));
+  return UseWriteResult(storage->Set(ValueStore::DEFAULTS, *input), storage);
 }
 
 void StorageStorageAreaSetFunction::GetQuotaLimitHeuristics(
     QuotaLimitHeuristics* heuristics) const {
   GetModificationQuotaLimitHeuristics(heuristics);
 }
 
 bool StorageStorageAreaRemoveFunction::RunWithStorage(ValueStore* storage) {
   base::Value* input = NULL;
   EXTENSION_FUNCTION_VALIDATE(args_->Get(0, &input));
 
   switch (input->GetType()) {
     case base::Value::TYPE_STRING: {
       std::string as_string;
       input->GetAsString(&as_string);
-      return UseWriteResult(storage->Remove(as_string));
+      return UseWriteResult(storage->Remove(as_string), storage);
     }
 
     case base::Value::TYPE_LIST: {
       std::vector<std::string> as_string_list;
       AddAllStringValues(*static_cast<base::ListValue*>(input),
                          &as_string_list);
-      return UseWriteResult(storage->Remove(as_string_list));
+      return UseWriteResult(storage->Remove(as_string_list), storage);
     }
 
     default:
       EXTENSION_FUNCTION_VALIDATE(false);
       return false;
   };
 }
 
 void StorageStorageAreaRemoveFunction::GetQuotaLimitHeuristics(
     QuotaLimitHeuristics* heuristics) const {
   GetModificationQuotaLimitHeuristics(heuristics);
 }
 
 bool StorageStorageAreaClearFunction::RunWithStorage(ValueStore* storage) {
-  return UseWriteResult(storage->Clear());
+  return UseWriteResult(storage->Clear(), storage);
 }
 
 void StorageStorageAreaClearFunction::GetQuotaLimitHeuristics(
     QuotaLimitHeuristics* heuristics) const {
   GetModificationQuotaLimitHeuristics(heuristics);
 }
 
 }  // namespace extensions