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