blob: 51270238e269290549f9f6e0eb37f660ec533890 [file] [log] [blame]
Linus Torvalds1da177e2005-04-16 15:20:36 -07001/*
2 * INET An implementation of the TCP/IP protocol suite for the LINUX
3 * operating system. INET is implemented using the BSD Socket
4 * interface as the means of communication with the user level.
5 *
6 * Generic socket support routines. Memory allocators, socket lock/release
7 * handler for protocols to use and generic option handler.
8 *
9 *
Jesper Juhl02c30a82005-05-05 16:16:16 -070010 * Authors: Ross Biro
Linus Torvalds1da177e2005-04-16 15:20:36 -070011 * Fred N. van Kempen, <waltje@uWalt.NL.Mugnet.ORG>
12 * Florian La Roche, <flla@stud.uni-sb.de>
13 * Alan Cox, <A.Cox@swansea.ac.uk>
14 *
15 * Fixes:
16 * Alan Cox : Numerous verify_area() problems
17 * Alan Cox : Connecting on a connecting socket
18 * now returns an error for tcp.
19 * Alan Cox : sock->protocol is set correctly.
20 * and is not sometimes left as 0.
21 * Alan Cox : connect handles icmp errors on a
22 * connect properly. Unfortunately there
23 * is a restart syscall nasty there. I
24 * can't match BSD without hacking the C
25 * library. Ideas urgently sought!
26 * Alan Cox : Disallow bind() to addresses that are
27 * not ours - especially broadcast ones!!
28 * Alan Cox : Socket 1024 _IS_ ok for users. (fencepost)
29 * Alan Cox : sock_wfree/sock_rfree don't destroy sockets,
30 * instead they leave that for the DESTROY timer.
31 * Alan Cox : Clean up error flag in accept
32 * Alan Cox : TCP ack handling is buggy, the DESTROY timer
33 * was buggy. Put a remove_sock() in the handler
34 * for memory when we hit 0. Also altered the timer
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +090035 * code. The ACK stuff can wait and needs major
Linus Torvalds1da177e2005-04-16 15:20:36 -070036 * TCP layer surgery.
37 * Alan Cox : Fixed TCP ack bug, removed remove sock
38 * and fixed timer/inet_bh race.
39 * Alan Cox : Added zapped flag for TCP
40 * Alan Cox : Move kfree_skb into skbuff.c and tidied up surplus code
41 * Alan Cox : for new sk_buff allocations wmalloc/rmalloc now call alloc_skb
42 * Alan Cox : kfree_s calls now are kfree_skbmem so we can track skb resources
43 * Alan Cox : Supports socket option broadcast now as does udp. Packet and raw need fixing.
44 * Alan Cox : Added RCVBUF,SNDBUF size setting. It suddenly occurred to me how easy it was so...
45 * Rick Sladkey : Relaxed UDP rules for matching packets.
46 * C.E.Hawkins : IFF_PROMISC/SIOCGHWADDR support
47 * Pauline Middelink : identd support
48 * Alan Cox : Fixed connect() taking signals I think.
49 * Alan Cox : SO_LINGER supported
50 * Alan Cox : Error reporting fixes
51 * Anonymous : inet_create tidied up (sk->reuse setting)
52 * Alan Cox : inet sockets don't set sk->type!
53 * Alan Cox : Split socket option code
54 * Alan Cox : Callbacks
55 * Alan Cox : Nagle flag for Charles & Johannes stuff
56 * Alex : Removed restriction on inet fioctl
57 * Alan Cox : Splitting INET from NET core
58 * Alan Cox : Fixed bogus SO_TYPE handling in getsockopt()
59 * Adam Caldwell : Missing return in SO_DONTROUTE/SO_DEBUG code
60 * Alan Cox : Split IP from generic code
61 * Alan Cox : New kfree_skbmem()
62 * Alan Cox : Make SO_DEBUG superuser only.
63 * Alan Cox : Allow anyone to clear SO_DEBUG
64 * (compatibility fix)
65 * Alan Cox : Added optimistic memory grabbing for AF_UNIX throughput.
66 * Alan Cox : Allocator for a socket is settable.
67 * Alan Cox : SO_ERROR includes soft errors.
68 * Alan Cox : Allow NULL arguments on some SO_ opts
69 * Alan Cox : Generic socket allocation to make hooks
70 * easier (suggested by Craig Metz).
71 * Michael Pall : SO_ERROR returns positive errno again
72 * Steve Whitehouse: Added default destructor to free
73 * protocol private data.
74 * Steve Whitehouse: Added various other default routines
75 * common to several socket families.
76 * Chris Evans : Call suser() check last on F_SETOWN
77 * Jay Schulist : Added SO_ATTACH_FILTER and SO_DETACH_FILTER.
78 * Andi Kleen : Add sock_kmalloc()/sock_kfree_s()
79 * Andi Kleen : Fix write_space callback
80 * Chris Evans : Security fixes - signedness again
81 * Arnaldo C. Melo : cleanups, use skb_queue_purge
82 *
83 * To Fix:
84 *
85 *
86 * This program is free software; you can redistribute it and/or
87 * modify it under the terms of the GNU General Public License
88 * as published by the Free Software Foundation; either version
89 * 2 of the License, or (at your option) any later version.
90 */
91
Joe Perchese005d192012-05-16 19:58:40 +000092#define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
93
Randy Dunlap4fc268d2006-01-11 12:17:47 -080094#include <linux/capability.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070095#include <linux/errno.h>
Richard Cochrancb820f82013-07-19 19:40:09 +020096#include <linux/errqueue.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070097#include <linux/types.h>
98#include <linux/socket.h>
99#include <linux/in.h>
100#include <linux/kernel.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -0700101#include <linux/module.h>
102#include <linux/proc_fs.h>
103#include <linux/seq_file.h>
104#include <linux/sched.h>
105#include <linux/timer.h>
106#include <linux/string.h>
107#include <linux/sockios.h>
108#include <linux/net.h>
109#include <linux/mm.h>
110#include <linux/slab.h>
111#include <linux/interrupt.h>
112#include <linux/poll.h>
113#include <linux/tcp.h>
114#include <linux/init.h>
Al Viroa1f8e7f72006-10-19 16:08:53 -0400115#include <linux/highmem.h>
Eric W. Biederman3f551f92010-06-13 03:28:59 +0000116#include <linux/user_namespace.h>
Ingo Molnarc5905af2012-02-24 08:31:31 +0100117#include <linux/static_key.h>
David S. Miller3969eb32012-01-09 13:44:23 -0800118#include <linux/memcontrol.h>
David S. Miller8c1ae102012-05-03 02:25:55 -0400119#include <linux/prefetch.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -0700120
121#include <asm/uaccess.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -0700122
123#include <linux/netdevice.h>
124#include <net/protocol.h>
125#include <linux/skbuff.h>
Eric W. Biederman457c4cb2007-09-12 12:01:34 +0200126#include <net/net_namespace.h>
Arnaldo Carvalho de Melo2e6599c2005-06-18 22:46:52 -0700127#include <net/request_sock.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -0700128#include <net/sock.h>
Patrick Ohly20d49472009-02-12 05:03:38 +0000129#include <linux/net_tstamp.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -0700130#include <net/xfrm.h>
131#include <linux/ipsec.h>
Herbert Xuf8451722010-05-24 00:12:34 -0700132#include <net/cls_cgroup.h>
Neil Horman5bc14212011-11-22 05:10:51 +0000133#include <net/netprio_cgroup.h>
Craig Gallekeb4cb002015-06-15 11:26:18 -0400134#include <linux/sock_diag.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -0700135
136#include <linux/filter.h>
Craig Gallek538950a2016-01-04 17:41:47 -0500137#include <net/sock_reuseport.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -0700138
Satoru Moriya3847ce32011-06-17 12:00:03 +0000139#include <trace/events/sock.h>
140
Linus Torvalds1da177e2005-04-16 15:20:36 -0700141#ifdef CONFIG_INET
142#include <net/tcp.h>
143#endif
144
Eliezer Tamir076bb0c2013-07-10 17:13:17 +0300145#include <net/busy_poll.h>
Eliezer Tamir06021292013-06-10 11:39:50 +0300146
Glauber Costa36b77a52011-12-16 00:51:59 +0000147static DEFINE_MUTEX(proto_list_mutex);
Glauber Costad1a4c0b2011-12-11 21:47:04 +0000148static LIST_HEAD(proto_list);
149
Eric W. Biedermana3b299d2014-04-23 14:26:56 -0700150/**
151 * sk_ns_capable - General socket capability test
152 * @sk: Socket to use a capability on or through
153 * @user_ns: The user namespace of the capability to use
154 * @cap: The capability to use
155 *
156 * Test to see if the opener of the socket had when the socket was
157 * created and the current process has the capability @cap in the user
158 * namespace @user_ns.
159 */
160bool sk_ns_capable(const struct sock *sk,
161 struct user_namespace *user_ns, int cap)
162{
163 return file_ns_capable(sk->sk_socket->file, user_ns, cap) &&
164 ns_capable(user_ns, cap);
165}
166EXPORT_SYMBOL(sk_ns_capable);
167
168/**
169 * sk_capable - Socket global capability test
170 * @sk: Socket to use a capability on or through
Masanari Iidae793c0f2014-09-04 23:44:36 +0900171 * @cap: The global capability to use
Eric W. Biedermana3b299d2014-04-23 14:26:56 -0700172 *
173 * Test to see if the opener of the socket had when the socket was
174 * created and the current process has the capability @cap in all user
175 * namespaces.
176 */
177bool sk_capable(const struct sock *sk, int cap)
178{
179 return sk_ns_capable(sk, &init_user_ns, cap);
180}
181EXPORT_SYMBOL(sk_capable);
182
183/**
184 * sk_net_capable - Network namespace socket capability test
185 * @sk: Socket to use a capability on or through
186 * @cap: The capability to use
187 *
Masanari Iidae793c0f2014-09-04 23:44:36 +0900188 * Test to see if the opener of the socket had when the socket was created
Eric W. Biedermana3b299d2014-04-23 14:26:56 -0700189 * and the current process has the capability @cap over the network namespace
190 * the socket is a member of.
191 */
192bool sk_net_capable(const struct sock *sk, int cap)
193{
194 return sk_ns_capable(sk, sock_net(sk)->user_ns, cap);
195}
196EXPORT_SYMBOL(sk_net_capable);
197
198
Andrew Mortonc255a452012-07-31 16:43:02 -0700199#ifdef CONFIG_MEMCG_KMEM
Glauber Costa1d62e432012-04-09 19:36:33 -0300200int mem_cgroup_sockets_init(struct mem_cgroup *memcg, struct cgroup_subsys *ss)
Glauber Costad1a4c0b2011-12-11 21:47:04 +0000201{
202 struct proto *proto;
203 int ret = 0;
204
Glauber Costa36b77a52011-12-16 00:51:59 +0000205 mutex_lock(&proto_list_mutex);
Glauber Costad1a4c0b2011-12-11 21:47:04 +0000206 list_for_each_entry(proto, &proto_list, node) {
207 if (proto->init_cgroup) {
Glauber Costa1d62e432012-04-09 19:36:33 -0300208 ret = proto->init_cgroup(memcg, ss);
Glauber Costad1a4c0b2011-12-11 21:47:04 +0000209 if (ret)
210 goto out;
211 }
212 }
213
Glauber Costa36b77a52011-12-16 00:51:59 +0000214 mutex_unlock(&proto_list_mutex);
Glauber Costad1a4c0b2011-12-11 21:47:04 +0000215 return ret;
216out:
217 list_for_each_entry_continue_reverse(proto, &proto_list, node)
218 if (proto->destroy_cgroup)
Glauber Costa1d62e432012-04-09 19:36:33 -0300219 proto->destroy_cgroup(memcg);
Glauber Costa36b77a52011-12-16 00:51:59 +0000220 mutex_unlock(&proto_list_mutex);
Glauber Costad1a4c0b2011-12-11 21:47:04 +0000221 return ret;
222}
223
Glauber Costa1d62e432012-04-09 19:36:33 -0300224void mem_cgroup_sockets_destroy(struct mem_cgroup *memcg)
Glauber Costad1a4c0b2011-12-11 21:47:04 +0000225{
226 struct proto *proto;
227
Glauber Costa36b77a52011-12-16 00:51:59 +0000228 mutex_lock(&proto_list_mutex);
Glauber Costad1a4c0b2011-12-11 21:47:04 +0000229 list_for_each_entry_reverse(proto, &proto_list, node)
230 if (proto->destroy_cgroup)
Glauber Costa1d62e432012-04-09 19:36:33 -0300231 proto->destroy_cgroup(memcg);
Glauber Costa36b77a52011-12-16 00:51:59 +0000232 mutex_unlock(&proto_list_mutex);
Glauber Costad1a4c0b2011-12-11 21:47:04 +0000233}
234#endif
235
Ingo Molnarda21f242006-07-03 00:25:12 -0700236/*
237 * Each address family might have different locking rules, so we have
238 * one slock key per address family:
239 */
Ingo Molnara5b5bb92006-07-03 00:25:35 -0700240static struct lock_class_key af_family_keys[AF_MAX];
241static struct lock_class_key af_family_slock_keys[AF_MAX];
242
stephen hemmingercbda4ea2013-02-22 07:59:10 +0000243#if defined(CONFIG_MEMCG_KMEM)
Ingo Molnarc5905af2012-02-24 08:31:31 +0100244struct static_key memcg_socket_limit_enabled;
Glauber Costae1aab162011-12-11 21:47:03 +0000245EXPORT_SYMBOL(memcg_socket_limit_enabled);
stephen hemmingercbda4ea2013-02-22 07:59:10 +0000246#endif
Glauber Costae1aab162011-12-11 21:47:03 +0000247
Ingo Molnara5b5bb92006-07-03 00:25:35 -0700248/*
249 * Make lock validator output more readable. (we pre-construct these
250 * strings build-time, so that runtime initialization of socket
251 * locks is fast):
252 */
Jan Engelhardt36cbd3d2009-08-05 10:42:58 -0700253static const char *const af_family_key_strings[AF_MAX+1] = {
Ingo Molnara5b5bb92006-07-03 00:25:35 -0700254 "sk_lock-AF_UNSPEC", "sk_lock-AF_UNIX" , "sk_lock-AF_INET" ,
255 "sk_lock-AF_AX25" , "sk_lock-AF_IPX" , "sk_lock-AF_APPLETALK",
256 "sk_lock-AF_NETROM", "sk_lock-AF_BRIDGE" , "sk_lock-AF_ATMPVC" ,
257 "sk_lock-AF_X25" , "sk_lock-AF_INET6" , "sk_lock-AF_ROSE" ,
258 "sk_lock-AF_DECnet", "sk_lock-AF_NETBEUI" , "sk_lock-AF_SECURITY" ,
259 "sk_lock-AF_KEY" , "sk_lock-AF_NETLINK" , "sk_lock-AF_PACKET" ,
260 "sk_lock-AF_ASH" , "sk_lock-AF_ECONET" , "sk_lock-AF_ATMSVC" ,
Andy Grovercbd151b2009-02-26 23:43:19 -0800261 "sk_lock-AF_RDS" , "sk_lock-AF_SNA" , "sk_lock-AF_IRDA" ,
Ingo Molnara5b5bb92006-07-03 00:25:35 -0700262 "sk_lock-AF_PPPOX" , "sk_lock-AF_WANPIPE" , "sk_lock-AF_LLC" ,
Oliver Hartkoppcd05acf2007-12-16 15:59:24 -0800263 "sk_lock-27" , "sk_lock-28" , "sk_lock-AF_CAN" ,
David Howells17926a72007-04-26 15:48:28 -0700264 "sk_lock-AF_TIPC" , "sk_lock-AF_BLUETOOTH", "sk_lock-IUCV" ,
Remi Denis-Courmontbce7b152008-09-22 19:51:15 -0700265 "sk_lock-AF_RXRPC" , "sk_lock-AF_ISDN" , "sk_lock-AF_PHONET" ,
Miloslav Trmač6f107b52010-12-08 14:35:34 +0800266 "sk_lock-AF_IEEE802154", "sk_lock-AF_CAIF" , "sk_lock-AF_ALG" ,
Federico Vaga456db6a2013-05-28 05:02:44 +0000267 "sk_lock-AF_NFC" , "sk_lock-AF_VSOCK" , "sk_lock-AF_MAX"
Ingo Molnara5b5bb92006-07-03 00:25:35 -0700268};
Jan Engelhardt36cbd3d2009-08-05 10:42:58 -0700269static const char *const af_family_slock_key_strings[AF_MAX+1] = {
Ingo Molnara5b5bb92006-07-03 00:25:35 -0700270 "slock-AF_UNSPEC", "slock-AF_UNIX" , "slock-AF_INET" ,
271 "slock-AF_AX25" , "slock-AF_IPX" , "slock-AF_APPLETALK",
272 "slock-AF_NETROM", "slock-AF_BRIDGE" , "slock-AF_ATMPVC" ,
273 "slock-AF_X25" , "slock-AF_INET6" , "slock-AF_ROSE" ,
274 "slock-AF_DECnet", "slock-AF_NETBEUI" , "slock-AF_SECURITY" ,
275 "slock-AF_KEY" , "slock-AF_NETLINK" , "slock-AF_PACKET" ,
276 "slock-AF_ASH" , "slock-AF_ECONET" , "slock-AF_ATMSVC" ,
Andy Grovercbd151b2009-02-26 23:43:19 -0800277 "slock-AF_RDS" , "slock-AF_SNA" , "slock-AF_IRDA" ,
Ingo Molnara5b5bb92006-07-03 00:25:35 -0700278 "slock-AF_PPPOX" , "slock-AF_WANPIPE" , "slock-AF_LLC" ,
Oliver Hartkoppcd05acf2007-12-16 15:59:24 -0800279 "slock-27" , "slock-28" , "slock-AF_CAN" ,
David Howells17926a72007-04-26 15:48:28 -0700280 "slock-AF_TIPC" , "slock-AF_BLUETOOTH", "slock-AF_IUCV" ,
Remi Denis-Courmontbce7b152008-09-22 19:51:15 -0700281 "slock-AF_RXRPC" , "slock-AF_ISDN" , "slock-AF_PHONET" ,
Miloslav Trmač6f107b52010-12-08 14:35:34 +0800282 "slock-AF_IEEE802154", "slock-AF_CAIF" , "slock-AF_ALG" ,
Federico Vaga456db6a2013-05-28 05:02:44 +0000283 "slock-AF_NFC" , "slock-AF_VSOCK" ,"slock-AF_MAX"
Ingo Molnara5b5bb92006-07-03 00:25:35 -0700284};
Jan Engelhardt36cbd3d2009-08-05 10:42:58 -0700285static const char *const af_family_clock_key_strings[AF_MAX+1] = {
Peter Zijlstra443aef02007-07-19 01:49:00 -0700286 "clock-AF_UNSPEC", "clock-AF_UNIX" , "clock-AF_INET" ,
287 "clock-AF_AX25" , "clock-AF_IPX" , "clock-AF_APPLETALK",
288 "clock-AF_NETROM", "clock-AF_BRIDGE" , "clock-AF_ATMPVC" ,
289 "clock-AF_X25" , "clock-AF_INET6" , "clock-AF_ROSE" ,
290 "clock-AF_DECnet", "clock-AF_NETBEUI" , "clock-AF_SECURITY" ,
291 "clock-AF_KEY" , "clock-AF_NETLINK" , "clock-AF_PACKET" ,
292 "clock-AF_ASH" , "clock-AF_ECONET" , "clock-AF_ATMSVC" ,
Andy Grovercbd151b2009-02-26 23:43:19 -0800293 "clock-AF_RDS" , "clock-AF_SNA" , "clock-AF_IRDA" ,
Peter Zijlstra443aef02007-07-19 01:49:00 -0700294 "clock-AF_PPPOX" , "clock-AF_WANPIPE" , "clock-AF_LLC" ,
Oliver Hartkoppb4942af2008-07-23 14:06:04 -0700295 "clock-27" , "clock-28" , "clock-AF_CAN" ,
David Howellse51f8022007-07-21 19:30:16 -0700296 "clock-AF_TIPC" , "clock-AF_BLUETOOTH", "clock-AF_IUCV" ,
Remi Denis-Courmontbce7b152008-09-22 19:51:15 -0700297 "clock-AF_RXRPC" , "clock-AF_ISDN" , "clock-AF_PHONET" ,
Miloslav Trmač6f107b52010-12-08 14:35:34 +0800298 "clock-AF_IEEE802154", "clock-AF_CAIF" , "clock-AF_ALG" ,
Federico Vaga456db6a2013-05-28 05:02:44 +0000299 "clock-AF_NFC" , "clock-AF_VSOCK" , "clock-AF_MAX"
Peter Zijlstra443aef02007-07-19 01:49:00 -0700300};
Ingo Molnarda21f242006-07-03 00:25:12 -0700301
302/*
303 * sk_callback_lock locking rules are per-address-family,
304 * so split the lock classes by using a per-AF key:
305 */
306static struct lock_class_key af_callback_keys[AF_MAX];
307
Linus Torvalds1da177e2005-04-16 15:20:36 -0700308/* Take into consideration the size of the struct sk_buff overhead in the
309 * determination of these values, since that is non-constant across
310 * platforms. This makes socket queueing behavior and performance
311 * not depend upon such differences.
312 */
313#define _SK_MEM_PACKETS 256
Eric Dumazet87fb4b72011-10-13 07:28:54 +0000314#define _SK_MEM_OVERHEAD SKB_TRUESIZE(256)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700315#define SK_WMEM_MAX (_SK_MEM_OVERHEAD * _SK_MEM_PACKETS)
316#define SK_RMEM_MAX (_SK_MEM_OVERHEAD * _SK_MEM_PACKETS)
317
318/* Run time adjustable parameters. */
Brian Haleyab32ea52006-09-22 14:15:41 -0700319__u32 sysctl_wmem_max __read_mostly = SK_WMEM_MAX;
Hans Schillstrom6d8ebc82012-04-30 08:13:50 +0200320EXPORT_SYMBOL(sysctl_wmem_max);
Brian Haleyab32ea52006-09-22 14:15:41 -0700321__u32 sysctl_rmem_max __read_mostly = SK_RMEM_MAX;
Hans Schillstrom6d8ebc82012-04-30 08:13:50 +0200322EXPORT_SYMBOL(sysctl_rmem_max);
Brian Haleyab32ea52006-09-22 14:15:41 -0700323__u32 sysctl_wmem_default __read_mostly = SK_WMEM_MAX;
324__u32 sysctl_rmem_default __read_mostly = SK_RMEM_MAX;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700325
Lucas De Marchi25985ed2011-03-30 22:57:33 -0300326/* Maximal space eaten by iovec or ancillary data plus some space */
Brian Haleyab32ea52006-09-22 14:15:41 -0700327int sysctl_optmem_max __read_mostly = sizeof(unsigned long)*(2*UIO_MAXIOV+512);
Eric Dumazet2a915252009-05-27 11:30:05 +0000328EXPORT_SYMBOL(sysctl_optmem_max);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700329
Willem de Bruijnb245be12015-01-30 13:29:32 -0500330int sysctl_tstamp_allow_data __read_mostly = 1;
331
Mel Gormanc93bdd02012-07-31 16:44:19 -0700332struct static_key memalloc_socks = STATIC_KEY_INIT_FALSE;
333EXPORT_SYMBOL_GPL(memalloc_socks);
334
Mel Gorman7cb02402012-07-31 16:44:16 -0700335/**
336 * sk_set_memalloc - sets %SOCK_MEMALLOC
337 * @sk: socket to set it on
338 *
339 * Set %SOCK_MEMALLOC on a socket for access to emergency reserves.
340 * It's the responsibility of the admin to adjust min_free_kbytes
341 * to meet the requirements
342 */
343void sk_set_memalloc(struct sock *sk)
344{
345 sock_set_flag(sk, SOCK_MEMALLOC);
346 sk->sk_allocation |= __GFP_MEMALLOC;
Mel Gormanc93bdd02012-07-31 16:44:19 -0700347 static_key_slow_inc(&memalloc_socks);
Mel Gorman7cb02402012-07-31 16:44:16 -0700348}
349EXPORT_SYMBOL_GPL(sk_set_memalloc);
350
351void sk_clear_memalloc(struct sock *sk)
352{
353 sock_reset_flag(sk, SOCK_MEMALLOC);
354 sk->sk_allocation &= ~__GFP_MEMALLOC;
Mel Gormanc93bdd02012-07-31 16:44:19 -0700355 static_key_slow_dec(&memalloc_socks);
Mel Gormanc76562b2012-07-31 16:44:41 -0700356
357 /*
358 * SOCK_MEMALLOC is allowed to ignore rmem limits to ensure forward
Mel Gorman5d753612015-06-10 21:02:04 -0400359 * progress of swapping. SOCK_MEMALLOC may be cleared while
360 * it has rmem allocations due to the last swapfile being deactivated
361 * but there is a risk that the socket is unusable due to exceeding
362 * the rmem limits. Reclaim the reserves and obey rmem limits again.
Mel Gormanc76562b2012-07-31 16:44:41 -0700363 */
Mel Gorman5d753612015-06-10 21:02:04 -0400364 sk_mem_reclaim(sk);
Mel Gorman7cb02402012-07-31 16:44:16 -0700365}
366EXPORT_SYMBOL_GPL(sk_clear_memalloc);
367
Mel Gormanb4b9e352012-07-31 16:44:26 -0700368int __sk_backlog_rcv(struct sock *sk, struct sk_buff *skb)
369{
370 int ret;
371 unsigned long pflags = current->flags;
372
373 /* these should have been dropped before queueing */
374 BUG_ON(!sock_flag(sk, SOCK_MEMALLOC));
375
376 current->flags |= PF_MEMALLOC;
377 ret = sk->sk_backlog_rcv(sk, skb);
378 tsk_restore_flags(current, pflags, PF_MEMALLOC);
379
380 return ret;
381}
382EXPORT_SYMBOL(__sk_backlog_rcv);
383
Linus Torvalds1da177e2005-04-16 15:20:36 -0700384static int sock_set_timeout(long *timeo_p, char __user *optval, int optlen)
385{
386 struct timeval tv;
387
388 if (optlen < sizeof(tv))
389 return -EINVAL;
390 if (copy_from_user(&tv, optval, sizeof(tv)))
391 return -EFAULT;
Vasily Averinba780732007-05-24 16:58:54 -0700392 if (tv.tv_usec < 0 || tv.tv_usec >= USEC_PER_SEC)
393 return -EDOM;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700394
Vasily Averinba780732007-05-24 16:58:54 -0700395 if (tv.tv_sec < 0) {
Andrew Morton6f11df82007-07-09 13:16:00 -0700396 static int warned __read_mostly;
397
Vasily Averinba780732007-05-24 16:58:54 -0700398 *timeo_p = 0;
Ilpo Järvinen50aab542008-05-02 16:20:10 -0700399 if (warned < 10 && net_ratelimit()) {
Vasily Averinba780732007-05-24 16:58:54 -0700400 warned++;
Joe Perchese005d192012-05-16 19:58:40 +0000401 pr_info("%s: `%s' (pid %d) tries to set negative timeout\n",
402 __func__, current->comm, task_pid_nr(current));
Ilpo Järvinen50aab542008-05-02 16:20:10 -0700403 }
Vasily Averinba780732007-05-24 16:58:54 -0700404 return 0;
405 }
Linus Torvalds1da177e2005-04-16 15:20:36 -0700406 *timeo_p = MAX_SCHEDULE_TIMEOUT;
407 if (tv.tv_sec == 0 && tv.tv_usec == 0)
408 return 0;
409 if (tv.tv_sec < (MAX_SCHEDULE_TIMEOUT/HZ - 1))
410 *timeo_p = tv.tv_sec*HZ + (tv.tv_usec+(1000000/HZ-1))/(1000000/HZ);
411 return 0;
412}
413
414static void sock_warn_obsolete_bsdism(const char *name)
415{
416 static int warned;
417 static char warncomm[TASK_COMM_LEN];
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +0900418 if (strcmp(warncomm, current->comm) && warned < 5) {
419 strcpy(warncomm, current->comm);
Joe Perchese005d192012-05-16 19:58:40 +0000420 pr_warn("process `%s' is using obsolete %s SO_BSDCOMPAT\n",
421 warncomm, name);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700422 warned++;
423 }
424}
425
Hannes Frederic Sowa080a2702015-10-26 13:51:37 +0100426static bool sock_needs_netstamp(const struct sock *sk)
427{
428 switch (sk->sk_family) {
429 case AF_UNSPEC:
430 case AF_UNIX:
431 return false;
432 default:
433 return true;
434 }
435}
436
Eric Dumazet08e29af2011-11-28 12:04:18 +0000437static void sock_disable_timestamp(struct sock *sk, unsigned long flags)
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +0900438{
Eric Dumazet08e29af2011-11-28 12:04:18 +0000439 if (sk->sk_flags & flags) {
440 sk->sk_flags &= ~flags;
Hannes Frederic Sowa080a2702015-10-26 13:51:37 +0100441 if (sock_needs_netstamp(sk) &&
442 !(sk->sk_flags & SK_FLAGS_TIMESTAMP))
Patrick Ohly20d49472009-02-12 05:03:38 +0000443 net_disable_timestamp();
Linus Torvalds1da177e2005-04-16 15:20:36 -0700444 }
445}
446
447
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800448int sock_queue_rcv_skb(struct sock *sk, struct sk_buff *skb)
449{
Eric Dumazet766e90372009-10-14 20:40:11 -0700450 int err;
Neil Horman3b885782009-10-12 13:26:31 -0700451 unsigned long flags;
452 struct sk_buff_head *list = &sk->sk_receive_queue;
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800453
Eric Dumazet0fd7bac2011-12-21 07:11:44 +0000454 if (atomic_read(&sk->sk_rmem_alloc) >= sk->sk_rcvbuf) {
Eric Dumazet766e90372009-10-14 20:40:11 -0700455 atomic_inc(&sk->sk_drops);
Satoru Moriya3847ce32011-06-17 12:00:03 +0000456 trace_sock_rcvqueue_full(sk, skb);
Eric Dumazet766e90372009-10-14 20:40:11 -0700457 return -ENOMEM;
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800458 }
459
Dmitry Mishinfda9ef52006-08-31 15:28:39 -0700460 err = sk_filter(sk, skb);
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800461 if (err)
Eric Dumazet766e90372009-10-14 20:40:11 -0700462 return err;
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800463
Mel Gormanc76562b2012-07-31 16:44:41 -0700464 if (!sk_rmem_schedule(sk, skb, skb->truesize)) {
Eric Dumazet766e90372009-10-14 20:40:11 -0700465 atomic_inc(&sk->sk_drops);
466 return -ENOBUFS;
Hideo Aoki3ab224b2007-12-31 00:11:19 -0800467 }
468
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800469 skb->dev = NULL;
470 skb_set_owner_r(skb, sk);
David S. Miller49ad9592008-12-17 22:11:38 -0800471
Eric Dumazet7fee2262010-05-11 23:19:48 +0000472 /* we escape from rcu protected region, make sure we dont leak
473 * a norefcounted dst
474 */
475 skb_dst_force(skb);
476
Neil Horman3b885782009-10-12 13:26:31 -0700477 spin_lock_irqsave(&list->lock, flags);
Eyal Birger3bc3b962015-03-01 14:58:30 +0200478 sock_skb_set_dropcount(sk, skb);
Neil Horman3b885782009-10-12 13:26:31 -0700479 __skb_queue_tail(list, skb);
480 spin_unlock_irqrestore(&list->lock, flags);
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800481
482 if (!sock_flag(sk, SOCK_DEAD))
David S. Miller676d2362014-04-11 16:15:36 -0400483 sk->sk_data_ready(sk);
Eric Dumazet766e90372009-10-14 20:40:11 -0700484 return 0;
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800485}
486EXPORT_SYMBOL(sock_queue_rcv_skb);
487
Arnaldo Carvalho de Melo58a5a7b2006-11-16 14:06:06 -0200488int sk_receive_skb(struct sock *sk, struct sk_buff *skb, const int nested)
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800489{
490 int rc = NET_RX_SUCCESS;
491
Dmitry Mishinfda9ef52006-08-31 15:28:39 -0700492 if (sk_filter(sk, skb))
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800493 goto discard_and_relse;
494
495 skb->dev = NULL;
496
Sorin Dumitru274f4822014-07-22 21:16:51 +0300497 if (sk_rcvqueues_full(sk, sk->sk_rcvbuf)) {
Eric Dumazetc3774112010-04-27 15:13:20 -0700498 atomic_inc(&sk->sk_drops);
499 goto discard_and_relse;
500 }
Arnaldo Carvalho de Melo58a5a7b2006-11-16 14:06:06 -0200501 if (nested)
502 bh_lock_sock_nested(sk);
503 else
504 bh_lock_sock(sk);
Ingo Molnara5b5bb92006-07-03 00:25:35 -0700505 if (!sock_owned_by_user(sk)) {
506 /*
507 * trylock + unlock semantics:
508 */
509 mutex_acquire(&sk->sk_lock.dep_map, 0, 1, _RET_IP_);
510
Peter Zijlstrac57943a2008-10-07 14:18:42 -0700511 rc = sk_backlog_rcv(sk, skb);
Ingo Molnara5b5bb92006-07-03 00:25:35 -0700512
513 mutex_release(&sk->sk_lock.dep_map, 1, _RET_IP_);
Eric Dumazetf545a382012-04-22 23:34:26 +0000514 } else if (sk_add_backlog(sk, skb, sk->sk_rcvbuf)) {
Zhu Yi8eae9392010-03-04 18:01:40 +0000515 bh_unlock_sock(sk);
516 atomic_inc(&sk->sk_drops);
517 goto discard_and_relse;
518 }
519
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800520 bh_unlock_sock(sk);
521out:
522 sock_put(sk);
523 return rc;
524discard_and_relse:
525 kfree_skb(skb);
526 goto out;
527}
528EXPORT_SYMBOL(sk_receive_skb);
529
530struct dst_entry *__sk_dst_check(struct sock *sk, u32 cookie)
531{
Eric Dumazetb6c67122010-04-08 23:03:29 +0000532 struct dst_entry *dst = __sk_dst_get(sk);
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800533
534 if (dst && dst->obsolete && dst->ops->check(dst, cookie) == NULL) {
Krishna Kumare022f0b2009-10-19 23:46:20 +0000535 sk_tx_queue_clear(sk);
Stephen Hemmingera9b3cd72011-08-01 16:19:00 +0000536 RCU_INIT_POINTER(sk->sk_dst_cache, NULL);
Denis Vlasenkof0088a52006-03-28 01:08:21 -0800537 dst_release(dst);
538 return NULL;
539 }
540
541 return dst;
542}
543EXPORT_SYMBOL(__sk_dst_check);
544
545struct dst_entry *sk_dst_check(struct sock *sk, u32 cookie)
546{
547 struct dst_entry *dst = sk_dst_get(sk);
548
549 if (dst && dst->obsolete && dst->ops->check(dst, cookie) == NULL) {
550 sk_dst_reset(sk);
551 dst_release(dst);
552 return NULL;
553 }
554
555 return dst;
556}
557EXPORT_SYMBOL(sk_dst_check);
558
Brian Haleyc91f6df2012-11-26 05:21:08 +0000559static int sock_setbindtodevice(struct sock *sk, char __user *optval,
560 int optlen)
David S. Miller48788092007-09-14 16:41:03 -0700561{
562 int ret = -ENOPROTOOPT;
563#ifdef CONFIG_NETDEVICES
YOSHIFUJI Hideaki3b1e0a62008-03-26 02:26:21 +0900564 struct net *net = sock_net(sk);
David S. Miller48788092007-09-14 16:41:03 -0700565 char devname[IFNAMSIZ];
566 int index;
567
568 /* Sorry... */
569 ret = -EPERM;
Eric W. Biederman5e1fccc2012-11-16 03:03:04 +0000570 if (!ns_capable(net->user_ns, CAP_NET_RAW))
David S. Miller48788092007-09-14 16:41:03 -0700571 goto out;
572
573 ret = -EINVAL;
574 if (optlen < 0)
575 goto out;
576
577 /* Bind this socket to a particular device like "eth0",
578 * as specified in the passed interface name. If the
579 * name is "" or the option length is zero the socket
580 * is not bound.
581 */
582 if (optlen > IFNAMSIZ - 1)
583 optlen = IFNAMSIZ - 1;
584 memset(devname, 0, sizeof(devname));
585
586 ret = -EFAULT;
587 if (copy_from_user(devname, optval, optlen))
588 goto out;
589
David S. Miller000ba2e2009-11-05 22:37:11 -0800590 index = 0;
591 if (devname[0] != '\0') {
Eric Dumazetbf8e56b2009-11-05 21:03:39 -0800592 struct net_device *dev;
David S. Miller48788092007-09-14 16:41:03 -0700593
Eric Dumazetbf8e56b2009-11-05 21:03:39 -0800594 rcu_read_lock();
595 dev = dev_get_by_name_rcu(net, devname);
596 if (dev)
597 index = dev->ifindex;
598 rcu_read_unlock();
David S. Miller48788092007-09-14 16:41:03 -0700599 ret = -ENODEV;
600 if (!dev)
601 goto out;
David S. Miller48788092007-09-14 16:41:03 -0700602 }
603
604 lock_sock(sk);
605 sk->sk_bound_dev_if = index;
606 sk_dst_reset(sk);
607 release_sock(sk);
608
609 ret = 0;
610
611out:
612#endif
613
614 return ret;
615}
616
Brian Haleyc91f6df2012-11-26 05:21:08 +0000617static int sock_getbindtodevice(struct sock *sk, char __user *optval,
618 int __user *optlen, int len)
619{
620 int ret = -ENOPROTOOPT;
621#ifdef CONFIG_NETDEVICES
622 struct net *net = sock_net(sk);
Brian Haleyc91f6df2012-11-26 05:21:08 +0000623 char devname[IFNAMSIZ];
Brian Haleyc91f6df2012-11-26 05:21:08 +0000624
625 if (sk->sk_bound_dev_if == 0) {
626 len = 0;
627 goto zero;
628 }
629
630 ret = -EINVAL;
631 if (len < IFNAMSIZ)
632 goto out;
633
Nicolas Schichan5dbe7c12013-06-26 17:23:42 +0200634 ret = netdev_get_name(net, devname, sk->sk_bound_dev_if);
635 if (ret)
Brian Haleyc91f6df2012-11-26 05:21:08 +0000636 goto out;
Brian Haleyc91f6df2012-11-26 05:21:08 +0000637
638 len = strlen(devname) + 1;
639
640 ret = -EFAULT;
641 if (copy_to_user(optval, devname, len))
642 goto out;
643
644zero:
645 ret = -EFAULT;
646 if (put_user(len, optlen))
647 goto out;
648
649 ret = 0;
650
651out:
652#endif
653
654 return ret;
655}
656
Pavel Emelyanovc0ef8772007-11-15 03:03:19 -0800657static inline void sock_valbool_flag(struct sock *sk, int bit, int valbool)
658{
659 if (valbool)
660 sock_set_flag(sk, bit);
661 else
662 sock_reset_flag(sk, bit);
663}
664
hannes@stressinduktion.orgf60e5992015-04-01 17:07:44 +0200665bool sk_mc_loop(struct sock *sk)
666{
667 if (dev_recursion_level())
668 return false;
669 if (!sk)
670 return true;
671 switch (sk->sk_family) {
672 case AF_INET:
673 return inet_sk(sk)->mc_loop;
674#if IS_ENABLED(CONFIG_IPV6)
675 case AF_INET6:
676 return inet6_sk(sk)->mc_loop;
677#endif
678 }
679 WARN_ON(1);
680 return true;
681}
682EXPORT_SYMBOL(sk_mc_loop);
683
Linus Torvalds1da177e2005-04-16 15:20:36 -0700684/*
685 * This is meant for all protocols to use and covers goings on
686 * at the socket level. Everything here is generic.
687 */
688
689int sock_setsockopt(struct socket *sock, int level, int optname,
David S. Millerb7058842009-09-30 16:12:20 -0700690 char __user *optval, unsigned int optlen)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700691{
Eric Dumazet2a915252009-05-27 11:30:05 +0000692 struct sock *sk = sock->sk;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700693 int val;
694 int valbool;
695 struct linger ling;
696 int ret = 0;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +0900697
Linus Torvalds1da177e2005-04-16 15:20:36 -0700698 /*
699 * Options without arguments
700 */
701
David S. Miller48788092007-09-14 16:41:03 -0700702 if (optname == SO_BINDTODEVICE)
Brian Haleyc91f6df2012-11-26 05:21:08 +0000703 return sock_setbindtodevice(sk, optval, optlen);
David S. Miller48788092007-09-14 16:41:03 -0700704
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700705 if (optlen < sizeof(int))
706 return -EINVAL;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +0900707
Linus Torvalds1da177e2005-04-16 15:20:36 -0700708 if (get_user(val, (int __user *)optval))
709 return -EFAULT;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +0900710
Eric Dumazet2a915252009-05-27 11:30:05 +0000711 valbool = val ? 1 : 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700712
713 lock_sock(sk);
714
Eric Dumazet2a915252009-05-27 11:30:05 +0000715 switch (optname) {
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700716 case SO_DEBUG:
Eric Dumazet2a915252009-05-27 11:30:05 +0000717 if (val && !capable(CAP_NET_ADMIN))
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700718 ret = -EACCES;
Eric Dumazet2a915252009-05-27 11:30:05 +0000719 else
Pavel Emelyanovc0ef8772007-11-15 03:03:19 -0800720 sock_valbool_flag(sk, SOCK_DBG, valbool);
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700721 break;
722 case SO_REUSEADDR:
Pavel Emelyanov4a17fd52012-04-19 03:39:36 +0000723 sk->sk_reuse = (valbool ? SK_CAN_REUSE : SK_NO_REUSE);
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700724 break;
Tom Herbert055dc212013-01-22 09:49:50 +0000725 case SO_REUSEPORT:
726 sk->sk_reuseport = valbool;
727 break;
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700728 case SO_TYPE:
Jan Engelhardt49c794e2009-08-04 07:28:28 +0000729 case SO_PROTOCOL:
Jan Engelhardt0d6038e2009-08-04 07:28:29 +0000730 case SO_DOMAIN:
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700731 case SO_ERROR:
732 ret = -ENOPROTOOPT;
733 break;
734 case SO_DONTROUTE:
Pavel Emelyanovc0ef8772007-11-15 03:03:19 -0800735 sock_valbool_flag(sk, SOCK_LOCALROUTE, valbool);
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700736 break;
737 case SO_BROADCAST:
738 sock_valbool_flag(sk, SOCK_BROADCAST, valbool);
739 break;
740 case SO_SNDBUF:
741 /* Don't error on this BSD doesn't and if you think
Eric Dumazet82981932012-04-26 20:07:59 +0000742 * about it this is right. Otherwise apps have to
743 * play 'guess the biggest size' games. RCVBUF/SNDBUF
744 * are treated in BSD as hints
745 */
746 val = min_t(u32, val, sysctl_wmem_max);
Patrick McHardyb0573de2005-08-09 19:30:51 -0700747set_sndbuf:
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700748 sk->sk_userlocks |= SOCK_SNDBUF_LOCK;
Eric Dumazet82981932012-04-26 20:07:59 +0000749 sk->sk_sndbuf = max_t(u32, val * 2, SOCK_MIN_SNDBUF);
750 /* Wake up sending tasks if we upped the value. */
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700751 sk->sk_write_space(sk);
752 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700753
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700754 case SO_SNDBUFFORCE:
755 if (!capable(CAP_NET_ADMIN)) {
756 ret = -EPERM;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700757 break;
758 }
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700759 goto set_sndbuf;
760
761 case SO_RCVBUF:
762 /* Don't error on this BSD doesn't and if you think
Eric Dumazet82981932012-04-26 20:07:59 +0000763 * about it this is right. Otherwise apps have to
764 * play 'guess the biggest size' games. RCVBUF/SNDBUF
765 * are treated in BSD as hints
766 */
767 val = min_t(u32, val, sysctl_rmem_max);
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700768set_rcvbuf:
769 sk->sk_userlocks |= SOCK_RCVBUF_LOCK;
770 /*
771 * We double it on the way in to account for
772 * "struct sk_buff" etc. overhead. Applications
773 * assume that the SO_RCVBUF setting they make will
774 * allow that much actual data to be received on that
775 * socket.
776 *
777 * Applications are unaware that "struct sk_buff" and
778 * other overheads allocate from the receive buffer
779 * during socket buffer allocation.
780 *
781 * And after considering the possible alternatives,
782 * returning the value we actually used in getsockopt
783 * is the most desirable behavior.
784 */
Eric Dumazet82981932012-04-26 20:07:59 +0000785 sk->sk_rcvbuf = max_t(u32, val * 2, SOCK_MIN_RCVBUF);
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700786 break;
787
788 case SO_RCVBUFFORCE:
789 if (!capable(CAP_NET_ADMIN)) {
790 ret = -EPERM;
791 break;
792 }
793 goto set_rcvbuf;
794
795 case SO_KEEPALIVE:
796#ifdef CONFIG_INET
Eric Dumazet3e109862012-09-24 07:00:11 +0000797 if (sk->sk_protocol == IPPROTO_TCP &&
798 sk->sk_type == SOCK_STREAM)
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700799 tcp_set_keepalive(sk, valbool);
800#endif
801 sock_valbool_flag(sk, SOCK_KEEPOPEN, valbool);
802 break;
803
804 case SO_OOBINLINE:
805 sock_valbool_flag(sk, SOCK_URGINLINE, valbool);
806 break;
807
808 case SO_NO_CHECK:
Tom Herbert28448b82014-05-23 08:47:19 -0700809 sk->sk_no_check_tx = valbool;
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700810 break;
811
812 case SO_PRIORITY:
Eric W. Biederman5e1fccc2012-11-16 03:03:04 +0000813 if ((val >= 0 && val <= 6) ||
814 ns_capable(sock_net(sk)->user_ns, CAP_NET_ADMIN))
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700815 sk->sk_priority = val;
816 else
817 ret = -EPERM;
818 break;
819
820 case SO_LINGER:
821 if (optlen < sizeof(ling)) {
822 ret = -EINVAL; /* 1003.1g */
823 break;
824 }
Eric Dumazet2a915252009-05-27 11:30:05 +0000825 if (copy_from_user(&ling, optval, sizeof(ling))) {
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700826 ret = -EFAULT;
827 break;
828 }
829 if (!ling.l_onoff)
830 sock_reset_flag(sk, SOCK_LINGER);
831 else {
832#if (BITS_PER_LONG == 32)
833 if ((unsigned int)ling.l_linger >= MAX_SCHEDULE_TIMEOUT/HZ)
834 sk->sk_lingertime = MAX_SCHEDULE_TIMEOUT;
835 else
836#endif
837 sk->sk_lingertime = (unsigned int)ling.l_linger * HZ;
838 sock_set_flag(sk, SOCK_LINGER);
839 }
840 break;
841
842 case SO_BSDCOMPAT:
843 sock_warn_obsolete_bsdism("setsockopt");
844 break;
845
846 case SO_PASSCRED:
847 if (valbool)
848 set_bit(SOCK_PASSCRED, &sock->flags);
849 else
850 clear_bit(SOCK_PASSCRED, &sock->flags);
851 break;
852
853 case SO_TIMESTAMP:
Eric Dumazet92f37fd2007-03-25 22:14:49 -0700854 case SO_TIMESTAMPNS:
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700855 if (valbool) {
Eric Dumazet92f37fd2007-03-25 22:14:49 -0700856 if (optname == SO_TIMESTAMP)
857 sock_reset_flag(sk, SOCK_RCVTSTAMPNS);
858 else
859 sock_set_flag(sk, SOCK_RCVTSTAMPNS);
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700860 sock_set_flag(sk, SOCK_RCVTSTAMP);
Patrick Ohly20d49472009-02-12 05:03:38 +0000861 sock_enable_timestamp(sk, SOCK_TIMESTAMP);
Eric Dumazet92f37fd2007-03-25 22:14:49 -0700862 } else {
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700863 sock_reset_flag(sk, SOCK_RCVTSTAMP);
Eric Dumazet92f37fd2007-03-25 22:14:49 -0700864 sock_reset_flag(sk, SOCK_RCVTSTAMPNS);
865 }
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700866 break;
867
Patrick Ohly20d49472009-02-12 05:03:38 +0000868 case SO_TIMESTAMPING:
869 if (val & ~SOF_TIMESTAMPING_MASK) {
Rémi Denis-Courmontf249fb72009-07-20 00:47:04 +0000870 ret = -EINVAL;
Patrick Ohly20d49472009-02-12 05:03:38 +0000871 break;
872 }
Willem de Bruijnb245be12015-01-30 13:29:32 -0500873
Willem de Bruijn09c2d252014-08-04 22:11:47 -0400874 if (val & SOF_TIMESTAMPING_OPT_ID &&
Willem de Bruijn4ed2d762014-08-04 22:11:49 -0400875 !(sk->sk_tsflags & SOF_TIMESTAMPING_OPT_ID)) {
WANG Congac5cc972015-12-16 23:39:04 -0800876 if (sk->sk_protocol == IPPROTO_TCP &&
877 sk->sk_type == SOCK_STREAM) {
Willem de Bruijn4ed2d762014-08-04 22:11:49 -0400878 if (sk->sk_state != TCP_ESTABLISHED) {
879 ret = -EINVAL;
880 break;
881 }
882 sk->sk_tskey = tcp_sk(sk)->snd_una;
883 } else {
884 sk->sk_tskey = 0;
885 }
886 }
Willem de Bruijnb9f40e22014-08-04 22:11:46 -0400887 sk->sk_tsflags = val;
Patrick Ohly20d49472009-02-12 05:03:38 +0000888 if (val & SOF_TIMESTAMPING_RX_SOFTWARE)
889 sock_enable_timestamp(sk,
890 SOCK_TIMESTAMPING_RX_SOFTWARE);
891 else
892 sock_disable_timestamp(sk,
Eric Dumazet08e29af2011-11-28 12:04:18 +0000893 (1UL << SOCK_TIMESTAMPING_RX_SOFTWARE));
Patrick Ohly20d49472009-02-12 05:03:38 +0000894 break;
895
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700896 case SO_RCVLOWAT:
897 if (val < 0)
898 val = INT_MAX;
899 sk->sk_rcvlowat = val ? : 1;
900 break;
901
902 case SO_RCVTIMEO:
903 ret = sock_set_timeout(&sk->sk_rcvtimeo, optval, optlen);
904 break;
905
906 case SO_SNDTIMEO:
907 ret = sock_set_timeout(&sk->sk_sndtimeo, optval, optlen);
908 break;
909
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700910 case SO_ATTACH_FILTER:
911 ret = -EINVAL;
912 if (optlen == sizeof(struct sock_fprog)) {
913 struct sock_fprog fprog;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700914
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700915 ret = -EFAULT;
916 if (copy_from_user(&fprog, optval, sizeof(fprog)))
Linus Torvalds1da177e2005-04-16 15:20:36 -0700917 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700918
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700919 ret = sk_attach_filter(&fprog, sk);
920 }
921 break;
922
Alexei Starovoitov89aa0752014-12-01 15:06:35 -0800923 case SO_ATTACH_BPF:
924 ret = -EINVAL;
925 if (optlen == sizeof(u32)) {
926 u32 ufd;
927
928 ret = -EFAULT;
929 if (copy_from_user(&ufd, optval, sizeof(ufd)))
930 break;
931
932 ret = sk_attach_bpf(ufd, sk);
933 }
934 break;
935
Craig Gallek538950a2016-01-04 17:41:47 -0500936 case SO_ATTACH_REUSEPORT_CBPF:
937 ret = -EINVAL;
938 if (optlen == sizeof(struct sock_fprog)) {
939 struct sock_fprog fprog;
940
941 ret = -EFAULT;
942 if (copy_from_user(&fprog, optval, sizeof(fprog)))
943 break;
944
945 ret = sk_reuseport_attach_filter(&fprog, sk);
946 }
947 break;
948
949 case SO_ATTACH_REUSEPORT_EBPF:
950 ret = -EINVAL;
951 if (optlen == sizeof(u32)) {
952 u32 ufd;
953
954 ret = -EFAULT;
955 if (copy_from_user(&ufd, optval, sizeof(ufd)))
956 break;
957
958 ret = sk_reuseport_attach_bpf(ufd, sk);
959 }
960 break;
961
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700962 case SO_DETACH_FILTER:
Pavel Emelyanov55b33322007-10-17 21:21:26 -0700963 ret = sk_detach_filter(sk);
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700964 break;
965
Vincent Bernatd59577b2013-01-16 22:55:49 +0100966 case SO_LOCK_FILTER:
967 if (sock_flag(sk, SOCK_FILTER_LOCKED) && !valbool)
968 ret = -EPERM;
969 else
970 sock_valbool_flag(sk, SOCK_FILTER_LOCKED, valbool);
971 break;
972
Stephen Hemmingere71a4782007-04-10 20:10:33 -0700973 case SO_PASSSEC:
974 if (valbool)
975 set_bit(SOCK_PASSSEC, &sock->flags);
976 else
977 clear_bit(SOCK_PASSSEC, &sock->flags);
978 break;
Laszlo Attila Toth4a19ec52008-01-30 19:08:16 -0800979 case SO_MARK:
Eric W. Biederman5e1fccc2012-11-16 03:03:04 +0000980 if (!ns_capable(sock_net(sk)->user_ns, CAP_NET_ADMIN))
Laszlo Attila Toth4a19ec52008-01-30 19:08:16 -0800981 ret = -EPERM;
Eric Dumazet2a915252009-05-27 11:30:05 +0000982 else
Laszlo Attila Toth4a19ec52008-01-30 19:08:16 -0800983 sk->sk_mark = val;
Laszlo Attila Toth4a19ec52008-01-30 19:08:16 -0800984 break;
Catherine Zhang877ce7c2006-06-29 12:27:47 -0700985
Neil Horman3b885782009-10-12 13:26:31 -0700986 case SO_RXQ_OVFL:
Johannes Berg8083f0f2011-10-07 03:30:20 +0000987 sock_valbool_flag(sk, SOCK_RXQ_OVFL, valbool);
Neil Horman3b885782009-10-12 13:26:31 -0700988 break;
Johannes Berg6e3e9392011-11-09 10:15:42 +0100989
990 case SO_WIFI_STATUS:
991 sock_valbool_flag(sk, SOCK_WIFI_STATUS, valbool);
992 break;
993
Pavel Emelyanovef64a542012-02-21 07:31:34 +0000994 case SO_PEEK_OFF:
995 if (sock->ops->set_peek_off)
Sasha Levin12663bf2013-12-07 17:26:27 -0500996 ret = sock->ops->set_peek_off(sk, val);
Pavel Emelyanovef64a542012-02-21 07:31:34 +0000997 else
998 ret = -EOPNOTSUPP;
999 break;
Ben Greear3bdc0eb2012-02-11 15:39:30 +00001000
1001 case SO_NOFCS:
1002 sock_valbool_flag(sk, SOCK_NOFCS, valbool);
1003 break;
1004
Keller, Jacob E7d4c04f2013-03-28 11:19:25 +00001005 case SO_SELECT_ERR_QUEUE:
1006 sock_valbool_flag(sk, SOCK_SELECT_ERR_QUEUE, valbool);
1007 break;
1008
Cong Wange0d10952013-08-01 11:10:25 +08001009#ifdef CONFIG_NET_RX_BUSY_POLL
Eliezer Tamir64b0dc52013-07-10 17:13:36 +03001010 case SO_BUSY_POLL:
Eliezer Tamirdafcc432013-06-14 16:33:57 +03001011 /* allow unprivileged users to decrease the value */
1012 if ((val > sk->sk_ll_usec) && !capable(CAP_NET_ADMIN))
1013 ret = -EPERM;
1014 else {
1015 if (val < 0)
1016 ret = -EINVAL;
1017 else
1018 sk->sk_ll_usec = val;
1019 }
1020 break;
1021#endif
Eric Dumazet62748f32013-09-24 08:20:52 -07001022
1023 case SO_MAX_PACING_RATE:
1024 sk->sk_max_pacing_rate = val;
1025 sk->sk_pacing_rate = min(sk->sk_pacing_rate,
1026 sk->sk_max_pacing_rate);
1027 break;
1028
Eric Dumazet70da2682015-10-08 19:33:21 -07001029 case SO_INCOMING_CPU:
1030 sk->sk_incoming_cpu = val;
1031 break;
1032
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001033 default:
1034 ret = -ENOPROTOOPT;
1035 break;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001036 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07001037 release_sock(sk);
1038 return ret;
1039}
Eric Dumazet2a915252009-05-27 11:30:05 +00001040EXPORT_SYMBOL(sock_setsockopt);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001041
1042
stephen hemminger8f098982014-01-03 09:17:14 -08001043static void cred_to_ucred(struct pid *pid, const struct cred *cred,
1044 struct ucred *ucred)
Eric W. Biederman3f551f92010-06-13 03:28:59 +00001045{
1046 ucred->pid = pid_vnr(pid);
1047 ucred->uid = ucred->gid = -1;
1048 if (cred) {
1049 struct user_namespace *current_ns = current_user_ns();
1050
Eric W. Biedermanb2e4f542012-05-23 16:39:45 -06001051 ucred->uid = from_kuid_munged(current_ns, cred->euid);
1052 ucred->gid = from_kgid_munged(current_ns, cred->egid);
Eric W. Biederman3f551f92010-06-13 03:28:59 +00001053 }
1054}
1055
Linus Torvalds1da177e2005-04-16 15:20:36 -07001056int sock_getsockopt(struct socket *sock, int level, int optname,
1057 char __user *optval, int __user *optlen)
1058{
1059 struct sock *sk = sock->sk;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001060
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001061 union {
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001062 int val;
1063 struct linger ling;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001064 struct timeval tm;
1065 } v;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001066
H Hartley Sweeten4d0392b2010-01-15 01:08:58 -08001067 int lv = sizeof(int);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001068 int len;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001069
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001070 if (get_user(len, optlen))
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001071 return -EFAULT;
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001072 if (len < 0)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001073 return -EINVAL;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001074
Eugene Teo50fee1d2009-02-23 15:38:41 -08001075 memset(&v, 0, sizeof(v));
Clément Lecignedf0bca02009-02-12 16:59:09 -08001076
Eric Dumazet2a915252009-05-27 11:30:05 +00001077 switch (optname) {
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001078 case SO_DEBUG:
1079 v.val = sock_flag(sk, SOCK_DBG);
1080 break;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001081
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001082 case SO_DONTROUTE:
1083 v.val = sock_flag(sk, SOCK_LOCALROUTE);
1084 break;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001085
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001086 case SO_BROADCAST:
Eric Dumazet1b23a5d2012-05-16 05:57:07 +00001087 v.val = sock_flag(sk, SOCK_BROADCAST);
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001088 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001089
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001090 case SO_SNDBUF:
1091 v.val = sk->sk_sndbuf;
1092 break;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001093
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001094 case SO_RCVBUF:
1095 v.val = sk->sk_rcvbuf;
1096 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001097
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001098 case SO_REUSEADDR:
1099 v.val = sk->sk_reuse;
1100 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001101
Tom Herbert055dc212013-01-22 09:49:50 +00001102 case SO_REUSEPORT:
1103 v.val = sk->sk_reuseport;
1104 break;
1105
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001106 case SO_KEEPALIVE:
Eric Dumazet1b23a5d2012-05-16 05:57:07 +00001107 v.val = sock_flag(sk, SOCK_KEEPOPEN);
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001108 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001109
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001110 case SO_TYPE:
1111 v.val = sk->sk_type;
1112 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001113
Jan Engelhardt49c794e2009-08-04 07:28:28 +00001114 case SO_PROTOCOL:
1115 v.val = sk->sk_protocol;
1116 break;
1117
Jan Engelhardt0d6038e2009-08-04 07:28:29 +00001118 case SO_DOMAIN:
1119 v.val = sk->sk_family;
1120 break;
1121
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001122 case SO_ERROR:
1123 v.val = -sock_error(sk);
Eric Dumazet2a915252009-05-27 11:30:05 +00001124 if (v.val == 0)
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001125 v.val = xchg(&sk->sk_err_soft, 0);
1126 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001127
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001128 case SO_OOBINLINE:
Eric Dumazet1b23a5d2012-05-16 05:57:07 +00001129 v.val = sock_flag(sk, SOCK_URGINLINE);
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001130 break;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001131
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001132 case SO_NO_CHECK:
Tom Herbert28448b82014-05-23 08:47:19 -07001133 v.val = sk->sk_no_check_tx;
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001134 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001135
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001136 case SO_PRIORITY:
1137 v.val = sk->sk_priority;
1138 break;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001139
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001140 case SO_LINGER:
1141 lv = sizeof(v.ling);
Eric Dumazet1b23a5d2012-05-16 05:57:07 +00001142 v.ling.l_onoff = sock_flag(sk, SOCK_LINGER);
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001143 v.ling.l_linger = sk->sk_lingertime / HZ;
1144 break;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001145
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001146 case SO_BSDCOMPAT:
1147 sock_warn_obsolete_bsdism("getsockopt");
1148 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001149
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001150 case SO_TIMESTAMP:
Eric Dumazet92f37fd2007-03-25 22:14:49 -07001151 v.val = sock_flag(sk, SOCK_RCVTSTAMP) &&
1152 !sock_flag(sk, SOCK_RCVTSTAMPNS);
1153 break;
1154
1155 case SO_TIMESTAMPNS:
1156 v.val = sock_flag(sk, SOCK_RCVTSTAMPNS);
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001157 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001158
Patrick Ohly20d49472009-02-12 05:03:38 +00001159 case SO_TIMESTAMPING:
Willem de Bruijnb9f40e22014-08-04 22:11:46 -04001160 v.val = sk->sk_tsflags;
Patrick Ohly20d49472009-02-12 05:03:38 +00001161 break;
1162
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001163 case SO_RCVTIMEO:
Eric Dumazet2a915252009-05-27 11:30:05 +00001164 lv = sizeof(struct timeval);
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001165 if (sk->sk_rcvtimeo == MAX_SCHEDULE_TIMEOUT) {
1166 v.tm.tv_sec = 0;
1167 v.tm.tv_usec = 0;
1168 } else {
1169 v.tm.tv_sec = sk->sk_rcvtimeo / HZ;
1170 v.tm.tv_usec = ((sk->sk_rcvtimeo % HZ) * 1000000) / HZ;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001171 }
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001172 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001173
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001174 case SO_SNDTIMEO:
Eric Dumazet2a915252009-05-27 11:30:05 +00001175 lv = sizeof(struct timeval);
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001176 if (sk->sk_sndtimeo == MAX_SCHEDULE_TIMEOUT) {
1177 v.tm.tv_sec = 0;
1178 v.tm.tv_usec = 0;
1179 } else {
1180 v.tm.tv_sec = sk->sk_sndtimeo / HZ;
1181 v.tm.tv_usec = ((sk->sk_sndtimeo % HZ) * 1000000) / HZ;
1182 }
1183 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001184
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001185 case SO_RCVLOWAT:
1186 v.val = sk->sk_rcvlowat;
1187 break;
Catherine Zhang877ce7c2006-06-29 12:27:47 -07001188
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001189 case SO_SNDLOWAT:
Eric Dumazet2a915252009-05-27 11:30:05 +00001190 v.val = 1;
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001191 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001192
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001193 case SO_PASSCRED:
Eric Dumazet82981932012-04-26 20:07:59 +00001194 v.val = !!test_bit(SOCK_PASSCRED, &sock->flags);
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001195 break;
1196
1197 case SO_PEERCRED:
Eric W. Biederman109f6e32010-06-13 03:30:14 +00001198 {
1199 struct ucred peercred;
1200 if (len > sizeof(peercred))
1201 len = sizeof(peercred);
1202 cred_to_ucred(sk->sk_peer_pid, sk->sk_peer_cred, &peercred);
1203 if (copy_to_user(optval, &peercred, len))
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001204 return -EFAULT;
1205 goto lenout;
Eric W. Biederman109f6e32010-06-13 03:30:14 +00001206 }
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001207
1208 case SO_PEERNAME:
1209 {
1210 char address[128];
1211
1212 if (sock->ops->getname(sock, (struct sockaddr *)address, &lv, 2))
1213 return -ENOTCONN;
1214 if (lv < len)
1215 return -EINVAL;
1216 if (copy_to_user(optval, address, len))
1217 return -EFAULT;
1218 goto lenout;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001219 }
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001220
1221 /* Dubious BSD thing... Probably nobody even uses it, but
1222 * the UNIX standard wants it for whatever reason... -DaveM
1223 */
1224 case SO_ACCEPTCONN:
1225 v.val = sk->sk_state == TCP_LISTEN;
1226 break;
1227
1228 case SO_PASSSEC:
Eric Dumazet82981932012-04-26 20:07:59 +00001229 v.val = !!test_bit(SOCK_PASSSEC, &sock->flags);
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001230 break;
1231
1232 case SO_PEERSEC:
1233 return security_socket_getpeersec_stream(sock, optval, optlen, len);
1234
Laszlo Attila Toth4a19ec52008-01-30 19:08:16 -08001235 case SO_MARK:
1236 v.val = sk->sk_mark;
1237 break;
1238
Neil Horman3b885782009-10-12 13:26:31 -07001239 case SO_RXQ_OVFL:
Eric Dumazet1b23a5d2012-05-16 05:57:07 +00001240 v.val = sock_flag(sk, SOCK_RXQ_OVFL);
Neil Horman3b885782009-10-12 13:26:31 -07001241 break;
1242
Johannes Berg6e3e9392011-11-09 10:15:42 +01001243 case SO_WIFI_STATUS:
Eric Dumazet1b23a5d2012-05-16 05:57:07 +00001244 v.val = sock_flag(sk, SOCK_WIFI_STATUS);
Johannes Berg6e3e9392011-11-09 10:15:42 +01001245 break;
1246
Pavel Emelyanovef64a542012-02-21 07:31:34 +00001247 case SO_PEEK_OFF:
1248 if (!sock->ops->set_peek_off)
1249 return -EOPNOTSUPP;
1250
1251 v.val = sk->sk_peek_off;
1252 break;
David S. Millerbc2f7992012-02-24 14:48:34 -05001253 case SO_NOFCS:
Eric Dumazet1b23a5d2012-05-16 05:57:07 +00001254 v.val = sock_flag(sk, SOCK_NOFCS);
David S. Millerbc2f7992012-02-24 14:48:34 -05001255 break;
Brian Haleyc91f6df2012-11-26 05:21:08 +00001256
Pavel Emelyanovf7b86bf2012-10-18 23:55:56 +00001257 case SO_BINDTODEVICE:
Brian Haleyc91f6df2012-11-26 05:21:08 +00001258 return sock_getbindtodevice(sk, optval, optlen, len);
1259
Pavel Emelyanova8fc9272012-11-01 02:01:48 +00001260 case SO_GET_FILTER:
1261 len = sk_get_filter(sk, (struct sock_filter __user *)optval, len);
1262 if (len < 0)
1263 return len;
1264
1265 goto lenout;
Brian Haleyc91f6df2012-11-26 05:21:08 +00001266
Vincent Bernatd59577b2013-01-16 22:55:49 +01001267 case SO_LOCK_FILTER:
1268 v.val = sock_flag(sk, SOCK_FILTER_LOCKED);
1269 break;
1270
Michal Sekletarea02f942014-01-17 17:09:45 +01001271 case SO_BPF_EXTENSIONS:
1272 v.val = bpf_tell_extensions();
1273 break;
1274
Keller, Jacob E7d4c04f2013-03-28 11:19:25 +00001275 case SO_SELECT_ERR_QUEUE:
1276 v.val = sock_flag(sk, SOCK_SELECT_ERR_QUEUE);
1277 break;
1278
Cong Wange0d10952013-08-01 11:10:25 +08001279#ifdef CONFIG_NET_RX_BUSY_POLL
Eliezer Tamir64b0dc52013-07-10 17:13:36 +03001280 case SO_BUSY_POLL:
Eliezer Tamirdafcc432013-06-14 16:33:57 +03001281 v.val = sk->sk_ll_usec;
1282 break;
1283#endif
1284
Eric Dumazet62748f32013-09-24 08:20:52 -07001285 case SO_MAX_PACING_RATE:
1286 v.val = sk->sk_max_pacing_rate;
1287 break;
1288
Eric Dumazet2c8c56e2014-11-11 05:54:28 -08001289 case SO_INCOMING_CPU:
1290 v.val = sk->sk_incoming_cpu;
1291 break;
1292
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001293 default:
YOSHIFUJI Hideaki/吉藤英明443b5992015-03-23 18:04:13 +09001294 /* We implement the SO_SNDLOWAT etc to not be settable
1295 * (1003.1g 7).
1296 */
Stephen Hemmingere71a4782007-04-10 20:10:33 -07001297 return -ENOPROTOOPT;
1298 }
1299
Linus Torvalds1da177e2005-04-16 15:20:36 -07001300 if (len > lv)
1301 len = lv;
1302 if (copy_to_user(optval, &v, len))
1303 return -EFAULT;
1304lenout:
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001305 if (put_user(len, optlen))
1306 return -EFAULT;
1307 return 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001308}
1309
Ingo Molnara5b5bb92006-07-03 00:25:35 -07001310/*
1311 * Initialize an sk_lock.
1312 *
1313 * (We also register the sk_lock with the lock validator.)
1314 */
Dave Jonesb6f99a22007-03-22 12:27:49 -07001315static inline void sock_lock_init(struct sock *sk)
Ingo Molnara5b5bb92006-07-03 00:25:35 -07001316{
Peter Zijlstraed075362006-12-06 20:35:24 -08001317 sock_lock_init_class_and_name(sk,
1318 af_family_slock_key_strings[sk->sk_family],
1319 af_family_slock_keys + sk->sk_family,
1320 af_family_key_strings[sk->sk_family],
1321 af_family_keys + sk->sk_family);
Ingo Molnara5b5bb92006-07-03 00:25:35 -07001322}
1323
Eric Dumazet4dc6dc72009-07-15 23:13:10 +00001324/*
1325 * Copy all fields from osk to nsk but nsk->sk_refcnt must not change yet,
1326 * even temporarly, because of RCU lookups. sk_node should also be left as is.
Eric Dumazet68835ab2010-11-30 19:04:07 +00001327 * We must not copy fields between sk_dontcopy_begin and sk_dontcopy_end
Eric Dumazet4dc6dc72009-07-15 23:13:10 +00001328 */
Pavel Emelyanovf1a6c4d2007-11-01 00:29:45 -07001329static void sock_copy(struct sock *nsk, const struct sock *osk)
1330{
1331#ifdef CONFIG_SECURITY_NETWORK
1332 void *sptr = nsk->sk_security;
1333#endif
Eric Dumazet68835ab2010-11-30 19:04:07 +00001334 memcpy(nsk, osk, offsetof(struct sock, sk_dontcopy_begin));
1335
1336 memcpy(&nsk->sk_dontcopy_end, &osk->sk_dontcopy_end,
1337 osk->sk_prot->obj_size - offsetof(struct sock, sk_dontcopy_end));
1338
Pavel Emelyanovf1a6c4d2007-11-01 00:29:45 -07001339#ifdef CONFIG_SECURITY_NETWORK
1340 nsk->sk_security = sptr;
1341 security_sk_clone(osk, nsk);
1342#endif
1343}
1344
Octavian Purdilafcbdf092010-12-16 14:26:56 -08001345void sk_prot_clear_portaddr_nulls(struct sock *sk, int size)
1346{
1347 unsigned long nulls1, nulls2;
1348
1349 nulls1 = offsetof(struct sock, __sk_common.skc_node.next);
1350 nulls2 = offsetof(struct sock, __sk_common.skc_portaddr_node.next);
1351 if (nulls1 > nulls2)
1352 swap(nulls1, nulls2);
1353
1354 if (nulls1 != 0)
1355 memset((char *)sk, 0, nulls1);
1356 memset((char *)sk + nulls1 + sizeof(void *), 0,
1357 nulls2 - nulls1 - sizeof(void *));
1358 memset((char *)sk + nulls2 + sizeof(void *), 0,
1359 size - nulls2 - sizeof(void *));
1360}
1361EXPORT_SYMBOL(sk_prot_clear_portaddr_nulls);
1362
Pavel Emelyanov2e4afe72007-11-01 00:36:26 -07001363static struct sock *sk_prot_alloc(struct proto *prot, gfp_t priority,
1364 int family)
Pavel Emelyanovc308c1b22007-11-01 00:33:50 -07001365{
1366 struct sock *sk;
1367 struct kmem_cache *slab;
1368
1369 slab = prot->slab;
Eric Dumazete912b112009-07-08 19:36:05 +00001370 if (slab != NULL) {
1371 sk = kmem_cache_alloc(slab, priority & ~__GFP_ZERO);
1372 if (!sk)
1373 return sk;
1374 if (priority & __GFP_ZERO) {
Octavian Purdilafcbdf092010-12-16 14:26:56 -08001375 if (prot->clear_sk)
1376 prot->clear_sk(sk, prot->obj_size);
1377 else
1378 sk_prot_clear_nulls(sk, prot->obj_size);
Eric Dumazete912b112009-07-08 19:36:05 +00001379 }
Octavian Purdilafcbdf092010-12-16 14:26:56 -08001380 } else
Pavel Emelyanovc308c1b22007-11-01 00:33:50 -07001381 sk = kmalloc(prot->obj_size, priority);
1382
Pavel Emelyanov2e4afe72007-11-01 00:36:26 -07001383 if (sk != NULL) {
Vegard Nossuma98b65a2009-02-26 14:46:57 +01001384 kmemcheck_annotate_bitfield(sk, flags);
1385
Pavel Emelyanov2e4afe72007-11-01 00:36:26 -07001386 if (security_sk_alloc(sk, family, priority))
1387 goto out_free;
1388
1389 if (!try_module_get(prot->owner))
1390 goto out_free_sec;
Krishna Kumare022f0b2009-10-19 23:46:20 +00001391 sk_tx_queue_clear(sk);
Tejun Heobd1060a2015-12-07 17:38:53 -05001392 cgroup_sk_alloc(&sk->sk_cgrp_data);
Pavel Emelyanov2e4afe72007-11-01 00:36:26 -07001393 }
1394
Pavel Emelyanovc308c1b22007-11-01 00:33:50 -07001395 return sk;
Pavel Emelyanov2e4afe72007-11-01 00:36:26 -07001396
1397out_free_sec:
1398 security_sk_free(sk);
1399out_free:
1400 if (slab != NULL)
1401 kmem_cache_free(slab, sk);
1402 else
1403 kfree(sk);
1404 return NULL;
Pavel Emelyanovc308c1b22007-11-01 00:33:50 -07001405}
1406
1407static void sk_prot_free(struct proto *prot, struct sock *sk)
1408{
1409 struct kmem_cache *slab;
Pavel Emelyanov2e4afe72007-11-01 00:36:26 -07001410 struct module *owner;
Pavel Emelyanovc308c1b22007-11-01 00:33:50 -07001411
Pavel Emelyanov2e4afe72007-11-01 00:36:26 -07001412 owner = prot->owner;
Pavel Emelyanovc308c1b22007-11-01 00:33:50 -07001413 slab = prot->slab;
Pavel Emelyanov2e4afe72007-11-01 00:36:26 -07001414
Tejun Heobd1060a2015-12-07 17:38:53 -05001415 cgroup_sk_free(&sk->sk_cgrp_data);
Pavel Emelyanov2e4afe72007-11-01 00:36:26 -07001416 security_sk_free(sk);
Pavel Emelyanovc308c1b22007-11-01 00:33:50 -07001417 if (slab != NULL)
1418 kmem_cache_free(slab, sk);
1419 else
1420 kfree(sk);
Pavel Emelyanov2e4afe72007-11-01 00:36:26 -07001421 module_put(owner);
Pavel Emelyanovc308c1b22007-11-01 00:33:50 -07001422}
1423
Linus Torvalds1da177e2005-04-16 15:20:36 -07001424/**
1425 * sk_alloc - All socket objects are allocated here
Randy Dunlapc4ea43c2007-10-12 21:17:49 -07001426 * @net: the applicable net namespace
Pavel Pisa4dc3b162005-05-01 08:59:25 -07001427 * @family: protocol family
1428 * @priority: for allocation (%GFP_KERNEL, %GFP_ATOMIC, etc)
1429 * @prot: struct proto associated with this new sock instance
Eric W. Biederman11aa9c22015-05-08 21:09:13 -05001430 * @kern: is this to be a kernel socket?
Linus Torvalds1da177e2005-04-16 15:20:36 -07001431 */
Eric W. Biederman1b8d7ae2007-10-08 23:24:22 -07001432struct sock *sk_alloc(struct net *net, int family, gfp_t priority,
Eric W. Biederman11aa9c22015-05-08 21:09:13 -05001433 struct proto *prot, int kern)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001434{
Pavel Emelyanovc308c1b22007-11-01 00:33:50 -07001435 struct sock *sk;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001436
Pavel Emelyanov154adbc2007-11-01 00:38:43 -07001437 sk = sk_prot_alloc(prot, priority | __GFP_ZERO, family);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001438 if (sk) {
Pavel Emelyanov154adbc2007-11-01 00:38:43 -07001439 sk->sk_family = family;
1440 /*
1441 * See comment in struct sock definition to understand
1442 * why we need sk_prot_creator -acme
1443 */
1444 sk->sk_prot = sk->sk_prot_creator = prot;
1445 sock_lock_init(sk);
Eric W. Biederman26abe142015-05-08 21:10:31 -05001446 sk->sk_net_refcnt = kern ? 0 : 1;
1447 if (likely(sk->sk_net_refcnt))
1448 get_net(net);
1449 sock_net_set(sk, net);
Jarek Poplawskid66ee052009-08-30 23:15:36 +00001450 atomic_set(&sk->sk_wmem_alloc, 1);
Herbert Xuf8451722010-05-24 00:12:34 -07001451
Tejun Heo2a56a1f2015-12-07 17:38:52 -05001452 sock_update_classid(&sk->sk_cgrp_data);
1453 sock_update_netprioidx(&sk->sk_cgrp_data);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001454 }
Frank Filza79af592005-09-27 15:23:38 -07001455
Pavel Emelyanov2e4afe72007-11-01 00:36:26 -07001456 return sk;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001457}
Eric Dumazet2a915252009-05-27 11:30:05 +00001458EXPORT_SYMBOL(sk_alloc);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001459
Craig Gallekeb4cb002015-06-15 11:26:18 -04001460void sk_destruct(struct sock *sk)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001461{
1462 struct sk_filter *filter;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001463
1464 if (sk->sk_destruct)
1465 sk->sk_destruct(sk);
1466
Paul E. McKenneya898def2010-02-22 17:04:49 -08001467 filter = rcu_dereference_check(sk->sk_filter,
1468 atomic_read(&sk->sk_wmem_alloc) == 0);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001469 if (filter) {
Pavel Emelyanov309dd5f2007-10-17 21:21:51 -07001470 sk_filter_uncharge(sk, filter);
Stephen Hemmingera9b3cd72011-08-01 16:19:00 +00001471 RCU_INIT_POINTER(sk->sk_filter, NULL);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001472 }
Craig Gallek538950a2016-01-04 17:41:47 -05001473 if (rcu_access_pointer(sk->sk_reuseport_cb))
1474 reuseport_detach_sock(sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001475
Eric Dumazet08e29af2011-11-28 12:04:18 +00001476 sock_disable_timestamp(sk, SK_FLAGS_TIMESTAMP);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001477
1478 if (atomic_read(&sk->sk_omem_alloc))
Joe Perchese005d192012-05-16 19:58:40 +00001479 pr_debug("%s: optmem leakage (%d bytes) detected\n",
1480 __func__, atomic_read(&sk->sk_omem_alloc));
Linus Torvalds1da177e2005-04-16 15:20:36 -07001481
Eric W. Biederman109f6e32010-06-13 03:30:14 +00001482 if (sk->sk_peer_cred)
1483 put_cred(sk->sk_peer_cred);
1484 put_pid(sk->sk_peer_pid);
Eric W. Biederman26abe142015-05-08 21:10:31 -05001485 if (likely(sk->sk_net_refcnt))
1486 put_net(sock_net(sk));
Pavel Emelyanovc308c1b22007-11-01 00:33:50 -07001487 sk_prot_free(sk->sk_prot_creator, sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001488}
Eric Dumazet2b85a342009-06-11 02:55:43 -07001489
Craig Gallekeb4cb002015-06-15 11:26:18 -04001490static void __sk_free(struct sock *sk)
1491{
Craig Gallekb9226222015-06-30 12:49:32 -04001492 if (unlikely(sock_diag_has_destroy_listeners(sk) && sk->sk_net_refcnt))
Craig Gallekeb4cb002015-06-15 11:26:18 -04001493 sock_diag_broadcast_destroy(sk);
1494 else
1495 sk_destruct(sk);
1496}
1497
Eric Dumazet2b85a342009-06-11 02:55:43 -07001498void sk_free(struct sock *sk)
1499{
1500 /*
Lucas De Marchi25985ed2011-03-30 22:57:33 -03001501 * We subtract one from sk_wmem_alloc and can know if
Eric Dumazet2b85a342009-06-11 02:55:43 -07001502 * some packets are still in some tx queue.
1503 * If not null, sock_wfree() will call __sk_free(sk) later
1504 */
1505 if (atomic_dec_and_test(&sk->sk_wmem_alloc))
1506 __sk_free(sk);
1507}
Eric Dumazet2a915252009-05-27 11:30:05 +00001508EXPORT_SYMBOL(sk_free);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001509
Stephen Rothwell475f1b52012-01-09 16:33:16 +11001510static void sk_update_clone(const struct sock *sk, struct sock *newsk)
1511{
1512 if (mem_cgroup_sockets_enabled && sk->sk_cgrp)
1513 sock_update_memcg(newsk);
1514}
1515
Eric Dumazete56c57d2011-11-08 17:07:07 -05001516/**
1517 * sk_clone_lock - clone a socket, and lock its clone
1518 * @sk: the socket to clone
1519 * @priority: for allocation (%GFP_KERNEL, %GFP_ATOMIC, etc)
1520 *
1521 * Caller must unlock socket even in error path (bh_unlock_sock(newsk))
1522 */
1523struct sock *sk_clone_lock(const struct sock *sk, const gfp_t priority)
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001524{
Pavel Emelyanov8fd1d172007-11-01 00:37:32 -07001525 struct sock *newsk;
Alexei Starovoitov278571b2014-07-30 20:34:12 -07001526 bool is_charged = true;
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001527
Pavel Emelyanov8fd1d172007-11-01 00:37:32 -07001528 newsk = sk_prot_alloc(sk->sk_prot, priority, sk->sk_family);
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001529 if (newsk != NULL) {
1530 struct sk_filter *filter;
1531
Venkat Yekkirala892c1412006-08-04 23:08:56 -07001532 sock_copy(newsk, sk);
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001533
1534 /* SANITY */
Sowmini Varadhan8a681732015-07-30 15:50:36 +02001535 if (likely(newsk->sk_net_refcnt))
1536 get_net(sock_net(newsk));
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001537 sk_node_init(&newsk->sk_node);
1538 sock_lock_init(newsk);
1539 bh_lock_sock(newsk);
Eric Dumazetfa438cc2007-03-04 16:05:44 -08001540 newsk->sk_backlog.head = newsk->sk_backlog.tail = NULL;
Zhu Yi8eae9392010-03-04 18:01:40 +00001541 newsk->sk_backlog.len = 0;
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001542
1543 atomic_set(&newsk->sk_rmem_alloc, 0);
Eric Dumazet2b85a342009-06-11 02:55:43 -07001544 /*
1545 * sk_wmem_alloc set to one (see sk_free() and sock_wfree())
1546 */
1547 atomic_set(&newsk->sk_wmem_alloc, 1);
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001548 atomic_set(&newsk->sk_omem_alloc, 0);
1549 skb_queue_head_init(&newsk->sk_receive_queue);
1550 skb_queue_head_init(&newsk->sk_write_queue);
1551
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001552 rwlock_init(&newsk->sk_callback_lock);
Peter Zijlstra443aef02007-07-19 01:49:00 -07001553 lockdep_set_class_and_name(&newsk->sk_callback_lock,
1554 af_callback_keys + newsk->sk_family,
1555 af_family_clock_key_strings[newsk->sk_family]);
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001556
1557 newsk->sk_dst_cache = NULL;
1558 newsk->sk_wmem_queued = 0;
1559 newsk->sk_forward_alloc = 0;
1560 newsk->sk_send_head = NULL;
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001561 newsk->sk_userlocks = sk->sk_userlocks & ~SOCK_BINDPORT_LOCK;
1562
1563 sock_reset_flag(newsk, SOCK_DONE);
1564 skb_queue_head_init(&newsk->sk_error_queue);
1565
Eric Dumazet0d7da9d2010-10-25 03:47:05 +00001566 filter = rcu_dereference_protected(newsk->sk_filter, 1);
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001567 if (filter != NULL)
Alexei Starovoitov278571b2014-07-30 20:34:12 -07001568 /* though it's an empty new sock, the charging may fail
1569 * if sysctl_optmem_max was changed between creation of
1570 * original socket and cloning
1571 */
1572 is_charged = sk_filter_charge(newsk, filter);
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001573
Eric Dumazetd188ba82015-12-08 07:22:02 -08001574 if (unlikely(!is_charged || xfrm_sk_clone_policy(newsk, sk))) {
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001575 /* It is still raw copy of parent, so invalidate
1576 * destructor and make plain sk_free() */
1577 newsk->sk_destruct = NULL;
Thomas Gleixnerb0691c82011-10-25 02:30:50 +00001578 bh_unlock_sock(newsk);
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001579 sk_free(newsk);
1580 newsk = NULL;
1581 goto out;
1582 }
1583
1584 newsk->sk_err = 0;
1585 newsk->sk_priority = 0;
Eric Dumazet2c8c56e2014-11-11 05:54:28 -08001586 newsk->sk_incoming_cpu = raw_smp_processor_id();
Eric Dumazet33cf7c92015-03-11 18:53:14 -07001587 atomic64_set(&newsk->sk_cookie, 0);
Eric Dumazet4dc6dc72009-07-15 23:13:10 +00001588 /*
1589 * Before updating sk_refcnt, we must commit prior changes to memory
1590 * (Documentation/RCU/rculist_nulls.txt for details)
1591 */
1592 smp_wmb();
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001593 atomic_set(&newsk->sk_refcnt, 2);
1594
1595 /*
1596 * Increment the counter in the same struct proto as the master
1597 * sock (sk_refcnt_debug_inc uses newsk->sk_prot->socks, that
1598 * is the same as sk->sk_prot->socks, as this field was copied
1599 * with memcpy).
1600 *
1601 * This _changes_ the previous behaviour, where
1602 * tcp_create_openreq_child always was incrementing the
1603 * equivalent to tcp_prot->socks (inet_sock_nr), so this have
1604 * to be taken into account in all callers. -acme
1605 */
1606 sk_refcnt_debug_inc(newsk);
David S. Miller972692e2008-06-17 22:41:38 -07001607 sk_set_socket(newsk, NULL);
Eric Dumazet43815482010-04-29 11:01:49 +00001608 newsk->sk_wq = NULL;
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001609
Glauber Costaf3f511e2012-01-05 20:16:39 +00001610 sk_update_clone(sk, newsk);
1611
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001612 if (newsk->sk_prot->sockets_allocated)
Glauber Costa180d8cd2011-12-11 21:47:02 +00001613 sk_sockets_allocated_inc(newsk);
Octavian Purdila704da5602010-01-08 00:00:09 -08001614
Hannes Frederic Sowa080a2702015-10-26 13:51:37 +01001615 if (sock_needs_netstamp(sk) &&
1616 newsk->sk_flags & SK_FLAGS_TIMESTAMP)
Octavian Purdila704da5602010-01-08 00:00:09 -08001617 net_enable_timestamp();
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001618 }
1619out:
1620 return newsk;
1621}
Eric Dumazete56c57d2011-11-08 17:07:07 -05001622EXPORT_SYMBOL_GPL(sk_clone_lock);
Arnaldo Carvalho de Melo87d11ce2005-08-09 20:10:12 -07001623
Andi Kleen99580892007-04-20 17:12:43 -07001624void sk_setup_caps(struct sock *sk, struct dst_entry *dst)
1625{
Eric Dumazetd6a4e262015-05-26 08:55:28 -07001626 u32 max_segs = 1;
1627
Eric Dumazet6bd4f352015-12-02 21:53:57 -08001628 sk_dst_set(sk, dst);
Andi Kleen99580892007-04-20 17:12:43 -07001629 sk->sk_route_caps = dst->dev->features;
1630 if (sk->sk_route_caps & NETIF_F_GSO)
Herbert Xu4fcd6b92007-05-31 22:15:50 -07001631 sk->sk_route_caps |= NETIF_F_GSO_SOFTWARE;
Eric Dumazeta4654192010-05-16 00:36:33 -07001632 sk->sk_route_caps &= ~sk->sk_route_nocaps;
Andi Kleen99580892007-04-20 17:12:43 -07001633 if (sk_can_gso(sk)) {
Peter P Waskiewicz Jr82cc1a72008-03-21 03:43:19 -07001634 if (dst->header_len) {
Andi Kleen99580892007-04-20 17:12:43 -07001635 sk->sk_route_caps &= ~NETIF_F_GSO_MASK;
Peter P Waskiewicz Jr82cc1a72008-03-21 03:43:19 -07001636 } else {
Andi Kleen99580892007-04-20 17:12:43 -07001637 sk->sk_route_caps |= NETIF_F_SG | NETIF_F_HW_CSUM;
Peter P Waskiewicz Jr82cc1a72008-03-21 03:43:19 -07001638 sk->sk_gso_max_size = dst->dev->gso_max_size;
Eric Dumazetd6a4e262015-05-26 08:55:28 -07001639 max_segs = max_t(u32, dst->dev->gso_max_segs, 1);
Peter P Waskiewicz Jr82cc1a72008-03-21 03:43:19 -07001640 }
Andi Kleen99580892007-04-20 17:12:43 -07001641 }
Eric Dumazetd6a4e262015-05-26 08:55:28 -07001642 sk->sk_gso_max_segs = max_segs;
Andi Kleen99580892007-04-20 17:12:43 -07001643}
1644EXPORT_SYMBOL_GPL(sk_setup_caps);
1645
Linus Torvalds1da177e2005-04-16 15:20:36 -07001646/*
1647 * Simple resource managers for sockets.
1648 */
1649
1650
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001651/*
1652 * Write buffer destructor automatically called from kfree_skb.
Linus Torvalds1da177e2005-04-16 15:20:36 -07001653 */
1654void sock_wfree(struct sk_buff *skb)
1655{
1656 struct sock *sk = skb->sk;
Eric Dumazetd99927f2009-09-24 10:49:24 +00001657 unsigned int len = skb->truesize;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001658
Eric Dumazetd99927f2009-09-24 10:49:24 +00001659 if (!sock_flag(sk, SOCK_USE_WRITE_QUEUE)) {
1660 /*
1661 * Keep a reference on sk_wmem_alloc, this will be released
1662 * after sk_write_space() call
1663 */
1664 atomic_sub(len - 1, &sk->sk_wmem_alloc);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001665 sk->sk_write_space(sk);
Eric Dumazetd99927f2009-09-24 10:49:24 +00001666 len = 1;
1667 }
Eric Dumazet2b85a342009-06-11 02:55:43 -07001668 /*
Eric Dumazetd99927f2009-09-24 10:49:24 +00001669 * if sk_wmem_alloc reaches 0, we must finish what sk_free()
1670 * could not do because of in-flight packets
Eric Dumazet2b85a342009-06-11 02:55:43 -07001671 */
Eric Dumazetd99927f2009-09-24 10:49:24 +00001672 if (atomic_sub_and_test(len, &sk->sk_wmem_alloc))
Eric Dumazet2b85a342009-06-11 02:55:43 -07001673 __sk_free(sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001674}
Eric Dumazet2a915252009-05-27 11:30:05 +00001675EXPORT_SYMBOL(sock_wfree);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001676
Eric Dumazet9e17f8a2015-11-01 15:36:55 -08001677void skb_set_owner_w(struct sk_buff *skb, struct sock *sk)
1678{
1679 skb_orphan(skb);
1680 skb->sk = sk;
1681#ifdef CONFIG_INET
1682 if (unlikely(!sk_fullsock(sk))) {
1683 skb->destructor = sock_edemux;
1684 sock_hold(sk);
1685 return;
1686 }
1687#endif
1688 skb->destructor = sock_wfree;
1689 skb_set_hash_from_sk(skb, sk);
1690 /*
1691 * We used to take a refcount on sk, but following operation
1692 * is enough to guarantee sk_free() wont free this sock until
1693 * all in-flight packets are completed
1694 */
1695 atomic_add(skb->truesize, &sk->sk_wmem_alloc);
1696}
1697EXPORT_SYMBOL(skb_set_owner_w);
1698
Eric Dumazetf2f872f2013-07-30 17:55:08 -07001699void skb_orphan_partial(struct sk_buff *skb)
1700{
1701 /* TCP stack sets skb->ooo_okay based on sk_wmem_alloc,
1702 * so we do not completely orphan skb, but transfert all
1703 * accounted bytes but one, to avoid unexpected reorders.
1704 */
1705 if (skb->destructor == sock_wfree
1706#ifdef CONFIG_INET
1707 || skb->destructor == tcp_wfree
1708#endif
1709 ) {
1710 atomic_sub(skb->truesize - 1, &skb->sk->sk_wmem_alloc);
1711 skb->truesize = 1;
1712 } else {
1713 skb_orphan(skb);
1714 }
1715}
1716EXPORT_SYMBOL(skb_orphan_partial);
1717
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001718/*
1719 * Read buffer destructor automatically called from kfree_skb.
Linus Torvalds1da177e2005-04-16 15:20:36 -07001720 */
1721void sock_rfree(struct sk_buff *skb)
1722{
1723 struct sock *sk = skb->sk;
Eric Dumazetd361fd52010-07-10 22:45:17 +00001724 unsigned int len = skb->truesize;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001725
Eric Dumazetd361fd52010-07-10 22:45:17 +00001726 atomic_sub(len, &sk->sk_rmem_alloc);
1727 sk_mem_uncharge(sk, len);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001728}
Eric Dumazet2a915252009-05-27 11:30:05 +00001729EXPORT_SYMBOL(sock_rfree);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001730
Oliver Hartkopp7768eed2015-03-10 19:03:46 +01001731/*
1732 * Buffer destructor for skbs that are not used directly in read or write
1733 * path, e.g. for error handler skbs. Automatically called from kfree_skb.
1734 */
Alexander Duyck62bccb82014-09-04 13:31:35 -04001735void sock_efree(struct sk_buff *skb)
1736{
1737 sock_put(skb->sk);
1738}
1739EXPORT_SYMBOL(sock_efree);
1740
Eric W. Biederman976d02012012-05-23 17:16:53 -06001741kuid_t sock_i_uid(struct sock *sk)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001742{
Eric W. Biederman976d02012012-05-23 17:16:53 -06001743 kuid_t uid;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001744
Eric Dumazetf064af12010-09-22 12:43:39 +00001745 read_lock_bh(&sk->sk_callback_lock);
Eric W. Biederman976d02012012-05-23 17:16:53 -06001746 uid = sk->sk_socket ? SOCK_INODE(sk->sk_socket)->i_uid : GLOBAL_ROOT_UID;
Eric Dumazetf064af12010-09-22 12:43:39 +00001747 read_unlock_bh(&sk->sk_callback_lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001748 return uid;
1749}
Eric Dumazet2a915252009-05-27 11:30:05 +00001750EXPORT_SYMBOL(sock_i_uid);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001751
1752unsigned long sock_i_ino(struct sock *sk)
1753{
1754 unsigned long ino;
1755
Eric Dumazetf064af12010-09-22 12:43:39 +00001756 read_lock_bh(&sk->sk_callback_lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001757 ino = sk->sk_socket ? SOCK_INODE(sk->sk_socket)->i_ino : 0;
Eric Dumazetf064af12010-09-22 12:43:39 +00001758 read_unlock_bh(&sk->sk_callback_lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001759 return ino;
1760}
Eric Dumazet2a915252009-05-27 11:30:05 +00001761EXPORT_SYMBOL(sock_i_ino);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001762
1763/*
1764 * Allocate a skb from the socket's send buffer.
1765 */
Victor Fusco86a76ca2005-07-08 14:57:47 -07001766struct sk_buff *sock_wmalloc(struct sock *sk, unsigned long size, int force,
Al Virodd0fc662005-10-07 07:46:04 +01001767 gfp_t priority)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001768{
1769 if (force || atomic_read(&sk->sk_wmem_alloc) < sk->sk_sndbuf) {
Eric Dumazet2a915252009-05-27 11:30:05 +00001770 struct sk_buff *skb = alloc_skb(size, priority);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001771 if (skb) {
1772 skb_set_owner_w(skb, sk);
1773 return skb;
1774 }
1775 }
1776 return NULL;
1777}
Eric Dumazet2a915252009-05-27 11:30:05 +00001778EXPORT_SYMBOL(sock_wmalloc);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001779
1780/*
Linus Torvalds1da177e2005-04-16 15:20:36 -07001781 * Allocate a memory block from the socket's option memory buffer.
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001782 */
Al Virodd0fc662005-10-07 07:46:04 +01001783void *sock_kmalloc(struct sock *sk, int size, gfp_t priority)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001784{
Eric Dumazet95c96172012-04-15 05:58:06 +00001785 if ((unsigned int)size <= sysctl_optmem_max &&
Linus Torvalds1da177e2005-04-16 15:20:36 -07001786 atomic_read(&sk->sk_omem_alloc) + size < sysctl_optmem_max) {
1787 void *mem;
1788 /* First do the add, to avoid the race if kmalloc
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001789 * might sleep.
Linus Torvalds1da177e2005-04-16 15:20:36 -07001790 */
1791 atomic_add(size, &sk->sk_omem_alloc);
1792 mem = kmalloc(size, priority);
1793 if (mem)
1794 return mem;
1795 atomic_sub(size, &sk->sk_omem_alloc);
1796 }
1797 return NULL;
1798}
Eric Dumazet2a915252009-05-27 11:30:05 +00001799EXPORT_SYMBOL(sock_kmalloc);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001800
Daniel Borkmann79e88652014-11-19 17:13:11 +01001801/* Free an option memory block. Note, we actually want the inline
1802 * here as this allows gcc to detect the nullify and fold away the
1803 * condition entirely.
Linus Torvalds1da177e2005-04-16 15:20:36 -07001804 */
Daniel Borkmann79e88652014-11-19 17:13:11 +01001805static inline void __sock_kfree_s(struct sock *sk, void *mem, int size,
1806 const bool nullify)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001807{
David S. Millere53da5f2014-10-14 17:02:37 -04001808 if (WARN_ON_ONCE(!mem))
1809 return;
Daniel Borkmann79e88652014-11-19 17:13:11 +01001810 if (nullify)
1811 kzfree(mem);
1812 else
1813 kfree(mem);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001814 atomic_sub(size, &sk->sk_omem_alloc);
1815}
Daniel Borkmann79e88652014-11-19 17:13:11 +01001816
1817void sock_kfree_s(struct sock *sk, void *mem, int size)
1818{
1819 __sock_kfree_s(sk, mem, size, false);
1820}
Eric Dumazet2a915252009-05-27 11:30:05 +00001821EXPORT_SYMBOL(sock_kfree_s);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001822
Daniel Borkmann79e88652014-11-19 17:13:11 +01001823void sock_kzfree_s(struct sock *sk, void *mem, int size)
1824{
1825 __sock_kfree_s(sk, mem, size, true);
1826}
1827EXPORT_SYMBOL(sock_kzfree_s);
1828
Linus Torvalds1da177e2005-04-16 15:20:36 -07001829/* It is almost wait_for_tcp_memory minus release_sock/lock_sock.
1830 I think, these locks should be removed for datagram sockets.
1831 */
Eric Dumazet2a915252009-05-27 11:30:05 +00001832static long sock_wait_for_wmem(struct sock *sk, long timeo)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001833{
1834 DEFINE_WAIT(wait);
1835
Eric Dumazet9cd3e072015-11-29 20:03:10 -08001836 sk_clear_bit(SOCKWQ_ASYNC_NOSPACE, sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001837 for (;;) {
1838 if (!timeo)
1839 break;
1840 if (signal_pending(current))
1841 break;
1842 set_bit(SOCK_NOSPACE, &sk->sk_socket->flags);
Eric Dumazetaa395142010-04-20 13:03:51 +00001843 prepare_to_wait(sk_sleep(sk), &wait, TASK_INTERRUPTIBLE);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001844 if (atomic_read(&sk->sk_wmem_alloc) < sk->sk_sndbuf)
1845 break;
1846 if (sk->sk_shutdown & SEND_SHUTDOWN)
1847 break;
1848 if (sk->sk_err)
1849 break;
1850 timeo = schedule_timeout(timeo);
1851 }
Eric Dumazetaa395142010-04-20 13:03:51 +00001852 finish_wait(sk_sleep(sk), &wait);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001853 return timeo;
1854}
1855
1856
1857/*
1858 * Generic send/receive buffer handlers
1859 */
1860
Herbert Xu4cc7f682009-02-04 16:55:54 -08001861struct sk_buff *sock_alloc_send_pskb(struct sock *sk, unsigned long header_len,
1862 unsigned long data_len, int noblock,
Eric Dumazet28d64272013-08-08 14:38:47 -07001863 int *errcode, int max_page_order)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001864{
Eric Dumazet2e4e4412014-09-17 04:49:49 -07001865 struct sk_buff *skb;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001866 long timeo;
1867 int err;
1868
Linus Torvalds1da177e2005-04-16 15:20:36 -07001869 timeo = sock_sndtimeo(sk, noblock);
Eric Dumazet2e4e4412014-09-17 04:49:49 -07001870 for (;;) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07001871 err = sock_error(sk);
1872 if (err != 0)
1873 goto failure;
1874
1875 err = -EPIPE;
1876 if (sk->sk_shutdown & SEND_SHUTDOWN)
1877 goto failure;
1878
Eric Dumazet2e4e4412014-09-17 04:49:49 -07001879 if (sk_wmem_alloc_get(sk) < sk->sk_sndbuf)
1880 break;
Eric Dumazet28d64272013-08-08 14:38:47 -07001881
Eric Dumazet9cd3e072015-11-29 20:03:10 -08001882 sk_set_bit(SOCKWQ_ASYNC_NOSPACE, sk);
Eric Dumazet2e4e4412014-09-17 04:49:49 -07001883 set_bit(SOCK_NOSPACE, &sk->sk_socket->flags);
1884 err = -EAGAIN;
1885 if (!timeo)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001886 goto failure;
Eric Dumazet2e4e4412014-09-17 04:49:49 -07001887 if (signal_pending(current))
1888 goto interrupted;
1889 timeo = sock_wait_for_wmem(sk, timeo);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001890 }
Eric Dumazet2e4e4412014-09-17 04:49:49 -07001891 skb = alloc_skb_with_frags(header_len, data_len, max_page_order,
1892 errcode, sk->sk_allocation);
1893 if (skb)
1894 skb_set_owner_w(skb, sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001895 return skb;
1896
1897interrupted:
1898 err = sock_intr_errno(timeo);
1899failure:
1900 *errcode = err;
1901 return NULL;
1902}
Herbert Xu4cc7f682009-02-04 16:55:54 -08001903EXPORT_SYMBOL(sock_alloc_send_pskb);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001904
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001905struct sk_buff *sock_alloc_send_skb(struct sock *sk, unsigned long size,
Linus Torvalds1da177e2005-04-16 15:20:36 -07001906 int noblock, int *errcode)
1907{
Eric Dumazet28d64272013-08-08 14:38:47 -07001908 return sock_alloc_send_pskb(sk, size, 0, noblock, errcode, 0);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001909}
Eric Dumazet2a915252009-05-27 11:30:05 +00001910EXPORT_SYMBOL(sock_alloc_send_skb);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001911
Edward Jeef28ea362015-10-08 14:56:48 -07001912int sock_cmsg_send(struct sock *sk, struct msghdr *msg,
1913 struct sockcm_cookie *sockc)
1914{
1915 struct cmsghdr *cmsg;
1916
1917 for_each_cmsghdr(cmsg, msg) {
1918 if (!CMSG_OK(msg, cmsg))
1919 return -EINVAL;
1920 if (cmsg->cmsg_level != SOL_SOCKET)
1921 continue;
1922 switch (cmsg->cmsg_type) {
1923 case SO_MARK:
1924 if (!ns_capable(sock_net(sk)->user_ns, CAP_NET_ADMIN))
1925 return -EPERM;
1926 if (cmsg->cmsg_len != CMSG_LEN(sizeof(u32)))
1927 return -EINVAL;
1928 sockc->mark = *(u32 *)CMSG_DATA(cmsg);
1929 break;
1930 default:
1931 return -EINVAL;
1932 }
1933 }
1934 return 0;
1935}
1936EXPORT_SYMBOL(sock_cmsg_send);
1937
Eric Dumazet5640f762012-09-23 23:04:42 +00001938/* On 32bit arches, an skb frag is limited to 2^15 */
1939#define SKB_FRAG_PAGE_ORDER get_order(32768)
1940
Eric Dumazet400dfd32013-10-17 16:27:07 -07001941/**
1942 * skb_page_frag_refill - check that a page_frag contains enough room
1943 * @sz: minimum size of the fragment we want to get
1944 * @pfrag: pointer to page_frag
Eric Dumazet82d5e2b2014-09-08 04:00:00 -07001945 * @gfp: priority for memory allocation
Eric Dumazet400dfd32013-10-17 16:27:07 -07001946 *
1947 * Note: While this allocator tries to use high order pages, there is
1948 * no guarantee that allocations succeed. Therefore, @sz MUST be
1949 * less or equal than PAGE_SIZE.
1950 */
Eric Dumazetd9b29382014-08-27 20:49:34 -07001951bool skb_page_frag_refill(unsigned int sz, struct page_frag *pfrag, gfp_t gfp)
Eric Dumazet5640f762012-09-23 23:04:42 +00001952{
Eric Dumazet5640f762012-09-23 23:04:42 +00001953 if (pfrag->page) {
1954 if (atomic_read(&pfrag->page->_count) == 1) {
1955 pfrag->offset = 0;
1956 return true;
1957 }
Eric Dumazet400dfd32013-10-17 16:27:07 -07001958 if (pfrag->offset + sz <= pfrag->size)
Eric Dumazet5640f762012-09-23 23:04:42 +00001959 return true;
1960 put_page(pfrag->page);
1961 }
1962
Eric Dumazetd9b29382014-08-27 20:49:34 -07001963 pfrag->offset = 0;
1964 if (SKB_FRAG_PAGE_ORDER) {
Mel Gormand0164ad2015-11-06 16:28:21 -08001965 /* Avoid direct reclaim but allow kswapd to wake */
1966 pfrag->page = alloc_pages((gfp & ~__GFP_DIRECT_RECLAIM) |
1967 __GFP_COMP | __GFP_NOWARN |
1968 __GFP_NORETRY,
Eric Dumazetd9b29382014-08-27 20:49:34 -07001969 SKB_FRAG_PAGE_ORDER);
Eric Dumazet5640f762012-09-23 23:04:42 +00001970 if (likely(pfrag->page)) {
Eric Dumazetd9b29382014-08-27 20:49:34 -07001971 pfrag->size = PAGE_SIZE << SKB_FRAG_PAGE_ORDER;
Eric Dumazet5640f762012-09-23 23:04:42 +00001972 return true;
1973 }
Eric Dumazetd9b29382014-08-27 20:49:34 -07001974 }
1975 pfrag->page = alloc_page(gfp);
1976 if (likely(pfrag->page)) {
1977 pfrag->size = PAGE_SIZE;
1978 return true;
1979 }
Eric Dumazet400dfd32013-10-17 16:27:07 -07001980 return false;
1981}
1982EXPORT_SYMBOL(skb_page_frag_refill);
1983
1984bool sk_page_frag_refill(struct sock *sk, struct page_frag *pfrag)
1985{
1986 if (likely(skb_page_frag_refill(32U, pfrag, sk->sk_allocation)))
1987 return true;
1988
Eric Dumazet5640f762012-09-23 23:04:42 +00001989 sk_enter_memory_pressure(sk);
1990 sk_stream_moderate_sndbuf(sk);
1991 return false;
1992}
1993EXPORT_SYMBOL(sk_page_frag_refill);
1994
Linus Torvalds1da177e2005-04-16 15:20:36 -07001995static void __lock_sock(struct sock *sk)
Namhyung Kimf39234d2010-09-08 03:48:48 +00001996 __releases(&sk->sk_lock.slock)
1997 __acquires(&sk->sk_lock.slock)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001998{
1999 DEFINE_WAIT(wait);
2000
Stephen Hemmingere71a4782007-04-10 20:10:33 -07002001 for (;;) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07002002 prepare_to_wait_exclusive(&sk->sk_lock.wq, &wait,
2003 TASK_UNINTERRUPTIBLE);
2004 spin_unlock_bh(&sk->sk_lock.slock);
2005 schedule();
2006 spin_lock_bh(&sk->sk_lock.slock);
Stephen Hemmingere71a4782007-04-10 20:10:33 -07002007 if (!sock_owned_by_user(sk))
Linus Torvalds1da177e2005-04-16 15:20:36 -07002008 break;
2009 }
2010 finish_wait(&sk->sk_lock.wq, &wait);
2011}
2012
2013static void __release_sock(struct sock *sk)
Namhyung Kimf39234d2010-09-08 03:48:48 +00002014 __releases(&sk->sk_lock.slock)
2015 __acquires(&sk->sk_lock.slock)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002016{
2017 struct sk_buff *skb = sk->sk_backlog.head;
2018
2019 do {
2020 sk->sk_backlog.head = sk->sk_backlog.tail = NULL;
2021 bh_unlock_sock(sk);
2022
2023 do {
2024 struct sk_buff *next = skb->next;
2025
Eric Dumazete4cbb022012-04-30 16:07:09 +00002026 prefetch(next);
Eric Dumazet7fee2262010-05-11 23:19:48 +00002027 WARN_ON_ONCE(skb_dst_is_noref(skb));
Linus Torvalds1da177e2005-04-16 15:20:36 -07002028 skb->next = NULL;
Peter Zijlstrac57943a2008-10-07 14:18:42 -07002029 sk_backlog_rcv(sk, skb);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002030
2031 /*
2032 * We are in process context here with softirqs
2033 * disabled, use cond_resched_softirq() to preempt.
2034 * This is safe to do because we've taken the backlog
2035 * queue private:
2036 */
2037 cond_resched_softirq();
2038
2039 skb = next;
2040 } while (skb != NULL);
2041
2042 bh_lock_sock(sk);
Stephen Hemmingere71a4782007-04-10 20:10:33 -07002043 } while ((skb = sk->sk_backlog.head) != NULL);
Zhu Yi8eae9392010-03-04 18:01:40 +00002044
2045 /*
2046 * Doing the zeroing here guarantee we can not loop forever
2047 * while a wild producer attempts to flood us.
2048 */
2049 sk->sk_backlog.len = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002050}
2051
2052/**
2053 * sk_wait_data - wait for data to arrive at sk_receive_queue
Pavel Pisa4dc3b162005-05-01 08:59:25 -07002054 * @sk: sock to wait on
2055 * @timeo: for how long
Sabrina Dubrocadfbafc92015-07-24 18:19:25 +02002056 * @skb: last skb seen on sk_receive_queue
Linus Torvalds1da177e2005-04-16 15:20:36 -07002057 *
2058 * Now socket state including sk->sk_err is changed only under lock,
2059 * hence we may omit checks after joining wait queue.
2060 * We check receive queue before schedule() only as optimization;
2061 * it is very likely that release_sock() added new data.
2062 */
Sabrina Dubrocadfbafc92015-07-24 18:19:25 +02002063int sk_wait_data(struct sock *sk, long *timeo, const struct sk_buff *skb)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002064{
2065 int rc;
2066 DEFINE_WAIT(wait);
2067
Eric Dumazetaa395142010-04-20 13:03:51 +00002068 prepare_to_wait(sk_sleep(sk), &wait, TASK_INTERRUPTIBLE);
Eric Dumazet9cd3e072015-11-29 20:03:10 -08002069 sk_set_bit(SOCKWQ_ASYNC_WAITDATA, sk);
Sabrina Dubrocadfbafc92015-07-24 18:19:25 +02002070 rc = sk_wait_event(sk, timeo, skb_peek_tail(&sk->sk_receive_queue) != skb);
Eric Dumazet9cd3e072015-11-29 20:03:10 -08002071 sk_clear_bit(SOCKWQ_ASYNC_WAITDATA, sk);
Eric Dumazetaa395142010-04-20 13:03:51 +00002072 finish_wait(sk_sleep(sk), &wait);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002073 return rc;
2074}
Linus Torvalds1da177e2005-04-16 15:20:36 -07002075EXPORT_SYMBOL(sk_wait_data);
2076
Hideo Aoki3ab224b2007-12-31 00:11:19 -08002077/**
2078 * __sk_mem_schedule - increase sk_forward_alloc and memory_allocated
2079 * @sk: socket
2080 * @size: memory size to allocate
2081 * @kind: allocation type
2082 *
2083 * If kind is SK_MEM_SEND, it means wmem allocation. Otherwise it means
2084 * rmem allocation. This function assumes that protocols which have
2085 * memory_pressure use sk_wmem_queued as write buffer accounting.
2086 */
2087int __sk_mem_schedule(struct sock *sk, int size, int kind)
2088{
2089 struct proto *prot = sk->sk_prot;
2090 int amt = sk_mem_pages(size);
Eric Dumazet8d987e52010-11-09 23:24:26 +00002091 long allocated;
Glauber Costae1aab162011-12-11 21:47:03 +00002092 int parent_status = UNDER_LIMIT;
Hideo Aoki3ab224b2007-12-31 00:11:19 -08002093
2094 sk->sk_forward_alloc += amt * SK_MEM_QUANTUM;
Glauber Costa180d8cd2011-12-11 21:47:02 +00002095
Glauber Costae1aab162011-12-11 21:47:03 +00002096 allocated = sk_memory_allocated_add(sk, amt, &parent_status);
Hideo Aoki3ab224b2007-12-31 00:11:19 -08002097
2098 /* Under limit. */
Glauber Costae1aab162011-12-11 21:47:03 +00002099 if (parent_status == UNDER_LIMIT &&
2100 allocated <= sk_prot_mem_limits(sk, 0)) {
Glauber Costa180d8cd2011-12-11 21:47:02 +00002101 sk_leave_memory_pressure(sk);
Hideo Aoki3ab224b2007-12-31 00:11:19 -08002102 return 1;
2103 }
2104
Glauber Costae1aab162011-12-11 21:47:03 +00002105 /* Under pressure. (we or our parents) */
2106 if ((parent_status > SOFT_LIMIT) ||
2107 allocated > sk_prot_mem_limits(sk, 1))
Glauber Costa180d8cd2011-12-11 21:47:02 +00002108 sk_enter_memory_pressure(sk);
Hideo Aoki3ab224b2007-12-31 00:11:19 -08002109
Glauber Costae1aab162011-12-11 21:47:03 +00002110 /* Over hard limit (we or our parents) */
2111 if ((parent_status == OVER_LIMIT) ||
2112 (allocated > sk_prot_mem_limits(sk, 2)))
Hideo Aoki3ab224b2007-12-31 00:11:19 -08002113 goto suppress_allocation;
2114
2115 /* guarantee minimum buffer size under pressure */
2116 if (kind == SK_MEM_RECV) {
2117 if (atomic_read(&sk->sk_rmem_alloc) < prot->sysctl_rmem[0])
2118 return 1;
Glauber Costa180d8cd2011-12-11 21:47:02 +00002119
Hideo Aoki3ab224b2007-12-31 00:11:19 -08002120 } else { /* SK_MEM_SEND */
2121 if (sk->sk_type == SOCK_STREAM) {
2122 if (sk->sk_wmem_queued < prot->sysctl_wmem[0])
2123 return 1;
2124 } else if (atomic_read(&sk->sk_wmem_alloc) <
2125 prot->sysctl_wmem[0])
2126 return 1;
2127 }
2128
Glauber Costa180d8cd2011-12-11 21:47:02 +00002129 if (sk_has_memory_pressure(sk)) {
Eric Dumazet17483762008-11-25 21:16:35 -08002130 int alloc;
2131
Glauber Costa180d8cd2011-12-11 21:47:02 +00002132 if (!sk_under_memory_pressure(sk))
Eric Dumazet17483762008-11-25 21:16:35 -08002133 return 1;
Glauber Costa180d8cd2011-12-11 21:47:02 +00002134 alloc = sk_sockets_allocated_read_positive(sk);
2135 if (sk_prot_mem_limits(sk, 2) > alloc *
Hideo Aoki3ab224b2007-12-31 00:11:19 -08002136 sk_mem_pages(sk->sk_wmem_queued +
2137 atomic_read(&sk->sk_rmem_alloc) +
2138 sk->sk_forward_alloc))
2139 return 1;
2140 }
2141
2142suppress_allocation:
2143
2144 if (kind == SK_MEM_SEND && sk->sk_type == SOCK_STREAM) {
2145 sk_stream_moderate_sndbuf(sk);
2146
2147 /* Fail only if socket is _under_ its sndbuf.
2148 * In this case we cannot block, so that we have to fail.
2149 */
2150 if (sk->sk_wmem_queued + size >= sk->sk_sndbuf)
2151 return 1;
2152 }
2153
Satoru Moriya3847ce32011-06-17 12:00:03 +00002154 trace_sock_exceed_buf_limit(sk, prot, allocated);
2155
Hideo Aoki3ab224b2007-12-31 00:11:19 -08002156 /* Alas. Undo changes. */
2157 sk->sk_forward_alloc -= amt * SK_MEM_QUANTUM;
Glauber Costa180d8cd2011-12-11 21:47:02 +00002158
Glauber Costa0e90b312012-01-20 04:57:16 +00002159 sk_memory_allocated_sub(sk, amt);
Glauber Costa180d8cd2011-12-11 21:47:02 +00002160
Hideo Aoki3ab224b2007-12-31 00:11:19 -08002161 return 0;
2162}
Hideo Aoki3ab224b2007-12-31 00:11:19 -08002163EXPORT_SYMBOL(__sk_mem_schedule);
2164
2165/**
Jean Sacren69dba9b2015-08-27 18:05:49 -06002166 * __sk_mem_reclaim - reclaim memory_allocated
Hideo Aoki3ab224b2007-12-31 00:11:19 -08002167 * @sk: socket
Eric Dumazet1a24e042015-05-15 12:39:25 -07002168 * @amount: number of bytes (rounded down to a SK_MEM_QUANTUM multiple)
Hideo Aoki3ab224b2007-12-31 00:11:19 -08002169 */
Eric Dumazet1a24e042015-05-15 12:39:25 -07002170void __sk_mem_reclaim(struct sock *sk, int amount)
Hideo Aoki3ab224b2007-12-31 00:11:19 -08002171{
Eric Dumazet1a24e042015-05-15 12:39:25 -07002172 amount >>= SK_MEM_QUANTUM_SHIFT;
2173 sk_memory_allocated_sub(sk, amount);
2174 sk->sk_forward_alloc -= amount << SK_MEM_QUANTUM_SHIFT;
Hideo Aoki3ab224b2007-12-31 00:11:19 -08002175
Glauber Costa180d8cd2011-12-11 21:47:02 +00002176 if (sk_under_memory_pressure(sk) &&
2177 (sk_memory_allocated(sk) < sk_prot_mem_limits(sk, 0)))
2178 sk_leave_memory_pressure(sk);
Hideo Aoki3ab224b2007-12-31 00:11:19 -08002179}
Hideo Aoki3ab224b2007-12-31 00:11:19 -08002180EXPORT_SYMBOL(__sk_mem_reclaim);
2181
2182
Linus Torvalds1da177e2005-04-16 15:20:36 -07002183/*
2184 * Set of default routines for initialising struct proto_ops when
2185 * the protocol does not support a particular function. In certain
2186 * cases where it makes no sense for a protocol to have a "do nothing"
2187 * function, some default processing is provided.
2188 */
2189
2190int sock_no_bind(struct socket *sock, struct sockaddr *saddr, int len)
2191{
2192 return -EOPNOTSUPP;
2193}
Eric Dumazet2a915252009-05-27 11:30:05 +00002194EXPORT_SYMBOL(sock_no_bind);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002195
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09002196int sock_no_connect(struct socket *sock, struct sockaddr *saddr,
Linus Torvalds1da177e2005-04-16 15:20:36 -07002197 int len, int flags)
2198{
2199 return -EOPNOTSUPP;
2200}
Eric Dumazet2a915252009-05-27 11:30:05 +00002201EXPORT_SYMBOL(sock_no_connect);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002202
2203int sock_no_socketpair(struct socket *sock1, struct socket *sock2)
2204{
2205 return -EOPNOTSUPP;
2206}
Eric Dumazet2a915252009-05-27 11:30:05 +00002207EXPORT_SYMBOL(sock_no_socketpair);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002208
2209int sock_no_accept(struct socket *sock, struct socket *newsock, int flags)
2210{
2211 return -EOPNOTSUPP;
2212}
Eric Dumazet2a915252009-05-27 11:30:05 +00002213EXPORT_SYMBOL(sock_no_accept);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002214
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09002215int sock_no_getname(struct socket *sock, struct sockaddr *saddr,
Linus Torvalds1da177e2005-04-16 15:20:36 -07002216 int *len, int peer)
2217{
2218 return -EOPNOTSUPP;
2219}
Eric Dumazet2a915252009-05-27 11:30:05 +00002220EXPORT_SYMBOL(sock_no_getname);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002221
Eric Dumazet2a915252009-05-27 11:30:05 +00002222unsigned int sock_no_poll(struct file *file, struct socket *sock, poll_table *pt)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002223{
2224 return 0;
2225}
Eric Dumazet2a915252009-05-27 11:30:05 +00002226EXPORT_SYMBOL(sock_no_poll);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002227
2228int sock_no_ioctl(struct socket *sock, unsigned int cmd, unsigned long arg)
2229{
2230 return -EOPNOTSUPP;
2231}
Eric Dumazet2a915252009-05-27 11:30:05 +00002232EXPORT_SYMBOL(sock_no_ioctl);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002233
2234int sock_no_listen(struct socket *sock, int backlog)
2235{
2236 return -EOPNOTSUPP;
2237}
Eric Dumazet2a915252009-05-27 11:30:05 +00002238EXPORT_SYMBOL(sock_no_listen);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002239
2240int sock_no_shutdown(struct socket *sock, int how)
2241{
2242 return -EOPNOTSUPP;
2243}
Eric Dumazet2a915252009-05-27 11:30:05 +00002244EXPORT_SYMBOL(sock_no_shutdown);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002245
2246int sock_no_setsockopt(struct socket *sock, int level, int optname,
David S. Millerb7058842009-09-30 16:12:20 -07002247 char __user *optval, unsigned int optlen)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002248{
2249 return -EOPNOTSUPP;
2250}
Eric Dumazet2a915252009-05-27 11:30:05 +00002251EXPORT_SYMBOL(sock_no_setsockopt);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002252
2253int sock_no_getsockopt(struct socket *sock, int level, int optname,
2254 char __user *optval, int __user *optlen)
2255{
2256 return -EOPNOTSUPP;
2257}
Eric Dumazet2a915252009-05-27 11:30:05 +00002258EXPORT_SYMBOL(sock_no_getsockopt);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002259
Ying Xue1b784142015-03-02 15:37:48 +08002260int sock_no_sendmsg(struct socket *sock, struct msghdr *m, size_t len)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002261{
2262 return -EOPNOTSUPP;
2263}
Eric Dumazet2a915252009-05-27 11:30:05 +00002264EXPORT_SYMBOL(sock_no_sendmsg);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002265
Ying Xue1b784142015-03-02 15:37:48 +08002266int sock_no_recvmsg(struct socket *sock, struct msghdr *m, size_t len,
2267 int flags)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002268{
2269 return -EOPNOTSUPP;
2270}
Eric Dumazet2a915252009-05-27 11:30:05 +00002271EXPORT_SYMBOL(sock_no_recvmsg);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002272
2273int sock_no_mmap(struct file *file, struct socket *sock, struct vm_area_struct *vma)
2274{
2275 /* Mirror missing mmap method error code */
2276 return -ENODEV;
2277}
Eric Dumazet2a915252009-05-27 11:30:05 +00002278EXPORT_SYMBOL(sock_no_mmap);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002279
2280ssize_t sock_no_sendpage(struct socket *sock, struct page *page, int offset, size_t size, int flags)
2281{
2282 ssize_t res;
2283 struct msghdr msg = {.msg_flags = flags};
2284 struct kvec iov;
2285 char *kaddr = kmap(page);
2286 iov.iov_base = kaddr + offset;
2287 iov.iov_len = size;
2288 res = kernel_sendmsg(sock, &msg, &iov, 1, size);
2289 kunmap(page);
2290 return res;
2291}
Eric Dumazet2a915252009-05-27 11:30:05 +00002292EXPORT_SYMBOL(sock_no_sendpage);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002293
2294/*
2295 * Default Socket Callbacks
2296 */
2297
2298static void sock_def_wakeup(struct sock *sk)
2299{
Eric Dumazet43815482010-04-29 11:01:49 +00002300 struct socket_wq *wq;
2301
2302 rcu_read_lock();
2303 wq = rcu_dereference(sk->sk_wq);
Herbert Xu1ce0bf52015-11-26 13:55:39 +08002304 if (skwq_has_sleeper(wq))
Eric Dumazet43815482010-04-29 11:01:49 +00002305 wake_up_interruptible_all(&wq->wait);
2306 rcu_read_unlock();
Linus Torvalds1da177e2005-04-16 15:20:36 -07002307}
2308
2309static void sock_def_error_report(struct sock *sk)
2310{
Eric Dumazet43815482010-04-29 11:01:49 +00002311 struct socket_wq *wq;
2312
2313 rcu_read_lock();
2314 wq = rcu_dereference(sk->sk_wq);
Herbert Xu1ce0bf52015-11-26 13:55:39 +08002315 if (skwq_has_sleeper(wq))
Eric Dumazet43815482010-04-29 11:01:49 +00002316 wake_up_interruptible_poll(&wq->wait, POLLERR);
Pavel Emelyanov8d8ad9d2007-11-26 20:10:50 +08002317 sk_wake_async(sk, SOCK_WAKE_IO, POLL_ERR);
Eric Dumazet43815482010-04-29 11:01:49 +00002318 rcu_read_unlock();
Linus Torvalds1da177e2005-04-16 15:20:36 -07002319}
2320
David S. Miller676d2362014-04-11 16:15:36 -04002321static void sock_def_readable(struct sock *sk)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002322{
Eric Dumazet43815482010-04-29 11:01:49 +00002323 struct socket_wq *wq;
2324
2325 rcu_read_lock();
2326 wq = rcu_dereference(sk->sk_wq);
Herbert Xu1ce0bf52015-11-26 13:55:39 +08002327 if (skwq_has_sleeper(wq))
Eric Dumazet2c6607c2011-01-06 10:54:29 -08002328 wake_up_interruptible_sync_poll(&wq->wait, POLLIN | POLLPRI |
Davide Libenzi37e55402009-03-31 15:24:21 -07002329 POLLRDNORM | POLLRDBAND);
Pavel Emelyanov8d8ad9d2007-11-26 20:10:50 +08002330 sk_wake_async(sk, SOCK_WAKE_WAITD, POLL_IN);
Eric Dumazet43815482010-04-29 11:01:49 +00002331 rcu_read_unlock();
Linus Torvalds1da177e2005-04-16 15:20:36 -07002332}
2333
2334static void sock_def_write_space(struct sock *sk)
2335{
Eric Dumazet43815482010-04-29 11:01:49 +00002336 struct socket_wq *wq;
2337
2338 rcu_read_lock();
Linus Torvalds1da177e2005-04-16 15:20:36 -07002339
2340 /* Do not wake up a writer until he can make "significant"
2341 * progress. --DaveM
2342 */
Stephen Hemmingere71a4782007-04-10 20:10:33 -07002343 if ((atomic_read(&sk->sk_wmem_alloc) << 1) <= sk->sk_sndbuf) {
Eric Dumazet43815482010-04-29 11:01:49 +00002344 wq = rcu_dereference(sk->sk_wq);
Herbert Xu1ce0bf52015-11-26 13:55:39 +08002345 if (skwq_has_sleeper(wq))
Eric Dumazet43815482010-04-29 11:01:49 +00002346 wake_up_interruptible_sync_poll(&wq->wait, POLLOUT |
Davide Libenzi37e55402009-03-31 15:24:21 -07002347 POLLWRNORM | POLLWRBAND);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002348
2349 /* Should agree with poll, otherwise some programs break */
2350 if (sock_writeable(sk))
Pavel Emelyanov8d8ad9d2007-11-26 20:10:50 +08002351 sk_wake_async(sk, SOCK_WAKE_SPACE, POLL_OUT);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002352 }
2353
Eric Dumazet43815482010-04-29 11:01:49 +00002354 rcu_read_unlock();
Linus Torvalds1da177e2005-04-16 15:20:36 -07002355}
2356
2357static void sock_def_destruct(struct sock *sk)
2358{
Linus Torvalds1da177e2005-04-16 15:20:36 -07002359}
2360
2361void sk_send_sigurg(struct sock *sk)
2362{
2363 if (sk->sk_socket && sk->sk_socket->file)
2364 if (send_sigurg(&sk->sk_socket->file->f_owner))
Pavel Emelyanov8d8ad9d2007-11-26 20:10:50 +08002365 sk_wake_async(sk, SOCK_WAKE_URG, POLL_PRI);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002366}
Eric Dumazet2a915252009-05-27 11:30:05 +00002367EXPORT_SYMBOL(sk_send_sigurg);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002368
2369void sk_reset_timer(struct sock *sk, struct timer_list* timer,
2370 unsigned long expires)
2371{
2372 if (!mod_timer(timer, expires))
2373 sock_hold(sk);
2374}
Linus Torvalds1da177e2005-04-16 15:20:36 -07002375EXPORT_SYMBOL(sk_reset_timer);
2376
2377void sk_stop_timer(struct sock *sk, struct timer_list* timer)
2378{
Ying Xue25cc4ae2013-02-03 20:32:57 +00002379 if (del_timer(timer))
Linus Torvalds1da177e2005-04-16 15:20:36 -07002380 __sock_put(sk);
2381}
Linus Torvalds1da177e2005-04-16 15:20:36 -07002382EXPORT_SYMBOL(sk_stop_timer);
2383
2384void sock_init_data(struct socket *sock, struct sock *sk)
2385{
2386 skb_queue_head_init(&sk->sk_receive_queue);
2387 skb_queue_head_init(&sk->sk_write_queue);
2388 skb_queue_head_init(&sk->sk_error_queue);
2389
2390 sk->sk_send_head = NULL;
2391
2392 init_timer(&sk->sk_timer);
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09002393
Linus Torvalds1da177e2005-04-16 15:20:36 -07002394 sk->sk_allocation = GFP_KERNEL;
2395 sk->sk_rcvbuf = sysctl_rmem_default;
2396 sk->sk_sndbuf = sysctl_wmem_default;
2397 sk->sk_state = TCP_CLOSE;
David S. Miller972692e2008-06-17 22:41:38 -07002398 sk_set_socket(sk, sock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002399
2400 sock_set_flag(sk, SOCK_ZAPPED);
2401
Stephen Hemmingere71a4782007-04-10 20:10:33 -07002402 if (sock) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07002403 sk->sk_type = sock->type;
Eric Dumazet43815482010-04-29 11:01:49 +00002404 sk->sk_wq = sock->wq;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002405 sock->sk = sk;
2406 } else
Eric Dumazet43815482010-04-29 11:01:49 +00002407 sk->sk_wq = NULL;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002408
Linus Torvalds1da177e2005-04-16 15:20:36 -07002409 rwlock_init(&sk->sk_callback_lock);
Peter Zijlstra443aef02007-07-19 01:49:00 -07002410 lockdep_set_class_and_name(&sk->sk_callback_lock,
2411 af_callback_keys + sk->sk_family,
2412 af_family_clock_key_strings[sk->sk_family]);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002413
2414 sk->sk_state_change = sock_def_wakeup;
2415 sk->sk_data_ready = sock_def_readable;
2416 sk->sk_write_space = sock_def_write_space;
2417 sk->sk_error_report = sock_def_error_report;
2418 sk->sk_destruct = sock_def_destruct;
2419
Eric Dumazet5640f762012-09-23 23:04:42 +00002420 sk->sk_frag.page = NULL;
2421 sk->sk_frag.offset = 0;
Pavel Emelyanovef64a542012-02-21 07:31:34 +00002422 sk->sk_peek_off = -1;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002423
Eric W. Biederman109f6e32010-06-13 03:30:14 +00002424 sk->sk_peer_pid = NULL;
2425 sk->sk_peer_cred = NULL;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002426 sk->sk_write_pending = 0;
2427 sk->sk_rcvlowat = 1;
2428 sk->sk_rcvtimeo = MAX_SCHEDULE_TIMEOUT;
2429 sk->sk_sndtimeo = MAX_SCHEDULE_TIMEOUT;
2430
Eric Dumazetf37f0af2008-04-13 21:39:26 -07002431 sk->sk_stamp = ktime_set(-1L, 0);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002432
Cong Wange0d10952013-08-01 11:10:25 +08002433#ifdef CONFIG_NET_RX_BUSY_POLL
Eliezer Tamir06021292013-06-10 11:39:50 +03002434 sk->sk_napi_id = 0;
Eliezer Tamir64b0dc52013-07-10 17:13:36 +03002435 sk->sk_ll_usec = sysctl_net_busy_read;
Eliezer Tamir06021292013-06-10 11:39:50 +03002436#endif
2437
Eric Dumazet62748f32013-09-24 08:20:52 -07002438 sk->sk_max_pacing_rate = ~0U;
Eric Dumazet7eec4172013-10-08 15:16:00 -07002439 sk->sk_pacing_rate = ~0U;
Eric Dumazet70da2682015-10-08 19:33:21 -07002440 sk->sk_incoming_cpu = -1;
Eric Dumazet4dc6dc72009-07-15 23:13:10 +00002441 /*
2442 * Before updating sk_refcnt, we must commit prior changes to memory
2443 * (Documentation/RCU/rculist_nulls.txt for details)
2444 */
2445 smp_wmb();
Linus Torvalds1da177e2005-04-16 15:20:36 -07002446 atomic_set(&sk->sk_refcnt, 1);
Wang Chen33c732c2007-11-13 20:30:01 -08002447 atomic_set(&sk->sk_drops, 0);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002448}
Eric Dumazet2a915252009-05-27 11:30:05 +00002449EXPORT_SYMBOL(sock_init_data);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002450
Harvey Harrisonb5606c22008-02-13 15:03:16 -08002451void lock_sock_nested(struct sock *sk, int subclass)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002452{
2453 might_sleep();
Ingo Molnara5b5bb92006-07-03 00:25:35 -07002454 spin_lock_bh(&sk->sk_lock.slock);
John Heffnerd2e91172007-09-12 10:44:19 +02002455 if (sk->sk_lock.owned)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002456 __lock_sock(sk);
John Heffnerd2e91172007-09-12 10:44:19 +02002457 sk->sk_lock.owned = 1;
Ingo Molnara5b5bb92006-07-03 00:25:35 -07002458 spin_unlock(&sk->sk_lock.slock);
2459 /*
2460 * The sk_lock has mutex_lock() semantics here:
2461 */
Peter Zijlstrafcc70d52006-11-08 22:44:35 -08002462 mutex_acquire(&sk->sk_lock.dep_map, subclass, 0, _RET_IP_);
Ingo Molnara5b5bb92006-07-03 00:25:35 -07002463 local_bh_enable();
Linus Torvalds1da177e2005-04-16 15:20:36 -07002464}
Peter Zijlstrafcc70d52006-11-08 22:44:35 -08002465EXPORT_SYMBOL(lock_sock_nested);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002466
Harvey Harrisonb5606c22008-02-13 15:03:16 -08002467void release_sock(struct sock *sk)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002468{
Ingo Molnara5b5bb92006-07-03 00:25:35 -07002469 /*
2470 * The sk_lock has mutex_unlock() semantics:
2471 */
2472 mutex_release(&sk->sk_lock.dep_map, 1, _RET_IP_);
2473
2474 spin_lock_bh(&sk->sk_lock.slock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002475 if (sk->sk_backlog.tail)
2476 __release_sock(sk);
Eric Dumazet46d3cea2012-07-11 05:50:31 +00002477
Eric Dumazetc3f9b012014-03-10 09:50:11 -07002478 /* Warning : release_cb() might need to release sk ownership,
2479 * ie call sock_release_ownership(sk) before us.
2480 */
Eric Dumazet46d3cea2012-07-11 05:50:31 +00002481 if (sk->sk_prot->release_cb)
2482 sk->sk_prot->release_cb(sk);
2483
Eric Dumazetc3f9b012014-03-10 09:50:11 -07002484 sock_release_ownership(sk);
Ingo Molnara5b5bb92006-07-03 00:25:35 -07002485 if (waitqueue_active(&sk->sk_lock.wq))
2486 wake_up(&sk->sk_lock.wq);
2487 spin_unlock_bh(&sk->sk_lock.slock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002488}
2489EXPORT_SYMBOL(release_sock);
2490
Eric Dumazet8a74ad62010-05-26 19:20:18 +00002491/**
2492 * lock_sock_fast - fast version of lock_sock
2493 * @sk: socket
2494 *
2495 * This version should be used for very small section, where process wont block
2496 * return false if fast path is taken
2497 * sk_lock.slock locked, owned = 0, BH disabled
2498 * return true if slow path is taken
2499 * sk_lock.slock unlocked, owned = 1, BH enabled
2500 */
2501bool lock_sock_fast(struct sock *sk)
2502{
2503 might_sleep();
2504 spin_lock_bh(&sk->sk_lock.slock);
2505
2506 if (!sk->sk_lock.owned)
2507 /*
2508 * Note : We must disable BH
2509 */
2510 return false;
2511
2512 __lock_sock(sk);
2513 sk->sk_lock.owned = 1;
2514 spin_unlock(&sk->sk_lock.slock);
2515 /*
2516 * The sk_lock has mutex_lock() semantics here:
2517 */
2518 mutex_acquire(&sk->sk_lock.dep_map, 0, 0, _RET_IP_);
2519 local_bh_enable();
2520 return true;
2521}
2522EXPORT_SYMBOL(lock_sock_fast);
2523
Linus Torvalds1da177e2005-04-16 15:20:36 -07002524int sock_get_timestamp(struct sock *sk, struct timeval __user *userstamp)
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09002525{
Eric Dumazetb7aa0bf2007-04-19 16:16:32 -07002526 struct timeval tv;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002527 if (!sock_flag(sk, SOCK_TIMESTAMP))
Patrick Ohly20d49472009-02-12 05:03:38 +00002528 sock_enable_timestamp(sk, SOCK_TIMESTAMP);
Eric Dumazetb7aa0bf2007-04-19 16:16:32 -07002529 tv = ktime_to_timeval(sk->sk_stamp);
2530 if (tv.tv_sec == -1)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002531 return -ENOENT;
Eric Dumazetb7aa0bf2007-04-19 16:16:32 -07002532 if (tv.tv_sec == 0) {
2533 sk->sk_stamp = ktime_get_real();
2534 tv = ktime_to_timeval(sk->sk_stamp);
2535 }
2536 return copy_to_user(userstamp, &tv, sizeof(tv)) ? -EFAULT : 0;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09002537}
Linus Torvalds1da177e2005-04-16 15:20:36 -07002538EXPORT_SYMBOL(sock_get_timestamp);
2539
Eric Dumazetae40eb12007-03-18 17:33:16 -07002540int sock_get_timestampns(struct sock *sk, struct timespec __user *userstamp)
2541{
2542 struct timespec ts;
2543 if (!sock_flag(sk, SOCK_TIMESTAMP))
Patrick Ohly20d49472009-02-12 05:03:38 +00002544 sock_enable_timestamp(sk, SOCK_TIMESTAMP);
Eric Dumazetae40eb12007-03-18 17:33:16 -07002545 ts = ktime_to_timespec(sk->sk_stamp);
2546 if (ts.tv_sec == -1)
2547 return -ENOENT;
2548 if (ts.tv_sec == 0) {
2549 sk->sk_stamp = ktime_get_real();
2550 ts = ktime_to_timespec(sk->sk_stamp);
2551 }
2552 return copy_to_user(userstamp, &ts, sizeof(ts)) ? -EFAULT : 0;
2553}
2554EXPORT_SYMBOL(sock_get_timestampns);
2555
Patrick Ohly20d49472009-02-12 05:03:38 +00002556void sock_enable_timestamp(struct sock *sk, int flag)
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09002557{
Patrick Ohly20d49472009-02-12 05:03:38 +00002558 if (!sock_flag(sk, flag)) {
Eric Dumazet08e29af2011-11-28 12:04:18 +00002559 unsigned long previous_flags = sk->sk_flags;
2560
Patrick Ohly20d49472009-02-12 05:03:38 +00002561 sock_set_flag(sk, flag);
2562 /*
2563 * we just set one of the two flags which require net
2564 * time stamping, but time stamping might have been on
2565 * already because of the other one
2566 */
Hannes Frederic Sowa080a2702015-10-26 13:51:37 +01002567 if (sock_needs_netstamp(sk) &&
2568 !(previous_flags & SK_FLAGS_TIMESTAMP))
Patrick Ohly20d49472009-02-12 05:03:38 +00002569 net_enable_timestamp();
Linus Torvalds1da177e2005-04-16 15:20:36 -07002570 }
2571}
Linus Torvalds1da177e2005-04-16 15:20:36 -07002572
Richard Cochrancb820f82013-07-19 19:40:09 +02002573int sock_recv_errqueue(struct sock *sk, struct msghdr *msg, int len,
2574 int level, int type)
2575{
2576 struct sock_exterr_skb *serr;
Willem de Bruijn364a9e92014-08-31 21:30:27 -04002577 struct sk_buff *skb;
Richard Cochrancb820f82013-07-19 19:40:09 +02002578 int copied, err;
2579
2580 err = -EAGAIN;
Willem de Bruijn364a9e92014-08-31 21:30:27 -04002581 skb = sock_dequeue_err_skb(sk);
Richard Cochrancb820f82013-07-19 19:40:09 +02002582 if (skb == NULL)
2583 goto out;
2584
2585 copied = skb->len;
2586 if (copied > len) {
2587 msg->msg_flags |= MSG_TRUNC;
2588 copied = len;
2589 }
David S. Miller51f3d022014-11-05 16:46:40 -05002590 err = skb_copy_datagram_msg(skb, 0, msg, copied);
Richard Cochrancb820f82013-07-19 19:40:09 +02002591 if (err)
2592 goto out_free_skb;
2593
2594 sock_recv_timestamp(msg, sk, skb);
2595
2596 serr = SKB_EXT_ERR(skb);
2597 put_cmsg(msg, level, type, sizeof(serr->ee), &serr->ee);
2598
2599 msg->msg_flags |= MSG_ERRQUEUE;
2600 err = copied;
2601
Richard Cochrancb820f82013-07-19 19:40:09 +02002602out_free_skb:
2603 kfree_skb(skb);
2604out:
2605 return err;
2606}
2607EXPORT_SYMBOL(sock_recv_errqueue);
2608
Linus Torvalds1da177e2005-04-16 15:20:36 -07002609/*
2610 * Get a socket option on an socket.
2611 *
2612 * FIX: POSIX 1003.1g is very ambiguous here. It states that
2613 * asynchronous errors should be reported by getsockopt. We assume
2614 * this means if you specify SO_ERROR (otherwise whats the point of it).
2615 */
2616int sock_common_getsockopt(struct socket *sock, int level, int optname,
2617 char __user *optval, int __user *optlen)
2618{
2619 struct sock *sk = sock->sk;
2620
2621 return sk->sk_prot->getsockopt(sk, level, optname, optval, optlen);
2622}
Linus Torvalds1da177e2005-04-16 15:20:36 -07002623EXPORT_SYMBOL(sock_common_getsockopt);
2624
Dmitry Mishin3fdadf72006-03-20 22:45:21 -08002625#ifdef CONFIG_COMPAT
Arnaldo Carvalho de Melo543d9cf2006-03-20 22:48:35 -08002626int compat_sock_common_getsockopt(struct socket *sock, int level, int optname,
2627 char __user *optval, int __user *optlen)
Dmitry Mishin3fdadf72006-03-20 22:45:21 -08002628{
2629 struct sock *sk = sock->sk;
2630
Johannes Berg1e51f952007-03-06 13:44:06 -08002631 if (sk->sk_prot->compat_getsockopt != NULL)
Arnaldo Carvalho de Melo543d9cf2006-03-20 22:48:35 -08002632 return sk->sk_prot->compat_getsockopt(sk, level, optname,
2633 optval, optlen);
Dmitry Mishin3fdadf72006-03-20 22:45:21 -08002634 return sk->sk_prot->getsockopt(sk, level, optname, optval, optlen);
2635}
2636EXPORT_SYMBOL(compat_sock_common_getsockopt);
2637#endif
2638
Ying Xue1b784142015-03-02 15:37:48 +08002639int sock_common_recvmsg(struct socket *sock, struct msghdr *msg, size_t size,
2640 int flags)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002641{
2642 struct sock *sk = sock->sk;
2643 int addr_len = 0;
2644 int err;
2645
Ying Xue1b784142015-03-02 15:37:48 +08002646 err = sk->sk_prot->recvmsg(sk, msg, size, flags & MSG_DONTWAIT,
Linus Torvalds1da177e2005-04-16 15:20:36 -07002647 flags & ~MSG_DONTWAIT, &addr_len);
2648 if (err >= 0)
2649 msg->msg_namelen = addr_len;
2650 return err;
2651}
Linus Torvalds1da177e2005-04-16 15:20:36 -07002652EXPORT_SYMBOL(sock_common_recvmsg);
2653
2654/*
2655 * Set socket options on an inet socket.
2656 */
2657int sock_common_setsockopt(struct socket *sock, int level, int optname,
David S. Millerb7058842009-09-30 16:12:20 -07002658 char __user *optval, unsigned int optlen)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002659{
2660 struct sock *sk = sock->sk;
2661
2662 return sk->sk_prot->setsockopt(sk, level, optname, optval, optlen);
2663}
Linus Torvalds1da177e2005-04-16 15:20:36 -07002664EXPORT_SYMBOL(sock_common_setsockopt);
2665
Dmitry Mishin3fdadf72006-03-20 22:45:21 -08002666#ifdef CONFIG_COMPAT
Arnaldo Carvalho de Melo543d9cf2006-03-20 22:48:35 -08002667int compat_sock_common_setsockopt(struct socket *sock, int level, int optname,
David S. Millerb7058842009-09-30 16:12:20 -07002668 char __user *optval, unsigned int optlen)
Dmitry Mishin3fdadf72006-03-20 22:45:21 -08002669{
2670 struct sock *sk = sock->sk;
2671
Arnaldo Carvalho de Melo543d9cf2006-03-20 22:48:35 -08002672 if (sk->sk_prot->compat_setsockopt != NULL)
2673 return sk->sk_prot->compat_setsockopt(sk, level, optname,
2674 optval, optlen);
Dmitry Mishin3fdadf72006-03-20 22:45:21 -08002675 return sk->sk_prot->setsockopt(sk, level, optname, optval, optlen);
2676}
2677EXPORT_SYMBOL(compat_sock_common_setsockopt);
2678#endif
2679
Linus Torvalds1da177e2005-04-16 15:20:36 -07002680void sk_common_release(struct sock *sk)
2681{
2682 if (sk->sk_prot->destroy)
2683 sk->sk_prot->destroy(sk);
2684
2685 /*
2686 * Observation: when sock_common_release is called, processes have
2687 * no access to socket. But net still has.
2688 * Step one, detach it from networking:
2689 *
2690 * A. Remove from hash tables.
2691 */
2692
2693 sk->sk_prot->unhash(sk);
2694
2695 /*
2696 * In this point socket cannot receive new packets, but it is possible
2697 * that some packets are in flight because some CPU runs receiver and
2698 * did hash table lookup before we unhashed socket. They will achieve
2699 * receive queue and will be purged by socket destructor.
2700 *
2701 * Also we still have packets pending on receive queue and probably,
2702 * our own packets waiting in device queues. sock_destroy will drain
2703 * receive queue, but transmitted packets will delay socket destruction
2704 * until the last reference will be released.
2705 */
2706
2707 sock_orphan(sk);
2708
2709 xfrm_sk_free_policy(sk);
2710
Arnaldo Carvalho de Meloe6848972005-08-09 19:45:38 -07002711 sk_refcnt_debug_release(sk);
Eric Dumazet5640f762012-09-23 23:04:42 +00002712
2713 if (sk->sk_frag.page) {
2714 put_page(sk->sk_frag.page);
2715 sk->sk_frag.page = NULL;
2716 }
2717
Linus Torvalds1da177e2005-04-16 15:20:36 -07002718 sock_put(sk);
2719}
Linus Torvalds1da177e2005-04-16 15:20:36 -07002720EXPORT_SYMBOL(sk_common_release);
2721
Pavel Emelyanov13ff3d62008-03-28 16:38:17 -07002722#ifdef CONFIG_PROC_FS
2723#define PROTO_INUSE_NR 64 /* should be enough for the first time */
Pavel Emelyanov1338d462008-03-28 16:38:43 -07002724struct prot_inuse {
2725 int val[PROTO_INUSE_NR];
2726};
Pavel Emelyanov13ff3d62008-03-28 16:38:17 -07002727
2728static DECLARE_BITMAP(proto_inuse_idx, PROTO_INUSE_NR);
Pavel Emelyanov70ee1152008-03-31 19:42:16 -07002729
2730#ifdef CONFIG_NET_NS
2731void sock_prot_inuse_add(struct net *net, struct proto *prot, int val)
2732{
Eric Dumazetd6d9ca02010-07-19 10:48:49 +00002733 __this_cpu_add(net->core.inuse->val[prot->inuse_idx], val);
Pavel Emelyanov70ee1152008-03-31 19:42:16 -07002734}
2735EXPORT_SYMBOL_GPL(sock_prot_inuse_add);
2736
2737int sock_prot_inuse_get(struct net *net, struct proto *prot)
2738{
2739 int cpu, idx = prot->inuse_idx;
2740 int res = 0;
2741
2742 for_each_possible_cpu(cpu)
2743 res += per_cpu_ptr(net->core.inuse, cpu)->val[idx];
2744
2745 return res >= 0 ? res : 0;
2746}
2747EXPORT_SYMBOL_GPL(sock_prot_inuse_get);
2748
Alexey Dobriyan2c8c1e72010-01-17 03:35:32 +00002749static int __net_init sock_inuse_init_net(struct net *net)
Pavel Emelyanov70ee1152008-03-31 19:42:16 -07002750{
2751 net->core.inuse = alloc_percpu(struct prot_inuse);
2752 return net->core.inuse ? 0 : -ENOMEM;
2753}
2754
Alexey Dobriyan2c8c1e72010-01-17 03:35:32 +00002755static void __net_exit sock_inuse_exit_net(struct net *net)
Pavel Emelyanov70ee1152008-03-31 19:42:16 -07002756{
2757 free_percpu(net->core.inuse);
2758}
2759
2760static struct pernet_operations net_inuse_ops = {
2761 .init = sock_inuse_init_net,
2762 .exit = sock_inuse_exit_net,
2763};
2764
2765static __init int net_inuse_init(void)
2766{
2767 if (register_pernet_subsys(&net_inuse_ops))
2768 panic("Cannot initialize net inuse counters");
2769
2770 return 0;
2771}
2772
2773core_initcall(net_inuse_init);
2774#else
Pavel Emelyanov1338d462008-03-28 16:38:43 -07002775static DEFINE_PER_CPU(struct prot_inuse, prot_inuse);
2776
Pavel Emelyanovc29a0bc2008-03-31 19:41:46 -07002777void sock_prot_inuse_add(struct net *net, struct proto *prot, int val)
Pavel Emelyanov1338d462008-03-28 16:38:43 -07002778{
Eric Dumazetd6d9ca02010-07-19 10:48:49 +00002779 __this_cpu_add(prot_inuse.val[prot->inuse_idx], val);
Pavel Emelyanov1338d462008-03-28 16:38:43 -07002780}
2781EXPORT_SYMBOL_GPL(sock_prot_inuse_add);
2782
Pavel Emelyanovc29a0bc2008-03-31 19:41:46 -07002783int sock_prot_inuse_get(struct net *net, struct proto *prot)
Pavel Emelyanov1338d462008-03-28 16:38:43 -07002784{
2785 int cpu, idx = prot->inuse_idx;
2786 int res = 0;
2787
2788 for_each_possible_cpu(cpu)
2789 res += per_cpu(prot_inuse, cpu).val[idx];
2790
2791 return res >= 0 ? res : 0;
2792}
2793EXPORT_SYMBOL_GPL(sock_prot_inuse_get);
Pavel Emelyanov70ee1152008-03-31 19:42:16 -07002794#endif
Pavel Emelyanov13ff3d62008-03-28 16:38:17 -07002795
2796static void assign_proto_idx(struct proto *prot)
2797{
2798 prot->inuse_idx = find_first_zero_bit(proto_inuse_idx, PROTO_INUSE_NR);
2799
2800 if (unlikely(prot->inuse_idx == PROTO_INUSE_NR - 1)) {
Joe Perchese005d192012-05-16 19:58:40 +00002801 pr_err("PROTO_INUSE_NR exhausted\n");
Pavel Emelyanov13ff3d62008-03-28 16:38:17 -07002802 return;
2803 }
2804
2805 set_bit(prot->inuse_idx, proto_inuse_idx);
2806}
2807
2808static void release_proto_idx(struct proto *prot)
2809{
2810 if (prot->inuse_idx != PROTO_INUSE_NR - 1)
2811 clear_bit(prot->inuse_idx, proto_inuse_idx);
2812}
2813#else
2814static inline void assign_proto_idx(struct proto *prot)
2815{
2816}
2817
2818static inline void release_proto_idx(struct proto *prot)
2819{
2820}
2821#endif
2822
Eric Dumazet0159dfd2015-03-12 16:44:07 -07002823static void req_prot_cleanup(struct request_sock_ops *rsk_prot)
2824{
2825 if (!rsk_prot)
2826 return;
2827 kfree(rsk_prot->slab_name);
2828 rsk_prot->slab_name = NULL;
Julia Lawalladf78ed2015-09-13 14:15:18 +02002829 kmem_cache_destroy(rsk_prot->slab);
2830 rsk_prot->slab = NULL;
Eric Dumazet0159dfd2015-03-12 16:44:07 -07002831}
2832
2833static int req_prot_init(const struct proto *prot)
2834{
2835 struct request_sock_ops *rsk_prot = prot->rsk_prot;
2836
2837 if (!rsk_prot)
2838 return 0;
2839
2840 rsk_prot->slab_name = kasprintf(GFP_KERNEL, "request_sock_%s",
2841 prot->name);
2842 if (!rsk_prot->slab_name)
2843 return -ENOMEM;
2844
2845 rsk_prot->slab = kmem_cache_create(rsk_prot->slab_name,
2846 rsk_prot->obj_size, 0,
Eric Dumazete96f78a2015-10-03 06:27:28 -07002847 prot->slab_flags, NULL);
Eric Dumazet0159dfd2015-03-12 16:44:07 -07002848
2849 if (!rsk_prot->slab) {
2850 pr_crit("%s: Can't create request sock SLAB cache!\n",
2851 prot->name);
2852 return -ENOMEM;
2853 }
2854 return 0;
2855}
2856
Linus Torvalds1da177e2005-04-16 15:20:36 -07002857int proto_register(struct proto *prot, int alloc_slab)
2858{
Linus Torvalds1da177e2005-04-16 15:20:36 -07002859 if (alloc_slab) {
2860 prot->slab = kmem_cache_create(prot->name, prot->obj_size, 0,
Eric Dumazet271b72c2008-10-29 02:11:14 -07002861 SLAB_HWCACHE_ALIGN | prot->slab_flags,
2862 NULL);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002863
2864 if (prot->slab == NULL) {
Joe Perchese005d192012-05-16 19:58:40 +00002865 pr_crit("%s: Can't create sock SLAB cache!\n",
2866 prot->name);
Pavel Emelyanov60e76632008-03-28 16:39:10 -07002867 goto out;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002868 }
Arnaldo Carvalho de Melo2e6599c2005-06-18 22:46:52 -07002869
Eric Dumazet0159dfd2015-03-12 16:44:07 -07002870 if (req_prot_init(prot))
2871 goto out_free_request_sock_slab;
Arnaldo Carvalho de Melo8feaf0c02005-08-09 20:09:30 -07002872
Arnaldo Carvalho de Melo6d6ee432005-12-13 23:25:19 -08002873 if (prot->twsk_prot != NULL) {
Alexey Dobriyanfaf23422010-02-17 09:34:12 +00002874 prot->twsk_prot->twsk_slab_name = kasprintf(GFP_KERNEL, "tw_sock_%s", prot->name);
Arnaldo Carvalho de Melo8feaf0c02005-08-09 20:09:30 -07002875
Catalin Marinas7e56b5d2008-11-21 16:45:22 -08002876 if (prot->twsk_prot->twsk_slab_name == NULL)
Arnaldo Carvalho de Melo8feaf0c02005-08-09 20:09:30 -07002877 goto out_free_request_sock_slab;
2878
Arnaldo Carvalho de Melo6d6ee432005-12-13 23:25:19 -08002879 prot->twsk_prot->twsk_slab =
Catalin Marinas7e56b5d2008-11-21 16:45:22 -08002880 kmem_cache_create(prot->twsk_prot->twsk_slab_name,
Arnaldo Carvalho de Melo6d6ee432005-12-13 23:25:19 -08002881 prot->twsk_prot->twsk_obj_size,
Eric Dumazet3ab5aee2008-11-16 19:40:17 -08002882 0,
Eric Dumazet52db70d2015-04-10 06:07:18 -07002883 prot->slab_flags,
Paul Mundt20c2df82007-07-20 10:11:58 +09002884 NULL);
Arnaldo Carvalho de Melo6d6ee432005-12-13 23:25:19 -08002885 if (prot->twsk_prot->twsk_slab == NULL)
Arnaldo Carvalho de Melo8feaf0c02005-08-09 20:09:30 -07002886 goto out_free_timewait_sock_slab_name;
2887 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07002888 }
2889
Glauber Costa36b77a52011-12-16 00:51:59 +00002890 mutex_lock(&proto_list_mutex);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002891 list_add(&prot->node, &proto_list);
Pavel Emelyanov13ff3d62008-03-28 16:38:17 -07002892 assign_proto_idx(prot);
Glauber Costa36b77a52011-12-16 00:51:59 +00002893 mutex_unlock(&proto_list_mutex);
Pavel Emelyanovb733c002007-11-07 02:23:38 -08002894 return 0;
2895
Arnaldo Carvalho de Melo8feaf0c02005-08-09 20:09:30 -07002896out_free_timewait_sock_slab_name:
Catalin Marinas7e56b5d2008-11-21 16:45:22 -08002897 kfree(prot->twsk_prot->twsk_slab_name);
Arnaldo Carvalho de Melo8feaf0c02005-08-09 20:09:30 -07002898out_free_request_sock_slab:
Eric Dumazet0159dfd2015-03-12 16:44:07 -07002899 req_prot_cleanup(prot->rsk_prot);
2900
Arnaldo Carvalho de Melo2e6599c2005-06-18 22:46:52 -07002901 kmem_cache_destroy(prot->slab);
2902 prot->slab = NULL;
Pavel Emelyanovb733c002007-11-07 02:23:38 -08002903out:
2904 return -ENOBUFS;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002905}
Linus Torvalds1da177e2005-04-16 15:20:36 -07002906EXPORT_SYMBOL(proto_register);
2907
2908void proto_unregister(struct proto *prot)
2909{
Glauber Costa36b77a52011-12-16 00:51:59 +00002910 mutex_lock(&proto_list_mutex);
Pavel Emelyanov13ff3d62008-03-28 16:38:17 -07002911 release_proto_idx(prot);
Patrick McHardy0a3f4352005-09-06 19:47:50 -07002912 list_del(&prot->node);
Glauber Costa36b77a52011-12-16 00:51:59 +00002913 mutex_unlock(&proto_list_mutex);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002914
Julia Lawalladf78ed2015-09-13 14:15:18 +02002915 kmem_cache_destroy(prot->slab);
2916 prot->slab = NULL;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002917
Eric Dumazet0159dfd2015-03-12 16:44:07 -07002918 req_prot_cleanup(prot->rsk_prot);
Arnaldo Carvalho de Melo2e6599c2005-06-18 22:46:52 -07002919
Arnaldo Carvalho de Melo6d6ee432005-12-13 23:25:19 -08002920 if (prot->twsk_prot != NULL && prot->twsk_prot->twsk_slab != NULL) {
Arnaldo Carvalho de Melo6d6ee432005-12-13 23:25:19 -08002921 kmem_cache_destroy(prot->twsk_prot->twsk_slab);
Catalin Marinas7e56b5d2008-11-21 16:45:22 -08002922 kfree(prot->twsk_prot->twsk_slab_name);
Arnaldo Carvalho de Melo6d6ee432005-12-13 23:25:19 -08002923 prot->twsk_prot->twsk_slab = NULL;
Arnaldo Carvalho de Melo8feaf0c02005-08-09 20:09:30 -07002924 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07002925}
Linus Torvalds1da177e2005-04-16 15:20:36 -07002926EXPORT_SYMBOL(proto_unregister);
2927
2928#ifdef CONFIG_PROC_FS
Linus Torvalds1da177e2005-04-16 15:20:36 -07002929static void *proto_seq_start(struct seq_file *seq, loff_t *pos)
Glauber Costa36b77a52011-12-16 00:51:59 +00002930 __acquires(proto_list_mutex)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002931{
Glauber Costa36b77a52011-12-16 00:51:59 +00002932 mutex_lock(&proto_list_mutex);
Pavel Emelianov60f04382007-07-09 13:15:14 -07002933 return seq_list_start_head(&proto_list, *pos);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002934}
2935
2936static void *proto_seq_next(struct seq_file *seq, void *v, loff_t *pos)
2937{
Pavel Emelianov60f04382007-07-09 13:15:14 -07002938 return seq_list_next(v, &proto_list, pos);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002939}
2940
2941static void proto_seq_stop(struct seq_file *seq, void *v)
Glauber Costa36b77a52011-12-16 00:51:59 +00002942 __releases(proto_list_mutex)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002943{
Glauber Costa36b77a52011-12-16 00:51:59 +00002944 mutex_unlock(&proto_list_mutex);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002945}
2946
2947static char proto_method_implemented(const void *method)
2948{
2949 return method == NULL ? 'n' : 'y';
2950}
Glauber Costa180d8cd2011-12-11 21:47:02 +00002951static long sock_prot_memory_allocated(struct proto *proto)
2952{
Jeffrin Josecb75a362012-04-25 19:17:29 +05302953 return proto->memory_allocated != NULL ? proto_memory_allocated(proto) : -1L;
Glauber Costa180d8cd2011-12-11 21:47:02 +00002954}
2955
2956static char *sock_prot_memory_pressure(struct proto *proto)
2957{
2958 return proto->memory_pressure != NULL ?
2959 proto_memory_pressure(proto) ? "yes" : "no" : "NI";
2960}
Linus Torvalds1da177e2005-04-16 15:20:36 -07002961
2962static void proto_seq_printf(struct seq_file *seq, struct proto *proto)
2963{
Glauber Costa180d8cd2011-12-11 21:47:02 +00002964
Eric Dumazet8d987e52010-11-09 23:24:26 +00002965 seq_printf(seq, "%-9s %4u %6d %6ld %-3s %6u %-3s %-10s "
Linus Torvalds1da177e2005-04-16 15:20:36 -07002966 "%2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c\n",
2967 proto->name,
2968 proto->obj_size,
Eric Dumazet14e943d2008-11-19 15:14:01 -08002969 sock_prot_inuse_get(seq_file_net(seq), proto),
Glauber Costa180d8cd2011-12-11 21:47:02 +00002970 sock_prot_memory_allocated(proto),
2971 sock_prot_memory_pressure(proto),
Linus Torvalds1da177e2005-04-16 15:20:36 -07002972 proto->max_header,
2973 proto->slab == NULL ? "no" : "yes",
2974 module_name(proto->owner),
2975 proto_method_implemented(proto->close),
2976 proto_method_implemented(proto->connect),
2977 proto_method_implemented(proto->disconnect),
2978 proto_method_implemented(proto->accept),
2979 proto_method_implemented(proto->ioctl),
2980 proto_method_implemented(proto->init),
2981 proto_method_implemented(proto->destroy),
2982 proto_method_implemented(proto->shutdown),
2983 proto_method_implemented(proto->setsockopt),
2984 proto_method_implemented(proto->getsockopt),
2985 proto_method_implemented(proto->sendmsg),
2986 proto_method_implemented(proto->recvmsg),
2987 proto_method_implemented(proto->sendpage),
2988 proto_method_implemented(proto->bind),
2989 proto_method_implemented(proto->backlog_rcv),
2990 proto_method_implemented(proto->hash),
2991 proto_method_implemented(proto->unhash),
2992 proto_method_implemented(proto->get_port),
2993 proto_method_implemented(proto->enter_memory_pressure));
2994}
2995
2996static int proto_seq_show(struct seq_file *seq, void *v)
2997{
Pavel Emelianov60f04382007-07-09 13:15:14 -07002998 if (v == &proto_list)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002999 seq_printf(seq, "%-9s %-4s %-8s %-6s %-5s %-7s %-4s %-10s %s",
3000 "protocol",
3001 "size",
3002 "sockets",
3003 "memory",
3004 "press",
3005 "maxhdr",
3006 "slab",
3007 "module",
3008 "cl co di ac io in de sh ss gs se re sp bi br ha uh gp em\n");
3009 else
Pavel Emelianov60f04382007-07-09 13:15:14 -07003010 proto_seq_printf(seq, list_entry(v, struct proto, node));
Linus Torvalds1da177e2005-04-16 15:20:36 -07003011 return 0;
3012}
3013
Stephen Hemmingerf6908082007-03-12 14:34:29 -07003014static const struct seq_operations proto_seq_ops = {
Linus Torvalds1da177e2005-04-16 15:20:36 -07003015 .start = proto_seq_start,
3016 .next = proto_seq_next,
3017 .stop = proto_seq_stop,
3018 .show = proto_seq_show,
3019};
3020
3021static int proto_seq_open(struct inode *inode, struct file *file)
3022{
Eric Dumazet14e943d2008-11-19 15:14:01 -08003023 return seq_open_net(inode, file, &proto_seq_ops,
3024 sizeof(struct seq_net_private));
Linus Torvalds1da177e2005-04-16 15:20:36 -07003025}
3026
Arjan van de Ven9a321442007-02-12 00:55:35 -08003027static const struct file_operations proto_seq_fops = {
Linus Torvalds1da177e2005-04-16 15:20:36 -07003028 .owner = THIS_MODULE,
3029 .open = proto_seq_open,
3030 .read = seq_read,
3031 .llseek = seq_lseek,
Eric Dumazet14e943d2008-11-19 15:14:01 -08003032 .release = seq_release_net,
3033};
3034
3035static __net_init int proto_init_net(struct net *net)
3036{
Gao fengd4beaa62013-02-18 01:34:54 +00003037 if (!proc_create("protocols", S_IRUGO, net->proc_net, &proto_seq_fops))
Eric Dumazet14e943d2008-11-19 15:14:01 -08003038 return -ENOMEM;
3039
3040 return 0;
3041}
3042
3043static __net_exit void proto_exit_net(struct net *net)
3044{
Gao fengece31ff2013-02-18 01:34:56 +00003045 remove_proc_entry("protocols", net->proc_net);
Eric Dumazet14e943d2008-11-19 15:14:01 -08003046}
3047
3048
3049static __net_initdata struct pernet_operations proto_net_ops = {
3050 .init = proto_init_net,
3051 .exit = proto_exit_net,
Linus Torvalds1da177e2005-04-16 15:20:36 -07003052};
3053
3054static int __init proto_init(void)
3055{
Eric Dumazet14e943d2008-11-19 15:14:01 -08003056 return register_pernet_subsys(&proto_net_ops);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003057}
3058
3059subsys_initcall(proto_init);
3060
3061#endif /* PROC_FS */