net/quic/quic_data_stream_test.cc

   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
   3 // found in the LICENSE file.
   4
   5 #include "net/quic/quic_data_stream.h"
   6
   7 #include "net/quic/quic_ack_notifier.h"
   8 #include "net/quic/quic_connection.h"
   9 #include "net/quic/quic_utils.h"
  10 #include "net/quic/quic_write_blocked_list.h"
  11 #include "net/quic/spdy_utils.h"
  12 #include "net/quic/test_tools/quic_flow_controller_peer.h"
  13 #include "net/quic/test_tools/quic_session_peer.h"
  14 #include "net/quic/test_tools/quic_test_utils.h"
  15 #include "net/quic/test_tools/reliable_quic_stream_peer.h"
  16 #include "net/test/gtest_util.h"
  17 #include "testing/gmock/include/gmock/gmock.h"
  18
  19 using base::StringPiece;
  20 using std::min;
  21 using testing::AnyNumber;
  22 using testing::InSequence;
  23 using testing::Return;
  24 using testing::SaveArg;
  25 using testing::StrictMock;
  26 using testing::_;
  27
  28 namespace net {
  29 namespace test {
  30 namespace {
  31
  32 const bool kIsServer = true;
  33 const bool kShouldProcessData = true;
  34
  35 class TestStream : public QuicDataStream {
  36  public:
  37   TestStream(QuicStreamId id,
  38              QuicSession* session,
  39              bool should_process_data)
  40       : QuicDataStream(id, session),
  41         should_process_data_(should_process_data) {}
  42
  43   virtual uint32 ProcessData(const char* data, uint32 data_len) OVERRIDE {
  44     EXPECT_NE(0u, data_len);
  45     DVLOG(1) << "ProcessData data_len: " << data_len;
  46     data_ += string(data, data_len);
  47     return should_process_data_ ? data_len : 0;
  48   }
  49
  50   using ReliableQuicStream::WriteOrBufferData;
  51   using ReliableQuicStream::CloseReadSide;
  52   using ReliableQuicStream::CloseWriteSide;
  53
  54   const string& data() const { return data_; }
  55
  56  private:
  57   bool should_process_data_;
  58   string data_;
  59 };
  60
  61 class QuicDataStreamTest : public ::testing::TestWithParam<QuicVersion> {
  62  public:
  63   QuicDataStreamTest() {
  64     headers_[":host"] = "www.google.com";
  65     headers_[":path"] = "/index.hml";
  66     headers_[":scheme"] = "https";
  67     headers_["cookie"] =
  68         "__utma=208381060.1228362404.1372200928.1372200928.1372200928.1; "
  69         "__utmc=160408618; "
  70         "GX=DQAAAOEAAACWJYdewdE9rIrW6qw3PtVi2-d729qaa-74KqOsM1NVQblK4VhX"
  71         "hoALMsy6HOdDad2Sz0flUByv7etmo3mLMidGrBoljqO9hSVA40SLqpG_iuKKSHX"
  72         "RW3Np4bq0F0SDGDNsW0DSmTS9ufMRrlpARJDS7qAI6M3bghqJp4eABKZiRqebHT"
  73         "pMU-RXvTI5D5oCF1vYxYofH_l1Kviuiy3oQ1kS1enqWgbhJ2t61_SNdv-1XJIS0"
  74         "O3YeHLmVCs62O6zp89QwakfAWK9d3IDQvVSJzCQsvxvNIvaZFa567MawWlXg0Rh"
  75         "1zFMi5vzcns38-8_Sns; "
  76         "GA=v*2%2Fmem*57968640*47239936%2Fmem*57968640*47114716%2Fno-nm-"
  77         "yj*15%2Fno-cc-yj*5%2Fpc-ch*133685%2Fpc-s-cr*133947%2Fpc-s-t*1339"
  78         "47%2Fno-nm-yj*4%2Fno-cc-yj*1%2Fceft-as*1%2Fceft-nqas*0%2Fad-ra-c"
  79         "v_p%2Fad-nr-cv_p-f*1%2Fad-v-cv_p*859%2Fad-ns-cv_p-f*1%2Ffn-v-ad%"
  80         "2Fpc-t*250%2Fpc-cm*461%2Fpc-s-cr*722%2Fpc-s-t*722%2Fau_p*4"
  81         "SICAID=AJKiYcHdKgxum7KMXG0ei2t1-W4OD1uW-ecNsCqC0wDuAXiDGIcT_HA2o1"
  82         "3Rs1UKCuBAF9g8rWNOFbxt8PSNSHFuIhOo2t6bJAVpCsMU5Laa6lewuTMYI8MzdQP"
  83         "ARHKyW-koxuhMZHUnGBJAM1gJODe0cATO_KGoX4pbbFxxJ5IicRxOrWK_5rU3cdy6"
  84         "edlR9FsEdH6iujMcHkbE5l18ehJDwTWmBKBzVD87naobhMMrF6VvnDGxQVGp9Ir_b"
  85         "Rgj3RWUoPumQVCxtSOBdX0GlJOEcDTNCzQIm9BSfetog_eP_TfYubKudt5eMsXmN6"
  86         "QnyXHeGeK2UINUzJ-D30AFcpqYgH9_1BvYSpi7fc7_ydBU8TaD8ZRxvtnzXqj0RfG"
  87         "tuHghmv3aD-uzSYJ75XDdzKdizZ86IG6Fbn1XFhYZM-fbHhm3mVEXnyRW4ZuNOLFk"
  88         "Fas6LMcVC6Q8QLlHYbXBpdNFuGbuZGUnav5C-2I_-46lL0NGg3GewxGKGHvHEfoyn"
  89         "EFFlEYHsBQ98rXImL8ySDycdLEFvBPdtctPmWCfTxwmoSMLHU2SCVDhbqMWU5b0yr"
  90         "JBCScs_ejbKaqBDoB7ZGxTvqlrB__2ZmnHHjCr8RgMRtKNtIeuZAo ";
  91   }
  92
  93   void Initialize(bool stream_should_process_data) {
  94     connection_ = new testing::StrictMock<MockConnection>(
  95         kIsServer, SupportedVersions(GetParam()));
  96     session_.reset(new testing::StrictMock<MockSession>(connection_));
  97     stream_.reset(new TestStream(kClientDataStreamId1, session_.get(),
  98                                  stream_should_process_data));
  99     stream2_.reset(new TestStream(kClientDataStreamId2, session_.get(),
 100                                   stream_should_process_data));
 101     write_blocked_list_ =
 102         QuicSessionPeer::GetWriteBlockedStreams(session_.get());
 103   }
 104
 105  protected:
 106   MockConnection* connection_;
 107   scoped_ptr<MockSession> session_;
 108   scoped_ptr<TestStream> stream_;
 109   scoped_ptr<TestStream> stream2_;
 110   SpdyHeaderBlock headers_;
 111   QuicWriteBlockedList* write_blocked_list_;
 112 };
 113
 114 INSTANTIATE_TEST_CASE_P(Tests, QuicDataStreamTest,
 115                         ::testing::ValuesIn(QuicSupportedVersions()));
 116
 117 TEST_P(QuicDataStreamTest, ProcessHeaders) {
 118   Initialize(kShouldProcessData);
 119
 120   string headers = SpdyUtils::SerializeUncompressedHeaders(headers_);
 121   stream_->OnStreamHeadersPriority(QuicUtils::HighestPriority());
 122   stream_->OnStreamHeaders(headers);
 123   EXPECT_EQ(headers, stream_->data());
 124   stream_->OnStreamHeadersComplete(false, headers.size());
 125   EXPECT_EQ(QuicUtils::HighestPriority(), stream_->EffectivePriority());
 126   EXPECT_EQ(headers, stream_->data());
 127   EXPECT_FALSE(stream_->IsDoneReading());
 128 }
 129
 130 TEST_P(QuicDataStreamTest, ProcessHeadersAndBody) {
 131   Initialize(kShouldProcessData);
 132
 133   string headers = SpdyUtils::SerializeUncompressedHeaders(headers_);
 134   string body = "this is the body";
 135
 136   stream_->OnStreamHeaders(headers);
 137   EXPECT_EQ(headers, stream_->data());
 138   stream_->OnStreamHeadersComplete(false, headers.size());
 139   QuicStreamFrame frame(kClientDataStreamId1, false, 0, MakeIOVector(body));
 140   stream_->OnStreamFrame(frame);
 141
 142   EXPECT_EQ(headers + body, stream_->data());
 143 }
 144
 145 TEST_P(QuicDataStreamTest, ProcessHeadersAndBodyFragments) {
 146   string headers = SpdyUtils::SerializeUncompressedHeaders(headers_);
 147   string body = "this is the body";
 148
 149   for (size_t fragment_size = 1; fragment_size < body.size();
 150        ++fragment_size) {
 151     Initialize(kShouldProcessData);
 152     for (size_t offset = 0; offset < headers.size();
 153          offset += fragment_size) {
 154       size_t remaining_data = headers.size() - offset;
 155       StringPiece fragment(headers.data() + offset,
 156                            min(fragment_size, remaining_data));
 157       stream_->OnStreamHeaders(fragment);
 158     }
 159     stream_->OnStreamHeadersComplete(false, headers.size());
 160     for (size_t offset = 0; offset < body.size(); offset += fragment_size) {
 161       size_t remaining_data = body.size() - offset;
 162       StringPiece fragment(body.data() + offset,
 163                            min(fragment_size, remaining_data));
 164       QuicStreamFrame frame(kClientDataStreamId1, false, offset,
 165                             MakeIOVector(fragment));
 166       stream_->OnStreamFrame(frame);
 167     }
 168     ASSERT_EQ(headers + body,
 169               stream_->data()) << "fragment_size: " << fragment_size;
 170   }
 171 }
 172
 173 TEST_P(QuicDataStreamTest, ProcessHeadersAndBodyFragmentsSplit) {
 174   string headers = SpdyUtils::SerializeUncompressedHeaders(headers_);
 175   string body = "this is the body";
 176
 177   for (size_t split_point = 1; split_point < body.size() - 1; ++split_point) {
 178     Initialize(kShouldProcessData);
 179     StringPiece headers1(headers.data(), split_point);
 180     stream_->OnStreamHeaders(headers1);
 181
 182     StringPiece headers2(headers.data() + split_point,
 183                          headers.size() - split_point);
 184     stream_->OnStreamHeaders(headers2);
 185     stream_->OnStreamHeadersComplete(false, headers.size());
 186
 187     StringPiece fragment1(body.data(), split_point);
 188     QuicStreamFrame frame1(kClientDataStreamId1, false, 0,
 189                            MakeIOVector(fragment1));
 190     stream_->OnStreamFrame(frame1);
 191
 192     StringPiece fragment2(body.data() + split_point,
 193                           body.size() - split_point);
 194     QuicStreamFrame frame2(kClientDataStreamId1, false, split_point,
 195                            MakeIOVector(fragment2));
 196     stream_->OnStreamFrame(frame2);
 197
 198     ASSERT_EQ(headers + body,
 199               stream_->data()) << "split_point: " << split_point;
 200   }
 201 }
 202
 203 TEST_P(QuicDataStreamTest, ProcessHeadersAndBodyReadv) {
 204   Initialize(!kShouldProcessData);
 205
 206   string headers = SpdyUtils::SerializeUncompressedHeaders(headers_);
 207   string body = "this is the body";
 208
 209   stream_->OnStreamHeaders(headers);
 210   EXPECT_EQ(headers, stream_->data());
 211   stream_->OnStreamHeadersComplete(false, headers.size());
 212   QuicStreamFrame frame(kClientDataStreamId1, false, 0, MakeIOVector(body));
 213   stream_->OnStreamFrame(frame);
 214
 215   char buffer[2048];
 216   ASSERT_LT(headers.length() + body.length(), arraysize(buffer));
 217   struct iovec vec;
 218   vec.iov_base = buffer;
 219   vec.iov_len = arraysize(buffer);
 220
 221   size_t bytes_read = stream_->Readv(&vec, 1);
 222   EXPECT_EQ(headers.length(), bytes_read);
 223   EXPECT_EQ(headers, string(buffer, bytes_read));
 224
 225   bytes_read = stream_->Readv(&vec, 1);
 226   EXPECT_EQ(body.length(), bytes_read);
 227   EXPECT_EQ(body, string(buffer, bytes_read));
 228 }
 229
 230 TEST_P(QuicDataStreamTest, ProcessHeadersAndBodyIncrementalReadv) {
 231   Initialize(!kShouldProcessData);
 232
 233   string headers = SpdyUtils::SerializeUncompressedHeaders(headers_);
 234   string body = "this is the body";
 235   stream_->OnStreamHeaders(headers);
 236   EXPECT_EQ(headers, stream_->data());
 237   stream_->OnStreamHeadersComplete(false, headers.size());
 238   QuicStreamFrame frame(kClientDataStreamId1, false, 0, MakeIOVector(body));
 239   stream_->OnStreamFrame(frame);
 240
 241   char buffer[1];
 242   struct iovec vec;
 243   vec.iov_base = buffer;
 244   vec.iov_len = arraysize(buffer);
 245
 246   string data = headers + body;
 247   for (size_t i = 0; i < data.length(); ++i) {
 248     size_t bytes_read = stream_->Readv(&vec, 1);
 249     ASSERT_EQ(1u, bytes_read);
 250     EXPECT_EQ(data.data()[i], buffer[0]);
 251   }
 252 }
 253
 254 TEST_P(QuicDataStreamTest, ProcessHeadersUsingReadvWithMultipleIovecs) {
 255   Initialize(!kShouldProcessData);
 256
 257   string headers = SpdyUtils::SerializeUncompressedHeaders(headers_);
 258   string body = "this is the body";
 259   stream_->OnStreamHeaders(headers);
 260   EXPECT_EQ(headers, stream_->data());
 261   stream_->OnStreamHeadersComplete(false, headers.size());
 262   QuicStreamFrame frame(kClientDataStreamId1, false, 0, MakeIOVector(body));
 263   stream_->OnStreamFrame(frame);
 264
 265   char buffer1[1];
 266   char buffer2[1];
 267   struct iovec vec[2];
 268   vec[0].iov_base = buffer1;
 269   vec[0].iov_len = arraysize(buffer1);
 270   vec[1].iov_base = buffer2;
 271   vec[1].iov_len = arraysize(buffer2);
 272   string data = headers + body;
 273   for (size_t i = 0; i < data.length(); i += 2) {
 274     size_t bytes_read = stream_->Readv(vec, 2);
 275     ASSERT_EQ(2u, bytes_read) << i;
 276     ASSERT_EQ(data.data()[i], buffer1[0]) << i;
 277     ASSERT_EQ(data.data()[i + 1], buffer2[0]) << i;
 278   }
 279 }
 280
 281 TEST_P(QuicDataStreamTest, StreamFlowControlBlocked) {
 282   // Tests that we send a BLOCKED frame to the peer when we attempt to write,
 283   // but are flow control blocked.
 284   if (GetParam() <= QUIC_VERSION_16) {
 285     return;
 286   }
 287
 288   Initialize(kShouldProcessData);
 289
 290   // Set a small flow control limit.
 291   const uint64 kWindow = 36;
 292   QuicFlowControllerPeer::SetSendWindowOffset(stream_->flow_controller(),
 293                                               kWindow);
 294   EXPECT_EQ(kWindow, QuicFlowControllerPeer::SendWindowOffset(
 295                          stream_->flow_controller()));
 296
 297   // Try to send more data than the flow control limit allows.
 298   string headers = SpdyUtils::SerializeUncompressedHeaders(headers_);
 299   string body;
 300   const uint64 kOverflow = 15;
 301   GenerateBody(&body, kWindow + kOverflow);
 302
 303   EXPECT_CALL(*connection_, SendBlocked(kClientDataStreamId1));
 304   EXPECT_CALL(*session_, WritevData(kClientDataStreamId1, _, _, _, _, _))
 305       .WillOnce(Return(QuicConsumedData(kWindow, true)));
 306   stream_->WriteOrBufferData(body, false, NULL);
 307
 308   // Should have sent as much as possible, resulting in no send window left.
 309   EXPECT_EQ(0u,
 310             QuicFlowControllerPeer::SendWindowSize(stream_->flow_controller()));
 311
 312   // And we should have queued the overflowed data.
 313   EXPECT_EQ(kOverflow,
 314             ReliableQuicStreamPeer::SizeOfQueuedData(stream_.get()));
 315 }
 316
 317 TEST_P(QuicDataStreamTest, StreamFlowControlNoWindowUpdateIfNotConsumed) {
 318   // The flow control receive window decreases whenever we add new bytes to the
 319   // sequencer, whether they are consumed immediately or buffered. However we
 320   // only send WINDOW_UPDATE frames based on increasing number of bytes
 321   // consumed.
 322   if (GetParam() <= QUIC_VERSION_16) {
 323     return;
 324   }
 325
 326   // Don't process data - it will be buffered instead.
 327   Initialize(!kShouldProcessData);
 328
 329   // Expect no WINDOW_UPDATE frames to be sent.
 330   EXPECT_CALL(*connection_, SendWindowUpdate(_, _)).Times(0);
 331
 332   // Set a small flow control receive window.
 333   const uint64 kWindow = 36;
 334   QuicFlowControllerPeer::SetReceiveWindowOffset(stream_->flow_controller(),
 335                                                  kWindow);
 336   QuicFlowControllerPeer::SetMaxReceiveWindow(stream_->flow_controller(),
 337                                               kWindow);
 338   EXPECT_EQ(kWindow, QuicFlowControllerPeer::ReceiveWindowOffset(
 339                          stream_->flow_controller()));
 340
 341   // Stream receives enough data to fill a fraction of the receive window.
 342   string headers = SpdyUtils::SerializeUncompressedHeaders(headers_);
 343   string body;
 344   GenerateBody(&body, kWindow / 3);
 345   stream_->OnStreamHeaders(headers);
 346   EXPECT_EQ(headers, stream_->data());
 347   stream_->OnStreamHeadersComplete(false, headers.size());
 348
 349   QuicStreamFrame frame1(kClientDataStreamId1, false, 0, MakeIOVector(body));
 350   stream_->OnStreamFrame(frame1);
 351   EXPECT_EQ(kWindow - (kWindow / 3), QuicFlowControllerPeer::ReceiveWindowSize(
 352                                          stream_->flow_controller()));
 353
 354   // Now receive another frame which results in the receive window being over
 355   // half full. This should all be buffered, decreasing the receive window but
 356   // not sending WINDOW_UPDATE.
 357   QuicStreamFrame frame2(kClientDataStreamId1, false, kWindow / 3,
 358                          MakeIOVector(body));
 359   stream_->OnStreamFrame(frame2);
 360   EXPECT_EQ(
 361       kWindow - (2 * kWindow / 3),
 362       QuicFlowControllerPeer::ReceiveWindowSize(stream_->flow_controller()));
 363 }
 364
 365 TEST_P(QuicDataStreamTest, StreamFlowControlWindowUpdate) {
 366   // Tests that on receipt of data, the stream updates its receive window offset
 367   // appropriately, and sends WINDOW_UPDATE frames when its receive window drops
 368   // too low.
 369   if (GetParam() <= QUIC_VERSION_16) {
 370     return;
 371   }
 372
 373   Initialize(kShouldProcessData);
 374
 375   // Set a small flow control limit.
 376   const uint64 kWindow = 36;
 377   QuicFlowControllerPeer::SetReceiveWindowOffset(stream_->flow_controller(),
 378                                                  kWindow);
 379   QuicFlowControllerPeer::SetMaxReceiveWindow(stream_->flow_controller(),
 380                                               kWindow);
 381   EXPECT_EQ(kWindow, QuicFlowControllerPeer::ReceiveWindowOffset(
 382                          stream_->flow_controller()));
 383
 384   // Stream receives enough data to fill a fraction of the receive window.
 385   string headers = SpdyUtils::SerializeUncompressedHeaders(headers_);
 386   string body;
 387   GenerateBody(&body, kWindow / 3);
 388   stream_->OnStreamHeaders(headers);
 389   EXPECT_EQ(headers, stream_->data());
 390   stream_->OnStreamHeadersComplete(false, headers.size());
 391
 392   QuicStreamFrame frame1(kClientDataStreamId1, false, 0, MakeIOVector(body));
 393   stream_->OnStreamFrame(frame1);
 394   EXPECT_EQ(kWindow - (kWindow / 3), QuicFlowControllerPeer::ReceiveWindowSize(
 395                                          stream_->flow_controller()));
 396
 397   // Now receive another frame which results in the receive window being over
 398   // half full.  This will trigger the stream to increase its receive window
 399   // offset and send a WINDOW_UPDATE. The result will be again an available
 400   // window of kWindow bytes.
 401   QuicStreamFrame frame2(kClientDataStreamId1, false, kWindow / 3,
 402                          MakeIOVector(body));
 403   EXPECT_CALL(*connection_,
 404               SendWindowUpdate(kClientDataStreamId1,
 405                                QuicFlowControllerPeer::ReceiveWindowOffset(
 406                                    stream_->flow_controller()) +
 407                                    2 * kWindow / 3));
 408   stream_->OnStreamFrame(frame2);
 409   EXPECT_EQ(kWindow, QuicFlowControllerPeer::ReceiveWindowSize(
 410                          stream_->flow_controller()));
 411 }
 412
 413 TEST_P(QuicDataStreamTest, ConnectionFlowControlWindowUpdate) {
 414   // Tests that on receipt of data, the connection updates its receive window
 415   // offset appropriately, and sends WINDOW_UPDATE frames when its receive
 416   // window drops too low.
 417   if (GetParam() < QUIC_VERSION_19) {
 418     return;
 419   }
 420   Initialize(kShouldProcessData);
 421
 422   // Set a small flow control limit for streams and connection.
 423   const uint64 kWindow = 36;
 424   QuicFlowControllerPeer::SetReceiveWindowOffset(stream_->flow_controller(),
 425                                                  kWindow);
 426   QuicFlowControllerPeer::SetMaxReceiveWindow(stream_->flow_controller(),
 427                                               kWindow);
 428   QuicFlowControllerPeer::SetReceiveWindowOffset(stream2_->flow_controller(),
 429                                                  kWindow);
 430   QuicFlowControllerPeer::SetMaxReceiveWindow(stream2_->flow_controller(),
 431                                               kWindow);
 432   QuicFlowControllerPeer::SetReceiveWindowOffset(session_->flow_controller(),
 433                                                  kWindow);
 434   QuicFlowControllerPeer::SetMaxReceiveWindow(session_->flow_controller(),
 435                                               kWindow);
 436
 437   // Supply headers to both streams so that they are happy to receive data.
 438   string headers = SpdyUtils::SerializeUncompressedHeaders(headers_);
 439   stream_->OnStreamHeaders(headers);
 440   stream_->OnStreamHeadersComplete(false, headers.size());
 441   stream2_->OnStreamHeaders(headers);
 442   stream2_->OnStreamHeadersComplete(false, headers.size());
 443
 444   // Each stream gets a quarter window of data. This should not trigger a
 445   // WINDOW_UPDATE for either stream, nor for the connection.
 446   string body;
 447   GenerateBody(&body, kWindow / 4);
 448   QuicStreamFrame frame1(kClientDataStreamId1, false, 0, MakeIOVector(body));
 449   stream_->OnStreamFrame(frame1);
 450   QuicStreamFrame frame2(kClientDataStreamId2, false, 0, MakeIOVector(body));
 451   stream2_->OnStreamFrame(frame2);
 452
 453   // Now receive a further single byte on one stream - again this does not
 454   // trigger a stream WINDOW_UPDATE, but now the connection flow control window
 455   // is over half full and thus a connection WINDOW_UPDATE is sent.
 456   EXPECT_CALL(*connection_, SendWindowUpdate(kClientDataStreamId1, _)).Times(0);
 457   EXPECT_CALL(*connection_, SendWindowUpdate(kClientDataStreamId2, _)).Times(0);
 458   EXPECT_CALL(*connection_,
 459               SendWindowUpdate(0, QuicFlowControllerPeer::ReceiveWindowOffset(
 460                                       session_->flow_controller()) +
 461                                       1 + kWindow / 2));
 462   QuicStreamFrame frame3(kClientDataStreamId1, false, (kWindow / 4),
 463                          MakeIOVector("a"));
 464   stream_->OnStreamFrame(frame3);
 465 }
 466
 467 TEST_P(QuicDataStreamTest, StreamFlowControlViolation) {
 468   // Tests that on if the peer sends too much data (i.e. violates the flow
 469   // control protocol), then we terminate the connection.
 470   if (GetParam() <= QUIC_VERSION_16) {
 471     return;
 472   }
 473
 474   // Stream should not process data, so that data gets buffered in the
 475   // sequencer, triggering flow control limits.
 476   Initialize(!kShouldProcessData);
 477
 478   // Set a small flow control limit.
 479   const uint64 kWindow = 50;
 480   QuicFlowControllerPeer::SetReceiveWindowOffset(stream_->flow_controller(),
 481                                                  kWindow);
 482
 483   string headers = SpdyUtils::SerializeUncompressedHeaders(headers_);
 484   stream_->OnStreamHeaders(headers);
 485   EXPECT_EQ(headers, stream_->data());
 486   stream_->OnStreamHeadersComplete(false, headers.size());
 487
 488   // Receive data to overflow the window, violating flow control.
 489   string body;
 490   GenerateBody(&body, kWindow + 1);
 491   QuicStreamFrame frame(kClientDataStreamId1, false, 0, MakeIOVector(body));
 492   EXPECT_CALL(*connection_,
 493               SendConnectionClose(QUIC_FLOW_CONTROL_RECEIVED_TOO_MUCH_DATA));
 494   stream_->OnStreamFrame(frame);
 495 }
 496
 497 TEST_P(QuicDataStreamTest, ConnectionFlowControlViolation) {
 498   // Tests that on if the peer sends too much data (i.e. violates the flow
 499   // control protocol), at the connection level (rather than the stream level)
 500   // then we terminate the connection.
 501   if (GetParam() < QUIC_VERSION_19) {
 502     return;
 503   }
 504
 505   // Stream should not process data, so that data gets buffered in the
 506   // sequencer, triggering flow control limits.
 507   Initialize(!kShouldProcessData);
 508
 509   // Set a small flow control window on streams, and connection.
 510   const uint64 kStreamWindow = 50;
 511   const uint64 kConnectionWindow = 10;
 512   QuicFlowControllerPeer::SetReceiveWindowOffset(stream_->flow_controller(),
 513                                                  kStreamWindow);
 514   QuicFlowControllerPeer::SetReceiveWindowOffset(session_->flow_controller(),
 515                                                  kConnectionWindow);
 516
 517   string headers = SpdyUtils::SerializeUncompressedHeaders(headers_);
 518   stream_->OnStreamHeaders(headers);
 519   EXPECT_EQ(headers, stream_->data());
 520   stream_->OnStreamHeadersComplete(false, headers.size());
 521
 522   // Send enough data to overflow the connection level flow control window.
 523   string body;
 524   GenerateBody(&body, kConnectionWindow + 1);
 525   EXPECT_LT(body.size(),  kStreamWindow);
 526   QuicStreamFrame frame(kClientDataStreamId1, false, 0, MakeIOVector(body));
 527
 528   EXPECT_CALL(*connection_,
 529               SendConnectionClose(QUIC_FLOW_CONTROL_RECEIVED_TOO_MUCH_DATA));
 530   stream_->OnStreamFrame(frame);
 531 }
 532
 533 TEST_P(QuicDataStreamTest, StreamFlowControlFinNotBlocked) {
 534   // An attempt to write a FIN with no data should not be flow control blocked,
 535   // even if the send window is 0.
 536   if (GetParam() <= QUIC_VERSION_16) {
 537     return;
 538   }
 539
 540   Initialize(kShouldProcessData);
 541
 542   // Set a flow control limit of zero.
 543   QuicFlowControllerPeer::SetReceiveWindowOffset(stream_->flow_controller(), 0);
 544   EXPECT_EQ(0u, QuicFlowControllerPeer::ReceiveWindowOffset(
 545                     stream_->flow_controller()));
 546
 547   // Send a frame with a FIN but no data. This should not be blocked.
 548   string body = "";
 549   bool fin = true;
 550
 551   EXPECT_CALL(*connection_, SendBlocked(kClientDataStreamId1)).Times(0);
 552   EXPECT_CALL(*session_, WritevData(kClientDataStreamId1, _, _, _, _, _))
 553       .WillOnce(Return(QuicConsumedData(0, fin)));
 554
 555   stream_->WriteOrBufferData(body, fin, NULL);
 556 }
 557
 558 }  // namespace
 559 }  // namespace test
 560 }  // namespace net