| Index: chrome/test/chrome_process_util_win.cc
|
| diff --git a/chrome/test/chrome_process_util_win.cc b/chrome/test/chrome_process_util_win.cc
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..b0e2e4029603ffe1d2c037d14906114e9f4cf70f
|
| --- /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;
|
| +}
|
|
|