Track whether a given user session has completed initialization, and use

components/user_manager/known_user.cc

 // Copyright 2015 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 "components/user_manager/known_user.h"
 
 #include <stddef.h>
 
 #include <memory>
 #include <utility>
@@ skipping 35 matching lines @@
 
 // Key of GAPS cookie.
 const char kGAPSCookie[] = "gaps_cookie";
 
 // Key of the reason for re-auth.
 const char kReauthReasonKey[] = "reauth_reason";
 
 // Key for the GaiaId migration status.
 const char kGaiaIdMigration[] = "gaia_id_migration";
 
+// Key of the boolean flag telling if user session has finished init yet.
+const char kProfileEverInitialized[] = "profile_ever_initialized";
+
 PrefService* GetLocalState() {
   if (!UserManager::IsInitialized())
     return nullptr;
 
   return UserManager::Get()->GetLocalState();
 }
 
 // Checks if values in |dict| correspond with |account_id| identity.
 bool UserMatches(const AccountId& account_id,
                  const base::DictionaryValue& dict) {
@@ skipping 378 matching lines @@
   SetBooleanPref(account_id, kUsingSAMLKey, using_saml);
 }
 
 bool IsUsingSAML(const AccountId& account_id) {
   bool using_saml;
   if (GetBooleanPref(account_id, kUsingSAMLKey, &using_saml))
     return using_saml;
   return false;
 }
 
+bool WasProfileEverInitialized(const AccountId& account_id) {
+  bool profile_ever_initialized;
+  if (GetBooleanPref(account_id, kProfileEverInitialized,
+                     &profile_ever_initialized))
+    return profile_ever_initialized;
+
+  // Sessions created before we started setting the session_initialized flag
+  // should default to "initialized = true".
+  LOG(WARNING) << "Treating unmigrated user as profile_ever_initialized=true";
+  return true;
+}
+
+void SetProfileEverInitialized(const AccountId& account_id, bool initialized) {
+  SetBooleanPref(account_id, kProfileEverInitialized, initialized);
+}
+
 void UpdateReauthReason(const AccountId& account_id, const int reauth_reason) {
   SetIntegerPref(account_id, kReauthReasonKey, reauth_reason);
 }
 
 bool FindReauthReason(const AccountId& account_id, int* out_value) {
   return GetIntegerPref(account_id, kReauthReasonKey, out_value);
 }
 
 void RemovePrefs(const AccountId& account_id) {
   PrefService* local_state = GetLocalState();
 
   // Local State may not be initialized in tests.
   if (!local_state)
     return;
 
   ListPrefUpdate update(local_state, kKnownUsers);
   for (size_t i = 0; i < update->GetSize(); ++i) {
     base::DictionaryValue* element = nullptr;
     if (update->GetDictionary(i, &element)) {
       if (UserMatches(account_id, *element)) {
         update->Remove(i, nullptr);
         break;
       }
     }
   }
 }
 
+// Exported so tests can call this from other components.
+void RemovePrefsForTesting(const AccountId& account_id) {
+  RemovePrefs(account_id);
+}
+
 void RegisterPrefs(PrefRegistrySimple* registry) {
   registry->RegisterListPref(kKnownUsers);
 }
 
 }  // namespace known_user
 }  // namespace user_manager