tcp_ecn.h
上传用户:lgb322
上传日期:2013-02-24
资源大小:30529k
文件大小:3k
源码类别:

嵌入式Linux

开发平台:

Unix_Linux

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