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 // TCP cubic send side congestion algorithm, emulates the behavior of
8 #ifndef NET_QUIC_CONGESTION_CONTROL_TCP_CUBIC_SENDER_H_
9 #define NET_QUIC_CONGESTION_CONTROL_TCP_CUBIC_SENDER_H_
11 #include "base/basictypes.h"
12 #include "base/compiler_specific.h"
13 #include "net/base/net_export.h"
14 #include "net/quic/congestion_control/cubic.h"
15 #include "net/quic/congestion_control/hybrid_slow_start.h"
16 #include "net/quic/congestion_control/send_algorithm_interface.h"
17 #include "net/quic/quic_bandwidth.h"
18 #include "net/quic/quic_connection_stats.h"
19 #include "net/quic/quic_protocol.h"
20 #include "net/quic/quic_time.h"
27 class TcpCubicSenderPeer;
30 class NET_EXPORT_PRIVATE TcpCubicSender : public SendAlgorithmInterface {
32 // Reno option and max_tcp_congestion_window are provided for testing.
33 TcpCubicSender(const QuicClock* clock,
34 const RttStats* rtt_stats,
36 QuicTcpCongestionWindow max_tcp_congestion_window,
37 QuicConnectionStats* stats);
38 virtual ~TcpCubicSender();
40 // Start implementation of SendAlgorithmInterface.
41 virtual void SetFromConfig(const QuicConfig& config, bool is_server) OVERRIDE;
42 virtual void OnIncomingQuicCongestionFeedbackFrame(
43 const QuicCongestionFeedbackFrame& feedback,
44 QuicTime feedback_receive_time) OVERRIDE;
45 virtual void OnCongestionEvent(bool rtt_updated,
46 QuicByteCount bytes_in_flight,
47 const CongestionMap& acked_packets,
48 const CongestionMap& lost_packets) OVERRIDE;
49 virtual bool OnPacketSent(QuicTime sent_time,
50 QuicByteCount bytes_in_flight,
51 QuicPacketSequenceNumber sequence_number,
53 HasRetransmittableData is_retransmittable) OVERRIDE;
54 virtual void OnRetransmissionTimeout(bool packets_retransmitted) OVERRIDE;
55 virtual void RevertRetransmissionTimeout() OVERRIDE;
56 virtual QuicTime::Delta TimeUntilSend(
58 QuicByteCount bytes_in_flight,
59 HasRetransmittableData has_retransmittable_data) const OVERRIDE;
60 virtual QuicBandwidth BandwidthEstimate() const OVERRIDE;
61 virtual bool HasReliableBandwidthEstimate() const OVERRIDE;
62 virtual QuicTime::Delta RetransmissionDelay() const OVERRIDE;
63 virtual QuicByteCount GetCongestionWindow() const OVERRIDE;
64 virtual bool InSlowStart() const OVERRIDE;
65 virtual QuicByteCount GetSlowStartThreshold() const OVERRIDE;
66 virtual CongestionControlType GetCongestionControlType() const OVERRIDE;
67 // End implementation of SendAlgorithmInterface.
70 friend class test::TcpCubicSenderPeer;
72 // TODO(ianswett): Remove these and migrate to OnCongestionEvent.
73 void OnPacketAcked(QuicPacketSequenceNumber acked_sequence_number,
74 QuicByteCount acked_bytes,
75 QuicByteCount bytes_in_flight);
76 void OnPacketLost(QuicPacketSequenceNumber largest_loss,
77 QuicByteCount bytes_in_flight);
79 QuicByteCount SendWindow() const;
80 void MaybeIncreaseCwnd(QuicPacketSequenceNumber acked_sequence_number,
81 QuicByteCount bytes_in_flight);
82 bool IsCwndLimited(QuicByteCount bytes_in_flight) const;
83 bool InRecovery() const;
84 // Methods for isolating PRR from the rest of TCP Cubic.
85 void PrrOnPacketLost(QuicByteCount bytes_in_flight);
86 void PrrOnPacketAcked(QuicByteCount acked_bytes);
87 QuicTime::Delta PrrTimeUntilSend(QuicByteCount bytes_in_flight) const;
90 HybridSlowStart hybrid_slow_start_;
92 const RttStats* rtt_stats_;
93 QuicConnectionStats* stats_;
95 // Reno provided for testing.
98 // ACK counter for the Reno implementation.
99 int64 congestion_window_count_;
101 // Receiver side advertised window.
102 QuicByteCount receive_window_;
104 // Bytes sent and acked since the last loss event. Used for PRR.
105 QuicByteCount prr_out_;
106 QuicByteCount prr_delivered_;
107 size_t ack_count_since_loss_;
109 // The congestion window before the last loss event.
110 QuicByteCount bytes_in_flight_before_loss_;
112 // Track the largest packet that has been sent.
113 QuicPacketSequenceNumber largest_sent_sequence_number_;
115 // Track the largest packet that has been acked.
116 QuicPacketSequenceNumber largest_acked_sequence_number_;
118 // Track the largest sequence number outstanding when a CWND cutback occurs.
119 QuicPacketSequenceNumber largest_sent_at_last_cutback_;
121 // Congestion window in packets.
122 QuicTcpCongestionWindow congestion_window_;
124 // Congestion window before the last loss event or RTO.
125 QuicByteCount previous_congestion_window_;
127 // Slow start congestion window in packets, aka ssthresh.
128 QuicTcpCongestionWindow slowstart_threshold_;
130 // Slow start threshold before the last loss event or RTO.
131 QuicTcpCongestionWindow previous_slowstart_threshold_;
133 // Whether the last loss event caused us to exit slowstart.
134 // Used for stats collection of slowstart_packets_lost
135 bool last_cutback_exited_slowstart_;
137 // Maximum number of outstanding packets for tcp.
138 QuicTcpCongestionWindow max_tcp_congestion_window_;
140 DISALLOW_COPY_AND_ASSIGN(TcpCubicSender);
145 #endif // NET_QUIC_CONGESTION_CONTROL_TCP_CUBIC_SENDER_H_