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 // Responsible for generating packets on behalf of a QuicConnection. | |
6 // Packets are serialized just-in-time. Control frames are queued. | |
7 // Ack and Feedback frames will be requested from the Connection | |
8 // just-in-time. When a packet needs to be sent, the Generator | |
9 // will serialize a packet and pass it to QuicConnection::SendOrQueuePacket() | |
10 // | |
11 // The Generator's mode of operation is controlled by two conditions: | |
12 // | |
13 // 1) Is the Delegate writable? | |
14 // | |
15 // If the Delegate is not writable, then no operations will cause | |
16 // a packet to be serialized. In particular: | |
17 // * SetShouldSendAck will simply record that an ack is to be sent. | |
18 // * AddControlFrame will enqueue the control frame. | |
19 // * ConsumeData will do nothing. | |
20 // | |
21 // If the Delegate is writable, then the behavior depends on the second | |
22 // condition: | |
23 // | |
24 // 2) Is the Generator in batch mode? | |
25 // | |
26 // If the Generator is NOT in batch mode, then each call to a write | |
27 // operation will serialize one or more packets. The contents will | |
28 // include any previous queued frames. If an ack should be sent | |
29 // but has not been sent, then the Delegate will be asked to create | |
30 // an Ack frame which will then be included in the packet. When | |
31 // the write call completes, the current packet will be serialized | |
32 // and sent to the Delegate, even if it is not full. | |
33 // | |
34 // If the Generator is in batch mode, then each write operation will | |
35 // add data to the "current" packet. When the current packet becomes | |
36 // full, it will be serialized and sent to the packet. When batch | |
37 // mode is ended via |FinishBatchOperations|, the current packet | |
38 // will be serialzied, even if it is not full. | |
39 // | |
40 // FEC behavior also depends on batch mode. In batch mode, FEC packets | |
41 // will be sent after |max_packets_per_group| have been sent, as well | |
42 // as after batch operations are complete. When not in batch mode, | |
43 // an FEC packet will be sent after each write call completes. | |
44 // | |
45 // TODO(rch): This behavior should probably be tuned. When not in batch | |
46 // mode, we should probably set a timer so that several independent | |
47 // operations can be grouped into the same FEC group. | |
48 // | |
49 // When an FEC packet is generated, it will be send to the Delegate, | |
50 // even if the Delegate has become unwritable after handling the | |
51 // data packet immediately proceeding the FEC packet. | |
52 | |
53 #ifndef NET_QUIC_QUIC_PACKET_GENERATOR_H_ | |
54 #define NET_QUIC_QUIC_PACKET_GENERATOR_H_ | |
55 | |
56 #include <list> | |
57 | |
58 #include "base/containers/hash_tables.h" | |
59 #include "net/quic/quic_ack_notifier.h" | |
60 #include "net/quic/quic_packet_creator.h" | |
61 #include "net/quic/quic_sent_packet_manager.h" | |
62 #include "net/quic/quic_types.h" | |
63 | |
64 namespace net { | |
65 | |
66 namespace test { | |
67 class QuicPacketGeneratorPeer; | |
68 } // namespace test | |
69 | |
70 class NET_EXPORT_PRIVATE QuicPacketGenerator { | |
71 public: | |
72 class NET_EXPORT_PRIVATE DelegateInterface { | |
73 public: | |
74 virtual ~DelegateInterface() {} | |
75 virtual bool ShouldGeneratePacket(TransmissionType transmission_type, | |
76 HasRetransmittableData retransmittable, | |
77 IsHandshake handshake) = 0; | |
78 virtual void PopulateAckFrame(QuicAckFrame* ack) = 0; | |
79 virtual void PopulateStopWaitingFrame( | |
80 QuicStopWaitingFrame* stop_waiting) = 0; | |
81 // Takes ownership of |packet.packet| and |packet.retransmittable_frames|. | |
82 virtual void OnSerializedPacket(const SerializedPacket& packet) = 0; | |
83 virtual void CloseConnection(QuicErrorCode error, bool from_peer) = 0; | |
84 }; | |
85 | |
86 // Interface which gets callbacks from the QuicPacketGenerator at interesting | |
87 // points. Implementations must not mutate the state of the generator | |
88 // as a result of these callbacks. | |
89 class NET_EXPORT_PRIVATE DebugDelegate { | |
90 public: | |
91 virtual ~DebugDelegate() {} | |
92 | |
93 // Called when a frame has been added to the current packet. | |
94 virtual void OnFrameAddedToPacket(const QuicFrame& frame) {} | |
95 }; | |
96 | |
97 QuicPacketGenerator(QuicConnectionId connection_id, | |
98 QuicFramer* framer, | |
99 QuicRandom* random_generator, | |
100 DelegateInterface* delegate); | |
101 | |
102 virtual ~QuicPacketGenerator(); | |
103 | |
104 // Called by the connection in the event of the congestion window changing. | |
105 void OnCongestionWindowChange(QuicPacketCount max_packets_in_flight); | |
106 | |
107 // Called by the connection when the RTT may have changed. | |
108 void OnRttChange(QuicTime::Delta rtt); | |
109 | |
110 // Indicates that an ACK frame should be sent. | |
111 // If |also_send_stop_waiting| is true, then it also indicates that a | |
112 // STOP_WAITING frame should be sent as well. | |
113 // The contents of the frame(s) will be generated via a call to the delegate | |
114 // CreateAckFrame() when the packet is serialized. | |
115 void SetShouldSendAck(bool also_send_stop_waiting); | |
116 | |
117 // Indicates that a STOP_WAITING frame should be sent. | |
118 void SetShouldSendStopWaiting(); | |
119 | |
120 void AddControlFrame(const QuicFrame& frame); | |
121 | |
122 // Given some data, may consume part or all of it and pass it to the | |
123 // packet creator to be serialized into packets. If not in batch | |
124 // mode, these packets will also be sent during this call. | |
125 // |delegate| (if not nullptr) will be informed once all packets sent as a | |
126 // result of this call are ACKed by the peer. | |
127 QuicConsumedData ConsumeData(QuicStreamId id, | |
128 const IOVector& data, | |
129 QuicStreamOffset offset, | |
130 bool fin, | |
131 FecProtection fec_protection, | |
132 QuicAckNotifier::DelegateInterface* delegate); | |
133 | |
134 // Indicates whether batch mode is currently enabled. | |
135 bool InBatchMode(); | |
136 // Disables flushing. | |
137 void StartBatchOperations(); | |
138 // Enables flushing and flushes queued data which can be sent. | |
139 void FinishBatchOperations(); | |
140 | |
141 // Flushes all queued frames, even frames which are not sendable. | |
142 void FlushAllQueuedFrames(); | |
143 | |
144 bool HasQueuedFrames() const; | |
145 | |
146 // Makes the framer not serialize the protocol version in sent packets. | |
147 void StopSendingVersion(); | |
148 | |
149 // Creates a version negotiation packet which supports |supported_versions|. | |
150 // Caller owns the created packet. Also, sets the entropy hash of the | |
151 // serialized packet to a random bool and returns that value as a member of | |
152 // SerializedPacket. | |
153 QuicEncryptedPacket* SerializeVersionNegotiationPacket( | |
154 const QuicVersionVector& supported_versions); | |
155 | |
156 | |
157 // Re-serializes frames with the original packet's sequence number length. | |
158 // Used for retransmitting packets to ensure they aren't too long. | |
159 // Caller must ensure that any open FEC group is closed before calling this | |
160 // method. | |
161 SerializedPacket ReserializeAllFrames( | |
162 const RetransmittableFrames& frames, | |
163 QuicSequenceNumberLength original_length); | |
164 | |
165 // Update the sequence number length to use in future packets as soon as it | |
166 // can be safely changed. | |
167 void UpdateSequenceNumberLength( | |
168 QuicPacketSequenceNumber least_packet_awaited_by_peer, | |
169 QuicPacketCount max_packets_in_flight); | |
170 | |
171 // Set the minimum number of bytes for the connection id length; | |
172 void SetConnectionIdLength(uint32 length); | |
173 | |
174 // Called when the FEC alarm fires. | |
175 void OnFecTimeout(); | |
176 | |
177 // Called after sending |sequence_number| to determine whether an FEC alarm | |
178 // should be set for sending out an FEC packet. Returns a positive and finite | |
179 // timeout if an FEC alarm should be set, and infinite if no alarm should be | |
180 // set. OnFecTimeout should be called to send the FEC packet when the alarm | |
181 // fires. | |
182 QuicTime::Delta GetFecTimeout(QuicPacketSequenceNumber sequence_number); | |
183 | |
184 // Sets the encryption level that will be applied to new packets. | |
185 void set_encryption_level(EncryptionLevel level); | |
186 | |
187 // Sequence number of the last created packet, or 0 if no packets have been | |
188 // created. | |
189 QuicPacketSequenceNumber sequence_number() const; | |
190 | |
191 QuicByteCount max_packet_length() const; | |
192 | |
193 void set_max_packet_length(QuicByteCount length); | |
194 | |
195 void set_debug_delegate(DebugDelegate* debug_delegate) { | |
196 debug_delegate_ = debug_delegate; | |
197 } | |
198 | |
199 private: | |
200 friend class test::QuicPacketGeneratorPeer; | |
201 | |
202 // Turn on FEC protection for subsequent packets in the generator. | |
203 // If no FEC group is currently open in the creator, this method flushes any | |
204 // queued frames in the generator and in the creator, and it then turns FEC on | |
205 // in the creator. This method may be called with an open FEC group in the | |
206 // creator, in which case, only the generator's state is altered. | |
207 void MaybeStartFecProtection(); | |
208 | |
209 // Serializes and calls the delegate on an FEC packet if one was under | |
210 // construction in the creator. When |force| is false, it relies on the | |
211 // creator being ready to send an FEC packet, otherwise FEC packet is sent | |
212 // as long as one is under construction in the creator. Also tries to turn | |
213 // off FEC protection in the creator if it's off in the generator. | |
214 void MaybeSendFecPacketAndCloseGroup(bool force); | |
215 | |
216 // Returns true if an FEC packet should be generated based on |force| and | |
217 // current state of the generator and the creator. | |
218 bool ShouldSendFecPacket(bool force); | |
219 | |
220 void SendQueuedFrames(bool flush); | |
221 | |
222 // Test to see if we have pending ack, or control frames. | |
223 bool HasPendingFrames() const; | |
224 // Test to see if the addition of a pending frame (which might be | |
225 // retransmittable) would still allow the resulting packet to be sent now. | |
226 bool CanSendWithNextPendingFrameAddition() const; | |
227 // Add exactly one pending frame, preferring ack frames over control frames. | |
228 bool AddNextPendingFrame(); | |
229 | |
230 bool AddFrame(const QuicFrame& frame); | |
231 | |
232 void SerializeAndSendPacket(); | |
233 | |
234 DelegateInterface* delegate_; | |
235 DebugDelegate* debug_delegate_; | |
236 | |
237 QuicPacketCreator packet_creator_; | |
238 QuicFrames queued_control_frames_; | |
239 | |
240 // True if batch mode is currently enabled. | |
241 bool batch_mode_; | |
242 | |
243 // Timeout used for FEC alarm. Can be set to zero initially or if the SRTT has | |
244 // not yet been set. | |
245 QuicTime::Delta fec_timeout_; | |
246 | |
247 // True if FEC protection is on. The creator may have an open FEC group even | |
248 // if this variable is false. | |
249 bool should_fec_protect_; | |
250 | |
251 // Flags to indicate the need for just-in-time construction of a frame. | |
252 bool should_send_ack_; | |
253 bool should_send_stop_waiting_; | |
254 // If we put a non-retransmittable frame (ack frame) in this packet, then we | |
255 // have to hold a reference to it until we flush (and serialize it). | |
256 // Retransmittable frames are referenced elsewhere so that they | |
257 // can later be (optionally) retransmitted. | |
258 QuicAckFrame pending_ack_frame_; | |
259 QuicStopWaitingFrame pending_stop_waiting_frame_; | |
260 // True if an ack or stop waiting frame is already queued, and should not be | |
261 // re-added. | |
262 bool ack_queued_; | |
263 bool stop_waiting_queued_; | |
264 | |
265 // Stores notifiers that should be attached to the next serialized packet. | |
266 std::list<QuicAckNotifier*> ack_notifiers_; | |
267 | |
268 DISALLOW_COPY_AND_ASSIGN(QuicPacketGenerator); | |
269 }; | |
270 | |
271 } // namespace net | |
272 | |
273 #endif // NET_QUIC_QUIC_PACKET_GENERATOR_H_ | |
OLD | NEW |