[mojo-edk] Bind a child token to child launches and port reservations.

mojo/edk/embedder/embedder_unittest.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 <stddef.h>
 #include <stdint.h>
 #include <string.h>
 
 #include <utility>
 
 #include "base/bind.h"
 #include "base/command_line.h"
 #include "base/files/file.h"
 #include "base/logging.h"
 #include "base/macros.h"
 #include "base/memory/shared_memory.h"
 #include "base/message_loop/message_loop.h"
+#include "base/process/process_handle.h"
 #include "base/synchronization/waitable_event.h"
 #include "base/test/test_timeouts.h"
 #include "mojo/edk/embedder/platform_channel_pair.h"
 #include "mojo/edk/embedder/test_embedder.h"
 #include "mojo/edk/system/test_utils.h"
 #include "mojo/edk/test/mojo_test_base.h"
 #include "mojo/public/c/system/core.h"
 #include "mojo/public/cpp/system/handle.h"
 #include "mojo/public/cpp/system/message_pipe.h"
 #include "testing/gtest/include/gtest/gtest.h"
@@ skipping 144 matching lines @@
 
   // Wait for |h0| to become readable and read a message from it.
   EXPECT_EQ(kBarBaz, ReadMessage(h0));
 
   ASSERT_EQ(MOJO_RESULT_OK, MojoClose(server_mp));
   ASSERT_EQ(MOJO_RESULT_OK, MojoClose(client_mp));
   ASSERT_EQ(MOJO_RESULT_OK, MojoClose(h0));
   ASSERT_EQ(MOJO_RESULT_OK, MojoClose(h1));
 }
 
+TEST_F(EmbedderTest, PipeSetup) {
+  std::string child_token = GenerateRandomToken();
+  std::string pipe_token = GenerateRandomToken();
+
+  ScopedMessagePipeHandle parent_mp =
+      CreateParentMessagePipe(pipe_token, child_token);
+  ScopedMessagePipeHandle child_mp =
+      CreateChildMessagePipe(pipe_token);
+
+  const std::string kHello = "hello";
+  WriteMessage(parent_mp.get().value(), kHello);
+
+  EXPECT_EQ(kHello, ReadMessage(child_mp.get().value()));
+}
+
+TEST_F(EmbedderTest, PipeSetup_LaunchDeath) {
+  PlatformChannelPair pair;
+
+  std::string child_token = GenerateRandomToken();
+  std::string pipe_token = GenerateRandomToken();
+
+  ScopedMessagePipeHandle parent_mp =
+      CreateParentMessagePipe(pipe_token, child_token);
+  ChildProcessLaunched(base::GetCurrentProcessHandle(), pair.PassServerHandle(),
+                       child_token);
+
+  // Close the remote end, simulating child death before the child connects to
+  // the reserved port.
+  ignore_result(pair.PassClientHandle());
+
+  EXPECT_EQ(MOJO_RESULT_OK, MojoWait(parent_mp.get().value(),
+                                     MOJO_HANDLE_SIGNAL_PEER_CLOSED,
+                                     MOJO_DEADLINE_INDEFINITE,
+                                     nullptr));
+}
+
+TEST_F(EmbedderTest, PipeSetup_LaunchFailure) {
+  PlatformChannelPair pair;
+
+  std::string child_token = GenerateRandomToken();
+  std::string pipe_token = GenerateRandomToken();
+
+  ScopedMessagePipeHandle parent_mp =
+      CreateParentMessagePipe(pipe_token, child_token);
+
+  ChildProcessLaunchFailed(child_token);
+  EXPECT_EQ(MOJO_RESULT_OK, MojoWait(parent_mp.get().value(),
+                                     MOJO_HANDLE_SIGNAL_PEER_CLOSED,
+                                     MOJO_DEADLINE_INDEFINITE,
+                                     nullptr));
+}
+
 // The sequence of messages sent is:
 //       server_mp   client_mp   mp0         mp1         mp2         mp3
 //   1.  "hello"
 //   2.              "world!"
 //   3.                          "FOO"
 //   4.  "Bar"+mp1
 //   5.  (close)
 //   6.              (close)
 //   7.                                                              "baz"
 //   8.                                                              (closed)
@@ skipping 306 matching lines @@
 #endif  // defined(OS_MACOSX) && !defined(OS_IOS)
 
 // TODO(vtl): Test immediate write & close.
 // TODO(vtl): Test broken-connection cases.
 
 #endif  // !defined(OS_IOS)
 
 }  // namespace
 }  // namespace edk
 }  // namespace mojo