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