blob: 2b854179ce02f83c3d536ff3bf8aa045f4004dbd [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]3021a5f2014-07-23 01:40:40211 &crypto_client_stream_factory_,
[email protected]89f79a42014-03-19 11:30:01212 make_scoped_ptr((QuicServerInfo*)NULL),
[email protected]3021a5f2014-07-23 01:40:40213 QuicServerId(kServerHostname, kServerPort,
214 false, PRIVACY_MODE_DISABLED),
215 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]3021a5f2014-07-23 01:40:40219 session_->InitializeSession();
[email protected]ed3fc15d2013-03-08 18:37:44220 session_->GetCryptoStream()->CryptoConnect();
[email protected]8ba81212013-05-03 13:11:48221 EXPECT_TRUE(session_->IsCryptoHandshakeConfirmed());
[email protected]6cca996b2013-01-25 07:43:36222 stream_.reset(use_closing_stream_ ?
[email protected]0b2294d32013-08-02 00:46:36223 new AutoClosingStream(session_->GetWeakPtr()) :
224 new QuicHttpStream(session_->GetWeakPtr()));
[email protected]98a9d1252014-04-04 00:43:59225 clock_.AdvanceTime(QuicTime::Delta::FromMilliseconds(20));
[email protected]6cca996b2013-01-25 07:43:36226 }
227
[email protected]1e960032013-12-20 19:00:20228 void SetRequest(const std::string& method,
229 const std::string& path,
230 RequestPriority priority) {
231 request_headers_ = maker_.GetRequestHeaders(method, "http", path);
[email protected]6cca996b2013-01-25 07:43:36232 }
233
[email protected]1e960032013-12-20 19:00:20234 void SetResponse(const std::string& status, const std::string& body) {
235 response_headers_ = maker_.GetResponseHeaders(status);
[email protected]92bf17c2014-03-03 21:14:03236 response_data_ = body;
[email protected]6cca996b2013-01-25 07:43:36237 }
[email protected]f702d572012-12-04 15:56:20238
[email protected]1e960032013-12-20 19:00:20239 scoped_ptr<QuicEncryptedPacket> ConstructDataPacket(
[email protected]f702d572012-12-04 15:56:20240 QuicPacketSequenceNumber sequence_number,
[email protected]e8ff26842013-03-22 21:02:05241 bool should_include_version,
[email protected]f702d572012-12-04 15:56:20242 bool fin,
243 QuicStreamOffset offset,
244 base::StringPiece data) {
[email protected]1e960032013-12-20 19:00:20245 return maker_.MakeDataPacket(
246 sequence_number, stream_id_, should_include_version, fin, offset, data);
[email protected]f702d572012-12-04 15:56:20247 }
248
[email protected]1e960032013-12-20 19:00:20249 scoped_ptr<QuicEncryptedPacket> ConstructRequestHeadersPacket(
250 QuicPacketSequenceNumber sequence_number,
251 bool fin) {
252 return maker_.MakeRequestHeadersPacket(
253 sequence_number, stream_id_, kIncludeVersion, fin, request_headers_);
254 }
255
256 scoped_ptr<QuicEncryptedPacket> ConstructResponseHeadersPacket(
257 QuicPacketSequenceNumber sequence_number,
258 bool fin) {
259 return maker_.MakeResponseHeadersPacket(
260 sequence_number, stream_id_, !kIncludeVersion, fin, response_headers_);
261 }
262
263 scoped_ptr<QuicEncryptedPacket> ConstructRstStreamPacket(
[email protected]06ff5152013-08-29 01:03:05264 QuicPacketSequenceNumber sequence_number) {
[email protected]1e960032013-12-20 19:00:20265 return maker_.MakeRstPacket(
[email protected]51cc1342014-04-18 23:44:37266 sequence_number, true, stream_id_,
267 AdjustErrorForVersion(QUIC_RST_FLOW_CONTROL_ACCOUNTING, GetParam()));
[email protected]06ff5152013-08-29 01:03:05268 }
269
[email protected]c5e1aca2014-01-30 04:03:04270 scoped_ptr<QuicEncryptedPacket> ConstructAckAndRstStreamPacket(
271 QuicPacketSequenceNumber sequence_number) {
272 return maker_.MakeAckAndRstPacket(
273 sequence_number, !kIncludeVersion, stream_id_, QUIC_STREAM_CANCELLED,
[email protected]08da9adb2014-04-24 08:33:31274 2, 1, !kIncludeCongestionFeedback);
[email protected]c5e1aca2014-01-30 04:03:04275 }
276
[email protected]1e960032013-12-20 19:00:20277 scoped_ptr<QuicEncryptedPacket> ConstructAckPacket(
[email protected]f702d572012-12-04 15:56:20278 QuicPacketSequenceNumber sequence_number,
[email protected]63534512012-12-23 18:49:00279 QuicPacketSequenceNumber largest_received,
280 QuicPacketSequenceNumber least_unacked) {
[email protected]1e960032013-12-20 19:00:20281 return maker_.MakeAckPacket(sequence_number, largest_received,
282 least_unacked, !kIncludeCongestionFeedback);
[email protected]63534512012-12-23 18:49:00283 }
284
[email protected]f702d572012-12-04 15:56:20285 BoundNetLog net_log_;
[email protected]63534512012-12-23 18:49:00286 bool use_closing_stream_;
[email protected]fee17f72013-02-03 07:47:41287 MockSendAlgorithm* send_algorithm_;
288 TestReceiveAlgorithm* receive_algorithm_;
[email protected]f702d572012-12-04 15:56:20289 scoped_refptr<TestTaskRunner> runner_;
[email protected]4356f0f2013-04-07 00:58:17290 scoped_ptr<MockWrite[]> mock_writes_;
[email protected]f702d572012-12-04 15:56:20291 MockClock clock_;
292 TestQuicConnection* connection_;
[email protected]2cfc6bb82013-10-27 03:40:44293 scoped_ptr<QuicConnectionHelper> helper_;
[email protected]f702d572012-12-04 15:56:20294 testing::StrictMock<MockConnectionVisitor> visitor_;
295 scoped_ptr<QuicHttpStream> stream_;
[email protected]cbd731e2013-10-24 00:20:39296 scoped_ptr<QuicDefaultPacketWriter> writer_;
[email protected]f702d572012-12-04 15:56:20297 scoped_ptr<QuicClientSession> session_;
[email protected]ef95114d2013-04-17 17:57:01298 QuicCryptoClientConfig crypto_config_;
[email protected]f702d572012-12-04 15:56:20299 TestCompletionCallback callback_;
300 HttpRequestInfo request_;
301 HttpRequestHeaders headers_;
302 HttpResponseInfo response_;
303 scoped_refptr<IOBufferWithSize> read_buffer_;
[email protected]1e960032013-12-20 19:00:20304 SpdyHeaderBlock request_headers_;
305 SpdyHeaderBlock response_headers_;
[email protected]6cca996b2013-01-25 07:43:36306 std::string request_data_;
307 std::string response_data_;
[email protected]f702d572012-12-04 15:56:20308
309 private:
[email protected]3aa9ca72014-02-27 19:39:43310 const QuicConnectionId connection_id_;
[email protected]1e960032013-12-20 19:00:20311 const QuicStreamId stream_id_;
312 QuicTestPacketMaker maker_;
[email protected]f702d572012-12-04 15:56:20313 IPEndPoint self_addr_;
314 IPEndPoint peer_addr_;
[email protected]457d6952013-12-13 09:24:58315 MockRandom random_generator_;
[email protected]e8ff26842013-03-22 21:02:05316 MockCryptoClientStreamFactory crypto_client_stream_factory_;
[email protected]f702d572012-12-04 15:56:20317 scoped_ptr<StaticSocketDataProvider> socket_data_;
318 std::vector<PacketToWrite> writes_;
319};
320
[email protected]1e960032013-12-20 19:00:20321INSTANTIATE_TEST_CASE_P(Version, QuicHttpStreamTest,
322 ::testing::ValuesIn(QuicSupportedVersions()));
323
324TEST_P(QuicHttpStreamTest, RenewStreamForAuth) {
[email protected]ed3fc15d2013-03-08 18:37:44325 Initialize();
[email protected]f702d572012-12-04 15:56:20326 EXPECT_EQ(NULL, stream_->RenewStreamForAuth());
327}
328
[email protected]1e960032013-12-20 19:00:20329TEST_P(QuicHttpStreamTest, CanFindEndOfResponse) {
[email protected]ed3fc15d2013-03-08 18:37:44330 Initialize();
[email protected]f702d572012-12-04 15:56:20331 EXPECT_TRUE(stream_->CanFindEndOfResponse());
332}
333
[email protected]1e960032013-12-20 19:00:20334TEST_P(QuicHttpStreamTest, IsConnectionReusable) {
[email protected]ed3fc15d2013-03-08 18:37:44335 Initialize();
[email protected]f702d572012-12-04 15:56:20336 EXPECT_FALSE(stream_->IsConnectionReusable());
337}
338
[email protected]1e960032013-12-20 19:00:20339TEST_P(QuicHttpStreamTest, GetRequest) {
340 SetRequest("GET", "/", DEFAULT_PRIORITY);
[email protected]92bf17c2014-03-03 21:14:03341 AddWrite(ConstructRequestHeadersPacket(1, kFin));
[email protected]f702d572012-12-04 15:56:20342 Initialize();
343
344 request_.method = "GET";
345 request_.url = GURL("http://www.google.com/");
346
[email protected]262eec82013-03-19 21:01:36347 EXPECT_EQ(OK, stream_->InitializeStream(&request_, DEFAULT_PRIORITY,
348 net_log_, callback_.callback()));
[email protected]f702d572012-12-04 15:56:20349 EXPECT_EQ(OK, stream_->SendRequest(headers_, &response_,
350 callback_.callback()));
[email protected]f702d572012-12-04 15:56:20351
352 // Ack the request.
[email protected]1e960032013-12-20 19:00:20353 ProcessPacket(ConstructAckPacket(1, 0, 0));
[email protected]f702d572012-12-04 15:56:20354
355 EXPECT_EQ(ERR_IO_PENDING,
356 stream_->ReadResponseHeaders(callback_.callback()));
357
[email protected]1e960032013-12-20 19:00:20358 SetResponse("404 Not Found", std::string());
[email protected]92bf17c2014-03-03 21:14:03359 ProcessPacket(ConstructResponseHeadersPacket(2, kFin));
[email protected]f702d572012-12-04 15:56:20360
361 // Now that the headers have been processed, the callback will return.
362 EXPECT_EQ(OK, callback_.WaitForResult());
[email protected]cadac622013-06-11 16:46:36363 ASSERT_TRUE(response_.headers.get());
[email protected]f702d572012-12-04 15:56:20364 EXPECT_EQ(404, response_.headers->response_code());
365 EXPECT_TRUE(response_.headers->HasHeaderValue("Content-Type", "text/plain"));
[email protected]6ed67432014-06-24 01:53:53366 EXPECT_FALSE(response_.response_time.is_null());
367 EXPECT_FALSE(response_.request_time.is_null());
[email protected]f702d572012-12-04 15:56:20368
369 // There is no body, so this should return immediately.
370 EXPECT_EQ(0, stream_->ReadResponseBody(read_buffer_.get(),
371 read_buffer_->size(),
372 callback_.callback()));
373 EXPECT_TRUE(stream_->IsResponseBodyComplete());
374 EXPECT_TRUE(AtEof());
375}
376
[email protected]3e7dca62013-09-10 16:14:23377// Regression test for http://crbug.com/288128
[email protected]1e960032013-12-20 19:00:20378TEST_P(QuicHttpStreamTest, GetRequestLargeResponse) {
379 SetRequest("GET", "/", DEFAULT_PRIORITY);
[email protected]92bf17c2014-03-03 21:14:03380 AddWrite(ConstructRequestHeadersPacket(1, kFin));
[email protected]3e7dca62013-09-10 16:14:23381 Initialize();
382
383 request_.method = "GET";
384 request_.url = GURL("http://www.google.com/");
385
386 EXPECT_EQ(OK, stream_->InitializeStream(&request_, DEFAULT_PRIORITY,
387 net_log_, callback_.callback()));
388 EXPECT_EQ(OK, stream_->SendRequest(headers_, &response_,
389 callback_.callback()));
[email protected]3e7dca62013-09-10 16:14:23390
391 // Ack the request.
[email protected]1e960032013-12-20 19:00:20392 ProcessPacket(ConstructAckPacket(1, 0, 0));
[email protected]3e7dca62013-09-10 16:14:23393
394 EXPECT_EQ(ERR_IO_PENDING,
395 stream_->ReadResponseHeaders(callback_.callback()));
396
397 SpdyHeaderBlock headers;
398 headers[":status"] = "200 OK";
399 headers[":version"] = "HTTP/1.1";
400 headers["content-type"] = "text/plain";
401 headers["big6"] = std::string(10000, 'x'); // Lots of x's.
402
403 std::string response = SpdyUtils::SerializeUncompressedHeaders(headers);
404 EXPECT_LT(4096u, response.length());
405 stream_->OnDataReceived(response.data(), response.length());
406 stream_->OnClose(QUIC_NO_ERROR);
407
408 // Now that the headers have been processed, the callback will return.
409 EXPECT_EQ(OK, callback_.WaitForResult());
410 ASSERT_TRUE(response_.headers.get());
411 EXPECT_EQ(200, response_.headers->response_code());
412 EXPECT_TRUE(response_.headers->HasHeaderValue("Content-Type", "text/plain"));
413
414 // There is no body, so this should return immediately.
415 EXPECT_EQ(0, stream_->ReadResponseBody(read_buffer_.get(),
416 read_buffer_->size(),
417 callback_.callback()));
418 EXPECT_TRUE(stream_->IsResponseBodyComplete());
419 EXPECT_TRUE(AtEof());
420}
421
[email protected]1e960032013-12-20 19:00:20422TEST_P(QuicHttpStreamTest, SendPostRequest) {
423 SetRequest("POST", "/", DEFAULT_PRIORITY);
[email protected]92bf17c2014-03-03 21:14:03424 AddWrite(ConstructRequestHeadersPacket(1, !kFin));
425 AddWrite(ConstructDataPacket(2, kIncludeVersion, kFin, 0, kUploadData));
[email protected]1e960032013-12-20 19:00:20426 AddWrite(ConstructAckPacket(3, 3, 1));
[email protected]f702d572012-12-04 15:56:20427
428 Initialize();
429
[email protected]b2d26cfd2012-12-11 10:36:06430 ScopedVector<UploadElementReader> element_readers;
431 element_readers.push_back(
432 new UploadBytesElementReader(kUploadData, strlen(kUploadData)));
[email protected]96c77a72013-09-24 09:49:20433 UploadDataStream upload_data_stream(element_readers.Pass(), 0);
[email protected]f702d572012-12-04 15:56:20434 request_.method = "POST";
435 request_.url = GURL("http://www.google.com/");
436 request_.upload_data_stream = &upload_data_stream;
[email protected]4db27d82012-12-20 11:50:24437 ASSERT_EQ(OK, request_.upload_data_stream->Init(CompletionCallback()));
[email protected]f702d572012-12-04 15:56:20438
[email protected]262eec82013-03-19 21:01:36439 EXPECT_EQ(OK, stream_->InitializeStream(&request_, DEFAULT_PRIORITY,
440 net_log_, callback_.callback()));
[email protected]f702d572012-12-04 15:56:20441 EXPECT_EQ(OK, stream_->SendRequest(headers_, &response_,
442 callback_.callback()));
[email protected]f702d572012-12-04 15:56:20443
444 // Ack both packets in the request.
[email protected]1e960032013-12-20 19:00:20445 ProcessPacket(ConstructAckPacket(1, 0, 0));
[email protected]f702d572012-12-04 15:56:20446
447 // Send the response headers (but not the body).
[email protected]1e960032013-12-20 19:00:20448 SetResponse("200 OK", std::string());
[email protected]92bf17c2014-03-03 21:14:03449 ProcessPacket(ConstructResponseHeadersPacket(2, !kFin));
[email protected]f702d572012-12-04 15:56:20450
451 // Since the headers have already arrived, this should return immediately.
452 EXPECT_EQ(OK, stream_->ReadResponseHeaders(callback_.callback()));
[email protected]cadac622013-06-11 16:46:36453 ASSERT_TRUE(response_.headers.get());
[email protected]f702d572012-12-04 15:56:20454 EXPECT_EQ(200, response_.headers->response_code());
455 EXPECT_TRUE(response_.headers->HasHeaderValue("Content-Type", "text/plain"));
456
457 // Send the response body.
458 const char kResponseBody[] = "Hello world!";
[email protected]92bf17c2014-03-03 21:14:03459 ProcessPacket(ConstructDataPacket(3, false, kFin, 0, kResponseBody));
[email protected]f702d572012-12-04 15:56:20460 // Since the body has already arrived, this should return immediately.
461 EXPECT_EQ(static_cast<int>(strlen(kResponseBody)),
462 stream_->ReadResponseBody(read_buffer_.get(), read_buffer_->size(),
463 callback_.callback()));
464
465 EXPECT_TRUE(stream_->IsResponseBodyComplete());
466 EXPECT_TRUE(AtEof());
467}
468
[email protected]1e960032013-12-20 19:00:20469TEST_P(QuicHttpStreamTest, SendChunkedPostRequest) {
470 SetRequest("POST", "/", DEFAULT_PRIORITY);
[email protected]c9e49a02013-02-26 05:56:47471 size_t chunk_size = strlen(kUploadData);
[email protected]92bf17c2014-03-03 21:14:03472 AddWrite(ConstructRequestHeadersPacket(1, !kFin));
473 AddWrite(ConstructDataPacket(2, kIncludeVersion, !kFin, 0, kUploadData));
474 AddWrite(ConstructDataPacket(3, kIncludeVersion, kFin, chunk_size,
475 kUploadData));
[email protected]1e960032013-12-20 19:00:20476 AddWrite(ConstructAckPacket(4, 3, 1));
[email protected]c9e49a02013-02-26 05:56:47477 Initialize();
478
479 UploadDataStream upload_data_stream(UploadDataStream::CHUNKED, 0);
480 upload_data_stream.AppendChunk(kUploadData, chunk_size, false);
481
482 request_.method = "POST";
483 request_.url = GURL("http://www.google.com/");
484 request_.upload_data_stream = &upload_data_stream;
485 ASSERT_EQ(OK, request_.upload_data_stream->Init(CompletionCallback()));
486
[email protected]262eec82013-03-19 21:01:36487 ASSERT_EQ(OK, stream_->InitializeStream(&request_, DEFAULT_PRIORITY,
488 net_log_, callback_.callback()));
[email protected]c9e49a02013-02-26 05:56:47489 ASSERT_EQ(ERR_IO_PENDING, stream_->SendRequest(headers_, &response_,
490 callback_.callback()));
[email protected]c9e49a02013-02-26 05:56:47491
492 upload_data_stream.AppendChunk(kUploadData, chunk_size, true);
493
494 // Ack both packets in the request.
[email protected]1e960032013-12-20 19:00:20495 ProcessPacket(ConstructAckPacket(1, 0, 0));
[email protected]c9e49a02013-02-26 05:56:47496
497 // Send the response headers (but not the body).
[email protected]1e960032013-12-20 19:00:20498 SetResponse("200 OK", std::string());
[email protected]92bf17c2014-03-03 21:14:03499 ProcessPacket(ConstructResponseHeadersPacket(2, !kFin));
[email protected]c9e49a02013-02-26 05:56:47500
501 // Since the headers have already arrived, this should return immediately.
502 ASSERT_EQ(OK, stream_->ReadResponseHeaders(callback_.callback()));
[email protected]cadac622013-06-11 16:46:36503 ASSERT_TRUE(response_.headers.get());
[email protected]c9e49a02013-02-26 05:56:47504 EXPECT_EQ(200, response_.headers->response_code());
505 EXPECT_TRUE(response_.headers->HasHeaderValue("Content-Type", "text/plain"));
506
507 // Send the response body.
508 const char kResponseBody[] = "Hello world!";
[email protected]1e960032013-12-20 19:00:20509 ProcessPacket(ConstructDataPacket(3, false, kFin, response_data_.length(),
510 kResponseBody));
[email protected]c9e49a02013-02-26 05:56:47511
512 // Since the body has already arrived, this should return immediately.
513 ASSERT_EQ(static_cast<int>(strlen(kResponseBody)),
514 stream_->ReadResponseBody(read_buffer_.get(), read_buffer_->size(),
515 callback_.callback()));
516
517 EXPECT_TRUE(stream_->IsResponseBodyComplete());
518 EXPECT_TRUE(AtEof());
519}
520
[email protected]1e960032013-12-20 19:00:20521TEST_P(QuicHttpStreamTest, DestroyedEarly) {
522 SetRequest("GET", "/", DEFAULT_PRIORITY);
[email protected]92bf17c2014-03-03 21:14:03523 AddWrite(ConstructRequestHeadersPacket(1, kFin));
[email protected]c5e1aca2014-01-30 04:03:04524 AddWrite(ConstructAckAndRstStreamPacket(2));
[email protected]63534512012-12-23 18:49:00525 use_closing_stream_ = true;
526 Initialize();
527
528 request_.method = "GET";
529 request_.url = GURL("http://www.google.com/");
530
[email protected]262eec82013-03-19 21:01:36531 EXPECT_EQ(OK, stream_->InitializeStream(&request_, DEFAULT_PRIORITY,
532 net_log_, callback_.callback()));
[email protected]63534512012-12-23 18:49:00533 EXPECT_EQ(OK, stream_->SendRequest(headers_, &response_,
[email protected]24e5bc52013-09-18 15:36:58534 callback_.callback()));
[email protected]63534512012-12-23 18:49:00535
536 // Ack the request.
[email protected]1e960032013-12-20 19:00:20537 ProcessPacket(ConstructAckPacket(1, 0, 0));
[email protected]63534512012-12-23 18:49:00538 EXPECT_EQ(ERR_IO_PENDING,
539 stream_->ReadResponseHeaders(callback_.callback()));
540
541 // Send the response with a body.
[email protected]1e960032013-12-20 19:00:20542 SetResponse("404 OK", "hello world!");
[email protected]63534512012-12-23 18:49:00543 // In the course of processing this packet, the QuicHttpStream close itself.
[email protected]92bf17c2014-03-03 21:14:03544 ProcessPacket(ConstructResponseHeadersPacket(2, kFin));
[email protected]63534512012-12-23 18:49:00545
546 EXPECT_TRUE(AtEof());
547}
548
[email protected]1e960032013-12-20 19:00:20549TEST_P(QuicHttpStreamTest, Priority) {
550 SetRequest("GET", "/", MEDIUM);
[email protected]92bf17c2014-03-03 21:14:03551 AddWrite(ConstructRequestHeadersPacket(1, kFin));
[email protected]c5e1aca2014-01-30 04:03:04552 AddWrite(ConstructAckAndRstStreamPacket(2));
[email protected]24e5bc52013-09-18 15:36:58553 use_closing_stream_ = true;
554 Initialize();
555
556 request_.method = "GET";
557 request_.url = GURL("http://www.google.com/");
558
559 EXPECT_EQ(OK, stream_->InitializeStream(&request_, MEDIUM,
560 net_log_, callback_.callback()));
561
562 // Check that priority is highest.
563 QuicReliableClientStream* reliable_stream =
564 QuicHttpStreamPeer::GetQuicReliableClientStream(stream_.get());
565 DCHECK(reliable_stream);
[email protected]9f0dcd4e2014-01-16 15:58:14566 DCHECK_EQ(QuicWriteBlockedList::kHighestPriority,
[email protected]24e5bc52013-09-18 15:36:58567 reliable_stream->EffectivePriority());
568
569 EXPECT_EQ(OK, stream_->SendRequest(headers_, &response_,
570 callback_.callback()));
[email protected]24e5bc52013-09-18 15:36:58571
572 // Check that priority has now dropped back to MEDIUM.
573 DCHECK_EQ(MEDIUM, ConvertQuicPriorityToRequestPriority(
574 reliable_stream->EffectivePriority()));
575
576 // Ack the request.
[email protected]1e960032013-12-20 19:00:20577 ProcessPacket(ConstructAckPacket(1, 0, 0));
[email protected]24e5bc52013-09-18 15:36:58578 EXPECT_EQ(ERR_IO_PENDING,
579 stream_->ReadResponseHeaders(callback_.callback()));
580
581 // Send the response with a body.
[email protected]1e960032013-12-20 19:00:20582 SetResponse("404 OK", "hello world!");
[email protected]24e5bc52013-09-18 15:36:58583 // In the course of processing this packet, the QuicHttpStream close itself.
[email protected]92bf17c2014-03-03 21:14:03584 ProcessPacket(ConstructResponseHeadersPacket(2, kFin));
[email protected]24e5bc52013-09-18 15:36:58585
586 EXPECT_TRUE(AtEof());
587}
588
[email protected]e1cca9a2013-09-20 17:14:44589// Regression test for http://crbug.com/294870
[email protected]1e960032013-12-20 19:00:20590TEST_P(QuicHttpStreamTest, CheckPriorityWithNoDelegate) {
591 SetRequest("GET", "/", MEDIUM);
[email protected]e1cca9a2013-09-20 17:14:44592 use_closing_stream_ = true;
[email protected]459a7402014-02-10 12:58:52593
[email protected]08da9adb2014-04-24 08:33:31594 AddWrite(ConstructRstStreamPacket(1));
[email protected]459a7402014-02-10 12:58:52595
[email protected]e1cca9a2013-09-20 17:14:44596 Initialize();
597
598 request_.method = "GET";
599 request_.url = GURL("http://www.google.com/");
600
601 EXPECT_EQ(OK, stream_->InitializeStream(&request_, MEDIUM,
602 net_log_, callback_.callback()));
603
604 // Check that priority is highest.
605 QuicReliableClientStream* reliable_stream =
606 QuicHttpStreamPeer::GetQuicReliableClientStream(stream_.get());
607 DCHECK(reliable_stream);
608 QuicReliableClientStream::Delegate* delegate = reliable_stream->GetDelegate();
609 DCHECK(delegate);
[email protected]9f0dcd4e2014-01-16 15:58:14610 DCHECK_EQ(QuicWriteBlockedList::kHighestPriority,
[email protected]e1cca9a2013-09-20 17:14:44611 reliable_stream->EffectivePriority());
612
613 // Set Delegate to NULL and make sure EffectivePriority returns highest
614 // priority.
615 reliable_stream->SetDelegate(NULL);
[email protected]9f0dcd4e2014-01-16 15:58:14616 DCHECK_EQ(QuicWriteBlockedList::kHighestPriority,
[email protected]e1cca9a2013-09-20 17:14:44617 reliable_stream->EffectivePriority());
618 reliable_stream->SetDelegate(delegate);
619}
620
[email protected]f702d572012-12-04 15:56:20621} // namespace test
[email protected]f702d572012-12-04 15:56:20622} // namespace net