Add custom interstitial for captive portals.

chrome/browser/captive_portal/captive_portal_tab_helper.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/captive_portal/captive_portal_tab_helper.h"
 
 #include "base/bind.h"
 #include "chrome/browser/captive_portal/captive_portal_login_detector.h"
 #include "chrome/browser/captive_portal/captive_portal_service_factory.h"
 #include "chrome/browser/captive_portal/captive_portal_tab_reloader.h"
@@ skipping 13 matching lines @@
 #include "content/public/browser/resource_request_details.h"
 #include "content/public/browser/web_contents.h"
 #include "net/base/net_errors.h"
 #include "net/ssl/ssl_info.h"
 
 using captive_portal::CaptivePortalResult;
 using content::ResourceType;
 
 DEFINE_WEB_CONTENTS_USER_DATA_KEY(CaptivePortalTabHelper);
 
+// The delay before displaying the SSL interstitial for cert errors.
+// - If a "captive portal detected" result arrives in this many seconds,
+//   a captive portal interstitial is displayed.
+// - Otherwise, an SSL interstitial is displayed.
+const int kDefaultSSLInterstitialDisplayDelay = 2;
+
 CaptivePortalTabHelper::CaptivePortalTabHelper(
     content::WebContents* web_contents)
     : content::WebContentsObserver(web_contents),
       // web_contents is NULL in unit tests.
       profile_(web_contents ? Profile::FromBrowserContext(
                                   web_contents->GetBrowserContext())
                             : NULL),
       tab_reloader_(
           new CaptivePortalTabReloader(
               profile_,
               web_contents,
-              base::Bind(&CaptivePortalTabHelper::OpenLoginTab,
-                         base::Unretained(this)))),
+              base::Bind(&CaptivePortalTabHelper::OpenLoginTabForWebContents,
+                         web_contents, false))),
       login_detector_(new CaptivePortalLoginDetector(profile_)),
       web_contents_(web_contents),
       pending_error_code_(net::OK),
+      ssl_error_delay_(
+          base::TimeDelta::FromSeconds(kDefaultSSLInterstitialDisplayDelay)),
       provisional_render_view_host_(NULL) {
   registrar_.Add(this,
                  chrome::NOTIFICATION_CAPTIVE_PORTAL_CHECK_RESULT,
                  content::Source<Profile>(profile_));
   registrar_.Add(this,
                  content::NOTIFICATION_RESOURCE_RECEIVED_REDIRECT,
                  content::Source<content::WebContents>(web_contents));
 }
 
 CaptivePortalTabHelper::~CaptivePortalTabHelper() {
@@ skipping 126 matching lines @@
 }
 
 void CaptivePortalTabHelper::OnSSLCertError(const net::SSLInfo& ssl_info) {
   tab_reloader_->OnSSLCertError(ssl_info);
 }
 
 bool CaptivePortalTabHelper::IsLoginTab() const {
   return login_detector_->is_login_tab();
 }
 
+base::TimeDelta CaptivePortalTabHelper::GetSSLErrorDelay() const {
+  return ssl_error_delay_;
+}
+
+// static
+void CaptivePortalTabHelper::OpenLoginTabForWebContents(
+    content::WebContents* web_contents,
+    bool focus) {
+  Browser* browser = chrome::FindBrowserWithWebContents(web_contents);
+
+  // If the Profile doesn't have a tabbed browser window open, do nothing.
+  if (!browser)
+    return;
+
+  // Check if the Profile's topmost browser window already has a login tab.
+  // If so, do nothing.
+  // TODO(mmenke):  Consider focusing that tab, at least if this is the tab
+  //                helper for the currently active tab for the profile.
+  for (int i = 0; i < browser->tab_strip_model()->count(); ++i) {
+    content::WebContents* contents =
+        browser->tab_strip_model()->GetWebContentsAt(i);
+    CaptivePortalTabHelper* captive_portal_tab_helper =
+        CaptivePortalTabHelper::FromWebContents(contents);
+    if (captive_portal_tab_helper->IsLoginTab()) {
+      if (focus)
+        browser->tab_strip_model()->ActivateTabAt(i, false);
+      return;
+    }
+  }
+
+  // Otherwise, open a login tab.  Only end up here when a captive portal result
+  // was received, so it's safe to assume profile has a CaptivePortalService.
+  content::WebContents* new_contents = chrome::AddSelectedTabWithURL(
+      browser,
+      CaptivePortalServiceFactory::GetForProfile(
+          browser->profile())->test_url(),
+      ui::PAGE_TRANSITION_TYPED);
+  CaptivePortalTabHelper* captive_portal_tab_helper =
+      CaptivePortalTabHelper::FromWebContents(new_contents);
+  captive_portal_tab_helper->SetIsLoginTab();
+}
+
 void CaptivePortalTabHelper::OnRedirect(int child_id,
                                         ResourceType resource_type,
                                         const GURL& new_url) {
   // Only main frame redirects for the provisional RenderViewHost matter.
   if (resource_type != content::RESOURCE_TYPE_MAIN_FRAME ||
       !provisional_render_view_host_ ||
       provisional_render_view_host_->GetProcess()->GetID() != child_id) {
     return;
   }
 
@@ skipping 18 matching lines @@
 
 void CaptivePortalTabHelper::SetIsLoginTab() {
   login_detector_->SetIsLoginTab();
 }
 
 void CaptivePortalTabHelper::SetTabReloaderForTest(
     CaptivePortalTabReloader* tab_reloader) {
   tab_reloader_.reset(tab_reloader);
 }
 
+void CaptivePortalTabHelper::SetSSLErrorDelayForTest(
+    base::TimeDelta ssl_error_delay) {
+  ssl_error_delay_ = ssl_error_delay;
+}
+
 CaptivePortalTabReloader* CaptivePortalTabHelper::GetTabReloaderForTest() {
   return tab_reloader_.get();
 }
-
-void CaptivePortalTabHelper::OpenLoginTab() {
-  Browser* browser = chrome::FindBrowserWithWebContents(web_contents_);
-
-  // If the Profile doesn't have a tabbed browser window open, do nothing.
-  if (!browser)
-    return;
-
-  // Check if the Profile's topmost browser window already has a login tab.
-  // If so, do nothing.
-  // TODO(mmenke):  Consider focusing that tab, at least if this is the tab
-  //                helper for the currently active tab for the profile.
-  for (int i = 0; i < browser->tab_strip_model()->count(); ++i) {
-    content::WebContents* web_contents =
-        browser->tab_strip_model()->GetWebContentsAt(i);
-    CaptivePortalTabHelper* captive_portal_tab_helper =
-        CaptivePortalTabHelper::FromWebContents(web_contents);
-    if (captive_portal_tab_helper->IsLoginTab())
-      return;
-  }
-
-  // Otherwise, open a login tab.  Only end up here when a captive portal result
-  // was received, so it's safe to assume |profile_| has a CaptivePortalService.
-  content::WebContents* web_contents = chrome::AddSelectedTabWithURL(
-          browser,
-          CaptivePortalServiceFactory::GetForProfile(profile_)->test_url(),
-          ui::PAGE_TRANSITION_TYPED);
-  CaptivePortalTabHelper* captive_portal_tab_helper =
-      CaptivePortalTabHelper::FromWebContents(web_contents);
-  captive_portal_tab_helper->SetIsLoginTab();
-}