quic_session.cc revision a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7
15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2012 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/quic/quic_session.h" 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 72a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "base/stl_util.h" 8eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch#include "net/quic/crypto/proof_verifier.h" 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/quic/quic_connection.h" 107dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch#include "net/ssl/ssl_info.h" 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using base::StringPiece; 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using base::hash_map; 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using base::hash_set; 152385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdochusing std::make_pair; 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using std::vector; 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace net { 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 202385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdochconst size_t kMaxPrematurelyClosedStreamsTracked = 20; 21424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles)const size_t kMaxZombieStreams = 20; 222385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch 23868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)#define ENDPOINT (is_server_ ? "Server: " : " Client: ") 24868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) 252a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// We want to make sure we delete any closed streams in a safe manner. 262a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// To avoid deleting a stream in mid-operation, we have a simple shim between 272a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// us and the stream, so we can delete any streams when we return from 282a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// processing. 292a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// 302a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// We could just override the base methods, but this makes it easier to make 312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// sure we don't miss any. 322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)class VisitorShim : public QuicConnectionVisitorInterface { 332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) public: 342a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) explicit VisitorShim(QuicSession* session) : session_(session) {} 352a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 36d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) virtual bool OnStreamFrames(const vector<QuicStreamFrame>& frames) OVERRIDE { 37d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) bool accepted = session_->OnStreamFrames(frames); 382a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) session_->PostProcessAfterData(); 392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return accepted; 402a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 412a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) virtual void OnRstStream(const QuicRstStreamFrame& frame) OVERRIDE { 422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) session_->OnRstStream(frame); 432a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) session_->PostProcessAfterData(); 442a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 452a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 462a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) virtual void OnGoAway(const QuicGoAwayFrame& frame) OVERRIDE { 472a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) session_->OnGoAway(frame); 482a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) session_->PostProcessAfterData(); 492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 502a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) virtual bool OnCanWrite() OVERRIDE { 522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) bool rc = session_->OnCanWrite(); 532a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) session_->PostProcessAfterData(); 542a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return rc; 552a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 562a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 57d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) virtual void OnSuccessfulVersionNegotiation( 58d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) const QuicVersion& version) OVERRIDE { 59d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) session_->OnSuccessfulVersionNegotiation(version); 60d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) } 61d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) 620f1bc08d4cfcc34181b0b5cbf065c40f687bf740Torne (Richard Coles) virtual void OnConfigNegotiated() OVERRIDE { 630f1bc08d4cfcc34181b0b5cbf065c40f687bf740Torne (Richard Coles) session_->OnConfigNegotiated(); 640f1bc08d4cfcc34181b0b5cbf065c40f687bf740Torne (Richard Coles) } 650f1bc08d4cfcc34181b0b5cbf065c40f687bf740Torne (Richard Coles) 661e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) virtual void OnConnectionClosed(QuicErrorCode error, 671e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) bool from_peer) OVERRIDE { 681e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) session_->OnConnectionClosed(error, from_peer); 692a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // The session will go away, so don't bother with cleanup. 702a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 712a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 72d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) virtual bool HasPendingHandshake() const OVERRIDE { 73d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) return session_->HasPendingHandshake(); 74d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) } 75d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) 762a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) private: 772a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) QuicSession* session_; 782a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)}; 792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 8090dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)QuicSession::QuicSession(QuicConnection* connection, 8190dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) const QuicConfig& config, 8290dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) bool is_server) 835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) : connection_(connection), 842a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) visitor_shim_(new VisitorShim(this)), 8590dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) config_(config), 86ba5b9a6411cb1792fd21f0a078d7a25cd1ceec16Ben Murdoch max_open_streams_(config_.max_streams_per_connection()), 875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) next_stream_id_(is_server ? 2 : 3), 885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) is_server_(is_server), 892a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) largest_peer_created_stream_id_(0), 90868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) error_(QUIC_NO_ERROR), 912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) goaway_received_(false), 92d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) goaway_sent_(false), 93d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) has_pending_handshake_(false) { 94868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) 95868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) connection_->set_visitor(visitor_shim_.get()); 960f1bc08d4cfcc34181b0b5cbf065c40f687bf740Torne (Richard Coles) connection_->SetFromConfig(config_); 97ba5b9a6411cb1792fd21f0a078d7a25cd1ceec16Ben Murdoch if (connection_->connected()) { 98ba5b9a6411cb1792fd21f0a078d7a25cd1ceec16Ben Murdoch connection_->SetOverallConnectionTimeout( 99ba5b9a6411cb1792fd21f0a078d7a25cd1ceec16Ben Murdoch config_.max_time_before_crypto_handshake()); 100ba5b9a6411cb1792fd21f0a078d7a25cd1ceec16Ben Murdoch } 1015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)QuicSession::~QuicSession() { 1042a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) STLDeleteElements(&closed_streams_); 1052a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) STLDeleteValues(&stream_map_); 1065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 108d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles)bool QuicSession::OnStreamFrames(const vector<QuicStreamFrame>& frames) { 1095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (size_t i = 0; i < frames.size(); ++i) { 1105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // TODO(rch) deal with the error case of stream id 0 1112385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch if (IsClosedStream(frames[i].stream_id)) { 1122385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch // If we get additional frames for a stream where we didn't process 1132385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch // headers, it's highly likely our compression context will end up 1142385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch // permanently out of sync with the peer's, so we give up and close the 1152385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch // connection. 1162385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch if (ContainsKey(prematurely_closed_streams_, frames[i].stream_id)) { 1172385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch connection()->SendConnectionClose( 1182385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch QUIC_STREAM_RST_BEFORE_HEADERS_DECOMPRESSED); 1192385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch return false; 1202385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch } 1212385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch continue; 1222385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch } 1235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ReliableQuicStream* stream = GetStream(frames[i].stream_id); 1255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (stream == NULL) return false; 1265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!stream->WillAcceptStreamFrame(frames[i])) return false; 1275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // TODO(alyssar) check against existing connection address: if changed, make 1295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // sure we update the connection. 1305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (size_t i = 0; i < frames.size(); ++i) { 133424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) QuicStreamId stream_id = frames[i].stream_id; 134424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) ReliableQuicStream* stream = GetStream(stream_id); 135424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) if (!stream) { 136424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) continue; 137424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) } 138424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) stream->OnStreamFrame(frames[i]); 139424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) 140424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) // If the stream had been prematurely closed, and the 141424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) // headers are now decompressed, then we are finally finished 142424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) // with this stream. 143424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) if (ContainsKey(zombie_streams_, stream_id) && 144424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) stream->headers_decompressed()) { 145424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) CloseZombieStream(stream_id); 1465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 148b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) 149b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) while (!decompression_blocked_streams_.empty()) { 150b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) QuicHeaderId header_id = decompression_blocked_streams_.begin()->first; 151868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) if (header_id != decompressor_.current_header_id()) { 152868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) break; 153b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) } 154868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) QuicStreamId stream_id = decompression_blocked_streams_.begin()->second; 155868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) decompression_blocked_streams_.erase(header_id); 156868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) ReliableQuicStream* stream = GetStream(stream_id); 157868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) if (!stream) { 158868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) connection()->SendConnectionClose( 159868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) QUIC_STREAM_RST_BEFORE_HEADERS_DECOMPRESSED); 160ba5b9a6411cb1792fd21f0a078d7a25cd1ceec16Ben Murdoch return false; 161868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) } 162868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) stream->OnDecompressorAvailable(); 163b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) } 1645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return true; 1655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void QuicSession::OnRstStream(const QuicRstStreamFrame& frame) { 1685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ReliableQuicStream* stream = GetStream(frame.stream_id); 1695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!stream) { 1705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; // Errors are handled by GetStream. 1715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 172424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) if (ContainsKey(zombie_streams_, stream->id())) { 173424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) // If this was a zombie stream then we close it out now. 174424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) CloseZombieStream(stream->id()); 175424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) // However, since the headers still have not been decompressed, we want to 176424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) // mark it a prematurely closed so that if we ever receive frames 177424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) // for this stream we can close the connection. 178424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) DCHECK(!stream->headers_decompressed()); 179424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) AddPrematurelyClosedStream(frame.stream_id); 180424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) return; 181424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) } 182f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) if (stream->stream_bytes_read() > 0 && !stream->headers_decompressed()) { 183f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) connection()->SendConnectionClose( 184f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) QUIC_STREAM_RST_BEFORE_HEADERS_DECOMPRESSED); 185f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) } 1862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) stream->OnStreamReset(frame.error_code); 1872a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 1882a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1892a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void QuicSession::OnGoAway(const QuicGoAwayFrame& frame) { 1902a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK(frame.last_good_stream_id < next_stream_id_); 1912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) goaway_received_ = true; 1925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1941e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles)void QuicSession::OnConnectionClosed(QuicErrorCode error, bool from_peer) { 195424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) DCHECK(!connection_->connected()); 196868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) if (error_ == QUIC_NO_ERROR) { 197868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) error_ = error; 198868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) } 199868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) 200a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) while (!stream_map_.empty()) { 2015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ReliableStreamMap::iterator it = stream_map_.begin(); 2025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) QuicStreamId id = it->first; 2031e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) it->second->OnConnectionClosed(error, from_peer); 2041e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) // The stream should call CloseStream as part of OnConnectionClosed. 2055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (stream_map_.find(id) != stream_map_.end()) { 2061e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) LOG(DFATAL) << ENDPOINT 2071e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) << "Stream failed to close under OnConnectionClosed"; 2085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CloseStream(id); 2095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)bool QuicSession::OnCanWrite() { 2142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // We latch this here rather than doing a traditional loop, because streams 2152a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // may be modifying the list as we loop. 2163551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) int remaining_writes = write_blocked_streams_.NumBlockedStreams(); 2172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2182a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) while (!connection_->HasQueuedData() && 2192a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) remaining_writes > 0) { 2203551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) DCHECK(write_blocked_streams_.HasWriteBlockedStreams()); 221f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) if (!write_blocked_streams_.HasWriteBlockedStreams()) { 222d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) LOG(DFATAL) << "WriteBlockedStream is missing"; 223d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) connection_->CloseConnection(QUIC_INTERNAL_ERROR, false); 224d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) return true; // We have no write blocked streams. 2253551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) } 226f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) int index = write_blocked_streams_.GetHighestPriorityWriteBlockedList(); 227d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) QuicStreamId stream_id = write_blocked_streams_.PopFront(index); 228d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) if (stream_id == kCryptoStreamId) { 229d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) has_pending_handshake_ = false; // We just popped it. 230d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) } 231d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) ReliableQuicStream* stream = GetStream(stream_id); 2322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (stream != NULL) { 2332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // If the stream can't write all bytes, it'll re-add itself to the blocked 2342a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // list. 2352a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) stream->OnCanWrite(); 2362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 2372a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) --remaining_writes; 2382a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 2392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2403551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) return !write_blocked_streams_.HasWriteBlockedStreams(); 2412a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 2422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 243d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles)bool QuicSession::HasPendingHandshake() const { 244d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) return has_pending_handshake_; 245d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles)} 246d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) 247f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles)QuicConsumedData QuicSession::WritevData( 248f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) QuicStreamId id, 249f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) const struct iovec* iov, 250f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) int iov_count, 251f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) QuicStreamOffset offset, 252f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) bool fin, 253f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) QuicAckNotifier::DelegateInterface* ack_notifier_delegate) { 2541e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) IOVector data; 2551e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) data.AppendIovec(iov, iov_count); 256f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) return connection_->SendStreamData(id, data, offset, fin, 257f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) ack_notifier_delegate); 2585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void QuicSession::SendRstStream(QuicStreamId id, 261c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) QuicRstStreamErrorCode error) { 2622a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) connection_->SendRstStream(id, error); 263424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) CloseStreamInner(id, true); 2645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2662a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void QuicSession::SendGoAway(QuicErrorCode error_code, const string& reason) { 2672a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) goaway_sent_ = true; 2682a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) connection_->SendGoAway(error_code, largest_peer_created_stream_id_, reason); 2692a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 2702a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void QuicSession::CloseStream(QuicStreamId stream_id) { 272424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) CloseStreamInner(stream_id, false); 273424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles)} 274424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) 275424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles)void QuicSession::CloseStreamInner(QuicStreamId stream_id, 276424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) bool locally_reset) { 277f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) DVLOG(1) << ENDPOINT << "Closing stream " << stream_id; 2785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ReliableStreamMap::iterator it = stream_map_.find(stream_id); 2805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (it == stream_map_.end()) { 281f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) DVLOG(1) << ENDPOINT << "Stream is already closed: " << stream_id; 2825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 2835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 284c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) ReliableQuicStream* stream = it->second; 285424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) if (connection_->connected() && !stream->headers_decompressed()) { 286424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) // If the stream is being closed locally (for example a client cancelling 287424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) // a request before receiving the response) then we need to make sure that 288424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) // we keep the stream alive long enough to process any response or 289424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) // RST_STREAM frames. 290424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) if (locally_reset && !is_server_) { 291424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) AddZombieStream(stream_id); 292424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) return; 2932385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch } 294424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) 295424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) // This stream has been closed before the headers were decompressed. 296424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) // This might cause problems with head of line blocking of headers. 297424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) // If the peer sent headers which were lost but we now close the stream 298424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) // we will never be able to decompress headers for other streams. 299424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) // To deal with this, we keep track of streams which have been closed 300424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) // prematurely. If we ever receive data frames for this steam, then we 301424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) // know there actually has been a problem and we close the connection. 302424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) AddPrematurelyClosedStream(stream->id()); 3032385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch } 3042a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) closed_streams_.push_back(it->second); 30558537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) if (ContainsKey(zombie_streams_, stream->id())) { 30658537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) zombie_streams_.erase(stream->id()); 30758537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) } 3085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) stream_map_.erase(it); 309c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) stream->OnClose(); 3105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 312424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles)void QuicSession::AddZombieStream(QuicStreamId stream_id) { 313424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) if (zombie_streams_.size() == kMaxZombieStreams) { 314424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) QuicStreamId oldest_zombie_stream_id = zombie_streams_.begin()->first; 315424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) CloseZombieStream(oldest_zombie_stream_id); 316424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) // However, since the headers still have not been decompressed, we want to 317424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) // mark it a prematurely closed so that if we ever receive frames 318424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) // for this stream we can close the connection. 319424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) AddPrematurelyClosedStream(oldest_zombie_stream_id); 320424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) } 321424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) zombie_streams_.insert(make_pair(stream_id, true)); 322424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles)} 323424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) 324424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles)void QuicSession::CloseZombieStream(QuicStreamId stream_id) { 325424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) DCHECK(ContainsKey(zombie_streams_, stream_id)); 326424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) zombie_streams_.erase(stream_id); 327424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) ReliableQuicStream* stream = GetStream(stream_id); 328424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) if (!stream) { 329424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) return; 330424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) } 331424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) stream_map_.erase(stream_id); 332424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) stream->OnClose(); 333424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) closed_streams_.push_back(stream); 334424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles)} 335424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) 336424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles)void QuicSession::AddPrematurelyClosedStream(QuicStreamId stream_id) { 337424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) if (prematurely_closed_streams_.size() == 338424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) kMaxPrematurelyClosedStreamsTracked) { 339424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) prematurely_closed_streams_.erase(prematurely_closed_streams_.begin()); 340424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) } 341424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) prematurely_closed_streams_.insert(make_pair(stream_id, true)); 342424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles)} 343424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) 344c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)bool QuicSession::IsEncryptionEstablished() { 345c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) return GetCryptoStream()->encryption_established(); 3465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 348c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)bool QuicSession::IsCryptoHandshakeConfirmed() { 349c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) return GetCryptoStream()->handshake_confirmed(); 350c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)} 351c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 3520f1bc08d4cfcc34181b0b5cbf065c40f687bf740Torne (Richard Coles)void QuicSession::OnConfigNegotiated() { 3530f1bc08d4cfcc34181b0b5cbf065c40f687bf740Torne (Richard Coles) connection_->SetFromConfig(config_); 3540f1bc08d4cfcc34181b0b5cbf065c40f687bf740Torne (Richard Coles)} 3550f1bc08d4cfcc34181b0b5cbf065c40f687bf740Torne (Richard Coles) 356c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)void QuicSession::OnCryptoHandshakeEvent(CryptoHandshakeEvent event) { 35790dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) switch (event) { 35890dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) // TODO(satyamshekhar): Move the logic of setting the encrypter/decrypter 35990dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) // to QuicSession since it is the glue. 36090dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) case ENCRYPTION_FIRST_ESTABLISHED: 36190dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) break; 36290dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 36390dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) case ENCRYPTION_REESTABLISHED: 36490dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) // Retransmit originally packets that were sent, since they can't be 36590dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) // decrypted by the peer. 366a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) connection_->RetransmitUnackedPackets(INITIAL_ENCRYPTION_ONLY); 36790dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) break; 36890dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 36990dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) case HANDSHAKE_CONFIRMED: 370868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) LOG_IF(DFATAL, !config_.negotiated()) << ENDPOINT 37190dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) << "Handshake confirmed without parameter negotiation."; 372868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) connection_->SetOverallConnectionTimeout(QuicTime::Delta::Infinite()); 37390dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) max_open_streams_ = config_.max_streams_per_connection(); 37490dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) break; 37590dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 37690dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) default: 377868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) LOG(ERROR) << ENDPOINT << "Got unknown handshake event: " << event; 378b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) } 3792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 3802a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3813551c9c881056c480085172ff9840cab31610854Torne (Richard Coles)void QuicSession::OnCryptoHandshakeMessageSent( 3823551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) const CryptoHandshakeMessage& message) { 3833551c9c881056c480085172ff9840cab31610854Torne (Richard Coles)} 3843551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) 3853551c9c881056c480085172ff9840cab31610854Torne (Richard Coles)void QuicSession::OnCryptoHandshakeMessageReceived( 3863551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) const CryptoHandshakeMessage& message) { 3873551c9c881056c480085172ff9840cab31610854Torne (Richard Coles)} 3883551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) 38990dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)QuicConfig* QuicSession::config() { 39090dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) return &config_; 39190dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)} 39290dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 3935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void QuicSession::ActivateStream(ReliableQuicStream* stream) { 394f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) DVLOG(1) << ENDPOINT << "num_streams: " << stream_map_.size() 395868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) << ". activating " << stream->id(); 396d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) DCHECK_EQ(stream_map_.count(stream->id()), 0u); 3975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) stream_map_[stream->id()] = stream; 3985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)QuicStreamId QuicSession::GetNextStreamId() { 4015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) QuicStreamId id = next_stream_id_; 4025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) next_stream_id_ += 2; 4035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return id; 4045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 4055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)ReliableQuicStream* QuicSession::GetStream(const QuicStreamId stream_id) { 4075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (stream_id == kCryptoStreamId) { 4085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return GetCryptoStream(); 4095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 4105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ReliableStreamMap::iterator it = stream_map_.find(stream_id); 4125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (it != stream_map_.end()) { 4135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return it->second; 4145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 4155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4162a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (IsClosedStream(stream_id)) { 4172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return NULL; 4182a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 4192a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 4205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (stream_id % 2 == next_stream_id_ % 2) { 4215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // We've received a frame for a locally-created stream that is not 4225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // currently active. This is an error. 4235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) connection()->SendConnectionClose(QUIC_PACKET_FOR_NONEXISTENT_STREAM); 4245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return NULL; 4255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 4265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return GetIncomingReliableStream(stream_id); 4285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 4295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)ReliableQuicStream* QuicSession::GetIncomingReliableStream( 4315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) QuicStreamId stream_id) { 4325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (IsClosedStream(stream_id)) { 4335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return NULL; 4345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 4355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (goaway_sent_) { 4372a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // We've already sent a GoAway 438c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) SendRstStream(stream_id, QUIC_STREAM_PEER_GOING_AWAY); 4392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return NULL; 4402a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 4412a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 4425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) implicitly_created_streams_.erase(stream_id); 4435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (stream_id > largest_peer_created_stream_id_) { 4445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // TODO(rch) add unit test for this 4455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (stream_id - largest_peer_created_stream_id_ > kMaxStreamIdDelta) { 4465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) connection()->SendConnectionClose(QUIC_INVALID_STREAM_ID); 4475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return NULL; 4485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 44958537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) if (largest_peer_created_stream_id_ == 0) { 45058537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) largest_peer_created_stream_id_= 1; 45158537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) } 45258537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) for (QuicStreamId id = largest_peer_created_stream_id_ + 2; 45358537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) id < stream_id; 45458537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) id += 2) { 45558537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) implicitly_created_streams_.insert(id); 4565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 4575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) largest_peer_created_stream_id_ = stream_id; 4585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 4595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ReliableQuicStream* stream = CreateIncomingReliableStream(stream_id); 4605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (stream == NULL) { 4615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return NULL; 4625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 4635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ActivateStream(stream); 4645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return stream; 4655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 4665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool QuicSession::IsClosedStream(QuicStreamId id) { 4685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK_NE(0u, id); 4695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (id == kCryptoStreamId) { 4705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return false; 4715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 472424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) if (ContainsKey(zombie_streams_, id)) { 473424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) return true; 474424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) } 475424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) if (ContainsKey(stream_map_, id)) { 4765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Stream is active 4775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return false; 4785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 4795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (id % 2 == next_stream_id_ % 2) { 4805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Locally created streams are strictly in-order. If the id is in the 4815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // range of created streams and it's not active, it must have been closed. 4825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return id < next_stream_id_; 4835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 4845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // For peer created streams, we also need to consider implicitly created 4855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // streams. 4862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return id <= largest_peer_created_stream_id_ && 4872a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) implicitly_created_streams_.count(id) == 0; 4882a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 4895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4902a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)size_t QuicSession::GetNumOpenStreams() const { 491424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) return stream_map_.size() + implicitly_created_streams_.size() - 492424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) zombie_streams_.size(); 4935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 4945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 495d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles)void QuicSession::MarkWriteBlocked(QuicStreamId id, QuicPriority priority) { 496d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) if (id == kCryptoStreamId) { 497d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) DCHECK(!has_pending_handshake_); 498d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) has_pending_handshake_ = true; 499d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) // TODO(jar): Be sure to use the highest priority for the crypto stream, 500d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) // perhaps by adding a "special" priority for it that is higher than 501d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) // kHighestPriority. 502d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) priority = kHighestPriority; 503d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) } 504d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) write_blocked_streams_.PushBack(id, priority); 5052a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 5062a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 507f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles)bool QuicSession::HasQueuedData() const { 508f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) return write_blocked_streams_.NumBlockedStreams() || 509f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) connection_->HasQueuedData(); 510f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles)} 511f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) 512b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles)void QuicSession::MarkDecompressionBlocked(QuicHeaderId header_id, 513b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) QuicStreamId stream_id) { 514b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) decompression_blocked_streams_[header_id] = stream_id; 515b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles)} 516b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) 5177dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdochbool QuicSession::GetSSLInfo(SSLInfo* ssl_info) { 5187dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch NOTIMPLEMENTED(); 5197dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch return false; 5207dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch} 5217dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch 5222a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void QuicSession::PostProcessAfterData() { 5232a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) STLDeleteElements(&closed_streams_); 5242a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) closed_streams_.clear(); 5252a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 5262a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 5275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace net 528