OLD | NEW |
1 // Copyright 2016 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 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 | 4 |
5 #include "chrome/test/base/mojo_test_connector.h" | 5 #include "chrome/test/base/mojo_test_connector.h" |
6 | 6 |
7 #include "base/bind.h" | 7 #include "base/bind.h" |
8 #include "base/callback.h" | 8 #include "base/callback.h" |
9 #include "base/command_line.h" | 9 #include "base/command_line.h" |
10 #include "base/memory/ptr_util.h" | 10 #include "base/memory/ptr_util.h" |
(...skipping 11 matching lines...) Expand all Loading... |
22 #include "services/shell/background/tests/test_catalog_store.h" | 22 #include "services/shell/background/tests/test_catalog_store.h" |
23 #include "services/shell/native_runner_delegate.h" | 23 #include "services/shell/native_runner_delegate.h" |
24 #include "services/shell/public/cpp/connector.h" | 24 #include "services/shell/public/cpp/connector.h" |
25 #include "services/shell/public/cpp/shell_client.h" | 25 #include "services/shell/public/cpp/shell_client.h" |
26 #include "services/shell/public/cpp/shell_connection.h" | 26 #include "services/shell/public/cpp/shell_connection.h" |
27 #include "services/shell/runner/common/client_util.h" | 27 #include "services/shell/runner/common/client_util.h" |
28 #include "services/shell/runner/common/switches.h" | 28 #include "services/shell/runner/common/switches.h" |
29 #include "services/shell/shell.h" | 29 #include "services/shell/shell.h" |
30 #include "services/shell/switches.h" | 30 #include "services/shell/switches.h" |
31 | 31 |
32 using mojo::shell::mojom::ShellClient; | 32 using shell::mojom::ShellClient; |
33 using mojo::shell::mojom::ShellClientPtr; | 33 using shell::mojom::ShellClientPtr; |
34 | 34 |
35 namespace { | 35 namespace { |
36 | 36 |
37 const char kTestRunnerName[] = "mojo:test-runner"; | 37 const char kTestRunnerName[] = "mojo:test-runner"; |
38 const char kTestName[] = "exe:chrome"; | 38 const char kTestName[] = "exe:chrome"; |
39 | 39 |
40 // Returns the Dictionary value of |parent| under the specified key, creating | 40 // Returns the Dictionary value of |parent| under the specified key, creating |
41 // and adding as necessary. | 41 // and adding as necessary. |
42 base::DictionaryValue* EnsureDictionary(base::DictionaryValue* parent, | 42 base::DictionaryValue* EnsureDictionary(base::DictionaryValue* parent, |
43 const char* key) { | 43 const char* key) { |
44 base::DictionaryValue* dictionary = nullptr; | 44 base::DictionaryValue* dictionary = nullptr; |
45 if (parent->GetDictionary(key, &dictionary)) | 45 if (parent->GetDictionary(key, &dictionary)) |
46 return dictionary; | 46 return dictionary; |
47 | 47 |
48 std::unique_ptr<base::DictionaryValue> owned_dictionary( | 48 std::unique_ptr<base::DictionaryValue> owned_dictionary( |
49 new base::DictionaryValue); | 49 new base::DictionaryValue); |
50 dictionary = owned_dictionary.get(); | 50 dictionary = owned_dictionary.get(); |
51 parent->Set(key, std::move(owned_dictionary)); | 51 parent->Set(key, std::move(owned_dictionary)); |
52 return dictionary; | 52 return dictionary; |
53 } | 53 } |
54 | 54 |
55 // This builds a permissive catalog with the addition of the 'instance_name' | 55 // This builds a permissive catalog with the addition of the 'instance_name' |
56 // permission. | 56 // permission. |
57 std::unique_ptr<mojo::shell::TestCatalogStore> BuildTestCatalogStore() { | 57 std::unique_ptr<shell::TestCatalogStore> BuildTestCatalogStore() { |
58 std::unique_ptr<base::ListValue> apps(new base::ListValue); | 58 std::unique_ptr<base::ListValue> apps(new base::ListValue); |
59 std::unique_ptr<base::DictionaryValue> test_app_config = | 59 std::unique_ptr<base::DictionaryValue> test_app_config = |
60 mojo::shell::BuildPermissiveSerializedAppInfo(kTestRunnerName, "test"); | 60 shell::BuildPermissiveSerializedAppInfo(kTestRunnerName, "test"); |
61 base::DictionaryValue* capabilities = | 61 base::DictionaryValue* capabilities = |
62 EnsureDictionary(test_app_config.get(), catalog::Store::kCapabilitiesKey); | 62 EnsureDictionary(test_app_config.get(), catalog::Store::kCapabilitiesKey); |
63 base::DictionaryValue* required_capabilities = | 63 base::DictionaryValue* required_capabilities = |
64 EnsureDictionary(capabilities, catalog::Store::kCapabilities_RequiredKey); | 64 EnsureDictionary(capabilities, catalog::Store::kCapabilities_RequiredKey); |
65 std::unique_ptr<base::ListValue> required_shell_classes(new base::ListValue); | 65 std::unique_ptr<base::ListValue> required_shell_classes(new base::ListValue); |
66 required_shell_classes->AppendString("instance_name"); | 66 required_shell_classes->AppendString("instance_name"); |
67 required_shell_classes->AppendString("client_process"); | 67 required_shell_classes->AppendString("client_process"); |
68 std::unique_ptr<base::DictionaryValue> shell_caps(new base::DictionaryValue); | 68 std::unique_ptr<base::DictionaryValue> shell_caps(new base::DictionaryValue); |
69 shell_caps->Set(catalog::Store::kCapabilities_ClassesKey, | 69 shell_caps->Set(catalog::Store::kCapabilities_ClassesKey, |
70 std::move(required_shell_classes)); | 70 std::move(required_shell_classes)); |
71 required_capabilities->Set("mojo:shell", std::move(shell_caps)); | 71 required_capabilities->Set("mojo:shell", std::move(shell_caps)); |
72 apps->Append(std::move(test_app_config)); | 72 apps->Append(std::move(test_app_config)); |
73 return base::WrapUnique(new mojo::shell::TestCatalogStore(std::move(apps))); | 73 return base::WrapUnique(new shell::TestCatalogStore(std::move(apps))); |
74 } | 74 } |
75 | 75 |
76 // BackgroundTestState maintains all the state necessary to bind the test to | 76 // BackgroundTestState maintains all the state necessary to bind the test to |
77 // mojo. This class is only used on the thread created by BackgroundShell. | 77 // mojo. This class is only used on the thread created by BackgroundShell. |
78 class BackgroundTestState { | 78 class BackgroundTestState { |
79 public: | 79 public: |
80 BackgroundTestState() {} | 80 BackgroundTestState() {} |
81 ~BackgroundTestState() {} | 81 ~BackgroundTestState() {} |
82 | 82 |
83 // Prepares the command line and other setup for connecting the test to mojo. | 83 // Prepares the command line and other setup for connecting the test to mojo. |
84 // Must be paired with a clal to ChildProcessLaunched(). | 84 // Must be paired with a clal to ChildProcessLaunched(). |
85 void Connect(base::CommandLine* command_line, | 85 void Connect(base::CommandLine* command_line, |
86 mojo::shell::Shell* shell, | 86 shell::Shell* shell, |
87 const std::string& instance, | 87 const std::string& instance, |
88 base::TestLauncher::LaunchOptions* test_launch_options) { | 88 base::TestLauncher::LaunchOptions* test_launch_options) { |
89 command_line->AppendSwitch(MojoTestConnector::kTestSwitch); | 89 command_line->AppendSwitch(MojoTestConnector::kTestSwitch); |
90 command_line->AppendSwitch(switches::kWaitForMojoShell); | 90 command_line->AppendSwitch(switches::kWaitForMojoShell); |
91 command_line->AppendSwitch(switches::kChildProcess); | 91 command_line->AppendSwitch(switches::kChildProcess); |
92 mojo_ipc_channel_.reset(new mojo::edk::PlatformChannelPair); | 92 mojo_ipc_channel_.reset(new mojo::edk::PlatformChannelPair); |
93 mojo_ipc_channel_->PrepareToPassClientHandleToChildProcess( | 93 mojo_ipc_channel_->PrepareToPassClientHandleToChildProcess( |
94 command_line, &handle_passing_info_); | 94 command_line, &handle_passing_info_); |
95 #if defined(OS_WIN) | 95 #if defined(OS_WIN) |
96 test_launch_options->inherit_handles = true; | 96 test_launch_options->inherit_handles = true; |
97 test_launch_options->handles_to_inherit = &handle_passing_info_; | 97 test_launch_options->handles_to_inherit = &handle_passing_info_; |
98 #if defined(OFFICIAL_BUILD) | 98 #if defined(OFFICIAL_BUILD) |
99 CHECK(false) << "Launching mojo process with inherit_handles is insecure!"; | 99 CHECK(false) << "Launching mojo process with inherit_handles is insecure!"; |
100 #endif | 100 #endif |
101 #elif defined(OS_POSIX) | 101 #elif defined(OS_POSIX) |
102 test_launch_options->fds_to_remap = &handle_passing_info_; | 102 test_launch_options->fds_to_remap = &handle_passing_info_; |
103 #else | 103 #else |
104 #error "Unsupported" | 104 #error "Unsupported" |
105 #endif | 105 #endif |
106 mojo::shell::mojom::ShellClientPtr client = | 106 shell::mojom::ShellClientPtr client = |
107 mojo::shell::PassShellClientRequestOnCommandLine(command_line); | 107 shell::PassShellClientRequestOnCommandLine(command_line); |
108 | 108 |
109 std::unique_ptr<mojo::shell::ConnectParams> params( | 109 std::unique_ptr<shell::ConnectParams> params(new shell::ConnectParams); |
110 new mojo::shell::ConnectParams); | 110 params->set_source(shell::CreateShellIdentity()); |
111 params->set_source(mojo::shell::CreateShellIdentity()); | |
112 params->set_target( | 111 params->set_target( |
113 mojo::Identity(kTestName, mojo::shell::mojom::kRootUserID, instance)); | 112 shell::Identity(kTestName, shell::mojom::kRootUserID, instance)); |
114 | 113 |
115 mojo::shell::mojom::ClientProcessConnectionPtr client_process_connection = | 114 shell::mojom::ClientProcessConnectionPtr client_process_connection = |
116 mojo::shell::mojom::ClientProcessConnection::New(); | 115 shell::mojom::ClientProcessConnection::New(); |
117 client_process_connection->shell_client = | 116 client_process_connection->shell_client = |
118 client.PassInterface().PassHandle(); | 117 client.PassInterface().PassHandle(); |
119 client_process_connection->pid_receiver_request = | 118 client_process_connection->pid_receiver_request = |
120 mojo::GetProxy(&pid_receiver_).PassMessagePipe(); | 119 mojo::GetProxy(&pid_receiver_).PassMessagePipe(); |
121 params->set_client_process_connection(std::move(client_process_connection)); | 120 params->set_client_process_connection(std::move(client_process_connection)); |
122 shell->Connect(std::move(params)); | 121 shell->Connect(std::move(params)); |
123 } | 122 } |
124 | 123 |
125 // Called after the test process has launched. Completes the registration done | 124 // Called after the test process has launched. Completes the registration done |
126 // in Connect(). | 125 // in Connect(). |
127 void ChildProcessLaunched(base::ProcessHandle handle, base::ProcessId pid) { | 126 void ChildProcessLaunched(base::ProcessHandle handle, base::ProcessId pid) { |
128 pid_receiver_->SetPID(pid); | 127 pid_receiver_->SetPID(pid); |
129 mojo_ipc_channel_->ChildProcessLaunched(); | 128 mojo_ipc_channel_->ChildProcessLaunched(); |
130 mojo::edk::ChildProcessLaunched( | 129 mojo::edk::ChildProcessLaunched( |
131 handle, mojo::edk::ScopedPlatformHandle(mojo::edk::PlatformHandle( | 130 handle, mojo::edk::ScopedPlatformHandle(mojo::edk::PlatformHandle( |
132 mojo_ipc_channel_->PassServerHandle().release().handle))); | 131 mojo_ipc_channel_->PassServerHandle().release().handle))); |
133 } | 132 } |
134 | 133 |
135 private: | 134 private: |
136 // Used to back the NodeChannel between the parent and child node. | 135 // Used to back the NodeChannel between the parent and child node. |
137 std::unique_ptr<mojo::edk::PlatformChannelPair> mojo_ipc_channel_; | 136 std::unique_ptr<mojo::edk::PlatformChannelPair> mojo_ipc_channel_; |
138 | 137 |
139 mojo::edk::HandlePassingInformation handle_passing_info_; | 138 mojo::edk::HandlePassingInformation handle_passing_info_; |
140 | 139 |
141 mojo::shell::mojom::PIDReceiverPtr pid_receiver_; | 140 shell::mojom::PIDReceiverPtr pid_receiver_; |
142 | 141 |
143 DISALLOW_COPY_AND_ASSIGN(BackgroundTestState); | 142 DISALLOW_COPY_AND_ASSIGN(BackgroundTestState); |
144 }; | 143 }; |
145 | 144 |
146 // Called used destroy BackgroundTestState on the background thread. | 145 // Called used destroy BackgroundTestState on the background thread. |
147 void DestroyBackgroundStateOnBackgroundThread( | 146 void DestroyBackgroundStateOnBackgroundThread( |
148 std::unique_ptr<BackgroundTestState> state, | 147 std::unique_ptr<BackgroundTestState> state, |
149 mojo::shell::Shell* shell) {} | 148 shell::Shell* shell) {} |
150 | 149 |
151 // State created per test. Manages creation of the corresponding | 150 // State created per test. Manages creation of the corresponding |
152 // BackgroundTestState and making sure processing runs on the right threads. | 151 // BackgroundTestState and making sure processing runs on the right threads. |
153 class MojoTestState : public content::TestState { | 152 class MojoTestState : public content::TestState { |
154 public: | 153 public: |
155 explicit MojoTestState(mojo::shell::BackgroundShell* background_shell) | 154 explicit MojoTestState(shell::BackgroundShell* background_shell) |
156 : background_shell_(background_shell) {} | 155 : background_shell_(background_shell) {} |
157 | 156 |
158 ~MojoTestState() override { | 157 ~MojoTestState() override { |
159 DCHECK(background_state_); | 158 DCHECK(background_state_); |
160 // BackgroundState needs to be destroyed on the background thread. We're | 159 // BackgroundState needs to be destroyed on the background thread. We're |
161 // guaranteed |background_shell_| has been created by the time we reach | 160 // guaranteed |background_shell_| has been created by the time we reach |
162 // here as Init() blocks until |background_shell_| has been created. | 161 // here as Init() blocks until |background_shell_| has been created. |
163 background_shell_->ExecuteOnShellThread( | 162 background_shell_->ExecuteOnShellThread( |
164 base::Bind(&DestroyBackgroundStateOnBackgroundThread, | 163 base::Bind(&DestroyBackgroundStateOnBackgroundThread, |
165 base::Passed(&background_state_))); | 164 base::Passed(&background_state_))); |
(...skipping 19 matching lines...) Expand all Loading... |
185 base::WaitableEvent signal(true, false); | 184 base::WaitableEvent signal(true, false); |
186 background_shell_->ExecuteOnShellThread( | 185 background_shell_->ExecuteOnShellThread( |
187 base::Bind(&MojoTestState::ChildProcessLaunchedOnBackgroundThread, | 186 base::Bind(&MojoTestState::ChildProcessLaunchedOnBackgroundThread, |
188 base::Unretained(this), handle, pid, &signal)); | 187 base::Unretained(this), handle, pid, &signal)); |
189 signal.Wait(); | 188 signal.Wait(); |
190 } | 189 } |
191 | 190 |
192 void ChildProcessLaunchedOnBackgroundThread(base::ProcessHandle handle, | 191 void ChildProcessLaunchedOnBackgroundThread(base::ProcessHandle handle, |
193 base::ProcessId pid, | 192 base::ProcessId pid, |
194 base::WaitableEvent* signal, | 193 base::WaitableEvent* signal, |
195 mojo::shell::Shell* shell) { | 194 shell::Shell* shell) { |
196 background_state_->ChildProcessLaunched(handle, pid); | 195 background_state_->ChildProcessLaunched(handle, pid); |
197 signal->Signal(); | 196 signal->Signal(); |
198 } | 197 } |
199 | 198 |
200 void BindOnBackgroundThread( | 199 void BindOnBackgroundThread( |
201 base::WaitableEvent* signal, | 200 base::WaitableEvent* signal, |
202 base::CommandLine* command_line, | 201 base::CommandLine* command_line, |
203 base::TestLauncher::LaunchOptions* test_launch_options, | 202 base::TestLauncher::LaunchOptions* test_launch_options, |
204 mojo::shell::Shell* shell) { | 203 shell::Shell* shell) { |
205 static int instance_id = 0; | 204 static int instance_id = 0; |
206 const std::string instance_name = | 205 const std::string instance_name = |
207 "instance-" + base::IntToString(instance_id++); | 206 "instance-" + base::IntToString(instance_id++); |
208 background_state_.reset(new BackgroundTestState); | 207 background_state_.reset(new BackgroundTestState); |
209 background_state_->Connect(command_line, shell, instance_name, | 208 background_state_->Connect(command_line, shell, instance_name, |
210 test_launch_options); | 209 test_launch_options); |
211 signal->Signal(); | 210 signal->Signal(); |
212 } | 211 } |
213 | 212 |
214 mojo::shell::BackgroundShell* background_shell_; | 213 shell::BackgroundShell* background_shell_; |
215 std::unique_ptr<BackgroundTestState> background_state_; | 214 std::unique_ptr<BackgroundTestState> background_state_; |
216 | 215 |
217 DISALLOW_COPY_AND_ASSIGN(MojoTestState); | 216 DISALLOW_COPY_AND_ASSIGN(MojoTestState); |
218 }; | 217 }; |
219 | 218 |
220 } // namespace | 219 } // namespace |
221 | 220 |
222 class MojoTestConnector::NativeRunnerDelegateImpl | 221 class MojoTestConnector::NativeRunnerDelegateImpl |
223 : public mojo::shell::NativeRunnerDelegate { | 222 : public shell::NativeRunnerDelegate { |
224 public: | 223 public: |
225 NativeRunnerDelegateImpl() {} | 224 NativeRunnerDelegateImpl() {} |
226 ~NativeRunnerDelegateImpl() override {} | 225 ~NativeRunnerDelegateImpl() override {} |
227 | 226 |
228 private: | 227 private: |
229 // mojo::shell::NativeRunnerDelegate: | 228 // shell::NativeRunnerDelegate: |
230 void AdjustCommandLineArgumentsForTarget( | 229 void AdjustCommandLineArgumentsForTarget( |
231 const mojo::Identity& target, | 230 const shell::Identity& target, |
232 base::CommandLine* command_line) override { | 231 base::CommandLine* command_line) override { |
233 if (target.name() == "exe:chrome") | 232 if (target.name() == "exe:chrome") |
234 command_line->AppendSwitch(switches::kWaitForMojoShell); | 233 command_line->AppendSwitch(switches::kWaitForMojoShell); |
235 } | 234 } |
236 | 235 |
237 DISALLOW_COPY_AND_ASSIGN(NativeRunnerDelegateImpl); | 236 DISALLOW_COPY_AND_ASSIGN(NativeRunnerDelegateImpl); |
238 }; | 237 }; |
239 | 238 |
240 // static | 239 // static |
241 const char MojoTestConnector::kTestSwitch[] = "is_test"; | 240 const char MojoTestConnector::kTestSwitch[] = "is_test"; |
242 | 241 |
243 MojoTestConnector::MojoTestConnector() {} | 242 MojoTestConnector::MojoTestConnector() {} |
244 | 243 |
245 mojo::shell::mojom::ShellClientRequest MojoTestConnector::Init() { | 244 shell::mojom::ShellClientRequest MojoTestConnector::Init() { |
246 native_runner_delegate_.reset(new NativeRunnerDelegateImpl); | 245 native_runner_delegate_.reset(new NativeRunnerDelegateImpl); |
247 std::unique_ptr<mojo::shell::BackgroundShell::InitParams> init_params( | 246 std::unique_ptr<shell::BackgroundShell::InitParams> init_params( |
248 new mojo::shell::BackgroundShell::InitParams); | 247 new shell::BackgroundShell::InitParams); |
249 init_params->catalog_store = BuildTestCatalogStore(); | 248 init_params->catalog_store = BuildTestCatalogStore(); |
250 // When running in single_process mode chrome initializes the edk. | 249 // When running in single_process mode chrome initializes the edk. |
251 init_params->init_edk = !base::CommandLine::ForCurrentProcess()->HasSwitch( | 250 init_params->init_edk = !base::CommandLine::ForCurrentProcess()->HasSwitch( |
252 content::kSingleProcessTestsFlag); | 251 content::kSingleProcessTestsFlag); |
253 init_params->native_runner_delegate = native_runner_delegate_.get(); | 252 init_params->native_runner_delegate = native_runner_delegate_.get(); |
254 background_shell_.Init(std::move(init_params)); | 253 background_shell_.Init(std::move(init_params)); |
255 return background_shell_.CreateShellClientRequest(kTestRunnerName); | 254 return background_shell_.CreateShellClientRequest(kTestRunnerName); |
256 } | 255 } |
257 | 256 |
258 MojoTestConnector::~MojoTestConnector() {} | 257 MojoTestConnector::~MojoTestConnector() {} |
259 | 258 |
260 std::unique_ptr<content::TestState> MojoTestConnector::PrepareForTest( | 259 std::unique_ptr<content::TestState> MojoTestConnector::PrepareForTest( |
261 base::CommandLine* command_line, | 260 base::CommandLine* command_line, |
262 base::TestLauncher::LaunchOptions* test_launch_options) { | 261 base::TestLauncher::LaunchOptions* test_launch_options) { |
263 std::unique_ptr<MojoTestState> test_state( | 262 std::unique_ptr<MojoTestState> test_state( |
264 new MojoTestState(&background_shell_)); | 263 new MojoTestState(&background_shell_)); |
265 test_state->Init(command_line, test_launch_options); | 264 test_state->Init(command_line, test_launch_options); |
266 return std::move(test_state); | 265 return std::move(test_state); |
267 } | 266 } |
OLD | NEW |