blob: 686ec69bfdb1a51dc6257872ec24c1c1de7cc619 [file] [log] [blame]
// Copyright (c) 2012 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.
#ifndef NET_THIRD_PARTY_QUIC_CORE_QUIC_STREAM_SEQUENCER_H_
#define NET_THIRD_PARTY_QUIC_CORE_QUIC_STREAM_SEQUENCER_H_
#include <cstddef>
#include <map>
#include "base/macros.h"
#include "net/third_party/quic/core/quic_packets.h"
#include "net/third_party/quic/core/quic_stream_sequencer_buffer.h"
#include "net/third_party/quic/platform/api/quic_export.h"
#include "net/third_party/quic/platform/api/quic_string.h"
namespace quic {
namespace test {
class QuicStreamSequencerPeer;
} // namespace test
// Buffers frames until we have something which can be passed
// up to the next layer.
class QUIC_EXPORT_PRIVATE QuicStreamSequencer {
public:
// Interface that thie Sequencer uses to communicate with the Stream.
class StreamInterface {
public:
virtual ~StreamInterface() = default;
// Called when new data is available to be read from the sequencer.
virtual void OnDataAvailable() = 0;
// Called when the end of the stream has been read.
virtual void OnFinRead() = 0;
// Called when bytes have been consumed from the sequencer.
virtual void AddBytesConsumed(QuicByteCount bytes) = 0;
// TODO(rch): Clean up this interface via OnUnrecoverableError and
// remove PeerAddressOfLatestPacket().
// Called when an error has occurred which should result in the stream
// being reset.
virtual void Reset(QuicRstStreamErrorCode error) = 0;
// Called when an error has occurred which should result in the connection
// being closed.
virtual void CloseConnectionWithDetails(QuicErrorCode error,
const QuicString& details) = 0;
// Returns the stream id of this stream.
virtual QuicStreamId id() const = 0;
// Returns the peer address of the last packet received for this stream.
virtual const QuicSocketAddress& PeerAddressOfLatestPacket() const = 0;
};
explicit QuicStreamSequencer(StreamInterface* quic_stream);
QuicStreamSequencer(const QuicStreamSequencer&) = delete;
QuicStreamSequencer(QuicStreamSequencer&&) = default;
QuicStreamSequencer& operator=(const QuicStreamSequencer&) = delete;
virtual ~QuicStreamSequencer();
// If the frame is the next one we need in order to process in-order data,
// ProcessData will be immediately called on the stream until all buffered
// data is processed or the stream fails to consume data. Any unconsumed
// data will be buffered. If the frame is not the next in line, it will be
// buffered.
void OnStreamFrame(const QuicStreamFrame& frame);
// If the frame is the next one we need in order to process in-order data,
// ProcessData will be immediately called on the crypto stream until all
// buffered data is processed or the crypto stream fails to consume data. Any
// unconsumed data will be buffered. If the frame is not the next in line, it
// will be buffered.
void OnCryptoFrame(const QuicCryptoFrame& frame);
// Once data is buffered, it's up to the stream to read it when the stream
// can handle more data. The following three functions make that possible.
// Fills in up to iov_len IOVECs with the next readable regions. Returns the
// number of iovs used. Non-destructive of the underlying data.
int GetReadableRegions(IOVEC* iov, size_t iov_len) const;
// Fills in one IOVEC with the next readable region. Returns false if there
// is no readable region available.
bool GetReadableRegion(IOVEC* iov) const;
// Fill in one IOVEC with the next unread region for the quic spdy stream.
// Returns false if no readable region is available.
bool PrefetchNextRegion(IOVEC* iov);
// Copies the data into the iov_len buffers provided. Returns the number of
// bytes read. Any buffered data no longer in use will be released.
// TODO(rch): remove this method and instead implement it as a helper method
// based on GetReadableRegions and MarkConsumed.
int Readv(const struct IOVEC* iov, size_t iov_len);
// Consumes |num_bytes| data. Used in conjunction with |GetReadableRegions|
// to do zero-copy reads.
void MarkConsumed(size_t num_bytes);
// Appends all of the readable data to |buffer| and marks all of the appended
// data as consumed.
void Read(QuicString* buffer);
// Returns true if the sequncer has bytes available for reading.
bool HasBytesToRead() const;
// Number of bytes available to read.
size_t ReadableBytes() const;
// Returns true if the sequencer has delivered the fin.
bool IsClosed() const;
// Calls |OnDataAvailable| on |stream_| if there is buffered data that can
// be processed, and causes |OnDataAvailable| to be called as new data
// arrives.
void SetUnblocked();
// Blocks processing of frames until |SetUnblocked| is called.
void SetBlockedUntilFlush();
// Sets the sequencer to discard all incoming data itself and not call
// |stream_->OnDataAvailable()|. |stream_->OnFinRead()| will be called
// automatically when the FIN is consumed (which may be immediately).
void StopReading();
// Free the memory of underlying buffer.
void ReleaseBuffer();
// Free the memory of underlying buffer when no bytes remain in it.
void ReleaseBufferIfEmpty();
// Number of bytes in the buffer right now.
size_t NumBytesBuffered() const;
// Number of bytes has been consumed.
QuicStreamOffset NumBytesConsumed() const;
QuicStreamOffset close_offset() const { return close_offset_; }
int num_frames_received() const { return num_frames_received_; }
int num_duplicate_frames_received() const {
return num_duplicate_frames_received_;
}
bool ignore_read_data() const { return ignore_read_data_; }
void set_level_triggered(bool level_triggered) {
level_triggered_ = level_triggered;
}
bool level_triggered() const { return level_triggered_; }
void set_stream(StreamInterface* stream) { stream_ = stream; }
// Returns string describing internal state.
const QuicString DebugString() const;
private:
friend class test::QuicStreamSequencerPeer;
// Deletes and records as consumed any buffered data that is now in-sequence.
// (To be called only after StopReading has been called.)
void FlushBufferedFrames();
// Wait until we've seen 'offset' bytes, and then terminate the stream.
void CloseStreamAtOffset(QuicStreamOffset offset);
// If we've received a FIN and have processed all remaining data, then inform
// the stream of FIN, and clear buffers.
bool MaybeCloseStream();
// Shared implementation between OnStreamFrame and OnCryptoFrame.
void OnFrameData(QuicStreamOffset byte_offset,
size_t data_len,
const char* data_buffer);
// The stream which owns this sequencer.
StreamInterface* stream_;
// Stores received data in offset order.
QuicStreamSequencerBuffer buffered_frames_;
// The offset, if any, we got a stream termination for. When this many bytes
// have been processed, the sequencer will be closed.
QuicStreamOffset close_offset_;
// If true, the sequencer is blocked from passing data to the stream and will
// buffer all new incoming data until FlushBufferedFrames is called.
bool blocked_;
// Count of the number of frames received.
int num_frames_received_;
// Count of the number of duplicate frames received.
int num_duplicate_frames_received_;
// If true, all incoming data will be discarded.
bool ignore_read_data_;
// If false, only call OnDataAvailable() when it becomes newly unblocked.
// Otherwise, call OnDataAvailable() when number of readable bytes changes.
bool level_triggered_;
// Latched value of quic_stop_reading_when_level_triggered flag. When true,
// the sequencer will discard incoming data (but not FIN bits) after
// StopReading is called, even in level_triggered_ mode.
const bool stop_reading_when_level_triggered_;
};
} // namespace quic
#endif // NET_THIRD_PARTY_QUIC_CORE_QUIC_STREAM_SEQUENCER_H_