OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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 "base/basictypes.h" | 5 #include "base/basictypes.h" |
6 #include "base/bind.h" | 6 #include "base/bind.h" |
7 #include "base/memory/ref_counted.h" | 7 #include "base/memory/ref_counted.h" |
8 #include "base/message_loop/message_loop.h" | 8 #include "base/message_loop/message_loop.h" |
9 #include "base/win/scoped_handle.h" | 9 #include "base/win/scoped_handle.h" |
10 #include "base/win/scoped_process_information.h" | 10 #include "base/win/scoped_process_information.h" |
(...skipping 244 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
255 task_runner_->PostTask( | 255 task_runner_->PostTask( |
256 FROM_HERE, | 256 FROM_HERE, |
257 base::Bind(&WorkerProcessLauncherTest::StopWorker, | 257 base::Bind(&WorkerProcessLauncherTest::StopWorker, |
258 base::Unretained(this))); | 258 base::Unretained(this))); |
259 } | 259 } |
260 | 260 |
261 void WorkerProcessLauncherTest::KillProcess() { | 261 void WorkerProcessLauncherTest::KillProcess() { |
262 event_handler_ = NULL; | 262 event_handler_ = NULL; |
263 | 263 |
264 if (worker_process_.IsValid()) { | 264 if (worker_process_.IsValid()) { |
265 TerminateProcess(worker_process_, CONTROL_C_EXIT); | 265 TerminateProcess(worker_process_.Get(), CONTROL_C_EXIT); |
266 worker_process_.Close(); | 266 worker_process_.Close(); |
267 } | 267 } |
268 } | 268 } |
269 | 269 |
270 void WorkerProcessLauncherTest::TerminateWorker(DWORD exit_code) { | 270 void WorkerProcessLauncherTest::TerminateWorker(DWORD exit_code) { |
271 if (worker_process_.IsValid()) | 271 if (worker_process_.IsValid()) |
272 TerminateProcess(worker_process_, exit_code); | 272 TerminateProcess(worker_process_.Get(), exit_code); |
273 } | 273 } |
274 | 274 |
275 void WorkerProcessLauncherTest::ConnectClient() { | 275 void WorkerProcessLauncherTest::ConnectClient() { |
276 channel_client_ = IPC::ChannelProxy::Create(IPC::ChannelHandle(channel_name_), | 276 channel_client_ = IPC::ChannelProxy::Create(IPC::ChannelHandle(channel_name_), |
277 IPC::Channel::MODE_CLIENT, | 277 IPC::Channel::MODE_CLIENT, |
278 &client_listener_, | 278 &client_listener_, |
279 task_runner_); | 279 task_runner_); |
280 | 280 |
281 // Pretend that |kLaunchSuccessTimeoutSeconds| passed since launching | 281 // Pretend that |kLaunchSuccessTimeoutSeconds| passed since launching |
282 // the worker process. This will make the backoff algorithm think that this | 282 // the worker process. This will make the backoff algorithm think that this |
(...skipping 72 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
355 base::win::ScopedProcessInformation process_information(temp_process_info); | 355 base::win::ScopedProcessInformation process_information(temp_process_info); |
356 worker_process_.Set(process_information.TakeProcessHandle()); | 356 worker_process_.Set(process_information.TakeProcessHandle()); |
357 ASSERT_TRUE(worker_process_.IsValid()); | 357 ASSERT_TRUE(worker_process_.IsValid()); |
358 | 358 |
359 channel_name_ = IPC::Channel::GenerateUniqueRandomChannelID(); | 359 channel_name_ = IPC::Channel::GenerateUniqueRandomChannelID(); |
360 ScopedHandle pipe; | 360 ScopedHandle pipe; |
361 ASSERT_TRUE(CreateIpcChannel(channel_name_, kIpcSecurityDescriptor, &pipe)); | 361 ASSERT_TRUE(CreateIpcChannel(channel_name_, kIpcSecurityDescriptor, &pipe)); |
362 | 362 |
363 // Wrap the pipe into an IPC channel. | 363 // Wrap the pipe into an IPC channel. |
364 channel_server_ = IPC::ChannelProxy::Create( | 364 channel_server_ = IPC::ChannelProxy::Create( |
365 IPC::ChannelHandle(pipe), IPC::Channel::MODE_SERVER, this, task_runner_); | 365 IPC::ChannelHandle(pipe.Get()), IPC::Channel::MODE_SERVER, this, |
| 366 task_runner_); |
366 | 367 |
367 HANDLE temp_handle; | 368 HANDLE temp_handle; |
368 ASSERT_TRUE(DuplicateHandle(GetCurrentProcess(), | 369 ASSERT_TRUE(DuplicateHandle(GetCurrentProcess(), |
369 worker_process_, | 370 worker_process_.Get(), |
370 GetCurrentProcess(), | 371 GetCurrentProcess(), |
371 &temp_handle, | 372 &temp_handle, |
372 0, | 373 0, |
373 FALSE, | 374 FALSE, |
374 DUPLICATE_SAME_ACCESS)); | 375 DUPLICATE_SAME_ACCESS)); |
375 ScopedHandle copy(temp_handle); | 376 ScopedHandle copy(temp_handle); |
376 | 377 |
377 event_handler_->OnProcessLaunched(copy.Pass()); | 378 event_handler_->OnProcessLaunched(copy.Pass()); |
378 } | 379 } |
379 | 380 |
(...skipping 142 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
522 EXPECT_CALL(client_listener_, OnCrash(_, _, _)) | 523 EXPECT_CALL(client_listener_, OnCrash(_, _, _)) |
523 .Times(1) | 524 .Times(1) |
524 .WillOnce(InvokeWithoutArgs( | 525 .WillOnce(InvokeWithoutArgs( |
525 this, &WorkerProcessLauncherTest::SendFakeMessageToLauncher)); | 526 this, &WorkerProcessLauncherTest::SendFakeMessageToLauncher)); |
526 | 527 |
527 StartWorker(); | 528 StartWorker(); |
528 message_loop_.Run(); | 529 message_loop_.Run(); |
529 } | 530 } |
530 | 531 |
531 } // namespace remoting | 532 } // namespace remoting |
OLD | NEW |