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 12const wchar_t kChromePipeName[] = L"\\\\.\\pipe\\ChromeCrashServices"; 13const wchar_t kGoogleUpdatePipeName[] = L"\\\\.\\pipe\\GoogleCrashServices\\"; 14const wchar_t kSystemPrincipalSid[] = L"S-1-5-18"; 15 16const 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 21extern "C" IMAGE_DOS_HEADER __ImageBase; 22 23// Builds a string representation of the user's SID and places it in user_sid. 24bool 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 54bool 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 73google_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 86google_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} 122