headless: Initial headless embedder API implementation

headless/lib/browser/headless_devtools.cc

+// Copyright 2015 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 "headless/lib/browser/headless_devtools.h"
+
+#include "base/files/file_path.h"
+#include "components/devtools_http_handler/devtools_http_handler.h"
+#include "components/devtools_http_handler/devtools_http_handler_delegate.h"
+#include "content/public/browser/browser_context.h"
+#include "content/public/browser/devtools_frontend_host.h"
+#include "content/public/browser/navigation_entry.h"
+#include "headless/lib/browser/headless_browser_context.h"
+#include "net/base/net_errors.h"
+#include "net/socket/tcp_server_socket.h"
+#include "ui/base/resource/resource_bundle.h"
+
+using devtools_http_handler::DevToolsHttpHandler;
+
+namespace headless {
+
+namespace {
+
+const int kBackLog = 10;
+
+class TCPServerSocketFactory : public DevToolsHttpHandler::ServerSocketFactory {
+ public:
+  TCPServerSocketFactory(const std::string& address, uint16_t port)
+      : address_(address), port_(port) {}
+
+ private:
+  // DevToolsHttpHandler::ServerSocketFactory implementation:
+  scoped_ptr<net::ServerSocket> CreateForHttpServer() override {
+    scoped_ptr<net::ServerSocket> socket(
+        new net::TCPServerSocket(nullptr, net::NetLog::Source()));
+    if (socket->ListenWithAddressAndPort(address_, port_, kBackLog) != net::OK)
+      return scoped_ptr<net::ServerSocket>();
+
+    return socket;
+  }
+
+  std::string address_;
+  uint16_t port_;
+
+  DISALLOW_COPY_AND_ASSIGN(TCPServerSocketFactory);
+};
+
+class HeadlessDevToolsDelegate
+    : public devtools_http_handler::DevToolsHttpHandlerDelegate {
+ public:
+  HeadlessDevToolsDelegate();
+  ~HeadlessDevToolsDelegate() override;
+
+  // devtools_http_handler::DevToolsHttpHandlerDelegate implementation:
+  std::string GetDiscoveryPageHTML() override;
+  std::string GetFrontendResource(const std::string& path) override;
+  std::string GetPageThumbnailData(const GURL& url) override;
+  content::DevToolsExternalAgentProxyDelegate* HandleWebSocketConnection(
+      const std::string& path) override;
+
+ private:
+  DISALLOW_COPY_AND_ASSIGN(HeadlessDevToolsDelegate);
+};
+
+HeadlessDevToolsDelegate::HeadlessDevToolsDelegate() {}
+
+HeadlessDevToolsDelegate::~HeadlessDevToolsDelegate() {}
+
+std::string HeadlessDevToolsDelegate::GetDiscoveryPageHTML() {
+  return std::string();
+}
+
+std::string HeadlessDevToolsDelegate::GetFrontendResource(
+    const std::string& path) {
+  return content::DevToolsFrontendHost::GetFrontendResource(path).as_string();
+}
+
+std::string HeadlessDevToolsDelegate::GetPageThumbnailData(const GURL& url) {
+  return std::string();
+}
+
+content::DevToolsExternalAgentProxyDelegate*
+HeadlessDevToolsDelegate::HandleWebSocketConnection(const std::string& path) {
+  return nullptr;
+}
+
+}  // namespace
+
+scoped_ptr<DevToolsHttpHandler> CreateLocalDevToolsHttpHandler(
+    HeadlessBrowserContext* browser_context) {
+  int port = browser_context->options().devtools_http_port;
+  DCHECK_GE(port, 0);
+  scoped_ptr<DevToolsHttpHandler::ServerSocketFactory> socket_factory(
+      new TCPServerSocketFactory("127.0.0.1", port));

[Ryan Sleevi, 2016/02/22 21:11:13]

Is this correct? Are you sure it should always be 127.0.0.1? For example, this
only supports IPv4, not IPv6 - which is certainly not desirable for a number of
cases.

At the least, this should be up-lifted to a file-level comment and documented.

[Sami, 2016/02/23 20:19:08]

Agreed, I think it would be better for this to be embedder-controlled. I've
added an option for it and moved this constant to the headless shell.

+  return make_scoped_ptr(new DevToolsHttpHandler(
+      std::move(socket_factory), std::string(), new HeadlessDevToolsDelegate(),
+      browser_context->GetPath(), base::FilePath(), std::string(),
+      browser_context->options().user_agent));
+}
+
+}  // namespace headless