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_http_stream.h" | 5 #include "net/quic/quic_http_stream.h" |
6 | 6 |
7 #include <stdint.h> | 7 #include <stdint.h> |
8 | 8 |
9 #include <vector> | 9 #include <vector> |
10 | 10 |
(...skipping 121 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
132 IoMode mode; | 132 IoMode mode; |
133 QuicEncryptedPacket* packet; | 133 QuicEncryptedPacket* packet; |
134 int rv; | 134 int rv; |
135 }; | 135 }; |
136 | 136 |
137 QuicHttpStreamTest() | 137 QuicHttpStreamTest() |
138 : net_log_(BoundNetLog()), | 138 : net_log_(BoundNetLog()), |
139 use_closing_stream_(false), | 139 use_closing_stream_(false), |
140 crypto_config_(CryptoTestUtils::ProofVerifierForTesting()), | 140 crypto_config_(CryptoTestUtils::ProofVerifierForTesting()), |
141 read_buffer_(new IOBufferWithSize(4096)), | 141 read_buffer_(new IOBufferWithSize(4096)), |
| 142 promise_id_(kServerDataStreamId1), |
| 143 stream_id_(kClientDataStreamId1), |
142 connection_id_(2), | 144 connection_id_(2), |
143 stream_id_(kClientDataStreamId1), | |
144 maker_(GetParam(), connection_id_, &clock_, kDefaultServerHostName), | 145 maker_(GetParam(), connection_id_, &clock_, kDefaultServerHostName), |
145 random_generator_(0) { | 146 random_generator_(0), |
| 147 response_offset_(0) { |
146 IPAddress ip; | 148 IPAddress ip; |
147 CHECK(ip.AssignFromIPLiteral("192.0.2.33")); | 149 CHECK(ip.AssignFromIPLiteral("192.0.2.33")); |
148 peer_addr_ = IPEndPoint(ip, 443); | 150 peer_addr_ = IPEndPoint(ip, 443); |
149 self_addr_ = IPEndPoint(ip, 8435); | 151 self_addr_ = IPEndPoint(ip, 8435); |
150 clock_.AdvanceTime(QuicTime::Delta::FromMilliseconds(20)); | 152 clock_.AdvanceTime(QuicTime::Delta::FromMilliseconds(20)); |
151 } | 153 } |
152 | 154 |
153 ~QuicHttpStreamTest() { | 155 ~QuicHttpStreamTest() { |
154 session_->CloseSessionOnError(ERR_ABORTED, QUIC_INTERNAL_ERROR); | 156 session_->CloseSessionOnError(ERR_ABORTED, QUIC_INTERNAL_ERROR); |
155 for (size_t i = 0; i < writes_.size(); i++) { | 157 for (size_t i = 0; i < writes_.size(); i++) { |
(...skipping 85 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
241 /*cert_verify_flags=*/0, DefaultQuicConfig(), &crypto_config_, | 243 /*cert_verify_flags=*/0, DefaultQuicConfig(), &crypto_config_, |
242 "CONNECTION_UNKNOWN", base::TimeTicks::Now(), &push_promise_index_, | 244 "CONNECTION_UNKNOWN", base::TimeTicks::Now(), &push_promise_index_, |
243 base::ThreadTaskRunnerHandle::Get().get(), | 245 base::ThreadTaskRunnerHandle::Get().get(), |
244 /*socket_performance_watcher=*/nullptr, nullptr)); | 246 /*socket_performance_watcher=*/nullptr, nullptr)); |
245 session_->Initialize(); | 247 session_->Initialize(); |
246 session_->GetCryptoStream()->CryptoConnect(); | 248 session_->GetCryptoStream()->CryptoConnect(); |
247 EXPECT_TRUE(session_->IsCryptoHandshakeConfirmed()); | 249 EXPECT_TRUE(session_->IsCryptoHandshakeConfirmed()); |
248 stream_.reset(use_closing_stream_ | 250 stream_.reset(use_closing_stream_ |
249 ? new AutoClosingStream(session_->GetWeakPtr()) | 251 ? new AutoClosingStream(session_->GetWeakPtr()) |
250 : new QuicHttpStream(session_->GetWeakPtr())); | 252 : new QuicHttpStream(session_->GetWeakPtr())); |
| 253 |
| 254 promised_stream_.reset(use_closing_stream_ |
| 255 ? new AutoClosingStream(session_->GetWeakPtr()) |
| 256 : new QuicHttpStream(session_->GetWeakPtr())); |
| 257 |
| 258 push_promise_[":path"] = "/bar"; |
| 259 push_promise_[":authority"] = "www.example.org"; |
| 260 push_promise_[":version"] = "HTTP/1.1"; |
| 261 push_promise_[":method"] = "GET"; |
| 262 push_promise_[":scheme"] = "https"; |
| 263 |
| 264 promised_response_[":status"] = "200 OK"; |
| 265 promised_response_[":version"] = "HTTP/1.1"; |
| 266 promised_response_["content-type"] = "text/plain"; |
| 267 |
| 268 promise_url_ = SpdyUtils::GetUrlFromHeaderBlock(push_promise_); |
| 269 |
| 270 serialized_push_promise_ = |
| 271 SpdyUtils::SerializeUncompressedHeaders(push_promise_); |
251 } | 272 } |
252 | 273 |
253 void SetRequest(const std::string& method, | 274 void SetRequest(const std::string& method, |
254 const std::string& path, | 275 const std::string& path, |
255 RequestPriority priority) { | 276 RequestPriority priority) { |
256 request_headers_ = maker_.GetRequestHeaders(method, "http", path); | 277 request_headers_ = maker_.GetRequestHeaders(method, "http", path); |
257 } | 278 } |
258 | 279 |
259 void SetResponse(const std::string& status, const std::string& body) { | 280 void SetResponse(const std::string& status, const std::string& body) { |
260 response_headers_ = maker_.GetResponseHeaders(status); | 281 response_headers_ = maker_.GetResponseHeaders(status); |
261 response_data_ = body; | 282 response_data_ = body; |
262 } | 283 } |
263 | 284 |
| 285 scoped_ptr<QuicEncryptedPacket> InnerConstructDataPacket( |
| 286 QuicPacketNumber packet_number, |
| 287 QuicStreamId stream_id, |
| 288 bool should_include_version, |
| 289 bool fin, |
| 290 QuicStreamOffset offset, |
| 291 base::StringPiece data) { |
| 292 return maker_.MakeDataPacket(packet_number, stream_id, |
| 293 should_include_version, fin, offset, data); |
| 294 } |
| 295 |
264 scoped_ptr<QuicEncryptedPacket> ConstructDataPacket( | 296 scoped_ptr<QuicEncryptedPacket> ConstructDataPacket( |
265 QuicPacketNumber packet_number, | 297 QuicPacketNumber packet_number, |
266 bool should_include_version, | 298 bool should_include_version, |
267 bool fin, | 299 bool fin, |
268 QuicStreamOffset offset, | 300 QuicStreamOffset offset, |
269 base::StringPiece data) { | 301 base::StringPiece data) { |
270 return maker_.MakeDataPacket(packet_number, stream_id_, | 302 return InnerConstructDataPacket(packet_number, stream_id_, |
271 should_include_version, fin, offset, data); | 303 should_include_version, fin, offset, data); |
| 304 } |
| 305 |
| 306 scoped_ptr<QuicEncryptedPacket> InnerConstructRequestHeadersPacket( |
| 307 QuicPacketNumber packet_number, |
| 308 QuicStreamId stream_id, |
| 309 bool should_include_version, |
| 310 bool fin, |
| 311 RequestPriority request_priority, |
| 312 size_t* spdy_headers_frame_length) { |
| 313 SpdyPriority priority = |
| 314 ConvertRequestPriorityToQuicPriority(request_priority); |
| 315 return maker_.MakeRequestHeadersPacket( |
| 316 packet_number, stream_id, should_include_version, fin, priority, |
| 317 request_headers_, spdy_headers_frame_length); |
272 } | 318 } |
273 | 319 |
274 scoped_ptr<QuicEncryptedPacket> ConstructRequestHeadersPacket( | 320 scoped_ptr<QuicEncryptedPacket> ConstructRequestHeadersPacket( |
275 QuicPacketNumber packet_number, | 321 QuicPacketNumber packet_number, |
276 bool fin, | 322 bool fin, |
277 RequestPriority request_priority, | 323 RequestPriority request_priority, |
278 size_t* spdy_headers_frame_length) { | 324 size_t* spdy_headers_frame_length) { |
279 SpdyPriority priority = | 325 return InnerConstructRequestHeadersPacket( |
280 ConvertRequestPriorityToQuicPriority(request_priority); | 326 packet_number, stream_id_, kIncludeVersion, fin, request_priority, |
281 return maker_.MakeRequestHeadersPacket( | 327 spdy_headers_frame_length); |
282 packet_number, stream_id_, kIncludeVersion, fin, priority, | 328 } |
283 request_headers_, spdy_headers_frame_length); | 329 |
| 330 scoped_ptr<QuicEncryptedPacket> InnerConstructResponseHeadersPacket( |
| 331 QuicPacketNumber packet_number, |
| 332 QuicStreamId stream_id, |
| 333 bool fin, |
| 334 size_t* spdy_headers_frame_length) { |
| 335 return maker_.MakeResponseHeadersPacket( |
| 336 packet_number, stream_id, !kIncludeVersion, fin, response_headers_, |
| 337 spdy_headers_frame_length, &response_offset_); |
284 } | 338 } |
285 | 339 |
286 scoped_ptr<QuicEncryptedPacket> ConstructResponseHeadersPacket( | 340 scoped_ptr<QuicEncryptedPacket> ConstructResponseHeadersPacket( |
287 QuicPacketNumber packet_number, | 341 QuicPacketNumber packet_number, |
288 bool fin, | 342 bool fin, |
289 size_t* spdy_headers_frame_length) { | 343 size_t* spdy_headers_frame_length) { |
290 return maker_.MakeResponseHeadersPacket( | 344 return InnerConstructResponseHeadersPacket(packet_number, stream_id_, fin, |
291 packet_number, stream_id_, !kIncludeVersion, fin, response_headers_, | 345 spdy_headers_frame_length); |
292 spdy_headers_frame_length); | |
293 } | 346 } |
294 | 347 |
295 scoped_ptr<QuicEncryptedPacket> ConstructRstStreamPacket( | 348 scoped_ptr<QuicEncryptedPacket> ConstructRstStreamPacket( |
296 QuicPacketNumber packet_number) { | 349 QuicPacketNumber packet_number) { |
297 return maker_.MakeRstPacket( | 350 return maker_.MakeRstPacket( |
298 packet_number, true, stream_id_, | 351 packet_number, true, stream_id_, |
299 AdjustErrorForVersion(QUIC_RST_ACKNOWLEDGEMENT, GetParam())); | 352 AdjustErrorForVersion(QUIC_RST_ACKNOWLEDGEMENT, GetParam())); |
300 } | 353 } |
301 | 354 |
302 scoped_ptr<QuicEncryptedPacket> ConstructRstStreamCancelledPacket( | 355 scoped_ptr<QuicEncryptedPacket> ConstructRstStreamCancelledPacket( |
303 QuicPacketNumber packet_number) { | 356 QuicPacketNumber packet_number) { |
304 return maker_.MakeRstPacket(packet_number, !kIncludeVersion, stream_id_, | 357 return maker_.MakeRstPacket(packet_number, !kIncludeVersion, stream_id_, |
305 QUIC_STREAM_CANCELLED); | 358 QUIC_STREAM_CANCELLED); |
306 } | 359 } |
307 | 360 |
| 361 scoped_ptr<QuicEncryptedPacket> ConstructRstStreamVaryMismatchPacket( |
| 362 QuicPacketNumber packet_number) { |
| 363 return maker_.MakeRstPacket(packet_number, !kIncludeVersion, promise_id_, |
| 364 QUIC_PROMISE_VARY_MISMATCH); |
| 365 } |
| 366 |
308 scoped_ptr<QuicEncryptedPacket> ConstructAckAndRstStreamPacket( | 367 scoped_ptr<QuicEncryptedPacket> ConstructAckAndRstStreamPacket( |
309 QuicPacketNumber packet_number) { | 368 QuicPacketNumber packet_number) { |
310 return maker_.MakeAckAndRstPacket(packet_number, !kIncludeVersion, | 369 return maker_.MakeAckAndRstPacket(packet_number, !kIncludeVersion, |
311 stream_id_, QUIC_STREAM_CANCELLED, 2, 1, | 370 stream_id_, QUIC_STREAM_CANCELLED, 2, 1, |
312 !kIncludeCongestionFeedback); | 371 !kIncludeCongestionFeedback); |
313 } | 372 } |
314 | 373 |
315 scoped_ptr<QuicEncryptedPacket> ConstructAckPacket( | 374 scoped_ptr<QuicEncryptedPacket> ConstructAckPacket( |
316 QuicPacketNumber packet_number, | 375 QuicPacketNumber packet_number, |
317 QuicPacketNumber largest_received, | 376 QuicPacketNumber largest_received, |
318 QuicPacketNumber least_unacked) { | 377 QuicPacketNumber least_unacked) { |
319 return maker_.MakeAckPacket(packet_number, largest_received, least_unacked, | 378 return maker_.MakeAckPacket(packet_number, largest_received, least_unacked, |
320 !kIncludeCongestionFeedback); | 379 !kIncludeCongestionFeedback); |
321 } | 380 } |
322 | 381 |
| 382 void ReceivePromise(QuicStreamId id) { |
| 383 QuicChromiumClientStream* stream = |
| 384 QuicHttpStreamPeer::GetQuicChromiumClientStream(stream_.get()); |
| 385 stream->OnStreamHeaders(serialized_push_promise_); |
| 386 |
| 387 stream->OnPromiseHeadersComplete(id, serialized_push_promise_.size()); |
| 388 } |
| 389 |
323 BoundNetLog net_log_; | 390 BoundNetLog net_log_; |
324 bool use_closing_stream_; | 391 bool use_closing_stream_; |
325 MockSendAlgorithm* send_algorithm_; | 392 MockSendAlgorithm* send_algorithm_; |
326 scoped_refptr<TestTaskRunner> runner_; | 393 scoped_refptr<TestTaskRunner> runner_; |
327 scoped_ptr<MockWrite[]> mock_writes_; | 394 scoped_ptr<MockWrite[]> mock_writes_; |
328 MockClock clock_; | 395 MockClock clock_; |
329 TestQuicConnection* connection_; | 396 TestQuicConnection* connection_; |
330 scoped_ptr<QuicChromiumConnectionHelper> helper_; | 397 scoped_ptr<QuicChromiumConnectionHelper> helper_; |
331 testing::StrictMock<MockConnectionVisitor> visitor_; | 398 testing::StrictMock<MockConnectionVisitor> visitor_; |
332 scoped_ptr<QuicHttpStream> stream_; | 399 scoped_ptr<QuicHttpStream> stream_; |
333 TransportSecurityState transport_security_state_; | 400 TransportSecurityState transport_security_state_; |
334 scoped_ptr<QuicChromiumClientSession> session_; | 401 scoped_ptr<QuicChromiumClientSession> session_; |
335 QuicCryptoClientConfig crypto_config_; | 402 QuicCryptoClientConfig crypto_config_; |
336 TestCompletionCallback callback_; | 403 TestCompletionCallback callback_; |
337 HttpRequestInfo request_; | 404 HttpRequestInfo request_; |
338 HttpRequestHeaders headers_; | 405 HttpRequestHeaders headers_; |
339 HttpResponseInfo response_; | 406 HttpResponseInfo response_; |
340 scoped_refptr<IOBufferWithSize> read_buffer_; | 407 scoped_refptr<IOBufferWithSize> read_buffer_; |
341 SpdyHeaderBlock request_headers_; | 408 SpdyHeaderBlock request_headers_; |
342 SpdyHeaderBlock response_headers_; | 409 SpdyHeaderBlock response_headers_; |
343 std::string request_data_; | 410 std::string request_data_; |
344 std::string response_data_; | 411 std::string response_data_; |
345 QuicClientPushPromiseIndex push_promise_index_; | 412 QuicClientPushPromiseIndex push_promise_index_; |
346 | 413 |
| 414 // For server push testing |
| 415 scoped_ptr<QuicHttpStream> promised_stream_; |
| 416 SpdyHeaderBlock push_promise_; |
| 417 SpdyHeaderBlock promised_response_; |
| 418 const QuicStreamId promise_id_; |
| 419 string promise_url_; |
| 420 string serialized_push_promise_; |
| 421 const QuicStreamId stream_id_; |
| 422 |
347 private: | 423 private: |
348 const QuicConnectionId connection_id_; | 424 const QuicConnectionId connection_id_; |
349 const QuicStreamId stream_id_; | |
350 QuicTestPacketMaker maker_; | 425 QuicTestPacketMaker maker_; |
351 IPEndPoint self_addr_; | 426 IPEndPoint self_addr_; |
352 IPEndPoint peer_addr_; | 427 IPEndPoint peer_addr_; |
353 MockRandom random_generator_; | 428 MockRandom random_generator_; |
354 ProofVerifyDetailsChromium verify_details_; | 429 ProofVerifyDetailsChromium verify_details_; |
355 MockCryptoClientStreamFactory crypto_client_stream_factory_; | 430 MockCryptoClientStreamFactory crypto_client_stream_factory_; |
356 scoped_ptr<StaticSocketDataProvider> socket_data_; | 431 scoped_ptr<StaticSocketDataProvider> socket_data_; |
357 std::vector<PacketToWrite> writes_; | 432 std::vector<PacketToWrite> writes_; |
| 433 QuicStreamOffset response_offset_; |
358 }; | 434 }; |
359 | 435 |
360 INSTANTIATE_TEST_CASE_P(Version, | 436 INSTANTIATE_TEST_CASE_P(Version, |
361 QuicHttpStreamTest, | 437 QuicHttpStreamTest, |
362 ::testing::ValuesIn(QuicSupportedVersions())); | 438 ::testing::ValuesIn(QuicSupportedVersions())); |
363 | 439 |
364 TEST_P(QuicHttpStreamTest, RenewStreamForAuth) { | 440 TEST_P(QuicHttpStreamTest, RenewStreamForAuth) { |
365 Initialize(); | 441 Initialize(); |
366 EXPECT_EQ(nullptr, stream_->RenewStreamForAuth()); | 442 EXPECT_EQ(nullptr, stream_->RenewStreamForAuth()); |
367 } | 443 } |
(...skipping 655 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
1023 request_.upload_data_stream = &upload_data_stream; | 1099 request_.upload_data_stream = &upload_data_stream; |
1024 ASSERT_EQ(OK, request_.upload_data_stream->Init( | 1100 ASSERT_EQ(OK, request_.upload_data_stream->Init( |
1025 TestCompletionCallback().callback())); | 1101 TestCompletionCallback().callback())); |
1026 | 1102 |
1027 ASSERT_EQ(OK, stream_->InitializeStream(&request_, DEFAULT_PRIORITY, net_log_, | 1103 ASSERT_EQ(OK, stream_->InitializeStream(&request_, DEFAULT_PRIORITY, net_log_, |
1028 callback_.callback())); | 1104 callback_.callback())); |
1029 ASSERT_EQ(ERR_QUIC_PROTOCOL_ERROR, | 1105 ASSERT_EQ(ERR_QUIC_PROTOCOL_ERROR, |
1030 stream_->SendRequest(headers_, &response_, callback_.callback())); | 1106 stream_->SendRequest(headers_, &response_, callback_.callback())); |
1031 } | 1107 } |
1032 | 1108 |
| 1109 TEST_P(QuicHttpStreamTest, ServerPushGetRequest) { |
| 1110 SetRequest("GET", "/", DEFAULT_PRIORITY); |
| 1111 Initialize(); |
| 1112 |
| 1113 // Initialize the first stream, for receiving the promise on. |
| 1114 request_.method = "GET"; |
| 1115 request_.url = GURL("http://www.example.org/"); |
| 1116 |
| 1117 EXPECT_EQ(OK, stream_->InitializeStream(&request_, DEFAULT_PRIORITY, net_log_, |
| 1118 callback_.callback())); |
| 1119 |
| 1120 // TODO(ckrasic) - could do this via constructing a PUSH_PROMISE |
| 1121 // packet, but does it matter? |
| 1122 ReceivePromise(promise_id_); |
| 1123 EXPECT_NE(session_->GetPromisedByUrl(promise_url_), nullptr); |
| 1124 |
| 1125 request_.url = GURL(promise_url_); |
| 1126 |
| 1127 // Make the second stream that will exercise the first step of the |
| 1128 // server push rendezvous mechanism. |
| 1129 EXPECT_EQ(OK, |
| 1130 promised_stream_->InitializeStream(&request_, DEFAULT_PRIORITY, |
| 1131 net_log_, callback_.callback())); |
| 1132 |
| 1133 // Receive the promised response headers. |
| 1134 response_headers_ = promised_response_; |
| 1135 size_t spdy_response_headers_frame_length; |
| 1136 ProcessPacket(InnerConstructResponseHeadersPacket( |
| 1137 1, promise_id_, false, &spdy_response_headers_frame_length)); |
| 1138 |
| 1139 // Receive the promised response body. |
| 1140 const char kResponseBody[] = "Hello world!"; |
| 1141 ProcessPacket( |
| 1142 InnerConstructDataPacket(2, promise_id_, false, kFin, 0, kResponseBody)); |
| 1143 |
| 1144 // Now sending a matching request will have successful rendezvous |
| 1145 // with the promised stream. |
| 1146 EXPECT_EQ(OK, promised_stream_->SendRequest(headers_, &response_, |
| 1147 callback_.callback())); |
| 1148 |
| 1149 EXPECT_EQ( |
| 1150 QuicHttpStreamPeer::GetQuicChromiumClientStream(promised_stream_.get()) |
| 1151 ->id(), |
| 1152 promise_id_); |
| 1153 |
| 1154 // The headers will be immediately available. |
| 1155 EXPECT_EQ(OK, promised_stream_->ReadResponseHeaders(callback_.callback())); |
| 1156 |
| 1157 // As will be the body. |
| 1158 EXPECT_EQ( |
| 1159 static_cast<int>(strlen(kResponseBody)), |
| 1160 promised_stream_->ReadResponseBody( |
| 1161 read_buffer_.get(), read_buffer_->size(), callback_.callback())); |
| 1162 EXPECT_TRUE(promised_stream_->IsResponseBodyComplete()); |
| 1163 EXPECT_TRUE(AtEof()); |
| 1164 |
| 1165 EXPECT_EQ(0, stream_->GetTotalSentBytes()); |
| 1166 EXPECT_EQ(0, stream_->GetTotalReceivedBytes()); |
| 1167 EXPECT_EQ(0, promised_stream_->GetTotalSentBytes()); |
| 1168 EXPECT_EQ(static_cast<int64_t>(spdy_response_headers_frame_length + |
| 1169 strlen(kResponseBody)), |
| 1170 promised_stream_->GetTotalReceivedBytes()); |
| 1171 } |
| 1172 |
| 1173 TEST_P(QuicHttpStreamTest, ServerPushGetRequestSlowResponse) { |
| 1174 SetRequest("GET", "/", DEFAULT_PRIORITY); |
| 1175 Initialize(); |
| 1176 |
| 1177 // Initialize the first stream, for receiving the promise on. |
| 1178 request_.method = "GET"; |
| 1179 request_.url = GURL("http://www.example.org/"); |
| 1180 |
| 1181 EXPECT_EQ(OK, stream_->InitializeStream(&request_, DEFAULT_PRIORITY, net_log_, |
| 1182 callback_.callback())); |
| 1183 |
| 1184 // TODO(ckrasic) - could do this via constructing a PUSH_PROMISE |
| 1185 // packet, but does it matter? |
| 1186 ReceivePromise(promise_id_); |
| 1187 EXPECT_NE(session_->GetPromisedByUrl(promise_url_), nullptr); |
| 1188 |
| 1189 request_.url = GURL(promise_url_); |
| 1190 |
| 1191 // Make the second stream that will exercise the first step of the |
| 1192 // server push rendezvous mechanism. |
| 1193 EXPECT_EQ(OK, |
| 1194 promised_stream_->InitializeStream(&request_, DEFAULT_PRIORITY, |
| 1195 net_log_, callback_.callback())); |
| 1196 |
| 1197 // Now sending a matching request will rendezvous with the promised |
| 1198 // stream, but pending secondary validation. |
| 1199 EXPECT_EQ(ERR_IO_PENDING, promised_stream_->SendRequest( |
| 1200 headers_, &response_, callback_.callback())); |
| 1201 |
| 1202 // Receive the promised response headers. |
| 1203 response_headers_ = promised_response_; |
| 1204 size_t spdy_response_headers_frame_length; |
| 1205 ProcessPacket(InnerConstructResponseHeadersPacket( |
| 1206 1, promise_id_, false, &spdy_response_headers_frame_length)); |
| 1207 |
| 1208 // Receive the promised response body. |
| 1209 const char kResponseBody[] = "Hello world!"; |
| 1210 ProcessPacket( |
| 1211 InnerConstructDataPacket(2, promise_id_, false, kFin, 0, kResponseBody)); |
| 1212 |
| 1213 base::MessageLoop::current()->RunUntilIdle(); |
| 1214 |
| 1215 // Rendezvous should have succeeded now, so the promised stream |
| 1216 // should point at our push stream, and we should be able read |
| 1217 // headers and data from it. |
| 1218 EXPECT_EQ(OK, callback_.WaitForResult()); |
| 1219 |
| 1220 EXPECT_EQ( |
| 1221 QuicHttpStreamPeer::GetQuicChromiumClientStream(promised_stream_.get()) |
| 1222 ->id(), |
| 1223 promise_id_); |
| 1224 |
| 1225 EXPECT_EQ(OK, promised_stream_->ReadResponseHeaders(callback_.callback())); |
| 1226 |
| 1227 EXPECT_EQ( |
| 1228 static_cast<int>(strlen(kResponseBody)), |
| 1229 promised_stream_->ReadResponseBody( |
| 1230 read_buffer_.get(), read_buffer_->size(), callback_.callback())); |
| 1231 |
| 1232 // Callback should return |
| 1233 EXPECT_TRUE(promised_stream_->IsResponseBodyComplete()); |
| 1234 EXPECT_TRUE(AtEof()); |
| 1235 |
| 1236 EXPECT_EQ(0, stream_->GetTotalSentBytes()); |
| 1237 EXPECT_EQ(0, stream_->GetTotalReceivedBytes()); |
| 1238 EXPECT_EQ(0, promised_stream_->GetTotalSentBytes()); |
| 1239 EXPECT_EQ(static_cast<int64_t>(spdy_response_headers_frame_length + |
| 1240 strlen(kResponseBody)), |
| 1241 promised_stream_->GetTotalReceivedBytes()); |
| 1242 } |
| 1243 |
| 1244 TEST_P(QuicHttpStreamTest, ServerPushCrossOriginOK) { |
| 1245 SetRequest("GET", "/", DEFAULT_PRIORITY); |
| 1246 Initialize(); |
| 1247 |
| 1248 // Initialize the first stream, for receiving the promise on. |
| 1249 request_.method = "GET"; |
| 1250 request_.url = GURL("http://www.example.org/"); |
| 1251 |
| 1252 EXPECT_EQ(OK, stream_->InitializeStream(&request_, DEFAULT_PRIORITY, net_log_, |
| 1253 callback_.callback())); |
| 1254 |
| 1255 // TODO(ckrasic) - could do this via constructing a PUSH_PROMISE |
| 1256 // packet, but does it matter? |
| 1257 |
| 1258 push_promise_[":authority"] = "mail.example.org"; |
| 1259 promise_url_ = SpdyUtils::GetUrlFromHeaderBlock(push_promise_); |
| 1260 serialized_push_promise_ = |
| 1261 SpdyUtils::SerializeUncompressedHeaders(push_promise_); |
| 1262 |
| 1263 ReceivePromise(promise_id_); |
| 1264 EXPECT_NE(session_->GetPromisedByUrl(promise_url_), nullptr); |
| 1265 |
| 1266 request_.url = GURL(promise_url_); |
| 1267 |
| 1268 // Make the second stream that will exercise the first step of the |
| 1269 // server push rendezvous mechanism. |
| 1270 EXPECT_EQ(OK, |
| 1271 promised_stream_->InitializeStream(&request_, DEFAULT_PRIORITY, |
| 1272 net_log_, callback_.callback())); |
| 1273 |
| 1274 // Receive the promised response headers. |
| 1275 response_headers_ = promised_response_; |
| 1276 size_t spdy_response_headers_frame_length; |
| 1277 ProcessPacket(InnerConstructResponseHeadersPacket( |
| 1278 1, promise_id_, false, &spdy_response_headers_frame_length)); |
| 1279 |
| 1280 // Receive the promised response body. |
| 1281 const char kResponseBody[] = "Hello world!"; |
| 1282 ProcessPacket( |
| 1283 InnerConstructDataPacket(2, promise_id_, false, kFin, 0, kResponseBody)); |
| 1284 |
| 1285 // Now sending a matching request will have successful rendezvous |
| 1286 // with the promised stream. |
| 1287 EXPECT_EQ(OK, promised_stream_->SendRequest(headers_, &response_, |
| 1288 callback_.callback())); |
| 1289 |
| 1290 EXPECT_EQ( |
| 1291 QuicHttpStreamPeer::GetQuicChromiumClientStream(promised_stream_.get()) |
| 1292 ->id(), |
| 1293 promise_id_); |
| 1294 |
| 1295 // The headers will be immediately available. |
| 1296 EXPECT_EQ(OK, promised_stream_->ReadResponseHeaders(callback_.callback())); |
| 1297 |
| 1298 // As will be the body. |
| 1299 EXPECT_EQ( |
| 1300 static_cast<int>(strlen(kResponseBody)), |
| 1301 promised_stream_->ReadResponseBody( |
| 1302 read_buffer_.get(), read_buffer_->size(), callback_.callback())); |
| 1303 EXPECT_TRUE(promised_stream_->IsResponseBodyComplete()); |
| 1304 EXPECT_TRUE(AtEof()); |
| 1305 |
| 1306 EXPECT_EQ(0, stream_->GetTotalSentBytes()); |
| 1307 EXPECT_EQ(0, stream_->GetTotalReceivedBytes()); |
| 1308 EXPECT_EQ(0, promised_stream_->GetTotalSentBytes()); |
| 1309 EXPECT_EQ(static_cast<int64_t>(spdy_response_headers_frame_length + |
| 1310 strlen(kResponseBody)), |
| 1311 promised_stream_->GetTotalReceivedBytes()); |
| 1312 } |
| 1313 |
| 1314 TEST_P(QuicHttpStreamTest, ServerPushCrossOriginFail) { |
| 1315 SetRequest("GET", "/", DEFAULT_PRIORITY); |
| 1316 Initialize(); |
| 1317 |
| 1318 // Initialize the first stream, for receiving the promise on. |
| 1319 request_.method = "GET"; |
| 1320 request_.url = GURL("http://www.example.org/"); |
| 1321 |
| 1322 EXPECT_EQ(OK, stream_->InitializeStream(&request_, DEFAULT_PRIORITY, net_log_, |
| 1323 callback_.callback())); |
| 1324 |
| 1325 // TODO(ckrasic) - could do this via constructing a PUSH_PROMISE |
| 1326 // packet, but does it matter? |
| 1327 push_promise_[":authority"] = "www.notexample.org"; |
| 1328 promise_url_ = SpdyUtils::GetUrlFromHeaderBlock(push_promise_); |
| 1329 serialized_push_promise_ = |
| 1330 SpdyUtils::SerializeUncompressedHeaders(push_promise_); |
| 1331 |
| 1332 ReceivePromise(promise_id_); |
| 1333 // The promise will have been rejected because the cert doesn't |
| 1334 // match. |
| 1335 EXPECT_EQ(session_->GetPromisedByUrl(promise_url_), nullptr); |
| 1336 } |
| 1337 |
| 1338 TEST_P(QuicHttpStreamTest, ServerPushVaryCheckOK) { |
| 1339 SetRequest("GET", "/", DEFAULT_PRIORITY); |
| 1340 Initialize(); |
| 1341 |
| 1342 // Initialize the first stream, for receiving the promise on. |
| 1343 request_.method = "GET"; |
| 1344 request_.url = GURL("http://www.example.org/"); |
| 1345 |
| 1346 EXPECT_EQ(OK, stream_->InitializeStream(&request_, DEFAULT_PRIORITY, net_log_, |
| 1347 callback_.callback())); |
| 1348 |
| 1349 push_promise_["accept-encoding"] = "gzip"; |
| 1350 serialized_push_promise_ = |
| 1351 SpdyUtils::SerializeUncompressedHeaders(push_promise_); |
| 1352 |
| 1353 // TODO(ckrasic) - could do this via constructing a PUSH_PROMISE |
| 1354 // packet, but does it matter? |
| 1355 ReceivePromise(promise_id_); |
| 1356 EXPECT_NE(session_->GetPromisedByUrl(promise_url_), nullptr); |
| 1357 |
| 1358 request_.url = GURL(promise_url_); |
| 1359 |
| 1360 // Make the second stream that will exercise the first step of the |
| 1361 // server push rendezvous mechanism. |
| 1362 EXPECT_EQ(OK, |
| 1363 promised_stream_->InitializeStream(&request_, DEFAULT_PRIORITY, |
| 1364 net_log_, callback_.callback())); |
| 1365 |
| 1366 headers_.SetHeader("accept-encoding", "gzip"); |
| 1367 |
| 1368 // Now sending a matching request will rendezvous with the promised |
| 1369 // stream, but pending secondary validation. |
| 1370 EXPECT_EQ(ERR_IO_PENDING, promised_stream_->SendRequest( |
| 1371 headers_, &response_, callback_.callback())); |
| 1372 |
| 1373 // Receive the promised response headers. |
| 1374 promised_response_["vary"] = "accept-encoding"; |
| 1375 response_headers_ = promised_response_; |
| 1376 size_t spdy_response_headers_frame_length; |
| 1377 ProcessPacket(InnerConstructResponseHeadersPacket( |
| 1378 1, promise_id_, false, &spdy_response_headers_frame_length)); |
| 1379 |
| 1380 // Receive the promised response body. |
| 1381 const char kResponseBody[] = "Hello world!"; |
| 1382 ProcessPacket( |
| 1383 InnerConstructDataPacket(2, promise_id_, false, kFin, 0, kResponseBody)); |
| 1384 |
| 1385 base::MessageLoop::current()->RunUntilIdle(); |
| 1386 |
| 1387 // Rendezvous should have succeeded now, so the promised stream |
| 1388 // should point at our push stream, and we should be able read |
| 1389 // headers and data from it. |
| 1390 EXPECT_EQ(OK, callback_.WaitForResult()); |
| 1391 |
| 1392 EXPECT_EQ( |
| 1393 QuicHttpStreamPeer::GetQuicChromiumClientStream(promised_stream_.get()) |
| 1394 ->id(), |
| 1395 promise_id_); |
| 1396 |
| 1397 EXPECT_EQ(OK, promised_stream_->ReadResponseHeaders(callback_.callback())); |
| 1398 |
| 1399 EXPECT_EQ( |
| 1400 static_cast<int>(strlen(kResponseBody)), |
| 1401 promised_stream_->ReadResponseBody( |
| 1402 read_buffer_.get(), read_buffer_->size(), callback_.callback())); |
| 1403 |
| 1404 // Callback should return |
| 1405 EXPECT_TRUE(promised_stream_->IsResponseBodyComplete()); |
| 1406 EXPECT_TRUE(AtEof()); |
| 1407 |
| 1408 EXPECT_EQ(0, stream_->GetTotalSentBytes()); |
| 1409 EXPECT_EQ(0, stream_->GetTotalReceivedBytes()); |
| 1410 EXPECT_EQ(0, promised_stream_->GetTotalSentBytes()); |
| 1411 EXPECT_EQ(static_cast<int64_t>(spdy_response_headers_frame_length + |
| 1412 strlen(kResponseBody)), |
| 1413 promised_stream_->GetTotalReceivedBytes()); |
| 1414 } |
| 1415 |
| 1416 TEST_P(QuicHttpStreamTest, ServerPushVaryCheckFail) { |
| 1417 SetRequest("GET", "/", DEFAULT_PRIORITY); |
| 1418 request_headers_[":scheme"] = "https"; |
| 1419 request_headers_[":path"] = "/bar"; |
| 1420 request_headers_["accept-encoding"] = "sdch"; |
| 1421 |
| 1422 size_t spdy_request_header_frame_length; |
| 1423 AddWrite(ConstructRstStreamVaryMismatchPacket(1)); |
| 1424 AddWrite(InnerConstructRequestHeadersPacket( |
| 1425 2, stream_id_ + 2, !kIncludeVersion, kFin, DEFAULT_PRIORITY, |
| 1426 &spdy_request_header_frame_length)); |
| 1427 AddWrite(ConstructAckPacket(3, 3, 1)); |
| 1428 AddWrite(ConstructRstStreamCancelledPacket(4)); |
| 1429 Initialize(); |
| 1430 |
| 1431 // Initialize the first stream, for receiving the promise on. |
| 1432 request_.method = "GET"; |
| 1433 request_.url = GURL("http://www.example.org/"); |
| 1434 |
| 1435 EXPECT_EQ(OK, stream_->InitializeStream(&request_, DEFAULT_PRIORITY, net_log_, |
| 1436 callback_.callback())); |
| 1437 |
| 1438 push_promise_["accept-encoding"] = "gzip"; |
| 1439 serialized_push_promise_ = |
| 1440 SpdyUtils::SerializeUncompressedHeaders(push_promise_); |
| 1441 |
| 1442 // TODO(ckrasic) - could do this via constructing a PUSH_PROMISE |
| 1443 // packet, but does it matter? |
| 1444 ReceivePromise(promise_id_); |
| 1445 EXPECT_NE(session_->GetPromisedByUrl(promise_url_), nullptr); |
| 1446 |
| 1447 request_.url = GURL(promise_url_); |
| 1448 |
| 1449 // Make the second stream that will exercise the first step of the |
| 1450 // server push rendezvous mechanism. |
| 1451 EXPECT_EQ(OK, |
| 1452 promised_stream_->InitializeStream(&request_, DEFAULT_PRIORITY, |
| 1453 net_log_, callback_.callback())); |
| 1454 |
| 1455 headers_.SetHeader("accept-encoding", "sdch"); |
| 1456 |
| 1457 // Now sending a matching request will rendezvous with the promised |
| 1458 // stream, but pending secondary validation. |
| 1459 EXPECT_EQ(ERR_IO_PENDING, promised_stream_->SendRequest( |
| 1460 headers_, &response_, callback_.callback())); |
| 1461 |
| 1462 // Receive the promised response headers. |
| 1463 promised_response_["vary"] = "accept-encoding"; |
| 1464 response_headers_ = promised_response_; |
| 1465 size_t spdy_response_headers_frame_length; |
| 1466 ProcessPacket(InnerConstructResponseHeadersPacket( |
| 1467 1, promise_id_, false, &spdy_response_headers_frame_length)); |
| 1468 |
| 1469 base::MessageLoop::current()->RunUntilIdle(); |
| 1470 |
| 1471 // Rendezvous should have failed due to vary mismatch, so the |
| 1472 // promised stream should have been aborted, and instead we have a |
| 1473 // new, regular client initiated stream. |
| 1474 EXPECT_EQ(OK, callback_.WaitForResult()); |
| 1475 |
| 1476 // Not a server-initiated stream. |
| 1477 EXPECT_NE( |
| 1478 QuicHttpStreamPeer::GetQuicChromiumClientStream(promised_stream_.get()) |
| 1479 ->id(), |
| 1480 promise_id_); |
| 1481 |
| 1482 // Instead, a new client-initiated stream. |
| 1483 EXPECT_EQ( |
| 1484 QuicHttpStreamPeer::GetQuicChromiumClientStream(promised_stream_.get()) |
| 1485 ->id(), |
| 1486 stream_id_ + 2); |
| 1487 |
| 1488 // After rendezvous failure, the push stream has been cancelled. |
| 1489 EXPECT_EQ(session_->GetPromisedByUrl(promise_url_), nullptr); |
| 1490 |
| 1491 // The rest of the test verifies that the retried as |
| 1492 // client-initiated version of |promised_stream_| works as intended. |
| 1493 |
| 1494 // Ack the request. |
| 1495 ProcessPacket(ConstructAckPacket(2, 0, 0)); |
| 1496 |
| 1497 SetResponse("404 Not Found", std::string()); |
| 1498 size_t spdy_response_header_frame_length; |
| 1499 ProcessPacket(InnerConstructResponseHeadersPacket( |
| 1500 3, stream_id_ + 2, kFin, &spdy_response_header_frame_length)); |
| 1501 |
| 1502 base::MessageLoop::current()->RunUntilIdle(); |
| 1503 |
| 1504 EXPECT_EQ(OK, promised_stream_->ReadResponseHeaders(callback_.callback())); |
| 1505 ASSERT_TRUE(response_.headers.get()); |
| 1506 EXPECT_EQ(404, response_.headers->response_code()); |
| 1507 EXPECT_TRUE(response_.headers->HasHeaderValue("Content-Type", "text/plain")); |
| 1508 EXPECT_FALSE(response_.response_time.is_null()); |
| 1509 EXPECT_FALSE(response_.request_time.is_null()); |
| 1510 |
| 1511 // There is no body, so this should return immediately. |
| 1512 EXPECT_EQ( |
| 1513 0, promised_stream_->ReadResponseBody( |
| 1514 read_buffer_.get(), read_buffer_->size(), callback_.callback())); |
| 1515 EXPECT_TRUE(promised_stream_->IsResponseBodyComplete()); |
| 1516 |
| 1517 stream_->Close(true); |
| 1518 |
| 1519 EXPECT_TRUE(AtEof()); |
| 1520 |
| 1521 // QuicHttpStream::GetTotalSent/ReceivedBytes currently only includes the |
| 1522 // headers and payload. |
| 1523 EXPECT_EQ(static_cast<int64_t>(spdy_request_header_frame_length), |
| 1524 promised_stream_->GetTotalSentBytes()); |
| 1525 EXPECT_EQ(static_cast<int64_t>(spdy_response_header_frame_length), |
| 1526 promised_stream_->GetTotalReceivedBytes()); |
| 1527 } |
| 1528 |
1033 } // namespace test | 1529 } // namespace test |
1034 } // namespace net | 1530 } // namespace net |
OLD | NEW |