| Index: net/quic/quic_session.cc
|
| diff --git a/net/quic/quic_session.cc b/net/quic/quic_session.cc
|
| deleted file mode 100644
|
| index f102845b07bc236adb316fac7f1552ab3ba90986..0000000000000000000000000000000000000000
|
| --- a/net/quic/quic_session.cc
|
| +++ /dev/null
|
| @@ -1,752 +0,0 @@
|
| -// 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.
|
| -
|
| -#include "net/quic/quic_session.h"
|
| -
|
| -#include "base/stl_util.h"
|
| -#include "net/quic/crypto/proof_verifier.h"
|
| -#include "net/quic/quic_connection.h"
|
| -#include "net/quic/quic_flow_controller.h"
|
| -#include "net/quic/quic_headers_stream.h"
|
| -#include "net/ssl/ssl_info.h"
|
| -
|
| -using base::StringPiece;
|
| -using base::hash_map;
|
| -using base::hash_set;
|
| -using std::make_pair;
|
| -using std::map;
|
| -using std::max;
|
| -using std::string;
|
| -using std::vector;
|
| -
|
| -namespace net {
|
| -
|
| -#define ENDPOINT (is_server() ? "Server: " : " Client: ")
|
| -
|
| -// We want to make sure we delete any closed streams in a safe manner.
|
| -// To avoid deleting a stream in mid-operation, we have a simple shim between
|
| -// us and the stream, so we can delete any streams when we return from
|
| -// processing.
|
| -//
|
| -// We could just override the base methods, but this makes it easier to make
|
| -// sure we don't miss any.
|
| -class VisitorShim : public QuicConnectionVisitorInterface {
|
| - public:
|
| - explicit VisitorShim(QuicSession* session) : session_(session) {}
|
| -
|
| - void OnStreamFrames(const vector<QuicStreamFrame>& frames) override {
|
| - session_->OnStreamFrames(frames);
|
| - session_->PostProcessAfterData();
|
| - }
|
| - void OnRstStream(const QuicRstStreamFrame& frame) override {
|
| - session_->OnRstStream(frame);
|
| - session_->PostProcessAfterData();
|
| - }
|
| -
|
| - void OnGoAway(const QuicGoAwayFrame& frame) override {
|
| - session_->OnGoAway(frame);
|
| - session_->PostProcessAfterData();
|
| - }
|
| -
|
| - void OnWindowUpdateFrames(
|
| - const vector<QuicWindowUpdateFrame>& frames) override {
|
| - session_->OnWindowUpdateFrames(frames);
|
| - session_->PostProcessAfterData();
|
| - }
|
| -
|
| - void OnBlockedFrames(const vector<QuicBlockedFrame>& frames) override {
|
| - session_->OnBlockedFrames(frames);
|
| - session_->PostProcessAfterData();
|
| - }
|
| -
|
| - void OnCanWrite() override {
|
| - session_->OnCanWrite();
|
| - session_->PostProcessAfterData();
|
| - }
|
| -
|
| - void OnCongestionWindowChange(QuicTime now) override {
|
| - session_->OnCongestionWindowChange(now);
|
| - }
|
| -
|
| - void OnSuccessfulVersionNegotiation(const QuicVersion& version) override {
|
| - session_->OnSuccessfulVersionNegotiation(version);
|
| - }
|
| -
|
| - void OnConnectionClosed(QuicErrorCode error, bool from_peer) override {
|
| - session_->OnConnectionClosed(error, from_peer);
|
| - // The session will go away, so don't bother with cleanup.
|
| - }
|
| -
|
| - void OnWriteBlocked() override { session_->OnWriteBlocked(); }
|
| -
|
| - bool WillingAndAbleToWrite() const override {
|
| - return session_->WillingAndAbleToWrite();
|
| - }
|
| -
|
| - bool HasPendingHandshake() const override {
|
| - return session_->HasPendingHandshake();
|
| - }
|
| -
|
| - bool HasOpenDataStreams() const override {
|
| - return session_->HasOpenDataStreams();
|
| - }
|
| -
|
| - private:
|
| - QuicSession* session_;
|
| -};
|
| -
|
| -QuicSession::QuicSession(QuicConnection* connection, const QuicConfig& config)
|
| - : connection_(connection),
|
| - visitor_shim_(new VisitorShim(this)),
|
| - config_(config),
|
| - max_open_streams_(config_.MaxStreamsPerConnection()),
|
| - next_stream_id_(is_server() ? 2 : 5),
|
| - largest_peer_created_stream_id_(0),
|
| - error_(QUIC_NO_ERROR),
|
| - flow_controller_(new QuicFlowController(
|
| - connection_.get(),
|
| - 0,
|
| - is_server(),
|
| - kMinimumFlowControlSendWindow,
|
| - config_.GetInitialSessionFlowControlWindowToSend(),
|
| - config_.GetInitialSessionFlowControlWindowToSend())),
|
| - goaway_received_(false),
|
| - goaway_sent_(false),
|
| - has_pending_handshake_(false) {
|
| -}
|
| -
|
| -void QuicSession::InitializeSession() {
|
| - connection_->set_visitor(visitor_shim_.get());
|
| - connection_->SetFromConfig(config_);
|
| - headers_stream_.reset(new QuicHeadersStream(this));
|
| -}
|
| -
|
| -QuicSession::~QuicSession() {
|
| - STLDeleteElements(&closed_streams_);
|
| - STLDeleteValues(&stream_map_);
|
| -
|
| - DLOG_IF(WARNING,
|
| - locally_closed_streams_highest_offset_.size() > max_open_streams_)
|
| - << "Surprisingly high number of locally closed streams still waiting for "
|
| - "final byte offset: " << locally_closed_streams_highest_offset_.size();
|
| -}
|
| -
|
| -void QuicSession::OnStreamFrames(const vector<QuicStreamFrame>& frames) {
|
| - for (size_t i = 0; i < frames.size(); ++i) {
|
| - // TODO(rch) deal with the error case of stream id 0.
|
| - const QuicStreamFrame& frame = frames[i];
|
| - QuicStreamId stream_id = frame.stream_id;
|
| - ReliableQuicStream* stream = GetStream(stream_id);
|
| - if (!stream) {
|
| - // The stream no longer exists, but we may still be interested in the
|
| - // final stream byte offset sent by the peer. A frame with a FIN can give
|
| - // us this offset.
|
| - if (frame.fin) {
|
| - QuicStreamOffset final_byte_offset =
|
| - frame.offset + frame.data.TotalBufferSize();
|
| - UpdateFlowControlOnFinalReceivedByteOffset(stream_id,
|
| - final_byte_offset);
|
| - }
|
| -
|
| - continue;
|
| - }
|
| - stream->OnStreamFrame(frames[i]);
|
| - }
|
| -}
|
| -
|
| -void QuicSession::OnStreamHeaders(QuicStreamId stream_id,
|
| - StringPiece headers_data) {
|
| - QuicDataStream* stream = GetDataStream(stream_id);
|
| - if (!stream) {
|
| - // It's quite possible to receive headers after a stream has been reset.
|
| - return;
|
| - }
|
| - stream->OnStreamHeaders(headers_data);
|
| -}
|
| -
|
| -void QuicSession::OnStreamHeadersPriority(QuicStreamId stream_id,
|
| - QuicPriority priority) {
|
| - QuicDataStream* stream = GetDataStream(stream_id);
|
| - if (!stream) {
|
| - // It's quite possible to receive headers after a stream has been reset.
|
| - return;
|
| - }
|
| - stream->OnStreamHeadersPriority(priority);
|
| -}
|
| -
|
| -void QuicSession::OnStreamHeadersComplete(QuicStreamId stream_id,
|
| - bool fin,
|
| - size_t frame_len) {
|
| - QuicDataStream* stream = GetDataStream(stream_id);
|
| - if (!stream) {
|
| - // It's quite possible to receive headers after a stream has been reset.
|
| - return;
|
| - }
|
| - stream->OnStreamHeadersComplete(fin, frame_len);
|
| -}
|
| -
|
| -void QuicSession::OnRstStream(const QuicRstStreamFrame& frame) {
|
| - if (frame.stream_id == kCryptoStreamId) {
|
| - connection()->SendConnectionCloseWithDetails(
|
| - QUIC_INVALID_STREAM_ID,
|
| - "Attempt to reset the crypto stream");
|
| - return;
|
| - }
|
| - if (frame.stream_id == kHeadersStreamId) {
|
| - connection()->SendConnectionCloseWithDetails(
|
| - QUIC_INVALID_STREAM_ID,
|
| - "Attempt to reset the headers stream");
|
| - return;
|
| - }
|
| -
|
| - QuicDataStream* stream = GetDataStream(frame.stream_id);
|
| - if (!stream) {
|
| - // The RST frame contains the final byte offset for the stream: we can now
|
| - // update the connection level flow controller if needed.
|
| - UpdateFlowControlOnFinalReceivedByteOffset(frame.stream_id,
|
| - frame.byte_offset);
|
| - return; // Errors are handled by GetStream.
|
| - }
|
| -
|
| - stream->OnStreamReset(frame);
|
| -}
|
| -
|
| -void QuicSession::OnGoAway(const QuicGoAwayFrame& frame) {
|
| - DCHECK(frame.last_good_stream_id < next_stream_id_);
|
| - goaway_received_ = true;
|
| -}
|
| -
|
| -void QuicSession::OnConnectionClosed(QuicErrorCode error, bool from_peer) {
|
| - DCHECK(!connection_->connected());
|
| - if (error_ == QUIC_NO_ERROR) {
|
| - error_ = error;
|
| - }
|
| -
|
| - while (!stream_map_.empty()) {
|
| - DataStreamMap::iterator it = stream_map_.begin();
|
| - QuicStreamId id = it->first;
|
| - it->second->OnConnectionClosed(error, from_peer);
|
| - // The stream should call CloseStream as part of OnConnectionClosed.
|
| - if (stream_map_.find(id) != stream_map_.end()) {
|
| - LOG(DFATAL) << ENDPOINT
|
| - << "Stream failed to close under OnConnectionClosed";
|
| - CloseStream(id);
|
| - }
|
| - }
|
| -}
|
| -
|
| -void QuicSession::OnSuccessfulVersionNegotiation(const QuicVersion& version) {
|
| - headers_stream_->OnSuccessfulVersionNegotiation(version);
|
| -}
|
| -
|
| -void QuicSession::OnWindowUpdateFrames(
|
| - const vector<QuicWindowUpdateFrame>& frames) {
|
| - bool connection_window_updated = false;
|
| - for (size_t i = 0; i < frames.size(); ++i) {
|
| - // Stream may be closed by the time we receive a WINDOW_UPDATE, so we can't
|
| - // assume that it still exists.
|
| - QuicStreamId stream_id = frames[i].stream_id;
|
| - if (stream_id == kConnectionLevelId) {
|
| - // This is a window update that applies to the connection, rather than an
|
| - // individual stream.
|
| - DVLOG(1) << ENDPOINT
|
| - << "Received connection level flow control window update with "
|
| - "byte offset: " << frames[i].byte_offset;
|
| - if (flow_controller_->UpdateSendWindowOffset(frames[i].byte_offset)) {
|
| - connection_window_updated = true;
|
| - }
|
| - continue;
|
| - }
|
| -
|
| - ReliableQuicStream* stream = GetStream(stream_id);
|
| - if (stream) {
|
| - stream->OnWindowUpdateFrame(frames[i]);
|
| - }
|
| - }
|
| -
|
| - // Connection level flow control window has increased, so blocked streams can
|
| - // write again.
|
| - if (connection_window_updated) {
|
| - OnCanWrite();
|
| - }
|
| -}
|
| -
|
| -void QuicSession::OnBlockedFrames(const vector<QuicBlockedFrame>& frames) {
|
| - for (size_t i = 0; i < frames.size(); ++i) {
|
| - // TODO(rjshade): Compare our flow control receive windows for specified
|
| - // streams: if we have a large window then maybe something
|
| - // had gone wrong with the flow control accounting.
|
| - DVLOG(1) << ENDPOINT << "Received BLOCKED frame with stream id: "
|
| - << frames[i].stream_id;
|
| - }
|
| -}
|
| -
|
| -void QuicSession::OnCanWrite() {
|
| - // We limit the number of writes to the number of pending streams. If more
|
| - // streams become pending, WillingAndAbleToWrite will be true, which will
|
| - // cause the connection to request resumption before yielding to other
|
| - // connections.
|
| - size_t num_writes = write_blocked_streams_.NumBlockedStreams();
|
| - if (flow_controller_->IsBlocked()) {
|
| - // If we are connection level flow control blocked, then only allow the
|
| - // crypto and headers streams to try writing as all other streams will be
|
| - // blocked.
|
| - num_writes = 0;
|
| - if (write_blocked_streams_.crypto_stream_blocked()) {
|
| - num_writes += 1;
|
| - }
|
| - if (write_blocked_streams_.headers_stream_blocked()) {
|
| - num_writes += 1;
|
| - }
|
| - }
|
| - if (num_writes == 0) {
|
| - return;
|
| - }
|
| -
|
| - QuicConnection::ScopedPacketBundler ack_bundler(
|
| - connection_.get(), QuicConnection::NO_ACK);
|
| - for (size_t i = 0; i < num_writes; ++i) {
|
| - if (!(write_blocked_streams_.HasWriteBlockedCryptoOrHeadersStream() ||
|
| - write_blocked_streams_.HasWriteBlockedDataStreams())) {
|
| - // Writing one stream removed another!? Something's broken.
|
| - LOG(DFATAL) << "WriteBlockedStream is missing";
|
| - connection_->CloseConnection(QUIC_INTERNAL_ERROR, false);
|
| - return;
|
| - }
|
| - if (!connection_->CanWriteStreamData()) {
|
| - return;
|
| - }
|
| - QuicStreamId stream_id = write_blocked_streams_.PopFront();
|
| - if (stream_id == kCryptoStreamId) {
|
| - has_pending_handshake_ = false; // We just popped it.
|
| - }
|
| - ReliableQuicStream* stream = GetStream(stream_id);
|
| - if (stream != nullptr && !stream->flow_controller()->IsBlocked()) {
|
| - // If the stream can't write all bytes, it'll re-add itself to the blocked
|
| - // list.
|
| - stream->OnCanWrite();
|
| - }
|
| - }
|
| -}
|
| -
|
| -bool QuicSession::WillingAndAbleToWrite() const {
|
| - // If the crypto or headers streams are blocked, we want to schedule a write -
|
| - // they don't get blocked by connection level flow control. Otherwise only
|
| - // schedule a write if we are not flow control blocked at the connection
|
| - // level.
|
| - return write_blocked_streams_.HasWriteBlockedCryptoOrHeadersStream() ||
|
| - (!flow_controller_->IsBlocked() &&
|
| - write_blocked_streams_.HasWriteBlockedDataStreams());
|
| -}
|
| -
|
| -bool QuicSession::HasPendingHandshake() const {
|
| - return has_pending_handshake_;
|
| -}
|
| -
|
| -bool QuicSession::HasOpenDataStreams() const {
|
| - return GetNumOpenStreams() > 0;
|
| -}
|
| -
|
| -QuicConsumedData QuicSession::WritevData(
|
| - QuicStreamId id,
|
| - const IOVector& data,
|
| - QuicStreamOffset offset,
|
| - bool fin,
|
| - FecProtection fec_protection,
|
| - QuicAckNotifier::DelegateInterface* ack_notifier_delegate) {
|
| - return connection_->SendStreamData(id, data, offset, fin, fec_protection,
|
| - ack_notifier_delegate);
|
| -}
|
| -
|
| -size_t QuicSession::WriteHeaders(
|
| - QuicStreamId id,
|
| - const SpdyHeaderBlock& headers,
|
| - bool fin,
|
| - QuicPriority priority,
|
| - QuicAckNotifier::DelegateInterface* ack_notifier_delegate) {
|
| - return headers_stream_->WriteHeaders(id, headers, fin, priority,
|
| - ack_notifier_delegate);
|
| -}
|
| -
|
| -void QuicSession::SendRstStream(QuicStreamId id,
|
| - QuicRstStreamErrorCode error,
|
| - QuicStreamOffset bytes_written) {
|
| - if (connection()->connected()) {
|
| - // Only send a RST_STREAM frame if still connected.
|
| - connection_->SendRstStream(id, error, bytes_written);
|
| - }
|
| - CloseStreamInner(id, true);
|
| -}
|
| -
|
| -void QuicSession::SendGoAway(QuicErrorCode error_code, const string& reason) {
|
| - if (goaway_sent_) {
|
| - return;
|
| - }
|
| - goaway_sent_ = true;
|
| - connection_->SendGoAway(error_code, largest_peer_created_stream_id_, reason);
|
| -}
|
| -
|
| -void QuicSession::CloseStream(QuicStreamId stream_id) {
|
| - CloseStreamInner(stream_id, false);
|
| -}
|
| -
|
| -void QuicSession::CloseStreamInner(QuicStreamId stream_id,
|
| - bool locally_reset) {
|
| - DVLOG(1) << ENDPOINT << "Closing stream " << stream_id;
|
| -
|
| - DataStreamMap::iterator it = stream_map_.find(stream_id);
|
| - if (it == stream_map_.end()) {
|
| - DVLOG(1) << ENDPOINT << "Stream is already closed: " << stream_id;
|
| - return;
|
| - }
|
| - QuicDataStream* stream = it->second;
|
| -
|
| - // Tell the stream that a RST has been sent.
|
| - if (locally_reset) {
|
| - stream->set_rst_sent(true);
|
| - }
|
| -
|
| - closed_streams_.push_back(it->second);
|
| -
|
| - // If we haven't received a FIN or RST for this stream, we need to keep track
|
| - // of the how many bytes the stream's flow controller believes it has
|
| - // received, for accurate connection level flow control accounting.
|
| - if (!stream->HasFinalReceivedByteOffset() &&
|
| - stream->flow_controller()->IsEnabled()) {
|
| - locally_closed_streams_highest_offset_[stream_id] =
|
| - stream->flow_controller()->highest_received_byte_offset();
|
| - }
|
| -
|
| - stream_map_.erase(it);
|
| - stream->OnClose();
|
| - // Decrease the number of streams being emulated when a new one is opened.
|
| - connection_->SetNumOpenStreams(stream_map_.size());
|
| -}
|
| -
|
| -void QuicSession::UpdateFlowControlOnFinalReceivedByteOffset(
|
| - QuicStreamId stream_id, QuicStreamOffset final_byte_offset) {
|
| - map<QuicStreamId, QuicStreamOffset>::iterator it =
|
| - locally_closed_streams_highest_offset_.find(stream_id);
|
| - if (it == locally_closed_streams_highest_offset_.end()) {
|
| - return;
|
| - }
|
| -
|
| - DVLOG(1) << ENDPOINT << "Received final byte offset " << final_byte_offset
|
| - << " for stream " << stream_id;
|
| - QuicByteCount offset_diff = final_byte_offset - it->second;
|
| - if (flow_controller_->UpdateHighestReceivedOffset(
|
| - flow_controller_->highest_received_byte_offset() + offset_diff)) {
|
| - // If the final offset violates flow control, close the connection now.
|
| - if (flow_controller_->FlowControlViolation()) {
|
| - connection_->SendConnectionClose(
|
| - QUIC_FLOW_CONTROL_RECEIVED_TOO_MUCH_DATA);
|
| - return;
|
| - }
|
| - }
|
| -
|
| - flow_controller_->AddBytesConsumed(offset_diff);
|
| - locally_closed_streams_highest_offset_.erase(it);
|
| -}
|
| -
|
| -bool QuicSession::IsEncryptionEstablished() {
|
| - return GetCryptoStream()->encryption_established();
|
| -}
|
| -
|
| -bool QuicSession::IsCryptoHandshakeConfirmed() {
|
| - return GetCryptoStream()->handshake_confirmed();
|
| -}
|
| -
|
| -void QuicSession::OnConfigNegotiated() {
|
| - connection_->SetFromConfig(config_);
|
| -
|
| - uint32 max_streams = config_.MaxStreamsPerConnection();
|
| - if (is_server()) {
|
| - // A server should accept a small number of additional streams beyond the
|
| - // limit sent to the client. This helps avoid early connection termination
|
| - // when FIN/RSTs for old streams are lost or arrive out of order.
|
| - // Use a minimum number of additional streams, or a percentage increase,
|
| - // whichever is larger.
|
| - max_streams =
|
| - max(max_streams + kMaxStreamsMinimumIncrement,
|
| - static_cast<uint32>(max_streams * kMaxStreamsMultiplier));
|
| - }
|
| - set_max_open_streams(max_streams);
|
| -
|
| - if (config_.HasReceivedInitialStreamFlowControlWindowBytes()) {
|
| - // Streams which were created before the SHLO was received (0-RTT
|
| - // requests) are now informed of the peer's initial flow control window.
|
| - OnNewStreamFlowControlWindow(
|
| - config_.ReceivedInitialStreamFlowControlWindowBytes());
|
| - }
|
| - if (config_.HasReceivedInitialSessionFlowControlWindowBytes()) {
|
| - OnNewSessionFlowControlWindow(
|
| - config_.ReceivedInitialSessionFlowControlWindowBytes());
|
| - }
|
| -}
|
| -
|
| -void QuicSession::OnNewStreamFlowControlWindow(QuicStreamOffset new_window) {
|
| - if (new_window < kMinimumFlowControlSendWindow) {
|
| - LOG(ERROR) << "Peer sent us an invalid stream flow control send window: "
|
| - << new_window
|
| - << ", below default: " << kMinimumFlowControlSendWindow;
|
| - if (connection_->connected()) {
|
| - connection_->SendConnectionClose(QUIC_FLOW_CONTROL_INVALID_WINDOW);
|
| - }
|
| - return;
|
| - }
|
| -
|
| - // Inform all existing streams about the new window.
|
| - GetCryptoStream()->UpdateSendWindowOffset(new_window);
|
| - headers_stream_->UpdateSendWindowOffset(new_window);
|
| - for (DataStreamMap::iterator it = stream_map_.begin();
|
| - it != stream_map_.end(); ++it) {
|
| - it->second->UpdateSendWindowOffset(new_window);
|
| - }
|
| -}
|
| -
|
| -void QuicSession::OnNewSessionFlowControlWindow(QuicStreamOffset new_window) {
|
| - if (new_window < kMinimumFlowControlSendWindow) {
|
| - LOG(ERROR) << "Peer sent us an invalid session flow control send window: "
|
| - << new_window
|
| - << ", below default: " << kMinimumFlowControlSendWindow;
|
| - if (connection_->connected()) {
|
| - connection_->SendConnectionClose(QUIC_FLOW_CONTROL_INVALID_WINDOW);
|
| - }
|
| - return;
|
| - }
|
| -
|
| - flow_controller_->UpdateSendWindowOffset(new_window);
|
| -}
|
| -
|
| -void QuicSession::OnCryptoHandshakeEvent(CryptoHandshakeEvent event) {
|
| - switch (event) {
|
| - // TODO(satyamshekhar): Move the logic of setting the encrypter/decrypter
|
| - // to QuicSession since it is the glue.
|
| - case ENCRYPTION_FIRST_ESTABLISHED:
|
| - break;
|
| -
|
| - case ENCRYPTION_REESTABLISHED:
|
| - // Retransmit originally packets that were sent, since they can't be
|
| - // decrypted by the peer.
|
| - connection_->RetransmitUnackedPackets(ALL_INITIAL_RETRANSMISSION);
|
| - break;
|
| -
|
| - case HANDSHAKE_CONFIRMED:
|
| - LOG_IF(DFATAL, !config_.negotiated()) << ENDPOINT
|
| - << "Handshake confirmed without parameter negotiation.";
|
| - // Discard originally encrypted packets, since they can't be decrypted by
|
| - // the peer.
|
| - connection_->NeuterUnencryptedPackets();
|
| - break;
|
| -
|
| - default:
|
| - LOG(ERROR) << ENDPOINT << "Got unknown handshake event: " << event;
|
| - }
|
| -}
|
| -
|
| -void QuicSession::OnCryptoHandshakeMessageSent(
|
| - const CryptoHandshakeMessage& message) {
|
| -}
|
| -
|
| -void QuicSession::OnCryptoHandshakeMessageReceived(
|
| - const CryptoHandshakeMessage& message) {
|
| -}
|
| -
|
| -QuicConfig* QuicSession::config() {
|
| - return &config_;
|
| -}
|
| -
|
| -void QuicSession::ActivateStream(QuicDataStream* stream) {
|
| - DVLOG(1) << ENDPOINT << "num_streams: " << stream_map_.size()
|
| - << ". activating " << stream->id();
|
| - DCHECK_EQ(stream_map_.count(stream->id()), 0u);
|
| - stream_map_[stream->id()] = stream;
|
| - // Increase the number of streams being emulated when a new one is opened.
|
| - connection_->SetNumOpenStreams(stream_map_.size());
|
| -}
|
| -
|
| -QuicStreamId QuicSession::GetNextStreamId() {
|
| - QuicStreamId id = next_stream_id_;
|
| - next_stream_id_ += 2;
|
| - return id;
|
| -}
|
| -
|
| -ReliableQuicStream* QuicSession::GetStream(const QuicStreamId stream_id) {
|
| - if (stream_id == kCryptoStreamId) {
|
| - return GetCryptoStream();
|
| - }
|
| - if (stream_id == kHeadersStreamId) {
|
| - return headers_stream_.get();
|
| - }
|
| - return GetDataStream(stream_id);
|
| -}
|
| -
|
| -QuicDataStream* QuicSession::GetDataStream(const QuicStreamId stream_id) {
|
| - if (stream_id == kCryptoStreamId) {
|
| - DLOG(FATAL) << "Attempt to call GetDataStream with the crypto stream id";
|
| - return nullptr;
|
| - }
|
| - if (stream_id == kHeadersStreamId) {
|
| - DLOG(FATAL) << "Attempt to call GetDataStream with the headers stream id";
|
| - return nullptr;
|
| - }
|
| -
|
| - DataStreamMap::iterator it = stream_map_.find(stream_id);
|
| - if (it != stream_map_.end()) {
|
| - return it->second;
|
| - }
|
| -
|
| - if (IsClosedStream(stream_id)) {
|
| - return nullptr;
|
| - }
|
| -
|
| - if (stream_id % 2 == next_stream_id_ % 2) {
|
| - // We've received a frame for a locally-created stream that is not
|
| - // currently active. This is an error.
|
| - if (connection()->connected()) {
|
| - connection()->SendConnectionClose(QUIC_PACKET_FOR_NONEXISTENT_STREAM);
|
| - }
|
| - return nullptr;
|
| - }
|
| -
|
| - return GetIncomingDataStream(stream_id);
|
| -}
|
| -
|
| -QuicDataStream* QuicSession::GetIncomingDataStream(QuicStreamId stream_id) {
|
| - if (IsClosedStream(stream_id)) {
|
| - return nullptr;
|
| - }
|
| -
|
| - implicitly_created_streams_.erase(stream_id);
|
| - if (stream_id > largest_peer_created_stream_id_) {
|
| - if (stream_id - largest_peer_created_stream_id_ > kMaxStreamIdDelta) {
|
| - // We may already have sent a connection close due to multiple reset
|
| - // streams in the same packet.
|
| - if (connection()->connected()) {
|
| - LOG(ERROR) << "Trying to get stream: " << stream_id
|
| - << ", largest peer created stream: "
|
| - << largest_peer_created_stream_id_
|
| - << ", max delta: " << kMaxStreamIdDelta;
|
| - connection()->SendConnectionClose(QUIC_INVALID_STREAM_ID);
|
| - }
|
| - return nullptr;
|
| - }
|
| - if (largest_peer_created_stream_id_ == 0) {
|
| - if (is_server()) {
|
| - largest_peer_created_stream_id_ = 3;
|
| - } else {
|
| - largest_peer_created_stream_id_ = 1;
|
| - }
|
| - }
|
| - for (QuicStreamId id = largest_peer_created_stream_id_ + 2;
|
| - id < stream_id;
|
| - id += 2) {
|
| - implicitly_created_streams_.insert(id);
|
| - }
|
| - largest_peer_created_stream_id_ = stream_id;
|
| - }
|
| - QuicDataStream* stream = CreateIncomingDataStream(stream_id);
|
| - if (stream == nullptr) {
|
| - return nullptr;
|
| - }
|
| - ActivateStream(stream);
|
| - return stream;
|
| -}
|
| -
|
| -void QuicSession::set_max_open_streams(size_t max_open_streams) {
|
| - DVLOG(1) << "Setting max_open_streams_ to " << max_open_streams;
|
| - max_open_streams_ = max_open_streams;
|
| -}
|
| -
|
| -bool QuicSession::IsClosedStream(QuicStreamId id) {
|
| - DCHECK_NE(0u, id);
|
| - if (id == kCryptoStreamId) {
|
| - return false;
|
| - }
|
| - if (id == kHeadersStreamId) {
|
| - return false;
|
| - }
|
| - if (ContainsKey(stream_map_, id)) {
|
| - // Stream is active
|
| - return false;
|
| - }
|
| - if (id % 2 == next_stream_id_ % 2) {
|
| - // Locally created streams are strictly in-order. If the id is in the
|
| - // range of created streams and it's not active, it must have been closed.
|
| - return id < next_stream_id_;
|
| - }
|
| - // For peer created streams, we also need to consider implicitly created
|
| - // streams.
|
| - return id <= largest_peer_created_stream_id_ &&
|
| - !ContainsKey(implicitly_created_streams_, id);
|
| -}
|
| -
|
| -size_t QuicSession::GetNumOpenStreams() const {
|
| - return stream_map_.size() + implicitly_created_streams_.size();
|
| -}
|
| -
|
| -void QuicSession::MarkWriteBlocked(QuicStreamId id, QuicPriority priority) {
|
| -#ifndef NDEBUG
|
| - ReliableQuicStream* stream = GetStream(id);
|
| - if (stream != nullptr) {
|
| - LOG_IF(DFATAL, priority != stream->EffectivePriority())
|
| - << ENDPOINT << "Stream " << id
|
| - << "Priorities do not match. Got: " << priority
|
| - << " Expected: " << stream->EffectivePriority();
|
| - } else {
|
| - LOG(DFATAL) << "Marking unknown stream " << id << " blocked.";
|
| - }
|
| -#endif
|
| -
|
| - if (id == kCryptoStreamId) {
|
| - DCHECK(!has_pending_handshake_);
|
| - has_pending_handshake_ = true;
|
| - // TODO(jar): Be sure to use the highest priority for the crypto stream,
|
| - // perhaps by adding a "special" priority for it that is higher than
|
| - // kHighestPriority.
|
| - priority = kHighestPriority;
|
| - }
|
| - write_blocked_streams_.PushBack(id, priority);
|
| -}
|
| -
|
| -bool QuicSession::HasDataToWrite() const {
|
| - return write_blocked_streams_.HasWriteBlockedCryptoOrHeadersStream() ||
|
| - write_blocked_streams_.HasWriteBlockedDataStreams() ||
|
| - connection_->HasQueuedData();
|
| -}
|
| -
|
| -bool QuicSession::GetSSLInfo(SSLInfo* ssl_info) const {
|
| - NOTIMPLEMENTED();
|
| - return false;
|
| -}
|
| -
|
| -void QuicSession::PostProcessAfterData() {
|
| - STLDeleteElements(&closed_streams_);
|
| -
|
| - if (connection()->connected() &&
|
| - locally_closed_streams_highest_offset_.size() > max_open_streams_) {
|
| - // A buggy client may fail to send FIN/RSTs. Don't tolerate this.
|
| - connection_->SendConnectionClose(QUIC_TOO_MANY_UNFINISHED_STREAMS);
|
| - }
|
| -}
|
| -
|
| -bool QuicSession::IsConnectionFlowControlBlocked() const {
|
| - return flow_controller_->IsBlocked();
|
| -}
|
| -
|
| -bool QuicSession::IsStreamFlowControlBlocked() {
|
| - if (headers_stream_->flow_controller()->IsBlocked() ||
|
| - GetCryptoStream()->flow_controller()->IsBlocked()) {
|
| - return true;
|
| - }
|
| - for (DataStreamMap::iterator it = stream_map_.begin();
|
| - it != stream_map_.end(); ++it) {
|
| - if (it->second->flow_controller()->IsBlocked()) {
|
| - return true;
|
| - }
|
| - }
|
| - return false;
|
| -}
|
| -
|
| -} // namespace net
|
|
|