Convert utility process WiFi Credentials IPC to mojo

chrome/utility/extensions/extensions_handler.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/utility/extensions/extensions_handler.h"
 
 #include <utility>
 
 #include "base/command_line.h"
 #include "base/path_service.h"
@@ skipping 14 matching lines @@
 #include "media/base/media.h"
 #include "mojo/public/cpp/bindings/strong_binding.h"
 #include "services/service_manager/public/cpp/interface_registry.h"
 #include "ui/base/ui_base_switches.h"
 
 #if !defined(MEDIA_DISABLE_FFMPEG)
 #include "media/base/media_file_checker.h"
 #endif
 
 #if defined(OS_WIN)
+#include "chrome/common/extensions/wifi_credentials.mojom.h"
 #include "chrome/utility/media_galleries/itunes_pref_parser_win.h"
 #include "components/wifi/wifi_service.h"
 #endif  // defined(OS_WIN)
 
 #if defined(OS_WIN) || defined(OS_MACOSX)
 #include "chrome/utility/media_galleries/iapps_xml_utils.h"
 #include "chrome/utility/media_galleries/itunes_library_parser.h"
 #include "chrome/utility/media_galleries/picasa_album_table_reader.h"
 #include "chrome/utility/media_galleries/picasa_albums_indexer.h"
 #endif  // defined(OS_WIN) || defined(OS_MACOSX)
@@ skipping 43 matching lines @@
       const std::vector<metadata::AttachedImage>& attached_images) {
     callback.Run(true, metadata.ToValue(), attached_images);
     ReleaseProcessIfNeeded();
   }
 
   ChromeContentUtilityClient* const utility_client_;
 
   DISALLOW_COPY_AND_ASSIGN(MediaParserImpl);
 };
 
+#if defined(OS_WIN)
+class WiFiCredentialsImpl : public extensions::mojom::WiFiCredentials {
+ public:
+  static void Create(extensions::mojom::WiFiCredentialsRequest request) {

[Sam McNally, 2017/01/16 04:16:20]

Constructors and destructors before methods.

[Noel Gordon, 2017/01/16 11:09:58]

Done.

+    mojo::MakeStrongBinding(base::MakeUnique<WiFiCredentialsImpl>(),
+                            std::move(request));
+  }
+
+  WiFiCredentialsImpl() = default;
+  ~WiFiCredentialsImpl() override = default;
+
+ private:
+  // extensions::mojom::WiFiCredentials:
+  void GetWiFiCredentials(const std::string& ssid,
+                          const GetWiFiCredentialsCallback& callback) override {
+    if (ssid == "chrome://test-wifi-get-key-from-system") {
+      callback.Run(true, ssid);  // test-mode: echo the ssid back in key_data.
+      return;
+    }
+
+    std::unique_ptr<wifi::WiFiService> wifi_service(
+        wifi::WiFiService::Create());
+    wifi_service->Initialize(nullptr);
+
+    std::string key_data;
+    std::string error;
+    wifi_service->GetKeyFromSystem(ssid, &key_data, &error);
+
+    const bool success = error.empty();
+    if (!success)
+      key_data.clear();

[Sam McNally, 2017/01/16 04:16:20]

This wasn't here before.

[Noel Gordon, 2017/01/16 11:09:58]

But is now because security.  If this API fails, the content of key_data is now
always empty. 

+
+    callback.Run(success, key_data);
+  }
+
+  DISALLOW_COPY_AND_ASSIGN(WiFiCredentialsImpl);
+};
+#endif  // defined(OS_WIN)
+
 }  // namespace
 
 namespace extensions {
 
 ExtensionsHandler::ExtensionsHandler() {
   ExtensionsClient::Set(ChromeExtensionsClient::GetInstance());
 }
 
 ExtensionsHandler::~ExtensionsHandler() {
 }
 
 // static
 void ExtensionsHandler::PreSandboxStartup() {
   // Initialize media libraries for media file validation.
   media::InitializeMediaLibrary();
 }
 
 // static
 void ExtensionsHandler::ExposeInterfacesToBrowser(
     service_manager::InterfaceRegistry* registry,
     ChromeContentUtilityClient* utility_client,
     bool running_elevated) {
   // If our process runs with elevated privileges, only add elevated
   // Mojo services to the interface registry.
-  if (running_elevated)
+  if (running_elevated) {
+#if defined(OS_WIN)
+    registry->AddInterface(base::Bind(&WiFiCredentialsImpl::Create));
+#endif
     return;
+  }
 
   registry->AddInterface(base::Bind(&MediaParserImpl::Create, utility_client));
 }
 
 bool ExtensionsHandler::OnMessageReceived(const IPC::Message& message) {
   bool handled = true;
   IPC_BEGIN_MESSAGE_MAP(ExtensionsHandler, message)
     IPC_MESSAGE_HANDLER(ChromeUtilityMsg_CheckMediaFile, OnCheckMediaFile)
 
 #if defined(OS_WIN)
     IPC_MESSAGE_HANDLER(ChromeUtilityMsg_ParseITunesPrefXml,
                         OnParseITunesPrefXml)
 #endif  // defined(OS_WIN)
 
 #if defined(OS_WIN) || defined(OS_MACOSX)
     IPC_MESSAGE_HANDLER(ChromeUtilityMsg_ParseITunesLibraryXmlFile,
                         OnParseITunesLibraryXmlFile)
     IPC_MESSAGE_HANDLER(ChromeUtilityMsg_ParsePicasaPMPDatabase,
                         OnParsePicasaPMPDatabase)
     IPC_MESSAGE_HANDLER(ChromeUtilityMsg_IndexPicasaAlbumsContents,
                         OnIndexPicasaAlbumsContents)
 #endif  // defined(OS_WIN) || defined(OS_MACOSX)
 
-#if defined(OS_WIN)
-    IPC_MESSAGE_HANDLER(ChromeUtilityHostMsg_GetWiFiCredentials,
-                        OnGetWiFiCredentials)
-#endif  // defined(OS_WIN)
-
     IPC_MESSAGE_UNHANDLED(handled = false)
   IPC_END_MESSAGE_MAP()
 
   return handled || utility_handler_.OnMessageReceived(message);
 }
 
 void ExtensionsHandler::OnCheckMediaFile(
     int64_t milliseconds_of_decoding,
     const IPC::PlatformFileForTransit& media_file) {
 #if !defined(MEDIA_DISABLE_FFMPEG)
@@ skipping 58 matching lines @@
     const std::vector<picasa::FolderINIContents>& folders_inis) {
   picasa::PicasaAlbumsIndexer indexer(album_uids);
   indexer.ParseFolderINI(folders_inis);
 
   Send(new ChromeUtilityHostMsg_IndexPicasaAlbumsContents_Finished(
       indexer.albums_images()));
   ReleaseProcessIfNeeded();
 }
 #endif  // defined(OS_WIN) || defined(OS_MACOSX)
 
-#if defined(OS_WIN)
-void ExtensionsHandler::OnGetWiFiCredentials(const std::string& network_guid) {
-  std::unique_ptr<wifi::WiFiService> wifi_service(wifi::WiFiService::Create());
-  wifi_service->Initialize(NULL);
-
-  std::string key_data;
-  std::string error;
-  wifi_service->GetKeyFromSystem(network_guid, &key_data, &error);
-
-  Send(new ChromeUtilityHostMsg_GotWiFiCredentials(key_data, error.empty()));
-}
-#endif  // defined(OS_WIN)
-
 }  // namespace extensions