Add Wi-Fi SSID to captive portal interstitial.

chrome/browser/ssl/captive_portal_blocking_page.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/ssl/captive_portal_blocking_page.h"
 
 #include "base/i18n/rtl.h"
 #include "base/metrics/histogram.h"
 #include "base/prefs/pref_service.h"
+#include "base/strings/string_util.h"
 #include "base/strings/utf_string_conversions.h"
 #include "base/values.h"
 #include "chrome/browser/captive_portal/captive_portal_tab_helper.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/common/pref_names.h"
 #include "components/captive_portal/captive_portal_detector.h"
+#include "components/wifi/wifi_service.h"
 #include "content/public/browser/web_contents.h"
 #include "grit/generated_resources.h"
 #include "net/base/net_util.h"
 #include "net/base/network_change_notifier.h"
 #include "ui/base/l10n/l10n_util.h"
 
 #if !defined(ENABLE_CAPTIVE_PORTAL_DETECTION)
 #error This file must be built with ENABLE_CAPTIVE_PORTAL_DETECTION flag.
 #endif
 
@@ skipping 14 matching lines @@
 
 bool IsWifiConnection() {
   // |net::NetworkChangeNotifier::GetConnectionType| isn't accurate on Linux and
   // Windows. See https://crbug.com/160537 for details.
   // TODO(meacer): Add heuristics to get a more accurate connection type on
   //               these platforms.
   return net::NetworkChangeNotifier::GetConnectionType() ==
       net::NetworkChangeNotifier::CONNECTION_WIFI;
 }
 
+std::string GetWiFiName() {
+  std::string ssid;
+#if defined(OS_WIN) || defined(OS_MACOSX)
+  scoped_ptr<wifi::WiFiService> wifi_service(wifi::WiFiService::Create());
+  wifi_service->Initialize(NULL);
+  std::string error;
+  wifi_service->GetConnectedNetworkSSID(&ssid, &error);
+  if (!error.empty())
+    return "";
+#endif
+  // TODO(meacer): Handle non UTF8 SSIDs.
+  if (!base::IsStringUTF8(ssid))
+    return "";
+  return ssid;
+}
+
 const char kOpenLoginPageCommand[] = "openLoginPage";
 
 } // namespace
 
 // static
 const void* CaptivePortalBlockingPage::kTypeForTesting =
     &CaptivePortalBlockingPage::kTypeForTesting;
 
 CaptivePortalBlockingPage::CaptivePortalBlockingPage(
     content::WebContents* web_contents,
@@ skipping 42 matching lines @@
   load_time_data->SetString("heading",
       l10n_util::GetStringUTF16(
           is_wifi_connection_ ?
           IDS_CAPTIVE_PORTAL_HEADING_WIFI :
           IDS_CAPTIVE_PORTAL_HEADING_WIRED));
 
   if (login_url_.spec() == captive_portal::CaptivePortalDetector::kDefaultURL) {
     // Captive portal may intercept requests without HTTP redirects, in which
     // case the login url would be the same as the captive portal detection url.
     // Don't show the login url in that case.
-    load_time_data->SetString(
-        "primaryParagraph",
-        l10n_util::GetStringUTF16(
-            is_wifi_connection_ ?
-            IDS_CAPTIVE_PORTAL_PRIMARY_PARAGRAPH_NO_LOGIN_URL_WIFI :
-            IDS_CAPTIVE_PORTAL_PRIMARY_PARAGRAPH_NO_LOGIN_URL_WIRED));
+    if (is_wifi_connection_) {
+      if (wifi_ssid_.empty())
+        wifi_ssid_ = GetWiFiName();
+      if (wifi_ssid_.empty()) {
+        load_time_data->SetString(
+            "primaryParagraph",
+            l10n_util::GetStringUTF16(
+                IDS_CAPTIVE_PORTAL_PRIMARY_PARAGRAPH_NO_LOGIN_URL_WIFI));
+      } else {
+        load_time_data->SetString(
+            "primaryParagraph",
+            l10n_util::GetStringFUTF16(
+                IDS_CAPTIVE_PORTAL_PRIMARY_PARAGRAPH_NO_LOGIN_URL_WIFI_SSID,
+                net::EscapeForHTML(base::UTF8ToUTF16(wifi_ssid_))));
+      }
+    } else {
+      // Non-WiFi connection:
+      load_time_data->SetString(
+          "primaryParagraph",
+          l10n_util::GetStringUTF16(
+              IDS_CAPTIVE_PORTAL_PRIMARY_PARAGRAPH_NO_LOGIN_URL_WIRED));
+    }
   } else {
+    // Portal redirection was done with HTTP redirects, show the login URL.
     std::string languages;
     Profile* profile = Profile::FromBrowserContext(
         web_contents()->GetBrowserContext());
     if (profile)
       languages = profile->GetPrefs()->GetString(prefs::kAcceptLanguages);
 
     base::string16 login_host = net::IDNToUnicode(login_url_.host(), languages);
     if (base::i18n::IsRTL())
       base::i18n::WrapStringWithLTRFormatting(&login_host);
-    load_time_data->SetString(
-        "primaryParagraph",
-        l10n_util::GetStringFUTF16(
-            is_wifi_connection_ ?
-            IDS_CAPTIVE_PORTAL_PRIMARY_PARAGRAPH_WIFI :
-            IDS_CAPTIVE_PORTAL_PRIMARY_PARAGRAPH_WIRED,
-            login_host));
+
+    if (is_wifi_connection_) {
+      if (wifi_ssid_.empty())
+        wifi_ssid_ = GetWiFiName();
+      if (wifi_ssid_.empty()) {
+        load_time_data->SetString(
+            "primaryParagraph",
+            l10n_util::GetStringFUTF16(
+                IDS_CAPTIVE_PORTAL_PRIMARY_PARAGRAPH_WIFI,
+                login_host));
+      } else {
+        load_time_data->SetString(
+            "primaryParagraph",
+                l10n_util::GetStringFUTF16(
+                IDS_CAPTIVE_PORTAL_PRIMARY_PARAGRAPH_WIFI_SSID,
+                net::EscapeForHTML(base::UTF8ToUTF16(wifi_ssid_)),
+                login_host));
+      }
+    } else {
+      // Non-WiFi connection:
+      load_time_data->SetString(
+          "primaryParagraph",
+          l10n_util::GetStringFUTF16(IDS_CAPTIVE_PORTAL_PRIMARY_PARAGRAPH_WIRED,
+                                     login_host));
+    }
   }
 
   // Fill the empty strings to avoid getting debug warnings.
   load_time_data->SetString("openDetails", base::string16());
   load_time_data->SetString("closeDetails", base::string16());
   load_time_data->SetString("explanationParagraph", base::string16());
   load_time_data->SetString("finalParagraph", base::string16());
 }
 
 void CaptivePortalBlockingPage::CommandReceived(const std::string& command) {
   // The response has quotes around it.
   if (command == std::string("\"") + kOpenLoginPageCommand + "\"") {
     RecordUMA(OPEN_LOGIN_PAGE);
     CaptivePortalTabHelper::OpenLoginTabForWebContents(web_contents(), true);
   }
 }