| 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 "services/service_manager/tests/util.h" | 5 #include "services/service_manager/tests/util.h" | 
| 6 | 6 | 
| 7 #include "base/base_paths.h" | 7 #include "base/base_paths.h" | 
| 8 #include "base/base_switches.h" | 8 #include "base/base_switches.h" | 
| 9 #include "base/bind.h" | 9 #include "base/bind.h" | 
| 10 #include "base/command_line.h" | 10 #include "base/command_line.h" | 
| 11 #include "base/files/file_path.h" | 11 #include "base/files/file_path.h" | 
| 12 #include "base/macros.h" | 12 #include "base/macros.h" | 
| 13 #include "base/message_loop/message_loop.h" | 13 #include "base/message_loop/message_loop.h" | 
| 14 #include "base/path_service.h" | 14 #include "base/path_service.h" | 
| 15 #include "base/process/process.h" | 15 #include "base/process/process.h" | 
| 16 #include "base/run_loop.h" | 16 #include "base/run_loop.h" | 
| 17 #include "mojo/edk/embedder/embedder.h" | 17 #include "mojo/edk/embedder/embedder.h" | 
| 18 #include "mojo/edk/embedder/platform_channel_pair.h" | 18 #include "mojo/edk/embedder/platform_channel_pair.h" | 
| 19 #include "mojo/edk/embedder/scoped_platform_handle.h" | 19 #include "mojo/edk/embedder/scoped_platform_handle.h" | 
| 20 #include "services/service_manager/public/cpp/connection.h" | 20 #include "services/service_manager/public/cpp/connection.h" | 
| 21 #include "services/service_manager/public/cpp/connector.h" | 21 #include "services/service_manager/public/cpp/connector.h" | 
| 22 #include "services/service_manager/public/interfaces/connector.mojom.h" | 22 #include "services/service_manager/public/interfaces/connector.mojom.h" | 
| 23 #include "services/service_manager/public/interfaces/service_factory.mojom.h" | 23 #include "services/service_manager/public/interfaces/service_factory.mojom.h" | 
| 24 #include "services/service_manager/runner/common/switches.h" | 24 #include "services/service_manager/runner/common/switches.h" | 
| 25 | 25 | 
| 26 namespace shell { | 26 namespace service_manager { | 
| 27 namespace test { | 27 namespace test { | 
| 28 | 28 | 
| 29 namespace { | 29 namespace { | 
| 30 | 30 | 
| 31 void QuitLoop(base::RunLoop* loop) { | 31 void QuitLoop(base::RunLoop* loop) { | 
| 32   loop->Quit(); | 32   loop->Quit(); | 
| 33 } | 33 } | 
| 34 | 34 | 
| 35 }  // namespace | 35 }  // namespace | 
| 36 | 36 | 
| 37 std::unique_ptr<Connection> LaunchAndConnectToProcess( | 37 std::unique_ptr<Connection> LaunchAndConnectToProcess( | 
| 38     const std::string& target_exe_name, | 38     const std::string& target_exe_name, | 
| 39     const Identity target, | 39     const Identity target, | 
| 40     shell::Connector* connector, | 40     service_manager::Connector* connector, | 
| 41     base::Process* process) { | 41     base::Process* process) { | 
| 42   base::FilePath target_path; | 42   base::FilePath target_path; | 
| 43   CHECK(base::PathService::Get(base::DIR_EXE, &target_path)); | 43   CHECK(base::PathService::Get(base::DIR_EXE, &target_path)); | 
| 44   target_path = target_path.AppendASCII(target_exe_name); | 44   target_path = target_path.AppendASCII(target_exe_name); | 
| 45 | 45 | 
| 46   base::CommandLine child_command_line(target_path); | 46   base::CommandLine child_command_line(target_path); | 
| 47   // Forward the wait-for-debugger flag but nothing else - we don't want to | 47   // Forward the wait-for-debugger flag but nothing else - we don't want to | 
| 48   // stamp on the platform-channel flag. | 48   // stamp on the platform-channel flag. | 
| 49   if (base::CommandLine::ForCurrentProcess()->HasSwitch( | 49   if (base::CommandLine::ForCurrentProcess()->HasSwitch( | 
| 50           switches::kWaitForDebugger)) { | 50           switches::kWaitForDebugger)) { | 
| (...skipping 11 matching lines...) Expand all  Loading... | 
| 62   // and pass that as well. | 62   // and pass that as well. | 
| 63   std::string primordial_pipe_token = mojo::edk::GenerateRandomToken(); | 63   std::string primordial_pipe_token = mojo::edk::GenerateRandomToken(); | 
| 64   child_command_line.AppendSwitchASCII(switches::kPrimordialPipeToken, | 64   child_command_line.AppendSwitchASCII(switches::kPrimordialPipeToken, | 
| 65                                         primordial_pipe_token); | 65                                         primordial_pipe_token); | 
| 66 | 66 | 
| 67   // Allocate the pipe locally. | 67   // Allocate the pipe locally. | 
| 68   std::string child_token = mojo::edk::GenerateRandomToken(); | 68   std::string child_token = mojo::edk::GenerateRandomToken(); | 
| 69   mojo::ScopedMessagePipeHandle pipe = | 69   mojo::ScopedMessagePipeHandle pipe = | 
| 70       mojo::edk::CreateParentMessagePipe(primordial_pipe_token, child_token); | 70       mojo::edk::CreateParentMessagePipe(primordial_pipe_token, child_token); | 
| 71 | 71 | 
| 72   shell::mojom::ServicePtr client; | 72   service_manager::mojom::ServicePtr client; | 
| 73   client.Bind( | 73   client.Bind(mojo::InterfacePtrInfo<service_manager::mojom::Service>( | 
| 74       mojo::InterfacePtrInfo<shell::mojom::Service>(std::move(pipe), 0u)); | 74       std::move(pipe), 0u)); | 
| 75   shell::mojom::PIDReceiverPtr receiver; | 75   service_manager::mojom::PIDReceiverPtr receiver; | 
| 76 | 76 | 
| 77   shell::Connector::ConnectParams params(target); | 77   service_manager::Connector::ConnectParams params(target); | 
| 78   params.set_client_process_connection(std::move(client), GetProxy(&receiver)); | 78   params.set_client_process_connection(std::move(client), GetProxy(&receiver)); | 
| 79   std::unique_ptr<shell::Connection> connection = connector->Connect(¶ms); | 79   std::unique_ptr<service_manager::Connection> connection = | 
|  | 80       connector->Connect(¶ms); | 
| 80   { | 81   { | 
| 81     base::RunLoop loop; | 82     base::RunLoop loop; | 
| 82     connection->AddConnectionCompletedClosure(base::Bind(&QuitLoop, &loop)); | 83     connection->AddConnectionCompletedClosure(base::Bind(&QuitLoop, &loop)); | 
| 83     base::MessageLoop::ScopedNestableTaskAllower allow( | 84     base::MessageLoop::ScopedNestableTaskAllower allow( | 
| 84         base::MessageLoop::current()); | 85         base::MessageLoop::current()); | 
| 85     loop.Run(); | 86     loop.Run(); | 
| 86   } | 87   } | 
| 87 | 88 | 
| 88   base::LaunchOptions options; | 89   base::LaunchOptions options; | 
| 89 #if defined(OS_WIN) | 90 #if defined(OS_WIN) | 
| 90   options.handles_to_inherit = &handle_passing_info; | 91   options.handles_to_inherit = &handle_passing_info; | 
| 91 #elif defined(OS_POSIX) | 92 #elif defined(OS_POSIX) | 
| 92   options.fds_to_remap = &handle_passing_info; | 93   options.fds_to_remap = &handle_passing_info; | 
| 93 #endif | 94 #endif | 
| 94   *process = base::LaunchProcess(child_command_line, options); | 95   *process = base::LaunchProcess(child_command_line, options); | 
| 95   DCHECK(process->IsValid()); | 96   DCHECK(process->IsValid()); | 
| 96   receiver->SetPID(process->Pid()); | 97   receiver->SetPID(process->Pid()); | 
| 97   mojo::edk::ChildProcessLaunched(process->Handle(), | 98   mojo::edk::ChildProcessLaunched(process->Handle(), | 
| 98                                   platform_channel_pair.PassServerHandle(), | 99                                   platform_channel_pair.PassServerHandle(), | 
| 99                                   child_token); | 100                                   child_token); | 
| 100   return connection; | 101   return connection; | 
| 101 } | 102 } | 
| 102 | 103 | 
| 103 }  // namespace test | 104 }  // namespace test | 
| 104 }  // namespace shell | 105 }  // namespace service_manager | 
| OLD | NEW | 
|---|