Index: headless/lib/headless_main_delegate.cc |
diff --git a/headless/lib/headless_main_delegate.cc b/headless/lib/headless_main_delegate.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..bd0dabceca29a9aa8204052758f8e5f8b75f0e91 |
--- /dev/null |
+++ b/headless/lib/headless_main_delegate.cc |
@@ -0,0 +1,118 @@ |
+// 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 |