chrome/browser/chrome_process_finder_win.cc

   1 // Copyright 2013 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/browser/chrome_process_finder_win.h"
   6
   7 #include <shellapi.h>
   8 #include <string>
   9
  10 #include "base/command_line.h"
  11 #include "base/files/file_path.h"
  12 #include "base/files/file_util.h"
  13 #include "base/logging.h"
  14 #include "base/process/process.h"
  15 #include "base/process/process_info.h"
  16 #include "base/strings/string_number_conversions.h"
  17 #include "base/strings/stringprintf.h"
  18 #include "base/strings/utf_string_conversions.h"
  19 #include "base/win/message_window.h"
  20 #include "base/win/scoped_handle.h"
  21 #include "base/win/win_util.h"
  22 #include "base/win/windows_version.h"
  23 #include "chrome/common/chrome_constants.h"
  24 #include "chrome/common/chrome_switches.h"
  25
  26
  27 namespace {
  28
  29 int timeout_in_milliseconds = 20 * 1000;
  30
  31 }  // namespace
  32
  33 namespace chrome {
  34
  35 HWND FindRunningChromeWindow(const base::FilePath& user_data_dir) {
  36   return base::win::MessageWindow::FindWindow(user_data_dir.value());
  37 }
  38
  39 NotifyChromeResult AttemptToNotifyRunningChrome(HWND remote_window,
  40                                                 bool fast_start) {
  41   DCHECK(remote_window);
  42   DWORD process_id = 0;
  43   DWORD thread_id = GetWindowThreadProcessId(remote_window, &process_id);
  44   if (!thread_id || !process_id)
  45     return NOTIFY_FAILED;
  46
  47   base::CommandLine command_line(*base::CommandLine::ForCurrentProcess());
  48   command_line.AppendSwitchASCII(
  49       switches::kOriginalProcessStartTime,
  50       base::Int64ToString(
  51           base::CurrentProcessInfo::CreationTime().ToInternalValue()));
  52
  53   if (fast_start)
  54     command_line.AppendSwitch(switches::kFastStart);
  55
  56   // Send the command line to the remote chrome window.
  57   // Format is "START\0<<<current directory>>>\0<<<commandline>>>".
  58   std::wstring to_send(L"START\0", 6);  // want the NULL in the string.
  59   base::FilePath cur_dir;
  60   if (!base::GetCurrentDirectory(&cur_dir))
  61     return NOTIFY_FAILED;
  62   to_send.append(cur_dir.value());
  63   to_send.append(L"\0", 1);  // Null separator.
  64   to_send.append(command_line.GetCommandLineString());
  65   to_send.append(L"\0", 1);  // Null separator.
  66
  67   // Allow the current running browser window to make itself the foreground
  68   // window (otherwise it will just flash in the taskbar).
  69   ::AllowSetForegroundWindow(process_id);
  70
  71   COPYDATASTRUCT cds;
  72   cds.dwData = 0;
  73   cds.cbData = static_cast<DWORD>((to_send.length() + 1) * sizeof(wchar_t));
  74   cds.lpData = const_cast<wchar_t*>(to_send.c_str());
  75   DWORD_PTR result = 0;
  76   if (::SendMessageTimeout(remote_window, WM_COPYDATA, NULL,
  77                            reinterpret_cast<LPARAM>(&cds), SMTO_ABORTIFHUNG,
  78                            timeout_in_milliseconds, &result)) {
  79     return result ? NOTIFY_SUCCESS : NOTIFY_FAILED;
  80   }
  81
  82   // It is possible that the process owning this window may have died by now.
  83   if (!::IsWindow(remote_window))
  84     return NOTIFY_FAILED;
  85
  86   // If the window couldn't be notified but still exists, assume it is hung.
  87   return NOTIFY_WINDOW_HUNG;
  88 }
  89
  90 base::TimeDelta SetNotificationTimeoutForTesting(base::TimeDelta new_timeout) {
  91   base::TimeDelta old_timeout =
  92       base::TimeDelta::FromMilliseconds(timeout_in_milliseconds);
  93   timeout_in_milliseconds = new_timeout.InMilliseconds();
  94   return old_timeout;
  95 }
  96
  97 }  // namespace chrome