headless: Add a headless shell

headless/lib/headless_main_delegate.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/headless_main_delegate.h"
+
+#include "base/command_line.h"
+#include "base/logging.h"
+#include "base/path_service.h"
+#include "base/trace_event/trace_event.h"
+#include "content/public/browser/browser_main_runner.h"
+#include "content/public/browser/content_browser_client.h"
+#include "content/public/common/content_client.h"
+#include "content/public/common/content_switches.h"
+#include "content/public/renderer/content_renderer_client.h"
+#include "content/public/utility/content_utility_client.h"
+#include "ui/base/resource/resource_bundle.h"
+
+namespace {
+
+void InitLogging() {
+  base::FilePath log_filename;
+  PathService::Get(base::DIR_EXE, &log_filename);
+  log_filename = log_filename.AppendASCII("headless_test_shell.log");
+  logging::LoggingSettings settings;
+  settings.logging_dest = logging::LOG_TO_ALL;
+  settings.log_file = log_filename.value().c_str();
+  settings.delete_old = logging::DELETE_OLD_LOG_FILE;
+  logging::InitLogging(settings);
+  logging::SetLogItems(true, true, true, true);
+}
+
+}  // namespace
+
+namespace headless {
+
+class HeadlessContentClient : public content::ContentClient {};
+
+class HeadlessContentBrowserClient : public content::ContentBrowserClient {};
+
+class HeadlessContentRendererClient : public content::ContentRendererClient {};
+
+class HeadlessContentUtilityClient : public content::ContentUtilityClient {};
+
+HeadlessMainDelegate::HeadlessMainDelegate() {}
+
+HeadlessMainDelegate::~HeadlessMainDelegate() {}
+
+bool HeadlessMainDelegate::BasicStartupComplete(int* exit_code) {
+  base::CommandLine& command_line = *base::CommandLine::ForCurrentProcess();
+
+  InitLogging();
+
+  command_line.AppendSwitch(switches::kNoSandbox);
+  command_line.AppendSwitch(switches::kSingleProcess);
+
+  // TODO(skyostil): Investigate using Mesa/SwiftShader for output.
+  command_line.AppendSwitch(switches::kDisableGpu);
+
+  content_client_.reset(new HeadlessContentClient);
+  SetContentClient(content_client_.get());
+
+  return false;
+}
+
+void HeadlessMainDelegate::PreSandboxStartup() {
+  InitializeResourceBundle();
+}
+
+int HeadlessMainDelegate::RunProcess(
+    const std::string& process_type,
+    const content::MainFunctionParams& main_function_params) {
+  if (!process_type.empty())
+    return -1;
+
+  base::trace_event::TraceLog::GetInstance()->SetProcessName("HeadlessBrowser");
+
+  scoped_ptr<content::BrowserMainRunner> browser_runner(
+      content::BrowserMainRunner::Create());
+  int exit_code = browser_runner->Initialize(main_function_params);
+  DCHECK_LT(exit_code, 0);
+  exit_code = browser_runner->Run();
+  browser_runner->Shutdown();
+  return exit_code;
+}
+
+void HeadlessMainDelegate::ZygoteForked() {
+  // TODO(skyostil): Disable the zygote host.
+}
+
+// static
+void HeadlessMainDelegate::InitializeResourceBundle() {
+  base::FilePath pak_file;
+  bool r = PathService::Get(base::DIR_MODULE, &pak_file);

[Peter Beverloo, 2015/10/27 14:35:51]

nit: const bool success?

+  DCHECK(r);
+  pak_file = pak_file.Append(FILE_PATH_LITERAL("content_shell.pak"));

[Peter Beverloo, 2015/10/27 14:35:51]

headless.pak

+  ui::ResourceBundle::InitSharedInstanceWithPakPath(pak_file);
+}
+
+content::ContentBrowserClient*
+HeadlessMainDelegate::CreateContentBrowserClient() {
+  browser_client_.reset(new HeadlessContentBrowserClient);
+  return browser_client_.get();
+}
+
+content::ContentRendererClient*
+HeadlessMainDelegate::CreateContentRendererClient() {
+  renderer_client_.reset(new HeadlessContentRendererClient);
+  return renderer_client_.get();
+}
+
+content::ContentUtilityClient*
+HeadlessMainDelegate::CreateContentUtilityClient() {
+  utility_client_.reset(new HeadlessContentUtilityClient);
+  return utility_client_.get();
+}
+
+}  // namespace headless