Mac OS X-specific implementation of Networking Private API.

components/wifi/wifi_test.cc

 // Copyright 2013 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 <stdio.h>
 #include <iostream>
 #include <string>
 
 #include "base/at_exit.h"
 #include "base/bind.h"
@@ skipping 54 matching lines @@
 
 WiFiTest::Result WiFiTest::Main(int argc, const char* argv[]) {
   if (!ParseCommandLine(argc, argv)) {
     fprintf(stderr,
             "usage: %s [--list]"
             " [--get_properties]"
             " [--connect]"
             " [--disconnect]"
             " [--network_guid=<network_guid>]"
             " [--frequency=0|2400|5000]"
+            " [--password=<password>]"
             " [<network_guid>]\n",
             argv[0]);
     return RESULT_WRONG_USAGE;
   }
 
   base::MessageLoopForIO loop;
   result_ = RESULT_PENDING;
 
   return result_;
 }
 
 bool WiFiTest::ParseCommandLine(int argc, const char* argv[]) {
   CommandLine::Init(argc, argv);
   const CommandLine& parsed_command_line = *CommandLine::ForCurrentProcess();
   std::string network_guid =
       parsed_command_line.GetSwitchValueASCII("network_guid");
+  std::string password =
+      parsed_command_line.GetSwitchValueASCII("password");
 
   if (parsed_command_line.GetArgs().size() == 1) {
 #if defined(OS_WIN)
     network_guid = WideToASCII(parsed_command_line.GetArgs()[0]);
 #else
     network_guid = parsed_command_line.GetArgs()[0];
 #endif
   }
 
 #if defined(OS_WIN)
   if (parsed_command_line.HasSwitch("debug"))
     MessageBoxA(NULL, __FUNCTION__, "Debug Me!", MB_OK);
 #endif
 
-#if defined(OS_WIN)
+#if defined(OS_WIN) || defined(OS_MACOSX)
   scoped_ptr<WiFiService> wifi_service(WiFiService::Create());
 #else
   scoped_ptr<WiFiService> wifi_service(WiFiService::CreateForTest());
 #endif
 
   wifi_service->Initialize(NULL);
 
   if (parsed_command_line.HasSwitch("list")) {
     ListValue network_list;
     wifi_service->GetVisibleNetworks(&network_list);
     std::cout << network_list;
     return true;
   }
 
   if (parsed_command_line.HasSwitch("get_properties")) {
     if (network_guid.length() > 0) {
       DictionaryValue properties;
       std::string error;
       wifi_service->GetProperties(network_guid, &properties, &error);
       std::cout << error << ":\n" << properties;
       return true;
     }
   }
 
+  if (parsed_command_line.HasSwitch("password")) {
+    if (network_guid.length() > 0) {
+      std::string error;
+      scoped_ptr<DictionaryValue> properties(new DictionaryValue());
+      properties->SetString("WiFi.Passphrase", password);
+      wifi_service->SetProperties(network_guid, properties.Pass(), &error);
+      // fall through to connect.
+    }
+  }
+
   if (parsed_command_line.HasSwitch("connect")) {
     if (network_guid.length() > 0) {
       std::string error;
       wifi_service->StartConnect(network_guid, &error);
       std::cout << error;
       return true;
     }
   }
 
   if (parsed_command_line.HasSwitch("disconnect")) {
     if (network_guid.length() > 0) {
       std::string error;
       wifi_service->StartDisconnect(network_guid, &error);
       std::cout << error;
       return true;
     }
   }
 
   return false;
 }
 
 }  // namespace wifi
 
 int main(int argc, const char* argv[]) {
   wifi::WiFiTest wifi_test;
   return wifi_test.Main(argc, argv);
 }