| OLD | NEW |
| 1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 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 // Manages the packet entropy calculation for both sent and received packets | 5 // Manages the packet entropy calculation for both sent and received packets |
| 6 // for a connection. | 6 // for a connection. |
| 7 | 7 |
| 8 #ifndef NET_QUIC_QUIC_SENT_ENTROPY_MANAGER_H_ | 8 #ifndef NET_QUIC_QUIC_SENT_ENTROPY_MANAGER_H_ |
| 9 #define NET_QUIC_QUIC_SENT_ENTROPY_MANAGER_H_ | 9 #define NET_QUIC_QUIC_SENT_ENTROPY_MANAGER_H_ |
| 10 | 10 |
| 11 #include <deque> |
| 12 |
| 11 #include "net/base/linked_hash_map.h" | 13 #include "net/base/linked_hash_map.h" |
| 12 #include "net/quic/quic_framer.h" | 14 #include "net/quic/quic_framer.h" |
| 13 #include "net/quic/quic_protocol.h" | 15 #include "net/quic/quic_protocol.h" |
| 14 | 16 |
| 15 namespace net { | 17 namespace net { |
| 16 | 18 |
| 17 namespace test { | 19 namespace test { |
| 18 class QuicConnectionPeer; | 20 class QuicConnectionPeer; |
| 19 } // namespace test | 21 } // namespace test |
| 20 | 22 |
| 21 // Records all sent packets by a connection to track the cumulative entropy of | 23 // Records all sent packets by a connection to track the cumulative entropy of |
| 22 // sent packets. It is used by the connection to validate an ack | 24 // sent packets. It is used by the connection to validate an ack |
| 23 // frame sent by the peer as a preventive measure against the optimistic ack | 25 // frame sent by the peer as a preventive measure against the optimistic ack |
| 24 // attack. | 26 // attack. |
| 25 class NET_EXPORT_PRIVATE QuicSentEntropyManager { | 27 class NET_EXPORT_PRIVATE QuicSentEntropyManager { |
| 26 public: | 28 public: |
| 27 QuicSentEntropyManager(); | 29 QuicSentEntropyManager(); |
| 28 virtual ~QuicSentEntropyManager(); | 30 virtual ~QuicSentEntropyManager(); |
| 29 | 31 |
| 30 // Record |entropy_hash| for sent packet corresponding to |sequence_number|. | 32 // Record |entropy_hash| for sent packet corresponding to |sequence_number|. |
| 31 void RecordPacketEntropyHash(QuicPacketSequenceNumber sequence_number, | 33 void RecordPacketEntropyHash(QuicPacketSequenceNumber sequence_number, |
| 32 QuicPacketEntropyHash entropy_hash); | 34 QuicPacketEntropyHash entropy_hash); |
| 33 | 35 |
| 34 QuicPacketEntropyHash EntropyHash( | 36 // Retrieves the cumulative entropy up to |sequence_number|. |
| 35 QuicPacketSequenceNumber sequence_number) const; | 37 // Must always be called with a monotonically increasing |sequence_number|. |
| 38 QuicPacketEntropyHash GetCumulativeEntropy( |
| 39 QuicPacketSequenceNumber sequence_number); |
| 36 | 40 |
| 37 // Returns true if |entropy_hash| matches the expected sent entropy hash | 41 // Returns true if |entropy_hash| matches the expected sent entropy hash |
| 38 // up to |sequence_number| removing sequence numbers from |missing_packets|. | 42 // up to |largest_observed| removing sequence numbers from |missing_packets|. |
| 39 bool IsValidEntropy(QuicPacketSequenceNumber sequence_number, | 43 // Must always be called with a monotonically increasing |largest_observed|. |
| 44 bool IsValidEntropy(QuicPacketSequenceNumber largest_observed, |
| 40 const SequenceNumberSet& missing_packets, | 45 const SequenceNumberSet& missing_packets, |
| 41 QuicPacketEntropyHash entropy_hash) const; | 46 QuicPacketEntropyHash entropy_hash); |
| 42 | 47 |
| 43 // Removes not required entries from |packets_entropy_| before | 48 // Removes unnecessary entries before |sequence_number|. |
| 44 // |sequence_number|. | |
| 45 void ClearEntropyBefore(QuicPacketSequenceNumber sequence_number); | 49 void ClearEntropyBefore(QuicPacketSequenceNumber sequence_number); |
| 46 | 50 |
| 47 private: | 51 private: |
| 48 friend class test::QuicConnectionPeer; | 52 friend class test::QuicConnectionPeer; |
| 49 | 53 |
| 50 typedef linked_hash_map<QuicPacketSequenceNumber, | 54 typedef std::deque<QuicPacketEntropyHash> SentEntropyMap; |
| 51 std::pair<QuicPacketEntropyHash, | |
| 52 QuicPacketEntropyHash> > SentEntropyMap; | |
| 53 | 55 |
| 54 // Linked hash map from sequence numbers to the sent entropy hash up to the | 56 struct CumulativeEntropy { |
| 55 // sequence number in the key. | 57 CumulativeEntropy() : sequence_number(0), entropy(0) {} |
| 58 |
| 59 QuicPacketSequenceNumber sequence_number; |
| 60 QuicPacketEntropyHash entropy; |
| 61 }; |
| 62 |
| 63 // Convenience methods to get the largest and smallest packets with entropies. |
| 64 QuicPacketSequenceNumber GetLargestPacketWithEntropy() const; |
| 65 QuicPacketSequenceNumber GetSmallestPacketWithEntropy() const; |
| 66 // Convenience method to get the entropy hash for |sequence_number|. |
| 67 QuicPacketEntropyHash GetPacketEntropy( |
| 68 QuicPacketSequenceNumber sequence_number) const; |
| 69 |
| 70 // Update the cumulative entropy to |sequence_number|. |
| 71 void UpdateCumulativeEntropy(QuicPacketSequenceNumber sequence_number, |
| 72 CumulativeEntropy* cumulative) const; |
| 73 |
| 74 // Maps sequence numbers to the sent entropy hash for the sequence number. |
| 56 SentEntropyMap packets_entropy_; | 75 SentEntropyMap packets_entropy_; |
| 76 QuicPacketSequenceNumber map_offset_; |
| 57 | 77 |
| 58 // Cumulative hash of entropy of all sent packets. | 78 // Cache the cumulative entropy for IsValidEntropy. |
| 59 QuicPacketEntropyHash packets_entropy_hash_; | 79 CumulativeEntropy last_valid_entropy_; |
| 80 |
| 81 // Cache the cumulative entropy for the sequence number used by EntropyHash. |
| 82 CumulativeEntropy last_cumulative_entropy_; |
| 60 | 83 |
| 61 DISALLOW_COPY_AND_ASSIGN(QuicSentEntropyManager); | 84 DISALLOW_COPY_AND_ASSIGN(QuicSentEntropyManager); |
| 62 }; | 85 }; |
| 63 | 86 |
| 64 } // namespace net | 87 } // namespace net |
| 65 | 88 |
| 66 #endif // NET_QUIC_QUIC_SENT_ENTROPY_MANAGER_H_ | 89 #endif // NET_QUIC_QUIC_SENT_ENTROPY_MANAGER_H_ |
| OLD | NEW |