blob: a82cec4fc9d43dce5d213190661840c91d51de38 [file] [log] [blame]
QUICHE teama6ef0a62019-03-07 20:34:33 -05001// 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
fayang4245c212019-11-05 13:33:46 -08005// Responsible for creating packets on behalf of a QuicConnection.
6// Packets are serialized just-in-time. Stream data and control frames will be
7// requested from the Connection just-in-time. Frames are accumulated into
8// "current" packet until no more frames can fit, then current packet gets
9// serialized and passed to connection via OnSerializedPacket().
10//
11// Whether a packet should be serialized is determined by whether delegate is
12// writable. If the Delegate is not writable, then no operations will cause
13// a packet to be serialized.
QUICHE teama6ef0a62019-03-07 20:34:33 -050014
15#ifndef QUICHE_QUIC_CORE_QUIC_PACKET_CREATOR_H_
16#define QUICHE_QUIC_CORE_QUIC_PACKET_CREATOR_H_
17
18#include <cstddef>
19#include <memory>
20#include <utility>
21#include <vector>
22
renjietangdbe98342019-10-18 11:00:57 -070023#include "net/third_party/quiche/src/quic/core/frames/quic_stream_frame.h"
danzhbf4836c2020-02-11 20:29:15 -080024#include "net/third_party/quiche/src/quic/core/quic_circular_deque.h"
fayang08750832019-10-24 11:25:34 -070025#include "net/third_party/quiche/src/quic/core/quic_coalesced_packet.h"
QUICHE teama6ef0a62019-03-07 20:34:33 -050026#include "net/third_party/quiche/src/quic/core/quic_framer.h"
27#include "net/third_party/quiche/src/quic/core/quic_packets.h"
QUICHE teama6ef0a62019-03-07 20:34:33 -050028#include "net/third_party/quiche/src/quic/core/quic_types.h"
29#include "net/third_party/quiche/src/quic/platform/api/quic_export.h"
fayang04590fc2020-08-31 10:37:55 -070030#include "net/third_party/quiche/src/quic/platform/api/quic_macros.h"
dmcardlecf0bfcf2019-12-13 08:08:21 -080031#include "net/third_party/quiche/src/common/platform/api/quiche_string_piece.h"
QUICHE teama6ef0a62019-03-07 20:34:33 -050032
33namespace quic {
34namespace test {
35class QuicPacketCreatorPeer;
36}
37
38class QUIC_EXPORT_PRIVATE QuicPacketCreator {
39 public:
40 // A delegate interface for further processing serialized packet.
ianswettb023c7b2019-05-06 12:38:10 -070041 class QUIC_EXPORT_PRIVATE DelegateInterface {
QUICHE teama6ef0a62019-03-07 20:34:33 -050042 public:
ianswettb023c7b2019-05-06 12:38:10 -070043 virtual ~DelegateInterface() {}
dschinazi66dea072019-04-09 11:41:06 -070044 // Get a buffer of kMaxOutgoingPacketSize bytes to serialize the next
wub50d4c712020-05-19 15:48:28 -070045 // packet. If the return value's buffer is nullptr, QuicPacketCreator will
46 // serialize on a stack buffer.
47 virtual QuicPacketBuffer GetPacketBuffer() = 0;
wub8a5dafa2020-05-13 12:30:17 -070048 // Called when a packet is serialized. Delegate take the ownership of
49 // |serialized_packet|.
50 virtual void OnSerializedPacket(SerializedPacket serialized_packet) = 0;
ianswettb023c7b2019-05-06 12:38:10 -070051
52 // Called when an unrecoverable error is encountered.
53 virtual void OnUnrecoverableError(QuicErrorCode error,
fkastenholz85f18902019-05-28 12:47:00 -070054 const std::string& error_details) = 0;
fayangcad11792019-09-16 13:11:44 -070055
56 // Consults delegate whether a packet should be generated.
57 virtual bool ShouldGeneratePacket(HasRetransmittableData retransmittable,
58 IsHandshake handshake) = 0;
59 // Called when there is data to be sent. Retrieves updated ACK frame from
60 // the delegate.
61 virtual const QuicFrames MaybeBundleAckOpportunistically() = 0;
fayang15042962020-07-01 12:14:29 -070062
63 // Returns the packet fate for serialized packets which will be handed over
64 // to delegate via OnSerializedPacket(). Called when a packet is about to be
65 // serialized.
66 virtual SerializedPacketFate GetSerializedPacketFate(
67 bool is_mtu_discovery,
68 EncryptionLevel encryption_level) = 0;
QUICHE teama6ef0a62019-03-07 20:34:33 -050069 };
70
71 // Interface which gets callbacks from the QuicPacketCreator at interesting
72 // points. Implementations must not mutate the state of the creator
73 // as a result of these callbacks.
74 class QUIC_EXPORT_PRIVATE DebugDelegate {
75 public:
76 virtual ~DebugDelegate() {}
77
78 // Called when a frame has been added to the current packet.
dschinazi17d42422019-06-18 16:35:07 -070079 virtual void OnFrameAddedToPacket(const QuicFrame& /*frame*/) {}
renjietangdbe98342019-10-18 11:00:57 -070080
81 // Called when a stream frame is coalesced with an existing stream frame.
82 // |frame| is the new stream frame.
83 virtual void OnStreamFrameCoalesced(const QuicStreamFrame& /*frame*/) {}
QUICHE teama6ef0a62019-03-07 20:34:33 -050084 };
85
danzh051bf772020-08-24 12:30:36 -070086 // Set the peer address which the serialized packet will be sent to during the
87 // scope of this object. Upon exiting the scope, the original peer address is
88 // restored.
89 class QUIC_EXPORT_PRIVATE ScopedPeerAddressContext {
90 public:
91 ScopedPeerAddressContext(QuicPacketCreator* creator,
92 QuicSocketAddress address);
93 ~ScopedPeerAddressContext();
94
95 private:
96 QuicPacketCreator* creator_;
97 QuicSocketAddress old_peer_address_;
98 };
99
dschinazi7b9278c2019-05-20 07:36:21 -0700100 QuicPacketCreator(QuicConnectionId server_connection_id,
QUICHE teama6ef0a62019-03-07 20:34:33 -0500101 QuicFramer* framer,
102 DelegateInterface* delegate);
dschinazi7b9278c2019-05-20 07:36:21 -0700103 QuicPacketCreator(QuicConnectionId server_connection_id,
QUICHE teama6ef0a62019-03-07 20:34:33 -0500104 QuicFramer* framer,
105 QuicRandom* random,
106 DelegateInterface* delegate);
107 QuicPacketCreator(const QuicPacketCreator&) = delete;
108 QuicPacketCreator& operator=(const QuicPacketCreator&) = delete;
109
110 ~QuicPacketCreator();
111
112 // Makes the framer not serialize the protocol version in sent packets.
113 void StopSendingVersion();
114
115 // SetDiversificationNonce sets the nonce that will be sent in each public
116 // header of packets encrypted at the initial encryption level. Should only
117 // be called by servers.
118 void SetDiversificationNonce(const DiversificationNonce& nonce);
119
120 // Update the packet number length to use in future packets as soon as it
121 // can be safely changed.
122 // TODO(fayang): Directly set packet number length instead of compute it in
123 // creator.
124 void UpdatePacketNumberLength(QuicPacketNumber least_packet_awaited_by_peer,
125 QuicPacketCount max_packets_in_flight);
126
fayang4c1c2362019-09-13 07:20:01 -0700127 // Skip |count| packet numbers.
128 void SkipNPacketNumbers(QuicPacketCount count,
129 QuicPacketNumber least_packet_awaited_by_peer,
130 QuicPacketCount max_packets_in_flight);
131
QUICHE teama6ef0a62019-03-07 20:34:33 -0500132 // The overhead the framing will add for a packet with one frame.
133 static size_t StreamFramePacketOverhead(
134 QuicTransportVersion version,
135 QuicConnectionIdLength destination_connection_id_length,
136 QuicConnectionIdLength source_connection_id_length,
137 bool include_version,
138 bool include_diversification_nonce,
139 QuicPacketNumberLength packet_number_length,
140 QuicVariableLengthIntegerLength retry_token_length_length,
141 QuicVariableLengthIntegerLength length_length,
142 QuicStreamOffset offset);
143
144 // Returns false and flushes all pending frames if current open packet is
145 // full.
146 // If current packet is not full, creates a stream frame that fits into the
147 // open packet and adds it to the packet.
fayang62b637b2019-09-16 08:40:49 -0700148 bool ConsumeDataToFillCurrentPacket(QuicStreamId id,
149 size_t data_size,
150 QuicStreamOffset offset,
151 bool fin,
152 bool needs_full_padding,
153 TransmissionType transmission_type,
154 QuicFrame* frame);
QUICHE teama6ef0a62019-03-07 20:34:33 -0500155
156 // Creates a CRYPTO frame that fits into the current packet (which must be
157 // empty) and adds it to the packet.
fayang62b637b2019-09-16 08:40:49 -0700158 bool ConsumeCryptoDataToFillCurrentPacket(EncryptionLevel level,
159 size_t write_length,
160 QuicStreamOffset offset,
161 bool needs_full_padding,
162 TransmissionType transmission_type,
163 QuicFrame* frame);
QUICHE teama6ef0a62019-03-07 20:34:33 -0500164
165 // Returns true if current open packet can accommodate more stream frames of
166 // stream |id| at |offset| and data length |data_size|, false otherwise.
fayang5a3cfcb2020-08-27 12:57:23 -0700167 // TODO(fayang): mark this const by moving RemoveSoftMaxPacketLength out.
QUICHE teama6ef0a62019-03-07 20:34:33 -0500168 bool HasRoomForStreamFrame(QuicStreamId id,
169 QuicStreamOffset offset,
170 size_t data_size);
171
172 // Returns true if current open packet can accommodate a message frame of
173 // |length|.
fayang5a3cfcb2020-08-27 12:57:23 -0700174 // TODO(fayang): mark this const by moving RemoveSoftMaxPacketLength out.
QUICHE teama6ef0a62019-03-07 20:34:33 -0500175 bool HasRoomForMessageFrame(QuicByteCount length);
176
QUICHE teama6ef0a62019-03-07 20:34:33 -0500177 // Serializes all added frames into a single packet and invokes the delegate_
178 // to further process the SerializedPacket.
fayang62b637b2019-09-16 08:40:49 -0700179 void FlushCurrentPacket();
QUICHE teama6ef0a62019-03-07 20:34:33 -0500180
181 // Optimized method to create a QuicStreamFrame and serialize it. Adds the
182 // QuicStreamFrame to the returned SerializedPacket. Sets
183 // |num_bytes_consumed| to the number of bytes consumed to create the
184 // QuicStreamFrame.
185 void CreateAndSerializeStreamFrame(QuicStreamId id,
186 size_t write_length,
187 QuicStreamOffset iov_offset,
188 QuicStreamOffset stream_offset,
189 bool fin,
190 TransmissionType transmission_type,
191 size_t* num_bytes_consumed);
192
193 // Returns true if there are frames pending to be serialized.
194 bool HasPendingFrames() const;
195
196 // Returns true if there are retransmittable frames pending to be serialized.
197 bool HasPendingRetransmittableFrames() const;
198
199 // Returns true if there are stream frames for |id| pending to be serialized.
200 bool HasPendingStreamFramesOfStream(QuicStreamId id) const;
201
202 // Returns the number of bytes which are available to be used by additional
203 // frames in the packet. Since stream frames are slightly smaller when they
204 // are the last frame in a packet, this method will return a different
205 // value than max_packet_size - PacketSize(), in this case.
fayang5a3cfcb2020-08-27 12:57:23 -0700206 size_t BytesFree() const;
QUICHE teama6ef0a62019-03-07 20:34:33 -0500207
208 // Returns the number of bytes that the packet will expand by if a new frame
209 // is added to the packet. If the last frame was a stream frame, it will
210 // expand slightly when a new frame is added, and this method returns the
211 // amount of expected expansion.
212 size_t ExpansionOnNewFrame() const;
213
fayang54a38e32020-06-26 07:01:15 -0700214 // Returns the number of bytes that the packet will expand by when a new frame
215 // is going to be added. |last_frame| is the last frame of the packet.
bnce5c11c02020-07-29 08:02:21 -0700216 static size_t ExpansionOnNewFrameWithLastFrame(const QuicFrame& last_frame,
217 QuicTransportVersion version);
fayang54a38e32020-06-26 07:01:15 -0700218
QUICHE teama6ef0a62019-03-07 20:34:33 -0500219 // Returns the number of bytes in the current packet, including the header,
220 // if serialized with the current frames. Adding a frame to the packet
221 // may change the serialized length of existing frames, as per the comment
222 // in BytesFree.
fayang5a3cfcb2020-08-27 12:57:23 -0700223 size_t PacketSize() const;
QUICHE teama6ef0a62019-03-07 20:34:33 -0500224
225 // Tries to add |frame| to the packet creator's list of frames to be
226 // serialized. If the frame does not fit into the current packet, flushes the
227 // packet and returns false.
renjietangb63005e2019-11-19 23:08:53 -0800228 bool AddFrame(const QuicFrame& frame, TransmissionType transmission_type);
QUICHE teama6ef0a62019-03-07 20:34:33 -0500229
230 // Identical to AddSavedFrame, but allows the frame to be padded.
231 bool AddPaddedSavedFrame(const QuicFrame& frame,
232 TransmissionType transmission_type);
233
234 // Creates a version negotiation packet which supports |supported_versions|.
235 std::unique_ptr<QuicEncryptedPacket> SerializeVersionNegotiationPacket(
236 bool ietf_quic,
dschinazi48ac9192019-07-31 00:07:26 -0700237 bool use_length_prefix,
QUICHE teama6ef0a62019-03-07 20:34:33 -0500238 const ParsedQuicVersionVector& supported_versions);
239
240 // Creates a connectivity probing packet for versions prior to version 99.
wub8a5dafa2020-05-13 12:30:17 -0700241 std::unique_ptr<SerializedPacket> SerializeConnectivityProbingPacket();
QUICHE teama6ef0a62019-03-07 20:34:33 -0500242
243 // Create connectivity probing request and response packets using PATH
244 // CHALLENGE and PATH RESPONSE frames, respectively, for version 99/IETF QUIC.
245 // SerializePathChallengeConnectivityProbingPacket will pad the packet to be
246 // MTU bytes long.
wub8a5dafa2020-05-13 12:30:17 -0700247 std::unique_ptr<SerializedPacket>
248 SerializePathChallengeConnectivityProbingPacket(QuicPathFrameBuffer* payload);
QUICHE teama6ef0a62019-03-07 20:34:33 -0500249
250 // If |is_padded| is true then SerializePathResponseConnectivityProbingPacket
251 // will pad the packet to be MTU bytes long, else it will not pad the packet.
252 // |payloads| is cleared.
wub8a5dafa2020-05-13 12:30:17 -0700253 std::unique_ptr<SerializedPacket>
254 SerializePathResponseConnectivityProbingPacket(
wuba750aab2020-02-10 06:43:15 -0800255 const QuicCircularDeque<QuicPathFrameBuffer>& payloads,
QUICHE teama6ef0a62019-03-07 20:34:33 -0500256 const bool is_padded);
257
258 // Returns a dummy packet that is valid but contains no useful information.
259 static SerializedPacket NoPacket();
260
QUICHE team2252b702019-05-14 23:55:14 -0400261 // Returns the destination connection ID to send over the wire.
262 QuicConnectionId GetDestinationConnectionId() const;
263
264 // Returns the source connection ID to send over the wire.
265 QuicConnectionId GetSourceConnectionId() const;
266
QUICHE teama6ef0a62019-03-07 20:34:33 -0500267 // Returns length of destination connection ID to send over the wire.
268 QuicConnectionIdLength GetDestinationConnectionIdLength() const;
269
270 // Returns length of source connection ID to send over the wire.
271 QuicConnectionIdLength GetSourceConnectionIdLength() const;
272
dschinazi7b9278c2019-05-20 07:36:21 -0700273 // Sets whether the server connection ID should be sent over the wire.
274 void SetServerConnectionIdIncluded(
275 QuicConnectionIdIncluded server_connection_id_included);
QUICHE teama6ef0a62019-03-07 20:34:33 -0500276
dschinazi8ff74822019-05-28 16:37:20 -0700277 // Update the server connection ID used in outgoing packets.
dschinazi7b9278c2019-05-20 07:36:21 -0700278 void SetServerConnectionId(QuicConnectionId server_connection_id);
QUICHE teamc65d1d12019-03-19 20:58:04 -0700279
dschinazi346b7ce2019-06-05 01:38:18 -0700280 // Update the client connection ID used in outgoing packets.
281 void SetClientConnectionId(QuicConnectionId client_connection_id);
282
QUICHE teama6ef0a62019-03-07 20:34:33 -0500283 // Sets the encryption level that will be applied to new packets.
dschinazi87aad8a2020-08-26 17:02:02 -0700284 void set_encryption_level(EncryptionLevel level);
dschinazi6458eb32020-06-23 12:38:41 -0700285 EncryptionLevel encryption_level() { return packet_.encryption_level; }
QUICHE teama6ef0a62019-03-07 20:34:33 -0500286
287 // packet number of the last created packet, or 0 if no packets have been
288 // created.
289 QuicPacketNumber packet_number() const { return packet_.packet_number; }
290
291 QuicByteCount max_packet_length() const { return max_packet_length_; }
292
293 bool has_ack() const { return packet_.has_ack; }
294
295 bool has_stop_waiting() const { return packet_.has_stop_waiting; }
296
297 // Sets the encrypter to use for the encryption level and updates the max
298 // plaintext size.
299 void SetEncrypter(EncryptionLevel level,
300 std::unique_ptr<QuicEncrypter> encrypter);
301
302 // Indicates whether the packet creator is in a state where it can change
303 // current maximum packet length.
304 bool CanSetMaxPacketLength() const;
305
306 // Sets the maximum packet length.
307 void SetMaxPacketLength(QuicByteCount length);
308
dschinazied459c02020-05-07 16:12:23 -0700309 // Sets the maximum DATAGRAM/MESSAGE frame size we can send.
310 void SetMaxDatagramFrameSize(QuicByteCount max_datagram_frame_size);
311
fayang2ab1e852019-11-04 11:24:36 -0800312 // Set a soft maximum packet length in the creator. If a packet cannot be
313 // successfully created, creator will remove the soft limit and use the actual
314 // max packet length.
315 void SetSoftMaxPacketLength(QuicByteCount length);
316
QUICHE teama6ef0a62019-03-07 20:34:33 -0500317 // Increases pending_padding_bytes by |size|. Pending padding will be sent by
318 // MaybeAddPadding().
319 void AddPendingPadding(QuicByteCount size);
320
dschinazi244f6dc2019-05-06 15:45:16 -0700321 // Sets the retry token to be sent over the wire in IETF Initial packets.
dmcardlecf0bfcf2019-12-13 08:08:21 -0800322 void SetRetryToken(quiche::QuicheStringPiece retry_token);
QUICHE teama6ef0a62019-03-07 20:34:33 -0500323
fayang18be79a2019-09-16 15:17:12 -0700324 // Consumes retransmittable control |frame|. Returns true if the frame is
325 // successfully consumed. Returns false otherwise.
326 bool ConsumeRetransmittableControlFrame(const QuicFrame& frame);
327
328 // Given some data, may consume part or all of it and pass it to the
329 // packet creator to be serialized into packets. If not in batch
330 // mode, these packets will also be sent during this call.
331 // When |state| is FIN_AND_PADDING, random padding of size [1, 256] will be
332 // added after stream frames. If current constructed packet cannot
333 // accommodate, the padding will overflow to the next packet(s).
334 QuicConsumedData ConsumeData(QuicStreamId id,
335 size_t write_length,
336 QuicStreamOffset offset,
337 StreamSendingState state);
338
339 // Sends as many data only packets as allowed by the send algorithm and the
340 // available iov.
341 // This path does not support padding, or bundling pending frames.
342 // In case we access this method from ConsumeData, total_bytes_consumed
343 // keeps track of how many bytes have already been consumed.
344 QuicConsumedData ConsumeDataFastPath(QuicStreamId id,
345 size_t write_length,
346 QuicStreamOffset offset,
347 bool fin,
348 size_t total_bytes_consumed);
349
350 // Consumes data for CRYPTO frames sent at |level| starting at |offset| for a
351 // total of |write_length| bytes, and returns the number of bytes consumed.
352 // The data is passed into the packet creator and serialized into one or more
353 // packets.
354 size_t ConsumeCryptoData(EncryptionLevel level,
355 size_t write_length,
356 QuicStreamOffset offset);
357
358 // Generates an MTU discovery packet of specified size.
359 void GenerateMtuDiscoveryPacket(QuicByteCount target_mtu);
360
361 // Called when there is data to be sent, Retrieves updated ACK frame from
362 // delegate_ and flushes it.
363 void MaybeBundleAckOpportunistically();
364
365 // Called to flush ACK and STOP_WAITING frames, returns false if the flush
366 // fails.
367 bool FlushAckFrame(const QuicFrames& frames);
368
369 // Adds a random amount of padding (between 1 to 256 bytes).
370 void AddRandomPadding();
371
372 // Attaches packet flusher.
373 void AttachPacketFlusher();
374
375 // Flushes everything, including current open packet and pending padding.
376 void Flush();
377
378 // Sends remaining pending padding.
379 // Pending paddings should only be sent when there is nothing else to send.
380 void SendRemainingPendingPadding();
381
382 // Set the minimum number of bytes for the server connection id length;
383 void SetServerConnectionIdLength(uint32_t length);
384
385 // Set transmission type of next constructed packets.
386 void SetTransmissionType(TransmissionType type);
387
388 // Tries to add a message frame containing |message| and returns the status.
389 MessageStatus AddMessageFrame(QuicMessageId message_id,
390 QuicMemSliceSpan message);
391
QUICHE teama6ef0a62019-03-07 20:34:33 -0500392 // Returns the largest payload that will fit into a single MESSAGE frame.
ianswettb239f862019-04-05 09:15:06 -0700393 QuicPacketLength GetCurrentLargestMessagePayload() const;
394 // Returns the largest payload that will fit into a single MESSAGE frame at
395 // any point during the connection. This assumes the version and
396 // connection ID lengths do not change.
397 QuicPacketLength GetGuaranteedLargestMessagePayload() const;
QUICHE teama6ef0a62019-03-07 20:34:33 -0500398
fayang354c9422019-05-21 08:10:35 -0700399 // Packet number of next created packet.
400 QuicPacketNumber NextSendingPacketNumber() const;
401
QUICHE teama6ef0a62019-03-07 20:34:33 -0500402 void set_debug_delegate(DebugDelegate* debug_delegate) {
403 debug_delegate_ = debug_delegate;
404 }
405
QUICHE teama6ef0a62019-03-07 20:34:33 -0500406 QuicByteCount pending_padding_bytes() const { return pending_padding_bytes_; }
407
dschinazi5c1d7d82020-07-29 16:42:50 -0700408 ParsedQuicVersion version() const { return framer_->version(); }
409
QUICHE teama6ef0a62019-03-07 20:34:33 -0500410 QuicTransportVersion transport_version() const {
411 return framer_->transport_version();
412 }
413
nharper55fa6132019-05-07 19:37:21 -0700414 // Returns the minimum size that the plaintext of a packet must be.
QUICHE team2252b702019-05-14 23:55:14 -0400415 static size_t MinPlaintextPacketSize(const ParsedQuicVersion& version);
nharper55fa6132019-05-07 19:37:21 -0700416
fayang18be79a2019-09-16 15:17:12 -0700417 // Indicates whether packet flusher is currently attached.
418 bool PacketFlusherAttached() const;
419
nharper3907ac22019-09-25 15:32:28 -0700420 void set_fully_pad_crypto_handshake_packets(bool new_value) {
fayang18be79a2019-09-16 15:17:12 -0700421 fully_pad_crypto_handshake_packets_ = new_value;
422 }
423
424 bool fully_pad_crypto_handshake_packets() const {
fayang18be79a2019-09-16 15:17:12 -0700425 return fully_pad_crypto_handshake_packets_;
426 }
427
renjietang4c704c82019-10-07 16:39:11 -0700428 // Serialize a probing packet that uses IETF QUIC's PATH CHALLENGE frame. Also
429 // fills the packet with padding.
430 size_t BuildPaddedPathChallengePacket(const QuicPacketHeader& header,
431 char* buffer,
432 size_t packet_length,
433 QuicPathFrameBuffer* payload,
434 QuicRandom* randomizer,
435 EncryptionLevel level);
436
437 // Serialize a probing response packet that uses IETF QUIC's PATH RESPONSE
438 // frame. Also fills the packet with padding if |is_padded| is
439 // true. |payloads| is always emptied, even if the packet can not be
440 // successfully built.
wuba750aab2020-02-10 06:43:15 -0800441 size_t BuildPathResponsePacket(
442 const QuicPacketHeader& header,
443 char* buffer,
444 size_t packet_length,
445 const QuicCircularDeque<QuicPathFrameBuffer>& payloads,
446 const bool is_padded,
447 EncryptionLevel level);
renjietang4c704c82019-10-07 16:39:11 -0700448
449 // Serializes a probing packet, which is a padded PING packet. Returns the
450 // length of the packet. Returns 0 if it fails to serialize.
451 size_t BuildConnectivityProbingPacket(const QuicPacketHeader& header,
452 char* buffer,
453 size_t packet_length,
454 EncryptionLevel level);
455
fayang08750832019-10-24 11:25:34 -0700456 // Serializes |coalesced| to provided |buffer|, returns coalesced packet
457 // length if serialization succeeds. Otherwise, returns 0.
458 size_t SerializeCoalescedPacket(const QuicCoalescedPacket& coalesced,
459 char* buffer,
460 size_t buffer_len);
461
fayangfe963c52020-07-16 06:56:09 -0700462 // Returns true if max_packet_length_ is currently a soft value.
463 bool HasSoftMaxPacketLength() const;
464
fayangfe963c52020-07-16 06:56:09 -0700465 bool coalesced_packet_of_higher_space() const {
466 return coalesced_packet_of_higher_space_;
467 }
468
danzh051bf772020-08-24 12:30:36 -0700469 // Use this address to sent to the peer from now on. If this address is
470 // different from the current one, flush all the queue frames first.
471 void SetDefaultPeerAddress(QuicSocketAddress address);
472
QUICHE teama6ef0a62019-03-07 20:34:33 -0500473 private:
474 friend class test::QuicPacketCreatorPeer;
475
fayang04590fc2020-08-31 10:37:55 -0700476 // Used to 1) clear queued_frames_, 2) report unrecoverable error (if
477 // serialization fails) upon exiting the scope.
478 class QUIC_EXPORT_PRIVATE ScopedSerializationFailureHandler {
fayange447bc62020-08-27 13:47:11 -0700479 public:
fayang04590fc2020-08-31 10:37:55 -0700480 explicit ScopedSerializationFailureHandler(QuicPacketCreator* creator);
481 ~ScopedSerializationFailureHandler();
fayange447bc62020-08-27 13:47:11 -0700482
483 private:
484 QuicPacketCreator* creator_; // Unowned.
485 };
486
QUICHE teama6ef0a62019-03-07 20:34:33 -0500487 // Creates a stream frame which fits into the current open packet. If
QUICHE teamf08778a2019-03-14 08:10:26 -0700488 // |data_size| is 0 and fin is true, the expected behavior is to consume
489 // the fin.
QUICHE teama6ef0a62019-03-07 20:34:33 -0500490 void CreateStreamFrame(QuicStreamId id,
QUICHE teamf08778a2019-03-14 08:10:26 -0700491 size_t data_size,
QUICHE teama6ef0a62019-03-07 20:34:33 -0500492 QuicStreamOffset offset,
493 bool fin,
494 QuicFrame* frame);
495
496 // Creates a CRYPTO frame which fits into the current open packet. Returns
497 // false if there isn't enough room in the current open packet for a CRYPTO
498 // frame, and true if there is.
499 bool CreateCryptoFrame(EncryptionLevel level,
500 size_t write_length,
501 QuicStreamOffset offset,
502 QuicFrame* frame);
503
504 void FillPacketHeader(QuicPacketHeader* header);
505
QUICHE teama6ef0a62019-03-07 20:34:33 -0500506 // Adds a padding frame to the current packet (if there is space) when (1)
507 // current packet needs full padding or (2) there are pending paddings.
508 void MaybeAddPadding();
509
510 // Serializes all frames which have been added and adds any which should be
511 // retransmitted to packet_.retransmittable_frames. All frames must fit into
fayang04590fc2020-08-31 10:37:55 -0700512 // a single packet. Returns true on success, otherwise, returns false.
513 // Fails if |encrypted_buffer| is not large enough for the encrypted packet.
514 QUIC_MUST_USE_RESULT bool SerializePacket(
515 QuicOwnedPacketBuffer encrypted_buffer,
516 size_t encrypted_buffer_len);
QUICHE teama6ef0a62019-03-07 20:34:33 -0500517
518 // Called after a new SerialiedPacket is created to call the delegate's
519 // OnSerializedPacket and reset state.
520 void OnSerializedPacket();
521
522 // Clears all fields of packet_ that should be cleared between serializations.
523 void ClearPacket();
524
fayang08750832019-10-24 11:25:34 -0700525 // Re-serialzes frames of ENCRYPTION_INITIAL packet in coalesced packet with
526 // the original packet's packet number and packet number length.
527 // |padding_size| indicates the size of necessary padding. Returns 0 if
528 // serialization fails.
529 size_t ReserializeInitialPacketInCoalescedPacket(
530 const SerializedPacket& packet,
531 size_t padding_size,
532 char* buffer,
533 size_t buffer_len);
534
renjietangdbe98342019-10-18 11:00:57 -0700535 // Tries to coalesce |frame| with the back of |queued_frames_|.
536 // Returns true on success.
537 bool MaybeCoalesceStreamFrame(const QuicStreamFrame& frame);
538
fayang2ab1e852019-11-04 11:24:36 -0800539 // Called to remove the soft max_packet_length and restores
540 // latched_hard_max_packet_length_ if the packet cannot accommodate a single
541 // frame. Returns true if the soft limit is successfully removed. Returns
542 // false if either there is no current soft limit or there are queued frames
543 // (such that the packet length cannot be changed).
544 bool RemoveSoftMaxPacketLength();
545
QUICHE teama6ef0a62019-03-07 20:34:33 -0500546 // Returns true if a diversification nonce should be included in the current
547 // packet's header.
548 bool IncludeNonceInPublicHeader() const;
549
550 // Returns true if version should be included in current packet's header.
551 bool IncludeVersionInHeader() const;
552
553 // Returns length of packet number to send over the wire.
554 // packet_.packet_number_length should never be read directly, use this
555 // function instead.
556 QuicPacketNumberLength GetPacketNumberLength() const;
557
nharper55fa6132019-05-07 19:37:21 -0700558 // Returns the size in bytes of the packet header.
559 size_t PacketHeaderSize() const;
560
QUICHE teama6ef0a62019-03-07 20:34:33 -0500561 // Returns whether the destination connection ID is sent over the wire.
562 QuicConnectionIdIncluded GetDestinationConnectionIdIncluded() const;
563
564 // Returns whether the source connection ID is sent over the wire.
565 QuicConnectionIdIncluded GetSourceConnectionIdIncluded() const;
566
567 // Returns length of the retry token variable length integer to send over the
568 // wire. Is non-zero for v99 IETF Initial packets.
569 QuicVariableLengthIntegerLength GetRetryTokenLengthLength() const;
570
571 // Returns the retry token to send over the wire, only sent in
572 // v99 IETF Initial packets.
dmcardlecf0bfcf2019-12-13 08:08:21 -0800573 quiche::QuicheStringPiece GetRetryToken() const;
QUICHE teama6ef0a62019-03-07 20:34:33 -0500574
575 // Returns length of the length variable length integer to send over the
576 // wire. Is non-zero for v99 IETF Initial, 0-RTT or Handshake packets.
577 QuicVariableLengthIntegerLength GetLengthLength() const;
578
ianswette28f0222019-04-04 13:31:22 -0700579 // Returns true if |frame| is a ClientHello.
580 bool StreamFrameIsClientHello(const QuicStreamFrame& frame) const;
QUICHE teama6ef0a62019-03-07 20:34:33 -0500581
582 // Returns true if packet under construction has IETF long header.
583 bool HasIetfLongHeader() const;
584
fayangfce2f722020-06-26 10:27:32 -0700585 // Get serialized frame length. Returns 0 if the frame does not fit into
586 // current packet.
587 size_t GetSerializedFrameLength(const QuicFrame& frame);
588
589 // Add extra padding to pending_padding_bytes_ to meet minimum plaintext
590 // packet size required for header protection.
591 void MaybeAddExtraPaddingForHeaderProtection();
592
fayang9320ca72020-08-03 13:02:59 -0700593 // Returns true and close connection if it attempts to send unencrypted data.
594 bool AttemptingToSendUnencryptedStreamData();
595
QUICHE teama6ef0a62019-03-07 20:34:33 -0500596 // Does not own these delegates or the framer.
597 DelegateInterface* delegate_;
598 DebugDelegate* debug_delegate_;
599 QuicFramer* framer_;
600 QuicRandom* random_;
601
602 // Controls whether version should be included while serializing the packet.
603 // send_version_in_packet_ should never be read directly, use
604 // IncludeVersionInHeader() instead.
605 bool send_version_in_packet_;
606 // If true, then |diversification_nonce_| will be included in the header of
607 // all packets created at the initial encryption level.
608 bool have_diversification_nonce_;
609 DiversificationNonce diversification_nonce_;
610 // Maximum length including headers and encryption (UDP payload length.)
611 QuicByteCount max_packet_length_;
612 size_t max_plaintext_size_;
dschinazi7b9278c2019-05-20 07:36:21 -0700613 // Whether the server_connection_id is sent over the wire.
614 QuicConnectionIdIncluded server_connection_id_included_;
QUICHE teama6ef0a62019-03-07 20:34:33 -0500615
616 // Frames to be added to the next SerializedPacket
617 QuicFrames queued_frames_;
618
fayang7a06f9b2020-06-24 10:23:19 -0700619 // Serialization size of header + frames. If there is no queued frames,
620 // packet_size_ is 0.
QUICHE teama6ef0a62019-03-07 20:34:33 -0500621 // TODO(ianswett): Move packet_size_ into SerializedPacket once
622 // QuicEncryptedPacket has been flattened into SerializedPacket.
623 size_t packet_size_;
dschinazi7b9278c2019-05-20 07:36:21 -0700624 QuicConnectionId server_connection_id_;
dschinazi346b7ce2019-06-05 01:38:18 -0700625 QuicConnectionId client_connection_id_;
QUICHE teama6ef0a62019-03-07 20:34:33 -0500626
627 // Packet used to invoke OnSerializedPacket.
628 SerializedPacket packet_;
629
630 // Retry token to send over the wire in v99 IETF Initial packets.
vasilvvc48c8712019-03-11 13:38:16 -0700631 std::string retry_token_;
QUICHE teama6ef0a62019-03-07 20:34:33 -0500632
633 // Pending padding bytes to send. Pending padding bytes will be sent in next
634 // packet(s) (after all other frames) if current constructed packet does not
635 // have room to send all of them.
636 QuicByteCount pending_padding_bytes_;
637
638 // Indicates whether current constructed packet needs full padding to max
639 // packet size. Please note, full padding does not consume pending padding
640 // bytes.
641 bool needs_full_padding_;
642
fayang18be79a2019-09-16 15:17:12 -0700643 // Transmission type of the next serialized packet.
644 TransmissionType next_transmission_type_;
645
646 // True if packet flusher is currently attached.
647 bool flusher_attached_;
648
649 // Whether crypto handshake packets should be fully padded.
650 bool fully_pad_crypto_handshake_packets_;
651
652 // Packet number of the first packet of a write operation. This gets set
653 // when the out-most flusher attaches and gets cleared when the out-most
654 // flusher detaches.
655 QuicPacketNumber write_start_packet_number_;
656
fayang2ab1e852019-11-04 11:24:36 -0800657 // If not 0, this latches the actual max_packet_length when
658 // SetSoftMaxPacketLength is called and max_packet_length_ gets
659 // set to a soft value.
660 QuicByteCount latched_hard_max_packet_length_;
dschinazied459c02020-05-07 16:12:23 -0700661
662 // The maximum length of a MESSAGE/DATAGRAM frame that our peer is willing to
663 // accept. There is no limit for QUIC_CRYPTO connections, but QUIC+TLS
664 // negotiates this during the handshake.
665 QuicByteCount max_datagram_frame_size_;
wub50d4c712020-05-19 15:48:28 -0700666
fayangfe963c52020-07-16 06:56:09 -0700667 const bool coalesced_packet_of_higher_space_ =
fayangdceb9a32020-07-29 10:57:04 -0700668 GetQuicReloadableFlag(quic_coalesced_packet_of_higher_space2);
fayang04590fc2020-08-31 10:37:55 -0700669
670 const bool close_connection_on_serialization_failure_ =
671 GetQuicReloadableFlag(quic_close_connection_on_serialization_failure);
QUICHE teama6ef0a62019-03-07 20:34:33 -0500672};
673
674} // namespace quic
675
676#endif // QUICHE_QUIC_CORE_QUIC_PACKET_CREATOR_H_