Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(205)

Side by Side Diff: net/quic/quic_http_stream_test.cc

Issue 1692253004: QUIC - chromium server push support. (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: Initial for-review version Created 4 years, 10 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
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
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
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
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
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698