1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
|
// 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 <string>
#include <vector>
#include "base/command_line.h"
#include "base/process_util.h"
#include "base/string_util.h"
// Yes, this is impossibly lame. This horrible hack is Good Enough, though,
// because it's not used in production code, but just for testing.
//
// We could make this better by creating a system through which all instances of
// Chromium can communicate. ProcessSingleton does that for Windows and Linux,
// but the Mac doesn't implement it as its system services handle it. It's not
// worth implementing just for this.
//
// We could do something similar to what Linux does, and use |fuser| to find a
// file that the app ordinarily opens within the data dir. However, fuser is
// broken on Leopard, and does not detect files that lsof shows are open.
//
// What's going on here is that during ui_tests, the Chromium application is
// launched using the --user-data-dir command line option. By examining the
// output of ps, we can find the appropriately-launched Chromium process. Note
// that this _does_ work for paths with spaces. The command line that ps gives
// is just the argv separated with spaces. There's no escaping spaces as a shell
// would do, so a straight string comparison will work just fine.
//
// TODO(avi):see if there is a better way
base::ProcessId ChromeBrowserProcessId(const FilePath& data_dir) {
std::vector<std::string> argv;
argv.push_back("ps");
argv.push_back("-xw");
std::string ps_output;
if (!base::GetAppOutput(CommandLine(argv), &ps_output))
return -1;
std::vector<std::string> lines;
SplitString(ps_output, '\n', &lines);
for (size_t i=0; i<lines.size(); ++i) {
const std::string& line = lines[i];
if (line.find(data_dir.value()) != std::string::npos &&
line.find("type=renderer") == std::string::npos) {
int pid = StringToInt(line); // pid is at beginning of line
return pid==0 ? -1 : pid;
}
}
return -1;
}
|