policy: Register OffTheRecordProfile at UserCloudPolicyManagerFactory.

chrome/browser/policy/cloud/user_cloud_policy_manager_factory.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/policy/cloud/user_cloud_policy_manager_factory.h"
 
 #include "base/logging.h"
 #include "base/message_loop/message_loop_proxy.h"
 #include "base/sequenced_task_runner.h"
 #include "chrome/browser/policy/cloud/cloud_external_data_manager.h"
@@ skipping 10 matching lines @@
 }
 
 // static
 UserCloudPolicyManager* UserCloudPolicyManagerFactory::GetForProfile(
     Profile* profile) {
   return GetInstance()->GetManagerForProfile(profile);
 }
 
 // static
 scoped_ptr<UserCloudPolicyManager>
-UserCloudPolicyManagerFactory::CreateForProfile(
+UserCloudPolicyManagerFactory::CreateForOriginalProfile(
     Profile* profile,
     bool force_immediate_load,
     scoped_refptr<base::SequencedTaskRunner> background_task_runner) {
-  return GetInstance()->CreateManagerForProfile(
+  return GetInstance()->CreateManagerForOriginalProfile(
       profile, force_immediate_load, background_task_runner);
 }
 
+// static
+UserCloudPolicyManager*
+UserCloudPolicyManagerFactory::RegisterForOffTheRecordProfile(
+    Profile* original_profile,
+    Profile* off_the_record_profile) {
+  return GetInstance()->RegisterManagerForOffTheRecordProfile(
+      original_profile, off_the_record_profile);
+}
+
+
 UserCloudPolicyManagerFactory::UserCloudPolicyManagerFactory()
     : BrowserContextKeyedBaseFactory(
         "UserCloudPolicyManager",
         BrowserContextDependencyManager::GetInstance()) {}
 
 UserCloudPolicyManagerFactory::~UserCloudPolicyManagerFactory() {}
 
 UserCloudPolicyManager* UserCloudPolicyManagerFactory::GetManagerForProfile(
     Profile* profile) {
-  // Get the manager for the original profile, since the PolicyService is
-  // also shared between the incognito Profile and the original Profile.
-  ManagerMap::const_iterator it = managers_.find(profile->GetOriginalProfile());
+  // In case |profile| is an incognito Profile, |manager_| will have a matching
+  // entry pointing to the PolicyService of the original Profile.
+  ManagerMap::const_iterator it = managers_.find(profile);
   return it != managers_.end() ? it->second : NULL;
 }
 
 scoped_ptr<UserCloudPolicyManager>
-    UserCloudPolicyManagerFactory::CreateManagerForProfile(
-        Profile* profile,
-        bool force_immediate_load,
-        scoped_refptr<base::SequencedTaskRunner> background_task_runner) {
+UserCloudPolicyManagerFactory::CreateManagerForOriginalProfile(
+    Profile* profile,
+    bool force_immediate_load,
+    scoped_refptr<base::SequencedTaskRunner> background_task_runner) {
   scoped_ptr<UserCloudPolicyStore> store(
       UserCloudPolicyStore::Create(profile, background_task_runner));
   if (force_immediate_load)
     store->LoadImmediately();
   scoped_ptr<UserCloudPolicyManager> manager(
       new UserCloudPolicyManager(profile,
                                  store.Pass(),
                                  scoped_ptr<CloudExternalDataManager>(),
                                  base::MessageLoopProxy::current()));
   manager->Init();
   return manager.Pass();
 }
 
+UserCloudPolicyManager*
+UserCloudPolicyManagerFactory::RegisterManagerForOffTheRecordProfile(
+    Profile* original_profile,
+    Profile* off_the_record_profile) {
+  // Register the PolicyService of the original Profile for the respective
+  // incognito Profile. See GetManagerForProfile above.
+  UserCloudPolicyManager* manager = GetManagerForProfile(original_profile);
+  Register(off_the_record_profile, manager);
+  return manager;
+}
+
 void UserCloudPolicyManagerFactory::BrowserContextShutdown(
     content::BrowserContext* context) {
   Profile* profile = static_cast<Profile*>(context);
   if (profile->IsOffTheRecord())
     return;
   UserCloudPolicyManager* manager = GetManagerForProfile(profile);
   if (manager)
     manager->Shutdown();
 }
 
@@ skipping 18 matching lines @@
   ManagerMap::iterator entry = managers_.find(profile);
   if (entry != managers_.end()) {
     DCHECK_EQ(instance, entry->second);
     managers_.erase(entry);
   } else {
     NOTREACHED();
   }
 }
 
 }  // namespace policy