// Copyright 2020 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/core/node_channel.h" #include "base/callback_helpers.h" #include "base/logging.h" #include "base/memory/scoped_refptr.h" #include "base/message_loop/message_pump_type.h" #include "base/test/task_environment.h" #include "base/threading/thread.h" #include "mojo/core/embedder/embedder.h" #include "mojo/core/test/mock_node_channel_delegate.h" #include "mojo/public/cpp/platform/platform_channel.h" #include "mojo/public/cpp/platform/platform_channel_endpoint.h" #include "testing/gtest/include/gtest/gtest.h" namespace mojo { namespace core { namespace { using NodeChannelTest = testing::Test; using ports::NodeName; using testing::_; scoped_refptr CreateNodeChannel(NodeChannel::Delegate* delegate, PlatformChannelEndpoint endpoint) { return NodeChannel::Create(delegate, ConnectionParams(std::move(endpoint)), Channel::HandlePolicy::kAcceptHandles, GetIOTaskRunner(), base::NullCallback()); } TEST_F(NodeChannelTest, DestructionIsSafe) { // Regression test for https://crbug.com/1081874. base::test::TaskEnvironment task_environment; PlatformChannel channel; MockNodeChannelDelegate local_delegate; auto local_channel = CreateNodeChannel(&local_delegate, channel.TakeLocalEndpoint()); local_channel->Start(); MockNodeChannelDelegate remote_delegate; auto remote_channel = CreateNodeChannel(&remote_delegate, channel.TakeRemoteEndpoint()); remote_channel->Start(); // Verify end-to-end operation const NodeName kRemoteNodeName{123, 456}; const NodeName kToken{987, 654}; base::RunLoop loop; EXPECT_CALL(local_delegate, OnAcceptInvitee(ports::kInvalidNodeName, kRemoteNodeName, kToken)) .WillRepeatedly([&] { loop.Quit(); }); remote_channel->AcceptInvitee(kRemoteNodeName, kToken); loop.Run(); // Now send another message to the local endpoint but tear it down // immediately. This will race with the message being received on the IO // thread, and although the corresponding delegate call may or may not // dispatch as a result, the race should still be memory-safe. remote_channel->AcceptInvitee(kRemoteNodeName, kToken); base::RunLoop error_loop; EXPECT_CALL(remote_delegate, OnChannelError).WillOnce([&] { error_loop.Quit(); }); local_channel.reset(); error_loop.Run(); } TEST_F(NodeChannelTest, MessagesCannotBeSmallerThanOldestVersion) { base::test::TaskEnvironment task_environment; PlatformChannel channel; MockNodeChannelDelegate local_delegate; auto local_channel = CreateNodeChannel(&local_delegate, channel.TakeLocalEndpoint()); local_channel->Start(); MockNodeChannelDelegate remote_delegate; auto remote_channel = CreateNodeChannel(&remote_delegate, channel.TakeRemoteEndpoint()); remote_channel->Start(); base::RunLoop loop; // It's a bad message and shouldn't be passed to the delegate. EXPECT_CALL(local_delegate, OnRequestPortMerge(_, _, _)).Times(0); // This good message should go through after. const NodeName kRemoteNodeName{123, 456}; const NodeName kToken{987, 654}; EXPECT_CALL(local_delegate, OnAcceptInvitee(ports::kInvalidNodeName, kRemoteNodeName, kToken)) .WillRepeatedly([&] { loop.Quit(); }); // 1 byte is not enough to contain the oldest version of the request port // merge payload, it should be discarded. int payload_size = 1; int capacity = /*sizeof(header)=*/8 + payload_size; auto message = Channel::Message::CreateMessage(capacity, capacity, /*num_handles=*/0); memset(message->mutable_payload(), 0, capacity); // Set the type of this message as REQUEST_PORT_MERGE (6) *reinterpret_cast(message->mutable_payload()) = 6; // This short message should be ignored. remote_channel->SendChannelMessage(std::move(message)); remote_channel->AcceptInvitee(kRemoteNodeName, kToken); loop.Run(); remote_channel->ShutDown(); local_channel->ShutDown(); } } // namespace } // namespace core } // namespace mojo