quic_session.cc revision d0247b1b59f9c528cb6df88b4f2b9afaf80d181e
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) 622a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) virtual void ConnectionClose(QuicErrorCode error, bool from_peer) OVERRIDE { 632a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) session_->ConnectionClose(error, from_peer); 642a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // The session will go away, so don't bother with cleanup. 652a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 662a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 67d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) virtual bool HasPendingHandshake() const OVERRIDE { 68d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) return session_->HasPendingHandshake(); 69d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) } 70d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) 712a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) private: 722a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) QuicSession* session_; 732a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)}; 742a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 7590dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)QuicSession::QuicSession(QuicConnection* connection, 7690dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) const QuicConfig& config, 7790dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) bool is_server) 785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) : connection_(connection), 792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) visitor_shim_(new VisitorShim(this)), 8090dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) config_(config), 81ba5b9a6411cb1792fd21f0a078d7a25cd1ceec16Ben Murdoch max_open_streams_(config_.max_streams_per_connection()), 825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) next_stream_id_(is_server ? 2 : 3), 835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) is_server_(is_server), 842a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) largest_peer_created_stream_id_(0), 85868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) error_(QUIC_NO_ERROR), 862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) goaway_received_(false), 87d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) goaway_sent_(false), 88d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) has_pending_handshake_(false) { 89868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) 90868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) connection_->set_visitor(visitor_shim_.get()); 91868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) connection_->SetIdleNetworkTimeout(config_.idle_connection_state_lifetime()); 92ba5b9a6411cb1792fd21f0a078d7a25cd1ceec16Ben Murdoch if (connection_->connected()) { 93ba5b9a6411cb1792fd21f0a078d7a25cd1ceec16Ben Murdoch connection_->SetOverallConnectionTimeout( 94ba5b9a6411cb1792fd21f0a078d7a25cd1ceec16Ben Murdoch config_.max_time_before_crypto_handshake()); 95ba5b9a6411cb1792fd21f0a078d7a25cd1ceec16Ben Murdoch } 96868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) // TODO(satyamshekhar): Set congestion control and ICSL also. 975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)QuicSession::~QuicSession() { 1002a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) STLDeleteElements(&closed_streams_); 1012a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) STLDeleteValues(&stream_map_); 1025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 104d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles)bool QuicSession::OnStreamFrames(const vector<QuicStreamFrame>& frames) { 1055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (size_t i = 0; i < frames.size(); ++i) { 1065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // TODO(rch) deal with the error case of stream id 0 1072385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch if (IsClosedStream(frames[i].stream_id)) { 1082385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch // If we get additional frames for a stream where we didn't process 1092385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch // headers, it's highly likely our compression context will end up 1102385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch // permanently out of sync with the peer's, so we give up and close the 1112385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch // connection. 1122385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch if (ContainsKey(prematurely_closed_streams_, frames[i].stream_id)) { 1132385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch connection()->SendConnectionClose( 1142385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch QUIC_STREAM_RST_BEFORE_HEADERS_DECOMPRESSED); 1152385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch return false; 1162385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch } 1172385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch continue; 1182385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch } 1195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ReliableQuicStream* stream = GetStream(frames[i].stream_id); 1215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (stream == NULL) return false; 1225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!stream->WillAcceptStreamFrame(frames[i])) return false; 1235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // TODO(alyssar) check against existing connection address: if changed, make 1255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // sure we update the connection. 1265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (size_t i = 0; i < frames.size(); ++i) { 129424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) QuicStreamId stream_id = frames[i].stream_id; 130424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) ReliableQuicStream* stream = GetStream(stream_id); 131424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) if (!stream) { 132424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) continue; 133424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) } 134424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) stream->OnStreamFrame(frames[i]); 135424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) 136424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) // If the stream had been prematurely closed, and the 137424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) // headers are now decompressed, then we are finally finished 138424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) // with this stream. 139424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) if (ContainsKey(zombie_streams_, stream_id) && 140424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) stream->headers_decompressed()) { 141424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) CloseZombieStream(stream_id); 1425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 144b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) 145b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) while (!decompression_blocked_streams_.empty()) { 146b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) QuicHeaderId header_id = decompression_blocked_streams_.begin()->first; 147868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) if (header_id != decompressor_.current_header_id()) { 148868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) break; 149b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) } 150868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) QuicStreamId stream_id = decompression_blocked_streams_.begin()->second; 151868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) decompression_blocked_streams_.erase(header_id); 152868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) ReliableQuicStream* stream = GetStream(stream_id); 153868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) if (!stream) { 154868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) connection()->SendConnectionClose( 155868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) QUIC_STREAM_RST_BEFORE_HEADERS_DECOMPRESSED); 156ba5b9a6411cb1792fd21f0a078d7a25cd1ceec16Ben Murdoch return false; 157868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) } 158868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) stream->OnDecompressorAvailable(); 159b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) } 1605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return true; 1615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void QuicSession::OnRstStream(const QuicRstStreamFrame& frame) { 1645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ReliableQuicStream* stream = GetStream(frame.stream_id); 1655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!stream) { 1665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; // Errors are handled by GetStream. 1675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 168424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) if (ContainsKey(zombie_streams_, stream->id())) { 169424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) // If this was a zombie stream then we close it out now. 170424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) CloseZombieStream(stream->id()); 171424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) // However, since the headers still have not been decompressed, we want to 172424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) // mark it a prematurely closed so that if we ever receive frames 173424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) // for this stream we can close the connection. 174424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) DCHECK(!stream->headers_decompressed()); 175424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) AddPrematurelyClosedStream(frame.stream_id); 176424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) return; 177424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) } 1782a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) stream->OnStreamReset(frame.error_code); 1792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 1802a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1812a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void QuicSession::OnGoAway(const QuicGoAwayFrame& frame) { 1822a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK(frame.last_good_stream_id < next_stream_id_); 1832a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) goaway_received_ = true; 1845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void QuicSession::ConnectionClose(QuicErrorCode error, bool from_peer) { 187424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) DCHECK(!connection_->connected()); 188868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) if (error_ == QUIC_NO_ERROR) { 189868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) error_ = error; 190868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) } 191868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) 1925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) while (stream_map_.size() != 0) { 1935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ReliableStreamMap::iterator it = stream_map_.begin(); 1945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) QuicStreamId id = it->first; 1955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) it->second->ConnectionClose(error, from_peer); 1965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // The stream should call CloseStream as part of ConnectionClose. 1975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (stream_map_.find(id) != stream_map_.end()) { 198868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) LOG(DFATAL) << ENDPOINT << "Stream failed to close under ConnectionClose"; 1995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CloseStream(id); 2005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2042a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)bool QuicSession::OnCanWrite() { 2052a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // We latch this here rather than doing a traditional loop, because streams 2062a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // may be modifying the list as we loop. 2073551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) int remaining_writes = write_blocked_streams_.NumBlockedStreams(); 2082a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2092a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) while (!connection_->HasQueuedData() && 2102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) remaining_writes > 0) { 2113551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) DCHECK(write_blocked_streams_.HasWriteBlockedStreams()); 2123551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) int index = write_blocked_streams_.GetHighestPriorityWriteBlockedList(); 213d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) if (index == -1) { 214d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) LOG(DFATAL) << "WriteBlockedStream is missing"; 215d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) connection_->CloseConnection(QUIC_INTERNAL_ERROR, false); 216d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) return true; // We have no write blocked streams. 2173551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) } 218d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) QuicStreamId stream_id = write_blocked_streams_.PopFront(index); 219d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) if (stream_id == kCryptoStreamId) { 220d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) has_pending_handshake_ = false; // We just popped it. 221d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) } 222d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) ReliableQuicStream* stream = GetStream(stream_id); 2232a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (stream != NULL) { 2242a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // If the stream can't write all bytes, it'll re-add itself to the blocked 2252a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // list. 2262a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) stream->OnCanWrite(); 2272a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 2282a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) --remaining_writes; 2292a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 2302a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2313551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) return !write_blocked_streams_.HasWriteBlockedStreams(); 2322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 2332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 234d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles)bool QuicSession::HasPendingHandshake() const { 235d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) return has_pending_handshake_; 236d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles)} 237d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) 238d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles)QuicConsumedData QuicSession::WritevData(QuicStreamId id, 239d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) const struct iovec* iov, 240d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) int iov_count, 241d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) QuicStreamOffset offset, 242d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) bool fin) { 243d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) return connection_->SendvStreamData(id, iov, iov_count, offset, fin); 2445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void QuicSession::SendRstStream(QuicStreamId id, 247c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) QuicRstStreamErrorCode error) { 2482a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) connection_->SendRstStream(id, error); 249424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) CloseStreamInner(id, true); 2505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void QuicSession::SendGoAway(QuicErrorCode error_code, const string& reason) { 2532a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) goaway_sent_ = true; 2542a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) connection_->SendGoAway(error_code, largest_peer_created_stream_id_, reason); 2552a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 2562a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void QuicSession::CloseStream(QuicStreamId stream_id) { 258424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) CloseStreamInner(stream_id, false); 259424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles)} 260424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) 261424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles)void QuicSession::CloseStreamInner(QuicStreamId stream_id, 262424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) bool locally_reset) { 263868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) DLOG(INFO) << ENDPOINT << "Closing stream " << stream_id; 2645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ReliableStreamMap::iterator it = stream_map_.find(stream_id); 2665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (it == stream_map_.end()) { 267868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) DLOG(INFO) << ENDPOINT << "Stream is already closed: " << stream_id; 2685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 2695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 270c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) ReliableQuicStream* stream = it->second; 271424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) if (connection_->connected() && !stream->headers_decompressed()) { 272424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) // If the stream is being closed locally (for example a client cancelling 273424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) // a request before receiving the response) then we need to make sure that 274424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) // we keep the stream alive long enough to process any response or 275424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) // RST_STREAM frames. 276424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) if (locally_reset && !is_server_) { 277424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) AddZombieStream(stream_id); 278424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) return; 2792385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch } 280424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) 281424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) // This stream has been closed before the headers were decompressed. 282424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) // This might cause problems with head of line blocking of headers. 283424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) // If the peer sent headers which were lost but we now close the stream 284424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) // we will never be able to decompress headers for other streams. 285424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) // To deal with this, we keep track of streams which have been closed 286424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) // prematurely. If we ever receive data frames for this steam, then we 287424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) // know there actually has been a problem and we close the connection. 288424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) AddPrematurelyClosedStream(stream->id()); 2892385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch } 2902a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) closed_streams_.push_back(it->second); 29158537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) if (ContainsKey(zombie_streams_, stream->id())) { 29258537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) zombie_streams_.erase(stream->id()); 29358537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) } 2945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) stream_map_.erase(it); 295c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) stream->OnClose(); 2965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 298424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles)void QuicSession::AddZombieStream(QuicStreamId stream_id) { 299424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) if (zombie_streams_.size() == kMaxZombieStreams) { 300424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) QuicStreamId oldest_zombie_stream_id = zombie_streams_.begin()->first; 301424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) CloseZombieStream(oldest_zombie_stream_id); 302424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) // However, since the headers still have not been decompressed, we want to 303424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) // mark it a prematurely closed so that if we ever receive frames 304424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) // for this stream we can close the connection. 305424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) AddPrematurelyClosedStream(oldest_zombie_stream_id); 306424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) } 307424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) zombie_streams_.insert(make_pair(stream_id, true)); 308424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles)} 309424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) 310424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles)void QuicSession::CloseZombieStream(QuicStreamId stream_id) { 311424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) DCHECK(ContainsKey(zombie_streams_, stream_id)); 312424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) zombie_streams_.erase(stream_id); 313424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) ReliableQuicStream* stream = GetStream(stream_id); 314424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) if (!stream) { 315424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) return; 316424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) } 317424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) stream_map_.erase(stream_id); 318424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) stream->OnClose(); 319424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) closed_streams_.push_back(stream); 320424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles)} 321424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) 322424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles)void QuicSession::AddPrematurelyClosedStream(QuicStreamId stream_id) { 323424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) if (prematurely_closed_streams_.size() == 324424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) kMaxPrematurelyClosedStreamsTracked) { 325424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) prematurely_closed_streams_.erase(prematurely_closed_streams_.begin()); 326424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) } 327424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) prematurely_closed_streams_.insert(make_pair(stream_id, true)); 328424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles)} 329424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) 330c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)bool QuicSession::IsEncryptionEstablished() { 331c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) return GetCryptoStream()->encryption_established(); 3325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 334c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)bool QuicSession::IsCryptoHandshakeConfirmed() { 335c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) return GetCryptoStream()->handshake_confirmed(); 336c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)} 337c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 338c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)void QuicSession::OnCryptoHandshakeEvent(CryptoHandshakeEvent event) { 33990dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) switch (event) { 34090dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) // TODO(satyamshekhar): Move the logic of setting the encrypter/decrypter 34190dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) // to QuicSession since it is the glue. 34290dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) case ENCRYPTION_FIRST_ESTABLISHED: 34390dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) break; 34490dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 34590dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) case ENCRYPTION_REESTABLISHED: 34690dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) // Retransmit originally packets that were sent, since they can't be 34790dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) // decrypted by the peer. 34890dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) connection_->RetransmitUnackedPackets( 34990dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) QuicConnection::INITIAL_ENCRYPTION_ONLY); 35090dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) break; 35190dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 35290dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) case HANDSHAKE_CONFIRMED: 353868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) LOG_IF(DFATAL, !config_.negotiated()) << ENDPOINT 35490dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) << "Handshake confirmed without parameter negotiation."; 355868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) connection_->SetIdleNetworkTimeout( 35690dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) config_.idle_connection_state_lifetime()); 357868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) connection_->SetOverallConnectionTimeout(QuicTime::Delta::Infinite()); 35890dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) max_open_streams_ = config_.max_streams_per_connection(); 35990dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) break; 36090dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 36190dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) default: 362868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) LOG(ERROR) << ENDPOINT << "Got unknown handshake event: " << event; 363b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) } 3642a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 3652a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3663551c9c881056c480085172ff9840cab31610854Torne (Richard Coles)void QuicSession::OnCryptoHandshakeMessageSent( 3673551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) const CryptoHandshakeMessage& message) { 3683551c9c881056c480085172ff9840cab31610854Torne (Richard Coles)} 3693551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) 3703551c9c881056c480085172ff9840cab31610854Torne (Richard Coles)void QuicSession::OnCryptoHandshakeMessageReceived( 3713551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) const CryptoHandshakeMessage& message) { 3723551c9c881056c480085172ff9840cab31610854Torne (Richard Coles)} 3733551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) 37490dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)QuicConfig* QuicSession::config() { 37590dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) return &config_; 37690dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)} 37790dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 3785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void QuicSession::ActivateStream(ReliableQuicStream* stream) { 379868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) DLOG(INFO) << ENDPOINT << "num_streams: " << stream_map_.size() 380868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) << ". activating " << stream->id(); 381d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) DCHECK_EQ(stream_map_.count(stream->id()), 0u); 3825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) stream_map_[stream->id()] = stream; 3835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)QuicStreamId QuicSession::GetNextStreamId() { 3865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) QuicStreamId id = next_stream_id_; 3875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) next_stream_id_ += 2; 3885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return id; 3895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)ReliableQuicStream* QuicSession::GetStream(const QuicStreamId stream_id) { 3925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (stream_id == kCryptoStreamId) { 3935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return GetCryptoStream(); 3945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ReliableStreamMap::iterator it = stream_map_.find(stream_id); 3975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (it != stream_map_.end()) { 3985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return it->second; 3995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 4005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4012a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (IsClosedStream(stream_id)) { 4022a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return NULL; 4032a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 4042a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 4055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (stream_id % 2 == next_stream_id_ % 2) { 4065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // We've received a frame for a locally-created stream that is not 4075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // currently active. This is an error. 4085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) connection()->SendConnectionClose(QUIC_PACKET_FOR_NONEXISTENT_STREAM); 4095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return NULL; 4105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 4115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return GetIncomingReliableStream(stream_id); 4135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 4145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)ReliableQuicStream* QuicSession::GetIncomingReliableStream( 4165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) QuicStreamId stream_id) { 4175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (IsClosedStream(stream_id)) { 4185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return NULL; 4195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 4205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (goaway_sent_) { 4222a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // We've already sent a GoAway 423c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) SendRstStream(stream_id, QUIC_STREAM_PEER_GOING_AWAY); 4242a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return NULL; 4252a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 4262a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 4275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) implicitly_created_streams_.erase(stream_id); 4285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (stream_id > largest_peer_created_stream_id_) { 4295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // TODO(rch) add unit test for this 4305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (stream_id - largest_peer_created_stream_id_ > kMaxStreamIdDelta) { 4315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) connection()->SendConnectionClose(QUIC_INVALID_STREAM_ID); 4325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return NULL; 4335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 43458537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) if (largest_peer_created_stream_id_ == 0) { 43558537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) largest_peer_created_stream_id_= 1; 43658537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) } 43758537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) for (QuicStreamId id = largest_peer_created_stream_id_ + 2; 43858537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) id < stream_id; 43958537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) id += 2) { 44058537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) implicitly_created_streams_.insert(id); 4415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 4425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) largest_peer_created_stream_id_ = stream_id; 4435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 4445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ReliableQuicStream* stream = CreateIncomingReliableStream(stream_id); 4455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (stream == NULL) { 4465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return NULL; 4475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 4485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ActivateStream(stream); 4495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return stream; 4505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 4515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool QuicSession::IsClosedStream(QuicStreamId id) { 4535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK_NE(0u, id); 4545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (id == kCryptoStreamId) { 4555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return false; 4565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 457424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) if (ContainsKey(zombie_streams_, id)) { 458424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) return true; 459424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) } 460424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) if (ContainsKey(stream_map_, id)) { 4615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Stream is active 4625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return false; 4635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 4645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (id % 2 == next_stream_id_ % 2) { 4655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Locally created streams are strictly in-order. If the id is in the 4665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // range of created streams and it's not active, it must have been closed. 4675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return id < next_stream_id_; 4685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 4695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // For peer created streams, we also need to consider implicitly created 4705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // streams. 4712a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return id <= largest_peer_created_stream_id_ && 4722a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) implicitly_created_streams_.count(id) == 0; 4732a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 4745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4752a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)size_t QuicSession::GetNumOpenStreams() const { 476424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) return stream_map_.size() + implicitly_created_streams_.size() - 477424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) zombie_streams_.size(); 4785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 4795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 480d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles)void QuicSession::MarkWriteBlocked(QuicStreamId id, QuicPriority priority) { 481d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) if (id == kCryptoStreamId) { 482d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) DCHECK(!has_pending_handshake_); 483d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) has_pending_handshake_ = true; 484d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) // TODO(jar): Be sure to use the highest priority for the crypto stream, 485d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) // perhaps by adding a "special" priority for it that is higher than 486d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) // kHighestPriority. 487d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) priority = kHighestPriority; 488d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) } 489d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) write_blocked_streams_.PushBack(id, priority); 4902a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 4912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 492b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles)void QuicSession::MarkDecompressionBlocked(QuicHeaderId header_id, 493b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) QuicStreamId stream_id) { 494b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) decompression_blocked_streams_[header_id] = stream_id; 495b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles)} 496b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) 4977dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdochbool QuicSession::GetSSLInfo(SSLInfo* ssl_info) { 4987dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch NOTIMPLEMENTED(); 4997dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch return false; 5007dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch} 5017dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch 5022a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void QuicSession::PostProcessAfterData() { 5032a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) STLDeleteElements(&closed_streams_); 5042a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) closed_streams_.clear(); 5052a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 5062a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 5075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace net 508