Move preference MACs to the protected preference stores.

chrome/browser/prefs/profile_pref_store_manager.cc

Index: chrome/browser/prefs/profile_pref_store_manager.cc
diff --git a/chrome/browser/prefs/profile_pref_store_manager.cc b/chrome/browser/prefs/profile_pref_store_manager.cc
index d5a64b05a4b8502064d0d1d536fce315db014b7f..0b56e3143bd290ac77eeaac1af6c0b50fd5d6f12 100644
--- a/chrome/browser/prefs/profile_pref_store_manager.cc
+++ b/chrome/browser/prefs/profile_pref_store_manager.cc
@@ -12,8 +12,10 @@
 #include "base/prefs/json_pref_store.h"
 #include "base/prefs/persistent_pref_store.h"
 #include "base/prefs/pref_registry_simple.h"
+#include "chrome/browser/prefs/dictionary_pref_store.h"
 #include "chrome/browser/prefs/pref_hash_store_impl.h"
 #include "chrome/browser/prefs/tracked/pref_service_hash_store_contents.h"
+#include "chrome/browser/prefs/tracked/pref_store_hash_store_contents.h"
 #include "chrome/browser/prefs/tracked/segregated_pref_store.h"
 #include "chrome/browser/prefs/tracked/tracked_preferences_migration.h"
 #include "chrome/common/chrome_constants.h"
@@ -22,175 +24,55 @@
 
 namespace {
 
-// An adaptor that allows a PrefHashStoreImpl to access a preference store
-// directly as a dictionary. Uses an equivalent layout to
-// PrefStoreHashStoreContents.
-class DictionaryHashStoreContents : public HashStoreContents {
+class MigrationDelegateImpl : public TrackedPreferencesMigrationDelegate {
  public:
-  // Instantiates a HashStoreContents that is a copy of |to_copy|. The copy is
-  // mutable but does not affect the original, nor is it persisted to disk in
-  // any other way.
-  explicit DictionaryHashStoreContents(const HashStoreContents& to_copy)
-      : hash_store_id_(to_copy.hash_store_id()),
-        super_mac_(to_copy.GetSuperMac()) {
-    if (to_copy.IsInitialized())
-      dictionary_.reset(to_copy.GetContents()->DeepCopy());
-    int version = 0;
-    if (to_copy.GetVersion(&version))
-      version_.reset(new int(version));
-  }
-
-  // HashStoreContents implementation
-  virtual std::string hash_store_id() const OVERRIDE { return hash_store_id_; }
-
-  virtual void Reset() OVERRIDE {
-    dictionary_.reset();
-    super_mac_.clear();
-    version_.reset();
-  }
-
-  virtual bool IsInitialized() const OVERRIDE {
-    return dictionary_;
-  }
-
-  virtual const base::DictionaryValue* GetContents() const OVERRIDE{
-    return dictionary_.get();
-  }
-
-  virtual scoped_ptr<MutableDictionary> GetMutableContents() OVERRIDE {
-    return scoped_ptr<MutableDictionary>(
-        new SimpleMutableDictionary(this));
-  }
-
-  virtual std::string GetSuperMac() const OVERRIDE { return super_mac_; }
-
-  virtual void SetSuperMac(const std::string& super_mac) OVERRIDE {
-    super_mac_ = super_mac;
-  }
-
-  virtual bool GetVersion(int* version) const OVERRIDE {
-    if (!version_)
-      return false;
-    *version = *version_;
-    return true;
-  }
-
-  virtual void SetVersion(int version) OVERRIDE {
-    version_.reset(new int(version));
-  }
-
-  virtual void CommitPendingWrite() OVERRIDE {}
-
- private:
-  class SimpleMutableDictionary
-      : public HashStoreContents::MutableDictionary {
-   public:
-    explicit SimpleMutableDictionary(DictionaryHashStoreContents* outer)
-        : outer_(outer) {}
-
-    virtual ~SimpleMutableDictionary() {}
-
-    // MutableDictionary implementation
-    virtual base::DictionaryValue* operator->() OVERRIDE {
-      if (!outer_->dictionary_)
-        outer_->dictionary_.reset(new base::DictionaryValue);
-      return outer_->dictionary_.get();
-    }
-
-   private:
-    DictionaryHashStoreContents* outer_;
-
-    DISALLOW_COPY_AND_ASSIGN(SimpleMutableDictionary);
-  };
-
-  const std::string hash_store_id_;
-  std::string super_mac_;
-  scoped_ptr<int> version_;
-  scoped_ptr<base::DictionaryValue> dictionary_;
-
-  DISALLOW_COPY_AND_ASSIGN(DictionaryHashStoreContents);
-};
-
-// An in-memory PrefStore backed by an immutable DictionaryValue.
-class DictionaryPrefStore : public PrefStore {
- public:
-  explicit DictionaryPrefStore(const base::DictionaryValue* dictionary)
-      : dictionary_(dictionary) {}
-
-  virtual bool GetValue(const std::string& key,
-                        const base::Value** result) const OVERRIDE {
-    const base::Value* tmp = NULL;
-    if (!dictionary_->Get(key, &tmp))
-      return false;
-
-    if (result)
-      *result = tmp;
-    return true;
-  }
+  MigrationDelegateImpl(const base::WeakPtr<JsonPrefStore>& pref_store,
+                        InterceptablePrefFilter* pref_filter,
+                        PrefHashStore* pref_hash_store)
+      : pref_store_(pref_store),
+        pref_filter_(pref_filter),
+        pref_hash_store_(pref_hash_store) {}
+
+  // TrackedPreferencesMigrationDelegate implementation
+  virtual bool IsValid() OVERRIDE;
+  virtual void CleanPreference(const std::string& key) OVERRIDE;
+  virtual void NotifyOnSuccessfulWrite(
+      const base::Closure& on_successful_write) OVERRIDE;
+  virtual void InterceptLoadedPreferences(const Intercept& intercept) OVERRIDE;
+  virtual PrefHashStore* GetPrefHashStore() OVERRIDE;
 
  private:
-  virtual ~DictionaryPrefStore() {}
+  base::WeakPtr<JsonPrefStore> pref_store_;
+  InterceptablePrefFilter* pref_filter_;
+  PrefHashStore* pref_hash_store_;
 
-  const base::DictionaryValue* dictionary_;
-
-  DISALLOW_COPY_AND_ASSIGN(DictionaryPrefStore);
+  DISALLOW_COPY_AND_ASSIGN(MigrationDelegateImpl);
 };
 
-// Waits for a PrefStore to be initialized and then initializes the
-// corresponding PrefHashStore.
-// The observer deletes itself when its work is completed.
-class InitializeHashStoreObserver : public PrefStore::Observer {
- public:
-  // Creates an observer that will initialize |pref_hash_store| with the
-  // contents of |pref_store| when the latter is fully loaded.
-  InitializeHashStoreObserver(
-      const std::vector<PrefHashFilter::TrackedPreferenceMetadata>&
-          tracking_configuration,
-      size_t reporting_ids_count,
-      const scoped_refptr<PrefStore>& pref_store,
-      scoped_ptr<PrefHashStoreImpl> pref_hash_store_impl)
-      : tracking_configuration_(tracking_configuration),
-        reporting_ids_count_(reporting_ids_count),
-        pref_store_(pref_store),
-        pref_hash_store_impl_(pref_hash_store_impl.Pass()) {}
-
-  virtual ~InitializeHashStoreObserver();
-
-  // PrefStore::Observer implementation.
-  virtual void OnPrefValueChanged(const std::string& key) OVERRIDE;
-  virtual void OnInitializationCompleted(bool succeeded) OVERRIDE;
-
- private:
-  const std::vector<PrefHashFilter::TrackedPreferenceMetadata>
-      tracking_configuration_;
-  const size_t reporting_ids_count_;
-  scoped_refptr<PrefStore> pref_store_;
-  scoped_ptr<PrefHashStoreImpl> pref_hash_store_impl_;
-
-  DISALLOW_COPY_AND_ASSIGN(InitializeHashStoreObserver);
-};
+bool MigrationDelegateImpl::IsValid() {
+  return pref_store_;
+}
+void MigrationDelegateImpl::CleanPreference(const std::string& key) {
+  if (pref_store_)
+    pref_store_->RemoveValueSilently(key);
+}
 
-InitializeHashStoreObserver::~InitializeHashStoreObserver() {}
+void MigrationDelegateImpl::NotifyOnSuccessfulWrite(
+    const base::Closure& on_successful_write) {
+  if (pref_store_)
+    pref_store_->RegisterOnNextSuccessfulWriteCallback(on_successful_write);
+}
 
-void InitializeHashStoreObserver::OnPrefValueChanged(const std::string& key) {}
+void MigrationDelegateImpl::InterceptLoadedPreferences(
+    const Intercept& intercept) {
+  // |pref_filter_| is owned by |pref_store_|.
+  if (pref_store_)
+    pref_filter_->InterceptNextFilterOnLoad(intercept);
+}
 
-void InitializeHashStoreObserver::OnInitializationCompleted(bool succeeded) {
-  // If we successfully loaded the preferences _and_ the PrefHashStoreImpl
-  // hasn't been initialized by someone else in the meantime, initialize it now.
-  const PrefHashStoreImpl::StoreVersion pre_update_version =
-      pref_hash_store_impl_->GetCurrentVersion();
-  if (succeeded && pre_update_version < PrefHashStoreImpl::VERSION_LATEST) {
-    PrefHashFilter(pref_hash_store_impl_.PassAs<PrefHashStore>(),
-                   tracking_configuration_,
-                   NULL,
-                   reporting_ids_count_).Initialize(*pref_store_);
-    UMA_HISTOGRAM_ENUMERATION(
-        "Settings.TrackedPreferencesAlternateStoreVersionUpdatedFrom",
-        pre_update_version,
-        PrefHashStoreImpl::VERSION_LATEST + 1);
-  }
-  pref_store_->RemoveObserver(this);
-  delete this;
+PrefHashStore* MigrationDelegateImpl::GetPrefHashStore() {
+  // |pref_hash_store_| is owned indirectly by |pref_store_|.
+  return pref_store_ ? pref_hash_store_ : NULL;
 }
 
 }  // namespace
@@ -231,6 +113,7 @@ void ProfilePrefStoreManager::RegisterPrefs(PrefRegistrySimple* registry) {
 void ProfilePrefStoreManager::RegisterProfilePrefs(
     user_prefs::PrefRegistrySyncable* registry) {
   PrefHashFilter::RegisterProfilePrefs(registry);
+  PrefStoreHashStoreContents::RegisterProfilePrefs(registry);
 }
 
 // static
@@ -254,11 +137,6 @@ void ProfilePrefStoreManager::ClearResetTime(PrefService* pref_service) {
   PrefHashFilter::ClearResetTime(pref_service);
 }
 
-void ProfilePrefStoreManager::ResetPrefHashStore() {
-  if (kPlatformSupportsPreferenceTracking)
-    GetPrefHashStoreImpl()->Reset();
-}
-
 PersistentPrefStore* ProfilePrefStoreManager::CreateProfilePrefStore(
     const scoped_refptr<base::SequencedTaskRunner>& io_task_runner,
     TrackedPreferenceValidationDelegate* validation_delegate) {
@@ -287,14 +165,22 @@ PersistentPrefStore* ProfilePrefStoreManager::CreateProfilePrefStore(
       unprotected_pref_names.insert(it->name);
     }
   }
+  scoped_ptr<PrefHashStoreImpl> unprotected_pref_hash_store =
+      GetPrefHashStoreImpl(false);
+  scoped_ptr<PrefHashStoreImpl> protected_pref_hash_store =
+      GetPrefHashStoreImpl(true);
+  PrefHashStoreImpl* raw_unprotected_pref_hash_store =
+      unprotected_pref_hash_store.get();
+  PrefHashStoreImpl* raw_protected_pref_hash_store =
+      protected_pref_hash_store.get();
 
   scoped_ptr<PrefHashFilter> unprotected_pref_hash_filter(
-      new PrefHashFilter(GetPrefHashStoreImpl().PassAs<PrefHashStore>(),
+      new PrefHashFilter(unprotected_pref_hash_store.PassAs<PrefHashStore>(),
                          unprotected_configuration,
                          validation_delegate,
                          reporting_ids_count_));
   scoped_ptr<PrefHashFilter> protected_pref_hash_filter(
-      new PrefHashFilter(GetPrefHashStoreImpl().PassAs<PrefHashStore>(),
+      new PrefHashFilter(protected_pref_hash_store.PassAs<PrefHashStore>(),
                          protected_configuration,
                          validation_delegate,
                          reporting_ids_count_));
@@ -313,50 +199,33 @@ PersistentPrefStore* ProfilePrefStoreManager::CreateProfilePrefStore(
       io_task_runner,
       protected_pref_hash_filter.PassAs<PrefFilter>()));
 
+  scoped_ptr<HashStoreContents> unprotected_hash_store_contents(
+      new PrefStoreHashStoreContents(unprotected_pref_store));
+  scoped_ptr<HashStoreContents> protected_hash_store_contents(
+      new PrefStoreHashStoreContents(protected_pref_store));
+
+  raw_unprotected_pref_hash_store->SetHashStoreContents(
+      unprotected_hash_store_contents.Pass());
+  raw_protected_pref_hash_store->SetHashStoreContents(
+      protected_hash_store_contents.Pass());
+
   SetupTrackedPreferencesMigration(
       unprotected_pref_names,
       protected_pref_names,
-      base::Bind(&JsonPrefStore::RemoveValueSilently,
-                 unprotected_pref_store->AsWeakPtr()),
-      base::Bind(&JsonPrefStore::RemoveValueSilently,
-                 protected_pref_store->AsWeakPtr()),
-      base::Bind(&JsonPrefStore::RegisterOnNextSuccessfulWriteCallback,
-                 unprotected_pref_store->AsWeakPtr()),
-      base::Bind(&JsonPrefStore::RegisterOnNextSuccessfulWriteCallback,
-                 protected_pref_store->AsWeakPtr()),
-      raw_unprotected_pref_hash_filter,
-      raw_protected_pref_hash_filter);
+      scoped_ptr<TrackedPreferencesMigrationDelegate>(
+          new MigrationDelegateImpl(unprotected_pref_store->AsWeakPtr(),
+                                    raw_unprotected_pref_hash_filter,
+                                    raw_unprotected_pref_hash_store)),
+      scoped_ptr<TrackedPreferencesMigrationDelegate>(
+          new MigrationDelegateImpl(protected_pref_store->AsWeakPtr(),
+                                    raw_protected_pref_hash_filter,
+                                    raw_protected_pref_hash_store)),
+      GetLegacyPrefHashStoreImpl().PassAs<PrefHashStore>());
 
   return new SegregatedPrefStore(unprotected_pref_store, protected_pref_store,
                                  protected_pref_names);
 }
 
-void ProfilePrefStoreManager::UpdateProfileHashStoreIfRequired(
-    const scoped_refptr<base::SequencedTaskRunner>& io_task_runner) {
-  if (!kPlatformSupportsPreferenceTracking)
-    return;
-  scoped_ptr<PrefHashStoreImpl> pref_hash_store_impl(GetPrefHashStoreImpl());
-  const PrefHashStoreImpl::StoreVersion current_version =
-      pref_hash_store_impl->GetCurrentVersion();
-  UMA_HISTOGRAM_ENUMERATION("Settings.TrackedPreferencesAlternateStoreVersion",
-                            current_version,
-                            PrefHashStoreImpl::VERSION_LATEST + 1);
-
-  // Update the pref hash store if it's not at the latest version.
-  if (current_version != PrefHashStoreImpl::VERSION_LATEST) {
-    scoped_refptr<JsonPrefStore> pref_store =
-        new JsonPrefStore(GetPrefFilePathFromProfilePath(profile_path_),
-                          io_task_runner,
-                          scoped_ptr<PrefFilter>());
-    pref_store->AddObserver(
-        new InitializeHashStoreObserver(tracking_configuration_,
-                                        reporting_ids_count_,
-                                        pref_store,
-                                        pref_hash_store_impl.Pass()));
-    pref_store->ReadPrefsAsync(NULL);
-  }
-}
-
 bool ProfilePrefStoreManager::InitializePrefsFromMasterPrefs(
     const base::DictionaryValue& master_prefs) {
   // Create the profile directory if it doesn't exist yet (very possible on
@@ -364,6 +233,21 @@ bool ProfilePrefStoreManager::InitializePrefsFromMasterPrefs(
   if (!base::CreateDirectory(profile_path_))
     return false;
 
+  scoped_ptr<base::DictionaryValue> protected_master_prefs;
+
+  if (kPlatformSupportsPreferenceTracking) {
+    protected_master_prefs.reset(master_prefs.DeepCopy());
+    scoped_refptr<WriteablePrefStore> pref_store(
+        new DictionaryPrefStore(protected_master_prefs.get()));
+    scoped_ptr<PrefHashStoreImpl> pref_hash_store(GetPrefHashStoreImpl(true));
+    pref_hash_store->SetHashStoreContents(scoped_ptr<HashStoreContents>(
+        new PrefStoreHashStoreContents(pref_store)));
+    PrefHashFilter(pref_hash_store.PassAs<PrefHashStore>(),
+                   tracking_configuration_,
+                   NULL,
+                   reporting_ids_count_).Initialize(*pref_store);
+  }
+
   // This will write out to a single combined file which will be immediately
   // migrated to two files on load.
   JSONFileValueSerializer serializer(
@@ -374,16 +258,8 @@ bool ProfilePrefStoreManager::InitializePrefsFromMasterPrefs(
   // complete before Chrome can start (as master preferences seed the Local
   // State and Preferences files). This won't trip ThreadIORestrictions as they
   // won't have kicked in yet on the main thread.
-  bool success = serializer.Serialize(master_prefs);
-
-  if (success && kPlatformSupportsPreferenceTracking) {
-    scoped_refptr<const PrefStore> pref_store(
-        new DictionaryPrefStore(&master_prefs));
-    PrefHashFilter(GetPrefHashStoreImpl().PassAs<PrefHashStore>(),
-                   tracking_configuration_,
-                   NULL,
-                   reporting_ids_count_).Initialize(*pref_store);
-  }
+  bool success = serializer.Serialize(
+      protected_master_prefs ? *protected_master_prefs : master_prefs);
 
   UMA_HISTOGRAM_BOOLEAN("Settings.InitializedFromMasterPrefs", success);
   return success;
@@ -395,7 +271,7 @@ ProfilePrefStoreManager::CreateDeprecatedCombinedProfilePrefStore(
   scoped_ptr<PrefFilter> pref_filter;
   if (kPlatformSupportsPreferenceTracking) {
     pref_filter.reset(
-        new PrefHashFilter(GetPrefHashStoreImpl().PassAs<PrefHashStore>(),
+        new PrefHashFilter(GetLegacyPrefHashStoreImpl().PassAs<PrefHashStore>(),
                            tracking_configuration_,
                            NULL,
                            reporting_ids_count_));
@@ -405,12 +281,22 @@ ProfilePrefStoreManager::CreateDeprecatedCombinedProfilePrefStore(
                            pref_filter.Pass());
 }
 
-scoped_ptr<PrefHashStoreImpl> ProfilePrefStoreManager::GetPrefHashStoreImpl() {
+scoped_ptr<PrefHashStoreImpl> ProfilePrefStoreManager::GetPrefHashStoreImpl(
+    bool use_super_mac) {
+  DCHECK(kPlatformSupportsPreferenceTracking);
+
+  return make_scoped_ptr(
+      new PrefHashStoreImpl(seed_, device_id_, use_super_mac));
+}
+
+scoped_ptr<PrefHashStoreImpl>
+ProfilePrefStoreManager::GetLegacyPrefHashStoreImpl() {
   DCHECK(kPlatformSupportsPreferenceTracking);
 
-  return make_scoped_ptr(new PrefHashStoreImpl(
-      seed_,
-      device_id_,
+  scoped_ptr<PrefHashStoreImpl> instance(
+      new PrefHashStoreImpl(seed_, device_id_, false));
+  instance->SetHashStoreContents(
       scoped_ptr<HashStoreContents>(new PrefServiceHashStoreContents(
-          profile_path_.AsUTF8Unsafe(), local_state_))));
+          profile_path_.AsUTF8Unsafe(), local_state_)));
+  return instance.Pass();
 }