| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2016 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "net/quic/core/quic_protocol.h" | 5 #include "net/quic/core/quic_frames.h" |
| 6 | |
| 7 #include "base/memory/ptr_util.h" | |
| 8 #include "base/strings/string_number_conversions.h" | |
| 9 #include "net/quic/core/quic_flags.h" | |
| 10 #include "net/quic/core/quic_utils.h" | |
| 11 #include "net/quic/core/quic_versions.h" | |
| 12 | 6 |
| 13 using base::StringPiece; | 7 using base::StringPiece; |
| 14 using std::map; | |
| 15 using std::numeric_limits; | |
| 16 using std::ostream; | 8 using std::ostream; |
| 17 using std::string; | 9 using std::string; |
| 18 | 10 |
| 19 namespace net { | 11 namespace net { |
| 20 | 12 |
| 21 size_t GetPacketHeaderSize(QuicVersion version, | |
| 22 const QuicPacketHeader& header) { | |
| 23 return GetPacketHeaderSize(version, header.public_header.connection_id_length, | |
| 24 header.public_header.version_flag, | |
| 25 header.public_header.multipath_flag, | |
| 26 header.public_header.nonce != nullptr, | |
| 27 header.public_header.packet_number_length); | |
| 28 } | |
| 29 | |
| 30 size_t GetPacketHeaderSize(QuicVersion version, | |
| 31 QuicConnectionIdLength connection_id_length, | |
| 32 bool include_version, | |
| 33 bool include_path_id, | |
| 34 bool include_diversification_nonce, | |
| 35 QuicPacketNumberLength packet_number_length) { | |
| 36 return kPublicFlagsSize + connection_id_length + | |
| 37 (include_version ? kQuicVersionSize : 0) + | |
| 38 (include_path_id ? kQuicPathIdSize : 0) + packet_number_length + | |
| 39 (include_diversification_nonce ? kDiversificationNonceSize : 0); | |
| 40 } | |
| 41 | |
| 42 size_t GetStartOfEncryptedData(QuicVersion version, | |
| 43 const QuicPacketHeader& header) { | |
| 44 return GetPacketHeaderSize(version, header); | |
| 45 } | |
| 46 | |
| 47 size_t GetStartOfEncryptedData(QuicVersion version, | |
| 48 QuicConnectionIdLength connection_id_length, | |
| 49 bool include_version, | |
| 50 bool include_path_id, | |
| 51 bool include_diversification_nonce, | |
| 52 QuicPacketNumberLength packet_number_length) { | |
| 53 // Encryption starts before private flags. | |
| 54 return GetPacketHeaderSize(version, connection_id_length, include_version, | |
| 55 include_path_id, include_diversification_nonce, | |
| 56 packet_number_length); | |
| 57 } | |
| 58 | |
| 59 QuicPacketPublicHeader::QuicPacketPublicHeader() | |
| 60 : connection_id(0), | |
| 61 connection_id_length(PACKET_8BYTE_CONNECTION_ID), | |
| 62 multipath_flag(false), | |
| 63 reset_flag(false), | |
| 64 version_flag(false), | |
| 65 packet_number_length(PACKET_6BYTE_PACKET_NUMBER), | |
| 66 nonce(nullptr) {} | |
| 67 | |
| 68 QuicPacketPublicHeader::QuicPacketPublicHeader( | |
| 69 const QuicPacketPublicHeader& other) = default; | |
| 70 | |
| 71 QuicPacketPublicHeader::~QuicPacketPublicHeader() {} | |
| 72 | |
| 73 QuicPacketHeader::QuicPacketHeader() | |
| 74 : packet_number(0), path_id(kDefaultPathId) {} | |
| 75 | |
| 76 QuicPacketHeader::QuicPacketHeader(const QuicPacketPublicHeader& header) | |
| 77 : public_header(header), packet_number(0), path_id(kDefaultPathId) {} | |
| 78 | |
| 79 QuicPacketHeader::QuicPacketHeader(const QuicPacketHeader& other) = default; | |
| 80 | |
| 81 QuicPublicResetPacket::QuicPublicResetPacket() | |
| 82 : nonce_proof(0), rejected_packet_number(0) {} | |
| 83 | |
| 84 QuicPublicResetPacket::QuicPublicResetPacket( | |
| 85 const QuicPacketPublicHeader& header) | |
| 86 : public_header(header), nonce_proof(0), rejected_packet_number(0) {} | |
| 87 | |
| 88 void StreamBufferDeleter::operator()(char* buffer) const { | 13 void StreamBufferDeleter::operator()(char* buffer) const { |
| 89 if (allocator_ != nullptr && buffer != nullptr) { | 14 if (allocator_ != nullptr && buffer != nullptr) { |
| 90 allocator_->Delete(buffer); | 15 allocator_->Delete(buffer); |
| 91 } | 16 } |
| 92 } | 17 } |
| 93 | 18 |
| 94 UniqueStreamBuffer NewStreamBuffer(QuicBufferAllocator* allocator, | 19 UniqueStreamBuffer NewStreamBuffer(QuicBufferAllocator* allocator, |
| 95 size_t size) { | 20 size_t size) { |
| 96 return UniqueStreamBuffer(allocator->New(size), | 21 return UniqueStreamBuffer(allocator->New(size), |
| 97 StreamBufferDeleter(allocator)); | 22 StreamBufferDeleter(allocator)); |
| (...skipping 41 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 139 offset(offset), | 64 offset(offset), |
| 140 buffer(std::move(buffer)) { | 65 buffer(std::move(buffer)) { |
| 141 if (this->buffer != nullptr) { | 66 if (this->buffer != nullptr) { |
| 142 DCHECK(data_buffer == nullptr); | 67 DCHECK(data_buffer == nullptr); |
| 143 this->data_buffer = this->buffer.get(); | 68 this->data_buffer = this->buffer.get(); |
| 144 } | 69 } |
| 145 } | 70 } |
| 146 | 71 |
| 147 QuicStreamFrame::~QuicStreamFrame() {} | 72 QuicStreamFrame::~QuicStreamFrame() {} |
| 148 | 73 |
| 149 ostream& operator<<(ostream& os, const QuicPacketHeader& header) { | |
| 150 os << "{ connection_id: " << header.public_header.connection_id | |
| 151 << ", connection_id_length: " << header.public_header.connection_id_length | |
| 152 << ", packet_number_length: " << header.public_header.packet_number_length | |
| 153 << ", multipath_flag: " << header.public_header.multipath_flag | |
| 154 << ", reset_flag: " << header.public_header.reset_flag | |
| 155 << ", version_flag: " << header.public_header.version_flag; | |
| 156 if (header.public_header.version_flag) { | |
| 157 os << ", version:"; | |
| 158 for (size_t i = 0; i < header.public_header.versions.size(); ++i) { | |
| 159 os << " "; | |
| 160 os << QuicVersionToString(header.public_header.versions[i]); | |
| 161 } | |
| 162 } | |
| 163 if (header.public_header.nonce != nullptr) { | |
| 164 os << ", diversification_nonce: " | |
| 165 << QuicUtils::HexEncode(StringPiece(header.public_header.nonce->data(), | |
| 166 header.public_header.nonce->size())); | |
| 167 } | |
| 168 os << ", path_id: " << static_cast<int>(header.path_id) | |
| 169 << ", packet_number: " << header.packet_number << " }\n"; | |
| 170 return os; | |
| 171 } | |
| 172 | |
| 173 bool IsAwaitingPacket(const QuicAckFrame& ack_frame, | 74 bool IsAwaitingPacket(const QuicAckFrame& ack_frame, |
| 174 QuicPacketNumber packet_number, | 75 QuicPacketNumber packet_number, |
| 175 QuicPacketNumber peer_least_packet_awaiting_ack) { | 76 QuicPacketNumber peer_least_packet_awaiting_ack) { |
| 176 return packet_number >= peer_least_packet_awaiting_ack && | 77 return packet_number >= peer_least_packet_awaiting_ack && |
| 177 !ack_frame.packets.Contains(packet_number); | 78 !ack_frame.packets.Contains(packet_number); |
| 178 } | 79 } |
| 179 | 80 |
| 180 QuicStopWaitingFrame::QuicStopWaitingFrame() | 81 QuicStopWaitingFrame::QuicStopWaitingFrame() |
| 181 : path_id(kDefaultPathId), least_unacked(0) {} | 82 : path_id(kDefaultPathId), least_unacked(0) {} |
| 182 | 83 |
| (...skipping 112 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 295 } | 196 } |
| 296 } | 197 } |
| 297 | 198 |
| 298 ostream& operator<<(ostream& os, const QuicStopWaitingFrame& sent_info) { | 199 ostream& operator<<(ostream& os, const QuicStopWaitingFrame& sent_info) { |
| 299 os << "{ least_unacked: " << sent_info.least_unacked << " }\n"; | 200 os << "{ least_unacked: " << sent_info.least_unacked << " }\n"; |
| 300 return os; | 201 return os; |
| 301 } | 202 } |
| 302 | 203 |
| 303 PacketNumberQueue::PacketNumberQueue() = default; | 204 PacketNumberQueue::PacketNumberQueue() = default; |
| 304 PacketNumberQueue::PacketNumberQueue(const PacketNumberQueue& other) = default; | 205 PacketNumberQueue::PacketNumberQueue(const PacketNumberQueue& other) = default; |
| 305 // TODO(rtenneti): on windows RValue reference gives errors. | 206 PacketNumberQueue::PacketNumberQueue(PacketNumberQueue&& other) = default; |
| 306 // PacketNumberQueue::PacketNumberQueue(PacketNumberQueue&& other) = default; | |
| 307 PacketNumberQueue::~PacketNumberQueue() {} | 207 PacketNumberQueue::~PacketNumberQueue() {} |
| 308 | 208 |
| 309 PacketNumberQueue& PacketNumberQueue::operator=( | 209 PacketNumberQueue& PacketNumberQueue::operator=( |
| 310 const PacketNumberQueue& other) = default; | 210 const PacketNumberQueue& other) = default; |
| 311 // TODO(rtenneti): on windows RValue reference gives errors. | 211 PacketNumberQueue& PacketNumberQueue::operator=(PacketNumberQueue&& other) = |
| 312 // PacketNumberQueue& PacketNumberQueue::operator=(PacketNumberQueue&& other) = | 212 default; |
| 313 // default; | |
| 314 | 213 |
| 315 void PacketNumberQueue::Add(QuicPacketNumber packet_number) { | 214 void PacketNumberQueue::Add(QuicPacketNumber packet_number) { |
| 316 packet_number_intervals_.Add(packet_number, packet_number + 1); | 215 packet_number_intervals_.Add(packet_number, packet_number + 1); |
| 317 } | 216 } |
| 318 | 217 |
| 319 void PacketNumberQueue::Add(QuicPacketNumber lower, QuicPacketNumber higher) { | 218 void PacketNumberQueue::Add(QuicPacketNumber lower, QuicPacketNumber higher) { |
| 320 packet_number_intervals_.Add(lower, higher); | 219 packet_number_intervals_.Add(lower, higher); |
| 321 } | 220 } |
| 322 | 221 |
| 323 void PacketNumberQueue::Remove(QuicPacketNumber packet_number) { | 222 void PacketNumberQueue::Remove(QuicPacketNumber packet_number) { |
| (...skipping 223 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 547 QuicGoAwayFrame::QuicGoAwayFrame() | 446 QuicGoAwayFrame::QuicGoAwayFrame() |
| 548 : error_code(QUIC_NO_ERROR), last_good_stream_id(0) {} | 447 : error_code(QUIC_NO_ERROR), last_good_stream_id(0) {} |
| 549 | 448 |
| 550 QuicGoAwayFrame::QuicGoAwayFrame(QuicErrorCode error_code, | 449 QuicGoAwayFrame::QuicGoAwayFrame(QuicErrorCode error_code, |
| 551 QuicStreamId last_good_stream_id, | 450 QuicStreamId last_good_stream_id, |
| 552 const string& reason) | 451 const string& reason) |
| 553 : error_code(error_code), | 452 : error_code(error_code), |
| 554 last_good_stream_id(last_good_stream_id), | 453 last_good_stream_id(last_good_stream_id), |
| 555 reason_phrase(reason) {} | 454 reason_phrase(reason) {} |
| 556 | 455 |
| 557 QuicData::QuicData(const char* buffer, size_t length) | |
| 558 : buffer_(buffer), length_(length), owns_buffer_(false) {} | |
| 559 | |
| 560 QuicData::QuicData(const char* buffer, size_t length, bool owns_buffer) | |
| 561 : buffer_(buffer), length_(length), owns_buffer_(owns_buffer) {} | |
| 562 | |
| 563 QuicData::~QuicData() { | |
| 564 if (owns_buffer_) { | |
| 565 delete[] const_cast<char*>(buffer_); | |
| 566 } | |
| 567 } | |
| 568 | |
| 569 QuicWindowUpdateFrame::QuicWindowUpdateFrame(QuicStreamId stream_id, | 456 QuicWindowUpdateFrame::QuicWindowUpdateFrame(QuicStreamId stream_id, |
| 570 QuicStreamOffset byte_offset) | 457 QuicStreamOffset byte_offset) |
| 571 : stream_id(stream_id), byte_offset(byte_offset) {} | 458 : stream_id(stream_id), byte_offset(byte_offset) {} |
| 572 | 459 |
| 573 QuicBlockedFrame::QuicBlockedFrame(QuicStreamId stream_id) | 460 QuicBlockedFrame::QuicBlockedFrame(QuicStreamId stream_id) |
| 574 : stream_id(stream_id) {} | 461 : stream_id(stream_id) {} |
| 575 | 462 |
| 576 QuicPathCloseFrame::QuicPathCloseFrame(QuicPathId path_id) : path_id(path_id) {} | 463 QuicPathCloseFrame::QuicPathCloseFrame(QuicPathId path_id) : path_id(path_id) {} |
| 577 | 464 |
| 578 QuicPacket::QuicPacket(char* buffer, | |
| 579 size_t length, | |
| 580 bool owns_buffer, | |
| 581 QuicConnectionIdLength connection_id_length, | |
| 582 bool includes_version, | |
| 583 bool includes_path_id, | |
| 584 bool includes_diversification_nonce, | |
| 585 QuicPacketNumberLength packet_number_length) | |
| 586 : QuicData(buffer, length, owns_buffer), | |
| 587 buffer_(buffer), | |
| 588 connection_id_length_(connection_id_length), | |
| 589 includes_version_(includes_version), | |
| 590 includes_path_id_(includes_path_id), | |
| 591 includes_diversification_nonce_(includes_diversification_nonce), | |
| 592 packet_number_length_(packet_number_length) {} | |
| 593 | |
| 594 QuicEncryptedPacket::QuicEncryptedPacket(const char* buffer, size_t length) | |
| 595 : QuicData(buffer, length) {} | |
| 596 | |
| 597 QuicEncryptedPacket::QuicEncryptedPacket(const char* buffer, | |
| 598 size_t length, | |
| 599 bool owns_buffer) | |
| 600 : QuicData(buffer, length, owns_buffer) {} | |
| 601 | |
| 602 std::unique_ptr<QuicEncryptedPacket> QuicEncryptedPacket::Clone() const { | |
| 603 char* buffer = new char[this->length()]; | |
| 604 memcpy(buffer, this->data(), this->length()); | |
| 605 return base::MakeUnique<QuicEncryptedPacket>(buffer, this->length(), true); | |
| 606 } | |
| 607 | |
| 608 ostream& operator<<(ostream& os, const QuicEncryptedPacket& s) { | |
| 609 os << s.length() << "-byte data"; | |
| 610 return os; | |
| 611 } | |
| 612 | |
| 613 QuicReceivedPacket::QuicReceivedPacket(const char* buffer, | |
| 614 size_t length, | |
| 615 QuicTime receipt_time) | |
| 616 : QuicEncryptedPacket(buffer, length), | |
| 617 receipt_time_(receipt_time), | |
| 618 ttl_(0) {} | |
| 619 | |
| 620 QuicReceivedPacket::QuicReceivedPacket(const char* buffer, | |
| 621 size_t length, | |
| 622 QuicTime receipt_time, | |
| 623 bool owns_buffer) | |
| 624 : QuicEncryptedPacket(buffer, length, owns_buffer), | |
| 625 receipt_time_(receipt_time), | |
| 626 ttl_(0) {} | |
| 627 | |
| 628 QuicReceivedPacket::QuicReceivedPacket(const char* buffer, | |
| 629 size_t length, | |
| 630 QuicTime receipt_time, | |
| 631 bool owns_buffer, | |
| 632 int ttl, | |
| 633 bool ttl_valid) | |
| 634 : QuicEncryptedPacket(buffer, length, owns_buffer), | |
| 635 receipt_time_(receipt_time), | |
| 636 ttl_(ttl_valid ? ttl : -1) {} | |
| 637 | |
| 638 std::unique_ptr<QuicReceivedPacket> QuicReceivedPacket::Clone() const { | |
| 639 char* buffer = new char[this->length()]; | |
| 640 memcpy(buffer, this->data(), this->length()); | |
| 641 return base::MakeUnique<QuicReceivedPacket>( | |
| 642 buffer, this->length(), receipt_time(), true, ttl(), ttl() >= 0); | |
| 643 } | |
| 644 | |
| 645 ostream& operator<<(ostream& os, const QuicReceivedPacket& s) { | |
| 646 os << s.length() << "-byte data"; | |
| 647 return os; | |
| 648 } | |
| 649 | |
| 650 StringPiece QuicPacket::AssociatedData(QuicVersion version) const { | |
| 651 return StringPiece( | |
| 652 data(), GetStartOfEncryptedData(version, connection_id_length_, | |
| 653 includes_version_, includes_path_id_, | |
| 654 includes_diversification_nonce_, | |
| 655 packet_number_length_)); | |
| 656 } | |
| 657 | |
| 658 StringPiece QuicPacket::Plaintext(QuicVersion version) const { | |
| 659 const size_t start_of_encrypted_data = GetStartOfEncryptedData( | |
| 660 version, connection_id_length_, includes_version_, includes_path_id_, | |
| 661 includes_diversification_nonce_, packet_number_length_); | |
| 662 return StringPiece(data() + start_of_encrypted_data, | |
| 663 length() - start_of_encrypted_data); | |
| 664 } | |
| 665 | |
| 666 AckListenerWrapper::AckListenerWrapper(QuicAckListenerInterface* listener, | |
| 667 QuicPacketLength data_length) | |
| 668 : ack_listener(listener), length(data_length) { | |
| 669 DCHECK(listener != nullptr); | |
| 670 } | |
| 671 | |
| 672 AckListenerWrapper::AckListenerWrapper(const AckListenerWrapper& other) = | |
| 673 default; | |
| 674 | |
| 675 AckListenerWrapper::~AckListenerWrapper() {} | |
| 676 | |
| 677 SerializedPacket::SerializedPacket(QuicPathId path_id, | |
| 678 QuicPacketNumber packet_number, | |
| 679 QuicPacketNumberLength packet_number_length, | |
| 680 const char* encrypted_buffer, | |
| 681 QuicPacketLength encrypted_length, | |
| 682 bool has_ack, | |
| 683 bool has_stop_waiting) | |
| 684 : encrypted_buffer(encrypted_buffer), | |
| 685 encrypted_length(encrypted_length), | |
| 686 has_crypto_handshake(NOT_HANDSHAKE), | |
| 687 num_padding_bytes(0), | |
| 688 path_id(path_id), | |
| 689 packet_number(packet_number), | |
| 690 packet_number_length(packet_number_length), | |
| 691 encryption_level(ENCRYPTION_NONE), | |
| 692 has_ack(has_ack), | |
| 693 has_stop_waiting(has_stop_waiting), | |
| 694 transmission_type(NOT_RETRANSMISSION), | |
| 695 original_path_id(kInvalidPathId), | |
| 696 original_packet_number(0) {} | |
| 697 | |
| 698 SerializedPacket::SerializedPacket(const SerializedPacket& other) = default; | |
| 699 | |
| 700 SerializedPacket::~SerializedPacket() {} | |
| 701 | |
| 702 TransmissionInfo::TransmissionInfo() | |
| 703 : encryption_level(ENCRYPTION_NONE), | |
| 704 packet_number_length(PACKET_1BYTE_PACKET_NUMBER), | |
| 705 bytes_sent(0), | |
| 706 sent_time(QuicTime::Zero()), | |
| 707 transmission_type(NOT_RETRANSMISSION), | |
| 708 in_flight(false), | |
| 709 is_unackable(false), | |
| 710 has_crypto_handshake(false), | |
| 711 num_padding_bytes(0), | |
| 712 retransmission(0) {} | |
| 713 | |
| 714 void ClearSerializedPacket(SerializedPacket* serialized_packet) { | |
| 715 if (!serialized_packet->retransmittable_frames.empty()) { | |
| 716 DeleteFrames(&serialized_packet->retransmittable_frames); | |
| 717 } | |
| 718 serialized_packet->encrypted_buffer = nullptr; | |
| 719 serialized_packet->encrypted_length = 0; | |
| 720 } | |
| 721 | |
| 722 char* CopyBuffer(const SerializedPacket& packet) { | |
| 723 char* dst_buffer = new char[packet.encrypted_length]; | |
| 724 memcpy(dst_buffer, packet.encrypted_buffer, packet.encrypted_length); | |
| 725 return dst_buffer; | |
| 726 } | |
| 727 | |
| 728 TransmissionInfo::TransmissionInfo(EncryptionLevel level, | |
| 729 QuicPacketNumberLength packet_number_length, | |
| 730 TransmissionType transmission_type, | |
| 731 QuicTime sent_time, | |
| 732 QuicPacketLength bytes_sent, | |
| 733 bool has_crypto_handshake, | |
| 734 int num_padding_bytes) | |
| 735 : encryption_level(level), | |
| 736 packet_number_length(packet_number_length), | |
| 737 bytes_sent(bytes_sent), | |
| 738 sent_time(sent_time), | |
| 739 transmission_type(transmission_type), | |
| 740 in_flight(false), | |
| 741 is_unackable(false), | |
| 742 has_crypto_handshake(has_crypto_handshake), | |
| 743 num_padding_bytes(num_padding_bytes), | |
| 744 retransmission(0) {} | |
| 745 | |
| 746 TransmissionInfo::TransmissionInfo(const TransmissionInfo& other) = default; | |
| 747 | |
| 748 TransmissionInfo::~TransmissionInfo() {} | |
| 749 | |
| 750 } // namespace net | 465 } // namespace net |
| OLD | NEW |