headless: Make it possible to override the DevTools server address

headless/app/headless_shell.cc

 // Copyright 2016 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 <memory>
 
 #include "base/bind.h"
 #include "base/callback.h"
 #include "base/command_line.h"
 #include "base/location.h"
@@ skipping 104 matching lines @@
   DISALLOW_COPY_AND_ASSIGN(HeadlessShell);
 };
 
 int main(int argc, const char** argv) {
   HeadlessShell shell;
   HeadlessBrowser::Options::Builder builder(argc, argv);
 
   // Enable devtools if requested.
   base::CommandLine command_line(argc, argv);
   if (command_line.HasSwitch(switches::kRemoteDebuggingPort)) {
+    std::string address = kDevToolsHttpServerAddress;
+    if (command_line.HasSwitch(headless::switches::kRemoteDebuggingAddress)) {
+      address = command_line.GetSwitchValueASCII(
+          headless::switches::kRemoteDebuggingAddress);
+      net::IPAddress parsed_address;
+      if (!net::ParseURLHostnameToAddress(address, &parsed_address)) {
+        LOG(ERROR) << "Invalid devtools server address";
+        return EXIT_FAILURE;
+      }
+    }
     int parsed_port;
     std::string port_str =
         command_line.GetSwitchValueASCII(switches::kRemoteDebuggingPort);
-    if (base::StringToInt(port_str, &parsed_port) &&
-        base::IsValueInRangeForNumericType<uint16_t>(parsed_port)) {
-      net::IPAddress devtools_address;
-      bool result =
-          devtools_address.AssignFromIPLiteral(kDevToolsHttpServerAddress);
-      DCHECK(result);
-      builder.EnableDevToolsServer(net::IPEndPoint(
-          devtools_address, base::checked_cast<uint16_t>(parsed_port)));
+    if (!base::StringToInt(port_str, &parsed_port) ||
+        !base::IsValueInRangeForNumericType<uint16_t>(parsed_port)) {
+      LOG(ERROR) << "Invalid devtools server port";
+      return EXIT_FAILURE;
     }
+    net::IPAddress devtools_address;
+    bool result = devtools_address.AssignFromIPLiteral(address);
+    DCHECK(result);
+    builder.EnableDevToolsServer(net::IPEndPoint(
+        devtools_address, base::checked_cast<uint16_t>(parsed_port)));
   }
 
   if (command_line.HasSwitch(headless::switches::kProxyServer)) {
     std::string proxy_server =
         command_line.GetSwitchValueASCII(headless::switches::kProxyServer);
     net::HostPortPair parsed_proxy_server =
         net::HostPortPair::FromString(proxy_server);
     if (parsed_proxy_server.host().empty() || !parsed_proxy_server.port()) {
       LOG(ERROR) << "Malformed proxy server url";
       return EXIT_FAILURE;
     }
     builder.SetProxyServer(parsed_proxy_server);
   }
 
   if (command_line.HasSwitch(switches::kHostResolverRules)) {
     builder.SetHostResolverRules(
         command_line.GetSwitchValueASCII(switches::kHostResolverRules));
   }
 
   return HeadlessBrowserMain(
       builder.Build(),
       base::Bind(&HeadlessShell::OnStart, base::Unretained(&shell)));
 }