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 256 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
267 LOG(DFATAL) << "data.empty() && !fin"; | 267 LOG(DFATAL) << "data.empty() && !fin"; |
268 return; | 268 return; |
269 } | 269 } |
270 | 270 |
271 if (fin_buffered_) { | 271 if (fin_buffered_) { |
272 LOG(DFATAL) << "Fin already buffered"; | 272 LOG(DFATAL) << "Fin already buffered"; |
273 return; | 273 return; |
274 } | 274 } |
275 | 275 |
276 scoped_refptr<ProxyAckNotifierDelegate> proxy_delegate; | 276 scoped_refptr<ProxyAckNotifierDelegate> proxy_delegate; |
277 if (ack_notifier_delegate != NULL) { | 277 if (ack_notifier_delegate != nullptr) { |
278 proxy_delegate = new ProxyAckNotifierDelegate(ack_notifier_delegate); | 278 proxy_delegate = new ProxyAckNotifierDelegate(ack_notifier_delegate); |
279 } | 279 } |
280 | 280 |
281 QuicConsumedData consumed_data(0, false); | 281 QuicConsumedData consumed_data(0, false); |
282 fin_buffered_ = fin; | 282 fin_buffered_ = fin; |
283 | 283 |
284 if (queued_data_.empty()) { | 284 if (queued_data_.empty()) { |
285 struct iovec iov(MakeIovec(data)); | 285 struct iovec iov(MakeIovec(data)); |
286 consumed_data = WritevData(&iov, 1, fin, proxy_delegate.get()); | 286 consumed_data = WritevData(&iov, 1, fin, proxy_delegate.get()); |
287 DCHECK_LE(consumed_data.bytes_consumed, data.length()); | 287 DCHECK_LE(consumed_data.bytes_consumed, data.length()); |
288 } | 288 } |
289 | 289 |
290 bool write_completed; | 290 bool write_completed; |
291 // If there's unconsumed data or an unconsumed fin, queue it. | 291 // If there's unconsumed data or an unconsumed fin, queue it. |
292 if (consumed_data.bytes_consumed < data.length() || | 292 if (consumed_data.bytes_consumed < data.length() || |
293 (fin && !consumed_data.fin_consumed)) { | 293 (fin && !consumed_data.fin_consumed)) { |
294 StringPiece remainder(data.substr(consumed_data.bytes_consumed)); | 294 StringPiece remainder(data.substr(consumed_data.bytes_consumed)); |
295 queued_data_.push_back(PendingData(remainder.as_string(), proxy_delegate)); | 295 queued_data_.push_back(PendingData(remainder.as_string(), proxy_delegate)); |
296 write_completed = false; | 296 write_completed = false; |
297 } else { | 297 } else { |
298 write_completed = true; | 298 write_completed = true; |
299 } | 299 } |
300 | 300 |
301 if ((proxy_delegate.get() != NULL) && | 301 if ((proxy_delegate.get() != nullptr) && |
302 (consumed_data.bytes_consumed > 0 || consumed_data.fin_consumed)) { | 302 (consumed_data.bytes_consumed > 0 || consumed_data.fin_consumed)) { |
303 proxy_delegate->WroteData(write_completed); | 303 proxy_delegate->WroteData(write_completed); |
304 } | 304 } |
305 } | 305 } |
306 | 306 |
307 void ReliableQuicStream::OnCanWrite() { | 307 void ReliableQuicStream::OnCanWrite() { |
308 bool fin = false; | 308 bool fin = false; |
309 while (!queued_data_.empty()) { | 309 while (!queued_data_.empty()) { |
310 PendingData* pending_data = &queued_data_.front(); | 310 PendingData* pending_data = &queued_data_.front(); |
311 ProxyAckNotifierDelegate* delegate = pending_data->delegate.get(); | 311 ProxyAckNotifierDelegate* delegate = pending_data->delegate.get(); |
312 if (queued_data_.size() == 1 && fin_buffered_) { | 312 if (queued_data_.size() == 1 && fin_buffered_) { |
313 fin = true; | 313 fin = true; |
314 } | 314 } |
315 struct iovec iov(MakeIovec(pending_data->data)); | 315 struct iovec iov(MakeIovec(pending_data->data)); |
316 QuicConsumedData consumed_data = WritevData(&iov, 1, fin, delegate); | 316 QuicConsumedData consumed_data = WritevData(&iov, 1, fin, delegate); |
317 if (consumed_data.bytes_consumed == pending_data->data.size() && | 317 if (consumed_data.bytes_consumed == pending_data->data.size() && |
318 fin == consumed_data.fin_consumed) { | 318 fin == consumed_data.fin_consumed) { |
319 queued_data_.pop_front(); | 319 queued_data_.pop_front(); |
320 if (delegate != NULL) { | 320 if (delegate != nullptr) { |
321 delegate->WroteData(true); | 321 delegate->WroteData(true); |
322 } | 322 } |
323 } else { | 323 } else { |
324 if (consumed_data.bytes_consumed > 0) { | 324 if (consumed_data.bytes_consumed > 0) { |
325 pending_data->data.erase(0, consumed_data.bytes_consumed); | 325 pending_data->data.erase(0, consumed_data.bytes_consumed); |
326 if (delegate != NULL) { | 326 if (delegate != nullptr) { |
327 delegate->WroteData(false); | 327 delegate->WroteData(false); |
328 } | 328 } |
329 } | 329 } |
330 break; | 330 break; |
331 } | 331 } |
332 } | 332 } |
333 } | 333 } |
334 | 334 |
335 void ReliableQuicStream::MaybeSendBlocked() { | 335 void ReliableQuicStream::MaybeSendBlocked() { |
336 flow_controller_.MaybeSendBlocked(); | 336 flow_controller_.MaybeSendBlocked(); |
(...skipping 198 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
535 | 535 |
536 bool ReliableQuicStream::IsFlowControlBlocked() { | 536 bool ReliableQuicStream::IsFlowControlBlocked() { |
537 if (flow_controller_.IsBlocked()) { | 537 if (flow_controller_.IsBlocked()) { |
538 return true; | 538 return true; |
539 } | 539 } |
540 return stream_contributes_to_connection_flow_control_ && | 540 return stream_contributes_to_connection_flow_control_ && |
541 connection_flow_controller_->IsBlocked(); | 541 connection_flow_controller_->IsBlocked(); |
542 } | 542 } |
543 | 543 |
544 } // namespace net | 544 } // namespace net |
OLD | NEW |