cregit-Linux how code gets into the kernel

Release 4.14 net/ipv4/tcp_nv.c

Directory: net/ipv4
/*
 * TCP NV: TCP with Congestion Avoidance
 *
 * TCP-NV is a successor of TCP-Vegas that has been developed to
 * deal with the issues that occur in modern networks.
 * Like TCP-Vegas, TCP-NV supports true congestion avoidance,
 * the ability to detect congestion before packet losses occur.
 * When congestion (queue buildup) starts to occur, TCP-NV
 * predicts what the cwnd size should be for the current
 * throughput and it reduces the cwnd proportionally to
 * the difference between the current cwnd and the predicted cwnd.
 *
 * NV is only recommeneded for traffic within a data center, and when
 * all the flows are NV (at least those within the data center). This
 * is due to the inherent unfairness between flows using losses to
 * detect congestion (congestion control) and those that use queue
 * buildup to detect congestion (congestion avoidance).
 *
 * Note: High NIC coalescence values may lower the performance of NV
 * due to the increased noise in RTT values. In particular, we have
 * seen issues with rx-frames values greater than 8.
 *
 * TODO:
 * 1) Add mechanism to deal with reverse congestion.
 */

#include <linux/mm.h>
#include <linux/module.h>
#include <linux/math64.h>
#include <net/tcp.h>
#include <linux/inet_diag.h>

/* TCP NV parameters
 *
 * nv_pad               Max number of queued packets allowed in network
 * nv_pad_buffer        Do not grow cwnd if this closed to nv_pad
 * nv_reset_period      How often (in) seconds)to reset min_rtt
 * nv_min_cwnd          Don't decrease cwnd below this if there are no losses
 * nv_cong_dec_mult     Decrease cwnd by X% (30%) of congestion when detected
 * nv_ssthresh_factor   On congestion set ssthresh to this * <desired cwnd> / 8
 * nv_rtt_factor        RTT averaging factor
 * nv_loss_dec_factor   Decrease cwnd by this (50%) when losses occur
 * nv_dec_eval_min_calls        Wait this many RTT measurements before dec cwnd
 * nv_inc_eval_min_calls        Wait this many RTT measurements before inc cwnd
 * nv_ssthresh_eval_min_calls   Wait this many RTT measurements before stopping
 *                              slow-start due to congestion
 * nv_stop_rtt_cnt      Only grow cwnd for this many RTTs after non-congestion
 * nv_rtt_min_cnt       Wait these many RTTs before making congesion decision
 * nv_cwnd_growth_rate_neg
 * nv_cwnd_growth_rate_pos
 *      How quickly to double growth rate (not rate) of cwnd when not
 *      congested. One value (nv_cwnd_growth_rate_neg) for when
 *      rate < 1 pkt/RTT (after losses). The other (nv_cwnd_growth_rate_pos)
 *      otherwise.
 */


static int nv_pad __read_mostly = 10;

static int nv_pad_buffer __read_mostly = 2;

static int nv_reset_period __read_mostly = 5; 
/* in seconds */

static int nv_min_cwnd __read_mostly = 2;

static int nv_cong_dec_mult __read_mostly = 30 * 128 / 100; 
/* = 30% */

static int nv_ssthresh_factor __read_mostly = 8; 
/* = 1 */

static int nv_rtt_factor __read_mostly = 128; 
/* = 1/2*old + 1/2*new */

static int nv_loss_dec_factor __read_mostly = 512; 
/* => 50% */

static int nv_cwnd_growth_rate_neg __read_mostly = 8;

static int nv_cwnd_growth_rate_pos __read_mostly; 
/* 0 => fixed like Reno */

static int nv_dec_eval_min_calls __read_mostly = 60;

static int nv_inc_eval_min_calls __read_mostly = 20;

static int nv_ssthresh_eval_min_calls __read_mostly = 30;

static int nv_stop_rtt_cnt __read_mostly = 10;

static int nv_rtt_min_cnt __read_mostly = 2;

module_param(nv_pad, int, 0644);
MODULE_PARM_DESC(nv_pad, "max queued packets allowed in network");
module_param(nv_reset_period, int, 0644);
MODULE_PARM_DESC(nv_reset_period, "nv_min_rtt reset period (secs)");
module_param(nv_min_cwnd, int, 0644);
MODULE_PARM_DESC(nv_min_cwnd, "NV will not decrease cwnd below this value"
		 " without losses");

/* TCP NV Parameters */

struct tcpnv {
	
unsigned long nv_min_rtt_reset_jiffies;  /* when to switch to
                                                  * nv_min_rtt_new */
	
s8  cwnd_growth_factor;	/* Current cwnd growth factor,
                                 * < 0 => less than 1 packet/RTT */
	
u8  available8;
	
u16 available16;
	
u8  nv_allow_cwnd_growth:1, /* whether cwnd can grow */
		
nv_reset:1,	    /* whether to reset values */
		
nv_catchup:1;	    /* whether we are growing because
                                     * of temporary cwnd decrease */
	
u8  nv_eval_call_cnt;	/* call count since last eval */
	
u8  nv_min_cwnd;	/* nv won't make a ca decision if cwnd is
                                 * smaller than this. It may grow to handle
                                 * TSO, LRO and interrupt coalescence because
                                 * with these a small cwnd cannot saturate
                                 * the link. Note that this is different from
                                 * the file local nv_min_cwnd */
	
u8  nv_rtt_cnt;		/* RTTs without making ca decision */;
	
u32 nv_last_rtt;	/* last rtt */
	
u32 nv_min_rtt;		/* active min rtt. Used to determine slope */
	
u32 nv_min_rtt_new;	/* min rtt for future use */
	
u32 nv_rtt_max_rate;	/* max rate seen during current RTT */
	
u32 nv_rtt_start_seq;	/* current RTT ends when packet arrives
                                 * acking beyond nv_rtt_start_seq */
	
u32 nv_last_snd_una;	/* Previous value of tp->snd_una. It is
                                 * used to determine bytes acked since last
                                 * call to bictcp_acked */
	
u32 nv_no_cong_cnt;	/* Consecutive no congestion decisions */
};


#define NV_INIT_RTT	  U32_MAX

#define NV_MIN_CWND	  4

#define NV_MIN_CWND_GROW  2

#define NV_TSO_CWND_BOUND 80


static inline void tcpnv_reset(struct tcpnv *ca, struct sock *sk) { struct tcp_sock *tp = tcp_sk(sk); ca->nv_reset = 0; ca->nv_no_cong_cnt = 0; ca->nv_rtt_cnt = 0; ca->nv_last_rtt = 0; ca->nv_rtt_max_rate = 0; ca->nv_rtt_start_seq = tp->snd_una; ca->nv_eval_call_cnt = 0; ca->nv_last_snd_una = tp->snd_una; }

Contributors

PersonTokensPropCommitsCommitProp
Lawrence Brakmo79100.00%1100.00%
Total79100.00%1100.00%


static void tcpnv_init(struct sock *sk) { struct tcpnv *ca = inet_csk_ca(sk); tcpnv_reset(ca, sk); ca->nv_allow_cwnd_growth = 1; ca->nv_min_rtt_reset_jiffies = jiffies + 2 * HZ; ca->nv_min_rtt = NV_INIT_RTT; ca->nv_min_rtt_new = NV_INIT_RTT; ca->nv_min_cwnd = NV_MIN_CWND; ca->nv_catchup = 0; ca->cwnd_growth_factor = 0; }

Contributors

PersonTokensPropCommitsCommitProp
Lawrence Brakmo74100.00%1100.00%
Total74100.00%1100.00%


static void tcpnv_cong_avoid(struct sock *sk, u32 ack, u32 acked) { struct tcp_sock *tp = tcp_sk(sk); struct tcpnv *ca = inet_csk_ca(sk); u32 cnt; if (!tcp_is_cwnd_limited(sk)) return; /* Only grow cwnd if NV has not detected congestion */ if (!ca->nv_allow_cwnd_growth) return; if (tcp_in_slow_start(tp)) { acked = tcp_slow_start(tp, acked); if (!acked) return; } if (ca->cwnd_growth_factor < 0) { cnt = tp->snd_cwnd << -ca->cwnd_growth_factor; tcp_cong_avoid_ai(tp, cnt, acked); } else { cnt = max(4U, tp->snd_cwnd >> ca->cwnd_growth_factor); tcp_cong_avoid_ai(tp, cnt, acked); } }

Contributors

PersonTokensPropCommitsCommitProp
Lawrence Brakmo139100.00%1100.00%
Total139100.00%1100.00%


static u32 tcpnv_recalc_ssthresh(struct sock *sk) { const struct tcp_sock *tp = tcp_sk(sk); return max((tp->snd_cwnd * nv_loss_dec_factor) >> 10, 2U); }

Contributors

PersonTokensPropCommitsCommitProp
Lawrence Brakmo38100.00%1100.00%
Total38100.00%1100.00%


static void tcpnv_state(struct sock *sk, u8 new_state) { struct tcpnv *ca = inet_csk_ca(sk); if (new_state == TCP_CA_Open && ca->nv_reset) { tcpnv_reset(ca, sk); } else if (new_state == TCP_CA_Loss || new_state == TCP_CA_CWR || new_state == TCP_CA_Recovery) { ca->nv_reset = 1; ca->nv_allow_cwnd_growth = 0; if (new_state == TCP_CA_Loss) { /* Reset cwnd growth factor to Reno value */ if (ca->cwnd_growth_factor > 0) ca->cwnd_growth_factor = 0; /* Decrease growth rate if allowed */ if (nv_cwnd_growth_rate_neg > 0 && ca->cwnd_growth_factor > -8) ca->cwnd_growth_factor--; } } }

Contributors

PersonTokensPropCommitsCommitProp
Lawrence Brakmo114100.00%1100.00%
Total114100.00%1100.00%

/* Do congestion avoidance calculations for TCP-NV */
static void tcpnv_acked(struct sock *sk, const struct ack_sample *sample) { const struct inet_connection_sock *icsk = inet_csk(sk); struct tcp_sock *tp = tcp_sk(sk); struct tcpnv *ca = inet_csk_ca(sk); unsigned long now = jiffies; s64 rate64 = 0; u32 rate, max_win, cwnd_by_slope; u32 avg_rtt; u32 bytes_acked = 0; /* Some calls are for duplicates without timetamps */ if (sample->rtt_us < 0) return; /* If not in TCP_CA_Open or TCP_CA_Disorder states, skip. */ if (icsk->icsk_ca_state != TCP_CA_Open && icsk->icsk_ca_state != TCP_CA_Disorder) return; /* Stop cwnd growth if we were in catch up mode */ if (ca->nv_catchup && tp->snd_cwnd >= nv_min_cwnd) { ca->nv_catchup = 0; ca->nv_allow_cwnd_growth = 0; } bytes_acked = tp->snd_una - ca->nv_last_snd_una; ca->nv_last_snd_una = tp->snd_una; if (sample->in_flight == 0) return; /* Calculate moving average of RTT */ if (nv_rtt_factor > 0) { if (ca->nv_last_rtt > 0) { avg_rtt = (((u64)sample->rtt_us) * nv_rtt_factor + ((u64)ca->nv_last_rtt) * (256 - nv_rtt_factor)) >> 8; } else { avg_rtt = sample->rtt_us; ca->nv_min_rtt = avg_rtt << 1; } ca->nv_last_rtt = avg_rtt; } else { avg_rtt = sample->rtt_us; } /* rate in 100's bits per second */ rate64 = ((u64)sample->in_flight) * 8000000; rate = (u32)div64_u64(rate64, (u64)(avg_rtt ?: 1) * 100); /* Remember the maximum rate seen during this RTT * Note: It may be more than one RTT. This function should be * called at least nv_dec_eval_min_calls times. */ if (ca->nv_rtt_max_rate < rate) ca->nv_rtt_max_rate = rate; /* We have valid information, increment counter */ if (ca->nv_eval_call_cnt < 255) ca->nv_eval_call_cnt++; /* update min rtt if necessary */ if (avg_rtt < ca->nv_min_rtt) ca->nv_min_rtt = avg_rtt; /* update future min_rtt if necessary */ if (avg_rtt < ca->nv_min_rtt_new) ca->nv_min_rtt_new = avg_rtt; /* nv_min_rtt is updated with the minimum (possibley averaged) rtt * seen in the last sysctl_tcp_nv_reset_period seconds (i.e. a * warm reset). This new nv_min_rtt will be continued to be updated * and be used for another sysctl_tcp_nv_reset_period seconds, * when it will be updated again. * In practice we introduce some randomness, so the actual period used * is chosen randomly from the range: * [sysctl_tcp_nv_reset_period*3/4, sysctl_tcp_nv_reset_period*5/4) */ if (time_after_eq(now, ca->nv_min_rtt_reset_jiffies)) { unsigned char rand; ca->nv_min_rtt = ca->nv_min_rtt_new; ca->nv_min_rtt_new = NV_INIT_RTT; get_random_bytes(&rand, 1); ca->nv_min_rtt_reset_jiffies = now + ((nv_reset_period * (384 + rand) * HZ) >> 9); /* Every so often we decrease ca->nv_min_cwnd in case previous * value is no longer accurate. */ ca->nv_min_cwnd = max(ca->nv_min_cwnd / 2, NV_MIN_CWND); } /* Once per RTT check if we need to do congestion avoidance */ if (before(ca->nv_rtt_start_seq, tp->snd_una)) { ca->nv_rtt_start_seq = tp->snd_nxt; if (ca->nv_rtt_cnt < 0xff) /* Increase counter for RTTs without CA decision */ ca->nv_rtt_cnt++; /* If this function is only called once within an RTT * the cwnd is probably too small (in some cases due to * tso, lro or interrupt coalescence), so we increase * ca->nv_min_cwnd. */ if (ca->nv_eval_call_cnt == 1 && bytes_acked >= (ca->nv_min_cwnd - 1) * tp->mss_cache && ca->nv_min_cwnd < (NV_TSO_CWND_BOUND + 1)) { ca->nv_min_cwnd = min(ca->nv_min_cwnd + NV_MIN_CWND_GROW, NV_TSO_CWND_BOUND + 1); ca->nv_rtt_start_seq = tp->snd_nxt + ca->nv_min_cwnd * tp->mss_cache; ca->nv_eval_call_cnt = 0; ca->nv_allow_cwnd_growth = 1; return; } /* Find the ideal cwnd for current rate from slope * slope = 80000.0 * mss / nv_min_rtt * cwnd_by_slope = nv_rtt_max_rate / slope */ cwnd_by_slope = (u32) div64_u64(((u64)ca->nv_rtt_max_rate) * ca->nv_min_rtt, (u64)(80000 * tp->mss_cache)); max_win = cwnd_by_slope + nv_pad; /* If cwnd > max_win, decrease cwnd * if cwnd < max_win, grow cwnd * else leave the same */ if (tp->snd_cwnd > max_win) { /* there is congestion, check that it is ok * to make a CA decision * 1. We should have at least nv_dec_eval_min_calls * data points before making a CA decision * 2. We only make a congesion decision after * nv_rtt_min_cnt RTTs */ if (ca->nv_rtt_cnt < nv_rtt_min_cnt) { return; } else if (tp->snd_ssthresh == TCP_INFINITE_SSTHRESH) { if (ca->nv_eval_call_cnt < nv_ssthresh_eval_min_calls) return; /* otherwise we will decrease cwnd */ } else if (ca->nv_eval_call_cnt < nv_dec_eval_min_calls) { if (ca->nv_allow_cwnd_growth && ca->nv_rtt_cnt > nv_stop_rtt_cnt) ca->nv_allow_cwnd_growth = 0; return; } /* We have enough data to determine we are congested */ ca->nv_allow_cwnd_growth = 0; tp->snd_ssthresh = (nv_ssthresh_factor * max_win) >> 3; if (tp->snd_cwnd - max_win > 2) { /* gap > 2, we do exponential cwnd decrease */ int dec; dec = max(2U, ((tp->snd_cwnd - max_win) * nv_cong_dec_mult) >> 7); tp->snd_cwnd -= dec; } else if (nv_cong_dec_mult > 0) { tp->snd_cwnd = max_win; } if (ca->cwnd_growth_factor > 0) ca->cwnd_growth_factor = 0; ca->nv_no_cong_cnt = 0; } else if (tp->snd_cwnd <= max_win - nv_pad_buffer) { /* There is no congestion, grow cwnd if allowed*/ if (ca->nv_eval_call_cnt < nv_inc_eval_min_calls) return; ca->nv_allow_cwnd_growth = 1; ca->nv_no_cong_cnt++; if (ca->cwnd_growth_factor < 0 && nv_cwnd_growth_rate_neg > 0 && ca->nv_no_cong_cnt > nv_cwnd_growth_rate_neg) { ca->cwnd_growth_factor++; ca->nv_no_cong_cnt = 0; } else if (ca->cwnd_growth_factor >= 0 && nv_cwnd_growth_rate_pos > 0 && ca->nv_no_cong_cnt > nv_cwnd_growth_rate_pos) { ca->cwnd_growth_factor++; ca->nv_no_cong_cnt = 0; } } else { /* cwnd is in-between, so do nothing */ return; } /* update state */ ca->nv_eval_call_cnt = 0; ca->nv_rtt_cnt = 0; ca->nv_rtt_max_rate = 0; /* Don't want to make cwnd < nv_min_cwnd * (it wasn't before, if it is now is because nv * decreased it). */ if (tp->snd_cwnd < nv_min_cwnd) tp->snd_cwnd = nv_min_cwnd; } }

Contributors

PersonTokensPropCommitsCommitProp
Lawrence Brakmo87199.54%150.00%
Konstantin Khlebnikov40.46%150.00%
Total875100.00%2100.00%

/* Extract info for Tcp socket info provided via netlink */
static size_t tcpnv_get_info(struct sock *sk, u32 ext, int *attr, union tcp_cc_info *info) { const struct tcpnv *ca = inet_csk_ca(sk); if (ext & (1 << (INET_DIAG_VEGASINFO - 1))) { info->vegas.tcpv_enabled = 1; info->vegas.tcpv_rttcnt = ca->nv_rtt_cnt; info->vegas.tcpv_rtt = ca->nv_last_rtt; info->vegas.tcpv_minrtt = ca->nv_min_rtt; *attr = INET_DIAG_VEGASINFO; return sizeof(struct tcpvegas_info); } return 0; }

Contributors

PersonTokensPropCommitsCommitProp
Lawrence Brakmo10299.03%150.00%
Stephen Hemminger10.97%150.00%
Total103100.00%2100.00%

static struct tcp_congestion_ops tcpnv __read_mostly = { .init = tcpnv_init, .ssthresh = tcpnv_recalc_ssthresh, .cong_avoid = tcpnv_cong_avoid, .set_state = tcpnv_state, .undo_cwnd = tcp_reno_undo_cwnd, .pkts_acked = tcpnv_acked, .get_info = tcpnv_get_info, .owner = THIS_MODULE, .name = "nv", };
static int __init tcpnv_register(void) { BUILD_BUG_ON(sizeof(struct tcpnv) > ICSK_CA_PRIV_SIZE); return tcp_register_congestion_control(&tcpnv); }

Contributors

PersonTokensPropCommitsCommitProp
Lawrence Brakmo27100.00%1100.00%
Total27100.00%1100.00%


static void __exit tcpnv_unregister(void) { tcp_unregister_congestion_control(&tcpnv); }

Contributors

PersonTokensPropCommitsCommitProp
Lawrence Brakmo15100.00%1100.00%
Total15100.00%1100.00%

module_init(tcpnv_register); module_exit(tcpnv_unregister); MODULE_AUTHOR("Lawrence Brakmo"); MODULE_LICENSE("GPL"); MODULE_DESCRIPTION("TCP NV"); MODULE_VERSION("1.0");

Overall Contributors

PersonTokensPropCommitsCommitProp
Lawrence Brakmo181699.67%125.00%
Konstantin Khlebnikov40.22%125.00%
Yuchung Cheng10.05%125.00%
Stephen Hemminger10.05%125.00%
Total1822100.00%4100.00%
Directory: net/ipv4
Information contained on this website is for historical information purposes only and does not indicate or represent copyright ownership.
Created with cregit.