| 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 |