hc
2024-05-09 b9d5c334faa47a75f1f28e72d203fc0334e8471d
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
#ifndef _TCP_DCTCP_H
#define _TCP_DCTCP_H
 
static inline void dctcp_ece_ack_cwr(struct sock *sk, u32 ce_state)
{
   struct tcp_sock *tp = tcp_sk(sk);
 
   if (ce_state == 1)
       tp->ecn_flags |= TCP_ECN_DEMAND_CWR;
   else
       tp->ecn_flags &= ~TCP_ECN_DEMAND_CWR;
}
 
/* Minimal DCTP CE state machine:
 *
 * S:    0 <- last pkt was non-CE
 *    1 <- last pkt was CE
 */
static inline void dctcp_ece_ack_update(struct sock *sk, enum tcp_ca_event evt,
                   u32 *prior_rcv_nxt, u32 *ce_state)
{
   u32 new_ce_state = (evt == CA_EVENT_ECN_IS_CE) ? 1 : 0;
 
   if (*ce_state != new_ce_state) {
       /* CE state has changed, force an immediate ACK to
        * reflect the new CE state. If an ACK was delayed,
        * send that first to reflect the prior CE state.
        */
       if (inet_csk(sk)->icsk_ack.pending & ICSK_ACK_TIMER) {
           dctcp_ece_ack_cwr(sk, *ce_state);
           __tcp_send_ack(sk, *prior_rcv_nxt);
       }
       inet_csk(sk)->icsk_ack.pending |= ICSK_ACK_NOW;
   }
   *prior_rcv_nxt = tcp_sk(sk)->rcv_nxt;
   *ce_state = new_ce_state;
   dctcp_ece_ack_cwr(sk, new_ce_state);
}
 
#endif