request_sock.h 6.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259
  1. /*
  2. * NET Generic infrastructure for Network protocols.
  3. *
  4. * Definitions for request_sock
  5. *
  6. * Authors: Arnaldo Carvalho de Melo <acme@conectiva.com.br>
  7. *
  8. * From code originally in include/net/tcp.h
  9. *
  10. * This program is free software; you can redistribute it and/or
  11. * modify it under the terms of the GNU General Public License
  12. * as published by the Free Software Foundation; either version
  13. * 2 of the License, or (at your option) any later version.
  14. */
  15. #ifndef _REQUEST_SOCK_H
  16. #define _REQUEST_SOCK_H
  17. #include <linux/slab.h>
  18. #include <linux/spinlock.h>
  19. #include <linux/types.h>
  20. #include <linux/bug.h>
  21. #include <net/sock.h>
  22. struct request_sock;
  23. struct sk_buff;
  24. struct dst_entry;
  25. struct proto;
  26. /* empty to "strongly type" an otherwise void parameter.
  27. */
  28. struct request_values {
  29. };
  30. struct request_sock_ops {
  31. int family;
  32. int obj_size;
  33. struct kmem_cache *slab;
  34. char *slab_name;
  35. int (*rtx_syn_ack)(struct sock *sk,
  36. struct request_sock *req,
  37. struct request_values *rvp);
  38. void (*send_ack)(struct sock *sk, struct sk_buff *skb,
  39. struct request_sock *req);
  40. void (*send_reset)(struct sock *sk,
  41. struct sk_buff *skb);
  42. void (*destructor)(struct request_sock *req);
  43. void (*syn_ack_timeout)(struct sock *sk,
  44. struct request_sock *req);
  45. };
  46. /* struct request_sock - mini sock to represent a connection request
  47. */
  48. struct request_sock {
  49. struct request_sock *dl_next; /* Must be first member! */
  50. u16 mss;
  51. u8 retrans;
  52. u8 cookie_ts; /* syncookie: encode tcpopts in timestamp */
  53. /* The following two fields can be easily recomputed I think -AK */
  54. u32 window_clamp; /* window clamp at creation time */
  55. u32 rcv_wnd; /* rcv_wnd offered first time */
  56. u32 ts_recent;
  57. unsigned long expires;
  58. const struct request_sock_ops *rsk_ops;
  59. struct sock *sk;
  60. u32 secid;
  61. u32 peer_secid;
  62. };
  63. static inline struct request_sock *reqsk_alloc(const struct request_sock_ops *ops)
  64. {
  65. struct request_sock *req = kmem_cache_alloc(ops->slab, GFP_ATOMIC);
  66. if (req != NULL)
  67. req->rsk_ops = ops;
  68. return req;
  69. }
  70. static inline void __reqsk_free(struct request_sock *req)
  71. {
  72. kmem_cache_free(req->rsk_ops->slab, req);
  73. }
  74. static inline void reqsk_free(struct request_sock *req)
  75. {
  76. req->rsk_ops->destructor(req);
  77. __reqsk_free(req);
  78. }
  79. extern int sysctl_max_syn_backlog;
  80. /** struct listen_sock - listen state
  81. *
  82. * @max_qlen_log - log_2 of maximal queued SYNs/REQUESTs
  83. */
  84. struct listen_sock {
  85. u8 max_qlen_log;
  86. u8 synflood_warned;
  87. /* 2 bytes hole, try to use */
  88. int qlen;
  89. int qlen_young;
  90. int clock_hand;
  91. u32 hash_rnd;
  92. u32 nr_table_entries;
  93. struct request_sock *syn_table[0];
  94. };
  95. /** struct request_sock_queue - queue of request_socks
  96. *
  97. * @rskq_accept_head - FIFO head of established children
  98. * @rskq_accept_tail - FIFO tail of established children
  99. * @rskq_defer_accept - User waits for some data after accept()
  100. * @syn_wait_lock - serializer
  101. *
  102. * %syn_wait_lock is necessary only to avoid proc interface having to grab the main
  103. * lock sock while browsing the listening hash (otherwise it's deadlock prone).
  104. *
  105. * This lock is acquired in read mode only from listening_get_next() seq_file
  106. * op and it's acquired in write mode _only_ from code that is actively
  107. * changing rskq_accept_head. All readers that are holding the master sock lock
  108. * don't need to grab this lock in read mode too as rskq_accept_head. writes
  109. * are always protected from the main sock lock.
  110. */
  111. struct request_sock_queue {
  112. struct request_sock *rskq_accept_head;
  113. struct request_sock *rskq_accept_tail;
  114. rwlock_t syn_wait_lock;
  115. u8 rskq_defer_accept;
  116. /* 3 bytes hole, try to pack */
  117. struct listen_sock *listen_opt;
  118. };
  119. extern int reqsk_queue_alloc(struct request_sock_queue *queue,
  120. unsigned int nr_table_entries);
  121. extern void __reqsk_queue_destroy(struct request_sock_queue *queue);
  122. extern void reqsk_queue_destroy(struct request_sock_queue *queue);
  123. static inline struct request_sock *
  124. reqsk_queue_yank_acceptq(struct request_sock_queue *queue)
  125. {
  126. struct request_sock *req = queue->rskq_accept_head;
  127. queue->rskq_accept_head = NULL;
  128. return req;
  129. }
  130. static inline int reqsk_queue_empty(struct request_sock_queue *queue)
  131. {
  132. return queue->rskq_accept_head == NULL;
  133. }
  134. static inline void reqsk_queue_unlink(struct request_sock_queue *queue,
  135. struct request_sock *req,
  136. struct request_sock **prev_req)
  137. {
  138. write_lock(&queue->syn_wait_lock);
  139. *prev_req = req->dl_next;
  140. write_unlock(&queue->syn_wait_lock);
  141. }
  142. static inline void reqsk_queue_add(struct request_sock_queue *queue,
  143. struct request_sock *req,
  144. struct sock *parent,
  145. struct sock *child)
  146. {
  147. req->sk = child;
  148. sk_acceptq_added(parent);
  149. if (queue->rskq_accept_head == NULL)
  150. queue->rskq_accept_head = req;
  151. else
  152. queue->rskq_accept_tail->dl_next = req;
  153. queue->rskq_accept_tail = req;
  154. req->dl_next = NULL;
  155. }
  156. static inline struct request_sock *reqsk_queue_remove(struct request_sock_queue *queue)
  157. {
  158. struct request_sock *req = queue->rskq_accept_head;
  159. WARN_ON(req == NULL);
  160. queue->rskq_accept_head = req->dl_next;
  161. if (queue->rskq_accept_head == NULL)
  162. queue->rskq_accept_tail = NULL;
  163. return req;
  164. }
  165. static inline struct sock *reqsk_queue_get_child(struct request_sock_queue *queue,
  166. struct sock *parent)
  167. {
  168. struct request_sock *req = reqsk_queue_remove(queue);
  169. struct sock *child = req->sk;
  170. WARN_ON(child == NULL);
  171. sk_acceptq_removed(parent);
  172. __reqsk_free(req);
  173. return child;
  174. }
  175. static inline int reqsk_queue_removed(struct request_sock_queue *queue,
  176. struct request_sock *req)
  177. {
  178. struct listen_sock *lopt = queue->listen_opt;
  179. if (req->retrans == 0)
  180. --lopt->qlen_young;
  181. return --lopt->qlen;
  182. }
  183. static inline int reqsk_queue_added(struct request_sock_queue *queue)
  184. {
  185. struct listen_sock *lopt = queue->listen_opt;
  186. const int prev_qlen = lopt->qlen;
  187. lopt->qlen_young++;
  188. lopt->qlen++;
  189. return prev_qlen;
  190. }
  191. static inline int reqsk_queue_len(const struct request_sock_queue *queue)
  192. {
  193. return queue->listen_opt != NULL ? queue->listen_opt->qlen : 0;
  194. }
  195. static inline int reqsk_queue_len_young(const struct request_sock_queue *queue)
  196. {
  197. return queue->listen_opt->qlen_young;
  198. }
  199. static inline int reqsk_queue_is_full(const struct request_sock_queue *queue)
  200. {
  201. return queue->listen_opt->qlen >> queue->listen_opt->max_qlen_log;
  202. }
  203. static inline void reqsk_queue_hash_req(struct request_sock_queue *queue,
  204. u32 hash, struct request_sock *req,
  205. unsigned long timeout)
  206. {
  207. struct listen_sock *lopt = queue->listen_opt;
  208. req->expires = jiffies + timeout;
  209. req->retrans = 0;
  210. req->sk = NULL;
  211. req->dl_next = lopt->syn_table[hash];
  212. write_lock(&queue->syn_wait_lock);
  213. lopt->syn_table[hash] = req;
  214. write_unlock(&queue->syn_wait_lock);
  215. }
  216. #endif /* _REQUEST_SOCK_H */