| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "net/quic/quic_packet_creator.h" | 5 #include "net/quic/quic_packet_creator.h" |
| 6 | 6 |
| 7 #include "base/logging.h" | 7 #include "base/logging.h" |
| 8 #include "net/quic/crypto/quic_random.h" | 8 #include "net/quic/crypto/quic_random.h" |
| 9 #include "net/quic/quic_fec_group.h" | 9 #include "net/quic/quic_fec_group.h" |
| 10 #include "net/quic/quic_utils.h" | 10 #include "net/quic/quic_utils.h" |
| (...skipping 12 matching lines...) Expand all Loading... |
| 23 bool is_server) | 23 bool is_server) |
| 24 : guid_(guid), | 24 : guid_(guid), |
| 25 framer_(framer), | 25 framer_(framer), |
| 26 random_generator_(random_generator), | 26 random_generator_(random_generator), |
| 27 sequence_number_(0), | 27 sequence_number_(0), |
| 28 fec_group_number_(0), | 28 fec_group_number_(0), |
| 29 is_server_(is_server), | 29 is_server_(is_server), |
| 30 send_version_in_packet_(!is_server), | 30 send_version_in_packet_(!is_server), |
| 31 packet_size_(GetPacketHeaderSize(options_.send_guid_length, | 31 packet_size_(GetPacketHeaderSize(options_.send_guid_length, |
| 32 send_version_in_packet_, | 32 send_version_in_packet_, |
| 33 options_.send_sequence_number_length, | |
| 34 NOT_IN_FEC_GROUP)) { | 33 NOT_IN_FEC_GROUP)) { |
| 35 framer_->set_fec_builder(this); | 34 framer_->set_fec_builder(this); |
| 36 } | 35 } |
| 37 | 36 |
| 38 QuicPacketCreator::~QuicPacketCreator() { | 37 QuicPacketCreator::~QuicPacketCreator() { |
| 39 } | 38 } |
| 40 | 39 |
| 41 void QuicPacketCreator::OnBuiltFecProtectedPayload( | 40 void QuicPacketCreator::OnBuiltFecProtectedPayload( |
| 42 const QuicPacketHeader& header, StringPiece payload) { | 41 const QuicPacketHeader& header, StringPiece payload) { |
| 43 if (fec_group_.get()) { | 42 if (fec_group_.get()) { |
| 44 fec_group_->Update(header, payload); | 43 fec_group_->Update(header, payload); |
| 45 } | 44 } |
| 46 } | 45 } |
| 47 | 46 |
| 48 bool QuicPacketCreator::ShouldSendFec(bool force_close) const { | 47 bool QuicPacketCreator::ShouldSendFec(bool force_close) const { |
| 49 return fec_group_.get() != NULL && fec_group_->NumReceivedPackets() > 0 && | 48 return fec_group_.get() != NULL && fec_group_->NumReceivedPackets() > 0 && |
| 50 (force_close || | 49 (force_close || |
| 51 fec_group_->NumReceivedPackets() >= options_.max_packets_per_fec_group); | 50 fec_group_->NumReceivedPackets() >= options_.max_packets_per_fec_group); |
| 52 } | 51 } |
| 53 | 52 |
| 54 void QuicPacketCreator::MaybeStartFEC() { | 53 void QuicPacketCreator::MaybeStartFEC() { |
| 55 if (options_.max_packets_per_fec_group > 0 && fec_group_.get() == NULL) { | 54 if (options_.max_packets_per_fec_group > 0 && fec_group_.get() == NULL) { |
| 56 DCHECK(queued_frames_.empty()); | 55 DCHECK(queued_frames_.empty()); |
| 57 // Set the fec group number to the sequence number of the next packet. | 56 // Set the fec group number to the sequence number of the next packet. |
| 58 fec_group_number_ = sequence_number() + 1; | 57 fec_group_number_ = sequence_number() + 1; |
| 59 fec_group_.reset(new QuicFecGroup()); | 58 fec_group_.reset(new QuicFecGroup()); |
| 60 packet_size_ = GetPacketHeaderSize(options_.send_guid_length, | 59 packet_size_ = GetPacketHeaderSize(options_.send_guid_length, |
| 61 send_version_in_packet_, | 60 send_version_in_packet_, |
| 62 options_.send_sequence_number_length, | |
| 63 IN_FEC_GROUP); | 61 IN_FEC_GROUP); |
| 64 DCHECK_LE(packet_size_, options_.max_packet_length); | 62 DCHECK_LE(packet_size_, options_.max_packet_length); |
| 65 } | 63 } |
| 66 } | 64 } |
| 67 | 65 |
| 68 // Stops serializing version of the protocol in packets sent after this call. | 66 // Stops serializing version of the protocol in packets sent after this call. |
| 69 // A packet that is already open might send kQuicVersionSize bytes less than the | 67 // A packet that is already open might send kQuicVersionSize bytes less than the |
| 70 // maximum packet size if we stop sending version before it is serialized. | 68 // maximum packet size if we stop sending version before it is serialized. |
| 71 void QuicPacketCreator::StopSendingVersion() { | 69 void QuicPacketCreator::StopSendingVersion() { |
| 72 DCHECK(send_version_in_packet_); | 70 DCHECK(send_version_in_packet_); |
| 73 send_version_in_packet_ = false; | 71 send_version_in_packet_ = false; |
| 74 if (packet_size_ > 0) { | 72 if (packet_size_ > 0) { |
| 75 DCHECK_LT(kQuicVersionSize, packet_size_); | 73 DCHECK_LT(kQuicVersionSize, packet_size_); |
| 76 packet_size_ -= kQuicVersionSize; | 74 packet_size_ -= kQuicVersionSize; |
| 77 } | 75 } |
| 78 } | 76 } |
| 79 | 77 |
| 80 bool QuicPacketCreator::HasRoomForStreamFrame() const { | 78 bool QuicPacketCreator::HasRoomForStreamFrame() const { |
| 81 return BytesFree() > QuicFramer::GetMinStreamFrameSize(); | 79 return BytesFree() > QuicFramer::GetMinStreamFrameSize(); |
| 82 } | 80 } |
| 83 | 81 |
| 84 // static | 82 // static |
| 85 size_t QuicPacketCreator::StreamFramePacketOverhead( | 83 size_t QuicPacketCreator::StreamFramePacketOverhead( |
| 86 int num_frames, | 84 int num_frames, |
| 87 QuicGuidLength guid_length, | 85 QuicGuidLength guid_length, |
| 88 bool include_version, | 86 bool include_version, |
| 89 QuicSequenceNumberLength sequence_number_length, | |
| 90 InFecGroup is_in_fec_group) { | 87 InFecGroup is_in_fec_group) { |
| 91 return GetPacketHeaderSize(guid_length, include_version, | 88 return GetPacketHeaderSize(guid_length, include_version, is_in_fec_group) + |
| 92 sequence_number_length, is_in_fec_group) + | |
| 93 QuicFramer::GetMinStreamFrameSize() * num_frames; | 89 QuicFramer::GetMinStreamFrameSize() * num_frames; |
| 94 } | 90 } |
| 95 | 91 |
| 96 size_t QuicPacketCreator::CreateStreamFrame(QuicStreamId id, | 92 size_t QuicPacketCreator::CreateStreamFrame(QuicStreamId id, |
| 97 StringPiece data, | 93 StringPiece data, |
| 98 QuicStreamOffset offset, | 94 QuicStreamOffset offset, |
| 99 bool fin, | 95 bool fin, |
| 100 QuicFrame* frame) { | 96 QuicFrame* frame) { |
| 101 DCHECK_GT(options_.max_packet_length, | 97 DCHECK_GT(options_.max_packet_length, |
| 102 StreamFramePacketOverhead( | 98 StreamFramePacketOverhead( |
| 103 1, PACKET_8BYTE_GUID, kIncludeVersion, | 99 1, PACKET_8BYTE_GUID, kIncludeVersion, IN_FEC_GROUP)); |
| 104 PACKET_6BYTE_SEQUENCE_NUMBER, IN_FEC_GROUP)); | |
| 105 DCHECK(HasRoomForStreamFrame()); | 100 DCHECK(HasRoomForStreamFrame()); |
| 106 | 101 |
| 107 const size_t free_bytes = BytesFree(); | 102 const size_t free_bytes = BytesFree(); |
| 108 size_t bytes_consumed = 0; | 103 size_t bytes_consumed = 0; |
| 109 | 104 |
| 110 if (data.size() != 0) { | 105 if (data.size() != 0) { |
| 111 size_t max_data_len = free_bytes - QuicFramer::GetMinStreamFrameSize(); | 106 size_t max_data_len = free_bytes - QuicFramer::GetMinStreamFrameSize(); |
| 112 bytes_consumed = min<size_t>(max_data_len, data.size()); | 107 bytes_consumed = min<size_t>(max_data_len, data.size()); |
| 113 | 108 |
| 114 bool set_fin = fin && bytes_consumed == data.size(); // Last frame. | 109 bool set_fin = fin && bytes_consumed == data.size(); // Last frame. |
| (...skipping 43 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 158 SerializedPacket QuicPacketCreator::SerializePacket() { | 153 SerializedPacket QuicPacketCreator::SerializePacket() { |
| 159 DCHECK_EQ(false, queued_frames_.empty()); | 154 DCHECK_EQ(false, queued_frames_.empty()); |
| 160 QuicPacketHeader header; | 155 QuicPacketHeader header; |
| 161 FillPacketHeader(fec_group_number_, false, false, &header); | 156 FillPacketHeader(fec_group_number_, false, false, &header); |
| 162 | 157 |
| 163 SerializedPacket serialized = framer_->ConstructFrameDataPacket( | 158 SerializedPacket serialized = framer_->ConstructFrameDataPacket( |
| 164 header, queued_frames_, packet_size_); | 159 header, queued_frames_, packet_size_); |
| 165 queued_frames_.clear(); | 160 queued_frames_.clear(); |
| 166 packet_size_ = GetPacketHeaderSize(options_.send_guid_length, | 161 packet_size_ = GetPacketHeaderSize(options_.send_guid_length, |
| 167 send_version_in_packet_, | 162 send_version_in_packet_, |
| 168 options_.send_sequence_number_length, | |
| 169 fec_group_.get() != NULL ? | 163 fec_group_.get() != NULL ? |
| 170 IN_FEC_GROUP : NOT_IN_FEC_GROUP); | 164 IN_FEC_GROUP : NOT_IN_FEC_GROUP); |
| 171 serialized.retransmittable_frames = queued_retransmittable_frames_.release(); | 165 serialized.retransmittable_frames = queued_retransmittable_frames_.release(); |
| 172 return serialized; | 166 return serialized; |
| 173 } | 167 } |
| 174 | 168 |
| 175 SerializedPacket QuicPacketCreator::SerializeFec() { | 169 SerializedPacket QuicPacketCreator::SerializeFec() { |
| 176 DCHECK_LT(0u, fec_group_->NumReceivedPackets()); | 170 DCHECK_LT(0u, fec_group_->NumReceivedPackets()); |
| 177 DCHECK_EQ(0u, queued_frames_.size()); | 171 DCHECK_EQ(0u, queued_frames_.size()); |
| 178 QuicPacketHeader header; | 172 QuicPacketHeader header; |
| 179 FillPacketHeader(fec_group_number_, true, | 173 FillPacketHeader(fec_group_number_, true, |
| 180 fec_group_->entropy_parity(), &header); | 174 fec_group_->entropy_parity(), &header); |
| 181 QuicFecData fec_data; | 175 QuicFecData fec_data; |
| 182 fec_data.fec_group = fec_group_->min_protected_packet(); | 176 fec_data.fec_group = fec_group_->min_protected_packet(); |
| 183 fec_data.redundancy = fec_group_->payload_parity(); | 177 fec_data.redundancy = fec_group_->payload_parity(); |
| 184 SerializedPacket serialized = framer_->ConstructFecPacket(header, fec_data); | 178 SerializedPacket serialized = framer_->ConstructFecPacket(header, fec_data); |
| 185 fec_group_.reset(NULL); | 179 fec_group_.reset(NULL); |
| 186 fec_group_number_ = 0; | 180 fec_group_number_ = 0; |
| 187 // Reset packet_size_, since the next packet may not have an FEC group. | 181 // Reset packet_size_, since the next packet may not have an FEC group. |
| 188 packet_size_ = GetPacketHeaderSize(options_.send_guid_length, | 182 packet_size_ = GetPacketHeaderSize(options_.send_guid_length, |
| 189 send_version_in_packet_, | 183 send_version_in_packet_, |
| 190 options_.send_sequence_number_length, | |
| 191 NOT_IN_FEC_GROUP); | 184 NOT_IN_FEC_GROUP); |
| 192 DCHECK(serialized.packet); | 185 DCHECK(serialized.packet); |
| 193 DCHECK_GE(options_.max_packet_length, serialized.packet->length()); | 186 DCHECK_GE(options_.max_packet_length, serialized.packet->length()); |
| 194 return serialized; | 187 return serialized; |
| 195 } | 188 } |
| 196 | 189 |
| 197 SerializedPacket QuicPacketCreator::SerializeConnectionClose( | 190 SerializedPacket QuicPacketCreator::SerializeConnectionClose( |
| 198 QuicConnectionCloseFrame* close_frame) { | 191 QuicConnectionCloseFrame* close_frame) { |
| 199 QuicFrames frames; | 192 QuicFrames frames; |
| 200 frames.push_back(QuicFrame(close_frame)); | 193 frames.push_back(QuicFrame(close_frame)); |
| (...skipping 68 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 269 queued_frames_.push_back( | 262 queued_frames_.push_back( |
| 270 queued_retransmittable_frames_->AddNonStreamFrame(frame)); | 263 queued_retransmittable_frames_->AddNonStreamFrame(frame)); |
| 271 } | 264 } |
| 272 } else { | 265 } else { |
| 273 queued_frames_.push_back(frame); | 266 queued_frames_.push_back(frame); |
| 274 } | 267 } |
| 275 return true; | 268 return true; |
| 276 } | 269 } |
| 277 | 270 |
| 278 } // namespace net | 271 } // namespace net |
| OLD | NEW |