| 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_packet_creator.h" | 5 #include "net/quic/quic_packet_creator.h" |
| 6 | 6 |
| 7 #include <stdint.h> | 7 #include <stdint.h> |
| 8 | 8 |
| 9 #include "base/macros.h" | 9 #include "base/macros.h" |
| 10 #include "base/stl_util.h" | 10 #include "base/stl_util.h" |
| (...skipping 91 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 102 if (serialized_packet == nullptr) { | 102 if (serialized_packet == nullptr) { |
| 103 return; | 103 return; |
| 104 } | 104 } |
| 105 | 105 |
| 106 serialized_packet_ = *serialized_packet; | 106 serialized_packet_ = *serialized_packet; |
| 107 serialized_packet_.packet = serialized_packet->packet->Clone(); | 107 serialized_packet_.packet = serialized_packet->packet->Clone(); |
| 108 delete serialized_packet->packet; | 108 delete serialized_packet->packet; |
| 109 serialized_packet->retransmittable_frames.clear(); | 109 serialized_packet->retransmittable_frames.clear(); |
| 110 } | 110 } |
| 111 | 111 |
| 112 void DeleteSerializedPacket() { ClearSerializedPacket(&serialized_packet_); } |
| 113 |
| 112 protected: | 114 protected: |
| 113 QuicPacketCreatorTest() | 115 QuicPacketCreatorTest() |
| 114 : server_framer_(SupportedVersions(GetParam().version), | 116 : server_framer_(SupportedVersions(GetParam().version), |
| 115 QuicTime::Zero(), | 117 QuicTime::Zero(), |
| 116 Perspective::IS_SERVER), | 118 Perspective::IS_SERVER), |
| 117 client_framer_(SupportedVersions(GetParam().version), | 119 client_framer_(SupportedVersions(GetParam().version), |
| 118 QuicTime::Zero(), | 120 QuicTime::Zero(), |
| 119 Perspective::IS_CLIENT), | 121 Perspective::IS_CLIENT), |
| 120 connection_id_(2), | 122 connection_id_(2), |
| 121 data_("foo"), | 123 data_("foo"), |
| (...skipping 19 matching lines...) Expand all Loading... |
| 141 ~QuicPacketCreatorTest() override {} | 143 ~QuicPacketCreatorTest() override {} |
| 142 | 144 |
| 143 SerializedPacket SerializeAllFrames(const QuicFrames& frames) { | 145 SerializedPacket SerializeAllFrames(const QuicFrames& frames) { |
| 144 SerializedPacket packet = QuicPacketCreatorPeer::SerializeAllFrames( | 146 SerializedPacket packet = QuicPacketCreatorPeer::SerializeAllFrames( |
| 145 &creator_, frames, buffer_, kMaxPacketSize); | 147 &creator_, frames, buffer_, kMaxPacketSize); |
| 146 EXPECT_EQ(QuicPacketCreatorPeer::GetEncryptionLevel(&creator_), | 148 EXPECT_EQ(QuicPacketCreatorPeer::GetEncryptionLevel(&creator_), |
| 147 packet.encryption_level); | 149 packet.encryption_level); |
| 148 return packet; | 150 return packet; |
| 149 } | 151 } |
| 150 | 152 |
| 151 void ProcessPacket(QuicEncryptedPacket* encrypted) { | 153 void ProcessPacket(const SerializedPacket& packet) { |
| 152 server_framer_.ProcessPacket(*encrypted); | 154 server_framer_.ProcessPacket(*packet.packet); |
| 153 } | 155 } |
| 154 | 156 |
| 155 void CheckStreamFrame(const QuicFrame& frame, | 157 void CheckStreamFrame(const QuicFrame& frame, |
| 156 QuicStreamId stream_id, | 158 QuicStreamId stream_id, |
| 157 const string& data, | 159 const string& data, |
| 158 QuicStreamOffset offset, | 160 QuicStreamOffset offset, |
| 159 bool fin) { | 161 bool fin) { |
| 160 EXPECT_EQ(STREAM_FRAME, frame.type); | 162 EXPECT_EQ(STREAM_FRAME, frame.type); |
| 161 ASSERT_TRUE(frame.stream_frame); | 163 ASSERT_TRUE(frame.stream_frame); |
| 162 EXPECT_EQ(stream_id, frame.stream_frame->stream_id); | 164 EXPECT_EQ(stream_id, frame.stream_frame->stream_id); |
| (...skipping 88 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 251 EXPECT_CALL(framer_visitor_, OnPacket()); | 253 EXPECT_CALL(framer_visitor_, OnPacket()); |
| 252 EXPECT_CALL(framer_visitor_, OnUnauthenticatedPublicHeader(_)); | 254 EXPECT_CALL(framer_visitor_, OnUnauthenticatedPublicHeader(_)); |
| 253 EXPECT_CALL(framer_visitor_, OnUnauthenticatedHeader(_)); | 255 EXPECT_CALL(framer_visitor_, OnUnauthenticatedHeader(_)); |
| 254 EXPECT_CALL(framer_visitor_, OnDecryptedPacket(_)); | 256 EXPECT_CALL(framer_visitor_, OnDecryptedPacket(_)); |
| 255 EXPECT_CALL(framer_visitor_, OnPacketHeader(_)); | 257 EXPECT_CALL(framer_visitor_, OnPacketHeader(_)); |
| 256 EXPECT_CALL(framer_visitor_, OnAckFrame(_)); | 258 EXPECT_CALL(framer_visitor_, OnAckFrame(_)); |
| 257 EXPECT_CALL(framer_visitor_, OnStreamFrame(_)); | 259 EXPECT_CALL(framer_visitor_, OnStreamFrame(_)); |
| 258 EXPECT_CALL(framer_visitor_, OnStreamFrame(_)); | 260 EXPECT_CALL(framer_visitor_, OnStreamFrame(_)); |
| 259 EXPECT_CALL(framer_visitor_, OnPacketComplete()); | 261 EXPECT_CALL(framer_visitor_, OnPacketComplete()); |
| 260 } | 262 } |
| 261 ProcessPacket(serialized.packet); | 263 ProcessPacket(serialized); |
| 262 delete serialized.packet; | 264 delete serialized.packet; |
| 263 } | 265 } |
| 264 } | 266 } |
| 265 | 267 |
| 266 TEST_P(QuicPacketCreatorTest, SerializeWithFEC) { | 268 TEST_P(QuicPacketCreatorTest, SerializeWithFEC) { |
| 267 // Send FEC packet every 6 packets. | 269 // Send FEC packet every 6 packets. |
| 268 creator_.set_max_packets_per_fec_group(6); | 270 creator_.set_max_packets_per_fec_group(6); |
| 269 // Should return false since we do not have enough packets in the FEC group to | 271 // Should return false since we do not have enough packets in the FEC group to |
| 270 // trigger an FEC packet. | 272 // trigger an FEC packet. |
| 271 ASSERT_FALSE(creator_.ShouldSendFec(/*force_close=*/false)); | 273 ASSERT_FALSE(creator_.ShouldSendFec(/*force_close=*/false)); |
| (...skipping 12 matching lines...) Expand all Loading... |
| 284 InSequence s; | 286 InSequence s; |
| 285 EXPECT_CALL(framer_visitor_, OnPacket()); | 287 EXPECT_CALL(framer_visitor_, OnPacket()); |
| 286 EXPECT_CALL(framer_visitor_, OnUnauthenticatedPublicHeader(_)); | 288 EXPECT_CALL(framer_visitor_, OnUnauthenticatedPublicHeader(_)); |
| 287 EXPECT_CALL(framer_visitor_, OnUnauthenticatedHeader(_)); | 289 EXPECT_CALL(framer_visitor_, OnUnauthenticatedHeader(_)); |
| 288 EXPECT_CALL(framer_visitor_, OnDecryptedPacket(_)); | 290 EXPECT_CALL(framer_visitor_, OnDecryptedPacket(_)); |
| 289 EXPECT_CALL(framer_visitor_, OnPacketHeader(_)); | 291 EXPECT_CALL(framer_visitor_, OnPacketHeader(_)); |
| 290 EXPECT_CALL(framer_visitor_, OnFecProtectedPayload(_)); | 292 EXPECT_CALL(framer_visitor_, OnFecProtectedPayload(_)); |
| 291 EXPECT_CALL(framer_visitor_, OnStreamFrame(_)); | 293 EXPECT_CALL(framer_visitor_, OnStreamFrame(_)); |
| 292 EXPECT_CALL(framer_visitor_, OnPacketComplete()); | 294 EXPECT_CALL(framer_visitor_, OnPacketComplete()); |
| 293 } | 295 } |
| 294 ProcessPacket(serialized_packet_.packet); | 296 ProcessPacket(serialized_packet_); |
| 295 ClearSerializedPacket(&serialized_packet_); | 297 DeleteSerializedPacket(); |
| 296 | 298 |
| 297 // Should return false since we do not have enough packets in the FEC group to | 299 // Should return false since we do not have enough packets in the FEC group to |
| 298 // trigger an FEC packet. | 300 // trigger an FEC packet. |
| 299 ASSERT_FALSE(creator_.ShouldSendFec(/*force_close=*/false)); | 301 ASSERT_FALSE(creator_.ShouldSendFec(/*force_close=*/false)); |
| 300 // Should return true since there are packets in the FEC group. | 302 // Should return true since there are packets in the FEC group. |
| 301 ASSERT_TRUE(creator_.ShouldSendFec(/*force_close=*/true)); | 303 ASSERT_TRUE(creator_.ShouldSendFec(/*force_close=*/true)); |
| 302 | 304 |
| 303 EXPECT_CALL(delegate_, OnSerializedPacket(_)) | 305 EXPECT_CALL(delegate_, OnSerializedPacket(_)) |
| 304 .WillOnce(Invoke(this, &QuicPacketCreatorTest::SaveSerializedPacket)); | 306 .WillOnce(Invoke(this, &QuicPacketCreatorTest::SaveSerializedPacket)); |
| 305 creator_.MaybeSendFecPacketAndCloseGroup(/*force_send_fec=*/true, | 307 creator_.MaybeSendFecPacketAndCloseGroup(/*force_send_fec=*/true, |
| 306 /*is_fec_timeout=*/false); | 308 /*is_fec_timeout=*/false); |
| 307 ASSERT_EQ(2u, serialized_packet_.packet_number); | 309 ASSERT_EQ(2u, serialized_packet_.packet_number); |
| 308 { | 310 { |
| 309 InSequence s; | 311 InSequence s; |
| 310 EXPECT_CALL(framer_visitor_, OnPacket()); | 312 EXPECT_CALL(framer_visitor_, OnPacket()); |
| 311 EXPECT_CALL(framer_visitor_, OnUnauthenticatedPublicHeader(_)); | 313 EXPECT_CALL(framer_visitor_, OnUnauthenticatedPublicHeader(_)); |
| 312 EXPECT_CALL(framer_visitor_, OnUnauthenticatedHeader(_)); | 314 EXPECT_CALL(framer_visitor_, OnUnauthenticatedHeader(_)); |
| 313 EXPECT_CALL(framer_visitor_, OnDecryptedPacket(_)); | 315 EXPECT_CALL(framer_visitor_, OnDecryptedPacket(_)); |
| 314 EXPECT_CALL(framer_visitor_, OnPacketHeader(_)); | 316 EXPECT_CALL(framer_visitor_, OnPacketHeader(_)); |
| 315 EXPECT_CALL(framer_visitor_, OnFecData(_)); | 317 EXPECT_CALL(framer_visitor_, OnFecData(_)); |
| 316 EXPECT_CALL(framer_visitor_, OnPacketComplete()); | 318 EXPECT_CALL(framer_visitor_, OnPacketComplete()); |
| 317 } | 319 } |
| 318 ProcessPacket(serialized_packet_.packet); | 320 ProcessPacket(serialized_packet_); |
| 319 ClearSerializedPacket(&serialized_packet_); | 321 DeleteSerializedPacket(); |
| 320 } | 322 } |
| 321 | 323 |
| 322 TEST_P(QuicPacketCreatorTest, SerializeChangingSequenceNumberLength) { | 324 TEST_P(QuicPacketCreatorTest, SerializeChangingSequenceNumberLength) { |
| 323 frames_.push_back(QuicFrame(new QuicAckFrame(MakeAckFrame(0u)))); | 325 frames_.push_back(QuicFrame(new QuicAckFrame(MakeAckFrame(0u)))); |
| 324 creator_.AddSavedFrame(frames_[0]); | 326 creator_.AddSavedFrame(frames_[0]); |
| 325 QuicPacketCreatorPeer::SetNextPacketNumberLength(&creator_, | 327 QuicPacketCreatorPeer::SetNextPacketNumberLength(&creator_, |
| 326 PACKET_4BYTE_PACKET_NUMBER); | 328 PACKET_4BYTE_PACKET_NUMBER); |
| 327 | 329 |
| 328 EXPECT_CALL(delegate_, OnSerializedPacket(_)) | 330 EXPECT_CALL(delegate_, OnSerializedPacket(_)) |
| 329 .WillRepeatedly( | 331 .WillRepeatedly( |
| 330 Invoke(this, &QuicPacketCreatorTest::SaveSerializedPacket)); | 332 Invoke(this, &QuicPacketCreatorTest::SaveSerializedPacket)); |
| 331 creator_.Flush(); | 333 creator_.Flush(); |
| 332 // The packet number length will not change mid-packet. | 334 // The packet number length will not change mid-packet. |
| 333 EXPECT_EQ(PACKET_1BYTE_PACKET_NUMBER, | 335 EXPECT_EQ(PACKET_1BYTE_PACKET_NUMBER, |
| 334 serialized_packet_.packet_number_length); | 336 serialized_packet_.packet_number_length); |
| 335 | 337 |
| 336 { | 338 { |
| 337 InSequence s; | 339 InSequence s; |
| 338 EXPECT_CALL(framer_visitor_, OnPacket()); | 340 EXPECT_CALL(framer_visitor_, OnPacket()); |
| 339 EXPECT_CALL(framer_visitor_, OnUnauthenticatedPublicHeader(_)); | 341 EXPECT_CALL(framer_visitor_, OnUnauthenticatedPublicHeader(_)); |
| 340 EXPECT_CALL(framer_visitor_, OnUnauthenticatedHeader(_)); | 342 EXPECT_CALL(framer_visitor_, OnUnauthenticatedHeader(_)); |
| 341 EXPECT_CALL(framer_visitor_, OnDecryptedPacket(_)); | 343 EXPECT_CALL(framer_visitor_, OnDecryptedPacket(_)); |
| 342 EXPECT_CALL(framer_visitor_, OnPacketHeader(_)); | 344 EXPECT_CALL(framer_visitor_, OnPacketHeader(_)); |
| 343 EXPECT_CALL(framer_visitor_, OnAckFrame(_)); | 345 EXPECT_CALL(framer_visitor_, OnAckFrame(_)); |
| 344 EXPECT_CALL(framer_visitor_, OnPacketComplete()); | 346 EXPECT_CALL(framer_visitor_, OnPacketComplete()); |
| 345 } | 347 } |
| 346 ProcessPacket(serialized_packet_.packet); | 348 ProcessPacket(serialized_packet_); |
| 347 ClearSerializedPacket(&serialized_packet_); | 349 DeleteSerializedPacket(); |
| 348 | 350 |
| 349 creator_.AddSavedFrame(frames_[0]); | 351 creator_.AddSavedFrame(frames_[0]); |
| 350 creator_.Flush(); | 352 creator_.Flush(); |
| 351 // Now the actual packet number length should have changed. | 353 // Now the actual packet number length should have changed. |
| 352 EXPECT_EQ(PACKET_4BYTE_PACKET_NUMBER, | 354 EXPECT_EQ(PACKET_4BYTE_PACKET_NUMBER, |
| 353 serialized_packet_.packet_number_length); | 355 serialized_packet_.packet_number_length); |
| 354 delete frames_[0].ack_frame; | 356 delete frames_[0].ack_frame; |
| 355 | 357 |
| 356 { | 358 { |
| 357 InSequence s; | 359 InSequence s; |
| 358 EXPECT_CALL(framer_visitor_, OnPacket()); | 360 EXPECT_CALL(framer_visitor_, OnPacket()); |
| 359 EXPECT_CALL(framer_visitor_, OnUnauthenticatedPublicHeader(_)); | 361 EXPECT_CALL(framer_visitor_, OnUnauthenticatedPublicHeader(_)); |
| 360 EXPECT_CALL(framer_visitor_, OnUnauthenticatedHeader(_)); | 362 EXPECT_CALL(framer_visitor_, OnUnauthenticatedHeader(_)); |
| 361 EXPECT_CALL(framer_visitor_, OnDecryptedPacket(_)); | 363 EXPECT_CALL(framer_visitor_, OnDecryptedPacket(_)); |
| 362 EXPECT_CALL(framer_visitor_, OnPacketHeader(_)); | 364 EXPECT_CALL(framer_visitor_, OnPacketHeader(_)); |
| 363 EXPECT_CALL(framer_visitor_, OnAckFrame(_)); | 365 EXPECT_CALL(framer_visitor_, OnAckFrame(_)); |
| 364 EXPECT_CALL(framer_visitor_, OnPacketComplete()); | 366 EXPECT_CALL(framer_visitor_, OnPacketComplete()); |
| 365 } | 367 } |
| 366 ProcessPacket(serialized_packet_.packet); | 368 ProcessPacket(serialized_packet_); |
| 367 ClearSerializedPacket(&serialized_packet_); | 369 DeleteSerializedPacket(); |
| 368 } | 370 } |
| 369 | 371 |
| 370 TEST_P(QuicPacketCreatorTest, ChangeSequenceNumberLengthMidPacket) { | 372 TEST_P(QuicPacketCreatorTest, ChangeSequenceNumberLengthMidPacket) { |
| 371 // Changing the packet number length with queued frames in the creator | 373 // Changing the packet number length with queued frames in the creator |
| 372 // should hold the change until after any currently queued frames are | 374 // should hold the change until after any currently queued frames are |
| 373 // serialized. | 375 // serialized. |
| 374 | 376 |
| 375 // Packet 1. | 377 // Packet 1. |
| 376 // Queue a frame in the creator. | 378 // Queue a frame in the creator. |
| 377 EXPECT_FALSE(creator_.HasPendingFrames()); | 379 EXPECT_FALSE(creator_.HasPendingFrames()); |
| (...skipping 26 matching lines...) Expand all Loading... |
| 404 EXPECT_CALL(framer_visitor_, OnPacket()); | 406 EXPECT_CALL(framer_visitor_, OnPacket()); |
| 405 EXPECT_CALL(framer_visitor_, OnUnauthenticatedPublicHeader(_)); | 407 EXPECT_CALL(framer_visitor_, OnUnauthenticatedPublicHeader(_)); |
| 406 EXPECT_CALL(framer_visitor_, OnUnauthenticatedHeader(_)); | 408 EXPECT_CALL(framer_visitor_, OnUnauthenticatedHeader(_)); |
| 407 EXPECT_CALL(framer_visitor_, OnDecryptedPacket(_)); | 409 EXPECT_CALL(framer_visitor_, OnDecryptedPacket(_)); |
| 408 EXPECT_CALL(framer_visitor_, OnPacketHeader(_)) | 410 EXPECT_CALL(framer_visitor_, OnPacketHeader(_)) |
| 409 .WillOnce(DoAll(SaveArg<0>(&header), Return(true))); | 411 .WillOnce(DoAll(SaveArg<0>(&header), Return(true))); |
| 410 EXPECT_CALL(framer_visitor_, OnAckFrame(_)); | 412 EXPECT_CALL(framer_visitor_, OnAckFrame(_)); |
| 411 EXPECT_CALL(framer_visitor_, OnStopWaitingFrame(_)); | 413 EXPECT_CALL(framer_visitor_, OnStopWaitingFrame(_)); |
| 412 EXPECT_CALL(framer_visitor_, OnPacketComplete()); | 414 EXPECT_CALL(framer_visitor_, OnPacketComplete()); |
| 413 } | 415 } |
| 414 ProcessPacket(serialized_packet_.packet); | 416 ProcessPacket(serialized_packet_); |
| 415 EXPECT_EQ(PACKET_1BYTE_PACKET_NUMBER, | 417 EXPECT_EQ(PACKET_1BYTE_PACKET_NUMBER, |
| 416 header.public_header.packet_number_length); | 418 header.public_header.packet_number_length); |
| 417 ClearSerializedPacket(&serialized_packet_); | 419 DeleteSerializedPacket(); |
| 418 | 420 |
| 419 // Packet 2. | 421 // Packet 2. |
| 420 EXPECT_FALSE(creator_.HasPendingFrames()); | 422 EXPECT_FALSE(creator_.HasPendingFrames()); |
| 421 // Generate Packet 2 with one frame -- packet number length should now | 423 // Generate Packet 2 with one frame -- packet number length should now |
| 422 // change to 4 bytes. | 424 // change to 4 bytes. |
| 423 EXPECT_TRUE(creator_.AddSavedFrame(QuicFrame(&stop_waiting_frame))); | 425 EXPECT_TRUE(creator_.AddSavedFrame(QuicFrame(&stop_waiting_frame))); |
| 424 EXPECT_TRUE(creator_.HasPendingFrames()); | 426 EXPECT_TRUE(creator_.HasPendingFrames()); |
| 425 | 427 |
| 426 // Ensure the packet is successfully created. | 428 // Ensure the packet is successfully created. |
| 427 creator_.Flush(); | 429 creator_.Flush(); |
| 428 ASSERT_TRUE(serialized_packet_.packet); | 430 ASSERT_TRUE(serialized_packet_.packet); |
| 429 EXPECT_EQ(PACKET_4BYTE_PACKET_NUMBER, | 431 EXPECT_EQ(PACKET_4BYTE_PACKET_NUMBER, |
| 430 serialized_packet_.packet_number_length); | 432 serialized_packet_.packet_number_length); |
| 431 | 433 |
| 432 // Verify that header in transmitted packet has 4 byte sequence length. | 434 // Verify that header in transmitted packet has 4 byte sequence length. |
| 433 { | 435 { |
| 434 InSequence s; | 436 InSequence s; |
| 435 EXPECT_CALL(framer_visitor_, OnPacket()); | 437 EXPECT_CALL(framer_visitor_, OnPacket()); |
| 436 EXPECT_CALL(framer_visitor_, OnUnauthenticatedPublicHeader(_)); | 438 EXPECT_CALL(framer_visitor_, OnUnauthenticatedPublicHeader(_)); |
| 437 EXPECT_CALL(framer_visitor_, OnUnauthenticatedHeader(_)); | 439 EXPECT_CALL(framer_visitor_, OnUnauthenticatedHeader(_)); |
| 438 EXPECT_CALL(framer_visitor_, OnDecryptedPacket(_)); | 440 EXPECT_CALL(framer_visitor_, OnDecryptedPacket(_)); |
| 439 EXPECT_CALL(framer_visitor_, OnPacketHeader(_)) | 441 EXPECT_CALL(framer_visitor_, OnPacketHeader(_)) |
| 440 .WillOnce(DoAll(SaveArg<0>(&header), Return(true))); | 442 .WillOnce(DoAll(SaveArg<0>(&header), Return(true))); |
| 441 EXPECT_CALL(framer_visitor_, OnStopWaitingFrame(_)); | 443 EXPECT_CALL(framer_visitor_, OnStopWaitingFrame(_)); |
| 442 EXPECT_CALL(framer_visitor_, OnPacketComplete()); | 444 EXPECT_CALL(framer_visitor_, OnPacketComplete()); |
| 443 } | 445 } |
| 444 ProcessPacket(serialized_packet_.packet); | 446 ProcessPacket(serialized_packet_); |
| 445 EXPECT_EQ(PACKET_4BYTE_PACKET_NUMBER, | 447 EXPECT_EQ(PACKET_4BYTE_PACKET_NUMBER, |
| 446 header.public_header.packet_number_length); | 448 header.public_header.packet_number_length); |
| 447 | 449 |
| 448 ClearSerializedPacket(&serialized_packet_); | 450 DeleteSerializedPacket(); |
| 449 delete ack_frame.ack_frame; | 451 delete ack_frame.ack_frame; |
| 450 } | 452 } |
| 451 | 453 |
| 452 TEST_P(QuicPacketCreatorTest, SerializeWithFECChangingSequenceNumberLength) { | 454 TEST_P(QuicPacketCreatorTest, SerializeWithFECChangingSequenceNumberLength) { |
| 453 // Test goal is to test the following sequence (P1 => generate Packet 1): | 455 // Test goal is to test the following sequence (P1 => generate Packet 1): |
| 454 // P1 <change seq num length> P2 FEC, | 456 // P1 <change seq num length> P2 FEC, |
| 455 // and we expect that packet number length should not change until the end | 457 // and we expect that packet number length should not change until the end |
| 456 // of the open FEC group. | 458 // of the open FEC group. |
| 457 | 459 |
| 458 // Send FEC packet every 6 packets. | 460 // Send FEC packet every 6 packets. |
| (...skipping 21 matching lines...) Expand all Loading... |
| 480 InSequence s; | 482 InSequence s; |
| 481 EXPECT_CALL(framer_visitor_, OnPacket()); | 483 EXPECT_CALL(framer_visitor_, OnPacket()); |
| 482 EXPECT_CALL(framer_visitor_, OnUnauthenticatedPublicHeader(_)); | 484 EXPECT_CALL(framer_visitor_, OnUnauthenticatedPublicHeader(_)); |
| 483 EXPECT_CALL(framer_visitor_, OnUnauthenticatedHeader(_)); | 485 EXPECT_CALL(framer_visitor_, OnUnauthenticatedHeader(_)); |
| 484 EXPECT_CALL(framer_visitor_, OnDecryptedPacket(_)); | 486 EXPECT_CALL(framer_visitor_, OnDecryptedPacket(_)); |
| 485 EXPECT_CALL(framer_visitor_, OnPacketHeader(_)); | 487 EXPECT_CALL(framer_visitor_, OnPacketHeader(_)); |
| 486 EXPECT_CALL(framer_visitor_, OnFecProtectedPayload(_)); | 488 EXPECT_CALL(framer_visitor_, OnFecProtectedPayload(_)); |
| 487 EXPECT_CALL(framer_visitor_, OnStreamFrame(_)); | 489 EXPECT_CALL(framer_visitor_, OnStreamFrame(_)); |
| 488 EXPECT_CALL(framer_visitor_, OnPacketComplete()); | 490 EXPECT_CALL(framer_visitor_, OnPacketComplete()); |
| 489 } | 491 } |
| 490 ProcessPacket(serialized_packet_.packet); | 492 ProcessPacket(serialized_packet_); |
| 491 ClearSerializedPacket(&serialized_packet_); | 493 DeleteSerializedPacket(); |
| 492 | 494 |
| 493 // Generate Packet 2. | 495 // Generate Packet 2. |
| 494 ASSERT_TRUE(creator_.ConsumeData(2u, io_vector, 0u, 0u, false, false, &frame, | 496 ASSERT_TRUE(creator_.ConsumeData(2u, io_vector, 0u, 0u, false, false, &frame, |
| 495 MUST_FEC_PROTECT)); | 497 MUST_FEC_PROTECT)); |
| 496 creator_.Flush(); | 498 creator_.Flush(); |
| 497 EXPECT_EQ(PACKET_1BYTE_PACKET_NUMBER, | 499 EXPECT_EQ(PACKET_1BYTE_PACKET_NUMBER, |
| 498 serialized_packet_.packet_number_length); | 500 serialized_packet_.packet_number_length); |
| 499 | 501 |
| 500 { | 502 { |
| 501 InSequence s; | 503 InSequence s; |
| 502 EXPECT_CALL(framer_visitor_, OnPacket()); | 504 EXPECT_CALL(framer_visitor_, OnPacket()); |
| 503 EXPECT_CALL(framer_visitor_, OnUnauthenticatedPublicHeader(_)); | 505 EXPECT_CALL(framer_visitor_, OnUnauthenticatedPublicHeader(_)); |
| 504 EXPECT_CALL(framer_visitor_, OnUnauthenticatedHeader(_)); | 506 EXPECT_CALL(framer_visitor_, OnUnauthenticatedHeader(_)); |
| 505 EXPECT_CALL(framer_visitor_, OnDecryptedPacket(_)); | 507 EXPECT_CALL(framer_visitor_, OnDecryptedPacket(_)); |
| 506 EXPECT_CALL(framer_visitor_, OnPacketHeader(_)); | 508 EXPECT_CALL(framer_visitor_, OnPacketHeader(_)); |
| 507 EXPECT_CALL(framer_visitor_, OnFecProtectedPayload(_)); | 509 EXPECT_CALL(framer_visitor_, OnFecProtectedPayload(_)); |
| 508 EXPECT_CALL(framer_visitor_, OnStreamFrame(_)); | 510 EXPECT_CALL(framer_visitor_, OnStreamFrame(_)); |
| 509 EXPECT_CALL(framer_visitor_, OnPacketComplete()); | 511 EXPECT_CALL(framer_visitor_, OnPacketComplete()); |
| 510 } | 512 } |
| 511 ProcessPacket(serialized_packet_.packet); | 513 ProcessPacket(serialized_packet_); |
| 512 ClearSerializedPacket(&serialized_packet_); | 514 DeleteSerializedPacket(); |
| 513 | 515 |
| 514 // Should return false since we do not have enough packets in the FEC group to | 516 // Should return false since we do not have enough packets in the FEC group to |
| 515 // trigger an FEC packet. | 517 // trigger an FEC packet. |
| 516 ASSERT_FALSE(creator_.ShouldSendFec(/*force_close=*/false)); | 518 ASSERT_FALSE(creator_.ShouldSendFec(/*force_close=*/false)); |
| 517 // Should return true since there are packets in the FEC group. | 519 // Should return true since there are packets in the FEC group. |
| 518 ASSERT_TRUE(creator_.ShouldSendFec(/*force_close=*/true)); | 520 ASSERT_TRUE(creator_.ShouldSendFec(/*force_close=*/true)); |
| 519 | 521 |
| 520 // Force generation of FEC packet. | 522 // Force generation of FEC packet. |
| 521 EXPECT_CALL(delegate_, OnSerializedPacket(_)) | 523 EXPECT_CALL(delegate_, OnSerializedPacket(_)) |
| 522 .WillOnce(Invoke(this, &QuicPacketCreatorTest::SaveSerializedPacket)); | 524 .WillOnce(Invoke(this, &QuicPacketCreatorTest::SaveSerializedPacket)); |
| 523 // Turn off FEC protection. | 525 // Turn off FEC protection. |
| 524 creator_.MaybeSendFecPacketAndCloseGroup(/*force_send_fec=*/true, | 526 creator_.MaybeSendFecPacketAndCloseGroup(/*force_send_fec=*/true, |
| 525 /*is_fec_timeout=*/false); | 527 /*is_fec_timeout=*/false); |
| 526 EXPECT_EQ(PACKET_1BYTE_PACKET_NUMBER, | 528 EXPECT_EQ(PACKET_1BYTE_PACKET_NUMBER, |
| 527 serialized_packet_.packet_number_length); | 529 serialized_packet_.packet_number_length); |
| 528 ASSERT_EQ(3u, serialized_packet_.packet_number); | 530 ASSERT_EQ(3u, serialized_packet_.packet_number); |
| 529 | 531 |
| 530 { | 532 { |
| 531 InSequence s; | 533 InSequence s; |
| 532 EXPECT_CALL(framer_visitor_, OnPacket()); | 534 EXPECT_CALL(framer_visitor_, OnPacket()); |
| 533 EXPECT_CALL(framer_visitor_, OnUnauthenticatedPublicHeader(_)); | 535 EXPECT_CALL(framer_visitor_, OnUnauthenticatedPublicHeader(_)); |
| 534 EXPECT_CALL(framer_visitor_, OnUnauthenticatedHeader(_)); | 536 EXPECT_CALL(framer_visitor_, OnUnauthenticatedHeader(_)); |
| 535 EXPECT_CALL(framer_visitor_, OnDecryptedPacket(_)); | 537 EXPECT_CALL(framer_visitor_, OnDecryptedPacket(_)); |
| 536 EXPECT_CALL(framer_visitor_, OnPacketHeader(_)); | 538 EXPECT_CALL(framer_visitor_, OnPacketHeader(_)); |
| 537 EXPECT_CALL(framer_visitor_, OnFecData(_)); | 539 EXPECT_CALL(framer_visitor_, OnFecData(_)); |
| 538 EXPECT_CALL(framer_visitor_, OnPacketComplete()); | 540 EXPECT_CALL(framer_visitor_, OnPacketComplete()); |
| 539 } | 541 } |
| 540 ProcessPacket(serialized_packet_.packet); | 542 ProcessPacket(serialized_packet_); |
| 541 ClearSerializedPacket(&serialized_packet_); | 543 DeleteSerializedPacket(); |
| 542 | 544 |
| 543 // Ensure the next FEC group starts using the new packet number length. | 545 // Ensure the next FEC group starts using the new packet number length. |
| 544 ASSERT_TRUE(creator_.ConsumeData(3u, io_vector, 0u, 0u, false, false, &frame, | 546 ASSERT_TRUE(creator_.ConsumeData(3u, io_vector, 0u, 0u, false, false, &frame, |
| 545 MUST_FEC_PROTECT)); | 547 MUST_FEC_PROTECT)); |
| 546 EXPECT_TRUE(QuicPacketCreatorPeer::IsFecProtected(&creator_)); | 548 EXPECT_TRUE(QuicPacketCreatorPeer::IsFecProtected(&creator_)); |
| 547 EXPECT_CALL(delegate_, OnSerializedPacket(_)) | 549 EXPECT_CALL(delegate_, OnSerializedPacket(_)) |
| 548 .WillOnce(Invoke(this, &QuicPacketCreatorTest::SaveSerializedPacket)); | 550 .WillOnce(Invoke(this, &QuicPacketCreatorTest::SaveSerializedPacket)); |
| 549 creator_.Flush(); | 551 creator_.Flush(); |
| 550 EXPECT_EQ(PACKET_4BYTE_PACKET_NUMBER, | 552 EXPECT_EQ(PACKET_4BYTE_PACKET_NUMBER, |
| 551 serialized_packet_.packet_number_length); | 553 serialized_packet_.packet_number_length); |
| 552 ClearSerializedPacket(&serialized_packet_); | 554 DeleteSerializedPacket(); |
| 553 } | 555 } |
| 554 | 556 |
| 555 TEST_P(QuicPacketCreatorTest, ReserializeFramesWithSequenceNumberLength) { | 557 TEST_P(QuicPacketCreatorTest, ReserializeFramesWithSequenceNumberLength) { |
| 556 // If the original packet number length, the current packet number | 558 // If the original packet number length, the current packet number |
| 557 // length, and the configured send packet number length are different, the | 559 // length, and the configured send packet number length are different, the |
| 558 // retransmit must sent with the original length and the others do not change. | 560 // retransmit must sent with the original length and the others do not change. |
| 559 QuicPacketCreatorPeer::SetNextPacketNumberLength(&creator_, | 561 QuicPacketCreatorPeer::SetNextPacketNumberLength(&creator_, |
| 560 PACKET_4BYTE_PACKET_NUMBER); | 562 PACKET_4BYTE_PACKET_NUMBER); |
| 561 QuicPacketCreatorPeer::SetPacketNumberLength(&creator_, | 563 QuicPacketCreatorPeer::SetPacketNumberLength(&creator_, |
| 562 PACKET_2BYTE_PACKET_NUMBER); | 564 PACKET_2BYTE_PACKET_NUMBER); |
| (...skipping 23 matching lines...) Expand all Loading... |
| 586 { | 588 { |
| 587 InSequence s; | 589 InSequence s; |
| 588 EXPECT_CALL(framer_visitor_, OnPacket()); | 590 EXPECT_CALL(framer_visitor_, OnPacket()); |
| 589 EXPECT_CALL(framer_visitor_, OnUnauthenticatedPublicHeader(_)); | 591 EXPECT_CALL(framer_visitor_, OnUnauthenticatedPublicHeader(_)); |
| 590 EXPECT_CALL(framer_visitor_, OnUnauthenticatedHeader(_)); | 592 EXPECT_CALL(framer_visitor_, OnUnauthenticatedHeader(_)); |
| 591 EXPECT_CALL(framer_visitor_, OnDecryptedPacket(_)); | 593 EXPECT_CALL(framer_visitor_, OnDecryptedPacket(_)); |
| 592 EXPECT_CALL(framer_visitor_, OnPacketHeader(_)); | 594 EXPECT_CALL(framer_visitor_, OnPacketHeader(_)); |
| 593 EXPECT_CALL(framer_visitor_, OnStreamFrame(_)); | 595 EXPECT_CALL(framer_visitor_, OnStreamFrame(_)); |
| 594 EXPECT_CALL(framer_visitor_, OnPacketComplete()); | 596 EXPECT_CALL(framer_visitor_, OnPacketComplete()); |
| 595 } | 597 } |
| 596 ProcessPacket(serialized.packet); | 598 ProcessPacket(serialized); |
| 597 delete serialized.packet; | 599 delete serialized.packet; |
| 598 delete stream_frame; | 600 delete stream_frame; |
| 599 } | 601 } |
| 600 | 602 |
| 601 TEST_P(QuicPacketCreatorTest, ReserializeCryptoFrameWithForwardSecurity) { | 603 TEST_P(QuicPacketCreatorTest, ReserializeCryptoFrameWithForwardSecurity) { |
| 602 QuicStreamFrame* stream_frame = | 604 QuicStreamFrame* stream_frame = |
| 603 new QuicStreamFrame(kCryptoStreamId, /*fin=*/false, 0u, StringPiece()); | 605 new QuicStreamFrame(kCryptoStreamId, /*fin=*/false, 0u, StringPiece()); |
| 604 QuicFrames frames; | 606 QuicFrames frames; |
| 605 frames.push_back(QuicFrame(stream_frame)); | 607 frames.push_back(QuicFrame(stream_frame)); |
| 606 creator_.set_encryption_level(ENCRYPTION_FORWARD_SECURE); | 608 creator_.set_encryption_level(ENCRYPTION_FORWARD_SECURE); |
| (...skipping 128 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 735 | 737 |
| 736 InSequence s; | 738 InSequence s; |
| 737 EXPECT_CALL(framer_visitor_, OnPacket()); | 739 EXPECT_CALL(framer_visitor_, OnPacket()); |
| 738 EXPECT_CALL(framer_visitor_, OnUnauthenticatedPublicHeader(_)); | 740 EXPECT_CALL(framer_visitor_, OnUnauthenticatedPublicHeader(_)); |
| 739 EXPECT_CALL(framer_visitor_, OnUnauthenticatedHeader(_)); | 741 EXPECT_CALL(framer_visitor_, OnUnauthenticatedHeader(_)); |
| 740 EXPECT_CALL(framer_visitor_, OnDecryptedPacket(_)); | 742 EXPECT_CALL(framer_visitor_, OnDecryptedPacket(_)); |
| 741 EXPECT_CALL(framer_visitor_, OnPacketHeader(_)); | 743 EXPECT_CALL(framer_visitor_, OnPacketHeader(_)); |
| 742 EXPECT_CALL(framer_visitor_, OnConnectionCloseFrame(_)); | 744 EXPECT_CALL(framer_visitor_, OnConnectionCloseFrame(_)); |
| 743 EXPECT_CALL(framer_visitor_, OnPacketComplete()); | 745 EXPECT_CALL(framer_visitor_, OnPacketComplete()); |
| 744 | 746 |
| 745 ProcessPacket(serialized.packet); | 747 ProcessPacket(serialized); |
| 746 delete serialized.packet; | 748 delete serialized.packet; |
| 747 } | 749 } |
| 748 | 750 |
| 749 TEST_P(QuicPacketCreatorTest, SwitchFecOnOffWithGroupInProgress) { | 751 TEST_P(QuicPacketCreatorTest, SwitchFecOnOffWithGroupInProgress) { |
| 750 // Send FEC packet every 6 packets. | 752 // Send FEC packet every 6 packets. |
| 751 creator_.set_max_packets_per_fec_group(6); | 753 creator_.set_max_packets_per_fec_group(6); |
| 752 // Turn on FEC protection. | 754 // Turn on FEC protection. |
| 753 QuicFrame frame; | 755 QuicFrame frame; |
| 754 QuicIOVector io_vector(MakeIOVector("test")); | 756 QuicIOVector io_vector(MakeIOVector("test")); |
| 755 ASSERT_TRUE(creator_.ConsumeData(1u, io_vector, 0u, 0u, false, false, &frame, | 757 ASSERT_TRUE(creator_.ConsumeData(1u, io_vector, 0u, 0u, false, false, &frame, |
| (...skipping 150 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 906 | 908 |
| 907 // BytesFree() returns bytes available for the next frame, which will | 909 // BytesFree() returns bytes available for the next frame, which will |
| 908 // be two bytes smaller since the stream frame would need to be grown. | 910 // be two bytes smaller since the stream frame would need to be grown. |
| 909 EXPECT_EQ(2u, creator_.ExpansionOnNewFrame()); | 911 EXPECT_EQ(2u, creator_.ExpansionOnNewFrame()); |
| 910 size_t expected_bytes_free = bytes_free < 3 ? 0 : bytes_free - 2; | 912 size_t expected_bytes_free = bytes_free < 3 ? 0 : bytes_free - 2; |
| 911 EXPECT_EQ(expected_bytes_free, creator_.BytesFree()) << "delta: " << delta; | 913 EXPECT_EQ(expected_bytes_free, creator_.BytesFree()) << "delta: " << delta; |
| 912 EXPECT_CALL(delegate_, OnSerializedPacket(_)) | 914 EXPECT_CALL(delegate_, OnSerializedPacket(_)) |
| 913 .WillOnce(Invoke(this, &QuicPacketCreatorTest::SaveSerializedPacket)); | 915 .WillOnce(Invoke(this, &QuicPacketCreatorTest::SaveSerializedPacket)); |
| 914 creator_.Flush(); | 916 creator_.Flush(); |
| 915 ASSERT_TRUE(serialized_packet_.packet); | 917 ASSERT_TRUE(serialized_packet_.packet); |
| 916 ClearSerializedPacket(&serialized_packet_); | 918 DeleteSerializedPacket(); |
| 917 } | 919 } |
| 918 } | 920 } |
| 919 | 921 |
| 920 TEST_P(QuicPacketCreatorTest, StreamFrameConsumptionWithFec) { | 922 TEST_P(QuicPacketCreatorTest, StreamFrameConsumptionWithFec) { |
| 921 // Send FEC packet every 6 packets. | 923 // Send FEC packet every 6 packets. |
| 922 creator_.set_max_packets_per_fec_group(6); | 924 creator_.set_max_packets_per_fec_group(6); |
| 923 // Turn on FEC protection. | 925 // Turn on FEC protection. |
| 924 QuicFrame frame; | 926 QuicFrame frame; |
| 925 QuicIOVector io_vector(MakeIOVector("test")); | 927 QuicIOVector io_vector(MakeIOVector("test")); |
| 926 ASSERT_TRUE(creator_.ConsumeData(1u, io_vector, 0u, 0u, false, false, &frame, | 928 ASSERT_TRUE(creator_.ConsumeData(1u, io_vector, 0u, 0u, false, false, &frame, |
| (...skipping 31 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 958 .WillOnce(Invoke(this, &QuicPacketCreatorTest::SaveSerializedPacket)); | 960 .WillOnce(Invoke(this, &QuicPacketCreatorTest::SaveSerializedPacket)); |
| 959 // Every 6th packet will generate an extra FEC packet. | 961 // Every 6th packet will generate an extra FEC packet. |
| 960 if (delta == -1 || delta == 5) { | 962 if (delta == -1 || delta == 5) { |
| 961 EXPECT_CALL(delegate_, OnSerializedPacket(_)) | 963 EXPECT_CALL(delegate_, OnSerializedPacket(_)) |
| 962 .WillOnce( | 964 .WillOnce( |
| 963 Invoke(this, &QuicPacketCreatorTest::ClearSerializedPacket)); | 965 Invoke(this, &QuicPacketCreatorTest::ClearSerializedPacket)); |
| 964 } | 966 } |
| 965 } | 967 } |
| 966 creator_.Flush(); | 968 creator_.Flush(); |
| 967 ASSERT_TRUE(serialized_packet_.packet); | 969 ASSERT_TRUE(serialized_packet_.packet); |
| 968 ClearSerializedPacket(&serialized_packet_); | 970 DeleteSerializedPacket(); |
| 969 } | 971 } |
| 970 } | 972 } |
| 971 | 973 |
| 972 TEST_P(QuicPacketCreatorTest, CryptoStreamFramePacketPadding) { | 974 TEST_P(QuicPacketCreatorTest, CryptoStreamFramePacketPadding) { |
| 973 // Compute the total overhead for a single frame in packet. | 975 // Compute the total overhead for a single frame in packet. |
| 974 const size_t overhead = GetPacketHeaderOverhead(NOT_IN_FEC_GROUP) + | 976 const size_t overhead = GetPacketHeaderOverhead(NOT_IN_FEC_GROUP) + |
| 975 GetEncryptionOverhead() + | 977 GetEncryptionOverhead() + |
| 976 GetStreamFrameOverhead(NOT_IN_FEC_GROUP); | 978 GetStreamFrameOverhead(NOT_IN_FEC_GROUP); |
| 977 ASSERT_GT(kMaxPacketSize, overhead); | 979 ASSERT_GT(kMaxPacketSize, overhead); |
| 978 size_t capacity = kDefaultMaxPacketSize - overhead; | 980 size_t capacity = kDefaultMaxPacketSize - overhead; |
| (...skipping 16 matching lines...) Expand all Loading... |
| 995 ASSERT_TRUE(serialized_packet_.packet); | 997 ASSERT_TRUE(serialized_packet_.packet); |
| 996 // If there is not enough space in the packet to fit a padding frame | 998 // If there is not enough space in the packet to fit a padding frame |
| 997 // (1 byte) and to expand the stream frame (another 2 bytes) the packet | 999 // (1 byte) and to expand the stream frame (another 2 bytes) the packet |
| 998 // will not be padded. | 1000 // will not be padded. |
| 999 if (bytes_free < 3) { | 1001 if (bytes_free < 3) { |
| 1000 EXPECT_EQ(kDefaultMaxPacketSize - bytes_free, | 1002 EXPECT_EQ(kDefaultMaxPacketSize - bytes_free, |
| 1001 serialized_packet_.packet->length()); | 1003 serialized_packet_.packet->length()); |
| 1002 } else { | 1004 } else { |
| 1003 EXPECT_EQ(kDefaultMaxPacketSize, serialized_packet_.packet->length()); | 1005 EXPECT_EQ(kDefaultMaxPacketSize, serialized_packet_.packet->length()); |
| 1004 } | 1006 } |
| 1005 ClearSerializedPacket(&serialized_packet_); | 1007 DeleteSerializedPacket(); |
| 1006 } | 1008 } |
| 1007 } | 1009 } |
| 1008 | 1010 |
| 1009 TEST_P(QuicPacketCreatorTest, NonCryptoStreamFramePacketNonPadding) { | 1011 TEST_P(QuicPacketCreatorTest, NonCryptoStreamFramePacketNonPadding) { |
| 1010 // Compute the total overhead for a single frame in packet. | 1012 // Compute the total overhead for a single frame in packet. |
| 1011 const size_t overhead = GetPacketHeaderOverhead(NOT_IN_FEC_GROUP) + | 1013 const size_t overhead = GetPacketHeaderOverhead(NOT_IN_FEC_GROUP) + |
| 1012 GetEncryptionOverhead() + | 1014 GetEncryptionOverhead() + |
| 1013 GetStreamFrameOverhead(NOT_IN_FEC_GROUP); | 1015 GetStreamFrameOverhead(NOT_IN_FEC_GROUP); |
| 1014 ASSERT_GT(kDefaultMaxPacketSize, overhead); | 1016 ASSERT_GT(kDefaultMaxPacketSize, overhead); |
| 1015 size_t capacity = kDefaultMaxPacketSize - overhead; | 1017 size_t capacity = kDefaultMaxPacketSize - overhead; |
| (...skipping 13 matching lines...) Expand all Loading... |
| 1029 size_t bytes_consumed = frame.stream_frame->frame_length; | 1031 size_t bytes_consumed = frame.stream_frame->frame_length; |
| 1030 EXPECT_LT(0u, bytes_consumed); | 1032 EXPECT_LT(0u, bytes_consumed); |
| 1031 creator_.Flush(); | 1033 creator_.Flush(); |
| 1032 ASSERT_TRUE(serialized_packet_.packet); | 1034 ASSERT_TRUE(serialized_packet_.packet); |
| 1033 if (bytes_free > 0) { | 1035 if (bytes_free > 0) { |
| 1034 EXPECT_EQ(kDefaultMaxPacketSize - bytes_free, | 1036 EXPECT_EQ(kDefaultMaxPacketSize - bytes_free, |
| 1035 serialized_packet_.packet->length()); | 1037 serialized_packet_.packet->length()); |
| 1036 } else { | 1038 } else { |
| 1037 EXPECT_EQ(kDefaultMaxPacketSize, serialized_packet_.packet->length()); | 1039 EXPECT_EQ(kDefaultMaxPacketSize, serialized_packet_.packet->length()); |
| 1038 } | 1040 } |
| 1039 ClearSerializedPacket(&serialized_packet_); | 1041 DeleteSerializedPacket(); |
| 1040 } | 1042 } |
| 1041 } | 1043 } |
| 1042 | 1044 |
| 1043 TEST_P(QuicPacketCreatorTest, SerializeVersionNegotiationPacket) { | 1045 TEST_P(QuicPacketCreatorTest, SerializeVersionNegotiationPacket) { |
| 1044 QuicFramerPeer::SetPerspective(&client_framer_, Perspective::IS_SERVER); | 1046 QuicFramerPeer::SetPerspective(&client_framer_, Perspective::IS_SERVER); |
| 1045 QuicVersionVector versions; | 1047 QuicVersionVector versions; |
| 1046 versions.push_back(test::QuicVersionMax()); | 1048 versions.push_back(test::QuicVersionMax()); |
| 1047 scoped_ptr<QuicEncryptedPacket> encrypted( | 1049 scoped_ptr<QuicEncryptedPacket> encrypted( |
| 1048 creator_.SerializeVersionNegotiationPacket(versions)); | 1050 creator_.SerializeVersionNegotiationPacket(versions)); |
| 1049 | 1051 |
| (...skipping 76 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 1126 InSequence s; | 1128 InSequence s; |
| 1127 EXPECT_CALL(framer_visitor_, OnPacket()); | 1129 EXPECT_CALL(framer_visitor_, OnPacket()); |
| 1128 EXPECT_CALL(framer_visitor_, OnUnauthenticatedPublicHeader(_)); | 1130 EXPECT_CALL(framer_visitor_, OnUnauthenticatedPublicHeader(_)); |
| 1129 EXPECT_CALL(framer_visitor_, OnUnauthenticatedHeader(_)); | 1131 EXPECT_CALL(framer_visitor_, OnUnauthenticatedHeader(_)); |
| 1130 EXPECT_CALL(framer_visitor_, OnDecryptedPacket(_)); | 1132 EXPECT_CALL(framer_visitor_, OnDecryptedPacket(_)); |
| 1131 EXPECT_CALL(framer_visitor_, OnPacketHeader(_)) | 1133 EXPECT_CALL(framer_visitor_, OnPacketHeader(_)) |
| 1132 .WillOnce(DoAll(SaveArg<0>(&header), Return(true))); | 1134 .WillOnce(DoAll(SaveArg<0>(&header), Return(true))); |
| 1133 EXPECT_CALL(framer_visitor_, OnStreamFrame(_)); | 1135 EXPECT_CALL(framer_visitor_, OnStreamFrame(_)); |
| 1134 EXPECT_CALL(framer_visitor_, OnPacketComplete()); | 1136 EXPECT_CALL(framer_visitor_, OnPacketComplete()); |
| 1135 } | 1137 } |
| 1136 ProcessPacket(serialized.packet); | 1138 ProcessPacket(serialized); |
| 1137 EXPECT_EQ(GetParam().version_serialization, | 1139 EXPECT_EQ(GetParam().version_serialization, |
| 1138 header.public_header.version_flag); | 1140 header.public_header.version_flag); |
| 1139 delete serialized.packet; | 1141 delete serialized.packet; |
| 1140 } | 1142 } |
| 1141 | 1143 |
| 1142 TEST_P(QuicPacketCreatorTest, ConsumeDataLargerThanOneStreamFrame) { | 1144 TEST_P(QuicPacketCreatorTest, ConsumeDataLargerThanOneStreamFrame) { |
| 1143 if (!GetParam().version_serialization) { | 1145 if (!GetParam().version_serialization) { |
| 1144 creator_.StopSendingVersion(); | 1146 creator_.StopSendingVersion(); |
| 1145 } | 1147 } |
| 1146 // A string larger than fits into a frame. | 1148 // A string larger than fits into a frame. |
| (...skipping 10 matching lines...) Expand all Loading... |
| 1157 EXPECT_CALL(delegate_, OnSerializedPacket(_)) | 1159 EXPECT_CALL(delegate_, OnSerializedPacket(_)) |
| 1158 .WillOnce(Invoke(this, &QuicPacketCreatorTest::SaveSerializedPacket)); | 1160 .WillOnce(Invoke(this, &QuicPacketCreatorTest::SaveSerializedPacket)); |
| 1159 ASSERT_TRUE(creator_.ConsumeData(1u, io_vector, 0u, 0u, true, false, &frame, | 1161 ASSERT_TRUE(creator_.ConsumeData(1u, io_vector, 0u, 0u, true, false, &frame, |
| 1160 MAY_FEC_PROTECT)); | 1162 MAY_FEC_PROTECT)); |
| 1161 ASSERT_TRUE(frame.stream_frame); | 1163 ASSERT_TRUE(frame.stream_frame); |
| 1162 size_t consumed = frame.stream_frame->frame_length; | 1164 size_t consumed = frame.stream_frame->frame_length; |
| 1163 EXPECT_EQ(payload_length, consumed); | 1165 EXPECT_EQ(payload_length, consumed); |
| 1164 const string payload(payload_length, 'a'); | 1166 const string payload(payload_length, 'a'); |
| 1165 CheckStreamFrame(frame, 1u, payload, 0u, false); | 1167 CheckStreamFrame(frame, 1u, payload, 0u, false); |
| 1166 creator_.Flush(); | 1168 creator_.Flush(); |
| 1167 ClearSerializedPacket(&serialized_packet_); | 1169 DeleteSerializedPacket(); |
| 1168 } | 1170 } |
| 1169 | 1171 |
| 1170 TEST_P(QuicPacketCreatorTest, AddFrameAndFlush) { | 1172 TEST_P(QuicPacketCreatorTest, AddFrameAndFlush) { |
| 1171 if (!GetParam().version_serialization) { | 1173 if (!GetParam().version_serialization) { |
| 1172 creator_.StopSendingVersion(); | 1174 creator_.StopSendingVersion(); |
| 1173 } | 1175 } |
| 1174 const size_t max_plaintext_size = | 1176 const size_t max_plaintext_size = |
| 1175 client_framer_.GetMaxPlaintextSize(creator_.max_packet_length()); | 1177 client_framer_.GetMaxPlaintextSize(creator_.max_packet_length()); |
| 1176 EXPECT_FALSE(creator_.HasPendingFrames()); | 1178 EXPECT_FALSE(creator_.HasPendingFrames()); |
| 1177 EXPECT_EQ(max_plaintext_size - | 1179 EXPECT_EQ(max_plaintext_size - |
| (...skipping 28 matching lines...) Expand all Loading... |
| 1206 .WillOnce(Invoke(this, &QuicPacketCreatorTest::SaveSerializedPacket)); | 1208 .WillOnce(Invoke(this, &QuicPacketCreatorTest::SaveSerializedPacket)); |
| 1207 EXPECT_FALSE(creator_.AddSavedFrame(QuicFrame(&ack_frame))); | 1209 EXPECT_FALSE(creator_.AddSavedFrame(QuicFrame(&ack_frame))); |
| 1208 | 1210 |
| 1209 // Ensure the packet is successfully created. | 1211 // Ensure the packet is successfully created. |
| 1210 ASSERT_TRUE(serialized_packet_.packet); | 1212 ASSERT_TRUE(serialized_packet_.packet); |
| 1211 ASSERT_FALSE(serialized_packet_.retransmittable_frames.empty()); | 1213 ASSERT_FALSE(serialized_packet_.retransmittable_frames.empty()); |
| 1212 const QuicFrames& retransmittable = serialized_packet_.retransmittable_frames; | 1214 const QuicFrames& retransmittable = serialized_packet_.retransmittable_frames; |
| 1213 ASSERT_EQ(1u, retransmittable.size()); | 1215 ASSERT_EQ(1u, retransmittable.size()); |
| 1214 EXPECT_EQ(STREAM_FRAME, retransmittable[0].type); | 1216 EXPECT_EQ(STREAM_FRAME, retransmittable[0].type); |
| 1215 ASSERT_TRUE(retransmittable[0].stream_frame); | 1217 ASSERT_TRUE(retransmittable[0].stream_frame); |
| 1216 ClearSerializedPacket(&serialized_packet_); | 1218 DeleteSerializedPacket(); |
| 1217 | 1219 |
| 1218 EXPECT_FALSE(creator_.HasPendingFrames()); | 1220 EXPECT_FALSE(creator_.HasPendingFrames()); |
| 1219 EXPECT_EQ(max_plaintext_size - | 1221 EXPECT_EQ(max_plaintext_size - |
| 1220 GetPacketHeaderSize( | 1222 GetPacketHeaderSize( |
| 1221 creator_.connection_id_length(), | 1223 creator_.connection_id_length(), |
| 1222 QuicPacketCreatorPeer::SendVersionInPacket(&creator_), | 1224 QuicPacketCreatorPeer::SendVersionInPacket(&creator_), |
| 1223 /*include_path_id=*/false, PACKET_1BYTE_PACKET_NUMBER, | 1225 /*include_path_id=*/false, PACKET_1BYTE_PACKET_NUMBER, |
| 1224 NOT_IN_FEC_GROUP), | 1226 NOT_IN_FEC_GROUP), |
| 1225 creator_.BytesFree()); | 1227 creator_.BytesFree()); |
| 1226 } | 1228 } |
| (...skipping 32 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 1259 // Ensure the packet is successfully created, and the packet size estimate | 1261 // Ensure the packet is successfully created, and the packet size estimate |
| 1260 // matches the serialized packet length. | 1262 // matches the serialized packet length. |
| 1261 EXPECT_CALL(entropy_calculator_, EntropyHash(_)).WillOnce(testing::Return(0)); | 1263 EXPECT_CALL(entropy_calculator_, EntropyHash(_)).WillOnce(testing::Return(0)); |
| 1262 EXPECT_CALL(delegate_, OnSerializedPacket(_)) | 1264 EXPECT_CALL(delegate_, OnSerializedPacket(_)) |
| 1263 .WillOnce(Invoke(this, &QuicPacketCreatorTest::SaveSerializedPacket)); | 1265 .WillOnce(Invoke(this, &QuicPacketCreatorTest::SaveSerializedPacket)); |
| 1264 size_t est_packet_size = creator_.PacketSize(); | 1266 size_t est_packet_size = creator_.PacketSize(); |
| 1265 creator_.Flush(); | 1267 creator_.Flush(); |
| 1266 ASSERT_TRUE(serialized_packet_.packet); | 1268 ASSERT_TRUE(serialized_packet_.packet); |
| 1267 EXPECT_EQ(est_packet_size, client_framer_.GetMaxPlaintextSize( | 1269 EXPECT_EQ(est_packet_size, client_framer_.GetMaxPlaintextSize( |
| 1268 serialized_packet_.packet->length())); | 1270 serialized_packet_.packet->length())); |
| 1269 ClearSerializedPacket(&serialized_packet_); | 1271 DeleteSerializedPacket(); |
| 1270 } | 1272 } |
| 1271 | 1273 |
| 1272 TEST_P(QuicPacketCreatorTest, SerializeTruncatedAckFrameWithSmallPacketSize) { | 1274 TEST_P(QuicPacketCreatorTest, SerializeTruncatedAckFrameWithSmallPacketSize) { |
| 1273 if (!GetParam().version_serialization) { | 1275 if (!GetParam().version_serialization) { |
| 1274 creator_.StopSendingVersion(); | 1276 creator_.StopSendingVersion(); |
| 1275 } | 1277 } |
| 1276 creator_.SetMaxPacketLength(500u); | 1278 creator_.SetMaxPacketLength(500u); |
| 1277 | 1279 |
| 1278 const size_t max_plaintext_size = | 1280 const size_t max_plaintext_size = |
| 1279 client_framer_.GetMaxPlaintextSize(creator_.max_packet_length()); | 1281 client_framer_.GetMaxPlaintextSize(creator_.max_packet_length()); |
| (...skipping 17 matching lines...) Expand all Loading... |
| 1297 // Ensure the packet is successfully created, and the packet size estimate | 1299 // Ensure the packet is successfully created, and the packet size estimate |
| 1298 // may not match the serialized packet length. | 1300 // may not match the serialized packet length. |
| 1299 EXPECT_CALL(entropy_calculator_, EntropyHash(_)).WillOnce(Return(0)); | 1301 EXPECT_CALL(entropy_calculator_, EntropyHash(_)).WillOnce(Return(0)); |
| 1300 size_t est_packet_size = creator_.PacketSize(); | 1302 size_t est_packet_size = creator_.PacketSize(); |
| 1301 EXPECT_CALL(delegate_, OnSerializedPacket(_)) | 1303 EXPECT_CALL(delegate_, OnSerializedPacket(_)) |
| 1302 .WillOnce(Invoke(this, &QuicPacketCreatorTest::SaveSerializedPacket)); | 1304 .WillOnce(Invoke(this, &QuicPacketCreatorTest::SaveSerializedPacket)); |
| 1303 creator_.Flush(); | 1305 creator_.Flush(); |
| 1304 ASSERT_TRUE(serialized_packet_.packet); | 1306 ASSERT_TRUE(serialized_packet_.packet); |
| 1305 EXPECT_GE(est_packet_size, client_framer_.GetMaxPlaintextSize( | 1307 EXPECT_GE(est_packet_size, client_framer_.GetMaxPlaintextSize( |
| 1306 serialized_packet_.packet->length())); | 1308 serialized_packet_.packet->length())); |
| 1307 ClearSerializedPacket(&serialized_packet_); | 1309 DeleteSerializedPacket(); |
| 1308 } | 1310 } |
| 1309 | 1311 |
| 1310 TEST_P(QuicPacketCreatorTest, EntropyFlag) { | 1312 TEST_P(QuicPacketCreatorTest, EntropyFlag) { |
| 1311 frames_.push_back( | 1313 frames_.push_back( |
| 1312 QuicFrame(new QuicStreamFrame(0u, false, 0u, StringPiece()))); | 1314 QuicFrame(new QuicStreamFrame(0u, false, 0u, StringPiece()))); |
| 1313 | 1315 |
| 1314 for (int i = 0; i < 2; ++i) { | 1316 for (int i = 0; i < 2; ++i) { |
| 1315 for (int j = 0; j < 64; ++j) { | 1317 for (int j = 0; j < 64; ++j) { |
| 1316 SerializedPacket serialized = SerializeAllFrames(frames_); | 1318 SerializedPacket serialized = SerializeAllFrames(frames_); |
| 1317 // Verify both BoolSource and hash algorithm. | 1319 // Verify both BoolSource and hash algorithm. |
| (...skipping 71 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 1389 // Should return true since there are packets in the FEC group. | 1391 // Should return true since there are packets in the FEC group. |
| 1390 ASSERT_TRUE(creator_.ShouldSendFec(/*force_close=*/true)); | 1392 ASSERT_TRUE(creator_.ShouldSendFec(/*force_close=*/true)); |
| 1391 | 1393 |
| 1392 // Change FEC policy, send FEC packet and close FEC group. | 1394 // Change FEC policy, send FEC packet and close FEC group. |
| 1393 creator_.set_fec_send_policy(FEC_ANY_TRIGGER); | 1395 creator_.set_fec_send_policy(FEC_ANY_TRIGGER); |
| 1394 EXPECT_CALL(delegate_, OnSerializedPacket(_)) | 1396 EXPECT_CALL(delegate_, OnSerializedPacket(_)) |
| 1395 .WillOnce(Invoke(this, &QuicPacketCreatorTest::SaveSerializedPacket)); | 1397 .WillOnce(Invoke(this, &QuicPacketCreatorTest::SaveSerializedPacket)); |
| 1396 creator_.MaybeSendFecPacketAndCloseGroup(/*force_send_fec=*/true, | 1398 creator_.MaybeSendFecPacketAndCloseGroup(/*force_send_fec=*/true, |
| 1397 /*is_fec_timeout=*/false); | 1399 /*is_fec_timeout=*/false); |
| 1398 ASSERT_EQ(3u, serialized_packet_.packet_number); | 1400 ASSERT_EQ(3u, serialized_packet_.packet_number); |
| 1399 ClearSerializedPacket(&serialized_packet_); | 1401 DeleteSerializedPacket(); |
| 1400 } | 1402 } |
| 1401 | 1403 |
| 1402 TEST_P(QuicPacketCreatorTest, ResetFecGroupWithQueuedFrames) { | 1404 TEST_P(QuicPacketCreatorTest, ResetFecGroupWithQueuedFrames) { |
| 1403 // Send FEC packet every 6 packets. | 1405 // Send FEC packet every 6 packets. |
| 1404 creator_.set_max_packets_per_fec_group(6); | 1406 creator_.set_max_packets_per_fec_group(6); |
| 1405 // Add a stream frame to the creator and turn on FEC protection. | 1407 // Add a stream frame to the creator and turn on FEC protection. |
| 1406 QuicFrame frame; | 1408 QuicFrame frame; |
| 1407 QuicIOVector io_vector(MakeIOVector("test")); | 1409 QuicIOVector io_vector(MakeIOVector("test")); |
| 1408 ASSERT_TRUE(creator_.ConsumeData(1u, io_vector, 0u, 0u, false, false, &frame, | 1410 ASSERT_TRUE(creator_.ConsumeData(1u, io_vector, 0u, 0u, false, false, &frame, |
| 1409 MUST_FEC_PROTECT)); | 1411 MUST_FEC_PROTECT)); |
| (...skipping 198 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 1608 EXPECT_TRUE(creator_.HasPendingFrames()); | 1610 EXPECT_TRUE(creator_.HasPendingFrames()); |
| 1609 EXPECT_EQ(0u, creator_.packet_number()); | 1611 EXPECT_EQ(0u, creator_.packet_number()); |
| 1610 EXPECT_CALL(delegate_, OnSerializedPacket(_)) | 1612 EXPECT_CALL(delegate_, OnSerializedPacket(_)) |
| 1611 .WillRepeatedly( | 1613 .WillRepeatedly( |
| 1612 Invoke(this, &QuicPacketCreatorTest::SaveSerializedPacket)); | 1614 Invoke(this, &QuicPacketCreatorTest::SaveSerializedPacket)); |
| 1613 creator_.Flush(); | 1615 creator_.Flush(); |
| 1614 EXPECT_FALSE(creator_.HasPendingFrames()); | 1616 EXPECT_FALSE(creator_.HasPendingFrames()); |
| 1615 EXPECT_EQ(1u, creator_.packet_number()); | 1617 EXPECT_EQ(1u, creator_.packet_number()); |
| 1616 // Verify serialized data packet's path id. | 1618 // Verify serialized data packet's path id. |
| 1617 EXPECT_EQ(kDefaultPathId, serialized_packet_.path_id); | 1619 EXPECT_EQ(kDefaultPathId, serialized_packet_.path_id); |
| 1618 ClearSerializedPacket(&serialized_packet_); | 1620 DeleteSerializedPacket(); |
| 1619 | 1621 |
| 1620 // Change to path 1. | 1622 // Change to path 1. |
| 1621 QuicPathId kPathId1 = 1; | 1623 QuicPathId kPathId1 = 1; |
| 1622 creator_.SetCurrentPath(kPathId1, 1, 0); | 1624 creator_.SetCurrentPath(kPathId1, 1, 0); |
| 1623 EXPECT_EQ(kPathId1, QuicPacketCreatorPeer::GetCurrentPath(&creator_)); | 1625 EXPECT_EQ(kPathId1, QuicPacketCreatorPeer::GetCurrentPath(&creator_)); |
| 1624 EXPECT_FALSE(creator_.HasPendingFrames()); | 1626 EXPECT_FALSE(creator_.HasPendingFrames()); |
| 1625 EXPECT_EQ(0u, creator_.packet_number()); | 1627 EXPECT_EQ(0u, creator_.packet_number()); |
| 1626 EXPECT_EQ(PACKET_1BYTE_PACKET_NUMBER, | 1628 EXPECT_EQ(PACKET_1BYTE_PACKET_NUMBER, |
| 1627 QuicPacketCreatorPeer::NextPacketNumberLength(&creator_)); | 1629 QuicPacketCreatorPeer::NextPacketNumberLength(&creator_)); |
| 1628 | 1630 |
| 1629 // Add a stream frame to the creator and flush the packet. | 1631 // Add a stream frame to the creator and flush the packet. |
| 1630 ASSERT_TRUE(creator_.ConsumeData(1u, io_vector, 0u, 0u, false, false, &frame, | 1632 ASSERT_TRUE(creator_.ConsumeData(1u, io_vector, 0u, 0u, false, false, &frame, |
| 1631 MAY_FEC_PROTECT)); | 1633 MAY_FEC_PROTECT)); |
| 1632 ASSERT_TRUE(frame.stream_frame); | 1634 ASSERT_TRUE(frame.stream_frame); |
| 1633 consumed = frame.stream_frame->frame_length; | 1635 consumed = frame.stream_frame->frame_length; |
| 1634 EXPECT_EQ(4u, consumed); | 1636 EXPECT_EQ(4u, consumed); |
| 1635 EXPECT_TRUE(creator_.HasPendingFrames()); | 1637 EXPECT_TRUE(creator_.HasPendingFrames()); |
| 1636 creator_.Flush(); | 1638 creator_.Flush(); |
| 1637 // Verify serialized data packet's path id. | 1639 // Verify serialized data packet's path id. |
| 1638 EXPECT_EQ(kPathId1, serialized_packet_.path_id); | 1640 EXPECT_EQ(kPathId1, serialized_packet_.path_id); |
| 1639 ClearSerializedPacket(&serialized_packet_); | 1641 DeleteSerializedPacket(); |
| 1640 } | 1642 } |
| 1641 | 1643 |
| 1642 TEST_P(QuicPacketCreatorTest, AddUnencryptedStreamDataClosesConnection) { | 1644 TEST_P(QuicPacketCreatorTest, AddUnencryptedStreamDataClosesConnection) { |
| 1643 FLAGS_quic_never_write_unencrypted_data = true; | 1645 FLAGS_quic_never_write_unencrypted_data = true; |
| 1644 EXPECT_CALL(delegate_, CloseConnection(_, _)); | 1646 EXPECT_CALL(delegate_, CloseConnection(_, _)); |
| 1645 QuicStreamFrame stream_frame(kHeadersStreamId, /*fin=*/false, 0u, | 1647 QuicStreamFrame stream_frame(kHeadersStreamId, /*fin=*/false, 0u, |
| 1646 StringPiece()); | 1648 StringPiece()); |
| 1647 EXPECT_DFATAL(creator_.AddSavedFrame(QuicFrame(&stream_frame)), | 1649 EXPECT_DFATAL(creator_.AddSavedFrame(QuicFrame(&stream_frame)), |
| 1648 "Cannot send stream data without encryption."); | 1650 "Cannot send stream data without encryption."); |
| 1649 } | 1651 } |
| (...skipping 44 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 1694 EXPECT_CALL(delegate_, CloseConnection(QUIC_UNENCRYPTED_FEC_DATA, _)); | 1696 EXPECT_CALL(delegate_, CloseConnection(QUIC_UNENCRYPTED_FEC_DATA, _)); |
| 1695 char seralized_fec_buffer[kMaxPacketSize]; | 1697 char seralized_fec_buffer[kMaxPacketSize]; |
| 1696 EXPECT_DFATAL(QuicPacketCreatorPeer::SerializeFec( | 1698 EXPECT_DFATAL(QuicPacketCreatorPeer::SerializeFec( |
| 1697 &creator_, seralized_fec_buffer, kMaxPacketSize), | 1699 &creator_, seralized_fec_buffer, kMaxPacketSize), |
| 1698 "SerializeFEC must be called with encryption."); | 1700 "SerializeFEC must be called with encryption."); |
| 1699 } | 1701 } |
| 1700 | 1702 |
| 1701 } // namespace | 1703 } // namespace |
| 1702 } // namespace test | 1704 } // namespace test |
| 1703 } // namespace net | 1705 } // namespace net |
| OLD | NEW |