1
2
3
4
5
6
7
8
9
10
11
12#include <linux/mm.h>
13#include <linux/module.h>
14#include <linux/skbuff.h>
15#include <linux/inet_diag.h>
16
17#include <net/tcp.h>
18
19
20
21
22#define V_PARAM_SHIFT 1
23static const int beta = 3 << V_PARAM_SHIFT;
24
25
26struct veno {
27 u8 doing_veno_now;
28 u16 cntrtt;
29 u32 minrtt;
30 u32 basertt;
31 u32 inc;
32 u32 diff;
33};
34
35
36
37
38
39
40
41
42
43
44static inline void veno_enable(struct sock *sk)
45{
46 struct veno *veno = inet_csk_ca(sk);
47
48
49 veno->doing_veno_now = 1;
50
51 veno->minrtt = 0x7fffffff;
52}
53
54static inline void veno_disable(struct sock *sk)
55{
56 struct veno *veno = inet_csk_ca(sk);
57
58
59 veno->doing_veno_now = 0;
60}
61
62static void tcp_veno_init(struct sock *sk)
63{
64 struct veno *veno = inet_csk_ca(sk);
65
66 veno->basertt = 0x7fffffff;
67 veno->inc = 1;
68 veno_enable(sk);
69}
70
71
72static void tcp_veno_pkts_acked(struct sock *sk, u32 cnt, s32 rtt_us)
73{
74 struct veno *veno = inet_csk_ca(sk);
75 u32 vrtt;
76
77 if (rtt_us < 0)
78 return;
79
80
81 vrtt = rtt_us + 1;
82
83
84 if (vrtt < veno->basertt)
85 veno->basertt = vrtt;
86
87
88
89
90 veno->minrtt = min(veno->minrtt, vrtt);
91 veno->cntrtt++;
92}
93
94static void tcp_veno_state(struct sock *sk, u8 ca_state)
95{
96 if (ca_state == TCP_CA_Open)
97 veno_enable(sk);
98 else
99 veno_disable(sk);
100}
101
102
103
104
105
106
107
108
109
110
111static void tcp_veno_cwnd_event(struct sock *sk, enum tcp_ca_event event)
112{
113 if (event == CA_EVENT_CWND_RESTART || event == CA_EVENT_TX_START)
114 tcp_veno_init(sk);
115}
116
117static void tcp_veno_cong_avoid(struct sock *sk, u32 ack, u32 in_flight)
118{
119 struct tcp_sock *tp = tcp_sk(sk);
120 struct veno *veno = inet_csk_ca(sk);
121
122 if (!veno->doing_veno_now) {
123 tcp_reno_cong_avoid(sk, ack, in_flight);
124 return;
125 }
126
127
128 if (!tcp_is_cwnd_limited(sk, in_flight))
129 return;
130
131
132 if (veno->cntrtt <= 2) {
133
134
135
136 tcp_reno_cong_avoid(sk, ack, in_flight);
137 } else {
138 u64 target_cwnd;
139 u32 rtt;
140
141
142
143
144
145 rtt = veno->minrtt;
146
147 target_cwnd = (tp->snd_cwnd * veno->basertt);
148 target_cwnd <<= V_PARAM_SHIFT;
149 do_div(target_cwnd, rtt);
150
151 veno->diff = (tp->snd_cwnd << V_PARAM_SHIFT) - target_cwnd;
152
153 if (tp->snd_cwnd <= tp->snd_ssthresh) {
154
155 tcp_slow_start(tp);
156 } else {
157
158 if (veno->diff < beta) {
159
160
161
162 tcp_cong_avoid_ai(tp, tp->snd_cwnd);
163 } else {
164
165
166
167 if (tp->snd_cwnd_cnt >= tp->snd_cwnd) {
168 if (veno->inc &&
169 tp->snd_cwnd < tp->snd_cwnd_clamp) {
170 tp->snd_cwnd++;
171 veno->inc = 0;
172 } else
173 veno->inc = 1;
174 tp->snd_cwnd_cnt = 0;
175 } else
176 tp->snd_cwnd_cnt++;
177 }
178
179 }
180 if (tp->snd_cwnd < 2)
181 tp->snd_cwnd = 2;
182 else if (tp->snd_cwnd > tp->snd_cwnd_clamp)
183 tp->snd_cwnd = tp->snd_cwnd_clamp;
184 }
185
186
187 veno->minrtt = 0x7fffffff;
188}
189
190
191static u32 tcp_veno_ssthresh(struct sock *sk)
192{
193 const struct tcp_sock *tp = tcp_sk(sk);
194 struct veno *veno = inet_csk_ca(sk);
195
196 if (veno->diff < beta)
197
198 return max(tp->snd_cwnd * 4 / 5, 2U);
199 else
200
201 return max(tp->snd_cwnd >> 1U, 2U);
202}
203
204static struct tcp_congestion_ops tcp_veno __read_mostly = {
205 .flags = TCP_CONG_RTT_STAMP,
206 .init = tcp_veno_init,
207 .ssthresh = tcp_veno_ssthresh,
208 .cong_avoid = tcp_veno_cong_avoid,
209 .pkts_acked = tcp_veno_pkts_acked,
210 .set_state = tcp_veno_state,
211 .cwnd_event = tcp_veno_cwnd_event,
212
213 .owner = THIS_MODULE,
214 .name = "veno",
215};
216
217static int __init tcp_veno_register(void)
218{
219 BUILD_BUG_ON(sizeof(struct veno) > ICSK_CA_PRIV_SIZE);
220 tcp_register_congestion_control(&tcp_veno);
221 return 0;
222}
223
224static void __exit tcp_veno_unregister(void)
225{
226 tcp_unregister_congestion_control(&tcp_veno);
227}
228
229module_init(tcp_veno_register);
230module_exit(tcp_veno_unregister);
231
232MODULE_AUTHOR("Bin Zhou, Cheng Peng Fu");
233MODULE_LICENSE("GPL");
234MODULE_DESCRIPTION("TCP Veno");
235