| 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/basictypes.h" | 7 #include "base/basictypes.h" |
| 8 #include "base/logging.h" | 8 #include "base/logging.h" |
| 9 #include "net/quic/crypto/quic_random.h" | 9 #include "net/quic/crypto/quic_random.h" |
| 10 #include "net/quic/quic_ack_notifier.h" | 10 #include "net/quic/quic_ack_notifier.h" |
| (...skipping 37 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 48 // Source of entropy. | 48 // Source of entropy. |
| 49 QuicRandom* random_; | 49 QuicRandom* random_; |
| 50 // Stored random bits. | 50 // Stored random bits. |
| 51 uint64 bit_bucket_; | 51 uint64 bit_bucket_; |
| 52 // The next available bit has "1" in the mask. Zero means empty bucket. | 52 // The next available bit has "1" in the mask. Zero means empty bucket. |
| 53 uint64 bit_mask_; | 53 uint64 bit_mask_; |
| 54 | 54 |
| 55 DISALLOW_COPY_AND_ASSIGN(QuicRandomBoolSource); | 55 DISALLOW_COPY_AND_ASSIGN(QuicRandomBoolSource); |
| 56 }; | 56 }; |
| 57 | 57 |
| 58 QuicPacketCreator::QuicPacketCreator(QuicGuid guid, | 58 QuicPacketCreator::QuicPacketCreator(QuicConnectionId connection_id, |
| 59 QuicFramer* framer, | 59 QuicFramer* framer, |
| 60 QuicRandom* random_generator, | 60 QuicRandom* random_generator, |
| 61 bool is_server) | 61 bool is_server) |
| 62 : guid_(guid), | 62 : connection_id_(connection_id), |
| 63 framer_(framer), | 63 framer_(framer), |
| 64 random_bool_source_(new QuicRandomBoolSource(random_generator)), | 64 random_bool_source_(new QuicRandomBoolSource(random_generator)), |
| 65 sequence_number_(0), | 65 sequence_number_(0), |
| 66 fec_group_number_(0), | 66 fec_group_number_(0), |
| 67 is_server_(is_server), | 67 is_server_(is_server), |
| 68 send_version_in_packet_(!is_server), | 68 send_version_in_packet_(!is_server), |
| 69 sequence_number_length_(options_.send_sequence_number_length), | 69 sequence_number_length_(options_.send_sequence_number_length), |
| 70 packet_size_(0) { | 70 packet_size_(0) { |
| 71 framer_->set_fec_builder(this); | 71 framer_->set_fec_builder(this); |
| 72 } | 72 } |
| (...skipping 57 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 130 | 130 |
| 131 bool QuicPacketCreator::HasRoomForStreamFrame(QuicStreamId id, | 131 bool QuicPacketCreator::HasRoomForStreamFrame(QuicStreamId id, |
| 132 QuicStreamOffset offset) const { | 132 QuicStreamOffset offset) const { |
| 133 return BytesFree() > | 133 return BytesFree() > |
| 134 QuicFramer::GetMinStreamFrameSize(framer_->version(), id, offset, true); | 134 QuicFramer::GetMinStreamFrameSize(framer_->version(), id, offset, true); |
| 135 } | 135 } |
| 136 | 136 |
| 137 // static | 137 // static |
| 138 size_t QuicPacketCreator::StreamFramePacketOverhead( | 138 size_t QuicPacketCreator::StreamFramePacketOverhead( |
| 139 QuicVersion version, | 139 QuicVersion version, |
| 140 QuicGuidLength guid_length, | 140 QuicConnectionIdLength connection_id_length, |
| 141 bool include_version, | 141 bool include_version, |
| 142 QuicSequenceNumberLength sequence_number_length, | 142 QuicSequenceNumberLength sequence_number_length, |
| 143 InFecGroup is_in_fec_group) { | 143 InFecGroup is_in_fec_group) { |
| 144 return GetPacketHeaderSize(guid_length, include_version, | 144 return GetPacketHeaderSize(connection_id_length, include_version, |
| 145 sequence_number_length, is_in_fec_group) + | 145 sequence_number_length, is_in_fec_group) + |
| 146 // Assumes this is a stream with a single lone packet. | 146 // Assumes this is a stream with a single lone packet. |
| 147 QuicFramer::GetMinStreamFrameSize(version, 1u, 0u, true); | 147 QuicFramer::GetMinStreamFrameSize(version, 1u, 0u, true); |
| 148 } | 148 } |
| 149 | 149 |
| 150 size_t QuicPacketCreator::CreateStreamFrame(QuicStreamId id, | 150 size_t QuicPacketCreator::CreateStreamFrame(QuicStreamId id, |
| 151 const IOVector& data, | 151 const IOVector& data, |
| 152 QuicStreamOffset offset, | 152 QuicStreamOffset offset, |
| 153 bool fin, | 153 bool fin, |
| 154 QuicFrame* frame) { | 154 QuicFrame* frame) { |
| 155 DCHECK_GT(options_.max_packet_length, | 155 DCHECK_GT(options_.max_packet_length, |
| 156 StreamFramePacketOverhead( | 156 StreamFramePacketOverhead( |
| 157 framer_->version(), PACKET_8BYTE_GUID, kIncludeVersion, | 157 framer_->version(), PACKET_8BYTE_CONNECTION_ID, kIncludeVersion, |
| 158 PACKET_6BYTE_SEQUENCE_NUMBER, IN_FEC_GROUP)); | 158 PACKET_6BYTE_SEQUENCE_NUMBER, IN_FEC_GROUP)); |
| 159 if (!HasRoomForStreamFrame(id, offset)) { | 159 if (!HasRoomForStreamFrame(id, offset)) { |
| 160 LOG(DFATAL) << "No room for Stream frame, BytesFree: " << BytesFree() | 160 LOG(DFATAL) << "No room for Stream frame, BytesFree: " << BytesFree() |
| 161 << " MinStreamFrameSize: " | 161 << " MinStreamFrameSize: " |
| 162 << QuicFramer::GetMinStreamFrameSize( | 162 << QuicFramer::GetMinStreamFrameSize( |
| 163 framer_->version(), id, offset, true); | 163 framer_->version(), id, offset, true); |
| 164 } | 164 } |
| 165 | 165 |
| 166 if (data.Empty()) { | 166 if (data.Empty()) { |
| 167 if (!fin) { | 167 if (!fin) { |
| (...skipping 124 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 292 } | 292 } |
| 293 | 293 |
| 294 size_t QuicPacketCreator::PacketSize() const { | 294 size_t QuicPacketCreator::PacketSize() const { |
| 295 if (queued_frames_.empty()) { | 295 if (queued_frames_.empty()) { |
| 296 // Only adjust the sequence number length when the FEC group is not open, | 296 // Only adjust the sequence number length when the FEC group is not open, |
| 297 // to ensure no packets in a group are too large. | 297 // to ensure no packets in a group are too large. |
| 298 if (fec_group_.get() == NULL || | 298 if (fec_group_.get() == NULL || |
| 299 fec_group_->NumReceivedPackets() == 0) { | 299 fec_group_->NumReceivedPackets() == 0) { |
| 300 sequence_number_length_ = options_.send_sequence_number_length; | 300 sequence_number_length_ = options_.send_sequence_number_length; |
| 301 } | 301 } |
| 302 packet_size_ = GetPacketHeaderSize(options_.send_guid_length, | 302 packet_size_ = GetPacketHeaderSize(options_.send_connection_id_length, |
| 303 send_version_in_packet_, | 303 send_version_in_packet_, |
| 304 sequence_number_length_, | 304 sequence_number_length_, |
| 305 options_.max_packets_per_fec_group == 0 ? | 305 options_.max_packets_per_fec_group == 0 ? |
| 306 NOT_IN_FEC_GROUP : IN_FEC_GROUP); | 306 NOT_IN_FEC_GROUP : IN_FEC_GROUP); |
| 307 } | 307 } |
| 308 return packet_size_; | 308 return packet_size_; |
| 309 } | 309 } |
| 310 | 310 |
| 311 bool QuicPacketCreator::AddSavedFrame(const QuicFrame& frame) { | 311 bool QuicPacketCreator::AddSavedFrame(const QuicFrame& frame) { |
| 312 return AddFrame(frame, true); | 312 return AddFrame(frame, true); |
| (...skipping 60 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 373 QuicConnectionCloseFrame* close_frame) { | 373 QuicConnectionCloseFrame* close_frame) { |
| 374 QuicFrames frames; | 374 QuicFrames frames; |
| 375 frames.push_back(QuicFrame(close_frame)); | 375 frames.push_back(QuicFrame(close_frame)); |
| 376 return SerializeAllFrames(frames); | 376 return SerializeAllFrames(frames); |
| 377 } | 377 } |
| 378 | 378 |
| 379 QuicEncryptedPacket* QuicPacketCreator::SerializeVersionNegotiationPacket( | 379 QuicEncryptedPacket* QuicPacketCreator::SerializeVersionNegotiationPacket( |
| 380 const QuicVersionVector& supported_versions) { | 380 const QuicVersionVector& supported_versions) { |
| 381 DCHECK(is_server_); | 381 DCHECK(is_server_); |
| 382 QuicPacketPublicHeader header; | 382 QuicPacketPublicHeader header; |
| 383 header.guid = guid_; | 383 header.connection_id = connection_id_; |
| 384 header.reset_flag = false; | 384 header.reset_flag = false; |
| 385 header.version_flag = true; | 385 header.version_flag = true; |
| 386 header.versions = supported_versions; | 386 header.versions = supported_versions; |
| 387 QuicEncryptedPacket* encrypted = | 387 QuicEncryptedPacket* encrypted = |
| 388 framer_->BuildVersionNegotiationPacket(header, supported_versions); | 388 framer_->BuildVersionNegotiationPacket(header, supported_versions); |
| 389 DCHECK(encrypted); | 389 DCHECK(encrypted); |
| 390 DCHECK_GE(options_.max_packet_length, encrypted->length()); | 390 DCHECK_GE(options_.max_packet_length, encrypted->length()); |
| 391 return encrypted; | 391 return encrypted; |
| 392 } | 392 } |
| 393 | 393 |
| 394 void QuicPacketCreator::FillPacketHeader(QuicFecGroupNumber fec_group, | 394 void QuicPacketCreator::FillPacketHeader(QuicFecGroupNumber fec_group, |
| 395 bool fec_flag, | 395 bool fec_flag, |
| 396 QuicPacketHeader* header) { | 396 QuicPacketHeader* header) { |
| 397 header->public_header.guid = guid_; | 397 header->public_header.connection_id = connection_id_; |
| 398 header->public_header.reset_flag = false; | 398 header->public_header.reset_flag = false; |
| 399 header->public_header.version_flag = send_version_in_packet_; | 399 header->public_header.version_flag = send_version_in_packet_; |
| 400 header->fec_flag = fec_flag; | 400 header->fec_flag = fec_flag; |
| 401 header->packet_sequence_number = ++sequence_number_; | 401 header->packet_sequence_number = ++sequence_number_; |
| 402 header->public_header.sequence_number_length = sequence_number_length_; | 402 header->public_header.sequence_number_length = sequence_number_length_; |
| 403 header->entropy_flag = random_bool_source_->RandBool(); | 403 header->entropy_flag = random_bool_source_->RandBool(); |
| 404 header->is_in_fec_group = fec_group == 0 ? NOT_IN_FEC_GROUP : IN_FEC_GROUP; | 404 header->is_in_fec_group = fec_group == 0 ? NOT_IN_FEC_GROUP : IN_FEC_GROUP; |
| 405 header->fec_group = fec_group; | 405 header->fec_group = fec_group; |
| 406 } | 406 } |
| 407 | 407 |
| (...skipping 62 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 470 if (!is_handshake) { | 470 if (!is_handshake) { |
| 471 return; | 471 return; |
| 472 } | 472 } |
| 473 | 473 |
| 474 QuicPaddingFrame padding; | 474 QuicPaddingFrame padding; |
| 475 bool success = AddFrame(QuicFrame(&padding), false); | 475 bool success = AddFrame(QuicFrame(&padding), false); |
| 476 DCHECK(success); | 476 DCHECK(success); |
| 477 } | 477 } |
| 478 | 478 |
| 479 } // namespace net | 479 } // namespace net |
| OLD | NEW |