James Robinson | 646469d | 2014-10-03 15:33:28 -0700 | [diff] [blame] | 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; |
James Robinson | d2015d9 | 2014-12-08 13:45:40 -0800 | [diff] [blame] | 21 | using std::string; |
James Robinson | 646469d | 2014-10-03 15:33:28 -0700 | [diff] [blame] | 22 | using testing::AnyNumber; |
| 23 | using testing::InSequence; |
| 24 | using testing::Return; |
| 25 | using testing::SaveArg; |
| 26 | using testing::StrictMock; |
| 27 | using testing::_; |
| 28 | |
| 29 | namespace net { |
| 30 | namespace test { |
| 31 | namespace { |
| 32 | |
| 33 | const bool kIsServer = true; |
| 34 | const bool kShouldProcessData = true; |
| 35 | |
| 36 | class TestStream : public QuicDataStream { |
| 37 | public: |
| 38 | TestStream(QuicStreamId id, |
| 39 | QuicSession* session, |
| 40 | bool should_process_data) |
| 41 | : QuicDataStream(id, session), |
| 42 | should_process_data_(should_process_data) {} |
| 43 | |
James Robinson | e1b30cf | 2014-10-21 12:25:40 -0700 | [diff] [blame] | 44 | uint32 ProcessData(const char* data, uint32 data_len) override { |
James Robinson | 646469d | 2014-10-03 15:33:28 -0700 | [diff] [blame] | 45 | EXPECT_NE(0u, data_len); |
| 46 | DVLOG(1) << "ProcessData data_len: " << data_len; |
| 47 | data_ += string(data, data_len); |
| 48 | return should_process_data_ ? data_len : 0; |
| 49 | } |
| 50 | |
| 51 | using ReliableQuicStream::WriteOrBufferData; |
| 52 | using ReliableQuicStream::CloseReadSide; |
| 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::TestWithParam<QuicVersion> { |
| 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 | kIsServer, SupportedVersions(GetParam())); |
| 97 | session_.reset(new testing::StrictMock<MockSession>(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 | MockConnection* connection_; |
| 108 | scoped_ptr<MockSession> session_; |
| 109 | scoped_ptr<TestStream> stream_; |
| 110 | scoped_ptr<TestStream> stream2_; |
| 111 | SpdyHeaderBlock headers_; |
| 112 | QuicWriteBlockedList* write_blocked_list_; |
| 113 | }; |
| 114 | |
| 115 | INSTANTIATE_TEST_CASE_P(Tests, QuicDataStreamTest, |
| 116 | ::testing::ValuesIn(QuicSupportedVersions())); |
| 117 | |
| 118 | TEST_P(QuicDataStreamTest, ProcessHeaders) { |
| 119 | Initialize(kShouldProcessData); |
| 120 | |
| 121 | string headers = SpdyUtils::SerializeUncompressedHeaders(headers_); |
| 122 | stream_->OnStreamHeadersPriority(QuicUtils::HighestPriority()); |
| 123 | stream_->OnStreamHeaders(headers); |
| 124 | EXPECT_EQ(headers, stream_->data()); |
| 125 | stream_->OnStreamHeadersComplete(false, headers.size()); |
| 126 | EXPECT_EQ(QuicUtils::HighestPriority(), stream_->EffectivePriority()); |
| 127 | EXPECT_EQ(headers, stream_->data()); |
| 128 | EXPECT_FALSE(stream_->IsDoneReading()); |
| 129 | } |
| 130 | |
| 131 | TEST_P(QuicDataStreamTest, ProcessHeadersAndBody) { |
| 132 | Initialize(kShouldProcessData); |
| 133 | |
| 134 | string headers = SpdyUtils::SerializeUncompressedHeaders(headers_); |
| 135 | string body = "this is the body"; |
| 136 | |
| 137 | stream_->OnStreamHeaders(headers); |
| 138 | EXPECT_EQ(headers, stream_->data()); |
| 139 | stream_->OnStreamHeadersComplete(false, headers.size()); |
| 140 | QuicStreamFrame frame(kClientDataStreamId1, false, 0, MakeIOVector(body)); |
| 141 | stream_->OnStreamFrame(frame); |
| 142 | |
| 143 | EXPECT_EQ(headers + body, stream_->data()); |
| 144 | } |
| 145 | |
| 146 | TEST_P(QuicDataStreamTest, ProcessHeadersAndBodyFragments) { |
| 147 | string headers = SpdyUtils::SerializeUncompressedHeaders(headers_); |
| 148 | string body = "this is the body"; |
| 149 | |
| 150 | for (size_t fragment_size = 1; fragment_size < body.size(); |
| 151 | ++fragment_size) { |
| 152 | Initialize(kShouldProcessData); |
| 153 | for (size_t offset = 0; offset < headers.size(); |
| 154 | offset += fragment_size) { |
| 155 | size_t remaining_data = headers.size() - offset; |
| 156 | StringPiece fragment(headers.data() + offset, |
| 157 | min(fragment_size, remaining_data)); |
| 158 | stream_->OnStreamHeaders(fragment); |
| 159 | } |
| 160 | stream_->OnStreamHeadersComplete(false, headers.size()); |
| 161 | for (size_t offset = 0; offset < body.size(); offset += fragment_size) { |
| 162 | size_t remaining_data = body.size() - offset; |
| 163 | StringPiece fragment(body.data() + offset, |
| 164 | min(fragment_size, remaining_data)); |
| 165 | QuicStreamFrame frame(kClientDataStreamId1, false, offset, |
| 166 | MakeIOVector(fragment)); |
| 167 | stream_->OnStreamFrame(frame); |
| 168 | } |
| 169 | ASSERT_EQ(headers + body, |
| 170 | stream_->data()) << "fragment_size: " << fragment_size; |
| 171 | } |
| 172 | } |
| 173 | |
| 174 | TEST_P(QuicDataStreamTest, ProcessHeadersAndBodyFragmentsSplit) { |
| 175 | string headers = SpdyUtils::SerializeUncompressedHeaders(headers_); |
| 176 | string body = "this is the body"; |
| 177 | |
| 178 | for (size_t split_point = 1; split_point < body.size() - 1; ++split_point) { |
| 179 | Initialize(kShouldProcessData); |
| 180 | StringPiece headers1(headers.data(), split_point); |
| 181 | stream_->OnStreamHeaders(headers1); |
| 182 | |
| 183 | StringPiece headers2(headers.data() + split_point, |
| 184 | headers.size() - split_point); |
| 185 | stream_->OnStreamHeaders(headers2); |
| 186 | stream_->OnStreamHeadersComplete(false, headers.size()); |
| 187 | |
| 188 | StringPiece fragment1(body.data(), split_point); |
| 189 | QuicStreamFrame frame1(kClientDataStreamId1, false, 0, |
| 190 | MakeIOVector(fragment1)); |
| 191 | stream_->OnStreamFrame(frame1); |
| 192 | |
| 193 | StringPiece fragment2(body.data() + split_point, |
| 194 | body.size() - split_point); |
| 195 | QuicStreamFrame frame2(kClientDataStreamId1, false, split_point, |
| 196 | MakeIOVector(fragment2)); |
| 197 | stream_->OnStreamFrame(frame2); |
| 198 | |
| 199 | ASSERT_EQ(headers + body, |
| 200 | stream_->data()) << "split_point: " << split_point; |
| 201 | } |
| 202 | } |
| 203 | |
| 204 | TEST_P(QuicDataStreamTest, ProcessHeadersAndBodyReadv) { |
| 205 | Initialize(!kShouldProcessData); |
| 206 | |
| 207 | string headers = SpdyUtils::SerializeUncompressedHeaders(headers_); |
| 208 | string body = "this is the body"; |
| 209 | |
| 210 | stream_->OnStreamHeaders(headers); |
| 211 | EXPECT_EQ(headers, stream_->data()); |
| 212 | stream_->OnStreamHeadersComplete(false, headers.size()); |
| 213 | QuicStreamFrame frame(kClientDataStreamId1, false, 0, MakeIOVector(body)); |
| 214 | stream_->OnStreamFrame(frame); |
| 215 | |
| 216 | char buffer[2048]; |
| 217 | ASSERT_LT(headers.length() + body.length(), arraysize(buffer)); |
| 218 | struct iovec vec; |
| 219 | vec.iov_base = buffer; |
| 220 | vec.iov_len = arraysize(buffer); |
| 221 | |
| 222 | size_t bytes_read = stream_->Readv(&vec, 1); |
| 223 | EXPECT_EQ(headers.length(), bytes_read); |
| 224 | EXPECT_EQ(headers, string(buffer, bytes_read)); |
| 225 | |
| 226 | bytes_read = stream_->Readv(&vec, 1); |
| 227 | EXPECT_EQ(body.length(), bytes_read); |
| 228 | EXPECT_EQ(body, string(buffer, bytes_read)); |
| 229 | } |
| 230 | |
| 231 | TEST_P(QuicDataStreamTest, ProcessHeadersAndBodyIncrementalReadv) { |
| 232 | Initialize(!kShouldProcessData); |
| 233 | |
| 234 | string headers = SpdyUtils::SerializeUncompressedHeaders(headers_); |
| 235 | string body = "this is the body"; |
| 236 | stream_->OnStreamHeaders(headers); |
| 237 | EXPECT_EQ(headers, stream_->data()); |
| 238 | stream_->OnStreamHeadersComplete(false, headers.size()); |
| 239 | QuicStreamFrame frame(kClientDataStreamId1, false, 0, MakeIOVector(body)); |
| 240 | stream_->OnStreamFrame(frame); |
| 241 | |
| 242 | char buffer[1]; |
| 243 | struct iovec vec; |
| 244 | vec.iov_base = buffer; |
| 245 | vec.iov_len = arraysize(buffer); |
| 246 | |
| 247 | string data = headers + body; |
| 248 | for (size_t i = 0; i < data.length(); ++i) { |
| 249 | size_t bytes_read = stream_->Readv(&vec, 1); |
| 250 | ASSERT_EQ(1u, bytes_read); |
| 251 | EXPECT_EQ(data.data()[i], buffer[0]); |
| 252 | } |
| 253 | } |
| 254 | |
| 255 | TEST_P(QuicDataStreamTest, ProcessHeadersUsingReadvWithMultipleIovecs) { |
| 256 | Initialize(!kShouldProcessData); |
| 257 | |
| 258 | string headers = SpdyUtils::SerializeUncompressedHeaders(headers_); |
| 259 | string body = "this is the body"; |
| 260 | stream_->OnStreamHeaders(headers); |
| 261 | EXPECT_EQ(headers, stream_->data()); |
| 262 | stream_->OnStreamHeadersComplete(false, headers.size()); |
| 263 | QuicStreamFrame frame(kClientDataStreamId1, false, 0, MakeIOVector(body)); |
| 264 | stream_->OnStreamFrame(frame); |
| 265 | |
| 266 | char buffer1[1]; |
| 267 | char buffer2[1]; |
| 268 | struct iovec vec[2]; |
| 269 | vec[0].iov_base = buffer1; |
| 270 | vec[0].iov_len = arraysize(buffer1); |
| 271 | vec[1].iov_base = buffer2; |
| 272 | vec[1].iov_len = arraysize(buffer2); |
| 273 | string data = headers + body; |
| 274 | for (size_t i = 0; i < data.length(); i += 2) { |
| 275 | size_t bytes_read = stream_->Readv(vec, 2); |
| 276 | ASSERT_EQ(2u, bytes_read) << i; |
| 277 | ASSERT_EQ(data.data()[i], buffer1[0]) << i; |
| 278 | ASSERT_EQ(data.data()[i + 1], buffer2[0]) << i; |
| 279 | } |
| 280 | } |
| 281 | |
| 282 | TEST_P(QuicDataStreamTest, StreamFlowControlBlocked) { |
| 283 | // Tests that we send a BLOCKED frame to the peer when we attempt to write, |
| 284 | // but are flow control blocked. |
| 285 | Initialize(kShouldProcessData); |
| 286 | |
| 287 | // Set a small flow control limit. |
| 288 | const uint64 kWindow = 36; |
| 289 | QuicFlowControllerPeer::SetSendWindowOffset(stream_->flow_controller(), |
| 290 | kWindow); |
| 291 | EXPECT_EQ(kWindow, QuicFlowControllerPeer::SendWindowOffset( |
| 292 | stream_->flow_controller())); |
| 293 | |
| 294 | // Try to send more data than the flow control limit allows. |
| 295 | string headers = SpdyUtils::SerializeUncompressedHeaders(headers_); |
| 296 | string body; |
| 297 | const uint64 kOverflow = 15; |
| 298 | GenerateBody(&body, kWindow + kOverflow); |
| 299 | |
| 300 | EXPECT_CALL(*connection_, SendBlocked(kClientDataStreamId1)); |
| 301 | EXPECT_CALL(*session_, WritevData(kClientDataStreamId1, _, _, _, _, _)) |
| 302 | .WillOnce(Return(QuicConsumedData(kWindow, true))); |
| 303 | stream_->WriteOrBufferData(body, false, nullptr); |
| 304 | |
| 305 | // Should have sent as much as possible, resulting in no send window left. |
| 306 | EXPECT_EQ(0u, |
| 307 | QuicFlowControllerPeer::SendWindowSize(stream_->flow_controller())); |
| 308 | |
| 309 | // And we should have queued the overflowed data. |
| 310 | EXPECT_EQ(kOverflow, |
| 311 | ReliableQuicStreamPeer::SizeOfQueuedData(stream_.get())); |
| 312 | } |
| 313 | |
| 314 | TEST_P(QuicDataStreamTest, StreamFlowControlNoWindowUpdateIfNotConsumed) { |
| 315 | // The flow control receive window decreases whenever we add new bytes to the |
| 316 | // sequencer, whether they are consumed immediately or buffered. However we |
| 317 | // only send WINDOW_UPDATE frames based on increasing number of bytes |
| 318 | // consumed. |
| 319 | |
| 320 | // Don't process data - it will be buffered instead. |
| 321 | Initialize(!kShouldProcessData); |
| 322 | |
| 323 | // Expect no WINDOW_UPDATE frames to be sent. |
| 324 | EXPECT_CALL(*connection_, SendWindowUpdate(_, _)).Times(0); |
| 325 | |
| 326 | // Set a small flow control receive window. |
| 327 | const uint64 kWindow = 36; |
| 328 | QuicFlowControllerPeer::SetReceiveWindowOffset(stream_->flow_controller(), |
| 329 | kWindow); |
| 330 | QuicFlowControllerPeer::SetMaxReceiveWindow(stream_->flow_controller(), |
| 331 | kWindow); |
| 332 | EXPECT_EQ(kWindow, QuicFlowControllerPeer::ReceiveWindowOffset( |
| 333 | stream_->flow_controller())); |
| 334 | |
| 335 | // Stream receives enough data to fill a fraction of the receive window. |
| 336 | string headers = SpdyUtils::SerializeUncompressedHeaders(headers_); |
| 337 | string body; |
| 338 | GenerateBody(&body, kWindow / 3); |
| 339 | stream_->OnStreamHeaders(headers); |
| 340 | EXPECT_EQ(headers, stream_->data()); |
| 341 | stream_->OnStreamHeadersComplete(false, headers.size()); |
| 342 | |
| 343 | QuicStreamFrame frame1(kClientDataStreamId1, false, 0, MakeIOVector(body)); |
| 344 | stream_->OnStreamFrame(frame1); |
| 345 | EXPECT_EQ(kWindow - (kWindow / 3), QuicFlowControllerPeer::ReceiveWindowSize( |
| 346 | stream_->flow_controller())); |
| 347 | |
| 348 | // Now receive another frame which results in the receive window being over |
| 349 | // half full. This should all be buffered, decreasing the receive window but |
| 350 | // not sending WINDOW_UPDATE. |
| 351 | QuicStreamFrame frame2(kClientDataStreamId1, false, kWindow / 3, |
| 352 | MakeIOVector(body)); |
| 353 | stream_->OnStreamFrame(frame2); |
| 354 | EXPECT_EQ( |
| 355 | kWindow - (2 * kWindow / 3), |
| 356 | QuicFlowControllerPeer::ReceiveWindowSize(stream_->flow_controller())); |
| 357 | } |
| 358 | |
| 359 | TEST_P(QuicDataStreamTest, StreamFlowControlWindowUpdate) { |
| 360 | // Tests that on receipt of data, the stream updates its receive window offset |
| 361 | // appropriately, and sends WINDOW_UPDATE frames when its receive window drops |
| 362 | // too low. |
| 363 | Initialize(kShouldProcessData); |
| 364 | |
| 365 | // Set a small flow control limit. |
| 366 | const uint64 kWindow = 36; |
| 367 | QuicFlowControllerPeer::SetReceiveWindowOffset(stream_->flow_controller(), |
| 368 | kWindow); |
| 369 | QuicFlowControllerPeer::SetMaxReceiveWindow(stream_->flow_controller(), |
| 370 | kWindow); |
| 371 | EXPECT_EQ(kWindow, QuicFlowControllerPeer::ReceiveWindowOffset( |
| 372 | stream_->flow_controller())); |
| 373 | |
| 374 | // Stream receives enough data to fill a fraction of the receive window. |
| 375 | string headers = SpdyUtils::SerializeUncompressedHeaders(headers_); |
| 376 | string body; |
| 377 | GenerateBody(&body, kWindow / 3); |
| 378 | stream_->OnStreamHeaders(headers); |
| 379 | EXPECT_EQ(headers, stream_->data()); |
| 380 | stream_->OnStreamHeadersComplete(false, headers.size()); |
| 381 | |
| 382 | QuicStreamFrame frame1(kClientDataStreamId1, false, 0, MakeIOVector(body)); |
| 383 | stream_->OnStreamFrame(frame1); |
| 384 | EXPECT_EQ(kWindow - (kWindow / 3), QuicFlowControllerPeer::ReceiveWindowSize( |
| 385 | stream_->flow_controller())); |
| 386 | |
| 387 | // Now receive another frame which results in the receive window being over |
| 388 | // half full. This will trigger the stream to increase its receive window |
| 389 | // offset and send a WINDOW_UPDATE. The result will be again an available |
| 390 | // window of kWindow bytes. |
| 391 | QuicStreamFrame frame2(kClientDataStreamId1, false, kWindow / 3, |
| 392 | MakeIOVector(body)); |
| 393 | EXPECT_CALL(*connection_, |
| 394 | SendWindowUpdate(kClientDataStreamId1, |
| 395 | QuicFlowControllerPeer::ReceiveWindowOffset( |
| 396 | stream_->flow_controller()) + |
| 397 | 2 * kWindow / 3)); |
| 398 | stream_->OnStreamFrame(frame2); |
| 399 | EXPECT_EQ(kWindow, QuicFlowControllerPeer::ReceiveWindowSize( |
| 400 | stream_->flow_controller())); |
| 401 | } |
| 402 | |
| 403 | TEST_P(QuicDataStreamTest, ConnectionFlowControlWindowUpdate) { |
| 404 | // Tests that on receipt of data, the connection updates its receive window |
| 405 | // offset appropriately, and sends WINDOW_UPDATE frames when its receive |
| 406 | // window drops too low. |
James Robinson | 646469d | 2014-10-03 15:33:28 -0700 | [diff] [blame] | 407 | Initialize(kShouldProcessData); |
| 408 | |
| 409 | // Set a small flow control limit for streams and connection. |
| 410 | const uint64 kWindow = 36; |
| 411 | QuicFlowControllerPeer::SetReceiveWindowOffset(stream_->flow_controller(), |
| 412 | kWindow); |
| 413 | QuicFlowControllerPeer::SetMaxReceiveWindow(stream_->flow_controller(), |
| 414 | kWindow); |
| 415 | QuicFlowControllerPeer::SetReceiveWindowOffset(stream2_->flow_controller(), |
| 416 | kWindow); |
| 417 | QuicFlowControllerPeer::SetMaxReceiveWindow(stream2_->flow_controller(), |
| 418 | kWindow); |
| 419 | QuicFlowControllerPeer::SetReceiveWindowOffset(session_->flow_controller(), |
| 420 | kWindow); |
| 421 | QuicFlowControllerPeer::SetMaxReceiveWindow(session_->flow_controller(), |
| 422 | kWindow); |
| 423 | |
| 424 | // Supply headers to both streams so that they are happy to receive data. |
| 425 | string headers = SpdyUtils::SerializeUncompressedHeaders(headers_); |
| 426 | stream_->OnStreamHeaders(headers); |
| 427 | stream_->OnStreamHeadersComplete(false, headers.size()); |
| 428 | stream2_->OnStreamHeaders(headers); |
| 429 | stream2_->OnStreamHeadersComplete(false, headers.size()); |
| 430 | |
| 431 | // Each stream gets a quarter window of data. This should not trigger a |
| 432 | // WINDOW_UPDATE for either stream, nor for the connection. |
| 433 | string body; |
| 434 | GenerateBody(&body, kWindow / 4); |
| 435 | QuicStreamFrame frame1(kClientDataStreamId1, false, 0, MakeIOVector(body)); |
| 436 | stream_->OnStreamFrame(frame1); |
| 437 | QuicStreamFrame frame2(kClientDataStreamId2, false, 0, MakeIOVector(body)); |
| 438 | stream2_->OnStreamFrame(frame2); |
| 439 | |
| 440 | // Now receive a further single byte on one stream - again this does not |
| 441 | // trigger a stream WINDOW_UPDATE, but now the connection flow control window |
| 442 | // is over half full and thus a connection WINDOW_UPDATE is sent. |
| 443 | EXPECT_CALL(*connection_, SendWindowUpdate(kClientDataStreamId1, _)).Times(0); |
| 444 | EXPECT_CALL(*connection_, SendWindowUpdate(kClientDataStreamId2, _)).Times(0); |
| 445 | EXPECT_CALL(*connection_, |
| 446 | SendWindowUpdate(0, QuicFlowControllerPeer::ReceiveWindowOffset( |
| 447 | session_->flow_controller()) + |
| 448 | 1 + kWindow / 2)); |
| 449 | QuicStreamFrame frame3(kClientDataStreamId1, false, (kWindow / 4), |
| 450 | MakeIOVector("a")); |
| 451 | stream_->OnStreamFrame(frame3); |
| 452 | } |
| 453 | |
| 454 | TEST_P(QuicDataStreamTest, StreamFlowControlViolation) { |
| 455 | // Tests that on if the peer sends too much data (i.e. violates the flow |
| 456 | // control protocol), then we terminate the connection. |
| 457 | |
| 458 | // Stream should not process data, so that data gets buffered in the |
| 459 | // sequencer, triggering flow control limits. |
| 460 | Initialize(!kShouldProcessData); |
| 461 | |
| 462 | // Set a small flow control limit. |
| 463 | const uint64 kWindow = 50; |
| 464 | QuicFlowControllerPeer::SetReceiveWindowOffset(stream_->flow_controller(), |
| 465 | kWindow); |
| 466 | |
| 467 | string headers = SpdyUtils::SerializeUncompressedHeaders(headers_); |
| 468 | stream_->OnStreamHeaders(headers); |
| 469 | EXPECT_EQ(headers, stream_->data()); |
| 470 | stream_->OnStreamHeadersComplete(false, headers.size()); |
| 471 | |
| 472 | // Receive data to overflow the window, violating flow control. |
| 473 | string body; |
| 474 | GenerateBody(&body, kWindow + 1); |
| 475 | QuicStreamFrame frame(kClientDataStreamId1, false, 0, MakeIOVector(body)); |
| 476 | EXPECT_CALL(*connection_, |
| 477 | SendConnectionClose(QUIC_FLOW_CONTROL_RECEIVED_TOO_MUCH_DATA)); |
| 478 | stream_->OnStreamFrame(frame); |
| 479 | } |
| 480 | |
| 481 | TEST_P(QuicDataStreamTest, ConnectionFlowControlViolation) { |
| 482 | // Tests that on if the peer sends too much data (i.e. violates the flow |
| 483 | // control protocol), at the connection level (rather than the stream level) |
| 484 | // then we terminate the connection. |
James Robinson | 646469d | 2014-10-03 15:33:28 -0700 | [diff] [blame] | 485 | |
| 486 | // Stream should not process data, so that data gets buffered in the |
| 487 | // sequencer, triggering flow control limits. |
| 488 | Initialize(!kShouldProcessData); |
| 489 | |
| 490 | // Set a small flow control window on streams, and connection. |
| 491 | const uint64 kStreamWindow = 50; |
| 492 | const uint64 kConnectionWindow = 10; |
| 493 | QuicFlowControllerPeer::SetReceiveWindowOffset(stream_->flow_controller(), |
| 494 | kStreamWindow); |
| 495 | QuicFlowControllerPeer::SetReceiveWindowOffset(session_->flow_controller(), |
| 496 | kConnectionWindow); |
| 497 | |
| 498 | string headers = SpdyUtils::SerializeUncompressedHeaders(headers_); |
| 499 | stream_->OnStreamHeaders(headers); |
| 500 | EXPECT_EQ(headers, stream_->data()); |
| 501 | stream_->OnStreamHeadersComplete(false, headers.size()); |
| 502 | |
| 503 | // Send enough data to overflow the connection level flow control window. |
| 504 | string body; |
| 505 | GenerateBody(&body, kConnectionWindow + 1); |
| 506 | EXPECT_LT(body.size(), kStreamWindow); |
| 507 | QuicStreamFrame frame(kClientDataStreamId1, false, 0, MakeIOVector(body)); |
| 508 | |
| 509 | EXPECT_CALL(*connection_, |
| 510 | SendConnectionClose(QUIC_FLOW_CONTROL_RECEIVED_TOO_MUCH_DATA)); |
| 511 | stream_->OnStreamFrame(frame); |
| 512 | } |
| 513 | |
| 514 | TEST_P(QuicDataStreamTest, StreamFlowControlFinNotBlocked) { |
| 515 | // An attempt to write a FIN with no data should not be flow control blocked, |
| 516 | // even if the send window is 0. |
| 517 | |
| 518 | Initialize(kShouldProcessData); |
| 519 | |
| 520 | // Set a flow control limit of zero. |
| 521 | QuicFlowControllerPeer::SetReceiveWindowOffset(stream_->flow_controller(), 0); |
| 522 | EXPECT_EQ(0u, QuicFlowControllerPeer::ReceiveWindowOffset( |
| 523 | stream_->flow_controller())); |
| 524 | |
| 525 | // Send a frame with a FIN but no data. This should not be blocked. |
| 526 | string body = ""; |
| 527 | bool fin = true; |
| 528 | |
| 529 | EXPECT_CALL(*connection_, SendBlocked(kClientDataStreamId1)).Times(0); |
| 530 | EXPECT_CALL(*session_, WritevData(kClientDataStreamId1, _, _, _, _, _)) |
| 531 | .WillOnce(Return(QuicConsumedData(0, fin))); |
| 532 | |
| 533 | stream_->WriteOrBufferData(body, fin, nullptr); |
| 534 | } |
| 535 | |
| 536 | } // namespace |
| 537 | } // namespace test |
| 538 | } // namespace net |