[email protected] | e3ce40a | 2012-01-31 03:03:03 | [diff] [blame] | 1 | // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
[email protected] | fc14cef | 2009-01-27 22:17:29 | [diff] [blame] | 2 | // Use of this source code is governed by a BSD-style license that can be |
| 3 | // found in the LICENSE file. |
| 4 | |
[email protected] | 7c47ae3e | 2009-02-18 00:34:21 | [diff] [blame] | 5 | #include "chrome/browser/process_singleton.h" |
[email protected] | fc14cef | 2009-01-27 22:17:29 | [diff] [blame] | 6 | |
[email protected] | b1d7844b | 2012-08-22 22:37:47 | [diff] [blame] | 7 | #include <shellapi.h> |
[email protected] | 6927f2e0 | 2012-09-18 00:13:49 | [diff] [blame] | 8 | #include <shobjidl.h> |
[email protected] | b1d7844b | 2012-08-22 22:37:47 | [diff] [blame] | 9 | |
[email protected] | fc14cef | 2009-01-27 22:17:29 | [diff] [blame] | 10 | #include "base/base_paths.h" |
| 11 | #include "base/command_line.h" |
[email protected] | f805fe8 | 2010-08-03 22:47:10 | [diff] [blame] | 12 | #include "base/file_path.h" |
[email protected] | 9e9b6e8e | 2009-12-02 08:45:01 | [diff] [blame] | 13 | #include "base/path_service.h" |
[email protected] | fc14cef | 2009-01-27 22:17:29 | [diff] [blame] | 14 | #include "base/process_util.h" |
[email protected] | f394bc6 | 2012-08-22 21:42:29 | [diff] [blame] | 15 | #include "base/stringprintf.h" |
[email protected] | 0f26d7b | 2011-01-05 19:10:44 | [diff] [blame] | 16 | #include "base/utf_string_conversions.h" |
[email protected] | f394bc6 | 2012-08-22 21:42:29 | [diff] [blame] | 17 | #include "base/win/metro.h" |
[email protected] | b3d791c9 | 2012-10-05 19:01:24 | [diff] [blame] | 18 | #include "base/win/registry.h" |
[email protected] | 6927f2e0 | 2012-09-18 00:13:49 | [diff] [blame] | 19 | #include "base/win/scoped_com_initializer.h" |
| 20 | #include "base/win/scoped_comptr.h" |
[email protected] | b90d7e80 | 2011-01-09 16:32:20 | [diff] [blame] | 21 | #include "base/win/scoped_handle.h" |
[email protected] | b3d791c9 | 2012-10-05 19:01:24 | [diff] [blame] | 22 | #include "base/win/win_util.h" |
[email protected] | 6927f2e0 | 2012-09-18 00:13:49 | [diff] [blame] | 23 | #include "base/win/windows_version.h" |
[email protected] | ecb924c | 2011-03-17 00:34:09 | [diff] [blame] | 24 | #include "base/win/wrapped_window_proc.h" |
[email protected] | e9613b5 | 2012-11-27 22:35:13 | [diff] [blame] | 25 | #include "chrome/browser/browser_process.h" |
[email protected] | 24590067 | 2012-10-13 02:27:11 | [diff] [blame] | 26 | #include "chrome/browser/shell_integration.h" |
[email protected] | b50892c5f | 2012-05-13 07:34:14 | [diff] [blame] | 27 | #include "chrome/browser/ui/simple_message_box.h" |
[email protected] | fc14cef | 2009-01-27 22:17:29 | [diff] [blame] | 28 | #include "chrome/common/chrome_constants.h" |
[email protected] | b3d791c9 | 2012-10-05 19:01:24 | [diff] [blame] | 29 | #include "chrome/common/chrome_paths.h" |
| 30 | #include "chrome/common/chrome_paths_internal.h" |
[email protected] | cfa118a | 2012-10-10 22:27:53 | [diff] [blame] | 31 | #include "chrome/common/chrome_switches.h" |
[email protected] | 6927f2e0 | 2012-09-18 00:13:49 | [diff] [blame] | 32 | #include "chrome/installer/util/browser_distribution.h" |
[email protected] | 78679969 | 2012-09-26 14:16:48 | [diff] [blame] | 33 | #include "chrome/installer/util/install_util.h" |
[email protected] | 6927f2e0 | 2012-09-18 00:13:49 | [diff] [blame] | 34 | #include "chrome/installer/util/shell_util.h" |
[email protected] | 0a19455 | 2011-09-14 17:53:35 | [diff] [blame] | 35 | #include "chrome/installer/util/wmi.h" |
[email protected] | b39ef1cb | 2011-10-25 04:46:55 | [diff] [blame] | 36 | #include "content/public/common/result_codes.h" |
[email protected] | 34ac8f3 | 2009-02-22 23:03:27 | [diff] [blame] | 37 | #include "grit/chromium_strings.h" |
| 38 | #include "grit/generated_resources.h" |
[email protected] | f394bc6 | 2012-08-22 21:42:29 | [diff] [blame] | 39 | #include "net/base/escape.h" |
[email protected] | c051a1b | 2011-01-21 23:30:17 | [diff] [blame] | 40 | #include "ui/base/l10n/l10n_util.h" |
[email protected] | e766106 | 2011-01-19 22:16:53 | [diff] [blame] | 41 | #include "ui/base/win/hwnd_util.h" |
[email protected] | fc14cef | 2009-01-27 22:17:29 | [diff] [blame] | 42 | |
| 43 | namespace { |
| 44 | |
[email protected] | 2b7ff5b9 | 2012-07-17 22:45:18 | [diff] [blame] | 45 | const char kLockfile[] = "lockfile"; |
| 46 | |
[email protected] | f394bc6 | 2012-08-22 21:42:29 | [diff] [blame] | 47 | const char kSearchUrl[] = |
| 48 | "http://www.google.com/search?q=%s&sourceid=chrome&ie=UTF-8"; |
| 49 | |
[email protected] | 14649ecd | 2012-12-05 01:00:24 | [diff] [blame] | 50 | const int kMetroChromeActivationTimeoutMs = 3000; |
| 51 | |
| 52 | // A helper class that acquires the given |mutex| while the AutoLockMutex is in |
| 53 | // scope. |
| 54 | class AutoLockMutex { |
| 55 | public: |
| 56 | explicit AutoLockMutex(HANDLE mutex) : mutex_(mutex) { |
| 57 | DWORD result = WaitForSingleObject(mutex_, INFINITE); |
| 58 | DPCHECK(result == WAIT_OBJECT_0) << "Result = " << result; |
| 59 | } |
| 60 | |
| 61 | ~AutoLockMutex() { |
| 62 | BOOL released = ReleaseMutex(mutex_); |
| 63 | DPCHECK(released); |
| 64 | } |
| 65 | |
| 66 | private: |
| 67 | HANDLE mutex_; |
| 68 | DISALLOW_COPY_AND_ASSIGN(AutoLockMutex); |
| 69 | }; |
| 70 | |
| 71 | // A helper class that releases the given |mutex| while the AutoUnlockMutex is |
| 72 | // in scope and immediately re-acquires it when going out of scope. |
| 73 | class AutoUnlockMutex { |
| 74 | public: |
| 75 | explicit AutoUnlockMutex(HANDLE mutex) : mutex_(mutex) { |
| 76 | BOOL released = ReleaseMutex(mutex_); |
| 77 | DPCHECK(released); |
| 78 | } |
| 79 | |
| 80 | ~AutoUnlockMutex() { |
| 81 | DWORD result = WaitForSingleObject(mutex_, INFINITE); |
| 82 | DPCHECK(result == WAIT_OBJECT_0) << "Result = " << result; |
| 83 | } |
| 84 | |
| 85 | private: |
| 86 | HANDLE mutex_; |
| 87 | DISALLOW_COPY_AND_ASSIGN(AutoUnlockMutex); |
| 88 | }; |
[email protected] | b3d791c9 | 2012-10-05 19:01:24 | [diff] [blame] | 89 | |
[email protected] | f891fb3 | 2009-04-08 00:20:32 | [diff] [blame] | 90 | // Checks the visibility of the enumerated window and signals once a visible |
[email protected] | fc14cef | 2009-01-27 22:17:29 | [diff] [blame] | 91 | // window has been found. |
| 92 | BOOL CALLBACK BrowserWindowEnumeration(HWND window, LPARAM param) { |
| 93 | bool* result = reinterpret_cast<bool*>(param); |
| 94 | *result = IsWindowVisible(window) != 0; |
| 95 | // Stops enumeration if a visible window has been found. |
| 96 | return !*result; |
| 97 | } |
| 98 | |
[email protected] | e3ce40a | 2012-01-31 03:03:03 | [diff] [blame] | 99 | // This function thunks to the object's version of the windowproc, taking in |
| 100 | // consideration that there are several messages being dispatched before |
| 101 | // WM_NCCREATE which we let windows handle. |
| 102 | LRESULT CALLBACK ThunkWndProc(HWND hwnd, UINT message, |
| 103 | WPARAM wparam, LPARAM lparam) { |
| 104 | ProcessSingleton* singleton = |
| 105 | reinterpret_cast<ProcessSingleton*>(ui::GetWindowUserData(hwnd)); |
| 106 | if (message == WM_NCCREATE) { |
| 107 | CREATESTRUCT* cs = reinterpret_cast<CREATESTRUCT*>(lparam); |
| 108 | singleton = reinterpret_cast<ProcessSingleton*>(cs->lpCreateParams); |
| 109 | CHECK(singleton); |
| 110 | ui::SetWindowUserData(hwnd, singleton); |
| 111 | } else if (!singleton) { |
| 112 | return ::DefWindowProc(hwnd, message, wparam, lparam); |
| 113 | } |
| 114 | return singleton->WndProc(hwnd, message, wparam, lparam); |
| 115 | } |
| 116 | |
[email protected] | edf04b51 | 2012-02-23 09:47:43 | [diff] [blame] | 117 | bool ParseCommandLine(const COPYDATASTRUCT* cds, |
| 118 | CommandLine* parsed_command_line, |
| 119 | FilePath* current_directory) { |
| 120 | // We should have enough room for the shortest command (min_message_size) |
| 121 | // and also be a multiple of wchar_t bytes. The shortest command |
| 122 | // possible is L"START\0\0" (empty current directory and command line). |
| 123 | static const int min_message_size = 7; |
| 124 | if (cds->cbData < min_message_size * sizeof(wchar_t) || |
| 125 | cds->cbData % sizeof(wchar_t) != 0) { |
| 126 | LOG(WARNING) << "Invalid WM_COPYDATA, length = " << cds->cbData; |
| 127 | return false; |
| 128 | } |
| 129 | |
| 130 | // We split the string into 4 parts on NULLs. |
| 131 | DCHECK(cds->lpData); |
| 132 | const std::wstring msg(static_cast<wchar_t*>(cds->lpData), |
| 133 | cds->cbData / sizeof(wchar_t)); |
| 134 | const std::wstring::size_type first_null = msg.find_first_of(L'\0'); |
| 135 | if (first_null == 0 || first_null == std::wstring::npos) { |
| 136 | // no NULL byte, don't know what to do |
| 137 | LOG(WARNING) << "Invalid WM_COPYDATA, length = " << msg.length() << |
| 138 | ", first null = " << first_null; |
| 139 | return false; |
| 140 | } |
| 141 | |
| 142 | // Decode the command, which is everything until the first NULL. |
| 143 | if (msg.substr(0, first_null) == L"START") { |
| 144 | // Another instance is starting parse the command line & do what it would |
| 145 | // have done. |
| 146 | VLOG(1) << "Handling STARTUP request from another process"; |
| 147 | const std::wstring::size_type second_null = |
| 148 | msg.find_first_of(L'\0', first_null + 1); |
| 149 | if (second_null == std::wstring::npos || |
| 150 | first_null == msg.length() - 1 || second_null == msg.length()) { |
| 151 | LOG(WARNING) << "Invalid format for start command, we need a string in 4 " |
| 152 | "parts separated by NULLs"; |
| 153 | return false; |
| 154 | } |
| 155 | |
| 156 | // Get current directory. |
| 157 | *current_directory = FilePath(msg.substr(first_null + 1, |
| 158 | second_null - first_null)); |
| 159 | |
| 160 | const std::wstring::size_type third_null = |
| 161 | msg.find_first_of(L'\0', second_null + 1); |
| 162 | if (third_null == std::wstring::npos || |
| 163 | third_null == msg.length()) { |
| 164 | LOG(WARNING) << "Invalid format for start command, we need a string in 4 " |
| 165 | "parts separated by NULLs"; |
| 166 | } |
| 167 | |
| 168 | // Get command line. |
| 169 | const std::wstring cmd_line = |
| 170 | msg.substr(second_null + 1, third_null - second_null); |
| 171 | *parsed_command_line = CommandLine::FromString(cmd_line); |
| 172 | return true; |
| 173 | } |
| 174 | return false; |
| 175 | } |
| 176 | |
[email protected] | 6927f2e0 | 2012-09-18 00:13:49 | [diff] [blame] | 177 | bool ActivateMetroChrome() { |
| 178 | FilePath chrome_exe; |
| 179 | if (!PathService::Get(base::FILE_EXE, &chrome_exe)) { |
| 180 | NOTREACHED() << "Failed to get chrome exe path"; |
| 181 | return false; |
| 182 | } |
| 183 | string16 app_id = ShellUtil::GetBrowserModelId( |
[email protected] | 78679969 | 2012-09-26 14:16:48 | [diff] [blame] | 184 | BrowserDistribution::GetDistribution(), |
| 185 | InstallUtil::IsPerUserInstall(chrome_exe.value().c_str())); |
[email protected] | 6927f2e0 | 2012-09-18 00:13:49 | [diff] [blame] | 186 | if (app_id.empty()) { |
| 187 | NOTREACHED() << "Failed to get chrome app user model id."; |
| 188 | return false; |
| 189 | } |
| 190 | |
| 191 | base::win::ScopedComPtr<IApplicationActivationManager> activation_manager; |
| 192 | HRESULT hr = activation_manager.CreateInstance( |
| 193 | CLSID_ApplicationActivationManager); |
| 194 | if (!activation_manager) { |
| 195 | NOTREACHED() << "Failed to cocreate activation manager. Error: " << hr; |
| 196 | return false; |
| 197 | } |
| 198 | |
| 199 | unsigned long pid = 0; |
| 200 | hr = activation_manager->ActivateApplication(app_id.c_str(), |
| 201 | L"open", |
| 202 | AO_NONE, |
| 203 | &pid); |
| 204 | if (FAILED(hr)) { |
| 205 | NOTREACHED() << "Failed to activate metro chrome. Error: " << hr; |
| 206 | return false; |
| 207 | } |
| 208 | return true; |
| 209 | } |
| 210 | |
[email protected] | b3d791c9 | 2012-10-05 19:01:24 | [diff] [blame] | 211 | // Returns true if Chrome needs to be relaunched into Windows 8 immersive mode. |
| 212 | // Following conditions apply:- |
| 213 | // 1. Windows 8 or greater. |
| 214 | // 2. Not in Windows 8 immersive mode. |
[email protected] | 24590067 | 2012-10-13 02:27:11 | [diff] [blame] | 215 | // 3. Chrome is default browser. |
| 216 | // 4. Process integrity level is not high. |
| 217 | // 5. The profile data directory is the default directory. |
| 218 | // 6. Last used mode was immersive/machine is a tablet. |
[email protected] | b3d791c9 | 2012-10-05 19:01:24 | [diff] [blame] | 219 | // TODO(ananta) |
| 220 | // Move this function to a common place as the Windows 8 delegate_execute |
| 221 | // handler can possibly use this. |
| 222 | bool ShouldLaunchInWindows8ImmersiveMode(const FilePath& user_data_dir) { |
| 223 | #if defined(USE_AURA) |
| 224 | return false; |
| 225 | #endif |
| 226 | |
| 227 | if (base::win::GetVersion() < base::win::VERSION_WIN8) |
| 228 | return false; |
| 229 | |
| 230 | if (base::win::IsProcessImmersive(base::GetCurrentProcessHandle())) |
| 231 | return false; |
| 232 | |
[email protected] | 8988665 | 2012-12-11 18:09:07 | [diff] [blame^] | 233 | if (ShellIntegration::GetDefaultBrowser() != ShellIntegration::IS_DEFAULT) |
[email protected] | 24590067 | 2012-10-13 02:27:11 | [diff] [blame] | 234 | return false; |
| 235 | |
[email protected] | b3d791c9 | 2012-10-05 19:01:24 | [diff] [blame] | 236 | base::IntegrityLevel integrity_level = base::INTEGRITY_UNKNOWN; |
| 237 | base::GetProcessIntegrityLevel(base::GetCurrentProcessHandle(), |
| 238 | &integrity_level); |
| 239 | if (integrity_level == base::HIGH_INTEGRITY) |
| 240 | return false; |
| 241 | |
| 242 | FilePath default_user_data_dir; |
| 243 | if (!chrome::GetDefaultUserDataDirectory(&default_user_data_dir)) |
| 244 | return false; |
| 245 | |
| 246 | if (default_user_data_dir != user_data_dir) |
| 247 | return false; |
| 248 | |
[email protected] | e212a306 | 2012-11-14 15:59:07 | [diff] [blame] | 249 | base::win::RegKey reg_key; |
[email protected] | 7e41600 | 2012-10-13 05:19:21 | [diff] [blame] | 250 | DWORD reg_value = 0; |
[email protected] | e212a306 | 2012-11-14 15:59:07 | [diff] [blame] | 251 | if (reg_key.Create(HKEY_CURRENT_USER, chrome::kMetroRegistryPath, |
| 252 | KEY_READ) == ERROR_SUCCESS && |
[email protected] | 7e41600 | 2012-10-13 05:19:21 | [diff] [blame] | 253 | reg_key.ReadValueDW(chrome::kLaunchModeValue, |
| 254 | ®_value) == ERROR_SUCCESS) { |
[email protected] | e212a306 | 2012-11-14 15:59:07 | [diff] [blame] | 255 | return reg_value == 1; |
[email protected] | b3d791c9 | 2012-10-05 19:01:24 | [diff] [blame] | 256 | } |
| 257 | return base::win::IsMachineATablet(); |
| 258 | } |
| 259 | |
[email protected] | fc14cef | 2009-01-27 22:17:29 | [diff] [blame] | 260 | } // namespace |
| 261 | |
[email protected] | 0a19455 | 2011-09-14 17:53:35 | [diff] [blame] | 262 | // Microsoft's Softricity virtualization breaks the sandbox processes. |
| 263 | // So, if we detect the Softricity DLL we use WMI Win32_Process.Create to |
| 264 | // break out of the virtualization environment. |
| 265 | // http://code.google.com/p/chromium/issues/detail?id=43650 |
| 266 | bool ProcessSingleton::EscapeVirtualization(const FilePath& user_data_dir) { |
| 267 | if (::GetModuleHandle(L"sftldr_wow64.dll") || |
| 268 | ::GetModuleHandle(L"sftldr.dll")) { |
| 269 | int process_id; |
| 270 | if (!installer::WMIProcess::Launch(GetCommandLineW(), &process_id)) |
| 271 | return false; |
| 272 | is_virtualized_ = true; |
| 273 | // The new window was spawned from WMI, and won't be in the foreground. |
| 274 | // So, first we sleep while the new chrome.exe instance starts (because |
| 275 | // WaitForInputIdle doesn't work here). Then we poll for up to two more |
| 276 | // seconds and make the window foreground if we find it (or we give up). |
| 277 | HWND hwnd = 0; |
| 278 | ::Sleep(90); |
| 279 | for (int tries = 200; tries; --tries) { |
| 280 | hwnd = FindWindowEx(HWND_MESSAGE, NULL, chrome::kMessageWindowClass, |
| 281 | user_data_dir.value().c_str()); |
| 282 | if (hwnd) { |
| 283 | ::SetForegroundWindow(hwnd); |
| 284 | break; |
| 285 | } |
| 286 | ::Sleep(10); |
| 287 | } |
| 288 | return true; |
| 289 | } |
| 290 | return false; |
| 291 | } |
| 292 | |
[email protected] | 7c47ae3e | 2009-02-18 00:34:21 | [diff] [blame] | 293 | ProcessSingleton::ProcessSingleton(const FilePath& user_data_dir) |
[email protected] | 0a19455 | 2011-09-14 17:53:35 | [diff] [blame] | 294 | : window_(NULL), locked_(false), foreground_window_(NULL), |
[email protected] | 14649ecd | 2012-12-05 01:00:24 | [diff] [blame] | 295 | is_virtualized_(false), lock_file_(INVALID_HANDLE_VALUE), |
| 296 | user_data_dir_(user_data_dir) { |
[email protected] | fc14cef | 2009-01-27 22:17:29 | [diff] [blame] | 297 | } |
| 298 | |
[email protected] | 7c47ae3e | 2009-02-18 00:34:21 | [diff] [blame] | 299 | ProcessSingleton::~ProcessSingleton() { |
[email protected] | a9b36c9 | 2012-06-18 08:47:57 | [diff] [blame] | 300 | // We need to unregister the window as late as possible so that we can detect |
| 301 | // another instance of chrome running. Otherwise we may end up writing out |
| 302 | // data while a new chrome is starting up. |
| 303 | if (window_) { |
| 304 | ::DestroyWindow(window_); |
| 305 | ::UnregisterClass(chrome::kMessageWindowClass, |
| 306 | base::GetModuleFromAddress(&ThunkWndProc)); |
| 307 | } |
[email protected] | 2b7ff5b9 | 2012-07-17 22:45:18 | [diff] [blame] | 308 | if (lock_file_ != INVALID_HANDLE_VALUE) |
| 309 | CloseHandle(lock_file_); |
[email protected] | fc14cef | 2009-01-27 22:17:29 | [diff] [blame] | 310 | } |
| 311 | |
[email protected] | 14649ecd | 2012-12-05 01:00:24 | [diff] [blame] | 312 | // Code roughly based on Mozilla. |
[email protected] | 9f20a6d0 | 2009-08-21 01:18:37 | [diff] [blame] | 313 | ProcessSingleton::NotifyResult ProcessSingleton::NotifyOtherProcess() { |
[email protected] | 0a19455 | 2011-09-14 17:53:35 | [diff] [blame] | 314 | if (is_virtualized_) |
| 315 | return PROCESS_NOTIFIED; // We already spawned the process in this case. |
[email protected] | e9613b5 | 2012-11-27 22:35:13 | [diff] [blame] | 316 | if (lock_file_ == INVALID_HANDLE_VALUE && !remote_window_) { |
[email protected] | 2b7ff5b9 | 2012-07-17 22:45:18 | [diff] [blame] | 317 | return LOCK_ERROR; |
[email protected] | e9613b5 | 2012-11-27 22:35:13 | [diff] [blame] | 318 | } else if (!remote_window_) { |
[email protected] | 9f20a6d0 | 2009-08-21 01:18:37 | [diff] [blame] | 319 | return PROCESS_NONE; |
[email protected] | e9613b5 | 2012-11-27 22:35:13 | [diff] [blame] | 320 | } |
[email protected] | fc14cef | 2009-01-27 22:17:29 | [diff] [blame] | 321 | |
[email protected] | f394bc6 | 2012-08-22 21:42:29 | [diff] [blame] | 322 | DWORD process_id = 0; |
| 323 | DWORD thread_id = GetWindowThreadProcessId(remote_window_, &process_id); |
| 324 | // It is possible that the process owning this window may have died by now. |
| 325 | if (!thread_id || !process_id) { |
| 326 | remote_window_ = NULL; |
| 327 | return PROCESS_NONE; |
| 328 | } |
| 329 | |
| 330 | if (base::win::IsMetroProcess()) { |
| 331 | // Interesting corner case. We are launched as a metro process but we |
| 332 | // found another chrome running. Since metro enforces single instance then |
| 333 | // the other chrome must be desktop chrome and this must be a search charm |
| 334 | // activation. This scenario is unique; other cases should be properly |
| 335 | // handled by the delegate_execute which will not activate a second chrome. |
| 336 | string16 terms; |
| 337 | base::win::MetroLaunchType launch = base::win::GetMetroLaunchParams(&terms); |
| 338 | if (launch != base::win::METRO_SEARCH) { |
| 339 | LOG(WARNING) << "In metro mode, but and launch is " << launch; |
| 340 | } else { |
| 341 | std::string query = net::EscapeQueryParamValue(UTF16ToUTF8(terms), true); |
| 342 | std::string url = base::StringPrintf(kSearchUrl, query.c_str()); |
| 343 | SHELLEXECUTEINFOA sei = { sizeof(sei) }; |
| 344 | sei.fMask = SEE_MASK_FLAG_LOG_USAGE; |
| 345 | sei.nShow = SW_SHOWNORMAL; |
| 346 | sei.lpFile = url.c_str(); |
| 347 | OutputDebugStringA(sei.lpFile); |
| 348 | sei.lpDirectory = ""; |
| 349 | ::ShellExecuteExA(&sei); |
| 350 | } |
| 351 | return PROCESS_NOTIFIED; |
| 352 | } |
[email protected] | ec6e7d7 | 2012-12-06 01:30:42 | [diff] [blame] | 353 | |
[email protected] | f394bc6 | 2012-08-22 21:42:29 | [diff] [blame] | 354 | // Non-metro mode, send our command line to the other chrome message window. |
[email protected] | fc14cef | 2009-01-27 22:17:29 | [diff] [blame] | 355 | // format is "START\0<<<current directory>>>\0<<<commandline>>>". |
| 356 | std::wstring to_send(L"START\0", 6); // want the NULL in the string. |
[email protected] | b969648 | 2010-11-30 23:56:18 | [diff] [blame] | 357 | FilePath cur_dir; |
[email protected] | fc14cef | 2009-01-27 22:17:29 | [diff] [blame] | 358 | if (!PathService::Get(base::DIR_CURRENT, &cur_dir)) |
[email protected] | 9f20a6d0 | 2009-08-21 01:18:37 | [diff] [blame] | 359 | return PROCESS_NONE; |
[email protected] | b969648 | 2010-11-30 23:56:18 | [diff] [blame] | 360 | to_send.append(cur_dir.value()); |
[email protected] | fc14cef | 2009-01-27 22:17:29 | [diff] [blame] | 361 | to_send.append(L"\0", 1); // Null separator. |
| 362 | to_send.append(GetCommandLineW()); |
| 363 | to_send.append(L"\0", 1); // Null separator. |
| 364 | |
[email protected] | ec6e7d7 | 2012-12-06 01:30:42 | [diff] [blame] | 365 | base::win::ScopedHandle process_handle; |
| 366 | if (base::win::GetVersion() >= base::win::VERSION_WIN8 && |
| 367 | base::OpenProcessHandleWithAccess( |
| 368 | process_id, PROCESS_QUERY_INFORMATION, |
| 369 | process_handle.Receive()) && |
| 370 | base::win::IsProcessImmersive(process_handle.Get())) { |
| 371 | ActivateMetroChrome(); |
| 372 | } |
| 373 | |
[email protected] | fc14cef | 2009-01-27 22:17:29 | [diff] [blame] | 374 | // Allow the current running browser window making itself the foreground |
| 375 | // window (otherwise it will just flash in the taskbar). |
[email protected] | fc14cef | 2009-01-27 22:17:29 | [diff] [blame] | 376 | AllowSetForegroundWindow(process_id); |
| 377 | |
[email protected] | fc14cef | 2009-01-27 22:17:29 | [diff] [blame] | 378 | COPYDATASTRUCT cds; |
| 379 | cds.dwData = 0; |
| 380 | cds.cbData = static_cast<DWORD>((to_send.length() + 1) * sizeof(wchar_t)); |
| 381 | cds.lpData = const_cast<wchar_t*>(to_send.c_str()); |
| 382 | DWORD_PTR result = 0; |
| 383 | if (SendMessageTimeout(remote_window_, |
| 384 | WM_COPYDATA, |
| 385 | NULL, |
| 386 | reinterpret_cast<LPARAM>(&cds), |
| 387 | SMTO_ABORTIFHUNG, |
[email protected] | 8b08cbd | 2009-08-04 05:34:19 | [diff] [blame] | 388 | kTimeoutInSeconds * 1000, |
[email protected] | fc14cef | 2009-01-27 22:17:29 | [diff] [blame] | 389 | &result)) { |
[email protected] | 0815b6d | 2009-02-11 00:39:37 | [diff] [blame] | 390 | // It is possible that the process owning this window may have died by now. |
| 391 | if (!result) { |
| 392 | remote_window_ = NULL; |
[email protected] | 9f20a6d0 | 2009-08-21 01:18:37 | [diff] [blame] | 393 | return PROCESS_NONE; |
[email protected] | 0815b6d | 2009-02-11 00:39:37 | [diff] [blame] | 394 | } |
[email protected] | 9f20a6d0 | 2009-08-21 01:18:37 | [diff] [blame] | 395 | return PROCESS_NOTIFIED; |
[email protected] | fc14cef | 2009-01-27 22:17:29 | [diff] [blame] | 396 | } |
| 397 | |
[email protected] | 0815b6d | 2009-02-11 00:39:37 | [diff] [blame] | 398 | // It is possible that the process owning this window may have died by now. |
| 399 | if (!IsWindow(remote_window_)) { |
| 400 | remote_window_ = NULL; |
[email protected] | 9f20a6d0 | 2009-08-21 01:18:37 | [diff] [blame] | 401 | return PROCESS_NONE; |
[email protected] | 0815b6d | 2009-02-11 00:39:37 | [diff] [blame] | 402 | } |
| 403 | |
[email protected] | fc14cef | 2009-01-27 22:17:29 | [diff] [blame] | 404 | // The window is hung. Scan for every window to find a visible one. |
| 405 | bool visible_window = false; |
| 406 | EnumThreadWindows(thread_id, |
| 407 | &BrowserWindowEnumeration, |
| 408 | reinterpret_cast<LPARAM>(&visible_window)); |
| 409 | |
| 410 | // If there is a visible browser window, ask the user before killing it. |
[email protected] | d3322029 | 2012-07-04 01:41:27 | [diff] [blame] | 411 | if (visible_window && chrome::ShowMessageBox(NULL, |
[email protected] | 5da155e | 2012-05-26 16:31:16 | [diff] [blame] | 412 | l10n_util::GetStringUTF16(IDS_PRODUCT_NAME), |
| 413 | l10n_util::GetStringUTF16(IDS_BROWSER_HUNGBROWSER_MESSAGE), |
[email protected] | d3322029 | 2012-07-04 01:41:27 | [diff] [blame] | 414 | chrome::MESSAGE_BOX_TYPE_QUESTION) == chrome::MESSAGE_BOX_RESULT_NO) { |
[email protected] | 5da155e | 2012-05-26 16:31:16 | [diff] [blame] | 415 | // The user denied. Quit silently. |
| 416 | return PROCESS_NOTIFIED; |
[email protected] | fc14cef | 2009-01-27 22:17:29 | [diff] [blame] | 417 | } |
| 418 | |
| 419 | // Time to take action. Kill the browser process. |
[email protected] | 1fcfb20 | 2011-07-19 19:53:14 | [diff] [blame] | 420 | base::KillProcessById(process_id, content::RESULT_CODE_HUNG, true); |
[email protected] | fc14cef | 2009-01-27 22:17:29 | [diff] [blame] | 421 | remote_window_ = NULL; |
[email protected] | 9f20a6d0 | 2009-08-21 01:18:37 | [diff] [blame] | 422 | return PROCESS_NONE; |
[email protected] | fc14cef | 2009-01-27 22:17:29 | [diff] [blame] | 423 | } |
| 424 | |
[email protected] | 5d36454 | 2012-04-05 07:15:39 | [diff] [blame] | 425 | ProcessSingleton::NotifyResult ProcessSingleton::NotifyOtherProcessOrCreate( |
| 426 | const NotificationCallback& notification_callback) { |
[email protected] | 14649ecd | 2012-12-05 01:00:24 | [diff] [blame] | 427 | ProcessSingleton::NotifyResult result = PROCESS_NONE; |
| 428 | if (!Create(notification_callback)) { |
| 429 | result = NotifyOtherProcess(); |
| 430 | if (result == PROCESS_NONE) |
| 431 | result = PROFILE_IN_USE; |
[email protected] | 351f7d4 | 2012-12-07 22:42:02 | [diff] [blame] | 432 | } else { |
| 433 | g_browser_process->PlatformSpecificCommandLineProcessing( |
| 434 | *CommandLine::ForCurrentProcess()); |
[email protected] | 14649ecd | 2012-12-05 01:00:24 | [diff] [blame] | 435 | } |
| 436 | return result; |
[email protected] | 4a44bc3 | 2010-05-28 22:22:44 | [diff] [blame] | 437 | } |
| 438 | |
[email protected] | 14649ecd | 2012-12-05 01:00:24 | [diff] [blame] | 439 | // Look for a Chrome instance that uses the same profile directory. If there |
| 440 | // isn't one, create a message window with its title set to the profile |
| 441 | // directory path. |
[email protected] | 5d36454 | 2012-04-05 07:15:39 | [diff] [blame] | 442 | bool ProcessSingleton::Create( |
| 443 | const NotificationCallback& notification_callback) { |
[email protected] | 5d36454 | 2012-04-05 07:15:39 | [diff] [blame] | 444 | DCHECK(notification_callback_.is_null()); |
| 445 | |
[email protected] | 14649ecd | 2012-12-05 01:00:24 | [diff] [blame] | 446 | static const wchar_t kMutexName[] = L"Local\\ChromeProcessSingletonStartup!"; |
| 447 | static const wchar_t kMetroActivationEventName[] = |
| 448 | L"Local\\ChromeProcessSingletonStartupMetroActivation!"; |
| 449 | |
| 450 | remote_window_ = FindWindowEx(HWND_MESSAGE, NULL, |
| 451 | chrome::kMessageWindowClass, |
| 452 | user_data_dir_.value().c_str()); |
| 453 | if (!remote_window_ && !EscapeVirtualization(user_data_dir_)) { |
| 454 | // Make sure we will be the one and only process creating the window. |
| 455 | // We use a named Mutex since we are protecting against multi-process |
| 456 | // access. As documented, it's clearer to NOT request ownership on creation |
| 457 | // since it isn't guaranteed we will get it. It is better to create it |
| 458 | // without ownership and explicitly get the ownership afterward. |
| 459 | base::win::ScopedHandle only_me(CreateMutex(NULL, FALSE, kMutexName)); |
| 460 | DPCHECK(only_me.IsValid()); |
| 461 | |
| 462 | AutoLockMutex auto_lock_only_me(only_me); |
| 463 | |
| 464 | // We now own the mutex so we are the only process that can create the |
| 465 | // window at this time, but we must still check if someone created it |
| 466 | // between the time where we looked for it above and the time the mutex |
| 467 | // was given to us. |
| 468 | remote_window_ = FindWindowEx(HWND_MESSAGE, NULL, |
| 469 | chrome::kMessageWindowClass, |
| 470 | user_data_dir_.value().c_str()); |
| 471 | |
| 472 | |
| 473 | // In Win8+, a new Chrome process launched in Desktop mode may need to be |
| 474 | // transmuted into Metro Chrome (see ShouldLaunchInWindows8ImmersiveMode for |
| 475 | // heuristics). To accomplish this, the current Chrome activates Metro |
| 476 | // Chrome, releases the startup mutex, and waits for metro Chrome to take |
| 477 | // the singleton. From that point onward, the command line for this Chrome |
| 478 | // process will be sent to Metro Chrome by the usual channels. |
| 479 | if (!remote_window_ && base::win::GetVersion() >= base::win::VERSION_WIN8 && |
| 480 | !base::win::IsMetroProcess()) { |
| 481 | // |metro_activation_event| is created right before activating a Metro |
| 482 | // Chrome (note that there can only be one Metro Chrome process; by OS |
| 483 | // design); all following Desktop processes will then wait for this event |
| 484 | // to be signaled by Metro Chrome which will do so as soon as it grabs |
| 485 | // this singleton (should any of the waiting processes timeout waiting for |
| 486 | // the signal they will try to grab the singleton for themselves which |
| 487 | // will result in a forced Desktop Chrome launch in the worst case). |
| 488 | base::win::ScopedHandle metro_activation_event( |
| 489 | OpenEvent(SYNCHRONIZE, FALSE, kMetroActivationEventName)); |
| 490 | if (!metro_activation_event.IsValid() && |
| 491 | ShouldLaunchInWindows8ImmersiveMode(user_data_dir_)) { |
| 492 | // No Metro activation is under way, but the desire is to launch in |
| 493 | // Metro mode: activate and rendez-vous with the activated process. |
| 494 | metro_activation_event.Set( |
| 495 | CreateEvent(NULL, TRUE, FALSE, kMetroActivationEventName)); |
| 496 | if (!ActivateMetroChrome()) { |
| 497 | // Failed to launch immersive Chrome, default to launching on Desktop. |
| 498 | LOG(ERROR) << "Failed to launch immersive chrome"; |
| 499 | metro_activation_event.Close(); |
| 500 | } |
| 501 | } |
| 502 | |
| 503 | if (metro_activation_event.IsValid()) { |
| 504 | // Release |only_me| (to let Metro Chrome grab this singleton) and wait |
| 505 | // until the event is signaled (i.e. Metro Chrome was successfully |
| 506 | // activated). Ignore timeout waiting for |metro_activation_event|. |
| 507 | { |
| 508 | AutoUnlockMutex auto_unlock_only_me(only_me); |
| 509 | |
| 510 | DWORD result = WaitForSingleObject(metro_activation_event, |
| 511 | kMetroChromeActivationTimeoutMs); |
| 512 | DPCHECK(result == WAIT_OBJECT_0 || result == WAIT_TIMEOUT) |
| 513 | << "Result = " << result; |
| 514 | } |
| 515 | |
| 516 | // Check if this singleton was successfully grabbed by another process |
| 517 | // (hopefully Metro Chrome). Failing to do so, this process will grab |
| 518 | // the singleton and launch in Desktop mode. |
| 519 | remote_window_ = FindWindowEx(HWND_MESSAGE, NULL, |
| 520 | chrome::kMessageWindowClass, |
| 521 | user_data_dir_.value().c_str()); |
| 522 | } |
| 523 | } |
| 524 | |
| 525 | if (!remote_window_) { |
| 526 | // We have to make sure there is no Chrome instance running on another |
| 527 | // machine that uses the same profile. |
| 528 | FilePath lock_file_path = user_data_dir_.AppendASCII(kLockfile); |
| 529 | lock_file_ = CreateFile(lock_file_path.value().c_str(), |
| 530 | GENERIC_WRITE, |
| 531 | FILE_SHARE_READ, |
| 532 | NULL, |
| 533 | CREATE_ALWAYS, |
| 534 | FILE_ATTRIBUTE_NORMAL | FILE_FLAG_DELETE_ON_CLOSE, |
| 535 | NULL); |
| 536 | DWORD error = GetLastError(); |
| 537 | LOG_IF(WARNING, lock_file_ != INVALID_HANDLE_VALUE && |
| 538 | error == ERROR_ALREADY_EXISTS) << "Lock file exists but is writable."; |
| 539 | LOG_IF(ERROR, lock_file_ == INVALID_HANDLE_VALUE) |
| 540 | << "Lock file can not be created! Error code: " << error; |
| 541 | |
| 542 | if (lock_file_ != INVALID_HANDLE_VALUE) { |
| 543 | HINSTANCE hinst = base::GetModuleFromAddress(&ThunkWndProc); |
| 544 | |
| 545 | WNDCLASSEX wc = {0}; |
| 546 | wc.cbSize = sizeof(wc); |
| 547 | wc.lpfnWndProc = base::win::WrappedWindowProc<ThunkWndProc>; |
| 548 | wc.hInstance = hinst; |
| 549 | wc.lpszClassName = chrome::kMessageWindowClass; |
| 550 | ATOM clazz = ::RegisterClassEx(&wc); |
| 551 | DCHECK(clazz); |
| 552 | |
| 553 | // Set the window's title to the path of our user data directory so |
| 554 | // other Chrome instances can decide if they should forward to us. |
| 555 | window_ = ::CreateWindow(MAKEINTATOM(clazz), |
| 556 | user_data_dir_.value().c_str(), |
| 557 | 0, 0, 0, 0, 0, HWND_MESSAGE, 0, hinst, this); |
| 558 | CHECK(window_); |
| 559 | } |
| 560 | |
| 561 | if (base::win::GetVersion() >= base::win::VERSION_WIN8) { |
| 562 | // Make sure no one is still waiting on Metro activation whether it |
| 563 | // succeeded (i.e., this is the Metro process) or failed. |
| 564 | base::win::ScopedHandle metro_activation_event( |
| 565 | OpenEvent(EVENT_MODIFY_STATE, FALSE, kMetroActivationEventName)); |
| 566 | if (metro_activation_event.IsValid()) |
| 567 | SetEvent(metro_activation_event); |
| 568 | } |
| 569 | } |
| 570 | } |
| 571 | |
[email protected] | 5d36454 | 2012-04-05 07:15:39 | [diff] [blame] | 572 | if (window_ != NULL) |
| 573 | notification_callback_ = notification_callback; |
| 574 | |
[email protected] | 9a18283 | 2012-02-10 18:45:58 | [diff] [blame] | 575 | return window_ != NULL; |
[email protected] | fc14cef | 2009-01-27 22:17:29 | [diff] [blame] | 576 | } |
| 577 | |
[email protected] | 9f20a6d0 | 2009-08-21 01:18:37 | [diff] [blame] | 578 | void ProcessSingleton::Cleanup() { |
| 579 | } |
| 580 | |
[email protected] | 7c47ae3e | 2009-02-18 00:34:21 | [diff] [blame] | 581 | LRESULT ProcessSingleton::OnCopyData(HWND hwnd, const COPYDATASTRUCT* cds) { |
[email protected] | 175a7a2 | 2009-05-03 15:57:53 | [diff] [blame] | 582 | // If locked, it means we are not ready to process this message because |
[email protected] | afd20c02 | 2010-06-10 00:48:20 | [diff] [blame] | 583 | // we are probably in a first run critical phase. |
[email protected] | 175a7a2 | 2009-05-03 15:57:53 | [diff] [blame] | 584 | if (locked_) { |
[email protected] | 95259c6 | 2011-10-25 23:23:53 | [diff] [blame] | 585 | #if defined(USE_AURA) |
| 586 | NOTIMPLEMENTED(); |
| 587 | #else |
[email protected] | 175a7a2 | 2009-05-03 15:57:53 | [diff] [blame] | 588 | // Attempt to place ourselves in the foreground / flash the task bar. |
[email protected] | edf04b51 | 2012-02-23 09:47:43 | [diff] [blame] | 589 | if (foreground_window_ != NULL && IsWindow(foreground_window_)) { |
[email protected] | 175a7a2 | 2009-05-03 15:57:53 | [diff] [blame] | 590 | SetForegroundWindow(foreground_window_); |
[email protected] | edf04b51 | 2012-02-23 09:47:43 | [diff] [blame] | 591 | } else { |
| 592 | // Read the command line and store it. It will be replayed when the |
| 593 | // ProcessSingleton becomes unlocked. |
| 594 | CommandLine parsed_command_line(CommandLine::NO_PROGRAM); |
| 595 | FilePath current_directory; |
| 596 | if (ParseCommandLine(cds, &parsed_command_line, ¤t_directory)) |
| 597 | saved_startup_messages_.push_back( |
| 598 | std::make_pair(parsed_command_line.argv(), current_directory)); |
| 599 | } |
[email protected] | 95259c6 | 2011-10-25 23:23:53 | [diff] [blame] | 600 | #endif |
[email protected] | 175a7a2 | 2009-05-03 15:57:53 | [diff] [blame] | 601 | return TRUE; |
| 602 | } |
| 603 | |
[email protected] | edf04b51 | 2012-02-23 09:47:43 | [diff] [blame] | 604 | CommandLine parsed_command_line(CommandLine::NO_PROGRAM); |
| 605 | FilePath current_directory; |
| 606 | if (!ParseCommandLine(cds, &parsed_command_line, ¤t_directory)) |
[email protected] | fc14cef | 2009-01-27 22:17:29 | [diff] [blame] | 607 | return TRUE; |
[email protected] | 5d36454 | 2012-04-05 07:15:39 | [diff] [blame] | 608 | return notification_callback_.Run(parsed_command_line, current_directory) ? |
| 609 | TRUE : FALSE; |
[email protected] | fc14cef | 2009-01-27 22:17:29 | [diff] [blame] | 610 | } |
| 611 | |
[email protected] | e3ce40a | 2012-01-31 03:03:03 | [diff] [blame] | 612 | LRESULT ProcessSingleton::WndProc(HWND hwnd, UINT message, |
| 613 | WPARAM wparam, LPARAM lparam) { |
[email protected] | fc14cef | 2009-01-27 22:17:29 | [diff] [blame] | 614 | switch (message) { |
| 615 | case WM_COPYDATA: |
| 616 | return OnCopyData(reinterpret_cast<HWND>(wparam), |
| 617 | reinterpret_cast<COPYDATASTRUCT*>(lparam)); |
| 618 | default: |
| 619 | break; |
| 620 | } |
| 621 | |
| 622 | return ::DefWindowProc(hwnd, message, wparam, lparam); |
| 623 | } |