net/quic/quic_ack_notifier_manager.cc

   1 // Copyright 2013 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 #include "net/quic/quic_ack_notifier_manager.h"
   6
   7 #include <stddef.h>
   8 #include <list>
   9 #include <map>
  10 #include <utility>
  11 #include <vector>
  12
  13 #include "base/stl_util.h"
  14 #include "net/quic/quic_ack_notifier.h"
  15 #include "net/quic/quic_flags.h"
  16 #include "net/quic/quic_protocol.h"
  17
  18 namespace net {
  19
  20 AckNotifierManager::AckNotifierManager() {}
  21
  22 AckNotifierManager::~AckNotifierManager() {
  23   for (const auto& pair : ack_notifier_map_) {
  24     for (QuicAckNotifier* notifier : pair.second) {
  25       if (notifier->OnPacketAbandoned()) {
  26         delete notifier;
  27       }
  28     }
  29   }
  30 }
  31
  32 void AckNotifierManager::OnPacketAcked(QuicPacketNumber packet_number,
  33                                        QuicTime::Delta delta_largest_observed) {
  34   // Inform all the registered AckNotifiers of the new ACK.
  35   auto map_it = ack_notifier_map_.find(packet_number);
  36   if (map_it == ack_notifier_map_.end()) {
  37     // No AckNotifier is interested in this packet number.
  38     return;
  39   }
  40
  41   // One or more AckNotifiers are registered as interested in this sequence
  42   // number. Iterate through them and call OnAck on each.
  43   for (QuicAckNotifier* ack_notifier : map_it->second) {
  44     if (ack_notifier->OnAck(delta_largest_observed)) {
  45       // If this has resulted in an empty AckNotifer, erase it.
  46       delete ack_notifier;
  47     }
  48   }
  49
  50   // Remove the packet number from the map as we have notified all the
  51   // registered AckNotifiers, and we won't see it again.
  52   ack_notifier_map_.erase(map_it);
  53 }
  54
  55 void AckNotifierManager::OnPacketRetransmitted(
  56     QuicPacketNumber old_packet_number,
  57     QuicPacketNumber new_packet_number,
  58     int packet_payload_size) {
  59   auto map_it = ack_notifier_map_.find(old_packet_number);
  60   if (map_it == ack_notifier_map_.end()) {
  61     // No AckNotifiers are interested in the old packet number.
  62     return;
  63   }
  64
  65   // Update the existing QuicAckNotifiers to the new packet number.
  66   AckNotifierList& ack_notifier_list = map_it->second;
  67   for (QuicAckNotifier* ack_notifier : ack_notifier_list) {
  68     ack_notifier->OnPacketRetransmitted(packet_payload_size);
  69   }
  70
  71   // The old packet number is no longer of interest, copy the updated
  72   // AckNotifiers to the new packet number before deleting the old.
  73   ack_notifier_map_[new_packet_number] = ack_notifier_list;
  74   ack_notifier_map_.erase(map_it);
  75 }
  76
  77 void AckNotifierManager::OnSerializedPacket(
  78     const SerializedPacket& serialized_packet) {
  79   // Inform each attached AckNotifier of the packet's serialization.
  80   AckNotifierList& notifier_list =
  81       ack_notifier_map_[serialized_packet.packet_number];
  82   for (QuicAckNotifier* notifier : serialized_packet.notifiers) {
  83     if (notifier == nullptr) {
  84       LOG(DFATAL) << "AckNotifier should not be nullptr.";
  85       continue;
  86     }
  87     notifier->OnSerializedPacket();
  88     notifier_list.push_back(notifier);
  89   }
  90 }
  91
  92 void AckNotifierManager::OnPacketRemoved(QuicPacketNumber packet_number) {
  93   // Determine if there are any notifiers interested in this packet.
  94   auto map_it = ack_notifier_map_.find(packet_number);
  95   if (map_it == ack_notifier_map_.end()) {
  96     return;
  97   }
  98
  99   // Notify all of the interested notifiers that the packet is abandoned.
 100   for (QuicAckNotifier* ack_notifier : map_it->second) {
 101     DCHECK(ack_notifier);
 102     if (ack_notifier->OnPacketAbandoned()) {
 103       // If this has resulted in an empty AckNotifer, erase it.
 104       delete ack_notifier;
 105     }
 106   }
 107
 108   // Remove the packet with given packet number from the map.
 109   ack_notifier_map_.erase(map_it);
 110 }
 111
 112 }  // namespace net