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 |