summaryrefslogtreecommitdiffstats
path: root/chrome/test/chrome_process_util_win.cc
diff options
context:
space:
mode:
authorestade@chromium.org <estade@chromium.org@0039d316-1c4b-4281-b951-d872f2087c98>2009-04-09 22:44:57 +0000
committerestade@chromium.org <estade@chromium.org@0039d316-1c4b-4281-b951-d872f2087c98>2009-04-09 22:44:57 +0000
commit6aa9349ef5c1110d9c096945840e6a105e8a8295 (patch)
tree8afce0e1f3aae77e05105cfbe0d24ea3bdae9106 /chrome/test/chrome_process_util_win.cc
parentb61236c6679615af0811d59d130c91397d8c6be7 (diff)
downloadchromium_src-6aa9349ef5c1110d9c096945840e6a105e8a8295.zip
chromium_src-6aa9349ef5c1110d9c096945840e6a105e8a8295.tar.gz
chromium_src-6aa9349ef5c1110d9c096945840e6a105e8a8295.tar.bz2
Replace chrome_process_filter with chrome_process_util.
- move code only used by tests to chrome/test - make a better, more portable abstraction For now, it still only works on Windows. But this is the first step to porting this part of code. Patch by phajdan.jr@chromium.org: <http://codereview.chromium.org/54003> Review URL: http://codereview.chromium.org/67004 git-svn-id: svn://svn.chromium.org/chrome/trunk/src@13476 0039d316-1c4b-4281-b951-d872f2087c98
Diffstat (limited to 'chrome/test/chrome_process_util_win.cc')
-rw-r--r--chrome/test/chrome_process_util_win.cc63
1 files changed, 63 insertions, 0 deletions
diff --git a/chrome/test/chrome_process_util_win.cc b/chrome/test/chrome_process_util_win.cc
new file mode 100644
index 0000000..b0e2e40
--- /dev/null
+++ b/chrome/test/chrome_process_util_win.cc
@@ -0,0 +1,63 @@
+// Copyright (c) 2009 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/test/chrome_process_util.h"
+
+#include <windows.h>
+
+#include <vector>
+
+#include "base/logging.h"
+#include "base/process_util.h"
+#include "chrome/common/chrome_constants.h"
+
+namespace {
+
+class ChromeProcessFilter : public base::ProcessFilter {
+ public:
+ explicit ChromeProcessFilter(base::ProcessId browser_pid)
+ : browser_pid_(browser_pid) {}
+
+ virtual bool Includes(base::ProcessId pid, base::ProcessId parent_pid) const {
+ // Match browser process itself and its children.
+ return browser_pid_ == pid || browser_pid_ == parent_pid;
+ }
+
+ private:
+ base::ProcessId browser_pid_;
+
+ DISALLOW_COPY_AND_ASSIGN(ChromeProcessFilter);
+};
+
+} // namespace
+
+base::ProcessId ChromeBrowserProcessId(const FilePath& data_dir) {
+ HWND message_window = FindWindowEx(HWND_MESSAGE, NULL,
+ chrome::kMessageWindowClass,
+ data_dir.value().c_str());
+ if (!message_window)
+ return -1;
+
+ DWORD browser_pid = -1;
+ GetWindowThreadProcessId(message_window, &browser_pid);
+
+ return browser_pid;
+}
+
+ChromeProcessList GetRunningChromeProcesses(const FilePath& data_dir) {
+ ChromeProcessList result;
+
+ base::ProcessId browser_pid = ChromeBrowserProcessId(data_dir);
+ if (browser_pid < 0)
+ return result;
+
+ ChromeProcessFilter filter(browser_pid);
+ base::NamedProcessIterator it(chrome::kBrowserProcessExecutableName, &filter);
+
+ const ProcessEntry* process_entry;
+ while (process_entry = it.NextProcessEntry())
+ result.push_back(process_entry->th32ProcessID);
+
+ return result;
+}