Index: mojo/edk/system/remote_message_pipe_unittest.cc |
diff --git a/mojo/edk/system/remote_message_pipe_unittest.cc b/mojo/edk/system/remote_message_pipe_unittest.cc |
deleted file mode 100644 |
index 1860c239e03e23663cfcb035fabe9adac3fcf5b8..0000000000000000000000000000000000000000 |
--- a/mojo/edk/system/remote_message_pipe_unittest.cc |
+++ /dev/null |
@@ -1,1376 +0,0 @@ |
-// 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 <stdint.h> |
-#include <stdio.h> |
-#include <string.h> |
- |
-#include <vector> |
- |
-#include "base/bind.h" |
-#include "base/files/file_path.h" |
-#include "base/files/file_util.h" |
-#include "base/files/scoped_file.h" |
-#include "base/files/scoped_temp_dir.h" |
-#include "base/location.h" |
-#include "base/logging.h" |
-#include "base/macros.h" |
-#include "base/message_loop/message_loop.h" |
-#include "base/test/test_io_thread.h" |
-#include "base/threading/platform_thread.h" // For |Sleep()|. |
-#include "build/build_config.h" // TODO(vtl): Remove this. |
-#include "mojo/edk/embedder/platform_channel_pair.h" |
-#include "mojo/edk/embedder/platform_shared_buffer.h" |
-#include "mojo/edk/embedder/scoped_platform_handle.h" |
-#include "mojo/edk/embedder/simple_platform_support.h" |
-#include "mojo/edk/system/channel.h" |
-#include "mojo/edk/system/channel_endpoint.h" |
-#include "mojo/edk/system/channel_endpoint_id.h" |
-#include "mojo/edk/system/incoming_endpoint.h" |
-#include "mojo/edk/system/message_pipe.h" |
-#include "mojo/edk/system/message_pipe_dispatcher.h" |
-#include "mojo/edk/system/platform_handle_dispatcher.h" |
-#include "mojo/edk/system/raw_channel.h" |
-#include "mojo/edk/system/shared_buffer_dispatcher.h" |
-#include "mojo/edk/system/test_utils.h" |
-#include "mojo/edk/system/waiter.h" |
-#include "mojo/edk/test/test_utils.h" |
-#include "testing/gtest/include/gtest/gtest.h" |
- |
-namespace mojo { |
-namespace system { |
-namespace { |
- |
-const MojoHandleSignals kAllSignals = MOJO_HANDLE_SIGNAL_READABLE | |
- MOJO_HANDLE_SIGNAL_WRITABLE | |
- MOJO_HANDLE_SIGNAL_PEER_CLOSED; |
- |
-class RemoteMessagePipeTest : public testing::Test { |
- public: |
- RemoteMessagePipeTest() : io_thread_(base::TestIOThread::kAutoStart) {} |
- ~RemoteMessagePipeTest() override {} |
- |
- void SetUp() override { |
- io_thread_.PostTaskAndWait( |
- FROM_HERE, base::Bind(&RemoteMessagePipeTest::SetUpOnIOThread, |
- base::Unretained(this))); |
- } |
- |
- void TearDown() override { |
- io_thread_.PostTaskAndWait( |
- FROM_HERE, base::Bind(&RemoteMessagePipeTest::TearDownOnIOThread, |
- base::Unretained(this))); |
- } |
- |
- protected: |
- // This connects the two given |ChannelEndpoint|s. It assumes/requires that |
- // this is the bootstrap case (i.e., no other message pipes have ever been |
- // hosted on the channel). |
- void BootstrapChannelEndpoints(scoped_refptr<ChannelEndpoint> ep0, |
- scoped_refptr<ChannelEndpoint> ep1) { |
- io_thread_.PostTaskAndWait( |
- FROM_HERE, |
- base::Bind(&RemoteMessagePipeTest::BootstrapChannelEndpointsOnIOThread, |
- base::Unretained(this), ep0, ep1)); |
- } |
- |
- // This bootstraps |ep| on |channels_[channel_index]|. It assumes/requires |
- // that this is the bootstrap case (i.e., no message pipes have ever been |
- // hosted on the channel). This returns *without* waiting. |
- void BootstrapChannelEndpointNoWait(unsigned channel_index, |
- scoped_refptr<ChannelEndpoint> ep) { |
- io_thread_.PostTask( |
- FROM_HERE, |
- base::Bind(&RemoteMessagePipeTest::BootstrapChannelEndpointOnIOThread, |
- base::Unretained(this), channel_index, ep)); |
- } |
- |
- void RestoreInitialState() { |
- io_thread_.PostTaskAndWait( |
- FROM_HERE, |
- base::Bind(&RemoteMessagePipeTest::RestoreInitialStateOnIOThread, |
- base::Unretained(this))); |
- } |
- |
- embedder::PlatformSupport* platform_support() { return &platform_support_; } |
- base::TestIOThread* io_thread() { return &io_thread_; } |
- // Warning: It's up to the caller to ensure that the returned channel |
- // is/remains valid. |
- Channel* channels(size_t i) { return channels_[i].get(); } |
- |
- private: |
- void SetUpOnIOThread() { |
- CHECK_EQ(base::MessageLoop::current(), io_thread()->message_loop()); |
- |
- embedder::PlatformChannelPair channel_pair; |
- platform_handles_[0] = channel_pair.PassServerHandle(); |
- platform_handles_[1] = channel_pair.PassClientHandle(); |
- } |
- |
- void TearDownOnIOThread() { |
- CHECK_EQ(base::MessageLoop::current(), io_thread()->message_loop()); |
- |
- if (channels_[0]) { |
- channels_[0]->Shutdown(); |
- channels_[0] = nullptr; |
- } |
- if (channels_[1]) { |
- channels_[1]->Shutdown(); |
- channels_[1] = nullptr; |
- } |
- } |
- |
- void CreateAndInitChannel(unsigned channel_index) { |
- CHECK_EQ(base::MessageLoop::current(), io_thread()->message_loop()); |
- CHECK(channel_index == 0 || channel_index == 1); |
- CHECK(!channels_[channel_index]); |
- |
- channels_[channel_index] = new Channel(&platform_support_); |
- channels_[channel_index]->Init( |
- RawChannel::Create(platform_handles_[channel_index].Pass())); |
- } |
- |
- void BootstrapChannelEndpointsOnIOThread(scoped_refptr<ChannelEndpoint> ep0, |
- scoped_refptr<ChannelEndpoint> ep1) { |
- CHECK_EQ(base::MessageLoop::current(), io_thread()->message_loop()); |
- |
- if (!channels_[0]) |
- CreateAndInitChannel(0); |
- if (!channels_[1]) |
- CreateAndInitChannel(1); |
- |
- channels_[0]->SetBootstrapEndpoint(ep0); |
- channels_[1]->SetBootstrapEndpoint(ep1); |
- } |
- |
- void BootstrapChannelEndpointOnIOThread(unsigned channel_index, |
- scoped_refptr<ChannelEndpoint> ep) { |
- CHECK_EQ(base::MessageLoop::current(), io_thread()->message_loop()); |
- CHECK(channel_index == 0 || channel_index == 1); |
- |
- CreateAndInitChannel(channel_index); |
- channels_[channel_index]->SetBootstrapEndpoint(ep); |
- } |
- |
- void RestoreInitialStateOnIOThread() { |
- CHECK_EQ(base::MessageLoop::current(), io_thread()->message_loop()); |
- |
- TearDownOnIOThread(); |
- SetUpOnIOThread(); |
- } |
- |
- embedder::SimplePlatformSupport platform_support_; |
- base::TestIOThread io_thread_; |
- embedder::ScopedPlatformHandle platform_handles_[2]; |
- scoped_refptr<Channel> channels_[2]; |
- |
- DISALLOW_COPY_AND_ASSIGN(RemoteMessagePipeTest); |
-}; |
- |
-TEST_F(RemoteMessagePipeTest, Basic) { |
- static const char kHello[] = "hello"; |
- static const char kWorld[] = "world!!!1!!!1!"; |
- char buffer[100] = {0}; |
- uint32_t buffer_size = static_cast<uint32_t>(sizeof(buffer)); |
- Waiter waiter; |
- HandleSignalsState hss; |
- uint32_t context = 0; |
- |
- // Connect message pipes. MP 0, port 1 will be attached to channel 0 and |
- // connected to MP 1, port 0, which will be attached to channel 1. This leaves |
- // MP 0, port 0 and MP 1, port 1 as the "user-facing" endpoints. |
- |
- scoped_refptr<ChannelEndpoint> ep0; |
- scoped_refptr<MessagePipe> mp0(MessagePipe::CreateLocalProxy(&ep0)); |
- scoped_refptr<ChannelEndpoint> ep1; |
- scoped_refptr<MessagePipe> mp1(MessagePipe::CreateProxyLocal(&ep1)); |
- BootstrapChannelEndpoints(ep0, ep1); |
- |
- // Write in one direction: MP 0, port 0 -> ... -> MP 1, port 1. |
- |
- // Prepare to wait on MP 1, port 1. (Add the waiter now. Otherwise, if we do |
- // it later, it might already be readable.) |
- waiter.Init(); |
- ASSERT_EQ( |
- MOJO_RESULT_OK, |
- mp1->AddAwakable(1, &waiter, MOJO_HANDLE_SIGNAL_READABLE, 123, nullptr)); |
- |
- // Write to MP 0, port 0. |
- EXPECT_EQ( |
- MOJO_RESULT_OK, |
- mp0->WriteMessage(0, UserPointer<const void>(kHello), sizeof(kHello), |
- nullptr, MOJO_WRITE_MESSAGE_FLAG_NONE)); |
- |
- // Wait. |
- EXPECT_EQ(MOJO_RESULT_OK, waiter.Wait(MOJO_DEADLINE_INDEFINITE, &context)); |
- EXPECT_EQ(123u, context); |
- hss = HandleSignalsState(); |
- mp1->RemoveAwakable(1, &waiter, &hss); |
- EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, |
- hss.satisfied_signals); |
- EXPECT_EQ(kAllSignals, hss.satisfiable_signals); |
- |
- // Read from MP 1, port 1. |
- EXPECT_EQ(MOJO_RESULT_OK, |
- mp1->ReadMessage(1, UserPointer<void>(buffer), |
- MakeUserPointer(&buffer_size), nullptr, nullptr, |
- MOJO_READ_MESSAGE_FLAG_NONE)); |
- EXPECT_EQ(sizeof(kHello), static_cast<size_t>(buffer_size)); |
- EXPECT_STREQ(kHello, buffer); |
- |
- // Write in the other direction: MP 1, port 1 -> ... -> MP 0, port 0. |
- |
- waiter.Init(); |
- ASSERT_EQ( |
- MOJO_RESULT_OK, |
- mp0->AddAwakable(0, &waiter, MOJO_HANDLE_SIGNAL_READABLE, 456, nullptr)); |
- |
- EXPECT_EQ( |
- MOJO_RESULT_OK, |
- mp1->WriteMessage(1, UserPointer<const void>(kWorld), sizeof(kWorld), |
- nullptr, MOJO_WRITE_MESSAGE_FLAG_NONE)); |
- |
- EXPECT_EQ(MOJO_RESULT_OK, waiter.Wait(MOJO_DEADLINE_INDEFINITE, &context)); |
- EXPECT_EQ(456u, context); |
- hss = HandleSignalsState(); |
- mp0->RemoveAwakable(0, &waiter, &hss); |
- EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, |
- hss.satisfied_signals); |
- EXPECT_EQ(kAllSignals, hss.satisfiable_signals); |
- |
- buffer_size = static_cast<uint32_t>(sizeof(buffer)); |
- EXPECT_EQ(MOJO_RESULT_OK, |
- mp0->ReadMessage(0, UserPointer<void>(buffer), |
- MakeUserPointer(&buffer_size), nullptr, nullptr, |
- MOJO_READ_MESSAGE_FLAG_NONE)); |
- EXPECT_EQ(sizeof(kWorld), static_cast<size_t>(buffer_size)); |
- EXPECT_STREQ(kWorld, buffer); |
- |
- // Close MP 0, port 0. |
- mp0->Close(0); |
- |
- // Try to wait for MP 1, port 1 to become readable. This will eventually fail |
- // when it realizes that MP 0, port 0 has been closed. (It may also fail |
- // immediately.) |
- waiter.Init(); |
- hss = HandleSignalsState(); |
- MojoResult result = |
- mp1->AddAwakable(1, &waiter, MOJO_HANDLE_SIGNAL_READABLE, 789, &hss); |
- if (result == MOJO_RESULT_OK) { |
- EXPECT_EQ(MOJO_RESULT_FAILED_PRECONDITION, |
- waiter.Wait(MOJO_DEADLINE_INDEFINITE, &context)); |
- EXPECT_EQ(789u, context); |
- hss = HandleSignalsState(); |
- mp1->RemoveAwakable(1, &waiter, &hss); |
- } |
- EXPECT_EQ(MOJO_HANDLE_SIGNAL_PEER_CLOSED, hss.satisfied_signals); |
- EXPECT_EQ(MOJO_HANDLE_SIGNAL_PEER_CLOSED, hss.satisfiable_signals); |
- |
- // And MP 1, port 1. |
- mp1->Close(1); |
-} |
- |
-TEST_F(RemoteMessagePipeTest, PeerClosed) { |
- Waiter waiter; |
- HandleSignalsState hss; |
- uint32_t context = 0; |
- |
- // Connect message pipes. MP 0, port 1 will be attached to channel 0 and |
- // connected to MP 1, port 0, which will be attached to channel 1. This leaves |
- // MP 0, port 0 and MP 1, port 1 as the "user-facing" endpoints. |
- |
- scoped_refptr<ChannelEndpoint> ep0; |
- scoped_refptr<MessagePipe> mp0(MessagePipe::CreateLocalProxy(&ep0)); |
- scoped_refptr<ChannelEndpoint> ep1; |
- scoped_refptr<MessagePipe> mp1(MessagePipe::CreateProxyLocal(&ep1)); |
- BootstrapChannelEndpoints(ep0, ep1); |
- |
- // Close MP 0, port 0. |
- mp0->Close(0); |
- |
- // Try to wait for MP 1, port 1 to be signaled with peer closed. |
- waiter.Init(); |
- hss = HandleSignalsState(); |
- MojoResult result = |
- mp1->AddAwakable(1, &waiter, MOJO_HANDLE_SIGNAL_PEER_CLOSED, 101, &hss); |
- if (result == MOJO_RESULT_OK) { |
- EXPECT_EQ(MOJO_RESULT_OK, waiter.Wait(MOJO_DEADLINE_INDEFINITE, &context)); |
- EXPECT_EQ(101u, context); |
- hss = HandleSignalsState(); |
- mp1->RemoveAwakable(1, &waiter, &hss); |
- } |
- EXPECT_EQ(MOJO_HANDLE_SIGNAL_PEER_CLOSED, hss.satisfied_signals); |
- EXPECT_EQ(MOJO_HANDLE_SIGNAL_PEER_CLOSED, hss.satisfiable_signals); |
- |
- // And MP 1, port 1. |
- mp1->Close(1); |
-} |
- |
-TEST_F(RemoteMessagePipeTest, Multiplex) { |
- static const char kHello[] = "hello"; |
- static const char kWorld[] = "world!!!1!!!1!"; |
- char buffer[100] = {0}; |
- uint32_t buffer_size = static_cast<uint32_t>(sizeof(buffer)); |
- Waiter waiter; |
- HandleSignalsState hss; |
- uint32_t context = 0; |
- |
- // Connect message pipes as in the |Basic| test. |
- |
- scoped_refptr<ChannelEndpoint> ep0; |
- scoped_refptr<MessagePipe> mp0(MessagePipe::CreateLocalProxy(&ep0)); |
- scoped_refptr<ChannelEndpoint> ep1; |
- scoped_refptr<MessagePipe> mp1(MessagePipe::CreateProxyLocal(&ep1)); |
- BootstrapChannelEndpoints(ep0, ep1); |
- |
- // Now put another message pipe on the channel. |
- |
- // Do this by creating a message pipe (for the |channels(0)| side) and |
- // attaching and running it, yielding the remote ID. A message is then sent |
- // via |ep0| (i.e., sent using |mp0|, port 0) with this remote ID. Upon |
- // receiving this message, |PassIncomingMessagePipe()| is used to obtain the |
- // message pipe on the other side. |
- scoped_refptr<MessagePipe> mp2(MessagePipe::CreateLocalLocal()); |
- ASSERT_TRUE(channels(0)); |
- size_t max_endpoint_info_size; |
- size_t max_platform_handle_count; |
- mp2->StartSerialize(1, channels(0), &max_endpoint_info_size, |
- &max_platform_handle_count); |
- EXPECT_GT(max_endpoint_info_size, 0u); |
- ASSERT_EQ(0u, max_platform_handle_count); |
- scoped_ptr<char[]> endpoint_info(new char[max_endpoint_info_size]); |
- size_t endpoint_info_size; |
- mp2->EndSerialize(1, channels(0), endpoint_info.get(), &endpoint_info_size, |
- nullptr); |
- EXPECT_EQ(max_endpoint_info_size, endpoint_info_size); |
- |
- waiter.Init(); |
- ASSERT_EQ( |
- MOJO_RESULT_OK, |
- mp1->AddAwakable(1, &waiter, MOJO_HANDLE_SIGNAL_READABLE, 123, nullptr)); |
- |
- EXPECT_EQ(MOJO_RESULT_OK, |
- mp0->WriteMessage(0, UserPointer<const void>(endpoint_info.get()), |
- static_cast<uint32_t>(endpoint_info_size), |
- nullptr, MOJO_WRITE_MESSAGE_FLAG_NONE)); |
- |
- EXPECT_EQ(MOJO_RESULT_OK, waiter.Wait(MOJO_DEADLINE_INDEFINITE, &context)); |
- EXPECT_EQ(123u, context); |
- hss = HandleSignalsState(); |
- mp1->RemoveAwakable(1, &waiter, &hss); |
- EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, |
- hss.satisfied_signals); |
- EXPECT_EQ(kAllSignals, hss.satisfiable_signals); |
- |
- EXPECT_EQ(endpoint_info_size, channels(1)->GetSerializedEndpointSize()); |
- scoped_ptr<char[]> received_endpoint_info(new char[endpoint_info_size]); |
- buffer_size = static_cast<uint32_t>(endpoint_info_size); |
- EXPECT_EQ(MOJO_RESULT_OK, |
- mp1->ReadMessage(1, UserPointer<void>(received_endpoint_info.get()), |
- MakeUserPointer(&buffer_size), nullptr, nullptr, |
- MOJO_READ_MESSAGE_FLAG_NONE)); |
- EXPECT_EQ(endpoint_info_size, static_cast<size_t>(buffer_size)); |
- EXPECT_EQ(0, memcmp(received_endpoint_info.get(), endpoint_info.get(), |
- endpoint_info_size)); |
- |
- // Warning: The local side of mp3 is port 0, not port 1. |
- scoped_refptr<IncomingEndpoint> incoming_endpoint = |
- channels(1)->DeserializeEndpoint(received_endpoint_info.get()); |
- ASSERT_TRUE(incoming_endpoint); |
- scoped_refptr<MessagePipe> mp3 = incoming_endpoint->ConvertToMessagePipe(); |
- ASSERT_TRUE(mp3); |
- |
- // Write: MP 2, port 0 -> MP 3, port 1. |
- |
- waiter.Init(); |
- ASSERT_EQ( |
- MOJO_RESULT_OK, |
- mp3->AddAwakable(0, &waiter, MOJO_HANDLE_SIGNAL_READABLE, 789, nullptr)); |
- |
- EXPECT_EQ( |
- MOJO_RESULT_OK, |
- mp2->WriteMessage(0, UserPointer<const void>(kHello), sizeof(kHello), |
- nullptr, MOJO_WRITE_MESSAGE_FLAG_NONE)); |
- |
- EXPECT_EQ(MOJO_RESULT_OK, waiter.Wait(MOJO_DEADLINE_INDEFINITE, &context)); |
- EXPECT_EQ(789u, context); |
- hss = HandleSignalsState(); |
- mp3->RemoveAwakable(0, &waiter, &hss); |
- EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, |
- hss.satisfied_signals); |
- EXPECT_EQ(kAllSignals, hss.satisfiable_signals); |
- |
- // Make sure there's nothing on MP 0, port 0 or MP 1, port 1 or MP 2, port 0. |
- buffer_size = static_cast<uint32_t>(sizeof(buffer)); |
- EXPECT_EQ(MOJO_RESULT_SHOULD_WAIT, |
- mp0->ReadMessage(0, UserPointer<void>(buffer), |
- MakeUserPointer(&buffer_size), nullptr, nullptr, |
- MOJO_READ_MESSAGE_FLAG_NONE)); |
- buffer_size = static_cast<uint32_t>(sizeof(buffer)); |
- EXPECT_EQ(MOJO_RESULT_SHOULD_WAIT, |
- mp1->ReadMessage(1, UserPointer<void>(buffer), |
- MakeUserPointer(&buffer_size), nullptr, nullptr, |
- MOJO_READ_MESSAGE_FLAG_NONE)); |
- buffer_size = static_cast<uint32_t>(sizeof(buffer)); |
- EXPECT_EQ(MOJO_RESULT_SHOULD_WAIT, |
- mp2->ReadMessage(0, UserPointer<void>(buffer), |
- MakeUserPointer(&buffer_size), nullptr, nullptr, |
- MOJO_READ_MESSAGE_FLAG_NONE)); |
- |
- // Read from MP 3, port 1. |
- buffer_size = static_cast<uint32_t>(sizeof(buffer)); |
- EXPECT_EQ(MOJO_RESULT_OK, |
- mp3->ReadMessage(0, UserPointer<void>(buffer), |
- MakeUserPointer(&buffer_size), nullptr, nullptr, |
- MOJO_READ_MESSAGE_FLAG_NONE)); |
- EXPECT_EQ(sizeof(kHello), static_cast<size_t>(buffer_size)); |
- EXPECT_STREQ(kHello, buffer); |
- |
- // Write: MP 0, port 0 -> MP 1, port 1 again. |
- |
- waiter.Init(); |
- ASSERT_EQ( |
- MOJO_RESULT_OK, |
- mp1->AddAwakable(1, &waiter, MOJO_HANDLE_SIGNAL_READABLE, 123, nullptr)); |
- |
- EXPECT_EQ( |
- MOJO_RESULT_OK, |
- mp0->WriteMessage(0, UserPointer<const void>(kWorld), sizeof(kWorld), |
- nullptr, MOJO_WRITE_MESSAGE_FLAG_NONE)); |
- |
- EXPECT_EQ(MOJO_RESULT_OK, waiter.Wait(MOJO_DEADLINE_INDEFINITE, &context)); |
- EXPECT_EQ(123u, context); |
- hss = HandleSignalsState(); |
- mp1->RemoveAwakable(1, &waiter, &hss); |
- EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, |
- hss.satisfied_signals); |
- EXPECT_EQ(kAllSignals | MOJO_HANDLE_SIGNAL_PEER_CLOSED, |
- hss.satisfiable_signals); |
- |
- // Make sure there's nothing on the other ports. |
- buffer_size = static_cast<uint32_t>(sizeof(buffer)); |
- EXPECT_EQ(MOJO_RESULT_SHOULD_WAIT, |
- mp0->ReadMessage(0, UserPointer<void>(buffer), |
- MakeUserPointer(&buffer_size), nullptr, nullptr, |
- MOJO_READ_MESSAGE_FLAG_NONE)); |
- buffer_size = static_cast<uint32_t>(sizeof(buffer)); |
- EXPECT_EQ(MOJO_RESULT_SHOULD_WAIT, |
- mp2->ReadMessage(0, UserPointer<void>(buffer), |
- MakeUserPointer(&buffer_size), nullptr, nullptr, |
- MOJO_READ_MESSAGE_FLAG_NONE)); |
- buffer_size = static_cast<uint32_t>(sizeof(buffer)); |
- EXPECT_EQ(MOJO_RESULT_SHOULD_WAIT, |
- mp3->ReadMessage(0, UserPointer<void>(buffer), |
- MakeUserPointer(&buffer_size), nullptr, nullptr, |
- MOJO_READ_MESSAGE_FLAG_NONE)); |
- |
- buffer_size = static_cast<uint32_t>(sizeof(buffer)); |
- EXPECT_EQ(MOJO_RESULT_OK, |
- mp1->ReadMessage(1, UserPointer<void>(buffer), |
- MakeUserPointer(&buffer_size), nullptr, nullptr, |
- MOJO_READ_MESSAGE_FLAG_NONE)); |
- EXPECT_EQ(sizeof(kWorld), static_cast<size_t>(buffer_size)); |
- EXPECT_STREQ(kWorld, buffer); |
- |
- mp0->Close(0); |
- mp1->Close(1); |
- mp2->Close(0); |
- mp3->Close(0); |
-} |
- |
-TEST_F(RemoteMessagePipeTest, CloseBeforeAttachAndRun) { |
- static const char kHello[] = "hello"; |
- char buffer[100] = {0}; |
- uint32_t buffer_size = static_cast<uint32_t>(sizeof(buffer)); |
- Waiter waiter; |
- HandleSignalsState hss; |
- uint32_t context = 0; |
- |
- // Connect message pipes. MP 0, port 1 will be attached to channel 0 and |
- // connected to MP 1, port 0, which will be attached to channel 1. This leaves |
- // MP 0, port 0 and MP 1, port 1 as the "user-facing" endpoints. |
- |
- scoped_refptr<ChannelEndpoint> ep0; |
- scoped_refptr<MessagePipe> mp0(MessagePipe::CreateLocalProxy(&ep0)); |
- |
- // Write to MP 0, port 0. |
- EXPECT_EQ( |
- MOJO_RESULT_OK, |
- mp0->WriteMessage(0, UserPointer<const void>(kHello), sizeof(kHello), |
- nullptr, MOJO_WRITE_MESSAGE_FLAG_NONE)); |
- |
- // Close MP 0, port 0 before it's even been attached to the channel and run. |
- mp0->Close(0); |
- |
- BootstrapChannelEndpointNoWait(0, ep0); |
- |
- scoped_refptr<ChannelEndpoint> ep1; |
- scoped_refptr<MessagePipe> mp1(MessagePipe::CreateProxyLocal(&ep1)); |
- |
- // Prepare to wait on MP 1, port 1. (Add the waiter now. Otherwise, if we do |
- // it later, it might already be readable.) |
- waiter.Init(); |
- ASSERT_EQ( |
- MOJO_RESULT_OK, |
- mp1->AddAwakable(1, &waiter, MOJO_HANDLE_SIGNAL_READABLE, 123, nullptr)); |
- |
- BootstrapChannelEndpointNoWait(1, ep1); |
- |
- // Wait. |
- EXPECT_EQ(MOJO_RESULT_OK, waiter.Wait(MOJO_DEADLINE_INDEFINITE, &context)); |
- EXPECT_EQ(123u, context); |
- hss = HandleSignalsState(); |
- // Note: MP 1, port 1 should definitely should be readable, but it may or may |
- // not appear as writable (there's a race, and it may not have noticed that |
- // the other side was closed yet -- e.g., inserting a sleep here would make it |
- // much more likely to notice that it's no longer writable). |
- mp1->RemoveAwakable(1, &waiter, &hss); |
- EXPECT_TRUE((hss.satisfied_signals & MOJO_HANDLE_SIGNAL_READABLE)); |
- EXPECT_TRUE((hss.satisfiable_signals & MOJO_HANDLE_SIGNAL_READABLE)); |
- |
- // Read from MP 1, port 1. |
- EXPECT_EQ(MOJO_RESULT_OK, |
- mp1->ReadMessage(1, UserPointer<void>(buffer), |
- MakeUserPointer(&buffer_size), nullptr, nullptr, |
- MOJO_READ_MESSAGE_FLAG_NONE)); |
- EXPECT_EQ(sizeof(kHello), static_cast<size_t>(buffer_size)); |
- EXPECT_STREQ(kHello, buffer); |
- |
- // And MP 1, port 1. |
- mp1->Close(1); |
-} |
- |
-TEST_F(RemoteMessagePipeTest, CloseBeforeConnect) { |
- static const char kHello[] = "hello"; |
- char buffer[100] = {0}; |
- uint32_t buffer_size = static_cast<uint32_t>(sizeof(buffer)); |
- Waiter waiter; |
- HandleSignalsState hss; |
- uint32_t context = 0; |
- |
- // Connect message pipes. MP 0, port 1 will be attached to channel 0 and |
- // connected to MP 1, port 0, which will be attached to channel 1. This leaves |
- // MP 0, port 0 and MP 1, port 1 as the "user-facing" endpoints. |
- |
- scoped_refptr<ChannelEndpoint> ep0; |
- scoped_refptr<MessagePipe> mp0(MessagePipe::CreateLocalProxy(&ep0)); |
- |
- // Write to MP 0, port 0. |
- EXPECT_EQ( |
- MOJO_RESULT_OK, |
- mp0->WriteMessage(0, UserPointer<const void>(kHello), sizeof(kHello), |
- nullptr, MOJO_WRITE_MESSAGE_FLAG_NONE)); |
- |
- BootstrapChannelEndpointNoWait(0, ep0); |
- |
- // Close MP 0, port 0 before channel 1 is even connected. |
- mp0->Close(0); |
- |
- scoped_refptr<ChannelEndpoint> ep1; |
- scoped_refptr<MessagePipe> mp1(MessagePipe::CreateProxyLocal(&ep1)); |
- |
- // Prepare to wait on MP 1, port 1. (Add the waiter now. Otherwise, if we do |
- // it later, it might already be readable.) |
- waiter.Init(); |
- ASSERT_EQ( |
- MOJO_RESULT_OK, |
- mp1->AddAwakable(1, &waiter, MOJO_HANDLE_SIGNAL_READABLE, 123, nullptr)); |
- |
- BootstrapChannelEndpointNoWait(1, ep1); |
- |
- // Wait. |
- EXPECT_EQ(MOJO_RESULT_OK, waiter.Wait(MOJO_DEADLINE_INDEFINITE, &context)); |
- EXPECT_EQ(123u, context); |
- hss = HandleSignalsState(); |
- // Note: MP 1, port 1 should definitely should be readable, but it may or may |
- // not appear as writable (there's a race, and it may not have noticed that |
- // the other side was closed yet -- e.g., inserting a sleep here would make it |
- // much more likely to notice that it's no longer writable). |
- mp1->RemoveAwakable(1, &waiter, &hss); |
- EXPECT_TRUE((hss.satisfied_signals & MOJO_HANDLE_SIGNAL_READABLE)); |
- EXPECT_TRUE((hss.satisfiable_signals & MOJO_HANDLE_SIGNAL_READABLE)); |
- |
- // Read from MP 1, port 1. |
- EXPECT_EQ(MOJO_RESULT_OK, |
- mp1->ReadMessage(1, UserPointer<void>(buffer), |
- MakeUserPointer(&buffer_size), nullptr, nullptr, |
- MOJO_READ_MESSAGE_FLAG_NONE)); |
- EXPECT_EQ(sizeof(kHello), static_cast<size_t>(buffer_size)); |
- EXPECT_STREQ(kHello, buffer); |
- |
- // And MP 1, port 1. |
- mp1->Close(1); |
-} |
- |
-TEST_F(RemoteMessagePipeTest, HandlePassing) { |
- static const char kHello[] = "hello"; |
- Waiter waiter; |
- HandleSignalsState hss; |
- uint32_t context = 0; |
- |
- scoped_refptr<ChannelEndpoint> ep0; |
- scoped_refptr<MessagePipe> mp0(MessagePipe::CreateLocalProxy(&ep0)); |
- scoped_refptr<ChannelEndpoint> ep1; |
- scoped_refptr<MessagePipe> mp1(MessagePipe::CreateProxyLocal(&ep1)); |
- BootstrapChannelEndpoints(ep0, ep1); |
- |
- // We'll try to pass this dispatcher. |
- scoped_refptr<MessagePipeDispatcher> dispatcher( |
- new MessagePipeDispatcher(MessagePipeDispatcher::kDefaultCreateOptions)); |
- scoped_refptr<MessagePipe> local_mp(MessagePipe::CreateLocalLocal()); |
- dispatcher->Init(local_mp, 0); |
- |
- // Prepare to wait on MP 1, port 1. (Add the waiter now. Otherwise, if we do |
- // it later, it might already be readable.) |
- waiter.Init(); |
- ASSERT_EQ( |
- MOJO_RESULT_OK, |
- mp1->AddAwakable(1, &waiter, MOJO_HANDLE_SIGNAL_READABLE, 123, nullptr)); |
- |
- // Write to MP 0, port 0. |
- { |
- DispatcherTransport transport( |
- test::DispatcherTryStartTransport(dispatcher.get())); |
- EXPECT_TRUE(transport.is_valid()); |
- |
- std::vector<DispatcherTransport> transports; |
- transports.push_back(transport); |
- EXPECT_EQ( |
- MOJO_RESULT_OK, |
- mp0->WriteMessage(0, UserPointer<const void>(kHello), sizeof(kHello), |
- &transports, MOJO_WRITE_MESSAGE_FLAG_NONE)); |
- transport.End(); |
- |
- // |dispatcher| should have been closed. This is |DCHECK()|ed when the |
- // |dispatcher| is destroyed. |
- EXPECT_TRUE(dispatcher->HasOneRef()); |
- dispatcher = nullptr; |
- } |
- |
- // Wait. |
- EXPECT_EQ(MOJO_RESULT_OK, waiter.Wait(MOJO_DEADLINE_INDEFINITE, &context)); |
- EXPECT_EQ(123u, context); |
- hss = HandleSignalsState(); |
- mp1->RemoveAwakable(1, &waiter, &hss); |
- EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, |
- hss.satisfied_signals); |
- EXPECT_EQ(kAllSignals, hss.satisfiable_signals); |
- |
- // Read from MP 1, port 1. |
- char read_buffer[100] = {0}; |
- uint32_t read_buffer_size = static_cast<uint32_t>(sizeof(read_buffer)); |
- DispatcherVector read_dispatchers; |
- uint32_t read_num_dispatchers = 10; // Maximum to get. |
- EXPECT_EQ( |
- MOJO_RESULT_OK, |
- mp1->ReadMessage(1, UserPointer<void>(read_buffer), |
- MakeUserPointer(&read_buffer_size), &read_dispatchers, |
- &read_num_dispatchers, MOJO_READ_MESSAGE_FLAG_NONE)); |
- EXPECT_EQ(sizeof(kHello), static_cast<size_t>(read_buffer_size)); |
- EXPECT_STREQ(kHello, read_buffer); |
- EXPECT_EQ(1u, read_dispatchers.size()); |
- EXPECT_EQ(1u, read_num_dispatchers); |
- ASSERT_TRUE(read_dispatchers[0]); |
- EXPECT_TRUE(read_dispatchers[0]->HasOneRef()); |
- |
- EXPECT_EQ(Dispatcher::kTypeMessagePipe, read_dispatchers[0]->GetType()); |
- dispatcher = static_cast<MessagePipeDispatcher*>(read_dispatchers[0].get()); |
- |
- // Add the waiter now, before it becomes readable to avoid a race. |
- waiter.Init(); |
- ASSERT_EQ(MOJO_RESULT_OK, |
- dispatcher->AddAwakable(&waiter, MOJO_HANDLE_SIGNAL_READABLE, 456, |
- nullptr)); |
- |
- // Write to "local_mp", port 1. |
- EXPECT_EQ( |
- MOJO_RESULT_OK, |
- local_mp->WriteMessage(1, UserPointer<const void>(kHello), sizeof(kHello), |
- nullptr, MOJO_WRITE_MESSAGE_FLAG_NONE)); |
- |
- // TODO(vtl): FIXME -- We (racily) crash if I close |dispatcher| immediately |
- // here. (We don't crash if I sleep and then close.) |
- |
- // Wait for the dispatcher to become readable. |
- EXPECT_EQ(MOJO_RESULT_OK, waiter.Wait(MOJO_DEADLINE_INDEFINITE, &context)); |
- EXPECT_EQ(456u, context); |
- hss = HandleSignalsState(); |
- dispatcher->RemoveAwakable(&waiter, &hss); |
- EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, |
- hss.satisfied_signals); |
- EXPECT_EQ(kAllSignals, hss.satisfiable_signals); |
- |
- // Read from the dispatcher. |
- memset(read_buffer, 0, sizeof(read_buffer)); |
- read_buffer_size = static_cast<uint32_t>(sizeof(read_buffer)); |
- EXPECT_EQ(MOJO_RESULT_OK, |
- dispatcher->ReadMessage(UserPointer<void>(read_buffer), |
- MakeUserPointer(&read_buffer_size), 0, |
- nullptr, MOJO_READ_MESSAGE_FLAG_NONE)); |
- EXPECT_EQ(sizeof(kHello), static_cast<size_t>(read_buffer_size)); |
- EXPECT_STREQ(kHello, read_buffer); |
- |
- // Prepare to wait on "local_mp", port 1. |
- waiter.Init(); |
- ASSERT_EQ(MOJO_RESULT_OK, |
- local_mp->AddAwakable(1, &waiter, MOJO_HANDLE_SIGNAL_READABLE, 789, |
- nullptr)); |
- |
- // Write to the dispatcher. |
- EXPECT_EQ(MOJO_RESULT_OK, dispatcher->WriteMessage( |
- UserPointer<const void>(kHello), sizeof(kHello), |
- nullptr, MOJO_WRITE_MESSAGE_FLAG_NONE)); |
- |
- // Wait. |
- EXPECT_EQ(MOJO_RESULT_OK, waiter.Wait(MOJO_DEADLINE_INDEFINITE, &context)); |
- EXPECT_EQ(789u, context); |
- hss = HandleSignalsState(); |
- local_mp->RemoveAwakable(1, &waiter, &hss); |
- EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, |
- hss.satisfied_signals); |
- EXPECT_EQ(kAllSignals, hss.satisfiable_signals); |
- |
- // Read from "local_mp", port 1. |
- memset(read_buffer, 0, sizeof(read_buffer)); |
- read_buffer_size = static_cast<uint32_t>(sizeof(read_buffer)); |
- EXPECT_EQ(MOJO_RESULT_OK, |
- local_mp->ReadMessage(1, UserPointer<void>(read_buffer), |
- MakeUserPointer(&read_buffer_size), nullptr, |
- nullptr, MOJO_READ_MESSAGE_FLAG_NONE)); |
- EXPECT_EQ(sizeof(kHello), static_cast<size_t>(read_buffer_size)); |
- EXPECT_STREQ(kHello, read_buffer); |
- |
- // TODO(vtl): Also test that messages queued up before the handle was sent are |
- // delivered properly. |
- |
- // Close everything that belongs to us. |
- mp0->Close(0); |
- mp1->Close(1); |
- EXPECT_EQ(MOJO_RESULT_OK, dispatcher->Close()); |
- // Note that |local_mp|'s port 0 belong to |dispatcher|, which was closed. |
- local_mp->Close(1); |
-} |
- |
-TEST_F(RemoteMessagePipeTest, HandlePassingHalfClosed) { |
- static const char kHello[] = "hello"; |
- static const char kWorld[] = "world!"; |
- Waiter waiter; |
- HandleSignalsState hss; |
- uint32_t context = 0; |
- |
- // We'll try to pass this dispatcher. |
- scoped_refptr<MessagePipeDispatcher> dispatcher( |
- new MessagePipeDispatcher(MessagePipeDispatcher::kDefaultCreateOptions)); |
- scoped_refptr<MessagePipe> local_mp(MessagePipe::CreateLocalLocal()); |
- dispatcher->Init(local_mp, 0); |
- |
- hss = local_mp->GetHandleSignalsState(0); |
- EXPECT_EQ(MOJO_HANDLE_SIGNAL_WRITABLE, hss.satisfied_signals); |
- EXPECT_EQ(kAllSignals, hss.satisfiable_signals); |
- // Write to the other end (|local_mp|, port 1), and then close it. |
- EXPECT_EQ( |
- MOJO_RESULT_OK, |
- local_mp->WriteMessage(1, UserPointer<const void>(kHello), sizeof(kHello), |
- nullptr, MOJO_WRITE_MESSAGE_FLAG_NONE)); |
- hss = local_mp->GetHandleSignalsState(0); |
- EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, |
- hss.satisfied_signals); |
- EXPECT_EQ(kAllSignals, hss.satisfiable_signals); |
- // Then the second message.... |
- EXPECT_EQ( |
- MOJO_RESULT_OK, |
- local_mp->WriteMessage(1, UserPointer<const void>(kWorld), sizeof(kWorld), |
- nullptr, MOJO_WRITE_MESSAGE_FLAG_NONE)); |
- hss = local_mp->GetHandleSignalsState(0); |
- EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, |
- hss.satisfied_signals); |
- EXPECT_EQ(kAllSignals, hss.satisfiable_signals); |
- // Then close it. |
- local_mp->Close(1); |
- |
- scoped_refptr<ChannelEndpoint> ep0; |
- scoped_refptr<MessagePipe> mp0(MessagePipe::CreateLocalProxy(&ep0)); |
- scoped_refptr<ChannelEndpoint> ep1; |
- scoped_refptr<MessagePipe> mp1(MessagePipe::CreateProxyLocal(&ep1)); |
- BootstrapChannelEndpoints(ep0, ep1); |
- |
- // Prepare to wait on MP 1, port 1. (Add the waiter now. Otherwise, if we do |
- // it later, it might already be readable.) |
- waiter.Init(); |
- ASSERT_EQ( |
- MOJO_RESULT_OK, |
- mp1->AddAwakable(1, &waiter, MOJO_HANDLE_SIGNAL_READABLE, 123, nullptr)); |
- |
- // Write to MP 0, port 0. |
- { |
- DispatcherTransport transport( |
- test::DispatcherTryStartTransport(dispatcher.get())); |
- EXPECT_TRUE(transport.is_valid()); |
- |
- std::vector<DispatcherTransport> transports; |
- transports.push_back(transport); |
- EXPECT_EQ( |
- MOJO_RESULT_OK, |
- mp0->WriteMessage(0, UserPointer<const void>(kHello), sizeof(kHello), |
- &transports, MOJO_WRITE_MESSAGE_FLAG_NONE)); |
- transport.End(); |
- |
- // |dispatcher| should have been closed. This is |DCHECK()|ed when the |
- // |dispatcher| is destroyed. |
- EXPECT_TRUE(dispatcher->HasOneRef()); |
- dispatcher = nullptr; |
- } |
- |
- // Wait. |
- EXPECT_EQ(MOJO_RESULT_OK, waiter.Wait(MOJO_DEADLINE_INDEFINITE, &context)); |
- EXPECT_EQ(123u, context); |
- hss = HandleSignalsState(); |
- mp1->RemoveAwakable(1, &waiter, &hss); |
- EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, |
- hss.satisfied_signals); |
- EXPECT_EQ(kAllSignals, hss.satisfiable_signals); |
- |
- // Read from MP 1, port 1. |
- char read_buffer[100] = {0}; |
- uint32_t read_buffer_size = static_cast<uint32_t>(sizeof(read_buffer)); |
- DispatcherVector read_dispatchers; |
- uint32_t read_num_dispatchers = 10; // Maximum to get. |
- EXPECT_EQ( |
- MOJO_RESULT_OK, |
- mp1->ReadMessage(1, UserPointer<void>(read_buffer), |
- MakeUserPointer(&read_buffer_size), &read_dispatchers, |
- &read_num_dispatchers, MOJO_READ_MESSAGE_FLAG_NONE)); |
- EXPECT_EQ(sizeof(kHello), static_cast<size_t>(read_buffer_size)); |
- EXPECT_STREQ(kHello, read_buffer); |
- EXPECT_EQ(1u, read_dispatchers.size()); |
- EXPECT_EQ(1u, read_num_dispatchers); |
- ASSERT_TRUE(read_dispatchers[0]); |
- EXPECT_TRUE(read_dispatchers[0]->HasOneRef()); |
- |
- EXPECT_EQ(Dispatcher::kTypeMessagePipe, read_dispatchers[0]->GetType()); |
- dispatcher = static_cast<MessagePipeDispatcher*>(read_dispatchers[0].get()); |
- |
- // |dispatcher| should already be readable and not writable. |
- hss = dispatcher->GetHandleSignalsState(); |
- EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_PEER_CLOSED, |
- hss.satisfied_signals); |
- EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_PEER_CLOSED, |
- hss.satisfiable_signals); |
- // So read from it. |
- memset(read_buffer, 0, sizeof(read_buffer)); |
- read_buffer_size = static_cast<uint32_t>(sizeof(read_buffer)); |
- EXPECT_EQ(MOJO_RESULT_OK, |
- dispatcher->ReadMessage(UserPointer<void>(read_buffer), |
- MakeUserPointer(&read_buffer_size), 0, |
- nullptr, MOJO_READ_MESSAGE_FLAG_NONE)); |
- EXPECT_EQ(sizeof(kHello), static_cast<size_t>(read_buffer_size)); |
- EXPECT_STREQ(kHello, read_buffer); |
- // It should still be readable. |
- hss = dispatcher->GetHandleSignalsState(); |
- EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_PEER_CLOSED, |
- hss.satisfied_signals); |
- EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_PEER_CLOSED, |
- hss.satisfiable_signals); |
- // So read from it. |
- memset(read_buffer, 0, sizeof(read_buffer)); |
- read_buffer_size = static_cast<uint32_t>(sizeof(read_buffer)); |
- EXPECT_EQ(MOJO_RESULT_OK, |
- dispatcher->ReadMessage(UserPointer<void>(read_buffer), |
- MakeUserPointer(&read_buffer_size), 0, |
- nullptr, MOJO_READ_MESSAGE_FLAG_NONE)); |
- EXPECT_EQ(sizeof(kWorld), static_cast<size_t>(read_buffer_size)); |
- EXPECT_STREQ(kWorld, read_buffer); |
- // Now it should no longer be readable. |
- hss = dispatcher->GetHandleSignalsState(); |
- EXPECT_EQ(MOJO_HANDLE_SIGNAL_PEER_CLOSED, hss.satisfied_signals); |
- EXPECT_EQ(MOJO_HANDLE_SIGNAL_PEER_CLOSED, hss.satisfiable_signals); |
- |
- // Close everything that belongs to us. |
- mp0->Close(0); |
- mp1->Close(1); |
- EXPECT_EQ(MOJO_RESULT_OK, dispatcher->Close()); |
-} |
- |
-#if defined(OS_POSIX) |
-#define MAYBE_SharedBufferPassing SharedBufferPassing |
-#else |
-// Not yet implemented (on Windows). |
-#define MAYBE_SharedBufferPassing DISABLED_SharedBufferPassing |
-#endif |
-TEST_F(RemoteMessagePipeTest, MAYBE_SharedBufferPassing) { |
- static const char kHello[] = "hello"; |
- Waiter waiter; |
- HandleSignalsState hss; |
- uint32_t context = 0; |
- |
- scoped_refptr<ChannelEndpoint> ep0; |
- scoped_refptr<MessagePipe> mp0(MessagePipe::CreateLocalProxy(&ep0)); |
- scoped_refptr<ChannelEndpoint> ep1; |
- scoped_refptr<MessagePipe> mp1(MessagePipe::CreateProxyLocal(&ep1)); |
- BootstrapChannelEndpoints(ep0, ep1); |
- |
- // We'll try to pass this dispatcher. |
- scoped_refptr<SharedBufferDispatcher> dispatcher; |
- EXPECT_EQ(MOJO_RESULT_OK, SharedBufferDispatcher::Create( |
- platform_support(), |
- SharedBufferDispatcher::kDefaultCreateOptions, |
- 100, &dispatcher)); |
- ASSERT_TRUE(dispatcher); |
- |
- // Make a mapping. |
- scoped_ptr<embedder::PlatformSharedBufferMapping> mapping0; |
- EXPECT_EQ(MOJO_RESULT_OK, dispatcher->MapBuffer( |
- 0, 100, MOJO_MAP_BUFFER_FLAG_NONE, &mapping0)); |
- ASSERT_TRUE(mapping0); |
- ASSERT_TRUE(mapping0->GetBase()); |
- ASSERT_EQ(100u, mapping0->GetLength()); |
- static_cast<char*>(mapping0->GetBase())[0] = 'A'; |
- static_cast<char*>(mapping0->GetBase())[50] = 'B'; |
- static_cast<char*>(mapping0->GetBase())[99] = 'C'; |
- |
- // Prepare to wait on MP 1, port 1. (Add the waiter now. Otherwise, if we do |
- // it later, it might already be readable.) |
- waiter.Init(); |
- ASSERT_EQ( |
- MOJO_RESULT_OK, |
- mp1->AddAwakable(1, &waiter, MOJO_HANDLE_SIGNAL_READABLE, 123, nullptr)); |
- |
- // Write to MP 0, port 0. |
- { |
- DispatcherTransport transport( |
- test::DispatcherTryStartTransport(dispatcher.get())); |
- EXPECT_TRUE(transport.is_valid()); |
- |
- std::vector<DispatcherTransport> transports; |
- transports.push_back(transport); |
- EXPECT_EQ( |
- MOJO_RESULT_OK, |
- mp0->WriteMessage(0, UserPointer<const void>(kHello), sizeof(kHello), |
- &transports, MOJO_WRITE_MESSAGE_FLAG_NONE)); |
- transport.End(); |
- |
- // |dispatcher| should have been closed. This is |DCHECK()|ed when the |
- // |dispatcher| is destroyed. |
- EXPECT_TRUE(dispatcher->HasOneRef()); |
- dispatcher = nullptr; |
- } |
- |
- // Wait. |
- EXPECT_EQ(MOJO_RESULT_OK, waiter.Wait(MOJO_DEADLINE_INDEFINITE, &context)); |
- EXPECT_EQ(123u, context); |
- hss = HandleSignalsState(); |
- mp1->RemoveAwakable(1, &waiter, &hss); |
- EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, |
- hss.satisfied_signals); |
- EXPECT_EQ(kAllSignals, hss.satisfiable_signals); |
- |
- // Read from MP 1, port 1. |
- char read_buffer[100] = {0}; |
- uint32_t read_buffer_size = static_cast<uint32_t>(sizeof(read_buffer)); |
- DispatcherVector read_dispatchers; |
- uint32_t read_num_dispatchers = 10; // Maximum to get. |
- EXPECT_EQ( |
- MOJO_RESULT_OK, |
- mp1->ReadMessage(1, UserPointer<void>(read_buffer), |
- MakeUserPointer(&read_buffer_size), &read_dispatchers, |
- &read_num_dispatchers, MOJO_READ_MESSAGE_FLAG_NONE)); |
- EXPECT_EQ(sizeof(kHello), static_cast<size_t>(read_buffer_size)); |
- EXPECT_STREQ(kHello, read_buffer); |
- EXPECT_EQ(1u, read_dispatchers.size()); |
- EXPECT_EQ(1u, read_num_dispatchers); |
- ASSERT_TRUE(read_dispatchers[0]); |
- EXPECT_TRUE(read_dispatchers[0]->HasOneRef()); |
- |
- EXPECT_EQ(Dispatcher::kTypeSharedBuffer, read_dispatchers[0]->GetType()); |
- dispatcher = static_cast<SharedBufferDispatcher*>(read_dispatchers[0].get()); |
- |
- // Make another mapping. |
- scoped_ptr<embedder::PlatformSharedBufferMapping> mapping1; |
- EXPECT_EQ(MOJO_RESULT_OK, dispatcher->MapBuffer( |
- 0, 100, MOJO_MAP_BUFFER_FLAG_NONE, &mapping1)); |
- ASSERT_TRUE(mapping1); |
- ASSERT_TRUE(mapping1->GetBase()); |
- ASSERT_EQ(100u, mapping1->GetLength()); |
- EXPECT_NE(mapping1->GetBase(), mapping0->GetBase()); |
- EXPECT_EQ('A', static_cast<char*>(mapping1->GetBase())[0]); |
- EXPECT_EQ('B', static_cast<char*>(mapping1->GetBase())[50]); |
- EXPECT_EQ('C', static_cast<char*>(mapping1->GetBase())[99]); |
- |
- // Write stuff either way. |
- static_cast<char*>(mapping1->GetBase())[1] = 'x'; |
- EXPECT_EQ('x', static_cast<char*>(mapping0->GetBase())[1]); |
- static_cast<char*>(mapping0->GetBase())[2] = 'y'; |
- EXPECT_EQ('y', static_cast<char*>(mapping1->GetBase())[2]); |
- |
- // Kill the first mapping; the second should still be valid. |
- mapping0.reset(); |
- EXPECT_EQ('A', static_cast<char*>(mapping1->GetBase())[0]); |
- |
- // Close everything that belongs to us. |
- mp0->Close(0); |
- mp1->Close(1); |
- EXPECT_EQ(MOJO_RESULT_OK, dispatcher->Close()); |
- |
- // The second mapping should still be good. |
- EXPECT_EQ('x', static_cast<char*>(mapping1->GetBase())[1]); |
-} |
- |
-#if defined(OS_POSIX) |
-#define MAYBE_PlatformHandlePassing PlatformHandlePassing |
-#else |
-// Not yet implemented (on Windows). |
-#define MAYBE_PlatformHandlePassing DISABLED_PlatformHandlePassing |
-#endif |
-TEST_F(RemoteMessagePipeTest, MAYBE_PlatformHandlePassing) { |
- base::ScopedTempDir temp_dir; |
- ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); |
- |
- static const char kHello[] = "hello"; |
- static const char kWorld[] = "world"; |
- Waiter waiter; |
- uint32_t context = 0; |
- HandleSignalsState hss; |
- |
- scoped_refptr<ChannelEndpoint> ep0; |
- scoped_refptr<MessagePipe> mp0(MessagePipe::CreateLocalProxy(&ep0)); |
- scoped_refptr<ChannelEndpoint> ep1; |
- scoped_refptr<MessagePipe> mp1(MessagePipe::CreateProxyLocal(&ep1)); |
- BootstrapChannelEndpoints(ep0, ep1); |
- |
- base::FilePath unused; |
- base::ScopedFILE fp( |
- base::CreateAndOpenTemporaryFileInDir(temp_dir.path(), &unused)); |
- EXPECT_EQ(sizeof(kHello), fwrite(kHello, 1, sizeof(kHello), fp.get())); |
- // We'll try to pass this dispatcher, which will cause a |PlatformHandle| to |
- // be passed. |
- scoped_refptr<PlatformHandleDispatcher> dispatcher( |
- new PlatformHandleDispatcher( |
- mojo::test::PlatformHandleFromFILE(fp.Pass()))); |
- |
- // Prepare to wait on MP 1, port 1. (Add the waiter now. Otherwise, if we do |
- // it later, it might already be readable.) |
- waiter.Init(); |
- ASSERT_EQ( |
- MOJO_RESULT_OK, |
- mp1->AddAwakable(1, &waiter, MOJO_HANDLE_SIGNAL_READABLE, 123, nullptr)); |
- |
- // Write to MP 0, port 0. |
- { |
- DispatcherTransport transport( |
- test::DispatcherTryStartTransport(dispatcher.get())); |
- EXPECT_TRUE(transport.is_valid()); |
- |
- std::vector<DispatcherTransport> transports; |
- transports.push_back(transport); |
- EXPECT_EQ( |
- MOJO_RESULT_OK, |
- mp0->WriteMessage(0, UserPointer<const void>(kWorld), sizeof(kWorld), |
- &transports, MOJO_WRITE_MESSAGE_FLAG_NONE)); |
- transport.End(); |
- |
- // |dispatcher| should have been closed. This is |DCHECK()|ed when the |
- // |dispatcher| is destroyed. |
- EXPECT_TRUE(dispatcher->HasOneRef()); |
- dispatcher = nullptr; |
- } |
- |
- // Wait. |
- EXPECT_EQ(MOJO_RESULT_OK, waiter.Wait(MOJO_DEADLINE_INDEFINITE, &context)); |
- EXPECT_EQ(123u, context); |
- hss = HandleSignalsState(); |
- mp1->RemoveAwakable(1, &waiter, &hss); |
- EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, |
- hss.satisfied_signals); |
- EXPECT_EQ(kAllSignals, hss.satisfiable_signals); |
- |
- // Read from MP 1, port 1. |
- char read_buffer[100] = {0}; |
- uint32_t read_buffer_size = static_cast<uint32_t>(sizeof(read_buffer)); |
- DispatcherVector read_dispatchers; |
- uint32_t read_num_dispatchers = 10; // Maximum to get. |
- EXPECT_EQ( |
- MOJO_RESULT_OK, |
- mp1->ReadMessage(1, UserPointer<void>(read_buffer), |
- MakeUserPointer(&read_buffer_size), &read_dispatchers, |
- &read_num_dispatchers, MOJO_READ_MESSAGE_FLAG_NONE)); |
- EXPECT_EQ(sizeof(kWorld), static_cast<size_t>(read_buffer_size)); |
- EXPECT_STREQ(kWorld, read_buffer); |
- EXPECT_EQ(1u, read_dispatchers.size()); |
- EXPECT_EQ(1u, read_num_dispatchers); |
- ASSERT_TRUE(read_dispatchers[0]); |
- EXPECT_TRUE(read_dispatchers[0]->HasOneRef()); |
- |
- EXPECT_EQ(Dispatcher::kTypePlatformHandle, read_dispatchers[0]->GetType()); |
- dispatcher = |
- static_cast<PlatformHandleDispatcher*>(read_dispatchers[0].get()); |
- |
- embedder::ScopedPlatformHandle h = dispatcher->PassPlatformHandle().Pass(); |
- EXPECT_TRUE(h.is_valid()); |
- |
- fp = mojo::test::FILEFromPlatformHandle(h.Pass(), "rb").Pass(); |
- EXPECT_FALSE(h.is_valid()); |
- EXPECT_TRUE(fp); |
- |
- rewind(fp.get()); |
- memset(read_buffer, 0, sizeof(read_buffer)); |
- EXPECT_EQ(sizeof(kHello), |
- fread(read_buffer, 1, sizeof(read_buffer), fp.get())); |
- EXPECT_STREQ(kHello, read_buffer); |
- |
- // Close everything that belongs to us. |
- mp0->Close(0); |
- mp1->Close(1); |
- EXPECT_EQ(MOJO_RESULT_OK, dispatcher->Close()); |
-} |
- |
-// Test racing closes (on each end). |
-// Note: A flaky failure would almost certainly indicate a problem in the code |
-// itself (not in the test). Also, any logged warnings/errors would also |
-// probably be indicative of bugs. |
-TEST_F(RemoteMessagePipeTest, RacingClosesStress) { |
- base::TimeDelta delay = base::TimeDelta::FromMilliseconds(5); |
- |
- for (unsigned i = 0; i < 256; i++) { |
- DVLOG(2) << "---------------------------------------- " << i; |
- scoped_refptr<ChannelEndpoint> ep0; |
- scoped_refptr<MessagePipe> mp0(MessagePipe::CreateLocalProxy(&ep0)); |
- BootstrapChannelEndpointNoWait(0, ep0); |
- |
- scoped_refptr<ChannelEndpoint> ep1; |
- scoped_refptr<MessagePipe> mp1(MessagePipe::CreateProxyLocal(&ep1)); |
- BootstrapChannelEndpointNoWait(1, ep1); |
- |
- if (i & 1u) { |
- io_thread()->task_runner()->PostTask( |
- FROM_HERE, base::Bind(&base::PlatformThread::Sleep, delay)); |
- } |
- if (i & 2u) |
- base::PlatformThread::Sleep(delay); |
- |
- mp0->Close(0); |
- |
- if (i & 4u) { |
- io_thread()->task_runner()->PostTask( |
- FROM_HERE, base::Bind(&base::PlatformThread::Sleep, delay)); |
- } |
- if (i & 8u) |
- base::PlatformThread::Sleep(delay); |
- |
- mp1->Close(1); |
- |
- RestoreInitialState(); |
- } |
-} |
- |
-// Tests passing an end of a message pipe over a remote message pipe, and then |
-// passing that end back. |
-// TODO(vtl): Also test passing a message pipe across two remote message pipes. |
-TEST_F(RemoteMessagePipeTest, PassMessagePipeHandleAcrossAndBack) { |
- static const char kHello[] = "hello"; |
- static const char kWorld[] = "world"; |
- Waiter waiter; |
- HandleSignalsState hss; |
- uint32_t context = 0; |
- |
- scoped_refptr<ChannelEndpoint> ep0; |
- scoped_refptr<MessagePipe> mp0(MessagePipe::CreateLocalProxy(&ep0)); |
- scoped_refptr<ChannelEndpoint> ep1; |
- scoped_refptr<MessagePipe> mp1(MessagePipe::CreateProxyLocal(&ep1)); |
- BootstrapChannelEndpoints(ep0, ep1); |
- |
- // We'll try to pass this dispatcher. |
- scoped_refptr<MessagePipeDispatcher> dispatcher( |
- new MessagePipeDispatcher(MessagePipeDispatcher::kDefaultCreateOptions)); |
- scoped_refptr<MessagePipe> local_mp(MessagePipe::CreateLocalLocal()); |
- dispatcher->Init(local_mp, 0); |
- |
- // Prepare to wait on MP 1, port 1. (Add the waiter now. Otherwise, if we do |
- // it later, it might already be readable.) |
- waiter.Init(); |
- ASSERT_EQ( |
- MOJO_RESULT_OK, |
- mp1->AddAwakable(1, &waiter, MOJO_HANDLE_SIGNAL_READABLE, 123, nullptr)); |
- |
- // Write to MP 0, port 0. |
- { |
- DispatcherTransport transport( |
- test::DispatcherTryStartTransport(dispatcher.get())); |
- EXPECT_TRUE(transport.is_valid()); |
- |
- std::vector<DispatcherTransport> transports; |
- transports.push_back(transport); |
- EXPECT_EQ( |
- MOJO_RESULT_OK, |
- mp0->WriteMessage(0, UserPointer<const void>(kHello), sizeof(kHello), |
- &transports, MOJO_WRITE_MESSAGE_FLAG_NONE)); |
- transport.End(); |
- |
- // |dispatcher| should have been closed. This is |DCHECK()|ed when the |
- // |dispatcher| is destroyed. |
- EXPECT_TRUE(dispatcher->HasOneRef()); |
- dispatcher = nullptr; |
- } |
- |
- // Wait. |
- EXPECT_EQ(MOJO_RESULT_OK, waiter.Wait(MOJO_DEADLINE_INDEFINITE, &context)); |
- EXPECT_EQ(123u, context); |
- hss = HandleSignalsState(); |
- mp1->RemoveAwakable(1, &waiter, &hss); |
- EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, |
- hss.satisfied_signals); |
- EXPECT_EQ(kAllSignals, hss.satisfiable_signals); |
- |
- // Read from MP 1, port 1. |
- char read_buffer[100] = {0}; |
- uint32_t read_buffer_size = static_cast<uint32_t>(sizeof(read_buffer)); |
- DispatcherVector read_dispatchers; |
- uint32_t read_num_dispatchers = 10; // Maximum to get. |
- EXPECT_EQ( |
- MOJO_RESULT_OK, |
- mp1->ReadMessage(1, UserPointer<void>(read_buffer), |
- MakeUserPointer(&read_buffer_size), &read_dispatchers, |
- &read_num_dispatchers, MOJO_READ_MESSAGE_FLAG_NONE)); |
- EXPECT_EQ(sizeof(kHello), static_cast<size_t>(read_buffer_size)); |
- EXPECT_STREQ(kHello, read_buffer); |
- EXPECT_EQ(1u, read_dispatchers.size()); |
- EXPECT_EQ(1u, read_num_dispatchers); |
- ASSERT_TRUE(read_dispatchers[0]); |
- EXPECT_TRUE(read_dispatchers[0]->HasOneRef()); |
- |
- EXPECT_EQ(Dispatcher::kTypeMessagePipe, read_dispatchers[0]->GetType()); |
- dispatcher = static_cast<MessagePipeDispatcher*>(read_dispatchers[0].get()); |
- read_dispatchers.clear(); |
- |
- // Now pass it back. |
- |
- // Prepare to wait on MP 0, port 0. (Add the waiter now. Otherwise, if we do |
- // it later, it might already be readable.) |
- waiter.Init(); |
- ASSERT_EQ( |
- MOJO_RESULT_OK, |
- mp0->AddAwakable(0, &waiter, MOJO_HANDLE_SIGNAL_READABLE, 456, nullptr)); |
- |
- // Write to MP 1, port 1. |
- { |
- DispatcherTransport transport( |
- test::DispatcherTryStartTransport(dispatcher.get())); |
- EXPECT_TRUE(transport.is_valid()); |
- |
- std::vector<DispatcherTransport> transports; |
- transports.push_back(transport); |
- EXPECT_EQ( |
- MOJO_RESULT_OK, |
- mp1->WriteMessage(1, UserPointer<const void>(kWorld), sizeof(kWorld), |
- &transports, MOJO_WRITE_MESSAGE_FLAG_NONE)); |
- transport.End(); |
- |
- // |dispatcher| should have been closed. This is |DCHECK()|ed when the |
- // |dispatcher| is destroyed. |
- EXPECT_TRUE(dispatcher->HasOneRef()); |
- dispatcher = nullptr; |
- } |
- |
- // Wait. |
- EXPECT_EQ(MOJO_RESULT_OK, waiter.Wait(MOJO_DEADLINE_INDEFINITE, &context)); |
- EXPECT_EQ(456u, context); |
- hss = HandleSignalsState(); |
- mp0->RemoveAwakable(0, &waiter, &hss); |
- EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, |
- hss.satisfied_signals); |
- EXPECT_EQ(kAllSignals, hss.satisfiable_signals); |
- |
- // Read from MP 0, port 0. |
- read_buffer_size = static_cast<uint32_t>(sizeof(read_buffer)); |
- read_num_dispatchers = 10; // Maximum to get. |
- EXPECT_EQ( |
- MOJO_RESULT_OK, |
- mp0->ReadMessage(0, UserPointer<void>(read_buffer), |
- MakeUserPointer(&read_buffer_size), &read_dispatchers, |
- &read_num_dispatchers, MOJO_READ_MESSAGE_FLAG_NONE)); |
- EXPECT_EQ(sizeof(kWorld), static_cast<size_t>(read_buffer_size)); |
- EXPECT_STREQ(kWorld, read_buffer); |
- EXPECT_EQ(1u, read_dispatchers.size()); |
- EXPECT_EQ(1u, read_num_dispatchers); |
- ASSERT_TRUE(read_dispatchers[0]); |
- EXPECT_TRUE(read_dispatchers[0]->HasOneRef()); |
- |
- EXPECT_EQ(Dispatcher::kTypeMessagePipe, read_dispatchers[0]->GetType()); |
- dispatcher = static_cast<MessagePipeDispatcher*>(read_dispatchers[0].get()); |
- read_dispatchers.clear(); |
- |
- // Add the waiter now, before it becomes readable to avoid a race. |
- waiter.Init(); |
- ASSERT_EQ(MOJO_RESULT_OK, |
- dispatcher->AddAwakable(&waiter, MOJO_HANDLE_SIGNAL_READABLE, 789, |
- nullptr)); |
- |
- // Write to "local_mp", port 1. |
- EXPECT_EQ( |
- MOJO_RESULT_OK, |
- local_mp->WriteMessage(1, UserPointer<const void>(kHello), sizeof(kHello), |
- nullptr, MOJO_WRITE_MESSAGE_FLAG_NONE)); |
- |
- // Wait for the dispatcher to become readable. |
- EXPECT_EQ(MOJO_RESULT_OK, waiter.Wait(MOJO_DEADLINE_INDEFINITE, &context)); |
- EXPECT_EQ(789u, context); |
- hss = HandleSignalsState(); |
- dispatcher->RemoveAwakable(&waiter, &hss); |
- EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, |
- hss.satisfied_signals); |
- EXPECT_EQ(kAllSignals, hss.satisfiable_signals); |
- |
- // Read from the dispatcher. |
- memset(read_buffer, 0, sizeof(read_buffer)); |
- read_buffer_size = static_cast<uint32_t>(sizeof(read_buffer)); |
- EXPECT_EQ(MOJO_RESULT_OK, |
- dispatcher->ReadMessage(UserPointer<void>(read_buffer), |
- MakeUserPointer(&read_buffer_size), 0, |
- nullptr, MOJO_READ_MESSAGE_FLAG_NONE)); |
- EXPECT_EQ(sizeof(kHello), static_cast<size_t>(read_buffer_size)); |
- EXPECT_STREQ(kHello, read_buffer); |
- |
- // Prepare to wait on "local_mp", port 1. |
- waiter.Init(); |
- ASSERT_EQ(MOJO_RESULT_OK, |
- local_mp->AddAwakable(1, &waiter, MOJO_HANDLE_SIGNAL_READABLE, 789, |
- nullptr)); |
- |
- // Write to the dispatcher. |
- EXPECT_EQ(MOJO_RESULT_OK, dispatcher->WriteMessage( |
- UserPointer<const void>(kHello), sizeof(kHello), |
- nullptr, MOJO_WRITE_MESSAGE_FLAG_NONE)); |
- |
- // Wait. |
- EXPECT_EQ(MOJO_RESULT_OK, waiter.Wait(MOJO_DEADLINE_INDEFINITE, &context)); |
- EXPECT_EQ(789u, context); |
- hss = HandleSignalsState(); |
- local_mp->RemoveAwakable(1, &waiter, &hss); |
- EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, |
- hss.satisfied_signals); |
- EXPECT_EQ(kAllSignals, hss.satisfiable_signals); |
- |
- // Read from "local_mp", port 1. |
- memset(read_buffer, 0, sizeof(read_buffer)); |
- read_buffer_size = static_cast<uint32_t>(sizeof(read_buffer)); |
- EXPECT_EQ(MOJO_RESULT_OK, |
- local_mp->ReadMessage(1, UserPointer<void>(read_buffer), |
- MakeUserPointer(&read_buffer_size), nullptr, |
- nullptr, MOJO_READ_MESSAGE_FLAG_NONE)); |
- EXPECT_EQ(sizeof(kHello), static_cast<size_t>(read_buffer_size)); |
- EXPECT_STREQ(kHello, read_buffer); |
- |
- // TODO(vtl): Also test the cases where messages are written and read (at |
- // various points) on the message pipe being passed around. |
- |
- // Close everything that belongs to us. |
- mp0->Close(0); |
- mp1->Close(1); |
- EXPECT_EQ(MOJO_RESULT_OK, dispatcher->Close()); |
- // Note that |local_mp|'s port 0 belong to |dispatcher|, which was closed. |
- local_mp->Close(1); |
-} |
- |
-} // namespace |
-} // namespace system |
-} // namespace mojo |