net/quic/test_tools/quic_sent_packet_manager_peer.h

   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 #ifndef NET_QUIC_TEST_TOOLS_QUIC_SENT_PACKET_MANAGER_PEER_H_
   6 #define NET_QUIC_TEST_TOOLS_QUIC_SENT_PACKET_MANAGER_PEER_H_
   7
   8 #include "net/quic/quic_protocol.h"
   9 #include "net/quic/quic_sent_packet_manager.h"
  10
  11 namespace net {
  12
  13 class SendAlgorithmInterface;
  14
  15 namespace test {
  16
  17 class QuicSentPacketManagerPeer {
  18  public:
  19   static void SetMaxTailLossProbes(
  20       QuicSentPacketManager* sent_packet_manager, size_t max_tail_loss_probes);
  21
  22   static void SetSendAlgorithm(QuicSentPacketManager* sent_packet_manager,
  23                                SendAlgorithmInterface* send_algorithm);
  24
  25   static void SetIsServer(QuicSentPacketManager* sent_packet_manager,
  26                           bool is_server);
  27
  28   static const LossDetectionInterface* GetLossAlgorithm(
  29       QuicSentPacketManager* sent_packet_manager);
  30
  31   static const SendAlgorithmInterface* GetCongestionControlAlgorithm(
  32       const QuicSentPacketManager& sent_packet_manager);
  33
  34   static void SetLossAlgorithm(QuicSentPacketManager* sent_packet_manager,
  35                                LossDetectionInterface* loss_detector);
  36
  37   static RttStats* GetRttStats(QuicSentPacketManager* sent_packet_manager);
  38
  39   static size_t GetNackCount(
  40       const QuicSentPacketManager* sent_packet_manager,
  41       QuicPacketSequenceNumber sequence_number);
  42
  43   static size_t GetPendingRetransmissionCount(
  44       const QuicSentPacketManager* sent_packet_manager);
  45
  46   static bool HasPendingPackets(
  47       const QuicSentPacketManager* sent_packet_manager);
  48
  49   static QuicTime GetSentTime(const QuicSentPacketManager* sent_packet_manager,
  50                               QuicPacketSequenceNumber sequence_number);
  51
  52   // Returns true if |sequence_number| is a retransmission of a packet.
  53   static bool IsRetransmission(QuicSentPacketManager* sent_packet_manager,
  54                                QuicPacketSequenceNumber sequence_number);
  55
  56   static void MarkForRetransmission(QuicSentPacketManager* sent_packet_manager,
  57                                     QuicPacketSequenceNumber sequence_number,
  58                                     TransmissionType transmission_type);
  59
  60   static QuicTime::Delta GetRetransmissionDelay(
  61       const QuicSentPacketManager* sent_packet_manager);
  62
  63   static bool HasUnackedCryptoPackets(
  64       const QuicSentPacketManager* sent_packet_manager);
  65
  66   static size_t GetNumRetransmittablePackets(
  67       const QuicSentPacketManager* sent_packet_manager);
  68
  69   static QuicByteCount GetBytesInFlight(
  70       const QuicSentPacketManager* sent_packet_manager);
  71
  72   static QuicSentPacketManager::NetworkChangeVisitor* GetNetworkChangeVisitor(
  73       const QuicSentPacketManager* sent_packet_manager);
  74
  75   static QuicSustainedBandwidthRecorder& GetBandwidthRecorder(
  76       QuicSentPacketManager* sent_packet_manager);
  77
  78  private:
  79   DISALLOW_COPY_AND_ASSIGN(QuicSentPacketManagerPeer);
  80 };
  81
  82 }  // namespace test
  83
  84 }  // namespace net
  85
  86 #endif  // NET_QUIC_TEST_TOOLS_QUIC_SENT_PACKET_MANAGER_PEER_H_