| OLD | NEW |
| (Empty) |
| 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 } | |
| OLD | NEW |