| 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 325 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 336 EXPECT_TRUE(event_handler_); | 336 EXPECT_TRUE(event_handler_); |
| 337 EXPECT_FALSE(worker_process_.IsValid()); | 337 EXPECT_FALSE(worker_process_.IsValid()); |
| 338 | 338 |
| 339 WCHAR notepad[MAX_PATH + 1]; | 339 WCHAR notepad[MAX_PATH + 1]; |
| 340 ASSERT_GT(ExpandEnvironmentStrings( | 340 ASSERT_GT(ExpandEnvironmentStrings( |
| 341 L"\045SystemRoot\045\\system32\\notepad.exe", notepad, MAX_PATH), 0u); | 341 L"\045SystemRoot\045\\system32\\notepad.exe", notepad, MAX_PATH), 0u); |
| 342 | 342 |
| 343 STARTUPINFOW startup_info = { 0 }; | 343 STARTUPINFOW startup_info = { 0 }; |
| 344 startup_info.cb = sizeof(startup_info); | 344 startup_info.cb = sizeof(startup_info); |
| 345 | 345 |
| 346 PROCESS_INFORMATION temp_process_info = {}; | 346 base::win::ScopedProcessInformation process_information; |
| 347 ASSERT_TRUE(CreateProcess(NULL, | 347 ASSERT_TRUE(CreateProcess(NULL, |
| 348 notepad, | 348 notepad, |
| 349 NULL, // default process attibutes | 349 NULL, // default process attibutes |
| 350 NULL, // default thread attibutes | 350 NULL, // default thread attibutes |
| 351 FALSE, // do not inherit handles | 351 FALSE, // do not inherit handles |
| 352 CREATE_SUSPENDED, | 352 CREATE_SUSPENDED, |
| 353 NULL, // no environment | 353 NULL, // no environment |
| 354 NULL, // default current directory | 354 NULL, // default current directory |
| 355 &startup_info, | 355 &startup_info, |
| 356 &temp_process_info)); | 356 process_information.Receive())); |
| 357 base::win::ScopedProcessInformation process_information(temp_process_info); | |
| 358 worker_process_.Set(process_information.TakeProcessHandle()); | 357 worker_process_.Set(process_information.TakeProcessHandle()); |
| 359 ASSERT_TRUE(worker_process_.IsValid()); | 358 ASSERT_TRUE(worker_process_.IsValid()); |
| 360 | 359 |
| 361 channel_name_ = IPC::Channel::GenerateUniqueRandomChannelID(); | 360 channel_name_ = IPC::Channel::GenerateUniqueRandomChannelID(); |
| 362 ScopedHandle pipe; | 361 ScopedHandle pipe; |
| 363 ASSERT_TRUE(CreateIpcChannel(channel_name_, kIpcSecurityDescriptor, &pipe)); | 362 ASSERT_TRUE(CreateIpcChannel(channel_name_, kIpcSecurityDescriptor, &pipe)); |
| 364 | 363 |
| 365 // Wrap the pipe into an IPC channel. | 364 // Wrap the pipe into an IPC channel. |
| 366 channel_server_.reset(new IPC::ChannelProxy( | 365 channel_server_.reset(new IPC::ChannelProxy( |
| 367 IPC::ChannelHandle(pipe), | 366 IPC::ChannelHandle(pipe), |
| 368 IPC::Channel::MODE_SERVER, | 367 IPC::Channel::MODE_SERVER, |
| 369 this, | 368 this, |
| 370 task_runner_)); | 369 task_runner_)); |
| 371 | 370 |
| 372 HANDLE temp_handle; | 371 ScopedHandle copy; |
| 373 ASSERT_TRUE(DuplicateHandle(GetCurrentProcess(), | 372 ASSERT_TRUE(DuplicateHandle(GetCurrentProcess(), |
| 374 worker_process_, | 373 worker_process_, |
| 375 GetCurrentProcess(), | 374 GetCurrentProcess(), |
| 376 &temp_handle, | 375 copy.Receive(), |
| 377 0, | 376 0, |
| 378 FALSE, | 377 FALSE, |
| 379 DUPLICATE_SAME_ACCESS)); | 378 DUPLICATE_SAME_ACCESS)); |
| 380 ScopedHandle copy(temp_handle); | |
| 381 | 379 |
| 382 event_handler_->OnProcessLaunched(copy.Pass()); | 380 event_handler_->OnProcessLaunched(copy.Pass()); |
| 383 } | 381 } |
| 384 | 382 |
| 385 TEST_F(WorkerProcessLauncherTest, Start) { | 383 TEST_F(WorkerProcessLauncherTest, Start) { |
| 386 EXPECT_CALL(*launcher_delegate_, LaunchProcess(_)) | 384 EXPECT_CALL(*launcher_delegate_, LaunchProcess(_)) |
| 387 .Times(1) | 385 .Times(1) |
| 388 .WillRepeatedly(Invoke(this, &WorkerProcessLauncherTest::LaunchProcess)); | 386 .WillRepeatedly(Invoke(this, &WorkerProcessLauncherTest::LaunchProcess)); |
| 389 | 387 |
| 390 EXPECT_CALL(server_listener_, OnChannelConnected(_)) | 388 EXPECT_CALL(server_listener_, OnChannelConnected(_)) |
| (...skipping 136 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 527 EXPECT_CALL(client_listener_, OnCrash(_, _, _)) | 525 EXPECT_CALL(client_listener_, OnCrash(_, _, _)) |
| 528 .Times(1) | 526 .Times(1) |
| 529 .WillOnce(InvokeWithoutArgs( | 527 .WillOnce(InvokeWithoutArgs( |
| 530 this, &WorkerProcessLauncherTest::SendFakeMessageToLauncher)); | 528 this, &WorkerProcessLauncherTest::SendFakeMessageToLauncher)); |
| 531 | 529 |
| 532 StartWorker(); | 530 StartWorker(); |
| 533 message_loop_.Run(); | 531 message_loop_.Run(); |
| 534 } | 532 } |
| 535 | 533 |
| 536 } // namespace remoting | 534 } // namespace remoting |
| OLD | NEW |