| // Copyright 2013 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 "chrome/browser/win/chrome_process_finder.h" |
| |
| #include <shellapi.h> |
| #include <string> |
| |
| #include "base/command_line.h" |
| #include "base/files/file_path.h" |
| #include "base/files/file_util.h" |
| #include "base/logging.h" |
| #include "base/numerics/safe_conversions.h" |
| #include "base/process/process.h" |
| #include "base/strings/string_number_conversions.h" |
| #include "base/strings/utf_string_conversions.h" |
| #include "base/win/message_window.h" |
| #include "base/win/scoped_handle.h" |
| #include "base/win/win_util.h" |
| #include "base/win/windows_version.h" |
| #include "chrome/common/chrome_constants.h" |
| #include "chrome/common/chrome_switches.h" |
| |
| |
| namespace { |
| |
| uint32_t g_timeout_in_milliseconds = 20 * 1000; |
| |
| } // namespace |
| |
| namespace chrome { |
| |
| HWND FindRunningChromeWindow(const base::FilePath& user_data_dir) { |
| return base::win::MessageWindow::FindWindow(user_data_dir.value()); |
| } |
| |
| NotifyChromeResult AttemptToNotifyRunningChrome(HWND remote_window, |
| bool fast_start) { |
| DCHECK(remote_window); |
| DWORD process_id = 0; |
| DWORD thread_id = GetWindowThreadProcessId(remote_window, &process_id); |
| if (!thread_id || !process_id) |
| return NOTIFY_FAILED; |
| |
| base::CommandLine command_line(*base::CommandLine::ForCurrentProcess()); |
| command_line.AppendSwitchASCII( |
| switches::kOriginalProcessStartTime, |
| base::Int64ToString( |
| base::Process::Current().CreationTime().ToInternalValue())); |
| |
| if (fast_start) |
| command_line.AppendSwitch(switches::kFastStart); |
| |
| // Send the command line to the remote chrome window. |
| // Format is "START\0<<<current directory>>>\0<<<commandline>>>". |
| std::wstring to_send(L"START\0", 6); // want the NULL in the string. |
| base::FilePath cur_dir; |
| if (!base::GetCurrentDirectory(&cur_dir)) |
| return NOTIFY_FAILED; |
| to_send.append(cur_dir.value()); |
| to_send.append(L"\0", 1); // Null separator. |
| to_send.append(command_line.GetCommandLineString()); |
| to_send.append(L"\0", 1); // Null separator. |
| |
| // Allow the current running browser window to make itself the foreground |
| // window (otherwise it will just flash in the taskbar). |
| ::AllowSetForegroundWindow(process_id); |
| |
| COPYDATASTRUCT cds; |
| cds.dwData = 0; |
| cds.cbData = static_cast<DWORD>((to_send.length() + 1) * sizeof(wchar_t)); |
| cds.lpData = const_cast<wchar_t*>(to_send.c_str()); |
| DWORD_PTR result = 0; |
| if (::SendMessageTimeout(remote_window, WM_COPYDATA, NULL, |
| reinterpret_cast<LPARAM>(&cds), SMTO_ABORTIFHUNG, |
| g_timeout_in_milliseconds, &result)) { |
| return result ? NOTIFY_SUCCESS : NOTIFY_FAILED; |
| } |
| |
| // If SendMessageTimeout failed to send message consider this as |
| // NOTIFY_FAILED. |
| if (::GetLastError() != ERROR_TIMEOUT) |
| return NOTIFY_FAILED; |
| |
| // It is possible that the process owning this window may have died by now. |
| if (!::IsWindow(remote_window)) |
| return NOTIFY_FAILED; |
| |
| // If the window couldn't be notified but still exists, assume it is hung. |
| return NOTIFY_WINDOW_HUNG; |
| } |
| |
| base::TimeDelta SetNotificationTimeoutForTesting(base::TimeDelta new_timeout) { |
| base::TimeDelta old_timeout = |
| base::TimeDelta::FromMilliseconds(g_timeout_in_milliseconds); |
| g_timeout_in_milliseconds = |
| base::checked_cast<uint32_t>(new_timeout.InMilliseconds()); |
| return old_timeout; |
| } |
| |
| } // namespace chrome |