Add support for reporting crashes in the Chrome installer. This basically inv...

chrome/installer/setup/setup_main.cc

 // Copyright (c) 2010 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 <windows.h>
 #include <msi.h>
 #include <shellapi.h>
 #include <shlobj.h>
 
 #include <string>
 
 #include "base/at_exit.h"
 #include "base/basictypes.h"
 #include "base/command_line.h"
 #include "base/file_util.h"
+#include "base/file_version_info.h"
 #include "base/path_service.h"
 #include "base/process_util.h"
 #include "base/scoped_handle_win.h"
 #include "base/string_number_conversions.h"
 #include "base/string_util.h"
 #include "base/utf_string_conversions.h"
 #include "base/values.h"
+#include "base/win_util.h"
 #include "base/win/windows_version.h"
+#include "breakpad/src/client/windows/handler/exception_handler.h"
 #include "chrome/common/chrome_switches.h"
 #include "chrome/installer/setup/install.h"
 #include "chrome/installer/setup/setup_constants.h"
 #include "chrome/installer/setup/setup_util.h"
 #include "chrome/installer/setup/uninstall.h"
 #include "chrome/installer/util/browser_distribution.h"
 #include "chrome/installer/util/delete_after_reboot_helper.h"
 #include "chrome/installer/util/delete_tree_work_item.h"
 #include "chrome/installer/util/google_update_settings.h"
 #include "chrome/installer/util/google_update_constants.h"
@@ skipping 11 matching lines @@
 
 #include "installer_util_strings.h"  // NOLINT
 
 using installer::Product;
 using installer::ProductPackageMapping;
 using installer::Products;
 using installer::Package;
 using installer::Packages;
 using installer::MasterPreferences;
 
+const wchar_t kChromePipeName[] = L"\\\\.\\pipe\\ChromeCrashServices";
+const wchar_t kGoogleUpdatePipeName[] = L"\\\\.\\pipe\\GoogleCrashServices\\";
+const wchar_t kSystemPrincipalSid[] = L"S-1-5-18";
+
+const MINIDUMP_TYPE kLargerDumpType = static_cast<MINIDUMP_TYPE>(
+    MiniDumpWithProcessThreadData |  // Get PEB and TEB.
+    MiniDumpWithUnloadedModules |  // Get unloaded modules when available.
+    MiniDumpWithIndirectlyReferencedMemory);  // Get memory referenced by stack.
+
 namespace {
 
 // This method unpacks and uncompresses the given archive file. For Chrome
 // install we are creating a uncompressed archive that contains all the files
 // needed for the installer. This uncompressed archive is later compressed.
 //
 // This method first uncompresses archive specified by parameter "archive"
 // and assumes that it will result in an uncompressed full archive file
 // (chrome.7z) or uncompressed archive patch file (chrome_patch.diff). If it
 // is patch file, it is applied to the old archive file that should be
@@ skipping 630 matching lines @@
     VLOG(1) << "Install distribution: Chrome";
     installations->AddDistribution(BrowserDistribution::CHROME_BROWSER, prefs);
   }
 
   if (prefs.install_chrome_frame()) {
     VLOG(1) << "Install distribution: Chrome Frame";
     installations->AddDistribution(BrowserDistribution::CHROME_FRAME, prefs);
   }
 }
 
+// Returns the Custom information for the client identified by the exe path
+// passed in. This information is used for crash reporting.
+google_breakpad::CustomClientInfo* GetCustomInfo(const wchar_t* exe_path) {
+  std::wstring product;
+  std::wstring version;
+  scoped_ptr<FileVersionInfo>
+      version_info(FileVersionInfo::CreateFileVersionInfo(FilePath(exe_path)));
+  if (version_info.get()) {
+    version = version_info->product_version();
+    product = version_info->product_short_name();
+  }
+
+  if (version.empty())
+    version = L"0.1.0.0";
+
+  if (product.empty())
+    product = L"Chrome installer";

[robertshield, 2010/12/22 19:02:00]

microscopic nit: "Chrome Installer" here and on line 73

[ananta, 2010/12/22 19:13:01]

Done.

+
+  static google_breakpad::CustomInfoEntry ver_entry(L"ver", version.c_str());
+  static google_breakpad::CustomInfoEntry prod_entry(L"prod", product.c_str());
+  static google_breakpad::CustomInfoEntry plat_entry(L"plat", L"Win32");
+  static google_breakpad::CustomInfoEntry type_entry(L"ptype",
+                                                     L"Chrome installer");
+  static google_breakpad::CustomInfoEntry entries[] = {
+      ver_entry, prod_entry, plat_entry, type_entry };
+  static google_breakpad::CustomClientInfo custom_info = {
+      entries, arraysize(entries) };
+  return &custom_info;
+}
+
+// Initialize crash reporting for this process. This involves connecting to
+// breakpad, etc.
+google_breakpad::ExceptionHandler* InitializeCrashReporting(
+    bool system_install) {
+  // Only report crashes if the user allows it.
+  if (!GoogleUpdateSettings::GetCollectStatsConsent())
+    return NULL;
+
+  // Get the alternate dump directory. We use the temp path.
+  FilePath temp_directory;
+  if (!file_util::GetTempDir(&temp_directory) || temp_directory.empty())
+    return NULL;
+
+  wchar_t exe_path[MAX_PATH * 2] = {0};
+  GetModuleFileName(NULL, exe_path, arraysize(exe_path));
+
+  // Build the pipe name. It can be either:
+  // System-wide install: "NamedPipe\GoogleCrashServices\S-1-5-18"
+  // Per-user install: "NamedPipe\GoogleCrashServices\<user SID>"
+  std::wstring user_sid = kSystemPrincipalSid;
+
+  if (system_install) {
+    user_sid = kSystemPrincipalSid;

[robertshield, 2010/12/22 19:02:00]

Don't need this line since the user_sid is initialized to the system sid above.

[ananta, 2010/12/22 19:13:01]

Done.

+  } else {
+    if (!win_util::GetUserSidString(&user_sid)) {
+      return NULL;
+    }
+  }
+
+  std::wstring pipe_name = kGoogleUpdatePipeName;
+  pipe_name += user_sid;
+
+  google_breakpad::ExceptionHandler* breakpad =
+      new google_breakpad::ExceptionHandler(
+          temp_directory.ToWStringHack(), NULL, NULL, NULL,
+          google_breakpad::ExceptionHandler::HANDLER_ALL, kLargerDumpType,
+          pipe_name.c_str(), GetCustomInfo(exe_path));
+  return breakpad;
+}
+
 }  // namespace
 
 int WINAPI wWinMain(HINSTANCE instance, HINSTANCE prev_instance,
                     wchar_t* command_line, int show_command) {
   // The exit manager is in charge of calling the dtors of singletons.
   base::AtExitManager exit_manager;
   CommandLine::Init(0, NULL);
 
   const MasterPreferences& prefs =
       installer::MasterPreferences::ForCurrentProcess();
   installer::InitInstallerLogging(prefs);
 
   const CommandLine& cmd_line = *CommandLine::ForCurrentProcess();
   VLOG(1) << "Command Line: " << cmd_line.command_line_string();
 
   VLOG(1) << "multi install is " << prefs.is_multi_install();
   bool system_install = false;
   prefs.GetBool(installer::master_preferences::kSystemLevel, &system_install);
   VLOG(1) << "system install is " << system_install;
 
+  google_breakpad::scoped_ptr<google_breakpad::ExceptionHandler> breakpad(
+      InitializeCrashReporting(system_install));
+
   ProductPackageMapping installations(prefs.is_multi_install(), system_install);
   PopulateInstallations(prefs, &installations);
 
   // Check to make sure current system is WinXP or later. If not, log
   // error message and get out.
   if (!InstallUtil::IsOSSupported()) {
     LOG(ERROR) << "Chrome only supports Windows XP or later.";
     installations.packages()[0]->WriteInstallerResult(
         installer::OS_NOT_SUPPORTED, IDS_INSTALL_OS_NOT_SUPPORTED_BASE, NULL);
     return installer::OS_NOT_SUPPORTED;
@@ skipping 96 matching lines @@
       // to pass through, since this is only returned on uninstall which is
       // never invoked directly by Google Update.
       return_code = InstallUtil::GetInstallReturnCode(install_status);
     }
   }
 
   VLOG(1) << "Installation complete, returning: " << return_code;
 
   return return_code;
 }