| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "net/quic/reliable_quic_stream.h" | 5 #include "net/quic/reliable_quic_stream.h" |
| 6 | 6 |
| 7 #include "base/logging.h" | 7 #include "base/logging.h" |
| 8 #include "net/quic/iovector.h" | 8 #include "net/quic/iovector.h" |
| 9 #include "net/quic/quic_flow_controller.h" | 9 #include "net/quic/quic_flow_controller.h" |
| 10 #include "net/quic/quic_session.h" | 10 #include "net/quic/quic_session.h" |
| (...skipping 27 matching lines...) Expand all Loading... |
| 38 class ReliableQuicStream::ProxyAckNotifierDelegate | 38 class ReliableQuicStream::ProxyAckNotifierDelegate |
| 39 : public QuicAckNotifier::DelegateInterface { | 39 : public QuicAckNotifier::DelegateInterface { |
| 40 public: | 40 public: |
| 41 explicit ProxyAckNotifierDelegate(DelegateInterface* delegate) | 41 explicit ProxyAckNotifierDelegate(DelegateInterface* delegate) |
| 42 : delegate_(delegate), | 42 : delegate_(delegate), |
| 43 pending_acks_(0), | 43 pending_acks_(0), |
| 44 wrote_last_data_(false), | 44 wrote_last_data_(false), |
| 45 num_original_packets_(0), | 45 num_original_packets_(0), |
| 46 num_original_bytes_(0), | 46 num_original_bytes_(0), |
| 47 num_retransmitted_packets_(0), | 47 num_retransmitted_packets_(0), |
| 48 num_retransmitted_bytes_(0) { | 48 num_retransmitted_bytes_(0) {} |
| 49 } | |
| 50 | 49 |
| 51 virtual void OnAckNotification(int num_original_packets, | 50 virtual void OnAckNotification( |
| 52 int num_original_bytes, | 51 int num_original_packets, |
| 53 int num_retransmitted_packets, | 52 int num_original_bytes, |
| 54 int num_retransmitted_bytes, | 53 int num_retransmitted_packets, |
| 55 QuicTime::Delta delta_largest_observed) | 54 int num_retransmitted_bytes, |
| 56 OVERRIDE { | 55 QuicTime::Delta delta_largest_observed) OVERRIDE { |
| 57 DCHECK_LT(0, pending_acks_); | 56 DCHECK_LT(0, pending_acks_); |
| 58 --pending_acks_; | 57 --pending_acks_; |
| 59 num_original_packets_ += num_original_packets; | 58 num_original_packets_ += num_original_packets; |
| 60 num_original_bytes_ += num_original_bytes; | 59 num_original_bytes_ += num_original_bytes; |
| 61 num_retransmitted_packets_ += num_retransmitted_packets; | 60 num_retransmitted_packets_ += num_retransmitted_packets; |
| 62 num_retransmitted_bytes_ += num_retransmitted_bytes; | 61 num_retransmitted_bytes_ += num_retransmitted_bytes; |
| 63 | 62 |
| 64 if (wrote_last_data_ && pending_acks_ == 0) { | 63 if (wrote_last_data_ && pending_acks_ == 0) { |
| 65 delegate_->OnAckNotification(num_original_packets_, | 64 delegate_->OnAckNotification(num_original_packets_, |
| 66 num_original_bytes_, | 65 num_original_bytes_, |
| 67 num_retransmitted_packets_, | 66 num_retransmitted_packets_, |
| 68 num_retransmitted_bytes_, | 67 num_retransmitted_bytes_, |
| 69 delta_largest_observed); | 68 delta_largest_observed); |
| 70 } | 69 } |
| 71 } | 70 } |
| 72 | 71 |
| 73 void WroteData(bool last_data) { | 72 void WroteData(bool last_data) { |
| 74 DCHECK(!wrote_last_data_); | 73 DCHECK(!wrote_last_data_); |
| 75 ++pending_acks_; | 74 ++pending_acks_; |
| 76 wrote_last_data_ = last_data; | 75 wrote_last_data_ = last_data; |
| 77 } | 76 } |
| 78 | 77 |
| 79 protected: | 78 protected: |
| 80 // Delegates are ref counted. | 79 // Delegates are ref counted. |
| 81 virtual ~ProxyAckNotifierDelegate() { | 80 virtual ~ProxyAckNotifierDelegate() {} |
| 82 } | |
| 83 | 81 |
| 84 private: | 82 private: |
| 85 // Original delegate. delegate_->OnAckNotification will be called when: | 83 // Original delegate. delegate_->OnAckNotification will be called when: |
| 86 // wrote_last_data_ == true and pending_acks_ == 0 | 84 // wrote_last_data_ == true and pending_acks_ == 0 |
| 87 scoped_refptr<DelegateInterface> delegate_; | 85 scoped_refptr<DelegateInterface> delegate_; |
| 88 | 86 |
| 89 // Number of outstanding acks. | 87 // Number of outstanding acks. |
| 90 int pending_acks_; | 88 int pending_acks_; |
| 91 | 89 |
| 92 // True if no pending writes remain. | 90 // True if no pending writes remain. |
| 93 bool wrote_last_data_; | 91 bool wrote_last_data_; |
| 94 | 92 |
| 95 // Accumulators. | 93 // Accumulators. |
| 96 int num_original_packets_; | 94 int num_original_packets_; |
| 97 int num_original_bytes_; | 95 int num_original_bytes_; |
| 98 int num_retransmitted_packets_; | 96 int num_retransmitted_packets_; |
| 99 int num_retransmitted_bytes_; | 97 int num_retransmitted_bytes_; |
| 100 | 98 |
| 101 DISALLOW_COPY_AND_ASSIGN(ProxyAckNotifierDelegate); | 99 DISALLOW_COPY_AND_ASSIGN(ProxyAckNotifierDelegate); |
| 102 }; | 100 }; |
| 103 | 101 |
| 104 ReliableQuicStream::PendingData::PendingData( | 102 ReliableQuicStream::PendingData::PendingData( |
| 105 string data_in, scoped_refptr<ProxyAckNotifierDelegate> delegate_in) | 103 string data_in, |
| 104 scoped_refptr<ProxyAckNotifierDelegate> delegate_in) |
| 106 : data(data_in), delegate(delegate_in) { | 105 : data(data_in), delegate(delegate_in) { |
| 107 } | 106 } |
| 108 | 107 |
| 109 ReliableQuicStream::PendingData::~PendingData() { | 108 ReliableQuicStream::PendingData::~PendingData() { |
| 110 } | 109 } |
| 111 | 110 |
| 112 ReliableQuicStream::ReliableQuicStream(QuicStreamId id, QuicSession* session) | 111 ReliableQuicStream::ReliableQuicStream(QuicStreamId id, QuicSession* session) |
| 113 : sequencer_(this), | 112 : sequencer_(this), |
| 114 id_(id), | 113 id_(id), |
| 115 session_(session), | 114 session_(session), |
| 116 stream_bytes_read_(0), | 115 stream_bytes_read_(0), |
| 117 stream_bytes_written_(0), | 116 stream_bytes_written_(0), |
| 118 stream_error_(QUIC_STREAM_NO_ERROR), | 117 stream_error_(QUIC_STREAM_NO_ERROR), |
| 119 connection_error_(QUIC_NO_ERROR), | 118 connection_error_(QUIC_NO_ERROR), |
| 120 read_side_closed_(false), | 119 read_side_closed_(false), |
| 121 write_side_closed_(false), | 120 write_side_closed_(false), |
| 122 fin_buffered_(false), | 121 fin_buffered_(false), |
| 123 fin_sent_(false), | 122 fin_sent_(false), |
| 124 rst_sent_(false), | 123 rst_sent_(false), |
| 125 is_server_(session_->is_server()), | 124 is_server_(session_->is_server()), |
| 126 flow_controller_( | 125 flow_controller_( |
| 127 session_->connection()->version(), | 126 session_->connection()->version(), |
| 128 id_, | 127 id_, |
| 129 is_server_, | 128 is_server_, |
| 130 session_->config()->HasReceivedInitialFlowControlWindowBytes() ? | 129 session_->config()->HasReceivedInitialFlowControlWindowBytes() |
| 131 session_->config()->ReceivedInitialFlowControlWindowBytes() : | 130 ? session_->config()->ReceivedInitialFlowControlWindowBytes() |
| 132 kDefaultFlowControlSendWindow, | 131 : kDefaultFlowControlSendWindow, |
| 133 session_->connection()->max_flow_control_receive_window_bytes(), | 132 session_->connection()->max_flow_control_receive_window_bytes(), |
| 134 session_->connection()->max_flow_control_receive_window_bytes()) { | 133 session_->connection()->max_flow_control_receive_window_bytes()) { |
| 135 } | 134 } |
| 136 | 135 |
| 137 ReliableQuicStream::~ReliableQuicStream() { | 136 ReliableQuicStream::~ReliableQuicStream() { |
| 138 } | 137 } |
| 139 | 138 |
| 140 bool ReliableQuicStream::OnStreamFrame(const QuicStreamFrame& frame) { | 139 bool ReliableQuicStream::OnStreamFrame(const QuicStreamFrame& frame) { |
| 141 if (read_side_closed_) { | 140 if (read_side_closed_) { |
| 142 DVLOG(1) << ENDPOINT << "Ignoring frame " << frame.stream_id; | 141 DVLOG(1) << ENDPOINT << "Ignoring frame " << frame.stream_id; |
| (...skipping 244 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 387 | 386 |
| 388 void ReliableQuicStream::OnClose() { | 387 void ReliableQuicStream::OnClose() { |
| 389 CloseReadSide(); | 388 CloseReadSide(); |
| 390 CloseWriteSide(); | 389 CloseWriteSide(); |
| 391 | 390 |
| 392 if (!fin_sent_ && !rst_sent_) { | 391 if (!fin_sent_ && !rst_sent_) { |
| 393 // For flow control accounting, we must tell the peer how many bytes we have | 392 // For flow control accounting, we must tell the peer how many bytes we have |
| 394 // written on this stream before termination. Done here if needed, using a | 393 // written on this stream before termination. Done here if needed, using a |
| 395 // RST frame. | 394 // RST frame. |
| 396 DVLOG(1) << ENDPOINT << "Sending RST in OnClose: " << id(); | 395 DVLOG(1) << ENDPOINT << "Sending RST in OnClose: " << id(); |
| 397 session_->SendRstStream(id(), QUIC_RST_FLOW_CONTROL_ACCOUNTING, | 396 session_->SendRstStream( |
| 398 stream_bytes_written_); | 397 id(), QUIC_RST_FLOW_CONTROL_ACCOUNTING, stream_bytes_written_); |
| 399 rst_sent_ = true; | 398 rst_sent_ = true; |
| 400 } | 399 } |
| 401 } | 400 } |
| 402 | 401 |
| 403 void ReliableQuicStream::OnWindowUpdateFrame( | 402 void ReliableQuicStream::OnWindowUpdateFrame( |
| 404 const QuicWindowUpdateFrame& frame) { | 403 const QuicWindowUpdateFrame& frame) { |
| 405 if (!flow_controller_.IsEnabled()) { | 404 if (!flow_controller_.IsEnabled()) { |
| 406 DLOG(DFATAL) << "Flow control not enabled! " << version(); | 405 DLOG(DFATAL) << "Flow control not enabled! " << version(); |
| 407 return; | 406 return; |
| 408 } | 407 } |
| 409 | 408 |
| 410 if (flow_controller_.UpdateSendWindowOffset(frame.byte_offset)) { | 409 if (flow_controller_.UpdateSendWindowOffset(frame.byte_offset)) { |
| 411 // We can write again! | 410 // We can write again! |
| 412 // TODO(rjshade): This does not respect priorities (e.g. multiple | 411 // TODO(rjshade): This does not respect priorities (e.g. multiple |
| 413 // outstanding POSTs are unblocked on arrival of | 412 // outstanding POSTs are unblocked on arrival of |
| 414 // SHLO with initial window). | 413 // SHLO with initial window). |
| 415 OnCanWrite(); | 414 OnCanWrite(); |
| 416 } | 415 } |
| 417 } | 416 } |
| 418 | 417 |
| 419 } // namespace net | 418 } // namespace net |
| OLD | NEW |