OLD | NEW |
1 // Copyright 2013 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "net/quic/quic_ack_notifier.h" | 5 #include "net/quic/quic_ack_notifier.h" |
6 | 6 |
7 #include <set> | 7 #include <set> |
8 | 8 |
9 #include "base/logging.h" | 9 #include "base/logging.h" |
10 #include "base/stl_util.h" | 10 #include "base/stl_util.h" |
11 | 11 |
12 using base::hash_map; | 12 using base::hash_map; |
13 using std::make_pair; | 13 using std::make_pair; |
14 | 14 |
15 namespace net { | 15 namespace net { |
16 | 16 |
17 QuicAckNotifier::PacketInfo::PacketInfo() : packet_payload_size(0) { | 17 QuicAckNotifier::PacketInfo::PacketInfo() : packet_payload_size(0) { |
18 } | 18 } |
19 | 19 |
20 QuicAckNotifier::PacketInfo::PacketInfo(int payload_size) | 20 QuicAckNotifier::PacketInfo::PacketInfo(int payload_size) |
21 : packet_payload_size(payload_size) { | 21 : packet_payload_size(payload_size) { |
22 } | 22 } |
23 | 23 |
24 QuicAckNotifier::DelegateInterface::DelegateInterface() {} | 24 QuicAckNotifier::DelegateInterface::DelegateInterface() { |
| 25 } |
25 | 26 |
26 QuicAckNotifier::DelegateInterface::~DelegateInterface() {} | 27 QuicAckNotifier::DelegateInterface::~DelegateInterface() { |
| 28 } |
27 | 29 |
28 QuicAckNotifier::QuicAckNotifier(DelegateInterface* delegate) | 30 QuicAckNotifier::QuicAckNotifier(DelegateInterface* delegate) |
29 : delegate_(delegate), | 31 : delegate_(delegate), |
30 original_packet_count_(0), | 32 original_packet_count_(0), |
31 original_byte_count_(0), | 33 original_byte_count_(0), |
32 retransmitted_packet_count_(0), | 34 retransmitted_packet_count_(0), |
33 retransmitted_byte_count_(0) { | 35 retransmitted_byte_count_(0) { |
34 DCHECK(delegate); | 36 DCHECK(delegate); |
35 } | 37 } |
36 | 38 |
37 QuicAckNotifier::~QuicAckNotifier() { | 39 QuicAckNotifier::~QuicAckNotifier() { |
38 } | 40 } |
39 | 41 |
40 void QuicAckNotifier::AddSequenceNumber( | 42 void QuicAckNotifier::AddSequenceNumber( |
41 const QuicPacketSequenceNumber& sequence_number, | 43 const QuicPacketSequenceNumber& sequence_number, |
42 int packet_payload_size) { | 44 int packet_payload_size) { |
43 sequence_numbers_.insert(make_pair(sequence_number, | 45 sequence_numbers_.insert( |
44 PacketInfo(packet_payload_size))); | 46 make_pair(sequence_number, PacketInfo(packet_payload_size))); |
45 ++original_packet_count_; | 47 ++original_packet_count_; |
46 original_byte_count_ += packet_payload_size; | 48 original_byte_count_ += packet_payload_size; |
47 } | 49 } |
48 | 50 |
49 bool QuicAckNotifier::OnAck(QuicPacketSequenceNumber sequence_number, | 51 bool QuicAckNotifier::OnAck(QuicPacketSequenceNumber sequence_number, |
50 QuicTime::Delta delta_largest_observed) { | 52 QuicTime::Delta delta_largest_observed) { |
51 DCHECK(ContainsKey(sequence_numbers_, sequence_number)); | 53 DCHECK(ContainsKey(sequence_numbers_, sequence_number)); |
52 sequence_numbers_.erase(sequence_number); | 54 sequence_numbers_.erase(sequence_number); |
53 if (IsEmpty()) { | 55 if (IsEmpty()) { |
54 // We have seen all the sequence numbers we were waiting for, trigger | 56 // We have seen all the sequence numbers we were waiting for, trigger |
55 // callback notification. | 57 // callback notification. |
56 delegate_->OnAckNotification( | 58 delegate_->OnAckNotification(original_packet_count_, |
57 original_packet_count_, original_byte_count_, | 59 original_byte_count_, |
58 retransmitted_packet_count_, retransmitted_byte_count_, | 60 retransmitted_packet_count_, |
59 delta_largest_observed); | 61 retransmitted_byte_count_, |
| 62 delta_largest_observed); |
60 return true; | 63 return true; |
61 } | 64 } |
62 return false; | 65 return false; |
63 } | 66 } |
64 | 67 |
65 void QuicAckNotifier::UpdateSequenceNumber( | 68 void QuicAckNotifier::UpdateSequenceNumber( |
66 QuicPacketSequenceNumber old_sequence_number, | 69 QuicPacketSequenceNumber old_sequence_number, |
67 QuicPacketSequenceNumber new_sequence_number) { | 70 QuicPacketSequenceNumber new_sequence_number) { |
68 DCHECK(!ContainsKey(sequence_numbers_, new_sequence_number)); | 71 DCHECK(!ContainsKey(sequence_numbers_, new_sequence_number)); |
69 | 72 |
70 PacketInfo packet_info; | 73 PacketInfo packet_info; |
71 hash_map<QuicPacketSequenceNumber, PacketInfo>::iterator it = | 74 hash_map<QuicPacketSequenceNumber, PacketInfo>::iterator it = |
72 sequence_numbers_.find(old_sequence_number); | 75 sequence_numbers_.find(old_sequence_number); |
73 if (it != sequence_numbers_.end()) { | 76 if (it != sequence_numbers_.end()) { |
74 packet_info = it->second; | 77 packet_info = it->second; |
75 sequence_numbers_.erase(it); | 78 sequence_numbers_.erase(it); |
76 } else { | 79 } else { |
77 DLOG(DFATAL) << "Old sequence number not found."; | 80 DLOG(DFATAL) << "Old sequence number not found."; |
78 } | 81 } |
79 | 82 |
80 ++retransmitted_packet_count_; | 83 ++retransmitted_packet_count_; |
81 retransmitted_byte_count_ += packet_info.packet_payload_size; | 84 retransmitted_byte_count_ += packet_info.packet_payload_size; |
82 sequence_numbers_.insert(make_pair(new_sequence_number, packet_info)); | 85 sequence_numbers_.insert(make_pair(new_sequence_number, packet_info)); |
83 } | 86 } |
84 | 87 |
85 }; // namespace net | 88 }; // namespace net |
OLD | NEW |