reliable_quic_stream.cc revision 6d86b77056ed63eb6871182f42a9fd5f07550f90
15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2012 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/quic/reliable_quic_stream.h" 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 75d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)#include "base/logging.h" 8e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch#include "net/quic/iovector.h" 90529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch#include "net/quic/quic_flow_controller.h" 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/quic/quic_session.h" 115d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)#include "net/quic/quic_write_blocked_list.h" 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using base::StringPiece; 14b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles)using std::min; 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace net { 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 185d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)#define ENDPOINT (is_server_ ? "Server: " : " Client: ") 195d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 20424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles)namespace { 21424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) 225d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)struct iovec MakeIovec(StringPiece data) { 235d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) struct iovec iov = {const_cast<char*>(data.data()), 245d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) static_cast<size_t>(data.size())}; 255d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) return iov; 26424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles)} 27424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) 286d86b77056ed63eb6871182f42a9fd5f07550f90Torne (Richard Coles)size_t GetInitialStreamFlowControlWindowToSend(QuicSession* session) { 296d86b77056ed63eb6871182f42a9fd5f07550f90Torne (Richard Coles) QuicVersion version = session->connection()->version(); 306d86b77056ed63eb6871182f42a9fd5f07550f90Torne (Richard Coles) if (version <= QUIC_VERSION_19) { 316d86b77056ed63eb6871182f42a9fd5f07550f90Torne (Richard Coles) return session->config()->GetInitialFlowControlWindowToSend(); 326d86b77056ed63eb6871182f42a9fd5f07550f90Torne (Richard Coles) } 336d86b77056ed63eb6871182f42a9fd5f07550f90Torne (Richard Coles) 346d86b77056ed63eb6871182f42a9fd5f07550f90Torne (Richard Coles) return session->config()->GetInitialStreamFlowControlWindowToSend(); 356d86b77056ed63eb6871182f42a9fd5f07550f90Torne (Richard Coles)} 366d86b77056ed63eb6871182f42a9fd5f07550f90Torne (Richard Coles) 376d86b77056ed63eb6871182f42a9fd5f07550f90Torne (Richard Coles)size_t GetReceivedFlowControlWindow(QuicSession* session) { 386d86b77056ed63eb6871182f42a9fd5f07550f90Torne (Richard Coles) QuicVersion version = session->connection()->version(); 396d86b77056ed63eb6871182f42a9fd5f07550f90Torne (Richard Coles) if (version <= QUIC_VERSION_19) { 406d86b77056ed63eb6871182f42a9fd5f07550f90Torne (Richard Coles) if (session->config()->HasReceivedInitialFlowControlWindowBytes()) { 416d86b77056ed63eb6871182f42a9fd5f07550f90Torne (Richard Coles) return session->config()->ReceivedInitialFlowControlWindowBytes(); 426d86b77056ed63eb6871182f42a9fd5f07550f90Torne (Richard Coles) } 436d86b77056ed63eb6871182f42a9fd5f07550f90Torne (Richard Coles) 446d86b77056ed63eb6871182f42a9fd5f07550f90Torne (Richard Coles) return kDefaultFlowControlSendWindow; 456d86b77056ed63eb6871182f42a9fd5f07550f90Torne (Richard Coles) } 466d86b77056ed63eb6871182f42a9fd5f07550f90Torne (Richard Coles) 476d86b77056ed63eb6871182f42a9fd5f07550f90Torne (Richard Coles) // Version must be >= QUIC_VERSION_20, so we check for stream specific flow 486d86b77056ed63eb6871182f42a9fd5f07550f90Torne (Richard Coles) // control window. 496d86b77056ed63eb6871182f42a9fd5f07550f90Torne (Richard Coles) if (session->config()->HasReceivedInitialStreamFlowControlWindowBytes()) { 506d86b77056ed63eb6871182f42a9fd5f07550f90Torne (Richard Coles) return session->config()->ReceivedInitialStreamFlowControlWindowBytes(); 516d86b77056ed63eb6871182f42a9fd5f07550f90Torne (Richard Coles) } 526d86b77056ed63eb6871182f42a9fd5f07550f90Torne (Richard Coles) 536d86b77056ed63eb6871182f42a9fd5f07550f90Torne (Richard Coles) return kDefaultFlowControlSendWindow; 546d86b77056ed63eb6871182f42a9fd5f07550f90Torne (Richard Coles)} 556d86b77056ed63eb6871182f42a9fd5f07550f90Torne (Richard Coles) 56424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles)} // namespace 57424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) 5823730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles)// Wrapper that aggregates OnAckNotifications for packets sent using 5923730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles)// WriteOrBufferData and delivers them to the original 6023730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles)// QuicAckNotifier::DelegateInterface after all bytes written using 6123730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles)// WriteOrBufferData are acked. This level of indirection is 6223730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles)// necessary because the delegate interface provides no mechanism that 6323730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles)// WriteOrBufferData can use to inform it that the write required 6423730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles)// multiple WritevData calls or that only part of the data has been 6523730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles)// sent out by the time ACKs start arriving. 6623730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles)class ReliableQuicStream::ProxyAckNotifierDelegate 6723730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) : public QuicAckNotifier::DelegateInterface { 6823730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) public: 6923730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) explicit ProxyAckNotifierDelegate(DelegateInterface* delegate) 7023730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) : delegate_(delegate), 7123730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) pending_acks_(0), 7223730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) wrote_last_data_(false), 7323730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) num_original_packets_(0), 7423730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) num_original_bytes_(0), 7523730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) num_retransmitted_packets_(0), 7623730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) num_retransmitted_bytes_(0) { 7723730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) } 7823730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) 7923730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) virtual void OnAckNotification(int num_original_packets, 8023730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) int num_original_bytes, 8123730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) int num_retransmitted_packets, 820529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch int num_retransmitted_bytes, 830529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch QuicTime::Delta delta_largest_observed) 840529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch OVERRIDE { 8523730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) DCHECK_LT(0, pending_acks_); 8623730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) --pending_acks_; 8723730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) num_original_packets_ += num_original_packets; 8823730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) num_original_bytes_ += num_original_bytes; 8923730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) num_retransmitted_packets_ += num_retransmitted_packets; 9023730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) num_retransmitted_bytes_ += num_retransmitted_bytes; 9123730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) 9223730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) if (wrote_last_data_ && pending_acks_ == 0) { 9323730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) delegate_->OnAckNotification(num_original_packets_, 9423730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) num_original_bytes_, 9523730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) num_retransmitted_packets_, 960529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch num_retransmitted_bytes_, 970529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch delta_largest_observed); 9823730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) } 9923730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) } 10023730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) 10123730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) void WroteData(bool last_data) { 10223730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) DCHECK(!wrote_last_data_); 10323730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) ++pending_acks_; 10423730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) wrote_last_data_ = last_data; 10523730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) } 10623730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) 10723730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) protected: 10823730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) // Delegates are ref counted. 109f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) virtual ~ProxyAckNotifierDelegate() OVERRIDE { 11023730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) } 11123730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) 11223730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) private: 11323730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) // Original delegate. delegate_->OnAckNotification will be called when: 11423730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) // wrote_last_data_ == true and pending_acks_ == 0 11523730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) scoped_refptr<DelegateInterface> delegate_; 11623730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) 11723730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) // Number of outstanding acks. 11823730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) int pending_acks_; 11923730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) 12023730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) // True if no pending writes remain. 12123730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) bool wrote_last_data_; 12223730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) 12323730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) // Accumulators. 12423730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) int num_original_packets_; 12523730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) int num_original_bytes_; 12623730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) int num_retransmitted_packets_; 12723730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) int num_retransmitted_bytes_; 12823730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) 12923730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) DISALLOW_COPY_AND_ASSIGN(ProxyAckNotifierDelegate); 13023730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles)}; 13123730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) 13223730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles)ReliableQuicStream::PendingData::PendingData( 13323730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) string data_in, scoped_refptr<ProxyAckNotifierDelegate> delegate_in) 13423730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) : data(data_in), delegate(delegate_in) { 13523730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles)} 13623730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) 13723730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles)ReliableQuicStream::PendingData::~PendingData() { 13823730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles)} 13923730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) 140e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen MurdochReliableQuicStream::ReliableQuicStream(QuicStreamId id, QuicSession* session) 1415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) : sequencer_(this), 1425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) id_(id), 1435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) session_(session), 1442a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) stream_bytes_read_(0), 1452a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) stream_bytes_written_(0), 146c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) stream_error_(QUIC_STREAM_NO_ERROR), 147c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) connection_error_(QUIC_NO_ERROR), 1485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) read_side_closed_(false), 1492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) write_side_closed_(false), 1502a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) fin_buffered_(false), 1511e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) fin_sent_(false), 152f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) fin_received_(false), 1535d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) rst_sent_(false), 154f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) rst_received_(false), 155f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) fec_policy_(FEC_PROTECT_OPTIONAL), 1560529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch is_server_(session_->is_server()), 1570529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch flow_controller_( 1586d86b77056ed63eb6871182f42a9fd5f07550f90Torne (Richard Coles) session_->connection(), id_, is_server_, 1596d86b77056ed63eb6871182f42a9fd5f07550f90Torne (Richard Coles) GetReceivedFlowControlWindow(session), 1606d86b77056ed63eb6871182f42a9fd5f07550f90Torne (Richard Coles) GetInitialStreamFlowControlWindowToSend(session), 1616d86b77056ed63eb6871182f42a9fd5f07550f90Torne (Richard Coles) GetInitialStreamFlowControlWindowToSend(session)), 162cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) connection_flow_controller_(session_->flow_controller()) { 1635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)ReliableQuicStream::~ReliableQuicStream() { 1665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1685c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liubool ReliableQuicStream::OnStreamFrame(const QuicStreamFrame& frame) { 1695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (read_side_closed_) { 1705c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu DVLOG(1) << ENDPOINT << "Ignoring frame " << frame.stream_id; 1715c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu // We don't want to be reading: blackhole the data. 1722a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return true; 1735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1745c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu 1755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (frame.stream_id != id_) { 1765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) LOG(ERROR) << "Error!"; 1775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return false; 1785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 179e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch 180f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) if (frame.fin) { 181f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) fin_received_ = true; 182f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) } 183f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 184e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch // This count include duplicate data received. 18546d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) size_t frame_payload_size = frame.data.TotalBufferSize(); 18646d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) stream_bytes_read_ += frame_payload_size; 18746d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) 18846d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) // Flow control is interested in tracking highest received offset. 18946d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) if (MaybeIncreaseHighestReceivedOffset(frame.offset + frame_payload_size)) { 19046d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) // As the highest received offset has changed, we should check to see if 19146d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) // this is a violation of flow control. 19246d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) if (flow_controller_.FlowControlViolation() || 19346d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) connection_flow_controller_->FlowControlViolation()) { 19446d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) session_->connection()->SendConnectionClose( 19546d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) QUIC_FLOW_CONTROL_RECEIVED_TOO_MUCH_DATA); 19646d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) return false; 19746d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) } 198e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch } 199e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch 20046d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) return sequencer_.OnStreamFrame(frame); 201e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch} 202e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch 203010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles)int ReliableQuicStream::num_frames_received() const { 204c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch return sequencer_.num_frames_received(); 205c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch} 206c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 207010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles)int ReliableQuicStream::num_duplicate_frames_received() const { 208c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch return sequencer_.num_duplicate_frames_received(); 209c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch} 210c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 2115d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)void ReliableQuicStream::OnStreamReset(const QuicRstStreamFrame& frame) { 212f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) rst_received_ = true; 21346d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) MaybeIncreaseHighestReceivedOffset(frame.byte_offset); 21446d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) 2155d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) stream_error_ = frame.error_code; 216a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) CloseWriteSide(); 217a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) CloseReadSide(); 2185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2201e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles)void ReliableQuicStream::OnConnectionClosed(QuicErrorCode error, 2211e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) bool from_peer) { 2227d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) if (read_side_closed_ && write_side_closed_) { 223c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) return; 224c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) } 225c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) if (error != QUIC_NO_ERROR) { 226c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) stream_error_ = QUIC_STREAM_CONNECTION_ERROR; 227c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) connection_error_ = error; 228c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) } 229c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 230a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) CloseWriteSide(); 231a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) CloseReadSide(); 2325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 234a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)void ReliableQuicStream::OnFinRead() { 235a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) DCHECK(sequencer_.IsClosed()); 2365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CloseReadSide(); 2375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 239a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)void ReliableQuicStream::Reset(QuicRstStreamErrorCode error) { 240a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) DCHECK_NE(QUIC_STREAM_NO_ERROR, error); 241c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) stream_error_ = error; 242a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) // Sending a RstStream results in calling CloseStream. 2435d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) session()->SendRstStream(id(), error, stream_bytes_written_); 2445d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) rst_sent_ = true; 2455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2471e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles)void ReliableQuicStream::CloseConnection(QuicErrorCode error) { 2481e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) session()->connection()->SendConnectionClose(error); 2491e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles)} 2501e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) 2511e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles)void ReliableQuicStream::CloseConnectionWithDetails(QuicErrorCode error, 2521e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) const string& details) { 2531e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) session()->connection()->SendConnectionCloseWithDetails(error, details); 2541e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles)} 2551e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) 256a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)QuicVersion ReliableQuicStream::version() const { 2575d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) return session()->connection()->version(); 258b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles)} 259b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) 26023730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles)void ReliableQuicStream::WriteOrBufferData( 26123730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) StringPiece data, 26223730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) bool fin, 26323730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) QuicAckNotifier::DelegateInterface* ack_notifier_delegate) { 2645d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) if (data.empty() && !fin) { 2655d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) LOG(DFATAL) << "data.empty() && !fin"; 2665d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) return; 267b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) } 268b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) 2695d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) if (fin_buffered_) { 2705d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) LOG(DFATAL) << "Fin already buffered"; 2715d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) return; 272b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) } 2732a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 27423730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) scoped_refptr<ProxyAckNotifierDelegate> proxy_delegate; 27523730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) if (ack_notifier_delegate != NULL) { 27623730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) proxy_delegate = new ProxyAckNotifierDelegate(ack_notifier_delegate); 27723730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) } 27823730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) 2792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) QuicConsumedData consumed_data(0, false); 2802a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) fin_buffered_ = fin; 2812a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2822a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (queued_data_.empty()) { 2835d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) struct iovec iov(MakeIovec(data)); 28423730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) consumed_data = WritevData(&iov, 1, fin, proxy_delegate.get()); 2852a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK_LE(consumed_data.bytes_consumed, data.length()); 2862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 2872a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 28823730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) bool write_completed; 2892a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // If there's unconsumed data or an unconsumed fin, queue it. 2902a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (consumed_data.bytes_consumed < data.length() || 2912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) (fin && !consumed_data.fin_consumed)) { 29223730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) StringPiece remainder(data.substr(consumed_data.bytes_consumed)); 29323730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) queued_data_.push_back(PendingData(remainder.as_string(), proxy_delegate)); 29423730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) write_completed = false; 29523730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) } else { 29623730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) write_completed = true; 29723730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) } 29823730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) 29923730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) if ((proxy_delegate.get() != NULL) && 30023730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) (consumed_data.bytes_consumed > 0 || consumed_data.fin_consumed)) { 30123730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) proxy_delegate->WroteData(write_completed); 3022a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 3032a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 3042a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3052a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void ReliableQuicStream::OnCanWrite() { 3062a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) bool fin = false; 3072a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) while (!queued_data_.empty()) { 30823730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) PendingData* pending_data = &queued_data_.front(); 30923730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) ProxyAckNotifierDelegate* delegate = pending_data->delegate.get(); 3102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (queued_data_.size() == 1 && fin_buffered_) { 3112a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) fin = true; 3122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 31323730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) struct iovec iov(MakeIovec(pending_data->data)); 31423730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) QuicConsumedData consumed_data = WritevData(&iov, 1, fin, delegate); 31523730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) if (consumed_data.bytes_consumed == pending_data->data.size() && 3162a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) fin == consumed_data.fin_consumed) { 3172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) queued_data_.pop_front(); 31823730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) if (delegate != NULL) { 31923730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) delegate->WroteData(true); 32023730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) } 3212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } else { 32223730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) if (consumed_data.bytes_consumed > 0) { 32323730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) pending_data->data.erase(0, consumed_data.bytes_consumed); 32423730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) if (delegate != NULL) { 32523730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) delegate->WroteData(false); 32623730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) } 32723730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) } 3282a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) break; 3292a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 3302a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 3312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 3322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 333010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles)void ReliableQuicStream::MaybeSendBlocked() { 33446d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) flow_controller_.MaybeSendBlocked(); 33546d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) connection_flow_controller_->MaybeSendBlocked(); 336010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) // If we are connection level flow control blocked, then add the stream 337010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) // to the write blocked list. It will be given a chance to write when a 338010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) // connection level WINDOW_UPDATE arrives. 339010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) if (connection_flow_controller_->IsBlocked() && 340010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) !flow_controller_.IsBlocked()) { 341010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) session_->MarkWriteBlocked(id(), EffectivePriority()); 342010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) } 343010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles)} 344010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) 3455d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)QuicConsumedData ReliableQuicStream::WritevData( 346f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) const struct iovec* iov, 347f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) int iov_count, 348f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) bool fin, 349f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) QuicAckNotifier::DelegateInterface* ack_notifier_delegate) { 3505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (write_side_closed_) { 3511e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) DLOG(ERROR) << ENDPOINT << "Attempt to write when the write side is closed"; 3522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return QuicConsumedData(0, false); 3535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 355e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch // How much data we want to write. 356e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch size_t write_length = TotalIovecLength(iov, iov_count); 357e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch 358e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch // A FIN with zero data payload should not be flow control blocked. 359e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch bool fin_with_zero_data = (fin && write_length == 0); 360e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch 3615c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu if (flow_controller_.IsEnabled()) { 3625c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu // How much data we are allowed to write from flow control. 363010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) uint64 send_window = flow_controller_.SendWindowSize(); 364010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) if (connection_flow_controller_->IsEnabled()) { 365010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) send_window = 366010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) min(send_window, connection_flow_controller_->SendWindowSize()); 367010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) } 3685c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu 369e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch if (send_window == 0 && !fin_with_zero_data) { 370e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch // Quick return if we can't send anything. 371010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) MaybeSendBlocked(); 372e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch return QuicConsumedData(0, false); 373e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch } 374e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch 375e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch if (write_length > send_window) { 376e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch // Don't send the FIN if we aren't going to send all the data. 377e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch fin = false; 378e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch 379e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch // Writing more data would be a violation of flow control. 380e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch write_length = send_window; 381e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch } 382d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) } 383a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) 384a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) // Fill an IOVector with bytes from the iovec. 385a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) IOVector data; 386a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) data.AppendIovecAtMostBytes(iov, iov_count, write_length); 387a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) 388f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) QuicConsumedData consumed_data = session()->WritevData( 389f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) id(), data, stream_bytes_written_, fin, GetFecProtection(), 390f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) ack_notifier_delegate); 3912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) stream_bytes_written_ += consumed_data.bytes_consumed; 392e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch 393010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) AddBytesSent(consumed_data.bytes_consumed); 3940529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch 395d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) if (consumed_data.bytes_consumed == write_length) { 3960529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch if (!fin_with_zero_data) { 397010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) MaybeSendBlocked(); 398e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch } 3992a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (fin && consumed_data.fin_consumed) { 4002a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) fin_sent_ = true; 4012a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) CloseWriteSide(); 402868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) } else if (fin && !consumed_data.fin_consumed) { 403d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) session_->MarkWriteBlocked(id(), EffectivePriority()); 4042a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 4052a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } else { 406d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) session_->MarkWriteBlocked(id(), EffectivePriority()); 4075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 4082a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return consumed_data; 4095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 4105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 411f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)FecProtection ReliableQuicStream::GetFecProtection() { 412f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) return fec_policy_ == FEC_PROTECT_ALWAYS ? MUST_FEC_PROTECT : MAY_FEC_PROTECT; 413f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)} 414f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 4155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void ReliableQuicStream::CloseReadSide() { 4162a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (read_side_closed_) { 4172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return; 4182a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 419f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) DVLOG(1) << ENDPOINT << "Done reading from stream " << id(); 4205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) read_side_closed_ = true; 4225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (write_side_closed_) { 423f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) DVLOG(1) << ENDPOINT << "Closing stream: " << id(); 4245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) session_->CloseStream(id()); 4255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 4265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 4275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void ReliableQuicStream::CloseWriteSide() { 4292a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (write_side_closed_) { 4302a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return; 4312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 432f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) DVLOG(1) << ENDPOINT << "Done writing to stream " << id(); 4335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) write_side_closed_ = true; 4355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (read_side_closed_) { 436f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) DVLOG(1) << ENDPOINT << "Closing stream: " << id(); 4375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) session_->CloseStream(id()); 4385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 4395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 4405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 441010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles)bool ReliableQuicStream::HasBufferedData() const { 44258537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) return !queued_data_.empty(); 44358537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles)} 44458537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) 4452a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void ReliableQuicStream::OnClose() { 446c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) CloseReadSide(); 447c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) CloseWriteSide(); 448c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 4495c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu if (!fin_sent_ && !rst_sent_) { 4505d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) // For flow control accounting, we must tell the peer how many bytes we have 4515d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) // written on this stream before termination. Done here if needed, using a 4525d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) // RST frame. 4535d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) DVLOG(1) << ENDPOINT << "Sending RST in OnClose: " << id(); 4540529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch session_->SendRstStream(id(), QUIC_RST_FLOW_CONTROL_ACCOUNTING, 4550529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch stream_bytes_written_); 4565d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) rst_sent_ = true; 457424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) } 45846d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) 45946d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) // We are closing the stream and will not process any further incoming bytes. 46046d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) // As there may be more bytes in flight and we need to ensure that both 46146d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) // endpoints have the same connection level flow control state, mark all 46246d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) // unreceived or buffered bytes as consumed. 46346d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) uint64 bytes_to_consume = flow_controller_.highest_received_byte_offset() - 46446d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) flow_controller_.bytes_consumed(); 46546d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) AddBytesConsumed(bytes_to_consume); 466424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles)} 467424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) 468e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdochvoid ReliableQuicStream::OnWindowUpdateFrame( 469e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch const QuicWindowUpdateFrame& frame) { 4700529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch if (!flow_controller_.IsEnabled()) { 471e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch DLOG(DFATAL) << "Flow control not enabled! " << version(); 472e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch return; 473e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch } 474e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch 4750529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch if (flow_controller_.UpdateSendWindowOffset(frame.byte_offset)) { 4760529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch // We can write again! 4770529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch // TODO(rjshade): This does not respect priorities (e.g. multiple 4780529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch // outstanding POSTs are unblocked on arrival of 4790529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch // SHLO with initial window). 480010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) // As long as the connection is not flow control blocked, we can write! 4810529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch OnCanWrite(); 482e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch } 483e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch} 484e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch 48546d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles)bool ReliableQuicStream::MaybeIncreaseHighestReceivedOffset(uint64 new_offset) { 486010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) if (flow_controller_.IsEnabled()) { 48746d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) uint64 increment = 48846d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) new_offset - flow_controller_.highest_received_byte_offset(); 48946d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) if (flow_controller_.UpdateHighestReceivedOffset(new_offset)) { 49046d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) // If |new_offset| increased the stream flow controller's highest received 49146d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) // offset, then we need to increase the connection flow controller's value 49246d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) // by the incremental difference. 49346d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) connection_flow_controller_->UpdateHighestReceivedOffset( 49446d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) connection_flow_controller_->highest_received_byte_offset() + 49546d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) increment); 49646d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) return true; 49746d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) } 498010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) } 49946d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) return false; 500010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles)} 501010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) 502010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles)void ReliableQuicStream::AddBytesSent(uint64 bytes) { 503010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) if (flow_controller_.IsEnabled()) { 504010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) flow_controller_.AddBytesSent(bytes); 505010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) connection_flow_controller_->AddBytesSent(bytes); 506010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) } 507010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles)} 508010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) 509010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles)void ReliableQuicStream::AddBytesConsumed(uint64 bytes) { 510010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) if (flow_controller_.IsEnabled()) { 51146d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) // Only adjust stream level flow controller if we are still reading. 51246d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) if (!read_side_closed_) { 51346d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) flow_controller_.AddBytesConsumed(bytes); 51446d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) } 51546d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) 516010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) connection_flow_controller_->AddBytesConsumed(bytes); 517010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) } 518010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles)} 519010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) 520010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles)bool ReliableQuicStream::IsFlowControlBlocked() { 521010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) return flow_controller_.IsBlocked() || 522010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) connection_flow_controller_->IsBlocked(); 523010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles)} 524010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) 5255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace net 526