reliable_quic_stream.cc revision 46d4c2bc3267f3f028f39e7e311b0f89aba2e4fd
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) 28424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles)} // namespace 29424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) 3023730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles)// Wrapper that aggregates OnAckNotifications for packets sent using 3123730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles)// WriteOrBufferData and delivers them to the original 3223730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles)// QuicAckNotifier::DelegateInterface after all bytes written using 3323730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles)// WriteOrBufferData are acked. This level of indirection is 3423730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles)// necessary because the delegate interface provides no mechanism that 3523730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles)// WriteOrBufferData can use to inform it that the write required 3623730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles)// multiple WritevData calls or that only part of the data has been 3723730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles)// sent out by the time ACKs start arriving. 3823730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles)class ReliableQuicStream::ProxyAckNotifierDelegate 3923730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) : public QuicAckNotifier::DelegateInterface { 4023730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) public: 4123730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) explicit ProxyAckNotifierDelegate(DelegateInterface* delegate) 4223730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) : delegate_(delegate), 4323730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) pending_acks_(0), 4423730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) wrote_last_data_(false), 4523730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) num_original_packets_(0), 4623730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) num_original_bytes_(0), 4723730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) num_retransmitted_packets_(0), 4823730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) num_retransmitted_bytes_(0) { 4923730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) } 5023730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) 5123730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) virtual void OnAckNotification(int num_original_packets, 5223730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) int num_original_bytes, 5323730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) int num_retransmitted_packets, 540529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch int num_retransmitted_bytes, 550529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch QuicTime::Delta delta_largest_observed) 560529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch OVERRIDE { 5723730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) DCHECK_LT(0, pending_acks_); 5823730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) --pending_acks_; 5923730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) num_original_packets_ += num_original_packets; 6023730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) num_original_bytes_ += num_original_bytes; 6123730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) num_retransmitted_packets_ += num_retransmitted_packets; 6223730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) num_retransmitted_bytes_ += num_retransmitted_bytes; 6323730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) 6423730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) if (wrote_last_data_ && pending_acks_ == 0) { 6523730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) delegate_->OnAckNotification(num_original_packets_, 6623730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) num_original_bytes_, 6723730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) num_retransmitted_packets_, 680529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch num_retransmitted_bytes_, 690529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch delta_largest_observed); 7023730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) } 7123730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) } 7223730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) 7323730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) void WroteData(bool last_data) { 7423730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) DCHECK(!wrote_last_data_); 7523730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) ++pending_acks_; 7623730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) wrote_last_data_ = last_data; 7723730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) } 7823730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) 7923730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) protected: 8023730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) // Delegates are ref counted. 8123730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) virtual ~ProxyAckNotifierDelegate() { 8223730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) } 8323730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) 8423730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) private: 8523730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) // Original delegate. delegate_->OnAckNotification will be called when: 8623730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) // wrote_last_data_ == true and pending_acks_ == 0 8723730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) scoped_refptr<DelegateInterface> delegate_; 8823730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) 8923730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) // Number of outstanding acks. 9023730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) int pending_acks_; 9123730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) 9223730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) // True if no pending writes remain. 9323730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) bool wrote_last_data_; 9423730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) 9523730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) // Accumulators. 9623730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) int num_original_packets_; 9723730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) int num_original_bytes_; 9823730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) int num_retransmitted_packets_; 9923730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) int num_retransmitted_bytes_; 10023730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) 10123730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) DISALLOW_COPY_AND_ASSIGN(ProxyAckNotifierDelegate); 10223730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles)}; 10323730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) 10423730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles)ReliableQuicStream::PendingData::PendingData( 10523730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) string data_in, scoped_refptr<ProxyAckNotifierDelegate> delegate_in) 10623730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) : data(data_in), delegate(delegate_in) { 10723730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles)} 10823730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) 10923730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles)ReliableQuicStream::PendingData::~PendingData() { 11023730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles)} 11123730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) 112e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen MurdochReliableQuicStream::ReliableQuicStream(QuicStreamId id, QuicSession* session) 1135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) : sequencer_(this), 1145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) id_(id), 1155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) session_(session), 1162a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) stream_bytes_read_(0), 1172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) stream_bytes_written_(0), 118c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) stream_error_(QUIC_STREAM_NO_ERROR), 119c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) connection_error_(QUIC_NO_ERROR), 1205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) read_side_closed_(false), 1212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) write_side_closed_(false), 1222a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) fin_buffered_(false), 1231e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) fin_sent_(false), 1245d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) rst_sent_(false), 1250529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch is_server_(session_->is_server()), 1260529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch flow_controller_( 12746d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) session_->connection(), 1280529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch id_, 1290529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch is_server_, 1300529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch session_->config()->HasReceivedInitialFlowControlWindowBytes() ? 1310529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch session_->config()->ReceivedInitialFlowControlWindowBytes() : 1320529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch kDefaultFlowControlSendWindow, 133cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) session_->max_flow_control_receive_window_bytes(), 134cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) session_->max_flow_control_receive_window_bytes()), 135cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) connection_flow_controller_(session_->flow_controller()) { 1365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)ReliableQuicStream::~ReliableQuicStream() { 1395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1415c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liubool ReliableQuicStream::OnStreamFrame(const QuicStreamFrame& frame) { 1425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (read_side_closed_) { 1435c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu DVLOG(1) << ENDPOINT << "Ignoring frame " << frame.stream_id; 1445c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu // We don't want to be reading: blackhole the data. 1452a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return true; 1465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1475c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu 1485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (frame.stream_id != id_) { 1495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) LOG(ERROR) << "Error!"; 1505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return false; 1515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 152e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch 153e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch // This count include duplicate data received. 15446d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) size_t frame_payload_size = frame.data.TotalBufferSize(); 15546d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) stream_bytes_read_ += frame_payload_size; 15646d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) 15746d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) // Flow control is interested in tracking highest received offset. 15846d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) if (MaybeIncreaseHighestReceivedOffset(frame.offset + frame_payload_size)) { 15946d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) // As the highest received offset has changed, we should check to see if 16046d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) // this is a violation of flow control. 16146d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) if (flow_controller_.FlowControlViolation() || 16246d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) connection_flow_controller_->FlowControlViolation()) { 16346d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) session_->connection()->SendConnectionClose( 16446d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) QUIC_FLOW_CONTROL_RECEIVED_TOO_MUCH_DATA); 16546d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) return false; 16646d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) } 167e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch } 168e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch 16946d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) return sequencer_.OnStreamFrame(frame); 170e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch} 171e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch 172010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles)int ReliableQuicStream::num_frames_received() const { 173c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch return sequencer_.num_frames_received(); 174c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch} 175c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 176010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles)int ReliableQuicStream::num_duplicate_frames_received() const { 177c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch return sequencer_.num_duplicate_frames_received(); 178c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch} 179c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 1805d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)void ReliableQuicStream::OnStreamReset(const QuicRstStreamFrame& frame) { 18146d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) MaybeIncreaseHighestReceivedOffset(frame.byte_offset); 18246d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) 1835d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) stream_error_ = frame.error_code; 184a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) CloseWriteSide(); 185a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) CloseReadSide(); 1865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1881e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles)void ReliableQuicStream::OnConnectionClosed(QuicErrorCode error, 1891e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) bool from_peer) { 1907d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) if (read_side_closed_ && write_side_closed_) { 191c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) return; 192c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) } 193c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) if (error != QUIC_NO_ERROR) { 194c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) stream_error_ = QUIC_STREAM_CONNECTION_ERROR; 195c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) connection_error_ = error; 196c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) } 197c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 198a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) CloseWriteSide(); 199a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) CloseReadSide(); 2005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 202a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)void ReliableQuicStream::OnFinRead() { 203a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) DCHECK(sequencer_.IsClosed()); 2045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CloseReadSide(); 2055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 207a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)void ReliableQuicStream::Reset(QuicRstStreamErrorCode error) { 208a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) DCHECK_NE(QUIC_STREAM_NO_ERROR, error); 209c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) stream_error_ = error; 210a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) // Sending a RstStream results in calling CloseStream. 2115d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) session()->SendRstStream(id(), error, stream_bytes_written_); 2125d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) rst_sent_ = true; 2135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2151e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles)void ReliableQuicStream::CloseConnection(QuicErrorCode error) { 2161e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) session()->connection()->SendConnectionClose(error); 2171e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles)} 2181e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) 2191e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles)void ReliableQuicStream::CloseConnectionWithDetails(QuicErrorCode error, 2201e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) const string& details) { 2211e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) session()->connection()->SendConnectionCloseWithDetails(error, details); 2221e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles)} 2231e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) 224a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)QuicVersion ReliableQuicStream::version() const { 2255d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) return session()->connection()->version(); 226b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles)} 227b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) 22823730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles)void ReliableQuicStream::WriteOrBufferData( 22923730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) StringPiece data, 23023730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) bool fin, 23123730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) QuicAckNotifier::DelegateInterface* ack_notifier_delegate) { 2325d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) if (data.empty() && !fin) { 2335d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) LOG(DFATAL) << "data.empty() && !fin"; 2345d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) return; 235b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) } 236b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) 2375d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) if (fin_buffered_) { 2385d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) LOG(DFATAL) << "Fin already buffered"; 2395d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) return; 240b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) } 2412a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 24223730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) scoped_refptr<ProxyAckNotifierDelegate> proxy_delegate; 24323730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) if (ack_notifier_delegate != NULL) { 24423730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) proxy_delegate = new ProxyAckNotifierDelegate(ack_notifier_delegate); 24523730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) } 24623730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) 2472a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) QuicConsumedData consumed_data(0, false); 2482a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) fin_buffered_ = fin; 2492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2502a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (queued_data_.empty()) { 2515d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) struct iovec iov(MakeIovec(data)); 25223730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) consumed_data = WritevData(&iov, 1, fin, proxy_delegate.get()); 2532a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK_LE(consumed_data.bytes_consumed, data.length()); 2542a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 2552a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 25623730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) bool write_completed; 2572a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // If there's unconsumed data or an unconsumed fin, queue it. 2582a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (consumed_data.bytes_consumed < data.length() || 2592a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) (fin && !consumed_data.fin_consumed)) { 26023730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) StringPiece remainder(data.substr(consumed_data.bytes_consumed)); 26123730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) queued_data_.push_back(PendingData(remainder.as_string(), proxy_delegate)); 26223730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) write_completed = false; 26323730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) } else { 26423730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) write_completed = true; 26523730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) } 26623730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) 26723730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) if ((proxy_delegate.get() != NULL) && 26823730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) (consumed_data.bytes_consumed > 0 || consumed_data.fin_consumed)) { 26923730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) proxy_delegate->WroteData(write_completed); 2702a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 2712a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 2722a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2732a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void ReliableQuicStream::OnCanWrite() { 2742a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) bool fin = false; 2752a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) while (!queued_data_.empty()) { 27623730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) PendingData* pending_data = &queued_data_.front(); 27723730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) ProxyAckNotifierDelegate* delegate = pending_data->delegate.get(); 2782a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (queued_data_.size() == 1 && fin_buffered_) { 2792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) fin = true; 2802a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 28123730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) struct iovec iov(MakeIovec(pending_data->data)); 28223730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) QuicConsumedData consumed_data = WritevData(&iov, 1, fin, delegate); 28323730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) if (consumed_data.bytes_consumed == pending_data->data.size() && 2842a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) fin == consumed_data.fin_consumed) { 2852a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) queued_data_.pop_front(); 28623730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) if (delegate != NULL) { 28723730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) delegate->WroteData(true); 28823730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) } 2892a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } else { 29023730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) if (consumed_data.bytes_consumed > 0) { 29123730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) pending_data->data.erase(0, consumed_data.bytes_consumed); 29223730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) if (delegate != NULL) { 29323730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) delegate->WroteData(false); 29423730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) } 29523730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) } 2962a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) break; 2972a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 2982a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 2992a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 3002a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 301010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles)void ReliableQuicStream::MaybeSendBlocked() { 30246d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) flow_controller_.MaybeSendBlocked(); 30346d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) connection_flow_controller_->MaybeSendBlocked(); 304010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) // If we are connection level flow control blocked, then add the stream 305010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) // to the write blocked list. It will be given a chance to write when a 306010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) // connection level WINDOW_UPDATE arrives. 307010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) if (connection_flow_controller_->IsBlocked() && 308010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) !flow_controller_.IsBlocked()) { 309010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) session_->MarkWriteBlocked(id(), EffectivePriority()); 310010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) } 311010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles)} 312010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) 3135d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)QuicConsumedData ReliableQuicStream::WritevData( 314f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) const struct iovec* iov, 315f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) int iov_count, 316f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) bool fin, 317f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) QuicAckNotifier::DelegateInterface* ack_notifier_delegate) { 3185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (write_side_closed_) { 3191e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) DLOG(ERROR) << ENDPOINT << "Attempt to write when the write side is closed"; 3202a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return QuicConsumedData(0, false); 3215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 323e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch // How much data we want to write. 324e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch size_t write_length = TotalIovecLength(iov, iov_count); 325e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch 326e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch // A FIN with zero data payload should not be flow control blocked. 327e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch bool fin_with_zero_data = (fin && write_length == 0); 328e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch 3295c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu if (flow_controller_.IsEnabled()) { 3305c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu // How much data we are allowed to write from flow control. 331010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) uint64 send_window = flow_controller_.SendWindowSize(); 332010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) if (connection_flow_controller_->IsEnabled()) { 333010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) send_window = 334010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) min(send_window, connection_flow_controller_->SendWindowSize()); 335010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) } 3365c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu 337e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch if (send_window == 0 && !fin_with_zero_data) { 338e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch // Quick return if we can't send anything. 339010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) MaybeSendBlocked(); 340e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch return QuicConsumedData(0, false); 341e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch } 342e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch 343e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch if (write_length > send_window) { 344e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch // Don't send the FIN if we aren't going to send all the data. 345e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch fin = false; 346e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch 347e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch // Writing more data would be a violation of flow control. 348e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch write_length = send_window; 349e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch } 350d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) } 351a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) 352a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) // Fill an IOVector with bytes from the iovec. 353a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) IOVector data; 354a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) data.AppendIovecAtMostBytes(iov, iov_count, write_length); 355a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) 356f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) QuicConsumedData consumed_data = session()->WritevData( 357a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) id(), data, stream_bytes_written_, fin, ack_notifier_delegate); 3582a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) stream_bytes_written_ += consumed_data.bytes_consumed; 359e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch 360010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) AddBytesSent(consumed_data.bytes_consumed); 3610529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch 362d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) if (consumed_data.bytes_consumed == write_length) { 3630529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch if (!fin_with_zero_data) { 364010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) MaybeSendBlocked(); 365e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch } 3662a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (fin && consumed_data.fin_consumed) { 3672a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) fin_sent_ = true; 3682a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) CloseWriteSide(); 369868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) } else if (fin && !consumed_data.fin_consumed) { 370d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) session_->MarkWriteBlocked(id(), EffectivePriority()); 3712a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 3722a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } else { 373d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) session_->MarkWriteBlocked(id(), EffectivePriority()); 3745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3752a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return consumed_data; 3765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void ReliableQuicStream::CloseReadSide() { 3792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (read_side_closed_) { 3802a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return; 3812a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 382f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) DVLOG(1) << ENDPOINT << "Done reading from stream " << id(); 3835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) read_side_closed_ = true; 3855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (write_side_closed_) { 386f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) DVLOG(1) << ENDPOINT << "Closing stream: " << id(); 3875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) session_->CloseStream(id()); 3885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void ReliableQuicStream::CloseWriteSide() { 3922a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (write_side_closed_) { 3932a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return; 3942a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 395f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) DVLOG(1) << ENDPOINT << "Done writing to stream " << id(); 3965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) write_side_closed_ = true; 3985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (read_side_closed_) { 399f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) DVLOG(1) << ENDPOINT << "Closing stream: " << id(); 4005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) session_->CloseStream(id()); 4015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 4025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 4035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 404010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles)bool ReliableQuicStream::HasBufferedData() const { 40558537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) return !queued_data_.empty(); 40658537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles)} 40758537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) 4082a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void ReliableQuicStream::OnClose() { 409c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) CloseReadSide(); 410c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) CloseWriteSide(); 411c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 4125c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu if (!fin_sent_ && !rst_sent_) { 4135d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) // For flow control accounting, we must tell the peer how many bytes we have 4145d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) // written on this stream before termination. Done here if needed, using a 4155d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) // RST frame. 4165d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) DVLOG(1) << ENDPOINT << "Sending RST in OnClose: " << id(); 4170529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch session_->SendRstStream(id(), QUIC_RST_FLOW_CONTROL_ACCOUNTING, 4180529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch stream_bytes_written_); 4195d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) rst_sent_ = true; 420424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) } 42146d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) 42246d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) // We are closing the stream and will not process any further incoming bytes. 42346d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) // As there may be more bytes in flight and we need to ensure that both 42446d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) // endpoints have the same connection level flow control state, mark all 42546d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) // unreceived or buffered bytes as consumed. 42646d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) uint64 bytes_to_consume = flow_controller_.highest_received_byte_offset() - 42746d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) flow_controller_.bytes_consumed(); 42846d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) AddBytesConsumed(bytes_to_consume); 429424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles)} 430424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) 431e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdochvoid ReliableQuicStream::OnWindowUpdateFrame( 432e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch const QuicWindowUpdateFrame& frame) { 4330529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch if (!flow_controller_.IsEnabled()) { 434e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch DLOG(DFATAL) << "Flow control not enabled! " << version(); 435e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch return; 436e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch } 437e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch 4380529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch if (flow_controller_.UpdateSendWindowOffset(frame.byte_offset)) { 4390529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch // We can write again! 4400529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch // TODO(rjshade): This does not respect priorities (e.g. multiple 4410529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch // outstanding POSTs are unblocked on arrival of 4420529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch // SHLO with initial window). 443010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) // As long as the connection is not flow control blocked, we can write! 4440529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch OnCanWrite(); 445e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch } 446e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch} 447e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch 44846d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles)bool ReliableQuicStream::MaybeIncreaseHighestReceivedOffset(uint64 new_offset) { 449010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) if (flow_controller_.IsEnabled()) { 45046d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) uint64 increment = 45146d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) new_offset - flow_controller_.highest_received_byte_offset(); 45246d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) if (flow_controller_.UpdateHighestReceivedOffset(new_offset)) { 45346d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) // If |new_offset| increased the stream flow controller's highest received 45446d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) // offset, then we need to increase the connection flow controller's value 45546d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) // by the incremental difference. 45646d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) connection_flow_controller_->UpdateHighestReceivedOffset( 45746d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) connection_flow_controller_->highest_received_byte_offset() + 45846d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) increment); 45946d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) return true; 46046d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) } 461010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) } 46246d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) return false; 463010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles)} 464010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) 465010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles)void ReliableQuicStream::AddBytesSent(uint64 bytes) { 466010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) if (flow_controller_.IsEnabled()) { 467010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) flow_controller_.AddBytesSent(bytes); 468010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) connection_flow_controller_->AddBytesSent(bytes); 469010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) } 470010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles)} 471010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) 472010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles)void ReliableQuicStream::AddBytesConsumed(uint64 bytes) { 473010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) if (flow_controller_.IsEnabled()) { 47446d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) // Only adjust stream level flow controller if we are still reading. 47546d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) if (!read_side_closed_) { 47646d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) flow_controller_.AddBytesConsumed(bytes); 47746d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) } 47846d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) 479010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) connection_flow_controller_->AddBytesConsumed(bytes); 480010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) } 481010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles)} 482010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) 483010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles)bool ReliableQuicStream::IsFlowControlBlocked() { 484010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) return flow_controller_.IsBlocked() || 485010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) connection_flow_controller_->IsBlocked(); 486010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles)} 487010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) 4885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace net 489