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