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)),

[mmenke, 2014/10/30 19:28:01]

I don't think this value makes any sense here - the class doesn't even use it
for anything.  Does the browser test have issues if it's non-zero?

[meacer, 2014/11/06 21:21:55]

Yes, for example SSLCertErrorLogin will fail if it's non zero.
I agree it doesn't look like it belongs here, but then the only alternative I
can think of is to introduce global state somewhere in SSLErrorHandler which
doesn't sound better. I'm happy to consider any other suggestions.

[mmenke, 2014/11/06 22:12:40]

I want to play around with this locally to better understand the tests a bit
better (I wrote them ages ago), and see if I can come up with another option. 
May not have time until tomorrow.

       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() {
 }
 
+base::TimeDelta CaptivePortalTabHelper::GetSSLErrorDelay() const {
+  return ssl_error_delay_;
+}
+
+void CaptivePortalTabHelper::SetSSLErrorDelayForTest(
+    base::TimeDelta ssl_error_delay) {
+  ssl_error_delay_ = ssl_error_delay;
+}
+
 void CaptivePortalTabHelper::RenderViewDeleted(
     content::RenderViewHost* render_view_host) {
   // This can happen when a cross-process navigation is aborted, either by
   // pressing stop or by starting a new cross-process navigation that can't
   // re-use |provisional_render_view_host_|.  May also happen on a crash.
   if (render_view_host == provisional_render_view_host_)
     OnLoadAborted();
 }
 
 void CaptivePortalTabHelper::DidStartProvisionalLoadForFrame(
@@ skipping 156 matching lines @@
 
 void CaptivePortalTabHelper::SetTabReloaderForTest(
     CaptivePortalTabReloader* tab_reloader) {
   tab_reloader_.reset(tab_reloader);
 }
 
 CaptivePortalTabReloader* CaptivePortalTabHelper::GetTabReloaderForTest() {
   return tab_reloader_.get();
 }
 
-void CaptivePortalTabHelper::OpenLoginTab() {
-  Browser* browser = chrome::FindBrowserWithWebContents(web_contents_);
+// static
+void CaptivePortalTabHelper::OpenLoginTabForWebContents(

[mmenke, 2014/10/30 19:28:01]

Definition order should match declaration order.

[meacer, 2014/11/06 21:21:55]

Done.

+    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* web_contents =
+    content::WebContents* contents =
         browser->tab_strip_model()->GetWebContentsAt(i);
     CaptivePortalTabHelper* captive_portal_tab_helper =
-        CaptivePortalTabHelper::FromWebContents(web_contents);
-    if (captive_portal_tab_helper->IsLoginTab())
+        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* web_contents = chrome::AddSelectedTabWithURL(
-          browser,
-          CaptivePortalServiceFactory::GetForProfile(profile_)->test_url(),
-          ui::PAGE_TRANSITION_TYPED);
+  // 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(web_contents);
+      CaptivePortalTabHelper::FromWebContents(new_contents);
   captive_portal_tab_helper->SetIsLoginTab();
 }