headless: Make it possible to configure an HTTP proxy

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 "base/bind.h"
 #include "base/callback.h"
 #include "base/command_line.h"
 #include "base/location.h"
 #include "base/memory/ref_counted.h"
 #include "base/numerics/safe_conversions.h"
 #include "base/strings/string_number_conversions.h"
 #include "content/public/common/content_switches.h"
+#include "headless/app/headless_shell_switches.h"
 #include "headless/public/headless_browser.h"
 #include "headless/public/headless_web_contents.h"
 #include "net/base/ip_address.h"
 #include "ui/gfx/geometry/size.h"
 
 using headless::HeadlessBrowser;
 using headless::HeadlessWebContents;
 
 namespace {
 // Address where to listen to incoming DevTools connections.
@@ skipping 38 matching lines @@
       web_contents_ = nullptr;
       browser_->Shutdown();
     }
   }
 
   // HeadlessWebContents::Observer implementation:
   void DocumentOnLoadCompletedInMainFrame() override {
     ShutdownIfNeeded();
   }
 
+  void DidFinishNavigation(bool success) override {}
+
  private:
   HeadlessBrowser* browser_;  // Not owned.
   scoped_ptr<HeadlessWebContents> web_contents_;
 
   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)) {
     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 (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);
+  }
+
   return HeadlessBrowserMain(
       builder.Build(),
       base::Bind(&HeadlessShell::OnStart, base::Unretained(&shell)));
 }