OLD | NEW |
1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 #include "components/browser_watcher/window_hang_monitor_win.h" | 4 #include "components/browser_watcher/window_hang_monitor_win.h" |
5 | 5 |
6 #include <utility> | 6 #include <utility> |
7 | 7 |
8 #include "base/callback.h" | 8 #include "base/callback.h" |
9 #include "base/files/file_util.h" | 9 #include "base/files/file_util.h" |
10 #include "base/location.h" | 10 #include "base/location.h" |
11 #include "base/logging.h" | 11 #include "base/logging.h" |
12 #include "base/message_loop/message_loop.h" | |
13 #include "base/strings/string_piece.h" | 12 #include "base/strings/string_piece.h" |
14 #include "base/strings/string_util.h" | 13 #include "base/strings/string_util.h" |
| 14 #include "base/threading/thread_task_runner_handle.h" |
15 #include "base/win/message_window.h" | 15 #include "base/win/message_window.h" |
16 | 16 |
17 namespace browser_watcher { | 17 namespace browser_watcher { |
18 | 18 |
19 namespace { | 19 namespace { |
20 | 20 |
21 // Returns true if the class name for |window| equals |str|. | 21 // Returns true if the class name for |window| equals |str|. |
22 bool WindowClassNameEqualsString(HWND window, base::StringPiece16 str) { | 22 bool WindowClassNameEqualsString(HWND window, base::StringPiece16 str) { |
23 wchar_t class_name[MAX_PATH]; | 23 wchar_t class_name[MAX_PATH]; |
24 int str_length = ::GetClassName(window, class_name, MAX_PATH); | 24 int str_length = ::GetClassName(window, class_name, MAX_PATH); |
(...skipping 43 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
68 if (outstanding_ping_) { | 68 if (outstanding_ping_) { |
69 // We have an outstanding ping, disable it and leak it intentionally as | 69 // We have an outstanding ping, disable it and leak it intentionally as |
70 // if the callback arrives eventually, it'll cause a use-after-free. | 70 // if the callback arrives eventually, it'll cause a use-after-free. |
71 outstanding_ping_->monitor = nullptr; | 71 outstanding_ping_->monitor = nullptr; |
72 outstanding_ping_ = nullptr; | 72 outstanding_ping_ = nullptr; |
73 } | 73 } |
74 } | 74 } |
75 | 75 |
76 void WindowHangMonitor::Initialize(base::Process process) { | 76 void WindowHangMonitor::Initialize(base::Process process) { |
77 window_process_ = std::move(process); | 77 window_process_ = std::move(process); |
78 timer_.SetTaskRunner(base::MessageLoop::current()->task_runner()); | 78 timer_.SetTaskRunner(base::ThreadTaskRunnerHandle::Get()); |
79 | 79 |
80 ScheduleFindWindow(); | 80 ScheduleFindWindow(); |
81 } | 81 } |
82 | 82 |
83 void WindowHangMonitor::ScheduleFindWindow() { | 83 void WindowHangMonitor::ScheduleFindWindow() { |
84 // TODO(erikwright): We could reduce the polling by using WaitForInputIdle, | 84 // TODO(erikwright): We could reduce the polling by using WaitForInputIdle, |
85 // but it is hard to test (requiring a non-Console executable). | 85 // but it is hard to test (requiring a non-Console executable). |
86 timer_.Start( | 86 timer_.Start( |
87 FROM_HERE, ping_interval_, | 87 FROM_HERE, ping_interval_, |
88 base::Bind(&WindowHangMonitor::PollForWindow, base::Unretained(this))); | 88 base::Bind(&WindowHangMonitor::PollForWindow, base::Unretained(this))); |
(...skipping 93 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
182 // at any point after we found it. | 182 // at any point after we found it. |
183 HWND hwnd = FindChromeMessageWindow(window_process_.Pid()); | 183 HWND hwnd = FindChromeMessageWindow(window_process_.Pid()); |
184 if (hwnd) { | 184 if (hwnd) { |
185 SendPing(hwnd); | 185 SendPing(hwnd); |
186 } else { | 186 } else { |
187 callback_.Run(WINDOW_VANISHED); | 187 callback_.Run(WINDOW_VANISHED); |
188 } | 188 } |
189 } | 189 } |
190 | 190 |
191 } // namespace browser_watcher | 191 } // namespace browser_watcher |
OLD | NEW |