| OLD | NEW |
| (Empty) |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | |
| 2 // Use of this source code is governed by a BSD-style license that can be | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #include "net/quic/quic_connection.h" | |
| 6 | |
| 7 #include "base/basictypes.h" | |
| 8 #include "base/bind.h" | |
| 9 #include "base/stl_util.h" | |
| 10 #include "net/base/net_errors.h" | |
| 11 #include "net/quic/congestion_control/loss_detection_interface.h" | |
| 12 #include "net/quic/congestion_control/send_algorithm_interface.h" | |
| 13 #include "net/quic/crypto/null_encrypter.h" | |
| 14 #include "net/quic/crypto/quic_decrypter.h" | |
| 15 #include "net/quic/crypto/quic_encrypter.h" | |
| 16 #include "net/quic/quic_ack_notifier.h" | |
| 17 #include "net/quic/quic_flags.h" | |
| 18 #include "net/quic/quic_protocol.h" | |
| 19 #include "net/quic/quic_utils.h" | |
| 20 #include "net/quic/test_tools/mock_clock.h" | |
| 21 #include "net/quic/test_tools/mock_random.h" | |
| 22 #include "net/quic/test_tools/quic_config_peer.h" | |
| 23 #include "net/quic/test_tools/quic_connection_peer.h" | |
| 24 #include "net/quic/test_tools/quic_framer_peer.h" | |
| 25 #include "net/quic/test_tools/quic_packet_creator_peer.h" | |
| 26 #include "net/quic/test_tools/quic_packet_generator_peer.h" | |
| 27 #include "net/quic/test_tools/quic_sent_packet_manager_peer.h" | |
| 28 #include "net/quic/test_tools/quic_test_utils.h" | |
| 29 #include "net/quic/test_tools/simple_quic_framer.h" | |
| 30 #include "net/test/gtest_util.h" | |
| 31 #include "testing/gmock/include/gmock/gmock.h" | |
| 32 #include "testing/gtest/include/gtest/gtest.h" | |
| 33 | |
| 34 using base::StringPiece; | |
| 35 using std::map; | |
| 36 using std::string; | |
| 37 using std::vector; | |
| 38 using testing::AnyNumber; | |
| 39 using testing::AtLeast; | |
| 40 using testing::ContainerEq; | |
| 41 using testing::Contains; | |
| 42 using testing::DoAll; | |
| 43 using testing::InSequence; | |
| 44 using testing::InvokeWithoutArgs; | |
| 45 using testing::NiceMock; | |
| 46 using testing::Ref; | |
| 47 using testing::Return; | |
| 48 using testing::SaveArg; | |
| 49 using testing::StrictMock; | |
| 50 using testing::_; | |
| 51 | |
| 52 namespace net { | |
| 53 namespace test { | |
| 54 namespace { | |
| 55 | |
| 56 const char data1[] = "foo"; | |
| 57 const char data2[] = "bar"; | |
| 58 | |
| 59 const bool kFin = true; | |
| 60 const bool kEntropyFlag = true; | |
| 61 | |
| 62 const QuicPacketEntropyHash kTestEntropyHash = 76; | |
| 63 | |
| 64 const int kDefaultRetransmissionTimeMs = 500; | |
| 65 | |
| 66 // TaggingEncrypter appends kTagSize bytes of |tag| to the end of each message. | |
| 67 class TaggingEncrypter : public QuicEncrypter { | |
| 68 public: | |
| 69 explicit TaggingEncrypter(uint8 tag) | |
| 70 : tag_(tag) { | |
| 71 } | |
| 72 | |
| 73 ~TaggingEncrypter() override {} | |
| 74 | |
| 75 // QuicEncrypter interface. | |
| 76 bool SetKey(StringPiece key) override { return true; } | |
| 77 | |
| 78 bool SetNoncePrefix(StringPiece nonce_prefix) override { return true; } | |
| 79 | |
| 80 bool Encrypt(StringPiece nonce, | |
| 81 StringPiece associated_data, | |
| 82 StringPiece plaintext, | |
| 83 unsigned char* output) override { | |
| 84 memcpy(output, plaintext.data(), plaintext.size()); | |
| 85 output += plaintext.size(); | |
| 86 memset(output, tag_, kTagSize); | |
| 87 return true; | |
| 88 } | |
| 89 | |
| 90 bool EncryptPacket(QuicPacketSequenceNumber sequence_number, | |
| 91 StringPiece associated_data, | |
| 92 StringPiece plaintext, | |
| 93 char* output, | |
| 94 size_t* output_length, | |
| 95 size_t max_output_length) override { | |
| 96 const size_t len = plaintext.size() + kTagSize; | |
| 97 if (max_output_length < len) { | |
| 98 return false; | |
| 99 } | |
| 100 Encrypt(StringPiece(), associated_data, plaintext, | |
| 101 reinterpret_cast<unsigned char*>(output)); | |
| 102 *output_length = len; | |
| 103 return true; | |
| 104 } | |
| 105 | |
| 106 size_t GetKeySize() const override { return 0; } | |
| 107 size_t GetNoncePrefixSize() const override { return 0; } | |
| 108 | |
| 109 size_t GetMaxPlaintextSize(size_t ciphertext_size) const override { | |
| 110 return ciphertext_size - kTagSize; | |
| 111 } | |
| 112 | |
| 113 size_t GetCiphertextSize(size_t plaintext_size) const override { | |
| 114 return plaintext_size + kTagSize; | |
| 115 } | |
| 116 | |
| 117 StringPiece GetKey() const override { return StringPiece(); } | |
| 118 | |
| 119 StringPiece GetNoncePrefix() const override { return StringPiece(); } | |
| 120 | |
| 121 private: | |
| 122 enum { | |
| 123 kTagSize = 12, | |
| 124 }; | |
| 125 | |
| 126 const uint8 tag_; | |
| 127 | |
| 128 DISALLOW_COPY_AND_ASSIGN(TaggingEncrypter); | |
| 129 }; | |
| 130 | |
| 131 // TaggingDecrypter ensures that the final kTagSize bytes of the message all | |
| 132 // have the same value and then removes them. | |
| 133 class TaggingDecrypter : public QuicDecrypter { | |
| 134 public: | |
| 135 ~TaggingDecrypter() override {} | |
| 136 | |
| 137 // QuicDecrypter interface | |
| 138 bool SetKey(StringPiece key) override { return true; } | |
| 139 | |
| 140 bool SetNoncePrefix(StringPiece nonce_prefix) override { return true; } | |
| 141 | |
| 142 bool DecryptPacket(QuicPacketSequenceNumber sequence_number, | |
| 143 const StringPiece& associated_data, | |
| 144 const StringPiece& ciphertext, | |
| 145 char* output, | |
| 146 size_t* output_length, | |
| 147 size_t max_output_length) override { | |
| 148 if (ciphertext.size() < kTagSize) { | |
| 149 return false; | |
| 150 } | |
| 151 if (!CheckTag(ciphertext, GetTag(ciphertext))) { | |
| 152 return false; | |
| 153 } | |
| 154 *output_length = ciphertext.size() - kTagSize; | |
| 155 memcpy(output, ciphertext.data(), *output_length); | |
| 156 return true; | |
| 157 } | |
| 158 | |
| 159 StringPiece GetKey() const override { return StringPiece(); } | |
| 160 StringPiece GetNoncePrefix() const override { return StringPiece(); } | |
| 161 | |
| 162 protected: | |
| 163 virtual uint8 GetTag(StringPiece ciphertext) { | |
| 164 return ciphertext.data()[ciphertext.size()-1]; | |
| 165 } | |
| 166 | |
| 167 private: | |
| 168 enum { | |
| 169 kTagSize = 12, | |
| 170 }; | |
| 171 | |
| 172 bool CheckTag(StringPiece ciphertext, uint8 tag) { | |
| 173 for (size_t i = ciphertext.size() - kTagSize; i < ciphertext.size(); i++) { | |
| 174 if (ciphertext.data()[i] != tag) { | |
| 175 return false; | |
| 176 } | |
| 177 } | |
| 178 | |
| 179 return true; | |
| 180 } | |
| 181 }; | |
| 182 | |
| 183 // StringTaggingDecrypter ensures that the final kTagSize bytes of the message | |
| 184 // match the expected value. | |
| 185 class StrictTaggingDecrypter : public TaggingDecrypter { | |
| 186 public: | |
| 187 explicit StrictTaggingDecrypter(uint8 tag) : tag_(tag) {} | |
| 188 ~StrictTaggingDecrypter() override {} | |
| 189 | |
| 190 // TaggingQuicDecrypter | |
| 191 uint8 GetTag(StringPiece ciphertext) override { return tag_; } | |
| 192 | |
| 193 private: | |
| 194 const uint8 tag_; | |
| 195 }; | |
| 196 | |
| 197 class TestConnectionHelper : public QuicConnectionHelperInterface { | |
| 198 public: | |
| 199 class TestAlarm : public QuicAlarm { | |
| 200 public: | |
| 201 explicit TestAlarm(QuicAlarm::Delegate* delegate) | |
| 202 : QuicAlarm(delegate) { | |
| 203 } | |
| 204 | |
| 205 void SetImpl() override {} | |
| 206 void CancelImpl() override {} | |
| 207 using QuicAlarm::Fire; | |
| 208 }; | |
| 209 | |
| 210 TestConnectionHelper(MockClock* clock, MockRandom* random_generator) | |
| 211 : clock_(clock), | |
| 212 random_generator_(random_generator) { | |
| 213 clock_->AdvanceTime(QuicTime::Delta::FromSeconds(1)); | |
| 214 } | |
| 215 | |
| 216 // QuicConnectionHelperInterface | |
| 217 const QuicClock* GetClock() const override { return clock_; } | |
| 218 | |
| 219 QuicRandom* GetRandomGenerator() override { return random_generator_; } | |
| 220 | |
| 221 QuicAlarm* CreateAlarm(QuicAlarm::Delegate* delegate) override { | |
| 222 return new TestAlarm(delegate); | |
| 223 } | |
| 224 | |
| 225 private: | |
| 226 MockClock* clock_; | |
| 227 MockRandom* random_generator_; | |
| 228 | |
| 229 DISALLOW_COPY_AND_ASSIGN(TestConnectionHelper); | |
| 230 }; | |
| 231 | |
| 232 class TestPacketWriter : public QuicPacketWriter { | |
| 233 public: | |
| 234 TestPacketWriter(QuicVersion version, MockClock *clock) | |
| 235 : version_(version), | |
| 236 framer_(SupportedVersions(version_)), | |
| 237 last_packet_size_(0), | |
| 238 write_blocked_(false), | |
| 239 block_on_next_write_(false), | |
| 240 is_write_blocked_data_buffered_(false), | |
| 241 final_bytes_of_last_packet_(0), | |
| 242 final_bytes_of_previous_packet_(0), | |
| 243 use_tagging_decrypter_(false), | |
| 244 packets_write_attempts_(0), | |
| 245 clock_(clock), | |
| 246 write_pause_time_delta_(QuicTime::Delta::Zero()) { | |
| 247 } | |
| 248 | |
| 249 // QuicPacketWriter interface | |
| 250 WriteResult WritePacket(const char* buffer, | |
| 251 size_t buf_len, | |
| 252 const IPAddressNumber& self_address, | |
| 253 const IPEndPoint& peer_address) override { | |
| 254 QuicEncryptedPacket packet(buffer, buf_len); | |
| 255 ++packets_write_attempts_; | |
| 256 | |
| 257 if (packet.length() >= sizeof(final_bytes_of_last_packet_)) { | |
| 258 final_bytes_of_previous_packet_ = final_bytes_of_last_packet_; | |
| 259 memcpy(&final_bytes_of_last_packet_, packet.data() + packet.length() - 4, | |
| 260 sizeof(final_bytes_of_last_packet_)); | |
| 261 } | |
| 262 | |
| 263 if (use_tagging_decrypter_) { | |
| 264 framer_.framer()->SetDecrypter(new TaggingDecrypter, ENCRYPTION_NONE); | |
| 265 } | |
| 266 EXPECT_TRUE(framer_.ProcessPacket(packet)); | |
| 267 if (block_on_next_write_) { | |
| 268 write_blocked_ = true; | |
| 269 block_on_next_write_ = false; | |
| 270 } | |
| 271 if (IsWriteBlocked()) { | |
| 272 return WriteResult(WRITE_STATUS_BLOCKED, -1); | |
| 273 } | |
| 274 last_packet_size_ = packet.length(); | |
| 275 | |
| 276 if (!write_pause_time_delta_.IsZero()) { | |
| 277 clock_->AdvanceTime(write_pause_time_delta_); | |
| 278 } | |
| 279 return WriteResult(WRITE_STATUS_OK, last_packet_size_); | |
| 280 } | |
| 281 | |
| 282 bool IsWriteBlockedDataBuffered() const override { | |
| 283 return is_write_blocked_data_buffered_; | |
| 284 } | |
| 285 | |
| 286 bool IsWriteBlocked() const override { return write_blocked_; } | |
| 287 | |
| 288 void SetWritable() override { write_blocked_ = false; } | |
| 289 | |
| 290 void BlockOnNextWrite() { block_on_next_write_ = true; } | |
| 291 | |
| 292 // Sets the amount of time that the writer should before the actual write. | |
| 293 void SetWritePauseTimeDelta(QuicTime::Delta delta) { | |
| 294 write_pause_time_delta_ = delta; | |
| 295 } | |
| 296 | |
| 297 const QuicPacketHeader& header() { return framer_.header(); } | |
| 298 | |
| 299 size_t frame_count() const { return framer_.num_frames(); } | |
| 300 | |
| 301 const vector<QuicAckFrame>& ack_frames() const { | |
| 302 return framer_.ack_frames(); | |
| 303 } | |
| 304 | |
| 305 const vector<QuicStopWaitingFrame>& stop_waiting_frames() const { | |
| 306 return framer_.stop_waiting_frames(); | |
| 307 } | |
| 308 | |
| 309 const vector<QuicConnectionCloseFrame>& connection_close_frames() const { | |
| 310 return framer_.connection_close_frames(); | |
| 311 } | |
| 312 | |
| 313 const vector<QuicStreamFrame>& stream_frames() const { | |
| 314 return framer_.stream_frames(); | |
| 315 } | |
| 316 | |
| 317 const vector<QuicPingFrame>& ping_frames() const { | |
| 318 return framer_.ping_frames(); | |
| 319 } | |
| 320 | |
| 321 size_t last_packet_size() { | |
| 322 return last_packet_size_; | |
| 323 } | |
| 324 | |
| 325 const QuicVersionNegotiationPacket* version_negotiation_packet() { | |
| 326 return framer_.version_negotiation_packet(); | |
| 327 } | |
| 328 | |
| 329 void set_is_write_blocked_data_buffered(bool buffered) { | |
| 330 is_write_blocked_data_buffered_ = buffered; | |
| 331 } | |
| 332 | |
| 333 void set_is_server(bool is_server) { | |
| 334 // We invert is_server here, because the framer needs to parse packets | |
| 335 // we send. | |
| 336 QuicFramerPeer::SetIsServer(framer_.framer(), !is_server); | |
| 337 } | |
| 338 | |
| 339 // final_bytes_of_last_packet_ returns the last four bytes of the previous | |
| 340 // packet as a little-endian, uint32. This is intended to be used with a | |
| 341 // TaggingEncrypter so that tests can determine which encrypter was used for | |
| 342 // a given packet. | |
| 343 uint32 final_bytes_of_last_packet() { return final_bytes_of_last_packet_; } | |
| 344 | |
| 345 // Returns the final bytes of the second to last packet. | |
| 346 uint32 final_bytes_of_previous_packet() { | |
| 347 return final_bytes_of_previous_packet_; | |
| 348 } | |
| 349 | |
| 350 void use_tagging_decrypter() { | |
| 351 use_tagging_decrypter_ = true; | |
| 352 } | |
| 353 | |
| 354 uint32 packets_write_attempts() { return packets_write_attempts_; } | |
| 355 | |
| 356 void Reset() { framer_.Reset(); } | |
| 357 | |
| 358 void SetSupportedVersions(const QuicVersionVector& versions) { | |
| 359 framer_.SetSupportedVersions(versions); | |
| 360 } | |
| 361 | |
| 362 private: | |
| 363 QuicVersion version_; | |
| 364 SimpleQuicFramer framer_; | |
| 365 size_t last_packet_size_; | |
| 366 bool write_blocked_; | |
| 367 bool block_on_next_write_; | |
| 368 bool is_write_blocked_data_buffered_; | |
| 369 uint32 final_bytes_of_last_packet_; | |
| 370 uint32 final_bytes_of_previous_packet_; | |
| 371 bool use_tagging_decrypter_; | |
| 372 uint32 packets_write_attempts_; | |
| 373 MockClock *clock_; | |
| 374 // If non-zero, the clock will pause during WritePacket for this amount of | |
| 375 // time. | |
| 376 QuicTime::Delta write_pause_time_delta_; | |
| 377 | |
| 378 DISALLOW_COPY_AND_ASSIGN(TestPacketWriter); | |
| 379 }; | |
| 380 | |
| 381 class TestConnection : public QuicConnection { | |
| 382 public: | |
| 383 TestConnection(QuicConnectionId connection_id, | |
| 384 IPEndPoint address, | |
| 385 TestConnectionHelper* helper, | |
| 386 const PacketWriterFactory& factory, | |
| 387 bool is_server, | |
| 388 QuicVersion version) | |
| 389 : QuicConnection(connection_id, | |
| 390 address, | |
| 391 helper, | |
| 392 factory, | |
| 393 /* owns_writer= */ false, | |
| 394 is_server, | |
| 395 /* is_secure= */ false, | |
| 396 SupportedVersions(version)) { | |
| 397 // Disable tail loss probes for most tests. | |
| 398 QuicSentPacketManagerPeer::SetMaxTailLossProbes( | |
| 399 QuicConnectionPeer::GetSentPacketManager(this), 0); | |
| 400 writer()->set_is_server(is_server); | |
| 401 } | |
| 402 | |
| 403 void SendAck() { | |
| 404 QuicConnectionPeer::SendAck(this); | |
| 405 } | |
| 406 | |
| 407 void SetSendAlgorithm(SendAlgorithmInterface* send_algorithm) { | |
| 408 QuicConnectionPeer::SetSendAlgorithm(this, send_algorithm); | |
| 409 } | |
| 410 | |
| 411 void SetLossAlgorithm(LossDetectionInterface* loss_algorithm) { | |
| 412 QuicSentPacketManagerPeer::SetLossAlgorithm( | |
| 413 QuicConnectionPeer::GetSentPacketManager(this), loss_algorithm); | |
| 414 } | |
| 415 | |
| 416 void SendPacket(EncryptionLevel level, | |
| 417 QuicPacketSequenceNumber sequence_number, | |
| 418 QuicPacket* packet, | |
| 419 QuicPacketEntropyHash entropy_hash, | |
| 420 HasRetransmittableData retransmittable) { | |
| 421 RetransmittableFrames* retransmittable_frames = | |
| 422 retransmittable == HAS_RETRANSMITTABLE_DATA | |
| 423 ? new RetransmittableFrames() | |
| 424 : nullptr; | |
| 425 QuicEncryptedPacket* encrypted = | |
| 426 QuicConnectionPeer::GetFramer(this) | |
| 427 ->EncryptPacket(ENCRYPTION_NONE, sequence_number, *packet); | |
| 428 delete packet; | |
| 429 OnSerializedPacket(SerializedPacket(sequence_number, | |
| 430 PACKET_6BYTE_SEQUENCE_NUMBER, encrypted, | |
| 431 entropy_hash, retransmittable_frames)); | |
| 432 } | |
| 433 | |
| 434 QuicConsumedData SendStreamDataWithString( | |
| 435 QuicStreamId id, | |
| 436 StringPiece data, | |
| 437 QuicStreamOffset offset, | |
| 438 bool fin, | |
| 439 QuicAckNotifier::DelegateInterface* delegate) { | |
| 440 return SendStreamDataWithStringHelper(id, data, offset, fin, | |
| 441 MAY_FEC_PROTECT, delegate); | |
| 442 } | |
| 443 | |
| 444 QuicConsumedData SendStreamDataWithStringWithFec( | |
| 445 QuicStreamId id, | |
| 446 StringPiece data, | |
| 447 QuicStreamOffset offset, | |
| 448 bool fin, | |
| 449 QuicAckNotifier::DelegateInterface* delegate) { | |
| 450 return SendStreamDataWithStringHelper(id, data, offset, fin, | |
| 451 MUST_FEC_PROTECT, delegate); | |
| 452 } | |
| 453 | |
| 454 QuicConsumedData SendStreamDataWithStringHelper( | |
| 455 QuicStreamId id, | |
| 456 StringPiece data, | |
| 457 QuicStreamOffset offset, | |
| 458 bool fin, | |
| 459 FecProtection fec_protection, | |
| 460 QuicAckNotifier::DelegateInterface* delegate) { | |
| 461 IOVector data_iov; | |
| 462 if (!data.empty()) { | |
| 463 data_iov.Append(const_cast<char*>(data.data()), data.size()); | |
| 464 } | |
| 465 return QuicConnection::SendStreamData(id, data_iov, offset, fin, | |
| 466 fec_protection, delegate); | |
| 467 } | |
| 468 | |
| 469 QuicConsumedData SendStreamData3() { | |
| 470 return SendStreamDataWithString(kClientDataStreamId1, "food", 0, !kFin, | |
| 471 nullptr); | |
| 472 } | |
| 473 | |
| 474 QuicConsumedData SendStreamData3WithFec() { | |
| 475 return SendStreamDataWithStringWithFec(kClientDataStreamId1, "food", 0, | |
| 476 !kFin, nullptr); | |
| 477 } | |
| 478 | |
| 479 QuicConsumedData SendStreamData5() { | |
| 480 return SendStreamDataWithString(kClientDataStreamId2, "food2", 0, !kFin, | |
| 481 nullptr); | |
| 482 } | |
| 483 | |
| 484 QuicConsumedData SendStreamData5WithFec() { | |
| 485 return SendStreamDataWithStringWithFec(kClientDataStreamId2, "food2", 0, | |
| 486 !kFin, nullptr); | |
| 487 } | |
| 488 // Ensures the connection can write stream data before writing. | |
| 489 QuicConsumedData EnsureWritableAndSendStreamData5() { | |
| 490 EXPECT_TRUE(CanWriteStreamData()); | |
| 491 return SendStreamData5(); | |
| 492 } | |
| 493 | |
| 494 // The crypto stream has special semantics so that it is not blocked by a | |
| 495 // congestion window limitation, and also so that it gets put into a separate | |
| 496 // packet (so that it is easier to reason about a crypto frame not being | |
| 497 // split needlessly across packet boundaries). As a result, we have separate | |
| 498 // tests for some cases for this stream. | |
| 499 QuicConsumedData SendCryptoStreamData() { | |
| 500 return SendStreamDataWithString(kCryptoStreamId, "chlo", 0, !kFin, nullptr); | |
| 501 } | |
| 502 | |
| 503 bool is_server() { | |
| 504 return QuicConnectionPeer::IsServer(this); | |
| 505 } | |
| 506 | |
| 507 void set_version(QuicVersion version) { | |
| 508 QuicConnectionPeer::GetFramer(this)->set_version(version); | |
| 509 } | |
| 510 | |
| 511 void SetSupportedVersions(const QuicVersionVector& versions) { | |
| 512 QuicConnectionPeer::GetFramer(this)->SetSupportedVersions(versions); | |
| 513 writer()->SetSupportedVersions(versions); | |
| 514 } | |
| 515 | |
| 516 void set_is_server(bool is_server) { | |
| 517 writer()->set_is_server(is_server); | |
| 518 QuicConnectionPeer::SetIsServer(this, is_server); | |
| 519 } | |
| 520 | |
| 521 TestConnectionHelper::TestAlarm* GetAckAlarm() { | |
| 522 return reinterpret_cast<TestConnectionHelper::TestAlarm*>( | |
| 523 QuicConnectionPeer::GetAckAlarm(this)); | |
| 524 } | |
| 525 | |
| 526 TestConnectionHelper::TestAlarm* GetPingAlarm() { | |
| 527 return reinterpret_cast<TestConnectionHelper::TestAlarm*>( | |
| 528 QuicConnectionPeer::GetPingAlarm(this)); | |
| 529 } | |
| 530 | |
| 531 TestConnectionHelper::TestAlarm* GetFecAlarm() { | |
| 532 return reinterpret_cast<TestConnectionHelper::TestAlarm*>( | |
| 533 QuicConnectionPeer::GetFecAlarm(this)); | |
| 534 } | |
| 535 | |
| 536 TestConnectionHelper::TestAlarm* GetResumeWritesAlarm() { | |
| 537 return reinterpret_cast<TestConnectionHelper::TestAlarm*>( | |
| 538 QuicConnectionPeer::GetResumeWritesAlarm(this)); | |
| 539 } | |
| 540 | |
| 541 TestConnectionHelper::TestAlarm* GetRetransmissionAlarm() { | |
| 542 return reinterpret_cast<TestConnectionHelper::TestAlarm*>( | |
| 543 QuicConnectionPeer::GetRetransmissionAlarm(this)); | |
| 544 } | |
| 545 | |
| 546 TestConnectionHelper::TestAlarm* GetSendAlarm() { | |
| 547 return reinterpret_cast<TestConnectionHelper::TestAlarm*>( | |
| 548 QuicConnectionPeer::GetSendAlarm(this)); | |
| 549 } | |
| 550 | |
| 551 TestConnectionHelper::TestAlarm* GetTimeoutAlarm() { | |
| 552 return reinterpret_cast<TestConnectionHelper::TestAlarm*>( | |
| 553 QuicConnectionPeer::GetTimeoutAlarm(this)); | |
| 554 } | |
| 555 | |
| 556 using QuicConnection::SelectMutualVersion; | |
| 557 | |
| 558 private: | |
| 559 TestPacketWriter* writer() { | |
| 560 return static_cast<TestPacketWriter*>(QuicConnection::writer()); | |
| 561 } | |
| 562 | |
| 563 DISALLOW_COPY_AND_ASSIGN(TestConnection); | |
| 564 }; | |
| 565 | |
| 566 // Used for testing packets revived from FEC packets. | |
| 567 class FecQuicConnectionDebugVisitor | |
| 568 : public QuicConnectionDebugVisitor { | |
| 569 public: | |
| 570 void OnRevivedPacket(const QuicPacketHeader& header, | |
| 571 StringPiece data) override { | |
| 572 revived_header_ = header; | |
| 573 } | |
| 574 | |
| 575 // Public accessor method. | |
| 576 QuicPacketHeader revived_header() const { | |
| 577 return revived_header_; | |
| 578 } | |
| 579 | |
| 580 private: | |
| 581 QuicPacketHeader revived_header_; | |
| 582 }; | |
| 583 | |
| 584 class MockPacketWriterFactory : public QuicConnection::PacketWriterFactory { | |
| 585 public: | |
| 586 explicit MockPacketWriterFactory(QuicPacketWriter* writer) { | |
| 587 ON_CALL(*this, Create(_)).WillByDefault(Return(writer)); | |
| 588 } | |
| 589 ~MockPacketWriterFactory() override {} | |
| 590 | |
| 591 MOCK_CONST_METHOD1(Create, QuicPacketWriter*(QuicConnection* connection)); | |
| 592 }; | |
| 593 | |
| 594 class QuicConnectionTest : public ::testing::TestWithParam<QuicVersion> { | |
| 595 protected: | |
| 596 QuicConnectionTest() | |
| 597 : connection_id_(42), | |
| 598 framer_(SupportedVersions(version()), QuicTime::Zero(), false), | |
| 599 peer_creator_(connection_id_, &framer_, &random_generator_), | |
| 600 send_algorithm_(new StrictMock<MockSendAlgorithm>), | |
| 601 loss_algorithm_(new MockLossAlgorithm()), | |
| 602 helper_(new TestConnectionHelper(&clock_, &random_generator_)), | |
| 603 writer_(new TestPacketWriter(version(), &clock_)), | |
| 604 factory_(writer_.get()), | |
| 605 connection_(connection_id_, | |
| 606 IPEndPoint(), | |
| 607 helper_.get(), | |
| 608 factory_, | |
| 609 false, | |
| 610 version()), | |
| 611 creator_(QuicConnectionPeer::GetPacketCreator(&connection_)), | |
| 612 generator_(QuicConnectionPeer::GetPacketGenerator(&connection_)), | |
| 613 manager_(QuicConnectionPeer::GetSentPacketManager(&connection_)), | |
| 614 frame1_(1, false, 0, MakeIOVector(data1)), | |
| 615 frame2_(1, false, 3, MakeIOVector(data2)), | |
| 616 sequence_number_length_(PACKET_6BYTE_SEQUENCE_NUMBER), | |
| 617 connection_id_length_(PACKET_8BYTE_CONNECTION_ID) { | |
| 618 connection_.set_visitor(&visitor_); | |
| 619 connection_.SetSendAlgorithm(send_algorithm_); | |
| 620 connection_.SetLossAlgorithm(loss_algorithm_); | |
| 621 framer_.set_received_entropy_calculator(&entropy_calculator_); | |
| 622 EXPECT_CALL( | |
| 623 *send_algorithm_, TimeUntilSend(_, _, _)).WillRepeatedly(Return( | |
| 624 QuicTime::Delta::Zero())); | |
| 625 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)) | |
| 626 .Times(AnyNumber()); | |
| 627 EXPECT_CALL(*send_algorithm_, RetransmissionDelay()).WillRepeatedly( | |
| 628 Return(QuicTime::Delta::Zero())); | |
| 629 EXPECT_CALL(*send_algorithm_, GetCongestionWindow()).WillRepeatedly( | |
| 630 Return(kMaxPacketSize)); | |
| 631 ON_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)) | |
| 632 .WillByDefault(Return(true)); | |
| 633 EXPECT_CALL(*send_algorithm_, HasReliableBandwidthEstimate()) | |
| 634 .Times(AnyNumber()); | |
| 635 EXPECT_CALL(*send_algorithm_, BandwidthEstimate()) | |
| 636 .Times(AnyNumber()) | |
| 637 .WillRepeatedly(Return(QuicBandwidth::Zero())); | |
| 638 EXPECT_CALL(*send_algorithm_, InSlowStart()).Times(AnyNumber()); | |
| 639 EXPECT_CALL(*send_algorithm_, InRecovery()).Times(AnyNumber()); | |
| 640 EXPECT_CALL(visitor_, WillingAndAbleToWrite()).Times(AnyNumber()); | |
| 641 EXPECT_CALL(visitor_, HasPendingHandshake()).Times(AnyNumber()); | |
| 642 EXPECT_CALL(visitor_, OnCanWrite()).Times(AnyNumber()); | |
| 643 EXPECT_CALL(visitor_, HasOpenDataStreams()).WillRepeatedly(Return(false)); | |
| 644 EXPECT_CALL(visitor_, OnCongestionWindowChange(_)).Times(AnyNumber()); | |
| 645 | |
| 646 EXPECT_CALL(*loss_algorithm_, GetLossTimeout()) | |
| 647 .WillRepeatedly(Return(QuicTime::Zero())); | |
| 648 EXPECT_CALL(*loss_algorithm_, DetectLostPackets(_, _, _, _)) | |
| 649 .WillRepeatedly(Return(SequenceNumberSet())); | |
| 650 } | |
| 651 | |
| 652 QuicVersion version() { | |
| 653 return GetParam(); | |
| 654 } | |
| 655 | |
| 656 QuicAckFrame* outgoing_ack() { | |
| 657 QuicConnectionPeer::PopulateAckFrame(&connection_, &ack_); | |
| 658 return &ack_; | |
| 659 } | |
| 660 | |
| 661 QuicStopWaitingFrame* stop_waiting() { | |
| 662 QuicConnectionPeer::PopulateStopWaitingFrame(&connection_, &stop_waiting_); | |
| 663 return &stop_waiting_; | |
| 664 } | |
| 665 | |
| 666 QuicPacketSequenceNumber least_unacked() { | |
| 667 if (writer_->stop_waiting_frames().empty()) { | |
| 668 return 0; | |
| 669 } | |
| 670 return writer_->stop_waiting_frames()[0].least_unacked; | |
| 671 } | |
| 672 | |
| 673 void use_tagging_decrypter() { | |
| 674 writer_->use_tagging_decrypter(); | |
| 675 } | |
| 676 | |
| 677 void ProcessPacket(QuicPacketSequenceNumber number) { | |
| 678 EXPECT_CALL(visitor_, OnStreamFrames(_)).Times(1); | |
| 679 ProcessDataPacket(number, 0, !kEntropyFlag); | |
| 680 } | |
| 681 | |
| 682 QuicPacketEntropyHash ProcessFramePacket(QuicFrame frame) { | |
| 683 QuicFrames frames; | |
| 684 frames.push_back(QuicFrame(frame)); | |
| 685 QuicPacketCreatorPeer::SetSendVersionInPacket(&peer_creator_, | |
| 686 connection_.is_server()); | |
| 687 SerializedPacket serialized_packet = | |
| 688 peer_creator_.SerializeAllFrames(frames); | |
| 689 scoped_ptr<QuicEncryptedPacket> encrypted(serialized_packet.packet); | |
| 690 connection_.ProcessUdpPacket(IPEndPoint(), IPEndPoint(), *encrypted); | |
| 691 return serialized_packet.entropy_hash; | |
| 692 } | |
| 693 | |
| 694 size_t ProcessDataPacket(QuicPacketSequenceNumber number, | |
| 695 QuicFecGroupNumber fec_group, | |
| 696 bool entropy_flag) { | |
| 697 return ProcessDataPacketAtLevel(number, fec_group, entropy_flag, | |
| 698 ENCRYPTION_NONE); | |
| 699 } | |
| 700 | |
| 701 size_t ProcessDataPacketAtLevel(QuicPacketSequenceNumber number, | |
| 702 QuicFecGroupNumber fec_group, | |
| 703 bool entropy_flag, | |
| 704 EncryptionLevel level) { | |
| 705 scoped_ptr<QuicPacket> packet(ConstructDataPacket(number, fec_group, | |
| 706 entropy_flag)); | |
| 707 scoped_ptr<QuicEncryptedPacket> encrypted(framer_.EncryptPacket( | |
| 708 level, number, *packet)); | |
| 709 connection_.ProcessUdpPacket(IPEndPoint(), IPEndPoint(), *encrypted); | |
| 710 return encrypted->length(); | |
| 711 } | |
| 712 | |
| 713 void ProcessPingPacket(QuicPacketSequenceNumber number) { | |
| 714 scoped_ptr<QuicPacket> packet(ConstructPingPacket(number)); | |
| 715 scoped_ptr<QuicEncryptedPacket> encrypted(framer_.EncryptPacket( | |
| 716 ENCRYPTION_NONE, number, *packet)); | |
| 717 connection_.ProcessUdpPacket(IPEndPoint(), IPEndPoint(), *encrypted); | |
| 718 } | |
| 719 | |
| 720 void ProcessClosePacket(QuicPacketSequenceNumber number, | |
| 721 QuicFecGroupNumber fec_group) { | |
| 722 scoped_ptr<QuicPacket> packet(ConstructClosePacket(number, fec_group)); | |
| 723 scoped_ptr<QuicEncryptedPacket> encrypted(framer_.EncryptPacket( | |
| 724 ENCRYPTION_NONE, number, *packet)); | |
| 725 connection_.ProcessUdpPacket(IPEndPoint(), IPEndPoint(), *encrypted); | |
| 726 } | |
| 727 | |
| 728 size_t ProcessFecProtectedPacket(QuicPacketSequenceNumber number, | |
| 729 bool expect_revival, bool entropy_flag) { | |
| 730 if (expect_revival) { | |
| 731 EXPECT_CALL(visitor_, OnStreamFrames(_)).Times(1); | |
| 732 } | |
| 733 EXPECT_CALL(visitor_, OnStreamFrames(_)).Times(1). | |
| 734 RetiresOnSaturation(); | |
| 735 return ProcessDataPacket(number, 1, entropy_flag); | |
| 736 } | |
| 737 | |
| 738 // Processes an FEC packet that covers the packets that would have been | |
| 739 // received. | |
| 740 size_t ProcessFecPacket(QuicPacketSequenceNumber number, | |
| 741 QuicPacketSequenceNumber min_protected_packet, | |
| 742 bool expect_revival, | |
| 743 bool entropy_flag, | |
| 744 QuicPacket* packet) { | |
| 745 if (expect_revival) { | |
| 746 EXPECT_CALL(visitor_, OnStreamFrames(_)).Times(1); | |
| 747 } | |
| 748 | |
| 749 // Construct the decrypted data packet so we can compute the correct | |
| 750 // redundancy. If |packet| has been provided then use that, otherwise | |
| 751 // construct a default data packet. | |
| 752 scoped_ptr<QuicPacket> data_packet; | |
| 753 if (packet) { | |
| 754 data_packet.reset(packet); | |
| 755 } else { | |
| 756 data_packet.reset(ConstructDataPacket(number, 1, !kEntropyFlag)); | |
| 757 } | |
| 758 | |
| 759 header_.public_header.connection_id = connection_id_; | |
| 760 header_.public_header.reset_flag = false; | |
| 761 header_.public_header.version_flag = false; | |
| 762 header_.public_header.sequence_number_length = sequence_number_length_; | |
| 763 header_.public_header.connection_id_length = connection_id_length_; | |
| 764 header_.packet_sequence_number = number; | |
| 765 header_.entropy_flag = entropy_flag; | |
| 766 header_.fec_flag = true; | |
| 767 header_.is_in_fec_group = IN_FEC_GROUP; | |
| 768 header_.fec_group = min_protected_packet; | |
| 769 QuicFecData fec_data; | |
| 770 fec_data.fec_group = header_.fec_group; | |
| 771 | |
| 772 // Since all data packets in this test have the same payload, the | |
| 773 // redundancy is either equal to that payload or the xor of that payload | |
| 774 // with itself, depending on the number of packets. | |
| 775 if (((number - min_protected_packet) % 2) == 0) { | |
| 776 for (size_t i = GetStartOfFecProtectedData( | |
| 777 header_.public_header.connection_id_length, | |
| 778 header_.public_header.version_flag, | |
| 779 header_.public_header.sequence_number_length); | |
| 780 i < data_packet->length(); ++i) { | |
| 781 data_packet->mutable_data()[i] ^= data_packet->data()[i]; | |
| 782 } | |
| 783 } | |
| 784 fec_data.redundancy = data_packet->FecProtectedData(); | |
| 785 | |
| 786 scoped_ptr<QuicPacket> fec_packet( | |
| 787 framer_.BuildFecPacket(header_, fec_data)); | |
| 788 scoped_ptr<QuicEncryptedPacket> encrypted( | |
| 789 framer_.EncryptPacket(ENCRYPTION_NONE, number, *fec_packet)); | |
| 790 | |
| 791 connection_.ProcessUdpPacket(IPEndPoint(), IPEndPoint(), *encrypted); | |
| 792 return encrypted->length(); | |
| 793 } | |
| 794 | |
| 795 QuicByteCount SendStreamDataToPeer(QuicStreamId id, | |
| 796 StringPiece data, | |
| 797 QuicStreamOffset offset, | |
| 798 bool fin, | |
| 799 QuicPacketSequenceNumber* last_packet) { | |
| 800 QuicByteCount packet_size; | |
| 801 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)) | |
| 802 .WillOnce(DoAll(SaveArg<3>(&packet_size), Return(true))); | |
| 803 connection_.SendStreamDataWithString(id, data, offset, fin, nullptr); | |
| 804 if (last_packet != nullptr) { | |
| 805 *last_packet = creator_->sequence_number(); | |
| 806 } | |
| 807 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)) | |
| 808 .Times(AnyNumber()); | |
| 809 return packet_size; | |
| 810 } | |
| 811 | |
| 812 void SendAckPacketToPeer() { | |
| 813 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)).Times(1); | |
| 814 connection_.SendAck(); | |
| 815 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)) | |
| 816 .Times(AnyNumber()); | |
| 817 } | |
| 818 | |
| 819 QuicPacketEntropyHash ProcessAckPacket(QuicAckFrame* frame) { | |
| 820 return ProcessFramePacket(QuicFrame(frame)); | |
| 821 } | |
| 822 | |
| 823 QuicPacketEntropyHash ProcessStopWaitingPacket(QuicStopWaitingFrame* frame) { | |
| 824 return ProcessFramePacket(QuicFrame(frame)); | |
| 825 } | |
| 826 | |
| 827 QuicPacketEntropyHash ProcessGoAwayPacket(QuicGoAwayFrame* frame) { | |
| 828 return ProcessFramePacket(QuicFrame(frame)); | |
| 829 } | |
| 830 | |
| 831 bool IsMissing(QuicPacketSequenceNumber number) { | |
| 832 return IsAwaitingPacket(*outgoing_ack(), number); | |
| 833 } | |
| 834 | |
| 835 QuicPacket* ConstructDataPacket(QuicPacketSequenceNumber number, | |
| 836 QuicFecGroupNumber fec_group, | |
| 837 bool entropy_flag) { | |
| 838 header_.public_header.connection_id = connection_id_; | |
| 839 header_.public_header.reset_flag = false; | |
| 840 header_.public_header.version_flag = false; | |
| 841 header_.public_header.sequence_number_length = sequence_number_length_; | |
| 842 header_.public_header.connection_id_length = connection_id_length_; | |
| 843 header_.entropy_flag = entropy_flag; | |
| 844 header_.fec_flag = false; | |
| 845 header_.packet_sequence_number = number; | |
| 846 header_.is_in_fec_group = fec_group == 0u ? NOT_IN_FEC_GROUP : IN_FEC_GROUP; | |
| 847 header_.fec_group = fec_group; | |
| 848 | |
| 849 QuicFrames frames; | |
| 850 QuicFrame frame(&frame1_); | |
| 851 frames.push_back(frame); | |
| 852 QuicPacket* packet = BuildUnsizedDataPacket(&framer_, header_, frames); | |
| 853 EXPECT_TRUE(packet != nullptr); | |
| 854 return packet; | |
| 855 } | |
| 856 | |
| 857 QuicPacket* ConstructPingPacket(QuicPacketSequenceNumber number) { | |
| 858 header_.public_header.connection_id = connection_id_; | |
| 859 header_.packet_sequence_number = number; | |
| 860 header_.public_header.reset_flag = false; | |
| 861 header_.public_header.version_flag = false; | |
| 862 header_.entropy_flag = false; | |
| 863 header_.fec_flag = false; | |
| 864 header_.is_in_fec_group = NOT_IN_FEC_GROUP; | |
| 865 header_.fec_group = 0; | |
| 866 | |
| 867 QuicPingFrame ping; | |
| 868 | |
| 869 QuicFrames frames; | |
| 870 QuicFrame frame(&ping); | |
| 871 frames.push_back(frame); | |
| 872 QuicPacket* packet = BuildUnsizedDataPacket(&framer_, header_, frames); | |
| 873 EXPECT_TRUE(packet != nullptr); | |
| 874 return packet; | |
| 875 } | |
| 876 | |
| 877 QuicPacket* ConstructClosePacket(QuicPacketSequenceNumber number, | |
| 878 QuicFecGroupNumber fec_group) { | |
| 879 header_.public_header.connection_id = connection_id_; | |
| 880 header_.packet_sequence_number = number; | |
| 881 header_.public_header.reset_flag = false; | |
| 882 header_.public_header.version_flag = false; | |
| 883 header_.entropy_flag = false; | |
| 884 header_.fec_flag = false; | |
| 885 header_.is_in_fec_group = fec_group == 0u ? NOT_IN_FEC_GROUP : IN_FEC_GROUP; | |
| 886 header_.fec_group = fec_group; | |
| 887 | |
| 888 QuicConnectionCloseFrame qccf; | |
| 889 qccf.error_code = QUIC_PEER_GOING_AWAY; | |
| 890 | |
| 891 QuicFrames frames; | |
| 892 QuicFrame frame(&qccf); | |
| 893 frames.push_back(frame); | |
| 894 QuicPacket* packet = BuildUnsizedDataPacket(&framer_, header_, frames); | |
| 895 EXPECT_TRUE(packet != nullptr); | |
| 896 return packet; | |
| 897 } | |
| 898 | |
| 899 QuicTime::Delta DefaultRetransmissionTime() { | |
| 900 return QuicTime::Delta::FromMilliseconds(kDefaultRetransmissionTimeMs); | |
| 901 } | |
| 902 | |
| 903 QuicTime::Delta DefaultDelayedAckTime() { | |
| 904 return QuicTime::Delta::FromMilliseconds(kMaxDelayedAckTimeMs); | |
| 905 } | |
| 906 | |
| 907 // Initialize a frame acknowledging all packets up to largest_observed. | |
| 908 const QuicAckFrame InitAckFrame(QuicPacketSequenceNumber largest_observed) { | |
| 909 QuicAckFrame frame(MakeAckFrame(largest_observed)); | |
| 910 if (largest_observed > 0) { | |
| 911 frame.entropy_hash = | |
| 912 QuicConnectionPeer::GetSentEntropyHash(&connection_, | |
| 913 largest_observed); | |
| 914 } | |
| 915 return frame; | |
| 916 } | |
| 917 | |
| 918 const QuicStopWaitingFrame InitStopWaitingFrame( | |
| 919 QuicPacketSequenceNumber least_unacked) { | |
| 920 QuicStopWaitingFrame frame; | |
| 921 frame.least_unacked = least_unacked; | |
| 922 return frame; | |
| 923 } | |
| 924 | |
| 925 // Explicitly nack a packet. | |
| 926 void NackPacket(QuicPacketSequenceNumber missing, QuicAckFrame* frame) { | |
| 927 frame->missing_packets.insert(missing); | |
| 928 frame->entropy_hash ^= | |
| 929 QuicConnectionPeer::PacketEntropy(&connection_, missing); | |
| 930 } | |
| 931 | |
| 932 // Undo nacking a packet within the frame. | |
| 933 void AckPacket(QuicPacketSequenceNumber arrived, QuicAckFrame* frame) { | |
| 934 EXPECT_THAT(frame->missing_packets, Contains(arrived)); | |
| 935 frame->missing_packets.erase(arrived); | |
| 936 frame->entropy_hash ^= | |
| 937 QuicConnectionPeer::PacketEntropy(&connection_, arrived); | |
| 938 } | |
| 939 | |
| 940 void TriggerConnectionClose() { | |
| 941 // Send an erroneous packet to close the connection. | |
| 942 EXPECT_CALL(visitor_, | |
| 943 OnConnectionClosed(QUIC_INVALID_PACKET_HEADER, false)); | |
| 944 // Call ProcessDataPacket rather than ProcessPacket, as we should not get a | |
| 945 // packet call to the visitor. | |
| 946 ProcessDataPacket(6000, 0, !kEntropyFlag); | |
| 947 EXPECT_FALSE(QuicConnectionPeer::GetConnectionClosePacket(&connection_) == | |
| 948 nullptr); | |
| 949 } | |
| 950 | |
| 951 void BlockOnNextWrite() { | |
| 952 writer_->BlockOnNextWrite(); | |
| 953 EXPECT_CALL(visitor_, OnWriteBlocked()).Times(AtLeast(1)); | |
| 954 } | |
| 955 | |
| 956 void SetWritePauseTimeDelta(QuicTime::Delta delta) { | |
| 957 writer_->SetWritePauseTimeDelta(delta); | |
| 958 } | |
| 959 | |
| 960 void CongestionBlockWrites() { | |
| 961 EXPECT_CALL(*send_algorithm_, | |
| 962 TimeUntilSend(_, _, _)).WillRepeatedly( | |
| 963 testing::Return(QuicTime::Delta::FromSeconds(1))); | |
| 964 } | |
| 965 | |
| 966 void CongestionUnblockWrites() { | |
| 967 EXPECT_CALL(*send_algorithm_, | |
| 968 TimeUntilSend(_, _, _)).WillRepeatedly( | |
| 969 testing::Return(QuicTime::Delta::Zero())); | |
| 970 } | |
| 971 | |
| 972 QuicConnectionId connection_id_; | |
| 973 QuicFramer framer_; | |
| 974 QuicPacketCreator peer_creator_; | |
| 975 MockEntropyCalculator entropy_calculator_; | |
| 976 | |
| 977 MockSendAlgorithm* send_algorithm_; | |
| 978 MockLossAlgorithm* loss_algorithm_; | |
| 979 MockClock clock_; | |
| 980 MockRandom random_generator_; | |
| 981 scoped_ptr<TestConnectionHelper> helper_; | |
| 982 scoped_ptr<TestPacketWriter> writer_; | |
| 983 NiceMock<MockPacketWriterFactory> factory_; | |
| 984 TestConnection connection_; | |
| 985 QuicPacketCreator* creator_; | |
| 986 QuicPacketGenerator* generator_; | |
| 987 QuicSentPacketManager* manager_; | |
| 988 StrictMock<MockConnectionVisitor> visitor_; | |
| 989 | |
| 990 QuicPacketHeader header_; | |
| 991 QuicStreamFrame frame1_; | |
| 992 QuicStreamFrame frame2_; | |
| 993 QuicAckFrame ack_; | |
| 994 QuicStopWaitingFrame stop_waiting_; | |
| 995 QuicSequenceNumberLength sequence_number_length_; | |
| 996 QuicConnectionIdLength connection_id_length_; | |
| 997 | |
| 998 private: | |
| 999 DISALLOW_COPY_AND_ASSIGN(QuicConnectionTest); | |
| 1000 }; | |
| 1001 | |
| 1002 // Run all end to end tests with all supported versions. | |
| 1003 INSTANTIATE_TEST_CASE_P(SupportedVersion, | |
| 1004 QuicConnectionTest, | |
| 1005 ::testing::ValuesIn(QuicSupportedVersions())); | |
| 1006 | |
| 1007 TEST_P(QuicConnectionTest, MaxPacketSize) { | |
| 1008 EXPECT_FALSE(connection_.is_server()); | |
| 1009 EXPECT_EQ(1350u, connection_.max_packet_length()); | |
| 1010 } | |
| 1011 | |
| 1012 TEST_P(QuicConnectionTest, SmallerServerMaxPacketSize) { | |
| 1013 ValueRestore<bool> old_flag(&FLAGS_quic_small_default_packet_size, true); | |
| 1014 QuicConnectionId connection_id = 42; | |
| 1015 bool kIsServer = true; | |
| 1016 TestConnection connection(connection_id, IPEndPoint(), helper_.get(), | |
| 1017 factory_, kIsServer, version()); | |
| 1018 EXPECT_TRUE(connection.is_server()); | |
| 1019 EXPECT_EQ(1000u, connection.max_packet_length()); | |
| 1020 } | |
| 1021 | |
| 1022 TEST_P(QuicConnectionTest, ServerMaxPacketSize) { | |
| 1023 ValueRestore<bool> old_flag(&FLAGS_quic_small_default_packet_size, false); | |
| 1024 QuicConnectionId connection_id = 42; | |
| 1025 bool kIsServer = true; | |
| 1026 TestConnection connection(connection_id, IPEndPoint(), helper_.get(), | |
| 1027 factory_, kIsServer, version()); | |
| 1028 EXPECT_TRUE(connection.is_server()); | |
| 1029 EXPECT_EQ(1350u, connection.max_packet_length()); | |
| 1030 } | |
| 1031 | |
| 1032 TEST_P(QuicConnectionTest, IncreaseServerMaxPacketSize) { | |
| 1033 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 1034 | |
| 1035 connection_.set_is_server(true); | |
| 1036 connection_.set_max_packet_length(1000); | |
| 1037 | |
| 1038 QuicPacketHeader header; | |
| 1039 header.public_header.connection_id = connection_id_; | |
| 1040 header.public_header.reset_flag = false; | |
| 1041 header.public_header.version_flag = true; | |
| 1042 header.entropy_flag = false; | |
| 1043 header.fec_flag = false; | |
| 1044 header.packet_sequence_number = 1; | |
| 1045 header.fec_group = 0; | |
| 1046 | |
| 1047 QuicFrames frames; | |
| 1048 QuicPaddingFrame padding; | |
| 1049 frames.push_back(QuicFrame(&frame1_)); | |
| 1050 frames.push_back(QuicFrame(&padding)); | |
| 1051 | |
| 1052 scoped_ptr<QuicPacket> packet( | |
| 1053 BuildUnsizedDataPacket(&framer_, header, frames)); | |
| 1054 scoped_ptr<QuicEncryptedPacket> encrypted( | |
| 1055 framer_.EncryptPacket(ENCRYPTION_NONE, 12, *packet)); | |
| 1056 EXPECT_EQ(kMaxPacketSize, encrypted->length()); | |
| 1057 | |
| 1058 framer_.set_version(version()); | |
| 1059 EXPECT_CALL(visitor_, OnStreamFrames(_)).Times(1); | |
| 1060 connection_.ProcessUdpPacket(IPEndPoint(), IPEndPoint(), *encrypted); | |
| 1061 | |
| 1062 EXPECT_EQ(kMaxPacketSize, connection_.max_packet_length()); | |
| 1063 } | |
| 1064 | |
| 1065 TEST_P(QuicConnectionTest, PacketsInOrder) { | |
| 1066 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 1067 | |
| 1068 ProcessPacket(1); | |
| 1069 EXPECT_EQ(1u, outgoing_ack()->largest_observed); | |
| 1070 EXPECT_EQ(0u, outgoing_ack()->missing_packets.size()); | |
| 1071 | |
| 1072 ProcessPacket(2); | |
| 1073 EXPECT_EQ(2u, outgoing_ack()->largest_observed); | |
| 1074 EXPECT_EQ(0u, outgoing_ack()->missing_packets.size()); | |
| 1075 | |
| 1076 ProcessPacket(3); | |
| 1077 EXPECT_EQ(3u, outgoing_ack()->largest_observed); | |
| 1078 EXPECT_EQ(0u, outgoing_ack()->missing_packets.size()); | |
| 1079 } | |
| 1080 | |
| 1081 TEST_P(QuicConnectionTest, PacketsOutOfOrder) { | |
| 1082 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 1083 | |
| 1084 ProcessPacket(3); | |
| 1085 EXPECT_EQ(3u, outgoing_ack()->largest_observed); | |
| 1086 EXPECT_TRUE(IsMissing(2)); | |
| 1087 EXPECT_TRUE(IsMissing(1)); | |
| 1088 | |
| 1089 ProcessPacket(2); | |
| 1090 EXPECT_EQ(3u, outgoing_ack()->largest_observed); | |
| 1091 EXPECT_FALSE(IsMissing(2)); | |
| 1092 EXPECT_TRUE(IsMissing(1)); | |
| 1093 | |
| 1094 ProcessPacket(1); | |
| 1095 EXPECT_EQ(3u, outgoing_ack()->largest_observed); | |
| 1096 EXPECT_FALSE(IsMissing(2)); | |
| 1097 EXPECT_FALSE(IsMissing(1)); | |
| 1098 } | |
| 1099 | |
| 1100 TEST_P(QuicConnectionTest, DuplicatePacket) { | |
| 1101 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 1102 | |
| 1103 ProcessPacket(3); | |
| 1104 EXPECT_EQ(3u, outgoing_ack()->largest_observed); | |
| 1105 EXPECT_TRUE(IsMissing(2)); | |
| 1106 EXPECT_TRUE(IsMissing(1)); | |
| 1107 | |
| 1108 // Send packet 3 again, but do not set the expectation that | |
| 1109 // the visitor OnStreamFrames() will be called. | |
| 1110 ProcessDataPacket(3, 0, !kEntropyFlag); | |
| 1111 EXPECT_EQ(3u, outgoing_ack()->largest_observed); | |
| 1112 EXPECT_TRUE(IsMissing(2)); | |
| 1113 EXPECT_TRUE(IsMissing(1)); | |
| 1114 } | |
| 1115 | |
| 1116 TEST_P(QuicConnectionTest, PacketsOutOfOrderWithAdditionsAndLeastAwaiting) { | |
| 1117 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 1118 | |
| 1119 ProcessPacket(3); | |
| 1120 EXPECT_EQ(3u, outgoing_ack()->largest_observed); | |
| 1121 EXPECT_TRUE(IsMissing(2)); | |
| 1122 EXPECT_TRUE(IsMissing(1)); | |
| 1123 | |
| 1124 ProcessPacket(2); | |
| 1125 EXPECT_EQ(3u, outgoing_ack()->largest_observed); | |
| 1126 EXPECT_TRUE(IsMissing(1)); | |
| 1127 | |
| 1128 ProcessPacket(5); | |
| 1129 EXPECT_EQ(5u, outgoing_ack()->largest_observed); | |
| 1130 EXPECT_TRUE(IsMissing(1)); | |
| 1131 EXPECT_TRUE(IsMissing(4)); | |
| 1132 | |
| 1133 // Pretend at this point the client has gotten acks for 2 and 3 and 1 is a | |
| 1134 // packet the peer will not retransmit. It indicates this by sending 'least | |
| 1135 // awaiting' is 4. The connection should then realize 1 will not be | |
| 1136 // retransmitted, and will remove it from the missing list. | |
| 1137 peer_creator_.set_sequence_number(5); | |
| 1138 QuicAckFrame frame = InitAckFrame(1); | |
| 1139 EXPECT_CALL(*send_algorithm_, OnCongestionEvent(_, _, _, _)); | |
| 1140 ProcessAckPacket(&frame); | |
| 1141 | |
| 1142 // Force an ack to be sent. | |
| 1143 SendAckPacketToPeer(); | |
| 1144 EXPECT_TRUE(IsMissing(4)); | |
| 1145 } | |
| 1146 | |
| 1147 TEST_P(QuicConnectionTest, RejectPacketTooFarOut) { | |
| 1148 EXPECT_CALL(visitor_, | |
| 1149 OnConnectionClosed(QUIC_INVALID_PACKET_HEADER, false)); | |
| 1150 // Call ProcessDataPacket rather than ProcessPacket, as we should not get a | |
| 1151 // packet call to the visitor. | |
| 1152 ProcessDataPacket(6000, 0, !kEntropyFlag); | |
| 1153 EXPECT_FALSE(QuicConnectionPeer::GetConnectionClosePacket(&connection_) == | |
| 1154 nullptr); | |
| 1155 } | |
| 1156 | |
| 1157 TEST_P(QuicConnectionTest, RejectUnencryptedStreamData) { | |
| 1158 // Process an unencrypted packet from the non-crypto stream. | |
| 1159 frame1_.stream_id = 3; | |
| 1160 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 1161 EXPECT_CALL(visitor_, OnConnectionClosed(QUIC_UNENCRYPTED_STREAM_DATA, | |
| 1162 false)); | |
| 1163 ProcessDataPacket(1, 0, !kEntropyFlag); | |
| 1164 EXPECT_FALSE(QuicConnectionPeer::GetConnectionClosePacket(&connection_) == | |
| 1165 nullptr); | |
| 1166 const vector<QuicConnectionCloseFrame>& connection_close_frames = | |
| 1167 writer_->connection_close_frames(); | |
| 1168 EXPECT_EQ(1u, connection_close_frames.size()); | |
| 1169 EXPECT_EQ(QUIC_UNENCRYPTED_STREAM_DATA, | |
| 1170 connection_close_frames[0].error_code); | |
| 1171 } | |
| 1172 | |
| 1173 TEST_P(QuicConnectionTest, TruncatedAck) { | |
| 1174 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 1175 QuicPacketSequenceNumber num_packets = 256 * 2 + 1; | |
| 1176 for (QuicPacketSequenceNumber i = 0; i < num_packets; ++i) { | |
| 1177 SendStreamDataToPeer(3, "foo", i * 3, !kFin, nullptr); | |
| 1178 } | |
| 1179 | |
| 1180 QuicAckFrame frame = InitAckFrame(num_packets); | |
| 1181 SequenceNumberSet lost_packets; | |
| 1182 // Create an ack with 256 nacks, none adjacent to one another. | |
| 1183 for (QuicPacketSequenceNumber i = 1; i <= 256; ++i) { | |
| 1184 NackPacket(i * 2, &frame); | |
| 1185 if (i < 256) { // Last packet is nacked, but not lost. | |
| 1186 lost_packets.insert(i * 2); | |
| 1187 } | |
| 1188 } | |
| 1189 EXPECT_CALL(*loss_algorithm_, DetectLostPackets(_, _, _, _)) | |
| 1190 .WillOnce(Return(lost_packets)); | |
| 1191 EXPECT_CALL(entropy_calculator_, EntropyHash(511)) | |
| 1192 .WillOnce(Return(static_cast<QuicPacketEntropyHash>(0))); | |
| 1193 EXPECT_CALL(*send_algorithm_, OnCongestionEvent(true, _, _, _)); | |
| 1194 ProcessAckPacket(&frame); | |
| 1195 | |
| 1196 // A truncated ack will not have the true largest observed. | |
| 1197 EXPECT_GT(num_packets, manager_->largest_observed()); | |
| 1198 | |
| 1199 AckPacket(192, &frame); | |
| 1200 | |
| 1201 // Removing one missing packet allows us to ack 192 and one more range, but | |
| 1202 // 192 has already been declared lost, so it doesn't register as an ack. | |
| 1203 EXPECT_CALL(*loss_algorithm_, DetectLostPackets(_, _, _, _)) | |
| 1204 .WillOnce(Return(SequenceNumberSet())); | |
| 1205 EXPECT_CALL(*send_algorithm_, OnCongestionEvent(true, _, _, _)); | |
| 1206 ProcessAckPacket(&frame); | |
| 1207 EXPECT_EQ(num_packets, manager_->largest_observed()); | |
| 1208 } | |
| 1209 | |
| 1210 TEST_P(QuicConnectionTest, AckReceiptCausesAckSendBadEntropy) { | |
| 1211 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 1212 | |
| 1213 ProcessPacket(1); | |
| 1214 // Delay sending, then queue up an ack. | |
| 1215 EXPECT_CALL(*send_algorithm_, | |
| 1216 TimeUntilSend(_, _, _)).WillOnce( | |
| 1217 testing::Return(QuicTime::Delta::FromMicroseconds(1))); | |
| 1218 QuicConnectionPeer::SendAck(&connection_); | |
| 1219 | |
| 1220 // Process an ack with a least unacked of the received ack. | |
| 1221 // This causes an ack to be sent when TimeUntilSend returns 0. | |
| 1222 EXPECT_CALL(*send_algorithm_, | |
| 1223 TimeUntilSend(_, _, _)).WillRepeatedly( | |
| 1224 testing::Return(QuicTime::Delta::Zero())); | |
| 1225 // Skip a packet and then record an ack. | |
| 1226 peer_creator_.set_sequence_number(2); | |
| 1227 QuicAckFrame frame = InitAckFrame(0); | |
| 1228 ProcessAckPacket(&frame); | |
| 1229 } | |
| 1230 | |
| 1231 TEST_P(QuicConnectionTest, OutOfOrderReceiptCausesAckSend) { | |
| 1232 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 1233 | |
| 1234 ProcessPacket(3); | |
| 1235 // Should ack immediately since we have missing packets. | |
| 1236 EXPECT_EQ(1u, writer_->packets_write_attempts()); | |
| 1237 | |
| 1238 ProcessPacket(2); | |
| 1239 // Should ack immediately since we have missing packets. | |
| 1240 EXPECT_EQ(2u, writer_->packets_write_attempts()); | |
| 1241 | |
| 1242 ProcessPacket(1); | |
| 1243 // Should ack immediately, since this fills the last hole. | |
| 1244 EXPECT_EQ(3u, writer_->packets_write_attempts()); | |
| 1245 | |
| 1246 ProcessPacket(4); | |
| 1247 // Should not cause an ack. | |
| 1248 EXPECT_EQ(3u, writer_->packets_write_attempts()); | |
| 1249 } | |
| 1250 | |
| 1251 TEST_P(QuicConnectionTest, AckReceiptCausesAckSend) { | |
| 1252 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 1253 | |
| 1254 QuicPacketSequenceNumber original; | |
| 1255 QuicByteCount packet_size; | |
| 1256 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)) | |
| 1257 .WillOnce(DoAll(SaveArg<2>(&original), SaveArg<3>(&packet_size), | |
| 1258 Return(true))); | |
| 1259 connection_.SendStreamDataWithString(3, "foo", 0, !kFin, nullptr); | |
| 1260 QuicAckFrame frame = InitAckFrame(original); | |
| 1261 NackPacket(original, &frame); | |
| 1262 // First nack triggers early retransmit. | |
| 1263 SequenceNumberSet lost_packets; | |
| 1264 lost_packets.insert(1); | |
| 1265 EXPECT_CALL(*loss_algorithm_, DetectLostPackets(_, _, _, _)) | |
| 1266 .WillOnce(Return(lost_packets)); | |
| 1267 EXPECT_CALL(*send_algorithm_, OnCongestionEvent(true, _, _, _)); | |
| 1268 QuicPacketSequenceNumber retransmission; | |
| 1269 EXPECT_CALL(*send_algorithm_, | |
| 1270 OnPacketSent(_, _, _, packet_size - kQuicVersionSize, _)) | |
| 1271 .WillOnce(DoAll(SaveArg<2>(&retransmission), Return(true))); | |
| 1272 | |
| 1273 ProcessAckPacket(&frame); | |
| 1274 | |
| 1275 QuicAckFrame frame2 = InitAckFrame(retransmission); | |
| 1276 NackPacket(original, &frame2); | |
| 1277 EXPECT_CALL(*send_algorithm_, OnCongestionEvent(true, _, _, _)); | |
| 1278 EXPECT_CALL(*loss_algorithm_, DetectLostPackets(_, _, _, _)) | |
| 1279 .WillOnce(Return(SequenceNumberSet())); | |
| 1280 ProcessAckPacket(&frame2); | |
| 1281 | |
| 1282 // Now if the peer sends an ack which still reports the retransmitted packet | |
| 1283 // as missing, that will bundle an ack with data after two acks in a row | |
| 1284 // indicate the high water mark needs to be raised. | |
| 1285 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, | |
| 1286 HAS_RETRANSMITTABLE_DATA)); | |
| 1287 connection_.SendStreamDataWithString(3, "foo", 3, !kFin, nullptr); | |
| 1288 // No ack sent. | |
| 1289 EXPECT_EQ(1u, writer_->frame_count()); | |
| 1290 EXPECT_EQ(1u, writer_->stream_frames().size()); | |
| 1291 | |
| 1292 // No more packet loss for the rest of the test. | |
| 1293 EXPECT_CALL(*loss_algorithm_, DetectLostPackets(_, _, _, _)) | |
| 1294 .WillRepeatedly(Return(SequenceNumberSet())); | |
| 1295 ProcessAckPacket(&frame2); | |
| 1296 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, | |
| 1297 HAS_RETRANSMITTABLE_DATA)); | |
| 1298 connection_.SendStreamDataWithString(3, "foo", 3, !kFin, nullptr); | |
| 1299 // Ack bundled. | |
| 1300 EXPECT_EQ(3u, writer_->frame_count()); | |
| 1301 EXPECT_EQ(1u, writer_->stream_frames().size()); | |
| 1302 EXPECT_FALSE(writer_->ack_frames().empty()); | |
| 1303 | |
| 1304 // But an ack with no missing packets will not send an ack. | |
| 1305 AckPacket(original, &frame2); | |
| 1306 ProcessAckPacket(&frame2); | |
| 1307 ProcessAckPacket(&frame2); | |
| 1308 } | |
| 1309 | |
| 1310 TEST_P(QuicConnectionTest, 20AcksCausesAckSend) { | |
| 1311 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 1312 | |
| 1313 SendStreamDataToPeer(1, "foo", 0, !kFin, nullptr); | |
| 1314 | |
| 1315 QuicAlarm* ack_alarm = QuicConnectionPeer::GetAckAlarm(&connection_); | |
| 1316 // But an ack with no missing packets will not send an ack. | |
| 1317 QuicAckFrame frame = InitAckFrame(1); | |
| 1318 EXPECT_CALL(*send_algorithm_, OnCongestionEvent(true, _, _, _)); | |
| 1319 EXPECT_CALL(*loss_algorithm_, DetectLostPackets(_, _, _, _)) | |
| 1320 .WillRepeatedly(Return(SequenceNumberSet())); | |
| 1321 for (int i = 0; i < 20; ++i) { | |
| 1322 EXPECT_FALSE(ack_alarm->IsSet()); | |
| 1323 ProcessAckPacket(&frame); | |
| 1324 } | |
| 1325 EXPECT_TRUE(ack_alarm->IsSet()); | |
| 1326 } | |
| 1327 | |
| 1328 TEST_P(QuicConnectionTest, LeastUnackedLower) { | |
| 1329 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 1330 | |
| 1331 SendStreamDataToPeer(1, "foo", 0, !kFin, nullptr); | |
| 1332 SendStreamDataToPeer(1, "bar", 3, !kFin, nullptr); | |
| 1333 SendStreamDataToPeer(1, "eep", 6, !kFin, nullptr); | |
| 1334 | |
| 1335 // Start out saying the least unacked is 2. | |
| 1336 peer_creator_.set_sequence_number(5); | |
| 1337 QuicStopWaitingFrame frame = InitStopWaitingFrame(2); | |
| 1338 ProcessStopWaitingPacket(&frame); | |
| 1339 | |
| 1340 // Change it to 1, but lower the sequence number to fake out-of-order packets. | |
| 1341 // This should be fine. | |
| 1342 peer_creator_.set_sequence_number(1); | |
| 1343 // The scheduler will not process out of order acks, but all packet processing | |
| 1344 // causes the connection to try to write. | |
| 1345 EXPECT_CALL(visitor_, OnCanWrite()); | |
| 1346 QuicStopWaitingFrame frame2 = InitStopWaitingFrame(1); | |
| 1347 ProcessStopWaitingPacket(&frame2); | |
| 1348 | |
| 1349 // Now claim it's one, but set the ordering so it was sent "after" the first | |
| 1350 // one. This should cause a connection error. | |
| 1351 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)); | |
| 1352 peer_creator_.set_sequence_number(7); | |
| 1353 EXPECT_CALL(visitor_, | |
| 1354 OnConnectionClosed(QUIC_INVALID_STOP_WAITING_DATA, false)); | |
| 1355 QuicStopWaitingFrame frame3 = InitStopWaitingFrame(1); | |
| 1356 ProcessStopWaitingPacket(&frame3); | |
| 1357 } | |
| 1358 | |
| 1359 TEST_P(QuicConnectionTest, TooManySentPackets) { | |
| 1360 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 1361 | |
| 1362 for (int i = 0; i < 1100; ++i) { | |
| 1363 SendStreamDataToPeer(1, "foo", 3 * i, !kFin, nullptr); | |
| 1364 } | |
| 1365 | |
| 1366 // Ack packet 1, which leaves more than the limit outstanding. | |
| 1367 EXPECT_CALL(*send_algorithm_, OnCongestionEvent(true, _, _, _)); | |
| 1368 EXPECT_CALL(visitor_, OnConnectionClosed( | |
| 1369 QUIC_TOO_MANY_OUTSTANDING_SENT_PACKETS, false)); | |
| 1370 // We're receive buffer limited, so the connection won't try to write more. | |
| 1371 EXPECT_CALL(visitor_, OnCanWrite()).Times(0); | |
| 1372 | |
| 1373 // Nack every packet except the last one, leaving a huge gap. | |
| 1374 QuicAckFrame frame1 = InitAckFrame(1100); | |
| 1375 for (QuicPacketSequenceNumber i = 1; i < 1100; ++i) { | |
| 1376 NackPacket(i, &frame1); | |
| 1377 } | |
| 1378 ProcessAckPacket(&frame1); | |
| 1379 } | |
| 1380 | |
| 1381 TEST_P(QuicConnectionTest, TooManyReceivedPackets) { | |
| 1382 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 1383 EXPECT_CALL(visitor_, OnConnectionClosed( | |
| 1384 QUIC_TOO_MANY_OUTSTANDING_RECEIVED_PACKETS, false)); | |
| 1385 | |
| 1386 // Miss every other packet for 1000 packets. | |
| 1387 for (QuicPacketSequenceNumber i = 1; i < 1000; ++i) { | |
| 1388 ProcessPacket(i * 2); | |
| 1389 if (!connection_.connected()) { | |
| 1390 break; | |
| 1391 } | |
| 1392 } | |
| 1393 } | |
| 1394 | |
| 1395 TEST_P(QuicConnectionTest, LargestObservedLower) { | |
| 1396 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 1397 | |
| 1398 SendStreamDataToPeer(1, "foo", 0, !kFin, nullptr); | |
| 1399 SendStreamDataToPeer(1, "bar", 3, !kFin, nullptr); | |
| 1400 SendStreamDataToPeer(1, "eep", 6, !kFin, nullptr); | |
| 1401 EXPECT_CALL(*send_algorithm_, OnCongestionEvent(true, _, _, _)); | |
| 1402 | |
| 1403 // Start out saying the largest observed is 2. | |
| 1404 QuicAckFrame frame1 = InitAckFrame(1); | |
| 1405 QuicAckFrame frame2 = InitAckFrame(2); | |
| 1406 ProcessAckPacket(&frame2); | |
| 1407 | |
| 1408 // Now change it to 1, and it should cause a connection error. | |
| 1409 EXPECT_CALL(visitor_, OnConnectionClosed(QUIC_INVALID_ACK_DATA, false)); | |
| 1410 EXPECT_CALL(visitor_, OnCanWrite()).Times(0); | |
| 1411 ProcessAckPacket(&frame1); | |
| 1412 } | |
| 1413 | |
| 1414 TEST_P(QuicConnectionTest, AckUnsentData) { | |
| 1415 // Ack a packet which has not been sent. | |
| 1416 EXPECT_CALL(visitor_, OnConnectionClosed(QUIC_INVALID_ACK_DATA, false)); | |
| 1417 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 1418 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)); | |
| 1419 QuicAckFrame frame(MakeAckFrame(1)); | |
| 1420 EXPECT_CALL(visitor_, OnCanWrite()).Times(0); | |
| 1421 ProcessAckPacket(&frame); | |
| 1422 } | |
| 1423 | |
| 1424 TEST_P(QuicConnectionTest, AckAll) { | |
| 1425 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 1426 ProcessPacket(1); | |
| 1427 | |
| 1428 peer_creator_.set_sequence_number(1); | |
| 1429 QuicAckFrame frame1 = InitAckFrame(0); | |
| 1430 ProcessAckPacket(&frame1); | |
| 1431 } | |
| 1432 | |
| 1433 TEST_P(QuicConnectionTest, SendingDifferentSequenceNumberLengthsBandwidth) { | |
| 1434 QuicPacketSequenceNumber last_packet; | |
| 1435 SendStreamDataToPeer(1, "foo", 0, !kFin, &last_packet); | |
| 1436 EXPECT_EQ(1u, last_packet); | |
| 1437 EXPECT_EQ(PACKET_1BYTE_SEQUENCE_NUMBER, | |
| 1438 creator_->next_sequence_number_length()); | |
| 1439 EXPECT_EQ(PACKET_1BYTE_SEQUENCE_NUMBER, | |
| 1440 writer_->header().public_header.sequence_number_length); | |
| 1441 | |
| 1442 EXPECT_CALL(*send_algorithm_, GetCongestionWindow()).WillRepeatedly( | |
| 1443 Return(kMaxPacketSize * 256)); | |
| 1444 | |
| 1445 SendStreamDataToPeer(1, "bar", 3, !kFin, &last_packet); | |
| 1446 EXPECT_EQ(2u, last_packet); | |
| 1447 EXPECT_EQ(PACKET_2BYTE_SEQUENCE_NUMBER, | |
| 1448 creator_->next_sequence_number_length()); | |
| 1449 // The 1 packet lag is due to the sequence number length being recalculated in | |
| 1450 // QuicConnection after a packet is sent. | |
| 1451 EXPECT_EQ(PACKET_1BYTE_SEQUENCE_NUMBER, | |
| 1452 writer_->header().public_header.sequence_number_length); | |
| 1453 | |
| 1454 EXPECT_CALL(*send_algorithm_, GetCongestionWindow()).WillRepeatedly( | |
| 1455 Return(kMaxPacketSize * 256 * 256)); | |
| 1456 | |
| 1457 SendStreamDataToPeer(1, "foo", 6, !kFin, &last_packet); | |
| 1458 EXPECT_EQ(3u, last_packet); | |
| 1459 EXPECT_EQ(PACKET_4BYTE_SEQUENCE_NUMBER, | |
| 1460 creator_->next_sequence_number_length()); | |
| 1461 EXPECT_EQ(PACKET_2BYTE_SEQUENCE_NUMBER, | |
| 1462 writer_->header().public_header.sequence_number_length); | |
| 1463 | |
| 1464 EXPECT_CALL(*send_algorithm_, GetCongestionWindow()).WillRepeatedly( | |
| 1465 Return(kMaxPacketSize * 256 * 256 * 256)); | |
| 1466 | |
| 1467 SendStreamDataToPeer(1, "bar", 9, !kFin, &last_packet); | |
| 1468 EXPECT_EQ(4u, last_packet); | |
| 1469 EXPECT_EQ(PACKET_4BYTE_SEQUENCE_NUMBER, | |
| 1470 creator_->next_sequence_number_length()); | |
| 1471 EXPECT_EQ(PACKET_4BYTE_SEQUENCE_NUMBER, | |
| 1472 writer_->header().public_header.sequence_number_length); | |
| 1473 | |
| 1474 EXPECT_CALL(*send_algorithm_, GetCongestionWindow()).WillRepeatedly( | |
| 1475 Return(kMaxPacketSize * 256 * 256 * 256 * 256)); | |
| 1476 | |
| 1477 SendStreamDataToPeer(1, "foo", 12, !kFin, &last_packet); | |
| 1478 EXPECT_EQ(5u, last_packet); | |
| 1479 EXPECT_EQ(PACKET_6BYTE_SEQUENCE_NUMBER, | |
| 1480 creator_->next_sequence_number_length()); | |
| 1481 EXPECT_EQ(PACKET_4BYTE_SEQUENCE_NUMBER, | |
| 1482 writer_->header().public_header.sequence_number_length); | |
| 1483 } | |
| 1484 | |
| 1485 // TODO(ianswett): Re-enable this test by finding a good way to test different | |
| 1486 // sequence number lengths without sending packets with giant gaps. | |
| 1487 TEST_P(QuicConnectionTest, | |
| 1488 DISABLED_SendingDifferentSequenceNumberLengthsUnackedDelta) { | |
| 1489 QuicPacketSequenceNumber last_packet; | |
| 1490 SendStreamDataToPeer(1, "foo", 0, !kFin, &last_packet); | |
| 1491 EXPECT_EQ(1u, last_packet); | |
| 1492 EXPECT_EQ(PACKET_1BYTE_SEQUENCE_NUMBER, | |
| 1493 creator_->next_sequence_number_length()); | |
| 1494 EXPECT_EQ(PACKET_1BYTE_SEQUENCE_NUMBER, | |
| 1495 writer_->header().public_header.sequence_number_length); | |
| 1496 | |
| 1497 creator_->set_sequence_number(100); | |
| 1498 | |
| 1499 SendStreamDataToPeer(1, "bar", 3, !kFin, &last_packet); | |
| 1500 EXPECT_EQ(PACKET_2BYTE_SEQUENCE_NUMBER, | |
| 1501 creator_->next_sequence_number_length()); | |
| 1502 EXPECT_EQ(PACKET_1BYTE_SEQUENCE_NUMBER, | |
| 1503 writer_->header().public_header.sequence_number_length); | |
| 1504 | |
| 1505 creator_->set_sequence_number(100 * 256); | |
| 1506 | |
| 1507 SendStreamDataToPeer(1, "foo", 6, !kFin, &last_packet); | |
| 1508 EXPECT_EQ(PACKET_4BYTE_SEQUENCE_NUMBER, | |
| 1509 creator_->next_sequence_number_length()); | |
| 1510 EXPECT_EQ(PACKET_2BYTE_SEQUENCE_NUMBER, | |
| 1511 writer_->header().public_header.sequence_number_length); | |
| 1512 | |
| 1513 creator_->set_sequence_number(100 * 256 * 256); | |
| 1514 | |
| 1515 SendStreamDataToPeer(1, "bar", 9, !kFin, &last_packet); | |
| 1516 EXPECT_EQ(PACKET_4BYTE_SEQUENCE_NUMBER, | |
| 1517 creator_->next_sequence_number_length()); | |
| 1518 EXPECT_EQ(PACKET_4BYTE_SEQUENCE_NUMBER, | |
| 1519 writer_->header().public_header.sequence_number_length); | |
| 1520 | |
| 1521 creator_->set_sequence_number(100 * 256 * 256 * 256); | |
| 1522 | |
| 1523 SendStreamDataToPeer(1, "foo", 12, !kFin, &last_packet); | |
| 1524 EXPECT_EQ(PACKET_6BYTE_SEQUENCE_NUMBER, | |
| 1525 creator_->next_sequence_number_length()); | |
| 1526 EXPECT_EQ(PACKET_4BYTE_SEQUENCE_NUMBER, | |
| 1527 writer_->header().public_header.sequence_number_length); | |
| 1528 } | |
| 1529 | |
| 1530 TEST_P(QuicConnectionTest, BasicSending) { | |
| 1531 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 1532 QuicPacketSequenceNumber last_packet; | |
| 1533 SendStreamDataToPeer(1, "foo", 0, !kFin, &last_packet); // Packet 1 | |
| 1534 EXPECT_EQ(1u, last_packet); | |
| 1535 SendAckPacketToPeer(); // Packet 2 | |
| 1536 | |
| 1537 EXPECT_EQ(1u, least_unacked()); | |
| 1538 | |
| 1539 SendAckPacketToPeer(); // Packet 3 | |
| 1540 EXPECT_EQ(1u, least_unacked()); | |
| 1541 | |
| 1542 SendStreamDataToPeer(1, "bar", 3, !kFin, &last_packet); // Packet 4 | |
| 1543 EXPECT_EQ(4u, last_packet); | |
| 1544 SendAckPacketToPeer(); // Packet 5 | |
| 1545 EXPECT_EQ(1u, least_unacked()); | |
| 1546 | |
| 1547 EXPECT_CALL(*send_algorithm_, OnCongestionEvent(true, _, _, _)); | |
| 1548 | |
| 1549 // Peer acks up to packet 3. | |
| 1550 QuicAckFrame frame = InitAckFrame(3); | |
| 1551 ProcessAckPacket(&frame); | |
| 1552 SendAckPacketToPeer(); // Packet 6 | |
| 1553 | |
| 1554 // As soon as we've acked one, we skip ack packets 2 and 3 and note lack of | |
| 1555 // ack for 4. | |
| 1556 EXPECT_EQ(4u, least_unacked()); | |
| 1557 | |
| 1558 EXPECT_CALL(*send_algorithm_, OnCongestionEvent(true, _, _, _)); | |
| 1559 | |
| 1560 // Peer acks up to packet 4, the last packet. | |
| 1561 QuicAckFrame frame2 = InitAckFrame(6); | |
| 1562 ProcessAckPacket(&frame2); // Acks don't instigate acks. | |
| 1563 | |
| 1564 // Verify that we did not send an ack. | |
| 1565 EXPECT_EQ(6u, writer_->header().packet_sequence_number); | |
| 1566 | |
| 1567 // So the last ack has not changed. | |
| 1568 EXPECT_EQ(4u, least_unacked()); | |
| 1569 | |
| 1570 // If we force an ack, we shouldn't change our retransmit state. | |
| 1571 SendAckPacketToPeer(); // Packet 7 | |
| 1572 EXPECT_EQ(7u, least_unacked()); | |
| 1573 | |
| 1574 // But if we send more data it should. | |
| 1575 SendStreamDataToPeer(1, "eep", 6, !kFin, &last_packet); // Packet 8 | |
| 1576 EXPECT_EQ(8u, last_packet); | |
| 1577 SendAckPacketToPeer(); // Packet 9 | |
| 1578 EXPECT_EQ(7u, least_unacked()); | |
| 1579 } | |
| 1580 | |
| 1581 // If FLAGS_quic_record_send_time_before_write is disabled, QuicConnection | |
| 1582 // should record the packet sen-tdime after the packet is sent. | |
| 1583 TEST_P(QuicConnectionTest, RecordSentTimeAfterPacketSent) { | |
| 1584 ValueRestore<bool> old_flag(&FLAGS_quic_record_send_time_before_write, false); | |
| 1585 // We're using a MockClock for the tests, so we have complete control over the | |
| 1586 // time. | |
| 1587 // Our recorded timestamp for the last packet sent time will be passed in to | |
| 1588 // the send_algorithm. Make sure that it is set to the correct value. | |
| 1589 QuicTime actual_recorded_send_time = QuicTime::Zero(); | |
| 1590 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)) | |
| 1591 .WillOnce(DoAll(SaveArg<0>(&actual_recorded_send_time), Return(true))); | |
| 1592 | |
| 1593 // First send without any pause and check the result. | |
| 1594 QuicTime expected_recorded_send_time = clock_.Now(); | |
| 1595 connection_.SendStreamDataWithString(1, "foo", 0, !kFin, nullptr); | |
| 1596 EXPECT_EQ(expected_recorded_send_time, actual_recorded_send_time) | |
| 1597 << "Expected time = " << expected_recorded_send_time.ToDebuggingValue() | |
| 1598 << ". Actual time = " << actual_recorded_send_time.ToDebuggingValue(); | |
| 1599 | |
| 1600 // Now pause during the write, and check the results. | |
| 1601 actual_recorded_send_time = QuicTime::Zero(); | |
| 1602 const QuicTime::Delta kWritePauseTimeDelta = | |
| 1603 QuicTime::Delta::FromMilliseconds(5000); | |
| 1604 SetWritePauseTimeDelta(kWritePauseTimeDelta); | |
| 1605 expected_recorded_send_time = clock_.Now().Add(kWritePauseTimeDelta); | |
| 1606 | |
| 1607 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)) | |
| 1608 .WillOnce(DoAll(SaveArg<0>(&actual_recorded_send_time), Return(true))); | |
| 1609 connection_.SendStreamDataWithString(2, "baz", 0, !kFin, nullptr); | |
| 1610 EXPECT_EQ(expected_recorded_send_time, actual_recorded_send_time) | |
| 1611 << "Expected time = " << expected_recorded_send_time.ToDebuggingValue() | |
| 1612 << ". Actual time = " << actual_recorded_send_time.ToDebuggingValue(); | |
| 1613 } | |
| 1614 | |
| 1615 // If FLAGS_quic_record_send_time_before_write is enabled, QuicConnection should | |
| 1616 // record the the packet sent-time prior to sending the packet. | |
| 1617 TEST_P(QuicConnectionTest, RecordSentTimeBeforePacketSent) { | |
| 1618 ValueRestore<bool> old_flag(&FLAGS_quic_record_send_time_before_write, true); | |
| 1619 // We're using a MockClock for the tests, so we have complete control over the | |
| 1620 // time. | |
| 1621 // Our recorded timestamp for the last packet sent time will be passed in to | |
| 1622 // the send_algorithm. Make sure that it is set to the correct value. | |
| 1623 QuicTime actual_recorded_send_time = QuicTime::Zero(); | |
| 1624 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)) | |
| 1625 .WillOnce(DoAll(SaveArg<0>(&actual_recorded_send_time), Return(true))); | |
| 1626 | |
| 1627 // First send without any pause and check the result. | |
| 1628 QuicTime expected_recorded_send_time = clock_.Now(); | |
| 1629 connection_.SendStreamDataWithString(1, "foo", 0, !kFin, nullptr); | |
| 1630 EXPECT_EQ(expected_recorded_send_time, actual_recorded_send_time) | |
| 1631 << "Expected time = " << expected_recorded_send_time.ToDebuggingValue() | |
| 1632 << ". Actual time = " << actual_recorded_send_time.ToDebuggingValue(); | |
| 1633 | |
| 1634 // Now pause during the write, and check the results. | |
| 1635 actual_recorded_send_time = QuicTime::Zero(); | |
| 1636 const QuicTime::Delta kWritePauseTimeDelta = | |
| 1637 QuicTime::Delta::FromMilliseconds(5000); | |
| 1638 SetWritePauseTimeDelta(kWritePauseTimeDelta); | |
| 1639 expected_recorded_send_time = clock_.Now(); | |
| 1640 | |
| 1641 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)) | |
| 1642 .WillOnce(DoAll(SaveArg<0>(&actual_recorded_send_time), Return(true))); | |
| 1643 connection_.SendStreamDataWithString(2, "baz", 0, !kFin, nullptr); | |
| 1644 EXPECT_EQ(expected_recorded_send_time, actual_recorded_send_time) | |
| 1645 << "Expected time = " << expected_recorded_send_time.ToDebuggingValue() | |
| 1646 << ". Actual time = " << actual_recorded_send_time.ToDebuggingValue(); | |
| 1647 } | |
| 1648 | |
| 1649 TEST_P(QuicConnectionTest, FECSending) { | |
| 1650 // All packets carry version info till version is negotiated. | |
| 1651 size_t payload_length; | |
| 1652 // GetPacketLengthForOneStream() assumes a stream offset of 0 in determining | |
| 1653 // packet length. The size of the offset field in a stream frame is 0 for | |
| 1654 // offset 0, and 2 for non-zero offsets up through 64K. Increase | |
| 1655 // max_packet_length by 2 so that subsequent packets containing subsequent | |
| 1656 // stream frames with non-zero offets will fit within the packet length. | |
| 1657 size_t length = 2 + GetPacketLengthForOneStream( | |
| 1658 connection_.version(), kIncludeVersion, | |
| 1659 PACKET_8BYTE_CONNECTION_ID, PACKET_1BYTE_SEQUENCE_NUMBER, | |
| 1660 IN_FEC_GROUP, &payload_length); | |
| 1661 creator_->set_max_packet_length(length); | |
| 1662 | |
| 1663 // Send 4 protected data packets, which should also trigger 1 FEC packet. | |
| 1664 EXPECT_CALL(*send_algorithm_, | |
| 1665 OnPacketSent(_, _, _, _, HAS_RETRANSMITTABLE_DATA)).Times(5); | |
| 1666 // The first stream frame will have 2 fewer overhead bytes than the other 3. | |
| 1667 const string payload(payload_length * 4 + 2, 'a'); | |
| 1668 connection_.SendStreamDataWithStringWithFec(1, payload, 0, !kFin, nullptr); | |
| 1669 // Expect the FEC group to be closed after SendStreamDataWithString. | |
| 1670 EXPECT_FALSE(creator_->IsFecGroupOpen()); | |
| 1671 EXPECT_FALSE(creator_->IsFecProtected()); | |
| 1672 } | |
| 1673 | |
| 1674 TEST_P(QuicConnectionTest, FECQueueing) { | |
| 1675 // All packets carry version info till version is negotiated. | |
| 1676 size_t payload_length; | |
| 1677 size_t length = GetPacketLengthForOneStream( | |
| 1678 connection_.version(), kIncludeVersion, | |
| 1679 PACKET_8BYTE_CONNECTION_ID, PACKET_1BYTE_SEQUENCE_NUMBER, | |
| 1680 IN_FEC_GROUP, &payload_length); | |
| 1681 creator_->set_max_packet_length(length); | |
| 1682 EXPECT_TRUE(creator_->IsFecEnabled()); | |
| 1683 | |
| 1684 EXPECT_EQ(0u, connection_.NumQueuedPackets()); | |
| 1685 BlockOnNextWrite(); | |
| 1686 const string payload(payload_length, 'a'); | |
| 1687 connection_.SendStreamDataWithStringWithFec(1, payload, 0, !kFin, nullptr); | |
| 1688 EXPECT_FALSE(creator_->IsFecGroupOpen()); | |
| 1689 EXPECT_FALSE(creator_->IsFecProtected()); | |
| 1690 // Expect the first data packet and the fec packet to be queued. | |
| 1691 EXPECT_EQ(2u, connection_.NumQueuedPackets()); | |
| 1692 } | |
| 1693 | |
| 1694 TEST_P(QuicConnectionTest, FECAlarmStoppedWhenFECPacketSent) { | |
| 1695 EXPECT_TRUE(creator_->IsFecEnabled()); | |
| 1696 EXPECT_EQ(0u, QuicSentPacketManagerPeer::GetBytesInFlight(manager_)); | |
| 1697 EXPECT_FALSE(connection_.GetFecAlarm()->IsSet()); | |
| 1698 | |
| 1699 creator_->set_max_packets_per_fec_group(2); | |
| 1700 | |
| 1701 // 1 Data packet. FEC alarm should be set. | |
| 1702 EXPECT_CALL(*send_algorithm_, | |
| 1703 OnPacketSent(_, _, 1u, _, HAS_RETRANSMITTABLE_DATA)).Times(1); | |
| 1704 connection_.SendStreamDataWithStringWithFec(3, "foo", 0, true, nullptr); | |
| 1705 EXPECT_TRUE(connection_.GetFecAlarm()->IsSet()); | |
| 1706 | |
| 1707 // Second data packet triggers FEC packet out. FEC alarm should not be set. | |
| 1708 EXPECT_CALL(*send_algorithm_, | |
| 1709 OnPacketSent(_, _, _, _, HAS_RETRANSMITTABLE_DATA)).Times(2); | |
| 1710 connection_.SendStreamDataWithStringWithFec(5, "foo", 0, true, nullptr); | |
| 1711 EXPECT_TRUE(writer_->header().fec_flag); | |
| 1712 EXPECT_FALSE(connection_.GetFecAlarm()->IsSet()); | |
| 1713 } | |
| 1714 | |
| 1715 TEST_P(QuicConnectionTest, FECAlarmStoppedOnConnectionClose) { | |
| 1716 EXPECT_TRUE(creator_->IsFecEnabled()); | |
| 1717 EXPECT_FALSE(connection_.GetFecAlarm()->IsSet()); | |
| 1718 creator_->set_max_packets_per_fec_group(100); | |
| 1719 | |
| 1720 // 1 Data packet. FEC alarm should be set. | |
| 1721 EXPECT_CALL(*send_algorithm_, | |
| 1722 OnPacketSent(_, _, 1u, _, HAS_RETRANSMITTABLE_DATA)).Times(1); | |
| 1723 connection_.SendStreamDataWithStringWithFec(3, "foo", 0, kFin, nullptr); | |
| 1724 EXPECT_TRUE(connection_.GetFecAlarm()->IsSet()); | |
| 1725 | |
| 1726 EXPECT_CALL(visitor_, OnConnectionClosed(QUIC_NO_ERROR, false)); | |
| 1727 // Closing connection should stop the FEC alarm. | |
| 1728 connection_.CloseConnection(QUIC_NO_ERROR, /*from_peer=*/false); | |
| 1729 EXPECT_FALSE(connection_.GetFecAlarm()->IsSet()); | |
| 1730 } | |
| 1731 | |
| 1732 TEST_P(QuicConnectionTest, RemoveFECFromInflightOnRetransmissionTimeout) { | |
| 1733 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 1734 EXPECT_TRUE(creator_->IsFecEnabled()); | |
| 1735 EXPECT_EQ(0u, QuicSentPacketManagerPeer::GetBytesInFlight(manager_)); | |
| 1736 EXPECT_FALSE(connection_.GetFecAlarm()->IsSet()); | |
| 1737 | |
| 1738 // 1 Data packet. FEC alarm should be set. | |
| 1739 EXPECT_CALL(*send_algorithm_, | |
| 1740 OnPacketSent(_, _, 1u, _, HAS_RETRANSMITTABLE_DATA)).Times(1); | |
| 1741 connection_.SendStreamDataWithStringWithFec(3, "foo", 0, !kFin, nullptr); | |
| 1742 EXPECT_TRUE(connection_.GetFecAlarm()->IsSet()); | |
| 1743 size_t protected_packet = | |
| 1744 QuicSentPacketManagerPeer::GetBytesInFlight(manager_); | |
| 1745 | |
| 1746 // Force FEC timeout to send FEC packet out. | |
| 1747 EXPECT_CALL(*send_algorithm_, | |
| 1748 OnPacketSent(_, _, 2u, _, HAS_RETRANSMITTABLE_DATA)).Times(1); | |
| 1749 connection_.GetFecAlarm()->Fire(); | |
| 1750 EXPECT_TRUE(writer_->header().fec_flag); | |
| 1751 | |
| 1752 size_t fec_packet = protected_packet; | |
| 1753 EXPECT_EQ(protected_packet + fec_packet, | |
| 1754 QuicSentPacketManagerPeer::GetBytesInFlight(manager_)); | |
| 1755 clock_.AdvanceTime(DefaultRetransmissionTime()); | |
| 1756 | |
| 1757 // On RTO, both data and FEC packets are removed from inflight, only the data | |
| 1758 // packet is retransmitted, and this retransmission (but not FEC) gets added | |
| 1759 // back into the inflight. | |
| 1760 EXPECT_CALL(*send_algorithm_, OnRetransmissionTimeout(true)); | |
| 1761 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)).Times(1); | |
| 1762 connection_.GetRetransmissionAlarm()->Fire(); | |
| 1763 | |
| 1764 // The retransmission of packet 1 will be 3 bytes smaller than packet 1, since | |
| 1765 // the first transmission will have 1 byte for FEC group number and 2 bytes of | |
| 1766 // stream frame size, which are absent in the retransmission. | |
| 1767 size_t retransmitted_packet = protected_packet - 3; | |
| 1768 if (FLAGS_quic_use_new_rto) { | |
| 1769 EXPECT_EQ(protected_packet + retransmitted_packet, | |
| 1770 QuicSentPacketManagerPeer::GetBytesInFlight(manager_)); | |
| 1771 } else { | |
| 1772 EXPECT_EQ(retransmitted_packet, | |
| 1773 QuicSentPacketManagerPeer::GetBytesInFlight(manager_)); | |
| 1774 } | |
| 1775 EXPECT_FALSE(connection_.GetFecAlarm()->IsSet()); | |
| 1776 | |
| 1777 // Receive ack for the retransmission. No data should be outstanding. | |
| 1778 QuicAckFrame ack = InitAckFrame(3); | |
| 1779 NackPacket(1, &ack); | |
| 1780 NackPacket(2, &ack); | |
| 1781 SequenceNumberSet lost_packets; | |
| 1782 if (FLAGS_quic_use_new_rto) { | |
| 1783 lost_packets.insert(1); | |
| 1784 } | |
| 1785 EXPECT_CALL(*loss_algorithm_, DetectLostPackets(_, _, _, _)) | |
| 1786 .WillOnce(Return(lost_packets)); | |
| 1787 EXPECT_CALL(*send_algorithm_, OnCongestionEvent(true, _, _, _)); | |
| 1788 ProcessAckPacket(&ack); | |
| 1789 | |
| 1790 // Ensure the alarm is not set since all packets have been acked or abandoned. | |
| 1791 EXPECT_FALSE(connection_.GetRetransmissionAlarm()->IsSet()); | |
| 1792 EXPECT_EQ(0u, QuicSentPacketManagerPeer::GetBytesInFlight(manager_)); | |
| 1793 } | |
| 1794 | |
| 1795 TEST_P(QuicConnectionTest, RemoveFECFromInflightOnLossRetransmission) { | |
| 1796 EXPECT_TRUE(creator_->IsFecEnabled()); | |
| 1797 EXPECT_FALSE(connection_.GetFecAlarm()->IsSet()); | |
| 1798 | |
| 1799 // 1 FEC-protected data packet. FEC alarm should be set. | |
| 1800 EXPECT_CALL(*send_algorithm_, | |
| 1801 OnPacketSent(_, _, _, _, HAS_RETRANSMITTABLE_DATA)).Times(1); | |
| 1802 connection_.SendStreamDataWithStringWithFec(3, "foo", 0, kFin, nullptr); | |
| 1803 EXPECT_TRUE(connection_.GetFecAlarm()->IsSet()); | |
| 1804 size_t protected_packet = | |
| 1805 QuicSentPacketManagerPeer::GetBytesInFlight(manager_); | |
| 1806 | |
| 1807 // Force FEC timeout to send FEC packet out. | |
| 1808 EXPECT_CALL(*send_algorithm_, | |
| 1809 OnPacketSent(_, _, _, _, HAS_RETRANSMITTABLE_DATA)).Times(1); | |
| 1810 connection_.GetFecAlarm()->Fire(); | |
| 1811 EXPECT_TRUE(writer_->header().fec_flag); | |
| 1812 size_t fec_packet = protected_packet; | |
| 1813 EXPECT_EQ(protected_packet + fec_packet, | |
| 1814 QuicSentPacketManagerPeer::GetBytesInFlight(manager_)); | |
| 1815 | |
| 1816 // Send more data to trigger NACKs. Note that all data starts at stream offset | |
| 1817 // 0 to ensure the same packet size, for ease of testing. | |
| 1818 EXPECT_CALL(*send_algorithm_, | |
| 1819 OnPacketSent(_, _, _, _, HAS_RETRANSMITTABLE_DATA)).Times(4); | |
| 1820 connection_.SendStreamDataWithString(5, "foo", 0, kFin, nullptr); | |
| 1821 connection_.SendStreamDataWithString(7, "foo", 0, kFin, nullptr); | |
| 1822 connection_.SendStreamDataWithString(9, "foo", 0, kFin, nullptr); | |
| 1823 connection_.SendStreamDataWithString(11, "foo", 0, kFin, nullptr); | |
| 1824 | |
| 1825 // An unprotected packet will be 3 bytes smaller than an FEC-protected packet, | |
| 1826 // since the protected packet will have 1 byte for FEC group number and | |
| 1827 // 2 bytes of stream frame size, which are absent in the unprotected packet. | |
| 1828 size_t unprotected_packet = protected_packet - 3; | |
| 1829 EXPECT_EQ(protected_packet + fec_packet + 4 * unprotected_packet, | |
| 1830 QuicSentPacketManagerPeer::GetBytesInFlight(manager_)); | |
| 1831 EXPECT_FALSE(connection_.GetFecAlarm()->IsSet()); | |
| 1832 | |
| 1833 // Ack data packets, and NACK FEC packet and one data packet. Triggers | |
| 1834 // NACK-based loss detection of both packets, but only data packet is | |
| 1835 // retransmitted and considered oustanding. | |
| 1836 QuicAckFrame ack = InitAckFrame(6); | |
| 1837 NackPacket(2, &ack); | |
| 1838 NackPacket(3, &ack); | |
| 1839 SequenceNumberSet lost_packets; | |
| 1840 lost_packets.insert(2); | |
| 1841 lost_packets.insert(3); | |
| 1842 EXPECT_CALL(*loss_algorithm_, DetectLostPackets(_, _, _, _)) | |
| 1843 .WillOnce(Return(lost_packets)); | |
| 1844 EXPECT_CALL(*send_algorithm_, OnCongestionEvent(true, _, _, _)); | |
| 1845 EXPECT_CALL(*send_algorithm_, | |
| 1846 OnPacketSent(_, _, _, _, HAS_RETRANSMITTABLE_DATA)).Times(1); | |
| 1847 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 1848 ProcessAckPacket(&ack); | |
| 1849 // On receiving this ack from the server, the client will no longer send | |
| 1850 // version number in subsequent packets, including in this retransmission. | |
| 1851 size_t unprotected_packet_no_version = unprotected_packet - 4; | |
| 1852 EXPECT_EQ(unprotected_packet_no_version, | |
| 1853 QuicSentPacketManagerPeer::GetBytesInFlight(manager_)); | |
| 1854 | |
| 1855 // Receive ack for the retransmission. No data should be outstanding. | |
| 1856 QuicAckFrame ack2 = InitAckFrame(7); | |
| 1857 NackPacket(2, &ack2); | |
| 1858 NackPacket(3, &ack2); | |
| 1859 SequenceNumberSet lost_packets2; | |
| 1860 EXPECT_CALL(*loss_algorithm_, DetectLostPackets(_, _, _, _)) | |
| 1861 .WillOnce(Return(lost_packets2)); | |
| 1862 EXPECT_CALL(*send_algorithm_, OnCongestionEvent(true, _, _, _)); | |
| 1863 ProcessAckPacket(&ack2); | |
| 1864 EXPECT_EQ(0u, QuicSentPacketManagerPeer::GetBytesInFlight(manager_)); | |
| 1865 } | |
| 1866 | |
| 1867 TEST_P(QuicConnectionTest, FECRemainsInflightOnTLPOfEarlierData) { | |
| 1868 // This test checks if TLP is sent correctly when a data and an FEC packet | |
| 1869 // are outstanding. TLP should be sent for the data packet when the | |
| 1870 // retransmission alarm fires. | |
| 1871 // Turn on TLP for this test. | |
| 1872 QuicSentPacketManagerPeer::SetMaxTailLossProbes(manager_, 1); | |
| 1873 EXPECT_TRUE(creator_->IsFecEnabled()); | |
| 1874 EXPECT_EQ(0u, QuicSentPacketManagerPeer::GetBytesInFlight(manager_)); | |
| 1875 EXPECT_FALSE(connection_.GetFecAlarm()->IsSet()); | |
| 1876 | |
| 1877 // 1 Data packet. FEC alarm should be set. | |
| 1878 EXPECT_CALL(*send_algorithm_, | |
| 1879 OnPacketSent(_, _, 1u, _, HAS_RETRANSMITTABLE_DATA)).Times(1); | |
| 1880 connection_.SendStreamDataWithStringWithFec(3, "foo", 0, kFin, nullptr); | |
| 1881 EXPECT_TRUE(connection_.GetFecAlarm()->IsSet()); | |
| 1882 size_t protected_packet = | |
| 1883 QuicSentPacketManagerPeer::GetBytesInFlight(manager_); | |
| 1884 EXPECT_LT(0u, protected_packet); | |
| 1885 | |
| 1886 // Force FEC timeout to send FEC packet out. | |
| 1887 EXPECT_CALL(*send_algorithm_, | |
| 1888 OnPacketSent(_, _, 2u, _, HAS_RETRANSMITTABLE_DATA)).Times(1); | |
| 1889 connection_.GetFecAlarm()->Fire(); | |
| 1890 EXPECT_TRUE(writer_->header().fec_flag); | |
| 1891 size_t fec_packet = protected_packet; | |
| 1892 EXPECT_EQ(protected_packet + fec_packet, | |
| 1893 QuicSentPacketManagerPeer::GetBytesInFlight(manager_)); | |
| 1894 | |
| 1895 // TLP alarm should be set. | |
| 1896 QuicTime retransmission_time = | |
| 1897 connection_.GetRetransmissionAlarm()->deadline(); | |
| 1898 EXPECT_NE(QuicTime::Zero(), retransmission_time); | |
| 1899 // Simulate the retransmission alarm firing and sending a TLP, so send | |
| 1900 // algorithm's OnRetransmissionTimeout is not called. | |
| 1901 clock_.AdvanceTime(retransmission_time.Subtract(clock_.Now())); | |
| 1902 EXPECT_CALL(*send_algorithm_, | |
| 1903 OnPacketSent(_, _, 3u, _, HAS_RETRANSMITTABLE_DATA)).Times(1); | |
| 1904 connection_.GetRetransmissionAlarm()->Fire(); | |
| 1905 // The TLP retransmission of packet 1 will be 3 bytes smaller than packet 1, | |
| 1906 // since packet 1 will have 1 byte for FEC group number and 2 bytes of stream | |
| 1907 // frame size, which are absent in the the TLP retransmission. | |
| 1908 size_t tlp_packet = protected_packet - 3; | |
| 1909 EXPECT_EQ(protected_packet + fec_packet + tlp_packet, | |
| 1910 QuicSentPacketManagerPeer::GetBytesInFlight(manager_)); | |
| 1911 } | |
| 1912 | |
| 1913 TEST_P(QuicConnectionTest, FECRemainsInflightOnTLPOfLaterData) { | |
| 1914 // Tests if TLP is sent correctly when data packet 1 and an FEC packet are | |
| 1915 // sent followed by data packet 2, and data packet 1 is acked. TLP should be | |
| 1916 // sent for data packet 2 when the retransmission alarm fires. Turn on TLP for | |
| 1917 // this test. | |
| 1918 QuicSentPacketManagerPeer::SetMaxTailLossProbes(manager_, 1); | |
| 1919 EXPECT_TRUE(creator_->IsFecEnabled()); | |
| 1920 EXPECT_EQ(0u, QuicSentPacketManagerPeer::GetBytesInFlight(manager_)); | |
| 1921 EXPECT_FALSE(connection_.GetFecAlarm()->IsSet()); | |
| 1922 | |
| 1923 // 1 Data packet. FEC alarm should be set. | |
| 1924 EXPECT_CALL(*send_algorithm_, | |
| 1925 OnPacketSent(_, _, 1u, _, HAS_RETRANSMITTABLE_DATA)).Times(1); | |
| 1926 connection_.SendStreamDataWithStringWithFec(3, "foo", 0, kFin, nullptr); | |
| 1927 EXPECT_TRUE(connection_.GetFecAlarm()->IsSet()); | |
| 1928 size_t protected_packet = | |
| 1929 QuicSentPacketManagerPeer::GetBytesInFlight(manager_); | |
| 1930 EXPECT_LT(0u, protected_packet); | |
| 1931 | |
| 1932 // Force FEC timeout to send FEC packet out. | |
| 1933 EXPECT_CALL(*send_algorithm_, | |
| 1934 OnPacketSent(_, _, 2u, _, HAS_RETRANSMITTABLE_DATA)).Times(1); | |
| 1935 connection_.GetFecAlarm()->Fire(); | |
| 1936 EXPECT_TRUE(writer_->header().fec_flag); | |
| 1937 // Protected data packet and FEC packet oustanding. | |
| 1938 size_t fec_packet = protected_packet; | |
| 1939 EXPECT_EQ(protected_packet + fec_packet, | |
| 1940 QuicSentPacketManagerPeer::GetBytesInFlight(manager_)); | |
| 1941 | |
| 1942 // Send 1 unprotected data packet. No FEC alarm should be set. | |
| 1943 EXPECT_CALL(*send_algorithm_, | |
| 1944 OnPacketSent(_, _, 3u, _, HAS_RETRANSMITTABLE_DATA)).Times(1); | |
| 1945 connection_.SendStreamDataWithString(5, "foo", 0, kFin, nullptr); | |
| 1946 EXPECT_FALSE(connection_.GetFecAlarm()->IsSet()); | |
| 1947 // Protected data packet, FEC packet, and unprotected data packet oustanding. | |
| 1948 // An unprotected packet will be 3 bytes smaller than an FEC-protected packet, | |
| 1949 // since the protected packet will have 1 byte for FEC group number and | |
| 1950 // 2 bytes of stream frame size, which are absent in the unprotected packet. | |
| 1951 size_t unprotected_packet = protected_packet - 3; | |
| 1952 EXPECT_EQ(protected_packet + fec_packet + unprotected_packet, | |
| 1953 QuicSentPacketManagerPeer::GetBytesInFlight(manager_)); | |
| 1954 | |
| 1955 // Receive ack for first data packet. FEC and second data packet are still | |
| 1956 // outstanding. | |
| 1957 QuicAckFrame ack = InitAckFrame(1); | |
| 1958 EXPECT_CALL(*send_algorithm_, OnCongestionEvent(true, _, _, _)); | |
| 1959 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 1960 ProcessAckPacket(&ack); | |
| 1961 // FEC packet and unprotected data packet oustanding. | |
| 1962 EXPECT_EQ(fec_packet + unprotected_packet, | |
| 1963 QuicSentPacketManagerPeer::GetBytesInFlight(manager_)); | |
| 1964 | |
| 1965 // TLP alarm should be set. | |
| 1966 QuicTime retransmission_time = | |
| 1967 connection_.GetRetransmissionAlarm()->deadline(); | |
| 1968 EXPECT_NE(QuicTime::Zero(), retransmission_time); | |
| 1969 // Simulate the retransmission alarm firing and sending a TLP, so send | |
| 1970 // algorithm's OnRetransmissionTimeout is not called. | |
| 1971 clock_.AdvanceTime(retransmission_time.Subtract(clock_.Now())); | |
| 1972 EXPECT_CALL(*send_algorithm_, | |
| 1973 OnPacketSent(_, _, 4u, _, HAS_RETRANSMITTABLE_DATA)).Times(1); | |
| 1974 connection_.GetRetransmissionAlarm()->Fire(); | |
| 1975 | |
| 1976 // Having received an ack from the server, the client will no longer send | |
| 1977 // version number in subsequent packets, including in this retransmission. | |
| 1978 size_t tlp_packet_no_version = unprotected_packet - 4; | |
| 1979 EXPECT_EQ(fec_packet + unprotected_packet + tlp_packet_no_version, | |
| 1980 QuicSentPacketManagerPeer::GetBytesInFlight(manager_)); | |
| 1981 } | |
| 1982 | |
| 1983 TEST_P(QuicConnectionTest, NoTLPForFECPacket) { | |
| 1984 // Turn on TLP for this test. | |
| 1985 QuicSentPacketManagerPeer::SetMaxTailLossProbes(manager_, 1); | |
| 1986 EXPECT_TRUE(creator_->IsFecEnabled()); | |
| 1987 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 1988 | |
| 1989 // Send 1 FEC-protected data packet. FEC alarm should be set. | |
| 1990 EXPECT_CALL(*send_algorithm_, | |
| 1991 OnPacketSent(_, _, _, _, HAS_RETRANSMITTABLE_DATA)).Times(1); | |
| 1992 connection_.SendStreamDataWithStringWithFec(3, "foo", 0, !kFin, nullptr); | |
| 1993 EXPECT_TRUE(connection_.GetFecAlarm()->IsSet()); | |
| 1994 // Force FEC timeout to send FEC packet out. | |
| 1995 EXPECT_CALL(*send_algorithm_, | |
| 1996 OnPacketSent(_, _, _, _, HAS_RETRANSMITTABLE_DATA)).Times(1); | |
| 1997 connection_.GetFecAlarm()->Fire(); | |
| 1998 EXPECT_TRUE(writer_->header().fec_flag); | |
| 1999 | |
| 2000 // Ack data packet, but not FEC packet. | |
| 2001 QuicAckFrame ack = InitAckFrame(1); | |
| 2002 EXPECT_CALL(*send_algorithm_, OnCongestionEvent(true, _, _, _)); | |
| 2003 ProcessAckPacket(&ack); | |
| 2004 | |
| 2005 // No TLP alarm for FEC, but retransmission alarm should be set for an RTO. | |
| 2006 EXPECT_LT(0u, QuicSentPacketManagerPeer::GetBytesInFlight(manager_)); | |
| 2007 EXPECT_TRUE(connection_.GetRetransmissionAlarm()->IsSet()); | |
| 2008 QuicTime rto_time = connection_.GetRetransmissionAlarm()->deadline(); | |
| 2009 EXPECT_NE(QuicTime::Zero(), rto_time); | |
| 2010 | |
| 2011 // Simulate the retransmission alarm firing. FEC packet is no longer | |
| 2012 // outstanding. | |
| 2013 if (!FLAGS_quic_use_new_rto) { | |
| 2014 EXPECT_CALL(*send_algorithm_, OnRetransmissionTimeout(false)); | |
| 2015 } | |
| 2016 clock_.AdvanceTime(rto_time.Subtract(clock_.Now())); | |
| 2017 connection_.GetRetransmissionAlarm()->Fire(); | |
| 2018 | |
| 2019 EXPECT_FALSE(connection_.GetRetransmissionAlarm()->IsSet()); | |
| 2020 EXPECT_EQ(0u, QuicSentPacketManagerPeer::GetBytesInFlight(manager_)); | |
| 2021 } | |
| 2022 | |
| 2023 TEST_P(QuicConnectionTest, FramePacking) { | |
| 2024 CongestionBlockWrites(); | |
| 2025 | |
| 2026 // Send an ack and two stream frames in 1 packet by queueing them. | |
| 2027 connection_.SendAck(); | |
| 2028 EXPECT_CALL(visitor_, OnCanWrite()).WillOnce(DoAll( | |
| 2029 IgnoreResult(InvokeWithoutArgs(&connection_, | |
| 2030 &TestConnection::SendStreamData3)), | |
| 2031 IgnoreResult(InvokeWithoutArgs(&connection_, | |
| 2032 &TestConnection::SendStreamData5)))); | |
| 2033 | |
| 2034 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)).Times(1); | |
| 2035 CongestionUnblockWrites(); | |
| 2036 connection_.GetSendAlarm()->Fire(); | |
| 2037 EXPECT_EQ(0u, connection_.NumQueuedPackets()); | |
| 2038 EXPECT_FALSE(connection_.HasQueuedData()); | |
| 2039 | |
| 2040 // Parse the last packet and ensure it's an ack and two stream frames from | |
| 2041 // two different streams. | |
| 2042 EXPECT_EQ(4u, writer_->frame_count()); | |
| 2043 EXPECT_FALSE(writer_->stop_waiting_frames().empty()); | |
| 2044 EXPECT_FALSE(writer_->ack_frames().empty()); | |
| 2045 ASSERT_EQ(2u, writer_->stream_frames().size()); | |
| 2046 EXPECT_EQ(kClientDataStreamId1, writer_->stream_frames()[0].stream_id); | |
| 2047 EXPECT_EQ(kClientDataStreamId2, writer_->stream_frames()[1].stream_id); | |
| 2048 } | |
| 2049 | |
| 2050 TEST_P(QuicConnectionTest, FramePackingNonCryptoThenCrypto) { | |
| 2051 CongestionBlockWrites(); | |
| 2052 | |
| 2053 // Send an ack and two stream frames (one non-crypto, then one crypto) in 2 | |
| 2054 // packets by queueing them. | |
| 2055 connection_.SendAck(); | |
| 2056 EXPECT_CALL(visitor_, OnCanWrite()).WillOnce(DoAll( | |
| 2057 IgnoreResult(InvokeWithoutArgs(&connection_, | |
| 2058 &TestConnection::SendStreamData3)), | |
| 2059 IgnoreResult(InvokeWithoutArgs(&connection_, | |
| 2060 &TestConnection::SendCryptoStreamData)))); | |
| 2061 | |
| 2062 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)).Times(2); | |
| 2063 CongestionUnblockWrites(); | |
| 2064 connection_.GetSendAlarm()->Fire(); | |
| 2065 EXPECT_EQ(0u, connection_.NumQueuedPackets()); | |
| 2066 EXPECT_FALSE(connection_.HasQueuedData()); | |
| 2067 | |
| 2068 // Parse the last packet and ensure it's the crypto stream frame. | |
| 2069 EXPECT_EQ(1u, writer_->frame_count()); | |
| 2070 ASSERT_EQ(1u, writer_->stream_frames().size()); | |
| 2071 EXPECT_EQ(kCryptoStreamId, writer_->stream_frames()[0].stream_id); | |
| 2072 } | |
| 2073 | |
| 2074 TEST_P(QuicConnectionTest, FramePackingCryptoThenNonCrypto) { | |
| 2075 CongestionBlockWrites(); | |
| 2076 | |
| 2077 // Send an ack and two stream frames (one crypto, then one non-crypto) in 2 | |
| 2078 // packets by queueing them. | |
| 2079 connection_.SendAck(); | |
| 2080 EXPECT_CALL(visitor_, OnCanWrite()).WillOnce(DoAll( | |
| 2081 IgnoreResult(InvokeWithoutArgs(&connection_, | |
| 2082 &TestConnection::SendCryptoStreamData)), | |
| 2083 IgnoreResult(InvokeWithoutArgs(&connection_, | |
| 2084 &TestConnection::SendStreamData3)))); | |
| 2085 | |
| 2086 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)).Times(2); | |
| 2087 CongestionUnblockWrites(); | |
| 2088 connection_.GetSendAlarm()->Fire(); | |
| 2089 EXPECT_EQ(0u, connection_.NumQueuedPackets()); | |
| 2090 EXPECT_FALSE(connection_.HasQueuedData()); | |
| 2091 | |
| 2092 // Parse the last packet and ensure it's the stream frame from stream 3. | |
| 2093 EXPECT_EQ(1u, writer_->frame_count()); | |
| 2094 ASSERT_EQ(1u, writer_->stream_frames().size()); | |
| 2095 EXPECT_EQ(kClientDataStreamId1, writer_->stream_frames()[0].stream_id); | |
| 2096 } | |
| 2097 | |
| 2098 TEST_P(QuicConnectionTest, FramePackingFEC) { | |
| 2099 EXPECT_TRUE(creator_->IsFecEnabled()); | |
| 2100 | |
| 2101 CongestionBlockWrites(); | |
| 2102 | |
| 2103 // Queue an ack and two stream frames. Ack gets flushed when FEC is turned on | |
| 2104 // for sending protected data; two stream frames are packed in 1 packet. | |
| 2105 EXPECT_CALL(visitor_, OnCanWrite()).WillOnce(DoAll( | |
| 2106 IgnoreResult(InvokeWithoutArgs( | |
| 2107 &connection_, &TestConnection::SendStreamData3WithFec)), | |
| 2108 IgnoreResult(InvokeWithoutArgs( | |
| 2109 &connection_, &TestConnection::SendStreamData5WithFec)))); | |
| 2110 connection_.SendAck(); | |
| 2111 | |
| 2112 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)).Times(2); | |
| 2113 CongestionUnblockWrites(); | |
| 2114 connection_.GetSendAlarm()->Fire(); | |
| 2115 EXPECT_EQ(0u, connection_.NumQueuedPackets()); | |
| 2116 EXPECT_FALSE(connection_.HasQueuedData()); | |
| 2117 | |
| 2118 // Parse the last packet and ensure it's in an fec group. | |
| 2119 EXPECT_EQ(2u, writer_->header().fec_group); | |
| 2120 EXPECT_EQ(2u, writer_->frame_count()); | |
| 2121 | |
| 2122 // FEC alarm should be set. | |
| 2123 EXPECT_TRUE(connection_.GetFecAlarm()->IsSet()); | |
| 2124 } | |
| 2125 | |
| 2126 TEST_P(QuicConnectionTest, FramePackingAckResponse) { | |
| 2127 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 2128 // Process a data packet to queue up a pending ack. | |
| 2129 EXPECT_CALL(visitor_, OnStreamFrames(_)).Times(1); | |
| 2130 ProcessDataPacket(1, 1, kEntropyFlag); | |
| 2131 | |
| 2132 EXPECT_CALL(visitor_, OnCanWrite()).WillOnce(DoAll( | |
| 2133 IgnoreResult(InvokeWithoutArgs(&connection_, | |
| 2134 &TestConnection::SendStreamData3)), | |
| 2135 IgnoreResult(InvokeWithoutArgs(&connection_, | |
| 2136 &TestConnection::SendStreamData5)))); | |
| 2137 | |
| 2138 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)).Times(1); | |
| 2139 | |
| 2140 // Process an ack to cause the visitor's OnCanWrite to be invoked. | |
| 2141 peer_creator_.set_sequence_number(2); | |
| 2142 QuicAckFrame ack_one = InitAckFrame(0); | |
| 2143 ProcessAckPacket(&ack_one); | |
| 2144 | |
| 2145 EXPECT_EQ(0u, connection_.NumQueuedPackets()); | |
| 2146 EXPECT_FALSE(connection_.HasQueuedData()); | |
| 2147 | |
| 2148 // Parse the last packet and ensure it's an ack and two stream frames from | |
| 2149 // two different streams. | |
| 2150 EXPECT_EQ(4u, writer_->frame_count()); | |
| 2151 EXPECT_FALSE(writer_->stop_waiting_frames().empty()); | |
| 2152 EXPECT_FALSE(writer_->ack_frames().empty()); | |
| 2153 ASSERT_EQ(2u, writer_->stream_frames().size()); | |
| 2154 EXPECT_EQ(kClientDataStreamId1, writer_->stream_frames()[0].stream_id); | |
| 2155 EXPECT_EQ(kClientDataStreamId2, writer_->stream_frames()[1].stream_id); | |
| 2156 } | |
| 2157 | |
| 2158 TEST_P(QuicConnectionTest, FramePackingSendv) { | |
| 2159 // Send data in 1 packet by writing multiple blocks in a single iovector | |
| 2160 // using writev. | |
| 2161 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)); | |
| 2162 | |
| 2163 char data[] = "ABCD"; | |
| 2164 IOVector data_iov; | |
| 2165 data_iov.AppendNoCoalesce(data, 2); | |
| 2166 data_iov.AppendNoCoalesce(data + 2, 2); | |
| 2167 connection_.SendStreamData(1, data_iov, 0, !kFin, MAY_FEC_PROTECT, nullptr); | |
| 2168 | |
| 2169 EXPECT_EQ(0u, connection_.NumQueuedPackets()); | |
| 2170 EXPECT_FALSE(connection_.HasQueuedData()); | |
| 2171 | |
| 2172 // Parse the last packet and ensure multiple iovector blocks have | |
| 2173 // been packed into a single stream frame from one stream. | |
| 2174 EXPECT_EQ(1u, writer_->frame_count()); | |
| 2175 EXPECT_EQ(1u, writer_->stream_frames().size()); | |
| 2176 QuicStreamFrame frame = writer_->stream_frames()[0]; | |
| 2177 EXPECT_EQ(1u, frame.stream_id); | |
| 2178 EXPECT_EQ("ABCD", string(static_cast<char*> | |
| 2179 (frame.data.iovec()[0].iov_base), | |
| 2180 (frame.data.iovec()[0].iov_len))); | |
| 2181 } | |
| 2182 | |
| 2183 TEST_P(QuicConnectionTest, FramePackingSendvQueued) { | |
| 2184 // Try to send two stream frames in 1 packet by using writev. | |
| 2185 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)); | |
| 2186 | |
| 2187 BlockOnNextWrite(); | |
| 2188 char data[] = "ABCD"; | |
| 2189 IOVector data_iov; | |
| 2190 data_iov.AppendNoCoalesce(data, 2); | |
| 2191 data_iov.AppendNoCoalesce(data + 2, 2); | |
| 2192 connection_.SendStreamData(1, data_iov, 0, !kFin, MAY_FEC_PROTECT, nullptr); | |
| 2193 | |
| 2194 EXPECT_EQ(1u, connection_.NumQueuedPackets()); | |
| 2195 EXPECT_TRUE(connection_.HasQueuedData()); | |
| 2196 | |
| 2197 // Unblock the writes and actually send. | |
| 2198 writer_->SetWritable(); | |
| 2199 connection_.OnCanWrite(); | |
| 2200 EXPECT_EQ(0u, connection_.NumQueuedPackets()); | |
| 2201 | |
| 2202 // Parse the last packet and ensure it's one stream frame from one stream. | |
| 2203 EXPECT_EQ(1u, writer_->frame_count()); | |
| 2204 EXPECT_EQ(1u, writer_->stream_frames().size()); | |
| 2205 EXPECT_EQ(1u, writer_->stream_frames()[0].stream_id); | |
| 2206 } | |
| 2207 | |
| 2208 TEST_P(QuicConnectionTest, SendingZeroBytes) { | |
| 2209 // Send a zero byte write with a fin using writev. | |
| 2210 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)); | |
| 2211 IOVector empty_iov; | |
| 2212 connection_.SendStreamData(1, empty_iov, 0, kFin, MAY_FEC_PROTECT, nullptr); | |
| 2213 | |
| 2214 EXPECT_EQ(0u, connection_.NumQueuedPackets()); | |
| 2215 EXPECT_FALSE(connection_.HasQueuedData()); | |
| 2216 | |
| 2217 // Parse the last packet and ensure it's one stream frame from one stream. | |
| 2218 EXPECT_EQ(1u, writer_->frame_count()); | |
| 2219 EXPECT_EQ(1u, writer_->stream_frames().size()); | |
| 2220 EXPECT_EQ(1u, writer_->stream_frames()[0].stream_id); | |
| 2221 EXPECT_TRUE(writer_->stream_frames()[0].fin); | |
| 2222 } | |
| 2223 | |
| 2224 TEST_P(QuicConnectionTest, OnCanWrite) { | |
| 2225 // Visitor's OnCanWrite will send data, but will have more pending writes. | |
| 2226 EXPECT_CALL(visitor_, OnCanWrite()).WillOnce(DoAll( | |
| 2227 IgnoreResult(InvokeWithoutArgs(&connection_, | |
| 2228 &TestConnection::SendStreamData3)), | |
| 2229 IgnoreResult(InvokeWithoutArgs(&connection_, | |
| 2230 &TestConnection::SendStreamData5)))); | |
| 2231 EXPECT_CALL(visitor_, WillingAndAbleToWrite()).WillOnce(Return(true)); | |
| 2232 EXPECT_CALL(*send_algorithm_, | |
| 2233 TimeUntilSend(_, _, _)).WillRepeatedly( | |
| 2234 testing::Return(QuicTime::Delta::Zero())); | |
| 2235 | |
| 2236 connection_.OnCanWrite(); | |
| 2237 | |
| 2238 // Parse the last packet and ensure it's the two stream frames from | |
| 2239 // two different streams. | |
| 2240 EXPECT_EQ(2u, writer_->frame_count()); | |
| 2241 EXPECT_EQ(2u, writer_->stream_frames().size()); | |
| 2242 EXPECT_EQ(kClientDataStreamId1, writer_->stream_frames()[0].stream_id); | |
| 2243 EXPECT_EQ(kClientDataStreamId2, writer_->stream_frames()[1].stream_id); | |
| 2244 } | |
| 2245 | |
| 2246 TEST_P(QuicConnectionTest, RetransmitOnNack) { | |
| 2247 QuicPacketSequenceNumber last_packet; | |
| 2248 QuicByteCount second_packet_size; | |
| 2249 SendStreamDataToPeer(3, "foo", 0, !kFin, &last_packet); // Packet 1 | |
| 2250 second_packet_size = | |
| 2251 SendStreamDataToPeer(3, "foos", 3, !kFin, &last_packet); // Packet 2 | |
| 2252 SendStreamDataToPeer(3, "fooos", 7, !kFin, &last_packet); // Packet 3 | |
| 2253 | |
| 2254 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 2255 | |
| 2256 // Don't lose a packet on an ack, and nothing is retransmitted. | |
| 2257 EXPECT_CALL(*send_algorithm_, OnCongestionEvent(true, _, _, _)); | |
| 2258 QuicAckFrame ack_one = InitAckFrame(1); | |
| 2259 ProcessAckPacket(&ack_one); | |
| 2260 | |
| 2261 // Lose a packet and ensure it triggers retransmission. | |
| 2262 QuicAckFrame nack_two = InitAckFrame(3); | |
| 2263 NackPacket(2, &nack_two); | |
| 2264 SequenceNumberSet lost_packets; | |
| 2265 lost_packets.insert(2); | |
| 2266 EXPECT_CALL(*loss_algorithm_, DetectLostPackets(_, _, _, _)) | |
| 2267 .WillOnce(Return(lost_packets)); | |
| 2268 EXPECT_CALL(*send_algorithm_, OnCongestionEvent(true, _, _, _)); | |
| 2269 EXPECT_CALL(*send_algorithm_, | |
| 2270 OnPacketSent(_, _, _, second_packet_size - kQuicVersionSize, _)). | |
| 2271 Times(1); | |
| 2272 ProcessAckPacket(&nack_two); | |
| 2273 } | |
| 2274 | |
| 2275 TEST_P(QuicConnectionTest, DiscardRetransmit) { | |
| 2276 QuicPacketSequenceNumber last_packet; | |
| 2277 SendStreamDataToPeer(1, "foo", 0, !kFin, &last_packet); // Packet 1 | |
| 2278 SendStreamDataToPeer(1, "foos", 3, !kFin, &last_packet); // Packet 2 | |
| 2279 SendStreamDataToPeer(1, "fooos", 7, !kFin, &last_packet); // Packet 3 | |
| 2280 | |
| 2281 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 2282 | |
| 2283 // Instigate a loss with an ack. | |
| 2284 QuicAckFrame nack_two = InitAckFrame(3); | |
| 2285 NackPacket(2, &nack_two); | |
| 2286 // The first nack should trigger a fast retransmission, but we'll be | |
| 2287 // write blocked, so the packet will be queued. | |
| 2288 BlockOnNextWrite(); | |
| 2289 SequenceNumberSet lost_packets; | |
| 2290 lost_packets.insert(2); | |
| 2291 EXPECT_CALL(*loss_algorithm_, DetectLostPackets(_, _, _, _)) | |
| 2292 .WillOnce(Return(lost_packets)); | |
| 2293 EXPECT_CALL(*send_algorithm_, OnCongestionEvent(true, _, _, _)); | |
| 2294 ProcessAckPacket(&nack_two); | |
| 2295 EXPECT_EQ(1u, connection_.NumQueuedPackets()); | |
| 2296 | |
| 2297 // Now, ack the previous transmission. | |
| 2298 EXPECT_CALL(*loss_algorithm_, DetectLostPackets(_, _, _, _)) | |
| 2299 .WillOnce(Return(SequenceNumberSet())); | |
| 2300 QuicAckFrame ack_all = InitAckFrame(3); | |
| 2301 ProcessAckPacket(&ack_all); | |
| 2302 | |
| 2303 // Unblock the socket and attempt to send the queued packets. However, | |
| 2304 // since the previous transmission has been acked, we will not | |
| 2305 // send the retransmission. | |
| 2306 EXPECT_CALL(*send_algorithm_, | |
| 2307 OnPacketSent(_, _, _, _, _)).Times(0); | |
| 2308 | |
| 2309 writer_->SetWritable(); | |
| 2310 connection_.OnCanWrite(); | |
| 2311 | |
| 2312 EXPECT_EQ(0u, connection_.NumQueuedPackets()); | |
| 2313 } | |
| 2314 | |
| 2315 TEST_P(QuicConnectionTest, RetransmitNackedLargestObserved) { | |
| 2316 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 2317 QuicPacketSequenceNumber largest_observed; | |
| 2318 QuicByteCount packet_size; | |
| 2319 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)) | |
| 2320 .WillOnce(DoAll(SaveArg<2>(&largest_observed), SaveArg<3>(&packet_size), | |
| 2321 Return(true))); | |
| 2322 connection_.SendStreamDataWithString(3, "foo", 0, !kFin, nullptr); | |
| 2323 | |
| 2324 QuicAckFrame frame = InitAckFrame(1); | |
| 2325 NackPacket(largest_observed, &frame); | |
| 2326 // The first nack should retransmit the largest observed packet. | |
| 2327 SequenceNumberSet lost_packets; | |
| 2328 lost_packets.insert(1); | |
| 2329 EXPECT_CALL(*loss_algorithm_, DetectLostPackets(_, _, _, _)) | |
| 2330 .WillOnce(Return(lost_packets)); | |
| 2331 EXPECT_CALL(*send_algorithm_, OnCongestionEvent(true, _, _, _)); | |
| 2332 EXPECT_CALL(*send_algorithm_, | |
| 2333 OnPacketSent(_, _, _, packet_size - kQuicVersionSize, _)); | |
| 2334 ProcessAckPacket(&frame); | |
| 2335 } | |
| 2336 | |
| 2337 TEST_P(QuicConnectionTest, QueueAfterTwoRTOs) { | |
| 2338 for (int i = 0; i < 10; ++i) { | |
| 2339 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)).Times(1); | |
| 2340 connection_.SendStreamDataWithString(3, "foo", i * 3, !kFin, nullptr); | |
| 2341 } | |
| 2342 | |
| 2343 // Block the writer and ensure they're queued. | |
| 2344 BlockOnNextWrite(); | |
| 2345 clock_.AdvanceTime(DefaultRetransmissionTime()); | |
| 2346 // Only one packet should be retransmitted. | |
| 2347 if (!FLAGS_quic_use_new_rto) { | |
| 2348 EXPECT_CALL(*send_algorithm_, OnRetransmissionTimeout(true)); | |
| 2349 } | |
| 2350 connection_.GetRetransmissionAlarm()->Fire(); | |
| 2351 EXPECT_TRUE(connection_.HasQueuedData()); | |
| 2352 | |
| 2353 // Unblock the writer. | |
| 2354 writer_->SetWritable(); | |
| 2355 clock_.AdvanceTime(QuicTime::Delta::FromMicroseconds( | |
| 2356 2 * DefaultRetransmissionTime().ToMicroseconds())); | |
| 2357 // Retransmit already retransmitted packets event though the sequence number | |
| 2358 // greater than the largest observed. | |
| 2359 if (FLAGS_quic_use_new_rto) { | |
| 2360 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)).Times(2); | |
| 2361 } else { | |
| 2362 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)).Times(10); | |
| 2363 } | |
| 2364 connection_.GetRetransmissionAlarm()->Fire(); | |
| 2365 connection_.OnCanWrite(); | |
| 2366 } | |
| 2367 | |
| 2368 TEST_P(QuicConnectionTest, WriteBlockedThenSent) { | |
| 2369 BlockOnNextWrite(); | |
| 2370 writer_->set_is_write_blocked_data_buffered(true); | |
| 2371 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)).Times(1); | |
| 2372 connection_.SendStreamDataWithString(1, "foo", 0, !kFin, nullptr); | |
| 2373 EXPECT_TRUE(connection_.GetRetransmissionAlarm()->IsSet()); | |
| 2374 | |
| 2375 writer_->SetWritable(); | |
| 2376 connection_.OnCanWrite(); | |
| 2377 EXPECT_TRUE(connection_.GetRetransmissionAlarm()->IsSet()); | |
| 2378 } | |
| 2379 | |
| 2380 TEST_P(QuicConnectionTest, RetransmitWriteBlockedAckedOriginalThenSent) { | |
| 2381 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 2382 connection_.SendStreamDataWithString(3, "foo", 0, !kFin, nullptr); | |
| 2383 EXPECT_TRUE(connection_.GetRetransmissionAlarm()->IsSet()); | |
| 2384 | |
| 2385 BlockOnNextWrite(); | |
| 2386 writer_->set_is_write_blocked_data_buffered(true); | |
| 2387 // Simulate the retransmission alarm firing. | |
| 2388 if (!FLAGS_quic_use_new_rto) { | |
| 2389 EXPECT_CALL(*send_algorithm_, OnRetransmissionTimeout(_)); | |
| 2390 } | |
| 2391 clock_.AdvanceTime(DefaultRetransmissionTime()); | |
| 2392 connection_.GetRetransmissionAlarm()->Fire(); | |
| 2393 | |
| 2394 // Ack the sent packet before the callback returns, which happens in | |
| 2395 // rare circumstances with write blocked sockets. | |
| 2396 QuicAckFrame ack = InitAckFrame(1); | |
| 2397 EXPECT_CALL(*send_algorithm_, OnCongestionEvent(true, _, _, _)); | |
| 2398 if (!FLAGS_quic_use_new_rto) { | |
| 2399 EXPECT_CALL(*send_algorithm_, RevertRetransmissionTimeout()); | |
| 2400 } | |
| 2401 ProcessAckPacket(&ack); | |
| 2402 | |
| 2403 writer_->SetWritable(); | |
| 2404 connection_.OnCanWrite(); | |
| 2405 // There is now a pending packet, but with no retransmittable frames. | |
| 2406 EXPECT_TRUE(connection_.GetRetransmissionAlarm()->IsSet()); | |
| 2407 EXPECT_FALSE(connection_.sent_packet_manager().HasRetransmittableFrames(2)); | |
| 2408 } | |
| 2409 | |
| 2410 TEST_P(QuicConnectionTest, AlarmsWhenWriteBlocked) { | |
| 2411 // Block the connection. | |
| 2412 BlockOnNextWrite(); | |
| 2413 connection_.SendStreamDataWithString(3, "foo", 0, !kFin, nullptr); | |
| 2414 EXPECT_EQ(1u, writer_->packets_write_attempts()); | |
| 2415 EXPECT_TRUE(writer_->IsWriteBlocked()); | |
| 2416 | |
| 2417 // Set the send and resumption alarms. Fire the alarms and ensure they don't | |
| 2418 // attempt to write. | |
| 2419 connection_.GetResumeWritesAlarm()->Set(clock_.ApproximateNow()); | |
| 2420 connection_.GetSendAlarm()->Set(clock_.ApproximateNow()); | |
| 2421 connection_.GetResumeWritesAlarm()->Fire(); | |
| 2422 connection_.GetSendAlarm()->Fire(); | |
| 2423 EXPECT_TRUE(writer_->IsWriteBlocked()); | |
| 2424 EXPECT_EQ(1u, writer_->packets_write_attempts()); | |
| 2425 } | |
| 2426 | |
| 2427 TEST_P(QuicConnectionTest, NoLimitPacketsPerNack) { | |
| 2428 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 2429 int offset = 0; | |
| 2430 // Send packets 1 to 15. | |
| 2431 for (int i = 0; i < 15; ++i) { | |
| 2432 SendStreamDataToPeer(1, "foo", offset, !kFin, nullptr); | |
| 2433 offset += 3; | |
| 2434 } | |
| 2435 | |
| 2436 // Ack 15, nack 1-14. | |
| 2437 SequenceNumberSet lost_packets; | |
| 2438 QuicAckFrame nack = InitAckFrame(15); | |
| 2439 for (int i = 1; i < 15; ++i) { | |
| 2440 NackPacket(i, &nack); | |
| 2441 lost_packets.insert(i); | |
| 2442 } | |
| 2443 | |
| 2444 // 14 packets have been NACK'd and lost. In TCP cubic, PRR limits | |
| 2445 // the retransmission rate in the case of burst losses. | |
| 2446 EXPECT_CALL(*loss_algorithm_, DetectLostPackets(_, _, _, _)) | |
| 2447 .WillOnce(Return(lost_packets)); | |
| 2448 EXPECT_CALL(*send_algorithm_, OnCongestionEvent(true, _, _, _)); | |
| 2449 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)).Times(14); | |
| 2450 ProcessAckPacket(&nack); | |
| 2451 } | |
| 2452 | |
| 2453 // Test sending multiple acks from the connection to the session. | |
| 2454 TEST_P(QuicConnectionTest, MultipleAcks) { | |
| 2455 QuicPacketSequenceNumber last_packet; | |
| 2456 SendStreamDataToPeer(1, "foo", 0, !kFin, &last_packet); // Packet 1 | |
| 2457 EXPECT_EQ(1u, last_packet); | |
| 2458 SendStreamDataToPeer(3, "foo", 0, !kFin, &last_packet); // Packet 2 | |
| 2459 EXPECT_EQ(2u, last_packet); | |
| 2460 SendAckPacketToPeer(); // Packet 3 | |
| 2461 SendStreamDataToPeer(5, "foo", 0, !kFin, &last_packet); // Packet 4 | |
| 2462 EXPECT_EQ(4u, last_packet); | |
| 2463 SendStreamDataToPeer(1, "foo", 3, !kFin, &last_packet); // Packet 5 | |
| 2464 EXPECT_EQ(5u, last_packet); | |
| 2465 SendStreamDataToPeer(3, "foo", 3, !kFin, &last_packet); // Packet 6 | |
| 2466 EXPECT_EQ(6u, last_packet); | |
| 2467 | |
| 2468 // Client will ack packets 1, 2, [!3], 4, 5. | |
| 2469 EXPECT_CALL(*send_algorithm_, OnCongestionEvent(true, _, _, _)); | |
| 2470 QuicAckFrame frame1 = InitAckFrame(5); | |
| 2471 NackPacket(3, &frame1); | |
| 2472 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 2473 ProcessAckPacket(&frame1); | |
| 2474 | |
| 2475 // Now the client implicitly acks 3, and explicitly acks 6. | |
| 2476 EXPECT_CALL(*send_algorithm_, OnCongestionEvent(true, _, _, _)); | |
| 2477 QuicAckFrame frame2 = InitAckFrame(6); | |
| 2478 ProcessAckPacket(&frame2); | |
| 2479 } | |
| 2480 | |
| 2481 TEST_P(QuicConnectionTest, DontLatchUnackedPacket) { | |
| 2482 SendStreamDataToPeer(1, "foo", 0, !kFin, nullptr); // Packet 1; | |
| 2483 // From now on, we send acks, so the send algorithm won't mark them pending. | |
| 2484 ON_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)) | |
| 2485 .WillByDefault(Return(false)); | |
| 2486 SendAckPacketToPeer(); // Packet 2 | |
| 2487 | |
| 2488 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 2489 EXPECT_CALL(*send_algorithm_, OnCongestionEvent(true, _, _, _)); | |
| 2490 QuicAckFrame frame = InitAckFrame(1); | |
| 2491 ProcessAckPacket(&frame); | |
| 2492 | |
| 2493 // Verify that our internal state has least-unacked as 2, because we're still | |
| 2494 // waiting for a potential ack for 2. | |
| 2495 | |
| 2496 EXPECT_EQ(2u, stop_waiting()->least_unacked); | |
| 2497 | |
| 2498 EXPECT_CALL(*send_algorithm_, OnCongestionEvent(true, _, _, _)); | |
| 2499 frame = InitAckFrame(2); | |
| 2500 ProcessAckPacket(&frame); | |
| 2501 EXPECT_EQ(3u, stop_waiting()->least_unacked); | |
| 2502 | |
| 2503 // When we send an ack, we make sure our least-unacked makes sense. In this | |
| 2504 // case since we're not waiting on an ack for 2 and all packets are acked, we | |
| 2505 // set it to 3. | |
| 2506 SendAckPacketToPeer(); // Packet 3 | |
| 2507 // Least_unacked remains at 3 until another ack is received. | |
| 2508 EXPECT_EQ(3u, stop_waiting()->least_unacked); | |
| 2509 // Check that the outgoing ack had its sequence number as least_unacked. | |
| 2510 EXPECT_EQ(3u, least_unacked()); | |
| 2511 | |
| 2512 // Ack the ack, which updates the rtt and raises the least unacked. | |
| 2513 EXPECT_CALL(*send_algorithm_, OnCongestionEvent(true, _, _, _)); | |
| 2514 frame = InitAckFrame(3); | |
| 2515 ProcessAckPacket(&frame); | |
| 2516 | |
| 2517 ON_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)) | |
| 2518 .WillByDefault(Return(true)); | |
| 2519 SendStreamDataToPeer(1, "bar", 3, false, nullptr); // Packet 4 | |
| 2520 EXPECT_EQ(4u, stop_waiting()->least_unacked); | |
| 2521 ON_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)) | |
| 2522 .WillByDefault(Return(false)); | |
| 2523 SendAckPacketToPeer(); // Packet 5 | |
| 2524 EXPECT_EQ(4u, least_unacked()); | |
| 2525 | |
| 2526 // Send two data packets at the end, and ensure if the last one is acked, | |
| 2527 // the least unacked is raised above the ack packets. | |
| 2528 ON_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)) | |
| 2529 .WillByDefault(Return(true)); | |
| 2530 SendStreamDataToPeer(1, "bar", 6, false, nullptr); // Packet 6 | |
| 2531 SendStreamDataToPeer(1, "bar", 9, false, nullptr); // Packet 7 | |
| 2532 | |
| 2533 EXPECT_CALL(*send_algorithm_, OnCongestionEvent(true, _, _, _)); | |
| 2534 frame = InitAckFrame(7); | |
| 2535 NackPacket(5, &frame); | |
| 2536 NackPacket(6, &frame); | |
| 2537 ProcessAckPacket(&frame); | |
| 2538 | |
| 2539 EXPECT_EQ(6u, stop_waiting()->least_unacked); | |
| 2540 } | |
| 2541 | |
| 2542 TEST_P(QuicConnectionTest, ReviveMissingPacketAfterFecPacket) { | |
| 2543 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 2544 | |
| 2545 // Don't send missing packet 1. | |
| 2546 ProcessFecPacket(2, 1, true, !kEntropyFlag, nullptr); | |
| 2547 // Entropy flag should be false, so entropy should be 0. | |
| 2548 EXPECT_EQ(0u, QuicConnectionPeer::ReceivedEntropyHash(&connection_, 2)); | |
| 2549 } | |
| 2550 | |
| 2551 TEST_P(QuicConnectionTest, ReviveMissingPacketWithVaryingSeqNumLengths) { | |
| 2552 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 2553 | |
| 2554 // Set up a debug visitor to the connection. | |
| 2555 FecQuicConnectionDebugVisitor* fec_visitor = | |
| 2556 new FecQuicConnectionDebugVisitor(); | |
| 2557 connection_.set_debug_visitor(fec_visitor); | |
| 2558 | |
| 2559 QuicPacketSequenceNumber fec_packet = 0; | |
| 2560 QuicSequenceNumberLength lengths[] = {PACKET_6BYTE_SEQUENCE_NUMBER, | |
| 2561 PACKET_4BYTE_SEQUENCE_NUMBER, | |
| 2562 PACKET_2BYTE_SEQUENCE_NUMBER, | |
| 2563 PACKET_1BYTE_SEQUENCE_NUMBER}; | |
| 2564 // For each sequence number length size, revive a packet and check sequence | |
| 2565 // number length in the revived packet. | |
| 2566 for (size_t i = 0; i < arraysize(lengths); ++i) { | |
| 2567 // Set sequence_number_length_ (for data and FEC packets). | |
| 2568 sequence_number_length_ = lengths[i]; | |
| 2569 fec_packet += 2; | |
| 2570 // Don't send missing packet, but send fec packet right after it. | |
| 2571 ProcessFecPacket(fec_packet, fec_packet - 1, true, !kEntropyFlag, nullptr); | |
| 2572 // Sequence number length in the revived header should be the same as | |
| 2573 // in the original data/fec packet headers. | |
| 2574 EXPECT_EQ(sequence_number_length_, fec_visitor->revived_header(). | |
| 2575 public_header.sequence_number_length); | |
| 2576 } | |
| 2577 } | |
| 2578 | |
| 2579 TEST_P(QuicConnectionTest, ReviveMissingPacketWithVaryingConnectionIdLengths) { | |
| 2580 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 2581 | |
| 2582 // Set up a debug visitor to the connection. | |
| 2583 FecQuicConnectionDebugVisitor* fec_visitor = | |
| 2584 new FecQuicConnectionDebugVisitor(); | |
| 2585 connection_.set_debug_visitor(fec_visitor); | |
| 2586 | |
| 2587 QuicPacketSequenceNumber fec_packet = 0; | |
| 2588 QuicConnectionIdLength lengths[] = {PACKET_8BYTE_CONNECTION_ID, | |
| 2589 PACKET_4BYTE_CONNECTION_ID, | |
| 2590 PACKET_1BYTE_CONNECTION_ID, | |
| 2591 PACKET_0BYTE_CONNECTION_ID}; | |
| 2592 // For each connection id length size, revive a packet and check connection | |
| 2593 // id length in the revived packet. | |
| 2594 for (size_t i = 0; i < arraysize(lengths); ++i) { | |
| 2595 // Set connection id length (for data and FEC packets). | |
| 2596 connection_id_length_ = lengths[i]; | |
| 2597 fec_packet += 2; | |
| 2598 // Don't send missing packet, but send fec packet right after it. | |
| 2599 ProcessFecPacket(fec_packet, fec_packet - 1, true, !kEntropyFlag, nullptr); | |
| 2600 // Connection id length in the revived header should be the same as | |
| 2601 // in the original data/fec packet headers. | |
| 2602 EXPECT_EQ(connection_id_length_, | |
| 2603 fec_visitor->revived_header().public_header.connection_id_length); | |
| 2604 } | |
| 2605 } | |
| 2606 | |
| 2607 TEST_P(QuicConnectionTest, ReviveMissingPacketAfterDataPacketThenFecPacket) { | |
| 2608 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 2609 | |
| 2610 ProcessFecProtectedPacket(1, false, kEntropyFlag); | |
| 2611 // Don't send missing packet 2. | |
| 2612 ProcessFecPacket(3, 1, true, !kEntropyFlag, nullptr); | |
| 2613 // Entropy flag should be true, so entropy should not be 0. | |
| 2614 EXPECT_NE(0u, QuicConnectionPeer::ReceivedEntropyHash(&connection_, 2)); | |
| 2615 } | |
| 2616 | |
| 2617 TEST_P(QuicConnectionTest, ReviveMissingPacketAfterDataPacketsThenFecPacket) { | |
| 2618 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 2619 | |
| 2620 ProcessFecProtectedPacket(1, false, !kEntropyFlag); | |
| 2621 // Don't send missing packet 2. | |
| 2622 ProcessFecProtectedPacket(3, false, !kEntropyFlag); | |
| 2623 ProcessFecPacket(4, 1, true, kEntropyFlag, nullptr); | |
| 2624 // Ensure QUIC no longer revives entropy for lost packets. | |
| 2625 EXPECT_EQ(0u, QuicConnectionPeer::ReceivedEntropyHash(&connection_, 2)); | |
| 2626 EXPECT_NE(0u, QuicConnectionPeer::ReceivedEntropyHash(&connection_, 4)); | |
| 2627 } | |
| 2628 | |
| 2629 TEST_P(QuicConnectionTest, ReviveMissingPacketAfterDataPacket) { | |
| 2630 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 2631 | |
| 2632 // Don't send missing packet 1. | |
| 2633 ProcessFecPacket(3, 1, false, !kEntropyFlag, nullptr); | |
| 2634 // Out of order. | |
| 2635 ProcessFecProtectedPacket(2, true, !kEntropyFlag); | |
| 2636 // Entropy flag should be false, so entropy should be 0. | |
| 2637 EXPECT_EQ(0u, QuicConnectionPeer::ReceivedEntropyHash(&connection_, 2)); | |
| 2638 } | |
| 2639 | |
| 2640 TEST_P(QuicConnectionTest, ReviveMissingPacketAfterDataPackets) { | |
| 2641 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 2642 | |
| 2643 ProcessFecProtectedPacket(1, false, !kEntropyFlag); | |
| 2644 // Don't send missing packet 2. | |
| 2645 ProcessFecPacket(6, 1, false, kEntropyFlag, nullptr); | |
| 2646 ProcessFecProtectedPacket(3, false, kEntropyFlag); | |
| 2647 ProcessFecProtectedPacket(4, false, kEntropyFlag); | |
| 2648 ProcessFecProtectedPacket(5, true, !kEntropyFlag); | |
| 2649 // Ensure entropy is not revived for the missing packet. | |
| 2650 EXPECT_EQ(0u, QuicConnectionPeer::ReceivedEntropyHash(&connection_, 2)); | |
| 2651 EXPECT_NE(0u, QuicConnectionPeer::ReceivedEntropyHash(&connection_, 3)); | |
| 2652 } | |
| 2653 | |
| 2654 TEST_P(QuicConnectionTest, TLP) { | |
| 2655 QuicSentPacketManagerPeer::SetMaxTailLossProbes(manager_, 1); | |
| 2656 | |
| 2657 SendStreamDataToPeer(3, "foo", 0, !kFin, nullptr); | |
| 2658 EXPECT_EQ(1u, stop_waiting()->least_unacked); | |
| 2659 QuicTime retransmission_time = | |
| 2660 connection_.GetRetransmissionAlarm()->deadline(); | |
| 2661 EXPECT_NE(QuicTime::Zero(), retransmission_time); | |
| 2662 | |
| 2663 EXPECT_EQ(1u, writer_->header().packet_sequence_number); | |
| 2664 // Simulate the retransmission alarm firing and sending a tlp, | |
| 2665 // so send algorithm's OnRetransmissionTimeout is not called. | |
| 2666 clock_.AdvanceTime(retransmission_time.Subtract(clock_.Now())); | |
| 2667 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, 2u, _, _)); | |
| 2668 connection_.GetRetransmissionAlarm()->Fire(); | |
| 2669 EXPECT_EQ(2u, writer_->header().packet_sequence_number); | |
| 2670 // We do not raise the high water mark yet. | |
| 2671 EXPECT_EQ(1u, stop_waiting()->least_unacked); | |
| 2672 } | |
| 2673 | |
| 2674 TEST_P(QuicConnectionTest, RTO) { | |
| 2675 QuicTime default_retransmission_time = clock_.ApproximateNow().Add( | |
| 2676 DefaultRetransmissionTime()); | |
| 2677 SendStreamDataToPeer(3, "foo", 0, !kFin, nullptr); | |
| 2678 EXPECT_EQ(1u, stop_waiting()->least_unacked); | |
| 2679 | |
| 2680 EXPECT_EQ(1u, writer_->header().packet_sequence_number); | |
| 2681 EXPECT_EQ(default_retransmission_time, | |
| 2682 connection_.GetRetransmissionAlarm()->deadline()); | |
| 2683 // Simulate the retransmission alarm firing. | |
| 2684 clock_.AdvanceTime(DefaultRetransmissionTime()); | |
| 2685 if (!FLAGS_quic_use_new_rto) { | |
| 2686 EXPECT_CALL(*send_algorithm_, OnRetransmissionTimeout(true)); | |
| 2687 } | |
| 2688 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, 2u, _, _)); | |
| 2689 connection_.GetRetransmissionAlarm()->Fire(); | |
| 2690 EXPECT_EQ(2u, writer_->header().packet_sequence_number); | |
| 2691 // We do not raise the high water mark yet. | |
| 2692 EXPECT_EQ(1u, stop_waiting()->least_unacked); | |
| 2693 } | |
| 2694 | |
| 2695 TEST_P(QuicConnectionTest, RTOWithSameEncryptionLevel) { | |
| 2696 QuicTime default_retransmission_time = clock_.ApproximateNow().Add( | |
| 2697 DefaultRetransmissionTime()); | |
| 2698 use_tagging_decrypter(); | |
| 2699 | |
| 2700 // A TaggingEncrypter puts kTagSize copies of the given byte (0x01 here) at | |
| 2701 // the end of the packet. We can test this to check which encrypter was used. | |
| 2702 connection_.SetEncrypter(ENCRYPTION_NONE, new TaggingEncrypter(0x01)); | |
| 2703 SendStreamDataToPeer(3, "foo", 0, !kFin, nullptr); | |
| 2704 EXPECT_EQ(0x01010101u, writer_->final_bytes_of_last_packet()); | |
| 2705 | |
| 2706 connection_.SetEncrypter(ENCRYPTION_INITIAL, new TaggingEncrypter(0x02)); | |
| 2707 connection_.SetDefaultEncryptionLevel(ENCRYPTION_INITIAL); | |
| 2708 SendStreamDataToPeer(3, "foo", 0, !kFin, nullptr); | |
| 2709 EXPECT_EQ(0x02020202u, writer_->final_bytes_of_last_packet()); | |
| 2710 | |
| 2711 EXPECT_EQ(default_retransmission_time, | |
| 2712 connection_.GetRetransmissionAlarm()->deadline()); | |
| 2713 { | |
| 2714 InSequence s; | |
| 2715 if (!FLAGS_quic_use_new_rto) { | |
| 2716 EXPECT_CALL(*send_algorithm_, OnRetransmissionTimeout(true)); | |
| 2717 } | |
| 2718 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, 3, _, _)); | |
| 2719 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, 4, _, _)); | |
| 2720 } | |
| 2721 | |
| 2722 // Simulate the retransmission alarm firing. | |
| 2723 clock_.AdvanceTime(DefaultRetransmissionTime()); | |
| 2724 connection_.GetRetransmissionAlarm()->Fire(); | |
| 2725 | |
| 2726 // Packet should have been sent with ENCRYPTION_NONE. | |
| 2727 EXPECT_EQ(0x01010101u, writer_->final_bytes_of_previous_packet()); | |
| 2728 | |
| 2729 // Packet should have been sent with ENCRYPTION_INITIAL. | |
| 2730 EXPECT_EQ(0x02020202u, writer_->final_bytes_of_last_packet()); | |
| 2731 } | |
| 2732 | |
| 2733 TEST_P(QuicConnectionTest, SendHandshakeMessages) { | |
| 2734 use_tagging_decrypter(); | |
| 2735 // A TaggingEncrypter puts kTagSize copies of the given byte (0x01 here) at | |
| 2736 // the end of the packet. We can test this to check which encrypter was used. | |
| 2737 connection_.SetEncrypter(ENCRYPTION_NONE, new TaggingEncrypter(0x01)); | |
| 2738 | |
| 2739 // Attempt to send a handshake message and have the socket block. | |
| 2740 EXPECT_CALL(*send_algorithm_, | |
| 2741 TimeUntilSend(_, _, _)).WillRepeatedly( | |
| 2742 testing::Return(QuicTime::Delta::Zero())); | |
| 2743 BlockOnNextWrite(); | |
| 2744 connection_.SendStreamDataWithString(1, "foo", 0, !kFin, nullptr); | |
| 2745 // The packet should be serialized, but not queued. | |
| 2746 EXPECT_EQ(1u, connection_.NumQueuedPackets()); | |
| 2747 | |
| 2748 // Switch to the new encrypter. | |
| 2749 connection_.SetEncrypter(ENCRYPTION_INITIAL, new TaggingEncrypter(0x02)); | |
| 2750 connection_.SetDefaultEncryptionLevel(ENCRYPTION_INITIAL); | |
| 2751 | |
| 2752 // Now become writeable and flush the packets. | |
| 2753 writer_->SetWritable(); | |
| 2754 EXPECT_CALL(visitor_, OnCanWrite()); | |
| 2755 connection_.OnCanWrite(); | |
| 2756 EXPECT_EQ(0u, connection_.NumQueuedPackets()); | |
| 2757 | |
| 2758 // Verify that the handshake packet went out at the null encryption. | |
| 2759 EXPECT_EQ(0x01010101u, writer_->final_bytes_of_last_packet()); | |
| 2760 } | |
| 2761 | |
| 2762 TEST_P(QuicConnectionTest, | |
| 2763 DropRetransmitsForNullEncryptedPacketAfterForwardSecure) { | |
| 2764 use_tagging_decrypter(); | |
| 2765 connection_.SetEncrypter(ENCRYPTION_NONE, new TaggingEncrypter(0x01)); | |
| 2766 QuicPacketSequenceNumber sequence_number; | |
| 2767 SendStreamDataToPeer(3, "foo", 0, !kFin, &sequence_number); | |
| 2768 | |
| 2769 // Simulate the retransmission alarm firing and the socket blocking. | |
| 2770 BlockOnNextWrite(); | |
| 2771 if (!FLAGS_quic_use_new_rto) { | |
| 2772 EXPECT_CALL(*send_algorithm_, OnRetransmissionTimeout(true)); | |
| 2773 } | |
| 2774 clock_.AdvanceTime(DefaultRetransmissionTime()); | |
| 2775 connection_.GetRetransmissionAlarm()->Fire(); | |
| 2776 | |
| 2777 // Go forward secure. | |
| 2778 connection_.SetEncrypter(ENCRYPTION_FORWARD_SECURE, | |
| 2779 new TaggingEncrypter(0x02)); | |
| 2780 connection_.SetDefaultEncryptionLevel(ENCRYPTION_FORWARD_SECURE); | |
| 2781 connection_.NeuterUnencryptedPackets(); | |
| 2782 | |
| 2783 EXPECT_EQ(QuicTime::Zero(), | |
| 2784 connection_.GetRetransmissionAlarm()->deadline()); | |
| 2785 // Unblock the socket and ensure that no packets are sent. | |
| 2786 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)).Times(0); | |
| 2787 writer_->SetWritable(); | |
| 2788 connection_.OnCanWrite(); | |
| 2789 } | |
| 2790 | |
| 2791 TEST_P(QuicConnectionTest, RetransmitPacketsWithInitialEncryption) { | |
| 2792 use_tagging_decrypter(); | |
| 2793 connection_.SetEncrypter(ENCRYPTION_NONE, new TaggingEncrypter(0x01)); | |
| 2794 connection_.SetDefaultEncryptionLevel(ENCRYPTION_NONE); | |
| 2795 | |
| 2796 SendStreamDataToPeer(1, "foo", 0, !kFin, nullptr); | |
| 2797 | |
| 2798 connection_.SetEncrypter(ENCRYPTION_INITIAL, new TaggingEncrypter(0x02)); | |
| 2799 connection_.SetDefaultEncryptionLevel(ENCRYPTION_INITIAL); | |
| 2800 | |
| 2801 SendStreamDataToPeer(2, "bar", 0, !kFin, nullptr); | |
| 2802 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)).Times(1); | |
| 2803 | |
| 2804 connection_.RetransmitUnackedPackets(ALL_INITIAL_RETRANSMISSION); | |
| 2805 } | |
| 2806 | |
| 2807 TEST_P(QuicConnectionTest, DelayForwardSecureEncryptionUntilClientIsReady) { | |
| 2808 // A TaggingEncrypter puts kTagSize copies of the given byte (0x02 here) at | |
| 2809 // the end of the packet. We can test this to check which encrypter was used. | |
| 2810 use_tagging_decrypter(); | |
| 2811 connection_.SetEncrypter(ENCRYPTION_INITIAL, new TaggingEncrypter(0x02)); | |
| 2812 connection_.SetDefaultEncryptionLevel(ENCRYPTION_INITIAL); | |
| 2813 SendAckPacketToPeer(); | |
| 2814 EXPECT_EQ(0x02020202u, writer_->final_bytes_of_last_packet()); | |
| 2815 | |
| 2816 // Set a forward-secure encrypter but do not make it the default, and verify | |
| 2817 // that it is not yet used. | |
| 2818 connection_.SetEncrypter(ENCRYPTION_FORWARD_SECURE, | |
| 2819 new TaggingEncrypter(0x03)); | |
| 2820 SendAckPacketToPeer(); | |
| 2821 EXPECT_EQ(0x02020202u, writer_->final_bytes_of_last_packet()); | |
| 2822 | |
| 2823 // Now simulate receipt of a forward-secure packet and verify that the | |
| 2824 // forward-secure encrypter is now used. | |
| 2825 connection_.OnDecryptedPacket(ENCRYPTION_FORWARD_SECURE); | |
| 2826 SendAckPacketToPeer(); | |
| 2827 EXPECT_EQ(0x03030303u, writer_->final_bytes_of_last_packet()); | |
| 2828 } | |
| 2829 | |
| 2830 TEST_P(QuicConnectionTest, DelayForwardSecureEncryptionUntilManyPacketSent) { | |
| 2831 // Set a congestion window of 10 packets. | |
| 2832 QuicPacketCount congestion_window = 10; | |
| 2833 EXPECT_CALL(*send_algorithm_, GetCongestionWindow()).WillRepeatedly( | |
| 2834 Return(congestion_window * kDefaultMaxPacketSize)); | |
| 2835 | |
| 2836 // A TaggingEncrypter puts kTagSize copies of the given byte (0x02 here) at | |
| 2837 // the end of the packet. We can test this to check which encrypter was used. | |
| 2838 use_tagging_decrypter(); | |
| 2839 connection_.SetEncrypter(ENCRYPTION_INITIAL, new TaggingEncrypter(0x02)); | |
| 2840 connection_.SetDefaultEncryptionLevel(ENCRYPTION_INITIAL); | |
| 2841 SendAckPacketToPeer(); | |
| 2842 EXPECT_EQ(0x02020202u, writer_->final_bytes_of_last_packet()); | |
| 2843 | |
| 2844 // Set a forward-secure encrypter but do not make it the default, and | |
| 2845 // verify that it is not yet used. | |
| 2846 connection_.SetEncrypter(ENCRYPTION_FORWARD_SECURE, | |
| 2847 new TaggingEncrypter(0x03)); | |
| 2848 SendAckPacketToPeer(); | |
| 2849 EXPECT_EQ(0x02020202u, writer_->final_bytes_of_last_packet()); | |
| 2850 | |
| 2851 // Now send a packet "Far enough" after the encrypter was set and verify that | |
| 2852 // the forward-secure encrypter is now used. | |
| 2853 for (uint64 i = 0; i < 3 * congestion_window - 1; ++i) { | |
| 2854 EXPECT_EQ(0x02020202u, writer_->final_bytes_of_last_packet()); | |
| 2855 SendAckPacketToPeer(); | |
| 2856 } | |
| 2857 EXPECT_EQ(0x03030303u, writer_->final_bytes_of_last_packet()); | |
| 2858 } | |
| 2859 | |
| 2860 TEST_P(QuicConnectionTest, BufferNonDecryptablePackets) { | |
| 2861 // SetFromConfig is always called after construction from InitializeSession. | |
| 2862 EXPECT_CALL(*send_algorithm_, SetFromConfig(_, _, _)); | |
| 2863 QuicConfig config; | |
| 2864 connection_.SetFromConfig(config); | |
| 2865 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 2866 use_tagging_decrypter(); | |
| 2867 | |
| 2868 const uint8 tag = 0x07; | |
| 2869 framer_.SetEncrypter(ENCRYPTION_INITIAL, new TaggingEncrypter(tag)); | |
| 2870 | |
| 2871 // Process an encrypted packet which can not yet be decrypted which should | |
| 2872 // result in the packet being buffered. | |
| 2873 ProcessDataPacketAtLevel(1, 0, kEntropyFlag, ENCRYPTION_INITIAL); | |
| 2874 | |
| 2875 // Transition to the new encryption state and process another encrypted packet | |
| 2876 // which should result in the original packet being processed. | |
| 2877 connection_.SetDecrypter(new StrictTaggingDecrypter(tag), | |
| 2878 ENCRYPTION_INITIAL); | |
| 2879 connection_.SetDefaultEncryptionLevel(ENCRYPTION_INITIAL); | |
| 2880 connection_.SetEncrypter(ENCRYPTION_INITIAL, new TaggingEncrypter(tag)); | |
| 2881 EXPECT_CALL(visitor_, OnStreamFrames(_)).Times(2); | |
| 2882 ProcessDataPacketAtLevel(2, 0, kEntropyFlag, ENCRYPTION_INITIAL); | |
| 2883 | |
| 2884 // Finally, process a third packet and note that we do not reprocess the | |
| 2885 // buffered packet. | |
| 2886 EXPECT_CALL(visitor_, OnStreamFrames(_)).Times(1); | |
| 2887 ProcessDataPacketAtLevel(3, 0, kEntropyFlag, ENCRYPTION_INITIAL); | |
| 2888 } | |
| 2889 | |
| 2890 TEST_P(QuicConnectionTest, Buffer100NonDecryptablePackets) { | |
| 2891 // SetFromConfig is always called after construction from InitializeSession. | |
| 2892 EXPECT_CALL(*send_algorithm_, SetFromConfig(_, _, _)); | |
| 2893 QuicConfig config; | |
| 2894 config.set_max_undecryptable_packets(100); | |
| 2895 connection_.SetFromConfig(config); | |
| 2896 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 2897 use_tagging_decrypter(); | |
| 2898 | |
| 2899 const uint8 tag = 0x07; | |
| 2900 framer_.SetEncrypter(ENCRYPTION_INITIAL, new TaggingEncrypter(tag)); | |
| 2901 | |
| 2902 // Process an encrypted packet which can not yet be decrypted which should | |
| 2903 // result in the packet being buffered. | |
| 2904 for (QuicPacketSequenceNumber i = 1; i <= 100; ++i) { | |
| 2905 ProcessDataPacketAtLevel(i, 0, kEntropyFlag, ENCRYPTION_INITIAL); | |
| 2906 } | |
| 2907 | |
| 2908 // Transition to the new encryption state and process another encrypted packet | |
| 2909 // which should result in the original packets being processed. | |
| 2910 connection_.SetDecrypter(new StrictTaggingDecrypter(tag), ENCRYPTION_INITIAL); | |
| 2911 connection_.SetDefaultEncryptionLevel(ENCRYPTION_INITIAL); | |
| 2912 connection_.SetEncrypter(ENCRYPTION_INITIAL, new TaggingEncrypter(tag)); | |
| 2913 EXPECT_CALL(visitor_, OnStreamFrames(_)).Times(101); | |
| 2914 ProcessDataPacketAtLevel(101, 0, kEntropyFlag, ENCRYPTION_INITIAL); | |
| 2915 | |
| 2916 // Finally, process a third packet and note that we do not reprocess the | |
| 2917 // buffered packet. | |
| 2918 EXPECT_CALL(visitor_, OnStreamFrames(_)).Times(1); | |
| 2919 ProcessDataPacketAtLevel(102, 0, kEntropyFlag, ENCRYPTION_INITIAL); | |
| 2920 } | |
| 2921 | |
| 2922 TEST_P(QuicConnectionTest, TestRetransmitOrder) { | |
| 2923 QuicByteCount first_packet_size; | |
| 2924 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)).WillOnce( | |
| 2925 DoAll(SaveArg<3>(&first_packet_size), Return(true))); | |
| 2926 | |
| 2927 connection_.SendStreamDataWithString(3, "first_packet", 0, !kFin, nullptr); | |
| 2928 QuicByteCount second_packet_size; | |
| 2929 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)).WillOnce( | |
| 2930 DoAll(SaveArg<3>(&second_packet_size), Return(true))); | |
| 2931 connection_.SendStreamDataWithString(3, "second_packet", 12, !kFin, nullptr); | |
| 2932 EXPECT_NE(first_packet_size, second_packet_size); | |
| 2933 // Advance the clock by huge time to make sure packets will be retransmitted. | |
| 2934 clock_.AdvanceTime(QuicTime::Delta::FromSeconds(10)); | |
| 2935 if (!FLAGS_quic_use_new_rto) { | |
| 2936 EXPECT_CALL(*send_algorithm_, OnRetransmissionTimeout(true)); | |
| 2937 } | |
| 2938 { | |
| 2939 InSequence s; | |
| 2940 EXPECT_CALL(*send_algorithm_, | |
| 2941 OnPacketSent(_, _, _, first_packet_size, _)); | |
| 2942 EXPECT_CALL(*send_algorithm_, | |
| 2943 OnPacketSent(_, _, _, second_packet_size, _)); | |
| 2944 } | |
| 2945 connection_.GetRetransmissionAlarm()->Fire(); | |
| 2946 | |
| 2947 // Advance again and expect the packets to be sent again in the same order. | |
| 2948 clock_.AdvanceTime(QuicTime::Delta::FromSeconds(20)); | |
| 2949 if (!FLAGS_quic_use_new_rto) { | |
| 2950 EXPECT_CALL(*send_algorithm_, OnRetransmissionTimeout(true)); | |
| 2951 } | |
| 2952 { | |
| 2953 InSequence s; | |
| 2954 EXPECT_CALL(*send_algorithm_, | |
| 2955 OnPacketSent(_, _, _, first_packet_size, _)); | |
| 2956 EXPECT_CALL(*send_algorithm_, | |
| 2957 OnPacketSent(_, _, _, second_packet_size, _)); | |
| 2958 } | |
| 2959 connection_.GetRetransmissionAlarm()->Fire(); | |
| 2960 } | |
| 2961 | |
| 2962 TEST_P(QuicConnectionTest, SetRTOAfterWritingToSocket) { | |
| 2963 BlockOnNextWrite(); | |
| 2964 connection_.SendStreamDataWithString(1, "foo", 0, !kFin, nullptr); | |
| 2965 // Make sure that RTO is not started when the packet is queued. | |
| 2966 EXPECT_FALSE(connection_.GetRetransmissionAlarm()->IsSet()); | |
| 2967 | |
| 2968 // Test that RTO is started once we write to the socket. | |
| 2969 writer_->SetWritable(); | |
| 2970 connection_.OnCanWrite(); | |
| 2971 EXPECT_TRUE(connection_.GetRetransmissionAlarm()->IsSet()); | |
| 2972 } | |
| 2973 | |
| 2974 TEST_P(QuicConnectionTest, DelayRTOWithAckReceipt) { | |
| 2975 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 2976 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)) | |
| 2977 .Times(2); | |
| 2978 connection_.SendStreamDataWithString(2, "foo", 0, !kFin, nullptr); | |
| 2979 connection_.SendStreamDataWithString(3, "bar", 0, !kFin, nullptr); | |
| 2980 QuicAlarm* retransmission_alarm = connection_.GetRetransmissionAlarm(); | |
| 2981 EXPECT_TRUE(retransmission_alarm->IsSet()); | |
| 2982 EXPECT_EQ(clock_.Now().Add(DefaultRetransmissionTime()), | |
| 2983 retransmission_alarm->deadline()); | |
| 2984 | |
| 2985 // Advance the time right before the RTO, then receive an ack for the first | |
| 2986 // packet to delay the RTO. | |
| 2987 clock_.AdvanceTime(DefaultRetransmissionTime()); | |
| 2988 EXPECT_CALL(*send_algorithm_, OnCongestionEvent(true, _, _, _)); | |
| 2989 QuicAckFrame ack = InitAckFrame(1); | |
| 2990 ProcessAckPacket(&ack); | |
| 2991 EXPECT_TRUE(retransmission_alarm->IsSet()); | |
| 2992 EXPECT_GT(retransmission_alarm->deadline(), clock_.Now()); | |
| 2993 | |
| 2994 // Move forward past the original RTO and ensure the RTO is still pending. | |
| 2995 clock_.AdvanceTime(DefaultRetransmissionTime().Multiply(2)); | |
| 2996 | |
| 2997 // Ensure the second packet gets retransmitted when it finally fires. | |
| 2998 EXPECT_TRUE(retransmission_alarm->IsSet()); | |
| 2999 EXPECT_LT(retransmission_alarm->deadline(), clock_.ApproximateNow()); | |
| 3000 if (!FLAGS_quic_use_new_rto) { | |
| 3001 EXPECT_CALL(*send_algorithm_, OnRetransmissionTimeout(true)); | |
| 3002 } | |
| 3003 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)); | |
| 3004 // Manually cancel the alarm to simulate a real test. | |
| 3005 connection_.GetRetransmissionAlarm()->Fire(); | |
| 3006 | |
| 3007 // The new retransmitted sequence number should set the RTO to a larger value | |
| 3008 // than previously. | |
| 3009 EXPECT_TRUE(retransmission_alarm->IsSet()); | |
| 3010 QuicTime next_rto_time = retransmission_alarm->deadline(); | |
| 3011 QuicTime expected_rto_time = | |
| 3012 connection_.sent_packet_manager().GetRetransmissionTime(); | |
| 3013 EXPECT_EQ(next_rto_time, expected_rto_time); | |
| 3014 } | |
| 3015 | |
| 3016 TEST_P(QuicConnectionTest, TestQueued) { | |
| 3017 EXPECT_EQ(0u, connection_.NumQueuedPackets()); | |
| 3018 BlockOnNextWrite(); | |
| 3019 connection_.SendStreamDataWithString(1, "foo", 0, !kFin, nullptr); | |
| 3020 EXPECT_EQ(1u, connection_.NumQueuedPackets()); | |
| 3021 | |
| 3022 // Unblock the writes and actually send. | |
| 3023 writer_->SetWritable(); | |
| 3024 connection_.OnCanWrite(); | |
| 3025 EXPECT_EQ(0u, connection_.NumQueuedPackets()); | |
| 3026 } | |
| 3027 | |
| 3028 TEST_P(QuicConnectionTest, CloseFecGroup) { | |
| 3029 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 3030 // Don't send missing packet 1. | |
| 3031 // Don't send missing packet 2. | |
| 3032 ProcessFecProtectedPacket(3, false, !kEntropyFlag); | |
| 3033 // Don't send missing FEC packet 3. | |
| 3034 ASSERT_EQ(1u, connection_.NumFecGroups()); | |
| 3035 | |
| 3036 // Now send non-fec protected ack packet and close the group. | |
| 3037 peer_creator_.set_sequence_number(4); | |
| 3038 QuicStopWaitingFrame frame = InitStopWaitingFrame(5); | |
| 3039 ProcessStopWaitingPacket(&frame); | |
| 3040 ASSERT_EQ(0u, connection_.NumFecGroups()); | |
| 3041 } | |
| 3042 | |
| 3043 TEST_P(QuicConnectionTest, InitialTimeout) { | |
| 3044 EXPECT_TRUE(connection_.connected()); | |
| 3045 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)).Times(AnyNumber()); | |
| 3046 EXPECT_FALSE(connection_.GetTimeoutAlarm()->IsSet()); | |
| 3047 | |
| 3048 // SetFromConfig sets the initial timeouts before negotiation. | |
| 3049 EXPECT_CALL(*send_algorithm_, SetFromConfig(_, _, _)); | |
| 3050 QuicConfig config; | |
| 3051 connection_.SetFromConfig(config); | |
| 3052 // Subtract a second from the idle timeout on the client side. | |
| 3053 QuicTime default_timeout = clock_.ApproximateNow().Add( | |
| 3054 QuicTime::Delta::FromSeconds(kInitialIdleTimeoutSecs - 1)); | |
| 3055 EXPECT_EQ(default_timeout, connection_.GetTimeoutAlarm()->deadline()); | |
| 3056 | |
| 3057 EXPECT_CALL(visitor_, OnConnectionClosed(QUIC_CONNECTION_TIMED_OUT, false)); | |
| 3058 // Simulate the timeout alarm firing. | |
| 3059 clock_.AdvanceTime( | |
| 3060 QuicTime::Delta::FromSeconds(kInitialIdleTimeoutSecs - 1)); | |
| 3061 connection_.GetTimeoutAlarm()->Fire(); | |
| 3062 | |
| 3063 EXPECT_FALSE(connection_.GetTimeoutAlarm()->IsSet()); | |
| 3064 EXPECT_FALSE(connection_.connected()); | |
| 3065 | |
| 3066 EXPECT_FALSE(connection_.GetAckAlarm()->IsSet()); | |
| 3067 EXPECT_FALSE(connection_.GetPingAlarm()->IsSet()); | |
| 3068 EXPECT_FALSE(connection_.GetFecAlarm()->IsSet()); | |
| 3069 EXPECT_FALSE(connection_.GetResumeWritesAlarm()->IsSet()); | |
| 3070 EXPECT_FALSE(connection_.GetRetransmissionAlarm()->IsSet()); | |
| 3071 EXPECT_FALSE(connection_.GetSendAlarm()->IsSet()); | |
| 3072 } | |
| 3073 | |
| 3074 TEST_P(QuicConnectionTest, OverallTimeout) { | |
| 3075 // Use a shorter overall connection timeout than idle timeout for this test. | |
| 3076 const QuicTime::Delta timeout = QuicTime::Delta::FromSeconds(5); | |
| 3077 connection_.SetNetworkTimeouts(timeout, timeout); | |
| 3078 EXPECT_TRUE(connection_.connected()); | |
| 3079 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)).Times(AnyNumber()); | |
| 3080 | |
| 3081 QuicTime overall_timeout = clock_.ApproximateNow().Add(timeout).Subtract( | |
| 3082 QuicTime::Delta::FromSeconds(1)); | |
| 3083 EXPECT_EQ(overall_timeout, connection_.GetTimeoutAlarm()->deadline()); | |
| 3084 EXPECT_TRUE(connection_.connected()); | |
| 3085 | |
| 3086 // Send and ack new data 3 seconds later to lengthen the idle timeout. | |
| 3087 SendStreamDataToPeer(1, "GET /", 0, kFin, nullptr); | |
| 3088 clock_.AdvanceTime(QuicTime::Delta::FromSeconds(3)); | |
| 3089 QuicAckFrame frame = InitAckFrame(1); | |
| 3090 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 3091 EXPECT_CALL(*send_algorithm_, OnCongestionEvent(true, _, _, _)); | |
| 3092 ProcessAckPacket(&frame); | |
| 3093 | |
| 3094 // Fire early to verify it wouldn't timeout yet. | |
| 3095 connection_.GetTimeoutAlarm()->Fire(); | |
| 3096 EXPECT_TRUE(connection_.GetTimeoutAlarm()->IsSet()); | |
| 3097 EXPECT_TRUE(connection_.connected()); | |
| 3098 | |
| 3099 clock_.AdvanceTime(timeout.Subtract(QuicTime::Delta::FromSeconds(2))); | |
| 3100 | |
| 3101 EXPECT_CALL(visitor_, | |
| 3102 OnConnectionClosed(QUIC_CONNECTION_OVERALL_TIMED_OUT, false)); | |
| 3103 // Simulate the timeout alarm firing. | |
| 3104 connection_.GetTimeoutAlarm()->Fire(); | |
| 3105 | |
| 3106 EXPECT_FALSE(connection_.GetTimeoutAlarm()->IsSet()); | |
| 3107 EXPECT_FALSE(connection_.connected()); | |
| 3108 | |
| 3109 EXPECT_FALSE(connection_.GetAckAlarm()->IsSet()); | |
| 3110 EXPECT_FALSE(connection_.GetPingAlarm()->IsSet()); | |
| 3111 EXPECT_FALSE(connection_.GetFecAlarm()->IsSet()); | |
| 3112 EXPECT_FALSE(connection_.GetResumeWritesAlarm()->IsSet()); | |
| 3113 EXPECT_FALSE(connection_.GetRetransmissionAlarm()->IsSet()); | |
| 3114 EXPECT_FALSE(connection_.GetSendAlarm()->IsSet()); | |
| 3115 } | |
| 3116 | |
| 3117 TEST_P(QuicConnectionTest, PingAfterSend) { | |
| 3118 EXPECT_TRUE(connection_.connected()); | |
| 3119 EXPECT_CALL(visitor_, HasOpenDataStreams()).WillRepeatedly(Return(true)); | |
| 3120 EXPECT_FALSE(connection_.GetPingAlarm()->IsSet()); | |
| 3121 | |
| 3122 // Advance to 5ms, and send a packet to the peer, which will set | |
| 3123 // the ping alarm. | |
| 3124 clock_.AdvanceTime(QuicTime::Delta::FromMilliseconds(5)); | |
| 3125 EXPECT_FALSE(connection_.GetRetransmissionAlarm()->IsSet()); | |
| 3126 SendStreamDataToPeer(1, "GET /", 0, kFin, nullptr); | |
| 3127 EXPECT_TRUE(connection_.GetPingAlarm()->IsSet()); | |
| 3128 EXPECT_EQ(clock_.ApproximateNow().Add(QuicTime::Delta::FromSeconds(15)), | |
| 3129 connection_.GetPingAlarm()->deadline()); | |
| 3130 | |
| 3131 // Now recevie and ACK of the previous packet, which will move the | |
| 3132 // ping alarm forward. | |
| 3133 clock_.AdvanceTime(QuicTime::Delta::FromMilliseconds(5)); | |
| 3134 QuicAckFrame frame = InitAckFrame(1); | |
| 3135 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 3136 EXPECT_CALL(*send_algorithm_, OnCongestionEvent(true, _, _, _)); | |
| 3137 ProcessAckPacket(&frame); | |
| 3138 EXPECT_TRUE(connection_.GetPingAlarm()->IsSet()); | |
| 3139 // The ping timer is set slightly less than 15 seconds in the future, because | |
| 3140 // of the 1s ping timer alarm granularity. | |
| 3141 EXPECT_EQ(clock_.ApproximateNow().Add(QuicTime::Delta::FromSeconds(15)) | |
| 3142 .Subtract(QuicTime::Delta::FromMilliseconds(5)), | |
| 3143 connection_.GetPingAlarm()->deadline()); | |
| 3144 | |
| 3145 writer_->Reset(); | |
| 3146 clock_.AdvanceTime(QuicTime::Delta::FromSeconds(15)); | |
| 3147 connection_.GetPingAlarm()->Fire(); | |
| 3148 EXPECT_EQ(1u, writer_->frame_count()); | |
| 3149 ASSERT_EQ(1u, writer_->ping_frames().size()); | |
| 3150 writer_->Reset(); | |
| 3151 | |
| 3152 EXPECT_CALL(visitor_, HasOpenDataStreams()).WillRepeatedly(Return(false)); | |
| 3153 clock_.AdvanceTime(QuicTime::Delta::FromMilliseconds(5)); | |
| 3154 SendAckPacketToPeer(); | |
| 3155 | |
| 3156 EXPECT_FALSE(connection_.GetPingAlarm()->IsSet()); | |
| 3157 } | |
| 3158 | |
| 3159 TEST_P(QuicConnectionTest, TimeoutAfterSend) { | |
| 3160 EXPECT_TRUE(connection_.connected()); | |
| 3161 EXPECT_CALL(*send_algorithm_, SetFromConfig(_, _, _)); | |
| 3162 QuicConfig config; | |
| 3163 connection_.SetFromConfig(config); | |
| 3164 EXPECT_FALSE(QuicConnectionPeer::IsSilentCloseEnabled(&connection_)); | |
| 3165 | |
| 3166 const QuicTime::Delta initial_idle_timeout = | |
| 3167 QuicTime::Delta::FromSeconds(kInitialIdleTimeoutSecs - 1); | |
| 3168 const QuicTime::Delta five_ms = QuicTime::Delta::FromMilliseconds(5); | |
| 3169 QuicTime default_timeout = clock_.ApproximateNow().Add(initial_idle_timeout); | |
| 3170 | |
| 3171 // When we send a packet, the timeout will change to 5ms + | |
| 3172 // kInitialIdleTimeoutSecs. | |
| 3173 clock_.AdvanceTime(five_ms); | |
| 3174 | |
| 3175 // Send an ack so we don't set the retransmission alarm. | |
| 3176 SendAckPacketToPeer(); | |
| 3177 EXPECT_EQ(default_timeout, connection_.GetTimeoutAlarm()->deadline()); | |
| 3178 | |
| 3179 // The original alarm will fire. We should not time out because we had a | |
| 3180 // network event at t=5ms. The alarm will reregister. | |
| 3181 clock_.AdvanceTime(initial_idle_timeout.Subtract(five_ms)); | |
| 3182 EXPECT_EQ(default_timeout, clock_.ApproximateNow()); | |
| 3183 connection_.GetTimeoutAlarm()->Fire(); | |
| 3184 EXPECT_TRUE(connection_.GetTimeoutAlarm()->IsSet()); | |
| 3185 EXPECT_TRUE(connection_.connected()); | |
| 3186 EXPECT_EQ(default_timeout.Add(five_ms), | |
| 3187 connection_.GetTimeoutAlarm()->deadline()); | |
| 3188 | |
| 3189 // This time, we should time out. | |
| 3190 EXPECT_CALL(visitor_, OnConnectionClosed(QUIC_CONNECTION_TIMED_OUT, false)); | |
| 3191 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)); | |
| 3192 clock_.AdvanceTime(five_ms); | |
| 3193 EXPECT_EQ(default_timeout.Add(five_ms), clock_.ApproximateNow()); | |
| 3194 connection_.GetTimeoutAlarm()->Fire(); | |
| 3195 EXPECT_FALSE(connection_.GetTimeoutAlarm()->IsSet()); | |
| 3196 EXPECT_FALSE(connection_.connected()); | |
| 3197 } | |
| 3198 | |
| 3199 TEST_P(QuicConnectionTest, TimeoutAfterSendSilentClose) { | |
| 3200 // Same test as above, but complete a handshake which enables silent close, | |
| 3201 // causing no connection close packet to be sent. | |
| 3202 EXPECT_TRUE(connection_.connected()); | |
| 3203 EXPECT_CALL(*send_algorithm_, SetFromConfig(_, _, _)); | |
| 3204 QuicConfig config; | |
| 3205 | |
| 3206 // Create a handshake message that also enables silent close. | |
| 3207 CryptoHandshakeMessage msg; | |
| 3208 string error_details; | |
| 3209 QuicConfig client_config; | |
| 3210 client_config.SetInitialStreamFlowControlWindowToSend( | |
| 3211 kInitialStreamFlowControlWindowForTest); | |
| 3212 client_config.SetInitialSessionFlowControlWindowToSend( | |
| 3213 kInitialSessionFlowControlWindowForTest); | |
| 3214 client_config.SetIdleConnectionStateLifetime( | |
| 3215 QuicTime::Delta::FromSeconds(kDefaultIdleTimeoutSecs), | |
| 3216 QuicTime::Delta::FromSeconds(kDefaultIdleTimeoutSecs)); | |
| 3217 client_config.ToHandshakeMessage(&msg); | |
| 3218 const QuicErrorCode error = | |
| 3219 config.ProcessPeerHello(msg, CLIENT, &error_details); | |
| 3220 EXPECT_EQ(QUIC_NO_ERROR, error); | |
| 3221 | |
| 3222 connection_.SetFromConfig(config); | |
| 3223 EXPECT_TRUE(QuicConnectionPeer::IsSilentCloseEnabled(&connection_)); | |
| 3224 | |
| 3225 const QuicTime::Delta default_idle_timeout = | |
| 3226 QuicTime::Delta::FromSeconds(kDefaultIdleTimeoutSecs - 1); | |
| 3227 const QuicTime::Delta five_ms = QuicTime::Delta::FromMilliseconds(5); | |
| 3228 QuicTime default_timeout = clock_.ApproximateNow().Add(default_idle_timeout); | |
| 3229 | |
| 3230 // When we send a packet, the timeout will change to 5ms + | |
| 3231 // kInitialIdleTimeoutSecs. | |
| 3232 clock_.AdvanceTime(five_ms); | |
| 3233 | |
| 3234 // Send an ack so we don't set the retransmission alarm. | |
| 3235 SendAckPacketToPeer(); | |
| 3236 EXPECT_EQ(default_timeout, connection_.GetTimeoutAlarm()->deadline()); | |
| 3237 | |
| 3238 // The original alarm will fire. We should not time out because we had a | |
| 3239 // network event at t=5ms. The alarm will reregister. | |
| 3240 clock_.AdvanceTime(default_idle_timeout.Subtract(five_ms)); | |
| 3241 EXPECT_EQ(default_timeout, clock_.ApproximateNow()); | |
| 3242 connection_.GetTimeoutAlarm()->Fire(); | |
| 3243 EXPECT_TRUE(connection_.GetTimeoutAlarm()->IsSet()); | |
| 3244 EXPECT_TRUE(connection_.connected()); | |
| 3245 EXPECT_EQ(default_timeout.Add(five_ms), | |
| 3246 connection_.GetTimeoutAlarm()->deadline()); | |
| 3247 | |
| 3248 // This time, we should time out. | |
| 3249 EXPECT_CALL(visitor_, OnConnectionClosed(QUIC_CONNECTION_TIMED_OUT, false)); | |
| 3250 clock_.AdvanceTime(five_ms); | |
| 3251 EXPECT_EQ(default_timeout.Add(five_ms), clock_.ApproximateNow()); | |
| 3252 connection_.GetTimeoutAlarm()->Fire(); | |
| 3253 EXPECT_FALSE(connection_.GetTimeoutAlarm()->IsSet()); | |
| 3254 EXPECT_FALSE(connection_.connected()); | |
| 3255 } | |
| 3256 | |
| 3257 TEST_P(QuicConnectionTest, SendScheduler) { | |
| 3258 // Test that if we send a packet without delay, it is not queued. | |
| 3259 QuicPacket* packet = ConstructDataPacket(1, 0, !kEntropyFlag); | |
| 3260 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)); | |
| 3261 connection_.SendPacket( | |
| 3262 ENCRYPTION_NONE, 1, packet, kTestEntropyHash, HAS_RETRANSMITTABLE_DATA); | |
| 3263 EXPECT_EQ(0u, connection_.NumQueuedPackets()); | |
| 3264 } | |
| 3265 | |
| 3266 TEST_P(QuicConnectionTest, SendSchedulerEAGAIN) { | |
| 3267 QuicPacket* packet = ConstructDataPacket(1, 0, !kEntropyFlag); | |
| 3268 BlockOnNextWrite(); | |
| 3269 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, 1, _, _)).Times(0); | |
| 3270 connection_.SendPacket( | |
| 3271 ENCRYPTION_NONE, 1, packet, kTestEntropyHash, HAS_RETRANSMITTABLE_DATA); | |
| 3272 EXPECT_EQ(1u, connection_.NumQueuedPackets()); | |
| 3273 } | |
| 3274 | |
| 3275 TEST_P(QuicConnectionTest, TestQueueLimitsOnSendStreamData) { | |
| 3276 // All packets carry version info till version is negotiated. | |
| 3277 size_t payload_length; | |
| 3278 size_t length = GetPacketLengthForOneStream( | |
| 3279 connection_.version(), kIncludeVersion, | |
| 3280 PACKET_8BYTE_CONNECTION_ID, PACKET_1BYTE_SEQUENCE_NUMBER, | |
| 3281 NOT_IN_FEC_GROUP, &payload_length); | |
| 3282 creator_->set_max_packet_length(length); | |
| 3283 | |
| 3284 // Queue the first packet. | |
| 3285 EXPECT_CALL(*send_algorithm_, | |
| 3286 TimeUntilSend(_, _, _)).WillOnce( | |
| 3287 testing::Return(QuicTime::Delta::FromMicroseconds(10))); | |
| 3288 const string payload(payload_length, 'a'); | |
| 3289 EXPECT_EQ(0u, connection_.SendStreamDataWithString(3, payload, 0, !kFin, | |
| 3290 nullptr).bytes_consumed); | |
| 3291 EXPECT_EQ(0u, connection_.NumQueuedPackets()); | |
| 3292 } | |
| 3293 | |
| 3294 TEST_P(QuicConnectionTest, LoopThroughSendingPackets) { | |
| 3295 // All packets carry version info till version is negotiated. | |
| 3296 size_t payload_length; | |
| 3297 // GetPacketLengthForOneStream() assumes a stream offset of 0 in determining | |
| 3298 // packet length. The size of the offset field in a stream frame is 0 for | |
| 3299 // offset 0, and 2 for non-zero offsets up through 16K. Increase | |
| 3300 // max_packet_length by 2 so that subsequent packets containing subsequent | |
| 3301 // stream frames with non-zero offets will fit within the packet length. | |
| 3302 size_t length = 2 + GetPacketLengthForOneStream( | |
| 3303 connection_.version(), kIncludeVersion, | |
| 3304 PACKET_8BYTE_CONNECTION_ID, PACKET_1BYTE_SEQUENCE_NUMBER, | |
| 3305 NOT_IN_FEC_GROUP, &payload_length); | |
| 3306 creator_->set_max_packet_length(length); | |
| 3307 | |
| 3308 // Queue the first packet. | |
| 3309 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)).Times(7); | |
| 3310 // The first stream frame will have 2 fewer overhead bytes than the other six. | |
| 3311 const string payload(payload_length * 7 + 2, 'a'); | |
| 3312 EXPECT_EQ(payload.size(), | |
| 3313 connection_.SendStreamDataWithString(1, payload, 0, !kFin, nullptr) | |
| 3314 .bytes_consumed); | |
| 3315 } | |
| 3316 | |
| 3317 TEST_P(QuicConnectionTest, LoopThroughSendingPacketsWithTruncation) { | |
| 3318 // Set up a larger payload than will fit in one packet. | |
| 3319 const string payload(connection_.max_packet_length(), 'a'); | |
| 3320 EXPECT_CALL(*send_algorithm_, SetFromConfig(_, _, _)).Times(AnyNumber()); | |
| 3321 | |
| 3322 // Now send some packets with no truncation. | |
| 3323 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)).Times(2); | |
| 3324 EXPECT_EQ(payload.size(), | |
| 3325 connection_.SendStreamDataWithString( | |
| 3326 3, payload, 0, !kFin, nullptr).bytes_consumed); | |
| 3327 // Track the size of the second packet here. The overhead will be the largest | |
| 3328 // we see in this test, due to the non-truncated CID. | |
| 3329 size_t non_truncated_packet_size = writer_->last_packet_size(); | |
| 3330 | |
| 3331 // Change to a 4 byte CID. | |
| 3332 QuicConfig config; | |
| 3333 QuicConfigPeer::SetReceivedBytesForConnectionId(&config, 4); | |
| 3334 connection_.SetFromConfig(config); | |
| 3335 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)).Times(2); | |
| 3336 EXPECT_EQ(payload.size(), | |
| 3337 connection_.SendStreamDataWithString( | |
| 3338 3, payload, 0, !kFin, nullptr).bytes_consumed); | |
| 3339 // Verify that we have 8 fewer bytes than in the non-truncated case. The | |
| 3340 // first packet got 4 bytes of extra payload due to the truncation, and the | |
| 3341 // headers here are also 4 byte smaller. | |
| 3342 EXPECT_EQ(non_truncated_packet_size, writer_->last_packet_size() + 8); | |
| 3343 | |
| 3344 | |
| 3345 // Change to a 1 byte CID. | |
| 3346 QuicConfigPeer::SetReceivedBytesForConnectionId(&config, 1); | |
| 3347 connection_.SetFromConfig(config); | |
| 3348 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)).Times(2); | |
| 3349 EXPECT_EQ(payload.size(), | |
| 3350 connection_.SendStreamDataWithString( | |
| 3351 3, payload, 0, !kFin, nullptr).bytes_consumed); | |
| 3352 // Just like above, we save 7 bytes on payload, and 7 on truncation. | |
| 3353 EXPECT_EQ(non_truncated_packet_size, writer_->last_packet_size() + 7 * 2); | |
| 3354 | |
| 3355 // Change to a 0 byte CID. | |
| 3356 QuicConfigPeer::SetReceivedBytesForConnectionId(&config, 0); | |
| 3357 connection_.SetFromConfig(config); | |
| 3358 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)).Times(2); | |
| 3359 EXPECT_EQ(payload.size(), | |
| 3360 connection_.SendStreamDataWithString( | |
| 3361 3, payload, 0, !kFin, nullptr).bytes_consumed); | |
| 3362 // Just like above, we save 8 bytes on payload, and 8 on truncation. | |
| 3363 EXPECT_EQ(non_truncated_packet_size, writer_->last_packet_size() + 8 * 2); | |
| 3364 } | |
| 3365 | |
| 3366 TEST_P(QuicConnectionTest, SendDelayedAck) { | |
| 3367 QuicTime ack_time = clock_.ApproximateNow().Add(DefaultDelayedAckTime()); | |
| 3368 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 3369 EXPECT_FALSE(connection_.GetAckAlarm()->IsSet()); | |
| 3370 const uint8 tag = 0x07; | |
| 3371 connection_.SetDecrypter(new StrictTaggingDecrypter(tag), | |
| 3372 ENCRYPTION_INITIAL); | |
| 3373 framer_.SetEncrypter(ENCRYPTION_INITIAL, new TaggingEncrypter(tag)); | |
| 3374 // Process a packet from the non-crypto stream. | |
| 3375 frame1_.stream_id = 3; | |
| 3376 | |
| 3377 // The same as ProcessPacket(1) except that ENCRYPTION_INITIAL is used | |
| 3378 // instead of ENCRYPTION_NONE. | |
| 3379 EXPECT_CALL(visitor_, OnStreamFrames(_)).Times(1); | |
| 3380 ProcessDataPacketAtLevel(1, 0, !kEntropyFlag, ENCRYPTION_INITIAL); | |
| 3381 | |
| 3382 // Check if delayed ack timer is running for the expected interval. | |
| 3383 EXPECT_TRUE(connection_.GetAckAlarm()->IsSet()); | |
| 3384 EXPECT_EQ(ack_time, connection_.GetAckAlarm()->deadline()); | |
| 3385 // Simulate delayed ack alarm firing. | |
| 3386 connection_.GetAckAlarm()->Fire(); | |
| 3387 // Check that ack is sent and that delayed ack alarm is reset. | |
| 3388 EXPECT_EQ(2u, writer_->frame_count()); | |
| 3389 EXPECT_FALSE(writer_->stop_waiting_frames().empty()); | |
| 3390 EXPECT_FALSE(writer_->ack_frames().empty()); | |
| 3391 EXPECT_FALSE(connection_.GetAckAlarm()->IsSet()); | |
| 3392 } | |
| 3393 | |
| 3394 TEST_P(QuicConnectionTest, SendDelayedAckOnHandshakeConfirmed) { | |
| 3395 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 3396 ProcessPacket(1); | |
| 3397 // Check that ack is sent and that delayed ack alarm is set. | |
| 3398 EXPECT_TRUE(connection_.GetAckAlarm()->IsSet()); | |
| 3399 QuicTime ack_time = clock_.ApproximateNow().Add(DefaultDelayedAckTime()); | |
| 3400 EXPECT_EQ(ack_time, connection_.GetAckAlarm()->deadline()); | |
| 3401 | |
| 3402 // Completing the handshake as the server does nothing. | |
| 3403 QuicConnectionPeer::SetIsServer(&connection_, true); | |
| 3404 connection_.OnHandshakeComplete(); | |
| 3405 EXPECT_TRUE(connection_.GetAckAlarm()->IsSet()); | |
| 3406 EXPECT_EQ(ack_time, connection_.GetAckAlarm()->deadline()); | |
| 3407 | |
| 3408 // Complete the handshake as the client decreases the delayed ack time to 0ms. | |
| 3409 QuicConnectionPeer::SetIsServer(&connection_, false); | |
| 3410 connection_.OnHandshakeComplete(); | |
| 3411 EXPECT_TRUE(connection_.GetAckAlarm()->IsSet()); | |
| 3412 EXPECT_EQ(clock_.ApproximateNow(), connection_.GetAckAlarm()->deadline()); | |
| 3413 } | |
| 3414 | |
| 3415 TEST_P(QuicConnectionTest, SendDelayedAckOnSecondPacket) { | |
| 3416 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 3417 ProcessPacket(1); | |
| 3418 ProcessPacket(2); | |
| 3419 // Check that ack is sent and that delayed ack alarm is reset. | |
| 3420 EXPECT_EQ(2u, writer_->frame_count()); | |
| 3421 EXPECT_FALSE(writer_->stop_waiting_frames().empty()); | |
| 3422 EXPECT_FALSE(writer_->ack_frames().empty()); | |
| 3423 EXPECT_FALSE(connection_.GetAckAlarm()->IsSet()); | |
| 3424 } | |
| 3425 | |
| 3426 TEST_P(QuicConnectionTest, NoAckOnOldNacks) { | |
| 3427 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 3428 // Drop one packet, triggering a sequence of acks. | |
| 3429 ProcessPacket(2); | |
| 3430 size_t frames_per_ack = 2; | |
| 3431 EXPECT_EQ(frames_per_ack, writer_->frame_count()); | |
| 3432 EXPECT_FALSE(writer_->ack_frames().empty()); | |
| 3433 writer_->Reset(); | |
| 3434 ProcessPacket(3); | |
| 3435 EXPECT_EQ(frames_per_ack, writer_->frame_count()); | |
| 3436 EXPECT_FALSE(writer_->ack_frames().empty()); | |
| 3437 writer_->Reset(); | |
| 3438 ProcessPacket(4); | |
| 3439 EXPECT_EQ(frames_per_ack, writer_->frame_count()); | |
| 3440 EXPECT_FALSE(writer_->ack_frames().empty()); | |
| 3441 writer_->Reset(); | |
| 3442 ProcessPacket(5); | |
| 3443 EXPECT_EQ(frames_per_ack, writer_->frame_count()); | |
| 3444 EXPECT_FALSE(writer_->ack_frames().empty()); | |
| 3445 writer_->Reset(); | |
| 3446 // Now only set the timer on the 6th packet, instead of sending another ack. | |
| 3447 ProcessPacket(6); | |
| 3448 EXPECT_EQ(0u, writer_->frame_count()); | |
| 3449 EXPECT_TRUE(connection_.GetAckAlarm()->IsSet()); | |
| 3450 } | |
| 3451 | |
| 3452 TEST_P(QuicConnectionTest, SendDelayedAckOnOutgoingPacket) { | |
| 3453 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 3454 ProcessPacket(1); | |
| 3455 connection_.SendStreamDataWithString(kClientDataStreamId1, "foo", 0, !kFin, | |
| 3456 nullptr); | |
| 3457 // Check that ack is bundled with outgoing data and that delayed ack | |
| 3458 // alarm is reset. | |
| 3459 EXPECT_EQ(3u, writer_->frame_count()); | |
| 3460 EXPECT_FALSE(writer_->stop_waiting_frames().empty()); | |
| 3461 EXPECT_FALSE(writer_->ack_frames().empty()); | |
| 3462 EXPECT_FALSE(connection_.GetAckAlarm()->IsSet()); | |
| 3463 } | |
| 3464 | |
| 3465 TEST_P(QuicConnectionTest, SendDelayedAckOnOutgoingCryptoPacket) { | |
| 3466 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 3467 ProcessPacket(1); | |
| 3468 connection_.SendStreamDataWithString(kCryptoStreamId, "foo", 0, !kFin, | |
| 3469 nullptr); | |
| 3470 // Check that ack is bundled with outgoing crypto data. | |
| 3471 EXPECT_EQ(3u, writer_->frame_count()); | |
| 3472 EXPECT_FALSE(writer_->ack_frames().empty()); | |
| 3473 EXPECT_FALSE(connection_.GetAckAlarm()->IsSet()); | |
| 3474 } | |
| 3475 | |
| 3476 TEST_P(QuicConnectionTest, BlockAndBufferOnFirstCHLOPacketOfTwo) { | |
| 3477 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 3478 ProcessPacket(1); | |
| 3479 BlockOnNextWrite(); | |
| 3480 writer_->set_is_write_blocked_data_buffered(true); | |
| 3481 connection_.SendStreamDataWithString(kCryptoStreamId, "foo", 0, !kFin, | |
| 3482 nullptr); | |
| 3483 EXPECT_TRUE(writer_->IsWriteBlocked()); | |
| 3484 EXPECT_FALSE(connection_.HasQueuedData()); | |
| 3485 connection_.SendStreamDataWithString(kCryptoStreamId, "bar", 3, !kFin, | |
| 3486 nullptr); | |
| 3487 EXPECT_TRUE(writer_->IsWriteBlocked()); | |
| 3488 EXPECT_TRUE(connection_.HasQueuedData()); | |
| 3489 } | |
| 3490 | |
| 3491 TEST_P(QuicConnectionTest, BundleAckForSecondCHLO) { | |
| 3492 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 3493 EXPECT_FALSE(connection_.GetAckAlarm()->IsSet()); | |
| 3494 EXPECT_CALL(visitor_, OnCanWrite()).WillOnce( | |
| 3495 IgnoreResult(InvokeWithoutArgs(&connection_, | |
| 3496 &TestConnection::SendCryptoStreamData))); | |
| 3497 // Process a packet from the crypto stream, which is frame1_'s default. | |
| 3498 // Receiving the CHLO as packet 2 first will cause the connection to | |
| 3499 // immediately send an ack, due to the packet gap. | |
| 3500 ProcessPacket(2); | |
| 3501 // Check that ack is sent and that delayed ack alarm is reset. | |
| 3502 EXPECT_EQ(3u, writer_->frame_count()); | |
| 3503 EXPECT_FALSE(writer_->stop_waiting_frames().empty()); | |
| 3504 EXPECT_EQ(1u, writer_->stream_frames().size()); | |
| 3505 EXPECT_FALSE(writer_->ack_frames().empty()); | |
| 3506 EXPECT_FALSE(connection_.GetAckAlarm()->IsSet()); | |
| 3507 } | |
| 3508 | |
| 3509 TEST_P(QuicConnectionTest, BundleAckWithDataOnIncomingAck) { | |
| 3510 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 3511 connection_.SendStreamDataWithString(kClientDataStreamId1, "foo", 0, !kFin, | |
| 3512 nullptr); | |
| 3513 connection_.SendStreamDataWithString(kClientDataStreamId1, "foo", 3, !kFin, | |
| 3514 nullptr); | |
| 3515 // Ack the second packet, which will retransmit the first packet. | |
| 3516 QuicAckFrame ack = InitAckFrame(2); | |
| 3517 NackPacket(1, &ack); | |
| 3518 SequenceNumberSet lost_packets; | |
| 3519 lost_packets.insert(1); | |
| 3520 EXPECT_CALL(*loss_algorithm_, DetectLostPackets(_, _, _, _)) | |
| 3521 .WillOnce(Return(lost_packets)); | |
| 3522 EXPECT_CALL(*send_algorithm_, OnCongestionEvent(true, _, _, _)); | |
| 3523 ProcessAckPacket(&ack); | |
| 3524 EXPECT_EQ(1u, writer_->frame_count()); | |
| 3525 EXPECT_EQ(1u, writer_->stream_frames().size()); | |
| 3526 writer_->Reset(); | |
| 3527 | |
| 3528 // Now ack the retransmission, which will both raise the high water mark | |
| 3529 // and see if there is more data to send. | |
| 3530 ack = InitAckFrame(3); | |
| 3531 NackPacket(1, &ack); | |
| 3532 EXPECT_CALL(*loss_algorithm_, DetectLostPackets(_, _, _, _)) | |
| 3533 .WillOnce(Return(SequenceNumberSet())); | |
| 3534 EXPECT_CALL(*send_algorithm_, OnCongestionEvent(true, _, _, _)); | |
| 3535 ProcessAckPacket(&ack); | |
| 3536 | |
| 3537 // Check that no packet is sent and the ack alarm isn't set. | |
| 3538 EXPECT_EQ(0u, writer_->frame_count()); | |
| 3539 EXPECT_FALSE(connection_.GetAckAlarm()->IsSet()); | |
| 3540 writer_->Reset(); | |
| 3541 | |
| 3542 // Send the same ack, but send both data and an ack together. | |
| 3543 ack = InitAckFrame(3); | |
| 3544 NackPacket(1, &ack); | |
| 3545 EXPECT_CALL(*loss_algorithm_, DetectLostPackets(_, _, _, _)) | |
| 3546 .WillOnce(Return(SequenceNumberSet())); | |
| 3547 EXPECT_CALL(visitor_, OnCanWrite()).WillOnce( | |
| 3548 IgnoreResult(InvokeWithoutArgs( | |
| 3549 &connection_, | |
| 3550 &TestConnection::EnsureWritableAndSendStreamData5))); | |
| 3551 ProcessAckPacket(&ack); | |
| 3552 | |
| 3553 // Check that ack is bundled with outgoing data and the delayed ack | |
| 3554 // alarm is reset. | |
| 3555 EXPECT_EQ(3u, writer_->frame_count()); | |
| 3556 EXPECT_FALSE(writer_->stop_waiting_frames().empty()); | |
| 3557 EXPECT_FALSE(writer_->ack_frames().empty()); | |
| 3558 EXPECT_EQ(1u, writer_->stream_frames().size()); | |
| 3559 EXPECT_FALSE(connection_.GetAckAlarm()->IsSet()); | |
| 3560 } | |
| 3561 | |
| 3562 TEST_P(QuicConnectionTest, NoAckSentForClose) { | |
| 3563 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 3564 ProcessPacket(1); | |
| 3565 EXPECT_CALL(visitor_, OnConnectionClosed(QUIC_PEER_GOING_AWAY, true)); | |
| 3566 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)).Times(0); | |
| 3567 ProcessClosePacket(2, 0); | |
| 3568 } | |
| 3569 | |
| 3570 TEST_P(QuicConnectionTest, SendWhenDisconnected) { | |
| 3571 EXPECT_TRUE(connection_.connected()); | |
| 3572 EXPECT_CALL(visitor_, OnConnectionClosed(QUIC_PEER_GOING_AWAY, false)); | |
| 3573 connection_.CloseConnection(QUIC_PEER_GOING_AWAY, false); | |
| 3574 EXPECT_FALSE(connection_.connected()); | |
| 3575 EXPECT_FALSE(connection_.CanWriteStreamData()); | |
| 3576 QuicPacket* packet = ConstructDataPacket(1, 0, !kEntropyFlag); | |
| 3577 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, 1, _, _)).Times(0); | |
| 3578 connection_.SendPacket( | |
| 3579 ENCRYPTION_NONE, 1, packet, kTestEntropyHash, HAS_RETRANSMITTABLE_DATA); | |
| 3580 } | |
| 3581 | |
| 3582 TEST_P(QuicConnectionTest, PublicReset) { | |
| 3583 QuicPublicResetPacket header; | |
| 3584 header.public_header.connection_id = connection_id_; | |
| 3585 header.public_header.reset_flag = true; | |
| 3586 header.public_header.version_flag = false; | |
| 3587 header.rejected_sequence_number = 10101; | |
| 3588 scoped_ptr<QuicEncryptedPacket> packet( | |
| 3589 framer_.BuildPublicResetPacket(header)); | |
| 3590 EXPECT_CALL(visitor_, OnConnectionClosed(QUIC_PUBLIC_RESET, true)); | |
| 3591 connection_.ProcessUdpPacket(IPEndPoint(), IPEndPoint(), *packet); | |
| 3592 } | |
| 3593 | |
| 3594 TEST_P(QuicConnectionTest, GoAway) { | |
| 3595 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 3596 | |
| 3597 QuicGoAwayFrame goaway; | |
| 3598 goaway.last_good_stream_id = 1; | |
| 3599 goaway.error_code = QUIC_PEER_GOING_AWAY; | |
| 3600 goaway.reason_phrase = "Going away."; | |
| 3601 EXPECT_CALL(visitor_, OnGoAway(_)); | |
| 3602 ProcessGoAwayPacket(&goaway); | |
| 3603 } | |
| 3604 | |
| 3605 TEST_P(QuicConnectionTest, WindowUpdate) { | |
| 3606 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 3607 | |
| 3608 QuicWindowUpdateFrame window_update; | |
| 3609 window_update.stream_id = 3; | |
| 3610 window_update.byte_offset = 1234; | |
| 3611 EXPECT_CALL(visitor_, OnWindowUpdateFrames(_)); | |
| 3612 ProcessFramePacket(QuicFrame(&window_update)); | |
| 3613 } | |
| 3614 | |
| 3615 TEST_P(QuicConnectionTest, Blocked) { | |
| 3616 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 3617 | |
| 3618 QuicBlockedFrame blocked; | |
| 3619 blocked.stream_id = 3; | |
| 3620 EXPECT_CALL(visitor_, OnBlockedFrames(_)); | |
| 3621 ProcessFramePacket(QuicFrame(&blocked)); | |
| 3622 } | |
| 3623 | |
| 3624 TEST_P(QuicConnectionTest, ZeroBytePacket) { | |
| 3625 // Don't close the connection for zero byte packets. | |
| 3626 EXPECT_CALL(visitor_, OnConnectionClosed(_, _)).Times(0); | |
| 3627 QuicEncryptedPacket encrypted(nullptr, 0); | |
| 3628 connection_.ProcessUdpPacket(IPEndPoint(), IPEndPoint(), encrypted); | |
| 3629 } | |
| 3630 | |
| 3631 TEST_P(QuicConnectionTest, MissingPacketsBeforeLeastUnacked) { | |
| 3632 // Set the sequence number of the ack packet to be least unacked (4). | |
| 3633 peer_creator_.set_sequence_number(3); | |
| 3634 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 3635 QuicStopWaitingFrame frame = InitStopWaitingFrame(4); | |
| 3636 ProcessStopWaitingPacket(&frame); | |
| 3637 EXPECT_TRUE(outgoing_ack()->missing_packets.empty()); | |
| 3638 } | |
| 3639 | |
| 3640 TEST_P(QuicConnectionTest, ReceivedEntropyHashCalculation) { | |
| 3641 EXPECT_CALL(visitor_, OnStreamFrames(_)).Times(AtLeast(1)); | |
| 3642 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 3643 ProcessDataPacket(1, 1, kEntropyFlag); | |
| 3644 ProcessDataPacket(4, 1, kEntropyFlag); | |
| 3645 ProcessDataPacket(3, 1, !kEntropyFlag); | |
| 3646 ProcessDataPacket(7, 1, kEntropyFlag); | |
| 3647 EXPECT_EQ(146u, outgoing_ack()->entropy_hash); | |
| 3648 } | |
| 3649 | |
| 3650 TEST_P(QuicConnectionTest, ReceivedEntropyHashCalculationHalfFEC) { | |
| 3651 // FEC packets should not change the entropy hash calculation. | |
| 3652 EXPECT_CALL(visitor_, OnStreamFrames(_)).Times(AtLeast(1)); | |
| 3653 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 3654 ProcessDataPacket(1, 1, kEntropyFlag); | |
| 3655 ProcessFecPacket(4, 1, false, kEntropyFlag, nullptr); | |
| 3656 ProcessDataPacket(3, 3, !kEntropyFlag); | |
| 3657 ProcessFecPacket(7, 3, false, kEntropyFlag, nullptr); | |
| 3658 EXPECT_EQ(146u, outgoing_ack()->entropy_hash); | |
| 3659 } | |
| 3660 | |
| 3661 TEST_P(QuicConnectionTest, UpdateEntropyForReceivedPackets) { | |
| 3662 EXPECT_CALL(visitor_, OnStreamFrames(_)).Times(AtLeast(1)); | |
| 3663 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 3664 ProcessDataPacket(1, 1, kEntropyFlag); | |
| 3665 ProcessDataPacket(5, 1, kEntropyFlag); | |
| 3666 ProcessDataPacket(4, 1, !kEntropyFlag); | |
| 3667 EXPECT_EQ(34u, outgoing_ack()->entropy_hash); | |
| 3668 // Make 4th packet my least unacked, and update entropy for 2, 3 packets. | |
| 3669 peer_creator_.set_sequence_number(5); | |
| 3670 QuicPacketEntropyHash six_packet_entropy_hash = 0; | |
| 3671 QuicPacketEntropyHash kRandomEntropyHash = 129u; | |
| 3672 QuicStopWaitingFrame frame = InitStopWaitingFrame(4); | |
| 3673 frame.entropy_hash = kRandomEntropyHash; | |
| 3674 if (ProcessStopWaitingPacket(&frame)) { | |
| 3675 six_packet_entropy_hash = 1 << 6; | |
| 3676 } | |
| 3677 | |
| 3678 EXPECT_EQ((kRandomEntropyHash + (1 << 5) + six_packet_entropy_hash), | |
| 3679 outgoing_ack()->entropy_hash); | |
| 3680 } | |
| 3681 | |
| 3682 TEST_P(QuicConnectionTest, UpdateEntropyHashUptoCurrentPacket) { | |
| 3683 EXPECT_CALL(visitor_, OnStreamFrames(_)).Times(AtLeast(1)); | |
| 3684 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 3685 ProcessDataPacket(1, 1, kEntropyFlag); | |
| 3686 ProcessDataPacket(5, 1, !kEntropyFlag); | |
| 3687 ProcessDataPacket(22, 1, kEntropyFlag); | |
| 3688 EXPECT_EQ(66u, outgoing_ack()->entropy_hash); | |
| 3689 peer_creator_.set_sequence_number(22); | |
| 3690 QuicPacketEntropyHash kRandomEntropyHash = 85u; | |
| 3691 // Current packet is the least unacked packet. | |
| 3692 QuicPacketEntropyHash ack_entropy_hash; | |
| 3693 QuicStopWaitingFrame frame = InitStopWaitingFrame(23); | |
| 3694 frame.entropy_hash = kRandomEntropyHash; | |
| 3695 ack_entropy_hash = ProcessStopWaitingPacket(&frame); | |
| 3696 EXPECT_EQ((kRandomEntropyHash + ack_entropy_hash), | |
| 3697 outgoing_ack()->entropy_hash); | |
| 3698 ProcessDataPacket(25, 1, kEntropyFlag); | |
| 3699 EXPECT_EQ((kRandomEntropyHash + ack_entropy_hash + (1 << (25 % 8))), | |
| 3700 outgoing_ack()->entropy_hash); | |
| 3701 } | |
| 3702 | |
| 3703 TEST_P(QuicConnectionTest, EntropyCalculationForTruncatedAck) { | |
| 3704 EXPECT_CALL(visitor_, OnStreamFrames(_)).Times(AtLeast(1)); | |
| 3705 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 3706 QuicPacketEntropyHash entropy[51]; | |
| 3707 entropy[0] = 0; | |
| 3708 for (int i = 1; i < 51; ++i) { | |
| 3709 bool should_send = i % 10 != 1; | |
| 3710 bool entropy_flag = (i & (i - 1)) != 0; | |
| 3711 if (!should_send) { | |
| 3712 entropy[i] = entropy[i - 1]; | |
| 3713 continue; | |
| 3714 } | |
| 3715 if (entropy_flag) { | |
| 3716 entropy[i] = entropy[i - 1] ^ (1 << (i % 8)); | |
| 3717 } else { | |
| 3718 entropy[i] = entropy[i - 1]; | |
| 3719 } | |
| 3720 ProcessDataPacket(i, 1, entropy_flag); | |
| 3721 } | |
| 3722 for (int i = 1; i < 50; ++i) { | |
| 3723 EXPECT_EQ(entropy[i], QuicConnectionPeer::ReceivedEntropyHash( | |
| 3724 &connection_, i)); | |
| 3725 } | |
| 3726 } | |
| 3727 | |
| 3728 TEST_P(QuicConnectionTest, ServerSendsVersionNegotiationPacket) { | |
| 3729 connection_.SetSupportedVersions(QuicSupportedVersions()); | |
| 3730 framer_.set_version_for_tests(QUIC_VERSION_UNSUPPORTED); | |
| 3731 | |
| 3732 QuicPacketHeader header; | |
| 3733 header.public_header.connection_id = connection_id_; | |
| 3734 header.public_header.reset_flag = false; | |
| 3735 header.public_header.version_flag = true; | |
| 3736 header.entropy_flag = false; | |
| 3737 header.fec_flag = false; | |
| 3738 header.packet_sequence_number = 12; | |
| 3739 header.fec_group = 0; | |
| 3740 | |
| 3741 QuicFrames frames; | |
| 3742 QuicFrame frame(&frame1_); | |
| 3743 frames.push_back(frame); | |
| 3744 scoped_ptr<QuicPacket> packet( | |
| 3745 BuildUnsizedDataPacket(&framer_, header, frames)); | |
| 3746 scoped_ptr<QuicEncryptedPacket> encrypted( | |
| 3747 framer_.EncryptPacket(ENCRYPTION_NONE, 12, *packet)); | |
| 3748 | |
| 3749 framer_.set_version(version()); | |
| 3750 connection_.set_is_server(true); | |
| 3751 connection_.ProcessUdpPacket(IPEndPoint(), IPEndPoint(), *encrypted); | |
| 3752 EXPECT_TRUE(writer_->version_negotiation_packet() != nullptr); | |
| 3753 | |
| 3754 size_t num_versions = arraysize(kSupportedQuicVersions); | |
| 3755 ASSERT_EQ(num_versions, | |
| 3756 writer_->version_negotiation_packet()->versions.size()); | |
| 3757 | |
| 3758 // We expect all versions in kSupportedQuicVersions to be | |
| 3759 // included in the packet. | |
| 3760 for (size_t i = 0; i < num_versions; ++i) { | |
| 3761 EXPECT_EQ(kSupportedQuicVersions[i], | |
| 3762 writer_->version_negotiation_packet()->versions[i]); | |
| 3763 } | |
| 3764 } | |
| 3765 | |
| 3766 TEST_P(QuicConnectionTest, ServerSendsVersionNegotiationPacketSocketBlocked) { | |
| 3767 connection_.SetSupportedVersions(QuicSupportedVersions()); | |
| 3768 framer_.set_version_for_tests(QUIC_VERSION_UNSUPPORTED); | |
| 3769 | |
| 3770 QuicPacketHeader header; | |
| 3771 header.public_header.connection_id = connection_id_; | |
| 3772 header.public_header.reset_flag = false; | |
| 3773 header.public_header.version_flag = true; | |
| 3774 header.entropy_flag = false; | |
| 3775 header.fec_flag = false; | |
| 3776 header.packet_sequence_number = 12; | |
| 3777 header.fec_group = 0; | |
| 3778 | |
| 3779 QuicFrames frames; | |
| 3780 QuicFrame frame(&frame1_); | |
| 3781 frames.push_back(frame); | |
| 3782 scoped_ptr<QuicPacket> packet( | |
| 3783 BuildUnsizedDataPacket(&framer_, header, frames)); | |
| 3784 scoped_ptr<QuicEncryptedPacket> encrypted( | |
| 3785 framer_.EncryptPacket(ENCRYPTION_NONE, 12, *packet)); | |
| 3786 | |
| 3787 framer_.set_version(version()); | |
| 3788 connection_.set_is_server(true); | |
| 3789 BlockOnNextWrite(); | |
| 3790 connection_.ProcessUdpPacket(IPEndPoint(), IPEndPoint(), *encrypted); | |
| 3791 EXPECT_EQ(0u, writer_->last_packet_size()); | |
| 3792 EXPECT_TRUE(connection_.HasQueuedData()); | |
| 3793 | |
| 3794 writer_->SetWritable(); | |
| 3795 connection_.OnCanWrite(); | |
| 3796 EXPECT_TRUE(writer_->version_negotiation_packet() != nullptr); | |
| 3797 | |
| 3798 size_t num_versions = arraysize(kSupportedQuicVersions); | |
| 3799 ASSERT_EQ(num_versions, | |
| 3800 writer_->version_negotiation_packet()->versions.size()); | |
| 3801 | |
| 3802 // We expect all versions in kSupportedQuicVersions to be | |
| 3803 // included in the packet. | |
| 3804 for (size_t i = 0; i < num_versions; ++i) { | |
| 3805 EXPECT_EQ(kSupportedQuicVersions[i], | |
| 3806 writer_->version_negotiation_packet()->versions[i]); | |
| 3807 } | |
| 3808 } | |
| 3809 | |
| 3810 TEST_P(QuicConnectionTest, | |
| 3811 ServerSendsVersionNegotiationPacketSocketBlockedDataBuffered) { | |
| 3812 connection_.SetSupportedVersions(QuicSupportedVersions()); | |
| 3813 framer_.set_version_for_tests(QUIC_VERSION_UNSUPPORTED); | |
| 3814 | |
| 3815 QuicPacketHeader header; | |
| 3816 header.public_header.connection_id = connection_id_; | |
| 3817 header.public_header.reset_flag = false; | |
| 3818 header.public_header.version_flag = true; | |
| 3819 header.entropy_flag = false; | |
| 3820 header.fec_flag = false; | |
| 3821 header.packet_sequence_number = 12; | |
| 3822 header.fec_group = 0; | |
| 3823 | |
| 3824 QuicFrames frames; | |
| 3825 QuicFrame frame(&frame1_); | |
| 3826 frames.push_back(frame); | |
| 3827 scoped_ptr<QuicPacket> packet( | |
| 3828 BuildUnsizedDataPacket(&framer_, header, frames)); | |
| 3829 scoped_ptr<QuicEncryptedPacket> encrypted( | |
| 3830 framer_.EncryptPacket(ENCRYPTION_NONE, 12, *packet)); | |
| 3831 | |
| 3832 framer_.set_version(version()); | |
| 3833 connection_.set_is_server(true); | |
| 3834 BlockOnNextWrite(); | |
| 3835 writer_->set_is_write_blocked_data_buffered(true); | |
| 3836 connection_.ProcessUdpPacket(IPEndPoint(), IPEndPoint(), *encrypted); | |
| 3837 EXPECT_EQ(0u, writer_->last_packet_size()); | |
| 3838 EXPECT_FALSE(connection_.HasQueuedData()); | |
| 3839 } | |
| 3840 | |
| 3841 TEST_P(QuicConnectionTest, ClientHandlesVersionNegotiation) { | |
| 3842 // Start out with some unsupported version. | |
| 3843 QuicConnectionPeer::GetFramer(&connection_)->set_version_for_tests( | |
| 3844 QUIC_VERSION_UNSUPPORTED); | |
| 3845 | |
| 3846 QuicPacketHeader header; | |
| 3847 header.public_header.connection_id = connection_id_; | |
| 3848 header.public_header.reset_flag = false; | |
| 3849 header.public_header.version_flag = true; | |
| 3850 header.entropy_flag = false; | |
| 3851 header.fec_flag = false; | |
| 3852 header.packet_sequence_number = 12; | |
| 3853 header.fec_group = 0; | |
| 3854 | |
| 3855 QuicVersionVector supported_versions; | |
| 3856 for (size_t i = 0; i < arraysize(kSupportedQuicVersions); ++i) { | |
| 3857 supported_versions.push_back(kSupportedQuicVersions[i]); | |
| 3858 } | |
| 3859 | |
| 3860 // Send a version negotiation packet. | |
| 3861 scoped_ptr<QuicEncryptedPacket> encrypted( | |
| 3862 framer_.BuildVersionNegotiationPacket( | |
| 3863 header.public_header, supported_versions)); | |
| 3864 connection_.ProcessUdpPacket(IPEndPoint(), IPEndPoint(), *encrypted); | |
| 3865 | |
| 3866 // Now force another packet. The connection should transition into | |
| 3867 // NEGOTIATED_VERSION state and tell the packet creator to StopSendingVersion. | |
| 3868 header.public_header.version_flag = false; | |
| 3869 QuicFrames frames; | |
| 3870 QuicFrame frame(&frame1_); | |
| 3871 frames.push_back(frame); | |
| 3872 scoped_ptr<QuicPacket> packet( | |
| 3873 BuildUnsizedDataPacket(&framer_, header, frames)); | |
| 3874 encrypted.reset(framer_.EncryptPacket(ENCRYPTION_NONE, 12, *packet)); | |
| 3875 EXPECT_CALL(visitor_, OnStreamFrames(_)).Times(1); | |
| 3876 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 3877 connection_.ProcessUdpPacket(IPEndPoint(), IPEndPoint(), *encrypted); | |
| 3878 | |
| 3879 ASSERT_FALSE(QuicPacketCreatorPeer::SendVersionInPacket(creator_)); | |
| 3880 } | |
| 3881 | |
| 3882 TEST_P(QuicConnectionTest, BadVersionNegotiation) { | |
| 3883 QuicPacketHeader header; | |
| 3884 header.public_header.connection_id = connection_id_; | |
| 3885 header.public_header.reset_flag = false; | |
| 3886 header.public_header.version_flag = true; | |
| 3887 header.entropy_flag = false; | |
| 3888 header.fec_flag = false; | |
| 3889 header.packet_sequence_number = 12; | |
| 3890 header.fec_group = 0; | |
| 3891 | |
| 3892 QuicVersionVector supported_versions; | |
| 3893 for (size_t i = 0; i < arraysize(kSupportedQuicVersions); ++i) { | |
| 3894 supported_versions.push_back(kSupportedQuicVersions[i]); | |
| 3895 } | |
| 3896 | |
| 3897 // Send a version negotiation packet with the version the client started with. | |
| 3898 // It should be rejected. | |
| 3899 EXPECT_CALL(visitor_, | |
| 3900 OnConnectionClosed(QUIC_INVALID_VERSION_NEGOTIATION_PACKET, | |
| 3901 false)); | |
| 3902 scoped_ptr<QuicEncryptedPacket> encrypted( | |
| 3903 framer_.BuildVersionNegotiationPacket( | |
| 3904 header.public_header, supported_versions)); | |
| 3905 connection_.ProcessUdpPacket(IPEndPoint(), IPEndPoint(), *encrypted); | |
| 3906 } | |
| 3907 | |
| 3908 TEST_P(QuicConnectionTest, CheckSendStats) { | |
| 3909 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)); | |
| 3910 connection_.SendStreamDataWithString(3, "first", 0, !kFin, nullptr); | |
| 3911 size_t first_packet_size = writer_->last_packet_size(); | |
| 3912 | |
| 3913 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)); | |
| 3914 connection_.SendStreamDataWithString(5, "second", 0, !kFin, nullptr); | |
| 3915 size_t second_packet_size = writer_->last_packet_size(); | |
| 3916 | |
| 3917 // 2 retransmissions due to rto, 1 due to explicit nack. | |
| 3918 EXPECT_CALL(*send_algorithm_, OnRetransmissionTimeout(true)); | |
| 3919 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)).Times(3); | |
| 3920 | |
| 3921 // Retransmit due to RTO. | |
| 3922 clock_.AdvanceTime(QuicTime::Delta::FromSeconds(10)); | |
| 3923 connection_.GetRetransmissionAlarm()->Fire(); | |
| 3924 | |
| 3925 // Retransmit due to explicit nacks. | |
| 3926 QuicAckFrame nack_three = InitAckFrame(4); | |
| 3927 NackPacket(3, &nack_three); | |
| 3928 NackPacket(1, &nack_three); | |
| 3929 SequenceNumberSet lost_packets; | |
| 3930 lost_packets.insert(1); | |
| 3931 lost_packets.insert(3); | |
| 3932 EXPECT_CALL(*loss_algorithm_, DetectLostPackets(_, _, _, _)) | |
| 3933 .WillOnce(Return(lost_packets)); | |
| 3934 EXPECT_CALL(*send_algorithm_, OnCongestionEvent(true, _, _, _)); | |
| 3935 EXPECT_CALL(visitor_, OnCanWrite()).Times(2); | |
| 3936 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 3937 if (!FLAGS_quic_use_new_rto) { | |
| 3938 EXPECT_CALL(*send_algorithm_, RevertRetransmissionTimeout()); | |
| 3939 } | |
| 3940 ProcessAckPacket(&nack_three); | |
| 3941 | |
| 3942 EXPECT_CALL(*send_algorithm_, BandwidthEstimate()).WillOnce( | |
| 3943 Return(QuicBandwidth::Zero())); | |
| 3944 | |
| 3945 const QuicConnectionStats& stats = connection_.GetStats(); | |
| 3946 EXPECT_EQ(3 * first_packet_size + 2 * second_packet_size - kQuicVersionSize, | |
| 3947 stats.bytes_sent); | |
| 3948 EXPECT_EQ(5u, stats.packets_sent); | |
| 3949 EXPECT_EQ(2 * first_packet_size + second_packet_size - kQuicVersionSize, | |
| 3950 stats.bytes_retransmitted); | |
| 3951 EXPECT_EQ(3u, stats.packets_retransmitted); | |
| 3952 EXPECT_EQ(1u, stats.rto_count); | |
| 3953 EXPECT_EQ(kDefaultMaxPacketSize, stats.max_packet_size); | |
| 3954 } | |
| 3955 | |
| 3956 TEST_P(QuicConnectionTest, CheckReceiveStats) { | |
| 3957 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 3958 | |
| 3959 size_t received_bytes = 0; | |
| 3960 received_bytes += ProcessFecProtectedPacket(1, false, !kEntropyFlag); | |
| 3961 received_bytes += ProcessFecProtectedPacket(3, false, !kEntropyFlag); | |
| 3962 // Should be counted against dropped packets. | |
| 3963 received_bytes += ProcessDataPacket(3, 1, !kEntropyFlag); | |
| 3964 received_bytes += ProcessFecPacket(4, 1, true, !kEntropyFlag, nullptr); | |
| 3965 | |
| 3966 EXPECT_CALL(*send_algorithm_, BandwidthEstimate()).WillOnce( | |
| 3967 Return(QuicBandwidth::Zero())); | |
| 3968 | |
| 3969 const QuicConnectionStats& stats = connection_.GetStats(); | |
| 3970 EXPECT_EQ(received_bytes, stats.bytes_received); | |
| 3971 EXPECT_EQ(4u, stats.packets_received); | |
| 3972 | |
| 3973 EXPECT_EQ(1u, stats.packets_revived); | |
| 3974 EXPECT_EQ(1u, stats.packets_dropped); | |
| 3975 } | |
| 3976 | |
| 3977 TEST_P(QuicConnectionTest, TestFecGroupLimits) { | |
| 3978 // Create and return a group for 1. | |
| 3979 ASSERT_TRUE(QuicConnectionPeer::GetFecGroup(&connection_, 1) != nullptr); | |
| 3980 | |
| 3981 // Create and return a group for 2. | |
| 3982 ASSERT_TRUE(QuicConnectionPeer::GetFecGroup(&connection_, 2) != nullptr); | |
| 3983 | |
| 3984 // Create and return a group for 4. This should remove 1 but not 2. | |
| 3985 ASSERT_TRUE(QuicConnectionPeer::GetFecGroup(&connection_, 4) != nullptr); | |
| 3986 ASSERT_TRUE(QuicConnectionPeer::GetFecGroup(&connection_, 1) == nullptr); | |
| 3987 ASSERT_TRUE(QuicConnectionPeer::GetFecGroup(&connection_, 2) != nullptr); | |
| 3988 | |
| 3989 // Create and return a group for 3. This will kill off 2. | |
| 3990 ASSERT_TRUE(QuicConnectionPeer::GetFecGroup(&connection_, 3) != nullptr); | |
| 3991 ASSERT_TRUE(QuicConnectionPeer::GetFecGroup(&connection_, 2) == nullptr); | |
| 3992 | |
| 3993 // Verify that adding 5 kills off 3, despite 4 being created before 3. | |
| 3994 ASSERT_TRUE(QuicConnectionPeer::GetFecGroup(&connection_, 5) != nullptr); | |
| 3995 ASSERT_TRUE(QuicConnectionPeer::GetFecGroup(&connection_, 4) != nullptr); | |
| 3996 ASSERT_TRUE(QuicConnectionPeer::GetFecGroup(&connection_, 3) == nullptr); | |
| 3997 } | |
| 3998 | |
| 3999 TEST_P(QuicConnectionTest, ProcessFramesIfPacketClosedConnection) { | |
| 4000 // Construct a packet with stream frame and connection close frame. | |
| 4001 header_.public_header.connection_id = connection_id_; | |
| 4002 header_.packet_sequence_number = 1; | |
| 4003 header_.public_header.reset_flag = false; | |
| 4004 header_.public_header.version_flag = false; | |
| 4005 header_.entropy_flag = false; | |
| 4006 header_.fec_flag = false; | |
| 4007 header_.fec_group = 0; | |
| 4008 | |
| 4009 QuicConnectionCloseFrame qccf; | |
| 4010 qccf.error_code = QUIC_PEER_GOING_AWAY; | |
| 4011 QuicFrame close_frame(&qccf); | |
| 4012 QuicFrame stream_frame(&frame1_); | |
| 4013 | |
| 4014 QuicFrames frames; | |
| 4015 frames.push_back(stream_frame); | |
| 4016 frames.push_back(close_frame); | |
| 4017 scoped_ptr<QuicPacket> packet( | |
| 4018 BuildUnsizedDataPacket(&framer_, header_, frames)); | |
| 4019 EXPECT_TRUE(nullptr != packet.get()); | |
| 4020 scoped_ptr<QuicEncryptedPacket> encrypted(framer_.EncryptPacket( | |
| 4021 ENCRYPTION_NONE, 1, *packet)); | |
| 4022 | |
| 4023 EXPECT_CALL(visitor_, OnConnectionClosed(QUIC_PEER_GOING_AWAY, true)); | |
| 4024 EXPECT_CALL(visitor_, OnStreamFrames(_)).Times(1); | |
| 4025 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 4026 | |
| 4027 connection_.ProcessUdpPacket(IPEndPoint(), IPEndPoint(), *encrypted); | |
| 4028 } | |
| 4029 | |
| 4030 TEST_P(QuicConnectionTest, SelectMutualVersion) { | |
| 4031 connection_.SetSupportedVersions(QuicSupportedVersions()); | |
| 4032 // Set the connection to speak the lowest quic version. | |
| 4033 connection_.set_version(QuicVersionMin()); | |
| 4034 EXPECT_EQ(QuicVersionMin(), connection_.version()); | |
| 4035 | |
| 4036 // Pass in available versions which includes a higher mutually supported | |
| 4037 // version. The higher mutually supported version should be selected. | |
| 4038 QuicVersionVector supported_versions; | |
| 4039 for (size_t i = 0; i < arraysize(kSupportedQuicVersions); ++i) { | |
| 4040 supported_versions.push_back(kSupportedQuicVersions[i]); | |
| 4041 } | |
| 4042 EXPECT_TRUE(connection_.SelectMutualVersion(supported_versions)); | |
| 4043 EXPECT_EQ(QuicVersionMax(), connection_.version()); | |
| 4044 | |
| 4045 // Expect that the lowest version is selected. | |
| 4046 // Ensure the lowest supported version is less than the max, unless they're | |
| 4047 // the same. | |
| 4048 EXPECT_LE(QuicVersionMin(), QuicVersionMax()); | |
| 4049 QuicVersionVector lowest_version_vector; | |
| 4050 lowest_version_vector.push_back(QuicVersionMin()); | |
| 4051 EXPECT_TRUE(connection_.SelectMutualVersion(lowest_version_vector)); | |
| 4052 EXPECT_EQ(QuicVersionMin(), connection_.version()); | |
| 4053 | |
| 4054 // Shouldn't be able to find a mutually supported version. | |
| 4055 QuicVersionVector unsupported_version; | |
| 4056 unsupported_version.push_back(QUIC_VERSION_UNSUPPORTED); | |
| 4057 EXPECT_FALSE(connection_.SelectMutualVersion(unsupported_version)); | |
| 4058 } | |
| 4059 | |
| 4060 TEST_P(QuicConnectionTest, ConnectionCloseWhenWritable) { | |
| 4061 EXPECT_FALSE(writer_->IsWriteBlocked()); | |
| 4062 | |
| 4063 // Send a packet. | |
| 4064 connection_.SendStreamDataWithString(1, "foo", 0, !kFin, nullptr); | |
| 4065 EXPECT_EQ(0u, connection_.NumQueuedPackets()); | |
| 4066 EXPECT_EQ(1u, writer_->packets_write_attempts()); | |
| 4067 | |
| 4068 TriggerConnectionClose(); | |
| 4069 EXPECT_EQ(2u, writer_->packets_write_attempts()); | |
| 4070 } | |
| 4071 | |
| 4072 TEST_P(QuicConnectionTest, ConnectionCloseGettingWriteBlocked) { | |
| 4073 BlockOnNextWrite(); | |
| 4074 TriggerConnectionClose(); | |
| 4075 EXPECT_EQ(1u, writer_->packets_write_attempts()); | |
| 4076 EXPECT_TRUE(writer_->IsWriteBlocked()); | |
| 4077 } | |
| 4078 | |
| 4079 TEST_P(QuicConnectionTest, ConnectionCloseWhenWriteBlocked) { | |
| 4080 BlockOnNextWrite(); | |
| 4081 connection_.SendStreamDataWithString(1, "foo", 0, !kFin, nullptr); | |
| 4082 EXPECT_EQ(1u, connection_.NumQueuedPackets()); | |
| 4083 EXPECT_EQ(1u, writer_->packets_write_attempts()); | |
| 4084 EXPECT_TRUE(writer_->IsWriteBlocked()); | |
| 4085 TriggerConnectionClose(); | |
| 4086 EXPECT_EQ(1u, writer_->packets_write_attempts()); | |
| 4087 } | |
| 4088 | |
| 4089 TEST_P(QuicConnectionTest, AckNotifierTriggerCallback) { | |
| 4090 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 4091 | |
| 4092 // Create a delegate which we expect to be called. | |
| 4093 scoped_refptr<MockAckNotifierDelegate> delegate(new MockAckNotifierDelegate); | |
| 4094 EXPECT_CALL(*delegate.get(), OnAckNotification(_, _, _)).Times(1); | |
| 4095 | |
| 4096 // Send some data, which will register the delegate to be notified. | |
| 4097 connection_.SendStreamDataWithString(1, "foo", 0, !kFin, delegate.get()); | |
| 4098 | |
| 4099 // Process an ACK from the server which should trigger the callback. | |
| 4100 EXPECT_CALL(*send_algorithm_, OnCongestionEvent(true, _, _, _)); | |
| 4101 QuicAckFrame frame = InitAckFrame(1); | |
| 4102 ProcessAckPacket(&frame); | |
| 4103 } | |
| 4104 | |
| 4105 TEST_P(QuicConnectionTest, AckNotifierFailToTriggerCallback) { | |
| 4106 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 4107 | |
| 4108 // Create a delegate which we don't expect to be called. | |
| 4109 scoped_refptr<MockAckNotifierDelegate> delegate(new MockAckNotifierDelegate); | |
| 4110 EXPECT_CALL(*delegate.get(), OnAckNotification(_, _, _)).Times(0); | |
| 4111 | |
| 4112 // Send some data, which will register the delegate to be notified. This will | |
| 4113 // not be ACKed and so the delegate should never be called. | |
| 4114 connection_.SendStreamDataWithString(1, "foo", 0, !kFin, delegate.get()); | |
| 4115 | |
| 4116 // Send some other data which we will ACK. | |
| 4117 connection_.SendStreamDataWithString(1, "foo", 0, !kFin, nullptr); | |
| 4118 connection_.SendStreamDataWithString(1, "bar", 0, !kFin, nullptr); | |
| 4119 | |
| 4120 // Now we receive ACK for packets 2 and 3, but importantly missing packet 1 | |
| 4121 // which we registered to be notified about. | |
| 4122 QuicAckFrame frame = InitAckFrame(3); | |
| 4123 NackPacket(1, &frame); | |
| 4124 SequenceNumberSet lost_packets; | |
| 4125 lost_packets.insert(1); | |
| 4126 EXPECT_CALL(*loss_algorithm_, DetectLostPackets(_, _, _, _)) | |
| 4127 .WillOnce(Return(lost_packets)); | |
| 4128 EXPECT_CALL(*send_algorithm_, OnCongestionEvent(true, _, _, _)); | |
| 4129 ProcessAckPacket(&frame); | |
| 4130 } | |
| 4131 | |
| 4132 TEST_P(QuicConnectionTest, AckNotifierCallbackAfterRetransmission) { | |
| 4133 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 4134 | |
| 4135 // Create a delegate which we expect to be called. | |
| 4136 scoped_refptr<MockAckNotifierDelegate> delegate(new MockAckNotifierDelegate); | |
| 4137 EXPECT_CALL(*delegate.get(), OnAckNotification(_, _, _)).Times(1); | |
| 4138 | |
| 4139 // Send four packets, and register to be notified on ACK of packet 2. | |
| 4140 connection_.SendStreamDataWithString(3, "foo", 0, !kFin, nullptr); | |
| 4141 connection_.SendStreamDataWithString(3, "bar", 0, !kFin, delegate.get()); | |
| 4142 connection_.SendStreamDataWithString(3, "baz", 0, !kFin, nullptr); | |
| 4143 connection_.SendStreamDataWithString(3, "qux", 0, !kFin, nullptr); | |
| 4144 | |
| 4145 // Now we receive ACK for packets 1, 3, and 4 and lose 2. | |
| 4146 QuicAckFrame frame = InitAckFrame(4); | |
| 4147 NackPacket(2, &frame); | |
| 4148 SequenceNumberSet lost_packets; | |
| 4149 lost_packets.insert(2); | |
| 4150 EXPECT_CALL(*loss_algorithm_, DetectLostPackets(_, _, _, _)) | |
| 4151 .WillOnce(Return(lost_packets)); | |
| 4152 EXPECT_CALL(*send_algorithm_, OnCongestionEvent(true, _, _, _)); | |
| 4153 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)); | |
| 4154 ProcessAckPacket(&frame); | |
| 4155 | |
| 4156 // Now we get an ACK for packet 5 (retransmitted packet 2), which should | |
| 4157 // trigger the callback. | |
| 4158 EXPECT_CALL(*loss_algorithm_, DetectLostPackets(_, _, _, _)) | |
| 4159 .WillRepeatedly(Return(SequenceNumberSet())); | |
| 4160 EXPECT_CALL(*send_algorithm_, OnCongestionEvent(true, _, _, _)); | |
| 4161 QuicAckFrame second_ack_frame = InitAckFrame(5); | |
| 4162 ProcessAckPacket(&second_ack_frame); | |
| 4163 } | |
| 4164 | |
| 4165 // AckNotifierCallback is triggered by the ack of a packet that timed | |
| 4166 // out and was retransmitted, even though the retransmission has a | |
| 4167 // different sequence number. | |
| 4168 TEST_P(QuicConnectionTest, AckNotifierCallbackForAckAfterRTO) { | |
| 4169 InSequence s; | |
| 4170 | |
| 4171 // Create a delegate which we expect to be called. | |
| 4172 scoped_refptr<MockAckNotifierDelegate> delegate( | |
| 4173 new StrictMock<MockAckNotifierDelegate>); | |
| 4174 | |
| 4175 QuicTime default_retransmission_time = clock_.ApproximateNow().Add( | |
| 4176 DefaultRetransmissionTime()); | |
| 4177 connection_.SendStreamDataWithString(3, "foo", 0, !kFin, delegate.get()); | |
| 4178 EXPECT_EQ(1u, stop_waiting()->least_unacked); | |
| 4179 | |
| 4180 EXPECT_EQ(1u, writer_->header().packet_sequence_number); | |
| 4181 EXPECT_EQ(default_retransmission_time, | |
| 4182 connection_.GetRetransmissionAlarm()->deadline()); | |
| 4183 // Simulate the retransmission alarm firing. | |
| 4184 clock_.AdvanceTime(DefaultRetransmissionTime()); | |
| 4185 if (!FLAGS_quic_use_new_rto) { | |
| 4186 EXPECT_CALL(*send_algorithm_, OnRetransmissionTimeout(true)); | |
| 4187 } | |
| 4188 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, 2u, _, _)); | |
| 4189 connection_.GetRetransmissionAlarm()->Fire(); | |
| 4190 EXPECT_EQ(2u, writer_->header().packet_sequence_number); | |
| 4191 // We do not raise the high water mark yet. | |
| 4192 EXPECT_EQ(1u, stop_waiting()->least_unacked); | |
| 4193 | |
| 4194 // Ack the original packet, which will revert the RTO. | |
| 4195 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 4196 EXPECT_CALL(*delegate.get(), OnAckNotification(1, _, _)); | |
| 4197 if (!FLAGS_quic_use_new_rto) { | |
| 4198 EXPECT_CALL(*send_algorithm_, RevertRetransmissionTimeout()); | |
| 4199 } | |
| 4200 EXPECT_CALL(*send_algorithm_, OnCongestionEvent(true, _, _, _)); | |
| 4201 QuicAckFrame ack_frame = InitAckFrame(1); | |
| 4202 ProcessAckPacket(&ack_frame); | |
| 4203 | |
| 4204 // Delegate is not notified again when the retransmit is acked. | |
| 4205 EXPECT_CALL(*send_algorithm_, OnCongestionEvent(true, _, _, _)); | |
| 4206 QuicAckFrame second_ack_frame = InitAckFrame(2); | |
| 4207 ProcessAckPacket(&second_ack_frame); | |
| 4208 } | |
| 4209 | |
| 4210 // AckNotifierCallback is triggered by the ack of a packet that was | |
| 4211 // previously nacked, even though the retransmission has a different | |
| 4212 // sequence number. | |
| 4213 TEST_P(QuicConnectionTest, AckNotifierCallbackForAckOfNackedPacket) { | |
| 4214 InSequence s; | |
| 4215 | |
| 4216 // Create a delegate which we expect to be called. | |
| 4217 scoped_refptr<MockAckNotifierDelegate> delegate( | |
| 4218 new StrictMock<MockAckNotifierDelegate>); | |
| 4219 | |
| 4220 // Send four packets, and register to be notified on ACK of packet 2. | |
| 4221 connection_.SendStreamDataWithString(3, "foo", 0, !kFin, nullptr); | |
| 4222 connection_.SendStreamDataWithString(3, "bar", 0, !kFin, delegate.get()); | |
| 4223 connection_.SendStreamDataWithString(3, "baz", 0, !kFin, nullptr); | |
| 4224 connection_.SendStreamDataWithString(3, "qux", 0, !kFin, nullptr); | |
| 4225 | |
| 4226 // Now we receive ACK for packets 1, 3, and 4 and lose 2. | |
| 4227 QuicAckFrame frame = InitAckFrame(4); | |
| 4228 NackPacket(2, &frame); | |
| 4229 SequenceNumberSet lost_packets; | |
| 4230 lost_packets.insert(2); | |
| 4231 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 4232 EXPECT_CALL(*loss_algorithm_, DetectLostPackets(_, _, _, _)) | |
| 4233 .WillOnce(Return(lost_packets)); | |
| 4234 EXPECT_CALL(*send_algorithm_, OnCongestionEvent(true, _, _, _)); | |
| 4235 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)); | |
| 4236 ProcessAckPacket(&frame); | |
| 4237 | |
| 4238 // Now we get an ACK for packet 2, which was previously nacked. | |
| 4239 SequenceNumberSet no_lost_packets; | |
| 4240 EXPECT_CALL(*delegate.get(), OnAckNotification(1, _, _)); | |
| 4241 EXPECT_CALL(*loss_algorithm_, DetectLostPackets(_, _, _, _)) | |
| 4242 .WillOnce(Return(no_lost_packets)); | |
| 4243 QuicAckFrame second_ack_frame = InitAckFrame(4); | |
| 4244 ProcessAckPacket(&second_ack_frame); | |
| 4245 | |
| 4246 // Verify that the delegate is not notified again when the | |
| 4247 // retransmit is acked. | |
| 4248 EXPECT_CALL(*loss_algorithm_, DetectLostPackets(_, _, _, _)) | |
| 4249 .WillOnce(Return(no_lost_packets)); | |
| 4250 EXPECT_CALL(*send_algorithm_, OnCongestionEvent(true, _, _, _)); | |
| 4251 QuicAckFrame third_ack_frame = InitAckFrame(5); | |
| 4252 ProcessAckPacket(&third_ack_frame); | |
| 4253 } | |
| 4254 | |
| 4255 TEST_P(QuicConnectionTest, AckNotifierFECTriggerCallback) { | |
| 4256 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 4257 | |
| 4258 // Create a delegate which we expect to be called. | |
| 4259 scoped_refptr<MockAckNotifierDelegate> delegate( | |
| 4260 new MockAckNotifierDelegate); | |
| 4261 EXPECT_CALL(*delegate.get(), OnAckNotification(_, _, _)).Times(1); | |
| 4262 | |
| 4263 // Send some data, which will register the delegate to be notified. | |
| 4264 connection_.SendStreamDataWithString(1, "foo", 0, !kFin, delegate.get()); | |
| 4265 connection_.SendStreamDataWithString(2, "bar", 0, !kFin, nullptr); | |
| 4266 | |
| 4267 // Process an ACK from the server with a revived packet, which should trigger | |
| 4268 // the callback. | |
| 4269 EXPECT_CALL(*send_algorithm_, OnCongestionEvent(true, _, _, _)); | |
| 4270 QuicAckFrame frame = InitAckFrame(2); | |
| 4271 NackPacket(1, &frame); | |
| 4272 frame.revived_packets.insert(1); | |
| 4273 ProcessAckPacket(&frame); | |
| 4274 // If the ack is processed again, the notifier should not be called again. | |
| 4275 ProcessAckPacket(&frame); | |
| 4276 } | |
| 4277 | |
| 4278 TEST_P(QuicConnectionTest, AckNotifierCallbackAfterFECRecovery) { | |
| 4279 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 4280 EXPECT_CALL(visitor_, OnCanWrite()); | |
| 4281 | |
| 4282 // Create a delegate which we expect to be called. | |
| 4283 scoped_refptr<MockAckNotifierDelegate> delegate(new MockAckNotifierDelegate); | |
| 4284 EXPECT_CALL(*delegate.get(), OnAckNotification(_, _, _)).Times(1); | |
| 4285 | |
| 4286 // Expect ACKs for 1 packet. | |
| 4287 EXPECT_CALL(*send_algorithm_, OnCongestionEvent(true, _, _, _)); | |
| 4288 | |
| 4289 // Send one packet, and register to be notified on ACK. | |
| 4290 connection_.SendStreamDataWithString(1, "foo", 0, !kFin, delegate.get()); | |
| 4291 | |
| 4292 // Ack packet gets dropped, but we receive an FEC packet that covers it. | |
| 4293 // Should recover the Ack packet and trigger the notification callback. | |
| 4294 QuicFrames frames; | |
| 4295 | |
| 4296 QuicAckFrame ack_frame = InitAckFrame(1); | |
| 4297 frames.push_back(QuicFrame(&ack_frame)); | |
| 4298 | |
| 4299 // Dummy stream frame to satisfy expectations set elsewhere. | |
| 4300 frames.push_back(QuicFrame(&frame1_)); | |
| 4301 | |
| 4302 QuicPacketHeader ack_header; | |
| 4303 ack_header.public_header.connection_id = connection_id_; | |
| 4304 ack_header.public_header.reset_flag = false; | |
| 4305 ack_header.public_header.version_flag = false; | |
| 4306 ack_header.entropy_flag = !kEntropyFlag; | |
| 4307 ack_header.fec_flag = true; | |
| 4308 ack_header.packet_sequence_number = 1; | |
| 4309 ack_header.is_in_fec_group = IN_FEC_GROUP; | |
| 4310 ack_header.fec_group = 1; | |
| 4311 | |
| 4312 QuicPacket* packet = BuildUnsizedDataPacket(&framer_, ack_header, frames); | |
| 4313 | |
| 4314 // Take the packet which contains the ACK frame, and construct and deliver an | |
| 4315 // FEC packet which allows the ACK packet to be recovered. | |
| 4316 ProcessFecPacket(2, 1, true, !kEntropyFlag, packet); | |
| 4317 } | |
| 4318 | |
| 4319 TEST_P(QuicConnectionTest, NetworkChangeVisitorCwndCallbackChangesFecState) { | |
| 4320 size_t max_packets_per_fec_group = creator_->max_packets_per_fec_group(); | |
| 4321 | |
| 4322 QuicSentPacketManager::NetworkChangeVisitor* visitor = | |
| 4323 QuicSentPacketManagerPeer::GetNetworkChangeVisitor(manager_); | |
| 4324 EXPECT_TRUE(visitor); | |
| 4325 | |
| 4326 // Increase FEC group size by increasing congestion window to a large number. | |
| 4327 EXPECT_CALL(*send_algorithm_, GetCongestionWindow()).WillRepeatedly( | |
| 4328 Return(1000 * kDefaultTCPMSS)); | |
| 4329 visitor->OnCongestionWindowChange(); | |
| 4330 EXPECT_LT(max_packets_per_fec_group, creator_->max_packets_per_fec_group()); | |
| 4331 } | |
| 4332 | |
| 4333 TEST_P(QuicConnectionTest, NetworkChangeVisitorConfigCallbackChangesFecState) { | |
| 4334 QuicSentPacketManager::NetworkChangeVisitor* visitor = | |
| 4335 QuicSentPacketManagerPeer::GetNetworkChangeVisitor(manager_); | |
| 4336 EXPECT_TRUE(visitor); | |
| 4337 EXPECT_EQ(QuicTime::Delta::Zero(), | |
| 4338 QuicPacketGeneratorPeer::GetFecTimeout(generator_)); | |
| 4339 | |
| 4340 // Verify that sending a config with a new initial rtt changes fec timeout. | |
| 4341 // Create and process a config with a non-zero initial RTT. | |
| 4342 EXPECT_CALL(*send_algorithm_, SetFromConfig(_, _, _)); | |
| 4343 QuicConfig config; | |
| 4344 config.SetInitialRoundTripTimeUsToSend(300000); | |
| 4345 connection_.SetFromConfig(config); | |
| 4346 EXPECT_LT(QuicTime::Delta::Zero(), | |
| 4347 QuicPacketGeneratorPeer::GetFecTimeout(generator_)); | |
| 4348 } | |
| 4349 | |
| 4350 TEST_P(QuicConnectionTest, NetworkChangeVisitorRttCallbackChangesFecState) { | |
| 4351 // Verify that sending a config with a new initial rtt changes fec timeout. | |
| 4352 QuicSentPacketManager::NetworkChangeVisitor* visitor = | |
| 4353 QuicSentPacketManagerPeer::GetNetworkChangeVisitor(manager_); | |
| 4354 EXPECT_TRUE(visitor); | |
| 4355 EXPECT_EQ(QuicTime::Delta::Zero(), | |
| 4356 QuicPacketGeneratorPeer::GetFecTimeout(generator_)); | |
| 4357 | |
| 4358 // Increase FEC timeout by increasing RTT. | |
| 4359 RttStats* rtt_stats = QuicSentPacketManagerPeer::GetRttStats(manager_); | |
| 4360 rtt_stats->UpdateRtt(QuicTime::Delta::FromMilliseconds(300), | |
| 4361 QuicTime::Delta::Zero(), QuicTime::Zero()); | |
| 4362 visitor->OnRttChange(); | |
| 4363 EXPECT_LT(QuicTime::Delta::Zero(), | |
| 4364 QuicPacketGeneratorPeer::GetFecTimeout(generator_)); | |
| 4365 } | |
| 4366 | |
| 4367 class MockQuicConnectionDebugVisitor | |
| 4368 : public QuicConnectionDebugVisitor { | |
| 4369 public: | |
| 4370 MOCK_METHOD1(OnFrameAddedToPacket, | |
| 4371 void(const QuicFrame&)); | |
| 4372 | |
| 4373 MOCK_METHOD6(OnPacketSent, | |
| 4374 void(const SerializedPacket&, | |
| 4375 QuicPacketSequenceNumber, | |
| 4376 EncryptionLevel, | |
| 4377 TransmissionType, | |
| 4378 const QuicEncryptedPacket&, | |
| 4379 QuicTime)); | |
| 4380 | |
| 4381 MOCK_METHOD3(OnPacketReceived, | |
| 4382 void(const IPEndPoint&, | |
| 4383 const IPEndPoint&, | |
| 4384 const QuicEncryptedPacket&)); | |
| 4385 | |
| 4386 MOCK_METHOD1(OnProtocolVersionMismatch, | |
| 4387 void(QuicVersion)); | |
| 4388 | |
| 4389 MOCK_METHOD1(OnPacketHeader, | |
| 4390 void(const QuicPacketHeader& header)); | |
| 4391 | |
| 4392 MOCK_METHOD1(OnStreamFrame, | |
| 4393 void(const QuicStreamFrame&)); | |
| 4394 | |
| 4395 MOCK_METHOD1(OnAckFrame, | |
| 4396 void(const QuicAckFrame& frame)); | |
| 4397 | |
| 4398 MOCK_METHOD1(OnStopWaitingFrame, | |
| 4399 void(const QuicStopWaitingFrame&)); | |
| 4400 | |
| 4401 MOCK_METHOD1(OnRstStreamFrame, | |
| 4402 void(const QuicRstStreamFrame&)); | |
| 4403 | |
| 4404 MOCK_METHOD1(OnConnectionCloseFrame, | |
| 4405 void(const QuicConnectionCloseFrame&)); | |
| 4406 | |
| 4407 MOCK_METHOD1(OnPublicResetPacket, | |
| 4408 void(const QuicPublicResetPacket&)); | |
| 4409 | |
| 4410 MOCK_METHOD1(OnVersionNegotiationPacket, | |
| 4411 void(const QuicVersionNegotiationPacket&)); | |
| 4412 | |
| 4413 MOCK_METHOD2(OnRevivedPacket, | |
| 4414 void(const QuicPacketHeader&, StringPiece payload)); | |
| 4415 }; | |
| 4416 | |
| 4417 TEST_P(QuicConnectionTest, OnPacketHeaderDebugVisitor) { | |
| 4418 QuicPacketHeader header; | |
| 4419 | |
| 4420 MockQuicConnectionDebugVisitor* debug_visitor = | |
| 4421 new MockQuicConnectionDebugVisitor(); | |
| 4422 connection_.set_debug_visitor(debug_visitor); | |
| 4423 EXPECT_CALL(*debug_visitor, OnPacketHeader(Ref(header))).Times(1); | |
| 4424 connection_.OnPacketHeader(header); | |
| 4425 } | |
| 4426 | |
| 4427 TEST_P(QuicConnectionTest, Pacing) { | |
| 4428 TestConnection server(connection_id_, IPEndPoint(), helper_.get(), | |
| 4429 factory_, /* is_server= */ true, version()); | |
| 4430 TestConnection client(connection_id_, IPEndPoint(), helper_.get(), | |
| 4431 factory_, /* is_server= */ false, version()); | |
| 4432 EXPECT_FALSE(client.sent_packet_manager().using_pacing()); | |
| 4433 EXPECT_FALSE(server.sent_packet_manager().using_pacing()); | |
| 4434 } | |
| 4435 | |
| 4436 TEST_P(QuicConnectionTest, ControlFramesInstigateAcks) { | |
| 4437 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | |
| 4438 | |
| 4439 // Send a WINDOW_UPDATE frame. | |
| 4440 QuicWindowUpdateFrame window_update; | |
| 4441 window_update.stream_id = 3; | |
| 4442 window_update.byte_offset = 1234; | |
| 4443 EXPECT_CALL(visitor_, OnWindowUpdateFrames(_)); | |
| 4444 ProcessFramePacket(QuicFrame(&window_update)); | |
| 4445 | |
| 4446 // Ensure that this has caused the ACK alarm to be set. | |
| 4447 QuicAlarm* ack_alarm = QuicConnectionPeer::GetAckAlarm(&connection_); | |
| 4448 EXPECT_TRUE(ack_alarm->IsSet()); | |
| 4449 | |
| 4450 // Cancel alarm, and try again with BLOCKED frame. | |
| 4451 ack_alarm->Cancel(); | |
| 4452 QuicBlockedFrame blocked; | |
| 4453 blocked.stream_id = 3; | |
| 4454 EXPECT_CALL(visitor_, OnBlockedFrames(_)); | |
| 4455 ProcessFramePacket(QuicFrame(&blocked)); | |
| 4456 EXPECT_TRUE(ack_alarm->IsSet()); | |
| 4457 } | |
| 4458 | |
| 4459 TEST_P(QuicConnectionTest, NoDataNoFin) { | |
| 4460 // Make sure that a call to SendStreamWithData, with no data and no FIN, does | |
| 4461 // not result in a QuicAckNotifier being used-after-free (fail under ASAN). | |
| 4462 // Regression test for b/18594622 | |
| 4463 scoped_refptr<MockAckNotifierDelegate> delegate(new MockAckNotifierDelegate); | |
| 4464 EXPECT_DFATAL( | |
| 4465 connection_.SendStreamDataWithString(3, "", 0, !kFin, delegate.get()), | |
| 4466 "Attempt to send empty stream frame"); | |
| 4467 } | |
| 4468 | |
| 4469 } // namespace | |
| 4470 } // namespace test | |
| 4471 } // namespace net | |
| OLD | NEW |