blob: 0b39e7ae43837e602ff473b641675fbf22b83b5c [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
Joe Perchese005d192012-05-16 19:58:40 +000092#define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
93
Randy Dunlap4fc268d2006-01-11 12:17:47 -080094#include <linux/capability.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070095#include <linux/errno.h>
Richard Cochrancb820f82013-07-19 19:40:09 +020096#include <linux/errqueue.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070097#include <linux/types.h>
98#include <linux/socket.h>
99#include <linux/in.h>
100#include <linux/kernel.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -0700101#include <linux/module.h>
102#include <linux/proc_fs.h>
103#include <linux/seq_file.h>
104#include <linux/sched.h>
105#include <linux/timer.h>
106#include <linux/string.h>
107#include <linux/sockios.h>
108#include <linux/net.h>
109#include <linux/mm.h>
110#include <linux/slab.h>
111#include <linux/interrupt.h>
112#include <linux/poll.h>
113#include <linux/tcp.h>
114#include <linux/init.h>
Al Viroa1f8e7f72006-10-19 16:08:53 -0400115#include <linux/highmem.h>
Eric W. Biederman3f551f92010-06-13 03:28:59 +0000116#include <linux/user_namespace.h>
Ingo Molnarc5905af2012-02-24 08:31:31 +0100117#include <linux/static_key.h>
David S. Miller3969eb32012-01-09 13:44:23 -0800118#include <linux/memcontrol.h>
David S. Miller8c1ae102012-05-03 02:25:55 -0400119#include <linux/prefetch.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -0700120
121#include <asm/uaccess.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -0700122
123#include <linux/netdevice.h>
124#include <net/protocol.h>
125#include <linux/skbuff.h>
Eric W. Biederman457c4cb2007-09-12 12:01:34 +0200126#include <net/net_namespace.h>
Arnaldo Carvalho de Melo2e6599c2005-06-18 22:46:52 -0700127#include <net/request_sock.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -0700128#include <net/sock.h>
Patrick Ohly20d49472009-02-12 05:03:38 +0000129#include <linux/net_tstamp.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -0700130#include <net/xfrm.h>
131#include <linux/ipsec.h>
Herbert Xuf8451722010-05-24 00:12:34 -0700132#include <net/cls_cgroup.h>
Neil Horman5bc14212011-11-22 05:10:51 +0000133#include <net/netprio_cgroup.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -0700134
135#include <linux/filter.h>
136
Satoru Moriya3847ce32011-06-17 12:00:03 +0000137#include <trace/events/sock.h>
138
Linus Torvalds1da177e2005-04-16 15:20:36 -0700139#ifdef CONFIG_INET
140#include <net/tcp.h>
141#endif
142
Eliezer Tamir076bb0c2013-07-10 17:13:17 +0300143#include <net/busy_poll.h>
Eliezer Tamir06021292013-06-10 11:39:50 +0300144
Glauber Costa36b77a52011-12-16 00:51:59 +0000145static DEFINE_MUTEX(proto_list_mutex);
Glauber Costad1a4c0b2011-12-11 21:47:04 +0000146static LIST_HEAD(proto_list);
147
Andrew Mortonc255a452012-07-31 16:43:02 -0700148#ifdef CONFIG_MEMCG_KMEM
Glauber Costa1d62e432012-04-09 19:36:33 -0300149int mem_cgroup_sockets_init(struct mem_cgroup *memcg, struct cgroup_subsys *ss)
Glauber Costad1a4c0b2011-12-11 21:47:04 +0000150{
151 struct proto *proto;
152 int ret = 0;
153
Glauber Costa36b77a52011-12-16 00:51:59 +0000154 mutex_lock(&proto_list_mutex);
Glauber Costad1a4c0b2011-12-11 21:47:04 +0000155 list_for_each_entry(proto, &proto_list, node) {
156 if (proto->init_cgroup) {
Glauber Costa1d62e432012-04-09 19:36:33 -0300157 ret = proto->init_cgroup(memcg, ss);
Glauber Costad1a4c0b2011-12-11 21:47:04 +0000158 if (ret)
159 goto out;
160 }
161 }
162
Glauber Costa36b77a52011-12-16 00:51:59 +0000163 mutex_unlock(&proto_list_mutex);
Glauber Costad1a4c0b2011-12-11 21:47:04 +0000164 return ret;
165out:
166 list_for_each_entry_continue_reverse(proto, &proto_list, node)
167 if (proto->destroy_cgroup)
Glauber Costa1d62e432012-04-09 19:36:33 -0300168 proto->destroy_cgroup(memcg);
Glauber Costa36b77a52011-12-16 00:51:59 +0000169 mutex_unlock(&proto_list_mutex);
Glauber Costad1a4c0b2011-12-11 21:47:04 +0000170 return ret;
171}
172
Glauber Costa1d62e432012-04-09 19:36:33 -0300173void mem_cgroup_sockets_destroy(struct mem_cgroup *memcg)
Glauber Costad1a4c0b2011-12-11 21:47:04 +0000174{
175 struct proto *proto;
176
Glauber Costa36b77a52011-12-16 00:51:59 +0000177 mutex_lock(&proto_list_mutex);
Glauber Costad1a4c0b2011-12-11 21:47:04 +0000178 list_for_each_entry_reverse(proto, &proto_list, node)
179 if (proto->destroy_cgroup)
Glauber Costa1d62e432012-04-09 19:36:33 -0300180 proto->destroy_cgroup(memcg);
Glauber Costa36b77a52011-12-16 00:51:59 +0000181 mutex_unlock(&proto_list_mutex);
Glauber Costad1a4c0b2011-12-11 21:47:04 +0000182}
183#endif
184
Ingo Molnarda21f242006-07-03 00:25:12 -0700185/*
186 * Each address family might have different locking rules, so we have
187 * one slock key per address family:
188 */
Ingo Molnara5b5bb92006-07-03 00:25:35 -0700189static struct lock_class_key af_family_keys[AF_MAX];
190static struct lock_class_key af_family_slock_keys[AF_MAX];
191
stephen hemmingercbda4ea2013-02-22 07:59:10 +0000192#if defined(CONFIG_MEMCG_KMEM)
Ingo Molnarc5905af2012-02-24 08:31:31 +0100193struct static_key memcg_socket_limit_enabled;
Glauber Costae1aab162011-12-11 21:47:03 +0000194EXPORT_SYMBOL(memcg_socket_limit_enabled);
stephen hemmingercbda4ea2013-02-22 07:59:10 +0000195#endif
Glauber Costae1aab162011-12-11 21:47:03 +0000196
Ingo Molnara5b5bb92006-07-03 00:25:35 -0700197/*
198 * Make lock validator output more readable. (we pre-construct these
199 * strings build-time, so that runtime initialization of socket
200 * locks is fast):
201 */
Jan Engelhardt36cbd3d2009-08-05 10:42:58 -0700202static const char *const af_family_key_strings[AF_MAX+1] = {
Ingo Molnara5b5bb92006-07-03 00:25:35 -0700203 "sk_lock-AF_UNSPEC", "sk_lock-AF_UNIX" , "sk_lock-AF_INET" ,
204 "sk_lock-AF_AX25" , "sk_lock-AF_IPX" , "sk_lock-AF_APPLETALK",
205 "sk_lock-AF_NETROM", "sk_lock-AF_BRIDGE" , "sk_lock-AF_ATMPVC" ,
206 "sk_lock-AF_X25" , "sk_lock-AF_INET6" , "sk_lock-AF_ROSE" ,
207 "sk_lock-AF_DECnet", "sk_lock-AF_NETBEUI" , "sk_lock-AF_SECURITY" ,
208 "sk_lock-AF_KEY" , "sk_lock-AF_NETLINK" , "sk_lock-AF_PACKET" ,
209 "sk_lock-AF_ASH" , "sk_lock-AF_ECONET" , "sk_lock-AF_ATMSVC" ,
Andy Grovercbd151b2009-02-26 23:43:19 -0800210 "sk_lock-AF_RDS" , "sk_lock-AF_SNA" , "sk_lock-AF_IRDA" ,
Ingo Molnara5b5bb92006-07-03 00:25:35 -0700211 "sk_lock-AF_PPPOX" , "sk_lock-AF_WANPIPE" , "sk_lock-AF_LLC" ,
Oliver Hartkoppcd05acf2007-12-16 15:59:24 -0800212 "sk_lock-27" , "sk_lock-28" , "sk_lock-AF_CAN" ,
David Howells17926a72007-04-26 15:48:28 -0700213 "sk_lock-AF_TIPC" , "sk_lock-AF_BLUETOOTH", "sk_lock-IUCV" ,
Remi Denis-Courmontbce7b152008-09-22 19:51:15 -0700214 "sk_lock-AF_RXRPC" , "sk_lock-AF_ISDN" , "sk_lock-AF_PHONET" ,
Miloslav Trmač6f107b52010-12-08 14:35:34 +0800215 "sk_lock-AF_IEEE802154", "sk_lock-AF_CAIF" , "sk_lock-AF_ALG" ,
Federico Vaga456db6a2013-05-28 05:02:44 +0000216 "sk_lock-AF_NFC" , "sk_lock-AF_VSOCK" , "sk_lock-AF_MAX"
Ingo Molnara5b5bb92006-07-03 00:25:35 -0700217};
Jan Engelhardt36cbd3d2009-08-05 10:42:58 -0700218static const char *const af_family_slock_key_strings[AF_MAX+1] = {
Ingo Molnara5b5bb92006-07-03 00:25:35 -0700219 "slock-AF_UNSPEC", "slock-AF_UNIX" , "slock-AF_INET" ,
220 "slock-AF_AX25" , "slock-AF_IPX" , "slock-AF_APPLETALK",
221 "slock-AF_NETROM", "slock-AF_BRIDGE" , "slock-AF_ATMPVC" ,
222 "slock-AF_X25" , "slock-AF_INET6" , "slock-AF_ROSE" ,
223 "slock-AF_DECnet", "slock-AF_NETBEUI" , "slock-AF_SECURITY" ,
224 "slock-AF_KEY" , "slock-AF_NETLINK" , "slock-AF_PACKET" ,
225 "slock-AF_ASH" , "slock-AF_ECONET" , "slock-AF_ATMSVC" ,
Andy Grovercbd151b2009-02-26 23:43:19 -0800226 "slock-AF_RDS" , "slock-AF_SNA" , "slock-AF_IRDA" ,
Ingo Molnara5b5bb92006-07-03 00:25:35 -0700227 "slock-AF_PPPOX" , "slock-AF_WANPIPE" , "slock-AF_LLC" ,
Oliver Hartkoppcd05acf2007-12-16 15:59:24 -0800228 "slock-27" , "slock-28" , "slock-AF_CAN" ,
David Howells17926a72007-04-26 15:48:28 -0700229 "slock-AF_TIPC" , "slock-AF_BLUETOOTH", "slock-AF_IUCV" ,
Remi Denis-Courmontbce7b152008-09-22 19:51:15 -0700230 "slock-AF_RXRPC" , "slock-AF_ISDN" , "slock-AF_PHONET" ,
Miloslav Trmač6f107b52010-12-08 14:35:34 +0800231 "slock-AF_IEEE802154", "slock-AF_CAIF" , "slock-AF_ALG" ,
Federico Vaga456db6a2013-05-28 05:02:44 +0000232 "slock-AF_NFC" , "slock-AF_VSOCK" ,"slock-AF_MAX"
Ingo Molnara5b5bb92006-07-03 00:25:35 -0700233};
Jan Engelhardt36cbd3d2009-08-05 10:42:58 -0700234static const char *const af_family_clock_key_strings[AF_MAX+1] = {
Peter Zijlstra443aef02007-07-19 01:49:00 -0700235 "clock-AF_UNSPEC", "clock-AF_UNIX" , "clock-AF_INET" ,
236 "clock-AF_AX25" , "clock-AF_IPX" , "clock-AF_APPLETALK",
237 "clock-AF_NETROM", "clock-AF_BRIDGE" , "clock-AF_ATMPVC" ,
238 "clock-AF_X25" , "clock-AF_INET6" , "clock-AF_ROSE" ,
239 "clock-AF_DECnet", "clock-AF_NETBEUI" , "clock-AF_SECURITY" ,
240 "clock-AF_KEY" , "clock-AF_NETLINK" , "clock-AF_PACKET" ,
241 "clock-AF_ASH" , "clock-AF_ECONET" , "clock-AF_ATMSVC" ,
Andy Grovercbd151b2009-02-26 23:43:19 -0800242 "clock-AF_RDS" , "clock-AF_SNA" , "clock-AF_IRDA" ,
Peter Zijlstra443aef02007-07-19 01:49:00 -0700243 "clock-AF_PPPOX" , "clock-AF_WANPIPE" , "clock-AF_LLC" ,
Oliver Hartkoppb4942af2008-07-23 14:06:04 -0700244 "clock-27" , "clock-28" , "clock-AF_CAN" ,
David Howellse51f8022007-07-21 19:30:16 -0700245 "clock-AF_TIPC" , "clock-AF_BLUETOOTH", "clock-AF_IUCV" ,
Remi Denis-Courmontbce7b152008-09-22 19:51:15 -0700246 "clock-AF_RXRPC" , "clock-AF_ISDN" , "clock-AF_PHONET" ,
Miloslav Trmač6f107b52010-12-08 14:35:34 +0800247 "clock-AF_IEEE802154", "clock-AF_CAIF" , "clock-AF_ALG" ,
Federico Vaga456db6a2013-05-28 05:02:44 +0000248 "clock-AF_NFC" , "clock-AF_VSOCK" , "clock-AF_MAX"
Peter Zijlstra443aef02007-07-19 01:49:00 -0700249};
Ingo Molnarda21f242006-07-03 00:25:12 -0700250
251/*
252 * sk_callback_lock locking rules are per-address-family,
253 * so split the lock classes by using a per-AF key:
254 */
255static struct lock_class_key af_callback_keys[AF_MAX];
256
Linus Torvalds1da177e2005-04-16 15:20:36 -0700257/* Take into consideration the size of the struct sk_buff overhead in the
258 * determination of these values, since that is non-constant across
259 * platforms. This makes socket queueing behavior and performance
260 * not depend upon such differences.
261 */
262#define _SK_MEM_PACKETS 256
Eric Dumazet87fb4b72011-10-13 07:28:54 +0000263#define _SK_MEM_OVERHEAD SKB_TRUESIZE(256)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700264#define SK_WMEM_MAX (_SK_MEM_OVERHEAD * _SK_MEM_PACKETS)
265#define SK_RMEM_MAX (_SK_MEM_OVERHEAD * _SK_MEM_PACKETS)
266
267/* Run time adjustable parameters. */
Brian Haleyab32ea52006-09-22 14:15:41 -0700268__u32 sysctl_wmem_max __read_mostly = SK_WMEM_MAX;
Hans Schillstrom6d8ebc82012-04-30 08:13:50 +0200269EXPORT_SYMBOL(sysctl_wmem_max);
Brian Haleyab32ea52006-09-22 14:15:41 -0700270__u32 sysctl_rmem_max __read_mostly = SK_RMEM_MAX;
Hans Schillstrom6d8ebc82012-04-30 08:13:50 +0200271EXPORT_SYMBOL(sysctl_rmem_max);
Brian Haleyab32ea52006-09-22 14:15:41 -0700272__u32 sysctl_wmem_default __read_mostly = SK_WMEM_MAX;
273__u32 sysctl_rmem_default __read_mostly = SK_RMEM_MAX;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700274
Lucas De Marchi25985ed2011-03-30 22:57:33 -0300275/* Maximal space eaten by iovec or ancillary data plus some space */
Brian Haleyab32ea52006-09-22 14:15:41 -0700276int sysctl_optmem_max __read_mostly = sizeof(unsigned long)*(2*UIO_MAXIOV+512);
Eric Dumazet2a915252009-05-27 11:30:05 +0000277EXPORT_SYMBOL(sysctl_optmem_max);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700278
Mel Gormanc93bdd02012-07-31 16:44:19 -0700279struct static_key memalloc_socks = STATIC_KEY_INIT_FALSE;
280EXPORT_SYMBOL_GPL(memalloc_socks);
281
Mel Gorman7cb02402012-07-31 16:44:16 -0700282/**
283 * sk_set_memalloc - sets %SOCK_MEMALLOC
284 * @sk: socket to set it on
285 *
286 * Set %SOCK_MEMALLOC on a socket for access to emergency reserves.
287 * It's the responsibility of the admin to adjust min_free_kbytes
288 * to meet the requirements
289 */
290void sk_set_memalloc(struct sock *sk)
291{
292 sock_set_flag(sk, SOCK_MEMALLOC);
293 sk->sk_allocation |= __GFP_MEMALLOC;
Mel Gormanc93bdd02012-07-31 16:44:19 -0700294 static_key_slow_inc(&memalloc_socks);
Mel Gorman7cb02402012-07-31 16:44:16 -0700295}
296EXPORT_SYMBOL_GPL(sk_set_memalloc);
297
298void sk_clear_memalloc(struct sock *sk)
299{
300 sock_reset_flag(sk, SOCK_MEMALLOC);
301 sk->sk_allocation &= ~__GFP_MEMALLOC;
Mel Gormanc93bdd02012-07-31 16:44:19 -0700302 static_key_slow_dec(&memalloc_socks);
Mel Gormanc76562b2012-07-31 16:44:41 -0700303
304 /*
305 * SOCK_MEMALLOC is allowed to ignore rmem limits to ensure forward
306 * progress of swapping. However, if SOCK_MEMALLOC is cleared while
307 * it has rmem allocations there is a risk that the user of the
308 * socket cannot make forward progress due to exceeding the rmem
309 * limits. By rights, sk_clear_memalloc() should only be called
310 * on sockets being torn down but warn and reset the accounting if
311 * that assumption breaks.
312 */
313 if (WARN_ON(sk->sk_forward_alloc))
314 sk_mem_reclaim(sk);
Mel Gorman7cb02402012-07-31 16:44:16 -0700315}
316EXPORT_SYMBOL_GPL(sk_clear_memalloc);
317
Mel Gormanb4b9e352012-07-31 16:44:26 -0700318int __sk_backlog_rcv(struct sock *sk, struct sk_buff *skb)
319{
320 int ret;
321 unsigned long pflags = current->flags;
322
323 /* these should have been dropped before queueing */
324 BUG_ON(!sock_flag(sk, SOCK_MEMALLOC));
325
326 current->flags |= PF_MEMALLOC;
327 ret = sk->sk_backlog_rcv(sk, skb);
328 tsk_restore_flags(current, pflags, PF_MEMALLOC);
329
330 return ret;
331}
332EXPORT_SYMBOL(__sk_backlog_rcv);
333
Linus Torvalds1da177e2005-04-16 15:20:36 -0700334static int sock_set_timeout(long *timeo_p, char __user *optval, int optlen)
335{
336 struct timeval tv;
337
338 if (optlen < sizeof(tv))
339 return -EINVAL;
340 if (copy_from_user(&tv, optval, sizeof(tv)))
341 return -EFAULT;
Vasily Averinba780732007-05-24 16:58:54 -0700342 if (tv.tv_usec < 0 || tv.tv_usec >= USEC_PER_SEC)
343 return -EDOM;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700344
Vasily Averinba780732007-05-24 16:58:54 -0700345 if (tv.tv_sec < 0) {
Andrew Morton6f11df82007-07-09 13:16:00 -0700346 static int warned __read_mostly;
347
Vasily Averinba780732007-05-24 16:58:54 -0700348 *timeo_p = 0;
Ilpo Järvinen50aab542008-05-02 16:20:10 -0700349 if (warned < 10 && net_ratelimit()) {
Vasily Averinba780732007-05-24 16:58:54 -0700350 warned++;
Joe Perchese005d192012-05-16 19:58:40 +0000351 pr_info("%s: `%s' (pid %d) tries to set negative timeout\n",
352 __func__, current->comm, task_pid_nr(current));
Ilpo Järvinen50aab542008-05-02 16:20:10 -0700353 }
Vasily Averinba780732007-05-24 16:58:54 -0700354 return 0;
355 }
Linus Torvalds1da177e2005-04-16 15:20:36 -0700356 *timeo_p = MAX_SCHEDULE_TIMEOUT;
357 if (tv.tv_sec == 0 && tv.tv_usec == 0)
358 return 0;
359 if (tv.tv_sec < (MAX_SCHEDULE_TIMEOUT/HZ - 1))
360 *timeo_p = tv.tv_sec*HZ + (tv.tv_usec+(1000000/HZ-1))/(1000000/HZ);
361 return 0;
362}
363
364static void sock_warn_obsolete_bsdism(const char *name)
365{
366 static int warned;
367 static char warncomm[TASK_COMM_LEN];
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +0900368 if (strcmp(warncomm, current->comm) && warned < 5) {
369 strcpy(warncomm, current->comm);
Joe Perchese005d192012-05-16 19:58:40 +0000370 pr_warn("process `%s' is using obsolete %s SO_BSDCOMPAT\n",
371 warncomm, name);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700372 warned++;
373 }
374}
375
Eric Dumazet08e29af2011-11-28 12:04:18 +0000376#define SK_FLAGS_TIMESTAMP ((1UL << SOCK_TIMESTAMP) | (1UL << SOCK_TIMESTAMPING_RX_SOFTWARE))
377
378static void sock_disable_timestamp(struct sock *sk, unsigned long flags)
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +0900379{
Eric Dumazet08e29af2011-11-28 12:04:18 +0000380 if (sk->sk_flags & flags) {
381 sk->sk_flags &= ~flags;
382 if (!(sk->sk_flags & SK_FLAGS_TIMESTAMP))
Patrick Ohly20d49472009-02-12 05:03:38 +0000383 net_disable_timestamp();
Linus Torvalds1da177e2005-04-16 15:20:36 -0700384 }
385}
386
387
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800388int sock_queue_rcv_skb(struct sock *sk, struct sk_buff *skb)
389{
Eric Dumazet766e90372009-10-14 20:40:11 -0700390 int err;
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800391 int skb_len;
Neil Horman3b885782009-10-12 13:26:31 -0700392 unsigned long flags;
393 struct sk_buff_head *list = &sk->sk_receive_queue;
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800394
Eric Dumazet0fd7bac2011-12-21 07:11:44 +0000395 if (atomic_read(&sk->sk_rmem_alloc) >= sk->sk_rcvbuf) {
Eric Dumazet766e90372009-10-14 20:40:11 -0700396 atomic_inc(&sk->sk_drops);
Satoru Moriya3847ce32011-06-17 12:00:03 +0000397 trace_sock_rcvqueue_full(sk, skb);
Eric Dumazet766e90372009-10-14 20:40:11 -0700398 return -ENOMEM;
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800399 }
400
Dmitry Mishinfda9ef52006-08-31 15:28:39 -0700401 err = sk_filter(sk, skb);
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800402 if (err)
Eric Dumazet766e90372009-10-14 20:40:11 -0700403 return err;
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800404
Mel Gormanc76562b2012-07-31 16:44:41 -0700405 if (!sk_rmem_schedule(sk, skb, skb->truesize)) {
Eric Dumazet766e90372009-10-14 20:40:11 -0700406 atomic_inc(&sk->sk_drops);
407 return -ENOBUFS;
Hideo Aoki3ab224b2007-12-31 00:11:19 -0800408 }
409
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800410 skb->dev = NULL;
411 skb_set_owner_r(skb, sk);
David S. Miller49ad9592008-12-17 22:11:38 -0800412
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800413 /* Cache the SKB length before we tack it onto the receive
414 * queue. Once it is added it no longer belongs to us and
415 * may be freed by other threads of control pulling packets
416 * from the queue.
417 */
418 skb_len = skb->len;
419
Eric Dumazet7fee2262010-05-11 23:19:48 +0000420 /* we escape from rcu protected region, make sure we dont leak
421 * a norefcounted dst
422 */
423 skb_dst_force(skb);
424
Neil Horman3b885782009-10-12 13:26:31 -0700425 spin_lock_irqsave(&list->lock, flags);
426 skb->dropcount = atomic_read(&sk->sk_drops);
427 __skb_queue_tail(list, skb);
428 spin_unlock_irqrestore(&list->lock, flags);
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800429
430 if (!sock_flag(sk, SOCK_DEAD))
431 sk->sk_data_ready(sk, skb_len);
Eric Dumazet766e90372009-10-14 20:40:11 -0700432 return 0;
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800433}
434EXPORT_SYMBOL(sock_queue_rcv_skb);
435
Arnaldo Carvalho de Melo58a5a7b2006-11-16 14:06:06 -0200436int sk_receive_skb(struct sock *sk, struct sk_buff *skb, const int nested)
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800437{
438 int rc = NET_RX_SUCCESS;
439
Dmitry Mishinfda9ef52006-08-31 15:28:39 -0700440 if (sk_filter(sk, skb))
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800441 goto discard_and_relse;
442
443 skb->dev = NULL;
444
Eric Dumazetf545a382012-04-22 23:34:26 +0000445 if (sk_rcvqueues_full(sk, skb, sk->sk_rcvbuf)) {
Eric Dumazetc3774112010-04-27 15:13:20 -0700446 atomic_inc(&sk->sk_drops);
447 goto discard_and_relse;
448 }
Arnaldo Carvalho de Melo58a5a7b2006-11-16 14:06:06 -0200449 if (nested)
450 bh_lock_sock_nested(sk);
451 else
452 bh_lock_sock(sk);
Ingo Molnara5b5bb92006-07-03 00:25:35 -0700453 if (!sock_owned_by_user(sk)) {
454 /*
455 * trylock + unlock semantics:
456 */
457 mutex_acquire(&sk->sk_lock.dep_map, 0, 1, _RET_IP_);
458
Peter Zijlstrac57943a2008-10-07 14:18:42 -0700459 rc = sk_backlog_rcv(sk, skb);
Ingo Molnara5b5bb92006-07-03 00:25:35 -0700460
461 mutex_release(&sk->sk_lock.dep_map, 1, _RET_IP_);
Eric Dumazetf545a382012-04-22 23:34:26 +0000462 } else if (sk_add_backlog(sk, skb, sk->sk_rcvbuf)) {
Zhu Yi8eae9392010-03-04 18:01:40 +0000463 bh_unlock_sock(sk);
464 atomic_inc(&sk->sk_drops);
465 goto discard_and_relse;
466 }
467
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800468 bh_unlock_sock(sk);
469out:
470 sock_put(sk);
471 return rc;
472discard_and_relse:
473 kfree_skb(skb);
474 goto out;
475}
476EXPORT_SYMBOL(sk_receive_skb);
477
Krishna Kumarea94ff32009-10-19 23:46:45 +0000478void sk_reset_txq(struct sock *sk)
479{
480 sk_tx_queue_clear(sk);
481}
482EXPORT_SYMBOL(sk_reset_txq);
483
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800484struct dst_entry *__sk_dst_check(struct sock *sk, u32 cookie)
485{
Eric Dumazetb6c67122010-04-08 23:03:29 +0000486 struct dst_entry *dst = __sk_dst_get(sk);
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800487
488 if (dst && dst->obsolete && dst->ops->check(dst, cookie) == NULL) {
Krishna Kumare022f0b2009-10-19 23:46:20 +0000489 sk_tx_queue_clear(sk);
Stephen Hemmingera9b3cd72011-08-01 16:19:00 +0000490 RCU_INIT_POINTER(sk->sk_dst_cache, NULL);
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800491 dst_release(dst);
492 return NULL;
493 }
494
495 return dst;
496}
497EXPORT_SYMBOL(__sk_dst_check);
498
499struct dst_entry *sk_dst_check(struct sock *sk, u32 cookie)
500{
501 struct dst_entry *dst = sk_dst_get(sk);
502
503 if (dst && dst->obsolete && dst->ops->check(dst, cookie) == NULL) {
504 sk_dst_reset(sk);
505 dst_release(dst);
506 return NULL;
507 }
508
509 return dst;
510}
511EXPORT_SYMBOL(sk_dst_check);
512
Brian Haleyc91f6df2012-11-26 05:21:08 +0000513static int sock_setbindtodevice(struct sock *sk, char __user *optval,
514 int optlen)
David S. Miller48788092007-09-14 16:41:03 -0700515{
516 int ret = -ENOPROTOOPT;
517#ifdef CONFIG_NETDEVICES
YOSHIFUJI Hideaki3b1e0a62008-03-26 02:26:21 +0900518 struct net *net = sock_net(sk);
David S. Miller48788092007-09-14 16:41:03 -0700519 char devname[IFNAMSIZ];
520 int index;
521
522 /* Sorry... */
523 ret = -EPERM;
Eric W. Biederman5e1fccc2012-11-16 03:03:04 +0000524 if (!ns_capable(net->user_ns, CAP_NET_RAW))
David S. Miller48788092007-09-14 16:41:03 -0700525 goto out;
526
527 ret = -EINVAL;
528 if (optlen < 0)
529 goto out;
530
531 /* Bind this socket to a particular device like "eth0",
532 * as specified in the passed interface name. If the
533 * name is "" or the option length is zero the socket
534 * is not bound.
535 */
536 if (optlen > IFNAMSIZ - 1)
537 optlen = IFNAMSIZ - 1;
538 memset(devname, 0, sizeof(devname));
539
540 ret = -EFAULT;
541 if (copy_from_user(devname, optval, optlen))
542 goto out;
543
David S. Miller000ba2e2009-11-05 22:37:11 -0800544 index = 0;
545 if (devname[0] != '\0') {
Eric Dumazetbf8e56b2009-11-05 21:03:39 -0800546 struct net_device *dev;
David S. Miller48788092007-09-14 16:41:03 -0700547
Eric Dumazetbf8e56b2009-11-05 21:03:39 -0800548 rcu_read_lock();
549 dev = dev_get_by_name_rcu(net, devname);
550 if (dev)
551 index = dev->ifindex;
552 rcu_read_unlock();
David S. Miller48788092007-09-14 16:41:03 -0700553 ret = -ENODEV;
554 if (!dev)
555 goto out;
David S. Miller48788092007-09-14 16:41:03 -0700556 }
557
558 lock_sock(sk);
559 sk->sk_bound_dev_if = index;
560 sk_dst_reset(sk);
561 release_sock(sk);
562
563 ret = 0;
564
565out:
566#endif
567
568 return ret;
569}
570
Brian Haleyc91f6df2012-11-26 05:21:08 +0000571static int sock_getbindtodevice(struct sock *sk, char __user *optval,
572 int __user *optlen, int len)
573{
574 int ret = -ENOPROTOOPT;
575#ifdef CONFIG_NETDEVICES
576 struct net *net = sock_net(sk);
Brian Haleyc91f6df2012-11-26 05:21:08 +0000577 char devname[IFNAMSIZ];
Brian Haleyc91f6df2012-11-26 05:21:08 +0000578
579 if (sk->sk_bound_dev_if == 0) {
580 len = 0;
581 goto zero;
582 }
583
584 ret = -EINVAL;
585 if (len < IFNAMSIZ)
586 goto out;
587
Nicolas Schichan5dbe7c12013-06-26 17:23:42 +0200588 ret = netdev_get_name(net, devname, sk->sk_bound_dev_if);
589 if (ret)
Brian Haleyc91f6df2012-11-26 05:21:08 +0000590 goto out;
Brian Haleyc91f6df2012-11-26 05:21:08 +0000591
592 len = strlen(devname) + 1;
593
594 ret = -EFAULT;
595 if (copy_to_user(optval, devname, len))
596 goto out;
597
598zero:
599 ret = -EFAULT;
600 if (put_user(len, optlen))
601 goto out;
602
603 ret = 0;
604
605out:
606#endif
607
608 return ret;
609}
610
Pavel Emelyanovc0ef8772007-11-15 03:03:19 -0800611static inline void sock_valbool_flag(struct sock *sk, int bit, int valbool)
612{
613 if (valbool)
614 sock_set_flag(sk, bit);
615 else
616 sock_reset_flag(sk, bit);
617}
618
Linus Torvalds1da177e2005-04-16 15:20:36 -0700619/*
620 * This is meant for all protocols to use and covers goings on
621 * at the socket level. Everything here is generic.
622 */
623
624int sock_setsockopt(struct socket *sock, int level, int optname,
David S. Millerb7058842009-09-30 16:12:20 -0700625 char __user *optval, unsigned int optlen)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700626{
Eric Dumazet2a915252009-05-27 11:30:05 +0000627 struct sock *sk = sock->sk;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700628 int val;
629 int valbool;
630 struct linger ling;
631 int ret = 0;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +0900632
Linus Torvalds1da177e2005-04-16 15:20:36 -0700633 /*
634 * Options without arguments
635 */
636
David S. Miller48788092007-09-14 16:41:03 -0700637 if (optname == SO_BINDTODEVICE)
Brian Haleyc91f6df2012-11-26 05:21:08 +0000638 return sock_setbindtodevice(sk, optval, optlen);
David S. Miller48788092007-09-14 16:41:03 -0700639
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700640 if (optlen < sizeof(int))
641 return -EINVAL;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +0900642
Linus Torvalds1da177e2005-04-16 15:20:36 -0700643 if (get_user(val, (int __user *)optval))
644 return -EFAULT;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +0900645
Eric Dumazet2a915252009-05-27 11:30:05 +0000646 valbool = val ? 1 : 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700647
648 lock_sock(sk);
649
Eric Dumazet2a915252009-05-27 11:30:05 +0000650 switch (optname) {
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700651 case SO_DEBUG:
Eric Dumazet2a915252009-05-27 11:30:05 +0000652 if (val && !capable(CAP_NET_ADMIN))
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700653 ret = -EACCES;
Eric Dumazet2a915252009-05-27 11:30:05 +0000654 else
Pavel Emelyanovc0ef8772007-11-15 03:03:19 -0800655 sock_valbool_flag(sk, SOCK_DBG, valbool);
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700656 break;
657 case SO_REUSEADDR:
Pavel Emelyanov4a17fd52012-04-19 03:39:36 +0000658 sk->sk_reuse = (valbool ? SK_CAN_REUSE : SK_NO_REUSE);
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700659 break;
Tom Herbert055dc212013-01-22 09:49:50 +0000660 case SO_REUSEPORT:
661 sk->sk_reuseport = valbool;
662 break;
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700663 case SO_TYPE:
Jan Engelhardt49c794e2009-08-04 07:28:28 +0000664 case SO_PROTOCOL:
Jan Engelhardt0d6038e2009-08-04 07:28:29 +0000665 case SO_DOMAIN:
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700666 case SO_ERROR:
667 ret = -ENOPROTOOPT;
668 break;
669 case SO_DONTROUTE:
Pavel Emelyanovc0ef8772007-11-15 03:03:19 -0800670 sock_valbool_flag(sk, SOCK_LOCALROUTE, valbool);
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700671 break;
672 case SO_BROADCAST:
673 sock_valbool_flag(sk, SOCK_BROADCAST, valbool);
674 break;
675 case SO_SNDBUF:
676 /* Don't error on this BSD doesn't and if you think
Eric Dumazet82981932012-04-26 20:07:59 +0000677 * about it this is right. Otherwise apps have to
678 * play 'guess the biggest size' games. RCVBUF/SNDBUF
679 * are treated in BSD as hints
680 */
681 val = min_t(u32, val, sysctl_wmem_max);
Patrick McHardyb0573de2005-08-09 19:30:51 -0700682set_sndbuf:
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700683 sk->sk_userlocks |= SOCK_SNDBUF_LOCK;
Eric Dumazet82981932012-04-26 20:07:59 +0000684 sk->sk_sndbuf = max_t(u32, val * 2, SOCK_MIN_SNDBUF);
685 /* Wake up sending tasks if we upped the value. */
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700686 sk->sk_write_space(sk);
687 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700688
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700689 case SO_SNDBUFFORCE:
690 if (!capable(CAP_NET_ADMIN)) {
691 ret = -EPERM;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700692 break;
693 }
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700694 goto set_sndbuf;
695
696 case SO_RCVBUF:
697 /* Don't error on this BSD doesn't and if you think
Eric Dumazet82981932012-04-26 20:07:59 +0000698 * about it this is right. Otherwise apps have to
699 * play 'guess the biggest size' games. RCVBUF/SNDBUF
700 * are treated in BSD as hints
701 */
702 val = min_t(u32, val, sysctl_rmem_max);
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700703set_rcvbuf:
704 sk->sk_userlocks |= SOCK_RCVBUF_LOCK;
705 /*
706 * We double it on the way in to account for
707 * "struct sk_buff" etc. overhead. Applications
708 * assume that the SO_RCVBUF setting they make will
709 * allow that much actual data to be received on that
710 * socket.
711 *
712 * Applications are unaware that "struct sk_buff" and
713 * other overheads allocate from the receive buffer
714 * during socket buffer allocation.
715 *
716 * And after considering the possible alternatives,
717 * returning the value we actually used in getsockopt
718 * is the most desirable behavior.
719 */
Eric Dumazet82981932012-04-26 20:07:59 +0000720 sk->sk_rcvbuf = max_t(u32, val * 2, SOCK_MIN_RCVBUF);
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700721 break;
722
723 case SO_RCVBUFFORCE:
724 if (!capable(CAP_NET_ADMIN)) {
725 ret = -EPERM;
726 break;
727 }
728 goto set_rcvbuf;
729
730 case SO_KEEPALIVE:
731#ifdef CONFIG_INET
Eric Dumazet3e109862012-09-24 07:00:11 +0000732 if (sk->sk_protocol == IPPROTO_TCP &&
733 sk->sk_type == SOCK_STREAM)
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700734 tcp_set_keepalive(sk, valbool);
735#endif
736 sock_valbool_flag(sk, SOCK_KEEPOPEN, valbool);
737 break;
738
739 case SO_OOBINLINE:
740 sock_valbool_flag(sk, SOCK_URGINLINE, valbool);
741 break;
742
743 case SO_NO_CHECK:
744 sk->sk_no_check = valbool;
745 break;
746
747 case SO_PRIORITY:
Eric W. Biederman5e1fccc2012-11-16 03:03:04 +0000748 if ((val >= 0 && val <= 6) ||
749 ns_capable(sock_net(sk)->user_ns, CAP_NET_ADMIN))
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700750 sk->sk_priority = val;
751 else
752 ret = -EPERM;
753 break;
754
755 case SO_LINGER:
756 if (optlen < sizeof(ling)) {
757 ret = -EINVAL; /* 1003.1g */
758 break;
759 }
Eric Dumazet2a915252009-05-27 11:30:05 +0000760 if (copy_from_user(&ling, optval, sizeof(ling))) {
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700761 ret = -EFAULT;
762 break;
763 }
764 if (!ling.l_onoff)
765 sock_reset_flag(sk, SOCK_LINGER);
766 else {
767#if (BITS_PER_LONG == 32)
768 if ((unsigned int)ling.l_linger >= MAX_SCHEDULE_TIMEOUT/HZ)
769 sk->sk_lingertime = MAX_SCHEDULE_TIMEOUT;
770 else
771#endif
772 sk->sk_lingertime = (unsigned int)ling.l_linger * HZ;
773 sock_set_flag(sk, SOCK_LINGER);
774 }
775 break;
776
777 case SO_BSDCOMPAT:
778 sock_warn_obsolete_bsdism("setsockopt");
779 break;
780
781 case SO_PASSCRED:
782 if (valbool)
783 set_bit(SOCK_PASSCRED, &sock->flags);
784 else
785 clear_bit(SOCK_PASSCRED, &sock->flags);
786 break;
787
788 case SO_TIMESTAMP:
Eric Dumazet92f37fd2007-03-25 22:14:49 -0700789 case SO_TIMESTAMPNS:
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700790 if (valbool) {
Eric Dumazet92f37fd2007-03-25 22:14:49 -0700791 if (optname == SO_TIMESTAMP)
792 sock_reset_flag(sk, SOCK_RCVTSTAMPNS);
793 else
794 sock_set_flag(sk, SOCK_RCVTSTAMPNS);
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700795 sock_set_flag(sk, SOCK_RCVTSTAMP);
Patrick Ohly20d49472009-02-12 05:03:38 +0000796 sock_enable_timestamp(sk, SOCK_TIMESTAMP);
Eric Dumazet92f37fd2007-03-25 22:14:49 -0700797 } else {
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700798 sock_reset_flag(sk, SOCK_RCVTSTAMP);
Eric Dumazet92f37fd2007-03-25 22:14:49 -0700799 sock_reset_flag(sk, SOCK_RCVTSTAMPNS);
800 }
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700801 break;
802
Patrick Ohly20d49472009-02-12 05:03:38 +0000803 case SO_TIMESTAMPING:
804 if (val & ~SOF_TIMESTAMPING_MASK) {
Rémi Denis-Courmontf249fb72009-07-20 00:47:04 +0000805 ret = -EINVAL;
Patrick Ohly20d49472009-02-12 05:03:38 +0000806 break;
807 }
808 sock_valbool_flag(sk, SOCK_TIMESTAMPING_TX_HARDWARE,
809 val & SOF_TIMESTAMPING_TX_HARDWARE);
810 sock_valbool_flag(sk, SOCK_TIMESTAMPING_TX_SOFTWARE,
811 val & SOF_TIMESTAMPING_TX_SOFTWARE);
812 sock_valbool_flag(sk, SOCK_TIMESTAMPING_RX_HARDWARE,
813 val & SOF_TIMESTAMPING_RX_HARDWARE);
814 if (val & SOF_TIMESTAMPING_RX_SOFTWARE)
815 sock_enable_timestamp(sk,
816 SOCK_TIMESTAMPING_RX_SOFTWARE);
817 else
818 sock_disable_timestamp(sk,
Eric Dumazet08e29af2011-11-28 12:04:18 +0000819 (1UL << SOCK_TIMESTAMPING_RX_SOFTWARE));
Patrick Ohly20d49472009-02-12 05:03:38 +0000820 sock_valbool_flag(sk, SOCK_TIMESTAMPING_SOFTWARE,
821 val & SOF_TIMESTAMPING_SOFTWARE);
822 sock_valbool_flag(sk, SOCK_TIMESTAMPING_SYS_HARDWARE,
823 val & SOF_TIMESTAMPING_SYS_HARDWARE);
824 sock_valbool_flag(sk, SOCK_TIMESTAMPING_RAW_HARDWARE,
825 val & SOF_TIMESTAMPING_RAW_HARDWARE);
826 break;
827
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700828 case SO_RCVLOWAT:
829 if (val < 0)
830 val = INT_MAX;
831 sk->sk_rcvlowat = val ? : 1;
832 break;
833
834 case SO_RCVTIMEO:
835 ret = sock_set_timeout(&sk->sk_rcvtimeo, optval, optlen);
836 break;
837
838 case SO_SNDTIMEO:
839 ret = sock_set_timeout(&sk->sk_sndtimeo, optval, optlen);
840 break;
841
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700842 case SO_ATTACH_FILTER:
843 ret = -EINVAL;
844 if (optlen == sizeof(struct sock_fprog)) {
845 struct sock_fprog fprog;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700846
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700847 ret = -EFAULT;
848 if (copy_from_user(&fprog, optval, sizeof(fprog)))
Linus Torvalds1da177e2005-04-16 15:20:36 -0700849 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700850
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700851 ret = sk_attach_filter(&fprog, sk);
852 }
853 break;
854
855 case SO_DETACH_FILTER:
Pavel Emelyanov55b33322007-10-17 21:21:26 -0700856 ret = sk_detach_filter(sk);
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700857 break;
858
Vincent Bernatd59577b2013-01-16 22:55:49 +0100859 case SO_LOCK_FILTER:
860 if (sock_flag(sk, SOCK_FILTER_LOCKED) && !valbool)
861 ret = -EPERM;
862 else
863 sock_valbool_flag(sk, SOCK_FILTER_LOCKED, valbool);
864 break;
865
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700866 case SO_PASSSEC:
867 if (valbool)
868 set_bit(SOCK_PASSSEC, &sock->flags);
869 else
870 clear_bit(SOCK_PASSSEC, &sock->flags);
871 break;
Laszlo Attila Toth4a19ec52008-01-30 19:08:16 -0800872 case SO_MARK:
Eric W. Biederman5e1fccc2012-11-16 03:03:04 +0000873 if (!ns_capable(sock_net(sk)->user_ns, CAP_NET_ADMIN))
Laszlo Attila Toth4a19ec52008-01-30 19:08:16 -0800874 ret = -EPERM;
Eric Dumazet2a915252009-05-27 11:30:05 +0000875 else
Laszlo Attila Toth4a19ec52008-01-30 19:08:16 -0800876 sk->sk_mark = val;
Laszlo Attila Toth4a19ec52008-01-30 19:08:16 -0800877 break;
Catherine Zhang877ce7c2006-06-29 12:27:47 -0700878
Linus Torvalds1da177e2005-04-16 15:20:36 -0700879 /* We implement the SO_SNDLOWAT etc to
880 not be settable (1003.1g 5.3) */
Neil Horman3b885782009-10-12 13:26:31 -0700881 case SO_RXQ_OVFL:
Johannes Berg8083f0f2011-10-07 03:30:20 +0000882 sock_valbool_flag(sk, SOCK_RXQ_OVFL, valbool);
Neil Horman3b885782009-10-12 13:26:31 -0700883 break;
Johannes Berg6e3e9392011-11-09 10:15:42 +0100884
885 case SO_WIFI_STATUS:
886 sock_valbool_flag(sk, SOCK_WIFI_STATUS, valbool);
887 break;
888
Pavel Emelyanovef64a542012-02-21 07:31:34 +0000889 case SO_PEEK_OFF:
890 if (sock->ops->set_peek_off)
891 sock->ops->set_peek_off(sk, val);
892 else
893 ret = -EOPNOTSUPP;
894 break;
Ben Greear3bdc0eb2012-02-11 15:39:30 +0000895
896 case SO_NOFCS:
897 sock_valbool_flag(sk, SOCK_NOFCS, valbool);
898 break;
899
Keller, Jacob E7d4c04f2013-03-28 11:19:25 +0000900 case SO_SELECT_ERR_QUEUE:
901 sock_valbool_flag(sk, SOCK_SELECT_ERR_QUEUE, valbool);
902 break;
903
Cong Wange0d10952013-08-01 11:10:25 +0800904#ifdef CONFIG_NET_RX_BUSY_POLL
Eliezer Tamir64b0dc52013-07-10 17:13:36 +0300905 case SO_BUSY_POLL:
Eliezer Tamirdafcc432013-06-14 16:33:57 +0300906 /* allow unprivileged users to decrease the value */
907 if ((val > sk->sk_ll_usec) && !capable(CAP_NET_ADMIN))
908 ret = -EPERM;
909 else {
910 if (val < 0)
911 ret = -EINVAL;
912 else
913 sk->sk_ll_usec = val;
914 }
915 break;
916#endif
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700917 default:
918 ret = -ENOPROTOOPT;
919 break;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +0900920 }
Linus Torvalds1da177e2005-04-16 15:20:36 -0700921 release_sock(sk);
922 return ret;
923}
Eric Dumazet2a915252009-05-27 11:30:05 +0000924EXPORT_SYMBOL(sock_setsockopt);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700925
926
Eric W. Biederman3f551f92010-06-13 03:28:59 +0000927void cred_to_ucred(struct pid *pid, const struct cred *cred,
928 struct ucred *ucred)
929{
930 ucred->pid = pid_vnr(pid);
931 ucred->uid = ucred->gid = -1;
932 if (cred) {
933 struct user_namespace *current_ns = current_user_ns();
934
Eric W. Biedermanb2e4f542012-05-23 16:39:45 -0600935 ucred->uid = from_kuid_munged(current_ns, cred->euid);
936 ucred->gid = from_kgid_munged(current_ns, cred->egid);
Eric W. Biederman3f551f92010-06-13 03:28:59 +0000937 }
938}
David S. Miller39247732010-06-16 16:18:25 -0700939EXPORT_SYMBOL_GPL(cred_to_ucred);
Eric W. Biederman3f551f92010-06-13 03:28:59 +0000940
Linus Torvalds1da177e2005-04-16 15:20:36 -0700941int sock_getsockopt(struct socket *sock, int level, int optname,
942 char __user *optval, int __user *optlen)
943{
944 struct sock *sk = sock->sk;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +0900945
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700946 union {
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +0900947 int val;
948 struct linger ling;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700949 struct timeval tm;
950 } v;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +0900951
H Hartley Sweeten4d0392b2010-01-15 01:08:58 -0800952 int lv = sizeof(int);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700953 int len;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +0900954
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700955 if (get_user(len, optlen))
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +0900956 return -EFAULT;
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700957 if (len < 0)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700958 return -EINVAL;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +0900959
Eugene Teo50fee1d2009-02-23 15:38:41 -0800960 memset(&v, 0, sizeof(v));
Clément Lecignedf0bca02009-02-12 16:59:09 -0800961
Eric Dumazet2a915252009-05-27 11:30:05 +0000962 switch (optname) {
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700963 case SO_DEBUG:
964 v.val = sock_flag(sk, SOCK_DBG);
965 break;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +0900966
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700967 case SO_DONTROUTE:
968 v.val = sock_flag(sk, SOCK_LOCALROUTE);
969 break;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +0900970
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700971 case SO_BROADCAST:
Eric Dumazet1b23a5d2012-05-16 05:57:07 +0000972 v.val = sock_flag(sk, SOCK_BROADCAST);
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700973 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700974
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700975 case SO_SNDBUF:
976 v.val = sk->sk_sndbuf;
977 break;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +0900978
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700979 case SO_RCVBUF:
980 v.val = sk->sk_rcvbuf;
981 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700982
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700983 case SO_REUSEADDR:
984 v.val = sk->sk_reuse;
985 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700986
Tom Herbert055dc212013-01-22 09:49:50 +0000987 case SO_REUSEPORT:
988 v.val = sk->sk_reuseport;
989 break;
990
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700991 case SO_KEEPALIVE:
Eric Dumazet1b23a5d2012-05-16 05:57:07 +0000992 v.val = sock_flag(sk, SOCK_KEEPOPEN);
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700993 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700994
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700995 case SO_TYPE:
996 v.val = sk->sk_type;
997 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700998
Jan Engelhardt49c794e2009-08-04 07:28:28 +0000999 case SO_PROTOCOL:
1000 v.val = sk->sk_protocol;
1001 break;
1002
Jan Engelhardt0d6038e2009-08-04 07:28:29 +00001003 case SO_DOMAIN:
1004 v.val = sk->sk_family;
1005 break;
1006
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001007 case SO_ERROR:
1008 v.val = -sock_error(sk);
Eric Dumazet2a915252009-05-27 11:30:05 +00001009 if (v.val == 0)
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001010 v.val = xchg(&sk->sk_err_soft, 0);
1011 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001012
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001013 case SO_OOBINLINE:
Eric Dumazet1b23a5d2012-05-16 05:57:07 +00001014 v.val = sock_flag(sk, SOCK_URGINLINE);
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001015 break;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001016
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001017 case SO_NO_CHECK:
1018 v.val = sk->sk_no_check;
1019 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001020
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001021 case SO_PRIORITY:
1022 v.val = sk->sk_priority;
1023 break;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001024
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001025 case SO_LINGER:
1026 lv = sizeof(v.ling);
Eric Dumazet1b23a5d2012-05-16 05:57:07 +00001027 v.ling.l_onoff = sock_flag(sk, SOCK_LINGER);
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001028 v.ling.l_linger = sk->sk_lingertime / HZ;
1029 break;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001030
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001031 case SO_BSDCOMPAT:
1032 sock_warn_obsolete_bsdism("getsockopt");
1033 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001034
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001035 case SO_TIMESTAMP:
Eric Dumazet92f37fd2007-03-25 22:14:49 -07001036 v.val = sock_flag(sk, SOCK_RCVTSTAMP) &&
1037 !sock_flag(sk, SOCK_RCVTSTAMPNS);
1038 break;
1039
1040 case SO_TIMESTAMPNS:
1041 v.val = sock_flag(sk, SOCK_RCVTSTAMPNS);
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001042 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001043
Patrick Ohly20d49472009-02-12 05:03:38 +00001044 case SO_TIMESTAMPING:
1045 v.val = 0;
1046 if (sock_flag(sk, SOCK_TIMESTAMPING_TX_HARDWARE))
1047 v.val |= SOF_TIMESTAMPING_TX_HARDWARE;
1048 if (sock_flag(sk, SOCK_TIMESTAMPING_TX_SOFTWARE))
1049 v.val |= SOF_TIMESTAMPING_TX_SOFTWARE;
1050 if (sock_flag(sk, SOCK_TIMESTAMPING_RX_HARDWARE))
1051 v.val |= SOF_TIMESTAMPING_RX_HARDWARE;
1052 if (sock_flag(sk, SOCK_TIMESTAMPING_RX_SOFTWARE))
1053 v.val |= SOF_TIMESTAMPING_RX_SOFTWARE;
1054 if (sock_flag(sk, SOCK_TIMESTAMPING_SOFTWARE))
1055 v.val |= SOF_TIMESTAMPING_SOFTWARE;
1056 if (sock_flag(sk, SOCK_TIMESTAMPING_SYS_HARDWARE))
1057 v.val |= SOF_TIMESTAMPING_SYS_HARDWARE;
1058 if (sock_flag(sk, SOCK_TIMESTAMPING_RAW_HARDWARE))
1059 v.val |= SOF_TIMESTAMPING_RAW_HARDWARE;
1060 break;
1061
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001062 case SO_RCVTIMEO:
Eric Dumazet2a915252009-05-27 11:30:05 +00001063 lv = sizeof(struct timeval);
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001064 if (sk->sk_rcvtimeo == MAX_SCHEDULE_TIMEOUT) {
1065 v.tm.tv_sec = 0;
1066 v.tm.tv_usec = 0;
1067 } else {
1068 v.tm.tv_sec = sk->sk_rcvtimeo / HZ;
1069 v.tm.tv_usec = ((sk->sk_rcvtimeo % HZ) * 1000000) / HZ;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001070 }
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001071 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001072
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001073 case SO_SNDTIMEO:
Eric Dumazet2a915252009-05-27 11:30:05 +00001074 lv = sizeof(struct timeval);
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001075 if (sk->sk_sndtimeo == MAX_SCHEDULE_TIMEOUT) {
1076 v.tm.tv_sec = 0;
1077 v.tm.tv_usec = 0;
1078 } else {
1079 v.tm.tv_sec = sk->sk_sndtimeo / HZ;
1080 v.tm.tv_usec = ((sk->sk_sndtimeo % HZ) * 1000000) / HZ;
1081 }
1082 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001083
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001084 case SO_RCVLOWAT:
1085 v.val = sk->sk_rcvlowat;
1086 break;
Catherine Zhang877ce7c2006-06-29 12:27:47 -07001087
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001088 case SO_SNDLOWAT:
Eric Dumazet2a915252009-05-27 11:30:05 +00001089 v.val = 1;
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001090 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001091
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001092 case SO_PASSCRED:
Eric Dumazet82981932012-04-26 20:07:59 +00001093 v.val = !!test_bit(SOCK_PASSCRED, &sock->flags);
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001094 break;
1095
1096 case SO_PEERCRED:
Eric W. Biederman109f6e32010-06-13 03:30:14 +00001097 {
1098 struct ucred peercred;
1099 if (len > sizeof(peercred))
1100 len = sizeof(peercred);
1101 cred_to_ucred(sk->sk_peer_pid, sk->sk_peer_cred, &peercred);
1102 if (copy_to_user(optval, &peercred, len))
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001103 return -EFAULT;
1104 goto lenout;
Eric W. Biederman109f6e32010-06-13 03:30:14 +00001105 }
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001106
1107 case SO_PEERNAME:
1108 {
1109 char address[128];
1110
1111 if (sock->ops->getname(sock, (struct sockaddr *)address, &lv, 2))
1112 return -ENOTCONN;
1113 if (lv < len)
1114 return -EINVAL;
1115 if (copy_to_user(optval, address, len))
1116 return -EFAULT;
1117 goto lenout;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001118 }
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001119
1120 /* Dubious BSD thing... Probably nobody even uses it, but
1121 * the UNIX standard wants it for whatever reason... -DaveM
1122 */
1123 case SO_ACCEPTCONN:
1124 v.val = sk->sk_state == TCP_LISTEN;
1125 break;
1126
1127 case SO_PASSSEC:
Eric Dumazet82981932012-04-26 20:07:59 +00001128 v.val = !!test_bit(SOCK_PASSSEC, &sock->flags);
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001129 break;
1130
1131 case SO_PEERSEC:
1132 return security_socket_getpeersec_stream(sock, optval, optlen, len);
1133
Laszlo Attila Toth4a19ec52008-01-30 19:08:16 -08001134 case SO_MARK:
1135 v.val = sk->sk_mark;
1136 break;
1137
Neil Horman3b885782009-10-12 13:26:31 -07001138 case SO_RXQ_OVFL:
Eric Dumazet1b23a5d2012-05-16 05:57:07 +00001139 v.val = sock_flag(sk, SOCK_RXQ_OVFL);
Neil Horman3b885782009-10-12 13:26:31 -07001140 break;
1141
Johannes Berg6e3e9392011-11-09 10:15:42 +01001142 case SO_WIFI_STATUS:
Eric Dumazet1b23a5d2012-05-16 05:57:07 +00001143 v.val = sock_flag(sk, SOCK_WIFI_STATUS);
Johannes Berg6e3e9392011-11-09 10:15:42 +01001144 break;
1145
Pavel Emelyanovef64a542012-02-21 07:31:34 +00001146 case SO_PEEK_OFF:
1147 if (!sock->ops->set_peek_off)
1148 return -EOPNOTSUPP;
1149
1150 v.val = sk->sk_peek_off;
1151 break;
David S. Millerbc2f7992012-02-24 14:48:34 -05001152 case SO_NOFCS:
Eric Dumazet1b23a5d2012-05-16 05:57:07 +00001153 v.val = sock_flag(sk, SOCK_NOFCS);
David S. Millerbc2f7992012-02-24 14:48:34 -05001154 break;
Brian Haleyc91f6df2012-11-26 05:21:08 +00001155
Pavel Emelyanovf7b86bf2012-10-18 23:55:56 +00001156 case SO_BINDTODEVICE:
Brian Haleyc91f6df2012-11-26 05:21:08 +00001157 return sock_getbindtodevice(sk, optval, optlen, len);
1158
Pavel Emelyanova8fc9272012-11-01 02:01:48 +00001159 case SO_GET_FILTER:
1160 len = sk_get_filter(sk, (struct sock_filter __user *)optval, len);
1161 if (len < 0)
1162 return len;
1163
1164 goto lenout;
Brian Haleyc91f6df2012-11-26 05:21:08 +00001165
Vincent Bernatd59577b2013-01-16 22:55:49 +01001166 case SO_LOCK_FILTER:
1167 v.val = sock_flag(sk, SOCK_FILTER_LOCKED);
1168 break;
1169
Keller, Jacob E7d4c04f2013-03-28 11:19:25 +00001170 case SO_SELECT_ERR_QUEUE:
1171 v.val = sock_flag(sk, SOCK_SELECT_ERR_QUEUE);
1172 break;
1173
Cong Wange0d10952013-08-01 11:10:25 +08001174#ifdef CONFIG_NET_RX_BUSY_POLL
Eliezer Tamir64b0dc52013-07-10 17:13:36 +03001175 case SO_BUSY_POLL:
Eliezer Tamirdafcc432013-06-14 16:33:57 +03001176 v.val = sk->sk_ll_usec;
1177 break;
1178#endif
1179
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001180 default:
1181 return -ENOPROTOOPT;
1182 }
1183
Linus Torvalds1da177e2005-04-16 15:20:36 -07001184 if (len > lv)
1185 len = lv;
1186 if (copy_to_user(optval, &v, len))
1187 return -EFAULT;
1188lenout:
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001189 if (put_user(len, optlen))
1190 return -EFAULT;
1191 return 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001192}
1193
Ingo Molnara5b5bb92006-07-03 00:25:35 -07001194/*
1195 * Initialize an sk_lock.
1196 *
1197 * (We also register the sk_lock with the lock validator.)
1198 */
Dave Jonesb6f99a22007-03-22 12:27:49 -07001199static inline void sock_lock_init(struct sock *sk)
Ingo Molnara5b5bb92006-07-03 00:25:35 -07001200{
Peter Zijlstraed075362006-12-06 20:35:24 -08001201 sock_lock_init_class_and_name(sk,
1202 af_family_slock_key_strings[sk->sk_family],
1203 af_family_slock_keys + sk->sk_family,
1204 af_family_key_strings[sk->sk_family],
1205 af_family_keys + sk->sk_family);
Ingo Molnara5b5bb92006-07-03 00:25:35 -07001206}
1207
Eric Dumazet4dc6dc72009-07-15 23:13:10 +00001208/*
1209 * Copy all fields from osk to nsk but nsk->sk_refcnt must not change yet,
1210 * even temporarly, because of RCU lookups. sk_node should also be left as is.
Eric Dumazet68835ab2010-11-30 19:04:07 +00001211 * We must not copy fields between sk_dontcopy_begin and sk_dontcopy_end
Eric Dumazet4dc6dc72009-07-15 23:13:10 +00001212 */
Pavel Emelyanovf1a6c4d2007-11-01 00:29:45 -07001213static void sock_copy(struct sock *nsk, const struct sock *osk)
1214{
1215#ifdef CONFIG_SECURITY_NETWORK
1216 void *sptr = nsk->sk_security;
1217#endif
Eric Dumazet68835ab2010-11-30 19:04:07 +00001218 memcpy(nsk, osk, offsetof(struct sock, sk_dontcopy_begin));
1219
1220 memcpy(&nsk->sk_dontcopy_end, &osk->sk_dontcopy_end,
1221 osk->sk_prot->obj_size - offsetof(struct sock, sk_dontcopy_end));
1222
Pavel Emelyanovf1a6c4d2007-11-01 00:29:45 -07001223#ifdef CONFIG_SECURITY_NETWORK
1224 nsk->sk_security = sptr;
1225 security_sk_clone(osk, nsk);
1226#endif
1227}
1228
Octavian Purdilafcbdf092010-12-16 14:26:56 -08001229void sk_prot_clear_portaddr_nulls(struct sock *sk, int size)
1230{
1231 unsigned long nulls1, nulls2;
1232
1233 nulls1 = offsetof(struct sock, __sk_common.skc_node.next);
1234 nulls2 = offsetof(struct sock, __sk_common.skc_portaddr_node.next);
1235 if (nulls1 > nulls2)
1236 swap(nulls1, nulls2);
1237
1238 if (nulls1 != 0)
1239 memset((char *)sk, 0, nulls1);
1240 memset((char *)sk + nulls1 + sizeof(void *), 0,
1241 nulls2 - nulls1 - sizeof(void *));
1242 memset((char *)sk + nulls2 + sizeof(void *), 0,
1243 size - nulls2 - sizeof(void *));
1244}
1245EXPORT_SYMBOL(sk_prot_clear_portaddr_nulls);
1246
Pavel Emelyanov2e4afe72007-11-01 00:36:26 -07001247static struct sock *sk_prot_alloc(struct proto *prot, gfp_t priority,
1248 int family)
Pavel Emelyanovc308c1b22007-11-01 00:33:50 -07001249{
1250 struct sock *sk;
1251 struct kmem_cache *slab;
1252
1253 slab = prot->slab;
Eric Dumazete912b112009-07-08 19:36:05 +00001254 if (slab != NULL) {
1255 sk = kmem_cache_alloc(slab, priority & ~__GFP_ZERO);
1256 if (!sk)
1257 return sk;
1258 if (priority & __GFP_ZERO) {
Octavian Purdilafcbdf092010-12-16 14:26:56 -08001259 if (prot->clear_sk)
1260 prot->clear_sk(sk, prot->obj_size);
1261 else
1262 sk_prot_clear_nulls(sk, prot->obj_size);
Eric Dumazete912b112009-07-08 19:36:05 +00001263 }
Octavian Purdilafcbdf092010-12-16 14:26:56 -08001264 } else
Pavel Emelyanovc308c1b22007-11-01 00:33:50 -07001265 sk = kmalloc(prot->obj_size, priority);
1266
Pavel Emelyanov2e4afe72007-11-01 00:36:26 -07001267 if (sk != NULL) {
Vegard Nossuma98b65a2009-02-26 14:46:57 +01001268 kmemcheck_annotate_bitfield(sk, flags);
1269
Pavel Emelyanov2e4afe72007-11-01 00:36:26 -07001270 if (security_sk_alloc(sk, family, priority))
1271 goto out_free;
1272
1273 if (!try_module_get(prot->owner))
1274 goto out_free_sec;
Krishna Kumare022f0b2009-10-19 23:46:20 +00001275 sk_tx_queue_clear(sk);
Pavel Emelyanov2e4afe72007-11-01 00:36:26 -07001276 }
1277
Pavel Emelyanovc308c1b22007-11-01 00:33:50 -07001278 return sk;
Pavel Emelyanov2e4afe72007-11-01 00:36:26 -07001279
1280out_free_sec:
1281 security_sk_free(sk);
1282out_free:
1283 if (slab != NULL)
1284 kmem_cache_free(slab, sk);
1285 else
1286 kfree(sk);
1287 return NULL;
Pavel Emelyanovc308c1b22007-11-01 00:33:50 -07001288}
1289
1290static void sk_prot_free(struct proto *prot, struct sock *sk)
1291{
1292 struct kmem_cache *slab;
Pavel Emelyanov2e4afe72007-11-01 00:36:26 -07001293 struct module *owner;
Pavel Emelyanovc308c1b22007-11-01 00:33:50 -07001294
Pavel Emelyanov2e4afe72007-11-01 00:36:26 -07001295 owner = prot->owner;
Pavel Emelyanovc308c1b22007-11-01 00:33:50 -07001296 slab = prot->slab;
Pavel Emelyanov2e4afe72007-11-01 00:36:26 -07001297
1298 security_sk_free(sk);
Pavel Emelyanovc308c1b22007-11-01 00:33:50 -07001299 if (slab != NULL)
1300 kmem_cache_free(slab, sk);
1301 else
1302 kfree(sk);
Pavel Emelyanov2e4afe72007-11-01 00:36:26 -07001303 module_put(owner);
Pavel Emelyanovc308c1b22007-11-01 00:33:50 -07001304}
1305
Daniel Wagner8fb974c2012-09-12 16:12:02 +02001306#if IS_ENABLED(CONFIG_NET_CLS_CGROUP)
Zefan Li211d2f972013-04-08 20:03:35 +00001307void sock_update_classid(struct sock *sk)
Herbert Xuf8451722010-05-24 00:12:34 -07001308{
Paul E. McKenney11441822010-10-06 17:15:35 -07001309 u32 classid;
Herbert Xuf8451722010-05-24 00:12:34 -07001310
Zefan Li211d2f972013-04-08 20:03:35 +00001311 classid = task_cls_classid(current);
Neil Horman3afa6d02012-08-20 07:59:10 +00001312 if (classid != sk->sk_classid)
Herbert Xuf8451722010-05-24 00:12:34 -07001313 sk->sk_classid = classid;
1314}
Herbert Xu82862742010-05-24 00:14:10 -07001315EXPORT_SYMBOL(sock_update_classid);
Daniel Wagner8fb974c2012-09-12 16:12:02 +02001316#endif
Neil Horman5bc14212011-11-22 05:10:51 +00001317
Daniel Wagner51e4e7f2012-09-12 16:12:03 +02001318#if IS_ENABLED(CONFIG_NETPRIO_CGROUP)
Zefan Li6ffd4642013-04-08 20:03:47 +00001319void sock_update_netprioidx(struct sock *sk)
Neil Horman5bc14212011-11-22 05:10:51 +00001320{
Neil Horman5bc14212011-11-22 05:10:51 +00001321 if (in_interrupt())
1322 return;
Neil Horman2b73bc62012-02-10 05:43:38 +00001323
Zefan Li6ffd4642013-04-08 20:03:47 +00001324 sk->sk_cgrp_prioidx = task_netprioidx(current);
Neil Horman5bc14212011-11-22 05:10:51 +00001325}
1326EXPORT_SYMBOL_GPL(sock_update_netprioidx);
Herbert Xuf8451722010-05-24 00:12:34 -07001327#endif
1328
Linus Torvalds1da177e2005-04-16 15:20:36 -07001329/**
1330 * sk_alloc - All socket objects are allocated here
Randy Dunlapc4ea43c2007-10-12 21:17:49 -07001331 * @net: the applicable net namespace
Pavel Pisa4dc3b162005-05-01 08:59:25 -07001332 * @family: protocol family
1333 * @priority: for allocation (%GFP_KERNEL, %GFP_ATOMIC, etc)
1334 * @prot: struct proto associated with this new sock instance
Linus Torvalds1da177e2005-04-16 15:20:36 -07001335 */
Eric W. Biederman1b8d7ae2007-10-08 23:24:22 -07001336struct sock *sk_alloc(struct net *net, int family, gfp_t priority,
Pavel Emelyanov6257ff22007-11-01 00:39:31 -07001337 struct proto *prot)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001338{
Pavel Emelyanovc308c1b22007-11-01 00:33:50 -07001339 struct sock *sk;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001340
Pavel Emelyanov154adbc2007-11-01 00:38:43 -07001341 sk = sk_prot_alloc(prot, priority | __GFP_ZERO, family);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001342 if (sk) {
Pavel Emelyanov154adbc2007-11-01 00:38:43 -07001343 sk->sk_family = family;
1344 /*
1345 * See comment in struct sock definition to understand
1346 * why we need sk_prot_creator -acme
1347 */
1348 sk->sk_prot = sk->sk_prot_creator = prot;
1349 sock_lock_init(sk);
YOSHIFUJI Hideaki3b1e0a62008-03-26 02:26:21 +09001350 sock_net_set(sk, get_net(net));
Jarek Poplawskid66ee052009-08-30 23:15:36 +00001351 atomic_set(&sk->sk_wmem_alloc, 1);
Herbert Xuf8451722010-05-24 00:12:34 -07001352
Zefan Li211d2f972013-04-08 20:03:35 +00001353 sock_update_classid(sk);
Zefan Li6ffd4642013-04-08 20:03:47 +00001354 sock_update_netprioidx(sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001355 }
Frank Filza79af592005-09-27 15:23:38 -07001356
Pavel Emelyanov2e4afe72007-11-01 00:36:26 -07001357 return sk;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001358}
Eric Dumazet2a915252009-05-27 11:30:05 +00001359EXPORT_SYMBOL(sk_alloc);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001360
Eric Dumazet2b85a342009-06-11 02:55:43 -07001361static void __sk_free(struct sock *sk)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001362{
1363 struct sk_filter *filter;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001364
1365 if (sk->sk_destruct)
1366 sk->sk_destruct(sk);
1367
Paul E. McKenneya898def2010-02-22 17:04:49 -08001368 filter = rcu_dereference_check(sk->sk_filter,
1369 atomic_read(&sk->sk_wmem_alloc) == 0);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001370 if (filter) {
Pavel Emelyanov309dd5f2007-10-17 21:21:51 -07001371 sk_filter_uncharge(sk, filter);
Stephen Hemmingera9b3cd72011-08-01 16:19:00 +00001372 RCU_INIT_POINTER(sk->sk_filter, NULL);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001373 }
1374
Eric Dumazet08e29af2011-11-28 12:04:18 +00001375 sock_disable_timestamp(sk, SK_FLAGS_TIMESTAMP);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001376
1377 if (atomic_read(&sk->sk_omem_alloc))
Joe Perchese005d192012-05-16 19:58:40 +00001378 pr_debug("%s: optmem leakage (%d bytes) detected\n",
1379 __func__, atomic_read(&sk->sk_omem_alloc));
Linus Torvalds1da177e2005-04-16 15:20:36 -07001380
Eric W. Biederman109f6e32010-06-13 03:30:14 +00001381 if (sk->sk_peer_cred)
1382 put_cred(sk->sk_peer_cred);
1383 put_pid(sk->sk_peer_pid);
YOSHIFUJI Hideaki3b1e0a62008-03-26 02:26:21 +09001384 put_net(sock_net(sk));
Pavel Emelyanovc308c1b22007-11-01 00:33:50 -07001385 sk_prot_free(sk->sk_prot_creator, sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001386}
Eric Dumazet2b85a342009-06-11 02:55:43 -07001387
1388void sk_free(struct sock *sk)
1389{
1390 /*
Lucas De Marchi25985ed2011-03-30 22:57:33 -03001391 * We subtract one from sk_wmem_alloc and can know if
Eric Dumazet2b85a342009-06-11 02:55:43 -07001392 * some packets are still in some tx queue.
1393 * If not null, sock_wfree() will call __sk_free(sk) later
1394 */
1395 if (atomic_dec_and_test(&sk->sk_wmem_alloc))
1396 __sk_free(sk);
1397}
Eric Dumazet2a915252009-05-27 11:30:05 +00001398EXPORT_SYMBOL(sk_free);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001399
Denis V. Lunevedf02082008-02-29 11:18:32 -08001400/*
Lucas De Marchi25985ed2011-03-30 22:57:33 -03001401 * Last sock_put should drop reference to sk->sk_net. It has already
1402 * been dropped in sk_change_net. Taking reference to stopping namespace
Denis V. Lunevedf02082008-02-29 11:18:32 -08001403 * is not an option.
Lucas De Marchi25985ed2011-03-30 22:57:33 -03001404 * Take reference to a socket to remove it from hash _alive_ and after that
Denis V. Lunevedf02082008-02-29 11:18:32 -08001405 * destroy it in the context of init_net.
1406 */
1407void sk_release_kernel(struct sock *sk)
1408{
1409 if (sk == NULL || sk->sk_socket == NULL)
1410 return;
1411
1412 sock_hold(sk);
1413 sock_release(sk->sk_socket);
Denis V. Lunev65a18ec2008-04-16 01:59:46 -07001414 release_net(sock_net(sk));
YOSHIFUJI Hideaki3b1e0a62008-03-26 02:26:21 +09001415 sock_net_set(sk, get_net(&init_net));
Denis V. Lunevedf02082008-02-29 11:18:32 -08001416 sock_put(sk);
1417}
David S. Miller45af1752008-02-29 11:33:19 -08001418EXPORT_SYMBOL(sk_release_kernel);
Denis V. Lunevedf02082008-02-29 11:18:32 -08001419
Stephen Rothwell475f1b52012-01-09 16:33:16 +11001420static void sk_update_clone(const struct sock *sk, struct sock *newsk)
1421{
1422 if (mem_cgroup_sockets_enabled && sk->sk_cgrp)
1423 sock_update_memcg(newsk);
1424}
1425
Eric Dumazete56c57d2011-11-08 17:07:07 -05001426/**
1427 * sk_clone_lock - clone a socket, and lock its clone
1428 * @sk: the socket to clone
1429 * @priority: for allocation (%GFP_KERNEL, %GFP_ATOMIC, etc)
1430 *
1431 * Caller must unlock socket even in error path (bh_unlock_sock(newsk))
1432 */
1433struct sock *sk_clone_lock(const struct sock *sk, const gfp_t priority)
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001434{
Pavel Emelyanov8fd1d172007-11-01 00:37:32 -07001435 struct sock *newsk;
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001436
Pavel Emelyanov8fd1d172007-11-01 00:37:32 -07001437 newsk = sk_prot_alloc(sk->sk_prot, priority, sk->sk_family);
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001438 if (newsk != NULL) {
1439 struct sk_filter *filter;
1440
Venkat Yekkirala892c1412006-08-04 23:08:56 -07001441 sock_copy(newsk, sk);
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001442
1443 /* SANITY */
YOSHIFUJI Hideaki3b1e0a62008-03-26 02:26:21 +09001444 get_net(sock_net(newsk));
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001445 sk_node_init(&newsk->sk_node);
1446 sock_lock_init(newsk);
1447 bh_lock_sock(newsk);
Eric Dumazetfa438cc2007-03-04 16:05:44 -08001448 newsk->sk_backlog.head = newsk->sk_backlog.tail = NULL;
Zhu Yi8eae9392010-03-04 18:01:40 +00001449 newsk->sk_backlog.len = 0;
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001450
1451 atomic_set(&newsk->sk_rmem_alloc, 0);
Eric Dumazet2b85a342009-06-11 02:55:43 -07001452 /*
1453 * sk_wmem_alloc set to one (see sk_free() and sock_wfree())
1454 */
1455 atomic_set(&newsk->sk_wmem_alloc, 1);
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001456 atomic_set(&newsk->sk_omem_alloc, 0);
1457 skb_queue_head_init(&newsk->sk_receive_queue);
1458 skb_queue_head_init(&newsk->sk_write_queue);
Chris Leech97fc2f02006-05-23 17:55:33 -07001459#ifdef CONFIG_NET_DMA
1460 skb_queue_head_init(&newsk->sk_async_wait_queue);
1461#endif
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001462
Eric Dumazetb6c67122010-04-08 23:03:29 +00001463 spin_lock_init(&newsk->sk_dst_lock);
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001464 rwlock_init(&newsk->sk_callback_lock);
Peter Zijlstra443aef02007-07-19 01:49:00 -07001465 lockdep_set_class_and_name(&newsk->sk_callback_lock,
1466 af_callback_keys + newsk->sk_family,
1467 af_family_clock_key_strings[newsk->sk_family]);
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001468
1469 newsk->sk_dst_cache = NULL;
1470 newsk->sk_wmem_queued = 0;
1471 newsk->sk_forward_alloc = 0;
1472 newsk->sk_send_head = NULL;
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001473 newsk->sk_userlocks = sk->sk_userlocks & ~SOCK_BINDPORT_LOCK;
1474
1475 sock_reset_flag(newsk, SOCK_DONE);
1476 skb_queue_head_init(&newsk->sk_error_queue);
1477
Eric Dumazet0d7da9d2010-10-25 03:47:05 +00001478 filter = rcu_dereference_protected(newsk->sk_filter, 1);
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001479 if (filter != NULL)
1480 sk_filter_charge(newsk, filter);
1481
1482 if (unlikely(xfrm_sk_clone_policy(newsk))) {
1483 /* It is still raw copy of parent, so invalidate
1484 * destructor and make plain sk_free() */
1485 newsk->sk_destruct = NULL;
Thomas Gleixnerb0691c82011-10-25 02:30:50 +00001486 bh_unlock_sock(newsk);
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001487 sk_free(newsk);
1488 newsk = NULL;
1489 goto out;
1490 }
1491
1492 newsk->sk_err = 0;
1493 newsk->sk_priority = 0;
Eric Dumazet4dc6dc72009-07-15 23:13:10 +00001494 /*
1495 * Before updating sk_refcnt, we must commit prior changes to memory
1496 * (Documentation/RCU/rculist_nulls.txt for details)
1497 */
1498 smp_wmb();
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001499 atomic_set(&newsk->sk_refcnt, 2);
1500
1501 /*
1502 * Increment the counter in the same struct proto as the master
1503 * sock (sk_refcnt_debug_inc uses newsk->sk_prot->socks, that
1504 * is the same as sk->sk_prot->socks, as this field was copied
1505 * with memcpy).
1506 *
1507 * This _changes_ the previous behaviour, where
1508 * tcp_create_openreq_child always was incrementing the
1509 * equivalent to tcp_prot->socks (inet_sock_nr), so this have
1510 * to be taken into account in all callers. -acme
1511 */
1512 sk_refcnt_debug_inc(newsk);
David S. Miller972692e2008-06-17 22:41:38 -07001513 sk_set_socket(newsk, NULL);
Eric Dumazet43815482010-04-29 11:01:49 +00001514 newsk->sk_wq = NULL;
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001515
Glauber Costaf3f511e2012-01-05 20:16:39 +00001516 sk_update_clone(sk, newsk);
1517
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001518 if (newsk->sk_prot->sockets_allocated)
Glauber Costa180d8cd2011-12-11 21:47:02 +00001519 sk_sockets_allocated_inc(newsk);
Octavian Purdila704da5602010-01-08 00:00:09 -08001520
Eric Dumazet08e29af2011-11-28 12:04:18 +00001521 if (newsk->sk_flags & SK_FLAGS_TIMESTAMP)
Octavian Purdila704da5602010-01-08 00:00:09 -08001522 net_enable_timestamp();
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001523 }
1524out:
1525 return newsk;
1526}
Eric Dumazete56c57d2011-11-08 17:07:07 -05001527EXPORT_SYMBOL_GPL(sk_clone_lock);
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001528
Andi Kleen99580892007-04-20 17:12:43 -07001529void sk_setup_caps(struct sock *sk, struct dst_entry *dst)
1530{
1531 __sk_dst_set(sk, dst);
1532 sk->sk_route_caps = dst->dev->features;
1533 if (sk->sk_route_caps & NETIF_F_GSO)
Herbert Xu4fcd6b92007-05-31 22:15:50 -07001534 sk->sk_route_caps |= NETIF_F_GSO_SOFTWARE;
Eric Dumazeta4654192010-05-16 00:36:33 -07001535 sk->sk_route_caps &= ~sk->sk_route_nocaps;
Andi Kleen99580892007-04-20 17:12:43 -07001536 if (sk_can_gso(sk)) {
Peter P Waskiewicz Jr82cc1a72008-03-21 03:43:19 -07001537 if (dst->header_len) {
Andi Kleen99580892007-04-20 17:12:43 -07001538 sk->sk_route_caps &= ~NETIF_F_GSO_MASK;
Peter P Waskiewicz Jr82cc1a72008-03-21 03:43:19 -07001539 } else {
Andi Kleen99580892007-04-20 17:12:43 -07001540 sk->sk_route_caps |= NETIF_F_SG | NETIF_F_HW_CSUM;
Peter P Waskiewicz Jr82cc1a72008-03-21 03:43:19 -07001541 sk->sk_gso_max_size = dst->dev->gso_max_size;
Ben Hutchings14853482012-07-30 16:11:42 +00001542 sk->sk_gso_max_segs = dst->dev->gso_max_segs;
Peter P Waskiewicz Jr82cc1a72008-03-21 03:43:19 -07001543 }
Andi Kleen99580892007-04-20 17:12:43 -07001544 }
1545}
1546EXPORT_SYMBOL_GPL(sk_setup_caps);
1547
Linus Torvalds1da177e2005-04-16 15:20:36 -07001548/*
1549 * Simple resource managers for sockets.
1550 */
1551
1552
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001553/*
1554 * Write buffer destructor automatically called from kfree_skb.
Linus Torvalds1da177e2005-04-16 15:20:36 -07001555 */
1556void sock_wfree(struct sk_buff *skb)
1557{
1558 struct sock *sk = skb->sk;
Eric Dumazetd99927f2009-09-24 10:49:24 +00001559 unsigned int len = skb->truesize;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001560
Eric Dumazetd99927f2009-09-24 10:49:24 +00001561 if (!sock_flag(sk, SOCK_USE_WRITE_QUEUE)) {
1562 /*
1563 * Keep a reference on sk_wmem_alloc, this will be released
1564 * after sk_write_space() call
1565 */
1566 atomic_sub(len - 1, &sk->sk_wmem_alloc);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001567 sk->sk_write_space(sk);
Eric Dumazetd99927f2009-09-24 10:49:24 +00001568 len = 1;
1569 }
Eric Dumazet2b85a342009-06-11 02:55:43 -07001570 /*
Eric Dumazetd99927f2009-09-24 10:49:24 +00001571 * if sk_wmem_alloc reaches 0, we must finish what sk_free()
1572 * could not do because of in-flight packets
Eric Dumazet2b85a342009-06-11 02:55:43 -07001573 */
Eric Dumazetd99927f2009-09-24 10:49:24 +00001574 if (atomic_sub_and_test(len, &sk->sk_wmem_alloc))
Eric Dumazet2b85a342009-06-11 02:55:43 -07001575 __sk_free(sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001576}
Eric Dumazet2a915252009-05-27 11:30:05 +00001577EXPORT_SYMBOL(sock_wfree);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001578
Eric Dumazetf2f872f2013-07-30 17:55:08 -07001579void skb_orphan_partial(struct sk_buff *skb)
1580{
1581 /* TCP stack sets skb->ooo_okay based on sk_wmem_alloc,
1582 * so we do not completely orphan skb, but transfert all
1583 * accounted bytes but one, to avoid unexpected reorders.
1584 */
1585 if (skb->destructor == sock_wfree
1586#ifdef CONFIG_INET
1587 || skb->destructor == tcp_wfree
1588#endif
1589 ) {
1590 atomic_sub(skb->truesize - 1, &skb->sk->sk_wmem_alloc);
1591 skb->truesize = 1;
1592 } else {
1593 skb_orphan(skb);
1594 }
1595}
1596EXPORT_SYMBOL(skb_orphan_partial);
1597
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001598/*
1599 * Read buffer destructor automatically called from kfree_skb.
Linus Torvalds1da177e2005-04-16 15:20:36 -07001600 */
1601void sock_rfree(struct sk_buff *skb)
1602{
1603 struct sock *sk = skb->sk;
Eric Dumazetd361fd52010-07-10 22:45:17 +00001604 unsigned int len = skb->truesize;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001605
Eric Dumazetd361fd52010-07-10 22:45:17 +00001606 atomic_sub(len, &sk->sk_rmem_alloc);
1607 sk_mem_uncharge(sk, len);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001608}
Eric Dumazet2a915252009-05-27 11:30:05 +00001609EXPORT_SYMBOL(sock_rfree);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001610
David S. Miller41063e92012-06-19 21:22:05 -07001611void sock_edemux(struct sk_buff *skb)
1612{
Eric Dumazete8123472012-09-02 23:57:18 +00001613 struct sock *sk = skb->sk;
1614
Randy Dunlap1c463e52012-09-10 09:13:07 -07001615#ifdef CONFIG_INET
Eric Dumazete8123472012-09-02 23:57:18 +00001616 if (sk->sk_state == TCP_TIME_WAIT)
1617 inet_twsk_put(inet_twsk(sk));
1618 else
Randy Dunlap1c463e52012-09-10 09:13:07 -07001619#endif
Eric Dumazete8123472012-09-02 23:57:18 +00001620 sock_put(sk);
David S. Miller41063e92012-06-19 21:22:05 -07001621}
1622EXPORT_SYMBOL(sock_edemux);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001623
Eric W. Biederman976d02012012-05-23 17:16:53 -06001624kuid_t sock_i_uid(struct sock *sk)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001625{
Eric W. Biederman976d02012012-05-23 17:16:53 -06001626 kuid_t uid;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001627
Eric Dumazetf064af12010-09-22 12:43:39 +00001628 read_lock_bh(&sk->sk_callback_lock);
Eric W. Biederman976d02012012-05-23 17:16:53 -06001629 uid = sk->sk_socket ? SOCK_INODE(sk->sk_socket)->i_uid : GLOBAL_ROOT_UID;
Eric Dumazetf064af12010-09-22 12:43:39 +00001630 read_unlock_bh(&sk->sk_callback_lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001631 return uid;
1632}
Eric Dumazet2a915252009-05-27 11:30:05 +00001633EXPORT_SYMBOL(sock_i_uid);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001634
1635unsigned long sock_i_ino(struct sock *sk)
1636{
1637 unsigned long ino;
1638
Eric Dumazetf064af12010-09-22 12:43:39 +00001639 read_lock_bh(&sk->sk_callback_lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001640 ino = sk->sk_socket ? SOCK_INODE(sk->sk_socket)->i_ino : 0;
Eric Dumazetf064af12010-09-22 12:43:39 +00001641 read_unlock_bh(&sk->sk_callback_lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001642 return ino;
1643}
Eric Dumazet2a915252009-05-27 11:30:05 +00001644EXPORT_SYMBOL(sock_i_ino);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001645
1646/*
1647 * Allocate a skb from the socket's send buffer.
1648 */
Victor Fusco86a76ca2005-07-08 14:57:47 -07001649struct sk_buff *sock_wmalloc(struct sock *sk, unsigned long size, int force,
Al Virodd0fc662005-10-07 07:46:04 +01001650 gfp_t priority)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001651{
1652 if (force || atomic_read(&sk->sk_wmem_alloc) < sk->sk_sndbuf) {
Eric Dumazet2a915252009-05-27 11:30:05 +00001653 struct sk_buff *skb = alloc_skb(size, priority);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001654 if (skb) {
1655 skb_set_owner_w(skb, sk);
1656 return skb;
1657 }
1658 }
1659 return NULL;
1660}
Eric Dumazet2a915252009-05-27 11:30:05 +00001661EXPORT_SYMBOL(sock_wmalloc);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001662
1663/*
1664 * Allocate a skb from the socket's receive buffer.
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001665 */
Victor Fusco86a76ca2005-07-08 14:57:47 -07001666struct sk_buff *sock_rmalloc(struct sock *sk, unsigned long size, int force,
Al Virodd0fc662005-10-07 07:46:04 +01001667 gfp_t priority)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001668{
1669 if (force || atomic_read(&sk->sk_rmem_alloc) < sk->sk_rcvbuf) {
1670 struct sk_buff *skb = alloc_skb(size, priority);
1671 if (skb) {
1672 skb_set_owner_r(skb, sk);
1673 return skb;
1674 }
1675 }
1676 return NULL;
1677}
1678
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001679/*
Linus Torvalds1da177e2005-04-16 15:20:36 -07001680 * Allocate a memory block from the socket's option memory buffer.
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001681 */
Al Virodd0fc662005-10-07 07:46:04 +01001682void *sock_kmalloc(struct sock *sk, int size, gfp_t priority)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001683{
Eric Dumazet95c96172012-04-15 05:58:06 +00001684 if ((unsigned int)size <= sysctl_optmem_max &&
Linus Torvalds1da177e2005-04-16 15:20:36 -07001685 atomic_read(&sk->sk_omem_alloc) + size < sysctl_optmem_max) {
1686 void *mem;
1687 /* First do the add, to avoid the race if kmalloc
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001688 * might sleep.
Linus Torvalds1da177e2005-04-16 15:20:36 -07001689 */
1690 atomic_add(size, &sk->sk_omem_alloc);
1691 mem = kmalloc(size, priority);
1692 if (mem)
1693 return mem;
1694 atomic_sub(size, &sk->sk_omem_alloc);
1695 }
1696 return NULL;
1697}
Eric Dumazet2a915252009-05-27 11:30:05 +00001698EXPORT_SYMBOL(sock_kmalloc);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001699
1700/*
1701 * Free an option memory block.
1702 */
1703void sock_kfree_s(struct sock *sk, void *mem, int size)
1704{
1705 kfree(mem);
1706 atomic_sub(size, &sk->sk_omem_alloc);
1707}
Eric Dumazet2a915252009-05-27 11:30:05 +00001708EXPORT_SYMBOL(sock_kfree_s);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001709
1710/* It is almost wait_for_tcp_memory minus release_sock/lock_sock.
1711 I think, these locks should be removed for datagram sockets.
1712 */
Eric Dumazet2a915252009-05-27 11:30:05 +00001713static long sock_wait_for_wmem(struct sock *sk, long timeo)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001714{
1715 DEFINE_WAIT(wait);
1716
1717 clear_bit(SOCK_ASYNC_NOSPACE, &sk->sk_socket->flags);
1718 for (;;) {
1719 if (!timeo)
1720 break;
1721 if (signal_pending(current))
1722 break;
1723 set_bit(SOCK_NOSPACE, &sk->sk_socket->flags);
Eric Dumazetaa395142010-04-20 13:03:51 +00001724 prepare_to_wait(sk_sleep(sk), &wait, TASK_INTERRUPTIBLE);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001725 if (atomic_read(&sk->sk_wmem_alloc) < sk->sk_sndbuf)
1726 break;
1727 if (sk->sk_shutdown & SEND_SHUTDOWN)
1728 break;
1729 if (sk->sk_err)
1730 break;
1731 timeo = schedule_timeout(timeo);
1732 }
Eric Dumazetaa395142010-04-20 13:03:51 +00001733 finish_wait(sk_sleep(sk), &wait);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001734 return timeo;
1735}
1736
1737
1738/*
1739 * Generic send/receive buffer handlers
1740 */
1741
Herbert Xu4cc7f682009-02-04 16:55:54 -08001742struct sk_buff *sock_alloc_send_pskb(struct sock *sk, unsigned long header_len,
1743 unsigned long data_len, int noblock,
Eric Dumazet28d64272013-08-08 14:38:47 -07001744 int *errcode, int max_page_order)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001745{
Eric Dumazet28d64272013-08-08 14:38:47 -07001746 struct sk_buff *skb = NULL;
1747 unsigned long chunk;
Al Viro7d877f32005-10-21 03:20:43 -04001748 gfp_t gfp_mask;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001749 long timeo;
1750 int err;
Jason Wangcc9b17a2012-05-30 21:18:10 +00001751 int npages = (data_len + (PAGE_SIZE - 1)) >> PAGE_SHIFT;
Eric Dumazet28d64272013-08-08 14:38:47 -07001752 struct page *page;
1753 int i;
Jason Wangcc9b17a2012-05-30 21:18:10 +00001754
1755 err = -EMSGSIZE;
1756 if (npages > MAX_SKB_FRAGS)
1757 goto failure;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001758
Linus Torvalds1da177e2005-04-16 15:20:36 -07001759 timeo = sock_sndtimeo(sk, noblock);
Eric Dumazet28d64272013-08-08 14:38:47 -07001760 while (!skb) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07001761 err = sock_error(sk);
1762 if (err != 0)
1763 goto failure;
1764
1765 err = -EPIPE;
1766 if (sk->sk_shutdown & SEND_SHUTDOWN)
1767 goto failure;
1768
Eric Dumazet28d64272013-08-08 14:38:47 -07001769 if (atomic_read(&sk->sk_wmem_alloc) >= sk->sk_sndbuf) {
1770 set_bit(SOCK_ASYNC_NOSPACE, &sk->sk_socket->flags);
1771 set_bit(SOCK_NOSPACE, &sk->sk_socket->flags);
1772 err = -EAGAIN;
1773 if (!timeo)
1774 goto failure;
1775 if (signal_pending(current))
1776 goto interrupted;
1777 timeo = sock_wait_for_wmem(sk, timeo);
1778 continue;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001779 }
Eric Dumazet28d64272013-08-08 14:38:47 -07001780
1781 err = -ENOBUFS;
1782 gfp_mask = sk->sk_allocation;
1783 if (gfp_mask & __GFP_WAIT)
1784 gfp_mask |= __GFP_REPEAT;
1785
1786 skb = alloc_skb(header_len, gfp_mask);
1787 if (!skb)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001788 goto failure;
Eric Dumazet28d64272013-08-08 14:38:47 -07001789
1790 skb->truesize += data_len;
1791
1792 for (i = 0; npages > 0; i++) {
1793 int order = max_page_order;
1794
1795 while (order) {
1796 if (npages >= 1 << order) {
1797 page = alloc_pages(sk->sk_allocation |
1798 __GFP_COMP | __GFP_NOWARN,
1799 order);
1800 if (page)
1801 goto fill_page;
1802 }
1803 order--;
1804 }
1805 page = alloc_page(sk->sk_allocation);
1806 if (!page)
1807 goto failure;
1808fill_page:
1809 chunk = min_t(unsigned long, data_len,
1810 PAGE_SIZE << order);
1811 skb_fill_page_desc(skb, i, page, 0, chunk);
1812 data_len -= chunk;
1813 npages -= 1 << order;
1814 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07001815 }
1816
1817 skb_set_owner_w(skb, sk);
1818 return skb;
1819
1820interrupted:
1821 err = sock_intr_errno(timeo);
1822failure:
Eric Dumazet28d64272013-08-08 14:38:47 -07001823 kfree_skb(skb);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001824 *errcode = err;
1825 return NULL;
1826}
Herbert Xu4cc7f682009-02-04 16:55:54 -08001827EXPORT_SYMBOL(sock_alloc_send_pskb);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001828
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001829struct sk_buff *sock_alloc_send_skb(struct sock *sk, unsigned long size,
Linus Torvalds1da177e2005-04-16 15:20:36 -07001830 int noblock, int *errcode)
1831{
Eric Dumazet28d64272013-08-08 14:38:47 -07001832 return sock_alloc_send_pskb(sk, size, 0, noblock, errcode, 0);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001833}
Eric Dumazet2a915252009-05-27 11:30:05 +00001834EXPORT_SYMBOL(sock_alloc_send_skb);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001835
Eric Dumazet5640f762012-09-23 23:04:42 +00001836/* On 32bit arches, an skb frag is limited to 2^15 */
1837#define SKB_FRAG_PAGE_ORDER get_order(32768)
1838
1839bool sk_page_frag_refill(struct sock *sk, struct page_frag *pfrag)
1840{
1841 int order;
1842
1843 if (pfrag->page) {
1844 if (atomic_read(&pfrag->page->_count) == 1) {
1845 pfrag->offset = 0;
1846 return true;
1847 }
1848 if (pfrag->offset < pfrag->size)
1849 return true;
1850 put_page(pfrag->page);
1851 }
1852
1853 /* We restrict high order allocations to users that can afford to wait */
1854 order = (sk->sk_allocation & __GFP_WAIT) ? SKB_FRAG_PAGE_ORDER : 0;
1855
1856 do {
1857 gfp_t gfp = sk->sk_allocation;
1858
1859 if (order)
1860 gfp |= __GFP_COMP | __GFP_NOWARN;
1861 pfrag->page = alloc_pages(gfp, order);
1862 if (likely(pfrag->page)) {
1863 pfrag->offset = 0;
1864 pfrag->size = PAGE_SIZE << order;
1865 return true;
1866 }
1867 } while (--order >= 0);
1868
1869 sk_enter_memory_pressure(sk);
1870 sk_stream_moderate_sndbuf(sk);
1871 return false;
1872}
1873EXPORT_SYMBOL(sk_page_frag_refill);
1874
Linus Torvalds1da177e2005-04-16 15:20:36 -07001875static void __lock_sock(struct sock *sk)
Namhyung Kimf39234d2010-09-08 03:48:48 +00001876 __releases(&sk->sk_lock.slock)
1877 __acquires(&sk->sk_lock.slock)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001878{
1879 DEFINE_WAIT(wait);
1880
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001881 for (;;) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07001882 prepare_to_wait_exclusive(&sk->sk_lock.wq, &wait,
1883 TASK_UNINTERRUPTIBLE);
1884 spin_unlock_bh(&sk->sk_lock.slock);
1885 schedule();
1886 spin_lock_bh(&sk->sk_lock.slock);
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001887 if (!sock_owned_by_user(sk))
Linus Torvalds1da177e2005-04-16 15:20:36 -07001888 break;
1889 }
1890 finish_wait(&sk->sk_lock.wq, &wait);
1891}
1892
1893static void __release_sock(struct sock *sk)
Namhyung Kimf39234d2010-09-08 03:48:48 +00001894 __releases(&sk->sk_lock.slock)
1895 __acquires(&sk->sk_lock.slock)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001896{
1897 struct sk_buff *skb = sk->sk_backlog.head;
1898
1899 do {
1900 sk->sk_backlog.head = sk->sk_backlog.tail = NULL;
1901 bh_unlock_sock(sk);
1902
1903 do {
1904 struct sk_buff *next = skb->next;
1905
Eric Dumazete4cbb022012-04-30 16:07:09 +00001906 prefetch(next);
Eric Dumazet7fee2262010-05-11 23:19:48 +00001907 WARN_ON_ONCE(skb_dst_is_noref(skb));
Linus Torvalds1da177e2005-04-16 15:20:36 -07001908 skb->next = NULL;
Peter Zijlstrac57943a2008-10-07 14:18:42 -07001909 sk_backlog_rcv(sk, skb);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001910
1911 /*
1912 * We are in process context here with softirqs
1913 * disabled, use cond_resched_softirq() to preempt.
1914 * This is safe to do because we've taken the backlog
1915 * queue private:
1916 */
1917 cond_resched_softirq();
1918
1919 skb = next;
1920 } while (skb != NULL);
1921
1922 bh_lock_sock(sk);
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001923 } while ((skb = sk->sk_backlog.head) != NULL);
Zhu Yi8eae9392010-03-04 18:01:40 +00001924
1925 /*
1926 * Doing the zeroing here guarantee we can not loop forever
1927 * while a wild producer attempts to flood us.
1928 */
1929 sk->sk_backlog.len = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001930}
1931
1932/**
1933 * sk_wait_data - wait for data to arrive at sk_receive_queue
Pavel Pisa4dc3b162005-05-01 08:59:25 -07001934 * @sk: sock to wait on
1935 * @timeo: for how long
Linus Torvalds1da177e2005-04-16 15:20:36 -07001936 *
1937 * Now socket state including sk->sk_err is changed only under lock,
1938 * hence we may omit checks after joining wait queue.
1939 * We check receive queue before schedule() only as optimization;
1940 * it is very likely that release_sock() added new data.
1941 */
1942int sk_wait_data(struct sock *sk, long *timeo)
1943{
1944 int rc;
1945 DEFINE_WAIT(wait);
1946
Eric Dumazetaa395142010-04-20 13:03:51 +00001947 prepare_to_wait(sk_sleep(sk), &wait, TASK_INTERRUPTIBLE);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001948 set_bit(SOCK_ASYNC_WAITDATA, &sk->sk_socket->flags);
1949 rc = sk_wait_event(sk, timeo, !skb_queue_empty(&sk->sk_receive_queue));
1950 clear_bit(SOCK_ASYNC_WAITDATA, &sk->sk_socket->flags);
Eric Dumazetaa395142010-04-20 13:03:51 +00001951 finish_wait(sk_sleep(sk), &wait);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001952 return rc;
1953}
Linus Torvalds1da177e2005-04-16 15:20:36 -07001954EXPORT_SYMBOL(sk_wait_data);
1955
Hideo Aoki3ab224b2007-12-31 00:11:19 -08001956/**
1957 * __sk_mem_schedule - increase sk_forward_alloc and memory_allocated
1958 * @sk: socket
1959 * @size: memory size to allocate
1960 * @kind: allocation type
1961 *
1962 * If kind is SK_MEM_SEND, it means wmem allocation. Otherwise it means
1963 * rmem allocation. This function assumes that protocols which have
1964 * memory_pressure use sk_wmem_queued as write buffer accounting.
1965 */
1966int __sk_mem_schedule(struct sock *sk, int size, int kind)
1967{
1968 struct proto *prot = sk->sk_prot;
1969 int amt = sk_mem_pages(size);
Eric Dumazet8d987e52010-11-09 23:24:26 +00001970 long allocated;
Glauber Costae1aab162011-12-11 21:47:03 +00001971 int parent_status = UNDER_LIMIT;
Hideo Aoki3ab224b2007-12-31 00:11:19 -08001972
1973 sk->sk_forward_alloc += amt * SK_MEM_QUANTUM;
Glauber Costa180d8cd2011-12-11 21:47:02 +00001974
Glauber Costae1aab162011-12-11 21:47:03 +00001975 allocated = sk_memory_allocated_add(sk, amt, &parent_status);
Hideo Aoki3ab224b2007-12-31 00:11:19 -08001976
1977 /* Under limit. */
Glauber Costae1aab162011-12-11 21:47:03 +00001978 if (parent_status == UNDER_LIMIT &&
1979 allocated <= sk_prot_mem_limits(sk, 0)) {
Glauber Costa180d8cd2011-12-11 21:47:02 +00001980 sk_leave_memory_pressure(sk);
Hideo Aoki3ab224b2007-12-31 00:11:19 -08001981 return 1;
1982 }
1983
Glauber Costae1aab162011-12-11 21:47:03 +00001984 /* Under pressure. (we or our parents) */
1985 if ((parent_status > SOFT_LIMIT) ||
1986 allocated > sk_prot_mem_limits(sk, 1))
Glauber Costa180d8cd2011-12-11 21:47:02 +00001987 sk_enter_memory_pressure(sk);
Hideo Aoki3ab224b2007-12-31 00:11:19 -08001988
Glauber Costae1aab162011-12-11 21:47:03 +00001989 /* Over hard limit (we or our parents) */
1990 if ((parent_status == OVER_LIMIT) ||
1991 (allocated > sk_prot_mem_limits(sk, 2)))
Hideo Aoki3ab224b2007-12-31 00:11:19 -08001992 goto suppress_allocation;
1993
1994 /* guarantee minimum buffer size under pressure */
1995 if (kind == SK_MEM_RECV) {
1996 if (atomic_read(&sk->sk_rmem_alloc) < prot->sysctl_rmem[0])
1997 return 1;
Glauber Costa180d8cd2011-12-11 21:47:02 +00001998
Hideo Aoki3ab224b2007-12-31 00:11:19 -08001999 } else { /* SK_MEM_SEND */
2000 if (sk->sk_type == SOCK_STREAM) {
2001 if (sk->sk_wmem_queued < prot->sysctl_wmem[0])
2002 return 1;
2003 } else if (atomic_read(&sk->sk_wmem_alloc) <
2004 prot->sysctl_wmem[0])
2005 return 1;
2006 }
2007
Glauber Costa180d8cd2011-12-11 21:47:02 +00002008 if (sk_has_memory_pressure(sk)) {
Eric Dumazet17483762008-11-25 21:16:35 -08002009 int alloc;
2010
Glauber Costa180d8cd2011-12-11 21:47:02 +00002011 if (!sk_under_memory_pressure(sk))
Eric Dumazet17483762008-11-25 21:16:35 -08002012 return 1;
Glauber Costa180d8cd2011-12-11 21:47:02 +00002013 alloc = sk_sockets_allocated_read_positive(sk);
2014 if (sk_prot_mem_limits(sk, 2) > alloc *
Hideo Aoki3ab224b2007-12-31 00:11:19 -08002015 sk_mem_pages(sk->sk_wmem_queued +
2016 atomic_read(&sk->sk_rmem_alloc) +
2017 sk->sk_forward_alloc))
2018 return 1;
2019 }
2020
2021suppress_allocation:
2022
2023 if (kind == SK_MEM_SEND && sk->sk_type == SOCK_STREAM) {
2024 sk_stream_moderate_sndbuf(sk);
2025
2026 /* Fail only if socket is _under_ its sndbuf.
2027 * In this case we cannot block, so that we have to fail.
2028 */
2029 if (sk->sk_wmem_queued + size >= sk->sk_sndbuf)
2030 return 1;
2031 }
2032
Satoru Moriya3847ce32011-06-17 12:00:03 +00002033 trace_sock_exceed_buf_limit(sk, prot, allocated);
2034
Hideo Aoki3ab224b2007-12-31 00:11:19 -08002035 /* Alas. Undo changes. */
2036 sk->sk_forward_alloc -= amt * SK_MEM_QUANTUM;
Glauber Costa180d8cd2011-12-11 21:47:02 +00002037
Glauber Costa0e90b312012-01-20 04:57:16 +00002038 sk_memory_allocated_sub(sk, amt);
Glauber Costa180d8cd2011-12-11 21:47:02 +00002039
Hideo Aoki3ab224b2007-12-31 00:11:19 -08002040 return 0;
2041}
Hideo Aoki3ab224b2007-12-31 00:11:19 -08002042EXPORT_SYMBOL(__sk_mem_schedule);
2043
2044/**
2045 * __sk_reclaim - reclaim memory_allocated
2046 * @sk: socket
2047 */
2048void __sk_mem_reclaim(struct sock *sk)
2049{
Glauber Costa180d8cd2011-12-11 21:47:02 +00002050 sk_memory_allocated_sub(sk,
Glauber Costa0e90b312012-01-20 04:57:16 +00002051 sk->sk_forward_alloc >> SK_MEM_QUANTUM_SHIFT);
Hideo Aoki3ab224b2007-12-31 00:11:19 -08002052 sk->sk_forward_alloc &= SK_MEM_QUANTUM - 1;
2053
Glauber Costa180d8cd2011-12-11 21:47:02 +00002054 if (sk_under_memory_pressure(sk) &&
2055 (sk_memory_allocated(sk) < sk_prot_mem_limits(sk, 0)))
2056 sk_leave_memory_pressure(sk);
Hideo Aoki3ab224b2007-12-31 00:11:19 -08002057}
Hideo Aoki3ab224b2007-12-31 00:11:19 -08002058EXPORT_SYMBOL(__sk_mem_reclaim);
2059
2060
Linus Torvalds1da177e2005-04-16 15:20:36 -07002061/*
2062 * Set of default routines for initialising struct proto_ops when
2063 * the protocol does not support a particular function. In certain
2064 * cases where it makes no sense for a protocol to have a "do nothing"
2065 * function, some default processing is provided.
2066 */
2067
2068int sock_no_bind(struct socket *sock, struct sockaddr *saddr, int len)
2069{
2070 return -EOPNOTSUPP;
2071}
Eric Dumazet2a915252009-05-27 11:30:05 +00002072EXPORT_SYMBOL(sock_no_bind);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002073
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09002074int sock_no_connect(struct socket *sock, struct sockaddr *saddr,
Linus Torvalds1da177e2005-04-16 15:20:36 -07002075 int len, int flags)
2076{
2077 return -EOPNOTSUPP;
2078}
Eric Dumazet2a915252009-05-27 11:30:05 +00002079EXPORT_SYMBOL(sock_no_connect);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002080
2081int sock_no_socketpair(struct socket *sock1, struct socket *sock2)
2082{
2083 return -EOPNOTSUPP;
2084}
Eric Dumazet2a915252009-05-27 11:30:05 +00002085EXPORT_SYMBOL(sock_no_socketpair);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002086
2087int sock_no_accept(struct socket *sock, struct socket *newsock, int flags)
2088{
2089 return -EOPNOTSUPP;
2090}
Eric Dumazet2a915252009-05-27 11:30:05 +00002091EXPORT_SYMBOL(sock_no_accept);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002092
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09002093int sock_no_getname(struct socket *sock, struct sockaddr *saddr,
Linus Torvalds1da177e2005-04-16 15:20:36 -07002094 int *len, int peer)
2095{
2096 return -EOPNOTSUPP;
2097}
Eric Dumazet2a915252009-05-27 11:30:05 +00002098EXPORT_SYMBOL(sock_no_getname);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002099
Eric Dumazet2a915252009-05-27 11:30:05 +00002100unsigned int sock_no_poll(struct file *file, struct socket *sock, poll_table *pt)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002101{
2102 return 0;
2103}
Eric Dumazet2a915252009-05-27 11:30:05 +00002104EXPORT_SYMBOL(sock_no_poll);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002105
2106int sock_no_ioctl(struct socket *sock, unsigned int cmd, unsigned long arg)
2107{
2108 return -EOPNOTSUPP;
2109}
Eric Dumazet2a915252009-05-27 11:30:05 +00002110EXPORT_SYMBOL(sock_no_ioctl);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002111
2112int sock_no_listen(struct socket *sock, int backlog)
2113{
2114 return -EOPNOTSUPP;
2115}
Eric Dumazet2a915252009-05-27 11:30:05 +00002116EXPORT_SYMBOL(sock_no_listen);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002117
2118int sock_no_shutdown(struct socket *sock, int how)
2119{
2120 return -EOPNOTSUPP;
2121}
Eric Dumazet2a915252009-05-27 11:30:05 +00002122EXPORT_SYMBOL(sock_no_shutdown);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002123
2124int sock_no_setsockopt(struct socket *sock, int level, int optname,
David S. Millerb7058842009-09-30 16:12:20 -07002125 char __user *optval, unsigned int optlen)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002126{
2127 return -EOPNOTSUPP;
2128}
Eric Dumazet2a915252009-05-27 11:30:05 +00002129EXPORT_SYMBOL(sock_no_setsockopt);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002130
2131int sock_no_getsockopt(struct socket *sock, int level, int optname,
2132 char __user *optval, int __user *optlen)
2133{
2134 return -EOPNOTSUPP;
2135}
Eric Dumazet2a915252009-05-27 11:30:05 +00002136EXPORT_SYMBOL(sock_no_getsockopt);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002137
2138int sock_no_sendmsg(struct kiocb *iocb, struct socket *sock, struct msghdr *m,
2139 size_t len)
2140{
2141 return -EOPNOTSUPP;
2142}
Eric Dumazet2a915252009-05-27 11:30:05 +00002143EXPORT_SYMBOL(sock_no_sendmsg);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002144
2145int sock_no_recvmsg(struct kiocb *iocb, struct socket *sock, struct msghdr *m,
2146 size_t len, int flags)
2147{
2148 return -EOPNOTSUPP;
2149}
Eric Dumazet2a915252009-05-27 11:30:05 +00002150EXPORT_SYMBOL(sock_no_recvmsg);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002151
2152int sock_no_mmap(struct file *file, struct socket *sock, struct vm_area_struct *vma)
2153{
2154 /* Mirror missing mmap method error code */
2155 return -ENODEV;
2156}
Eric Dumazet2a915252009-05-27 11:30:05 +00002157EXPORT_SYMBOL(sock_no_mmap);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002158
2159ssize_t sock_no_sendpage(struct socket *sock, struct page *page, int offset, size_t size, int flags)
2160{
2161 ssize_t res;
2162 struct msghdr msg = {.msg_flags = flags};
2163 struct kvec iov;
2164 char *kaddr = kmap(page);
2165 iov.iov_base = kaddr + offset;
2166 iov.iov_len = size;
2167 res = kernel_sendmsg(sock, &msg, &iov, 1, size);
2168 kunmap(page);
2169 return res;
2170}
Eric Dumazet2a915252009-05-27 11:30:05 +00002171EXPORT_SYMBOL(sock_no_sendpage);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002172
2173/*
2174 * Default Socket Callbacks
2175 */
2176
2177static void sock_def_wakeup(struct sock *sk)
2178{
Eric Dumazet43815482010-04-29 11:01:49 +00002179 struct socket_wq *wq;
2180
2181 rcu_read_lock();
2182 wq = rcu_dereference(sk->sk_wq);
2183 if (wq_has_sleeper(wq))
2184 wake_up_interruptible_all(&wq->wait);
2185 rcu_read_unlock();
Linus Torvalds1da177e2005-04-16 15:20:36 -07002186}
2187
2188static void sock_def_error_report(struct sock *sk)
2189{
Eric Dumazet43815482010-04-29 11:01:49 +00002190 struct socket_wq *wq;
2191
2192 rcu_read_lock();
2193 wq = rcu_dereference(sk->sk_wq);
2194 if (wq_has_sleeper(wq))
2195 wake_up_interruptible_poll(&wq->wait, POLLERR);
Pavel Emelyanov8d8ad9d2007-11-26 20:10:50 +08002196 sk_wake_async(sk, SOCK_WAKE_IO, POLL_ERR);
Eric Dumazet43815482010-04-29 11:01:49 +00002197 rcu_read_unlock();
Linus Torvalds1da177e2005-04-16 15:20:36 -07002198}
2199
2200static void sock_def_readable(struct sock *sk, int len)
2201{
Eric Dumazet43815482010-04-29 11:01:49 +00002202 struct socket_wq *wq;
2203
2204 rcu_read_lock();
2205 wq = rcu_dereference(sk->sk_wq);
2206 if (wq_has_sleeper(wq))
Eric Dumazet2c6607c2011-01-06 10:54:29 -08002207 wake_up_interruptible_sync_poll(&wq->wait, POLLIN | POLLPRI |
Davide Libenzi37e55402009-03-31 15:24:21 -07002208 POLLRDNORM | POLLRDBAND);
Pavel Emelyanov8d8ad9d2007-11-26 20:10:50 +08002209 sk_wake_async(sk, SOCK_WAKE_WAITD, POLL_IN);
Eric Dumazet43815482010-04-29 11:01:49 +00002210 rcu_read_unlock();
Linus Torvalds1da177e2005-04-16 15:20:36 -07002211}
2212
2213static void sock_def_write_space(struct sock *sk)
2214{
Eric Dumazet43815482010-04-29 11:01:49 +00002215 struct socket_wq *wq;
2216
2217 rcu_read_lock();
Linus Torvalds1da177e2005-04-16 15:20:36 -07002218
2219 /* Do not wake up a writer until he can make "significant"
2220 * progress. --DaveM
2221 */
Stephen Hemmingere71a4782007-04-10 20:10:33 -07002222 if ((atomic_read(&sk->sk_wmem_alloc) << 1) <= sk->sk_sndbuf) {
Eric Dumazet43815482010-04-29 11:01:49 +00002223 wq = rcu_dereference(sk->sk_wq);
2224 if (wq_has_sleeper(wq))
2225 wake_up_interruptible_sync_poll(&wq->wait, POLLOUT |
Davide Libenzi37e55402009-03-31 15:24:21 -07002226 POLLWRNORM | POLLWRBAND);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002227
2228 /* Should agree with poll, otherwise some programs break */
2229 if (sock_writeable(sk))
Pavel Emelyanov8d8ad9d2007-11-26 20:10:50 +08002230 sk_wake_async(sk, SOCK_WAKE_SPACE, POLL_OUT);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002231 }
2232
Eric Dumazet43815482010-04-29 11:01:49 +00002233 rcu_read_unlock();
Linus Torvalds1da177e2005-04-16 15:20:36 -07002234}
2235
2236static void sock_def_destruct(struct sock *sk)
2237{
Jesper Juhla51482b2005-11-08 09:41:34 -08002238 kfree(sk->sk_protinfo);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002239}
2240
2241void sk_send_sigurg(struct sock *sk)
2242{
2243 if (sk->sk_socket && sk->sk_socket->file)
2244 if (send_sigurg(&sk->sk_socket->file->f_owner))
Pavel Emelyanov8d8ad9d2007-11-26 20:10:50 +08002245 sk_wake_async(sk, SOCK_WAKE_URG, POLL_PRI);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002246}
Eric Dumazet2a915252009-05-27 11:30:05 +00002247EXPORT_SYMBOL(sk_send_sigurg);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002248
2249void sk_reset_timer(struct sock *sk, struct timer_list* timer,
2250 unsigned long expires)
2251{
2252 if (!mod_timer(timer, expires))
2253 sock_hold(sk);
2254}
Linus Torvalds1da177e2005-04-16 15:20:36 -07002255EXPORT_SYMBOL(sk_reset_timer);
2256
2257void sk_stop_timer(struct sock *sk, struct timer_list* timer)
2258{
Ying Xue25cc4ae2013-02-03 20:32:57 +00002259 if (del_timer(timer))
Linus Torvalds1da177e2005-04-16 15:20:36 -07002260 __sock_put(sk);
2261}
Linus Torvalds1da177e2005-04-16 15:20:36 -07002262EXPORT_SYMBOL(sk_stop_timer);
2263
2264void sock_init_data(struct socket *sock, struct sock *sk)
2265{
2266 skb_queue_head_init(&sk->sk_receive_queue);
2267 skb_queue_head_init(&sk->sk_write_queue);
2268 skb_queue_head_init(&sk->sk_error_queue);
Chris Leech97fc2f02006-05-23 17:55:33 -07002269#ifdef CONFIG_NET_DMA
2270 skb_queue_head_init(&sk->sk_async_wait_queue);
2271#endif
Linus Torvalds1da177e2005-04-16 15:20:36 -07002272
2273 sk->sk_send_head = NULL;
2274
2275 init_timer(&sk->sk_timer);
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09002276
Linus Torvalds1da177e2005-04-16 15:20:36 -07002277 sk->sk_allocation = GFP_KERNEL;
2278 sk->sk_rcvbuf = sysctl_rmem_default;
2279 sk->sk_sndbuf = sysctl_wmem_default;
2280 sk->sk_state = TCP_CLOSE;
David S. Miller972692e2008-06-17 22:41:38 -07002281 sk_set_socket(sk, sock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002282
2283 sock_set_flag(sk, SOCK_ZAPPED);
2284
Stephen Hemmingere71a4782007-04-10 20:10:33 -07002285 if (sock) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07002286 sk->sk_type = sock->type;
Eric Dumazet43815482010-04-29 11:01:49 +00002287 sk->sk_wq = sock->wq;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002288 sock->sk = sk;
2289 } else
Eric Dumazet43815482010-04-29 11:01:49 +00002290 sk->sk_wq = NULL;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002291
Eric Dumazetb6c67122010-04-08 23:03:29 +00002292 spin_lock_init(&sk->sk_dst_lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002293 rwlock_init(&sk->sk_callback_lock);
Peter Zijlstra443aef02007-07-19 01:49:00 -07002294 lockdep_set_class_and_name(&sk->sk_callback_lock,
2295 af_callback_keys + sk->sk_family,
2296 af_family_clock_key_strings[sk->sk_family]);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002297
2298 sk->sk_state_change = sock_def_wakeup;
2299 sk->sk_data_ready = sock_def_readable;
2300 sk->sk_write_space = sock_def_write_space;
2301 sk->sk_error_report = sock_def_error_report;
2302 sk->sk_destruct = sock_def_destruct;
2303
Eric Dumazet5640f762012-09-23 23:04:42 +00002304 sk->sk_frag.page = NULL;
2305 sk->sk_frag.offset = 0;
Pavel Emelyanovef64a542012-02-21 07:31:34 +00002306 sk->sk_peek_off = -1;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002307
Eric W. Biederman109f6e32010-06-13 03:30:14 +00002308 sk->sk_peer_pid = NULL;
2309 sk->sk_peer_cred = NULL;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002310 sk->sk_write_pending = 0;
2311 sk->sk_rcvlowat = 1;
2312 sk->sk_rcvtimeo = MAX_SCHEDULE_TIMEOUT;
2313 sk->sk_sndtimeo = MAX_SCHEDULE_TIMEOUT;
2314
Eric Dumazetf37f0af2008-04-13 21:39:26 -07002315 sk->sk_stamp = ktime_set(-1L, 0);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002316
Cong Wange0d10952013-08-01 11:10:25 +08002317#ifdef CONFIG_NET_RX_BUSY_POLL
Eliezer Tamir06021292013-06-10 11:39:50 +03002318 sk->sk_napi_id = 0;
Eliezer Tamir64b0dc52013-07-10 17:13:36 +03002319 sk->sk_ll_usec = sysctl_net_busy_read;
Eliezer Tamir06021292013-06-10 11:39:50 +03002320#endif
2321
Eric Dumazet7eec4172013-10-08 15:16:00 -07002322 sk->sk_pacing_rate = ~0U;
Eric Dumazet4dc6dc72009-07-15 23:13:10 +00002323 /*
2324 * Before updating sk_refcnt, we must commit prior changes to memory
2325 * (Documentation/RCU/rculist_nulls.txt for details)
2326 */
2327 smp_wmb();
Linus Torvalds1da177e2005-04-16 15:20:36 -07002328 atomic_set(&sk->sk_refcnt, 1);
Wang Chen33c732c2007-11-13 20:30:01 -08002329 atomic_set(&sk->sk_drops, 0);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002330}
Eric Dumazet2a915252009-05-27 11:30:05 +00002331EXPORT_SYMBOL(sock_init_data);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002332
Harvey Harrisonb5606c22008-02-13 15:03:16 -08002333void lock_sock_nested(struct sock *sk, int subclass)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002334{
2335 might_sleep();
Ingo Molnara5b5bb92006-07-03 00:25:35 -07002336 spin_lock_bh(&sk->sk_lock.slock);
John Heffnerd2e91172007-09-12 10:44:19 +02002337 if (sk->sk_lock.owned)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002338 __lock_sock(sk);
John Heffnerd2e91172007-09-12 10:44:19 +02002339 sk->sk_lock.owned = 1;
Ingo Molnara5b5bb92006-07-03 00:25:35 -07002340 spin_unlock(&sk->sk_lock.slock);
2341 /*
2342 * The sk_lock has mutex_lock() semantics here:
2343 */
Peter Zijlstrafcc70d52006-11-08 22:44:35 -08002344 mutex_acquire(&sk->sk_lock.dep_map, subclass, 0, _RET_IP_);
Ingo Molnara5b5bb92006-07-03 00:25:35 -07002345 local_bh_enable();
Linus Torvalds1da177e2005-04-16 15:20:36 -07002346}
Peter Zijlstrafcc70d52006-11-08 22:44:35 -08002347EXPORT_SYMBOL(lock_sock_nested);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002348
Harvey Harrisonb5606c22008-02-13 15:03:16 -08002349void release_sock(struct sock *sk)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002350{
Ingo Molnara5b5bb92006-07-03 00:25:35 -07002351 /*
2352 * The sk_lock has mutex_unlock() semantics:
2353 */
2354 mutex_release(&sk->sk_lock.dep_map, 1, _RET_IP_);
2355
2356 spin_lock_bh(&sk->sk_lock.slock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002357 if (sk->sk_backlog.tail)
2358 __release_sock(sk);
Eric Dumazet46d3cea2012-07-11 05:50:31 +00002359
2360 if (sk->sk_prot->release_cb)
2361 sk->sk_prot->release_cb(sk);
2362
John Heffnerd2e91172007-09-12 10:44:19 +02002363 sk->sk_lock.owned = 0;
Ingo Molnara5b5bb92006-07-03 00:25:35 -07002364 if (waitqueue_active(&sk->sk_lock.wq))
2365 wake_up(&sk->sk_lock.wq);
2366 spin_unlock_bh(&sk->sk_lock.slock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002367}
2368EXPORT_SYMBOL(release_sock);
2369
Eric Dumazet8a74ad62010-05-26 19:20:18 +00002370/**
2371 * lock_sock_fast - fast version of lock_sock
2372 * @sk: socket
2373 *
2374 * This version should be used for very small section, where process wont block
2375 * return false if fast path is taken
2376 * sk_lock.slock locked, owned = 0, BH disabled
2377 * return true if slow path is taken
2378 * sk_lock.slock unlocked, owned = 1, BH enabled
2379 */
2380bool lock_sock_fast(struct sock *sk)
2381{
2382 might_sleep();
2383 spin_lock_bh(&sk->sk_lock.slock);
2384
2385 if (!sk->sk_lock.owned)
2386 /*
2387 * Note : We must disable BH
2388 */
2389 return false;
2390
2391 __lock_sock(sk);
2392 sk->sk_lock.owned = 1;
2393 spin_unlock(&sk->sk_lock.slock);
2394 /*
2395 * The sk_lock has mutex_lock() semantics here:
2396 */
2397 mutex_acquire(&sk->sk_lock.dep_map, 0, 0, _RET_IP_);
2398 local_bh_enable();
2399 return true;
2400}
2401EXPORT_SYMBOL(lock_sock_fast);
2402
Linus Torvalds1da177e2005-04-16 15:20:36 -07002403int sock_get_timestamp(struct sock *sk, struct timeval __user *userstamp)
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09002404{
Eric Dumazetb7aa0bf2007-04-19 16:16:32 -07002405 struct timeval tv;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002406 if (!sock_flag(sk, SOCK_TIMESTAMP))
Patrick Ohly20d49472009-02-12 05:03:38 +00002407 sock_enable_timestamp(sk, SOCK_TIMESTAMP);
Eric Dumazetb7aa0bf2007-04-19 16:16:32 -07002408 tv = ktime_to_timeval(sk->sk_stamp);
2409 if (tv.tv_sec == -1)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002410 return -ENOENT;
Eric Dumazetb7aa0bf2007-04-19 16:16:32 -07002411 if (tv.tv_sec == 0) {
2412 sk->sk_stamp = ktime_get_real();
2413 tv = ktime_to_timeval(sk->sk_stamp);
2414 }
2415 return copy_to_user(userstamp, &tv, sizeof(tv)) ? -EFAULT : 0;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09002416}
Linus Torvalds1da177e2005-04-16 15:20:36 -07002417EXPORT_SYMBOL(sock_get_timestamp);
2418
Eric Dumazetae40eb12007-03-18 17:33:16 -07002419int sock_get_timestampns(struct sock *sk, struct timespec __user *userstamp)
2420{
2421 struct timespec ts;
2422 if (!sock_flag(sk, SOCK_TIMESTAMP))
Patrick Ohly20d49472009-02-12 05:03:38 +00002423 sock_enable_timestamp(sk, SOCK_TIMESTAMP);
Eric Dumazetae40eb12007-03-18 17:33:16 -07002424 ts = ktime_to_timespec(sk->sk_stamp);
2425 if (ts.tv_sec == -1)
2426 return -ENOENT;
2427 if (ts.tv_sec == 0) {
2428 sk->sk_stamp = ktime_get_real();
2429 ts = ktime_to_timespec(sk->sk_stamp);
2430 }
2431 return copy_to_user(userstamp, &ts, sizeof(ts)) ? -EFAULT : 0;
2432}
2433EXPORT_SYMBOL(sock_get_timestampns);
2434
Patrick Ohly20d49472009-02-12 05:03:38 +00002435void sock_enable_timestamp(struct sock *sk, int flag)
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09002436{
Patrick Ohly20d49472009-02-12 05:03:38 +00002437 if (!sock_flag(sk, flag)) {
Eric Dumazet08e29af2011-11-28 12:04:18 +00002438 unsigned long previous_flags = sk->sk_flags;
2439
Patrick Ohly20d49472009-02-12 05:03:38 +00002440 sock_set_flag(sk, flag);
2441 /*
2442 * we just set one of the two flags which require net
2443 * time stamping, but time stamping might have been on
2444 * already because of the other one
2445 */
Eric Dumazet08e29af2011-11-28 12:04:18 +00002446 if (!(previous_flags & SK_FLAGS_TIMESTAMP))
Patrick Ohly20d49472009-02-12 05:03:38 +00002447 net_enable_timestamp();
Linus Torvalds1da177e2005-04-16 15:20:36 -07002448 }
2449}
Linus Torvalds1da177e2005-04-16 15:20:36 -07002450
Richard Cochrancb820f82013-07-19 19:40:09 +02002451int sock_recv_errqueue(struct sock *sk, struct msghdr *msg, int len,
2452 int level, int type)
2453{
2454 struct sock_exterr_skb *serr;
2455 struct sk_buff *skb, *skb2;
2456 int copied, err;
2457
2458 err = -EAGAIN;
2459 skb = skb_dequeue(&sk->sk_error_queue);
2460 if (skb == NULL)
2461 goto out;
2462
2463 copied = skb->len;
2464 if (copied > len) {
2465 msg->msg_flags |= MSG_TRUNC;
2466 copied = len;
2467 }
2468 err = skb_copy_datagram_iovec(skb, 0, msg->msg_iov, copied);
2469 if (err)
2470 goto out_free_skb;
2471
2472 sock_recv_timestamp(msg, sk, skb);
2473
2474 serr = SKB_EXT_ERR(skb);
2475 put_cmsg(msg, level, type, sizeof(serr->ee), &serr->ee);
2476
2477 msg->msg_flags |= MSG_ERRQUEUE;
2478 err = copied;
2479
2480 /* Reset and regenerate socket error */
2481 spin_lock_bh(&sk->sk_error_queue.lock);
2482 sk->sk_err = 0;
2483 if ((skb2 = skb_peek(&sk->sk_error_queue)) != NULL) {
2484 sk->sk_err = SKB_EXT_ERR(skb2)->ee.ee_errno;
2485 spin_unlock_bh(&sk->sk_error_queue.lock);
2486 sk->sk_error_report(sk);
2487 } else
2488 spin_unlock_bh(&sk->sk_error_queue.lock);
2489
2490out_free_skb:
2491 kfree_skb(skb);
2492out:
2493 return err;
2494}
2495EXPORT_SYMBOL(sock_recv_errqueue);
2496
Linus Torvalds1da177e2005-04-16 15:20:36 -07002497/*
2498 * Get a socket option on an socket.
2499 *
2500 * FIX: POSIX 1003.1g is very ambiguous here. It states that
2501 * asynchronous errors should be reported by getsockopt. We assume
2502 * this means if you specify SO_ERROR (otherwise whats the point of it).
2503 */
2504int sock_common_getsockopt(struct socket *sock, int level, int optname,
2505 char __user *optval, int __user *optlen)
2506{
2507 struct sock *sk = sock->sk;
2508
2509 return sk->sk_prot->getsockopt(sk, level, optname, optval, optlen);
2510}
Linus Torvalds1da177e2005-04-16 15:20:36 -07002511EXPORT_SYMBOL(sock_common_getsockopt);
2512
Dmitry Mishin3fdadf72006-03-20 22:45:21 -08002513#ifdef CONFIG_COMPAT
Arnaldo Carvalho de Melo543d9cf2006-03-20 22:48:35 -08002514int compat_sock_common_getsockopt(struct socket *sock, int level, int optname,
2515 char __user *optval, int __user *optlen)
Dmitry Mishin3fdadf72006-03-20 22:45:21 -08002516{
2517 struct sock *sk = sock->sk;
2518
Johannes Berg1e51f952007-03-06 13:44:06 -08002519 if (sk->sk_prot->compat_getsockopt != NULL)
Arnaldo Carvalho de Melo543d9cf2006-03-20 22:48:35 -08002520 return sk->sk_prot->compat_getsockopt(sk, level, optname,
2521 optval, optlen);
Dmitry Mishin3fdadf72006-03-20 22:45:21 -08002522 return sk->sk_prot->getsockopt(sk, level, optname, optval, optlen);
2523}
2524EXPORT_SYMBOL(compat_sock_common_getsockopt);
2525#endif
2526
Linus Torvalds1da177e2005-04-16 15:20:36 -07002527int sock_common_recvmsg(struct kiocb *iocb, struct socket *sock,
2528 struct msghdr *msg, size_t size, int flags)
2529{
2530 struct sock *sk = sock->sk;
2531 int addr_len = 0;
2532 int err;
2533
2534 err = sk->sk_prot->recvmsg(iocb, sk, msg, size, flags & MSG_DONTWAIT,
2535 flags & ~MSG_DONTWAIT, &addr_len);
2536 if (err >= 0)
2537 msg->msg_namelen = addr_len;
2538 return err;
2539}
Linus Torvalds1da177e2005-04-16 15:20:36 -07002540EXPORT_SYMBOL(sock_common_recvmsg);
2541
2542/*
2543 * Set socket options on an inet socket.
2544 */
2545int sock_common_setsockopt(struct socket *sock, int level, int optname,
David S. Millerb7058842009-09-30 16:12:20 -07002546 char __user *optval, unsigned int optlen)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002547{
2548 struct sock *sk = sock->sk;
2549
2550 return sk->sk_prot->setsockopt(sk, level, optname, optval, optlen);
2551}
Linus Torvalds1da177e2005-04-16 15:20:36 -07002552EXPORT_SYMBOL(sock_common_setsockopt);
2553
Dmitry Mishin3fdadf72006-03-20 22:45:21 -08002554#ifdef CONFIG_COMPAT
Arnaldo Carvalho de Melo543d9cf2006-03-20 22:48:35 -08002555int compat_sock_common_setsockopt(struct socket *sock, int level, int optname,
David S. Millerb7058842009-09-30 16:12:20 -07002556 char __user *optval, unsigned int optlen)
Dmitry Mishin3fdadf72006-03-20 22:45:21 -08002557{
2558 struct sock *sk = sock->sk;
2559
Arnaldo Carvalho de Melo543d9cf2006-03-20 22:48:35 -08002560 if (sk->sk_prot->compat_setsockopt != NULL)
2561 return sk->sk_prot->compat_setsockopt(sk, level, optname,
2562 optval, optlen);
Dmitry Mishin3fdadf72006-03-20 22:45:21 -08002563 return sk->sk_prot->setsockopt(sk, level, optname, optval, optlen);
2564}
2565EXPORT_SYMBOL(compat_sock_common_setsockopt);
2566#endif
2567
Linus Torvalds1da177e2005-04-16 15:20:36 -07002568void sk_common_release(struct sock *sk)
2569{
2570 if (sk->sk_prot->destroy)
2571 sk->sk_prot->destroy(sk);
2572
2573 /*
2574 * Observation: when sock_common_release is called, processes have
2575 * no access to socket. But net still has.
2576 * Step one, detach it from networking:
2577 *
2578 * A. Remove from hash tables.
2579 */
2580
2581 sk->sk_prot->unhash(sk);
2582
2583 /*
2584 * In this point socket cannot receive new packets, but it is possible
2585 * that some packets are in flight because some CPU runs receiver and
2586 * did hash table lookup before we unhashed socket. They will achieve
2587 * receive queue and will be purged by socket destructor.
2588 *
2589 * Also we still have packets pending on receive queue and probably,
2590 * our own packets waiting in device queues. sock_destroy will drain
2591 * receive queue, but transmitted packets will delay socket destruction
2592 * until the last reference will be released.
2593 */
2594
2595 sock_orphan(sk);
2596
2597 xfrm_sk_free_policy(sk);
2598
Arnaldo Carvalho de Meloe6848972005-08-09 19:45:38 -07002599 sk_refcnt_debug_release(sk);
Eric Dumazet5640f762012-09-23 23:04:42 +00002600
2601 if (sk->sk_frag.page) {
2602 put_page(sk->sk_frag.page);
2603 sk->sk_frag.page = NULL;
2604 }
2605
Linus Torvalds1da177e2005-04-16 15:20:36 -07002606 sock_put(sk);
2607}
Linus Torvalds1da177e2005-04-16 15:20:36 -07002608EXPORT_SYMBOL(sk_common_release);
2609
Pavel Emelyanov13ff3d62008-03-28 16:38:17 -07002610#ifdef CONFIG_PROC_FS
2611#define PROTO_INUSE_NR 64 /* should be enough for the first time */
Pavel Emelyanov1338d462008-03-28 16:38:43 -07002612struct prot_inuse {
2613 int val[PROTO_INUSE_NR];
2614};
Pavel Emelyanov13ff3d62008-03-28 16:38:17 -07002615
2616static DECLARE_BITMAP(proto_inuse_idx, PROTO_INUSE_NR);
Pavel Emelyanov70ee1152008-03-31 19:42:16 -07002617
2618#ifdef CONFIG_NET_NS
2619void sock_prot_inuse_add(struct net *net, struct proto *prot, int val)
2620{
Eric Dumazetd6d9ca02010-07-19 10:48:49 +00002621 __this_cpu_add(net->core.inuse->val[prot->inuse_idx], val);
Pavel Emelyanov70ee1152008-03-31 19:42:16 -07002622}
2623EXPORT_SYMBOL_GPL(sock_prot_inuse_add);
2624
2625int sock_prot_inuse_get(struct net *net, struct proto *prot)
2626{
2627 int cpu, idx = prot->inuse_idx;
2628 int res = 0;
2629
2630 for_each_possible_cpu(cpu)
2631 res += per_cpu_ptr(net->core.inuse, cpu)->val[idx];
2632
2633 return res >= 0 ? res : 0;
2634}
2635EXPORT_SYMBOL_GPL(sock_prot_inuse_get);
2636
Alexey Dobriyan2c8c1e72010-01-17 03:35:32 +00002637static int __net_init sock_inuse_init_net(struct net *net)
Pavel Emelyanov70ee1152008-03-31 19:42:16 -07002638{
2639 net->core.inuse = alloc_percpu(struct prot_inuse);
2640 return net->core.inuse ? 0 : -ENOMEM;
2641}
2642
Alexey Dobriyan2c8c1e72010-01-17 03:35:32 +00002643static void __net_exit sock_inuse_exit_net(struct net *net)
Pavel Emelyanov70ee1152008-03-31 19:42:16 -07002644{
2645 free_percpu(net->core.inuse);
2646}
2647
2648static struct pernet_operations net_inuse_ops = {
2649 .init = sock_inuse_init_net,
2650 .exit = sock_inuse_exit_net,
2651};
2652
2653static __init int net_inuse_init(void)
2654{
2655 if (register_pernet_subsys(&net_inuse_ops))
2656 panic("Cannot initialize net inuse counters");
2657
2658 return 0;
2659}
2660
2661core_initcall(net_inuse_init);
2662#else
Pavel Emelyanov1338d462008-03-28 16:38:43 -07002663static DEFINE_PER_CPU(struct prot_inuse, prot_inuse);
2664
Pavel Emelyanovc29a0bc2008-03-31 19:41:46 -07002665void sock_prot_inuse_add(struct net *net, struct proto *prot, int val)
Pavel Emelyanov1338d462008-03-28 16:38:43 -07002666{
Eric Dumazetd6d9ca02010-07-19 10:48:49 +00002667 __this_cpu_add(prot_inuse.val[prot->inuse_idx], val);
Pavel Emelyanov1338d462008-03-28 16:38:43 -07002668}
2669EXPORT_SYMBOL_GPL(sock_prot_inuse_add);
2670
Pavel Emelyanovc29a0bc2008-03-31 19:41:46 -07002671int sock_prot_inuse_get(struct net *net, struct proto *prot)
Pavel Emelyanov1338d462008-03-28 16:38:43 -07002672{
2673 int cpu, idx = prot->inuse_idx;
2674 int res = 0;
2675
2676 for_each_possible_cpu(cpu)
2677 res += per_cpu(prot_inuse, cpu).val[idx];
2678
2679 return res >= 0 ? res : 0;
2680}
2681EXPORT_SYMBOL_GPL(sock_prot_inuse_get);
Pavel Emelyanov70ee1152008-03-31 19:42:16 -07002682#endif
Pavel Emelyanov13ff3d62008-03-28 16:38:17 -07002683
2684static void assign_proto_idx(struct proto *prot)
2685{
2686 prot->inuse_idx = find_first_zero_bit(proto_inuse_idx, PROTO_INUSE_NR);
2687
2688 if (unlikely(prot->inuse_idx == PROTO_INUSE_NR - 1)) {
Joe Perchese005d192012-05-16 19:58:40 +00002689 pr_err("PROTO_INUSE_NR exhausted\n");
Pavel Emelyanov13ff3d62008-03-28 16:38:17 -07002690 return;
2691 }
2692
2693 set_bit(prot->inuse_idx, proto_inuse_idx);
2694}
2695
2696static void release_proto_idx(struct proto *prot)
2697{
2698 if (prot->inuse_idx != PROTO_INUSE_NR - 1)
2699 clear_bit(prot->inuse_idx, proto_inuse_idx);
2700}
2701#else
2702static inline void assign_proto_idx(struct proto *prot)
2703{
2704}
2705
2706static inline void release_proto_idx(struct proto *prot)
2707{
2708}
2709#endif
2710
Linus Torvalds1da177e2005-04-16 15:20:36 -07002711int proto_register(struct proto *prot, int alloc_slab)
2712{
Linus Torvalds1da177e2005-04-16 15:20:36 -07002713 if (alloc_slab) {
2714 prot->slab = kmem_cache_create(prot->name, prot->obj_size, 0,
Eric Dumazet271b72c2008-10-29 02:11:14 -07002715 SLAB_HWCACHE_ALIGN | prot->slab_flags,
2716 NULL);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002717
2718 if (prot->slab == NULL) {
Joe Perchese005d192012-05-16 19:58:40 +00002719 pr_crit("%s: Can't create sock SLAB cache!\n",
2720 prot->name);
Pavel Emelyanov60e76632008-03-28 16:39:10 -07002721 goto out;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002722 }
Arnaldo Carvalho de Melo2e6599c2005-06-18 22:46:52 -07002723
2724 if (prot->rsk_prot != NULL) {
Alexey Dobriyanfaf23422010-02-17 09:34:12 +00002725 prot->rsk_prot->slab_name = kasprintf(GFP_KERNEL, "request_sock_%s", prot->name);
Catalin Marinas7e56b5d2008-11-21 16:45:22 -08002726 if (prot->rsk_prot->slab_name == NULL)
Arnaldo Carvalho de Melo2e6599c2005-06-18 22:46:52 -07002727 goto out_free_sock_slab;
2728
Catalin Marinas7e56b5d2008-11-21 16:45:22 -08002729 prot->rsk_prot->slab = kmem_cache_create(prot->rsk_prot->slab_name,
Arnaldo Carvalho de Melo2e6599c2005-06-18 22:46:52 -07002730 prot->rsk_prot->obj_size, 0,
Paul Mundt20c2df82007-07-20 10:11:58 +09002731 SLAB_HWCACHE_ALIGN, NULL);
Arnaldo Carvalho de Melo2e6599c2005-06-18 22:46:52 -07002732
2733 if (prot->rsk_prot->slab == NULL) {
Joe Perchese005d192012-05-16 19:58:40 +00002734 pr_crit("%s: Can't create request sock SLAB cache!\n",
2735 prot->name);
Arnaldo Carvalho de Melo2e6599c2005-06-18 22:46:52 -07002736 goto out_free_request_sock_slab_name;
2737 }
2738 }
Arnaldo Carvalho de Melo8feaf0c02005-08-09 20:09:30 -07002739
Arnaldo Carvalho de Melo6d6ee432005-12-13 23:25:19 -08002740 if (prot->twsk_prot != NULL) {
Alexey Dobriyanfaf23422010-02-17 09:34:12 +00002741 prot->twsk_prot->twsk_slab_name = kasprintf(GFP_KERNEL, "tw_sock_%s", prot->name);
Arnaldo Carvalho de Melo8feaf0c02005-08-09 20:09:30 -07002742
Catalin Marinas7e56b5d2008-11-21 16:45:22 -08002743 if (prot->twsk_prot->twsk_slab_name == NULL)
Arnaldo Carvalho de Melo8feaf0c02005-08-09 20:09:30 -07002744 goto out_free_request_sock_slab;
2745
Arnaldo Carvalho de Melo6d6ee432005-12-13 23:25:19 -08002746 prot->twsk_prot->twsk_slab =
Catalin Marinas7e56b5d2008-11-21 16:45:22 -08002747 kmem_cache_create(prot->twsk_prot->twsk_slab_name,
Arnaldo Carvalho de Melo6d6ee432005-12-13 23:25:19 -08002748 prot->twsk_prot->twsk_obj_size,
Eric Dumazet3ab5aee2008-11-16 19:40:17 -08002749 0,
2750 SLAB_HWCACHE_ALIGN |
2751 prot->slab_flags,
Paul Mundt20c2df82007-07-20 10:11:58 +09002752 NULL);
Arnaldo Carvalho de Melo6d6ee432005-12-13 23:25:19 -08002753 if (prot->twsk_prot->twsk_slab == NULL)
Arnaldo Carvalho de Melo8feaf0c02005-08-09 20:09:30 -07002754 goto out_free_timewait_sock_slab_name;
2755 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07002756 }
2757
Glauber Costa36b77a52011-12-16 00:51:59 +00002758 mutex_lock(&proto_list_mutex);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002759 list_add(&prot->node, &proto_list);
Pavel Emelyanov13ff3d62008-03-28 16:38:17 -07002760 assign_proto_idx(prot);
Glauber Costa36b77a52011-12-16 00:51:59 +00002761 mutex_unlock(&proto_list_mutex);
Pavel Emelyanovb733c002007-11-07 02:23:38 -08002762 return 0;
2763
Arnaldo Carvalho de Melo8feaf0c02005-08-09 20:09:30 -07002764out_free_timewait_sock_slab_name:
Catalin Marinas7e56b5d2008-11-21 16:45:22 -08002765 kfree(prot->twsk_prot->twsk_slab_name);
Arnaldo Carvalho de Melo8feaf0c02005-08-09 20:09:30 -07002766out_free_request_sock_slab:
2767 if (prot->rsk_prot && prot->rsk_prot->slab) {
2768 kmem_cache_destroy(prot->rsk_prot->slab);
2769 prot->rsk_prot->slab = NULL;
2770 }
Arnaldo Carvalho de Melo2e6599c2005-06-18 22:46:52 -07002771out_free_request_sock_slab_name:
Dan Carpenter72150e92010-03-06 01:04:45 +00002772 if (prot->rsk_prot)
2773 kfree(prot->rsk_prot->slab_name);
Arnaldo Carvalho de Melo2e6599c2005-06-18 22:46:52 -07002774out_free_sock_slab:
2775 kmem_cache_destroy(prot->slab);
2776 prot->slab = NULL;
Pavel Emelyanovb733c002007-11-07 02:23:38 -08002777out:
2778 return -ENOBUFS;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002779}
Linus Torvalds1da177e2005-04-16 15:20:36 -07002780EXPORT_SYMBOL(proto_register);
2781
2782void proto_unregister(struct proto *prot)
2783{
Glauber Costa36b77a52011-12-16 00:51:59 +00002784 mutex_lock(&proto_list_mutex);
Pavel Emelyanov13ff3d62008-03-28 16:38:17 -07002785 release_proto_idx(prot);
Patrick McHardy0a3f4352005-09-06 19:47:50 -07002786 list_del(&prot->node);
Glauber Costa36b77a52011-12-16 00:51:59 +00002787 mutex_unlock(&proto_list_mutex);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002788
2789 if (prot->slab != NULL) {
2790 kmem_cache_destroy(prot->slab);
2791 prot->slab = NULL;
2792 }
2793
Arnaldo Carvalho de Melo2e6599c2005-06-18 22:46:52 -07002794 if (prot->rsk_prot != NULL && prot->rsk_prot->slab != NULL) {
Arnaldo Carvalho de Melo2e6599c2005-06-18 22:46:52 -07002795 kmem_cache_destroy(prot->rsk_prot->slab);
Catalin Marinas7e56b5d2008-11-21 16:45:22 -08002796 kfree(prot->rsk_prot->slab_name);
Arnaldo Carvalho de Melo2e6599c2005-06-18 22:46:52 -07002797 prot->rsk_prot->slab = NULL;
2798 }
2799
Arnaldo Carvalho de Melo6d6ee432005-12-13 23:25:19 -08002800 if (prot->twsk_prot != NULL && prot->twsk_prot->twsk_slab != NULL) {
Arnaldo Carvalho de Melo6d6ee432005-12-13 23:25:19 -08002801 kmem_cache_destroy(prot->twsk_prot->twsk_slab);
Catalin Marinas7e56b5d2008-11-21 16:45:22 -08002802 kfree(prot->twsk_prot->twsk_slab_name);
Arnaldo Carvalho de Melo6d6ee432005-12-13 23:25:19 -08002803 prot->twsk_prot->twsk_slab = NULL;
Arnaldo Carvalho de Melo8feaf0c02005-08-09 20:09:30 -07002804 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07002805}
Linus Torvalds1da177e2005-04-16 15:20:36 -07002806EXPORT_SYMBOL(proto_unregister);
2807
2808#ifdef CONFIG_PROC_FS
Linus Torvalds1da177e2005-04-16 15:20:36 -07002809static void *proto_seq_start(struct seq_file *seq, loff_t *pos)
Glauber Costa36b77a52011-12-16 00:51:59 +00002810 __acquires(proto_list_mutex)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002811{
Glauber Costa36b77a52011-12-16 00:51:59 +00002812 mutex_lock(&proto_list_mutex);
Pavel Emelianov60f04382007-07-09 13:15:14 -07002813 return seq_list_start_head(&proto_list, *pos);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002814}
2815
2816static void *proto_seq_next(struct seq_file *seq, void *v, loff_t *pos)
2817{
Pavel Emelianov60f04382007-07-09 13:15:14 -07002818 return seq_list_next(v, &proto_list, pos);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002819}
2820
2821static void proto_seq_stop(struct seq_file *seq, void *v)
Glauber Costa36b77a52011-12-16 00:51:59 +00002822 __releases(proto_list_mutex)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002823{
Glauber Costa36b77a52011-12-16 00:51:59 +00002824 mutex_unlock(&proto_list_mutex);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002825}
2826
2827static char proto_method_implemented(const void *method)
2828{
2829 return method == NULL ? 'n' : 'y';
2830}
Glauber Costa180d8cd2011-12-11 21:47:02 +00002831static long sock_prot_memory_allocated(struct proto *proto)
2832{
Jeffrin Josecb75a362012-04-25 19:17:29 +05302833 return proto->memory_allocated != NULL ? proto_memory_allocated(proto) : -1L;
Glauber Costa180d8cd2011-12-11 21:47:02 +00002834}
2835
2836static char *sock_prot_memory_pressure(struct proto *proto)
2837{
2838 return proto->memory_pressure != NULL ?
2839 proto_memory_pressure(proto) ? "yes" : "no" : "NI";
2840}
Linus Torvalds1da177e2005-04-16 15:20:36 -07002841
2842static void proto_seq_printf(struct seq_file *seq, struct proto *proto)
2843{
Glauber Costa180d8cd2011-12-11 21:47:02 +00002844
Eric Dumazet8d987e52010-11-09 23:24:26 +00002845 seq_printf(seq, "%-9s %4u %6d %6ld %-3s %6u %-3s %-10s "
Linus Torvalds1da177e2005-04-16 15:20:36 -07002846 "%2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c\n",
2847 proto->name,
2848 proto->obj_size,
Eric Dumazet14e943d2008-11-19 15:14:01 -08002849 sock_prot_inuse_get(seq_file_net(seq), proto),
Glauber Costa180d8cd2011-12-11 21:47:02 +00002850 sock_prot_memory_allocated(proto),
2851 sock_prot_memory_pressure(proto),
Linus Torvalds1da177e2005-04-16 15:20:36 -07002852 proto->max_header,
2853 proto->slab == NULL ? "no" : "yes",
2854 module_name(proto->owner),
2855 proto_method_implemented(proto->close),
2856 proto_method_implemented(proto->connect),
2857 proto_method_implemented(proto->disconnect),
2858 proto_method_implemented(proto->accept),
2859 proto_method_implemented(proto->ioctl),
2860 proto_method_implemented(proto->init),
2861 proto_method_implemented(proto->destroy),
2862 proto_method_implemented(proto->shutdown),
2863 proto_method_implemented(proto->setsockopt),
2864 proto_method_implemented(proto->getsockopt),
2865 proto_method_implemented(proto->sendmsg),
2866 proto_method_implemented(proto->recvmsg),
2867 proto_method_implemented(proto->sendpage),
2868 proto_method_implemented(proto->bind),
2869 proto_method_implemented(proto->backlog_rcv),
2870 proto_method_implemented(proto->hash),
2871 proto_method_implemented(proto->unhash),
2872 proto_method_implemented(proto->get_port),
2873 proto_method_implemented(proto->enter_memory_pressure));
2874}
2875
2876static int proto_seq_show(struct seq_file *seq, void *v)
2877{
Pavel Emelianov60f04382007-07-09 13:15:14 -07002878 if (v == &proto_list)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002879 seq_printf(seq, "%-9s %-4s %-8s %-6s %-5s %-7s %-4s %-10s %s",
2880 "protocol",
2881 "size",
2882 "sockets",
2883 "memory",
2884 "press",
2885 "maxhdr",
2886 "slab",
2887 "module",
2888 "cl co di ac io in de sh ss gs se re sp bi br ha uh gp em\n");
2889 else
Pavel Emelianov60f04382007-07-09 13:15:14 -07002890 proto_seq_printf(seq, list_entry(v, struct proto, node));
Linus Torvalds1da177e2005-04-16 15:20:36 -07002891 return 0;
2892}
2893
Stephen Hemmingerf6908082007-03-12 14:34:29 -07002894static const struct seq_operations proto_seq_ops = {
Linus Torvalds1da177e2005-04-16 15:20:36 -07002895 .start = proto_seq_start,
2896 .next = proto_seq_next,
2897 .stop = proto_seq_stop,
2898 .show = proto_seq_show,
2899};
2900
2901static int proto_seq_open(struct inode *inode, struct file *file)
2902{
Eric Dumazet14e943d2008-11-19 15:14:01 -08002903 return seq_open_net(inode, file, &proto_seq_ops,
2904 sizeof(struct seq_net_private));
Linus Torvalds1da177e2005-04-16 15:20:36 -07002905}
2906
Arjan van de Ven9a321442007-02-12 00:55:35 -08002907static const struct file_operations proto_seq_fops = {
Linus Torvalds1da177e2005-04-16 15:20:36 -07002908 .owner = THIS_MODULE,
2909 .open = proto_seq_open,
2910 .read = seq_read,
2911 .llseek = seq_lseek,
Eric Dumazet14e943d2008-11-19 15:14:01 -08002912 .release = seq_release_net,
2913};
2914
2915static __net_init int proto_init_net(struct net *net)
2916{
Gao fengd4beaa62013-02-18 01:34:54 +00002917 if (!proc_create("protocols", S_IRUGO, net->proc_net, &proto_seq_fops))
Eric Dumazet14e943d2008-11-19 15:14:01 -08002918 return -ENOMEM;
2919
2920 return 0;
2921}
2922
2923static __net_exit void proto_exit_net(struct net *net)
2924{
Gao fengece31ff2013-02-18 01:34:56 +00002925 remove_proc_entry("protocols", net->proc_net);
Eric Dumazet14e943d2008-11-19 15:14:01 -08002926}
2927
2928
2929static __net_initdata struct pernet_operations proto_net_ops = {
2930 .init = proto_init_net,
2931 .exit = proto_exit_net,
Linus Torvalds1da177e2005-04-16 15:20:36 -07002932};
2933
2934static int __init proto_init(void)
2935{
Eric Dumazet14e943d2008-11-19 15:14:01 -08002936 return register_pernet_subsys(&proto_net_ops);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002937}
2938
2939subsys_initcall(proto_init);
2940
2941#endif /* PROC_FS */