[Sync] Cache encrypted types info in ProfileSyncService

chrome/browser/sync/util/cryptographer.cc

 // Copyright (c) 2011 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 "base/base64.h"
 #include "chrome/browser/sync/util/cryptographer.h"
 #include "chrome/browser/password_manager/encryptor.h"
 
 namespace browser_sync {
 
 const char kNigoriTag[] = "google_chrome_nigori";
 
 // We name a particular Nigori instance (ie. a triplet consisting of a hostname,
 // a username, and a password) by calling Permute on this string. Since the
 // output of Permute is always the same for a given triplet, clients will always
 // assign the same name to a particular triplet.
 const char kNigoriKeyName[] = "nigori-key";
 
+Cryptographer::Observer::~Observer() {}
+
 Cryptographer::Cryptographer()
     : default_nigori_(NULL),
       encrypt_everything_(false) {
   syncable::ModelTypeSet sensitive_types = SensitiveTypes();
   encrypted_types_.insert(sensitive_types.begin(), sensitive_types.end());
 }
 
 Cryptographer::~Cryptographer() {}
 
+void Cryptographer::AddObserver(Observer* observer) {
+  observers_.AddObserver(observer);
+}
+
+void Cryptographer::RemoveObserver(Observer* observer) {
+  observers_.RemoveObserver(observer);
+}
+
 void Cryptographer::Bootstrap(const std::string& restored_bootstrap_token) {
   if (is_initialized()) {
     NOTREACHED();
     return;
   }
 
   scoped_ptr<Nigori> nigori(UnpackBootstrapToken(restored_bootstrap_token));
   if (nigori.get())
     AddKeyImpl(nigori.release());
 }
@@ skipping 229 matching lines @@
   syncable::ModelTypeSet types;
   // Both of these have their own encryption schemes, but we include them
   // anyways.
   types.insert(syncable::PASSWORDS);
   types.insert(syncable::NIGORI);
   return types;
 }
 
 void Cryptographer::UpdateEncryptedTypesFromNigori(
     const sync_pb::NigoriSpecifics& nigori) {
-  encrypted_types_.clear();
   if (nigori.encrypt_everything()) {
     set_encrypt_everything();
     return;
   }
+
+  syncable::ModelTypeSet encrypted_types(SensitiveTypes());
   if (nigori.encrypt_bookmarks())
-    encrypted_types_.insert(syncable::BOOKMARKS);
+    encrypted_types.insert(syncable::BOOKMARKS);
   if (nigori.encrypt_preferences())
-    encrypted_types_.insert(syncable::PREFERENCES);
+    encrypted_types.insert(syncable::PREFERENCES);
   if (nigori.encrypt_autofill_profile())
-    encrypted_types_.insert(syncable::AUTOFILL_PROFILE);
+    encrypted_types.insert(syncable::AUTOFILL_PROFILE);
   if (nigori.encrypt_autofill())
-    encrypted_types_.insert(syncable::AUTOFILL);
+    encrypted_types.insert(syncable::AUTOFILL);
   if (nigori.encrypt_themes())
-    encrypted_types_.insert(syncable::THEMES);
+    encrypted_types.insert(syncable::THEMES);
   if (nigori.encrypt_typed_urls())
-    encrypted_types_.insert(syncable::TYPED_URLS);
+    encrypted_types.insert(syncable::TYPED_URLS);
   if (nigori.encrypt_extension_settings())
-    encrypted_types_.insert(syncable::EXTENSION_SETTINGS);
+    encrypted_types.insert(syncable::EXTENSION_SETTINGS);
   if (nigori.encrypt_extensions())
-    encrypted_types_.insert(syncable::EXTENSIONS);
+    encrypted_types.insert(syncable::EXTENSIONS);
   if (nigori.encrypt_search_engines())
-    encrypted_types_.insert(syncable::SEARCH_ENGINES);
+    encrypted_types.insert(syncable::SEARCH_ENGINES);
   if (nigori.encrypt_sessions())
-    encrypted_types_.insert(syncable::SESSIONS);
+    encrypted_types.insert(syncable::SESSIONS);
   if (nigori.encrypt_apps())
-    encrypted_types_.insert(syncable::APPS);
+    encrypted_types.insert(syncable::APPS);
   if (nigori.encrypt_app_notifications())
-    encrypted_types_.insert(syncable::APP_NOTIFICATIONS);
+    encrypted_types.insert(syncable::APP_NOTIFICATIONS);
 
   // Note: the initial version with encryption did not support the
   // encrypt_everything field. If anything more than the sensitive types were
   // encrypted, it meant we were encrypting everything.
-  syncable::ModelTypeSet sensitive_types = SensitiveTypes();
-  encrypted_types_.insert(sensitive_types.begin(), sensitive_types.end());
   if (!nigori.has_encrypt_everything() &&
-      encrypted_types_.size() > sensitive_types.size()) {
+      encrypted_types.size() > SensitiveTypes().size()) {
     set_encrypt_everything();
+    return;
   }
+
+  SetEncryptedTypes(encrypted_types);
 }
 
 void Cryptographer::UpdateNigoriFromEncryptedTypes(
     sync_pb::NigoriSpecifics* nigori) const {
   nigori->set_encrypt_everything(encrypt_everything_);
   nigori->set_encrypt_bookmarks(
       encrypted_types_.count(syncable::BOOKMARKS) > 0);
   nigori->set_encrypt_preferences(
       encrypted_types_.count(syncable::PREFERENCES) > 0);
   nigori->set_encrypt_autofill_profile(
       encrypted_types_.count(syncable::AUTOFILL_PROFILE) > 0);
   nigori->set_encrypt_autofill(encrypted_types_.count(syncable::AUTOFILL) > 0);
   nigori->set_encrypt_themes(encrypted_types_.count(syncable::THEMES) > 0);
   nigori->set_encrypt_typed_urls(
       encrypted_types_.count(syncable::TYPED_URLS) > 0);
   nigori->set_encrypt_extension_settings(
       encrypted_types_.count(syncable::EXTENSION_SETTINGS) > 0);
   nigori->set_encrypt_extensions(
       encrypted_types_.count(syncable::EXTENSIONS) > 0);
   nigori->set_encrypt_search_engines(
       encrypted_types_.count(syncable::SEARCH_ENGINES) > 0);
   nigori->set_encrypt_sessions(encrypted_types_.count(syncable::SESSIONS) > 0);
   nigori->set_encrypt_apps(encrypted_types_.count(syncable::APPS) > 0);
   nigori->set_encrypt_app_notifications(
       encrypted_types_.count(syncable::APP_NOTIFICATIONS) > 0);
 }
 
 void Cryptographer::set_encrypt_everything() {
+  if (encrypt_everything_) {
+    DCHECK(encrypted_types_ == syncable::GetAllRealModelTypes());
+    return;
+  }
   encrypt_everything_ = true;
+  // Change |encrypted_types_| directly to avoid sending more than one
+  // notification.
   encrypted_types_ = syncable::GetAllRealModelTypes();
+  EmitEncryptedTypesChangedNotification();
 }
 
 bool Cryptographer::encrypt_everything() const {
   return encrypt_everything_;
 }
 
-void Cryptographer::SetEncryptedTypes(syncable::ModelTypeSet new_types) {
-  encrypted_types_.insert(new_types.begin(), new_types.end());
+syncable::ModelTypeSet Cryptographer::GetEncryptedTypes() const {
+  return encrypted_types_;
 }
 
-syncable::ModelTypeSet Cryptographer::GetEncryptedTypes() const {
-  return encrypted_types_;
+void Cryptographer::SetEncryptedTypesForTest(
+    const syncable::ModelTypeSet& encrypted_types) {
+  SetEncryptedTypes(encrypted_types);
+}
+
+void Cryptographer::SetEncryptedTypes(
+    const syncable::ModelTypeSet& encrypted_types) {
+  if (encrypted_types_ == encrypted_types) {
+    return;
+  }
+  encrypted_types_ = encrypted_types;
+  EmitEncryptedTypesChangedNotification();
+}
+
+void Cryptographer::EmitEncryptedTypesChangedNotification() {
+  FOR_EACH_OBSERVER(
+      Observer, observers_,
+      OnEncryptedTypesChanged(encrypted_types_, encrypt_everything_));
 }
 
 void Cryptographer::InstallKeys(const std::string& default_key_name,
                                 const sync_pb::NigoriKeyBag& bag) {
   int key_size = bag.key_size();
   for (int i = 0; i < key_size; ++i) {
     const sync_pb::NigoriKey key = bag.key(i);
     // Only use this key if we don't already know about it.
     if (nigoris_.end() == nigoris_.find(key.name())) {
       scoped_ptr<Nigori> new_nigori(new Nigori);
       if (!new_nigori->InitByImport(key.user_key(),
                                     key.encryption_key(),
                                     key.mac_key())) {
         NOTREACHED();
         continue;
       }
       nigoris_[key.name()] = make_linked_ptr(new_nigori.release());
     }
   }
   DCHECK(nigoris_.end() != nigoris_.find(default_key_name));
   default_nigori_ = &*nigoris_.find(default_key_name);
 }
 
 }  // namespace browser_sync