blob: 002939cfc069d2fe5eabe9b8ec4cefc1e20b2112 [file] [log] [blame]
Linus Torvalds1da177e2005-04-16 15:20:36 -07001/*
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 * Generic socket support routines. Memory allocators, socket lock/release
7 * handler for protocols to use and generic option handler.
8 *
9 *
Jesper Juhl02c30a82005-05-05 16:16:16 -070010 * Authors: Ross Biro
Linus Torvalds1da177e2005-04-16 15:20:36 -070011 * Fred N. van Kempen, <waltje@uWalt.NL.Mugnet.ORG>
12 * Florian La Roche, <flla@stud.uni-sb.de>
13 * Alan Cox, <A.Cox@swansea.ac.uk>
14 *
15 * Fixes:
16 * Alan Cox : Numerous verify_area() problems
17 * Alan Cox : Connecting on a connecting socket
18 * now returns an error for tcp.
19 * Alan Cox : sock->protocol is set correctly.
20 * and is not sometimes left as 0.
21 * Alan Cox : connect handles icmp errors on a
22 * connect properly. Unfortunately there
23 * is a restart syscall nasty there. I
24 * can't match BSD without hacking the C
25 * library. Ideas urgently sought!
26 * Alan Cox : Disallow bind() to addresses that are
27 * not ours - especially broadcast ones!!
28 * Alan Cox : Socket 1024 _IS_ ok for users. (fencepost)
29 * Alan Cox : sock_wfree/sock_rfree don't destroy sockets,
30 * instead they leave that for the DESTROY timer.
31 * Alan Cox : Clean up error flag in accept
32 * Alan Cox : TCP ack handling is buggy, the DESTROY timer
33 * was buggy. Put a remove_sock() in the handler
34 * for memory when we hit 0. Also altered the timer
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +090035 * code. The ACK stuff can wait and needs major
Linus Torvalds1da177e2005-04-16 15:20:36 -070036 * TCP layer surgery.
37 * Alan Cox : Fixed TCP ack bug, removed remove sock
38 * and fixed timer/inet_bh race.
39 * Alan Cox : Added zapped flag for TCP
40 * Alan Cox : Move kfree_skb into skbuff.c and tidied up surplus code
41 * Alan Cox : for new sk_buff allocations wmalloc/rmalloc now call alloc_skb
42 * Alan Cox : kfree_s calls now are kfree_skbmem so we can track skb resources
43 * Alan Cox : Supports socket option broadcast now as does udp. Packet and raw need fixing.
44 * Alan Cox : Added RCVBUF,SNDBUF size setting. It suddenly occurred to me how easy it was so...
45 * Rick Sladkey : Relaxed UDP rules for matching packets.
46 * C.E.Hawkins : IFF_PROMISC/SIOCGHWADDR support
47 * Pauline Middelink : identd support
48 * Alan Cox : Fixed connect() taking signals I think.
49 * Alan Cox : SO_LINGER supported
50 * Alan Cox : Error reporting fixes
51 * Anonymous : inet_create tidied up (sk->reuse setting)
52 * Alan Cox : inet sockets don't set sk->type!
53 * Alan Cox : Split socket option code
54 * Alan Cox : Callbacks
55 * Alan Cox : Nagle flag for Charles & Johannes stuff
56 * Alex : Removed restriction on inet fioctl
57 * Alan Cox : Splitting INET from NET core
58 * Alan Cox : Fixed bogus SO_TYPE handling in getsockopt()
59 * Adam Caldwell : Missing return in SO_DONTROUTE/SO_DEBUG code
60 * Alan Cox : Split IP from generic code
61 * Alan Cox : New kfree_skbmem()
62 * Alan Cox : Make SO_DEBUG superuser only.
63 * Alan Cox : Allow anyone to clear SO_DEBUG
64 * (compatibility fix)
65 * Alan Cox : Added optimistic memory grabbing for AF_UNIX throughput.
66 * Alan Cox : Allocator for a socket is settable.
67 * Alan Cox : SO_ERROR includes soft errors.
68 * Alan Cox : Allow NULL arguments on some SO_ opts
69 * Alan Cox : Generic socket allocation to make hooks
70 * easier (suggested by Craig Metz).
71 * Michael Pall : SO_ERROR returns positive errno again
72 * Steve Whitehouse: Added default destructor to free
73 * protocol private data.
74 * Steve Whitehouse: Added various other default routines
75 * common to several socket families.
76 * Chris Evans : Call suser() check last on F_SETOWN
77 * Jay Schulist : Added SO_ATTACH_FILTER and SO_DETACH_FILTER.
78 * Andi Kleen : Add sock_kmalloc()/sock_kfree_s()
79 * Andi Kleen : Fix write_space callback
80 * Chris Evans : Security fixes - signedness again
81 * Arnaldo C. Melo : cleanups, use skb_queue_purge
82 *
83 * To Fix:
84 *
85 *
86 * This program is free software; you can redistribute it and/or
87 * modify it under the terms of the GNU General Public License
88 * as published by the Free Software Foundation; either version
89 * 2 of the License, or (at your option) any later version.
90 */
91
Randy Dunlap4fc268d2006-01-11 12:17:47 -080092#include <linux/capability.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070093#include <linux/errno.h>
94#include <linux/types.h>
95#include <linux/socket.h>
96#include <linux/in.h>
97#include <linux/kernel.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070098#include <linux/module.h>
99#include <linux/proc_fs.h>
100#include <linux/seq_file.h>
101#include <linux/sched.h>
102#include <linux/timer.h>
103#include <linux/string.h>
104#include <linux/sockios.h>
105#include <linux/net.h>
106#include <linux/mm.h>
107#include <linux/slab.h>
108#include <linux/interrupt.h>
109#include <linux/poll.h>
110#include <linux/tcp.h>
111#include <linux/init.h>
Al Viroa1f8e7f72006-10-19 16:08:53 -0400112#include <linux/highmem.h>
Eric W. Biederman3f551f92010-06-13 03:28:59 +0000113#include <linux/user_namespace.h>
Glauber Costae1aab162011-12-11 21:47:03 +0000114#include <linux/jump_label.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -0700115
116#include <asm/uaccess.h>
117#include <asm/system.h>
118
119#include <linux/netdevice.h>
120#include <net/protocol.h>
121#include <linux/skbuff.h>
Eric W. Biederman457c4cb2007-09-12 12:01:34 +0200122#include <net/net_namespace.h>
Arnaldo Carvalho de Melo2e6599c2005-06-18 22:46:52 -0700123#include <net/request_sock.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -0700124#include <net/sock.h>
Patrick Ohly20d49472009-02-12 05:03:38 +0000125#include <linux/net_tstamp.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -0700126#include <net/xfrm.h>
127#include <linux/ipsec.h>
Herbert Xuf8451722010-05-24 00:12:34 -0700128#include <net/cls_cgroup.h>
Neil Horman5bc14212011-11-22 05:10:51 +0000129#include <net/netprio_cgroup.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -0700130
131#include <linux/filter.h>
132
Satoru Moriya3847ce32011-06-17 12:00:03 +0000133#include <trace/events/sock.h>
134
Linus Torvalds1da177e2005-04-16 15:20:36 -0700135#ifdef CONFIG_INET
136#include <net/tcp.h>
137#endif
138
Glauber Costa36b77a52011-12-16 00:51:59 +0000139static DEFINE_MUTEX(proto_list_mutex);
Glauber Costad1a4c0b2011-12-11 21:47:04 +0000140static LIST_HEAD(proto_list);
141
142#ifdef CONFIG_CGROUP_MEM_RES_CTLR_KMEM
143int mem_cgroup_sockets_init(struct cgroup *cgrp, struct cgroup_subsys *ss)
144{
145 struct proto *proto;
146 int ret = 0;
147
Glauber Costa36b77a52011-12-16 00:51:59 +0000148 mutex_lock(&proto_list_mutex);
Glauber Costad1a4c0b2011-12-11 21:47:04 +0000149 list_for_each_entry(proto, &proto_list, node) {
150 if (proto->init_cgroup) {
151 ret = proto->init_cgroup(cgrp, ss);
152 if (ret)
153 goto out;
154 }
155 }
156
Glauber Costa36b77a52011-12-16 00:51:59 +0000157 mutex_unlock(&proto_list_mutex);
Glauber Costad1a4c0b2011-12-11 21:47:04 +0000158 return ret;
159out:
160 list_for_each_entry_continue_reverse(proto, &proto_list, node)
161 if (proto->destroy_cgroup)
162 proto->destroy_cgroup(cgrp, ss);
Glauber Costa36b77a52011-12-16 00:51:59 +0000163 mutex_unlock(&proto_list_mutex);
Glauber Costad1a4c0b2011-12-11 21:47:04 +0000164 return ret;
165}
166
167void mem_cgroup_sockets_destroy(struct cgroup *cgrp, struct cgroup_subsys *ss)
168{
169 struct proto *proto;
170
Glauber Costa36b77a52011-12-16 00:51:59 +0000171 mutex_lock(&proto_list_mutex);
Glauber Costad1a4c0b2011-12-11 21:47:04 +0000172 list_for_each_entry_reverse(proto, &proto_list, node)
173 if (proto->destroy_cgroup)
174 proto->destroy_cgroup(cgrp, ss);
Glauber Costa36b77a52011-12-16 00:51:59 +0000175 mutex_unlock(&proto_list_mutex);
Glauber Costad1a4c0b2011-12-11 21:47:04 +0000176}
177#endif
178
Ingo Molnarda21f242006-07-03 00:25:12 -0700179/*
180 * Each address family might have different locking rules, so we have
181 * one slock key per address family:
182 */
Ingo Molnara5b5bb9a2006-07-03 00:25:35 -0700183static struct lock_class_key af_family_keys[AF_MAX];
184static struct lock_class_key af_family_slock_keys[AF_MAX];
185
Glauber Costae1aab162011-12-11 21:47:03 +0000186struct jump_label_key memcg_socket_limit_enabled;
187EXPORT_SYMBOL(memcg_socket_limit_enabled);
188
Ingo Molnara5b5bb9a2006-07-03 00:25:35 -0700189/*
190 * Make lock validator output more readable. (we pre-construct these
191 * strings build-time, so that runtime initialization of socket
192 * locks is fast):
193 */
Jan Engelhardt36cbd3d2009-08-05 10:42:58 -0700194static const char *const af_family_key_strings[AF_MAX+1] = {
Ingo Molnara5b5bb9a2006-07-03 00:25:35 -0700195 "sk_lock-AF_UNSPEC", "sk_lock-AF_UNIX" , "sk_lock-AF_INET" ,
196 "sk_lock-AF_AX25" , "sk_lock-AF_IPX" , "sk_lock-AF_APPLETALK",
197 "sk_lock-AF_NETROM", "sk_lock-AF_BRIDGE" , "sk_lock-AF_ATMPVC" ,
198 "sk_lock-AF_X25" , "sk_lock-AF_INET6" , "sk_lock-AF_ROSE" ,
199 "sk_lock-AF_DECnet", "sk_lock-AF_NETBEUI" , "sk_lock-AF_SECURITY" ,
200 "sk_lock-AF_KEY" , "sk_lock-AF_NETLINK" , "sk_lock-AF_PACKET" ,
201 "sk_lock-AF_ASH" , "sk_lock-AF_ECONET" , "sk_lock-AF_ATMSVC" ,
Andy Grovercbd151b2009-02-26 23:43:19 -0800202 "sk_lock-AF_RDS" , "sk_lock-AF_SNA" , "sk_lock-AF_IRDA" ,
Ingo Molnara5b5bb9a2006-07-03 00:25:35 -0700203 "sk_lock-AF_PPPOX" , "sk_lock-AF_WANPIPE" , "sk_lock-AF_LLC" ,
Oliver Hartkoppcd05acf2007-12-16 15:59:24 -0800204 "sk_lock-27" , "sk_lock-28" , "sk_lock-AF_CAN" ,
David Howells17926a72007-04-26 15:48:28 -0700205 "sk_lock-AF_TIPC" , "sk_lock-AF_BLUETOOTH", "sk_lock-IUCV" ,
Remi Denis-Courmontbce7b152008-09-22 19:51:15 -0700206 "sk_lock-AF_RXRPC" , "sk_lock-AF_ISDN" , "sk_lock-AF_PHONET" ,
Miloslav Trmač6f107b52010-12-08 14:35:34 +0800207 "sk_lock-AF_IEEE802154", "sk_lock-AF_CAIF" , "sk_lock-AF_ALG" ,
Aloisio Almeida Jrc7fe3b52011-07-01 19:31:35 -0300208 "sk_lock-AF_NFC" , "sk_lock-AF_MAX"
Ingo Molnara5b5bb9a2006-07-03 00:25:35 -0700209};
Jan Engelhardt36cbd3d2009-08-05 10:42:58 -0700210static const char *const af_family_slock_key_strings[AF_MAX+1] = {
Ingo Molnara5b5bb9a2006-07-03 00:25:35 -0700211 "slock-AF_UNSPEC", "slock-AF_UNIX" , "slock-AF_INET" ,
212 "slock-AF_AX25" , "slock-AF_IPX" , "slock-AF_APPLETALK",
213 "slock-AF_NETROM", "slock-AF_BRIDGE" , "slock-AF_ATMPVC" ,
214 "slock-AF_X25" , "slock-AF_INET6" , "slock-AF_ROSE" ,
215 "slock-AF_DECnet", "slock-AF_NETBEUI" , "slock-AF_SECURITY" ,
216 "slock-AF_KEY" , "slock-AF_NETLINK" , "slock-AF_PACKET" ,
217 "slock-AF_ASH" , "slock-AF_ECONET" , "slock-AF_ATMSVC" ,
Andy Grovercbd151b2009-02-26 23:43:19 -0800218 "slock-AF_RDS" , "slock-AF_SNA" , "slock-AF_IRDA" ,
Ingo Molnara5b5bb9a2006-07-03 00:25:35 -0700219 "slock-AF_PPPOX" , "slock-AF_WANPIPE" , "slock-AF_LLC" ,
Oliver Hartkoppcd05acf2007-12-16 15:59:24 -0800220 "slock-27" , "slock-28" , "slock-AF_CAN" ,
David Howells17926a72007-04-26 15:48:28 -0700221 "slock-AF_TIPC" , "slock-AF_BLUETOOTH", "slock-AF_IUCV" ,
Remi Denis-Courmontbce7b152008-09-22 19:51:15 -0700222 "slock-AF_RXRPC" , "slock-AF_ISDN" , "slock-AF_PHONET" ,
Miloslav Trmač6f107b52010-12-08 14:35:34 +0800223 "slock-AF_IEEE802154", "slock-AF_CAIF" , "slock-AF_ALG" ,
Aloisio Almeida Jrc7fe3b52011-07-01 19:31:35 -0300224 "slock-AF_NFC" , "slock-AF_MAX"
Ingo Molnara5b5bb9a2006-07-03 00:25:35 -0700225};
Jan Engelhardt36cbd3d2009-08-05 10:42:58 -0700226static const char *const af_family_clock_key_strings[AF_MAX+1] = {
Peter Zijlstra443aef0e2007-07-19 01:49:00 -0700227 "clock-AF_UNSPEC", "clock-AF_UNIX" , "clock-AF_INET" ,
228 "clock-AF_AX25" , "clock-AF_IPX" , "clock-AF_APPLETALK",
229 "clock-AF_NETROM", "clock-AF_BRIDGE" , "clock-AF_ATMPVC" ,
230 "clock-AF_X25" , "clock-AF_INET6" , "clock-AF_ROSE" ,
231 "clock-AF_DECnet", "clock-AF_NETBEUI" , "clock-AF_SECURITY" ,
232 "clock-AF_KEY" , "clock-AF_NETLINK" , "clock-AF_PACKET" ,
233 "clock-AF_ASH" , "clock-AF_ECONET" , "clock-AF_ATMSVC" ,
Andy Grovercbd151b2009-02-26 23:43:19 -0800234 "clock-AF_RDS" , "clock-AF_SNA" , "clock-AF_IRDA" ,
Peter Zijlstra443aef0e2007-07-19 01:49:00 -0700235 "clock-AF_PPPOX" , "clock-AF_WANPIPE" , "clock-AF_LLC" ,
Oliver Hartkoppb4942af2008-07-23 14:06:04 -0700236 "clock-27" , "clock-28" , "clock-AF_CAN" ,
David Howellse51f8022007-07-21 19:30:16 -0700237 "clock-AF_TIPC" , "clock-AF_BLUETOOTH", "clock-AF_IUCV" ,
Remi Denis-Courmontbce7b152008-09-22 19:51:15 -0700238 "clock-AF_RXRPC" , "clock-AF_ISDN" , "clock-AF_PHONET" ,
Miloslav Trmač6f107b52010-12-08 14:35:34 +0800239 "clock-AF_IEEE802154", "clock-AF_CAIF" , "clock-AF_ALG" ,
Aloisio Almeida Jrc7fe3b52011-07-01 19:31:35 -0300240 "clock-AF_NFC" , "clock-AF_MAX"
Peter Zijlstra443aef0e2007-07-19 01:49:00 -0700241};
Ingo Molnarda21f242006-07-03 00:25:12 -0700242
243/*
244 * sk_callback_lock locking rules are per-address-family,
245 * so split the lock classes by using a per-AF key:
246 */
247static struct lock_class_key af_callback_keys[AF_MAX];
248
Linus Torvalds1da177e2005-04-16 15:20:36 -0700249/* Take into consideration the size of the struct sk_buff overhead in the
250 * determination of these values, since that is non-constant across
251 * platforms. This makes socket queueing behavior and performance
252 * not depend upon such differences.
253 */
254#define _SK_MEM_PACKETS 256
Eric Dumazet87fb4b72011-10-13 07:28:54 +0000255#define _SK_MEM_OVERHEAD SKB_TRUESIZE(256)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700256#define SK_WMEM_MAX (_SK_MEM_OVERHEAD * _SK_MEM_PACKETS)
257#define SK_RMEM_MAX (_SK_MEM_OVERHEAD * _SK_MEM_PACKETS)
258
259/* Run time adjustable parameters. */
Brian Haleyab32ea52006-09-22 14:15:41 -0700260__u32 sysctl_wmem_max __read_mostly = SK_WMEM_MAX;
261__u32 sysctl_rmem_max __read_mostly = SK_RMEM_MAX;
262__u32 sysctl_wmem_default __read_mostly = SK_WMEM_MAX;
263__u32 sysctl_rmem_default __read_mostly = SK_RMEM_MAX;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700264
Lucas De Marchi25985ed2011-03-30 22:57:33 -0300265/* Maximal space eaten by iovec or ancillary data plus some space */
Brian Haleyab32ea52006-09-22 14:15:41 -0700266int sysctl_optmem_max __read_mostly = sizeof(unsigned long)*(2*UIO_MAXIOV+512);
Eric Dumazet2a915252009-05-27 11:30:05 +0000267EXPORT_SYMBOL(sysctl_optmem_max);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700268
Neil Horman5bc14212011-11-22 05:10:51 +0000269#if defined(CONFIG_CGROUPS)
270#if !defined(CONFIG_NET_CLS_CGROUP)
Herbert Xuf8451722010-05-24 00:12:34 -0700271int net_cls_subsys_id = -1;
272EXPORT_SYMBOL_GPL(net_cls_subsys_id);
273#endif
Neil Horman5bc14212011-11-22 05:10:51 +0000274#if !defined(CONFIG_NETPRIO_CGROUP)
275int net_prio_subsys_id = -1;
276EXPORT_SYMBOL_GPL(net_prio_subsys_id);
277#endif
278#endif
Herbert Xuf8451722010-05-24 00:12:34 -0700279
Linus Torvalds1da177e2005-04-16 15:20:36 -0700280static int sock_set_timeout(long *timeo_p, char __user *optval, int optlen)
281{
282 struct timeval tv;
283
284 if (optlen < sizeof(tv))
285 return -EINVAL;
286 if (copy_from_user(&tv, optval, sizeof(tv)))
287 return -EFAULT;
Vasily Averinba780732007-05-24 16:58:54 -0700288 if (tv.tv_usec < 0 || tv.tv_usec >= USEC_PER_SEC)
289 return -EDOM;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700290
Vasily Averinba780732007-05-24 16:58:54 -0700291 if (tv.tv_sec < 0) {
Andrew Morton6f11df82007-07-09 13:16:00 -0700292 static int warned __read_mostly;
293
Vasily Averinba780732007-05-24 16:58:54 -0700294 *timeo_p = 0;
Ilpo Järvinen50aab542008-05-02 16:20:10 -0700295 if (warned < 10 && net_ratelimit()) {
Vasily Averinba780732007-05-24 16:58:54 -0700296 warned++;
297 printk(KERN_INFO "sock_set_timeout: `%s' (pid %d) "
298 "tries to set negative timeout\n",
Pavel Emelyanovba25f9d2007-10-18 23:40:40 -0700299 current->comm, task_pid_nr(current));
Ilpo Järvinen50aab542008-05-02 16:20:10 -0700300 }
Vasily Averinba780732007-05-24 16:58:54 -0700301 return 0;
302 }
Linus Torvalds1da177e2005-04-16 15:20:36 -0700303 *timeo_p = MAX_SCHEDULE_TIMEOUT;
304 if (tv.tv_sec == 0 && tv.tv_usec == 0)
305 return 0;
306 if (tv.tv_sec < (MAX_SCHEDULE_TIMEOUT/HZ - 1))
307 *timeo_p = tv.tv_sec*HZ + (tv.tv_usec+(1000000/HZ-1))/(1000000/HZ);
308 return 0;
309}
310
311static void sock_warn_obsolete_bsdism(const char *name)
312{
313 static int warned;
314 static char warncomm[TASK_COMM_LEN];
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +0900315 if (strcmp(warncomm, current->comm) && warned < 5) {
316 strcpy(warncomm, current->comm);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700317 printk(KERN_WARNING "process `%s' is using obsolete "
318 "%s SO_BSDCOMPAT\n", warncomm, name);
319 warned++;
320 }
321}
322
Eric Dumazet08e29af2011-11-28 12:04:18 +0000323#define SK_FLAGS_TIMESTAMP ((1UL << SOCK_TIMESTAMP) | (1UL << SOCK_TIMESTAMPING_RX_SOFTWARE))
324
325static void sock_disable_timestamp(struct sock *sk, unsigned long flags)
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +0900326{
Eric Dumazet08e29af2011-11-28 12:04:18 +0000327 if (sk->sk_flags & flags) {
328 sk->sk_flags &= ~flags;
329 if (!(sk->sk_flags & SK_FLAGS_TIMESTAMP))
Patrick Ohly20d49472009-02-12 05:03:38 +0000330 net_disable_timestamp();
Linus Torvalds1da177e2005-04-16 15:20:36 -0700331 }
332}
333
334
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800335int sock_queue_rcv_skb(struct sock *sk, struct sk_buff *skb)
336{
Eric Dumazet766e90372009-10-14 20:40:11 -0700337 int err;
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800338 int skb_len;
Neil Horman3b885782009-10-12 13:26:31 -0700339 unsigned long flags;
340 struct sk_buff_head *list = &sk->sk_receive_queue;
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800341
Eric Dumazet0fd7bac2011-12-21 07:11:44 +0000342 if (atomic_read(&sk->sk_rmem_alloc) >= sk->sk_rcvbuf) {
Eric Dumazet766e90372009-10-14 20:40:11 -0700343 atomic_inc(&sk->sk_drops);
Satoru Moriya3847ce32011-06-17 12:00:03 +0000344 trace_sock_rcvqueue_full(sk, skb);
Eric Dumazet766e90372009-10-14 20:40:11 -0700345 return -ENOMEM;
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800346 }
347
Dmitry Mishinfda9ef52006-08-31 15:28:39 -0700348 err = sk_filter(sk, skb);
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800349 if (err)
Eric Dumazet766e90372009-10-14 20:40:11 -0700350 return err;
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800351
Hideo Aoki3ab224b2007-12-31 00:11:19 -0800352 if (!sk_rmem_schedule(sk, skb->truesize)) {
Eric Dumazet766e90372009-10-14 20:40:11 -0700353 atomic_inc(&sk->sk_drops);
354 return -ENOBUFS;
Hideo Aoki3ab224b2007-12-31 00:11:19 -0800355 }
356
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800357 skb->dev = NULL;
358 skb_set_owner_r(skb, sk);
David S. Miller49ad9592008-12-17 22:11:38 -0800359
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800360 /* Cache the SKB length before we tack it onto the receive
361 * queue. Once it is added it no longer belongs to us and
362 * may be freed by other threads of control pulling packets
363 * from the queue.
364 */
365 skb_len = skb->len;
366
Eric Dumazet7fee2262010-05-11 23:19:48 +0000367 /* we escape from rcu protected region, make sure we dont leak
368 * a norefcounted dst
369 */
370 skb_dst_force(skb);
371
Neil Horman3b885782009-10-12 13:26:31 -0700372 spin_lock_irqsave(&list->lock, flags);
373 skb->dropcount = atomic_read(&sk->sk_drops);
374 __skb_queue_tail(list, skb);
375 spin_unlock_irqrestore(&list->lock, flags);
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800376
377 if (!sock_flag(sk, SOCK_DEAD))
378 sk->sk_data_ready(sk, skb_len);
Eric Dumazet766e90372009-10-14 20:40:11 -0700379 return 0;
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800380}
381EXPORT_SYMBOL(sock_queue_rcv_skb);
382
Arnaldo Carvalho de Melo58a5a7b2006-11-16 14:06:06 -0200383int sk_receive_skb(struct sock *sk, struct sk_buff *skb, const int nested)
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800384{
385 int rc = NET_RX_SUCCESS;
386
Dmitry Mishinfda9ef52006-08-31 15:28:39 -0700387 if (sk_filter(sk, skb))
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800388 goto discard_and_relse;
389
390 skb->dev = NULL;
391
Eric Dumazetc3774112010-04-27 15:13:20 -0700392 if (sk_rcvqueues_full(sk, skb)) {
393 atomic_inc(&sk->sk_drops);
394 goto discard_and_relse;
395 }
Arnaldo Carvalho de Melo58a5a7b2006-11-16 14:06:06 -0200396 if (nested)
397 bh_lock_sock_nested(sk);
398 else
399 bh_lock_sock(sk);
Ingo Molnara5b5bb9a2006-07-03 00:25:35 -0700400 if (!sock_owned_by_user(sk)) {
401 /*
402 * trylock + unlock semantics:
403 */
404 mutex_acquire(&sk->sk_lock.dep_map, 0, 1, _RET_IP_);
405
Peter Zijlstrac57943a2008-10-07 14:18:42 -0700406 rc = sk_backlog_rcv(sk, skb);
Ingo Molnara5b5bb9a2006-07-03 00:25:35 -0700407
408 mutex_release(&sk->sk_lock.dep_map, 1, _RET_IP_);
Zhu Yia3a858f2010-03-04 18:01:47 +0000409 } else if (sk_add_backlog(sk, skb)) {
Zhu Yi8eae9392010-03-04 18:01:40 +0000410 bh_unlock_sock(sk);
411 atomic_inc(&sk->sk_drops);
412 goto discard_and_relse;
413 }
414
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800415 bh_unlock_sock(sk);
416out:
417 sock_put(sk);
418 return rc;
419discard_and_relse:
420 kfree_skb(skb);
421 goto out;
422}
423EXPORT_SYMBOL(sk_receive_skb);
424
Krishna Kumarea94ff32009-10-19 23:46:45 +0000425void sk_reset_txq(struct sock *sk)
426{
427 sk_tx_queue_clear(sk);
428}
429EXPORT_SYMBOL(sk_reset_txq);
430
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800431struct dst_entry *__sk_dst_check(struct sock *sk, u32 cookie)
432{
Eric Dumazetb6c67122010-04-08 23:03:29 +0000433 struct dst_entry *dst = __sk_dst_get(sk);
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800434
435 if (dst && dst->obsolete && dst->ops->check(dst, cookie) == NULL) {
Krishna Kumare022f0b2009-10-19 23:46:20 +0000436 sk_tx_queue_clear(sk);
Stephen Hemmingera9b3cd72011-08-01 16:19:00 +0000437 RCU_INIT_POINTER(sk->sk_dst_cache, NULL);
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800438 dst_release(dst);
439 return NULL;
440 }
441
442 return dst;
443}
444EXPORT_SYMBOL(__sk_dst_check);
445
446struct dst_entry *sk_dst_check(struct sock *sk, u32 cookie)
447{
448 struct dst_entry *dst = sk_dst_get(sk);
449
450 if (dst && dst->obsolete && dst->ops->check(dst, cookie) == NULL) {
451 sk_dst_reset(sk);
452 dst_release(dst);
453 return NULL;
454 }
455
456 return dst;
457}
458EXPORT_SYMBOL(sk_dst_check);
459
David S. Miller48788092007-09-14 16:41:03 -0700460static int sock_bindtodevice(struct sock *sk, char __user *optval, int optlen)
461{
462 int ret = -ENOPROTOOPT;
463#ifdef CONFIG_NETDEVICES
YOSHIFUJI Hideaki3b1e0a62008-03-26 02:26:21 +0900464 struct net *net = sock_net(sk);
David S. Miller48788092007-09-14 16:41:03 -0700465 char devname[IFNAMSIZ];
466 int index;
467
468 /* Sorry... */
469 ret = -EPERM;
470 if (!capable(CAP_NET_RAW))
471 goto out;
472
473 ret = -EINVAL;
474 if (optlen < 0)
475 goto out;
476
477 /* Bind this socket to a particular device like "eth0",
478 * as specified in the passed interface name. If the
479 * name is "" or the option length is zero the socket
480 * is not bound.
481 */
482 if (optlen > IFNAMSIZ - 1)
483 optlen = IFNAMSIZ - 1;
484 memset(devname, 0, sizeof(devname));
485
486 ret = -EFAULT;
487 if (copy_from_user(devname, optval, optlen))
488 goto out;
489
David S. Miller000ba2e2009-11-05 22:37:11 -0800490 index = 0;
491 if (devname[0] != '\0') {
Eric Dumazetbf8e56b2009-11-05 21:03:39 -0800492 struct net_device *dev;
David S. Miller48788092007-09-14 16:41:03 -0700493
Eric Dumazetbf8e56b2009-11-05 21:03:39 -0800494 rcu_read_lock();
495 dev = dev_get_by_name_rcu(net, devname);
496 if (dev)
497 index = dev->ifindex;
498 rcu_read_unlock();
David S. Miller48788092007-09-14 16:41:03 -0700499 ret = -ENODEV;
500 if (!dev)
501 goto out;
David S. Miller48788092007-09-14 16:41:03 -0700502 }
503
504 lock_sock(sk);
505 sk->sk_bound_dev_if = index;
506 sk_dst_reset(sk);
507 release_sock(sk);
508
509 ret = 0;
510
511out:
512#endif
513
514 return ret;
515}
516
Pavel Emelyanovc0ef8772007-11-15 03:03:19 -0800517static inline void sock_valbool_flag(struct sock *sk, int bit, int valbool)
518{
519 if (valbool)
520 sock_set_flag(sk, bit);
521 else
522 sock_reset_flag(sk, bit);
523}
524
Linus Torvalds1da177e2005-04-16 15:20:36 -0700525/*
526 * This is meant for all protocols to use and covers goings on
527 * at the socket level. Everything here is generic.
528 */
529
530int sock_setsockopt(struct socket *sock, int level, int optname,
David S. Millerb7058842009-09-30 16:12:20 -0700531 char __user *optval, unsigned int optlen)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700532{
Eric Dumazet2a915252009-05-27 11:30:05 +0000533 struct sock *sk = sock->sk;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700534 int val;
535 int valbool;
536 struct linger ling;
537 int ret = 0;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +0900538
Linus Torvalds1da177e2005-04-16 15:20:36 -0700539 /*
540 * Options without arguments
541 */
542
David S. Miller48788092007-09-14 16:41:03 -0700543 if (optname == SO_BINDTODEVICE)
544 return sock_bindtodevice(sk, optval, optlen);
545
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700546 if (optlen < sizeof(int))
547 return -EINVAL;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +0900548
Linus Torvalds1da177e2005-04-16 15:20:36 -0700549 if (get_user(val, (int __user *)optval))
550 return -EFAULT;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +0900551
Eric Dumazet2a915252009-05-27 11:30:05 +0000552 valbool = val ? 1 : 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700553
554 lock_sock(sk);
555
Eric Dumazet2a915252009-05-27 11:30:05 +0000556 switch (optname) {
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700557 case SO_DEBUG:
Eric Dumazet2a915252009-05-27 11:30:05 +0000558 if (val && !capable(CAP_NET_ADMIN))
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700559 ret = -EACCES;
Eric Dumazet2a915252009-05-27 11:30:05 +0000560 else
Pavel Emelyanovc0ef8772007-11-15 03:03:19 -0800561 sock_valbool_flag(sk, SOCK_DBG, valbool);
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700562 break;
563 case SO_REUSEADDR:
564 sk->sk_reuse = valbool;
565 break;
566 case SO_TYPE:
Jan Engelhardt49c794e2009-08-04 07:28:28 +0000567 case SO_PROTOCOL:
Jan Engelhardt0d6038e2009-08-04 07:28:29 +0000568 case SO_DOMAIN:
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700569 case SO_ERROR:
570 ret = -ENOPROTOOPT;
571 break;
572 case SO_DONTROUTE:
Pavel Emelyanovc0ef8772007-11-15 03:03:19 -0800573 sock_valbool_flag(sk, SOCK_LOCALROUTE, valbool);
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700574 break;
575 case SO_BROADCAST:
576 sock_valbool_flag(sk, SOCK_BROADCAST, valbool);
577 break;
578 case SO_SNDBUF:
579 /* Don't error on this BSD doesn't and if you think
580 about it this is right. Otherwise apps have to
581 play 'guess the biggest size' games. RCVBUF/SNDBUF
582 are treated in BSD as hints */
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +0900583
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700584 if (val > sysctl_wmem_max)
585 val = sysctl_wmem_max;
Patrick McHardyb0573de2005-08-09 19:30:51 -0700586set_sndbuf:
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700587 sk->sk_userlocks |= SOCK_SNDBUF_LOCK;
588 if ((val * 2) < SOCK_MIN_SNDBUF)
589 sk->sk_sndbuf = SOCK_MIN_SNDBUF;
590 else
591 sk->sk_sndbuf = val * 2;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700592
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700593 /*
594 * Wake up sending tasks if we
595 * upped the value.
596 */
597 sk->sk_write_space(sk);
598 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700599
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700600 case SO_SNDBUFFORCE:
601 if (!capable(CAP_NET_ADMIN)) {
602 ret = -EPERM;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700603 break;
604 }
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700605 goto set_sndbuf;
606
607 case SO_RCVBUF:
608 /* Don't error on this BSD doesn't and if you think
609 about it this is right. Otherwise apps have to
610 play 'guess the biggest size' games. RCVBUF/SNDBUF
611 are treated in BSD as hints */
612
613 if (val > sysctl_rmem_max)
614 val = sysctl_rmem_max;
615set_rcvbuf:
616 sk->sk_userlocks |= SOCK_RCVBUF_LOCK;
617 /*
618 * We double it on the way in to account for
619 * "struct sk_buff" etc. overhead. Applications
620 * assume that the SO_RCVBUF setting they make will
621 * allow that much actual data to be received on that
622 * socket.
623 *
624 * Applications are unaware that "struct sk_buff" and
625 * other overheads allocate from the receive buffer
626 * during socket buffer allocation.
627 *
628 * And after considering the possible alternatives,
629 * returning the value we actually used in getsockopt
630 * is the most desirable behavior.
631 */
632 if ((val * 2) < SOCK_MIN_RCVBUF)
633 sk->sk_rcvbuf = SOCK_MIN_RCVBUF;
634 else
635 sk->sk_rcvbuf = val * 2;
636 break;
637
638 case SO_RCVBUFFORCE:
639 if (!capable(CAP_NET_ADMIN)) {
640 ret = -EPERM;
641 break;
642 }
643 goto set_rcvbuf;
644
645 case SO_KEEPALIVE:
646#ifdef CONFIG_INET
647 if (sk->sk_protocol == IPPROTO_TCP)
648 tcp_set_keepalive(sk, valbool);
649#endif
650 sock_valbool_flag(sk, SOCK_KEEPOPEN, valbool);
651 break;
652
653 case SO_OOBINLINE:
654 sock_valbool_flag(sk, SOCK_URGINLINE, valbool);
655 break;
656
657 case SO_NO_CHECK:
658 sk->sk_no_check = valbool;
659 break;
660
661 case SO_PRIORITY:
662 if ((val >= 0 && val <= 6) || capable(CAP_NET_ADMIN))
663 sk->sk_priority = val;
664 else
665 ret = -EPERM;
666 break;
667
668 case SO_LINGER:
669 if (optlen < sizeof(ling)) {
670 ret = -EINVAL; /* 1003.1g */
671 break;
672 }
Eric Dumazet2a915252009-05-27 11:30:05 +0000673 if (copy_from_user(&ling, optval, sizeof(ling))) {
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700674 ret = -EFAULT;
675 break;
676 }
677 if (!ling.l_onoff)
678 sock_reset_flag(sk, SOCK_LINGER);
679 else {
680#if (BITS_PER_LONG == 32)
681 if ((unsigned int)ling.l_linger >= MAX_SCHEDULE_TIMEOUT/HZ)
682 sk->sk_lingertime = MAX_SCHEDULE_TIMEOUT;
683 else
684#endif
685 sk->sk_lingertime = (unsigned int)ling.l_linger * HZ;
686 sock_set_flag(sk, SOCK_LINGER);
687 }
688 break;
689
690 case SO_BSDCOMPAT:
691 sock_warn_obsolete_bsdism("setsockopt");
692 break;
693
694 case SO_PASSCRED:
695 if (valbool)
696 set_bit(SOCK_PASSCRED, &sock->flags);
697 else
698 clear_bit(SOCK_PASSCRED, &sock->flags);
699 break;
700
701 case SO_TIMESTAMP:
Eric Dumazet92f37fd2007-03-25 22:14:49 -0700702 case SO_TIMESTAMPNS:
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700703 if (valbool) {
Eric Dumazet92f37fd2007-03-25 22:14:49 -0700704 if (optname == SO_TIMESTAMP)
705 sock_reset_flag(sk, SOCK_RCVTSTAMPNS);
706 else
707 sock_set_flag(sk, SOCK_RCVTSTAMPNS);
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700708 sock_set_flag(sk, SOCK_RCVTSTAMP);
Patrick Ohly20d49472009-02-12 05:03:38 +0000709 sock_enable_timestamp(sk, SOCK_TIMESTAMP);
Eric Dumazet92f37fd2007-03-25 22:14:49 -0700710 } else {
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700711 sock_reset_flag(sk, SOCK_RCVTSTAMP);
Eric Dumazet92f37fd2007-03-25 22:14:49 -0700712 sock_reset_flag(sk, SOCK_RCVTSTAMPNS);
713 }
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700714 break;
715
Patrick Ohly20d49472009-02-12 05:03:38 +0000716 case SO_TIMESTAMPING:
717 if (val & ~SOF_TIMESTAMPING_MASK) {
Rémi Denis-Courmontf249fb72009-07-20 00:47:04 +0000718 ret = -EINVAL;
Patrick Ohly20d49472009-02-12 05:03:38 +0000719 break;
720 }
721 sock_valbool_flag(sk, SOCK_TIMESTAMPING_TX_HARDWARE,
722 val & SOF_TIMESTAMPING_TX_HARDWARE);
723 sock_valbool_flag(sk, SOCK_TIMESTAMPING_TX_SOFTWARE,
724 val & SOF_TIMESTAMPING_TX_SOFTWARE);
725 sock_valbool_flag(sk, SOCK_TIMESTAMPING_RX_HARDWARE,
726 val & SOF_TIMESTAMPING_RX_HARDWARE);
727 if (val & SOF_TIMESTAMPING_RX_SOFTWARE)
728 sock_enable_timestamp(sk,
729 SOCK_TIMESTAMPING_RX_SOFTWARE);
730 else
731 sock_disable_timestamp(sk,
Eric Dumazet08e29af2011-11-28 12:04:18 +0000732 (1UL << SOCK_TIMESTAMPING_RX_SOFTWARE));
Patrick Ohly20d49472009-02-12 05:03:38 +0000733 sock_valbool_flag(sk, SOCK_TIMESTAMPING_SOFTWARE,
734 val & SOF_TIMESTAMPING_SOFTWARE);
735 sock_valbool_flag(sk, SOCK_TIMESTAMPING_SYS_HARDWARE,
736 val & SOF_TIMESTAMPING_SYS_HARDWARE);
737 sock_valbool_flag(sk, SOCK_TIMESTAMPING_RAW_HARDWARE,
738 val & SOF_TIMESTAMPING_RAW_HARDWARE);
739 break;
740
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700741 case SO_RCVLOWAT:
742 if (val < 0)
743 val = INT_MAX;
744 sk->sk_rcvlowat = val ? : 1;
745 break;
746
747 case SO_RCVTIMEO:
748 ret = sock_set_timeout(&sk->sk_rcvtimeo, optval, optlen);
749 break;
750
751 case SO_SNDTIMEO:
752 ret = sock_set_timeout(&sk->sk_sndtimeo, optval, optlen);
753 break;
754
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700755 case SO_ATTACH_FILTER:
756 ret = -EINVAL;
757 if (optlen == sizeof(struct sock_fprog)) {
758 struct sock_fprog fprog;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700759
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700760 ret = -EFAULT;
761 if (copy_from_user(&fprog, optval, sizeof(fprog)))
Linus Torvalds1da177e2005-04-16 15:20:36 -0700762 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700763
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700764 ret = sk_attach_filter(&fprog, sk);
765 }
766 break;
767
768 case SO_DETACH_FILTER:
Pavel Emelyanov55b33322007-10-17 21:21:26 -0700769 ret = sk_detach_filter(sk);
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700770 break;
771
772 case SO_PASSSEC:
773 if (valbool)
774 set_bit(SOCK_PASSSEC, &sock->flags);
775 else
776 clear_bit(SOCK_PASSSEC, &sock->flags);
777 break;
Laszlo Attila Toth4a19ec52008-01-30 19:08:16 -0800778 case SO_MARK:
779 if (!capable(CAP_NET_ADMIN))
780 ret = -EPERM;
Eric Dumazet2a915252009-05-27 11:30:05 +0000781 else
Laszlo Attila Toth4a19ec52008-01-30 19:08:16 -0800782 sk->sk_mark = val;
Laszlo Attila Toth4a19ec52008-01-30 19:08:16 -0800783 break;
Catherine Zhang877ce7c2006-06-29 12:27:47 -0700784
Linus Torvalds1da177e2005-04-16 15:20:36 -0700785 /* We implement the SO_SNDLOWAT etc to
786 not be settable (1003.1g 5.3) */
Neil Horman3b885782009-10-12 13:26:31 -0700787 case SO_RXQ_OVFL:
Johannes Berg8083f0f2011-10-07 03:30:20 +0000788 sock_valbool_flag(sk, SOCK_RXQ_OVFL, valbool);
Neil Horman3b885782009-10-12 13:26:31 -0700789 break;
Johannes Berg6e3e9392011-11-09 10:15:42 +0100790
791 case SO_WIFI_STATUS:
792 sock_valbool_flag(sk, SOCK_WIFI_STATUS, valbool);
793 break;
794
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700795 default:
796 ret = -ENOPROTOOPT;
797 break;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +0900798 }
Linus Torvalds1da177e2005-04-16 15:20:36 -0700799 release_sock(sk);
800 return ret;
801}
Eric Dumazet2a915252009-05-27 11:30:05 +0000802EXPORT_SYMBOL(sock_setsockopt);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700803
804
Eric W. Biederman3f551f92010-06-13 03:28:59 +0000805void cred_to_ucred(struct pid *pid, const struct cred *cred,
806 struct ucred *ucred)
807{
808 ucred->pid = pid_vnr(pid);
809 ucred->uid = ucred->gid = -1;
810 if (cred) {
811 struct user_namespace *current_ns = current_user_ns();
812
813 ucred->uid = user_ns_map_uid(current_ns, cred, cred->euid);
814 ucred->gid = user_ns_map_gid(current_ns, cred, cred->egid);
815 }
816}
David S. Miller39247732010-06-16 16:18:25 -0700817EXPORT_SYMBOL_GPL(cred_to_ucred);
Eric W. Biederman3f551f92010-06-13 03:28:59 +0000818
Linus Torvalds1da177e2005-04-16 15:20:36 -0700819int sock_getsockopt(struct socket *sock, int level, int optname,
820 char __user *optval, int __user *optlen)
821{
822 struct sock *sk = sock->sk;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +0900823
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700824 union {
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +0900825 int val;
826 struct linger ling;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700827 struct timeval tm;
828 } v;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +0900829
H Hartley Sweeten4d0392b2010-01-15 01:08:58 -0800830 int lv = sizeof(int);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700831 int len;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +0900832
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700833 if (get_user(len, optlen))
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +0900834 return -EFAULT;
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700835 if (len < 0)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700836 return -EINVAL;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +0900837
Eugene Teo50fee1d2009-02-23 15:38:41 -0800838 memset(&v, 0, sizeof(v));
Clément Lecignedf0bca02009-02-12 16:59:09 -0800839
Eric Dumazet2a915252009-05-27 11:30:05 +0000840 switch (optname) {
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700841 case SO_DEBUG:
842 v.val = sock_flag(sk, SOCK_DBG);
843 break;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +0900844
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700845 case SO_DONTROUTE:
846 v.val = sock_flag(sk, SOCK_LOCALROUTE);
847 break;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +0900848
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700849 case SO_BROADCAST:
850 v.val = !!sock_flag(sk, SOCK_BROADCAST);
851 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700852
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700853 case SO_SNDBUF:
854 v.val = sk->sk_sndbuf;
855 break;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +0900856
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700857 case SO_RCVBUF:
858 v.val = sk->sk_rcvbuf;
859 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700860
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700861 case SO_REUSEADDR:
862 v.val = sk->sk_reuse;
863 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700864
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700865 case SO_KEEPALIVE:
866 v.val = !!sock_flag(sk, SOCK_KEEPOPEN);
867 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700868
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700869 case SO_TYPE:
870 v.val = sk->sk_type;
871 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700872
Jan Engelhardt49c794e2009-08-04 07:28:28 +0000873 case SO_PROTOCOL:
874 v.val = sk->sk_protocol;
875 break;
876
Jan Engelhardt0d6038e2009-08-04 07:28:29 +0000877 case SO_DOMAIN:
878 v.val = sk->sk_family;
879 break;
880
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700881 case SO_ERROR:
882 v.val = -sock_error(sk);
Eric Dumazet2a915252009-05-27 11:30:05 +0000883 if (v.val == 0)
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700884 v.val = xchg(&sk->sk_err_soft, 0);
885 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700886
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700887 case SO_OOBINLINE:
888 v.val = !!sock_flag(sk, SOCK_URGINLINE);
889 break;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +0900890
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700891 case SO_NO_CHECK:
892 v.val = sk->sk_no_check;
893 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700894
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700895 case SO_PRIORITY:
896 v.val = sk->sk_priority;
897 break;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +0900898
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700899 case SO_LINGER:
900 lv = sizeof(v.ling);
901 v.ling.l_onoff = !!sock_flag(sk, SOCK_LINGER);
902 v.ling.l_linger = sk->sk_lingertime / HZ;
903 break;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +0900904
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700905 case SO_BSDCOMPAT:
906 sock_warn_obsolete_bsdism("getsockopt");
907 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700908
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700909 case SO_TIMESTAMP:
Eric Dumazet92f37fd2007-03-25 22:14:49 -0700910 v.val = sock_flag(sk, SOCK_RCVTSTAMP) &&
911 !sock_flag(sk, SOCK_RCVTSTAMPNS);
912 break;
913
914 case SO_TIMESTAMPNS:
915 v.val = sock_flag(sk, SOCK_RCVTSTAMPNS);
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700916 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700917
Patrick Ohly20d49472009-02-12 05:03:38 +0000918 case SO_TIMESTAMPING:
919 v.val = 0;
920 if (sock_flag(sk, SOCK_TIMESTAMPING_TX_HARDWARE))
921 v.val |= SOF_TIMESTAMPING_TX_HARDWARE;
922 if (sock_flag(sk, SOCK_TIMESTAMPING_TX_SOFTWARE))
923 v.val |= SOF_TIMESTAMPING_TX_SOFTWARE;
924 if (sock_flag(sk, SOCK_TIMESTAMPING_RX_HARDWARE))
925 v.val |= SOF_TIMESTAMPING_RX_HARDWARE;
926 if (sock_flag(sk, SOCK_TIMESTAMPING_RX_SOFTWARE))
927 v.val |= SOF_TIMESTAMPING_RX_SOFTWARE;
928 if (sock_flag(sk, SOCK_TIMESTAMPING_SOFTWARE))
929 v.val |= SOF_TIMESTAMPING_SOFTWARE;
930 if (sock_flag(sk, SOCK_TIMESTAMPING_SYS_HARDWARE))
931 v.val |= SOF_TIMESTAMPING_SYS_HARDWARE;
932 if (sock_flag(sk, SOCK_TIMESTAMPING_RAW_HARDWARE))
933 v.val |= SOF_TIMESTAMPING_RAW_HARDWARE;
934 break;
935
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700936 case SO_RCVTIMEO:
Eric Dumazet2a915252009-05-27 11:30:05 +0000937 lv = sizeof(struct timeval);
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700938 if (sk->sk_rcvtimeo == MAX_SCHEDULE_TIMEOUT) {
939 v.tm.tv_sec = 0;
940 v.tm.tv_usec = 0;
941 } else {
942 v.tm.tv_sec = sk->sk_rcvtimeo / HZ;
943 v.tm.tv_usec = ((sk->sk_rcvtimeo % HZ) * 1000000) / HZ;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700944 }
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700945 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700946
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700947 case SO_SNDTIMEO:
Eric Dumazet2a915252009-05-27 11:30:05 +0000948 lv = sizeof(struct timeval);
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700949 if (sk->sk_sndtimeo == MAX_SCHEDULE_TIMEOUT) {
950 v.tm.tv_sec = 0;
951 v.tm.tv_usec = 0;
952 } else {
953 v.tm.tv_sec = sk->sk_sndtimeo / HZ;
954 v.tm.tv_usec = ((sk->sk_sndtimeo % HZ) * 1000000) / HZ;
955 }
956 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700957
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700958 case SO_RCVLOWAT:
959 v.val = sk->sk_rcvlowat;
960 break;
Catherine Zhang877ce7c2006-06-29 12:27:47 -0700961
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700962 case SO_SNDLOWAT:
Eric Dumazet2a915252009-05-27 11:30:05 +0000963 v.val = 1;
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700964 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700965
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700966 case SO_PASSCRED:
967 v.val = test_bit(SOCK_PASSCRED, &sock->flags) ? 1 : 0;
968 break;
969
970 case SO_PEERCRED:
Eric W. Biederman109f6e32010-06-13 03:30:14 +0000971 {
972 struct ucred peercred;
973 if (len > sizeof(peercred))
974 len = sizeof(peercred);
975 cred_to_ucred(sk->sk_peer_pid, sk->sk_peer_cred, &peercred);
976 if (copy_to_user(optval, &peercred, len))
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700977 return -EFAULT;
978 goto lenout;
Eric W. Biederman109f6e32010-06-13 03:30:14 +0000979 }
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700980
981 case SO_PEERNAME:
982 {
983 char address[128];
984
985 if (sock->ops->getname(sock, (struct sockaddr *)address, &lv, 2))
986 return -ENOTCONN;
987 if (lv < len)
988 return -EINVAL;
989 if (copy_to_user(optval, address, len))
990 return -EFAULT;
991 goto lenout;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700992 }
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700993
994 /* Dubious BSD thing... Probably nobody even uses it, but
995 * the UNIX standard wants it for whatever reason... -DaveM
996 */
997 case SO_ACCEPTCONN:
998 v.val = sk->sk_state == TCP_LISTEN;
999 break;
1000
1001 case SO_PASSSEC:
1002 v.val = test_bit(SOCK_PASSSEC, &sock->flags) ? 1 : 0;
1003 break;
1004
1005 case SO_PEERSEC:
1006 return security_socket_getpeersec_stream(sock, optval, optlen, len);
1007
Laszlo Attila Toth4a19ec52008-01-30 19:08:16 -08001008 case SO_MARK:
1009 v.val = sk->sk_mark;
1010 break;
1011
Neil Horman3b885782009-10-12 13:26:31 -07001012 case SO_RXQ_OVFL:
1013 v.val = !!sock_flag(sk, SOCK_RXQ_OVFL);
1014 break;
1015
Johannes Berg6e3e9392011-11-09 10:15:42 +01001016 case SO_WIFI_STATUS:
1017 v.val = !!sock_flag(sk, SOCK_WIFI_STATUS);
1018 break;
1019
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001020 default:
1021 return -ENOPROTOOPT;
1022 }
1023
Linus Torvalds1da177e2005-04-16 15:20:36 -07001024 if (len > lv)
1025 len = lv;
1026 if (copy_to_user(optval, &v, len))
1027 return -EFAULT;
1028lenout:
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001029 if (put_user(len, optlen))
1030 return -EFAULT;
1031 return 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001032}
1033
Ingo Molnara5b5bb9a2006-07-03 00:25:35 -07001034/*
1035 * Initialize an sk_lock.
1036 *
1037 * (We also register the sk_lock with the lock validator.)
1038 */
Dave Jonesb6f99a22007-03-22 12:27:49 -07001039static inline void sock_lock_init(struct sock *sk)
Ingo Molnara5b5bb9a2006-07-03 00:25:35 -07001040{
Peter Zijlstraed075362006-12-06 20:35:24 -08001041 sock_lock_init_class_and_name(sk,
1042 af_family_slock_key_strings[sk->sk_family],
1043 af_family_slock_keys + sk->sk_family,
1044 af_family_key_strings[sk->sk_family],
1045 af_family_keys + sk->sk_family);
Ingo Molnara5b5bb9a2006-07-03 00:25:35 -07001046}
1047
Eric Dumazet4dc6dc72009-07-15 23:13:10 +00001048/*
1049 * Copy all fields from osk to nsk but nsk->sk_refcnt must not change yet,
1050 * even temporarly, because of RCU lookups. sk_node should also be left as is.
Eric Dumazet68835ab2010-11-30 19:04:07 +00001051 * We must not copy fields between sk_dontcopy_begin and sk_dontcopy_end
Eric Dumazet4dc6dc72009-07-15 23:13:10 +00001052 */
Pavel Emelyanovf1a6c4d2007-11-01 00:29:45 -07001053static void sock_copy(struct sock *nsk, const struct sock *osk)
1054{
1055#ifdef CONFIG_SECURITY_NETWORK
1056 void *sptr = nsk->sk_security;
1057#endif
Eric Dumazet68835ab2010-11-30 19:04:07 +00001058 memcpy(nsk, osk, offsetof(struct sock, sk_dontcopy_begin));
1059
1060 memcpy(&nsk->sk_dontcopy_end, &osk->sk_dontcopy_end,
1061 osk->sk_prot->obj_size - offsetof(struct sock, sk_dontcopy_end));
1062
Pavel Emelyanovf1a6c4d2007-11-01 00:29:45 -07001063#ifdef CONFIG_SECURITY_NETWORK
1064 nsk->sk_security = sptr;
1065 security_sk_clone(osk, nsk);
1066#endif
1067}
1068
Octavian Purdilafcbdf092010-12-16 14:26:56 -08001069/*
1070 * caches using SLAB_DESTROY_BY_RCU should let .next pointer from nulls nodes
1071 * un-modified. Special care is taken when initializing object to zero.
1072 */
1073static inline void sk_prot_clear_nulls(struct sock *sk, int size)
1074{
1075 if (offsetof(struct sock, sk_node.next) != 0)
1076 memset(sk, 0, offsetof(struct sock, sk_node.next));
1077 memset(&sk->sk_node.pprev, 0,
1078 size - offsetof(struct sock, sk_node.pprev));
1079}
1080
1081void sk_prot_clear_portaddr_nulls(struct sock *sk, int size)
1082{
1083 unsigned long nulls1, nulls2;
1084
1085 nulls1 = offsetof(struct sock, __sk_common.skc_node.next);
1086 nulls2 = offsetof(struct sock, __sk_common.skc_portaddr_node.next);
1087 if (nulls1 > nulls2)
1088 swap(nulls1, nulls2);
1089
1090 if (nulls1 != 0)
1091 memset((char *)sk, 0, nulls1);
1092 memset((char *)sk + nulls1 + sizeof(void *), 0,
1093 nulls2 - nulls1 - sizeof(void *));
1094 memset((char *)sk + nulls2 + sizeof(void *), 0,
1095 size - nulls2 - sizeof(void *));
1096}
1097EXPORT_SYMBOL(sk_prot_clear_portaddr_nulls);
1098
Pavel Emelyanov2e4afe72007-11-01 00:36:26 -07001099static struct sock *sk_prot_alloc(struct proto *prot, gfp_t priority,
1100 int family)
Pavel Emelyanovc308c1b22007-11-01 00:33:50 -07001101{
1102 struct sock *sk;
1103 struct kmem_cache *slab;
1104
1105 slab = prot->slab;
Eric Dumazete912b112009-07-08 19:36:05 +00001106 if (slab != NULL) {
1107 sk = kmem_cache_alloc(slab, priority & ~__GFP_ZERO);
1108 if (!sk)
1109 return sk;
1110 if (priority & __GFP_ZERO) {
Octavian Purdilafcbdf092010-12-16 14:26:56 -08001111 if (prot->clear_sk)
1112 prot->clear_sk(sk, prot->obj_size);
1113 else
1114 sk_prot_clear_nulls(sk, prot->obj_size);
Eric Dumazete912b112009-07-08 19:36:05 +00001115 }
Octavian Purdilafcbdf092010-12-16 14:26:56 -08001116 } else
Pavel Emelyanovc308c1b22007-11-01 00:33:50 -07001117 sk = kmalloc(prot->obj_size, priority);
1118
Pavel Emelyanov2e4afe72007-11-01 00:36:26 -07001119 if (sk != NULL) {
Vegard Nossuma98b65a2009-02-26 14:46:57 +01001120 kmemcheck_annotate_bitfield(sk, flags);
1121
Pavel Emelyanov2e4afe72007-11-01 00:36:26 -07001122 if (security_sk_alloc(sk, family, priority))
1123 goto out_free;
1124
1125 if (!try_module_get(prot->owner))
1126 goto out_free_sec;
Krishna Kumare022f0b2009-10-19 23:46:20 +00001127 sk_tx_queue_clear(sk);
Pavel Emelyanov2e4afe72007-11-01 00:36:26 -07001128 }
1129
Pavel Emelyanovc308c1b22007-11-01 00:33:50 -07001130 return sk;
Pavel Emelyanov2e4afe72007-11-01 00:36:26 -07001131
1132out_free_sec:
1133 security_sk_free(sk);
1134out_free:
1135 if (slab != NULL)
1136 kmem_cache_free(slab, sk);
1137 else
1138 kfree(sk);
1139 return NULL;
Pavel Emelyanovc308c1b22007-11-01 00:33:50 -07001140}
1141
1142static void sk_prot_free(struct proto *prot, struct sock *sk)
1143{
1144 struct kmem_cache *slab;
Pavel Emelyanov2e4afe72007-11-01 00:36:26 -07001145 struct module *owner;
Pavel Emelyanovc308c1b22007-11-01 00:33:50 -07001146
Pavel Emelyanov2e4afe72007-11-01 00:36:26 -07001147 owner = prot->owner;
Pavel Emelyanovc308c1b22007-11-01 00:33:50 -07001148 slab = prot->slab;
Pavel Emelyanov2e4afe72007-11-01 00:36:26 -07001149
1150 security_sk_free(sk);
Pavel Emelyanovc308c1b22007-11-01 00:33:50 -07001151 if (slab != NULL)
1152 kmem_cache_free(slab, sk);
1153 else
1154 kfree(sk);
Pavel Emelyanov2e4afe72007-11-01 00:36:26 -07001155 module_put(owner);
Pavel Emelyanovc308c1b22007-11-01 00:33:50 -07001156}
1157
Herbert Xuf8451722010-05-24 00:12:34 -07001158#ifdef CONFIG_CGROUPS
1159void sock_update_classid(struct sock *sk)
1160{
Paul E. McKenney11441822010-10-06 17:15:35 -07001161 u32 classid;
Herbert Xuf8451722010-05-24 00:12:34 -07001162
Paul E. McKenney11441822010-10-06 17:15:35 -07001163 rcu_read_lock(); /* doing current task, which cannot vanish. */
1164 classid = task_cls_classid(current);
1165 rcu_read_unlock();
Herbert Xuf8451722010-05-24 00:12:34 -07001166 if (classid && classid != sk->sk_classid)
1167 sk->sk_classid = classid;
1168}
Herbert Xu82862742010-05-24 00:14:10 -07001169EXPORT_SYMBOL(sock_update_classid);
Neil Horman5bc14212011-11-22 05:10:51 +00001170
1171void sock_update_netprioidx(struct sock *sk)
1172{
1173 struct cgroup_netprio_state *state;
1174 if (in_interrupt())
1175 return;
1176 rcu_read_lock();
1177 state = task_netprio_state(current);
1178 sk->sk_cgrp_prioidx = state ? state->prioidx : 0;
1179 rcu_read_unlock();
1180}
1181EXPORT_SYMBOL_GPL(sock_update_netprioidx);
Herbert Xuf8451722010-05-24 00:12:34 -07001182#endif
1183
Linus Torvalds1da177e2005-04-16 15:20:36 -07001184/**
1185 * sk_alloc - All socket objects are allocated here
Randy Dunlapc4ea43c2007-10-12 21:17:49 -07001186 * @net: the applicable net namespace
Pavel Pisa4dc3b162005-05-01 08:59:25 -07001187 * @family: protocol family
1188 * @priority: for allocation (%GFP_KERNEL, %GFP_ATOMIC, etc)
1189 * @prot: struct proto associated with this new sock instance
Linus Torvalds1da177e2005-04-16 15:20:36 -07001190 */
Eric W. Biederman1b8d7ae2007-10-08 23:24:22 -07001191struct sock *sk_alloc(struct net *net, int family, gfp_t priority,
Pavel Emelyanov6257ff22007-11-01 00:39:31 -07001192 struct proto *prot)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001193{
Pavel Emelyanovc308c1b22007-11-01 00:33:50 -07001194 struct sock *sk;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001195
Pavel Emelyanov154adbc2007-11-01 00:38:43 -07001196 sk = sk_prot_alloc(prot, priority | __GFP_ZERO, family);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001197 if (sk) {
Pavel Emelyanov154adbc2007-11-01 00:38:43 -07001198 sk->sk_family = family;
1199 /*
1200 * See comment in struct sock definition to understand
1201 * why we need sk_prot_creator -acme
1202 */
1203 sk->sk_prot = sk->sk_prot_creator = prot;
1204 sock_lock_init(sk);
YOSHIFUJI Hideaki3b1e0a62008-03-26 02:26:21 +09001205 sock_net_set(sk, get_net(net));
Jarek Poplawskid66ee052009-08-30 23:15:36 +00001206 atomic_set(&sk->sk_wmem_alloc, 1);
Herbert Xuf8451722010-05-24 00:12:34 -07001207
1208 sock_update_classid(sk);
Neil Horman5bc14212011-11-22 05:10:51 +00001209 sock_update_netprioidx(sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001210 }
Frank Filza79af592005-09-27 15:23:38 -07001211
Pavel Emelyanov2e4afe72007-11-01 00:36:26 -07001212 return sk;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001213}
Eric Dumazet2a915252009-05-27 11:30:05 +00001214EXPORT_SYMBOL(sk_alloc);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001215
Eric Dumazet2b85a342009-06-11 02:55:43 -07001216static void __sk_free(struct sock *sk)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001217{
1218 struct sk_filter *filter;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001219
1220 if (sk->sk_destruct)
1221 sk->sk_destruct(sk);
1222
Paul E. McKenneya898def2010-02-22 17:04:49 -08001223 filter = rcu_dereference_check(sk->sk_filter,
1224 atomic_read(&sk->sk_wmem_alloc) == 0);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001225 if (filter) {
Pavel Emelyanov309dd5f2007-10-17 21:21:51 -07001226 sk_filter_uncharge(sk, filter);
Stephen Hemmingera9b3cd72011-08-01 16:19:00 +00001227 RCU_INIT_POINTER(sk->sk_filter, NULL);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001228 }
1229
Eric Dumazet08e29af2011-11-28 12:04:18 +00001230 sock_disable_timestamp(sk, SK_FLAGS_TIMESTAMP);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001231
1232 if (atomic_read(&sk->sk_omem_alloc))
1233 printk(KERN_DEBUG "%s: optmem leakage (%d bytes) detected.\n",
Harvey Harrison0dc47872008-03-05 20:47:47 -08001234 __func__, atomic_read(&sk->sk_omem_alloc));
Linus Torvalds1da177e2005-04-16 15:20:36 -07001235
Eric W. Biederman109f6e32010-06-13 03:30:14 +00001236 if (sk->sk_peer_cred)
1237 put_cred(sk->sk_peer_cred);
1238 put_pid(sk->sk_peer_pid);
YOSHIFUJI Hideaki3b1e0a62008-03-26 02:26:21 +09001239 put_net(sock_net(sk));
Pavel Emelyanovc308c1b22007-11-01 00:33:50 -07001240 sk_prot_free(sk->sk_prot_creator, sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001241}
Eric Dumazet2b85a342009-06-11 02:55:43 -07001242
1243void sk_free(struct sock *sk)
1244{
1245 /*
Lucas De Marchi25985ed2011-03-30 22:57:33 -03001246 * We subtract one from sk_wmem_alloc and can know if
Eric Dumazet2b85a342009-06-11 02:55:43 -07001247 * some packets are still in some tx queue.
1248 * If not null, sock_wfree() will call __sk_free(sk) later
1249 */
1250 if (atomic_dec_and_test(&sk->sk_wmem_alloc))
1251 __sk_free(sk);
1252}
Eric Dumazet2a915252009-05-27 11:30:05 +00001253EXPORT_SYMBOL(sk_free);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001254
Denis V. Lunevedf02082008-02-29 11:18:32 -08001255/*
Lucas De Marchi25985ed2011-03-30 22:57:33 -03001256 * Last sock_put should drop reference to sk->sk_net. It has already
1257 * been dropped in sk_change_net. Taking reference to stopping namespace
Denis V. Lunevedf02082008-02-29 11:18:32 -08001258 * is not an option.
Lucas De Marchi25985ed2011-03-30 22:57:33 -03001259 * Take reference to a socket to remove it from hash _alive_ and after that
Denis V. Lunevedf02082008-02-29 11:18:32 -08001260 * destroy it in the context of init_net.
1261 */
1262void sk_release_kernel(struct sock *sk)
1263{
1264 if (sk == NULL || sk->sk_socket == NULL)
1265 return;
1266
1267 sock_hold(sk);
1268 sock_release(sk->sk_socket);
Denis V. Lunev65a18ec2008-04-16 01:59:46 -07001269 release_net(sock_net(sk));
YOSHIFUJI Hideaki3b1e0a62008-03-26 02:26:21 +09001270 sock_net_set(sk, get_net(&init_net));
Denis V. Lunevedf02082008-02-29 11:18:32 -08001271 sock_put(sk);
1272}
David S. Miller45af1752008-02-29 11:33:19 -08001273EXPORT_SYMBOL(sk_release_kernel);
Denis V. Lunevedf02082008-02-29 11:18:32 -08001274
Eric Dumazete56c57d2011-11-08 17:07:07 -05001275/**
1276 * sk_clone_lock - clone a socket, and lock its clone
1277 * @sk: the socket to clone
1278 * @priority: for allocation (%GFP_KERNEL, %GFP_ATOMIC, etc)
1279 *
1280 * Caller must unlock socket even in error path (bh_unlock_sock(newsk))
1281 */
1282struct sock *sk_clone_lock(const struct sock *sk, const gfp_t priority)
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001283{
Pavel Emelyanov8fd1d172007-11-01 00:37:32 -07001284 struct sock *newsk;
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001285
Pavel Emelyanov8fd1d172007-11-01 00:37:32 -07001286 newsk = sk_prot_alloc(sk->sk_prot, priority, sk->sk_family);
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001287 if (newsk != NULL) {
1288 struct sk_filter *filter;
1289
Venkat Yekkirala892c1412006-08-04 23:08:56 -07001290 sock_copy(newsk, sk);
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001291
1292 /* SANITY */
YOSHIFUJI Hideaki3b1e0a62008-03-26 02:26:21 +09001293 get_net(sock_net(newsk));
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001294 sk_node_init(&newsk->sk_node);
1295 sock_lock_init(newsk);
1296 bh_lock_sock(newsk);
Eric Dumazetfa438cc2007-03-04 16:05:44 -08001297 newsk->sk_backlog.head = newsk->sk_backlog.tail = NULL;
Zhu Yi8eae9392010-03-04 18:01:40 +00001298 newsk->sk_backlog.len = 0;
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001299
1300 atomic_set(&newsk->sk_rmem_alloc, 0);
Eric Dumazet2b85a342009-06-11 02:55:43 -07001301 /*
1302 * sk_wmem_alloc set to one (see sk_free() and sock_wfree())
1303 */
1304 atomic_set(&newsk->sk_wmem_alloc, 1);
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001305 atomic_set(&newsk->sk_omem_alloc, 0);
1306 skb_queue_head_init(&newsk->sk_receive_queue);
1307 skb_queue_head_init(&newsk->sk_write_queue);
Chris Leech97fc2f02006-05-23 17:55:33 -07001308#ifdef CONFIG_NET_DMA
1309 skb_queue_head_init(&newsk->sk_async_wait_queue);
1310#endif
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001311
Eric Dumazetb6c67122010-04-08 23:03:29 +00001312 spin_lock_init(&newsk->sk_dst_lock);
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001313 rwlock_init(&newsk->sk_callback_lock);
Peter Zijlstra443aef0e2007-07-19 01:49:00 -07001314 lockdep_set_class_and_name(&newsk->sk_callback_lock,
1315 af_callback_keys + newsk->sk_family,
1316 af_family_clock_key_strings[newsk->sk_family]);
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001317
1318 newsk->sk_dst_cache = NULL;
1319 newsk->sk_wmem_queued = 0;
1320 newsk->sk_forward_alloc = 0;
1321 newsk->sk_send_head = NULL;
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001322 newsk->sk_userlocks = sk->sk_userlocks & ~SOCK_BINDPORT_LOCK;
1323
1324 sock_reset_flag(newsk, SOCK_DONE);
1325 skb_queue_head_init(&newsk->sk_error_queue);
1326
Eric Dumazet0d7da9d2010-10-25 03:47:05 +00001327 filter = rcu_dereference_protected(newsk->sk_filter, 1);
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001328 if (filter != NULL)
1329 sk_filter_charge(newsk, filter);
1330
1331 if (unlikely(xfrm_sk_clone_policy(newsk))) {
1332 /* It is still raw copy of parent, so invalidate
1333 * destructor and make plain sk_free() */
1334 newsk->sk_destruct = NULL;
Thomas Gleixnerb0691c82011-10-25 02:30:50 +00001335 bh_unlock_sock(newsk);
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001336 sk_free(newsk);
1337 newsk = NULL;
1338 goto out;
1339 }
1340
1341 newsk->sk_err = 0;
1342 newsk->sk_priority = 0;
Eric Dumazet4dc6dc72009-07-15 23:13:10 +00001343 /*
1344 * Before updating sk_refcnt, we must commit prior changes to memory
1345 * (Documentation/RCU/rculist_nulls.txt for details)
1346 */
1347 smp_wmb();
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001348 atomic_set(&newsk->sk_refcnt, 2);
1349
1350 /*
1351 * Increment the counter in the same struct proto as the master
1352 * sock (sk_refcnt_debug_inc uses newsk->sk_prot->socks, that
1353 * is the same as sk->sk_prot->socks, as this field was copied
1354 * with memcpy).
1355 *
1356 * This _changes_ the previous behaviour, where
1357 * tcp_create_openreq_child always was incrementing the
1358 * equivalent to tcp_prot->socks (inet_sock_nr), so this have
1359 * to be taken into account in all callers. -acme
1360 */
1361 sk_refcnt_debug_inc(newsk);
David S. Miller972692e2008-06-17 22:41:38 -07001362 sk_set_socket(newsk, NULL);
Eric Dumazet43815482010-04-29 11:01:49 +00001363 newsk->sk_wq = NULL;
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001364
1365 if (newsk->sk_prot->sockets_allocated)
Glauber Costa180d8cd2011-12-11 21:47:02 +00001366 sk_sockets_allocated_inc(newsk);
Octavian Purdila704da5602010-01-08 00:00:09 -08001367
Eric Dumazet08e29af2011-11-28 12:04:18 +00001368 if (newsk->sk_flags & SK_FLAGS_TIMESTAMP)
Octavian Purdila704da5602010-01-08 00:00:09 -08001369 net_enable_timestamp();
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001370 }
1371out:
1372 return newsk;
1373}
Eric Dumazete56c57d2011-11-08 17:07:07 -05001374EXPORT_SYMBOL_GPL(sk_clone_lock);
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001375
Andi Kleen99580892007-04-20 17:12:43 -07001376void sk_setup_caps(struct sock *sk, struct dst_entry *dst)
1377{
1378 __sk_dst_set(sk, dst);
1379 sk->sk_route_caps = dst->dev->features;
1380 if (sk->sk_route_caps & NETIF_F_GSO)
Herbert Xu4fcd6b92007-05-31 22:15:50 -07001381 sk->sk_route_caps |= NETIF_F_GSO_SOFTWARE;
Eric Dumazeta4654192010-05-16 00:36:33 -07001382 sk->sk_route_caps &= ~sk->sk_route_nocaps;
Andi Kleen99580892007-04-20 17:12:43 -07001383 if (sk_can_gso(sk)) {
Peter P Waskiewicz Jr82cc1a72008-03-21 03:43:19 -07001384 if (dst->header_len) {
Andi Kleen99580892007-04-20 17:12:43 -07001385 sk->sk_route_caps &= ~NETIF_F_GSO_MASK;
Peter P Waskiewicz Jr82cc1a72008-03-21 03:43:19 -07001386 } else {
Andi Kleen99580892007-04-20 17:12:43 -07001387 sk->sk_route_caps |= NETIF_F_SG | NETIF_F_HW_CSUM;
Peter P Waskiewicz Jr82cc1a72008-03-21 03:43:19 -07001388 sk->sk_gso_max_size = dst->dev->gso_max_size;
1389 }
Andi Kleen99580892007-04-20 17:12:43 -07001390 }
1391}
1392EXPORT_SYMBOL_GPL(sk_setup_caps);
1393
Linus Torvalds1da177e2005-04-16 15:20:36 -07001394void __init sk_init(void)
1395{
Jan Beulich44813742009-09-21 17:03:05 -07001396 if (totalram_pages <= 4096) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07001397 sysctl_wmem_max = 32767;
1398 sysctl_rmem_max = 32767;
1399 sysctl_wmem_default = 32767;
1400 sysctl_rmem_default = 32767;
Jan Beulich44813742009-09-21 17:03:05 -07001401 } else if (totalram_pages >= 131072) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07001402 sysctl_wmem_max = 131071;
1403 sysctl_rmem_max = 131071;
1404 }
1405}
1406
1407/*
1408 * Simple resource managers for sockets.
1409 */
1410
1411
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001412/*
1413 * Write buffer destructor automatically called from kfree_skb.
Linus Torvalds1da177e2005-04-16 15:20:36 -07001414 */
1415void sock_wfree(struct sk_buff *skb)
1416{
1417 struct sock *sk = skb->sk;
Eric Dumazetd99927f2009-09-24 10:49:24 +00001418 unsigned int len = skb->truesize;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001419
Eric Dumazetd99927f2009-09-24 10:49:24 +00001420 if (!sock_flag(sk, SOCK_USE_WRITE_QUEUE)) {
1421 /*
1422 * Keep a reference on sk_wmem_alloc, this will be released
1423 * after sk_write_space() call
1424 */
1425 atomic_sub(len - 1, &sk->sk_wmem_alloc);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001426 sk->sk_write_space(sk);
Eric Dumazetd99927f2009-09-24 10:49:24 +00001427 len = 1;
1428 }
Eric Dumazet2b85a342009-06-11 02:55:43 -07001429 /*
Eric Dumazetd99927f2009-09-24 10:49:24 +00001430 * if sk_wmem_alloc reaches 0, we must finish what sk_free()
1431 * could not do because of in-flight packets
Eric Dumazet2b85a342009-06-11 02:55:43 -07001432 */
Eric Dumazetd99927f2009-09-24 10:49:24 +00001433 if (atomic_sub_and_test(len, &sk->sk_wmem_alloc))
Eric Dumazet2b85a342009-06-11 02:55:43 -07001434 __sk_free(sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001435}
Eric Dumazet2a915252009-05-27 11:30:05 +00001436EXPORT_SYMBOL(sock_wfree);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001437
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001438/*
1439 * Read buffer destructor automatically called from kfree_skb.
Linus Torvalds1da177e2005-04-16 15:20:36 -07001440 */
1441void sock_rfree(struct sk_buff *skb)
1442{
1443 struct sock *sk = skb->sk;
Eric Dumazetd361fd52010-07-10 22:45:17 +00001444 unsigned int len = skb->truesize;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001445
Eric Dumazetd361fd52010-07-10 22:45:17 +00001446 atomic_sub(len, &sk->sk_rmem_alloc);
1447 sk_mem_uncharge(sk, len);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001448}
Eric Dumazet2a915252009-05-27 11:30:05 +00001449EXPORT_SYMBOL(sock_rfree);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001450
1451
1452int sock_i_uid(struct sock *sk)
1453{
1454 int uid;
1455
Eric Dumazetf064af12010-09-22 12:43:39 +00001456 read_lock_bh(&sk->sk_callback_lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001457 uid = sk->sk_socket ? SOCK_INODE(sk->sk_socket)->i_uid : 0;
Eric Dumazetf064af12010-09-22 12:43:39 +00001458 read_unlock_bh(&sk->sk_callback_lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001459 return uid;
1460}
Eric Dumazet2a915252009-05-27 11:30:05 +00001461EXPORT_SYMBOL(sock_i_uid);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001462
1463unsigned long sock_i_ino(struct sock *sk)
1464{
1465 unsigned long ino;
1466
Eric Dumazetf064af12010-09-22 12:43:39 +00001467 read_lock_bh(&sk->sk_callback_lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001468 ino = sk->sk_socket ? SOCK_INODE(sk->sk_socket)->i_ino : 0;
Eric Dumazetf064af12010-09-22 12:43:39 +00001469 read_unlock_bh(&sk->sk_callback_lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001470 return ino;
1471}
Eric Dumazet2a915252009-05-27 11:30:05 +00001472EXPORT_SYMBOL(sock_i_ino);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001473
1474/*
1475 * Allocate a skb from the socket's send buffer.
1476 */
Victor Fusco86a76ca2005-07-08 14:57:47 -07001477struct sk_buff *sock_wmalloc(struct sock *sk, unsigned long size, int force,
Al Virodd0fc662005-10-07 07:46:04 +01001478 gfp_t priority)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001479{
1480 if (force || atomic_read(&sk->sk_wmem_alloc) < sk->sk_sndbuf) {
Eric Dumazet2a915252009-05-27 11:30:05 +00001481 struct sk_buff *skb = alloc_skb(size, priority);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001482 if (skb) {
1483 skb_set_owner_w(skb, sk);
1484 return skb;
1485 }
1486 }
1487 return NULL;
1488}
Eric Dumazet2a915252009-05-27 11:30:05 +00001489EXPORT_SYMBOL(sock_wmalloc);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001490
1491/*
1492 * Allocate a skb from the socket's receive buffer.
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001493 */
Victor Fusco86a76ca2005-07-08 14:57:47 -07001494struct sk_buff *sock_rmalloc(struct sock *sk, unsigned long size, int force,
Al Virodd0fc662005-10-07 07:46:04 +01001495 gfp_t priority)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001496{
1497 if (force || atomic_read(&sk->sk_rmem_alloc) < sk->sk_rcvbuf) {
1498 struct sk_buff *skb = alloc_skb(size, priority);
1499 if (skb) {
1500 skb_set_owner_r(skb, sk);
1501 return skb;
1502 }
1503 }
1504 return NULL;
1505}
1506
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001507/*
Linus Torvalds1da177e2005-04-16 15:20:36 -07001508 * Allocate a memory block from the socket's option memory buffer.
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001509 */
Al Virodd0fc662005-10-07 07:46:04 +01001510void *sock_kmalloc(struct sock *sk, int size, gfp_t priority)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001511{
1512 if ((unsigned)size <= sysctl_optmem_max &&
1513 atomic_read(&sk->sk_omem_alloc) + size < sysctl_optmem_max) {
1514 void *mem;
1515 /* First do the add, to avoid the race if kmalloc
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001516 * might sleep.
Linus Torvalds1da177e2005-04-16 15:20:36 -07001517 */
1518 atomic_add(size, &sk->sk_omem_alloc);
1519 mem = kmalloc(size, priority);
1520 if (mem)
1521 return mem;
1522 atomic_sub(size, &sk->sk_omem_alloc);
1523 }
1524 return NULL;
1525}
Eric Dumazet2a915252009-05-27 11:30:05 +00001526EXPORT_SYMBOL(sock_kmalloc);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001527
1528/*
1529 * Free an option memory block.
1530 */
1531void sock_kfree_s(struct sock *sk, void *mem, int size)
1532{
1533 kfree(mem);
1534 atomic_sub(size, &sk->sk_omem_alloc);
1535}
Eric Dumazet2a915252009-05-27 11:30:05 +00001536EXPORT_SYMBOL(sock_kfree_s);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001537
1538/* It is almost wait_for_tcp_memory minus release_sock/lock_sock.
1539 I think, these locks should be removed for datagram sockets.
1540 */
Eric Dumazet2a915252009-05-27 11:30:05 +00001541static long sock_wait_for_wmem(struct sock *sk, long timeo)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001542{
1543 DEFINE_WAIT(wait);
1544
1545 clear_bit(SOCK_ASYNC_NOSPACE, &sk->sk_socket->flags);
1546 for (;;) {
1547 if (!timeo)
1548 break;
1549 if (signal_pending(current))
1550 break;
1551 set_bit(SOCK_NOSPACE, &sk->sk_socket->flags);
Eric Dumazetaa395142010-04-20 13:03:51 +00001552 prepare_to_wait(sk_sleep(sk), &wait, TASK_INTERRUPTIBLE);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001553 if (atomic_read(&sk->sk_wmem_alloc) < sk->sk_sndbuf)
1554 break;
1555 if (sk->sk_shutdown & SEND_SHUTDOWN)
1556 break;
1557 if (sk->sk_err)
1558 break;
1559 timeo = schedule_timeout(timeo);
1560 }
Eric Dumazetaa395142010-04-20 13:03:51 +00001561 finish_wait(sk_sleep(sk), &wait);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001562 return timeo;
1563}
1564
1565
1566/*
1567 * Generic send/receive buffer handlers
1568 */
1569
Herbert Xu4cc7f682009-02-04 16:55:54 -08001570struct sk_buff *sock_alloc_send_pskb(struct sock *sk, unsigned long header_len,
1571 unsigned long data_len, int noblock,
1572 int *errcode)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001573{
1574 struct sk_buff *skb;
Al Viro7d877f32005-10-21 03:20:43 -04001575 gfp_t gfp_mask;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001576 long timeo;
1577 int err;
1578
1579 gfp_mask = sk->sk_allocation;
1580 if (gfp_mask & __GFP_WAIT)
1581 gfp_mask |= __GFP_REPEAT;
1582
1583 timeo = sock_sndtimeo(sk, noblock);
1584 while (1) {
1585 err = sock_error(sk);
1586 if (err != 0)
1587 goto failure;
1588
1589 err = -EPIPE;
1590 if (sk->sk_shutdown & SEND_SHUTDOWN)
1591 goto failure;
1592
1593 if (atomic_read(&sk->sk_wmem_alloc) < sk->sk_sndbuf) {
Larry Woodmandb38c1792006-11-03 16:05:45 -08001594 skb = alloc_skb(header_len, gfp_mask);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001595 if (skb) {
1596 int npages;
1597 int i;
1598
1599 /* No pages, we're done... */
1600 if (!data_len)
1601 break;
1602
1603 npages = (data_len + (PAGE_SIZE - 1)) >> PAGE_SHIFT;
1604 skb->truesize += data_len;
1605 skb_shinfo(skb)->nr_frags = npages;
1606 for (i = 0; i < npages; i++) {
1607 struct page *page;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001608
1609 page = alloc_pages(sk->sk_allocation, 0);
1610 if (!page) {
1611 err = -ENOBUFS;
1612 skb_shinfo(skb)->nr_frags = i;
1613 kfree_skb(skb);
1614 goto failure;
1615 }
1616
Ian Campbellea2ab692011-08-22 23:44:58 +00001617 __skb_fill_page_desc(skb, i,
1618 page, 0,
1619 (data_len >= PAGE_SIZE ?
1620 PAGE_SIZE :
1621 data_len));
Linus Torvalds1da177e2005-04-16 15:20:36 -07001622 data_len -= PAGE_SIZE;
1623 }
1624
1625 /* Full success... */
1626 break;
1627 }
1628 err = -ENOBUFS;
1629 goto failure;
1630 }
1631 set_bit(SOCK_ASYNC_NOSPACE, &sk->sk_socket->flags);
1632 set_bit(SOCK_NOSPACE, &sk->sk_socket->flags);
1633 err = -EAGAIN;
1634 if (!timeo)
1635 goto failure;
1636 if (signal_pending(current))
1637 goto interrupted;
1638 timeo = sock_wait_for_wmem(sk, timeo);
1639 }
1640
1641 skb_set_owner_w(skb, sk);
1642 return skb;
1643
1644interrupted:
1645 err = sock_intr_errno(timeo);
1646failure:
1647 *errcode = err;
1648 return NULL;
1649}
Herbert Xu4cc7f682009-02-04 16:55:54 -08001650EXPORT_SYMBOL(sock_alloc_send_pskb);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001651
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001652struct sk_buff *sock_alloc_send_skb(struct sock *sk, unsigned long size,
Linus Torvalds1da177e2005-04-16 15:20:36 -07001653 int noblock, int *errcode)
1654{
1655 return sock_alloc_send_pskb(sk, size, 0, noblock, errcode);
1656}
Eric Dumazet2a915252009-05-27 11:30:05 +00001657EXPORT_SYMBOL(sock_alloc_send_skb);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001658
1659static void __lock_sock(struct sock *sk)
Namhyung Kimf39234d2010-09-08 03:48:48 +00001660 __releases(&sk->sk_lock.slock)
1661 __acquires(&sk->sk_lock.slock)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001662{
1663 DEFINE_WAIT(wait);
1664
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001665 for (;;) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07001666 prepare_to_wait_exclusive(&sk->sk_lock.wq, &wait,
1667 TASK_UNINTERRUPTIBLE);
1668 spin_unlock_bh(&sk->sk_lock.slock);
1669 schedule();
1670 spin_lock_bh(&sk->sk_lock.slock);
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001671 if (!sock_owned_by_user(sk))
Linus Torvalds1da177e2005-04-16 15:20:36 -07001672 break;
1673 }
1674 finish_wait(&sk->sk_lock.wq, &wait);
1675}
1676
1677static void __release_sock(struct sock *sk)
Namhyung Kimf39234d2010-09-08 03:48:48 +00001678 __releases(&sk->sk_lock.slock)
1679 __acquires(&sk->sk_lock.slock)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001680{
1681 struct sk_buff *skb = sk->sk_backlog.head;
1682
1683 do {
1684 sk->sk_backlog.head = sk->sk_backlog.tail = NULL;
1685 bh_unlock_sock(sk);
1686
1687 do {
1688 struct sk_buff *next = skb->next;
1689
Eric Dumazet7fee2262010-05-11 23:19:48 +00001690 WARN_ON_ONCE(skb_dst_is_noref(skb));
Linus Torvalds1da177e2005-04-16 15:20:36 -07001691 skb->next = NULL;
Peter Zijlstrac57943a2008-10-07 14:18:42 -07001692 sk_backlog_rcv(sk, skb);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001693
1694 /*
1695 * We are in process context here with softirqs
1696 * disabled, use cond_resched_softirq() to preempt.
1697 * This is safe to do because we've taken the backlog
1698 * queue private:
1699 */
1700 cond_resched_softirq();
1701
1702 skb = next;
1703 } while (skb != NULL);
1704
1705 bh_lock_sock(sk);
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001706 } while ((skb = sk->sk_backlog.head) != NULL);
Zhu Yi8eae9392010-03-04 18:01:40 +00001707
1708 /*
1709 * Doing the zeroing here guarantee we can not loop forever
1710 * while a wild producer attempts to flood us.
1711 */
1712 sk->sk_backlog.len = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001713}
1714
1715/**
1716 * sk_wait_data - wait for data to arrive at sk_receive_queue
Pavel Pisa4dc3b162005-05-01 08:59:25 -07001717 * @sk: sock to wait on
1718 * @timeo: for how long
Linus Torvalds1da177e2005-04-16 15:20:36 -07001719 *
1720 * Now socket state including sk->sk_err is changed only under lock,
1721 * hence we may omit checks after joining wait queue.
1722 * We check receive queue before schedule() only as optimization;
1723 * it is very likely that release_sock() added new data.
1724 */
1725int sk_wait_data(struct sock *sk, long *timeo)
1726{
1727 int rc;
1728 DEFINE_WAIT(wait);
1729
Eric Dumazetaa395142010-04-20 13:03:51 +00001730 prepare_to_wait(sk_sleep(sk), &wait, TASK_INTERRUPTIBLE);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001731 set_bit(SOCK_ASYNC_WAITDATA, &sk->sk_socket->flags);
1732 rc = sk_wait_event(sk, timeo, !skb_queue_empty(&sk->sk_receive_queue));
1733 clear_bit(SOCK_ASYNC_WAITDATA, &sk->sk_socket->flags);
Eric Dumazetaa395142010-04-20 13:03:51 +00001734 finish_wait(sk_sleep(sk), &wait);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001735 return rc;
1736}
Linus Torvalds1da177e2005-04-16 15:20:36 -07001737EXPORT_SYMBOL(sk_wait_data);
1738
Hideo Aoki3ab224b2007-12-31 00:11:19 -08001739/**
1740 * __sk_mem_schedule - increase sk_forward_alloc and memory_allocated
1741 * @sk: socket
1742 * @size: memory size to allocate
1743 * @kind: allocation type
1744 *
1745 * If kind is SK_MEM_SEND, it means wmem allocation. Otherwise it means
1746 * rmem allocation. This function assumes that protocols which have
1747 * memory_pressure use sk_wmem_queued as write buffer accounting.
1748 */
1749int __sk_mem_schedule(struct sock *sk, int size, int kind)
1750{
1751 struct proto *prot = sk->sk_prot;
1752 int amt = sk_mem_pages(size);
Eric Dumazet8d987e52010-11-09 23:24:26 +00001753 long allocated;
Glauber Costae1aab162011-12-11 21:47:03 +00001754 int parent_status = UNDER_LIMIT;
Hideo Aoki3ab224b2007-12-31 00:11:19 -08001755
1756 sk->sk_forward_alloc += amt * SK_MEM_QUANTUM;
Glauber Costa180d8cd2011-12-11 21:47:02 +00001757
Glauber Costae1aab162011-12-11 21:47:03 +00001758 allocated = sk_memory_allocated_add(sk, amt, &parent_status);
Hideo Aoki3ab224b2007-12-31 00:11:19 -08001759
1760 /* Under limit. */
Glauber Costae1aab162011-12-11 21:47:03 +00001761 if (parent_status == UNDER_LIMIT &&
1762 allocated <= sk_prot_mem_limits(sk, 0)) {
Glauber Costa180d8cd2011-12-11 21:47:02 +00001763 sk_leave_memory_pressure(sk);
Hideo Aoki3ab224b2007-12-31 00:11:19 -08001764 return 1;
1765 }
1766
Glauber Costae1aab162011-12-11 21:47:03 +00001767 /* Under pressure. (we or our parents) */
1768 if ((parent_status > SOFT_LIMIT) ||
1769 allocated > sk_prot_mem_limits(sk, 1))
Glauber Costa180d8cd2011-12-11 21:47:02 +00001770 sk_enter_memory_pressure(sk);
Hideo Aoki3ab224b2007-12-31 00:11:19 -08001771
Glauber Costae1aab162011-12-11 21:47:03 +00001772 /* Over hard limit (we or our parents) */
1773 if ((parent_status == OVER_LIMIT) ||
1774 (allocated > sk_prot_mem_limits(sk, 2)))
Hideo Aoki3ab224b2007-12-31 00:11:19 -08001775 goto suppress_allocation;
1776
1777 /* guarantee minimum buffer size under pressure */
1778 if (kind == SK_MEM_RECV) {
1779 if (atomic_read(&sk->sk_rmem_alloc) < prot->sysctl_rmem[0])
1780 return 1;
Glauber Costa180d8cd2011-12-11 21:47:02 +00001781
Hideo Aoki3ab224b2007-12-31 00:11:19 -08001782 } else { /* SK_MEM_SEND */
1783 if (sk->sk_type == SOCK_STREAM) {
1784 if (sk->sk_wmem_queued < prot->sysctl_wmem[0])
1785 return 1;
1786 } else if (atomic_read(&sk->sk_wmem_alloc) <
1787 prot->sysctl_wmem[0])
1788 return 1;
1789 }
1790
Glauber Costa180d8cd2011-12-11 21:47:02 +00001791 if (sk_has_memory_pressure(sk)) {
Eric Dumazet17483762008-11-25 21:16:35 -08001792 int alloc;
1793
Glauber Costa180d8cd2011-12-11 21:47:02 +00001794 if (!sk_under_memory_pressure(sk))
Eric Dumazet17483762008-11-25 21:16:35 -08001795 return 1;
Glauber Costa180d8cd2011-12-11 21:47:02 +00001796 alloc = sk_sockets_allocated_read_positive(sk);
1797 if (sk_prot_mem_limits(sk, 2) > alloc *
Hideo Aoki3ab224b2007-12-31 00:11:19 -08001798 sk_mem_pages(sk->sk_wmem_queued +
1799 atomic_read(&sk->sk_rmem_alloc) +
1800 sk->sk_forward_alloc))
1801 return 1;
1802 }
1803
1804suppress_allocation:
1805
1806 if (kind == SK_MEM_SEND && sk->sk_type == SOCK_STREAM) {
1807 sk_stream_moderate_sndbuf(sk);
1808
1809 /* Fail only if socket is _under_ its sndbuf.
1810 * In this case we cannot block, so that we have to fail.
1811 */
1812 if (sk->sk_wmem_queued + size >= sk->sk_sndbuf)
1813 return 1;
1814 }
1815
Satoru Moriya3847ce32011-06-17 12:00:03 +00001816 trace_sock_exceed_buf_limit(sk, prot, allocated);
1817
Hideo Aoki3ab224b2007-12-31 00:11:19 -08001818 /* Alas. Undo changes. */
1819 sk->sk_forward_alloc -= amt * SK_MEM_QUANTUM;
Glauber Costa180d8cd2011-12-11 21:47:02 +00001820
Glauber Costae1aab162011-12-11 21:47:03 +00001821 sk_memory_allocated_sub(sk, amt, parent_status);
Glauber Costa180d8cd2011-12-11 21:47:02 +00001822
Hideo Aoki3ab224b2007-12-31 00:11:19 -08001823 return 0;
1824}
Hideo Aoki3ab224b2007-12-31 00:11:19 -08001825EXPORT_SYMBOL(__sk_mem_schedule);
1826
1827/**
1828 * __sk_reclaim - reclaim memory_allocated
1829 * @sk: socket
1830 */
1831void __sk_mem_reclaim(struct sock *sk)
1832{
Glauber Costa180d8cd2011-12-11 21:47:02 +00001833 sk_memory_allocated_sub(sk,
Glauber Costae1aab162011-12-11 21:47:03 +00001834 sk->sk_forward_alloc >> SK_MEM_QUANTUM_SHIFT, 0);
Hideo Aoki3ab224b2007-12-31 00:11:19 -08001835 sk->sk_forward_alloc &= SK_MEM_QUANTUM - 1;
1836
Glauber Costa180d8cd2011-12-11 21:47:02 +00001837 if (sk_under_memory_pressure(sk) &&
1838 (sk_memory_allocated(sk) < sk_prot_mem_limits(sk, 0)))
1839 sk_leave_memory_pressure(sk);
Hideo Aoki3ab224b2007-12-31 00:11:19 -08001840}
Hideo Aoki3ab224b2007-12-31 00:11:19 -08001841EXPORT_SYMBOL(__sk_mem_reclaim);
1842
1843
Linus Torvalds1da177e2005-04-16 15:20:36 -07001844/*
1845 * Set of default routines for initialising struct proto_ops when
1846 * the protocol does not support a particular function. In certain
1847 * cases where it makes no sense for a protocol to have a "do nothing"
1848 * function, some default processing is provided.
1849 */
1850
1851int sock_no_bind(struct socket *sock, struct sockaddr *saddr, int len)
1852{
1853 return -EOPNOTSUPP;
1854}
Eric Dumazet2a915252009-05-27 11:30:05 +00001855EXPORT_SYMBOL(sock_no_bind);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001856
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001857int sock_no_connect(struct socket *sock, struct sockaddr *saddr,
Linus Torvalds1da177e2005-04-16 15:20:36 -07001858 int len, int flags)
1859{
1860 return -EOPNOTSUPP;
1861}
Eric Dumazet2a915252009-05-27 11:30:05 +00001862EXPORT_SYMBOL(sock_no_connect);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001863
1864int sock_no_socketpair(struct socket *sock1, struct socket *sock2)
1865{
1866 return -EOPNOTSUPP;
1867}
Eric Dumazet2a915252009-05-27 11:30:05 +00001868EXPORT_SYMBOL(sock_no_socketpair);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001869
1870int sock_no_accept(struct socket *sock, struct socket *newsock, int flags)
1871{
1872 return -EOPNOTSUPP;
1873}
Eric Dumazet2a915252009-05-27 11:30:05 +00001874EXPORT_SYMBOL(sock_no_accept);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001875
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001876int sock_no_getname(struct socket *sock, struct sockaddr *saddr,
Linus Torvalds1da177e2005-04-16 15:20:36 -07001877 int *len, int peer)
1878{
1879 return -EOPNOTSUPP;
1880}
Eric Dumazet2a915252009-05-27 11:30:05 +00001881EXPORT_SYMBOL(sock_no_getname);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001882
Eric Dumazet2a915252009-05-27 11:30:05 +00001883unsigned int sock_no_poll(struct file *file, struct socket *sock, poll_table *pt)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001884{
1885 return 0;
1886}
Eric Dumazet2a915252009-05-27 11:30:05 +00001887EXPORT_SYMBOL(sock_no_poll);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001888
1889int sock_no_ioctl(struct socket *sock, unsigned int cmd, unsigned long arg)
1890{
1891 return -EOPNOTSUPP;
1892}
Eric Dumazet2a915252009-05-27 11:30:05 +00001893EXPORT_SYMBOL(sock_no_ioctl);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001894
1895int sock_no_listen(struct socket *sock, int backlog)
1896{
1897 return -EOPNOTSUPP;
1898}
Eric Dumazet2a915252009-05-27 11:30:05 +00001899EXPORT_SYMBOL(sock_no_listen);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001900
1901int sock_no_shutdown(struct socket *sock, int how)
1902{
1903 return -EOPNOTSUPP;
1904}
Eric Dumazet2a915252009-05-27 11:30:05 +00001905EXPORT_SYMBOL(sock_no_shutdown);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001906
1907int sock_no_setsockopt(struct socket *sock, int level, int optname,
David S. Millerb7058842009-09-30 16:12:20 -07001908 char __user *optval, unsigned int optlen)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001909{
1910 return -EOPNOTSUPP;
1911}
Eric Dumazet2a915252009-05-27 11:30:05 +00001912EXPORT_SYMBOL(sock_no_setsockopt);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001913
1914int sock_no_getsockopt(struct socket *sock, int level, int optname,
1915 char __user *optval, int __user *optlen)
1916{
1917 return -EOPNOTSUPP;
1918}
Eric Dumazet2a915252009-05-27 11:30:05 +00001919EXPORT_SYMBOL(sock_no_getsockopt);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001920
1921int sock_no_sendmsg(struct kiocb *iocb, struct socket *sock, struct msghdr *m,
1922 size_t len)
1923{
1924 return -EOPNOTSUPP;
1925}
Eric Dumazet2a915252009-05-27 11:30:05 +00001926EXPORT_SYMBOL(sock_no_sendmsg);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001927
1928int sock_no_recvmsg(struct kiocb *iocb, struct socket *sock, struct msghdr *m,
1929 size_t len, int flags)
1930{
1931 return -EOPNOTSUPP;
1932}
Eric Dumazet2a915252009-05-27 11:30:05 +00001933EXPORT_SYMBOL(sock_no_recvmsg);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001934
1935int sock_no_mmap(struct file *file, struct socket *sock, struct vm_area_struct *vma)
1936{
1937 /* Mirror missing mmap method error code */
1938 return -ENODEV;
1939}
Eric Dumazet2a915252009-05-27 11:30:05 +00001940EXPORT_SYMBOL(sock_no_mmap);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001941
1942ssize_t sock_no_sendpage(struct socket *sock, struct page *page, int offset, size_t size, int flags)
1943{
1944 ssize_t res;
1945 struct msghdr msg = {.msg_flags = flags};
1946 struct kvec iov;
1947 char *kaddr = kmap(page);
1948 iov.iov_base = kaddr + offset;
1949 iov.iov_len = size;
1950 res = kernel_sendmsg(sock, &msg, &iov, 1, size);
1951 kunmap(page);
1952 return res;
1953}
Eric Dumazet2a915252009-05-27 11:30:05 +00001954EXPORT_SYMBOL(sock_no_sendpage);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001955
1956/*
1957 * Default Socket Callbacks
1958 */
1959
1960static void sock_def_wakeup(struct sock *sk)
1961{
Eric Dumazet43815482010-04-29 11:01:49 +00001962 struct socket_wq *wq;
1963
1964 rcu_read_lock();
1965 wq = rcu_dereference(sk->sk_wq);
1966 if (wq_has_sleeper(wq))
1967 wake_up_interruptible_all(&wq->wait);
1968 rcu_read_unlock();
Linus Torvalds1da177e2005-04-16 15:20:36 -07001969}
1970
1971static void sock_def_error_report(struct sock *sk)
1972{
Eric Dumazet43815482010-04-29 11:01:49 +00001973 struct socket_wq *wq;
1974
1975 rcu_read_lock();
1976 wq = rcu_dereference(sk->sk_wq);
1977 if (wq_has_sleeper(wq))
1978 wake_up_interruptible_poll(&wq->wait, POLLERR);
Pavel Emelyanov8d8ad9d2007-11-26 20:10:50 +08001979 sk_wake_async(sk, SOCK_WAKE_IO, POLL_ERR);
Eric Dumazet43815482010-04-29 11:01:49 +00001980 rcu_read_unlock();
Linus Torvalds1da177e2005-04-16 15:20:36 -07001981}
1982
1983static void sock_def_readable(struct sock *sk, int len)
1984{
Eric Dumazet43815482010-04-29 11:01:49 +00001985 struct socket_wq *wq;
1986
1987 rcu_read_lock();
1988 wq = rcu_dereference(sk->sk_wq);
1989 if (wq_has_sleeper(wq))
Eric Dumazet2c6607c2011-01-06 10:54:29 -08001990 wake_up_interruptible_sync_poll(&wq->wait, POLLIN | POLLPRI |
Davide Libenzi37e55402009-03-31 15:24:21 -07001991 POLLRDNORM | POLLRDBAND);
Pavel Emelyanov8d8ad9d2007-11-26 20:10:50 +08001992 sk_wake_async(sk, SOCK_WAKE_WAITD, POLL_IN);
Eric Dumazet43815482010-04-29 11:01:49 +00001993 rcu_read_unlock();
Linus Torvalds1da177e2005-04-16 15:20:36 -07001994}
1995
1996static void sock_def_write_space(struct sock *sk)
1997{
Eric Dumazet43815482010-04-29 11:01:49 +00001998 struct socket_wq *wq;
1999
2000 rcu_read_lock();
Linus Torvalds1da177e2005-04-16 15:20:36 -07002001
2002 /* Do not wake up a writer until he can make "significant"
2003 * progress. --DaveM
2004 */
Stephen Hemmingere71a4782007-04-10 20:10:33 -07002005 if ((atomic_read(&sk->sk_wmem_alloc) << 1) <= sk->sk_sndbuf) {
Eric Dumazet43815482010-04-29 11:01:49 +00002006 wq = rcu_dereference(sk->sk_wq);
2007 if (wq_has_sleeper(wq))
2008 wake_up_interruptible_sync_poll(&wq->wait, POLLOUT |
Davide Libenzi37e55402009-03-31 15:24:21 -07002009 POLLWRNORM | POLLWRBAND);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002010
2011 /* Should agree with poll, otherwise some programs break */
2012 if (sock_writeable(sk))
Pavel Emelyanov8d8ad9d2007-11-26 20:10:50 +08002013 sk_wake_async(sk, SOCK_WAKE_SPACE, POLL_OUT);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002014 }
2015
Eric Dumazet43815482010-04-29 11:01:49 +00002016 rcu_read_unlock();
Linus Torvalds1da177e2005-04-16 15:20:36 -07002017}
2018
2019static void sock_def_destruct(struct sock *sk)
2020{
Jesper Juhla51482b2005-11-08 09:41:34 -08002021 kfree(sk->sk_protinfo);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002022}
2023
2024void sk_send_sigurg(struct sock *sk)
2025{
2026 if (sk->sk_socket && sk->sk_socket->file)
2027 if (send_sigurg(&sk->sk_socket->file->f_owner))
Pavel Emelyanov8d8ad9d2007-11-26 20:10:50 +08002028 sk_wake_async(sk, SOCK_WAKE_URG, POLL_PRI);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002029}
Eric Dumazet2a915252009-05-27 11:30:05 +00002030EXPORT_SYMBOL(sk_send_sigurg);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002031
2032void sk_reset_timer(struct sock *sk, struct timer_list* timer,
2033 unsigned long expires)
2034{
2035 if (!mod_timer(timer, expires))
2036 sock_hold(sk);
2037}
Linus Torvalds1da177e2005-04-16 15:20:36 -07002038EXPORT_SYMBOL(sk_reset_timer);
2039
2040void sk_stop_timer(struct sock *sk, struct timer_list* timer)
2041{
2042 if (timer_pending(timer) && del_timer(timer))
2043 __sock_put(sk);
2044}
Linus Torvalds1da177e2005-04-16 15:20:36 -07002045EXPORT_SYMBOL(sk_stop_timer);
2046
2047void sock_init_data(struct socket *sock, struct sock *sk)
2048{
2049 skb_queue_head_init(&sk->sk_receive_queue);
2050 skb_queue_head_init(&sk->sk_write_queue);
2051 skb_queue_head_init(&sk->sk_error_queue);
Chris Leech97fc2f02006-05-23 17:55:33 -07002052#ifdef CONFIG_NET_DMA
2053 skb_queue_head_init(&sk->sk_async_wait_queue);
2054#endif
Linus Torvalds1da177e2005-04-16 15:20:36 -07002055
2056 sk->sk_send_head = NULL;
2057
2058 init_timer(&sk->sk_timer);
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09002059
Linus Torvalds1da177e2005-04-16 15:20:36 -07002060 sk->sk_allocation = GFP_KERNEL;
2061 sk->sk_rcvbuf = sysctl_rmem_default;
2062 sk->sk_sndbuf = sysctl_wmem_default;
2063 sk->sk_state = TCP_CLOSE;
David S. Miller972692e2008-06-17 22:41:38 -07002064 sk_set_socket(sk, sock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002065
2066 sock_set_flag(sk, SOCK_ZAPPED);
2067
Stephen Hemmingere71a4782007-04-10 20:10:33 -07002068 if (sock) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07002069 sk->sk_type = sock->type;
Eric Dumazet43815482010-04-29 11:01:49 +00002070 sk->sk_wq = sock->wq;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002071 sock->sk = sk;
2072 } else
Eric Dumazet43815482010-04-29 11:01:49 +00002073 sk->sk_wq = NULL;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002074
Eric Dumazetb6c67122010-04-08 23:03:29 +00002075 spin_lock_init(&sk->sk_dst_lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002076 rwlock_init(&sk->sk_callback_lock);
Peter Zijlstra443aef0e2007-07-19 01:49:00 -07002077 lockdep_set_class_and_name(&sk->sk_callback_lock,
2078 af_callback_keys + sk->sk_family,
2079 af_family_clock_key_strings[sk->sk_family]);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002080
2081 sk->sk_state_change = sock_def_wakeup;
2082 sk->sk_data_ready = sock_def_readable;
2083 sk->sk_write_space = sock_def_write_space;
2084 sk->sk_error_report = sock_def_error_report;
2085 sk->sk_destruct = sock_def_destruct;
2086
2087 sk->sk_sndmsg_page = NULL;
2088 sk->sk_sndmsg_off = 0;
2089
Eric W. Biederman109f6e32010-06-13 03:30:14 +00002090 sk->sk_peer_pid = NULL;
2091 sk->sk_peer_cred = NULL;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002092 sk->sk_write_pending = 0;
2093 sk->sk_rcvlowat = 1;
2094 sk->sk_rcvtimeo = MAX_SCHEDULE_TIMEOUT;
2095 sk->sk_sndtimeo = MAX_SCHEDULE_TIMEOUT;
2096
Eric Dumazetf37f0af2008-04-13 21:39:26 -07002097 sk->sk_stamp = ktime_set(-1L, 0);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002098
Eric Dumazet4dc6dc72009-07-15 23:13:10 +00002099 /*
2100 * Before updating sk_refcnt, we must commit prior changes to memory
2101 * (Documentation/RCU/rculist_nulls.txt for details)
2102 */
2103 smp_wmb();
Linus Torvalds1da177e2005-04-16 15:20:36 -07002104 atomic_set(&sk->sk_refcnt, 1);
Wang Chen33c732c2007-11-13 20:30:01 -08002105 atomic_set(&sk->sk_drops, 0);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002106}
Eric Dumazet2a915252009-05-27 11:30:05 +00002107EXPORT_SYMBOL(sock_init_data);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002108
Harvey Harrisonb5606c22008-02-13 15:03:16 -08002109void lock_sock_nested(struct sock *sk, int subclass)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002110{
2111 might_sleep();
Ingo Molnara5b5bb9a2006-07-03 00:25:35 -07002112 spin_lock_bh(&sk->sk_lock.slock);
John Heffnerd2e91172007-09-12 10:44:19 +02002113 if (sk->sk_lock.owned)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002114 __lock_sock(sk);
John Heffnerd2e91172007-09-12 10:44:19 +02002115 sk->sk_lock.owned = 1;
Ingo Molnara5b5bb9a2006-07-03 00:25:35 -07002116 spin_unlock(&sk->sk_lock.slock);
2117 /*
2118 * The sk_lock has mutex_lock() semantics here:
2119 */
Peter Zijlstrafcc70d52006-11-08 22:44:35 -08002120 mutex_acquire(&sk->sk_lock.dep_map, subclass, 0, _RET_IP_);
Ingo Molnara5b5bb9a2006-07-03 00:25:35 -07002121 local_bh_enable();
Linus Torvalds1da177e2005-04-16 15:20:36 -07002122}
Peter Zijlstrafcc70d52006-11-08 22:44:35 -08002123EXPORT_SYMBOL(lock_sock_nested);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002124
Harvey Harrisonb5606c22008-02-13 15:03:16 -08002125void release_sock(struct sock *sk)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002126{
Ingo Molnara5b5bb9a2006-07-03 00:25:35 -07002127 /*
2128 * The sk_lock has mutex_unlock() semantics:
2129 */
2130 mutex_release(&sk->sk_lock.dep_map, 1, _RET_IP_);
2131
2132 spin_lock_bh(&sk->sk_lock.slock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002133 if (sk->sk_backlog.tail)
2134 __release_sock(sk);
John Heffnerd2e91172007-09-12 10:44:19 +02002135 sk->sk_lock.owned = 0;
Ingo Molnara5b5bb9a2006-07-03 00:25:35 -07002136 if (waitqueue_active(&sk->sk_lock.wq))
2137 wake_up(&sk->sk_lock.wq);
2138 spin_unlock_bh(&sk->sk_lock.slock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002139}
2140EXPORT_SYMBOL(release_sock);
2141
Eric Dumazet8a74ad62010-05-26 19:20:18 +00002142/**
2143 * lock_sock_fast - fast version of lock_sock
2144 * @sk: socket
2145 *
2146 * This version should be used for very small section, where process wont block
2147 * return false if fast path is taken
2148 * sk_lock.slock locked, owned = 0, BH disabled
2149 * return true if slow path is taken
2150 * sk_lock.slock unlocked, owned = 1, BH enabled
2151 */
2152bool lock_sock_fast(struct sock *sk)
2153{
2154 might_sleep();
2155 spin_lock_bh(&sk->sk_lock.slock);
2156
2157 if (!sk->sk_lock.owned)
2158 /*
2159 * Note : We must disable BH
2160 */
2161 return false;
2162
2163 __lock_sock(sk);
2164 sk->sk_lock.owned = 1;
2165 spin_unlock(&sk->sk_lock.slock);
2166 /*
2167 * The sk_lock has mutex_lock() semantics here:
2168 */
2169 mutex_acquire(&sk->sk_lock.dep_map, 0, 0, _RET_IP_);
2170 local_bh_enable();
2171 return true;
2172}
2173EXPORT_SYMBOL(lock_sock_fast);
2174
Linus Torvalds1da177e2005-04-16 15:20:36 -07002175int sock_get_timestamp(struct sock *sk, struct timeval __user *userstamp)
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09002176{
Eric Dumazetb7aa0bf2007-04-19 16:16:32 -07002177 struct timeval tv;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002178 if (!sock_flag(sk, SOCK_TIMESTAMP))
Patrick Ohly20d49472009-02-12 05:03:38 +00002179 sock_enable_timestamp(sk, SOCK_TIMESTAMP);
Eric Dumazetb7aa0bf2007-04-19 16:16:32 -07002180 tv = ktime_to_timeval(sk->sk_stamp);
2181 if (tv.tv_sec == -1)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002182 return -ENOENT;
Eric Dumazetb7aa0bf2007-04-19 16:16:32 -07002183 if (tv.tv_sec == 0) {
2184 sk->sk_stamp = ktime_get_real();
2185 tv = ktime_to_timeval(sk->sk_stamp);
2186 }
2187 return copy_to_user(userstamp, &tv, sizeof(tv)) ? -EFAULT : 0;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09002188}
Linus Torvalds1da177e2005-04-16 15:20:36 -07002189EXPORT_SYMBOL(sock_get_timestamp);
2190
Eric Dumazetae40eb12007-03-18 17:33:16 -07002191int sock_get_timestampns(struct sock *sk, struct timespec __user *userstamp)
2192{
2193 struct timespec ts;
2194 if (!sock_flag(sk, SOCK_TIMESTAMP))
Patrick Ohly20d49472009-02-12 05:03:38 +00002195 sock_enable_timestamp(sk, SOCK_TIMESTAMP);
Eric Dumazetae40eb12007-03-18 17:33:16 -07002196 ts = ktime_to_timespec(sk->sk_stamp);
2197 if (ts.tv_sec == -1)
2198 return -ENOENT;
2199 if (ts.tv_sec == 0) {
2200 sk->sk_stamp = ktime_get_real();
2201 ts = ktime_to_timespec(sk->sk_stamp);
2202 }
2203 return copy_to_user(userstamp, &ts, sizeof(ts)) ? -EFAULT : 0;
2204}
2205EXPORT_SYMBOL(sock_get_timestampns);
2206
Patrick Ohly20d49472009-02-12 05:03:38 +00002207void sock_enable_timestamp(struct sock *sk, int flag)
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09002208{
Patrick Ohly20d49472009-02-12 05:03:38 +00002209 if (!sock_flag(sk, flag)) {
Eric Dumazet08e29af2011-11-28 12:04:18 +00002210 unsigned long previous_flags = sk->sk_flags;
2211
Patrick Ohly20d49472009-02-12 05:03:38 +00002212 sock_set_flag(sk, flag);
2213 /*
2214 * we just set one of the two flags which require net
2215 * time stamping, but time stamping might have been on
2216 * already because of the other one
2217 */
Eric Dumazet08e29af2011-11-28 12:04:18 +00002218 if (!(previous_flags & SK_FLAGS_TIMESTAMP))
Patrick Ohly20d49472009-02-12 05:03:38 +00002219 net_enable_timestamp();
Linus Torvalds1da177e2005-04-16 15:20:36 -07002220 }
2221}
Linus Torvalds1da177e2005-04-16 15:20:36 -07002222
2223/*
2224 * Get a socket option on an socket.
2225 *
2226 * FIX: POSIX 1003.1g is very ambiguous here. It states that
2227 * asynchronous errors should be reported by getsockopt. We assume
2228 * this means if you specify SO_ERROR (otherwise whats the point of it).
2229 */
2230int sock_common_getsockopt(struct socket *sock, int level, int optname,
2231 char __user *optval, int __user *optlen)
2232{
2233 struct sock *sk = sock->sk;
2234
2235 return sk->sk_prot->getsockopt(sk, level, optname, optval, optlen);
2236}
Linus Torvalds1da177e2005-04-16 15:20:36 -07002237EXPORT_SYMBOL(sock_common_getsockopt);
2238
Dmitry Mishin3fdadf72006-03-20 22:45:21 -08002239#ifdef CONFIG_COMPAT
Arnaldo Carvalho de Melo543d9cf2006-03-20 22:48:35 -08002240int compat_sock_common_getsockopt(struct socket *sock, int level, int optname,
2241 char __user *optval, int __user *optlen)
Dmitry Mishin3fdadf72006-03-20 22:45:21 -08002242{
2243 struct sock *sk = sock->sk;
2244
Johannes Berg1e51f952007-03-06 13:44:06 -08002245 if (sk->sk_prot->compat_getsockopt != NULL)
Arnaldo Carvalho de Melo543d9cf2006-03-20 22:48:35 -08002246 return sk->sk_prot->compat_getsockopt(sk, level, optname,
2247 optval, optlen);
Dmitry Mishin3fdadf72006-03-20 22:45:21 -08002248 return sk->sk_prot->getsockopt(sk, level, optname, optval, optlen);
2249}
2250EXPORT_SYMBOL(compat_sock_common_getsockopt);
2251#endif
2252
Linus Torvalds1da177e2005-04-16 15:20:36 -07002253int sock_common_recvmsg(struct kiocb *iocb, struct socket *sock,
2254 struct msghdr *msg, size_t size, int flags)
2255{
2256 struct sock *sk = sock->sk;
2257 int addr_len = 0;
2258 int err;
2259
2260 err = sk->sk_prot->recvmsg(iocb, sk, msg, size, flags & MSG_DONTWAIT,
2261 flags & ~MSG_DONTWAIT, &addr_len);
2262 if (err >= 0)
2263 msg->msg_namelen = addr_len;
2264 return err;
2265}
Linus Torvalds1da177e2005-04-16 15:20:36 -07002266EXPORT_SYMBOL(sock_common_recvmsg);
2267
2268/*
2269 * Set socket options on an inet socket.
2270 */
2271int sock_common_setsockopt(struct socket *sock, int level, int optname,
David S. Millerb7058842009-09-30 16:12:20 -07002272 char __user *optval, unsigned int optlen)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002273{
2274 struct sock *sk = sock->sk;
2275
2276 return sk->sk_prot->setsockopt(sk, level, optname, optval, optlen);
2277}
Linus Torvalds1da177e2005-04-16 15:20:36 -07002278EXPORT_SYMBOL(sock_common_setsockopt);
2279
Dmitry Mishin3fdadf72006-03-20 22:45:21 -08002280#ifdef CONFIG_COMPAT
Arnaldo Carvalho de Melo543d9cf2006-03-20 22:48:35 -08002281int compat_sock_common_setsockopt(struct socket *sock, int level, int optname,
David S. Millerb7058842009-09-30 16:12:20 -07002282 char __user *optval, unsigned int optlen)
Dmitry Mishin3fdadf72006-03-20 22:45:21 -08002283{
2284 struct sock *sk = sock->sk;
2285
Arnaldo Carvalho de Melo543d9cf2006-03-20 22:48:35 -08002286 if (sk->sk_prot->compat_setsockopt != NULL)
2287 return sk->sk_prot->compat_setsockopt(sk, level, optname,
2288 optval, optlen);
Dmitry Mishin3fdadf72006-03-20 22:45:21 -08002289 return sk->sk_prot->setsockopt(sk, level, optname, optval, optlen);
2290}
2291EXPORT_SYMBOL(compat_sock_common_setsockopt);
2292#endif
2293
Linus Torvalds1da177e2005-04-16 15:20:36 -07002294void sk_common_release(struct sock *sk)
2295{
2296 if (sk->sk_prot->destroy)
2297 sk->sk_prot->destroy(sk);
2298
2299 /*
2300 * Observation: when sock_common_release is called, processes have
2301 * no access to socket. But net still has.
2302 * Step one, detach it from networking:
2303 *
2304 * A. Remove from hash tables.
2305 */
2306
2307 sk->sk_prot->unhash(sk);
2308
2309 /*
2310 * In this point socket cannot receive new packets, but it is possible
2311 * that some packets are in flight because some CPU runs receiver and
2312 * did hash table lookup before we unhashed socket. They will achieve
2313 * receive queue and will be purged by socket destructor.
2314 *
2315 * Also we still have packets pending on receive queue and probably,
2316 * our own packets waiting in device queues. sock_destroy will drain
2317 * receive queue, but transmitted packets will delay socket destruction
2318 * until the last reference will be released.
2319 */
2320
2321 sock_orphan(sk);
2322
2323 xfrm_sk_free_policy(sk);
2324
Arnaldo Carvalho de Meloe6848972005-08-09 19:45:38 -07002325 sk_refcnt_debug_release(sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002326 sock_put(sk);
2327}
Linus Torvalds1da177e2005-04-16 15:20:36 -07002328EXPORT_SYMBOL(sk_common_release);
2329
Pavel Emelyanov13ff3d62008-03-28 16:38:17 -07002330#ifdef CONFIG_PROC_FS
2331#define PROTO_INUSE_NR 64 /* should be enough for the first time */
Pavel Emelyanov1338d462008-03-28 16:38:43 -07002332struct prot_inuse {
2333 int val[PROTO_INUSE_NR];
2334};
Pavel Emelyanov13ff3d62008-03-28 16:38:17 -07002335
2336static DECLARE_BITMAP(proto_inuse_idx, PROTO_INUSE_NR);
Pavel Emelyanov70ee1152008-03-31 19:42:16 -07002337
2338#ifdef CONFIG_NET_NS
2339void sock_prot_inuse_add(struct net *net, struct proto *prot, int val)
2340{
Eric Dumazetd6d9ca02010-07-19 10:48:49 +00002341 __this_cpu_add(net->core.inuse->val[prot->inuse_idx], val);
Pavel Emelyanov70ee1152008-03-31 19:42:16 -07002342}
2343EXPORT_SYMBOL_GPL(sock_prot_inuse_add);
2344
2345int sock_prot_inuse_get(struct net *net, struct proto *prot)
2346{
2347 int cpu, idx = prot->inuse_idx;
2348 int res = 0;
2349
2350 for_each_possible_cpu(cpu)
2351 res += per_cpu_ptr(net->core.inuse, cpu)->val[idx];
2352
2353 return res >= 0 ? res : 0;
2354}
2355EXPORT_SYMBOL_GPL(sock_prot_inuse_get);
2356
Alexey Dobriyan2c8c1e72010-01-17 03:35:32 +00002357static int __net_init sock_inuse_init_net(struct net *net)
Pavel Emelyanov70ee1152008-03-31 19:42:16 -07002358{
2359 net->core.inuse = alloc_percpu(struct prot_inuse);
2360 return net->core.inuse ? 0 : -ENOMEM;
2361}
2362
Alexey Dobriyan2c8c1e72010-01-17 03:35:32 +00002363static void __net_exit sock_inuse_exit_net(struct net *net)
Pavel Emelyanov70ee1152008-03-31 19:42:16 -07002364{
2365 free_percpu(net->core.inuse);
2366}
2367
2368static struct pernet_operations net_inuse_ops = {
2369 .init = sock_inuse_init_net,
2370 .exit = sock_inuse_exit_net,
2371};
2372
2373static __init int net_inuse_init(void)
2374{
2375 if (register_pernet_subsys(&net_inuse_ops))
2376 panic("Cannot initialize net inuse counters");
2377
2378 return 0;
2379}
2380
2381core_initcall(net_inuse_init);
2382#else
Pavel Emelyanov1338d462008-03-28 16:38:43 -07002383static DEFINE_PER_CPU(struct prot_inuse, prot_inuse);
2384
Pavel Emelyanovc29a0bc2008-03-31 19:41:46 -07002385void sock_prot_inuse_add(struct net *net, struct proto *prot, int val)
Pavel Emelyanov1338d462008-03-28 16:38:43 -07002386{
Eric Dumazetd6d9ca02010-07-19 10:48:49 +00002387 __this_cpu_add(prot_inuse.val[prot->inuse_idx], val);
Pavel Emelyanov1338d462008-03-28 16:38:43 -07002388}
2389EXPORT_SYMBOL_GPL(sock_prot_inuse_add);
2390
Pavel Emelyanovc29a0bc2008-03-31 19:41:46 -07002391int sock_prot_inuse_get(struct net *net, struct proto *prot)
Pavel Emelyanov1338d462008-03-28 16:38:43 -07002392{
2393 int cpu, idx = prot->inuse_idx;
2394 int res = 0;
2395
2396 for_each_possible_cpu(cpu)
2397 res += per_cpu(prot_inuse, cpu).val[idx];
2398
2399 return res >= 0 ? res : 0;
2400}
2401EXPORT_SYMBOL_GPL(sock_prot_inuse_get);
Pavel Emelyanov70ee1152008-03-31 19:42:16 -07002402#endif
Pavel Emelyanov13ff3d62008-03-28 16:38:17 -07002403
2404static void assign_proto_idx(struct proto *prot)
2405{
2406 prot->inuse_idx = find_first_zero_bit(proto_inuse_idx, PROTO_INUSE_NR);
2407
2408 if (unlikely(prot->inuse_idx == PROTO_INUSE_NR - 1)) {
2409 printk(KERN_ERR "PROTO_INUSE_NR exhausted\n");
2410 return;
2411 }
2412
2413 set_bit(prot->inuse_idx, proto_inuse_idx);
2414}
2415
2416static void release_proto_idx(struct proto *prot)
2417{
2418 if (prot->inuse_idx != PROTO_INUSE_NR - 1)
2419 clear_bit(prot->inuse_idx, proto_inuse_idx);
2420}
2421#else
2422static inline void assign_proto_idx(struct proto *prot)
2423{
2424}
2425
2426static inline void release_proto_idx(struct proto *prot)
2427{
2428}
2429#endif
2430
Linus Torvalds1da177e2005-04-16 15:20:36 -07002431int proto_register(struct proto *prot, int alloc_slab)
2432{
Linus Torvalds1da177e2005-04-16 15:20:36 -07002433 if (alloc_slab) {
2434 prot->slab = kmem_cache_create(prot->name, prot->obj_size, 0,
Eric Dumazet271b72c2008-10-29 02:11:14 -07002435 SLAB_HWCACHE_ALIGN | prot->slab_flags,
2436 NULL);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002437
2438 if (prot->slab == NULL) {
2439 printk(KERN_CRIT "%s: Can't create sock SLAB cache!\n",
2440 prot->name);
Pavel Emelyanov60e76632008-03-28 16:39:10 -07002441 goto out;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002442 }
Arnaldo Carvalho de Melo2e6599c2005-06-18 22:46:52 -07002443
2444 if (prot->rsk_prot != NULL) {
Alexey Dobriyanfaf23422010-02-17 09:34:12 +00002445 prot->rsk_prot->slab_name = kasprintf(GFP_KERNEL, "request_sock_%s", prot->name);
Catalin Marinas7e56b5d2008-11-21 16:45:22 -08002446 if (prot->rsk_prot->slab_name == NULL)
Arnaldo Carvalho de Melo2e6599c2005-06-18 22:46:52 -07002447 goto out_free_sock_slab;
2448
Catalin Marinas7e56b5d2008-11-21 16:45:22 -08002449 prot->rsk_prot->slab = kmem_cache_create(prot->rsk_prot->slab_name,
Arnaldo Carvalho de Melo2e6599c2005-06-18 22:46:52 -07002450 prot->rsk_prot->obj_size, 0,
Paul Mundt20c2df82007-07-20 10:11:58 +09002451 SLAB_HWCACHE_ALIGN, NULL);
Arnaldo Carvalho de Melo2e6599c2005-06-18 22:46:52 -07002452
2453 if (prot->rsk_prot->slab == NULL) {
2454 printk(KERN_CRIT "%s: Can't create request sock SLAB cache!\n",
2455 prot->name);
2456 goto out_free_request_sock_slab_name;
2457 }
2458 }
Arnaldo Carvalho de Melo8feaf0c02005-08-09 20:09:30 -07002459
Arnaldo Carvalho de Melo6d6ee432005-12-13 23:25:19 -08002460 if (prot->twsk_prot != NULL) {
Alexey Dobriyanfaf23422010-02-17 09:34:12 +00002461 prot->twsk_prot->twsk_slab_name = kasprintf(GFP_KERNEL, "tw_sock_%s", prot->name);
Arnaldo Carvalho de Melo8feaf0c02005-08-09 20:09:30 -07002462
Catalin Marinas7e56b5d2008-11-21 16:45:22 -08002463 if (prot->twsk_prot->twsk_slab_name == NULL)
Arnaldo Carvalho de Melo8feaf0c02005-08-09 20:09:30 -07002464 goto out_free_request_sock_slab;
2465
Arnaldo Carvalho de Melo6d6ee432005-12-13 23:25:19 -08002466 prot->twsk_prot->twsk_slab =
Catalin Marinas7e56b5d2008-11-21 16:45:22 -08002467 kmem_cache_create(prot->twsk_prot->twsk_slab_name,
Arnaldo Carvalho de Melo6d6ee432005-12-13 23:25:19 -08002468 prot->twsk_prot->twsk_obj_size,
Eric Dumazet3ab5aee2008-11-16 19:40:17 -08002469 0,
2470 SLAB_HWCACHE_ALIGN |
2471 prot->slab_flags,
Paul Mundt20c2df82007-07-20 10:11:58 +09002472 NULL);
Arnaldo Carvalho de Melo6d6ee432005-12-13 23:25:19 -08002473 if (prot->twsk_prot->twsk_slab == NULL)
Arnaldo Carvalho de Melo8feaf0c02005-08-09 20:09:30 -07002474 goto out_free_timewait_sock_slab_name;
2475 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07002476 }
2477
Glauber Costa36b77a52011-12-16 00:51:59 +00002478 mutex_lock(&proto_list_mutex);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002479 list_add(&prot->node, &proto_list);
Pavel Emelyanov13ff3d62008-03-28 16:38:17 -07002480 assign_proto_idx(prot);
Glauber Costa36b77a52011-12-16 00:51:59 +00002481 mutex_unlock(&proto_list_mutex);
Pavel Emelyanovb733c002007-11-07 02:23:38 -08002482 return 0;
2483
Arnaldo Carvalho de Melo8feaf0c02005-08-09 20:09:30 -07002484out_free_timewait_sock_slab_name:
Catalin Marinas7e56b5d2008-11-21 16:45:22 -08002485 kfree(prot->twsk_prot->twsk_slab_name);
Arnaldo Carvalho de Melo8feaf0c02005-08-09 20:09:30 -07002486out_free_request_sock_slab:
2487 if (prot->rsk_prot && prot->rsk_prot->slab) {
2488 kmem_cache_destroy(prot->rsk_prot->slab);
2489 prot->rsk_prot->slab = NULL;
2490 }
Arnaldo Carvalho de Melo2e6599c2005-06-18 22:46:52 -07002491out_free_request_sock_slab_name:
Dan Carpenter72150e92010-03-06 01:04:45 +00002492 if (prot->rsk_prot)
2493 kfree(prot->rsk_prot->slab_name);
Arnaldo Carvalho de Melo2e6599c2005-06-18 22:46:52 -07002494out_free_sock_slab:
2495 kmem_cache_destroy(prot->slab);
2496 prot->slab = NULL;
Pavel Emelyanovb733c002007-11-07 02:23:38 -08002497out:
2498 return -ENOBUFS;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002499}
Linus Torvalds1da177e2005-04-16 15:20:36 -07002500EXPORT_SYMBOL(proto_register);
2501
2502void proto_unregister(struct proto *prot)
2503{
Glauber Costa36b77a52011-12-16 00:51:59 +00002504 mutex_lock(&proto_list_mutex);
Pavel Emelyanov13ff3d62008-03-28 16:38:17 -07002505 release_proto_idx(prot);
Patrick McHardy0a3f4352005-09-06 19:47:50 -07002506 list_del(&prot->node);
Glauber Costa36b77a52011-12-16 00:51:59 +00002507 mutex_unlock(&proto_list_mutex);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002508
2509 if (prot->slab != NULL) {
2510 kmem_cache_destroy(prot->slab);
2511 prot->slab = NULL;
2512 }
2513
Arnaldo Carvalho de Melo2e6599c2005-06-18 22:46:52 -07002514 if (prot->rsk_prot != NULL && prot->rsk_prot->slab != NULL) {
Arnaldo Carvalho de Melo2e6599c2005-06-18 22:46:52 -07002515 kmem_cache_destroy(prot->rsk_prot->slab);
Catalin Marinas7e56b5d2008-11-21 16:45:22 -08002516 kfree(prot->rsk_prot->slab_name);
Arnaldo Carvalho de Melo2e6599c2005-06-18 22:46:52 -07002517 prot->rsk_prot->slab = NULL;
2518 }
2519
Arnaldo Carvalho de Melo6d6ee432005-12-13 23:25:19 -08002520 if (prot->twsk_prot != NULL && prot->twsk_prot->twsk_slab != NULL) {
Arnaldo Carvalho de Melo6d6ee432005-12-13 23:25:19 -08002521 kmem_cache_destroy(prot->twsk_prot->twsk_slab);
Catalin Marinas7e56b5d2008-11-21 16:45:22 -08002522 kfree(prot->twsk_prot->twsk_slab_name);
Arnaldo Carvalho de Melo6d6ee432005-12-13 23:25:19 -08002523 prot->twsk_prot->twsk_slab = NULL;
Arnaldo Carvalho de Melo8feaf0c02005-08-09 20:09:30 -07002524 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07002525}
Linus Torvalds1da177e2005-04-16 15:20:36 -07002526EXPORT_SYMBOL(proto_unregister);
2527
2528#ifdef CONFIG_PROC_FS
Linus Torvalds1da177e2005-04-16 15:20:36 -07002529static void *proto_seq_start(struct seq_file *seq, loff_t *pos)
Glauber Costa36b77a52011-12-16 00:51:59 +00002530 __acquires(proto_list_mutex)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002531{
Glauber Costa36b77a52011-12-16 00:51:59 +00002532 mutex_lock(&proto_list_mutex);
Pavel Emelianov60f04382007-07-09 13:15:14 -07002533 return seq_list_start_head(&proto_list, *pos);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002534}
2535
2536static void *proto_seq_next(struct seq_file *seq, void *v, loff_t *pos)
2537{
Pavel Emelianov60f04382007-07-09 13:15:14 -07002538 return seq_list_next(v, &proto_list, pos);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002539}
2540
2541static void proto_seq_stop(struct seq_file *seq, void *v)
Glauber Costa36b77a52011-12-16 00:51:59 +00002542 __releases(proto_list_mutex)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002543{
Glauber Costa36b77a52011-12-16 00:51:59 +00002544 mutex_unlock(&proto_list_mutex);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002545}
2546
2547static char proto_method_implemented(const void *method)
2548{
2549 return method == NULL ? 'n' : 'y';
2550}
Glauber Costa180d8cd2011-12-11 21:47:02 +00002551static long sock_prot_memory_allocated(struct proto *proto)
2552{
2553 return proto->memory_allocated != NULL ? proto_memory_allocated(proto): -1L;
2554}
2555
2556static char *sock_prot_memory_pressure(struct proto *proto)
2557{
2558 return proto->memory_pressure != NULL ?
2559 proto_memory_pressure(proto) ? "yes" : "no" : "NI";
2560}
Linus Torvalds1da177e2005-04-16 15:20:36 -07002561
2562static void proto_seq_printf(struct seq_file *seq, struct proto *proto)
2563{
Glauber Costa180d8cd2011-12-11 21:47:02 +00002564
Eric Dumazet8d987e52010-11-09 23:24:26 +00002565 seq_printf(seq, "%-9s %4u %6d %6ld %-3s %6u %-3s %-10s "
Linus Torvalds1da177e2005-04-16 15:20:36 -07002566 "%2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c\n",
2567 proto->name,
2568 proto->obj_size,
Eric Dumazet14e943d2008-11-19 15:14:01 -08002569 sock_prot_inuse_get(seq_file_net(seq), proto),
Glauber Costa180d8cd2011-12-11 21:47:02 +00002570 sock_prot_memory_allocated(proto),
2571 sock_prot_memory_pressure(proto),
Linus Torvalds1da177e2005-04-16 15:20:36 -07002572 proto->max_header,
2573 proto->slab == NULL ? "no" : "yes",
2574 module_name(proto->owner),
2575 proto_method_implemented(proto->close),
2576 proto_method_implemented(proto->connect),
2577 proto_method_implemented(proto->disconnect),
2578 proto_method_implemented(proto->accept),
2579 proto_method_implemented(proto->ioctl),
2580 proto_method_implemented(proto->init),
2581 proto_method_implemented(proto->destroy),
2582 proto_method_implemented(proto->shutdown),
2583 proto_method_implemented(proto->setsockopt),
2584 proto_method_implemented(proto->getsockopt),
2585 proto_method_implemented(proto->sendmsg),
2586 proto_method_implemented(proto->recvmsg),
2587 proto_method_implemented(proto->sendpage),
2588 proto_method_implemented(proto->bind),
2589 proto_method_implemented(proto->backlog_rcv),
2590 proto_method_implemented(proto->hash),
2591 proto_method_implemented(proto->unhash),
2592 proto_method_implemented(proto->get_port),
2593 proto_method_implemented(proto->enter_memory_pressure));
2594}
2595
2596static int proto_seq_show(struct seq_file *seq, void *v)
2597{
Pavel Emelianov60f04382007-07-09 13:15:14 -07002598 if (v == &proto_list)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002599 seq_printf(seq, "%-9s %-4s %-8s %-6s %-5s %-7s %-4s %-10s %s",
2600 "protocol",
2601 "size",
2602 "sockets",
2603 "memory",
2604 "press",
2605 "maxhdr",
2606 "slab",
2607 "module",
2608 "cl co di ac io in de sh ss gs se re sp bi br ha uh gp em\n");
2609 else
Pavel Emelianov60f04382007-07-09 13:15:14 -07002610 proto_seq_printf(seq, list_entry(v, struct proto, node));
Linus Torvalds1da177e2005-04-16 15:20:36 -07002611 return 0;
2612}
2613
Stephen Hemmingerf6908082007-03-12 14:34:29 -07002614static const struct seq_operations proto_seq_ops = {
Linus Torvalds1da177e2005-04-16 15:20:36 -07002615 .start = proto_seq_start,
2616 .next = proto_seq_next,
2617 .stop = proto_seq_stop,
2618 .show = proto_seq_show,
2619};
2620
2621static int proto_seq_open(struct inode *inode, struct file *file)
2622{
Eric Dumazet14e943d2008-11-19 15:14:01 -08002623 return seq_open_net(inode, file, &proto_seq_ops,
2624 sizeof(struct seq_net_private));
Linus Torvalds1da177e2005-04-16 15:20:36 -07002625}
2626
Arjan van de Ven9a321442007-02-12 00:55:35 -08002627static const struct file_operations proto_seq_fops = {
Linus Torvalds1da177e2005-04-16 15:20:36 -07002628 .owner = THIS_MODULE,
2629 .open = proto_seq_open,
2630 .read = seq_read,
2631 .llseek = seq_lseek,
Eric Dumazet14e943d2008-11-19 15:14:01 -08002632 .release = seq_release_net,
2633};
2634
2635static __net_init int proto_init_net(struct net *net)
2636{
2637 if (!proc_net_fops_create(net, "protocols", S_IRUGO, &proto_seq_fops))
2638 return -ENOMEM;
2639
2640 return 0;
2641}
2642
2643static __net_exit void proto_exit_net(struct net *net)
2644{
2645 proc_net_remove(net, "protocols");
2646}
2647
2648
2649static __net_initdata struct pernet_operations proto_net_ops = {
2650 .init = proto_init_net,
2651 .exit = proto_exit_net,
Linus Torvalds1da177e2005-04-16 15:20:36 -07002652};
2653
2654static int __init proto_init(void)
2655{
Eric Dumazet14e943d2008-11-19 15:14:01 -08002656 return register_pernet_subsys(&proto_net_ops);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002657}
2658
2659subsys_initcall(proto_init);
2660
2661#endif /* PROC_FS */