diff options
author | Allan Sandfeld Jensen <allan.jensen@qt.io> | 2019-05-24 11:40:17 +0200 |
---|---|---|
committer | Allan Sandfeld Jensen <allan.jensen@qt.io> | 2019-05-24 12:42:11 +0000 |
commit | 5d87695f37678f96492b258bbab36486c59866b4 (patch) | |
tree | be9783bbaf04fb930c4d74ca9c00b5e7954c8bc6 /chromium/net/third_party/quiche/src/quic/core/quic_stream_send_buffer.cc | |
parent | 6c11fb357ec39bf087b8b632e2b1e375aef1b38b (diff) | |
download | qtwebengine-chromium-5d87695f37678f96492b258bbab36486c59866b4.tar.gz |
BASELINE: Update Chromium to 75.0.3770.56
Change-Id: I86d2007fd27a45d5797eee06f4c9369b8b50ac4f
Reviewed-by: Alexandru Croitor <alexandru.croitor@qt.io>
Diffstat (limited to 'chromium/net/third_party/quiche/src/quic/core/quic_stream_send_buffer.cc')
-rw-r--r-- | chromium/net/third_party/quiche/src/quic/core/quic_stream_send_buffer.cc | 308 |
1 files changed, 308 insertions, 0 deletions
diff --git a/chromium/net/third_party/quiche/src/quic/core/quic_stream_send_buffer.cc b/chromium/net/third_party/quiche/src/quic/core/quic_stream_send_buffer.cc new file mode 100644 index 00000000000..ead192dd6fe --- /dev/null +++ b/chromium/net/third_party/quiche/src/quic/core/quic_stream_send_buffer.cc @@ -0,0 +1,308 @@ +// Copyright (c) 2017 The Chromium Authors. All rights reserved. +// Use of this source code is governed by a BSD-style license that can be +// found in the LICENSE file. + +#include <algorithm> + +#include "net/third_party/quiche/src/quic/core/quic_data_writer.h" +#include "net/third_party/quiche/src/quic/core/quic_interval.h" +#include "net/third_party/quiche/src/quic/core/quic_stream_send_buffer.h" +#include "net/third_party/quiche/src/quic/core/quic_utils.h" +#include "net/third_party/quiche/src/quic/platform/api/quic_bug_tracker.h" +#include "net/third_party/quiche/src/quic/platform/api/quic_flag_utils.h" +#include "net/third_party/quiche/src/quic/platform/api/quic_flags.h" +#include "net/third_party/quiche/src/quic/platform/api/quic_logging.h" + +namespace quic { + +namespace { + +struct CompareOffset { + bool operator()(const BufferedSlice& slice, QuicStreamOffset offset) const { + return slice.offset + slice.slice.length() < offset; + } +}; + +} // namespace + +BufferedSlice::BufferedSlice(QuicMemSlice mem_slice, QuicStreamOffset offset) + : slice(std::move(mem_slice)), offset(offset) {} + +BufferedSlice::BufferedSlice(BufferedSlice&& other) = default; + +BufferedSlice& BufferedSlice::operator=(BufferedSlice&& other) = default; + +BufferedSlice::~BufferedSlice() {} + +bool StreamPendingRetransmission::operator==( + const StreamPendingRetransmission& other) const { + return offset == other.offset && length == other.length; +} + +QuicStreamSendBuffer::QuicStreamSendBuffer(QuicBufferAllocator* allocator) + : stream_offset_(0), + allocator_(allocator), + stream_bytes_written_(0), + stream_bytes_outstanding_(0), + write_index_(-1) {} + +QuicStreamSendBuffer::~QuicStreamSendBuffer() {} + +void QuicStreamSendBuffer::SaveStreamData(const struct iovec* iov, + int iov_count, + size_t iov_offset, + QuicByteCount data_length) { + DCHECK_LT(0u, data_length); + // Latch the maximum data slice size. + const QuicByteCount max_data_slice_size = + GetQuicFlag(FLAGS_quic_send_buffer_max_data_slice_size); + while (data_length > 0) { + size_t slice_len = std::min(data_length, max_data_slice_size); + QuicMemSlice slice(allocator_, slice_len); + QuicUtils::CopyToBuffer(iov, iov_count, iov_offset, slice_len, + const_cast<char*>(slice.data())); + SaveMemSlice(std::move(slice)); + data_length -= slice_len; + iov_offset += slice_len; + } +} + +void QuicStreamSendBuffer::SaveMemSlice(QuicMemSlice slice) { + QUIC_DVLOG(2) << "Save slice offset " << stream_offset_ << " length " + << slice.length(); + if (slice.empty()) { + QUIC_BUG << "Try to save empty MemSlice to send buffer."; + return; + } + size_t length = slice.length(); + buffered_slices_.emplace_back(std::move(slice), stream_offset_); + if (write_index_ == -1) { + write_index_ = buffered_slices_.size() - 1; + } + stream_offset_ += length; +} + +QuicByteCount QuicStreamSendBuffer::SaveMemSliceSpan(QuicMemSliceSpan span) { + return span.ConsumeAll( + [&](QuicMemSlice slice) { SaveMemSlice(std::move(slice)); }); +} + +void QuicStreamSendBuffer::OnStreamDataConsumed(size_t bytes_consumed) { + stream_bytes_written_ += bytes_consumed; + stream_bytes_outstanding_ += bytes_consumed; +} + +bool QuicStreamSendBuffer::WriteStreamData(QuicStreamOffset offset, + QuicByteCount data_length, + QuicDataWriter* writer) { + bool write_index_hit = false; + QuicDeque<BufferedSlice>::iterator slice_it = + write_index_ == -1 + ? buffered_slices_.begin() + // Assume with write_index, write mostly starts from indexed slice. + : buffered_slices_.begin() + write_index_; + if (write_index_ != -1) { + if (offset >= slice_it->offset + slice_it->slice.length()) { + QUIC_BUG << "Tried to write data out of sequence."; + return false; + } + // Determine if write actually happens at indexed slice. + if (offset >= slice_it->offset) { + write_index_hit = true; + } else { + // Write index missed, move iterator to the beginning. + slice_it = buffered_slices_.begin(); + } + } + + for (; slice_it != buffered_slices_.end(); ++slice_it) { + if (data_length == 0 || offset < slice_it->offset) { + break; + } + if (offset >= slice_it->offset + slice_it->slice.length()) { + continue; + } + QuicByteCount slice_offset = offset - slice_it->offset; + QuicByteCount available_bytes_in_slice = + slice_it->slice.length() - slice_offset; + QuicByteCount copy_length = std::min(data_length, available_bytes_in_slice); + if (!writer->WriteBytes(slice_it->slice.data() + slice_offset, + copy_length)) { + QUIC_BUG << "Writer fails to write."; + return false; + } + offset += copy_length; + data_length -= copy_length; + + if (write_index_hit && copy_length == available_bytes_in_slice) { + // Finished writing all data in current slice, advance write index for + // next write. + ++write_index_; + } + } + + if (write_index_hit && + static_cast<size_t>(write_index_) == buffered_slices_.size()) { + // Already write to the end off buffer. + QUIC_DVLOG(2) << "Finish writing out all buffered data."; + write_index_ = -1; + } + + return data_length == 0; +} + +bool QuicStreamSendBuffer::OnStreamDataAcked( + QuicStreamOffset offset, + QuicByteCount data_length, + QuicByteCount* newly_acked_length) { + *newly_acked_length = 0; + if (data_length == 0) { + return true; + } + if (bytes_acked_.Empty() || offset >= bytes_acked_.rbegin()->max() || + bytes_acked_.IsDisjoint( + QuicInterval<QuicStreamOffset>(offset, offset + data_length))) { + // Optimization for the typical case, when all data is newly acked. + if (stream_bytes_outstanding_ < data_length) { + return false; + } + bytes_acked_.Add(offset, offset + data_length); + *newly_acked_length = data_length; + stream_bytes_outstanding_ -= data_length; + pending_retransmissions_.Difference(offset, offset + data_length); + if (!FreeMemSlices(offset, offset + data_length)) { + return false; + } + CleanUpBufferedSlices(); + return true; + } + // Exit if no new data gets acked. + if (bytes_acked_.Contains(offset, offset + data_length)) { + return true; + } + // Execute the slow path if newly acked data fill in existing holes. + QuicIntervalSet<QuicStreamOffset> newly_acked(offset, offset + data_length); + newly_acked.Difference(bytes_acked_); + for (const auto& interval : newly_acked) { + *newly_acked_length += (interval.max() - interval.min()); + } + if (stream_bytes_outstanding_ < *newly_acked_length) { + return false; + } + stream_bytes_outstanding_ -= *newly_acked_length; + bytes_acked_.Add(offset, offset + data_length); + pending_retransmissions_.Difference(offset, offset + data_length); + if (newly_acked.Empty()) { + return true; + } + if (!FreeMemSlices(newly_acked.begin()->min(), newly_acked.rbegin()->max())) { + return false; + } + CleanUpBufferedSlices(); + return true; +} + +void QuicStreamSendBuffer::OnStreamDataLost(QuicStreamOffset offset, + QuicByteCount data_length) { + if (data_length == 0) { + return; + } + QuicIntervalSet<QuicStreamOffset> bytes_lost(offset, offset + data_length); + bytes_lost.Difference(bytes_acked_); + if (bytes_lost.Empty()) { + return; + } + for (const auto& lost : bytes_lost) { + pending_retransmissions_.Add(lost.min(), lost.max()); + } +} + +void QuicStreamSendBuffer::OnStreamDataRetransmitted( + QuicStreamOffset offset, + QuicByteCount data_length) { + if (data_length == 0) { + return; + } + pending_retransmissions_.Difference(offset, offset + data_length); +} + +bool QuicStreamSendBuffer::HasPendingRetransmission() const { + return !pending_retransmissions_.Empty(); +} + +StreamPendingRetransmission QuicStreamSendBuffer::NextPendingRetransmission() + const { + if (HasPendingRetransmission()) { + const auto pending = pending_retransmissions_.begin(); + return {pending->min(), pending->max() - pending->min()}; + } + QUIC_BUG << "NextPendingRetransmission is called unexpected with no " + "pending retransmissions."; + return {0, 0}; +} + +bool QuicStreamSendBuffer::FreeMemSlices(QuicStreamOffset start, + QuicStreamOffset end) { + auto it = buffered_slices_.begin(); + // Find it, such that buffered_slices_[it - 1].end < start <= + // buffered_slices_[it].end. + if (it == buffered_slices_.end() || it->slice.empty()) { + QUIC_BUG << "Trying to ack stream data [" << start << ", " << end << "), " + << (it == buffered_slices_.end() + ? "and there is no outstanding data." + : "and the first slice is empty."); + return false; + } + if (start >= it->offset + it->slice.length() || start < it->offset) { + // Slow path that not the earliest outstanding data gets acked. + it = std::lower_bound(buffered_slices_.begin(), buffered_slices_.end(), + start, CompareOffset()); + } + if (it == buffered_slices_.end() || it->slice.empty()) { + QUIC_BUG << "Offset " << start + << " does not exist or it has already been acked."; + return false; + } + for (; it != buffered_slices_.end(); ++it) { + if (it->offset >= end) { + break; + } + if (!it->slice.empty() && + bytes_acked_.Contains(it->offset, it->offset + it->slice.length())) { + it->slice.Reset(); + } + } + return true; +} + +void QuicStreamSendBuffer::CleanUpBufferedSlices() { + while (!buffered_slices_.empty() && buffered_slices_.front().slice.empty()) { + // Remove data which stops waiting for acks. Please note, mem slices can + // be released out of order, but send buffer is cleaned up in order. + QUIC_BUG_IF(write_index_ == 0) + << "Fail to advance current_write_slice_. It points to the slice " + "whose data has all be written and ACK'ed or ignored. " + "current_write_slice_ offset " + << buffered_slices_[write_index_].offset << " length " + << buffered_slices_[write_index_].slice.length(); + if (write_index_ > 0) { + // If write index is pointing to any slice, reduce the index as the + // slices are all shifted to the left by one. + --write_index_; + } + buffered_slices_.pop_front(); + } +} + +bool QuicStreamSendBuffer::IsStreamDataOutstanding( + QuicStreamOffset offset, + QuicByteCount data_length) const { + return data_length > 0 && + !bytes_acked_.Contains(offset, offset + data_length); +} + +size_t QuicStreamSendBuffer::size() const { + return buffered_slices_.size(); +} + +} // namespace quic |