| #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 |
