xref: /aosp_15_r20/external/cronet/net/quic/quic_http_stream.cc (revision 6777b5387eb2ff775bb5750e3f5d96f37fb7352b)
1*6777b538SAndroid Build Coastguard Worker // Copyright 2012 The Chromium Authors
2*6777b538SAndroid Build Coastguard Worker // Use of this source code is governed by a BSD-style license that can be
3*6777b538SAndroid Build Coastguard Worker // found in the LICENSE file.
4*6777b538SAndroid Build Coastguard Worker 
5*6777b538SAndroid Build Coastguard Worker #include "net/quic/quic_http_stream.h"
6*6777b538SAndroid Build Coastguard Worker 
7*6777b538SAndroid Build Coastguard Worker #include <set>
8*6777b538SAndroid Build Coastguard Worker #include <string_view>
9*6777b538SAndroid Build Coastguard Worker #include <utility>
10*6777b538SAndroid Build Coastguard Worker 
11*6777b538SAndroid Build Coastguard Worker #include "base/auto_reset.h"
12*6777b538SAndroid Build Coastguard Worker #include "base/functional/bind.h"
13*6777b538SAndroid Build Coastguard Worker #include "base/metrics/histogram_functions.h"
14*6777b538SAndroid Build Coastguard Worker #include "base/strings/string_split.h"
15*6777b538SAndroid Build Coastguard Worker #include "base/task/single_thread_task_runner.h"
16*6777b538SAndroid Build Coastguard Worker #include "net/base/features.h"
17*6777b538SAndroid Build Coastguard Worker #include "net/base/ip_endpoint.h"
18*6777b538SAndroid Build Coastguard Worker #include "net/base/load_flags.h"
19*6777b538SAndroid Build Coastguard Worker #include "net/base/net_errors.h"
20*6777b538SAndroid Build Coastguard Worker #include "net/http/http_response_headers.h"
21*6777b538SAndroid Build Coastguard Worker #include "net/http/http_status_code.h"
22*6777b538SAndroid Build Coastguard Worker #include "net/http/http_util.h"
23*6777b538SAndroid Build Coastguard Worker #include "net/log/net_log.h"
24*6777b538SAndroid Build Coastguard Worker #include "net/log/net_log_event_type.h"
25*6777b538SAndroid Build Coastguard Worker #include "net/log/net_log_source.h"
26*6777b538SAndroid Build Coastguard Worker #include "net/quic/quic_http_utils.h"
27*6777b538SAndroid Build Coastguard Worker #include "net/spdy/spdy_http_utils.h"
28*6777b538SAndroid Build Coastguard Worker #include "net/ssl/ssl_info.h"
29*6777b538SAndroid Build Coastguard Worker #include "net/third_party/quiche/src/quiche/quic/core/http/spdy_utils.h"
30*6777b538SAndroid Build Coastguard Worker #include "net/third_party/quiche/src/quiche/quic/core/quic_stream_sequencer.h"
31*6777b538SAndroid Build Coastguard Worker #include "net/third_party/quiche/src/quiche/quic/core/quic_utils.h"
32*6777b538SAndroid Build Coastguard Worker #include "net/third_party/quiche/src/quiche/spdy/core/spdy_frame_builder.h"
33*6777b538SAndroid Build Coastguard Worker #include "net/third_party/quiche/src/quiche/spdy/core/spdy_framer.h"
34*6777b538SAndroid Build Coastguard Worker #include "url/origin.h"
35*6777b538SAndroid Build Coastguard Worker #include "url/scheme_host_port.h"
36*6777b538SAndroid Build Coastguard Worker 
37*6777b538SAndroid Build Coastguard Worker namespace net {
38*6777b538SAndroid Build Coastguard Worker 
QuicHttpStream(std::unique_ptr<QuicChromiumClientSession::Handle> session,std::set<std::string> dns_aliases)39*6777b538SAndroid Build Coastguard Worker QuicHttpStream::QuicHttpStream(
40*6777b538SAndroid Build Coastguard Worker     std::unique_ptr<QuicChromiumClientSession::Handle> session,
41*6777b538SAndroid Build Coastguard Worker     std::set<std::string> dns_aliases)
42*6777b538SAndroid Build Coastguard Worker     : MultiplexedHttpStream(std::move(session)),
43*6777b538SAndroid Build Coastguard Worker       dns_aliases_(std::move(dns_aliases)) {}
44*6777b538SAndroid Build Coastguard Worker 
~QuicHttpStream()45*6777b538SAndroid Build Coastguard Worker QuicHttpStream::~QuicHttpStream() {
46*6777b538SAndroid Build Coastguard Worker   CHECK(!in_loop_);
47*6777b538SAndroid Build Coastguard Worker   Close(false);
48*6777b538SAndroid Build Coastguard Worker }
49*6777b538SAndroid Build Coastguard Worker 
ConnectionInfoFromQuicVersion(quic::ParsedQuicVersion quic_version)50*6777b538SAndroid Build Coastguard Worker HttpConnectionInfo QuicHttpStream::ConnectionInfoFromQuicVersion(
51*6777b538SAndroid Build Coastguard Worker     quic::ParsedQuicVersion quic_version) {
52*6777b538SAndroid Build Coastguard Worker   switch (quic_version.transport_version) {
53*6777b538SAndroid Build Coastguard Worker     case quic::QUIC_VERSION_UNSUPPORTED:
54*6777b538SAndroid Build Coastguard Worker       return HttpConnectionInfo::kQUIC_UNKNOWN_VERSION;
55*6777b538SAndroid Build Coastguard Worker     case quic::QUIC_VERSION_46:
56*6777b538SAndroid Build Coastguard Worker       return HttpConnectionInfo::kQUIC_46;
57*6777b538SAndroid Build Coastguard Worker     case quic::QUIC_VERSION_IETF_DRAFT_29:
58*6777b538SAndroid Build Coastguard Worker       DCHECK(quic_version.UsesTls());
59*6777b538SAndroid Build Coastguard Worker       return HttpConnectionInfo::kQUIC_DRAFT_29;
60*6777b538SAndroid Build Coastguard Worker     case quic::QUIC_VERSION_IETF_RFC_V1:
61*6777b538SAndroid Build Coastguard Worker       DCHECK(quic_version.UsesTls());
62*6777b538SAndroid Build Coastguard Worker       return HttpConnectionInfo::kQUIC_RFC_V1;
63*6777b538SAndroid Build Coastguard Worker     case quic::QUIC_VERSION_RESERVED_FOR_NEGOTIATION:
64*6777b538SAndroid Build Coastguard Worker       return HttpConnectionInfo::kQUIC_999;
65*6777b538SAndroid Build Coastguard Worker     case quic::QUIC_VERSION_IETF_RFC_V2:
66*6777b538SAndroid Build Coastguard Worker       DCHECK(quic_version.UsesTls());
67*6777b538SAndroid Build Coastguard Worker       return HttpConnectionInfo::kQUIC_2_DRAFT_8;
68*6777b538SAndroid Build Coastguard Worker   }
69*6777b538SAndroid Build Coastguard Worker   NOTREACHED();
70*6777b538SAndroid Build Coastguard Worker   return HttpConnectionInfo::kQUIC_UNKNOWN_VERSION;
71*6777b538SAndroid Build Coastguard Worker }
72*6777b538SAndroid Build Coastguard Worker 
RegisterRequest(const HttpRequestInfo * request_info)73*6777b538SAndroid Build Coastguard Worker void QuicHttpStream::RegisterRequest(const HttpRequestInfo* request_info) {
74*6777b538SAndroid Build Coastguard Worker   DCHECK(request_info);
75*6777b538SAndroid Build Coastguard Worker   DCHECK(request_info->traffic_annotation.is_valid());
76*6777b538SAndroid Build Coastguard Worker   request_info_ = request_info;
77*6777b538SAndroid Build Coastguard Worker }
78*6777b538SAndroid Build Coastguard Worker 
InitializeStream(bool can_send_early,RequestPriority priority,const NetLogWithSource & stream_net_log,CompletionOnceCallback callback)79*6777b538SAndroid Build Coastguard Worker int QuicHttpStream::InitializeStream(bool can_send_early,
80*6777b538SAndroid Build Coastguard Worker                                      RequestPriority priority,
81*6777b538SAndroid Build Coastguard Worker                                      const NetLogWithSource& stream_net_log,
82*6777b538SAndroid Build Coastguard Worker                                      CompletionOnceCallback callback) {
83*6777b538SAndroid Build Coastguard Worker   CHECK(callback_.is_null());
84*6777b538SAndroid Build Coastguard Worker   DCHECK(request_info_);
85*6777b538SAndroid Build Coastguard Worker   DCHECK(!stream_);
86*6777b538SAndroid Build Coastguard Worker 
87*6777b538SAndroid Build Coastguard Worker   // HttpNetworkTransaction will retry any request that fails with
88*6777b538SAndroid Build Coastguard Worker   // ERR_QUIC_HANDSHAKE_FAILED. It will retry any request with
89*6777b538SAndroid Build Coastguard Worker   // ERR_CONNECTION_CLOSED so long as the connection has been used for other
90*6777b538SAndroid Build Coastguard Worker   // streams first and headers have not yet been received.
91*6777b538SAndroid Build Coastguard Worker   if (!quic_session()->IsConnected()) {
92*6777b538SAndroid Build Coastguard Worker     return GetResponseStatus();
93*6777b538SAndroid Build Coastguard Worker   }
94*6777b538SAndroid Build Coastguard Worker 
95*6777b538SAndroid Build Coastguard Worker   stream_net_log.AddEventReferencingSource(
96*6777b538SAndroid Build Coastguard Worker       NetLogEventType::HTTP_STREAM_REQUEST_BOUND_TO_QUIC_SESSION,
97*6777b538SAndroid Build Coastguard Worker       quic_session()->net_log().source());
98*6777b538SAndroid Build Coastguard Worker   stream_net_log.AddEventWithIntParams(
99*6777b538SAndroid Build Coastguard Worker       NetLogEventType::QUIC_CONNECTION_MIGRATION_MODE,
100*6777b538SAndroid Build Coastguard Worker       "connection_migration_mode",
101*6777b538SAndroid Build Coastguard Worker       static_cast<int>(quic_session()->connection_migration_mode()));
102*6777b538SAndroid Build Coastguard Worker 
103*6777b538SAndroid Build Coastguard Worker   stream_net_log_ = stream_net_log;
104*6777b538SAndroid Build Coastguard Worker   can_send_early_ = can_send_early;
105*6777b538SAndroid Build Coastguard Worker   request_time_ = base::Time::Now();
106*6777b538SAndroid Build Coastguard Worker   priority_ = priority;
107*6777b538SAndroid Build Coastguard Worker 
108*6777b538SAndroid Build Coastguard Worker   SaveSSLInfo();
109*6777b538SAndroid Build Coastguard Worker 
110*6777b538SAndroid Build Coastguard Worker   next_state_ = STATE_REQUEST_STREAM;
111*6777b538SAndroid Build Coastguard Worker   int rv = DoLoop(OK);
112*6777b538SAndroid Build Coastguard Worker   if (rv == ERR_IO_PENDING) {
113*6777b538SAndroid Build Coastguard Worker     callback_ = std::move(callback);
114*6777b538SAndroid Build Coastguard Worker   }
115*6777b538SAndroid Build Coastguard Worker 
116*6777b538SAndroid Build Coastguard Worker   return MapStreamError(rv);
117*6777b538SAndroid Build Coastguard Worker }
118*6777b538SAndroid Build Coastguard Worker 
SendRequest(const HttpRequestHeaders & request_headers,HttpResponseInfo * response,CompletionOnceCallback callback)119*6777b538SAndroid Build Coastguard Worker int QuicHttpStream::SendRequest(const HttpRequestHeaders& request_headers,
120*6777b538SAndroid Build Coastguard Worker                                 HttpResponseInfo* response,
121*6777b538SAndroid Build Coastguard Worker                                 CompletionOnceCallback callback) {
122*6777b538SAndroid Build Coastguard Worker   CHECK(!request_body_stream_);
123*6777b538SAndroid Build Coastguard Worker   CHECK(!response_info_);
124*6777b538SAndroid Build Coastguard Worker   CHECK(callback_.is_null());
125*6777b538SAndroid Build Coastguard Worker   CHECK(!callback.is_null());
126*6777b538SAndroid Build Coastguard Worker   CHECK(response);
127*6777b538SAndroid Build Coastguard Worker 
128*6777b538SAndroid Build Coastguard Worker   if (!stream_ || !quic_session()->IsConnected()) {
129*6777b538SAndroid Build Coastguard Worker     return GetResponseStatus();
130*6777b538SAndroid Build Coastguard Worker   }
131*6777b538SAndroid Build Coastguard Worker 
132*6777b538SAndroid Build Coastguard Worker   // Store the serialized request headers.
133*6777b538SAndroid Build Coastguard Worker   CreateSpdyHeadersFromHttpRequest(*request_info_, priority_, request_headers,
134*6777b538SAndroid Build Coastguard Worker                                    &request_headers_);
135*6777b538SAndroid Build Coastguard Worker 
136*6777b538SAndroid Build Coastguard Worker   // Store the request body.
137*6777b538SAndroid Build Coastguard Worker   request_body_stream_ = request_info_->upload_data_stream;
138*6777b538SAndroid Build Coastguard Worker   if (request_body_stream_) {
139*6777b538SAndroid Build Coastguard Worker     // TODO(rch): Can we be more precise about when to allocate
140*6777b538SAndroid Build Coastguard Worker     // raw_request_body_buf_. Removed the following check. DoReadRequestBody()
141*6777b538SAndroid Build Coastguard Worker     // was being called even if we didn't yet allocate raw_request_body_buf_.
142*6777b538SAndroid Build Coastguard Worker     //   && (request_body_stream_->size() ||
143*6777b538SAndroid Build Coastguard Worker     //       request_body_stream_->is_chunked()))
144*6777b538SAndroid Build Coastguard Worker     // Set the body buffer size to be the size of the body clamped
145*6777b538SAndroid Build Coastguard Worker     // into the range [10 * quic::kMaxOutgoingPacketSize, 256 *
146*6777b538SAndroid Build Coastguard Worker     // quic::kMaxOutgoingPacketSize]. With larger bodies, larger buffers reduce
147*6777b538SAndroid Build Coastguard Worker     // CPU usage.
148*6777b538SAndroid Build Coastguard Worker     raw_request_body_buf_ =
149*6777b538SAndroid Build Coastguard Worker         base::MakeRefCounted<IOBufferWithSize>(static_cast<size_t>(
150*6777b538SAndroid Build Coastguard Worker             std::max(10 * quic::kMaxOutgoingPacketSize,
151*6777b538SAndroid Build Coastguard Worker                      std::min(request_body_stream_->size(),
152*6777b538SAndroid Build Coastguard Worker                               256 * quic::kMaxOutgoingPacketSize))));
153*6777b538SAndroid Build Coastguard Worker     // The request body buffer is empty at first.
154*6777b538SAndroid Build Coastguard Worker     request_body_buf_ =
155*6777b538SAndroid Build Coastguard Worker         base::MakeRefCounted<DrainableIOBuffer>(raw_request_body_buf_, 0);
156*6777b538SAndroid Build Coastguard Worker   }
157*6777b538SAndroid Build Coastguard Worker 
158*6777b538SAndroid Build Coastguard Worker   // Store the response info.
159*6777b538SAndroid Build Coastguard Worker   response_info_ = response;
160*6777b538SAndroid Build Coastguard Worker 
161*6777b538SAndroid Build Coastguard Worker   // Put the peer's IP address and port into the response.
162*6777b538SAndroid Build Coastguard Worker   IPEndPoint address;
163*6777b538SAndroid Build Coastguard Worker   int rv = quic_session()->GetPeerAddress(&address);
164*6777b538SAndroid Build Coastguard Worker   if (rv != OK) {
165*6777b538SAndroid Build Coastguard Worker     return rv;
166*6777b538SAndroid Build Coastguard Worker   }
167*6777b538SAndroid Build Coastguard Worker   response_info_->remote_endpoint = address;
168*6777b538SAndroid Build Coastguard Worker 
169*6777b538SAndroid Build Coastguard Worker   next_state_ = STATE_SET_REQUEST_PRIORITY;
170*6777b538SAndroid Build Coastguard Worker   rv = DoLoop(OK);
171*6777b538SAndroid Build Coastguard Worker 
172*6777b538SAndroid Build Coastguard Worker   if (rv == ERR_IO_PENDING) {
173*6777b538SAndroid Build Coastguard Worker     callback_ = std::move(callback);
174*6777b538SAndroid Build Coastguard Worker   }
175*6777b538SAndroid Build Coastguard Worker 
176*6777b538SAndroid Build Coastguard Worker   return rv > 0 ? OK : MapStreamError(rv);
177*6777b538SAndroid Build Coastguard Worker }
178*6777b538SAndroid Build Coastguard Worker 
ReadResponseHeaders(CompletionOnceCallback callback)179*6777b538SAndroid Build Coastguard Worker int QuicHttpStream::ReadResponseHeaders(CompletionOnceCallback callback) {
180*6777b538SAndroid Build Coastguard Worker   CHECK(callback_.is_null());
181*6777b538SAndroid Build Coastguard Worker   CHECK(!callback.is_null());
182*6777b538SAndroid Build Coastguard Worker 
183*6777b538SAndroid Build Coastguard Worker   int rv = stream_->ReadInitialHeaders(
184*6777b538SAndroid Build Coastguard Worker       &response_header_block_,
185*6777b538SAndroid Build Coastguard Worker       base::BindOnce(&QuicHttpStream::OnReadResponseHeadersComplete,
186*6777b538SAndroid Build Coastguard Worker                      weak_factory_.GetWeakPtr()));
187*6777b538SAndroid Build Coastguard Worker 
188*6777b538SAndroid Build Coastguard Worker   if (rv == ERR_IO_PENDING) {
189*6777b538SAndroid Build Coastguard Worker     // Still waiting for the response, return IO_PENDING.
190*6777b538SAndroid Build Coastguard Worker     CHECK(callback_.is_null());
191*6777b538SAndroid Build Coastguard Worker     callback_ = std::move(callback);
192*6777b538SAndroid Build Coastguard Worker     return ERR_IO_PENDING;
193*6777b538SAndroid Build Coastguard Worker   }
194*6777b538SAndroid Build Coastguard Worker 
195*6777b538SAndroid Build Coastguard Worker   if (rv < 0) {
196*6777b538SAndroid Build Coastguard Worker     return MapStreamError(rv);
197*6777b538SAndroid Build Coastguard Worker   }
198*6777b538SAndroid Build Coastguard Worker 
199*6777b538SAndroid Build Coastguard Worker   // Check if we already have the response headers. If so, return synchronously.
200*6777b538SAndroid Build Coastguard Worker   if (response_headers_received_) {
201*6777b538SAndroid Build Coastguard Worker     return OK;
202*6777b538SAndroid Build Coastguard Worker   }
203*6777b538SAndroid Build Coastguard Worker 
204*6777b538SAndroid Build Coastguard Worker   headers_bytes_received_ += rv;
205*6777b538SAndroid Build Coastguard Worker   return ProcessResponseHeaders(response_header_block_);
206*6777b538SAndroid Build Coastguard Worker }
207*6777b538SAndroid Build Coastguard Worker 
ReadResponseBody(IOBuffer * buf,int buf_len,CompletionOnceCallback callback)208*6777b538SAndroid Build Coastguard Worker int QuicHttpStream::ReadResponseBody(IOBuffer* buf,
209*6777b538SAndroid Build Coastguard Worker                                      int buf_len,
210*6777b538SAndroid Build Coastguard Worker                                      CompletionOnceCallback callback) {
211*6777b538SAndroid Build Coastguard Worker   CHECK(callback_.is_null());
212*6777b538SAndroid Build Coastguard Worker   CHECK(!callback.is_null());
213*6777b538SAndroid Build Coastguard Worker   CHECK(!user_buffer_.get());
214*6777b538SAndroid Build Coastguard Worker   CHECK_EQ(0, user_buffer_len_);
215*6777b538SAndroid Build Coastguard Worker 
216*6777b538SAndroid Build Coastguard Worker   // Invalidate HttpRequestInfo pointer. This is to allow the stream to be
217*6777b538SAndroid Build Coastguard Worker   // shared across multiple transactions which might require this
218*6777b538SAndroid Build Coastguard Worker   // stream to outlive the request_info_'s owner.
219*6777b538SAndroid Build Coastguard Worker   // Only allowed when Read state machine starts. It is safe to reset it at
220*6777b538SAndroid Build Coastguard Worker   // this point since request_info_->upload_data_stream is also not needed
221*6777b538SAndroid Build Coastguard Worker   // anymore.
222*6777b538SAndroid Build Coastguard Worker   request_info_ = nullptr;
223*6777b538SAndroid Build Coastguard Worker 
224*6777b538SAndroid Build Coastguard Worker   // If the stream is already closed, there is no body to read.
225*6777b538SAndroid Build Coastguard Worker   if (stream_->IsDoneReading()) {
226*6777b538SAndroid Build Coastguard Worker     return HandleReadComplete(OK);
227*6777b538SAndroid Build Coastguard Worker   }
228*6777b538SAndroid Build Coastguard Worker 
229*6777b538SAndroid Build Coastguard Worker   int rv = stream_->ReadBody(buf, buf_len,
230*6777b538SAndroid Build Coastguard Worker                              base::BindOnce(&QuicHttpStream::OnReadBodyComplete,
231*6777b538SAndroid Build Coastguard Worker                                             weak_factory_.GetWeakPtr()));
232*6777b538SAndroid Build Coastguard Worker   if (rv == ERR_IO_PENDING) {
233*6777b538SAndroid Build Coastguard Worker     callback_ = std::move(callback);
234*6777b538SAndroid Build Coastguard Worker     user_buffer_ = buf;
235*6777b538SAndroid Build Coastguard Worker     user_buffer_len_ = buf_len;
236*6777b538SAndroid Build Coastguard Worker     return ERR_IO_PENDING;
237*6777b538SAndroid Build Coastguard Worker   }
238*6777b538SAndroid Build Coastguard Worker 
239*6777b538SAndroid Build Coastguard Worker   if (rv < 0) {
240*6777b538SAndroid Build Coastguard Worker     return MapStreamError(rv);
241*6777b538SAndroid Build Coastguard Worker   }
242*6777b538SAndroid Build Coastguard Worker 
243*6777b538SAndroid Build Coastguard Worker   return HandleReadComplete(rv);
244*6777b538SAndroid Build Coastguard Worker }
245*6777b538SAndroid Build Coastguard Worker 
Close(bool)246*6777b538SAndroid Build Coastguard Worker void QuicHttpStream::Close(bool /*not_reusable*/) {
247*6777b538SAndroid Build Coastguard Worker   session_error_ = ERR_ABORTED;
248*6777b538SAndroid Build Coastguard Worker   SaveResponseStatus();
249*6777b538SAndroid Build Coastguard Worker   // Note: the not_reusable flag has no meaning for QUIC streams.
250*6777b538SAndroid Build Coastguard Worker   if (stream_) {
251*6777b538SAndroid Build Coastguard Worker     stream_->Reset(quic::QUIC_STREAM_CANCELLED);
252*6777b538SAndroid Build Coastguard Worker   }
253*6777b538SAndroid Build Coastguard Worker   ResetStream();
254*6777b538SAndroid Build Coastguard Worker }
255*6777b538SAndroid Build Coastguard Worker 
IsResponseBodyComplete() const256*6777b538SAndroid Build Coastguard Worker bool QuicHttpStream::IsResponseBodyComplete() const {
257*6777b538SAndroid Build Coastguard Worker   return next_state_ == STATE_OPEN && stream_->IsDoneReading();
258*6777b538SAndroid Build Coastguard Worker }
259*6777b538SAndroid Build Coastguard Worker 
IsConnectionReused() const260*6777b538SAndroid Build Coastguard Worker bool QuicHttpStream::IsConnectionReused() const {
261*6777b538SAndroid Build Coastguard Worker   // TODO(rch): do something smarter here.
262*6777b538SAndroid Build Coastguard Worker   return stream_ && stream_->id() > 1;
263*6777b538SAndroid Build Coastguard Worker }
264*6777b538SAndroid Build Coastguard Worker 
GetTotalReceivedBytes() const265*6777b538SAndroid Build Coastguard Worker int64_t QuicHttpStream::GetTotalReceivedBytes() const {
266*6777b538SAndroid Build Coastguard Worker   if (stream_) {
267*6777b538SAndroid Build Coastguard Worker     DCHECK_LE(stream_->NumBytesConsumed(), stream_->stream_bytes_read());
268*6777b538SAndroid Build Coastguard Worker     // Only count the uniquely received bytes.
269*6777b538SAndroid Build Coastguard Worker     return stream_->NumBytesConsumed();
270*6777b538SAndroid Build Coastguard Worker   }
271*6777b538SAndroid Build Coastguard Worker   return closed_stream_received_bytes_;
272*6777b538SAndroid Build Coastguard Worker }
273*6777b538SAndroid Build Coastguard Worker 
GetTotalSentBytes() const274*6777b538SAndroid Build Coastguard Worker int64_t QuicHttpStream::GetTotalSentBytes() const {
275*6777b538SAndroid Build Coastguard Worker   if (stream_) {
276*6777b538SAndroid Build Coastguard Worker     return stream_->stream_bytes_written();
277*6777b538SAndroid Build Coastguard Worker   }
278*6777b538SAndroid Build Coastguard Worker   return closed_stream_sent_bytes_;
279*6777b538SAndroid Build Coastguard Worker }
280*6777b538SAndroid Build Coastguard Worker 
GetLoadTimingInfo(LoadTimingInfo * load_timing_info) const281*6777b538SAndroid Build Coastguard Worker bool QuicHttpStream::GetLoadTimingInfo(LoadTimingInfo* load_timing_info) const {
282*6777b538SAndroid Build Coastguard Worker   bool is_first_stream = closed_is_first_stream_;
283*6777b538SAndroid Build Coastguard Worker   if (stream_) {
284*6777b538SAndroid Build Coastguard Worker     is_first_stream = stream_->IsFirstStream();
285*6777b538SAndroid Build Coastguard Worker     load_timing_info->first_early_hints_time =
286*6777b538SAndroid Build Coastguard Worker         stream_->first_early_hints_time();
287*6777b538SAndroid Build Coastguard Worker     load_timing_info->receive_non_informational_headers_start =
288*6777b538SAndroid Build Coastguard Worker         stream_->headers_received_start_time();
289*6777b538SAndroid Build Coastguard Worker     load_timing_info->receive_headers_start =
290*6777b538SAndroid Build Coastguard Worker         load_timing_info->first_early_hints_time.is_null()
291*6777b538SAndroid Build Coastguard Worker             ? load_timing_info->receive_non_informational_headers_start
292*6777b538SAndroid Build Coastguard Worker             : load_timing_info->first_early_hints_time;
293*6777b538SAndroid Build Coastguard Worker   }
294*6777b538SAndroid Build Coastguard Worker 
295*6777b538SAndroid Build Coastguard Worker   if (is_first_stream) {
296*6777b538SAndroid Build Coastguard Worker     load_timing_info->socket_reused = false;
297*6777b538SAndroid Build Coastguard Worker     load_timing_info->connect_timing = connect_timing_;
298*6777b538SAndroid Build Coastguard Worker   } else {
299*6777b538SAndroid Build Coastguard Worker     load_timing_info->socket_reused = true;
300*6777b538SAndroid Build Coastguard Worker   }
301*6777b538SAndroid Build Coastguard Worker   return true;
302*6777b538SAndroid Build Coastguard Worker }
303*6777b538SAndroid Build Coastguard Worker 
GetAlternativeService(AlternativeService * alternative_service) const304*6777b538SAndroid Build Coastguard Worker bool QuicHttpStream::GetAlternativeService(
305*6777b538SAndroid Build Coastguard Worker     AlternativeService* alternative_service) const {
306*6777b538SAndroid Build Coastguard Worker   alternative_service->protocol = kProtoQUIC;
307*6777b538SAndroid Build Coastguard Worker   const url::SchemeHostPort& destination = quic_session()->destination();
308*6777b538SAndroid Build Coastguard Worker   alternative_service->host = destination.host();
309*6777b538SAndroid Build Coastguard Worker   alternative_service->port = destination.port();
310*6777b538SAndroid Build Coastguard Worker   return true;
311*6777b538SAndroid Build Coastguard Worker }
312*6777b538SAndroid Build Coastguard Worker 
PopulateNetErrorDetails(NetErrorDetails * details)313*6777b538SAndroid Build Coastguard Worker void QuicHttpStream::PopulateNetErrorDetails(NetErrorDetails* details) {
314*6777b538SAndroid Build Coastguard Worker   details->connection_info =
315*6777b538SAndroid Build Coastguard Worker       ConnectionInfoFromQuicVersion(quic_session()->GetQuicVersion());
316*6777b538SAndroid Build Coastguard Worker   quic_session()->PopulateNetErrorDetails(details);
317*6777b538SAndroid Build Coastguard Worker   if (quic_session()->OneRttKeysAvailable() && stream_ &&
318*6777b538SAndroid Build Coastguard Worker       stream_->connection_error() != quic::QUIC_NO_ERROR) {
319*6777b538SAndroid Build Coastguard Worker     details->quic_connection_error = stream_->connection_error();
320*6777b538SAndroid Build Coastguard Worker   }
321*6777b538SAndroid Build Coastguard Worker }
322*6777b538SAndroid Build Coastguard Worker 
SetPriority(RequestPriority priority)323*6777b538SAndroid Build Coastguard Worker void QuicHttpStream::SetPriority(RequestPriority priority) {
324*6777b538SAndroid Build Coastguard Worker   priority_ = priority;
325*6777b538SAndroid Build Coastguard Worker }
326*6777b538SAndroid Build Coastguard Worker 
OnReadResponseHeadersComplete(int rv)327*6777b538SAndroid Build Coastguard Worker void QuicHttpStream::OnReadResponseHeadersComplete(int rv) {
328*6777b538SAndroid Build Coastguard Worker   DCHECK(callback_);
329*6777b538SAndroid Build Coastguard Worker   DCHECK(!response_headers_received_);
330*6777b538SAndroid Build Coastguard Worker   if (rv > 0) {
331*6777b538SAndroid Build Coastguard Worker     headers_bytes_received_ += rv;
332*6777b538SAndroid Build Coastguard Worker     rv = ProcessResponseHeaders(response_header_block_);
333*6777b538SAndroid Build Coastguard Worker   }
334*6777b538SAndroid Build Coastguard Worker   if (rv != ERR_IO_PENDING && !callback_.is_null()) {
335*6777b538SAndroid Build Coastguard Worker     DoCallback(rv);
336*6777b538SAndroid Build Coastguard Worker   }
337*6777b538SAndroid Build Coastguard Worker }
338*6777b538SAndroid Build Coastguard Worker 
GetDnsAliases() const339*6777b538SAndroid Build Coastguard Worker const std::set<std::string>& QuicHttpStream::GetDnsAliases() const {
340*6777b538SAndroid Build Coastguard Worker   return dns_aliases_;
341*6777b538SAndroid Build Coastguard Worker }
342*6777b538SAndroid Build Coastguard Worker 
GetAcceptChViaAlps() const343*6777b538SAndroid Build Coastguard Worker std::string_view QuicHttpStream::GetAcceptChViaAlps() const {
344*6777b538SAndroid Build Coastguard Worker   if (!request_info_) {
345*6777b538SAndroid Build Coastguard Worker     return {};
346*6777b538SAndroid Build Coastguard Worker   }
347*6777b538SAndroid Build Coastguard Worker 
348*6777b538SAndroid Build Coastguard Worker   return session()->GetAcceptChViaAlps(url::SchemeHostPort(request_info_->url));
349*6777b538SAndroid Build Coastguard Worker }
350*6777b538SAndroid Build Coastguard Worker 
351*6777b538SAndroid Build Coastguard Worker std::optional<HttpStream::QuicErrorDetails>
GetQuicErrorDetails() const352*6777b538SAndroid Build Coastguard Worker QuicHttpStream::GetQuicErrorDetails() const {
353*6777b538SAndroid Build Coastguard Worker   QuicErrorDetails details;
354*6777b538SAndroid Build Coastguard Worker   if (stream_) {
355*6777b538SAndroid Build Coastguard Worker     details.connection_error = stream_->connection_error();
356*6777b538SAndroid Build Coastguard Worker     details.stream_error = stream_->stream_error();
357*6777b538SAndroid Build Coastguard Worker     details.connection_wire_error = stream_->connection_wire_error();
358*6777b538SAndroid Build Coastguard Worker     details.ietf_application_error = stream_->ietf_application_error();
359*6777b538SAndroid Build Coastguard Worker   } else {
360*6777b538SAndroid Build Coastguard Worker     details.connection_error = connection_error_;
361*6777b538SAndroid Build Coastguard Worker     details.stream_error = stream_error_;
362*6777b538SAndroid Build Coastguard Worker     details.connection_wire_error = connection_wire_error_;
363*6777b538SAndroid Build Coastguard Worker     details.ietf_application_error = ietf_application_error_;
364*6777b538SAndroid Build Coastguard Worker   }
365*6777b538SAndroid Build Coastguard Worker   return details;
366*6777b538SAndroid Build Coastguard Worker }
367*6777b538SAndroid Build Coastguard Worker 
ReadTrailingHeaders()368*6777b538SAndroid Build Coastguard Worker void QuicHttpStream::ReadTrailingHeaders() {
369*6777b538SAndroid Build Coastguard Worker   int rv = stream_->ReadTrailingHeaders(
370*6777b538SAndroid Build Coastguard Worker       &trailing_header_block_,
371*6777b538SAndroid Build Coastguard Worker       base::BindOnce(&QuicHttpStream::OnReadTrailingHeadersComplete,
372*6777b538SAndroid Build Coastguard Worker                      weak_factory_.GetWeakPtr()));
373*6777b538SAndroid Build Coastguard Worker 
374*6777b538SAndroid Build Coastguard Worker   if (rv != ERR_IO_PENDING) {
375*6777b538SAndroid Build Coastguard Worker     OnReadTrailingHeadersComplete(rv);
376*6777b538SAndroid Build Coastguard Worker   }
377*6777b538SAndroid Build Coastguard Worker }
378*6777b538SAndroid Build Coastguard Worker 
OnReadTrailingHeadersComplete(int rv)379*6777b538SAndroid Build Coastguard Worker void QuicHttpStream::OnReadTrailingHeadersComplete(int rv) {
380*6777b538SAndroid Build Coastguard Worker   DCHECK(response_headers_received_);
381*6777b538SAndroid Build Coastguard Worker   if (rv > 0) {
382*6777b538SAndroid Build Coastguard Worker     headers_bytes_received_ += rv;
383*6777b538SAndroid Build Coastguard Worker   }
384*6777b538SAndroid Build Coastguard Worker 
385*6777b538SAndroid Build Coastguard Worker   // QuicHttpStream ignores trailers.
386*6777b538SAndroid Build Coastguard Worker   if (stream_->IsDoneReading()) {
387*6777b538SAndroid Build Coastguard Worker     // Close the read side. If the write side has been closed, this will
388*6777b538SAndroid Build Coastguard Worker     // invoke QuicHttpStream::OnClose to reset the stream.
389*6777b538SAndroid Build Coastguard Worker     stream_->OnFinRead();
390*6777b538SAndroid Build Coastguard Worker     SetResponseStatus(OK);
391*6777b538SAndroid Build Coastguard Worker   }
392*6777b538SAndroid Build Coastguard Worker }
393*6777b538SAndroid Build Coastguard Worker 
OnIOComplete(int rv)394*6777b538SAndroid Build Coastguard Worker void QuicHttpStream::OnIOComplete(int rv) {
395*6777b538SAndroid Build Coastguard Worker   rv = DoLoop(rv);
396*6777b538SAndroid Build Coastguard Worker 
397*6777b538SAndroid Build Coastguard Worker   if (rv != ERR_IO_PENDING && !callback_.is_null()) {
398*6777b538SAndroid Build Coastguard Worker     DoCallback(rv);
399*6777b538SAndroid Build Coastguard Worker   }
400*6777b538SAndroid Build Coastguard Worker }
401*6777b538SAndroid Build Coastguard Worker 
DoCallback(int rv)402*6777b538SAndroid Build Coastguard Worker void QuicHttpStream::DoCallback(int rv) {
403*6777b538SAndroid Build Coastguard Worker   CHECK_NE(rv, ERR_IO_PENDING);
404*6777b538SAndroid Build Coastguard Worker   CHECK(!callback_.is_null());
405*6777b538SAndroid Build Coastguard Worker   CHECK(!in_loop_);
406*6777b538SAndroid Build Coastguard Worker 
407*6777b538SAndroid Build Coastguard Worker   // The client callback can do anything, including destroying this class,
408*6777b538SAndroid Build Coastguard Worker   // so any pending callback must be issued after everything else is done.
409*6777b538SAndroid Build Coastguard Worker   std::move(callback_).Run(MapStreamError(rv));
410*6777b538SAndroid Build Coastguard Worker }
411*6777b538SAndroid Build Coastguard Worker 
DoLoop(int rv)412*6777b538SAndroid Build Coastguard Worker int QuicHttpStream::DoLoop(int rv) {
413*6777b538SAndroid Build Coastguard Worker   CHECK(!in_loop_);
414*6777b538SAndroid Build Coastguard Worker   base::AutoReset<bool> auto_reset_in_loop(&in_loop_, true);
415*6777b538SAndroid Build Coastguard Worker   std::unique_ptr<quic::QuicConnection::ScopedPacketFlusher> packet_flusher =
416*6777b538SAndroid Build Coastguard Worker       quic_session()->CreatePacketBundler();
417*6777b538SAndroid Build Coastguard Worker   do {
418*6777b538SAndroid Build Coastguard Worker     State state = next_state_;
419*6777b538SAndroid Build Coastguard Worker     next_state_ = STATE_NONE;
420*6777b538SAndroid Build Coastguard Worker     switch (state) {
421*6777b538SAndroid Build Coastguard Worker       case STATE_REQUEST_STREAM:
422*6777b538SAndroid Build Coastguard Worker         CHECK_EQ(OK, rv);
423*6777b538SAndroid Build Coastguard Worker         rv = DoRequestStream();
424*6777b538SAndroid Build Coastguard Worker         break;
425*6777b538SAndroid Build Coastguard Worker       case STATE_REQUEST_STREAM_COMPLETE:
426*6777b538SAndroid Build Coastguard Worker         rv = DoRequestStreamComplete(rv);
427*6777b538SAndroid Build Coastguard Worker         break;
428*6777b538SAndroid Build Coastguard Worker       case STATE_SET_REQUEST_PRIORITY:
429*6777b538SAndroid Build Coastguard Worker         CHECK_EQ(OK, rv);
430*6777b538SAndroid Build Coastguard Worker         rv = DoSetRequestPriority();
431*6777b538SAndroid Build Coastguard Worker         break;
432*6777b538SAndroid Build Coastguard Worker       case STATE_SEND_HEADERS:
433*6777b538SAndroid Build Coastguard Worker         CHECK_EQ(OK, rv);
434*6777b538SAndroid Build Coastguard Worker         rv = DoSendHeaders();
435*6777b538SAndroid Build Coastguard Worker         break;
436*6777b538SAndroid Build Coastguard Worker       case STATE_SEND_HEADERS_COMPLETE:
437*6777b538SAndroid Build Coastguard Worker         rv = DoSendHeadersComplete(rv);
438*6777b538SAndroid Build Coastguard Worker         break;
439*6777b538SAndroid Build Coastguard Worker       case STATE_READ_REQUEST_BODY:
440*6777b538SAndroid Build Coastguard Worker         CHECK_EQ(OK, rv);
441*6777b538SAndroid Build Coastguard Worker         rv = DoReadRequestBody();
442*6777b538SAndroid Build Coastguard Worker         break;
443*6777b538SAndroid Build Coastguard Worker       case STATE_READ_REQUEST_BODY_COMPLETE:
444*6777b538SAndroid Build Coastguard Worker         rv = DoReadRequestBodyComplete(rv);
445*6777b538SAndroid Build Coastguard Worker         break;
446*6777b538SAndroid Build Coastguard Worker       case STATE_SEND_BODY:
447*6777b538SAndroid Build Coastguard Worker         CHECK_EQ(OK, rv);
448*6777b538SAndroid Build Coastguard Worker         rv = DoSendBody();
449*6777b538SAndroid Build Coastguard Worker         break;
450*6777b538SAndroid Build Coastguard Worker       case STATE_SEND_BODY_COMPLETE:
451*6777b538SAndroid Build Coastguard Worker         rv = DoSendBodyComplete(rv);
452*6777b538SAndroid Build Coastguard Worker         break;
453*6777b538SAndroid Build Coastguard Worker       case STATE_OPEN:
454*6777b538SAndroid Build Coastguard Worker         CHECK_EQ(OK, rv);
455*6777b538SAndroid Build Coastguard Worker         break;
456*6777b538SAndroid Build Coastguard Worker       default:
457*6777b538SAndroid Build Coastguard Worker         NOTREACHED() << "next_state_: " << next_state_;
458*6777b538SAndroid Build Coastguard Worker         break;
459*6777b538SAndroid Build Coastguard Worker     }
460*6777b538SAndroid Build Coastguard Worker   } while (next_state_ != STATE_NONE && next_state_ != STATE_OPEN &&
461*6777b538SAndroid Build Coastguard Worker            rv != ERR_IO_PENDING);
462*6777b538SAndroid Build Coastguard Worker 
463*6777b538SAndroid Build Coastguard Worker   return rv;
464*6777b538SAndroid Build Coastguard Worker }
465*6777b538SAndroid Build Coastguard Worker 
DoRequestStream()466*6777b538SAndroid Build Coastguard Worker int QuicHttpStream::DoRequestStream() {
467*6777b538SAndroid Build Coastguard Worker   next_state_ = STATE_REQUEST_STREAM_COMPLETE;
468*6777b538SAndroid Build Coastguard Worker 
469*6777b538SAndroid Build Coastguard Worker   return quic_session()->RequestStream(
470*6777b538SAndroid Build Coastguard Worker       !can_send_early_,
471*6777b538SAndroid Build Coastguard Worker       base::BindOnce(&QuicHttpStream::OnIOComplete, weak_factory_.GetWeakPtr()),
472*6777b538SAndroid Build Coastguard Worker       NetworkTrafficAnnotationTag(request_info_->traffic_annotation));
473*6777b538SAndroid Build Coastguard Worker }
474*6777b538SAndroid Build Coastguard Worker 
DoRequestStreamComplete(int rv)475*6777b538SAndroid Build Coastguard Worker int QuicHttpStream::DoRequestStreamComplete(int rv) {
476*6777b538SAndroid Build Coastguard Worker   DCHECK(rv == OK || !stream_);
477*6777b538SAndroid Build Coastguard Worker   if (rv != OK) {
478*6777b538SAndroid Build Coastguard Worker     session_error_ = rv;
479*6777b538SAndroid Build Coastguard Worker     return GetResponseStatus();
480*6777b538SAndroid Build Coastguard Worker   }
481*6777b538SAndroid Build Coastguard Worker 
482*6777b538SAndroid Build Coastguard Worker   stream_ = quic_session()->ReleaseStream();
483*6777b538SAndroid Build Coastguard Worker   DCHECK(stream_);
484*6777b538SAndroid Build Coastguard Worker   if (!stream_->IsOpen()) {
485*6777b538SAndroid Build Coastguard Worker     session_error_ = ERR_CONNECTION_CLOSED;
486*6777b538SAndroid Build Coastguard Worker     return GetResponseStatus();
487*6777b538SAndroid Build Coastguard Worker   }
488*6777b538SAndroid Build Coastguard Worker 
489*6777b538SAndroid Build Coastguard Worker   if (request_info_->load_flags &
490*6777b538SAndroid Build Coastguard Worker       LOAD_DISABLE_CONNECTION_MIGRATION_TO_CELLULAR) {
491*6777b538SAndroid Build Coastguard Worker     stream_->DisableConnectionMigrationToCellularNetwork();
492*6777b538SAndroid Build Coastguard Worker   }
493*6777b538SAndroid Build Coastguard Worker 
494*6777b538SAndroid Build Coastguard Worker   DCHECK(response_info_ == nullptr);
495*6777b538SAndroid Build Coastguard Worker 
496*6777b538SAndroid Build Coastguard Worker   return OK;
497*6777b538SAndroid Build Coastguard Worker }
498*6777b538SAndroid Build Coastguard Worker 
DoSetRequestPriority()499*6777b538SAndroid Build Coastguard Worker int QuicHttpStream::DoSetRequestPriority() {
500*6777b538SAndroid Build Coastguard Worker   // Set priority according to request
501*6777b538SAndroid Build Coastguard Worker   DCHECK(stream_);
502*6777b538SAndroid Build Coastguard Worker   DCHECK(response_info_);
503*6777b538SAndroid Build Coastguard Worker   DCHECK(request_info_);
504*6777b538SAndroid Build Coastguard Worker 
505*6777b538SAndroid Build Coastguard Worker   uint8_t urgency = ConvertRequestPriorityToQuicPriority(priority_);
506*6777b538SAndroid Build Coastguard Worker   bool incremental = request_info_->priority_incremental;
507*6777b538SAndroid Build Coastguard Worker   stream_->SetPriority(
508*6777b538SAndroid Build Coastguard Worker       quic::QuicStreamPriority(quic::HttpStreamPriority{urgency, incremental}));
509*6777b538SAndroid Build Coastguard Worker   next_state_ = STATE_SEND_HEADERS;
510*6777b538SAndroid Build Coastguard Worker   return OK;
511*6777b538SAndroid Build Coastguard Worker }
512*6777b538SAndroid Build Coastguard Worker 
DoSendHeaders()513*6777b538SAndroid Build Coastguard Worker int QuicHttpStream::DoSendHeaders() {
514*6777b538SAndroid Build Coastguard Worker   uint8_t urgency = ConvertRequestPriorityToQuicPriority(priority_);
515*6777b538SAndroid Build Coastguard Worker   bool incremental = request_info_->priority_incremental;
516*6777b538SAndroid Build Coastguard Worker   quic::QuicStreamPriority priority(
517*6777b538SAndroid Build Coastguard Worker       quic::HttpStreamPriority{urgency, incremental});
518*6777b538SAndroid Build Coastguard Worker   // Log the actual request with the URL Request's net log.
519*6777b538SAndroid Build Coastguard Worker   stream_net_log_.AddEvent(
520*6777b538SAndroid Build Coastguard Worker       NetLogEventType::HTTP_TRANSACTION_QUIC_SEND_REQUEST_HEADERS,
521*6777b538SAndroid Build Coastguard Worker       [&](NetLogCaptureMode capture_mode) {
522*6777b538SAndroid Build Coastguard Worker         return QuicRequestNetLogParams(stream_->id(), &request_headers_,
523*6777b538SAndroid Build Coastguard Worker                                        priority, capture_mode);
524*6777b538SAndroid Build Coastguard Worker       });
525*6777b538SAndroid Build Coastguard Worker   DispatchRequestHeadersCallback(request_headers_);
526*6777b538SAndroid Build Coastguard Worker   bool has_upload_data = request_body_stream_ != nullptr;
527*6777b538SAndroid Build Coastguard Worker 
528*6777b538SAndroid Build Coastguard Worker   next_state_ = STATE_SEND_HEADERS_COMPLETE;
529*6777b538SAndroid Build Coastguard Worker   int rv = stream_->WriteHeaders(std::move(request_headers_), !has_upload_data,
530*6777b538SAndroid Build Coastguard Worker                                  nullptr);
531*6777b538SAndroid Build Coastguard Worker   if (rv > 0) {
532*6777b538SAndroid Build Coastguard Worker     headers_bytes_sent_ += rv;
533*6777b538SAndroid Build Coastguard Worker   }
534*6777b538SAndroid Build Coastguard Worker 
535*6777b538SAndroid Build Coastguard Worker   request_headers_ = spdy::Http2HeaderBlock();
536*6777b538SAndroid Build Coastguard Worker   return rv;
537*6777b538SAndroid Build Coastguard Worker }
538*6777b538SAndroid Build Coastguard Worker 
DoSendHeadersComplete(int rv)539*6777b538SAndroid Build Coastguard Worker int QuicHttpStream::DoSendHeadersComplete(int rv) {
540*6777b538SAndroid Build Coastguard Worker   if (rv < 0) {
541*6777b538SAndroid Build Coastguard Worker     return rv;
542*6777b538SAndroid Build Coastguard Worker   }
543*6777b538SAndroid Build Coastguard Worker 
544*6777b538SAndroid Build Coastguard Worker   next_state_ = request_body_stream_ ? STATE_READ_REQUEST_BODY : STATE_OPEN;
545*6777b538SAndroid Build Coastguard Worker 
546*6777b538SAndroid Build Coastguard Worker   return OK;
547*6777b538SAndroid Build Coastguard Worker }
548*6777b538SAndroid Build Coastguard Worker 
DoReadRequestBody()549*6777b538SAndroid Build Coastguard Worker int QuicHttpStream::DoReadRequestBody() {
550*6777b538SAndroid Build Coastguard Worker   next_state_ = STATE_READ_REQUEST_BODY_COMPLETE;
551*6777b538SAndroid Build Coastguard Worker   return request_body_stream_->Read(
552*6777b538SAndroid Build Coastguard Worker       raw_request_body_buf_.get(), raw_request_body_buf_->size(),
553*6777b538SAndroid Build Coastguard Worker       base::BindOnce(&QuicHttpStream::OnIOComplete,
554*6777b538SAndroid Build Coastguard Worker                      weak_factory_.GetWeakPtr()));
555*6777b538SAndroid Build Coastguard Worker }
556*6777b538SAndroid Build Coastguard Worker 
DoReadRequestBodyComplete(int rv)557*6777b538SAndroid Build Coastguard Worker int QuicHttpStream::DoReadRequestBodyComplete(int rv) {
558*6777b538SAndroid Build Coastguard Worker   // |rv| is the result of read from the request body from the last call to
559*6777b538SAndroid Build Coastguard Worker   // DoSendBody().
560*6777b538SAndroid Build Coastguard Worker   if (rv < 0) {
561*6777b538SAndroid Build Coastguard Worker     stream_->Reset(quic::QUIC_ERROR_PROCESSING_STREAM);
562*6777b538SAndroid Build Coastguard Worker     ResetStream();
563*6777b538SAndroid Build Coastguard Worker     return rv;
564*6777b538SAndroid Build Coastguard Worker   }
565*6777b538SAndroid Build Coastguard Worker 
566*6777b538SAndroid Build Coastguard Worker   request_body_buf_ =
567*6777b538SAndroid Build Coastguard Worker       base::MakeRefCounted<DrainableIOBuffer>(raw_request_body_buf_, rv);
568*6777b538SAndroid Build Coastguard Worker   if (rv == 0) {  // Reached the end.
569*6777b538SAndroid Build Coastguard Worker     DCHECK(request_body_stream_->IsEOF());
570*6777b538SAndroid Build Coastguard Worker   }
571*6777b538SAndroid Build Coastguard Worker 
572*6777b538SAndroid Build Coastguard Worker   next_state_ = STATE_SEND_BODY;
573*6777b538SAndroid Build Coastguard Worker   return OK;
574*6777b538SAndroid Build Coastguard Worker }
575*6777b538SAndroid Build Coastguard Worker 
DoSendBody()576*6777b538SAndroid Build Coastguard Worker int QuicHttpStream::DoSendBody() {
577*6777b538SAndroid Build Coastguard Worker   CHECK(request_body_stream_);
578*6777b538SAndroid Build Coastguard Worker   CHECK(request_body_buf_.get());
579*6777b538SAndroid Build Coastguard Worker   const bool eof = request_body_stream_->IsEOF();
580*6777b538SAndroid Build Coastguard Worker   int len = request_body_buf_->BytesRemaining();
581*6777b538SAndroid Build Coastguard Worker   if (len > 0 || eof) {
582*6777b538SAndroid Build Coastguard Worker     next_state_ = STATE_SEND_BODY_COMPLETE;
583*6777b538SAndroid Build Coastguard Worker     std::string_view data(request_body_buf_->data(), len);
584*6777b538SAndroid Build Coastguard Worker     return stream_->WriteStreamData(
585*6777b538SAndroid Build Coastguard Worker         data, eof,
586*6777b538SAndroid Build Coastguard Worker         base::BindOnce(&QuicHttpStream::OnIOComplete,
587*6777b538SAndroid Build Coastguard Worker                        weak_factory_.GetWeakPtr()));
588*6777b538SAndroid Build Coastguard Worker   }
589*6777b538SAndroid Build Coastguard Worker 
590*6777b538SAndroid Build Coastguard Worker   next_state_ = STATE_OPEN;
591*6777b538SAndroid Build Coastguard Worker   return OK;
592*6777b538SAndroid Build Coastguard Worker }
593*6777b538SAndroid Build Coastguard Worker 
DoSendBodyComplete(int rv)594*6777b538SAndroid Build Coastguard Worker int QuicHttpStream::DoSendBodyComplete(int rv) {
595*6777b538SAndroid Build Coastguard Worker   if (rv < 0) {
596*6777b538SAndroid Build Coastguard Worker     return rv;
597*6777b538SAndroid Build Coastguard Worker   }
598*6777b538SAndroid Build Coastguard Worker 
599*6777b538SAndroid Build Coastguard Worker   request_body_buf_->DidConsume(request_body_buf_->BytesRemaining());
600*6777b538SAndroid Build Coastguard Worker 
601*6777b538SAndroid Build Coastguard Worker   if (!request_body_stream_->IsEOF()) {
602*6777b538SAndroid Build Coastguard Worker     next_state_ = STATE_READ_REQUEST_BODY;
603*6777b538SAndroid Build Coastguard Worker     return OK;
604*6777b538SAndroid Build Coastguard Worker   }
605*6777b538SAndroid Build Coastguard Worker 
606*6777b538SAndroid Build Coastguard Worker   next_state_ = STATE_OPEN;
607*6777b538SAndroid Build Coastguard Worker   return OK;
608*6777b538SAndroid Build Coastguard Worker }
609*6777b538SAndroid Build Coastguard Worker 
ProcessResponseHeaders(const spdy::Http2HeaderBlock & headers)610*6777b538SAndroid Build Coastguard Worker int QuicHttpStream::ProcessResponseHeaders(
611*6777b538SAndroid Build Coastguard Worker     const spdy::Http2HeaderBlock& headers) {
612*6777b538SAndroid Build Coastguard Worker   const int rv = SpdyHeadersToHttpResponse(headers, response_info_);
613*6777b538SAndroid Build Coastguard Worker   base::UmaHistogramBoolean("Net.QuicHttpStream.ProcessResponseHeaderSuccess",
614*6777b538SAndroid Build Coastguard Worker                             rv == OK);
615*6777b538SAndroid Build Coastguard Worker   if (rv != OK) {
616*6777b538SAndroid Build Coastguard Worker     DLOG(WARNING) << "Invalid headers";
617*6777b538SAndroid Build Coastguard Worker     return ERR_QUIC_PROTOCOL_ERROR;
618*6777b538SAndroid Build Coastguard Worker   }
619*6777b538SAndroid Build Coastguard Worker 
620*6777b538SAndroid Build Coastguard Worker   if (response_info_->headers->response_code() == HTTP_EARLY_HINTS) {
621*6777b538SAndroid Build Coastguard Worker     DCHECK(!response_headers_received_);
622*6777b538SAndroid Build Coastguard Worker     headers_bytes_received_ = 0;
623*6777b538SAndroid Build Coastguard Worker     return OK;
624*6777b538SAndroid Build Coastguard Worker   }
625*6777b538SAndroid Build Coastguard Worker 
626*6777b538SAndroid Build Coastguard Worker   response_info_->connection_info =
627*6777b538SAndroid Build Coastguard Worker       ConnectionInfoFromQuicVersion(quic_session()->GetQuicVersion());
628*6777b538SAndroid Build Coastguard Worker   response_info_->was_alpn_negotiated = true;
629*6777b538SAndroid Build Coastguard Worker   response_info_->alpn_negotiated_protocol =
630*6777b538SAndroid Build Coastguard Worker       HttpConnectionInfoToString(response_info_->connection_info);
631*6777b538SAndroid Build Coastguard Worker   response_info_->response_time = base::Time::Now();
632*6777b538SAndroid Build Coastguard Worker   response_info_->request_time = request_time_;
633*6777b538SAndroid Build Coastguard Worker   response_headers_received_ = true;
634*6777b538SAndroid Build Coastguard Worker 
635*6777b538SAndroid Build Coastguard Worker   // Populate |connect_timing_| when response headers are received. This should
636*6777b538SAndroid Build Coastguard Worker   // take care of 0-RTT where request is sent before handshake is confirmed.
637*6777b538SAndroid Build Coastguard Worker   connect_timing_ = quic_session()->GetConnectTiming();
638*6777b538SAndroid Build Coastguard Worker 
639*6777b538SAndroid Build Coastguard Worker   base::SingleThreadTaskRunner::GetCurrentDefault()->PostTask(
640*6777b538SAndroid Build Coastguard Worker       FROM_HERE, base::BindOnce(&QuicHttpStream::ReadTrailingHeaders,
641*6777b538SAndroid Build Coastguard Worker                                 weak_factory_.GetWeakPtr()));
642*6777b538SAndroid Build Coastguard Worker 
643*6777b538SAndroid Build Coastguard Worker   if (stream_->IsDoneReading()) {
644*6777b538SAndroid Build Coastguard Worker     session_error_ = OK;
645*6777b538SAndroid Build Coastguard Worker     SaveResponseStatus();
646*6777b538SAndroid Build Coastguard Worker     stream_->OnFinRead();
647*6777b538SAndroid Build Coastguard Worker   }
648*6777b538SAndroid Build Coastguard Worker 
649*6777b538SAndroid Build Coastguard Worker   return OK;
650*6777b538SAndroid Build Coastguard Worker }
651*6777b538SAndroid Build Coastguard Worker 
OnReadBodyComplete(int rv)652*6777b538SAndroid Build Coastguard Worker void QuicHttpStream::OnReadBodyComplete(int rv) {
653*6777b538SAndroid Build Coastguard Worker   CHECK(callback_);
654*6777b538SAndroid Build Coastguard Worker   user_buffer_ = nullptr;
655*6777b538SAndroid Build Coastguard Worker   user_buffer_len_ = 0;
656*6777b538SAndroid Build Coastguard Worker   rv = HandleReadComplete(rv);
657*6777b538SAndroid Build Coastguard Worker   DoCallback(rv);
658*6777b538SAndroid Build Coastguard Worker }
659*6777b538SAndroid Build Coastguard Worker 
HandleReadComplete(int rv)660*6777b538SAndroid Build Coastguard Worker int QuicHttpStream::HandleReadComplete(int rv) {
661*6777b538SAndroid Build Coastguard Worker   if (stream_->IsDoneReading()) {
662*6777b538SAndroid Build Coastguard Worker     stream_->OnFinRead();
663*6777b538SAndroid Build Coastguard Worker     SetResponseStatus(OK);
664*6777b538SAndroid Build Coastguard Worker     ResetStream();
665*6777b538SAndroid Build Coastguard Worker   }
666*6777b538SAndroid Build Coastguard Worker   return rv;
667*6777b538SAndroid Build Coastguard Worker }
668*6777b538SAndroid Build Coastguard Worker 
ResetStream()669*6777b538SAndroid Build Coastguard Worker void QuicHttpStream::ResetStream() {
670*6777b538SAndroid Build Coastguard Worker   // If |request_body_stream_| is non-NULL, Reset it, to abort any in progress
671*6777b538SAndroid Build Coastguard Worker   // read.
672*6777b538SAndroid Build Coastguard Worker   if (request_body_stream_) {
673*6777b538SAndroid Build Coastguard Worker     request_body_stream_->Reset();
674*6777b538SAndroid Build Coastguard Worker   }
675*6777b538SAndroid Build Coastguard Worker 
676*6777b538SAndroid Build Coastguard Worker   if (!stream_) {
677*6777b538SAndroid Build Coastguard Worker     return;
678*6777b538SAndroid Build Coastguard Worker   }
679*6777b538SAndroid Build Coastguard Worker 
680*6777b538SAndroid Build Coastguard Worker   DCHECK_LE(stream_->NumBytesConsumed(), stream_->stream_bytes_read());
681*6777b538SAndroid Build Coastguard Worker   // Only count the uniquely received bytes.
682*6777b538SAndroid Build Coastguard Worker   closed_stream_received_bytes_ = stream_->NumBytesConsumed();
683*6777b538SAndroid Build Coastguard Worker   closed_stream_sent_bytes_ = stream_->stream_bytes_written();
684*6777b538SAndroid Build Coastguard Worker   closed_is_first_stream_ = stream_->IsFirstStream();
685*6777b538SAndroid Build Coastguard Worker   connection_error_ = stream_->connection_error();
686*6777b538SAndroid Build Coastguard Worker   stream_error_ = stream_->stream_error();
687*6777b538SAndroid Build Coastguard Worker   connection_wire_error_ = stream_->connection_wire_error();
688*6777b538SAndroid Build Coastguard Worker   ietf_application_error_ = stream_->ietf_application_error();
689*6777b538SAndroid Build Coastguard Worker }
690*6777b538SAndroid Build Coastguard Worker 
MapStreamError(int rv)691*6777b538SAndroid Build Coastguard Worker int QuicHttpStream::MapStreamError(int rv) {
692*6777b538SAndroid Build Coastguard Worker   if (rv == ERR_QUIC_PROTOCOL_ERROR && !quic_session()->OneRttKeysAvailable()) {
693*6777b538SAndroid Build Coastguard Worker     return ERR_QUIC_HANDSHAKE_FAILED;
694*6777b538SAndroid Build Coastguard Worker   }
695*6777b538SAndroid Build Coastguard Worker   return rv;
696*6777b538SAndroid Build Coastguard Worker }
697*6777b538SAndroid Build Coastguard Worker 
GetResponseStatus()698*6777b538SAndroid Build Coastguard Worker int QuicHttpStream::GetResponseStatus() {
699*6777b538SAndroid Build Coastguard Worker   SaveResponseStatus();
700*6777b538SAndroid Build Coastguard Worker   return response_status_;
701*6777b538SAndroid Build Coastguard Worker }
702*6777b538SAndroid Build Coastguard Worker 
SaveResponseStatus()703*6777b538SAndroid Build Coastguard Worker void QuicHttpStream::SaveResponseStatus() {
704*6777b538SAndroid Build Coastguard Worker   if (!has_response_status_) {
705*6777b538SAndroid Build Coastguard Worker     SetResponseStatus(ComputeResponseStatus());
706*6777b538SAndroid Build Coastguard Worker   }
707*6777b538SAndroid Build Coastguard Worker }
708*6777b538SAndroid Build Coastguard Worker 
SetResponseStatus(int response_status)709*6777b538SAndroid Build Coastguard Worker void QuicHttpStream::SetResponseStatus(int response_status) {
710*6777b538SAndroid Build Coastguard Worker   has_response_status_ = true;
711*6777b538SAndroid Build Coastguard Worker   response_status_ = response_status;
712*6777b538SAndroid Build Coastguard Worker }
713*6777b538SAndroid Build Coastguard Worker 
ComputeResponseStatus() const714*6777b538SAndroid Build Coastguard Worker int QuicHttpStream::ComputeResponseStatus() const {
715*6777b538SAndroid Build Coastguard Worker   DCHECK(!has_response_status_);
716*6777b538SAndroid Build Coastguard Worker 
717*6777b538SAndroid Build Coastguard Worker   // If the handshake has failed this will be handled by the QuicSessionPool
718*6777b538SAndroid Build Coastguard Worker   // and HttpStreamFactory to mark QUIC as broken if TCP is actually working.
719*6777b538SAndroid Build Coastguard Worker   if (!quic_session()->OneRttKeysAvailable()) {
720*6777b538SAndroid Build Coastguard Worker     return ERR_QUIC_HANDSHAKE_FAILED;
721*6777b538SAndroid Build Coastguard Worker   }
722*6777b538SAndroid Build Coastguard Worker 
723*6777b538SAndroid Build Coastguard Worker   // If the session was aborted by a higher layer, simply use that error code.
724*6777b538SAndroid Build Coastguard Worker   if (session_error_ != ERR_UNEXPECTED) {
725*6777b538SAndroid Build Coastguard Worker     return session_error_;
726*6777b538SAndroid Build Coastguard Worker   }
727*6777b538SAndroid Build Coastguard Worker 
728*6777b538SAndroid Build Coastguard Worker   // If |response_info_| is null then the request has not been sent, so
729*6777b538SAndroid Build Coastguard Worker   // return ERR_CONNECTION_CLOSED to permit HttpNetworkTransaction to
730*6777b538SAndroid Build Coastguard Worker   // retry the request.
731*6777b538SAndroid Build Coastguard Worker   if (!response_info_) {
732*6777b538SAndroid Build Coastguard Worker     return ERR_CONNECTION_CLOSED;
733*6777b538SAndroid Build Coastguard Worker   }
734*6777b538SAndroid Build Coastguard Worker 
735*6777b538SAndroid Build Coastguard Worker   base::UmaHistogramEnumeration("Net.QuicHttpStream.ResponseStatus",
736*6777b538SAndroid Build Coastguard Worker                                 stream_->stream_error(),
737*6777b538SAndroid Build Coastguard Worker                                 quic::QUIC_STREAM_LAST_ERROR);
738*6777b538SAndroid Build Coastguard Worker 
739*6777b538SAndroid Build Coastguard Worker   return ERR_QUIC_PROTOCOL_ERROR;
740*6777b538SAndroid Build Coastguard Worker }
741*6777b538SAndroid Build Coastguard Worker 
SetRequestIdempotency(Idempotency idempotency)742*6777b538SAndroid Build Coastguard Worker void QuicHttpStream::SetRequestIdempotency(Idempotency idempotency) {
743*6777b538SAndroid Build Coastguard Worker   if (stream_ == nullptr) {
744*6777b538SAndroid Build Coastguard Worker     return;
745*6777b538SAndroid Build Coastguard Worker   }
746*6777b538SAndroid Build Coastguard Worker   stream_->SetRequestIdempotency(idempotency);
747*6777b538SAndroid Build Coastguard Worker }
748*6777b538SAndroid Build Coastguard Worker 
749*6777b538SAndroid Build Coastguard Worker }  // namespace net
750