Added ConfigurationPolicyProvider::RefreshPolicies.

chrome/browser/policy/asynchronous_policy_provider.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 "chrome/browser/policy/asynchronous_policy_provider.h"
 
 #include "base/bind.h"
 #include "chrome/browser/policy/asynchronous_policy_loader.h"
 #include "chrome/browser/policy/policy_map.h"
+#include "content/public/browser/browser_thread.h"
+
+using content::BrowserThread;
 
 namespace policy {
 
 AsynchronousPolicyProvider::AsynchronousPolicyProvider(
     const PolicyDefinitionList* policy_list,
     scoped_refptr<AsynchronousPolicyLoader> loader)
     : ConfigurationPolicyProvider(policy_list),
-      loader_(loader) {
+      loader_(loader),
+      ALLOW_THIS_IN_INITIALIZER_LIST(weak_ptr_factory_(this)) {
   loader_->Init(
-      base::Bind(&AsynchronousPolicyProvider::NotifyPolicyUpdated,
+      base::Bind(&AsynchronousPolicyProvider::OnUnexpectedReload,
                  base::Unretained(this)));
 }
 
 AsynchronousPolicyProvider::~AsynchronousPolicyProvider() {
   DCHECK(CalledOnValidThread());
   // |loader_| won't invoke its callback anymore after Stop(), therefore
   // Unretained(this) is safe in the ctor.
   loader_->Stop();
 }
 
-void AsynchronousPolicyProvider::ForceReload() {
-  loader_->Reload(true);
-}
-
 bool AsynchronousPolicyProvider::ProvideInternal(PolicyMap* map) {
   DCHECK(CalledOnValidThread());
-  DCHECK(loader_->policy());
+  if (!loader_->policy())
+    return false;
   map->LoadFrom(loader_->policy(), policy_definition_list());
   return true;
 }
 
+void AsynchronousPolicyProvider::RefreshPolicies() {
+  DCHECK(CalledOnValidThread());
+  // Invalidate any pending callbacks issued from here.
+  weak_ptr_factory_.InvalidateWeakPtrs();
+  // Post a new callback with a fresh WeakPtr to Reload.
+  BrowserThread::PostTask(
+      BrowserThread::FILE, FROM_HERE,
+      base::Bind(&AsynchronousPolicyLoader::Reload,
+                 loader_,
+                 base::Bind(&AsynchronousPolicyProvider::OnRefreshReload,
+                            weak_ptr_factory_.GetWeakPtr()),
+                 true));
+}
+
+void AsynchronousPolicyProvider::OnRefreshReload() {
+  DCHECK(CalledOnValidThread());
+  NotifyPolicyUpdated();
+}
+
+void AsynchronousPolicyProvider::OnUnexpectedReload() {
+  DCHECK(CalledOnValidThread());
+  // Don't notify now if we're waiting for a Reload requested from
+  // RefreshPolicies.
+  if (weak_ptr_factory_.HasWeakPtrs())
+    return;
+  NotifyPolicyUpdated();
+}
+
 }  // namespace policy