quic_packet_creator.cc revision f8ee788a64d60abd8f2d742a5fdedde054ecd910
15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2012 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/quic/quic_packet_creator.h" 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 75d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)#include "base/basictypes.h" 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/logging.h" 92a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "net/quic/crypto/quic_random.h" 1058537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles)#include "net/quic/quic_ack_notifier.h" 112a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "net/quic/quic_fec_group.h" 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/quic/quic_utils.h" 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using base::StringPiece; 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using std::make_pair; 16d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles)using std::max; 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using std::min; 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using std::pair; 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using std::vector; 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace net { 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 235d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)// A QuicRandom wrapper that gets a bucket of entropy and distributes it 245d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)// bit-by-bit. Replenishes the bucket as needed. Not thread-safe. Expose this 255d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)// class if single bit randomness is needed elsewhere. 265d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)class QuicRandomBoolSource { 275d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) public: 285d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) // random: Source of entropy. Not owned. 295d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) explicit QuicRandomBoolSource(QuicRandom* random) 305d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) : random_(random), 315d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) bit_bucket_(0), 325d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) bit_mask_(0) {} 335d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 345d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) ~QuicRandomBoolSource() {} 355d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 365d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) // Returns the next random bit from the bucket. 375d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) bool RandBool() { 385d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) if (bit_mask_ == 0) { 395d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) bit_bucket_ = random_->RandUint64(); 405d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) bit_mask_ = 1; 415d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) } 425d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) bool result = ((bit_bucket_ & bit_mask_) != 0); 435d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) bit_mask_ <<= 1; 445d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) return result; 455d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) } 465d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 475d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) private: 485d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) // Source of entropy. 495d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) QuicRandom* random_; 505d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) // Stored random bits. 515d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) uint64 bit_bucket_; 525d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) // The next available bit has "1" in the mask. Zero means empty bucket. 535d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) uint64 bit_mask_; 545d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 555d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) DISALLOW_COPY_AND_ASSIGN(QuicRandomBoolSource); 565d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)}; 575d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 58a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)QuicPacketCreator::QuicPacketCreator(QuicConnectionId connection_id, 592a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) QuicFramer* framer, 60f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) QuicRandom* random_generator) 61a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) : connection_id_(connection_id), 62010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) encryption_level_(ENCRYPTION_NONE), 635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) framer_(framer), 645d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) random_bool_source_(new QuicRandomBoolSource(random_generator)), 655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) sequence_number_(0), 66cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) should_fec_protect_(false), 672a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) fec_group_number_(0), 68f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) send_version_in_packet_(!framer->is_server()), 6946d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) max_packet_length_(kDefaultMaxPacketSize), 7046d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) max_packets_per_fec_group_(0), 7146d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) connection_id_length_(PACKET_8BYTE_CONNECTION_ID), 7246d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) next_sequence_number_length_(PACKET_1BYTE_SEQUENCE_NUMBER), 7346d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) sequence_number_length_(next_sequence_number_length_), 743551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) packet_size_(0) { 75d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) framer_->set_fec_builder(this); 765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)QuicPacketCreator::~QuicPacketCreator() { 795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void QuicPacketCreator::OnBuiltFecProtectedPayload( 822a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const QuicPacketHeader& header, StringPiece payload) { 835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (fec_group_.get()) { 8468043e1e95eeb07d5cae7aca370b26518b0867d6Torne (Richard Coles) DCHECK_NE(0u, header.fec_group); 85010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) fec_group_->Update(encryption_level_, header, payload); 865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 892a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)bool QuicPacketCreator::ShouldSendFec(bool force_close) const { 90f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) DCHECK(!HasPendingFrames()); 91868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) return fec_group_.get() != NULL && fec_group_->NumReceivedPackets() > 0 && 92cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) (force_close || fec_group_->NumReceivedPackets() >= 9346d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) max_packets_per_fec_group_); 942a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 96f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)bool QuicPacketCreator::IsFecGroupOpen() const { 97f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) return ShouldSendFec(true); 98f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)} 99f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 100cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles)void QuicPacketCreator::StartFecProtectingPackets() { 101cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) if (!IsFecEnabled()) { 102cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) LOG(DFATAL) << "Cannot start FEC protection when FEC is not enabled."; 103cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) return; 1042a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 105cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) // TODO(jri): This currently requires that the generator flush out any 106cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) // pending frames when FEC protection is turned on. If current packet can be 107cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) // converted to an FEC protected packet, do it. This will require the 108cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) // generator to check if the resulting expansion still allows the incoming 109cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) // frame to be added to the packet. 110cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) if (HasPendingFrames()) { 111cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) LOG(DFATAL) << "Cannot start FEC protection with pending frames."; 112cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) return; 113cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) } 114cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) DCHECK(!should_fec_protect_); 115cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) should_fec_protect_ = true; 116cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles)} 117cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) 118cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles)void QuicPacketCreator::StopFecProtectingPackets() { 119cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) if (fec_group_.get() != NULL) { 120cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) LOG(DFATAL) << "Cannot stop FEC protection with open FEC group."; 121cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) return; 122cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) } 123cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) DCHECK(should_fec_protect_); 124cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) should_fec_protect_ = false; 125cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) fec_group_number_ = 0; 126cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles)} 127cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) 128cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles)bool QuicPacketCreator::IsFecProtected() const { 129cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) return should_fec_protect_; 130cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles)} 131cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) 132cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles)bool QuicPacketCreator::IsFecEnabled() const { 13346d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) return max_packets_per_fec_group_ > 0; 134cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles)} 135cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) 136cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles)InFecGroup QuicPacketCreator::MaybeUpdateLengthsAndStartFec() { 137cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) if (fec_group_.get() != NULL) { 138cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) // Don't update any lengths when an FEC group is open, to ensure same 139cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) // packet header size in all packets within a group. 140cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) return IN_FEC_GROUP; 141cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) } 142cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) if (!queued_frames_.empty()) { 143cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) // Don't change creator state if there are frames queued. 144cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) return fec_group_.get() == NULL ? NOT_IN_FEC_GROUP : IN_FEC_GROUP; 145cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) } 14646d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) 14746d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) // Update sequence number length only on packet and FEC group boundaries. 14846d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) sequence_number_length_ = next_sequence_number_length_; 149cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) 150cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) if (!should_fec_protect_) { 151cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) return NOT_IN_FEC_GROUP; 152cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) } 153cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) // Start a new FEC group since protection is on. Set the fec group number to 154cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) // the sequence number of the next packet. 155cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) fec_group_number_ = sequence_number() + 1; 156cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) fec_group_.reset(new QuicFecGroup()); 157cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) return IN_FEC_GROUP; 1582a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 1595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Stops serializing version of the protocol in packets sent after this call. 1612a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// A packet that is already open might send kQuicVersionSize bytes less than the 1622a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// maximum packet size if we stop sending version before it is serialized. 1632a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void QuicPacketCreator::StopSendingVersion() { 1642a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK(send_version_in_packet_); 1652a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) send_version_in_packet_ = false; 1662a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (packet_size_ > 0) { 1672a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK_LT(kQuicVersionSize, packet_size_); 1682a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) packet_size_ -= kQuicVersionSize; 1695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1702a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 1712a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 172d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles)void QuicPacketCreator::UpdateSequenceNumberLength( 173d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) QuicPacketSequenceNumber least_packet_awaited_by_peer, 1740529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch QuicByteCount congestion_window) { 175d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) DCHECK_LE(least_packet_awaited_by_peer, sequence_number_ + 1); 176d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) // Since the packet creator will not change sequence number length mid FEC 177d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) // group, include the size of an FEC group to be safe. 178d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) const QuicPacketSequenceNumber current_delta = 17946d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) max_packets_per_fec_group_ + sequence_number_ + 1 180d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) - least_packet_awaited_by_peer; 1810529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch const uint64 congestion_window_packets = 18246d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) congestion_window / max_packet_length_; 1830529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch const uint64 delta = max(current_delta, congestion_window_packets); 18446d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) next_sequence_number_length_ = 1850f1bc08d4cfcc34181b0b5cbf065c40f687bf740Torne (Richard Coles) QuicFramer::GetMinSequenceNumberLength(delta * 4); 186d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles)} 187d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) 1882385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdochbool QuicPacketCreator::HasRoomForStreamFrame(QuicStreamId id, 1892385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch QuicStreamOffset offset) const { 190c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch // TODO(jri): This is a simple safe decision for now, but make 191c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch // is_in_fec_group a parameter. Same as with all public methods in 192c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch // QuicPacketCreator. 1932385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch return BytesFree() > 194c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch QuicFramer::GetMinStreamFrameSize(framer_->version(), id, offset, true, 195cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) should_fec_protect_ ? IN_FEC_GROUP : 196cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) NOT_IN_FEC_GROUP); 1972a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 1982a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1992a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// static 200868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)size_t QuicPacketCreator::StreamFramePacketOverhead( 2012385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch QuicVersion version, 202a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) QuicConnectionIdLength connection_id_length, 203868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) bool include_version, 204868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) QuicSequenceNumberLength sequence_number_length, 205cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) QuicStreamOffset offset, 206868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) InFecGroup is_in_fec_group) { 207a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) return GetPacketHeaderSize(connection_id_length, include_version, 208868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) sequence_number_length, is_in_fec_group) + 2097dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch // Assumes this is a stream with a single lone packet. 210cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) QuicFramer::GetMinStreamFrameSize(version, 1u, offset, true, 211cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) is_in_fec_group); 2122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 2132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)size_t QuicPacketCreator::CreateStreamFrame(QuicStreamId id, 215f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) const IOVector& data, 2162a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) QuicStreamOffset offset, 2172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) bool fin, 2182a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) QuicFrame* frame) { 21946d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) DCHECK_GT(max_packet_length_, StreamFramePacketOverhead( 220a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) framer_->version(), PACKET_8BYTE_CONNECTION_ID, kIncludeVersion, 221cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) PACKET_6BYTE_SEQUENCE_NUMBER, offset, IN_FEC_GROUP)); 222cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) 223cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) InFecGroup is_in_fec_group = MaybeUpdateLengthsAndStartFec(); 224c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 225a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) LOG_IF(DFATAL, !HasRoomForStreamFrame(id, offset)) 226a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) << "No room for Stream frame, BytesFree: " << BytesFree() 227a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) << " MinStreamFrameSize: " 228a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) << QuicFramer::GetMinStreamFrameSize( 229c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch framer_->version(), id, offset, true, is_in_fec_group); 2302a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 231f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) if (data.Empty()) { 232a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) LOG_IF(DFATAL, !fin) 233a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) << "Creating a stream frame with no data or fin."; 2341e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) // Create a new packet for the fin, if necessary. 235f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) *frame = QuicFrame(new QuicStreamFrame(id, true, offset, data)); 2361e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) return 0; 2371e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) } 2381e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) 239f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) const size_t data_size = data.TotalBufferSize(); 240c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch size_t min_frame_size = QuicFramer::GetMinStreamFrameSize( 241c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch framer_->version(), id, offset, /*last_frame_in_packet=*/ true, 242c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch is_in_fec_group); 243c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch size_t bytes_consumed = min<size_t>(BytesFree() - min_frame_size, data_size); 2445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 245f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) bool set_fin = fin && bytes_consumed == data_size; // Last frame. 246f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) IOVector frame_data; 247f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) frame_data.AppendIovecAtMostBytes(data.iovec(), data.Size(), 248f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) bytes_consumed); 249f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) DCHECK_EQ(frame_data.TotalBufferSize(), bytes_consumed); 250f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) *frame = QuicFrame(new QuicStreamFrame(id, set_fin, offset, frame_data)); 2512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return bytes_consumed; 2522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 2535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 25458537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles)size_t QuicPacketCreator::CreateStreamFrameWithNotifier( 25558537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) QuicStreamId id, 256f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) const IOVector& data, 25758537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) QuicStreamOffset offset, 25858537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) bool fin, 25958537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) QuicAckNotifier* notifier, 26058537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) QuicFrame* frame) { 26158537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) size_t bytes_consumed = CreateStreamFrame(id, data, offset, fin, frame); 26258537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) 26358537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) // The frame keeps track of the QuicAckNotifier until it is serialized into 26458537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) // a packet. At that point the notifier is informed of the sequence number 26558537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) // of the packet that this frame was eventually sent in. 26658537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) frame->stream_frame->notifier = notifier; 26758537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) 26858537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) return bytes_consumed; 26958537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles)} 27058537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) 2713551c9c881056c480085172ff9840cab31610854Torne (Richard Coles)SerializedPacket QuicPacketCreator::ReserializeAllFrames( 2723551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) const QuicFrames& frames, 2733551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) QuicSequenceNumberLength original_length) { 274cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) DCHECK(fec_group_.get() == NULL); 275cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) const QuicSequenceNumberLength saved_length = sequence_number_length_; 27646d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) const QuicSequenceNumberLength saved_next_length = 27746d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) next_sequence_number_length_; 278cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) const bool saved_should_fec_protect = should_fec_protect_; 27968043e1e95eeb07d5cae7aca370b26518b0867d6Torne (Richard Coles) 280cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) // Temporarily set the sequence number length and stop FEC protection. 2813551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) sequence_number_length_ = original_length; 28246d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) next_sequence_number_length_ = original_length; 283cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) should_fec_protect_ = false; 28468043e1e95eeb07d5cae7aca370b26518b0867d6Torne (Richard Coles) 285cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) // Serialize the packet and restore the FEC and sequence number length state. 2863551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) SerializedPacket serialized_packet = SerializeAllFrames(frames); 287cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) sequence_number_length_ = saved_length; 28846d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) next_sequence_number_length_ = saved_next_length; 289cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) should_fec_protect_ = saved_should_fec_protect; 29068043e1e95eeb07d5cae7aca370b26518b0867d6Torne (Richard Coles) 2913551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) return serialized_packet; 2923551c9c881056c480085172ff9840cab31610854Torne (Richard Coles)} 2933551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) 2942a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)SerializedPacket QuicPacketCreator::SerializeAllFrames( 2952a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const QuicFrames& frames) { 2962a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // TODO(satyamshekhar): Verify that this DCHECK won't fail. What about queued 2972a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // frames from SendStreamData()[send_stream_should_flush_ == false && 2982a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // data.empty() == true] and retransmit due to RTO. 2992a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK_EQ(0u, queued_frames_.size()); 300a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) LOG_IF(DFATAL, frames.empty()) 301a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) << "Attempt to serialize empty packet"; 3022a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) for (size_t i = 0; i < frames.size(); ++i) { 3032a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) bool success = AddFrame(frames[i], false); 3042a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK(success); 3055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3062a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) SerializedPacket packet = SerializePacket(); 3072a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK(packet.retransmittable_frames == NULL); 3082a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return packet; 3095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 311f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)bool QuicPacketCreator::HasPendingFrames() const { 3122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return !queued_frames_.empty(); 3132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 3145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 315c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdochsize_t QuicPacketCreator::ExpansionOnNewFrame() const { 316c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch // If packet is FEC protected, there's no expansion. 317cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) if (should_fec_protect_) { 318c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch return 0; 319c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch } 320c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch // If the last frame in the packet is a stream frame, then it will expand to 321c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch // include the stream_length field when a new frame is added. 322c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch bool has_trailing_stream_frame = 323c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch !queued_frames_.empty() && queued_frames_.back().type == STREAM_FRAME; 324c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch return has_trailing_stream_frame ? kQuicStreamPayloadLengthSize : 0; 325c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch} 326c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 3272a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)size_t QuicPacketCreator::BytesFree() const { 3282a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const size_t max_plaintext_size = 32946d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) framer_->GetMaxPlaintextSize(max_packet_length_); 3301e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) DCHECK_GE(max_plaintext_size, PacketSize()); 331c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch return max_plaintext_size - min(max_plaintext_size, PacketSize() 332c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch + ExpansionOnNewFrame()); 3333551c9c881056c480085172ff9840cab31610854Torne (Richard Coles)} 3343551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) 3353551c9c881056c480085172ff9840cab31610854Torne (Richard Coles)size_t QuicPacketCreator::PacketSize() const { 33646d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) if (!queued_frames_.empty()) { 33746d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) return packet_size_; 3383551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) } 33946d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) if (fec_group_.get() == NULL) { 34046d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) // Update sequence number length on packet and FEC boundary. 34146d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) sequence_number_length_ = next_sequence_number_length_; 34246d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) } 34346d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) packet_size_ = GetPacketHeaderSize( 34446d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) connection_id_length_, send_version_in_packet_, sequence_number_length_, 34546d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) should_fec_protect_ ? IN_FEC_GROUP : NOT_IN_FEC_GROUP); 3463551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) return packet_size_; 3475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)bool QuicPacketCreator::AddSavedFrame(const QuicFrame& frame) { 3502a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return AddFrame(frame, true); 3512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 3525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3532a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)SerializedPacket QuicPacketCreator::SerializePacket() { 354a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) LOG_IF(DFATAL, queued_frames_.empty()) 355a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) << "Attempt to serialize empty packet"; 356cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) DCHECK_GE(sequence_number_ + 1, fec_group_number_); 3575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) QuicPacketHeader header; 358cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) FillPacketHeader(should_fec_protect_ ? fec_group_number_ : 0, false, &header); 3592a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3605d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) MaybeAddPadding(); 3611e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) 3621e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) size_t max_plaintext_size = 36346d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) framer_->GetMaxPlaintextSize(max_packet_length_); 3641e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) DCHECK_GE(max_plaintext_size, packet_size_); 365cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) // ACK Frames will be truncated only if they're the only frame in the packet, 366cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) // and if packet_size_ was set to max_plaintext_size. If truncation occurred, 367cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) // then GetSerializedFrameLength will have returned all bytes free. 368cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) bool possibly_truncated = packet_size_ == max_plaintext_size && 369cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) queued_frames_.size() == 1 && 370cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) queued_frames_.back().type == ACK_FRAME; 37158537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) SerializedPacket serialized = 37258537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) framer_->BuildDataPacket(header, queued_frames_, packet_size_); 373a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) LOG_IF(DFATAL, !serialized.packet) 374a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) << "Failed to serialize " << queued_frames_.size() << " frames."; 3751e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) // Because of possible truncation, we can't be confident that our 3761e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) // packet size calculation worked correctly. 377cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) if (!possibly_truncated) { 3781e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) DCHECK_EQ(packet_size_, serialized.packet->length()); 379cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) } 3803551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) packet_size_ = 0; 3812a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) queued_frames_.clear(); 3822a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) serialized.retransmittable_frames = queued_retransmittable_frames_.release(); 3832a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return serialized; 3845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)SerializedPacket QuicPacketCreator::SerializeFec() { 3870529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch if (fec_group_.get() == NULL || fec_group_->NumReceivedPackets() <= 0) { 3880529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch LOG(DFATAL) << "SerializeFEC called but no group or zero packets in group."; 3890529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch // TODO(jri): Make this a public method of framer? 3900529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch SerializedPacket kNoPacket(0, PACKET_1BYTE_SEQUENCE_NUMBER, NULL, 0, NULL); 3910529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch return kNoPacket; 3920529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch } 3932a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK_EQ(0u, queued_frames_.size()); 3945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) QuicPacketHeader header; 3955d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) FillPacketHeader(fec_group_number_, true, &header); 3962a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) QuicFecData fec_data; 3972a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) fec_data.fec_group = fec_group_->min_protected_packet(); 3982a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) fec_data.redundancy = fec_group_->payload_parity(); 3992385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch SerializedPacket serialized = framer_->BuildFecPacket(header, fec_data); 4002a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) fec_group_.reset(NULL); 4013551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) packet_size_ = 0; 402a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) LOG_IF(DFATAL, !serialized.packet) 403a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) << "Failed to serialize fec packet for group:" << fec_data.fec_group; 40446d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) DCHECK_GE(max_packet_length_, serialized.packet->length()); 4052a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return serialized; 4062a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 4075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4082a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)SerializedPacket QuicPacketCreator::SerializeConnectionClose( 4092a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) QuicConnectionCloseFrame* close_frame) { 4105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) QuicFrames frames; 4112a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) frames.push_back(QuicFrame(close_frame)); 4122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return SerializeAllFrames(frames); 4135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 4145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4152a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)QuicEncryptedPacket* QuicPacketCreator::SerializeVersionNegotiationPacket( 416558790d6acca3451cf3a6b497803a5f07d0bec58Ben Murdoch const QuicVersionVector& supported_versions) { 417f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) DCHECK(framer_->is_server()); 4182a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) QuicPacketPublicHeader header; 419a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) header.connection_id = connection_id_; 4202a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) header.reset_flag = false; 4212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) header.version_flag = true; 4222a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) header.versions = supported_versions; 4232a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) QuicEncryptedPacket* encrypted = 4242385ea399aae016c0806a4f9ef3c9cfe3d2a39dfBen Murdoch framer_->BuildVersionNegotiationPacket(header, supported_versions); 4252a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK(encrypted); 42646d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) DCHECK_GE(max_packet_length_, encrypted->length()); 4272a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return encrypted; 4285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 4295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void QuicPacketCreator::FillPacketHeader(QuicFecGroupNumber fec_group, 4312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) bool fec_flag, 4325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) QuicPacketHeader* header) { 433a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) header->public_header.connection_id = connection_id_; 4342a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) header->public_header.reset_flag = false; 4352a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) header->public_header.version_flag = send_version_in_packet_; 4362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) header->fec_flag = fec_flag; 4375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) header->packet_sequence_number = ++sequence_number_; 4383551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) header->public_header.sequence_number_length = sequence_number_length_; 4395d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) header->entropy_flag = random_bool_source_->RandBool(); 440868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) header->is_in_fec_group = fec_group == 0 ? NOT_IN_FEC_GROUP : IN_FEC_GROUP; 4415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) header->fec_group = fec_group; 4425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 4435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4442a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)bool QuicPacketCreator::ShouldRetransmit(const QuicFrame& frame) { 445a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) switch (frame.type) { 446a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) case ACK_FRAME: 447a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) case CONGESTION_FEEDBACK_FRAME: 448a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) case PADDING_FRAME: 449a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) case STOP_WAITING_FRAME: 450a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) return false; 451a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) default: 452a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) return true; 453a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) } 4542a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 4552a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 4562a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)bool QuicPacketCreator::AddFrame(const QuicFrame& frame, 4572a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) bool save_retransmittable_frames) { 4585d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) DVLOG(1) << "Adding frame: " << frame; 459cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) InFecGroup is_in_fec_group = MaybeUpdateLengthsAndStartFec(); 460cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) 4612a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) size_t frame_len = framer_->GetSerializedFrameLength( 462c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch frame, BytesFree(), queued_frames_.empty(), true, is_in_fec_group, 463cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) sequence_number_length_); 4642a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (frame_len == 0) { 4652a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return false; 4662a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 4673551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) DCHECK_LT(0u, packet_size_); 468c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch packet_size_ += ExpansionOnNewFrame() + frame_len; 469c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 4702a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (save_retransmittable_frames && ShouldRetransmit(frame)) { 4712a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (queued_retransmittable_frames_.get() == NULL) { 4722a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) queued_retransmittable_frames_.reset(new RetransmittableFrames()); 4732a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 4742a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (frame.type == STREAM_FRAME) { 4752a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) queued_frames_.push_back( 4762a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) queued_retransmittable_frames_->AddStreamFrame(frame.stream_frame)); 4772a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } else { 4782a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) queued_frames_.push_back( 4792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) queued_retransmittable_frames_->AddNonStreamFrame(frame)); 4802a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 4812a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } else { 4822a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) queued_frames_.push_back(frame); 4832a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 4842a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return true; 4852a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 4862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 4871e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles)void QuicPacketCreator::MaybeAddPadding() { 4881e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) if (BytesFree() == 0) { 4891e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) // Don't pad full packets. 4901e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) return; 4911e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) } 4921e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) 4931e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) // If any of the frames in the current packet are on the crypto stream 4941e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) // then they contain handshake messagses, and we should pad them. 4951e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) bool is_handshake = false; 4961e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) for (size_t i = 0; i < queued_frames_.size(); ++i) { 4971e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) if (queued_frames_[i].type == STREAM_FRAME && 4981e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) queued_frames_[i].stream_frame->stream_id == kCryptoStreamId) { 4991e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) is_handshake = true; 5001e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) break; 5011e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) } 5021e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) } 5031e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) if (!is_handshake) { 5041e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) return; 5051e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) } 5061e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) 5071e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) QuicPaddingFrame padding; 5081e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) bool success = AddFrame(QuicFrame(&padding), false); 5091e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) DCHECK(success); 5101e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles)} 5111e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) 5125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace net 513