1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. 2 // Use of this source code is governed by a BSD-style license that can be 3 // found in the LICENSE file. 4 // 5 // The entity that handles framing writes for a Quic client or server. 6 // Each QuicSession will have a connection associated with it. 7 // 8 // On the server side, the Dispatcher handles the raw reads, and hands off 9 // packets via ProcessUdpPacket for framing and processing. 10 // 11 // On the client side, the Connection handles the raw reads, as well as the 12 // processing. 13 // 14 // Note: this class is not thread-safe. 15 16 #ifndef NET_QUIC_QUIC_CONNECTION_H_ 17 #define NET_QUIC_QUIC_CONNECTION_H_ 18 19 #include <stddef.h> 20 #include <deque> 21 #include <list> 22 #include <map> 23 #include <queue> 24 #include <string> 25 #include <vector> 26 27 #include "base/logging.h" 28 #include "net/base/iovec.h" 29 #include "net/base/ip_endpoint.h" 30 #include "net/quic/iovector.h" 31 #include "net/quic/quic_ack_notifier.h" 32 #include "net/quic/quic_ack_notifier_manager.h" 33 #include "net/quic/quic_alarm.h" 34 #include "net/quic/quic_blocked_writer_interface.h" 35 #include "net/quic/quic_connection_stats.h" 36 #include "net/quic/quic_packet_creator.h" 37 #include "net/quic/quic_packet_generator.h" 38 #include "net/quic/quic_packet_writer.h" 39 #include "net/quic/quic_protocol.h" 40 #include "net/quic/quic_received_packet_manager.h" 41 #include "net/quic/quic_sent_entropy_manager.h" 42 #include "net/quic/quic_sent_packet_manager.h" 43 #include "net/quic/quic_time.h" 44 #include "net/quic/quic_types.h" 45 46 namespace net { 47 48 class QuicClock; 49 class QuicConfig; 50 class QuicConnection; 51 class QuicDecrypter; 52 class QuicEncrypter; 53 class QuicFecGroup; 54 class QuicRandom; 55 56 namespace test { 57 class PacketSavingConnection; 58 class QuicConnectionPeer; 59 } // namespace test 60 61 // Class that receives callbacks from the connection when frames are received 62 // and when other interesting events happen. 63 class NET_EXPORT_PRIVATE QuicConnectionVisitorInterface { 64 public: ~QuicConnectionVisitorInterface()65 virtual ~QuicConnectionVisitorInterface() {} 66 67 // A simple visitor interface for dealing with data frames. 68 virtual void OnStreamFrames(const std::vector<QuicStreamFrame>& frames) = 0; 69 70 // The session should process all WINDOW_UPDATE frames, adjusting both stream 71 // and connection level flow control windows. 72 virtual void OnWindowUpdateFrames( 73 const std::vector<QuicWindowUpdateFrame>& frames) = 0; 74 75 // BLOCKED frames tell us that the peer believes it is flow control blocked on 76 // a specified stream. If the session at this end disagrees, something has 77 // gone wrong with our flow control accounting. 78 virtual void OnBlockedFrames(const std::vector<QuicBlockedFrame>& frames) = 0; 79 80 // Called when the stream is reset by the peer. 81 virtual void OnRstStream(const QuicRstStreamFrame& frame) = 0; 82 83 // Called when the connection is going away according to the peer. 84 virtual void OnGoAway(const QuicGoAwayFrame& frame) = 0; 85 86 // Called when the connection is closed either locally by the framer, or 87 // remotely by the peer. 88 virtual void OnConnectionClosed(QuicErrorCode error, bool from_peer) = 0; 89 90 // Called when the connection failed to write because the socket was blocked. 91 virtual void OnWriteBlocked() = 0; 92 93 // Called once a specific QUIC version is agreed by both endpoints. 94 virtual void OnSuccessfulVersionNegotiation(const QuicVersion& version) = 0; 95 96 // Called when a blocked socket becomes writable. 97 virtual void OnCanWrite() = 0; 98 99 // Called when the connection experiences a change in congestion window. 100 virtual void OnCongestionWindowChange(QuicTime now) = 0; 101 102 // Called to ask if the visitor wants to schedule write resumption as it both 103 // has pending data to write, and is able to write (e.g. based on flow control 104 // limits). 105 // Writes may be pending because they were write-blocked, congestion-throttled 106 // or yielded to other connections. 107 virtual bool WillingAndAbleToWrite() const = 0; 108 109 // Called to ask if any handshake messages are pending in this visitor. 110 virtual bool HasPendingHandshake() const = 0; 111 112 // Called to ask if any streams are open in this visitor, excluding the 113 // reserved crypto and headers stream. 114 virtual bool HasOpenDataStreams() const = 0; 115 }; 116 117 // Interface which gets callbacks from the QuicConnection at interesting 118 // points. Implementations must not mutate the state of the connection 119 // as a result of these callbacks. 120 class NET_EXPORT_PRIVATE QuicConnectionDebugVisitor 121 : public QuicPacketGenerator::DebugDelegate, 122 public QuicSentPacketManager::DebugDelegate { 123 public: ~QuicConnectionDebugVisitor()124 virtual ~QuicConnectionDebugVisitor() {} 125 126 // Called when a packet has been sent. OnPacketSent(QuicPacketSequenceNumber sequence_number,EncryptionLevel level,TransmissionType transmission_type,const QuicEncryptedPacket & packet,WriteResult result)127 virtual void OnPacketSent(QuicPacketSequenceNumber sequence_number, 128 EncryptionLevel level, 129 TransmissionType transmission_type, 130 const QuicEncryptedPacket& packet, 131 WriteResult result) {} 132 133 // Called when the contents of a packet have been retransmitted as 134 // a new packet. OnPacketRetransmitted(QuicPacketSequenceNumber old_sequence_number,QuicPacketSequenceNumber new_sequence_number)135 virtual void OnPacketRetransmitted( 136 QuicPacketSequenceNumber old_sequence_number, 137 QuicPacketSequenceNumber new_sequence_number) {} 138 139 // Called when a packet has been received, but before it is 140 // validated or parsed. OnPacketReceived(const IPEndPoint & self_address,const IPEndPoint & peer_address,const QuicEncryptedPacket & packet)141 virtual void OnPacketReceived(const IPEndPoint& self_address, 142 const IPEndPoint& peer_address, 143 const QuicEncryptedPacket& packet) {} 144 145 // Called when a packet is received with a connection id that does not 146 // match the ID of this connection. OnIncorrectConnectionId(QuicConnectionId connection_id)147 virtual void OnIncorrectConnectionId( 148 QuicConnectionId connection_id) {} 149 150 // Called when an undecryptable packet has been received. OnUndecryptablePacket()151 virtual void OnUndecryptablePacket() {} 152 153 // Called when a duplicate packet has been received. OnDuplicatePacket(QuicPacketSequenceNumber sequence_number)154 virtual void OnDuplicatePacket(QuicPacketSequenceNumber sequence_number) {} 155 156 // Called when the protocol version on the received packet doensn't match 157 // current protocol version of the connection. OnProtocolVersionMismatch(QuicVersion version)158 virtual void OnProtocolVersionMismatch(QuicVersion version) {} 159 160 // Called when the complete header of a packet has been parsed. OnPacketHeader(const QuicPacketHeader & header)161 virtual void OnPacketHeader(const QuicPacketHeader& header) {} 162 163 // Called when a StreamFrame has been parsed. OnStreamFrame(const QuicStreamFrame & frame)164 virtual void OnStreamFrame(const QuicStreamFrame& frame) {} 165 166 // Called when a AckFrame has been parsed. OnAckFrame(const QuicAckFrame & frame)167 virtual void OnAckFrame(const QuicAckFrame& frame) {} 168 169 // Called when a CongestionFeedbackFrame has been parsed. OnCongestionFeedbackFrame(const QuicCongestionFeedbackFrame & frame)170 virtual void OnCongestionFeedbackFrame( 171 const QuicCongestionFeedbackFrame& frame) {} 172 173 // Called when a StopWaitingFrame has been parsed. OnStopWaitingFrame(const QuicStopWaitingFrame & frame)174 virtual void OnStopWaitingFrame(const QuicStopWaitingFrame& frame) {} 175 176 // Called when a Ping has been parsed. OnPingFrame(const QuicPingFrame & frame)177 virtual void OnPingFrame(const QuicPingFrame& frame) {} 178 179 // Called when a GoAway has been parsed. OnGoAwayFrame(const QuicGoAwayFrame & frame)180 virtual void OnGoAwayFrame(const QuicGoAwayFrame& frame) {} 181 182 // Called when a RstStreamFrame has been parsed. OnRstStreamFrame(const QuicRstStreamFrame & frame)183 virtual void OnRstStreamFrame(const QuicRstStreamFrame& frame) {} 184 185 // Called when a ConnectionCloseFrame has been parsed. OnConnectionCloseFrame(const QuicConnectionCloseFrame & frame)186 virtual void OnConnectionCloseFrame( 187 const QuicConnectionCloseFrame& frame) {} 188 189 // Called when a WindowUpdate has been parsed. OnWindowUpdateFrame(const QuicWindowUpdateFrame & frame)190 virtual void OnWindowUpdateFrame(const QuicWindowUpdateFrame& frame) {} 191 192 // Called when a BlockedFrame has been parsed. OnBlockedFrame(const QuicBlockedFrame & frame)193 virtual void OnBlockedFrame(const QuicBlockedFrame& frame) {} 194 195 // Called when a public reset packet has been received. OnPublicResetPacket(const QuicPublicResetPacket & packet)196 virtual void OnPublicResetPacket(const QuicPublicResetPacket& packet) {} 197 198 // Called when a version negotiation packet has been received. OnVersionNegotiationPacket(const QuicVersionNegotiationPacket & packet)199 virtual void OnVersionNegotiationPacket( 200 const QuicVersionNegotiationPacket& packet) {} 201 202 // Called after a packet has been successfully parsed which results 203 // in the revival of a packet via FEC. OnRevivedPacket(const QuicPacketHeader & revived_header,base::StringPiece payload)204 virtual void OnRevivedPacket(const QuicPacketHeader& revived_header, 205 base::StringPiece payload) {} 206 207 // Called when the connection is closed. OnConnectionClosed(QuicErrorCode error,bool from_peer)208 virtual void OnConnectionClosed(QuicErrorCode error, bool from_peer) {} 209 210 // Called when the version negotiation is successful. OnSuccessfulVersionNegotiation(const QuicVersion & version)211 virtual void OnSuccessfulVersionNegotiation(const QuicVersion& version) {} 212 }; 213 214 class NET_EXPORT_PRIVATE QuicConnectionHelperInterface { 215 public: ~QuicConnectionHelperInterface()216 virtual ~QuicConnectionHelperInterface() {} 217 218 // Returns a QuicClock to be used for all time related functions. 219 virtual const QuicClock* GetClock() const = 0; 220 221 // Returns a QuicRandom to be used for all random number related functions. 222 virtual QuicRandom* GetRandomGenerator() = 0; 223 224 // Creates a new platform-specific alarm which will be configured to 225 // notify |delegate| when the alarm fires. Caller takes ownership 226 // of the new alarm, which will not yet be "set" to fire. 227 virtual QuicAlarm* CreateAlarm(QuicAlarm::Delegate* delegate) = 0; 228 }; 229 230 class NET_EXPORT_PRIVATE QuicConnection 231 : public QuicFramerVisitorInterface, 232 public QuicBlockedWriterInterface, 233 public QuicPacketGenerator::DelegateInterface, 234 public QuicSentPacketManager::NetworkChangeVisitor { 235 public: 236 enum AckBundling { 237 NO_ACK = 0, 238 SEND_ACK = 1, 239 BUNDLE_PENDING_ACK = 2, 240 }; 241 242 class PacketWriterFactory { 243 public: ~PacketWriterFactory()244 virtual ~PacketWriterFactory() {} 245 246 virtual QuicPacketWriter* Create(QuicConnection* connection) const = 0; 247 }; 248 249 // Constructs a new QuicConnection for |connection_id| and |address|. Invokes 250 // writer_factory->Create() to get a writer; |owns_writer| specifies whether 251 // the connection takes ownership of the returned writer. |helper| must 252 // outlive this connection. 253 QuicConnection(QuicConnectionId connection_id, 254 IPEndPoint address, 255 QuicConnectionHelperInterface* helper, 256 const PacketWriterFactory& writer_factory, 257 bool owns_writer, 258 bool is_server, 259 const QuicVersionVector& supported_versions); 260 virtual ~QuicConnection(); 261 262 // Sets connection parameters from the supplied |config|. 263 void SetFromConfig(const QuicConfig& config); 264 265 // Send the data in |data| to the peer in as few packets as possible. 266 // Returns a pair with the number of bytes consumed from data, and a boolean 267 // indicating if the fin bit was consumed. This does not indicate the data 268 // has been sent on the wire: it may have been turned into a packet and queued 269 // if the socket was unexpectedly blocked. |fec_protection| indicates if 270 // data is to be FEC protected. Note that data that is sent immediately 271 // following MUST_FEC_PROTECT data may get protected by falling within the 272 // same FEC group. 273 // If |delegate| is provided, then it will be informed once ACKs have been 274 // received for all the packets written in this call. 275 // The |delegate| is not owned by the QuicConnection and must outlive it. 276 QuicConsumedData SendStreamData(QuicStreamId id, 277 const IOVector& data, 278 QuicStreamOffset offset, 279 bool fin, 280 FecProtection fec_protection, 281 QuicAckNotifier::DelegateInterface* delegate); 282 283 // Send a RST_STREAM frame to the peer. 284 virtual void SendRstStream(QuicStreamId id, 285 QuicRstStreamErrorCode error, 286 QuicStreamOffset bytes_written); 287 288 // Send a BLOCKED frame to the peer. 289 virtual void SendBlocked(QuicStreamId id); 290 291 // Send a WINDOW_UPDATE frame to the peer. 292 virtual void SendWindowUpdate(QuicStreamId id, 293 QuicStreamOffset byte_offset); 294 295 // Sends the connection close packet without affecting the state of the 296 // connection. This should only be called if the session is actively being 297 // destroyed: otherwise call SendConnectionCloseWithDetails instead. 298 virtual void SendConnectionClosePacket(QuicErrorCode error, 299 const std::string& details); 300 301 // Sends a connection close frame to the peer, and closes the connection by 302 // calling CloseConnection(notifying the visitor as it does so). 303 virtual void SendConnectionClose(QuicErrorCode error); 304 virtual void SendConnectionCloseWithDetails(QuicErrorCode error, 305 const std::string& details); 306 // Notifies the visitor of the close and marks the connection as disconnected. 307 virtual void CloseConnection(QuicErrorCode error, bool from_peer) OVERRIDE; 308 virtual void SendGoAway(QuicErrorCode error, 309 QuicStreamId last_good_stream_id, 310 const std::string& reason); 311 312 // Returns statistics tracked for this connection. 313 const QuicConnectionStats& GetStats(); 314 315 // Processes an incoming UDP packet (consisting of a QuicEncryptedPacket) from 316 // the peer. If processing this packet permits a packet to be revived from 317 // its FEC group that packet will be revived and processed. 318 virtual void ProcessUdpPacket(const IPEndPoint& self_address, 319 const IPEndPoint& peer_address, 320 const QuicEncryptedPacket& packet); 321 322 // QuicBlockedWriterInterface 323 // Called when the underlying connection becomes writable to allow queued 324 // writes to happen. 325 virtual void OnCanWrite() OVERRIDE; 326 327 // Called when an error occurs while attempting to write a packet to the 328 // network. 329 void OnWriteError(int error_code); 330 331 // If the socket is not blocked, writes queued packets. 332 void WriteIfNotBlocked(); 333 334 // The version of the protocol this connection is using. version()335 QuicVersion version() const { return framer_.version(); } 336 337 // The versions of the protocol that this connection supports. supported_versions()338 const QuicVersionVector& supported_versions() const { 339 return framer_.supported_versions(); 340 } 341 342 // From QuicFramerVisitorInterface 343 virtual void OnError(QuicFramer* framer) OVERRIDE; 344 virtual bool OnProtocolVersionMismatch(QuicVersion received_version) OVERRIDE; 345 virtual void OnPacket() OVERRIDE; 346 virtual void OnPublicResetPacket( 347 const QuicPublicResetPacket& packet) OVERRIDE; 348 virtual void OnVersionNegotiationPacket( 349 const QuicVersionNegotiationPacket& packet) OVERRIDE; 350 virtual void OnRevivedPacket() OVERRIDE; 351 virtual bool OnUnauthenticatedPublicHeader( 352 const QuicPacketPublicHeader& header) OVERRIDE; 353 virtual bool OnUnauthenticatedHeader(const QuicPacketHeader& header) OVERRIDE; 354 virtual void OnDecryptedPacket(EncryptionLevel level) OVERRIDE; 355 virtual bool OnPacketHeader(const QuicPacketHeader& header) OVERRIDE; 356 virtual void OnFecProtectedPayload(base::StringPiece payload) OVERRIDE; 357 virtual bool OnStreamFrame(const QuicStreamFrame& frame) OVERRIDE; 358 virtual bool OnAckFrame(const QuicAckFrame& frame) OVERRIDE; 359 virtual bool OnCongestionFeedbackFrame( 360 const QuicCongestionFeedbackFrame& frame) OVERRIDE; 361 virtual bool OnStopWaitingFrame(const QuicStopWaitingFrame& frame) OVERRIDE; 362 virtual bool OnPingFrame(const QuicPingFrame& frame) OVERRIDE; 363 virtual bool OnRstStreamFrame(const QuicRstStreamFrame& frame) OVERRIDE; 364 virtual bool OnConnectionCloseFrame( 365 const QuicConnectionCloseFrame& frame) OVERRIDE; 366 virtual bool OnGoAwayFrame(const QuicGoAwayFrame& frame) OVERRIDE; 367 virtual bool OnWindowUpdateFrame(const QuicWindowUpdateFrame& frame) OVERRIDE; 368 virtual bool OnBlockedFrame(const QuicBlockedFrame& frame) OVERRIDE; 369 virtual void OnFecData(const QuicFecData& fec) OVERRIDE; 370 virtual void OnPacketComplete() OVERRIDE; 371 372 // QuicPacketGenerator::DelegateInterface 373 virtual bool ShouldGeneratePacket(TransmissionType transmission_type, 374 HasRetransmittableData retransmittable, 375 IsHandshake handshake) OVERRIDE; 376 virtual QuicAckFrame* CreateAckFrame() OVERRIDE; 377 virtual QuicCongestionFeedbackFrame* CreateFeedbackFrame() OVERRIDE; 378 virtual QuicStopWaitingFrame* CreateStopWaitingFrame() OVERRIDE; 379 virtual void OnSerializedPacket(const SerializedPacket& packet) OVERRIDE; 380 381 // QuicSentPacketManager::NetworkChangeVisitor 382 virtual void OnCongestionWindowChange( 383 QuicByteCount congestion_window) OVERRIDE; 384 385 // Called by the crypto stream when the handshake completes. In the server's 386 // case this is when the SHLO has been ACKed. Clients call this on receipt of 387 // the SHLO. 388 void OnHandshakeComplete(); 389 390 // Accessors set_visitor(QuicConnectionVisitorInterface * visitor)391 void set_visitor(QuicConnectionVisitorInterface* visitor) { 392 visitor_ = visitor; 393 } 394 // This method takes ownership of |debug_visitor|. set_debug_visitor(QuicConnectionDebugVisitor * debug_visitor)395 void set_debug_visitor(QuicConnectionDebugVisitor* debug_visitor) { 396 debug_visitor_.reset(debug_visitor); 397 packet_generator_.set_debug_delegate(debug_visitor); 398 sent_packet_manager_.set_debug_delegate(debug_visitor); 399 } self_address()400 const IPEndPoint& self_address() const { return self_address_; } peer_address()401 const IPEndPoint& peer_address() const { return peer_address_; } connection_id()402 QuicConnectionId connection_id() const { return connection_id_; } clock()403 const QuicClock* clock() const { return clock_; } random_generator()404 QuicRandom* random_generator() const { return random_generator_; } 405 size_t max_packet_length() const; 406 void set_max_packet_length(size_t length); 407 connected()408 bool connected() const { return connected_; } 409 410 // Must only be called on client connections. server_supported_versions()411 const QuicVersionVector& server_supported_versions() const { 412 DCHECK(!is_server_); 413 return server_supported_versions_; 414 } 415 NumFecGroups()416 size_t NumFecGroups() const { return group_map_.size(); } 417 418 // Testing only. NumQueuedPackets()419 size_t NumQueuedPackets() const { return queued_packets_.size(); } 420 ReleaseConnectionClosePacket()421 QuicEncryptedPacket* ReleaseConnectionClosePacket() { 422 return connection_close_packet_.release(); 423 } 424 425 // Returns true if the underlying UDP socket is writable, there is 426 // no queued data and the connection is not congestion-control 427 // blocked. 428 bool CanWriteStreamData(); 429 430 // Returns true if the connection has queued packets or frames. 431 bool HasQueuedData() const; 432 433 // Sets (or resets) the idle state connection timeout. Also, checks and times 434 // out the connection if network timer has expired for |timeout|. 435 void SetIdleNetworkTimeout(QuicTime::Delta timeout); 436 // Sets (or resets) the total time delta the connection can be alive for. 437 // Also, checks and times out the connection if timer has expired for 438 // |timeout|. Used to limit the time a connection can be alive before crypto 439 // handshake finishes. 440 void SetOverallConnectionTimeout(QuicTime::Delta timeout); 441 442 // If the connection has timed out, this will close the connection and return 443 // true. Otherwise, it will return false and will reset the timeout alarm. 444 bool CheckForTimeout(); 445 446 // Sends a ping, and resets the ping alarm. 447 void SendPing(); 448 449 // Sets up a packet with an QuicAckFrame and sends it out. 450 void SendAck(); 451 452 // Called when an RTO fires. Resets the retransmission alarm if there are 453 // remaining unacked packets. 454 void OnRetransmissionTimeout(); 455 456 // Retransmits all unacked packets with retransmittable frames if 457 // |retransmission_type| is ALL_UNACKED_PACKETS, otherwise retransmits only 458 // initially encrypted packets. Used when the negotiated protocol version is 459 // different from what was initially assumed and when the initial encryption 460 // changes. 461 void RetransmitUnackedPackets(TransmissionType retransmission_type); 462 463 // Calls |sent_packet_manager_|'s NeuterUnencryptedPackets. Used when the 464 // connection becomes forward secure and hasn't received acks for all packets. 465 void NeuterUnencryptedPackets(); 466 467 // Changes the encrypter used for level |level| to |encrypter|. The function 468 // takes ownership of |encrypter|. 469 void SetEncrypter(EncryptionLevel level, QuicEncrypter* encrypter); 470 const QuicEncrypter* encrypter(EncryptionLevel level) const; 471 472 // SetDefaultEncryptionLevel sets the encryption level that will be applied 473 // to new packets. 474 void SetDefaultEncryptionLevel(EncryptionLevel level); 475 476 // SetDecrypter sets the primary decrypter, replacing any that already exists, 477 // and takes ownership. If an alternative decrypter is in place then the 478 // function DCHECKs. This is intended for cases where one knows that future 479 // packets will be using the new decrypter and the previous decrypter is now 480 // obsolete. |level| indicates the encryption level of the new decrypter. 481 void SetDecrypter(QuicDecrypter* decrypter, EncryptionLevel level); 482 483 // SetAlternativeDecrypter sets a decrypter that may be used to decrypt 484 // future packets and takes ownership of it. |level| indicates the encryption 485 // level of the decrypter. If |latch_once_used| is true, then the first time 486 // that the decrypter is successful it will replace the primary decrypter. 487 // Otherwise both decrypters will remain active and the primary decrypter 488 // will be the one last used. 489 void SetAlternativeDecrypter(QuicDecrypter* decrypter, 490 EncryptionLevel level, 491 bool latch_once_used); 492 493 const QuicDecrypter* decrypter() const; 494 const QuicDecrypter* alternative_decrypter() const; 495 is_server()496 bool is_server() const { return is_server_; } 497 498 // Returns the underlying sent packet manager. sent_packet_manager()499 const QuicSentPacketManager& sent_packet_manager() const { 500 return sent_packet_manager_; 501 } 502 503 bool CanWrite(HasRetransmittableData retransmittable); 504 505 // Stores current batch state for connection, puts the connection 506 // into batch mode, and destruction restores the stored batch state. 507 // While the bundler is in scope, any generated frames are bundled 508 // as densely as possible into packets. In addition, this bundler 509 // can be configured to ensure that an ACK frame is included in the 510 // first packet created, if there's new ack information to be sent. 511 class ScopedPacketBundler { 512 public: 513 // In addition to all outgoing frames being bundled when the 514 // bundler is in scope, setting |include_ack| to true ensures that 515 // an ACK frame is opportunistically bundled with the first 516 // outgoing packet. 517 ScopedPacketBundler(QuicConnection* connection, AckBundling send_ack); 518 ~ScopedPacketBundler(); 519 520 private: 521 QuicConnection* connection_; 522 bool already_in_batch_mode_; 523 }; 524 525 protected: 526 // Packets which have not been written to the wire. 527 // Owns the QuicPacket* packet. 528 struct QueuedPacket { 529 QueuedPacket(SerializedPacket packet, 530 EncryptionLevel level); 531 QueuedPacket(SerializedPacket packet, 532 EncryptionLevel level, 533 TransmissionType transmission_type, 534 QuicPacketSequenceNumber original_sequence_number); 535 536 SerializedPacket serialized_packet; 537 const EncryptionLevel encryption_level; 538 TransmissionType transmission_type; 539 // The packet's original sequence number if it is a retransmission. 540 // Otherwise it must be 0. 541 QuicPacketSequenceNumber original_sequence_number; 542 }; 543 544 // Do any work which logically would be done in OnPacket but can not be 545 // safely done until the packet is validated. Returns true if the packet 546 // can be handled, false otherwise. 547 virtual bool ProcessValidatedPacket(); 548 549 // Send a packet to the peer, and takes ownership of the packet if the packet 550 // cannot be written immediately. 551 virtual void SendOrQueuePacket(QueuedPacket packet); 552 helper()553 QuicConnectionHelperInterface* helper() { return helper_; } 554 555 // Selects and updates the version of the protocol being used by selecting a 556 // version from |available_versions| which is also supported. Returns true if 557 // such a version exists, false otherwise. 558 bool SelectMutualVersion(const QuicVersionVector& available_versions); 559 writer()560 QuicPacketWriter* writer() { return writer_; } 561 peer_port_changed()562 bool peer_port_changed() const { return peer_port_changed_; } 563 sequence_number_of_last_sent_packet()564 QuicPacketSequenceNumber sequence_number_of_last_sent_packet() const { 565 return sequence_number_of_last_sent_packet_; 566 } 567 568 private: 569 friend class test::QuicConnectionPeer; 570 friend class test::PacketSavingConnection; 571 572 typedef std::list<QueuedPacket> QueuedPacketList; 573 typedef std::map<QuicFecGroupNumber, QuicFecGroup*> FecGroupMap; 574 575 // Writes the given packet to socket, encrypted with packet's 576 // encryption_level. Returns true on successful write, and false if the writer 577 // was blocked and the write needs to be tried again. Notifies the 578 // SentPacketManager when the write is successful and sets 579 // retransmittable frames to NULL. 580 // Saves the connection close packet for later transmission, even if the 581 // writer is write blocked. 582 bool WritePacket(QueuedPacket* packet); 583 584 // Does the main work of WritePacket, but does not delete the packet or 585 // retransmittable frames upon success. 586 bool WritePacketInner(QueuedPacket* packet); 587 588 // Make sure an ack we got from our peer is sane. 589 bool ValidateAckFrame(const QuicAckFrame& incoming_ack); 590 591 // Make sure a stop waiting we got from our peer is sane. 592 bool ValidateStopWaitingFrame(const QuicStopWaitingFrame& stop_waiting); 593 594 // Sends a version negotiation packet to the peer. 595 void SendVersionNegotiationPacket(); 596 597 // Clears any accumulated frames from the last received packet. 598 void ClearLastFrames(); 599 600 // Writes as many queued packets as possible. The connection must not be 601 // blocked when this is called. 602 void WriteQueuedPackets(); 603 604 // Writes as many pending retransmissions as possible. 605 void WritePendingRetransmissions(); 606 607 // Returns true if the packet should be discarded and not sent. 608 bool ShouldDiscardPacket(const QueuedPacket& packet); 609 610 // Queues |packet| in the hopes that it can be decrypted in the 611 // future, when a new key is installed. 612 void QueueUndecryptablePacket(const QuicEncryptedPacket& packet); 613 614 // Attempts to process any queued undecryptable packets. 615 void MaybeProcessUndecryptablePackets(); 616 617 // If a packet can be revived from the current FEC group, then 618 // revive and process the packet. 619 void MaybeProcessRevivedPacket(); 620 621 void ProcessAckFrame(const QuicAckFrame& incoming_ack); 622 623 void ProcessStopWaitingFrame(const QuicStopWaitingFrame& stop_waiting); 624 625 // Update |stop_waiting| for an outgoing ack. 626 void UpdateStopWaiting(QuicStopWaitingFrame* stop_waiting); 627 628 // Queues an ack or sets the ack alarm when an incoming packet arrives that 629 // should be acked. 630 void MaybeQueueAck(); 631 632 // Checks if the last packet should instigate an ack. 633 bool ShouldLastPacketInstigateAck() const; 634 635 // Checks if the peer is waiting for packets that have been given up on, and 636 // therefore an ack frame should be sent with a larger least_unacked. 637 void UpdateStopWaitingCount(); 638 639 // Sends any packets which are a response to the last packet, including both 640 // acks and pending writes if an ack opened the congestion window. 641 void MaybeSendInResponseToPacket(); 642 643 // Gets the least unacked sequence number, which is the next sequence number 644 // to be sent if there are no outstanding packets. 645 QuicPacketSequenceNumber GetLeastUnacked() const; 646 647 // Get the FEC group associate with the last processed packet or NULL, if the 648 // group has already been deleted. 649 QuicFecGroup* GetFecGroup(); 650 651 // Closes any FEC groups protecting packets before |sequence_number|. 652 void CloseFecGroupsBefore(QuicPacketSequenceNumber sequence_number); 653 654 // Sets the ping alarm to the appropriate value, if any. 655 void SetPingAlarm(); 656 657 // On arrival of a new packet, checks to see if the socket addresses have 658 // changed since the last packet we saw on this connection. 659 void CheckForAddressMigration(const IPEndPoint& self_address, 660 const IPEndPoint& peer_address); 661 662 HasRetransmittableData IsRetransmittable(const QueuedPacket& packet); 663 bool IsConnectionClose(QueuedPacket packet); 664 665 QuicFramer framer_; 666 QuicConnectionHelperInterface* helper_; // Not owned. 667 QuicPacketWriter* writer_; // Owned or not depending on |owns_writer_|. 668 bool owns_writer_; 669 EncryptionLevel encryption_level_; 670 const QuicClock* clock_; 671 QuicRandom* random_generator_; 672 673 const QuicConnectionId connection_id_; 674 // Address on the last successfully processed packet received from the 675 // client. 676 IPEndPoint self_address_; 677 IPEndPoint peer_address_; 678 // Used to store latest peer port to possibly migrate to later. 679 int migrating_peer_port_; 680 681 bool last_packet_revived_; // True if the last packet was revived from FEC. 682 size_t last_size_; // Size of the last received packet. 683 EncryptionLevel last_decrypted_packet_level_; 684 QuicPacketHeader last_header_; 685 std::vector<QuicStreamFrame> last_stream_frames_; 686 std::vector<QuicAckFrame> last_ack_frames_; 687 std::vector<QuicCongestionFeedbackFrame> last_congestion_frames_; 688 std::vector<QuicStopWaitingFrame> last_stop_waiting_frames_; 689 std::vector<QuicRstStreamFrame> last_rst_frames_; 690 std::vector<QuicGoAwayFrame> last_goaway_frames_; 691 std::vector<QuicWindowUpdateFrame> last_window_update_frames_; 692 std::vector<QuicBlockedFrame> last_blocked_frames_; 693 std::vector<QuicPingFrame> last_ping_frames_; 694 std::vector<QuicConnectionCloseFrame> last_close_frames_; 695 696 QuicCongestionFeedbackFrame outgoing_congestion_feedback_; 697 698 // Track some peer state so we can do less bookkeeping 699 // Largest sequence sent by the peer which had an ack frame (latest ack info). 700 QuicPacketSequenceNumber largest_seen_packet_with_ack_; 701 702 // Largest sequence number sent by the peer which had a stop waiting frame. 703 QuicPacketSequenceNumber largest_seen_packet_with_stop_waiting_; 704 705 // Collection of packets which were received before encryption was 706 // established, but which could not be decrypted. We buffer these on 707 // the assumption that they could not be processed because they were 708 // sent with the INITIAL encryption and the CHLO message was lost. 709 std::deque<QuicEncryptedPacket*> undecryptable_packets_; 710 711 // When the version negotiation packet could not be sent because the socket 712 // was not writable, this is set to true. 713 bool pending_version_negotiation_packet_; 714 715 // When packets could not be sent because the socket was not writable, 716 // they are added to this list. All corresponding frames are in 717 // unacked_packets_ if they are to be retransmitted. 718 QueuedPacketList queued_packets_; 719 720 // Contains the connection close packet if the connection has been closed. 721 scoped_ptr<QuicEncryptedPacket> connection_close_packet_; 722 723 FecGroupMap group_map_; 724 725 QuicReceivedPacketManager received_packet_manager_; 726 QuicSentEntropyManager sent_entropy_manager_; 727 728 // Indicates whether an ack should be sent the next time we try to write. 729 bool ack_queued_; 730 // Indicates how many consecutive packets have arrived without sending an ack. 731 uint32 num_packets_received_since_last_ack_sent_; 732 // Indicates how many consecutive times an ack has arrived which indicates 733 // the peer needs to stop waiting for some packets. 734 int stop_waiting_count_; 735 736 // An alarm that fires when an ACK should be sent to the peer. 737 scoped_ptr<QuicAlarm> ack_alarm_; 738 // An alarm that fires when a packet needs to be retransmitted. 739 scoped_ptr<QuicAlarm> retransmission_alarm_; 740 // An alarm that is scheduled when the sent scheduler requires a 741 // a delay before sending packets and fires when the packet may be sent. 742 scoped_ptr<QuicAlarm> send_alarm_; 743 // An alarm that is scheduled when the connection can still write and there 744 // may be more data to send. 745 scoped_ptr<QuicAlarm> resume_writes_alarm_; 746 // An alarm that fires when the connection may have timed out. 747 scoped_ptr<QuicAlarm> timeout_alarm_; 748 // An alarm that fires when a ping should be sent. 749 scoped_ptr<QuicAlarm> ping_alarm_; 750 751 QuicConnectionVisitorInterface* visitor_; 752 scoped_ptr<QuicConnectionDebugVisitor> debug_visitor_; 753 QuicPacketGenerator packet_generator_; 754 755 // Network idle time before we kill of this connection. 756 QuicTime::Delta idle_network_timeout_; 757 // Overall connection timeout. 758 QuicTime::Delta overall_connection_timeout_; 759 760 // Statistics for this session. 761 QuicConnectionStats stats_; 762 763 // The time that we got a packet for this connection. 764 // This is used for timeouts, and does not indicate the packet was processed. 765 QuicTime time_of_last_received_packet_; 766 767 // The last time a new (non-retransmitted) packet was sent for this 768 // connection. 769 QuicTime time_of_last_sent_new_packet_; 770 771 // Sequence number of the last sent packet. Packets are guaranteed to be sent 772 // in sequence number order. 773 QuicPacketSequenceNumber sequence_number_of_last_sent_packet_; 774 775 // Sent packet manager which tracks the status of packets sent by this 776 // connection and contains the send and receive algorithms to determine when 777 // to send packets. 778 QuicSentPacketManager sent_packet_manager_; 779 780 // The state of connection in version negotiation finite state machine. 781 QuicVersionNegotiationState version_negotiation_state_; 782 783 // Tracks if the connection was created by the server. 784 bool is_server_; 785 786 // True by default. False if we've received or sent an explicit connection 787 // close. 788 bool connected_; 789 790 // Set to true if the UDP packet headers have a new IP address for the peer. 791 // If true, do not perform connection migration. 792 bool peer_ip_changed_; 793 794 // Set to true if the UDP packet headers have a new port for the peer. 795 // If true, and the IP has not changed, then we can migrate the connection. 796 bool peer_port_changed_; 797 798 // Set to true if the UDP packet headers are addressed to a different IP. 799 // We do not support connection migration when the self IP changed. 800 bool self_ip_changed_; 801 802 // Set to true if the UDP packet headers are addressed to a different port. 803 // We do not support connection migration when the self port changed. 804 bool self_port_changed_; 805 806 // If non-empty this contains the set of versions received in a 807 // version negotiation packet. 808 QuicVersionVector server_supported_versions_; 809 810 DISALLOW_COPY_AND_ASSIGN(QuicConnection); 811 }; 812 813 } // namespace net 814 815 #endif // NET_QUIC_QUIC_CONNECTION_H_ 816