| // Copyright (c) 2012 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 "net/third_party/quic/core/quic_session.h" |
| |
| #include <cstdint> |
| #include <set> |
| #include <utility> |
| |
| #include "net/third_party/quic/core/crypto/crypto_protocol.h" |
| #include "net/third_party/quic/core/crypto/null_encrypter.h" |
| #include "net/third_party/quic/core/quic_crypto_stream.h" |
| #include "net/third_party/quic/core/quic_data_writer.h" |
| #include "net/third_party/quic/core/quic_error_codes.h" |
| #include "net/third_party/quic/core/quic_packets.h" |
| #include "net/third_party/quic/core/quic_stream.h" |
| #include "net/third_party/quic/core/quic_utils.h" |
| #include "net/third_party/quic/platform/api/quic_arraysize.h" |
| #include "net/third_party/quic/platform/api/quic_expect_bug.h" |
| #include "net/third_party/quic/platform/api/quic_flags.h" |
| #include "net/third_party/quic/platform/api/quic_map_util.h" |
| #include "net/third_party/quic/platform/api/quic_mem_slice_storage.h" |
| #include "net/third_party/quic/platform/api/quic_ptr_util.h" |
| #include "net/third_party/quic/platform/api/quic_str_cat.h" |
| #include "net/third_party/quic/platform/api/quic_string.h" |
| #include "net/third_party/quic/platform/api/quic_string_piece.h" |
| #include "net/third_party/quic/platform/api/quic_test.h" |
| #include "net/third_party/quic/platform/api/quic_test_mem_slice_vector.h" |
| #include "net/third_party/quic/test_tools/mock_quic_session_visitor.h" |
| #include "net/third_party/quic/test_tools/quic_config_peer.h" |
| #include "net/third_party/quic/test_tools/quic_connection_peer.h" |
| #include "net/third_party/quic/test_tools/quic_flow_controller_peer.h" |
| #include "net/third_party/quic/test_tools/quic_session_peer.h" |
| #include "net/third_party/quic/test_tools/quic_stream_peer.h" |
| #include "net/third_party/quic/test_tools/quic_stream_send_buffer_peer.h" |
| #include "net/third_party/quic/test_tools/quic_test_utils.h" |
| #include "testing/gmock/include/gmock/gmock.h" |
| |
| using spdy::kV3HighestPriority; |
| using spdy::SpdyPriority; |
| using testing::_; |
| using testing::AtLeast; |
| using testing::InSequence; |
| using testing::Invoke; |
| using testing::NiceMock; |
| using testing::Return; |
| using testing::StrictMock; |
| using testing::WithArg; |
| |
| namespace quic { |
| namespace test { |
| namespace { |
| |
| class TestCryptoStream : public QuicCryptoStream, public QuicCryptoHandshaker { |
| public: |
| explicit TestCryptoStream(QuicSession* session) |
| : QuicCryptoStream(session), |
| QuicCryptoHandshaker(this, session), |
| encryption_established_(false), |
| handshake_confirmed_(false), |
| params_(new QuicCryptoNegotiatedParameters) {} |
| |
| void OnHandshakeMessage(const CryptoHandshakeMessage& /*message*/) override { |
| encryption_established_ = true; |
| handshake_confirmed_ = true; |
| CryptoHandshakeMessage msg; |
| QuicString error_details; |
| session()->config()->SetInitialStreamFlowControlWindowToSend( |
| kInitialStreamFlowControlWindowForTest); |
| session()->config()->SetInitialSessionFlowControlWindowToSend( |
| kInitialSessionFlowControlWindowForTest); |
| session()->config()->ToHandshakeMessage(&msg); |
| const QuicErrorCode error = |
| session()->config()->ProcessPeerHello(msg, CLIENT, &error_details); |
| EXPECT_EQ(QUIC_NO_ERROR, error); |
| session()->OnConfigNegotiated(); |
| session()->connection()->SetDefaultEncryptionLevel( |
| ENCRYPTION_FORWARD_SECURE); |
| session()->OnCryptoHandshakeEvent(QuicSession::HANDSHAKE_CONFIRMED); |
| } |
| |
| // QuicCryptoStream implementation |
| bool encryption_established() const override { |
| return encryption_established_; |
| } |
| bool handshake_confirmed() const override { return handshake_confirmed_; } |
| const QuicCryptoNegotiatedParameters& crypto_negotiated_params() |
| const override { |
| return *params_; |
| } |
| CryptoMessageParser* crypto_message_parser() override { |
| return QuicCryptoHandshaker::crypto_message_parser(); |
| } |
| |
| MOCK_METHOD0(OnCanWrite, void()); |
| bool HasPendingCryptoRetransmission() override { return false; } |
| |
| MOCK_CONST_METHOD0(HasPendingRetransmission, bool()); |
| |
| private: |
| using QuicCryptoStream::session; |
| |
| bool encryption_established_; |
| bool handshake_confirmed_; |
| QuicReferenceCountedPointer<QuicCryptoNegotiatedParameters> params_; |
| }; |
| |
| class TestStream : public QuicStream { |
| public: |
| TestStream(QuicStreamId id, QuicSession* session, StreamType type) |
| : QuicStream(id, session, /*is_static=*/false, type) {} |
| |
| TestStream(PendingStream pending, StreamType type) |
| : QuicStream(std::move(pending), type) {} |
| |
| using QuicStream::CloseReadSide; |
| using QuicStream::CloseWriteSide; |
| using QuicStream::WriteMemSlices; |
| using QuicStream::WritevData; |
| |
| void OnDataAvailable() override {} |
| |
| MOCK_METHOD0(OnCanWrite, void()); |
| MOCK_METHOD3(RetransmitStreamData, |
| bool(QuicStreamOffset, QuicByteCount, bool)); |
| |
| MOCK_CONST_METHOD0(HasPendingRetransmission, bool()); |
| MOCK_METHOD1(OnStopSending, void(uint16_t code)); |
| }; |
| |
| class TestSession : public QuicSession { |
| public: |
| explicit TestSession(QuicConnection* connection, |
| MockQuicSessionVisitor* session_visitor) |
| : QuicSession(connection, |
| session_visitor, |
| DefaultQuicConfig(), |
| CurrentSupportedVersions()), |
| crypto_stream_(this), |
| writev_consumes_all_data_(false), |
| should_buffer_incoming_streams_(false), |
| num_incoming_streams_created_(0) { |
| Initialize(); |
| this->connection()->SetEncrypter( |
| ENCRYPTION_FORWARD_SECURE, |
| QuicMakeUnique<NullEncrypter>(connection->perspective())); |
| } |
| |
| ~TestSession() override { delete connection(); } |
| |
| TestCryptoStream* GetMutableCryptoStream() override { |
| return &crypto_stream_; |
| } |
| |
| const TestCryptoStream* GetCryptoStream() const override { |
| return &crypto_stream_; |
| } |
| |
| TestStream* CreateOutgoingBidirectionalStream() { |
| QuicStreamId id = GetNextOutgoingBidirectionalStreamId(); |
| if (id == |
| QuicUtils::GetInvalidStreamId(connection()->transport_version())) { |
| return nullptr; |
| } |
| TestStream* stream = new TestStream(id, this, BIDIRECTIONAL); |
| ActivateStream(QuicWrapUnique(stream)); |
| return stream; |
| } |
| |
| TestStream* CreateOutgoingUnidirectionalStream() { |
| TestStream* stream = new TestStream(GetNextOutgoingUnidirectionalStreamId(), |
| this, WRITE_UNIDIRECTIONAL); |
| ActivateStream(QuicWrapUnique(stream)); |
| return stream; |
| } |
| |
| TestStream* CreateIncomingStream(QuicStreamId id) override { |
| // Enforce the limit on the number of open streams. |
| if (GetNumOpenIncomingStreams() + 1 > |
| max_open_incoming_bidirectional_streams() && |
| connection()->transport_version() != QUIC_VERSION_99) { |
| // No need to do this test for version 99; it's done by |
| // QuicSession::GetOrCreateDynamicStream. |
| connection()->CloseConnection( |
| QUIC_TOO_MANY_OPEN_STREAMS, "Too many streams!", |
| ConnectionCloseBehavior::SEND_CONNECTION_CLOSE_PACKET); |
| return nullptr; |
| } |
| |
| TestStream* stream = |
| new TestStream(id, this, |
| DetermineStreamType( |
| id, connection()->transport_version(), perspective(), |
| /*is_incoming=*/true, BIDIRECTIONAL)); |
| ActivateStream(QuicWrapUnique(stream)); |
| ++num_incoming_streams_created_; |
| return stream; |
| } |
| |
| TestStream* CreateIncomingStream(PendingStream pending) override { |
| QuicStreamId id = pending.id(); |
| TestStream* stream = |
| new TestStream(std::move(pending), |
| DetermineStreamType( |
| id, connection()->transport_version(), perspective(), |
| /*is_incoming=*/true, BIDIRECTIONAL)); |
| ActivateStream(QuicWrapUnique(stream)); |
| ++num_incoming_streams_created_; |
| return stream; |
| } |
| |
| bool IsClosedStream(QuicStreamId id) { |
| return QuicSession::IsClosedStream(id); |
| } |
| |
| QuicStream* GetOrCreateDynamicStream(QuicStreamId stream_id) { |
| return QuicSession::GetOrCreateDynamicStream(stream_id); |
| } |
| |
| bool ShouldKeepConnectionAlive() const override { |
| return GetNumOpenDynamicStreams() > 0; |
| } |
| |
| QuicConsumedData WritevData(QuicStream* stream, |
| QuicStreamId id, |
| size_t write_length, |
| QuicStreamOffset offset, |
| StreamSendingState state) override { |
| bool fin = state != NO_FIN; |
| QuicConsumedData consumed(write_length, fin); |
| if (!writev_consumes_all_data_) { |
| consumed = |
| QuicSession::WritevData(stream, id, write_length, offset, state); |
| } |
| if (fin && consumed.fin_consumed) { |
| stream->set_fin_sent(true); |
| } |
| QuicSessionPeer::GetWriteBlockedStreams(this)->UpdateBytesForStream( |
| id, consumed.bytes_consumed); |
| return consumed; |
| } |
| |
| MOCK_METHOD0(OnCanCreateNewOutgoingStream, void()); |
| |
| void set_writev_consumes_all_data(bool val) { |
| writev_consumes_all_data_ = val; |
| } |
| |
| QuicConsumedData SendStreamData(QuicStream* stream) { |
| struct IOVEC iov; |
| if (stream->id() != |
| QuicUtils::GetCryptoStreamId(connection()->transport_version()) && |
| this->connection()->encryption_level() != ENCRYPTION_FORWARD_SECURE) { |
| this->connection()->SetDefaultEncryptionLevel(ENCRYPTION_FORWARD_SECURE); |
| } |
| MakeIOVector("not empty", &iov); |
| QuicStreamPeer::SendBuffer(stream).SaveStreamData(&iov, 1, 0, 9); |
| QuicConsumedData consumed = WritevData(stream, stream->id(), 9, 0, FIN); |
| QuicStreamPeer::SendBuffer(stream).OnStreamDataConsumed( |
| consumed.bytes_consumed); |
| return consumed; |
| } |
| |
| bool ClearControlFrame(const QuicFrame& frame) { |
| DeleteFrame(&const_cast<QuicFrame&>(frame)); |
| return true; |
| } |
| |
| bool SaveFrame(const QuicFrame& frame) { |
| save_frame_ = frame; |
| DeleteFrame(&const_cast<QuicFrame&>(frame)); |
| return true; |
| } |
| |
| const QuicFrame& save_frame() { return save_frame_; } |
| |
| QuicConsumedData SendLargeFakeData(QuicStream* stream, int bytes) { |
| DCHECK(writev_consumes_all_data_); |
| return WritevData(stream, stream->id(), bytes, 0, FIN); |
| } |
| |
| bool ShouldBufferIncomingStream(QuicStreamId id) const override { |
| return should_buffer_incoming_streams_; |
| } |
| |
| void set_should_buffer_incoming_streams(bool should_buffer_incoming_streams) { |
| should_buffer_incoming_streams_ = should_buffer_incoming_streams; |
| } |
| |
| int num_incoming_streams_created() const { |
| return num_incoming_streams_created_; |
| } |
| |
| using QuicSession::ActivateStream; |
| using QuicSession::closed_streams; |
| using QuicSession::zombie_streams; |
| |
| private: |
| StrictMock<TestCryptoStream> crypto_stream_; |
| |
| bool writev_consumes_all_data_; |
| bool should_buffer_incoming_streams_; |
| QuicFrame save_frame_; |
| int num_incoming_streams_created_; |
| }; |
| |
| class QuicSessionTestBase : public QuicTestWithParam<ParsedQuicVersion> { |
| protected: |
| explicit QuicSessionTestBase(Perspective perspective) |
| : connection_( |
| new StrictMock<MockQuicConnection>(&helper_, |
| &alarm_factory_, |
| perspective, |
| SupportedVersions(GetParam()))), |
| session_(connection_, &session_visitor_) { |
| session_.config()->SetInitialStreamFlowControlWindowToSend( |
| kInitialStreamFlowControlWindowForTest); |
| session_.config()->SetInitialSessionFlowControlWindowToSend( |
| kInitialSessionFlowControlWindowForTest); |
| connection_->AdvanceTime(QuicTime::Delta::FromSeconds(1)); |
| TestCryptoStream* crypto_stream = session_.GetMutableCryptoStream(); |
| EXPECT_CALL(*crypto_stream, HasPendingRetransmission()) |
| .Times(testing::AnyNumber()); |
| } |
| |
| void CheckClosedStreams() { |
| for (QuicStreamId i = |
| QuicUtils::GetCryptoStreamId(connection_->transport_version()); |
| i < 100; i++) { |
| if (!QuicContainsKey(closed_streams_, i)) { |
| EXPECT_FALSE(session_.IsClosedStream(i)) << " stream id: " << i; |
| } else { |
| EXPECT_TRUE(session_.IsClosedStream(i)) << " stream id: " << i; |
| } |
| } |
| } |
| |
| void CloseStream(QuicStreamId id) { |
| if (session_.connection()->transport_version() == QUIC_VERSION_99 && |
| QuicUtils::GetStreamType(id, session_.perspective(), |
| session_.IsIncomingStream(id)) == |
| READ_UNIDIRECTIONAL) { |
| // Verify reset is not sent for READ_UNIDIRECTIONAL streams. |
| EXPECT_CALL(*connection_, SendControlFrame(_)).Times(0); |
| EXPECT_CALL(*connection_, OnStreamReset(_, _)).Times(0); |
| } else { |
| // Verify reset IS sent for BIDIRECTIONAL streams. |
| if (session_.connection()->transport_version() == QUIC_VERSION_99) { |
| // Once for the RST_STREAM, Once for the STOP_SENDING |
| EXPECT_CALL(*connection_, SendControlFrame(_)) |
| .Times(2) |
| .WillRepeatedly(Invoke(&session_, &TestSession::ClearControlFrame)); |
| } else { |
| EXPECT_CALL(*connection_, SendControlFrame(_)) |
| .WillOnce(Invoke(&session_, &TestSession::ClearControlFrame)); |
| } |
| EXPECT_CALL(*connection_, OnStreamReset(id, _)); |
| } |
| session_.CloseStream(id); |
| closed_streams_.insert(id); |
| } |
| |
| QuicTransportVersion transport_version() const { |
| return connection_->transport_version(); |
| } |
| |
| QuicStreamId GetNthClientInitiatedBidirectionalId(int n) { |
| return QuicUtils::GetFirstBidirectionalStreamId( |
| connection_->transport_version(), Perspective::IS_CLIENT) + |
| QuicUtils::StreamIdDelta(connection_->transport_version()) * n; |
| } |
| |
| QuicStreamId GetNthClientInitiatedUnidirectionalId(int n) { |
| return QuicUtils::GetFirstUnidirectionalStreamId( |
| connection_->transport_version(), Perspective::IS_CLIENT) + |
| QuicUtils::StreamIdDelta(connection_->transport_version()) * n; |
| } |
| |
| QuicStreamId GetNthServerInitiatedBidirectionalId(int n) { |
| return QuicUtils::GetFirstBidirectionalStreamId( |
| connection_->transport_version(), Perspective::IS_SERVER) + |
| QuicUtils::StreamIdDelta(connection_->transport_version()) * n; |
| } |
| |
| QuicStreamId GetNthServerInitiatedUnidirectionalId(int n) { |
| return QuicUtils::GetFirstUnidirectionalStreamId( |
| connection_->transport_version(), Perspective::IS_SERVER) + |
| QuicUtils::StreamIdDelta(connection_->transport_version()) * n; |
| } |
| |
| MockQuicConnectionHelper helper_; |
| MockAlarmFactory alarm_factory_; |
| NiceMock<MockQuicSessionVisitor> session_visitor_; |
| StrictMock<MockQuicConnection>* connection_; |
| TestSession session_; |
| std::set<QuicStreamId> closed_streams_; |
| }; |
| |
| class QuicSessionTestServer : public QuicSessionTestBase { |
| public: |
| // CheckMultiPathResponse validates that a written packet |
| // contains both expected path responses. |
| WriteResult CheckMultiPathResponse(const char* buffer, |
| size_t buf_len, |
| const QuicIpAddress& self_address, |
| const QuicSocketAddress& peer_address, |
| PerPacketOptions* options) { |
| QuicEncryptedPacket packet(buffer, buf_len); |
| { |
| InSequence s; |
| EXPECT_CALL(framer_visitor_, OnPacket()); |
| EXPECT_CALL(framer_visitor_, OnUnauthenticatedPublicHeader(_)); |
| EXPECT_CALL(framer_visitor_, OnUnauthenticatedHeader(_)); |
| EXPECT_CALL(framer_visitor_, OnDecryptedPacket(_)); |
| EXPECT_CALL(framer_visitor_, OnPacketHeader(_)); |
| EXPECT_CALL(framer_visitor_, OnPathResponseFrame(_)) |
| .WillOnce( |
| WithArg<0>(Invoke([this](const QuicPathResponseFrame& frame) { |
| EXPECT_EQ(path_frame_buffer1_, frame.data_buffer); |
| return true; |
| }))); |
| EXPECT_CALL(framer_visitor_, OnPathResponseFrame(_)) |
| .WillOnce( |
| WithArg<0>(Invoke([this](const QuicPathResponseFrame& frame) { |
| EXPECT_EQ(path_frame_buffer2_, frame.data_buffer); |
| return true; |
| }))); |
| EXPECT_CALL(framer_visitor_, OnPacketComplete()); |
| } |
| client_framer_.ProcessPacket(packet); |
| return WriteResult(WRITE_STATUS_OK, 0); |
| } |
| |
| protected: |
| QuicSessionTestServer() |
| : QuicSessionTestBase(Perspective::IS_SERVER), |
| path_frame_buffer1_({0, 1, 2, 3, 4, 5, 6, 7}), |
| path_frame_buffer2_({8, 9, 10, 11, 12, 13, 14, 15}), |
| client_framer_(SupportedVersions(GetParam()), |
| QuicTime::Zero(), |
| Perspective::IS_CLIENT) { |
| client_framer_.set_visitor(&framer_visitor_); |
| } |
| |
| QuicPathFrameBuffer path_frame_buffer1_; |
| QuicPathFrameBuffer path_frame_buffer2_; |
| StrictMock<MockFramerVisitor> framer_visitor_; |
| // Framer used to process packets sent by server. |
| QuicFramer client_framer_; |
| }; |
| |
| INSTANTIATE_TEST_SUITE_P(Tests, |
| QuicSessionTestServer, |
| ::testing::ValuesIn(AllSupportedVersions())); |
| |
| TEST_P(QuicSessionTestServer, PeerAddress) { |
| EXPECT_EQ(QuicSocketAddress(QuicIpAddress::Loopback4(), kTestPort), |
| session_.peer_address()); |
| } |
| |
| TEST_P(QuicSessionTestServer, SelfAddress) { |
| EXPECT_TRUE(session_.self_address().IsInitialized()); |
| } |
| |
| TEST_P(QuicSessionTestServer, DontCallOnWriteBlockedForDisconnectedConnection) { |
| EXPECT_CALL(*connection_, CloseConnection(_, _, _)) |
| .WillOnce( |
| Invoke(connection_, &MockQuicConnection::ReallyCloseConnection)); |
| connection_->CloseConnection(QUIC_NO_ERROR, "Everything is fine.", |
| ConnectionCloseBehavior::SILENT_CLOSE); |
| ASSERT_FALSE(connection_->connected()); |
| |
| if (GetQuicReloadableFlag( |
| quic_connection_do_not_add_to_write_blocked_list_if_disconnected)) { |
| EXPECT_CALL(session_visitor_, OnWriteBlocked(_)).Times(0); |
| } else { |
| EXPECT_CALL(session_visitor_, OnWriteBlocked(_)).Times(1); |
| } |
| session_.OnWriteBlocked(); |
| } |
| |
| TEST_P(QuicSessionTestServer, IsCryptoHandshakeConfirmed) { |
| EXPECT_FALSE(session_.IsCryptoHandshakeConfirmed()); |
| CryptoHandshakeMessage message; |
| session_.GetMutableCryptoStream()->OnHandshakeMessage(message); |
| EXPECT_TRUE(session_.IsCryptoHandshakeConfirmed()); |
| } |
| |
| TEST_P(QuicSessionTestServer, IsClosedStreamDefault) { |
| // Ensure that no streams are initially closed. |
| for (QuicStreamId i = |
| QuicUtils::GetCryptoStreamId(connection_->transport_version()); |
| i < 100; i++) { |
| EXPECT_FALSE(session_.IsClosedStream(i)) << "stream id: " << i; |
| } |
| } |
| |
| TEST_P(QuicSessionTestServer, AvailableBidirectionalStreams) { |
| ASSERT_TRUE(session_.GetOrCreateDynamicStream( |
| GetNthClientInitiatedBidirectionalId(3)) != nullptr); |
| // Smaller bidirectional streams should be available. |
| EXPECT_TRUE(QuicSessionPeer::IsStreamAvailable( |
| &session_, GetNthClientInitiatedBidirectionalId(1))); |
| EXPECT_TRUE(QuicSessionPeer::IsStreamAvailable( |
| &session_, GetNthClientInitiatedBidirectionalId(2))); |
| ASSERT_TRUE(session_.GetOrCreateDynamicStream( |
| GetNthClientInitiatedBidirectionalId(2)) != nullptr); |
| ASSERT_TRUE(session_.GetOrCreateDynamicStream( |
| GetNthClientInitiatedBidirectionalId(1)) != nullptr); |
| } |
| |
| TEST_P(QuicSessionTestServer, AvailableUnidirectionalStreams) { |
| ASSERT_TRUE(session_.GetOrCreateDynamicStream( |
| GetNthClientInitiatedUnidirectionalId(3)) != nullptr); |
| // Smaller unidirectional streams should be available. |
| EXPECT_TRUE(QuicSessionPeer::IsStreamAvailable( |
| &session_, GetNthClientInitiatedUnidirectionalId(1))); |
| EXPECT_TRUE(QuicSessionPeer::IsStreamAvailable( |
| &session_, GetNthClientInitiatedUnidirectionalId(2))); |
| ASSERT_TRUE(session_.GetOrCreateDynamicStream( |
| GetNthClientInitiatedUnidirectionalId(2)) != nullptr); |
| ASSERT_TRUE(session_.GetOrCreateDynamicStream( |
| GetNthClientInitiatedUnidirectionalId(1)) != nullptr); |
| } |
| |
| TEST_P(QuicSessionTestServer, MaxAvailableBidirectionalStreams) { |
| if (transport_version() == QUIC_VERSION_99) { |
| EXPECT_EQ(session_.max_open_incoming_bidirectional_streams(), |
| session_.MaxAvailableBidirectionalStreams()); |
| } else { |
| // The protocol specification requires that there can be at least 10 times |
| // as many available streams as the connection's maximum open streams. |
| EXPECT_EQ(session_.max_open_incoming_bidirectional_streams() * |
| kMaxAvailableStreamsMultiplier, |
| session_.MaxAvailableBidirectionalStreams()); |
| } |
| } |
| |
| TEST_P(QuicSessionTestServer, MaxAvailableUnidirectionalStreams) { |
| if (transport_version() == QUIC_VERSION_99) { |
| EXPECT_EQ(session_.max_open_incoming_unidirectional_streams(), |
| session_.MaxAvailableUnidirectionalStreams()); |
| } else { |
| // The protocol specification requires that there can be at least 10 times |
| // as many available streams as the connection's maximum open streams. |
| EXPECT_EQ(session_.max_open_incoming_unidirectional_streams() * |
| kMaxAvailableStreamsMultiplier, |
| session_.MaxAvailableUnidirectionalStreams()); |
| } |
| } |
| |
| TEST_P(QuicSessionTestServer, IsClosedBidirectionalStreamLocallyCreated) { |
| TestStream* stream2 = session_.CreateOutgoingBidirectionalStream(); |
| EXPECT_EQ(GetNthServerInitiatedBidirectionalId(0), stream2->id()); |
| TestStream* stream4 = session_.CreateOutgoingBidirectionalStream(); |
| EXPECT_EQ(GetNthServerInitiatedBidirectionalId(1), stream4->id()); |
| |
| CheckClosedStreams(); |
| CloseStream(GetNthServerInitiatedBidirectionalId(0)); |
| CheckClosedStreams(); |
| CloseStream(GetNthServerInitiatedBidirectionalId(1)); |
| CheckClosedStreams(); |
| } |
| |
| TEST_P(QuicSessionTestServer, IsClosedUnidirectionalStreamLocallyCreated) { |
| TestStream* stream2 = session_.CreateOutgoingUnidirectionalStream(); |
| EXPECT_EQ(GetNthServerInitiatedUnidirectionalId(0), stream2->id()); |
| TestStream* stream4 = session_.CreateOutgoingUnidirectionalStream(); |
| EXPECT_EQ(GetNthServerInitiatedUnidirectionalId(1), stream4->id()); |
| |
| CheckClosedStreams(); |
| CloseStream(GetNthServerInitiatedUnidirectionalId(0)); |
| CheckClosedStreams(); |
| CloseStream(GetNthServerInitiatedUnidirectionalId(1)); |
| CheckClosedStreams(); |
| } |
| |
| TEST_P(QuicSessionTestServer, IsClosedBidirectionalStreamPeerCreated) { |
| QuicStreamId stream_id1 = GetNthClientInitiatedBidirectionalId(0); |
| QuicStreamId stream_id2 = GetNthClientInitiatedBidirectionalId(1); |
| session_.GetOrCreateDynamicStream(stream_id1); |
| session_.GetOrCreateDynamicStream(stream_id2); |
| |
| CheckClosedStreams(); |
| CloseStream(stream_id1); |
| CheckClosedStreams(); |
| CloseStream(stream_id2); |
| // Create a stream, and make another available. |
| QuicStream* stream3 = session_.GetOrCreateDynamicStream( |
| stream_id2 + |
| 2 * QuicUtils::StreamIdDelta(connection_->transport_version())); |
| CheckClosedStreams(); |
| // Close one, but make sure the other is still not closed |
| CloseStream(stream3->id()); |
| CheckClosedStreams(); |
| } |
| |
| TEST_P(QuicSessionTestServer, IsClosedUnidirectionalStreamPeerCreated) { |
| QuicStreamId stream_id1 = GetNthClientInitiatedUnidirectionalId(0); |
| QuicStreamId stream_id2 = GetNthClientInitiatedUnidirectionalId(1); |
| session_.GetOrCreateDynamicStream(stream_id1); |
| session_.GetOrCreateDynamicStream(stream_id2); |
| |
| CheckClosedStreams(); |
| CloseStream(stream_id1); |
| CheckClosedStreams(); |
| CloseStream(stream_id2); |
| // Create a stream, and make another available. |
| QuicStream* stream3 = session_.GetOrCreateDynamicStream( |
| stream_id2 + |
| 2 * QuicUtils::StreamIdDelta(connection_->transport_version())); |
| CheckClosedStreams(); |
| // Close one, but make sure the other is still not closed |
| CloseStream(stream3->id()); |
| CheckClosedStreams(); |
| } |
| |
| TEST_P(QuicSessionTestServer, MaximumAvailableOpenedBidirectionalStreams) { |
| QuicStreamId stream_id = GetNthClientInitiatedBidirectionalId(0); |
| session_.GetOrCreateDynamicStream(stream_id); |
| EXPECT_CALL(*connection_, CloseConnection(_, _, _)).Times(0); |
| EXPECT_NE( |
| nullptr, |
| session_.GetOrCreateDynamicStream(GetNthClientInitiatedBidirectionalId( |
| session_.max_open_incoming_bidirectional_streams() - 1))); |
| } |
| |
| TEST_P(QuicSessionTestServer, MaximumAvailableOpenedUnidirectionalStreams) { |
| QuicStreamId stream_id = GetNthClientInitiatedUnidirectionalId(0); |
| session_.GetOrCreateDynamicStream(stream_id); |
| EXPECT_CALL(*connection_, CloseConnection(_, _, _)).Times(0); |
| EXPECT_NE( |
| nullptr, |
| session_.GetOrCreateDynamicStream(GetNthClientInitiatedUnidirectionalId( |
| session_.max_open_incoming_unidirectional_streams() - 1))); |
| } |
| |
| TEST_P(QuicSessionTestServer, TooManyAvailableBidirectionalStreams) { |
| QuicStreamId stream_id1 = GetNthClientInitiatedBidirectionalId(0); |
| QuicStreamId stream_id2; |
| EXPECT_NE(nullptr, session_.GetOrCreateDynamicStream(stream_id1)); |
| // A stream ID which is too large to create. |
| stream_id2 = GetNthClientInitiatedBidirectionalId( |
| session_.MaxAvailableBidirectionalStreams() + 2); |
| if (transport_version() == QUIC_VERSION_99) { |
| // V99 terminates the connection with invalid stream id |
| EXPECT_CALL(*connection_, CloseConnection(QUIC_INVALID_STREAM_ID, _, _)); |
| } else { |
| // other versions terminate the connection with |
| // QUIC_TOO_MANY_AVAILABLE_STREAMS. |
| EXPECT_CALL(*connection_, |
| CloseConnection(QUIC_TOO_MANY_AVAILABLE_STREAMS, _, _)); |
| } |
| EXPECT_EQ(nullptr, session_.GetOrCreateDynamicStream(stream_id2)); |
| } |
| |
| TEST_P(QuicSessionTestServer, TooManyAvailableUnidirectionalStreams) { |
| QuicStreamId stream_id1 = GetNthClientInitiatedUnidirectionalId(0); |
| QuicStreamId stream_id2; |
| EXPECT_NE(nullptr, session_.GetOrCreateDynamicStream(stream_id1)); |
| // A stream ID which is too large to create. |
| stream_id2 = GetNthClientInitiatedUnidirectionalId( |
| session_.MaxAvailableUnidirectionalStreams() + 2); |
| if (transport_version() == QUIC_VERSION_99) { |
| // V99 terminates the connection with invalid stream id |
| EXPECT_CALL(*connection_, CloseConnection(QUIC_INVALID_STREAM_ID, _, _)); |
| } else { |
| // other versions terminate the connection with |
| // QUIC_TOO_MANY_AVAILABLE_STREAMS. |
| EXPECT_CALL(*connection_, |
| CloseConnection(QUIC_TOO_MANY_AVAILABLE_STREAMS, _, _)); |
| } |
| EXPECT_EQ(nullptr, session_.GetOrCreateDynamicStream(stream_id2)); |
| } |
| |
| TEST_P(QuicSessionTestServer, ManyAvailableBidirectionalStreams) { |
| // When max_open_streams_ is 200, should be able to create 200 streams |
| // out-of-order, that is, creating the one with the largest stream ID first. |
| QuicSessionPeer::SetMaxOpenIncomingStreams(&session_, 200); |
| QuicStreamId stream_id = GetNthClientInitiatedBidirectionalId(0); |
| // Create one stream. |
| EXPECT_NE(nullptr, session_.GetOrCreateDynamicStream(stream_id)); |
| EXPECT_CALL(*connection_, CloseConnection(_, _, _)).Times(0); |
| |
| // Create the largest stream ID of a threatened total of 200 streams. |
| // GetNth... starts at 0, so for 200 streams, get the 199th. |
| EXPECT_NE(nullptr, session_.GetOrCreateDynamicStream( |
| GetNthClientInitiatedBidirectionalId(199))); |
| } |
| |
| TEST_P(QuicSessionTestServer, ManyAvailableUnidirectionalStreams) { |
| // When max_open_streams_ is 200, should be able to create 200 streams |
| // out-of-order, that is, creating the one with the largest stream ID first. |
| QuicSessionPeer::SetMaxOpenIncomingStreams(&session_, 200); |
| QuicStreamId stream_id = GetNthClientInitiatedUnidirectionalId(0); |
| // Create one stream. |
| EXPECT_NE(nullptr, session_.GetOrCreateDynamicStream(stream_id)); |
| EXPECT_CALL(*connection_, CloseConnection(_, _, _)).Times(0); |
| |
| // Create the largest stream ID of a threatened total of 200 streams. |
| // GetNth... starts at 0, so for 200 streams, get the 199th. |
| EXPECT_NE(nullptr, session_.GetOrCreateDynamicStream( |
| GetNthClientInitiatedUnidirectionalId(199))); |
| } |
| |
| TEST_P(QuicSessionTestServer, DebugDFatalIfMarkingClosedStreamWriteBlocked) { |
| // EXPECT_QUIC_BUG tests are expensive so only run one instance of them. |
| if (GetParam() != AllSupportedVersions()[0]) { |
| return; |
| } |
| |
| TestStream* stream2 = session_.CreateOutgoingBidirectionalStream(); |
| QuicStreamId closed_stream_id = stream2->id(); |
| // Close the stream. |
| EXPECT_CALL(*connection_, SendControlFrame(_)); |
| EXPECT_CALL(*connection_, OnStreamReset(closed_stream_id, _)); |
| stream2->Reset(QUIC_BAD_APPLICATION_PAYLOAD); |
| QuicString msg = |
| QuicStrCat("Marking unknown stream ", closed_stream_id, " blocked."); |
| EXPECT_QUIC_BUG(session_.MarkConnectionLevelWriteBlocked(closed_stream_id), |
| msg); |
| } |
| |
| TEST_P(QuicSessionTestServer, OnCanWrite) { |
| session_.set_writev_consumes_all_data(true); |
| TestStream* stream2 = session_.CreateOutgoingBidirectionalStream(); |
| TestStream* stream4 = session_.CreateOutgoingBidirectionalStream(); |
| TestStream* stream6 = session_.CreateOutgoingBidirectionalStream(); |
| |
| session_.MarkConnectionLevelWriteBlocked(stream2->id()); |
| session_.MarkConnectionLevelWriteBlocked(stream6->id()); |
| session_.MarkConnectionLevelWriteBlocked(stream4->id()); |
| |
| InSequence s; |
| |
| // Reregister, to test the loop limit. |
| EXPECT_CALL(*stream2, OnCanWrite()).WillOnce(Invoke([this, stream2]() { |
| session_.SendStreamData(stream2); |
| session_.MarkConnectionLevelWriteBlocked(stream2->id()); |
| })); |
| // 2 will get called a second time as it didn't finish its block |
| EXPECT_CALL(*stream2, OnCanWrite()).WillOnce(Invoke([this, stream2]() { |
| session_.SendStreamData(stream2); |
| })); |
| EXPECT_CALL(*stream6, OnCanWrite()).WillOnce(Invoke([this, stream6]() { |
| session_.SendStreamData(stream6); |
| })); |
| // 4 will not get called, as we exceeded the loop limit. |
| session_.OnCanWrite(); |
| EXPECT_TRUE(session_.WillingAndAbleToWrite()); |
| } |
| |
| TEST_P(QuicSessionTestServer, TestBatchedWrites) { |
| session_.set_writev_consumes_all_data(true); |
| TestStream* stream2 = session_.CreateOutgoingBidirectionalStream(); |
| TestStream* stream4 = session_.CreateOutgoingBidirectionalStream(); |
| TestStream* stream6 = session_.CreateOutgoingBidirectionalStream(); |
| |
| session_.set_writev_consumes_all_data(true); |
| session_.MarkConnectionLevelWriteBlocked(stream2->id()); |
| session_.MarkConnectionLevelWriteBlocked(stream4->id()); |
| |
| // With two sessions blocked, we should get two write calls. They should both |
| // go to the first stream as it will only write 6k and mark itself blocked |
| // again. |
| InSequence s; |
| EXPECT_CALL(*stream2, OnCanWrite()).WillOnce(Invoke([this, stream2]() { |
| session_.SendLargeFakeData(stream2, 6000); |
| session_.MarkConnectionLevelWriteBlocked(stream2->id()); |
| })); |
| EXPECT_CALL(*stream2, OnCanWrite()).WillOnce(Invoke([this, stream2]() { |
| session_.SendLargeFakeData(stream2, 6000); |
| session_.MarkConnectionLevelWriteBlocked(stream2->id()); |
| })); |
| session_.OnCanWrite(); |
| |
| // We should get one more call for stream2, at which point it has used its |
| // write quota and we move over to stream 4. |
| EXPECT_CALL(*stream2, OnCanWrite()).WillOnce(Invoke([this, stream2]() { |
| session_.SendLargeFakeData(stream2, 6000); |
| session_.MarkConnectionLevelWriteBlocked(stream2->id()); |
| })); |
| EXPECT_CALL(*stream4, OnCanWrite()).WillOnce(Invoke([this, stream4]() { |
| session_.SendLargeFakeData(stream4, 6000); |
| session_.MarkConnectionLevelWriteBlocked(stream4->id()); |
| })); |
| session_.OnCanWrite(); |
| |
| // Now let stream 4 do the 2nd of its 3 writes, but add a block for a high |
| // priority stream 6. 4 should be preempted. 6 will write but *not* block so |
| // will cede back to 4. |
| stream6->SetPriority(kV3HighestPriority); |
| EXPECT_CALL(*stream4, OnCanWrite()) |
| .WillOnce(Invoke([this, stream4, stream6]() { |
| session_.SendLargeFakeData(stream4, 6000); |
| session_.MarkConnectionLevelWriteBlocked(stream4->id()); |
| session_.MarkConnectionLevelWriteBlocked(stream6->id()); |
| })); |
| EXPECT_CALL(*stream6, OnCanWrite()) |
| .WillOnce(Invoke([this, stream4, stream6]() { |
| session_.SendStreamData(stream6); |
| session_.SendLargeFakeData(stream4, 6000); |
| })); |
| session_.OnCanWrite(); |
| |
| // Stream4 alread did 6k worth of writes, so after doing another 12k it should |
| // cede and 2 should resume. |
| EXPECT_CALL(*stream4, OnCanWrite()).WillOnce(Invoke([this, stream4]() { |
| session_.SendLargeFakeData(stream4, 12000); |
| session_.MarkConnectionLevelWriteBlocked(stream4->id()); |
| })); |
| EXPECT_CALL(*stream2, OnCanWrite()).WillOnce(Invoke([this, stream2]() { |
| session_.SendLargeFakeData(stream2, 6000); |
| session_.MarkConnectionLevelWriteBlocked(stream2->id()); |
| })); |
| session_.OnCanWrite(); |
| } |
| |
| TEST_P(QuicSessionTestServer, OnCanWriteBundlesStreams) { |
| // Encryption needs to be established before data can be sent. |
| CryptoHandshakeMessage msg; |
| MockPacketWriter* writer = static_cast<MockPacketWriter*>( |
| QuicConnectionPeer::GetWriter(session_.connection())); |
| session_.GetMutableCryptoStream()->OnHandshakeMessage(msg); |
| |
| // Drive congestion control manually. |
| MockSendAlgorithm* send_algorithm = new StrictMock<MockSendAlgorithm>; |
| QuicConnectionPeer::SetSendAlgorithm(session_.connection(), send_algorithm); |
| |
| TestStream* stream2 = session_.CreateOutgoingBidirectionalStream(); |
| TestStream* stream4 = session_.CreateOutgoingBidirectionalStream(); |
| TestStream* stream6 = session_.CreateOutgoingBidirectionalStream(); |
| |
| session_.MarkConnectionLevelWriteBlocked(stream2->id()); |
| session_.MarkConnectionLevelWriteBlocked(stream6->id()); |
| session_.MarkConnectionLevelWriteBlocked(stream4->id()); |
| |
| EXPECT_CALL(*send_algorithm, CanSend(_)).WillRepeatedly(Return(true)); |
| EXPECT_CALL(*send_algorithm, GetCongestionWindow()) |
| .WillRepeatedly(Return(kMaxPacketSize * 10)); |
| EXPECT_CALL(*send_algorithm, InRecovery()).WillRepeatedly(Return(false)); |
| EXPECT_CALL(*stream2, OnCanWrite()).WillOnce(Invoke([this, stream2]() { |
| session_.SendStreamData(stream2); |
| })); |
| EXPECT_CALL(*stream4, OnCanWrite()).WillOnce(Invoke([this, stream4]() { |
| session_.SendStreamData(stream4); |
| })); |
| EXPECT_CALL(*stream6, OnCanWrite()).WillOnce(Invoke([this, stream6]() { |
| session_.SendStreamData(stream6); |
| })); |
| |
| // Expect that we only send one packet, the writes from different streams |
| // should be bundled together. |
| EXPECT_CALL(*writer, WritePacket(_, _, _, _, _)) |
| .WillOnce(Return(WriteResult(WRITE_STATUS_OK, 0))); |
| EXPECT_CALL(*send_algorithm, OnPacketSent(_, _, _, _, _)); |
| EXPECT_CALL(*send_algorithm, OnApplicationLimited(_)); |
| session_.OnCanWrite(); |
| EXPECT_FALSE(session_.WillingAndAbleToWrite()); |
| } |
| |
| TEST_P(QuicSessionTestServer, OnCanWriteCongestionControlBlocks) { |
| session_.set_writev_consumes_all_data(true); |
| InSequence s; |
| |
| // Drive congestion control manually. |
| MockSendAlgorithm* send_algorithm = new StrictMock<MockSendAlgorithm>; |
| QuicConnectionPeer::SetSendAlgorithm(session_.connection(), send_algorithm); |
| |
| TestStream* stream2 = session_.CreateOutgoingBidirectionalStream(); |
| TestStream* stream4 = session_.CreateOutgoingBidirectionalStream(); |
| TestStream* stream6 = session_.CreateOutgoingBidirectionalStream(); |
| |
| session_.MarkConnectionLevelWriteBlocked(stream2->id()); |
| session_.MarkConnectionLevelWriteBlocked(stream6->id()); |
| session_.MarkConnectionLevelWriteBlocked(stream4->id()); |
| |
| EXPECT_CALL(*send_algorithm, CanSend(_)).WillOnce(Return(true)); |
| EXPECT_CALL(*stream2, OnCanWrite()).WillOnce(Invoke([this, stream2]() { |
| session_.SendStreamData(stream2); |
| })); |
| EXPECT_CALL(*send_algorithm, CanSend(_)).WillOnce(Return(true)); |
| EXPECT_CALL(*stream6, OnCanWrite()).WillOnce(Invoke([this, stream6]() { |
| session_.SendStreamData(stream6); |
| })); |
| EXPECT_CALL(*send_algorithm, CanSend(_)).WillOnce(Return(false)); |
| // stream4->OnCanWrite is not called. |
| |
| session_.OnCanWrite(); |
| EXPECT_TRUE(session_.WillingAndAbleToWrite()); |
| |
| // Still congestion-control blocked. |
| EXPECT_CALL(*send_algorithm, CanSend(_)).WillOnce(Return(false)); |
| session_.OnCanWrite(); |
| EXPECT_TRUE(session_.WillingAndAbleToWrite()); |
| |
| // stream4->OnCanWrite is called once the connection stops being |
| // congestion-control blocked. |
| EXPECT_CALL(*send_algorithm, CanSend(_)).WillOnce(Return(true)); |
| EXPECT_CALL(*stream4, OnCanWrite()).WillOnce(Invoke([this, stream4]() { |
| session_.SendStreamData(stream4); |
| })); |
| EXPECT_CALL(*send_algorithm, OnApplicationLimited(_)); |
| session_.OnCanWrite(); |
| EXPECT_FALSE(session_.WillingAndAbleToWrite()); |
| } |
| |
| TEST_P(QuicSessionTestServer, OnCanWriteWriterBlocks) { |
| // Drive congestion control manually in order to ensure that |
| // application-limited signaling is handled correctly. |
| MockSendAlgorithm* send_algorithm = new StrictMock<MockSendAlgorithm>; |
| QuicConnectionPeer::SetSendAlgorithm(session_.connection(), send_algorithm); |
| EXPECT_CALL(*send_algorithm, CanSend(_)).WillRepeatedly(Return(true)); |
| |
| // Drive packet writer manually. |
| MockPacketWriter* writer = static_cast<MockPacketWriter*>( |
| QuicConnectionPeer::GetWriter(session_.connection())); |
| EXPECT_CALL(*writer, IsWriteBlocked()).WillRepeatedly(Return(true)); |
| EXPECT_CALL(*writer, WritePacket(_, _, _, _, _)).Times(0); |
| |
| TestStream* stream2 = session_.CreateOutgoingBidirectionalStream(); |
| |
| session_.MarkConnectionLevelWriteBlocked(stream2->id()); |
| |
| EXPECT_CALL(*stream2, OnCanWrite()).Times(0); |
| EXPECT_CALL(*send_algorithm, OnApplicationLimited(_)).Times(0); |
| |
| session_.OnCanWrite(); |
| EXPECT_TRUE(session_.WillingAndAbleToWrite()); |
| } |
| |
| TEST_P(QuicSessionTestServer, BufferedHandshake) { |
| session_.set_writev_consumes_all_data(true); |
| EXPECT_FALSE(session_.HasPendingHandshake()); // Default value. |
| |
| // Test that blocking other streams does not change our status. |
| TestStream* stream2 = session_.CreateOutgoingBidirectionalStream(); |
| session_.MarkConnectionLevelWriteBlocked(stream2->id()); |
| EXPECT_FALSE(session_.HasPendingHandshake()); |
| |
| TestStream* stream3 = session_.CreateOutgoingBidirectionalStream(); |
| session_.MarkConnectionLevelWriteBlocked(stream3->id()); |
| EXPECT_FALSE(session_.HasPendingHandshake()); |
| |
| // Blocking (due to buffering of) the Crypto stream is detected. |
| session_.MarkConnectionLevelWriteBlocked( |
| QuicUtils::GetCryptoStreamId(connection_->transport_version())); |
| EXPECT_TRUE(session_.HasPendingHandshake()); |
| |
| TestStream* stream4 = session_.CreateOutgoingBidirectionalStream(); |
| session_.MarkConnectionLevelWriteBlocked(stream4->id()); |
| EXPECT_TRUE(session_.HasPendingHandshake()); |
| |
| InSequence s; |
| // Force most streams to re-register, which is common scenario when we block |
| // the Crypto stream, and only the crypto stream can "really" write. |
| |
| // Due to prioritization, we *should* be asked to write the crypto stream |
| // first. |
| // Don't re-register the crypto stream (which signals complete writing). |
| TestCryptoStream* crypto_stream = session_.GetMutableCryptoStream(); |
| EXPECT_CALL(*crypto_stream, OnCanWrite()); |
| |
| EXPECT_CALL(*stream2, OnCanWrite()).WillOnce(Invoke([this, stream2]() { |
| session_.SendStreamData(stream2); |
| })); |
| EXPECT_CALL(*stream3, OnCanWrite()).WillOnce(Invoke([this, stream3]() { |
| session_.SendStreamData(stream3); |
| })); |
| EXPECT_CALL(*stream4, OnCanWrite()).WillOnce(Invoke([this, stream4]() { |
| session_.SendStreamData(stream4); |
| session_.MarkConnectionLevelWriteBlocked(stream4->id()); |
| })); |
| |
| session_.OnCanWrite(); |
| EXPECT_TRUE(session_.WillingAndAbleToWrite()); |
| EXPECT_FALSE(session_.HasPendingHandshake()); // Crypto stream wrote. |
| } |
| |
| TEST_P(QuicSessionTestServer, OnCanWriteWithClosedStream) { |
| session_.set_writev_consumes_all_data(true); |
| TestStream* stream2 = session_.CreateOutgoingBidirectionalStream(); |
| TestStream* stream4 = session_.CreateOutgoingBidirectionalStream(); |
| TestStream* stream6 = session_.CreateOutgoingBidirectionalStream(); |
| |
| session_.MarkConnectionLevelWriteBlocked(stream2->id()); |
| session_.MarkConnectionLevelWriteBlocked(stream6->id()); |
| session_.MarkConnectionLevelWriteBlocked(stream4->id()); |
| CloseStream(stream6->id()); |
| |
| InSequence s; |
| EXPECT_CALL(*connection_, SendControlFrame(_)) |
| .WillRepeatedly(Invoke(&session_, &TestSession::ClearControlFrame)); |
| EXPECT_CALL(*stream2, OnCanWrite()).WillOnce(Invoke([this, stream2]() { |
| session_.SendStreamData(stream2); |
| })); |
| EXPECT_CALL(*stream4, OnCanWrite()).WillOnce(Invoke([this, stream4]() { |
| session_.SendStreamData(stream4); |
| })); |
| session_.OnCanWrite(); |
| EXPECT_FALSE(session_.WillingAndAbleToWrite()); |
| } |
| |
| TEST_P(QuicSessionTestServer, OnCanWriteLimitsNumWritesIfFlowControlBlocked) { |
| // Drive congestion control manually in order to ensure that |
| // application-limited signaling is handled correctly. |
| MockSendAlgorithm* send_algorithm = new StrictMock<MockSendAlgorithm>; |
| QuicConnectionPeer::SetSendAlgorithm(session_.connection(), send_algorithm); |
| EXPECT_CALL(*send_algorithm, CanSend(_)).WillRepeatedly(Return(true)); |
| |
| // Ensure connection level flow control blockage. |
| QuicFlowControllerPeer::SetSendWindowOffset(session_.flow_controller(), 0); |
| EXPECT_TRUE(session_.flow_controller()->IsBlocked()); |
| EXPECT_TRUE(session_.IsConnectionFlowControlBlocked()); |
| EXPECT_FALSE(session_.IsStreamFlowControlBlocked()); |
| |
| // Mark the crypto and headers streams as write blocked, we expect them to be |
| // allowed to write later. |
| session_.MarkConnectionLevelWriteBlocked( |
| QuicUtils::GetCryptoStreamId(connection_->transport_version())); |
| |
| // Create a data stream, and although it is write blocked we never expect it |
| // to be allowed to write as we are connection level flow control blocked. |
| TestStream* stream = session_.CreateOutgoingBidirectionalStream(); |
| session_.MarkConnectionLevelWriteBlocked(stream->id()); |
| EXPECT_CALL(*stream, OnCanWrite()).Times(0); |
| |
| // The crypto and headers streams should be called even though we are |
| // connection flow control blocked. |
| TestCryptoStream* crypto_stream = session_.GetMutableCryptoStream(); |
| EXPECT_CALL(*crypto_stream, OnCanWrite()); |
| |
| // After the crypto and header streams perform a write, the connection will be |
| // blocked by the flow control, hence it should become application-limited. |
| EXPECT_CALL(*send_algorithm, OnApplicationLimited(_)); |
| |
| session_.OnCanWrite(); |
| EXPECT_FALSE(session_.WillingAndAbleToWrite()); |
| } |
| |
| TEST_P(QuicSessionTestServer, SendGoAway) { |
| if (transport_version() == QUIC_VERSION_99) { |
| // GoAway frames are not in version 99 |
| return; |
| } |
| MockPacketWriter* writer = static_cast<MockPacketWriter*>( |
| QuicConnectionPeer::GetWriter(session_.connection())); |
| EXPECT_CALL(*writer, WritePacket(_, _, _, _, _)) |
| .WillOnce(Return(WriteResult(WRITE_STATUS_OK, 0))); |
| |
| EXPECT_CALL(*connection_, SendControlFrame(_)) |
| .WillOnce( |
| Invoke(connection_, &MockQuicConnection::ReallySendControlFrame)); |
| session_.SendGoAway(QUIC_PEER_GOING_AWAY, "Going Away."); |
| EXPECT_TRUE(session_.goaway_sent()); |
| |
| const QuicStreamId kTestStreamId = 5u; |
| EXPECT_CALL(*connection_, SendControlFrame(_)).Times(0); |
| EXPECT_CALL(*connection_, |
| OnStreamReset(kTestStreamId, QUIC_STREAM_PEER_GOING_AWAY)) |
| .Times(0); |
| EXPECT_TRUE(session_.GetOrCreateDynamicStream(kTestStreamId)); |
| } |
| |
| TEST_P(QuicSessionTestServer, DoNotSendGoAwayTwice) { |
| if (transport_version() == QUIC_VERSION_99) { |
| // TODO(b/118808809): Enable this test for version 99 when GOAWAY is |
| // supported. |
| return; |
| } |
| EXPECT_CALL(*connection_, SendControlFrame(_)) |
| .WillOnce(Invoke(&session_, &TestSession::ClearControlFrame)); |
| session_.SendGoAway(QUIC_PEER_GOING_AWAY, "Going Away."); |
| EXPECT_TRUE(session_.goaway_sent()); |
| session_.SendGoAway(QUIC_PEER_GOING_AWAY, "Going Away."); |
| } |
| |
| TEST_P(QuicSessionTestServer, InvalidGoAway) { |
| if (transport_version() == QUIC_VERSION_99) { |
| // TODO(b/118808809): Enable this test for version 99 when GOAWAY is |
| // supported. |
| return; |
| } |
| QuicGoAwayFrame go_away(kInvalidControlFrameId, QUIC_PEER_GOING_AWAY, |
| session_.next_outgoing_bidirectional_stream_id(), ""); |
| session_.OnGoAway(go_away); |
| } |
| |
| // Test that server session will send a connectivity probe in response to a |
| // connectivity probe on the same path. |
| TEST_P(QuicSessionTestServer, ServerReplyToConnectivityProbe) { |
| QuicSocketAddress old_peer_address = |
| QuicSocketAddress(QuicIpAddress::Loopback4(), kTestPort); |
| EXPECT_EQ(old_peer_address, session_.peer_address()); |
| |
| QuicSocketAddress new_peer_address = |
| QuicSocketAddress(QuicIpAddress::Loopback4(), kTestPort + 1); |
| |
| MockPacketWriter* writer = static_cast<MockPacketWriter*>( |
| QuicConnectionPeer::GetWriter(session_.connection())); |
| EXPECT_CALL(*writer, WritePacket(_, _, _, new_peer_address, _)) |
| .WillOnce(Return(WriteResult(WRITE_STATUS_OK, 0))); |
| EXPECT_CALL(*connection_, SendConnectivityProbingResponsePacket(_)) |
| .WillOnce(Invoke( |
| connection_, |
| &MockQuicConnection::ReallySendConnectivityProbingResponsePacket)); |
| if (transport_version() == QUIC_VERSION_99) { |
| // Need to explicitly do this to emulate the reception of a PathChallenge, |
| // which stores its payload for use in generating the response. |
| connection_->OnPathChallengeFrame( |
| QuicPathChallengeFrame(0, path_frame_buffer1_)); |
| } |
| session_.OnConnectivityProbeReceived(session_.self_address(), |
| new_peer_address); |
| EXPECT_EQ(old_peer_address, session_.peer_address()); |
| } |
| |
| // Same as above, but check that if there are two PATH_CHALLENGE frames in the |
| // packet, the response has both of them AND we do not do migration. This for |
| // V99 only. |
| TEST_P(QuicSessionTestServer, ServerReplyToConnectivityProbes) { |
| if (transport_version() != QUIC_VERSION_99) { |
| return; |
| } |
| QuicSocketAddress old_peer_address = |
| QuicSocketAddress(QuicIpAddress::Loopback4(), kTestPort); |
| EXPECT_EQ(old_peer_address, session_.peer_address()); |
| |
| MockPacketWriter* writer = static_cast<MockPacketWriter*>( |
| QuicConnectionPeer::GetWriter(session_.connection())); |
| // CheckMultiPathResponse validates that the written packet |
| // contains both path responses. |
| EXPECT_CALL(*writer, WritePacket(_, _, _, old_peer_address, _)) |
| .WillOnce(Invoke(this, &QuicSessionTestServer::CheckMultiPathResponse)); |
| |
| EXPECT_CALL(*connection_, SendConnectivityProbingResponsePacket(_)) |
| .WillOnce(Invoke( |
| connection_, |
| &MockQuicConnection::ReallySendConnectivityProbingResponsePacket)); |
| // Need to explicitly do this to emulate the reception of a PathChallenge, |
| // which stores its payload for use in generating the response. |
| connection_->OnPathChallengeFrame( |
| QuicPathChallengeFrame(0, path_frame_buffer1_)); |
| connection_->OnPathChallengeFrame( |
| QuicPathChallengeFrame(0, path_frame_buffer2_)); |
| session_.OnConnectivityProbeReceived(session_.self_address(), |
| old_peer_address); |
| } |
| |
| TEST_P(QuicSessionTestServer, IncreasedTimeoutAfterCryptoHandshake) { |
| EXPECT_EQ(kInitialIdleTimeoutSecs + 3, |
| QuicConnectionPeer::GetNetworkTimeout(connection_).ToSeconds()); |
| CryptoHandshakeMessage msg; |
| session_.GetMutableCryptoStream()->OnHandshakeMessage(msg); |
| EXPECT_EQ(kMaximumIdleTimeoutSecs + 3, |
| QuicConnectionPeer::GetNetworkTimeout(connection_).ToSeconds()); |
| } |
| |
| TEST_P(QuicSessionTestServer, OnStreamFrameFinStaticStreamId) { |
| // Send two bytes of payload. |
| QuicStreamFrame data1( |
| QuicUtils::GetCryptoStreamId(connection_->transport_version()), true, 0, |
| QuicStringPiece("HT")); |
| EXPECT_CALL(*connection_, |
| CloseConnection( |
| QUIC_INVALID_STREAM_ID, "Attempt to close a static stream", |
| ConnectionCloseBehavior::SEND_CONNECTION_CLOSE_PACKET)); |
| session_.OnStreamFrame(data1); |
| } |
| |
| TEST_P(QuicSessionTestServer, OnRstStreamStaticStreamId) { |
| // Send two bytes of payload. |
| QuicRstStreamFrame rst1( |
| kInvalidControlFrameId, |
| QuicUtils::GetCryptoStreamId(connection_->transport_version()), |
| QUIC_ERROR_PROCESSING_STREAM, 0); |
| EXPECT_CALL(*connection_, |
| CloseConnection( |
| QUIC_INVALID_STREAM_ID, "Attempt to reset a static stream", |
| ConnectionCloseBehavior::SEND_CONNECTION_CLOSE_PACKET)); |
| session_.OnRstStream(rst1); |
| } |
| |
| TEST_P(QuicSessionTestServer, OnStreamFrameInvalidStreamId) { |
| // Send two bytes of payload. |
| QuicStreamFrame data1( |
| QuicUtils::GetInvalidStreamId(connection_->transport_version()), true, 0, |
| QuicStringPiece("HT")); |
| EXPECT_CALL(*connection_, |
| CloseConnection( |
| QUIC_INVALID_STREAM_ID, "Recevied data for an invalid stream", |
| ConnectionCloseBehavior::SEND_CONNECTION_CLOSE_PACKET)); |
| session_.OnStreamFrame(data1); |
| } |
| |
| TEST_P(QuicSessionTestServer, OnRstStreamInvalidStreamId) { |
| // Send two bytes of payload. |
| QuicRstStreamFrame rst1( |
| kInvalidControlFrameId, |
| QuicUtils::GetInvalidStreamId(connection_->transport_version()), |
| QUIC_ERROR_PROCESSING_STREAM, 0); |
| EXPECT_CALL(*connection_, |
| CloseConnection( |
| QUIC_INVALID_STREAM_ID, "Recevied data for an invalid stream", |
| ConnectionCloseBehavior::SEND_CONNECTION_CLOSE_PACKET)); |
| session_.OnRstStream(rst1); |
| } |
| |
| TEST_P(QuicSessionTestServer, HandshakeUnblocksFlowControlBlockedStream) { |
| // Test that if a stream is flow control blocked, then on receipt of the SHLO |
| // containing a suitable send window offset, the stream becomes unblocked. |
| |
| // Ensure that Writev consumes all the data it is given (simulate no socket |
| // blocking). |
| session_.set_writev_consumes_all_data(true); |
| |
| // Create a stream, and send enough data to make it flow control blocked. |
| TestStream* stream2 = session_.CreateOutgoingBidirectionalStream(); |
| QuicString body(kMinimumFlowControlSendWindow, '.'); |
| EXPECT_FALSE(stream2->flow_controller()->IsBlocked()); |
| EXPECT_FALSE(session_.IsConnectionFlowControlBlocked()); |
| EXPECT_FALSE(session_.IsStreamFlowControlBlocked()); |
| EXPECT_CALL(*connection_, SendControlFrame(_)).Times(AtLeast(1)); |
| stream2->WriteOrBufferData(body, false, nullptr); |
| EXPECT_TRUE(stream2->flow_controller()->IsBlocked()); |
| EXPECT_TRUE(session_.IsConnectionFlowControlBlocked()); |
| EXPECT_TRUE(session_.IsStreamFlowControlBlocked()); |
| |
| // Now complete the crypto handshake, resulting in an increased flow control |
| // send window. |
| CryptoHandshakeMessage msg; |
| session_.GetMutableCryptoStream()->OnHandshakeMessage(msg); |
| EXPECT_TRUE(QuicSessionPeer::IsStreamWriteBlocked(&session_, stream2->id())); |
| // Stream is now unblocked. |
| EXPECT_FALSE(stream2->flow_controller()->IsBlocked()); |
| EXPECT_FALSE(session_.IsConnectionFlowControlBlocked()); |
| EXPECT_FALSE(session_.IsStreamFlowControlBlocked()); |
| } |
| |
| TEST_P(QuicSessionTestServer, HandshakeUnblocksFlowControlBlockedCryptoStream) { |
| if (GetParam().transport_version >= QUIC_VERSION_47) { |
| // QUIC version 47 onwards uses CRYPTO frames for the handshake, so this |
| // test doesn't make sense for those versions since CRYPTO frames aren't |
| // flow controlled. |
| return; |
| } |
| // Test that if the crypto stream is flow control blocked, then if the SHLO |
| // contains a larger send window offset, the stream becomes unblocked. |
| session_.set_writev_consumes_all_data(true); |
| TestCryptoStream* crypto_stream = session_.GetMutableCryptoStream(); |
| EXPECT_FALSE(crypto_stream->flow_controller()->IsBlocked()); |
| EXPECT_FALSE(session_.IsConnectionFlowControlBlocked()); |
| EXPECT_FALSE(session_.IsStreamFlowControlBlocked()); |
| EXPECT_FALSE(session_.IsConnectionFlowControlBlocked()); |
| EXPECT_FALSE(session_.IsStreamFlowControlBlocked()); |
| EXPECT_CALL(*connection_, SendControlFrame(_)) |
| .WillOnce(Invoke(&session_, &TestSession::ClearControlFrame)); |
| for (QuicStreamId i = 0; |
| !crypto_stream->flow_controller()->IsBlocked() && i < 1000u; i++) { |
| EXPECT_FALSE(session_.IsConnectionFlowControlBlocked()); |
| EXPECT_FALSE(session_.IsStreamFlowControlBlocked()); |
| QuicStreamOffset offset = crypto_stream->stream_bytes_written(); |
| QuicConfig config; |
| CryptoHandshakeMessage crypto_message; |
| config.ToHandshakeMessage(&crypto_message); |
| crypto_stream->SendHandshakeMessage(crypto_message); |
| char buf[1000]; |
| QuicDataWriter writer(1000, buf, NETWORK_BYTE_ORDER); |
| crypto_stream->WriteStreamData(offset, crypto_message.size(), &writer); |
| } |
| EXPECT_TRUE(crypto_stream->flow_controller()->IsBlocked()); |
| EXPECT_FALSE(session_.IsConnectionFlowControlBlocked()); |
| EXPECT_TRUE(session_.IsStreamFlowControlBlocked()); |
| EXPECT_FALSE(session_.HasDataToWrite()); |
| EXPECT_TRUE(crypto_stream->HasBufferedData()); |
| |
| // Now complete the crypto handshake, resulting in an increased flow control |
| // send window. |
| CryptoHandshakeMessage msg; |
| session_.GetMutableCryptoStream()->OnHandshakeMessage(msg); |
| EXPECT_TRUE(QuicSessionPeer::IsStreamWriteBlocked( |
| &session_, |
| QuicUtils::GetCryptoStreamId(connection_->transport_version()))); |
| // Stream is now unblocked and will no longer have buffered data. |
| EXPECT_FALSE(crypto_stream->flow_controller()->IsBlocked()); |
| EXPECT_FALSE(session_.IsConnectionFlowControlBlocked()); |
| EXPECT_FALSE(session_.IsStreamFlowControlBlocked()); |
| } |
| |
| TEST_P(QuicSessionTestServer, ConnectionFlowControlAccountingRstOutOfOrder) { |
| // Test that when we receive an out of order stream RST we correctly adjust |
| // our connection level flow control receive window. |
| // On close, the stream should mark as consumed all bytes between the highest |
| // byte consumed so far and the final byte offset from the RST frame. |
| TestStream* stream = session_.CreateOutgoingBidirectionalStream(); |
| |
| const QuicStreamOffset kByteOffset = |
| 1 + kInitialSessionFlowControlWindowForTest / 2; |
| |
| EXPECT_CALL(*connection_, SendControlFrame(_)) |
| .Times(2) |
| .WillRepeatedly(Invoke(&session_, &TestSession::ClearControlFrame)); |
| EXPECT_CALL(*connection_, OnStreamReset(stream->id(), _)); |
| |
| QuicRstStreamFrame rst_frame(kInvalidControlFrameId, stream->id(), |
| QUIC_STREAM_CANCELLED, kByteOffset); |
| session_.OnRstStream(rst_frame); |
| if (transport_version() == QUIC_VERSION_99) { |
| // The test is predicated on the stream being fully closed. For V99, the |
| // RST_STREAM only does one side (the read side from the perspective of the |
| // node receiving the RST_STREAM). This is needed to fully close the |
| // stream and therefore fulfill all of the expects. |
| QuicStopSendingFrame frame(kInvalidControlFrameId, stream->id(), |
| QUIC_STREAM_CANCELLED); |
| EXPECT_TRUE(session_.OnStopSendingFrame(frame)); |
| } |
| EXPECT_EQ(kByteOffset, session_.flow_controller()->bytes_consumed()); |
| } |
| |
| TEST_P(QuicSessionTestServer, ConnectionFlowControlAccountingFinAndLocalReset) { |
| // Test the situation where we receive a FIN on a stream, and before we fully |
| // consume all the data from the sequencer buffer we locally RST the stream. |
| // The bytes between highest consumed byte, and the final byte offset that we |
| // determined when the FIN arrived, should be marked as consumed at the |
| // connection level flow controller when the stream is reset. |
| TestStream* stream = session_.CreateOutgoingBidirectionalStream(); |
| |
| const QuicStreamOffset kByteOffset = |
| kInitialSessionFlowControlWindowForTest / 2 - 1; |
| QuicStreamFrame frame(stream->id(), true, kByteOffset, "."); |
| session_.OnStreamFrame(frame); |
| EXPECT_TRUE(connection_->connected()); |
| |
| EXPECT_EQ(0u, stream->flow_controller()->bytes_consumed()); |
| EXPECT_EQ(kByteOffset + frame.data_length, |
| stream->flow_controller()->highest_received_byte_offset()); |
| |
| // Reset stream locally. |
| EXPECT_CALL(*connection_, SendControlFrame(_)); |
| EXPECT_CALL(*connection_, OnStreamReset(stream->id(), _)); |
| stream->Reset(QUIC_STREAM_CANCELLED); |
| EXPECT_EQ(kByteOffset + frame.data_length, |
| session_.flow_controller()->bytes_consumed()); |
| } |
| |
| TEST_P(QuicSessionTestServer, ConnectionFlowControlAccountingFinAfterRst) { |
| // Test that when we RST the stream (and tear down stream state), and then |
| // receive a FIN from the peer, we correctly adjust our connection level flow |
| // control receive window. |
| |
| // Connection starts with some non-zero highest received byte offset, |
| // due to other active streams. |
| const uint64_t kInitialConnectionBytesConsumed = 567; |
| const uint64_t kInitialConnectionHighestReceivedOffset = 1234; |
| EXPECT_LT(kInitialConnectionBytesConsumed, |
| kInitialConnectionHighestReceivedOffset); |
| session_.flow_controller()->UpdateHighestReceivedOffset( |
| kInitialConnectionHighestReceivedOffset); |
| session_.flow_controller()->AddBytesConsumed(kInitialConnectionBytesConsumed); |
| |
| // Reset our stream: this results in the stream being closed locally. |
| TestStream* stream = session_.CreateOutgoingBidirectionalStream(); |
| EXPECT_CALL(*connection_, SendControlFrame(_)); |
| EXPECT_CALL(*connection_, OnStreamReset(stream->id(), _)); |
| stream->Reset(QUIC_STREAM_CANCELLED); |
| |
| // Now receive a response from the peer with a FIN. We should handle this by |
| // adjusting the connection level flow control receive window to take into |
| // account the total number of bytes sent by the peer. |
| const QuicStreamOffset kByteOffset = 5678; |
| QuicString body = "hello"; |
| QuicStreamFrame frame(stream->id(), true, kByteOffset, QuicStringPiece(body)); |
| session_.OnStreamFrame(frame); |
| |
| QuicStreamOffset total_stream_bytes_sent_by_peer = |
| kByteOffset + body.length(); |
| EXPECT_EQ(kInitialConnectionBytesConsumed + total_stream_bytes_sent_by_peer, |
| session_.flow_controller()->bytes_consumed()); |
| EXPECT_EQ( |
| kInitialConnectionHighestReceivedOffset + total_stream_bytes_sent_by_peer, |
| session_.flow_controller()->highest_received_byte_offset()); |
| } |
| |
| TEST_P(QuicSessionTestServer, ConnectionFlowControlAccountingRstAfterRst) { |
| // Test that when we RST the stream (and tear down stream state), and then |
| // receive a RST from the peer, we correctly adjust our connection level flow |
| // control receive window. |
| |
| // Connection starts with some non-zero highest received byte offset, |
| // due to other active streams. |
| const uint64_t kInitialConnectionBytesConsumed = 567; |
| const uint64_t kInitialConnectionHighestReceivedOffset = 1234; |
| EXPECT_LT(kInitialConnectionBytesConsumed, |
| kInitialConnectionHighestReceivedOffset); |
| session_.flow_controller()->UpdateHighestReceivedOffset( |
| kInitialConnectionHighestReceivedOffset); |
| session_.flow_controller()->AddBytesConsumed(kInitialConnectionBytesConsumed); |
| |
| // Reset our stream: this results in the stream being closed locally. |
| TestStream* stream = session_.CreateOutgoingBidirectionalStream(); |
| EXPECT_CALL(*connection_, SendControlFrame(_)); |
| EXPECT_CALL(*connection_, OnStreamReset(stream->id(), _)); |
| stream->Reset(QUIC_STREAM_CANCELLED); |
| EXPECT_TRUE(QuicStreamPeer::read_side_closed(stream)); |
| |
| // Now receive a RST from the peer. We should handle this by adjusting the |
| // connection level flow control receive window to take into account the total |
| // number of bytes sent by the peer. |
| const QuicStreamOffset kByteOffset = 5678; |
| QuicRstStreamFrame rst_frame(kInvalidControlFrameId, stream->id(), |
| QUIC_STREAM_CANCELLED, kByteOffset); |
| session_.OnRstStream(rst_frame); |
| |
| EXPECT_EQ(kInitialConnectionBytesConsumed + kByteOffset, |
| session_.flow_controller()->bytes_consumed()); |
| EXPECT_EQ(kInitialConnectionHighestReceivedOffset + kByteOffset, |
| session_.flow_controller()->highest_received_byte_offset()); |
| } |
| |
| TEST_P(QuicSessionTestServer, InvalidStreamFlowControlWindowInHandshake) { |
| // Test that receipt of an invalid (< default) stream flow control window from |
| // the peer results in the connection being torn down. |
| const uint32_t kInvalidWindow = kMinimumFlowControlSendWindow - 1; |
| QuicConfigPeer::SetReceivedInitialStreamFlowControlWindow(session_.config(), |
| kInvalidWindow); |
| |
| EXPECT_CALL(*connection_, |
| CloseConnection(QUIC_FLOW_CONTROL_INVALID_WINDOW, _, _)); |
| session_.OnConfigNegotiated(); |
| } |
| |
| TEST_P(QuicSessionTestServer, InvalidSessionFlowControlWindowInHandshake) { |
| // Test that receipt of an invalid (< default) session flow control window |
| // from the peer results in the connection being torn down. |
| const uint32_t kInvalidWindow = kMinimumFlowControlSendWindow - 1; |
| QuicConfigPeer::SetReceivedInitialSessionFlowControlWindow(session_.config(), |
| kInvalidWindow); |
| |
| EXPECT_CALL(*connection_, |
| CloseConnection(QUIC_FLOW_CONTROL_INVALID_WINDOW, _, _)); |
| session_.OnConfigNegotiated(); |
| } |
| |
| // Test negotiation of custom server initial flow control window. |
| TEST_P(QuicSessionTestServer, CustomFlowControlWindow) { |
| QuicTagVector copt; |
| copt.push_back(kIFW7); |
| QuicConfigPeer::SetReceivedConnectionOptions(session_.config(), copt); |
| |
| session_.OnConfigNegotiated(); |
| EXPECT_EQ(192 * 1024u, QuicFlowControllerPeer::ReceiveWindowSize( |
| session_.flow_controller())); |
| } |
| |
| TEST_P(QuicSessionTestServer, FlowControlWithInvalidFinalOffset) { |
| // Test that if we receive a stream RST with a highest byte offset that |
| // violates flow control, that we close the connection. |
| const uint64_t kLargeOffset = kInitialSessionFlowControlWindowForTest + 1; |
| EXPECT_CALL(*connection_, |
| CloseConnection(QUIC_FLOW_CONTROL_RECEIVED_TOO_MUCH_DATA, _, _)) |
| .Times(2); |
| |
| // Check that stream frame + FIN results in connection close. |
| TestStream* stream = session_.CreateOutgoingBidirectionalStream(); |
| EXPECT_CALL(*connection_, SendControlFrame(_)); |
| EXPECT_CALL(*connection_, OnStreamReset(stream->id(), _)); |
| stream->Reset(QUIC_STREAM_CANCELLED); |
| QuicStreamFrame frame(stream->id(), true, kLargeOffset, QuicStringPiece()); |
| session_.OnStreamFrame(frame); |
| |
| // Check that RST results in connection close. |
| QuicRstStreamFrame rst_frame(kInvalidControlFrameId, stream->id(), |
| QUIC_STREAM_CANCELLED, kLargeOffset); |
| session_.OnRstStream(rst_frame); |
| } |
| |
| TEST_P(QuicSessionTestServer, TooManyUnfinishedStreamsCauseServerRejectStream) { |
| // If a buggy/malicious peer creates too many streams that are not ended |
| // with a FIN or RST then we send an RST to refuse streams. For V99 the |
| // connection is closed. |
| const QuicStreamId kMaxStreams = 5; |
| QuicSessionPeer::SetMaxOpenIncomingStreams(&session_, kMaxStreams); |
| const QuicStreamId kFirstStreamId = GetNthClientInitiatedBidirectionalId(0); |
| const QuicStreamId kFinalStreamId = |
| GetNthClientInitiatedBidirectionalId(kMaxStreams); |
| // Create kMaxStreams data streams, and close them all without receiving a |
| // FIN or a RST_STREAM from the client. |
| for (QuicStreamId i = kFirstStreamId; i < kFinalStreamId; |
| i += QuicUtils::StreamIdDelta(connection_->transport_version())) { |
| QuicStreamFrame data1(i, false, 0, QuicStringPiece("HT")); |
| session_.OnStreamFrame(data1); |
| // EXPECT_EQ(1u, session_.GetNumOpenStreams()); |
| if (transport_version() == QUIC_VERSION_99) { |
| // Expect two control frames, RST STREAM and STOP SENDING |
| EXPECT_CALL(*connection_, SendControlFrame(_)) |
| .Times(2) |
| .WillRepeatedly(Invoke(&session_, &TestSession::ClearControlFrame)); |
| } else { |
| // Expect one control frame, just RST STREAM |
| EXPECT_CALL(*connection_, SendControlFrame(_)) |
| .WillOnce(Invoke(&session_, &TestSession::ClearControlFrame)); |
| } |
| // Close stream. Should not make new streams available since |
| // the stream is not finished. |
| EXPECT_CALL(*connection_, OnStreamReset(i, _)); |
| session_.CloseStream(i); |
| } |
| |
| if (transport_version() == QUIC_VERSION_99) { |
| EXPECT_CALL(*connection_, CloseConnection(QUIC_INVALID_STREAM_ID, |
| "Stream id 24 above 20", _)); |
| } else { |
| EXPECT_CALL(*connection_, SendControlFrame(_)).Times(1); |
| EXPECT_CALL(*connection_, |
| OnStreamReset(kFinalStreamId, QUIC_REFUSED_STREAM)) |
| .Times(1); |
| } |
| // Create one more data streams to exceed limit of open stream. |
| QuicStreamFrame data1(kFinalStreamId, false, 0, QuicStringPiece("HT")); |
| session_.OnStreamFrame(data1); |
| } |
| |
| TEST_P(QuicSessionTestServer, DrainingStreamsDoNotCountAsOpenedOutgoing) { |
| // Verify that a draining stream (which has received a FIN but not consumed |
| // it) does not count against the open quota (because it is closed from the |
| // protocol point of view). |
| TestStream* stream = session_.CreateOutgoingBidirectionalStream(); |
| QuicStreamId stream_id = stream->id(); |
| QuicStreamFrame data1(stream_id, true, 0, QuicStringPiece("HT")); |
| session_.OnStreamFrame(data1); |
| EXPECT_CALL(session_, OnCanCreateNewOutgoingStream()).Times(1); |
| session_.StreamDraining(stream_id); |
| } |
| |
| TEST_P(QuicSessionTestServer, NoPendingStreams) { |
| session_.set_should_buffer_incoming_streams(false); |
| |
| QuicStreamId stream_id = QuicUtils::GetFirstUnidirectionalStreamId( |
| transport_version(), Perspective::IS_CLIENT); |
| QuicStreamFrame data1(stream_id, true, 10, QuicStringPiece("HT")); |
| session_.OnStreamFrame(data1); |
| EXPECT_EQ(1, session_.num_incoming_streams_created()); |
| |
| QuicStreamFrame data2(stream_id, false, 0, QuicStringPiece("HT")); |
| session_.OnStreamFrame(data2); |
| EXPECT_EQ(1, session_.num_incoming_streams_created()); |
| } |
| |
| TEST_P(QuicSessionTestServer, PendingStreams) { |
| if (connection_->transport_version() != QUIC_VERSION_99) { |
| return; |
| } |
| session_.set_should_buffer_incoming_streams(true); |
| |
| QuicStreamId stream_id = QuicUtils::GetFirstUnidirectionalStreamId( |
| transport_version(), Perspective::IS_CLIENT); |
| QuicStreamFrame data1(stream_id, true, 10, QuicStringPiece("HT")); |
| session_.OnStreamFrame(data1); |
| EXPECT_EQ(0, session_.num_incoming_streams_created()); |
| |
| QuicStreamFrame data2(stream_id, false, 0, QuicStringPiece("HT")); |
| session_.OnStreamFrame(data2); |
| EXPECT_EQ(1, session_.num_incoming_streams_created()); |
| } |
| |
| TEST_P(QuicSessionTestServer, RstPendingStreams) { |
| if (connection_->transport_version() != QUIC_VERSION_99) { |
| return; |
| } |
| session_.set_should_buffer_incoming_streams(true); |
| |
| QuicStreamId stream_id = QuicUtils::GetFirstUnidirectionalStreamId( |
| transport_version(), Perspective::IS_CLIENT); |
| QuicStreamFrame data1(stream_id, true, 10, QuicStringPiece("HT")); |
| session_.OnStreamFrame(data1); |
| EXPECT_EQ(0, session_.num_incoming_streams_created()); |
| |
| EXPECT_CALL(session_, OnCanCreateNewOutgoingStream()).Times(1); |
| EXPECT_CALL(*connection_, SendControlFrame(_)).Times(1); |
| EXPECT_CALL(*connection_, OnStreamReset(stream_id, QUIC_RST_ACKNOWLEDGEMENT)) |
| .Times(1); |
| QuicRstStreamFrame rst1(kInvalidControlFrameId, stream_id, |
| QUIC_ERROR_PROCESSING_STREAM, 12); |
| session_.OnRstStream(rst1); |
| EXPECT_EQ(0, session_.num_incoming_streams_created()); |
| |
| QuicStreamFrame data2(stream_id, false, 0, QuicStringPiece("HT")); |
| session_.OnStreamFrame(data2); |
| EXPECT_EQ(0, session_.num_incoming_streams_created()); |
| } |
| |
| TEST_P(QuicSessionTestServer, DrainingStreamsDoNotCountAsOpened) { |
| // Verify that a draining stream (which has received a FIN but not consumed |
| // it) does not count against the open quota (because it is closed from the |
| // protocol point of view). |
| if (transport_version() == QUIC_VERSION_99) { |
| // On v99, we will expect to see a MAX_STREAM_ID go out when there are not |
| // enough streams to create the next one. |
| EXPECT_CALL(*connection_, SendControlFrame(_)).Times(1); |
| } else { |
| EXPECT_CALL(*connection_, SendControlFrame(_)).Times(0); |
| } |
| EXPECT_CALL(*connection_, OnStreamReset(_, QUIC_REFUSED_STREAM)).Times(0); |
| const QuicStreamId kMaxStreams = 5; |
| QuicSessionPeer::SetMaxOpenIncomingStreams(&session_, kMaxStreams); |
| |
| // Create kMaxStreams + 1 data streams, and mark them draining. |
| const QuicStreamId kFirstStreamId = GetNthClientInitiatedBidirectionalId(0); |
| const QuicStreamId kFinalStreamId = |
| GetNthClientInitiatedBidirectionalId(2 * kMaxStreams + 1); |
| for (QuicStreamId i = kFirstStreamId; i < kFinalStreamId; |
| i += QuicUtils::StreamIdDelta(connection_->transport_version())) { |
| QuicStreamFrame data1(i, true, 0, QuicStringPiece("HT")); |
| session_.OnStreamFrame(data1); |
| EXPECT_EQ(1u, session_.GetNumOpenIncomingStreams()); |
| session_.StreamDraining(i); |
| EXPECT_EQ(0u, session_.GetNumOpenIncomingStreams()); |
| } |
| } |
| |
| class QuicSessionTestClient : public QuicSessionTestBase { |
| protected: |
| QuicSessionTestClient() : QuicSessionTestBase(Perspective::IS_CLIENT) {} |
| }; |
| |
| INSTANTIATE_TEST_SUITE_P(Tests, |
| QuicSessionTestClient, |
| ::testing::ValuesIn(AllSupportedVersions())); |
| |
| TEST_P(QuicSessionTestClient, AvailableBidirectionalStreamsClient) { |
| ASSERT_TRUE(session_.GetOrCreateDynamicStream( |
| GetNthServerInitiatedBidirectionalId(2)) != nullptr); |
| // Smaller bidirectional streams should be available. |
| EXPECT_TRUE(QuicSessionPeer::IsStreamAvailable( |
| &session_, GetNthServerInitiatedBidirectionalId(0))); |
| EXPECT_TRUE(QuicSessionPeer::IsStreamAvailable( |
| &session_, GetNthServerInitiatedBidirectionalId(1))); |
| ASSERT_TRUE(session_.GetOrCreateDynamicStream( |
| GetNthServerInitiatedBidirectionalId(0)) != nullptr); |
| ASSERT_TRUE(session_.GetOrCreateDynamicStream( |
| GetNthServerInitiatedBidirectionalId(1)) != nullptr); |
| // And 5 should be not available. |
| EXPECT_FALSE(QuicSessionPeer::IsStreamAvailable( |
| &session_, GetNthClientInitiatedBidirectionalId(1))); |
| } |
| |
| TEST_P(QuicSessionTestClient, AvailableUnidirectionalStreamsClient) { |
| ASSERT_TRUE(session_.GetOrCreateDynamicStream( |
| GetNthServerInitiatedUnidirectionalId(2)) != nullptr); |
| // Smaller unidirectional streams should be available. |
| EXPECT_TRUE(QuicSessionPeer::IsStreamAvailable( |
| &session_, GetNthServerInitiatedUnidirectionalId(0))); |
| EXPECT_TRUE(QuicSessionPeer::IsStreamAvailable( |
| &session_, GetNthServerInitiatedUnidirectionalId(1))); |
| ASSERT_TRUE(session_.GetOrCreateDynamicStream( |
| GetNthServerInitiatedUnidirectionalId(0)) != nullptr); |
| ASSERT_TRUE(session_.GetOrCreateDynamicStream( |
| GetNthServerInitiatedUnidirectionalId(1)) != nullptr); |
| // And 5 should be not available. |
| EXPECT_FALSE(QuicSessionPeer::IsStreamAvailable( |
| &session_, GetNthClientInitiatedUnidirectionalId(1))); |
| } |
| |
| TEST_P(QuicSessionTestClient, RecordFinAfterReadSideClosed) { |
| // Verify that an incoming FIN is recorded in a stream object even if the read |
| // side has been closed. This prevents an entry from being made in |
| // locally_closed_streams_highest_offset_ (which will never be deleted). |
| TestStream* stream = session_.CreateOutgoingBidirectionalStream(); |
| QuicStreamId stream_id = stream->id(); |
| |
| // Close the read side manually. |
| QuicStreamPeer::CloseReadSide(stream); |
| |
| // Receive a stream data frame with FIN. |
| QuicStreamFrame frame(stream_id, true, 0, QuicStringPiece()); |
| session_.OnStreamFrame(frame); |
| EXPECT_TRUE(stream->fin_received()); |
| |
| // Reset stream locally. |
| EXPECT_CALL(*connection_, SendControlFrame(_)); |
| EXPECT_CALL(*connection_, OnStreamReset(stream->id(), _)); |
| stream->Reset(QUIC_STREAM_CANCELLED); |
| EXPECT_TRUE(QuicStreamPeer::read_side_closed(stream)); |
| |
| EXPECT_TRUE(connection_->connected()); |
| EXPECT_TRUE(QuicSessionPeer::IsStreamClosed(&session_, stream_id)); |
| EXPECT_FALSE(QuicSessionPeer::IsStreamCreated(&session_, stream_id)); |
| |
| // The stream is not waiting for the arrival of the peer's final offset as it |
| // was received with the FIN earlier. |
| EXPECT_EQ( |
| 0u, |
| QuicSessionPeer::GetLocallyClosedStreamsHighestOffset(&session_).size()); |
| } |
| |
| TEST_P(QuicSessionTestServer, ZombieStreams) { |
| TestStream* stream2 = session_.CreateOutgoingBidirectionalStream(); |
| QuicStreamPeer::SetStreamBytesWritten(3, stream2); |
| EXPECT_TRUE(stream2->IsWaitingForAcks()); |
| |
| EXPECT_CALL(*connection_, SendControlFrame(_)); |
| EXPECT_CALL(*connection_, OnStreamReset(stream2->id(), _)); |
| session_.CloseStream(stream2->id()); |
| EXPECT_FALSE(QuicContainsKey(session_.zombie_streams(), stream2->id())); |
| ASSERT_EQ(1u, session_.closed_streams()->size()); |
| EXPECT_EQ(stream2->id(), session_.closed_streams()->front()->id()); |
| session_.OnStreamDoneWaitingForAcks(stream2->id()); |
| EXPECT_FALSE(QuicContainsKey(session_.zombie_streams(), stream2->id())); |
| EXPECT_EQ(1u, session_.closed_streams()->size()); |
| EXPECT_EQ(stream2->id(), session_.closed_streams()->front()->id()); |
| } |
| |
| TEST_P(QuicSessionTestServer, RstStreamReceivedAfterRstStreamSent) { |
| TestStream* stream2 = session_.CreateOutgoingBidirectionalStream(); |
| QuicStreamPeer::SetStreamBytesWritten(3, stream2); |
| EXPECT_TRUE(stream2->IsWaitingForAcks()); |
| |
| EXPECT_CALL(*connection_, SendControlFrame(_)); |
| EXPECT_CALL(*connection_, OnStreamReset(stream2->id(), _)); |
| EXPECT_CALL(session_, OnCanCreateNewOutgoingStream()).Times(0); |
| stream2->Reset(quic::QUIC_STREAM_CANCELLED); |
| |
| QuicRstStreamFrame rst1(kInvalidControlFrameId, stream2->id(), |
| QUIC_ERROR_PROCESSING_STREAM, 0); |
| if (transport_version() != QUIC_VERSION_99) { |
| EXPECT_CALL(session_, OnCanCreateNewOutgoingStream()).Times(1); |
| } |
| session_.OnRstStream(rst1); |
| } |
| |
| // Regression test of b/71548958. |
| TEST_P(QuicSessionTestServer, TestZombieStreams) { |
| session_.set_writev_consumes_all_data(true); |
| |
| TestStream* stream2 = session_.CreateOutgoingBidirectionalStream(); |
| QuicString body(100, '.'); |
| stream2->WriteOrBufferData(body, false, nullptr); |
| EXPECT_TRUE(stream2->IsWaitingForAcks()); |
| EXPECT_EQ(1u, QuicStreamPeer::SendBuffer(stream2).size()); |
| |
| QuicRstStreamFrame rst_frame(kInvalidControlFrameId, stream2->id(), |
| QUIC_STREAM_CANCELLED, 1234); |
| // Just for the RST_STREAM |
| EXPECT_CALL(*connection_, SendControlFrame(_)) |
| .WillOnce(Invoke(&session_, &TestSession::ClearControlFrame)); |
| if (transport_version() == QUIC_VERSION_99) { |
| EXPECT_CALL(*connection_, |
| OnStreamReset(stream2->id(), QUIC_STREAM_CANCELLED)); |
| } else { |
| EXPECT_CALL(*connection_, |
| OnStreamReset(stream2->id(), QUIC_RST_ACKNOWLEDGEMENT)); |
| } |
| stream2->OnStreamReset(rst_frame); |
| |
| if (transport_version() == QUIC_VERSION_99) { |
| // The test is predicated on the stream being fully closed. For V99, the |
| // RST_STREAM only does one side (the read side from the perspective of the |
| // node receiving the RST_STREAM). This is needed to fully close the |
| // stream and therefore fulfill all of the expects. |
| QuicStopSendingFrame frame(kInvalidControlFrameId, stream2->id(), |
| QUIC_STREAM_CANCELLED); |
| EXPECT_TRUE(session_.OnStopSendingFrame(frame)); |
| } |
| EXPECT_FALSE(QuicContainsKey(session_.zombie_streams(), stream2->id())); |
| ASSERT_EQ(1u, session_.closed_streams()->size()); |
| EXPECT_EQ(stream2->id(), session_.closed_streams()->front()->id()); |
| |
| TestStream* stream4 = session_.CreateOutgoingBidirectionalStream(); |
| if (transport_version() == QUIC_VERSION_99) { |
| // Once for the RST_STREAM, once for the STOP_SENDING |
| EXPECT_CALL(*connection_, SendControlFrame(_)) |
| .Times(2) |
| .WillRepeatedly(Invoke(&session_, &TestSession::ClearControlFrame)); |
| } else { |
| // Just for the RST_STREAM |
| EXPECT_CALL(*connection_, SendControlFrame(_)).Times(1); |
| } |
| EXPECT_CALL(*connection_, |
| OnStreamReset(stream4->id(), QUIC_STREAM_CANCELLED)); |
| stream4->WriteOrBufferData(body, false, nullptr); |
| // Note well: Reset() actually closes the stream in both directions. For |
| // GOOGLE QUIC it sends a RST_STREAM (which does a 2-way close), for IETF |
| // QUIC/V99 it sends both a RST_STREAM and a STOP_SENDING (each of which |
| // closes in only one direction). |
| stream4->Reset(QUIC_STREAM_CANCELLED); |
| EXPECT_FALSE(QuicContainsKey(session_.zombie_streams(), stream4->id())); |
| EXPECT_EQ(2u, session_.closed_streams()->size()); |
| } |
| |
| TEST_P(QuicSessionTestServer, OnStreamFrameLost) { |
| QuicConnectionPeer::SetSessionDecidesWhatToWrite(connection_); |
| InSequence s; |
| |
| // Drive congestion control manually. |
| MockSendAlgorithm* send_algorithm = new StrictMock<MockSendAlgorithm>; |
| QuicConnectionPeer::SetSendAlgorithm(session_.connection(), send_algorithm); |
| |
| TestCryptoStream* crypto_stream = session_.GetMutableCryptoStream(); |
| TestStream* stream2 = session_.CreateOutgoingBidirectionalStream(); |
| TestStream* stream4 = session_.CreateOutgoingBidirectionalStream(); |
| |
| QuicStreamFrame frame1( |
| QuicUtils::GetCryptoStreamId(connection_->transport_version()), false, 0, |
| 1300); |
| QuicStreamFrame frame2(stream2->id(), false, 0, 9); |
| QuicStreamFrame frame3(stream4->id(), false, 0, 9); |
| |
| // Lost data on cryption stream, streams 2 and 4. |
| EXPECT_CALL(*stream4, HasPendingRetransmission()).WillOnce(Return(true)); |
| if (connection_->transport_version() < QUIC_VERSION_47) { |
| EXPECT_CALL(*crypto_stream, HasPendingRetransmission()) |
| .WillOnce(Return(true)); |
| } |
| EXPECT_CALL(*stream2, HasPendingRetransmission()).WillOnce(Return(true)); |
| session_.OnFrameLost(QuicFrame(frame3)); |
| if (connection_->transport_version() < QUIC_VERSION_47) { |
| session_.OnFrameLost(QuicFrame(frame1)); |
| } else { |
| QuicCryptoFrame crypto_frame(ENCRYPTION_NONE, 0, 1300); |
| session_.OnFrameLost(QuicFrame(&crypto_frame)); |
| } |
| session_.OnFrameLost(QuicFrame(frame2)); |
| EXPECT_TRUE(session_.WillingAndAbleToWrite()); |
| |
| // Mark streams 2 and 4 write blocked. |
| session_.MarkConnectionLevelWriteBlocked(stream2->id()); |
| session_.MarkConnectionLevelWriteBlocked(stream4->id()); |
| |
| // Lost data is retransmitted before new data, and retransmissions for crypto |
| // stream go first. |
| // Do not check congestion window when crypto stream has lost data. |
| EXPECT_CALL(*send_algorithm, CanSend(_)).Times(0); |
| if (connection_->transport_version() < QUIC_VERSION_47) { |
| EXPECT_CALL(*crypto_stream, OnCanWrite()); |
| EXPECT_CALL(*crypto_stream, HasPendingRetransmission()) |
| .WillOnce(Return(false)); |
| } |
| // Check congestion window for non crypto streams. |
| EXPECT_CALL(*send_algorithm, CanSend(_)).WillOnce(Return(true)); |
| EXPECT_CALL(*stream4, OnCanWrite()); |
| EXPECT_CALL(*stream4, HasPendingRetransmission()).WillOnce(Return(false)); |
| // Connection is blocked. |
| EXPECT_CALL(*send_algorithm, CanSend(_)).WillRepeatedly(Return(false)); |
| |
| session_.OnCanWrite(); |
| EXPECT_TRUE(session_.WillingAndAbleToWrite()); |
| |
| // Unblock connection. |
| // Stream 2 retransmits lost data. |
| EXPECT_CALL(*send_algorithm, CanSend(_)).WillOnce(Return(true)); |
| EXPECT_CALL(*stream2, OnCanWrite()); |
| EXPECT_CALL(*stream2, HasPendingRetransmission()).WillOnce(Return(false)); |
| EXPECT_CALL(*send_algorithm, CanSend(_)).WillOnce(Return(true)); |
| // Stream 2 sends new data. |
| EXPECT_CALL(*stream2, OnCanWrite()); |
| EXPECT_CALL(*send_algorithm, CanSend(_)).WillOnce(Return(true)); |
| EXPECT_CALL(*stream4, OnCanWrite()); |
| EXPECT_CALL(*send_algorithm, OnApplicationLimited(_)); |
| |
| session_.OnCanWrite(); |
| EXPECT_FALSE(session_.WillingAndAbleToWrite()); |
| } |
| |
| TEST_P(QuicSessionTestServer, DonotRetransmitDataOfClosedStreams) { |
| QuicConnectionPeer::SetSessionDecidesWhatToWrite(connection_); |
| InSequence s; |
| |
| TestStream* stream2 = session_.CreateOutgoingBidirectionalStream(); |
| TestStream* stream4 = session_.CreateOutgoingBidirectionalStream(); |
| TestStream* stream6 = session_.CreateOutgoingBidirectionalStream(); |
| |
| QuicStreamFrame frame1(stream2->id(), false, 0, 9); |
| QuicStreamFrame frame2(stream4->id(), false, 0, 9); |
| QuicStreamFrame frame3(stream6->id(), false, 0, 9); |
| |
| EXPECT_CALL(*stream6, HasPendingRetransmission()).WillOnce(Return(true)); |
| EXPECT_CALL(*stream4, HasPendingRetransmission()).WillOnce(Return(true)); |
| EXPECT_CALL(*stream2, HasPendingRetransmission()).WillOnce(Return(true)); |
| session_.OnFrameLost(QuicFrame(frame3)); |
| session_.OnFrameLost(QuicFrame(frame2)); |
| session_.OnFrameLost(QuicFrame(frame1)); |
| |
| session_.MarkConnectionLevelWriteBlocked(stream2->id()); |
| session_.MarkConnectionLevelWriteBlocked(stream4->id()); |
| session_.MarkConnectionLevelWriteBlocked(stream6->id()); |
| |
| // Reset stream 4 locally. |
| EXPECT_CALL(*connection_, SendControlFrame(_)); |
| EXPECT_CALL(*connection_, OnStreamReset(stream4->id(), _)); |
| stream4->Reset(QUIC_STREAM_CANCELLED); |
| |
| // Verify stream 4 is removed from streams with lost data list. |
| EXPECT_CALL(*stream6, OnCanWrite()); |
| EXPECT_CALL(*stream6, HasPendingRetransmission()).WillOnce(Return(false)); |
| EXPECT_CALL(*stream2, OnCanWrite()); |
| EXPECT_CALL(*stream2, HasPendingRetransmission()).WillOnce(Return(false)); |
| EXPECT_CALL(*connection_, SendControlFrame(_)) |
| .WillRepeatedly(Invoke(&session_, &TestSession::ClearControlFrame)); |
| EXPECT_CALL(*stream2, OnCanWrite()); |
| EXPECT_CALL(*stream6, OnCanWrite()); |
| session_.OnCanWrite(); |
| } |
| |
| TEST_P(QuicSessionTestServer, RetransmitFrames) { |
| QuicConnectionPeer::SetSessionDecidesWhatToWrite(connection_); |
| MockSendAlgorithm* send_algorithm = new StrictMock<MockSendAlgorithm>; |
| QuicConnectionPeer::SetSendAlgorithm(session_.connection(), send_algorithm); |
| InSequence s; |
| |
| TestStream* stream2 = session_.CreateOutgoingBidirectionalStream(); |
| TestStream* stream4 = session_.CreateOutgoingBidirectionalStream(); |
| TestStream* stream6 = session_.CreateOutgoingBidirectionalStream(); |
| EXPECT_CALL(*connection_, SendControlFrame(_)) |
| .WillOnce(Invoke(&session_, &TestSession::ClearControlFrame)); |
| session_.SendWindowUpdate(stream2->id(), 9); |
| |
| QuicStreamFrame frame1(stream2->id(), false, 0, 9); |
| QuicStreamFrame frame2(stream4->id(), false, 0, 9); |
| QuicStreamFrame frame3(stream6->id(), false, 0, 9); |
| QuicWindowUpdateFrame window_update(1, stream2->id(), 9); |
| QuicFrames frames; |
| frames.push_back(QuicFrame(frame1)); |
| frames.push_back(QuicFrame(&window_update)); |
| frames.push_back(QuicFrame(frame2)); |
| frames.push_back(QuicFrame(frame3)); |
| EXPECT_FALSE(session_.WillingAndAbleToWrite()); |
| |
| EXPECT_CALL(*stream2, RetransmitStreamData(_, _, _)).WillOnce(Return(true)); |
| EXPECT_CALL(*connection_, SendControlFrame(_)) |
| .WillOnce(Invoke(&session_, &TestSession::ClearControlFrame)); |
| EXPECT_CALL(*stream4, RetransmitStreamData(_, _, _)).WillOnce(Return(true)); |
| EXPECT_CALL(*stream6, RetransmitStreamData(_, _, _)).WillOnce(Return(true)); |
| EXPECT_CALL(*send_algorithm, OnApplicationLimited(_)); |
| session_.RetransmitFrames(frames, TLP_RETRANSMISSION); |
| } |
| |
| // Regression test of b/110082001. |
| TEST_P(QuicSessionTestServer, RetransmitLostDataCausesConnectionClose) { |
| // This test mimics the scenario when a dynamic stream retransmits lost data |
| // and causes connection close. |
| QuicConnectionPeer::SetSessionDecidesWhatToWrite(connection_); |
| TestStream* stream = session_.CreateOutgoingBidirectionalStream(); |
| QuicStreamFrame frame(stream->id(), false, 0, 9); |
| |
| EXPECT_CALL(*stream, HasPendingRetransmission()) |
| .Times(2) |
| .WillOnce(Return(true)) |
| .WillOnce(Return(false)); |
| session_.OnFrameLost(QuicFrame(frame)); |
| // Retransmit stream data causes connection close. Stream has not sent fin |
| // yet, so an RST is sent. |
| EXPECT_CALL(*stream, OnCanWrite()) |
| .WillOnce(Invoke(stream, &QuicStream::OnClose)); |
| if (transport_version() == QUIC_VERSION_99) { |
| // Once for the RST_STREAM, once for the STOP_SENDING |
| EXPECT_CALL(*connection_, SendControlFrame(_)) |
| .Times(2) |
| .WillRepeatedly(Invoke(&session_, &TestSession::SaveFrame)); |
| } else { |
| // Just for the RST_STREAM |
| EXPECT_CALL(*connection_, SendControlFrame(_)) |
| .WillOnce(Invoke(&session_, &TestSession::SaveFrame)); |
| } |
| EXPECT_CALL(*connection_, OnStreamReset(stream->id(), _)); |
| session_.OnCanWrite(); |
| } |
| |
| TEST_P(QuicSessionTestServer, SendMessage) { |
| // Cannot send message when encryption is not established. |
| EXPECT_FALSE(session_.IsCryptoHandshakeConfirmed()); |
| quic::QuicMemSliceStorage storage(nullptr, 0, nullptr, 0); |
| EXPECT_EQ(MessageResult(MESSAGE_STATUS_ENCRYPTION_NOT_ESTABLISHED, 0), |
| session_.SendMessage( |
| MakeSpan(connection_->helper()->GetStreamSendBufferAllocator(), |
| "", &storage))); |
| |
| // Finish handshake. |
| CryptoHandshakeMessage handshake_message; |
| session_.GetMutableCryptoStream()->OnHandshakeMessage(handshake_message); |
| EXPECT_TRUE(session_.IsCryptoHandshakeConfirmed()); |
| |
| QuicStringPiece message; |
| EXPECT_CALL(*connection_, SendMessage(1, _)) |
| .WillOnce(Return(MESSAGE_STATUS_SUCCESS)); |
| EXPECT_EQ(MessageResult(MESSAGE_STATUS_SUCCESS, 1), |
| session_.SendMessage( |
| MakeSpan(connection_->helper()->GetStreamSendBufferAllocator(), |
| message, &storage))); |
| // Verify message_id increases. |
| EXPECT_CALL(*connection_, SendMessage(2, _)) |
| .WillOnce(Return(MESSAGE_STATUS_TOO_LARGE)); |
| EXPECT_EQ(MessageResult(MESSAGE_STATUS_TOO_LARGE, 0), |
| session_.SendMessage( |
| MakeSpan(connection_->helper()->GetStreamSendBufferAllocator(), |
| message, &storage))); |
| // Verify unsent message does not consume a message_id. |
| EXPECT_CALL(*connection_, SendMessage(2, _)) |
| .WillOnce(Return(MESSAGE_STATUS_SUCCESS)); |
| EXPECT_EQ(MessageResult(MESSAGE_STATUS_SUCCESS, 2), |
| session_.SendMessage( |
| MakeSpan(connection_->helper()->GetStreamSendBufferAllocator(), |
| message, &storage))); |
| |
| QuicMessageFrame frame(1); |
| QuicMessageFrame frame2(2); |
| EXPECT_FALSE(session_.IsFrameOutstanding(QuicFrame(&frame))); |
| EXPECT_FALSE(session_.IsFrameOutstanding(QuicFrame(&frame2))); |
| |
| // Lost message 2. |
| session_.OnMessageLost(2); |
| EXPECT_FALSE(session_.IsFrameOutstanding(QuicFrame(&frame2))); |
| |
| // message 1 gets acked. |
| session_.OnMessageAcked(1); |
| EXPECT_FALSE(session_.IsFrameOutstanding(QuicFrame(&frame))); |
| } |
| |
| // Regression test of b/115323618. |
| TEST_P(QuicSessionTestServer, LocallyResetZombieStreams) { |
| QuicConnectionPeer::SetSessionDecidesWhatToWrite(connection_); |
| |
| session_.set_writev_consumes_all_data(true); |
| TestStream* stream2 = session_.CreateOutgoingBidirectionalStream(); |
| QuicString body(100, '.'); |
| stream2->CloseReadSide(); |
| stream2->WriteOrBufferData(body, true, nullptr); |
| EXPECT_TRUE(stream2->IsWaitingForAcks()); |
| // Verify stream2 is a zombie streams. |
| EXPECT_TRUE(QuicContainsKey(session_.zombie_streams(), stream2->id())); |
| |
| QuicStreamFrame frame(stream2->id(), true, 0, 100); |
| EXPECT_CALL(*stream2, HasPendingRetransmission()) |
| .WillRepeatedly(Return(true)); |
| session_.OnFrameLost(QuicFrame(frame)); |
| |
| // Reset stream2 locally. |
| EXPECT_CALL(*connection_, SendControlFrame(_)) |
| .WillRepeatedly(Invoke(&session_, &TestSession::ClearControlFrame)); |
| EXPECT_CALL(*connection_, OnStreamReset(stream2->id(), _)); |
| stream2->Reset(QUIC_STREAM_CANCELLED); |
| |
| // Verify stream 2 gets closed. |
| EXPECT_FALSE(QuicContainsKey(session_.zombie_streams(), stream2->id())); |
| EXPECT_TRUE(session_.IsClosedStream(stream2->id())); |
| EXPECT_CALL(*stream2, OnCanWrite()).Times(0); |
| session_.OnCanWrite(); |
| } |
| |
| TEST_P(QuicSessionTestServer, CleanUpClosedStreamsAlarm) { |
| EXPECT_FALSE( |
| QuicSessionPeer::GetCleanUpClosedStreamsAlarm(&session_)->IsSet()); |
| |
| session_.set_writev_consumes_all_data(true); |
| TestStream* stream2 = session_.CreateOutgoingBidirectionalStream(); |
| EXPECT_FALSE(stream2->IsWaitingForAcks()); |
| |
| EXPECT_CALL(*connection_, SendControlFrame(_)); |
| EXPECT_CALL(*connection_, OnStreamReset(stream2->id(), _)); |
| session_.CloseStream(stream2->id()); |
| EXPECT_FALSE(QuicContainsKey(session_.zombie_streams(), stream2->id())); |
| EXPECT_EQ(1u, session_.closed_streams()->size()); |
| EXPECT_TRUE( |
| QuicSessionPeer::GetCleanUpClosedStreamsAlarm(&session_)->IsSet()); |
| |
| alarm_factory_.FireAlarm( |
| QuicSessionPeer::GetCleanUpClosedStreamsAlarm(&session_)); |
| EXPECT_TRUE(session_.closed_streams()->empty()); |
| } |
| |
| TEST_P(QuicSessionTestServer, WriteUnidirectionalStream) { |
| session_.set_writev_consumes_all_data(true); |
| TestStream* stream4 = new TestStream(GetNthServerInitiatedUnidirectionalId(1), |
| &session_, WRITE_UNIDIRECTIONAL); |
| session_.ActivateStream(QuicWrapUnique(stream4)); |
| QuicString body(100, '.'); |
| stream4->WriteOrBufferData(body, false, nullptr); |
| EXPECT_FALSE(QuicContainsKey(session_.zombie_streams(), stream4->id())); |
| stream4->WriteOrBufferData(body, true, nullptr); |
| EXPECT_TRUE(QuicContainsKey(session_.zombie_streams(), stream4->id())); |
| } |
| |
| TEST_P(QuicSessionTestServer, ReceivedDataOnWriteUnidirectionalStream) { |
| TestStream* stream4 = new TestStream(GetNthServerInitiatedUnidirectionalId(1), |
| &session_, WRITE_UNIDIRECTIONAL); |
| session_.ActivateStream(QuicWrapUnique(stream4)); |
| |
| EXPECT_CALL( |
| *connection_, |
| CloseConnection(QUIC_DATA_RECEIVED_ON_WRITE_UNIDIRECTIONAL_STREAM, _, _)) |
| .Times(1); |
| QuicStreamFrame stream_frame(GetNthServerInitiatedUnidirectionalId(1), false, |
| 0, 2); |
| session_.OnStreamFrame(stream_frame); |
| } |
| |
| TEST_P(QuicSessionTestServer, ReadUnidirectionalStream) { |
| TestStream* stream4 = new TestStream(GetNthClientInitiatedUnidirectionalId(1), |
| &session_, READ_UNIDIRECTIONAL); |
| session_.ActivateStream(QuicWrapUnique(stream4)); |
| EXPECT_FALSE(stream4->IsWaitingForAcks()); |
| // Discard all incoming data. |
| stream4->StopReading(); |
| |
| QuicString data(100, '.'); |
| QuicStreamFrame stream_frame(GetNthClientInitiatedUnidirectionalId(1), false, |
| 0, data); |
| stream4->OnStreamFrame(stream_frame); |
| EXPECT_TRUE(session_.closed_streams()->empty()); |
| |
| QuicStreamFrame stream_frame2(GetNthClientInitiatedUnidirectionalId(1), true, |
| 100, data); |
| stream4->OnStreamFrame(stream_frame2); |
| EXPECT_EQ(1u, session_.closed_streams()->size()); |
| } |
| |
| TEST_P(QuicSessionTestServer, WriteOrBufferDataOnReadUnidirectionalStream) { |
| TestStream* stream4 = new TestStream(GetNthClientInitiatedUnidirectionalId(1), |
| &session_, READ_UNIDIRECTIONAL); |
| session_.ActivateStream(QuicWrapUnique(stream4)); |
| |
| EXPECT_CALL(*connection_, |
| CloseConnection( |
| QUIC_TRY_TO_WRITE_DATA_ON_READ_UNIDIRECTIONAL_STREAM, _, _)) |
| .Times(1); |
| QuicString body(100, '.'); |
| stream4->WriteOrBufferData(body, false, nullptr); |
| } |
| |
| TEST_P(QuicSessionTestServer, WritevDataOnReadUnidirectionalStream) { |
| TestStream* stream4 = new TestStream(GetNthClientInitiatedUnidirectionalId(1), |
| &session_, READ_UNIDIRECTIONAL); |
| session_.ActivateStream(QuicWrapUnique(stream4)); |
| |
| EXPECT_CALL(*connection_, |
| CloseConnection( |
| QUIC_TRY_TO_WRITE_DATA_ON_READ_UNIDIRECTIONAL_STREAM, _, _)) |
| .Times(1); |
| QuicString body(100, '.'); |
| struct IOVEC iov = {const_cast<char*>(body.data()), body.length()}; |
| QuicMemSliceStorage storage( |
| &iov, 1, session_.connection()->helper()->GetStreamSendBufferAllocator(), |
| 1024); |
| stream4->WriteMemSlices(storage.ToSpan(), false); |
| } |
| |
| TEST_P(QuicSessionTestServer, WriteMemSlicesOnReadUnidirectionalStream) { |
| TestStream* stream4 = new TestStream(GetNthClientInitiatedUnidirectionalId(1), |
| &session_, READ_UNIDIRECTIONAL); |
| session_.ActivateStream(QuicWrapUnique(stream4)); |
| |
| EXPECT_CALL(*connection_, |
| CloseConnection( |
| QUIC_TRY_TO_WRITE_DATA_ON_READ_UNIDIRECTIONAL_STREAM, _, _)) |
| .Times(1); |
| char data[1024]; |
| std::vector<std::pair<char*, size_t>> buffers; |
| buffers.push_back(std::make_pair(data, QUIC_ARRAYSIZE(data))); |
| buffers.push_back(std::make_pair(data, QUIC_ARRAYSIZE(data))); |
| QuicTestMemSliceVector vector(buffers); |
| stream4->WriteMemSlices(vector.span(), false); |
| } |
| |
| // Test code that tests that an incoming stream frame with a new (not previously |
| // seen) stream id is acceptable. The ID must not be larger than has been |
| // advertised. It may be equal to what has been advertised. These tests |
| // invoke QuicStreamIdManager::MaybeIncreaseLargestPeerStreamId by calling |
| // QuicSession::OnStreamFrame in order to check that all the steps are connected |
| // properly and that nothing in the call path interferes with the check. |
| // First test make sure that streams with ids below the limit are accepted. |
| TEST_P(QuicSessionTestServer, NewStreamIdBelowLimit) { |
| if (transport_version() != QUIC_VERSION_99) { |
| // Applicable only to V99 |
| return; |
| } |
| QuicStreamId bidirectional_stream_id = |
| QuicSessionPeer::v99_streamid_manager(&session_) |
| ->advertised_max_allowed_incoming_bidirectional_stream_id() - |
| kV99StreamIdIncrement; |
| QuicStreamFrame bidirectional_stream_frame(bidirectional_stream_id, false, 0, |
| "Random String"); |
| EXPECT_CALL(*connection_, CloseConnection(_, _, _)).Times(0); |
| session_.OnStreamFrame(bidirectional_stream_frame); |
| |
| QuicStreamId unidirectional_stream_id = |
| QuicSessionPeer::v99_streamid_manager(&session_) |
| ->advertised_max_allowed_incoming_unidirectional_stream_id() - |
| kV99StreamIdIncrement; |
| QuicStreamFrame unidirectional_stream_frame(unidirectional_stream_id, false, |
| 0, "Random String"); |
| EXPECT_CALL(*connection_, CloseConnection(_, _, _)).Times(0); |
| session_.OnStreamFrame(unidirectional_stream_frame); |
| } |
| |
| // Accept a stream with an ID that equals the limit. |
| TEST_P(QuicSessionTestServer, NewStreamIdAtLimit) { |
| if (transport_version() != QUIC_VERSION_99) { |
| // Applicable only to V99 |
| return; |
| } |
| QuicStreamId bidirectional_stream_id = |
| QuicSessionPeer::v99_streamid_manager(&session_) |
| ->advertised_max_allowed_incoming_bidirectional_stream_id(); |
| QuicStreamFrame bidirectional_stream_frame(bidirectional_stream_id, false, 0, |
| "Random String"); |
| EXPECT_CALL(*connection_, CloseConnection(_, _, _)).Times(0); |
| session_.OnStreamFrame(bidirectional_stream_frame); |
| |
| QuicStreamId unidirectional_stream_id = |
| QuicSessionPeer::v99_streamid_manager(&session_) |
| ->advertised_max_allowed_incoming_unidirectional_stream_id(); |
| QuicStreamFrame unidirectional_stream_frame(unidirectional_stream_id, false, |
| 0, "Random String"); |
| EXPECT_CALL(*connection_, CloseConnection(_, _, _)).Times(0); |
| session_.OnStreamFrame(unidirectional_stream_frame); |
| } |
| |
| // Close the connection if the id exceeds the limit. |
| TEST_P(QuicSessionTestServer, NewStreamIdAboveLimit) { |
| if (transport_version() != QUIC_VERSION_99) { |
| // Applicable only to V99 |
| return; |
| } |
| QuicStreamId bidirectional_stream_id = |
| QuicSessionPeer::v99_streamid_manager(&session_) |
| ->advertised_max_allowed_incoming_bidirectional_stream_id() + |
| kV99StreamIdIncrement; |
| QuicStreamFrame bidirectional_stream_frame(bidirectional_stream_id, false, 0, |
| "Random String"); |
| EXPECT_CALL(*connection_, CloseConnection(QUIC_INVALID_STREAM_ID, |
| "Stream id 404 above 400", _)); |
| session_.OnStreamFrame(bidirectional_stream_frame); |
| |
| QuicStreamId unidirectional_stream_id = |
| QuicSessionPeer::v99_streamid_manager(&session_) |
| ->advertised_max_allowed_incoming_unidirectional_stream_id() + |
| kV99StreamIdIncrement; |
| QuicStreamFrame unidirectional_stream_frame(unidirectional_stream_id, false, |
| 0, "Random String"); |
| EXPECT_CALL(*connection_, CloseConnection(QUIC_INVALID_STREAM_ID, |
| "Stream id 402 above 398", _)); |
| session_.OnStreamFrame(unidirectional_stream_frame); |
| } |
| |
| // Check that the OnStopSendingFrame upcall handles bad input properly |
| // First test checks that invalid stream ids are handled. |
| TEST_P(QuicSessionTestServer, OnStopSendingInputInvalidStreamId) { |
| if (transport_version() != QUIC_VERSION_99) { |
| // Applicable only to V99 |
| return; |
| } |
| // Check that "invalid" stream ids are rejected. |
| // Note that the notion of an invalid stream id is Google-specific. |
| QuicStopSendingFrame frame(1, -1, 123); |
| EXPECT_CALL( |
| *connection_, |
| CloseConnection(QUIC_INVALID_STREAM_ID, |
| "Received STOP_SENDING for an invalid stream", _)); |
| EXPECT_FALSE(session_.OnStopSendingFrame(frame)); |
| } |
| |
| // Second test, streams in the static stream map are not subject to |
| // STOP_SENDING; it's ignored. |
| TEST_P(QuicSessionTestServer, OnStopSendingInputStaticStreams) { |
| if (transport_version() != QUIC_VERSION_99) { |
| // Applicable only to V99 |
| return; |
| } |
| // Check that a stream id in the static stream map is ignored. |
| // Note that the notion of a static stream is Google-specific. |
| QuicStopSendingFrame frame(1, 0, 123); |
| EXPECT_CALL(*connection_, |
| CloseConnection(QUIC_INVALID_STREAM_ID, |
| "Received STOP_SENDING for a static stream", _)); |
| EXPECT_FALSE(session_.OnStopSendingFrame(frame)); |
| } |
| |
| // Third test, if stream id specifies a closed stream: |
| // return true and do not close the connection. |
| TEST_P(QuicSessionTestServer, OnStopSendingInputClosedStream) { |
| if (transport_version() != QUIC_VERSION_99) { |
| // Applicable only to V99 |
| return; |
| } |
| |
| TestStream* stream = session_.CreateOutgoingBidirectionalStream(); |
| QuicStreamId stream_id = stream->id(); |
| // Expect these as side effect of the close operations. |
| EXPECT_CALL(*connection_, SendControlFrame(_)); |
| EXPECT_CALL(*connection_, OnStreamReset(_, _)); |
| stream->CloseWriteSide(); |
| stream->CloseReadSide(); |
| QuicStopSendingFrame frame(1, stream_id, 123); |
| EXPECT_CALL(*connection_, CloseConnection(_, _, _)).Times(0); |
| EXPECT_TRUE(session_.OnStopSendingFrame(frame)); |
| } |
| |
| // Fourth test, if stream id specifies a nonexistent stream, return false and |
| // close the connection |
| TEST_P(QuicSessionTestServer, OnStopSendingInputNonExistentStream) { |
| if (transport_version() != QUIC_VERSION_99) { |
| // Applicable only to V99 |
| return; |
| } |
| |
| QuicStopSendingFrame frame(1, GetNthServerInitiatedBidirectionalId(123456), |
| 123); |
| EXPECT_CALL( |
| *connection_, |
| CloseConnection(IETF_QUIC_PROTOCOL_VIOLATION, |
| "Received STOP_SENDING for a non-existent stream", _)) |
| .Times(1); |
| EXPECT_FALSE(session_.OnStopSendingFrame(frame)); |
| } |
| |
| // For a valid stream, ensure that all works |
| TEST_P(QuicSessionTestServer, OnStopSendingInputValidStream) { |
| if (transport_version() != QUIC_VERSION_99) { |
| // Applicable only to V99 |
| return; |
| } |
| |
| TestStream* stream = session_.CreateOutgoingBidirectionalStream(); |
| |
| // Ensure that the stream starts out open in both directions. |
| EXPECT_FALSE(QuicStreamPeer::write_side_closed(stream)); |
| EXPECT_FALSE(QuicStreamPeer::read_side_closed(stream)); |
| |
| QuicStreamId stream_id = stream->id(); |
| QuicStopSendingFrame frame(1, stream_id, 123); |
| EXPECT_CALL(*stream, OnStopSending(123)); |
| // Expect a reset to come back out. |
| EXPECT_CALL(*connection_, SendControlFrame(_)); |
| EXPECT_CALL( |
| *connection_, |
| OnStreamReset(stream_id, static_cast<QuicRstStreamErrorCode>(123))); |
| EXPECT_TRUE(session_.OnStopSendingFrame(frame)); |
| // When the STOP_SENDING is received, the node generates a RST_STREAM, |
| // which closes the stream in the write direction. Ensure this. |
| EXPECT_FALSE(QuicStreamPeer::read_side_closed(stream)); |
| EXPECT_TRUE(QuicStreamPeer::write_side_closed(stream)); |
| } |
| |
| } // namespace |
| } // namespace test |
| } // namespace quic |