Land Recent QUIC Changes.
[chromium-blink-merge.git] / net / quic / quic_framer.cc
blobb0b3afcb3a05fa52697e3bad1e2277d29f83226c
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.
5 #include "net/quic/quic_framer.h"
7 #include "base/containers/hash_tables.h"
8 #include "base/stl_util.h"
9 #include "net/quic/crypto/crypto_framer.h"
10 #include "net/quic/crypto/crypto_handshake_message.h"
11 #include "net/quic/crypto/crypto_protocol.h"
12 #include "net/quic/crypto/quic_decrypter.h"
13 #include "net/quic/crypto/quic_encrypter.h"
14 #include "net/quic/quic_data_reader.h"
15 #include "net/quic/quic_data_writer.h"
16 #include "net/quic/quic_flags.h"
17 #include "net/quic/quic_socket_address_coder.h"
19 using base::StringPiece;
20 using std::make_pair;
21 using std::map;
22 using std::max;
23 using std::min;
24 using std::numeric_limits;
25 using std::string;
27 namespace net {
29 namespace {
31 // Mask to select the lowest 48 bits of a sequence number.
32 const QuicPacketSequenceNumber k6ByteSequenceNumberMask =
33 GG_UINT64_C(0x0000FFFFFFFFFFFF);
34 const QuicPacketSequenceNumber k4ByteSequenceNumberMask =
35 GG_UINT64_C(0x00000000FFFFFFFF);
36 const QuicPacketSequenceNumber k2ByteSequenceNumberMask =
37 GG_UINT64_C(0x000000000000FFFF);
38 const QuicPacketSequenceNumber k1ByteSequenceNumberMask =
39 GG_UINT64_C(0x00000000000000FF);
41 const QuicConnectionId k1ByteConnectionIdMask = GG_UINT64_C(0x00000000000000FF);
42 const QuicConnectionId k4ByteConnectionIdMask = GG_UINT64_C(0x00000000FFFFFFFF);
44 // Number of bits the sequence number length bits are shifted from the right
45 // edge of the public header.
46 const uint8 kPublicHeaderSequenceNumberShift = 4;
48 // New Frame Types, QUIC v. >= 10:
49 // There are two interpretations for the Frame Type byte in the QUIC protocol,
50 // resulting in two Frame Types: Special Frame Types and Regular Frame Types.
52 // Regular Frame Types use the Frame Type byte simply. Currently defined
53 // Regular Frame Types are:
54 // Padding : 0b 00000000 (0x00)
55 // ResetStream : 0b 00000001 (0x01)
56 // ConnectionClose : 0b 00000010 (0x02)
57 // GoAway : 0b 00000011 (0x03)
58 // WindowUpdate : 0b 00000100 (0x04)
59 // Blocked : 0b 00000101 (0x05)
61 // Special Frame Types encode both a Frame Type and corresponding flags
62 // all in the Frame Type byte. Currently defined Special Frame Types are:
63 // Stream : 0b 1xxxxxxx
64 // Ack : 0b 01xxxxxx
65 // CongestionFeedback : 0b 001xxxxx
67 // Semantics of the flag bits above (the x bits) depends on the frame type.
69 // Masks to determine if the frame type is a special use
70 // and for specific special frame types.
71 const uint8 kQuicFrameTypeSpecialMask = 0xE0; // 0b 11100000
72 const uint8 kQuicFrameTypeStreamMask = 0x80;
73 const uint8 kQuicFrameTypeAckMask = 0x40;
74 const uint8 kQuicFrameTypeCongestionFeedbackMask = 0x20;
76 // Stream frame relative shifts and masks for interpreting the stream flags.
77 // StreamID may be 1, 2, 3, or 4 bytes.
78 const uint8 kQuicStreamIdShift = 2;
79 const uint8 kQuicStreamIDLengthMask = 0x03;
81 // Offset may be 0, 2, 3, 4, 5, 6, 7, 8 bytes.
82 const uint8 kQuicStreamOffsetShift = 3;
83 const uint8 kQuicStreamOffsetMask = 0x07;
85 // Data length may be 0 or 2 bytes.
86 const uint8 kQuicStreamDataLengthShift = 1;
87 const uint8 kQuicStreamDataLengthMask = 0x01;
89 // Fin bit may be set or not.
90 const uint8 kQuicStreamFinShift = 1;
91 const uint8 kQuicStreamFinMask = 0x01;
93 // Sequence number size shift used in AckFrames.
94 const uint8 kQuicSequenceNumberLengthShift = 2;
96 // Acks may be truncated.
97 const uint8 kQuicAckTruncatedShift = 1;
98 const uint8 kQuicAckTruncatedMask = 0x01;
100 // Acks may not have any nacks.
101 const uint8 kQuicHasNacksMask = 0x01;
103 // Returns the absolute value of the difference between |a| and |b|.
104 QuicPacketSequenceNumber Delta(QuicPacketSequenceNumber a,
105 QuicPacketSequenceNumber b) {
106 // Since these are unsigned numbers, we can't just return abs(a - b)
107 if (a < b) {
108 return b - a;
110 return a - b;
113 QuicPacketSequenceNumber ClosestTo(QuicPacketSequenceNumber target,
114 QuicPacketSequenceNumber a,
115 QuicPacketSequenceNumber b) {
116 return (Delta(target, a) < Delta(target, b)) ? a : b;
119 QuicSequenceNumberLength ReadSequenceNumberLength(uint8 flags) {
120 switch (flags & PACKET_FLAGS_6BYTE_SEQUENCE) {
121 case PACKET_FLAGS_6BYTE_SEQUENCE:
122 return PACKET_6BYTE_SEQUENCE_NUMBER;
123 case PACKET_FLAGS_4BYTE_SEQUENCE:
124 return PACKET_4BYTE_SEQUENCE_NUMBER;
125 case PACKET_FLAGS_2BYTE_SEQUENCE:
126 return PACKET_2BYTE_SEQUENCE_NUMBER;
127 case PACKET_FLAGS_1BYTE_SEQUENCE:
128 return PACKET_1BYTE_SEQUENCE_NUMBER;
129 default:
130 LOG(DFATAL) << "Unreachable case statement.";
131 return PACKET_6BYTE_SEQUENCE_NUMBER;
135 } // namespace
137 bool QuicFramerVisitorInterface::OnWindowUpdateFrame(
138 const QuicWindowUpdateFrame& frame) {
139 return true;
142 bool QuicFramerVisitorInterface::OnBlockedFrame(const QuicBlockedFrame& frame) {
143 return true;
146 QuicFramer::QuicFramer(const QuicVersionVector& supported_versions,
147 QuicTime creation_time,
148 bool is_server)
149 : visitor_(nullptr),
150 fec_builder_(nullptr),
151 entropy_calculator_(nullptr),
152 error_(QUIC_NO_ERROR),
153 last_sequence_number_(0),
154 last_serialized_connection_id_(0),
155 supported_versions_(supported_versions),
156 decrypter_level_(ENCRYPTION_NONE),
157 alternative_decrypter_level_(ENCRYPTION_NONE),
158 alternative_decrypter_latch_(false),
159 is_server_(is_server),
160 validate_flags_(true),
161 creation_time_(creation_time),
162 last_timestamp_(QuicTime::Delta::Zero()) {
163 DCHECK(!supported_versions.empty());
164 quic_version_ = supported_versions_[0];
165 decrypter_.reset(QuicDecrypter::Create(kNULL));
166 encrypter_[ENCRYPTION_NONE].reset(QuicEncrypter::Create(kNULL));
169 QuicFramer::~QuicFramer() {}
171 // static
172 size_t QuicFramer::GetMinStreamFrameSize(QuicStreamId stream_id,
173 QuicStreamOffset offset,
174 bool last_frame_in_packet,
175 InFecGroup is_in_fec_group) {
176 bool no_stream_frame_length = last_frame_in_packet &&
177 is_in_fec_group == NOT_IN_FEC_GROUP;
178 return kQuicFrameTypeSize + GetStreamIdSize(stream_id) +
179 GetStreamOffsetSize(offset) +
180 (no_stream_frame_length ? 0 : kQuicStreamPayloadLengthSize);
183 // static
184 size_t QuicFramer::GetMinAckFrameSize(
185 QuicSequenceNumberLength sequence_number_length,
186 QuicSequenceNumberLength largest_observed_length) {
187 return kQuicFrameTypeSize + kQuicEntropyHashSize +
188 largest_observed_length + kQuicDeltaTimeLargestObservedSize;
191 // static
192 size_t QuicFramer::GetStopWaitingFrameSize(
193 QuicSequenceNumberLength sequence_number_length) {
194 return kQuicFrameTypeSize + kQuicEntropyHashSize +
195 sequence_number_length;
198 // static
199 size_t QuicFramer::GetMinRstStreamFrameSize() {
200 return kQuicFrameTypeSize + kQuicMaxStreamIdSize +
201 kQuicMaxStreamOffsetSize + kQuicErrorCodeSize +
202 kQuicErrorDetailsLengthSize;
205 // static
206 size_t QuicFramer::GetMinConnectionCloseFrameSize() {
207 return kQuicFrameTypeSize + kQuicErrorCodeSize + kQuicErrorDetailsLengthSize;
210 // static
211 size_t QuicFramer::GetMinGoAwayFrameSize() {
212 return kQuicFrameTypeSize + kQuicErrorCodeSize + kQuicErrorDetailsLengthSize +
213 kQuicMaxStreamIdSize;
216 // static
217 size_t QuicFramer::GetWindowUpdateFrameSize() {
218 return kQuicFrameTypeSize + kQuicMaxStreamIdSize + kQuicMaxStreamOffsetSize;
221 // static
222 size_t QuicFramer::GetBlockedFrameSize() {
223 return kQuicFrameTypeSize + kQuicMaxStreamIdSize;
226 // static
227 size_t QuicFramer::GetStreamIdSize(QuicStreamId stream_id) {
228 // Sizes are 1 through 4 bytes.
229 for (int i = 1; i <= 4; ++i) {
230 stream_id >>= 8;
231 if (stream_id == 0) {
232 return i;
235 LOG(DFATAL) << "Failed to determine StreamIDSize.";
236 return 4;
239 // static
240 size_t QuicFramer::GetStreamOffsetSize(QuicStreamOffset offset) {
241 // 0 is a special case.
242 if (offset == 0) {
243 return 0;
245 // 2 through 8 are the remaining sizes.
246 offset >>= 8;
247 for (int i = 2; i <= 8; ++i) {
248 offset >>= 8;
249 if (offset == 0) {
250 return i;
253 LOG(DFATAL) << "Failed to determine StreamOffsetSize.";
254 return 8;
257 // static
258 size_t QuicFramer::GetVersionNegotiationPacketSize(size_t number_versions) {
259 return kPublicFlagsSize + PACKET_8BYTE_CONNECTION_ID +
260 number_versions * kQuicVersionSize;
263 bool QuicFramer::IsSupportedVersion(const QuicVersion version) const {
264 for (size_t i = 0; i < supported_versions_.size(); ++i) {
265 if (version == supported_versions_[i]) {
266 return true;
269 return false;
272 size_t QuicFramer::GetSerializedFrameLength(
273 const QuicFrame& frame,
274 size_t free_bytes,
275 bool first_frame,
276 bool last_frame,
277 InFecGroup is_in_fec_group,
278 QuicSequenceNumberLength sequence_number_length) {
279 if (frame.type == PADDING_FRAME) {
280 // PADDING implies end of packet.
281 return free_bytes;
283 size_t frame_len =
284 ComputeFrameLength(frame, last_frame, is_in_fec_group,
285 sequence_number_length);
286 if (frame_len <= free_bytes) {
287 // Frame fits within packet. Note that acks may be truncated.
288 return frame_len;
290 // Only truncate the first frame in a packet, so if subsequent ones go
291 // over, stop including more frames.
292 if (!first_frame) {
293 return 0;
295 bool can_truncate = frame.type == ACK_FRAME &&
296 free_bytes >= GetMinAckFrameSize(PACKET_6BYTE_SEQUENCE_NUMBER,
297 PACKET_6BYTE_SEQUENCE_NUMBER);
298 if (can_truncate) {
299 // Truncate the frame so the packet will not exceed kMaxPacketSize.
300 // Note that we may not use every byte of the writer in this case.
301 DVLOG(1) << "Truncating large frame, free bytes: " << free_bytes;
302 return free_bytes;
304 if (!FLAGS_quic_allow_oversized_packets_for_test) {
305 return 0;
307 LOG(DFATAL) << "Packet size too small to fit frame.";
308 return frame_len;
311 QuicFramer::AckFrameInfo::AckFrameInfo() : max_delta(0) {}
313 QuicFramer::AckFrameInfo::~AckFrameInfo() {}
315 QuicPacketEntropyHash QuicFramer::GetPacketEntropyHash(
316 const QuicPacketHeader& header) const {
317 return header.entropy_flag << (header.packet_sequence_number % 8);
320 SerializedPacket QuicFramer::BuildDataPacket(
321 const QuicPacketHeader& header,
322 const QuicFrames& frames,
323 size_t packet_size) {
324 QuicDataWriter writer(packet_size);
325 const SerializedPacket kNoPacket(0, PACKET_1BYTE_SEQUENCE_NUMBER, nullptr, 0,
326 nullptr);
327 if (!AppendPacketHeader(header, &writer)) {
328 LOG(DFATAL) << "AppendPacketHeader failed";
329 return kNoPacket;
332 for (size_t i = 0; i < frames.size(); ++i) {
333 const QuicFrame& frame = frames[i];
335 // Determine if we should write stream frame length in header.
336 const bool no_stream_frame_length =
337 (header.is_in_fec_group == NOT_IN_FEC_GROUP) &&
338 (i == frames.size() - 1);
339 if (!AppendTypeByte(frame, no_stream_frame_length, &writer)) {
340 LOG(DFATAL) << "AppendTypeByte failed";
341 return kNoPacket;
344 switch (frame.type) {
345 case PADDING_FRAME:
346 writer.WritePadding();
347 break;
348 case STREAM_FRAME:
349 if (!AppendStreamFrame(
350 *frame.stream_frame, no_stream_frame_length, &writer)) {
351 LOG(DFATAL) << "AppendStreamFrame failed";
352 return kNoPacket;
354 break;
355 case ACK_FRAME:
356 if (!AppendAckFrameAndTypeByte(
357 header, *frame.ack_frame, &writer)) {
358 LOG(DFATAL) << "AppendAckFrameAndTypeByte failed";
359 return kNoPacket;
361 break;
362 case CONGESTION_FEEDBACK_FRAME:
363 if (!AppendCongestionFeedbackFrame(
364 *frame.congestion_feedback_frame, &writer)) {
365 LOG(DFATAL) << "AppendCongestionFeedbackFrame failed";
366 return kNoPacket;
368 break;
369 case STOP_WAITING_FRAME:
370 if (!AppendStopWaitingFrame(
371 header, *frame.stop_waiting_frame, &writer)) {
372 LOG(DFATAL) << "AppendStopWaitingFrame failed";
373 return kNoPacket;
375 break;
376 case PING_FRAME:
377 // Ping has no payload.
378 break;
379 case RST_STREAM_FRAME:
380 if (!AppendRstStreamFrame(*frame.rst_stream_frame, &writer)) {
381 LOG(DFATAL) << "AppendRstStreamFrame failed";
382 return kNoPacket;
384 break;
385 case CONNECTION_CLOSE_FRAME:
386 if (!AppendConnectionCloseFrame(
387 *frame.connection_close_frame, &writer)) {
388 LOG(DFATAL) << "AppendConnectionCloseFrame failed";
389 return kNoPacket;
391 break;
392 case GOAWAY_FRAME:
393 if (!AppendGoAwayFrame(*frame.goaway_frame, &writer)) {
394 LOG(DFATAL) << "AppendGoAwayFrame failed";
395 return kNoPacket;
397 break;
398 case WINDOW_UPDATE_FRAME:
399 if (!AppendWindowUpdateFrame(*frame.window_update_frame, &writer)) {
400 LOG(DFATAL) << "AppendWindowUpdateFrame failed";
401 return kNoPacket;
403 break;
404 case BLOCKED_FRAME:
405 if (!AppendBlockedFrame(*frame.blocked_frame, &writer)) {
406 LOG(DFATAL) << "AppendBlockedFrame failed";
407 return kNoPacket;
409 break;
410 default:
411 RaiseError(QUIC_INVALID_FRAME_DATA);
412 LOG(DFATAL) << "QUIC_INVALID_FRAME_DATA";
413 return kNoPacket;
417 // Save the length before writing, because take clears it.
418 const size_t len = writer.length();
419 // Less than or equal because truncated acks end up with max_plaintex_size
420 // length, even though they're typically slightly shorter.
421 DCHECK_LE(len, packet_size);
422 QuicPacket* packet = QuicPacket::NewDataPacket(
423 writer.take(), len, true, header.public_header.connection_id_length,
424 header.public_header.version_flag,
425 header.public_header.sequence_number_length);
427 if (fec_builder_) {
428 fec_builder_->OnBuiltFecProtectedPayload(header,
429 packet->FecProtectedData());
432 return SerializedPacket(header.packet_sequence_number,
433 header.public_header.sequence_number_length, packet,
434 GetPacketEntropyHash(header), nullptr);
437 SerializedPacket QuicFramer::BuildFecPacket(const QuicPacketHeader& header,
438 const QuicFecData& fec) {
439 DCHECK_EQ(IN_FEC_GROUP, header.is_in_fec_group);
440 DCHECK_NE(0u, header.fec_group);
441 size_t len = GetPacketHeaderSize(header);
442 len += fec.redundancy.length();
444 QuicDataWriter writer(len);
445 const SerializedPacket kNoPacket(0, PACKET_1BYTE_SEQUENCE_NUMBER, nullptr, 0,
446 nullptr);
447 if (!AppendPacketHeader(header, &writer)) {
448 LOG(DFATAL) << "AppendPacketHeader failed";
449 return kNoPacket;
452 if (!writer.WriteBytes(fec.redundancy.data(), fec.redundancy.length())) {
453 LOG(DFATAL) << "Failed to add FEC";
454 return kNoPacket;
457 return SerializedPacket(
458 header.packet_sequence_number,
459 header.public_header.sequence_number_length,
460 QuicPacket::NewFecPacket(writer.take(), len, true,
461 header.public_header.connection_id_length,
462 header.public_header.version_flag,
463 header.public_header.sequence_number_length),
464 GetPacketEntropyHash(header), nullptr);
467 // static
468 QuicEncryptedPacket* QuicFramer::BuildPublicResetPacket(
469 const QuicPublicResetPacket& packet) {
470 DCHECK(packet.public_header.reset_flag);
472 CryptoHandshakeMessage reset;
473 reset.set_tag(kPRST);
474 reset.SetValue(kRNON, packet.nonce_proof);
475 reset.SetValue(kRSEQ, packet.rejected_sequence_number);
476 if (!packet.client_address.address().empty()) {
477 // packet.client_address is non-empty.
478 QuicSocketAddressCoder address_coder(packet.client_address);
479 string serialized_address = address_coder.Encode();
480 if (serialized_address.empty()) {
481 return nullptr;
483 reset.SetStringPiece(kCADR, serialized_address);
485 const QuicData& reset_serialized = reset.GetSerialized();
487 size_t len =
488 kPublicFlagsSize + PACKET_8BYTE_CONNECTION_ID + reset_serialized.length();
489 QuicDataWriter writer(len);
491 uint8 flags = static_cast<uint8>(PACKET_PUBLIC_FLAGS_RST |
492 PACKET_PUBLIC_FLAGS_8BYTE_CONNECTION_ID);
493 if (!writer.WriteUInt8(flags)) {
494 return nullptr;
497 if (!writer.WriteUInt64(packet.public_header.connection_id)) {
498 return nullptr;
501 if (!writer.WriteBytes(reset_serialized.data(), reset_serialized.length())) {
502 return nullptr;
505 return new QuicEncryptedPacket(writer.take(), len, true);
508 QuicEncryptedPacket* QuicFramer::BuildVersionNegotiationPacket(
509 const QuicPacketPublicHeader& header,
510 const QuicVersionVector& supported_versions) {
511 DCHECK(header.version_flag);
512 size_t len = GetVersionNegotiationPacketSize(supported_versions.size());
513 QuicDataWriter writer(len);
515 uint8 flags = static_cast<uint8>(PACKET_PUBLIC_FLAGS_VERSION |
516 PACKET_PUBLIC_FLAGS_8BYTE_CONNECTION_ID);
517 if (!writer.WriteUInt8(flags)) {
518 return nullptr;
521 if (!writer.WriteUInt64(header.connection_id)) {
522 return nullptr;
525 for (size_t i = 0; i < supported_versions.size(); ++i) {
526 if (!writer.WriteUInt32(QuicVersionToQuicTag(supported_versions[i]))) {
527 return nullptr;
531 return new QuicEncryptedPacket(writer.take(), len, true);
534 bool QuicFramer::ProcessPacket(const QuicEncryptedPacket& packet) {
535 DCHECK(!reader_.get());
536 reader_.reset(new QuicDataReader(packet.data(), packet.length()));
538 visitor_->OnPacket();
540 // First parse the public header.
541 QuicPacketPublicHeader public_header;
542 if (!ProcessPublicHeader(&public_header)) {
543 DLOG(WARNING) << "Unable to process public header.";
544 DCHECK_NE("", detailed_error_);
545 return RaiseError(QUIC_INVALID_PACKET_HEADER);
548 if (!visitor_->OnUnauthenticatedPublicHeader(public_header)) {
549 // The visitor suppresses further processing of the packet.
550 reader_.reset(nullptr);
551 return true;
554 if (is_server_ && public_header.version_flag &&
555 public_header.versions[0] != quic_version_) {
556 if (!visitor_->OnProtocolVersionMismatch(public_header.versions[0])) {
557 reader_.reset(nullptr);
558 return true;
562 bool rv;
563 if (!is_server_ && public_header.version_flag) {
564 rv = ProcessVersionNegotiationPacket(&public_header);
565 } else if (public_header.reset_flag) {
566 rv = ProcessPublicResetPacket(public_header);
567 } else {
568 rv = ProcessDataPacket(public_header, packet);
571 reader_.reset(nullptr);
572 return rv;
575 bool QuicFramer::ProcessVersionNegotiationPacket(
576 QuicPacketPublicHeader* public_header) {
577 DCHECK(!is_server_);
578 // Try reading at least once to raise error if the packet is invalid.
579 do {
580 QuicTag version;
581 if (!reader_->ReadBytes(&version, kQuicVersionSize)) {
582 set_detailed_error("Unable to read supported version in negotiation.");
583 return RaiseError(QUIC_INVALID_VERSION_NEGOTIATION_PACKET);
585 public_header->versions.push_back(QuicTagToQuicVersion(version));
586 } while (!reader_->IsDoneReading());
588 visitor_->OnVersionNegotiationPacket(*public_header);
589 return true;
592 bool QuicFramer::ProcessDataPacket(
593 const QuicPacketPublicHeader& public_header,
594 const QuicEncryptedPacket& packet) {
595 QuicPacketHeader header(public_header);
596 if (!ProcessPacketHeader(&header, packet)) {
597 DLOG(WARNING) << "Unable to process data packet header.";
598 return false;
601 if (!visitor_->OnPacketHeader(header)) {
602 // The visitor suppresses further processing of the packet.
603 return true;
606 if (packet.length() > kMaxPacketSize) {
607 DLOG(WARNING) << "Packet too large: " << packet.length();
608 return RaiseError(QUIC_PACKET_TOO_LARGE);
611 // Handle the payload.
612 if (!header.fec_flag) {
613 if (header.is_in_fec_group == IN_FEC_GROUP) {
614 StringPiece payload = reader_->PeekRemainingPayload();
615 visitor_->OnFecProtectedPayload(payload);
617 if (!ProcessFrameData(header)) {
618 DCHECK_NE(QUIC_NO_ERROR, error_); // ProcessFrameData sets the error.
619 DLOG(WARNING) << "Unable to process frame data.";
620 return false;
622 } else {
623 QuicFecData fec_data;
624 fec_data.fec_group = header.fec_group;
625 fec_data.redundancy = reader_->ReadRemainingPayload();
626 visitor_->OnFecData(fec_data);
629 visitor_->OnPacketComplete();
630 return true;
633 bool QuicFramer::ProcessPublicResetPacket(
634 const QuicPacketPublicHeader& public_header) {
635 QuicPublicResetPacket packet(public_header);
637 scoped_ptr<CryptoHandshakeMessage> reset(
638 CryptoFramer::ParseMessage(reader_->ReadRemainingPayload()));
639 if (!reset.get()) {
640 set_detailed_error("Unable to read reset message.");
641 return RaiseError(QUIC_INVALID_PUBLIC_RST_PACKET);
643 if (reset->tag() != kPRST) {
644 set_detailed_error("Incorrect message tag.");
645 return RaiseError(QUIC_INVALID_PUBLIC_RST_PACKET);
648 if (reset->GetUint64(kRNON, &packet.nonce_proof) != QUIC_NO_ERROR) {
649 set_detailed_error("Unable to read nonce proof.");
650 return RaiseError(QUIC_INVALID_PUBLIC_RST_PACKET);
652 // TODO(satyamshekhar): validate nonce to protect against DoS.
654 if (reset->GetUint64(kRSEQ, &packet.rejected_sequence_number) !=
655 QUIC_NO_ERROR) {
656 set_detailed_error("Unable to read rejected sequence number.");
657 return RaiseError(QUIC_INVALID_PUBLIC_RST_PACKET);
660 StringPiece address;
661 if (reset->GetStringPiece(kCADR, &address)) {
662 QuicSocketAddressCoder address_coder;
663 if (address_coder.Decode(address.data(), address.length())) {
664 packet.client_address = IPEndPoint(address_coder.ip(),
665 address_coder.port());
669 visitor_->OnPublicResetPacket(packet);
670 return true;
673 bool QuicFramer::ProcessRevivedPacket(QuicPacketHeader* header,
674 StringPiece payload) {
675 DCHECK(!reader_.get());
677 visitor_->OnRevivedPacket();
679 header->entropy_hash = GetPacketEntropyHash(*header);
681 if (!visitor_->OnPacketHeader(*header)) {
682 return true;
685 if (payload.length() > kMaxPacketSize) {
686 set_detailed_error("Revived packet too large.");
687 return RaiseError(QUIC_PACKET_TOO_LARGE);
690 reader_.reset(new QuicDataReader(payload.data(), payload.length()));
691 if (!ProcessFrameData(*header)) {
692 DCHECK_NE(QUIC_NO_ERROR, error_); // ProcessFrameData sets the error.
693 DLOG(WARNING) << "Unable to process frame data.";
694 return false;
697 visitor_->OnPacketComplete();
698 reader_.reset(nullptr);
699 return true;
702 bool QuicFramer::AppendPacketHeader(const QuicPacketHeader& header,
703 QuicDataWriter* writer) {
704 DVLOG(1) << "Appending header: " << header;
705 DCHECK(header.fec_group > 0 || header.is_in_fec_group == NOT_IN_FEC_GROUP);
706 uint8 public_flags = 0;
707 if (header.public_header.reset_flag) {
708 public_flags |= PACKET_PUBLIC_FLAGS_RST;
710 if (header.public_header.version_flag) {
711 public_flags |= PACKET_PUBLIC_FLAGS_VERSION;
714 public_flags |=
715 GetSequenceNumberFlags(header.public_header.sequence_number_length)
716 << kPublicHeaderSequenceNumberShift;
718 switch (header.public_header.connection_id_length) {
719 case PACKET_0BYTE_CONNECTION_ID:
720 if (!writer->WriteUInt8(
721 public_flags | PACKET_PUBLIC_FLAGS_0BYTE_CONNECTION_ID)) {
722 return false;
724 break;
725 case PACKET_1BYTE_CONNECTION_ID:
726 if (!writer->WriteUInt8(
727 public_flags | PACKET_PUBLIC_FLAGS_1BYTE_CONNECTION_ID)) {
728 return false;
730 if (!writer->WriteUInt8(
731 header.public_header.connection_id & k1ByteConnectionIdMask)) {
732 return false;
734 break;
735 case PACKET_4BYTE_CONNECTION_ID:
736 if (!writer->WriteUInt8(
737 public_flags | PACKET_PUBLIC_FLAGS_4BYTE_CONNECTION_ID)) {
738 return false;
740 if (!writer->WriteUInt32(
741 header.public_header.connection_id & k4ByteConnectionIdMask)) {
742 return false;
744 break;
745 case PACKET_8BYTE_CONNECTION_ID:
746 if (!writer->WriteUInt8(
747 public_flags | PACKET_PUBLIC_FLAGS_8BYTE_CONNECTION_ID)) {
748 return false;
750 if (!writer->WriteUInt64(header.public_header.connection_id)) {
751 return false;
753 break;
755 last_serialized_connection_id_ = header.public_header.connection_id;
757 if (header.public_header.version_flag) {
758 DCHECK(!is_server_);
759 writer->WriteUInt32(QuicVersionToQuicTag(quic_version_));
762 if (!AppendPacketSequenceNumber(header.public_header.sequence_number_length,
763 header.packet_sequence_number, writer)) {
764 return false;
767 uint8 private_flags = 0;
768 if (header.entropy_flag) {
769 private_flags |= PACKET_PRIVATE_FLAGS_ENTROPY;
771 if (header.is_in_fec_group == IN_FEC_GROUP) {
772 private_flags |= PACKET_PRIVATE_FLAGS_FEC_GROUP;
774 if (header.fec_flag) {
775 private_flags |= PACKET_PRIVATE_FLAGS_FEC;
777 if (!writer->WriteUInt8(private_flags)) {
778 return false;
781 // The FEC group number is the sequence number of the first fec
782 // protected packet, or 0 if this packet is not protected.
783 if (header.is_in_fec_group == IN_FEC_GROUP) {
784 DCHECK_LE(header.fec_group, header.packet_sequence_number);
785 DCHECK_LT(header.packet_sequence_number - header.fec_group, 255u);
786 // Offset from the current packet sequence number to the first fec
787 // protected packet.
788 uint8 first_fec_protected_packet_offset =
789 static_cast<uint8>(header.packet_sequence_number - header.fec_group);
790 if (!writer->WriteBytes(&first_fec_protected_packet_offset, 1)) {
791 return false;
795 return true;
798 const QuicTime::Delta QuicFramer::CalculateTimestampFromWire(
799 uint32 time_delta_us) {
800 // The new time_delta might have wrapped to the next epoch, or it
801 // might have reverse wrapped to the previous epoch, or it might
802 // remain in the same epoch. Select the time closest to the previous
803 // time.
805 // epoch_delta is the delta between epochs. A delta is 4 bytes of
806 // microseconds.
807 const uint64 epoch_delta = GG_UINT64_C(1) << 32;
808 uint64 epoch = last_timestamp_.ToMicroseconds() & ~(epoch_delta - 1);
809 // Wrapping is safe here because a wrapped value will not be ClosestTo below.
810 uint64 prev_epoch = epoch - epoch_delta;
811 uint64 next_epoch = epoch + epoch_delta;
813 uint64 time = ClosestTo(last_timestamp_.ToMicroseconds(),
814 epoch + time_delta_us,
815 ClosestTo(last_timestamp_.ToMicroseconds(),
816 prev_epoch + time_delta_us,
817 next_epoch + time_delta_us));
819 return QuicTime::Delta::FromMicroseconds(time);
822 QuicPacketSequenceNumber QuicFramer::CalculatePacketSequenceNumberFromWire(
823 QuicSequenceNumberLength sequence_number_length,
824 QuicPacketSequenceNumber packet_sequence_number) const {
825 // The new sequence number might have wrapped to the next epoch, or
826 // it might have reverse wrapped to the previous epoch, or it might
827 // remain in the same epoch. Select the sequence number closest to the
828 // next expected sequence number, the previous sequence number plus 1.
830 // epoch_delta is the delta between epochs the sequence number was serialized
831 // with, so the correct value is likely the same epoch as the last sequence
832 // number or an adjacent epoch.
833 const QuicPacketSequenceNumber epoch_delta =
834 GG_UINT64_C(1) << (8 * sequence_number_length);
835 QuicPacketSequenceNumber next_sequence_number = last_sequence_number_ + 1;
836 QuicPacketSequenceNumber epoch = last_sequence_number_ & ~(epoch_delta - 1);
837 QuicPacketSequenceNumber prev_epoch = epoch - epoch_delta;
838 QuicPacketSequenceNumber next_epoch = epoch + epoch_delta;
840 return ClosestTo(next_sequence_number,
841 epoch + packet_sequence_number,
842 ClosestTo(next_sequence_number,
843 prev_epoch + packet_sequence_number,
844 next_epoch + packet_sequence_number));
847 bool QuicFramer::ProcessPublicHeader(
848 QuicPacketPublicHeader* public_header) {
849 uint8 public_flags;
850 if (!reader_->ReadBytes(&public_flags, 1)) {
851 set_detailed_error("Unable to read public flags.");
852 return false;
855 public_header->reset_flag = (public_flags & PACKET_PUBLIC_FLAGS_RST) != 0;
856 public_header->version_flag =
857 (public_flags & PACKET_PUBLIC_FLAGS_VERSION) != 0;
859 if (validate_flags_ &&
860 !public_header->version_flag && public_flags > PACKET_PUBLIC_FLAGS_MAX) {
861 set_detailed_error("Illegal public flags value.");
862 return false;
865 if (public_header->reset_flag && public_header->version_flag) {
866 set_detailed_error("Got version flag in reset packet");
867 return false;
870 switch (public_flags & PACKET_PUBLIC_FLAGS_8BYTE_CONNECTION_ID) {
871 case PACKET_PUBLIC_FLAGS_8BYTE_CONNECTION_ID:
872 if (!reader_->ReadUInt64(&public_header->connection_id)) {
873 set_detailed_error("Unable to read ConnectionId.");
874 return false;
876 public_header->connection_id_length = PACKET_8BYTE_CONNECTION_ID;
877 break;
878 case PACKET_PUBLIC_FLAGS_4BYTE_CONNECTION_ID:
879 // If the connection_id is truncated, expect to read the last serialized
880 // connection_id.
881 if (!reader_->ReadBytes(&public_header->connection_id,
882 PACKET_4BYTE_CONNECTION_ID)) {
883 set_detailed_error("Unable to read ConnectionId.");
884 return false;
886 if (last_serialized_connection_id_ &&
887 (public_header->connection_id & k4ByteConnectionIdMask) !=
888 (last_serialized_connection_id_ & k4ByteConnectionIdMask)) {
889 set_detailed_error("Truncated 4 byte ConnectionId does not match "
890 "previous connection_id.");
891 return false;
893 public_header->connection_id_length = PACKET_4BYTE_CONNECTION_ID;
894 public_header->connection_id = last_serialized_connection_id_;
895 break;
896 case PACKET_PUBLIC_FLAGS_1BYTE_CONNECTION_ID:
897 if (!reader_->ReadBytes(&public_header->connection_id,
898 PACKET_1BYTE_CONNECTION_ID)) {
899 set_detailed_error("Unable to read ConnectionId.");
900 return false;
902 if (last_serialized_connection_id_ &&
903 (public_header->connection_id & k1ByteConnectionIdMask) !=
904 (last_serialized_connection_id_ & k1ByteConnectionIdMask)) {
905 set_detailed_error("Truncated 1 byte ConnectionId does not match "
906 "previous connection_id.");
907 return false;
909 public_header->connection_id_length = PACKET_1BYTE_CONNECTION_ID;
910 public_header->connection_id = last_serialized_connection_id_;
911 break;
912 case PACKET_PUBLIC_FLAGS_0BYTE_CONNECTION_ID:
913 public_header->connection_id_length = PACKET_0BYTE_CONNECTION_ID;
914 public_header->connection_id = last_serialized_connection_id_;
915 break;
918 public_header->sequence_number_length =
919 ReadSequenceNumberLength(
920 public_flags >> kPublicHeaderSequenceNumberShift);
922 // Read the version only if the packet is from the client.
923 // version flag from the server means version negotiation packet.
924 if (public_header->version_flag && is_server_) {
925 QuicTag version_tag;
926 if (!reader_->ReadUInt32(&version_tag)) {
927 set_detailed_error("Unable to read protocol version.");
928 return false;
931 // If the version from the new packet is the same as the version of this
932 // framer, then the public flags should be set to something we understand.
933 // If not, this raises an error.
934 QuicVersion version = QuicTagToQuicVersion(version_tag);
935 if (version == quic_version_ && public_flags > PACKET_PUBLIC_FLAGS_MAX) {
936 set_detailed_error("Illegal public flags value.");
937 return false;
939 public_header->versions.push_back(version);
941 return true;
944 // static
945 QuicSequenceNumberLength QuicFramer::GetMinSequenceNumberLength(
946 QuicPacketSequenceNumber sequence_number) {
947 if (sequence_number < 1 << (PACKET_1BYTE_SEQUENCE_NUMBER * 8)) {
948 return PACKET_1BYTE_SEQUENCE_NUMBER;
949 } else if (sequence_number < 1 << (PACKET_2BYTE_SEQUENCE_NUMBER * 8)) {
950 return PACKET_2BYTE_SEQUENCE_NUMBER;
951 } else if (sequence_number <
952 GG_UINT64_C(1) << (PACKET_4BYTE_SEQUENCE_NUMBER * 8)) {
953 return PACKET_4BYTE_SEQUENCE_NUMBER;
954 } else {
955 return PACKET_6BYTE_SEQUENCE_NUMBER;
959 // static
960 uint8 QuicFramer::GetSequenceNumberFlags(
961 QuicSequenceNumberLength sequence_number_length) {
962 switch (sequence_number_length) {
963 case PACKET_1BYTE_SEQUENCE_NUMBER:
964 return PACKET_FLAGS_1BYTE_SEQUENCE;
965 case PACKET_2BYTE_SEQUENCE_NUMBER:
966 return PACKET_FLAGS_2BYTE_SEQUENCE;
967 case PACKET_4BYTE_SEQUENCE_NUMBER:
968 return PACKET_FLAGS_4BYTE_SEQUENCE;
969 case PACKET_6BYTE_SEQUENCE_NUMBER:
970 return PACKET_FLAGS_6BYTE_SEQUENCE;
971 default:
972 LOG(DFATAL) << "Unreachable case statement.";
973 return PACKET_FLAGS_6BYTE_SEQUENCE;
977 // static
978 QuicFramer::AckFrameInfo QuicFramer::GetAckFrameInfo(
979 const QuicAckFrame& frame) {
980 AckFrameInfo ack_info;
981 if (frame.missing_packets.empty()) {
982 return ack_info;
984 DCHECK_GE(frame.largest_observed, *frame.missing_packets.rbegin());
985 size_t cur_range_length = 0;
986 SequenceNumberSet::const_iterator iter = frame.missing_packets.begin();
987 QuicPacketSequenceNumber last_missing = *iter;
988 ++iter;
989 for (; iter != frame.missing_packets.end(); ++iter) {
990 if (cur_range_length < numeric_limits<uint8>::max() &&
991 *iter == (last_missing + 1)) {
992 ++cur_range_length;
993 } else {
994 ack_info.nack_ranges[last_missing - cur_range_length] =
995 static_cast<uint8>(cur_range_length);
996 cur_range_length = 0;
998 ack_info.max_delta = max(ack_info.max_delta, *iter - last_missing);
999 last_missing = *iter;
1001 // Include the last nack range.
1002 ack_info.nack_ranges[last_missing - cur_range_length] =
1003 static_cast<uint8>(cur_range_length);
1004 // Include the range to the largest observed.
1005 ack_info.max_delta =
1006 max(ack_info.max_delta, frame.largest_observed - last_missing);
1007 return ack_info;
1010 bool QuicFramer::ProcessPacketHeader(
1011 QuicPacketHeader* header,
1012 const QuicEncryptedPacket& packet) {
1013 if (!ProcessPacketSequenceNumber(header->public_header.sequence_number_length,
1014 &header->packet_sequence_number)) {
1015 set_detailed_error("Unable to read sequence number.");
1016 return RaiseError(QUIC_INVALID_PACKET_HEADER);
1019 if (header->packet_sequence_number == 0u) {
1020 set_detailed_error("Packet sequence numbers cannot be 0.");
1021 return RaiseError(QUIC_INVALID_PACKET_HEADER);
1024 if (!visitor_->OnUnauthenticatedHeader(*header)) {
1025 return false;
1028 if (!DecryptPayload(*header, packet)) {
1029 set_detailed_error("Unable to decrypt payload.");
1030 return RaiseError(QUIC_DECRYPTION_FAILURE);
1033 uint8 private_flags;
1034 if (!reader_->ReadBytes(&private_flags, 1)) {
1035 set_detailed_error("Unable to read private flags.");
1036 return RaiseError(QUIC_INVALID_PACKET_HEADER);
1039 if (private_flags > PACKET_PRIVATE_FLAGS_MAX) {
1040 set_detailed_error("Illegal private flags value.");
1041 return RaiseError(QUIC_INVALID_PACKET_HEADER);
1044 header->entropy_flag = (private_flags & PACKET_PRIVATE_FLAGS_ENTROPY) != 0;
1045 header->fec_flag = (private_flags & PACKET_PRIVATE_FLAGS_FEC) != 0;
1047 if ((private_flags & PACKET_PRIVATE_FLAGS_FEC_GROUP) != 0) {
1048 header->is_in_fec_group = IN_FEC_GROUP;
1049 uint8 first_fec_protected_packet_offset;
1050 if (!reader_->ReadBytes(&first_fec_protected_packet_offset, 1)) {
1051 set_detailed_error("Unable to read first fec protected packet offset.");
1052 return RaiseError(QUIC_INVALID_PACKET_HEADER);
1054 if (first_fec_protected_packet_offset >= header->packet_sequence_number) {
1055 set_detailed_error("First fec protected packet offset must be less "
1056 "than the sequence number.");
1057 return RaiseError(QUIC_INVALID_PACKET_HEADER);
1059 header->fec_group =
1060 header->packet_sequence_number - first_fec_protected_packet_offset;
1063 header->entropy_hash = GetPacketEntropyHash(*header);
1064 // Set the last sequence number after we have decrypted the packet
1065 // so we are confident is not attacker controlled.
1066 last_sequence_number_ = header->packet_sequence_number;
1067 return true;
1070 bool QuicFramer::ProcessPacketSequenceNumber(
1071 QuicSequenceNumberLength sequence_number_length,
1072 QuicPacketSequenceNumber* sequence_number) {
1073 QuicPacketSequenceNumber wire_sequence_number = 0u;
1074 if (!reader_->ReadBytes(&wire_sequence_number, sequence_number_length)) {
1075 return false;
1078 // TODO(ianswett): Explore the usefulness of trying multiple sequence numbers
1079 // in case the first guess is incorrect.
1080 *sequence_number =
1081 CalculatePacketSequenceNumberFromWire(sequence_number_length,
1082 wire_sequence_number);
1083 return true;
1086 bool QuicFramer::ProcessFrameData(const QuicPacketHeader& header) {
1087 if (reader_->IsDoneReading()) {
1088 set_detailed_error("Packet has no frames.");
1089 return RaiseError(QUIC_MISSING_PAYLOAD);
1091 while (!reader_->IsDoneReading()) {
1092 uint8 frame_type;
1093 if (!reader_->ReadBytes(&frame_type, 1)) {
1094 set_detailed_error("Unable to read frame type.");
1095 return RaiseError(QUIC_INVALID_FRAME_DATA);
1098 if (frame_type & kQuicFrameTypeSpecialMask) {
1099 // Stream Frame
1100 if (frame_type & kQuicFrameTypeStreamMask) {
1101 QuicStreamFrame frame;
1102 if (!ProcessStreamFrame(frame_type, &frame)) {
1103 return RaiseError(QUIC_INVALID_STREAM_DATA);
1105 if (!visitor_->OnStreamFrame(frame)) {
1106 DVLOG(1) << "Visitor asked to stop further processing.";
1107 // Returning true since there was no parsing error.
1108 return true;
1110 continue;
1113 // Ack Frame
1114 if (frame_type & kQuicFrameTypeAckMask) {
1115 QuicAckFrame frame;
1116 if (!ProcessAckFrame(frame_type, &frame)) {
1117 return RaiseError(QUIC_INVALID_ACK_DATA);
1119 if (!visitor_->OnAckFrame(frame)) {
1120 DVLOG(1) << "Visitor asked to stop further processing.";
1121 // Returning true since there was no parsing error.
1122 return true;
1124 continue;
1127 // Congestion Feedback Frame
1128 if (frame_type & kQuicFrameTypeCongestionFeedbackMask) {
1129 if (quic_version_ > QUIC_VERSION_22) {
1130 set_detailed_error("Congestion Feedback Frame has been deprecated.");
1131 DLOG(WARNING) << "Congestion Feedback Frame has been deprecated.";
1133 QuicCongestionFeedbackFrame frame;
1134 if (!ProcessCongestionFeedbackFrame(&frame)) {
1135 return RaiseError(QUIC_INVALID_CONGESTION_FEEDBACK_DATA);
1137 if (!visitor_->OnCongestionFeedbackFrame(frame)) {
1138 DVLOG(1) << "Visitor asked to stop further processing.";
1139 // Returning true since there was no parsing error.
1140 return true;
1142 continue;
1145 // This was a special frame type that did not match any
1146 // of the known ones. Error.
1147 set_detailed_error("Illegal frame type.");
1148 DLOG(WARNING) << "Illegal frame type: "
1149 << static_cast<int>(frame_type);
1150 return RaiseError(QUIC_INVALID_FRAME_DATA);
1153 switch (frame_type) {
1154 case PADDING_FRAME:
1155 // We're done with the packet.
1156 return true;
1158 case RST_STREAM_FRAME: {
1159 QuicRstStreamFrame frame;
1160 if (!ProcessRstStreamFrame(&frame)) {
1161 return RaiseError(QUIC_INVALID_RST_STREAM_DATA);
1163 if (!visitor_->OnRstStreamFrame(frame)) {
1164 DVLOG(1) << "Visitor asked to stop further processing.";
1165 // Returning true since there was no parsing error.
1166 return true;
1168 continue;
1171 case CONNECTION_CLOSE_FRAME: {
1172 QuicConnectionCloseFrame frame;
1173 if (!ProcessConnectionCloseFrame(&frame)) {
1174 return RaiseError(QUIC_INVALID_CONNECTION_CLOSE_DATA);
1177 if (!visitor_->OnConnectionCloseFrame(frame)) {
1178 DVLOG(1) << "Visitor asked to stop further processing.";
1179 // Returning true since there was no parsing error.
1180 return true;
1182 continue;
1185 case GOAWAY_FRAME: {
1186 QuicGoAwayFrame goaway_frame;
1187 if (!ProcessGoAwayFrame(&goaway_frame)) {
1188 return RaiseError(QUIC_INVALID_GOAWAY_DATA);
1190 if (!visitor_->OnGoAwayFrame(goaway_frame)) {
1191 DVLOG(1) << "Visitor asked to stop further processing.";
1192 // Returning true since there was no parsing error.
1193 return true;
1195 continue;
1198 case WINDOW_UPDATE_FRAME: {
1199 QuicWindowUpdateFrame window_update_frame;
1200 if (!ProcessWindowUpdateFrame(&window_update_frame)) {
1201 return RaiseError(QUIC_INVALID_WINDOW_UPDATE_DATA);
1203 if (!visitor_->OnWindowUpdateFrame(window_update_frame)) {
1204 DVLOG(1) << "Visitor asked to stop further processing.";
1205 // Returning true since there was no parsing error.
1206 return true;
1208 continue;
1211 case BLOCKED_FRAME: {
1212 QuicBlockedFrame blocked_frame;
1213 if (!ProcessBlockedFrame(&blocked_frame)) {
1214 return RaiseError(QUIC_INVALID_BLOCKED_DATA);
1216 if (!visitor_->OnBlockedFrame(blocked_frame)) {
1217 DVLOG(1) << "Visitor asked to stop further processing.";
1218 // Returning true since there was no parsing error.
1219 return true;
1221 continue;
1224 case STOP_WAITING_FRAME: {
1225 QuicStopWaitingFrame stop_waiting_frame;
1226 if (!ProcessStopWaitingFrame(header, &stop_waiting_frame)) {
1227 return RaiseError(QUIC_INVALID_STOP_WAITING_DATA);
1229 if (!visitor_->OnStopWaitingFrame(stop_waiting_frame)) {
1230 DVLOG(1) << "Visitor asked to stop further processing.";
1231 // Returning true since there was no parsing error.
1232 return true;
1234 continue;
1236 case PING_FRAME: {
1237 // Ping has no payload.
1238 QuicPingFrame ping_frame;
1239 if (!visitor_->OnPingFrame(ping_frame)) {
1240 DVLOG(1) << "Visitor asked to stop further processing.";
1241 // Returning true since there was no parsing error.
1242 return true;
1244 continue;
1247 default:
1248 set_detailed_error("Illegal frame type.");
1249 DLOG(WARNING) << "Illegal frame type: "
1250 << static_cast<int>(frame_type);
1251 return RaiseError(QUIC_INVALID_FRAME_DATA);
1255 return true;
1258 bool QuicFramer::ProcessStreamFrame(uint8 frame_type,
1259 QuicStreamFrame* frame) {
1260 uint8 stream_flags = frame_type;
1262 stream_flags &= ~kQuicFrameTypeStreamMask;
1264 // Read from right to left: StreamID, Offset, Data Length, Fin.
1265 const uint8 stream_id_length = (stream_flags & kQuicStreamIDLengthMask) + 1;
1266 stream_flags >>= kQuicStreamIdShift;
1268 uint8 offset_length = (stream_flags & kQuicStreamOffsetMask);
1269 // There is no encoding for 1 byte, only 0 and 2 through 8.
1270 if (offset_length > 0) {
1271 offset_length += 1;
1273 stream_flags >>= kQuicStreamOffsetShift;
1275 bool has_data_length =
1276 (stream_flags & kQuicStreamDataLengthMask) == kQuicStreamDataLengthMask;
1277 stream_flags >>= kQuicStreamDataLengthShift;
1279 frame->fin = (stream_flags & kQuicStreamFinMask) == kQuicStreamFinShift;
1281 frame->stream_id = 0;
1282 if (!reader_->ReadBytes(&frame->stream_id, stream_id_length)) {
1283 set_detailed_error("Unable to read stream_id.");
1284 return false;
1287 frame->offset = 0;
1288 if (!reader_->ReadBytes(&frame->offset, offset_length)) {
1289 set_detailed_error("Unable to read offset.");
1290 return false;
1293 StringPiece frame_data;
1294 if (has_data_length) {
1295 if (!reader_->ReadStringPiece16(&frame_data)) {
1296 set_detailed_error("Unable to read frame data.");
1297 return false;
1299 } else {
1300 if (!reader_->ReadStringPiece(&frame_data, reader_->BytesRemaining())) {
1301 set_detailed_error("Unable to read frame data.");
1302 return false;
1305 // Point frame to the right data.
1306 frame->data.Clear();
1307 if (!frame_data.empty()) {
1308 frame->data.Append(const_cast<char*>(frame_data.data()), frame_data.size());
1311 return true;
1314 bool QuicFramer::ProcessAckFrame(uint8 frame_type, QuicAckFrame* ack_frame) {
1315 // Determine the three lengths from the frame type: largest observed length,
1316 // missing sequence number length, and missing range length.
1317 const QuicSequenceNumberLength missing_sequence_number_length =
1318 ReadSequenceNumberLength(frame_type);
1319 frame_type >>= kQuicSequenceNumberLengthShift;
1320 const QuicSequenceNumberLength largest_observed_sequence_number_length =
1321 ReadSequenceNumberLength(frame_type);
1322 frame_type >>= kQuicSequenceNumberLengthShift;
1323 ack_frame->is_truncated = frame_type & kQuicAckTruncatedMask;
1324 frame_type >>= kQuicAckTruncatedShift;
1325 bool has_nacks = frame_type & kQuicHasNacksMask;
1327 if (!reader_->ReadBytes(&ack_frame->entropy_hash, 1)) {
1328 set_detailed_error("Unable to read entropy hash for received packets.");
1329 return false;
1332 if (!reader_->ReadBytes(&ack_frame->largest_observed,
1333 largest_observed_sequence_number_length)) {
1334 set_detailed_error("Unable to read largest observed.");
1335 return false;
1338 uint64 delta_time_largest_observed_us;
1339 if (!reader_->ReadUFloat16(&delta_time_largest_observed_us)) {
1340 set_detailed_error("Unable to read delta time largest observed.");
1341 return false;
1344 if (delta_time_largest_observed_us == kUFloat16MaxValue) {
1345 ack_frame->delta_time_largest_observed = QuicTime::Delta::Infinite();
1346 } else {
1347 ack_frame->delta_time_largest_observed =
1348 QuicTime::Delta::FromMicroseconds(delta_time_largest_observed_us);
1351 if (!ProcessTimestampsInAckFrame(ack_frame)) {
1352 return false;
1355 if (!has_nacks) {
1356 return true;
1359 uint8 num_missing_ranges;
1360 if (!reader_->ReadBytes(&num_missing_ranges, 1)) {
1361 set_detailed_error("Unable to read num missing packet ranges.");
1362 return false;
1365 QuicPacketSequenceNumber last_sequence_number = ack_frame->largest_observed;
1366 for (size_t i = 0; i < num_missing_ranges; ++i) {
1367 QuicPacketSequenceNumber missing_delta = 0;
1368 if (!reader_->ReadBytes(&missing_delta, missing_sequence_number_length)) {
1369 set_detailed_error("Unable to read missing sequence number delta.");
1370 return false;
1372 last_sequence_number -= missing_delta;
1373 QuicPacketSequenceNumber range_length = 0;
1374 if (!reader_->ReadBytes(&range_length, PACKET_1BYTE_SEQUENCE_NUMBER)) {
1375 set_detailed_error("Unable to read missing sequence number range.");
1376 return false;
1378 for (size_t i = 0; i <= range_length; ++i) {
1379 ack_frame->missing_packets.insert(last_sequence_number - i);
1381 // Subtract an extra 1 to ensure ranges are represented efficiently and
1382 // can't overlap by 1 sequence number. This allows a missing_delta of 0
1383 // to represent an adjacent nack range.
1384 last_sequence_number -= (range_length + 1);
1387 // Parse the revived packets list.
1388 uint8 num_revived_packets;
1389 if (!reader_->ReadBytes(&num_revived_packets, 1)) {
1390 set_detailed_error("Unable to read num revived packets.");
1391 return false;
1394 for (size_t i = 0; i < num_revived_packets; ++i) {
1395 QuicPacketSequenceNumber revived_packet = 0;
1396 if (!reader_->ReadBytes(&revived_packet,
1397 largest_observed_sequence_number_length)) {
1398 set_detailed_error("Unable to read revived packet.");
1399 return false;
1402 ack_frame->revived_packets.insert(revived_packet);
1405 return true;
1408 bool QuicFramer::ProcessTimestampsInAckFrame(QuicAckFrame* ack_frame) {
1409 if (version() > QUIC_VERSION_22 && !ack_frame->is_truncated) {
1410 uint8 num_received_packets;
1411 if (!reader_->ReadBytes(&num_received_packets, 1)) {
1412 set_detailed_error("Unable to read num received packets.");
1413 return false;
1416 if (num_received_packets > 0) {
1417 uint8 delta_from_largest_observed;
1418 if (!reader_->ReadBytes(&delta_from_largest_observed,
1419 PACKET_1BYTE_SEQUENCE_NUMBER)) {
1420 set_detailed_error(
1421 "Unable to read sequence delta in received packets.");
1422 return false;
1424 QuicPacketSequenceNumber seq_num = ack_frame->largest_observed -
1425 delta_from_largest_observed;
1427 // Time delta from the framer creation.
1428 uint32 time_delta_us;
1429 if (!reader_->ReadBytes(&time_delta_us, sizeof(time_delta_us))) {
1430 set_detailed_error("Unable to read time delta in received packets.");
1431 return false;
1434 last_timestamp_ = CalculateTimestampFromWire(time_delta_us);
1436 ack_frame->received_packet_times.push_back(
1437 make_pair(seq_num, creation_time_.Add(last_timestamp_)));
1439 for (uint8 i = 1; i < num_received_packets; ++i) {
1440 if (!reader_->ReadBytes(&delta_from_largest_observed,
1441 PACKET_1BYTE_SEQUENCE_NUMBER)) {
1442 set_detailed_error(
1443 "Unable to read sequence delta in received packets.");
1444 return false;
1446 seq_num = ack_frame->largest_observed - delta_from_largest_observed;
1448 // Time delta from the previous timestamp.
1449 uint64 incremental_time_delta_us;
1450 if (!reader_->ReadUFloat16(&incremental_time_delta_us)) {
1451 set_detailed_error(
1452 "Unable to read incremental time delta in received packets.");
1453 return false;
1456 last_timestamp_ = last_timestamp_.Add(
1457 QuicTime::Delta::FromMicroseconds(incremental_time_delta_us));
1458 ack_frame->received_packet_times.push_back(
1459 make_pair(seq_num, creation_time_.Add(last_timestamp_)));
1463 return true;
1466 bool QuicFramer::ProcessStopWaitingFrame(const QuicPacketHeader& header,
1467 QuicStopWaitingFrame* stop_waiting) {
1468 if (!reader_->ReadBytes(&stop_waiting->entropy_hash, 1)) {
1469 set_detailed_error("Unable to read entropy hash for sent packets.");
1470 return false;
1473 QuicPacketSequenceNumber least_unacked_delta = 0;
1474 if (!reader_->ReadBytes(&least_unacked_delta,
1475 header.public_header.sequence_number_length)) {
1476 set_detailed_error("Unable to read least unacked delta.");
1477 return false;
1479 DCHECK_GE(header.packet_sequence_number, least_unacked_delta);
1480 stop_waiting->least_unacked =
1481 header.packet_sequence_number - least_unacked_delta;
1483 return true;
1486 bool QuicFramer::ProcessCongestionFeedbackFrame(
1487 QuicCongestionFeedbackFrame* frame) {
1488 uint8 feedback_type;
1489 if (!reader_->ReadBytes(&feedback_type, 1)) {
1490 set_detailed_error("Unable to read congestion feedback type.");
1491 return false;
1493 frame->type =
1494 static_cast<CongestionFeedbackType>(feedback_type);
1496 switch (frame->type) {
1497 case kTCP: {
1498 CongestionFeedbackMessageTCP* tcp = &frame->tcp;
1499 uint16 receive_window = 0;
1500 if (!reader_->ReadUInt16(&receive_window)) {
1501 set_detailed_error("Unable to read receive window.");
1502 return false;
1504 // Simple bit packing, don't send the 4 least significant bits.
1505 tcp->receive_window = static_cast<QuicByteCount>(receive_window) << 4;
1506 break;
1508 default:
1509 set_detailed_error("Illegal congestion feedback type.");
1510 DLOG(WARNING) << "Illegal congestion feedback type: "
1511 << frame->type;
1512 return RaiseError(QUIC_INVALID_FRAME_DATA);
1515 return true;
1518 bool QuicFramer::ProcessRstStreamFrame(QuicRstStreamFrame* frame) {
1519 if (!reader_->ReadUInt32(&frame->stream_id)) {
1520 set_detailed_error("Unable to read stream_id.");
1521 return false;
1524 if (!reader_->ReadUInt64(&frame->byte_offset)) {
1525 set_detailed_error("Unable to read rst stream sent byte offset.");
1526 return false;
1529 uint32 error_code;
1530 if (!reader_->ReadUInt32(&error_code)) {
1531 set_detailed_error("Unable to read rst stream error code.");
1532 return false;
1535 if (error_code >= QUIC_STREAM_LAST_ERROR) {
1536 set_detailed_error("Invalid rst stream error code.");
1537 return false;
1540 frame->error_code = static_cast<QuicRstStreamErrorCode>(error_code);
1542 StringPiece error_details;
1543 if (!reader_->ReadStringPiece16(&error_details)) {
1544 set_detailed_error("Unable to read rst stream error details.");
1545 return false;
1547 frame->error_details = error_details.as_string();
1549 return true;
1552 bool QuicFramer::ProcessConnectionCloseFrame(QuicConnectionCloseFrame* frame) {
1553 uint32 error_code;
1554 if (!reader_->ReadUInt32(&error_code)) {
1555 set_detailed_error("Unable to read connection close error code.");
1556 return false;
1559 if (error_code >= QUIC_LAST_ERROR) {
1560 set_detailed_error("Invalid error code.");
1561 return false;
1564 frame->error_code = static_cast<QuicErrorCode>(error_code);
1566 StringPiece error_details;
1567 if (!reader_->ReadStringPiece16(&error_details)) {
1568 set_detailed_error("Unable to read connection close error details.");
1569 return false;
1571 frame->error_details = error_details.as_string();
1573 return true;
1576 bool QuicFramer::ProcessGoAwayFrame(QuicGoAwayFrame* frame) {
1577 uint32 error_code;
1578 if (!reader_->ReadUInt32(&error_code)) {
1579 set_detailed_error("Unable to read go away error code.");
1580 return false;
1582 frame->error_code = static_cast<QuicErrorCode>(error_code);
1584 if (error_code >= QUIC_LAST_ERROR) {
1585 set_detailed_error("Invalid error code.");
1586 return false;
1589 uint32 stream_id;
1590 if (!reader_->ReadUInt32(&stream_id)) {
1591 set_detailed_error("Unable to read last good stream id.");
1592 return false;
1594 frame->last_good_stream_id = static_cast<QuicStreamId>(stream_id);
1596 StringPiece reason_phrase;
1597 if (!reader_->ReadStringPiece16(&reason_phrase)) {
1598 set_detailed_error("Unable to read goaway reason.");
1599 return false;
1601 frame->reason_phrase = reason_phrase.as_string();
1603 return true;
1606 bool QuicFramer::ProcessWindowUpdateFrame(QuicWindowUpdateFrame* frame) {
1607 if (!reader_->ReadUInt32(&frame->stream_id)) {
1608 set_detailed_error("Unable to read stream_id.");
1609 return false;
1612 if (!reader_->ReadUInt64(&frame->byte_offset)) {
1613 set_detailed_error("Unable to read window byte_offset.");
1614 return false;
1617 return true;
1620 bool QuicFramer::ProcessBlockedFrame(QuicBlockedFrame* frame) {
1621 if (!reader_->ReadUInt32(&frame->stream_id)) {
1622 set_detailed_error("Unable to read stream_id.");
1623 return false;
1626 return true;
1629 // static
1630 StringPiece QuicFramer::GetAssociatedDataFromEncryptedPacket(
1631 const QuicEncryptedPacket& encrypted,
1632 QuicConnectionIdLength connection_id_length,
1633 bool includes_version,
1634 QuicSequenceNumberLength sequence_number_length) {
1635 return StringPiece(
1636 encrypted.data() + kStartOfHashData, GetStartOfEncryptedData(
1637 connection_id_length, includes_version, sequence_number_length)
1638 - kStartOfHashData);
1641 void QuicFramer::SetDecrypter(QuicDecrypter* decrypter,
1642 EncryptionLevel level) {
1643 DCHECK(alternative_decrypter_.get() == nullptr);
1644 DCHECK_GE(level, decrypter_level_);
1645 decrypter_.reset(decrypter);
1646 decrypter_level_ = level;
1649 void QuicFramer::SetAlternativeDecrypter(QuicDecrypter* decrypter,
1650 EncryptionLevel level,
1651 bool latch_once_used) {
1652 alternative_decrypter_.reset(decrypter);
1653 alternative_decrypter_level_ = level;
1654 alternative_decrypter_latch_ = latch_once_used;
1657 const QuicDecrypter* QuicFramer::decrypter() const {
1658 return decrypter_.get();
1661 const QuicDecrypter* QuicFramer::alternative_decrypter() const {
1662 return alternative_decrypter_.get();
1665 void QuicFramer::SetEncrypter(EncryptionLevel level,
1666 QuicEncrypter* encrypter) {
1667 DCHECK_GE(level, 0);
1668 DCHECK_LT(level, NUM_ENCRYPTION_LEVELS);
1669 encrypter_[level].reset(encrypter);
1672 const QuicEncrypter* QuicFramer::encrypter(EncryptionLevel level) const {
1673 DCHECK_GE(level, 0);
1674 DCHECK_LT(level, NUM_ENCRYPTION_LEVELS);
1675 DCHECK(encrypter_[level].get() != nullptr);
1676 return encrypter_[level].get();
1679 QuicEncryptedPacket* QuicFramer::EncryptPacket(
1680 EncryptionLevel level,
1681 QuicPacketSequenceNumber packet_sequence_number,
1682 const QuicPacket& packet) {
1683 DCHECK(encrypter_[level].get() != nullptr);
1685 scoped_ptr<QuicData> out(encrypter_[level]->EncryptPacket(
1686 packet_sequence_number, packet.AssociatedData(), packet.Plaintext()));
1687 if (out.get() == nullptr) {
1688 RaiseError(QUIC_ENCRYPTION_FAILURE);
1689 return nullptr;
1691 StringPiece header_data = packet.BeforePlaintext();
1692 size_t len = header_data.length() + out->length();
1693 char* buffer = new char[len];
1694 // TODO(rch): eliminate this buffer copy by passing in a buffer to Encrypt().
1695 memcpy(buffer, header_data.data(), header_data.length());
1696 memcpy(buffer + header_data.length(), out->data(), out->length());
1697 return new QuicEncryptedPacket(buffer, len, true);
1700 size_t QuicFramer::GetMaxPlaintextSize(size_t ciphertext_size) {
1701 // In order to keep the code simple, we don't have the current encryption
1702 // level to hand. Both the NullEncrypter and AES-GCM have a tag length of 12.
1703 size_t min_plaintext_size = ciphertext_size;
1705 for (int i = ENCRYPTION_NONE; i < NUM_ENCRYPTION_LEVELS; i++) {
1706 if (encrypter_[i].get() != nullptr) {
1707 size_t size = encrypter_[i]->GetMaxPlaintextSize(ciphertext_size);
1708 if (size < min_plaintext_size) {
1709 min_plaintext_size = size;
1714 return min_plaintext_size;
1717 bool QuicFramer::DecryptPayload(const QuicPacketHeader& header,
1718 const QuicEncryptedPacket& packet) {
1719 StringPiece encrypted;
1720 if (!reader_->ReadStringPiece(&encrypted, reader_->BytesRemaining())) {
1721 return false;
1723 DCHECK(decrypter_.get() != nullptr);
1724 decrypted_.reset(decrypter_->DecryptPacket(
1725 header.packet_sequence_number,
1726 GetAssociatedDataFromEncryptedPacket(
1727 packet,
1728 header.public_header.connection_id_length,
1729 header.public_header.version_flag,
1730 header.public_header.sequence_number_length),
1731 encrypted));
1732 if (decrypted_.get() != nullptr) {
1733 visitor_->OnDecryptedPacket(decrypter_level_);
1734 } else if (alternative_decrypter_.get() != nullptr) {
1735 decrypted_.reset(alternative_decrypter_->DecryptPacket(
1736 header.packet_sequence_number,
1737 GetAssociatedDataFromEncryptedPacket(
1738 packet,
1739 header.public_header.connection_id_length,
1740 header.public_header.version_flag,
1741 header.public_header.sequence_number_length),
1742 encrypted));
1743 if (decrypted_.get() != nullptr) {
1744 visitor_->OnDecryptedPacket(alternative_decrypter_level_);
1745 if (alternative_decrypter_latch_) {
1746 // Switch to the alternative decrypter and latch so that we cannot
1747 // switch back.
1748 decrypter_.reset(alternative_decrypter_.release());
1749 decrypter_level_ = alternative_decrypter_level_;
1750 alternative_decrypter_level_ = ENCRYPTION_NONE;
1751 } else {
1752 // Switch the alternative decrypter so that we use it first next time.
1753 decrypter_.swap(alternative_decrypter_);
1754 EncryptionLevel level = alternative_decrypter_level_;
1755 alternative_decrypter_level_ = decrypter_level_;
1756 decrypter_level_ = level;
1761 if (decrypted_.get() == nullptr) {
1762 DLOG(WARNING) << "DecryptPacket failed for sequence_number:"
1763 << header.packet_sequence_number;
1764 return false;
1767 reader_.reset(new QuicDataReader(decrypted_->data(), decrypted_->length()));
1768 return true;
1771 size_t QuicFramer::GetAckFrameSize(
1772 const QuicAckFrame& ack,
1773 QuicSequenceNumberLength sequence_number_length) {
1774 AckFrameInfo ack_info = GetAckFrameInfo(ack);
1775 QuicSequenceNumberLength largest_observed_length =
1776 GetMinSequenceNumberLength(ack.largest_observed);
1777 QuicSequenceNumberLength missing_sequence_number_length =
1778 GetMinSequenceNumberLength(ack_info.max_delta);
1780 size_t ack_size = GetMinAckFrameSize(sequence_number_length,
1781 largest_observed_length);
1782 if (!ack_info.nack_ranges.empty()) {
1783 ack_size += kNumberOfNackRangesSize + kNumberOfRevivedPacketsSize;
1784 ack_size += min(ack_info.nack_ranges.size(), kMaxNackRanges) *
1785 (missing_sequence_number_length + PACKET_1BYTE_SEQUENCE_NUMBER);
1786 ack_size += min(ack.revived_packets.size(),
1787 kMaxRevivedPackets) * largest_observed_length;
1790 // In version 23, if the ack will be truncated due to too many nack ranges,
1791 // then do not include the number of timestamps (1 byte).
1792 if (version() > QUIC_VERSION_22 &&
1793 ack_info.nack_ranges.size() <= kMaxNackRanges) {
1794 // 1 byte for the number of timestamps.
1795 ack_size += 1;
1796 if (ack.received_packet_times.size() > 0) {
1797 // 1 byte for sequence number, 4 bytes for timestamp for the first
1798 // packet.
1799 ack_size += 5;
1801 // 1 byte for sequence number, 2 bytes for timestamp for the other
1802 // packets.
1803 ack_size += 3 * (ack.received_packet_times.size() - 1);
1807 return ack_size;
1810 size_t QuicFramer::ComputeFrameLength(
1811 const QuicFrame& frame,
1812 bool last_frame_in_packet,
1813 InFecGroup is_in_fec_group,
1814 QuicSequenceNumberLength sequence_number_length) {
1815 switch (frame.type) {
1816 case STREAM_FRAME:
1817 return GetMinStreamFrameSize(frame.stream_frame->stream_id,
1818 frame.stream_frame->offset,
1819 last_frame_in_packet,
1820 is_in_fec_group) +
1821 frame.stream_frame->data.TotalBufferSize();
1822 case ACK_FRAME: {
1823 return GetAckFrameSize(*frame.ack_frame, sequence_number_length);
1825 case CONGESTION_FEEDBACK_FRAME: {
1826 size_t len = kQuicFrameTypeSize;
1827 const QuicCongestionFeedbackFrame& congestion_feedback =
1828 *frame.congestion_feedback_frame;
1829 len += 1; // Congestion feedback type.
1831 switch (congestion_feedback.type) {
1832 case kTCP:
1833 len += 2; // Receive window.
1834 break;
1835 default:
1836 set_detailed_error("Illegal feedback type.");
1837 DVLOG(1) << "Illegal feedback type: " << congestion_feedback.type;
1838 break;
1840 return len;
1842 case STOP_WAITING_FRAME:
1843 return GetStopWaitingFrameSize(sequence_number_length);
1844 case PING_FRAME:
1845 // Ping has no payload.
1846 return kQuicFrameTypeSize;
1847 case RST_STREAM_FRAME:
1848 return GetMinRstStreamFrameSize() +
1849 frame.rst_stream_frame->error_details.size();
1850 case CONNECTION_CLOSE_FRAME:
1851 return GetMinConnectionCloseFrameSize() +
1852 frame.connection_close_frame->error_details.size();
1853 case GOAWAY_FRAME:
1854 return GetMinGoAwayFrameSize() + frame.goaway_frame->reason_phrase.size();
1855 case WINDOW_UPDATE_FRAME:
1856 return GetWindowUpdateFrameSize();
1857 case BLOCKED_FRAME:
1858 return GetBlockedFrameSize();
1859 case PADDING_FRAME:
1860 DCHECK(false);
1861 return 0;
1862 case NUM_FRAME_TYPES:
1863 DCHECK(false);
1864 return 0;
1867 // Not reachable, but some Chrome compilers can't figure that out. *sigh*
1868 DCHECK(false);
1869 return 0;
1872 bool QuicFramer::AppendTypeByte(const QuicFrame& frame,
1873 bool no_stream_frame_length,
1874 QuicDataWriter* writer) {
1875 uint8 type_byte = 0;
1876 switch (frame.type) {
1877 case STREAM_FRAME: {
1878 if (frame.stream_frame == nullptr) {
1879 LOG(DFATAL) << "Failed to append STREAM frame with no stream_frame.";
1881 // Fin bit.
1882 type_byte |= frame.stream_frame->fin ? kQuicStreamFinMask : 0;
1884 // Data Length bit.
1885 type_byte <<= kQuicStreamDataLengthShift;
1886 type_byte |= no_stream_frame_length ? 0: kQuicStreamDataLengthMask;
1888 // Offset 3 bits.
1889 type_byte <<= kQuicStreamOffsetShift;
1890 const size_t offset_len = GetStreamOffsetSize(frame.stream_frame->offset);
1891 if (offset_len > 0) {
1892 type_byte |= offset_len - 1;
1895 // stream id 2 bits.
1896 type_byte <<= kQuicStreamIdShift;
1897 type_byte |= GetStreamIdSize(frame.stream_frame->stream_id) - 1;
1898 type_byte |= kQuicFrameTypeStreamMask; // Set Stream Frame Type to 1.
1899 break;
1901 case ACK_FRAME:
1902 return true;
1903 case CONGESTION_FEEDBACK_FRAME: {
1904 // TODO(ianswett): Use extra 5 bits in the congestion feedback framing.
1905 type_byte = kQuicFrameTypeCongestionFeedbackMask;
1906 break;
1908 default:
1909 type_byte = static_cast<uint8>(frame.type);
1910 break;
1913 return writer->WriteUInt8(type_byte);
1916 // static
1917 bool QuicFramer::AppendPacketSequenceNumber(
1918 QuicSequenceNumberLength sequence_number_length,
1919 QuicPacketSequenceNumber packet_sequence_number,
1920 QuicDataWriter* writer) {
1921 // Ensure the entire sequence number can be written.
1922 if (writer->capacity() - writer->length() <
1923 static_cast<size_t>(sequence_number_length)) {
1924 return false;
1926 switch (sequence_number_length) {
1927 case PACKET_1BYTE_SEQUENCE_NUMBER:
1928 return writer->WriteUInt8(
1929 packet_sequence_number & k1ByteSequenceNumberMask);
1930 break;
1931 case PACKET_2BYTE_SEQUENCE_NUMBER:
1932 return writer->WriteUInt16(
1933 packet_sequence_number & k2ByteSequenceNumberMask);
1934 break;
1935 case PACKET_4BYTE_SEQUENCE_NUMBER:
1936 return writer->WriteUInt32(
1937 packet_sequence_number & k4ByteSequenceNumberMask);
1938 break;
1939 case PACKET_6BYTE_SEQUENCE_NUMBER:
1940 return writer->WriteUInt48(
1941 packet_sequence_number & k6ByteSequenceNumberMask);
1942 break;
1943 default:
1944 DCHECK(false) << "sequence_number_length: " << sequence_number_length;
1945 return false;
1949 bool QuicFramer::AppendStreamFrame(
1950 const QuicStreamFrame& frame,
1951 bool no_stream_frame_length,
1952 QuicDataWriter* writer) {
1953 if (!writer->WriteBytes(&frame.stream_id, GetStreamIdSize(frame.stream_id))) {
1954 LOG(DFATAL) << "Writing stream id size failed.";
1955 return false;
1957 if (!writer->WriteBytes(&frame.offset, GetStreamOffsetSize(frame.offset))) {
1958 LOG(DFATAL) << "Writing offset size failed.";
1959 return false;
1961 if (!no_stream_frame_length) {
1962 if ((frame.data.TotalBufferSize() > numeric_limits<uint16>::max()) ||
1963 !writer->WriteUInt16(
1964 static_cast<uint16>(frame.data.TotalBufferSize()))) {
1965 LOG(DFATAL) << "Writing stream frame length failed";
1966 return false;
1970 if (!writer->WriteIOVector(frame.data)) {
1971 LOG(DFATAL) << "Writing frame data failed.";
1972 return false;
1974 return true;
1977 // static
1978 void QuicFramer::set_version(const QuicVersion version) {
1979 DCHECK(IsSupportedVersion(version)) << QuicVersionToString(version);
1980 quic_version_ = version;
1983 bool QuicFramer::AppendAckFrameAndTypeByte(
1984 const QuicPacketHeader& header,
1985 const QuicAckFrame& frame,
1986 QuicDataWriter* writer) {
1987 AckFrameInfo ack_info = GetAckFrameInfo(frame);
1988 QuicPacketSequenceNumber ack_largest_observed = frame.largest_observed;
1989 QuicSequenceNumberLength largest_observed_length =
1990 GetMinSequenceNumberLength(ack_largest_observed);
1991 QuicSequenceNumberLength missing_sequence_number_length =
1992 GetMinSequenceNumberLength(ack_info.max_delta);
1993 // Determine whether we need to truncate ranges.
1994 size_t available_range_bytes = writer->capacity() - writer->length() -
1995 kNumberOfRevivedPacketsSize - kNumberOfNackRangesSize -
1996 GetMinAckFrameSize(header.public_header.sequence_number_length,
1997 largest_observed_length);
1998 size_t max_num_ranges = available_range_bytes /
1999 (missing_sequence_number_length + PACKET_1BYTE_SEQUENCE_NUMBER);
2000 max_num_ranges = min(kMaxNackRanges, max_num_ranges);
2001 bool truncated = ack_info.nack_ranges.size() > max_num_ranges;
2002 DVLOG_IF(1, truncated) << "Truncating ack from "
2003 << ack_info.nack_ranges.size() << " ranges to "
2004 << max_num_ranges;
2005 // Write out the type byte by setting the low order bits and doing shifts
2006 // to make room for the next bit flags to be set.
2007 // Whether there are any nacks.
2008 uint8 type_byte = ack_info.nack_ranges.empty() ? 0 : kQuicHasNacksMask;
2010 // truncating bit.
2011 type_byte <<= kQuicAckTruncatedShift;
2012 type_byte |= truncated ? kQuicAckTruncatedMask : 0;
2014 // Largest observed sequence number length.
2015 type_byte <<= kQuicSequenceNumberLengthShift;
2016 type_byte |= GetSequenceNumberFlags(largest_observed_length);
2018 // Missing sequence number length.
2019 type_byte <<= kQuicSequenceNumberLengthShift;
2020 type_byte |= GetSequenceNumberFlags(missing_sequence_number_length);
2022 type_byte |= kQuicFrameTypeAckMask;
2024 if (!writer->WriteUInt8(type_byte)) {
2025 return false;
2028 QuicPacketEntropyHash ack_entropy_hash = frame.entropy_hash;
2029 NackRangeMap::reverse_iterator ack_iter = ack_info.nack_ranges.rbegin();
2030 if (truncated) {
2031 // Skip the nack ranges which the truncated ack won't include and set
2032 // a correct largest observed for the truncated ack.
2033 for (size_t i = 1; i < (ack_info.nack_ranges.size() - max_num_ranges);
2034 ++i) {
2035 ++ack_iter;
2037 // If the last range is followed by acks, include them.
2038 // If the last range is followed by another range, specify the end of the
2039 // range as the largest_observed.
2040 ack_largest_observed = ack_iter->first - 1;
2041 // Also update the entropy so it matches the largest observed.
2042 ack_entropy_hash = entropy_calculator_->EntropyHash(ack_largest_observed);
2043 ++ack_iter;
2046 if (!writer->WriteUInt8(ack_entropy_hash)) {
2047 return false;
2050 if (!AppendPacketSequenceNumber(largest_observed_length,
2051 ack_largest_observed, writer)) {
2052 return false;
2055 uint64 delta_time_largest_observed_us = kUFloat16MaxValue;
2056 if (!frame.delta_time_largest_observed.IsInfinite()) {
2057 DCHECK_LE(0u, frame.delta_time_largest_observed.ToMicroseconds());
2058 delta_time_largest_observed_us =
2059 frame.delta_time_largest_observed.ToMicroseconds();
2062 if (!writer->WriteUFloat16(delta_time_largest_observed_us)) {
2063 return false;
2066 // Timestamp goes at the end of the required fields.
2067 if (version() > QUIC_VERSION_22 && !truncated) {
2068 if (!AppendTimestampToAckFrame(frame, writer)) {
2069 return false;
2073 if (ack_info.nack_ranges.empty()) {
2074 return true;
2077 const uint8 num_missing_ranges =
2078 static_cast<uint8>(min(ack_info.nack_ranges.size(), max_num_ranges));
2079 if (!writer->WriteBytes(&num_missing_ranges, 1)) {
2080 return false;
2083 int num_ranges_written = 0;
2084 QuicPacketSequenceNumber last_sequence_written = ack_largest_observed;
2085 for (; ack_iter != ack_info.nack_ranges.rend(); ++ack_iter) {
2086 // Calculate the delta to the last number in the range.
2087 QuicPacketSequenceNumber missing_delta =
2088 last_sequence_written - (ack_iter->first + ack_iter->second);
2089 if (!AppendPacketSequenceNumber(missing_sequence_number_length,
2090 missing_delta, writer)) {
2091 return false;
2093 if (!AppendPacketSequenceNumber(PACKET_1BYTE_SEQUENCE_NUMBER,
2094 ack_iter->second, writer)) {
2095 return false;
2097 // Subtract 1 so a missing_delta of 0 means an adjacent range.
2098 last_sequence_written = ack_iter->first - 1;
2099 ++num_ranges_written;
2101 DCHECK_EQ(num_missing_ranges, num_ranges_written);
2103 // Append revived packets.
2104 // If not all the revived packets fit, only mention the ones that do.
2105 uint8 num_revived_packets =
2106 static_cast<uint8>(min(frame.revived_packets.size(), kMaxRevivedPackets));
2107 num_revived_packets = static_cast<uint8>(min(
2108 static_cast<size_t>(num_revived_packets),
2109 (writer->capacity() - writer->length()) / largest_observed_length));
2110 if (!writer->WriteBytes(&num_revived_packets, 1)) {
2111 return false;
2114 SequenceNumberSet::const_iterator iter = frame.revived_packets.begin();
2115 for (int i = 0; i < num_revived_packets; ++i, ++iter) {
2116 LOG_IF(DFATAL, !ContainsKey(frame.missing_packets, *iter));
2117 if (!AppendPacketSequenceNumber(largest_observed_length,
2118 *iter, writer)) {
2119 return false;
2123 return true;
2126 bool QuicFramer::AppendCongestionFeedbackFrame(
2127 const QuicCongestionFeedbackFrame& frame,
2128 QuicDataWriter* writer) {
2129 if (!writer->WriteBytes(&frame.type, 1)) {
2130 return false;
2133 switch (frame.type) {
2134 case kTCP: {
2135 const CongestionFeedbackMessageTCP& tcp = frame.tcp;
2136 DCHECK_LE(tcp.receive_window, 1u << 20);
2137 // Simple bit packing, don't send the 4 least significant bits.
2138 uint16 receive_window = static_cast<uint16>(tcp.receive_window >> 4);
2139 if (!writer->WriteUInt16(receive_window)) {
2140 return false;
2142 break;
2144 default:
2145 return false;
2148 return true;
2151 bool QuicFramer::AppendTimestampToAckFrame(const QuicAckFrame& frame,
2152 QuicDataWriter* writer) {
2153 DCHECK_GE(version(), QUIC_VERSION_23);
2154 DCHECK_GE(numeric_limits<uint8>::max(), frame.received_packet_times.size());
2155 // num_received_packets is only 1 byte.
2156 if (frame.received_packet_times.size() > numeric_limits<uint8>::max()) {
2157 return false;
2160 uint8 num_received_packets = frame.received_packet_times.size();
2162 if (!writer->WriteBytes(&num_received_packets, 1)) {
2163 return false;
2165 if (num_received_packets == 0) {
2166 return true;
2169 PacketTimeList::const_iterator it = frame.received_packet_times.begin();
2170 QuicPacketSequenceNumber sequence_number = it->first;
2171 QuicPacketSequenceNumber delta_from_largest_observed =
2172 frame.largest_observed - sequence_number;
2174 DCHECK_GE(numeric_limits<uint8>::max(), delta_from_largest_observed);
2175 if (delta_from_largest_observed > numeric_limits<uint8>::max()) {
2176 return false;
2179 if (!writer->WriteUInt8(
2180 delta_from_largest_observed & k1ByteSequenceNumberMask)) {
2181 return false;
2184 // Use the lowest 4 bytes of the time delta from the creation_time_.
2185 const uint64 time_epoch_delta_us = GG_UINT64_C(1) << 32;
2186 uint32 time_delta_us =
2187 static_cast<uint32>(it->second.Subtract(creation_time_).ToMicroseconds()
2188 & (time_epoch_delta_us - 1));
2189 if (!writer->WriteBytes(&time_delta_us, sizeof(time_delta_us))) {
2190 return false;
2193 QuicTime prev_time = it->second;
2195 for (++it; it != frame.received_packet_times.end(); ++it) {
2196 sequence_number = it->first;
2197 delta_from_largest_observed = frame.largest_observed - sequence_number;
2199 if (delta_from_largest_observed > numeric_limits<uint8>::max()) {
2200 return false;
2203 if (!writer->WriteUInt8(
2204 delta_from_largest_observed & k1ByteSequenceNumberMask)) {
2205 return false;
2208 uint64 time_delta_us = it->second.Subtract(prev_time).ToMicroseconds();
2209 prev_time = it->second;
2210 if (!writer->WriteUFloat16(time_delta_us)) {
2211 return false;
2214 return true;
2217 bool QuicFramer::AppendStopWaitingFrame(
2218 const QuicPacketHeader& header,
2219 const QuicStopWaitingFrame& frame,
2220 QuicDataWriter* writer) {
2221 DCHECK_GE(header.packet_sequence_number, frame.least_unacked);
2222 const QuicPacketSequenceNumber least_unacked_delta =
2223 header.packet_sequence_number - frame.least_unacked;
2224 const QuicPacketSequenceNumber length_shift =
2225 header.public_header.sequence_number_length * 8;
2226 if (!writer->WriteUInt8(frame.entropy_hash)) {
2227 LOG(DFATAL) << " hash failed";
2228 return false;
2231 if (least_unacked_delta >> length_shift > 0) {
2232 LOG(DFATAL) << "sequence_number_length "
2233 << header.public_header.sequence_number_length
2234 << " is too small for least_unacked_delta: "
2235 << least_unacked_delta;
2236 return false;
2238 if (!AppendPacketSequenceNumber(header.public_header.sequence_number_length,
2239 least_unacked_delta, writer)) {
2240 LOG(DFATAL) << " seq failed: "
2241 << header.public_header.sequence_number_length;
2242 return false;
2245 return true;
2248 bool QuicFramer::AppendRstStreamFrame(
2249 const QuicRstStreamFrame& frame,
2250 QuicDataWriter* writer) {
2251 if (!writer->WriteUInt32(frame.stream_id)) {
2252 return false;
2255 if (!writer->WriteUInt64(frame.byte_offset)) {
2256 return false;
2259 uint32 error_code = static_cast<uint32>(frame.error_code);
2260 if (!writer->WriteUInt32(error_code)) {
2261 return false;
2264 if (!writer->WriteStringPiece16(frame.error_details)) {
2265 return false;
2267 return true;
2270 bool QuicFramer::AppendConnectionCloseFrame(
2271 const QuicConnectionCloseFrame& frame,
2272 QuicDataWriter* writer) {
2273 uint32 error_code = static_cast<uint32>(frame.error_code);
2274 if (!writer->WriteUInt32(error_code)) {
2275 return false;
2277 if (!writer->WriteStringPiece16(frame.error_details)) {
2278 return false;
2280 return true;
2283 bool QuicFramer::AppendGoAwayFrame(const QuicGoAwayFrame& frame,
2284 QuicDataWriter* writer) {
2285 uint32 error_code = static_cast<uint32>(frame.error_code);
2286 if (!writer->WriteUInt32(error_code)) {
2287 return false;
2289 uint32 stream_id = static_cast<uint32>(frame.last_good_stream_id);
2290 if (!writer->WriteUInt32(stream_id)) {
2291 return false;
2293 if (!writer->WriteStringPiece16(frame.reason_phrase)) {
2294 return false;
2296 return true;
2299 bool QuicFramer::AppendWindowUpdateFrame(const QuicWindowUpdateFrame& frame,
2300 QuicDataWriter* writer) {
2301 uint32 stream_id = static_cast<uint32>(frame.stream_id);
2302 if (!writer->WriteUInt32(stream_id)) {
2303 return false;
2305 if (!writer->WriteUInt64(frame.byte_offset)) {
2306 return false;
2308 return true;
2311 bool QuicFramer::AppendBlockedFrame(const QuicBlockedFrame& frame,
2312 QuicDataWriter* writer) {
2313 uint32 stream_id = static_cast<uint32>(frame.stream_id);
2314 if (!writer->WriteUInt32(stream_id)) {
2315 return false;
2317 return true;
2320 bool QuicFramer::RaiseError(QuicErrorCode error) {
2321 DVLOG(1) << "Error detail: " << detailed_error_;
2322 set_error(error);
2323 visitor_->OnError(this);
2324 reader_.reset(nullptr);
2325 return false;
2328 } // namespace net