blob: aa9d9d1670c252333b093360e9a304ac519b4de3 [file] [log] [blame]
// 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 <list>
#include <map>
#include "base/containers/hash_tables.h"
#include "net/quic/quic_protocol.h"
namespace net {
class QuicAckNotifier;
namespace test {
class AckNotifierManagerPeer;
}
// 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 OnPacketRetransmitted(QuicPacketSequenceNumber old_sequence_number,
QuicPacketSequenceNumber new_sequence_number,
int packet_payload_size);
// 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);
// This method is invoked when a packet is removed from the list of unacked
// packets, and it is no longer necessary to keep track of the notifier.
void OnPacketRemoved(QuicPacketSequenceNumber sequence_number);
private:
friend class test::AckNotifierManagerPeer;
typedef std::list<QuicAckNotifier*> AckNotifierList;
// TODO(ianswett): Further improvement may come from changing this to a deque.
typedef base::hash_map<QuicPacketSequenceNumber, AckNotifierList>
AckNotifierMap;
// 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.
// When the last reference is removed from the map, the notifier is deleted.
AckNotifierMap ack_notifier_map_;
DISALLOW_COPY_AND_ASSIGN(AckNotifierManager);
};
} // namespace net
#endif // NET_QUIC_QUIC_ACK_NOTIFIER_MANAGER_H_