OLD | NEW |
1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 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_data_stream.h" | 5 #include "net/quic/quic_data_stream.h" |
6 | 6 |
7 #include "net/quic/quic_ack_notifier.h" | 7 #include "net/quic/quic_ack_notifier.h" |
8 #include "net/quic/quic_connection.h" | 8 #include "net/quic/quic_connection.h" |
9 #include "net/quic/quic_flags.h" | 9 #include "net/quic/quic_flags.h" |
10 #include "net/quic/quic_utils.h" | 10 #include "net/quic/quic_utils.h" |
(...skipping 267 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
278 ASSERT_EQ(data.data()[i + 1], buffer2[0]) << i; | 278 ASSERT_EQ(data.data()[i + 1], buffer2[0]) << i; |
279 } | 279 } |
280 } | 280 } |
281 | 281 |
282 TEST_P(QuicDataStreamTest, StreamFlowControlBlocked) { | 282 TEST_P(QuicDataStreamTest, StreamFlowControlBlocked) { |
283 // Tests that we send a BLOCKED frame to the peer when we attempt to write, | 283 // Tests that we send a BLOCKED frame to the peer when we attempt to write, |
284 // but are flow control blocked. | 284 // but are flow control blocked. |
285 if (GetParam() < QUIC_VERSION_17) { | 285 if (GetParam() < QUIC_VERSION_17) { |
286 return; | 286 return; |
287 } | 287 } |
288 ValueRestore<bool> old_flag(&FLAGS_enable_quic_stream_flow_control_2, true); | |
289 | 288 |
290 Initialize(kShouldProcessData); | 289 Initialize(kShouldProcessData); |
291 | 290 |
292 // Set a small flow control limit. | 291 // Set a small flow control limit. |
293 const uint64 kWindow = 36; | 292 const uint64 kWindow = 36; |
294 QuicFlowControllerPeer::SetSendWindowOffset(stream_->flow_controller(), | 293 QuicFlowControllerPeer::SetSendWindowOffset(stream_->flow_controller(), |
295 kWindow); | 294 kWindow); |
296 EXPECT_EQ(kWindow, QuicFlowControllerPeer::SendWindowOffset( | 295 EXPECT_EQ(kWindow, QuicFlowControllerPeer::SendWindowOffset( |
297 stream_->flow_controller())); | 296 stream_->flow_controller())); |
298 | 297 |
(...skipping 18 matching lines...) Expand all Loading... |
317 } | 316 } |
318 | 317 |
319 TEST_P(QuicDataStreamTest, StreamFlowControlNoWindowUpdateIfNotConsumed) { | 318 TEST_P(QuicDataStreamTest, StreamFlowControlNoWindowUpdateIfNotConsumed) { |
320 // The flow control receive window decreases whenever we add new bytes to the | 319 // The flow control receive window decreases whenever we add new bytes to the |
321 // sequencer, whether they are consumed immediately or buffered. However we | 320 // sequencer, whether they are consumed immediately or buffered. However we |
322 // only send WINDOW_UPDATE frames based on increasing number of bytes | 321 // only send WINDOW_UPDATE frames based on increasing number of bytes |
323 // consumed. | 322 // consumed. |
324 if (GetParam() < QUIC_VERSION_17) { | 323 if (GetParam() < QUIC_VERSION_17) { |
325 return; | 324 return; |
326 } | 325 } |
327 ValueRestore<bool> old_flag(&FLAGS_enable_quic_stream_flow_control_2, true); | |
328 | 326 |
329 // Don't process data - it will be buffered instead. | 327 // Don't process data - it will be buffered instead. |
330 Initialize(!kShouldProcessData); | 328 Initialize(!kShouldProcessData); |
331 | 329 |
332 // Expect no WINDOW_UPDATE frames to be sent. | 330 // Expect no WINDOW_UPDATE frames to be sent. |
333 EXPECT_CALL(*connection_, SendWindowUpdate(_, _)).Times(0); | 331 EXPECT_CALL(*connection_, SendWindowUpdate(_, _)).Times(0); |
334 | 332 |
335 // Set a small flow control receive window. | 333 // Set a small flow control receive window. |
336 const uint64 kWindow = 36; | 334 const uint64 kWindow = 36; |
337 QuicFlowControllerPeer::SetReceiveWindowOffset(stream_->flow_controller(), | 335 QuicFlowControllerPeer::SetReceiveWindowOffset(stream_->flow_controller(), |
(...skipping 27 matching lines...) Expand all Loading... |
365 QuicFlowControllerPeer::ReceiveWindowSize(stream_->flow_controller())); | 363 QuicFlowControllerPeer::ReceiveWindowSize(stream_->flow_controller())); |
366 } | 364 } |
367 | 365 |
368 TEST_P(QuicDataStreamTest, StreamFlowControlWindowUpdate) { | 366 TEST_P(QuicDataStreamTest, StreamFlowControlWindowUpdate) { |
369 // Tests that on receipt of data, the stream updates its receive window offset | 367 // Tests that on receipt of data, the stream updates its receive window offset |
370 // appropriately, and sends WINDOW_UPDATE frames when its receive window drops | 368 // appropriately, and sends WINDOW_UPDATE frames when its receive window drops |
371 // too low. | 369 // too low. |
372 if (GetParam() < QUIC_VERSION_17) { | 370 if (GetParam() < QUIC_VERSION_17) { |
373 return; | 371 return; |
374 } | 372 } |
375 ValueRestore<bool> old_flag(&FLAGS_enable_quic_stream_flow_control_2, true); | |
376 | 373 |
377 Initialize(kShouldProcessData); | 374 Initialize(kShouldProcessData); |
378 | 375 |
379 // Set a small flow control limit. | 376 // Set a small flow control limit. |
380 const uint64 kWindow = 36; | 377 const uint64 kWindow = 36; |
381 QuicFlowControllerPeer::SetReceiveWindowOffset(stream_->flow_controller(), | 378 QuicFlowControllerPeer::SetReceiveWindowOffset(stream_->flow_controller(), |
382 kWindow); | 379 kWindow); |
383 QuicFlowControllerPeer::SetMaxReceiveWindow(stream_->flow_controller(), | 380 QuicFlowControllerPeer::SetMaxReceiveWindow(stream_->flow_controller(), |
384 kWindow); | 381 kWindow); |
385 EXPECT_EQ(kWindow, QuicFlowControllerPeer::ReceiveWindowOffset( | 382 EXPECT_EQ(kWindow, QuicFlowControllerPeer::ReceiveWindowOffset( |
(...skipping 28 matching lines...) Expand all Loading... |
414 stream_->flow_controller())); | 411 stream_->flow_controller())); |
415 } | 412 } |
416 | 413 |
417 TEST_P(QuicDataStreamTest, ConnectionFlowControlWindowUpdate) { | 414 TEST_P(QuicDataStreamTest, ConnectionFlowControlWindowUpdate) { |
418 // Tests that on receipt of data, the connection updates its receive window | 415 // Tests that on receipt of data, the connection updates its receive window |
419 // offset appropriately, and sends WINDOW_UPDATE frames when its receive | 416 // offset appropriately, and sends WINDOW_UPDATE frames when its receive |
420 // window drops too low. | 417 // window drops too low. |
421 if (GetParam() < QUIC_VERSION_19) { | 418 if (GetParam() < QUIC_VERSION_19) { |
422 return; | 419 return; |
423 } | 420 } |
424 ValueRestore<bool> old_flag2(&FLAGS_enable_quic_stream_flow_control_2, true); | |
425 ValueRestore<bool> old_flag(&FLAGS_enable_quic_connection_flow_control_2, | 421 ValueRestore<bool> old_flag(&FLAGS_enable_quic_connection_flow_control_2, |
426 true); | 422 true); |
427 | 423 |
428 Initialize(kShouldProcessData); | 424 Initialize(kShouldProcessData); |
429 | 425 |
430 // Set a small flow control limit for streams and connection. | 426 // Set a small flow control limit for streams and connection. |
431 const uint64 kWindow = 36; | 427 const uint64 kWindow = 36; |
432 QuicFlowControllerPeer::SetReceiveWindowOffset(stream_->flow_controller(), | 428 QuicFlowControllerPeer::SetReceiveWindowOffset(stream_->flow_controller(), |
433 kWindow); | 429 kWindow); |
434 QuicFlowControllerPeer::SetMaxReceiveWindow(stream_->flow_controller(), | 430 QuicFlowControllerPeer::SetMaxReceiveWindow(stream_->flow_controller(), |
(...skipping 36 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
471 MakeIOVector("a")); | 467 MakeIOVector("a")); |
472 stream_->OnStreamFrame(frame3); | 468 stream_->OnStreamFrame(frame3); |
473 } | 469 } |
474 | 470 |
475 TEST_P(QuicDataStreamTest, StreamFlowControlViolation) { | 471 TEST_P(QuicDataStreamTest, StreamFlowControlViolation) { |
476 // Tests that on if the peer sends too much data (i.e. violates the flow | 472 // Tests that on if the peer sends too much data (i.e. violates the flow |
477 // control protocol), then we terminate the connection. | 473 // control protocol), then we terminate the connection. |
478 if (GetParam() < QUIC_VERSION_17) { | 474 if (GetParam() < QUIC_VERSION_17) { |
479 return; | 475 return; |
480 } | 476 } |
481 ValueRestore<bool> old_flag(&FLAGS_enable_quic_stream_flow_control_2, true); | |
482 | 477 |
483 // Stream should not process data, so that data gets buffered in the | 478 // Stream should not process data, so that data gets buffered in the |
484 // sequencer, triggering flow control limits. | 479 // sequencer, triggering flow control limits. |
485 Initialize(!kShouldProcessData); | 480 Initialize(!kShouldProcessData); |
486 | 481 |
487 // Set a small flow control limit. | 482 // Set a small flow control limit. |
488 const uint64 kWindow = 50; | 483 const uint64 kWindow = 50; |
489 QuicFlowControllerPeer::SetReceiveWindowOffset(stream_->flow_controller(), | 484 QuicFlowControllerPeer::SetReceiveWindowOffset(stream_->flow_controller(), |
490 kWindow); | 485 kWindow); |
491 | 486 |
(...skipping 11 matching lines...) Expand all Loading... |
503 stream_->OnStreamFrame(frame); | 498 stream_->OnStreamFrame(frame); |
504 } | 499 } |
505 | 500 |
506 TEST_P(QuicDataStreamTest, ConnectionFlowControlViolation) { | 501 TEST_P(QuicDataStreamTest, ConnectionFlowControlViolation) { |
507 // Tests that on if the peer sends too much data (i.e. violates the flow | 502 // Tests that on if the peer sends too much data (i.e. violates the flow |
508 // control protocol), at the connection level (rather than the stream level) | 503 // control protocol), at the connection level (rather than the stream level) |
509 // then we terminate the connection. | 504 // then we terminate the connection. |
510 if (GetParam() < QUIC_VERSION_19) { | 505 if (GetParam() < QUIC_VERSION_19) { |
511 return; | 506 return; |
512 } | 507 } |
513 ValueRestore<bool> old_flag2(&FLAGS_enable_quic_stream_flow_control_2, true); | |
514 ValueRestore<bool> old_flag(&FLAGS_enable_quic_connection_flow_control_2, | 508 ValueRestore<bool> old_flag(&FLAGS_enable_quic_connection_flow_control_2, |
515 true); | 509 true); |
516 | 510 |
517 // Stream should not process data, so that data gets buffered in the | 511 // Stream should not process data, so that data gets buffered in the |
518 // sequencer, triggering flow control limits. | 512 // sequencer, triggering flow control limits. |
519 Initialize(!kShouldProcessData); | 513 Initialize(!kShouldProcessData); |
520 | 514 |
521 // Set a small flow control window on streams, and connection. | 515 // Set a small flow control window on streams, and connection. |
522 const uint64 kStreamWindow = 50; | 516 const uint64 kStreamWindow = 50; |
523 const uint64 kConnectionWindow = 10; | 517 const uint64 kConnectionWindow = 10; |
(...skipping 17 matching lines...) Expand all Loading... |
541 SendConnectionClose(QUIC_FLOW_CONTROL_RECEIVED_TOO_MUCH_DATA)); | 535 SendConnectionClose(QUIC_FLOW_CONTROL_RECEIVED_TOO_MUCH_DATA)); |
542 stream_->OnStreamFrame(frame); | 536 stream_->OnStreamFrame(frame); |
543 } | 537 } |
544 | 538 |
545 TEST_P(QuicDataStreamTest, StreamFlowControlFinNotBlocked) { | 539 TEST_P(QuicDataStreamTest, StreamFlowControlFinNotBlocked) { |
546 // An attempt to write a FIN with no data should not be flow control blocked, | 540 // An attempt to write a FIN with no data should not be flow control blocked, |
547 // even if the send window is 0. | 541 // even if the send window is 0. |
548 if (GetParam() < QUIC_VERSION_17) { | 542 if (GetParam() < QUIC_VERSION_17) { |
549 return; | 543 return; |
550 } | 544 } |
551 ValueRestore<bool> old_flag(&FLAGS_enable_quic_stream_flow_control_2, true); | |
552 | 545 |
553 Initialize(kShouldProcessData); | 546 Initialize(kShouldProcessData); |
554 | 547 |
555 // Set a flow control limit of zero. | 548 // Set a flow control limit of zero. |
556 QuicFlowControllerPeer::SetReceiveWindowOffset(stream_->flow_controller(), 0); | 549 QuicFlowControllerPeer::SetReceiveWindowOffset(stream_->flow_controller(), 0); |
557 EXPECT_EQ(0u, QuicFlowControllerPeer::ReceiveWindowOffset( | 550 EXPECT_EQ(0u, QuicFlowControllerPeer::ReceiveWindowOffset( |
558 stream_->flow_controller())); | 551 stream_->flow_controller())); |
559 | 552 |
560 // Send a frame with a FIN but no data. This should not be blocked. | 553 // Send a frame with a FIN but no data. This should not be blocked. |
561 string body = ""; | 554 string body = ""; |
562 bool fin = true; | 555 bool fin = true; |
563 | 556 |
564 EXPECT_CALL(*connection_, SendBlocked(kClientDataStreamId1)).Times(0); | 557 EXPECT_CALL(*connection_, SendBlocked(kClientDataStreamId1)).Times(0); |
565 EXPECT_CALL(*session_, WritevData(kClientDataStreamId1, _, _, _, _, _)) | 558 EXPECT_CALL(*session_, WritevData(kClientDataStreamId1, _, _, _, _, _)) |
566 .WillOnce(Return(QuicConsumedData(0, fin))); | 559 .WillOnce(Return(QuicConsumedData(0, fin))); |
567 | 560 |
568 stream_->WriteOrBufferData(body, fin, NULL); | 561 stream_->WriteOrBufferData(body, fin, NULL); |
569 } | 562 } |
570 | 563 |
571 } // namespace | 564 } // namespace |
572 } // namespace test | 565 } // namespace test |
573 } // namespace net | 566 } // namespace net |
OLD | NEW |