blob: c0643d4458d89d24681ea9a90b8fcb94b770cac5 [file] [log] [blame]
[email protected]f702d572012-12-04 15:56:201// 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
3// found in the LICENSE file.
4
5#include "net/quic/quic_http_stream.h"
6
7#include <vector>
8
9#include "net/base/net_errors.h"
10#include "net/base/test_completion_callback.h"
[email protected]b2d26cfd2012-12-11 10:36:0611#include "net/base/upload_bytes_element_reader.h"
[email protected]f702d572012-12-04 15:56:2012#include "net/base/upload_data_stream.h"
13#include "net/http/http_response_headers.h"
[email protected]fee17f72013-02-03 07:47:4114#include "net/quic/congestion_control/receive_algorithm_interface.h"
15#include "net/quic/congestion_control/send_algorithm_interface.h"
[email protected]e8ff26842013-03-22 21:02:0516#include "net/quic/crypto/crypto_protocol.h"
[email protected]4df69842013-02-27 06:32:1617#include "net/quic/crypto/quic_decrypter.h"
18#include "net/quic/crypto/quic_encrypter.h"
[email protected]17bf15c2014-03-14 10:08:0419#include "net/quic/crypto/quic_server_info.h"
[email protected]f702d572012-12-04 15:56:2020#include "net/quic/quic_client_session.h"
21#include "net/quic/quic_connection.h"
22#include "net/quic/quic_connection_helper.h"
[email protected]cbd731e2013-10-24 00:20:3923#include "net/quic/quic_default_packet_writer.h"
[email protected]24e5bc52013-09-18 15:36:5824#include "net/quic/quic_http_utils.h"
25#include "net/quic/quic_reliable_client_stream.h"
[email protected]9f0dcd4e2014-01-16 15:58:1426#include "net/quic/quic_write_blocked_list.h"
[email protected]3e7dca62013-09-10 16:14:2327#include "net/quic/spdy_utils.h"
[email protected]f702d572012-12-04 15:56:2028#include "net/quic/test_tools/mock_clock.h"
[email protected]e8ff26842013-03-22 21:02:0529#include "net/quic/test_tools/mock_crypto_client_stream_factory.h"
[email protected]9db443912013-02-25 05:27:0330#include "net/quic/test_tools/mock_random.h"
[email protected]b1f287d2012-12-22 17:25:3931#include "net/quic/test_tools/quic_connection_peer.h"
[email protected]1e960032013-12-20 19:00:2032#include "net/quic/test_tools/quic_test_packet_maker.h"
[email protected]f702d572012-12-04 15:56:2033#include "net/quic/test_tools/quic_test_utils.h"
34#include "net/quic/test_tools/test_task_runner.h"
35#include "net/socket/socket_test_util.h"
[email protected]6cca996b2013-01-25 07:43:3636#include "net/spdy/spdy_frame_builder.h"
37#include "net/spdy/spdy_framer.h"
38#include "net/spdy/spdy_http_utils.h"
39#include "net/spdy/spdy_protocol.h"
[email protected]f702d572012-12-04 15:56:2040#include "testing/gmock/include/gmock/gmock.h"
41#include "testing/gtest/include/gtest/gtest.h"
42
43using testing::_;
[email protected]06ff5152013-08-29 01:03:0544using testing::AnyNumber;
45using testing::Return;
[email protected]f702d572012-12-04 15:56:2046
47namespace net {
[email protected]f702d572012-12-04 15:56:2048namespace test {
[email protected]f702d572012-12-04 15:56:2049namespace {
50
51const char kUploadData[] = "hello world!";
[email protected]e4c3ea62014-03-15 00:45:1452const char kServerHostname[] = "www.google.com";
53const uint16 kServerPort = 80;
[email protected]f702d572012-12-04 15:56:2054
55class TestQuicConnection : public QuicConnection {
56 public:
[email protected]1e960032013-12-20 19:00:2057 TestQuicConnection(const QuicVersionVector& versions,
[email protected]3aa9ca72014-02-27 19:39:4358 QuicConnectionId connection_id,
[email protected]f702d572012-12-04 15:56:2059 IPEndPoint address,
[email protected]cbd731e2013-10-24 00:20:3960 QuicConnectionHelper* helper,
61 QuicPacketWriter* writer)
[email protected]3aa9ca72014-02-27 19:39:4362 : QuicConnection(connection_id, address, helper, writer, false,
[email protected]ce7bb1412014-05-17 15:51:3363 versions) {
[email protected]f702d572012-12-04 15:56:2064 }
65
[email protected]fee17f72013-02-03 07:47:4166 void SetSendAlgorithm(SendAlgorithmInterface* send_algorithm) {
67 QuicConnectionPeer::SetSendAlgorithm(this, send_algorithm);
[email protected]f702d572012-12-04 15:56:2068 }
[email protected]26f3f8e2012-12-13 21:07:1969
[email protected]fee17f72013-02-03 07:47:4170 void SetReceiveAlgorithm(ReceiveAlgorithmInterface* receive_algorithm) {
71 QuicConnectionPeer::SetReceiveAlgorithm(this, receive_algorithm);
[email protected]26f3f8e2012-12-13 21:07:1972 }
73};
74
[email protected]fee17f72013-02-03 07:47:4175class TestReceiveAlgorithm : public ReceiveAlgorithmInterface {
[email protected]26f3f8e2012-12-13 21:07:1976 public:
[email protected]1e960032013-12-20 19:00:2077 virtual bool GenerateCongestionFeedback(
78 QuicCongestionFeedbackFrame* /*congestion_feedback*/) {
79 return false;
[email protected]26f3f8e2012-12-13 21:07:1980 }
81
[email protected]6ae6e342014-02-06 02:21:4282 MOCK_METHOD3(RecordIncomingPacket,
83 void(QuicByteCount, QuicPacketSequenceNumber, QuicTime));
[email protected]f702d572012-12-04 15:56:2084};
85
[email protected]63534512012-12-23 18:49:0086// Subclass of QuicHttpStream that closes itself when the first piece of data
87// is received.
88class AutoClosingStream : public QuicHttpStream {
89 public:
[email protected]0b2294d32013-08-02 00:46:3690 explicit AutoClosingStream(const base::WeakPtr<QuicClientSession>& session)
91 : QuicHttpStream(session) {
[email protected]63534512012-12-23 18:49:0092 }
93
[email protected]46fadfd2013-02-06 09:40:1694 virtual int OnDataReceived(const char* data, int length) OVERRIDE {
[email protected]63534512012-12-23 18:49:0095 Close(false);
96 return OK;
97 }
98};
99
[email protected]f702d572012-12-04 15:56:20100} // namespace
101
[email protected]24e5bc52013-09-18 15:36:58102class QuicHttpStreamPeer {
103 public:
104 static QuicReliableClientStream* GetQuicReliableClientStream(
105 QuicHttpStream* stream) {
106 return stream->stream_;
107 }
108};
109
[email protected]1e960032013-12-20 19:00:20110class QuicHttpStreamTest : public ::testing::TestWithParam<QuicVersion> {
[email protected]f702d572012-12-04 15:56:20111 protected:
[email protected]1e960032013-12-20 19:00:20112 static const bool kFin = true;
113 static const bool kIncludeVersion = true;
114 static const bool kIncludeCongestionFeedback = true;
115
[email protected]f702d572012-12-04 15:56:20116 // Holds a packet to be written to the wire, and the IO mode that should
117 // be used by the mock socket when performing the write.
118 struct PacketToWrite {
119 PacketToWrite(IoMode mode, QuicEncryptedPacket* packet)
120 : mode(mode),
121 packet(packet) {
122 }
123 IoMode mode;
124 QuicEncryptedPacket* packet;
125 };
126
127 QuicHttpStreamTest()
128 : net_log_(BoundNetLog()),
[email protected]63534512012-12-23 18:49:00129 use_closing_stream_(false),
[email protected]f702d572012-12-04 15:56:20130 read_buffer_(new IOBufferWithSize(4096)),
[email protected]3aa9ca72014-02-27 19:39:43131 connection_id_(2),
[email protected]66ae5962014-05-22 11:13:05132 stream_id_(kClientDataStreamId1),
[email protected]3aa9ca72014-02-27 19:39:43133 maker_(GetParam(), connection_id_),
[email protected]1e960032013-12-20 19:00:20134 random_generator_(0) {
[email protected]f702d572012-12-04 15:56:20135 IPAddressNumber ip;
136 CHECK(ParseIPLiteralToNumber("192.0.2.33", &ip));
137 peer_addr_ = IPEndPoint(ip, 443);
138 self_addr_ = IPEndPoint(ip, 8435);
[email protected]f702d572012-12-04 15:56:20139 }
140
141 ~QuicHttpStreamTest() {
[email protected]4d283b32013-10-17 12:57:27142 session_->CloseSessionOnError(ERR_ABORTED);
[email protected]f702d572012-12-04 15:56:20143 for (size_t i = 0; i < writes_.size(); i++) {
144 delete writes_[i].packet;
145 }
146 }
147
148 // Adds a packet to the list of expected writes.
[email protected]1e960032013-12-20 19:00:20149 void AddWrite(scoped_ptr<QuicEncryptedPacket> packet) {
150 writes_.push_back(PacketToWrite(SYNCHRONOUS, packet.release()));
[email protected]f702d572012-12-04 15:56:20151 }
152
153 // Returns the packet to be written at position |pos|.
154 QuicEncryptedPacket* GetWrite(size_t pos) {
155 return writes_[pos].packet;
156 }
157
158 bool AtEof() {
159 return socket_data_->at_read_eof() && socket_data_->at_write_eof();
160 }
161
[email protected]1e960032013-12-20 19:00:20162 void ProcessPacket(scoped_ptr<QuicEncryptedPacket> packet) {
163 connection_->ProcessUdpPacket(self_addr_, peer_addr_, *packet);
[email protected]f702d572012-12-04 15:56:20164 }
165
166 // Configures the test fixture to use the list of expected writes.
167 void Initialize() {
168 mock_writes_.reset(new MockWrite[writes_.size()]);
169 for (size_t i = 0; i < writes_.size(); i++) {
170 mock_writes_[i] = MockWrite(writes_[i].mode,
171 writes_[i].packet->data(),
172 writes_[i].packet->length());
173 };
174
175 socket_data_.reset(new StaticSocketDataProvider(NULL, 0, mock_writes_.get(),
176 writes_.size()));
177
[email protected]e13201d82012-12-12 05:00:32178 MockUDPClientSocket* socket = new MockUDPClientSocket(socket_data_.get(),
179 net_log_.net_log());
180 socket->Connect(peer_addr_);
[email protected]f702d572012-12-04 15:56:20181 runner_ = new TestTaskRunner(&clock_);
[email protected]fee17f72013-02-03 07:47:41182 send_algorithm_ = new MockSendAlgorithm();
[email protected]1e960032013-12-20 19:00:20183 receive_algorithm_ = new TestReceiveAlgorithm();
[email protected]6ae6e342014-02-06 02:21:42184 EXPECT_CALL(*receive_algorithm_, RecordIncomingPacket(_, _, _)).
[email protected]06ff5152013-08-29 01:03:05185 Times(AnyNumber());
[email protected]efecff92013-09-24 07:49:23186 EXPECT_CALL(*send_algorithm_,
[email protected]ef0da582014-05-09 07:16:30187 OnPacketSent(_, _, _, _, _)).WillRepeatedly(Return(true));
[email protected]48878092013-07-26 14:51:56188 EXPECT_CALL(*send_algorithm_, RetransmissionDelay()).WillRepeatedly(
[email protected]06ff5152013-08-29 01:03:05189 Return(QuicTime::Delta::Zero()));
[email protected]08da9adb2014-04-24 08:33:31190 EXPECT_CALL(*send_algorithm_, GetCongestionWindow()).WillRepeatedly(
191 Return(kMaxPacketSize));
[email protected]77b5d50b2014-05-07 22:48:48192 EXPECT_CALL(*send_algorithm_, TimeUntilSend(_, _, _)).
[email protected]06ff5152013-08-29 01:03:05193 WillRepeatedly(Return(QuicTime::Delta::Zero()));
[email protected]ea825e02013-08-21 18:12:45194 EXPECT_CALL(*send_algorithm_, BandwidthEstimate()).WillRepeatedly(
[email protected]06ff5152013-08-29 01:03:05195 Return(QuicBandwidth::Zero()));
[email protected]62f90632013-11-01 13:07:11196 EXPECT_CALL(*send_algorithm_, SetFromConfig(_, _)).Times(AnyNumber());
[email protected]2cfc6bb82013-10-27 03:40:44197 helper_.reset(new QuicConnectionHelper(runner_.get(), &clock_,
198 &random_generator_));
[email protected]cbd731e2013-10-24 00:20:39199 writer_.reset(new QuicDefaultPacketWriter(socket));
[email protected]3aa9ca72014-02-27 19:39:43200 connection_ = new TestQuicConnection(SupportedVersions(GetParam()),
201 connection_id_, peer_addr_,
202 helper_.get(), writer_.get());
[email protected]f702d572012-12-04 15:56:20203 connection_->set_visitor(&visitor_);
[email protected]fee17f72013-02-03 07:47:41204 connection_->SetSendAlgorithm(send_algorithm_);
205 connection_->SetReceiveAlgorithm(receive_algorithm_);
[email protected]ef95114d2013-04-17 17:57:01206 crypto_config_.SetDefaults();
[email protected]18ccfdb2013-08-15 00:13:44207 session_.reset(
208 new QuicClientSession(connection_,
[email protected]cbd731e2013-10-24 00:20:39209 scoped_ptr<DatagramClientSocket>(socket),
210 writer_.Pass(), NULL,
[email protected]18ccfdb2013-08-15 00:13:44211 &crypto_client_stream_factory_,
[email protected]89f79a42014-03-19 11:30:01212 make_scoped_ptr((QuicServerInfo*)NULL),
[email protected]257f24f2014-04-01 09:15:37213 QuicServerId(kServerHostname, kServerPort,
214 false, PRIVACY_MODE_DISABLED),
[email protected]a5b98172014-06-18 07:01:59215 DefaultQuicConfig(), &crypto_config_,
[email protected]65768442014-06-06 23:37:03216 base::MessageLoop::current()->
217 message_loop_proxy().get(),
[email protected]ce7bb1412014-05-17 15:51:33218 NULL));
[email protected]ed3fc15d2013-03-08 18:37:44219 session_->GetCryptoStream()->CryptoConnect();
[email protected]8ba81212013-05-03 13:11:48220 EXPECT_TRUE(session_->IsCryptoHandshakeConfirmed());
[email protected]6cca996b2013-01-25 07:43:36221 stream_.reset(use_closing_stream_ ?
[email protected]0b2294d32013-08-02 00:46:36222 new AutoClosingStream(session_->GetWeakPtr()) :
223 new QuicHttpStream(session_->GetWeakPtr()));
[email protected]98a9d1252014-04-04 00:43:59224 clock_.AdvanceTime(QuicTime::Delta::FromMilliseconds(20));
[email protected]6cca996b2013-01-25 07:43:36225 }
226
[email protected]1e960032013-12-20 19:00:20227 void SetRequest(const std::string& method,
228 const std::string& path,
229 RequestPriority priority) {
230 request_headers_ = maker_.GetRequestHeaders(method, "http", path);
[email protected]6cca996b2013-01-25 07:43:36231 }
232
[email protected]1e960032013-12-20 19:00:20233 void SetResponse(const std::string& status, const std::string& body) {
234 response_headers_ = maker_.GetResponseHeaders(status);
[email protected]92bf17c2014-03-03 21:14:03235 response_data_ = body;
[email protected]6cca996b2013-01-25 07:43:36236 }
[email protected]f702d572012-12-04 15:56:20237
[email protected]1e960032013-12-20 19:00:20238 scoped_ptr<QuicEncryptedPacket> ConstructDataPacket(
[email protected]f702d572012-12-04 15:56:20239 QuicPacketSequenceNumber sequence_number,
[email protected]e8ff26842013-03-22 21:02:05240 bool should_include_version,
[email protected]f702d572012-12-04 15:56:20241 bool fin,
242 QuicStreamOffset offset,
243 base::StringPiece data) {
[email protected]1e960032013-12-20 19:00:20244 return maker_.MakeDataPacket(
245 sequence_number, stream_id_, should_include_version, fin, offset, data);
[email protected]f702d572012-12-04 15:56:20246 }
247
[email protected]1e960032013-12-20 19:00:20248 scoped_ptr<QuicEncryptedPacket> ConstructRequestHeadersPacket(
249 QuicPacketSequenceNumber sequence_number,
250 bool fin) {
251 return maker_.MakeRequestHeadersPacket(
252 sequence_number, stream_id_, kIncludeVersion, fin, request_headers_);
253 }
254
255 scoped_ptr<QuicEncryptedPacket> ConstructResponseHeadersPacket(
256 QuicPacketSequenceNumber sequence_number,
257 bool fin) {
258 return maker_.MakeResponseHeadersPacket(
259 sequence_number, stream_id_, !kIncludeVersion, fin, response_headers_);
260 }
261
262 scoped_ptr<QuicEncryptedPacket> ConstructRstStreamPacket(
[email protected]06ff5152013-08-29 01:03:05263 QuicPacketSequenceNumber sequence_number) {
[email protected]1e960032013-12-20 19:00:20264 return maker_.MakeRstPacket(
[email protected]51cc1342014-04-18 23:44:37265 sequence_number, true, stream_id_,
266 AdjustErrorForVersion(QUIC_RST_FLOW_CONTROL_ACCOUNTING, GetParam()));
[email protected]06ff5152013-08-29 01:03:05267 }
268
[email protected]c5e1aca2014-01-30 04:03:04269 scoped_ptr<QuicEncryptedPacket> ConstructAckAndRstStreamPacket(
270 QuicPacketSequenceNumber sequence_number) {
271 return maker_.MakeAckAndRstPacket(
272 sequence_number, !kIncludeVersion, stream_id_, QUIC_STREAM_CANCELLED,
[email protected]08da9adb2014-04-24 08:33:31273 2, 1, !kIncludeCongestionFeedback);
[email protected]c5e1aca2014-01-30 04:03:04274 }
275
[email protected]1e960032013-12-20 19:00:20276 scoped_ptr<QuicEncryptedPacket> ConstructAckPacket(
[email protected]f702d572012-12-04 15:56:20277 QuicPacketSequenceNumber sequence_number,
[email protected]63534512012-12-23 18:49:00278 QuicPacketSequenceNumber largest_received,
279 QuicPacketSequenceNumber least_unacked) {
[email protected]1e960032013-12-20 19:00:20280 return maker_.MakeAckPacket(sequence_number, largest_received,
281 least_unacked, !kIncludeCongestionFeedback);
[email protected]63534512012-12-23 18:49:00282 }
283
[email protected]f702d572012-12-04 15:56:20284 BoundNetLog net_log_;
[email protected]63534512012-12-23 18:49:00285 bool use_closing_stream_;
[email protected]fee17f72013-02-03 07:47:41286 MockSendAlgorithm* send_algorithm_;
287 TestReceiveAlgorithm* receive_algorithm_;
[email protected]f702d572012-12-04 15:56:20288 scoped_refptr<TestTaskRunner> runner_;
[email protected]4356f0f2013-04-07 00:58:17289 scoped_ptr<MockWrite[]> mock_writes_;
[email protected]f702d572012-12-04 15:56:20290 MockClock clock_;
291 TestQuicConnection* connection_;
[email protected]2cfc6bb82013-10-27 03:40:44292 scoped_ptr<QuicConnectionHelper> helper_;
[email protected]f702d572012-12-04 15:56:20293 testing::StrictMock<MockConnectionVisitor> visitor_;
294 scoped_ptr<QuicHttpStream> stream_;
[email protected]cbd731e2013-10-24 00:20:39295 scoped_ptr<QuicDefaultPacketWriter> writer_;
[email protected]f702d572012-12-04 15:56:20296 scoped_ptr<QuicClientSession> session_;
[email protected]ef95114d2013-04-17 17:57:01297 QuicCryptoClientConfig crypto_config_;
[email protected]f702d572012-12-04 15:56:20298 TestCompletionCallback callback_;
299 HttpRequestInfo request_;
300 HttpRequestHeaders headers_;
301 HttpResponseInfo response_;
302 scoped_refptr<IOBufferWithSize> read_buffer_;
[email protected]1e960032013-12-20 19:00:20303 SpdyHeaderBlock request_headers_;
304 SpdyHeaderBlock response_headers_;
[email protected]6cca996b2013-01-25 07:43:36305 std::string request_data_;
306 std::string response_data_;
[email protected]f702d572012-12-04 15:56:20307
308 private:
[email protected]3aa9ca72014-02-27 19:39:43309 const QuicConnectionId connection_id_;
[email protected]1e960032013-12-20 19:00:20310 const QuicStreamId stream_id_;
311 QuicTestPacketMaker maker_;
[email protected]f702d572012-12-04 15:56:20312 IPEndPoint self_addr_;
313 IPEndPoint peer_addr_;
[email protected]457d6952013-12-13 09:24:58314 MockRandom random_generator_;
[email protected]e8ff26842013-03-22 21:02:05315 MockCryptoClientStreamFactory crypto_client_stream_factory_;
[email protected]f702d572012-12-04 15:56:20316 scoped_ptr<StaticSocketDataProvider> socket_data_;
317 std::vector<PacketToWrite> writes_;
318};
319
[email protected]1e960032013-12-20 19:00:20320INSTANTIATE_TEST_CASE_P(Version, QuicHttpStreamTest,
321 ::testing::ValuesIn(QuicSupportedVersions()));
322
323TEST_P(QuicHttpStreamTest, RenewStreamForAuth) {
[email protected]ed3fc15d2013-03-08 18:37:44324 Initialize();
[email protected]f702d572012-12-04 15:56:20325 EXPECT_EQ(NULL, stream_->RenewStreamForAuth());
326}
327
[email protected]1e960032013-12-20 19:00:20328TEST_P(QuicHttpStreamTest, CanFindEndOfResponse) {
[email protected]ed3fc15d2013-03-08 18:37:44329 Initialize();
[email protected]f702d572012-12-04 15:56:20330 EXPECT_TRUE(stream_->CanFindEndOfResponse());
331}
332
[email protected]1e960032013-12-20 19:00:20333TEST_P(QuicHttpStreamTest, IsConnectionReusable) {
[email protected]ed3fc15d2013-03-08 18:37:44334 Initialize();
[email protected]f702d572012-12-04 15:56:20335 EXPECT_FALSE(stream_->IsConnectionReusable());
336}
337
[email protected]1e960032013-12-20 19:00:20338TEST_P(QuicHttpStreamTest, GetRequest) {
339 SetRequest("GET", "/", DEFAULT_PRIORITY);
[email protected]92bf17c2014-03-03 21:14:03340 AddWrite(ConstructRequestHeadersPacket(1, kFin));
[email protected]f702d572012-12-04 15:56:20341 Initialize();
342
343 request_.method = "GET";
344 request_.url = GURL("http://www.google.com/");
345
[email protected]262eec82013-03-19 21:01:36346 EXPECT_EQ(OK, stream_->InitializeStream(&request_, DEFAULT_PRIORITY,
347 net_log_, callback_.callback()));
[email protected]f702d572012-12-04 15:56:20348 EXPECT_EQ(OK, stream_->SendRequest(headers_, &response_,
349 callback_.callback()));
[email protected]f702d572012-12-04 15:56:20350
351 // Ack the request.
[email protected]1e960032013-12-20 19:00:20352 ProcessPacket(ConstructAckPacket(1, 0, 0));
[email protected]f702d572012-12-04 15:56:20353
354 EXPECT_EQ(ERR_IO_PENDING,
355 stream_->ReadResponseHeaders(callback_.callback()));
356
[email protected]1e960032013-12-20 19:00:20357 SetResponse("404 Not Found", std::string());
[email protected]92bf17c2014-03-03 21:14:03358 ProcessPacket(ConstructResponseHeadersPacket(2, kFin));
[email protected]f702d572012-12-04 15:56:20359
360 // Now that the headers have been processed, the callback will return.
361 EXPECT_EQ(OK, callback_.WaitForResult());
[email protected]cadac622013-06-11 16:46:36362 ASSERT_TRUE(response_.headers.get());
[email protected]f702d572012-12-04 15:56:20363 EXPECT_EQ(404, response_.headers->response_code());
364 EXPECT_TRUE(response_.headers->HasHeaderValue("Content-Type", "text/plain"));
[email protected]6ed67432014-06-24 01:53:53365 EXPECT_FALSE(response_.response_time.is_null());
366 EXPECT_FALSE(response_.request_time.is_null());
[email protected]f702d572012-12-04 15:56:20367
368 // There is no body, so this should return immediately.
369 EXPECT_EQ(0, stream_->ReadResponseBody(read_buffer_.get(),
370 read_buffer_->size(),
371 callback_.callback()));
372 EXPECT_TRUE(stream_->IsResponseBodyComplete());
373 EXPECT_TRUE(AtEof());
374}
375
[email protected]3e7dca62013-09-10 16:14:23376// Regression test for http://crbug.com/288128
[email protected]1e960032013-12-20 19:00:20377TEST_P(QuicHttpStreamTest, GetRequestLargeResponse) {
378 SetRequest("GET", "/", DEFAULT_PRIORITY);
[email protected]92bf17c2014-03-03 21:14:03379 AddWrite(ConstructRequestHeadersPacket(1, kFin));
[email protected]3e7dca62013-09-10 16:14:23380 Initialize();
381
382 request_.method = "GET";
383 request_.url = GURL("http://www.google.com/");
384
385 EXPECT_EQ(OK, stream_->InitializeStream(&request_, DEFAULT_PRIORITY,
386 net_log_, callback_.callback()));
387 EXPECT_EQ(OK, stream_->SendRequest(headers_, &response_,
388 callback_.callback()));
[email protected]3e7dca62013-09-10 16:14:23389
390 // Ack the request.
[email protected]1e960032013-12-20 19:00:20391 ProcessPacket(ConstructAckPacket(1, 0, 0));
[email protected]3e7dca62013-09-10 16:14:23392
393 EXPECT_EQ(ERR_IO_PENDING,
394 stream_->ReadResponseHeaders(callback_.callback()));
395
396 SpdyHeaderBlock headers;
397 headers[":status"] = "200 OK";
398 headers[":version"] = "HTTP/1.1";
399 headers["content-type"] = "text/plain";
400 headers["big6"] = std::string(10000, 'x'); // Lots of x's.
401
402 std::string response = SpdyUtils::SerializeUncompressedHeaders(headers);
403 EXPECT_LT(4096u, response.length());
404 stream_->OnDataReceived(response.data(), response.length());
405 stream_->OnClose(QUIC_NO_ERROR);
406
407 // Now that the headers have been processed, the callback will return.
408 EXPECT_EQ(OK, callback_.WaitForResult());
409 ASSERT_TRUE(response_.headers.get());
410 EXPECT_EQ(200, response_.headers->response_code());
411 EXPECT_TRUE(response_.headers->HasHeaderValue("Content-Type", "text/plain"));
412
413 // There is no body, so this should return immediately.
414 EXPECT_EQ(0, stream_->ReadResponseBody(read_buffer_.get(),
415 read_buffer_->size(),
416 callback_.callback()));
417 EXPECT_TRUE(stream_->IsResponseBodyComplete());
418 EXPECT_TRUE(AtEof());
419}
420
[email protected]1e960032013-12-20 19:00:20421TEST_P(QuicHttpStreamTest, SendPostRequest) {
422 SetRequest("POST", "/", DEFAULT_PRIORITY);
[email protected]92bf17c2014-03-03 21:14:03423 AddWrite(ConstructRequestHeadersPacket(1, !kFin));
424 AddWrite(ConstructDataPacket(2, kIncludeVersion, kFin, 0, kUploadData));
[email protected]1e960032013-12-20 19:00:20425 AddWrite(ConstructAckPacket(3, 3, 1));
[email protected]f702d572012-12-04 15:56:20426
427 Initialize();
428
[email protected]b2d26cfd2012-12-11 10:36:06429 ScopedVector<UploadElementReader> element_readers;
430 element_readers.push_back(
431 new UploadBytesElementReader(kUploadData, strlen(kUploadData)));
[email protected]96c77a72013-09-24 09:49:20432 UploadDataStream upload_data_stream(element_readers.Pass(), 0);
[email protected]f702d572012-12-04 15:56:20433 request_.method = "POST";
434 request_.url = GURL("http://www.google.com/");
435 request_.upload_data_stream = &upload_data_stream;
[email protected]4db27d82012-12-20 11:50:24436 ASSERT_EQ(OK, request_.upload_data_stream->Init(CompletionCallback()));
[email protected]f702d572012-12-04 15:56:20437
[email protected]262eec82013-03-19 21:01:36438 EXPECT_EQ(OK, stream_->InitializeStream(&request_, DEFAULT_PRIORITY,
439 net_log_, callback_.callback()));
[email protected]f702d572012-12-04 15:56:20440 EXPECT_EQ(OK, stream_->SendRequest(headers_, &response_,
441 callback_.callback()));
[email protected]f702d572012-12-04 15:56:20442
443 // Ack both packets in the request.
[email protected]1e960032013-12-20 19:00:20444 ProcessPacket(ConstructAckPacket(1, 0, 0));
[email protected]f702d572012-12-04 15:56:20445
446 // Send the response headers (but not the body).
[email protected]1e960032013-12-20 19:00:20447 SetResponse("200 OK", std::string());
[email protected]92bf17c2014-03-03 21:14:03448 ProcessPacket(ConstructResponseHeadersPacket(2, !kFin));
[email protected]f702d572012-12-04 15:56:20449
450 // Since the headers have already arrived, this should return immediately.
451 EXPECT_EQ(OK, stream_->ReadResponseHeaders(callback_.callback()));
[email protected]cadac622013-06-11 16:46:36452 ASSERT_TRUE(response_.headers.get());
[email protected]f702d572012-12-04 15:56:20453 EXPECT_EQ(200, response_.headers->response_code());
454 EXPECT_TRUE(response_.headers->HasHeaderValue("Content-Type", "text/plain"));
455
456 // Send the response body.
457 const char kResponseBody[] = "Hello world!";
[email protected]92bf17c2014-03-03 21:14:03458 ProcessPacket(ConstructDataPacket(3, false, kFin, 0, kResponseBody));
[email protected]f702d572012-12-04 15:56:20459 // Since the body has already arrived, this should return immediately.
460 EXPECT_EQ(static_cast<int>(strlen(kResponseBody)),
461 stream_->ReadResponseBody(read_buffer_.get(), read_buffer_->size(),
462 callback_.callback()));
463
464 EXPECT_TRUE(stream_->IsResponseBodyComplete());
465 EXPECT_TRUE(AtEof());
466}
467
[email protected]1e960032013-12-20 19:00:20468TEST_P(QuicHttpStreamTest, SendChunkedPostRequest) {
469 SetRequest("POST", "/", DEFAULT_PRIORITY);
[email protected]c9e49a02013-02-26 05:56:47470 size_t chunk_size = strlen(kUploadData);
[email protected]92bf17c2014-03-03 21:14:03471 AddWrite(ConstructRequestHeadersPacket(1, !kFin));
472 AddWrite(ConstructDataPacket(2, kIncludeVersion, !kFin, 0, kUploadData));
473 AddWrite(ConstructDataPacket(3, kIncludeVersion, kFin, chunk_size,
474 kUploadData));
[email protected]1e960032013-12-20 19:00:20475 AddWrite(ConstructAckPacket(4, 3, 1));
[email protected]c9e49a02013-02-26 05:56:47476 Initialize();
477
478 UploadDataStream upload_data_stream(UploadDataStream::CHUNKED, 0);
479 upload_data_stream.AppendChunk(kUploadData, chunk_size, false);
480
481 request_.method = "POST";
482 request_.url = GURL("http://www.google.com/");
483 request_.upload_data_stream = &upload_data_stream;
484 ASSERT_EQ(OK, request_.upload_data_stream->Init(CompletionCallback()));
485
[email protected]262eec82013-03-19 21:01:36486 ASSERT_EQ(OK, stream_->InitializeStream(&request_, DEFAULT_PRIORITY,
487 net_log_, callback_.callback()));
[email protected]c9e49a02013-02-26 05:56:47488 ASSERT_EQ(ERR_IO_PENDING, stream_->SendRequest(headers_, &response_,
489 callback_.callback()));
[email protected]c9e49a02013-02-26 05:56:47490
491 upload_data_stream.AppendChunk(kUploadData, chunk_size, true);
492
493 // Ack both packets in the request.
[email protected]1e960032013-12-20 19:00:20494 ProcessPacket(ConstructAckPacket(1, 0, 0));
[email protected]c9e49a02013-02-26 05:56:47495
496 // Send the response headers (but not the body).
[email protected]1e960032013-12-20 19:00:20497 SetResponse("200 OK", std::string());
[email protected]92bf17c2014-03-03 21:14:03498 ProcessPacket(ConstructResponseHeadersPacket(2, !kFin));
[email protected]c9e49a02013-02-26 05:56:47499
500 // Since the headers have already arrived, this should return immediately.
501 ASSERT_EQ(OK, stream_->ReadResponseHeaders(callback_.callback()));
[email protected]cadac622013-06-11 16:46:36502 ASSERT_TRUE(response_.headers.get());
[email protected]c9e49a02013-02-26 05:56:47503 EXPECT_EQ(200, response_.headers->response_code());
504 EXPECT_TRUE(response_.headers->HasHeaderValue("Content-Type", "text/plain"));
505
506 // Send the response body.
507 const char kResponseBody[] = "Hello world!";
[email protected]1e960032013-12-20 19:00:20508 ProcessPacket(ConstructDataPacket(3, false, kFin, response_data_.length(),
509 kResponseBody));
[email protected]c9e49a02013-02-26 05:56:47510
511 // Since the body has already arrived, this should return immediately.
512 ASSERT_EQ(static_cast<int>(strlen(kResponseBody)),
513 stream_->ReadResponseBody(read_buffer_.get(), read_buffer_->size(),
514 callback_.callback()));
515
516 EXPECT_TRUE(stream_->IsResponseBodyComplete());
517 EXPECT_TRUE(AtEof());
518}
519
[email protected]1e960032013-12-20 19:00:20520TEST_P(QuicHttpStreamTest, DestroyedEarly) {
521 SetRequest("GET", "/", DEFAULT_PRIORITY);
[email protected]92bf17c2014-03-03 21:14:03522 AddWrite(ConstructRequestHeadersPacket(1, kFin));
[email protected]c5e1aca2014-01-30 04:03:04523 AddWrite(ConstructAckAndRstStreamPacket(2));
[email protected]63534512012-12-23 18:49:00524 use_closing_stream_ = true;
525 Initialize();
526
527 request_.method = "GET";
528 request_.url = GURL("http://www.google.com/");
529
[email protected]262eec82013-03-19 21:01:36530 EXPECT_EQ(OK, stream_->InitializeStream(&request_, DEFAULT_PRIORITY,
531 net_log_, callback_.callback()));
[email protected]63534512012-12-23 18:49:00532 EXPECT_EQ(OK, stream_->SendRequest(headers_, &response_,
[email protected]24e5bc52013-09-18 15:36:58533 callback_.callback()));
[email protected]63534512012-12-23 18:49:00534
535 // Ack the request.
[email protected]1e960032013-12-20 19:00:20536 ProcessPacket(ConstructAckPacket(1, 0, 0));
[email protected]63534512012-12-23 18:49:00537 EXPECT_EQ(ERR_IO_PENDING,
538 stream_->ReadResponseHeaders(callback_.callback()));
539
540 // Send the response with a body.
[email protected]1e960032013-12-20 19:00:20541 SetResponse("404 OK", "hello world!");
[email protected]63534512012-12-23 18:49:00542 // In the course of processing this packet, the QuicHttpStream close itself.
[email protected]92bf17c2014-03-03 21:14:03543 ProcessPacket(ConstructResponseHeadersPacket(2, kFin));
[email protected]63534512012-12-23 18:49:00544
545 EXPECT_TRUE(AtEof());
546}
547
[email protected]1e960032013-12-20 19:00:20548TEST_P(QuicHttpStreamTest, Priority) {
549 SetRequest("GET", "/", MEDIUM);
[email protected]92bf17c2014-03-03 21:14:03550 AddWrite(ConstructRequestHeadersPacket(1, kFin));
[email protected]c5e1aca2014-01-30 04:03:04551 AddWrite(ConstructAckAndRstStreamPacket(2));
[email protected]24e5bc52013-09-18 15:36:58552 use_closing_stream_ = true;
553 Initialize();
554
555 request_.method = "GET";
556 request_.url = GURL("http://www.google.com/");
557
558 EXPECT_EQ(OK, stream_->InitializeStream(&request_, MEDIUM,
559 net_log_, callback_.callback()));
560
561 // Check that priority is highest.
562 QuicReliableClientStream* reliable_stream =
563 QuicHttpStreamPeer::GetQuicReliableClientStream(stream_.get());
564 DCHECK(reliable_stream);
[email protected]9f0dcd4e2014-01-16 15:58:14565 DCHECK_EQ(QuicWriteBlockedList::kHighestPriority,
[email protected]24e5bc52013-09-18 15:36:58566 reliable_stream->EffectivePriority());
567
568 EXPECT_EQ(OK, stream_->SendRequest(headers_, &response_,
569 callback_.callback()));
[email protected]24e5bc52013-09-18 15:36:58570
571 // Check that priority has now dropped back to MEDIUM.
572 DCHECK_EQ(MEDIUM, ConvertQuicPriorityToRequestPriority(
573 reliable_stream->EffectivePriority()));
574
575 // Ack the request.
[email protected]1e960032013-12-20 19:00:20576 ProcessPacket(ConstructAckPacket(1, 0, 0));
[email protected]24e5bc52013-09-18 15:36:58577 EXPECT_EQ(ERR_IO_PENDING,
578 stream_->ReadResponseHeaders(callback_.callback()));
579
580 // Send the response with a body.
[email protected]1e960032013-12-20 19:00:20581 SetResponse("404 OK", "hello world!");
[email protected]24e5bc52013-09-18 15:36:58582 // In the course of processing this packet, the QuicHttpStream close itself.
[email protected]92bf17c2014-03-03 21:14:03583 ProcessPacket(ConstructResponseHeadersPacket(2, kFin));
[email protected]24e5bc52013-09-18 15:36:58584
585 EXPECT_TRUE(AtEof());
586}
587
[email protected]e1cca9a2013-09-20 17:14:44588// Regression test for http://crbug.com/294870
[email protected]1e960032013-12-20 19:00:20589TEST_P(QuicHttpStreamTest, CheckPriorityWithNoDelegate) {
590 SetRequest("GET", "/", MEDIUM);
[email protected]e1cca9a2013-09-20 17:14:44591 use_closing_stream_ = true;
[email protected]459a7402014-02-10 12:58:52592
[email protected]08da9adb2014-04-24 08:33:31593 AddWrite(ConstructRstStreamPacket(1));
[email protected]459a7402014-02-10 12:58:52594
[email protected]e1cca9a2013-09-20 17:14:44595 Initialize();
596
597 request_.method = "GET";
598 request_.url = GURL("http://www.google.com/");
599
600 EXPECT_EQ(OK, stream_->InitializeStream(&request_, MEDIUM,
601 net_log_, callback_.callback()));
602
603 // Check that priority is highest.
604 QuicReliableClientStream* reliable_stream =
605 QuicHttpStreamPeer::GetQuicReliableClientStream(stream_.get());
606 DCHECK(reliable_stream);
607 QuicReliableClientStream::Delegate* delegate = reliable_stream->GetDelegate();
608 DCHECK(delegate);
[email protected]9f0dcd4e2014-01-16 15:58:14609 DCHECK_EQ(QuicWriteBlockedList::kHighestPriority,
[email protected]e1cca9a2013-09-20 17:14:44610 reliable_stream->EffectivePriority());
611
612 // Set Delegate to NULL and make sure EffectivePriority returns highest
613 // priority.
614 reliable_stream->SetDelegate(NULL);
[email protected]9f0dcd4e2014-01-16 15:58:14615 DCHECK_EQ(QuicWriteBlockedList::kHighestPriority,
[email protected]e1cca9a2013-09-20 17:14:44616 reliable_stream->EffectivePriority());
617 reliable_stream->SetDelegate(delegate);
618}
619
[email protected]f702d572012-12-04 15:56:20620} // namespace test
[email protected]f702d572012-12-04 15:56:20621} // namespace net