net/core/sock.c

   1 // SPDX-License-Identifier: GPL-2.0-or-later
   2 /*
   3  * INET         An implementation of the TCP/IP protocol suite for the LINUX
   4  *              operating system.  INET is implemented using the  BSD Socket
   5  *              interface as the means of communication with the user level.
   6  *
   7  *              Generic socket support routines. Memory allocators, socket lock/release
   8  *              handler for protocols to use and generic option handler.
   9  *
  10  * Authors:     Ross Biro
  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
  35  *                                      code. The ACK stuff can wait and needs major
  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 #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
  87
  88 #include <asm/unaligned.h>
  89 #include <linux/capability.h>
  90 #include <linux/errno.h>
  91 #include <linux/errqueue.h>
  92 #include <linux/types.h>
  93 #include <linux/socket.h>
  94 #include <linux/in.h>
  95 #include <linux/kernel.h>
  96 #include <linux/module.h>
  97 #include <linux/proc_fs.h>
  98 #include <linux/seq_file.h>
  99 #include <linux/sched.h>
 100 #include <linux/sched/mm.h>
 101 #include <linux/timer.h>
 102 #include <linux/string.h>
 103 #include <linux/sockios.h>
 104 #include <linux/net.h>
 105 #include <linux/mm.h>
 106 #include <linux/slab.h>
 107 #include <linux/interrupt.h>
 108 #include <linux/poll.h>
 109 #include <linux/tcp.h>
 110 #include <linux/init.h>
 111 #include <linux/highmem.h>
 112 #include <linux/user_namespace.h>
 113 #include <linux/static_key.h>
 114 #include <linux/memcontrol.h>
 115 #include <linux/prefetch.h>
 116 #include <linux/compat.h>
 117
 118 #include <linux/uaccess.h>
 119
 120 #include <linux/netdevice.h>
 121 #include <net/protocol.h>
 122 #include <linux/skbuff.h>
 123 #include <net/net_namespace.h>
 124 #include <net/request_sock.h>
 125 #include <net/sock.h>
 126 #include <linux/net_tstamp.h>
 127 #include <net/xfrm.h>
 128 #include <linux/ipsec.h>
 129 #include <net/cls_cgroup.h>
 130 #include <net/netprio_cgroup.h>
 131 #include <linux/sock_diag.h>
 132
 133 #include <linux/filter.h>
 134 #include <net/sock_reuseport.h>
 135 #include <net/bpf_sk_storage.h>
 136
 137 #include <trace/events/sock.h>
 138
 139 #include <net/tcp.h>
 140 #include <net/busy_poll.h>
 141
 142 #include <linux/ethtool.h>
 143
 144 #include "dev.h"
 145
 146 static DEFINE_MUTEX(proto_list_mutex);
 147 static LIST_HEAD(proto_list);
 148
 149 static void sock_def_write_space_wfree(struct sock *sk);
 150 static void sock_def_write_space(struct sock *sk);
 151
 152 /**
 153  * sk_ns_capable - General socket capability test
 154  * @sk: Socket to use a capability on or through
 155  * @user_ns: The user namespace of the capability to use
 156  * @cap: The capability to use
 157  *
 158  * Test to see if the opener of the socket had when the socket was
 159  * created and the current process has the capability @cap in the user
 160  * namespace @user_ns.
 161  */
 162 bool sk_ns_capable(const struct sock *sk,
 163                    struct user_namespace *user_ns, int cap)
 164 {
 165         return file_ns_capable(sk->sk_socket->file, user_ns, cap) &&
 166                 ns_capable(user_ns, cap);
 167 }
 168 EXPORT_SYMBOL(sk_ns_capable);
 169
 170 /**
 171  * sk_capable - Socket global capability test
 172  * @sk: Socket to use a capability on or through
 173  * @cap: The global capability to use
 174  *
 175  * Test to see if the opener of the socket had when the socket was
 176  * created and the current process has the capability @cap in all user
 177  * namespaces.
 178  */
 179 bool sk_capable(const struct sock *sk, int cap)
 180 {
 181         return sk_ns_capable(sk, &init_user_ns, cap);
 182 }
 183 EXPORT_SYMBOL(sk_capable);
 184
 185 /**
 186  * sk_net_capable - Network namespace socket capability test
 187  * @sk: Socket to use a capability on or through
 188  * @cap: The capability to use
 189  *
 190  * Test to see if the opener of the socket had when the socket was created
 191  * and the current process has the capability @cap over the network namespace
 192  * the socket is a member of.
 193  */
 194 bool sk_net_capable(const struct sock *sk, int cap)
 195 {
 196         return sk_ns_capable(sk, sock_net(sk)->user_ns, cap);
 197 }
 198 EXPORT_SYMBOL(sk_net_capable);
 199
 200 /*
 201  * Each address family might have different locking rules, so we have
 202  * one slock key per address family and separate keys for internal and
 203  * userspace sockets.
 204  */
 205 static struct lock_class_key af_family_keys[AF_MAX];
 206 static struct lock_class_key af_family_kern_keys[AF_MAX];
 207 static struct lock_class_key af_family_slock_keys[AF_MAX];
 208 static struct lock_class_key af_family_kern_slock_keys[AF_MAX];
 209
 210 /*
 211  * Make lock validator output more readable. (we pre-construct these
 212  * strings build-time, so that runtime initialization of socket
 213  * locks is fast):
 214  */
 215
 216 #define _sock_locks(x)                                            \
 217   x "AF_UNSPEC",        x "AF_UNIX"     ,       x "AF_INET"     , \
 218   x "AF_AX25"  ,        x "AF_IPX"      ,       x "AF_APPLETALK", \
 219   x "AF_NETROM",        x "AF_BRIDGE"   ,       x "AF_ATMPVC"   , \
 220   x "AF_X25"   ,        x "AF_INET6"    ,       x "AF_ROSE"     , \
 221   x "AF_DECnet",        x "AF_NETBEUI"  ,       x "AF_SECURITY" , \
 222   x "AF_KEY"   ,        x "AF_NETLINK"  ,       x "AF_PACKET"   , \
 223   x "AF_ASH"   ,        x "AF_ECONET"   ,       x "AF_ATMSVC"   , \
 224   x "AF_RDS"   ,        x "AF_SNA"      ,       x "AF_IRDA"     , \
 225   x "AF_PPPOX" ,        x "AF_WANPIPE"  ,       x "AF_LLC"      , \
 226   x "27"       ,        x "28"          ,       x "AF_CAN"      , \
 227   x "AF_TIPC"  ,        x "AF_BLUETOOTH",       x "IUCV"        , \
 228   x "AF_RXRPC" ,        x "AF_ISDN"     ,       x "AF_PHONET"   , \
 229   x "AF_IEEE802154",    x "AF_CAIF"     ,       x "AF_ALG"      , \
 230   x "AF_NFC"   ,        x "AF_VSOCK"    ,       x "AF_KCM"      , \
 231   x "AF_QIPCRTR",       x "AF_SMC"      ,       x "AF_XDP"      , \
 232   x "AF_MCTP"  , \
 233   x "AF_MAX"
 234
 235 static const char *const af_family_key_strings[AF_MAX+1] = {
 236         _sock_locks("sk_lock-")
 237 };
 238 static const char *const af_family_slock_key_strings[AF_MAX+1] = {
 239         _sock_locks("slock-")
 240 };
 241 static const char *const af_family_clock_key_strings[AF_MAX+1] = {
 242         _sock_locks("clock-")
 243 };
 244
 245 static const char *const af_family_kern_key_strings[AF_MAX+1] = {
 246         _sock_locks("k-sk_lock-")
 247 };
 248 static const char *const af_family_kern_slock_key_strings[AF_MAX+1] = {
 249         _sock_locks("k-slock-")
 250 };
 251 static const char *const af_family_kern_clock_key_strings[AF_MAX+1] = {
 252         _sock_locks("k-clock-")
 253 };
 254 static const char *const af_family_rlock_key_strings[AF_MAX+1] = {
 255         _sock_locks("rlock-")
 256 };
 257 static const char *const af_family_wlock_key_strings[AF_MAX+1] = {
 258         _sock_locks("wlock-")
 259 };
 260 static const char *const af_family_elock_key_strings[AF_MAX+1] = {
 261         _sock_locks("elock-")
 262 };
 263
 264 /*
 265  * sk_callback_lock and sk queues locking rules are per-address-family,
 266  * so split the lock classes by using a per-AF key:
 267  */
 268 static struct lock_class_key af_callback_keys[AF_MAX];
 269 static struct lock_class_key af_rlock_keys[AF_MAX];
 270 static struct lock_class_key af_wlock_keys[AF_MAX];
 271 static struct lock_class_key af_elock_keys[AF_MAX];
 272 static struct lock_class_key af_kern_callback_keys[AF_MAX];
 273
 274 /* Run time adjustable parameters. */
 275 __u32 sysctl_wmem_max __read_mostly = SK_WMEM_MAX;
 276 EXPORT_SYMBOL(sysctl_wmem_max);
 277 __u32 sysctl_rmem_max __read_mostly = SK_RMEM_MAX;
 278 EXPORT_SYMBOL(sysctl_rmem_max);
 279 __u32 sysctl_wmem_default __read_mostly = SK_WMEM_MAX;
 280 __u32 sysctl_rmem_default __read_mostly = SK_RMEM_MAX;
 281
 282 /* Maximal space eaten by iovec or ancillary data plus some space */
 283 int sysctl_optmem_max __read_mostly = sizeof(unsigned long)*(2*UIO_MAXIOV+512);
 284 EXPORT_SYMBOL(sysctl_optmem_max);
 285
 286 int sysctl_tstamp_allow_data __read_mostly = 1;
 287
 288 DEFINE_STATIC_KEY_FALSE(memalloc_socks_key);
 289 EXPORT_SYMBOL_GPL(memalloc_socks_key);
 290
 291 /**
 292  * sk_set_memalloc - sets %SOCK_MEMALLOC
 293  * @sk: socket to set it on
 294  *
 295  * Set %SOCK_MEMALLOC on a socket for access to emergency reserves.
 296  * It's the responsibility of the admin to adjust min_free_kbytes
 297  * to meet the requirements
 298  */
 299 void sk_set_memalloc(struct sock *sk)
 300 {
 301         sock_set_flag(sk, SOCK_MEMALLOC);
 302         sk->sk_allocation |= __GFP_MEMALLOC;
 303         static_branch_inc(&memalloc_socks_key);
 304 }
 305 EXPORT_SYMBOL_GPL(sk_set_memalloc);
 306
 307 void sk_clear_memalloc(struct sock *sk)
 308 {
 309         sock_reset_flag(sk, SOCK_MEMALLOC);
 310         sk->sk_allocation &= ~__GFP_MEMALLOC;
 311         static_branch_dec(&memalloc_socks_key);
 312
 313         /*
 314          * SOCK_MEMALLOC is allowed to ignore rmem limits to ensure forward
 315          * progress of swapping. SOCK_MEMALLOC may be cleared while
 316          * it has rmem allocations due to the last swapfile being deactivated
 317          * but there is a risk that the socket is unusable due to exceeding
 318          * the rmem limits. Reclaim the reserves and obey rmem limits again.
 319          */
 320         sk_mem_reclaim(sk);
 321 }
 322 EXPORT_SYMBOL_GPL(sk_clear_memalloc);
 323
 324 int __sk_backlog_rcv(struct sock *sk, struct sk_buff *skb)
 325 {
 326         int ret;
 327         unsigned int noreclaim_flag;
 328
 329         /* these should have been dropped before queueing */
 330         BUG_ON(!sock_flag(sk, SOCK_MEMALLOC));
 331
 332         noreclaim_flag = memalloc_noreclaim_save();
 333         ret = INDIRECT_CALL_INET(sk->sk_backlog_rcv,
 334                                  tcp_v6_do_rcv,
 335                                  tcp_v4_do_rcv,
 336                                  sk, skb);
 337         memalloc_noreclaim_restore(noreclaim_flag);
 338
 339         return ret;
 340 }
 341 EXPORT_SYMBOL(__sk_backlog_rcv);
 342
 343 void sk_error_report(struct sock *sk)
 344 {
 345         sk->sk_error_report(sk);
 346
 347         switch (sk->sk_family) {
 348         case AF_INET:
 349                 fallthrough;
 350         case AF_INET6:
 351                 trace_inet_sk_error_report(sk);
 352                 break;
 353         default:
 354                 break;
 355         }
 356 }
 357 EXPORT_SYMBOL(sk_error_report);
 358
 359 int sock_get_timeout(long timeo, void *optval, bool old_timeval)
 360 {
 361         struct __kernel_sock_timeval tv;
 362
 363         if (timeo == MAX_SCHEDULE_TIMEOUT) {
 364                 tv.tv_sec = 0;
 365                 tv.tv_usec = 0;
 366         } else {
 367                 tv.tv_sec = timeo / HZ;
 368                 tv.tv_usec = ((timeo % HZ) * USEC_PER_SEC) / HZ;
 369         }
 370
 371         if (old_timeval && in_compat_syscall() && !COMPAT_USE_64BIT_TIME) {
 372                 struct old_timeval32 tv32 = { tv.tv_sec, tv.tv_usec };
 373                 *(struct old_timeval32 *)optval = tv32;
 374                 return sizeof(tv32);
 375         }
 376
 377         if (old_timeval) {
 378                 struct __kernel_old_timeval old_tv;
 379                 old_tv.tv_sec = tv.tv_sec;
 380                 old_tv.tv_usec = tv.tv_usec;
 381                 *(struct __kernel_old_timeval *)optval = old_tv;
 382                 return sizeof(old_tv);
 383         }
 384
 385         *(struct __kernel_sock_timeval *)optval = tv;
 386         return sizeof(tv);
 387 }
 388 EXPORT_SYMBOL(sock_get_timeout);
 389
 390 int sock_copy_user_timeval(struct __kernel_sock_timeval *tv,
 391                            sockptr_t optval, int optlen, bool old_timeval)
 392 {
 393         if (old_timeval && in_compat_syscall() && !COMPAT_USE_64BIT_TIME) {
 394                 struct old_timeval32 tv32;
 395
 396                 if (optlen < sizeof(tv32))
 397                         return -EINVAL;
 398
 399                 if (copy_from_sockptr(&tv32, optval, sizeof(tv32)))
 400                         return -EFAULT;
 401                 tv->tv_sec = tv32.tv_sec;
 402                 tv->tv_usec = tv32.tv_usec;
 403         } else if (old_timeval) {
 404                 struct __kernel_old_timeval old_tv;
 405
 406                 if (optlen < sizeof(old_tv))
 407                         return -EINVAL;
 408                 if (copy_from_sockptr(&old_tv, optval, sizeof(old_tv)))
 409                         return -EFAULT;
 410                 tv->tv_sec = old_tv.tv_sec;
 411                 tv->tv_usec = old_tv.tv_usec;
 412         } else {
 413                 if (optlen < sizeof(*tv))
 414                         return -EINVAL;
 415                 if (copy_from_sockptr(tv, optval, sizeof(*tv)))
 416                         return -EFAULT;
 417         }
 418
 419         return 0;
 420 }
 421 EXPORT_SYMBOL(sock_copy_user_timeval);
 422
 423 static int sock_set_timeout(long *timeo_p, sockptr_t optval, int optlen,
 424                             bool old_timeval)
 425 {
 426         struct __kernel_sock_timeval tv;
 427         int err = sock_copy_user_timeval(&tv, optval, optlen, old_timeval);
 428
 429         if (err)
 430                 return err;
 431
 432         if (tv.tv_usec < 0 || tv.tv_usec >= USEC_PER_SEC)
 433                 return -EDOM;
 434
 435         if (tv.tv_sec < 0) {
 436                 static int warned __read_mostly;
 437
 438                 *timeo_p = 0;
 439                 if (warned < 10 && net_ratelimit()) {
 440                         warned++;
 441                         pr_info("%s: `%s' (pid %d) tries to set negative timeout\n",
 442                                 __func__, current->comm, task_pid_nr(current));
 443                 }
 444                 return 0;
 445         }
 446         *timeo_p = MAX_SCHEDULE_TIMEOUT;
 447         if (tv.tv_sec == 0 && tv.tv_usec == 0)
 448                 return 0;
 449         if (tv.tv_sec < (MAX_SCHEDULE_TIMEOUT / HZ - 1))
 450                 *timeo_p = tv.tv_sec * HZ + DIV_ROUND_UP((unsigned long)tv.tv_usec, USEC_PER_SEC / HZ);
 451         return 0;
 452 }
 453
 454 static bool sock_needs_netstamp(const struct sock *sk)
 455 {
 456         switch (sk->sk_family) {
 457         case AF_UNSPEC:
 458         case AF_UNIX:
 459                 return false;
 460         default:
 461                 return true;
 462         }
 463 }
 464
 465 static void sock_disable_timestamp(struct sock *sk, unsigned long flags)
 466 {
 467         if (sk->sk_flags & flags) {
 468                 sk->sk_flags &= ~flags;
 469                 if (sock_needs_netstamp(sk) &&
 470                     !(sk->sk_flags & SK_FLAGS_TIMESTAMP))
 471                         net_disable_timestamp();
 472         }
 473 }
 474
 475
 476 int __sock_queue_rcv_skb(struct sock *sk, struct sk_buff *skb)
 477 {
 478         unsigned long flags;
 479         struct sk_buff_head *list = &sk->sk_receive_queue;
 480
 481         if (atomic_read(&sk->sk_rmem_alloc) >= sk->sk_rcvbuf) {
 482                 atomic_inc(&sk->sk_drops);
 483                 trace_sock_rcvqueue_full(sk, skb);
 484                 return -ENOMEM;
 485         }
 486
 487         if (!sk_rmem_schedule(sk, skb, skb->truesize)) {
 488                 atomic_inc(&sk->sk_drops);
 489                 return -ENOBUFS;
 490         }
 491
 492         skb->dev = NULL;
 493         skb_set_owner_r(skb, sk);
 494
 495         /* we escape from rcu protected region, make sure we dont leak
 496          * a norefcounted dst
 497          */
 498         skb_dst_force(skb);
 499
 500         spin_lock_irqsave(&list->lock, flags);
 501         sock_skb_set_dropcount(sk, skb);
 502         __skb_queue_tail(list, skb);
 503         spin_unlock_irqrestore(&list->lock, flags);
 504
 505         if (!sock_flag(sk, SOCK_DEAD))
 506                 sk->sk_data_ready(sk);
 507         return 0;
 508 }
 509 EXPORT_SYMBOL(__sock_queue_rcv_skb);
 510
 511 int sock_queue_rcv_skb_reason(struct sock *sk, struct sk_buff *skb,
 512                               enum skb_drop_reason *reason)
 513 {
 514         enum skb_drop_reason drop_reason;
 515         int err;
 516
 517         err = sk_filter(sk, skb);
 518         if (err) {
 519                 drop_reason = SKB_DROP_REASON_SOCKET_FILTER;
 520                 goto out;
 521         }
 522         err = __sock_queue_rcv_skb(sk, skb);
 523         switch (err) {
 524         case -ENOMEM:
 525                 drop_reason = SKB_DROP_REASON_SOCKET_RCVBUFF;
 526                 break;
 527         case -ENOBUFS:
 528                 drop_reason = SKB_DROP_REASON_PROTO_MEM;
 529                 break;
 530         default:
 531                 drop_reason = SKB_NOT_DROPPED_YET;
 532                 break;
 533         }
 534 out:
 535         if (reason)
 536                 *reason = drop_reason;
 537         return err;
 538 }
 539 EXPORT_SYMBOL(sock_queue_rcv_skb_reason);
 540
 541 int __sk_receive_skb(struct sock *sk, struct sk_buff *skb,
 542                      const int nested, unsigned int trim_cap, bool refcounted)
 543 {
 544         int rc = NET_RX_SUCCESS;
 545
 546         if (sk_filter_trim_cap(sk, skb, trim_cap))
 547                 goto discard_and_relse;
 548
 549         skb->dev = NULL;
 550
 551         if (sk_rcvqueues_full(sk, sk->sk_rcvbuf)) {
 552                 atomic_inc(&sk->sk_drops);
 553                 goto discard_and_relse;
 554         }
 555         if (nested)
 556                 bh_lock_sock_nested(sk);
 557         else
 558                 bh_lock_sock(sk);
 559         if (!sock_owned_by_user(sk)) {
 560                 /*
 561                  * trylock + unlock semantics:
 562                  */
 563                 mutex_acquire(&sk->sk_lock.dep_map, 0, 1, _RET_IP_);
 564
 565                 rc = sk_backlog_rcv(sk, skb);
 566
 567                 mutex_release(&sk->sk_lock.dep_map, _RET_IP_);
 568         } else if (sk_add_backlog(sk, skb, READ_ONCE(sk->sk_rcvbuf))) {
 569                 bh_unlock_sock(sk);
 570                 atomic_inc(&sk->sk_drops);
 571                 goto discard_and_relse;
 572         }
 573
 574         bh_unlock_sock(sk);
 575 out:
 576         if (refcounted)
 577                 sock_put(sk);
 578         return rc;
 579 discard_and_relse:
 580         kfree_skb(skb);
 581         goto out;
 582 }
 583 EXPORT_SYMBOL(__sk_receive_skb);
 584
 585 INDIRECT_CALLABLE_DECLARE(struct dst_entry *ip6_dst_check(struct dst_entry *,
 586                                                           u32));
 587 INDIRECT_CALLABLE_DECLARE(struct dst_entry *ipv4_dst_check(struct dst_entry *,
 588                                                            u32));
 589 struct dst_entry *__sk_dst_check(struct sock *sk, u32 cookie)
 590 {
 591         struct dst_entry *dst = __sk_dst_get(sk);
 592
 593         if (dst && dst->obsolete &&
 594             INDIRECT_CALL_INET(dst->ops->check, ip6_dst_check, ipv4_dst_check,
 595                                dst, cookie) == NULL) {
 596                 sk_tx_queue_clear(sk);
 597                 sk->sk_dst_pending_confirm = 0;
 598                 RCU_INIT_POINTER(sk->sk_dst_cache, NULL);
 599                 dst_release(dst);
 600                 return NULL;
 601         }
 602
 603         return dst;
 604 }
 605 EXPORT_SYMBOL(__sk_dst_check);
 606
 607 struct dst_entry *sk_dst_check(struct sock *sk, u32 cookie)
 608 {
 609         struct dst_entry *dst = sk_dst_get(sk);
 610
 611         if (dst && dst->obsolete &&
 612             INDIRECT_CALL_INET(dst->ops->check, ip6_dst_check, ipv4_dst_check,
 613                                dst, cookie) == NULL) {
 614                 sk_dst_reset(sk);
 615                 dst_release(dst);
 616                 return NULL;
 617         }
 618
 619         return dst;
 620 }
 621 EXPORT_SYMBOL(sk_dst_check);
 622
 623 static int sock_bindtoindex_locked(struct sock *sk, int ifindex)
 624 {
 625         int ret = -ENOPROTOOPT;
 626 #ifdef CONFIG_NETDEVICES
 627         struct net *net = sock_net(sk);
 628
 629         /* Sorry... */
 630         ret = -EPERM;
 631         if (sk->sk_bound_dev_if && !ns_capable(net->user_ns, CAP_NET_RAW))
 632                 goto out;
 633
 634         ret = -EINVAL;
 635         if (ifindex < 0)
 636                 goto out;
 637
 638         /* Paired with all READ_ONCE() done locklessly. */
 639         WRITE_ONCE(sk->sk_bound_dev_if, ifindex);
 640
 641         if (sk->sk_prot->rehash)
 642                 sk->sk_prot->rehash(sk);
 643         sk_dst_reset(sk);
 644
 645         ret = 0;
 646
 647 out:
 648 #endif
 649
 650         return ret;
 651 }
 652
 653 int sock_bindtoindex(struct sock *sk, int ifindex, bool lock_sk)
 654 {
 655         int ret;
 656
 657         if (lock_sk)
 658                 lock_sock(sk);
 659         ret = sock_bindtoindex_locked(sk, ifindex);
 660         if (lock_sk)
 661                 release_sock(sk);
 662
 663         return ret;
 664 }
 665 EXPORT_SYMBOL(sock_bindtoindex);
 666
 667 static int sock_setbindtodevice(struct sock *sk, sockptr_t optval, int optlen)
 668 {
 669         int ret = -ENOPROTOOPT;
 670 #ifdef CONFIG_NETDEVICES
 671         struct net *net = sock_net(sk);
 672         char devname[IFNAMSIZ];
 673         int index;
 674
 675         ret = -EINVAL;
 676         if (optlen < 0)
 677                 goto out;
 678
 679         /* Bind this socket to a particular device like "eth0",
 680          * as specified in the passed interface name. If the
 681          * name is "" or the option length is zero the socket
 682          * is not bound.
 683          */
 684         if (optlen > IFNAMSIZ - 1)
 685                 optlen = IFNAMSIZ - 1;
 686         memset(devname, 0, sizeof(devname));
 687
 688         ret = -EFAULT;
 689         if (copy_from_sockptr(devname, optval, optlen))
 690                 goto out;
 691
 692         index = 0;
 693         if (devname[0] != '\0') {
 694                 struct net_device *dev;
 695
 696                 rcu_read_lock();
 697                 dev = dev_get_by_name_rcu(net, devname);
 698                 if (dev)
 699                         index = dev->ifindex;
 700                 rcu_read_unlock();
 701                 ret = -ENODEV;
 702                 if (!dev)
 703                         goto out;
 704         }
 705
 706         sockopt_lock_sock(sk);
 707         ret = sock_bindtoindex_locked(sk, index);
 708         sockopt_release_sock(sk);
 709 out:
 710 #endif
 711
 712         return ret;
 713 }
 714
 715 static int sock_getbindtodevice(struct sock *sk, sockptr_t optval,
 716                                 sockptr_t optlen, int len)
 717 {
 718         int ret = -ENOPROTOOPT;
 719 #ifdef CONFIG_NETDEVICES
 720         int bound_dev_if = READ_ONCE(sk->sk_bound_dev_if);
 721         struct net *net = sock_net(sk);
 722         char devname[IFNAMSIZ];
 723
 724         if (bound_dev_if == 0) {
 725                 len = 0;
 726                 goto zero;
 727         }
 728
 729         ret = -EINVAL;
 730         if (len < IFNAMSIZ)
 731                 goto out;
 732
 733         ret = netdev_get_name(net, devname, bound_dev_if);
 734         if (ret)
 735                 goto out;
 736
 737         len = strlen(devname) + 1;
 738
 739         ret = -EFAULT;
 740         if (copy_to_sockptr(optval, devname, len))
 741                 goto out;
 742
 743 zero:
 744         ret = -EFAULT;
 745         if (copy_to_sockptr(optlen, &len, sizeof(int)))
 746                 goto out;
 747
 748         ret = 0;
 749
 750 out:
 751 #endif
 752
 753         return ret;
 754 }
 755
 756 bool sk_mc_loop(struct sock *sk)
 757 {
 758         if (dev_recursion_level())
 759                 return false;
 760         if (!sk)
 761                 return true;
 762         switch (sk->sk_family) {
 763         case AF_INET:
 764                 return inet_sk(sk)->mc_loop;
 765 #if IS_ENABLED(CONFIG_IPV6)
 766         case AF_INET6:
 767                 return inet6_sk(sk)->mc_loop;
 768 #endif
 769         }
 770         WARN_ON_ONCE(1);
 771         return true;
 772 }
 773 EXPORT_SYMBOL(sk_mc_loop);
 774
 775 void sock_set_reuseaddr(struct sock *sk)
 776 {
 777         lock_sock(sk);
 778         sk->sk_reuse = SK_CAN_REUSE;
 779         release_sock(sk);
 780 }
 781 EXPORT_SYMBOL(sock_set_reuseaddr);
 782
 783 void sock_set_reuseport(struct sock *sk)
 784 {
 785         lock_sock(sk);
 786         sk->sk_reuseport = true;
 787         release_sock(sk);
 788 }
 789 EXPORT_SYMBOL(sock_set_reuseport);
 790
 791 void sock_no_linger(struct sock *sk)
 792 {
 793         lock_sock(sk);
 794         sk->sk_lingertime = 0;
 795         sock_set_flag(sk, SOCK_LINGER);
 796         release_sock(sk);
 797 }
 798 EXPORT_SYMBOL(sock_no_linger);
 799
 800 void sock_set_priority(struct sock *sk, u32 priority)
 801 {
 802         lock_sock(sk);
 803         sk->sk_priority = priority;
 804         release_sock(sk);
 805 }
 806 EXPORT_SYMBOL(sock_set_priority);
 807
 808 void sock_set_sndtimeo(struct sock *sk, s64 secs)
 809 {
 810         lock_sock(sk);
 811         if (secs && secs < MAX_SCHEDULE_TIMEOUT / HZ - 1)
 812                 sk->sk_sndtimeo = secs * HZ;
 813         else
 814                 sk->sk_sndtimeo = MAX_SCHEDULE_TIMEOUT;
 815         release_sock(sk);
 816 }
 817 EXPORT_SYMBOL(sock_set_sndtimeo);
 818
 819 static void __sock_set_timestamps(struct sock *sk, bool val, bool new, bool ns)
 820 {
 821         if (val)  {
 822                 sock_valbool_flag(sk, SOCK_TSTAMP_NEW, new);
 823                 sock_valbool_flag(sk, SOCK_RCVTSTAMPNS, ns);
 824                 sock_set_flag(sk, SOCK_RCVTSTAMP);
 825                 sock_enable_timestamp(sk, SOCK_TIMESTAMP);
 826         } else {
 827                 sock_reset_flag(sk, SOCK_RCVTSTAMP);
 828                 sock_reset_flag(sk, SOCK_RCVTSTAMPNS);
 829         }
 830 }
 831
 832 void sock_enable_timestamps(struct sock *sk)
 833 {
 834         lock_sock(sk);
 835         __sock_set_timestamps(sk, true, false, true);
 836         release_sock(sk);
 837 }
 838 EXPORT_SYMBOL(sock_enable_timestamps);
 839
 840 void sock_set_timestamp(struct sock *sk, int optname, bool valbool)
 841 {
 842         switch (optname) {
 843         case SO_TIMESTAMP_OLD:
 844                 __sock_set_timestamps(sk, valbool, false, false);
 845                 break;
 846         case SO_TIMESTAMP_NEW:
 847                 __sock_set_timestamps(sk, valbool, true, false);
 848                 break;
 849         case SO_TIMESTAMPNS_OLD:
 850                 __sock_set_timestamps(sk, valbool, false, true);
 851                 break;
 852         case SO_TIMESTAMPNS_NEW:
 853                 __sock_set_timestamps(sk, valbool, true, true);
 854                 break;
 855         }
 856 }
 857
 858 static int sock_timestamping_bind_phc(struct sock *sk, int phc_index)
 859 {
 860         struct net *net = sock_net(sk);
 861         struct net_device *dev = NULL;
 862         bool match = false;
 863         int *vclock_index;
 864         int i, num;
 865
 866         if (sk->sk_bound_dev_if)
 867                 dev = dev_get_by_index(net, sk->sk_bound_dev_if);
 868
 869         if (!dev) {
 870                 pr_err("%s: sock not bind to device\n", __func__);
 871                 return -EOPNOTSUPP;
 872         }
 873
 874         num = ethtool_get_phc_vclocks(dev, &vclock_index);
 875         dev_put(dev);
 876
 877         for (i = 0; i < num; i++) {
 878                 if (*(vclock_index + i) == phc_index) {
 879                         match = true;
 880                         break;
 881                 }
 882         }
 883
 884         if (num > 0)
 885                 kfree(vclock_index);
 886
 887         if (!match)
 888                 return -EINVAL;
 889
 890         sk->sk_bind_phc = phc_index;
 891
 892         return 0;
 893 }
 894
 895 int sock_set_timestamping(struct sock *sk, int optname,
 896                           struct so_timestamping timestamping)
 897 {
 898         int val = timestamping.flags;
 899         int ret;
 900
 901         if (val & ~SOF_TIMESTAMPING_MASK)
 902                 return -EINVAL;
 903
 904         if (val & SOF_TIMESTAMPING_OPT_ID &&
 905             !(sk->sk_tsflags & SOF_TIMESTAMPING_OPT_ID)) {
 906                 if (sk_is_tcp(sk)) {
 907                         if ((1 << sk->sk_state) &
 908                             (TCPF_CLOSE | TCPF_LISTEN))
 909                                 return -EINVAL;
 910                         atomic_set(&sk->sk_tskey, tcp_sk(sk)->snd_una);
 911                 } else {
 912                         atomic_set(&sk->sk_tskey, 0);
 913                 }
 914         }
 915
 916         if (val & SOF_TIMESTAMPING_OPT_STATS &&
 917             !(val & SOF_TIMESTAMPING_OPT_TSONLY))
 918                 return -EINVAL;
 919
 920         if (val & SOF_TIMESTAMPING_BIND_PHC) {
 921                 ret = sock_timestamping_bind_phc(sk, timestamping.bind_phc);
 922                 if (ret)
 923                         return ret;
 924         }
 925
 926         sk->sk_tsflags = val;
 927         sock_valbool_flag(sk, SOCK_TSTAMP_NEW, optname == SO_TIMESTAMPING_NEW);
 928
 929         if (val & SOF_TIMESTAMPING_RX_SOFTWARE)
 930                 sock_enable_timestamp(sk,
 931                                       SOCK_TIMESTAMPING_RX_SOFTWARE);
 932         else
 933                 sock_disable_timestamp(sk,
 934                                        (1UL << SOCK_TIMESTAMPING_RX_SOFTWARE));
 935         return 0;
 936 }
 937
 938 void sock_set_keepalive(struct sock *sk)
 939 {
 940         lock_sock(sk);
 941         if (sk->sk_prot->keepalive)
 942                 sk->sk_prot->keepalive(sk, true);
 943         sock_valbool_flag(sk, SOCK_KEEPOPEN, true);
 944         release_sock(sk);
 945 }
 946 EXPORT_SYMBOL(sock_set_keepalive);
 947
 948 static void __sock_set_rcvbuf(struct sock *sk, int val)
 949 {
 950         /* Ensure val * 2 fits into an int, to prevent max_t() from treating it
 951          * as a negative value.
 952          */
 953         val = min_t(int, val, INT_MAX / 2);
 954         sk->sk_userlocks |= SOCK_RCVBUF_LOCK;
 955
 956         /* We double it on the way in to account for "struct sk_buff" etc.
 957          * overhead.   Applications assume that the SO_RCVBUF setting they make
 958          * will allow that much actual data to be received on that socket.
 959          *
 960          * Applications are unaware that "struct sk_buff" and other overheads
 961          * allocate from the receive buffer during socket buffer allocation.
 962          *
 963          * And after considering the possible alternatives, returning the value
 964          * we actually used in getsockopt is the most desirable behavior.
 965          */
 966         WRITE_ONCE(sk->sk_rcvbuf, max_t(int, val * 2, SOCK_MIN_RCVBUF));
 967 }
 968
 969 void sock_set_rcvbuf(struct sock *sk, int val)
 970 {
 971         lock_sock(sk);
 972         __sock_set_rcvbuf(sk, val);
 973         release_sock(sk);
 974 }
 975 EXPORT_SYMBOL(sock_set_rcvbuf);
 976
 977 static void __sock_set_mark(struct sock *sk, u32 val)
 978 {
 979         if (val != sk->sk_mark) {
 980                 sk->sk_mark = val;
 981                 sk_dst_reset(sk);
 982         }
 983 }
 984
 985 void sock_set_mark(struct sock *sk, u32 val)
 986 {
 987         lock_sock(sk);
 988         __sock_set_mark(sk, val);
 989         release_sock(sk);
 990 }
 991 EXPORT_SYMBOL(sock_set_mark);
 992
 993 static void sock_release_reserved_memory(struct sock *sk, int bytes)
 994 {
 995         /* Round down bytes to multiple of pages */
 996         bytes = round_down(bytes, PAGE_SIZE);
 997
 998         WARN_ON(bytes > sk->sk_reserved_mem);
 999         sk->sk_reserved_mem -= bytes;
1000         sk_mem_reclaim(sk);
1001 }
1002
1003 static int sock_reserve_memory(struct sock *sk, int bytes)
1004 {
1005         long allocated;
1006         bool charged;
1007         int pages;
1008
1009         if (!mem_cgroup_sockets_enabled || !sk->sk_memcg || !sk_has_account(sk))
1010                 return -EOPNOTSUPP;
1011
1012         if (!bytes)
1013                 return 0;
1014
1015         pages = sk_mem_pages(bytes);
1016
1017         /* pre-charge to memcg */
1018         charged = mem_cgroup_charge_skmem(sk->sk_memcg, pages,
1019                                           GFP_KERNEL | __GFP_RETRY_MAYFAIL);
1020         if (!charged)
1021                 return -ENOMEM;
1022
1023         /* pre-charge to forward_alloc */
1024         sk_memory_allocated_add(sk, pages);
1025         allocated = sk_memory_allocated(sk);
1026         /* If the system goes into memory pressure with this
1027          * precharge, give up and return error.
1028          */
1029         if (allocated > sk_prot_mem_limits(sk, 1)) {
1030                 sk_memory_allocated_sub(sk, pages);
1031                 mem_cgroup_uncharge_skmem(sk->sk_memcg, pages);
1032                 return -ENOMEM;
1033         }
1034         sk->sk_forward_alloc += pages << PAGE_SHIFT;
1035
1036         sk->sk_reserved_mem += pages << PAGE_SHIFT;
1037
1038         return 0;
1039 }
1040
1041 void sockopt_lock_sock(struct sock *sk)
1042 {
1043         /* When current->bpf_ctx is set, the setsockopt is called from
1044          * a bpf prog.  bpf has ensured the sk lock has been
1045          * acquired before calling setsockopt().
1046          */
1047         if (has_current_bpf_ctx())
1048                 return;
1049
1050         lock_sock(sk);
1051 }
1052 EXPORT_SYMBOL(sockopt_lock_sock);
1053
1054 void sockopt_release_sock(struct sock *sk)
1055 {
1056         if (has_current_bpf_ctx())
1057                 return;
1058
1059         release_sock(sk);
1060 }
1061 EXPORT_SYMBOL(sockopt_release_sock);
1062
1063 bool sockopt_ns_capable(struct user_namespace *ns, int cap)
1064 {
1065         return has_current_bpf_ctx() || ns_capable(ns, cap);
1066 }
1067 EXPORT_SYMBOL(sockopt_ns_capable);
1068
1069 bool sockopt_capable(int cap)
1070 {
1071         return has_current_bpf_ctx() || capable(cap);
1072 }
1073 EXPORT_SYMBOL(sockopt_capable);
1074
1075 /*
1076  *      This is meant for all protocols to use and covers goings on
1077  *      at the socket level. Everything here is generic.
1078  */
1079
1080 int sk_setsockopt(struct sock *sk, int level, int optname,
1081                   sockptr_t optval, unsigned int optlen)
1082 {
1083         struct so_timestamping timestamping;
1084         struct socket *sock = sk->sk_socket;
1085         struct sock_txtime sk_txtime;
1086         int val;
1087         int valbool;
1088         struct linger ling;
1089         int ret = 0;
1090
1091         /*
1092          *      Options without arguments
1093          */
1094
1095         if (optname == SO_BINDTODEVICE)
1096                 return sock_setbindtodevice(sk, optval, optlen);
1097
1098         if (optlen < sizeof(int))
1099                 return -EINVAL;
1100
1101         if (copy_from_sockptr(&val, optval, sizeof(val)))
1102                 return -EFAULT;
1103
1104         valbool = val ? 1 : 0;
1105
1106         sockopt_lock_sock(sk);
1107
1108         switch (optname) {
1109         case SO_DEBUG:
1110                 if (val && !sockopt_capable(CAP_NET_ADMIN))
1111                         ret = -EACCES;
1112                 else
1113                         sock_valbool_flag(sk, SOCK_DBG, valbool);
1114                 break;
1115         case SO_REUSEADDR:
1116                 sk->sk_reuse = (valbool ? SK_CAN_REUSE : SK_NO_REUSE);
1117                 break;
1118         case SO_REUSEPORT:
1119                 sk->sk_reuseport = valbool;
1120                 break;
1121         case SO_TYPE:
1122         case SO_PROTOCOL:
1123         case SO_DOMAIN:
1124         case SO_ERROR:
1125                 ret = -ENOPROTOOPT;
1126                 break;
1127         case SO_DONTROUTE:
1128                 sock_valbool_flag(sk, SOCK_LOCALROUTE, valbool);
1129                 sk_dst_reset(sk);
1130                 break;
1131         case SO_BROADCAST:
1132                 sock_valbool_flag(sk, SOCK_BROADCAST, valbool);
1133                 break;
1134         case SO_SNDBUF:
1135                 /* Don't error on this BSD doesn't and if you think
1136                  * about it this is right. Otherwise apps have to
1137                  * play 'guess the biggest size' games. RCVBUF/SNDBUF
1138                  * are treated in BSD as hints
1139                  */
1140                 val = min_t(u32, val, READ_ONCE(sysctl_wmem_max));
1141 set_sndbuf:
1142                 /* Ensure val * 2 fits into an int, to prevent max_t()
1143                  * from treating it as a negative value.
1144                  */
1145                 val = min_t(int, val, INT_MAX / 2);
1146                 sk->sk_userlocks |= SOCK_SNDBUF_LOCK;
1147                 WRITE_ONCE(sk->sk_sndbuf,
1148                            max_t(int, val * 2, SOCK_MIN_SNDBUF));
1149                 /* Wake up sending tasks if we upped the value. */
1150                 sk->sk_write_space(sk);
1151                 break;
1152
1153         case SO_SNDBUFFORCE:
1154                 if (!sockopt_capable(CAP_NET_ADMIN)) {
1155                         ret = -EPERM;
1156                         break;
1157                 }
1158
1159                 /* No negative values (to prevent underflow, as val will be
1160                  * multiplied by 2).
1161                  */
1162                 if (val < 0)
1163                         val = 0;
1164                 goto set_sndbuf;
1165
1166         case SO_RCVBUF:
1167                 /* Don't error on this BSD doesn't and if you think
1168                  * about it this is right. Otherwise apps have to
1169                  * play 'guess the biggest size' games. RCVBUF/SNDBUF
1170                  * are treated in BSD as hints
1171                  */
1172                 __sock_set_rcvbuf(sk, min_t(u32, val, READ_ONCE(sysctl_rmem_max)));
1173                 break;
1174
1175         case SO_RCVBUFFORCE:
1176                 if (!sockopt_capable(CAP_NET_ADMIN)) {
1177                         ret = -EPERM;
1178                         break;
1179                 }
1180
1181                 /* No negative values (to prevent underflow, as val will be
1182                  * multiplied by 2).
1183                  */
1184                 __sock_set_rcvbuf(sk, max(val, 0));
1185                 break;
1186
1187         case SO_KEEPALIVE:
1188                 if (sk->sk_prot->keepalive)
1189                         sk->sk_prot->keepalive(sk, valbool);
1190                 sock_valbool_flag(sk, SOCK_KEEPOPEN, valbool);
1191                 break;
1192
1193         case SO_OOBINLINE:
1194                 sock_valbool_flag(sk, SOCK_URGINLINE, valbool);
1195                 break;
1196
1197         case SO_NO_CHECK:
1198                 sk->sk_no_check_tx = valbool;
1199                 break;
1200
1201         case SO_PRIORITY:
1202                 if ((val >= 0 && val <= 6) ||
1203                     sockopt_ns_capable(sock_net(sk)->user_ns, CAP_NET_RAW) ||
1204                     sockopt_ns_capable(sock_net(sk)->user_ns, CAP_NET_ADMIN))
1205                         sk->sk_priority = val;
1206                 else
1207                         ret = -EPERM;
1208                 break;
1209
1210         case SO_LINGER:
1211                 if (optlen < sizeof(ling)) {
1212                         ret = -EINVAL;  /* 1003.1g */
1213                         break;
1214                 }
1215                 if (copy_from_sockptr(&ling, optval, sizeof(ling))) {
1216                         ret = -EFAULT;
1217                         break;
1218                 }
1219                 if (!ling.l_onoff)
1220                         sock_reset_flag(sk, SOCK_LINGER);
1221                 else {
1222 #if (BITS_PER_LONG == 32)
1223                         if ((unsigned int)ling.l_linger >= MAX_SCHEDULE_TIMEOUT/HZ)
1224                                 sk->sk_lingertime = MAX_SCHEDULE_TIMEOUT;
1225                         else
1226 #endif
1227                                 sk->sk_lingertime = (unsigned int)ling.l_linger * HZ;
1228                         sock_set_flag(sk, SOCK_LINGER);
1229                 }
1230                 break;
1231
1232         case SO_BSDCOMPAT:
1233                 break;
1234
1235         case SO_PASSCRED:
1236                 if (valbool)
1237                         set_bit(SOCK_PASSCRED, &sock->flags);
1238                 else
1239                         clear_bit(SOCK_PASSCRED, &sock->flags);
1240                 break;
1241
1242         case SO_TIMESTAMP_OLD:
1243         case SO_TIMESTAMP_NEW:
1244         case SO_TIMESTAMPNS_OLD:
1245         case SO_TIMESTAMPNS_NEW:
1246                 sock_set_timestamp(sk, optname, valbool);
1247                 break;
1248
1249         case SO_TIMESTAMPING_NEW:
1250         case SO_TIMESTAMPING_OLD:
1251                 if (optlen == sizeof(timestamping)) {
1252                         if (copy_from_sockptr(&timestamping, optval,
1253                                               sizeof(timestamping))) {
1254                                 ret = -EFAULT;
1255                                 break;
1256                         }
1257                 } else {
1258                         memset(&timestamping, 0, sizeof(timestamping));
1259                         timestamping.flags = val;
1260                 }
1261                 ret = sock_set_timestamping(sk, optname, timestamping);
1262                 break;
1263
1264         case SO_RCVLOWAT:
1265                 if (val < 0)
1266                         val = INT_MAX;
1267                 if (sock && sock->ops->set_rcvlowat)
1268                         ret = sock->ops->set_rcvlowat(sk, val);
1269                 else
1270                         WRITE_ONCE(sk->sk_rcvlowat, val ? : 1);
1271                 break;
1272
1273         case SO_RCVTIMEO_OLD:
1274         case SO_RCVTIMEO_NEW:
1275                 ret = sock_set_timeout(&sk->sk_rcvtimeo, optval,
1276                                        optlen, optname == SO_RCVTIMEO_OLD);
1277                 break;
1278
1279         case SO_SNDTIMEO_OLD:
1280         case SO_SNDTIMEO_NEW:
1281                 ret = sock_set_timeout(&sk->sk_sndtimeo, optval,
1282                                        optlen, optname == SO_SNDTIMEO_OLD);
1283                 break;
1284
1285         case SO_ATTACH_FILTER: {
1286                 struct sock_fprog fprog;
1287
1288                 ret = copy_bpf_fprog_from_user(&fprog, optval, optlen);
1289                 if (!ret)
1290                         ret = sk_attach_filter(&fprog, sk);
1291                 break;
1292         }
1293         case SO_ATTACH_BPF:
1294                 ret = -EINVAL;
1295                 if (optlen == sizeof(u32)) {
1296                         u32 ufd;
1297
1298                         ret = -EFAULT;
1299                         if (copy_from_sockptr(&ufd, optval, sizeof(ufd)))
1300                                 break;
1301
1302                         ret = sk_attach_bpf(ufd, sk);
1303                 }
1304                 break;
1305
1306         case SO_ATTACH_REUSEPORT_CBPF: {
1307                 struct sock_fprog fprog;
1308
1309                 ret = copy_bpf_fprog_from_user(&fprog, optval, optlen);
1310                 if (!ret)
1311                         ret = sk_reuseport_attach_filter(&fprog, sk);
1312                 break;
1313         }
1314         case SO_ATTACH_REUSEPORT_EBPF:
1315                 ret = -EINVAL;
1316                 if (optlen == sizeof(u32)) {
1317                         u32 ufd;
1318
1319                         ret = -EFAULT;
1320                         if (copy_from_sockptr(&ufd, optval, sizeof(ufd)))
1321                                 break;
1322
1323                         ret = sk_reuseport_attach_bpf(ufd, sk);
1324                 }
1325                 break;
1326
1327         case SO_DETACH_REUSEPORT_BPF:
1328                 ret = reuseport_detach_prog(sk);
1329                 break;
1330
1331         case SO_DETACH_FILTER:
1332                 ret = sk_detach_filter(sk);
1333                 break;
1334
1335         case SO_LOCK_FILTER:
1336                 if (sock_flag(sk, SOCK_FILTER_LOCKED) && !valbool)
1337                         ret = -EPERM;
1338                 else
1339                         sock_valbool_flag(sk, SOCK_FILTER_LOCKED, valbool);
1340                 break;
1341
1342         case SO_PASSSEC:
1343                 if (valbool)
1344                         set_bit(SOCK_PASSSEC, &sock->flags);
1345                 else
1346                         clear_bit(SOCK_PASSSEC, &sock->flags);
1347                 break;
1348         case SO_MARK:
1349                 if (!sockopt_ns_capable(sock_net(sk)->user_ns, CAP_NET_RAW) &&
1350                     !sockopt_ns_capable(sock_net(sk)->user_ns, CAP_NET_ADMIN)) {
1351                         ret = -EPERM;
1352                         break;
1353                 }
1354
1355                 __sock_set_mark(sk, val);
1356                 break;
1357         case SO_RCVMARK:
1358                 if (!sockopt_ns_capable(sock_net(sk)->user_ns, CAP_NET_RAW) &&
1359                     !sockopt_ns_capable(sock_net(sk)->user_ns, CAP_NET_ADMIN)) {
1360                         ret = -EPERM;
1361                         break;
1362                 }
1363
1364                 sock_valbool_flag(sk, SOCK_RCVMARK, valbool);
1365                 break;
1366
1367         case SO_RXQ_OVFL:
1368                 sock_valbool_flag(sk, SOCK_RXQ_OVFL, valbool);
1369                 break;
1370
1371         case SO_WIFI_STATUS:
1372                 sock_valbool_flag(sk, SOCK_WIFI_STATUS, valbool);
1373                 break;
1374
1375         case SO_PEEK_OFF:
1376                 if (sock->ops->set_peek_off)
1377                         ret = sock->ops->set_peek_off(sk, val);
1378                 else
1379                         ret = -EOPNOTSUPP;
1380                 break;
1381
1382         case SO_NOFCS:
1383                 sock_valbool_flag(sk, SOCK_NOFCS, valbool);
1384                 break;
1385
1386         case SO_SELECT_ERR_QUEUE:
1387                 sock_valbool_flag(sk, SOCK_SELECT_ERR_QUEUE, valbool);
1388                 break;
1389
1390 #ifdef CONFIG_NET_RX_BUSY_POLL
1391         case SO_BUSY_POLL:
1392                 /* allow unprivileged users to decrease the value */
1393                 if ((val > sk->sk_ll_usec) && !sockopt_capable(CAP_NET_ADMIN))
1394                         ret = -EPERM;
1395                 else {
1396                         if (val < 0)
1397                                 ret = -EINVAL;
1398                         else
1399                                 WRITE_ONCE(sk->sk_ll_usec, val);
1400                 }
1401                 break;
1402         case SO_PREFER_BUSY_POLL:
1403                 if (valbool && !sockopt_capable(CAP_NET_ADMIN))
1404                         ret = -EPERM;
1405                 else
1406                         WRITE_ONCE(sk->sk_prefer_busy_poll, valbool);
1407                 break;
1408         case SO_BUSY_POLL_BUDGET:
1409                 if (val > READ_ONCE(sk->sk_busy_poll_budget) && !sockopt_capable(CAP_NET_ADMIN)) {
1410                         ret = -EPERM;
1411                 } else {
1412                         if (val < 0 || val > U16_MAX)
1413                                 ret = -EINVAL;
1414                         else
1415                                 WRITE_ONCE(sk->sk_busy_poll_budget, val);
1416                 }
1417                 break;
1418 #endif
1419
1420         case SO_MAX_PACING_RATE:
1421                 {
1422                 unsigned long ulval = (val == ~0U) ? ~0UL : (unsigned int)val;
1423
1424                 if (sizeof(ulval) != sizeof(val) &&
1425                     optlen >= sizeof(ulval) &&
1426                     copy_from_sockptr(&ulval, optval, sizeof(ulval))) {
1427                         ret = -EFAULT;
1428                         break;
1429                 }
1430                 if (ulval != ~0UL)
1431                         cmpxchg(&sk->sk_pacing_status,
1432                                 SK_PACING_NONE,
1433                                 SK_PACING_NEEDED);
1434                 sk->sk_max_pacing_rate = ulval;
1435                 sk->sk_pacing_rate = min(sk->sk_pacing_rate, ulval);
1436                 break;
1437                 }
1438         case SO_INCOMING_CPU:
1439                 reuseport_update_incoming_cpu(sk, val);
1440                 break;
1441
1442         case SO_CNX_ADVICE:
1443                 if (val == 1)
1444                         dst_negative_advice(sk);
1445                 break;
1446
1447         case SO_ZEROCOPY:
1448                 if (sk->sk_family == PF_INET || sk->sk_family == PF_INET6) {
1449                         if (!(sk_is_tcp(sk) ||
1450                               (sk->sk_type == SOCK_DGRAM &&
1451                                sk->sk_protocol == IPPROTO_UDP)))
1452                                 ret = -EOPNOTSUPP;
1453                 } else if (sk->sk_family != PF_RDS) {
1454                         ret = -EOPNOTSUPP;
1455                 }
1456                 if (!ret) {
1457                         if (val < 0 || val > 1)
1458                                 ret = -EINVAL;
1459                         else
1460                                 sock_valbool_flag(sk, SOCK_ZEROCOPY, valbool);
1461                 }
1462                 break;
1463
1464         case SO_TXTIME:
1465                 if (optlen != sizeof(struct sock_txtime)) {
1466                         ret = -EINVAL;
1467                         break;
1468                 } else if (copy_from_sockptr(&sk_txtime, optval,
1469                            sizeof(struct sock_txtime))) {
1470                         ret = -EFAULT;
1471                         break;
1472                 } else if (sk_txtime.flags & ~SOF_TXTIME_FLAGS_MASK) {
1473                         ret = -EINVAL;
1474                         break;
1475                 }
1476                 /* CLOCK_MONOTONIC is only used by sch_fq, and this packet
1477                  * scheduler has enough safe guards.
1478                  */
1479                 if (sk_txtime.clockid != CLOCK_MONOTONIC &&
1480                     !sockopt_ns_capable(sock_net(sk)->user_ns, CAP_NET_ADMIN)) {
1481                         ret = -EPERM;
1482                         break;
1483                 }
1484                 sock_valbool_flag(sk, SOCK_TXTIME, true);
1485                 sk->sk_clockid = sk_txtime.clockid;
1486                 sk->sk_txtime_deadline_mode =
1487                         !!(sk_txtime.flags & SOF_TXTIME_DEADLINE_MODE);
1488                 sk->sk_txtime_report_errors =
1489                         !!(sk_txtime.flags & SOF_TXTIME_REPORT_ERRORS);
1490                 break;
1491
1492         case SO_BINDTOIFINDEX:
1493                 ret = sock_bindtoindex_locked(sk, val);
1494                 break;
1495
1496         case SO_BUF_LOCK:
1497                 if (val & ~SOCK_BUF_LOCK_MASK) {
1498                         ret = -EINVAL;
1499                         break;
1500                 }
1501                 sk->sk_userlocks = val | (sk->sk_userlocks &
1502                                           ~SOCK_BUF_LOCK_MASK);
1503                 break;
1504
1505         case SO_RESERVE_MEM:
1506         {
1507                 int delta;
1508
1509                 if (val < 0) {
1510                         ret = -EINVAL;
1511                         break;
1512                 }
1513
1514                 delta = val - sk->sk_reserved_mem;
1515                 if (delta < 0)
1516                         sock_release_reserved_memory(sk, -delta);
1517                 else
1518                         ret = sock_reserve_memory(sk, delta);
1519                 break;
1520         }
1521
1522         case SO_TXREHASH:
1523                 if (val < -1 || val > 1) {
1524                         ret = -EINVAL;
1525                         break;
1526                 }
1527                 if ((u8)val == SOCK_TXREHASH_DEFAULT)
1528                         val = READ_ONCE(sock_net(sk)->core.sysctl_txrehash);
1529                 /* Paired with READ_ONCE() in tcp_rtx_synack() */
1530                 WRITE_ONCE(sk->sk_txrehash, (u8)val);
1531                 break;
1532
1533         default:
1534                 ret = -ENOPROTOOPT;
1535                 break;
1536         }
1537         sockopt_release_sock(sk);
1538         return ret;
1539 }
1540
1541 int sock_setsockopt(struct socket *sock, int level, int optname,
1542                     sockptr_t optval, unsigned int optlen)
1543 {
1544         return sk_setsockopt(sock->sk, level, optname,
1545                              optval, optlen);
1546 }
1547 EXPORT_SYMBOL(sock_setsockopt);
1548
1549 static const struct cred *sk_get_peer_cred(struct sock *sk)
1550 {
1551         const struct cred *cred;
1552
1553         spin_lock(&sk->sk_peer_lock);
1554         cred = get_cred(sk->sk_peer_cred);
1555         spin_unlock(&sk->sk_peer_lock);
1556
1557         return cred;
1558 }
1559
1560 static void cred_to_ucred(struct pid *pid, const struct cred *cred,
1561                           struct ucred *ucred)
1562 {
1563         ucred->pid = pid_vnr(pid);
1564         ucred->uid = ucred->gid = -1;
1565         if (cred) {
1566                 struct user_namespace *current_ns = current_user_ns();
1567
1568                 ucred->uid = from_kuid_munged(current_ns, cred->euid);
1569                 ucred->gid = from_kgid_munged(current_ns, cred->egid);
1570         }
1571 }
1572
1573 static int groups_to_user(sockptr_t dst, const struct group_info *src)
1574 {
1575         struct user_namespace *user_ns = current_user_ns();
1576         int i;
1577
1578         for (i = 0; i < src->ngroups; i++) {
1579                 gid_t gid = from_kgid_munged(user_ns, src->gid[i]);
1580
1581                 if (copy_to_sockptr_offset(dst, i * sizeof(gid), &gid, sizeof(gid)))
1582                         return -EFAULT;
1583         }
1584
1585         return 0;
1586 }
1587
1588 int sk_getsockopt(struct sock *sk, int level, int optname,
1589                   sockptr_t optval, sockptr_t optlen)
1590 {
1591         struct socket *sock = sk->sk_socket;
1592
1593         union {
1594                 int val;
1595                 u64 val64;
1596                 unsigned long ulval;
1597                 struct linger ling;
1598                 struct old_timeval32 tm32;
1599                 struct __kernel_old_timeval tm;
1600                 struct  __kernel_sock_timeval stm;
1601                 struct sock_txtime txtime;
1602                 struct so_timestamping timestamping;
1603         } v;
1604
1605         int lv = sizeof(int);
1606         int len;
1607
1608         if (copy_from_sockptr(&len, optlen, sizeof(int)))
1609                 return -EFAULT;
1610         if (len < 0)
1611                 return -EINVAL;
1612
1613         memset(&v, 0, sizeof(v));
1614
1615         switch (optname) {
1616         case SO_DEBUG:
1617                 v.val = sock_flag(sk, SOCK_DBG);
1618                 break;
1619
1620         case SO_DONTROUTE:
1621                 v.val = sock_flag(sk, SOCK_LOCALROUTE);
1622                 break;
1623
1624         case SO_BROADCAST:
1625                 v.val = sock_flag(sk, SOCK_BROADCAST);
1626                 break;
1627
1628         case SO_SNDBUF:
1629                 v.val = sk->sk_sndbuf;
1630                 break;
1631
1632         case SO_RCVBUF:
1633                 v.val = sk->sk_rcvbuf;
1634                 break;
1635
1636         case SO_REUSEADDR:
1637                 v.val = sk->sk_reuse;
1638                 break;
1639
1640         case SO_REUSEPORT:
1641                 v.val = sk->sk_reuseport;
1642                 break;
1643
1644         case SO_KEEPALIVE:
1645                 v.val = sock_flag(sk, SOCK_KEEPOPEN);
1646                 break;
1647
1648         case SO_TYPE:
1649                 v.val = sk->sk_type;
1650                 break;
1651
1652         case SO_PROTOCOL:
1653                 v.val = sk->sk_protocol;
1654                 break;
1655
1656         case SO_DOMAIN:
1657                 v.val = sk->sk_family;
1658                 break;
1659
1660         case SO_ERROR:
1661                 v.val = -sock_error(sk);
1662                 if (v.val == 0)
1663                         v.val = xchg(&sk->sk_err_soft, 0);
1664                 break;
1665
1666         case SO_OOBINLINE:
1667                 v.val = sock_flag(sk, SOCK_URGINLINE);
1668                 break;
1669
1670         case SO_NO_CHECK:
1671                 v.val = sk->sk_no_check_tx;
1672                 break;
1673
1674         case SO_PRIORITY:
1675                 v.val = sk->sk_priority;
1676                 break;
1677
1678         case SO_LINGER:
1679                 lv              = sizeof(v.ling);
1680                 v.ling.l_onoff  = sock_flag(sk, SOCK_LINGER);
1681                 v.ling.l_linger = sk->sk_lingertime / HZ;
1682                 break;
1683
1684         case SO_BSDCOMPAT:
1685                 break;
1686
1687         case SO_TIMESTAMP_OLD:
1688                 v.val = sock_flag(sk, SOCK_RCVTSTAMP) &&
1689                                 !sock_flag(sk, SOCK_TSTAMP_NEW) &&
1690                                 !sock_flag(sk, SOCK_RCVTSTAMPNS);
1691                 break;
1692
1693         case SO_TIMESTAMPNS_OLD:
1694                 v.val = sock_flag(sk, SOCK_RCVTSTAMPNS) && !sock_flag(sk, SOCK_TSTAMP_NEW);
1695                 break;
1696
1697         case SO_TIMESTAMP_NEW:
1698                 v.val = sock_flag(sk, SOCK_RCVTSTAMP) && sock_flag(sk, SOCK_TSTAMP_NEW);
1699                 break;
1700
1701         case SO_TIMESTAMPNS_NEW:
1702                 v.val = sock_flag(sk, SOCK_RCVTSTAMPNS) && sock_flag(sk, SOCK_TSTAMP_NEW);
1703                 break;
1704
1705         case SO_TIMESTAMPING_OLD:
1706                 lv = sizeof(v.timestamping);
1707                 v.timestamping.flags = sk->sk_tsflags;
1708                 v.timestamping.bind_phc = sk->sk_bind_phc;
1709                 break;
1710
1711         case SO_RCVTIMEO_OLD:
1712         case SO_RCVTIMEO_NEW:
1713                 lv = sock_get_timeout(sk->sk_rcvtimeo, &v, SO_RCVTIMEO_OLD == optname);
1714                 break;
1715
1716         case SO_SNDTIMEO_OLD:
1717         case SO_SNDTIMEO_NEW:
1718                 lv = sock_get_timeout(sk->sk_sndtimeo, &v, SO_SNDTIMEO_OLD == optname);
1719                 break;
1720
1721         case SO_RCVLOWAT:
1722                 v.val = sk->sk_rcvlowat;
1723                 break;
1724
1725         case SO_SNDLOWAT:
1726                 v.val = 1;
1727                 break;
1728
1729         case SO_PASSCRED:
1730                 v.val = !!test_bit(SOCK_PASSCRED, &sock->flags);
1731                 break;
1732
1733         case SO_PEERCRED:
1734         {
1735                 struct ucred peercred;
1736                 if (len > sizeof(peercred))
1737                         len = sizeof(peercred);
1738
1739                 spin_lock(&sk->sk_peer_lock);
1740                 cred_to_ucred(sk->sk_peer_pid, sk->sk_peer_cred, &peercred);
1741                 spin_unlock(&sk->sk_peer_lock);
1742
1743                 if (copy_to_sockptr(optval, &peercred, len))
1744                         return -EFAULT;
1745                 goto lenout;
1746         }
1747
1748         case SO_PEERGROUPS:
1749         {
1750                 const struct cred *cred;
1751                 int ret, n;
1752
1753                 cred = sk_get_peer_cred(sk);
1754                 if (!cred)
1755                         return -ENODATA;
1756
1757                 n = cred->group_info->ngroups;
1758                 if (len < n * sizeof(gid_t)) {
1759                         len = n * sizeof(gid_t);
1760                         put_cred(cred);
1761                         return copy_to_sockptr(optlen, &len, sizeof(int)) ? -EFAULT : -ERANGE;
1762                 }
1763                 len = n * sizeof(gid_t);
1764
1765                 ret = groups_to_user(optval, cred->group_info);
1766                 put_cred(cred);
1767                 if (ret)
1768                         return ret;
1769                 goto lenout;
1770         }
1771
1772         case SO_PEERNAME:
1773         {
1774                 char address[128];
1775
1776                 lv = sock->ops->getname(sock, (struct sockaddr *)address, 2);
1777                 if (lv < 0)
1778                         return -ENOTCONN;
1779                 if (lv < len)
1780                         return -EINVAL;
1781                 if (copy_to_sockptr(optval, address, len))
1782                         return -EFAULT;
1783                 goto lenout;
1784         }
1785
1786         /* Dubious BSD thing... Probably nobody even uses it, but
1787          * the UNIX standard wants it for whatever reason... -DaveM
1788          */
1789         case SO_ACCEPTCONN:
1790                 v.val = sk->sk_state == TCP_LISTEN;
1791                 break;
1792
1793         case SO_PASSSEC:
1794                 v.val = !!test_bit(SOCK_PASSSEC, &sock->flags);
1795                 break;
1796
1797         case SO_PEERSEC:
1798                 return security_socket_getpeersec_stream(sock, optval.user, optlen.user, len);
1799
1800         case SO_MARK:
1801                 v.val = sk->sk_mark;
1802                 break;
1803
1804         case SO_RCVMARK:
1805                 v.val = sock_flag(sk, SOCK_RCVMARK);
1806                 break;
1807
1808         case SO_RXQ_OVFL:
1809                 v.val = sock_flag(sk, SOCK_RXQ_OVFL);
1810                 break;
1811
1812         case SO_WIFI_STATUS:
1813                 v.val = sock_flag(sk, SOCK_WIFI_STATUS);
1814                 break;
1815
1816         case SO_PEEK_OFF:
1817                 if (!sock->ops->set_peek_off)
1818                         return -EOPNOTSUPP;
1819
1820                 v.val = sk->sk_peek_off;
1821                 break;
1822         case SO_NOFCS:
1823                 v.val = sock_flag(sk, SOCK_NOFCS);
1824                 break;
1825
1826         case SO_BINDTODEVICE:
1827                 return sock_getbindtodevice(sk, optval, optlen, len);
1828
1829         case SO_GET_FILTER:
1830                 len = sk_get_filter(sk, optval, len);
1831                 if (len < 0)
1832                         return len;
1833
1834                 goto lenout;
1835
1836         case SO_LOCK_FILTER:
1837                 v.val = sock_flag(sk, SOCK_FILTER_LOCKED);
1838                 break;
1839
1840         case SO_BPF_EXTENSIONS:
1841                 v.val = bpf_tell_extensions();
1842                 break;
1843
1844         case SO_SELECT_ERR_QUEUE:
1845                 v.val = sock_flag(sk, SOCK_SELECT_ERR_QUEUE);
1846                 break;
1847
1848 #ifdef CONFIG_NET_RX_BUSY_POLL
1849         case SO_BUSY_POLL:
1850                 v.val = sk->sk_ll_usec;
1851                 break;
1852         case SO_PREFER_BUSY_POLL:
1853                 v.val = READ_ONCE(sk->sk_prefer_busy_poll);
1854                 break;
1855 #endif
1856
1857         case SO_MAX_PACING_RATE:
1858                 if (sizeof(v.ulval) != sizeof(v.val) && len >= sizeof(v.ulval)) {
1859                         lv = sizeof(v.ulval);
1860                         v.ulval = sk->sk_max_pacing_rate;
1861                 } else {
1862                         /* 32bit version */
1863                         v.val = min_t(unsigned long, sk->sk_max_pacing_rate, ~0U);
1864                 }
1865                 break;
1866
1867         case SO_INCOMING_CPU:
1868                 v.val = READ_ONCE(sk->sk_incoming_cpu);
1869                 break;
1870
1871         case SO_MEMINFO:
1872         {
1873                 u32 meminfo[SK_MEMINFO_VARS];
1874
1875                 sk_get_meminfo(sk, meminfo);
1876
1877                 len = min_t(unsigned int, len, sizeof(meminfo));
1878                 if (copy_to_sockptr(optval, &meminfo, len))
1879                         return -EFAULT;
1880
1881                 goto lenout;
1882         }
1883
1884 #ifdef CONFIG_NET_RX_BUSY_POLL
1885         case SO_INCOMING_NAPI_ID:
1886                 v.val = READ_ONCE(sk->sk_napi_id);
1887
1888                 /* aggregate non-NAPI IDs down to 0 */
1889                 if (v.val < MIN_NAPI_ID)
1890                         v.val = 0;
1891
1892                 break;
1893 #endif
1894
1895         case SO_COOKIE:
1896                 lv = sizeof(u64);
1897                 if (len < lv)
1898                         return -EINVAL;
1899                 v.val64 = sock_gen_cookie(sk);
1900                 break;
1901
1902         case SO_ZEROCOPY:
1903                 v.val = sock_flag(sk, SOCK_ZEROCOPY);
1904                 break;
1905
1906         case SO_TXTIME:
1907                 lv = sizeof(v.txtime);
1908                 v.txtime.clockid = sk->sk_clockid;
1909                 v.txtime.flags |= sk->sk_txtime_deadline_mode ?
1910                                   SOF_TXTIME_DEADLINE_MODE : 0;
1911                 v.txtime.flags |= sk->sk_txtime_report_errors ?
1912                                   SOF_TXTIME_REPORT_ERRORS : 0;
1913                 break;
1914
1915         case SO_BINDTOIFINDEX:
1916                 v.val = READ_ONCE(sk->sk_bound_dev_if);
1917                 break;
1918
1919         case SO_NETNS_COOKIE:
1920                 lv = sizeof(u64);
1921                 if (len != lv)
1922                         return -EINVAL;
1923                 v.val64 = sock_net(sk)->net_cookie;
1924                 break;
1925
1926         case SO_BUF_LOCK:
1927                 v.val = sk->sk_userlocks & SOCK_BUF_LOCK_MASK;
1928                 break;
1929
1930         case SO_RESERVE_MEM:
1931                 v.val = sk->sk_reserved_mem;
1932                 break;
1933
1934         case SO_TXREHASH:
1935                 v.val = sk->sk_txrehash;
1936                 break;
1937
1938         default:
1939                 /* We implement the SO_SNDLOWAT etc to not be settable
1940                  * (1003.1g 7).
1941                  */
1942                 return -ENOPROTOOPT;
1943         }
1944
1945         if (len > lv)
1946                 len = lv;
1947         if (copy_to_sockptr(optval, &v, len))
1948                 return -EFAULT;
1949 lenout:
1950         if (copy_to_sockptr(optlen, &len, sizeof(int)))
1951                 return -EFAULT;
1952         return 0;
1953 }
1954
1955 int sock_getsockopt(struct socket *sock, int level, int optname,
1956                     char __user *optval, int __user *optlen)
1957 {
1958         return sk_getsockopt(sock->sk, level, optname,
1959                              USER_SOCKPTR(optval),
1960                              USER_SOCKPTR(optlen));
1961 }
1962
1963 /*
1964  * Initialize an sk_lock.
1965  *
1966  * (We also register the sk_lock with the lock validator.)
1967  */
1968 static inline void sock_lock_init(struct sock *sk)
1969 {
1970         if (sk->sk_kern_sock)
1971                 sock_lock_init_class_and_name(
1972                         sk,
1973                         af_family_kern_slock_key_strings[sk->sk_family],
1974                         af_family_kern_slock_keys + sk->sk_family,
1975                         af_family_kern_key_strings[sk->sk_family],
1976                         af_family_kern_keys + sk->sk_family);
1977         else
1978                 sock_lock_init_class_and_name(
1979                         sk,
1980                         af_family_slock_key_strings[sk->sk_family],
1981                         af_family_slock_keys + sk->sk_family,
1982                         af_family_key_strings[sk->sk_family],
1983                         af_family_keys + sk->sk_family);
1984 }
1985
1986 /*
1987  * Copy all fields from osk to nsk but nsk->sk_refcnt must not change yet,
1988  * even temporarly, because of RCU lookups. sk_node should also be left as is.
1989  * We must not copy fields between sk_dontcopy_begin and sk_dontcopy_end
1990  */
1991 static void sock_copy(struct sock *nsk, const struct sock *osk)
1992 {
1993         const struct proto *prot = READ_ONCE(osk->sk_prot);
1994 #ifdef CONFIG_SECURITY_NETWORK
1995         void *sptr = nsk->sk_security;
1996 #endif
1997
1998         /* If we move sk_tx_queue_mapping out of the private section,
1999          * we must check if sk_tx_queue_clear() is called after
2000          * sock_copy() in sk_clone_lock().
2001          */
2002         BUILD_BUG_ON(offsetof(struct sock, sk_tx_queue_mapping) <
2003                      offsetof(struct sock, sk_dontcopy_begin) ||
2004                      offsetof(struct sock, sk_tx_queue_mapping) >=
2005                      offsetof(struct sock, sk_dontcopy_end));
2006
2007         memcpy(nsk, osk, offsetof(struct sock, sk_dontcopy_begin));
2008
2009         memcpy(&nsk->sk_dontcopy_end, &osk->sk_dontcopy_end,
2010                prot->obj_size - offsetof(struct sock, sk_dontcopy_end));
2011
2012 #ifdef CONFIG_SECURITY_NETWORK
2013         nsk->sk_security = sptr;
2014         security_sk_clone(osk, nsk);
2015 #endif
2016 }
2017
2018 static struct sock *sk_prot_alloc(struct proto *prot, gfp_t priority,
2019                 int family)
2020 {
2021         struct sock *sk;
2022         struct kmem_cache *slab;
2023
2024         slab = prot->slab;
2025         if (slab != NULL) {
2026                 sk = kmem_cache_alloc(slab, priority & ~__GFP_ZERO);
2027                 if (!sk)
2028                         return sk;
2029                 if (want_init_on_alloc(priority))
2030                         sk_prot_clear_nulls(sk, prot->obj_size);
2031         } else
2032                 sk = kmalloc(prot->obj_size, priority);
2033
2034         if (sk != NULL) {
2035                 if (security_sk_alloc(sk, family, priority))
2036                         goto out_free;
2037
2038                 if (!try_module_get(prot->owner))
2039                         goto out_free_sec;
2040         }
2041
2042         return sk;
2043
2044 out_free_sec:
2045         security_sk_free(sk);
2046 out_free:
2047         if (slab != NULL)
2048                 kmem_cache_free(slab, sk);
2049         else
2050                 kfree(sk);
2051         return NULL;
2052 }
2053
2054 static void sk_prot_free(struct proto *prot, struct sock *sk)
2055 {
2056         struct kmem_cache *slab;
2057         struct module *owner;
2058
2059         owner = prot->owner;
2060         slab = prot->slab;
2061
2062         cgroup_sk_free(&sk->sk_cgrp_data);
2063         mem_cgroup_sk_free(sk);
2064         security_sk_free(sk);
2065         if (slab != NULL)
2066                 kmem_cache_free(slab, sk);
2067         else
2068                 kfree(sk);
2069         module_put(owner);
2070 }
2071
2072 /**
2073  *      sk_alloc - All socket objects are allocated here
2074  *      @net: the applicable net namespace
2075  *      @family: protocol family
2076  *      @priority: for allocation (%GFP_KERNEL, %GFP_ATOMIC, etc)
2077  *      @prot: struct proto associated with this new sock instance
2078  *      @kern: is this to be a kernel socket?
2079  */
2080 struct sock *sk_alloc(struct net *net, int family, gfp_t priority,
2081                       struct proto *prot, int kern)
2082 {
2083         struct sock *sk;
2084
2085         sk = sk_prot_alloc(prot, priority | __GFP_ZERO, family);
2086         if (sk) {
2087                 sk->sk_family = family;
2088                 /*
2089                  * See comment in struct sock definition to understand
2090                  * why we need sk_prot_creator -acme
2091                  */
2092                 sk->sk_prot = sk->sk_prot_creator = prot;
2093                 sk->sk_kern_sock = kern;
2094                 sock_lock_init(sk);
2095                 sk->sk_net_refcnt = kern ? 0 : 1;
2096                 if (likely(sk->sk_net_refcnt)) {
2097                         get_net_track(net, &sk->ns_tracker, priority);
2098                         sock_inuse_add(net, 1);
2099                 }
2100
2101                 sock_net_set(sk, net);
2102                 refcount_set(&sk->sk_wmem_alloc, 1);
2103
2104                 mem_cgroup_sk_alloc(sk);
2105                 cgroup_sk_alloc(&sk->sk_cgrp_data);
2106                 sock_update_classid(&sk->sk_cgrp_data);
2107                 sock_update_netprioidx(&sk->sk_cgrp_data);
2108                 sk_tx_queue_clear(sk);
2109         }
2110
2111         return sk;
2112 }
2113 EXPORT_SYMBOL(sk_alloc);
2114
2115 /* Sockets having SOCK_RCU_FREE will call this function after one RCU
2116  * grace period. This is the case for UDP sockets and TCP listeners.
2117  */
2118 static void __sk_destruct(struct rcu_head *head)
2119 {
2120         struct sock *sk = container_of(head, struct sock, sk_rcu);
2121         struct sk_filter *filter;
2122
2123         if (sk->sk_destruct)
2124                 sk->sk_destruct(sk);
2125
2126         filter = rcu_dereference_check(sk->sk_filter,
2127                                        refcount_read(&sk->sk_wmem_alloc) == 0);
2128         if (filter) {
2129                 sk_filter_uncharge(sk, filter);
2130                 RCU_INIT_POINTER(sk->sk_filter, NULL);
2131         }
2132
2133         sock_disable_timestamp(sk, SK_FLAGS_TIMESTAMP);
2134
2135 #ifdef CONFIG_BPF_SYSCALL
2136         bpf_sk_storage_free(sk);
2137 #endif
2138
2139         if (atomic_read(&sk->sk_omem_alloc))
2140                 pr_debug("%s: optmem leakage (%d bytes) detected\n",
2141                          __func__, atomic_read(&sk->sk_omem_alloc));
2142
2143         if (sk->sk_frag.page) {
2144                 put_page(sk->sk_frag.page);
2145                 sk->sk_frag.page = NULL;
2146         }
2147
2148         /* We do not need to acquire sk->sk_peer_lock, we are the last user. */
2149         put_cred(sk->sk_peer_cred);
2150         put_pid(sk->sk_peer_pid);
2151
2152         if (likely(sk->sk_net_refcnt))
2153                 put_net_track(sock_net(sk), &sk->ns_tracker);
2154         sk_prot_free(sk->sk_prot_creator, sk);
2155 }
2156
2157 void sk_destruct(struct sock *sk)
2158 {
2159         bool use_call_rcu = sock_flag(sk, SOCK_RCU_FREE);
2160
2161         if (rcu_access_pointer(sk->sk_reuseport_cb)) {
2162                 reuseport_detach_sock(sk);
2163                 use_call_rcu = true;
2164         }
2165
2166         if (use_call_rcu)
2167                 call_rcu(&sk->sk_rcu, __sk_destruct);
2168         else
2169                 __sk_destruct(&sk->sk_rcu);
2170 }
2171
2172 static void __sk_free(struct sock *sk)
2173 {
2174         if (likely(sk->sk_net_refcnt))
2175                 sock_inuse_add(sock_net(sk), -1);
2176
2177         if (unlikely(sk->sk_net_refcnt && sock_diag_has_destroy_listeners(sk)))
2178                 sock_diag_broadcast_destroy(sk);
2179         else
2180                 sk_destruct(sk);
2181 }
2182
2183 void sk_free(struct sock *sk)
2184 {
2185         /*
2186          * We subtract one from sk_wmem_alloc and can know if
2187          * some packets are still in some tx queue.
2188          * If not null, sock_wfree() will call __sk_free(sk) later
2189          */
2190         if (refcount_dec_and_test(&sk->sk_wmem_alloc))
2191                 __sk_free(sk);
2192 }
2193 EXPORT_SYMBOL(sk_free);
2194
2195 static void sk_init_common(struct sock *sk)
2196 {
2197         skb_queue_head_init(&sk->sk_receive_queue);
2198         skb_queue_head_init(&sk->sk_write_queue);
2199         skb_queue_head_init(&sk->sk_error_queue);
2200
2201         rwlock_init(&sk->sk_callback_lock);
2202         lockdep_set_class_and_name(&sk->sk_receive_queue.lock,
2203                         af_rlock_keys + sk->sk_family,
2204                         af_family_rlock_key_strings[sk->sk_family]);
2205         lockdep_set_class_and_name(&sk->sk_write_queue.lock,
2206                         af_wlock_keys + sk->sk_family,
2207                         af_family_wlock_key_strings[sk->sk_family]);
2208         lockdep_set_class_and_name(&sk->sk_error_queue.lock,
2209                         af_elock_keys + sk->sk_family,
2210                         af_family_elock_key_strings[sk->sk_family]);
2211         lockdep_set_class_and_name(&sk->sk_callback_lock,
2212                         af_callback_keys + sk->sk_family,
2213                         af_family_clock_key_strings[sk->sk_family]);
2214 }
2215
2216 /**
2217  *      sk_clone_lock - clone a socket, and lock its clone
2218  *      @sk: the socket to clone
2219  *      @priority: for allocation (%GFP_KERNEL, %GFP_ATOMIC, etc)
2220  *
2221  *      Caller must unlock socket even in error path (bh_unlock_sock(newsk))
2222  */
2223 struct sock *sk_clone_lock(const struct sock *sk, const gfp_t priority)
2224 {
2225         struct proto *prot = READ_ONCE(sk->sk_prot);
2226         struct sk_filter *filter;
2227         bool is_charged = true;
2228         struct sock *newsk;
2229
2230         newsk = sk_prot_alloc(prot, priority, sk->sk_family);
2231         if (!newsk)
2232                 goto out;
2233
2234         sock_copy(newsk, sk);
2235
2236         newsk->sk_prot_creator = prot;
2237
2238         /* SANITY */
2239         if (likely(newsk->sk_net_refcnt)) {
2240                 get_net_track(sock_net(newsk), &newsk->ns_tracker, priority);
2241                 sock_inuse_add(sock_net(newsk), 1);
2242         }
2243         sk_node_init(&newsk->sk_node);
2244         sock_lock_init(newsk);
2245         bh_lock_sock(newsk);
2246         newsk->sk_backlog.head  = newsk->sk_backlog.tail = NULL;
2247         newsk->sk_backlog.len = 0;
2248
2249         atomic_set(&newsk->sk_rmem_alloc, 0);
2250
2251         /* sk_wmem_alloc set to one (see sk_free() and sock_wfree()) */
2252         refcount_set(&newsk->sk_wmem_alloc, 1);
2253
2254         atomic_set(&newsk->sk_omem_alloc, 0);
2255         sk_init_common(newsk);
2256
2257         newsk->sk_dst_cache     = NULL;
2258         newsk->sk_dst_pending_confirm = 0;
2259         newsk->sk_wmem_queued   = 0;
2260         newsk->sk_forward_alloc = 0;
2261         newsk->sk_reserved_mem  = 0;
2262         atomic_set(&newsk->sk_drops, 0);
2263         newsk->sk_send_head     = NULL;
2264         newsk->sk_userlocks     = sk->sk_userlocks & ~SOCK_BINDPORT_LOCK;
2265         atomic_set(&newsk->sk_zckey, 0);
2266
2267         sock_reset_flag(newsk, SOCK_DONE);
2268
2269         /* sk->sk_memcg will be populated at accept() time */
2270         newsk->sk_memcg = NULL;
2271
2272         cgroup_sk_clone(&newsk->sk_cgrp_data);
2273
2274         rcu_read_lock();
2275         filter = rcu_dereference(sk->sk_filter);
2276         if (filter != NULL)
2277                 /* though it's an empty new sock, the charging may fail
2278                  * if sysctl_optmem_max was changed between creation of
2279                  * original socket and cloning
2280                  */
2281                 is_charged = sk_filter_charge(newsk, filter);
2282         RCU_INIT_POINTER(newsk->sk_filter, filter);
2283         rcu_read_unlock();
2284
2285         if (unlikely(!is_charged || xfrm_sk_clone_policy(newsk, sk))) {
2286                 /* We need to make sure that we don't uncharge the new
2287                  * socket if we couldn't charge it in the first place
2288                  * as otherwise we uncharge the parent's filter.
2289                  */
2290                 if (!is_charged)
2291                         RCU_INIT_POINTER(newsk->sk_filter, NULL);
2292                 sk_free_unlock_clone(newsk);
2293                 newsk = NULL;
2294                 goto out;
2295         }
2296         RCU_INIT_POINTER(newsk->sk_reuseport_cb, NULL);
2297
2298         if (bpf_sk_storage_clone(sk, newsk)) {
2299                 sk_free_unlock_clone(newsk);
2300                 newsk = NULL;
2301                 goto out;
2302         }
2303
2304         /* Clear sk_user_data if parent had the pointer tagged
2305          * as not suitable for copying when cloning.
2306          */
2307         if (sk_user_data_is_nocopy(newsk))
2308                 newsk->sk_user_data = NULL;
2309
2310         newsk->sk_err      = 0;
2311         newsk->sk_err_soft = 0;
2312         newsk->sk_priority = 0;
2313         newsk->sk_incoming_cpu = raw_smp_processor_id();
2314
2315         /* Before updating sk_refcnt, we must commit prior changes to memory
2316          * (Documentation/RCU/rculist_nulls.rst for details)
2317          */
2318         smp_wmb();
2319         refcount_set(&newsk->sk_refcnt, 2);
2320
2321         /* Increment the counter in the same struct proto as the master
2322          * sock (sk_refcnt_debug_inc uses newsk->sk_prot->socks, that
2323          * is the same as sk->sk_prot->socks, as this field was copied
2324          * with memcpy).
2325          *
2326          * This _changes_ the previous behaviour, where
2327          * tcp_create_openreq_child always was incrementing the
2328          * equivalent to tcp_prot->socks (inet_sock_nr), so this have
2329          * to be taken into account in all callers. -acme
2330          */
2331         sk_refcnt_debug_inc(newsk);
2332         sk_set_socket(newsk, NULL);
2333         sk_tx_queue_clear(newsk);
2334         RCU_INIT_POINTER(newsk->sk_wq, NULL);
2335
2336         if (newsk->sk_prot->sockets_allocated)
2337                 sk_sockets_allocated_inc(newsk);
2338
2339         if (sock_needs_netstamp(sk) && newsk->sk_flags & SK_FLAGS_TIMESTAMP)
2340                 net_enable_timestamp();
2341 out:
2342         return newsk;
2343 }
2344 EXPORT_SYMBOL_GPL(sk_clone_lock);
2345
2346 void sk_free_unlock_clone(struct sock *sk)
2347 {
2348         /* It is still raw copy of parent, so invalidate
2349          * destructor and make plain sk_free() */
2350         sk->sk_destruct = NULL;
2351         bh_unlock_sock(sk);
2352         sk_free(sk);
2353 }
2354 EXPORT_SYMBOL_GPL(sk_free_unlock_clone);
2355
2356 static void sk_trim_gso_size(struct sock *sk)
2357 {
2358         if (sk->sk_gso_max_size <= GSO_LEGACY_MAX_SIZE)
2359                 return;
2360 #if IS_ENABLED(CONFIG_IPV6)
2361         if (sk->sk_family == AF_INET6 &&
2362             sk_is_tcp(sk) &&
2363             !ipv6_addr_v4mapped(&sk->sk_v6_rcv_saddr))
2364                 return;
2365 #endif
2366         sk->sk_gso_max_size = GSO_LEGACY_MAX_SIZE;
2367 }
2368
2369 void sk_setup_caps(struct sock *sk, struct dst_entry *dst)
2370 {
2371         u32 max_segs = 1;
2372
2373         sk_dst_set(sk, dst);
2374         sk->sk_route_caps = dst->dev->features;
2375         if (sk_is_tcp(sk))
2376                 sk->sk_route_caps |= NETIF_F_GSO;
2377         if (sk->sk_route_caps & NETIF_F_GSO)
2378                 sk->sk_route_caps |= NETIF_F_GSO_SOFTWARE;
2379         if (unlikely(sk->sk_gso_disabled))
2380                 sk->sk_route_caps &= ~NETIF_F_GSO_MASK;
2381         if (sk_can_gso(sk)) {
2382                 if (dst->header_len && !xfrm_dst_offload_ok(dst)) {
2383                         sk->sk_route_caps &= ~NETIF_F_GSO_MASK;
2384                 } else {
2385                         sk->sk_route_caps |= NETIF_F_SG | NETIF_F_HW_CSUM;
2386                         /* pairs with the WRITE_ONCE() in netif_set_gso_max_size() */
2387                         sk->sk_gso_max_size = READ_ONCE(dst->dev->gso_max_size);
2388                         sk_trim_gso_size(sk);
2389                         sk->sk_gso_max_size -= (MAX_TCP_HEADER + 1);
2390                         /* pairs with the WRITE_ONCE() in netif_set_gso_max_segs() */
2391                         max_segs = max_t(u32, READ_ONCE(dst->dev->gso_max_segs), 1);
2392                 }
2393         }
2394         sk->sk_gso_max_segs = max_segs;
2395 }
2396 EXPORT_SYMBOL_GPL(sk_setup_caps);
2397
2398 /*
2399  *      Simple resource managers for sockets.
2400  */
2401
2402
2403 /*
2404  * Write buffer destructor automatically called from kfree_skb.
2405  */
2406 void sock_wfree(struct sk_buff *skb)
2407 {
2408         struct sock *sk = skb->sk;
2409         unsigned int len = skb->truesize;
2410         bool free;
2411
2412         if (!sock_flag(sk, SOCK_USE_WRITE_QUEUE)) {
2413                 if (sock_flag(sk, SOCK_RCU_FREE) &&
2414                     sk->sk_write_space == sock_def_write_space) {
2415                         rcu_read_lock();
2416                         free = refcount_sub_and_test(len, &sk->sk_wmem_alloc);
2417                         sock_def_write_space_wfree(sk);
2418                         rcu_read_unlock();
2419                         if (unlikely(free))
2420                                 __sk_free(sk);
2421                         return;
2422                 }
2423
2424                 /*
2425                  * Keep a reference on sk_wmem_alloc, this will be released
2426                  * after sk_write_space() call
2427                  */
2428                 WARN_ON(refcount_sub_and_test(len - 1, &sk->sk_wmem_alloc));
2429                 sk->sk_write_space(sk);
2430                 len = 1;
2431         }
2432         /*
2433          * if sk_wmem_alloc reaches 0, we must finish what sk_free()
2434          * could not do because of in-flight packets
2435          */
2436         if (refcount_sub_and_test(len, &sk->sk_wmem_alloc))
2437                 __sk_free(sk);
2438 }
2439 EXPORT_SYMBOL(sock_wfree);
2440
2441 /* This variant of sock_wfree() is used by TCP,
2442  * since it sets SOCK_USE_WRITE_QUEUE.
2443  */
2444 void __sock_wfree(struct sk_buff *skb)
2445 {
2446         struct sock *sk = skb->sk;
2447
2448         if (refcount_sub_and_test(skb->truesize, &sk->sk_wmem_alloc))
2449                 __sk_free(sk);
2450 }
2451
2452 void skb_set_owner_w(struct sk_buff *skb, struct sock *sk)
2453 {
2454         skb_orphan(skb);
2455         skb->sk = sk;
2456 #ifdef CONFIG_INET
2457         if (unlikely(!sk_fullsock(sk))) {
2458                 skb->destructor = sock_edemux;
2459                 sock_hold(sk);
2460                 return;
2461         }
2462 #endif
2463         skb->destructor = sock_wfree;
2464         skb_set_hash_from_sk(skb, sk);
2465         /*
2466          * We used to take a refcount on sk, but following operation
2467          * is enough to guarantee sk_free() wont free this sock until
2468          * all in-flight packets are completed
2469          */
2470         refcount_add(skb->truesize, &sk->sk_wmem_alloc);
2471 }
2472 EXPORT_SYMBOL(skb_set_owner_w);
2473
2474 static bool can_skb_orphan_partial(const struct sk_buff *skb)
2475 {
2476 #ifdef CONFIG_TLS_DEVICE
2477         /* Drivers depend on in-order delivery for crypto offload,
2478          * partial orphan breaks out-of-order-OK logic.
2479          */
2480         if (skb->decrypted)
2481                 return false;
2482 #endif
2483         return (skb->destructor == sock_wfree ||
2484                 (IS_ENABLED(CONFIG_INET) && skb->destructor == tcp_wfree));
2485 }
2486
2487 /* This helper is used by netem, as it can hold packets in its
2488  * delay queue. We want to allow the owner socket to send more
2489  * packets, as if they were already TX completed by a typical driver.
2490  * But we also want to keep skb->sk set because some packet schedulers
2491  * rely on it (sch_fq for example).
2492  */
2493 void skb_orphan_partial(struct sk_buff *skb)
2494 {
2495         if (skb_is_tcp_pure_ack(skb))
2496                 return;
2497
2498         if (can_skb_orphan_partial(skb) && skb_set_owner_sk_safe(skb, skb->sk))
2499                 return;
2500
2501         skb_orphan(skb);
2502 }
2503 EXPORT_SYMBOL(skb_orphan_partial);
2504
2505 /*
2506  * Read buffer destructor automatically called from kfree_skb.
2507  */
2508 void sock_rfree(struct sk_buff *skb)
2509 {
2510         struct sock *sk = skb->sk;
2511         unsigned int len = skb->truesize;
2512
2513         atomic_sub(len, &sk->sk_rmem_alloc);
2514         sk_mem_uncharge(sk, len);
2515 }
2516 EXPORT_SYMBOL(sock_rfree);
2517
2518 /*
2519  * Buffer destructor for skbs that are not used directly in read or write
2520  * path, e.g. for error handler skbs. Automatically called from kfree_skb.
2521  */
2522 void sock_efree(struct sk_buff *skb)
2523 {
2524         sock_put(skb->sk);
2525 }
2526 EXPORT_SYMBOL(sock_efree);
2527
2528 /* Buffer destructor for prefetch/receive path where reference count may
2529  * not be held, e.g. for listen sockets.
2530  */
2531 #ifdef CONFIG_INET
2532 void sock_pfree(struct sk_buff *skb)
2533 {
2534         if (sk_is_refcounted(skb->sk))
2535                 sock_gen_put(skb->sk);
2536 }
2537 EXPORT_SYMBOL(sock_pfree);
2538 #endif /* CONFIG_INET */
2539
2540 kuid_t sock_i_uid(struct sock *sk)
2541 {
2542         kuid_t uid;
2543
2544         read_lock_bh(&sk->sk_callback_lock);
2545         uid = sk->sk_socket ? SOCK_INODE(sk->sk_socket)->i_uid : GLOBAL_ROOT_UID;
2546         read_unlock_bh(&sk->sk_callback_lock);
2547         return uid;
2548 }
2549 EXPORT_SYMBOL(sock_i_uid);
2550
2551 unsigned long sock_i_ino(struct sock *sk)
2552 {
2553         unsigned long ino;
2554
2555         read_lock_bh(&sk->sk_callback_lock);
2556         ino = sk->sk_socket ? SOCK_INODE(sk->sk_socket)->i_ino : 0;
2557         read_unlock_bh(&sk->sk_callback_lock);
2558         return ino;
2559 }
2560 EXPORT_SYMBOL(sock_i_ino);
2561
2562 /*
2563  * Allocate a skb from the socket's send buffer.
2564  */
2565 struct sk_buff *sock_wmalloc(struct sock *sk, unsigned long size, int force,
2566                              gfp_t priority)
2567 {
2568         if (force ||
2569             refcount_read(&sk->sk_wmem_alloc) < READ_ONCE(sk->sk_sndbuf)) {
2570                 struct sk_buff *skb = alloc_skb(size, priority);
2571
2572                 if (skb) {
2573                         skb_set_owner_w(skb, sk);
2574                         return skb;
2575                 }
2576         }
2577         return NULL;
2578 }
2579 EXPORT_SYMBOL(sock_wmalloc);
2580
2581 static void sock_ofree(struct sk_buff *skb)
2582 {
2583         struct sock *sk = skb->sk;
2584
2585         atomic_sub(skb->truesize, &sk->sk_omem_alloc);
2586 }
2587
2588 struct sk_buff *sock_omalloc(struct sock *sk, unsigned long size,
2589                              gfp_t priority)
2590 {
2591         struct sk_buff *skb;
2592
2593         /* small safe race: SKB_TRUESIZE may differ from final skb->truesize */
2594         if (atomic_read(&sk->sk_omem_alloc) + SKB_TRUESIZE(size) >
2595             READ_ONCE(sysctl_optmem_max))
2596                 return NULL;
2597
2598         skb = alloc_skb(size, priority);
2599         if (!skb)
2600                 return NULL;
2601
2602         atomic_add(skb->truesize, &sk->sk_omem_alloc);
2603         skb->sk = sk;
2604         skb->destructor = sock_ofree;
2605         return skb;
2606 }
2607
2608 /*
2609  * Allocate a memory block from the socket's option memory buffer.
2610  */
2611 void *sock_kmalloc(struct sock *sk, int size, gfp_t priority)
2612 {
2613         int optmem_max = READ_ONCE(sysctl_optmem_max);
2614
2615         if ((unsigned int)size <= optmem_max &&
2616             atomic_read(&sk->sk_omem_alloc) + size < optmem_max) {
2617                 void *mem;
2618                 /* First do the add, to avoid the race if kmalloc
2619                  * might sleep.
2620                  */
2621                 atomic_add(size, &sk->sk_omem_alloc);
2622                 mem = kmalloc(size, priority);
2623                 if (mem)
2624                         return mem;
2625                 atomic_sub(size, &sk->sk_omem_alloc);
2626         }
2627         return NULL;
2628 }
2629 EXPORT_SYMBOL(sock_kmalloc);
2630
2631 /* Free an option memory block. Note, we actually want the inline
2632  * here as this allows gcc to detect the nullify and fold away the
2633  * condition entirely.
2634  */
2635 static inline void __sock_kfree_s(struct sock *sk, void *mem, int size,
2636                                   const bool nullify)
2637 {
2638         if (WARN_ON_ONCE(!mem))
2639                 return;
2640         if (nullify)
2641                 kfree_sensitive(mem);
2642         else
2643                 kfree(mem);
2644         atomic_sub(size, &sk->sk_omem_alloc);
2645 }
2646
2647 void sock_kfree_s(struct sock *sk, void *mem, int size)
2648 {
2649         __sock_kfree_s(sk, mem, size, false);
2650 }
2651 EXPORT_SYMBOL(sock_kfree_s);
2652
2653 void sock_kzfree_s(struct sock *sk, void *mem, int size)
2654 {
2655         __sock_kfree_s(sk, mem, size, true);
2656 }
2657 EXPORT_SYMBOL(sock_kzfree_s);
2658
2659 /* It is almost wait_for_tcp_memory minus release_sock/lock_sock.
2660    I think, these locks should be removed for datagram sockets.
2661  */
2662 static long sock_wait_for_wmem(struct sock *sk, long timeo)
2663 {
2664         DEFINE_WAIT(wait);
2665
2666         sk_clear_bit(SOCKWQ_ASYNC_NOSPACE, sk);
2667         for (;;) {
2668                 if (!timeo)
2669                         break;
2670                 if (signal_pending(current))
2671                         break;
2672                 set_bit(SOCK_NOSPACE, &sk->sk_socket->flags);
2673                 prepare_to_wait(sk_sleep(sk), &wait, TASK_INTERRUPTIBLE);
2674                 if (refcount_read(&sk->sk_wmem_alloc) < READ_ONCE(sk->sk_sndbuf))
2675                         break;
2676                 if (sk->sk_shutdown & SEND_SHUTDOWN)
2677                         break;
2678                 if (sk->sk_err)
2679                         break;
2680                 timeo = schedule_timeout(timeo);
2681         }
2682         finish_wait(sk_sleep(sk), &wait);
2683         return timeo;
2684 }
2685
2686
2687 /*
2688  *      Generic send/receive buffer handlers
2689  */
2690
2691 struct sk_buff *sock_alloc_send_pskb(struct sock *sk, unsigned long header_len,
2692                                      unsigned long data_len, int noblock,
2693                                      int *errcode, int max_page_order)
2694 {
2695         struct sk_buff *skb;
2696         long timeo;
2697         int err;
2698
2699         timeo = sock_sndtimeo(sk, noblock);
2700         for (;;) {
2701                 err = sock_error(sk);
2702                 if (err != 0)
2703                         goto failure;
2704
2705                 err = -EPIPE;
2706                 if (sk->sk_shutdown & SEND_SHUTDOWN)
2707                         goto failure;
2708
2709                 if (sk_wmem_alloc_get(sk) < READ_ONCE(sk->sk_sndbuf))
2710                         break;
2711
2712                 sk_set_bit(SOCKWQ_ASYNC_NOSPACE, sk);
2713                 set_bit(SOCK_NOSPACE, &sk->sk_socket->flags);
2714                 err = -EAGAIN;
2715                 if (!timeo)
2716                         goto failure;
2717                 if (signal_pending(current))
2718                         goto interrupted;
2719                 timeo = sock_wait_for_wmem(sk, timeo);
2720         }
2721         skb = alloc_skb_with_frags(header_len, data_len, max_page_order,
2722                                    errcode, sk->sk_allocation);
2723         if (skb)
2724                 skb_set_owner_w(skb, sk);
2725         return skb;
2726
2727 interrupted:
2728         err = sock_intr_errno(timeo);
2729 failure:
2730         *errcode = err;
2731         return NULL;
2732 }
2733 EXPORT_SYMBOL(sock_alloc_send_pskb);
2734
2735 int __sock_cmsg_send(struct sock *sk, struct msghdr *msg, struct cmsghdr *cmsg,
2736                      struct sockcm_cookie *sockc)
2737 {
2738         u32 tsflags;
2739
2740         switch (cmsg->cmsg_type) {
2741         case SO_MARK:
2742                 if (!ns_capable(sock_net(sk)->user_ns, CAP_NET_RAW) &&
2743                     !ns_capable(sock_net(sk)->user_ns, CAP_NET_ADMIN))
2744                         return -EPERM;
2745                 if (cmsg->cmsg_len != CMSG_LEN(sizeof(u32)))
2746                         return -EINVAL;
2747                 sockc->mark = *(u32 *)CMSG_DATA(cmsg);
2748                 break;
2749         case SO_TIMESTAMPING_OLD:
2750                 if (cmsg->cmsg_len != CMSG_LEN(sizeof(u32)))
2751                         return -EINVAL;
2752
2753                 tsflags = *(u32 *)CMSG_DATA(cmsg);
2754                 if (tsflags & ~SOF_TIMESTAMPING_TX_RECORD_MASK)
2755                         return -EINVAL;
2756
2757                 sockc->tsflags &= ~SOF_TIMESTAMPING_TX_RECORD_MASK;
2758                 sockc->tsflags |= tsflags;
2759                 break;
2760         case SCM_TXTIME:
2761                 if (!sock_flag(sk, SOCK_TXTIME))
2762                         return -EINVAL;
2763                 if (cmsg->cmsg_len != CMSG_LEN(sizeof(u64)))
2764                         return -EINVAL;
2765                 sockc->transmit_time = get_unaligned((u64 *)CMSG_DATA(cmsg));
2766                 break;
2767         /* SCM_RIGHTS and SCM_CREDENTIALS are semantically in SOL_UNIX. */
2768         case SCM_RIGHTS:
2769         case SCM_CREDENTIALS:
2770                 break;
2771         default:
2772                 return -EINVAL;
2773         }
2774         return 0;
2775 }
2776 EXPORT_SYMBOL(__sock_cmsg_send);
2777
2778 int sock_cmsg_send(struct sock *sk, struct msghdr *msg,
2779                    struct sockcm_cookie *sockc)
2780 {
2781         struct cmsghdr *cmsg;
2782         int ret;
2783
2784         for_each_cmsghdr(cmsg, msg) {
2785                 if (!CMSG_OK(msg, cmsg))
2786                         return -EINVAL;
2787                 if (cmsg->cmsg_level != SOL_SOCKET)
2788                         continue;
2789                 ret = __sock_cmsg_send(sk, msg, cmsg, sockc);
2790                 if (ret)
2791                         return ret;
2792         }
2793         return 0;
2794 }
2795 EXPORT_SYMBOL(sock_cmsg_send);
2796
2797 static void sk_enter_memory_pressure(struct sock *sk)
2798 {
2799         if (!sk->sk_prot->enter_memory_pressure)
2800                 return;
2801
2802         sk->sk_prot->enter_memory_pressure(sk);
2803 }
2804
2805 static void sk_leave_memory_pressure(struct sock *sk)
2806 {
2807         if (sk->sk_prot->leave_memory_pressure) {
2808                 INDIRECT_CALL_INET_1(sk->sk_prot->leave_memory_pressure,
2809                                      tcp_leave_memory_pressure, sk);
2810         } else {
2811                 unsigned long *memory_pressure = sk->sk_prot->memory_pressure;
2812
2813                 if (memory_pressure && READ_ONCE(*memory_pressure))
2814                         WRITE_ONCE(*memory_pressure, 0);
2815         }
2816 }
2817
2818 DEFINE_STATIC_KEY_FALSE(net_high_order_alloc_disable_key);
2819
2820 /**
2821  * skb_page_frag_refill - check that a page_frag contains enough room
2822  * @sz: minimum size of the fragment we want to get
2823  * @pfrag: pointer to page_frag
2824  * @gfp: priority for memory allocation
2825  *
2826  * Note: While this allocator tries to use high order pages, there is
2827  * no guarantee that allocations succeed. Therefore, @sz MUST be
2828  * less or equal than PAGE_SIZE.
2829  */
2830 bool skb_page_frag_refill(unsigned int sz, struct page_frag *pfrag, gfp_t gfp)
2831 {
2832         if (pfrag->page) {
2833                 if (page_ref_count(pfrag->page) == 1) {
2834                         pfrag->offset = 0;
2835                         return true;
2836                 }
2837                 if (pfrag->offset + sz <= pfrag->size)
2838                         return true;
2839                 put_page(pfrag->page);
2840         }
2841
2842         pfrag->offset = 0;
2843         if (SKB_FRAG_PAGE_ORDER &&
2844             !static_branch_unlikely(&net_high_order_alloc_disable_key)) {
2845                 /* Avoid direct reclaim but allow kswapd to wake */
2846                 pfrag->page = alloc_pages((gfp & ~__GFP_DIRECT_RECLAIM) |
2847                                           __GFP_COMP | __GFP_NOWARN |
2848                                           __GFP_NORETRY,
2849                                           SKB_FRAG_PAGE_ORDER);
2850                 if (likely(pfrag->page)) {
2851                         pfrag->size = PAGE_SIZE << SKB_FRAG_PAGE_ORDER;
2852                         return true;
2853                 }
2854         }
2855         pfrag->page = alloc_page(gfp);
2856         if (likely(pfrag->page)) {
2857                 pfrag->size = PAGE_SIZE;
2858                 return true;
2859         }
2860         return false;
2861 }
2862 EXPORT_SYMBOL(skb_page_frag_refill);
2863
2864 bool sk_page_frag_refill(struct sock *sk, struct page_frag *pfrag)
2865 {
2866         if (likely(skb_page_frag_refill(32U, pfrag, sk->sk_allocation)))
2867                 return true;
2868
2869         sk_enter_memory_pressure(sk);
2870         sk_stream_moderate_sndbuf(sk);
2871         return false;
2872 }
2873 EXPORT_SYMBOL(sk_page_frag_refill);
2874
2875 void __lock_sock(struct sock *sk)
2876         __releases(&sk->sk_lock.slock)
2877         __acquires(&sk->sk_lock.slock)
2878 {
2879         DEFINE_WAIT(wait);
2880
2881         for (;;) {
2882                 prepare_to_wait_exclusive(&sk->sk_lock.wq, &wait,
2883                                         TASK_UNINTERRUPTIBLE);
2884                 spin_unlock_bh(&sk->sk_lock.slock);
2885                 schedule();
2886                 spin_lock_bh(&sk->sk_lock.slock);
2887                 if (!sock_owned_by_user(sk))
2888                         break;
2889         }
2890         finish_wait(&sk->sk_lock.wq, &wait);
2891 }
2892
2893 void __release_sock(struct sock *sk)
2894         __releases(&sk->sk_lock.slock)
2895         __acquires(&sk->sk_lock.slock)
2896 {
2897         struct sk_buff *skb, *next;
2898
2899         while ((skb = sk->sk_backlog.head) != NULL) {
2900                 sk->sk_backlog.head = sk->sk_backlog.tail = NULL;
2901
2902                 spin_unlock_bh(&sk->sk_lock.slock);
2903
2904                 do {
2905                         next = skb->next;
2906                         prefetch(next);
2907                         DEBUG_NET_WARN_ON_ONCE(skb_dst_is_noref(skb));
2908                         skb_mark_not_on_list(skb);
2909                         sk_backlog_rcv(sk, skb);
2910
2911                         cond_resched();
2912
2913                         skb = next;
2914                 } while (skb != NULL);
2915
2916                 spin_lock_bh(&sk->sk_lock.slock);
2917         }
2918
2919         /*
2920          * Doing the zeroing here guarantee we can not loop forever
2921          * while a wild producer attempts to flood us.
2922          */
2923         sk->sk_backlog.len = 0;
2924 }
2925
2926 void __sk_flush_backlog(struct sock *sk)
2927 {
2928         spin_lock_bh(&sk->sk_lock.slock);
2929         __release_sock(sk);
2930         spin_unlock_bh(&sk->sk_lock.slock);
2931 }
2932 EXPORT_SYMBOL_GPL(__sk_flush_backlog);
2933
2934 /**
2935  * sk_wait_data - wait for data to arrive at sk_receive_queue
2936  * @sk:    sock to wait on
2937  * @timeo: for how long
2938  * @skb:   last skb seen on sk_receive_queue
2939  *
2940  * Now socket state including sk->sk_err is changed only under lock,
2941  * hence we may omit checks after joining wait queue.
2942  * We check receive queue before schedule() only as optimization;
2943  * it is very likely that release_sock() added new data.
2944  */
2945 int sk_wait_data(struct sock *sk, long *timeo, const struct sk_buff *skb)
2946 {
2947         DEFINE_WAIT_FUNC(wait, woken_wake_function);
2948         int rc;
2949
2950         add_wait_queue(sk_sleep(sk), &wait);
2951         sk_set_bit(SOCKWQ_ASYNC_WAITDATA, sk);
2952         rc = sk_wait_event(sk, timeo, skb_peek_tail(&sk->sk_receive_queue) != skb, &wait);
2953         sk_clear_bit(SOCKWQ_ASYNC_WAITDATA, sk);
2954         remove_wait_queue(sk_sleep(sk), &wait);
2955         return rc;
2956 }
2957 EXPORT_SYMBOL(sk_wait_data);
2958
2959 /**
2960  *      __sk_mem_raise_allocated - increase memory_allocated
2961  *      @sk: socket
2962  *      @size: memory size to allocate
2963  *      @amt: pages to allocate
2964  *      @kind: allocation type
2965  *
2966  *      Similar to __sk_mem_schedule(), but does not update sk_forward_alloc
2967  */
2968 int __sk_mem_raise_allocated(struct sock *sk, int size, int amt, int kind)
2969 {
2970         bool memcg_charge = mem_cgroup_sockets_enabled && sk->sk_memcg;
2971         struct proto *prot = sk->sk_prot;
2972         bool charged = true;
2973         long allocated;
2974
2975         sk_memory_allocated_add(sk, amt);
2976         allocated = sk_memory_allocated(sk);
2977         if (memcg_charge &&
2978             !(charged = mem_cgroup_charge_skmem(sk->sk_memcg, amt,
2979                                                 gfp_memcg_charge())))
2980                 goto suppress_allocation;
2981
2982         /* Under limit. */
2983         if (allocated <= sk_prot_mem_limits(sk, 0)) {
2984                 sk_leave_memory_pressure(sk);
2985                 return 1;
2986         }
2987
2988         /* Under pressure. */
2989         if (allocated > sk_prot_mem_limits(sk, 1))
2990                 sk_enter_memory_pressure(sk);
2991
2992         /* Over hard limit. */
2993         if (allocated > sk_prot_mem_limits(sk, 2))
2994                 goto suppress_allocation;
2995
2996         /* guarantee minimum buffer size under pressure */
2997         if (kind == SK_MEM_RECV) {
2998                 if (atomic_read(&sk->sk_rmem_alloc) < sk_get_rmem0(sk, prot))
2999                         return 1;
3000
3001         } else { /* SK_MEM_SEND */
3002                 int wmem0 = sk_get_wmem0(sk, prot);
3003
3004                 if (sk->sk_type == SOCK_STREAM) {
3005                         if (sk->sk_wmem_queued < wmem0)
3006                                 return 1;
3007                 } else if (refcount_read(&sk->sk_wmem_alloc) < wmem0) {
3008                                 return 1;
3009                 }
3010         }
3011
3012         if (sk_has_memory_pressure(sk)) {
3013                 u64 alloc;
3014
3015                 if (!sk_under_memory_pressure(sk))
3016                         return 1;
3017                 alloc = sk_sockets_allocated_read_positive(sk);
3018                 if (sk_prot_mem_limits(sk, 2) > alloc *
3019                     sk_mem_pages(sk->sk_wmem_queued +
3020                                  atomic_read(&sk->sk_rmem_alloc) +
3021                                  sk->sk_forward_alloc))
3022                         return 1;
3023         }
3024
3025 suppress_allocation:
3026
3027         if (kind == SK_MEM_SEND && sk->sk_type == SOCK_STREAM) {
3028                 sk_stream_moderate_sndbuf(sk);
3029
3030                 /* Fail only if socket is _under_ its sndbuf.
3031                  * In this case we cannot block, so that we have to fail.
3032                  */
3033                 if (sk->sk_wmem_queued + size >= sk->sk_sndbuf) {
3034                         /* Force charge with __GFP_NOFAIL */
3035                         if (memcg_charge && !charged) {
3036                                 mem_cgroup_charge_skmem(sk->sk_memcg, amt,
3037                                         gfp_memcg_charge() | __GFP_NOFAIL);
3038                         }
3039                         return 1;
3040                 }
3041         }
3042
3043         if (kind == SK_MEM_SEND || (kind == SK_MEM_RECV && charged))
3044                 trace_sock_exceed_buf_limit(sk, prot, allocated, kind);
3045
3046         sk_memory_allocated_sub(sk, amt);
3047
3048         if (memcg_charge && charged)
3049                 mem_cgroup_uncharge_skmem(sk->sk_memcg, amt);
3050
3051         return 0;
3052 }
3053
3054 /**
3055  *      __sk_mem_schedule - increase sk_forward_alloc and memory_allocated
3056  *      @sk: socket
3057  *      @size: memory size to allocate
3058  *      @kind: allocation type
3059  *
3060  *      If kind is SK_MEM_SEND, it means wmem allocation. Otherwise it means
3061  *      rmem allocation. This function assumes that protocols which have
3062  *      memory_pressure use sk_wmem_queued as write buffer accounting.
3063  */
3064 int __sk_mem_schedule(struct sock *sk, int size, int kind)
3065 {
3066         int ret, amt = sk_mem_pages(size);
3067
3068         sk->sk_forward_alloc += amt << PAGE_SHIFT;
3069         ret = __sk_mem_raise_allocated(sk, size, amt, kind);
3070         if (!ret)
3071                 sk->sk_forward_alloc -= amt << PAGE_SHIFT;
3072         return ret;
3073 }
3074 EXPORT_SYMBOL(__sk_mem_schedule);
3075
3076 /**
3077  *      __sk_mem_reduce_allocated - reclaim memory_allocated
3078  *      @sk: socket
3079  *      @amount: number of quanta
3080  *
3081  *      Similar to __sk_mem_reclaim(), but does not update sk_forward_alloc
3082  */
3083 void __sk_mem_reduce_allocated(struct sock *sk, int amount)
3084 {
3085         sk_memory_allocated_sub(sk, amount);
3086
3087         if (mem_cgroup_sockets_enabled && sk->sk_memcg)
3088                 mem_cgroup_uncharge_skmem(sk->sk_memcg, amount);
3089
3090         if (sk_under_memory_pressure(sk) &&
3091             (sk_memory_allocated(sk) < sk_prot_mem_limits(sk, 0)))
3092                 sk_leave_memory_pressure(sk);
3093 }
3094
3095 /**
3096  *      __sk_mem_reclaim - reclaim sk_forward_alloc and memory_allocated
3097  *      @sk: socket
3098  *      @amount: number of bytes (rounded down to a PAGE_SIZE multiple)
3099  */
3100 void __sk_mem_reclaim(struct sock *sk, int amount)
3101 {
3102         amount >>= PAGE_SHIFT;
3103         sk->sk_forward_alloc -= amount << PAGE_SHIFT;
3104         __sk_mem_reduce_allocated(sk, amount);
3105 }
3106 EXPORT_SYMBOL(__sk_mem_reclaim);
3107
3108 int sk_set_peek_off(struct sock *sk, int val)
3109 {
3110         sk->sk_peek_off = val;
3111         return 0;
3112 }
3113 EXPORT_SYMBOL_GPL(sk_set_peek_off);
3114
3115 /*
3116  * Set of default routines for initialising struct proto_ops when
3117  * the protocol does not support a particular function. In certain
3118  * cases where it makes no sense for a protocol to have a "do nothing"
3119  * function, some default processing is provided.
3120  */
3121
3122 int sock_no_bind(struct socket *sock, struct sockaddr *saddr, int len)
3123 {
3124         return -EOPNOTSUPP;
3125 }
3126 EXPORT_SYMBOL(sock_no_bind);
3127
3128 int sock_no_connect(struct socket *sock, struct sockaddr *saddr,
3129                     int len, int flags)
3130 {
3131         return -EOPNOTSUPP;
3132 }
3133 EXPORT_SYMBOL(sock_no_connect);
3134
3135 int sock_no_socketpair(struct socket *sock1, struct socket *sock2)
3136 {
3137         return -EOPNOTSUPP;
3138 }
3139 EXPORT_SYMBOL(sock_no_socketpair);
3140
3141 int sock_no_accept(struct socket *sock, struct socket *newsock, int flags,
3142                    bool kern)
3143 {
3144         return -EOPNOTSUPP;
3145 }
3146 EXPORT_SYMBOL(sock_no_accept);
3147
3148 int sock_no_getname(struct socket *sock, struct sockaddr *saddr,
3149                     int peer)
3150 {
3151         return -EOPNOTSUPP;
3152 }
3153 EXPORT_SYMBOL(sock_no_getname);
3154
3155 int sock_no_ioctl(struct socket *sock, unsigned int cmd, unsigned long arg)
3156 {
3157         return -EOPNOTSUPP;
3158 }
3159 EXPORT_SYMBOL(sock_no_ioctl);
3160
3161 int sock_no_listen(struct socket *sock, int backlog)
3162 {
3163         return -EOPNOTSUPP;
3164 }
3165 EXPORT_SYMBOL(sock_no_listen);
3166
3167 int sock_no_shutdown(struct socket *sock, int how)
3168 {
3169         return -EOPNOTSUPP;
3170 }
3171 EXPORT_SYMBOL(sock_no_shutdown);
3172
3173 int sock_no_sendmsg(struct socket *sock, struct msghdr *m, size_t len)
3174 {
3175         return -EOPNOTSUPP;
3176 }
3177 EXPORT_SYMBOL(sock_no_sendmsg);
3178
3179 int sock_no_sendmsg_locked(struct sock *sk, struct msghdr *m, size_t len)
3180 {
3181         return -EOPNOTSUPP;
3182 }
3183 EXPORT_SYMBOL(sock_no_sendmsg_locked);
3184
3185 int sock_no_recvmsg(struct socket *sock, struct msghdr *m, size_t len,
3186                     int flags)
3187 {
3188         return -EOPNOTSUPP;
3189 }
3190 EXPORT_SYMBOL(sock_no_recvmsg);
3191
3192 int sock_no_mmap(struct file *file, struct socket *sock, struct vm_area_struct *vma)
3193 {
3194         /* Mirror missing mmap method error code */
3195         return -ENODEV;
3196 }
3197 EXPORT_SYMBOL(sock_no_mmap);
3198
3199 /*
3200  * When a file is received (via SCM_RIGHTS, etc), we must bump the
3201  * various sock-based usage counts.
3202  */
3203 void __receive_sock(struct file *file)
3204 {
3205         struct socket *sock;
3206
3207         sock = sock_from_file(file);
3208         if (sock) {
3209                 sock_update_netprioidx(&sock->sk->sk_cgrp_data);
3210                 sock_update_classid(&sock->sk->sk_cgrp_data);
3211         }
3212 }
3213
3214 ssize_t sock_no_sendpage(struct socket *sock, struct page *page, int offset, size_t size, int flags)
3215 {
3216         ssize_t res;
3217         struct msghdr msg = {.msg_flags = flags};
3218         struct kvec iov;
3219         char *kaddr = kmap(page);
3220         iov.iov_base = kaddr + offset;
3221         iov.iov_len = size;
3222         res = kernel_sendmsg(sock, &msg, &iov, 1, size);
3223         kunmap(page);
3224         return res;
3225 }
3226 EXPORT_SYMBOL(sock_no_sendpage);
3227
3228 ssize_t sock_no_sendpage_locked(struct sock *sk, struct page *page,
3229                                 int offset, size_t size, int flags)
3230 {
3231         ssize_t res;
3232         struct msghdr msg = {.msg_flags = flags};
3233         struct kvec iov;
3234         char *kaddr = kmap(page);
3235
3236         iov.iov_base = kaddr + offset;
3237         iov.iov_len = size;
3238         res = kernel_sendmsg_locked(sk, &msg, &iov, 1, size);
3239         kunmap(page);
3240         return res;
3241 }
3242 EXPORT_SYMBOL(sock_no_sendpage_locked);
3243
3244 /*
3245  *      Default Socket Callbacks
3246  */
3247
3248 static void sock_def_wakeup(struct sock *sk)
3249 {
3250         struct socket_wq *wq;
3251
3252         rcu_read_lock();
3253         wq = rcu_dereference(sk->sk_wq);
3254         if (skwq_has_sleeper(wq))
3255                 wake_up_interruptible_all(&wq->wait);
3256         rcu_read_unlock();
3257 }
3258
3259 static void sock_def_error_report(struct sock *sk)
3260 {
3261         struct socket_wq *wq;
3262
3263         rcu_read_lock();
3264         wq = rcu_dereference(sk->sk_wq);
3265         if (skwq_has_sleeper(wq))
3266                 wake_up_interruptible_poll(&wq->wait, EPOLLERR);
3267         sk_wake_async(sk, SOCK_WAKE_IO, POLL_ERR);
3268         rcu_read_unlock();
3269 }
3270
3271 void sock_def_readable(struct sock *sk)
3272 {
3273         struct socket_wq *wq;
3274
3275         rcu_read_lock();
3276         wq = rcu_dereference(sk->sk_wq);
3277         if (skwq_has_sleeper(wq))
3278                 wake_up_interruptible_sync_poll(&wq->wait, EPOLLIN | EPOLLPRI |
3279                                                 EPOLLRDNORM | EPOLLRDBAND);
3280         sk_wake_async(sk, SOCK_WAKE_WAITD, POLL_IN);
3281         rcu_read_unlock();
3282 }
3283
3284 static void sock_def_write_space(struct sock *sk)
3285 {
3286         struct socket_wq *wq;
3287
3288         rcu_read_lock();
3289
3290         /* Do not wake up a writer until he can make "significant"
3291          * progress.  --DaveM
3292          */
3293         if (sock_writeable(sk)) {
3294                 wq = rcu_dereference(sk->sk_wq);
3295                 if (skwq_has_sleeper(wq))
3296                         wake_up_interruptible_sync_poll(&wq->wait, EPOLLOUT |
3297                                                 EPOLLWRNORM | EPOLLWRBAND);
3298
3299                 /* Should agree with poll, otherwise some programs break */
3300                 sk_wake_async(sk, SOCK_WAKE_SPACE, POLL_OUT);
3301         }
3302
3303         rcu_read_unlock();
3304 }
3305
3306 /* An optimised version of sock_def_write_space(), should only be called
3307  * for SOCK_RCU_FREE sockets under RCU read section and after putting
3308  * ->sk_wmem_alloc.
3309  */
3310 static void sock_def_write_space_wfree(struct sock *sk)
3311 {
3312         /* Do not wake up a writer until he can make "significant"
3313          * progress.  --DaveM
3314          */
3315         if (sock_writeable(sk)) {
3316                 struct socket_wq *wq = rcu_dereference(sk->sk_wq);
3317
3318                 /* rely on refcount_sub from sock_wfree() */
3319                 smp_mb__after_atomic();
3320                 if (wq && waitqueue_active(&wq->wait))
3321                         wake_up_interruptible_sync_poll(&wq->wait, EPOLLOUT |
3322                                                 EPOLLWRNORM | EPOLLWRBAND);
3323
3324                 /* Should agree with poll, otherwise some programs break */
3325                 sk_wake_async(sk, SOCK_WAKE_SPACE, POLL_OUT);
3326         }
3327 }
3328
3329 static void sock_def_destruct(struct sock *sk)
3330 {
3331 }
3332
3333 void sk_send_sigurg(struct sock *sk)
3334 {
3335         if (sk->sk_socket && sk->sk_socket->file)
3336                 if (send_sigurg(&sk->sk_socket->file->f_owner))
3337                         sk_wake_async(sk, SOCK_WAKE_URG, POLL_PRI);
3338 }
3339 EXPORT_SYMBOL(sk_send_sigurg);
3340
3341 void sk_reset_timer(struct sock *sk, struct timer_list* timer,
3342                     unsigned long expires)
3343 {
3344         if (!mod_timer(timer, expires))
3345                 sock_hold(sk);
3346 }
3347 EXPORT_SYMBOL(sk_reset_timer);
3348
3349 void sk_stop_timer(struct sock *sk, struct timer_list* timer)
3350 {
3351         if (del_timer(timer))
3352                 __sock_put(sk);
3353 }
3354 EXPORT_SYMBOL(sk_stop_timer);
3355
3356 void sk_stop_timer_sync(struct sock *sk, struct timer_list *timer)
3357 {
3358         if (del_timer_sync(timer))
3359                 __sock_put(sk);
3360 }
3361 EXPORT_SYMBOL(sk_stop_timer_sync);
3362
3363 void sock_init_data_uid(struct socket *sock, struct sock *sk, kuid_t uid)
3364 {
3365         sk_init_common(sk);
3366         sk->sk_send_head        =       NULL;
3367
3368         timer_setup(&sk->sk_timer, NULL, 0);
3369
3370         sk->sk_allocation       =       GFP_KERNEL;
3371         sk->sk_rcvbuf           =       READ_ONCE(sysctl_rmem_default);
3372         sk->sk_sndbuf           =       READ_ONCE(sysctl_wmem_default);
3373         sk->sk_state            =       TCP_CLOSE;
3374         sk_set_socket(sk, sock);
3375
3376         sock_set_flag(sk, SOCK_ZAPPED);
3377
3378         if (sock) {
3379                 sk->sk_type     =       sock->type;
3380                 RCU_INIT_POINTER(sk->sk_wq, &sock->wq);
3381                 sock->sk        =       sk;
3382         } else {
3383                 RCU_INIT_POINTER(sk->sk_wq, NULL);
3384         }
3385         sk->sk_uid      =       uid;
3386
3387         rwlock_init(&sk->sk_callback_lock);
3388         if (sk->sk_kern_sock)
3389                 lockdep_set_class_and_name(
3390                         &sk->sk_callback_lock,
3391                         af_kern_callback_keys + sk->sk_family,
3392                         af_family_kern_clock_key_strings[sk->sk_family]);
3393         else
3394                 lockdep_set_class_and_name(
3395                         &sk->sk_callback_lock,
3396                         af_callback_keys + sk->sk_family,
3397                         af_family_clock_key_strings[sk->sk_family]);
3398
3399         sk->sk_state_change     =       sock_def_wakeup;
3400         sk->sk_data_ready       =       sock_def_readable;
3401         sk->sk_write_space      =       sock_def_write_space;
3402         sk->sk_error_report     =       sock_def_error_report;
3403         sk->sk_destruct         =       sock_def_destruct;
3404
3405         sk->sk_frag.page        =       NULL;
3406         sk->sk_frag.offset      =       0;
3407         sk->sk_peek_off         =       -1;
3408
3409         sk->sk_peer_pid         =       NULL;
3410         sk->sk_peer_cred        =       NULL;
3411         spin_lock_init(&sk->sk_peer_lock);
3412
3413         sk->sk_write_pending    =       0;
3414         sk->sk_rcvlowat         =       1;
3415         sk->sk_rcvtimeo         =       MAX_SCHEDULE_TIMEOUT;
3416         sk->sk_sndtimeo         =       MAX_SCHEDULE_TIMEOUT;
3417
3418         sk->sk_stamp = SK_DEFAULT_STAMP;
3419 #if BITS_PER_LONG==32
3420         seqlock_init(&sk->sk_stamp_seq);
3421 #endif
3422         atomic_set(&sk->sk_zckey, 0);
3423
3424 #ifdef CONFIG_NET_RX_BUSY_POLL
3425         sk->sk_napi_id          =       0;
3426         sk->sk_ll_usec          =       READ_ONCE(sysctl_net_busy_read);
3427 #endif
3428
3429         sk->sk_max_pacing_rate = ~0UL;
3430         sk->sk_pacing_rate = ~0UL;
3431         WRITE_ONCE(sk->sk_pacing_shift, 10);
3432         sk->sk_incoming_cpu = -1;
3433
3434         sk_rx_queue_clear(sk);
3435         /*
3436          * Before updating sk_refcnt, we must commit prior changes to memory
3437          * (Documentation/RCU/rculist_nulls.rst for details)
3438          */
3439         smp_wmb();
3440         refcount_set(&sk->sk_refcnt, 1);
3441         atomic_set(&sk->sk_drops, 0);
3442 }
3443 EXPORT_SYMBOL(sock_init_data_uid);
3444
3445 void sock_init_data(struct socket *sock, struct sock *sk)
3446 {
3447         kuid_t uid = sock ?
3448                 SOCK_INODE(sock)->i_uid :
3449                 make_kuid(sock_net(sk)->user_ns, 0);
3450
3451         sock_init_data_uid(sock, sk, uid);
3452 }
3453 EXPORT_SYMBOL(sock_init_data);
3454
3455 void lock_sock_nested(struct sock *sk, int subclass)
3456 {
3457         /* The sk_lock has mutex_lock() semantics here. */
3458         mutex_acquire(&sk->sk_lock.dep_map, subclass, 0, _RET_IP_);
3459
3460         might_sleep();
3461         spin_lock_bh(&sk->sk_lock.slock);
3462         if (sock_owned_by_user_nocheck(sk))
3463                 __lock_sock(sk);
3464         sk->sk_lock.owned = 1;
3465         spin_unlock_bh(&sk->sk_lock.slock);
3466 }
3467 EXPORT_SYMBOL(lock_sock_nested);
3468
3469 void release_sock(struct sock *sk)
3470 {
3471         spin_lock_bh(&sk->sk_lock.slock);
3472         if (sk->sk_backlog.tail)
3473                 __release_sock(sk);
3474
3475         /* Warning : release_cb() might need to release sk ownership,
3476          * ie call sock_release_ownership(sk) before us.
3477          */
3478         if (sk->sk_prot->release_cb)
3479                 sk->sk_prot->release_cb(sk);
3480
3481         sock_release_ownership(sk);
3482         if (waitqueue_active(&sk->sk_lock.wq))
3483                 wake_up(&sk->sk_lock.wq);
3484         spin_unlock_bh(&sk->sk_lock.slock);
3485 }
3486 EXPORT_SYMBOL(release_sock);
3487
3488 bool __lock_sock_fast(struct sock *sk) __acquires(&sk->sk_lock.slock)
3489 {
3490         might_sleep();
3491         spin_lock_bh(&sk->sk_lock.slock);
3492
3493         if (!sock_owned_by_user_nocheck(sk)) {
3494                 /*
3495                  * Fast path return with bottom halves disabled and
3496                  * sock::sk_lock.slock held.
3497                  *
3498                  * The 'mutex' is not contended and holding
3499                  * sock::sk_lock.slock prevents all other lockers to
3500                  * proceed so the corresponding unlock_sock_fast() can
3501                  * avoid the slow path of release_sock() completely and
3502                  * just release slock.
3503                  *
3504                  * From a semantical POV this is equivalent to 'acquiring'
3505                  * the 'mutex', hence the corresponding lockdep
3506                  * mutex_release() has to happen in the fast path of
3507                  * unlock_sock_fast().
3508                  */
3509                 return false;
3510         }
3511
3512         __lock_sock(sk);
3513         sk->sk_lock.owned = 1;
3514         __acquire(&sk->sk_lock.slock);
3515         spin_unlock_bh(&sk->sk_lock.slock);
3516         return true;
3517 }
3518 EXPORT_SYMBOL(__lock_sock_fast);
3519
3520 int sock_gettstamp(struct socket *sock, void __user *userstamp,
3521                    bool timeval, bool time32)
3522 {
3523         struct sock *sk = sock->sk;
3524         struct timespec64 ts;
3525
3526         sock_enable_timestamp(sk, SOCK_TIMESTAMP);
3527         ts = ktime_to_timespec64(sock_read_timestamp(sk));
3528         if (ts.tv_sec == -1)
3529                 return -ENOENT;
3530         if (ts.tv_sec == 0) {
3531                 ktime_t kt = ktime_get_real();
3532                 sock_write_timestamp(sk, kt);
3533                 ts = ktime_to_timespec64(kt);
3534         }
3535
3536         if (timeval)
3537                 ts.tv_nsec /= 1000;
3538
3539 #ifdef CONFIG_COMPAT_32BIT_TIME
3540         if (time32)
3541                 return put_old_timespec32(&ts, userstamp);
3542 #endif
3543 #ifdef CONFIG_SPARC64
3544         /* beware of padding in sparc64 timeval */
3545         if (timeval && !in_compat_syscall()) {
3546                 struct __kernel_old_timeval __user tv = {
3547                         .tv_sec = ts.tv_sec,
3548                         .tv_usec = ts.tv_nsec,
3549                 };
3550                 if (copy_to_user(userstamp, &tv, sizeof(tv)))
3551                         return -EFAULT;
3552                 return 0;
3553         }
3554 #endif
3555         return put_timespec64(&ts, userstamp);
3556 }
3557 EXPORT_SYMBOL(sock_gettstamp);
3558
3559 void sock_enable_timestamp(struct sock *sk, enum sock_flags flag)
3560 {
3561         if (!sock_flag(sk, flag)) {
3562                 unsigned long previous_flags = sk->sk_flags;
3563
3564                 sock_set_flag(sk, flag);
3565                 /*
3566                  * we just set one of the two flags which require net
3567                  * time stamping, but time stamping might have been on
3568                  * already because of the other one
3569                  */
3570                 if (sock_needs_netstamp(sk) &&
3571                     !(previous_flags & SK_FLAGS_TIMESTAMP))
3572                         net_enable_timestamp();
3573         }
3574 }
3575
3576 int sock_recv_errqueue(struct sock *sk, struct msghdr *msg, int len,
3577                        int level, int type)
3578 {
3579         struct sock_exterr_skb *serr;
3580         struct sk_buff *skb;
3581         int copied, err;
3582
3583         err = -EAGAIN;
3584         skb = sock_dequeue_err_skb(sk);
3585         if (skb == NULL)
3586                 goto out;
3587
3588         copied = skb->len;
3589         if (copied > len) {
3590                 msg->msg_flags |= MSG_TRUNC;
3591                 copied = len;
3592         }
3593         err = skb_copy_datagram_msg(skb, 0, msg, copied);
3594         if (err)
3595                 goto out_free_skb;
3596
3597         sock_recv_timestamp(msg, sk, skb);
3598
3599         serr = SKB_EXT_ERR(skb);
3600         put_cmsg(msg, level, type, sizeof(serr->ee), &serr->ee);
3601
3602         msg->msg_flags |= MSG_ERRQUEUE;
3603         err = copied;
3604
3605 out_free_skb:
3606         kfree_skb(skb);
3607 out:
3608         return err;
3609 }
3610 EXPORT_SYMBOL(sock_recv_errqueue);
3611
3612 /*
3613  *      Get a socket option on an socket.
3614  *
3615  *      FIX: POSIX 1003.1g is very ambiguous here. It states that
3616  *      asynchronous errors should be reported by getsockopt. We assume
3617  *      this means if you specify SO_ERROR (otherwise whats the point of it).
3618  */
3619 int sock_common_getsockopt(struct socket *sock, int level, int optname,
3620                            char __user *optval, int __user *optlen)
3621 {
3622         struct sock *sk = sock->sk;
3623
3624         /* IPV6_ADDRFORM can change sk->sk_prot under us. */
3625         return READ_ONCE(sk->sk_prot)->getsockopt(sk, level, optname, optval, optlen);
3626 }
3627 EXPORT_SYMBOL(sock_common_getsockopt);
3628
3629 int sock_common_recvmsg(struct socket *sock, struct msghdr *msg, size_t size,
3630                         int flags)
3631 {
3632         struct sock *sk = sock->sk;
3633         int addr_len = 0;
3634         int err;
3635
3636         err = sk->sk_prot->recvmsg(sk, msg, size, flags, &addr_len);
3637         if (err >= 0)
3638                 msg->msg_namelen = addr_len;
3639         return err;
3640 }
3641 EXPORT_SYMBOL(sock_common_recvmsg);
3642
3643 /*
3644  *      Set socket options on an inet socket.
3645  */
3646 int sock_common_setsockopt(struct socket *sock, int level, int optname,
3647                            sockptr_t optval, unsigned int optlen)
3648 {
3649         struct sock *sk = sock->sk;
3650
3651         /* IPV6_ADDRFORM can change sk->sk_prot under us. */
3652         return READ_ONCE(sk->sk_prot)->setsockopt(sk, level, optname, optval, optlen);
3653 }
3654 EXPORT_SYMBOL(sock_common_setsockopt);
3655
3656 void sk_common_release(struct sock *sk)
3657 {
3658         if (sk->sk_prot->destroy)
3659                 sk->sk_prot->destroy(sk);
3660
3661         /*
3662          * Observation: when sk_common_release is called, processes have
3663          * no access to socket. But net still has.
3664          * Step one, detach it from networking:
3665          *
3666          * A. Remove from hash tables.
3667          */
3668
3669         sk->sk_prot->unhash(sk);
3670
3671         /*
3672          * In this point socket cannot receive new packets, but it is possible
3673          * that some packets are in flight because some CPU runs receiver and
3674          * did hash table lookup before we unhashed socket. They will achieve
3675          * receive queue and will be purged by socket destructor.
3676          *
3677          * Also we still have packets pending on receive queue and probably,
3678          * our own packets waiting in device queues. sock_destroy will drain
3679          * receive queue, but transmitted packets will delay socket destruction
3680          * until the last reference will be released.
3681          */
3682
3683         sock_orphan(sk);
3684
3685         xfrm_sk_free_policy(sk);
3686
3687         sk_refcnt_debug_release(sk);
3688
3689         sock_put(sk);
3690 }
3691 EXPORT_SYMBOL(sk_common_release);
3692
3693 void sk_get_meminfo(const struct sock *sk, u32 *mem)
3694 {
3695         memset(mem, 0, sizeof(*mem) * SK_MEMINFO_VARS);
3696
3697         mem[SK_MEMINFO_RMEM_ALLOC] = sk_rmem_alloc_get(sk);
3698         mem[SK_MEMINFO_RCVBUF] = READ_ONCE(sk->sk_rcvbuf);
3699         mem[SK_MEMINFO_WMEM_ALLOC] = sk_wmem_alloc_get(sk);
3700         mem[SK_MEMINFO_SNDBUF] = READ_ONCE(sk->sk_sndbuf);
3701         mem[SK_MEMINFO_FWD_ALLOC] = sk->sk_forward_alloc;
3702         mem[SK_MEMINFO_WMEM_QUEUED] = READ_ONCE(sk->sk_wmem_queued);
3703         mem[SK_MEMINFO_OPTMEM] = atomic_read(&sk->sk_omem_alloc);
3704         mem[SK_MEMINFO_BACKLOG] = READ_ONCE(sk->sk_backlog.len);
3705         mem[SK_MEMINFO_DROPS] = atomic_read(&sk->sk_drops);
3706 }
3707
3708 #ifdef CONFIG_PROC_FS
3709 static DECLARE_BITMAP(proto_inuse_idx, PROTO_INUSE_NR);
3710
3711 int sock_prot_inuse_get(struct net *net, struct proto *prot)
3712 {
3713         int cpu, idx = prot->inuse_idx;
3714         int res = 0;
3715
3716         for_each_possible_cpu(cpu)
3717                 res += per_cpu_ptr(net->core.prot_inuse, cpu)->val[idx];
3718
3719         return res >= 0 ? res : 0;
3720 }
3721 EXPORT_SYMBOL_GPL(sock_prot_inuse_get);
3722
3723 int sock_inuse_get(struct net *net)
3724 {
3725         int cpu, res = 0;
3726
3727         for_each_possible_cpu(cpu)
3728                 res += per_cpu_ptr(net->core.prot_inuse, cpu)->all;
3729
3730         return res;
3731 }
3732
3733 EXPORT_SYMBOL_GPL(sock_inuse_get);
3734
3735 static int __net_init sock_inuse_init_net(struct net *net)
3736 {
3737         net->core.prot_inuse = alloc_percpu(struct prot_inuse);
3738         if (net->core.prot_inuse == NULL)
3739                 return -ENOMEM;
3740         return 0;
3741 }
3742
3743 static void __net_exit sock_inuse_exit_net(struct net *net)
3744 {
3745         free_percpu(net->core.prot_inuse);
3746 }
3747
3748 static struct pernet_operations net_inuse_ops = {
3749         .init = sock_inuse_init_net,
3750         .exit = sock_inuse_exit_net,
3751 };
3752
3753 static __init int net_inuse_init(void)
3754 {
3755         if (register_pernet_subsys(&net_inuse_ops))
3756                 panic("Cannot initialize net inuse counters");
3757
3758         return 0;
3759 }
3760
3761 core_initcall(net_inuse_init);
3762
3763 static int assign_proto_idx(struct proto *prot)
3764 {
3765         prot->inuse_idx = find_first_zero_bit(proto_inuse_idx, PROTO_INUSE_NR);
3766
3767         if (unlikely(prot->inuse_idx == PROTO_INUSE_NR - 1)) {
3768                 pr_err("PROTO_INUSE_NR exhausted\n");
3769                 return -ENOSPC;
3770         }
3771
3772         set_bit(prot->inuse_idx, proto_inuse_idx);
3773         return 0;
3774 }
3775
3776 static void release_proto_idx(struct proto *prot)
3777 {
3778         if (prot->inuse_idx != PROTO_INUSE_NR - 1)
3779                 clear_bit(prot->inuse_idx, proto_inuse_idx);
3780 }
3781 #else
3782 static inline int assign_proto_idx(struct proto *prot)
3783 {
3784         return 0;
3785 }
3786
3787 static inline void release_proto_idx(struct proto *prot)
3788 {
3789 }
3790
3791 #endif
3792
3793 static void tw_prot_cleanup(struct timewait_sock_ops *twsk_prot)
3794 {
3795         if (!twsk_prot)
3796                 return;
3797         kfree(twsk_prot->twsk_slab_name);
3798         twsk_prot->twsk_slab_name = NULL;
3799         kmem_cache_destroy(twsk_prot->twsk_slab);
3800         twsk_prot->twsk_slab = NULL;
3801 }
3802
3803 static int tw_prot_init(const struct proto *prot)
3804 {
3805         struct timewait_sock_ops *twsk_prot = prot->twsk_prot;
3806
3807         if (!twsk_prot)
3808                 return 0;
3809
3810         twsk_prot->twsk_slab_name = kasprintf(GFP_KERNEL, "tw_sock_%s",
3811                                               prot->name);
3812         if (!twsk_prot->twsk_slab_name)
3813                 return -ENOMEM;
3814
3815         twsk_prot->twsk_slab =
3816                 kmem_cache_create(twsk_prot->twsk_slab_name,
3817                                   twsk_prot->twsk_obj_size, 0,
3818                                   SLAB_ACCOUNT | prot->slab_flags,
3819                                   NULL);
3820         if (!twsk_prot->twsk_slab) {
3821                 pr_crit("%s: Can't create timewait sock SLAB cache!\n",
3822                         prot->name);
3823                 return -ENOMEM;
3824         }
3825
3826         return 0;
3827 }
3828
3829 static void req_prot_cleanup(struct request_sock_ops *rsk_prot)
3830 {
3831         if (!rsk_prot)
3832                 return;
3833         kfree(rsk_prot->slab_name);
3834         rsk_prot->slab_name = NULL;
3835         kmem_cache_destroy(rsk_prot->slab);
3836         rsk_prot->slab = NULL;
3837 }
3838
3839 static int req_prot_init(const struct proto *prot)
3840 {
3841         struct request_sock_ops *rsk_prot = prot->rsk_prot;
3842
3843         if (!rsk_prot)
3844                 return 0;
3845
3846         rsk_prot->slab_name = kasprintf(GFP_KERNEL, "request_sock_%s",
3847                                         prot->name);
3848         if (!rsk_prot->slab_name)
3849                 return -ENOMEM;
3850
3851         rsk_prot->slab = kmem_cache_create(rsk_prot->slab_name,
3852                                            rsk_prot->obj_size, 0,
3853                                            SLAB_ACCOUNT | prot->slab_flags,
3854                                            NULL);
3855
3856         if (!rsk_prot->slab) {
3857                 pr_crit("%s: Can't create request sock SLAB cache!\n",
3858                         prot->name);
3859                 return -ENOMEM;
3860         }
3861         return 0;
3862 }
3863
3864 int proto_register(struct proto *prot, int alloc_slab)
3865 {
3866         int ret = -ENOBUFS;
3867
3868         if (prot->memory_allocated && !prot->sysctl_mem) {
3869                 pr_err("%s: missing sysctl_mem\n", prot->name);
3870                 return -EINVAL;
3871         }
3872         if (prot->memory_allocated && !prot->per_cpu_fw_alloc) {
3873                 pr_err("%s: missing per_cpu_fw_alloc\n", prot->name);
3874                 return -EINVAL;
3875         }
3876         if (alloc_slab) {
3877                 prot->slab = kmem_cache_create_usercopy(prot->name,
3878                                         prot->obj_size, 0,
3879                                         SLAB_HWCACHE_ALIGN | SLAB_ACCOUNT |
3880                                         prot->slab_flags,
3881                                         prot->useroffset, prot->usersize,
3882                                         NULL);
3883
3884                 if (prot->slab == NULL) {
3885                         pr_crit("%s: Can't create sock SLAB cache!\n",
3886                                 prot->name);
3887                         goto out;
3888                 }
3889
3890                 if (req_prot_init(prot))
3891                         goto out_free_request_sock_slab;
3892
3893                 if (tw_prot_init(prot))
3894                         goto out_free_timewait_sock_slab;
3895         }
3896
3897         mutex_lock(&proto_list_mutex);
3898         ret = assign_proto_idx(prot);
3899         if (ret) {
3900                 mutex_unlock(&proto_list_mutex);
3901                 goto out_free_timewait_sock_slab;
3902         }
3903         list_add(&prot->node, &proto_list);
3904         mutex_unlock(&proto_list_mutex);
3905         return ret;
3906
3907 out_free_timewait_sock_slab:
3908         if (alloc_slab)
3909                 tw_prot_cleanup(prot->twsk_prot);
3910 out_free_request_sock_slab:
3911         if (alloc_slab) {
3912                 req_prot_cleanup(prot->rsk_prot);
3913
3914                 kmem_cache_destroy(prot->slab);
3915                 prot->slab = NULL;
3916         }
3917 out:
3918         return ret;
3919 }
3920 EXPORT_SYMBOL(proto_register);
3921
3922 void proto_unregister(struct proto *prot)
3923 {
3924         mutex_lock(&proto_list_mutex);
3925         release_proto_idx(prot);
3926         list_del(&prot->node);
3927         mutex_unlock(&proto_list_mutex);
3928
3929         kmem_cache_destroy(prot->slab);
3930         prot->slab = NULL;
3931
3932         req_prot_cleanup(prot->rsk_prot);
3933         tw_prot_cleanup(prot->twsk_prot);
3934 }
3935 EXPORT_SYMBOL(proto_unregister);
3936
3937 int sock_load_diag_module(int family, int protocol)
3938 {
3939         if (!protocol) {
3940                 if (!sock_is_registered(family))
3941                         return -ENOENT;
3942
3943                 return request_module("net-pf-%d-proto-%d-type-%d", PF_NETLINK,
3944                                       NETLINK_SOCK_DIAG, family);
3945         }
3946
3947 #ifdef CONFIG_INET
3948         if (family == AF_INET &&
3949             protocol != IPPROTO_RAW &&
3950             protocol < MAX_INET_PROTOS &&
3951             !rcu_access_pointer(inet_protos[protocol]))
3952                 return -ENOENT;
3953 #endif
3954
3955         return request_module("net-pf-%d-proto-%d-type-%d-%d", PF_NETLINK,
3956                               NETLINK_SOCK_DIAG, family, protocol);
3957 }
3958 EXPORT_SYMBOL(sock_load_diag_module);
3959
3960 #ifdef CONFIG_PROC_FS
3961 static void *proto_seq_start(struct seq_file *seq, loff_t *pos)
3962         __acquires(proto_list_mutex)
3963 {
3964         mutex_lock(&proto_list_mutex);
3965         return seq_list_start_head(&proto_list, *pos);
3966 }
3967
3968 static void *proto_seq_next(struct seq_file *seq, void *v, loff_t *pos)
3969 {
3970         return seq_list_next(v, &proto_list, pos);
3971 }
3972
3973 static void proto_seq_stop(struct seq_file *seq, void *v)
3974         __releases(proto_list_mutex)
3975 {
3976         mutex_unlock(&proto_list_mutex);
3977 }
3978
3979 static char proto_method_implemented(const void *method)
3980 {
3981         return method == NULL ? 'n' : 'y';
3982 }
3983 static long sock_prot_memory_allocated(struct proto *proto)
3984 {
3985         return proto->memory_allocated != NULL ? proto_memory_allocated(proto) : -1L;
3986 }
3987
3988 static const char *sock_prot_memory_pressure(struct proto *proto)
3989 {
3990         return proto->memory_pressure != NULL ?
3991         proto_memory_pressure(proto) ? "yes" : "no" : "NI";
3992 }
3993
3994 static void proto_seq_printf(struct seq_file *seq, struct proto *proto)
3995 {
3996
3997         seq_printf(seq, "%-9s %4u %6d  %6ld   %-3s %6u   %-3s  %-10s "
3998                         "%2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c %2c\n",
3999                    proto->name,
4000                    proto->obj_size,
4001                    sock_prot_inuse_get(seq_file_net(seq), proto),
4002                    sock_prot_memory_allocated(proto),
4003                    sock_prot_memory_pressure(proto),
4004                    proto->max_header,
4005                    proto->slab == NULL ? "no" : "yes",
4006                    module_name(proto->owner),
4007                    proto_method_implemented(proto->close),
4008                    proto_method_implemented(proto->connect),
4009                    proto_method_implemented(proto->disconnect),
4010                    proto_method_implemented(proto->accept),
4011                    proto_method_implemented(proto->ioctl),
4012                    proto_method_implemented(proto->init),
4013                    proto_method_implemented(proto->destroy),
4014                    proto_method_implemented(proto->shutdown),
4015                    proto_method_implemented(proto->setsockopt),
4016                    proto_method_implemented(proto->getsockopt),
4017                    proto_method_implemented(proto->sendmsg),
4018                    proto_method_implemented(proto->recvmsg),
4019                    proto_method_implemented(proto->sendpage),
4020                    proto_method_implemented(proto->bind),
4021                    proto_method_implemented(proto->backlog_rcv),
4022                    proto_method_implemented(proto->hash),
4023                    proto_method_implemented(proto->unhash),
4024                    proto_method_implemented(proto->get_port),
4025                    proto_method_implemented(proto->enter_memory_pressure));
4026 }
4027
4028 static int proto_seq_show(struct seq_file *seq, void *v)
4029 {
4030         if (v == &proto_list)
4031                 seq_printf(seq, "%-9s %-4s %-8s %-6s %-5s %-7s %-4s %-10s %s",
4032                            "protocol",
4033                            "size",
4034                            "sockets",
4035                            "memory",
4036                            "press",
4037                            "maxhdr",
4038                            "slab",
4039                            "module",
4040                            "cl co di ac io in de sh ss gs se re sp bi br ha uh gp em\n");
4041         else
4042                 proto_seq_printf(seq, list_entry(v, struct proto, node));
4043         return 0;
4044 }
4045
4046 static const struct seq_operations proto_seq_ops = {
4047         .start  = proto_seq_start,
4048         .next   = proto_seq_next,
4049         .stop   = proto_seq_stop,
4050         .show   = proto_seq_show,
4051 };
4052
4053 static __net_init int proto_init_net(struct net *net)
4054 {
4055         if (!proc_create_net("protocols", 0444, net->proc_net, &proto_seq_ops,
4056                         sizeof(struct seq_net_private)))
4057                 return -ENOMEM;
4058
4059         return 0;
4060 }
4061
4062 static __net_exit void proto_exit_net(struct net *net)
4063 {
4064         remove_proc_entry("protocols", net->proc_net);
4065 }
4066
4067
4068 static __net_initdata struct pernet_operations proto_net_ops = {
4069         .init = proto_init_net,
4070         .exit = proto_exit_net,
4071 };
4072
4073 static int __init proto_init(void)
4074 {
4075         return register_pernet_subsys(&proto_net_ops);
4076 }
4077
4078 subsys_initcall(proto_init);
4079
4080 #endif /* PROC_FS */
4081
4082 #ifdef CONFIG_NET_RX_BUSY_POLL
4083 bool sk_busy_loop_end(void *p, unsigned long start_time)
4084 {
4085         struct sock *sk = p;
4086
4087         return !skb_queue_empty_lockless(&sk->sk_receive_queue) ||
4088                sk_busy_loop_timeout(sk, start_time);
4089 }
4090 EXPORT_SYMBOL(sk_busy_loop_end);
4091 #endif /* CONFIG_NET_RX_BUSY_POLL */
4092
4093 int sock_bind_add(struct sock *sk, struct sockaddr *addr, int addr_len)
4094 {
4095         if (!sk->sk_prot->bind_add)
4096                 return -EOPNOTSUPP;
4097         return sk->sk_prot->bind_add(sk, addr, addr_len);
4098 }
4099 EXPORT_SYMBOL(sock_bind_add);