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 |