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/quic_session.h" | 8 #include "net/quic/quic_session.h" |
9 #include "net/quic/quic_spdy_decompressor.h" | 9 #include "net/quic/quic_spdy_decompressor.h" |
10 #include "net/quic/quic_write_blocked_list.h" | 10 #include "net/quic/quic_write_blocked_list.h" |
(...skipping 164 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
175 if (write_side_closed_) { | 175 if (write_side_closed_) { |
176 DLOG(ERROR) << ENDPOINT << "Attempt to write when the write side is closed"; | 176 DLOG(ERROR) << ENDPOINT << "Attempt to write when the write side is closed"; |
177 return QuicConsumedData(0, false); | 177 return QuicConsumedData(0, false); |
178 } | 178 } |
179 | 179 |
180 size_t write_length = 0u; | 180 size_t write_length = 0u; |
181 for (int i = 0; i < iov_count; ++i) { | 181 for (int i = 0; i < iov_count; ++i) { |
182 write_length += iov[i].iov_len; | 182 write_length += iov[i].iov_len; |
183 // TODO(rjshade): Maybe block write based on available flow control window. | 183 // TODO(rjshade): Maybe block write based on available flow control window. |
184 } | 184 } |
| 185 |
| 186 // Fill an IOVector with bytes from the iovec. |
| 187 IOVector data; |
| 188 data.AppendIovecAtMostBytes(iov, iov_count, write_length); |
| 189 |
185 QuicConsumedData consumed_data = session()->WritevData( | 190 QuicConsumedData consumed_data = session()->WritevData( |
186 id(), iov, iov_count, stream_bytes_written_, fin, ack_notifier_delegate); | 191 id(), data, stream_bytes_written_, fin, ack_notifier_delegate); |
187 stream_bytes_written_ += consumed_data.bytes_consumed; | 192 stream_bytes_written_ += consumed_data.bytes_consumed; |
188 if (consumed_data.bytes_consumed == write_length) { | 193 if (consumed_data.bytes_consumed == write_length) { |
189 if (fin && consumed_data.fin_consumed) { | 194 if (fin && consumed_data.fin_consumed) { |
190 fin_sent_ = true; | 195 fin_sent_ = true; |
191 CloseWriteSide(); | 196 CloseWriteSide(); |
192 } else if (fin && !consumed_data.fin_consumed) { | 197 } else if (fin && !consumed_data.fin_consumed) { |
193 session_->MarkWriteBlocked(id(), EffectivePriority()); | 198 session_->MarkWriteBlocked(id(), EffectivePriority()); |
194 } | 199 } |
195 } else { | 200 } else { |
196 session_->MarkWriteBlocked(id(), EffectivePriority()); | 201 session_->MarkWriteBlocked(id(), EffectivePriority()); |
(...skipping 40 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
237 // For flow control accounting, we must tell the peer how many bytes we have | 242 // For flow control accounting, we must tell the peer how many bytes we have |
238 // written on this stream before termination. Done here if needed, using a | 243 // written on this stream before termination. Done here if needed, using a |
239 // RST frame. | 244 // RST frame. |
240 DVLOG(1) << ENDPOINT << "Sending RST in OnClose: " << id(); | 245 DVLOG(1) << ENDPOINT << "Sending RST in OnClose: " << id(); |
241 session_->SendRstStream(id(), QUIC_STREAM_NO_ERROR, stream_bytes_written_); | 246 session_->SendRstStream(id(), QUIC_STREAM_NO_ERROR, stream_bytes_written_); |
242 rst_sent_ = true; | 247 rst_sent_ = true; |
243 } | 248 } |
244 } | 249 } |
245 | 250 |
246 } // namespace net | 251 } // namespace net |
OLD | NEW |