Stop autologin when the WebContents is destroyed.

chrome/browser/ui/auto_login_info_bar_delegate.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/ui/auto_login_info_bar_delegate.h"
 
 #include "base/bind.h"
 #include "base/logging.h"
 #include "base/message_loop.h"
 #include "base/metrics/histogram.h"
@@ skipping 59 matching lines @@
 
   HISTOGRAM_MAX
 };
 
 // AutoLoginRedirector --------------------------------------------------------
 
 // This class is created by the AutoLoginInfoBarDelegate when the user wishes to
 // auto-login.  It holds context information needed while re-issuing service
 // tokens using the TokenService, gets the browser cookies with the TokenAuth
 // API, and finally redirects the user to the correct page.
-class AutoLoginRedirector : public UbertokenConsumer {
+class AutoLoginRedirector : public UbertokenConsumer,
+                            public content::NotificationObserver {
  public:
   AutoLoginRedirector(NavigationController* navigation_controller,
                       const std::string& args);
   virtual ~AutoLoginRedirector();
 
  private:
   // Overriden from UbertokenConsumer:
   virtual void OnUbertokenSuccess(const std::string& token) OVERRIDE;
   virtual void OnUbertokenFailure(const GoogleServiceAuthError& error) OVERRIDE;
 
+  // Implementation of content::NotificationObserver
+  virtual void Observe(int type,
+                       const content::NotificationSource& source,
+                       const content::NotificationDetails& details) OVERRIDE;
+
   // Redirect tab to MergeSession URL, logging the user in and navigating
   // to the desired page.
   void RedirectToMergeSession(const std::string& token);
 
   NavigationController* navigation_controller_;
   const std::string args_;
   scoped_ptr<UbertokenFetcher> ubertoken_fetcher_;
 
+  // For listening to NavigationController destruction.
+  content::NotificationRegistrar registrar_;
+
   DISALLOW_COPY_AND_ASSIGN(AutoLoginRedirector);
 };
 
 AutoLoginRedirector::AutoLoginRedirector(
     NavigationController* navigation_controller,
     const std::string& args)
     : navigation_controller_(navigation_controller),
       args_(args) {
   ubertoken_fetcher_.reset(new UbertokenFetcher(
       Profile::FromBrowserContext(navigation_controller_->GetBrowserContext()),
       this));
+  registrar_.Add(this,
+                 content::NOTIFICATION_WEB_CONTENTS_DESTROYED,
+                 content::Source<content::WebContents>(
+                     navigation_controller_->GetWebContents()));
   ubertoken_fetcher_->StartFetchingToken();
 }
 
 AutoLoginRedirector::~AutoLoginRedirector() {
 }
 
+void AutoLoginRedirector::Observe(int type,
+                                  const NotificationSource& source,
+                                  const NotificationDetails& details) {
+  DCHECK(type == content::NOTIFICATION_WEB_CONTENTS_DESTROYED);
+  // The WebContents that started this has been destroyed. The request must be
+  // cancelled and this object must be deleted.
+  ubertoken_fetcher_.reset();
+  MessageLoop::current()->DeleteSoon(FROM_HERE, this);
+}
+
 void AutoLoginRedirector::OnUbertokenSuccess(const std::string& token) {
   RedirectToMergeSession(token);
   MessageLoop::current()->DeleteSoon(FROM_HERE, this);
 }
 
 void AutoLoginRedirector::OnUbertokenFailure(
     const GoogleServiceAuthError& error) {
   LOG(WARNING) << "AutoLoginRedirector: token request failed";
   MessageLoop::current()->DeleteSoon(FROM_HERE, this);
 }
@@ skipping 91 matching lines @@
 void AutoLoginInfoBarDelegate::RecordHistogramAction(int action) {
   UMA_HISTOGRAM_ENUMERATION("AutoLogin.Regular", action, HISTOGRAM_MAX);
 }
 
 void AutoLoginInfoBarDelegate::Observe(int type,
                                        const NotificationSource& source,
                                        const NotificationDetails& details) {
   DCHECK_EQ(chrome::NOTIFICATION_GOOGLE_SIGNED_OUT, type);
   owner()->RemoveInfoBar(this);
 }