[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 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 |
| 3 | // found in the LICENSE file. |
| 4 | |
rch | 675757b | 2016-07-29 16:40:11 | [diff] [blame] | 5 | #include "net/quic/chromium/quic_http_stream.h" |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 6 | |
sclittle | 1edeeb2 | 2015-09-02 20:46:10 | [diff] [blame] | 7 | #include <stdint.h> |
| 8 | |
danakj | ad1777e | 2016-04-16 00:56:42 | [diff] [blame] | 9 | #include <memory> |
bnc | 086b39e1 | 2016-06-24 13:05:26 | [diff] [blame] | 10 | #include <utility> |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 11 | |
danakj | ad1777e | 2016-04-16 00:56:42 | [diff] [blame] | 12 | #include "base/memory/ptr_util.h" |
fdoray | 92e35a7 | 2016-06-10 15:54:55 | [diff] [blame] | 13 | #include "base/run_loop.h" |
xunjieli | 188bd40 | 2016-03-12 00:17:25 | [diff] [blame] | 14 | #include "base/strings/string_number_conversions.h" |
gab | f767595f | 2016-05-11 18:50:35 | [diff] [blame] | 15 | #include "base/threading/thread_task_runner_handle.h" |
xunjieli | 84adaab | 2016-09-20 01:12:28 | [diff] [blame] | 16 | #include "base/time/time.h" |
mmenke | cbc2b71 | 2014-10-09 20:29:07 | [diff] [blame] | 17 | #include "net/base/chunked_upload_data_stream.h" |
| 18 | #include "net/base/elements_upload_data_stream.h" |
xunjieli | 84adaab | 2016-09-20 01:12:28 | [diff] [blame] | 19 | #include "net/base/load_timing_info.h" |
| 20 | #include "net/base/load_timing_info_test_util.h" |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 21 | #include "net/base/net_errors.h" |
| 22 | #include "net/base/test_completion_callback.h" |
[email protected] | b2d26cfd | 2012-12-11 10:36:06 | [diff] [blame] | 23 | #include "net/base/upload_bytes_element_reader.h" |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 24 | #include "net/http/http_response_headers.h" |
[email protected] | 5db45220 | 2014-08-19 05:22:15 | [diff] [blame] | 25 | #include "net/http/transport_security_state.h" |
mikecirone | 8b85c43 | 2016-09-08 19:11:00 | [diff] [blame] | 26 | #include "net/log/net_log_event_type.h" |
xunjieli | 5fafe14 | 2016-03-23 23:32:54 | [diff] [blame] | 27 | #include "net/log/test_net_log.h" |
| 28 | #include "net/log/test_net_log_util.h" |
rch | 675757b | 2016-07-29 16:40:11 | [diff] [blame] | 29 | #include "net/quic/chromium/crypto/proof_verifier_chromium.h" |
rch | 223465c | 2016-12-08 21:17:29 | [diff] [blame] | 30 | #include "net/quic/chromium/mock_crypto_client_stream_factory.h" |
rch | 675757b | 2016-07-29 16:40:11 | [diff] [blame] | 31 | #include "net/quic/chromium/quic_chromium_alarm_factory.h" |
| 32 | #include "net/quic/chromium/quic_chromium_connection_helper.h" |
| 33 | #include "net/quic/chromium/quic_chromium_packet_reader.h" |
| 34 | #include "net/quic/chromium/quic_chromium_packet_writer.h" |
rch | 223465c | 2016-12-08 21:17:29 | [diff] [blame] | 35 | #include "net/quic/chromium/quic_http_utils.h" |
rch | faf834e | 2016-12-21 17:55:02 | [diff] [blame] | 36 | #include "net/quic/chromium/quic_server_info.h" |
rch | 223465c | 2016-12-08 21:17:29 | [diff] [blame] | 37 | #include "net/quic/chromium/quic_test_packet_maker.h" |
rch | 1e36936 | 2017-04-03 19:44:51 | [diff] [blame] | 38 | #include "net/quic/chromium/test_task_runner.h" |
rch | d4db7c15 | 2016-07-29 21:58:12 | [diff] [blame] | 39 | #include "net/quic/core/congestion_control/send_algorithm_interface.h" |
| 40 | #include "net/quic/core/crypto/crypto_protocol.h" |
| 41 | #include "net/quic/core/crypto/quic_decrypter.h" |
| 42 | #include "net/quic/core/crypto/quic_encrypter.h" |
rch | d4db7c15 | 2016-07-29 21:58:12 | [diff] [blame] | 43 | #include "net/quic/core/quic_connection.h" |
rch | d4db7c15 | 2016-07-29 21:58:12 | [diff] [blame] | 44 | #include "net/quic/core/quic_write_blocked_list.h" |
| 45 | #include "net/quic/core/spdy_utils.h" |
Michael Warres | ed8ebd5 | 2017-12-22 22:42:04 | [diff] [blame] | 46 | #include "net/quic/core/tls_client_handshaker.h" |
fayang | 19c3077 | 2017-03-09 15:06:17 | [diff] [blame] | 47 | #include "net/quic/platform/api/quic_string_piece.h" |
rch | 1fe2eeb | 2015-10-26 14:45:57 | [diff] [blame] | 48 | #include "net/quic/test_tools/crypto_test_utils.h" |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 49 | #include "net/quic/test_tools/mock_clock.h" |
[email protected] | 9db44391 | 2013-02-25 05:27:03 | [diff] [blame] | 50 | #include "net/quic/test_tools/mock_random.h" |
[email protected] | b1f287d | 2012-12-22 17:25:39 | [diff] [blame] | 51 | #include "net/quic/test_tools/quic_connection_peer.h" |
ckrasic | bf2f59c | 2017-05-04 23:54:36 | [diff] [blame] | 52 | #include "net/quic/test_tools/quic_spdy_session_peer.h" |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 53 | #include "net/quic/test_tools/quic_test_utils.h" |
tbansal | ca83c00 | 2016-04-28 20:56:28 | [diff] [blame] | 54 | #include "net/socket/socket_performance_watcher.h" |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 55 | #include "net/socket/socket_test_util.h" |
bnc | 8f8f7d30 | 2017-04-24 18:08:06 | [diff] [blame] | 56 | #include "net/spdy/chromium/spdy_http_utils.h" |
| 57 | #include "net/spdy/core/spdy_frame_builder.h" |
| 58 | #include "net/spdy/core/spdy_framer.h" |
| 59 | #include "net/spdy/core/spdy_protocol.h" |
rch | 03b7a20 | 2016-02-05 00:54:20 | [diff] [blame] | 60 | #include "net/test/cert_test_util.h" |
robpercival | 214763f | 2016-07-01 23:27:01 | [diff] [blame] | 61 | #include "net/test/gtest_util.h" |
rsleevi | a69c79a | 2016-06-22 03:28:43 | [diff] [blame] | 62 | #include "net/test/test_data_directory.h" |
Ramin Halavati | 683bcaa9 | 2018-02-14 08:42:39 | [diff] [blame] | 63 | #include "net/traffic_annotation/network_traffic_annotation_test_helper.h" |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 64 | #include "testing/gmock/include/gmock/gmock.h" |
| 65 | #include "testing/gtest/include/gtest/gtest.h" |
| 66 | |
bnc | 614a92d3 | 2016-04-04 13:56:07 | [diff] [blame] | 67 | using std::string; |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 68 | using testing::_; |
[email protected] | 06ff515 | 2013-08-29 01:03:05 | [diff] [blame] | 69 | using testing::AnyNumber; |
| 70 | using testing::Return; |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 71 | |
| 72 | namespace net { |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 73 | namespace test { |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 74 | namespace { |
| 75 | |
[email protected] | 16ba774 | 2014-08-22 00:57:25 | [diff] [blame] | 76 | const char kUploadData[] = "Really nifty data!"; |
rch | 9ae5b3b | 2016-02-11 00:36:29 | [diff] [blame] | 77 | const char kDefaultServerHostName[] = "www.example.org"; |
rch | cd37901 | 2017-04-12 21:53:32 | [diff] [blame] | 78 | const uint16_t kDefaultServerPort = 443; |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 79 | |
| 80 | class TestQuicConnection : public QuicConnection { |
| 81 | public: |
Dan Zhang | fefaf5b | 2017-12-11 17:06:24 | [diff] [blame] | 82 | TestQuicConnection(const ParsedQuicVersionVector& versions, |
[email protected] | 3aa9ca7 | 2014-02-27 19:39:43 | [diff] [blame] | 83 | QuicConnectionId connection_id, |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 84 | IPEndPoint address, |
rch | 12fef55 | 2016-01-15 16:26:31 | [diff] [blame] | 85 | QuicChromiumConnectionHelper* helper, |
rch | 16c74d1d | 2016-04-22 06:14:07 | [diff] [blame] | 86 | QuicChromiumAlarmFactory* alarm_factory, |
jdorfman | 90d185f3 | 2016-01-15 13:22:47 | [diff] [blame] | 87 | QuicPacketWriter* writer) |
[email protected] | 66cd2d6 | 2014-08-01 18:42:39 | [diff] [blame] | 88 | : QuicConnection(connection_id, |
fayang | 91ca201 | 2016-11-22 07:42:46 | [diff] [blame] | 89 | QuicSocketAddress(QuicSocketAddressImpl(address)), |
[email protected] | 66cd2d6 | 2014-08-01 18:42:39 | [diff] [blame] | 90 | helper, |
rch | 16c74d1d | 2016-04-22 06:14:07 | [diff] [blame] | 91 | alarm_factory, |
jdorfman | 90d185f3 | 2016-01-15 13:22:47 | [diff] [blame] | 92 | writer, |
rtenneti | 6f48aa9 | 2015-03-16 02:18:48 | [diff] [blame] | 93 | true /* owns_writer */, |
| 94 | Perspective::IS_CLIENT, |
rtenneti | 6f48aa9 | 2015-03-16 02:18:48 | [diff] [blame] | 95 | versions) {} |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 96 | |
[email protected] | fee17f7 | 2013-02-03 07:47:41 | [diff] [blame] | 97 | void SetSendAlgorithm(SendAlgorithmInterface* send_algorithm) { |
fayang | 754ab657 | 2017-01-18 20:39:04 | [diff] [blame] | 98 | QuicConnectionPeer::SetSendAlgorithm(this, send_algorithm); |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 99 | } |
| 100 | }; |
| 101 | |
maksim.sisov | 84e20c9 | 2016-06-23 08:49:34 | [diff] [blame] | 102 | // UploadDataStream that always returns errors on data read. |
| 103 | class ReadErrorUploadDataStream : public UploadDataStream { |
| 104 | public: |
| 105 | enum class FailureMode { SYNC, ASYNC }; |
| 106 | |
| 107 | explicit ReadErrorUploadDataStream(FailureMode mode) |
| 108 | : UploadDataStream(true, 0), async_(mode), weak_factory_(this) {} |
| 109 | ~ReadErrorUploadDataStream() override {} |
| 110 | |
| 111 | private: |
| 112 | void CompleteRead() { UploadDataStream::OnReadCompleted(ERR_FAILED); } |
| 113 | |
| 114 | // UploadDataStream implementation: |
tfarina | 4283411 | 2016-09-22 13:38:20 | [diff] [blame] | 115 | int InitInternal(const NetLogWithSource& net_log) override { return OK; } |
maksim.sisov | 84e20c9 | 2016-06-23 08:49:34 | [diff] [blame] | 116 | |
| 117 | int ReadInternal(IOBuffer* buf, int buf_len) override { |
| 118 | if (async_ == FailureMode::ASYNC) { |
| 119 | base::ThreadTaskRunnerHandle::Get()->PostTask( |
| 120 | FROM_HERE, base::Bind(&ReadErrorUploadDataStream::CompleteRead, |
| 121 | weak_factory_.GetWeakPtr())); |
| 122 | return ERR_IO_PENDING; |
| 123 | } |
| 124 | return ERR_FAILED; |
| 125 | } |
| 126 | |
| 127 | void ResetInternal() override {} |
| 128 | |
| 129 | const FailureMode async_; |
| 130 | |
| 131 | base::WeakPtrFactory<ReadErrorUploadDataStream> weak_factory_; |
| 132 | |
| 133 | DISALLOW_COPY_AND_ASSIGN(ReadErrorUploadDataStream); |
| 134 | }; |
| 135 | |
xunjieli | 8dff50b | 2016-07-22 14:19:06 | [diff] [blame] | 136 | // A Callback that deletes the QuicHttpStream. |
| 137 | class DeleteStreamCallback : public TestCompletionCallbackBase { |
| 138 | public: |
| 139 | DeleteStreamCallback(std::unique_ptr<QuicHttpStream> stream) |
| 140 | : stream_(std::move(stream)), |
| 141 | callback_(base::Bind(&DeleteStreamCallback::DeleteStream, |
| 142 | base::Unretained(this))) {} |
| 143 | |
| 144 | const CompletionCallback& callback() const { return callback_; } |
| 145 | |
| 146 | private: |
| 147 | void DeleteStream(int result) { |
| 148 | stream_.reset(); |
| 149 | SetResult(result); |
| 150 | } |
| 151 | |
| 152 | std::unique_ptr<QuicHttpStream> stream_; |
| 153 | CompletionCallback callback_; |
| 154 | }; |
| 155 | |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 156 | } // namespace |
| 157 | |
[email protected] | 24e5bc5 | 2013-09-18 15:36:58 | [diff] [blame] | 158 | class QuicHttpStreamPeer { |
| 159 | public: |
rch | 08e19857 | 2017-05-09 16:56:55 | [diff] [blame] | 160 | static QuicChromiumClientStream::Handle* GetQuicChromiumClientStream( |
[email protected] | 24e5bc5 | 2013-09-18 15:36:58 | [diff] [blame] | 161 | QuicHttpStream* stream) { |
rch | 08e19857 | 2017-05-09 16:56:55 | [diff] [blame] | 162 | return stream->stream_.get(); |
[email protected] | 24e5bc5 | 2013-09-18 15:36:58 | [diff] [blame] | 163 | } |
| 164 | }; |
| 165 | |
Michael Warres | 74ee3ce | 2017-10-09 15:26:37 | [diff] [blame] | 166 | class QuicHttpStreamTest |
Yixin Wang | 079ad54 | 2018-01-11 04:06:05 | [diff] [blame] | 167 | : public ::testing::TestWithParam<std::tuple<QuicTransportVersion, bool>> { |
rch | fb47f71 | 2017-05-21 03:24:00 | [diff] [blame] | 168 | public: |
| 169 | void CloseStream(QuicHttpStream* stream, int /*rv*/) { stream->Close(false); } |
| 170 | |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 171 | protected: |
[email protected] | 1e96003 | 2013-12-20 19:00:20 | [diff] [blame] | 172 | static const bool kFin = true; |
| 173 | static const bool kIncludeVersion = true; |
| 174 | static const bool kIncludeCongestionFeedback = true; |
| 175 | |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 176 | // Holds a packet to be written to the wire, and the IO mode that should |
| 177 | // be used by the mock socket when performing the write. |
| 178 | struct PacketToWrite { |
jokulik | f2bd55c5 | 2016-03-24 22:35:30 | [diff] [blame] | 179 | PacketToWrite(IoMode mode, QuicReceivedPacket* packet) |
rjshade | d5ced07 | 2015-12-18 19:26:02 | [diff] [blame] | 180 | : mode(mode), packet(packet) {} |
rtenneti | 15656ae | 2016-01-23 03:05:03 | [diff] [blame] | 181 | PacketToWrite(IoMode mode, int rv) : mode(mode), packet(nullptr), rv(rv) {} |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 182 | IoMode mode; |
jokulik | f2bd55c5 | 2016-03-24 22:35:30 | [diff] [blame] | 183 | QuicReceivedPacket* packet; |
rtenneti | 15656ae | 2016-01-23 03:05:03 | [diff] [blame] | 184 | int rv; |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 185 | }; |
| 186 | |
| 187 | QuicHttpStreamTest() |
Yixin Wang | 079ad54 | 2018-01-11 04:06:05 | [diff] [blame] | 188 | : version_(std::get<0>(GetParam())), |
| 189 | client_headers_include_h2_stream_dependency_(std::get<1>(GetParam())), |
| 190 | crypto_config_(crypto_test_utils::ProofVerifierForTesting(), |
Michael Warres | ed8ebd5 | 2017-12-22 22:42:04 | [diff] [blame] | 191 | TlsClientHandshaker::CreateSslCtx()), |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 192 | read_buffer_(new IOBufferWithSize(4096)), |
ckrasic | bf2f59c | 2017-05-04 23:54:36 | [diff] [blame] | 193 | promise_id_(GetNthServerInitiatedStreamId(0)), |
| 194 | stream_id_(GetNthClientInitiatedStreamId(0)), |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 195 | connection_id_(2), |
Yixin Wang | 079ad54 | 2018-01-11 04:06:05 | [diff] [blame] | 196 | client_maker_(version_, |
alyssar | 2adf3ac | 2016-05-03 17:12:58 | [diff] [blame] | 197 | connection_id_, |
| 198 | &clock_, |
| 199 | kDefaultServerHostName, |
Yixin Wang | 079ad54 | 2018-01-11 04:06:05 | [diff] [blame] | 200 | Perspective::IS_CLIENT, |
| 201 | client_headers_include_h2_stream_dependency_), |
| 202 | server_maker_(version_, |
zhongyi | 5dbfdd4 | 2017-06-20 05:22:15 | [diff] [blame] | 203 | connection_id_, |
alyssar | 2adf3ac | 2016-05-03 17:12:58 | [diff] [blame] | 204 | &clock_, |
| 205 | kDefaultServerHostName, |
Yixin Wang | 079ad54 | 2018-01-11 04:06:05 | [diff] [blame] | 206 | Perspective::IS_SERVER, |
| 207 | false), |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 208 | random_generator_(0), |
| 209 | response_offset_(0) { |
martijn | 21968ea | 2016-02-24 18:46:20 | [diff] [blame] | 210 | IPAddress ip(192, 0, 2, 33); |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 211 | peer_addr_ = IPEndPoint(ip, 443); |
| 212 | self_addr_ = IPEndPoint(ip, 8435); |
rtenneti | 4b06ae7 | 2014-08-26 03:43:43 | [diff] [blame] | 213 | clock_.AdvanceTime(QuicTime::Delta::FromMilliseconds(20)); |
Ramin Halavati | 683bcaa9 | 2018-02-14 08:42:39 | [diff] [blame] | 214 | request_.traffic_annotation = |
| 215 | MutableNetworkTrafficAnnotationTag(TRAFFIC_ANNOTATION_FOR_TESTS); |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 216 | } |
| 217 | |
| 218 | ~QuicHttpStreamTest() { |
rtenneti | 85dcfac2 | 2015-03-27 20:22:19 | [diff] [blame] | 219 | session_->CloseSessionOnError(ERR_ABORTED, QUIC_INTERNAL_ERROR); |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 220 | for (size_t i = 0; i < writes_.size(); i++) { |
| 221 | delete writes_[i].packet; |
| 222 | } |
| 223 | } |
| 224 | |
| 225 | // Adds a packet to the list of expected writes. |
danakj | ad1777e | 2016-04-16 00:56:42 | [diff] [blame] | 226 | void AddWrite(std::unique_ptr<QuicReceivedPacket> packet) { |
[email protected] | 1e96003 | 2013-12-20 19:00:20 | [diff] [blame] | 227 | writes_.push_back(PacketToWrite(SYNCHRONOUS, packet.release())); |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 228 | } |
| 229 | |
rtenneti | 15656ae | 2016-01-23 03:05:03 | [diff] [blame] | 230 | void AddWrite(IoMode mode, int rv) { |
| 231 | writes_.push_back(PacketToWrite(mode, rv)); |
| 232 | } |
| 233 | |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 234 | // Returns the packet to be written at position |pos|. |
jokulik | f2bd55c5 | 2016-03-24 22:35:30 | [diff] [blame] | 235 | QuicReceivedPacket* GetWrite(size_t pos) { return writes_[pos].packet; } |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 236 | |
| 237 | bool AtEof() { |
rch | 37de576c | 2015-05-17 20:28:17 | [diff] [blame] | 238 | return socket_data_->AllReadDataConsumed() && |
| 239 | socket_data_->AllWriteDataConsumed(); |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 240 | } |
| 241 | |
danakj | ad1777e | 2016-04-16 00:56:42 | [diff] [blame] | 242 | void ProcessPacket(std::unique_ptr<QuicReceivedPacket> packet) { |
fayang | 91ca201 | 2016-11-22 07:42:46 | [diff] [blame] | 243 | connection_->ProcessUdpPacket( |
| 244 | QuicSocketAddress(QuicSocketAddressImpl(self_addr_)), |
| 245 | QuicSocketAddress(QuicSocketAddressImpl(peer_addr_)), *packet); |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 246 | } |
| 247 | |
| 248 | // Configures the test fixture to use the list of expected writes. |
| 249 | void Initialize() { |
| 250 | mock_writes_.reset(new MockWrite[writes_.size()]); |
| 251 | for (size_t i = 0; i < writes_.size(); i++) { |
rtenneti | 15656ae | 2016-01-23 03:05:03 | [diff] [blame] | 252 | if (writes_[i].packet == nullptr) { |
| 253 | mock_writes_[i] = MockWrite(writes_[i].mode, writes_[i].rv, i); |
| 254 | } else { |
| 255 | mock_writes_[i] = MockWrite(writes_[i].mode, writes_[i].packet->data(), |
| 256 | writes_[i].packet->length()); |
| 257 | } |
bnc | 614a92d3 | 2016-04-04 13:56:07 | [diff] [blame] | 258 | } |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 259 | |
rtenneti | be63573 | 2014-10-02 22:51:42 | [diff] [blame] | 260 | socket_data_.reset(new StaticSocketDataProvider( |
| 261 | nullptr, 0, mock_writes_.get(), writes_.size())); |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 262 | |
danakj | ad1777e | 2016-04-16 00:56:42 | [diff] [blame] | 263 | std::unique_ptr<MockUDPClientSocket> socket(new MockUDPClientSocket( |
xunjieli | b53b38c | 2016-03-24 15:54:36 | [diff] [blame] | 264 | socket_data_.get(), net_log_.bound().net_log())); |
[email protected] | e13201d8 | 2012-12-12 05:00:32 | [diff] [blame] | 265 | socket->Connect(peer_addr_); |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 266 | runner_ = new TestTaskRunner(&clock_); |
[email protected] | fee17f7 | 2013-02-03 07:47:41 | [diff] [blame] | 267 | send_algorithm_ = new MockSendAlgorithm(); |
rch | 7dd1570 | 2015-07-01 18:57:57 | [diff] [blame] | 268 | EXPECT_CALL(*send_algorithm_, InRecovery()).WillRepeatedly(Return(false)); |
| 269 | EXPECT_CALL(*send_algorithm_, InSlowStart()).WillRepeatedly(Return(false)); |
rjshade | d5ced07 | 2015-12-18 19:26:02 | [diff] [blame] | 270 | EXPECT_CALL(*send_algorithm_, OnPacketSent(_, _, _, _, _)) |
Dan Zhang | f1147017 | 2017-09-18 22:02:09 | [diff] [blame] | 271 | .Times(testing::AtLeast(1)); |
rtenneti | 44f4a2e | 2015-08-07 14:00:07 | [diff] [blame] | 272 | EXPECT_CALL(*send_algorithm_, GetCongestionWindow()) |
| 273 | .WillRepeatedly(Return(kMaxPacketSize)); |
jokulik | 0e0a00c3 | 2016-06-13 21:51:58 | [diff] [blame] | 274 | EXPECT_CALL(*send_algorithm_, PacingRate(_)) |
rtenneti | 44f4a2e | 2015-08-07 14:00:07 | [diff] [blame] | 275 | .WillRepeatedly(Return(QuicBandwidth::Zero())); |
Michael Warres | 74ee3ce | 2017-10-09 15:26:37 | [diff] [blame] | 276 | EXPECT_CALL(*send_algorithm_, CanSend(_)).WillRepeatedly(Return(true)); |
rtenneti | 44f4a2e | 2015-08-07 14:00:07 | [diff] [blame] | 277 | EXPECT_CALL(*send_algorithm_, BandwidthEstimate()) |
| 278 | .WillRepeatedly(Return(QuicBandwidth::Zero())); |
rch | 1e543ec | 2015-03-29 07:04:40 | [diff] [blame] | 279 | EXPECT_CALL(*send_algorithm_, SetFromConfig(_, _)).Times(AnyNumber()); |
rch | 08cae03 | 2017-03-29 00:59:15 | [diff] [blame] | 280 | EXPECT_CALL(*send_algorithm_, OnApplicationLimited(_)).Times(AnyNumber()); |
rch | 0d5bcf6 | 2017-05-24 22:48:45 | [diff] [blame] | 281 | EXPECT_CALL(*send_algorithm_, GetCongestionControlType()) |
| 282 | .Times(AnyNumber()); |
rch | 16c74d1d | 2016-04-22 06:14:07 | [diff] [blame] | 283 | helper_.reset( |
| 284 | new QuicChromiumConnectionHelper(&clock_, &random_generator_)); |
| 285 | alarm_factory_.reset(new QuicChromiumAlarmFactory(runner_.get(), &clock_)); |
| 286 | |
Michael Warres | 74ee3ce | 2017-10-09 15:26:37 | [diff] [blame] | 287 | connection_ = new TestQuicConnection( |
Dan Zhang | fefaf5b | 2017-12-11 17:06:24 | [diff] [blame] | 288 | SupportedVersions( |
Yixin Wang | 079ad54 | 2018-01-11 04:06:05 | [diff] [blame] | 289 | net::ParsedQuicVersion(net::PROTOCOL_QUIC_CRYPTO, version_)), |
Dan Zhang | fefaf5b | 2017-12-11 17:06:24 | [diff] [blame] | 290 | connection_id_, peer_addr_, helper_.get(), alarm_factory_.get(), |
Ryan Hamilton | 9edcf1a | 2017-11-22 05:55:17 | [diff] [blame] | 291 | new QuicChromiumPacketWriter( |
| 292 | socket.get(), base::ThreadTaskRunnerHandle::Get().get())); |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 293 | connection_->set_visitor(&visitor_); |
[email protected] | fee17f7 | 2013-02-03 07:47:41 | [diff] [blame] | 294 | connection_->SetSendAlgorithm(send_algorithm_); |
rch | 03b7a20 | 2016-02-05 00:54:20 | [diff] [blame] | 295 | |
| 296 | // Load a certificate that is valid for *.example.org |
| 297 | scoped_refptr<X509Certificate> test_cert( |
| 298 | ImportCertFromFile(GetTestCertsDirectory(), "wildcard.pem")); |
| 299 | EXPECT_TRUE(test_cert.get()); |
| 300 | |
| 301 | verify_details_.cert_verify_result.verified_cert = test_cert; |
| 302 | verify_details_.cert_verify_result.is_issued_by_known_root = true; |
| 303 | crypto_client_stream_factory_.AddProofVerifyDetails(&verify_details_); |
| 304 | |
xunjieli | 84adaab | 2016-09-20 01:12:28 | [diff] [blame] | 305 | base::TimeTicks dns_end = base::TimeTicks::Now(); |
| 306 | base::TimeTicks dns_start = dns_end - base::TimeDelta::FromMilliseconds(1); |
ckrasic | 4f9d88d | 2015-07-22 22:23:16 | [diff] [blame] | 307 | session_.reset(new QuicChromiumClientSession( |
xunjieli | b53b38c | 2016-03-24 15:54:36 | [diff] [blame] | 308 | connection_, std::move(socket), |
rtenneti | 1cd3b16 | 2015-09-29 02:58:28 | [diff] [blame] | 309 | /*stream_factory=*/nullptr, &crypto_client_stream_factory_, &clock_, |
bnc | 614a92d3 | 2016-04-04 13:56:07 | [diff] [blame] | 310 | &transport_security_state_, |
danakj | ad1777e | 2016-04-16 00:56:42 | [diff] [blame] | 311 | base::WrapUnique(static_cast<QuicServerInfo*>(nullptr)), |
Paul Jensen | 8e3c5d3 | 2018-02-19 17:06:33 | [diff] [blame^] | 312 | QuicSessionKey(kDefaultServerHostName, kDefaultServerPort, |
| 313 | PRIVACY_MODE_DISABLED, SocketTag()), |
Zhongyi Shi | 22e91af | 2017-10-21 00:53:57 | [diff] [blame] | 314 | /*require_confirmation=*/false, /*migrate_session_early*/ false, |
Zhongyi Shi | 5f587cc | 2017-11-21 23:24:17 | [diff] [blame] | 315 | /*migrate_session_on_network_change*/ false, |
Zhongyi Shi | f4683a3 | 2017-12-01 00:03:28 | [diff] [blame] | 316 | /*migrate_session_early_v2*/ false, |
Zhongyi Shi | 5f587cc | 2017-11-21 23:24:17 | [diff] [blame] | 317 | /*migrate_session_on_network_change_v2*/ false, |
Zhongyi Shi | 73f23ca87 | 2017-12-13 18:37:13 | [diff] [blame] | 318 | base::TimeDelta::FromSeconds(kMaxTimeOnNonDefaultNetworkSecs), |
Zhongyi Shi | 8b1e43f | 2017-12-13 20:46:30 | [diff] [blame] | 319 | kMaxMigrationsToNonDefaultNetworkOnPathDegrading, |
Zhongyi Shi | 5f587cc | 2017-11-21 23:24:17 | [diff] [blame] | 320 | kQuicYieldAfterPacketsRead, |
rtenneti | 1cd3b16 | 2015-09-29 02:58:28 | [diff] [blame] | 321 | QuicTime::Delta::FromMilliseconds(kQuicYieldAfterDurationMilliseconds), |
Yixin Wang | 079ad54 | 2018-01-11 04:06:05 | [diff] [blame] | 322 | client_headers_include_h2_stream_dependency_, /*cert_verify_flags=*/0, |
| 323 | DefaultQuicConfig(), &crypto_config_, "CONNECTION_UNKNOWN", dns_start, |
| 324 | dns_end, &push_promise_index_, nullptr, |
xunjieli | 84adaab | 2016-09-20 01:12:28 | [diff] [blame] | 325 | base::ThreadTaskRunnerHandle::Get().get(), |
xunjieli | 5fafe14 | 2016-03-23 23:32:54 | [diff] [blame] | 326 | /*socket_performance_watcher=*/nullptr, net_log_.bound().net_log())); |
rtenneti | d39bd76 | 2015-06-12 01:05:52 | [diff] [blame] | 327 | session_->Initialize(); |
xunjieli | 100937eb5 | 2016-09-15 20:09:37 | [diff] [blame] | 328 | TestCompletionCallback callback; |
rch | f0b18c8a | 2017-05-05 19:31:57 | [diff] [blame] | 329 | |
rch | 433bf5f | 2017-02-14 04:10:47 | [diff] [blame] | 330 | session_->CryptoConnect(callback.callback()); |
Ryan Hamilton | 6c2a2a8 | 2017-12-15 02:06:28 | [diff] [blame] | 331 | stream_ = std::make_unique<QuicHttpStream>( |
| 332 | session_->CreateHandle(HostPortPair("www.example.org", 443))); |
| 333 | promised_stream_ = std::make_unique<QuicHttpStream>( |
| 334 | session_->CreateHandle(HostPortPair("www.example.org", 443))); |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 335 | push_promise_[":path"] = "/bar"; |
| 336 | push_promise_[":authority"] = "www.example.org"; |
| 337 | push_promise_[":version"] = "HTTP/1.1"; |
| 338 | push_promise_[":method"] = "GET"; |
| 339 | push_promise_[":scheme"] = "https"; |
| 340 | |
| 341 | promised_response_[":status"] = "200 OK"; |
| 342 | promised_response_[":version"] = "HTTP/1.1"; |
| 343 | promised_response_["content-type"] = "text/plain"; |
| 344 | |
Yixin Wang | 81c1118 | 2018-02-05 20:09:57 | [diff] [blame] | 345 | promise_url_ = SpdyUtils::GetPromisedUrlFromHeaderBlock(push_promise_); |
[email protected] | 6cca996b | 2013-01-25 07:43:36 | [diff] [blame] | 346 | } |
| 347 | |
bnc | 614a92d3 | 2016-04-04 13:56:07 | [diff] [blame] | 348 | void SetRequest(const string& method, |
| 349 | const string& path, |
[email protected] | 1e96003 | 2013-12-20 19:00:20 | [diff] [blame] | 350 | RequestPriority priority) { |
rch | cd37901 | 2017-04-12 21:53:32 | [diff] [blame] | 351 | request_headers_ = client_maker_.GetRequestHeaders(method, "https", path); |
[email protected] | 6cca996b | 2013-01-25 07:43:36 | [diff] [blame] | 352 | } |
| 353 | |
bnc | 614a92d3 | 2016-04-04 13:56:07 | [diff] [blame] | 354 | void SetResponse(const string& status, const string& body) { |
alyssar | 2adf3ac | 2016-05-03 17:12:58 | [diff] [blame] | 355 | response_headers_ = server_maker_.GetResponseHeaders(status); |
[email protected] | 92bf17c | 2014-03-03 21:14:03 | [diff] [blame] | 356 | response_data_ = body; |
[email protected] | 6cca996b | 2013-01-25 07:43:36 | [diff] [blame] | 357 | } |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 358 | |
danakj | ad1777e | 2016-04-16 00:56:42 | [diff] [blame] | 359 | std::unique_ptr<QuicReceivedPacket> InnerConstructDataPacket( |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 360 | QuicPacketNumber packet_number, |
| 361 | QuicStreamId stream_id, |
| 362 | bool should_include_version, |
| 363 | bool fin, |
| 364 | QuicStreamOffset offset, |
fayang | 19c3077 | 2017-03-09 15:06:17 | [diff] [blame] | 365 | QuicStringPiece data, |
alyssar | 2adf3ac | 2016-05-03 17:12:58 | [diff] [blame] | 366 | QuicTestPacketMaker* maker) { |
| 367 | return maker->MakeDataPacket(packet_number, stream_id, |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 368 | should_include_version, fin, offset, data); |
| 369 | } |
| 370 | |
alyssar | 2adf3ac | 2016-05-03 17:12:58 | [diff] [blame] | 371 | std::unique_ptr<QuicReceivedPacket> ConstructClientDataPacket( |
rtenneti | a004d33 | 2015-08-28 06:44:57 | [diff] [blame] | 372 | QuicPacketNumber packet_number, |
[email protected] | e8ff2684 | 2013-03-22 21:02:05 | [diff] [blame] | 373 | bool should_include_version, |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 374 | bool fin, |
| 375 | QuicStreamOffset offset, |
fayang | 19c3077 | 2017-03-09 15:06:17 | [diff] [blame] | 376 | QuicStringPiece data) { |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 377 | return InnerConstructDataPacket(packet_number, stream_id_, |
alyssar | 2adf3ac | 2016-05-03 17:12:58 | [diff] [blame] | 378 | should_include_version, fin, offset, data, |
| 379 | &client_maker_); |
| 380 | } |
| 381 | |
| 382 | std::unique_ptr<QuicReceivedPacket> ConstructServerDataPacket( |
| 383 | QuicPacketNumber packet_number, |
| 384 | bool should_include_version, |
| 385 | bool fin, |
| 386 | QuicStreamOffset offset, |
fayang | 19c3077 | 2017-03-09 15:06:17 | [diff] [blame] | 387 | QuicStringPiece data) { |
alyssar | 2adf3ac | 2016-05-03 17:12:58 | [diff] [blame] | 388 | return InnerConstructDataPacket(packet_number, stream_id_, |
| 389 | should_include_version, fin, offset, data, |
| 390 | &server_maker_); |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 391 | } |
| 392 | |
danakj | ad1777e | 2016-04-16 00:56:42 | [diff] [blame] | 393 | std::unique_ptr<QuicReceivedPacket> InnerConstructRequestHeadersPacket( |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 394 | QuicPacketNumber packet_number, |
| 395 | QuicStreamId stream_id, |
| 396 | bool should_include_version, |
| 397 | bool fin, |
| 398 | RequestPriority request_priority, |
xunjieli | 100937eb5 | 2016-09-15 20:09:37 | [diff] [blame] | 399 | size_t* spdy_headers_frame_length, |
| 400 | QuicStreamOffset* offset) { |
Yixin Wang | 7a3f1b8d | 2018-01-17 21:40:48 | [diff] [blame] | 401 | return InnerConstructRequestHeadersPacket( |
| 402 | packet_number, stream_id, should_include_version, fin, request_priority, |
| 403 | 0, spdy_headers_frame_length, offset); |
| 404 | } |
| 405 | |
| 406 | std::unique_ptr<QuicReceivedPacket> InnerConstructRequestHeadersPacket( |
| 407 | QuicPacketNumber packet_number, |
| 408 | QuicStreamId stream_id, |
| 409 | bool should_include_version, |
| 410 | bool fin, |
| 411 | RequestPriority request_priority, |
| 412 | QuicStreamId parent_stream_id, |
| 413 | size_t* spdy_headers_frame_length, |
| 414 | QuicStreamOffset* offset) { |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 415 | SpdyPriority priority = |
| 416 | ConvertRequestPriorityToQuicPriority(request_priority); |
alyssar | 2adf3ac | 2016-05-03 17:12:58 | [diff] [blame] | 417 | return client_maker_.MakeRequestHeadersPacket( |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 418 | packet_number, stream_id, should_include_version, fin, priority, |
Yixin Wang | 7a3f1b8d | 2018-01-17 21:40:48 | [diff] [blame] | 419 | std::move(request_headers_), parent_stream_id, |
| 420 | spdy_headers_frame_length, offset); |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 421 | } |
| 422 | |
danakj | ad1777e | 2016-04-16 00:56:42 | [diff] [blame] | 423 | std::unique_ptr<QuicReceivedPacket> ConstructRequestHeadersPacket( |
rtenneti | a004d33 | 2015-08-28 06:44:57 | [diff] [blame] | 424 | QuicPacketNumber packet_number, |
rtenneti | f4bdb54 | 2015-01-21 14:33:05 | [diff] [blame] | 425 | bool fin, |
sclittle | c4dc1a3 | 2015-09-24 00:15:45 | [diff] [blame] | 426 | RequestPriority request_priority, |
| 427 | size_t* spdy_headers_frame_length) { |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 428 | return InnerConstructRequestHeadersPacket( |
| 429 | packet_number, stream_id_, kIncludeVersion, fin, request_priority, |
xunjieli | 100937eb5 | 2016-09-15 20:09:37 | [diff] [blame] | 430 | spdy_headers_frame_length, nullptr); |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 431 | } |
| 432 | |
danakj | ad1777e | 2016-04-16 00:56:42 | [diff] [blame] | 433 | std::unique_ptr<QuicReceivedPacket> InnerConstructResponseHeadersPacket( |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 434 | QuicPacketNumber packet_number, |
| 435 | QuicStreamId stream_id, |
| 436 | bool fin, |
| 437 | size_t* spdy_headers_frame_length) { |
alyssar | 2adf3ac | 2016-05-03 17:12:58 | [diff] [blame] | 438 | return server_maker_.MakeResponseHeadersPacket( |
bnc | 086b39e1 | 2016-06-24 13:05:26 | [diff] [blame] | 439 | packet_number, stream_id, !kIncludeVersion, fin, |
| 440 | std::move(response_headers_), spdy_headers_frame_length, |
| 441 | &response_offset_); |
[email protected] | 1e96003 | 2013-12-20 19:00:20 | [diff] [blame] | 442 | } |
| 443 | |
danakj | ad1777e | 2016-04-16 00:56:42 | [diff] [blame] | 444 | std::unique_ptr<QuicReceivedPacket> ConstructResponseHeadersPacket( |
rtenneti | a004d33 | 2015-08-28 06:44:57 | [diff] [blame] | 445 | QuicPacketNumber packet_number, |
sclittle | c4dc1a3 | 2015-09-24 00:15:45 | [diff] [blame] | 446 | bool fin, |
| 447 | size_t* spdy_headers_frame_length) { |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 448 | return InnerConstructResponseHeadersPacket(packet_number, stream_id_, fin, |
| 449 | spdy_headers_frame_length); |
[email protected] | 1e96003 | 2013-12-20 19:00:20 | [diff] [blame] | 450 | } |
| 451 | |
danakj | ad1777e | 2016-04-16 00:56:42 | [diff] [blame] | 452 | std::unique_ptr<QuicReceivedPacket> ConstructResponseHeadersPacketWithOffset( |
xunjieli | 34291fe1 | 2016-03-02 13:58:38 | [diff] [blame] | 453 | QuicPacketNumber packet_number, |
| 454 | bool fin, |
| 455 | size_t* spdy_headers_frame_length, |
| 456 | QuicStreamOffset* offset) { |
alyssar | 2adf3ac | 2016-05-03 17:12:58 | [diff] [blame] | 457 | return server_maker_.MakeResponseHeadersPacket( |
bnc | 086b39e1 | 2016-06-24 13:05:26 | [diff] [blame] | 458 | packet_number, stream_id_, !kIncludeVersion, fin, |
| 459 | std::move(response_headers_), spdy_headers_frame_length, offset); |
xunjieli | 34291fe1 | 2016-03-02 13:58:38 | [diff] [blame] | 460 | } |
| 461 | |
danakj | ad1777e | 2016-04-16 00:56:42 | [diff] [blame] | 462 | std::unique_ptr<QuicReceivedPacket> ConstructResponseTrailersPacket( |
xunjieli | 34291fe1 | 2016-03-02 13:58:38 | [diff] [blame] | 463 | QuicPacketNumber packet_number, |
| 464 | bool fin, |
bnc | 086b39e1 | 2016-06-24 13:05:26 | [diff] [blame] | 465 | SpdyHeaderBlock trailers, |
xunjieli | 34291fe1 | 2016-03-02 13:58:38 | [diff] [blame] | 466 | size_t* spdy_headers_frame_length, |
| 467 | QuicStreamOffset* offset) { |
alyssar | 2adf3ac | 2016-05-03 17:12:58 | [diff] [blame] | 468 | return server_maker_.MakeResponseHeadersPacket( |
bnc | 086b39e1 | 2016-06-24 13:05:26 | [diff] [blame] | 469 | packet_number, stream_id_, !kIncludeVersion, fin, std::move(trailers), |
alyssar | 2adf3ac | 2016-05-03 17:12:58 | [diff] [blame] | 470 | spdy_headers_frame_length, offset); |
xunjieli | 34291fe1 | 2016-03-02 13:58:38 | [diff] [blame] | 471 | } |
| 472 | |
alyssar | 2adf3ac | 2016-05-03 17:12:58 | [diff] [blame] | 473 | std::unique_ptr<QuicReceivedPacket> ConstructClientRstStreamPacket( |
rtenneti | a004d33 | 2015-08-28 06:44:57 | [diff] [blame] | 474 | QuicPacketNumber packet_number) { |
alyssar | a72f535 | 2016-10-20 12:45:16 | [diff] [blame] | 475 | return client_maker_.MakeRstPacket(packet_number, true, stream_id_, |
| 476 | QUIC_RST_ACKNOWLEDGEMENT); |
[email protected] | 06ff515 | 2013-08-29 01:03:05 | [diff] [blame] | 477 | } |
| 478 | |
alyssar | 2adf3ac | 2016-05-03 17:12:58 | [diff] [blame] | 479 | std::unique_ptr<QuicReceivedPacket> ConstructClientRstStreamCancelledPacket( |
rtenneti | a004d33 | 2015-08-28 06:44:57 | [diff] [blame] | 480 | QuicPacketNumber packet_number) { |
alyssar | 2adf3ac | 2016-05-03 17:12:58 | [diff] [blame] | 481 | return client_maker_.MakeRstPacket(packet_number, !kIncludeVersion, |
| 482 | stream_id_, QUIC_STREAM_CANCELLED); |
rtenneti | 6bd660b | 2015-07-18 00:19:53 | [diff] [blame] | 483 | } |
| 484 | |
alyssar | 2adf3ac | 2016-05-03 17:12:58 | [diff] [blame] | 485 | std::unique_ptr<QuicReceivedPacket> |
| 486 | ConstructClientRstStreamVaryMismatchPacket(QuicPacketNumber packet_number) { |
| 487 | return client_maker_.MakeRstPacket(packet_number, !kIncludeVersion, |
| 488 | promise_id_, QUIC_PROMISE_VARY_MISMATCH); |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 489 | } |
| 490 | |
danakj | ad1777e | 2016-04-16 00:56:42 | [diff] [blame] | 491 | std::unique_ptr<QuicReceivedPacket> ConstructAckAndRstStreamPacket( |
xunjieli | 34291fe1 | 2016-03-02 13:58:38 | [diff] [blame] | 492 | QuicPacketNumber packet_number, |
| 493 | QuicPacketNumber largest_received, |
wangyix | 6444ffe | 2017-04-25 17:49:49 | [diff] [blame] | 494 | QuicPacketNumber smallest_received, |
| 495 | QuicPacketNumber least_unacked) { |
alyssar | 2adf3ac | 2016-05-03 17:12:58 | [diff] [blame] | 496 | return client_maker_.MakeAckAndRstPacket( |
xunjieli | 34291fe1 | 2016-03-02 13:58:38 | [diff] [blame] | 497 | packet_number, !kIncludeVersion, stream_id_, QUIC_STREAM_CANCELLED, |
wangyix | 6444ffe | 2017-04-25 17:49:49 | [diff] [blame] | 498 | largest_received, smallest_received, least_unacked, |
xunjieli | 34291fe1 | 2016-03-02 13:58:38 | [diff] [blame] | 499 | !kIncludeCongestionFeedback); |
| 500 | } |
| 501 | |
maksim.sisov | 84e20c9 | 2016-06-23 08:49:34 | [diff] [blame] | 502 | std::unique_ptr<QuicReceivedPacket> ConstructClientRstStreamErrorPacket( |
| 503 | QuicPacketNumber packet_number, |
| 504 | bool include_version) { |
| 505 | return client_maker_.MakeRstPacket(packet_number, include_version, |
| 506 | stream_id_, |
| 507 | QUIC_ERROR_PROCESSING_STREAM); |
| 508 | } |
| 509 | |
danakj | ad1777e | 2016-04-16 00:56:42 | [diff] [blame] | 510 | std::unique_ptr<QuicReceivedPacket> ConstructAckAndRstStreamPacket( |
rtenneti | a004d33 | 2015-08-28 06:44:57 | [diff] [blame] | 511 | QuicPacketNumber packet_number) { |
xunjieli | 34291fe1 | 2016-03-02 13:58:38 | [diff] [blame] | 512 | return ConstructAckAndRstStreamPacket(packet_number, 2, 1, 1); |
[email protected] | c5e1aca | 2014-01-30 04:03:04 | [diff] [blame] | 513 | } |
| 514 | |
alyssar | 2adf3ac | 2016-05-03 17:12:58 | [diff] [blame] | 515 | std::unique_ptr<QuicReceivedPacket> ConstructClientAckPacket( |
rtenneti | a004d33 | 2015-08-28 06:44:57 | [diff] [blame] | 516 | QuicPacketNumber packet_number, |
| 517 | QuicPacketNumber largest_received, |
wangyix | 6444ffe | 2017-04-25 17:49:49 | [diff] [blame] | 518 | QuicPacketNumber smallest_received, |
rtenneti | a004d33 | 2015-08-28 06:44:57 | [diff] [blame] | 519 | QuicPacketNumber least_unacked) { |
alyssar | 2adf3ac | 2016-05-03 17:12:58 | [diff] [blame] | 520 | return client_maker_.MakeAckPacket(packet_number, largest_received, |
wangyix | 6444ffe | 2017-04-25 17:49:49 | [diff] [blame] | 521 | smallest_received, least_unacked, |
alyssar | 2adf3ac | 2016-05-03 17:12:58 | [diff] [blame] | 522 | !kIncludeCongestionFeedback); |
| 523 | } |
| 524 | |
| 525 | std::unique_ptr<QuicReceivedPacket> ConstructServerAckPacket( |
| 526 | QuicPacketNumber packet_number, |
| 527 | QuicPacketNumber largest_received, |
wangyix | 6444ffe | 2017-04-25 17:49:49 | [diff] [blame] | 528 | QuicPacketNumber smallest_received, |
alyssar | 2adf3ac | 2016-05-03 17:12:58 | [diff] [blame] | 529 | QuicPacketNumber least_unacked) { |
| 530 | return server_maker_.MakeAckPacket(packet_number, largest_received, |
wangyix | 6444ffe | 2017-04-25 17:49:49 | [diff] [blame] | 531 | smallest_received, least_unacked, |
alyssar | 2adf3ac | 2016-05-03 17:12:58 | [diff] [blame] | 532 | !kIncludeCongestionFeedback); |
[email protected] | 6353451 | 2012-12-23 18:49:00 | [diff] [blame] | 533 | } |
| 534 | |
Yixin Wang | b470bc88 | 2018-02-15 18:43:57 | [diff] [blame] | 535 | std::unique_ptr<QuicReceivedPacket> ConstructClientPriorityPacket( |
| 536 | QuicPacketNumber packet_number, |
| 537 | bool should_include_version, |
| 538 | QuicStreamId id, |
| 539 | QuicStreamId parent_stream_id, |
| 540 | RequestPriority request_priority, |
| 541 | QuicStreamOffset* header_stream_offset) { |
| 542 | return client_maker_.MakePriorityPacket( |
| 543 | packet_number, should_include_version, id, parent_stream_id, |
| 544 | ConvertRequestPriorityToQuicPriority(request_priority), |
| 545 | header_stream_offset); |
| 546 | } |
| 547 | |
rch | 5cb52246 | 2017-04-25 20:18:36 | [diff] [blame] | 548 | std::unique_ptr<QuicReceivedPacket> ConstructInitialSettingsPacket( |
fayang | 3bcb8b50 | 2016-12-07 21:44:37 | [diff] [blame] | 549 | QuicStreamOffset* offset) { |
rch | 5cb52246 | 2017-04-25 20:18:36 | [diff] [blame] | 550 | return client_maker_.MakeInitialSettingsPacket(1, offset); |
fayang | 3bcb8b50 | 2016-12-07 21:44:37 | [diff] [blame] | 551 | } |
| 552 | |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 553 | void ReceivePromise(QuicStreamId id) { |
ckrasic | 32b17dcd | 2016-10-31 06:15:35 | [diff] [blame] | 554 | auto headers = AsHeaderList(push_promise_); |
rch | 08e19857 | 2017-05-09 16:56:55 | [diff] [blame] | 555 | QuicChromiumClientStream::Handle* stream = |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 556 | QuicHttpStreamPeer::GetQuicChromiumClientStream(stream_.get()); |
ckrasic | 32b17dcd | 2016-10-31 06:15:35 | [diff] [blame] | 557 | stream->OnPromiseHeaderList(id, headers.uncompressed_header_bytes(), |
| 558 | headers); |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 559 | } |
| 560 | |
xunjieli | 84adaab | 2016-09-20 01:12:28 | [diff] [blame] | 561 | void ExpectLoadTimingValid(const LoadTimingInfo& load_timing_info, |
xunjieli | 100937eb5 | 2016-09-15 20:09:37 | [diff] [blame] | 562 | bool session_reused) { |
| 563 | EXPECT_EQ(session_reused, load_timing_info.socket_reused); |
xunjieli | 84adaab | 2016-09-20 01:12:28 | [diff] [blame] | 564 | if (session_reused) { |
| 565 | ExpectConnectTimingHasNoTimes(load_timing_info.connect_timing); |
| 566 | } else { |
| 567 | ExpectConnectTimingHasTimes( |
| 568 | load_timing_info.connect_timing, |
| 569 | CONNECT_TIMING_HAS_SSL_TIMES | CONNECT_TIMING_HAS_DNS_TIMES); |
| 570 | } |
| 571 | ExpectLoadTimingHasOnlyConnectionTimes(load_timing_info); |
xunjieli | 100937eb5 | 2016-09-15 20:09:37 | [diff] [blame] | 572 | } |
| 573 | |
ckrasic | bf2f59c | 2017-05-04 23:54:36 | [diff] [blame] | 574 | QuicStreamId GetNthClientInitiatedStreamId(int n) { |
Yixin Wang | 079ad54 | 2018-01-11 04:06:05 | [diff] [blame] | 575 | return test::GetNthClientInitiatedStreamId(version_, n); |
ckrasic | bf2f59c | 2017-05-04 23:54:36 | [diff] [blame] | 576 | } |
| 577 | |
| 578 | QuicStreamId GetNthServerInitiatedStreamId(int n) { |
Yixin Wang | 079ad54 | 2018-01-11 04:06:05 | [diff] [blame] | 579 | return test::GetNthServerInitiatedStreamId(version_, n); |
ckrasic | bf2f59c | 2017-05-04 23:54:36 | [diff] [blame] | 580 | } |
| 581 | |
Yixin Wang | 079ad54 | 2018-01-11 04:06:05 | [diff] [blame] | 582 | const QuicTransportVersion version_; |
| 583 | const bool client_headers_include_h2_stream_dependency_; |
| 584 | |
xunjieli | 5fafe14 | 2016-03-23 23:32:54 | [diff] [blame] | 585 | BoundTestNetLog net_log_; |
[email protected] | fee17f7 | 2013-02-03 07:47:41 | [diff] [blame] | 586 | MockSendAlgorithm* send_algorithm_; |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 587 | scoped_refptr<TestTaskRunner> runner_; |
danakj | ad1777e | 2016-04-16 00:56:42 | [diff] [blame] | 588 | std::unique_ptr<MockWrite[]> mock_writes_; |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 589 | MockClock clock_; |
| 590 | TestQuicConnection* connection_; |
danakj | ad1777e | 2016-04-16 00:56:42 | [diff] [blame] | 591 | std::unique_ptr<QuicChromiumConnectionHelper> helper_; |
rch | 16c74d1d | 2016-04-22 06:14:07 | [diff] [blame] | 592 | std::unique_ptr<QuicChromiumAlarmFactory> alarm_factory_; |
ckrasic | a7fd124 | 2016-05-14 20:36:01 | [diff] [blame] | 593 | testing::StrictMock<MockQuicConnectionVisitor> visitor_; |
rch | 97827ee | 2017-05-24 23:49:12 | [diff] [blame] | 594 | std::unique_ptr<UploadDataStream> upload_data_stream_; |
danakj | ad1777e | 2016-04-16 00:56:42 | [diff] [blame] | 595 | std::unique_ptr<QuicHttpStream> stream_; |
[email protected] | 5db45220 | 2014-08-19 05:22:15 | [diff] [blame] | 596 | TransportSecurityState transport_security_state_; |
danakj | ad1777e | 2016-04-16 00:56:42 | [diff] [blame] | 597 | std::unique_ptr<QuicChromiumClientSession> session_; |
[email protected] | ef95114d | 2013-04-17 17:57:01 | [diff] [blame] | 598 | QuicCryptoClientConfig crypto_config_; |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 599 | TestCompletionCallback callback_; |
| 600 | HttpRequestInfo request_; |
| 601 | HttpRequestHeaders headers_; |
| 602 | HttpResponseInfo response_; |
| 603 | scoped_refptr<IOBufferWithSize> read_buffer_; |
[email protected] | 1e96003 | 2013-12-20 19:00:20 | [diff] [blame] | 604 | SpdyHeaderBlock request_headers_; |
| 605 | SpdyHeaderBlock response_headers_; |
bnc | 614a92d3 | 2016-04-04 13:56:07 | [diff] [blame] | 606 | string request_data_; |
| 607 | string response_data_; |
ckrasic | 244375a3 | 2016-02-04 21:21:22 | [diff] [blame] | 608 | QuicClientPushPromiseIndex push_promise_index_; |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 609 | |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 610 | // For server push testing |
danakj | ad1777e | 2016-04-16 00:56:42 | [diff] [blame] | 611 | std::unique_ptr<QuicHttpStream> promised_stream_; |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 612 | SpdyHeaderBlock push_promise_; |
| 613 | SpdyHeaderBlock promised_response_; |
| 614 | const QuicStreamId promise_id_; |
| 615 | string promise_url_; |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 616 | const QuicStreamId stream_id_; |
| 617 | |
[email protected] | 3aa9ca7 | 2014-02-27 19:39:43 | [diff] [blame] | 618 | const QuicConnectionId connection_id_; |
alyssar | 2adf3ac | 2016-05-03 17:12:58 | [diff] [blame] | 619 | QuicTestPacketMaker client_maker_; |
| 620 | QuicTestPacketMaker server_maker_; |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 621 | IPEndPoint self_addr_; |
| 622 | IPEndPoint peer_addr_; |
[email protected] | 457d695 | 2013-12-13 09:24:58 | [diff] [blame] | 623 | MockRandom random_generator_; |
rch | 03b7a20 | 2016-02-05 00:54:20 | [diff] [blame] | 624 | ProofVerifyDetailsChromium verify_details_; |
[email protected] | e8ff2684 | 2013-03-22 21:02:05 | [diff] [blame] | 625 | MockCryptoClientStreamFactory crypto_client_stream_factory_; |
danakj | ad1777e | 2016-04-16 00:56:42 | [diff] [blame] | 626 | std::unique_ptr<StaticSocketDataProvider> socket_data_; |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 627 | std::vector<PacketToWrite> writes_; |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 628 | QuicStreamOffset response_offset_; |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 629 | }; |
| 630 | |
Yixin Wang | 079ad54 | 2018-01-11 04:06:05 | [diff] [blame] | 631 | INSTANTIATE_TEST_CASE_P( |
| 632 | VersionIncludeStreamDependencySequnece, |
| 633 | QuicHttpStreamTest, |
| 634 | ::testing::Combine(::testing::ValuesIn(AllSupportedTransportVersions()), |
| 635 | ::testing::Bool())); |
[email protected] | 1e96003 | 2013-12-20 19:00:20 | [diff] [blame] | 636 | |
| 637 | TEST_P(QuicHttpStreamTest, RenewStreamForAuth) { |
[email protected] | ed3fc15d | 2013-03-08 18:37:44 | [diff] [blame] | 638 | Initialize(); |
rtenneti | be63573 | 2014-10-02 22:51:42 | [diff] [blame] | 639 | EXPECT_EQ(nullptr, stream_->RenewStreamForAuth()); |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 640 | } |
| 641 | |
mmenke | bd84c39 | 2015-09-02 14:12:34 | [diff] [blame] | 642 | TEST_P(QuicHttpStreamTest, CanReuseConnection) { |
[email protected] | ed3fc15d | 2013-03-08 18:37:44 | [diff] [blame] | 643 | Initialize(); |
mmenke | bd84c39 | 2015-09-02 14:12:34 | [diff] [blame] | 644 | EXPECT_FALSE(stream_->CanReuseConnection()); |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 645 | } |
| 646 | |
jri | 231c297 | 2016-03-08 19:50:11 | [diff] [blame] | 647 | TEST_P(QuicHttpStreamTest, DisableConnectionMigrationForStream) { |
| 648 | request_.load_flags |= LOAD_DISABLE_CONNECTION_MIGRATION; |
| 649 | Initialize(); |
xunjieli | 5fafe14 | 2016-03-23 23:32:54 | [diff] [blame] | 650 | EXPECT_EQ(OK, |
Steven Valdez | b4ff041 | 2018-01-18 22:39:27 | [diff] [blame] | 651 | stream_->InitializeStream(&request_, false, DEFAULT_PRIORITY, |
xunjieli | 5fafe14 | 2016-03-23 23:32:54 | [diff] [blame] | 652 | net_log_.bound(), callback_.callback())); |
rch | 08e19857 | 2017-05-09 16:56:55 | [diff] [blame] | 653 | QuicChromiumClientStream::Handle* client_stream = |
jri | 231c297 | 2016-03-08 19:50:11 | [diff] [blame] | 654 | QuicHttpStreamPeer::GetQuicChromiumClientStream(stream_.get()); |
| 655 | EXPECT_FALSE(client_stream->can_migrate()); |
| 656 | } |
| 657 | |
[email protected] | 1e96003 | 2013-12-20 19:00:20 | [diff] [blame] | 658 | TEST_P(QuicHttpStreamTest, GetRequest) { |
| 659 | SetRequest("GET", "/", DEFAULT_PRIORITY); |
sclittle | c4dc1a3 | 2015-09-24 00:15:45 | [diff] [blame] | 660 | size_t spdy_request_header_frame_length; |
fayang | 3bcb8b50 | 2016-12-07 21:44:37 | [diff] [blame] | 661 | QuicStreamOffset header_stream_offset = 0; |
rch | 5cb52246 | 2017-04-25 20:18:36 | [diff] [blame] | 662 | AddWrite(ConstructInitialSettingsPacket(&header_stream_offset)); |
fayang | 3bcb8b50 | 2016-12-07 21:44:37 | [diff] [blame] | 663 | AddWrite(InnerConstructRequestHeadersPacket( |
ckrasic | bf2f59c | 2017-05-04 23:54:36 | [diff] [blame] | 664 | 2, GetNthClientInitiatedStreamId(0), kIncludeVersion, kFin, |
| 665 | DEFAULT_PRIORITY, &spdy_request_header_frame_length, |
| 666 | &header_stream_offset)); |
fayang | 3bcb8b50 | 2016-12-07 21:44:37 | [diff] [blame] | 667 | |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 668 | Initialize(); |
| 669 | |
| 670 | request_.method = "GET"; |
rch | cd37901 | 2017-04-12 21:53:32 | [diff] [blame] | 671 | request_.url = GURL("https://www.example.org/"); |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 672 | |
xunjieli | 100937eb5 | 2016-09-15 20:09:37 | [diff] [blame] | 673 | // Make sure getting load timing from the stream early does not crash. |
| 674 | LoadTimingInfo load_timing_info; |
| 675 | EXPECT_TRUE(stream_->GetLoadTimingInfo(&load_timing_info)); |
| 676 | |
xunjieli | 5fafe14 | 2016-03-23 23:32:54 | [diff] [blame] | 677 | EXPECT_EQ(OK, |
Steven Valdez | b4ff041 | 2018-01-18 22:39:27 | [diff] [blame] | 678 | stream_->InitializeStream(&request_, true, DEFAULT_PRIORITY, |
xunjieli | 5fafe14 | 2016-03-23 23:32:54 | [diff] [blame] | 679 | net_log_.bound(), callback_.callback())); |
rjshade | d5ced07 | 2015-12-18 19:26:02 | [diff] [blame] | 680 | EXPECT_EQ(OK, |
| 681 | stream_->SendRequest(headers_, &response_, callback_.callback())); |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 682 | |
| 683 | // Ack the request. |
wangyix | 6444ffe | 2017-04-25 17:49:49 | [diff] [blame] | 684 | ProcessPacket(ConstructServerAckPacket(1, 0, 0, 0)); |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 685 | |
robpercival | 214763f | 2016-07-01 23:27:01 | [diff] [blame] | 686 | EXPECT_THAT(stream_->ReadResponseHeaders(callback_.callback()), |
| 687 | IsError(ERR_IO_PENDING)); |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 688 | |
bnc | 614a92d3 | 2016-04-04 13:56:07 | [diff] [blame] | 689 | SetResponse("404 Not Found", string()); |
sclittle | c4dc1a3 | 2015-09-24 00:15:45 | [diff] [blame] | 690 | size_t spdy_response_header_frame_length; |
| 691 | ProcessPacket(ConstructResponseHeadersPacket( |
| 692 | 2, kFin, &spdy_response_header_frame_length)); |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 693 | |
| 694 | // Now that the headers have been processed, the callback will return. |
robpercival | 214763f | 2016-07-01 23:27:01 | [diff] [blame] | 695 | EXPECT_THAT(callback_.WaitForResult(), IsOk()); |
[email protected] | cadac62 | 2013-06-11 16:46:36 | [diff] [blame] | 696 | ASSERT_TRUE(response_.headers.get()); |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 697 | EXPECT_EQ(404, response_.headers->response_code()); |
| 698 | EXPECT_TRUE(response_.headers->HasHeaderValue("Content-Type", "text/plain")); |
[email protected] | 6ed6743 | 2014-06-24 01:53:53 | [diff] [blame] | 699 | EXPECT_FALSE(response_.response_time.is_null()); |
| 700 | EXPECT_FALSE(response_.request_time.is_null()); |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 701 | |
| 702 | // There is no body, so this should return immediately. |
rjshade | d5ced07 | 2015-12-18 19:26:02 | [diff] [blame] | 703 | EXPECT_EQ(0, |
| 704 | stream_->ReadResponseBody(read_buffer_.get(), read_buffer_->size(), |
| 705 | callback_.callback())); |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 706 | EXPECT_TRUE(stream_->IsResponseBodyComplete()); |
| 707 | EXPECT_TRUE(AtEof()); |
sclittle | 1edeeb2 | 2015-09-02 20:46:10 | [diff] [blame] | 708 | |
xunjieli | 100937eb5 | 2016-09-15 20:09:37 | [diff] [blame] | 709 | EXPECT_TRUE(stream_->GetLoadTimingInfo(&load_timing_info)); |
xunjieli | 84adaab | 2016-09-20 01:12:28 | [diff] [blame] | 710 | ExpectLoadTimingValid(load_timing_info, /*session_reused=*/false); |
xunjieli | 100937eb5 | 2016-09-15 20:09:37 | [diff] [blame] | 711 | |
sclittle | 1edeeb2 | 2015-09-02 20:46:10 | [diff] [blame] | 712 | // QuicHttpStream::GetTotalSent/ReceivedBytes currently only includes the |
sclittle | c4dc1a3 | 2015-09-24 00:15:45 | [diff] [blame] | 713 | // headers and payload. |
| 714 | EXPECT_EQ(static_cast<int64_t>(spdy_request_header_frame_length), |
| 715 | stream_->GetTotalSentBytes()); |
| 716 | EXPECT_EQ(static_cast<int64_t>(spdy_response_header_frame_length), |
| 717 | stream_->GetTotalReceivedBytes()); |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 718 | } |
| 719 | |
xunjieli | 100937eb5 | 2016-09-15 20:09:37 | [diff] [blame] | 720 | TEST_P(QuicHttpStreamTest, LoadTimingTwoRequests) { |
| 721 | SetRequest("GET", "/", DEFAULT_PRIORITY); |
| 722 | size_t spdy_request_header_frame_length; |
| 723 | |
| 724 | QuicStreamOffset offset = 0; |
rch | 5cb52246 | 2017-04-25 20:18:36 | [diff] [blame] | 725 | AddWrite(ConstructInitialSettingsPacket(&offset)); |
xunjieli | 100937eb5 | 2016-09-15 20:09:37 | [diff] [blame] | 726 | AddWrite(InnerConstructRequestHeadersPacket( |
ckrasic | bf2f59c | 2017-05-04 23:54:36 | [diff] [blame] | 727 | 2, GetNthClientInitiatedStreamId(0), kIncludeVersion, kFin, |
| 728 | DEFAULT_PRIORITY, &spdy_request_header_frame_length, &offset)); |
xunjieli | 100937eb5 | 2016-09-15 20:09:37 | [diff] [blame] | 729 | |
| 730 | // SetRequest() again for second request as |request_headers_| was moved. |
| 731 | SetRequest("GET", "/", DEFAULT_PRIORITY); |
| 732 | AddWrite(InnerConstructRequestHeadersPacket( |
ckrasic | bf2f59c | 2017-05-04 23:54:36 | [diff] [blame] | 733 | 3, GetNthClientInitiatedStreamId(1), kIncludeVersion, kFin, |
Yixin Wang | 7a3f1b8d | 2018-01-17 21:40:48 | [diff] [blame] | 734 | DEFAULT_PRIORITY, GetNthClientInitiatedStreamId(0), |
| 735 | &spdy_request_header_frame_length, &offset)); |
wangyix | 6444ffe | 2017-04-25 17:49:49 | [diff] [blame] | 736 | AddWrite(ConstructClientAckPacket(4, 3, 1, 1)); // Ack the responses. |
xunjieli | 100937eb5 | 2016-09-15 20:09:37 | [diff] [blame] | 737 | |
| 738 | Initialize(); |
| 739 | |
| 740 | request_.method = "GET"; |
rch | cd37901 | 2017-04-12 21:53:32 | [diff] [blame] | 741 | request_.url = GURL("https://www.example.org/"); |
xunjieli | 100937eb5 | 2016-09-15 20:09:37 | [diff] [blame] | 742 | // Start first request. |
| 743 | EXPECT_EQ(OK, |
Steven Valdez | b4ff041 | 2018-01-18 22:39:27 | [diff] [blame] | 744 | stream_->InitializeStream(&request_, true, DEFAULT_PRIORITY, |
xunjieli | 100937eb5 | 2016-09-15 20:09:37 | [diff] [blame] | 745 | net_log_.bound(), callback_.callback())); |
| 746 | EXPECT_EQ(OK, |
| 747 | stream_->SendRequest(headers_, &response_, callback_.callback())); |
| 748 | |
| 749 | // Start a second request. |
Ryan Hamilton | 6c2a2a8 | 2017-12-15 02:06:28 | [diff] [blame] | 750 | QuicHttpStream stream2( |
| 751 | session_->CreateHandle(HostPortPair("www.example.org", 443))); |
xunjieli | 100937eb5 | 2016-09-15 20:09:37 | [diff] [blame] | 752 | TestCompletionCallback callback2; |
| 753 | EXPECT_EQ(OK, |
Steven Valdez | b4ff041 | 2018-01-18 22:39:27 | [diff] [blame] | 754 | stream2.InitializeStream(&request_, true, DEFAULT_PRIORITY, |
xunjieli | 100937eb5 | 2016-09-15 20:09:37 | [diff] [blame] | 755 | net_log_.bound(), callback2.callback())); |
| 756 | EXPECT_EQ(OK, |
| 757 | stream2.SendRequest(headers_, &response_, callback2.callback())); |
| 758 | |
| 759 | // Ack both requests. |
wangyix | 6444ffe | 2017-04-25 17:49:49 | [diff] [blame] | 760 | ProcessPacket(ConstructServerAckPacket(1, 0, 0, 0)); |
xunjieli | 100937eb5 | 2016-09-15 20:09:37 | [diff] [blame] | 761 | |
| 762 | EXPECT_THAT(stream_->ReadResponseHeaders(callback_.callback()), |
| 763 | IsError(ERR_IO_PENDING)); |
| 764 | size_t spdy_response_header_frame_length; |
| 765 | SetResponse("200 OK", string()); |
| 766 | ProcessPacket(InnerConstructResponseHeadersPacket( |
ckrasic | bf2f59c | 2017-05-04 23:54:36 | [diff] [blame] | 767 | 2, GetNthClientInitiatedStreamId(0), kFin, |
| 768 | &spdy_response_header_frame_length)); |
xunjieli | 100937eb5 | 2016-09-15 20:09:37 | [diff] [blame] | 769 | |
| 770 | // Now that the headers have been processed, the callback will return. |
| 771 | EXPECT_THAT(callback_.WaitForResult(), IsOk()); |
| 772 | EXPECT_EQ(200, response_.headers->response_code()); |
| 773 | |
| 774 | // There is no body, so this should return immediately. |
| 775 | EXPECT_EQ(0, |
| 776 | stream_->ReadResponseBody(read_buffer_.get(), read_buffer_->size(), |
| 777 | callback_.callback())); |
| 778 | EXPECT_TRUE(stream_->IsResponseBodyComplete()); |
| 779 | |
| 780 | LoadTimingInfo load_timing_info; |
| 781 | EXPECT_TRUE(stream_->GetLoadTimingInfo(&load_timing_info)); |
xunjieli | 84adaab | 2016-09-20 01:12:28 | [diff] [blame] | 782 | ExpectLoadTimingValid(load_timing_info, /*session_reused=*/false); |
xunjieli | 100937eb5 | 2016-09-15 20:09:37 | [diff] [blame] | 783 | |
| 784 | // SetResponse() again for second request as |response_headers_| was moved. |
| 785 | SetResponse("200 OK", string()); |
| 786 | EXPECT_THAT(stream2.ReadResponseHeaders(callback2.callback()), |
| 787 | IsError(ERR_IO_PENDING)); |
| 788 | |
| 789 | ProcessPacket(InnerConstructResponseHeadersPacket( |
ckrasic | bf2f59c | 2017-05-04 23:54:36 | [diff] [blame] | 790 | 3, GetNthClientInitiatedStreamId(1), kFin, |
| 791 | &spdy_response_header_frame_length)); |
xunjieli | 100937eb5 | 2016-09-15 20:09:37 | [diff] [blame] | 792 | |
| 793 | EXPECT_THAT(callback2.WaitForResult(), IsOk()); |
| 794 | |
| 795 | // There is no body, so this should return immediately. |
| 796 | EXPECT_EQ(0, |
| 797 | stream2.ReadResponseBody(read_buffer_.get(), read_buffer_->size(), |
| 798 | callback2.callback())); |
| 799 | EXPECT_TRUE(stream2.IsResponseBodyComplete()); |
| 800 | |
| 801 | LoadTimingInfo load_timing_info2; |
| 802 | EXPECT_TRUE(stream2.GetLoadTimingInfo(&load_timing_info2)); |
xunjieli | 84adaab | 2016-09-20 01:12:28 | [diff] [blame] | 803 | ExpectLoadTimingValid(load_timing_info2, /*session_reused=*/true); |
xunjieli | 100937eb5 | 2016-09-15 20:09:37 | [diff] [blame] | 804 | } |
| 805 | |
xunjieli | 34291fe1 | 2016-03-02 13:58:38 | [diff] [blame] | 806 | // QuicHttpStream does not currently support trailers. It should ignore |
| 807 | // trailers upon receiving them. |
| 808 | TEST_P(QuicHttpStreamTest, GetRequestWithTrailers) { |
| 809 | SetRequest("GET", "/", DEFAULT_PRIORITY); |
| 810 | size_t spdy_request_header_frame_length; |
fayang | 3bcb8b50 | 2016-12-07 21:44:37 | [diff] [blame] | 811 | QuicStreamOffset header_stream_offset = 0; |
rch | 5cb52246 | 2017-04-25 20:18:36 | [diff] [blame] | 812 | AddWrite(ConstructInitialSettingsPacket(&header_stream_offset)); |
fayang | 3bcb8b50 | 2016-12-07 21:44:37 | [diff] [blame] | 813 | AddWrite(InnerConstructRequestHeadersPacket( |
ckrasic | bf2f59c | 2017-05-04 23:54:36 | [diff] [blame] | 814 | 2, GetNthClientInitiatedStreamId(0), kIncludeVersion, kFin, |
| 815 | DEFAULT_PRIORITY, &spdy_request_header_frame_length, |
| 816 | &header_stream_offset)); |
wangyix | 6444ffe | 2017-04-25 17:49:49 | [diff] [blame] | 817 | AddWrite(ConstructClientAckPacket(3, 3, 1, 1)); // Ack the data packet. |
xunjieli | 34291fe1 | 2016-03-02 13:58:38 | [diff] [blame] | 818 | |
| 819 | Initialize(); |
| 820 | |
| 821 | request_.method = "GET"; |
rch | cd37901 | 2017-04-12 21:53:32 | [diff] [blame] | 822 | request_.url = GURL("https://www.example.org/"); |
xunjieli | 34291fe1 | 2016-03-02 13:58:38 | [diff] [blame] | 823 | |
xunjieli | 5fafe14 | 2016-03-23 23:32:54 | [diff] [blame] | 824 | EXPECT_EQ(OK, |
Steven Valdez | b4ff041 | 2018-01-18 22:39:27 | [diff] [blame] | 825 | stream_->InitializeStream(&request_, true, DEFAULT_PRIORITY, |
xunjieli | 5fafe14 | 2016-03-23 23:32:54 | [diff] [blame] | 826 | net_log_.bound(), callback_.callback())); |
| 827 | |
xunjieli | 34291fe1 | 2016-03-02 13:58:38 | [diff] [blame] | 828 | EXPECT_EQ(OK, |
| 829 | stream_->SendRequest(headers_, &response_, callback_.callback())); |
xunjieli | 34291fe1 | 2016-03-02 13:58:38 | [diff] [blame] | 830 | // Ack the request. |
wangyix | 6444ffe | 2017-04-25 17:49:49 | [diff] [blame] | 831 | ProcessPacket(ConstructServerAckPacket(1, 0, 0, 0)); |
xunjieli | 34291fe1 | 2016-03-02 13:58:38 | [diff] [blame] | 832 | |
robpercival | 214763f | 2016-07-01 23:27:01 | [diff] [blame] | 833 | EXPECT_THAT(stream_->ReadResponseHeaders(callback_.callback()), |
| 834 | IsError(ERR_IO_PENDING)); |
xunjieli | 34291fe1 | 2016-03-02 13:58:38 | [diff] [blame] | 835 | |
bnc | 614a92d3 | 2016-04-04 13:56:07 | [diff] [blame] | 836 | SetResponse("200 OK", string()); |
xunjieli | 34291fe1 | 2016-03-02 13:58:38 | [diff] [blame] | 837 | |
| 838 | // Send the response headers. |
| 839 | size_t spdy_response_header_frame_length; |
| 840 | QuicStreamOffset offset = 0; |
| 841 | ProcessPacket(ConstructResponseHeadersPacketWithOffset( |
| 842 | 2, !kFin, &spdy_response_header_frame_length, &offset)); |
| 843 | // Now that the headers have been processed, the callback will return. |
robpercival | 214763f | 2016-07-01 23:27:01 | [diff] [blame] | 844 | EXPECT_THAT(callback_.WaitForResult(), IsOk()); |
xunjieli | 34291fe1 | 2016-03-02 13:58:38 | [diff] [blame] | 845 | ASSERT_TRUE(response_.headers.get()); |
| 846 | EXPECT_EQ(200, response_.headers->response_code()); |
| 847 | EXPECT_TRUE(response_.headers->HasHeaderValue("Content-Type", "text/plain")); |
| 848 | EXPECT_FALSE(response_.response_time.is_null()); |
| 849 | EXPECT_FALSE(response_.request_time.is_null()); |
| 850 | |
| 851 | // Send the response body. |
| 852 | const char kResponseBody[] = "Hello world!"; |
| 853 | ProcessPacket( |
alyssar | 2adf3ac | 2016-05-03 17:12:58 | [diff] [blame] | 854 | ConstructServerDataPacket(3, false, !kFin, /*offset=*/0, kResponseBody)); |
xunjieli | 34291fe1 | 2016-03-02 13:58:38 | [diff] [blame] | 855 | SpdyHeaderBlock trailers; |
| 856 | size_t spdy_trailers_frame_length; |
| 857 | trailers["foo"] = "bar"; |
xunjieli | 188bd40 | 2016-03-12 00:17:25 | [diff] [blame] | 858 | trailers[kFinalOffsetHeaderKey] = base::IntToString(strlen(kResponseBody)); |
xunjieli | 34291fe1 | 2016-03-02 13:58:38 | [diff] [blame] | 859 | ProcessPacket(ConstructResponseTrailersPacket( |
bnc | 086b39e1 | 2016-06-24 13:05:26 | [diff] [blame] | 860 | 4, kFin, std::move(trailers), &spdy_trailers_frame_length, &offset)); |
xunjieli | 34291fe1 | 2016-03-02 13:58:38 | [diff] [blame] | 861 | |
| 862 | // Make sure trailers are processed. |
fdoray | 92e35a7 | 2016-06-10 15:54:55 | [diff] [blame] | 863 | base::RunLoop().RunUntilIdle(); |
xunjieli | 34291fe1 | 2016-03-02 13:58:38 | [diff] [blame] | 864 | |
| 865 | EXPECT_EQ(static_cast<int>(strlen(kResponseBody)), |
| 866 | stream_->ReadResponseBody(read_buffer_.get(), read_buffer_->size(), |
| 867 | callback_.callback())); |
| 868 | EXPECT_TRUE(stream_->IsResponseBodyComplete()); |
| 869 | |
| 870 | EXPECT_EQ(OK, |
| 871 | stream_->ReadResponseBody(read_buffer_.get(), read_buffer_->size(), |
| 872 | callback_.callback())); |
| 873 | |
| 874 | EXPECT_TRUE(stream_->IsResponseBodyComplete()); |
| 875 | EXPECT_TRUE(AtEof()); |
| 876 | |
| 877 | // QuicHttpStream::GetTotalSent/ReceivedBytes currently only includes the |
| 878 | // headers and payload. |
| 879 | EXPECT_EQ(static_cast<int64_t>(spdy_request_header_frame_length), |
| 880 | stream_->GetTotalSentBytes()); |
| 881 | EXPECT_EQ( |
| 882 | static_cast<int64_t>(spdy_response_header_frame_length + |
| 883 | strlen(kResponseBody) + +spdy_trailers_frame_length), |
| 884 | stream_->GetTotalReceivedBytes()); |
xunjieli | 5fafe14 | 2016-03-23 23:32:54 | [diff] [blame] | 885 | // Check that NetLog was filled as expected. |
| 886 | TestNetLogEntry::List entries; |
| 887 | net_log_.GetEntries(&entries); |
| 888 | size_t pos = ExpectLogContainsSomewhere( |
| 889 | entries, /*min_offset=*/0, |
mikecirone | 8b85c43 | 2016-09-08 19:11:00 | [diff] [blame] | 890 | NetLogEventType::QUIC_CHROMIUM_CLIENT_STREAM_SEND_REQUEST_HEADERS, |
| 891 | NetLogEventPhase::NONE); |
xunjieli | 5fafe14 | 2016-03-23 23:32:54 | [diff] [blame] | 892 | pos = ExpectLogContainsSomewhere( |
| 893 | entries, /*min_offset=*/pos, |
mikecirone | 8b85c43 | 2016-09-08 19:11:00 | [diff] [blame] | 894 | NetLogEventType::QUIC_CHROMIUM_CLIENT_STREAM_SEND_REQUEST_HEADERS, |
| 895 | NetLogEventPhase::NONE); |
xunjieli | 5fafe14 | 2016-03-23 23:32:54 | [diff] [blame] | 896 | ExpectLogContainsSomewhere( |
| 897 | entries, /*min_offset=*/pos, |
mikecirone | 8b85c43 | 2016-09-08 19:11:00 | [diff] [blame] | 898 | NetLogEventType::QUIC_CHROMIUM_CLIENT_STREAM_SEND_REQUEST_HEADERS, |
| 899 | NetLogEventPhase::NONE); |
xunjieli | 34291fe1 | 2016-03-02 13:58:38 | [diff] [blame] | 900 | } |
| 901 | |
[email protected] | 3e7dca6 | 2013-09-10 16:14:23 | [diff] [blame] | 902 | // Regression test for http://crbug.com/288128 |
[email protected] | 1e96003 | 2013-12-20 19:00:20 | [diff] [blame] | 903 | TEST_P(QuicHttpStreamTest, GetRequestLargeResponse) { |
| 904 | SetRequest("GET", "/", DEFAULT_PRIORITY); |
sclittle | c4dc1a3 | 2015-09-24 00:15:45 | [diff] [blame] | 905 | size_t spdy_request_headers_frame_length; |
fayang | 3bcb8b50 | 2016-12-07 21:44:37 | [diff] [blame] | 906 | QuicStreamOffset header_stream_offset = 0; |
rch | 5cb52246 | 2017-04-25 20:18:36 | [diff] [blame] | 907 | AddWrite(ConstructInitialSettingsPacket(&header_stream_offset)); |
fayang | 3bcb8b50 | 2016-12-07 21:44:37 | [diff] [blame] | 908 | AddWrite(InnerConstructRequestHeadersPacket( |
ckrasic | bf2f59c | 2017-05-04 23:54:36 | [diff] [blame] | 909 | 2, GetNthClientInitiatedStreamId(0), kIncludeVersion, kFin, |
| 910 | DEFAULT_PRIORITY, &spdy_request_headers_frame_length, |
| 911 | &header_stream_offset)); |
[email protected] | 3e7dca6 | 2013-09-10 16:14:23 | [diff] [blame] | 912 | Initialize(); |
| 913 | |
| 914 | request_.method = "GET"; |
rch | cd37901 | 2017-04-12 21:53:32 | [diff] [blame] | 915 | request_.url = GURL("https://www.example.org/"); |
[email protected] | 3e7dca6 | 2013-09-10 16:14:23 | [diff] [blame] | 916 | |
xunjieli | 5fafe14 | 2016-03-23 23:32:54 | [diff] [blame] | 917 | EXPECT_EQ(OK, |
Steven Valdez | b4ff041 | 2018-01-18 22:39:27 | [diff] [blame] | 918 | stream_->InitializeStream(&request_, true, DEFAULT_PRIORITY, |
xunjieli | 5fafe14 | 2016-03-23 23:32:54 | [diff] [blame] | 919 | net_log_.bound(), callback_.callback())); |
rjshade | d5ced07 | 2015-12-18 19:26:02 | [diff] [blame] | 920 | EXPECT_EQ(OK, |
| 921 | stream_->SendRequest(headers_, &response_, callback_.callback())); |
[email protected] | 3e7dca6 | 2013-09-10 16:14:23 | [diff] [blame] | 922 | |
| 923 | // Ack the request. |
wangyix | 6444ffe | 2017-04-25 17:49:49 | [diff] [blame] | 924 | ProcessPacket(ConstructServerAckPacket(1, 0, 0, 0)); |
[email protected] | 3e7dca6 | 2013-09-10 16:14:23 | [diff] [blame] | 925 | |
robpercival | 214763f | 2016-07-01 23:27:01 | [diff] [blame] | 926 | EXPECT_THAT(stream_->ReadResponseHeaders(callback_.callback()), |
| 927 | IsError(ERR_IO_PENDING)); |
[email protected] | 3e7dca6 | 2013-09-10 16:14:23 | [diff] [blame] | 928 | |
bnc | 086b39e1 | 2016-06-24 13:05:26 | [diff] [blame] | 929 | response_headers_[":status"] = "200 OK"; |
| 930 | response_headers_[":version"] = "HTTP/1.1"; |
| 931 | response_headers_["content-type"] = "text/plain"; |
| 932 | response_headers_["big6"] = string(1000, 'x'); // Lots of x's. |
[email protected] | 3e7dca6 | 2013-09-10 16:14:23 | [diff] [blame] | 933 | |
sclittle | c4dc1a3 | 2015-09-24 00:15:45 | [diff] [blame] | 934 | size_t spdy_response_headers_frame_length; |
| 935 | ProcessPacket(ConstructResponseHeadersPacket( |
| 936 | 2, kFin, &spdy_response_headers_frame_length)); |
[email protected] | 3e7dca6 | 2013-09-10 16:14:23 | [diff] [blame] | 937 | |
| 938 | // Now that the headers have been processed, the callback will return. |
robpercival | 214763f | 2016-07-01 23:27:01 | [diff] [blame] | 939 | EXPECT_THAT(callback_.WaitForResult(), IsOk()); |
[email protected] | 3e7dca6 | 2013-09-10 16:14:23 | [diff] [blame] | 940 | ASSERT_TRUE(response_.headers.get()); |
| 941 | EXPECT_EQ(200, response_.headers->response_code()); |
| 942 | EXPECT_TRUE(response_.headers->HasHeaderValue("Content-Type", "text/plain")); |
| 943 | |
| 944 | // There is no body, so this should return immediately. |
rjshade | d5ced07 | 2015-12-18 19:26:02 | [diff] [blame] | 945 | EXPECT_EQ(0, |
| 946 | stream_->ReadResponseBody(read_buffer_.get(), read_buffer_->size(), |
| 947 | callback_.callback())); |
[email protected] | 3e7dca6 | 2013-09-10 16:14:23 | [diff] [blame] | 948 | EXPECT_TRUE(stream_->IsResponseBodyComplete()); |
| 949 | EXPECT_TRUE(AtEof()); |
sclittle | 1edeeb2 | 2015-09-02 20:46:10 | [diff] [blame] | 950 | |
| 951 | // QuicHttpStream::GetTotalSent/ReceivedBytes currently only includes the |
sclittle | c4dc1a3 | 2015-09-24 00:15:45 | [diff] [blame] | 952 | // headers and payload. |
| 953 | EXPECT_EQ(static_cast<int64_t>(spdy_request_headers_frame_length), |
| 954 | stream_->GetTotalSentBytes()); |
| 955 | EXPECT_EQ(static_cast<int64_t>(spdy_response_headers_frame_length), |
| 956 | stream_->GetTotalReceivedBytes()); |
[email protected] | 3e7dca6 | 2013-09-10 16:14:23 | [diff] [blame] | 957 | } |
| 958 | |
rch | f9f103cbc | 2014-08-30 05:28:04 | [diff] [blame] | 959 | // Regression test for http://crbug.com/409101 |
| 960 | TEST_P(QuicHttpStreamTest, SessionClosedBeforeSendRequest) { |
| 961 | SetRequest("GET", "/", DEFAULT_PRIORITY); |
| 962 | Initialize(); |
| 963 | |
| 964 | request_.method = "GET"; |
rch | cd37901 | 2017-04-12 21:53:32 | [diff] [blame] | 965 | request_.url = GURL("https://www.example.org/"); |
rch | f9f103cbc | 2014-08-30 05:28:04 | [diff] [blame] | 966 | |
xunjieli | 5fafe14 | 2016-03-23 23:32:54 | [diff] [blame] | 967 | EXPECT_EQ(OK, |
Steven Valdez | b4ff041 | 2018-01-18 22:39:27 | [diff] [blame] | 968 | stream_->InitializeStream(&request_, true, DEFAULT_PRIORITY, |
xunjieli | 5fafe14 | 2016-03-23 23:32:54 | [diff] [blame] | 969 | net_log_.bound(), callback_.callback())); |
rch | f9f103cbc | 2014-08-30 05:28:04 | [diff] [blame] | 970 | |
jri | 78ec06a | 2016-03-31 18:19:40 | [diff] [blame] | 971 | session_->connection()->CloseConnection( |
| 972 | QUIC_NO_ERROR, "test", ConnectionCloseBehavior::SILENT_CLOSE); |
rch | f9f103cbc | 2014-08-30 05:28:04 | [diff] [blame] | 973 | |
| 974 | EXPECT_EQ(ERR_CONNECTION_CLOSED, |
rjshade | d5ced07 | 2015-12-18 19:26:02 | [diff] [blame] | 975 | stream_->SendRequest(headers_, &response_, callback_.callback())); |
sclittle | 1edeeb2 | 2015-09-02 20:46:10 | [diff] [blame] | 976 | |
| 977 | EXPECT_EQ(0, stream_->GetTotalSentBytes()); |
| 978 | EXPECT_EQ(0, stream_->GetTotalReceivedBytes()); |
rch | f9f103cbc | 2014-08-30 05:28:04 | [diff] [blame] | 979 | } |
| 980 | |
rch | 03b7a20 | 2016-02-05 00:54:20 | [diff] [blame] | 981 | // Regression test for http://crbug.com/584441 |
| 982 | TEST_P(QuicHttpStreamTest, GetSSLInfoAfterSessionClosed) { |
| 983 | SetRequest("GET", "/", DEFAULT_PRIORITY); |
| 984 | Initialize(); |
| 985 | |
| 986 | request_.method = "GET"; |
rch | cd37901 | 2017-04-12 21:53:32 | [diff] [blame] | 987 | request_.url = GURL("https://www.example.org/"); |
rch | 03b7a20 | 2016-02-05 00:54:20 | [diff] [blame] | 988 | |
xunjieli | 5fafe14 | 2016-03-23 23:32:54 | [diff] [blame] | 989 | EXPECT_EQ(OK, |
Steven Valdez | b4ff041 | 2018-01-18 22:39:27 | [diff] [blame] | 990 | stream_->InitializeStream(&request_, true, DEFAULT_PRIORITY, |
xunjieli | 5fafe14 | 2016-03-23 23:32:54 | [diff] [blame] | 991 | net_log_.bound(), callback_.callback())); |
rch | 03b7a20 | 2016-02-05 00:54:20 | [diff] [blame] | 992 | |
| 993 | SSLInfo ssl_info; |
rch | 11565e0 | 2016-02-09 20:13:47 | [diff] [blame] | 994 | EXPECT_FALSE(ssl_info.is_valid()); |
rch | 03b7a20 | 2016-02-05 00:54:20 | [diff] [blame] | 995 | stream_->GetSSLInfo(&ssl_info); |
rch | 11565e0 | 2016-02-09 20:13:47 | [diff] [blame] | 996 | EXPECT_TRUE(ssl_info.is_valid()); |
rch | 03b7a20 | 2016-02-05 00:54:20 | [diff] [blame] | 997 | |
jri | 78ec06a | 2016-03-31 18:19:40 | [diff] [blame] | 998 | session_->connection()->CloseConnection( |
| 999 | QUIC_NO_ERROR, "test", ConnectionCloseBehavior::SILENT_CLOSE); |
rch | 03b7a20 | 2016-02-05 00:54:20 | [diff] [blame] | 1000 | |
rch | 11565e0 | 2016-02-09 20:13:47 | [diff] [blame] | 1001 | SSLInfo ssl_info2; |
| 1002 | stream_->GetSSLInfo(&ssl_info2); |
| 1003 | EXPECT_TRUE(ssl_info2.is_valid()); |
rch | 03b7a20 | 2016-02-05 00:54:20 | [diff] [blame] | 1004 | } |
| 1005 | |
rch | cd37901 | 2017-04-12 21:53:32 | [diff] [blame] | 1006 | TEST_P(QuicHttpStreamTest, GetAlternativeService) { |
| 1007 | SetRequest("GET", "/", DEFAULT_PRIORITY); |
| 1008 | Initialize(); |
| 1009 | |
| 1010 | request_.method = "GET"; |
| 1011 | request_.url = GURL("https://www.example.org/"); |
| 1012 | |
| 1013 | EXPECT_EQ(OK, |
Steven Valdez | b4ff041 | 2018-01-18 22:39:27 | [diff] [blame] | 1014 | stream_->InitializeStream(&request_, true, DEFAULT_PRIORITY, |
rch | cd37901 | 2017-04-12 21:53:32 | [diff] [blame] | 1015 | net_log_.bound(), callback_.callback())); |
| 1016 | |
| 1017 | AlternativeService alternative_service; |
| 1018 | EXPECT_TRUE(stream_->GetAlternativeService(&alternative_service)); |
| 1019 | EXPECT_EQ(AlternativeService(kProtoQUIC, "www.example.org", 443), |
| 1020 | alternative_service); |
| 1021 | |
| 1022 | session_->connection()->CloseConnection( |
| 1023 | QUIC_NO_ERROR, "test", ConnectionCloseBehavior::SILENT_CLOSE); |
| 1024 | |
| 1025 | AlternativeService alternative_service2; |
| 1026 | EXPECT_TRUE(stream_->GetAlternativeService(&alternative_service2)); |
| 1027 | EXPECT_EQ(AlternativeService(kProtoQUIC, "www.example.org", 443), |
| 1028 | alternative_service2); |
| 1029 | } |
| 1030 | |
zhongyi | ca364fbb | 2015-12-12 03:39:12 | [diff] [blame] | 1031 | TEST_P(QuicHttpStreamTest, LogGranularQuicConnectionError) { |
| 1032 | SetRequest("GET", "/", DEFAULT_PRIORITY); |
| 1033 | size_t spdy_request_headers_frame_length; |
fayang | 3bcb8b50 | 2016-12-07 21:44:37 | [diff] [blame] | 1034 | QuicStreamOffset header_stream_offset = 0; |
rch | 5cb52246 | 2017-04-25 20:18:36 | [diff] [blame] | 1035 | AddWrite(ConstructInitialSettingsPacket(&header_stream_offset)); |
fayang | 3bcb8b50 | 2016-12-07 21:44:37 | [diff] [blame] | 1036 | AddWrite(InnerConstructRequestHeadersPacket( |
ckrasic | bf2f59c | 2017-05-04 23:54:36 | [diff] [blame] | 1037 | 2, GetNthClientInitiatedStreamId(0), kIncludeVersion, kFin, |
| 1038 | DEFAULT_PRIORITY, &spdy_request_headers_frame_length, |
| 1039 | &header_stream_offset)); |
fayang | 3bcb8b50 | 2016-12-07 21:44:37 | [diff] [blame] | 1040 | AddWrite(ConstructAckAndRstStreamPacket(3)); |
zhongyi | ca364fbb | 2015-12-12 03:39:12 | [diff] [blame] | 1041 | Initialize(); |
| 1042 | |
| 1043 | request_.method = "GET"; |
rch | cd37901 | 2017-04-12 21:53:32 | [diff] [blame] | 1044 | request_.url = GURL("https://www.example.org/"); |
zhongyi | ca364fbb | 2015-12-12 03:39:12 | [diff] [blame] | 1045 | |
xunjieli | 5fafe14 | 2016-03-23 23:32:54 | [diff] [blame] | 1046 | EXPECT_EQ(OK, |
Steven Valdez | b4ff041 | 2018-01-18 22:39:27 | [diff] [blame] | 1047 | stream_->InitializeStream(&request_, true, DEFAULT_PRIORITY, |
xunjieli | 5fafe14 | 2016-03-23 23:32:54 | [diff] [blame] | 1048 | net_log_.bound(), callback_.callback())); |
zhongyi | ca364fbb | 2015-12-12 03:39:12 | [diff] [blame] | 1049 | EXPECT_EQ(OK, |
| 1050 | stream_->SendRequest(headers_, &response_, callback_.callback())); |
| 1051 | |
| 1052 | // Ack the request. |
wangyix | 6444ffe | 2017-04-25 17:49:49 | [diff] [blame] | 1053 | ProcessPacket(ConstructServerAckPacket(1, 0, 0, 0)); |
robpercival | 214763f | 2016-07-01 23:27:01 | [diff] [blame] | 1054 | EXPECT_THAT(stream_->ReadResponseHeaders(callback_.callback()), |
| 1055 | IsError(ERR_IO_PENDING)); |
zhongyi | ca364fbb | 2015-12-12 03:39:12 | [diff] [blame] | 1056 | |
rch | 1c5b74a | 2016-06-23 22:10:55 | [diff] [blame] | 1057 | QuicConnectionCloseFrame frame; |
| 1058 | frame.error_code = QUIC_PEER_GOING_AWAY; |
| 1059 | session_->connection()->OnConnectionCloseFrame(frame); |
zhongyi | ca364fbb | 2015-12-12 03:39:12 | [diff] [blame] | 1060 | |
| 1061 | NetErrorDetails details; |
| 1062 | EXPECT_EQ(QUIC_NO_ERROR, details.quic_connection_error); |
| 1063 | stream_->PopulateNetErrorDetails(&details); |
| 1064 | EXPECT_EQ(QUIC_PEER_GOING_AWAY, details.quic_connection_error); |
| 1065 | } |
| 1066 | |
Ryan Hamilton | e316e48 | 2017-08-17 02:48:53 | [diff] [blame] | 1067 | TEST_P(QuicHttpStreamTest, LogGranularQuicErrorIfHandshakeNotConfirmed) { |
rch | 617e065 | 2017-04-26 17:57:51 | [diff] [blame] | 1068 | // By default the test setup defaults handshake to be confirmed. Manually set |
| 1069 | // it to be not confirmed. |
rch | 617e065 | 2017-04-26 17:57:51 | [diff] [blame] | 1070 | crypto_client_stream_factory_.set_handshake_mode( |
| 1071 | MockCryptoClientStream::ZERO_RTT); |
| 1072 | |
zhongyi | ca364fbb | 2015-12-12 03:39:12 | [diff] [blame] | 1073 | SetRequest("GET", "/", DEFAULT_PRIORITY); |
| 1074 | size_t spdy_request_headers_frame_length; |
fayang | 3bcb8b50 | 2016-12-07 21:44:37 | [diff] [blame] | 1075 | QuicStreamOffset header_stream_offset = 0; |
fayang | 3bcb8b50 | 2016-12-07 21:44:37 | [diff] [blame] | 1076 | AddWrite(InnerConstructRequestHeadersPacket( |
ckrasic | bf2f59c | 2017-05-04 23:54:36 | [diff] [blame] | 1077 | 1, GetNthClientInitiatedStreamId(0), kIncludeVersion, kFin, |
| 1078 | DEFAULT_PRIORITY, &spdy_request_headers_frame_length, |
| 1079 | &header_stream_offset)); |
zhongyi | ca364fbb | 2015-12-12 03:39:12 | [diff] [blame] | 1080 | Initialize(); |
| 1081 | |
| 1082 | request_.method = "GET"; |
rch | cd37901 | 2017-04-12 21:53:32 | [diff] [blame] | 1083 | request_.url = GURL("https://www.example.org/"); |
zhongyi | ca364fbb | 2015-12-12 03:39:12 | [diff] [blame] | 1084 | |
xunjieli | 5fafe14 | 2016-03-23 23:32:54 | [diff] [blame] | 1085 | EXPECT_EQ(OK, |
Steven Valdez | b4ff041 | 2018-01-18 22:39:27 | [diff] [blame] | 1086 | stream_->InitializeStream(&request_, true, DEFAULT_PRIORITY, |
xunjieli | 5fafe14 | 2016-03-23 23:32:54 | [diff] [blame] | 1087 | net_log_.bound(), callback_.callback())); |
zhongyi | ca364fbb | 2015-12-12 03:39:12 | [diff] [blame] | 1088 | EXPECT_EQ(OK, |
| 1089 | stream_->SendRequest(headers_, &response_, callback_.callback())); |
| 1090 | |
| 1091 | // Ack the request. |
wangyix | 6444ffe | 2017-04-25 17:49:49 | [diff] [blame] | 1092 | ProcessPacket(ConstructServerAckPacket(1, 0, 0, 0)); |
robpercival | 214763f | 2016-07-01 23:27:01 | [diff] [blame] | 1093 | EXPECT_THAT(stream_->ReadResponseHeaders(callback_.callback()), |
| 1094 | IsError(ERR_IO_PENDING)); |
zhongyi | ca364fbb | 2015-12-12 03:39:12 | [diff] [blame] | 1095 | |
rch | 1c5b74a | 2016-06-23 22:10:55 | [diff] [blame] | 1096 | QuicConnectionCloseFrame frame; |
| 1097 | frame.error_code = QUIC_PEER_GOING_AWAY; |
| 1098 | session_->connection()->OnConnectionCloseFrame(frame); |
zhongyi | ca364fbb | 2015-12-12 03:39:12 | [diff] [blame] | 1099 | |
| 1100 | NetErrorDetails details; |
zhongyi | ca364fbb | 2015-12-12 03:39:12 | [diff] [blame] | 1101 | stream_->PopulateNetErrorDetails(&details); |
Ryan Hamilton | e316e48 | 2017-08-17 02:48:53 | [diff] [blame] | 1102 | EXPECT_EQ(QUIC_PEER_GOING_AWAY, details.quic_connection_error); |
zhongyi | ca364fbb | 2015-12-12 03:39:12 | [diff] [blame] | 1103 | } |
| 1104 | |
rch | 11a114a | 2014-09-04 23:41:59 | [diff] [blame] | 1105 | // Regression test for http://crbug.com/409871 |
| 1106 | TEST_P(QuicHttpStreamTest, SessionClosedBeforeReadResponseHeaders) { |
| 1107 | SetRequest("GET", "/", DEFAULT_PRIORITY); |
sclittle | c4dc1a3 | 2015-09-24 00:15:45 | [diff] [blame] | 1108 | size_t spdy_request_headers_frame_length; |
fayang | 3bcb8b50 | 2016-12-07 21:44:37 | [diff] [blame] | 1109 | QuicStreamOffset header_stream_offset = 0; |
rch | 5cb52246 | 2017-04-25 20:18:36 | [diff] [blame] | 1110 | AddWrite(ConstructInitialSettingsPacket(&header_stream_offset)); |
fayang | 3bcb8b50 | 2016-12-07 21:44:37 | [diff] [blame] | 1111 | AddWrite(InnerConstructRequestHeadersPacket( |
ckrasic | bf2f59c | 2017-05-04 23:54:36 | [diff] [blame] | 1112 | 2, GetNthClientInitiatedStreamId(0), kIncludeVersion, kFin, |
| 1113 | DEFAULT_PRIORITY, &spdy_request_headers_frame_length, |
| 1114 | &header_stream_offset)); |
rch | 11a114a | 2014-09-04 23:41:59 | [diff] [blame] | 1115 | Initialize(); |
| 1116 | |
| 1117 | request_.method = "GET"; |
rch | cd37901 | 2017-04-12 21:53:32 | [diff] [blame] | 1118 | request_.url = GURL("https://www.example.org/"); |
rch | 11a114a | 2014-09-04 23:41:59 | [diff] [blame] | 1119 | |
xunjieli | 5fafe14 | 2016-03-23 23:32:54 | [diff] [blame] | 1120 | EXPECT_EQ(OK, |
Steven Valdez | b4ff041 | 2018-01-18 22:39:27 | [diff] [blame] | 1121 | stream_->InitializeStream(&request_, true, DEFAULT_PRIORITY, |
xunjieli | 5fafe14 | 2016-03-23 23:32:54 | [diff] [blame] | 1122 | net_log_.bound(), callback_.callback())); |
rch | 11a114a | 2014-09-04 23:41:59 | [diff] [blame] | 1123 | |
rjshade | d5ced07 | 2015-12-18 19:26:02 | [diff] [blame] | 1124 | EXPECT_EQ(OK, |
| 1125 | stream_->SendRequest(headers_, &response_, callback_.callback())); |
rch | 11a114a | 2014-09-04 23:41:59 | [diff] [blame] | 1126 | |
jri | 78ec06a | 2016-03-31 18:19:40 | [diff] [blame] | 1127 | session_->connection()->CloseConnection( |
| 1128 | QUIC_NO_ERROR, "test", ConnectionCloseBehavior::SILENT_CLOSE); |
rch | 11a114a | 2014-09-04 23:41:59 | [diff] [blame] | 1129 | |
| 1130 | EXPECT_NE(OK, stream_->ReadResponseHeaders(callback_.callback())); |
sclittle | 1edeeb2 | 2015-09-02 20:46:10 | [diff] [blame] | 1131 | |
| 1132 | // QuicHttpStream::GetTotalSent/ReceivedBytes currently only includes the |
sclittle | c4dc1a3 | 2015-09-24 00:15:45 | [diff] [blame] | 1133 | // headers and payload. |
| 1134 | EXPECT_EQ(static_cast<int64_t>(spdy_request_headers_frame_length), |
| 1135 | stream_->GetTotalSentBytes()); |
sclittle | 1edeeb2 | 2015-09-02 20:46:10 | [diff] [blame] | 1136 | EXPECT_EQ(0, stream_->GetTotalReceivedBytes()); |
rch | 11a114a | 2014-09-04 23:41:59 | [diff] [blame] | 1137 | } |
| 1138 | |
[email protected] | 1e96003 | 2013-12-20 19:00:20 | [diff] [blame] | 1139 | TEST_P(QuicHttpStreamTest, SendPostRequest) { |
| 1140 | SetRequest("POST", "/", DEFAULT_PRIORITY); |
sclittle | c4dc1a3 | 2015-09-24 00:15:45 | [diff] [blame] | 1141 | size_t spdy_request_headers_frame_length; |
fayang | 3bcb8b50 | 2016-12-07 21:44:37 | [diff] [blame] | 1142 | QuicStreamOffset header_stream_offset = 0; |
rch | 5cb52246 | 2017-04-25 20:18:36 | [diff] [blame] | 1143 | AddWrite(ConstructInitialSettingsPacket(&header_stream_offset)); |
fayang | 3bcb8b50 | 2016-12-07 21:44:37 | [diff] [blame] | 1144 | AddWrite(InnerConstructRequestHeadersPacket( |
ckrasic | bf2f59c | 2017-05-04 23:54:36 | [diff] [blame] | 1145 | 2, GetNthClientInitiatedStreamId(0), kIncludeVersion, !kFin, |
| 1146 | DEFAULT_PRIORITY, &spdy_request_headers_frame_length, |
| 1147 | &header_stream_offset)); |
fayang | 3bcb8b50 | 2016-12-07 21:44:37 | [diff] [blame] | 1148 | AddWrite(ConstructClientDataPacket(3, kIncludeVersion, kFin, 0, kUploadData)); |
wangyix | 6444ffe | 2017-04-25 17:49:49 | [diff] [blame] | 1149 | AddWrite(ConstructClientAckPacket(4, 3, 1, 1)); |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 1150 | |
| 1151 | Initialize(); |
| 1152 | |
danakj | ad1777e | 2016-04-16 00:56:42 | [diff] [blame] | 1153 | std::vector<std::unique_ptr<UploadElementReader>> element_readers; |
Jeremy Roman | 0579ed6 | 2017-08-29 15:56:19 | [diff] [blame] | 1154 | element_readers.push_back(std::make_unique<UploadBytesElementReader>( |
ricea | 2deef68 | 2016-09-09 08:04:07 | [diff] [blame] | 1155 | kUploadData, strlen(kUploadData))); |
rch | 97827ee | 2017-05-24 23:49:12 | [diff] [blame] | 1156 | upload_data_stream_ = |
Jeremy Roman | 0579ed6 | 2017-08-29 15:56:19 | [diff] [blame] | 1157 | std::make_unique<ElementsUploadDataStream>(std::move(element_readers), 0); |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 1158 | request_.method = "POST"; |
rch | cd37901 | 2017-04-12 21:53:32 | [diff] [blame] | 1159 | request_.url = GURL("https://www.example.org/"); |
rch | 97827ee | 2017-05-24 23:49:12 | [diff] [blame] | 1160 | request_.upload_data_stream = upload_data_stream_.get(); |
tfarina | 4283411 | 2016-09-22 13:38:20 | [diff] [blame] | 1161 | ASSERT_THAT(request_.upload_data_stream->Init(CompletionCallback(), |
| 1162 | NetLogWithSource()), |
| 1163 | IsOk()); |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 1164 | |
xunjieli | 5fafe14 | 2016-03-23 23:32:54 | [diff] [blame] | 1165 | EXPECT_EQ(OK, |
Steven Valdez | b4ff041 | 2018-01-18 22:39:27 | [diff] [blame] | 1166 | stream_->InitializeStream(&request_, false, DEFAULT_PRIORITY, |
xunjieli | 5fafe14 | 2016-03-23 23:32:54 | [diff] [blame] | 1167 | net_log_.bound(), callback_.callback())); |
rjshade | d5ced07 | 2015-12-18 19:26:02 | [diff] [blame] | 1168 | EXPECT_EQ(OK, |
| 1169 | stream_->SendRequest(headers_, &response_, callback_.callback())); |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 1170 | |
| 1171 | // Ack both packets in the request. |
wangyix | 6444ffe | 2017-04-25 17:49:49 | [diff] [blame] | 1172 | ProcessPacket(ConstructServerAckPacket(1, 0, 0, 0)); |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 1173 | |
| 1174 | // Send the response headers (but not the body). |
bnc | 614a92d3 | 2016-04-04 13:56:07 | [diff] [blame] | 1175 | SetResponse("200 OK", string()); |
sclittle | c4dc1a3 | 2015-09-24 00:15:45 | [diff] [blame] | 1176 | size_t spdy_response_headers_frame_length; |
| 1177 | ProcessPacket(ConstructResponseHeadersPacket( |
| 1178 | 2, !kFin, &spdy_response_headers_frame_length)); |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 1179 | |
rch | fb47f71 | 2017-05-21 03:24:00 | [diff] [blame] | 1180 | // The headers have already arrived. |
| 1181 | EXPECT_THAT(stream_->ReadResponseHeaders(callback_.callback()), IsOk()); |
[email protected] | cadac62 | 2013-06-11 16:46:36 | [diff] [blame] | 1182 | ASSERT_TRUE(response_.headers.get()); |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 1183 | EXPECT_EQ(200, response_.headers->response_code()); |
| 1184 | EXPECT_TRUE(response_.headers->HasHeaderValue("Content-Type", "text/plain")); |
| 1185 | |
| 1186 | // Send the response body. |
| 1187 | const char kResponseBody[] = "Hello world!"; |
alyssar | 2adf3ac | 2016-05-03 17:12:58 | [diff] [blame] | 1188 | ProcessPacket(ConstructServerDataPacket(3, false, kFin, 0, kResponseBody)); |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 1189 | // Since the body has already arrived, this should return immediately. |
| 1190 | EXPECT_EQ(static_cast<int>(strlen(kResponseBody)), |
| 1191 | stream_->ReadResponseBody(read_buffer_.get(), read_buffer_->size(), |
| 1192 | callback_.callback())); |
Ryan Hamilton | 2ef0a9c | 2017-07-25 03:18:29 | [diff] [blame] | 1193 | EXPECT_EQ(0, |
| 1194 | stream_->ReadResponseBody(read_buffer_.get(), read_buffer_->size(), |
| 1195 | callback_.callback())); |
| 1196 | |
| 1197 | EXPECT_TRUE(stream_->IsResponseBodyComplete()); |
| 1198 | EXPECT_TRUE(AtEof()); |
| 1199 | |
| 1200 | // QuicHttpStream::GetTotalSent/ReceivedBytes currently only includes the |
| 1201 | // headers and payload. |
| 1202 | EXPECT_EQ(static_cast<int64_t>(spdy_request_headers_frame_length + |
| 1203 | strlen(kUploadData)), |
| 1204 | stream_->GetTotalSentBytes()); |
| 1205 | EXPECT_EQ(static_cast<int64_t>(spdy_response_headers_frame_length + |
| 1206 | strlen(kResponseBody)), |
| 1207 | stream_->GetTotalReceivedBytes()); |
| 1208 | } |
| 1209 | |
| 1210 | TEST_P(QuicHttpStreamTest, SendPostRequestAndReceiveSoloFin) { |
| 1211 | SetRequest("POST", "/", DEFAULT_PRIORITY); |
| 1212 | size_t spdy_request_headers_frame_length; |
| 1213 | QuicStreamOffset header_stream_offset = 0; |
| 1214 | AddWrite(ConstructInitialSettingsPacket(&header_stream_offset)); |
| 1215 | AddWrite(InnerConstructRequestHeadersPacket( |
| 1216 | 2, GetNthClientInitiatedStreamId(0), kIncludeVersion, !kFin, |
| 1217 | DEFAULT_PRIORITY, &spdy_request_headers_frame_length, |
| 1218 | &header_stream_offset)); |
| 1219 | AddWrite(ConstructClientDataPacket(3, kIncludeVersion, kFin, 0, kUploadData)); |
| 1220 | AddWrite(ConstructClientAckPacket(4, 3, 1, 1)); |
| 1221 | |
| 1222 | Initialize(); |
| 1223 | |
| 1224 | std::vector<std::unique_ptr<UploadElementReader>> element_readers; |
Jeremy Roman | 0579ed6 | 2017-08-29 15:56:19 | [diff] [blame] | 1225 | element_readers.push_back(std::make_unique<UploadBytesElementReader>( |
Ryan Hamilton | 2ef0a9c | 2017-07-25 03:18:29 | [diff] [blame] | 1226 | kUploadData, strlen(kUploadData))); |
| 1227 | upload_data_stream_ = |
Jeremy Roman | 0579ed6 | 2017-08-29 15:56:19 | [diff] [blame] | 1228 | std::make_unique<ElementsUploadDataStream>(std::move(element_readers), 0); |
Ryan Hamilton | 2ef0a9c | 2017-07-25 03:18:29 | [diff] [blame] | 1229 | request_.method = "POST"; |
| 1230 | request_.url = GURL("https://www.example.org/"); |
| 1231 | request_.upload_data_stream = upload_data_stream_.get(); |
| 1232 | ASSERT_THAT(request_.upload_data_stream->Init(CompletionCallback(), |
| 1233 | NetLogWithSource()), |
| 1234 | IsOk()); |
| 1235 | |
| 1236 | EXPECT_EQ(OK, |
Steven Valdez | b4ff041 | 2018-01-18 22:39:27 | [diff] [blame] | 1237 | stream_->InitializeStream(&request_, false, DEFAULT_PRIORITY, |
Ryan Hamilton | 2ef0a9c | 2017-07-25 03:18:29 | [diff] [blame] | 1238 | net_log_.bound(), callback_.callback())); |
| 1239 | EXPECT_EQ(OK, |
| 1240 | stream_->SendRequest(headers_, &response_, callback_.callback())); |
| 1241 | |
| 1242 | // Ack both packets in the request. |
| 1243 | ProcessPacket(ConstructServerAckPacket(1, 0, 0, 0)); |
| 1244 | |
| 1245 | // Send the response headers (but not the body). |
| 1246 | SetResponse("200 OK", string()); |
| 1247 | size_t spdy_response_headers_frame_length; |
| 1248 | ProcessPacket(ConstructResponseHeadersPacket( |
| 1249 | 2, !kFin, &spdy_response_headers_frame_length)); |
| 1250 | |
| 1251 | // The headers have already arrived. |
| 1252 | EXPECT_THAT(stream_->ReadResponseHeaders(callback_.callback()), IsOk()); |
| 1253 | ASSERT_TRUE(response_.headers.get()); |
| 1254 | EXPECT_EQ(200, response_.headers->response_code()); |
| 1255 | EXPECT_TRUE(response_.headers->HasHeaderValue("Content-Type", "text/plain")); |
| 1256 | |
| 1257 | // Send the response body. |
| 1258 | const char kResponseBody[] = "Hello world!"; |
| 1259 | ProcessPacket(ConstructServerDataPacket(3, false, !kFin, 0, kResponseBody)); |
| 1260 | // Since the body has already arrived, this should return immediately. |
| 1261 | EXPECT_EQ(static_cast<int>(strlen(kResponseBody)), |
| 1262 | stream_->ReadResponseBody(read_buffer_.get(), read_buffer_->size(), |
| 1263 | callback_.callback())); |
| 1264 | ProcessPacket(ConstructServerDataPacket(4, false, kFin, |
| 1265 | arraysize(kResponseBody) - 1, "")); |
| 1266 | EXPECT_EQ(0, |
| 1267 | stream_->ReadResponseBody(read_buffer_.get(), read_buffer_->size(), |
| 1268 | callback_.callback())); |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 1269 | |
| 1270 | EXPECT_TRUE(stream_->IsResponseBodyComplete()); |
| 1271 | EXPECT_TRUE(AtEof()); |
sclittle | 1edeeb2 | 2015-09-02 20:46:10 | [diff] [blame] | 1272 | |
| 1273 | // QuicHttpStream::GetTotalSent/ReceivedBytes currently only includes the |
sclittle | c4dc1a3 | 2015-09-24 00:15:45 | [diff] [blame] | 1274 | // headers and payload. |
| 1275 | EXPECT_EQ(static_cast<int64_t>(spdy_request_headers_frame_length + |
| 1276 | strlen(kUploadData)), |
sclittle | 1edeeb2 | 2015-09-02 20:46:10 | [diff] [blame] | 1277 | stream_->GetTotalSentBytes()); |
sclittle | c4dc1a3 | 2015-09-24 00:15:45 | [diff] [blame] | 1278 | EXPECT_EQ(static_cast<int64_t>(spdy_response_headers_frame_length + |
| 1279 | strlen(kResponseBody)), |
sclittle | 1edeeb2 | 2015-09-02 20:46:10 | [diff] [blame] | 1280 | stream_->GetTotalReceivedBytes()); |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 1281 | } |
| 1282 | |
[email protected] | 1e96003 | 2013-12-20 19:00:20 | [diff] [blame] | 1283 | TEST_P(QuicHttpStreamTest, SendChunkedPostRequest) { |
| 1284 | SetRequest("POST", "/", DEFAULT_PRIORITY); |
[email protected] | c9e49a0 | 2013-02-26 05:56:47 | [diff] [blame] | 1285 | size_t chunk_size = strlen(kUploadData); |
sclittle | c4dc1a3 | 2015-09-24 00:15:45 | [diff] [blame] | 1286 | size_t spdy_request_headers_frame_length; |
fayang | 3bcb8b50 | 2016-12-07 21:44:37 | [diff] [blame] | 1287 | QuicStreamOffset header_stream_offset = 0; |
rch | 5cb52246 | 2017-04-25 20:18:36 | [diff] [blame] | 1288 | AddWrite(ConstructInitialSettingsPacket(&header_stream_offset)); |
fayang | 3bcb8b50 | 2016-12-07 21:44:37 | [diff] [blame] | 1289 | AddWrite(InnerConstructRequestHeadersPacket( |
ckrasic | bf2f59c | 2017-05-04 23:54:36 | [diff] [blame] | 1290 | 2, GetNthClientInitiatedStreamId(0), kIncludeVersion, !kFin, |
| 1291 | DEFAULT_PRIORITY, &spdy_request_headers_frame_length, |
| 1292 | &header_stream_offset)); |
rjshade | d5ced07 | 2015-12-18 19:26:02 | [diff] [blame] | 1293 | AddWrite( |
fayang | 3bcb8b50 | 2016-12-07 21:44:37 | [diff] [blame] | 1294 | ConstructClientDataPacket(3, kIncludeVersion, !kFin, 0, kUploadData)); |
| 1295 | AddWrite(ConstructClientDataPacket(4, kIncludeVersion, kFin, chunk_size, |
alyssar | 2adf3ac | 2016-05-03 17:12:58 | [diff] [blame] | 1296 | kUploadData)); |
wangyix | 6444ffe | 2017-04-25 17:49:49 | [diff] [blame] | 1297 | AddWrite(ConstructClientAckPacket(5, 3, 1, 1)); |
[email protected] | c9e49a0 | 2013-02-26 05:56:47 | [diff] [blame] | 1298 | Initialize(); |
| 1299 | |
Jeremy Roman | 0579ed6 | 2017-08-29 15:56:19 | [diff] [blame] | 1300 | upload_data_stream_ = std::make_unique<ChunkedUploadDataStream>(0); |
rch | 97827ee | 2017-05-24 23:49:12 | [diff] [blame] | 1301 | auto* chunked_upload_stream = |
| 1302 | static_cast<ChunkedUploadDataStream*>(upload_data_stream_.get()); |
| 1303 | chunked_upload_stream->AppendData(kUploadData, chunk_size, false); |
[email protected] | c9e49a0 | 2013-02-26 05:56:47 | [diff] [blame] | 1304 | |
| 1305 | request_.method = "POST"; |
rch | cd37901 | 2017-04-12 21:53:32 | [diff] [blame] | 1306 | request_.url = GURL("https://www.example.org/"); |
rch | 97827ee | 2017-05-24 23:49:12 | [diff] [blame] | 1307 | request_.upload_data_stream = upload_data_stream_.get(); |
mmenke | cbc2b71 | 2014-10-09 20:29:07 | [diff] [blame] | 1308 | ASSERT_EQ(OK, request_.upload_data_stream->Init( |
tfarina | 4283411 | 2016-09-22 13:38:20 | [diff] [blame] | 1309 | TestCompletionCallback().callback(), NetLogWithSource())); |
[email protected] | c9e49a0 | 2013-02-26 05:56:47 | [diff] [blame] | 1310 | |
xunjieli | 5fafe14 | 2016-03-23 23:32:54 | [diff] [blame] | 1311 | ASSERT_EQ(OK, |
Steven Valdez | b4ff041 | 2018-01-18 22:39:27 | [diff] [blame] | 1312 | stream_->InitializeStream(&request_, false, DEFAULT_PRIORITY, |
xunjieli | 5fafe14 | 2016-03-23 23:32:54 | [diff] [blame] | 1313 | net_log_.bound(), callback_.callback())); |
rjshade | d5ced07 | 2015-12-18 19:26:02 | [diff] [blame] | 1314 | ASSERT_EQ(ERR_IO_PENDING, |
| 1315 | stream_->SendRequest(headers_, &response_, callback_.callback())); |
[email protected] | c9e49a0 | 2013-02-26 05:56:47 | [diff] [blame] | 1316 | |
rch | 97827ee | 2017-05-24 23:49:12 | [diff] [blame] | 1317 | chunked_upload_stream->AppendData(kUploadData, chunk_size, true); |
robpercival | 214763f | 2016-07-01 23:27:01 | [diff] [blame] | 1318 | EXPECT_THAT(callback_.WaitForResult(), IsOk()); |
[email protected] | c9e49a0 | 2013-02-26 05:56:47 | [diff] [blame] | 1319 | |
| 1320 | // Ack both packets in the request. |
wangyix | 6444ffe | 2017-04-25 17:49:49 | [diff] [blame] | 1321 | ProcessPacket(ConstructServerAckPacket(1, 0, 0, 0)); |
[email protected] | c9e49a0 | 2013-02-26 05:56:47 | [diff] [blame] | 1322 | |
| 1323 | // Send the response headers (but not the body). |
bnc | 614a92d3 | 2016-04-04 13:56:07 | [diff] [blame] | 1324 | SetResponse("200 OK", string()); |
sclittle | c4dc1a3 | 2015-09-24 00:15:45 | [diff] [blame] | 1325 | size_t spdy_response_headers_frame_length; |
| 1326 | ProcessPacket(ConstructResponseHeadersPacket( |
| 1327 | 2, !kFin, &spdy_response_headers_frame_length)); |
[email protected] | c9e49a0 | 2013-02-26 05:56:47 | [diff] [blame] | 1328 | |
rch | fb47f71 | 2017-05-21 03:24:00 | [diff] [blame] | 1329 | // The headers have already arrived. |
| 1330 | EXPECT_THAT(stream_->ReadResponseHeaders(callback_.callback()), IsOk()); |
[email protected] | cadac62 | 2013-06-11 16:46:36 | [diff] [blame] | 1331 | ASSERT_TRUE(response_.headers.get()); |
[email protected] | c9e49a0 | 2013-02-26 05:56:47 | [diff] [blame] | 1332 | EXPECT_EQ(200, response_.headers->response_code()); |
| 1333 | EXPECT_TRUE(response_.headers->HasHeaderValue("Content-Type", "text/plain")); |
| 1334 | |
| 1335 | // Send the response body. |
| 1336 | const char kResponseBody[] = "Hello world!"; |
alyssar | 2adf3ac | 2016-05-03 17:12:58 | [diff] [blame] | 1337 | ProcessPacket(ConstructServerDataPacket( |
| 1338 | 3, false, kFin, response_data_.length(), kResponseBody)); |
[email protected] | c9e49a0 | 2013-02-26 05:56:47 | [diff] [blame] | 1339 | |
| 1340 | // Since the body has already arrived, this should return immediately. |
| 1341 | ASSERT_EQ(static_cast<int>(strlen(kResponseBody)), |
| 1342 | stream_->ReadResponseBody(read_buffer_.get(), read_buffer_->size(), |
| 1343 | callback_.callback())); |
| 1344 | |
| 1345 | EXPECT_TRUE(stream_->IsResponseBodyComplete()); |
| 1346 | EXPECT_TRUE(AtEof()); |
sclittle | 1edeeb2 | 2015-09-02 20:46:10 | [diff] [blame] | 1347 | |
| 1348 | // QuicHttpStream::GetTotalSent/ReceivedBytes currently only includes the |
sclittle | c4dc1a3 | 2015-09-24 00:15:45 | [diff] [blame] | 1349 | // headers and payload. |
| 1350 | EXPECT_EQ(static_cast<int64_t>(spdy_request_headers_frame_length + |
| 1351 | strlen(kUploadData) * 2), |
sclittle | 1edeeb2 | 2015-09-02 20:46:10 | [diff] [blame] | 1352 | stream_->GetTotalSentBytes()); |
sclittle | c4dc1a3 | 2015-09-24 00:15:45 | [diff] [blame] | 1353 | EXPECT_EQ(static_cast<int64_t>(spdy_response_headers_frame_length + |
| 1354 | strlen(kResponseBody)), |
sclittle | 1edeeb2 | 2015-09-02 20:46:10 | [diff] [blame] | 1355 | stream_->GetTotalReceivedBytes()); |
[email protected] | c9e49a0 | 2013-02-26 05:56:47 | [diff] [blame] | 1356 | } |
| 1357 | |
[email protected] | 16ba774 | 2014-08-22 00:57:25 | [diff] [blame] | 1358 | TEST_P(QuicHttpStreamTest, SendChunkedPostRequestWithFinalEmptyDataPacket) { |
| 1359 | SetRequest("POST", "/", DEFAULT_PRIORITY); |
| 1360 | size_t chunk_size = strlen(kUploadData); |
sclittle | c4dc1a3 | 2015-09-24 00:15:45 | [diff] [blame] | 1361 | size_t spdy_request_headers_frame_length; |
fayang | 3bcb8b50 | 2016-12-07 21:44:37 | [diff] [blame] | 1362 | QuicStreamOffset header_stream_offset = 0; |
rch | 5cb52246 | 2017-04-25 20:18:36 | [diff] [blame] | 1363 | AddWrite(ConstructInitialSettingsPacket(&header_stream_offset)); |
fayang | 3bcb8b50 | 2016-12-07 21:44:37 | [diff] [blame] | 1364 | AddWrite(InnerConstructRequestHeadersPacket( |
ckrasic | bf2f59c | 2017-05-04 23:54:36 | [diff] [blame] | 1365 | 2, GetNthClientInitiatedStreamId(0), kIncludeVersion, !kFin, |
| 1366 | DEFAULT_PRIORITY, &spdy_request_headers_frame_length, |
| 1367 | &header_stream_offset)); |
alyssar | 2adf3ac | 2016-05-03 17:12:58 | [diff] [blame] | 1368 | AddWrite( |
fayang | 3bcb8b50 | 2016-12-07 21:44:37 | [diff] [blame] | 1369 | ConstructClientDataPacket(3, kIncludeVersion, !kFin, 0, kUploadData)); |
| 1370 | AddWrite(ConstructClientDataPacket(4, kIncludeVersion, kFin, chunk_size, "")); |
wangyix | 6444ffe | 2017-04-25 17:49:49 | [diff] [blame] | 1371 | AddWrite(ConstructClientAckPacket(5, 3, 1, 1)); |
[email protected] | 16ba774 | 2014-08-22 00:57:25 | [diff] [blame] | 1372 | Initialize(); |
| 1373 | |
Jeremy Roman | 0579ed6 | 2017-08-29 15:56:19 | [diff] [blame] | 1374 | upload_data_stream_ = std::make_unique<ChunkedUploadDataStream>(0); |
rch | 97827ee | 2017-05-24 23:49:12 | [diff] [blame] | 1375 | auto* chunked_upload_stream = |
| 1376 | static_cast<ChunkedUploadDataStream*>(upload_data_stream_.get()); |
| 1377 | chunked_upload_stream->AppendData(kUploadData, chunk_size, false); |
[email protected] | 16ba774 | 2014-08-22 00:57:25 | [diff] [blame] | 1378 | |
| 1379 | request_.method = "POST"; |
rch | cd37901 | 2017-04-12 21:53:32 | [diff] [blame] | 1380 | request_.url = GURL("https://www.example.org/"); |
rch | 97827ee | 2017-05-24 23:49:12 | [diff] [blame] | 1381 | request_.upload_data_stream = upload_data_stream_.get(); |
mmenke | cbc2b71 | 2014-10-09 20:29:07 | [diff] [blame] | 1382 | ASSERT_EQ(OK, request_.upload_data_stream->Init( |
tfarina | 4283411 | 2016-09-22 13:38:20 | [diff] [blame] | 1383 | TestCompletionCallback().callback(), NetLogWithSource())); |
[email protected] | 16ba774 | 2014-08-22 00:57:25 | [diff] [blame] | 1384 | |
xunjieli | 5fafe14 | 2016-03-23 23:32:54 | [diff] [blame] | 1385 | ASSERT_EQ(OK, |
Steven Valdez | b4ff041 | 2018-01-18 22:39:27 | [diff] [blame] | 1386 | stream_->InitializeStream(&request_, false, DEFAULT_PRIORITY, |
xunjieli | 5fafe14 | 2016-03-23 23:32:54 | [diff] [blame] | 1387 | net_log_.bound(), callback_.callback())); |
rjshade | d5ced07 | 2015-12-18 19:26:02 | [diff] [blame] | 1388 | ASSERT_EQ(ERR_IO_PENDING, |
| 1389 | stream_->SendRequest(headers_, &response_, callback_.callback())); |
[email protected] | 16ba774 | 2014-08-22 00:57:25 | [diff] [blame] | 1390 | |
rch | 97827ee | 2017-05-24 23:49:12 | [diff] [blame] | 1391 | chunked_upload_stream->AppendData(nullptr, 0, true); |
robpercival | 214763f | 2016-07-01 23:27:01 | [diff] [blame] | 1392 | EXPECT_THAT(callback_.WaitForResult(), IsOk()); |
[email protected] | 16ba774 | 2014-08-22 00:57:25 | [diff] [blame] | 1393 | |
wangyix | 6444ffe | 2017-04-25 17:49:49 | [diff] [blame] | 1394 | ProcessPacket(ConstructServerAckPacket(1, 0, 0, 0)); |
[email protected] | 16ba774 | 2014-08-22 00:57:25 | [diff] [blame] | 1395 | |
| 1396 | // Send the response headers (but not the body). |
bnc | 614a92d3 | 2016-04-04 13:56:07 | [diff] [blame] | 1397 | SetResponse("200 OK", string()); |
sclittle | c4dc1a3 | 2015-09-24 00:15:45 | [diff] [blame] | 1398 | size_t spdy_response_headers_frame_length; |
| 1399 | ProcessPacket(ConstructResponseHeadersPacket( |
| 1400 | 2, !kFin, &spdy_response_headers_frame_length)); |
[email protected] | 16ba774 | 2014-08-22 00:57:25 | [diff] [blame] | 1401 | |
rch | fb47f71 | 2017-05-21 03:24:00 | [diff] [blame] | 1402 | // The headers have already arrived. |
| 1403 | EXPECT_THAT(stream_->ReadResponseHeaders(callback_.callback()), IsOk()); |
[email protected] | 16ba774 | 2014-08-22 00:57:25 | [diff] [blame] | 1404 | ASSERT_TRUE(response_.headers.get()); |
| 1405 | EXPECT_EQ(200, response_.headers->response_code()); |
| 1406 | EXPECT_TRUE(response_.headers->HasHeaderValue("Content-Type", "text/plain")); |
| 1407 | |
| 1408 | // Send the response body. |
| 1409 | const char kResponseBody[] = "Hello world!"; |
alyssar | 2adf3ac | 2016-05-03 17:12:58 | [diff] [blame] | 1410 | ProcessPacket(ConstructServerDataPacket( |
| 1411 | 3, false, kFin, response_data_.length(), kResponseBody)); |
[email protected] | 16ba774 | 2014-08-22 00:57:25 | [diff] [blame] | 1412 | |
rch | b27683c | 2015-07-29 23:53:50 | [diff] [blame] | 1413 | // The body has arrived, but it is delivered asynchronously |
[email protected] | 16ba774 | 2014-08-22 00:57:25 | [diff] [blame] | 1414 | ASSERT_EQ(static_cast<int>(strlen(kResponseBody)), |
| 1415 | stream_->ReadResponseBody(read_buffer_.get(), read_buffer_->size(), |
| 1416 | callback_.callback())); |
[email protected] | 16ba774 | 2014-08-22 00:57:25 | [diff] [blame] | 1417 | EXPECT_TRUE(stream_->IsResponseBodyComplete()); |
| 1418 | EXPECT_TRUE(AtEof()); |
sclittle | 1edeeb2 | 2015-09-02 20:46:10 | [diff] [blame] | 1419 | |
| 1420 | // QuicHttpStream::GetTotalSent/ReceivedBytes currently only includes the |
sclittle | c4dc1a3 | 2015-09-24 00:15:45 | [diff] [blame] | 1421 | // headers and payload. |
| 1422 | EXPECT_EQ(static_cast<int64_t>(spdy_request_headers_frame_length + |
| 1423 | strlen(kUploadData)), |
sclittle | 1edeeb2 | 2015-09-02 20:46:10 | [diff] [blame] | 1424 | stream_->GetTotalSentBytes()); |
sclittle | c4dc1a3 | 2015-09-24 00:15:45 | [diff] [blame] | 1425 | EXPECT_EQ(static_cast<int64_t>(spdy_response_headers_frame_length + |
| 1426 | strlen(kResponseBody)), |
sclittle | 1edeeb2 | 2015-09-02 20:46:10 | [diff] [blame] | 1427 | stream_->GetTotalReceivedBytes()); |
[email protected] | 16ba774 | 2014-08-22 00:57:25 | [diff] [blame] | 1428 | } |
| 1429 | |
| 1430 | TEST_P(QuicHttpStreamTest, SendChunkedPostRequestWithOneEmptyDataPacket) { |
| 1431 | SetRequest("POST", "/", DEFAULT_PRIORITY); |
sclittle | c4dc1a3 | 2015-09-24 00:15:45 | [diff] [blame] | 1432 | size_t spdy_request_headers_frame_length; |
fayang | 3bcb8b50 | 2016-12-07 21:44:37 | [diff] [blame] | 1433 | QuicStreamOffset header_stream_offset = 0; |
rch | 5cb52246 | 2017-04-25 20:18:36 | [diff] [blame] | 1434 | AddWrite(ConstructInitialSettingsPacket(&header_stream_offset)); |
fayang | 3bcb8b50 | 2016-12-07 21:44:37 | [diff] [blame] | 1435 | AddWrite(InnerConstructRequestHeadersPacket( |
ckrasic | bf2f59c | 2017-05-04 23:54:36 | [diff] [blame] | 1436 | 2, GetNthClientInitiatedStreamId(0), kIncludeVersion, !kFin, |
| 1437 | DEFAULT_PRIORITY, &spdy_request_headers_frame_length, |
| 1438 | &header_stream_offset)); |
fayang | 3bcb8b50 | 2016-12-07 21:44:37 | [diff] [blame] | 1439 | AddWrite(ConstructClientDataPacket(3, kIncludeVersion, kFin, 0, "")); |
wangyix | 6444ffe | 2017-04-25 17:49:49 | [diff] [blame] | 1440 | AddWrite(ConstructClientAckPacket(4, 3, 1, 1)); |
[email protected] | 16ba774 | 2014-08-22 00:57:25 | [diff] [blame] | 1441 | Initialize(); |
| 1442 | |
Jeremy Roman | 0579ed6 | 2017-08-29 15:56:19 | [diff] [blame] | 1443 | upload_data_stream_ = std::make_unique<ChunkedUploadDataStream>(0); |
rch | 97827ee | 2017-05-24 23:49:12 | [diff] [blame] | 1444 | auto* chunked_upload_stream = |
| 1445 | static_cast<ChunkedUploadDataStream*>(upload_data_stream_.get()); |
[email protected] | 16ba774 | 2014-08-22 00:57:25 | [diff] [blame] | 1446 | |
| 1447 | request_.method = "POST"; |
rch | cd37901 | 2017-04-12 21:53:32 | [diff] [blame] | 1448 | request_.url = GURL("https://www.example.org/"); |
rch | 97827ee | 2017-05-24 23:49:12 | [diff] [blame] | 1449 | request_.upload_data_stream = upload_data_stream_.get(); |
mmenke | cbc2b71 | 2014-10-09 20:29:07 | [diff] [blame] | 1450 | ASSERT_EQ(OK, request_.upload_data_stream->Init( |
tfarina | 4283411 | 2016-09-22 13:38:20 | [diff] [blame] | 1451 | TestCompletionCallback().callback(), NetLogWithSource())); |
[email protected] | 16ba774 | 2014-08-22 00:57:25 | [diff] [blame] | 1452 | |
xunjieli | 5fafe14 | 2016-03-23 23:32:54 | [diff] [blame] | 1453 | ASSERT_EQ(OK, |
Steven Valdez | b4ff041 | 2018-01-18 22:39:27 | [diff] [blame] | 1454 | stream_->InitializeStream(&request_, false, DEFAULT_PRIORITY, |
xunjieli | 5fafe14 | 2016-03-23 23:32:54 | [diff] [blame] | 1455 | net_log_.bound(), callback_.callback())); |
rjshade | d5ced07 | 2015-12-18 19:26:02 | [diff] [blame] | 1456 | ASSERT_EQ(ERR_IO_PENDING, |
| 1457 | stream_->SendRequest(headers_, &response_, callback_.callback())); |
[email protected] | 16ba774 | 2014-08-22 00:57:25 | [diff] [blame] | 1458 | |
rch | 97827ee | 2017-05-24 23:49:12 | [diff] [blame] | 1459 | chunked_upload_stream->AppendData(nullptr, 0, true); |
robpercival | 214763f | 2016-07-01 23:27:01 | [diff] [blame] | 1460 | EXPECT_THAT(callback_.WaitForResult(), IsOk()); |
[email protected] | 16ba774 | 2014-08-22 00:57:25 | [diff] [blame] | 1461 | |
wangyix | 6444ffe | 2017-04-25 17:49:49 | [diff] [blame] | 1462 | ProcessPacket(ConstructServerAckPacket(1, 0, 0, 0)); |
[email protected] | 16ba774 | 2014-08-22 00:57:25 | [diff] [blame] | 1463 | |
| 1464 | // Send the response headers (but not the body). |
bnc | 614a92d3 | 2016-04-04 13:56:07 | [diff] [blame] | 1465 | SetResponse("200 OK", string()); |
sclittle | c4dc1a3 | 2015-09-24 00:15:45 | [diff] [blame] | 1466 | size_t spdy_response_headers_frame_length; |
| 1467 | ProcessPacket(ConstructResponseHeadersPacket( |
| 1468 | 2, !kFin, &spdy_response_headers_frame_length)); |
[email protected] | 16ba774 | 2014-08-22 00:57:25 | [diff] [blame] | 1469 | |
rch | fb47f71 | 2017-05-21 03:24:00 | [diff] [blame] | 1470 | // The headers have already arrived. |
| 1471 | EXPECT_THAT(stream_->ReadResponseHeaders(callback_.callback()), IsOk()); |
[email protected] | 16ba774 | 2014-08-22 00:57:25 | [diff] [blame] | 1472 | ASSERT_TRUE(response_.headers.get()); |
| 1473 | EXPECT_EQ(200, response_.headers->response_code()); |
| 1474 | EXPECT_TRUE(response_.headers->HasHeaderValue("Content-Type", "text/plain")); |
| 1475 | |
| 1476 | // Send the response body. |
| 1477 | const char kResponseBody[] = "Hello world!"; |
alyssar | 2adf3ac | 2016-05-03 17:12:58 | [diff] [blame] | 1478 | ProcessPacket(ConstructServerDataPacket( |
| 1479 | 3, false, kFin, response_data_.length(), kResponseBody)); |
[email protected] | 16ba774 | 2014-08-22 00:57:25 | [diff] [blame] | 1480 | |
rch | b27683c | 2015-07-29 23:53:50 | [diff] [blame] | 1481 | // The body has arrived, but it is delivered asynchronously |
[email protected] | 16ba774 | 2014-08-22 00:57:25 | [diff] [blame] | 1482 | ASSERT_EQ(static_cast<int>(strlen(kResponseBody)), |
| 1483 | stream_->ReadResponseBody(read_buffer_.get(), read_buffer_->size(), |
| 1484 | callback_.callback())); |
| 1485 | |
| 1486 | EXPECT_TRUE(stream_->IsResponseBodyComplete()); |
| 1487 | EXPECT_TRUE(AtEof()); |
sclittle | 1edeeb2 | 2015-09-02 20:46:10 | [diff] [blame] | 1488 | |
| 1489 | // QuicHttpStream::GetTotalSent/ReceivedBytes currently only includes the |
sclittle | c4dc1a3 | 2015-09-24 00:15:45 | [diff] [blame] | 1490 | // headers and payload. |
| 1491 | EXPECT_EQ(static_cast<int64_t>(spdy_request_headers_frame_length), |
| 1492 | stream_->GetTotalSentBytes()); |
| 1493 | EXPECT_EQ(static_cast<int64_t>(spdy_response_headers_frame_length + |
| 1494 | strlen(kResponseBody)), |
sclittle | 1edeeb2 | 2015-09-02 20:46:10 | [diff] [blame] | 1495 | stream_->GetTotalReceivedBytes()); |
[email protected] | 16ba774 | 2014-08-22 00:57:25 | [diff] [blame] | 1496 | } |
| 1497 | |
[email protected] | 1e96003 | 2013-12-20 19:00:20 | [diff] [blame] | 1498 | TEST_P(QuicHttpStreamTest, DestroyedEarly) { |
| 1499 | SetRequest("GET", "/", DEFAULT_PRIORITY); |
sclittle | c4dc1a3 | 2015-09-24 00:15:45 | [diff] [blame] | 1500 | size_t spdy_request_headers_frame_length; |
fayang | 3bcb8b50 | 2016-12-07 21:44:37 | [diff] [blame] | 1501 | QuicStreamOffset header_stream_offset = 0; |
rch | 5cb52246 | 2017-04-25 20:18:36 | [diff] [blame] | 1502 | AddWrite(ConstructInitialSettingsPacket(&header_stream_offset)); |
fayang | 3bcb8b50 | 2016-12-07 21:44:37 | [diff] [blame] | 1503 | AddWrite(InnerConstructRequestHeadersPacket( |
ckrasic | bf2f59c | 2017-05-04 23:54:36 | [diff] [blame] | 1504 | 2, GetNthClientInitiatedStreamId(0), kIncludeVersion, kFin, |
| 1505 | DEFAULT_PRIORITY, &spdy_request_headers_frame_length, |
| 1506 | &header_stream_offset)); |
fayang | 3bcb8b50 | 2016-12-07 21:44:37 | [diff] [blame] | 1507 | AddWrite(ConstructAckAndRstStreamPacket(3)); |
[email protected] | 6353451 | 2012-12-23 18:49:00 | [diff] [blame] | 1508 | Initialize(); |
| 1509 | |
| 1510 | request_.method = "GET"; |
rch | cd37901 | 2017-04-12 21:53:32 | [diff] [blame] | 1511 | request_.url = GURL("https://www.example.org/"); |
[email protected] | 6353451 | 2012-12-23 18:49:00 | [diff] [blame] | 1512 | |
xunjieli | 5fafe14 | 2016-03-23 23:32:54 | [diff] [blame] | 1513 | EXPECT_EQ(OK, |
Steven Valdez | b4ff041 | 2018-01-18 22:39:27 | [diff] [blame] | 1514 | stream_->InitializeStream(&request_, true, DEFAULT_PRIORITY, |
xunjieli | 5fafe14 | 2016-03-23 23:32:54 | [diff] [blame] | 1515 | net_log_.bound(), callback_.callback())); |
rjshade | d5ced07 | 2015-12-18 19:26:02 | [diff] [blame] | 1516 | EXPECT_EQ(OK, |
| 1517 | stream_->SendRequest(headers_, &response_, callback_.callback())); |
[email protected] | 6353451 | 2012-12-23 18:49:00 | [diff] [blame] | 1518 | |
| 1519 | // Ack the request. |
wangyix | 6444ffe | 2017-04-25 17:49:49 | [diff] [blame] | 1520 | ProcessPacket(ConstructServerAckPacket(1, 0, 0, 0)); |
rch | fb47f71 | 2017-05-21 03:24:00 | [diff] [blame] | 1521 | EXPECT_THAT(stream_->ReadResponseHeaders( |
| 1522 | base::Bind(&QuicHttpStreamTest::CloseStream, |
| 1523 | base::Unretained(this), stream_.get())), |
robpercival | 214763f | 2016-07-01 23:27:01 | [diff] [blame] | 1524 | IsError(ERR_IO_PENDING)); |
[email protected] | 6353451 | 2012-12-23 18:49:00 | [diff] [blame] | 1525 | |
| 1526 | // Send the response with a body. |
[email protected] | 1e96003 | 2013-12-20 19:00:20 | [diff] [blame] | 1527 | SetResponse("404 OK", "hello world!"); |
[email protected] | 6353451 | 2012-12-23 18:49:00 | [diff] [blame] | 1528 | // In the course of processing this packet, the QuicHttpStream close itself. |
rch | fb47f71 | 2017-05-21 03:24:00 | [diff] [blame] | 1529 | size_t response_size = 0; |
rch | 1bcfddf2 | 2017-06-03 00:26:29 | [diff] [blame] | 1530 | ProcessPacket(ConstructResponseHeadersPacket(2, !kFin, &response_size)); |
[email protected] | 6353451 | 2012-12-23 18:49:00 | [diff] [blame] | 1531 | |
fdoray | 92e35a7 | 2016-06-10 15:54:55 | [diff] [blame] | 1532 | base::RunLoop().RunUntilIdle(); |
rch | b27683c | 2015-07-29 23:53:50 | [diff] [blame] | 1533 | |
[email protected] | 6353451 | 2012-12-23 18:49:00 | [diff] [blame] | 1534 | EXPECT_TRUE(AtEof()); |
sclittle | 1edeeb2 | 2015-09-02 20:46:10 | [diff] [blame] | 1535 | |
| 1536 | // QuicHttpStream::GetTotalSent/ReceivedBytes currently only includes the |
sclittle | c4dc1a3 | 2015-09-24 00:15:45 | [diff] [blame] | 1537 | // headers and payload. |
| 1538 | EXPECT_EQ(static_cast<int64_t>(spdy_request_headers_frame_length), |
| 1539 | stream_->GetTotalSentBytes()); |
rch | fb47f71 | 2017-05-21 03:24:00 | [diff] [blame] | 1540 | // The stream was closed after receiving the headers. |
| 1541 | EXPECT_EQ(static_cast<int64_t>(response_size), |
| 1542 | stream_->GetTotalReceivedBytes()); |
[email protected] | 6353451 | 2012-12-23 18:49:00 | [diff] [blame] | 1543 | } |
| 1544 | |
[email protected] | 1e96003 | 2013-12-20 19:00:20 | [diff] [blame] | 1545 | TEST_P(QuicHttpStreamTest, Priority) { |
| 1546 | SetRequest("GET", "/", MEDIUM); |
sclittle | c4dc1a3 | 2015-09-24 00:15:45 | [diff] [blame] | 1547 | size_t spdy_request_headers_frame_length; |
fayang | 3bcb8b50 | 2016-12-07 21:44:37 | [diff] [blame] | 1548 | QuicStreamOffset header_stream_offset = 0; |
rch | 5cb52246 | 2017-04-25 20:18:36 | [diff] [blame] | 1549 | AddWrite(ConstructInitialSettingsPacket(&header_stream_offset)); |
fayang | 3bcb8b50 | 2016-12-07 21:44:37 | [diff] [blame] | 1550 | AddWrite(InnerConstructRequestHeadersPacket( |
ckrasic | bf2f59c | 2017-05-04 23:54:36 | [diff] [blame] | 1551 | 2, GetNthClientInitiatedStreamId(0), kIncludeVersion, kFin, MEDIUM, |
fayang | 3bcb8b50 | 2016-12-07 21:44:37 | [diff] [blame] | 1552 | &spdy_request_headers_frame_length, &header_stream_offset)); |
[email protected] | 24e5bc5 | 2013-09-18 15:36:58 | [diff] [blame] | 1553 | Initialize(); |
| 1554 | |
| 1555 | request_.method = "GET"; |
rch | cd37901 | 2017-04-12 21:53:32 | [diff] [blame] | 1556 | request_.url = GURL("https://www.example.org/"); |
[email protected] | 24e5bc5 | 2013-09-18 15:36:58 | [diff] [blame] | 1557 | |
Steven Valdez | b4ff041 | 2018-01-18 22:39:27 | [diff] [blame] | 1558 | EXPECT_EQ(OK, |
| 1559 | stream_->InitializeStream(&request_, true, MEDIUM, net_log_.bound(), |
| 1560 | callback_.callback())); |
[email protected] | 24e5bc5 | 2013-09-18 15:36:58 | [diff] [blame] | 1561 | |
| 1562 | // Check that priority is highest. |
rch | 08e19857 | 2017-05-09 16:56:55 | [diff] [blame] | 1563 | QuicChromiumClientStream::Handle* reliable_stream = |
rch | 12fef55 | 2016-01-15 16:26:31 | [diff] [blame] | 1564 | QuicHttpStreamPeer::GetQuicChromiumClientStream(stream_.get()); |
[email protected] | 24e5bc5 | 2013-09-18 15:36:58 | [diff] [blame] | 1565 | DCHECK(reliable_stream); |
fayang | a64c1a9 | 2016-02-13 01:55:58 | [diff] [blame] | 1566 | DCHECK_EQ(kV3HighestPriority, reliable_stream->priority()); |
[email protected] | 24e5bc5 | 2013-09-18 15:36:58 | [diff] [blame] | 1567 | |
rjshade | d5ced07 | 2015-12-18 19:26:02 | [diff] [blame] | 1568 | EXPECT_EQ(OK, |
| 1569 | stream_->SendRequest(headers_, &response_, callback_.callback())); |
[email protected] | 24e5bc5 | 2013-09-18 15:36:58 | [diff] [blame] | 1570 | |
| 1571 | // Check that priority has now dropped back to MEDIUM. |
ianswett | 0888cff | 2015-11-24 17:42:16 | [diff] [blame] | 1572 | DCHECK_EQ(MEDIUM, |
fayang | a64c1a9 | 2016-02-13 01:55:58 | [diff] [blame] | 1573 | ConvertQuicPriorityToRequestPriority(reliable_stream->priority())); |
[email protected] | 24e5bc5 | 2013-09-18 15:36:58 | [diff] [blame] | 1574 | |
| 1575 | // Ack the request. |
wangyix | 6444ffe | 2017-04-25 17:49:49 | [diff] [blame] | 1576 | ProcessPacket(ConstructServerAckPacket(1, 0, 0, 0)); |
robpercival | 214763f | 2016-07-01 23:27:01 | [diff] [blame] | 1577 | EXPECT_THAT(stream_->ReadResponseHeaders(callback_.callback()), |
| 1578 | IsError(ERR_IO_PENDING)); |
[email protected] | 24e5bc5 | 2013-09-18 15:36:58 | [diff] [blame] | 1579 | |
| 1580 | // Send the response with a body. |
[email protected] | 1e96003 | 2013-12-20 19:00:20 | [diff] [blame] | 1581 | SetResponse("404 OK", "hello world!"); |
rch | fb47f71 | 2017-05-21 03:24:00 | [diff] [blame] | 1582 | size_t response_size = 0; |
| 1583 | ProcessPacket(ConstructResponseHeadersPacket(2, kFin, &response_size)); |
[email protected] | 24e5bc5 | 2013-09-18 15:36:58 | [diff] [blame] | 1584 | |
rch | fb47f71 | 2017-05-21 03:24:00 | [diff] [blame] | 1585 | EXPECT_EQ(OK, callback_.WaitForResult()); |
rch | b27683c | 2015-07-29 23:53:50 | [diff] [blame] | 1586 | |
[email protected] | 24e5bc5 | 2013-09-18 15:36:58 | [diff] [blame] | 1587 | EXPECT_TRUE(AtEof()); |
sclittle | 1edeeb2 | 2015-09-02 20:46:10 | [diff] [blame] | 1588 | |
| 1589 | // QuicHttpStream::GetTotalSent/ReceivedBytes currently only includes the |
sclittle | c4dc1a3 | 2015-09-24 00:15:45 | [diff] [blame] | 1590 | // headers and payload. |
| 1591 | EXPECT_EQ(static_cast<int64_t>(spdy_request_headers_frame_length), |
| 1592 | stream_->GetTotalSentBytes()); |
rch | fb47f71 | 2017-05-21 03:24:00 | [diff] [blame] | 1593 | EXPECT_EQ(static_cast<int64_t>(response_size), |
| 1594 | stream_->GetTotalReceivedBytes()); |
[email protected] | 24e5bc5 | 2013-09-18 15:36:58 | [diff] [blame] | 1595 | } |
| 1596 | |
xunjieli | 8dff50b | 2016-07-22 14:19:06 | [diff] [blame] | 1597 | TEST_P(QuicHttpStreamTest, SessionClosedDuringDoLoop) { |
| 1598 | SetRequest("POST", "/", DEFAULT_PRIORITY); |
| 1599 | size_t spdy_request_headers_frame_length; |
fayang | 3bcb8b50 | 2016-12-07 21:44:37 | [diff] [blame] | 1600 | QuicStreamOffset header_stream_offset = 0; |
rch | 5cb52246 | 2017-04-25 20:18:36 | [diff] [blame] | 1601 | AddWrite(ConstructInitialSettingsPacket(&header_stream_offset)); |
fayang | 3bcb8b50 | 2016-12-07 21:44:37 | [diff] [blame] | 1602 | AddWrite(InnerConstructRequestHeadersPacket( |
ckrasic | bf2f59c | 2017-05-04 23:54:36 | [diff] [blame] | 1603 | 2, GetNthClientInitiatedStreamId(0), kIncludeVersion, !kFin, |
| 1604 | DEFAULT_PRIORITY, &spdy_request_headers_frame_length, |
| 1605 | &header_stream_offset)); |
xunjieli | 8dff50b | 2016-07-22 14:19:06 | [diff] [blame] | 1606 | AddWrite( |
fayang | 3bcb8b50 | 2016-12-07 21:44:37 | [diff] [blame] | 1607 | ConstructClientDataPacket(3, kIncludeVersion, !kFin, 0, kUploadData)); |
xunjieli | 8dff50b | 2016-07-22 14:19:06 | [diff] [blame] | 1608 | // Second data write will result in a synchronous failure which will close |
| 1609 | // the session. |
| 1610 | AddWrite(SYNCHRONOUS, ERR_FAILED); |
| 1611 | Initialize(); |
| 1612 | |
Jeremy Roman | 0579ed6 | 2017-08-29 15:56:19 | [diff] [blame] | 1613 | upload_data_stream_ = std::make_unique<ChunkedUploadDataStream>(0); |
rch | 97827ee | 2017-05-24 23:49:12 | [diff] [blame] | 1614 | auto* chunked_upload_stream = |
| 1615 | static_cast<ChunkedUploadDataStream*>(upload_data_stream_.get()); |
xunjieli | 8dff50b | 2016-07-22 14:19:06 | [diff] [blame] | 1616 | |
| 1617 | request_.method = "POST"; |
rch | cd37901 | 2017-04-12 21:53:32 | [diff] [blame] | 1618 | request_.url = GURL("https://www.example.org/"); |
rch | 97827ee | 2017-05-24 23:49:12 | [diff] [blame] | 1619 | request_.upload_data_stream = upload_data_stream_.get(); |
xunjieli | 8dff50b | 2016-07-22 14:19:06 | [diff] [blame] | 1620 | ASSERT_EQ(OK, request_.upload_data_stream->Init( |
tfarina | 4283411 | 2016-09-22 13:38:20 | [diff] [blame] | 1621 | TestCompletionCallback().callback(), NetLogWithSource())); |
xunjieli | 8dff50b | 2016-07-22 14:19:06 | [diff] [blame] | 1622 | |
| 1623 | size_t chunk_size = strlen(kUploadData); |
rch | 97827ee | 2017-05-24 23:49:12 | [diff] [blame] | 1624 | chunked_upload_stream->AppendData(kUploadData, chunk_size, false); |
xunjieli | 8dff50b | 2016-07-22 14:19:06 | [diff] [blame] | 1625 | ASSERT_EQ(OK, |
Steven Valdez | b4ff041 | 2018-01-18 22:39:27 | [diff] [blame] | 1626 | stream_->InitializeStream(&request_, false, DEFAULT_PRIORITY, |
xunjieli | 8dff50b | 2016-07-22 14:19:06 | [diff] [blame] | 1627 | net_log_.bound(), callback_.callback())); |
| 1628 | QuicHttpStream* stream = stream_.get(); |
| 1629 | DeleteStreamCallback delete_stream_callback(std::move(stream_)); |
| 1630 | // SendRequest() completes asynchronously after the final chunk is added. |
| 1631 | ASSERT_EQ(ERR_IO_PENDING, |
| 1632 | stream->SendRequest(headers_, &response_, callback_.callback())); |
rch | 97827ee | 2017-05-24 23:49:12 | [diff] [blame] | 1633 | chunked_upload_stream->AppendData(kUploadData, chunk_size, true); |
xunjieli | 8dff50b | 2016-07-22 14:19:06 | [diff] [blame] | 1634 | int rv = callback_.WaitForResult(); |
| 1635 | EXPECT_EQ(ERR_QUIC_PROTOCOL_ERROR, rv); |
| 1636 | } |
| 1637 | |
rtenneti | 15656ae | 2016-01-23 03:05:03 | [diff] [blame] | 1638 | TEST_P(QuicHttpStreamTest, SessionClosedBeforeSendHeadersComplete) { |
| 1639 | SetRequest("POST", "/", DEFAULT_PRIORITY); |
fayang | 3bcb8b50 | 2016-12-07 21:44:37 | [diff] [blame] | 1640 | QuicStreamOffset header_stream_offset = 0; |
rch | 5cb52246 | 2017-04-25 20:18:36 | [diff] [blame] | 1641 | AddWrite(ConstructInitialSettingsPacket(&header_stream_offset)); |
rtenneti | 15656ae | 2016-01-23 03:05:03 | [diff] [blame] | 1642 | AddWrite(SYNCHRONOUS, ERR_FAILED); |
| 1643 | Initialize(); |
| 1644 | |
Jeremy Roman | 0579ed6 | 2017-08-29 15:56:19 | [diff] [blame] | 1645 | upload_data_stream_ = std::make_unique<ChunkedUploadDataStream>(0); |
rtenneti | 15656ae | 2016-01-23 03:05:03 | [diff] [blame] | 1646 | |
| 1647 | request_.method = "POST"; |
rch | cd37901 | 2017-04-12 21:53:32 | [diff] [blame] | 1648 | request_.url = GURL("https://www.example.org/"); |
rch | 97827ee | 2017-05-24 23:49:12 | [diff] [blame] | 1649 | request_.upload_data_stream = upload_data_stream_.get(); |
rtenneti | 15656ae | 2016-01-23 03:05:03 | [diff] [blame] | 1650 | ASSERT_EQ(OK, request_.upload_data_stream->Init( |
tfarina | 4283411 | 2016-09-22 13:38:20 | [diff] [blame] | 1651 | TestCompletionCallback().callback(), NetLogWithSource())); |
rtenneti | 15656ae | 2016-01-23 03:05:03 | [diff] [blame] | 1652 | |
xunjieli | 5fafe14 | 2016-03-23 23:32:54 | [diff] [blame] | 1653 | ASSERT_EQ(OK, |
Steven Valdez | b4ff041 | 2018-01-18 22:39:27 | [diff] [blame] | 1654 | stream_->InitializeStream(&request_, false, DEFAULT_PRIORITY, |
xunjieli | 5fafe14 | 2016-03-23 23:32:54 | [diff] [blame] | 1655 | net_log_.bound(), callback_.callback())); |
rtenneti | 15656ae | 2016-01-23 03:05:03 | [diff] [blame] | 1656 | ASSERT_EQ(ERR_QUIC_PROTOCOL_ERROR, |
| 1657 | stream_->SendRequest(headers_, &response_, callback_.callback())); |
mmenke | ffff364 | 2017-06-15 17:37:24 | [diff] [blame] | 1658 | |
| 1659 | EXPECT_LE(0, stream_->GetTotalSentBytes()); |
| 1660 | EXPECT_EQ(0, stream_->GetTotalReceivedBytes()); |
rtenneti | 15656ae | 2016-01-23 03:05:03 | [diff] [blame] | 1661 | } |
| 1662 | |
| 1663 | TEST_P(QuicHttpStreamTest, SessionClosedBeforeSendBodyComplete) { |
| 1664 | SetRequest("POST", "/", DEFAULT_PRIORITY); |
| 1665 | size_t spdy_request_headers_frame_length; |
fayang | 3bcb8b50 | 2016-12-07 21:44:37 | [diff] [blame] | 1666 | QuicStreamOffset header_stream_offset = 0; |
rch | 5cb52246 | 2017-04-25 20:18:36 | [diff] [blame] | 1667 | AddWrite(ConstructInitialSettingsPacket(&header_stream_offset)); |
fayang | 3bcb8b50 | 2016-12-07 21:44:37 | [diff] [blame] | 1668 | AddWrite(InnerConstructRequestHeadersPacket( |
ckrasic | bf2f59c | 2017-05-04 23:54:36 | [diff] [blame] | 1669 | 2, GetNthClientInitiatedStreamId(0), kIncludeVersion, !kFin, |
| 1670 | DEFAULT_PRIORITY, &spdy_request_headers_frame_length, |
| 1671 | &header_stream_offset)); |
rtenneti | 15656ae | 2016-01-23 03:05:03 | [diff] [blame] | 1672 | AddWrite(SYNCHRONOUS, ERR_FAILED); |
| 1673 | Initialize(); |
| 1674 | |
Jeremy Roman | 0579ed6 | 2017-08-29 15:56:19 | [diff] [blame] | 1675 | upload_data_stream_ = std::make_unique<ChunkedUploadDataStream>(0); |
rch | 97827ee | 2017-05-24 23:49:12 | [diff] [blame] | 1676 | auto* chunked_upload_stream = |
| 1677 | static_cast<ChunkedUploadDataStream*>(upload_data_stream_.get()); |
rtenneti | 15656ae | 2016-01-23 03:05:03 | [diff] [blame] | 1678 | size_t chunk_size = strlen(kUploadData); |
rch | 97827ee | 2017-05-24 23:49:12 | [diff] [blame] | 1679 | chunked_upload_stream->AppendData(kUploadData, chunk_size, false); |
rtenneti | 15656ae | 2016-01-23 03:05:03 | [diff] [blame] | 1680 | |
| 1681 | request_.method = "POST"; |
rch | cd37901 | 2017-04-12 21:53:32 | [diff] [blame] | 1682 | request_.url = GURL("https://www.example.org/"); |
rch | 97827ee | 2017-05-24 23:49:12 | [diff] [blame] | 1683 | request_.upload_data_stream = upload_data_stream_.get(); |
rtenneti | 15656ae | 2016-01-23 03:05:03 | [diff] [blame] | 1684 | ASSERT_EQ(OK, request_.upload_data_stream->Init( |
tfarina | 4283411 | 2016-09-22 13:38:20 | [diff] [blame] | 1685 | TestCompletionCallback().callback(), NetLogWithSource())); |
rtenneti | 15656ae | 2016-01-23 03:05:03 | [diff] [blame] | 1686 | |
xunjieli | 5fafe14 | 2016-03-23 23:32:54 | [diff] [blame] | 1687 | ASSERT_EQ(OK, |
Steven Valdez | b4ff041 | 2018-01-18 22:39:27 | [diff] [blame] | 1688 | stream_->InitializeStream(&request_, false, DEFAULT_PRIORITY, |
xunjieli | 5fafe14 | 2016-03-23 23:32:54 | [diff] [blame] | 1689 | net_log_.bound(), callback_.callback())); |
rtenneti | 15656ae | 2016-01-23 03:05:03 | [diff] [blame] | 1690 | ASSERT_EQ(ERR_QUIC_PROTOCOL_ERROR, |
| 1691 | stream_->SendRequest(headers_, &response_, callback_.callback())); |
| 1692 | } |
| 1693 | |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 1694 | TEST_P(QuicHttpStreamTest, ServerPushGetRequest) { |
| 1695 | SetRequest("GET", "/", DEFAULT_PRIORITY); |
| 1696 | Initialize(); |
| 1697 | |
| 1698 | // Initialize the first stream, for receiving the promise on. |
| 1699 | request_.method = "GET"; |
rch | cd37901 | 2017-04-12 21:53:32 | [diff] [blame] | 1700 | request_.url = GURL("https://www.example.org/"); |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 1701 | |
xunjieli | 5fafe14 | 2016-03-23 23:32:54 | [diff] [blame] | 1702 | EXPECT_EQ(OK, |
Steven Valdez | b4ff041 | 2018-01-18 22:39:27 | [diff] [blame] | 1703 | stream_->InitializeStream(&request_, true, DEFAULT_PRIORITY, |
xunjieli | 5fafe14 | 2016-03-23 23:32:54 | [diff] [blame] | 1704 | net_log_.bound(), callback_.callback())); |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 1705 | |
| 1706 | // TODO(ckrasic) - could do this via constructing a PUSH_PROMISE |
| 1707 | // packet, but does it matter? |
| 1708 | ReceivePromise(promise_id_); |
| 1709 | EXPECT_NE(session_->GetPromisedByUrl(promise_url_), nullptr); |
| 1710 | |
| 1711 | request_.url = GURL(promise_url_); |
| 1712 | |
| 1713 | // Make the second stream that will exercise the first step of the |
| 1714 | // server push rendezvous mechanism. |
Steven Valdez | b4ff041 | 2018-01-18 22:39:27 | [diff] [blame] | 1715 | EXPECT_EQ(OK, promised_stream_->InitializeStream( |
| 1716 | &request_, true, DEFAULT_PRIORITY, net_log_.bound(), |
| 1717 | callback_.callback())); |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 1718 | |
| 1719 | // Receive the promised response headers. |
bnc | 94893a7 | 2016-06-30 13:45:25 | [diff] [blame] | 1720 | response_headers_ = promised_response_.Clone(); |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 1721 | size_t spdy_response_headers_frame_length; |
| 1722 | ProcessPacket(InnerConstructResponseHeadersPacket( |
| 1723 | 1, promise_id_, false, &spdy_response_headers_frame_length)); |
| 1724 | |
| 1725 | // Receive the promised response body. |
| 1726 | const char kResponseBody[] = "Hello world!"; |
alyssar | 2adf3ac | 2016-05-03 17:12:58 | [diff] [blame] | 1727 | ProcessPacket(InnerConstructDataPacket(2, promise_id_, false, kFin, 0, |
| 1728 | kResponseBody, &server_maker_)); |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 1729 | |
| 1730 | // Now sending a matching request will have successful rendezvous |
| 1731 | // with the promised stream. |
| 1732 | EXPECT_EQ(OK, promised_stream_->SendRequest(headers_, &response_, |
| 1733 | callback_.callback())); |
| 1734 | |
| 1735 | EXPECT_EQ( |
| 1736 | QuicHttpStreamPeer::GetQuicChromiumClientStream(promised_stream_.get()) |
| 1737 | ->id(), |
| 1738 | promise_id_); |
| 1739 | |
| 1740 | // The headers will be immediately available. |
robpercival | 214763f | 2016-07-01 23:27:01 | [diff] [blame] | 1741 | EXPECT_THAT(promised_stream_->ReadResponseHeaders(callback_.callback()), |
| 1742 | IsOk()); |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 1743 | |
| 1744 | // As will be the body. |
| 1745 | EXPECT_EQ( |
| 1746 | static_cast<int>(strlen(kResponseBody)), |
| 1747 | promised_stream_->ReadResponseBody( |
| 1748 | read_buffer_.get(), read_buffer_->size(), callback_.callback())); |
| 1749 | EXPECT_TRUE(promised_stream_->IsResponseBodyComplete()); |
| 1750 | EXPECT_TRUE(AtEof()); |
| 1751 | |
| 1752 | EXPECT_EQ(0, stream_->GetTotalSentBytes()); |
| 1753 | EXPECT_EQ(0, stream_->GetTotalReceivedBytes()); |
| 1754 | EXPECT_EQ(0, promised_stream_->GetTotalSentBytes()); |
| 1755 | EXPECT_EQ(static_cast<int64_t>(spdy_response_headers_frame_length + |
| 1756 | strlen(kResponseBody)), |
| 1757 | promised_stream_->GetTotalReceivedBytes()); |
| 1758 | } |
| 1759 | |
| 1760 | TEST_P(QuicHttpStreamTest, ServerPushGetRequestSlowResponse) { |
| 1761 | SetRequest("GET", "/", DEFAULT_PRIORITY); |
| 1762 | Initialize(); |
| 1763 | |
| 1764 | // Initialize the first stream, for receiving the promise on. |
| 1765 | request_.method = "GET"; |
rch | cd37901 | 2017-04-12 21:53:32 | [diff] [blame] | 1766 | request_.url = GURL("https://www.example.org/"); |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 1767 | |
xunjieli | 5fafe14 | 2016-03-23 23:32:54 | [diff] [blame] | 1768 | EXPECT_EQ(OK, |
Steven Valdez | b4ff041 | 2018-01-18 22:39:27 | [diff] [blame] | 1769 | stream_->InitializeStream(&request_, true, DEFAULT_PRIORITY, |
xunjieli | 5fafe14 | 2016-03-23 23:32:54 | [diff] [blame] | 1770 | net_log_.bound(), callback_.callback())); |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 1771 | |
| 1772 | // TODO(ckrasic) - could do this via constructing a PUSH_PROMISE |
| 1773 | // packet, but does it matter? |
| 1774 | ReceivePromise(promise_id_); |
| 1775 | EXPECT_NE(session_->GetPromisedByUrl(promise_url_), nullptr); |
| 1776 | |
| 1777 | request_.url = GURL(promise_url_); |
| 1778 | |
| 1779 | // Make the second stream that will exercise the first step of the |
| 1780 | // server push rendezvous mechanism. |
Steven Valdez | b4ff041 | 2018-01-18 22:39:27 | [diff] [blame] | 1781 | EXPECT_EQ(OK, promised_stream_->InitializeStream( |
| 1782 | &request_, true, DEFAULT_PRIORITY, net_log_.bound(), |
| 1783 | callback_.callback())); |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 1784 | |
| 1785 | // Now sending a matching request will rendezvous with the promised |
| 1786 | // stream, but pending secondary validation. |
| 1787 | EXPECT_EQ(ERR_IO_PENDING, promised_stream_->SendRequest( |
| 1788 | headers_, &response_, callback_.callback())); |
| 1789 | |
| 1790 | // Receive the promised response headers. |
bnc | 94893a7 | 2016-06-30 13:45:25 | [diff] [blame] | 1791 | response_headers_ = promised_response_.Clone(); |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 1792 | size_t spdy_response_headers_frame_length; |
| 1793 | ProcessPacket(InnerConstructResponseHeadersPacket( |
| 1794 | 1, promise_id_, false, &spdy_response_headers_frame_length)); |
| 1795 | |
| 1796 | // Receive the promised response body. |
| 1797 | const char kResponseBody[] = "Hello world!"; |
alyssar | 2adf3ac | 2016-05-03 17:12:58 | [diff] [blame] | 1798 | ProcessPacket(InnerConstructDataPacket(2, promise_id_, false, kFin, 0, |
| 1799 | kResponseBody, &server_maker_)); |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 1800 | |
fdoray | 92e35a7 | 2016-06-10 15:54:55 | [diff] [blame] | 1801 | base::RunLoop().RunUntilIdle(); |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 1802 | |
| 1803 | // Rendezvous should have succeeded now, so the promised stream |
| 1804 | // should point at our push stream, and we should be able read |
| 1805 | // headers and data from it. |
robpercival | 214763f | 2016-07-01 23:27:01 | [diff] [blame] | 1806 | EXPECT_THAT(callback_.WaitForResult(), IsOk()); |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 1807 | |
| 1808 | EXPECT_EQ( |
| 1809 | QuicHttpStreamPeer::GetQuicChromiumClientStream(promised_stream_.get()) |
| 1810 | ->id(), |
| 1811 | promise_id_); |
| 1812 | |
robpercival | 214763f | 2016-07-01 23:27:01 | [diff] [blame] | 1813 | EXPECT_THAT(promised_stream_->ReadResponseHeaders(callback_.callback()), |
| 1814 | IsOk()); |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 1815 | |
| 1816 | EXPECT_EQ( |
| 1817 | static_cast<int>(strlen(kResponseBody)), |
| 1818 | promised_stream_->ReadResponseBody( |
| 1819 | read_buffer_.get(), read_buffer_->size(), callback_.callback())); |
| 1820 | |
| 1821 | // Callback should return |
| 1822 | EXPECT_TRUE(promised_stream_->IsResponseBodyComplete()); |
| 1823 | EXPECT_TRUE(AtEof()); |
| 1824 | |
| 1825 | EXPECT_EQ(0, stream_->GetTotalSentBytes()); |
| 1826 | EXPECT_EQ(0, stream_->GetTotalReceivedBytes()); |
| 1827 | EXPECT_EQ(0, promised_stream_->GetTotalSentBytes()); |
| 1828 | EXPECT_EQ(static_cast<int64_t>(spdy_response_headers_frame_length + |
| 1829 | strlen(kResponseBody)), |
| 1830 | promised_stream_->GetTotalReceivedBytes()); |
| 1831 | } |
| 1832 | |
ckrasic | 2c63f9b | 2016-08-16 23:54:07 | [diff] [blame] | 1833 | // Verify fix for crbug.com/637349 |
| 1834 | TEST_P(QuicHttpStreamTest, ServerPushCancelHttpStreamBeforeResponse) { |
| 1835 | SetRequest("GET", "/", DEFAULT_PRIORITY); |
| 1836 | Initialize(); |
| 1837 | |
| 1838 | // Initialize the first stream, for receiving the promise on. |
| 1839 | request_.method = "GET"; |
rch | cd37901 | 2017-04-12 21:53:32 | [diff] [blame] | 1840 | request_.url = GURL("https://www.example.org/"); |
ckrasic | 2c63f9b | 2016-08-16 23:54:07 | [diff] [blame] | 1841 | |
| 1842 | EXPECT_EQ(OK, |
Steven Valdez | b4ff041 | 2018-01-18 22:39:27 | [diff] [blame] | 1843 | stream_->InitializeStream(&request_, true, DEFAULT_PRIORITY, |
ckrasic | 2c63f9b | 2016-08-16 23:54:07 | [diff] [blame] | 1844 | net_log_.bound(), callback_.callback())); |
| 1845 | |
| 1846 | // TODO(ckrasic) - could do this via constructing a PUSH_PROMISE |
| 1847 | // packet, but does it matter? |
| 1848 | ReceivePromise(promise_id_); |
| 1849 | EXPECT_NE(session_->GetPromisedByUrl(promise_url_), nullptr); |
| 1850 | |
| 1851 | request_.url = GURL(promise_url_); |
| 1852 | |
| 1853 | // Make the second stream that will exercise the first step of the |
| 1854 | // server push rendezvous mechanism. |
Steven Valdez | b4ff041 | 2018-01-18 22:39:27 | [diff] [blame] | 1855 | EXPECT_EQ(OK, promised_stream_->InitializeStream( |
| 1856 | &request_, true, DEFAULT_PRIORITY, net_log_.bound(), |
| 1857 | callback_.callback())); |
ckrasic | 2c63f9b | 2016-08-16 23:54:07 | [diff] [blame] | 1858 | |
| 1859 | // Now sending a matching request will rendezvous with the promised |
| 1860 | // stream, but pending secondary validation. |
| 1861 | EXPECT_EQ(ERR_IO_PENDING, promised_stream_->SendRequest( |
| 1862 | headers_, &response_, callback_.callback())); |
| 1863 | |
| 1864 | base::RunLoop().RunUntilIdle(); |
| 1865 | |
| 1866 | // Cause of FinalValidation() crash as per bug. |
| 1867 | promised_stream_.reset(); |
| 1868 | |
| 1869 | // Receive the promised response headers. |
| 1870 | response_headers_ = promised_response_.Clone(); |
| 1871 | size_t spdy_response_headers_frame_length; |
| 1872 | ProcessPacket(InnerConstructResponseHeadersPacket( |
| 1873 | 1, promise_id_, false, &spdy_response_headers_frame_length)); |
| 1874 | } |
| 1875 | |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 1876 | TEST_P(QuicHttpStreamTest, ServerPushCrossOriginOK) { |
| 1877 | SetRequest("GET", "/", DEFAULT_PRIORITY); |
| 1878 | Initialize(); |
| 1879 | |
| 1880 | // Initialize the first stream, for receiving the promise on. |
| 1881 | request_.method = "GET"; |
rch | cd37901 | 2017-04-12 21:53:32 | [diff] [blame] | 1882 | request_.url = GURL("https://www.example.org/"); |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 1883 | |
xunjieli | 5fafe14 | 2016-03-23 23:32:54 | [diff] [blame] | 1884 | EXPECT_EQ(OK, |
Steven Valdez | b4ff041 | 2018-01-18 22:39:27 | [diff] [blame] | 1885 | stream_->InitializeStream(&request_, true, DEFAULT_PRIORITY, |
xunjieli | 5fafe14 | 2016-03-23 23:32:54 | [diff] [blame] | 1886 | net_log_.bound(), callback_.callback())); |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 1887 | |
| 1888 | // TODO(ckrasic) - could do this via constructing a PUSH_PROMISE |
| 1889 | // packet, but does it matter? |
| 1890 | |
| 1891 | push_promise_[":authority"] = "mail.example.org"; |
Yixin Wang | 81c1118 | 2018-02-05 20:09:57 | [diff] [blame] | 1892 | promise_url_ = SpdyUtils::GetPromisedUrlFromHeaderBlock(push_promise_); |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 1893 | |
| 1894 | ReceivePromise(promise_id_); |
| 1895 | EXPECT_NE(session_->GetPromisedByUrl(promise_url_), nullptr); |
| 1896 | |
| 1897 | request_.url = GURL(promise_url_); |
| 1898 | |
| 1899 | // Make the second stream that will exercise the first step of the |
| 1900 | // server push rendezvous mechanism. |
Steven Valdez | b4ff041 | 2018-01-18 22:39:27 | [diff] [blame] | 1901 | EXPECT_EQ(OK, promised_stream_->InitializeStream( |
| 1902 | &request_, true, DEFAULT_PRIORITY, net_log_.bound(), |
| 1903 | callback_.callback())); |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 1904 | |
| 1905 | // Receive the promised response headers. |
bnc | 94893a7 | 2016-06-30 13:45:25 | [diff] [blame] | 1906 | response_headers_ = promised_response_.Clone(); |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 1907 | size_t spdy_response_headers_frame_length; |
| 1908 | ProcessPacket(InnerConstructResponseHeadersPacket( |
| 1909 | 1, promise_id_, false, &spdy_response_headers_frame_length)); |
| 1910 | |
| 1911 | // Receive the promised response body. |
| 1912 | const char kResponseBody[] = "Hello world!"; |
alyssar | 2adf3ac | 2016-05-03 17:12:58 | [diff] [blame] | 1913 | ProcessPacket(InnerConstructDataPacket(2, promise_id_, false, kFin, 0, |
| 1914 | kResponseBody, &server_maker_)); |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 1915 | |
| 1916 | // Now sending a matching request will have successful rendezvous |
| 1917 | // with the promised stream. |
| 1918 | EXPECT_EQ(OK, promised_stream_->SendRequest(headers_, &response_, |
| 1919 | callback_.callback())); |
| 1920 | |
| 1921 | EXPECT_EQ( |
| 1922 | QuicHttpStreamPeer::GetQuicChromiumClientStream(promised_stream_.get()) |
| 1923 | ->id(), |
| 1924 | promise_id_); |
| 1925 | |
| 1926 | // The headers will be immediately available. |
robpercival | 214763f | 2016-07-01 23:27:01 | [diff] [blame] | 1927 | EXPECT_THAT(promised_stream_->ReadResponseHeaders(callback_.callback()), |
| 1928 | IsOk()); |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 1929 | |
| 1930 | // As will be the body. |
| 1931 | EXPECT_EQ( |
| 1932 | static_cast<int>(strlen(kResponseBody)), |
| 1933 | promised_stream_->ReadResponseBody( |
| 1934 | read_buffer_.get(), read_buffer_->size(), callback_.callback())); |
| 1935 | EXPECT_TRUE(promised_stream_->IsResponseBodyComplete()); |
| 1936 | EXPECT_TRUE(AtEof()); |
| 1937 | |
| 1938 | EXPECT_EQ(0, stream_->GetTotalSentBytes()); |
| 1939 | EXPECT_EQ(0, stream_->GetTotalReceivedBytes()); |
| 1940 | EXPECT_EQ(0, promised_stream_->GetTotalSentBytes()); |
| 1941 | EXPECT_EQ(static_cast<int64_t>(spdy_response_headers_frame_length + |
| 1942 | strlen(kResponseBody)), |
| 1943 | promised_stream_->GetTotalReceivedBytes()); |
| 1944 | } |
| 1945 | |
| 1946 | TEST_P(QuicHttpStreamTest, ServerPushCrossOriginFail) { |
| 1947 | SetRequest("GET", "/", DEFAULT_PRIORITY); |
| 1948 | Initialize(); |
| 1949 | |
| 1950 | // Initialize the first stream, for receiving the promise on. |
| 1951 | request_.method = "GET"; |
rch | cd37901 | 2017-04-12 21:53:32 | [diff] [blame] | 1952 | request_.url = GURL("https://www.example.org/"); |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 1953 | |
xunjieli | 5fafe14 | 2016-03-23 23:32:54 | [diff] [blame] | 1954 | EXPECT_EQ(OK, |
Steven Valdez | b4ff041 | 2018-01-18 22:39:27 | [diff] [blame] | 1955 | stream_->InitializeStream(&request_, true, DEFAULT_PRIORITY, |
xunjieli | 5fafe14 | 2016-03-23 23:32:54 | [diff] [blame] | 1956 | net_log_.bound(), callback_.callback())); |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 1957 | |
| 1958 | // TODO(ckrasic) - could do this via constructing a PUSH_PROMISE |
| 1959 | // packet, but does it matter? |
| 1960 | push_promise_[":authority"] = "www.notexample.org"; |
Yixin Wang | 81c1118 | 2018-02-05 20:09:57 | [diff] [blame] | 1961 | promise_url_ = SpdyUtils::GetPromisedUrlFromHeaderBlock(push_promise_); |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 1962 | |
| 1963 | ReceivePromise(promise_id_); |
| 1964 | // The promise will have been rejected because the cert doesn't |
| 1965 | // match. |
| 1966 | EXPECT_EQ(session_->GetPromisedByUrl(promise_url_), nullptr); |
| 1967 | } |
| 1968 | |
| 1969 | TEST_P(QuicHttpStreamTest, ServerPushVaryCheckOK) { |
| 1970 | SetRequest("GET", "/", DEFAULT_PRIORITY); |
| 1971 | Initialize(); |
| 1972 | |
| 1973 | // Initialize the first stream, for receiving the promise on. |
| 1974 | request_.method = "GET"; |
rch | cd37901 | 2017-04-12 21:53:32 | [diff] [blame] | 1975 | request_.url = GURL("https://www.example.org/"); |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 1976 | |
xunjieli | 5fafe14 | 2016-03-23 23:32:54 | [diff] [blame] | 1977 | EXPECT_EQ(OK, |
Steven Valdez | b4ff041 | 2018-01-18 22:39:27 | [diff] [blame] | 1978 | stream_->InitializeStream(&request_, true, DEFAULT_PRIORITY, |
xunjieli | 5fafe14 | 2016-03-23 23:32:54 | [diff] [blame] | 1979 | net_log_.bound(), callback_.callback())); |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 1980 | |
| 1981 | push_promise_["accept-encoding"] = "gzip"; |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 1982 | |
| 1983 | // TODO(ckrasic) - could do this via constructing a PUSH_PROMISE |
| 1984 | // packet, but does it matter? |
| 1985 | ReceivePromise(promise_id_); |
| 1986 | EXPECT_NE(session_->GetPromisedByUrl(promise_url_), nullptr); |
| 1987 | |
| 1988 | request_.url = GURL(promise_url_); |
| 1989 | |
| 1990 | // Make the second stream that will exercise the first step of the |
| 1991 | // server push rendezvous mechanism. |
Steven Valdez | b4ff041 | 2018-01-18 22:39:27 | [diff] [blame] | 1992 | EXPECT_EQ(OK, promised_stream_->InitializeStream( |
| 1993 | &request_, true, DEFAULT_PRIORITY, net_log_.bound(), |
| 1994 | callback_.callback())); |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 1995 | |
| 1996 | headers_.SetHeader("accept-encoding", "gzip"); |
| 1997 | |
| 1998 | // Now sending a matching request will rendezvous with the promised |
| 1999 | // stream, but pending secondary validation. |
| 2000 | EXPECT_EQ(ERR_IO_PENDING, promised_stream_->SendRequest( |
| 2001 | headers_, &response_, callback_.callback())); |
| 2002 | |
| 2003 | // Receive the promised response headers. |
| 2004 | promised_response_["vary"] = "accept-encoding"; |
bnc | 94893a7 | 2016-06-30 13:45:25 | [diff] [blame] | 2005 | response_headers_ = promised_response_.Clone(); |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 2006 | size_t spdy_response_headers_frame_length; |
| 2007 | ProcessPacket(InnerConstructResponseHeadersPacket( |
| 2008 | 1, promise_id_, false, &spdy_response_headers_frame_length)); |
| 2009 | |
| 2010 | // Receive the promised response body. |
| 2011 | const char kResponseBody[] = "Hello world!"; |
alyssar | 2adf3ac | 2016-05-03 17:12:58 | [diff] [blame] | 2012 | ProcessPacket(InnerConstructDataPacket(2, promise_id_, false, kFin, 0, |
| 2013 | kResponseBody, &server_maker_)); |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 2014 | |
fdoray | 92e35a7 | 2016-06-10 15:54:55 | [diff] [blame] | 2015 | base::RunLoop().RunUntilIdle(); |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 2016 | |
| 2017 | // Rendezvous should have succeeded now, so the promised stream |
| 2018 | // should point at our push stream, and we should be able read |
| 2019 | // headers and data from it. |
robpercival | 214763f | 2016-07-01 23:27:01 | [diff] [blame] | 2020 | EXPECT_THAT(callback_.WaitForResult(), IsOk()); |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 2021 | |
| 2022 | EXPECT_EQ( |
| 2023 | QuicHttpStreamPeer::GetQuicChromiumClientStream(promised_stream_.get()) |
| 2024 | ->id(), |
| 2025 | promise_id_); |
| 2026 | |
robpercival | 214763f | 2016-07-01 23:27:01 | [diff] [blame] | 2027 | EXPECT_THAT(promised_stream_->ReadResponseHeaders(callback_.callback()), |
| 2028 | IsOk()); |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 2029 | |
| 2030 | EXPECT_EQ( |
| 2031 | static_cast<int>(strlen(kResponseBody)), |
| 2032 | promised_stream_->ReadResponseBody( |
| 2033 | read_buffer_.get(), read_buffer_->size(), callback_.callback())); |
| 2034 | |
| 2035 | // Callback should return |
| 2036 | EXPECT_TRUE(promised_stream_->IsResponseBodyComplete()); |
| 2037 | EXPECT_TRUE(AtEof()); |
| 2038 | |
| 2039 | EXPECT_EQ(0, stream_->GetTotalSentBytes()); |
| 2040 | EXPECT_EQ(0, stream_->GetTotalReceivedBytes()); |
| 2041 | EXPECT_EQ(0, promised_stream_->GetTotalSentBytes()); |
| 2042 | EXPECT_EQ(static_cast<int64_t>(spdy_response_headers_frame_length + |
| 2043 | strlen(kResponseBody)), |
| 2044 | promised_stream_->GetTotalReceivedBytes()); |
| 2045 | } |
| 2046 | |
| 2047 | TEST_P(QuicHttpStreamTest, ServerPushVaryCheckFail) { |
| 2048 | SetRequest("GET", "/", DEFAULT_PRIORITY); |
| 2049 | request_headers_[":scheme"] = "https"; |
| 2050 | request_headers_[":path"] = "/bar"; |
| 2051 | request_headers_["accept-encoding"] = "sdch"; |
| 2052 | |
| 2053 | size_t spdy_request_header_frame_length; |
fayang | 3bcb8b50 | 2016-12-07 21:44:37 | [diff] [blame] | 2054 | QuicStreamOffset header_stream_offset = 0; |
rch | 5cb52246 | 2017-04-25 20:18:36 | [diff] [blame] | 2055 | AddWrite(ConstructInitialSettingsPacket(&header_stream_offset)); |
Yixin Wang | b470bc88 | 2018-02-15 18:43:57 | [diff] [blame] | 2056 | |
| 2057 | QuicPacketNumber client_packet_number = 2; |
| 2058 | if (client_headers_include_h2_stream_dependency_ && |
| 2059 | version_ > QUIC_VERSION_42) { |
| 2060 | AddWrite(ConstructClientPriorityPacket( |
| 2061 | client_packet_number++, kIncludeVersion, promise_id_, 0, |
| 2062 | DEFAULT_PRIORITY, &header_stream_offset)); |
| 2063 | } |
| 2064 | AddWrite(ConstructClientRstStreamVaryMismatchPacket(client_packet_number++)); |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 2065 | AddWrite(InnerConstructRequestHeadersPacket( |
Yixin Wang | b470bc88 | 2018-02-15 18:43:57 | [diff] [blame] | 2066 | client_packet_number++, stream_id_ + 2, !kIncludeVersion, kFin, |
| 2067 | DEFAULT_PRIORITY, promise_id_, &spdy_request_header_frame_length, |
| 2068 | &header_stream_offset)); |
| 2069 | AddWrite(ConstructClientAckPacket(client_packet_number++, 3, 1, 1)); |
| 2070 | AddWrite(ConstructClientRstStreamCancelledPacket(client_packet_number++)); |
| 2071 | |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 2072 | Initialize(); |
| 2073 | |
| 2074 | // Initialize the first stream, for receiving the promise on. |
| 2075 | request_.method = "GET"; |
rch | cd37901 | 2017-04-12 21:53:32 | [diff] [blame] | 2076 | request_.url = GURL("https://www.example.org/"); |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 2077 | |
xunjieli | 5fafe14 | 2016-03-23 23:32:54 | [diff] [blame] | 2078 | EXPECT_EQ(OK, |
Steven Valdez | b4ff041 | 2018-01-18 22:39:27 | [diff] [blame] | 2079 | stream_->InitializeStream(&request_, true, DEFAULT_PRIORITY, |
xunjieli | 5fafe14 | 2016-03-23 23:32:54 | [diff] [blame] | 2080 | net_log_.bound(), callback_.callback())); |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 2081 | |
| 2082 | push_promise_["accept-encoding"] = "gzip"; |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 2083 | |
| 2084 | // TODO(ckrasic) - could do this via constructing a PUSH_PROMISE |
| 2085 | // packet, but does it matter? |
| 2086 | ReceivePromise(promise_id_); |
| 2087 | EXPECT_NE(session_->GetPromisedByUrl(promise_url_), nullptr); |
| 2088 | |
| 2089 | request_.url = GURL(promise_url_); |
| 2090 | |
| 2091 | // Make the second stream that will exercise the first step of the |
| 2092 | // server push rendezvous mechanism. |
Steven Valdez | b4ff041 | 2018-01-18 22:39:27 | [diff] [blame] | 2093 | EXPECT_EQ(OK, promised_stream_->InitializeStream( |
| 2094 | &request_, true, DEFAULT_PRIORITY, net_log_.bound(), |
| 2095 | callback_.callback())); |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 2096 | |
| 2097 | headers_.SetHeader("accept-encoding", "sdch"); |
| 2098 | |
| 2099 | // Now sending a matching request will rendezvous with the promised |
| 2100 | // stream, but pending secondary validation. |
| 2101 | EXPECT_EQ(ERR_IO_PENDING, promised_stream_->SendRequest( |
| 2102 | headers_, &response_, callback_.callback())); |
| 2103 | |
| 2104 | // Receive the promised response headers. |
| 2105 | promised_response_["vary"] = "accept-encoding"; |
bnc | 94893a7 | 2016-06-30 13:45:25 | [diff] [blame] | 2106 | response_headers_ = promised_response_.Clone(); |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 2107 | size_t spdy_response_headers_frame_length; |
| 2108 | ProcessPacket(InnerConstructResponseHeadersPacket( |
| 2109 | 1, promise_id_, false, &spdy_response_headers_frame_length)); |
| 2110 | |
fdoray | 92e35a7 | 2016-06-10 15:54:55 | [diff] [blame] | 2111 | base::RunLoop().RunUntilIdle(); |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 2112 | |
| 2113 | // Rendezvous should have failed due to vary mismatch, so the |
| 2114 | // promised stream should have been aborted, and instead we have a |
| 2115 | // new, regular client initiated stream. |
robpercival | 214763f | 2016-07-01 23:27:01 | [diff] [blame] | 2116 | EXPECT_THAT(callback_.WaitForResult(), IsOk()); |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 2117 | |
| 2118 | // Not a server-initiated stream. |
| 2119 | EXPECT_NE( |
| 2120 | QuicHttpStreamPeer::GetQuicChromiumClientStream(promised_stream_.get()) |
| 2121 | ->id(), |
| 2122 | promise_id_); |
| 2123 | |
| 2124 | // Instead, a new client-initiated stream. |
| 2125 | EXPECT_EQ( |
| 2126 | QuicHttpStreamPeer::GetQuicChromiumClientStream(promised_stream_.get()) |
| 2127 | ->id(), |
| 2128 | stream_id_ + 2); |
| 2129 | |
| 2130 | // After rendezvous failure, the push stream has been cancelled. |
| 2131 | EXPECT_EQ(session_->GetPromisedByUrl(promise_url_), nullptr); |
| 2132 | |
| 2133 | // The rest of the test verifies that the retried as |
| 2134 | // client-initiated version of |promised_stream_| works as intended. |
| 2135 | |
| 2136 | // Ack the request. |
wangyix | 6444ffe | 2017-04-25 17:49:49 | [diff] [blame] | 2137 | ProcessPacket(ConstructServerAckPacket(2, 0, 0, 0)); |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 2138 | |
bnc | 614a92d3 | 2016-04-04 13:56:07 | [diff] [blame] | 2139 | SetResponse("404 Not Found", string()); |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 2140 | size_t spdy_response_header_frame_length; |
| 2141 | ProcessPacket(InnerConstructResponseHeadersPacket( |
| 2142 | 3, stream_id_ + 2, kFin, &spdy_response_header_frame_length)); |
| 2143 | |
fdoray | 92e35a7 | 2016-06-10 15:54:55 | [diff] [blame] | 2144 | base::RunLoop().RunUntilIdle(); |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 2145 | |
robpercival | 214763f | 2016-07-01 23:27:01 | [diff] [blame] | 2146 | EXPECT_THAT(promised_stream_->ReadResponseHeaders(callback_.callback()), |
| 2147 | IsOk()); |
ckrasic | 3865ee0f | 2016-02-29 22:04:56 | [diff] [blame] | 2148 | ASSERT_TRUE(response_.headers.get()); |
| 2149 | EXPECT_EQ(404, response_.headers->response_code()); |
| 2150 | EXPECT_TRUE(response_.headers->HasHeaderValue("Content-Type", "text/plain")); |
| 2151 | EXPECT_FALSE(response_.response_time.is_null()); |
| 2152 | EXPECT_FALSE(response_.request_time.is_null()); |
| 2153 | |
| 2154 | // There is no body, so this should return immediately. |
| 2155 | EXPECT_EQ( |
| 2156 | 0, promised_stream_->ReadResponseBody( |
| 2157 | read_buffer_.get(), read_buffer_->size(), callback_.callback())); |
| 2158 | EXPECT_TRUE(promised_stream_->IsResponseBodyComplete()); |
| 2159 | |
| 2160 | stream_->Close(true); |
| 2161 | |
| 2162 | EXPECT_TRUE(AtEof()); |
| 2163 | |
| 2164 | // QuicHttpStream::GetTotalSent/ReceivedBytes currently only includes the |
| 2165 | // headers and payload. |
| 2166 | EXPECT_EQ(static_cast<int64_t>(spdy_request_header_frame_length), |
| 2167 | promised_stream_->GetTotalSentBytes()); |
| 2168 | EXPECT_EQ(static_cast<int64_t>(spdy_response_header_frame_length), |
| 2169 | promised_stream_->GetTotalReceivedBytes()); |
| 2170 | } |
| 2171 | |
maksim.sisov | 84e20c9 | 2016-06-23 08:49:34 | [diff] [blame] | 2172 | TEST_P(QuicHttpStreamTest, DataReadErrorSynchronous) { |
| 2173 | SetRequest("POST", "/", DEFAULT_PRIORITY); |
| 2174 | size_t spdy_request_headers_frame_length; |
fayang | 3bcb8b50 | 2016-12-07 21:44:37 | [diff] [blame] | 2175 | QuicStreamOffset header_stream_offset = 0; |
rch | 5cb52246 | 2017-04-25 20:18:36 | [diff] [blame] | 2176 | AddWrite(ConstructInitialSettingsPacket(&header_stream_offset)); |
fayang | 3bcb8b50 | 2016-12-07 21:44:37 | [diff] [blame] | 2177 | AddWrite(InnerConstructRequestHeadersPacket( |
ckrasic | bf2f59c | 2017-05-04 23:54:36 | [diff] [blame] | 2178 | 2, GetNthClientInitiatedStreamId(0), kIncludeVersion, !kFin, |
| 2179 | DEFAULT_PRIORITY, &spdy_request_headers_frame_length, |
| 2180 | &header_stream_offset)); |
fayang | 3bcb8b50 | 2016-12-07 21:44:37 | [diff] [blame] | 2181 | AddWrite(ConstructClientRstStreamErrorPacket(3, kIncludeVersion)); |
maksim.sisov | 84e20c9 | 2016-06-23 08:49:34 | [diff] [blame] | 2182 | |
| 2183 | Initialize(); |
| 2184 | |
Jeremy Roman | 0579ed6 | 2017-08-29 15:56:19 | [diff] [blame] | 2185 | upload_data_stream_ = std::make_unique<ReadErrorUploadDataStream>( |
maksim.sisov | 84e20c9 | 2016-06-23 08:49:34 | [diff] [blame] | 2186 | ReadErrorUploadDataStream::FailureMode::SYNC); |
| 2187 | request_.method = "POST"; |
rch | cd37901 | 2017-04-12 21:53:32 | [diff] [blame] | 2188 | request_.url = GURL("https://www.example.org/"); |
rch | 97827ee | 2017-05-24 23:49:12 | [diff] [blame] | 2189 | request_.upload_data_stream = upload_data_stream_.get(); |
maksim.sisov | 84e20c9 | 2016-06-23 08:49:34 | [diff] [blame] | 2190 | ASSERT_EQ(OK, request_.upload_data_stream->Init( |
tfarina | 4283411 | 2016-09-22 13:38:20 | [diff] [blame] | 2191 | TestCompletionCallback().callback(), NetLogWithSource())); |
maksim.sisov | 84e20c9 | 2016-06-23 08:49:34 | [diff] [blame] | 2192 | |
| 2193 | EXPECT_EQ(OK, |
Steven Valdez | b4ff041 | 2018-01-18 22:39:27 | [diff] [blame] | 2194 | stream_->InitializeStream(&request_, false, DEFAULT_PRIORITY, |
maksim.sisov | 84e20c9 | 2016-06-23 08:49:34 | [diff] [blame] | 2195 | net_log_.bound(), callback_.callback())); |
| 2196 | |
| 2197 | int result = stream_->SendRequest(headers_, &response_, callback_.callback()); |
robpercival | 214763f | 2016-07-01 23:27:01 | [diff] [blame] | 2198 | EXPECT_THAT(result, IsError(ERR_FAILED)); |
maksim.sisov | 84e20c9 | 2016-06-23 08:49:34 | [diff] [blame] | 2199 | |
| 2200 | EXPECT_TRUE(AtEof()); |
| 2201 | |
| 2202 | // QuicHttpStream::GetTotalSent/ReceivedBytes includes only headers. |
| 2203 | EXPECT_EQ(static_cast<int64_t>(spdy_request_headers_frame_length), |
| 2204 | stream_->GetTotalSentBytes()); |
| 2205 | EXPECT_EQ(0, stream_->GetTotalReceivedBytes()); |
| 2206 | } |
| 2207 | |
| 2208 | TEST_P(QuicHttpStreamTest, DataReadErrorAsynchronous) { |
| 2209 | SetRequest("POST", "/", DEFAULT_PRIORITY); |
| 2210 | size_t spdy_request_headers_frame_length; |
fayang | 3bcb8b50 | 2016-12-07 21:44:37 | [diff] [blame] | 2211 | QuicStreamOffset header_stream_offset = 0; |
rch | 5cb52246 | 2017-04-25 20:18:36 | [diff] [blame] | 2212 | AddWrite(ConstructInitialSettingsPacket(&header_stream_offset)); |
fayang | 3bcb8b50 | 2016-12-07 21:44:37 | [diff] [blame] | 2213 | AddWrite(InnerConstructRequestHeadersPacket( |
ckrasic | bf2f59c | 2017-05-04 23:54:36 | [diff] [blame] | 2214 | 2, GetNthClientInitiatedStreamId(0), kIncludeVersion, !kFin, |
| 2215 | DEFAULT_PRIORITY, &spdy_request_headers_frame_length, |
| 2216 | &header_stream_offset)); |
fayang | 3bcb8b50 | 2016-12-07 21:44:37 | [diff] [blame] | 2217 | AddWrite(ConstructClientRstStreamErrorPacket(3, !kIncludeVersion)); |
maksim.sisov | 84e20c9 | 2016-06-23 08:49:34 | [diff] [blame] | 2218 | |
| 2219 | Initialize(); |
| 2220 | |
Jeremy Roman | 0579ed6 | 2017-08-29 15:56:19 | [diff] [blame] | 2221 | upload_data_stream_ = std::make_unique<ReadErrorUploadDataStream>( |
maksim.sisov | 84e20c9 | 2016-06-23 08:49:34 | [diff] [blame] | 2222 | ReadErrorUploadDataStream::FailureMode::ASYNC); |
| 2223 | request_.method = "POST"; |
rch | cd37901 | 2017-04-12 21:53:32 | [diff] [blame] | 2224 | request_.url = GURL("https://www.example.org/"); |
rch | 97827ee | 2017-05-24 23:49:12 | [diff] [blame] | 2225 | request_.upload_data_stream = upload_data_stream_.get(); |
maksim.sisov | 84e20c9 | 2016-06-23 08:49:34 | [diff] [blame] | 2226 | ASSERT_EQ(OK, request_.upload_data_stream->Init( |
tfarina | 4283411 | 2016-09-22 13:38:20 | [diff] [blame] | 2227 | TestCompletionCallback().callback(), NetLogWithSource())); |
maksim.sisov | 84e20c9 | 2016-06-23 08:49:34 | [diff] [blame] | 2228 | |
| 2229 | EXPECT_EQ(OK, |
Steven Valdez | b4ff041 | 2018-01-18 22:39:27 | [diff] [blame] | 2230 | stream_->InitializeStream(&request_, false, DEFAULT_PRIORITY, |
maksim.sisov | 84e20c9 | 2016-06-23 08:49:34 | [diff] [blame] | 2231 | net_log_.bound(), callback_.callback())); |
| 2232 | |
| 2233 | int result = stream_->SendRequest(headers_, &response_, callback_.callback()); |
| 2234 | |
wangyix | 6444ffe | 2017-04-25 17:49:49 | [diff] [blame] | 2235 | ProcessPacket(ConstructServerAckPacket(1, 0, 0, 0)); |
maksim.sisov | 84e20c9 | 2016-06-23 08:49:34 | [diff] [blame] | 2236 | SetResponse("200 OK", string()); |
| 2237 | |
robpercival | 214763f | 2016-07-01 23:27:01 | [diff] [blame] | 2238 | EXPECT_THAT(result, IsError(ERR_IO_PENDING)); |
| 2239 | EXPECT_THAT(callback_.GetResult(result), IsError(ERR_FAILED)); |
maksim.sisov | 84e20c9 | 2016-06-23 08:49:34 | [diff] [blame] | 2240 | |
| 2241 | EXPECT_TRUE(AtEof()); |
| 2242 | |
| 2243 | // QuicHttpStream::GetTotalSent/ReceivedBytes includes only headers. |
| 2244 | EXPECT_EQ(static_cast<int64_t>(spdy_request_headers_frame_length), |
| 2245 | stream_->GetTotalSentBytes()); |
| 2246 | EXPECT_EQ(0, stream_->GetTotalReceivedBytes()); |
| 2247 | } |
| 2248 | |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 2249 | } // namespace test |
[email protected] | f702d57 | 2012-12-04 15:56:20 | [diff] [blame] | 2250 | } // namespace net |