| // Copyright 2013 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #ifndef NET_QUIC_QUIC_ACK_NOTIFIER_MANAGER_H_ |
| #define NET_QUIC_QUIC_ACK_NOTIFIER_MANAGER_H_ |
| |
| #include <map> |
| |
| #include "base/containers/hash_tables.h" |
| #include "net/quic/quic_protocol.h" |
| |
| namespace net { |
| |
| class QuicAckNotifier; |
| |
| // The AckNotifierManager is used by the QuicSentPacketManager to keep track of |
| // all the AckNotifiers currently active. It owns the AckNotifiers which it gets |
| // from the serialized packets passed into OnSerializedPacket. It maintains both |
| // a set of AckNotifiers and a map from sequence number to AckNotifier the sake |
| // of efficiency - we can quickly check the map to see if any AckNotifiers are |
| // interested in a given sequence number. |
| class NET_EXPORT_PRIVATE AckNotifierManager { |
| public: |
| AckNotifierManager(); |
| virtual ~AckNotifierManager(); |
| |
| // Called when the connection receives a new AckFrame. If |sequence_number| |
| // exists in ack_notifier_map_ then the corresponding AckNotifiers will have |
| // their OnAck method called. |
| void OnPacketAcked(QuicPacketSequenceNumber sequence_number, |
| QuicTime::Delta delta_largest_observed); |
| |
| // If a packet has been retransmitted with a new sequence number, then this |
| // will be called. It updates the mapping in ack_notifier_map_, and also |
| // updates the internal set of sequence numbers in each matching AckNotifier. |
| void UpdateSequenceNumber(QuicPacketSequenceNumber old_sequence_number, |
| QuicPacketSequenceNumber new_sequence_number); |
| |
| // This is called after a packet has been serialized, is ready to be sent, and |
| // contains retransmittable frames (which may have associated AckNotifiers). |
| // If any of the retransmittable frames included in |serialized_packet| have |
| // AckNotifiers registered, then add them to our internal map and additionally |
| // inform the AckNotifier of the sequence number which it should track. |
| void OnSerializedPacket(const SerializedPacket& serialized_packet); |
| |
| private: |
| typedef base::hash_set<QuicAckNotifier*> AckNotifierSet; |
| typedef std::map<QuicPacketSequenceNumber, AckNotifierSet> AckNotifierMap; |
| |
| // On every ACK frame received by the connection, all the ack_notifiers_ will |
| // be told which sequeunce numbers were ACKed. |
| // Once a given QuicAckNotifier has seen all the sequence numbers it is |
| // interested in, it will be deleted, and removed from this set. |
| // Owns the AckNotifiers in this set. |
| AckNotifierSet ack_notifiers_; |
| |
| // Maps from sequence number to the AckNotifiers which are registered |
| // for that sequence number. On receipt of an ACK for a given sequence |
| // number, call OnAck for all mapped AckNotifiers. |
| // Does not own the AckNotifiers. |
| AckNotifierMap ack_notifier_map_; |
| |
| DISALLOW_COPY_AND_ASSIGN(AckNotifierManager); |
| }; |
| |
| } // namespace net |
| |
| #endif // NET_QUIC_QUIC_ACK_NOTIFIER_MANAGER_H_ |