| // Copyright (c) 2018 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 "quiche/quic/test_tools/simple_session_notifier.h" |
| |
| #include <utility> |
| |
| #include "quiche/quic/core/crypto/null_encrypter.h" |
| #include "quiche/quic/core/quic_utils.h" |
| #include "quiche/quic/platform/api/quic_test.h" |
| #include "quiche/quic/test_tools/quic_connection_peer.h" |
| #include "quiche/quic/test_tools/quic_test_utils.h" |
| #include "quiche/quic/test_tools/simple_data_producer.h" |
| |
| using testing::_; |
| using testing::InSequence; |
| using testing::Return; |
| using testing::StrictMock; |
| |
| namespace quic { |
| namespace test { |
| namespace { |
| |
| class MockQuicConnectionWithSendStreamData : public MockQuicConnection { |
| public: |
| MockQuicConnectionWithSendStreamData(MockQuicConnectionHelper* helper, |
| MockAlarmFactory* alarm_factory, |
| Perspective perspective) |
| : MockQuicConnection(helper, alarm_factory, perspective) {} |
| |
| MOCK_METHOD(QuicConsumedData, SendStreamData, |
| (QuicStreamId id, size_t write_length, QuicStreamOffset offset, |
| StreamSendingState state), |
| (override)); |
| }; |
| |
| class SimpleSessionNotifierTest : public QuicTest { |
| public: |
| SimpleSessionNotifierTest() |
| : connection_(&helper_, &alarm_factory_, Perspective::IS_CLIENT), |
| notifier_(&connection_) { |
| connection_.set_visitor(&visitor_); |
| connection_.SetSessionNotifier(¬ifier_); |
| EXPECT_FALSE(notifier_.WillingToWrite()); |
| EXPECT_EQ(0u, notifier_.StreamBytesSent()); |
| EXPECT_FALSE(notifier_.HasBufferedStreamData()); |
| } |
| |
| bool ControlFrameConsumed(const QuicFrame& frame) { |
| DeleteFrame(&const_cast<QuicFrame&>(frame)); |
| return true; |
| } |
| |
| MockQuicConnectionHelper helper_; |
| MockAlarmFactory alarm_factory_; |
| MockQuicConnectionVisitor visitor_; |
| StrictMock<MockQuicConnectionWithSendStreamData> connection_; |
| SimpleSessionNotifier notifier_; |
| }; |
| |
| TEST_F(SimpleSessionNotifierTest, WriteOrBufferData) { |
| InSequence s; |
| EXPECT_CALL(connection_, SendStreamData(3, 1024, 0, NO_FIN)) |
| .WillOnce(Return(QuicConsumedData(1024, false))); |
| notifier_.WriteOrBufferData(3, 1024, NO_FIN); |
| EXPECT_EQ(0u, notifier_.StreamBytesToSend()); |
| EXPECT_CALL(connection_, SendStreamData(5, 512, 0, NO_FIN)) |
| .WillOnce(Return(QuicConsumedData(512, false))); |
| notifier_.WriteOrBufferData(5, 512, NO_FIN); |
| EXPECT_FALSE(notifier_.WillingToWrite()); |
| // Connection is blocked. |
| EXPECT_CALL(connection_, SendStreamData(5, 512, 512, FIN)) |
| .WillOnce(Return(QuicConsumedData(256, false))); |
| notifier_.WriteOrBufferData(5, 512, FIN); |
| EXPECT_TRUE(notifier_.WillingToWrite()); |
| EXPECT_EQ(1792u, notifier_.StreamBytesSent()); |
| EXPECT_EQ(256u, notifier_.StreamBytesToSend()); |
| EXPECT_TRUE(notifier_.HasBufferedStreamData()); |
| |
| // New data cannot be sent as connection is blocked. |
| EXPECT_CALL(connection_, SendStreamData(7, 1024, 0, FIN)).Times(0); |
| notifier_.WriteOrBufferData(7, 1024, FIN); |
| EXPECT_EQ(1792u, notifier_.StreamBytesSent()); |
| } |
| |
| TEST_F(SimpleSessionNotifierTest, WriteOrBufferRstStream) { |
| InSequence s; |
| EXPECT_CALL(connection_, SendStreamData(5, 1024, 0, FIN)) |
| .WillOnce(Return(QuicConsumedData(1024, true))); |
| notifier_.WriteOrBufferData(5, 1024, FIN); |
| EXPECT_TRUE(notifier_.StreamIsWaitingForAcks(5)); |
| EXPECT_TRUE(notifier_.HasUnackedStreamData()); |
| |
| // Reset stream 5 with no error. |
| EXPECT_CALL(connection_, SendControlFrame(_)) |
| .WillRepeatedly( |
| Invoke(this, &SimpleSessionNotifierTest::ControlFrameConsumed)); |
| notifier_.WriteOrBufferRstStream(5, QUIC_STREAM_NO_ERROR, 1024); |
| // Verify stream 5 is waiting for acks. |
| EXPECT_TRUE(notifier_.StreamIsWaitingForAcks(5)); |
| EXPECT_TRUE(notifier_.HasUnackedStreamData()); |
| |
| // Reset stream 5 with error. |
| notifier_.WriteOrBufferRstStream(5, QUIC_ERROR_PROCESSING_STREAM, 1024); |
| EXPECT_FALSE(notifier_.StreamIsWaitingForAcks(5)); |
| EXPECT_FALSE(notifier_.HasUnackedStreamData()); |
| } |
| |
| TEST_F(SimpleSessionNotifierTest, WriteOrBufferPing) { |
| InSequence s; |
| // Write ping when connection is not write blocked. |
| EXPECT_CALL(connection_, SendControlFrame(_)) |
| .WillRepeatedly( |
| Invoke(this, &SimpleSessionNotifierTest::ControlFrameConsumed)); |
| notifier_.WriteOrBufferPing(); |
| EXPECT_EQ(0u, notifier_.StreamBytesToSend()); |
| EXPECT_FALSE(notifier_.WillingToWrite()); |
| |
| // Write stream data and cause the connection to be write blocked. |
| EXPECT_CALL(connection_, SendStreamData(3, 1024, 0, NO_FIN)) |
| .WillOnce(Return(QuicConsumedData(1024, false))); |
| notifier_.WriteOrBufferData(3, 1024, NO_FIN); |
| EXPECT_EQ(0u, notifier_.StreamBytesToSend()); |
| EXPECT_CALL(connection_, SendStreamData(5, 512, 0, NO_FIN)) |
| .WillOnce(Return(QuicConsumedData(256, false))); |
| notifier_.WriteOrBufferData(5, 512, NO_FIN); |
| EXPECT_TRUE(notifier_.WillingToWrite()); |
| |
| // Connection is blocked. |
| EXPECT_CALL(connection_, SendControlFrame(_)).Times(0); |
| notifier_.WriteOrBufferPing(); |
| } |
| |
| TEST_F(SimpleSessionNotifierTest, NeuterUnencryptedData) { |
| if (QuicVersionUsesCryptoFrames(connection_.transport_version())) { |
| // This test writes crypto data through crypto streams. It won't work when |
| // crypto frames are used instead. |
| return; |
| } |
| InSequence s; |
| // Send crypto data [0, 1024) in ENCRYPTION_INITIAL. |
| connection_.SetDefaultEncryptionLevel(ENCRYPTION_INITIAL); |
| EXPECT_CALL(connection_, SendStreamData(QuicUtils::GetCryptoStreamId( |
| connection_.transport_version()), |
| 1024, 0, NO_FIN)) |
| .WillOnce(Return(QuicConsumedData(1024, false))); |
| notifier_.WriteOrBufferData( |
| QuicUtils::GetCryptoStreamId(connection_.transport_version()), 1024, |
| NO_FIN); |
| // Send crypto data [1024, 2048) in ENCRYPTION_ZERO_RTT. |
| connection_.SetDefaultEncryptionLevel(ENCRYPTION_ZERO_RTT); |
| EXPECT_CALL(connection_, SendStreamData(QuicUtils::GetCryptoStreamId( |
| connection_.transport_version()), |
| 1024, 1024, NO_FIN)) |
| .WillOnce(Return(QuicConsumedData(1024, false))); |
| notifier_.WriteOrBufferData( |
| QuicUtils::GetCryptoStreamId(connection_.transport_version()), 1024, |
| NO_FIN); |
| // Ack [1024, 2048). |
| QuicStreamFrame stream_frame( |
| QuicUtils::GetCryptoStreamId(connection_.transport_version()), false, |
| 1024, 1024); |
| notifier_.OnFrameAcked(QuicFrame(stream_frame), QuicTime::Delta::Zero(), |
| QuicTime::Zero()); |
| EXPECT_TRUE(notifier_.StreamIsWaitingForAcks( |
| QuicUtils::GetCryptoStreamId(connection_.transport_version()))); |
| EXPECT_TRUE(notifier_.HasUnackedStreamData()); |
| |
| // Neuters unencrypted data. |
| notifier_.NeuterUnencryptedData(); |
| EXPECT_FALSE(notifier_.StreamIsWaitingForAcks( |
| QuicUtils::GetCryptoStreamId(connection_.transport_version()))); |
| EXPECT_FALSE(notifier_.HasUnackedStreamData()); |
| } |
| |
| TEST_F(SimpleSessionNotifierTest, OnCanWrite) { |
| if (QuicVersionUsesCryptoFrames(connection_.transport_version())) { |
| // This test writes crypto data through crypto streams. It won't work when |
| // crypto frames are used instead. |
| return; |
| } |
| InSequence s; |
| // Send crypto data [0, 1024) in ENCRYPTION_INITIAL. |
| connection_.SetDefaultEncryptionLevel(ENCRYPTION_INITIAL); |
| EXPECT_CALL(connection_, SendStreamData(QuicUtils::GetCryptoStreamId( |
| connection_.transport_version()), |
| 1024, 0, NO_FIN)) |
| .WillOnce(Return(QuicConsumedData(1024, false))); |
| notifier_.WriteOrBufferData( |
| QuicUtils::GetCryptoStreamId(connection_.transport_version()), 1024, |
| NO_FIN); |
| |
| // Send crypto data [1024, 2048) in ENCRYPTION_ZERO_RTT. |
| connection_.SetDefaultEncryptionLevel(ENCRYPTION_ZERO_RTT); |
| EXPECT_CALL(connection_, SendStreamData(QuicUtils::GetCryptoStreamId( |
| connection_.transport_version()), |
| 1024, 1024, NO_FIN)) |
| .WillOnce(Return(QuicConsumedData(1024, false))); |
| notifier_.WriteOrBufferData( |
| QuicUtils::GetCryptoStreamId(connection_.transport_version()), 1024, |
| NO_FIN); |
| // Send stream 3 [0, 1024) and connection is blocked. |
| EXPECT_CALL(connection_, SendStreamData(3, 1024, 0, FIN)) |
| .WillOnce(Return(QuicConsumedData(512, false))); |
| notifier_.WriteOrBufferData(3, 1024, FIN); |
| // Send stream 5 [0, 1024). |
| EXPECT_CALL(connection_, SendStreamData(5, _, _, _)).Times(0); |
| notifier_.WriteOrBufferData(5, 1024, NO_FIN); |
| // Reset stream 5 with error. |
| EXPECT_CALL(connection_, SendControlFrame(_)).Times(0); |
| notifier_.WriteOrBufferRstStream(5, QUIC_ERROR_PROCESSING_STREAM, 1024); |
| |
| // Lost crypto data [500, 1500) and stream 3 [0, 512). |
| QuicStreamFrame frame1( |
| QuicUtils::GetCryptoStreamId(connection_.transport_version()), false, 500, |
| 1000); |
| QuicStreamFrame frame2(3, false, 0, 512); |
| notifier_.OnFrameLost(QuicFrame(frame1)); |
| notifier_.OnFrameLost(QuicFrame(frame2)); |
| |
| // Connection becomes writable. |
| // Lost crypto data gets retransmitted as [500, 1024) and [1024, 1500), as |
| // they are in different encryption levels. |
| EXPECT_CALL(connection_, SendStreamData(QuicUtils::GetCryptoStreamId( |
| connection_.transport_version()), |
| 524, 500, NO_FIN)) |
| .WillOnce(Return(QuicConsumedData(524, false))); |
| EXPECT_CALL(connection_, SendStreamData(QuicUtils::GetCryptoStreamId( |
| connection_.transport_version()), |
| 476, 1024, NO_FIN)) |
| .WillOnce(Return(QuicConsumedData(476, false))); |
| // Lost stream 3 data gets retransmitted. |
| EXPECT_CALL(connection_, SendStreamData(3, 512, 0, NO_FIN)) |
| .WillOnce(Return(QuicConsumedData(512, false))); |
| // Buffered control frames get sent. |
| EXPECT_CALL(connection_, SendControlFrame(_)) |
| .WillOnce(Invoke(this, &SimpleSessionNotifierTest::ControlFrameConsumed)); |
| // Buffered stream 3 data [512, 1024) gets sent. |
| EXPECT_CALL(connection_, SendStreamData(3, 512, 512, FIN)) |
| .WillOnce(Return(QuicConsumedData(512, true))); |
| notifier_.OnCanWrite(); |
| EXPECT_FALSE(notifier_.WillingToWrite()); |
| } |
| |
| TEST_F(SimpleSessionNotifierTest, OnCanWriteCryptoFrames) { |
| if (!QuicVersionUsesCryptoFrames(connection_.transport_version())) { |
| return; |
| } |
| SimpleDataProducer producer; |
| connection_.SetDataProducer(&producer); |
| InSequence s; |
| // Send crypto data [0, 1024) in ENCRYPTION_INITIAL. |
| connection_.SetDefaultEncryptionLevel(ENCRYPTION_INITIAL); |
| EXPECT_CALL(connection_, SendCryptoData(ENCRYPTION_INITIAL, 1024, 0)) |
| .WillOnce(Invoke(&connection_, |
| &MockQuicConnection::QuicConnection_SendCryptoData)); |
| EXPECT_CALL(connection_, CloseConnection(QUIC_PACKET_WRITE_ERROR, _, _)); |
| std::string crypto_data1(1024, 'a'); |
| producer.SaveCryptoData(ENCRYPTION_INITIAL, 0, crypto_data1); |
| std::string crypto_data2(524, 'a'); |
| producer.SaveCryptoData(ENCRYPTION_INITIAL, 500, crypto_data2); |
| notifier_.WriteCryptoData(ENCRYPTION_INITIAL, 1024, 0); |
| // Send crypto data [1024, 2048) in ENCRYPTION_ZERO_RTT. |
| connection_.SetEncrypter(ENCRYPTION_ZERO_RTT, std::make_unique<NullEncrypter>( |
| Perspective::IS_CLIENT)); |
| connection_.SetDefaultEncryptionLevel(ENCRYPTION_ZERO_RTT); |
| EXPECT_CALL(connection_, SendCryptoData(ENCRYPTION_ZERO_RTT, 1024, 0)) |
| .WillOnce(Invoke(&connection_, |
| &MockQuicConnection::QuicConnection_SendCryptoData)); |
| std::string crypto_data3(1024, 'a'); |
| producer.SaveCryptoData(ENCRYPTION_ZERO_RTT, 0, crypto_data3); |
| notifier_.WriteCryptoData(ENCRYPTION_ZERO_RTT, 1024, 0); |
| // Send stream 3 [0, 1024) and connection is blocked. |
| EXPECT_CALL(connection_, SendStreamData(3, 1024, 0, FIN)) |
| .WillOnce(Return(QuicConsumedData(512, false))); |
| notifier_.WriteOrBufferData(3, 1024, FIN); |
| // Send stream 5 [0, 1024). |
| EXPECT_CALL(connection_, SendStreamData(5, _, _, _)).Times(0); |
| notifier_.WriteOrBufferData(5, 1024, NO_FIN); |
| // Reset stream 5 with error. |
| EXPECT_CALL(connection_, SendControlFrame(_)).Times(0); |
| notifier_.WriteOrBufferRstStream(5, QUIC_ERROR_PROCESSING_STREAM, 1024); |
| |
| // Lost crypto data [500, 1500) and stream 3 [0, 512). |
| QuicCryptoFrame crypto_frame1(ENCRYPTION_INITIAL, 500, 524); |
| QuicCryptoFrame crypto_frame2(ENCRYPTION_ZERO_RTT, 0, 476); |
| QuicStreamFrame stream3_frame(3, false, 0, 512); |
| notifier_.OnFrameLost(QuicFrame(&crypto_frame1)); |
| notifier_.OnFrameLost(QuicFrame(&crypto_frame2)); |
| notifier_.OnFrameLost(QuicFrame(stream3_frame)); |
| |
| // Connection becomes writable. |
| // Lost crypto data gets retransmitted as [500, 1024) and [1024, 1500), as |
| // they are in different encryption levels. |
| EXPECT_CALL(connection_, SendCryptoData(ENCRYPTION_INITIAL, 524, 500)) |
| .WillOnce(Invoke(&connection_, |
| &MockQuicConnection::QuicConnection_SendCryptoData)); |
| EXPECT_CALL(connection_, SendCryptoData(ENCRYPTION_ZERO_RTT, 476, 0)) |
| .WillOnce(Invoke(&connection_, |
| &MockQuicConnection::QuicConnection_SendCryptoData)); |
| // Lost stream 3 data gets retransmitted. |
| EXPECT_CALL(connection_, SendStreamData(3, 512, 0, NO_FIN)) |
| .WillOnce(Return(QuicConsumedData(512, false))); |
| // Buffered control frames get sent. |
| EXPECT_CALL(connection_, SendControlFrame(_)) |
| .WillOnce(Invoke(this, &SimpleSessionNotifierTest::ControlFrameConsumed)); |
| // Buffered stream 3 data [512, 1024) gets sent. |
| EXPECT_CALL(connection_, SendStreamData(3, 512, 512, FIN)) |
| .WillOnce(Return(QuicConsumedData(512, true))); |
| notifier_.OnCanWrite(); |
| EXPECT_FALSE(notifier_.WillingToWrite()); |
| } |
| |
| TEST_F(SimpleSessionNotifierTest, RetransmitFrames) { |
| InSequence s; |
| connection_.SetEncrypter( |
| ENCRYPTION_FORWARD_SECURE, |
| std::make_unique<NullEncrypter>(Perspective::IS_CLIENT)); |
| // Send stream 3 data [0, 10) and fin. |
| EXPECT_CALL(connection_, SendStreamData(3, 10, 0, FIN)) |
| .WillOnce(Return(QuicConsumedData(10, true))); |
| notifier_.WriteOrBufferData(3, 10, FIN); |
| QuicStreamFrame frame1(3, true, 0, 10); |
| // Send stream 5 [0, 10) and fin. |
| EXPECT_CALL(connection_, SendStreamData(5, 10, 0, FIN)) |
| .WillOnce(Return(QuicConsumedData(10, true))); |
| notifier_.WriteOrBufferData(5, 10, FIN); |
| QuicStreamFrame frame2(5, true, 0, 10); |
| // Reset stream 5 with no error. |
| EXPECT_CALL(connection_, SendControlFrame(_)) |
| .WillOnce(Invoke(this, &SimpleSessionNotifierTest::ControlFrameConsumed)); |
| notifier_.WriteOrBufferRstStream(5, QUIC_STREAM_NO_ERROR, 10); |
| |
| // Ack stream 3 [3, 7), and stream 5 [8, 10). |
| QuicStreamFrame ack_frame1(3, false, 3, 4); |
| QuicStreamFrame ack_frame2(5, false, 8, 2); |
| notifier_.OnFrameAcked(QuicFrame(ack_frame1), QuicTime::Delta::Zero(), |
| QuicTime::Zero()); |
| notifier_.OnFrameAcked(QuicFrame(ack_frame2), QuicTime::Delta::Zero(), |
| QuicTime::Zero()); |
| EXPECT_FALSE(notifier_.WillingToWrite()); |
| |
| // Force to send. |
| QuicRstStreamFrame rst_stream(1, 5, QUIC_STREAM_NO_ERROR, 10); |
| QuicFrames frames; |
| frames.push_back(QuicFrame(frame2)); |
| frames.push_back(QuicFrame(&rst_stream)); |
| frames.push_back(QuicFrame(frame1)); |
| // stream 5 data [0, 8), fin only are retransmitted. |
| EXPECT_CALL(connection_, SendStreamData(5, 8, 0, NO_FIN)) |
| .WillOnce(Return(QuicConsumedData(8, false))); |
| EXPECT_CALL(connection_, SendStreamData(5, 0, 10, FIN)) |
| .WillOnce(Return(QuicConsumedData(0, true))); |
| // rst_stream is retransmitted. |
| EXPECT_CALL(connection_, SendControlFrame(_)) |
| .WillOnce(Invoke(this, &SimpleSessionNotifierTest::ControlFrameConsumed)); |
| // stream 3 data [0, 3) is retransmitted and connection is blocked. |
| EXPECT_CALL(connection_, SendStreamData(3, 3, 0, NO_FIN)) |
| .WillOnce(Return(QuicConsumedData(2, false))); |
| notifier_.RetransmitFrames(frames, PTO_RETRANSMISSION); |
| EXPECT_FALSE(notifier_.WillingToWrite()); |
| } |
| |
| } // namespace |
| } // namespace test |
| } // namespace quic |