[NET]: Add ECN support for TSO
[linux-2.6.git] / include / net / tcp_ecn.h
1 #ifndef _NET_TCP_ECN_H_
2 #define _NET_TCP_ECN_H_ 1
3
4 #include <net/inet_ecn.h>
5 #include <net/request_sock.h>
6
7 #define TCP_HP_BITS (~(TCP_RESERVED_BITS|TCP_FLAG_PSH))
8
9 #define TCP_ECN_OK              1
10 #define TCP_ECN_QUEUE_CWR       2
11 #define TCP_ECN_DEMAND_CWR      4
12
13 static inline void TCP_ECN_queue_cwr(struct tcp_sock *tp)
14 {
15         if (tp->ecn_flags&TCP_ECN_OK)
16                 tp->ecn_flags |= TCP_ECN_QUEUE_CWR;
17 }
18
19
20 /* Output functions */
21
22 static inline void TCP_ECN_send_synack(struct tcp_sock *tp,
23                                        struct sk_buff *skb)
24 {
25         TCP_SKB_CB(skb)->flags &= ~TCPCB_FLAG_CWR;
26         if (!(tp->ecn_flags&TCP_ECN_OK))
27                 TCP_SKB_CB(skb)->flags &= ~TCPCB_FLAG_ECE;
28 }
29
30 static inline void TCP_ECN_send_syn(struct sock *sk, struct tcp_sock *tp,
31                                     struct sk_buff *skb)
32 {
33         tp->ecn_flags = 0;
34         if (sysctl_tcp_ecn) {
35                 TCP_SKB_CB(skb)->flags |= TCPCB_FLAG_ECE|TCPCB_FLAG_CWR;
36                 tp->ecn_flags = TCP_ECN_OK;
37         }
38 }
39
40 static __inline__ void
41 TCP_ECN_make_synack(struct request_sock *req, struct tcphdr *th)
42 {
43         if (inet_rsk(req)->ecn_ok)
44                 th->ece = 1;
45 }
46
47 static inline void TCP_ECN_send(struct sock *sk, struct tcp_sock *tp,
48                                 struct sk_buff *skb, int tcp_header_len)
49 {
50         if (tp->ecn_flags & TCP_ECN_OK) {
51                 /* Not-retransmitted data segment: set ECT and inject CWR. */
52                 if (skb->len != tcp_header_len &&
53                     !before(TCP_SKB_CB(skb)->seq, tp->snd_nxt)) {
54                         INET_ECN_xmit(sk);
55                         if (tp->ecn_flags&TCP_ECN_QUEUE_CWR) {
56                                 tp->ecn_flags &= ~TCP_ECN_QUEUE_CWR;
57                                 skb->h.th->cwr = 1;
58                                 if (skb_shinfo(skb)->gso_type & SKB_GSO_TCPV4)
59                                         skb_shinfo(skb)->gso_type |=
60                                                 SKB_GSO_TCPV4_ECN;
61                         }
62                 } else {
63                         /* ACK or retransmitted segment: clear ECT|CE */
64                         INET_ECN_dontxmit(sk);
65                 }
66                 if (tp->ecn_flags & TCP_ECN_DEMAND_CWR)
67                         skb->h.th->ece = 1;
68         }
69 }
70
71 /* Input functions */
72
73 static inline void TCP_ECN_accept_cwr(struct tcp_sock *tp, struct sk_buff *skb)
74 {
75         if (skb->h.th->cwr)
76                 tp->ecn_flags &= ~TCP_ECN_DEMAND_CWR;
77 }
78
79 static inline void TCP_ECN_withdraw_cwr(struct tcp_sock *tp)
80 {
81         tp->ecn_flags &= ~TCP_ECN_DEMAND_CWR;
82 }
83
84 static inline void TCP_ECN_check_ce(struct tcp_sock *tp, struct sk_buff *skb)
85 {
86         if (tp->ecn_flags&TCP_ECN_OK) {
87                 if (INET_ECN_is_ce(TCP_SKB_CB(skb)->flags))
88                         tp->ecn_flags |= TCP_ECN_DEMAND_CWR;
89                 /* Funny extension: if ECT is not set on a segment,
90                  * it is surely retransmit. It is not in ECN RFC,
91                  * but Linux follows this rule. */
92                 else if (INET_ECN_is_not_ect((TCP_SKB_CB(skb)->flags)))
93                         tcp_enter_quickack_mode((struct sock *)tp);
94         }
95 }
96
97 static inline void TCP_ECN_rcv_synack(struct tcp_sock *tp, struct tcphdr *th)
98 {
99         if ((tp->ecn_flags&TCP_ECN_OK) && (!th->ece || th->cwr))
100                 tp->ecn_flags &= ~TCP_ECN_OK;
101 }
102
103 static inline void TCP_ECN_rcv_syn(struct tcp_sock *tp, struct tcphdr *th)
104 {
105         if ((tp->ecn_flags&TCP_ECN_OK) && (!th->ece || !th->cwr))
106                 tp->ecn_flags &= ~TCP_ECN_OK;
107 }
108
109 static inline int TCP_ECN_rcv_ecn_echo(struct tcp_sock *tp, struct tcphdr *th)
110 {
111         if (th->ece && !th->syn && (tp->ecn_flags&TCP_ECN_OK))
112                 return 1;
113         return 0;
114 }
115
116 static inline void TCP_ECN_openreq_child(struct tcp_sock *tp,
117                                          struct request_sock *req)
118 {
119         tp->ecn_flags = inet_rsk(req)->ecn_ok ? TCP_ECN_OK : 0;
120 }
121
122 static __inline__ void
123 TCP_ECN_create_request(struct request_sock *req, struct tcphdr *th)
124 {
125         if (sysctl_tcp_ecn && th->ece && th->cwr)
126                 inet_rsk(req)->ecn_ok = 1;
127 }
128
129 #endif