quic_session.cc revision 2385ea399aae016c0806a4f9ef3c9cfe3d2a39df
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; 212385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch 22868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)#define ENDPOINT (is_server_ ? "Server: " : " Client: ") 23868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) 242a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// We want to make sure we delete any closed streams in a safe manner. 252a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// To avoid deleting a stream in mid-operation, we have a simple shim between 262a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// us and the stream, so we can delete any streams when we return from 272a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// processing. 282a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// 292a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// We could just override the base methods, but this makes it easier to make 302a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// sure we don't miss any. 312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)class VisitorShim : public QuicConnectionVisitorInterface { 322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) public: 332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) explicit VisitorShim(QuicSession* session) : session_(session) {} 342a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 352a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) virtual bool OnPacket(const IPEndPoint& self_address, 362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const IPEndPoint& peer_address, 372a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const QuicPacketHeader& header, 382a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const vector<QuicStreamFrame>& frame) OVERRIDE { 392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) bool accepted = session_->OnPacket(self_address, peer_address, header, 402a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) frame); 412a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) session_->PostProcessAfterData(); 422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return accepted; 432a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 442a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) virtual void OnRstStream(const QuicRstStreamFrame& frame) OVERRIDE { 452a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) session_->OnRstStream(frame); 462a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) session_->PostProcessAfterData(); 472a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 482a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) virtual void OnGoAway(const QuicGoAwayFrame& frame) OVERRIDE { 502a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) session_->OnGoAway(frame); 512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) session_->PostProcessAfterData(); 522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 532a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 542a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) virtual void OnAck(const SequenceNumberSet& acked_packets) OVERRIDE { 552a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) session_->OnAck(acked_packets); 562a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) session_->PostProcessAfterData(); 572a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 582a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 592a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) virtual bool OnCanWrite() OVERRIDE { 602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) bool rc = session_->OnCanWrite(); 612a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) session_->PostProcessAfterData(); 622a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return rc; 632a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 642a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 652a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) virtual void ConnectionClose(QuicErrorCode error, bool from_peer) OVERRIDE { 662a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) session_->ConnectionClose(error, from_peer); 672a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // The session will go away, so don't bother with cleanup. 682a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 692a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 702a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) private: 712a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) QuicSession* session_; 722a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)}; 732a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 7490dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)QuicSession::QuicSession(QuicConnection* connection, 7590dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) const QuicConfig& config, 7690dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) bool is_server) 775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) : connection_(connection), 782a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) visitor_shim_(new VisitorShim(this)), 7990dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) config_(config), 805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) max_open_streams_(kDefaultMaxStreamsPerConnection), 815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) next_stream_id_(is_server ? 2 : 3), 825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) is_server_(is_server), 832a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) largest_peer_created_stream_id_(0), 84868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) error_(QUIC_NO_ERROR), 852a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) goaway_received_(false), 862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) goaway_sent_(false) { 87868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) set_max_open_streams(config_.max_streams_per_connection()); 88868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) 89868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) connection_->set_visitor(visitor_shim_.get()); 90868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) connection_->SetIdleNetworkTimeout(config_.idle_connection_state_lifetime()); 91868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) connection_->SetOverallConnectionTimeout( 92868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) config_.max_time_before_crypto_handshake()); 93868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) // TODO(satyamshekhar): Set congestion control and ICSL also. 945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)QuicSession::~QuicSession() { 972a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) STLDeleteElements(&closed_streams_); 982a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) STLDeleteValues(&stream_map_); 995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool QuicSession::OnPacket(const IPEndPoint& self_address, 1025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const IPEndPoint& peer_address, 1035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const QuicPacketHeader& header, 1045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const vector<QuicStreamFrame>& frames) { 1052a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (header.public_header.guid != connection()->guid()) { 106868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) DLOG(INFO) << ENDPOINT << "Got packet header for invalid GUID: " 1072a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) << header.public_header.guid; 1085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return false; 1095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1102385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch 1115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (size_t i = 0; i < frames.size(); ++i) { 1125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // TODO(rch) deal with the error case of stream id 0 1132385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch if (IsClosedStream(frames[i].stream_id)) { 1142385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch // If we get additional frames for a stream where we didn't process 1152385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch // headers, it's highly likely our compression context will end up 1162385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch // permanently out of sync with the peer's, so we give up and close the 1172385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch // connection. 1182385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch if (ContainsKey(prematurely_closed_streams_, frames[i].stream_id)) { 1192385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch connection()->SendConnectionClose( 1202385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch QUIC_STREAM_RST_BEFORE_HEADERS_DECOMPRESSED); 1212385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch return false; 1222385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch } 1232385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch continue; 1242385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch } 1255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ReliableQuicStream* stream = GetStream(frames[i].stream_id); 1275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (stream == NULL) return false; 1285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!stream->WillAcceptStreamFrame(frames[i])) return false; 1295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // TODO(alyssar) check against existing connection address: if changed, make 1315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // sure we update the connection. 1325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (size_t i = 0; i < frames.size(); ++i) { 1355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ReliableQuicStream* stream = GetStream(frames[i].stream_id); 1365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (stream) { 1375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) stream->OnStreamFrame(frames[i]); 1385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 140b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) 141b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) while (!decompression_blocked_streams_.empty()) { 142b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) QuicHeaderId header_id = decompression_blocked_streams_.begin()->first; 143868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) if (header_id != decompressor_.current_header_id()) { 144868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) break; 145b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) } 146868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) QuicStreamId stream_id = decompression_blocked_streams_.begin()->second; 147868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) decompression_blocked_streams_.erase(header_id); 148868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) ReliableQuicStream* stream = GetStream(stream_id); 149868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) if (!stream) { 150868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) connection()->SendConnectionClose( 151868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) QUIC_STREAM_RST_BEFORE_HEADERS_DECOMPRESSED); 152868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) } 153868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) stream->OnDecompressorAvailable(); 154b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) } 1555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return true; 1565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void QuicSession::OnRstStream(const QuicRstStreamFrame& frame) { 1595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ReliableQuicStream* stream = GetStream(frame.stream_id); 1605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!stream) { 1615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; // Errors are handled by GetStream. 1625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1632a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) stream->OnStreamReset(frame.error_code); 1642a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 1652a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1662a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void QuicSession::OnGoAway(const QuicGoAwayFrame& frame) { 1672a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK(frame.last_good_stream_id < next_stream_id_); 1682a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) goaway_received_ = true; 1695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void QuicSession::ConnectionClose(QuicErrorCode error, bool from_peer) { 172868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) if (error_ == QUIC_NO_ERROR) { 173868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) error_ = error; 174868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) } 175868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) 1765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) while (stream_map_.size() != 0) { 1775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ReliableStreamMap::iterator it = stream_map_.begin(); 1785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) QuicStreamId id = it->first; 1795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) it->second->ConnectionClose(error, from_peer); 1805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // The stream should call CloseStream as part of ConnectionClose. 1815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (stream_map_.find(id) != stream_map_.end()) { 182868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) LOG(DFATAL) << ENDPOINT << "Stream failed to close under ConnectionClose"; 1835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CloseStream(id); 1845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1882a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)bool QuicSession::OnCanWrite() { 1892a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // We latch this here rather than doing a traditional loop, because streams 1902a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // may be modifying the list as we loop. 1912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) int remaining_writes = write_blocked_streams_.NumObjects(); 1922a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1932a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) while (!connection_->HasQueuedData() && 1942a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) remaining_writes > 0) { 1952a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK(!write_blocked_streams_.IsEmpty()); 1962a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ReliableQuicStream* stream = 1972a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) GetStream(write_blocked_streams_.GetNextBlockedObject()); 1982a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (stream != NULL) { 1992a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // If the stream can't write all bytes, it'll re-add itself to the blocked 2002a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // list. 2012a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) stream->OnCanWrite(); 2022a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 2032a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) --remaining_writes; 2042a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 2052a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2062a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return write_blocked_streams_.IsEmpty(); 2072a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 2082a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2092a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)QuicConsumedData QuicSession::WriteData(QuicStreamId id, 2102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) StringPiece data, 2112a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) QuicStreamOffset offset, 2122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) bool fin) { 2132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return connection_->SendStreamData(id, data, offset, fin); 2145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void QuicSession::SendRstStream(QuicStreamId id, 217c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) QuicRstStreamErrorCode error) { 2182a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) connection_->SendRstStream(id, error); 2195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CloseStream(id); 2205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2222a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void QuicSession::SendGoAway(QuicErrorCode error_code, const string& reason) { 2232a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) goaway_sent_ = true; 2242a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) connection_->SendGoAway(error_code, largest_peer_created_stream_id_, reason); 2252a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 2262a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void QuicSession::CloseStream(QuicStreamId stream_id) { 228868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) DLOG(INFO) << ENDPOINT << "Closing stream " << stream_id; 2295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ReliableStreamMap::iterator it = stream_map_.find(stream_id); 2315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (it == stream_map_.end()) { 232868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) DLOG(INFO) << ENDPOINT << "Stream is already closed: " << stream_id; 2335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 2345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 235c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) ReliableQuicStream* stream = it->second; 2362385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch if (!stream->headers_decompressed()) { 2372385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch if (prematurely_closed_streams_.size() == 2382385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch kMaxPrematurelyClosedStreamsTracked) { 2392385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch prematurely_closed_streams_.erase(prematurely_closed_streams_.begin()); 2402385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch } 2412385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch prematurely_closed_streams_.insert(make_pair(stream->id(), true)); 2422385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch } 2432a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) closed_streams_.push_back(it->second); 2445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) stream_map_.erase(it); 245c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) stream->OnClose(); 2465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 248c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)bool QuicSession::IsEncryptionEstablished() { 249c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) return GetCryptoStream()->encryption_established(); 2505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 252c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)bool QuicSession::IsCryptoHandshakeConfirmed() { 253c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) return GetCryptoStream()->handshake_confirmed(); 254c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)} 255c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 256c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)void QuicSession::OnCryptoHandshakeEvent(CryptoHandshakeEvent event) { 25790dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) switch (event) { 25890dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) // TODO(satyamshekhar): Move the logic of setting the encrypter/decrypter 25990dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) // to QuicSession since it is the glue. 26090dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) case ENCRYPTION_FIRST_ESTABLISHED: 26190dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) break; 26290dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 26390dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) case ENCRYPTION_REESTABLISHED: 26490dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) // Retransmit originally packets that were sent, since they can't be 26590dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) // decrypted by the peer. 26690dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) connection_->RetransmitUnackedPackets( 26790dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) QuicConnection::INITIAL_ENCRYPTION_ONLY); 26890dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) break; 26990dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 27090dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) case HANDSHAKE_CONFIRMED: 271868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) LOG_IF(DFATAL, !config_.negotiated()) << ENDPOINT 27290dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) << "Handshake confirmed without parameter negotiation."; 273868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) connection_->SetIdleNetworkTimeout( 27490dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) config_.idle_connection_state_lifetime()); 275868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) connection_->SetOverallConnectionTimeout(QuicTime::Delta::Infinite()); 27690dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) max_open_streams_ = config_.max_streams_per_connection(); 27790dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) break; 27890dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 27990dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) default: 280868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) LOG(ERROR) << ENDPOINT << "Got unknown handshake event: " << event; 281b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) } 2822a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 2832a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 28490dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)QuicConfig* QuicSession::config() { 28590dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) return &config_; 28690dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)} 28790dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 2885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void QuicSession::ActivateStream(ReliableQuicStream* stream) { 289868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) DLOG(INFO) << ENDPOINT << "num_streams: " << stream_map_.size() 290868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) << ". activating " << stream->id(); 2915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(stream_map_.count(stream->id()) == 0); 2925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) stream_map_[stream->id()] = stream; 2935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)QuicStreamId QuicSession::GetNextStreamId() { 2965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) QuicStreamId id = next_stream_id_; 2975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) next_stream_id_ += 2; 2985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return id; 2995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)ReliableQuicStream* QuicSession::GetStream(const QuicStreamId stream_id) { 3025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (stream_id == kCryptoStreamId) { 3035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return GetCryptoStream(); 3045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ReliableStreamMap::iterator it = stream_map_.find(stream_id); 3075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (it != stream_map_.end()) { 3085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return it->second; 3095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3112a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (IsClosedStream(stream_id)) { 3122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return NULL; 3132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 3142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (stream_id % 2 == next_stream_id_ % 2) { 3165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // We've received a frame for a locally-created stream that is not 3175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // currently active. This is an error. 3185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) connection()->SendConnectionClose(QUIC_PACKET_FOR_NONEXISTENT_STREAM); 3195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return NULL; 3205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return GetIncomingReliableStream(stream_id); 3235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)ReliableQuicStream* QuicSession::GetIncomingReliableStream( 3265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) QuicStreamId stream_id) { 3275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (IsClosedStream(stream_id)) { 3285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return NULL; 3295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (goaway_sent_) { 3322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // We've already sent a GoAway 333c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) SendRstStream(stream_id, QUIC_STREAM_PEER_GOING_AWAY); 3342a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return NULL; 3352a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 3362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) implicitly_created_streams_.erase(stream_id); 3385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (stream_id > largest_peer_created_stream_id_) { 3395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // TODO(rch) add unit test for this 3405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (stream_id - largest_peer_created_stream_id_ > kMaxStreamIdDelta) { 3415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) connection()->SendConnectionClose(QUIC_INVALID_STREAM_ID); 3425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return NULL; 3435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (largest_peer_created_stream_id_ != 0) { 3455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (QuicStreamId id = largest_peer_created_stream_id_ + 2; 3465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) id < stream_id; 3475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) id += 2) { 3485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) implicitly_created_streams_.insert(id); 3495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) largest_peer_created_stream_id_ = stream_id; 3525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ReliableQuicStream* stream = CreateIncomingReliableStream(stream_id); 3545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (stream == NULL) { 3555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return NULL; 3565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ActivateStream(stream); 3585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return stream; 3595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool QuicSession::IsClosedStream(QuicStreamId id) { 3625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK_NE(0u, id); 3635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (id == kCryptoStreamId) { 3645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return false; 3655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (stream_map_.count(id) != 0) { 3675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Stream is active 3685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return false; 3695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (id % 2 == next_stream_id_ % 2) { 3715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Locally created streams are strictly in-order. If the id is in the 3725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // range of created streams and it's not active, it must have been closed. 3735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return id < next_stream_id_; 3745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // For peer created streams, we also need to consider implicitly created 3765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // streams. 3772a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return id <= largest_peer_created_stream_id_ && 3782a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) implicitly_created_streams_.count(id) == 0; 3792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 3805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3812a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)size_t QuicSession::GetNumOpenStreams() const { 3825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return stream_map_.size() + implicitly_created_streams_.size(); 3835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3852a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void QuicSession::MarkWriteBlocked(QuicStreamId id) { 3862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) write_blocked_streams_.AddBlockedObject(id); 3872a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 3882a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 389b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles)void QuicSession::MarkDecompressionBlocked(QuicHeaderId header_id, 390b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) QuicStreamId stream_id) { 391b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) decompression_blocked_streams_[header_id] = stream_id; 392b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles)} 393b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) 3947dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdochbool QuicSession::GetSSLInfo(SSLInfo* ssl_info) { 3957dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch NOTIMPLEMENTED(); 3967dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch return false; 3977dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch} 3987dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch 3992a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void QuicSession::PostProcessAfterData() { 4002a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) STLDeleteElements(&closed_streams_); 4012a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) closed_streams_.clear(); 4022a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 4032a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 4045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace net 405