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