projects / linux-2.6-block.git / blame
? search:
summary | shortlog | log | commit | commitdiff | tree
blob | blame (incremental) | history | HEAD
bridge: vlan: Use rcu_dereference instead of rtnl_dereference
[linux-2.6-block.git] / net / core / sock.c
CommitLineData
1da177e4
LT		 1/*
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 *
02c30a84 10 * Authors: Ross Biro
1da177e4
LT		 11 * 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
4ec93edb 35 * code. The ACK stuff can wait and needs major
1da177e4
LT		 36 * 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
e005d193
JP		 92#define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
93
4fc268d2 94#include <linux/capability.h>
1da177e4 95#include <linux/errno.h>
cb820f8e 96#include <linux/errqueue.h>
1da177e4
LT		 97#include <linux/types.h>
98#include <linux/socket.h>
99#include <linux/in.h>
100#include <linux/kernel.h>
1da177e4
LT		 101#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>
a1f8e7f7 115#include <linux/highmem.h>
3f551f94 116#include <linux/user_namespace.h>
c5905afb 117#include <linux/static_key.h>
3969eb38 118#include <linux/memcontrol.h>
8c1ae10d 119#include <linux/prefetch.h>
1da177e4
LT		 120
121#include <asm/uaccess.h>
1da177e4
LT		 122
123#include <linux/netdevice.h>
124#include <net/protocol.h>
125#include <linux/skbuff.h>
457c4cbc 126#include <net/net_namespace.h>
2e6599cb 127#include <net/request_sock.h>
1da177e4 128#include <net/sock.h>
20d49473 129#include <linux/net_tstamp.h>
1da177e4
LT		 130#include <net/xfrm.h>
131#include <linux/ipsec.h>
f8451725 132#include <net/cls_cgroup.h>
5bc1421e 133#include <net/netprio_cgroup.h>
eb4cb008 134#include <linux/sock_diag.h>
1da177e4
LT		 135
136#include <linux/filter.h>
137
3847ce32
SM		 138#include <trace/events/sock.h>
139
1da177e4
LT		 140#ifdef CONFIG_INET
141#include <net/tcp.h>
142#endif
143
076bb0c8 144#include <net/busy_poll.h>
06021292 145
36b77a52 146static DEFINE_MUTEX(proto_list_mutex);
d1a4c0b3
GC		 147static LIST_HEAD(proto_list);
148
a3b299da
EB		 149/**
150 * sk_ns_capable - General socket capability test
151 * @sk: Socket to use a capability on or through
152 * @user_ns: The user namespace of the capability to use
153 * @cap: The capability to use
154 *
155 * Test to see if the opener of the socket had when the socket was
156 * created and the current process has the capability @cap in the user
157 * namespace @user_ns.
158 */
159bool sk_ns_capable(const struct sock *sk,
160 struct user_namespace *user_ns, int cap)
161{
162 return file_ns_capable(sk->sk_socket->file, user_ns, cap) &&
163 ns_capable(user_ns, cap);
164}
165EXPORT_SYMBOL(sk_ns_capable);
166
167/**
168 * sk_capable - Socket global capability test
169 * @sk: Socket to use a capability on or through
e793c0f7 170 * @cap: The global capability to use
a3b299da
EB		 171 *
172 * Test to see if the opener of the socket had when the socket was
173 * created and the current process has the capability @cap in all user
174 * namespaces.
175 */
176bool sk_capable(const struct sock *sk, int cap)
177{
178 return sk_ns_capable(sk, &init_user_ns, cap);
179}
180EXPORT_SYMBOL(sk_capable);
181
182/**
183 * sk_net_capable - Network namespace socket capability test
184 * @sk: Socket to use a capability on or through
185 * @cap: The capability to use
186 *
e793c0f7 187 * Test to see if the opener of the socket had when the socket was created
a3b299da
EB		 188 * and the current process has the capability @cap over the network namespace
189 * the socket is a member of.
190 */
191bool sk_net_capable(const struct sock *sk, int cap)
192{
193 return sk_ns_capable(sk, sock_net(sk)->user_ns, cap);
194}
195EXPORT_SYMBOL(sk_net_capable);
196
197
c255a458 198#ifdef CONFIG_MEMCG_KMEM
1d62e436 199int mem_cgroup_sockets_init(struct mem_cgroup *memcg, struct cgroup_subsys *ss)
d1a4c0b3
GC		 200{
201 struct proto *proto;
202 int ret = 0;
203
36b77a52 204 mutex_lock(&proto_list_mutex);
d1a4c0b3
GC		 205 list_for_each_entry(proto, &proto_list, node) {
206 if (proto->init_cgroup) {
1d62e436 207 ret = proto->init_cgroup(memcg, ss);
d1a4c0b3
GC		 208 if (ret)
209 goto out;
210 }
211 }
212
36b77a52 213 mutex_unlock(&proto_list_mutex);
d1a4c0b3
GC		 214 return ret;
215out:
216 list_for_each_entry_continue_reverse(proto, &proto_list, node)
217 if (proto->destroy_cgroup)
1d62e436 218 proto->destroy_cgroup(memcg);
36b77a52 219 mutex_unlock(&proto_list_mutex);
d1a4c0b3
GC		 220 return ret;
221}
222
1d62e436 223void mem_cgroup_sockets_destroy(struct mem_cgroup *memcg)
d1a4c0b3
GC		 224{
225 struct proto *proto;
226
36b77a52 227 mutex_lock(&proto_list_mutex);
d1a4c0b3
GC		 228 list_for_each_entry_reverse(proto, &proto_list, node)
229 if (proto->destroy_cgroup)
1d62e436 230 proto->destroy_cgroup(memcg);
36b77a52 231 mutex_unlock(&proto_list_mutex);
d1a4c0b3
GC		 232}
233#endif
234
da21f24d
IM		 235/*
236 * Each address family might have different locking rules, so we have
237 * one slock key per address family:
238 */
a5b5bb9a
IM		 239static struct lock_class_key af_family_keys[AF_MAX];
240static struct lock_class_key af_family_slock_keys[AF_MAX];
241
cbda4eaf 242#if defined(CONFIG_MEMCG_KMEM)
c5905afb 243struct static_key memcg_socket_limit_enabled;
e1aab161 244EXPORT_SYMBOL(memcg_socket_limit_enabled);
cbda4eaf 245#endif
e1aab161 246
a5b5bb9a
IM		 247/*
248 * Make lock validator output more readable. (we pre-construct these
249 * strings build-time, so that runtime initialization of socket
250 * locks is fast):
251 */
36cbd3dc 252static const char *const af_family_key_strings[AF_MAX+1] = {
a5b5bb9a
IM		 253 "sk_lock-AF_UNSPEC", "sk_lock-AF_UNIX" , "sk_lock-AF_INET" ,
254 "sk_lock-AF_AX25" , "sk_lock-AF_IPX" , "sk_lock-AF_APPLETALK",
255 "sk_lock-AF_NETROM", "sk_lock-AF_BRIDGE" , "sk_lock-AF_ATMPVC" ,
256 "sk_lock-AF_X25" , "sk_lock-AF_INET6" , "sk_lock-AF_ROSE" ,
257 "sk_lock-AF_DECnet", "sk_lock-AF_NETBEUI" , "sk_lock-AF_SECURITY" ,
258 "sk_lock-AF_KEY" , "sk_lock-AF_NETLINK" , "sk_lock-AF_PACKET" ,
259 "sk_lock-AF_ASH" , "sk_lock-AF_ECONET" , "sk_lock-AF_ATMSVC" ,
cbd151bf 260 "sk_lock-AF_RDS" , "sk_lock-AF_SNA" , "sk_lock-AF_IRDA" ,
a5b5bb9a 261 "sk_lock-AF_PPPOX" , "sk_lock-AF_WANPIPE" , "sk_lock-AF_LLC" ,
cd05acfe 262 "sk_lock-27" , "sk_lock-28" , "sk_lock-AF_CAN" ,
17926a79 263 "sk_lock-AF_TIPC" , "sk_lock-AF_BLUETOOTH", "sk_lock-IUCV" ,
bce7b154 264 "sk_lock-AF_RXRPC" , "sk_lock-AF_ISDN" , "sk_lock-AF_PHONET" ,
6f107b58 265 "sk_lock-AF_IEEE802154", "sk_lock-AF_CAIF" , "sk_lock-AF_ALG" ,
456db6a4 266 "sk_lock-AF_NFC" , "sk_lock-AF_VSOCK" , "sk_lock-AF_MAX"
a5b5bb9a 267};
36cbd3dc 268static const char *const af_family_slock_key_strings[AF_MAX+1] = {
a5b5bb9a
IM		 269 "slock-AF_UNSPEC", "slock-AF_UNIX" , "slock-AF_INET" ,
270 "slock-AF_AX25" , "slock-AF_IPX" , "slock-AF_APPLETALK",
271 "slock-AF_NETROM", "slock-AF_BRIDGE" , "slock-AF_ATMPVC" ,
272 "slock-AF_X25" , "slock-AF_INET6" , "slock-AF_ROSE" ,
273 "slock-AF_DECnet", "slock-AF_NETBEUI" , "slock-AF_SECURITY" ,
274 "slock-AF_KEY" , "slock-AF_NETLINK" , "slock-AF_PACKET" ,
275 "slock-AF_ASH" , "slock-AF_ECONET" , "slock-AF_ATMSVC" ,
cbd151bf 276 "slock-AF_RDS" , "slock-AF_SNA" , "slock-AF_IRDA" ,
a5b5bb9a 277 "slock-AF_PPPOX" , "slock-AF_WANPIPE" , "slock-AF_LLC" ,
cd05acfe 278 "slock-27" , "slock-28" , "slock-AF_CAN" ,
17926a79 279 "slock-AF_TIPC" , "slock-AF_BLUETOOTH", "slock-AF_IUCV" ,
bce7b154 280 "slock-AF_RXRPC" , "slock-AF_ISDN" , "slock-AF_PHONET" ,
6f107b58 281 "slock-AF_IEEE802154", "slock-AF_CAIF" , "slock-AF_ALG" ,
456db6a4 282 "slock-AF_NFC" , "slock-AF_VSOCK" ,"slock-AF_MAX"
a5b5bb9a 283};
36cbd3dc 284static const char *const af_family_clock_key_strings[AF_MAX+1] = {
443aef0e
PZ		 285 "clock-AF_UNSPEC", "clock-AF_UNIX" , "clock-AF_INET" ,
286 "clock-AF_AX25" , "clock-AF_IPX" , "clock-AF_APPLETALK",
287 "clock-AF_NETROM", "clock-AF_BRIDGE" , "clock-AF_ATMPVC" ,
288 "clock-AF_X25" , "clock-AF_INET6" , "clock-AF_ROSE" ,
289 "clock-AF_DECnet", "clock-AF_NETBEUI" , "clock-AF_SECURITY" ,
290 "clock-AF_KEY" , "clock-AF_NETLINK" , "clock-AF_PACKET" ,
291 "clock-AF_ASH" , "clock-AF_ECONET" , "clock-AF_ATMSVC" ,
cbd151bf 292 "clock-AF_RDS" , "clock-AF_SNA" , "clock-AF_IRDA" ,
443aef0e 293 "clock-AF_PPPOX" , "clock-AF_WANPIPE" , "clock-AF_LLC" ,
b4942af6 294 "clock-27" , "clock-28" , "clock-AF_CAN" ,
e51f802b 295 "clock-AF_TIPC" , "clock-AF_BLUETOOTH", "clock-AF_IUCV" ,
bce7b154 296 "clock-AF_RXRPC" , "clock-AF_ISDN" , "clock-AF_PHONET" ,
6f107b58 297 "clock-AF_IEEE802154", "clock-AF_CAIF" , "clock-AF_ALG" ,
456db6a4 298 "clock-AF_NFC" , "clock-AF_VSOCK" , "clock-AF_MAX"
443aef0e 299};
da21f24d
IM		 300
301/*
302 * sk_callback_lock locking rules are per-address-family,
303 * so split the lock classes by using a per-AF key:
304 */
305static struct lock_class_key af_callback_keys[AF_MAX];
306
1da177e4
LT		 307/* Take into consideration the size of the struct sk_buff overhead in the
308 * determination of these values, since that is non-constant across
309 * platforms. This makes socket queueing behavior and performance
310 * not depend upon such differences.
311 */
312#define _SK_MEM_PACKETS 256
87fb4b7b 313#define _SK_MEM_OVERHEAD SKB_TRUESIZE(256)
1da177e4
LT		 314#define SK_WMEM_MAX (_SK_MEM_OVERHEAD * _SK_MEM_PACKETS)
315#define SK_RMEM_MAX (_SK_MEM_OVERHEAD * _SK_MEM_PACKETS)
316
317/* Run time adjustable parameters. */
ab32ea5d 318__u32 sysctl_wmem_max __read_mostly = SK_WMEM_MAX;
6d8ebc8a 319EXPORT_SYMBOL(sysctl_wmem_max);
ab32ea5d 320__u32 sysctl_rmem_max __read_mostly = SK_RMEM_MAX;
6d8ebc8a 321EXPORT_SYMBOL(sysctl_rmem_max);
ab32ea5d
BH		 322__u32 sysctl_wmem_default __read_mostly = SK_WMEM_MAX;
323__u32 sysctl_rmem_default __read_mostly = SK_RMEM_MAX;
1da177e4 324
25985edc 325/* Maximal space eaten by iovec or ancillary data plus some space */
ab32ea5d 326int sysctl_optmem_max __read_mostly = sizeof(unsigned long)*(2*UIO_MAXIOV+512);
2a91525c 327EXPORT_SYMBOL(sysctl_optmem_max);
1da177e4 328
b245be1f
WB		 329int sysctl_tstamp_allow_data __read_mostly = 1;
330
c93bdd0e
MG		 331struct static_key memalloc_socks = STATIC_KEY_INIT_FALSE;
332EXPORT_SYMBOL_GPL(memalloc_socks);
333
7cb02404
MG		 334/**
335 * sk_set_memalloc - sets %SOCK_MEMALLOC
336 * @sk: socket to set it on
337 *
338 * Set %SOCK_MEMALLOC on a socket for access to emergency reserves.
339 * It's the responsibility of the admin to adjust min_free_kbytes
340 * to meet the requirements
341 */
342void sk_set_memalloc(struct sock *sk)
343{
344 sock_set_flag(sk, SOCK_MEMALLOC);
345 sk->sk_allocation |= __GFP_MEMALLOC;
c93bdd0e 346 static_key_slow_inc(&memalloc_socks);
7cb02404
MG		 347}
348EXPORT_SYMBOL_GPL(sk_set_memalloc);
349
350void sk_clear_memalloc(struct sock *sk)
351{
352 sock_reset_flag(sk, SOCK_MEMALLOC);
353 sk->sk_allocation &= ~__GFP_MEMALLOC;
c93bdd0e 354 static_key_slow_dec(&memalloc_socks);
c76562b6
MG		 355
356 /*
357 * SOCK_MEMALLOC is allowed to ignore rmem limits to ensure forward
5d753610
MG		 358 * progress of swapping. SOCK_MEMALLOC may be cleared while
359 * it has rmem allocations due to the last swapfile being deactivated
360 * but there is a risk that the socket is unusable due to exceeding
361 * the rmem limits. Reclaim the reserves and obey rmem limits again.
c76562b6 362 */
5d753610 363 sk_mem_reclaim(sk);
7cb02404
MG		 364}
365EXPORT_SYMBOL_GPL(sk_clear_memalloc);
366
b4b9e355
MG		 367int __sk_backlog_rcv(struct sock *sk, struct sk_buff *skb)
368{
369 int ret;
370 unsigned long pflags = current->flags;
371
372 /* these should have been dropped before queueing */
373 BUG_ON(!sock_flag(sk, SOCK_MEMALLOC));
374
375 current->flags |= PF_MEMALLOC;
376 ret = sk->sk_backlog_rcv(sk, skb);
377 tsk_restore_flags(current, pflags, PF_MEMALLOC);
378
379 return ret;
380}
381EXPORT_SYMBOL(__sk_backlog_rcv);
382
1da177e4
LT		 383static int sock_set_timeout(long *timeo_p, char __user *optval, int optlen)
384{
385 struct timeval tv;
386
387 if (optlen < sizeof(tv))
388 return -EINVAL;
389 if (copy_from_user(&tv, optval, sizeof(tv)))
390 return -EFAULT;
ba78073e
VA		 391 if (tv.tv_usec < 0 || tv.tv_usec >= USEC_PER_SEC)
392 return -EDOM;
1da177e4 393
ba78073e 394 if (tv.tv_sec < 0) {
6f11df83
AM		 395 static int warned __read_mostly;
396
ba78073e 397 *timeo_p = 0;
50aab54f 398 if (warned < 10 && net_ratelimit()) {
ba78073e 399 warned++;
e005d193
JP		 400 pr_info("%s: `%s' (pid %d) tries to set negative timeout\n",
401 __func__, current->comm, task_pid_nr(current));
50aab54f 402 }
ba78073e
VA		 403 return 0;
404 }
1da177e4
LT		 405 *timeo_p = MAX_SCHEDULE_TIMEOUT;
406 if (tv.tv_sec == 0 && tv.tv_usec == 0)
407 return 0;
408 if (tv.tv_sec < (MAX_SCHEDULE_TIMEOUT/HZ - 1))
409 *timeo_p = tv.tv_sec*HZ + (tv.tv_usec+(1000000/HZ-1))/(1000000/HZ);
410 return 0;
411}
412
413static void sock_warn_obsolete_bsdism(const char *name)
414{
415 static int warned;
416 static char warncomm[TASK_COMM_LEN];
4ec93edb
YH		 417 if (strcmp(warncomm, current->comm) && warned < 5) {
418 strcpy(warncomm, current->comm);
e005d193
JP		 419 pr_warn("process `%s' is using obsolete %s SO_BSDCOMPAT\n",
420 warncomm, name);
1da177e4
LT		 421 warned++;
422 }
423}
424
080a270f
HFS		 425static bool sock_needs_netstamp(const struct sock *sk)
426{
427 switch (sk->sk_family) {
428 case AF_UNSPEC:
429 case AF_UNIX:
430 return false;
431 default:
432 return true;
433 }
434}
435
08e29af3
ED		 436#define SK_FLAGS_TIMESTAMP ((1UL << SOCK_TIMESTAMP) | (1UL << SOCK_TIMESTAMPING_RX_SOFTWARE))
437
438static void sock_disable_timestamp(struct sock *sk, unsigned long flags)
4ec93edb 439{
08e29af3
ED		 440 if (sk->sk_flags & flags) {
441 sk->sk_flags &= ~flags;
080a270f
HFS		 442 if (sock_needs_netstamp(sk) &&
443 !(sk->sk_flags & SK_FLAGS_TIMESTAMP))
20d49473 444 net_disable_timestamp();
1da177e4
LT		 445 }
446}
447
448
f0088a50
DV		 449int sock_queue_rcv_skb(struct sock *sk, struct sk_buff *skb)
450{
766e9037 451 int err;
3b885787
NH		 452 unsigned long flags;
453 struct sk_buff_head *list = &sk->sk_receive_queue;
f0088a50 454
0fd7bac6 455 if (atomic_read(&sk->sk_rmem_alloc) >= sk->sk_rcvbuf) {
766e9037 456 atomic_inc(&sk->sk_drops);
3847ce32 457 trace_sock_rcvqueue_full(sk, skb);
766e9037 458 return -ENOMEM;
f0088a50
DV		 459 }
460
fda9ef5d 461 err = sk_filter(sk, skb);
f0088a50 462 if (err)
766e9037 463 return err;
f0088a50 464
c76562b6 465 if (!sk_rmem_schedule(sk, skb, skb->truesize)) {
766e9037
ED		 466 atomic_inc(&sk->sk_drops);
467 return -ENOBUFS;
3ab224be
HA		 468 }
469
f0088a50
DV		 470 skb->dev = NULL;
471 skb_set_owner_r(skb, sk);
49ad9599 472
7fee226a
ED		 473 /* we escape from rcu protected region, make sure we dont leak
474 * a norefcounted dst
475 */
476 skb_dst_force(skb);
477
3b885787 478 spin_lock_irqsave(&list->lock, flags);
3bc3b96f 479 sock_skb_set_dropcount(sk, skb);
3b885787
NH		 480 __skb_queue_tail(list, skb);
481 spin_unlock_irqrestore(&list->lock, flags);
f0088a50
DV		 482
483 if (!sock_flag(sk, SOCK_DEAD))
676d2369 484 sk->sk_data_ready(sk);
766e9037 485 return 0;
f0088a50
DV		 486}
487EXPORT_SYMBOL(sock_queue_rcv_skb);
488
58a5a7b9 489int sk_receive_skb(struct sock *sk, struct sk_buff *skb, const int nested)
f0088a50
DV		 490{
491 int rc = NET_RX_SUCCESS;
492
fda9ef5d 493 if (sk_filter(sk, skb))
f0088a50
DV		 494 goto discard_and_relse;
495
496 skb->dev = NULL;
497
274f482d 498 if (sk_rcvqueues_full(sk, sk->sk_rcvbuf)) {
c377411f
ED		 499 atomic_inc(&sk->sk_drops);
500 goto discard_and_relse;
501 }
58a5a7b9
ACM		 502 if (nested)
503 bh_lock_sock_nested(sk);
504 else
505 bh_lock_sock(sk);
a5b5bb9a
IM		 506 if (!sock_owned_by_user(sk)) {
507 /*
508 * trylock + unlock semantics:
509 */
510 mutex_acquire(&sk->sk_lock.dep_map, 0, 1, _RET_IP_);
511
c57943a1 512 rc = sk_backlog_rcv(sk, skb);
a5b5bb9a
IM		 513
514 mutex_release(&sk->sk_lock.dep_map, 1, _RET_IP_);
f545a38f 515 } else if (sk_add_backlog(sk, skb, sk->sk_rcvbuf)) {
8eae939f
ZY		 516 bh_unlock_sock(sk);
517 atomic_inc(&sk->sk_drops);
518 goto discard_and_relse;
519 }
520
f0088a50
DV		 521 bh_unlock_sock(sk);
522out:
523 sock_put(sk);
524 return rc;
525discard_and_relse:
526 kfree_skb(skb);
527 goto out;
528}
529EXPORT_SYMBOL(sk_receive_skb);
530
531struct dst_entry *__sk_dst_check(struct sock *sk, u32 cookie)
532{
b6c6712a 533 struct dst_entry *dst = __sk_dst_get(sk);
f0088a50
DV		 534
535 if (dst && dst->obsolete && dst->ops->check(dst, cookie) == NULL) {
e022f0b4 536 sk_tx_queue_clear(sk);
a9b3cd7f 537 RCU_INIT_POINTER(sk->sk_dst_cache, NULL);
f0088a50
DV		 538 dst_release(dst);
539 return NULL;
540 }
541
542 return dst;
543}
544EXPORT_SYMBOL(__sk_dst_check);
545
546struct dst_entry *sk_dst_check(struct sock *sk, u32 cookie)
547{
548 struct dst_entry *dst = sk_dst_get(sk);
549
550 if (dst && dst->obsolete && dst->ops->check(dst, cookie) == NULL) {
551 sk_dst_reset(sk);
552 dst_release(dst);
553 return NULL;
554 }
555
556 return dst;
557}
558EXPORT_SYMBOL(sk_dst_check);
559
c91f6df2
BH		 560static int sock_setbindtodevice(struct sock *sk, char __user *optval,
561 int optlen)
4878809f
DM		 562{
563 int ret = -ENOPROTOOPT;
564#ifdef CONFIG_NETDEVICES
3b1e0a65 565 struct net *net = sock_net(sk);
4878809f
DM		 566 char devname[IFNAMSIZ];
567 int index;
568
569 /* Sorry... */
570 ret = -EPERM;
5e1fccc0 571 if (!ns_capable(net->user_ns, CAP_NET_RAW))
4878809f
DM		 572 goto out;
573
574 ret = -EINVAL;
575 if (optlen < 0)
576 goto out;
577
578 /* Bind this socket to a particular device like "eth0",
579 * as specified in the passed interface name. If the
580 * name is "" or the option length is zero the socket
581 * is not bound.
582 */
583 if (optlen > IFNAMSIZ - 1)
584 optlen = IFNAMSIZ - 1;
585 memset(devname, 0, sizeof(devname));
586
587 ret = -EFAULT;
588 if (copy_from_user(devname, optval, optlen))
589 goto out;
590
000ba2e4
DM		 591 index = 0;
592 if (devname[0] != '\0') {
bf8e56bf 593 struct net_device *dev;
4878809f 594
bf8e56bf
ED		 595 rcu_read_lock();
596 dev = dev_get_by_name_rcu(net, devname);
597 if (dev)
598 index = dev->ifindex;
599 rcu_read_unlock();
4878809f
DM		 600 ret = -ENODEV;
601 if (!dev)
602 goto out;
4878809f
DM		 603 }
604
605 lock_sock(sk);
606 sk->sk_bound_dev_if = index;
607 sk_dst_reset(sk);
608 release_sock(sk);
609
610 ret = 0;
611
612out:
613#endif
614
615 return ret;
616}
617
c91f6df2
BH		 618static int sock_getbindtodevice(struct sock *sk, char __user *optval,
619 int __user *optlen, int len)
620{
621 int ret = -ENOPROTOOPT;
622#ifdef CONFIG_NETDEVICES
623 struct net *net = sock_net(sk);
c91f6df2 624 char devname[IFNAMSIZ];
c91f6df2
BH		 625
626 if (sk->sk_bound_dev_if == 0) {
627 len = 0;
628 goto zero;
629 }
630
631 ret = -EINVAL;
632 if (len < IFNAMSIZ)
633 goto out;
634
5dbe7c17
NS		 635 ret = netdev_get_name(net, devname, sk->sk_bound_dev_if);
636 if (ret)
c91f6df2 637 goto out;
c91f6df2
BH		 638
639 len = strlen(devname) + 1;
640
641 ret = -EFAULT;
642 if (copy_to_user(optval, devname, len))
643 goto out;
644
645zero:
646 ret = -EFAULT;
647 if (put_user(len, optlen))
648 goto out;
649
650 ret = 0;
651
652out:
653#endif
654
655 return ret;
656}
657
c0ef877b
PE		 658static inline void sock_valbool_flag(struct sock *sk, int bit, int valbool)
659{
660 if (valbool)
661 sock_set_flag(sk, bit);
662 else
663 sock_reset_flag(sk, bit);
664}
665
f60e5990 666bool sk_mc_loop(struct sock *sk)
667{
668 if (dev_recursion_level())
669 return false;
670 if (!sk)
671 return true;
672 switch (sk->sk_family) {
673 case AF_INET:
674 return inet_sk(sk)->mc_loop;
675#if IS_ENABLED(CONFIG_IPV6)
676 case AF_INET6:
677 return inet6_sk(sk)->mc_loop;
678#endif
679 }
680 WARN_ON(1);
681 return true;
682}
683EXPORT_SYMBOL(sk_mc_loop);
684
1da177e4
LT		 685/*
686 * This is meant for all protocols to use and covers goings on
687 * at the socket level. Everything here is generic.
688 */
689
690int sock_setsockopt(struct socket *sock, int level, int optname,
b7058842 691 char __user *optval, unsigned int optlen)
1da177e4 692{
2a91525c 693 struct sock *sk = sock->sk;
1da177e4
LT		 694 int val;
695 int valbool;
696 struct linger ling;
697 int ret = 0;
4ec93edb 698
1da177e4
LT		 699 /*
700 * Options without arguments
701 */
702
4878809f 703 if (optname == SO_BINDTODEVICE)
c91f6df2 704 return sock_setbindtodevice(sk, optval, optlen);
4878809f 705
e71a4783
SH		 706 if (optlen < sizeof(int))
707 return -EINVAL;
4ec93edb 708
1da177e4
LT		 709 if (get_user(val, (int __user *)optval))
710 return -EFAULT;
4ec93edb 711
2a91525c 712 valbool = val ? 1 : 0;
1da177e4
LT		 713
714 lock_sock(sk);
715
2a91525c 716 switch (optname) {
e71a4783 717 case SO_DEBUG:
2a91525c 718 if (val && !capable(CAP_NET_ADMIN))
e71a4783 719 ret = -EACCES;
2a91525c 720 else
c0ef877b 721 sock_valbool_flag(sk, SOCK_DBG, valbool);
e71a4783
SH		 722 break;
723 case SO_REUSEADDR:
4a17fd52 724 sk->sk_reuse = (valbool ? SK_CAN_REUSE : SK_NO_REUSE);
e71a4783 725 break;
055dc21a
TH		 726 case SO_REUSEPORT:
727 sk->sk_reuseport = valbool;
728 break;
e71a4783 729 case SO_TYPE:
49c794e9 730 case SO_PROTOCOL:
0d6038ee 731 case SO_DOMAIN:
e71a4783
SH		 732 case SO_ERROR:
733 ret = -ENOPROTOOPT;
734 break;
735 case SO_DONTROUTE:
c0ef877b 736 sock_valbool_flag(sk, SOCK_LOCALROUTE, valbool);
e71a4783
SH		 737 break;
738 case SO_BROADCAST:
739 sock_valbool_flag(sk, SOCK_BROADCAST, valbool);
740 break;
741 case SO_SNDBUF:
742 /* Don't error on this BSD doesn't and if you think
82981930
ED		 743 * about it this is right. Otherwise apps have to
744 * play 'guess the biggest size' games. RCVBUF/SNDBUF
745 * are treated in BSD as hints
746 */
747 val = min_t(u32, val, sysctl_wmem_max);
b0573dea 748set_sndbuf:
e71a4783 749 sk->sk_userlocks |= SOCK_SNDBUF_LOCK;
82981930
ED		 750 sk->sk_sndbuf = max_t(u32, val * 2, SOCK_MIN_SNDBUF);
751 /* Wake up sending tasks if we upped the value. */
e71a4783
SH		 752 sk->sk_write_space(sk);
753 break;
1da177e4 754
e71a4783
SH		 755 case SO_SNDBUFFORCE:
756 if (!capable(CAP_NET_ADMIN)) {
757 ret = -EPERM;
758 break;
759 }
760 goto set_sndbuf;
b0573dea 761
e71a4783
SH		 762 case SO_RCVBUF:
763 /* Don't error on this BSD doesn't and if you think
82981930
ED		 764 * about it this is right. Otherwise apps have to
765 * play 'guess the biggest size' games. RCVBUF/SNDBUF
766 * are treated in BSD as hints
767 */
768 val = min_t(u32, val, sysctl_rmem_max);
b0573dea 769set_rcvbuf:
e71a4783
SH		 770 sk->sk_userlocks |= SOCK_RCVBUF_LOCK;
771 /*
772 * We double it on the way in to account for
773 * "struct sk_buff" etc. overhead. Applications
774 * assume that the SO_RCVBUF setting they make will
775 * allow that much actual data to be received on that
776 * socket.
777 *
778 * Applications are unaware that "struct sk_buff" and
779 * other overheads allocate from the receive buffer
780 * during socket buffer allocation.
781 *
782 * And after considering the possible alternatives,
783 * returning the value we actually used in getsockopt
784 * is the most desirable behavior.
785 */
82981930 786 sk->sk_rcvbuf = max_t(u32, val * 2, SOCK_MIN_RCVBUF);
e71a4783
SH		 787 break;
788
789 case SO_RCVBUFFORCE:
790 if (!capable(CAP_NET_ADMIN)) {
791 ret = -EPERM;
1da177e4 792 break;
e71a4783
SH		 793 }
794 goto set_rcvbuf;
1da177e4 795
e71a4783 796 case SO_KEEPALIVE:
1da177e4 797#ifdef CONFIG_INET
3e10986d
ED		 798 if (sk->sk_protocol == IPPROTO_TCP &&
799 sk->sk_type == SOCK_STREAM)
e71a4783 800 tcp_set_keepalive(sk, valbool);
1da177e4 801#endif
e71a4783
SH		 802 sock_valbool_flag(sk, SOCK_KEEPOPEN, valbool);
803 break;
804
805 case SO_OOBINLINE:
806 sock_valbool_flag(sk, SOCK_URGINLINE, valbool);
807 break;
808
809 case SO_NO_CHECK:
28448b80 810 sk->sk_no_check_tx = valbool;
e71a4783
SH		 811 break;
812
813 case SO_PRIORITY:
5e1fccc0
EB		 814 if ((val >= 0 && val <= 6) ||
815 ns_capable(sock_net(sk)->user_ns, CAP_NET_ADMIN))
e71a4783
SH		 816 sk->sk_priority = val;
817 else
818 ret = -EPERM;
819 break;
820
821 case SO_LINGER:
822 if (optlen < sizeof(ling)) {
823 ret = -EINVAL; /* 1003.1g */
1da177e4 824 break;
e71a4783 825 }
2a91525c 826 if (copy_from_user(&ling, optval, sizeof(ling))) {
e71a4783 827 ret = -EFAULT;
1da177e4 828 break;
e71a4783
SH		 829 }
830 if (!ling.l_onoff)
831 sock_reset_flag(sk, SOCK_LINGER);
832 else {
1da177e4 833#if (BITS_PER_LONG == 32)
e71a4783
SH		 834 if ((unsigned int)ling.l_linger >= MAX_SCHEDULE_TIMEOUT/HZ)
835 sk->sk_lingertime = MAX_SCHEDULE_TIMEOUT;
1da177e4 836 else
e71a4783
SH		 837#endif
838 sk->sk_lingertime = (unsigned int)ling.l_linger * HZ;
839 sock_set_flag(sk, SOCK_LINGER);
840 }
841 break;
842
843 case SO_BSDCOMPAT:
844 sock_warn_obsolete_bsdism("setsockopt");
845 break;
846
847 case SO_PASSCRED:
848 if (valbool)
849 set_bit(SOCK_PASSCRED, &sock->flags);
850 else
851 clear_bit(SOCK_PASSCRED, &sock->flags);
852 break;
853
854 case SO_TIMESTAMP:
92f37fd2 855 case SO_TIMESTAMPNS:
e71a4783 856 if (valbool) {
92f37fd2
ED		 857 if (optname == SO_TIMESTAMP)
858 sock_reset_flag(sk, SOCK_RCVTSTAMPNS);
859 else
860 sock_set_flag(sk, SOCK_RCVTSTAMPNS);
e71a4783 861 sock_set_flag(sk, SOCK_RCVTSTAMP);
20d49473 862 sock_enable_timestamp(sk, SOCK_TIMESTAMP);
92f37fd2 863 } else {
e71a4783 864 sock_reset_flag(sk, SOCK_RCVTSTAMP);
92f37fd2
ED		 865 sock_reset_flag(sk, SOCK_RCVTSTAMPNS);
866 }
e71a4783
SH		 867 break;
868
20d49473
PO		 869 case SO_TIMESTAMPING:
870 if (val & ~SOF_TIMESTAMPING_MASK) {
f249fb78 871 ret = -EINVAL;
20d49473
PO		 872 break;
873 }
b245be1f 874
09c2d251 875 if (val & SOF_TIMESTAMPING_OPT_ID &&
4ed2d765
WB		 876 !(sk->sk_tsflags & SOF_TIMESTAMPING_OPT_ID)) {
877 if (sk->sk_protocol == IPPROTO_TCP) {
878 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 }
b9f40e21 887 sk->sk_tsflags = val;
20d49473
PO		 888 if (val & SOF_TIMESTAMPING_RX_SOFTWARE)
889 sock_enable_timestamp(sk,
890 SOCK_TIMESTAMPING_RX_SOFTWARE);
891 else
892 sock_disable_timestamp(sk,
08e29af3 893 (1UL << SOCK_TIMESTAMPING_RX_SOFTWARE));
20d49473
PO		 894 break;
895
e71a4783
SH		 896 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;
1da177e4 909
e71a4783
SH		 910 case SO_ATTACH_FILTER:
911 ret = -EINVAL;
912 if (optlen == sizeof(struct sock_fprog)) {
913 struct sock_fprog fprog;
1da177e4 914
e71a4783
SH		 915 ret = -EFAULT;
916 if (copy_from_user(&fprog, optval, sizeof(fprog)))
1da177e4 917 break;
e71a4783
SH		 918
919 ret = sk_attach_filter(&fprog, sk);
920 }
921 break;
922
89aa0758
AS		 923 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
e71a4783 936 case SO_DETACH_FILTER:
55b33325 937 ret = sk_detach_filter(sk);
e71a4783 938 break;
1da177e4 939
d59577b6
VB		 940 case SO_LOCK_FILTER:
941 if (sock_flag(sk, SOCK_FILTER_LOCKED) && !valbool)
942 ret = -EPERM;
943 else
944 sock_valbool_flag(sk, SOCK_FILTER_LOCKED, valbool);
945 break;
946
e71a4783
SH		 947 case SO_PASSSEC:
948 if (valbool)
949 set_bit(SOCK_PASSSEC, &sock->flags);
950 else
951 clear_bit(SOCK_PASSSEC, &sock->flags);
952 break;
4a19ec58 953 case SO_MARK:
5e1fccc0 954 if (!ns_capable(sock_net(sk)->user_ns, CAP_NET_ADMIN))
4a19ec58 955 ret = -EPERM;
2a91525c 956 else
4a19ec58 957 sk->sk_mark = val;
4a19ec58 958 break;
877ce7c1 959
3b885787 960 case SO_RXQ_OVFL:
8083f0fc 961 sock_valbool_flag(sk, SOCK_RXQ_OVFL, valbool);
3b885787 962 break;
6e3e939f
JB		 963
964 case SO_WIFI_STATUS:
965 sock_valbool_flag(sk, SOCK_WIFI_STATUS, valbool);
966 break;
967
ef64a54f
PE		 968 case SO_PEEK_OFF:
969 if (sock->ops->set_peek_off)
12663bfc 970 ret = sock->ops->set_peek_off(sk, val);
ef64a54f
PE		 971 else
972 ret = -EOPNOTSUPP;
973 break;
3bdc0eba
BG		 974
975 case SO_NOFCS:
976 sock_valbool_flag(sk, SOCK_NOFCS, valbool);
977 break;
978
7d4c04fc
KJ		 979 case SO_SELECT_ERR_QUEUE:
980 sock_valbool_flag(sk, SOCK_SELECT_ERR_QUEUE, valbool);
981 break;
982
e0d1095a 983#ifdef CONFIG_NET_RX_BUSY_POLL
64b0dc51 984 case SO_BUSY_POLL:
dafcc438
ET		 985 /* allow unprivileged users to decrease the value */
986 if ((val > sk->sk_ll_usec) && !capable(CAP_NET_ADMIN))
987 ret = -EPERM;
988 else {
989 if (val < 0)
990 ret = -EINVAL;
991 else
992 sk->sk_ll_usec = val;
993 }
994 break;
995#endif
62748f32
ED		 996
997 case SO_MAX_PACING_RATE:
998 sk->sk_max_pacing_rate = val;
999 sk->sk_pacing_rate = min(sk->sk_pacing_rate,
1000 sk->sk_max_pacing_rate);
1001 break;
1002
70da268b
ED		 1003 case SO_INCOMING_CPU:
1004 sk->sk_incoming_cpu = val;
1005 break;
1006
e71a4783
SH		 1007 default:
1008 ret = -ENOPROTOOPT;
1009 break;
4ec93edb 1010 }
1da177e4
LT		 1011 release_sock(sk);
1012 return ret;
1013}
2a91525c 1014EXPORT_SYMBOL(sock_setsockopt);
1da177e4
LT		 1015
1016
8f09898b 1017static void cred_to_ucred(struct pid *pid, const struct cred *cred,
1018 struct ucred *ucred)
3f551f94
EB		 1019{
1020 ucred->pid = pid_vnr(pid);
1021 ucred->uid = ucred->gid = -1;
1022 if (cred) {
1023 struct user_namespace *current_ns = current_user_ns();
1024
b2e4f544
EB		 1025 ucred->uid = from_kuid_munged(current_ns, cred->euid);
1026 ucred->gid = from_kgid_munged(current_ns, cred->egid);
3f551f94
EB		 1027 }
1028}
1029
1da177e4
LT		 1030int sock_getsockopt(struct socket *sock, int level, int optname,
1031 char __user *optval, int __user *optlen)
1032{
1033 struct sock *sk = sock->sk;
4ec93edb 1034
e71a4783 1035 union {
4ec93edb
YH		 1036 int val;
1037 struct linger ling;
1da177e4
LT		 1038 struct timeval tm;
1039 } v;
4ec93edb 1040
4d0392be 1041 int lv = sizeof(int);
1da177e4 1042 int len;
4ec93edb 1043
e71a4783 1044 if (get_user(len, optlen))
4ec93edb 1045 return -EFAULT;
e71a4783 1046 if (len < 0)
1da177e4 1047 return -EINVAL;
4ec93edb 1048
50fee1de 1049 memset(&v, 0, sizeof(v));
df0bca04 1050
2a91525c 1051 switch (optname) {
e71a4783
SH		 1052 case SO_DEBUG:
1053 v.val = sock_flag(sk, SOCK_DBG);
1054 break;
1055
1056 case SO_DONTROUTE:
1057 v.val = sock_flag(sk, SOCK_LOCALROUTE);
1058 break;
1059
1060 case SO_BROADCAST:
1b23a5df 1061 v.val = sock_flag(sk, SOCK_BROADCAST);
e71a4783
SH		 1062 break;
1063
1064 case SO_SNDBUF:
1065 v.val = sk->sk_sndbuf;
1066 break;
1067
1068 case SO_RCVBUF:
1069 v.val = sk->sk_rcvbuf;
1070 break;
1071
1072 case SO_REUSEADDR:
1073 v.val = sk->sk_reuse;
1074 break;
1075
055dc21a
TH		 1076 case SO_REUSEPORT:
1077 v.val = sk->sk_reuseport;
1078 break;
1079
e71a4783 1080 case SO_KEEPALIVE:
1b23a5df 1081 v.val = sock_flag(sk, SOCK_KEEPOPEN);
e71a4783
SH		 1082 break;
1083
1084 case SO_TYPE:
1085 v.val = sk->sk_type;
1086 break;
1087
49c794e9
JE		 1088 case SO_PROTOCOL:
1089 v.val = sk->sk_protocol;
1090 break;
1091
0d6038ee
JE		 1092 case SO_DOMAIN:
1093 v.val = sk->sk_family;
1094 break;
1095
e71a4783
SH		 1096 case SO_ERROR:
1097 v.val = -sock_error(sk);
2a91525c 1098 if (v.val == 0)
e71a4783
SH		 1099 v.val = xchg(&sk->sk_err_soft, 0);
1100 break;
1101
1102 case SO_OOBINLINE:
1b23a5df 1103 v.val = sock_flag(sk, SOCK_URGINLINE);
e71a4783
SH		 1104 break;
1105
1106 case SO_NO_CHECK:
28448b80 1107 v.val = sk->sk_no_check_tx;
e71a4783
SH		 1108 break;
1109
1110 case SO_PRIORITY:
1111 v.val = sk->sk_priority;
1112 break;
1113
1114 case SO_LINGER:
1115 lv = sizeof(v.ling);
1b23a5df 1116 v.ling.l_onoff = sock_flag(sk, SOCK_LINGER);
e71a4783
SH		 1117 v.ling.l_linger = sk->sk_lingertime / HZ;
1118 break;
1119
1120 case SO_BSDCOMPAT:
1121 sock_warn_obsolete_bsdism("getsockopt");
1122 break;
1123
1124 case SO_TIMESTAMP:
92f37fd2
ED		 1125 v.val = sock_flag(sk, SOCK_RCVTSTAMP) &&
1126 !sock_flag(sk, SOCK_RCVTSTAMPNS);
1127 break;
1128
1129 case SO_TIMESTAMPNS:
1130 v.val = sock_flag(sk, SOCK_RCVTSTAMPNS);
e71a4783
SH		 1131 break;
1132
20d49473 1133 case SO_TIMESTAMPING:
b9f40e21 1134 v.val = sk->sk_tsflags;
20d49473
PO		 1135 break;
1136
e71a4783 1137 case SO_RCVTIMEO:
2a91525c 1138 lv = sizeof(struct timeval);
e71a4783
SH		 1139 if (sk->sk_rcvtimeo == MAX_SCHEDULE_TIMEOUT) {
1140 v.tm.tv_sec = 0;
1141 v.tm.tv_usec = 0;
1142 } else {
1143 v.tm.tv_sec = sk->sk_rcvtimeo / HZ;
1144 v.tm.tv_usec = ((sk->sk_rcvtimeo % HZ) * 1000000) / HZ;
1145 }
1146 break;
1147
1148 case SO_SNDTIMEO:
2a91525c 1149 lv = sizeof(struct timeval);
e71a4783
SH		 1150 if (sk->sk_sndtimeo == MAX_SCHEDULE_TIMEOUT) {
1151 v.tm.tv_sec = 0;
1152 v.tm.tv_usec = 0;
1153 } else {
1154 v.tm.tv_sec = sk->sk_sndtimeo / HZ;
1155 v.tm.tv_usec = ((sk->sk_sndtimeo % HZ) * 1000000) / HZ;
1156 }
1157 break;
1da177e4 1158
e71a4783
SH		 1159 case SO_RCVLOWAT:
1160 v.val = sk->sk_rcvlowat;
1161 break;
1da177e4 1162
e71a4783 1163 case SO_SNDLOWAT:
2a91525c 1164 v.val = 1;
e71a4783 1165 break;
1da177e4 1166
e71a4783 1167 case SO_PASSCRED:
82981930 1168 v.val = !!test_bit(SOCK_PASSCRED, &sock->flags);
e71a4783 1169 break;
1da177e4 1170
e71a4783 1171 case SO_PEERCRED:
109f6e39
EB		 1172 {
1173 struct ucred peercred;
1174 if (len > sizeof(peercred))
1175 len = sizeof(peercred);
1176 cred_to_ucred(sk->sk_peer_pid, sk->sk_peer_cred, &peercred);
1177 if (copy_to_user(optval, &peercred, len))
e71a4783
SH		 1178 return -EFAULT;
1179 goto lenout;
109f6e39 1180 }
1da177e4 1181
e71a4783
SH		 1182 case SO_PEERNAME:
1183 {
1184 char address[128];
1185
1186 if (sock->ops->getname(sock, (struct sockaddr *)address, &lv, 2))
1187 return -ENOTCONN;
1188 if (lv < len)
1189 return -EINVAL;
1190 if (copy_to_user(optval, address, len))
1191 return -EFAULT;
1192 goto lenout;
1193 }
1da177e4 1194
e71a4783
SH		 1195 /* Dubious BSD thing... Probably nobody even uses it, but
1196 * the UNIX standard wants it for whatever reason... -DaveM
1197 */
1198 case SO_ACCEPTCONN:
1199 v.val = sk->sk_state == TCP_LISTEN;
1200 break;
1da177e4 1201
e71a4783 1202 case SO_PASSSEC:
82981930 1203 v.val = !!test_bit(SOCK_PASSSEC, &sock->flags);
e71a4783 1204 break;
877ce7c1 1205
e71a4783
SH		 1206 case SO_PEERSEC:
1207 return security_socket_getpeersec_stream(sock, optval, optlen, len);
1da177e4 1208
4a19ec58
LAT		 1209 case SO_MARK:
1210 v.val = sk->sk_mark;
1211 break;
1212
3b885787 1213 case SO_RXQ_OVFL:
1b23a5df 1214 v.val = sock_flag(sk, SOCK_RXQ_OVFL);
3b885787
NH		 1215 break;
1216
6e3e939f 1217 case SO_WIFI_STATUS:
1b23a5df 1218 v.val = sock_flag(sk, SOCK_WIFI_STATUS);
6e3e939f
JB		 1219 break;
1220
ef64a54f
PE		 1221 case SO_PEEK_OFF:
1222 if (!sock->ops->set_peek_off)
1223 return -EOPNOTSUPP;
1224
1225 v.val = sk->sk_peek_off;
1226 break;
bc2f7996 1227 case SO_NOFCS:
1b23a5df 1228 v.val = sock_flag(sk, SOCK_NOFCS);
bc2f7996 1229 break;
c91f6df2 1230
f7b86bfe 1231 case SO_BINDTODEVICE:
c91f6df2
BH		 1232 return sock_getbindtodevice(sk, optval, optlen, len);
1233
a8fc9277
PE		 1234 case SO_GET_FILTER:
1235 len = sk_get_filter(sk, (struct sock_filter __user *)optval, len);
1236 if (len < 0)
1237 return len;
1238
1239 goto lenout;
c91f6df2 1240
d59577b6
VB		 1241 case SO_LOCK_FILTER:
1242 v.val = sock_flag(sk, SOCK_FILTER_LOCKED);
1243 break;
1244
ea02f941
MS		 1245 case SO_BPF_EXTENSIONS:
1246 v.val = bpf_tell_extensions();
1247 break;
1248
7d4c04fc
KJ		 1249 case SO_SELECT_ERR_QUEUE:
1250 v.val = sock_flag(sk, SOCK_SELECT_ERR_QUEUE);
1251 break;
1252
e0d1095a 1253#ifdef CONFIG_NET_RX_BUSY_POLL
64b0dc51 1254 case SO_BUSY_POLL:
dafcc438
ET		 1255 v.val = sk->sk_ll_usec;
1256 break;
1257#endif
1258
62748f32
ED		 1259 case SO_MAX_PACING_RATE:
1260 v.val = sk->sk_max_pacing_rate;
1261 break;
1262
2c8c56e1
ED		 1263 case SO_INCOMING_CPU:
1264 v.val = sk->sk_incoming_cpu;
1265 break;
1266
e71a4783 1267 default:
443b5991
YH		 1268 /* We implement the SO_SNDLOWAT etc to not be settable
1269 * (1003.1g 7).
1270 */
e71a4783 1271 return -ENOPROTOOPT;
1da177e4 1272 }
e71a4783 1273
1da177e4
LT		 1274 if (len > lv)
1275 len = lv;
1276 if (copy_to_user(optval, &v, len))
1277 return -EFAULT;
1278lenout:
4ec93edb
YH		 1279 if (put_user(len, optlen))
1280 return -EFAULT;
1281 return 0;
1da177e4
LT		 1282}
1283
a5b5bb9a
IM		 1284/*
1285 * Initialize an sk_lock.
1286 *
1287 * (We also register the sk_lock with the lock validator.)
1288 */
b6f99a21 1289static inline void sock_lock_init(struct sock *sk)
a5b5bb9a 1290{
ed07536e
PZ		 1291 sock_lock_init_class_and_name(sk,
1292 af_family_slock_key_strings[sk->sk_family],
1293 af_family_slock_keys + sk->sk_family,
1294 af_family_key_strings[sk->sk_family],
1295 af_family_keys + sk->sk_family);
a5b5bb9a
IM		 1296}
1297
4dc6dc71
ED		 1298/*
1299 * Copy all fields from osk to nsk but nsk->sk_refcnt must not change yet,
1300 * even temporarly, because of RCU lookups. sk_node should also be left as is.
68835aba 1301 * We must not copy fields between sk_dontcopy_begin and sk_dontcopy_end
4dc6dc71 1302 */
f1a6c4da
PE		 1303static void sock_copy(struct sock *nsk, const struct sock *osk)
1304{
1305#ifdef CONFIG_SECURITY_NETWORK
1306 void *sptr = nsk->sk_security;
1307#endif
68835aba
ED		 1308 memcpy(nsk, osk, offsetof(struct sock, sk_dontcopy_begin));
1309
1310 memcpy(&nsk->sk_dontcopy_end, &osk->sk_dontcopy_end,
1311 osk->sk_prot->obj_size - offsetof(struct sock, sk_dontcopy_end));
1312
f1a6c4da
PE		 1313#ifdef CONFIG_SECURITY_NETWORK
1314 nsk->sk_security = sptr;
1315 security_sk_clone(osk, nsk);
1316#endif
1317}
1318
fcbdf09d
OP		 1319void sk_prot_clear_portaddr_nulls(struct sock *sk, int size)
1320{
1321 unsigned long nulls1, nulls2;
1322
1323 nulls1 = offsetof(struct sock, __sk_common.skc_node.next);
1324 nulls2 = offsetof(struct sock, __sk_common.skc_portaddr_node.next);
1325 if (nulls1 > nulls2)
1326 swap(nulls1, nulls2);
1327
1328 if (nulls1 != 0)
1329 memset((char *)sk, 0, nulls1);
1330 memset((char *)sk + nulls1 + sizeof(void *), 0,
1331 nulls2 - nulls1 - sizeof(void *));
1332 memset((char *)sk + nulls2 + sizeof(void *), 0,
1333 size - nulls2 - sizeof(void *));
1334}
1335EXPORT_SYMBOL(sk_prot_clear_portaddr_nulls);
1336
2e4afe7b
PE		 1337static struct sock *sk_prot_alloc(struct proto *prot, gfp_t priority,
1338 int family)
c308c1b2
PE		 1339{
1340 struct sock *sk;
1341 struct kmem_cache *slab;
1342
1343 slab = prot->slab;
e912b114
ED		 1344 if (slab != NULL) {
1345 sk = kmem_cache_alloc(slab, priority & ~__GFP_ZERO);
1346 if (!sk)
1347 return sk;
1348 if (priority & __GFP_ZERO) {
fcbdf09d
OP		 1349 if (prot->clear_sk)
1350 prot->clear_sk(sk, prot->obj_size);
1351 else
1352 sk_prot_clear_nulls(sk, prot->obj_size);
e912b114 1353 }
fcbdf09d 1354 } else
c308c1b2
PE		 1355 sk = kmalloc(prot->obj_size, priority);
1356
2e4afe7b 1357 if (sk != NULL) {
a98b65a3
VN		 1358 kmemcheck_annotate_bitfield(sk, flags);
1359
2e4afe7b
PE		 1360 if (security_sk_alloc(sk, family, priority))
1361 goto out_free;
1362
1363 if (!try_module_get(prot->owner))
1364 goto out_free_sec;
e022f0b4 1365 sk_tx_queue_clear(sk);
2e4afe7b
PE		 1366 }
1367
c308c1b2 1368 return sk;
2e4afe7b
PE		 1369
1370out_free_sec:
1371 security_sk_free(sk);
1372out_free:
1373 if (slab != NULL)
1374 kmem_cache_free(slab, sk);
1375 else
1376 kfree(sk);
1377 return NULL;
c308c1b2
PE		 1378}
1379
1380static void sk_prot_free(struct proto *prot, struct sock *sk)
1381{
1382 struct kmem_cache *slab;
2e4afe7b 1383 struct module *owner;
c308c1b2 1384
2e4afe7b 1385 owner = prot->owner;
c308c1b2 1386 slab = prot->slab;
2e4afe7b
PE		 1387
1388 security_sk_free(sk);
c308c1b2
PE		 1389 if (slab != NULL)
1390 kmem_cache_free(slab, sk);
1391 else
1392 kfree(sk);
2e4afe7b 1393 module_put(owner);
c308c1b2
PE		 1394}
1395
86f8515f 1396#if IS_ENABLED(CONFIG_CGROUP_NET_PRIO)
6ffd4641 1397void sock_update_netprioidx(struct sock *sk)
5bc1421e 1398{
5bc1421e
NH		 1399 if (in_interrupt())
1400 return;
2b73bc65 1401
6ffd4641 1402 sk->sk_cgrp_prioidx = task_netprioidx(current);
5bc1421e
NH		 1403}
1404EXPORT_SYMBOL_GPL(sock_update_netprioidx);
f8451725
HX		 1405#endif
1406
1da177e4
LT		 1407/**
1408 * sk_alloc - All socket objects are allocated here
c4ea43c5 1409 * @net: the applicable net namespace
4dc3b16b
PP		 1410 * @family: protocol family
1411 * @priority: for allocation (%GFP_KERNEL, %GFP_ATOMIC, etc)
1412 * @prot: struct proto associated with this new sock instance
11aa9c28 1413 * @kern: is this to be a kernel socket?
1da177e4 1414 */
1b8d7ae4 1415struct sock *sk_alloc(struct net *net, int family, gfp_t priority,
11aa9c28 1416 struct proto *prot, int kern)
1da177e4 1417{
c308c1b2 1418 struct sock *sk;
1da177e4 1419
154adbc8 1420 sk = sk_prot_alloc(prot, priority | __GFP_ZERO, family);
1da177e4 1421 if (sk) {
154adbc8
PE		 1422 sk->sk_family = family;
1423 /*
1424 * See comment in struct sock definition to understand
1425 * why we need sk_prot_creator -acme
1426 */
1427 sk->sk_prot = sk->sk_prot_creator = prot;
1428 sock_lock_init(sk);
26abe143
EB		 1429 sk->sk_net_refcnt = kern ? 0 : 1;
1430 if (likely(sk->sk_net_refcnt))
1431 get_net(net);
1432 sock_net_set(sk, net);
d66ee058 1433 atomic_set(&sk->sk_wmem_alloc, 1);
f8451725 1434
211d2f97 1435 sock_update_classid(sk);
6ffd4641 1436 sock_update_netprioidx(sk);
1da177e4 1437 }
a79af59e 1438
2e4afe7b 1439 return sk;
1da177e4 1440}
2a91525c 1441EXPORT_SYMBOL(sk_alloc);
1da177e4 1442
eb4cb008 1443void sk_destruct(struct sock *sk)
1da177e4
LT		 1444{
1445 struct sk_filter *filter;
1da177e4
LT		 1446
1447 if (sk->sk_destruct)
1448 sk->sk_destruct(sk);
1449
a898def2
PM		 1450 filter = rcu_dereference_check(sk->sk_filter,
1451 atomic_read(&sk->sk_wmem_alloc) == 0);
1da177e4 1452 if (filter) {
309dd5fc 1453 sk_filter_uncharge(sk, filter);
a9b3cd7f 1454 RCU_INIT_POINTER(sk->sk_filter, NULL);
1da177e4
LT		 1455 }
1456
08e29af3 1457 sock_disable_timestamp(sk, SK_FLAGS_TIMESTAMP);
1da177e4
LT		 1458
1459 if (atomic_read(&sk->sk_omem_alloc))
e005d193
JP		 1460 pr_debug("%s: optmem leakage (%d bytes) detected\n",
1461 __func__, atomic_read(&sk->sk_omem_alloc));
1da177e4 1462
109f6e39
EB		 1463 if (sk->sk_peer_cred)
1464 put_cred(sk->sk_peer_cred);
1465 put_pid(sk->sk_peer_pid);
26abe143
EB		 1466 if (likely(sk->sk_net_refcnt))
1467 put_net(sock_net(sk));
c308c1b2 1468 sk_prot_free(sk->sk_prot_creator, sk);
1da177e4 1469}
2b85a34e 1470
eb4cb008
CG		 1471static void __sk_free(struct sock *sk)
1472{
b922622e 1473 if (unlikely(sock_diag_has_destroy_listeners(sk) && sk->sk_net_refcnt))
eb4cb008
CG		 1474 sock_diag_broadcast_destroy(sk);
1475 else
1476 sk_destruct(sk);
1477}
1478
2b85a34e
ED		 1479void sk_free(struct sock *sk)
1480{
1481 /*
25985edc 1482 * We subtract one from sk_wmem_alloc and can know if
2b85a34e
ED		 1483 * some packets are still in some tx queue.
1484 * If not null, sock_wfree() will call __sk_free(sk) later
1485 */
1486 if (atomic_dec_and_test(&sk->sk_wmem_alloc))
1487 __sk_free(sk);
1488}
2a91525c 1489EXPORT_SYMBOL(sk_free);
1da177e4 1490
475f1b52
SR		 1491static void sk_update_clone(const struct sock *sk, struct sock *newsk)
1492{
1493 if (mem_cgroup_sockets_enabled && sk->sk_cgrp)
1494 sock_update_memcg(newsk);
1495}
1496
e56c57d0
ED		 1497/**
1498 * sk_clone_lock - clone a socket, and lock its clone
1499 * @sk: the socket to clone
1500 * @priority: for allocation (%GFP_KERNEL, %GFP_ATOMIC, etc)
1501 *
1502 * Caller must unlock socket even in error path (bh_unlock_sock(newsk))
1503 */
1504struct sock *sk_clone_lock(const struct sock *sk, const gfp_t priority)
87d11ceb 1505{
8fd1d178 1506 struct sock *newsk;
278571ba 1507 bool is_charged = true;
87d11ceb 1508
8fd1d178 1509 newsk = sk_prot_alloc(sk->sk_prot, priority, sk->sk_family);
87d11ceb
ACM		 1510 if (newsk != NULL) {
1511 struct sk_filter *filter;
1512
892c141e 1513 sock_copy(newsk, sk);
87d11ceb
ACM		 1514
1515 /* SANITY */
8a681736
SV		 1516 if (likely(newsk->sk_net_refcnt))
1517 get_net(sock_net(newsk));
87d11ceb
ACM		 1518 sk_node_init(&newsk->sk_node);
1519 sock_lock_init(newsk);
1520 bh_lock_sock(newsk);
fa438ccf 1521 newsk->sk_backlog.head = newsk->sk_backlog.tail = NULL;
8eae939f 1522 newsk->sk_backlog.len = 0;
87d11ceb
ACM		 1523
1524 atomic_set(&newsk->sk_rmem_alloc, 0);
2b85a34e
ED		 1525 /*
1526 * sk_wmem_alloc set to one (see sk_free() and sock_wfree())
1527 */
1528 atomic_set(&newsk->sk_wmem_alloc, 1);
87d11ceb
ACM		 1529 atomic_set(&newsk->sk_omem_alloc, 0);
1530 skb_queue_head_init(&newsk->sk_receive_queue);
1531 skb_queue_head_init(&newsk->sk_write_queue);
1532
b6c6712a 1533 spin_lock_init(&newsk->sk_dst_lock);
87d11ceb 1534 rwlock_init(&newsk->sk_callback_lock);
443aef0e
PZ		 1535 lockdep_set_class_and_name(&newsk->sk_callback_lock,
1536 af_callback_keys + newsk->sk_family,
1537 af_family_clock_key_strings[newsk->sk_family]);
87d11ceb
ACM		 1538
1539 newsk->sk_dst_cache = NULL;
1540 newsk->sk_wmem_queued = 0;
1541 newsk->sk_forward_alloc = 0;
1542 newsk->sk_send_head = NULL;
87d11ceb
ACM		 1543 newsk->sk_userlocks = sk->sk_userlocks & ~SOCK_BINDPORT_LOCK;
1544
1545 sock_reset_flag(newsk, SOCK_DONE);
1546 skb_queue_head_init(&newsk->sk_error_queue);
1547
0d7da9dd 1548 filter = rcu_dereference_protected(newsk->sk_filter, 1);
87d11ceb 1549 if (filter != NULL)
278571ba
AS		 1550 /* though it's an empty new sock, the charging may fail
1551 * if sysctl_optmem_max was changed between creation of
1552 * original socket and cloning
1553 */
1554 is_charged = sk_filter_charge(newsk, filter);
87d11ceb 1555
278571ba 1556 if (unlikely(!is_charged || xfrm_sk_clone_policy(newsk))) {
87d11ceb
ACM		 1557 /* It is still raw copy of parent, so invalidate
1558 * destructor and make plain sk_free() */
1559 newsk->sk_destruct = NULL;
b0691c8e 1560 bh_unlock_sock(newsk);
87d11ceb
ACM		 1561 sk_free(newsk);
1562 newsk = NULL;
1563 goto out;
1564 }
1565
1566 newsk->sk_err = 0;
1567 newsk->sk_priority = 0;
2c8c56e1 1568 newsk->sk_incoming_cpu = raw_smp_processor_id();
33cf7c90 1569 atomic64_set(&newsk->sk_cookie, 0);
4dc6dc71
ED		 1570 /*
1571 * Before updating sk_refcnt, we must commit prior changes to memory
1572 * (Documentation/RCU/rculist_nulls.txt for details)
1573 */
1574 smp_wmb();
87d11ceb
ACM		 1575 atomic_set(&newsk->sk_refcnt, 2);
1576
1577 /*
1578 * Increment the counter in the same struct proto as the master
1579 * sock (sk_refcnt_debug_inc uses newsk->sk_prot->socks, that
1580 * is the same as sk->sk_prot->socks, as this field was copied
1581 * with memcpy).
1582 *
1583 * This _changes_ the previous behaviour, where
1584 * tcp_create_openreq_child always was incrementing the
1585 * equivalent to tcp_prot->socks (inet_sock_nr), so this have
1586 * to be taken into account in all callers. -acme
1587 */
1588 sk_refcnt_debug_inc(newsk);
972692e0 1589 sk_set_socket(newsk, NULL);
43815482 1590 newsk->sk_wq = NULL;
87d11ceb 1591
f3f511e1
GC		 1592 sk_update_clone(sk, newsk);
1593
87d11ceb 1594 if (newsk->sk_prot->sockets_allocated)
180d8cd9 1595 sk_sockets_allocated_inc(newsk);
704da560 1596
080a270f
HFS		 1597 if (sock_needs_netstamp(sk) &&
1598 newsk->sk_flags & SK_FLAGS_TIMESTAMP)
704da560 1599 net_enable_timestamp();
87d11ceb
ACM		 1600 }
1601out:
1602 return newsk;
1603}
e56c57d0 1604EXPORT_SYMBOL_GPL(sk_clone_lock);
87d11ceb 1605
9958089a
AK		 1606void sk_setup_caps(struct sock *sk, struct dst_entry *dst)
1607{
d6a4e26a
ED		 1608 u32 max_segs = 1;
1609
9958089a
AK		 1610 __sk_dst_set(sk, dst);
1611 sk->sk_route_caps = dst->dev->features;
1612 if (sk->sk_route_caps & NETIF_F_GSO)
4fcd6b99 1613 sk->sk_route_caps |= NETIF_F_GSO_SOFTWARE;
a465419b 1614 sk->sk_route_caps &= ~sk->sk_route_nocaps;
9958089a 1615 if (sk_can_gso(sk)) {
82cc1a7a 1616 if (dst->header_len) {
9958089a 1617 sk->sk_route_caps &= ~NETIF_F_GSO_MASK;
82cc1a7a 1618 } else {
9958089a 1619 sk->sk_route_caps |= NETIF_F_SG | NETIF_F_HW_CSUM;
82cc1a7a 1620 sk->sk_gso_max_size = dst->dev->gso_max_size;
d6a4e26a 1621 max_segs = max_t(u32, dst->dev->gso_max_segs, 1);
82cc1a7a 1622 }
9958089a 1623 }
d6a4e26a 1624 sk->sk_gso_max_segs = max_segs;
9958089a
AK		 1625}
1626EXPORT_SYMBOL_GPL(sk_setup_caps);
1627
1da177e4
LT		 1628/*
1629 * Simple resource managers for sockets.
1630 */
1631
1632
4ec93edb
YH		 1633/*
1634 * Write buffer destructor automatically called from kfree_skb.
1da177e4
LT		 1635 */
1636void sock_wfree(struct sk_buff *skb)
1637{
1638 struct sock *sk = skb->sk;
d99927f4 1639 unsigned int len = skb->truesize;
1da177e4 1640
d99927f4
ED		 1641 if (!sock_flag(sk, SOCK_USE_WRITE_QUEUE)) {
1642 /*
1643 * Keep a reference on sk_wmem_alloc, this will be released
1644 * after sk_write_space() call
1645 */
1646 atomic_sub(len - 1, &sk->sk_wmem_alloc);
1da177e4 1647 sk->sk_write_space(sk);
d99927f4
ED		 1648 len = 1;
1649 }
2b85a34e 1650 /*
d99927f4
ED		 1651 * if sk_wmem_alloc reaches 0, we must finish what sk_free()
1652 * could not do because of in-flight packets
2b85a34e 1653 */
d99927f4 1654 if (atomic_sub_and_test(len, &sk->sk_wmem_alloc))
2b85a34e 1655 __sk_free(sk);
1da177e4 1656}
2a91525c 1657EXPORT_SYMBOL(sock_wfree);
1da177e4 1658
f2f872f9
ED		 1659void skb_orphan_partial(struct sk_buff *skb)
1660{
1661 /* TCP stack sets skb->ooo_okay based on sk_wmem_alloc,
1662 * so we do not completely orphan skb, but transfert all
1663 * accounted bytes but one, to avoid unexpected reorders.
1664 */
1665 if (skb->destructor == sock_wfree
1666#ifdef CONFIG_INET
1667 || skb->destructor == tcp_wfree
1668#endif
1669 ) {
1670 atomic_sub(skb->truesize - 1, &skb->sk->sk_wmem_alloc);
1671 skb->truesize = 1;
1672 } else {
1673 skb_orphan(skb);
1674 }
1675}
1676EXPORT_SYMBOL(skb_orphan_partial);
1677
4ec93edb
YH		 1678/*
1679 * Read buffer destructor automatically called from kfree_skb.
1da177e4
LT		 1680 */
1681void sock_rfree(struct sk_buff *skb)
1682{
1683 struct sock *sk = skb->sk;
d361fd59 1684 unsigned int len = skb->truesize;
1da177e4 1685
d361fd59
ED		 1686 atomic_sub(len, &sk->sk_rmem_alloc);
1687 sk_mem_uncharge(sk, len);
1da177e4 1688}
2a91525c 1689EXPORT_SYMBOL(sock_rfree);
1da177e4 1690
7768eed8
OH		 1691/*
1692 * Buffer destructor for skbs that are not used directly in read or write
1693 * path, e.g. for error handler skbs. Automatically called from kfree_skb.
1694 */
62bccb8c
AD		 1695void sock_efree(struct sk_buff *skb)
1696{
1697 sock_put(skb->sk);
1698}
1699EXPORT_SYMBOL(sock_efree);
1700
976d0201 1701kuid_t sock_i_uid(struct sock *sk)
1da177e4 1702{
976d0201 1703 kuid_t uid;
1da177e4 1704
f064af1e 1705 read_lock_bh(&sk->sk_callback_lock);
976d0201 1706 uid = sk->sk_socket ? SOCK_INODE(sk->sk_socket)->i_uid : GLOBAL_ROOT_UID;
f064af1e 1707 read_unlock_bh(&sk->sk_callback_lock);
1da177e4
LT		 1708 return uid;
1709}
2a91525c 1710EXPORT_SYMBOL(sock_i_uid);
1da177e4
LT		 1711
1712unsigned long sock_i_ino(struct sock *sk)
1713{
1714 unsigned long ino;
1715
f064af1e 1716 read_lock_bh(&sk->sk_callback_lock);
1da177e4 1717 ino = sk->sk_socket ? SOCK_INODE(sk->sk_socket)->i_ino : 0;
f064af1e 1718 read_unlock_bh(&sk->sk_callback_lock);
1da177e4
LT		 1719 return ino;
1720}
2a91525c 1721EXPORT_SYMBOL(sock_i_ino);
1da177e4
LT		 1722
1723/*
1724 * Allocate a skb from the socket's send buffer.
1725 */
86a76caf 1726struct sk_buff *sock_wmalloc(struct sock *sk, unsigned long size, int force,
dd0fc66f 1727 gfp_t priority)
1da177e4
LT		 1728{
1729 if (force || atomic_read(&sk->sk_wmem_alloc) < sk->sk_sndbuf) {
2a91525c 1730 struct sk_buff *skb = alloc_skb(size, priority);
1da177e4
LT		 1731 if (skb) {
1732 skb_set_owner_w(skb, sk);
1733 return skb;
1734 }
1735 }
1736 return NULL;
1737}
2a91525c 1738EXPORT_SYMBOL(sock_wmalloc);
1da177e4 1739
4ec93edb 1740/*
1da177e4 1741 * Allocate a memory block from the socket's option memory buffer.
4ec93edb 1742 */
dd0fc66f 1743void *sock_kmalloc(struct sock *sk, int size, gfp_t priority)
1da177e4 1744{
95c96174 1745 if ((unsigned int)size <= sysctl_optmem_max &&
1da177e4
LT		 1746 atomic_read(&sk->sk_omem_alloc) + size < sysctl_optmem_max) {
1747 void *mem;
1748 /* First do the add, to avoid the race if kmalloc
4ec93edb 1749 * might sleep.
1da177e4
LT		 1750 */
1751 atomic_add(size, &sk->sk_omem_alloc);
1752 mem = kmalloc(size, priority);
1753 if (mem)
1754 return mem;
1755 atomic_sub(size, &sk->sk_omem_alloc);
1756 }
1757 return NULL;
1758}
2a91525c 1759EXPORT_SYMBOL(sock_kmalloc);
1da177e4 1760
79e88659
DB		 1761/* Free an option memory block. Note, we actually want the inline
1762 * here as this allows gcc to detect the nullify and fold away the
1763 * condition entirely.
1da177e4 1764 */
79e88659
DB		 1765static inline void __sock_kfree_s(struct sock *sk, void *mem, int size,
1766 const bool nullify)
1da177e4 1767{
e53da5fb
DM		 1768 if (WARN_ON_ONCE(!mem))
1769 return;
79e88659
DB		 1770 if (nullify)
1771 kzfree(mem);
1772 else
1773 kfree(mem);
1da177e4
LT		 1774 atomic_sub(size, &sk->sk_omem_alloc);
1775}
79e88659
DB		 1776
1777void sock_kfree_s(struct sock *sk, void *mem, int size)
1778{
1779 __sock_kfree_s(sk, mem, size, false);
1780}
2a91525c 1781EXPORT_SYMBOL(sock_kfree_s);
1da177e4 1782
79e88659
DB		 1783void sock_kzfree_s(struct sock *sk, void *mem, int size)
1784{
1785 __sock_kfree_s(sk, mem, size, true);
1786}
1787EXPORT_SYMBOL(sock_kzfree_s);
1788
1da177e4
LT		 1789/* It is almost wait_for_tcp_memory minus release_sock/lock_sock.
1790 I think, these locks should be removed for datagram sockets.
1791 */
2a91525c 1792static long sock_wait_for_wmem(struct sock *sk, long timeo)
1da177e4
LT		 1793{
1794 DEFINE_WAIT(wait);
1795
1796 clear_bit(SOCK_ASYNC_NOSPACE, &sk->sk_socket->flags);
1797 for (;;) {
1798 if (!timeo)
1799 break;
1800 if (signal_pending(current))
1801 break;
1802 set_bit(SOCK_NOSPACE, &sk->sk_socket->flags);
aa395145 1803 prepare_to_wait(sk_sleep(sk), &wait, TASK_INTERRUPTIBLE);
1da177e4
LT		 1804 if (atomic_read(&sk->sk_wmem_alloc) < sk->sk_sndbuf)
1805 break;
1806 if (sk->sk_shutdown & SEND_SHUTDOWN)
1807 break;
1808 if (sk->sk_err)
1809 break;
1810 timeo = schedule_timeout(timeo);
1811 }
aa395145 1812 finish_wait(sk_sleep(sk), &wait);
1da177e4
LT		 1813 return timeo;
1814}
1815
1816
1817/*
1818 * Generic send/receive buffer handlers
1819 */
1820
4cc7f68d
HX		 1821struct sk_buff *sock_alloc_send_pskb(struct sock *sk, unsigned long header_len,
1822 unsigned long data_len, int noblock,
28d64271 1823 int *errcode, int max_page_order)
1da177e4 1824{
2e4e4410 1825 struct sk_buff *skb;
1da177e4
LT		 1826 long timeo;
1827 int err;
1828
1da177e4 1829 timeo = sock_sndtimeo(sk, noblock);
2e4e4410 1830 for (;;) {
1da177e4
LT		 1831 err = sock_error(sk);
1832 if (err != 0)
1833 goto failure;
1834
1835 err = -EPIPE;
1836 if (sk->sk_shutdown & SEND_SHUTDOWN)
1837 goto failure;
1838
2e4e4410
ED		 1839 if (sk_wmem_alloc_get(sk) < sk->sk_sndbuf)
1840 break;
28d64271 1841
2e4e4410
ED		 1842 set_bit(SOCK_ASYNC_NOSPACE, &sk->sk_socket->flags);
1843 set_bit(SOCK_NOSPACE, &sk->sk_socket->flags);
1844 err = -EAGAIN;
1845 if (!timeo)
1da177e4 1846 goto failure;
2e4e4410
ED		 1847 if (signal_pending(current))
1848 goto interrupted;
1849 timeo = sock_wait_for_wmem(sk, timeo);
1da177e4 1850 }
2e4e4410
ED		 1851 skb = alloc_skb_with_frags(header_len, data_len, max_page_order,
1852 errcode, sk->sk_allocation);
1853 if (skb)
1854 skb_set_owner_w(skb, sk);
1da177e4
LT		 1855 return skb;
1856
1857interrupted:
1858 err = sock_intr_errno(timeo);
1859failure:
1860 *errcode = err;
1861 return NULL;
1862}
4cc7f68d 1863EXPORT_SYMBOL(sock_alloc_send_pskb);
1da177e4 1864
4ec93edb 1865struct sk_buff *sock_alloc_send_skb(struct sock *sk, unsigned long size,
1da177e4
LT		 1866 int noblock, int *errcode)
1867{
28d64271 1868 return sock_alloc_send_pskb(sk, size, 0, noblock, errcode, 0);
1da177e4 1869}
2a91525c 1870EXPORT_SYMBOL(sock_alloc_send_skb);
1da177e4 1871
f28ea365
EJ		 1872int sock_cmsg_send(struct sock *sk, struct msghdr *msg,
1873 struct sockcm_cookie *sockc)
1874{
1875 struct cmsghdr *cmsg;
1876
1877 for_each_cmsghdr(cmsg, msg) {
1878 if (!CMSG_OK(msg, cmsg))
1879 return -EINVAL;
1880 if (cmsg->cmsg_level != SOL_SOCKET)
1881 continue;
1882 switch (cmsg->cmsg_type) {
1883 case SO_MARK:
1884 if (!ns_capable(sock_net(sk)->user_ns, CAP_NET_ADMIN))
1885 return -EPERM;
1886 if (cmsg->cmsg_len != CMSG_LEN(sizeof(u32)))
1887 return -EINVAL;
1888 sockc->mark = *(u32 *)CMSG_DATA(cmsg);
1889 break;
1890 default:
1891 return -EINVAL;
1892 }
1893 }
1894 return 0;
1895}
1896EXPORT_SYMBOL(sock_cmsg_send);
1897
5640f768
ED		 1898/* On 32bit arches, an skb frag is limited to 2^15 */
1899#define SKB_FRAG_PAGE_ORDER get_order(32768)
1900
400dfd3a
ED		 1901/**
1902 * skb_page_frag_refill - check that a page_frag contains enough room
1903 * @sz: minimum size of the fragment we want to get
1904 * @pfrag: pointer to page_frag
82d5e2b8 1905 * @gfp: priority for memory allocation
400dfd3a
ED		 1906 *
1907 * Note: While this allocator tries to use high order pages, there is
1908 * no guarantee that allocations succeed. Therefore, @sz MUST be
1909 * less or equal than PAGE_SIZE.
1910 */
d9b2938a 1911bool skb_page_frag_refill(unsigned int sz, struct page_frag *pfrag, gfp_t gfp)
5640f768 1912{
5640f768
ED		 1913 if (pfrag->page) {
1914 if (atomic_read(&pfrag->page->_count) == 1) {
1915 pfrag->offset = 0;
1916 return true;
1917 }
400dfd3a 1918 if (pfrag->offset + sz <= pfrag->size)
5640f768
ED		 1919 return true;
1920 put_page(pfrag->page);
1921 }
1922
d9b2938a
ED		 1923 pfrag->offset = 0;
1924 if (SKB_FRAG_PAGE_ORDER) {
fb05e7a8 1925 pfrag->page = alloc_pages((gfp & ~__GFP_WAIT) | __GFP_COMP |
d9b2938a
ED		 1926 __GFP_NOWARN | __GFP_NORETRY,
1927 SKB_FRAG_PAGE_ORDER);
5640f768 1928 if (likely(pfrag->page)) {
d9b2938a 1929 pfrag->size = PAGE_SIZE << SKB_FRAG_PAGE_ORDER;
5640f768
ED		 1930 return true;
1931 }
d9b2938a
ED		 1932 }
1933 pfrag->page = alloc_page(gfp);
1934 if (likely(pfrag->page)) {
1935 pfrag->size = PAGE_SIZE;
1936 return true;
1937 }
400dfd3a
ED		 1938 return false;
1939}
1940EXPORT_SYMBOL(skb_page_frag_refill);
1941
1942bool sk_page_frag_refill(struct sock *sk, struct page_frag *pfrag)
1943{
1944 if (likely(skb_page_frag_refill(32U, pfrag, sk->sk_allocation)))
1945 return true;
1946
5640f768
ED		 1947 sk_enter_memory_pressure(sk);
1948 sk_stream_moderate_sndbuf(sk);
1949 return false;
1950}
1951EXPORT_SYMBOL(sk_page_frag_refill);
1952
1da177e4 1953static void __lock_sock(struct sock *sk)
f39234d6
NK		 1954 __releases(&sk->sk_lock.slock)
1955 __acquires(&sk->sk_lock.slock)
1da177e4
LT		 1956{
1957 DEFINE_WAIT(wait);
1958
e71a4783 1959 for (;;) {
1da177e4
LT		 1960 prepare_to_wait_exclusive(&sk->sk_lock.wq, &wait,
1961 TASK_UNINTERRUPTIBLE);
1962 spin_unlock_bh(&sk->sk_lock.slock);
1963 schedule();
1964 spin_lock_bh(&sk->sk_lock.slock);
e71a4783 1965 if (!sock_owned_by_user(sk))
1da177e4
LT		 1966 break;
1967 }
1968 finish_wait(&sk->sk_lock.wq, &wait);
1969}
1970
1971static void __release_sock(struct sock *sk)
f39234d6
NK		 1972 __releases(&sk->sk_lock.slock)
1973 __acquires(&sk->sk_lock.slock)
1da177e4
LT		 1974{
1975 struct sk_buff *skb = sk->sk_backlog.head;
1976
1977 do {
1978 sk->sk_backlog.head = sk->sk_backlog.tail = NULL;
1979 bh_unlock_sock(sk);
1980
1981 do {
1982 struct sk_buff *next = skb->next;
1983
e4cbb02a 1984 prefetch(next);
7fee226a 1985 WARN_ON_ONCE(skb_dst_is_noref(skb));
1da177e4 1986 skb->next = NULL;
c57943a1 1987 sk_backlog_rcv(sk, skb);
1da177e4
LT		 1988
1989 /*
1990 * We are in process context here with softirqs
1991 * disabled, use cond_resched_softirq() to preempt.
1992 * This is safe to do because we've taken the backlog
1993 * queue private:
1994 */
1995 cond_resched_softirq();
1996
1997 skb = next;
1998 } while (skb != NULL);
1999
2000 bh_lock_sock(sk);
e71a4783 2001 } while ((skb = sk->sk_backlog.head) != NULL);
8eae939f
ZY		 2002
2003 /*
2004 * Doing the zeroing here guarantee we can not loop forever
2005 * while a wild producer attempts to flood us.
2006 */
2007 sk->sk_backlog.len = 0;
1da177e4
LT		 2008}
2009
2010/**
2011 * sk_wait_data - wait for data to arrive at sk_receive_queue
4dc3b16b
PP		 2012 * @sk: sock to wait on
2013 * @timeo: for how long
dfbafc99 2014 * @skb: last skb seen on sk_receive_queue
1da177e4
LT		 2015 *
2016 * Now socket state including sk->sk_err is changed only under lock,
2017 * hence we may omit checks after joining wait queue.
2018 * We check receive queue before schedule() only as optimization;
2019 * it is very likely that release_sock() added new data.
2020 */
dfbafc99 2021int sk_wait_data(struct sock *sk, long *timeo, const struct sk_buff *skb)
1da177e4
LT		 2022{
2023 int rc;
2024 DEFINE_WAIT(wait);
2025
aa395145 2026 prepare_to_wait(sk_sleep(sk), &wait, TASK_INTERRUPTIBLE);
1da177e4 2027 set_bit(SOCK_ASYNC_WAITDATA, &sk->sk_socket->flags);
dfbafc99 2028 rc = sk_wait_event(sk, timeo, skb_peek_tail(&sk->sk_receive_queue) != skb);
1da177e4 2029 clear_bit(SOCK_ASYNC_WAITDATA, &sk->sk_socket->flags);
aa395145 2030 finish_wait(sk_sleep(sk), &wait);
1da177e4
LT		 2031 return rc;
2032}
1da177e4
LT		 2033EXPORT_SYMBOL(sk_wait_data);
2034
3ab224be
HA		 2035/**
2036 * __sk_mem_schedule - increase sk_forward_alloc and memory_allocated
2037 * @sk: socket
2038 * @size: memory size to allocate
2039 * @kind: allocation type
2040 *
2041 * If kind is SK_MEM_SEND, it means wmem allocation. Otherwise it means
2042 * rmem allocation. This function assumes that protocols which have
2043 * memory_pressure use sk_wmem_queued as write buffer accounting.
2044 */
2045int __sk_mem_schedule(struct sock *sk, int size, int kind)
2046{
2047 struct proto *prot = sk->sk_prot;
2048 int amt = sk_mem_pages(size);
8d987e5c 2049 long allocated;
e1aab161 2050 int parent_status = UNDER_LIMIT;
3ab224be
HA		 2051
2052 sk->sk_forward_alloc += amt * SK_MEM_QUANTUM;
180d8cd9 2053
e1aab161 2054 allocated = sk_memory_allocated_add(sk, amt, &parent_status);
3ab224be
HA		 2055
2056 /* Under limit. */
e1aab161
GC		 2057 if (parent_status == UNDER_LIMIT &&
2058 allocated <= sk_prot_mem_limits(sk, 0)) {
180d8cd9 2059 sk_leave_memory_pressure(sk);
3ab224be
HA		 2060 return 1;
2061 }
2062
e1aab161
GC		 2063 /* Under pressure. (we or our parents) */
2064 if ((parent_status > SOFT_LIMIT) ||
2065 allocated > sk_prot_mem_limits(sk, 1))
180d8cd9 2066 sk_enter_memory_pressure(sk);
3ab224be 2067
e1aab161
GC		 2068 /* Over hard limit (we or our parents) */
2069 if ((parent_status == OVER_LIMIT) ||
2070 (allocated > sk_prot_mem_limits(sk, 2)))
3ab224be
HA		 2071 goto suppress_allocation;
2072
2073 /* guarantee minimum buffer size under pressure */
2074 if (kind == SK_MEM_RECV) {
2075 if (atomic_read(&sk->sk_rmem_alloc) < prot->sysctl_rmem[0])
2076 return 1;
180d8cd9 2077
3ab224be
HA		 2078 } else { /* SK_MEM_SEND */
2079 if (sk->sk_type == SOCK_STREAM) {
2080 if (sk->sk_wmem_queued < prot->sysctl_wmem[0])
2081 return 1;
2082 } else if (atomic_read(&sk->sk_wmem_alloc) <
2083 prot->sysctl_wmem[0])
2084 return 1;
2085 }
2086
180d8cd9 2087 if (sk_has_memory_pressure(sk)) {
1748376b
ED		 2088 int alloc;
2089
180d8cd9 2090 if (!sk_under_memory_pressure(sk))
1748376b 2091 return 1;
180d8cd9
GC		 2092 alloc = sk_sockets_allocated_read_positive(sk);
2093 if (sk_prot_mem_limits(sk, 2) > alloc *
3ab224be
HA		 2094 sk_mem_pages(sk->sk_wmem_queued +
2095 atomic_read(&sk->sk_rmem_alloc) +
2096 sk->sk_forward_alloc))
2097 return 1;
2098 }
2099
2100suppress_allocation:
2101
2102 if (kind == SK_MEM_SEND && sk->sk_type == SOCK_STREAM) {
2103 sk_stream_moderate_sndbuf(sk);
2104
2105 /* Fail only if socket is _under_ its sndbuf.
2106 * In this case we cannot block, so that we have to fail.
2107 */
2108 if (sk->sk_wmem_queued + size >= sk->sk_sndbuf)
2109 return 1;
2110 }
2111
3847ce32
SM		 2112 trace_sock_exceed_buf_limit(sk, prot, allocated);
2113
3ab224be
HA		 2114 /* Alas. Undo changes. */
2115 sk->sk_forward_alloc -= amt * SK_MEM_QUANTUM;
180d8cd9 2116
0e90b31f 2117 sk_memory_allocated_sub(sk, amt);
180d8cd9 2118
3ab224be
HA		 2119 return 0;
2120}
3ab224be
HA		 2121EXPORT_SYMBOL(__sk_mem_schedule);
2122
2123/**
69dba9bb 2124 * __sk_mem_reclaim - reclaim memory_allocated
3ab224be 2125 * @sk: socket
1a24e04e 2126 * @amount: number of bytes (rounded down to a SK_MEM_QUANTUM multiple)
3ab224be 2127 */
1a24e04e 2128void __sk_mem_reclaim(struct sock *sk, int amount)
3ab224be 2129{
1a24e04e
ED		 2130 amount >>= SK_MEM_QUANTUM_SHIFT;
2131 sk_memory_allocated_sub(sk, amount);
2132 sk->sk_forward_alloc -= amount << SK_MEM_QUANTUM_SHIFT;
3ab224be 2133
180d8cd9
GC		 2134 if (sk_under_memory_pressure(sk) &&
2135 (sk_memory_allocated(sk) < sk_prot_mem_limits(sk, 0)))
2136 sk_leave_memory_pressure(sk);
3ab224be 2137}
3ab224be
HA		 2138EXPORT_SYMBOL(__sk_mem_reclaim);
2139
2140
1da177e4
LT		 2141/*
2142 * Set of default routines for initialising struct proto_ops when
2143 * the protocol does not support a particular function. In certain
2144 * cases where it makes no sense for a protocol to have a "do nothing"
2145 * function, some default processing is provided.
2146 */
2147
2148int sock_no_bind(struct socket *sock, struct sockaddr *saddr, int len)
2149{
2150 return -EOPNOTSUPP;
2151}
2a91525c 2152EXPORT_SYMBOL(sock_no_bind);
1da177e4 2153
4ec93edb 2154int sock_no_connect(struct socket *sock, struct sockaddr *saddr,
1da177e4
LT		 2155 int len, int flags)
2156{
2157 return -EOPNOTSUPP;
2158}
2a91525c 2159EXPORT_SYMBOL(sock_no_connect);
1da177e4
LT		 2160
2161int sock_no_socketpair(struct socket *sock1, struct socket *sock2)
2162{
2163 return -EOPNOTSUPP;
2164}
2a91525c 2165EXPORT_SYMBOL(sock_no_socketpair);
1da177e4
LT		 2166
2167int sock_no_accept(struct socket *sock, struct socket *newsock, int flags)
2168{
2169 return -EOPNOTSUPP;
2170}
2a91525c 2171EXPORT_SYMBOL(sock_no_accept);
1da177e4 2172
4ec93edb 2173int sock_no_getname(struct socket *sock, struct sockaddr *saddr,
1da177e4
LT		 2174 int *len, int peer)
2175{
2176 return -EOPNOTSUPP;
2177}
2a91525c 2178EXPORT_SYMBOL(sock_no_getname);
1da177e4 2179
2a91525c 2180unsigned int sock_no_poll(struct file *file, struct socket *sock, poll_table *pt)
1da177e4
LT		 2181{
2182 return 0;
2183}
2a91525c 2184EXPORT_SYMBOL(sock_no_poll);
1da177e4
LT		 2185
2186int sock_no_ioctl(struct socket *sock, unsigned int cmd, unsigned long arg)
2187{
2188 return -EOPNOTSUPP;
2189}
2a91525c 2190EXPORT_SYMBOL(sock_no_ioctl);
1da177e4
LT		 2191
2192int sock_no_listen(struct socket *sock, int backlog)
2193{
2194 return -EOPNOTSUPP;
2195}
2a91525c 2196EXPORT_SYMBOL(sock_no_listen);
1da177e4
LT		 2197
2198int sock_no_shutdown(struct socket *sock, int how)
2199{
2200 return -EOPNOTSUPP;
2201}
2a91525c 2202EXPORT_SYMBOL(sock_no_shutdown);
1da177e4
LT		 2203
2204int sock_no_setsockopt(struct socket *sock, int level, int optname,
b7058842 2205 char __user *optval, unsigned int optlen)
1da177e4
LT		 2206{
2207 return -EOPNOTSUPP;
2208}
2a91525c 2209EXPORT_SYMBOL(sock_no_setsockopt);
1da177e4
LT		 2210
2211int sock_no_getsockopt(struct socket *sock, int level, int optname,
2212 char __user *optval, int __user *optlen)
2213{
2214 return -EOPNOTSUPP;
2215}
2a91525c 2216EXPORT_SYMBOL(sock_no_getsockopt);
1da177e4 2217
1b784140 2218int sock_no_sendmsg(struct socket *sock, struct msghdr *m, size_t len)
1da177e4
LT		 2219{
2220 return -EOPNOTSUPP;
2221}
2a91525c 2222EXPORT_SYMBOL(sock_no_sendmsg);
1da177e4 2223
1b784140
YX		 2224int sock_no_recvmsg(struct socket *sock, struct msghdr *m, size_t len,
2225 int flags)
1da177e4
LT		 2226{
2227 return -EOPNOTSUPP;
2228}
2a91525c 2229EXPORT_SYMBOL(sock_no_recvmsg);
1da177e4
LT		 2230
2231int sock_no_mmap(struct file *file, struct socket *sock, struct vm_area_struct *vma)
2232{
2233 /* Mirror missing mmap method error code */
2234 return -ENODEV;
2235}
2a91525c 2236EXPORT_SYMBOL(sock_no_mmap);
1da177e4
LT		 2237
2238ssize_t sock_no_sendpage(struct socket *sock, struct page *page, int offset, size_t size, int flags)
2239{
2240 ssize_t res;
2241 struct msghdr msg = {.msg_flags = flags};
2242 struct kvec iov;
2243 char *kaddr = kmap(page);
2244 iov.iov_base = kaddr + offset;
2245 iov.iov_len = size;
2246 res = kernel_sendmsg(sock, &msg, &iov, 1, size);
2247 kunmap(page);
2248 return res;
2249}
2a91525c 2250EXPORT_SYMBOL(sock_no_sendpage);
1da177e4
LT		 2251
2252/*
2253 * Default Socket Callbacks
2254 */
2255
2256static void sock_def_wakeup(struct sock *sk)
2257{
43815482
ED		 2258 struct socket_wq *wq;
2259
2260 rcu_read_lock();
2261 wq = rcu_dereference(sk->sk_wq);
2262 if (wq_has_sleeper(wq))
2263 wake_up_interruptible_all(&wq->wait);
2264 rcu_read_unlock();
1da177e4
LT		 2265}
2266
2267static void sock_def_error_report(struct sock *sk)
2268{
43815482
ED		 2269 struct socket_wq *wq;
2270
2271 rcu_read_lock();
2272 wq = rcu_dereference(sk->sk_wq);
2273 if (wq_has_sleeper(wq))
2274 wake_up_interruptible_poll(&wq->wait, POLLERR);
8d8ad9d7 2275 sk_wake_async(sk, SOCK_WAKE_IO, POLL_ERR);
43815482 2276 rcu_read_unlock();
1da177e4
LT		 2277}
2278
676d2369 2279static void sock_def_readable(struct sock *sk)
1da177e4 2280{
43815482
ED		 2281 struct socket_wq *wq;
2282
2283 rcu_read_lock();
2284 wq = rcu_dereference(sk->sk_wq);
2285 if (wq_has_sleeper(wq))
2c6607c6 2286 wake_up_interruptible_sync_poll(&wq->wait, POLLIN | POLLPRI |
37e5540b 2287 POLLRDNORM | POLLRDBAND);
8d8ad9d7 2288 sk_wake_async(sk, SOCK_WAKE_WAITD, POLL_IN);
43815482 2289 rcu_read_unlock();
1da177e4
LT		 2290}
2291
2292static void sock_def_write_space(struct sock *sk)
2293{
43815482
ED		 2294 struct socket_wq *wq;
2295
2296 rcu_read_lock();
1da177e4
LT		 2297
2298 /* Do not wake up a writer until he can make "significant"
2299 * progress. --DaveM
2300 */
e71a4783 2301 if ((atomic_read(&sk->sk_wmem_alloc) << 1) <= sk->sk_sndbuf) {
43815482
ED		 2302 wq = rcu_dereference(sk->sk_wq);
2303 if (wq_has_sleeper(wq))
2304 wake_up_interruptible_sync_poll(&wq->wait, POLLOUT |
37e5540b 2305 POLLWRNORM | POLLWRBAND);
1da177e4
LT		 2306
2307 /* Should agree with poll, otherwise some programs break */
2308 if (sock_writeable(sk))
8d8ad9d7 2309 sk_wake_async(sk, SOCK_WAKE_SPACE, POLL_OUT);
1da177e4
LT		 2310 }
2311
43815482 2312 rcu_read_unlock();
1da177e4
LT		 2313}
2314
2315static void sock_def_destruct(struct sock *sk)
2316{
1da177e4
LT		 2317}
2318
2319void sk_send_sigurg(struct sock *sk)
2320{
2321 if (sk->sk_socket && sk->sk_socket->file)
2322 if (send_sigurg(&sk->sk_socket->file->f_owner))
8d8ad9d7 2323 sk_wake_async(sk, SOCK_WAKE_URG, POLL_PRI);
1da177e4 2324}
2a91525c 2325EXPORT_SYMBOL(sk_send_sigurg);
1da177e4
LT		 2326
2327void sk_reset_timer(struct sock *sk, struct timer_list* timer,
2328 unsigned long expires)
2329{
2330 if (!mod_timer(timer, expires))
2331 sock_hold(sk);
2332}
1da177e4
LT		 2333EXPORT_SYMBOL(sk_reset_timer);
2334
2335void sk_stop_timer(struct sock *sk, struct timer_list* timer)
2336{
25cc4ae9 2337 if (del_timer(timer))
1da177e4
LT		 2338 __sock_put(sk);
2339}
1da177e4
LT		 2340EXPORT_SYMBOL(sk_stop_timer);
2341
2342void sock_init_data(struct socket *sock, struct sock *sk)
2343{
2344 skb_queue_head_init(&sk->sk_receive_queue);
2345 skb_queue_head_init(&sk->sk_write_queue);
2346 skb_queue_head_init(&sk->sk_error_queue);
2347
2348 sk->sk_send_head = NULL;
2349
2350 init_timer(&sk->sk_timer);
4ec93edb 2351
1da177e4
LT		 2352 sk->sk_allocation = GFP_KERNEL;
2353 sk->sk_rcvbuf = sysctl_rmem_default;
2354 sk->sk_sndbuf = sysctl_wmem_default;
2355 sk->sk_state = TCP_CLOSE;
972692e0 2356 sk_set_socket(sk, sock);
1da177e4
LT		 2357
2358 sock_set_flag(sk, SOCK_ZAPPED);
2359
e71a4783 2360 if (sock) {
1da177e4 2361 sk->sk_type = sock->type;
43815482 2362 sk->sk_wq = sock->wq;
1da177e4
LT		 2363 sock->sk = sk;
2364 } else
43815482 2365 sk->sk_wq = NULL;
1da177e4 2366
b6c6712a 2367 spin_lock_init(&sk->sk_dst_lock);
1da177e4 2368 rwlock_init(&sk->sk_callback_lock);
443aef0e
PZ		 2369 lockdep_set_class_and_name(&sk->sk_callback_lock,
2370 af_callback_keys + sk->sk_family,
2371 af_family_clock_key_strings[sk->sk_family]);
1da177e4
LT		 2372
2373 sk->sk_state_change = sock_def_wakeup;
2374 sk->sk_data_ready = sock_def_readable;
2375 sk->sk_write_space = sock_def_write_space;
2376 sk->sk_error_report = sock_def_error_report;
2377 sk->sk_destruct = sock_def_destruct;
2378
5640f768
ED		 2379 sk->sk_frag.page = NULL;
2380 sk->sk_frag.offset = 0;
ef64a54f 2381 sk->sk_peek_off = -1;
1da177e4 2382
109f6e39
EB		 2383 sk->sk_peer_pid = NULL;
2384 sk->sk_peer_cred = NULL;
1da177e4
LT		 2385 sk->sk_write_pending = 0;
2386 sk->sk_rcvlowat = 1;
2387 sk->sk_rcvtimeo = MAX_SCHEDULE_TIMEOUT;
2388 sk->sk_sndtimeo = MAX_SCHEDULE_TIMEOUT;
2389
f37f0afb 2390 sk->sk_stamp = ktime_set(-1L, 0);
1da177e4 2391
e0d1095a 2392#ifdef CONFIG_NET_RX_BUSY_POLL
06021292 2393 sk->sk_napi_id = 0;
64b0dc51 2394 sk->sk_ll_usec = sysctl_net_busy_read;
06021292
ET		 2395#endif
2396
62748f32 2397 sk->sk_max_pacing_rate = ~0U;
7eec4174 2398 sk->sk_pacing_rate = ~0U;
70da268b 2399 sk->sk_incoming_cpu = -1;
4dc6dc71
ED		 2400 /*
2401 * Before updating sk_refcnt, we must commit prior changes to memory
2402 * (Documentation/RCU/rculist_nulls.txt for details)
2403 */
2404 smp_wmb();
1da177e4 2405 atomic_set(&sk->sk_refcnt, 1);
33c732c3 2406 atomic_set(&sk->sk_drops, 0);
1da177e4 2407}
2a91525c 2408EXPORT_SYMBOL(sock_init_data);
1da177e4 2409
b5606c2d 2410void lock_sock_nested(struct sock *sk, int subclass)
1da177e4
LT		 2411{
2412 might_sleep();
a5b5bb9a 2413 spin_lock_bh(&sk->sk_lock.slock);
d2e9117c 2414 if (sk->sk_lock.owned)
1da177e4 2415 __lock_sock(sk);
d2e9117c 2416 sk->sk_lock.owned = 1;
a5b5bb9a
IM		 2417 spin_unlock(&sk->sk_lock.slock);
2418 /*
2419 * The sk_lock has mutex_lock() semantics here:
2420 */
fcc70d5f 2421 mutex_acquire(&sk->sk_lock.dep_map, subclass, 0, _RET_IP_);
a5b5bb9a 2422 local_bh_enable();
1da177e4 2423}
fcc70d5f 2424EXPORT_SYMBOL(lock_sock_nested);
1da177e4 2425
b5606c2d 2426void release_sock(struct sock *sk)
1da177e4 2427{
a5b5bb9a
IM		 2428 /*
2429 * The sk_lock has mutex_unlock() semantics:
2430 */
2431 mutex_release(&sk->sk_lock.dep_map, 1, _RET_IP_);
2432
2433 spin_lock_bh(&sk->sk_lock.slock);
1da177e4
LT		 2434 if (sk->sk_backlog.tail)
2435 __release_sock(sk);
46d3ceab 2436
c3f9b018
ED		 2437 /* Warning : release_cb() might need to release sk ownership,
2438 * ie call sock_release_ownership(sk) before us.
2439 */
46d3ceab
ED		 2440 if (sk->sk_prot->release_cb)
2441 sk->sk_prot->release_cb(sk);
2442
c3f9b018 2443 sock_release_ownership(sk);
a5b5bb9a
IM		 2444 if (waitqueue_active(&sk->sk_lock.wq))
2445 wake_up(&sk->sk_lock.wq);
2446 spin_unlock_bh(&sk->sk_lock.slock);
1da177e4
LT		 2447}
2448EXPORT_SYMBOL(release_sock);
2449
8a74ad60
ED		 2450/**
2451 * lock_sock_fast - fast version of lock_sock
2452 * @sk: socket
2453 *
2454 * This version should be used for very small section, where process wont block
2455 * return false if fast path is taken
2456 * sk_lock.slock locked, owned = 0, BH disabled
2457 * return true if slow path is taken
2458 * sk_lock.slock unlocked, owned = 1, BH enabled
2459 */
2460bool lock_sock_fast(struct sock *sk)
2461{
2462 might_sleep();
2463 spin_lock_bh(&sk->sk_lock.slock);
2464
2465 if (!sk->sk_lock.owned)
2466 /*
2467 * Note : We must disable BH
2468 */
2469 return false;
2470
2471 __lock_sock(sk);
2472 sk->sk_lock.owned = 1;
2473 spin_unlock(&sk->sk_lock.slock);
2474 /*
2475 * The sk_lock has mutex_lock() semantics here:
2476 */
2477 mutex_acquire(&sk->sk_lock.dep_map, 0, 0, _RET_IP_);
2478 local_bh_enable();
2479 return true;
2480}
2481EXPORT_SYMBOL(lock_sock_fast);
2482
1da177e4 2483int sock_get_timestamp(struct sock *sk, struct timeval __user *userstamp)
4ec93edb 2484{
b7aa0bf7 2485 struct timeval tv;
1da177e4 2486 if (!sock_flag(sk, SOCK_TIMESTAMP))
20d49473 2487 sock_enable_timestamp(sk, SOCK_TIMESTAMP);
b7aa0bf7
ED		 2488 tv = ktime_to_timeval(sk->sk_stamp);
2489 if (tv.tv_sec == -1)
1da177e4 2490 return -ENOENT;
b7aa0bf7
ED		 2491 if (tv.tv_sec == 0) {
2492 sk->sk_stamp = ktime_get_real();
2493 tv = ktime_to_timeval(sk->sk_stamp);
2494 }
2495 return copy_to_user(userstamp, &tv, sizeof(tv)) ? -EFAULT : 0;
4ec93edb 2496}
1da177e4
LT		 2497EXPORT_SYMBOL(sock_get_timestamp);
2498
ae40eb1e
ED		 2499int sock_get_timestampns(struct sock *sk, struct timespec __user *userstamp)
2500{
2501 struct timespec ts;
2502 if (!sock_flag(sk, SOCK_TIMESTAMP))
20d49473 2503 sock_enable_timestamp(sk, SOCK_TIMESTAMP);
ae40eb1e
ED		 2504 ts = ktime_to_timespec(sk->sk_stamp);
2505 if (ts.tv_sec == -1)
2506 return -ENOENT;
2507 if (ts.tv_sec == 0) {
2508 sk->sk_stamp = ktime_get_real();
2509 ts = ktime_to_timespec(sk->sk_stamp);
2510 }
2511 return copy_to_user(userstamp, &ts, sizeof(ts)) ? -EFAULT : 0;
2512}
2513EXPORT_SYMBOL(sock_get_timestampns);
2514
20d49473 2515void sock_enable_timestamp(struct sock *sk, int flag)
4ec93edb 2516{
20d49473 2517 if (!sock_flag(sk, flag)) {
08e29af3
ED		 2518 unsigned long previous_flags = sk->sk_flags;
2519
20d49473
PO		 2520 sock_set_flag(sk, flag);
2521 /*
2522 * we just set one of the two flags which require net
2523 * time stamping, but time stamping might have been on
2524 * already because of the other one
2525 */
080a270f
HFS		 2526 if (sock_needs_netstamp(sk) &&
2527 !(previous_flags & SK_FLAGS_TIMESTAMP))
20d49473 2528 net_enable_timestamp();
1da177e4
LT		 2529 }
2530}
1da177e4 2531
cb820f8e
RC		 2532int sock_recv_errqueue(struct sock *sk, struct msghdr *msg, int len,
2533 int level, int type)
2534{
2535 struct sock_exterr_skb *serr;
364a9e93 2536 struct sk_buff *skb;
cb820f8e
RC		 2537 int copied, err;
2538
2539 err = -EAGAIN;
364a9e93 2540 skb = sock_dequeue_err_skb(sk);
cb820f8e
RC		 2541 if (skb == NULL)
2542 goto out;
2543
2544 copied = skb->len;
2545 if (copied > len) {
2546 msg->msg_flags |= MSG_TRUNC;
2547 copied = len;
2548 }
51f3d02b 2549 err = skb_copy_datagram_msg(skb, 0, msg, copied);
cb820f8e
RC		 2550 if (err)
2551 goto out_free_skb;
2552
2553 sock_recv_timestamp(msg, sk, skb);
2554
2555 serr = SKB_EXT_ERR(skb);
2556 put_cmsg(msg, level, type, sizeof(serr->ee), &serr->ee);
2557
2558 msg->msg_flags |= MSG_ERRQUEUE;
2559 err = copied;
2560
cb820f8e
RC		 2561out_free_skb:
2562 kfree_skb(skb);
2563out:
2564 return err;
2565}
2566EXPORT_SYMBOL(sock_recv_errqueue);
2567
1da177e4
LT		 2568/*
2569 * Get a socket option on an socket.
2570 *
2571 * FIX: POSIX 1003.1g is very ambiguous here. It states that
2572 * asynchronous errors should be reported by getsockopt. We assume
2573 * this means if you specify SO_ERROR (otherwise whats the point of it).
2574 */
2575int sock_common_getsockopt(struct socket *sock, int level, int optname,
2576 char __user *optval, int __user *optlen)
2577{
2578 struct sock *sk = sock->sk;
2579
2580 return sk->sk_prot->getsockopt(sk, level, optname, optval, optlen);
2581}
1da177e4
LT		 2582EXPORT_SYMBOL(sock_common_getsockopt);
2583
3fdadf7d 2584#ifdef CONFIG_COMPAT
543d9cfe
ACM		 2585int compat_sock_common_getsockopt(struct socket *sock, int level, int optname,
2586 char __user *optval, int __user *optlen)
3fdadf7d
DM		 2587{
2588 struct sock *sk = sock->sk;
2589
1e51f951 2590 if (sk->sk_prot->compat_getsockopt != NULL)
543d9cfe
ACM		 2591 return sk->sk_prot->compat_getsockopt(sk, level, optname,
2592 optval, optlen);
3fdadf7d
DM		 2593 return sk->sk_prot->getsockopt(sk, level, optname, optval, optlen);
2594}
2595EXPORT_SYMBOL(compat_sock_common_getsockopt);
2596#endif
2597
1b784140
YX		 2598int sock_common_recvmsg(struct socket *sock, struct msghdr *msg, size_t size,
2599 int flags)
1da177e4
LT		 2600{
2601 struct sock *sk = sock->sk;
2602 int addr_len = 0;
2603 int err;
2604
1b784140 2605 err = sk->sk_prot->recvmsg(sk, msg, size, flags & MSG_DONTWAIT,
1da177e4
LT		 2606 flags & ~MSG_DONTWAIT, &addr_len);
2607 if (err >= 0)
2608 msg->msg_namelen = addr_len;
2609 return err;
2610}
1da177e4
LT		 2611EXPORT_SYMBOL(sock_common_recvmsg);
2612
2613/*
2614 * Set socket options on an inet socket.
2615 */
2616int sock_common_setsockopt(struct socket *sock, int level, int optname,
b7058842 2617 char __user *optval, unsigned int optlen)
1da177e4
LT		 2618{
2619 struct sock *sk = sock->sk;
2620
2621 return sk->sk_prot->setsockopt(sk, level, optname, optval, optlen);
2622}
1da177e4
LT		 2623EXPORT_SYMBOL(sock_common_setsockopt);
2624
3fdadf7d 2625#ifdef CONFIG_COMPAT
543d9cfe 2626int compat_sock_common_setsockopt(struct socket *sock, int level, int optname,
b7058842 2627 char __user *optval, unsigned int optlen)
3fdadf7d
DM		 2628{
2629 struct sock *sk = sock->sk;
2630
543d9cfe
ACM		 2631 if (sk->sk_prot->compat_setsockopt != NULL)
2632 return sk->sk_prot->compat_setsockopt(sk, level, optname,
2633 optval, optlen);
3fdadf7d
DM		 2634 return sk->sk_prot->setsockopt(sk, level, optname, optval, optlen);
2635}
2636EXPORT_SYMBOL(compat_sock_common_setsockopt);
2637#endif
2638
1da177e4
LT		 2639void sk_common_release(struct sock *sk)
2640{
2641 if (sk->sk_prot->destroy)
2642 sk->sk_prot->destroy(sk);
2643
2644 /*
2645 * Observation: when sock_common_release is called, processes have
2646 * no access to socket. But net still has.
2647 * Step one, detach it from networking:
2648 *
2649 * A. Remove from hash tables.
2650 */
2651
2652 sk->sk_prot->unhash(sk);
2653
2654 /*
2655 * In this point socket cannot receive new packets, but it is possible
2656 * that some packets are in flight because some CPU runs receiver and
2657 * did hash table lookup before we unhashed socket. They will achieve
2658 * receive queue and will be purged by socket destructor.
2659 *
2660 * Also we still have packets pending on receive queue and probably,
2661 * our own packets waiting in device queues. sock_destroy will drain
2662 * receive queue, but transmitted packets will delay socket destruction
2663 * until the last reference will be released.
2664 */
2665
2666 sock_orphan(sk);
2667
2668 xfrm_sk_free_policy(sk);
2669
e6848976 2670 sk_refcnt_debug_release(sk);
5640f768
ED		 2671
2672 if (sk->sk_frag.page) {
2673 put_page(sk->sk_frag.page);
2674 sk->sk_frag.page = NULL;
2675 }
2676
1da177e4
LT		 2677 sock_put(sk);
2678}
1da177e4
LT		 2679EXPORT_SYMBOL(sk_common_release);
2680
13ff3d6f
PE		 2681#ifdef CONFIG_PROC_FS
2682#define PROTO_INUSE_NR 64 /* should be enough for the first time */
1338d466
PE		 2683struct prot_inuse {
2684 int val[PROTO_INUSE_NR];
2685};
13ff3d6f
PE		 2686
2687static DECLARE_BITMAP(proto_inuse_idx, PROTO_INUSE_NR);
70ee1159
PE		 2688
2689#ifdef CONFIG_NET_NS
2690void sock_prot_inuse_add(struct net *net, struct proto *prot, int val)
2691{
d6d9ca0f 2692 __this_cpu_add(net->core.inuse->val[prot->inuse_idx], val);
70ee1159
PE		 2693}
2694EXPORT_SYMBOL_GPL(sock_prot_inuse_add);
2695
2696int sock_prot_inuse_get(struct net *net, struct proto *prot)
2697{
2698 int cpu, idx = prot->inuse_idx;
2699 int res = 0;
2700
2701 for_each_possible_cpu(cpu)
2702 res += per_cpu_ptr(net->core.inuse, cpu)->val[idx];
2703
2704 return res >= 0 ? res : 0;
2705}
2706EXPORT_SYMBOL_GPL(sock_prot_inuse_get);
2707
2c8c1e72 2708static int __net_init sock_inuse_init_net(struct net *net)
70ee1159
PE		 2709{
2710 net->core.inuse = alloc_percpu(struct prot_inuse);
2711 return net->core.inuse ? 0 : -ENOMEM;
2712}
2713
2c8c1e72 2714static void __net_exit sock_inuse_exit_net(struct net *net)
70ee1159
PE		 2715{
2716 free_percpu(net->core.inuse);
2717}
2718
2719static struct pernet_operations net_inuse_ops = {
2720 .init = sock_inuse_init_net,
2721 .exit = sock_inuse_exit_net,
2722};
2723
2724static __init int net_inuse_init(void)
2725{
2726 if (register_pernet_subsys(&net_inuse_ops))
2727 panic("Cannot initialize net inuse counters");
2728
2729 return 0;
2730}
2731
2732core_initcall(net_inuse_init);
2733#else
1338d466
PE		 2734static DEFINE_PER_CPU(struct prot_inuse, prot_inuse);
2735
c29a0bc4 2736void sock_prot_inuse_add(struct net *net, struct proto *prot, int val)
1338d466 2737{
d6d9ca0f 2738 __this_cpu_add(prot_inuse.val[prot->inuse_idx], val);
1338d466
PE		 2739}
2740EXPORT_SYMBOL_GPL(sock_prot_inuse_add);
2741
c29a0bc4 2742int sock_prot_inuse_get(struct net *net, struct proto *prot)
1338d466
PE		 2743{
2744 int cpu, idx = prot->inuse_idx;
2745 int res = 0;
2746
2747 for_each_possible_cpu(cpu)
2748 res += per_cpu(prot_inuse, cpu).val[idx];
2749
2750 return res >= 0 ? res : 0;
2751}
2752EXPORT_SYMBOL_GPL(sock_prot_inuse_get);
70ee1159 2753#endif
13ff3d6f
PE		 2754
2755static void assign_proto_idx(struct proto *prot)
2756{
2757 prot->inuse_idx = find_first_zero_bit(proto_inuse_idx, PROTO_INUSE_NR);
2758
2759 if (unlikely(prot->inuse_idx == PROTO_INUSE_NR - 1)) {
e005d193 2760 pr_err("PROTO_INUSE_NR exhausted\n");
13ff3d6f
PE		 2761 return;
2762 }
2763
2764 set_bit(prot->inuse_idx, proto_inuse_idx);
2765}
2766
2767static void release_proto_idx(struct proto *prot)
2768{
2769 if (prot->inuse_idx != PROTO_INUSE_NR - 1)
2770 clear_bit(prot->inuse_idx, proto_inuse_idx);
2771}
2772#else
2773static inline void assign_proto_idx(struct proto *prot)
2774{
2775}
2776
2777static inline void release_proto_idx(struct proto *prot)
2778{
2779}
2780#endif
2781
0159dfd3
ED		 2782static void req_prot_cleanup(struct request_sock_ops *rsk_prot)
2783{
2784 if (!rsk_prot)
2785 return;
2786 kfree(rsk_prot->slab_name);
2787 rsk_prot->slab_name = NULL;
adf78eda
JL		 2788 kmem_cache_destroy(rsk_prot->slab);
2789 rsk_prot->slab = NULL;
0159dfd3
ED		 2790}
2791
2792static int req_prot_init(const struct proto *prot)
2793{
2794 struct request_sock_ops *rsk_prot = prot->rsk_prot;
2795
2796 if (!rsk_prot)
2797 return 0;
2798
2799 rsk_prot->slab_name = kasprintf(GFP_KERNEL, "request_sock_%s",
2800 prot->name);
2801 if (!rsk_prot->slab_name)
2802 return -ENOMEM;
2803
2804 rsk_prot->slab = kmem_cache_create(rsk_prot->slab_name,
2805 rsk_prot->obj_size, 0,
e96f78ab 2806 prot->slab_flags, NULL);
0159dfd3
ED		 2807
2808 if (!rsk_prot->slab) {
2809 pr_crit("%s: Can't create request sock SLAB cache!\n",
2810 prot->name);
2811 return -ENOMEM;
2812 }
2813 return 0;
2814}
2815
b733c007
PE		 2816int proto_register(struct proto *prot, int alloc_slab)
2817{
1da177e4
LT		 2818 if (alloc_slab) {
2819 prot->slab = kmem_cache_create(prot->name, prot->obj_size, 0,
271b72c7
ED		 2820 SLAB_HWCACHE_ALIGN | prot->slab_flags,
2821 NULL);
1da177e4
LT		 2822
2823 if (prot->slab == NULL) {
e005d193
JP		 2824 pr_crit("%s: Can't create sock SLAB cache!\n",
2825 prot->name);
60e7663d 2826 goto out;
1da177e4 2827 }
2e6599cb 2828
0159dfd3
ED		 2829 if (req_prot_init(prot))
2830 goto out_free_request_sock_slab;
8feaf0c0 2831
6d6ee43e 2832 if (prot->twsk_prot != NULL) {
faf23422 2833 prot->twsk_prot->twsk_slab_name = kasprintf(GFP_KERNEL, "tw_sock_%s", prot->name);
8feaf0c0 2834
7e56b5d6 2835 if (prot->twsk_prot->twsk_slab_name == NULL)
8feaf0c0
ACM		 2836 goto out_free_request_sock_slab;
2837
6d6ee43e 2838 prot->twsk_prot->twsk_slab =
7e56b5d6 2839 kmem_cache_create(prot->twsk_prot->twsk_slab_name,
6d6ee43e 2840 prot->twsk_prot->twsk_obj_size,
3ab5aee7 2841 0,
52db70dc 2842 prot->slab_flags,
20c2df83 2843 NULL);
6d6ee43e 2844 if (prot->twsk_prot->twsk_slab == NULL)
8feaf0c0
ACM		 2845 goto out_free_timewait_sock_slab_name;
2846 }
1da177e4
LT		 2847 }
2848
36b77a52 2849 mutex_lock(&proto_list_mutex);
1da177e4 2850 list_add(&prot->node, &proto_list);
13ff3d6f 2851 assign_proto_idx(prot);
36b77a52 2852 mutex_unlock(&proto_list_mutex);
b733c007
PE		 2853 return 0;
2854
8feaf0c0 2855out_free_timewait_sock_slab_name:
7e56b5d6 2856 kfree(prot->twsk_prot->twsk_slab_name);
8feaf0c0 2857out_free_request_sock_slab:
0159dfd3
ED		 2858 req_prot_cleanup(prot->rsk_prot);
2859
2e6599cb
ACM		 2860 kmem_cache_destroy(prot->slab);
2861 prot->slab = NULL;
b733c007
PE		 2862out:
2863 return -ENOBUFS;
1da177e4 2864}
1da177e4
LT		 2865EXPORT_SYMBOL(proto_register);
2866
2867void proto_unregister(struct proto *prot)
2868{
36b77a52 2869 mutex_lock(&proto_list_mutex);
13ff3d6f 2870 release_proto_idx(prot);
0a3f4358 2871 list_del(&prot->node);
36b77a52 2872 mutex_unlock(&proto_list_mutex);
1da177e4 2873
adf78eda
JL		 2874 kmem_cache_destroy(prot->slab);
2875 prot->slab = NULL;
1da177e4 2876
0159dfd3 2877 req_prot_cleanup(prot->rsk_prot);
2e6599cb 2878
6d6ee43e 2879 if (prot->twsk_prot != NULL && prot->twsk_prot->twsk_slab != NULL) {
6d6ee43e 2880 kmem_cache_destroy(prot->twsk_prot->twsk_slab);
7e56b5d6 2881 kfree(prot->twsk_prot->twsk_slab_name);
6d6ee43e 2882 prot->twsk_prot->twsk_slab = NULL;
8feaf0c0 2883 }
1da177e4 2884}
1da177e4
LT		 2885EXPORT_SYMBOL(proto_unregister);
2886
2887#ifdef CONFIG_PROC_FS
1da177e4 2888static void *proto_seq_start(struct seq_file *seq, loff_t *pos)
36b77a52 2889 __acquires(proto_list_mutex)
1da177e4 2890{
36b77a52 2891 mutex_lock(&proto_list_mutex);
60f0438a 2892 return seq_list_start_head(&proto_list, *pos);
1da177e4
LT		 2893}
2894
2895static void *proto_seq_next(struct seq_file *seq, void *v, loff_t *pos)
2896{
60f0438a 2897 return seq_list_next(v, &proto_list, pos);
1da177e4
LT		 2898}
2899
2900static void proto_seq_stop(struct seq_file *seq, void *v)
36b77a52 2901 __releases(proto_list_mutex)
1da177e4 2902{
36b77a52 2903 mutex_unlock(&proto_list_mutex);
1da177e4
LT		 2904}
2905
2906static char proto_method_implemented(const void *method)
2907{
2908 return method == NULL ? 'n' : 'y';
2909}
180d8cd9
GC		 2910static long sock_prot_memory_allocated(struct proto *proto)
2911{
cb75a36c 2912 return proto->memory_allocated != NULL ? proto_memory_allocated(proto) : -1L;
180d8cd9
GC		 2913}
2914
2915static char *sock_prot_memory_pressure(struct proto *proto)
2916{
2917 return proto->memory_pressure != NULL ?
2918 proto_memory_pressure(proto) ? "yes" : "no" : "NI";
2919}
1da177e4
LT		 2920
2921static void proto_seq_printf(struct seq_file *seq, struct proto *proto)
2922{
180d8cd9 2923
8d987e5c 2924 seq_printf(seq, "%-9s %4u %6d %6ld %-3s %6u %-3s %-10s "
1da177e4
LT		 2925 "%2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c\n",
2926 proto->name,
2927 proto->obj_size,
14e943db 2928 sock_prot_inuse_get(seq_file_net(seq), proto),
180d8cd9
GC		 2929 sock_prot_memory_allocated(proto),
2930 sock_prot_memory_pressure(proto),
1da177e4
LT		 2931 proto->max_header,
2932 proto->slab == NULL ? "no" : "yes",
2933 module_name(proto->owner),
2934 proto_method_implemented(proto->close),
2935 proto_method_implemented(proto->connect),
2936 proto_method_implemented(proto->disconnect),
2937 proto_method_implemented(proto->accept),
2938 proto_method_implemented(proto->ioctl),
2939 proto_method_implemented(proto->init),
2940 proto_method_implemented(proto->destroy),
2941 proto_method_implemented(proto->shutdown),
2942 proto_method_implemented(proto->setsockopt),
2943 proto_method_implemented(proto->getsockopt),
2944 proto_method_implemented(proto->sendmsg),
2945 proto_method_implemented(proto->recvmsg),
2946 proto_method_implemented(proto->sendpage),
2947 proto_method_implemented(proto->bind),
2948 proto_method_implemented(proto->backlog_rcv),
2949 proto_method_implemented(proto->hash),
2950 proto_method_implemented(proto->unhash),
2951 proto_method_implemented(proto->get_port),
2952 proto_method_implemented(proto->enter_memory_pressure));
2953}
2954
2955static int proto_seq_show(struct seq_file *seq, void *v)
2956{
60f0438a 2957 if (v == &proto_list)
1da177e4
LT		 2958 seq_printf(seq, "%-9s %-4s %-8s %-6s %-5s %-7s %-4s %-10s %s",
2959 "protocol",
2960 "size",
2961 "sockets",
2962 "memory",
2963 "press",
2964 "maxhdr",
2965 "slab",
2966 "module",
2967 "cl co di ac io in de sh ss gs se re sp bi br ha uh gp em\n");
2968 else
60f0438a 2969 proto_seq_printf(seq, list_entry(v, struct proto, node));
1da177e4
LT		 2970 return 0;
2971}
2972
f690808e 2973static const struct seq_operations proto_seq_ops = {
1da177e4
LT		 2974 .start = proto_seq_start,
2975 .next = proto_seq_next,
2976 .stop = proto_seq_stop,
2977 .show = proto_seq_show,
2978};
2979
2980static int proto_seq_open(struct inode *inode, struct file *file)
2981{
14e943db
ED		 2982 return seq_open_net(inode, file, &proto_seq_ops,
2983 sizeof(struct seq_net_private));
1da177e4
LT		 2984}
2985
9a32144e 2986static const struct file_operations proto_seq_fops = {
1da177e4
LT		 2987 .owner = THIS_MODULE,
2988 .open = proto_seq_open,
2989 .read = seq_read,
2990 .llseek = seq_lseek,
14e943db
ED		 2991 .release = seq_release_net,
2992};
2993
2994static __net_init int proto_init_net(struct net *net)
2995{
d4beaa66 2996 if (!proc_create("protocols", S_IRUGO, net->proc_net, &proto_seq_fops))
14e943db
ED		 2997 return -ENOMEM;
2998
2999 return 0;
3000}
3001
3002static __net_exit void proto_exit_net(struct net *net)
3003{
ece31ffd 3004 remove_proc_entry("protocols", net->proc_net);
14e943db
ED		 3005}
3006
3007
3008static __net_initdata struct pernet_operations proto_net_ops = {
3009 .init = proto_init_net,
3010 .exit = proto_exit_net,
1da177e4
LT		 3011};
3012
3013static int __init proto_init(void)
3014{
14e943db 3015 return register_pernet_subsys(&proto_net_ops);
1da177e4
LT		 3016}
3017
3018subsys_initcall(proto_init);
3019
3020#endif /* PROC_FS */
Linux 6.x block layer and io_uring tree(s)