blob: 3e98cc51ce7746eb72c36e45b28d2b744861422e [file] [log] [blame]
// Copyright 2015 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 "remoting/protocol/quic_channel_factory.h"
#include "base/bind.h"
#include "base/message_loop/message_loop.h"
#include "base/run_loop.h"
#include "net/base/io_buffer.h"
#include "net/base/net_errors.h"
#include "net/base/test_completion_callback.h"
#include "net/quic/p2p/quic_p2p_session.h"
#include "net/quic/p2p/quic_p2p_stream.h"
#include "net/socket/socket.h"
#include "remoting/base/constants.h"
#include "remoting/protocol/connection_tester.h"
#include "remoting/protocol/fake_datagram_socket.h"
#include "remoting/protocol/p2p_stream_socket.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"
using testing::_;
using testing::AtMost;
using testing::InvokeWithoutArgs;
namespace remoting {
namespace protocol {
namespace {
const int kMessageSize = 1024;
const int kMessages = 100;
const char kTestSessionId[] = "123123";
const char kTestChannelName[] = "test";
const char kTestChannelName2[] = "test2";
} // namespace
class QuicChannelFactoryTest : public testing::Test,
public testing::WithParamInterface<bool> {
public:
void DeleteAll() {
host_channel1_.reset();
host_channel2_.reset();
client_channel1_.reset();
client_channel2_.reset();
host_quic_.reset();
client_quic_.reset();
}
void FailedReadDeleteAll(int result) {
EXPECT_NE(net::OK, result);
DeleteAll();
}
void OnChannelConnected(scoped_ptr<P2PStreamSocket>* storage,
int* counter,
base::RunLoop* run_loop,
scoped_ptr<P2PStreamSocket> socket) {
*storage = socket.Pass();
if (counter) {
--(*counter);
EXPECT_GE(*counter, 0);
if (*counter == 0)
run_loop->Quit();
}
}
void OnChannelConnectedExpectFail(scoped_ptr<P2PStreamSocket> socket) {
EXPECT_FALSE(socket);
host_quic_->CancelChannelCreation(kTestChannelName2);
DeleteAll();
}
void OnChannelConnectedNotReached(scoped_ptr<P2PStreamSocket> socket) {
NOTREACHED();
}
protected:
void TearDown() override {
DeleteAll();
// QuicChannelFactory destroys the internals asynchronously. Run all pending
// tasks to avoid leaking memory.
base::RunLoop().RunUntilIdle();
}
void Initialize() {
host_base_channel_factory_.PairWith(&client_base_channel_factory_);
host_base_channel_factory_.set_asynchronous_create(GetParam());
client_base_channel_factory_.set_asynchronous_create(GetParam());
host_quic_.reset(new QuicChannelFactory(kTestSessionId, true));
client_quic_.reset(new QuicChannelFactory(kTestSessionId, false));
session_initiate_message_ =
client_quic_->CreateSessionInitiateConfigMessage();
EXPECT_TRUE(host_quic_->ProcessSessionInitiateConfigMessage(
session_initiate_message_));
session_accept_message_ = host_quic_->CreateSessionAcceptConfigMessage();
EXPECT_TRUE(client_quic_->ProcessSessionAcceptConfigMessage(
session_accept_message_));
const char kTestSharedSecret[] = "Shared Secret";
host_quic_->Start(&host_base_channel_factory_, kTestSharedSecret);
client_quic_->Start(&client_base_channel_factory_, kTestSharedSecret);
FakeDatagramSocket* host_base_channel =
host_base_channel_factory_.GetFakeChannel(kQuicChannelName);
if (host_base_channel)
host_base_channel->set_async_send(GetParam());
FakeDatagramSocket* client_base_channel =
client_base_channel_factory_.GetFakeChannel(kQuicChannelName);
if (client_base_channel)
client_base_channel->set_async_send(GetParam());
}
void CreateChannel(const std::string& name,
scoped_ptr<P2PStreamSocket>* host_channel,
scoped_ptr<P2PStreamSocket>* client_channel) {
int counter = 2;
base::RunLoop run_loop;
host_quic_->CreateChannel(
name,
base::Bind(&QuicChannelFactoryTest::OnChannelConnected,
base::Unretained(this), host_channel, &counter, &run_loop));
client_quic_->CreateChannel(
name, base::Bind(&QuicChannelFactoryTest::OnChannelConnected,
base::Unretained(this), client_channel, &counter,
&run_loop));
run_loop.Run();
EXPECT_TRUE(host_channel->get());
EXPECT_TRUE(client_channel->get());
}
scoped_refptr<net::IOBufferWithSize> CreateTestBuffer(int size) {
scoped_refptr<net::IOBufferWithSize> result =
new net::IOBufferWithSize(size);
for (int i = 0; i < size; ++i) {
result->data()[i] = rand() % 256;
}
return result;
}
base::MessageLoop message_loop_;
FakeDatagramChannelFactory host_base_channel_factory_;
FakeDatagramChannelFactory client_base_channel_factory_;
scoped_ptr<QuicChannelFactory> host_quic_;
scoped_ptr<QuicChannelFactory> client_quic_;
scoped_ptr<P2PStreamSocket> host_channel1_;
scoped_ptr<P2PStreamSocket> client_channel1_;
scoped_ptr<P2PStreamSocket> host_channel2_;
scoped_ptr<P2PStreamSocket> client_channel2_;
std::string session_initiate_message_;
std::string session_accept_message_;
};
INSTANTIATE_TEST_CASE_P(SyncWrite,
QuicChannelFactoryTest,
::testing::Values(false));
INSTANTIATE_TEST_CASE_P(AsyncWrite,
QuicChannelFactoryTest,
::testing::Values(true));
TEST_P(QuicChannelFactoryTest, OneChannel) {
Initialize();
scoped_ptr<P2PStreamSocket> host_channel;
scoped_ptr<P2PStreamSocket> client_channel;
ASSERT_NO_FATAL_FAILURE(
CreateChannel(kTestChannelName, &host_channel, &client_channel));
StreamConnectionTester tester(host_channel.get(), client_channel.get(),
kMessageSize, kMessages);
tester.Start();
message_loop_.Run();
tester.CheckResults();
}
TEST_P(QuicChannelFactoryTest, TwoChannels) {
Initialize();
scoped_ptr<P2PStreamSocket> host_channel1_;
scoped_ptr<P2PStreamSocket> client_channel1_;
ASSERT_NO_FATAL_FAILURE(
CreateChannel(kTestChannelName, &host_channel1_, &client_channel1_));
scoped_ptr<P2PStreamSocket> host_channel2_;
scoped_ptr<P2PStreamSocket> client_channel2_;
ASSERT_NO_FATAL_FAILURE(
CreateChannel(kTestChannelName2, &host_channel2_, &client_channel2_));
StreamConnectionTester tester1(host_channel1_.get(), client_channel1_.get(),
kMessageSize, kMessages);
StreamConnectionTester tester2(host_channel2_.get(), client_channel2_.get(),
kMessageSize, kMessages);
tester1.Start();
tester2.Start();
while (!tester1.done() || !tester2.done()) {
message_loop_.Run();
}
tester1.CheckResults();
tester2.CheckResults();
}
TEST_P(QuicChannelFactoryTest, SendFail) {
Initialize();
scoped_ptr<P2PStreamSocket> host_channel1_;
scoped_ptr<P2PStreamSocket> client_channel1_;
ASSERT_NO_FATAL_FAILURE(
CreateChannel(kTestChannelName, &host_channel1_, &client_channel1_));
scoped_ptr<P2PStreamSocket> host_channel2_;
scoped_ptr<P2PStreamSocket> client_channel2_;
ASSERT_NO_FATAL_FAILURE(
CreateChannel(kTestChannelName2, &host_channel2_, &client_channel2_));
host_base_channel_factory_.GetFakeChannel(kQuicChannelName)
->set_next_send_error(net::ERR_FAILED);
scoped_refptr<net::IOBufferWithSize> buf = CreateTestBuffer(100);
// Try writing to a channel. This should result in all stream being closed due
// to an error.
{
net::TestCompletionCallback write_cb_1;
host_channel1_->Write(buf.get(), buf->size(), write_cb_1.callback());
base::RunLoop().RunUntilIdle();
}
// Repeated attempt to write should result in an error.
{
net::TestCompletionCallback write_cb_1;
net::TestCompletionCallback write_cb_2;
EXPECT_NE(net::OK, host_channel1_->Write(buf.get(), buf->size(),
write_cb_1.callback()));
EXPECT_FALSE(write_cb_1.have_result());
EXPECT_NE(net::OK, host_channel1_->Write(buf.get(), buf->size(),
write_cb_2.callback()));
EXPECT_FALSE(write_cb_2.have_result());
}
}
TEST_P(QuicChannelFactoryTest, DeleteWhenFailed) {
Initialize();
ASSERT_NO_FATAL_FAILURE(
CreateChannel(kTestChannelName, &host_channel1_, &client_channel1_));
ASSERT_NO_FATAL_FAILURE(
CreateChannel(kTestChannelName2, &host_channel2_, &client_channel2_));
host_base_channel_factory_.GetFakeChannel(kQuicChannelName)
->set_next_send_error(net::ERR_FAILED);
scoped_refptr<net::IOBufferWithSize> read_buf =
new net::IOBufferWithSize(100);
EXPECT_EQ(net::ERR_IO_PENDING,
host_channel1_->Read(
read_buf.get(), read_buf->size(),
base::Bind(&QuicChannelFactoryTest::FailedReadDeleteAll,
base::Unretained(this))));
// Try writing to a channel. This should result it DeleteAll() called and the
// connection torn down.
scoped_refptr<net::IOBufferWithSize> buf = CreateTestBuffer(100);
net::TestCompletionCallback write_cb_1;
host_channel1_->Write(buf.get(), buf->size(), write_cb_1.callback());
base::RunLoop().RunUntilIdle();
// Check that the connection was torn down.
EXPECT_FALSE(host_quic_);
}
TEST_P(QuicChannelFactoryTest, SessionFail) {
host_base_channel_factory_.set_fail_create(true);
Initialize();
host_quic_->CreateChannel(
kTestChannelName,
base::Bind(&QuicChannelFactoryTest::OnChannelConnectedExpectFail,
base::Unretained(this)));
// host_quic_ may be destroyed at this point in sync mode.
if (host_quic_) {
host_quic_->CreateChannel(
kTestChannelName2,
base::Bind(&QuicChannelFactoryTest::OnChannelConnectedNotReached,
base::Unretained(this)));
}
base::RunLoop().RunUntilIdle();
// Check that DeleteAll() was called and the connection was torn down.
EXPECT_FALSE(host_quic_);
}
// Verify that the host just ignores incoming stream with unexpected name.
TEST_P(QuicChannelFactoryTest, UnknownName) {
Initialize();
// Create a new channel from the client side.
client_quic_->CreateChannel(
kTestChannelName, base::Bind(&QuicChannelFactoryTest::OnChannelConnected,
base::Unretained(this), &client_channel1_,
nullptr, nullptr));
base::RunLoop().RunUntilIdle();
EXPECT_EQ(0U, host_quic_->GetP2PSessionForTests()->GetNumOpenStreams());
}
// Verify that incoming streams that have received only partial name are
// destroyed correctly.
TEST_P(QuicChannelFactoryTest, SendPartialName) {
Initialize();
base::RunLoop().RunUntilIdle();
net::QuicP2PSession* session = client_quic_->GetP2PSessionForTests();
net::QuicP2PStream* stream = session->CreateOutgoingDynamicStream();
std::string name = kTestChannelName;
// Send only half of the name to the host.
stream->WriteHeader(std::string(1, static_cast<char>(name.size())) +
name.substr(0, name.size() / 2));
base::RunLoop().RunUntilIdle();
// Host should have received the new stream and is still waiting for the name.
EXPECT_EQ(1U, host_quic_->GetP2PSessionForTests()->GetNumOpenStreams());
session->CloseStream(stream->id());
base::RunLoop().RunUntilIdle();
// Verify that the stream was closed on the host side.
EXPECT_EQ(0U, host_quic_->GetP2PSessionForTests()->GetNumOpenStreams());
// Create another stream with only partial name and tear down connection while
// it's still pending.
stream = session->CreateOutgoingDynamicStream();
stream->WriteHeader(std::string(1, static_cast<char>(name.size())) +
name.substr(0, name.size() / 2));
base::RunLoop().RunUntilIdle();
EXPECT_EQ(1U, host_quic_->GetP2PSessionForTests()->GetNumOpenStreams());
}
// Verify that correct HKDF input suffix is used to generate encryption keys.
TEST_P(QuicChannelFactoryTest, HkdfInputSuffix) {
Initialize();
base::RunLoop().RunUntilIdle();
net::QuicCryptoStream* crypto_stream =
reinterpret_cast<net::QuicCryptoStream*>(
host_quic_->GetP2PSessionForTests()->GetStream(net::kCryptoStreamId));
const std::string& suffix =
crypto_stream->crypto_negotiated_params().hkdf_input_suffix;
EXPECT_EQ(strlen(kTestSessionId) + 1 + strlen(kQuicChannelName) + 1 +
session_initiate_message_.size() +
session_accept_message_.size(),
suffix.size());
EXPECT_EQ(std::string(kTestSessionId) + '\0' + kQuicChannelName + '\0' +
session_initiate_message_ + session_accept_message_,
suffix);
}
} // namespace protocol
} // namespace remoting