udp.h 8.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273
  1. /*
  2. * INET An implementation of the TCP/IP protocol suite for the LINUX
  3. * operating system. INET is implemented using the BSD Socket
  4. * interface as the means of communication with the user level.
  5. *
  6. * Definitions for the UDP module.
  7. *
  8. * Version: @(#)udp.h 1.0.2 05/07/93
  9. *
  10. * Authors: Ross Biro
  11. * Fred N. van Kempen, <waltje@uWalt.NL.Mugnet.ORG>
  12. *
  13. * Fixes:
  14. * Alan Cox : Turned on udp checksums. I don't want to
  15. * chase 'memory corruption' bugs that aren't!
  16. *
  17. * This program is free software; you can redistribute it and/or
  18. * modify it under the terms of the GNU General Public License
  19. * as published by the Free Software Foundation; either version
  20. * 2 of the License, or (at your option) any later version.
  21. */
  22. #ifndef _UDP_H
  23. #define _UDP_H
  24. #include <linux/list.h>
  25. #include <linux/bug.h>
  26. #include <net/inet_sock.h>
  27. #include <net/sock.h>
  28. #include <net/snmp.h>
  29. #include <net/ip.h>
  30. #include <linux/ipv6.h>
  31. #include <linux/seq_file.h>
  32. #include <linux/poll.h>
  33. /**
  34. * struct udp_skb_cb - UDP(-Lite) private variables
  35. *
  36. * @header: private variables used by IPv4/IPv6
  37. * @cscov: checksum coverage length (UDP-Lite only)
  38. * @partial_cov: if set indicates partial csum coverage
  39. */
  40. struct udp_skb_cb {
  41. union {
  42. struct inet_skb_parm h4;
  43. #if IS_ENABLED(CONFIG_IPV6)
  44. struct inet6_skb_parm h6;
  45. #endif
  46. } header;
  47. __u16 cscov;
  48. __u8 partial_cov;
  49. };
  50. #define UDP_SKB_CB(__skb) ((struct udp_skb_cb *)((__skb)->cb))
  51. /**
  52. * struct udp_hslot - UDP hash slot
  53. *
  54. * @head: head of list of sockets
  55. * @count: number of sockets in 'head' list
  56. * @lock: spinlock protecting changes to head/count
  57. */
  58. struct udp_hslot {
  59. struct hlist_nulls_head head;
  60. int count;
  61. spinlock_t lock;
  62. } __attribute__((aligned(2 * sizeof(long))));
  63. /**
  64. * struct udp_table - UDP table
  65. *
  66. * @hash: hash table, sockets are hashed on (local port)
  67. * @hash2: hash table, sockets are hashed on (local port, local address)
  68. * @mask: number of slots in hash tables, minus 1
  69. * @log: log2(number of slots in hash table)
  70. */
  71. struct udp_table {
  72. struct udp_hslot *hash;
  73. struct udp_hslot *hash2;
  74. unsigned int mask;
  75. unsigned int log;
  76. };
  77. extern struct udp_table udp_table;
  78. extern void udp_table_init(struct udp_table *, const char *);
  79. static inline struct udp_hslot *udp_hashslot(struct udp_table *table,
  80. struct net *net, unsigned int num)
  81. {
  82. return &table->hash[udp_hashfn(net, num, table->mask)];
  83. }
  84. /*
  85. * For secondary hash, net_hash_mix() is performed before calling
  86. * udp_hashslot2(), this explains difference with udp_hashslot()
  87. */
  88. static inline struct udp_hslot *udp_hashslot2(struct udp_table *table,
  89. unsigned int hash)
  90. {
  91. return &table->hash2[hash & table->mask];
  92. }
  93. /* Note: this must match 'valbool' in sock_setsockopt */
  94. #define UDP_CSUM_NOXMIT 1
  95. /* Used by SunRPC/xprt layer. */
  96. #define UDP_CSUM_NORCV 2
  97. /* Default, as per the RFC, is to always do csums. */
  98. #define UDP_CSUM_DEFAULT 0
  99. extern struct proto udp_prot;
  100. extern atomic_long_t udp_memory_allocated;
  101. /* sysctl variables for udp */
  102. extern long sysctl_udp_mem[3];
  103. extern int sysctl_udp_rmem_min;
  104. extern int sysctl_udp_wmem_min;
  105. struct sk_buff;
  106. /*
  107. * Generic checksumming routines for UDP(-Lite) v4 and v6
  108. */
  109. static inline __sum16 __udp_lib_checksum_complete(struct sk_buff *skb)
  110. {
  111. return __skb_checksum_complete_head(skb, UDP_SKB_CB(skb)->cscov);
  112. }
  113. static inline int udp_lib_checksum_complete(struct sk_buff *skb)
  114. {
  115. return !skb_csum_unnecessary(skb) &&
  116. __udp_lib_checksum_complete(skb);
  117. }
  118. /**
  119. * udp_csum_outgoing - compute UDPv4/v6 checksum over fragments
  120. * @sk: socket we are writing to
  121. * @skb: sk_buff containing the filled-in UDP header
  122. * (checksum field must be zeroed out)
  123. */
  124. static inline __wsum udp_csum_outgoing(struct sock *sk, struct sk_buff *skb)
  125. {
  126. __wsum csum = csum_partial(skb_transport_header(skb),
  127. sizeof(struct udphdr), 0);
  128. skb_queue_walk(&sk->sk_write_queue, skb) {
  129. csum = csum_add(csum, skb->csum);
  130. }
  131. return csum;
  132. }
  133. static inline __wsum udp_csum(struct sk_buff *skb)
  134. {
  135. __wsum csum = csum_partial(skb_transport_header(skb),
  136. sizeof(struct udphdr), skb->csum);
  137. for (skb = skb_shinfo(skb)->frag_list; skb; skb = skb->next) {
  138. csum = csum_add(csum, skb->csum);
  139. }
  140. return csum;
  141. }
  142. /* hash routines shared between UDPv4/6 and UDP-Litev4/6 */
  143. static inline void udp_lib_hash(struct sock *sk)
  144. {
  145. BUG();
  146. }
  147. extern void udp_lib_unhash(struct sock *sk);
  148. extern void udp_lib_rehash(struct sock *sk, u16 new_hash);
  149. static inline void udp_lib_close(struct sock *sk, long timeout)
  150. {
  151. sk_common_release(sk);
  152. }
  153. extern int udp_lib_get_port(struct sock *sk, unsigned short snum,
  154. int (*)(const struct sock *,const struct sock *),
  155. unsigned int hash2_nulladdr);
  156. /* net/ipv4/udp.c */
  157. extern int udp_get_port(struct sock *sk, unsigned short snum,
  158. int (*saddr_cmp)(const struct sock *,
  159. const struct sock *));
  160. extern void udp_err(struct sk_buff *, u32);
  161. int udp_abort(struct sock *sk, int err);
  162. extern int udp_sendmsg(struct kiocb *iocb, struct sock *sk,
  163. struct msghdr *msg, size_t len);
  164. extern int udp_push_pending_frames(struct sock *sk);
  165. extern void udp_flush_pending_frames(struct sock *sk);
  166. extern int udp_rcv(struct sk_buff *skb);
  167. extern int udp_ioctl(struct sock *sk, int cmd, unsigned long arg);
  168. extern int udp_disconnect(struct sock *sk, int flags);
  169. extern unsigned int udp_poll(struct file *file, struct socket *sock,
  170. poll_table *wait);
  171. extern int udp_lib_getsockopt(struct sock *sk, int level, int optname,
  172. char __user *optval, int __user *optlen);
  173. extern int udp_lib_setsockopt(struct sock *sk, int level, int optname,
  174. char __user *optval, unsigned int optlen,
  175. int (*push_pending_frames)(struct sock *));
  176. extern struct sock *udp4_lib_lookup(struct net *net, __be32 saddr, __be16 sport,
  177. __be32 daddr, __be16 dport,
  178. int dif);
  179. extern struct sock *__udp4_lib_lookup(struct net *net, __be32 saddr, __be16 sport,
  180. __be32 daddr, __be16 dport,
  181. int dif, struct udp_table *tbl);
  182. extern struct sock *udp6_lib_lookup(struct net *net, const struct in6_addr *saddr, __be16 sport,
  183. const struct in6_addr *daddr, __be16 dport,
  184. int dif);
  185. extern struct sock *__udp6_lib_lookup(struct net *net, const struct in6_addr *saddr, __be16 sport,
  186. const struct in6_addr *daddr, __be16 dport,
  187. int dif, struct udp_table *tbl);
  188. /*
  189. * SNMP statistics for UDP and UDP-Lite
  190. */
  191. #define UDP_INC_STATS_USER(net, field, is_udplite) do { \
  192. if (is_udplite) SNMP_INC_STATS_USER((net)->mib.udplite_statistics, field); \
  193. else SNMP_INC_STATS_USER((net)->mib.udp_statistics, field); } while(0)
  194. #define UDP_INC_STATS_BH(net, field, is_udplite) do { \
  195. if (is_udplite) SNMP_INC_STATS_BH((net)->mib.udplite_statistics, field); \
  196. else SNMP_INC_STATS_BH((net)->mib.udp_statistics, field); } while(0)
  197. #define UDP6_INC_STATS_BH(net, field, is_udplite) do { \
  198. if (is_udplite) SNMP_INC_STATS_BH((net)->mib.udplite_stats_in6, field);\
  199. else SNMP_INC_STATS_BH((net)->mib.udp_stats_in6, field); \
  200. } while(0)
  201. #define UDP6_INC_STATS_USER(net, field, __lite) do { \
  202. if (__lite) SNMP_INC_STATS_USER((net)->mib.udplite_stats_in6, field); \
  203. else SNMP_INC_STATS_USER((net)->mib.udp_stats_in6, field); \
  204. } while(0)
  205. #if IS_ENABLED(CONFIG_IPV6)
  206. #define UDPX_INC_STATS_BH(sk, field) \
  207. do { \
  208. if ((sk)->sk_family == AF_INET) \
  209. UDP_INC_STATS_BH(sock_net(sk), field, 0); \
  210. else \
  211. UDP6_INC_STATS_BH(sock_net(sk), field, 0); \
  212. } while (0);
  213. #else
  214. #define UDPX_INC_STATS_BH(sk, field) UDP_INC_STATS_BH(sock_net(sk), field, 0)
  215. #endif
  216. /* /proc */
  217. int udp_seq_open(struct inode *inode, struct file *file);
  218. struct udp_seq_afinfo {
  219. char *name;
  220. sa_family_t family;
  221. struct udp_table *udp_table;
  222. const struct file_operations *seq_fops;
  223. struct seq_operations seq_ops;
  224. };
  225. struct udp_iter_state {
  226. struct seq_net_private p;
  227. sa_family_t family;
  228. int bucket;
  229. struct udp_table *udp_table;
  230. };
  231. #ifdef CONFIG_PROC_FS
  232. extern int udp_proc_register(struct net *net, struct udp_seq_afinfo *afinfo);
  233. extern void udp_proc_unregister(struct net *net, struct udp_seq_afinfo *afinfo);
  234. extern int udp4_proc_init(void);
  235. extern void udp4_proc_exit(void);
  236. #endif
  237. extern void udp_init(void);
  238. extern int udp4_ufo_send_check(struct sk_buff *skb);
  239. extern struct sk_buff *udp4_ufo_fragment(struct sk_buff *skb,
  240. netdev_features_t features);
  241. #endif /* _UDP_H */