| OLD | NEW |
| (Empty) |
| 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 QuicReceivedPacketManager* GetReceivedPacketManager( | |
| 57 QuicConnection* connection); | |
| 58 | |
| 59 static QuicTime::Delta GetNetworkTimeout(QuicConnection* connection); | |
| 60 | |
| 61 static bool IsSavedForRetransmission( | |
| 62 QuicConnection* connection, | |
| 63 QuicPacketSequenceNumber sequence_number); | |
| 64 | |
| 65 static bool IsRetransmission(QuicConnection* connection, | |
| 66 QuicPacketSequenceNumber sequence_number); | |
| 67 | |
| 68 static QuicPacketEntropyHash GetSentEntropyHash( | |
| 69 QuicConnection* connection, | |
| 70 QuicPacketSequenceNumber sequence_number); | |
| 71 | |
| 72 static QuicPacketEntropyHash PacketEntropy( | |
| 73 QuicConnection* connection, | |
| 74 QuicPacketSequenceNumber sequence_number); | |
| 75 | |
| 76 static QuicPacketEntropyHash ReceivedEntropyHash( | |
| 77 QuicConnection* connection, | |
| 78 QuicPacketSequenceNumber sequence_number); | |
| 79 | |
| 80 static bool IsServer(QuicConnection* connection); | |
| 81 | |
| 82 static void SetIsServer(QuicConnection* connection, bool is_server); | |
| 83 | |
| 84 static void SetSelfAddress(QuicConnection* connection, | |
| 85 const IPEndPoint& self_address); | |
| 86 | |
| 87 static void SetPeerAddress(QuicConnection* connection, | |
| 88 const IPEndPoint& peer_address); | |
| 89 | |
| 90 static bool IsSilentCloseEnabled(QuicConnection* connection); | |
| 91 | |
| 92 static void SwapCrypters(QuicConnection* connection, QuicFramer* framer); | |
| 93 | |
| 94 static QuicConnectionHelperInterface* GetHelper(QuicConnection* connection); | |
| 95 | |
| 96 static QuicFramer* GetFramer(QuicConnection* connection); | |
| 97 | |
| 98 // Set last_header_->fec_group = fec_group and return connection->GetFecGroup | |
| 99 static QuicFecGroup* GetFecGroup(QuicConnection* connection, int fec_group); | |
| 100 | |
| 101 static QuicAlarm* GetAckAlarm(QuicConnection* connection); | |
| 102 static QuicAlarm* GetPingAlarm(QuicConnection* connection); | |
| 103 static QuicAlarm* GetFecAlarm(QuicConnection* connection); | |
| 104 static QuicAlarm* GetResumeWritesAlarm(QuicConnection* connection); | |
| 105 static QuicAlarm* GetRetransmissionAlarm(QuicConnection* connection); | |
| 106 static QuicAlarm* GetSendAlarm(QuicConnection* connection); | |
| 107 static QuicAlarm* GetTimeoutAlarm(QuicConnection* connection); | |
| 108 | |
| 109 static QuicPacketWriter* GetWriter(QuicConnection* connection); | |
| 110 // If |owns_writer| is true, takes ownership of |writer|. | |
| 111 static void SetWriter(QuicConnection* connection, | |
| 112 QuicPacketWriter* writer, | |
| 113 bool owns_writer); | |
| 114 static void CloseConnection(QuicConnection* connection); | |
| 115 static QuicEncryptedPacket* GetConnectionClosePacket( | |
| 116 QuicConnection* connection); | |
| 117 | |
| 118 static void SetSupportedVersions(QuicConnection* connection, | |
| 119 QuicVersionVector versions); | |
| 120 | |
| 121 static QuicPacketHeader* GetLastHeader(QuicConnection* connection); | |
| 122 | |
| 123 static void SetSequenceNumberOfLastSentPacket( | |
| 124 QuicConnection* connection, QuicPacketSequenceNumber number); | |
| 125 | |
| 126 static QuicConnectionStats* GetStats(QuicConnection* connection); | |
| 127 | |
| 128 private: | |
| 129 DISALLOW_COPY_AND_ASSIGN(QuicConnectionPeer); | |
| 130 }; | |
| 131 | |
| 132 } // namespace test | |
| 133 | |
| 134 } // namespace net | |
| 135 | |
| 136 #endif // NET_QUIC_TEST_TOOLS_QUIC_CONNECTION_PEER_H_ | |
| OLD | NEW |