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 // Accumulates frames for the next packet until more frames no longer fit or | |
6 // it's time to create a packet from them. Also provides packet creation of | |
7 // FEC packets based on previously created packets. | |
8 | |
9 #ifndef NET_QUIC_QUIC_PACKET_CREATOR_H_ | |
10 #define NET_QUIC_QUIC_PACKET_CREATOR_H_ | |
11 | |
12 #include <utility> | |
13 #include <vector> | |
14 | |
15 #include "base/memory/scoped_ptr.h" | |
16 #include "base/strings/string_piece.h" | |
17 #include "net/quic/quic_fec_group.h" | |
18 #include "net/quic/quic_framer.h" | |
19 #include "net/quic/quic_protocol.h" | |
20 | |
21 namespace net { | |
22 namespace test { | |
23 class QuicPacketCreatorPeer; | |
24 } | |
25 | |
26 class QuicAckNotifier; | |
27 class QuicRandom; | |
28 class QuicRandomBoolSource; | |
29 | |
30 class NET_EXPORT_PRIVATE QuicPacketCreator : public QuicFecBuilderInterface { | |
31 public: | |
32 // QuicRandom* required for packet entropy. | |
33 QuicPacketCreator(QuicConnectionId connection_id, | |
34 QuicFramer* framer, | |
35 QuicRandom* random_generator); | |
36 | |
37 ~QuicPacketCreator() override; | |
38 | |
39 // QuicFecBuilderInterface | |
40 void OnBuiltFecProtectedPayload(const QuicPacketHeader& header, | |
41 base::StringPiece payload) override; | |
42 | |
43 // Turn on FEC protection for subsequently created packets. FEC should be | |
44 // enabled first (max_packets_per_fec_group should be non-zero) for FEC | |
45 // protection to start. | |
46 void StartFecProtectingPackets(); | |
47 | |
48 // Turn off FEC protection for subsequently created packets. If the creator | |
49 // has any open FEC group, call will fail. It is the caller's responsibility | |
50 // to flush out FEC packets in generation, and to verify with ShouldSendFec() | |
51 // that there is no open FEC group. | |
52 void StopFecProtectingPackets(); | |
53 | |
54 // Checks if it's time to send an FEC packet. |force_close| forces this to | |
55 // return true if an FEC group is open. | |
56 bool ShouldSendFec(bool force_close) const; | |
57 | |
58 // Returns true if an FEC packet is under construction. | |
59 bool IsFecGroupOpen() const; | |
60 | |
61 // Makes the framer not serialize the protocol version in sent packets. | |
62 void StopSendingVersion(); | |
63 | |
64 // Update the sequence number length to use in future packets as soon as it | |
65 // can be safely changed. | |
66 void UpdateSequenceNumberLength( | |
67 QuicPacketSequenceNumber least_packet_awaited_by_peer, | |
68 QuicPacketCount max_packets_in_flight); | |
69 | |
70 // The overhead the framing will add for a packet with one frame. | |
71 static size_t StreamFramePacketOverhead( | |
72 QuicConnectionIdLength connection_id_length, | |
73 bool include_version, | |
74 QuicSequenceNumberLength sequence_number_length, | |
75 QuicStreamOffset offset, | |
76 InFecGroup is_in_fec_group); | |
77 | |
78 bool HasRoomForStreamFrame(QuicStreamId id, QuicStreamOffset offset) const; | |
79 | |
80 // Converts a raw payload to a frame which fits into the currently open | |
81 // packet if there is one. Returns the number of bytes consumed from data. | |
82 // If data is empty and fin is true, the expected behavior is to consume the | |
83 // fin but return 0. | |
84 size_t CreateStreamFrame(QuicStreamId id, | |
85 const IOVector& data, | |
86 QuicStreamOffset offset, | |
87 bool fin, | |
88 QuicFrame* frame); | |
89 | |
90 // Serializes all frames into a single packet. All frames must fit into a | |
91 // single packet. Also, sets the entropy hash of the serialized packet to a | |
92 // random bool and returns that value as a member of SerializedPacket. | |
93 // Never returns a RetransmittableFrames in SerializedPacket. | |
94 SerializedPacket SerializeAllFrames(const QuicFrames& frames); | |
95 | |
96 // Re-serializes frames with the original packet's sequence number length. | |
97 // Used for retransmitting packets to ensure they aren't too long. | |
98 // Caller must ensure that any open FEC group is closed before calling this | |
99 // method. | |
100 SerializedPacket ReserializeAllFrames( | |
101 const RetransmittableFrames& frames, | |
102 QuicSequenceNumberLength original_length); | |
103 | |
104 // Returns true if there are frames pending to be serialized. | |
105 bool HasPendingFrames() const; | |
106 | |
107 // Returns true if there are retransmittable frames pending to be serialized. | |
108 bool HasPendingRetransmittableFrames() const; | |
109 | |
110 // TODO(jri): Remove this method. | |
111 // Returns whether FEC protection is currently enabled. Note: Enabled does not | |
112 // mean that an FEC group is currently active; i.e., IsFecProtected() may | |
113 // still return false. | |
114 bool IsFecEnabled() const; | |
115 | |
116 // Returns true if subsequent packets will be FEC protected. Note: True does | |
117 // not mean that an FEC packet is currently under construction; i.e., | |
118 // fec_group_.get() may still be nullptr, until MaybeStartFec() is called. | |
119 bool IsFecProtected() const; | |
120 | |
121 // Returns the number of bytes which are available to be used by additional | |
122 // frames in the packet. Since stream frames are slightly smaller when they | |
123 // are the last frame in a packet, this method will return a different | |
124 // value than max_packet_size - PacketSize(), in this case. | |
125 size_t BytesFree() const; | |
126 | |
127 // Returns the number of bytes that the packet will expand by if a new frame | |
128 // is added to the packet. If the last frame was a stream frame, it will | |
129 // expand slightly when a new frame is added, and this method returns the | |
130 // amount of expected expansion. If the packet is in an FEC group, no | |
131 // expansion happens and this method always returns zero. | |
132 size_t ExpansionOnNewFrame() const; | |
133 | |
134 // Returns the number of bytes in the current packet, including the header, | |
135 // if serialized with the current frames. Adding a frame to the packet | |
136 // may change the serialized length of existing frames, as per the comment | |
137 // in BytesFree. | |
138 size_t PacketSize() const; | |
139 | |
140 // TODO(jri): AddSavedFrame calls AddFrame, which only saves the frame | |
141 // if it is a stream frame, not other types of frames. Fix this API; | |
142 // add a AddNonSavedFrame method. | |
143 // Adds |frame| to the packet creator's list of frames to be serialized. | |
144 // Returns false if the frame doesn't fit into the current packet. | |
145 bool AddSavedFrame(const QuicFrame& frame); | |
146 | |
147 // Serializes all frames which have been added and adds any which should be | |
148 // retransmitted to |retransmittable_frames| if it's not nullptr. All frames | |
149 // must fit into a single packet. Sets the entropy hash of the serialized | |
150 // packet to a random bool and returns that value as a member of | |
151 // SerializedPacket. Also, sets |serialized_frames| in the SerializedPacket to | |
152 // the corresponding RetransmittableFrames if any frames are to be | |
153 // retransmitted. | |
154 SerializedPacket SerializePacket(); | |
155 | |
156 // Packetize FEC data. All frames must fit into a single packet. Also, sets | |
157 // the entropy hash of the serialized packet to a random bool and returns | |
158 // that value as a member of SerializedPacket. | |
159 SerializedPacket SerializeFec(); | |
160 | |
161 // Creates a packet with connection close frame. Caller owns the created | |
162 // packet. Also, sets the entropy hash of the serialized packet to a random | |
163 // bool and returns that value as a member of SerializedPacket. | |
164 SerializedPacket SerializeConnectionClose( | |
165 QuicConnectionCloseFrame* close_frame); | |
166 | |
167 // Creates a version negotiation packet which supports |supported_versions|. | |
168 // Caller owns the created packet. Also, sets the entropy hash of the | |
169 // serialized packet to a random bool and returns that value as a member of | |
170 // SerializedPacket. | |
171 QuicEncryptedPacket* SerializeVersionNegotiationPacket( | |
172 const QuicVersionVector& supported_versions); | |
173 | |
174 // Sets the encryption level that will be applied to new packets. | |
175 void set_encryption_level(EncryptionLevel level) { | |
176 encryption_level_ = level; | |
177 } | |
178 | |
179 // Sequence number of the last created packet, or 0 if no packets have been | |
180 // created. | |
181 QuicPacketSequenceNumber sequence_number() const { | |
182 return sequence_number_; | |
183 } | |
184 | |
185 void set_sequence_number(QuicPacketSequenceNumber s) { | |
186 sequence_number_ = s; | |
187 } | |
188 | |
189 QuicConnectionIdLength connection_id_length() const { | |
190 return connection_id_length_; | |
191 } | |
192 | |
193 void set_connection_id_length(QuicConnectionIdLength length) { | |
194 connection_id_length_ = length; | |
195 } | |
196 | |
197 QuicSequenceNumberLength next_sequence_number_length() const { | |
198 return next_sequence_number_length_; | |
199 } | |
200 | |
201 void set_next_sequence_number_length(QuicSequenceNumberLength length) { | |
202 next_sequence_number_length_ = length; | |
203 } | |
204 | |
205 QuicByteCount max_packet_length() const { | |
206 return max_packet_length_; | |
207 } | |
208 | |
209 void set_max_packet_length(QuicByteCount length) { | |
210 // |max_packet_length_| should not be changed mid-packet or mid-FEC group. | |
211 DCHECK(fec_group_.get() == nullptr && queued_frames_.empty()); | |
212 max_packet_length_ = length; | |
213 } | |
214 | |
215 // Returns current max number of packets covered by an FEC group. | |
216 size_t max_packets_per_fec_group() const { | |
217 return max_packets_per_fec_group_; | |
218 } | |
219 | |
220 // Sets creator's max number of packets covered by an FEC group. | |
221 // Note: While there are no constraints on |max_packets_per_fec_group|, | |
222 // this setter enforces a min value of kLowestMaxPacketsPerFecGroup. | |
223 // To turn off FEC protection, use StopFecProtectingPackets(). | |
224 void set_max_packets_per_fec_group(size_t max_packets_per_fec_group); | |
225 | |
226 // Returns the currently open FEC group's number. If there isn't an open FEC | |
227 // group, returns the last closed FEC group number. Returns 0 when FEC is | |
228 // disabled or no FEC group has been created yet. | |
229 QuicFecGroupNumber fec_group_number() { return fec_group_number_; } | |
230 | |
231 private: | |
232 friend class test::QuicPacketCreatorPeer; | |
233 | |
234 static bool ShouldRetransmit(const QuicFrame& frame); | |
235 | |
236 // Updates sequence number and max packet lengths on a packet or FEC group | |
237 // boundary. | |
238 void MaybeUpdateLengths(); | |
239 | |
240 // Updates lengths and also starts an FEC group if FEC protection is on and | |
241 // there is not already an FEC group open. | |
242 InFecGroup MaybeUpdateLengthsAndStartFec(); | |
243 | |
244 void FillPacketHeader(QuicFecGroupNumber fec_group, | |
245 bool fec_flag, | |
246 QuicPacketHeader* header); | |
247 | |
248 // Allows a frame to be added without creating retransmittable frames. | |
249 // Particularly useful for retransmits using SerializeAllFrames(). | |
250 bool AddFrame(const QuicFrame& frame, bool save_retransmittable_frames); | |
251 | |
252 // Adds a padding frame to the current packet only if the current packet | |
253 // contains a handshake message, and there is sufficient room to fit a | |
254 // padding frame. | |
255 void MaybeAddPadding(); | |
256 | |
257 QuicConnectionId connection_id_; | |
258 EncryptionLevel encryption_level_; | |
259 QuicFramer* framer_; | |
260 scoped_ptr<QuicRandomBoolSource> random_bool_source_; | |
261 QuicPacketSequenceNumber sequence_number_; | |
262 // If true, any created packets will be FEC protected. | |
263 bool should_fec_protect_; | |
264 QuicFecGroupNumber fec_group_number_; | |
265 scoped_ptr<QuicFecGroup> fec_group_; | |
266 // Controls whether protocol version should be included while serializing the | |
267 // packet. | |
268 bool send_version_in_packet_; | |
269 // Maximum length including headers and encryption (UDP payload length.) | |
270 QuicByteCount max_packet_length_; | |
271 // 0 indicates FEC is disabled. | |
272 size_t max_packets_per_fec_group_; | |
273 // Length of connection_id to send over the wire. | |
274 QuicConnectionIdLength connection_id_length_; | |
275 // Staging variable to hold next packet sequence number length. When sequence | |
276 // number length is to be changed, this variable holds the new length until | |
277 // a packet or FEC group boundary, when the creator's sequence_number_length_ | |
278 // can be changed to this new value. | |
279 QuicSequenceNumberLength next_sequence_number_length_; | |
280 // Sequence number length for the current packet and for the current FEC group | |
281 // when FEC is enabled. Mutable so PacketSize() can adjust it when the packet | |
282 // is empty. | |
283 mutable QuicSequenceNumberLength sequence_number_length_; | |
284 // packet_size_ is mutable because it's just a cache of the current size. | |
285 // packet_size should never be read directly, use PacketSize() instead. | |
286 mutable size_t packet_size_; | |
287 QuicFrames queued_frames_; | |
288 scoped_ptr<RetransmittableFrames> queued_retransmittable_frames_; | |
289 | |
290 DISALLOW_COPY_AND_ASSIGN(QuicPacketCreator); | |
291 }; | |
292 | |
293 } // namespace net | |
294 | |
295 #endif // NET_QUIC_QUIC_PACKET_CREATOR_H_ | |
OLD | NEW |