headless: Add support for minidump generation on Linux

headless/lib/headless_content_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_content_main_delegate.h"
 
+#include "base/base_switches.h"
 #include "base/command_line.h"
 #include "base/files/file_util.h"
+#include "base/lazy_instance.h"
 #include "base/path_service.h"
 #include "base/run_loop.h"
 #include "base/strings/string_number_conversions.h"
 #include "base/trace_event/trace_event.h"
+#include "components/crash/content/app/breakpad_linux.h"
 #include "content/public/browser/browser_main_runner.h"
 #include "content/public/common/content_switches.h"
 #include "headless/lib/browser/headless_browser_impl.h"
 #include "headless/lib/browser/headless_content_browser_client.h"
+#include "headless/lib/headless_crash_reporter_client.h"
+#include "headless/lib/headless_macros.h"
 #include "ui/base/resource/resource_bundle.h"
 #include "ui/base/ui_base_switches.h"
 #include "ui/gfx/switches.h"
 #include "ui/gl/gl_switches.h"
 #include "ui/ozone/public/ozone_switches.h"
 
 #ifdef HEADLESS_USE_EMBEDDED_RESOURCES
 #include "headless/embedded_resource_pak.h"
 #endif
 
 namespace headless {
 namespace {
 // Keep in sync with content/common/content_constants_internal.h.
 // TODO(skyostil): Add a tracing test for this.
 const int kTraceEventBrowserProcessSortIndex = -6;
 
 HeadlessContentMainDelegate* g_current_headless_content_main_delegate = nullptr;
+
+base::LazyInstance<HeadlessCrashReporterClient>::Leaky g_headless_crash_client =
+    LAZY_INSTANCE_INITIALIZER;
 }  // namespace
 
 HeadlessContentMainDelegate::HeadlessContentMainDelegate(
     std::unique_ptr<HeadlessBrowserImpl> browser)
     : content_client_(browser->options()), browser_(std::move(browser)) {
   DCHECK(!g_current_headless_content_main_delegate);
   g_current_headless_content_main_delegate = this;
 }
 
 HeadlessContentMainDelegate::~HeadlessContentMainDelegate() {
@@ skipping 80 matching lines @@
 
   settings.logging_dest = log_mode;
   settings.log_file = log_path.value().c_str();
   settings.lock_log = logging::DONT_LOCK_LOG_FILE;
   settings.delete_old = process_type.empty() ? logging::DELETE_OLD_LOG_FILE
                                              : logging::APPEND_TO_OLD_LOG_FILE;
   bool success = logging::InitLogging(settings);
   DCHECK(success);
 }
 
+void HeadlessContentMainDelegate::InitCrashReporter(
+    const base::CommandLine& command_line) {
+  const std::string process_type =
+      command_line.GetSwitchValueASCII(switches::kProcessType);
+  crash_reporter::SetCrashReporterClient(g_headless_crash_client.Pointer());
+  g_headless_crash_client.Pointer()->set_crash_dumps_dir(
+      browser_->options()->crash_dumps_dir);
+
+  if (!browser_->options()->enable_crash_reporter) {
+    DCHECK(!breakpad::IsCrashReporterEnabled());
+    return;
+  }
+#if defined(HEADLESS_USE_BREAKPAD)
+  if (process_type != switches::kZygoteProcess)
+    breakpad::InitCrashReporter(process_type);
+#endif  // defined(HEADLESS_USE_BREAKPAD)
+}
+
 void HeadlessContentMainDelegate::PreSandboxStartup() {
-  InitLogging(*base::CommandLine::ForCurrentProcess());
+  const base::CommandLine& command_line(
+      *base::CommandLine::ForCurrentProcess());
+  const std::string process_type =
+      command_line.GetSwitchValueASCII(switches::kProcessType);
+  InitLogging(command_line);
+  InitCrashReporter(command_line);
   InitializeResourceBundle();
 }
 
 int HeadlessContentMainDelegate::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");
@@ skipping 10 matching lines @@
   browser_->RunOnStartCallback();
   browser_runner->Run();
   browser_.reset();
   browser_runner->Shutdown();
 
   // Return value >=0 here to disable calling content::BrowserMain.
   return 0;
 }
 
 void HeadlessContentMainDelegate::ZygoteForked() {
-  // TODO(skyostil): Disable the zygote host.
+  const base::CommandLine& command_line(
+      *base::CommandLine::ForCurrentProcess());
+  const std::string process_type =
+      command_line.GetSwitchValueASCII(switches::kProcessType);
+  // Unconditionally try to turn on crash reporting since we do not have access
+  // to the latest browser options at this point when testing. Breakpad will
+  // bail out gracefully if the browser process hasn't enabled crash reporting.
+  breakpad::InitCrashReporter(process_type);
 }
 
 // static
 HeadlessContentMainDelegate* HeadlessContentMainDelegate::GetInstance() {
   return g_current_headless_content_main_delegate;
 }
 
 // static
 void HeadlessContentMainDelegate::InitializeResourceBundle() {
   base::FilePath dir_module;
@@ skipping 24 matching lines @@
 #endif
 }
 
 content::ContentBrowserClient*
 HeadlessContentMainDelegate::CreateContentBrowserClient() {
   browser_client_.reset(new HeadlessContentBrowserClient(browser_.get()));
   return browser_client_.get();
 }
 
 }  // namespace headless