blob: e16a5db853c6f455b0ac826744d0ee5e96a44863 [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>
Craig Gallekeb4cb002015-06-15 11:26:18 -0400134#include <linux/sock_diag.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -0700135
136#include <linux/filter.h>
Craig Gallek538950a2016-01-04 17:41:47 -0500137#include <net/sock_reuseport.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -0700138
Satoru Moriya3847ce32011-06-17 12:00:03 +0000139#include <trace/events/sock.h>
140
Linus Torvalds1da177e2005-04-16 15:20:36 -0700141#ifdef CONFIG_INET
142#include <net/tcp.h>
143#endif
144
Eliezer Tamir076bb0c2013-07-10 17:13:17 +0300145#include <net/busy_poll.h>
Eliezer Tamir06021292013-06-10 11:39:50 +0300146
Glauber Costa36b77a52011-12-16 00:51:59 +0000147static DEFINE_MUTEX(proto_list_mutex);
Glauber Costad1a4c0b2011-12-11 21:47:04 +0000148static LIST_HEAD(proto_list);
149
Eric W. Biedermana3b299d2014-04-23 14:26:56 -0700150/**
151 * sk_ns_capable - General socket capability test
152 * @sk: Socket to use a capability on or through
153 * @user_ns: The user namespace of the capability to use
154 * @cap: The capability to use
155 *
156 * Test to see if the opener of the socket had when the socket was
157 * created and the current process has the capability @cap in the user
158 * namespace @user_ns.
159 */
160bool sk_ns_capable(const struct sock *sk,
161 struct user_namespace *user_ns, int cap)
162{
163 return file_ns_capable(sk->sk_socket->file, user_ns, cap) &&
164 ns_capable(user_ns, cap);
165}
166EXPORT_SYMBOL(sk_ns_capable);
167
168/**
169 * sk_capable - Socket global capability test
170 * @sk: Socket to use a capability on or through
Masanari Iidae793c0f2014-09-04 23:44:36 +0900171 * @cap: The global capability to use
Eric W. Biedermana3b299d2014-04-23 14:26:56 -0700172 *
173 * Test to see if the opener of the socket had when the socket was
174 * created and the current process has the capability @cap in all user
175 * namespaces.
176 */
177bool sk_capable(const struct sock *sk, int cap)
178{
179 return sk_ns_capable(sk, &init_user_ns, cap);
180}
181EXPORT_SYMBOL(sk_capable);
182
183/**
184 * sk_net_capable - Network namespace socket capability test
185 * @sk: Socket to use a capability on or through
186 * @cap: The capability to use
187 *
Masanari Iidae793c0f2014-09-04 23:44:36 +0900188 * Test to see if the opener of the socket had when the socket was created
Eric W. Biedermana3b299d2014-04-23 14:26:56 -0700189 * and the current process has the capability @cap over the network namespace
190 * the socket is a member of.
191 */
192bool sk_net_capable(const struct sock *sk, int cap)
193{
194 return sk_ns_capable(sk, sock_net(sk)->user_ns, cap);
195}
196EXPORT_SYMBOL(sk_net_capable);
197
Ingo Molnarda21f242006-07-03 00:25:12 -0700198/*
199 * Each address family might have different locking rules, so we have
200 * one slock key per address family:
201 */
Ingo Molnara5b5bb92006-07-03 00:25:35 -0700202static struct lock_class_key af_family_keys[AF_MAX];
203static struct lock_class_key af_family_slock_keys[AF_MAX];
204
Ingo Molnara5b5bb92006-07-03 00:25:35 -0700205/*
206 * Make lock validator output more readable. (we pre-construct these
207 * strings build-time, so that runtime initialization of socket
208 * locks is fast):
209 */
Jan Engelhardt36cbd3d2009-08-05 10:42:58 -0700210static const char *const af_family_key_strings[AF_MAX+1] = {
Ingo Molnara5b5bb92006-07-03 00:25:35 -0700211 "sk_lock-AF_UNSPEC", "sk_lock-AF_UNIX" , "sk_lock-AF_INET" ,
212 "sk_lock-AF_AX25" , "sk_lock-AF_IPX" , "sk_lock-AF_APPLETALK",
213 "sk_lock-AF_NETROM", "sk_lock-AF_BRIDGE" , "sk_lock-AF_ATMPVC" ,
214 "sk_lock-AF_X25" , "sk_lock-AF_INET6" , "sk_lock-AF_ROSE" ,
215 "sk_lock-AF_DECnet", "sk_lock-AF_NETBEUI" , "sk_lock-AF_SECURITY" ,
216 "sk_lock-AF_KEY" , "sk_lock-AF_NETLINK" , "sk_lock-AF_PACKET" ,
217 "sk_lock-AF_ASH" , "sk_lock-AF_ECONET" , "sk_lock-AF_ATMSVC" ,
Andy Grovercbd151b2009-02-26 23:43:19 -0800218 "sk_lock-AF_RDS" , "sk_lock-AF_SNA" , "sk_lock-AF_IRDA" ,
Ingo Molnara5b5bb92006-07-03 00:25:35 -0700219 "sk_lock-AF_PPPOX" , "sk_lock-AF_WANPIPE" , "sk_lock-AF_LLC" ,
Oliver Hartkoppcd05acf2007-12-16 15:59:24 -0800220 "sk_lock-27" , "sk_lock-28" , "sk_lock-AF_CAN" ,
David Howells17926a72007-04-26 15:48:28 -0700221 "sk_lock-AF_TIPC" , "sk_lock-AF_BLUETOOTH", "sk_lock-IUCV" ,
Remi Denis-Courmontbce7b152008-09-22 19:51:15 -0700222 "sk_lock-AF_RXRPC" , "sk_lock-AF_ISDN" , "sk_lock-AF_PHONET" ,
Miloslav Trmač6f107b52010-12-08 14:35:34 +0800223 "sk_lock-AF_IEEE802154", "sk_lock-AF_CAIF" , "sk_lock-AF_ALG" ,
Dexuan Cui0a1a37b2016-04-05 07:41:11 -0700224 "sk_lock-AF_NFC" , "sk_lock-AF_VSOCK" , "sk_lock-AF_KCM" ,
225 "sk_lock-AF_MAX"
Ingo Molnara5b5bb92006-07-03 00:25:35 -0700226};
Jan Engelhardt36cbd3d2009-08-05 10:42:58 -0700227static const char *const af_family_slock_key_strings[AF_MAX+1] = {
Ingo Molnara5b5bb92006-07-03 00:25:35 -0700228 "slock-AF_UNSPEC", "slock-AF_UNIX" , "slock-AF_INET" ,
229 "slock-AF_AX25" , "slock-AF_IPX" , "slock-AF_APPLETALK",
230 "slock-AF_NETROM", "slock-AF_BRIDGE" , "slock-AF_ATMPVC" ,
231 "slock-AF_X25" , "slock-AF_INET6" , "slock-AF_ROSE" ,
232 "slock-AF_DECnet", "slock-AF_NETBEUI" , "slock-AF_SECURITY" ,
233 "slock-AF_KEY" , "slock-AF_NETLINK" , "slock-AF_PACKET" ,
234 "slock-AF_ASH" , "slock-AF_ECONET" , "slock-AF_ATMSVC" ,
Andy Grovercbd151b2009-02-26 23:43:19 -0800235 "slock-AF_RDS" , "slock-AF_SNA" , "slock-AF_IRDA" ,
Ingo Molnara5b5bb92006-07-03 00:25:35 -0700236 "slock-AF_PPPOX" , "slock-AF_WANPIPE" , "slock-AF_LLC" ,
Oliver Hartkoppcd05acf2007-12-16 15:59:24 -0800237 "slock-27" , "slock-28" , "slock-AF_CAN" ,
David Howells17926a72007-04-26 15:48:28 -0700238 "slock-AF_TIPC" , "slock-AF_BLUETOOTH", "slock-AF_IUCV" ,
Remi Denis-Courmontbce7b152008-09-22 19:51:15 -0700239 "slock-AF_RXRPC" , "slock-AF_ISDN" , "slock-AF_PHONET" ,
Miloslav Trmač6f107b52010-12-08 14:35:34 +0800240 "slock-AF_IEEE802154", "slock-AF_CAIF" , "slock-AF_ALG" ,
Dexuan Cui0a1a37b2016-04-05 07:41:11 -0700241 "slock-AF_NFC" , "slock-AF_VSOCK" ,"slock-AF_KCM" ,
242 "slock-AF_MAX"
Ingo Molnara5b5bb92006-07-03 00:25:35 -0700243};
Jan Engelhardt36cbd3d2009-08-05 10:42:58 -0700244static const char *const af_family_clock_key_strings[AF_MAX+1] = {
Peter Zijlstra443aef0e2007-07-19 01:49:00 -0700245 "clock-AF_UNSPEC", "clock-AF_UNIX" , "clock-AF_INET" ,
246 "clock-AF_AX25" , "clock-AF_IPX" , "clock-AF_APPLETALK",
247 "clock-AF_NETROM", "clock-AF_BRIDGE" , "clock-AF_ATMPVC" ,
248 "clock-AF_X25" , "clock-AF_INET6" , "clock-AF_ROSE" ,
249 "clock-AF_DECnet", "clock-AF_NETBEUI" , "clock-AF_SECURITY" ,
250 "clock-AF_KEY" , "clock-AF_NETLINK" , "clock-AF_PACKET" ,
251 "clock-AF_ASH" , "clock-AF_ECONET" , "clock-AF_ATMSVC" ,
Andy Grovercbd151b2009-02-26 23:43:19 -0800252 "clock-AF_RDS" , "clock-AF_SNA" , "clock-AF_IRDA" ,
Peter Zijlstra443aef0e2007-07-19 01:49:00 -0700253 "clock-AF_PPPOX" , "clock-AF_WANPIPE" , "clock-AF_LLC" ,
Oliver Hartkoppb4942af2008-07-23 14:06:04 -0700254 "clock-27" , "clock-28" , "clock-AF_CAN" ,
David Howellse51f8022007-07-21 19:30:16 -0700255 "clock-AF_TIPC" , "clock-AF_BLUETOOTH", "clock-AF_IUCV" ,
Remi Denis-Courmontbce7b152008-09-22 19:51:15 -0700256 "clock-AF_RXRPC" , "clock-AF_ISDN" , "clock-AF_PHONET" ,
Miloslav Trmač6f107b52010-12-08 14:35:34 +0800257 "clock-AF_IEEE802154", "clock-AF_CAIF" , "clock-AF_ALG" ,
Dexuan Cui0a1a37b2016-04-05 07:41:11 -0700258 "clock-AF_NFC" , "clock-AF_VSOCK" , "clock-AF_KCM" ,
259 "clock-AF_MAX"
Peter Zijlstra443aef0e2007-07-19 01:49:00 -0700260};
Ingo Molnarda21f242006-07-03 00:25:12 -0700261
262/*
263 * sk_callback_lock locking rules are per-address-family,
264 * so split the lock classes by using a per-AF key:
265 */
266static struct lock_class_key af_callback_keys[AF_MAX];
267
Linus Torvalds1da177e2005-04-16 15:20:36 -0700268/* Take into consideration the size of the struct sk_buff overhead in the
269 * determination of these values, since that is non-constant across
270 * platforms. This makes socket queueing behavior and performance
271 * not depend upon such differences.
272 */
273#define _SK_MEM_PACKETS 256
Eric Dumazet87fb4b72011-10-13 07:28:54 +0000274#define _SK_MEM_OVERHEAD SKB_TRUESIZE(256)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700275#define SK_WMEM_MAX (_SK_MEM_OVERHEAD * _SK_MEM_PACKETS)
276#define SK_RMEM_MAX (_SK_MEM_OVERHEAD * _SK_MEM_PACKETS)
277
278/* Run time adjustable parameters. */
Brian Haleyab32ea52006-09-22 14:15:41 -0700279__u32 sysctl_wmem_max __read_mostly = SK_WMEM_MAX;
Hans Schillstrom6d8ebc82012-04-30 08:13:50 +0200280EXPORT_SYMBOL(sysctl_wmem_max);
Brian Haleyab32ea52006-09-22 14:15:41 -0700281__u32 sysctl_rmem_max __read_mostly = SK_RMEM_MAX;
Hans Schillstrom6d8ebc82012-04-30 08:13:50 +0200282EXPORT_SYMBOL(sysctl_rmem_max);
Brian Haleyab32ea52006-09-22 14:15:41 -0700283__u32 sysctl_wmem_default __read_mostly = SK_WMEM_MAX;
284__u32 sysctl_rmem_default __read_mostly = SK_RMEM_MAX;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700285
Lucas De Marchi25985ed2011-03-30 22:57:33 -0300286/* Maximal space eaten by iovec or ancillary data plus some space */
Brian Haleyab32ea52006-09-22 14:15:41 -0700287int sysctl_optmem_max __read_mostly = sizeof(unsigned long)*(2*UIO_MAXIOV+512);
Eric Dumazet2a915252009-05-27 11:30:05 +0000288EXPORT_SYMBOL(sysctl_optmem_max);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700289
Willem de Bruijnb245be12015-01-30 13:29:32 -0500290int sysctl_tstamp_allow_data __read_mostly = 1;
291
Mel Gormanc93bdd02012-07-31 16:44:19 -0700292struct static_key memalloc_socks = STATIC_KEY_INIT_FALSE;
293EXPORT_SYMBOL_GPL(memalloc_socks);
294
Mel Gorman7cb02402012-07-31 16:44:16 -0700295/**
296 * sk_set_memalloc - sets %SOCK_MEMALLOC
297 * @sk: socket to set it on
298 *
299 * Set %SOCK_MEMALLOC on a socket for access to emergency reserves.
300 * It's the responsibility of the admin to adjust min_free_kbytes
301 * to meet the requirements
302 */
303void sk_set_memalloc(struct sock *sk)
304{
305 sock_set_flag(sk, SOCK_MEMALLOC);
306 sk->sk_allocation |= __GFP_MEMALLOC;
Mel Gormanc93bdd02012-07-31 16:44:19 -0700307 static_key_slow_inc(&memalloc_socks);
Mel Gorman7cb02402012-07-31 16:44:16 -0700308}
309EXPORT_SYMBOL_GPL(sk_set_memalloc);
310
311void sk_clear_memalloc(struct sock *sk)
312{
313 sock_reset_flag(sk, SOCK_MEMALLOC);
314 sk->sk_allocation &= ~__GFP_MEMALLOC;
Mel Gormanc93bdd02012-07-31 16:44:19 -0700315 static_key_slow_dec(&memalloc_socks);
Mel Gormanc76562b2012-07-31 16:44:41 -0700316
317 /*
318 * SOCK_MEMALLOC is allowed to ignore rmem limits to ensure forward
Mel Gorman5d753612015-06-10 21:02:04 -0400319 * progress of swapping. SOCK_MEMALLOC may be cleared while
320 * it has rmem allocations due to the last swapfile being deactivated
321 * but there is a risk that the socket is unusable due to exceeding
322 * the rmem limits. Reclaim the reserves and obey rmem limits again.
Mel Gormanc76562b2012-07-31 16:44:41 -0700323 */
Mel Gorman5d753612015-06-10 21:02:04 -0400324 sk_mem_reclaim(sk);
Mel Gorman7cb02402012-07-31 16:44:16 -0700325}
326EXPORT_SYMBOL_GPL(sk_clear_memalloc);
327
Mel Gormanb4b9e352012-07-31 16:44:26 -0700328int __sk_backlog_rcv(struct sock *sk, struct sk_buff *skb)
329{
330 int ret;
331 unsigned long pflags = current->flags;
332
333 /* these should have been dropped before queueing */
334 BUG_ON(!sock_flag(sk, SOCK_MEMALLOC));
335
336 current->flags |= PF_MEMALLOC;
337 ret = sk->sk_backlog_rcv(sk, skb);
338 tsk_restore_flags(current, pflags, PF_MEMALLOC);
339
340 return ret;
341}
342EXPORT_SYMBOL(__sk_backlog_rcv);
343
Linus Torvalds1da177e2005-04-16 15:20:36 -0700344static int sock_set_timeout(long *timeo_p, char __user *optval, int optlen)
345{
346 struct timeval tv;
347
348 if (optlen < sizeof(tv))
349 return -EINVAL;
350 if (copy_from_user(&tv, optval, sizeof(tv)))
351 return -EFAULT;
Vasily Averinba780732007-05-24 16:58:54 -0700352 if (tv.tv_usec < 0 || tv.tv_usec >= USEC_PER_SEC)
353 return -EDOM;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700354
Vasily Averinba780732007-05-24 16:58:54 -0700355 if (tv.tv_sec < 0) {
Andrew Morton6f11df82007-07-09 13:16:00 -0700356 static int warned __read_mostly;
357
Vasily Averinba780732007-05-24 16:58:54 -0700358 *timeo_p = 0;
Ilpo Järvinen50aab542008-05-02 16:20:10 -0700359 if (warned < 10 && net_ratelimit()) {
Vasily Averinba780732007-05-24 16:58:54 -0700360 warned++;
Joe Perchese005d192012-05-16 19:58:40 +0000361 pr_info("%s: `%s' (pid %d) tries to set negative timeout\n",
362 __func__, current->comm, task_pid_nr(current));
Ilpo Järvinen50aab542008-05-02 16:20:10 -0700363 }
Vasily Averinba780732007-05-24 16:58:54 -0700364 return 0;
365 }
Linus Torvalds1da177e2005-04-16 15:20:36 -0700366 *timeo_p = MAX_SCHEDULE_TIMEOUT;
367 if (tv.tv_sec == 0 && tv.tv_usec == 0)
368 return 0;
369 if (tv.tv_sec < (MAX_SCHEDULE_TIMEOUT/HZ - 1))
370 *timeo_p = tv.tv_sec*HZ + (tv.tv_usec+(1000000/HZ-1))/(1000000/HZ);
371 return 0;
372}
373
374static void sock_warn_obsolete_bsdism(const char *name)
375{
376 static int warned;
377 static char warncomm[TASK_COMM_LEN];
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +0900378 if (strcmp(warncomm, current->comm) && warned < 5) {
379 strcpy(warncomm, current->comm);
Joe Perchese005d192012-05-16 19:58:40 +0000380 pr_warn("process `%s' is using obsolete %s SO_BSDCOMPAT\n",
381 warncomm, name);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700382 warned++;
383 }
384}
385
Hannes Frederic Sowa080a2702015-10-26 13:51:37 +0100386static bool sock_needs_netstamp(const struct sock *sk)
387{
388 switch (sk->sk_family) {
389 case AF_UNSPEC:
390 case AF_UNIX:
391 return false;
392 default:
393 return true;
394 }
395}
396
Eric Dumazet08e29af2011-11-28 12:04:18 +0000397static void sock_disable_timestamp(struct sock *sk, unsigned long flags)
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +0900398{
Eric Dumazet08e29af2011-11-28 12:04:18 +0000399 if (sk->sk_flags & flags) {
400 sk->sk_flags &= ~flags;
Hannes Frederic Sowa080a2702015-10-26 13:51:37 +0100401 if (sock_needs_netstamp(sk) &&
402 !(sk->sk_flags & SK_FLAGS_TIMESTAMP))
Patrick Ohly20d49472009-02-12 05:03:38 +0000403 net_disable_timestamp();
Linus Torvalds1da177e2005-04-16 15:20:36 -0700404 }
405}
406
407
samanthakumare6afc8a2016-04-05 12:41:15 -0400408int __sock_queue_rcv_skb(struct sock *sk, struct sk_buff *skb)
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800409{
Neil Horman3b885782009-10-12 13:26:31 -0700410 unsigned long flags;
411 struct sk_buff_head *list = &sk->sk_receive_queue;
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800412
Eric Dumazet0fd7bac2011-12-21 07:11:44 +0000413 if (atomic_read(&sk->sk_rmem_alloc) >= sk->sk_rcvbuf) {
Eric Dumazet766e90372009-10-14 20:40:11 -0700414 atomic_inc(&sk->sk_drops);
Satoru Moriya3847ce32011-06-17 12:00:03 +0000415 trace_sock_rcvqueue_full(sk, skb);
Eric Dumazet766e90372009-10-14 20:40:11 -0700416 return -ENOMEM;
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800417 }
418
Mel Gormanc76562b2012-07-31 16:44:41 -0700419 if (!sk_rmem_schedule(sk, skb, skb->truesize)) {
Eric Dumazet766e90372009-10-14 20:40:11 -0700420 atomic_inc(&sk->sk_drops);
421 return -ENOBUFS;
Hideo Aoki3ab224b2007-12-31 00:11:19 -0800422 }
423
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800424 skb->dev = NULL;
425 skb_set_owner_r(skb, sk);
David S. Miller49ad9592008-12-17 22:11:38 -0800426
Eric Dumazet7fee2262010-05-11 23:19:48 +0000427 /* we escape from rcu protected region, make sure we dont leak
428 * a norefcounted dst
429 */
430 skb_dst_force(skb);
431
Neil Horman3b885782009-10-12 13:26:31 -0700432 spin_lock_irqsave(&list->lock, flags);
Eyal Birger3bc3b962015-03-01 14:58:30 +0200433 sock_skb_set_dropcount(sk, skb);
Neil Horman3b885782009-10-12 13:26:31 -0700434 __skb_queue_tail(list, skb);
435 spin_unlock_irqrestore(&list->lock, flags);
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800436
437 if (!sock_flag(sk, SOCK_DEAD))
David S. Miller676d2362014-04-11 16:15:36 -0400438 sk->sk_data_ready(sk);
Eric Dumazet766e90372009-10-14 20:40:11 -0700439 return 0;
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800440}
samanthakumare6afc8a2016-04-05 12:41:15 -0400441EXPORT_SYMBOL(__sock_queue_rcv_skb);
442
443int sock_queue_rcv_skb(struct sock *sk, struct sk_buff *skb)
444{
445 int err;
446
447 err = sk_filter(sk, skb);
448 if (err)
449 return err;
450
451 return __sock_queue_rcv_skb(sk, skb);
452}
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800453EXPORT_SYMBOL(sock_queue_rcv_skb);
454
Arnaldo Carvalho de Melo58a5a7b2006-11-16 14:06:06 -0200455int sk_receive_skb(struct sock *sk, struct sk_buff *skb, const int nested)
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800456{
457 int rc = NET_RX_SUCCESS;
458
Dmitry Mishinfda9ef52006-08-31 15:28:39 -0700459 if (sk_filter(sk, skb))
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800460 goto discard_and_relse;
461
462 skb->dev = NULL;
463
Sorin Dumitru274f4822014-07-22 21:16:51 +0300464 if (sk_rcvqueues_full(sk, sk->sk_rcvbuf)) {
Eric Dumazetc3774112010-04-27 15:13:20 -0700465 atomic_inc(&sk->sk_drops);
466 goto discard_and_relse;
467 }
Arnaldo Carvalho de Melo58a5a7b2006-11-16 14:06:06 -0200468 if (nested)
469 bh_lock_sock_nested(sk);
470 else
471 bh_lock_sock(sk);
Ingo Molnara5b5bb92006-07-03 00:25:35 -0700472 if (!sock_owned_by_user(sk)) {
473 /*
474 * trylock + unlock semantics:
475 */
476 mutex_acquire(&sk->sk_lock.dep_map, 0, 1, _RET_IP_);
477
Peter Zijlstrac57943a2008-10-07 14:18:42 -0700478 rc = sk_backlog_rcv(sk, skb);
Ingo Molnara5b5bb92006-07-03 00:25:35 -0700479
480 mutex_release(&sk->sk_lock.dep_map, 1, _RET_IP_);
Eric Dumazetf545a382012-04-22 23:34:26 +0000481 } else if (sk_add_backlog(sk, skb, sk->sk_rcvbuf)) {
Zhu Yi8eae9392010-03-04 18:01:40 +0000482 bh_unlock_sock(sk);
483 atomic_inc(&sk->sk_drops);
484 goto discard_and_relse;
485 }
486
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800487 bh_unlock_sock(sk);
488out:
489 sock_put(sk);
490 return rc;
491discard_and_relse:
492 kfree_skb(skb);
493 goto out;
494}
495EXPORT_SYMBOL(sk_receive_skb);
496
497struct dst_entry *__sk_dst_check(struct sock *sk, u32 cookie)
498{
Eric Dumazetb6c67122010-04-08 23:03:29 +0000499 struct dst_entry *dst = __sk_dst_get(sk);
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800500
501 if (dst && dst->obsolete && dst->ops->check(dst, cookie) == NULL) {
Krishna Kumare022f0b2009-10-19 23:46:20 +0000502 sk_tx_queue_clear(sk);
Stephen Hemmingera9b3cd72011-08-01 16:19:00 +0000503 RCU_INIT_POINTER(sk->sk_dst_cache, NULL);
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800504 dst_release(dst);
505 return NULL;
506 }
507
508 return dst;
509}
510EXPORT_SYMBOL(__sk_dst_check);
511
512struct dst_entry *sk_dst_check(struct sock *sk, u32 cookie)
513{
514 struct dst_entry *dst = sk_dst_get(sk);
515
516 if (dst && dst->obsolete && dst->ops->check(dst, cookie) == NULL) {
517 sk_dst_reset(sk);
518 dst_release(dst);
519 return NULL;
520 }
521
522 return dst;
523}
524EXPORT_SYMBOL(sk_dst_check);
525
Brian Haleyc91f6df2012-11-26 05:21:08 +0000526static int sock_setbindtodevice(struct sock *sk, char __user *optval,
527 int optlen)
David S. Miller48788092007-09-14 16:41:03 -0700528{
529 int ret = -ENOPROTOOPT;
530#ifdef CONFIG_NETDEVICES
YOSHIFUJI Hideaki3b1e0a62008-03-26 02:26:21 +0900531 struct net *net = sock_net(sk);
David S. Miller48788092007-09-14 16:41:03 -0700532 char devname[IFNAMSIZ];
533 int index;
534
535 /* Sorry... */
536 ret = -EPERM;
Eric W. Biederman5e1fccc2012-11-16 03:03:04 +0000537 if (!ns_capable(net->user_ns, CAP_NET_RAW))
David S. Miller48788092007-09-14 16:41:03 -0700538 goto out;
539
540 ret = -EINVAL;
541 if (optlen < 0)
542 goto out;
543
544 /* Bind this socket to a particular device like "eth0",
545 * as specified in the passed interface name. If the
546 * name is "" or the option length is zero the socket
547 * is not bound.
548 */
549 if (optlen > IFNAMSIZ - 1)
550 optlen = IFNAMSIZ - 1;
551 memset(devname, 0, sizeof(devname));
552
553 ret = -EFAULT;
554 if (copy_from_user(devname, optval, optlen))
555 goto out;
556
David S. Miller000ba2e2009-11-05 22:37:11 -0800557 index = 0;
558 if (devname[0] != '\0') {
Eric Dumazetbf8e56b2009-11-05 21:03:39 -0800559 struct net_device *dev;
David S. Miller48788092007-09-14 16:41:03 -0700560
Eric Dumazetbf8e56b2009-11-05 21:03:39 -0800561 rcu_read_lock();
562 dev = dev_get_by_name_rcu(net, devname);
563 if (dev)
564 index = dev->ifindex;
565 rcu_read_unlock();
David S. Miller48788092007-09-14 16:41:03 -0700566 ret = -ENODEV;
567 if (!dev)
568 goto out;
David S. Miller48788092007-09-14 16:41:03 -0700569 }
570
571 lock_sock(sk);
572 sk->sk_bound_dev_if = index;
573 sk_dst_reset(sk);
574 release_sock(sk);
575
576 ret = 0;
577
578out:
579#endif
580
581 return ret;
582}
583
Brian Haleyc91f6df2012-11-26 05:21:08 +0000584static int sock_getbindtodevice(struct sock *sk, char __user *optval,
585 int __user *optlen, int len)
586{
587 int ret = -ENOPROTOOPT;
588#ifdef CONFIG_NETDEVICES
589 struct net *net = sock_net(sk);
Brian Haleyc91f6df2012-11-26 05:21:08 +0000590 char devname[IFNAMSIZ];
Brian Haleyc91f6df2012-11-26 05:21:08 +0000591
592 if (sk->sk_bound_dev_if == 0) {
593 len = 0;
594 goto zero;
595 }
596
597 ret = -EINVAL;
598 if (len < IFNAMSIZ)
599 goto out;
600
Nicolas Schichan5dbe7c12013-06-26 17:23:42 +0200601 ret = netdev_get_name(net, devname, sk->sk_bound_dev_if);
602 if (ret)
Brian Haleyc91f6df2012-11-26 05:21:08 +0000603 goto out;
Brian Haleyc91f6df2012-11-26 05:21:08 +0000604
605 len = strlen(devname) + 1;
606
607 ret = -EFAULT;
608 if (copy_to_user(optval, devname, len))
609 goto out;
610
611zero:
612 ret = -EFAULT;
613 if (put_user(len, optlen))
614 goto out;
615
616 ret = 0;
617
618out:
619#endif
620
621 return ret;
622}
623
Pavel Emelyanovc0ef8772007-11-15 03:03:19 -0800624static inline void sock_valbool_flag(struct sock *sk, int bit, int valbool)
625{
626 if (valbool)
627 sock_set_flag(sk, bit);
628 else
629 sock_reset_flag(sk, bit);
630}
631
hannes@stressinduktion.orgf60e5992015-04-01 17:07:44 +0200632bool sk_mc_loop(struct sock *sk)
633{
634 if (dev_recursion_level())
635 return false;
636 if (!sk)
637 return true;
638 switch (sk->sk_family) {
639 case AF_INET:
640 return inet_sk(sk)->mc_loop;
641#if IS_ENABLED(CONFIG_IPV6)
642 case AF_INET6:
643 return inet6_sk(sk)->mc_loop;
644#endif
645 }
646 WARN_ON(1);
647 return true;
648}
649EXPORT_SYMBOL(sk_mc_loop);
650
Linus Torvalds1da177e2005-04-16 15:20:36 -0700651/*
652 * This is meant for all protocols to use and covers goings on
653 * at the socket level. Everything here is generic.
654 */
655
656int sock_setsockopt(struct socket *sock, int level, int optname,
David S. Millerb7058842009-09-30 16:12:20 -0700657 char __user *optval, unsigned int optlen)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700658{
Eric Dumazet2a915252009-05-27 11:30:05 +0000659 struct sock *sk = sock->sk;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700660 int val;
661 int valbool;
662 struct linger ling;
663 int ret = 0;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +0900664
Linus Torvalds1da177e2005-04-16 15:20:36 -0700665 /*
666 * Options without arguments
667 */
668
David S. Miller48788092007-09-14 16:41:03 -0700669 if (optname == SO_BINDTODEVICE)
Brian Haleyc91f6df2012-11-26 05:21:08 +0000670 return sock_setbindtodevice(sk, optval, optlen);
David S. Miller48788092007-09-14 16:41:03 -0700671
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700672 if (optlen < sizeof(int))
673 return -EINVAL;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +0900674
Linus Torvalds1da177e2005-04-16 15:20:36 -0700675 if (get_user(val, (int __user *)optval))
676 return -EFAULT;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +0900677
Eric Dumazet2a915252009-05-27 11:30:05 +0000678 valbool = val ? 1 : 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700679
680 lock_sock(sk);
681
Eric Dumazet2a915252009-05-27 11:30:05 +0000682 switch (optname) {
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700683 case SO_DEBUG:
Eric Dumazet2a915252009-05-27 11:30:05 +0000684 if (val && !capable(CAP_NET_ADMIN))
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700685 ret = -EACCES;
Eric Dumazet2a915252009-05-27 11:30:05 +0000686 else
Pavel Emelyanovc0ef8772007-11-15 03:03:19 -0800687 sock_valbool_flag(sk, SOCK_DBG, valbool);
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700688 break;
689 case SO_REUSEADDR:
Pavel Emelyanov4a17fd52012-04-19 03:39:36 +0000690 sk->sk_reuse = (valbool ? SK_CAN_REUSE : SK_NO_REUSE);
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700691 break;
Tom Herbert055dc212013-01-22 09:49:50 +0000692 case SO_REUSEPORT:
693 sk->sk_reuseport = valbool;
694 break;
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700695 case SO_TYPE:
Jan Engelhardt49c794e2009-08-04 07:28:28 +0000696 case SO_PROTOCOL:
Jan Engelhardt0d6038e2009-08-04 07:28:29 +0000697 case SO_DOMAIN:
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700698 case SO_ERROR:
699 ret = -ENOPROTOOPT;
700 break;
701 case SO_DONTROUTE:
Pavel Emelyanovc0ef8772007-11-15 03:03:19 -0800702 sock_valbool_flag(sk, SOCK_LOCALROUTE, valbool);
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700703 break;
704 case SO_BROADCAST:
705 sock_valbool_flag(sk, SOCK_BROADCAST, valbool);
706 break;
707 case SO_SNDBUF:
708 /* Don't error on this BSD doesn't and if you think
Eric Dumazet82981932012-04-26 20:07:59 +0000709 * about it this is right. Otherwise apps have to
710 * play 'guess the biggest size' games. RCVBUF/SNDBUF
711 * are treated in BSD as hints
712 */
713 val = min_t(u32, val, sysctl_wmem_max);
Patrick McHardyb0573de2005-08-09 19:30:51 -0700714set_sndbuf:
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700715 sk->sk_userlocks |= SOCK_SNDBUF_LOCK;
Eric Dumazet82981932012-04-26 20:07:59 +0000716 sk->sk_sndbuf = max_t(u32, val * 2, SOCK_MIN_SNDBUF);
717 /* Wake up sending tasks if we upped the value. */
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700718 sk->sk_write_space(sk);
719 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700720
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700721 case SO_SNDBUFFORCE:
722 if (!capable(CAP_NET_ADMIN)) {
723 ret = -EPERM;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700724 break;
725 }
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700726 goto set_sndbuf;
727
728 case SO_RCVBUF:
729 /* Don't error on this BSD doesn't and if you think
Eric Dumazet82981932012-04-26 20:07:59 +0000730 * about it this is right. Otherwise apps have to
731 * play 'guess the biggest size' games. RCVBUF/SNDBUF
732 * are treated in BSD as hints
733 */
734 val = min_t(u32, val, sysctl_rmem_max);
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700735set_rcvbuf:
736 sk->sk_userlocks |= SOCK_RCVBUF_LOCK;
737 /*
738 * We double it on the way in to account for
739 * "struct sk_buff" etc. overhead. Applications
740 * assume that the SO_RCVBUF setting they make will
741 * allow that much actual data to be received on that
742 * socket.
743 *
744 * Applications are unaware that "struct sk_buff" and
745 * other overheads allocate from the receive buffer
746 * during socket buffer allocation.
747 *
748 * And after considering the possible alternatives,
749 * returning the value we actually used in getsockopt
750 * is the most desirable behavior.
751 */
Eric Dumazet82981932012-04-26 20:07:59 +0000752 sk->sk_rcvbuf = max_t(u32, val * 2, SOCK_MIN_RCVBUF);
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700753 break;
754
755 case SO_RCVBUFFORCE:
756 if (!capable(CAP_NET_ADMIN)) {
757 ret = -EPERM;
758 break;
759 }
760 goto set_rcvbuf;
761
762 case SO_KEEPALIVE:
763#ifdef CONFIG_INET
Eric Dumazet3e109862012-09-24 07:00:11 +0000764 if (sk->sk_protocol == IPPROTO_TCP &&
765 sk->sk_type == SOCK_STREAM)
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700766 tcp_set_keepalive(sk, valbool);
767#endif
768 sock_valbool_flag(sk, SOCK_KEEPOPEN, valbool);
769 break;
770
771 case SO_OOBINLINE:
772 sock_valbool_flag(sk, SOCK_URGINLINE, valbool);
773 break;
774
775 case SO_NO_CHECK:
Tom Herbert28448b82014-05-23 08:47:19 -0700776 sk->sk_no_check_tx = valbool;
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700777 break;
778
779 case SO_PRIORITY:
Eric W. Biederman5e1fccc2012-11-16 03:03:04 +0000780 if ((val >= 0 && val <= 6) ||
781 ns_capable(sock_net(sk)->user_ns, CAP_NET_ADMIN))
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700782 sk->sk_priority = val;
783 else
784 ret = -EPERM;
785 break;
786
787 case SO_LINGER:
788 if (optlen < sizeof(ling)) {
789 ret = -EINVAL; /* 1003.1g */
790 break;
791 }
Eric Dumazet2a915252009-05-27 11:30:05 +0000792 if (copy_from_user(&ling, optval, sizeof(ling))) {
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700793 ret = -EFAULT;
794 break;
795 }
796 if (!ling.l_onoff)
797 sock_reset_flag(sk, SOCK_LINGER);
798 else {
799#if (BITS_PER_LONG == 32)
800 if ((unsigned int)ling.l_linger >= MAX_SCHEDULE_TIMEOUT/HZ)
801 sk->sk_lingertime = MAX_SCHEDULE_TIMEOUT;
802 else
803#endif
804 sk->sk_lingertime = (unsigned int)ling.l_linger * HZ;
805 sock_set_flag(sk, SOCK_LINGER);
806 }
807 break;
808
809 case SO_BSDCOMPAT:
810 sock_warn_obsolete_bsdism("setsockopt");
811 break;
812
813 case SO_PASSCRED:
814 if (valbool)
815 set_bit(SOCK_PASSCRED, &sock->flags);
816 else
817 clear_bit(SOCK_PASSCRED, &sock->flags);
818 break;
819
820 case SO_TIMESTAMP:
Eric Dumazet92f37fd2007-03-25 22:14:49 -0700821 case SO_TIMESTAMPNS:
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700822 if (valbool) {
Eric Dumazet92f37fd2007-03-25 22:14:49 -0700823 if (optname == SO_TIMESTAMP)
824 sock_reset_flag(sk, SOCK_RCVTSTAMPNS);
825 else
826 sock_set_flag(sk, SOCK_RCVTSTAMPNS);
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700827 sock_set_flag(sk, SOCK_RCVTSTAMP);
Patrick Ohly20d49472009-02-12 05:03:38 +0000828 sock_enable_timestamp(sk, SOCK_TIMESTAMP);
Eric Dumazet92f37fd2007-03-25 22:14:49 -0700829 } else {
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700830 sock_reset_flag(sk, SOCK_RCVTSTAMP);
Eric Dumazet92f37fd2007-03-25 22:14:49 -0700831 sock_reset_flag(sk, SOCK_RCVTSTAMPNS);
832 }
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700833 break;
834
Patrick Ohly20d49472009-02-12 05:03:38 +0000835 case SO_TIMESTAMPING:
836 if (val & ~SOF_TIMESTAMPING_MASK) {
Rémi Denis-Courmontf249fb72009-07-20 00:47:04 +0000837 ret = -EINVAL;
Patrick Ohly20d49472009-02-12 05:03:38 +0000838 break;
839 }
Willem de Bruijnb245be12015-01-30 13:29:32 -0500840
Willem de Bruijn09c2d252014-08-04 22:11:47 -0400841 if (val & SOF_TIMESTAMPING_OPT_ID &&
Willem de Bruijn4ed2d762014-08-04 22:11:49 -0400842 !(sk->sk_tsflags & SOF_TIMESTAMPING_OPT_ID)) {
WANG Congac5cc972015-12-16 23:39:04 -0800843 if (sk->sk_protocol == IPPROTO_TCP &&
844 sk->sk_type == SOCK_STREAM) {
Soheil Hassas Yeganeh6db8b962016-04-02 23:08:07 -0400845 if ((1 << sk->sk_state) &
846 (TCPF_CLOSE | TCPF_LISTEN)) {
Willem de Bruijn4ed2d762014-08-04 22:11:49 -0400847 ret = -EINVAL;
848 break;
849 }
850 sk->sk_tskey = tcp_sk(sk)->snd_una;
851 } else {
852 sk->sk_tskey = 0;
853 }
854 }
Willem de Bruijnb9f40e22014-08-04 22:11:46 -0400855 sk->sk_tsflags = val;
Patrick Ohly20d49472009-02-12 05:03:38 +0000856 if (val & SOF_TIMESTAMPING_RX_SOFTWARE)
857 sock_enable_timestamp(sk,
858 SOCK_TIMESTAMPING_RX_SOFTWARE);
859 else
860 sock_disable_timestamp(sk,
Eric Dumazet08e29af2011-11-28 12:04:18 +0000861 (1UL << SOCK_TIMESTAMPING_RX_SOFTWARE));
Patrick Ohly20d49472009-02-12 05:03:38 +0000862 break;
863
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700864 case SO_RCVLOWAT:
865 if (val < 0)
866 val = INT_MAX;
867 sk->sk_rcvlowat = val ? : 1;
868 break;
869
870 case SO_RCVTIMEO:
871 ret = sock_set_timeout(&sk->sk_rcvtimeo, optval, optlen);
872 break;
873
874 case SO_SNDTIMEO:
875 ret = sock_set_timeout(&sk->sk_sndtimeo, optval, optlen);
876 break;
877
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700878 case SO_ATTACH_FILTER:
879 ret = -EINVAL;
880 if (optlen == sizeof(struct sock_fprog)) {
881 struct sock_fprog fprog;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700882
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700883 ret = -EFAULT;
884 if (copy_from_user(&fprog, optval, sizeof(fprog)))
Linus Torvalds1da177e2005-04-16 15:20:36 -0700885 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700886
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700887 ret = sk_attach_filter(&fprog, sk);
888 }
889 break;
890
Alexei Starovoitov89aa0752014-12-01 15:06:35 -0800891 case SO_ATTACH_BPF:
892 ret = -EINVAL;
893 if (optlen == sizeof(u32)) {
894 u32 ufd;
895
896 ret = -EFAULT;
897 if (copy_from_user(&ufd, optval, sizeof(ufd)))
898 break;
899
900 ret = sk_attach_bpf(ufd, sk);
901 }
902 break;
903
Craig Gallek538950a2016-01-04 17:41:47 -0500904 case SO_ATTACH_REUSEPORT_CBPF:
905 ret = -EINVAL;
906 if (optlen == sizeof(struct sock_fprog)) {
907 struct sock_fprog fprog;
908
909 ret = -EFAULT;
910 if (copy_from_user(&fprog, optval, sizeof(fprog)))
911 break;
912
913 ret = sk_reuseport_attach_filter(&fprog, sk);
914 }
915 break;
916
917 case SO_ATTACH_REUSEPORT_EBPF:
918 ret = -EINVAL;
919 if (optlen == sizeof(u32)) {
920 u32 ufd;
921
922 ret = -EFAULT;
923 if (copy_from_user(&ufd, optval, sizeof(ufd)))
924 break;
925
926 ret = sk_reuseport_attach_bpf(ufd, sk);
927 }
928 break;
929
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700930 case SO_DETACH_FILTER:
Pavel Emelyanov55b33322007-10-17 21:21:26 -0700931 ret = sk_detach_filter(sk);
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700932 break;
933
Vincent Bernatd59577b2013-01-16 22:55:49 +0100934 case SO_LOCK_FILTER:
935 if (sock_flag(sk, SOCK_FILTER_LOCKED) && !valbool)
936 ret = -EPERM;
937 else
938 sock_valbool_flag(sk, SOCK_FILTER_LOCKED, valbool);
939 break;
940
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700941 case SO_PASSSEC:
942 if (valbool)
943 set_bit(SOCK_PASSSEC, &sock->flags);
944 else
945 clear_bit(SOCK_PASSSEC, &sock->flags);
946 break;
Laszlo Attila Toth4a19ec52008-01-30 19:08:16 -0800947 case SO_MARK:
Eric W. Biederman5e1fccc2012-11-16 03:03:04 +0000948 if (!ns_capable(sock_net(sk)->user_ns, CAP_NET_ADMIN))
Laszlo Attila Toth4a19ec52008-01-30 19:08:16 -0800949 ret = -EPERM;
Eric Dumazet2a915252009-05-27 11:30:05 +0000950 else
Laszlo Attila Toth4a19ec52008-01-30 19:08:16 -0800951 sk->sk_mark = val;
Laszlo Attila Toth4a19ec52008-01-30 19:08:16 -0800952 break;
Catherine Zhang877ce7c2006-06-29 12:27:47 -0700953
Neil Horman3b885782009-10-12 13:26:31 -0700954 case SO_RXQ_OVFL:
Johannes Berg8083f0f2011-10-07 03:30:20 +0000955 sock_valbool_flag(sk, SOCK_RXQ_OVFL, valbool);
Neil Horman3b885782009-10-12 13:26:31 -0700956 break;
Johannes Berg6e3e9392011-11-09 10:15:42 +0100957
958 case SO_WIFI_STATUS:
959 sock_valbool_flag(sk, SOCK_WIFI_STATUS, valbool);
960 break;
961
Pavel Emelyanovef64a542012-02-21 07:31:34 +0000962 case SO_PEEK_OFF:
963 if (sock->ops->set_peek_off)
Sasha Levin12663bf2013-12-07 17:26:27 -0500964 ret = sock->ops->set_peek_off(sk, val);
Pavel Emelyanovef64a542012-02-21 07:31:34 +0000965 else
966 ret = -EOPNOTSUPP;
967 break;
Ben Greear3bdc0eb2012-02-11 15:39:30 +0000968
969 case SO_NOFCS:
970 sock_valbool_flag(sk, SOCK_NOFCS, valbool);
971 break;
972
Keller, Jacob E7d4c04f2013-03-28 11:19:25 +0000973 case SO_SELECT_ERR_QUEUE:
974 sock_valbool_flag(sk, SOCK_SELECT_ERR_QUEUE, valbool);
975 break;
976
Cong Wange0d10952013-08-01 11:10:25 +0800977#ifdef CONFIG_NET_RX_BUSY_POLL
Eliezer Tamir64b0dc52013-07-10 17:13:36 +0300978 case SO_BUSY_POLL:
Eliezer Tamirdafcc432013-06-14 16:33:57 +0300979 /* allow unprivileged users to decrease the value */
980 if ((val > sk->sk_ll_usec) && !capable(CAP_NET_ADMIN))
981 ret = -EPERM;
982 else {
983 if (val < 0)
984 ret = -EINVAL;
985 else
986 sk->sk_ll_usec = val;
987 }
988 break;
989#endif
Eric Dumazet62748f32013-09-24 08:20:52 -0700990
991 case SO_MAX_PACING_RATE:
992 sk->sk_max_pacing_rate = val;
993 sk->sk_pacing_rate = min(sk->sk_pacing_rate,
994 sk->sk_max_pacing_rate);
995 break;
996
Eric Dumazet70da2682015-10-08 19:33:21 -0700997 case SO_INCOMING_CPU:
998 sk->sk_incoming_cpu = val;
999 break;
1000
Tom Herberta87cb3e2016-02-24 10:02:52 -08001001 case SO_CNX_ADVICE:
1002 if (val == 1)
1003 dst_negative_advice(sk);
1004 break;
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001005 default:
1006 ret = -ENOPROTOOPT;
1007 break;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001008 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07001009 release_sock(sk);
1010 return ret;
1011}
Eric Dumazet2a915252009-05-27 11:30:05 +00001012EXPORT_SYMBOL(sock_setsockopt);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001013
1014
stephen hemminger8f098982014-01-03 09:17:14 -08001015static void cred_to_ucred(struct pid *pid, const struct cred *cred,
1016 struct ucred *ucred)
Eric W. Biederman3f551f92010-06-13 03:28:59 +00001017{
1018 ucred->pid = pid_vnr(pid);
1019 ucred->uid = ucred->gid = -1;
1020 if (cred) {
1021 struct user_namespace *current_ns = current_user_ns();
1022
Eric W. Biedermanb2e4f542012-05-23 16:39:45 -06001023 ucred->uid = from_kuid_munged(current_ns, cred->euid);
1024 ucred->gid = from_kgid_munged(current_ns, cred->egid);
Eric W. Biederman3f551f92010-06-13 03:28:59 +00001025 }
1026}
1027
Linus Torvalds1da177e2005-04-16 15:20:36 -07001028int sock_getsockopt(struct socket *sock, int level, int optname,
1029 char __user *optval, int __user *optlen)
1030{
1031 struct sock *sk = sock->sk;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001032
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001033 union {
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001034 int val;
1035 struct linger ling;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001036 struct timeval tm;
1037 } v;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001038
H Hartley Sweeten4d0392b2010-01-15 01:08:58 -08001039 int lv = sizeof(int);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001040 int len;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001041
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001042 if (get_user(len, optlen))
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001043 return -EFAULT;
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001044 if (len < 0)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001045 return -EINVAL;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001046
Eugene Teo50fee1d2009-02-23 15:38:41 -08001047 memset(&v, 0, sizeof(v));
Clément Lecignedf0bca02009-02-12 16:59:09 -08001048
Eric Dumazet2a915252009-05-27 11:30:05 +00001049 switch (optname) {
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001050 case SO_DEBUG:
1051 v.val = sock_flag(sk, SOCK_DBG);
1052 break;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001053
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001054 case SO_DONTROUTE:
1055 v.val = sock_flag(sk, SOCK_LOCALROUTE);
1056 break;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001057
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001058 case SO_BROADCAST:
Eric Dumazet1b23a5d2012-05-16 05:57:07 +00001059 v.val = sock_flag(sk, SOCK_BROADCAST);
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001060 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001061
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001062 case SO_SNDBUF:
1063 v.val = sk->sk_sndbuf;
1064 break;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001065
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001066 case SO_RCVBUF:
1067 v.val = sk->sk_rcvbuf;
1068 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001069
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001070 case SO_REUSEADDR:
1071 v.val = sk->sk_reuse;
1072 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001073
Tom Herbert055dc212013-01-22 09:49:50 +00001074 case SO_REUSEPORT:
1075 v.val = sk->sk_reuseport;
1076 break;
1077
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001078 case SO_KEEPALIVE:
Eric Dumazet1b23a5d2012-05-16 05:57:07 +00001079 v.val = sock_flag(sk, SOCK_KEEPOPEN);
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001080 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001081
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001082 case SO_TYPE:
1083 v.val = sk->sk_type;
1084 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001085
Jan Engelhardt49c794e2009-08-04 07:28:28 +00001086 case SO_PROTOCOL:
1087 v.val = sk->sk_protocol;
1088 break;
1089
Jan Engelhardt0d6038e2009-08-04 07:28:29 +00001090 case SO_DOMAIN:
1091 v.val = sk->sk_family;
1092 break;
1093
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001094 case SO_ERROR:
1095 v.val = -sock_error(sk);
Eric Dumazet2a915252009-05-27 11:30:05 +00001096 if (v.val == 0)
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001097 v.val = xchg(&sk->sk_err_soft, 0);
1098 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001099
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001100 case SO_OOBINLINE:
Eric Dumazet1b23a5d2012-05-16 05:57:07 +00001101 v.val = sock_flag(sk, SOCK_URGINLINE);
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001102 break;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001103
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001104 case SO_NO_CHECK:
Tom Herbert28448b82014-05-23 08:47:19 -07001105 v.val = sk->sk_no_check_tx;
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001106 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001107
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001108 case SO_PRIORITY:
1109 v.val = sk->sk_priority;
1110 break;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001111
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001112 case SO_LINGER:
1113 lv = sizeof(v.ling);
Eric Dumazet1b23a5d2012-05-16 05:57:07 +00001114 v.ling.l_onoff = sock_flag(sk, SOCK_LINGER);
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001115 v.ling.l_linger = sk->sk_lingertime / HZ;
1116 break;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001117
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001118 case SO_BSDCOMPAT:
1119 sock_warn_obsolete_bsdism("getsockopt");
1120 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001121
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001122 case SO_TIMESTAMP:
Eric Dumazet92f37fd2007-03-25 22:14:49 -07001123 v.val = sock_flag(sk, SOCK_RCVTSTAMP) &&
1124 !sock_flag(sk, SOCK_RCVTSTAMPNS);
1125 break;
1126
1127 case SO_TIMESTAMPNS:
1128 v.val = sock_flag(sk, SOCK_RCVTSTAMPNS);
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001129 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001130
Patrick Ohly20d49472009-02-12 05:03:38 +00001131 case SO_TIMESTAMPING:
Willem de Bruijnb9f40e22014-08-04 22:11:46 -04001132 v.val = sk->sk_tsflags;
Patrick Ohly20d49472009-02-12 05:03:38 +00001133 break;
1134
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001135 case SO_RCVTIMEO:
Eric Dumazet2a915252009-05-27 11:30:05 +00001136 lv = sizeof(struct timeval);
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001137 if (sk->sk_rcvtimeo == MAX_SCHEDULE_TIMEOUT) {
1138 v.tm.tv_sec = 0;
1139 v.tm.tv_usec = 0;
1140 } else {
1141 v.tm.tv_sec = sk->sk_rcvtimeo / HZ;
1142 v.tm.tv_usec = ((sk->sk_rcvtimeo % HZ) * 1000000) / HZ;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001143 }
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001144 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001145
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001146 case SO_SNDTIMEO:
Eric Dumazet2a915252009-05-27 11:30:05 +00001147 lv = sizeof(struct timeval);
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001148 if (sk->sk_sndtimeo == MAX_SCHEDULE_TIMEOUT) {
1149 v.tm.tv_sec = 0;
1150 v.tm.tv_usec = 0;
1151 } else {
1152 v.tm.tv_sec = sk->sk_sndtimeo / HZ;
1153 v.tm.tv_usec = ((sk->sk_sndtimeo % HZ) * 1000000) / HZ;
1154 }
1155 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001156
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001157 case SO_RCVLOWAT:
1158 v.val = sk->sk_rcvlowat;
1159 break;
Catherine Zhang877ce7c2006-06-29 12:27:47 -07001160
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001161 case SO_SNDLOWAT:
Eric Dumazet2a915252009-05-27 11:30:05 +00001162 v.val = 1;
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001163 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001164
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001165 case SO_PASSCRED:
Eric Dumazet82981932012-04-26 20:07:59 +00001166 v.val = !!test_bit(SOCK_PASSCRED, &sock->flags);
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001167 break;
1168
1169 case SO_PEERCRED:
Eric W. Biederman109f6e32010-06-13 03:30:14 +00001170 {
1171 struct ucred peercred;
1172 if (len > sizeof(peercred))
1173 len = sizeof(peercred);
1174 cred_to_ucred(sk->sk_peer_pid, sk->sk_peer_cred, &peercred);
1175 if (copy_to_user(optval, &peercred, len))
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001176 return -EFAULT;
1177 goto lenout;
Eric W. Biederman109f6e32010-06-13 03:30:14 +00001178 }
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001179
1180 case SO_PEERNAME:
1181 {
1182 char address[128];
1183
1184 if (sock->ops->getname(sock, (struct sockaddr *)address, &lv, 2))
1185 return -ENOTCONN;
1186 if (lv < len)
1187 return -EINVAL;
1188 if (copy_to_user(optval, address, len))
1189 return -EFAULT;
1190 goto lenout;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001191 }
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001192
1193 /* Dubious BSD thing... Probably nobody even uses it, but
1194 * the UNIX standard wants it for whatever reason... -DaveM
1195 */
1196 case SO_ACCEPTCONN:
1197 v.val = sk->sk_state == TCP_LISTEN;
1198 break;
1199
1200 case SO_PASSSEC:
Eric Dumazet82981932012-04-26 20:07:59 +00001201 v.val = !!test_bit(SOCK_PASSSEC, &sock->flags);
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001202 break;
1203
1204 case SO_PEERSEC:
1205 return security_socket_getpeersec_stream(sock, optval, optlen, len);
1206
Laszlo Attila Toth4a19ec52008-01-30 19:08:16 -08001207 case SO_MARK:
1208 v.val = sk->sk_mark;
1209 break;
1210
Neil Horman3b885782009-10-12 13:26:31 -07001211 case SO_RXQ_OVFL:
Eric Dumazet1b23a5d2012-05-16 05:57:07 +00001212 v.val = sock_flag(sk, SOCK_RXQ_OVFL);
Neil Horman3b885782009-10-12 13:26:31 -07001213 break;
1214
Johannes Berg6e3e9392011-11-09 10:15:42 +01001215 case SO_WIFI_STATUS:
Eric Dumazet1b23a5d2012-05-16 05:57:07 +00001216 v.val = sock_flag(sk, SOCK_WIFI_STATUS);
Johannes Berg6e3e9392011-11-09 10:15:42 +01001217 break;
1218
Pavel Emelyanovef64a542012-02-21 07:31:34 +00001219 case SO_PEEK_OFF:
1220 if (!sock->ops->set_peek_off)
1221 return -EOPNOTSUPP;
1222
1223 v.val = sk->sk_peek_off;
1224 break;
David S. Millerbc2f7992012-02-24 14:48:34 -05001225 case SO_NOFCS:
Eric Dumazet1b23a5d2012-05-16 05:57:07 +00001226 v.val = sock_flag(sk, SOCK_NOFCS);
David S. Millerbc2f7992012-02-24 14:48:34 -05001227 break;
Brian Haleyc91f6df2012-11-26 05:21:08 +00001228
Pavel Emelyanovf7b86bf2012-10-18 23:55:56 +00001229 case SO_BINDTODEVICE:
Brian Haleyc91f6df2012-11-26 05:21:08 +00001230 return sock_getbindtodevice(sk, optval, optlen, len);
1231
Pavel Emelyanova8fc9272012-11-01 02:01:48 +00001232 case SO_GET_FILTER:
1233 len = sk_get_filter(sk, (struct sock_filter __user *)optval, len);
1234 if (len < 0)
1235 return len;
1236
1237 goto lenout;
Brian Haleyc91f6df2012-11-26 05:21:08 +00001238
Vincent Bernatd59577b2013-01-16 22:55:49 +01001239 case SO_LOCK_FILTER:
1240 v.val = sock_flag(sk, SOCK_FILTER_LOCKED);
1241 break;
1242
Michal Sekletarea02f942014-01-17 17:09:45 +01001243 case SO_BPF_EXTENSIONS:
1244 v.val = bpf_tell_extensions();
1245 break;
1246
Keller, Jacob E7d4c04f2013-03-28 11:19:25 +00001247 case SO_SELECT_ERR_QUEUE:
1248 v.val = sock_flag(sk, SOCK_SELECT_ERR_QUEUE);
1249 break;
1250
Cong Wange0d10952013-08-01 11:10:25 +08001251#ifdef CONFIG_NET_RX_BUSY_POLL
Eliezer Tamir64b0dc52013-07-10 17:13:36 +03001252 case SO_BUSY_POLL:
Eliezer Tamirdafcc432013-06-14 16:33:57 +03001253 v.val = sk->sk_ll_usec;
1254 break;
1255#endif
1256
Eric Dumazet62748f32013-09-24 08:20:52 -07001257 case SO_MAX_PACING_RATE:
1258 v.val = sk->sk_max_pacing_rate;
1259 break;
1260
Eric Dumazet2c8c56e2014-11-11 05:54:28 -08001261 case SO_INCOMING_CPU:
1262 v.val = sk->sk_incoming_cpu;
1263 break;
1264
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001265 default:
YOSHIFUJI Hideaki/吉藤英明443b5992015-03-23 18:04:13 +09001266 /* We implement the SO_SNDLOWAT etc to not be settable
1267 * (1003.1g 7).
1268 */
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001269 return -ENOPROTOOPT;
1270 }
1271
Linus Torvalds1da177e2005-04-16 15:20:36 -07001272 if (len > lv)
1273 len = lv;
1274 if (copy_to_user(optval, &v, len))
1275 return -EFAULT;
1276lenout:
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001277 if (put_user(len, optlen))
1278 return -EFAULT;
1279 return 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001280}
1281
Ingo Molnara5b5bb92006-07-03 00:25:35 -07001282/*
1283 * Initialize an sk_lock.
1284 *
1285 * (We also register the sk_lock with the lock validator.)
1286 */
Dave Jonesb6f99a22007-03-22 12:27:49 -07001287static inline void sock_lock_init(struct sock *sk)
Ingo Molnara5b5bb92006-07-03 00:25:35 -07001288{
Peter Zijlstraed075362006-12-06 20:35:24 -08001289 sock_lock_init_class_and_name(sk,
1290 af_family_slock_key_strings[sk->sk_family],
1291 af_family_slock_keys + sk->sk_family,
1292 af_family_key_strings[sk->sk_family],
1293 af_family_keys + sk->sk_family);
Ingo Molnara5b5bb92006-07-03 00:25:35 -07001294}
1295
Eric Dumazet4dc6dc72009-07-15 23:13:10 +00001296/*
1297 * Copy all fields from osk to nsk but nsk->sk_refcnt must not change yet,
1298 * even temporarly, because of RCU lookups. sk_node should also be left as is.
Eric Dumazet68835ab2010-11-30 19:04:07 +00001299 * We must not copy fields between sk_dontcopy_begin and sk_dontcopy_end
Eric Dumazet4dc6dc72009-07-15 23:13:10 +00001300 */
Pavel Emelyanovf1a6c4d2007-11-01 00:29:45 -07001301static void sock_copy(struct sock *nsk, const struct sock *osk)
1302{
1303#ifdef CONFIG_SECURITY_NETWORK
1304 void *sptr = nsk->sk_security;
1305#endif
Eric Dumazet68835ab2010-11-30 19:04:07 +00001306 memcpy(nsk, osk, offsetof(struct sock, sk_dontcopy_begin));
1307
1308 memcpy(&nsk->sk_dontcopy_end, &osk->sk_dontcopy_end,
1309 osk->sk_prot->obj_size - offsetof(struct sock, sk_dontcopy_end));
1310
Pavel Emelyanovf1a6c4d2007-11-01 00:29:45 -07001311#ifdef CONFIG_SECURITY_NETWORK
1312 nsk->sk_security = sptr;
1313 security_sk_clone(osk, nsk);
1314#endif
1315}
1316
Octavian Purdilafcbdf092010-12-16 14:26:56 -08001317void sk_prot_clear_portaddr_nulls(struct sock *sk, int size)
1318{
1319 unsigned long nulls1, nulls2;
1320
1321 nulls1 = offsetof(struct sock, __sk_common.skc_node.next);
1322 nulls2 = offsetof(struct sock, __sk_common.skc_portaddr_node.next);
1323 if (nulls1 > nulls2)
1324 swap(nulls1, nulls2);
1325
1326 if (nulls1 != 0)
1327 memset((char *)sk, 0, nulls1);
1328 memset((char *)sk + nulls1 + sizeof(void *), 0,
1329 nulls2 - nulls1 - sizeof(void *));
1330 memset((char *)sk + nulls2 + sizeof(void *), 0,
1331 size - nulls2 - sizeof(void *));
1332}
1333EXPORT_SYMBOL(sk_prot_clear_portaddr_nulls);
1334
Pavel Emelyanov2e4afe72007-11-01 00:36:26 -07001335static struct sock *sk_prot_alloc(struct proto *prot, gfp_t priority,
1336 int family)
Pavel Emelyanovc308c1b22007-11-01 00:33:50 -07001337{
1338 struct sock *sk;
1339 struct kmem_cache *slab;
1340
1341 slab = prot->slab;
Eric Dumazete912b112009-07-08 19:36:05 +00001342 if (slab != NULL) {
1343 sk = kmem_cache_alloc(slab, priority & ~__GFP_ZERO);
1344 if (!sk)
1345 return sk;
1346 if (priority & __GFP_ZERO) {
Octavian Purdilafcbdf092010-12-16 14:26:56 -08001347 if (prot->clear_sk)
1348 prot->clear_sk(sk, prot->obj_size);
1349 else
1350 sk_prot_clear_nulls(sk, prot->obj_size);
Eric Dumazete912b112009-07-08 19:36:05 +00001351 }
Octavian Purdilafcbdf092010-12-16 14:26:56 -08001352 } else
Pavel Emelyanovc308c1b22007-11-01 00:33:50 -07001353 sk = kmalloc(prot->obj_size, priority);
1354
Pavel Emelyanov2e4afe72007-11-01 00:36:26 -07001355 if (sk != NULL) {
Vegard Nossuma98b65a2009-02-26 14:46:57 +01001356 kmemcheck_annotate_bitfield(sk, flags);
1357
Pavel Emelyanov2e4afe72007-11-01 00:36:26 -07001358 if (security_sk_alloc(sk, family, priority))
1359 goto out_free;
1360
1361 if (!try_module_get(prot->owner))
1362 goto out_free_sec;
Krishna Kumare022f0b2009-10-19 23:46:20 +00001363 sk_tx_queue_clear(sk);
Tejun Heobd1060a2015-12-07 17:38:53 -05001364 cgroup_sk_alloc(&sk->sk_cgrp_data);
Pavel Emelyanov2e4afe72007-11-01 00:36:26 -07001365 }
1366
Pavel Emelyanovc308c1b22007-11-01 00:33:50 -07001367 return sk;
Pavel Emelyanov2e4afe72007-11-01 00:36:26 -07001368
1369out_free_sec:
1370 security_sk_free(sk);
1371out_free:
1372 if (slab != NULL)
1373 kmem_cache_free(slab, sk);
1374 else
1375 kfree(sk);
1376 return NULL;
Pavel Emelyanovc308c1b22007-11-01 00:33:50 -07001377}
1378
1379static void sk_prot_free(struct proto *prot, struct sock *sk)
1380{
1381 struct kmem_cache *slab;
Pavel Emelyanov2e4afe72007-11-01 00:36:26 -07001382 struct module *owner;
Pavel Emelyanovc308c1b22007-11-01 00:33:50 -07001383
Pavel Emelyanov2e4afe72007-11-01 00:36:26 -07001384 owner = prot->owner;
Pavel Emelyanovc308c1b22007-11-01 00:33:50 -07001385 slab = prot->slab;
Pavel Emelyanov2e4afe72007-11-01 00:36:26 -07001386
Tejun Heobd1060a2015-12-07 17:38:53 -05001387 cgroup_sk_free(&sk->sk_cgrp_data);
Pavel Emelyanov2e4afe72007-11-01 00:36:26 -07001388 security_sk_free(sk);
Pavel Emelyanovc308c1b22007-11-01 00:33:50 -07001389 if (slab != NULL)
1390 kmem_cache_free(slab, sk);
1391 else
1392 kfree(sk);
Pavel Emelyanov2e4afe72007-11-01 00:36:26 -07001393 module_put(owner);
Pavel Emelyanovc308c1b22007-11-01 00:33:50 -07001394}
1395
Linus Torvalds1da177e2005-04-16 15:20:36 -07001396/**
1397 * sk_alloc - All socket objects are allocated here
Randy Dunlapc4ea43c2007-10-12 21:17:49 -07001398 * @net: the applicable net namespace
Pavel Pisa4dc3b162005-05-01 08:59:25 -07001399 * @family: protocol family
1400 * @priority: for allocation (%GFP_KERNEL, %GFP_ATOMIC, etc)
1401 * @prot: struct proto associated with this new sock instance
Eric W. Biederman11aa9c22015-05-08 21:09:13 -05001402 * @kern: is this to be a kernel socket?
Linus Torvalds1da177e2005-04-16 15:20:36 -07001403 */
Eric W. Biederman1b8d7ae2007-10-08 23:24:22 -07001404struct sock *sk_alloc(struct net *net, int family, gfp_t priority,
Eric W. Biederman11aa9c22015-05-08 21:09:13 -05001405 struct proto *prot, int kern)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001406{
Pavel Emelyanovc308c1b22007-11-01 00:33:50 -07001407 struct sock *sk;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001408
Pavel Emelyanov154adbc2007-11-01 00:38:43 -07001409 sk = sk_prot_alloc(prot, priority | __GFP_ZERO, family);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001410 if (sk) {
Pavel Emelyanov154adbc2007-11-01 00:38:43 -07001411 sk->sk_family = family;
1412 /*
1413 * See comment in struct sock definition to understand
1414 * why we need sk_prot_creator -acme
1415 */
1416 sk->sk_prot = sk->sk_prot_creator = prot;
1417 sock_lock_init(sk);
Eric W. Biederman26abe142015-05-08 21:10:31 -05001418 sk->sk_net_refcnt = kern ? 0 : 1;
1419 if (likely(sk->sk_net_refcnt))
1420 get_net(net);
1421 sock_net_set(sk, net);
Jarek Poplawskid66ee052009-08-30 23:15:36 +00001422 atomic_set(&sk->sk_wmem_alloc, 1);
Herbert Xuf8451722010-05-24 00:12:34 -07001423
Tejun Heo2a56a1f2015-12-07 17:38:52 -05001424 sock_update_classid(&sk->sk_cgrp_data);
1425 sock_update_netprioidx(&sk->sk_cgrp_data);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001426 }
Frank Filza79af592005-09-27 15:23:38 -07001427
Pavel Emelyanov2e4afe72007-11-01 00:36:26 -07001428 return sk;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001429}
Eric Dumazet2a915252009-05-27 11:30:05 +00001430EXPORT_SYMBOL(sk_alloc);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001431
Eric Dumazeta4298e42016-04-01 08:52:12 -07001432/* Sockets having SOCK_RCU_FREE will call this function after one RCU
1433 * grace period. This is the case for UDP sockets and TCP listeners.
1434 */
1435static void __sk_destruct(struct rcu_head *head)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001436{
Eric Dumazeta4298e42016-04-01 08:52:12 -07001437 struct sock *sk = container_of(head, struct sock, sk_rcu);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001438 struct sk_filter *filter;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001439
1440 if (sk->sk_destruct)
1441 sk->sk_destruct(sk);
1442
Paul E. McKenneya898def2010-02-22 17:04:49 -08001443 filter = rcu_dereference_check(sk->sk_filter,
1444 atomic_read(&sk->sk_wmem_alloc) == 0);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001445 if (filter) {
Pavel Emelyanov309dd5f2007-10-17 21:21:51 -07001446 sk_filter_uncharge(sk, filter);
Stephen Hemmingera9b3cd72011-08-01 16:19:00 +00001447 RCU_INIT_POINTER(sk->sk_filter, NULL);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001448 }
Craig Gallek538950a2016-01-04 17:41:47 -05001449 if (rcu_access_pointer(sk->sk_reuseport_cb))
1450 reuseport_detach_sock(sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001451
Eric Dumazet08e29af2011-11-28 12:04:18 +00001452 sock_disable_timestamp(sk, SK_FLAGS_TIMESTAMP);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001453
1454 if (atomic_read(&sk->sk_omem_alloc))
Joe Perchese005d192012-05-16 19:58:40 +00001455 pr_debug("%s: optmem leakage (%d bytes) detected\n",
1456 __func__, atomic_read(&sk->sk_omem_alloc));
Linus Torvalds1da177e2005-04-16 15:20:36 -07001457
Eric W. Biederman109f6e32010-06-13 03:30:14 +00001458 if (sk->sk_peer_cred)
1459 put_cred(sk->sk_peer_cred);
1460 put_pid(sk->sk_peer_pid);
Eric W. Biederman26abe142015-05-08 21:10:31 -05001461 if (likely(sk->sk_net_refcnt))
1462 put_net(sock_net(sk));
Pavel Emelyanovc308c1b22007-11-01 00:33:50 -07001463 sk_prot_free(sk->sk_prot_creator, sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001464}
Eric Dumazet2b85a342009-06-11 02:55:43 -07001465
Eric Dumazeta4298e42016-04-01 08:52:12 -07001466void sk_destruct(struct sock *sk)
1467{
1468 if (sock_flag(sk, SOCK_RCU_FREE))
1469 call_rcu(&sk->sk_rcu, __sk_destruct);
1470 else
1471 __sk_destruct(&sk->sk_rcu);
1472}
1473
Craig Gallekeb4cb002015-06-15 11:26:18 -04001474static void __sk_free(struct sock *sk)
1475{
Craig Gallekb9226222015-06-30 12:49:32 -04001476 if (unlikely(sock_diag_has_destroy_listeners(sk) && sk->sk_net_refcnt))
Craig Gallekeb4cb002015-06-15 11:26:18 -04001477 sock_diag_broadcast_destroy(sk);
1478 else
1479 sk_destruct(sk);
1480}
1481
Eric Dumazet2b85a342009-06-11 02:55:43 -07001482void sk_free(struct sock *sk)
1483{
1484 /*
Lucas De Marchi25985ed2011-03-30 22:57:33 -03001485 * We subtract one from sk_wmem_alloc and can know if
Eric Dumazet2b85a342009-06-11 02:55:43 -07001486 * some packets are still in some tx queue.
1487 * If not null, sock_wfree() will call __sk_free(sk) later
1488 */
1489 if (atomic_dec_and_test(&sk->sk_wmem_alloc))
1490 __sk_free(sk);
1491}
Eric Dumazet2a915252009-05-27 11:30:05 +00001492EXPORT_SYMBOL(sk_free);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001493
Eric Dumazete56c57d2011-11-08 17:07:07 -05001494/**
1495 * sk_clone_lock - clone a socket, and lock its clone
1496 * @sk: the socket to clone
1497 * @priority: for allocation (%GFP_KERNEL, %GFP_ATOMIC, etc)
1498 *
1499 * Caller must unlock socket even in error path (bh_unlock_sock(newsk))
1500 */
1501struct sock *sk_clone_lock(const struct sock *sk, const gfp_t priority)
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001502{
Pavel Emelyanov8fd1d172007-11-01 00:37:32 -07001503 struct sock *newsk;
Alexei Starovoitov278571b2014-07-30 20:34:12 -07001504 bool is_charged = true;
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001505
Pavel Emelyanov8fd1d172007-11-01 00:37:32 -07001506 newsk = sk_prot_alloc(sk->sk_prot, priority, sk->sk_family);
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001507 if (newsk != NULL) {
1508 struct sk_filter *filter;
1509
Venkat Yekkirala892c1412006-08-04 23:08:56 -07001510 sock_copy(newsk, sk);
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001511
1512 /* SANITY */
Sowmini Varadhan8a681732015-07-30 15:50:36 +02001513 if (likely(newsk->sk_net_refcnt))
1514 get_net(sock_net(newsk));
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001515 sk_node_init(&newsk->sk_node);
1516 sock_lock_init(newsk);
1517 bh_lock_sock(newsk);
Eric Dumazetfa438cc2007-03-04 16:05:44 -08001518 newsk->sk_backlog.head = newsk->sk_backlog.tail = NULL;
Zhu Yi8eae9392010-03-04 18:01:40 +00001519 newsk->sk_backlog.len = 0;
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001520
1521 atomic_set(&newsk->sk_rmem_alloc, 0);
Eric Dumazet2b85a342009-06-11 02:55:43 -07001522 /*
1523 * sk_wmem_alloc set to one (see sk_free() and sock_wfree())
1524 */
1525 atomic_set(&newsk->sk_wmem_alloc, 1);
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001526 atomic_set(&newsk->sk_omem_alloc, 0);
1527 skb_queue_head_init(&newsk->sk_receive_queue);
1528 skb_queue_head_init(&newsk->sk_write_queue);
1529
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001530 rwlock_init(&newsk->sk_callback_lock);
Peter Zijlstra443aef0e2007-07-19 01:49:00 -07001531 lockdep_set_class_and_name(&newsk->sk_callback_lock,
1532 af_callback_keys + newsk->sk_family,
1533 af_family_clock_key_strings[newsk->sk_family]);
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001534
1535 newsk->sk_dst_cache = NULL;
1536 newsk->sk_wmem_queued = 0;
1537 newsk->sk_forward_alloc = 0;
Eric Dumazet9caad862016-04-01 08:52:20 -07001538 atomic_set(&newsk->sk_drops, 0);
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001539 newsk->sk_send_head = NULL;
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001540 newsk->sk_userlocks = sk->sk_userlocks & ~SOCK_BINDPORT_LOCK;
1541
1542 sock_reset_flag(newsk, SOCK_DONE);
1543 skb_queue_head_init(&newsk->sk_error_queue);
1544
Eric Dumazet0d7da9d2010-10-25 03:47:05 +00001545 filter = rcu_dereference_protected(newsk->sk_filter, 1);
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001546 if (filter != NULL)
Alexei Starovoitov278571b2014-07-30 20:34:12 -07001547 /* though it's an empty new sock, the charging may fail
1548 * if sysctl_optmem_max was changed between creation of
1549 * original socket and cloning
1550 */
1551 is_charged = sk_filter_charge(newsk, filter);
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001552
Eric Dumazetd188ba82015-12-08 07:22:02 -08001553 if (unlikely(!is_charged || xfrm_sk_clone_policy(newsk, sk))) {
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001554 /* It is still raw copy of parent, so invalidate
1555 * destructor and make plain sk_free() */
1556 newsk->sk_destruct = NULL;
Thomas Gleixnerb0691c82011-10-25 02:30:50 +00001557 bh_unlock_sock(newsk);
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001558 sk_free(newsk);
1559 newsk = NULL;
1560 goto out;
1561 }
Craig Gallekfa463492016-02-10 11:50:39 -05001562 RCU_INIT_POINTER(newsk->sk_reuseport_cb, NULL);
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001563
1564 newsk->sk_err = 0;
1565 newsk->sk_priority = 0;
Eric Dumazet2c8c56e2014-11-11 05:54:28 -08001566 newsk->sk_incoming_cpu = raw_smp_processor_id();
Eric Dumazet33cf7c92015-03-11 18:53:14 -07001567 atomic64_set(&newsk->sk_cookie, 0);
Eric Dumazet4dc6dc72009-07-15 23:13:10 +00001568 /*
1569 * Before updating sk_refcnt, we must commit prior changes to memory
1570 * (Documentation/RCU/rculist_nulls.txt for details)
1571 */
1572 smp_wmb();
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001573 atomic_set(&newsk->sk_refcnt, 2);
1574
1575 /*
1576 * Increment the counter in the same struct proto as the master
1577 * sock (sk_refcnt_debug_inc uses newsk->sk_prot->socks, that
1578 * is the same as sk->sk_prot->socks, as this field was copied
1579 * with memcpy).
1580 *
1581 * This _changes_ the previous behaviour, where
1582 * tcp_create_openreq_child always was incrementing the
1583 * equivalent to tcp_prot->socks (inet_sock_nr), so this have
1584 * to be taken into account in all callers. -acme
1585 */
1586 sk_refcnt_debug_inc(newsk);
David S. Miller972692e2008-06-17 22:41:38 -07001587 sk_set_socket(newsk, NULL);
Eric Dumazet43815482010-04-29 11:01:49 +00001588 newsk->sk_wq = NULL;
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001589
Johannes Weinerbaac50b2016-01-14 15:21:17 -08001590 if (mem_cgroup_sockets_enabled && sk->sk_memcg)
Johannes Weiner3d596f72016-01-14 15:21:05 -08001591 sock_update_memcg(newsk);
Glauber Costaf3f511e2012-01-05 20:16:39 +00001592
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001593 if (newsk->sk_prot->sockets_allocated)
Glauber Costa180d8cd2011-12-11 21:47:02 +00001594 sk_sockets_allocated_inc(newsk);
Octavian Purdila704da5602010-01-08 00:00:09 -08001595
Hannes Frederic Sowa080a2702015-10-26 13:51:37 +01001596 if (sock_needs_netstamp(sk) &&
1597 newsk->sk_flags & SK_FLAGS_TIMESTAMP)
Octavian Purdila704da5602010-01-08 00:00:09 -08001598 net_enable_timestamp();
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001599 }
1600out:
1601 return newsk;
1602}
Eric Dumazete56c57d2011-11-08 17:07:07 -05001603EXPORT_SYMBOL_GPL(sk_clone_lock);
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001604
Andi Kleen99580892007-04-20 17:12:43 -07001605void sk_setup_caps(struct sock *sk, struct dst_entry *dst)
1606{
Eric Dumazetd6a4e262015-05-26 08:55:28 -07001607 u32 max_segs = 1;
1608
Eric Dumazet6bd4f352015-12-02 21:53:57 -08001609 sk_dst_set(sk, dst);
Andi Kleen99580892007-04-20 17:12:43 -07001610 sk->sk_route_caps = dst->dev->features;
1611 if (sk->sk_route_caps & NETIF_F_GSO)
Herbert Xu4fcd6b92007-05-31 22:15:50 -07001612 sk->sk_route_caps |= NETIF_F_GSO_SOFTWARE;
Eric Dumazeta4654192010-05-16 00:36:33 -07001613 sk->sk_route_caps &= ~sk->sk_route_nocaps;
Andi Kleen99580892007-04-20 17:12:43 -07001614 if (sk_can_gso(sk)) {
Peter P Waskiewicz Jr82cc1a72008-03-21 03:43:19 -07001615 if (dst->header_len) {
Andi Kleen99580892007-04-20 17:12:43 -07001616 sk->sk_route_caps &= ~NETIF_F_GSO_MASK;
Peter P Waskiewicz Jr82cc1a72008-03-21 03:43:19 -07001617 } else {
Andi Kleen99580892007-04-20 17:12:43 -07001618 sk->sk_route_caps |= NETIF_F_SG | NETIF_F_HW_CSUM;
Peter P Waskiewicz Jr82cc1a72008-03-21 03:43:19 -07001619 sk->sk_gso_max_size = dst->dev->gso_max_size;
Eric Dumazetd6a4e262015-05-26 08:55:28 -07001620 max_segs = max_t(u32, dst->dev->gso_max_segs, 1);
Peter P Waskiewicz Jr82cc1a72008-03-21 03:43:19 -07001621 }
Andi Kleen99580892007-04-20 17:12:43 -07001622 }
Eric Dumazetd6a4e262015-05-26 08:55:28 -07001623 sk->sk_gso_max_segs = max_segs;
Andi Kleen99580892007-04-20 17:12:43 -07001624}
1625EXPORT_SYMBOL_GPL(sk_setup_caps);
1626
Linus Torvalds1da177e2005-04-16 15:20:36 -07001627/*
1628 * Simple resource managers for sockets.
1629 */
1630
1631
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001632/*
1633 * Write buffer destructor automatically called from kfree_skb.
Linus Torvalds1da177e2005-04-16 15:20:36 -07001634 */
1635void sock_wfree(struct sk_buff *skb)
1636{
1637 struct sock *sk = skb->sk;
Eric Dumazetd99927f2009-09-24 10:49:24 +00001638 unsigned int len = skb->truesize;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001639
Eric Dumazetd99927f2009-09-24 10:49:24 +00001640 if (!sock_flag(sk, SOCK_USE_WRITE_QUEUE)) {
1641 /*
1642 * Keep a reference on sk_wmem_alloc, this will be released
1643 * after sk_write_space() call
1644 */
1645 atomic_sub(len - 1, &sk->sk_wmem_alloc);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001646 sk->sk_write_space(sk);
Eric Dumazetd99927f2009-09-24 10:49:24 +00001647 len = 1;
1648 }
Eric Dumazet2b85a342009-06-11 02:55:43 -07001649 /*
Eric Dumazetd99927f2009-09-24 10:49:24 +00001650 * if sk_wmem_alloc reaches 0, we must finish what sk_free()
1651 * could not do because of in-flight packets
Eric Dumazet2b85a342009-06-11 02:55:43 -07001652 */
Eric Dumazetd99927f2009-09-24 10:49:24 +00001653 if (atomic_sub_and_test(len, &sk->sk_wmem_alloc))
Eric Dumazet2b85a342009-06-11 02:55:43 -07001654 __sk_free(sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001655}
Eric Dumazet2a915252009-05-27 11:30:05 +00001656EXPORT_SYMBOL(sock_wfree);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001657
Eric Dumazet9e17f8a2015-11-01 15:36:55 -08001658void skb_set_owner_w(struct sk_buff *skb, struct sock *sk)
1659{
1660 skb_orphan(skb);
1661 skb->sk = sk;
1662#ifdef CONFIG_INET
1663 if (unlikely(!sk_fullsock(sk))) {
1664 skb->destructor = sock_edemux;
1665 sock_hold(sk);
1666 return;
1667 }
1668#endif
1669 skb->destructor = sock_wfree;
1670 skb_set_hash_from_sk(skb, sk);
1671 /*
1672 * We used to take a refcount on sk, but following operation
1673 * is enough to guarantee sk_free() wont free this sock until
1674 * all in-flight packets are completed
1675 */
1676 atomic_add(skb->truesize, &sk->sk_wmem_alloc);
1677}
1678EXPORT_SYMBOL(skb_set_owner_w);
1679
Eric Dumazetf2f872f2013-07-30 17:55:08 -07001680void skb_orphan_partial(struct sk_buff *skb)
1681{
1682 /* TCP stack sets skb->ooo_okay based on sk_wmem_alloc,
1683 * so we do not completely orphan skb, but transfert all
1684 * accounted bytes but one, to avoid unexpected reorders.
1685 */
1686 if (skb->destructor == sock_wfree
1687#ifdef CONFIG_INET
1688 || skb->destructor == tcp_wfree
1689#endif
1690 ) {
1691 atomic_sub(skb->truesize - 1, &skb->sk->sk_wmem_alloc);
1692 skb->truesize = 1;
1693 } else {
1694 skb_orphan(skb);
1695 }
1696}
1697EXPORT_SYMBOL(skb_orphan_partial);
1698
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001699/*
1700 * Read buffer destructor automatically called from kfree_skb.
Linus Torvalds1da177e2005-04-16 15:20:36 -07001701 */
1702void sock_rfree(struct sk_buff *skb)
1703{
1704 struct sock *sk = skb->sk;
Eric Dumazetd361fd52010-07-10 22:45:17 +00001705 unsigned int len = skb->truesize;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001706
Eric Dumazetd361fd52010-07-10 22:45:17 +00001707 atomic_sub(len, &sk->sk_rmem_alloc);
1708 sk_mem_uncharge(sk, len);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001709}
Eric Dumazet2a915252009-05-27 11:30:05 +00001710EXPORT_SYMBOL(sock_rfree);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001711
Oliver Hartkopp7768eed2015-03-10 19:03:46 +01001712/*
1713 * Buffer destructor for skbs that are not used directly in read or write
1714 * path, e.g. for error handler skbs. Automatically called from kfree_skb.
1715 */
Alexander Duyck62bccb82014-09-04 13:31:35 -04001716void sock_efree(struct sk_buff *skb)
1717{
1718 sock_put(skb->sk);
1719}
1720EXPORT_SYMBOL(sock_efree);
1721
Eric W. Biederman976d02012012-05-23 17:16:53 -06001722kuid_t sock_i_uid(struct sock *sk)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001723{
Eric W. Biederman976d02012012-05-23 17:16:53 -06001724 kuid_t uid;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001725
Eric Dumazetf064af12010-09-22 12:43:39 +00001726 read_lock_bh(&sk->sk_callback_lock);
Eric W. Biederman976d02012012-05-23 17:16:53 -06001727 uid = sk->sk_socket ? SOCK_INODE(sk->sk_socket)->i_uid : GLOBAL_ROOT_UID;
Eric Dumazetf064af12010-09-22 12:43:39 +00001728 read_unlock_bh(&sk->sk_callback_lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001729 return uid;
1730}
Eric Dumazet2a915252009-05-27 11:30:05 +00001731EXPORT_SYMBOL(sock_i_uid);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001732
1733unsigned long sock_i_ino(struct sock *sk)
1734{
1735 unsigned long ino;
1736
Eric Dumazetf064af12010-09-22 12:43:39 +00001737 read_lock_bh(&sk->sk_callback_lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001738 ino = sk->sk_socket ? SOCK_INODE(sk->sk_socket)->i_ino : 0;
Eric Dumazetf064af12010-09-22 12:43:39 +00001739 read_unlock_bh(&sk->sk_callback_lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001740 return ino;
1741}
Eric Dumazet2a915252009-05-27 11:30:05 +00001742EXPORT_SYMBOL(sock_i_ino);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001743
1744/*
1745 * Allocate a skb from the socket's send buffer.
1746 */
Victor Fusco86a76ca2005-07-08 14:57:47 -07001747struct sk_buff *sock_wmalloc(struct sock *sk, unsigned long size, int force,
Al Virodd0fc662005-10-07 07:46:04 +01001748 gfp_t priority)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001749{
1750 if (force || atomic_read(&sk->sk_wmem_alloc) < sk->sk_sndbuf) {
Eric Dumazet2a915252009-05-27 11:30:05 +00001751 struct sk_buff *skb = alloc_skb(size, priority);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001752 if (skb) {
1753 skb_set_owner_w(skb, sk);
1754 return skb;
1755 }
1756 }
1757 return NULL;
1758}
Eric Dumazet2a915252009-05-27 11:30:05 +00001759EXPORT_SYMBOL(sock_wmalloc);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001760
1761/*
Linus Torvalds1da177e2005-04-16 15:20:36 -07001762 * Allocate a memory block from the socket's option memory buffer.
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001763 */
Al Virodd0fc662005-10-07 07:46:04 +01001764void *sock_kmalloc(struct sock *sk, int size, gfp_t priority)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001765{
Eric Dumazet95c96172012-04-15 05:58:06 +00001766 if ((unsigned int)size <= sysctl_optmem_max &&
Linus Torvalds1da177e2005-04-16 15:20:36 -07001767 atomic_read(&sk->sk_omem_alloc) + size < sysctl_optmem_max) {
1768 void *mem;
1769 /* First do the add, to avoid the race if kmalloc
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001770 * might sleep.
Linus Torvalds1da177e2005-04-16 15:20:36 -07001771 */
1772 atomic_add(size, &sk->sk_omem_alloc);
1773 mem = kmalloc(size, priority);
1774 if (mem)
1775 return mem;
1776 atomic_sub(size, &sk->sk_omem_alloc);
1777 }
1778 return NULL;
1779}
Eric Dumazet2a915252009-05-27 11:30:05 +00001780EXPORT_SYMBOL(sock_kmalloc);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001781
Daniel Borkmann79e88652014-11-19 17:13:11 +01001782/* Free an option memory block. Note, we actually want the inline
1783 * here as this allows gcc to detect the nullify and fold away the
1784 * condition entirely.
Linus Torvalds1da177e2005-04-16 15:20:36 -07001785 */
Daniel Borkmann79e88652014-11-19 17:13:11 +01001786static inline void __sock_kfree_s(struct sock *sk, void *mem, int size,
1787 const bool nullify)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001788{
David S. Millere53da5f2014-10-14 17:02:37 -04001789 if (WARN_ON_ONCE(!mem))
1790 return;
Daniel Borkmann79e88652014-11-19 17:13:11 +01001791 if (nullify)
1792 kzfree(mem);
1793 else
1794 kfree(mem);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001795 atomic_sub(size, &sk->sk_omem_alloc);
1796}
Daniel Borkmann79e88652014-11-19 17:13:11 +01001797
1798void sock_kfree_s(struct sock *sk, void *mem, int size)
1799{
1800 __sock_kfree_s(sk, mem, size, false);
1801}
Eric Dumazet2a915252009-05-27 11:30:05 +00001802EXPORT_SYMBOL(sock_kfree_s);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001803
Daniel Borkmann79e88652014-11-19 17:13:11 +01001804void sock_kzfree_s(struct sock *sk, void *mem, int size)
1805{
1806 __sock_kfree_s(sk, mem, size, true);
1807}
1808EXPORT_SYMBOL(sock_kzfree_s);
1809
Linus Torvalds1da177e2005-04-16 15:20:36 -07001810/* It is almost wait_for_tcp_memory minus release_sock/lock_sock.
1811 I think, these locks should be removed for datagram sockets.
1812 */
Eric Dumazet2a915252009-05-27 11:30:05 +00001813static long sock_wait_for_wmem(struct sock *sk, long timeo)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001814{
1815 DEFINE_WAIT(wait);
1816
Eric Dumazet9cd3e072015-11-29 20:03:10 -08001817 sk_clear_bit(SOCKWQ_ASYNC_NOSPACE, sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001818 for (;;) {
1819 if (!timeo)
1820 break;
1821 if (signal_pending(current))
1822 break;
1823 set_bit(SOCK_NOSPACE, &sk->sk_socket->flags);
Eric Dumazetaa395142010-04-20 13:03:51 +00001824 prepare_to_wait(sk_sleep(sk), &wait, TASK_INTERRUPTIBLE);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001825 if (atomic_read(&sk->sk_wmem_alloc) < sk->sk_sndbuf)
1826 break;
1827 if (sk->sk_shutdown & SEND_SHUTDOWN)
1828 break;
1829 if (sk->sk_err)
1830 break;
1831 timeo = schedule_timeout(timeo);
1832 }
Eric Dumazetaa395142010-04-20 13:03:51 +00001833 finish_wait(sk_sleep(sk), &wait);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001834 return timeo;
1835}
1836
1837
1838/*
1839 * Generic send/receive buffer handlers
1840 */
1841
Herbert Xu4cc7f682009-02-04 16:55:54 -08001842struct sk_buff *sock_alloc_send_pskb(struct sock *sk, unsigned long header_len,
1843 unsigned long data_len, int noblock,
Eric Dumazet28d64272013-08-08 14:38:47 -07001844 int *errcode, int max_page_order)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001845{
Eric Dumazet2e4e4412014-09-17 04:49:49 -07001846 struct sk_buff *skb;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001847 long timeo;
1848 int err;
1849
Linus Torvalds1da177e2005-04-16 15:20:36 -07001850 timeo = sock_sndtimeo(sk, noblock);
Eric Dumazet2e4e4412014-09-17 04:49:49 -07001851 for (;;) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07001852 err = sock_error(sk);
1853 if (err != 0)
1854 goto failure;
1855
1856 err = -EPIPE;
1857 if (sk->sk_shutdown & SEND_SHUTDOWN)
1858 goto failure;
1859
Eric Dumazet2e4e4412014-09-17 04:49:49 -07001860 if (sk_wmem_alloc_get(sk) < sk->sk_sndbuf)
1861 break;
Eric Dumazet28d64272013-08-08 14:38:47 -07001862
Eric Dumazet9cd3e072015-11-29 20:03:10 -08001863 sk_set_bit(SOCKWQ_ASYNC_NOSPACE, sk);
Eric Dumazet2e4e4412014-09-17 04:49:49 -07001864 set_bit(SOCK_NOSPACE, &sk->sk_socket->flags);
1865 err = -EAGAIN;
1866 if (!timeo)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001867 goto failure;
Eric Dumazet2e4e4412014-09-17 04:49:49 -07001868 if (signal_pending(current))
1869 goto interrupted;
1870 timeo = sock_wait_for_wmem(sk, timeo);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001871 }
Eric Dumazet2e4e4412014-09-17 04:49:49 -07001872 skb = alloc_skb_with_frags(header_len, data_len, max_page_order,
1873 errcode, sk->sk_allocation);
1874 if (skb)
1875 skb_set_owner_w(skb, sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001876 return skb;
1877
1878interrupted:
1879 err = sock_intr_errno(timeo);
1880failure:
1881 *errcode = err;
1882 return NULL;
1883}
Herbert Xu4cc7f682009-02-04 16:55:54 -08001884EXPORT_SYMBOL(sock_alloc_send_pskb);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001885
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001886struct sk_buff *sock_alloc_send_skb(struct sock *sk, unsigned long size,
Linus Torvalds1da177e2005-04-16 15:20:36 -07001887 int noblock, int *errcode)
1888{
Eric Dumazet28d64272013-08-08 14:38:47 -07001889 return sock_alloc_send_pskb(sk, size, 0, noblock, errcode, 0);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001890}
Eric Dumazet2a915252009-05-27 11:30:05 +00001891EXPORT_SYMBOL(sock_alloc_send_skb);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001892
Willem de Bruijn39771b12016-04-02 23:08:06 -04001893int __sock_cmsg_send(struct sock *sk, struct msghdr *msg, struct cmsghdr *cmsg,
1894 struct sockcm_cookie *sockc)
1895{
Soheil Hassas Yeganeh3dd17e62016-04-02 23:08:09 -04001896 u32 tsflags;
1897
Willem de Bruijn39771b12016-04-02 23:08:06 -04001898 switch (cmsg->cmsg_type) {
1899 case SO_MARK:
1900 if (!ns_capable(sock_net(sk)->user_ns, CAP_NET_ADMIN))
1901 return -EPERM;
1902 if (cmsg->cmsg_len != CMSG_LEN(sizeof(u32)))
1903 return -EINVAL;
1904 sockc->mark = *(u32 *)CMSG_DATA(cmsg);
1905 break;
Soheil Hassas Yeganeh3dd17e62016-04-02 23:08:09 -04001906 case SO_TIMESTAMPING:
1907 if (cmsg->cmsg_len != CMSG_LEN(sizeof(u32)))
1908 return -EINVAL;
1909
1910 tsflags = *(u32 *)CMSG_DATA(cmsg);
1911 if (tsflags & ~SOF_TIMESTAMPING_TX_RECORD_MASK)
1912 return -EINVAL;
1913
1914 sockc->tsflags &= ~SOF_TIMESTAMPING_TX_RECORD_MASK;
1915 sockc->tsflags |= tsflags;
1916 break;
Willem de Bruijn39771b12016-04-02 23:08:06 -04001917 default:
1918 return -EINVAL;
1919 }
1920 return 0;
1921}
1922EXPORT_SYMBOL(__sock_cmsg_send);
1923
Edward Jeef28ea362015-10-08 14:56:48 -07001924int sock_cmsg_send(struct sock *sk, struct msghdr *msg,
1925 struct sockcm_cookie *sockc)
1926{
1927 struct cmsghdr *cmsg;
Willem de Bruijn39771b12016-04-02 23:08:06 -04001928 int ret;
Edward Jeef28ea362015-10-08 14:56:48 -07001929
1930 for_each_cmsghdr(cmsg, msg) {
1931 if (!CMSG_OK(msg, cmsg))
1932 return -EINVAL;
1933 if (cmsg->cmsg_level != SOL_SOCKET)
1934 continue;
Willem de Bruijn39771b12016-04-02 23:08:06 -04001935 ret = __sock_cmsg_send(sk, msg, cmsg, sockc);
1936 if (ret)
1937 return ret;
Edward Jeef28ea362015-10-08 14:56:48 -07001938 }
1939 return 0;
1940}
1941EXPORT_SYMBOL(sock_cmsg_send);
1942
Eric Dumazet5640f762012-09-23 23:04:42 +00001943/* On 32bit arches, an skb frag is limited to 2^15 */
1944#define SKB_FRAG_PAGE_ORDER get_order(32768)
1945
Eric Dumazet400dfd32013-10-17 16:27:07 -07001946/**
1947 * skb_page_frag_refill - check that a page_frag contains enough room
1948 * @sz: minimum size of the fragment we want to get
1949 * @pfrag: pointer to page_frag
Eric Dumazet82d5e2b2014-09-08 04:00:00 -07001950 * @gfp: priority for memory allocation
Eric Dumazet400dfd32013-10-17 16:27:07 -07001951 *
1952 * Note: While this allocator tries to use high order pages, there is
1953 * no guarantee that allocations succeed. Therefore, @sz MUST be
1954 * less or equal than PAGE_SIZE.
1955 */
Eric Dumazetd9b29382014-08-27 20:49:34 -07001956bool skb_page_frag_refill(unsigned int sz, struct page_frag *pfrag, gfp_t gfp)
Eric Dumazet5640f762012-09-23 23:04:42 +00001957{
Eric Dumazet5640f762012-09-23 23:04:42 +00001958 if (pfrag->page) {
Joonsoo Kimfe896d12016-03-17 14:19:26 -07001959 if (page_ref_count(pfrag->page) == 1) {
Eric Dumazet5640f762012-09-23 23:04:42 +00001960 pfrag->offset = 0;
1961 return true;
1962 }
Eric Dumazet400dfd32013-10-17 16:27:07 -07001963 if (pfrag->offset + sz <= pfrag->size)
Eric Dumazet5640f762012-09-23 23:04:42 +00001964 return true;
1965 put_page(pfrag->page);
1966 }
1967
Eric Dumazetd9b29382014-08-27 20:49:34 -07001968 pfrag->offset = 0;
1969 if (SKB_FRAG_PAGE_ORDER) {
Mel Gormand0164ad2015-11-06 16:28:21 -08001970 /* Avoid direct reclaim but allow kswapd to wake */
1971 pfrag->page = alloc_pages((gfp & ~__GFP_DIRECT_RECLAIM) |
1972 __GFP_COMP | __GFP_NOWARN |
1973 __GFP_NORETRY,
Eric Dumazetd9b29382014-08-27 20:49:34 -07001974 SKB_FRAG_PAGE_ORDER);
Eric Dumazet5640f762012-09-23 23:04:42 +00001975 if (likely(pfrag->page)) {
Eric Dumazetd9b29382014-08-27 20:49:34 -07001976 pfrag->size = PAGE_SIZE << SKB_FRAG_PAGE_ORDER;
Eric Dumazet5640f762012-09-23 23:04:42 +00001977 return true;
1978 }
Eric Dumazetd9b29382014-08-27 20:49:34 -07001979 }
1980 pfrag->page = alloc_page(gfp);
1981 if (likely(pfrag->page)) {
1982 pfrag->size = PAGE_SIZE;
1983 return true;
1984 }
Eric Dumazet400dfd32013-10-17 16:27:07 -07001985 return false;
1986}
1987EXPORT_SYMBOL(skb_page_frag_refill);
1988
1989bool sk_page_frag_refill(struct sock *sk, struct page_frag *pfrag)
1990{
1991 if (likely(skb_page_frag_refill(32U, pfrag, sk->sk_allocation)))
1992 return true;
1993
Eric Dumazet5640f762012-09-23 23:04:42 +00001994 sk_enter_memory_pressure(sk);
1995 sk_stream_moderate_sndbuf(sk);
1996 return false;
1997}
1998EXPORT_SYMBOL(sk_page_frag_refill);
1999
Linus Torvalds1da177e2005-04-16 15:20:36 -07002000static void __lock_sock(struct sock *sk)
Namhyung Kimf39234d2010-09-08 03:48:48 +00002001 __releases(&sk->sk_lock.slock)
2002 __acquires(&sk->sk_lock.slock)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002003{
2004 DEFINE_WAIT(wait);
2005
Stephen Hemmingere71a4782007-04-10 20:10:33 -07002006 for (;;) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07002007 prepare_to_wait_exclusive(&sk->sk_lock.wq, &wait,
2008 TASK_UNINTERRUPTIBLE);
2009 spin_unlock_bh(&sk->sk_lock.slock);
2010 schedule();
2011 spin_lock_bh(&sk->sk_lock.slock);
Stephen Hemmingere71a4782007-04-10 20:10:33 -07002012 if (!sock_owned_by_user(sk))
Linus Torvalds1da177e2005-04-16 15:20:36 -07002013 break;
2014 }
2015 finish_wait(&sk->sk_lock.wq, &wait);
2016}
2017
2018static void __release_sock(struct sock *sk)
Namhyung Kimf39234d2010-09-08 03:48:48 +00002019 __releases(&sk->sk_lock.slock)
2020 __acquires(&sk->sk_lock.slock)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002021{
2022 struct sk_buff *skb = sk->sk_backlog.head;
2023
2024 do {
2025 sk->sk_backlog.head = sk->sk_backlog.tail = NULL;
2026 bh_unlock_sock(sk);
2027
2028 do {
2029 struct sk_buff *next = skb->next;
2030
Eric Dumazete4cbb022012-04-30 16:07:09 +00002031 prefetch(next);
Eric Dumazet7fee2262010-05-11 23:19:48 +00002032 WARN_ON_ONCE(skb_dst_is_noref(skb));
Linus Torvalds1da177e2005-04-16 15:20:36 -07002033 skb->next = NULL;
Peter Zijlstrac57943a2008-10-07 14:18:42 -07002034 sk_backlog_rcv(sk, skb);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002035
2036 /*
2037 * We are in process context here with softirqs
2038 * disabled, use cond_resched_softirq() to preempt.
2039 * This is safe to do because we've taken the backlog
2040 * queue private:
2041 */
2042 cond_resched_softirq();
2043
2044 skb = next;
2045 } while (skb != NULL);
2046
2047 bh_lock_sock(sk);
Stephen Hemmingere71a4782007-04-10 20:10:33 -07002048 } while ((skb = sk->sk_backlog.head) != NULL);
Zhu Yi8eae9392010-03-04 18:01:40 +00002049
2050 /*
2051 * Doing the zeroing here guarantee we can not loop forever
2052 * while a wild producer attempts to flood us.
2053 */
2054 sk->sk_backlog.len = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002055}
2056
2057/**
2058 * sk_wait_data - wait for data to arrive at sk_receive_queue
Pavel Pisa4dc3b162005-05-01 08:59:25 -07002059 * @sk: sock to wait on
2060 * @timeo: for how long
Sabrina Dubrocadfbafc92015-07-24 18:19:25 +02002061 * @skb: last skb seen on sk_receive_queue
Linus Torvalds1da177e2005-04-16 15:20:36 -07002062 *
2063 * Now socket state including sk->sk_err is changed only under lock,
2064 * hence we may omit checks after joining wait queue.
2065 * We check receive queue before schedule() only as optimization;
2066 * it is very likely that release_sock() added new data.
2067 */
Sabrina Dubrocadfbafc92015-07-24 18:19:25 +02002068int sk_wait_data(struct sock *sk, long *timeo, const struct sk_buff *skb)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002069{
2070 int rc;
2071 DEFINE_WAIT(wait);
2072
Eric Dumazetaa395142010-04-20 13:03:51 +00002073 prepare_to_wait(sk_sleep(sk), &wait, TASK_INTERRUPTIBLE);
Eric Dumazet9cd3e072015-11-29 20:03:10 -08002074 sk_set_bit(SOCKWQ_ASYNC_WAITDATA, sk);
Sabrina Dubrocadfbafc92015-07-24 18:19:25 +02002075 rc = sk_wait_event(sk, timeo, skb_peek_tail(&sk->sk_receive_queue) != skb);
Eric Dumazet9cd3e072015-11-29 20:03:10 -08002076 sk_clear_bit(SOCKWQ_ASYNC_WAITDATA, sk);
Eric Dumazetaa395142010-04-20 13:03:51 +00002077 finish_wait(sk_sleep(sk), &wait);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002078 return rc;
2079}
Linus Torvalds1da177e2005-04-16 15:20:36 -07002080EXPORT_SYMBOL(sk_wait_data);
2081
Hideo Aoki3ab224b2007-12-31 00:11:19 -08002082/**
2083 * __sk_mem_schedule - increase sk_forward_alloc and memory_allocated
2084 * @sk: socket
2085 * @size: memory size to allocate
2086 * @kind: allocation type
2087 *
2088 * If kind is SK_MEM_SEND, it means wmem allocation. Otherwise it means
2089 * rmem allocation. This function assumes that protocols which have
2090 * memory_pressure use sk_wmem_queued as write buffer accounting.
2091 */
2092int __sk_mem_schedule(struct sock *sk, int size, int kind)
2093{
2094 struct proto *prot = sk->sk_prot;
2095 int amt = sk_mem_pages(size);
Eric Dumazet8d987e52010-11-09 23:24:26 +00002096 long allocated;
Hideo Aoki3ab224b2007-12-31 00:11:19 -08002097
2098 sk->sk_forward_alloc += amt * SK_MEM_QUANTUM;
Glauber Costa180d8cd2011-12-11 21:47:02 +00002099
Johannes Weinere8056052016-01-14 15:21:14 -08002100 allocated = sk_memory_allocated_add(sk, amt);
2101
Johannes Weinerbaac50b2016-01-14 15:21:17 -08002102 if (mem_cgroup_sockets_enabled && sk->sk_memcg &&
2103 !mem_cgroup_charge_skmem(sk->sk_memcg, amt))
Johannes Weinere8056052016-01-14 15:21:14 -08002104 goto suppress_allocation;
Hideo Aoki3ab224b2007-12-31 00:11:19 -08002105
2106 /* Under limit. */
Johannes Weinere8056052016-01-14 15:21:14 -08002107 if (allocated <= sk_prot_mem_limits(sk, 0)) {
Glauber Costa180d8cd2011-12-11 21:47:02 +00002108 sk_leave_memory_pressure(sk);
Hideo Aoki3ab224b2007-12-31 00:11:19 -08002109 return 1;
2110 }
2111
Johannes Weinere8056052016-01-14 15:21:14 -08002112 /* Under pressure. */
2113 if (allocated > sk_prot_mem_limits(sk, 1))
Glauber Costa180d8cd2011-12-11 21:47:02 +00002114 sk_enter_memory_pressure(sk);
Hideo Aoki3ab224b2007-12-31 00:11:19 -08002115
Johannes Weinere8056052016-01-14 15:21:14 -08002116 /* Over hard limit. */
2117 if (allocated > sk_prot_mem_limits(sk, 2))
Hideo Aoki3ab224b2007-12-31 00:11:19 -08002118 goto suppress_allocation;
2119
2120 /* guarantee minimum buffer size under pressure */
2121 if (kind == SK_MEM_RECV) {
2122 if (atomic_read(&sk->sk_rmem_alloc) < prot->sysctl_rmem[0])
2123 return 1;
Glauber Costa180d8cd2011-12-11 21:47:02 +00002124
Hideo Aoki3ab224b2007-12-31 00:11:19 -08002125 } else { /* SK_MEM_SEND */
2126 if (sk->sk_type == SOCK_STREAM) {
2127 if (sk->sk_wmem_queued < prot->sysctl_wmem[0])
2128 return 1;
2129 } else if (atomic_read(&sk->sk_wmem_alloc) <
2130 prot->sysctl_wmem[0])
2131 return 1;
2132 }
2133
Glauber Costa180d8cd2011-12-11 21:47:02 +00002134 if (sk_has_memory_pressure(sk)) {
Eric Dumazet17483762008-11-25 21:16:35 -08002135 int alloc;
2136
Glauber Costa180d8cd2011-12-11 21:47:02 +00002137 if (!sk_under_memory_pressure(sk))
Eric Dumazet17483762008-11-25 21:16:35 -08002138 return 1;
Glauber Costa180d8cd2011-12-11 21:47:02 +00002139 alloc = sk_sockets_allocated_read_positive(sk);
2140 if (sk_prot_mem_limits(sk, 2) > alloc *
Hideo Aoki3ab224b2007-12-31 00:11:19 -08002141 sk_mem_pages(sk->sk_wmem_queued +
2142 atomic_read(&sk->sk_rmem_alloc) +
2143 sk->sk_forward_alloc))
2144 return 1;
2145 }
2146
2147suppress_allocation:
2148
2149 if (kind == SK_MEM_SEND && sk->sk_type == SOCK_STREAM) {
2150 sk_stream_moderate_sndbuf(sk);
2151
2152 /* Fail only if socket is _under_ its sndbuf.
2153 * In this case we cannot block, so that we have to fail.
2154 */
2155 if (sk->sk_wmem_queued + size >= sk->sk_sndbuf)
2156 return 1;
2157 }
2158
Satoru Moriya3847ce32011-06-17 12:00:03 +00002159 trace_sock_exceed_buf_limit(sk, prot, allocated);
2160
Hideo Aoki3ab224b2007-12-31 00:11:19 -08002161 /* Alas. Undo changes. */
2162 sk->sk_forward_alloc -= amt * SK_MEM_QUANTUM;
Glauber Costa180d8cd2011-12-11 21:47:02 +00002163
Glauber Costa0e90b312012-01-20 04:57:16 +00002164 sk_memory_allocated_sub(sk, amt);
Glauber Costa180d8cd2011-12-11 21:47:02 +00002165
Johannes Weinerbaac50b2016-01-14 15:21:17 -08002166 if (mem_cgroup_sockets_enabled && sk->sk_memcg)
2167 mem_cgroup_uncharge_skmem(sk->sk_memcg, amt);
Johannes Weinere8056052016-01-14 15:21:14 -08002168
Hideo Aoki3ab224b2007-12-31 00:11:19 -08002169 return 0;
2170}
Hideo Aoki3ab224b2007-12-31 00:11:19 -08002171EXPORT_SYMBOL(__sk_mem_schedule);
2172
2173/**
Jean Sacren69dba9b2015-08-27 18:05:49 -06002174 * __sk_mem_reclaim - reclaim memory_allocated
Hideo Aoki3ab224b2007-12-31 00:11:19 -08002175 * @sk: socket
Eric Dumazet1a24e042015-05-15 12:39:25 -07002176 * @amount: number of bytes (rounded down to a SK_MEM_QUANTUM multiple)
Hideo Aoki3ab224b2007-12-31 00:11:19 -08002177 */
Eric Dumazet1a24e042015-05-15 12:39:25 -07002178void __sk_mem_reclaim(struct sock *sk, int amount)
Hideo Aoki3ab224b2007-12-31 00:11:19 -08002179{
Eric Dumazet1a24e042015-05-15 12:39:25 -07002180 amount >>= SK_MEM_QUANTUM_SHIFT;
2181 sk_memory_allocated_sub(sk, amount);
2182 sk->sk_forward_alloc -= amount << SK_MEM_QUANTUM_SHIFT;
Hideo Aoki3ab224b2007-12-31 00:11:19 -08002183
Johannes Weinerbaac50b2016-01-14 15:21:17 -08002184 if (mem_cgroup_sockets_enabled && sk->sk_memcg)
2185 mem_cgroup_uncharge_skmem(sk->sk_memcg, amount);
Johannes Weinere8056052016-01-14 15:21:14 -08002186
Glauber Costa180d8cd2011-12-11 21:47:02 +00002187 if (sk_under_memory_pressure(sk) &&
2188 (sk_memory_allocated(sk) < sk_prot_mem_limits(sk, 0)))
2189 sk_leave_memory_pressure(sk);
Hideo Aoki3ab224b2007-12-31 00:11:19 -08002190}
Hideo Aoki3ab224b2007-12-31 00:11:19 -08002191EXPORT_SYMBOL(__sk_mem_reclaim);
2192
samanthakumar627d2d62016-04-05 12:41:16 -04002193int sk_set_peek_off(struct sock *sk, int val)
2194{
2195 if (val < 0)
2196 return -EINVAL;
2197
2198 sk->sk_peek_off = val;
2199 return 0;
2200}
2201EXPORT_SYMBOL_GPL(sk_set_peek_off);
Hideo Aoki3ab224b2007-12-31 00:11:19 -08002202
Linus Torvalds1da177e2005-04-16 15:20:36 -07002203/*
2204 * Set of default routines for initialising struct proto_ops when
2205 * the protocol does not support a particular function. In certain
2206 * cases where it makes no sense for a protocol to have a "do nothing"
2207 * function, some default processing is provided.
2208 */
2209
2210int sock_no_bind(struct socket *sock, struct sockaddr *saddr, int len)
2211{
2212 return -EOPNOTSUPP;
2213}
Eric Dumazet2a915252009-05-27 11:30:05 +00002214EXPORT_SYMBOL(sock_no_bind);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002215
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09002216int sock_no_connect(struct socket *sock, struct sockaddr *saddr,
Linus Torvalds1da177e2005-04-16 15:20:36 -07002217 int len, int flags)
2218{
2219 return -EOPNOTSUPP;
2220}
Eric Dumazet2a915252009-05-27 11:30:05 +00002221EXPORT_SYMBOL(sock_no_connect);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002222
2223int sock_no_socketpair(struct socket *sock1, struct socket *sock2)
2224{
2225 return -EOPNOTSUPP;
2226}
Eric Dumazet2a915252009-05-27 11:30:05 +00002227EXPORT_SYMBOL(sock_no_socketpair);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002228
2229int sock_no_accept(struct socket *sock, struct socket *newsock, int flags)
2230{
2231 return -EOPNOTSUPP;
2232}
Eric Dumazet2a915252009-05-27 11:30:05 +00002233EXPORT_SYMBOL(sock_no_accept);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002234
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09002235int sock_no_getname(struct socket *sock, struct sockaddr *saddr,
Linus Torvalds1da177e2005-04-16 15:20:36 -07002236 int *len, int peer)
2237{
2238 return -EOPNOTSUPP;
2239}
Eric Dumazet2a915252009-05-27 11:30:05 +00002240EXPORT_SYMBOL(sock_no_getname);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002241
Eric Dumazet2a915252009-05-27 11:30:05 +00002242unsigned int sock_no_poll(struct file *file, struct socket *sock, poll_table *pt)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002243{
2244 return 0;
2245}
Eric Dumazet2a915252009-05-27 11:30:05 +00002246EXPORT_SYMBOL(sock_no_poll);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002247
2248int sock_no_ioctl(struct socket *sock, unsigned int cmd, unsigned long arg)
2249{
2250 return -EOPNOTSUPP;
2251}
Eric Dumazet2a915252009-05-27 11:30:05 +00002252EXPORT_SYMBOL(sock_no_ioctl);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002253
2254int sock_no_listen(struct socket *sock, int backlog)
2255{
2256 return -EOPNOTSUPP;
2257}
Eric Dumazet2a915252009-05-27 11:30:05 +00002258EXPORT_SYMBOL(sock_no_listen);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002259
2260int sock_no_shutdown(struct socket *sock, int how)
2261{
2262 return -EOPNOTSUPP;
2263}
Eric Dumazet2a915252009-05-27 11:30:05 +00002264EXPORT_SYMBOL(sock_no_shutdown);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002265
2266int sock_no_setsockopt(struct socket *sock, int level, int optname,
David S. Millerb7058842009-09-30 16:12:20 -07002267 char __user *optval, unsigned int optlen)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002268{
2269 return -EOPNOTSUPP;
2270}
Eric Dumazet2a915252009-05-27 11:30:05 +00002271EXPORT_SYMBOL(sock_no_setsockopt);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002272
2273int sock_no_getsockopt(struct socket *sock, int level, int optname,
2274 char __user *optval, int __user *optlen)
2275{
2276 return -EOPNOTSUPP;
2277}
Eric Dumazet2a915252009-05-27 11:30:05 +00002278EXPORT_SYMBOL(sock_no_getsockopt);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002279
Ying Xue1b784142015-03-02 15:37:48 +08002280int sock_no_sendmsg(struct socket *sock, struct msghdr *m, size_t len)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002281{
2282 return -EOPNOTSUPP;
2283}
Eric Dumazet2a915252009-05-27 11:30:05 +00002284EXPORT_SYMBOL(sock_no_sendmsg);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002285
Ying Xue1b784142015-03-02 15:37:48 +08002286int sock_no_recvmsg(struct socket *sock, struct msghdr *m, size_t len,
2287 int flags)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002288{
2289 return -EOPNOTSUPP;
2290}
Eric Dumazet2a915252009-05-27 11:30:05 +00002291EXPORT_SYMBOL(sock_no_recvmsg);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002292
2293int sock_no_mmap(struct file *file, struct socket *sock, struct vm_area_struct *vma)
2294{
2295 /* Mirror missing mmap method error code */
2296 return -ENODEV;
2297}
Eric Dumazet2a915252009-05-27 11:30:05 +00002298EXPORT_SYMBOL(sock_no_mmap);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002299
2300ssize_t sock_no_sendpage(struct socket *sock, struct page *page, int offset, size_t size, int flags)
2301{
2302 ssize_t res;
2303 struct msghdr msg = {.msg_flags = flags};
2304 struct kvec iov;
2305 char *kaddr = kmap(page);
2306 iov.iov_base = kaddr + offset;
2307 iov.iov_len = size;
2308 res = kernel_sendmsg(sock, &msg, &iov, 1, size);
2309 kunmap(page);
2310 return res;
2311}
Eric Dumazet2a915252009-05-27 11:30:05 +00002312EXPORT_SYMBOL(sock_no_sendpage);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002313
2314/*
2315 * Default Socket Callbacks
2316 */
2317
2318static void sock_def_wakeup(struct sock *sk)
2319{
Eric Dumazet43815482010-04-29 11:01:49 +00002320 struct socket_wq *wq;
2321
2322 rcu_read_lock();
2323 wq = rcu_dereference(sk->sk_wq);
Herbert Xu1ce0bf52015-11-26 13:55:39 +08002324 if (skwq_has_sleeper(wq))
Eric Dumazet43815482010-04-29 11:01:49 +00002325 wake_up_interruptible_all(&wq->wait);
2326 rcu_read_unlock();
Linus Torvalds1da177e2005-04-16 15:20:36 -07002327}
2328
2329static void sock_def_error_report(struct sock *sk)
2330{
Eric Dumazet43815482010-04-29 11:01:49 +00002331 struct socket_wq *wq;
2332
2333 rcu_read_lock();
2334 wq = rcu_dereference(sk->sk_wq);
Herbert Xu1ce0bf52015-11-26 13:55:39 +08002335 if (skwq_has_sleeper(wq))
Eric Dumazet43815482010-04-29 11:01:49 +00002336 wake_up_interruptible_poll(&wq->wait, POLLERR);
Pavel Emelyanov8d8ad9d2007-11-26 20:10:50 +08002337 sk_wake_async(sk, SOCK_WAKE_IO, POLL_ERR);
Eric Dumazet43815482010-04-29 11:01:49 +00002338 rcu_read_unlock();
Linus Torvalds1da177e2005-04-16 15:20:36 -07002339}
2340
David S. Miller676d2362014-04-11 16:15:36 -04002341static void sock_def_readable(struct sock *sk)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002342{
Eric Dumazet43815482010-04-29 11:01:49 +00002343 struct socket_wq *wq;
2344
2345 rcu_read_lock();
2346 wq = rcu_dereference(sk->sk_wq);
Herbert Xu1ce0bf52015-11-26 13:55:39 +08002347 if (skwq_has_sleeper(wq))
Eric Dumazet2c6607c2011-01-06 10:54:29 -08002348 wake_up_interruptible_sync_poll(&wq->wait, POLLIN | POLLPRI |
Davide Libenzi37e55402009-03-31 15:24:21 -07002349 POLLRDNORM | POLLRDBAND);
Pavel Emelyanov8d8ad9d2007-11-26 20:10:50 +08002350 sk_wake_async(sk, SOCK_WAKE_WAITD, POLL_IN);
Eric Dumazet43815482010-04-29 11:01:49 +00002351 rcu_read_unlock();
Linus Torvalds1da177e2005-04-16 15:20:36 -07002352}
2353
2354static void sock_def_write_space(struct sock *sk)
2355{
Eric Dumazet43815482010-04-29 11:01:49 +00002356 struct socket_wq *wq;
2357
2358 rcu_read_lock();
Linus Torvalds1da177e2005-04-16 15:20:36 -07002359
2360 /* Do not wake up a writer until he can make "significant"
2361 * progress. --DaveM
2362 */
Stephen Hemmingere71a4782007-04-10 20:10:33 -07002363 if ((atomic_read(&sk->sk_wmem_alloc) << 1) <= sk->sk_sndbuf) {
Eric Dumazet43815482010-04-29 11:01:49 +00002364 wq = rcu_dereference(sk->sk_wq);
Herbert Xu1ce0bf52015-11-26 13:55:39 +08002365 if (skwq_has_sleeper(wq))
Eric Dumazet43815482010-04-29 11:01:49 +00002366 wake_up_interruptible_sync_poll(&wq->wait, POLLOUT |
Davide Libenzi37e55402009-03-31 15:24:21 -07002367 POLLWRNORM | POLLWRBAND);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002368
2369 /* Should agree with poll, otherwise some programs break */
2370 if (sock_writeable(sk))
Pavel Emelyanov8d8ad9d2007-11-26 20:10:50 +08002371 sk_wake_async(sk, SOCK_WAKE_SPACE, POLL_OUT);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002372 }
2373
Eric Dumazet43815482010-04-29 11:01:49 +00002374 rcu_read_unlock();
Linus Torvalds1da177e2005-04-16 15:20:36 -07002375}
2376
2377static void sock_def_destruct(struct sock *sk)
2378{
Linus Torvalds1da177e2005-04-16 15:20:36 -07002379}
2380
2381void sk_send_sigurg(struct sock *sk)
2382{
2383 if (sk->sk_socket && sk->sk_socket->file)
2384 if (send_sigurg(&sk->sk_socket->file->f_owner))
Pavel Emelyanov8d8ad9d2007-11-26 20:10:50 +08002385 sk_wake_async(sk, SOCK_WAKE_URG, POLL_PRI);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002386}
Eric Dumazet2a915252009-05-27 11:30:05 +00002387EXPORT_SYMBOL(sk_send_sigurg);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002388
2389void sk_reset_timer(struct sock *sk, struct timer_list* timer,
2390 unsigned long expires)
2391{
2392 if (!mod_timer(timer, expires))
2393 sock_hold(sk);
2394}
Linus Torvalds1da177e2005-04-16 15:20:36 -07002395EXPORT_SYMBOL(sk_reset_timer);
2396
2397void sk_stop_timer(struct sock *sk, struct timer_list* timer)
2398{
Ying Xue25cc4ae2013-02-03 20:32:57 +00002399 if (del_timer(timer))
Linus Torvalds1da177e2005-04-16 15:20:36 -07002400 __sock_put(sk);
2401}
Linus Torvalds1da177e2005-04-16 15:20:36 -07002402EXPORT_SYMBOL(sk_stop_timer);
2403
2404void sock_init_data(struct socket *sock, struct sock *sk)
2405{
2406 skb_queue_head_init(&sk->sk_receive_queue);
2407 skb_queue_head_init(&sk->sk_write_queue);
2408 skb_queue_head_init(&sk->sk_error_queue);
2409
2410 sk->sk_send_head = NULL;
2411
2412 init_timer(&sk->sk_timer);
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09002413
Linus Torvalds1da177e2005-04-16 15:20:36 -07002414 sk->sk_allocation = GFP_KERNEL;
2415 sk->sk_rcvbuf = sysctl_rmem_default;
2416 sk->sk_sndbuf = sysctl_wmem_default;
2417 sk->sk_state = TCP_CLOSE;
David S. Miller972692e2008-06-17 22:41:38 -07002418 sk_set_socket(sk, sock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002419
2420 sock_set_flag(sk, SOCK_ZAPPED);
2421
Stephen Hemmingere71a4782007-04-10 20:10:33 -07002422 if (sock) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07002423 sk->sk_type = sock->type;
Eric Dumazet43815482010-04-29 11:01:49 +00002424 sk->sk_wq = sock->wq;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002425 sock->sk = sk;
2426 } else
Eric Dumazet43815482010-04-29 11:01:49 +00002427 sk->sk_wq = NULL;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002428
Linus Torvalds1da177e2005-04-16 15:20:36 -07002429 rwlock_init(&sk->sk_callback_lock);
Peter Zijlstra443aef0e2007-07-19 01:49:00 -07002430 lockdep_set_class_and_name(&sk->sk_callback_lock,
2431 af_callback_keys + sk->sk_family,
2432 af_family_clock_key_strings[sk->sk_family]);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002433
2434 sk->sk_state_change = sock_def_wakeup;
2435 sk->sk_data_ready = sock_def_readable;
2436 sk->sk_write_space = sock_def_write_space;
2437 sk->sk_error_report = sock_def_error_report;
2438 sk->sk_destruct = sock_def_destruct;
2439
Eric Dumazet5640f762012-09-23 23:04:42 +00002440 sk->sk_frag.page = NULL;
2441 sk->sk_frag.offset = 0;
Pavel Emelyanovef64a542012-02-21 07:31:34 +00002442 sk->sk_peek_off = -1;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002443
Eric W. Biederman109f6e32010-06-13 03:30:14 +00002444 sk->sk_peer_pid = NULL;
2445 sk->sk_peer_cred = NULL;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002446 sk->sk_write_pending = 0;
2447 sk->sk_rcvlowat = 1;
2448 sk->sk_rcvtimeo = MAX_SCHEDULE_TIMEOUT;
2449 sk->sk_sndtimeo = MAX_SCHEDULE_TIMEOUT;
2450
Eric Dumazetf37f0af2008-04-13 21:39:26 -07002451 sk->sk_stamp = ktime_set(-1L, 0);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002452
Cong Wange0d10952013-08-01 11:10:25 +08002453#ifdef CONFIG_NET_RX_BUSY_POLL
Eliezer Tamir06021292013-06-10 11:39:50 +03002454 sk->sk_napi_id = 0;
Eliezer Tamir64b0dc52013-07-10 17:13:36 +03002455 sk->sk_ll_usec = sysctl_net_busy_read;
Eliezer Tamir06021292013-06-10 11:39:50 +03002456#endif
2457
Eric Dumazet62748f32013-09-24 08:20:52 -07002458 sk->sk_max_pacing_rate = ~0U;
Eric Dumazet7eec4172013-10-08 15:16:00 -07002459 sk->sk_pacing_rate = ~0U;
Eric Dumazet70da2682015-10-08 19:33:21 -07002460 sk->sk_incoming_cpu = -1;
Eric Dumazet4dc6dc72009-07-15 23:13:10 +00002461 /*
2462 * Before updating sk_refcnt, we must commit prior changes to memory
2463 * (Documentation/RCU/rculist_nulls.txt for details)
2464 */
2465 smp_wmb();
Linus Torvalds1da177e2005-04-16 15:20:36 -07002466 atomic_set(&sk->sk_refcnt, 1);
Wang Chen33c732c2007-11-13 20:30:01 -08002467 atomic_set(&sk->sk_drops, 0);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002468}
Eric Dumazet2a915252009-05-27 11:30:05 +00002469EXPORT_SYMBOL(sock_init_data);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002470
Harvey Harrisonb5606c22008-02-13 15:03:16 -08002471void lock_sock_nested(struct sock *sk, int subclass)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002472{
2473 might_sleep();
Ingo Molnara5b5bb92006-07-03 00:25:35 -07002474 spin_lock_bh(&sk->sk_lock.slock);
John Heffnerd2e91172007-09-12 10:44:19 +02002475 if (sk->sk_lock.owned)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002476 __lock_sock(sk);
John Heffnerd2e91172007-09-12 10:44:19 +02002477 sk->sk_lock.owned = 1;
Ingo Molnara5b5bb92006-07-03 00:25:35 -07002478 spin_unlock(&sk->sk_lock.slock);
2479 /*
2480 * The sk_lock has mutex_lock() semantics here:
2481 */
Peter Zijlstrafcc70d52006-11-08 22:44:35 -08002482 mutex_acquire(&sk->sk_lock.dep_map, subclass, 0, _RET_IP_);
Ingo Molnara5b5bb92006-07-03 00:25:35 -07002483 local_bh_enable();
Linus Torvalds1da177e2005-04-16 15:20:36 -07002484}
Peter Zijlstrafcc70d52006-11-08 22:44:35 -08002485EXPORT_SYMBOL(lock_sock_nested);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002486
Harvey Harrisonb5606c22008-02-13 15:03:16 -08002487void release_sock(struct sock *sk)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002488{
Ingo Molnara5b5bb92006-07-03 00:25:35 -07002489 spin_lock_bh(&sk->sk_lock.slock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002490 if (sk->sk_backlog.tail)
2491 __release_sock(sk);
Eric Dumazet46d3cea2012-07-11 05:50:31 +00002492
Eric Dumazetc3f9b012014-03-10 09:50:11 -07002493 /* Warning : release_cb() might need to release sk ownership,
2494 * ie call sock_release_ownership(sk) before us.
2495 */
Eric Dumazet46d3cea2012-07-11 05:50:31 +00002496 if (sk->sk_prot->release_cb)
2497 sk->sk_prot->release_cb(sk);
2498
Eric Dumazetc3f9b012014-03-10 09:50:11 -07002499 sock_release_ownership(sk);
Ingo Molnara5b5bb92006-07-03 00:25:35 -07002500 if (waitqueue_active(&sk->sk_lock.wq))
2501 wake_up(&sk->sk_lock.wq);
2502 spin_unlock_bh(&sk->sk_lock.slock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002503}
2504EXPORT_SYMBOL(release_sock);
2505
Eric Dumazet8a74ad62010-05-26 19:20:18 +00002506/**
2507 * lock_sock_fast - fast version of lock_sock
2508 * @sk: socket
2509 *
2510 * This version should be used for very small section, where process wont block
2511 * return false if fast path is taken
2512 * sk_lock.slock locked, owned = 0, BH disabled
2513 * return true if slow path is taken
2514 * sk_lock.slock unlocked, owned = 1, BH enabled
2515 */
2516bool lock_sock_fast(struct sock *sk)
2517{
2518 might_sleep();
2519 spin_lock_bh(&sk->sk_lock.slock);
2520
2521 if (!sk->sk_lock.owned)
2522 /*
2523 * Note : We must disable BH
2524 */
2525 return false;
2526
2527 __lock_sock(sk);
2528 sk->sk_lock.owned = 1;
2529 spin_unlock(&sk->sk_lock.slock);
2530 /*
2531 * The sk_lock has mutex_lock() semantics here:
2532 */
2533 mutex_acquire(&sk->sk_lock.dep_map, 0, 0, _RET_IP_);
2534 local_bh_enable();
2535 return true;
2536}
2537EXPORT_SYMBOL(lock_sock_fast);
2538
Linus Torvalds1da177e2005-04-16 15:20:36 -07002539int sock_get_timestamp(struct sock *sk, struct timeval __user *userstamp)
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09002540{
Eric Dumazetb7aa0bf2007-04-19 16:16:32 -07002541 struct timeval tv;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002542 if (!sock_flag(sk, SOCK_TIMESTAMP))
Patrick Ohly20d49472009-02-12 05:03:38 +00002543 sock_enable_timestamp(sk, SOCK_TIMESTAMP);
Eric Dumazetb7aa0bf2007-04-19 16:16:32 -07002544 tv = ktime_to_timeval(sk->sk_stamp);
2545 if (tv.tv_sec == -1)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002546 return -ENOENT;
Eric Dumazetb7aa0bf2007-04-19 16:16:32 -07002547 if (tv.tv_sec == 0) {
2548 sk->sk_stamp = ktime_get_real();
2549 tv = ktime_to_timeval(sk->sk_stamp);
2550 }
2551 return copy_to_user(userstamp, &tv, sizeof(tv)) ? -EFAULT : 0;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09002552}
Linus Torvalds1da177e2005-04-16 15:20:36 -07002553EXPORT_SYMBOL(sock_get_timestamp);
2554
Eric Dumazetae40eb12007-03-18 17:33:16 -07002555int sock_get_timestampns(struct sock *sk, struct timespec __user *userstamp)
2556{
2557 struct timespec ts;
2558 if (!sock_flag(sk, SOCK_TIMESTAMP))
Patrick Ohly20d49472009-02-12 05:03:38 +00002559 sock_enable_timestamp(sk, SOCK_TIMESTAMP);
Eric Dumazetae40eb12007-03-18 17:33:16 -07002560 ts = ktime_to_timespec(sk->sk_stamp);
2561 if (ts.tv_sec == -1)
2562 return -ENOENT;
2563 if (ts.tv_sec == 0) {
2564 sk->sk_stamp = ktime_get_real();
2565 ts = ktime_to_timespec(sk->sk_stamp);
2566 }
2567 return copy_to_user(userstamp, &ts, sizeof(ts)) ? -EFAULT : 0;
2568}
2569EXPORT_SYMBOL(sock_get_timestampns);
2570
Patrick Ohly20d49472009-02-12 05:03:38 +00002571void sock_enable_timestamp(struct sock *sk, int flag)
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09002572{
Patrick Ohly20d49472009-02-12 05:03:38 +00002573 if (!sock_flag(sk, flag)) {
Eric Dumazet08e29af2011-11-28 12:04:18 +00002574 unsigned long previous_flags = sk->sk_flags;
2575
Patrick Ohly20d49472009-02-12 05:03:38 +00002576 sock_set_flag(sk, flag);
2577 /*
2578 * we just set one of the two flags which require net
2579 * time stamping, but time stamping might have been on
2580 * already because of the other one
2581 */
Hannes Frederic Sowa080a2702015-10-26 13:51:37 +01002582 if (sock_needs_netstamp(sk) &&
2583 !(previous_flags & SK_FLAGS_TIMESTAMP))
Patrick Ohly20d49472009-02-12 05:03:38 +00002584 net_enable_timestamp();
Linus Torvalds1da177e2005-04-16 15:20:36 -07002585 }
2586}
Linus Torvalds1da177e2005-04-16 15:20:36 -07002587
Richard Cochrancb820f82013-07-19 19:40:09 +02002588int sock_recv_errqueue(struct sock *sk, struct msghdr *msg, int len,
2589 int level, int type)
2590{
2591 struct sock_exterr_skb *serr;
Willem de Bruijn364a9e92014-08-31 21:30:27 -04002592 struct sk_buff *skb;
Richard Cochrancb820f82013-07-19 19:40:09 +02002593 int copied, err;
2594
2595 err = -EAGAIN;
Willem de Bruijn364a9e92014-08-31 21:30:27 -04002596 skb = sock_dequeue_err_skb(sk);
Richard Cochrancb820f82013-07-19 19:40:09 +02002597 if (skb == NULL)
2598 goto out;
2599
2600 copied = skb->len;
2601 if (copied > len) {
2602 msg->msg_flags |= MSG_TRUNC;
2603 copied = len;
2604 }
David S. Miller51f3d022014-11-05 16:46:40 -05002605 err = skb_copy_datagram_msg(skb, 0, msg, copied);
Richard Cochrancb820f82013-07-19 19:40:09 +02002606 if (err)
2607 goto out_free_skb;
2608
2609 sock_recv_timestamp(msg, sk, skb);
2610
2611 serr = SKB_EXT_ERR(skb);
2612 put_cmsg(msg, level, type, sizeof(serr->ee), &serr->ee);
2613
2614 msg->msg_flags |= MSG_ERRQUEUE;
2615 err = copied;
2616
Richard Cochrancb820f82013-07-19 19:40:09 +02002617out_free_skb:
2618 kfree_skb(skb);
2619out:
2620 return err;
2621}
2622EXPORT_SYMBOL(sock_recv_errqueue);
2623
Linus Torvalds1da177e2005-04-16 15:20:36 -07002624/*
2625 * Get a socket option on an socket.
2626 *
2627 * FIX: POSIX 1003.1g is very ambiguous here. It states that
2628 * asynchronous errors should be reported by getsockopt. We assume
2629 * this means if you specify SO_ERROR (otherwise whats the point of it).
2630 */
2631int sock_common_getsockopt(struct socket *sock, int level, int optname,
2632 char __user *optval, int __user *optlen)
2633{
2634 struct sock *sk = sock->sk;
2635
2636 return sk->sk_prot->getsockopt(sk, level, optname, optval, optlen);
2637}
Linus Torvalds1da177e2005-04-16 15:20:36 -07002638EXPORT_SYMBOL(sock_common_getsockopt);
2639
Dmitry Mishin3fdadf72006-03-20 22:45:21 -08002640#ifdef CONFIG_COMPAT
Arnaldo Carvalho de Melo543d9cf2006-03-20 22:48:35 -08002641int compat_sock_common_getsockopt(struct socket *sock, int level, int optname,
2642 char __user *optval, int __user *optlen)
Dmitry Mishin3fdadf72006-03-20 22:45:21 -08002643{
2644 struct sock *sk = sock->sk;
2645
Johannes Berg1e51f952007-03-06 13:44:06 -08002646 if (sk->sk_prot->compat_getsockopt != NULL)
Arnaldo Carvalho de Melo543d9cf2006-03-20 22:48:35 -08002647 return sk->sk_prot->compat_getsockopt(sk, level, optname,
2648 optval, optlen);
Dmitry Mishin3fdadf72006-03-20 22:45:21 -08002649 return sk->sk_prot->getsockopt(sk, level, optname, optval, optlen);
2650}
2651EXPORT_SYMBOL(compat_sock_common_getsockopt);
2652#endif
2653
Ying Xue1b784142015-03-02 15:37:48 +08002654int sock_common_recvmsg(struct socket *sock, struct msghdr *msg, size_t size,
2655 int flags)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002656{
2657 struct sock *sk = sock->sk;
2658 int addr_len = 0;
2659 int err;
2660
Ying Xue1b784142015-03-02 15:37:48 +08002661 err = sk->sk_prot->recvmsg(sk, msg, size, flags & MSG_DONTWAIT,
Linus Torvalds1da177e2005-04-16 15:20:36 -07002662 flags & ~MSG_DONTWAIT, &addr_len);
2663 if (err >= 0)
2664 msg->msg_namelen = addr_len;
2665 return err;
2666}
Linus Torvalds1da177e2005-04-16 15:20:36 -07002667EXPORT_SYMBOL(sock_common_recvmsg);
2668
2669/*
2670 * Set socket options on an inet socket.
2671 */
2672int sock_common_setsockopt(struct socket *sock, int level, int optname,
David S. Millerb7058842009-09-30 16:12:20 -07002673 char __user *optval, unsigned int optlen)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002674{
2675 struct sock *sk = sock->sk;
2676
2677 return sk->sk_prot->setsockopt(sk, level, optname, optval, optlen);
2678}
Linus Torvalds1da177e2005-04-16 15:20:36 -07002679EXPORT_SYMBOL(sock_common_setsockopt);
2680
Dmitry Mishin3fdadf72006-03-20 22:45:21 -08002681#ifdef CONFIG_COMPAT
Arnaldo Carvalho de Melo543d9cf2006-03-20 22:48:35 -08002682int compat_sock_common_setsockopt(struct socket *sock, int level, int optname,
David S. Millerb7058842009-09-30 16:12:20 -07002683 char __user *optval, unsigned int optlen)
Dmitry Mishin3fdadf72006-03-20 22:45:21 -08002684{
2685 struct sock *sk = sock->sk;
2686
Arnaldo Carvalho de Melo543d9cf2006-03-20 22:48:35 -08002687 if (sk->sk_prot->compat_setsockopt != NULL)
2688 return sk->sk_prot->compat_setsockopt(sk, level, optname,
2689 optval, optlen);
Dmitry Mishin3fdadf72006-03-20 22:45:21 -08002690 return sk->sk_prot->setsockopt(sk, level, optname, optval, optlen);
2691}
2692EXPORT_SYMBOL(compat_sock_common_setsockopt);
2693#endif
2694
Linus Torvalds1da177e2005-04-16 15:20:36 -07002695void sk_common_release(struct sock *sk)
2696{
2697 if (sk->sk_prot->destroy)
2698 sk->sk_prot->destroy(sk);
2699
2700 /*
2701 * Observation: when sock_common_release is called, processes have
2702 * no access to socket. But net still has.
2703 * Step one, detach it from networking:
2704 *
2705 * A. Remove from hash tables.
2706 */
2707
2708 sk->sk_prot->unhash(sk);
2709
2710 /*
2711 * In this point socket cannot receive new packets, but it is possible
2712 * that some packets are in flight because some CPU runs receiver and
2713 * did hash table lookup before we unhashed socket. They will achieve
2714 * receive queue and will be purged by socket destructor.
2715 *
2716 * Also we still have packets pending on receive queue and probably,
2717 * our own packets waiting in device queues. sock_destroy will drain
2718 * receive queue, but transmitted packets will delay socket destruction
2719 * until the last reference will be released.
2720 */
2721
2722 sock_orphan(sk);
2723
2724 xfrm_sk_free_policy(sk);
2725
Arnaldo Carvalho de Meloe6848972005-08-09 19:45:38 -07002726 sk_refcnt_debug_release(sk);
Eric Dumazet5640f762012-09-23 23:04:42 +00002727
2728 if (sk->sk_frag.page) {
2729 put_page(sk->sk_frag.page);
2730 sk->sk_frag.page = NULL;
2731 }
2732
Linus Torvalds1da177e2005-04-16 15:20:36 -07002733 sock_put(sk);
2734}
Linus Torvalds1da177e2005-04-16 15:20:36 -07002735EXPORT_SYMBOL(sk_common_release);
2736
Pavel Emelyanov13ff3d62008-03-28 16:38:17 -07002737#ifdef CONFIG_PROC_FS
2738#define PROTO_INUSE_NR 64 /* should be enough for the first time */
Pavel Emelyanov1338d462008-03-28 16:38:43 -07002739struct prot_inuse {
2740 int val[PROTO_INUSE_NR];
2741};
Pavel Emelyanov13ff3d62008-03-28 16:38:17 -07002742
2743static DECLARE_BITMAP(proto_inuse_idx, PROTO_INUSE_NR);
Pavel Emelyanov70ee1152008-03-31 19:42:16 -07002744
2745#ifdef CONFIG_NET_NS
2746void sock_prot_inuse_add(struct net *net, struct proto *prot, int val)
2747{
Eric Dumazetd6d9ca02010-07-19 10:48:49 +00002748 __this_cpu_add(net->core.inuse->val[prot->inuse_idx], val);
Pavel Emelyanov70ee1152008-03-31 19:42:16 -07002749}
2750EXPORT_SYMBOL_GPL(sock_prot_inuse_add);
2751
2752int sock_prot_inuse_get(struct net *net, struct proto *prot)
2753{
2754 int cpu, idx = prot->inuse_idx;
2755 int res = 0;
2756
2757 for_each_possible_cpu(cpu)
2758 res += per_cpu_ptr(net->core.inuse, cpu)->val[idx];
2759
2760 return res >= 0 ? res : 0;
2761}
2762EXPORT_SYMBOL_GPL(sock_prot_inuse_get);
2763
Alexey Dobriyan2c8c1e72010-01-17 03:35:32 +00002764static int __net_init sock_inuse_init_net(struct net *net)
Pavel Emelyanov70ee1152008-03-31 19:42:16 -07002765{
2766 net->core.inuse = alloc_percpu(struct prot_inuse);
2767 return net->core.inuse ? 0 : -ENOMEM;
2768}
2769
Alexey Dobriyan2c8c1e72010-01-17 03:35:32 +00002770static void __net_exit sock_inuse_exit_net(struct net *net)
Pavel Emelyanov70ee1152008-03-31 19:42:16 -07002771{
2772 free_percpu(net->core.inuse);
2773}
2774
2775static struct pernet_operations net_inuse_ops = {
2776 .init = sock_inuse_init_net,
2777 .exit = sock_inuse_exit_net,
2778};
2779
2780static __init int net_inuse_init(void)
2781{
2782 if (register_pernet_subsys(&net_inuse_ops))
2783 panic("Cannot initialize net inuse counters");
2784
2785 return 0;
2786}
2787
2788core_initcall(net_inuse_init);
2789#else
Pavel Emelyanov1338d462008-03-28 16:38:43 -07002790static DEFINE_PER_CPU(struct prot_inuse, prot_inuse);
2791
Pavel Emelyanovc29a0bc2008-03-31 19:41:46 -07002792void sock_prot_inuse_add(struct net *net, struct proto *prot, int val)
Pavel Emelyanov1338d462008-03-28 16:38:43 -07002793{
Eric Dumazetd6d9ca02010-07-19 10:48:49 +00002794 __this_cpu_add(prot_inuse.val[prot->inuse_idx], val);
Pavel Emelyanov1338d462008-03-28 16:38:43 -07002795}
2796EXPORT_SYMBOL_GPL(sock_prot_inuse_add);
2797
Pavel Emelyanovc29a0bc2008-03-31 19:41:46 -07002798int sock_prot_inuse_get(struct net *net, struct proto *prot)
Pavel Emelyanov1338d462008-03-28 16:38:43 -07002799{
2800 int cpu, idx = prot->inuse_idx;
2801 int res = 0;
2802
2803 for_each_possible_cpu(cpu)
2804 res += per_cpu(prot_inuse, cpu).val[idx];
2805
2806 return res >= 0 ? res : 0;
2807}
2808EXPORT_SYMBOL_GPL(sock_prot_inuse_get);
Pavel Emelyanov70ee1152008-03-31 19:42:16 -07002809#endif
Pavel Emelyanov13ff3d62008-03-28 16:38:17 -07002810
2811static void assign_proto_idx(struct proto *prot)
2812{
2813 prot->inuse_idx = find_first_zero_bit(proto_inuse_idx, PROTO_INUSE_NR);
2814
2815 if (unlikely(prot->inuse_idx == PROTO_INUSE_NR - 1)) {
Joe Perchese005d192012-05-16 19:58:40 +00002816 pr_err("PROTO_INUSE_NR exhausted\n");
Pavel Emelyanov13ff3d62008-03-28 16:38:17 -07002817 return;
2818 }
2819
2820 set_bit(prot->inuse_idx, proto_inuse_idx);
2821}
2822
2823static void release_proto_idx(struct proto *prot)
2824{
2825 if (prot->inuse_idx != PROTO_INUSE_NR - 1)
2826 clear_bit(prot->inuse_idx, proto_inuse_idx);
2827}
2828#else
2829static inline void assign_proto_idx(struct proto *prot)
2830{
2831}
2832
2833static inline void release_proto_idx(struct proto *prot)
2834{
2835}
2836#endif
2837
Eric Dumazet0159dfd2015-03-12 16:44:07 -07002838static void req_prot_cleanup(struct request_sock_ops *rsk_prot)
2839{
2840 if (!rsk_prot)
2841 return;
2842 kfree(rsk_prot->slab_name);
2843 rsk_prot->slab_name = NULL;
Julia Lawalladf78ed2015-09-13 14:15:18 +02002844 kmem_cache_destroy(rsk_prot->slab);
2845 rsk_prot->slab = NULL;
Eric Dumazet0159dfd2015-03-12 16:44:07 -07002846}
2847
2848static int req_prot_init(const struct proto *prot)
2849{
2850 struct request_sock_ops *rsk_prot = prot->rsk_prot;
2851
2852 if (!rsk_prot)
2853 return 0;
2854
2855 rsk_prot->slab_name = kasprintf(GFP_KERNEL, "request_sock_%s",
2856 prot->name);
2857 if (!rsk_prot->slab_name)
2858 return -ENOMEM;
2859
2860 rsk_prot->slab = kmem_cache_create(rsk_prot->slab_name,
2861 rsk_prot->obj_size, 0,
Eric Dumazete96f78a2015-10-03 06:27:28 -07002862 prot->slab_flags, NULL);
Eric Dumazet0159dfd2015-03-12 16:44:07 -07002863
2864 if (!rsk_prot->slab) {
2865 pr_crit("%s: Can't create request sock SLAB cache!\n",
2866 prot->name);
2867 return -ENOMEM;
2868 }
2869 return 0;
2870}
2871
Linus Torvalds1da177e2005-04-16 15:20:36 -07002872int proto_register(struct proto *prot, int alloc_slab)
2873{
Linus Torvalds1da177e2005-04-16 15:20:36 -07002874 if (alloc_slab) {
2875 prot->slab = kmem_cache_create(prot->name, prot->obj_size, 0,
Eric Dumazet271b72c2008-10-29 02:11:14 -07002876 SLAB_HWCACHE_ALIGN | prot->slab_flags,
2877 NULL);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002878
2879 if (prot->slab == NULL) {
Joe Perchese005d192012-05-16 19:58:40 +00002880 pr_crit("%s: Can't create sock SLAB cache!\n",
2881 prot->name);
Pavel Emelyanov60e76632008-03-28 16:39:10 -07002882 goto out;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002883 }
Arnaldo Carvalho de Melo2e6599c2005-06-18 22:46:52 -07002884
Eric Dumazet0159dfd2015-03-12 16:44:07 -07002885 if (req_prot_init(prot))
2886 goto out_free_request_sock_slab;
Arnaldo Carvalho de Melo8feaf0c02005-08-09 20:09:30 -07002887
Arnaldo Carvalho de Melo6d6ee432005-12-13 23:25:19 -08002888 if (prot->twsk_prot != NULL) {
Alexey Dobriyanfaf23422010-02-17 09:34:12 +00002889 prot->twsk_prot->twsk_slab_name = kasprintf(GFP_KERNEL, "tw_sock_%s", prot->name);
Arnaldo Carvalho de Melo8feaf0c02005-08-09 20:09:30 -07002890
Catalin Marinas7e56b5d2008-11-21 16:45:22 -08002891 if (prot->twsk_prot->twsk_slab_name == NULL)
Arnaldo Carvalho de Melo8feaf0c02005-08-09 20:09:30 -07002892 goto out_free_request_sock_slab;
2893
Arnaldo Carvalho de Melo6d6ee432005-12-13 23:25:19 -08002894 prot->twsk_prot->twsk_slab =
Catalin Marinas7e56b5d2008-11-21 16:45:22 -08002895 kmem_cache_create(prot->twsk_prot->twsk_slab_name,
Arnaldo Carvalho de Melo6d6ee432005-12-13 23:25:19 -08002896 prot->twsk_prot->twsk_obj_size,
Eric Dumazet3ab5aee2008-11-16 19:40:17 -08002897 0,
Eric Dumazet52db70d2015-04-10 06:07:18 -07002898 prot->slab_flags,
Paul Mundt20c2df82007-07-20 10:11:58 +09002899 NULL);
Arnaldo Carvalho de Melo6d6ee432005-12-13 23:25:19 -08002900 if (prot->twsk_prot->twsk_slab == NULL)
Arnaldo Carvalho de Melo8feaf0c02005-08-09 20:09:30 -07002901 goto out_free_timewait_sock_slab_name;
2902 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07002903 }
2904
Glauber Costa36b77a52011-12-16 00:51:59 +00002905 mutex_lock(&proto_list_mutex);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002906 list_add(&prot->node, &proto_list);
Pavel Emelyanov13ff3d62008-03-28 16:38:17 -07002907 assign_proto_idx(prot);
Glauber Costa36b77a52011-12-16 00:51:59 +00002908 mutex_unlock(&proto_list_mutex);
Pavel Emelyanovb733c002007-11-07 02:23:38 -08002909 return 0;
2910
Arnaldo Carvalho de Melo8feaf0c02005-08-09 20:09:30 -07002911out_free_timewait_sock_slab_name:
Catalin Marinas7e56b5d2008-11-21 16:45:22 -08002912 kfree(prot->twsk_prot->twsk_slab_name);
Arnaldo Carvalho de Melo8feaf0c02005-08-09 20:09:30 -07002913out_free_request_sock_slab:
Eric Dumazet0159dfd2015-03-12 16:44:07 -07002914 req_prot_cleanup(prot->rsk_prot);
2915
Arnaldo Carvalho de Melo2e6599c2005-06-18 22:46:52 -07002916 kmem_cache_destroy(prot->slab);
2917 prot->slab = NULL;
Pavel Emelyanovb733c002007-11-07 02:23:38 -08002918out:
2919 return -ENOBUFS;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002920}
Linus Torvalds1da177e2005-04-16 15:20:36 -07002921EXPORT_SYMBOL(proto_register);
2922
2923void proto_unregister(struct proto *prot)
2924{
Glauber Costa36b77a52011-12-16 00:51:59 +00002925 mutex_lock(&proto_list_mutex);
Pavel Emelyanov13ff3d62008-03-28 16:38:17 -07002926 release_proto_idx(prot);
Patrick McHardy0a3f4352005-09-06 19:47:50 -07002927 list_del(&prot->node);
Glauber Costa36b77a52011-12-16 00:51:59 +00002928 mutex_unlock(&proto_list_mutex);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002929
Julia Lawalladf78ed2015-09-13 14:15:18 +02002930 kmem_cache_destroy(prot->slab);
2931 prot->slab = NULL;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002932
Eric Dumazet0159dfd2015-03-12 16:44:07 -07002933 req_prot_cleanup(prot->rsk_prot);
Arnaldo Carvalho de Melo2e6599c2005-06-18 22:46:52 -07002934
Arnaldo Carvalho de Melo6d6ee432005-12-13 23:25:19 -08002935 if (prot->twsk_prot != NULL && prot->twsk_prot->twsk_slab != NULL) {
Arnaldo Carvalho de Melo6d6ee432005-12-13 23:25:19 -08002936 kmem_cache_destroy(prot->twsk_prot->twsk_slab);
Catalin Marinas7e56b5d2008-11-21 16:45:22 -08002937 kfree(prot->twsk_prot->twsk_slab_name);
Arnaldo Carvalho de Melo6d6ee432005-12-13 23:25:19 -08002938 prot->twsk_prot->twsk_slab = NULL;
Arnaldo Carvalho de Melo8feaf0c02005-08-09 20:09:30 -07002939 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07002940}
Linus Torvalds1da177e2005-04-16 15:20:36 -07002941EXPORT_SYMBOL(proto_unregister);
2942
2943#ifdef CONFIG_PROC_FS
Linus Torvalds1da177e2005-04-16 15:20:36 -07002944static void *proto_seq_start(struct seq_file *seq, loff_t *pos)
Glauber Costa36b77a52011-12-16 00:51:59 +00002945 __acquires(proto_list_mutex)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002946{
Glauber Costa36b77a52011-12-16 00:51:59 +00002947 mutex_lock(&proto_list_mutex);
Pavel Emelianov60f04382007-07-09 13:15:14 -07002948 return seq_list_start_head(&proto_list, *pos);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002949}
2950
2951static void *proto_seq_next(struct seq_file *seq, void *v, loff_t *pos)
2952{
Pavel Emelianov60f04382007-07-09 13:15:14 -07002953 return seq_list_next(v, &proto_list, pos);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002954}
2955
2956static void proto_seq_stop(struct seq_file *seq, void *v)
Glauber Costa36b77a52011-12-16 00:51:59 +00002957 __releases(proto_list_mutex)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002958{
Glauber Costa36b77a52011-12-16 00:51:59 +00002959 mutex_unlock(&proto_list_mutex);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002960}
2961
2962static char proto_method_implemented(const void *method)
2963{
2964 return method == NULL ? 'n' : 'y';
2965}
Glauber Costa180d8cd2011-12-11 21:47:02 +00002966static long sock_prot_memory_allocated(struct proto *proto)
2967{
Jeffrin Josecb75a362012-04-25 19:17:29 +05302968 return proto->memory_allocated != NULL ? proto_memory_allocated(proto) : -1L;
Glauber Costa180d8cd2011-12-11 21:47:02 +00002969}
2970
2971static char *sock_prot_memory_pressure(struct proto *proto)
2972{
2973 return proto->memory_pressure != NULL ?
2974 proto_memory_pressure(proto) ? "yes" : "no" : "NI";
2975}
Linus Torvalds1da177e2005-04-16 15:20:36 -07002976
2977static void proto_seq_printf(struct seq_file *seq, struct proto *proto)
2978{
Glauber Costa180d8cd2011-12-11 21:47:02 +00002979
Eric Dumazet8d987e52010-11-09 23:24:26 +00002980 seq_printf(seq, "%-9s %4u %6d %6ld %-3s %6u %-3s %-10s "
Linus Torvalds1da177e2005-04-16 15:20:36 -07002981 "%2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c\n",
2982 proto->name,
2983 proto->obj_size,
Eric Dumazet14e943d2008-11-19 15:14:01 -08002984 sock_prot_inuse_get(seq_file_net(seq), proto),
Glauber Costa180d8cd2011-12-11 21:47:02 +00002985 sock_prot_memory_allocated(proto),
2986 sock_prot_memory_pressure(proto),
Linus Torvalds1da177e2005-04-16 15:20:36 -07002987 proto->max_header,
2988 proto->slab == NULL ? "no" : "yes",
2989 module_name(proto->owner),
2990 proto_method_implemented(proto->close),
2991 proto_method_implemented(proto->connect),
2992 proto_method_implemented(proto->disconnect),
2993 proto_method_implemented(proto->accept),
2994 proto_method_implemented(proto->ioctl),
2995 proto_method_implemented(proto->init),
2996 proto_method_implemented(proto->destroy),
2997 proto_method_implemented(proto->shutdown),
2998 proto_method_implemented(proto->setsockopt),
2999 proto_method_implemented(proto->getsockopt),
3000 proto_method_implemented(proto->sendmsg),
3001 proto_method_implemented(proto->recvmsg),
3002 proto_method_implemented(proto->sendpage),
3003 proto_method_implemented(proto->bind),
3004 proto_method_implemented(proto->backlog_rcv),
3005 proto_method_implemented(proto->hash),
3006 proto_method_implemented(proto->unhash),
3007 proto_method_implemented(proto->get_port),
3008 proto_method_implemented(proto->enter_memory_pressure));
3009}
3010
3011static int proto_seq_show(struct seq_file *seq, void *v)
3012{
Pavel Emelianov60f04382007-07-09 13:15:14 -07003013 if (v == &proto_list)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003014 seq_printf(seq, "%-9s %-4s %-8s %-6s %-5s %-7s %-4s %-10s %s",
3015 "protocol",
3016 "size",
3017 "sockets",
3018 "memory",
3019 "press",
3020 "maxhdr",
3021 "slab",
3022 "module",
3023 "cl co di ac io in de sh ss gs se re sp bi br ha uh gp em\n");
3024 else
Pavel Emelianov60f04382007-07-09 13:15:14 -07003025 proto_seq_printf(seq, list_entry(v, struct proto, node));
Linus Torvalds1da177e2005-04-16 15:20:36 -07003026 return 0;
3027}
3028
Stephen Hemmingerf6908082007-03-12 14:34:29 -07003029static const struct seq_operations proto_seq_ops = {
Linus Torvalds1da177e2005-04-16 15:20:36 -07003030 .start = proto_seq_start,
3031 .next = proto_seq_next,
3032 .stop = proto_seq_stop,
3033 .show = proto_seq_show,
3034};
3035
3036static int proto_seq_open(struct inode *inode, struct file *file)
3037{
Eric Dumazet14e943d2008-11-19 15:14:01 -08003038 return seq_open_net(inode, file, &proto_seq_ops,
3039 sizeof(struct seq_net_private));
Linus Torvalds1da177e2005-04-16 15:20:36 -07003040}
3041
Arjan van de Ven9a321442007-02-12 00:55:35 -08003042static const struct file_operations proto_seq_fops = {
Linus Torvalds1da177e2005-04-16 15:20:36 -07003043 .owner = THIS_MODULE,
3044 .open = proto_seq_open,
3045 .read = seq_read,
3046 .llseek = seq_lseek,
Eric Dumazet14e943d2008-11-19 15:14:01 -08003047 .release = seq_release_net,
3048};
3049
3050static __net_init int proto_init_net(struct net *net)
3051{
Gao fengd4beaa62013-02-18 01:34:54 +00003052 if (!proc_create("protocols", S_IRUGO, net->proc_net, &proto_seq_fops))
Eric Dumazet14e943d2008-11-19 15:14:01 -08003053 return -ENOMEM;
3054
3055 return 0;
3056}
3057
3058static __net_exit void proto_exit_net(struct net *net)
3059{
Gao fengece31ff2013-02-18 01:34:56 +00003060 remove_proc_entry("protocols", net->proc_net);
Eric Dumazet14e943d2008-11-19 15:14:01 -08003061}
3062
3063
3064static __net_initdata struct pernet_operations proto_net_ops = {
3065 .init = proto_init_net,
3066 .exit = proto_exit_net,
Linus Torvalds1da177e2005-04-16 15:20:36 -07003067};
3068
3069static int __init proto_init(void)
3070{
Eric Dumazet14e943d2008-11-19 15:14:01 -08003071 return register_pernet_subsys(&proto_net_ops);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003072}
3073
3074subsys_initcall(proto_init);
3075
3076#endif /* PROC_FS */