Index: components/nacl/broker/nacl_broker_listener.cc |
diff --git a/components/nacl/broker/nacl_broker_listener.cc b/components/nacl/broker/nacl_broker_listener.cc |
index f21e32f48db331f1539954d9c83e5185bb1ca1c5..0adbfe2fa9a642f71fb715d39600e68347eea91e 100644 |
--- a/components/nacl/broker/nacl_broker_listener.cc |
+++ b/components/nacl/broker/nacl_broker_listener.cc |
@@ -12,16 +12,23 @@ |
#include "base/process/launch.h" |
#include "base/process/process.h" |
#include "base/process/process_handle.h" |
+#include "base/strings/string_number_conversions.h" |
#include "base/threading/thread_task_runner_handle.h" |
+#include "base/win/win_util.h" |
#include "components/nacl/common/nacl_cmd_line.h" |
#include "components/nacl/common/nacl_debug_exception_handler_win.h" |
#include "components/nacl/common/nacl_messages.h" |
#include "components/nacl/common/nacl_switches.h" |
#include "content/public/common/content_switches.h" |
+#include "content/public/common/mojo_channel_switches.h" |
#include "content/public/common/sandbox_init.h" |
#include "ipc/attachment_broker_unprivileged.h" |
#include "ipc/ipc_channel.h" |
#include "ipc/ipc_switches.h" |
+#include "mojo/edk/embedder/embedder.h" |
+#include "mojo/edk/embedder/platform_channel_pair.h" |
+#include "mojo/edk/embedder/scoped_ipc_support.h" |
+#include "mojo/public/cpp/system/message_pipe.h" |
#include "sandbox/win/src/sandbox_policy.h" |
namespace { |
@@ -34,6 +41,14 @@ void SendReply(IPC::Channel* channel, int32_t pid, bool result) { |
NaClBrokerListener::NaClBrokerListener() { |
IPC::AttachmentBrokerUnprivileged::CreateBrokerIfNeeded(); |
+ mojo::edk::Init(); |
Mark Seaborn
2016/07/08 20:26:19
How about putting this global-init stuff in NaClBr
Anand Mistry (off Chromium)
2016/07/12 07:26:23
Done.
|
+ mojo_ipc_support_.reset(new mojo::edk::ScopedIPCSupport( |
Mark Seaborn
2016/07/08 20:26:19
mojo_ipc_support_ is never read, so I don't think
Anand Mistry (off Chromium)
2016/07/12 07:26:23
It needs to exist for the life of this object. But
|
+ base::MessageLoop::current()->task_runner())); |
+ mojo::edk::ScopedPlatformHandle platform_channel( |
+ mojo::edk::PlatformChannelPair::PassClientHandleFromParentProcess( |
+ *base::CommandLine::ForCurrentProcess())); |
+ DCHECK(platform_channel.is_valid()); |
+ mojo::edk::SetParentPipeHandle(std::move(platform_channel)); |
} |
NaClBrokerListener::~NaClBrokerListener() { |
@@ -43,10 +58,14 @@ NaClBrokerListener::~NaClBrokerListener() { |
} |
void NaClBrokerListener::Listen() { |
- std::string channel_name = |
- base::CommandLine::ForCurrentProcess()->GetSwitchValueASCII( |
- switches::kProcessChannelID); |
- channel_ = IPC::Channel::CreateClient(channel_name, this); |
+ mojo::ScopedMessagePipeHandle handle( |
+ mojo::edk::CreateChildMessagePipe( |
+ base::CommandLine::ForCurrentProcess()->GetSwitchValueASCII( |
+ switches::kMojoChannelToken))); |
+ DCHECK(handle.is_valid()); |
+ IPC::ChannelHandle channel_handle(handle.release()); |
+ |
+ channel_ = IPC::Channel::CreateClient(channel_handle, this); |
IPC::AttachmentBroker* broker = IPC::AttachmentBroker::GetGlobal(); |
if (broker && !broker->IsPrivilegedBroker()) |
broker->RegisterBrokerCommunicationChannel(channel_.get()); |
@@ -95,6 +114,10 @@ void NaClBrokerListener::OnLaunchLoaderThroughBroker( |
const std::string& loader_channel_id) { |
Mark Seaborn
2016/07/08 20:26:19
Should this be called "token" rather than "id" now
Anand Mistry (off Chromium)
2016/07/12 07:26:23
Done.
|
base::ProcessHandle loader_handle_in_browser = 0; |
+ mojo::ScopedMessagePipeHandle loader_handle( |
Mark Seaborn
2016/07/08 20:26:19
Nit: With this addition, there's "loader_handle_in
Anand Mistry (off Chromium)
2016/07/12 07:26:23
s/handle/message_pipe
|
+ mojo::edk::CreateChildMessagePipe(loader_channel_id)); |
+ DCHECK(loader_handle.is_valid()); |
+ |
// Create the path to the nacl broker/loader executable - it's the executable |
// this code is running in. |
base::FilePath exe_path; |
@@ -106,14 +129,35 @@ void NaClBrokerListener::OnLaunchLoaderThroughBroker( |
cmd_line->AppendSwitchASCII(switches::kProcessType, |
switches::kNaClLoaderProcess); |
- cmd_line->AppendSwitchASCII(switches::kProcessChannelID, |
- loader_channel_id); |
+ // Mojo IPC setup. |
Mark Seaborn
2016/07/08 20:26:19
This is a largish chunk of plumbing for creating a
Anand Mistry (off Chromium)
2016/07/12 07:26:23
Most Chrome code uses content::ChildProcessLaunche
|
+ mojo::edk::PlatformChannelPair channel_pair; |
+ mojo::edk::ScopedPlatformHandle parent_handle = |
+ channel_pair.PassServerHandle(); |
+ mojo::edk::ScopedPlatformHandle client_handle = |
+ channel_pair.PassClientHandle(); |
+ base::HandlesToInheritVector handles; |
+ handles.push_back(client_handle.get().handle); |
+ cmd_line->AppendSwitchASCII( |
+ mojo::edk::PlatformChannelPair::kMojoPlatformChannelHandleSwitch, |
+ base::UintToString(base::win::HandleToUint32(handles[0]))); |
+ const std::string mojo_child_token = mojo::edk::GenerateRandomToken(); |
+ const std::string mojo_channel_token = mojo::edk::GenerateRandomToken(); |
+ mojo::ScopedMessagePipeHandle host_handle = |
+ mojo::edk::CreateParentMessagePipe(mojo_channel_token, |
+ mojo_child_token); |
+ cmd_line->AppendSwitchASCII(switches::kMojoChannelToken, |
+ mojo_channel_token); |
+ CHECK_EQ(MOJO_RESULT_OK, mojo::FuseMessagePipes(std::move(loader_handle), |
Mark Seaborn
2016/07/08 20:26:19
Does this cause messages to be forwarded by the Na
Anand Mistry (off Chromium)
2016/07/12 07:26:23
Yes, Mojo does "path shortening". It may take a fe
|
+ std::move(host_handle))); |
base::Process loader_process; |
sandbox::ResultCode result = content::StartSandboxedProcess( |
- this, cmd_line, base::HandlesToInheritVector(), &loader_process); |
+ this, cmd_line, handles, &loader_process); |
if (result == sandbox::SBOX_ALL_OK) { |
+ mojo::edk::ChildProcessLaunched(loader_process.Handle(), |
+ std::move(parent_handle), |
+ mojo_child_token); |
// Note: PROCESS_DUP_HANDLE is necessary here, because: |
// 1) The current process is the broker, which is the loader's parent. |
// 2) The browser is not the loader's parent, and so only gets the |
@@ -127,6 +171,8 @@ void NaClBrokerListener::OnLaunchLoaderThroughBroker( |
browser_process_.Handle(), &loader_handle_in_browser, |
PROCESS_DUP_HANDLE | PROCESS_QUERY_INFORMATION | PROCESS_TERMINATE, |
FALSE, 0); |
+ } else { |
+ mojo::edk::ChildProcessLaunchFailed(mojo_child_token); |
} |
} |
channel_->Send(new NaClProcessMsg_LoaderLaunched(loader_channel_id, |
Mark Seaborn
2016/07/08 20:26:19
Is loader_channel_id "consumed" by the call to moj
Anand Mistry (off Chromium)
2016/07/12 07:26:23
Yes. Just like before.
|