blob: 2c4f574168fbdcebc0cc82c0c8a36214992d6224 [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
Linus Torvalds7c0f6ba2016-12-24 11:46:01 -0800121#include <linux/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
David Howellscdfbabf2017-03-09 08:09:05 +0000200 * one slock key per address family and separate keys for internal and
201 * userspace sockets.
Ingo Molnarda21f242006-07-03 00:25:12 -0700202 */
Ingo Molnara5b5bb92006-07-03 00:25:35 -0700203static struct lock_class_key af_family_keys[AF_MAX];
David Howellscdfbabf2017-03-09 08:09:05 +0000204static struct lock_class_key af_family_kern_keys[AF_MAX];
Ingo Molnara5b5bb92006-07-03 00:25:35 -0700205static struct lock_class_key af_family_slock_keys[AF_MAX];
David Howellscdfbabf2017-03-09 08:09:05 +0000206static struct lock_class_key af_family_kern_slock_keys[AF_MAX];
Ingo Molnara5b5bb92006-07-03 00:25:35 -0700207
Ingo Molnara5b5bb92006-07-03 00:25:35 -0700208/*
209 * Make lock validator output more readable. (we pre-construct these
210 * strings build-time, so that runtime initialization of socket
211 * locks is fast):
212 */
David Howellscdfbabf2017-03-09 08:09:05 +0000213
214#define _sock_locks(x) \
215 x "AF_UNSPEC", x "AF_UNIX" , x "AF_INET" , \
216 x "AF_AX25" , x "AF_IPX" , x "AF_APPLETALK", \
217 x "AF_NETROM", x "AF_BRIDGE" , x "AF_ATMPVC" , \
218 x "AF_X25" , x "AF_INET6" , x "AF_ROSE" , \
219 x "AF_DECnet", x "AF_NETBEUI" , x "AF_SECURITY" , \
220 x "AF_KEY" , x "AF_NETLINK" , x "AF_PACKET" , \
221 x "AF_ASH" , x "AF_ECONET" , x "AF_ATMSVC" , \
222 x "AF_RDS" , x "AF_SNA" , x "AF_IRDA" , \
223 x "AF_PPPOX" , x "AF_WANPIPE" , x "AF_LLC" , \
224 x "27" , x "28" , x "AF_CAN" , \
225 x "AF_TIPC" , x "AF_BLUETOOTH", x "IUCV" , \
226 x "AF_RXRPC" , x "AF_ISDN" , x "AF_PHONET" , \
227 x "AF_IEEE802154", x "AF_CAIF" , x "AF_ALG" , \
228 x "AF_NFC" , x "AF_VSOCK" , x "AF_KCM" , \
229 x "AF_QIPCRTR", x "AF_SMC" , x "AF_MAX"
230
Jan Engelhardt36cbd3d2009-08-05 10:42:58 -0700231static const char *const af_family_key_strings[AF_MAX+1] = {
David Howellscdfbabf2017-03-09 08:09:05 +0000232 _sock_locks("sk_lock-")
Ingo Molnara5b5bb92006-07-03 00:25:35 -0700233};
Jan Engelhardt36cbd3d2009-08-05 10:42:58 -0700234static const char *const af_family_slock_key_strings[AF_MAX+1] = {
David Howellscdfbabf2017-03-09 08:09:05 +0000235 _sock_locks("slock-")
Ingo Molnara5b5bb92006-07-03 00:25:35 -0700236};
Jan Engelhardt36cbd3d2009-08-05 10:42:58 -0700237static const char *const af_family_clock_key_strings[AF_MAX+1] = {
David Howellscdfbabf2017-03-09 08:09:05 +0000238 _sock_locks("clock-")
239};
240
241static const char *const af_family_kern_key_strings[AF_MAX+1] = {
242 _sock_locks("k-sk_lock-")
243};
244static const char *const af_family_kern_slock_key_strings[AF_MAX+1] = {
245 _sock_locks("k-slock-")
246};
247static const char *const af_family_kern_clock_key_strings[AF_MAX+1] = {
248 _sock_locks("k-clock-")
Peter Zijlstra443aef02007-07-19 01:49:00 -0700249};
Ingo Molnarda21f242006-07-03 00:25:12 -0700250
251/*
252 * sk_callback_lock locking rules are per-address-family,
253 * so split the lock classes by using a per-AF key:
254 */
255static struct lock_class_key af_callback_keys[AF_MAX];
David Howellscdfbabf2017-03-09 08:09:05 +0000256static struct lock_class_key af_kern_callback_keys[AF_MAX];
Ingo Molnarda21f242006-07-03 00:25:12 -0700257
Linus Torvalds1da177e2005-04-16 15:20:36 -0700258/* Take into consideration the size of the struct sk_buff overhead in the
259 * determination of these values, since that is non-constant across
260 * platforms. This makes socket queueing behavior and performance
261 * not depend upon such differences.
262 */
263#define _SK_MEM_PACKETS 256
Eric Dumazet87fb4b72011-10-13 07:28:54 +0000264#define _SK_MEM_OVERHEAD SKB_TRUESIZE(256)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700265#define SK_WMEM_MAX (_SK_MEM_OVERHEAD * _SK_MEM_PACKETS)
266#define SK_RMEM_MAX (_SK_MEM_OVERHEAD * _SK_MEM_PACKETS)
267
268/* Run time adjustable parameters. */
Brian Haleyab32ea52006-09-22 14:15:41 -0700269__u32 sysctl_wmem_max __read_mostly = SK_WMEM_MAX;
Hans Schillstrom6d8ebc82012-04-30 08:13:50 +0200270EXPORT_SYMBOL(sysctl_wmem_max);
Brian Haleyab32ea52006-09-22 14:15:41 -0700271__u32 sysctl_rmem_max __read_mostly = SK_RMEM_MAX;
Hans Schillstrom6d8ebc82012-04-30 08:13:50 +0200272EXPORT_SYMBOL(sysctl_rmem_max);
Brian Haleyab32ea52006-09-22 14:15:41 -0700273__u32 sysctl_wmem_default __read_mostly = SK_WMEM_MAX;
274__u32 sysctl_rmem_default __read_mostly = SK_RMEM_MAX;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700275
Lucas De Marchi25985ed2011-03-30 22:57:33 -0300276/* Maximal space eaten by iovec or ancillary data plus some space */
Brian Haleyab32ea52006-09-22 14:15:41 -0700277int sysctl_optmem_max __read_mostly = sizeof(unsigned long)*(2*UIO_MAXIOV+512);
Eric Dumazet2a915252009-05-27 11:30:05 +0000278EXPORT_SYMBOL(sysctl_optmem_max);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700279
Willem de Bruijnb245be12015-01-30 13:29:32 -0500280int sysctl_tstamp_allow_data __read_mostly = 1;
281
Mel Gormanc93bdd02012-07-31 16:44:19 -0700282struct static_key memalloc_socks = STATIC_KEY_INIT_FALSE;
283EXPORT_SYMBOL_GPL(memalloc_socks);
284
Mel Gorman7cb02402012-07-31 16:44:16 -0700285/**
286 * sk_set_memalloc - sets %SOCK_MEMALLOC
287 * @sk: socket to set it on
288 *
289 * Set %SOCK_MEMALLOC on a socket for access to emergency reserves.
290 * It's the responsibility of the admin to adjust min_free_kbytes
291 * to meet the requirements
292 */
293void sk_set_memalloc(struct sock *sk)
294{
295 sock_set_flag(sk, SOCK_MEMALLOC);
296 sk->sk_allocation |= __GFP_MEMALLOC;
Mel Gormanc93bdd02012-07-31 16:44:19 -0700297 static_key_slow_inc(&memalloc_socks);
Mel Gorman7cb02402012-07-31 16:44:16 -0700298}
299EXPORT_SYMBOL_GPL(sk_set_memalloc);
300
301void sk_clear_memalloc(struct sock *sk)
302{
303 sock_reset_flag(sk, SOCK_MEMALLOC);
304 sk->sk_allocation &= ~__GFP_MEMALLOC;
Mel Gormanc93bdd02012-07-31 16:44:19 -0700305 static_key_slow_dec(&memalloc_socks);
Mel Gormanc76562b2012-07-31 16:44:41 -0700306
307 /*
308 * SOCK_MEMALLOC is allowed to ignore rmem limits to ensure forward
Mel Gorman5d753612015-06-10 21:02:04 -0400309 * progress of swapping. SOCK_MEMALLOC may be cleared while
310 * it has rmem allocations due to the last swapfile being deactivated
311 * but there is a risk that the socket is unusable due to exceeding
312 * the rmem limits. Reclaim the reserves and obey rmem limits again.
Mel Gormanc76562b2012-07-31 16:44:41 -0700313 */
Mel Gorman5d753612015-06-10 21:02:04 -0400314 sk_mem_reclaim(sk);
Mel Gorman7cb02402012-07-31 16:44:16 -0700315}
316EXPORT_SYMBOL_GPL(sk_clear_memalloc);
317
Mel Gormanb4b9e352012-07-31 16:44:26 -0700318int __sk_backlog_rcv(struct sock *sk, struct sk_buff *skb)
319{
320 int ret;
321 unsigned long pflags = current->flags;
322
323 /* these should have been dropped before queueing */
324 BUG_ON(!sock_flag(sk, SOCK_MEMALLOC));
325
326 current->flags |= PF_MEMALLOC;
327 ret = sk->sk_backlog_rcv(sk, skb);
328 tsk_restore_flags(current, pflags, PF_MEMALLOC);
329
330 return ret;
331}
332EXPORT_SYMBOL(__sk_backlog_rcv);
333
Linus Torvalds1da177e2005-04-16 15:20:36 -0700334static int sock_set_timeout(long *timeo_p, char __user *optval, int optlen)
335{
336 struct timeval tv;
337
338 if (optlen < sizeof(tv))
339 return -EINVAL;
340 if (copy_from_user(&tv, optval, sizeof(tv)))
341 return -EFAULT;
Vasily Averinba780732007-05-24 16:58:54 -0700342 if (tv.tv_usec < 0 || tv.tv_usec >= USEC_PER_SEC)
343 return -EDOM;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700344
Vasily Averinba780732007-05-24 16:58:54 -0700345 if (tv.tv_sec < 0) {
Andrew Morton6f11df82007-07-09 13:16:00 -0700346 static int warned __read_mostly;
347
Vasily Averinba780732007-05-24 16:58:54 -0700348 *timeo_p = 0;
Ilpo Järvinen50aab542008-05-02 16:20:10 -0700349 if (warned < 10 && net_ratelimit()) {
Vasily Averinba780732007-05-24 16:58:54 -0700350 warned++;
Joe Perchese005d192012-05-16 19:58:40 +0000351 pr_info("%s: `%s' (pid %d) tries to set negative timeout\n",
352 __func__, current->comm, task_pid_nr(current));
Ilpo Järvinen50aab542008-05-02 16:20:10 -0700353 }
Vasily Averinba780732007-05-24 16:58:54 -0700354 return 0;
355 }
Linus Torvalds1da177e2005-04-16 15:20:36 -0700356 *timeo_p = MAX_SCHEDULE_TIMEOUT;
357 if (tv.tv_sec == 0 && tv.tv_usec == 0)
358 return 0;
359 if (tv.tv_sec < (MAX_SCHEDULE_TIMEOUT/HZ - 1))
Gao Feng8ccde4c2017-02-21 17:09:19 +0800360 *timeo_p = tv.tv_sec * HZ + DIV_ROUND_UP(tv.tv_usec, USEC_PER_SEC / HZ);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700361 return 0;
362}
363
364static void sock_warn_obsolete_bsdism(const char *name)
365{
366 static int warned;
367 static char warncomm[TASK_COMM_LEN];
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +0900368 if (strcmp(warncomm, current->comm) && warned < 5) {
369 strcpy(warncomm, current->comm);
Joe Perchese005d192012-05-16 19:58:40 +0000370 pr_warn("process `%s' is using obsolete %s SO_BSDCOMPAT\n",
371 warncomm, name);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700372 warned++;
373 }
374}
375
Hannes Frederic Sowa080a2702015-10-26 13:51:37 +0100376static bool sock_needs_netstamp(const struct sock *sk)
377{
378 switch (sk->sk_family) {
379 case AF_UNSPEC:
380 case AF_UNIX:
381 return false;
382 default:
383 return true;
384 }
385}
386
Eric Dumazet08e29af2011-11-28 12:04:18 +0000387static void sock_disable_timestamp(struct sock *sk, unsigned long flags)
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +0900388{
Eric Dumazet08e29af2011-11-28 12:04:18 +0000389 if (sk->sk_flags & flags) {
390 sk->sk_flags &= ~flags;
Hannes Frederic Sowa080a2702015-10-26 13:51:37 +0100391 if (sock_needs_netstamp(sk) &&
392 !(sk->sk_flags & SK_FLAGS_TIMESTAMP))
Patrick Ohly20d49472009-02-12 05:03:38 +0000393 net_disable_timestamp();
Linus Torvalds1da177e2005-04-16 15:20:36 -0700394 }
395}
396
397
samanthakumare6afc8a2016-04-05 12:41:15 -0400398int __sock_queue_rcv_skb(struct sock *sk, struct sk_buff *skb)
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800399{
Neil Horman3b885782009-10-12 13:26:31 -0700400 unsigned long flags;
401 struct sk_buff_head *list = &sk->sk_receive_queue;
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800402
Eric Dumazet0fd7bac2011-12-21 07:11:44 +0000403 if (atomic_read(&sk->sk_rmem_alloc) >= sk->sk_rcvbuf) {
Eric Dumazet766e90372009-10-14 20:40:11 -0700404 atomic_inc(&sk->sk_drops);
Satoru Moriya3847ce32011-06-17 12:00:03 +0000405 trace_sock_rcvqueue_full(sk, skb);
Eric Dumazet766e90372009-10-14 20:40:11 -0700406 return -ENOMEM;
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800407 }
408
Mel Gormanc76562b2012-07-31 16:44:41 -0700409 if (!sk_rmem_schedule(sk, skb, skb->truesize)) {
Eric Dumazet766e90372009-10-14 20:40:11 -0700410 atomic_inc(&sk->sk_drops);
411 return -ENOBUFS;
Hideo Aoki3ab224b2007-12-31 00:11:19 -0800412 }
413
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800414 skb->dev = NULL;
415 skb_set_owner_r(skb, sk);
David S. Miller49ad9592008-12-17 22:11:38 -0800416
Eric Dumazet7fee2262010-05-11 23:19:48 +0000417 /* we escape from rcu protected region, make sure we dont leak
418 * a norefcounted dst
419 */
420 skb_dst_force(skb);
421
Neil Horman3b885782009-10-12 13:26:31 -0700422 spin_lock_irqsave(&list->lock, flags);
Eyal Birger3bc3b962015-03-01 14:58:30 +0200423 sock_skb_set_dropcount(sk, skb);
Neil Horman3b885782009-10-12 13:26:31 -0700424 __skb_queue_tail(list, skb);
425 spin_unlock_irqrestore(&list->lock, flags);
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800426
427 if (!sock_flag(sk, SOCK_DEAD))
David S. Miller676d2362014-04-11 16:15:36 -0400428 sk->sk_data_ready(sk);
Eric Dumazet766e90372009-10-14 20:40:11 -0700429 return 0;
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800430}
samanthakumare6afc8a2016-04-05 12:41:15 -0400431EXPORT_SYMBOL(__sock_queue_rcv_skb);
432
433int sock_queue_rcv_skb(struct sock *sk, struct sk_buff *skb)
434{
435 int err;
436
437 err = sk_filter(sk, skb);
438 if (err)
439 return err;
440
441 return __sock_queue_rcv_skb(sk, skb);
442}
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800443EXPORT_SYMBOL(sock_queue_rcv_skb);
444
Willem de Bruijn4f0c40d92016-07-12 18:18:57 -0400445int __sk_receive_skb(struct sock *sk, struct sk_buff *skb,
Eric Dumazetc3f24cf2016-11-02 17:14:41 -0700446 const int nested, unsigned int trim_cap, bool refcounted)
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800447{
448 int rc = NET_RX_SUCCESS;
449
Willem de Bruijn4f0c40d92016-07-12 18:18:57 -0400450 if (sk_filter_trim_cap(sk, skb, trim_cap))
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800451 goto discard_and_relse;
452
453 skb->dev = NULL;
454
Sorin Dumitru274f4822014-07-22 21:16:51 +0300455 if (sk_rcvqueues_full(sk, sk->sk_rcvbuf)) {
Eric Dumazetc3774112010-04-27 15:13:20 -0700456 atomic_inc(&sk->sk_drops);
457 goto discard_and_relse;
458 }
Arnaldo Carvalho de Melo58a5a7b2006-11-16 14:06:06 -0200459 if (nested)
460 bh_lock_sock_nested(sk);
461 else
462 bh_lock_sock(sk);
Ingo Molnara5b5bb92006-07-03 00:25:35 -0700463 if (!sock_owned_by_user(sk)) {
464 /*
465 * trylock + unlock semantics:
466 */
467 mutex_acquire(&sk->sk_lock.dep_map, 0, 1, _RET_IP_);
468
Peter Zijlstrac57943a2008-10-07 14:18:42 -0700469 rc = sk_backlog_rcv(sk, skb);
Ingo Molnara5b5bb92006-07-03 00:25:35 -0700470
471 mutex_release(&sk->sk_lock.dep_map, 1, _RET_IP_);
Eric Dumazetf545a382012-04-22 23:34:26 +0000472 } else if (sk_add_backlog(sk, skb, sk->sk_rcvbuf)) {
Zhu Yi8eae9392010-03-04 18:01:40 +0000473 bh_unlock_sock(sk);
474 atomic_inc(&sk->sk_drops);
475 goto discard_and_relse;
476 }
477
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800478 bh_unlock_sock(sk);
479out:
Eric Dumazetc3f24cf2016-11-02 17:14:41 -0700480 if (refcounted)
481 sock_put(sk);
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800482 return rc;
483discard_and_relse:
484 kfree_skb(skb);
485 goto out;
486}
Willem de Bruijn4f0c40d92016-07-12 18:18:57 -0400487EXPORT_SYMBOL(__sk_receive_skb);
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800488
489struct dst_entry *__sk_dst_check(struct sock *sk, u32 cookie)
490{
Eric Dumazetb6c67122010-04-08 23:03:29 +0000491 struct dst_entry *dst = __sk_dst_get(sk);
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800492
493 if (dst && dst->obsolete && dst->ops->check(dst, cookie) == NULL) {
Krishna Kumare022f0b2009-10-19 23:46:20 +0000494 sk_tx_queue_clear(sk);
Julian Anastasov9b8805a2017-02-06 23:14:11 +0200495 sk->sk_dst_pending_confirm = 0;
Stephen Hemmingera9b3cd72011-08-01 16:19:00 +0000496 RCU_INIT_POINTER(sk->sk_dst_cache, NULL);
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800497 dst_release(dst);
498 return NULL;
499 }
500
501 return dst;
502}
503EXPORT_SYMBOL(__sk_dst_check);
504
505struct dst_entry *sk_dst_check(struct sock *sk, u32 cookie)
506{
507 struct dst_entry *dst = sk_dst_get(sk);
508
509 if (dst && dst->obsolete && dst->ops->check(dst, cookie) == NULL) {
510 sk_dst_reset(sk);
511 dst_release(dst);
512 return NULL;
513 }
514
515 return dst;
516}
517EXPORT_SYMBOL(sk_dst_check);
518
Brian Haleyc91f6df2012-11-26 05:21:08 +0000519static int sock_setbindtodevice(struct sock *sk, char __user *optval,
520 int optlen)
David S. Miller48788092007-09-14 16:41:03 -0700521{
522 int ret = -ENOPROTOOPT;
523#ifdef CONFIG_NETDEVICES
YOSHIFUJI Hideaki3b1e0a62008-03-26 02:26:21 +0900524 struct net *net = sock_net(sk);
David S. Miller48788092007-09-14 16:41:03 -0700525 char devname[IFNAMSIZ];
526 int index;
527
528 /* Sorry... */
529 ret = -EPERM;
Eric W. Biederman5e1fccc2012-11-16 03:03:04 +0000530 if (!ns_capable(net->user_ns, CAP_NET_RAW))
David S. Miller48788092007-09-14 16:41:03 -0700531 goto out;
532
533 ret = -EINVAL;
534 if (optlen < 0)
535 goto out;
536
537 /* Bind this socket to a particular device like "eth0",
538 * as specified in the passed interface name. If the
539 * name is "" or the option length is zero the socket
540 * is not bound.
541 */
542 if (optlen > IFNAMSIZ - 1)
543 optlen = IFNAMSIZ - 1;
544 memset(devname, 0, sizeof(devname));
545
546 ret = -EFAULT;
547 if (copy_from_user(devname, optval, optlen))
548 goto out;
549
David S. Miller000ba2e2009-11-05 22:37:11 -0800550 index = 0;
551 if (devname[0] != '\0') {
Eric Dumazetbf8e56b2009-11-05 21:03:39 -0800552 struct net_device *dev;
David S. Miller48788092007-09-14 16:41:03 -0700553
Eric Dumazetbf8e56b2009-11-05 21:03:39 -0800554 rcu_read_lock();
555 dev = dev_get_by_name_rcu(net, devname);
556 if (dev)
557 index = dev->ifindex;
558 rcu_read_unlock();
David S. Miller48788092007-09-14 16:41:03 -0700559 ret = -ENODEV;
560 if (!dev)
561 goto out;
David S. Miller48788092007-09-14 16:41:03 -0700562 }
563
564 lock_sock(sk);
565 sk->sk_bound_dev_if = index;
566 sk_dst_reset(sk);
567 release_sock(sk);
568
569 ret = 0;
570
571out:
572#endif
573
574 return ret;
575}
576
Brian Haleyc91f6df2012-11-26 05:21:08 +0000577static int sock_getbindtodevice(struct sock *sk, char __user *optval,
578 int __user *optlen, int len)
579{
580 int ret = -ENOPROTOOPT;
581#ifdef CONFIG_NETDEVICES
582 struct net *net = sock_net(sk);
Brian Haleyc91f6df2012-11-26 05:21:08 +0000583 char devname[IFNAMSIZ];
Brian Haleyc91f6df2012-11-26 05:21:08 +0000584
585 if (sk->sk_bound_dev_if == 0) {
586 len = 0;
587 goto zero;
588 }
589
590 ret = -EINVAL;
591 if (len < IFNAMSIZ)
592 goto out;
593
Nicolas Schichan5dbe7c12013-06-26 17:23:42 +0200594 ret = netdev_get_name(net, devname, sk->sk_bound_dev_if);
595 if (ret)
Brian Haleyc91f6df2012-11-26 05:21:08 +0000596 goto out;
Brian Haleyc91f6df2012-11-26 05:21:08 +0000597
598 len = strlen(devname) + 1;
599
600 ret = -EFAULT;
601 if (copy_to_user(optval, devname, len))
602 goto out;
603
604zero:
605 ret = -EFAULT;
606 if (put_user(len, optlen))
607 goto out;
608
609 ret = 0;
610
611out:
612#endif
613
614 return ret;
615}
616
Pavel Emelyanovc0ef8772007-11-15 03:03:19 -0800617static inline void sock_valbool_flag(struct sock *sk, int bit, int valbool)
618{
619 if (valbool)
620 sock_set_flag(sk, bit);
621 else
622 sock_reset_flag(sk, bit);
623}
624
hannes@stressinduktion.orgf60e5992015-04-01 17:07:44 +0200625bool sk_mc_loop(struct sock *sk)
626{
627 if (dev_recursion_level())
628 return false;
629 if (!sk)
630 return true;
631 switch (sk->sk_family) {
632 case AF_INET:
633 return inet_sk(sk)->mc_loop;
634#if IS_ENABLED(CONFIG_IPV6)
635 case AF_INET6:
636 return inet6_sk(sk)->mc_loop;
637#endif
638 }
639 WARN_ON(1);
640 return true;
641}
642EXPORT_SYMBOL(sk_mc_loop);
643
Linus Torvalds1da177e2005-04-16 15:20:36 -0700644/*
645 * This is meant for all protocols to use and covers goings on
646 * at the socket level. Everything here is generic.
647 */
648
649int sock_setsockopt(struct socket *sock, int level, int optname,
David S. Millerb7058842009-09-30 16:12:20 -0700650 char __user *optval, unsigned int optlen)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700651{
Eric Dumazet2a915252009-05-27 11:30:05 +0000652 struct sock *sk = sock->sk;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700653 int val;
654 int valbool;
655 struct linger ling;
656 int ret = 0;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +0900657
Linus Torvalds1da177e2005-04-16 15:20:36 -0700658 /*
659 * Options without arguments
660 */
661
David S. Miller48788092007-09-14 16:41:03 -0700662 if (optname == SO_BINDTODEVICE)
Brian Haleyc91f6df2012-11-26 05:21:08 +0000663 return sock_setbindtodevice(sk, optval, optlen);
David S. Miller48788092007-09-14 16:41:03 -0700664
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700665 if (optlen < sizeof(int))
666 return -EINVAL;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +0900667
Linus Torvalds1da177e2005-04-16 15:20:36 -0700668 if (get_user(val, (int __user *)optval))
669 return -EFAULT;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +0900670
Eric Dumazet2a915252009-05-27 11:30:05 +0000671 valbool = val ? 1 : 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700672
673 lock_sock(sk);
674
Eric Dumazet2a915252009-05-27 11:30:05 +0000675 switch (optname) {
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700676 case SO_DEBUG:
Eric Dumazet2a915252009-05-27 11:30:05 +0000677 if (val && !capable(CAP_NET_ADMIN))
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700678 ret = -EACCES;
Eric Dumazet2a915252009-05-27 11:30:05 +0000679 else
Pavel Emelyanovc0ef8772007-11-15 03:03:19 -0800680 sock_valbool_flag(sk, SOCK_DBG, valbool);
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700681 break;
682 case SO_REUSEADDR:
Pavel Emelyanov4a17fd52012-04-19 03:39:36 +0000683 sk->sk_reuse = (valbool ? SK_CAN_REUSE : SK_NO_REUSE);
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700684 break;
Tom Herbert055dc212013-01-22 09:49:50 +0000685 case SO_REUSEPORT:
686 sk->sk_reuseport = valbool;
687 break;
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700688 case SO_TYPE:
Jan Engelhardt49c794e2009-08-04 07:28:28 +0000689 case SO_PROTOCOL:
Jan Engelhardt0d6038e2009-08-04 07:28:29 +0000690 case SO_DOMAIN:
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700691 case SO_ERROR:
692 ret = -ENOPROTOOPT;
693 break;
694 case SO_DONTROUTE:
Pavel Emelyanovc0ef8772007-11-15 03:03:19 -0800695 sock_valbool_flag(sk, SOCK_LOCALROUTE, valbool);
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700696 break;
697 case SO_BROADCAST:
698 sock_valbool_flag(sk, SOCK_BROADCAST, valbool);
699 break;
700 case SO_SNDBUF:
701 /* Don't error on this BSD doesn't and if you think
Eric Dumazet82981932012-04-26 20:07:59 +0000702 * about it this is right. Otherwise apps have to
703 * play 'guess the biggest size' games. RCVBUF/SNDBUF
704 * are treated in BSD as hints
705 */
706 val = min_t(u32, val, sysctl_wmem_max);
Patrick McHardyb0573de2005-08-09 19:30:51 -0700707set_sndbuf:
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700708 sk->sk_userlocks |= SOCK_SNDBUF_LOCK;
Eric Dumazetb98b0bc2016-12-02 09:44:53 -0800709 sk->sk_sndbuf = max_t(int, val * 2, SOCK_MIN_SNDBUF);
Eric Dumazet82981932012-04-26 20:07:59 +0000710 /* Wake up sending tasks if we upped the value. */
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700711 sk->sk_write_space(sk);
712 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700713
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700714 case SO_SNDBUFFORCE:
715 if (!capable(CAP_NET_ADMIN)) {
716 ret = -EPERM;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700717 break;
718 }
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700719 goto set_sndbuf;
720
721 case SO_RCVBUF:
722 /* Don't error on this BSD doesn't and if you think
Eric Dumazet82981932012-04-26 20:07:59 +0000723 * about it this is right. Otherwise apps have to
724 * play 'guess the biggest size' games. RCVBUF/SNDBUF
725 * are treated in BSD as hints
726 */
727 val = min_t(u32, val, sysctl_rmem_max);
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700728set_rcvbuf:
729 sk->sk_userlocks |= SOCK_RCVBUF_LOCK;
730 /*
731 * We double it on the way in to account for
732 * "struct sk_buff" etc. overhead. Applications
733 * assume that the SO_RCVBUF setting they make will
734 * allow that much actual data to be received on that
735 * socket.
736 *
737 * Applications are unaware that "struct sk_buff" and
738 * other overheads allocate from the receive buffer
739 * during socket buffer allocation.
740 *
741 * And after considering the possible alternatives,
742 * returning the value we actually used in getsockopt
743 * is the most desirable behavior.
744 */
Eric Dumazetb98b0bc2016-12-02 09:44:53 -0800745 sk->sk_rcvbuf = max_t(int, val * 2, SOCK_MIN_RCVBUF);
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700746 break;
747
748 case SO_RCVBUFFORCE:
749 if (!capable(CAP_NET_ADMIN)) {
750 ret = -EPERM;
751 break;
752 }
753 goto set_rcvbuf;
754
755 case SO_KEEPALIVE:
Ursula Braun4b9d07a2017-01-09 16:55:12 +0100756 if (sk->sk_prot->keepalive)
757 sk->sk_prot->keepalive(sk, valbool);
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700758 sock_valbool_flag(sk, SOCK_KEEPOPEN, valbool);
759 break;
760
761 case SO_OOBINLINE:
762 sock_valbool_flag(sk, SOCK_URGINLINE, valbool);
763 break;
764
765 case SO_NO_CHECK:
Tom Herbert28448b82014-05-23 08:47:19 -0700766 sk->sk_no_check_tx = valbool;
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700767 break;
768
769 case SO_PRIORITY:
Eric W. Biederman5e1fccc2012-11-16 03:03:04 +0000770 if ((val >= 0 && val <= 6) ||
771 ns_capable(sock_net(sk)->user_ns, CAP_NET_ADMIN))
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700772 sk->sk_priority = val;
773 else
774 ret = -EPERM;
775 break;
776
777 case SO_LINGER:
778 if (optlen < sizeof(ling)) {
779 ret = -EINVAL; /* 1003.1g */
780 break;
781 }
Eric Dumazet2a915252009-05-27 11:30:05 +0000782 if (copy_from_user(&ling, optval, sizeof(ling))) {
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700783 ret = -EFAULT;
784 break;
785 }
786 if (!ling.l_onoff)
787 sock_reset_flag(sk, SOCK_LINGER);
788 else {
789#if (BITS_PER_LONG == 32)
790 if ((unsigned int)ling.l_linger >= MAX_SCHEDULE_TIMEOUT/HZ)
791 sk->sk_lingertime = MAX_SCHEDULE_TIMEOUT;
792 else
793#endif
794 sk->sk_lingertime = (unsigned int)ling.l_linger * HZ;
795 sock_set_flag(sk, SOCK_LINGER);
796 }
797 break;
798
799 case SO_BSDCOMPAT:
800 sock_warn_obsolete_bsdism("setsockopt");
801 break;
802
803 case SO_PASSCRED:
804 if (valbool)
805 set_bit(SOCK_PASSCRED, &sock->flags);
806 else
807 clear_bit(SOCK_PASSCRED, &sock->flags);
808 break;
809
810 case SO_TIMESTAMP:
Eric Dumazet92f37fd2007-03-25 22:14:49 -0700811 case SO_TIMESTAMPNS:
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700812 if (valbool) {
Eric Dumazet92f37fd2007-03-25 22:14:49 -0700813 if (optname == SO_TIMESTAMP)
814 sock_reset_flag(sk, SOCK_RCVTSTAMPNS);
815 else
816 sock_set_flag(sk, SOCK_RCVTSTAMPNS);
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700817 sock_set_flag(sk, SOCK_RCVTSTAMP);
Patrick Ohly20d49472009-02-12 05:03:38 +0000818 sock_enable_timestamp(sk, SOCK_TIMESTAMP);
Eric Dumazet92f37fd2007-03-25 22:14:49 -0700819 } else {
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700820 sock_reset_flag(sk, SOCK_RCVTSTAMP);
Eric Dumazet92f37fd2007-03-25 22:14:49 -0700821 sock_reset_flag(sk, SOCK_RCVTSTAMPNS);
822 }
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700823 break;
824
Patrick Ohly20d49472009-02-12 05:03:38 +0000825 case SO_TIMESTAMPING:
826 if (val & ~SOF_TIMESTAMPING_MASK) {
Rémi Denis-Courmontf249fb72009-07-20 00:47:04 +0000827 ret = -EINVAL;
Patrick Ohly20d49472009-02-12 05:03:38 +0000828 break;
829 }
Willem de Bruijnb245be12015-01-30 13:29:32 -0500830
Willem de Bruijn09c2d252014-08-04 22:11:47 -0400831 if (val & SOF_TIMESTAMPING_OPT_ID &&
Willem de Bruijn4ed2d762014-08-04 22:11:49 -0400832 !(sk->sk_tsflags & SOF_TIMESTAMPING_OPT_ID)) {
WANG Congac5cc972015-12-16 23:39:04 -0800833 if (sk->sk_protocol == IPPROTO_TCP &&
834 sk->sk_type == SOCK_STREAM) {
Soheil Hassas Yeganeh6db8b962016-04-02 23:08:07 -0400835 if ((1 << sk->sk_state) &
836 (TCPF_CLOSE | TCPF_LISTEN)) {
Willem de Bruijn4ed2d762014-08-04 22:11:49 -0400837 ret = -EINVAL;
838 break;
839 }
840 sk->sk_tskey = tcp_sk(sk)->snd_una;
841 } else {
842 sk->sk_tskey = 0;
843 }
844 }
Francis Yan1c885802016-11-27 23:07:18 -0800845
846 if (val & SOF_TIMESTAMPING_OPT_STATS &&
847 !(val & SOF_TIMESTAMPING_OPT_TSONLY)) {
848 ret = -EINVAL;
849 break;
850 }
851
Willem de Bruijnb9f40e22014-08-04 22:11:46 -0400852 sk->sk_tsflags = val;
Patrick Ohly20d49472009-02-12 05:03:38 +0000853 if (val & SOF_TIMESTAMPING_RX_SOFTWARE)
854 sock_enable_timestamp(sk,
855 SOCK_TIMESTAMPING_RX_SOFTWARE);
856 else
857 sock_disable_timestamp(sk,
Eric Dumazet08e29af2011-11-28 12:04:18 +0000858 (1UL << SOCK_TIMESTAMPING_RX_SOFTWARE));
Patrick Ohly20d49472009-02-12 05:03:38 +0000859 break;
860
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700861 case SO_RCVLOWAT:
862 if (val < 0)
863 val = INT_MAX;
864 sk->sk_rcvlowat = val ? : 1;
865 break;
866
867 case SO_RCVTIMEO:
868 ret = sock_set_timeout(&sk->sk_rcvtimeo, optval, optlen);
869 break;
870
871 case SO_SNDTIMEO:
872 ret = sock_set_timeout(&sk->sk_sndtimeo, optval, optlen);
873 break;
874
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700875 case SO_ATTACH_FILTER:
876 ret = -EINVAL;
877 if (optlen == sizeof(struct sock_fprog)) {
878 struct sock_fprog fprog;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700879
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700880 ret = -EFAULT;
881 if (copy_from_user(&fprog, optval, sizeof(fprog)))
Linus Torvalds1da177e2005-04-16 15:20:36 -0700882 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700883
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700884 ret = sk_attach_filter(&fprog, sk);
885 }
886 break;
887
Alexei Starovoitov89aa0752014-12-01 15:06:35 -0800888 case SO_ATTACH_BPF:
889 ret = -EINVAL;
890 if (optlen == sizeof(u32)) {
891 u32 ufd;
892
893 ret = -EFAULT;
894 if (copy_from_user(&ufd, optval, sizeof(ufd)))
895 break;
896
897 ret = sk_attach_bpf(ufd, sk);
898 }
899 break;
900
Craig Gallek538950a2016-01-04 17:41:47 -0500901 case SO_ATTACH_REUSEPORT_CBPF:
902 ret = -EINVAL;
903 if (optlen == sizeof(struct sock_fprog)) {
904 struct sock_fprog fprog;
905
906 ret = -EFAULT;
907 if (copy_from_user(&fprog, optval, sizeof(fprog)))
908 break;
909
910 ret = sk_reuseport_attach_filter(&fprog, sk);
911 }
912 break;
913
914 case SO_ATTACH_REUSEPORT_EBPF:
915 ret = -EINVAL;
916 if (optlen == sizeof(u32)) {
917 u32 ufd;
918
919 ret = -EFAULT;
920 if (copy_from_user(&ufd, optval, sizeof(ufd)))
921 break;
922
923 ret = sk_reuseport_attach_bpf(ufd, sk);
924 }
925 break;
926
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700927 case SO_DETACH_FILTER:
Pavel Emelyanov55b33322007-10-17 21:21:26 -0700928 ret = sk_detach_filter(sk);
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700929 break;
930
Vincent Bernatd59577b2013-01-16 22:55:49 +0100931 case SO_LOCK_FILTER:
932 if (sock_flag(sk, SOCK_FILTER_LOCKED) && !valbool)
933 ret = -EPERM;
934 else
935 sock_valbool_flag(sk, SOCK_FILTER_LOCKED, valbool);
936 break;
937
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700938 case SO_PASSSEC:
939 if (valbool)
940 set_bit(SOCK_PASSSEC, &sock->flags);
941 else
942 clear_bit(SOCK_PASSSEC, &sock->flags);
943 break;
Laszlo Attila Toth4a19ec52008-01-30 19:08:16 -0800944 case SO_MARK:
Eric W. Biederman5e1fccc2012-11-16 03:03:04 +0000945 if (!ns_capable(sock_net(sk)->user_ns, CAP_NET_ADMIN))
Laszlo Attila Toth4a19ec52008-01-30 19:08:16 -0800946 ret = -EPERM;
Eric Dumazet2a915252009-05-27 11:30:05 +0000947 else
Laszlo Attila Toth4a19ec52008-01-30 19:08:16 -0800948 sk->sk_mark = val;
Laszlo Attila Toth4a19ec52008-01-30 19:08:16 -0800949 break;
Catherine Zhang877ce7c2006-06-29 12:27:47 -0700950
Neil Horman3b885782009-10-12 13:26:31 -0700951 case SO_RXQ_OVFL:
Johannes Berg8083f0f2011-10-07 03:30:20 +0000952 sock_valbool_flag(sk, SOCK_RXQ_OVFL, valbool);
Neil Horman3b885782009-10-12 13:26:31 -0700953 break;
Johannes Berg6e3e9392011-11-09 10:15:42 +0100954
955 case SO_WIFI_STATUS:
956 sock_valbool_flag(sk, SOCK_WIFI_STATUS, valbool);
957 break;
958
Pavel Emelyanovef64a542012-02-21 07:31:34 +0000959 case SO_PEEK_OFF:
960 if (sock->ops->set_peek_off)
Sasha Levin12663bf2013-12-07 17:26:27 -0500961 ret = sock->ops->set_peek_off(sk, val);
Pavel Emelyanovef64a542012-02-21 07:31:34 +0000962 else
963 ret = -EOPNOTSUPP;
964 break;
Ben Greear3bdc0eb2012-02-11 15:39:30 +0000965
966 case SO_NOFCS:
967 sock_valbool_flag(sk, SOCK_NOFCS, valbool);
968 break;
969
Keller, Jacob E7d4c04f2013-03-28 11:19:25 +0000970 case SO_SELECT_ERR_QUEUE:
971 sock_valbool_flag(sk, SOCK_SELECT_ERR_QUEUE, valbool);
972 break;
973
Cong Wange0d10952013-08-01 11:10:25 +0800974#ifdef CONFIG_NET_RX_BUSY_POLL
Eliezer Tamir64b0dc52013-07-10 17:13:36 +0300975 case SO_BUSY_POLL:
Eliezer Tamirdafcc432013-06-14 16:33:57 +0300976 /* allow unprivileged users to decrease the value */
977 if ((val > sk->sk_ll_usec) && !capable(CAP_NET_ADMIN))
978 ret = -EPERM;
979 else {
980 if (val < 0)
981 ret = -EINVAL;
982 else
983 sk->sk_ll_usec = val;
984 }
985 break;
986#endif
Eric Dumazet62748f32013-09-24 08:20:52 -0700987
988 case SO_MAX_PACING_RATE:
989 sk->sk_max_pacing_rate = val;
990 sk->sk_pacing_rate = min(sk->sk_pacing_rate,
991 sk->sk_max_pacing_rate);
992 break;
993
Eric Dumazet70da2682015-10-08 19:33:21 -0700994 case SO_INCOMING_CPU:
995 sk->sk_incoming_cpu = val;
996 break;
997
Tom Herberta87cb3e2016-02-24 10:02:52 -0800998 case SO_CNX_ADVICE:
999 if (val == 1)
1000 dst_negative_advice(sk);
1001 break;
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001002 default:
1003 ret = -ENOPROTOOPT;
1004 break;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001005 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07001006 release_sock(sk);
1007 return ret;
1008}
Eric Dumazet2a915252009-05-27 11:30:05 +00001009EXPORT_SYMBOL(sock_setsockopt);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001010
1011
stephen hemminger8f098982014-01-03 09:17:14 -08001012static void cred_to_ucred(struct pid *pid, const struct cred *cred,
1013 struct ucred *ucred)
Eric W. Biederman3f551f92010-06-13 03:28:59 +00001014{
1015 ucred->pid = pid_vnr(pid);
1016 ucred->uid = ucred->gid = -1;
1017 if (cred) {
1018 struct user_namespace *current_ns = current_user_ns();
1019
Eric W. Biedermanb2e4f542012-05-23 16:39:45 -06001020 ucred->uid = from_kuid_munged(current_ns, cred->euid);
1021 ucred->gid = from_kgid_munged(current_ns, cred->egid);
Eric W. Biederman3f551f92010-06-13 03:28:59 +00001022 }
1023}
1024
Linus Torvalds1da177e2005-04-16 15:20:36 -07001025int sock_getsockopt(struct socket *sock, int level, int optname,
1026 char __user *optval, int __user *optlen)
1027{
1028 struct sock *sk = sock->sk;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001029
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001030 union {
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001031 int val;
1032 struct linger ling;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001033 struct timeval tm;
1034 } v;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001035
H Hartley Sweeten4d0392b2010-01-15 01:08:58 -08001036 int lv = sizeof(int);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001037 int len;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001038
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001039 if (get_user(len, optlen))
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001040 return -EFAULT;
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001041 if (len < 0)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001042 return -EINVAL;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001043
Eugene Teo50fee1d2009-02-23 15:38:41 -08001044 memset(&v, 0, sizeof(v));
Clément Lecignedf0bca02009-02-12 16:59:09 -08001045
Eric Dumazet2a915252009-05-27 11:30:05 +00001046 switch (optname) {
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001047 case SO_DEBUG:
1048 v.val = sock_flag(sk, SOCK_DBG);
1049 break;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001050
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001051 case SO_DONTROUTE:
1052 v.val = sock_flag(sk, SOCK_LOCALROUTE);
1053 break;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001054
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001055 case SO_BROADCAST:
Eric Dumazet1b23a5d2012-05-16 05:57:07 +00001056 v.val = sock_flag(sk, SOCK_BROADCAST);
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001057 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001058
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001059 case SO_SNDBUF:
1060 v.val = sk->sk_sndbuf;
1061 break;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001062
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001063 case SO_RCVBUF:
1064 v.val = sk->sk_rcvbuf;
1065 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001066
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001067 case SO_REUSEADDR:
1068 v.val = sk->sk_reuse;
1069 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001070
Tom Herbert055dc212013-01-22 09:49:50 +00001071 case SO_REUSEPORT:
1072 v.val = sk->sk_reuseport;
1073 break;
1074
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001075 case SO_KEEPALIVE:
Eric Dumazet1b23a5d2012-05-16 05:57:07 +00001076 v.val = sock_flag(sk, SOCK_KEEPOPEN);
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001077 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001078
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001079 case SO_TYPE:
1080 v.val = sk->sk_type;
1081 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001082
Jan Engelhardt49c794e2009-08-04 07:28:28 +00001083 case SO_PROTOCOL:
1084 v.val = sk->sk_protocol;
1085 break;
1086
Jan Engelhardt0d6038e2009-08-04 07:28:29 +00001087 case SO_DOMAIN:
1088 v.val = sk->sk_family;
1089 break;
1090
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001091 case SO_ERROR:
1092 v.val = -sock_error(sk);
Eric Dumazet2a915252009-05-27 11:30:05 +00001093 if (v.val == 0)
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001094 v.val = xchg(&sk->sk_err_soft, 0);
1095 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001096
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001097 case SO_OOBINLINE:
Eric Dumazet1b23a5d2012-05-16 05:57:07 +00001098 v.val = sock_flag(sk, SOCK_URGINLINE);
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001099 break;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001100
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001101 case SO_NO_CHECK:
Tom Herbert28448b82014-05-23 08:47:19 -07001102 v.val = sk->sk_no_check_tx;
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001103 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001104
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001105 case SO_PRIORITY:
1106 v.val = sk->sk_priority;
1107 break;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001108
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001109 case SO_LINGER:
1110 lv = sizeof(v.ling);
Eric Dumazet1b23a5d2012-05-16 05:57:07 +00001111 v.ling.l_onoff = sock_flag(sk, SOCK_LINGER);
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001112 v.ling.l_linger = sk->sk_lingertime / HZ;
1113 break;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001114
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001115 case SO_BSDCOMPAT:
1116 sock_warn_obsolete_bsdism("getsockopt");
1117 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001118
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001119 case SO_TIMESTAMP:
Eric Dumazet92f37fd2007-03-25 22:14:49 -07001120 v.val = sock_flag(sk, SOCK_RCVTSTAMP) &&
1121 !sock_flag(sk, SOCK_RCVTSTAMPNS);
1122 break;
1123
1124 case SO_TIMESTAMPNS:
1125 v.val = sock_flag(sk, SOCK_RCVTSTAMPNS);
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001126 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001127
Patrick Ohly20d49472009-02-12 05:03:38 +00001128 case SO_TIMESTAMPING:
Willem de Bruijnb9f40e22014-08-04 22:11:46 -04001129 v.val = sk->sk_tsflags;
Patrick Ohly20d49472009-02-12 05:03:38 +00001130 break;
1131
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001132 case SO_RCVTIMEO:
Eric Dumazet2a915252009-05-27 11:30:05 +00001133 lv = sizeof(struct timeval);
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001134 if (sk->sk_rcvtimeo == MAX_SCHEDULE_TIMEOUT) {
1135 v.tm.tv_sec = 0;
1136 v.tm.tv_usec = 0;
1137 } else {
1138 v.tm.tv_sec = sk->sk_rcvtimeo / HZ;
Gao Feng8ccde4c2017-02-21 17:09:19 +08001139 v.tm.tv_usec = ((sk->sk_rcvtimeo % HZ) * USEC_PER_SEC) / HZ;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001140 }
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001141 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001142
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001143 case SO_SNDTIMEO:
Eric Dumazet2a915252009-05-27 11:30:05 +00001144 lv = sizeof(struct timeval);
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001145 if (sk->sk_sndtimeo == MAX_SCHEDULE_TIMEOUT) {
1146 v.tm.tv_sec = 0;
1147 v.tm.tv_usec = 0;
1148 } else {
1149 v.tm.tv_sec = sk->sk_sndtimeo / HZ;
Gao Feng8ccde4c2017-02-21 17:09:19 +08001150 v.tm.tv_usec = ((sk->sk_sndtimeo % HZ) * USEC_PER_SEC) / HZ;
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001151 }
1152 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001153
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001154 case SO_RCVLOWAT:
1155 v.val = sk->sk_rcvlowat;
1156 break;
Catherine Zhang877ce7c2006-06-29 12:27:47 -07001157
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001158 case SO_SNDLOWAT:
Eric Dumazet2a915252009-05-27 11:30:05 +00001159 v.val = 1;
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001160 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001161
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001162 case SO_PASSCRED:
Eric Dumazet82981932012-04-26 20:07:59 +00001163 v.val = !!test_bit(SOCK_PASSCRED, &sock->flags);
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001164 break;
1165
1166 case SO_PEERCRED:
Eric W. Biederman109f6e32010-06-13 03:30:14 +00001167 {
1168 struct ucred peercred;
1169 if (len > sizeof(peercred))
1170 len = sizeof(peercred);
1171 cred_to_ucred(sk->sk_peer_pid, sk->sk_peer_cred, &peercred);
1172 if (copy_to_user(optval, &peercred, len))
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001173 return -EFAULT;
1174 goto lenout;
Eric W. Biederman109f6e32010-06-13 03:30:14 +00001175 }
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001176
1177 case SO_PEERNAME:
1178 {
1179 char address[128];
1180
1181 if (sock->ops->getname(sock, (struct sockaddr *)address, &lv, 2))
1182 return -ENOTCONN;
1183 if (lv < len)
1184 return -EINVAL;
1185 if (copy_to_user(optval, address, len))
1186 return -EFAULT;
1187 goto lenout;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001188 }
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001189
1190 /* Dubious BSD thing... Probably nobody even uses it, but
1191 * the UNIX standard wants it for whatever reason... -DaveM
1192 */
1193 case SO_ACCEPTCONN:
1194 v.val = sk->sk_state == TCP_LISTEN;
1195 break;
1196
1197 case SO_PASSSEC:
Eric Dumazet82981932012-04-26 20:07:59 +00001198 v.val = !!test_bit(SOCK_PASSSEC, &sock->flags);
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001199 break;
1200
1201 case SO_PEERSEC:
1202 return security_socket_getpeersec_stream(sock, optval, optlen, len);
1203
Laszlo Attila Toth4a19ec52008-01-30 19:08:16 -08001204 case SO_MARK:
1205 v.val = sk->sk_mark;
1206 break;
1207
Neil Horman3b885782009-10-12 13:26:31 -07001208 case SO_RXQ_OVFL:
Eric Dumazet1b23a5d2012-05-16 05:57:07 +00001209 v.val = sock_flag(sk, SOCK_RXQ_OVFL);
Neil Horman3b885782009-10-12 13:26:31 -07001210 break;
1211
Johannes Berg6e3e9392011-11-09 10:15:42 +01001212 case SO_WIFI_STATUS:
Eric Dumazet1b23a5d2012-05-16 05:57:07 +00001213 v.val = sock_flag(sk, SOCK_WIFI_STATUS);
Johannes Berg6e3e9392011-11-09 10:15:42 +01001214 break;
1215
Pavel Emelyanovef64a542012-02-21 07:31:34 +00001216 case SO_PEEK_OFF:
1217 if (!sock->ops->set_peek_off)
1218 return -EOPNOTSUPP;
1219
1220 v.val = sk->sk_peek_off;
1221 break;
David S. Millerbc2f7992012-02-24 14:48:34 -05001222 case SO_NOFCS:
Eric Dumazet1b23a5d2012-05-16 05:57:07 +00001223 v.val = sock_flag(sk, SOCK_NOFCS);
David S. Millerbc2f7992012-02-24 14:48:34 -05001224 break;
Brian Haleyc91f6df2012-11-26 05:21:08 +00001225
Pavel Emelyanovf7b86bf2012-10-18 23:55:56 +00001226 case SO_BINDTODEVICE:
Brian Haleyc91f6df2012-11-26 05:21:08 +00001227 return sock_getbindtodevice(sk, optval, optlen, len);
1228
Pavel Emelyanova8fc9272012-11-01 02:01:48 +00001229 case SO_GET_FILTER:
1230 len = sk_get_filter(sk, (struct sock_filter __user *)optval, len);
1231 if (len < 0)
1232 return len;
1233
1234 goto lenout;
Brian Haleyc91f6df2012-11-26 05:21:08 +00001235
Vincent Bernatd59577b2013-01-16 22:55:49 +01001236 case SO_LOCK_FILTER:
1237 v.val = sock_flag(sk, SOCK_FILTER_LOCKED);
1238 break;
1239
Michal Sekletarea02f942014-01-17 17:09:45 +01001240 case SO_BPF_EXTENSIONS:
1241 v.val = bpf_tell_extensions();
1242 break;
1243
Keller, Jacob E7d4c04f2013-03-28 11:19:25 +00001244 case SO_SELECT_ERR_QUEUE:
1245 v.val = sock_flag(sk, SOCK_SELECT_ERR_QUEUE);
1246 break;
1247
Cong Wange0d10952013-08-01 11:10:25 +08001248#ifdef CONFIG_NET_RX_BUSY_POLL
Eliezer Tamir64b0dc52013-07-10 17:13:36 +03001249 case SO_BUSY_POLL:
Eliezer Tamirdafcc432013-06-14 16:33:57 +03001250 v.val = sk->sk_ll_usec;
1251 break;
1252#endif
1253
Eric Dumazet62748f32013-09-24 08:20:52 -07001254 case SO_MAX_PACING_RATE:
1255 v.val = sk->sk_max_pacing_rate;
1256 break;
1257
Eric Dumazet2c8c56e2014-11-11 05:54:28 -08001258 case SO_INCOMING_CPU:
1259 v.val = sk->sk_incoming_cpu;
1260 break;
1261
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001262 default:
YOSHIFUJI Hideaki/吉藤英明443b5992015-03-23 18:04:13 +09001263 /* We implement the SO_SNDLOWAT etc to not be settable
1264 * (1003.1g 7).
1265 */
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001266 return -ENOPROTOOPT;
1267 }
1268
Linus Torvalds1da177e2005-04-16 15:20:36 -07001269 if (len > lv)
1270 len = lv;
1271 if (copy_to_user(optval, &v, len))
1272 return -EFAULT;
1273lenout:
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001274 if (put_user(len, optlen))
1275 return -EFAULT;
1276 return 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001277}
1278
Ingo Molnara5b5bb92006-07-03 00:25:35 -07001279/*
1280 * Initialize an sk_lock.
1281 *
1282 * (We also register the sk_lock with the lock validator.)
1283 */
Dave Jonesb6f99a22007-03-22 12:27:49 -07001284static inline void sock_lock_init(struct sock *sk)
Ingo Molnara5b5bb92006-07-03 00:25:35 -07001285{
David Howellscdfbabf2017-03-09 08:09:05 +00001286 if (sk->sk_kern_sock)
1287 sock_lock_init_class_and_name(
1288 sk,
1289 af_family_kern_slock_key_strings[sk->sk_family],
1290 af_family_kern_slock_keys + sk->sk_family,
1291 af_family_kern_key_strings[sk->sk_family],
1292 af_family_kern_keys + sk->sk_family);
1293 else
1294 sock_lock_init_class_and_name(
1295 sk,
Peter Zijlstraed075362006-12-06 20:35:24 -08001296 af_family_slock_key_strings[sk->sk_family],
1297 af_family_slock_keys + sk->sk_family,
1298 af_family_key_strings[sk->sk_family],
1299 af_family_keys + sk->sk_family);
Ingo Molnara5b5bb92006-07-03 00:25:35 -07001300}
1301
Eric Dumazet4dc6dc72009-07-15 23:13:10 +00001302/*
1303 * Copy all fields from osk to nsk but nsk->sk_refcnt must not change yet,
1304 * even temporarly, because of RCU lookups. sk_node should also be left as is.
Eric Dumazet68835ab2010-11-30 19:04:07 +00001305 * We must not copy fields between sk_dontcopy_begin and sk_dontcopy_end
Eric Dumazet4dc6dc72009-07-15 23:13:10 +00001306 */
Pavel Emelyanovf1a6c4d2007-11-01 00:29:45 -07001307static void sock_copy(struct sock *nsk, const struct sock *osk)
1308{
1309#ifdef CONFIG_SECURITY_NETWORK
1310 void *sptr = nsk->sk_security;
1311#endif
Eric Dumazet68835ab2010-11-30 19:04:07 +00001312 memcpy(nsk, osk, offsetof(struct sock, sk_dontcopy_begin));
1313
1314 memcpy(&nsk->sk_dontcopy_end, &osk->sk_dontcopy_end,
1315 osk->sk_prot->obj_size - offsetof(struct sock, sk_dontcopy_end));
1316
Pavel Emelyanovf1a6c4d2007-11-01 00:29:45 -07001317#ifdef CONFIG_SECURITY_NETWORK
1318 nsk->sk_security = sptr;
1319 security_sk_clone(osk, nsk);
1320#endif
1321}
1322
Pavel Emelyanov2e4afe72007-11-01 00:36:26 -07001323static struct sock *sk_prot_alloc(struct proto *prot, gfp_t priority,
1324 int family)
Pavel Emelyanovc308c1b22007-11-01 00:33:50 -07001325{
1326 struct sock *sk;
1327 struct kmem_cache *slab;
1328
1329 slab = prot->slab;
Eric Dumazete912b112009-07-08 19:36:05 +00001330 if (slab != NULL) {
1331 sk = kmem_cache_alloc(slab, priority & ~__GFP_ZERO);
1332 if (!sk)
1333 return sk;
Eric Dumazetba2489b2016-08-23 11:39:29 -07001334 if (priority & __GFP_ZERO)
1335 sk_prot_clear_nulls(sk, prot->obj_size);
Octavian Purdilafcbdf092010-12-16 14:26:56 -08001336 } else
Pavel Emelyanovc308c1b22007-11-01 00:33:50 -07001337 sk = kmalloc(prot->obj_size, priority);
1338
Pavel Emelyanov2e4afe72007-11-01 00:36:26 -07001339 if (sk != NULL) {
Vegard Nossuma98b65a2009-02-26 14:46:57 +01001340 kmemcheck_annotate_bitfield(sk, flags);
1341
Pavel Emelyanov2e4afe72007-11-01 00:36:26 -07001342 if (security_sk_alloc(sk, family, priority))
1343 goto out_free;
1344
1345 if (!try_module_get(prot->owner))
1346 goto out_free_sec;
Krishna Kumare022f0b2009-10-19 23:46:20 +00001347 sk_tx_queue_clear(sk);
Pavel Emelyanov2e4afe72007-11-01 00:36:26 -07001348 }
1349
Pavel Emelyanovc308c1b22007-11-01 00:33:50 -07001350 return sk;
Pavel Emelyanov2e4afe72007-11-01 00:36:26 -07001351
1352out_free_sec:
1353 security_sk_free(sk);
1354out_free:
1355 if (slab != NULL)
1356 kmem_cache_free(slab, sk);
1357 else
1358 kfree(sk);
1359 return NULL;
Pavel Emelyanovc308c1b22007-11-01 00:33:50 -07001360}
1361
1362static void sk_prot_free(struct proto *prot, struct sock *sk)
1363{
1364 struct kmem_cache *slab;
Pavel Emelyanov2e4afe72007-11-01 00:36:26 -07001365 struct module *owner;
Pavel Emelyanovc308c1b22007-11-01 00:33:50 -07001366
Pavel Emelyanov2e4afe72007-11-01 00:36:26 -07001367 owner = prot->owner;
Pavel Emelyanovc308c1b22007-11-01 00:33:50 -07001368 slab = prot->slab;
Pavel Emelyanov2e4afe72007-11-01 00:36:26 -07001369
Tejun Heobd1060a2015-12-07 17:38:53 -05001370 cgroup_sk_free(&sk->sk_cgrp_data);
Johannes Weiner2d758072016-10-07 17:00:58 -07001371 mem_cgroup_sk_free(sk);
Pavel Emelyanov2e4afe72007-11-01 00:36:26 -07001372 security_sk_free(sk);
Pavel Emelyanovc308c1b22007-11-01 00:33:50 -07001373 if (slab != NULL)
1374 kmem_cache_free(slab, sk);
1375 else
1376 kfree(sk);
Pavel Emelyanov2e4afe72007-11-01 00:36:26 -07001377 module_put(owner);
Pavel Emelyanovc308c1b22007-11-01 00:33:50 -07001378}
1379
Linus Torvalds1da177e2005-04-16 15:20:36 -07001380/**
1381 * sk_alloc - All socket objects are allocated here
Randy Dunlapc4ea43c2007-10-12 21:17:49 -07001382 * @net: the applicable net namespace
Pavel Pisa4dc3b162005-05-01 08:59:25 -07001383 * @family: protocol family
1384 * @priority: for allocation (%GFP_KERNEL, %GFP_ATOMIC, etc)
1385 * @prot: struct proto associated with this new sock instance
Eric W. Biederman11aa9c22015-05-08 21:09:13 -05001386 * @kern: is this to be a kernel socket?
Linus Torvalds1da177e2005-04-16 15:20:36 -07001387 */
Eric W. Biederman1b8d7ae2007-10-08 23:24:22 -07001388struct sock *sk_alloc(struct net *net, int family, gfp_t priority,
Eric W. Biederman11aa9c22015-05-08 21:09:13 -05001389 struct proto *prot, int kern)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001390{
Pavel Emelyanovc308c1b22007-11-01 00:33:50 -07001391 struct sock *sk;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001392
Pavel Emelyanov154adbc2007-11-01 00:38:43 -07001393 sk = sk_prot_alloc(prot, priority | __GFP_ZERO, family);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001394 if (sk) {
Pavel Emelyanov154adbc2007-11-01 00:38:43 -07001395 sk->sk_family = family;
1396 /*
1397 * See comment in struct sock definition to understand
1398 * why we need sk_prot_creator -acme
1399 */
1400 sk->sk_prot = sk->sk_prot_creator = prot;
David Howellscdfbabf2017-03-09 08:09:05 +00001401 sk->sk_kern_sock = kern;
Pavel Emelyanov154adbc2007-11-01 00:38:43 -07001402 sock_lock_init(sk);
Eric W. Biederman26abe142015-05-08 21:10:31 -05001403 sk->sk_net_refcnt = kern ? 0 : 1;
1404 if (likely(sk->sk_net_refcnt))
1405 get_net(net);
1406 sock_net_set(sk, net);
Jarek Poplawskid66ee052009-08-30 23:15:36 +00001407 atomic_set(&sk->sk_wmem_alloc, 1);
Herbert Xuf8451722010-05-24 00:12:34 -07001408
Johannes Weiner2d758072016-10-07 17:00:58 -07001409 mem_cgroup_sk_alloc(sk);
Johannes Weinerd979a392016-09-19 14:44:38 -07001410 cgroup_sk_alloc(&sk->sk_cgrp_data);
Tejun Heo2a56a1f2015-12-07 17:38:52 -05001411 sock_update_classid(&sk->sk_cgrp_data);
1412 sock_update_netprioidx(&sk->sk_cgrp_data);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001413 }
Frank Filza79af592005-09-27 15:23:38 -07001414
Pavel Emelyanov2e4afe72007-11-01 00:36:26 -07001415 return sk;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001416}
Eric Dumazet2a915252009-05-27 11:30:05 +00001417EXPORT_SYMBOL(sk_alloc);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001418
Eric Dumazeta4298e42016-04-01 08:52:12 -07001419/* Sockets having SOCK_RCU_FREE will call this function after one RCU
1420 * grace period. This is the case for UDP sockets and TCP listeners.
1421 */
1422static void __sk_destruct(struct rcu_head *head)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001423{
Eric Dumazeta4298e42016-04-01 08:52:12 -07001424 struct sock *sk = container_of(head, struct sock, sk_rcu);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001425 struct sk_filter *filter;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001426
1427 if (sk->sk_destruct)
1428 sk->sk_destruct(sk);
1429
Paul E. McKenneya898def2010-02-22 17:04:49 -08001430 filter = rcu_dereference_check(sk->sk_filter,
1431 atomic_read(&sk->sk_wmem_alloc) == 0);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001432 if (filter) {
Pavel Emelyanov309dd5f2007-10-17 21:21:51 -07001433 sk_filter_uncharge(sk, filter);
Stephen Hemmingera9b3cd72011-08-01 16:19:00 +00001434 RCU_INIT_POINTER(sk->sk_filter, NULL);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001435 }
Craig Gallek538950a2016-01-04 17:41:47 -05001436 if (rcu_access_pointer(sk->sk_reuseport_cb))
1437 reuseport_detach_sock(sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001438
Eric Dumazet08e29af2011-11-28 12:04:18 +00001439 sock_disable_timestamp(sk, SK_FLAGS_TIMESTAMP);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001440
1441 if (atomic_read(&sk->sk_omem_alloc))
Joe Perchese005d192012-05-16 19:58:40 +00001442 pr_debug("%s: optmem leakage (%d bytes) detected\n",
1443 __func__, atomic_read(&sk->sk_omem_alloc));
Linus Torvalds1da177e2005-04-16 15:20:36 -07001444
Eric Dumazet22a0e182017-03-15 13:21:28 -07001445 if (sk->sk_frag.page) {
1446 put_page(sk->sk_frag.page);
1447 sk->sk_frag.page = NULL;
1448 }
1449
Eric W. Biederman109f6e32010-06-13 03:30:14 +00001450 if (sk->sk_peer_cred)
1451 put_cred(sk->sk_peer_cred);
1452 put_pid(sk->sk_peer_pid);
Eric W. Biederman26abe142015-05-08 21:10:31 -05001453 if (likely(sk->sk_net_refcnt))
1454 put_net(sock_net(sk));
Pavel Emelyanovc308c1b22007-11-01 00:33:50 -07001455 sk_prot_free(sk->sk_prot_creator, sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001456}
Eric Dumazet2b85a342009-06-11 02:55:43 -07001457
Eric Dumazeta4298e42016-04-01 08:52:12 -07001458void sk_destruct(struct sock *sk)
1459{
1460 if (sock_flag(sk, SOCK_RCU_FREE))
1461 call_rcu(&sk->sk_rcu, __sk_destruct);
1462 else
1463 __sk_destruct(&sk->sk_rcu);
1464}
1465
Craig Gallekeb4cb002015-06-15 11:26:18 -04001466static void __sk_free(struct sock *sk)
1467{
Craig Gallekb9226222015-06-30 12:49:32 -04001468 if (unlikely(sock_diag_has_destroy_listeners(sk) && sk->sk_net_refcnt))
Craig Gallekeb4cb002015-06-15 11:26:18 -04001469 sock_diag_broadcast_destroy(sk);
1470 else
1471 sk_destruct(sk);
1472}
1473
Eric Dumazet2b85a342009-06-11 02:55:43 -07001474void sk_free(struct sock *sk)
1475{
1476 /*
Lucas De Marchi25985ed2011-03-30 22:57:33 -03001477 * We subtract one from sk_wmem_alloc and can know if
Eric Dumazet2b85a342009-06-11 02:55:43 -07001478 * some packets are still in some tx queue.
1479 * If not null, sock_wfree() will call __sk_free(sk) later
1480 */
1481 if (atomic_dec_and_test(&sk->sk_wmem_alloc))
1482 __sk_free(sk);
1483}
Eric Dumazet2a915252009-05-27 11:30:05 +00001484EXPORT_SYMBOL(sk_free);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001485
Eric Dumazete56c57d2011-11-08 17:07:07 -05001486/**
1487 * sk_clone_lock - clone a socket, and lock its clone
1488 * @sk: the socket to clone
1489 * @priority: for allocation (%GFP_KERNEL, %GFP_ATOMIC, etc)
1490 *
1491 * Caller must unlock socket even in error path (bh_unlock_sock(newsk))
1492 */
1493struct sock *sk_clone_lock(const struct sock *sk, const gfp_t priority)
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001494{
Pavel Emelyanov8fd1d172007-11-01 00:37:32 -07001495 struct sock *newsk;
Alexei Starovoitov278571b2014-07-30 20:34:12 -07001496 bool is_charged = true;
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001497
Pavel Emelyanov8fd1d172007-11-01 00:37:32 -07001498 newsk = sk_prot_alloc(sk->sk_prot, priority, sk->sk_family);
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001499 if (newsk != NULL) {
1500 struct sk_filter *filter;
1501
Venkat Yekkirala892c1412006-08-04 23:08:56 -07001502 sock_copy(newsk, sk);
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001503
1504 /* SANITY */
Sowmini Varadhan8a681732015-07-30 15:50:36 +02001505 if (likely(newsk->sk_net_refcnt))
1506 get_net(sock_net(newsk));
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001507 sk_node_init(&newsk->sk_node);
1508 sock_lock_init(newsk);
1509 bh_lock_sock(newsk);
Eric Dumazetfa438cc2007-03-04 16:05:44 -08001510 newsk->sk_backlog.head = newsk->sk_backlog.tail = NULL;
Zhu Yi8eae9392010-03-04 18:01:40 +00001511 newsk->sk_backlog.len = 0;
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001512
1513 atomic_set(&newsk->sk_rmem_alloc, 0);
Eric Dumazet2b85a342009-06-11 02:55:43 -07001514 /*
1515 * sk_wmem_alloc set to one (see sk_free() and sock_wfree())
1516 */
1517 atomic_set(&newsk->sk_wmem_alloc, 1);
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001518 atomic_set(&newsk->sk_omem_alloc, 0);
1519 skb_queue_head_init(&newsk->sk_receive_queue);
1520 skb_queue_head_init(&newsk->sk_write_queue);
1521
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001522 rwlock_init(&newsk->sk_callback_lock);
Peter Zijlstra443aef02007-07-19 01:49:00 -07001523 lockdep_set_class_and_name(&newsk->sk_callback_lock,
1524 af_callback_keys + newsk->sk_family,
1525 af_family_clock_key_strings[newsk->sk_family]);
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001526
1527 newsk->sk_dst_cache = NULL;
Julian Anastasov9b8805a2017-02-06 23:14:11 +02001528 newsk->sk_dst_pending_confirm = 0;
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001529 newsk->sk_wmem_queued = 0;
1530 newsk->sk_forward_alloc = 0;
Eric Dumazet9caad862016-04-01 08:52:20 -07001531 atomic_set(&newsk->sk_drops, 0);
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001532 newsk->sk_send_head = NULL;
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001533 newsk->sk_userlocks = sk->sk_userlocks & ~SOCK_BINDPORT_LOCK;
1534
1535 sock_reset_flag(newsk, SOCK_DONE);
1536 skb_queue_head_init(&newsk->sk_error_queue);
1537
Eric Dumazet0d7da9d2010-10-25 03:47:05 +00001538 filter = rcu_dereference_protected(newsk->sk_filter, 1);
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001539 if (filter != NULL)
Alexei Starovoitov278571b2014-07-30 20:34:12 -07001540 /* though it's an empty new sock, the charging may fail
1541 * if sysctl_optmem_max was changed between creation of
1542 * original socket and cloning
1543 */
1544 is_charged = sk_filter_charge(newsk, filter);
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001545
Eric Dumazetd188ba82015-12-08 07:22:02 -08001546 if (unlikely(!is_charged || xfrm_sk_clone_policy(newsk, sk))) {
Daniel Borkmanna97e50c2017-03-22 13:08:08 +01001547 /* We need to make sure that we don't uncharge the new
1548 * socket if we couldn't charge it in the first place
1549 * as otherwise we uncharge the parent's filter.
1550 */
1551 if (!is_charged)
1552 RCU_INIT_POINTER(newsk->sk_filter, NULL);
Arnaldo Carvalho de Melo94352d42017-03-01 16:35:08 -03001553 sk_free_unlock_clone(newsk);
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001554 newsk = NULL;
1555 goto out;
1556 }
Craig Gallekfa463492016-02-10 11:50:39 -05001557 RCU_INIT_POINTER(newsk->sk_reuseport_cb, NULL);
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001558
1559 newsk->sk_err = 0;
Eric Dumazete551c322016-10-28 13:40:24 -07001560 newsk->sk_err_soft = 0;
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001561 newsk->sk_priority = 0;
Eric Dumazet2c8c56e2014-11-11 05:54:28 -08001562 newsk->sk_incoming_cpu = raw_smp_processor_id();
Eric Dumazet33cf7c92015-03-11 18:53:14 -07001563 atomic64_set(&newsk->sk_cookie, 0);
Johannes Weinerd979a392016-09-19 14:44:38 -07001564
Johannes Weiner2d758072016-10-07 17:00:58 -07001565 mem_cgroup_sk_alloc(newsk);
Johannes Weinerd979a392016-09-19 14:44:38 -07001566 cgroup_sk_alloc(&newsk->sk_cgrp_data);
1567
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
1590 if (newsk->sk_prot->sockets_allocated)
Glauber Costa180d8cd2011-12-11 21:47:02 +00001591 sk_sockets_allocated_inc(newsk);
Octavian Purdila704da5602010-01-08 00:00:09 -08001592
Hannes Frederic Sowa080a2702015-10-26 13:51:37 +01001593 if (sock_needs_netstamp(sk) &&
1594 newsk->sk_flags & SK_FLAGS_TIMESTAMP)
Octavian Purdila704da5602010-01-08 00:00:09 -08001595 net_enable_timestamp();
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001596 }
1597out:
1598 return newsk;
1599}
Eric Dumazete56c57d2011-11-08 17:07:07 -05001600EXPORT_SYMBOL_GPL(sk_clone_lock);
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001601
Arnaldo Carvalho de Melo94352d42017-03-01 16:35:08 -03001602void sk_free_unlock_clone(struct sock *sk)
1603{
1604 /* It is still raw copy of parent, so invalidate
1605 * destructor and make plain sk_free() */
1606 sk->sk_destruct = NULL;
1607 bh_unlock_sock(sk);
1608 sk_free(sk);
1609}
1610EXPORT_SYMBOL_GPL(sk_free_unlock_clone);
1611
Andi Kleen99580892007-04-20 17:12:43 -07001612void sk_setup_caps(struct sock *sk, struct dst_entry *dst)
1613{
Eric Dumazetd6a4e262015-05-26 08:55:28 -07001614 u32 max_segs = 1;
1615
Eric Dumazet6bd4f352015-12-02 21:53:57 -08001616 sk_dst_set(sk, dst);
Andi Kleen99580892007-04-20 17:12:43 -07001617 sk->sk_route_caps = dst->dev->features;
1618 if (sk->sk_route_caps & NETIF_F_GSO)
Herbert Xu4fcd6b92007-05-31 22:15:50 -07001619 sk->sk_route_caps |= NETIF_F_GSO_SOFTWARE;
Eric Dumazeta4654192010-05-16 00:36:33 -07001620 sk->sk_route_caps &= ~sk->sk_route_nocaps;
Andi Kleen99580892007-04-20 17:12:43 -07001621 if (sk_can_gso(sk)) {
Peter P Waskiewicz Jr82cc1a72008-03-21 03:43:19 -07001622 if (dst->header_len) {
Andi Kleen99580892007-04-20 17:12:43 -07001623 sk->sk_route_caps &= ~NETIF_F_GSO_MASK;
Peter P Waskiewicz Jr82cc1a72008-03-21 03:43:19 -07001624 } else {
Andi Kleen99580892007-04-20 17:12:43 -07001625 sk->sk_route_caps |= NETIF_F_SG | NETIF_F_HW_CSUM;
Peter P Waskiewicz Jr82cc1a72008-03-21 03:43:19 -07001626 sk->sk_gso_max_size = dst->dev->gso_max_size;
Eric Dumazetd6a4e262015-05-26 08:55:28 -07001627 max_segs = max_t(u32, dst->dev->gso_max_segs, 1);
Peter P Waskiewicz Jr82cc1a72008-03-21 03:43:19 -07001628 }
Andi Kleen99580892007-04-20 17:12:43 -07001629 }
Eric Dumazetd6a4e262015-05-26 08:55:28 -07001630 sk->sk_gso_max_segs = max_segs;
Andi Kleen99580892007-04-20 17:12:43 -07001631}
1632EXPORT_SYMBOL_GPL(sk_setup_caps);
1633
Linus Torvalds1da177e2005-04-16 15:20:36 -07001634/*
1635 * Simple resource managers for sockets.
1636 */
1637
1638
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001639/*
1640 * Write buffer destructor automatically called from kfree_skb.
Linus Torvalds1da177e2005-04-16 15:20:36 -07001641 */
1642void sock_wfree(struct sk_buff *skb)
1643{
1644 struct sock *sk = skb->sk;
Eric Dumazetd99927f2009-09-24 10:49:24 +00001645 unsigned int len = skb->truesize;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001646
Eric Dumazetd99927f2009-09-24 10:49:24 +00001647 if (!sock_flag(sk, SOCK_USE_WRITE_QUEUE)) {
1648 /*
1649 * Keep a reference on sk_wmem_alloc, this will be released
1650 * after sk_write_space() call
1651 */
1652 atomic_sub(len - 1, &sk->sk_wmem_alloc);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001653 sk->sk_write_space(sk);
Eric Dumazetd99927f2009-09-24 10:49:24 +00001654 len = 1;
1655 }
Eric Dumazet2b85a342009-06-11 02:55:43 -07001656 /*
Eric Dumazetd99927f2009-09-24 10:49:24 +00001657 * if sk_wmem_alloc reaches 0, we must finish what sk_free()
1658 * could not do because of in-flight packets
Eric Dumazet2b85a342009-06-11 02:55:43 -07001659 */
Eric Dumazetd99927f2009-09-24 10:49:24 +00001660 if (atomic_sub_and_test(len, &sk->sk_wmem_alloc))
Eric Dumazet2b85a342009-06-11 02:55:43 -07001661 __sk_free(sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001662}
Eric Dumazet2a915252009-05-27 11:30:05 +00001663EXPORT_SYMBOL(sock_wfree);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001664
Eric Dumazet1d2077a2016-05-02 10:56:27 -07001665/* This variant of sock_wfree() is used by TCP,
1666 * since it sets SOCK_USE_WRITE_QUEUE.
1667 */
1668void __sock_wfree(struct sk_buff *skb)
1669{
1670 struct sock *sk = skb->sk;
1671
1672 if (atomic_sub_and_test(skb->truesize, &sk->sk_wmem_alloc))
1673 __sk_free(sk);
1674}
1675
Eric Dumazet9e17f8a2015-11-01 15:36:55 -08001676void skb_set_owner_w(struct sk_buff *skb, struct sock *sk)
1677{
1678 skb_orphan(skb);
1679 skb->sk = sk;
1680#ifdef CONFIG_INET
1681 if (unlikely(!sk_fullsock(sk))) {
1682 skb->destructor = sock_edemux;
1683 sock_hold(sk);
1684 return;
1685 }
1686#endif
1687 skb->destructor = sock_wfree;
1688 skb_set_hash_from_sk(skb, sk);
1689 /*
1690 * We used to take a refcount on sk, but following operation
1691 * is enough to guarantee sk_free() wont free this sock until
1692 * all in-flight packets are completed
1693 */
1694 atomic_add(skb->truesize, &sk->sk_wmem_alloc);
1695}
1696EXPORT_SYMBOL(skb_set_owner_w);
1697
Eric Dumazet1d2077a2016-05-02 10:56:27 -07001698/* This helper is used by netem, as it can hold packets in its
1699 * delay queue. We want to allow the owner socket to send more
1700 * packets, as if they were already TX completed by a typical driver.
1701 * But we also want to keep skb->sk set because some packet schedulers
1702 * rely on it (sch_fq for example). So we set skb->truesize to a small
1703 * amount (1) and decrease sk_wmem_alloc accordingly.
1704 */
Eric Dumazetf2f872f2013-07-30 17:55:08 -07001705void skb_orphan_partial(struct sk_buff *skb)
1706{
Eric Dumazet1d2077a2016-05-02 10:56:27 -07001707 /* If this skb is a TCP pure ACK or already went here,
1708 * we have nothing to do. 2 is already a very small truesize.
1709 */
1710 if (skb->truesize <= 2)
1711 return;
1712
Eric Dumazetf2f872f2013-07-30 17:55:08 -07001713 /* TCP stack sets skb->ooo_okay based on sk_wmem_alloc,
1714 * so we do not completely orphan skb, but transfert all
1715 * accounted bytes but one, to avoid unexpected reorders.
1716 */
1717 if (skb->destructor == sock_wfree
1718#ifdef CONFIG_INET
1719 || skb->destructor == tcp_wfree
1720#endif
1721 ) {
1722 atomic_sub(skb->truesize - 1, &skb->sk->sk_wmem_alloc);
1723 skb->truesize = 1;
1724 } else {
1725 skb_orphan(skb);
1726 }
1727}
1728EXPORT_SYMBOL(skb_orphan_partial);
1729
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001730/*
1731 * Read buffer destructor automatically called from kfree_skb.
Linus Torvalds1da177e2005-04-16 15:20:36 -07001732 */
1733void sock_rfree(struct sk_buff *skb)
1734{
1735 struct sock *sk = skb->sk;
Eric Dumazetd361fd52010-07-10 22:45:17 +00001736 unsigned int len = skb->truesize;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001737
Eric Dumazetd361fd52010-07-10 22:45:17 +00001738 atomic_sub(len, &sk->sk_rmem_alloc);
1739 sk_mem_uncharge(sk, len);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001740}
Eric Dumazet2a915252009-05-27 11:30:05 +00001741EXPORT_SYMBOL(sock_rfree);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001742
Oliver Hartkopp7768eed2015-03-10 19:03:46 +01001743/*
1744 * Buffer destructor for skbs that are not used directly in read or write
1745 * path, e.g. for error handler skbs. Automatically called from kfree_skb.
1746 */
Alexander Duyck62bccb82014-09-04 13:31:35 -04001747void sock_efree(struct sk_buff *skb)
1748{
1749 sock_put(skb->sk);
1750}
1751EXPORT_SYMBOL(sock_efree);
1752
Eric W. Biederman976d02012012-05-23 17:16:53 -06001753kuid_t sock_i_uid(struct sock *sk)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001754{
Eric W. Biederman976d02012012-05-23 17:16:53 -06001755 kuid_t uid;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001756
Eric Dumazetf064af12010-09-22 12:43:39 +00001757 read_lock_bh(&sk->sk_callback_lock);
Eric W. Biederman976d02012012-05-23 17:16:53 -06001758 uid = sk->sk_socket ? SOCK_INODE(sk->sk_socket)->i_uid : GLOBAL_ROOT_UID;
Eric Dumazetf064af12010-09-22 12:43:39 +00001759 read_unlock_bh(&sk->sk_callback_lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001760 return uid;
1761}
Eric Dumazet2a915252009-05-27 11:30:05 +00001762EXPORT_SYMBOL(sock_i_uid);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001763
1764unsigned long sock_i_ino(struct sock *sk)
1765{
1766 unsigned long ino;
1767
Eric Dumazetf064af12010-09-22 12:43:39 +00001768 read_lock_bh(&sk->sk_callback_lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001769 ino = sk->sk_socket ? SOCK_INODE(sk->sk_socket)->i_ino : 0;
Eric Dumazetf064af12010-09-22 12:43:39 +00001770 read_unlock_bh(&sk->sk_callback_lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001771 return ino;
1772}
Eric Dumazet2a915252009-05-27 11:30:05 +00001773EXPORT_SYMBOL(sock_i_ino);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001774
1775/*
1776 * Allocate a skb from the socket's send buffer.
1777 */
Victor Fusco86a76ca2005-07-08 14:57:47 -07001778struct sk_buff *sock_wmalloc(struct sock *sk, unsigned long size, int force,
Al Virodd0fc662005-10-07 07:46:04 +01001779 gfp_t priority)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001780{
1781 if (force || atomic_read(&sk->sk_wmem_alloc) < sk->sk_sndbuf) {
Eric Dumazet2a915252009-05-27 11:30:05 +00001782 struct sk_buff *skb = alloc_skb(size, priority);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001783 if (skb) {
1784 skb_set_owner_w(skb, sk);
1785 return skb;
1786 }
1787 }
1788 return NULL;
1789}
Eric Dumazet2a915252009-05-27 11:30:05 +00001790EXPORT_SYMBOL(sock_wmalloc);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001791
1792/*
Linus Torvalds1da177e2005-04-16 15:20:36 -07001793 * Allocate a memory block from the socket's option memory buffer.
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001794 */
Al Virodd0fc662005-10-07 07:46:04 +01001795void *sock_kmalloc(struct sock *sk, int size, gfp_t priority)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001796{
Eric Dumazet95c96172012-04-15 05:58:06 +00001797 if ((unsigned int)size <= sysctl_optmem_max &&
Linus Torvalds1da177e2005-04-16 15:20:36 -07001798 atomic_read(&sk->sk_omem_alloc) + size < sysctl_optmem_max) {
1799 void *mem;
1800 /* First do the add, to avoid the race if kmalloc
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001801 * might sleep.
Linus Torvalds1da177e2005-04-16 15:20:36 -07001802 */
1803 atomic_add(size, &sk->sk_omem_alloc);
1804 mem = kmalloc(size, priority);
1805 if (mem)
1806 return mem;
1807 atomic_sub(size, &sk->sk_omem_alloc);
1808 }
1809 return NULL;
1810}
Eric Dumazet2a915252009-05-27 11:30:05 +00001811EXPORT_SYMBOL(sock_kmalloc);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001812
Daniel Borkmann79e88652014-11-19 17:13:11 +01001813/* Free an option memory block. Note, we actually want the inline
1814 * here as this allows gcc to detect the nullify and fold away the
1815 * condition entirely.
Linus Torvalds1da177e2005-04-16 15:20:36 -07001816 */
Daniel Borkmann79e88652014-11-19 17:13:11 +01001817static inline void __sock_kfree_s(struct sock *sk, void *mem, int size,
1818 const bool nullify)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001819{
David S. Millere53da5f2014-10-14 17:02:37 -04001820 if (WARN_ON_ONCE(!mem))
1821 return;
Daniel Borkmann79e88652014-11-19 17:13:11 +01001822 if (nullify)
1823 kzfree(mem);
1824 else
1825 kfree(mem);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001826 atomic_sub(size, &sk->sk_omem_alloc);
1827}
Daniel Borkmann79e88652014-11-19 17:13:11 +01001828
1829void sock_kfree_s(struct sock *sk, void *mem, int size)
1830{
1831 __sock_kfree_s(sk, mem, size, false);
1832}
Eric Dumazet2a915252009-05-27 11:30:05 +00001833EXPORT_SYMBOL(sock_kfree_s);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001834
Daniel Borkmann79e88652014-11-19 17:13:11 +01001835void sock_kzfree_s(struct sock *sk, void *mem, int size)
1836{
1837 __sock_kfree_s(sk, mem, size, true);
1838}
1839EXPORT_SYMBOL(sock_kzfree_s);
1840
Linus Torvalds1da177e2005-04-16 15:20:36 -07001841/* It is almost wait_for_tcp_memory minus release_sock/lock_sock.
1842 I think, these locks should be removed for datagram sockets.
1843 */
Eric Dumazet2a915252009-05-27 11:30:05 +00001844static long sock_wait_for_wmem(struct sock *sk, long timeo)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001845{
1846 DEFINE_WAIT(wait);
1847
Eric Dumazet9cd3e072015-11-29 20:03:10 -08001848 sk_clear_bit(SOCKWQ_ASYNC_NOSPACE, sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001849 for (;;) {
1850 if (!timeo)
1851 break;
1852 if (signal_pending(current))
1853 break;
1854 set_bit(SOCK_NOSPACE, &sk->sk_socket->flags);
Eric Dumazetaa395142010-04-20 13:03:51 +00001855 prepare_to_wait(sk_sleep(sk), &wait, TASK_INTERRUPTIBLE);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001856 if (atomic_read(&sk->sk_wmem_alloc) < sk->sk_sndbuf)
1857 break;
1858 if (sk->sk_shutdown & SEND_SHUTDOWN)
1859 break;
1860 if (sk->sk_err)
1861 break;
1862 timeo = schedule_timeout(timeo);
1863 }
Eric Dumazetaa395142010-04-20 13:03:51 +00001864 finish_wait(sk_sleep(sk), &wait);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001865 return timeo;
1866}
1867
1868
1869/*
1870 * Generic send/receive buffer handlers
1871 */
1872
Herbert Xu4cc7f682009-02-04 16:55:54 -08001873struct sk_buff *sock_alloc_send_pskb(struct sock *sk, unsigned long header_len,
1874 unsigned long data_len, int noblock,
Eric Dumazet28d64272013-08-08 14:38:47 -07001875 int *errcode, int max_page_order)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001876{
Eric Dumazet2e4e4412014-09-17 04:49:49 -07001877 struct sk_buff *skb;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001878 long timeo;
1879 int err;
1880
Linus Torvalds1da177e2005-04-16 15:20:36 -07001881 timeo = sock_sndtimeo(sk, noblock);
Eric Dumazet2e4e4412014-09-17 04:49:49 -07001882 for (;;) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07001883 err = sock_error(sk);
1884 if (err != 0)
1885 goto failure;
1886
1887 err = -EPIPE;
1888 if (sk->sk_shutdown & SEND_SHUTDOWN)
1889 goto failure;
1890
Eric Dumazet2e4e4412014-09-17 04:49:49 -07001891 if (sk_wmem_alloc_get(sk) < sk->sk_sndbuf)
1892 break;
Eric Dumazet28d64272013-08-08 14:38:47 -07001893
Eric Dumazet9cd3e072015-11-29 20:03:10 -08001894 sk_set_bit(SOCKWQ_ASYNC_NOSPACE, sk);
Eric Dumazet2e4e4412014-09-17 04:49:49 -07001895 set_bit(SOCK_NOSPACE, &sk->sk_socket->flags);
1896 err = -EAGAIN;
1897 if (!timeo)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001898 goto failure;
Eric Dumazet2e4e4412014-09-17 04:49:49 -07001899 if (signal_pending(current))
1900 goto interrupted;
1901 timeo = sock_wait_for_wmem(sk, timeo);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001902 }
Eric Dumazet2e4e4412014-09-17 04:49:49 -07001903 skb = alloc_skb_with_frags(header_len, data_len, max_page_order,
1904 errcode, sk->sk_allocation);
1905 if (skb)
1906 skb_set_owner_w(skb, sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001907 return skb;
1908
1909interrupted:
1910 err = sock_intr_errno(timeo);
1911failure:
1912 *errcode = err;
1913 return NULL;
1914}
Herbert Xu4cc7f682009-02-04 16:55:54 -08001915EXPORT_SYMBOL(sock_alloc_send_pskb);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001916
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001917struct sk_buff *sock_alloc_send_skb(struct sock *sk, unsigned long size,
Linus Torvalds1da177e2005-04-16 15:20:36 -07001918 int noblock, int *errcode)
1919{
Eric Dumazet28d64272013-08-08 14:38:47 -07001920 return sock_alloc_send_pskb(sk, size, 0, noblock, errcode, 0);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001921}
Eric Dumazet2a915252009-05-27 11:30:05 +00001922EXPORT_SYMBOL(sock_alloc_send_skb);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001923
Willem de Bruijn39771b12016-04-02 23:08:06 -04001924int __sock_cmsg_send(struct sock *sk, struct msghdr *msg, struct cmsghdr *cmsg,
1925 struct sockcm_cookie *sockc)
1926{
Soheil Hassas Yeganeh3dd17e62016-04-02 23:08:09 -04001927 u32 tsflags;
1928
Willem de Bruijn39771b12016-04-02 23:08:06 -04001929 switch (cmsg->cmsg_type) {
1930 case SO_MARK:
1931 if (!ns_capable(sock_net(sk)->user_ns, CAP_NET_ADMIN))
1932 return -EPERM;
1933 if (cmsg->cmsg_len != CMSG_LEN(sizeof(u32)))
1934 return -EINVAL;
1935 sockc->mark = *(u32 *)CMSG_DATA(cmsg);
1936 break;
Soheil Hassas Yeganeh3dd17e62016-04-02 23:08:09 -04001937 case SO_TIMESTAMPING:
1938 if (cmsg->cmsg_len != CMSG_LEN(sizeof(u32)))
1939 return -EINVAL;
1940
1941 tsflags = *(u32 *)CMSG_DATA(cmsg);
1942 if (tsflags & ~SOF_TIMESTAMPING_TX_RECORD_MASK)
1943 return -EINVAL;
1944
1945 sockc->tsflags &= ~SOF_TIMESTAMPING_TX_RECORD_MASK;
1946 sockc->tsflags |= tsflags;
1947 break;
Soheil Hassas Yeganeh779f1ed2016-07-11 16:51:26 -04001948 /* SCM_RIGHTS and SCM_CREDENTIALS are semantically in SOL_UNIX. */
1949 case SCM_RIGHTS:
1950 case SCM_CREDENTIALS:
1951 break;
Willem de Bruijn39771b12016-04-02 23:08:06 -04001952 default:
1953 return -EINVAL;
1954 }
1955 return 0;
1956}
1957EXPORT_SYMBOL(__sock_cmsg_send);
1958
Edward Jeef28ea362015-10-08 14:56:48 -07001959int sock_cmsg_send(struct sock *sk, struct msghdr *msg,
1960 struct sockcm_cookie *sockc)
1961{
1962 struct cmsghdr *cmsg;
Willem de Bruijn39771b12016-04-02 23:08:06 -04001963 int ret;
Edward Jeef28ea362015-10-08 14:56:48 -07001964
1965 for_each_cmsghdr(cmsg, msg) {
1966 if (!CMSG_OK(msg, cmsg))
1967 return -EINVAL;
1968 if (cmsg->cmsg_level != SOL_SOCKET)
1969 continue;
Willem de Bruijn39771b12016-04-02 23:08:06 -04001970 ret = __sock_cmsg_send(sk, msg, cmsg, sockc);
1971 if (ret)
1972 return ret;
Edward Jeef28ea362015-10-08 14:56:48 -07001973 }
1974 return 0;
1975}
1976EXPORT_SYMBOL(sock_cmsg_send);
1977
Eric Dumazet5640f762012-09-23 23:04:42 +00001978/* On 32bit arches, an skb frag is limited to 2^15 */
1979#define SKB_FRAG_PAGE_ORDER get_order(32768)
1980
Eric Dumazet400dfd32013-10-17 16:27:07 -07001981/**
1982 * skb_page_frag_refill - check that a page_frag contains enough room
1983 * @sz: minimum size of the fragment we want to get
1984 * @pfrag: pointer to page_frag
Eric Dumazet82d5e2b2014-09-08 04:00:00 -07001985 * @gfp: priority for memory allocation
Eric Dumazet400dfd32013-10-17 16:27:07 -07001986 *
1987 * Note: While this allocator tries to use high order pages, there is
1988 * no guarantee that allocations succeed. Therefore, @sz MUST be
1989 * less or equal than PAGE_SIZE.
1990 */
Eric Dumazetd9b29382014-08-27 20:49:34 -07001991bool skb_page_frag_refill(unsigned int sz, struct page_frag *pfrag, gfp_t gfp)
Eric Dumazet5640f762012-09-23 23:04:42 +00001992{
Eric Dumazet5640f762012-09-23 23:04:42 +00001993 if (pfrag->page) {
Joonsoo Kimfe896d12016-03-17 14:19:26 -07001994 if (page_ref_count(pfrag->page) == 1) {
Eric Dumazet5640f762012-09-23 23:04:42 +00001995 pfrag->offset = 0;
1996 return true;
1997 }
Eric Dumazet400dfd32013-10-17 16:27:07 -07001998 if (pfrag->offset + sz <= pfrag->size)
Eric Dumazet5640f762012-09-23 23:04:42 +00001999 return true;
2000 put_page(pfrag->page);
2001 }
2002
Eric Dumazetd9b29382014-08-27 20:49:34 -07002003 pfrag->offset = 0;
2004 if (SKB_FRAG_PAGE_ORDER) {
Mel Gormand0164ad2015-11-06 16:28:21 -08002005 /* Avoid direct reclaim but allow kswapd to wake */
2006 pfrag->page = alloc_pages((gfp & ~__GFP_DIRECT_RECLAIM) |
2007 __GFP_COMP | __GFP_NOWARN |
2008 __GFP_NORETRY,
Eric Dumazetd9b29382014-08-27 20:49:34 -07002009 SKB_FRAG_PAGE_ORDER);
Eric Dumazet5640f762012-09-23 23:04:42 +00002010 if (likely(pfrag->page)) {
Eric Dumazetd9b29382014-08-27 20:49:34 -07002011 pfrag->size = PAGE_SIZE << SKB_FRAG_PAGE_ORDER;
Eric Dumazet5640f762012-09-23 23:04:42 +00002012 return true;
2013 }
Eric Dumazetd9b29382014-08-27 20:49:34 -07002014 }
2015 pfrag->page = alloc_page(gfp);
2016 if (likely(pfrag->page)) {
2017 pfrag->size = PAGE_SIZE;
2018 return true;
2019 }
Eric Dumazet400dfd32013-10-17 16:27:07 -07002020 return false;
2021}
2022EXPORT_SYMBOL(skb_page_frag_refill);
2023
2024bool sk_page_frag_refill(struct sock *sk, struct page_frag *pfrag)
2025{
2026 if (likely(skb_page_frag_refill(32U, pfrag, sk->sk_allocation)))
2027 return true;
2028
Eric Dumazet5640f762012-09-23 23:04:42 +00002029 sk_enter_memory_pressure(sk);
2030 sk_stream_moderate_sndbuf(sk);
2031 return false;
2032}
2033EXPORT_SYMBOL(sk_page_frag_refill);
2034
Linus Torvalds1da177e2005-04-16 15:20:36 -07002035static void __lock_sock(struct sock *sk)
Namhyung Kimf39234d2010-09-08 03:48:48 +00002036 __releases(&sk->sk_lock.slock)
2037 __acquires(&sk->sk_lock.slock)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002038{
2039 DEFINE_WAIT(wait);
2040
Stephen Hemmingere71a4782007-04-10 20:10:33 -07002041 for (;;) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07002042 prepare_to_wait_exclusive(&sk->sk_lock.wq, &wait,
2043 TASK_UNINTERRUPTIBLE);
2044 spin_unlock_bh(&sk->sk_lock.slock);
2045 schedule();
2046 spin_lock_bh(&sk->sk_lock.slock);
Stephen Hemmingere71a4782007-04-10 20:10:33 -07002047 if (!sock_owned_by_user(sk))
Linus Torvalds1da177e2005-04-16 15:20:36 -07002048 break;
2049 }
2050 finish_wait(&sk->sk_lock.wq, &wait);
2051}
2052
2053static void __release_sock(struct sock *sk)
Namhyung Kimf39234d2010-09-08 03:48:48 +00002054 __releases(&sk->sk_lock.slock)
2055 __acquires(&sk->sk_lock.slock)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002056{
Eric Dumazet5413d1b2016-04-29 14:16:52 -07002057 struct sk_buff *skb, *next;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002058
Eric Dumazet5413d1b2016-04-29 14:16:52 -07002059 while ((skb = sk->sk_backlog.head) != NULL) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07002060 sk->sk_backlog.head = sk->sk_backlog.tail = NULL;
Eric Dumazet5413d1b2016-04-29 14:16:52 -07002061
2062 spin_unlock_bh(&sk->sk_lock.slock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002063
2064 do {
Eric Dumazet5413d1b2016-04-29 14:16:52 -07002065 next = skb->next;
Eric Dumazete4cbb022012-04-30 16:07:09 +00002066 prefetch(next);
Eric Dumazet7fee2262010-05-11 23:19:48 +00002067 WARN_ON_ONCE(skb_dst_is_noref(skb));
Linus Torvalds1da177e2005-04-16 15:20:36 -07002068 skb->next = NULL;
Peter Zijlstrac57943a2008-10-07 14:18:42 -07002069 sk_backlog_rcv(sk, skb);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002070
Eric Dumazet5413d1b2016-04-29 14:16:52 -07002071 cond_resched();
Linus Torvalds1da177e2005-04-16 15:20:36 -07002072
2073 skb = next;
2074 } while (skb != NULL);
2075
Eric Dumazet5413d1b2016-04-29 14:16:52 -07002076 spin_lock_bh(&sk->sk_lock.slock);
2077 }
Zhu Yi8eae9392010-03-04 18:01:40 +00002078
2079 /*
2080 * Doing the zeroing here guarantee we can not loop forever
2081 * while a wild producer attempts to flood us.
2082 */
2083 sk->sk_backlog.len = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002084}
2085
Eric Dumazetd41a69f2016-04-29 14:16:53 -07002086void __sk_flush_backlog(struct sock *sk)
2087{
2088 spin_lock_bh(&sk->sk_lock.slock);
2089 __release_sock(sk);
2090 spin_unlock_bh(&sk->sk_lock.slock);
2091}
2092
Linus Torvalds1da177e2005-04-16 15:20:36 -07002093/**
2094 * sk_wait_data - wait for data to arrive at sk_receive_queue
Pavel Pisa4dc3b162005-05-01 08:59:25 -07002095 * @sk: sock to wait on
2096 * @timeo: for how long
Sabrina Dubrocadfbafc92015-07-24 18:19:25 +02002097 * @skb: last skb seen on sk_receive_queue
Linus Torvalds1da177e2005-04-16 15:20:36 -07002098 *
2099 * Now socket state including sk->sk_err is changed only under lock,
2100 * hence we may omit checks after joining wait queue.
2101 * We check receive queue before schedule() only as optimization;
2102 * it is very likely that release_sock() added new data.
2103 */
Sabrina Dubrocadfbafc92015-07-24 18:19:25 +02002104int sk_wait_data(struct sock *sk, long *timeo, const struct sk_buff *skb)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002105{
WANG Congd9dc8b02016-11-11 10:20:50 -08002106 DEFINE_WAIT_FUNC(wait, woken_wake_function);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002107 int rc;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002108
WANG Congd9dc8b02016-11-11 10:20:50 -08002109 add_wait_queue(sk_sleep(sk), &wait);
Eric Dumazet9cd3e072015-11-29 20:03:10 -08002110 sk_set_bit(SOCKWQ_ASYNC_WAITDATA, sk);
WANG Congd9dc8b02016-11-11 10:20:50 -08002111 rc = sk_wait_event(sk, timeo, skb_peek_tail(&sk->sk_receive_queue) != skb, &wait);
Eric Dumazet9cd3e072015-11-29 20:03:10 -08002112 sk_clear_bit(SOCKWQ_ASYNC_WAITDATA, sk);
WANG Congd9dc8b02016-11-11 10:20:50 -08002113 remove_wait_queue(sk_sleep(sk), &wait);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002114 return rc;
2115}
Linus Torvalds1da177e2005-04-16 15:20:36 -07002116EXPORT_SYMBOL(sk_wait_data);
2117
Hideo Aoki3ab224b2007-12-31 00:11:19 -08002118/**
Paolo Abenif8c3bf02016-10-21 13:55:45 +02002119 * __sk_mem_raise_allocated - increase memory_allocated
Hideo Aoki3ab224b2007-12-31 00:11:19 -08002120 * @sk: socket
2121 * @size: memory size to allocate
Paolo Abenif8c3bf02016-10-21 13:55:45 +02002122 * @amt: pages to allocate
Hideo Aoki3ab224b2007-12-31 00:11:19 -08002123 * @kind: allocation type
2124 *
Paolo Abenif8c3bf02016-10-21 13:55:45 +02002125 * Similar to __sk_mem_schedule(), but does not update sk_forward_alloc
Hideo Aoki3ab224b2007-12-31 00:11:19 -08002126 */
Paolo Abenif8c3bf02016-10-21 13:55:45 +02002127int __sk_mem_raise_allocated(struct sock *sk, int size, int amt, int kind)
Hideo Aoki3ab224b2007-12-31 00:11:19 -08002128{
2129 struct proto *prot = sk->sk_prot;
Paolo Abenif8c3bf02016-10-21 13:55:45 +02002130 long allocated = sk_memory_allocated_add(sk, amt);
Johannes Weinere8056052016-01-14 15:21:14 -08002131
Johannes Weinerbaac50b2016-01-14 15:21:17 -08002132 if (mem_cgroup_sockets_enabled && sk->sk_memcg &&
2133 !mem_cgroup_charge_skmem(sk->sk_memcg, amt))
Johannes Weinere8056052016-01-14 15:21:14 -08002134 goto suppress_allocation;
Hideo Aoki3ab224b2007-12-31 00:11:19 -08002135
2136 /* Under limit. */
Johannes Weinere8056052016-01-14 15:21:14 -08002137 if (allocated <= sk_prot_mem_limits(sk, 0)) {
Glauber Costa180d8cd2011-12-11 21:47:02 +00002138 sk_leave_memory_pressure(sk);
Hideo Aoki3ab224b2007-12-31 00:11:19 -08002139 return 1;
2140 }
2141
Johannes Weinere8056052016-01-14 15:21:14 -08002142 /* Under pressure. */
2143 if (allocated > sk_prot_mem_limits(sk, 1))
Glauber Costa180d8cd2011-12-11 21:47:02 +00002144 sk_enter_memory_pressure(sk);
Hideo Aoki3ab224b2007-12-31 00:11:19 -08002145
Johannes Weinere8056052016-01-14 15:21:14 -08002146 /* Over hard limit. */
2147 if (allocated > sk_prot_mem_limits(sk, 2))
Hideo Aoki3ab224b2007-12-31 00:11:19 -08002148 goto suppress_allocation;
2149
2150 /* guarantee minimum buffer size under pressure */
2151 if (kind == SK_MEM_RECV) {
2152 if (atomic_read(&sk->sk_rmem_alloc) < prot->sysctl_rmem[0])
2153 return 1;
Glauber Costa180d8cd2011-12-11 21:47:02 +00002154
Hideo Aoki3ab224b2007-12-31 00:11:19 -08002155 } else { /* SK_MEM_SEND */
2156 if (sk->sk_type == SOCK_STREAM) {
2157 if (sk->sk_wmem_queued < prot->sysctl_wmem[0])
2158 return 1;
2159 } else if (atomic_read(&sk->sk_wmem_alloc) <
2160 prot->sysctl_wmem[0])
2161 return 1;
2162 }
2163
Glauber Costa180d8cd2011-12-11 21:47:02 +00002164 if (sk_has_memory_pressure(sk)) {
Eric Dumazet17483762008-11-25 21:16:35 -08002165 int alloc;
2166
Glauber Costa180d8cd2011-12-11 21:47:02 +00002167 if (!sk_under_memory_pressure(sk))
Eric Dumazet17483762008-11-25 21:16:35 -08002168 return 1;
Glauber Costa180d8cd2011-12-11 21:47:02 +00002169 alloc = sk_sockets_allocated_read_positive(sk);
2170 if (sk_prot_mem_limits(sk, 2) > alloc *
Hideo Aoki3ab224b2007-12-31 00:11:19 -08002171 sk_mem_pages(sk->sk_wmem_queued +
2172 atomic_read(&sk->sk_rmem_alloc) +
2173 sk->sk_forward_alloc))
2174 return 1;
2175 }
2176
2177suppress_allocation:
2178
2179 if (kind == SK_MEM_SEND && sk->sk_type == SOCK_STREAM) {
2180 sk_stream_moderate_sndbuf(sk);
2181
2182 /* Fail only if socket is _under_ its sndbuf.
2183 * In this case we cannot block, so that we have to fail.
2184 */
2185 if (sk->sk_wmem_queued + size >= sk->sk_sndbuf)
2186 return 1;
2187 }
2188
Satoru Moriya3847ce32011-06-17 12:00:03 +00002189 trace_sock_exceed_buf_limit(sk, prot, allocated);
2190
Glauber Costa0e90b312012-01-20 04:57:16 +00002191 sk_memory_allocated_sub(sk, amt);
Glauber Costa180d8cd2011-12-11 21:47:02 +00002192
Johannes Weinerbaac50b2016-01-14 15:21:17 -08002193 if (mem_cgroup_sockets_enabled && sk->sk_memcg)
2194 mem_cgroup_uncharge_skmem(sk->sk_memcg, amt);
Johannes Weinere8056052016-01-14 15:21:14 -08002195
Hideo Aoki3ab224b2007-12-31 00:11:19 -08002196 return 0;
2197}
Paolo Abenif8c3bf02016-10-21 13:55:45 +02002198EXPORT_SYMBOL(__sk_mem_raise_allocated);
2199
2200/**
2201 * __sk_mem_schedule - increase sk_forward_alloc and memory_allocated
2202 * @sk: socket
2203 * @size: memory size to allocate
2204 * @kind: allocation type
2205 *
2206 * If kind is SK_MEM_SEND, it means wmem allocation. Otherwise it means
2207 * rmem allocation. This function assumes that protocols which have
2208 * memory_pressure use sk_wmem_queued as write buffer accounting.
2209 */
2210int __sk_mem_schedule(struct sock *sk, int size, int kind)
2211{
2212 int ret, amt = sk_mem_pages(size);
2213
2214 sk->sk_forward_alloc += amt << SK_MEM_QUANTUM_SHIFT;
2215 ret = __sk_mem_raise_allocated(sk, size, amt, kind);
2216 if (!ret)
2217 sk->sk_forward_alloc -= amt << SK_MEM_QUANTUM_SHIFT;
2218 return ret;
2219}
Hideo Aoki3ab224b2007-12-31 00:11:19 -08002220EXPORT_SYMBOL(__sk_mem_schedule);
2221
2222/**
Paolo Abenif8c3bf02016-10-21 13:55:45 +02002223 * __sk_mem_reduce_allocated - reclaim memory_allocated
Hideo Aoki3ab224b2007-12-31 00:11:19 -08002224 * @sk: socket
Paolo Abenif8c3bf02016-10-21 13:55:45 +02002225 * @amount: number of quanta
2226 *
2227 * Similar to __sk_mem_reclaim(), but does not update sk_forward_alloc
Hideo Aoki3ab224b2007-12-31 00:11:19 -08002228 */
Paolo Abenif8c3bf02016-10-21 13:55:45 +02002229void __sk_mem_reduce_allocated(struct sock *sk, int amount)
Hideo Aoki3ab224b2007-12-31 00:11:19 -08002230{
Eric Dumazet1a24e042015-05-15 12:39:25 -07002231 sk_memory_allocated_sub(sk, amount);
Hideo Aoki3ab224b2007-12-31 00:11:19 -08002232
Johannes Weinerbaac50b2016-01-14 15:21:17 -08002233 if (mem_cgroup_sockets_enabled && sk->sk_memcg)
2234 mem_cgroup_uncharge_skmem(sk->sk_memcg, amount);
Johannes Weinere8056052016-01-14 15:21:14 -08002235
Glauber Costa180d8cd2011-12-11 21:47:02 +00002236 if (sk_under_memory_pressure(sk) &&
2237 (sk_memory_allocated(sk) < sk_prot_mem_limits(sk, 0)))
2238 sk_leave_memory_pressure(sk);
Hideo Aoki3ab224b2007-12-31 00:11:19 -08002239}
Paolo Abenif8c3bf02016-10-21 13:55:45 +02002240EXPORT_SYMBOL(__sk_mem_reduce_allocated);
2241
2242/**
2243 * __sk_mem_reclaim - reclaim sk_forward_alloc and memory_allocated
2244 * @sk: socket
2245 * @amount: number of bytes (rounded down to a SK_MEM_QUANTUM multiple)
2246 */
2247void __sk_mem_reclaim(struct sock *sk, int amount)
2248{
2249 amount >>= SK_MEM_QUANTUM_SHIFT;
2250 sk->sk_forward_alloc -= amount << SK_MEM_QUANTUM_SHIFT;
2251 __sk_mem_reduce_allocated(sk, amount);
2252}
Hideo Aoki3ab224b2007-12-31 00:11:19 -08002253EXPORT_SYMBOL(__sk_mem_reclaim);
2254
samanthakumar627d2d62016-04-05 12:41:16 -04002255int sk_set_peek_off(struct sock *sk, int val)
2256{
2257 if (val < 0)
2258 return -EINVAL;
2259
2260 sk->sk_peek_off = val;
2261 return 0;
2262}
2263EXPORT_SYMBOL_GPL(sk_set_peek_off);
Hideo Aoki3ab224b2007-12-31 00:11:19 -08002264
Linus Torvalds1da177e2005-04-16 15:20:36 -07002265/*
2266 * Set of default routines for initialising struct proto_ops when
2267 * the protocol does not support a particular function. In certain
2268 * cases where it makes no sense for a protocol to have a "do nothing"
2269 * function, some default processing is provided.
2270 */
2271
2272int sock_no_bind(struct socket *sock, struct sockaddr *saddr, int len)
2273{
2274 return -EOPNOTSUPP;
2275}
Eric Dumazet2a915252009-05-27 11:30:05 +00002276EXPORT_SYMBOL(sock_no_bind);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002277
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09002278int sock_no_connect(struct socket *sock, struct sockaddr *saddr,
Linus Torvalds1da177e2005-04-16 15:20:36 -07002279 int len, int flags)
2280{
2281 return -EOPNOTSUPP;
2282}
Eric Dumazet2a915252009-05-27 11:30:05 +00002283EXPORT_SYMBOL(sock_no_connect);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002284
2285int sock_no_socketpair(struct socket *sock1, struct socket *sock2)
2286{
2287 return -EOPNOTSUPP;
2288}
Eric Dumazet2a915252009-05-27 11:30:05 +00002289EXPORT_SYMBOL(sock_no_socketpair);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002290
David Howellscdfbabf2017-03-09 08:09:05 +00002291int sock_no_accept(struct socket *sock, struct socket *newsock, int flags,
2292 bool kern)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002293{
2294 return -EOPNOTSUPP;
2295}
Eric Dumazet2a915252009-05-27 11:30:05 +00002296EXPORT_SYMBOL(sock_no_accept);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002297
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09002298int sock_no_getname(struct socket *sock, struct sockaddr *saddr,
Linus Torvalds1da177e2005-04-16 15:20:36 -07002299 int *len, int peer)
2300{
2301 return -EOPNOTSUPP;
2302}
Eric Dumazet2a915252009-05-27 11:30:05 +00002303EXPORT_SYMBOL(sock_no_getname);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002304
Eric Dumazet2a915252009-05-27 11:30:05 +00002305unsigned int sock_no_poll(struct file *file, struct socket *sock, poll_table *pt)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002306{
2307 return 0;
2308}
Eric Dumazet2a915252009-05-27 11:30:05 +00002309EXPORT_SYMBOL(sock_no_poll);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002310
2311int sock_no_ioctl(struct socket *sock, unsigned int cmd, unsigned long arg)
2312{
2313 return -EOPNOTSUPP;
2314}
Eric Dumazet2a915252009-05-27 11:30:05 +00002315EXPORT_SYMBOL(sock_no_ioctl);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002316
2317int sock_no_listen(struct socket *sock, int backlog)
2318{
2319 return -EOPNOTSUPP;
2320}
Eric Dumazet2a915252009-05-27 11:30:05 +00002321EXPORT_SYMBOL(sock_no_listen);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002322
2323int sock_no_shutdown(struct socket *sock, int how)
2324{
2325 return -EOPNOTSUPP;
2326}
Eric Dumazet2a915252009-05-27 11:30:05 +00002327EXPORT_SYMBOL(sock_no_shutdown);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002328
2329int sock_no_setsockopt(struct socket *sock, int level, int optname,
David S. Millerb7058842009-09-30 16:12:20 -07002330 char __user *optval, unsigned int optlen)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002331{
2332 return -EOPNOTSUPP;
2333}
Eric Dumazet2a915252009-05-27 11:30:05 +00002334EXPORT_SYMBOL(sock_no_setsockopt);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002335
2336int sock_no_getsockopt(struct socket *sock, int level, int optname,
2337 char __user *optval, int __user *optlen)
2338{
2339 return -EOPNOTSUPP;
2340}
Eric Dumazet2a915252009-05-27 11:30:05 +00002341EXPORT_SYMBOL(sock_no_getsockopt);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002342
Ying Xue1b784142015-03-02 15:37:48 +08002343int sock_no_sendmsg(struct socket *sock, struct msghdr *m, size_t len)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002344{
2345 return -EOPNOTSUPP;
2346}
Eric Dumazet2a915252009-05-27 11:30:05 +00002347EXPORT_SYMBOL(sock_no_sendmsg);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002348
Ying Xue1b784142015-03-02 15:37:48 +08002349int sock_no_recvmsg(struct socket *sock, struct msghdr *m, size_t len,
2350 int flags)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002351{
2352 return -EOPNOTSUPP;
2353}
Eric Dumazet2a915252009-05-27 11:30:05 +00002354EXPORT_SYMBOL(sock_no_recvmsg);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002355
2356int sock_no_mmap(struct file *file, struct socket *sock, struct vm_area_struct *vma)
2357{
2358 /* Mirror missing mmap method error code */
2359 return -ENODEV;
2360}
Eric Dumazet2a915252009-05-27 11:30:05 +00002361EXPORT_SYMBOL(sock_no_mmap);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002362
2363ssize_t sock_no_sendpage(struct socket *sock, struct page *page, int offset, size_t size, int flags)
2364{
2365 ssize_t res;
2366 struct msghdr msg = {.msg_flags = flags};
2367 struct kvec iov;
2368 char *kaddr = kmap(page);
2369 iov.iov_base = kaddr + offset;
2370 iov.iov_len = size;
2371 res = kernel_sendmsg(sock, &msg, &iov, 1, size);
2372 kunmap(page);
2373 return res;
2374}
Eric Dumazet2a915252009-05-27 11:30:05 +00002375EXPORT_SYMBOL(sock_no_sendpage);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002376
2377/*
2378 * Default Socket Callbacks
2379 */
2380
2381static void sock_def_wakeup(struct sock *sk)
2382{
Eric Dumazet43815482010-04-29 11:01:49 +00002383 struct socket_wq *wq;
2384
2385 rcu_read_lock();
2386 wq = rcu_dereference(sk->sk_wq);
Herbert Xu1ce0bf52015-11-26 13:55:39 +08002387 if (skwq_has_sleeper(wq))
Eric Dumazet43815482010-04-29 11:01:49 +00002388 wake_up_interruptible_all(&wq->wait);
2389 rcu_read_unlock();
Linus Torvalds1da177e2005-04-16 15:20:36 -07002390}
2391
2392static void sock_def_error_report(struct sock *sk)
2393{
Eric Dumazet43815482010-04-29 11:01:49 +00002394 struct socket_wq *wq;
2395
2396 rcu_read_lock();
2397 wq = rcu_dereference(sk->sk_wq);
Herbert Xu1ce0bf52015-11-26 13:55:39 +08002398 if (skwq_has_sleeper(wq))
Eric Dumazet43815482010-04-29 11:01:49 +00002399 wake_up_interruptible_poll(&wq->wait, POLLERR);
Pavel Emelyanov8d8ad9d2007-11-26 20:10:50 +08002400 sk_wake_async(sk, SOCK_WAKE_IO, POLL_ERR);
Eric Dumazet43815482010-04-29 11:01:49 +00002401 rcu_read_unlock();
Linus Torvalds1da177e2005-04-16 15:20:36 -07002402}
2403
David S. Miller676d2362014-04-11 16:15:36 -04002404static void sock_def_readable(struct sock *sk)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002405{
Eric Dumazet43815482010-04-29 11:01:49 +00002406 struct socket_wq *wq;
2407
2408 rcu_read_lock();
2409 wq = rcu_dereference(sk->sk_wq);
Herbert Xu1ce0bf52015-11-26 13:55:39 +08002410 if (skwq_has_sleeper(wq))
Eric Dumazet2c6607c2011-01-06 10:54:29 -08002411 wake_up_interruptible_sync_poll(&wq->wait, POLLIN | POLLPRI |
Davide Libenzi37e55402009-03-31 15:24:21 -07002412 POLLRDNORM | POLLRDBAND);
Pavel Emelyanov8d8ad9d2007-11-26 20:10:50 +08002413 sk_wake_async(sk, SOCK_WAKE_WAITD, POLL_IN);
Eric Dumazet43815482010-04-29 11:01:49 +00002414 rcu_read_unlock();
Linus Torvalds1da177e2005-04-16 15:20:36 -07002415}
2416
2417static void sock_def_write_space(struct sock *sk)
2418{
Eric Dumazet43815482010-04-29 11:01:49 +00002419 struct socket_wq *wq;
2420
2421 rcu_read_lock();
Linus Torvalds1da177e2005-04-16 15:20:36 -07002422
2423 /* Do not wake up a writer until he can make "significant"
2424 * progress. --DaveM
2425 */
Stephen Hemmingere71a4782007-04-10 20:10:33 -07002426 if ((atomic_read(&sk->sk_wmem_alloc) << 1) <= sk->sk_sndbuf) {
Eric Dumazet43815482010-04-29 11:01:49 +00002427 wq = rcu_dereference(sk->sk_wq);
Herbert Xu1ce0bf52015-11-26 13:55:39 +08002428 if (skwq_has_sleeper(wq))
Eric Dumazet43815482010-04-29 11:01:49 +00002429 wake_up_interruptible_sync_poll(&wq->wait, POLLOUT |
Davide Libenzi37e55402009-03-31 15:24:21 -07002430 POLLWRNORM | POLLWRBAND);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002431
2432 /* Should agree with poll, otherwise some programs break */
2433 if (sock_writeable(sk))
Pavel Emelyanov8d8ad9d2007-11-26 20:10:50 +08002434 sk_wake_async(sk, SOCK_WAKE_SPACE, POLL_OUT);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002435 }
2436
Eric Dumazet43815482010-04-29 11:01:49 +00002437 rcu_read_unlock();
Linus Torvalds1da177e2005-04-16 15:20:36 -07002438}
2439
2440static void sock_def_destruct(struct sock *sk)
2441{
Linus Torvalds1da177e2005-04-16 15:20:36 -07002442}
2443
2444void sk_send_sigurg(struct sock *sk)
2445{
2446 if (sk->sk_socket && sk->sk_socket->file)
2447 if (send_sigurg(&sk->sk_socket->file->f_owner))
Pavel Emelyanov8d8ad9d2007-11-26 20:10:50 +08002448 sk_wake_async(sk, SOCK_WAKE_URG, POLL_PRI);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002449}
Eric Dumazet2a915252009-05-27 11:30:05 +00002450EXPORT_SYMBOL(sk_send_sigurg);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002451
2452void sk_reset_timer(struct sock *sk, struct timer_list* timer,
2453 unsigned long expires)
2454{
2455 if (!mod_timer(timer, expires))
2456 sock_hold(sk);
2457}
Linus Torvalds1da177e2005-04-16 15:20:36 -07002458EXPORT_SYMBOL(sk_reset_timer);
2459
2460void sk_stop_timer(struct sock *sk, struct timer_list* timer)
2461{
Ying Xue25cc4ae2013-02-03 20:32:57 +00002462 if (del_timer(timer))
Linus Torvalds1da177e2005-04-16 15:20:36 -07002463 __sock_put(sk);
2464}
Linus Torvalds1da177e2005-04-16 15:20:36 -07002465EXPORT_SYMBOL(sk_stop_timer);
2466
2467void sock_init_data(struct socket *sock, struct sock *sk)
2468{
2469 skb_queue_head_init(&sk->sk_receive_queue);
2470 skb_queue_head_init(&sk->sk_write_queue);
2471 skb_queue_head_init(&sk->sk_error_queue);
2472
2473 sk->sk_send_head = NULL;
2474
2475 init_timer(&sk->sk_timer);
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09002476
Linus Torvalds1da177e2005-04-16 15:20:36 -07002477 sk->sk_allocation = GFP_KERNEL;
2478 sk->sk_rcvbuf = sysctl_rmem_default;
2479 sk->sk_sndbuf = sysctl_wmem_default;
2480 sk->sk_state = TCP_CLOSE;
David S. Miller972692e2008-06-17 22:41:38 -07002481 sk_set_socket(sk, sock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002482
2483 sock_set_flag(sk, SOCK_ZAPPED);
2484
Stephen Hemmingere71a4782007-04-10 20:10:33 -07002485 if (sock) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07002486 sk->sk_type = sock->type;
Eric Dumazet43815482010-04-29 11:01:49 +00002487 sk->sk_wq = sock->wq;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002488 sock->sk = sk;
Lorenzo Colitti86741ec2016-11-04 02:23:41 +09002489 sk->sk_uid = SOCK_INODE(sock)->i_uid;
2490 } else {
Eric Dumazet43815482010-04-29 11:01:49 +00002491 sk->sk_wq = NULL;
Lorenzo Colitti86741ec2016-11-04 02:23:41 +09002492 sk->sk_uid = make_kuid(sock_net(sk)->user_ns, 0);
2493 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07002494
Linus Torvalds1da177e2005-04-16 15:20:36 -07002495 rwlock_init(&sk->sk_callback_lock);
David Howellscdfbabf2017-03-09 08:09:05 +00002496 if (sk->sk_kern_sock)
2497 lockdep_set_class_and_name(
2498 &sk->sk_callback_lock,
2499 af_kern_callback_keys + sk->sk_family,
2500 af_family_kern_clock_key_strings[sk->sk_family]);
2501 else
2502 lockdep_set_class_and_name(
2503 &sk->sk_callback_lock,
Peter Zijlstra443aef02007-07-19 01:49:00 -07002504 af_callback_keys + sk->sk_family,
2505 af_family_clock_key_strings[sk->sk_family]);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002506
2507 sk->sk_state_change = sock_def_wakeup;
2508 sk->sk_data_ready = sock_def_readable;
2509 sk->sk_write_space = sock_def_write_space;
2510 sk->sk_error_report = sock_def_error_report;
2511 sk->sk_destruct = sock_def_destruct;
2512
Eric Dumazet5640f762012-09-23 23:04:42 +00002513 sk->sk_frag.page = NULL;
2514 sk->sk_frag.offset = 0;
Pavel Emelyanovef64a542012-02-21 07:31:34 +00002515 sk->sk_peek_off = -1;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002516
Eric W. Biederman109f6e32010-06-13 03:30:14 +00002517 sk->sk_peer_pid = NULL;
2518 sk->sk_peer_cred = NULL;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002519 sk->sk_write_pending = 0;
2520 sk->sk_rcvlowat = 1;
2521 sk->sk_rcvtimeo = MAX_SCHEDULE_TIMEOUT;
2522 sk->sk_sndtimeo = MAX_SCHEDULE_TIMEOUT;
2523
Eric Dumazetf37f0af2008-04-13 21:39:26 -07002524 sk->sk_stamp = ktime_set(-1L, 0);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002525
Cong Wange0d10952013-08-01 11:10:25 +08002526#ifdef CONFIG_NET_RX_BUSY_POLL
Eliezer Tamir06021292013-06-10 11:39:50 +03002527 sk->sk_napi_id = 0;
Eliezer Tamir64b0dc52013-07-10 17:13:36 +03002528 sk->sk_ll_usec = sysctl_net_busy_read;
Eliezer Tamir06021292013-06-10 11:39:50 +03002529#endif
2530
Eric Dumazet62748f32013-09-24 08:20:52 -07002531 sk->sk_max_pacing_rate = ~0U;
Eric Dumazet7eec4172013-10-08 15:16:00 -07002532 sk->sk_pacing_rate = ~0U;
Eric Dumazet70da2682015-10-08 19:33:21 -07002533 sk->sk_incoming_cpu = -1;
Eric Dumazet4dc6dc72009-07-15 23:13:10 +00002534 /*
2535 * Before updating sk_refcnt, we must commit prior changes to memory
2536 * (Documentation/RCU/rculist_nulls.txt for details)
2537 */
2538 smp_wmb();
Linus Torvalds1da177e2005-04-16 15:20:36 -07002539 atomic_set(&sk->sk_refcnt, 1);
Wang Chen33c732c2007-11-13 20:30:01 -08002540 atomic_set(&sk->sk_drops, 0);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002541}
Eric Dumazet2a915252009-05-27 11:30:05 +00002542EXPORT_SYMBOL(sock_init_data);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002543
Harvey Harrisonb5606c22008-02-13 15:03:16 -08002544void lock_sock_nested(struct sock *sk, int subclass)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002545{
2546 might_sleep();
Ingo Molnara5b5bb92006-07-03 00:25:35 -07002547 spin_lock_bh(&sk->sk_lock.slock);
John Heffnerd2e91172007-09-12 10:44:19 +02002548 if (sk->sk_lock.owned)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002549 __lock_sock(sk);
John Heffnerd2e91172007-09-12 10:44:19 +02002550 sk->sk_lock.owned = 1;
Ingo Molnara5b5bb92006-07-03 00:25:35 -07002551 spin_unlock(&sk->sk_lock.slock);
2552 /*
2553 * The sk_lock has mutex_lock() semantics here:
2554 */
Peter Zijlstrafcc70d52006-11-08 22:44:35 -08002555 mutex_acquire(&sk->sk_lock.dep_map, subclass, 0, _RET_IP_);
Ingo Molnara5b5bb92006-07-03 00:25:35 -07002556 local_bh_enable();
Linus Torvalds1da177e2005-04-16 15:20:36 -07002557}
Peter Zijlstrafcc70d52006-11-08 22:44:35 -08002558EXPORT_SYMBOL(lock_sock_nested);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002559
Harvey Harrisonb5606c22008-02-13 15:03:16 -08002560void release_sock(struct sock *sk)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002561{
Ingo Molnara5b5bb92006-07-03 00:25:35 -07002562 spin_lock_bh(&sk->sk_lock.slock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002563 if (sk->sk_backlog.tail)
2564 __release_sock(sk);
Eric Dumazet46d3cea2012-07-11 05:50:31 +00002565
Eric Dumazetc3f9b012014-03-10 09:50:11 -07002566 /* Warning : release_cb() might need to release sk ownership,
2567 * ie call sock_release_ownership(sk) before us.
2568 */
Eric Dumazet46d3cea2012-07-11 05:50:31 +00002569 if (sk->sk_prot->release_cb)
2570 sk->sk_prot->release_cb(sk);
2571
Eric Dumazetc3f9b012014-03-10 09:50:11 -07002572 sock_release_ownership(sk);
Ingo Molnara5b5bb92006-07-03 00:25:35 -07002573 if (waitqueue_active(&sk->sk_lock.wq))
2574 wake_up(&sk->sk_lock.wq);
2575 spin_unlock_bh(&sk->sk_lock.slock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002576}
2577EXPORT_SYMBOL(release_sock);
2578
Eric Dumazet8a74ad62010-05-26 19:20:18 +00002579/**
2580 * lock_sock_fast - fast version of lock_sock
2581 * @sk: socket
2582 *
2583 * This version should be used for very small section, where process wont block
2584 * return false if fast path is taken
2585 * sk_lock.slock locked, owned = 0, BH disabled
2586 * return true if slow path is taken
2587 * sk_lock.slock unlocked, owned = 1, BH enabled
2588 */
2589bool lock_sock_fast(struct sock *sk)
2590{
2591 might_sleep();
2592 spin_lock_bh(&sk->sk_lock.slock);
2593
2594 if (!sk->sk_lock.owned)
2595 /*
2596 * Note : We must disable BH
2597 */
2598 return false;
2599
2600 __lock_sock(sk);
2601 sk->sk_lock.owned = 1;
2602 spin_unlock(&sk->sk_lock.slock);
2603 /*
2604 * The sk_lock has mutex_lock() semantics here:
2605 */
2606 mutex_acquire(&sk->sk_lock.dep_map, 0, 0, _RET_IP_);
2607 local_bh_enable();
2608 return true;
2609}
2610EXPORT_SYMBOL(lock_sock_fast);
2611
Linus Torvalds1da177e2005-04-16 15:20:36 -07002612int sock_get_timestamp(struct sock *sk, struct timeval __user *userstamp)
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09002613{
Eric Dumazetb7aa0bf2007-04-19 16:16:32 -07002614 struct timeval tv;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002615 if (!sock_flag(sk, SOCK_TIMESTAMP))
Patrick Ohly20d49472009-02-12 05:03:38 +00002616 sock_enable_timestamp(sk, SOCK_TIMESTAMP);
Eric Dumazetb7aa0bf2007-04-19 16:16:32 -07002617 tv = ktime_to_timeval(sk->sk_stamp);
2618 if (tv.tv_sec == -1)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002619 return -ENOENT;
Eric Dumazetb7aa0bf2007-04-19 16:16:32 -07002620 if (tv.tv_sec == 0) {
2621 sk->sk_stamp = ktime_get_real();
2622 tv = ktime_to_timeval(sk->sk_stamp);
2623 }
2624 return copy_to_user(userstamp, &tv, sizeof(tv)) ? -EFAULT : 0;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09002625}
Linus Torvalds1da177e2005-04-16 15:20:36 -07002626EXPORT_SYMBOL(sock_get_timestamp);
2627
Eric Dumazetae40eb12007-03-18 17:33:16 -07002628int sock_get_timestampns(struct sock *sk, struct timespec __user *userstamp)
2629{
2630 struct timespec ts;
2631 if (!sock_flag(sk, SOCK_TIMESTAMP))
Patrick Ohly20d49472009-02-12 05:03:38 +00002632 sock_enable_timestamp(sk, SOCK_TIMESTAMP);
Eric Dumazetae40eb12007-03-18 17:33:16 -07002633 ts = ktime_to_timespec(sk->sk_stamp);
2634 if (ts.tv_sec == -1)
2635 return -ENOENT;
2636 if (ts.tv_sec == 0) {
2637 sk->sk_stamp = ktime_get_real();
2638 ts = ktime_to_timespec(sk->sk_stamp);
2639 }
2640 return copy_to_user(userstamp, &ts, sizeof(ts)) ? -EFAULT : 0;
2641}
2642EXPORT_SYMBOL(sock_get_timestampns);
2643
Patrick Ohly20d49472009-02-12 05:03:38 +00002644void sock_enable_timestamp(struct sock *sk, int flag)
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09002645{
Patrick Ohly20d49472009-02-12 05:03:38 +00002646 if (!sock_flag(sk, flag)) {
Eric Dumazet08e29af2011-11-28 12:04:18 +00002647 unsigned long previous_flags = sk->sk_flags;
2648
Patrick Ohly20d49472009-02-12 05:03:38 +00002649 sock_set_flag(sk, flag);
2650 /*
2651 * we just set one of the two flags which require net
2652 * time stamping, but time stamping might have been on
2653 * already because of the other one
2654 */
Hannes Frederic Sowa080a2702015-10-26 13:51:37 +01002655 if (sock_needs_netstamp(sk) &&
2656 !(previous_flags & SK_FLAGS_TIMESTAMP))
Patrick Ohly20d49472009-02-12 05:03:38 +00002657 net_enable_timestamp();
Linus Torvalds1da177e2005-04-16 15:20:36 -07002658 }
2659}
Linus Torvalds1da177e2005-04-16 15:20:36 -07002660
Richard Cochrancb820f82013-07-19 19:40:09 +02002661int sock_recv_errqueue(struct sock *sk, struct msghdr *msg, int len,
2662 int level, int type)
2663{
2664 struct sock_exterr_skb *serr;
Willem de Bruijn364a9e92014-08-31 21:30:27 -04002665 struct sk_buff *skb;
Richard Cochrancb820f82013-07-19 19:40:09 +02002666 int copied, err;
2667
2668 err = -EAGAIN;
Willem de Bruijn364a9e92014-08-31 21:30:27 -04002669 skb = sock_dequeue_err_skb(sk);
Richard Cochrancb820f82013-07-19 19:40:09 +02002670 if (skb == NULL)
2671 goto out;
2672
2673 copied = skb->len;
2674 if (copied > len) {
2675 msg->msg_flags |= MSG_TRUNC;
2676 copied = len;
2677 }
David S. Miller51f3d022014-11-05 16:46:40 -05002678 err = skb_copy_datagram_msg(skb, 0, msg, copied);
Richard Cochrancb820f82013-07-19 19:40:09 +02002679 if (err)
2680 goto out_free_skb;
2681
2682 sock_recv_timestamp(msg, sk, skb);
2683
2684 serr = SKB_EXT_ERR(skb);
2685 put_cmsg(msg, level, type, sizeof(serr->ee), &serr->ee);
2686
2687 msg->msg_flags |= MSG_ERRQUEUE;
2688 err = copied;
2689
Richard Cochrancb820f82013-07-19 19:40:09 +02002690out_free_skb:
2691 kfree_skb(skb);
2692out:
2693 return err;
2694}
2695EXPORT_SYMBOL(sock_recv_errqueue);
2696
Linus Torvalds1da177e2005-04-16 15:20:36 -07002697/*
2698 * Get a socket option on an socket.
2699 *
2700 * FIX: POSIX 1003.1g is very ambiguous here. It states that
2701 * asynchronous errors should be reported by getsockopt. We assume
2702 * this means if you specify SO_ERROR (otherwise whats the point of it).
2703 */
2704int sock_common_getsockopt(struct socket *sock, int level, int optname,
2705 char __user *optval, int __user *optlen)
2706{
2707 struct sock *sk = sock->sk;
2708
2709 return sk->sk_prot->getsockopt(sk, level, optname, optval, optlen);
2710}
Linus Torvalds1da177e2005-04-16 15:20:36 -07002711EXPORT_SYMBOL(sock_common_getsockopt);
2712
Dmitry Mishin3fdadf72006-03-20 22:45:21 -08002713#ifdef CONFIG_COMPAT
Arnaldo Carvalho de Melo543d9cf2006-03-20 22:48:35 -08002714int compat_sock_common_getsockopt(struct socket *sock, int level, int optname,
2715 char __user *optval, int __user *optlen)
Dmitry Mishin3fdadf72006-03-20 22:45:21 -08002716{
2717 struct sock *sk = sock->sk;
2718
Johannes Berg1e51f952007-03-06 13:44:06 -08002719 if (sk->sk_prot->compat_getsockopt != NULL)
Arnaldo Carvalho de Melo543d9cf2006-03-20 22:48:35 -08002720 return sk->sk_prot->compat_getsockopt(sk, level, optname,
2721 optval, optlen);
Dmitry Mishin3fdadf72006-03-20 22:45:21 -08002722 return sk->sk_prot->getsockopt(sk, level, optname, optval, optlen);
2723}
2724EXPORT_SYMBOL(compat_sock_common_getsockopt);
2725#endif
2726
Ying Xue1b784142015-03-02 15:37:48 +08002727int sock_common_recvmsg(struct socket *sock, struct msghdr *msg, size_t size,
2728 int flags)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002729{
2730 struct sock *sk = sock->sk;
2731 int addr_len = 0;
2732 int err;
2733
Ying Xue1b784142015-03-02 15:37:48 +08002734 err = sk->sk_prot->recvmsg(sk, msg, size, flags & MSG_DONTWAIT,
Linus Torvalds1da177e2005-04-16 15:20:36 -07002735 flags & ~MSG_DONTWAIT, &addr_len);
2736 if (err >= 0)
2737 msg->msg_namelen = addr_len;
2738 return err;
2739}
Linus Torvalds1da177e2005-04-16 15:20:36 -07002740EXPORT_SYMBOL(sock_common_recvmsg);
2741
2742/*
2743 * Set socket options on an inet socket.
2744 */
2745int sock_common_setsockopt(struct socket *sock, int level, int optname,
David S. Millerb7058842009-09-30 16:12:20 -07002746 char __user *optval, unsigned int optlen)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002747{
2748 struct sock *sk = sock->sk;
2749
2750 return sk->sk_prot->setsockopt(sk, level, optname, optval, optlen);
2751}
Linus Torvalds1da177e2005-04-16 15:20:36 -07002752EXPORT_SYMBOL(sock_common_setsockopt);
2753
Dmitry Mishin3fdadf72006-03-20 22:45:21 -08002754#ifdef CONFIG_COMPAT
Arnaldo Carvalho de Melo543d9cf2006-03-20 22:48:35 -08002755int compat_sock_common_setsockopt(struct socket *sock, int level, int optname,
David S. Millerb7058842009-09-30 16:12:20 -07002756 char __user *optval, unsigned int optlen)
Dmitry Mishin3fdadf72006-03-20 22:45:21 -08002757{
2758 struct sock *sk = sock->sk;
2759
Arnaldo Carvalho de Melo543d9cf2006-03-20 22:48:35 -08002760 if (sk->sk_prot->compat_setsockopt != NULL)
2761 return sk->sk_prot->compat_setsockopt(sk, level, optname,
2762 optval, optlen);
Dmitry Mishin3fdadf72006-03-20 22:45:21 -08002763 return sk->sk_prot->setsockopt(sk, level, optname, optval, optlen);
2764}
2765EXPORT_SYMBOL(compat_sock_common_setsockopt);
2766#endif
2767
Linus Torvalds1da177e2005-04-16 15:20:36 -07002768void sk_common_release(struct sock *sk)
2769{
2770 if (sk->sk_prot->destroy)
2771 sk->sk_prot->destroy(sk);
2772
2773 /*
2774 * Observation: when sock_common_release is called, processes have
2775 * no access to socket. But net still has.
2776 * Step one, detach it from networking:
2777 *
2778 * A. Remove from hash tables.
2779 */
2780
2781 sk->sk_prot->unhash(sk);
2782
2783 /*
2784 * In this point socket cannot receive new packets, but it is possible
2785 * that some packets are in flight because some CPU runs receiver and
2786 * did hash table lookup before we unhashed socket. They will achieve
2787 * receive queue and will be purged by socket destructor.
2788 *
2789 * Also we still have packets pending on receive queue and probably,
2790 * our own packets waiting in device queues. sock_destroy will drain
2791 * receive queue, but transmitted packets will delay socket destruction
2792 * until the last reference will be released.
2793 */
2794
2795 sock_orphan(sk);
2796
2797 xfrm_sk_free_policy(sk);
2798
Arnaldo Carvalho de Meloe6848972005-08-09 19:45:38 -07002799 sk_refcnt_debug_release(sk);
Eric Dumazet5640f762012-09-23 23:04:42 +00002800
Linus Torvalds1da177e2005-04-16 15:20:36 -07002801 sock_put(sk);
2802}
Linus Torvalds1da177e2005-04-16 15:20:36 -07002803EXPORT_SYMBOL(sk_common_release);
2804
Pavel Emelyanov13ff3d62008-03-28 16:38:17 -07002805#ifdef CONFIG_PROC_FS
2806#define PROTO_INUSE_NR 64 /* should be enough for the first time */
Pavel Emelyanov1338d462008-03-28 16:38:43 -07002807struct prot_inuse {
2808 int val[PROTO_INUSE_NR];
2809};
Pavel Emelyanov13ff3d62008-03-28 16:38:17 -07002810
2811static DECLARE_BITMAP(proto_inuse_idx, PROTO_INUSE_NR);
Pavel Emelyanov70ee1152008-03-31 19:42:16 -07002812
2813#ifdef CONFIG_NET_NS
2814void sock_prot_inuse_add(struct net *net, struct proto *prot, int val)
2815{
Eric Dumazetd6d9ca02010-07-19 10:48:49 +00002816 __this_cpu_add(net->core.inuse->val[prot->inuse_idx], val);
Pavel Emelyanov70ee1152008-03-31 19:42:16 -07002817}
2818EXPORT_SYMBOL_GPL(sock_prot_inuse_add);
2819
2820int sock_prot_inuse_get(struct net *net, struct proto *prot)
2821{
2822 int cpu, idx = prot->inuse_idx;
2823 int res = 0;
2824
2825 for_each_possible_cpu(cpu)
2826 res += per_cpu_ptr(net->core.inuse, cpu)->val[idx];
2827
2828 return res >= 0 ? res : 0;
2829}
2830EXPORT_SYMBOL_GPL(sock_prot_inuse_get);
2831
Alexey Dobriyan2c8c1e72010-01-17 03:35:32 +00002832static int __net_init sock_inuse_init_net(struct net *net)
Pavel Emelyanov70ee1152008-03-31 19:42:16 -07002833{
2834 net->core.inuse = alloc_percpu(struct prot_inuse);
2835 return net->core.inuse ? 0 : -ENOMEM;
2836}
2837
Alexey Dobriyan2c8c1e72010-01-17 03:35:32 +00002838static void __net_exit sock_inuse_exit_net(struct net *net)
Pavel Emelyanov70ee1152008-03-31 19:42:16 -07002839{
2840 free_percpu(net->core.inuse);
2841}
2842
2843static struct pernet_operations net_inuse_ops = {
2844 .init = sock_inuse_init_net,
2845 .exit = sock_inuse_exit_net,
2846};
2847
2848static __init int net_inuse_init(void)
2849{
2850 if (register_pernet_subsys(&net_inuse_ops))
2851 panic("Cannot initialize net inuse counters");
2852
2853 return 0;
2854}
2855
2856core_initcall(net_inuse_init);
2857#else
Pavel Emelyanov1338d462008-03-28 16:38:43 -07002858static DEFINE_PER_CPU(struct prot_inuse, prot_inuse);
2859
Pavel Emelyanovc29a0bc2008-03-31 19:41:46 -07002860void sock_prot_inuse_add(struct net *net, struct proto *prot, int val)
Pavel Emelyanov1338d462008-03-28 16:38:43 -07002861{
Eric Dumazetd6d9ca02010-07-19 10:48:49 +00002862 __this_cpu_add(prot_inuse.val[prot->inuse_idx], val);
Pavel Emelyanov1338d462008-03-28 16:38:43 -07002863}
2864EXPORT_SYMBOL_GPL(sock_prot_inuse_add);
2865
Pavel Emelyanovc29a0bc2008-03-31 19:41:46 -07002866int sock_prot_inuse_get(struct net *net, struct proto *prot)
Pavel Emelyanov1338d462008-03-28 16:38:43 -07002867{
2868 int cpu, idx = prot->inuse_idx;
2869 int res = 0;
2870
2871 for_each_possible_cpu(cpu)
2872 res += per_cpu(prot_inuse, cpu).val[idx];
2873
2874 return res >= 0 ? res : 0;
2875}
2876EXPORT_SYMBOL_GPL(sock_prot_inuse_get);
Pavel Emelyanov70ee1152008-03-31 19:42:16 -07002877#endif
Pavel Emelyanov13ff3d62008-03-28 16:38:17 -07002878
2879static void assign_proto_idx(struct proto *prot)
2880{
2881 prot->inuse_idx = find_first_zero_bit(proto_inuse_idx, PROTO_INUSE_NR);
2882
2883 if (unlikely(prot->inuse_idx == PROTO_INUSE_NR - 1)) {
Joe Perchese005d192012-05-16 19:58:40 +00002884 pr_err("PROTO_INUSE_NR exhausted\n");
Pavel Emelyanov13ff3d62008-03-28 16:38:17 -07002885 return;
2886 }
2887
2888 set_bit(prot->inuse_idx, proto_inuse_idx);
2889}
2890
2891static void release_proto_idx(struct proto *prot)
2892{
2893 if (prot->inuse_idx != PROTO_INUSE_NR - 1)
2894 clear_bit(prot->inuse_idx, proto_inuse_idx);
2895}
2896#else
2897static inline void assign_proto_idx(struct proto *prot)
2898{
2899}
2900
2901static inline void release_proto_idx(struct proto *prot)
2902{
2903}
2904#endif
2905
Eric Dumazet0159dfd2015-03-12 16:44:07 -07002906static void req_prot_cleanup(struct request_sock_ops *rsk_prot)
2907{
2908 if (!rsk_prot)
2909 return;
2910 kfree(rsk_prot->slab_name);
2911 rsk_prot->slab_name = NULL;
Julia Lawalladf78ed2015-09-13 14:15:18 +02002912 kmem_cache_destroy(rsk_prot->slab);
2913 rsk_prot->slab = NULL;
Eric Dumazet0159dfd2015-03-12 16:44:07 -07002914}
2915
2916static int req_prot_init(const struct proto *prot)
2917{
2918 struct request_sock_ops *rsk_prot = prot->rsk_prot;
2919
2920 if (!rsk_prot)
2921 return 0;
2922
2923 rsk_prot->slab_name = kasprintf(GFP_KERNEL, "request_sock_%s",
2924 prot->name);
2925 if (!rsk_prot->slab_name)
2926 return -ENOMEM;
2927
2928 rsk_prot->slab = kmem_cache_create(rsk_prot->slab_name,
2929 rsk_prot->obj_size, 0,
Eric Dumazete96f78a2015-10-03 06:27:28 -07002930 prot->slab_flags, NULL);
Eric Dumazet0159dfd2015-03-12 16:44:07 -07002931
2932 if (!rsk_prot->slab) {
2933 pr_crit("%s: Can't create request sock SLAB cache!\n",
2934 prot->name);
2935 return -ENOMEM;
2936 }
2937 return 0;
2938}
2939
Linus Torvalds1da177e2005-04-16 15:20:36 -07002940int proto_register(struct proto *prot, int alloc_slab)
2941{
Linus Torvalds1da177e2005-04-16 15:20:36 -07002942 if (alloc_slab) {
2943 prot->slab = kmem_cache_create(prot->name, prot->obj_size, 0,
Eric Dumazet271b72c2008-10-29 02:11:14 -07002944 SLAB_HWCACHE_ALIGN | prot->slab_flags,
2945 NULL);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002946
2947 if (prot->slab == NULL) {
Joe Perchese005d192012-05-16 19:58:40 +00002948 pr_crit("%s: Can't create sock SLAB cache!\n",
2949 prot->name);
Pavel Emelyanov60e76632008-03-28 16:39:10 -07002950 goto out;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002951 }
Arnaldo Carvalho de Melo2e6599c2005-06-18 22:46:52 -07002952
Eric Dumazet0159dfd2015-03-12 16:44:07 -07002953 if (req_prot_init(prot))
2954 goto out_free_request_sock_slab;
Arnaldo Carvalho de Melo8feaf0c02005-08-09 20:09:30 -07002955
Arnaldo Carvalho de Melo6d6ee432005-12-13 23:25:19 -08002956 if (prot->twsk_prot != NULL) {
Alexey Dobriyanfaf23422010-02-17 09:34:12 +00002957 prot->twsk_prot->twsk_slab_name = kasprintf(GFP_KERNEL, "tw_sock_%s", prot->name);
Arnaldo Carvalho de Melo8feaf0c02005-08-09 20:09:30 -07002958
Catalin Marinas7e56b5d2008-11-21 16:45:22 -08002959 if (prot->twsk_prot->twsk_slab_name == NULL)
Arnaldo Carvalho de Melo8feaf0c02005-08-09 20:09:30 -07002960 goto out_free_request_sock_slab;
2961
Arnaldo Carvalho de Melo6d6ee432005-12-13 23:25:19 -08002962 prot->twsk_prot->twsk_slab =
Catalin Marinas7e56b5d2008-11-21 16:45:22 -08002963 kmem_cache_create(prot->twsk_prot->twsk_slab_name,
Arnaldo Carvalho de Melo6d6ee432005-12-13 23:25:19 -08002964 prot->twsk_prot->twsk_obj_size,
Eric Dumazet3ab5aee2008-11-16 19:40:17 -08002965 0,
Eric Dumazet52db70d2015-04-10 06:07:18 -07002966 prot->slab_flags,
Paul Mundt20c2df82007-07-20 10:11:58 +09002967 NULL);
Arnaldo Carvalho de Melo6d6ee432005-12-13 23:25:19 -08002968 if (prot->twsk_prot->twsk_slab == NULL)
Arnaldo Carvalho de Melo8feaf0c02005-08-09 20:09:30 -07002969 goto out_free_timewait_sock_slab_name;
2970 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07002971 }
2972
Glauber Costa36b77a52011-12-16 00:51:59 +00002973 mutex_lock(&proto_list_mutex);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002974 list_add(&prot->node, &proto_list);
Pavel Emelyanov13ff3d62008-03-28 16:38:17 -07002975 assign_proto_idx(prot);
Glauber Costa36b77a52011-12-16 00:51:59 +00002976 mutex_unlock(&proto_list_mutex);
Pavel Emelyanovb733c002007-11-07 02:23:38 -08002977 return 0;
2978
Arnaldo Carvalho de Melo8feaf0c02005-08-09 20:09:30 -07002979out_free_timewait_sock_slab_name:
Catalin Marinas7e56b5d2008-11-21 16:45:22 -08002980 kfree(prot->twsk_prot->twsk_slab_name);
Arnaldo Carvalho de Melo8feaf0c02005-08-09 20:09:30 -07002981out_free_request_sock_slab:
Eric Dumazet0159dfd2015-03-12 16:44:07 -07002982 req_prot_cleanup(prot->rsk_prot);
2983
Arnaldo Carvalho de Melo2e6599c2005-06-18 22:46:52 -07002984 kmem_cache_destroy(prot->slab);
2985 prot->slab = NULL;
Pavel Emelyanovb733c002007-11-07 02:23:38 -08002986out:
2987 return -ENOBUFS;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002988}
Linus Torvalds1da177e2005-04-16 15:20:36 -07002989EXPORT_SYMBOL(proto_register);
2990
2991void proto_unregister(struct proto *prot)
2992{
Glauber Costa36b77a52011-12-16 00:51:59 +00002993 mutex_lock(&proto_list_mutex);
Pavel Emelyanov13ff3d62008-03-28 16:38:17 -07002994 release_proto_idx(prot);
Patrick McHardy0a3f4352005-09-06 19:47:50 -07002995 list_del(&prot->node);
Glauber Costa36b77a52011-12-16 00:51:59 +00002996 mutex_unlock(&proto_list_mutex);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002997
Julia Lawalladf78ed2015-09-13 14:15:18 +02002998 kmem_cache_destroy(prot->slab);
2999 prot->slab = NULL;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003000
Eric Dumazet0159dfd2015-03-12 16:44:07 -07003001 req_prot_cleanup(prot->rsk_prot);
Arnaldo Carvalho de Melo2e6599c2005-06-18 22:46:52 -07003002
Arnaldo Carvalho de Melo6d6ee432005-12-13 23:25:19 -08003003 if (prot->twsk_prot != NULL && prot->twsk_prot->twsk_slab != NULL) {
Arnaldo Carvalho de Melo6d6ee432005-12-13 23:25:19 -08003004 kmem_cache_destroy(prot->twsk_prot->twsk_slab);
Catalin Marinas7e56b5d2008-11-21 16:45:22 -08003005 kfree(prot->twsk_prot->twsk_slab_name);
Arnaldo Carvalho de Melo6d6ee432005-12-13 23:25:19 -08003006 prot->twsk_prot->twsk_slab = NULL;
Arnaldo Carvalho de Melo8feaf0c02005-08-09 20:09:30 -07003007 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07003008}
Linus Torvalds1da177e2005-04-16 15:20:36 -07003009EXPORT_SYMBOL(proto_unregister);
3010
3011#ifdef CONFIG_PROC_FS
Linus Torvalds1da177e2005-04-16 15:20:36 -07003012static void *proto_seq_start(struct seq_file *seq, loff_t *pos)
Glauber Costa36b77a52011-12-16 00:51:59 +00003013 __acquires(proto_list_mutex)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003014{
Glauber Costa36b77a52011-12-16 00:51:59 +00003015 mutex_lock(&proto_list_mutex);
Pavel Emelianov60f04382007-07-09 13:15:14 -07003016 return seq_list_start_head(&proto_list, *pos);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003017}
3018
3019static void *proto_seq_next(struct seq_file *seq, void *v, loff_t *pos)
3020{
Pavel Emelianov60f04382007-07-09 13:15:14 -07003021 return seq_list_next(v, &proto_list, pos);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003022}
3023
3024static void proto_seq_stop(struct seq_file *seq, void *v)
Glauber Costa36b77a52011-12-16 00:51:59 +00003025 __releases(proto_list_mutex)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003026{
Glauber Costa36b77a52011-12-16 00:51:59 +00003027 mutex_unlock(&proto_list_mutex);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003028}
3029
3030static char proto_method_implemented(const void *method)
3031{
3032 return method == NULL ? 'n' : 'y';
3033}
Glauber Costa180d8cd2011-12-11 21:47:02 +00003034static long sock_prot_memory_allocated(struct proto *proto)
3035{
Jeffrin Josecb75a362012-04-25 19:17:29 +05303036 return proto->memory_allocated != NULL ? proto_memory_allocated(proto) : -1L;
Glauber Costa180d8cd2011-12-11 21:47:02 +00003037}
3038
3039static char *sock_prot_memory_pressure(struct proto *proto)
3040{
3041 return proto->memory_pressure != NULL ?
3042 proto_memory_pressure(proto) ? "yes" : "no" : "NI";
3043}
Linus Torvalds1da177e2005-04-16 15:20:36 -07003044
3045static void proto_seq_printf(struct seq_file *seq, struct proto *proto)
3046{
Glauber Costa180d8cd2011-12-11 21:47:02 +00003047
Eric Dumazet8d987e52010-11-09 23:24:26 +00003048 seq_printf(seq, "%-9s %4u %6d %6ld %-3s %6u %-3s %-10s "
Linus Torvalds1da177e2005-04-16 15:20:36 -07003049 "%2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c\n",
3050 proto->name,
3051 proto->obj_size,
Eric Dumazet14e943d2008-11-19 15:14:01 -08003052 sock_prot_inuse_get(seq_file_net(seq), proto),
Glauber Costa180d8cd2011-12-11 21:47:02 +00003053 sock_prot_memory_allocated(proto),
3054 sock_prot_memory_pressure(proto),
Linus Torvalds1da177e2005-04-16 15:20:36 -07003055 proto->max_header,
3056 proto->slab == NULL ? "no" : "yes",
3057 module_name(proto->owner),
3058 proto_method_implemented(proto->close),
3059 proto_method_implemented(proto->connect),
3060 proto_method_implemented(proto->disconnect),
3061 proto_method_implemented(proto->accept),
3062 proto_method_implemented(proto->ioctl),
3063 proto_method_implemented(proto->init),
3064 proto_method_implemented(proto->destroy),
3065 proto_method_implemented(proto->shutdown),
3066 proto_method_implemented(proto->setsockopt),
3067 proto_method_implemented(proto->getsockopt),
3068 proto_method_implemented(proto->sendmsg),
3069 proto_method_implemented(proto->recvmsg),
3070 proto_method_implemented(proto->sendpage),
3071 proto_method_implemented(proto->bind),
3072 proto_method_implemented(proto->backlog_rcv),
3073 proto_method_implemented(proto->hash),
3074 proto_method_implemented(proto->unhash),
3075 proto_method_implemented(proto->get_port),
3076 proto_method_implemented(proto->enter_memory_pressure));
3077}
3078
3079static int proto_seq_show(struct seq_file *seq, void *v)
3080{
Pavel Emelianov60f04382007-07-09 13:15:14 -07003081 if (v == &proto_list)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003082 seq_printf(seq, "%-9s %-4s %-8s %-6s %-5s %-7s %-4s %-10s %s",
3083 "protocol",
3084 "size",
3085 "sockets",
3086 "memory",
3087 "press",
3088 "maxhdr",
3089 "slab",
3090 "module",
3091 "cl co di ac io in de sh ss gs se re sp bi br ha uh gp em\n");
3092 else
Pavel Emelianov60f04382007-07-09 13:15:14 -07003093 proto_seq_printf(seq, list_entry(v, struct proto, node));
Linus Torvalds1da177e2005-04-16 15:20:36 -07003094 return 0;
3095}
3096
Stephen Hemmingerf6908082007-03-12 14:34:29 -07003097static const struct seq_operations proto_seq_ops = {
Linus Torvalds1da177e2005-04-16 15:20:36 -07003098 .start = proto_seq_start,
3099 .next = proto_seq_next,
3100 .stop = proto_seq_stop,
3101 .show = proto_seq_show,
3102};
3103
3104static int proto_seq_open(struct inode *inode, struct file *file)
3105{
Eric Dumazet14e943d2008-11-19 15:14:01 -08003106 return seq_open_net(inode, file, &proto_seq_ops,
3107 sizeof(struct seq_net_private));
Linus Torvalds1da177e2005-04-16 15:20:36 -07003108}
3109
Arjan van de Ven9a321442007-02-12 00:55:35 -08003110static const struct file_operations proto_seq_fops = {
Linus Torvalds1da177e2005-04-16 15:20:36 -07003111 .owner = THIS_MODULE,
3112 .open = proto_seq_open,
3113 .read = seq_read,
3114 .llseek = seq_lseek,
Eric Dumazet14e943d2008-11-19 15:14:01 -08003115 .release = seq_release_net,
3116};
3117
3118static __net_init int proto_init_net(struct net *net)
3119{
Gao fengd4beaa62013-02-18 01:34:54 +00003120 if (!proc_create("protocols", S_IRUGO, net->proc_net, &proto_seq_fops))
Eric Dumazet14e943d2008-11-19 15:14:01 -08003121 return -ENOMEM;
3122
3123 return 0;
3124}
3125
3126static __net_exit void proto_exit_net(struct net *net)
3127{
Gao fengece31ff2013-02-18 01:34:56 +00003128 remove_proc_entry("protocols", net->proc_net);
Eric Dumazet14e943d2008-11-19 15:14:01 -08003129}
3130
3131
3132static __net_initdata struct pernet_operations proto_net_ops = {
3133 .init = proto_init_net,
3134 .exit = proto_exit_net,
Linus Torvalds1da177e2005-04-16 15:20:36 -07003135};
3136
3137static int __init proto_init(void)
3138{
Eric Dumazet14e943d2008-11-19 15:14:01 -08003139 return register_pernet_subsys(&proto_net_ops);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003140}
3141
3142subsys_initcall(proto_init);
3143
3144#endif /* PROC_FS */