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 23d57369da7c6519fef57db42085f7b42d4c8845c1Torne (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, 81d57369da7c6519fef57db42085f7b42d4c8845c1Torne (Richard Coles) const QuicConfig& config) 825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) : connection_(connection), 832a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) visitor_shim_(new VisitorShim(this)), 8490dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) config_(config), 85ba5b9a6411cb1792fd21f0a078d7a25cd1ceec16Ben Murdoch max_open_streams_(config_.max_streams_per_connection()), 86d57369da7c6519fef57db42085f7b42d4c8845c1Torne (Richard Coles) next_stream_id_(is_server() ? 2 : 3), 872a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) largest_peer_created_stream_id_(0), 88868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) error_(QUIC_NO_ERROR), 892a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) goaway_received_(false), 90d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) goaway_sent_(false), 91d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) has_pending_handshake_(false) { 92868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) 93868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) connection_->set_visitor(visitor_shim_.get()); 940f1bc08d4cfcc34181b0b5cbf065c40f687bf740Torne (Richard Coles) connection_->SetFromConfig(config_); 95ba5b9a6411cb1792fd21f0a078d7a25cd1ceec16Ben Murdoch if (connection_->connected()) { 96ba5b9a6411cb1792fd21f0a078d7a25cd1ceec16Ben Murdoch connection_->SetOverallConnectionTimeout( 97ba5b9a6411cb1792fd21f0a078d7a25cd1ceec16Ben Murdoch config_.max_time_before_crypto_handshake()); 98ba5b9a6411cb1792fd21f0a078d7a25cd1ceec16Ben Murdoch } 995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)QuicSession::~QuicSession() { 1022a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) STLDeleteElements(&closed_streams_); 1032a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) STLDeleteValues(&stream_map_); 1045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 106d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles)bool QuicSession::OnStreamFrames(const vector<QuicStreamFrame>& frames) { 1075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (size_t i = 0; i < frames.size(); ++i) { 1085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // TODO(rch) deal with the error case of stream id 0 1092385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch if (IsClosedStream(frames[i].stream_id)) { 1102385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch // If we get additional frames for a stream where we didn't process 1112385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch // headers, it's highly likely our compression context will end up 1122385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch // permanently out of sync with the peer's, so we give up and close the 1132385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch // connection. 1142385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch if (ContainsKey(prematurely_closed_streams_, frames[i].stream_id)) { 1152385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch connection()->SendConnectionClose( 1162385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch QUIC_STREAM_RST_BEFORE_HEADERS_DECOMPRESSED); 1172385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch return false; 1182385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch } 1192385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch continue; 1202385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch } 1215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ReliableQuicStream* stream = GetStream(frames[i].stream_id); 1235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (stream == NULL) return false; 1245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!stream->WillAcceptStreamFrame(frames[i])) return false; 1255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // TODO(alyssar) check against existing connection address: if changed, make 1275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // sure we update the connection. 1285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (size_t i = 0; i < frames.size(); ++i) { 131424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) QuicStreamId stream_id = frames[i].stream_id; 132424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) ReliableQuicStream* stream = GetStream(stream_id); 133424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) if (!stream) { 134424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) continue; 135424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) } 136424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) stream->OnStreamFrame(frames[i]); 137424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) 138d57369da7c6519fef57db42085f7b42d4c8845c1Torne (Richard Coles) // If the stream is a data stream had been prematurely closed, and the 139424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) // headers are now decompressed, then we are finally finished 140424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) // with this stream. 141424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) if (ContainsKey(zombie_streams_, stream_id) && 142d57369da7c6519fef57db42085f7b42d4c8845c1Torne (Richard Coles) static_cast<QuicDataStream*>(stream)->headers_decompressed()) { 143424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) CloseZombieStream(stream_id); 1445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 146b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) 147b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) while (!decompression_blocked_streams_.empty()) { 148b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) QuicHeaderId header_id = decompression_blocked_streams_.begin()->first; 149868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) if (header_id != decompressor_.current_header_id()) { 150868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) break; 151b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) } 152868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) QuicStreamId stream_id = decompression_blocked_streams_.begin()->second; 153868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) decompression_blocked_streams_.erase(header_id); 154d57369da7c6519fef57db42085f7b42d4c8845c1Torne (Richard Coles) QuicDataStream* stream = GetDataStream(stream_id); 155868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) if (!stream) { 156868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) connection()->SendConnectionClose( 157868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) QUIC_STREAM_RST_BEFORE_HEADERS_DECOMPRESSED); 158ba5b9a6411cb1792fd21f0a078d7a25cd1ceec16Ben Murdoch return false; 159868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) } 160868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) stream->OnDecompressorAvailable(); 161b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) } 1625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return true; 1635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void QuicSession::OnRstStream(const QuicRstStreamFrame& frame) { 166d57369da7c6519fef57db42085f7b42d4c8845c1Torne (Richard Coles) if (frame.stream_id == kCryptoStreamId) { 167d57369da7c6519fef57db42085f7b42d4c8845c1Torne (Richard Coles) connection()->SendConnectionCloseWithDetails( 168d57369da7c6519fef57db42085f7b42d4c8845c1Torne (Richard Coles) QUIC_INVALID_STREAM_ID, 169d57369da7c6519fef57db42085f7b42d4c8845c1Torne (Richard Coles) "Attempt to reset the crypto stream"); 170d57369da7c6519fef57db42085f7b42d4c8845c1Torne (Richard Coles) return; 171d57369da7c6519fef57db42085f7b42d4c8845c1Torne (Richard Coles) } 172d57369da7c6519fef57db42085f7b42d4c8845c1Torne (Richard Coles) QuicDataStream* stream = GetDataStream(frame.stream_id); 1735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!stream) { 1745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; // Errors are handled by GetStream. 1755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 176424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) if (ContainsKey(zombie_streams_, stream->id())) { 177424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) // If this was a zombie stream then we close it out now. 178424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) CloseZombieStream(stream->id()); 179424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) // However, since the headers still have not been decompressed, we want to 180424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) // mark it a prematurely closed so that if we ever receive frames 181424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) // for this stream we can close the connection. 182424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) DCHECK(!stream->headers_decompressed()); 183424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) AddPrematurelyClosedStream(frame.stream_id); 184424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) return; 185424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) } 186f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) if (stream->stream_bytes_read() > 0 && !stream->headers_decompressed()) { 187f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) connection()->SendConnectionClose( 188f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) QUIC_STREAM_RST_BEFORE_HEADERS_DECOMPRESSED); 189f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) } 1902a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) stream->OnStreamReset(frame.error_code); 1912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 1922a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1932a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void QuicSession::OnGoAway(const QuicGoAwayFrame& frame) { 1942a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK(frame.last_good_stream_id < next_stream_id_); 1952a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) goaway_received_ = true; 1965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1981e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles)void QuicSession::OnConnectionClosed(QuicErrorCode error, bool from_peer) { 199424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) DCHECK(!connection_->connected()); 200868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) if (error_ == QUIC_NO_ERROR) { 201868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) error_ = error; 202868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) } 203868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) 204a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) while (!stream_map_.empty()) { 205d57369da7c6519fef57db42085f7b42d4c8845c1Torne (Richard Coles) DataStreamMap::iterator it = stream_map_.begin(); 2065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) QuicStreamId id = it->first; 2071e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) it->second->OnConnectionClosed(error, from_peer); 2081e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) // The stream should call CloseStream as part of OnConnectionClosed. 2095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (stream_map_.find(id) != stream_map_.end()) { 2101e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) LOG(DFATAL) << ENDPOINT 2111e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) << "Stream failed to close under OnConnectionClosed"; 2125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CloseStream(id); 2135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)bool QuicSession::OnCanWrite() { 2182a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // We latch this here rather than doing a traditional loop, because streams 2192a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // may be modifying the list as we loop. 2203551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) int remaining_writes = write_blocked_streams_.NumBlockedStreams(); 2212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2222a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) while (!connection_->HasQueuedData() && 2232a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) remaining_writes > 0) { 2243551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) DCHECK(write_blocked_streams_.HasWriteBlockedStreams()); 225f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) if (!write_blocked_streams_.HasWriteBlockedStreams()) { 226d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) LOG(DFATAL) << "WriteBlockedStream is missing"; 227d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) connection_->CloseConnection(QUIC_INTERNAL_ERROR, false); 228d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) return true; // We have no write blocked streams. 2293551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) } 230f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) int index = write_blocked_streams_.GetHighestPriorityWriteBlockedList(); 231d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) QuicStreamId stream_id = write_blocked_streams_.PopFront(index); 232d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) if (stream_id == kCryptoStreamId) { 233d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) has_pending_handshake_ = false; // We just popped it. 234d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) } 235d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) ReliableQuicStream* stream = GetStream(stream_id); 2362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (stream != NULL) { 2372a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // If the stream can't write all bytes, it'll re-add itself to the blocked 2382a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // list. 2392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) stream->OnCanWrite(); 2402a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 2412a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) --remaining_writes; 2422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 2432a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2443551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) return !write_blocked_streams_.HasWriteBlockedStreams(); 2452a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 2462a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 247d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles)bool QuicSession::HasPendingHandshake() const { 248d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) return has_pending_handshake_; 249d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles)} 250d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) 251f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles)QuicConsumedData QuicSession::WritevData( 252f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) QuicStreamId id, 253f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) const struct iovec* iov, 254f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) int iov_count, 255f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) QuicStreamOffset offset, 256f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) bool fin, 257f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) QuicAckNotifier::DelegateInterface* ack_notifier_delegate) { 2581e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) IOVector data; 2591e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) data.AppendIovec(iov, iov_count); 260f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) return connection_->SendStreamData(id, data, offset, fin, 261f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) ack_notifier_delegate); 2625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void QuicSession::SendRstStream(QuicStreamId id, 265c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) QuicRstStreamErrorCode error) { 2662a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) connection_->SendRstStream(id, error); 267424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) CloseStreamInner(id, true); 2685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2702a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void QuicSession::SendGoAway(QuicErrorCode error_code, const string& reason) { 2712a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) goaway_sent_ = true; 2722a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) connection_->SendGoAway(error_code, largest_peer_created_stream_id_, reason); 2732a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 2742a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void QuicSession::CloseStream(QuicStreamId stream_id) { 276424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) CloseStreamInner(stream_id, false); 277424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles)} 278424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) 279424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles)void QuicSession::CloseStreamInner(QuicStreamId stream_id, 280424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) bool locally_reset) { 281f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) DVLOG(1) << ENDPOINT << "Closing stream " << stream_id; 2825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 283d57369da7c6519fef57db42085f7b42d4c8845c1Torne (Richard Coles) DataStreamMap::iterator it = stream_map_.find(stream_id); 2845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (it == stream_map_.end()) { 285f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) DVLOG(1) << ENDPOINT << "Stream is already closed: " << stream_id; 2865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 2875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 288d57369da7c6519fef57db42085f7b42d4c8845c1Torne (Richard Coles) QuicDataStream* stream = it->second; 289424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) if (connection_->connected() && !stream->headers_decompressed()) { 290424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) // If the stream is being closed locally (for example a client cancelling 291424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) // a request before receiving the response) then we need to make sure that 292424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) // we keep the stream alive long enough to process any response or 293424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) // RST_STREAM frames. 294d57369da7c6519fef57db42085f7b42d4c8845c1Torne (Richard Coles) if (locally_reset && !is_server()) { 295424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) AddZombieStream(stream_id); 296424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) return; 2972385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch } 298424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) 299424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) // This stream has been closed before the headers were decompressed. 300424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) // This might cause problems with head of line blocking of headers. 301424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) // If the peer sent headers which were lost but we now close the stream 302424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) // we will never be able to decompress headers for other streams. 303424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) // To deal with this, we keep track of streams which have been closed 304424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) // prematurely. If we ever receive data frames for this steam, then we 305424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) // know there actually has been a problem and we close the connection. 306424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) AddPrematurelyClosedStream(stream->id()); 3072385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch } 3082a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) closed_streams_.push_back(it->second); 30958537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) if (ContainsKey(zombie_streams_, stream->id())) { 31058537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) zombie_streams_.erase(stream->id()); 31158537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) } 3125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) stream_map_.erase(it); 313c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) stream->OnClose(); 3145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 316424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles)void QuicSession::AddZombieStream(QuicStreamId stream_id) { 317424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) if (zombie_streams_.size() == kMaxZombieStreams) { 318424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) QuicStreamId oldest_zombie_stream_id = zombie_streams_.begin()->first; 319424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) CloseZombieStream(oldest_zombie_stream_id); 320424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) // However, since the headers still have not been decompressed, we want to 321424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) // mark it a prematurely closed so that if we ever receive frames 322424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) // for this stream we can close the connection. 323424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) AddPrematurelyClosedStream(oldest_zombie_stream_id); 324424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) } 325424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) zombie_streams_.insert(make_pair(stream_id, true)); 326424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles)} 327424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) 328424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles)void QuicSession::CloseZombieStream(QuicStreamId stream_id) { 329424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) DCHECK(ContainsKey(zombie_streams_, stream_id)); 330424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) zombie_streams_.erase(stream_id); 331d57369da7c6519fef57db42085f7b42d4c8845c1Torne (Richard Coles) QuicDataStream* stream = GetDataStream(stream_id); 332424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) if (!stream) { 333424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) return; 334424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) } 335424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) stream_map_.erase(stream_id); 336424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) stream->OnClose(); 337424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) closed_streams_.push_back(stream); 338424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles)} 339424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) 340424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles)void QuicSession::AddPrematurelyClosedStream(QuicStreamId stream_id) { 341424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) if (prematurely_closed_streams_.size() == 342424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) kMaxPrematurelyClosedStreamsTracked) { 343424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) prematurely_closed_streams_.erase(prematurely_closed_streams_.begin()); 344424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) } 345424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) prematurely_closed_streams_.insert(make_pair(stream_id, true)); 346424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles)} 347424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) 348c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)bool QuicSession::IsEncryptionEstablished() { 349c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) return GetCryptoStream()->encryption_established(); 3505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 352c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)bool QuicSession::IsCryptoHandshakeConfirmed() { 353c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) return GetCryptoStream()->handshake_confirmed(); 354c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)} 355c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 3560f1bc08d4cfcc34181b0b5cbf065c40f687bf740Torne (Richard Coles)void QuicSession::OnConfigNegotiated() { 3570f1bc08d4cfcc34181b0b5cbf065c40f687bf740Torne (Richard Coles) connection_->SetFromConfig(config_); 3580f1bc08d4cfcc34181b0b5cbf065c40f687bf740Torne (Richard Coles)} 3590f1bc08d4cfcc34181b0b5cbf065c40f687bf740Torne (Richard Coles) 360c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)void QuicSession::OnCryptoHandshakeEvent(CryptoHandshakeEvent event) { 36190dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) switch (event) { 36290dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) // TODO(satyamshekhar): Move the logic of setting the encrypter/decrypter 36390dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) // to QuicSession since it is the glue. 36490dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) case ENCRYPTION_FIRST_ESTABLISHED: 36590dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) break; 36690dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 36790dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) case ENCRYPTION_REESTABLISHED: 36890dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) // Retransmit originally packets that were sent, since they can't be 36990dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) // decrypted by the peer. 370a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) connection_->RetransmitUnackedPackets(INITIAL_ENCRYPTION_ONLY); 37190dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) break; 37290dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 37390dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) case HANDSHAKE_CONFIRMED: 374868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) LOG_IF(DFATAL, !config_.negotiated()) << ENDPOINT 37590dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) << "Handshake confirmed without parameter negotiation."; 376868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) connection_->SetOverallConnectionTimeout(QuicTime::Delta::Infinite()); 37790dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) max_open_streams_ = config_.max_streams_per_connection(); 37890dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) break; 37990dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 38090dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) default: 381868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) LOG(ERROR) << ENDPOINT << "Got unknown handshake event: " << event; 382b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) } 3832a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 3842a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3853551c9c881056c480085172ff9840cab31610854Torne (Richard Coles)void QuicSession::OnCryptoHandshakeMessageSent( 3863551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) const CryptoHandshakeMessage& message) { 3873551c9c881056c480085172ff9840cab31610854Torne (Richard Coles)} 3883551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) 3893551c9c881056c480085172ff9840cab31610854Torne (Richard Coles)void QuicSession::OnCryptoHandshakeMessageReceived( 3903551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) const CryptoHandshakeMessage& message) { 3913551c9c881056c480085172ff9840cab31610854Torne (Richard Coles)} 3923551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) 39390dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)QuicConfig* QuicSession::config() { 39490dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) return &config_; 39590dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)} 39690dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 397d57369da7c6519fef57db42085f7b42d4c8845c1Torne (Richard Coles)void QuicSession::ActivateStream(QuicDataStream* stream) { 398f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) DVLOG(1) << ENDPOINT << "num_streams: " << stream_map_.size() 399868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) << ". activating " << stream->id(); 400d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) DCHECK_EQ(stream_map_.count(stream->id()), 0u); 4015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) stream_map_[stream->id()] = stream; 4025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 4035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)QuicStreamId QuicSession::GetNextStreamId() { 4055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) QuicStreamId id = next_stream_id_; 4065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) next_stream_id_ += 2; 4075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return id; 4085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 4095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)ReliableQuicStream* QuicSession::GetStream(const QuicStreamId stream_id) { 4115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (stream_id == kCryptoStreamId) { 4125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return GetCryptoStream(); 4135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 414d57369da7c6519fef57db42085f7b42d4c8845c1Torne (Richard Coles) return GetDataStream(stream_id); 415d57369da7c6519fef57db42085f7b42d4c8845c1Torne (Richard Coles)} 416d57369da7c6519fef57db42085f7b42d4c8845c1Torne (Richard Coles) 417d57369da7c6519fef57db42085f7b42d4c8845c1Torne (Richard Coles)QuicDataStream* QuicSession::GetDataStream(const QuicStreamId stream_id) { 418d57369da7c6519fef57db42085f7b42d4c8845c1Torne (Richard Coles) if (stream_id == kCryptoStreamId) { 419d57369da7c6519fef57db42085f7b42d4c8845c1Torne (Richard Coles) DLOG(FATAL) << "Attempt to call GetDataStream with the crypto stream id"; 420d57369da7c6519fef57db42085f7b42d4c8845c1Torne (Richard Coles) return NULL; 421d57369da7c6519fef57db42085f7b42d4c8845c1Torne (Richard Coles) } 4225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 423d57369da7c6519fef57db42085f7b42d4c8845c1Torne (Richard Coles) DataStreamMap::iterator it = stream_map_.find(stream_id); 4245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (it != stream_map_.end()) { 4255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return it->second; 4265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 4275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4282a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (IsClosedStream(stream_id)) { 4292a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return NULL; 4302a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 4312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 4325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (stream_id % 2 == next_stream_id_ % 2) { 4335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // We've received a frame for a locally-created stream that is not 4345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // currently active. This is an error. 4355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) connection()->SendConnectionClose(QUIC_PACKET_FOR_NONEXISTENT_STREAM); 4365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return NULL; 4375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 4385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return GetIncomingReliableStream(stream_id); 4405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 4415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 442d57369da7c6519fef57db42085f7b42d4c8845c1Torne (Richard Coles)QuicDataStream* QuicSession::GetIncomingReliableStream( 4435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) QuicStreamId stream_id) { 4445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (IsClosedStream(stream_id)) { 4455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return NULL; 4465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 4475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4482a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (goaway_sent_) { 4492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // We've already sent a GoAway 450c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) SendRstStream(stream_id, QUIC_STREAM_PEER_GOING_AWAY); 4512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return NULL; 4522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 4532a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 4545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) implicitly_created_streams_.erase(stream_id); 4555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (stream_id > largest_peer_created_stream_id_) { 4565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // TODO(rch) add unit test for this 4575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (stream_id - largest_peer_created_stream_id_ > kMaxStreamIdDelta) { 4585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) connection()->SendConnectionClose(QUIC_INVALID_STREAM_ID); 4595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return NULL; 4605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 46158537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) if (largest_peer_created_stream_id_ == 0) { 46258537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) largest_peer_created_stream_id_= 1; 46358537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) } 46458537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) for (QuicStreamId id = largest_peer_created_stream_id_ + 2; 46558537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) id < stream_id; 46658537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) id += 2) { 46758537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) implicitly_created_streams_.insert(id); 4685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 4695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) largest_peer_created_stream_id_ = stream_id; 4705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 471d57369da7c6519fef57db42085f7b42d4c8845c1Torne (Richard Coles) QuicDataStream* stream = CreateIncomingDataStream(stream_id); 4725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (stream == NULL) { 4735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return NULL; 4745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 4755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ActivateStream(stream); 4765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return stream; 4775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 4785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool QuicSession::IsClosedStream(QuicStreamId id) { 4805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK_NE(0u, id); 4815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (id == kCryptoStreamId) { 4825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return false; 4835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 484424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) if (ContainsKey(zombie_streams_, id)) { 485424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) return true; 486424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) } 487424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) if (ContainsKey(stream_map_, id)) { 4885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Stream is active 4895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return false; 4905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 4915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (id % 2 == next_stream_id_ % 2) { 4925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Locally created streams are strictly in-order. If the id is in the 4935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // range of created streams and it's not active, it must have been closed. 4945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return id < next_stream_id_; 4955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 4965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // For peer created streams, we also need to consider implicitly created 4975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // streams. 4982a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return id <= largest_peer_created_stream_id_ && 4992a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) implicitly_created_streams_.count(id) == 0; 5002a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 5015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5022a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)size_t QuicSession::GetNumOpenStreams() const { 503424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) return stream_map_.size() + implicitly_created_streams_.size() - 504424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) zombie_streams_.size(); 5055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 5065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 507d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles)void QuicSession::MarkWriteBlocked(QuicStreamId id, QuicPriority priority) { 508d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) if (id == kCryptoStreamId) { 509d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) DCHECK(!has_pending_handshake_); 510d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) has_pending_handshake_ = true; 511d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) // TODO(jar): Be sure to use the highest priority for the crypto stream, 512d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) // perhaps by adding a "special" priority for it that is higher than 513d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) // kHighestPriority. 514d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) priority = kHighestPriority; 515d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) } 516d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) write_blocked_streams_.PushBack(id, priority); 5172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 5182a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 519f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles)bool QuicSession::HasQueuedData() const { 520f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) return write_blocked_streams_.NumBlockedStreams() || 521f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) connection_->HasQueuedData(); 522f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles)} 523f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) 524b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles)void QuicSession::MarkDecompressionBlocked(QuicHeaderId header_id, 525b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) QuicStreamId stream_id) { 526b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) decompression_blocked_streams_[header_id] = stream_id; 527b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles)} 528b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) 5297dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdochbool QuicSession::GetSSLInfo(SSLInfo* ssl_info) { 5307dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch NOTIMPLEMENTED(); 5317dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch return false; 5327dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch} 5337dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch 5342a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void QuicSession::PostProcessAfterData() { 5352a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) STLDeleteElements(&closed_streams_); 5362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) closed_streams_.clear(); 5372a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 5382a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 5395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace net 540