OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "net/quic/quic_connection.h" | 5 #include "net/quic/quic_connection.h" |
6 | 6 |
7 #include "base/basictypes.h" | 7 #include "base/basictypes.h" |
8 #include "base/bind.h" | 8 #include "base/bind.h" |
9 #include "base/stl_util.h" | 9 #include "base/stl_util.h" |
10 #include "net/base/net_errors.h" | 10 #include "net/base/net_errors.h" |
(...skipping 260 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
271 public: | 271 public: |
272 TestPacketWriter() | 272 TestPacketWriter() |
273 : last_packet_size_(0), | 273 : last_packet_size_(0), |
274 blocked_(false), | 274 blocked_(false), |
275 is_write_blocked_data_buffered_(false), | 275 is_write_blocked_data_buffered_(false), |
276 is_server_(true), | 276 is_server_(true), |
277 use_tagging_decrypter_(false), | 277 use_tagging_decrypter_(false), |
278 packets_write_attempts_(0) { | 278 packets_write_attempts_(0) { |
279 } | 279 } |
280 | 280 |
281 virtual ~TestPacketWriter() { | |
282 STLDeleteElements(&stream_data_); | |
283 } | |
284 | |
285 // QuicPacketWriter | 281 // QuicPacketWriter |
286 virtual WriteResult WritePacket( | 282 virtual WriteResult WritePacket( |
287 const char* buffer, size_t buf_len, | 283 const char* buffer, size_t buf_len, |
288 const IPAddressNumber& self_address, | 284 const IPAddressNumber& self_address, |
289 const IPEndPoint& peer_address, | 285 const IPEndPoint& peer_address, |
290 QuicBlockedWriterInterface* blocked_writer) OVERRIDE { | 286 QuicBlockedWriterInterface* blocked_writer) OVERRIDE { |
291 QuicEncryptedPacket packet(buffer, buf_len); | 287 QuicEncryptedPacket packet(buffer, buf_len); |
292 ++packets_write_attempts_; | 288 ++packets_write_attempts_; |
293 | 289 |
294 if (packet.length() >= sizeof(final_bytes_of_last_packet_)) { | 290 if (packet.length() >= sizeof(final_bytes_of_last_packet_)) { |
295 memcpy(&final_bytes_of_last_packet_, packet.data() + packet.length() - 4, | 291 memcpy(&final_bytes_of_last_packet_, packet.data() + packet.length() - 4, |
296 sizeof(final_bytes_of_last_packet_)); | 292 sizeof(final_bytes_of_last_packet_)); |
297 } | 293 } |
298 | 294 |
299 QuicFramer framer(QuicSupportedVersions(), QuicTime::Zero(), !is_server_); | 295 QuicFramer framer(QuicSupportedVersions(), QuicTime::Zero(), !is_server_); |
300 if (use_tagging_decrypter_) { | 296 if (use_tagging_decrypter_) { |
301 framer.SetDecrypter(new TaggingDecrypter); | 297 framer.SetDecrypter(new TaggingDecrypter); |
302 } | 298 } |
303 FramerVisitorCapturingFrames visitor; | 299 visitor_.Reset(); |
304 framer.set_visitor(&visitor); | 300 framer.set_visitor(&visitor_); |
305 EXPECT_TRUE(framer.ProcessPacket(packet)); | 301 EXPECT_TRUE(framer.ProcessPacket(packet)); |
306 header_ = *visitor.header(); | |
307 frame_count_ = visitor.frame_count(); | |
308 if (visitor.ack()) { | |
309 ack_.reset(new QuicAckFrame(*visitor.ack())); | |
310 } | |
311 if (visitor.feedback()) { | |
312 feedback_.reset(new QuicCongestionFeedbackFrame(*visitor.feedback())); | |
313 } | |
314 if (visitor.stream_frames() != NULL && !visitor.stream_frames()->empty()) { | |
315 stream_frames_ = *visitor.stream_frames(); | |
316 // Also make a copy of underlying data, since the data that the frames in | |
317 // |stream_frames_| point to is bound to the |visitor|'s scope. | |
318 for (size_t i = 0; i < stream_frames_.size(); ++i) { | |
319 stream_data_.push_back(new string(*visitor.stream_data()[i])); | |
320 IOVector data; | |
321 data.Append(const_cast<char*>(stream_data_.back()->data()), | |
322 stream_data_.back()->size()); | |
323 stream_frames_[i].data = data; | |
324 } | |
325 } | |
326 if (visitor.version_negotiation_packet() != NULL) { | |
327 version_negotiation_packet_.reset(new QuicVersionNegotiationPacket( | |
328 *visitor.version_negotiation_packet())); | |
329 } | |
330 if (blocked_) { | 302 if (blocked_) { |
331 return WriteResult(WRITE_STATUS_BLOCKED, -1); | 303 return WriteResult(WRITE_STATUS_BLOCKED, -1); |
332 } | 304 } |
333 last_packet_size_ = packet.length(); | 305 last_packet_size_ = packet.length(); |
334 return WriteResult(WRITE_STATUS_OK, last_packet_size_); | 306 return WriteResult(WRITE_STATUS_OK, last_packet_size_); |
335 } | 307 } |
336 | 308 |
337 virtual bool IsWriteBlockedDataBuffered() const OVERRIDE { | 309 virtual bool IsWriteBlockedDataBuffered() const OVERRIDE { |
338 return is_write_blocked_data_buffered_; | 310 return is_write_blocked_data_buffered_; |
339 } | 311 } |
340 | 312 |
341 QuicPacketHeader* header() { return &header_; } | 313 QuicPacketHeader* header() { return visitor_.header(); } |
342 | 314 |
343 size_t frame_count() const { return frame_count_; } | 315 size_t frame_count() const { return visitor_.frame_count(); } |
344 | 316 |
345 QuicAckFrame* ack() { return ack_.get(); } | 317 QuicAckFrame* ack() { return visitor_.ack(); } |
346 | 318 |
347 QuicCongestionFeedbackFrame* feedback() { return feedback_.get(); } | 319 QuicCongestionFeedbackFrame* feedback() { return visitor_.feedback(); } |
| 320 |
| 321 QuicConnectionCloseFrame* close() { return visitor_.close(); } |
348 | 322 |
349 const vector<QuicStreamFrame>* stream_frames() const { | 323 const vector<QuicStreamFrame>* stream_frames() const { |
350 return &stream_frames_; | 324 return visitor_.stream_frames(); |
351 } | 325 } |
352 | 326 |
353 size_t last_packet_size() { | 327 size_t last_packet_size() { |
354 return last_packet_size_; | 328 return last_packet_size_; |
355 } | 329 } |
356 | 330 |
357 QuicVersionNegotiationPacket* version_negotiation_packet() { | 331 QuicVersionNegotiationPacket* version_negotiation_packet() { |
358 return version_negotiation_packet_.get(); | 332 return visitor_.version_negotiation_packet(); |
359 } | 333 } |
360 | 334 |
361 void set_blocked(bool blocked) { blocked_ = blocked; } | 335 void set_blocked(bool blocked) { blocked_ = blocked; } |
362 | 336 |
363 void set_is_write_blocked_data_buffered(bool buffered) { | 337 void set_is_write_blocked_data_buffered(bool buffered) { |
364 is_write_blocked_data_buffered_ = buffered; | 338 is_write_blocked_data_buffered_ = buffered; |
365 } | 339 } |
366 | 340 |
367 void set_is_server(bool is_server) { is_server_ = is_server; } | 341 void set_is_server(bool is_server) { is_server_ = is_server; } |
368 | 342 |
369 // final_bytes_of_last_packet_ returns the last four bytes of the previous | 343 // final_bytes_of_last_packet_ returns the last four bytes of the previous |
370 // packet as a little-endian, uint32. This is intended to be used with a | 344 // packet as a little-endian, uint32. This is intended to be used with a |
371 // TaggingEncrypter so that tests can determine which encrypter was used for | 345 // TaggingEncrypter so that tests can determine which encrypter was used for |
372 // a given packet. | 346 // a given packet. |
373 uint32 final_bytes_of_last_packet() { return final_bytes_of_last_packet_; } | 347 uint32 final_bytes_of_last_packet() { return final_bytes_of_last_packet_; } |
374 | 348 |
375 void use_tagging_decrypter() { | 349 void use_tagging_decrypter() { |
376 use_tagging_decrypter_ = true; | 350 use_tagging_decrypter_ = true; |
377 } | 351 } |
378 | 352 |
379 uint32 packets_write_attempts() { return packets_write_attempts_; } | 353 uint32 packets_write_attempts() { return packets_write_attempts_; } |
380 | 354 |
381 private: | 355 private: |
382 QuicPacketHeader header_; | 356 FramerVisitorCapturingFrames visitor_; |
383 size_t frame_count_; | |
384 scoped_ptr<QuicAckFrame> ack_; | |
385 scoped_ptr<QuicCongestionFeedbackFrame> feedback_; | |
386 vector<QuicStreamFrame> stream_frames_; | |
387 vector<string*> stream_data_; | |
388 scoped_ptr<QuicVersionNegotiationPacket> version_negotiation_packet_; | |
389 size_t last_packet_size_; | 357 size_t last_packet_size_; |
390 bool blocked_; | 358 bool blocked_; |
391 bool is_write_blocked_data_buffered_; | 359 bool is_write_blocked_data_buffered_; |
392 bool is_server_; | 360 bool is_server_; |
393 uint32 final_bytes_of_last_packet_; | 361 uint32 final_bytes_of_last_packet_; |
394 bool use_tagging_decrypter_; | 362 bool use_tagging_decrypter_; |
395 uint32 packets_write_attempts_; | 363 uint32 packets_write_attempts_; |
396 | 364 |
397 DISALLOW_COPY_AND_ASSIGN(TestPacketWriter); | 365 DISALLOW_COPY_AND_ASSIGN(TestPacketWriter); |
398 }; | 366 }; |
(...skipping 126 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
525 : guid_(42), | 493 : guid_(42), |
526 framer_(QuicSupportedVersions(), QuicTime::Zero(), false), | 494 framer_(QuicSupportedVersions(), QuicTime::Zero(), false), |
527 creator_(guid_, &framer_, QuicRandom::GetInstance(), false), | 495 creator_(guid_, &framer_, QuicRandom::GetInstance(), false), |
528 send_algorithm_(new StrictMock<MockSendAlgorithm>), | 496 send_algorithm_(new StrictMock<MockSendAlgorithm>), |
529 helper_(new TestConnectionHelper(&clock_, &random_generator_)), | 497 helper_(new TestConnectionHelper(&clock_, &random_generator_)), |
530 writer_(new TestPacketWriter()), | 498 writer_(new TestPacketWriter()), |
531 connection_(guid_, IPEndPoint(), helper_.get(), writer_.get(), false), | 499 connection_(guid_, IPEndPoint(), helper_.get(), writer_.get(), false), |
532 frame1_(1, false, 0, MakeIOVector(data1)), | 500 frame1_(1, false, 0, MakeIOVector(data1)), |
533 frame2_(1, false, 3, MakeIOVector(data2)), | 501 frame2_(1, false, 3, MakeIOVector(data2)), |
534 accept_packet_(true) { | 502 accept_packet_(true) { |
535 // TODO(rtenneti): remove g_* flags. | |
536 FLAGS_track_retransmission_history = true; | |
537 connection_.set_visitor(&visitor_); | 503 connection_.set_visitor(&visitor_); |
538 connection_.SetSendAlgorithm(send_algorithm_); | 504 connection_.SetSendAlgorithm(send_algorithm_); |
539 framer_.set_received_entropy_calculator(&entropy_calculator_); | 505 framer_.set_received_entropy_calculator(&entropy_calculator_); |
540 // Simplify tests by not sending feedback unless specifically configured. | 506 // Simplify tests by not sending feedback unless specifically configured. |
541 SetFeedback(NULL); | 507 SetFeedback(NULL); |
542 EXPECT_CALL( | 508 EXPECT_CALL( |
543 *send_algorithm_, TimeUntilSend(_, _, _, _)).WillRepeatedly(Return( | 509 *send_algorithm_, TimeUntilSend(_, _, _, _)).WillRepeatedly(Return( |
544 QuicTime::Delta::Zero())); | 510 QuicTime::Delta::Zero())); |
545 EXPECT_CALL(*receive_algorithm_, | 511 EXPECT_CALL(*receive_algorithm_, |
546 RecordIncomingPacket(_, _, _, _)).Times(AnyNumber()); | 512 RecordIncomingPacket(_, _, _, _)).Times(AnyNumber()); |
547 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)) | 513 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)) |
548 .Times(AnyNumber()); | 514 .Times(AnyNumber()); |
549 EXPECT_CALL(*send_algorithm_, RetransmissionDelay()).WillRepeatedly( | 515 EXPECT_CALL(*send_algorithm_, RetransmissionDelay()).WillRepeatedly( |
550 Return(QuicTime::Delta::Zero())); | 516 Return(QuicTime::Delta::Zero())); |
551 EXPECT_CALL(*send_algorithm_, BandwidthEstimate()).WillRepeatedly(Return( | 517 EXPECT_CALL(*send_algorithm_, BandwidthEstimate()).WillRepeatedly(Return( |
552 QuicBandwidth::FromKBitsPerSecond(100))); | 518 QuicBandwidth::FromKBitsPerSecond(100))); |
553 EXPECT_CALL(*send_algorithm_, SmoothedRtt()).WillRepeatedly(Return( | 519 EXPECT_CALL(*send_algorithm_, SmoothedRtt()).WillRepeatedly(Return( |
554 QuicTime::Delta::FromMilliseconds(100))); | 520 QuicTime::Delta::FromMilliseconds(100))); |
555 ON_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)) | 521 ON_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)) |
556 .WillByDefault(Return(true)); | 522 .WillByDefault(Return(true)); |
557 EXPECT_CALL(visitor_, HasPendingHandshake()).Times(AnyNumber()); | 523 EXPECT_CALL(visitor_, HasPendingHandshake()).Times(AnyNumber()); |
558 EXPECT_CALL(visitor_, OnCanWrite()).Times(AnyNumber()).WillRepeatedly( | 524 EXPECT_CALL(visitor_, OnCanWrite()).Times(AnyNumber()).WillRepeatedly( |
559 Return(true)); | 525 Return(true)); |
560 } | 526 } |
561 | 527 |
562 void SetUp() { | |
563 FLAGS_bundle_ack_with_outgoing_packet = GetParam(); | |
564 } | |
565 | |
566 QuicAckFrame* outgoing_ack() { | 528 QuicAckFrame* outgoing_ack() { |
567 outgoing_ack_.reset(QuicConnectionPeer::CreateAckFrame(&connection_)); | 529 outgoing_ack_.reset(QuicConnectionPeer::CreateAckFrame(&connection_)); |
568 return outgoing_ack_.get(); | 530 return outgoing_ack_.get(); |
569 } | 531 } |
570 | 532 |
571 QuicAckFrame* last_ack() { | 533 QuicAckFrame* last_ack() { |
572 return writer_->ack(); | 534 return writer_->ack(); |
573 } | 535 } |
574 | 536 |
575 QuicCongestionFeedbackFrame* last_feedback() { | 537 QuicCongestionFeedbackFrame* last_feedback() { |
576 return writer_->feedback(); | 538 return writer_->feedback(); |
577 } | 539 } |
578 | 540 |
| 541 QuicConnectionCloseFrame* last_close() { |
| 542 return writer_->close(); |
| 543 } |
| 544 |
579 QuicPacketHeader* last_header() { | 545 QuicPacketHeader* last_header() { |
580 return writer_->header(); | 546 return writer_->header(); |
581 } | 547 } |
582 | 548 |
583 size_t last_sent_packet_size() { | 549 size_t last_sent_packet_size() { |
584 return writer_->last_packet_size(); | 550 return writer_->last_packet_size(); |
585 } | 551 } |
586 | 552 |
587 uint32 final_bytes_of_last_packet() { | 553 uint32 final_bytes_of_last_packet() { |
588 return writer_->final_bytes_of_last_packet(); | 554 return writer_->final_bytes_of_last_packet(); |
(...skipping 226 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
815 QuicPacketHeader header_; | 781 QuicPacketHeader header_; |
816 QuicStreamFrame frame1_; | 782 QuicStreamFrame frame1_; |
817 QuicStreamFrame frame2_; | 783 QuicStreamFrame frame2_; |
818 scoped_ptr<QuicAckFrame> outgoing_ack_; | 784 scoped_ptr<QuicAckFrame> outgoing_ack_; |
819 bool accept_packet_; | 785 bool accept_packet_; |
820 | 786 |
821 private: | 787 private: |
822 DISALLOW_COPY_AND_ASSIGN(QuicConnectionTest); | 788 DISALLOW_COPY_AND_ASSIGN(QuicConnectionTest); |
823 }; | 789 }; |
824 | 790 |
825 INSTANTIATE_TEST_CASE_P(BundleAckWithPacket, | 791 TEST_F(QuicConnectionTest, PacketsInOrder) { |
826 QuicConnectionTest, | |
827 ::testing::Values(false, true)); | |
828 | |
829 TEST_P(QuicConnectionTest, PacketsInOrder) { | |
830 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | 792 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); |
831 | 793 |
832 ProcessPacket(1); | 794 ProcessPacket(1); |
833 EXPECT_EQ(1u, outgoing_ack()->received_info.largest_observed); | 795 EXPECT_EQ(1u, outgoing_ack()->received_info.largest_observed); |
834 EXPECT_EQ(0u, outgoing_ack()->received_info.missing_packets.size()); | 796 EXPECT_EQ(0u, outgoing_ack()->received_info.missing_packets.size()); |
835 | 797 |
836 ProcessPacket(2); | 798 ProcessPacket(2); |
837 EXPECT_EQ(2u, outgoing_ack()->received_info.largest_observed); | 799 EXPECT_EQ(2u, outgoing_ack()->received_info.largest_observed); |
838 EXPECT_EQ(0u, outgoing_ack()->received_info.missing_packets.size()); | 800 EXPECT_EQ(0u, outgoing_ack()->received_info.missing_packets.size()); |
839 | 801 |
840 ProcessPacket(3); | 802 ProcessPacket(3); |
841 EXPECT_EQ(3u, outgoing_ack()->received_info.largest_observed); | 803 EXPECT_EQ(3u, outgoing_ack()->received_info.largest_observed); |
842 EXPECT_EQ(0u, outgoing_ack()->received_info.missing_packets.size()); | 804 EXPECT_EQ(0u, outgoing_ack()->received_info.missing_packets.size()); |
843 } | 805 } |
844 | 806 |
845 TEST_P(QuicConnectionTest, PacketsRejected) { | 807 TEST_F(QuicConnectionTest, PacketsRejected) { |
846 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | 808 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); |
847 | 809 |
848 ProcessPacket(1); | 810 ProcessPacket(1); |
849 EXPECT_EQ(1u, outgoing_ack()->received_info.largest_observed); | 811 EXPECT_EQ(1u, outgoing_ack()->received_info.largest_observed); |
850 EXPECT_EQ(0u, outgoing_ack()->received_info.missing_packets.size()); | 812 EXPECT_EQ(0u, outgoing_ack()->received_info.missing_packets.size()); |
851 | 813 |
852 accept_packet_ = false; | 814 accept_packet_ = false; |
853 ProcessPacket(2); | 815 ProcessPacket(2); |
854 // We should not have an ack for two. | 816 // We should not have an ack for two. |
855 EXPECT_EQ(1u, outgoing_ack()->received_info.largest_observed); | 817 EXPECT_EQ(1u, outgoing_ack()->received_info.largest_observed); |
856 EXPECT_EQ(0u, outgoing_ack()->received_info.missing_packets.size()); | 818 EXPECT_EQ(0u, outgoing_ack()->received_info.missing_packets.size()); |
857 } | 819 } |
858 | 820 |
859 TEST_P(QuicConnectionTest, PacketsOutOfOrder) { | 821 TEST_F(QuicConnectionTest, PacketsOutOfOrder) { |
860 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | 822 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); |
861 | 823 |
862 ProcessPacket(3); | 824 ProcessPacket(3); |
863 EXPECT_EQ(3u, outgoing_ack()->received_info.largest_observed); | 825 EXPECT_EQ(3u, outgoing_ack()->received_info.largest_observed); |
864 EXPECT_TRUE(IsMissing(2)); | 826 EXPECT_TRUE(IsMissing(2)); |
865 EXPECT_TRUE(IsMissing(1)); | 827 EXPECT_TRUE(IsMissing(1)); |
866 | 828 |
867 ProcessPacket(2); | 829 ProcessPacket(2); |
868 EXPECT_EQ(3u, outgoing_ack()->received_info.largest_observed); | 830 EXPECT_EQ(3u, outgoing_ack()->received_info.largest_observed); |
869 EXPECT_FALSE(IsMissing(2)); | 831 EXPECT_FALSE(IsMissing(2)); |
870 EXPECT_TRUE(IsMissing(1)); | 832 EXPECT_TRUE(IsMissing(1)); |
871 | 833 |
872 ProcessPacket(1); | 834 ProcessPacket(1); |
873 EXPECT_EQ(3u, outgoing_ack()->received_info.largest_observed); | 835 EXPECT_EQ(3u, outgoing_ack()->received_info.largest_observed); |
874 EXPECT_FALSE(IsMissing(2)); | 836 EXPECT_FALSE(IsMissing(2)); |
875 EXPECT_FALSE(IsMissing(1)); | 837 EXPECT_FALSE(IsMissing(1)); |
876 } | 838 } |
877 | 839 |
878 TEST_P(QuicConnectionTest, DuplicatePacket) { | 840 TEST_F(QuicConnectionTest, DuplicatePacket) { |
879 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | 841 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); |
880 | 842 |
881 ProcessPacket(3); | 843 ProcessPacket(3); |
882 EXPECT_EQ(3u, outgoing_ack()->received_info.largest_observed); | 844 EXPECT_EQ(3u, outgoing_ack()->received_info.largest_observed); |
883 EXPECT_TRUE(IsMissing(2)); | 845 EXPECT_TRUE(IsMissing(2)); |
884 EXPECT_TRUE(IsMissing(1)); | 846 EXPECT_TRUE(IsMissing(1)); |
885 | 847 |
886 // Send packet 3 again, but do not set the expectation that | 848 // Send packet 3 again, but do not set the expectation that |
887 // the visitor OnStreamFrames() will be called. | 849 // the visitor OnStreamFrames() will be called. |
888 ProcessDataPacket(3, 0, !kEntropyFlag); | 850 ProcessDataPacket(3, 0, !kEntropyFlag); |
889 EXPECT_EQ(3u, outgoing_ack()->received_info.largest_observed); | 851 EXPECT_EQ(3u, outgoing_ack()->received_info.largest_observed); |
890 EXPECT_TRUE(IsMissing(2)); | 852 EXPECT_TRUE(IsMissing(2)); |
891 EXPECT_TRUE(IsMissing(1)); | 853 EXPECT_TRUE(IsMissing(1)); |
892 } | 854 } |
893 | 855 |
894 TEST_P(QuicConnectionTest, PacketsOutOfOrderWithAdditionsAndLeastAwaiting) { | 856 TEST_F(QuicConnectionTest, PacketsOutOfOrderWithAdditionsAndLeastAwaiting) { |
895 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | 857 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); |
896 | 858 |
897 ProcessPacket(3); | 859 ProcessPacket(3); |
898 EXPECT_EQ(3u, outgoing_ack()->received_info.largest_observed); | 860 EXPECT_EQ(3u, outgoing_ack()->received_info.largest_observed); |
899 EXPECT_TRUE(IsMissing(2)); | 861 EXPECT_TRUE(IsMissing(2)); |
900 EXPECT_TRUE(IsMissing(1)); | 862 EXPECT_TRUE(IsMissing(1)); |
901 | 863 |
902 ProcessPacket(2); | 864 ProcessPacket(2); |
903 EXPECT_EQ(3u, outgoing_ack()->received_info.largest_observed); | 865 EXPECT_EQ(3u, outgoing_ack()->received_info.largest_observed); |
904 EXPECT_TRUE(IsMissing(1)); | 866 EXPECT_TRUE(IsMissing(1)); |
905 | 867 |
906 ProcessPacket(5); | 868 ProcessPacket(5); |
907 EXPECT_EQ(5u, outgoing_ack()->received_info.largest_observed); | 869 EXPECT_EQ(5u, outgoing_ack()->received_info.largest_observed); |
908 EXPECT_TRUE(IsMissing(1)); | 870 EXPECT_TRUE(IsMissing(1)); |
909 EXPECT_TRUE(IsMissing(4)); | 871 EXPECT_TRUE(IsMissing(4)); |
910 | 872 |
911 // Pretend at this point the client has gotten acks for 2 and 3 and 1 is a | 873 // Pretend at this point the client has gotten acks for 2 and 3 and 1 is a |
912 // packet the peer will not retransmit. It indicates this by sending 'least | 874 // packet the peer will not retransmit. It indicates this by sending 'least |
913 // awaiting' is 4. The connection should then realize 1 will not be | 875 // awaiting' is 4. The connection should then realize 1 will not be |
914 // retransmitted, and will remove it from the missing list. | 876 // retransmitted, and will remove it from the missing list. |
915 creator_.set_sequence_number(5); | 877 creator_.set_sequence_number(5); |
916 QuicAckFrame frame(0, QuicTime::Zero(), 4); | 878 QuicAckFrame frame(0, QuicTime::Zero(), 4); |
917 ProcessAckPacket(&frame); | 879 ProcessAckPacket(&frame); |
918 | 880 |
919 // Force an ack to be sent. | 881 // Force an ack to be sent. |
920 SendAckPacketToPeer(); | 882 SendAckPacketToPeer(); |
921 EXPECT_TRUE(IsMissing(4)); | 883 EXPECT_TRUE(IsMissing(4)); |
922 } | 884 } |
923 | 885 |
924 TEST_P(QuicConnectionTest, RejectPacketTooFarOut) { | 886 TEST_F(QuicConnectionTest, RejectPacketTooFarOut) { |
925 // Call ProcessDataPacket rather than ProcessPacket, as we should not get a | 887 // Call ProcessDataPacket rather than ProcessPacket, as we should not get a |
926 // packet call to the visitor. | 888 // packet call to the visitor. |
927 EXPECT_CALL(visitor_, OnConnectionClosed(QUIC_INVALID_PACKET_HEADER, false)); | 889 EXPECT_CALL(visitor_, OnConnectionClosed(QUIC_INVALID_PACKET_HEADER, false)); |
928 ProcessDataPacket(6000, 0, !kEntropyFlag); | 890 ProcessDataPacket(6000, 0, !kEntropyFlag); |
929 } | 891 } |
930 | 892 |
931 // TODO(rtenneti): Delete this when QUIC_VERSION_11 is deprecated. | 893 // TODO(rtenneti): Delete this when QUIC_VERSION_11 is deprecated. |
932 TEST_P(QuicConnectionTest, TruncatedAck11) { | 894 TEST_F(QuicConnectionTest, TruncatedAck11) { |
933 if (QuicVersionMax() > QUIC_VERSION_11) { | 895 if (QuicVersionMax() > QUIC_VERSION_11) { |
934 return; | 896 return; |
935 } | 897 } |
936 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | 898 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); |
937 EXPECT_CALL(*send_algorithm_, OnIncomingAck(_, _, _)).Times(2); | 899 EXPECT_CALL(*send_algorithm_, OnIncomingAck(_, _, _)).Times(2); |
938 EXPECT_CALL(*send_algorithm_, OnIncomingLoss(_, _)).Times(1); | 900 EXPECT_CALL(*send_algorithm_, OnIncomingLoss(_, _)).Times(1); |
939 for (int i = 0; i < 200; ++i) { | 901 for (int i = 0; i < 200; ++i) { |
940 SendStreamDataToPeer(1, "foo", i * 3, !kFin, NULL); | 902 SendStreamDataToPeer(1, "foo", i * 3, !kFin, NULL); |
941 } | 903 } |
942 | 904 |
(...skipping 10 matching lines...) Expand all Loading... |
953 | 915 |
954 frame.received_info.missing_packets.erase(192); | 916 frame.received_info.missing_packets.erase(192); |
955 frame.received_info.entropy_hash = | 917 frame.received_info.entropy_hash = |
956 QuicConnectionPeer::GetSentEntropyHash(&connection_, 193) ^ | 918 QuicConnectionPeer::GetSentEntropyHash(&connection_, 193) ^ |
957 QuicConnectionPeer::GetSentEntropyHash(&connection_, 191); | 919 QuicConnectionPeer::GetSentEntropyHash(&connection_, 191); |
958 | 920 |
959 ProcessAckPacket(&frame); | 921 ProcessAckPacket(&frame); |
960 EXPECT_FALSE(QuicConnectionPeer::GetReceivedTruncatedAck(&connection_)); | 922 EXPECT_FALSE(QuicConnectionPeer::GetReceivedTruncatedAck(&connection_)); |
961 } | 923 } |
962 | 924 |
963 TEST_P(QuicConnectionTest, TruncatedAck) { | 925 TEST_F(QuicConnectionTest, TruncatedAck) { |
964 // TODO(rtenneti): Delete this when QUIC_VERSION_11 is deprecated. | 926 // TODO(rtenneti): Delete this when QUIC_VERSION_11 is deprecated. |
965 if (QuicVersionMax() <= QUIC_VERSION_11) { | 927 if (QuicVersionMax() <= QUIC_VERSION_11) { |
966 return; | 928 return; |
967 } | 929 } |
968 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | 930 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); |
969 EXPECT_CALL(*send_algorithm_, OnIncomingAck(_, _, _)).Times(256); | 931 EXPECT_CALL(*send_algorithm_, OnIncomingAck(_, _, _)).Times(256); |
970 EXPECT_CALL(*send_algorithm_, OnIncomingLoss(_, _)).Times(2); | 932 EXPECT_CALL(*send_algorithm_, OnIncomingLoss(_, _)).Times(2); |
971 int num_packets = 256 * 2 + 1; | 933 int num_packets = 256 * 2 + 1; |
972 for (int i = 0; i < num_packets; ++i) { | 934 for (int i = 0; i < num_packets; ++i) { |
973 SendStreamDataToPeer(1, "foo", i * 3, !kFin, NULL); | 935 SendStreamDataToPeer(1, "foo", i * 3, !kFin, NULL); |
(...skipping 11 matching lines...) Expand all Loading... |
985 EXPECT_TRUE(QuicConnectionPeer::GetReceivedTruncatedAck(&connection_)); | 947 EXPECT_TRUE(QuicConnectionPeer::GetReceivedTruncatedAck(&connection_)); |
986 | 948 |
987 frame.received_info.missing_packets.erase(192); | 949 frame.received_info.missing_packets.erase(192); |
988 | 950 |
989 // Removing one missing packet allows us to ack 192 and one more range. | 951 // Removing one missing packet allows us to ack 192 and one more range. |
990 EXPECT_CALL(*send_algorithm_, OnIncomingAck(_, _, _)).Times(2); | 952 EXPECT_CALL(*send_algorithm_, OnIncomingAck(_, _, _)).Times(2); |
991 ProcessAckPacket(&frame); | 953 ProcessAckPacket(&frame); |
992 EXPECT_FALSE(QuicConnectionPeer::GetReceivedTruncatedAck(&connection_)); | 954 EXPECT_FALSE(QuicConnectionPeer::GetReceivedTruncatedAck(&connection_)); |
993 } | 955 } |
994 | 956 |
995 TEST_P(QuicConnectionTest, AckReceiptCausesAckSendBadEntropy) { | 957 TEST_F(QuicConnectionTest, AckReceiptCausesAckSendBadEntropy) { |
996 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | 958 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); |
997 | 959 |
998 ProcessPacket(1); | 960 ProcessPacket(1); |
999 // Delay sending, then queue up an ack. | 961 // Delay sending, then queue up an ack. |
1000 EXPECT_CALL(*send_algorithm_, | 962 EXPECT_CALL(*send_algorithm_, |
1001 TimeUntilSend(_, NOT_RETRANSMISSION, _, _)).WillOnce( | 963 TimeUntilSend(_, NOT_RETRANSMISSION, _, _)).WillOnce( |
1002 testing::Return(QuicTime::Delta::FromMicroseconds(1))); | 964 testing::Return(QuicTime::Delta::FromMicroseconds(1))); |
1003 QuicConnectionPeer::SendAck(&connection_); | 965 QuicConnectionPeer::SendAck(&connection_); |
1004 | 966 |
1005 // Process an ack with a least unacked of the received ack. | 967 // Process an ack with a least unacked of the received ack. |
1006 // This causes an ack to be sent when TimeUntilSend returns 0. | 968 // This causes an ack to be sent when TimeUntilSend returns 0. |
1007 EXPECT_CALL(*send_algorithm_, | 969 EXPECT_CALL(*send_algorithm_, |
1008 TimeUntilSend(_, NOT_RETRANSMISSION, _, _)).WillRepeatedly( | 970 TimeUntilSend(_, NOT_RETRANSMISSION, _, _)).WillRepeatedly( |
1009 testing::Return(QuicTime::Delta::Zero())); | 971 testing::Return(QuicTime::Delta::Zero())); |
1010 // Skip a packet and then record an ack. | 972 // Skip a packet and then record an ack. |
1011 creator_.set_sequence_number(2); | 973 creator_.set_sequence_number(2); |
1012 QuicAckFrame frame(0, QuicTime::Zero(), 3); | 974 QuicAckFrame frame(0, QuicTime::Zero(), 3); |
1013 ProcessAckPacket(&frame); | 975 ProcessAckPacket(&frame); |
1014 } | 976 } |
1015 | 977 |
1016 TEST_P(QuicConnectionTest, OutOfOrderReceiptCausesAckSend) { | 978 TEST_F(QuicConnectionTest, OutOfOrderReceiptCausesAckSend) { |
1017 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | 979 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); |
1018 | 980 |
1019 ProcessPacket(3); | 981 ProcessPacket(3); |
1020 // Should ack immediately since we have missing packets. | 982 // Should ack immediately since we have missing packets. |
1021 EXPECT_EQ(1u, writer_->packets_write_attempts()); | 983 EXPECT_EQ(1u, writer_->packets_write_attempts()); |
1022 | 984 |
1023 ProcessPacket(2); | 985 ProcessPacket(2); |
1024 // Should ack immediately since we have missing packets. | 986 // Should ack immediately since we have missing packets. |
1025 EXPECT_EQ(2u, writer_->packets_write_attempts()); | 987 EXPECT_EQ(2u, writer_->packets_write_attempts()); |
1026 | 988 |
1027 ProcessPacket(1); | 989 ProcessPacket(1); |
1028 // Should ack immediately, since this fills the last hole. | 990 // Should ack immediately, since this fills the last hole. |
1029 EXPECT_EQ(3u, writer_->packets_write_attempts()); | 991 EXPECT_EQ(3u, writer_->packets_write_attempts()); |
1030 | 992 |
1031 ProcessPacket(4); | 993 ProcessPacket(4); |
1032 // Should not cause an ack. | 994 // Should not cause an ack. |
1033 EXPECT_EQ(3u, writer_->packets_write_attempts()); | 995 EXPECT_EQ(3u, writer_->packets_write_attempts()); |
1034 } | 996 } |
1035 | 997 |
1036 TEST_P(QuicConnectionTest, AckReceiptCausesAckSend) { | 998 TEST_F(QuicConnectionTest, AckReceiptCausesAckSend) { |
1037 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | 999 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); |
1038 EXPECT_CALL(*send_algorithm_, OnIncomingLoss(_, _)).Times(1); | 1000 EXPECT_CALL(*send_algorithm_, OnIncomingLoss(_, _)).Times(1); |
1039 QuicPacketSequenceNumber original; | 1001 QuicPacketSequenceNumber original; |
1040 QuicByteCount packet_size; | 1002 QuicByteCount packet_size; |
1041 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, NOT_RETRANSMISSION, _)) | 1003 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, NOT_RETRANSMISSION, _)) |
1042 .WillOnce(DoAll(SaveArg<1>(&original), SaveArg<2>(&packet_size), | 1004 .WillOnce(DoAll(SaveArg<1>(&original), SaveArg<2>(&packet_size), |
1043 Return(true))); | 1005 Return(true))); |
1044 EXPECT_CALL(*send_algorithm_, OnPacketAbandoned(1, _)).Times(1); | 1006 EXPECT_CALL(*send_algorithm_, OnPacketAbandoned(1, _)).Times(1); |
1045 connection_.SendStreamDataWithString(3, "foo", 0, !kFin, NULL); | 1007 connection_.SendStreamDataWithString(3, "foo", 0, !kFin, NULL); |
1046 QuicAckFrame frame(original, QuicTime::Zero(), 1); | 1008 QuicAckFrame frame(original, QuicTime::Zero(), 1); |
(...skipping 24 matching lines...) Expand all Loading... |
1071 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, NOT_RETRANSMISSION, _)); | 1033 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, NOT_RETRANSMISSION, _)); |
1072 ProcessAckPacket(&frame2); | 1034 ProcessAckPacket(&frame2); |
1073 ProcessAckPacket(&frame2); | 1035 ProcessAckPacket(&frame2); |
1074 | 1036 |
1075 // But an ack with no missing packets will not send an ack. | 1037 // But an ack with no missing packets will not send an ack. |
1076 frame2.received_info.missing_packets.clear(); | 1038 frame2.received_info.missing_packets.clear(); |
1077 ProcessAckPacket(&frame2); | 1039 ProcessAckPacket(&frame2); |
1078 ProcessAckPacket(&frame2); | 1040 ProcessAckPacket(&frame2); |
1079 } | 1041 } |
1080 | 1042 |
1081 TEST_P(QuicConnectionTest, LeastUnackedLower) { | 1043 TEST_F(QuicConnectionTest, LeastUnackedLower) { |
1082 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | 1044 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); |
1083 | 1045 |
1084 SendStreamDataToPeer(1, "foo", 0, !kFin, NULL); | 1046 SendStreamDataToPeer(1, "foo", 0, !kFin, NULL); |
1085 SendStreamDataToPeer(1, "bar", 3, !kFin, NULL); | 1047 SendStreamDataToPeer(1, "bar", 3, !kFin, NULL); |
1086 SendStreamDataToPeer(1, "eep", 6, !kFin, NULL); | 1048 SendStreamDataToPeer(1, "eep", 6, !kFin, NULL); |
1087 | 1049 |
1088 // Start out saying the least unacked is 2. | 1050 // Start out saying the least unacked is 2. |
1089 creator_.set_sequence_number(5); | 1051 creator_.set_sequence_number(5); |
1090 QuicAckFrame frame(0, QuicTime::Zero(), 2); | 1052 QuicAckFrame frame(0, QuicTime::Zero(), 2); |
1091 ProcessAckPacket(&frame); | 1053 ProcessAckPacket(&frame); |
1092 | 1054 |
1093 // Change it to 1, but lower the sequence number to fake out-of-order packets. | 1055 // Change it to 1, but lower the sequence number to fake out-of-order packets. |
1094 // This should be fine. | 1056 // This should be fine. |
1095 creator_.set_sequence_number(1); | 1057 creator_.set_sequence_number(1); |
1096 QuicAckFrame frame2(0, QuicTime::Zero(), 1); | 1058 QuicAckFrame frame2(0, QuicTime::Zero(), 1); |
1097 // The scheduler will not process out of order acks. | 1059 // The scheduler will not process out of order acks. |
1098 EXPECT_CALL(visitor_, OnCanWrite()).Times(0); | 1060 EXPECT_CALL(visitor_, OnCanWrite()).Times(0); |
1099 ProcessAckPacket(&frame2); | 1061 ProcessAckPacket(&frame2); |
1100 | 1062 |
1101 // Now claim it's one, but set the ordering so it was sent "after" the first | 1063 // Now claim it's one, but set the ordering so it was sent "after" the first |
1102 // one. This should cause a connection error. | 1064 // one. This should cause a connection error. |
1103 EXPECT_CALL(visitor_, OnConnectionClosed(QUIC_INVALID_ACK_DATA, false)); | 1065 EXPECT_CALL(visitor_, OnConnectionClosed(QUIC_INVALID_ACK_DATA, false)); |
1104 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)); | 1066 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)); |
1105 creator_.set_sequence_number(7); | 1067 creator_.set_sequence_number(7); |
1106 ProcessAckPacket(&frame2); | 1068 ProcessAckPacket(&frame2); |
1107 } | 1069 } |
1108 | 1070 |
1109 TEST_P(QuicConnectionTest, LargestObservedLower) { | 1071 TEST_F(QuicConnectionTest, LargestObservedLower) { |
1110 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | 1072 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); |
1111 | 1073 |
1112 SendStreamDataToPeer(1, "foo", 0, !kFin, NULL); | 1074 SendStreamDataToPeer(1, "foo", 0, !kFin, NULL); |
1113 SendStreamDataToPeer(1, "bar", 3, !kFin, NULL); | 1075 SendStreamDataToPeer(1, "bar", 3, !kFin, NULL); |
1114 SendStreamDataToPeer(1, "eep", 6, !kFin, NULL); | 1076 SendStreamDataToPeer(1, "eep", 6, !kFin, NULL); |
1115 EXPECT_CALL(*send_algorithm_, OnIncomingAck(_, _, _)).Times(2); | 1077 EXPECT_CALL(*send_algorithm_, OnIncomingAck(_, _, _)).Times(2); |
1116 | 1078 |
1117 // Start out saying the largest observed is 2. | 1079 // Start out saying the largest observed is 2. |
1118 QuicAckFrame frame(2, QuicTime::Zero(), 0); | 1080 QuicAckFrame frame(2, QuicTime::Zero(), 0); |
1119 frame.received_info.entropy_hash = QuicConnectionPeer::GetSentEntropyHash( | 1081 frame.received_info.entropy_hash = QuicConnectionPeer::GetSentEntropyHash( |
1120 &connection_, 2); | 1082 &connection_, 2); |
1121 ProcessAckPacket(&frame); | 1083 ProcessAckPacket(&frame); |
1122 | 1084 |
1123 // Now change it to 1, and it should cause a connection error. | 1085 // Now change it to 1, and it should cause a connection error. |
1124 QuicAckFrame frame2(1, QuicTime::Zero(), 0); | 1086 QuicAckFrame frame2(1, QuicTime::Zero(), 0); |
1125 EXPECT_CALL(visitor_, OnConnectionClosed(QUIC_INVALID_ACK_DATA, false)); | 1087 EXPECT_CALL(visitor_, OnConnectionClosed(QUIC_INVALID_ACK_DATA, false)); |
1126 EXPECT_CALL(visitor_, OnCanWrite()).Times(0); | 1088 EXPECT_CALL(visitor_, OnCanWrite()).Times(0); |
1127 ProcessAckPacket(&frame2); | 1089 ProcessAckPacket(&frame2); |
1128 } | 1090 } |
1129 | 1091 |
1130 TEST_P(QuicConnectionTest, AckUnsentData) { | 1092 TEST_F(QuicConnectionTest, AckUnsentData) { |
1131 // Ack a packet which has not been sent. | 1093 // Ack a packet which has not been sent. |
1132 EXPECT_CALL(visitor_, OnConnectionClosed(QUIC_INVALID_ACK_DATA, false)); | 1094 EXPECT_CALL(visitor_, OnConnectionClosed(QUIC_INVALID_ACK_DATA, false)); |
1133 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | 1095 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); |
1134 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)); | 1096 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)); |
1135 QuicAckFrame frame(1, QuicTime::Zero(), 0); | 1097 QuicAckFrame frame(1, QuicTime::Zero(), 0); |
1136 EXPECT_CALL(visitor_, OnCanWrite()).Times(0); | 1098 EXPECT_CALL(visitor_, OnCanWrite()).Times(0); |
1137 ProcessAckPacket(&frame); | 1099 ProcessAckPacket(&frame); |
1138 } | 1100 } |
1139 | 1101 |
1140 TEST_P(QuicConnectionTest, AckAll) { | 1102 TEST_F(QuicConnectionTest, AckAll) { |
1141 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | 1103 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); |
1142 ProcessPacket(1); | 1104 ProcessPacket(1); |
1143 | 1105 |
1144 creator_.set_sequence_number(1); | 1106 creator_.set_sequence_number(1); |
1145 QuicAckFrame frame1(0, QuicTime::Zero(), 1); | 1107 QuicAckFrame frame1(0, QuicTime::Zero(), 1); |
1146 ProcessAckPacket(&frame1); | 1108 ProcessAckPacket(&frame1); |
1147 } | 1109 } |
1148 | 1110 |
1149 TEST_P(QuicConnectionTest, SendingDifferentSequenceNumberLengthsBandwidth) { | 1111 TEST_F(QuicConnectionTest, SendingDifferentSequenceNumberLengthsBandwidth) { |
1150 EXPECT_CALL(*send_algorithm_, BandwidthEstimate()).WillOnce(Return( | 1112 EXPECT_CALL(*send_algorithm_, BandwidthEstimate()).WillOnce(Return( |
1151 QuicBandwidth::FromKBitsPerSecond(1000))); | 1113 QuicBandwidth::FromKBitsPerSecond(1000))); |
1152 | 1114 |
1153 QuicPacketSequenceNumber last_packet; | 1115 QuicPacketSequenceNumber last_packet; |
1154 SendStreamDataToPeer(1, "foo", 0, !kFin, &last_packet); | 1116 SendStreamDataToPeer(1, "foo", 0, !kFin, &last_packet); |
1155 EXPECT_EQ(1u, last_packet); | 1117 EXPECT_EQ(1u, last_packet); |
1156 EXPECT_EQ(PACKET_1BYTE_SEQUENCE_NUMBER, | 1118 EXPECT_EQ(PACKET_1BYTE_SEQUENCE_NUMBER, |
1157 connection_.options()->send_sequence_number_length); | 1119 connection_.options()->send_sequence_number_length); |
1158 EXPECT_EQ(PACKET_1BYTE_SEQUENCE_NUMBER, | 1120 EXPECT_EQ(PACKET_1BYTE_SEQUENCE_NUMBER, |
1159 last_header()->public_header.sequence_number_length); | 1121 last_header()->public_header.sequence_number_length); |
(...skipping 34 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
1194 QuicBandwidth::FromKBitsPerSecond(1000ll * 256 * 256 * 256 * 256))); | 1156 QuicBandwidth::FromKBitsPerSecond(1000ll * 256 * 256 * 256 * 256))); |
1195 | 1157 |
1196 SendStreamDataToPeer(1, "foo", 12, !kFin, &last_packet); | 1158 SendStreamDataToPeer(1, "foo", 12, !kFin, &last_packet); |
1197 EXPECT_EQ(5u, last_packet); | 1159 EXPECT_EQ(5u, last_packet); |
1198 EXPECT_EQ(PACKET_6BYTE_SEQUENCE_NUMBER, | 1160 EXPECT_EQ(PACKET_6BYTE_SEQUENCE_NUMBER, |
1199 connection_.options()->send_sequence_number_length); | 1161 connection_.options()->send_sequence_number_length); |
1200 EXPECT_EQ(PACKET_4BYTE_SEQUENCE_NUMBER, | 1162 EXPECT_EQ(PACKET_4BYTE_SEQUENCE_NUMBER, |
1201 last_header()->public_header.sequence_number_length); | 1163 last_header()->public_header.sequence_number_length); |
1202 } | 1164 } |
1203 | 1165 |
1204 TEST_P(QuicConnectionTest, SendingDifferentSequenceNumberLengthsUnackedDelta) { | 1166 TEST_F(QuicConnectionTest, SendingDifferentSequenceNumberLengthsUnackedDelta) { |
1205 QuicPacketSequenceNumber last_packet; | 1167 QuicPacketSequenceNumber last_packet; |
1206 SendStreamDataToPeer(1, "foo", 0, !kFin, &last_packet); | 1168 SendStreamDataToPeer(1, "foo", 0, !kFin, &last_packet); |
1207 EXPECT_EQ(1u, last_packet); | 1169 EXPECT_EQ(1u, last_packet); |
1208 EXPECT_EQ(PACKET_1BYTE_SEQUENCE_NUMBER, | 1170 EXPECT_EQ(PACKET_1BYTE_SEQUENCE_NUMBER, |
1209 connection_.options()->send_sequence_number_length); | 1171 connection_.options()->send_sequence_number_length); |
1210 EXPECT_EQ(PACKET_1BYTE_SEQUENCE_NUMBER, | 1172 EXPECT_EQ(PACKET_1BYTE_SEQUENCE_NUMBER, |
1211 last_header()->public_header.sequence_number_length); | 1173 last_header()->public_header.sequence_number_length); |
1212 | 1174 |
1213 QuicConnectionPeer::GetPacketCreator(&connection_)->set_sequence_number(100); | 1175 QuicConnectionPeer::GetPacketCreator(&connection_)->set_sequence_number(100); |
1214 | 1176 |
(...skipping 24 matching lines...) Expand all Loading... |
1239 QuicConnectionPeer::GetPacketCreator(&connection_)->set_sequence_number( | 1201 QuicConnectionPeer::GetPacketCreator(&connection_)->set_sequence_number( |
1240 100 * 256 * 256 * 256); | 1202 100 * 256 * 256 * 256); |
1241 | 1203 |
1242 SendStreamDataToPeer(1, "foo", 12, !kFin, &last_packet); | 1204 SendStreamDataToPeer(1, "foo", 12, !kFin, &last_packet); |
1243 EXPECT_EQ(PACKET_6BYTE_SEQUENCE_NUMBER, | 1205 EXPECT_EQ(PACKET_6BYTE_SEQUENCE_NUMBER, |
1244 connection_.options()->send_sequence_number_length); | 1206 connection_.options()->send_sequence_number_length); |
1245 EXPECT_EQ(PACKET_4BYTE_SEQUENCE_NUMBER, | 1207 EXPECT_EQ(PACKET_4BYTE_SEQUENCE_NUMBER, |
1246 last_header()->public_header.sequence_number_length); | 1208 last_header()->public_header.sequence_number_length); |
1247 } | 1209 } |
1248 | 1210 |
1249 TEST_P(QuicConnectionTest, BasicSending) { | 1211 TEST_F(QuicConnectionTest, BasicSending) { |
1250 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | 1212 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); |
1251 EXPECT_CALL(*send_algorithm_, OnIncomingAck(_, _, _)).Times(6); | 1213 EXPECT_CALL(*send_algorithm_, OnIncomingAck(_, _, _)).Times(6); |
1252 QuicPacketSequenceNumber last_packet; | 1214 QuicPacketSequenceNumber last_packet; |
1253 SendStreamDataToPeer(1, "foo", 0, !kFin, &last_packet); // Packet 1 | 1215 SendStreamDataToPeer(1, "foo", 0, !kFin, &last_packet); // Packet 1 |
1254 EXPECT_EQ(1u, last_packet); | 1216 EXPECT_EQ(1u, last_packet); |
1255 SendAckPacketToPeer(); // Packet 2 | 1217 SendAckPacketToPeer(); // Packet 2 |
1256 | 1218 |
1257 EXPECT_EQ(1u, last_ack()->sent_info.least_unacked); | 1219 EXPECT_EQ(1u, last_ack()->sent_info.least_unacked); |
1258 | 1220 |
1259 SendAckPacketToPeer(); // Packet 3 | 1221 SendAckPacketToPeer(); // Packet 3 |
(...skipping 31 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
1291 SendAckPacketToPeer(); // Packet 7 | 1253 SendAckPacketToPeer(); // Packet 7 |
1292 EXPECT_EQ(7u, last_ack()->sent_info.least_unacked); | 1254 EXPECT_EQ(7u, last_ack()->sent_info.least_unacked); |
1293 | 1255 |
1294 // But if we send more data it should. | 1256 // But if we send more data it should. |
1295 SendStreamDataToPeer(1, "eep", 6, !kFin, &last_packet); // Packet 8 | 1257 SendStreamDataToPeer(1, "eep", 6, !kFin, &last_packet); // Packet 8 |
1296 EXPECT_EQ(8u, last_packet); | 1258 EXPECT_EQ(8u, last_packet); |
1297 SendAckPacketToPeer(); // Packet 9 | 1259 SendAckPacketToPeer(); // Packet 9 |
1298 EXPECT_EQ(8u, last_ack()->sent_info.least_unacked); | 1260 EXPECT_EQ(8u, last_ack()->sent_info.least_unacked); |
1299 } | 1261 } |
1300 | 1262 |
1301 TEST_P(QuicConnectionTest, FECSending) { | 1263 TEST_F(QuicConnectionTest, FECSending) { |
1302 // All packets carry version info till version is negotiated. | 1264 // All packets carry version info till version is negotiated. |
1303 size_t payload_length; | 1265 size_t payload_length; |
1304 connection_.options()->max_packet_length = | 1266 connection_.options()->max_packet_length = |
1305 GetPacketLengthForOneStream( | 1267 GetPacketLengthForOneStream( |
1306 connection_.version(), kIncludeVersion, PACKET_1BYTE_SEQUENCE_NUMBER, | 1268 connection_.version(), kIncludeVersion, PACKET_1BYTE_SEQUENCE_NUMBER, |
1307 IN_FEC_GROUP, &payload_length); | 1269 IN_FEC_GROUP, &payload_length); |
1308 // And send FEC every two packets. | 1270 // And send FEC every two packets. |
1309 connection_.options()->max_packets_per_fec_group = 2; | 1271 connection_.options()->max_packets_per_fec_group = 2; |
1310 | 1272 |
1311 // Send 4 data packets and 2 FEC packets. | 1273 // Send 4 data packets and 2 FEC packets. |
1312 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)).Times(6); | 1274 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)).Times(6); |
1313 // The first stream frame will consume 2 fewer bytes than the other three. | 1275 // The first stream frame will consume 2 fewer bytes than the other three. |
1314 const string payload(payload_length * 4 - 6, 'a'); | 1276 const string payload(payload_length * 4 - 6, 'a'); |
1315 connection_.SendStreamDataWithString(1, payload, 0, !kFin, NULL); | 1277 connection_.SendStreamDataWithString(1, payload, 0, !kFin, NULL); |
1316 // Expect the FEC group to be closed after SendStreamDataWithString. | 1278 // Expect the FEC group to be closed after SendStreamDataWithString. |
1317 EXPECT_FALSE(creator_.ShouldSendFec(true)); | 1279 EXPECT_FALSE(creator_.ShouldSendFec(true)); |
1318 } | 1280 } |
1319 | 1281 |
1320 TEST_P(QuicConnectionTest, FECQueueing) { | 1282 TEST_F(QuicConnectionTest, FECQueueing) { |
1321 // All packets carry version info till version is negotiated. | 1283 // All packets carry version info till version is negotiated. |
1322 size_t payload_length; | 1284 size_t payload_length; |
1323 connection_.options()->max_packet_length = | 1285 connection_.options()->max_packet_length = |
1324 GetPacketLengthForOneStream( | 1286 GetPacketLengthForOneStream( |
1325 connection_.version(), kIncludeVersion, PACKET_1BYTE_SEQUENCE_NUMBER, | 1287 connection_.version(), kIncludeVersion, PACKET_1BYTE_SEQUENCE_NUMBER, |
1326 IN_FEC_GROUP, &payload_length); | 1288 IN_FEC_GROUP, &payload_length); |
1327 // And send FEC every two packets. | 1289 // And send FEC every two packets. |
1328 connection_.options()->max_packets_per_fec_group = 2; | 1290 connection_.options()->max_packets_per_fec_group = 2; |
1329 | 1291 |
1330 EXPECT_EQ(0u, connection_.NumQueuedPackets()); | 1292 EXPECT_EQ(0u, connection_.NumQueuedPackets()); |
1331 writer_->set_blocked(true); | 1293 writer_->set_blocked(true); |
1332 const string payload(payload_length, 'a'); | 1294 const string payload(payload_length, 'a'); |
1333 connection_.SendStreamDataWithString(1, payload, 0, !kFin, NULL); | 1295 connection_.SendStreamDataWithString(1, payload, 0, !kFin, NULL); |
1334 EXPECT_FALSE(creator_.ShouldSendFec(true)); | 1296 EXPECT_FALSE(creator_.ShouldSendFec(true)); |
1335 // Expect the first data packet and the fec packet to be queued. | 1297 // Expect the first data packet and the fec packet to be queued. |
1336 EXPECT_EQ(2u, connection_.NumQueuedPackets()); | 1298 EXPECT_EQ(2u, connection_.NumQueuedPackets()); |
1337 } | 1299 } |
1338 | 1300 |
1339 TEST_P(QuicConnectionTest, AbandonFECFromCongestionWindow) { | 1301 TEST_F(QuicConnectionTest, AbandonFECFromCongestionWindow) { |
1340 connection_.options()->max_packets_per_fec_group = 1; | 1302 connection_.options()->max_packets_per_fec_group = 1; |
1341 // 1 Data and 1 FEC packet. | 1303 // 1 Data and 1 FEC packet. |
1342 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)).Times(2); | 1304 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)).Times(2); |
1343 connection_.SendStreamDataWithString(1, "foo", 0, !kFin, NULL); | 1305 connection_.SendStreamDataWithString(1, "foo", 0, !kFin, NULL); |
1344 | 1306 |
1345 const QuicTime::Delta retransmission_time = | 1307 const QuicTime::Delta retransmission_time = |
1346 QuicTime::Delta::FromMilliseconds(5000); | 1308 QuicTime::Delta::FromMilliseconds(5000); |
1347 clock_.AdvanceTime(retransmission_time); | 1309 clock_.AdvanceTime(retransmission_time); |
1348 | 1310 |
1349 // Abandon FEC packet. | 1311 // Abandon FEC packet. |
1350 EXPECT_CALL(*send_algorithm_, OnPacketAbandoned(_, _)).Times(1); | 1312 EXPECT_CALL(*send_algorithm_, OnPacketAbandoned(_, _)).Times(1); |
1351 EXPECT_CALL(visitor_, OnCanWrite()); | 1313 EXPECT_CALL(visitor_, OnCanWrite()); |
1352 connection_.OnAbandonFecTimeout(); | 1314 connection_.OnAbandonFecTimeout(); |
1353 } | 1315 } |
1354 | 1316 |
1355 TEST_P(QuicConnectionTest, DontAbandonAckedFEC) { | 1317 TEST_F(QuicConnectionTest, DontAbandonAckedFEC) { |
1356 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | 1318 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); |
1357 connection_.options()->max_packets_per_fec_group = 1; | 1319 connection_.options()->max_packets_per_fec_group = 1; |
1358 const QuicPacketSequenceNumber sequence_number = | 1320 const QuicPacketSequenceNumber sequence_number = |
1359 QuicConnectionPeer::GetPacketCreator(&connection_)->sequence_number() + 1; | 1321 QuicConnectionPeer::GetPacketCreator(&connection_)->sequence_number() + 1; |
1360 | 1322 |
1361 // 1 Data and 1 FEC packet. | 1323 // 1 Data and 1 FEC packet. |
1362 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)).Times(2); | 1324 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)).Times(2); |
1363 connection_.SendStreamDataWithString(1, "foo", 0, !kFin, NULL); | 1325 connection_.SendStreamDataWithString(1, "foo", 0, !kFin, NULL); |
1364 | 1326 |
1365 QuicAckFrame ack_fec(2, QuicTime::Zero(), 1); | 1327 QuicAckFrame ack_fec(2, QuicTime::Zero(), 1); |
(...skipping 10 matching lines...) Expand all Loading... |
1376 | 1338 |
1377 clock_.AdvanceTime(DefaultRetransmissionTime()); | 1339 clock_.AdvanceTime(DefaultRetransmissionTime()); |
1378 | 1340 |
1379 // Abandon only data packet, FEC has been acked. | 1341 // Abandon only data packet, FEC has been acked. |
1380 EXPECT_CALL(*send_algorithm_, OnPacketAbandoned(sequence_number, _)).Times(1); | 1342 EXPECT_CALL(*send_algorithm_, OnPacketAbandoned(sequence_number, _)).Times(1); |
1381 // Send only data packet. | 1343 // Send only data packet. |
1382 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)).Times(1); | 1344 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)).Times(1); |
1383 connection_.GetRetransmissionAlarm()->Fire(); | 1345 connection_.GetRetransmissionAlarm()->Fire(); |
1384 } | 1346 } |
1385 | 1347 |
1386 TEST_P(QuicConnectionTest, FramePacking) { | 1348 TEST_F(QuicConnectionTest, FramePacking) { |
1387 // Block the connection. | 1349 // Block the connection. |
1388 connection_.GetSendAlarm()->Set( | 1350 connection_.GetSendAlarm()->Set( |
1389 clock_.ApproximateNow().Add(QuicTime::Delta::FromSeconds(1))); | 1351 clock_.ApproximateNow().Add(QuicTime::Delta::FromSeconds(1))); |
1390 | 1352 |
1391 // Send an ack and two stream frames in 1 packet by queueing them. | 1353 // Send an ack and two stream frames in 1 packet by queueing them. |
1392 connection_.SendAck(); | 1354 connection_.SendAck(); |
1393 EXPECT_CALL(visitor_, OnCanWrite()).WillOnce(DoAll( | 1355 EXPECT_CALL(visitor_, OnCanWrite()).WillOnce(DoAll( |
1394 IgnoreResult(InvokeWithoutArgs(&connection_, | 1356 IgnoreResult(InvokeWithoutArgs(&connection_, |
1395 &TestConnection::SendStreamData3)), | 1357 &TestConnection::SendStreamData3)), |
1396 IgnoreResult(InvokeWithoutArgs(&connection_, | 1358 IgnoreResult(InvokeWithoutArgs(&connection_, |
(...skipping 10 matching lines...) Expand all Loading... |
1407 | 1369 |
1408 // Parse the last packet and ensure it's an ack and two stream frames from | 1370 // Parse the last packet and ensure it's an ack and two stream frames from |
1409 // two different streams. | 1371 // two different streams. |
1410 EXPECT_EQ(3u, writer_->frame_count()); | 1372 EXPECT_EQ(3u, writer_->frame_count()); |
1411 EXPECT_TRUE(writer_->ack()); | 1373 EXPECT_TRUE(writer_->ack()); |
1412 EXPECT_EQ(2u, writer_->stream_frames()->size()); | 1374 EXPECT_EQ(2u, writer_->stream_frames()->size()); |
1413 EXPECT_EQ(kStreamId3, (*writer_->stream_frames())[0].stream_id); | 1375 EXPECT_EQ(kStreamId3, (*writer_->stream_frames())[0].stream_id); |
1414 EXPECT_EQ(kStreamId5, (*writer_->stream_frames())[1].stream_id); | 1376 EXPECT_EQ(kStreamId5, (*writer_->stream_frames())[1].stream_id); |
1415 } | 1377 } |
1416 | 1378 |
1417 TEST_P(QuicConnectionTest, FramePackingNonCryptoThenCrypto) { | 1379 TEST_F(QuicConnectionTest, FramePackingNonCryptoThenCrypto) { |
1418 // Block the connection. | 1380 // Block the connection. |
1419 connection_.GetSendAlarm()->Set( | 1381 connection_.GetSendAlarm()->Set( |
1420 clock_.ApproximateNow().Add(QuicTime::Delta::FromSeconds(1))); | 1382 clock_.ApproximateNow().Add(QuicTime::Delta::FromSeconds(1))); |
1421 | 1383 |
1422 // Send an ack and two stream frames (one non-crypto, then one crypto) in 2 | 1384 // Send an ack and two stream frames (one non-crypto, then one crypto) in 2 |
1423 // packets by queueing them. | 1385 // packets by queueing them. |
1424 connection_.SendAck(); | 1386 connection_.SendAck(); |
1425 EXPECT_CALL(visitor_, OnCanWrite()).WillOnce(DoAll( | 1387 EXPECT_CALL(visitor_, OnCanWrite()).WillOnce(DoAll( |
1426 IgnoreResult(InvokeWithoutArgs(&connection_, | 1388 IgnoreResult(InvokeWithoutArgs(&connection_, |
1427 &TestConnection::SendStreamData3)), | 1389 &TestConnection::SendStreamData3)), |
1428 IgnoreResult(InvokeWithoutArgs(&connection_, | 1390 IgnoreResult(InvokeWithoutArgs(&connection_, |
1429 &TestConnection::SendCryptoStreamData)), | 1391 &TestConnection::SendCryptoStreamData)), |
1430 Return(true))); | 1392 Return(true))); |
1431 | 1393 |
1432 EXPECT_CALL(*send_algorithm_, | 1394 EXPECT_CALL(*send_algorithm_, |
1433 OnPacketSent(_, _, _, NOT_RETRANSMISSION, _)) | 1395 OnPacketSent(_, _, _, NOT_RETRANSMISSION, _)) |
1434 .Times(2); | 1396 .Times(2); |
1435 // Unblock the connection. | 1397 // Unblock the connection. |
1436 connection_.GetSendAlarm()->Fire(); | 1398 connection_.GetSendAlarm()->Fire(); |
1437 EXPECT_EQ(0u, connection_.NumQueuedPackets()); | 1399 EXPECT_EQ(0u, connection_.NumQueuedPackets()); |
1438 EXPECT_FALSE(connection_.HasQueuedData()); | 1400 EXPECT_FALSE(connection_.HasQueuedData()); |
1439 | 1401 |
1440 // Parse the last packet and ensure it's the crypto stream frame. | 1402 // Parse the last packet and ensure it's the crypto stream frame. |
1441 EXPECT_EQ(1u, writer_->frame_count()); | 1403 EXPECT_EQ(1u, writer_->frame_count()); |
1442 EXPECT_TRUE(writer_->ack()); | |
1443 EXPECT_EQ(1u, writer_->stream_frames()->size()); | 1404 EXPECT_EQ(1u, writer_->stream_frames()->size()); |
1444 EXPECT_EQ(kCryptoStreamId, (*writer_->stream_frames())[0].stream_id); | 1405 EXPECT_EQ(kCryptoStreamId, (*writer_->stream_frames())[0].stream_id); |
1445 } | 1406 } |
1446 | 1407 |
1447 TEST_P(QuicConnectionTest, FramePackingCryptoThenNonCrypto) { | 1408 TEST_F(QuicConnectionTest, FramePackingCryptoThenNonCrypto) { |
1448 // Block the connection. | 1409 // Block the connection. |
1449 connection_.GetSendAlarm()->Set( | 1410 connection_.GetSendAlarm()->Set( |
1450 clock_.ApproximateNow().Add(QuicTime::Delta::FromSeconds(1))); | 1411 clock_.ApproximateNow().Add(QuicTime::Delta::FromSeconds(1))); |
1451 | 1412 |
1452 // Send an ack and two stream frames (one crypto, then one non-crypto) in 3 | 1413 // Send an ack and two stream frames (one crypto, then one non-crypto) in 3 |
1453 // packets by queueing them. | 1414 // packets by queueing them. |
1454 connection_.SendAck(); | 1415 connection_.SendAck(); |
1455 EXPECT_CALL(visitor_, OnCanWrite()).WillOnce(DoAll( | 1416 EXPECT_CALL(visitor_, OnCanWrite()).WillOnce(DoAll( |
1456 IgnoreResult(InvokeWithoutArgs(&connection_, | 1417 IgnoreResult(InvokeWithoutArgs(&connection_, |
1457 &TestConnection::SendCryptoStreamData)), | 1418 &TestConnection::SendCryptoStreamData)), |
1458 IgnoreResult(InvokeWithoutArgs(&connection_, | 1419 IgnoreResult(InvokeWithoutArgs(&connection_, |
1459 &TestConnection::SendStreamData3)), | 1420 &TestConnection::SendStreamData3)), |
1460 Return(true))); | 1421 Return(true))); |
1461 | 1422 |
1462 EXPECT_CALL(*send_algorithm_, | 1423 EXPECT_CALL(*send_algorithm_, |
1463 OnPacketSent(_, _, _, NOT_RETRANSMISSION, _)) | 1424 OnPacketSent(_, _, _, NOT_RETRANSMISSION, _)) |
1464 .Times(3); | 1425 .Times(3); |
1465 // Unblock the connection. | 1426 // Unblock the connection. |
1466 connection_.GetSendAlarm()->Fire(); | 1427 connection_.GetSendAlarm()->Fire(); |
1467 EXPECT_EQ(0u, connection_.NumQueuedPackets()); | 1428 EXPECT_EQ(0u, connection_.NumQueuedPackets()); |
1468 EXPECT_FALSE(connection_.HasQueuedData()); | 1429 EXPECT_FALSE(connection_.HasQueuedData()); |
1469 | 1430 |
1470 // Parse the last packet and ensure it's the stream frame from stream 3. | 1431 // Parse the last packet and ensure it's the stream frame from stream 3. |
1471 EXPECT_EQ(1u, writer_->frame_count()); | 1432 EXPECT_EQ(1u, writer_->frame_count()); |
1472 EXPECT_TRUE(writer_->ack()); | |
1473 EXPECT_EQ(1u, writer_->stream_frames()->size()); | 1433 EXPECT_EQ(1u, writer_->stream_frames()->size()); |
1474 EXPECT_EQ(kStreamId3, (*writer_->stream_frames())[0].stream_id); | 1434 EXPECT_EQ(kStreamId3, (*writer_->stream_frames())[0].stream_id); |
1475 } | 1435 } |
1476 | 1436 |
1477 TEST_P(QuicConnectionTest, FramePackingFEC) { | 1437 TEST_F(QuicConnectionTest, FramePackingFEC) { |
1478 // Enable fec. | 1438 // Enable fec. |
1479 connection_.options()->max_packets_per_fec_group = 6; | 1439 connection_.options()->max_packets_per_fec_group = 6; |
1480 // Block the connection. | 1440 // Block the connection. |
1481 connection_.GetSendAlarm()->Set( | 1441 connection_.GetSendAlarm()->Set( |
1482 clock_.ApproximateNow().Add(QuicTime::Delta::FromSeconds(1))); | 1442 clock_.ApproximateNow().Add(QuicTime::Delta::FromSeconds(1))); |
1483 | 1443 |
1484 // Send an ack and two stream frames in 1 packet by queueing them. | 1444 // Send an ack and two stream frames in 1 packet by queueing them. |
1485 connection_.SendAck(); | 1445 connection_.SendAck(); |
1486 EXPECT_CALL(visitor_, OnCanWrite()).WillOnce(DoAll( | 1446 EXPECT_CALL(visitor_, OnCanWrite()).WillOnce(DoAll( |
1487 IgnoreResult(InvokeWithoutArgs(&connection_, | 1447 IgnoreResult(InvokeWithoutArgs(&connection_, |
1488 &TestConnection::SendStreamData3)), | 1448 &TestConnection::SendStreamData3)), |
1489 IgnoreResult(InvokeWithoutArgs(&connection_, | 1449 IgnoreResult(InvokeWithoutArgs(&connection_, |
1490 &TestConnection::SendStreamData5)), | 1450 &TestConnection::SendStreamData5)), |
1491 Return(true))); | 1451 Return(true))); |
1492 | 1452 |
1493 EXPECT_CALL(*send_algorithm_, | 1453 EXPECT_CALL(*send_algorithm_, |
1494 OnPacketSent(_, _, _, NOT_RETRANSMISSION, _)).Times(2); | 1454 OnPacketSent(_, _, _, NOT_RETRANSMISSION, _)).Times(2); |
1495 // Unblock the connection. | 1455 // Unblock the connection. |
1496 connection_.GetSendAlarm()->Fire(); | 1456 connection_.GetSendAlarm()->Fire(); |
1497 EXPECT_EQ(0u, connection_.NumQueuedPackets()); | 1457 EXPECT_EQ(0u, connection_.NumQueuedPackets()); |
1498 EXPECT_FALSE(connection_.HasQueuedData()); | 1458 EXPECT_FALSE(connection_.HasQueuedData()); |
1499 | 1459 |
1500 // Parse the last packet and ensure it's in an fec group. | 1460 // Parse the last packet and ensure it's in an fec group. |
1501 EXPECT_EQ(1u, writer_->header()->fec_group); | 1461 EXPECT_EQ(1u, writer_->header()->fec_group); |
1502 EXPECT_EQ(0u, writer_->frame_count()); | 1462 EXPECT_EQ(0u, writer_->frame_count()); |
1503 } | 1463 } |
1504 | 1464 |
1505 TEST_P(QuicConnectionTest, FramePackingAckResponse) { | 1465 TEST_F(QuicConnectionTest, FramePackingAckResponse) { |
1506 if (!GetParam()) { | |
1507 // This test depends on BundleAckWithPacket being true. | |
1508 return; | |
1509 } | |
1510 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | 1466 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); |
1511 | |
1512 // Process a data packet to queue up a pending ack. | 1467 // Process a data packet to queue up a pending ack. |
1513 EXPECT_CALL(visitor_, OnStreamFrames(_)).WillOnce(Return(true)); | 1468 EXPECT_CALL(visitor_, OnStreamFrames(_)).WillOnce(Return(true)); |
1514 ProcessDataPacket(1, 1, kEntropyFlag); | 1469 ProcessDataPacket(1, 1, kEntropyFlag); |
1515 | 1470 |
1516 EXPECT_CALL(visitor_, OnCanWrite()).WillOnce(DoAll( | 1471 EXPECT_CALL(visitor_, OnCanWrite()).WillOnce(DoAll( |
1517 IgnoreResult(InvokeWithoutArgs(&connection_, | 1472 IgnoreResult(InvokeWithoutArgs(&connection_, |
1518 &TestConnection::SendStreamData3)), | 1473 &TestConnection::SendStreamData3)), |
1519 IgnoreResult(InvokeWithoutArgs(&connection_, | 1474 IgnoreResult(InvokeWithoutArgs(&connection_, |
1520 &TestConnection::SendStreamData5)), | 1475 &TestConnection::SendStreamData5)), |
1521 Return(true))); | 1476 Return(true))); |
(...skipping 12 matching lines...) Expand all Loading... |
1534 | 1489 |
1535 // Parse the last packet and ensure it's an ack and two stream frames from | 1490 // Parse the last packet and ensure it's an ack and two stream frames from |
1536 // two different streams. | 1491 // two different streams. |
1537 EXPECT_EQ(3u, writer_->frame_count()); | 1492 EXPECT_EQ(3u, writer_->frame_count()); |
1538 EXPECT_TRUE(writer_->ack()); | 1493 EXPECT_TRUE(writer_->ack()); |
1539 ASSERT_EQ(2u, writer_->stream_frames()->size()); | 1494 ASSERT_EQ(2u, writer_->stream_frames()->size()); |
1540 EXPECT_EQ(kStreamId3, (*writer_->stream_frames())[0].stream_id); | 1495 EXPECT_EQ(kStreamId3, (*writer_->stream_frames())[0].stream_id); |
1541 EXPECT_EQ(kStreamId5, (*writer_->stream_frames())[1].stream_id); | 1496 EXPECT_EQ(kStreamId5, (*writer_->stream_frames())[1].stream_id); |
1542 } | 1497 } |
1543 | 1498 |
1544 TEST_P(QuicConnectionTest, FramePackingSendv) { | 1499 TEST_F(QuicConnectionTest, FramePackingSendv) { |
1545 // Send data in 1 packet by writing multiple blocks in a single iovector | 1500 // Send data in 1 packet by writing multiple blocks in a single iovector |
1546 // using writev. | 1501 // using writev. |
1547 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, NOT_RETRANSMISSION, _)); | 1502 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, NOT_RETRANSMISSION, _)); |
1548 | 1503 |
1549 char data[] = "ABCD"; | 1504 char data[] = "ABCD"; |
1550 IOVector data_iov; | 1505 IOVector data_iov; |
1551 data_iov.AppendNoCoalesce(data, 2); | 1506 data_iov.AppendNoCoalesce(data, 2); |
1552 data_iov.AppendNoCoalesce(data + 2, 2); | 1507 data_iov.AppendNoCoalesce(data + 2, 2); |
1553 connection_.SendStreamData(1, data_iov, 0, !kFin, NULL); | 1508 connection_.SendStreamData(1, data_iov, 0, !kFin, NULL); |
1554 | 1509 |
1555 EXPECT_EQ(0u, connection_.NumQueuedPackets()); | 1510 EXPECT_EQ(0u, connection_.NumQueuedPackets()); |
1556 EXPECT_FALSE(connection_.HasQueuedData()); | 1511 EXPECT_FALSE(connection_.HasQueuedData()); |
1557 | 1512 |
1558 // Parse the last packet and ensure multiple iovector blocks have | 1513 // Parse the last packet and ensure multiple iovector blocks have |
1559 // been packed into a single stream frame from one stream. | 1514 // been packed into a single stream frame from one stream. |
1560 EXPECT_EQ(1u, writer_->frame_count()); | 1515 EXPECT_EQ(1u, writer_->frame_count()); |
1561 EXPECT_EQ(1u, writer_->stream_frames()->size()); | 1516 EXPECT_EQ(1u, writer_->stream_frames()->size()); |
1562 QuicStreamFrame frame = (*writer_->stream_frames())[0]; | 1517 QuicStreamFrame frame = (*writer_->stream_frames())[0]; |
1563 EXPECT_EQ(1u, frame.stream_id); | 1518 EXPECT_EQ(1u, frame.stream_id); |
1564 EXPECT_EQ("ABCD", string(static_cast<char*> | 1519 EXPECT_EQ("ABCD", string(static_cast<char*> |
1565 (frame.data.iovec()[0].iov_base), | 1520 (frame.data.iovec()[0].iov_base), |
1566 (frame.data.iovec()[0].iov_len))); | 1521 (frame.data.iovec()[0].iov_len))); |
1567 } | 1522 } |
1568 | 1523 |
1569 TEST_P(QuicConnectionTest, FramePackingSendvQueued) { | 1524 TEST_F(QuicConnectionTest, FramePackingSendvQueued) { |
1570 // Try to send two stream frames in 1 packet by using writev. | 1525 // Try to send two stream frames in 1 packet by using writev. |
1571 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, NOT_RETRANSMISSION, _)); | 1526 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, NOT_RETRANSMISSION, _)); |
1572 | 1527 |
1573 writer_->set_blocked(true); | 1528 writer_->set_blocked(true); |
1574 char data[] = "ABCD"; | 1529 char data[] = "ABCD"; |
1575 IOVector data_iov; | 1530 IOVector data_iov; |
1576 data_iov.AppendNoCoalesce(data, 2); | 1531 data_iov.AppendNoCoalesce(data, 2); |
1577 data_iov.AppendNoCoalesce(data + 2, 2); | 1532 data_iov.AppendNoCoalesce(data + 2, 2); |
1578 connection_.SendStreamData(1, data_iov, 0, !kFin, NULL); | 1533 connection_.SendStreamData(1, data_iov, 0, !kFin, NULL); |
1579 | 1534 |
1580 EXPECT_EQ(1u, connection_.NumQueuedPackets()); | 1535 EXPECT_EQ(1u, connection_.NumQueuedPackets()); |
1581 EXPECT_TRUE(connection_.HasQueuedData()); | 1536 EXPECT_TRUE(connection_.HasQueuedData()); |
1582 | 1537 |
1583 // Attempt to send all packets, but since we're actually still | 1538 // Attempt to send all packets, but since we're actually still |
1584 // blocked, they should all remain queued. | 1539 // blocked, they should all remain queued. |
1585 EXPECT_FALSE(connection_.OnCanWrite()); | 1540 EXPECT_FALSE(connection_.OnCanWrite()); |
1586 EXPECT_EQ(1u, connection_.NumQueuedPackets()); | 1541 EXPECT_EQ(1u, connection_.NumQueuedPackets()); |
1587 | 1542 |
1588 // Unblock the writes and actually send. | 1543 // Unblock the writes and actually send. |
1589 writer_->set_blocked(false); | 1544 writer_->set_blocked(false); |
1590 EXPECT_TRUE(connection_.OnCanWrite()); | 1545 EXPECT_TRUE(connection_.OnCanWrite()); |
1591 EXPECT_EQ(0u, connection_.NumQueuedPackets()); | 1546 EXPECT_EQ(0u, connection_.NumQueuedPackets()); |
1592 | 1547 |
1593 // Parse the last packet and ensure it's one stream frame from one stream. | 1548 // Parse the last packet and ensure it's one stream frame from one stream. |
1594 EXPECT_EQ(1u, writer_->frame_count()); | 1549 EXPECT_EQ(1u, writer_->frame_count()); |
1595 EXPECT_EQ(1u, writer_->stream_frames()->size()); | 1550 EXPECT_EQ(1u, writer_->stream_frames()->size()); |
1596 EXPECT_EQ(1u, (*writer_->stream_frames())[0].stream_id); | 1551 EXPECT_EQ(1u, (*writer_->stream_frames())[0].stream_id); |
1597 } | 1552 } |
1598 | 1553 |
1599 TEST_P(QuicConnectionTest, SendingZeroBytes) { | 1554 TEST_F(QuicConnectionTest, SendingZeroBytes) { |
1600 // Send a zero byte write with a fin using writev. | 1555 // Send a zero byte write with a fin using writev. |
1601 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, NOT_RETRANSMISSION, _)); | 1556 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, NOT_RETRANSMISSION, _)); |
1602 IOVector empty_iov; | 1557 IOVector empty_iov; |
1603 connection_.SendStreamData(1, empty_iov, 0, kFin, NULL); | 1558 connection_.SendStreamData(1, empty_iov, 0, kFin, NULL); |
1604 | 1559 |
1605 EXPECT_EQ(0u, connection_.NumQueuedPackets()); | 1560 EXPECT_EQ(0u, connection_.NumQueuedPackets()); |
1606 EXPECT_FALSE(connection_.HasQueuedData()); | 1561 EXPECT_FALSE(connection_.HasQueuedData()); |
1607 | 1562 |
1608 // Parse the last packet and ensure it's one stream frame from one stream. | 1563 // Parse the last packet and ensure it's one stream frame from one stream. |
1609 EXPECT_EQ(1u, writer_->frame_count()); | 1564 EXPECT_EQ(1u, writer_->frame_count()); |
1610 EXPECT_EQ(1u, writer_->stream_frames()->size()); | 1565 EXPECT_EQ(1u, writer_->stream_frames()->size()); |
1611 EXPECT_EQ(1u, (*writer_->stream_frames())[0].stream_id); | 1566 EXPECT_EQ(1u, (*writer_->stream_frames())[0].stream_id); |
1612 EXPECT_TRUE((*writer_->stream_frames())[0].fin); | 1567 EXPECT_TRUE((*writer_->stream_frames())[0].fin); |
1613 } | 1568 } |
1614 | 1569 |
1615 TEST_P(QuicConnectionTest, OnCanWrite) { | 1570 TEST_F(QuicConnectionTest, OnCanWrite) { |
1616 // Visitor's OnCanWill send data, but will return false. | 1571 // Visitor's OnCanWill send data, but will return false. |
1617 EXPECT_CALL(visitor_, OnCanWrite()).WillOnce(DoAll( | 1572 EXPECT_CALL(visitor_, OnCanWrite()).WillOnce(DoAll( |
1618 IgnoreResult(InvokeWithoutArgs(&connection_, | 1573 IgnoreResult(InvokeWithoutArgs(&connection_, |
1619 &TestConnection::SendStreamData3)), | 1574 &TestConnection::SendStreamData3)), |
1620 IgnoreResult(InvokeWithoutArgs(&connection_, | 1575 IgnoreResult(InvokeWithoutArgs(&connection_, |
1621 &TestConnection::SendStreamData5)), | 1576 &TestConnection::SendStreamData5)), |
1622 Return(false))); | 1577 Return(false))); |
1623 | 1578 |
1624 EXPECT_CALL(*send_algorithm_, | 1579 EXPECT_CALL(*send_algorithm_, |
1625 TimeUntilSend(_, NOT_RETRANSMISSION, _, _)).WillRepeatedly( | 1580 TimeUntilSend(_, NOT_RETRANSMISSION, _, _)).WillRepeatedly( |
1626 testing::Return(QuicTime::Delta::Zero())); | 1581 testing::Return(QuicTime::Delta::Zero())); |
1627 | 1582 |
1628 // Unblock the connection. | 1583 // Unblock the connection. |
1629 connection_.OnCanWrite(); | 1584 connection_.OnCanWrite(); |
1630 // Parse the last packet and ensure it's the two stream frames from | 1585 // Parse the last packet and ensure it's the two stream frames from |
1631 // two different streams. | 1586 // two different streams. |
1632 EXPECT_EQ(2u, writer_->frame_count()); | 1587 EXPECT_EQ(2u, writer_->frame_count()); |
1633 EXPECT_EQ(2u, writer_->stream_frames()->size()); | 1588 EXPECT_EQ(2u, writer_->stream_frames()->size()); |
1634 EXPECT_EQ(kStreamId3, (*writer_->stream_frames())[0].stream_id); | 1589 EXPECT_EQ(kStreamId3, (*writer_->stream_frames())[0].stream_id); |
1635 EXPECT_EQ(kStreamId5, (*writer_->stream_frames())[1].stream_id); | 1590 EXPECT_EQ(kStreamId5, (*writer_->stream_frames())[1].stream_id); |
1636 } | 1591 } |
1637 | 1592 |
1638 TEST_P(QuicConnectionTest, RetransmitOnNack) { | 1593 TEST_F(QuicConnectionTest, RetransmitOnNack) { |
1639 EXPECT_CALL(*send_algorithm_, OnIncomingAck(_, _, _)).Times(2); | 1594 EXPECT_CALL(*send_algorithm_, OnIncomingAck(_, _, _)).Times(2); |
1640 EXPECT_CALL(*send_algorithm_, OnIncomingLoss(_, _)).Times(1); | 1595 EXPECT_CALL(*send_algorithm_, OnIncomingLoss(_, _)).Times(1); |
1641 EXPECT_CALL(*send_algorithm_, OnPacketAbandoned(2, _)).Times(1); | 1596 EXPECT_CALL(*send_algorithm_, OnPacketAbandoned(2, _)).Times(1); |
1642 QuicPacketSequenceNumber last_packet; | 1597 QuicPacketSequenceNumber last_packet; |
1643 QuicByteCount second_packet_size; | 1598 QuicByteCount second_packet_size; |
1644 SendStreamDataToPeer(3, "foo", 0, !kFin, &last_packet); // Packet 1 | 1599 SendStreamDataToPeer(3, "foo", 0, !kFin, &last_packet); // Packet 1 |
1645 second_packet_size = | 1600 second_packet_size = |
1646 SendStreamDataToPeer(3, "foos", 3, !kFin, &last_packet); // Packet 2 | 1601 SendStreamDataToPeer(3, "foos", 3, !kFin, &last_packet); // Packet 2 |
1647 SendStreamDataToPeer(3, "fooos", 7, !kFin, &last_packet); // Packet 3 | 1602 SendStreamDataToPeer(3, "fooos", 7, !kFin, &last_packet); // Packet 3 |
1648 | 1603 |
(...skipping 19 matching lines...) Expand all Loading... |
1668 ProcessAckPacket(&nack_two); | 1623 ProcessAckPacket(&nack_two); |
1669 ProcessAckPacket(&nack_two); | 1624 ProcessAckPacket(&nack_two); |
1670 | 1625 |
1671 // The third nack should trigger a retransimission. | 1626 // The third nack should trigger a retransimission. |
1672 EXPECT_CALL(*send_algorithm_, | 1627 EXPECT_CALL(*send_algorithm_, |
1673 OnPacketSent(_, _, second_packet_size - kQuicVersionSize, | 1628 OnPacketSent(_, _, second_packet_size - kQuicVersionSize, |
1674 NACK_RETRANSMISSION, _)).Times(1); | 1629 NACK_RETRANSMISSION, _)).Times(1); |
1675 ProcessAckPacket(&nack_two); | 1630 ProcessAckPacket(&nack_two); |
1676 } | 1631 } |
1677 | 1632 |
1678 TEST_P(QuicConnectionTest, DiscardRetransmit) { | 1633 TEST_F(QuicConnectionTest, DiscardRetransmit) { |
1679 EXPECT_CALL(*send_algorithm_, OnIncomingAck(_, _, _)).Times(2); | 1634 EXPECT_CALL(*send_algorithm_, OnIncomingAck(_, _, _)).Times(2); |
1680 EXPECT_CALL(*send_algorithm_, OnIncomingLoss(_, _)).Times(1); | 1635 EXPECT_CALL(*send_algorithm_, OnIncomingLoss(_, _)).Times(1); |
1681 EXPECT_CALL(*send_algorithm_, OnPacketAbandoned(2, _)).Times(1); | 1636 EXPECT_CALL(*send_algorithm_, OnPacketAbandoned(2, _)).Times(1); |
1682 QuicPacketSequenceNumber last_packet; | 1637 QuicPacketSequenceNumber last_packet; |
1683 SendStreamDataToPeer(1, "foo", 0, !kFin, &last_packet); // Packet 1 | 1638 SendStreamDataToPeer(1, "foo", 0, !kFin, &last_packet); // Packet 1 |
1684 SendStreamDataToPeer(1, "foos", 3, !kFin, &last_packet); // Packet 2 | 1639 SendStreamDataToPeer(1, "foos", 3, !kFin, &last_packet); // Packet 2 |
1685 SendStreamDataToPeer(1, "fooos", 7, !kFin, &last_packet); // Packet 3 | 1640 SendStreamDataToPeer(1, "fooos", 7, !kFin, &last_packet); // Packet 3 |
1686 | 1641 |
1687 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | 1642 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); |
1688 | 1643 |
(...skipping 35 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
1724 // send the retransmission. | 1679 // send the retransmission. |
1725 EXPECT_CALL(*send_algorithm_, | 1680 EXPECT_CALL(*send_algorithm_, |
1726 OnPacketSent(_, _, _, _, _)).Times(0); | 1681 OnPacketSent(_, _, _, _, _)).Times(0); |
1727 | 1682 |
1728 writer_->set_blocked(false); | 1683 writer_->set_blocked(false); |
1729 connection_.OnCanWrite(); | 1684 connection_.OnCanWrite(); |
1730 | 1685 |
1731 EXPECT_EQ(0u, connection_.NumQueuedPackets()); | 1686 EXPECT_EQ(0u, connection_.NumQueuedPackets()); |
1732 } | 1687 } |
1733 | 1688 |
1734 TEST_P(QuicConnectionTest, RetransmitNackedLargestObserved) { | 1689 TEST_F(QuicConnectionTest, RetransmitNackedLargestObserved) { |
1735 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | 1690 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); |
1736 EXPECT_CALL(*send_algorithm_, OnIncomingLoss(_, _)).Times(1); | 1691 EXPECT_CALL(*send_algorithm_, OnIncomingLoss(_, _)).Times(1); |
1737 QuicPacketSequenceNumber largest_observed; | 1692 QuicPacketSequenceNumber largest_observed; |
1738 QuicByteCount packet_size; | 1693 QuicByteCount packet_size; |
1739 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, NOT_RETRANSMISSION, _)) | 1694 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, NOT_RETRANSMISSION, _)) |
1740 .WillOnce(DoAll(SaveArg<1>(&largest_observed), SaveArg<2>(&packet_size), | 1695 .WillOnce(DoAll(SaveArg<1>(&largest_observed), SaveArg<2>(&packet_size), |
1741 Return(true))); | 1696 Return(true))); |
1742 EXPECT_CALL(*send_algorithm_, OnPacketAbandoned(1, _)).Times(1); | 1697 EXPECT_CALL(*send_algorithm_, OnPacketAbandoned(1, _)).Times(1); |
1743 connection_.SendStreamDataWithString(3, "foo", 0, !kFin, NULL); | 1698 connection_.SendStreamDataWithString(3, "foo", 0, !kFin, NULL); |
1744 QuicAckFrame frame(1, QuicTime::Zero(), largest_observed); | 1699 QuicAckFrame frame(1, QuicTime::Zero(), largest_observed); |
1745 frame.received_info.missing_packets.insert(largest_observed); | 1700 frame.received_info.missing_packets.insert(largest_observed); |
1746 frame.received_info.entropy_hash = QuicConnectionPeer::GetSentEntropyHash( | 1701 frame.received_info.entropy_hash = QuicConnectionPeer::GetSentEntropyHash( |
1747 &connection_, largest_observed - 1); | 1702 &connection_, largest_observed - 1); |
1748 ProcessAckPacket(&frame); | 1703 ProcessAckPacket(&frame); |
1749 ProcessAckPacket(&frame); | 1704 ProcessAckPacket(&frame); |
1750 // Third nack should retransmit the largest observed packet. | 1705 // Third nack should retransmit the largest observed packet. |
1751 EXPECT_CALL(*send_algorithm_, | 1706 EXPECT_CALL(*send_algorithm_, |
1752 OnPacketSent(_, _, packet_size - kQuicVersionSize, | 1707 OnPacketSent(_, _, packet_size - kQuicVersionSize, |
1753 NACK_RETRANSMISSION, _)); | 1708 NACK_RETRANSMISSION, _)); |
1754 ProcessAckPacket(&frame); | 1709 ProcessAckPacket(&frame); |
1755 } | 1710 } |
1756 | 1711 |
1757 TEST_P(QuicConnectionTest, QueueAfterTwoRTOs) { | 1712 TEST_F(QuicConnectionTest, QueueAfterTwoRTOs) { |
1758 for (int i = 0; i < 10; ++i) { | 1713 for (int i = 0; i < 10; ++i) { |
1759 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)).Times(1); | 1714 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)).Times(1); |
1760 connection_.SendStreamDataWithString(1, "foo", i * 3, !kFin, NULL); | 1715 connection_.SendStreamDataWithString(1, "foo", i * 3, !kFin, NULL); |
1761 } | 1716 } |
1762 | 1717 |
1763 // Block the congestion window and ensure they're queued. | 1718 // Block the congestion window and ensure they're queued. |
1764 writer_->set_blocked(true); | 1719 writer_->set_blocked(true); |
1765 clock_.AdvanceTime(DefaultRetransmissionTime()); | 1720 clock_.AdvanceTime(DefaultRetransmissionTime()); |
1766 // Only one packet should be retransmitted. | 1721 // Only one packet should be retransmitted. |
1767 EXPECT_CALL(*send_algorithm_, OnPacketAbandoned(_, _)).Times(10); | 1722 EXPECT_CALL(*send_algorithm_, OnPacketAbandoned(_, _)).Times(10); |
1768 connection_.GetRetransmissionAlarm()->Fire(); | 1723 connection_.GetRetransmissionAlarm()->Fire(); |
1769 EXPECT_TRUE(connection_.HasQueuedData()); | 1724 EXPECT_TRUE(connection_.HasQueuedData()); |
1770 | 1725 |
1771 // Unblock the congestion window. | 1726 // Unblock the congestion window. |
1772 writer_->set_blocked(false); | 1727 writer_->set_blocked(false); |
1773 clock_.AdvanceTime(QuicTime::Delta::FromMicroseconds( | 1728 clock_.AdvanceTime(QuicTime::Delta::FromMicroseconds( |
1774 2 * DefaultRetransmissionTime().ToMicroseconds())); | 1729 2 * DefaultRetransmissionTime().ToMicroseconds())); |
1775 // Retransmit already retransmitted packets event though the sequence number | 1730 // Retransmit already retransmitted packets event though the sequence number |
1776 // greater than the largest observed. | 1731 // greater than the largest observed. |
1777 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)).Times(10); | 1732 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)).Times(10); |
1778 connection_.GetRetransmissionAlarm()->Fire(); | 1733 connection_.GetRetransmissionAlarm()->Fire(); |
1779 connection_.OnCanWrite(); | 1734 connection_.OnCanWrite(); |
1780 } | 1735 } |
1781 | 1736 |
1782 TEST_P(QuicConnectionTest, WriteBlockedThenSent) { | 1737 TEST_F(QuicConnectionTest, WriteBlockedThenSent) { |
1783 writer_->set_blocked(true); | 1738 writer_->set_blocked(true); |
1784 | 1739 |
1785 writer_->set_is_write_blocked_data_buffered(true); | 1740 writer_->set_is_write_blocked_data_buffered(true); |
1786 connection_.SendStreamDataWithString(1, "foo", 0, !kFin, NULL); | 1741 connection_.SendStreamDataWithString(1, "foo", 0, !kFin, NULL); |
1787 EXPECT_FALSE(connection_.GetRetransmissionAlarm()->IsSet()); | 1742 EXPECT_FALSE(connection_.GetRetransmissionAlarm()->IsSet()); |
1788 | 1743 |
1789 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)).Times(1); | 1744 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)).Times(1); |
1790 connection_.OnPacketSent(WriteResult(WRITE_STATUS_OK, 0)); | 1745 connection_.OnPacketSent(WriteResult(WRITE_STATUS_OK, 0)); |
1791 EXPECT_TRUE(connection_.GetRetransmissionAlarm()->IsSet()); | 1746 EXPECT_TRUE(connection_.GetRetransmissionAlarm()->IsSet()); |
1792 } | 1747 } |
1793 | 1748 |
1794 TEST_P(QuicConnectionTest, ResumptionAlarmThenWriteBlocked) { | 1749 TEST_F(QuicConnectionTest, ResumptionAlarmThenWriteBlocked) { |
1795 // Set the send and resumption alarm, then block the connection. | 1750 // Set the send and resumption alarm, then block the connection. |
1796 connection_.GetResumeWritesAlarm()->Set(clock_.ApproximateNow()); | 1751 connection_.GetResumeWritesAlarm()->Set(clock_.ApproximateNow()); |
1797 connection_.GetSendAlarm()->Set(clock_.ApproximateNow()); | 1752 connection_.GetSendAlarm()->Set(clock_.ApproximateNow()); |
1798 QuicConnectionPeer::SetIsWriteBlocked(&connection_, true); | 1753 QuicConnectionPeer::SetIsWriteBlocked(&connection_, true); |
1799 | 1754 |
1800 // Fire the alarms and ensure the connection is still write blocked. | 1755 // Fire the alarms and ensure the connection is still write blocked. |
1801 connection_.GetResumeWritesAlarm()->Fire(); | 1756 connection_.GetResumeWritesAlarm()->Fire(); |
1802 connection_.GetSendAlarm()->Fire(); | 1757 connection_.GetSendAlarm()->Fire(); |
1803 EXPECT_TRUE(QuicConnectionPeer::IsWriteBlocked(&connection_)); | 1758 EXPECT_TRUE(QuicConnectionPeer::IsWriteBlocked(&connection_)); |
1804 } | 1759 } |
1805 | 1760 |
1806 TEST_P(QuicConnectionTest, LimitPacketsPerNack) { | 1761 TEST_F(QuicConnectionTest, LimitPacketsPerNack) { |
1807 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | 1762 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); |
1808 EXPECT_CALL(*send_algorithm_, OnIncomingAck(12, _, _)).Times(1); | 1763 EXPECT_CALL(*send_algorithm_, OnIncomingAck(12, _, _)).Times(1); |
1809 EXPECT_CALL(*send_algorithm_, OnIncomingLoss(_, _)).Times(1); | 1764 EXPECT_CALL(*send_algorithm_, OnIncomingLoss(_, _)).Times(1); |
1810 EXPECT_CALL(*send_algorithm_, OnPacketAbandoned(_, _)).Times(11); | 1765 EXPECT_CALL(*send_algorithm_, OnPacketAbandoned(_, _)).Times(11); |
1811 int offset = 0; | 1766 int offset = 0; |
1812 // Send packets 1 to 12. | 1767 // Send packets 1 to 12. |
1813 for (int i = 0; i < 12; ++i) { | 1768 for (int i = 0; i < 12; ++i) { |
1814 SendStreamDataToPeer(1, "foo", offset, !kFin, NULL); | 1769 SendStreamDataToPeer(1, "foo", offset, !kFin, NULL); |
1815 offset += 3; | 1770 offset += 3; |
1816 } | 1771 } |
(...skipping 14 matching lines...) Expand all Loading... |
1831 // The third call should trigger retransmitting 10 packets. | 1786 // The third call should trigger retransmitting 10 packets. |
1832 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)).Times(10); | 1787 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)).Times(10); |
1833 ProcessAckPacket(&nack); | 1788 ProcessAckPacket(&nack); |
1834 | 1789 |
1835 // The fourth call should trigger retransmitting the 11th packet. | 1790 // The fourth call should trigger retransmitting the 11th packet. |
1836 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)).Times(1); | 1791 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)).Times(1); |
1837 ProcessAckPacket(&nack); | 1792 ProcessAckPacket(&nack); |
1838 } | 1793 } |
1839 | 1794 |
1840 // Test sending multiple acks from the connection to the session. | 1795 // Test sending multiple acks from the connection to the session. |
1841 TEST_P(QuicConnectionTest, MultipleAcks) { | 1796 TEST_F(QuicConnectionTest, MultipleAcks) { |
1842 EXPECT_CALL(*send_algorithm_, OnIncomingAck(_, _, _)).Times(6); | 1797 EXPECT_CALL(*send_algorithm_, OnIncomingAck(_, _, _)).Times(6); |
1843 EXPECT_CALL(*send_algorithm_, OnIncomingLoss(_, _)).Times(1); | 1798 EXPECT_CALL(*send_algorithm_, OnIncomingLoss(_, _)).Times(1); |
1844 QuicPacketSequenceNumber last_packet; | 1799 QuicPacketSequenceNumber last_packet; |
1845 SendStreamDataToPeer(1, "foo", 0, !kFin, &last_packet); // Packet 1 | 1800 SendStreamDataToPeer(1, "foo", 0, !kFin, &last_packet); // Packet 1 |
1846 EXPECT_EQ(1u, last_packet); | 1801 EXPECT_EQ(1u, last_packet); |
1847 SendStreamDataToPeer(3, "foo", 0, !kFin, &last_packet); // Packet 2 | 1802 SendStreamDataToPeer(3, "foo", 0, !kFin, &last_packet); // Packet 2 |
1848 EXPECT_EQ(2u, last_packet); | 1803 EXPECT_EQ(2u, last_packet); |
1849 SendAckPacketToPeer(); // Packet 3 | 1804 SendAckPacketToPeer(); // Packet 3 |
1850 SendStreamDataToPeer(5, "foo", 0, !kFin, &last_packet); // Packet 4 | 1805 SendStreamDataToPeer(5, "foo", 0, !kFin, &last_packet); // Packet 4 |
1851 EXPECT_EQ(4u, last_packet); | 1806 EXPECT_EQ(4u, last_packet); |
(...skipping 15 matching lines...) Expand all Loading... |
1867 ProcessAckPacket(&frame1); | 1822 ProcessAckPacket(&frame1); |
1868 | 1823 |
1869 // Now the client implicitly acks 2, and explicitly acks 6. | 1824 // Now the client implicitly acks 2, and explicitly acks 6. |
1870 QuicAckFrame frame2(6, QuicTime::Zero(), 0); | 1825 QuicAckFrame frame2(6, QuicTime::Zero(), 0); |
1871 frame2.received_info.entropy_hash = | 1826 frame2.received_info.entropy_hash = |
1872 QuicConnectionPeer::GetSentEntropyHash(&connection_, 6); | 1827 QuicConnectionPeer::GetSentEntropyHash(&connection_, 6); |
1873 | 1828 |
1874 ProcessAckPacket(&frame2); | 1829 ProcessAckPacket(&frame2); |
1875 } | 1830 } |
1876 | 1831 |
1877 TEST_P(QuicConnectionTest, DontLatchUnackedPacket) { | 1832 TEST_F(QuicConnectionTest, DontLatchUnackedPacket) { |
1878 EXPECT_CALL(*send_algorithm_, OnIncomingAck(_, _, _)).Times(1); | 1833 EXPECT_CALL(*send_algorithm_, OnIncomingAck(_, _, _)).Times(1); |
1879 SendStreamDataToPeer(1, "foo", 0, !kFin, NULL); // Packet 1; | 1834 SendStreamDataToPeer(1, "foo", 0, !kFin, NULL); // Packet 1; |
1880 SendAckPacketToPeer(); // Packet 2 | 1835 SendAckPacketToPeer(); // Packet 2 |
1881 | 1836 |
1882 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | 1837 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); |
1883 QuicAckFrame frame(1, QuicTime::Zero(), 0); | 1838 QuicAckFrame frame(1, QuicTime::Zero(), 0); |
1884 frame.received_info.entropy_hash = QuicConnectionPeer::GetSentEntropyHash( | 1839 frame.received_info.entropy_hash = QuicConnectionPeer::GetSentEntropyHash( |
1885 &connection_, 1); | 1840 &connection_, 1); |
1886 ProcessAckPacket(&frame); | 1841 ProcessAckPacket(&frame); |
1887 | 1842 |
1888 // Verify that our internal state has least-unacked as 3. | 1843 // Verify that our internal state has least-unacked as 3. |
1889 EXPECT_EQ(3u, outgoing_ack()->sent_info.least_unacked); | 1844 EXPECT_EQ(3u, outgoing_ack()->sent_info.least_unacked); |
1890 | 1845 |
1891 // When we send an ack, we make sure our least-unacked makes sense. In this | 1846 // When we send an ack, we make sure our least-unacked makes sense. In this |
1892 // case since we're not waiting on an ack for 2 and all packets are acked, we | 1847 // case since we're not waiting on an ack for 2 and all packets are acked, we |
1893 // set it to 3. | 1848 // set it to 3. |
1894 SendAckPacketToPeer(); // Packet 3 | 1849 SendAckPacketToPeer(); // Packet 3 |
1895 // Since this was an ack packet, we set least_unacked to 4. | 1850 // Since this was an ack packet, we set least_unacked to 4. |
1896 EXPECT_EQ(4u, outgoing_ack()->sent_info.least_unacked); | 1851 EXPECT_EQ(4u, outgoing_ack()->sent_info.least_unacked); |
1897 // Check that the outgoing ack had its sequence number as least_unacked. | 1852 // Check that the outgoing ack had its sequence number as least_unacked. |
1898 EXPECT_EQ(3u, last_ack()->sent_info.least_unacked); | 1853 EXPECT_EQ(3u, last_ack()->sent_info.least_unacked); |
1899 | 1854 |
1900 SendStreamDataToPeer(1, "bar", 3, false, NULL); // Packet 4 | 1855 SendStreamDataToPeer(1, "bar", 3, false, NULL); // Packet 4 |
1901 EXPECT_EQ(4u, outgoing_ack()->sent_info.least_unacked); | 1856 EXPECT_EQ(4u, outgoing_ack()->sent_info.least_unacked); |
1902 SendAckPacketToPeer(); // Packet 5 | 1857 SendAckPacketToPeer(); // Packet 5 |
1903 EXPECT_EQ(4u, last_ack()->sent_info.least_unacked); | 1858 EXPECT_EQ(4u, last_ack()->sent_info.least_unacked); |
1904 } | 1859 } |
1905 | 1860 |
1906 TEST_P(QuicConnectionTest, ReviveMissingPacketAfterFecPacket) { | 1861 TEST_F(QuicConnectionTest, ReviveMissingPacketAfterFecPacket) { |
1907 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | 1862 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); |
1908 | 1863 |
1909 // Don't send missing packet 1. | 1864 // Don't send missing packet 1. |
1910 ProcessFecPacket(2, 1, true, !kEntropyFlag, NULL); | 1865 ProcessFecPacket(2, 1, true, !kEntropyFlag, NULL); |
1911 // Entropy flag should be false, so entropy should be 0. | 1866 // Entropy flag should be false, so entropy should be 0. |
1912 EXPECT_EQ(0u, QuicConnectionPeer::ReceivedEntropyHash(&connection_, 2)); | 1867 EXPECT_EQ(0u, QuicConnectionPeer::ReceivedEntropyHash(&connection_, 2)); |
1913 } | 1868 } |
1914 | 1869 |
1915 TEST_P(QuicConnectionTest, ReviveMissingPacketAfterDataPacketThenFecPacket) { | 1870 TEST_F(QuicConnectionTest, ReviveMissingPacketAfterDataPacketThenFecPacket) { |
1916 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | 1871 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); |
1917 | 1872 |
1918 ProcessFecProtectedPacket(1, false, kEntropyFlag); | 1873 ProcessFecProtectedPacket(1, false, kEntropyFlag); |
1919 // Don't send missing packet 2. | 1874 // Don't send missing packet 2. |
1920 ProcessFecPacket(3, 1, true, !kEntropyFlag, NULL); | 1875 ProcessFecPacket(3, 1, true, !kEntropyFlag, NULL); |
1921 // Entropy flag should be true, so entropy should not be 0. | 1876 // Entropy flag should be true, so entropy should not be 0. |
1922 EXPECT_NE(0u, QuicConnectionPeer::ReceivedEntropyHash(&connection_, 2)); | 1877 EXPECT_NE(0u, QuicConnectionPeer::ReceivedEntropyHash(&connection_, 2)); |
1923 } | 1878 } |
1924 | 1879 |
1925 TEST_P(QuicConnectionTest, ReviveMissingPacketAfterDataPacketsThenFecPacket) { | 1880 TEST_F(QuicConnectionTest, ReviveMissingPacketAfterDataPacketsThenFecPacket) { |
1926 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | 1881 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); |
1927 | 1882 |
1928 ProcessFecProtectedPacket(1, false, !kEntropyFlag); | 1883 ProcessFecProtectedPacket(1, false, !kEntropyFlag); |
1929 // Don't send missing packet 2. | 1884 // Don't send missing packet 2. |
1930 ProcessFecProtectedPacket(3, false, !kEntropyFlag); | 1885 ProcessFecProtectedPacket(3, false, !kEntropyFlag); |
1931 ProcessFecPacket(4, 1, true, kEntropyFlag, NULL); | 1886 ProcessFecPacket(4, 1, true, kEntropyFlag, NULL); |
1932 // Entropy flag should be true, so entropy should not be 0. | 1887 // Entropy flag should be true, so entropy should not be 0. |
1933 EXPECT_NE(0u, QuicConnectionPeer::ReceivedEntropyHash(&connection_, 2)); | 1888 EXPECT_NE(0u, QuicConnectionPeer::ReceivedEntropyHash(&connection_, 2)); |
1934 } | 1889 } |
1935 | 1890 |
1936 TEST_P(QuicConnectionTest, ReviveMissingPacketAfterDataPacket) { | 1891 TEST_F(QuicConnectionTest, ReviveMissingPacketAfterDataPacket) { |
1937 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | 1892 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); |
1938 | 1893 |
1939 // Don't send missing packet 1. | 1894 // Don't send missing packet 1. |
1940 ProcessFecPacket(3, 1, false, !kEntropyFlag, NULL); | 1895 ProcessFecPacket(3, 1, false, !kEntropyFlag, NULL); |
1941 // Out of order. | 1896 // Out of order. |
1942 ProcessFecProtectedPacket(2, true, !kEntropyFlag); | 1897 ProcessFecProtectedPacket(2, true, !kEntropyFlag); |
1943 // Entropy flag should be false, so entropy should be 0. | 1898 // Entropy flag should be false, so entropy should be 0. |
1944 EXPECT_EQ(0u, QuicConnectionPeer::ReceivedEntropyHash(&connection_, 2)); | 1899 EXPECT_EQ(0u, QuicConnectionPeer::ReceivedEntropyHash(&connection_, 2)); |
1945 } | 1900 } |
1946 | 1901 |
1947 TEST_P(QuicConnectionTest, ReviveMissingPacketAfterDataPackets) { | 1902 TEST_F(QuicConnectionTest, ReviveMissingPacketAfterDataPackets) { |
1948 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | 1903 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); |
1949 | 1904 |
1950 ProcessFecProtectedPacket(1, false, !kEntropyFlag); | 1905 ProcessFecProtectedPacket(1, false, !kEntropyFlag); |
1951 // Don't send missing packet 2. | 1906 // Don't send missing packet 2. |
1952 ProcessFecPacket(6, 1, false, kEntropyFlag, NULL); | 1907 ProcessFecPacket(6, 1, false, kEntropyFlag, NULL); |
1953 ProcessFecProtectedPacket(3, false, kEntropyFlag); | 1908 ProcessFecProtectedPacket(3, false, kEntropyFlag); |
1954 ProcessFecProtectedPacket(4, false, kEntropyFlag); | 1909 ProcessFecProtectedPacket(4, false, kEntropyFlag); |
1955 ProcessFecProtectedPacket(5, true, !kEntropyFlag); | 1910 ProcessFecProtectedPacket(5, true, !kEntropyFlag); |
1956 // Entropy flag should be true, so entropy should be 0. | 1911 // Entropy flag should be true, so entropy should be 0. |
1957 EXPECT_NE(0u, QuicConnectionPeer::ReceivedEntropyHash(&connection_, 2)); | 1912 EXPECT_NE(0u, QuicConnectionPeer::ReceivedEntropyHash(&connection_, 2)); |
1958 } | 1913 } |
1959 | 1914 |
1960 TEST_P(QuicConnectionTest, TestRetransmit) { | 1915 TEST_F(QuicConnectionTest, TestRetransmit) { |
1961 QuicTime default_retransmission_time = clock_.ApproximateNow().Add( | 1916 QuicTime default_retransmission_time = clock_.ApproximateNow().Add( |
1962 DefaultRetransmissionTime()); | 1917 DefaultRetransmissionTime()); |
1963 SendStreamDataToPeer(1, "foo", 0, !kFin, NULL); | 1918 SendStreamDataToPeer(1, "foo", 0, !kFin, NULL); |
1964 EXPECT_EQ(1u, outgoing_ack()->sent_info.least_unacked); | 1919 EXPECT_EQ(1u, outgoing_ack()->sent_info.least_unacked); |
1965 | 1920 |
1966 EXPECT_EQ(1u, last_header()->packet_sequence_number); | 1921 EXPECT_EQ(1u, last_header()->packet_sequence_number); |
1967 EXPECT_EQ(default_retransmission_time, | 1922 EXPECT_EQ(default_retransmission_time, |
1968 connection_.GetRetransmissionAlarm()->deadline()); | 1923 connection_.GetRetransmissionAlarm()->deadline()); |
1969 // Simulate the retransimission alarm firing. | 1924 // Simulate the retransimission alarm firing. |
1970 clock_.AdvanceTime(DefaultRetransmissionTime()); | 1925 clock_.AdvanceTime(DefaultRetransmissionTime()); |
1971 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)); | 1926 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)); |
1972 EXPECT_CALL(*send_algorithm_, OnPacketAbandoned(1, _)).Times(1); | 1927 EXPECT_CALL(*send_algorithm_, OnPacketAbandoned(1, _)).Times(1); |
1973 connection_.RetransmitPacket(1, RTO_RETRANSMISSION); | 1928 connection_.RetransmitPacket(1, RTO_RETRANSMISSION); |
1974 EXPECT_EQ(2u, last_header()->packet_sequence_number); | 1929 EXPECT_EQ(2u, last_header()->packet_sequence_number); |
1975 // We do not raise the high water mark yet. | 1930 // We do not raise the high water mark yet. |
1976 EXPECT_EQ(1u, outgoing_ack()->sent_info.least_unacked); | 1931 EXPECT_EQ(1u, outgoing_ack()->sent_info.least_unacked); |
1977 } | 1932 } |
1978 | 1933 |
1979 TEST_P(QuicConnectionTest, RetransmitWithSameEncryptionLevel) { | 1934 TEST_F(QuicConnectionTest, RetransmitWithSameEncryptionLevel) { |
1980 QuicTime default_retransmission_time = clock_.ApproximateNow().Add( | 1935 QuicTime default_retransmission_time = clock_.ApproximateNow().Add( |
1981 DefaultRetransmissionTime()); | 1936 DefaultRetransmissionTime()); |
1982 use_tagging_decrypter(); | 1937 use_tagging_decrypter(); |
1983 | 1938 |
1984 // A TaggingEncrypter puts kTagSize copies of the given byte (0x01 here) at | 1939 // A TaggingEncrypter puts kTagSize copies of the given byte (0x01 here) at |
1985 // the end of the packet. We can test this to check which encrypter was used. | 1940 // the end of the packet. We can test this to check which encrypter was used. |
1986 connection_.SetEncrypter(ENCRYPTION_NONE, new TaggingEncrypter(0x01)); | 1941 connection_.SetEncrypter(ENCRYPTION_NONE, new TaggingEncrypter(0x01)); |
1987 SendStreamDataToPeer(1, "foo", 0, !kFin, NULL); | 1942 SendStreamDataToPeer(1, "foo", 0, !kFin, NULL); |
1988 EXPECT_EQ(0x01010101u, final_bytes_of_last_packet()); | 1943 EXPECT_EQ(0x01010101u, final_bytes_of_last_packet()); |
1989 | 1944 |
(...skipping 12 matching lines...) Expand all Loading... |
2002 connection_.RetransmitPacket(1, RTO_RETRANSMISSION); | 1957 connection_.RetransmitPacket(1, RTO_RETRANSMISSION); |
2003 // Packet should have been sent with ENCRYPTION_NONE. | 1958 // Packet should have been sent with ENCRYPTION_NONE. |
2004 EXPECT_EQ(0x01010101u, final_bytes_of_last_packet()); | 1959 EXPECT_EQ(0x01010101u, final_bytes_of_last_packet()); |
2005 | 1960 |
2006 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)); | 1961 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)); |
2007 connection_.RetransmitPacket(2, RTO_RETRANSMISSION); | 1962 connection_.RetransmitPacket(2, RTO_RETRANSMISSION); |
2008 // Packet should have been sent with ENCRYPTION_INITIAL. | 1963 // Packet should have been sent with ENCRYPTION_INITIAL. |
2009 EXPECT_EQ(0x02020202u, final_bytes_of_last_packet()); | 1964 EXPECT_EQ(0x02020202u, final_bytes_of_last_packet()); |
2010 } | 1965 } |
2011 | 1966 |
2012 TEST_P(QuicConnectionTest, SendHandshakeMessages) { | 1967 TEST_F(QuicConnectionTest, SendHandshakeMessages) { |
2013 use_tagging_decrypter(); | 1968 use_tagging_decrypter(); |
2014 // A TaggingEncrypter puts kTagSize copies of the given byte (0x01 here) at | 1969 // A TaggingEncrypter puts kTagSize copies of the given byte (0x01 here) at |
2015 // the end of the packet. We can test this to check which encrypter was used. | 1970 // the end of the packet. We can test this to check which encrypter was used. |
2016 connection_.SetEncrypter(ENCRYPTION_NONE, new TaggingEncrypter(0x01)); | 1971 connection_.SetEncrypter(ENCRYPTION_NONE, new TaggingEncrypter(0x01)); |
2017 | 1972 |
2018 // Attempt to send a handshake message while the congestion manager | 1973 // Attempt to send a handshake message while the congestion manager |
2019 // does not permit sending. | 1974 // does not permit sending. |
2020 EXPECT_CALL(*send_algorithm_, | 1975 EXPECT_CALL(*send_algorithm_, |
2021 TimeUntilSend(_, _, _, IS_HANDSHAKE)).WillRepeatedly( | 1976 TimeUntilSend(_, _, _, IS_HANDSHAKE)).WillRepeatedly( |
2022 testing::Return(QuicTime::Delta::Infinite())); | 1977 testing::Return(QuicTime::Delta::Infinite())); |
(...skipping 10 matching lines...) Expand all Loading... |
2033 TimeUntilSend(_, _, _, IS_HANDSHAKE)).WillRepeatedly( | 1988 TimeUntilSend(_, _, _, IS_HANDSHAKE)).WillRepeatedly( |
2034 testing::Return(QuicTime::Delta::Zero())); | 1989 testing::Return(QuicTime::Delta::Zero())); |
2035 EXPECT_CALL(visitor_, OnCanWrite()); | 1990 EXPECT_CALL(visitor_, OnCanWrite()); |
2036 connection_.OnCanWrite(); | 1991 connection_.OnCanWrite(); |
2037 EXPECT_EQ(0u, connection_.NumQueuedPackets()); | 1992 EXPECT_EQ(0u, connection_.NumQueuedPackets()); |
2038 | 1993 |
2039 // Verify that the handshake packet went out at the null encryption. | 1994 // Verify that the handshake packet went out at the null encryption. |
2040 EXPECT_EQ(0x01010101u, final_bytes_of_last_packet()); | 1995 EXPECT_EQ(0x01010101u, final_bytes_of_last_packet()); |
2041 } | 1996 } |
2042 | 1997 |
2043 TEST_P(QuicConnectionTest, | 1998 TEST_F(QuicConnectionTest, |
2044 DropRetransmitsForNullEncryptedPacketAfterForwardSecure) { | 1999 DropRetransmitsForNullEncryptedPacketAfterForwardSecure) { |
2045 use_tagging_decrypter(); | 2000 use_tagging_decrypter(); |
2046 connection_.SetEncrypter(ENCRYPTION_NONE, new TaggingEncrypter(0x01)); | 2001 connection_.SetEncrypter(ENCRYPTION_NONE, new TaggingEncrypter(0x01)); |
2047 QuicPacketSequenceNumber sequence_number; | 2002 QuicPacketSequenceNumber sequence_number; |
2048 SendStreamDataToPeer(1, "foo", 0, !kFin, &sequence_number); | 2003 SendStreamDataToPeer(1, "foo", 0, !kFin, &sequence_number); |
2049 | 2004 |
2050 connection_.SetEncrypter(ENCRYPTION_FORWARD_SECURE, | 2005 connection_.SetEncrypter(ENCRYPTION_FORWARD_SECURE, |
2051 new TaggingEncrypter(0x02)); | 2006 new TaggingEncrypter(0x02)); |
2052 connection_.SetDefaultEncryptionLevel(ENCRYPTION_FORWARD_SECURE); | 2007 connection_.SetDefaultEncryptionLevel(ENCRYPTION_FORWARD_SECURE); |
2053 | 2008 |
2054 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)).Times(0); | 2009 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)).Times(0); |
2055 EXPECT_CALL(*send_algorithm_, OnPacketAbandoned(sequence_number, _)).Times(1); | 2010 EXPECT_CALL(*send_algorithm_, OnPacketAbandoned(sequence_number, _)).Times(1); |
2056 | 2011 |
2057 QuicTime default_retransmission_time = clock_.ApproximateNow().Add( | 2012 QuicTime default_retransmission_time = clock_.ApproximateNow().Add( |
2058 DefaultRetransmissionTime()); | 2013 DefaultRetransmissionTime()); |
2059 | 2014 |
2060 EXPECT_EQ(default_retransmission_time, | 2015 EXPECT_EQ(default_retransmission_time, |
2061 connection_.GetRetransmissionAlarm()->deadline()); | 2016 connection_.GetRetransmissionAlarm()->deadline()); |
2062 // Simulate the retransimission alarm firing. | 2017 // Simulate the retransimission alarm firing. |
2063 clock_.AdvanceTime(DefaultRetransmissionTime()); | 2018 clock_.AdvanceTime(DefaultRetransmissionTime()); |
2064 connection_.GetRetransmissionAlarm()->Fire(); | 2019 connection_.GetRetransmissionAlarm()->Fire(); |
2065 } | 2020 } |
2066 | 2021 |
2067 TEST_P(QuicConnectionTest, RetransmitPacketsWithInitialEncryption) { | 2022 TEST_F(QuicConnectionTest, RetransmitPacketsWithInitialEncryption) { |
2068 use_tagging_decrypter(); | 2023 use_tagging_decrypter(); |
2069 connection_.SetEncrypter(ENCRYPTION_NONE, new TaggingEncrypter(0x01)); | 2024 connection_.SetEncrypter(ENCRYPTION_NONE, new TaggingEncrypter(0x01)); |
2070 connection_.SetDefaultEncryptionLevel(ENCRYPTION_NONE); | 2025 connection_.SetDefaultEncryptionLevel(ENCRYPTION_NONE); |
2071 | 2026 |
2072 SendStreamDataToPeer(1, "foo", 0, !kFin, NULL); | 2027 SendStreamDataToPeer(1, "foo", 0, !kFin, NULL); |
2073 | 2028 |
2074 connection_.SetEncrypter(ENCRYPTION_INITIAL, new TaggingEncrypter(0x02)); | 2029 connection_.SetEncrypter(ENCRYPTION_INITIAL, new TaggingEncrypter(0x02)); |
2075 connection_.SetDefaultEncryptionLevel(ENCRYPTION_INITIAL); | 2030 connection_.SetDefaultEncryptionLevel(ENCRYPTION_INITIAL); |
2076 | 2031 |
2077 SendStreamDataToPeer(2, "bar", 0, !kFin, NULL); | 2032 SendStreamDataToPeer(2, "bar", 0, !kFin, NULL); |
2078 | 2033 |
2079 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)).Times(1); | 2034 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)).Times(1); |
2080 EXPECT_CALL(*send_algorithm_, OnPacketAbandoned(_, _)).Times(1); | 2035 EXPECT_CALL(*send_algorithm_, OnPacketAbandoned(_, _)).Times(1); |
2081 | 2036 |
2082 connection_.RetransmitUnackedPackets(QuicConnection::INITIAL_ENCRYPTION_ONLY); | 2037 connection_.RetransmitUnackedPackets(QuicConnection::INITIAL_ENCRYPTION_ONLY); |
2083 } | 2038 } |
2084 | 2039 |
2085 TEST_P(QuicConnectionTest, BufferNonDecryptablePackets) { | 2040 TEST_F(QuicConnectionTest, BufferNonDecryptablePackets) { |
2086 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | 2041 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); |
2087 use_tagging_decrypter(); | 2042 use_tagging_decrypter(); |
2088 | 2043 |
2089 const uint8 tag = 0x07; | 2044 const uint8 tag = 0x07; |
2090 framer_.SetEncrypter(ENCRYPTION_INITIAL, new TaggingEncrypter(tag)); | 2045 framer_.SetEncrypter(ENCRYPTION_INITIAL, new TaggingEncrypter(tag)); |
2091 | 2046 |
2092 // Process an encrypted packet which can not yet be decrypted | 2047 // Process an encrypted packet which can not yet be decrypted |
2093 // which should result in the packet being buffered. | 2048 // which should result in the packet being buffered. |
2094 ProcessDataPacketAtLevel(1, false, kEntropyFlag, ENCRYPTION_INITIAL); | 2049 ProcessDataPacketAtLevel(1, false, kEntropyFlag, ENCRYPTION_INITIAL); |
2095 | 2050 |
2096 // Transition to the new encryption state and process another | 2051 // Transition to the new encryption state and process another |
2097 // encrypted packet which should result in the original packet being | 2052 // encrypted packet which should result in the original packet being |
2098 // processed. | 2053 // processed. |
2099 connection_.SetDecrypter(new StrictTaggingDecrypter(tag)); | 2054 connection_.SetDecrypter(new StrictTaggingDecrypter(tag)); |
2100 connection_.SetDefaultEncryptionLevel(ENCRYPTION_INITIAL); | 2055 connection_.SetDefaultEncryptionLevel(ENCRYPTION_INITIAL); |
2101 connection_.SetEncrypter(ENCRYPTION_INITIAL, new TaggingEncrypter(tag)); | 2056 connection_.SetEncrypter(ENCRYPTION_INITIAL, new TaggingEncrypter(tag)); |
2102 EXPECT_CALL(visitor_, OnStreamFrames(_)).Times(2).WillRepeatedly( | 2057 EXPECT_CALL(visitor_, OnStreamFrames(_)).Times(2).WillRepeatedly( |
2103 Return(true)); | 2058 Return(true)); |
2104 ProcessDataPacketAtLevel(2, false, kEntropyFlag, ENCRYPTION_INITIAL); | 2059 ProcessDataPacketAtLevel(2, false, kEntropyFlag, ENCRYPTION_INITIAL); |
2105 | 2060 |
2106 // Finally, process a third packet and note that we do not | 2061 // Finally, process a third packet and note that we do not |
2107 // reprocess the buffered packet. | 2062 // reprocess the buffered packet. |
2108 EXPECT_CALL(visitor_, OnStreamFrames(_)).WillOnce(Return(true)); | 2063 EXPECT_CALL(visitor_, OnStreamFrames(_)).WillOnce(Return(true)); |
2109 ProcessDataPacketAtLevel(3, false, kEntropyFlag, ENCRYPTION_INITIAL); | 2064 ProcessDataPacketAtLevel(3, false, kEntropyFlag, ENCRYPTION_INITIAL); |
2110 } | 2065 } |
2111 | 2066 |
2112 TEST_P(QuicConnectionTest, TestRetransmitOrder) { | 2067 TEST_F(QuicConnectionTest, TestRetransmitOrder) { |
2113 QuicByteCount first_packet_size; | 2068 QuicByteCount first_packet_size; |
2114 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)).WillOnce( | 2069 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)).WillOnce( |
2115 DoAll(SaveArg<2>(&first_packet_size), Return(true))); | 2070 DoAll(SaveArg<2>(&first_packet_size), Return(true))); |
2116 | 2071 |
2117 connection_.SendStreamDataWithString(3, "first_packet", 0, !kFin, NULL); | 2072 connection_.SendStreamDataWithString(3, "first_packet", 0, !kFin, NULL); |
2118 QuicByteCount second_packet_size; | 2073 QuicByteCount second_packet_size; |
2119 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)).WillOnce( | 2074 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)).WillOnce( |
2120 DoAll(SaveArg<2>(&second_packet_size), Return(true))); | 2075 DoAll(SaveArg<2>(&second_packet_size), Return(true))); |
2121 connection_.SendStreamDataWithString(3, "second_packet", 12, !kFin, NULL); | 2076 connection_.SendStreamDataWithString(3, "second_packet", 12, !kFin, NULL); |
2122 EXPECT_NE(first_packet_size, second_packet_size); | 2077 EXPECT_NE(first_packet_size, second_packet_size); |
(...skipping 15 matching lines...) Expand all Loading... |
2138 { | 2093 { |
2139 InSequence s; | 2094 InSequence s; |
2140 EXPECT_CALL(*send_algorithm_, | 2095 EXPECT_CALL(*send_algorithm_, |
2141 OnPacketSent(_, _, first_packet_size, _, _)); | 2096 OnPacketSent(_, _, first_packet_size, _, _)); |
2142 EXPECT_CALL(*send_algorithm_, | 2097 EXPECT_CALL(*send_algorithm_, |
2143 OnPacketSent(_, _, second_packet_size, _, _)); | 2098 OnPacketSent(_, _, second_packet_size, _, _)); |
2144 } | 2099 } |
2145 connection_.GetRetransmissionAlarm()->Fire(); | 2100 connection_.GetRetransmissionAlarm()->Fire(); |
2146 } | 2101 } |
2147 | 2102 |
2148 TEST_P(QuicConnectionTest, TestRetransmissionCountCalculation) { | 2103 TEST_F(QuicConnectionTest, TestRetransmissionTracking) { |
2149 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | 2104 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); |
2150 QuicPacketSequenceNumber original_sequence_number; | 2105 QuicPacketSequenceNumber original_sequence_number; |
2151 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, NOT_RETRANSMISSION, _)) | 2106 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, NOT_RETRANSMISSION, _)) |
2152 .WillOnce(DoAll(SaveArg<1>(&original_sequence_number), Return(true))); | 2107 .WillOnce(DoAll(SaveArg<1>(&original_sequence_number), Return(true))); |
2153 connection_.SendStreamDataWithString(1, "foo", 0, !kFin, NULL); | 2108 connection_.SendStreamDataWithString(1, "foo", 0, !kFin, NULL); |
2154 | 2109 |
2155 EXPECT_TRUE(QuicConnectionPeer::IsSavedForRetransmission( | 2110 EXPECT_TRUE(QuicConnectionPeer::IsSavedForRetransmission( |
2156 &connection_, original_sequence_number)); | 2111 &connection_, original_sequence_number)); |
2157 EXPECT_EQ(0u, QuicConnectionPeer::GetRetransmissionCount( | 2112 EXPECT_FALSE(QuicConnectionPeer::IsRetransmission( |
2158 &connection_, original_sequence_number)); | 2113 &connection_, original_sequence_number)); |
2159 // Force retransmission due to RTO. | 2114 // Force retransmission due to RTO. |
2160 clock_.AdvanceTime(QuicTime::Delta::FromSeconds(10)); | 2115 clock_.AdvanceTime(QuicTime::Delta::FromSeconds(10)); |
2161 EXPECT_CALL(*send_algorithm_, | 2116 EXPECT_CALL(*send_algorithm_, |
2162 OnPacketAbandoned(original_sequence_number, _)).Times(1); | 2117 OnPacketAbandoned(original_sequence_number, _)).Times(1); |
2163 QuicPacketSequenceNumber rto_sequence_number; | 2118 QuicPacketSequenceNumber rto_sequence_number; |
2164 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, RTO_RETRANSMISSION, _)) | 2119 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, RTO_RETRANSMISSION, _)) |
2165 .WillOnce(DoAll(SaveArg<1>(&rto_sequence_number), Return(true))); | 2120 .WillOnce(DoAll(SaveArg<1>(&rto_sequence_number), Return(true))); |
2166 connection_.GetRetransmissionAlarm()->Fire(); | 2121 connection_.GetRetransmissionAlarm()->Fire(); |
2167 EXPECT_FALSE(QuicConnectionPeer::IsSavedForRetransmission( | 2122 EXPECT_FALSE(QuicConnectionPeer::IsSavedForRetransmission( |
2168 &connection_, original_sequence_number)); | 2123 &connection_, original_sequence_number)); |
2169 ASSERT_TRUE(QuicConnectionPeer::IsSavedForRetransmission( | 2124 ASSERT_TRUE(QuicConnectionPeer::IsSavedForRetransmission( |
2170 &connection_, rto_sequence_number)); | 2125 &connection_, rto_sequence_number)); |
2171 EXPECT_EQ(1u, QuicConnectionPeer::GetRetransmissionCount( | 2126 EXPECT_TRUE(QuicConnectionPeer::IsRetransmission( |
2172 &connection_, rto_sequence_number)); | 2127 &connection_, rto_sequence_number)); |
2173 // Once by explicit nack. | 2128 // Once by explicit nack. |
2174 EXPECT_CALL(*send_algorithm_, OnIncomingLoss(_, _)).Times(1); | 2129 EXPECT_CALL(*send_algorithm_, OnIncomingLoss(_, _)).Times(1); |
2175 EXPECT_CALL(*send_algorithm_, | 2130 EXPECT_CALL(*send_algorithm_, |
2176 OnPacketAbandoned(rto_sequence_number, _)).Times(1); | 2131 OnPacketAbandoned(rto_sequence_number, _)).Times(1); |
2177 QuicPacketSequenceNumber nack_sequence_number = 0; | 2132 QuicPacketSequenceNumber nack_sequence_number = 0; |
2178 // Ack packets might generate some other packets, which are not | 2133 // Ack packets might generate some other packets, which are not |
2179 // retransmissions. (More ack packets). | 2134 // retransmissions. (More ack packets). |
2180 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, NOT_RETRANSMISSION, _)) | 2135 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, NOT_RETRANSMISSION, _)) |
2181 .Times(AnyNumber()); | 2136 .Times(AnyNumber()); |
2182 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, NACK_RETRANSMISSION, _)) | 2137 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, NACK_RETRANSMISSION, _)) |
2183 .WillOnce(DoAll(SaveArg<1>(&nack_sequence_number), Return(true))); | 2138 .WillOnce(DoAll(SaveArg<1>(&nack_sequence_number), Return(true))); |
2184 QuicAckFrame ack(rto_sequence_number, QuicTime::Zero(), 0); | 2139 QuicAckFrame ack(rto_sequence_number, QuicTime::Zero(), 0); |
2185 // Ack the retransmitted packet. | 2140 // Ack the retransmitted packet. |
2186 ack.received_info.missing_packets.insert(original_sequence_number); | 2141 ack.received_info.missing_packets.insert(original_sequence_number); |
2187 ack.received_info.missing_packets.insert(rto_sequence_number); | 2142 ack.received_info.missing_packets.insert(rto_sequence_number); |
2188 ack.received_info.entropy_hash = QuicConnectionPeer::GetSentEntropyHash( | 2143 ack.received_info.entropy_hash = QuicConnectionPeer::GetSentEntropyHash( |
2189 &connection_, rto_sequence_number - 1); | 2144 &connection_, rto_sequence_number - 1); |
2190 for (int i = 0; i < 3; i++) { | 2145 for (int i = 0; i < 3; i++) { |
2191 ProcessAckPacket(&ack); | 2146 ProcessAckPacket(&ack); |
2192 } | 2147 } |
2193 ASSERT_NE(0u, nack_sequence_number); | 2148 ASSERT_NE(0u, nack_sequence_number); |
2194 EXPECT_FALSE(QuicConnectionPeer::IsSavedForRetransmission( | 2149 EXPECT_FALSE(QuicConnectionPeer::IsSavedForRetransmission( |
2195 &connection_, rto_sequence_number)); | 2150 &connection_, rto_sequence_number)); |
2196 ASSERT_TRUE(QuicConnectionPeer::IsSavedForRetransmission( | 2151 ASSERT_TRUE(QuicConnectionPeer::IsSavedForRetransmission( |
2197 &connection_, nack_sequence_number)); | 2152 &connection_, nack_sequence_number)); |
2198 EXPECT_EQ(2u, QuicConnectionPeer::GetRetransmissionCount( | 2153 EXPECT_TRUE(QuicConnectionPeer::IsRetransmission( |
2199 &connection_, nack_sequence_number)); | 2154 &connection_, nack_sequence_number)); |
2200 } | 2155 } |
2201 | 2156 |
2202 TEST_P(QuicConnectionTest, SetRTOAfterWritingToSocket) { | 2157 TEST_F(QuicConnectionTest, SetRTOAfterWritingToSocket) { |
2203 writer_->set_blocked(true); | 2158 writer_->set_blocked(true); |
2204 connection_.SendStreamDataWithString(1, "foo", 0, !kFin, NULL); | 2159 connection_.SendStreamDataWithString(1, "foo", 0, !kFin, NULL); |
2205 // Make sure that RTO is not started when the packet is queued. | 2160 // Make sure that RTO is not started when the packet is queued. |
2206 EXPECT_FALSE(connection_.GetRetransmissionAlarm()->IsSet()); | 2161 EXPECT_FALSE(connection_.GetRetransmissionAlarm()->IsSet()); |
2207 | 2162 |
2208 // Test that RTO is started once we write to the socket. | 2163 // Test that RTO is started once we write to the socket. |
2209 writer_->set_blocked(false); | 2164 writer_->set_blocked(false); |
2210 connection_.OnCanWrite(); | 2165 connection_.OnCanWrite(); |
2211 EXPECT_TRUE(connection_.GetRetransmissionAlarm()->IsSet()); | 2166 EXPECT_TRUE(connection_.GetRetransmissionAlarm()->IsSet()); |
2212 } | 2167 } |
2213 | 2168 |
2214 TEST_P(QuicConnectionTest, DelayRTOWithAckReceipt) { | 2169 TEST_F(QuicConnectionTest, DelayRTOWithAckReceipt) { |
2215 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | 2170 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); |
2216 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, NOT_RETRANSMISSION, _)) | 2171 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, NOT_RETRANSMISSION, _)) |
2217 .Times(2); | 2172 .Times(2); |
2218 connection_.SendStreamDataWithString(1, "foo", 0, !kFin, NULL); | 2173 connection_.SendStreamDataWithString(1, "foo", 0, !kFin, NULL); |
2219 connection_.SendStreamDataWithString(2, "bar", 0, !kFin, NULL); | 2174 connection_.SendStreamDataWithString(2, "bar", 0, !kFin, NULL); |
2220 QuicAlarm* retransmission_alarm = connection_.GetRetransmissionAlarm(); | 2175 QuicAlarm* retransmission_alarm = connection_.GetRetransmissionAlarm(); |
2221 EXPECT_TRUE(retransmission_alarm->IsSet()); | 2176 EXPECT_TRUE(retransmission_alarm->IsSet()); |
2222 | 2177 |
2223 // Advance the time right before the RTO, then receive an ack for the first | 2178 // Advance the time right before the RTO, then receive an ack for the first |
2224 // packet to delay the RTO. | 2179 // packet to delay the RTO. |
(...skipping 18 matching lines...) Expand all Loading... |
2243 | 2198 |
2244 // The new retransmitted sequence number should set the RTO to a larger value | 2199 // The new retransmitted sequence number should set the RTO to a larger value |
2245 // than previously. | 2200 // than previously. |
2246 EXPECT_TRUE(retransmission_alarm->IsSet()); | 2201 EXPECT_TRUE(retransmission_alarm->IsSet()); |
2247 QuicTime next_rto_time = retransmission_alarm->deadline(); | 2202 QuicTime next_rto_time = retransmission_alarm->deadline(); |
2248 QuicTime::Delta expected_rto = | 2203 QuicTime::Delta expected_rto = |
2249 connection_.congestion_manager().GetRetransmissionDelay(1, 1); | 2204 connection_.congestion_manager().GetRetransmissionDelay(1, 1); |
2250 EXPECT_EQ(next_rto_time, clock_.ApproximateNow().Add(expected_rto)); | 2205 EXPECT_EQ(next_rto_time, clock_.ApproximateNow().Add(expected_rto)); |
2251 } | 2206 } |
2252 | 2207 |
2253 TEST_P(QuicConnectionTest, TestQueued) { | 2208 TEST_F(QuicConnectionTest, TestQueued) { |
2254 EXPECT_EQ(0u, connection_.NumQueuedPackets()); | 2209 EXPECT_EQ(0u, connection_.NumQueuedPackets()); |
2255 writer_->set_blocked(true); | 2210 writer_->set_blocked(true); |
2256 connection_.SendStreamDataWithString(1, "foo", 0, !kFin, NULL); | 2211 connection_.SendStreamDataWithString(1, "foo", 0, !kFin, NULL); |
2257 EXPECT_EQ(1u, connection_.NumQueuedPackets()); | 2212 EXPECT_EQ(1u, connection_.NumQueuedPackets()); |
2258 | 2213 |
2259 // Attempt to send all packets, but since we're actually still | 2214 // Attempt to send all packets, but since we're actually still |
2260 // blocked, they should all remain queued. | 2215 // blocked, they should all remain queued. |
2261 EXPECT_FALSE(connection_.OnCanWrite()); | 2216 EXPECT_FALSE(connection_.OnCanWrite()); |
2262 EXPECT_EQ(1u, connection_.NumQueuedPackets()); | 2217 EXPECT_EQ(1u, connection_.NumQueuedPackets()); |
2263 | 2218 |
2264 // Unblock the writes and actually send. | 2219 // Unblock the writes and actually send. |
2265 writer_->set_blocked(false); | 2220 writer_->set_blocked(false); |
2266 EXPECT_TRUE(connection_.OnCanWrite()); | 2221 EXPECT_TRUE(connection_.OnCanWrite()); |
2267 EXPECT_EQ(0u, connection_.NumQueuedPackets()); | 2222 EXPECT_EQ(0u, connection_.NumQueuedPackets()); |
2268 } | 2223 } |
2269 | 2224 |
2270 TEST_P(QuicConnectionTest, CloseFecGroup) { | 2225 TEST_F(QuicConnectionTest, CloseFecGroup) { |
2271 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | 2226 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); |
2272 // Don't send missing packet 1. | 2227 // Don't send missing packet 1. |
2273 // Don't send missing packet 2. | 2228 // Don't send missing packet 2. |
2274 ProcessFecProtectedPacket(3, false, !kEntropyFlag); | 2229 ProcessFecProtectedPacket(3, false, !kEntropyFlag); |
2275 // Don't send missing FEC packet 3. | 2230 // Don't send missing FEC packet 3. |
2276 ASSERT_EQ(1u, connection_.NumFecGroups()); | 2231 ASSERT_EQ(1u, connection_.NumFecGroups()); |
2277 | 2232 |
2278 // Now send non-fec protected ack packet and close the group. | 2233 // Now send non-fec protected ack packet and close the group. |
2279 QuicAckFrame frame(0, QuicTime::Zero(), 5); | 2234 QuicAckFrame frame(0, QuicTime::Zero(), 5); |
2280 creator_.set_sequence_number(4); | 2235 creator_.set_sequence_number(4); |
2281 ProcessAckPacket(&frame); | 2236 ProcessAckPacket(&frame); |
2282 ASSERT_EQ(0u, connection_.NumFecGroups()); | 2237 ASSERT_EQ(0u, connection_.NumFecGroups()); |
2283 } | 2238 } |
2284 | 2239 |
2285 TEST_P(QuicConnectionTest, NoQuicCongestionFeedbackFrame) { | 2240 TEST_F(QuicConnectionTest, NoQuicCongestionFeedbackFrame) { |
2286 SendAckPacketToPeer(); | 2241 SendAckPacketToPeer(); |
2287 EXPECT_TRUE(last_feedback() == NULL); | 2242 EXPECT_TRUE(last_feedback() == NULL); |
2288 } | 2243 } |
2289 | 2244 |
2290 TEST_P(QuicConnectionTest, WithQuicCongestionFeedbackFrame) { | 2245 TEST_F(QuicConnectionTest, WithQuicCongestionFeedbackFrame) { |
2291 QuicCongestionFeedbackFrame info; | 2246 QuicCongestionFeedbackFrame info; |
2292 info.type = kFixRate; | 2247 info.type = kFixRate; |
2293 info.fix_rate.bitrate = QuicBandwidth::FromBytesPerSecond(123); | 2248 info.fix_rate.bitrate = QuicBandwidth::FromBytesPerSecond(123); |
2294 SetFeedback(&info); | 2249 SetFeedback(&info); |
2295 | 2250 |
2296 SendAckPacketToPeer(); | 2251 SendAckPacketToPeer(); |
2297 EXPECT_EQ(kFixRate, last_feedback()->type); | 2252 EXPECT_EQ(kFixRate, last_feedback()->type); |
2298 EXPECT_EQ(info.fix_rate.bitrate, last_feedback()->fix_rate.bitrate); | 2253 EXPECT_EQ(info.fix_rate.bitrate, last_feedback()->fix_rate.bitrate); |
2299 } | 2254 } |
2300 | 2255 |
2301 TEST_P(QuicConnectionTest, UpdateQuicCongestionFeedbackFrame) { | 2256 TEST_F(QuicConnectionTest, UpdateQuicCongestionFeedbackFrame) { |
2302 SendAckPacketToPeer(); | 2257 SendAckPacketToPeer(); |
2303 EXPECT_CALL(*receive_algorithm_, RecordIncomingPacket(_, _, _, _)); | 2258 EXPECT_CALL(*receive_algorithm_, RecordIncomingPacket(_, _, _, _)); |
2304 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | 2259 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); |
2305 ProcessPacket(1); | 2260 ProcessPacket(1); |
2306 } | 2261 } |
2307 | 2262 |
2308 TEST_P(QuicConnectionTest, DontUpdateQuicCongestionFeedbackFrameForRevived) { | 2263 TEST_F(QuicConnectionTest, DontUpdateQuicCongestionFeedbackFrameForRevived) { |
2309 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | 2264 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); |
2310 SendAckPacketToPeer(); | 2265 SendAckPacketToPeer(); |
2311 // Process an FEC packet, and revive the missing data packet | 2266 // Process an FEC packet, and revive the missing data packet |
2312 // but only contact the receive_algorithm once. | 2267 // but only contact the receive_algorithm once. |
2313 EXPECT_CALL(*receive_algorithm_, RecordIncomingPacket(_, _, _, _)); | 2268 EXPECT_CALL(*receive_algorithm_, RecordIncomingPacket(_, _, _, _)); |
2314 ProcessFecPacket(2, 1, true, !kEntropyFlag, NULL); | 2269 ProcessFecPacket(2, 1, true, !kEntropyFlag, NULL); |
2315 } | 2270 } |
2316 | 2271 |
2317 TEST_P(QuicConnectionTest, InitialTimeout) { | 2272 TEST_F(QuicConnectionTest, InitialTimeout) { |
2318 EXPECT_TRUE(connection_.connected()); | 2273 EXPECT_TRUE(connection_.connected()); |
2319 EXPECT_CALL(visitor_, OnConnectionClosed(QUIC_CONNECTION_TIMED_OUT, false)); | 2274 EXPECT_CALL(visitor_, OnConnectionClosed(QUIC_CONNECTION_TIMED_OUT, false)); |
2320 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)); | 2275 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)); |
2321 | 2276 |
2322 QuicTime default_timeout = clock_.ApproximateNow().Add( | 2277 QuicTime default_timeout = clock_.ApproximateNow().Add( |
2323 QuicTime::Delta::FromSeconds(kDefaultInitialTimeoutSecs)); | 2278 QuicTime::Delta::FromSeconds(kDefaultInitialTimeoutSecs)); |
2324 EXPECT_EQ(default_timeout, connection_.GetTimeoutAlarm()->deadline()); | 2279 EXPECT_EQ(default_timeout, connection_.GetTimeoutAlarm()->deadline()); |
2325 | 2280 |
2326 // Simulate the timeout alarm firing. | 2281 // Simulate the timeout alarm firing. |
2327 clock_.AdvanceTime( | 2282 clock_.AdvanceTime( |
2328 QuicTime::Delta::FromSeconds(kDefaultInitialTimeoutSecs)); | 2283 QuicTime::Delta::FromSeconds(kDefaultInitialTimeoutSecs)); |
2329 connection_.GetTimeoutAlarm()->Fire(); | 2284 connection_.GetTimeoutAlarm()->Fire(); |
2330 EXPECT_FALSE(connection_.GetTimeoutAlarm()->IsSet()); | 2285 EXPECT_FALSE(connection_.GetTimeoutAlarm()->IsSet()); |
2331 EXPECT_FALSE(connection_.connected()); | 2286 EXPECT_FALSE(connection_.connected()); |
2332 | 2287 |
2333 EXPECT_FALSE(connection_.GetAckAlarm()->IsSet()); | 2288 EXPECT_FALSE(connection_.GetAckAlarm()->IsSet()); |
2334 EXPECT_FALSE(connection_.GetResumeWritesAlarm()->IsSet()); | 2289 EXPECT_FALSE(connection_.GetResumeWritesAlarm()->IsSet()); |
2335 EXPECT_FALSE(connection_.GetRetransmissionAlarm()->IsSet()); | 2290 EXPECT_FALSE(connection_.GetRetransmissionAlarm()->IsSet()); |
2336 EXPECT_FALSE(connection_.GetSendAlarm()->IsSet()); | 2291 EXPECT_FALSE(connection_.GetSendAlarm()->IsSet()); |
2337 EXPECT_FALSE(connection_.GetTimeoutAlarm()->IsSet()); | 2292 EXPECT_FALSE(connection_.GetTimeoutAlarm()->IsSet()); |
2338 } | 2293 } |
2339 | 2294 |
2340 TEST_P(QuicConnectionTest, TimeoutAfterSend) { | 2295 TEST_F(QuicConnectionTest, TimeoutAfterSend) { |
2341 EXPECT_TRUE(connection_.connected()); | 2296 EXPECT_TRUE(connection_.connected()); |
2342 | 2297 |
2343 QuicTime default_timeout = clock_.ApproximateNow().Add( | 2298 QuicTime default_timeout = clock_.ApproximateNow().Add( |
2344 QuicTime::Delta::FromSeconds(kDefaultInitialTimeoutSecs)); | 2299 QuicTime::Delta::FromSeconds(kDefaultInitialTimeoutSecs)); |
2345 | 2300 |
2346 // When we send a packet, the timeout will change to 5000 + | 2301 // When we send a packet, the timeout will change to 5000 + |
2347 // kDefaultInitialTimeoutSecs. | 2302 // kDefaultInitialTimeoutSecs. |
2348 clock_.AdvanceTime(QuicTime::Delta::FromMilliseconds(5)); | 2303 clock_.AdvanceTime(QuicTime::Delta::FromMilliseconds(5)); |
2349 | 2304 |
2350 // Send an ack so we don't set the retransimission alarm. | 2305 // Send an ack so we don't set the retransimission alarm. |
(...skipping 16 matching lines...) Expand all Loading... |
2367 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)); | 2322 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)); |
2368 clock_.AdvanceTime(QuicTime::Delta::FromMilliseconds(5)); | 2323 clock_.AdvanceTime(QuicTime::Delta::FromMilliseconds(5)); |
2369 EXPECT_EQ(default_timeout.Add(QuicTime::Delta::FromMilliseconds(5)), | 2324 EXPECT_EQ(default_timeout.Add(QuicTime::Delta::FromMilliseconds(5)), |
2370 clock_.ApproximateNow()); | 2325 clock_.ApproximateNow()); |
2371 connection_.GetTimeoutAlarm()->Fire(); | 2326 connection_.GetTimeoutAlarm()->Fire(); |
2372 EXPECT_FALSE(connection_.GetTimeoutAlarm()->IsSet()); | 2327 EXPECT_FALSE(connection_.GetTimeoutAlarm()->IsSet()); |
2373 EXPECT_FALSE(connection_.connected()); | 2328 EXPECT_FALSE(connection_.connected()); |
2374 } | 2329 } |
2375 | 2330 |
2376 // TODO(ianswett): Add scheduler tests when should_retransmit is false. | 2331 // TODO(ianswett): Add scheduler tests when should_retransmit is false. |
2377 TEST_P(QuicConnectionTest, SendScheduler) { | 2332 TEST_F(QuicConnectionTest, SendScheduler) { |
2378 // Test that if we send a packet without delay, it is not queued. | 2333 // Test that if we send a packet without delay, it is not queued. |
2379 QuicPacket* packet = ConstructDataPacket(1, 0, !kEntropyFlag); | 2334 QuicPacket* packet = ConstructDataPacket(1, 0, !kEntropyFlag); |
2380 EXPECT_CALL(*send_algorithm_, | 2335 EXPECT_CALL(*send_algorithm_, |
2381 TimeUntilSend(_, NOT_RETRANSMISSION, _, _)).WillOnce( | 2336 TimeUntilSend(_, NOT_RETRANSMISSION, _, _)).WillOnce( |
2382 testing::Return(QuicTime::Delta::Zero())); | 2337 testing::Return(QuicTime::Delta::Zero())); |
2383 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)); | 2338 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)); |
2384 connection_.SendPacket( | 2339 connection_.SendPacket( |
2385 ENCRYPTION_NONE, 1, packet, kTestEntropyHash, HAS_RETRANSMITTABLE_DATA); | 2340 ENCRYPTION_NONE, 1, packet, kTestEntropyHash, HAS_RETRANSMITTABLE_DATA); |
2386 EXPECT_EQ(0u, connection_.NumQueuedPackets()); | 2341 EXPECT_EQ(0u, connection_.NumQueuedPackets()); |
2387 } | 2342 } |
2388 | 2343 |
2389 TEST_P(QuicConnectionTest, SendSchedulerDelay) { | 2344 TEST_F(QuicConnectionTest, SendSchedulerDelay) { |
2390 // Test that if we send a packet with a delay, it ends up queued. | 2345 // Test that if we send a packet with a delay, it ends up queued. |
2391 QuicPacket* packet = ConstructDataPacket(1, 0, !kEntropyFlag); | 2346 QuicPacket* packet = ConstructDataPacket(1, 0, !kEntropyFlag); |
2392 EXPECT_CALL(*send_algorithm_, | 2347 EXPECT_CALL(*send_algorithm_, |
2393 TimeUntilSend(_, NOT_RETRANSMISSION, _, _)).WillOnce( | 2348 TimeUntilSend(_, NOT_RETRANSMISSION, _, _)).WillOnce( |
2394 testing::Return(QuicTime::Delta::FromMicroseconds(1))); | 2349 testing::Return(QuicTime::Delta::FromMicroseconds(1))); |
2395 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, 1, _, _, _)).Times(0); | 2350 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, 1, _, _, _)).Times(0); |
2396 connection_.SendPacket( | 2351 connection_.SendPacket( |
2397 ENCRYPTION_NONE, 1, packet, kTestEntropyHash, HAS_RETRANSMITTABLE_DATA); | 2352 ENCRYPTION_NONE, 1, packet, kTestEntropyHash, HAS_RETRANSMITTABLE_DATA); |
2398 EXPECT_EQ(1u, connection_.NumQueuedPackets()); | 2353 EXPECT_EQ(1u, connection_.NumQueuedPackets()); |
2399 } | 2354 } |
2400 | 2355 |
2401 TEST_P(QuicConnectionTest, SendSchedulerForce) { | 2356 TEST_F(QuicConnectionTest, SendSchedulerForce) { |
2402 // Test that if we force send a packet, it is not queued. | 2357 // Test that if we force send a packet, it is not queued. |
2403 QuicPacket* packet = ConstructDataPacket(1, 0, !kEntropyFlag); | 2358 QuicPacket* packet = ConstructDataPacket(1, 0, !kEntropyFlag); |
2404 EXPECT_CALL(*send_algorithm_, | 2359 EXPECT_CALL(*send_algorithm_, |
2405 TimeUntilSend(_, NACK_RETRANSMISSION, _, _)).Times(0); | 2360 TimeUntilSend(_, NACK_RETRANSMISSION, _, _)).Times(0); |
2406 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)); | 2361 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)); |
2407 connection_.SendPacket( | 2362 connection_.SendPacket( |
2408 ENCRYPTION_NONE, 1, packet, kTestEntropyHash, HAS_RETRANSMITTABLE_DATA); | 2363 ENCRYPTION_NONE, 1, packet, kTestEntropyHash, HAS_RETRANSMITTABLE_DATA); |
2409 // XXX: fixme. was: connection_.SendPacket(1, packet, kForce); | 2364 // XXX: fixme. was: connection_.SendPacket(1, packet, kForce); |
2410 EXPECT_EQ(0u, connection_.NumQueuedPackets()); | 2365 EXPECT_EQ(0u, connection_.NumQueuedPackets()); |
2411 } | 2366 } |
2412 | 2367 |
2413 TEST_P(QuicConnectionTest, SendSchedulerEAGAIN) { | 2368 TEST_F(QuicConnectionTest, SendSchedulerEAGAIN) { |
2414 QuicPacket* packet = ConstructDataPacket(1, 0, !kEntropyFlag); | 2369 QuicPacket* packet = ConstructDataPacket(1, 0, !kEntropyFlag); |
2415 writer_->set_blocked(true); | 2370 writer_->set_blocked(true); |
2416 EXPECT_CALL(*send_algorithm_, | 2371 EXPECT_CALL(*send_algorithm_, |
2417 TimeUntilSend(_, NOT_RETRANSMISSION, _, _)).WillOnce( | 2372 TimeUntilSend(_, NOT_RETRANSMISSION, _, _)).WillOnce( |
2418 testing::Return(QuicTime::Delta::Zero())); | 2373 testing::Return(QuicTime::Delta::Zero())); |
2419 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, 1, _, _, _)).Times(0); | 2374 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, 1, _, _, _)).Times(0); |
2420 connection_.SendPacket( | 2375 connection_.SendPacket( |
2421 ENCRYPTION_NONE, 1, packet, kTestEntropyHash, HAS_RETRANSMITTABLE_DATA); | 2376 ENCRYPTION_NONE, 1, packet, kTestEntropyHash, HAS_RETRANSMITTABLE_DATA); |
2422 EXPECT_EQ(1u, connection_.NumQueuedPackets()); | 2377 EXPECT_EQ(1u, connection_.NumQueuedPackets()); |
2423 } | 2378 } |
2424 | 2379 |
2425 TEST_P(QuicConnectionTest, SendSchedulerDelayThenSend) { | 2380 TEST_F(QuicConnectionTest, SendSchedulerDelayThenSend) { |
2426 // Test that if we send a packet with a delay, it ends up queued. | 2381 // Test that if we send a packet with a delay, it ends up queued. |
2427 QuicPacket* packet = ConstructDataPacket(1, 0, !kEntropyFlag); | 2382 QuicPacket* packet = ConstructDataPacket(1, 0, !kEntropyFlag); |
2428 EXPECT_CALL(*send_algorithm_, | 2383 EXPECT_CALL(*send_algorithm_, |
2429 TimeUntilSend(_, NOT_RETRANSMISSION, _, _)).WillOnce( | 2384 TimeUntilSend(_, NOT_RETRANSMISSION, _, _)).WillOnce( |
2430 testing::Return(QuicTime::Delta::FromMicroseconds(1))); | 2385 testing::Return(QuicTime::Delta::FromMicroseconds(1))); |
2431 connection_.SendPacket( | 2386 connection_.SendPacket( |
2432 ENCRYPTION_NONE, 1, packet, kTestEntropyHash, HAS_RETRANSMITTABLE_DATA); | 2387 ENCRYPTION_NONE, 1, packet, kTestEntropyHash, HAS_RETRANSMITTABLE_DATA); |
2433 EXPECT_EQ(1u, connection_.NumQueuedPackets()); | 2388 EXPECT_EQ(1u, connection_.NumQueuedPackets()); |
2434 | 2389 |
2435 // Advance the clock to fire the alarm, and configure the scheduler | 2390 // Advance the clock to fire the alarm, and configure the scheduler |
2436 // to permit the packet to be sent. | 2391 // to permit the packet to be sent. |
2437 EXPECT_CALL(*send_algorithm_, | 2392 EXPECT_CALL(*send_algorithm_, |
2438 TimeUntilSend(_, NOT_RETRANSMISSION, _, _)).WillRepeatedly( | 2393 TimeUntilSend(_, NOT_RETRANSMISSION, _, _)).WillRepeatedly( |
2439 testing::Return(QuicTime::Delta::Zero())); | 2394 testing::Return(QuicTime::Delta::Zero())); |
2440 clock_.AdvanceTime(QuicTime::Delta::FromMicroseconds(1)); | 2395 clock_.AdvanceTime(QuicTime::Delta::FromMicroseconds(1)); |
2441 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)); | 2396 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)); |
2442 connection_.GetSendAlarm()->Fire(); | 2397 connection_.GetSendAlarm()->Fire(); |
2443 EXPECT_EQ(0u, connection_.NumQueuedPackets()); | 2398 EXPECT_EQ(0u, connection_.NumQueuedPackets()); |
2444 } | 2399 } |
2445 | 2400 |
2446 TEST_P(QuicConnectionTest, SendSchedulerDelayThenRetransmit) { | 2401 TEST_F(QuicConnectionTest, SendSchedulerDelayThenRetransmit) { |
2447 EXPECT_CALL(*send_algorithm_, TimeUntilSend(_, NOT_RETRANSMISSION, _, _)) | 2402 EXPECT_CALL(*send_algorithm_, TimeUntilSend(_, NOT_RETRANSMISSION, _, _)) |
2448 .WillRepeatedly(testing::Return(QuicTime::Delta::Zero())); | 2403 .WillRepeatedly(testing::Return(QuicTime::Delta::Zero())); |
2449 EXPECT_CALL(*send_algorithm_, OnPacketAbandoned(1, _)).Times(1); | 2404 EXPECT_CALL(*send_algorithm_, OnPacketAbandoned(1, _)).Times(1); |
2450 EXPECT_CALL(*send_algorithm_, | 2405 EXPECT_CALL(*send_algorithm_, |
2451 OnPacketSent(_, 1, _, NOT_RETRANSMISSION, _)); | 2406 OnPacketSent(_, 1, _, NOT_RETRANSMISSION, _)); |
2452 connection_.SendStreamDataWithString(1, "foo", 0, !kFin, NULL); | 2407 connection_.SendStreamDataWithString(1, "foo", 0, !kFin, NULL); |
2453 EXPECT_EQ(0u, connection_.NumQueuedPackets()); | 2408 EXPECT_EQ(0u, connection_.NumQueuedPackets()); |
2454 // Advance the time for retransmission of lost packet. | 2409 // Advance the time for retransmission of lost packet. |
2455 clock_.AdvanceTime(QuicTime::Delta::FromMilliseconds(501)); | 2410 clock_.AdvanceTime(QuicTime::Delta::FromMilliseconds(501)); |
2456 // Test that if we send a retransmit with a delay, it ends up queued in the | 2411 // Test that if we send a retransmit with a delay, it ends up queued in the |
(...skipping 11 matching lines...) Expand all Loading... |
2468 WillRepeatedly(testing::Return(QuicTime::Delta::Zero())); | 2423 WillRepeatedly(testing::Return(QuicTime::Delta::Zero())); |
2469 | 2424 |
2470 // Ensure the scheduler is notified this is a retransmit. | 2425 // Ensure the scheduler is notified this is a retransmit. |
2471 EXPECT_CALL(*send_algorithm_, | 2426 EXPECT_CALL(*send_algorithm_, |
2472 OnPacketSent(_, _, _, RTO_RETRANSMISSION, _)); | 2427 OnPacketSent(_, _, _, RTO_RETRANSMISSION, _)); |
2473 clock_.AdvanceTime(QuicTime::Delta::FromMicroseconds(1)); | 2428 clock_.AdvanceTime(QuicTime::Delta::FromMicroseconds(1)); |
2474 connection_.GetSendAlarm()->Fire(); | 2429 connection_.GetSendAlarm()->Fire(); |
2475 EXPECT_EQ(0u, connection_.NumQueuedPackets()); | 2430 EXPECT_EQ(0u, connection_.NumQueuedPackets()); |
2476 } | 2431 } |
2477 | 2432 |
2478 TEST_P(QuicConnectionTest, SendSchedulerDelayAndQueue) { | 2433 TEST_F(QuicConnectionTest, SendSchedulerDelayAndQueue) { |
2479 QuicPacket* packet = ConstructDataPacket(1, 0, !kEntropyFlag); | 2434 QuicPacket* packet = ConstructDataPacket(1, 0, !kEntropyFlag); |
2480 EXPECT_CALL(*send_algorithm_, | 2435 EXPECT_CALL(*send_algorithm_, |
2481 TimeUntilSend(_, NOT_RETRANSMISSION, _, _)).WillOnce( | 2436 TimeUntilSend(_, NOT_RETRANSMISSION, _, _)).WillOnce( |
2482 testing::Return(QuicTime::Delta::FromMicroseconds(1))); | 2437 testing::Return(QuicTime::Delta::FromMicroseconds(1))); |
2483 connection_.SendPacket( | 2438 connection_.SendPacket( |
2484 ENCRYPTION_NONE, 1, packet, kTestEntropyHash, HAS_RETRANSMITTABLE_DATA); | 2439 ENCRYPTION_NONE, 1, packet, kTestEntropyHash, HAS_RETRANSMITTABLE_DATA); |
2485 EXPECT_EQ(1u, connection_.NumQueuedPackets()); | 2440 EXPECT_EQ(1u, connection_.NumQueuedPackets()); |
2486 | 2441 |
2487 // Attempt to send another packet and make sure that it gets queued. | 2442 // Attempt to send another packet and make sure that it gets queued. |
2488 packet = ConstructDataPacket(2, 0, !kEntropyFlag); | 2443 packet = ConstructDataPacket(2, 0, !kEntropyFlag); |
2489 connection_.SendPacket( | 2444 connection_.SendPacket( |
2490 ENCRYPTION_NONE, 2, packet, kTestEntropyHash, HAS_RETRANSMITTABLE_DATA); | 2445 ENCRYPTION_NONE, 2, packet, kTestEntropyHash, HAS_RETRANSMITTABLE_DATA); |
2491 EXPECT_EQ(2u, connection_.NumQueuedPackets()); | 2446 EXPECT_EQ(2u, connection_.NumQueuedPackets()); |
2492 } | 2447 } |
2493 | 2448 |
2494 TEST_P(QuicConnectionTest, SendSchedulerDelayThenAckAndSend) { | 2449 TEST_F(QuicConnectionTest, SendSchedulerDelayThenAckAndSend) { |
2495 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | 2450 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); |
2496 QuicPacket* packet = ConstructDataPacket(1, 0, !kEntropyFlag); | 2451 QuicPacket* packet = ConstructDataPacket(1, 0, !kEntropyFlag); |
2497 EXPECT_CALL(*send_algorithm_, | 2452 EXPECT_CALL(*send_algorithm_, |
2498 TimeUntilSend(_, NOT_RETRANSMISSION, _, _)).WillOnce( | 2453 TimeUntilSend(_, NOT_RETRANSMISSION, _, _)).WillOnce( |
2499 testing::Return(QuicTime::Delta::FromMicroseconds(10))); | 2454 testing::Return(QuicTime::Delta::FromMicroseconds(10))); |
2500 connection_.SendPacket( | 2455 connection_.SendPacket( |
2501 ENCRYPTION_NONE, 1, packet, kTestEntropyHash, HAS_RETRANSMITTABLE_DATA); | 2456 ENCRYPTION_NONE, 1, packet, kTestEntropyHash, HAS_RETRANSMITTABLE_DATA); |
2502 EXPECT_EQ(1u, connection_.NumQueuedPackets()); | 2457 EXPECT_EQ(1u, connection_.NumQueuedPackets()); |
2503 | 2458 |
2504 // Now send non-retransmitting information, that we're not going to | 2459 // Now send non-retransmitting information, that we're not going to |
2505 // retransmit 3. The far end should stop waiting for it. | 2460 // retransmit 3. The far end should stop waiting for it. |
2506 QuicAckFrame frame(0, QuicTime::Zero(), 1); | 2461 QuicAckFrame frame(0, QuicTime::Zero(), 1); |
2507 EXPECT_CALL(*send_algorithm_, | 2462 EXPECT_CALL(*send_algorithm_, |
2508 TimeUntilSend(_, NOT_RETRANSMISSION, _, _)).WillRepeatedly( | 2463 TimeUntilSend(_, NOT_RETRANSMISSION, _, _)).WillRepeatedly( |
2509 testing::Return(QuicTime::Delta::Zero())); | 2464 testing::Return(QuicTime::Delta::Zero())); |
2510 EXPECT_CALL(*send_algorithm_, | 2465 EXPECT_CALL(*send_algorithm_, |
2511 OnPacketSent(_, _, _, _, _)); | 2466 OnPacketSent(_, _, _, _, _)); |
2512 ProcessAckPacket(&frame); | 2467 ProcessAckPacket(&frame); |
2513 | 2468 |
2514 EXPECT_EQ(0u, connection_.NumQueuedPackets()); | 2469 EXPECT_EQ(0u, connection_.NumQueuedPackets()); |
2515 // Ensure alarm is not set | 2470 // Ensure alarm is not set |
2516 EXPECT_FALSE(connection_.GetSendAlarm()->IsSet()); | 2471 EXPECT_FALSE(connection_.GetSendAlarm()->IsSet()); |
2517 } | 2472 } |
2518 | 2473 |
2519 TEST_P(QuicConnectionTest, SendSchedulerDelayThenAckAndHold) { | 2474 TEST_F(QuicConnectionTest, SendSchedulerDelayThenAckAndHold) { |
2520 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | 2475 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); |
2521 QuicPacket* packet = ConstructDataPacket(1, 0, !kEntropyFlag); | 2476 QuicPacket* packet = ConstructDataPacket(1, 0, !kEntropyFlag); |
2522 EXPECT_CALL(*send_algorithm_, | 2477 EXPECT_CALL(*send_algorithm_, |
2523 TimeUntilSend(_, NOT_RETRANSMISSION, _, _)).WillOnce( | 2478 TimeUntilSend(_, NOT_RETRANSMISSION, _, _)).WillOnce( |
2524 testing::Return(QuicTime::Delta::FromMicroseconds(10))); | 2479 testing::Return(QuicTime::Delta::FromMicroseconds(10))); |
2525 connection_.SendPacket( | 2480 connection_.SendPacket( |
2526 ENCRYPTION_NONE, 1, packet, kTestEntropyHash, HAS_RETRANSMITTABLE_DATA); | 2481 ENCRYPTION_NONE, 1, packet, kTestEntropyHash, HAS_RETRANSMITTABLE_DATA); |
2527 EXPECT_EQ(1u, connection_.NumQueuedPackets()); | 2482 EXPECT_EQ(1u, connection_.NumQueuedPackets()); |
2528 | 2483 |
2529 // Now send non-retransmitting information, that we're not going to | 2484 // Now send non-retransmitting information, that we're not going to |
2530 // retransmit 3. The far end should stop waiting for it. | 2485 // retransmit 3. The far end should stop waiting for it. |
2531 QuicAckFrame frame(0, QuicTime::Zero(), 1); | 2486 QuicAckFrame frame(0, QuicTime::Zero(), 1); |
2532 EXPECT_CALL(*send_algorithm_, | 2487 EXPECT_CALL(*send_algorithm_, |
2533 TimeUntilSend(_, NOT_RETRANSMISSION, _, _)).WillOnce( | 2488 TimeUntilSend(_, NOT_RETRANSMISSION, _, _)).WillOnce( |
2534 testing::Return(QuicTime::Delta::FromMicroseconds(1))); | 2489 testing::Return(QuicTime::Delta::FromMicroseconds(1))); |
2535 ProcessAckPacket(&frame); | 2490 ProcessAckPacket(&frame); |
2536 | 2491 |
2537 EXPECT_EQ(1u, connection_.NumQueuedPackets()); | 2492 EXPECT_EQ(1u, connection_.NumQueuedPackets()); |
2538 } | 2493 } |
2539 | 2494 |
2540 TEST_P(QuicConnectionTest, SendSchedulerDelayThenOnCanWrite) { | 2495 TEST_F(QuicConnectionTest, SendSchedulerDelayThenOnCanWrite) { |
2541 QuicPacket* packet = ConstructDataPacket(1, 0, !kEntropyFlag); | 2496 QuicPacket* packet = ConstructDataPacket(1, 0, !kEntropyFlag); |
2542 EXPECT_CALL(*send_algorithm_, | 2497 EXPECT_CALL(*send_algorithm_, |
2543 TimeUntilSend(_, NOT_RETRANSMISSION, _, _)).WillOnce( | 2498 TimeUntilSend(_, NOT_RETRANSMISSION, _, _)).WillOnce( |
2544 testing::Return(QuicTime::Delta::FromMicroseconds(10))); | 2499 testing::Return(QuicTime::Delta::FromMicroseconds(10))); |
2545 connection_.SendPacket( | 2500 connection_.SendPacket( |
2546 ENCRYPTION_NONE, 1, packet, kTestEntropyHash, HAS_RETRANSMITTABLE_DATA); | 2501 ENCRYPTION_NONE, 1, packet, kTestEntropyHash, HAS_RETRANSMITTABLE_DATA); |
2547 EXPECT_EQ(1u, connection_.NumQueuedPackets()); | 2502 EXPECT_EQ(1u, connection_.NumQueuedPackets()); |
2548 | 2503 |
2549 // OnCanWrite should not send the packet (because of the delay) | 2504 // OnCanWrite should not send the packet (because of the delay) |
2550 // but should still return true. | 2505 // but should still return true. |
2551 EXPECT_TRUE(connection_.OnCanWrite()); | 2506 EXPECT_TRUE(connection_.OnCanWrite()); |
2552 EXPECT_EQ(1u, connection_.NumQueuedPackets()); | 2507 EXPECT_EQ(1u, connection_.NumQueuedPackets()); |
2553 } | 2508 } |
2554 | 2509 |
2555 TEST_P(QuicConnectionTest, TestQueueLimitsOnSendStreamData) { | 2510 TEST_F(QuicConnectionTest, TestQueueLimitsOnSendStreamData) { |
2556 // All packets carry version info till version is negotiated. | 2511 // All packets carry version info till version is negotiated. |
2557 size_t payload_length; | 2512 size_t payload_length; |
2558 connection_.options()->max_packet_length = | 2513 connection_.options()->max_packet_length = |
2559 GetPacketLengthForOneStream( | 2514 GetPacketLengthForOneStream( |
2560 connection_.version(), kIncludeVersion, PACKET_1BYTE_SEQUENCE_NUMBER, | 2515 connection_.version(), kIncludeVersion, PACKET_1BYTE_SEQUENCE_NUMBER, |
2561 NOT_IN_FEC_GROUP, &payload_length); | 2516 NOT_IN_FEC_GROUP, &payload_length); |
2562 | 2517 |
2563 // Queue the first packet. | 2518 // Queue the first packet. |
2564 EXPECT_CALL(*send_algorithm_, | 2519 EXPECT_CALL(*send_algorithm_, |
2565 TimeUntilSend(_, NOT_RETRANSMISSION, _, _)).WillOnce( | 2520 TimeUntilSend(_, NOT_RETRANSMISSION, _, _)).WillOnce( |
2566 testing::Return(QuicTime::Delta::FromMicroseconds(10))); | 2521 testing::Return(QuicTime::Delta::FromMicroseconds(10))); |
2567 const string payload(payload_length, 'a'); | 2522 const string payload(payload_length, 'a'); |
2568 EXPECT_EQ(0u, | 2523 EXPECT_EQ(0u, |
2569 connection_.SendStreamDataWithString(3, payload, 0, | 2524 connection_.SendStreamDataWithString(3, payload, 0, |
2570 !kFin, NULL).bytes_consumed); | 2525 !kFin, NULL).bytes_consumed); |
2571 EXPECT_EQ(0u, connection_.NumQueuedPackets()); | 2526 EXPECT_EQ(0u, connection_.NumQueuedPackets()); |
2572 } | 2527 } |
2573 | 2528 |
2574 TEST_P(QuicConnectionTest, LoopThroughSendingPackets) { | 2529 TEST_F(QuicConnectionTest, LoopThroughSendingPackets) { |
2575 // All packets carry version info till version is negotiated. | 2530 // All packets carry version info till version is negotiated. |
2576 size_t payload_length; | 2531 size_t payload_length; |
2577 connection_.options()->max_packet_length = | 2532 connection_.options()->max_packet_length = |
2578 GetPacketLengthForOneStream( | 2533 GetPacketLengthForOneStream( |
2579 connection_.version(), kIncludeVersion, PACKET_1BYTE_SEQUENCE_NUMBER, | 2534 connection_.version(), kIncludeVersion, PACKET_1BYTE_SEQUENCE_NUMBER, |
2580 NOT_IN_FEC_GROUP, &payload_length); | 2535 NOT_IN_FEC_GROUP, &payload_length); |
2581 | 2536 |
2582 // Queue the first packet. | 2537 // Queue the first packet. |
2583 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)).Times(7); | 2538 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)).Times(7); |
2584 // The first stream frame will consume 2 fewer bytes than the other six. | 2539 // The first stream frame will consume 2 fewer bytes than the other six. |
2585 const string payload(payload_length * 7 - 12, 'a'); | 2540 const string payload(payload_length * 7 - 12, 'a'); |
2586 EXPECT_EQ(payload.size(), | 2541 EXPECT_EQ(payload.size(), |
2587 connection_.SendStreamDataWithString(1, payload, 0, | 2542 connection_.SendStreamDataWithString(1, payload, 0, |
2588 !kFin, NULL).bytes_consumed); | 2543 !kFin, NULL).bytes_consumed); |
2589 } | 2544 } |
2590 | 2545 |
2591 TEST_P(QuicConnectionTest, SendDelayedAckOnTimer) { | 2546 TEST_F(QuicConnectionTest, SendDelayedAckOnTimer) { |
2592 QuicTime ack_time = clock_.ApproximateNow().Add(DefaultDelayedAckTime()); | 2547 QuicTime ack_time = clock_.ApproximateNow().Add(DefaultDelayedAckTime()); |
2593 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | 2548 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); |
2594 EXPECT_FALSE(connection_.GetAckAlarm()->IsSet()); | 2549 EXPECT_FALSE(connection_.GetAckAlarm()->IsSet()); |
2595 ProcessPacket(1); | 2550 ProcessPacket(1); |
2596 // Check if delayed ack timer is running for the expected interval. | 2551 // Check if delayed ack timer is running for the expected interval. |
2597 EXPECT_TRUE(connection_.GetAckAlarm()->IsSet()); | 2552 EXPECT_TRUE(connection_.GetAckAlarm()->IsSet()); |
2598 EXPECT_EQ(ack_time, connection_.GetAckAlarm()->deadline()); | 2553 EXPECT_EQ(ack_time, connection_.GetAckAlarm()->deadline()); |
2599 // Simulate delayed ack alarm firing. | 2554 // Simulate delayed ack alarm firing. |
2600 connection_.GetAckAlarm()->Fire(); | 2555 connection_.GetAckAlarm()->Fire(); |
2601 // Check that ack is sent and that delayed ack alarm is reset. | 2556 // Check that ack is sent and that delayed ack alarm is reset. |
2602 EXPECT_EQ(1u, writer_->frame_count()); | 2557 EXPECT_EQ(1u, writer_->frame_count()); |
2603 EXPECT_TRUE(writer_->ack()); | 2558 EXPECT_TRUE(writer_->ack()); |
2604 EXPECT_FALSE(connection_.GetAckAlarm()->IsSet()); | 2559 EXPECT_FALSE(connection_.GetAckAlarm()->IsSet()); |
2605 } | 2560 } |
2606 | 2561 |
2607 TEST_P(QuicConnectionTest, SendDelayedAckOnSecondPacket) { | 2562 TEST_F(QuicConnectionTest, SendDelayedAckOnSecondPacket) { |
2608 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | 2563 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); |
2609 ProcessPacket(1); | 2564 ProcessPacket(1); |
2610 ProcessPacket(2); | 2565 ProcessPacket(2); |
2611 // Check that ack is sent and that delayed ack alarm is reset. | 2566 // Check that ack is sent and that delayed ack alarm is reset. |
2612 EXPECT_EQ(1u, writer_->frame_count()); | 2567 EXPECT_EQ(1u, writer_->frame_count()); |
2613 EXPECT_TRUE(writer_->ack()); | 2568 EXPECT_TRUE(writer_->ack()); |
2614 EXPECT_FALSE(connection_.GetAckAlarm()->IsSet()); | 2569 EXPECT_FALSE(connection_.GetAckAlarm()->IsSet()); |
2615 } | 2570 } |
2616 | 2571 |
2617 TEST_P(QuicConnectionTest, SendDelayedAckOnOutgoingPacket) { | 2572 TEST_F(QuicConnectionTest, SendDelayedAckOnOutgoingPacket) { |
2618 if (!FLAGS_bundle_ack_with_outgoing_packet) { | |
2619 // This test specifically tests ack bundling behavior. | |
2620 return; | |
2621 } | |
2622 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | 2573 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); |
2623 ProcessPacket(1); | 2574 ProcessPacket(1); |
2624 connection_.SendStreamDataWithString(kStreamId3, "foo", 0, !kFin, NULL); | 2575 connection_.SendStreamDataWithString(kStreamId3, "foo", 0, !kFin, NULL); |
2625 // Check that ack is bundled with outgoing data and that delayed ack | 2576 // Check that ack is bundled with outgoing data and that delayed ack |
2626 // alarm is reset. | 2577 // alarm is reset. |
2627 EXPECT_EQ(2u, writer_->frame_count()); | 2578 EXPECT_EQ(2u, writer_->frame_count()); |
2628 EXPECT_TRUE(writer_->ack()); | 2579 EXPECT_TRUE(writer_->ack()); |
2629 EXPECT_FALSE(connection_.GetAckAlarm()->IsSet()); | 2580 EXPECT_FALSE(connection_.GetAckAlarm()->IsSet()); |
2630 } | 2581 } |
2631 | 2582 |
2632 TEST_P(QuicConnectionTest, DontSendDelayedAckOnOutgoingCryptoPacket) { | 2583 TEST_F(QuicConnectionTest, DontSendDelayedAckOnOutgoingCryptoPacket) { |
2633 if (!FLAGS_bundle_ack_with_outgoing_packet) { | |
2634 // This test specifically tests ack bundling behavior. | |
2635 return; | |
2636 } | |
2637 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | 2584 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); |
2638 ProcessPacket(1); | 2585 ProcessPacket(1); |
2639 connection_.SendStreamDataWithString(kCryptoStreamId, "foo", 0, !kFin, NULL); | 2586 connection_.SendStreamDataWithString(kCryptoStreamId, "foo", 0, !kFin, NULL); |
2640 // Check that ack is not bundled with outgoing data. | 2587 // Check that ack is not bundled with outgoing data. |
2641 EXPECT_EQ(1u, writer_->frame_count()); | 2588 EXPECT_EQ(1u, writer_->frame_count()); |
2642 EXPECT_FALSE(writer_->ack()); | 2589 EXPECT_FALSE(writer_->ack()); |
2643 EXPECT_TRUE(connection_.GetAckAlarm()->IsSet()); | 2590 EXPECT_TRUE(connection_.GetAckAlarm()->IsSet()); |
2644 } | 2591 } |
2645 | 2592 |
2646 TEST_P(QuicConnectionTest, NoAckForClose) { | 2593 TEST_F(QuicConnectionTest, NoAckForClose) { |
2647 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | 2594 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); |
2648 ProcessPacket(1); | 2595 ProcessPacket(1); |
2649 EXPECT_CALL(*send_algorithm_, OnIncomingAck(_, _, _)).Times(0); | 2596 EXPECT_CALL(*send_algorithm_, OnIncomingAck(_, _, _)).Times(0); |
2650 EXPECT_CALL(visitor_, OnConnectionClosed(QUIC_PEER_GOING_AWAY, true)); | 2597 EXPECT_CALL(visitor_, OnConnectionClosed(QUIC_PEER_GOING_AWAY, true)); |
2651 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)).Times(0); | 2598 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)).Times(0); |
2652 ProcessClosePacket(2, 0); | 2599 ProcessClosePacket(2, 0); |
2653 } | 2600 } |
2654 | 2601 |
2655 TEST_P(QuicConnectionTest, SendWhenDisconnected) { | 2602 TEST_F(QuicConnectionTest, SendWhenDisconnected) { |
2656 EXPECT_TRUE(connection_.connected()); | 2603 EXPECT_TRUE(connection_.connected()); |
2657 EXPECT_CALL(visitor_, OnConnectionClosed(QUIC_PEER_GOING_AWAY, false)); | 2604 EXPECT_CALL(visitor_, OnConnectionClosed(QUIC_PEER_GOING_AWAY, false)); |
2658 connection_.CloseConnection(QUIC_PEER_GOING_AWAY, false); | 2605 connection_.CloseConnection(QUIC_PEER_GOING_AWAY, false); |
2659 EXPECT_FALSE(connection_.connected()); | 2606 EXPECT_FALSE(connection_.connected()); |
2660 QuicPacket* packet = ConstructDataPacket(1, 0, !kEntropyFlag); | 2607 QuicPacket* packet = ConstructDataPacket(1, 0, !kEntropyFlag); |
2661 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, 1, _, _, _)).Times(0); | 2608 EXPECT_CALL(*send_algorithm_, OnPacketSent(_, 1, _, _, _)).Times(0); |
2662 connection_.SendPacket( | 2609 connection_.SendPacket( |
2663 ENCRYPTION_NONE, 1, packet, kTestEntropyHash, HAS_RETRANSMITTABLE_DATA); | 2610 ENCRYPTION_NONE, 1, packet, kTestEntropyHash, HAS_RETRANSMITTABLE_DATA); |
2664 } | 2611 } |
2665 | 2612 |
2666 TEST_P(QuicConnectionTest, PublicReset) { | 2613 TEST_F(QuicConnectionTest, PublicReset) { |
2667 QuicPublicResetPacket header; | 2614 QuicPublicResetPacket header; |
2668 header.public_header.guid = guid_; | 2615 header.public_header.guid = guid_; |
2669 header.public_header.reset_flag = true; | 2616 header.public_header.reset_flag = true; |
2670 header.public_header.version_flag = false; | 2617 header.public_header.version_flag = false; |
2671 header.rejected_sequence_number = 10101; | 2618 header.rejected_sequence_number = 10101; |
2672 scoped_ptr<QuicEncryptedPacket> packet( | 2619 scoped_ptr<QuicEncryptedPacket> packet( |
2673 framer_.BuildPublicResetPacket(header)); | 2620 framer_.BuildPublicResetPacket(header)); |
2674 EXPECT_CALL(visitor_, OnConnectionClosed(QUIC_PUBLIC_RESET, true)); | 2621 EXPECT_CALL(visitor_, OnConnectionClosed(QUIC_PUBLIC_RESET, true)); |
2675 connection_.ProcessUdpPacket(IPEndPoint(), IPEndPoint(), *packet); | 2622 connection_.ProcessUdpPacket(IPEndPoint(), IPEndPoint(), *packet); |
2676 } | 2623 } |
2677 | 2624 |
2678 TEST_P(QuicConnectionTest, GoAway) { | 2625 TEST_F(QuicConnectionTest, GoAway) { |
2679 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | 2626 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); |
2680 | 2627 |
2681 QuicGoAwayFrame goaway; | 2628 QuicGoAwayFrame goaway; |
2682 goaway.last_good_stream_id = 1; | 2629 goaway.last_good_stream_id = 1; |
2683 goaway.error_code = QUIC_PEER_GOING_AWAY; | 2630 goaway.error_code = QUIC_PEER_GOING_AWAY; |
2684 goaway.reason_phrase = "Going away."; | 2631 goaway.reason_phrase = "Going away."; |
2685 EXPECT_CALL(visitor_, OnGoAway(_)); | 2632 EXPECT_CALL(visitor_, OnGoAway(_)); |
2686 ProcessGoAwayPacket(&goaway); | 2633 ProcessGoAwayPacket(&goaway); |
2687 } | 2634 } |
2688 | 2635 |
2689 TEST_P(QuicConnectionTest, MissingPacketsBeforeLeastUnacked) { | 2636 TEST_F(QuicConnectionTest, InvalidPacket) { |
| 2637 EXPECT_CALL(visitor_, |
| 2638 OnConnectionClosed(QUIC_INVALID_PACKET_HEADER, false)); |
| 2639 QuicEncryptedPacket encrypted(NULL, 0); |
| 2640 connection_.ProcessUdpPacket(IPEndPoint(), IPEndPoint(), encrypted); |
| 2641 // The connection close packet should have error details. |
| 2642 ASSERT_TRUE(last_close() != NULL); |
| 2643 EXPECT_EQ("Unable to read public flags.", last_close()->error_details); |
| 2644 } |
| 2645 |
| 2646 TEST_F(QuicConnectionTest, MissingPacketsBeforeLeastUnacked) { |
2690 QuicAckFrame ack(0, QuicTime::Zero(), 4); | 2647 QuicAckFrame ack(0, QuicTime::Zero(), 4); |
2691 // Set the sequence number of the ack packet to be least unacked (4). | 2648 // Set the sequence number of the ack packet to be least unacked (4). |
2692 creator_.set_sequence_number(3); | 2649 creator_.set_sequence_number(3); |
2693 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | 2650 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); |
2694 ProcessAckPacket(&ack); | 2651 ProcessAckPacket(&ack); |
2695 EXPECT_TRUE(outgoing_ack()->received_info.missing_packets.empty()); | 2652 EXPECT_TRUE(outgoing_ack()->received_info.missing_packets.empty()); |
2696 } | 2653 } |
2697 | 2654 |
2698 TEST_P(QuicConnectionTest, ReceivedEntropyHashCalculation) { | 2655 TEST_F(QuicConnectionTest, ReceivedEntropyHashCalculation) { |
2699 EXPECT_CALL(visitor_, OnStreamFrames(_)).WillRepeatedly(Return(true)); | 2656 EXPECT_CALL(visitor_, OnStreamFrames(_)).WillRepeatedly(Return(true)); |
2700 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | 2657 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); |
2701 ProcessDataPacket(1, 1, kEntropyFlag); | 2658 ProcessDataPacket(1, 1, kEntropyFlag); |
2702 ProcessDataPacket(4, 1, kEntropyFlag); | 2659 ProcessDataPacket(4, 1, kEntropyFlag); |
2703 ProcessDataPacket(3, 1, !kEntropyFlag); | 2660 ProcessDataPacket(3, 1, !kEntropyFlag); |
2704 ProcessDataPacket(7, 1, kEntropyFlag); | 2661 ProcessDataPacket(7, 1, kEntropyFlag); |
2705 EXPECT_EQ(146u, outgoing_ack()->received_info.entropy_hash); | 2662 EXPECT_EQ(146u, outgoing_ack()->received_info.entropy_hash); |
2706 } | 2663 } |
2707 | 2664 |
2708 TEST_P(QuicConnectionTest, UpdateEntropyForReceivedPackets) { | 2665 TEST_F(QuicConnectionTest, UpdateEntropyForReceivedPackets) { |
2709 EXPECT_CALL(visitor_, OnStreamFrames(_)).WillRepeatedly(Return(true)); | 2666 EXPECT_CALL(visitor_, OnStreamFrames(_)).WillRepeatedly(Return(true)); |
2710 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | 2667 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); |
2711 ProcessDataPacket(1, 1, kEntropyFlag); | 2668 ProcessDataPacket(1, 1, kEntropyFlag); |
2712 ProcessDataPacket(5, 1, kEntropyFlag); | 2669 ProcessDataPacket(5, 1, kEntropyFlag); |
2713 ProcessDataPacket(4, 1, !kEntropyFlag); | 2670 ProcessDataPacket(4, 1, !kEntropyFlag); |
2714 EXPECT_EQ(34u, outgoing_ack()->received_info.entropy_hash); | 2671 EXPECT_EQ(34u, outgoing_ack()->received_info.entropy_hash); |
2715 // Make 4th packet my least unacked, and update entropy for 2, 3 packets. | 2672 // Make 4th packet my least unacked, and update entropy for 2, 3 packets. |
2716 QuicAckFrame ack(0, QuicTime::Zero(), 4); | 2673 QuicAckFrame ack(0, QuicTime::Zero(), 4); |
2717 QuicPacketEntropyHash kRandomEntropyHash = 129u; | 2674 QuicPacketEntropyHash kRandomEntropyHash = 129u; |
2718 ack.sent_info.entropy_hash = kRandomEntropyHash; | 2675 ack.sent_info.entropy_hash = kRandomEntropyHash; |
2719 creator_.set_sequence_number(5); | 2676 creator_.set_sequence_number(5); |
2720 QuicPacketEntropyHash six_packet_entropy_hash = 0; | 2677 QuicPacketEntropyHash six_packet_entropy_hash = 0; |
2721 if (ProcessAckPacket(&ack)) { | 2678 if (ProcessAckPacket(&ack)) { |
2722 six_packet_entropy_hash = 1 << 6; | 2679 six_packet_entropy_hash = 1 << 6; |
2723 } | 2680 } |
2724 | 2681 |
2725 EXPECT_EQ((kRandomEntropyHash + (1 << 5) + six_packet_entropy_hash), | 2682 EXPECT_EQ((kRandomEntropyHash + (1 << 5) + six_packet_entropy_hash), |
2726 outgoing_ack()->received_info.entropy_hash); | 2683 outgoing_ack()->received_info.entropy_hash); |
2727 } | 2684 } |
2728 | 2685 |
2729 TEST_P(QuicConnectionTest, UpdateEntropyHashUptoCurrentPacket) { | 2686 TEST_F(QuicConnectionTest, UpdateEntropyHashUptoCurrentPacket) { |
2730 EXPECT_CALL(visitor_, OnStreamFrames(_)).WillRepeatedly(Return(true)); | 2687 EXPECT_CALL(visitor_, OnStreamFrames(_)).WillRepeatedly(Return(true)); |
2731 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | 2688 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); |
2732 ProcessDataPacket(1, 1, kEntropyFlag); | 2689 ProcessDataPacket(1, 1, kEntropyFlag); |
2733 ProcessDataPacket(5, 1, !kEntropyFlag); | 2690 ProcessDataPacket(5, 1, !kEntropyFlag); |
2734 ProcessDataPacket(22, 1, kEntropyFlag); | 2691 ProcessDataPacket(22, 1, kEntropyFlag); |
2735 EXPECT_EQ(66u, outgoing_ack()->received_info.entropy_hash); | 2692 EXPECT_EQ(66u, outgoing_ack()->received_info.entropy_hash); |
2736 creator_.set_sequence_number(22); | 2693 creator_.set_sequence_number(22); |
2737 QuicPacketEntropyHash kRandomEntropyHash = 85u; | 2694 QuicPacketEntropyHash kRandomEntropyHash = 85u; |
2738 // Current packet is the least unacked packet. | 2695 // Current packet is the least unacked packet. |
2739 QuicAckFrame ack(0, QuicTime::Zero(), 23); | 2696 QuicAckFrame ack(0, QuicTime::Zero(), 23); |
2740 ack.sent_info.entropy_hash = kRandomEntropyHash; | 2697 ack.sent_info.entropy_hash = kRandomEntropyHash; |
2741 QuicPacketEntropyHash ack_entropy_hash = ProcessAckPacket(&ack); | 2698 QuicPacketEntropyHash ack_entropy_hash = ProcessAckPacket(&ack); |
2742 EXPECT_EQ((kRandomEntropyHash + ack_entropy_hash), | 2699 EXPECT_EQ((kRandomEntropyHash + ack_entropy_hash), |
2743 outgoing_ack()->received_info.entropy_hash); | 2700 outgoing_ack()->received_info.entropy_hash); |
2744 ProcessDataPacket(25, 1, kEntropyFlag); | 2701 ProcessDataPacket(25, 1, kEntropyFlag); |
2745 EXPECT_EQ((kRandomEntropyHash + ack_entropy_hash + (1 << (25 % 8))), | 2702 EXPECT_EQ((kRandomEntropyHash + ack_entropy_hash + (1 << (25 % 8))), |
2746 outgoing_ack()->received_info.entropy_hash); | 2703 outgoing_ack()->received_info.entropy_hash); |
2747 } | 2704 } |
2748 | 2705 |
2749 TEST_P(QuicConnectionTest, EntropyCalculationForTruncatedAck) { | 2706 TEST_F(QuicConnectionTest, EntropyCalculationForTruncatedAck) { |
2750 EXPECT_CALL(visitor_, OnStreamFrames(_)).WillRepeatedly(Return(true)); | 2707 EXPECT_CALL(visitor_, OnStreamFrames(_)).WillRepeatedly(Return(true)); |
2751 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | 2708 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); |
2752 QuicPacketEntropyHash entropy[51]; | 2709 QuicPacketEntropyHash entropy[51]; |
2753 entropy[0] = 0; | 2710 entropy[0] = 0; |
2754 for (int i = 1; i < 51; ++i) { | 2711 for (int i = 1; i < 51; ++i) { |
2755 bool should_send = i % 10 != 0; | 2712 bool should_send = i % 10 != 0; |
2756 bool entropy_flag = (i & (i - 1)) != 0; | 2713 bool entropy_flag = (i & (i - 1)) != 0; |
2757 if (!should_send) { | 2714 if (!should_send) { |
2758 entropy[i] = entropy[i - 1]; | 2715 entropy[i] = entropy[i - 1]; |
2759 continue; | 2716 continue; |
2760 } | 2717 } |
2761 if (entropy_flag) { | 2718 if (entropy_flag) { |
2762 entropy[i] = entropy[i - 1] ^ (1 << (i % 8)); | 2719 entropy[i] = entropy[i - 1] ^ (1 << (i % 8)); |
2763 } else { | 2720 } else { |
2764 entropy[i] = entropy[i - 1]; | 2721 entropy[i] = entropy[i - 1]; |
2765 } | 2722 } |
2766 ProcessDataPacket(i, 1, entropy_flag); | 2723 ProcessDataPacket(i, 1, entropy_flag); |
2767 } | 2724 } |
2768 // Till 50 since 50th packet is not sent. | 2725 // Till 50 since 50th packet is not sent. |
2769 for (int i = 1; i < 50; ++i) { | 2726 for (int i = 1; i < 50; ++i) { |
2770 EXPECT_EQ(entropy[i], QuicConnectionPeer::ReceivedEntropyHash( | 2727 EXPECT_EQ(entropy[i], QuicConnectionPeer::ReceivedEntropyHash( |
2771 &connection_, i)); | 2728 &connection_, i)); |
2772 } | 2729 } |
2773 } | 2730 } |
2774 | 2731 |
2775 TEST_P(QuicConnectionTest, CheckSentEntropyHash) { | 2732 TEST_F(QuicConnectionTest, CheckSentEntropyHash) { |
2776 creator_.set_sequence_number(1); | 2733 creator_.set_sequence_number(1); |
2777 SequenceNumberSet missing_packets; | 2734 SequenceNumberSet missing_packets; |
2778 QuicPacketEntropyHash entropy_hash = 0; | 2735 QuicPacketEntropyHash entropy_hash = 0; |
2779 QuicPacketSequenceNumber max_sequence_number = 51; | 2736 QuicPacketSequenceNumber max_sequence_number = 51; |
2780 for (QuicPacketSequenceNumber i = 1; i <= max_sequence_number; ++i) { | 2737 for (QuicPacketSequenceNumber i = 1; i <= max_sequence_number; ++i) { |
2781 bool is_missing = i % 10 != 0; | 2738 bool is_missing = i % 10 != 0; |
2782 bool entropy_flag = (i & (i - 1)) != 0; | 2739 bool entropy_flag = (i & (i - 1)) != 0; |
2783 QuicPacketEntropyHash packet_entropy_hash = 0; | 2740 QuicPacketEntropyHash packet_entropy_hash = 0; |
2784 if (entropy_flag) { | 2741 if (entropy_flag) { |
2785 packet_entropy_hash = 1 << (i % 8); | 2742 packet_entropy_hash = 1 << (i % 8); |
2786 } | 2743 } |
2787 QuicPacket* packet = ConstructDataPacket(i, 0, entropy_flag); | 2744 QuicPacket* packet = ConstructDataPacket(i, 0, entropy_flag); |
2788 connection_.SendPacket( | 2745 connection_.SendPacket( |
2789 ENCRYPTION_NONE, i, packet, packet_entropy_hash, | 2746 ENCRYPTION_NONE, i, packet, packet_entropy_hash, |
2790 HAS_RETRANSMITTABLE_DATA); | 2747 HAS_RETRANSMITTABLE_DATA); |
2791 | 2748 |
2792 if (is_missing) { | 2749 if (is_missing) { |
2793 missing_packets.insert(i); | 2750 missing_packets.insert(i); |
2794 continue; | 2751 continue; |
2795 } | 2752 } |
2796 | 2753 |
2797 entropy_hash ^= packet_entropy_hash; | 2754 entropy_hash ^= packet_entropy_hash; |
2798 } | 2755 } |
2799 EXPECT_TRUE(QuicConnectionPeer::IsValidEntropy( | 2756 EXPECT_TRUE(QuicConnectionPeer::IsValidEntropy( |
2800 &connection_, max_sequence_number, missing_packets, entropy_hash)) | 2757 &connection_, max_sequence_number, missing_packets, entropy_hash)) |
2801 << ""; | 2758 << ""; |
2802 } | 2759 } |
2803 | 2760 |
2804 TEST_P(QuicConnectionTest, ServerSendsVersionNegotiationPacket) { | 2761 TEST_F(QuicConnectionTest, ServerSendsVersionNegotiationPacket) { |
2805 framer_.set_version_for_tests(QUIC_VERSION_UNSUPPORTED); | 2762 framer_.set_version_for_tests(QUIC_VERSION_UNSUPPORTED); |
2806 | 2763 |
2807 QuicPacketHeader header; | 2764 QuicPacketHeader header; |
2808 header.public_header.guid = guid_; | 2765 header.public_header.guid = guid_; |
2809 header.public_header.reset_flag = false; | 2766 header.public_header.reset_flag = false; |
2810 header.public_header.version_flag = true; | 2767 header.public_header.version_flag = true; |
2811 header.entropy_flag = false; | 2768 header.entropy_flag = false; |
2812 header.fec_flag = false; | 2769 header.fec_flag = false; |
2813 header.packet_sequence_number = 12; | 2770 header.packet_sequence_number = 12; |
2814 header.fec_group = 0; | 2771 header.fec_group = 0; |
(...skipping 16 matching lines...) Expand all Loading... |
2831 writer_->version_negotiation_packet()->versions.size()); | 2788 writer_->version_negotiation_packet()->versions.size()); |
2832 | 2789 |
2833 // We expect all versions in kSupportedQuicVersions to be | 2790 // We expect all versions in kSupportedQuicVersions to be |
2834 // included in the packet. | 2791 // included in the packet. |
2835 for (size_t i = 0; i < num_versions; ++i) { | 2792 for (size_t i = 0; i < num_versions; ++i) { |
2836 EXPECT_EQ(kSupportedQuicVersions[i], | 2793 EXPECT_EQ(kSupportedQuicVersions[i], |
2837 writer_->version_negotiation_packet()->versions[i]); | 2794 writer_->version_negotiation_packet()->versions[i]); |
2838 } | 2795 } |
2839 } | 2796 } |
2840 | 2797 |
2841 TEST_P(QuicConnectionTest, ServerSendsVersionNegotiationPacketSocketBlocked) { | 2798 TEST_F(QuicConnectionTest, ServerSendsVersionNegotiationPacketSocketBlocked) { |
2842 framer_.set_version_for_tests(QUIC_VERSION_UNSUPPORTED); | 2799 framer_.set_version_for_tests(QUIC_VERSION_UNSUPPORTED); |
2843 | 2800 |
2844 QuicPacketHeader header; | 2801 QuicPacketHeader header; |
2845 header.public_header.guid = guid_; | 2802 header.public_header.guid = guid_; |
2846 header.public_header.reset_flag = false; | 2803 header.public_header.reset_flag = false; |
2847 header.public_header.version_flag = true; | 2804 header.public_header.version_flag = true; |
2848 header.entropy_flag = false; | 2805 header.entropy_flag = false; |
2849 header.fec_flag = false; | 2806 header.fec_flag = false; |
2850 header.packet_sequence_number = 12; | 2807 header.packet_sequence_number = 12; |
2851 header.fec_group = 0; | 2808 header.fec_group = 0; |
(...skipping 23 matching lines...) Expand all Loading... |
2875 writer_->version_negotiation_packet()->versions.size()); | 2832 writer_->version_negotiation_packet()->versions.size()); |
2876 | 2833 |
2877 // We expect all versions in kSupportedQuicVersions to be | 2834 // We expect all versions in kSupportedQuicVersions to be |
2878 // included in the packet. | 2835 // included in the packet. |
2879 for (size_t i = 0; i < num_versions; ++i) { | 2836 for (size_t i = 0; i < num_versions; ++i) { |
2880 EXPECT_EQ(kSupportedQuicVersions[i], | 2837 EXPECT_EQ(kSupportedQuicVersions[i], |
2881 writer_->version_negotiation_packet()->versions[i]); | 2838 writer_->version_negotiation_packet()->versions[i]); |
2882 } | 2839 } |
2883 } | 2840 } |
2884 | 2841 |
2885 TEST_P(QuicConnectionTest, | 2842 TEST_F(QuicConnectionTest, |
2886 ServerSendsVersionNegotiationPacketSocketBlockedDataBuffered) { | 2843 ServerSendsVersionNegotiationPacketSocketBlockedDataBuffered) { |
2887 framer_.set_version_for_tests(QUIC_VERSION_UNSUPPORTED); | 2844 framer_.set_version_for_tests(QUIC_VERSION_UNSUPPORTED); |
2888 | 2845 |
2889 QuicPacketHeader header; | 2846 QuicPacketHeader header; |
2890 header.public_header.guid = guid_; | 2847 header.public_header.guid = guid_; |
2891 header.public_header.reset_flag = false; | 2848 header.public_header.reset_flag = false; |
2892 header.public_header.version_flag = true; | 2849 header.public_header.version_flag = true; |
2893 header.entropy_flag = false; | 2850 header.entropy_flag = false; |
2894 header.fec_flag = false; | 2851 header.fec_flag = false; |
2895 header.packet_sequence_number = 12; | 2852 header.packet_sequence_number = 12; |
(...skipping 10 matching lines...) Expand all Loading... |
2906 framer_.set_version(QuicVersionMax()); | 2863 framer_.set_version(QuicVersionMax()); |
2907 connection_.set_is_server(true); | 2864 connection_.set_is_server(true); |
2908 writer_->set_blocked(true); | 2865 writer_->set_blocked(true); |
2909 writer_->set_is_write_blocked_data_buffered(true); | 2866 writer_->set_is_write_blocked_data_buffered(true); |
2910 connection_.ProcessUdpPacket(IPEndPoint(), IPEndPoint(), *encrypted); | 2867 connection_.ProcessUdpPacket(IPEndPoint(), IPEndPoint(), *encrypted); |
2911 EXPECT_EQ(0u, writer_->last_packet_size()); | 2868 EXPECT_EQ(0u, writer_->last_packet_size()); |
2912 EXPECT_FALSE(connection_.HasQueuedData()); | 2869 EXPECT_FALSE(connection_.HasQueuedData()); |
2913 EXPECT_TRUE(QuicConnectionPeer::IsWriteBlocked(&connection_)); | 2870 EXPECT_TRUE(QuicConnectionPeer::IsWriteBlocked(&connection_)); |
2914 } | 2871 } |
2915 | 2872 |
2916 TEST_P(QuicConnectionTest, ClientHandlesVersionNegotiation) { | 2873 TEST_F(QuicConnectionTest, ClientHandlesVersionNegotiation) { |
2917 // Start out with some unsupported version. | 2874 // Start out with some unsupported version. |
2918 QuicConnectionPeer::GetFramer(&connection_)->set_version_for_tests( | 2875 QuicConnectionPeer::GetFramer(&connection_)->set_version_for_tests( |
2919 QUIC_VERSION_UNSUPPORTED); | 2876 QUIC_VERSION_UNSUPPORTED); |
2920 | 2877 |
2921 QuicPacketHeader header; | 2878 QuicPacketHeader header; |
2922 header.public_header.guid = guid_; | 2879 header.public_header.guid = guid_; |
2923 header.public_header.reset_flag = false; | 2880 header.public_header.reset_flag = false; |
2924 header.public_header.version_flag = true; | 2881 header.public_header.version_flag = true; |
2925 header.entropy_flag = false; | 2882 header.entropy_flag = false; |
2926 header.fec_flag = false; | 2883 header.fec_flag = false; |
(...skipping 21 matching lines...) Expand all Loading... |
2948 framer_.BuildUnsizedDataPacket(header, frames).packet); | 2905 framer_.BuildUnsizedDataPacket(header, frames).packet); |
2949 encrypted.reset(framer_.EncryptPacket(ENCRYPTION_NONE, 12, *packet)); | 2906 encrypted.reset(framer_.EncryptPacket(ENCRYPTION_NONE, 12, *packet)); |
2950 EXPECT_CALL(visitor_, OnStreamFrames(_)).Times(1); | 2907 EXPECT_CALL(visitor_, OnStreamFrames(_)).Times(1); |
2951 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | 2908 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); |
2952 connection_.ProcessUdpPacket(IPEndPoint(), IPEndPoint(), *encrypted); | 2909 connection_.ProcessUdpPacket(IPEndPoint(), IPEndPoint(), *encrypted); |
2953 | 2910 |
2954 ASSERT_FALSE(QuicPacketCreatorPeer::SendVersionInPacket( | 2911 ASSERT_FALSE(QuicPacketCreatorPeer::SendVersionInPacket( |
2955 QuicConnectionPeer::GetPacketCreator(&connection_))); | 2912 QuicConnectionPeer::GetPacketCreator(&connection_))); |
2956 } | 2913 } |
2957 | 2914 |
2958 TEST_P(QuicConnectionTest, BadVersionNegotiation) { | 2915 TEST_F(QuicConnectionTest, BadVersionNegotiation) { |
2959 QuicPacketHeader header; | 2916 QuicPacketHeader header; |
2960 header.public_header.guid = guid_; | 2917 header.public_header.guid = guid_; |
2961 header.public_header.reset_flag = false; | 2918 header.public_header.reset_flag = false; |
2962 header.public_header.version_flag = true; | 2919 header.public_header.version_flag = true; |
2963 header.entropy_flag = false; | 2920 header.entropy_flag = false; |
2964 header.fec_flag = false; | 2921 header.fec_flag = false; |
2965 header.packet_sequence_number = 12; | 2922 header.packet_sequence_number = 12; |
2966 header.fec_group = 0; | 2923 header.fec_group = 0; |
2967 | 2924 |
2968 QuicVersionVector supported_versions; | 2925 QuicVersionVector supported_versions; |
2969 for (size_t i = 0; i < arraysize(kSupportedQuicVersions); ++i) { | 2926 for (size_t i = 0; i < arraysize(kSupportedQuicVersions); ++i) { |
2970 supported_versions.push_back(kSupportedQuicVersions[i]); | 2927 supported_versions.push_back(kSupportedQuicVersions[i]); |
2971 } | 2928 } |
2972 | 2929 |
2973 // Send a version negotiation packet with the version the client started with. | 2930 // Send a version negotiation packet with the version the client started with. |
2974 // It should be rejected. | 2931 // It should be rejected. |
2975 EXPECT_CALL(visitor_, | 2932 EXPECT_CALL(visitor_, |
2976 OnConnectionClosed(QUIC_INVALID_VERSION_NEGOTIATION_PACKET, | 2933 OnConnectionClosed(QUIC_INVALID_VERSION_NEGOTIATION_PACKET, |
2977 false)); | 2934 false)); |
2978 scoped_ptr<QuicEncryptedPacket> encrypted( | 2935 scoped_ptr<QuicEncryptedPacket> encrypted( |
2979 framer_.BuildVersionNegotiationPacket( | 2936 framer_.BuildVersionNegotiationPacket( |
2980 header.public_header, supported_versions)); | 2937 header.public_header, supported_versions)); |
2981 connection_.ProcessUdpPacket(IPEndPoint(), IPEndPoint(), *encrypted); | 2938 connection_.ProcessUdpPacket(IPEndPoint(), IPEndPoint(), *encrypted); |
2982 } | 2939 } |
2983 | 2940 |
2984 TEST_P(QuicConnectionTest, CheckSendStats) { | 2941 TEST_F(QuicConnectionTest, CheckSendStats) { |
2985 EXPECT_CALL(*send_algorithm_, | 2942 EXPECT_CALL(*send_algorithm_, |
2986 OnPacketSent(_, _, _, NOT_RETRANSMISSION, _)); | 2943 OnPacketSent(_, _, _, NOT_RETRANSMISSION, _)); |
2987 connection_.SendStreamDataWithString(3, "first", 0, !kFin, NULL); | 2944 connection_.SendStreamDataWithString(3, "first", 0, !kFin, NULL); |
2988 size_t first_packet_size = last_sent_packet_size(); | 2945 size_t first_packet_size = last_sent_packet_size(); |
2989 | 2946 |
2990 EXPECT_CALL(*send_algorithm_, | 2947 EXPECT_CALL(*send_algorithm_, |
2991 OnPacketSent(_, _, _, NOT_RETRANSMISSION, _)); | 2948 OnPacketSent(_, _, _, NOT_RETRANSMISSION, _)); |
2992 connection_.SendStreamDataWithString(5, "second", 0, !kFin, NULL); | 2949 connection_.SendStreamDataWithString(5, "second", 0, !kFin, NULL); |
2993 size_t second_packet_size = last_sent_packet_size(); | 2950 size_t second_packet_size = last_sent_packet_size(); |
2994 | 2951 |
(...skipping 36 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
3031 const QuicConnectionStats& stats = connection_.GetStats(); | 2988 const QuicConnectionStats& stats = connection_.GetStats(); |
3032 EXPECT_EQ(3 * first_packet_size + 2 * second_packet_size - kQuicVersionSize, | 2989 EXPECT_EQ(3 * first_packet_size + 2 * second_packet_size - kQuicVersionSize, |
3033 stats.bytes_sent); | 2990 stats.bytes_sent); |
3034 EXPECT_EQ(5u, stats.packets_sent); | 2991 EXPECT_EQ(5u, stats.packets_sent); |
3035 EXPECT_EQ(2 * first_packet_size + second_packet_size - kQuicVersionSize, | 2992 EXPECT_EQ(2 * first_packet_size + second_packet_size - kQuicVersionSize, |
3036 stats.bytes_retransmitted); | 2993 stats.bytes_retransmitted); |
3037 EXPECT_EQ(3u, stats.packets_retransmitted); | 2994 EXPECT_EQ(3u, stats.packets_retransmitted); |
3038 EXPECT_EQ(1u, stats.rto_count); | 2995 EXPECT_EQ(1u, stats.rto_count); |
3039 } | 2996 } |
3040 | 2997 |
3041 TEST_P(QuicConnectionTest, CheckReceiveStats) { | 2998 TEST_F(QuicConnectionTest, CheckReceiveStats) { |
3042 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | 2999 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); |
3043 | 3000 |
3044 size_t received_bytes = 0; | 3001 size_t received_bytes = 0; |
3045 received_bytes += ProcessFecProtectedPacket(1, false, !kEntropyFlag); | 3002 received_bytes += ProcessFecProtectedPacket(1, false, !kEntropyFlag); |
3046 received_bytes += ProcessFecProtectedPacket(3, false, !kEntropyFlag); | 3003 received_bytes += ProcessFecProtectedPacket(3, false, !kEntropyFlag); |
3047 // Should be counted against dropped packets. | 3004 // Should be counted against dropped packets. |
3048 received_bytes += ProcessDataPacket(3, 1, !kEntropyFlag); | 3005 received_bytes += ProcessDataPacket(3, 1, !kEntropyFlag); |
3049 received_bytes += ProcessFecPacket(4, 1, true, !kEntropyFlag, NULL); | 3006 received_bytes += ProcessFecPacket(4, 1, true, !kEntropyFlag, NULL); |
3050 | 3007 |
3051 EXPECT_CALL(*send_algorithm_, SmoothedRtt()).WillOnce( | 3008 EXPECT_CALL(*send_algorithm_, SmoothedRtt()).WillOnce( |
3052 Return(QuicTime::Delta::Zero())); | 3009 Return(QuicTime::Delta::Zero())); |
3053 EXPECT_CALL(*send_algorithm_, BandwidthEstimate()).WillOnce( | 3010 EXPECT_CALL(*send_algorithm_, BandwidthEstimate()).WillOnce( |
3054 Return(QuicBandwidth::Zero())); | 3011 Return(QuicBandwidth::Zero())); |
3055 | 3012 |
3056 const QuicConnectionStats& stats = connection_.GetStats(); | 3013 const QuicConnectionStats& stats = connection_.GetStats(); |
3057 EXPECT_EQ(received_bytes, stats.bytes_received); | 3014 EXPECT_EQ(received_bytes, stats.bytes_received); |
3058 EXPECT_EQ(4u, stats.packets_received); | 3015 EXPECT_EQ(4u, stats.packets_received); |
3059 | 3016 |
3060 EXPECT_EQ(1u, stats.packets_revived); | 3017 EXPECT_EQ(1u, stats.packets_revived); |
3061 EXPECT_EQ(1u, stats.packets_dropped); | 3018 EXPECT_EQ(1u, stats.packets_dropped); |
3062 } | 3019 } |
3063 | 3020 |
3064 TEST_P(QuicConnectionTest, TestFecGroupLimits) { | 3021 TEST_F(QuicConnectionTest, TestFecGroupLimits) { |
3065 // Create and return a group for 1. | 3022 // Create and return a group for 1. |
3066 ASSERT_TRUE(QuicConnectionPeer::GetFecGroup(&connection_, 1) != NULL); | 3023 ASSERT_TRUE(QuicConnectionPeer::GetFecGroup(&connection_, 1) != NULL); |
3067 | 3024 |
3068 // Create and return a group for 2. | 3025 // Create and return a group for 2. |
3069 ASSERT_TRUE(QuicConnectionPeer::GetFecGroup(&connection_, 2) != NULL); | 3026 ASSERT_TRUE(QuicConnectionPeer::GetFecGroup(&connection_, 2) != NULL); |
3070 | 3027 |
3071 // Create and return a group for 4. This should remove 1 but not 2. | 3028 // Create and return a group for 4. This should remove 1 but not 2. |
3072 ASSERT_TRUE(QuicConnectionPeer::GetFecGroup(&connection_, 4) != NULL); | 3029 ASSERT_TRUE(QuicConnectionPeer::GetFecGroup(&connection_, 4) != NULL); |
3073 ASSERT_TRUE(QuicConnectionPeer::GetFecGroup(&connection_, 1) == NULL); | 3030 ASSERT_TRUE(QuicConnectionPeer::GetFecGroup(&connection_, 1) == NULL); |
3074 ASSERT_TRUE(QuicConnectionPeer::GetFecGroup(&connection_, 2) != NULL); | 3031 ASSERT_TRUE(QuicConnectionPeer::GetFecGroup(&connection_, 2) != NULL); |
3075 | 3032 |
3076 // Create and return a group for 3. This will kill off 2. | 3033 // Create and return a group for 3. This will kill off 2. |
3077 ASSERT_TRUE(QuicConnectionPeer::GetFecGroup(&connection_, 3) != NULL); | 3034 ASSERT_TRUE(QuicConnectionPeer::GetFecGroup(&connection_, 3) != NULL); |
3078 ASSERT_TRUE(QuicConnectionPeer::GetFecGroup(&connection_, 2) == NULL); | 3035 ASSERT_TRUE(QuicConnectionPeer::GetFecGroup(&connection_, 2) == NULL); |
3079 | 3036 |
3080 // Verify that adding 5 kills off 3, despite 4 being created before 3. | 3037 // Verify that adding 5 kills off 3, despite 4 being created before 3. |
3081 ASSERT_TRUE(QuicConnectionPeer::GetFecGroup(&connection_, 5) != NULL); | 3038 ASSERT_TRUE(QuicConnectionPeer::GetFecGroup(&connection_, 5) != NULL); |
3082 ASSERT_TRUE(QuicConnectionPeer::GetFecGroup(&connection_, 4) != NULL); | 3039 ASSERT_TRUE(QuicConnectionPeer::GetFecGroup(&connection_, 4) != NULL); |
3083 ASSERT_TRUE(QuicConnectionPeer::GetFecGroup(&connection_, 3) == NULL); | 3040 ASSERT_TRUE(QuicConnectionPeer::GetFecGroup(&connection_, 3) == NULL); |
3084 } | 3041 } |
3085 | 3042 |
3086 TEST_P(QuicConnectionTest, ProcessFramesIfPacketClosedConnection) { | 3043 TEST_F(QuicConnectionTest, ProcessFramesIfPacketClosedConnection) { |
3087 // Construct a packet with stream frame and connection close frame. | 3044 // Construct a packet with stream frame and connection close frame. |
3088 header_.public_header.guid = guid_; | 3045 header_.public_header.guid = guid_; |
3089 header_.packet_sequence_number = 1; | 3046 header_.packet_sequence_number = 1; |
3090 header_.public_header.reset_flag = false; | 3047 header_.public_header.reset_flag = false; |
3091 header_.public_header.version_flag = false; | 3048 header_.public_header.version_flag = false; |
3092 header_.entropy_flag = false; | 3049 header_.entropy_flag = false; |
3093 header_.fec_flag = false; | 3050 header_.fec_flag = false; |
3094 header_.fec_group = 0; | 3051 header_.fec_group = 0; |
3095 | 3052 |
3096 QuicConnectionCloseFrame qccf; | 3053 QuicConnectionCloseFrame qccf; |
(...skipping 11 matching lines...) Expand all Loading... |
3108 scoped_ptr<QuicEncryptedPacket> encrypted(framer_.EncryptPacket( | 3065 scoped_ptr<QuicEncryptedPacket> encrypted(framer_.EncryptPacket( |
3109 ENCRYPTION_NONE, 1, *packet)); | 3066 ENCRYPTION_NONE, 1, *packet)); |
3110 | 3067 |
3111 EXPECT_CALL(visitor_, OnConnectionClosed(QUIC_PEER_GOING_AWAY, true)); | 3068 EXPECT_CALL(visitor_, OnConnectionClosed(QUIC_PEER_GOING_AWAY, true)); |
3112 EXPECT_CALL(visitor_, OnStreamFrames(_)).Times(1); | 3069 EXPECT_CALL(visitor_, OnStreamFrames(_)).Times(1); |
3113 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | 3070 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); |
3114 | 3071 |
3115 connection_.ProcessUdpPacket(IPEndPoint(), IPEndPoint(), *encrypted); | 3072 connection_.ProcessUdpPacket(IPEndPoint(), IPEndPoint(), *encrypted); |
3116 } | 3073 } |
3117 | 3074 |
3118 TEST_P(QuicConnectionTest, SelectMutualVersion) { | 3075 TEST_F(QuicConnectionTest, SelectMutualVersion) { |
3119 // Set the connection to speak the lowest quic version. | 3076 // Set the connection to speak the lowest quic version. |
3120 connection_.set_version(QuicVersionMin()); | 3077 connection_.set_version(QuicVersionMin()); |
3121 EXPECT_EQ(QuicVersionMin(), connection_.version()); | 3078 EXPECT_EQ(QuicVersionMin(), connection_.version()); |
3122 | 3079 |
3123 // Pass in available versions which includes a higher mutually supported | 3080 // Pass in available versions which includes a higher mutually supported |
3124 // version. The higher mutually supported version should be selected. | 3081 // version. The higher mutually supported version should be selected. |
3125 QuicVersionVector supported_versions; | 3082 QuicVersionVector supported_versions; |
3126 for (size_t i = 0; i < arraysize(kSupportedQuicVersions); ++i) { | 3083 for (size_t i = 0; i < arraysize(kSupportedQuicVersions); ++i) { |
3127 supported_versions.push_back(kSupportedQuicVersions[i]); | 3084 supported_versions.push_back(kSupportedQuicVersions[i]); |
3128 } | 3085 } |
3129 EXPECT_TRUE(connection_.SelectMutualVersion(supported_versions)); | 3086 EXPECT_TRUE(connection_.SelectMutualVersion(supported_versions)); |
3130 EXPECT_EQ(QuicVersionMax(), connection_.version()); | 3087 EXPECT_EQ(QuicVersionMax(), connection_.version()); |
3131 | 3088 |
3132 // Expect that the lowest version is selected. | 3089 // Expect that the lowest version is selected. |
3133 // Ensure the lowest supported version is less than the max, unless they're | 3090 // Ensure the lowest supported version is less than the max, unless they're |
3134 // the same. | 3091 // the same. |
3135 EXPECT_LE(QuicVersionMin(), QuicVersionMax()); | 3092 EXPECT_LE(QuicVersionMin(), QuicVersionMax()); |
3136 QuicVersionVector lowest_version_vector; | 3093 QuicVersionVector lowest_version_vector; |
3137 lowest_version_vector.push_back(QuicVersionMin()); | 3094 lowest_version_vector.push_back(QuicVersionMin()); |
3138 EXPECT_TRUE(connection_.SelectMutualVersion(lowest_version_vector)); | 3095 EXPECT_TRUE(connection_.SelectMutualVersion(lowest_version_vector)); |
3139 EXPECT_EQ(QuicVersionMin(), connection_.version()); | 3096 EXPECT_EQ(QuicVersionMin(), connection_.version()); |
3140 | 3097 |
3141 // Shouldn't be able to find a mutually supported version. | 3098 // Shouldn't be able to find a mutually supported version. |
3142 QuicVersionVector unsupported_version; | 3099 QuicVersionVector unsupported_version; |
3143 unsupported_version.push_back(QUIC_VERSION_UNSUPPORTED); | 3100 unsupported_version.push_back(QUIC_VERSION_UNSUPPORTED); |
3144 EXPECT_FALSE(connection_.SelectMutualVersion(unsupported_version)); | 3101 EXPECT_FALSE(connection_.SelectMutualVersion(unsupported_version)); |
3145 } | 3102 } |
3146 | 3103 |
3147 TEST_P(QuicConnectionTest, ConnectionCloseWhenNotWriteBlocked) { | 3104 TEST_F(QuicConnectionTest, ConnectionCloseWhenNotWriteBlocked) { |
3148 writer_->set_blocked(false); // Already default. | 3105 writer_->set_blocked(false); // Already default. |
3149 | 3106 |
3150 // Send a packet (but write will not block). | 3107 // Send a packet (but write will not block). |
3151 connection_.SendStreamDataWithString(1, "foo", 0, !kFin, NULL); | 3108 connection_.SendStreamDataWithString(1, "foo", 0, !kFin, NULL); |
3152 EXPECT_EQ(0u, connection_.NumQueuedPackets()); | 3109 EXPECT_EQ(0u, connection_.NumQueuedPackets()); |
3153 EXPECT_EQ(1u, writer_->packets_write_attempts()); | 3110 EXPECT_EQ(1u, writer_->packets_write_attempts()); |
3154 | 3111 |
3155 // Send an erroneous packet to close the connection. | 3112 // Send an erroneous packet to close the connection. |
3156 EXPECT_CALL(visitor_, OnConnectionClosed(QUIC_INVALID_PACKET_HEADER, false)); | 3113 EXPECT_CALL(visitor_, OnConnectionClosed(QUIC_INVALID_PACKET_HEADER, false)); |
3157 ProcessDataPacket(6000, 0, !kEntropyFlag); | 3114 ProcessDataPacket(6000, 0, !kEntropyFlag); |
3158 EXPECT_EQ(2u, writer_->packets_write_attempts()); | 3115 EXPECT_EQ(2u, writer_->packets_write_attempts()); |
3159 } | 3116 } |
3160 | 3117 |
3161 TEST_P(QuicConnectionTest, ConnectionCloseWhenWriteBlocked) { | 3118 TEST_F(QuicConnectionTest, ConnectionCloseWhenWriteBlocked) { |
3162 EXPECT_EQ(0u, connection_.NumQueuedPackets()); | 3119 EXPECT_EQ(0u, connection_.NumQueuedPackets()); |
3163 writer_->set_blocked(true); | 3120 writer_->set_blocked(true); |
3164 | 3121 |
3165 // Send a packet to so that write will really block. | 3122 // Send a packet to so that write will really block. |
3166 connection_.SendStreamDataWithString(1, "foo", 0, !kFin, NULL); | 3123 connection_.SendStreamDataWithString(1, "foo", 0, !kFin, NULL); |
3167 EXPECT_EQ(1u, connection_.NumQueuedPackets()); | 3124 EXPECT_EQ(1u, connection_.NumQueuedPackets()); |
3168 EXPECT_EQ(1u, writer_->packets_write_attempts()); | 3125 EXPECT_EQ(1u, writer_->packets_write_attempts()); |
3169 | 3126 |
3170 // Send an erroneous packet to close the connection. | 3127 // Send an erroneous packet to close the connection. |
3171 EXPECT_CALL(visitor_, OnConnectionClosed(QUIC_INVALID_PACKET_HEADER, false)); | 3128 EXPECT_CALL(visitor_, OnConnectionClosed(QUIC_INVALID_PACKET_HEADER, false)); |
3172 ProcessDataPacket(6000, 0, !kEntropyFlag); | 3129 ProcessDataPacket(6000, 0, !kEntropyFlag); |
3173 EXPECT_EQ(1u, writer_->packets_write_attempts()); | 3130 EXPECT_EQ(1u, writer_->packets_write_attempts()); |
3174 } | 3131 } |
3175 | 3132 |
3176 TEST_P(QuicConnectionTest, ConnectionCloseWhenNothingPending) { | 3133 TEST_F(QuicConnectionTest, ConnectionCloseWhenNothingPending) { |
3177 writer_->set_blocked(true); | 3134 writer_->set_blocked(true); |
3178 | 3135 |
3179 // Send an erroneous packet to close the connection. | 3136 // Send an erroneous packet to close the connection. |
3180 EXPECT_CALL(visitor_, OnConnectionClosed(QUIC_INVALID_PACKET_HEADER, false)); | 3137 EXPECT_CALL(visitor_, OnConnectionClosed(QUIC_INVALID_PACKET_HEADER, false)); |
3181 ProcessDataPacket(6000, 0, !kEntropyFlag); | 3138 ProcessDataPacket(6000, 0, !kEntropyFlag); |
3182 EXPECT_EQ(1u, writer_->packets_write_attempts()); | 3139 EXPECT_EQ(1u, writer_->packets_write_attempts()); |
3183 } | 3140 } |
3184 | 3141 |
3185 TEST_P(QuicConnectionTest, AckNotifierTriggerCallback) { | 3142 TEST_F(QuicConnectionTest, AckNotifierTriggerCallback) { |
3186 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | 3143 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); |
3187 | 3144 |
3188 // Create a delegate which we expect to be called. | 3145 // Create a delegate which we expect to be called. |
3189 MockAckNotifierDelegate delegate; | 3146 MockAckNotifierDelegate delegate; |
3190 EXPECT_CALL(delegate, OnAckNotification()).Times(1);; | 3147 EXPECT_CALL(delegate, OnAckNotification()).Times(1);; |
3191 | 3148 |
3192 // Send some data, which will register the delegate to be notified. | 3149 // Send some data, which will register the delegate to be notified. |
3193 connection_.SendStreamDataWithString(1, "foo", 0, !kFin, &delegate); | 3150 connection_.SendStreamDataWithString(1, "foo", 0, !kFin, &delegate); |
3194 | 3151 |
3195 // Process an ACK from the server which should trigger the callback. | 3152 // Process an ACK from the server which should trigger the callback. |
3196 EXPECT_CALL(*send_algorithm_, OnIncomingAck(_, _, _)).Times(1); | 3153 EXPECT_CALL(*send_algorithm_, OnIncomingAck(_, _, _)).Times(1); |
3197 QuicAckFrame frame(1, QuicTime::Zero(), 0); | 3154 QuicAckFrame frame(1, QuicTime::Zero(), 0); |
3198 ProcessAckPacket(&frame); | 3155 ProcessAckPacket(&frame); |
3199 } | 3156 } |
3200 | 3157 |
3201 TEST_P(QuicConnectionTest, AckNotifierFailToTriggerCallback) { | 3158 TEST_F(QuicConnectionTest, AckNotifierFailToTriggerCallback) { |
3202 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | 3159 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); |
3203 | 3160 |
3204 // Create a delegate which we don't expect to be called. | 3161 // Create a delegate which we don't expect to be called. |
3205 MockAckNotifierDelegate delegate; | 3162 MockAckNotifierDelegate delegate; |
3206 EXPECT_CALL(delegate, OnAckNotification()).Times(0);; | 3163 EXPECT_CALL(delegate, OnAckNotification()).Times(0);; |
3207 | 3164 |
3208 EXPECT_CALL(*send_algorithm_, OnIncomingAck(_, _, _)).Times(2); | 3165 EXPECT_CALL(*send_algorithm_, OnIncomingAck(_, _, _)).Times(2); |
3209 EXPECT_CALL(*send_algorithm_, OnIncomingLoss(_, _)).Times(1); | 3166 EXPECT_CALL(*send_algorithm_, OnIncomingLoss(_, _)).Times(1); |
3210 | 3167 |
3211 // Send some data, which will register the delegate to be notified. This will | 3168 // Send some data, which will register the delegate to be notified. This will |
3212 // not be ACKed and so the delegate should never be called. | 3169 // not be ACKed and so the delegate should never be called. |
3213 connection_.SendStreamDataWithString(1, "foo", 0, !kFin, &delegate); | 3170 connection_.SendStreamDataWithString(1, "foo", 0, !kFin, &delegate); |
3214 | 3171 |
3215 // Send some other data which we will ACK. | 3172 // Send some other data which we will ACK. |
3216 connection_.SendStreamDataWithString(1, "foo", 0, !kFin, NULL); | 3173 connection_.SendStreamDataWithString(1, "foo", 0, !kFin, NULL); |
3217 connection_.SendStreamDataWithString(1, "bar", 0, !kFin, NULL); | 3174 connection_.SendStreamDataWithString(1, "bar", 0, !kFin, NULL); |
3218 | 3175 |
3219 // Now we receive ACK for packets 2 and 3, but importantly missing packet 1 | 3176 // Now we receive ACK for packets 2 and 3, but importantly missing packet 1 |
3220 // which we registered to be notified about. | 3177 // which we registered to be notified about. |
3221 QuicAckFrame frame(3, QuicTime::Zero(), 0); | 3178 QuicAckFrame frame(3, QuicTime::Zero(), 0); |
3222 frame.received_info.missing_packets.insert(1); | 3179 frame.received_info.missing_packets.insert(1); |
3223 ProcessAckPacket(&frame); | 3180 ProcessAckPacket(&frame); |
3224 } | 3181 } |
3225 | 3182 |
3226 TEST_P(QuicConnectionTest, AckNotifierCallbackAfterRetransmission) { | 3183 TEST_F(QuicConnectionTest, AckNotifierCallbackAfterRetransmission) { |
3227 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | 3184 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); |
3228 | 3185 |
3229 // Create a delegate which we expect to be called. | 3186 // Create a delegate which we expect to be called. |
3230 MockAckNotifierDelegate delegate; | 3187 MockAckNotifierDelegate delegate; |
3231 EXPECT_CALL(delegate, OnAckNotification()).Times(1);; | 3188 EXPECT_CALL(delegate, OnAckNotification()).Times(1);; |
3232 | 3189 |
3233 // In total expect ACKs for all 4 packets. | 3190 // In total expect ACKs for all 4 packets. |
3234 EXPECT_CALL(*send_algorithm_, OnIncomingAck(_, _, _)).Times(4); | 3191 EXPECT_CALL(*send_algorithm_, OnIncomingAck(_, _, _)).Times(4); |
3235 | 3192 |
3236 // We will lose the second packet. | 3193 // We will lose the second packet. |
(...skipping 19 matching lines...) Expand all Loading... |
3256 connection_.GetRetransmissionAlarm()->Fire(); | 3213 connection_.GetRetransmissionAlarm()->Fire(); |
3257 | 3214 |
3258 // Now we get an ACK for packet 5 (retransmitted packet 2), which should | 3215 // Now we get an ACK for packet 5 (retransmitted packet 2), which should |
3259 // trigger the callback. | 3216 // trigger the callback. |
3260 QuicAckFrame second_ack_frame(5, QuicTime::Zero(), 0); | 3217 QuicAckFrame second_ack_frame(5, QuicTime::Zero(), 0); |
3261 ProcessAckPacket(&second_ack_frame); | 3218 ProcessAckPacket(&second_ack_frame); |
3262 } | 3219 } |
3263 | 3220 |
3264 // TODO(rjshade): Add a similar test that FEC recovery on peer (and resulting | 3221 // TODO(rjshade): Add a similar test that FEC recovery on peer (and resulting |
3265 // ACK) triggers notification on our end. | 3222 // ACK) triggers notification on our end. |
3266 TEST_P(QuicConnectionTest, AckNotifierCallbackAfterFECRecovery) { | 3223 TEST_F(QuicConnectionTest, AckNotifierCallbackAfterFECRecovery) { |
3267 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); | 3224 EXPECT_CALL(visitor_, OnSuccessfulVersionNegotiation(_)); |
3268 EXPECT_CALL(visitor_, OnCanWrite()).Times(1).WillOnce(Return(true)); | 3225 EXPECT_CALL(visitor_, OnCanWrite()).Times(1).WillOnce(Return(true)); |
3269 | 3226 |
3270 // Create a delegate which we expect to be called. | 3227 // Create a delegate which we expect to be called. |
3271 MockAckNotifierDelegate delegate; | 3228 MockAckNotifierDelegate delegate; |
3272 EXPECT_CALL(delegate, OnAckNotification()).Times(1);; | 3229 EXPECT_CALL(delegate, OnAckNotification()).Times(1);; |
3273 | 3230 |
3274 // Expect ACKs for 1 packet. | 3231 // Expect ACKs for 1 packet. |
3275 EXPECT_CALL(*send_algorithm_, OnIncomingAck(_, _, _)).Times(1); | 3232 EXPECT_CALL(*send_algorithm_, OnIncomingAck(_, _, _)).Times(1); |
3276 | 3233 |
(...skipping 73 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
3350 MOCK_METHOD1(OnPublicResetPacket, | 3307 MOCK_METHOD1(OnPublicResetPacket, |
3351 void(const QuicPublicResetPacket&)); | 3308 void(const QuicPublicResetPacket&)); |
3352 | 3309 |
3353 MOCK_METHOD1(OnVersionNegotiationPacket, | 3310 MOCK_METHOD1(OnVersionNegotiationPacket, |
3354 void(const QuicVersionNegotiationPacket&)); | 3311 void(const QuicVersionNegotiationPacket&)); |
3355 | 3312 |
3356 MOCK_METHOD2(OnRevivedPacket, | 3313 MOCK_METHOD2(OnRevivedPacket, |
3357 void(const QuicPacketHeader&, StringPiece payload)); | 3314 void(const QuicPacketHeader&, StringPiece payload)); |
3358 }; | 3315 }; |
3359 | 3316 |
3360 TEST_P(QuicConnectionTest, OnPacketHeaderDebugVisitor) { | 3317 TEST_F(QuicConnectionTest, OnPacketHeaderDebugVisitor) { |
3361 QuicPacketHeader header; | 3318 QuicPacketHeader header; |
3362 | 3319 |
3363 scoped_ptr<MockQuicConnectionDebugVisitor> | 3320 scoped_ptr<MockQuicConnectionDebugVisitor> |
3364 debug_visitor(new StrictMock<MockQuicConnectionDebugVisitor>); | 3321 debug_visitor(new StrictMock<MockQuicConnectionDebugVisitor>); |
3365 connection_.set_debug_visitor(debug_visitor.get()); | 3322 connection_.set_debug_visitor(debug_visitor.get()); |
3366 EXPECT_CALL(*debug_visitor, OnPacketHeader(Ref(header))).Times(1); | 3323 EXPECT_CALL(*debug_visitor, OnPacketHeader(Ref(header))).Times(1); |
3367 connection_.OnPacketHeader(header); | 3324 connection_.OnPacketHeader(header); |
3368 } | 3325 } |
3369 | 3326 |
3370 } // namespace | 3327 } // namespace |
3371 } // namespace test | 3328 } // namespace test |
3372 } // namespace net | 3329 } // namespace net |
OLD | NEW |