Index: remoting/protocol/quic_channel_factory_unittest.cc |
diff --git a/remoting/protocol/quic_channel_factory_unittest.cc b/remoting/protocol/quic_channel_factory_unittest.cc |
deleted file mode 100644 |
index 3e98cc51ce7746eb72c36e45b28d2b744861422e..0000000000000000000000000000000000000000 |
--- a/remoting/protocol/quic_channel_factory_unittest.cc |
+++ /dev/null |
@@ -1,379 +0,0 @@ |
-// 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 |