net/quic/test_tools/quic_connection_peer.h

   1 // Copyright (c) 2012 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_CONNECTION_PEER_H_
   6 #define NET_QUIC_TEST_TOOLS_QUIC_CONNECTION_PEER_H_
   7
   8 #include "base/basictypes.h"
   9 #include "net/base/ip_endpoint.h"
  10 #include "net/quic/quic_connection_stats.h"
  11 #include "net/quic/quic_protocol.h"
  12
  13 namespace net {
  14
  15 struct QuicAckFrame;
  16 struct QuicPacketHeader;
  17 class QuicAlarm;
  18 class QuicConnection;
  19 class QuicConnectionHelperInterface;
  20 class QuicConnectionVisitorInterface;
  21 class QuicEncryptedPacket;
  22 class QuicFecGroup;
  23 class QuicFramer;
  24 class QuicPacketCreator;
  25 class QuicPacketGenerator;
  26 class QuicPacketWriter;
  27 class QuicReceivedPacketManager;
  28 class QuicSentPacketManager;
  29 class SendAlgorithmInterface;
  30
  31 namespace test {
  32
  33 // Peer to make public a number of otherwise private QuicConnection methods.
  34 class QuicConnectionPeer {
  35  public:
  36   static void SendAck(QuicConnection* connection);
  37
  38   static void SetSendAlgorithm(QuicConnection* connection,
  39                                SendAlgorithmInterface* send_algorithm);
  40
  41   static void PopulateAckFrame(QuicConnection* connection, QuicAckFrame* ack);
  42
  43   static void PopulateStopWaitingFrame(QuicConnection* connection,
  44                                        QuicStopWaitingFrame* stop_waiting);
  45
  46   static QuicConnectionVisitorInterface* GetVisitor(
  47       QuicConnection* connection);
  48
  49   static QuicPacketCreator* GetPacketCreator(QuicConnection* connection);
  50
  51   static QuicPacketGenerator* GetPacketGenerator(QuicConnection* connection);
  52
  53   static QuicSentPacketManager* GetSentPacketManager(
  54       QuicConnection* connection);
  55
  56   static QuicTime::Delta GetNetworkTimeout(QuicConnection* connection);
  57
  58   static QuicPacketEntropyHash GetSentEntropyHash(
  59       QuicConnection* connection,
  60       QuicPacketNumber packet_number);
  61
  62   static QuicPacketEntropyHash PacketEntropy(QuicConnection* connection,
  63                                              QuicPacketNumber packet_number);
  64
  65   static QuicPacketEntropyHash ReceivedEntropyHash(
  66       QuicConnection* connection,
  67       QuicPacketNumber packet_number);
  68
  69   static void SetPerspective(QuicConnection* connection,
  70                              Perspective perspective);
  71
  72   static void SetSelfAddress(QuicConnection* connection,
  73                              const IPEndPoint& self_address);
  74
  75   static void SetPeerAddress(QuicConnection* connection,
  76                              const IPEndPoint& peer_address);
  77
  78   static bool IsSilentCloseEnabled(QuicConnection* connection);
  79
  80   static void SwapCrypters(QuicConnection* connection, QuicFramer* framer);
  81
  82   static QuicConnectionHelperInterface* GetHelper(QuicConnection* connection);
  83
  84   static QuicFramer* GetFramer(QuicConnection* connection);
  85
  86   // Set last_header_->fec_group = fec_group and return connection->GetFecGroup
  87   static QuicFecGroup* GetFecGroup(QuicConnection* connection, int fec_group);
  88
  89   static QuicAlarm* GetAckAlarm(QuicConnection* connection);
  90   static QuicAlarm* GetPingAlarm(QuicConnection* connection);
  91   static QuicAlarm* GetFecAlarm(QuicConnection* connection);
  92   static QuicAlarm* GetResumeWritesAlarm(QuicConnection* connection);
  93   static QuicAlarm* GetRetransmissionAlarm(QuicConnection* connection);
  94   static QuicAlarm* GetSendAlarm(QuicConnection* connection);
  95   static QuicAlarm* GetTimeoutAlarm(QuicConnection* connection);
  96   static QuicAlarm* GetMtuDiscoveryAlarm(QuicConnection* connection);
  97
  98   static QuicPacketWriter* GetWriter(QuicConnection* connection);
  99   // If |owns_writer| is true, takes ownership of |writer|.
 100   static void SetWriter(QuicConnection* connection,
 101                         QuicPacketWriter* writer,
 102                         bool owns_writer);
 103   static void CloseConnection(QuicConnection* connection);
 104   static QuicEncryptedPacket* GetConnectionClosePacket(
 105       QuicConnection* connection);
 106
 107   static QuicPacketHeader* GetLastHeader(QuicConnection* connection);
 108
 109   static void SetPacketNumberOfLastSentPacket(QuicConnection* connection,
 110                                               QuicPacketNumber number);
 111
 112   static QuicConnectionStats* GetStats(QuicConnection* connection);
 113
 114   static QuicPacketCount GetPacketsBetweenMtuProbes(QuicConnection* connection);
 115
 116   static void SetPacketsBetweenMtuProbes(QuicConnection* connection,
 117                                          QuicPacketCount packets);
 118   static void SetNextMtuProbeAt(QuicConnection* connection,
 119                                 QuicPacketNumber number);
 120
 121  private:
 122   DISALLOW_COPY_AND_ASSIGN(QuicConnectionPeer);
 123 };
 124
 125 }  // namespace test
 126
 127 }  // namespace net
 128
 129 #endif  // NET_QUIC_TEST_TOOLS_QUIC_CONNECTION_PEER_H_