src/chrome_frame/crash_server_init.cc

   1 // Copyright (c) 2011 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "chrome_frame/crash_server_init.h"
   6
   7 #include <sddl.h>
   8 #include <Shlobj.h>
   9 #include <stdlib.h>
  10 #include "version.h"  // NOLINT
  11
  12 const wchar_t kChromePipeName[] = L"\\\\.\\pipe\\ChromeCrashServices";
  13 const wchar_t kGoogleUpdatePipeName[] = L"\\\\.\\pipe\\GoogleCrashServices\\";
  14 const wchar_t kSystemPrincipalSid[] = L"S-1-5-18";
  15
  16 const MINIDUMP_TYPE kLargerDumpType = static_cast<MINIDUMP_TYPE>(
  17     MiniDumpWithProcessThreadData |  // Get PEB and TEB.
  18     MiniDumpWithUnloadedModules |  // Get unloaded modules when available.
  19     MiniDumpWithIndirectlyReferencedMemory);  // Get memory referenced by stack.
  20
  21 extern "C" IMAGE_DOS_HEADER __ImageBase;
  22
  23 // Builds a string representation of the user's SID and places it in user_sid.
  24 bool GetUserSidString(std::wstring* user_sid) {
  25   bool success = false;
  26   if (user_sid) {
  27     struct {
  28       TOKEN_USER token_user;
  29       BYTE buffer[SECURITY_MAX_SID_SIZE];
  30     } token_info_buffer;
  31
  32     HANDLE token = NULL;
  33     if (OpenProcessToken(GetCurrentProcess(), TOKEN_QUERY, &token)) {
  34       DWORD out_size;
  35       if (GetTokenInformation(token, TokenUser, &token_info_buffer.token_user,
  36                               sizeof(token_info_buffer), &out_size)) {
  37         wchar_t* user_sid_value = NULL;
  38         if (token_info_buffer.token_user.User.Sid &&
  39             ConvertSidToStringSid(token_info_buffer.token_user.User.Sid,
  40                                   &user_sid_value)) {
  41           *user_sid = user_sid_value;
  42           LocalFree(user_sid_value);
  43           user_sid_value = NULL;
  44           success = true;
  45         }
  46       }
  47       CloseHandle(token);
  48     }
  49   }
  50
  51   return success;
  52 }
  53
  54 bool IsRunningSystemInstall() {
  55   wchar_t exe_path[MAX_PATH * 2] = {0};
  56   GetModuleFileName(reinterpret_cast<HMODULE>(&__ImageBase),
  57                     exe_path,
  58                     _countof(exe_path));
  59
  60   bool is_system = false;
  61
  62   wchar_t program_files_path[MAX_PATH] = {0};
  63   if (SUCCEEDED(SHGetFolderPath(NULL, CSIDL_PROGRAM_FILES, NULL,
  64                                 SHGFP_TYPE_CURRENT, program_files_path))) {
  65     if (wcsstr(exe_path, program_files_path) == exe_path) {
  66       is_system = true;
  67     }
  68   }
  69
  70   return is_system;
  71 }
  72
  73 google_breakpad::CustomClientInfo* GetCustomInfo() {
  74   static google_breakpad::CustomInfoEntry ver_entry(
  75       L"ver", TEXT(CHROME_VERSION_STRING));
  76   static google_breakpad::CustomInfoEntry prod_entry(L"prod", L"ChromeFrame");
  77   static google_breakpad::CustomInfoEntry plat_entry(L"plat", L"Win32");
  78   static google_breakpad::CustomInfoEntry type_entry(L"ptype", L"chrome_frame");
  79   static google_breakpad::CustomInfoEntry entries[] = {
  80       ver_entry, prod_entry, plat_entry, type_entry };
  81   static google_breakpad::CustomClientInfo custom_info = {
  82       entries, ARRAYSIZE(entries) };
  83   return &custom_info;
  84 }
  85
  86 google_breakpad::ExceptionHandler* InitializeCrashReporting(
  87     CrashReportingMode mode) {
  88   wchar_t temp_path[MAX_PATH + 1] = {0};
  89   DWORD path_len = ::GetTempPath(MAX_PATH, temp_path);
  90
  91   std::wstring pipe_name;
  92   if (mode == HEADLESS) {
  93     // This flag is used for testing, connect to the test crash service.
  94     pipe_name = kChromePipeName;
  95   } else {
  96     // Otherwise, build a pipe name corresponding to either user or
  97     // system-level Omaha.
  98     pipe_name = kGoogleUpdatePipeName;
  99     if (IsRunningSystemInstall()) {
 100       pipe_name += kSystemPrincipalSid;
 101     } else {
 102       std::wstring user_sid;
 103       if (GetUserSidString(&user_sid)) {
 104         pipe_name += user_sid;
 105       } else {
 106         // We don't think we're a system install, but we couldn't get the
 107         // user SID. Try connecting to the system-level crash service as a
 108         // last ditch effort.
 109         pipe_name += kSystemPrincipalSid;
 110       }
 111     }
 112   }
 113
 114   google_breakpad::ExceptionHandler* breakpad =
 115       new google_breakpad::ExceptionHandler(
 116           temp_path, NULL, NULL, NULL,
 117           google_breakpad::ExceptionHandler::HANDLER_ALL, kLargerDumpType,
 118           pipe_name.c_str(), GetCustomInfo());
 119
 120   return breakpad;
 121 }