Create a broker interface for the new Mojo EDK so that the browser can create and duplicate messa...

mojo/edk/embedder/embedder.cc

 // Copyright 2014 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "mojo/edk/embedder/embedder.h"
 
 #include "base/atomicops.h"
 #include "base/bind.h"
 #include "base/bind_helpers.h"
 #include "base/location.h"
 #include "base/logging.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/message_loop/message_loop.h"
 #include "base/task_runner.h"
 #include "mojo/edk/embedder/embedder_internal.h"
+#include "mojo/edk/embedder/platform_channel_pair.h"
 #include "mojo/edk/embedder/process_delegate.h"
 #include "mojo/edk/embedder/simple_platform_support.h"
 #include "mojo/edk/system/configuration.h"
 #include "mojo/edk/system/core.h"
 #include "mojo/edk/system/message_pipe_dispatcher.h"
 #include "mojo/edk/system/platform_handle_dispatcher.h"
 
+#if defined(OS_WIN)
+#include "mojo/edk/system/child_token_serializer_win.h"
+#include "mojo/edk/system/parent_token_serializer_state_win.h"
+#include "mojo/edk/system/parent_token_serializer_win.h"
+#include "mojo/edk/system/simple_token_serializer_win.h"
+#endif
+
 namespace mojo {
 namespace edk {
 
 // TODO(jam): move into annonymous namespace. Keep outside for debugging in VS
 // temporarily.
 int g_channel_count = 0;
 bool g_wait_for_no_more_channels = false;
 base::TaskRunner* g_delegate_task_runner = nullptr;  // Used at shutdown.
 
 namespace {
 
 // Note: Called on the I/O thread.
 void ShutdownIPCSupportHelper(bool wait_for_no_more_channels) {
   if (wait_for_no_more_channels && g_channel_count) {
     g_wait_for_no_more_channels = true;
     return;
   }
 
   g_delegate_task_runner->PostTask(
       FROM_HERE, base::Bind(&ProcessDelegate::OnShutdownComplete,
                             base::Unretained(internal::g_process_delegate)));
   g_delegate_task_runner = nullptr;
 }
 
 }  // namespace
 
 namespace internal {
 
 // Declared in embedder_internal.h.
+#if defined(OS_WIN)
+TokenSerializer* g_token_serializer = nullptr;
+#endif
 PlatformSupport* g_platform_support = nullptr;
 Core* g_core = nullptr;
 
 ProcessDelegate* g_process_delegate;
 base::TaskRunner* g_io_thread_task_runner = nullptr;
 
 Core* GetCore() {
   return g_core;
 }
 
@@ skipping 13 matching lines @@
     ShutdownIPCSupportHelper(false);
   }
 }
 
 }  // namespace internal
 
 void SetMaxMessageSize(size_t bytes) {
   GetMutableConfiguration()->max_message_num_bytes = bytes;
 }
 
+#if defined(OS_WIN)
+void PreInitializeParentProcess() {
+  ParentTokenSerializerState::GetInstance();
+}
+
+void PreInitializeChildProcess() {
+  ChildTokenSerializer::GetInstance();
+}
+
+HANDLE ChildProcessLaunched(HANDLE child_process) {
+  PlatformChannelPair token_channel;
+  new ParentTokenSerializer(child_process, token_channel.PassServerHandle());
+  return token_channel.PassClientHandle().release().handle;
+}
+
+void ChildProcessLaunched(HANDLE child_process, HANDLE server_pipe) {
+  new ParentTokenSerializer(
+      child_process, ScopedPlatformHandle(PlatformHandle(server_pipe)));
+}
+
+void SetParentPipeHandle(HANDLE pipe) {
+  ScopedPlatformHandle handle;
+  handle.reset(PlatformHandle(pipe));
+  ChildTokenSerializer::GetInstance()->
+      SetParentTokenSerializerHandle(handle.Pass());
+}
+#endif
+
 void Init() {
+#if defined(OS_WIN)
+  if (!internal::g_token_serializer)
+    internal::g_token_serializer = new SimpleTokenSerializer;
+#endif
+
   DCHECK(!internal::g_platform_support);
   internal::g_platform_support = new SimplePlatformSupport();
 
   DCHECK(!internal::g_core);
   internal::g_core = new Core(internal::g_platform_support);
 }
 
 MojoResult AsyncWait(MojoHandle handle,
                      MojoHandleSignals signals,
                      const base::Callback<void(MojoResult)>& callback) {
@@ skipping 74 matching lines @@
   CHECK(rv.is_valid());
   dispatcher->Init(platform_handle.Pass(), nullptr, 0, nullptr, 0, nullptr,
                    nullptr);
   // TODO(vtl): The |.Pass()| below is only needed due to an MSVS bug; remove it
   // once that's fixed.
   return rv.Pass();
 }
 
 }  // namespace edk
 }  // namespace mojo