quic_session.cc revision 7dbb3d5cf0c15f500944d211057644d6a2f37371
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) 75c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu#include "base/stl_util.h" 82a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "net/quic/crypto/proof_verifier.h" 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/quic/quic_connection.h" 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/ssl/ssl_info.h" 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using base::StringPiece; 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using base::hash_map; 142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)using base::hash_set; 1568043e1e95eeb07d5cae7aca370b26518b0867d6Torne (Richard Coles)using std::vector; 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 173551c9c881056c480085172ff9840cab31610854Torne (Richard Coles)namespace net { 181320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 192a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#define ENDPOINT (is_server_ ? "Server: " : " Client: ") 202a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 211e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles)// We want to make sure we delete any closed streams in a safe manner. 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// To avoid deleting a stream in mid-operation, we have a simple shim between 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// us and the stream, so we can delete any streams when we return from 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// processing. 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// 26a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)// We could just override the base methods, but this makes it easier to make 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// sure we don't miss any. 28cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles)class VisitorShim : public QuicConnectionVisitorInterface { 29cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) public: 30cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) explicit VisitorShim(QuicSession* session) : session_(session) {} 31f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 32f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) virtual bool OnPacket(const IPEndPoint& self_address, 335f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) const IPEndPoint& peer_address, 3423730a6e56a168d1879203e4b3819bb36e3d8f1fTorne (Richard Coles) const QuicPacketHeader& header, 355d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) const vector<QuicStreamFrame>& frame) OVERRIDE { 363551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) bool accepted = session_->OnPacket(self_address, peer_address, header, 370529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch frame); 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) session_->PostProcessAfterData(); 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return accepted; 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void OnRstStream(const QuicRstStreamFrame& frame) OVERRIDE { 422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) session_->OnRstStream(frame); 432a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) session_->PostProcessAfterData(); 442a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void OnGoAway(const QuicGoAwayFrame& frame) OVERRIDE { 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) session_->OnGoAway(frame); 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) session_->PostProcessAfterData(); 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void OnAck(const SequenceNumberSet& acked_packets) OVERRIDE { 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) session_->OnAck(acked_packets); 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) session_->PostProcessAfterData(); 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual bool OnCanWrite() OVERRIDE { 575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool rc = session_->OnCanWrite(); 585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) session_->PostProcessAfterData(); 595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return rc; 605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void ConnectionClose(QuicErrorCode error, bool from_peer) OVERRIDE { 635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) session_->ConnectionClose(error, from_peer); 644e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) // The session will go away, so don't bother with cleanup. 655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 682a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) QuicSession* session_; 692a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)}; 705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)QuicSession::QuicSession(QuicConnection* connection, 725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const QuicConfig& config, 735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool is_server) 742a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) : connection_(connection), 752a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) visitor_shim_(new VisitorShim(this)), 762a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) config_(config), 772a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) max_open_streams_(kDefaultMaxStreamsPerConnection), 785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) next_stream_id_(is_server ? 2 : 3), 792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) is_server_(is_server), 805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) largest_peer_created_stream_id_(0), 815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) error_(QUIC_NO_ERROR), 825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) goaway_received_(false), 835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) goaway_sent_(false) { 842a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) set_max_open_streams(config_.max_streams_per_connection()); 852a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) connection_->set_visitor(visitor_shim_.get()); 872a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) connection_->SetIdleNetworkTimeout(config_.idle_connection_state_lifetime()); 882a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) connection_->SetOverallConnectionTimeout( 892a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) config_.max_time_before_crypto_handshake()); 905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // TODO(satyamshekhar): Set congestion control and ICSL also. 915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)QuicSession::~QuicSession() { 945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) STLDeleteElements(&closed_streams_); 955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) STLDeleteValues(&stream_map_); 965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 972a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool QuicSession::OnPacket(const IPEndPoint& self_address, 992a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const IPEndPoint& peer_address, 1005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const QuicPacketHeader& header, 1012a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const vector<QuicStreamFrame>& frames) { 1022a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (header.public_header.guid != connection()->guid()) { 1032a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DLOG(INFO) << ENDPOINT << "Got packet header for invalid GUID: " 1042a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) << header.public_header.guid; 1052a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return false; 1062a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 1072a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) for (size_t i = 0; i < frames.size(); ++i) { 1082a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // TODO(rch) deal with the error case of stream id 0 1095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (IsClosedStream(frames[i].stream_id)) continue; 1102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1112a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ReliableQuicStream* stream = GetStream(frames[i].stream_id); 1122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (stream == NULL) return false; 1132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (!stream->WillAcceptStreamFrame(frames[i])) return false; 1142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1152a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // TODO(alyssar) check against existing connection address: if changed, make 1165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // sure we update the connection. 1172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 1182a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1192a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) for (size_t i = 0; i < frames.size(); ++i) { 1202a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ReliableQuicStream* stream = GetStream(frames[i].stream_id); 1212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (stream) { 1222a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) stream->OnStreamFrame(frames[i]); 1232a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 1242a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 1252a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) while (!decompression_blocked_streams_.empty()) { 127f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) QuicHeaderId header_id = decompression_blocked_streams_.begin()->first; 128c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch if (header_id != decompressor_.current_header_id()) { 129c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch break; 1302a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 1312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) QuicStreamId stream_id = decompression_blocked_streams_.begin()->second; 1325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) decompression_blocked_streams_.erase(header_id); 1337d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) ReliableQuicStream* stream = GetStream(stream_id); 1347d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) if (!stream) { 1357d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) connection()->SendConnectionClose( 1365d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) QUIC_STREAM_RST_BEFORE_HEADERS_DECOMPRESSED); 1375d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) } 138f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) stream->OnDecompressorAvailable(); 139c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch } 140c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch return true; 1412a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 1422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1432a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void QuicSession::OnRstStream(const QuicRstStreamFrame& frame) { 1445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ReliableQuicStream* stream = GetStream(frame.stream_id); 1455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!stream) { 1465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; // Errors are handled by GetStream. 1475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) stream->OnStreamReset(frame.error_code); 1495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void QuicSession::OnGoAway(const QuicGoAwayFrame& frame) { 1525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(frame.last_good_stream_id < next_stream_id_); 1535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) goaway_received_ = true; 1545d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)} 1555d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 156b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles)void QuicSession::ConnectionClose(QuicErrorCode error, bool from_peer) { 1575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (error_ == QUIC_NO_ERROR) { 1585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) error_ = error; 1595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) while (stream_map_.size() != 0) { 1625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ReliableStreamMap::iterator it = stream_map_.begin(); 1635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) QuicStreamId id = it->first; 1645d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) it->second->ConnectionClose(error, from_peer); 1655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // The stream should call CloseStream as part of ConnectionClose. 1665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (stream_map_.find(id) != stream_map_.end()) { 1672a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) LOG(DFATAL) << ENDPOINT << "Stream failed to close under ConnectionClose"; 1685d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) CloseStream(id); 1695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1705d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) } 1715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1722a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1732a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)bool QuicSession::OnCanWrite() { 1742a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // We latch this here rather than doing a traditional loop, because streams 1752a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // may be modifying the list as we loop. 1762a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) int remaining_writes = write_blocked_streams_.NumObjects(); 1772a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1782a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) while (!connection_->HasQueuedData() && 1792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) remaining_writes > 0) { 1802a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK(!write_blocked_streams_.IsEmpty()); 181c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch ReliableQuicStream* stream = 1822a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) GetStream(write_blocked_streams_.GetNextBlockedObject()); 1832a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (stream != NULL) { 1842a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // If the stream can't write all bytes, it'll re-add itself to the blocked 1855d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) // list. 1862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) stream->OnCanWrite(); 187c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch } 1885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) --remaining_writes; 1895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return write_blocked_streams_.IsEmpty(); 1925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1932a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1942a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)QuicConsumedData QuicSession::WriteData(QuicStreamId id, 1952a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) StringPiece data, 1962a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) QuicStreamOffset offset, 1975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool fin) { 1985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return connection_->SendStreamData(id, data, offset, fin); 1995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void QuicSession::SendRstStream(QuicStreamId id, 2025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) QuicRstStreamErrorCode error) { 2035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) connection_->SendRstStream(id, error); 2045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CloseStream(id); 2055d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)} 2065d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 2075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void QuicSession::SendGoAway(QuicErrorCode error_code, const string& reason) { 2085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) goaway_sent_ = true; 2095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) connection_->SendGoAway(error_code, largest_peer_created_stream_id_, reason); 2105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 211a3f7b4e666c476898878fa745f637129375cd889Ben Murdoch 212a3f7b4e666c476898878fa745f637129375cd889Ben Murdochvoid QuicSession::CloseStream(QuicStreamId stream_id) { 213a3f7b4e666c476898878fa745f637129375cd889Ben Murdoch DLOG(INFO) << ENDPOINT << "Closing stream " << stream_id; 214a3f7b4e666c476898878fa745f637129375cd889Ben Murdoch 2155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ReliableStreamMap::iterator it = stream_map_.find(stream_id); 2165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (it == stream_map_.end()) { 2175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DLOG(INFO) << ENDPOINT << "Stream is already closed: " << stream_id; 2185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 2195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ReliableQuicStream* stream = it->second; 221a3f7b4e666c476898878fa745f637129375cd889Ben Murdoch closed_streams_.push_back(it->second); 222a3f7b4e666c476898878fa745f637129375cd889Ben Murdoch stream_map_.erase(it); 223a3f7b4e666c476898878fa745f637129375cd889Ben Murdoch stream->OnClose(); 224a3f7b4e666c476898878fa745f637129375cd889Ben Murdoch} 225a3f7b4e666c476898878fa745f637129375cd889Ben Murdoch 226a3f7b4e666c476898878fa745f637129375cd889Ben Murdochbool QuicSession::IsEncryptionEstablished() { 227a3f7b4e666c476898878fa745f637129375cd889Ben Murdoch return GetCryptoStream()->encryption_established(); 228a3f7b4e666c476898878fa745f637129375cd889Ben Murdoch} 229a3f7b4e666c476898878fa745f637129375cd889Ben Murdoch 2305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool QuicSession::IsCryptoHandshakeConfirmed() { 2315d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) return GetCryptoStream()->handshake_confirmed(); 2325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void QuicSession::OnCryptoHandshakeEvent(CryptoHandshakeEvent event) { 2355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) switch (event) { 2365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // TODO(satyamshekhar): Move the logic of setting the encrypter/decrypter 2375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // to QuicSession since it is the glue. 2385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) case ENCRYPTION_FIRST_ESTABLISHED: 2395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) break; 2405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) case ENCRYPTION_REESTABLISHED: 24268043e1e95eeb07d5cae7aca370b26518b0867d6Torne (Richard Coles) // Retransmit originally packets that were sent, since they can't be 2435d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) // decrypted by the peer. 2445d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) connection_->RetransmitUnackedPackets( 2455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) QuicConnection::INITIAL_ENCRYPTION_ONLY); 2465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) break; 2475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2485d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) case HANDSHAKE_CONFIRMED: 2495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) LOG_IF(DFATAL, !config_.negotiated()) << ENDPOINT 2505d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) << "Handshake confirmed without parameter negotiation."; 2515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) connection_->SetIdleNetworkTimeout( 2522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) config_.idle_connection_state_lifetime()); 2532a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) connection_->SetOverallConnectionTimeout(QuicTime::Delta::Infinite()); 2542a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) max_open_streams_ = config_.max_streams_per_connection(); 2552a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) break; 256c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 2572a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) default: 2582a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) LOG(ERROR) << ENDPOINT << "Got unknown handshake event: " << event; 2592a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 2602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 261c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 2625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)QuicConfig* QuicSession::config() { 2635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return &config_; 2647d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles)} 2657d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) 2667d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles)void QuicSession::ActivateStream(ReliableQuicStream* stream) { 2672a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DLOG(INFO) << ENDPOINT << "num_streams: " << stream_map_.size() 2685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) << ". activating " << stream->id(); 2695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(stream_map_.count(stream->id()) == 0); 270f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) stream_map_[stream->id()] = stream; 2715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)QuicStreamId QuicSession::GetNextStreamId() { 2745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) QuicStreamId id = next_stream_id_; 2755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) next_stream_id_ += 2; 2765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return id; 2775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)ReliableQuicStream* QuicSession::GetStream(const QuicStreamId stream_id) { 2805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (stream_id == kCryptoStreamId) { 2815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return GetCryptoStream(); 2825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ReliableStreamMap::iterator it = stream_map_.find(stream_id); 2855f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) if (it != stream_map_.end()) { 2865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return it->second; 2875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (IsClosedStream(stream_id)) { 2905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return NULL; 2915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (stream_id % 2 == next_stream_id_ % 2) { 2945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // We've received a frame for a locally-created stream that is not 2955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // currently active. This is an error. 2965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) connection()->SendConnectionClose(QUIC_PACKET_FOR_NONEXISTENT_STREAM); 2975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return NULL; 2985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 299c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 3005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return GetIncomingReliableStream(stream_id); 3012a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 302a02191e04bc25c4935f804f2c080ae28663d096dBen Murdoch 303c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen MurdochReliableQuicStream* QuicSession::GetIncomingReliableStream( 304c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch QuicStreamId stream_id) { 3055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (IsClosedStream(stream_id)) { 3065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return NULL; 3075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 308a02191e04bc25c4935f804f2c080ae28663d096dBen Murdoch 309c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch if (goaway_sent_) { 310c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch // We've already sent a GoAway 311c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch SendRstStream(stream_id, QUIC_STREAM_PEER_GOING_AWAY); 3125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return NULL; 313c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) } 3145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3152a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) implicitly_created_streams_.erase(stream_id); 316a02191e04bc25c4935f804f2c080ae28663d096dBen Murdoch if (stream_id > largest_peer_created_stream_id_) { 317c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch // TODO(rch) add unit test for this 318c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch if (stream_id - largest_peer_created_stream_id_ > kMaxStreamIdDelta) { 3195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) connection()->SendConnectionClose(QUIC_INVALID_STREAM_ID); 320a02191e04bc25c4935f804f2c080ae28663d096dBen Murdoch return NULL; 321a02191e04bc25c4935f804f2c080ae28663d096dBen Murdoch } 3225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (largest_peer_created_stream_id_ != 0) { 3235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (QuicStreamId id = largest_peer_created_stream_id_ + 2; 324a02191e04bc25c4935f804f2c080ae28663d096dBen Murdoch id < stream_id; 325c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch id += 2) { 326c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch implicitly_created_streams_.insert(id); 327c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch } 328a02191e04bc25c4935f804f2c080ae28663d096dBen Murdoch } 329c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch largest_peer_created_stream_id_ = stream_id; 3305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ReliableQuicStream* stream = CreateIncomingReliableStream(stream_id); 3325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (stream == NULL) { 333c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) return NULL; 3345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ActivateStream(stream); 3362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return stream; 3372a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 3382a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)bool QuicSession::IsClosedStream(QuicStreamId id) { 3402a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK_NE(0u, id); 3412a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (id == kCryptoStreamId) { 3422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return false; 3432a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 34490dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) if (stream_map_.count(id) != 0) { 345c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Stream is active 3462a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return false; 347c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch } 348a02191e04bc25c4935f804f2c080ae28663d096dBen Murdoch if (id % 2 == next_stream_id_ % 2) { 349c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch // Locally created streams are strictly in-order. If the id is in the 350c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch // range of created streams and it's not active, it must have been closed. 3512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return id < next_stream_id_; 352a02191e04bc25c4935f804f2c080ae28663d096dBen Murdoch } 353a02191e04bc25c4935f804f2c080ae28663d096dBen Murdoch // For peer created streams, we also need to consider implicitly created 3542a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // streams. 3552a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return id <= largest_peer_created_stream_id_ && 356a02191e04bc25c4935f804f2c080ae28663d096dBen Murdoch implicitly_created_streams_.count(id) == 0; 357c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch} 358c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 359c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdochsize_t QuicSession::GetNumOpenStreams() const { 3602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return stream_map_.size() + implicitly_created_streams_.size(); 3612a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 3622a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3632a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void QuicSession::MarkWriteBlocked(QuicStreamId id) { 3642a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) write_blocked_streams_.AddBlockedObject(id); 3652a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 3662a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 36790dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)void QuicSession::MarkDecompressionBlocked(QuicHeaderId header_id, 368c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) QuicStreamId stream_id) { 369a02191e04bc25c4935f804f2c080ae28663d096dBen Murdoch decompression_blocked_streams_[header_id] = stream_id; 370c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch} 371c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 3722a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)bool QuicSession::GetSSLInfo(SSLInfo* ssl_info) { 3732a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) NOTIMPLEMENTED(); 3742a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return false; 375a02191e04bc25c4935f804f2c080ae28663d096dBen Murdoch} 376c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 377c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdochvoid QuicSession::PostProcessAfterData() { 378c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch STLDeleteElements(&closed_streams_); 3792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) closed_streams_.clear(); 3802a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 381bbcdd45c55eb7c4641ab97aef9889b0fc828e7d3Ben Murdoch 382bbcdd45c55eb7c4641ab97aef9889b0fc828e7d3Ben Murdoch} // namespace net 383bbcdd45c55eb7c4641ab97aef9889b0fc828e7d3Ben Murdoch