Added CaptivePortal Interstitial to chrome://interstitials

chrome/browser/ui/webui/interstitials/interstitial_ui.cc

 // Copyright 2014 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/webui/interstitials/interstitial_ui.h"
 
 #include "base/strings/string_number_conversions.h"
 #include "base/strings/string_util.h"
 #include "chrome/browser/browser_process.h"
+#include "chrome/browser/captive_portal/captive_portal_service.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/safe_browsing/safe_browsing_blocking_page.h"
 #include "chrome/browser/safe_browsing/safe_browsing_service.h"
+#include "chrome/browser/ssl/captive_portal_blocking_page.h"
 #include "chrome/browser/ssl/ssl_blocking_page.h"
 #include "chrome/common/url_constants.h"
 #include "chrome/grit/browser_resources.h"
 #include "content/public/browser/interstitial_page_delegate.h"
 #include "content/public/browser/web_contents.h"
 #include "content/public/browser/web_ui.h"
 #include "content/public/browser/web_ui_controller.h"
 #include "content/public/browser/web_ui_data_source.h"
 #include "net/base/net_errors.h"
 #include "net/base/url_util.h"
@@ skipping 16 matching lines @@
       const std::string& path,
       int render_process_id,
       int render_frame_id,
       const content::URLDataSource::GotDataCallback& callback) override;
 
  private:
   content::WebContents* web_contents_;
   DISALLOW_COPY_AND_ASSIGN(InterstitialHTMLSource);
 };
 
+class FakeConnectionInfoDelegate : public CaptivePortalBlockingPage::Delegate {
+ public:
+  FakeConnectionInfoDelegate(bool is_wifi_connection, std::string wifi_ssid)
+      : is_wifi_connection_(is_wifi_connection), wifi_ssid_(wifi_ssid) {}
+  ~FakeConnectionInfoDelegate() override {}
+
+  bool IsWifiConnection() const override { return is_wifi_connection_; }
+  std::string GetWiFiSSID() const override { return wifi_ssid_; }
+
+ private:
+  const bool is_wifi_connection_;

[meacer, 2015/06/19 21:23:18]

no need for const for bool.

[Bhanu Dev, 2015/06/19 22:35:51]

Done.

+  const std::string wifi_ssid_;

[meacer, 2015/06/19 21:23:18]

Use a reference here: |const std::string& wifi_ssid_|

+
+  DISALLOW_COPY_AND_ASSIGN(FakeConnectionInfoDelegate);
+};

[Bhanu Dev, 2015/06/19 21:09:03]

I am not sure if this is the correct place to place the definition of a class,
since this is only associated with the CreateCaptivePortalBlockingPage function
and not others.

[meacer, 2015/06/19 21:23:18]

Looks okay, it's obvious from the base class name where this will be used.

[Bhanu Dev, 2015/06/19 22:35:51]

Done.

+
 SSLBlockingPage* CreateSSLBlockingPage(content::WebContents* web_contents) {
   // Random parameters for SSL blocking page.
   int cert_error = net::ERR_CERT_CONTAINS_ERRORS;
   GURL request_url("https://example.com");
   bool overridable = false;
   bool strict_enforcement = false;
   base::Time time_triggered_ = base::Time::NowFromSystemTime();
   std::string url_param;
   if (net::GetValueForKeyInQuery(web_contents->GetURL(),
                                  "url",
@@ skipping 73 matching lines @@
   SafeBrowsingBlockingPage::UnsafeResource resource;
   resource.url = request_url;
   resource.threat_type =  threat_type;
   // Create a blocking page without showing the interstitial.
   return SafeBrowsingBlockingPage::CreateBlockingPage(
       g_browser_process->safe_browsing_service()->ui_manager().get(),
       web_contents,
       resource);
 }
 
+CaptivePortalBlockingPage* CreateCaptivePortalBlockingPage(
+    content::WebContents* web_contents) {
+  bool is_wifi_connection = 0;

[meacer, 2015/06/19 21:23:19]

false instead of 0.

[Bhanu Dev, 2015/06/19 22:35:51]

Done.

+  GURL landing_url("https://captive.portal/login");
+  GURL request_url("https://google.com");
+  std::string wifi_ssid;

[meacer, 2015/06/19 21:23:18]

If no "wifi_name" query param is passed this will be empty (while your html file
doesn't do that, it's possible someone manually enters a URL such as
chrome://interstitials/captiveportal?is_wifi=1). You should assign a default
value to this.

+
+  std::string request_url_param;
+  if (net::GetValueForKeyInQuery(web_contents->GetURL(), "url",
+                                 &request_url_param)) {
+    if (GURL(request_url_param).is_valid())
+      request_url = GURL(request_url_param);
+  }
+  std::string landing_url_param;
+  if (net::GetValueForKeyInQuery(web_contents->GetURL(), "landing_page",
+                                 &landing_url_param)) {
+    if (GURL(landing_url_param).is_valid())
+      landing_url = GURL(landing_url_param);
+  }
+  std::string wifi_connection_param;
+  if (net::GetValueForKeyInQuery(web_contents->GetURL(), "is_wifi",
+                                 &wifi_connection_param)) {
+    is_wifi_connection = wifi_connection_param == "1";
+  }
+  std::string wifi_ssid_param;
+  if (net::GetValueForKeyInQuery(web_contents->GetURL(), "wifi_name",
+                                 &wifi_ssid_param)) {
+    wifi_ssid = wifi_ssid_param;
+  }
+  FakeConnectionInfoDelegate* delegate =
+      new FakeConnectionInfoDelegate(is_wifi_connection, wifi_ssid);
+  net::SSLInfo ssl_info;
+  ssl_info.cert = new net::X509Certificate(
+      request_url.host(), "CA", base::Time::Max(), base::Time::Max());
+  CaptivePortalBlockingPage* blocking_page = new CaptivePortalBlockingPage(
+      web_contents, request_url, landing_url, nullptr, ssl_info,
+      base::Callback<void(bool)>());
+  blocking_page->SetDelegateForTesting(delegate);

[Bhanu Dev, 2015/06/19 21:09:03]

This is a testing function. I am not sure if can I use it here.

[meacer, 2015/06/19 21:23:19]

Oh, right :( Let's make it a non-test method and add a comment saying it should
only be used in tests and chrome://interstitials page. That will require you to
change a couple of test files.

+  return blocking_page;
+}
+
 } //  namespace
 
 InterstitialUI::InterstitialUI(content::WebUI* web_ui)
     : WebUIController(web_ui) {
   scoped_ptr<InterstitialHTMLSource> html_source(
       new InterstitialHTMLSource(web_ui->GetWebContents()));
   Profile* profile = Profile::FromWebUI(web_ui);
   content::URLDataSource::Add(profile, html_source.release());
 }
 
@@ skipping 27 matching lines @@
 void InterstitialHTMLSource::StartDataRequest(
     const std::string& path,
     int render_process_id,
     int render_frame_id,
     const content::URLDataSource::GotDataCallback& callback) {
   scoped_ptr<content::InterstitialPageDelegate> interstitial_delegate;
   if (StartsWithASCII(path, "ssl", true)) {
     interstitial_delegate.reset(CreateSSLBlockingPage(web_contents_));
   } else if (StartsWithASCII(path, "safebrowsing", true)) {
     interstitial_delegate.reset(CreateSafeBrowsingBlockingPage(web_contents_));
+  } else if (StartsWithASCII(path, "captiveportal", true)) {
+    interstitial_delegate.reset(CreateCaptivePortalBlockingPage(web_contents_));
   }
 
   std::string html;
   if (interstitial_delegate.get()) {
     html = interstitial_delegate.get()->GetHTMLContents();
   } else {
     html = ResourceBundle::GetSharedInstance()
                           .GetRawDataResource(IDR_SECURITY_INTERSTITIAL_UI_HTML)
                           .as_string();
   }
   scoped_refptr<base::RefCountedString> html_bytes = new base::RefCountedString;
   html_bytes->data().assign(html.begin(), html.end());
   callback.Run(html_bytes.get());
 }