LCOV - code coverage report
Current view: top level - net/ipv4 - inet_connection_sock.c (source / functions) Hit Total Coverage
Test: Real Lines: 52 405 12.8 %
Date: 2020-10-17 15:46:16 Functions: 0 36 0.0 %
Legend: Neither, QEMU, Real, Both Branches: 0 0 -

           Branch data     Line data    Source code
       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                 :            :  *              Support for INET connection oriented protocols.
       8                 :            :  *
       9                 :            :  * Authors:     See the TCP sources
      10                 :            :  */
      11                 :            : 
      12                 :            : #include <linux/module.h>
      13                 :            : #include <linux/jhash.h>
      14                 :            : 
      15                 :            : #include <net/inet_connection_sock.h>
      16                 :            : #include <net/inet_hashtables.h>
      17                 :            : #include <net/inet_timewait_sock.h>
      18                 :            : #include <net/ip.h>
      19                 :            : #include <net/route.h>
      20                 :            : #include <net/tcp_states.h>
      21                 :            : #include <net/xfrm.h>
      22                 :            : #include <net/tcp.h>
      23                 :            : #include <net/sock_reuseport.h>
      24                 :            : #include <net/addrconf.h>
      25                 :            : 
      26                 :            : #if IS_ENABLED(CONFIG_IPV6)
      27                 :            : /* match_sk*_wildcard == true:  IPV6_ADDR_ANY equals to any IPv6 addresses
      28                 :            :  *                              if IPv6 only, and any IPv4 addresses
      29                 :            :  *                              if not IPv6 only
      30                 :            :  * match_sk*_wildcard == false: addresses must be exactly the same, i.e.
      31                 :            :  *                              IPV6_ADDR_ANY only equals to IPV6_ADDR_ANY,
      32                 :            :  *                              and 0.0.0.0 equals to 0.0.0.0 only
      33                 :            :  */
      34                 :          3 : static bool ipv6_rcv_saddr_equal(const struct in6_addr *sk1_rcv_saddr6,
      35                 :            :                                  const struct in6_addr *sk2_rcv_saddr6,
      36                 :            :                                  __be32 sk1_rcv_saddr, __be32 sk2_rcv_saddr,
      37                 :            :                                  bool sk1_ipv6only, bool sk2_ipv6only,
      38                 :            :                                  bool match_sk1_wildcard,
      39                 :            :                                  bool match_sk2_wildcard)
      40                 :            : {
      41                 :            :         int addr_type = ipv6_addr_type(sk1_rcv_saddr6);
      42                 :          3 :         int addr_type2 = sk2_rcv_saddr6 ? ipv6_addr_type(sk2_rcv_saddr6) : IPV6_ADDR_MAPPED;
      43                 :            : 
      44                 :            :         /* if both are mapped, treat as IPv4 */
      45                 :          3 :         if (addr_type == IPV6_ADDR_MAPPED && addr_type2 == IPV6_ADDR_MAPPED) {
      46                 :          0 :                 if (!sk2_ipv6only) {
      47                 :          0 :                         if (sk1_rcv_saddr == sk2_rcv_saddr)
      48                 :            :                                 return true;
      49                 :          0 :                         return (match_sk1_wildcard && !sk1_rcv_saddr) ||
      50                 :          0 :                                 (match_sk2_wildcard && !sk2_rcv_saddr);
      51                 :            :                 }
      52                 :            :                 return false;
      53                 :            :         }
      54                 :            : 
      55                 :          3 :         if (addr_type == IPV6_ADDR_ANY && addr_type2 == IPV6_ADDR_ANY)
      56                 :            :                 return true;
      57                 :            : 
      58                 :          3 :         if (addr_type2 == IPV6_ADDR_ANY && match_sk2_wildcard &&
      59                 :          0 :             !(sk2_ipv6only && addr_type == IPV6_ADDR_MAPPED))
      60                 :            :                 return true;
      61                 :            : 
      62                 :          3 :         if (addr_type == IPV6_ADDR_ANY && match_sk1_wildcard &&
      63                 :          3 :             !(sk1_ipv6only && addr_type2 == IPV6_ADDR_MAPPED))
      64                 :            :                 return true;
      65                 :            : 
      66                 :          3 :         if (sk2_rcv_saddr6 &&
      67                 :            :             ipv6_addr_equal(sk1_rcv_saddr6, sk2_rcv_saddr6))
      68                 :            :                 return true;
      69                 :            : 
      70                 :          3 :         return false;
      71                 :            : }
      72                 :            : #endif
      73                 :            : 
      74                 :            : /* match_sk*_wildcard == true:  0.0.0.0 equals to any IPv4 addresses
      75                 :            :  * match_sk*_wildcard == false: addresses must be exactly the same, i.e.
      76                 :            :  *                              0.0.0.0 only equals to 0.0.0.0
      77                 :            :  */
      78                 :            : static bool ipv4_rcv_saddr_equal(__be32 sk1_rcv_saddr, __be32 sk2_rcv_saddr,
      79                 :            :                                  bool sk2_ipv6only, bool match_sk1_wildcard,
      80                 :            :                                  bool match_sk2_wildcard)
      81                 :            : {
      82                 :          3 :         if (!sk2_ipv6only) {
      83                 :          3 :                 if (sk1_rcv_saddr == sk2_rcv_saddr)
      84                 :            :                         return true;
      85                 :          0 :                 return (match_sk1_wildcard && !sk1_rcv_saddr) ||
      86                 :          0 :                         (match_sk2_wildcard && !sk2_rcv_saddr);
      87                 :            :         }
      88                 :            :         return false;
      89                 :            : }
      90                 :            : 
      91                 :          3 : bool inet_rcv_saddr_equal(const struct sock *sk, const struct sock *sk2,
      92                 :            :                           bool match_wildcard)
      93                 :            : {
      94                 :            : #if IS_ENABLED(CONFIG_IPV6)
      95                 :          3 :         if (sk->sk_family == AF_INET6)
      96                 :          3 :                 return ipv6_rcv_saddr_equal(&sk->sk_v6_rcv_saddr,
      97                 :            :                                             inet6_rcv_saddr(sk2),
      98                 :            :                                             sk->sk_rcv_saddr,
      99                 :            :                                             sk2->sk_rcv_saddr,
     100                 :          3 :                                             ipv6_only_sock(sk),
     101                 :          3 :                                             ipv6_only_sock(sk2),
     102                 :            :                                             match_wildcard,
     103                 :            :                                             match_wildcard);
     104                 :            : #endif
     105                 :          3 :         return ipv4_rcv_saddr_equal(sk->sk_rcv_saddr, sk2->sk_rcv_saddr,
     106                 :          3 :                                     ipv6_only_sock(sk2), match_wildcard,
     107                 :            :                                     match_wildcard);
     108                 :            : }
     109                 :            : EXPORT_SYMBOL(inet_rcv_saddr_equal);
     110                 :            : 
     111                 :          1 : bool inet_rcv_saddr_any(const struct sock *sk)
     112                 :            : {
     113                 :            : #if IS_ENABLED(CONFIG_IPV6)
     114                 :          1 :         if (sk->sk_family == AF_INET6)
     115                 :          0 :                 return ipv6_addr_any(&sk->sk_v6_rcv_saddr);
     116                 :            : #endif
     117                 :          1 :         return !sk->sk_rcv_saddr;
     118                 :            : }
     119                 :            : 
     120                 :          3 : void inet_get_local_port_range(struct net *net, int *low, int *high)
     121                 :            : {
     122                 :            :         unsigned int seq;
     123                 :            : 
     124                 :            :         do {
     125                 :            :                 seq = read_seqbegin(&net->ipv4.ip_local_ports.lock);
     126                 :            : 
     127                 :          3 :                 *low = net->ipv4.ip_local_ports.range[0];
     128                 :          3 :                 *high = net->ipv4.ip_local_ports.range[1];
     129                 :          3 :         } while (read_seqretry(&net->ipv4.ip_local_ports.lock, seq));
     130                 :          3 : }
     131                 :            : EXPORT_SYMBOL(inet_get_local_port_range);
     132                 :            : 
     133                 :          0 : static int inet_csk_bind_conflict(const struct sock *sk,
     134                 :            :                                   const struct inet_bind_bucket *tb,
     135                 :            :                                   bool relax, bool reuseport_ok)
     136                 :            : {
     137                 :            :         struct sock *sk2;
     138                 :          0 :         bool reuse = sk->sk_reuse;
     139                 :          0 :         bool reuseport = !!sk->sk_reuseport && reuseport_ok;
     140                 :          0 :         kuid_t uid = sock_i_uid((struct sock *)sk);
     141                 :            : 
     142                 :            :         /*
     143                 :            :          * Unlike other sk lookup places we do not check
     144                 :            :          * for sk_net here, since _all_ the socks listed
     145                 :            :          * in tb->owners list belong to the same net - the
     146                 :            :          * one this bucket belongs to.
     147                 :            :          */
     148                 :            : 
     149                 :          0 :         sk_for_each_bound(sk2, &tb->owners) {
     150                 :          0 :                 if (sk != sk2 &&
     151                 :          0 :                     (!sk->sk_bound_dev_if ||
     152                 :          0 :                      !sk2->sk_bound_dev_if ||
     153                 :            :                      sk->sk_bound_dev_if == sk2->sk_bound_dev_if)) {
     154                 :          0 :                         if ((!reuse || !sk2->sk_reuse ||
     155                 :          0 :                             sk2->sk_state == TCP_LISTEN) &&
     156                 :          0 :                             (!reuseport || !sk2->sk_reuseport ||
     157                 :          0 :                              rcu_access_pointer(sk->sk_reuseport_cb) ||
     158                 :          0 :                              (sk2->sk_state != TCP_TIME_WAIT &&
     159                 :          0 :                              !uid_eq(uid, sock_i_uid(sk2))))) {
     160                 :          0 :                                 if (inet_rcv_saddr_equal(sk, sk2, true))
     161                 :            :                                         break;
     162                 :            :                         }
     163                 :          0 :                         if (!relax && reuse && sk2->sk_reuse &&
     164                 :          0 :                             sk2->sk_state != TCP_LISTEN) {
     165                 :          0 :                                 if (inet_rcv_saddr_equal(sk, sk2, true))
     166                 :            :                                         break;
     167                 :            :                         }
     168                 :            :                 }
     169                 :            :         }
     170                 :          0 :         return sk2 != NULL;
     171                 :            : }
     172                 :            : 
     173                 :            : /*
     174                 :            :  * Find an open port number for the socket.  Returns with the
     175                 :            :  * inet_bind_hashbucket lock held.
     176                 :            :  */
     177                 :            : static struct inet_bind_hashbucket *
     178                 :          0 : inet_csk_find_open_port(struct sock *sk, struct inet_bind_bucket **tb_ret, int *port_ret)
     179                 :            : {
     180                 :          0 :         struct inet_hashinfo *hinfo = sk->sk_prot->h.hashinfo;
     181                 :            :         int port = 0;
     182                 :            :         struct inet_bind_hashbucket *head;
     183                 :            :         struct net *net = sock_net(sk);
     184                 :            :         int i, low, high, attempt_half;
     185                 :            :         struct inet_bind_bucket *tb;
     186                 :            :         u32 remaining, offset;
     187                 :            :         int l3mdev;
     188                 :            : 
     189                 :            :         l3mdev = inet_sk_bound_l3mdev(sk);
     190                 :          0 :         attempt_half = (sk->sk_reuse == SK_CAN_REUSE) ? 1 : 0;
     191                 :            : other_half_scan:
     192                 :            :         inet_get_local_port_range(net, &low, &high);
     193                 :          0 :         high++; /* [32768, 60999] -> [32768, 61000[ */
     194                 :          0 :         if (high - low < 4)
     195                 :            :                 attempt_half = 0;
     196                 :          0 :         if (attempt_half) {
     197                 :          0 :                 int half = low + (((high - low) >> 2) << 1);
     198                 :            : 
     199                 :          0 :                 if (attempt_half == 1)
     200                 :            :                         high = half;
     201                 :            :                 else
     202                 :            :                         low = half;
     203                 :            :         }
     204                 :          0 :         remaining = high - low;
     205                 :          0 :         if (likely(remaining > 1))
     206                 :          0 :                 remaining &= ~1U;
     207                 :            : 
     208                 :          0 :         offset = prandom_u32() % remaining;
     209                 :            :         /* __inet_hash_connect() favors ports having @low parity
     210                 :            :          * We do the opposite to not pollute connect() users.
     211                 :            :          */
     212                 :          0 :         offset |= 1U;
     213                 :            : 
     214                 :            : other_parity_scan:
     215                 :          0 :         port = low + offset;
     216                 :          0 :         for (i = 0; i < remaining; i += 2, port += 2) {
     217                 :          0 :                 if (unlikely(port >= high))
     218                 :          0 :                         port -= remaining;
     219                 :          0 :                 if (inet_is_local_reserved_port(net, port))
     220                 :          0 :                         continue;
     221                 :          0 :                 head = &hinfo->bhash[inet_bhashfn(net, port,
     222                 :            :                                                   hinfo->bhash_size)];
     223                 :            :                 spin_lock_bh(&head->lock);
     224                 :          0 :                 inet_bind_bucket_for_each(tb, &head->chain)
     225                 :          0 :                         if (net_eq(ib_net(tb), net) && tb->l3mdev == l3mdev &&
     226                 :          0 :                             tb->port == port) {
     227                 :          0 :                                 if (!inet_csk_bind_conflict(sk, tb, false, false))
     228                 :            :                                         goto success;
     229                 :            :                                 goto next_port;
     230                 :            :                         }
     231                 :            :                 tb = NULL;
     232                 :            :                 goto success;
     233                 :            : next_port:
     234                 :            :                 spin_unlock_bh(&head->lock);
     235                 :          0 :                 cond_resched();
     236                 :            :         }
     237                 :            : 
     238                 :          0 :         offset--;
     239                 :          0 :         if (!(offset & 1))
     240                 :            :                 goto other_parity_scan;
     241                 :            : 
     242                 :          0 :         if (attempt_half == 1) {
     243                 :            :                 /* OK we now try the upper half of the range */
     244                 :            :                 attempt_half = 2;
     245                 :            :                 goto other_half_scan;
     246                 :            :         }
     247                 :            :         return NULL;
     248                 :            : success:
     249                 :          0 :         *port_ret = port;
     250                 :          0 :         *tb_ret = tb;
     251                 :          0 :         return head;
     252                 :            : }
     253                 :            : 
     254                 :          0 : static inline int sk_reuseport_match(struct inet_bind_bucket *tb,
     255                 :            :                                      struct sock *sk)
     256                 :            : {
     257                 :          0 :         kuid_t uid = sock_i_uid(sk);
     258                 :            : 
     259                 :          0 :         if (tb->fastreuseport <= 0)
     260                 :            :                 return 0;
     261                 :          0 :         if (!sk->sk_reuseport)
     262                 :            :                 return 0;
     263                 :          0 :         if (rcu_access_pointer(sk->sk_reuseport_cb))
     264                 :            :                 return 0;
     265                 :          0 :         if (!uid_eq(tb->fastuid, uid))
     266                 :            :                 return 0;
     267                 :            :         /* We only need to check the rcv_saddr if this tb was once marked
     268                 :            :          * without fastreuseport and then was reset, as we can only know that
     269                 :            :          * the fast_*rcv_saddr doesn't have any conflicts with the socks on the
     270                 :            :          * owners list.
     271                 :            :          */
     272                 :          0 :         if (tb->fastreuseport == FASTREUSEPORT_ANY)
     273                 :            :                 return 1;
     274                 :            : #if IS_ENABLED(CONFIG_IPV6)
     275                 :          0 :         if (tb->fast_sk_family == AF_INET6)
     276                 :          0 :                 return ipv6_rcv_saddr_equal(&tb->fast_v6_rcv_saddr,
     277                 :            :                                             inet6_rcv_saddr(sk),
     278                 :            :                                             tb->fast_rcv_saddr,
     279                 :            :                                             sk->sk_rcv_saddr,
     280                 :            :                                             tb->fast_ipv6_only,
     281                 :          0 :                                             ipv6_only_sock(sk), true, false);
     282                 :            : #endif
     283                 :          0 :         return ipv4_rcv_saddr_equal(tb->fast_rcv_saddr, sk->sk_rcv_saddr,
     284                 :          0 :                                     ipv6_only_sock(sk), true, false);
     285                 :            : }
     286                 :            : 
     287                 :          0 : void inet_csk_update_fastreuse(struct inet_bind_bucket *tb,
     288                 :            :                                struct sock *sk)
     289                 :            : {
     290                 :          0 :         kuid_t uid = sock_i_uid(sk);
     291                 :          0 :         bool reuse = sk->sk_reuse && sk->sk_state != TCP_LISTEN;
     292                 :            : 
     293                 :          0 :         if (hlist_empty(&tb->owners)) {
     294                 :          0 :                 tb->fastreuse = reuse;
     295                 :          0 :                 if (sk->sk_reuseport) {
     296                 :          0 :                         tb->fastreuseport = FASTREUSEPORT_ANY;
     297                 :          0 :                         tb->fastuid = uid;
     298                 :          0 :                         tb->fast_rcv_saddr = sk->sk_rcv_saddr;
     299                 :          0 :                         tb->fast_ipv6_only = ipv6_only_sock(sk);
     300                 :          0 :                         tb->fast_sk_family = sk->sk_family;
     301                 :            : #if IS_ENABLED(CONFIG_IPV6)
     302                 :          0 :                         tb->fast_v6_rcv_saddr = sk->sk_v6_rcv_saddr;
     303                 :            : #endif
     304                 :            :                 } else {
     305                 :          0 :                         tb->fastreuseport = 0;
     306                 :            :                 }
     307                 :            :         } else {
     308                 :          0 :                 if (!reuse)
     309                 :          0 :                         tb->fastreuse = 0;
     310                 :          0 :                 if (sk->sk_reuseport) {
     311                 :            :                         /* We didn't match or we don't have fastreuseport set on
     312                 :            :                          * the tb, but we have sk_reuseport set on this socket
     313                 :            :                          * and we know that there are no bind conflicts with
     314                 :            :                          * this socket in this tb, so reset our tb's reuseport
     315                 :            :                          * settings so that any subsequent sockets that match
     316                 :            :                          * our current socket will be put on the fast path.
     317                 :            :                          *
     318                 :            :                          * If we reset we need to set FASTREUSEPORT_STRICT so we
     319                 :            :                          * do extra checking for all subsequent sk_reuseport
     320                 :            :                          * socks.
     321                 :            :                          */
     322                 :          0 :                         if (!sk_reuseport_match(tb, sk)) {
     323                 :          0 :                                 tb->fastreuseport = FASTREUSEPORT_STRICT;
     324                 :          0 :                                 tb->fastuid = uid;
     325                 :          0 :                                 tb->fast_rcv_saddr = sk->sk_rcv_saddr;
     326                 :          0 :                                 tb->fast_ipv6_only = ipv6_only_sock(sk);
     327                 :          0 :                                 tb->fast_sk_family = sk->sk_family;
     328                 :            : #if IS_ENABLED(CONFIG_IPV6)
     329                 :          0 :                                 tb->fast_v6_rcv_saddr = sk->sk_v6_rcv_saddr;
     330                 :            : #endif
     331                 :            :                         }
     332                 :            :                 } else {
     333                 :          0 :                         tb->fastreuseport = 0;
     334                 :            :                 }
     335                 :            :         }
     336                 :          0 : }
     337                 :            : 
     338                 :            : /* Obtain a reference to a local port for the given sock,
     339                 :            :  * if snum is zero it means select any available local port.
     340                 :            :  * We try to allocate an odd port (and leave even ports for connect())
     341                 :            :  */
     342                 :          0 : int inet_csk_get_port(struct sock *sk, unsigned short snum)
     343                 :            : {
     344                 :          0 :         bool reuse = sk->sk_reuse && sk->sk_state != TCP_LISTEN;
     345                 :          0 :         struct inet_hashinfo *hinfo = sk->sk_prot->h.hashinfo;
     346                 :          0 :         int ret = 1, port = snum;
     347                 :            :         struct inet_bind_hashbucket *head;
     348                 :            :         struct net *net = sock_net(sk);
     349                 :          0 :         struct inet_bind_bucket *tb = NULL;
     350                 :            :         int l3mdev;
     351                 :            : 
     352                 :            :         l3mdev = inet_sk_bound_l3mdev(sk);
     353                 :            : 
     354                 :          0 :         if (!port) {
     355                 :          0 :                 head = inet_csk_find_open_port(sk, &tb, &port);
     356                 :          0 :                 if (!head)
     357                 :            :                         return ret;
     358                 :          0 :                 if (!tb)
     359                 :            :                         goto tb_not_found;
     360                 :            :                 goto success;
     361                 :            :         }
     362                 :          0 :         head = &hinfo->bhash[inet_bhashfn(net, port,
     363                 :            :                                           hinfo->bhash_size)];
     364                 :            :         spin_lock_bh(&head->lock);
     365                 :          0 :         inet_bind_bucket_for_each(tb, &head->chain)
     366                 :          0 :                 if (net_eq(ib_net(tb), net) && tb->l3mdev == l3mdev &&
     367                 :          0 :                     tb->port == port)
     368                 :            :                         goto tb_found;
     369                 :            : tb_not_found:
     370                 :          0 :         tb = inet_bind_bucket_create(hinfo->bind_bucket_cachep,
     371                 :            :                                      net, head, port, l3mdev);
     372                 :          0 :         if (!tb)
     373                 :            :                 goto fail_unlock;
     374                 :            : tb_found:
     375                 :          0 :         if (!hlist_empty(&tb->owners)) {
     376                 :          0 :                 if (sk->sk_reuse == SK_FORCE_REUSE)
     377                 :            :                         goto success;
     378                 :            : 
     379                 :          0 :                 if ((tb->fastreuse > 0 && reuse) ||
     380                 :          0 :                     sk_reuseport_match(tb, sk))
     381                 :            :                         goto success;
     382                 :          0 :                 if (inet_csk_bind_conflict(sk, tb, true, true))
     383                 :            :                         goto fail_unlock;
     384                 :            :         }
     385                 :            : success:
     386                 :          0 :         inet_csk_update_fastreuse(tb, sk);
     387                 :            : 
     388                 :          0 :         if (!inet_csk(sk)->icsk_bind_hash)
     389                 :          0 :                 inet_bind_hash(sk, tb, port);
     390                 :          0 :         WARN_ON(inet_csk(sk)->icsk_bind_hash != tb);
     391                 :            :         ret = 0;
     392                 :            : 
     393                 :            : fail_unlock:
     394                 :            :         spin_unlock_bh(&head->lock);
     395                 :          0 :         return ret;
     396                 :            : }
     397                 :            : EXPORT_SYMBOL_GPL(inet_csk_get_port);
     398                 :            : 
     399                 :            : /*
     400                 :            :  * Wait for an incoming connection, avoid race conditions. This must be called
     401                 :            :  * with the socket locked.
     402                 :            :  */
     403                 :          0 : static int inet_csk_wait_for_connect(struct sock *sk, long timeo)
     404                 :            : {
     405                 :            :         struct inet_connection_sock *icsk = inet_csk(sk);
     406                 :          0 :         DEFINE_WAIT(wait);
     407                 :            :         int err;
     408                 :            : 
     409                 :            :         /*
     410                 :            :          * True wake-one mechanism for incoming connections: only
     411                 :            :          * one process gets woken up, not the 'whole herd'.
     412                 :            :          * Since we do not 'race & poll' for established sockets
     413                 :            :          * anymore, the common case will execute the loop only once.
     414                 :            :          *
     415                 :            :          * Subtle issue: "add_wait_queue_exclusive()" will be added
     416                 :            :          * after any current non-exclusive waiters, and we know that
     417                 :            :          * it will always _stay_ after any new non-exclusive waiters
     418                 :            :          * because all non-exclusive waiters are added at the
     419                 :            :          * beginning of the wait-queue. As such, it's ok to "drop"
     420                 :            :          * our exclusiveness temporarily when we get woken up without
     421                 :            :          * having to remove and re-insert us on the wait queue.
     422                 :            :          */
     423                 :            :         for (;;) {
     424                 :          0 :                 prepare_to_wait_exclusive(sk_sleep(sk), &wait,
     425                 :            :                                           TASK_INTERRUPTIBLE);
     426                 :          0 :                 release_sock(sk);
     427                 :          0 :                 if (reqsk_queue_empty(&icsk->icsk_accept_queue))
     428                 :          0 :                         timeo = schedule_timeout(timeo);
     429                 :            :                 sched_annotate_sleep();
     430                 :            :                 lock_sock(sk);
     431                 :            :                 err = 0;
     432                 :          0 :                 if (!reqsk_queue_empty(&icsk->icsk_accept_queue))
     433                 :            :                         break;
     434                 :            :                 err = -EINVAL;
     435                 :          0 :                 if (sk->sk_state != TCP_LISTEN)
     436                 :            :                         break;
     437                 :            :                 err = sock_intr_errno(timeo);
     438                 :          0 :                 if (signal_pending(current))
     439                 :            :                         break;
     440                 :            :                 err = -EAGAIN;
     441                 :          0 :                 if (!timeo)
     442                 :            :                         break;
     443                 :            :         }
     444                 :          0 :         finish_wait(sk_sleep(sk), &wait);
     445                 :          0 :         return err;
     446                 :            : }
     447                 :            : 
     448                 :            : /*
     449                 :            :  * This will accept the next outstanding connection.
     450                 :            :  */
     451                 :          0 : struct sock *inet_csk_accept(struct sock *sk, int flags, int *err, bool kern)
     452                 :            : {
     453                 :            :         struct inet_connection_sock *icsk = inet_csk(sk);
     454                 :          0 :         struct request_sock_queue *queue = &icsk->icsk_accept_queue;
     455                 :            :         struct request_sock *req;
     456                 :            :         struct sock *newsk;
     457                 :            :         int error;
     458                 :            : 
     459                 :            :         lock_sock(sk);
     460                 :            : 
     461                 :            :         /* We need to make sure that this socket is listening,
     462                 :            :          * and that it has something pending.
     463                 :            :          */
     464                 :            :         error = -EINVAL;
     465                 :          0 :         if (sk->sk_state != TCP_LISTEN)
     466                 :            :                 goto out_err;
     467                 :            : 
     468                 :            :         /* Find already established connection */
     469                 :          0 :         if (reqsk_queue_empty(queue)) {
     470                 :          0 :                 long timeo = sock_rcvtimeo(sk, flags & O_NONBLOCK);
     471                 :            : 
     472                 :            :                 /* If this is a non blocking socket don't sleep */
     473                 :            :                 error = -EAGAIN;
     474                 :          0 :                 if (!timeo)
     475                 :            :                         goto out_err;
     476                 :            : 
     477                 :          0 :                 error = inet_csk_wait_for_connect(sk, timeo);
     478                 :          0 :                 if (error)
     479                 :            :                         goto out_err;
     480                 :            :         }
     481                 :          0 :         req = reqsk_queue_remove(queue, sk);
     482                 :          0 :         newsk = req->sk;
     483                 :            : 
     484                 :          0 :         if (sk->sk_protocol == IPPROTO_TCP &&
     485                 :          0 :             tcp_rsk(req)->tfo_listener) {
     486                 :            :                 spin_lock_bh(&queue->fastopenq.lock);
     487                 :          0 :                 if (tcp_rsk(req)->tfo_listener) {
     488                 :            :                         /* We are still waiting for the final ACK from 3WHS
     489                 :            :                          * so can't free req now. Instead, we set req->sk to
     490                 :            :                          * NULL to signify that the child socket is taken
     491                 :            :                          * so reqsk_fastopen_remove() will free the req
     492                 :            :                          * when 3WHS finishes (or is aborted).
     493                 :            :                          */
     494                 :          0 :                         req->sk = NULL;
     495                 :            :                         req = NULL;
     496                 :            :                 }
     497                 :            :                 spin_unlock_bh(&queue->fastopenq.lock);
     498                 :            :         }
     499                 :            : 
     500                 :            : out:
     501                 :          0 :         release_sock(sk);
     502                 :          0 :         if (newsk && mem_cgroup_sockets_enabled) {
     503                 :            :                 int amt;
     504                 :            : 
     505                 :            :                 /* atomically get the memory usage, set and charge the
     506                 :            :                  * newsk->sk_memcg.
     507                 :            :                  */
     508                 :            :                 lock_sock(newsk);
     509                 :            : 
     510                 :            :                 /* The socket has not been accepted yet, no need to look at
     511                 :            :                  * newsk->sk_wmem_queued.
     512                 :            :                  */
     513                 :          0 :                 amt = sk_mem_pages(newsk->sk_forward_alloc +
     514                 :          0 :                                    atomic_read(&newsk->sk_rmem_alloc));
     515                 :          0 :                 mem_cgroup_sk_alloc(newsk);
     516                 :          0 :                 if (newsk->sk_memcg && amt)
     517                 :          0 :                         mem_cgroup_charge_skmem(newsk->sk_memcg, amt);
     518                 :            : 
     519                 :          0 :                 release_sock(newsk);
     520                 :            :         }
     521                 :          0 :         if (req)
     522                 :          0 :                 reqsk_put(req);
     523                 :          0 :         return newsk;
     524                 :            : out_err:
     525                 :            :         newsk = NULL;
     526                 :            :         req = NULL;
     527                 :          0 :         *err = error;
     528                 :          0 :         goto out;
     529                 :            : }
     530                 :            : EXPORT_SYMBOL(inet_csk_accept);
     531                 :            : 
     532                 :            : /*
     533                 :            :  * Using different timers for retransmit, delayed acks and probes
     534                 :            :  * We may wish use just one timer maintaining a list of expire jiffies
     535                 :            :  * to optimize.
     536                 :            :  */
     537                 :          1 : void inet_csk_init_xmit_timers(struct sock *sk,
     538                 :            :                                void (*retransmit_handler)(struct timer_list *t),
     539                 :            :                                void (*delack_handler)(struct timer_list *t),
     540                 :            :                                void (*keepalive_handler)(struct timer_list *t))
     541                 :            : {
     542                 :            :         struct inet_connection_sock *icsk = inet_csk(sk);
     543                 :            : 
     544                 :          1 :         timer_setup(&icsk->icsk_retransmit_timer, retransmit_handler, 0);
     545                 :          1 :         timer_setup(&icsk->icsk_delack_timer, delack_handler, 0);
     546                 :          1 :         timer_setup(&sk->sk_timer, keepalive_handler, 0);
     547                 :          1 :         icsk->icsk_pending = icsk->icsk_ack.pending = 0;
     548                 :          1 : }
     549                 :            : EXPORT_SYMBOL(inet_csk_init_xmit_timers);
     550                 :            : 
     551                 :          1 : void inet_csk_clear_xmit_timers(struct sock *sk)
     552                 :            : {
     553                 :            :         struct inet_connection_sock *icsk = inet_csk(sk);
     554                 :            : 
     555                 :          1 :         icsk->icsk_pending = icsk->icsk_ack.pending = icsk->icsk_ack.blocked = 0;
     556                 :            : 
     557                 :          1 :         sk_stop_timer(sk, &icsk->icsk_retransmit_timer);
     558                 :          1 :         sk_stop_timer(sk, &icsk->icsk_delack_timer);
     559                 :          1 :         sk_stop_timer(sk, &sk->sk_timer);
     560                 :          1 : }
     561                 :            : EXPORT_SYMBOL(inet_csk_clear_xmit_timers);
     562                 :            : 
     563                 :          0 : void inet_csk_delete_keepalive_timer(struct sock *sk)
     564                 :            : {
     565                 :          0 :         sk_stop_timer(sk, &sk->sk_timer);
     566                 :          0 : }
     567                 :            : EXPORT_SYMBOL(inet_csk_delete_keepalive_timer);
     568                 :            : 
     569                 :          1 : void inet_csk_reset_keepalive_timer(struct sock *sk, unsigned long len)
     570                 :            : {
     571                 :          1 :         sk_reset_timer(sk, &sk->sk_timer, jiffies + len);
     572                 :          1 : }
     573                 :            : EXPORT_SYMBOL(inet_csk_reset_keepalive_timer);
     574                 :            : 
     575                 :          0 : struct dst_entry *inet_csk_route_req(const struct sock *sk,
     576                 :            :                                      struct flowi4 *fl4,
     577                 :            :                                      const struct request_sock *req)
     578                 :            : {
     579                 :            :         const struct inet_request_sock *ireq = inet_rsk(req);
     580                 :            :         struct net *net = read_pnet(&ireq->ireq_net);
     581                 :            :         struct ip_options_rcu *opt;
     582                 :            :         struct rtable *rt;
     583                 :            : 
     584                 :            :         rcu_read_lock();
     585                 :          0 :         opt = rcu_dereference(ireq->ireq_opt);
     586                 :            : 
     587                 :          0 :         flowi4_init_output(fl4, ireq->ir_iif, ireq->ir_mark,
     588                 :          0 :                            RT_CONN_FLAGS(sk), RT_SCOPE_UNIVERSE,
     589                 :            :                            sk->sk_protocol, inet_sk_flowi_flags(sk),
     590                 :          0 :                            (opt && opt->opt.srr) ? opt->opt.faddr : ireq->ir_rmt_addr,
     591                 :            :                            ireq->ir_loc_addr, ireq->ir_rmt_port,
     592                 :            :                            htons(ireq->ir_num), sk->sk_uid);
     593                 :          0 :         security_req_classify_flow(req, flowi4_to_flowi(fl4));
     594                 :          0 :         rt = ip_route_output_flow(net, fl4, sk);
     595                 :          0 :         if (IS_ERR(rt))
     596                 :            :                 goto no_route;
     597                 :          0 :         if (opt && opt->opt.is_strictroute && rt->rt_uses_gateway)
     598                 :            :                 goto route_err;
     599                 :            :         rcu_read_unlock();
     600                 :          0 :         return &rt->dst;
     601                 :            : 
     602                 :            : route_err:
     603                 :            :         ip_rt_put(rt);
     604                 :            : no_route:
     605                 :            :         rcu_read_unlock();
     606                 :          0 :         __IP_INC_STATS(net, IPSTATS_MIB_OUTNOROUTES);
     607                 :          0 :         return NULL;
     608                 :            : }
     609                 :            : EXPORT_SYMBOL_GPL(inet_csk_route_req);
     610                 :            : 
     611                 :          0 : struct dst_entry *inet_csk_route_child_sock(const struct sock *sk,
     612                 :            :                                             struct sock *newsk,
     613                 :            :                                             const struct request_sock *req)
     614                 :            : {
     615                 :            :         const struct inet_request_sock *ireq = inet_rsk(req);
     616                 :            :         struct net *net = read_pnet(&ireq->ireq_net);
     617                 :            :         struct inet_sock *newinet = inet_sk(newsk);
     618                 :            :         struct ip_options_rcu *opt;
     619                 :            :         struct flowi4 *fl4;
     620                 :            :         struct rtable *rt;
     621                 :            : 
     622                 :          0 :         opt = rcu_dereference(ireq->ireq_opt);
     623                 :          0 :         fl4 = &newinet->cork.fl.u.ip4;
     624                 :            : 
     625                 :          0 :         flowi4_init_output(fl4, ireq->ir_iif, ireq->ir_mark,
     626                 :          0 :                            RT_CONN_FLAGS(sk), RT_SCOPE_UNIVERSE,
     627                 :            :                            sk->sk_protocol, inet_sk_flowi_flags(sk),
     628                 :          0 :                            (opt && opt->opt.srr) ? opt->opt.faddr : ireq->ir_rmt_addr,
     629                 :            :                            ireq->ir_loc_addr, ireq->ir_rmt_port,
     630                 :            :                            htons(ireq->ir_num), sk->sk_uid);
     631                 :          0 :         security_req_classify_flow(req, flowi4_to_flowi(fl4));
     632                 :          0 :         rt = ip_route_output_flow(net, fl4, sk);
     633                 :          0 :         if (IS_ERR(rt))
     634                 :            :                 goto no_route;
     635                 :          0 :         if (opt && opt->opt.is_strictroute && rt->rt_uses_gateway)
     636                 :            :                 goto route_err;
     637                 :          0 :         return &rt->dst;
     638                 :            : 
     639                 :            : route_err:
     640                 :            :         ip_rt_put(rt);
     641                 :            : no_route:
     642                 :          0 :         __IP_INC_STATS(net, IPSTATS_MIB_OUTNOROUTES);
     643                 :          0 :         return NULL;
     644                 :            : }
     645                 :            : EXPORT_SYMBOL_GPL(inet_csk_route_child_sock);
     646                 :            : 
     647                 :            : #if IS_ENABLED(CONFIG_IPV6)
     648                 :            : #define AF_INET_FAMILY(fam) ((fam) == AF_INET)
     649                 :            : #else
     650                 :            : #define AF_INET_FAMILY(fam) true
     651                 :            : #endif
     652                 :            : 
     653                 :            : /* Decide when to expire the request and when to resend SYN-ACK */
     654                 :          0 : static inline void syn_ack_recalc(struct request_sock *req, const int thresh,
     655                 :            :                                   const int max_retries,
     656                 :            :                                   const u8 rskq_defer_accept,
     657                 :            :                                   int *expire, int *resend)
     658                 :            : {
     659                 :          0 :         if (!rskq_defer_accept) {
     660                 :          0 :                 *expire = req->num_timeout >= thresh;
     661                 :          0 :                 *resend = 1;
     662                 :          0 :                 return;
     663                 :            :         }
     664                 :          0 :         *expire = req->num_timeout >= thresh &&
     665                 :          0 :                   (!inet_rsk(req)->acked || req->num_timeout >= max_retries);
     666                 :            :         /*
     667                 :            :          * Do not resend while waiting for data after ACK,
     668                 :            :          * start to resend on end of deferring period to give
     669                 :            :          * last chance for data or ACK to create established socket.
     670                 :            :          */
     671                 :          0 :         *resend = !inet_rsk(req)->acked ||
     672                 :          0 :                   req->num_timeout >= rskq_defer_accept - 1;
     673                 :            : }
     674                 :            : 
     675                 :          0 : int inet_rtx_syn_ack(const struct sock *parent, struct request_sock *req)
     676                 :            : {
     677                 :          0 :         int err = req->rsk_ops->rtx_syn_ack(parent, req);
     678                 :            : 
     679                 :          0 :         if (!err)
     680                 :          0 :                 req->num_retrans++;
     681                 :          0 :         return err;
     682                 :            : }
     683                 :            : EXPORT_SYMBOL(inet_rtx_syn_ack);
     684                 :            : 
     685                 :            : /* return true if req was found in the ehash table */
     686                 :          0 : static bool reqsk_queue_unlink(struct request_sock *req)
     687                 :            : {
     688                 :          0 :         struct inet_hashinfo *hashinfo = req_to_sk(req)->sk_prot->h.hashinfo;
     689                 :            :         bool found = false;
     690                 :            : 
     691                 :          0 :         if (sk_hashed(req_to_sk(req))) {
     692                 :          0 :                 spinlock_t *lock = inet_ehash_lockp(hashinfo, req->rsk_hash);
     693                 :            : 
     694                 :            :                 spin_lock(lock);
     695                 :            :                 found = __sk_nulls_del_node_init_rcu(req_to_sk(req));
     696                 :            :                 spin_unlock(lock);
     697                 :            :         }
     698                 :          0 :         if (timer_pending(&req->rsk_timer) && del_timer_sync(&req->rsk_timer))
     699                 :          0 :                 reqsk_put(req);
     700                 :          0 :         return found;
     701                 :            : }
     702                 :            : 
     703                 :          0 : void inet_csk_reqsk_queue_drop(struct sock *sk, struct request_sock *req)
     704                 :            : {
     705                 :          0 :         if (reqsk_queue_unlink(req)) {
     706                 :          0 :                 reqsk_queue_removed(&inet_csk(sk)->icsk_accept_queue, req);
     707                 :          0 :                 reqsk_put(req);
     708                 :            :         }
     709                 :          0 : }
     710                 :            : EXPORT_SYMBOL(inet_csk_reqsk_queue_drop);
     711                 :            : 
     712                 :          0 : void inet_csk_reqsk_queue_drop_and_put(struct sock *sk, struct request_sock *req)
     713                 :            : {
     714                 :          0 :         inet_csk_reqsk_queue_drop(sk, req);
     715                 :          0 :         reqsk_put(req);
     716                 :          0 : }
     717                 :            : EXPORT_SYMBOL(inet_csk_reqsk_queue_drop_and_put);
     718                 :            : 
     719                 :          0 : static void reqsk_timer_handler(struct timer_list *t)
     720                 :            : {
     721                 :          0 :         struct request_sock *req = from_timer(req, t, rsk_timer);
     722                 :          0 :         struct sock *sk_listener = req->rsk_listener;
     723                 :            :         struct net *net = sock_net(sk_listener);
     724                 :            :         struct inet_connection_sock *icsk = inet_csk(sk_listener);
     725                 :            :         struct request_sock_queue *queue = &icsk->icsk_accept_queue;
     726                 :          0 :         int qlen, expire = 0, resend = 0;
     727                 :            :         int max_retries, thresh;
     728                 :            :         u8 defer_accept;
     729                 :            : 
     730                 :          0 :         if (inet_sk_state_load(sk_listener) != TCP_LISTEN)
     731                 :            :                 goto drop;
     732                 :            : 
     733                 :          0 :         max_retries = icsk->icsk_syn_retries ? : net->ipv4.sysctl_tcp_synack_retries;
     734                 :            :         thresh = max_retries;
     735                 :            :         /* Normally all the openreqs are young and become mature
     736                 :            :          * (i.e. converted to established socket) for first timeout.
     737                 :            :          * If synack was not acknowledged for 1 second, it means
     738                 :            :          * one of the following things: synack was lost, ack was lost,
     739                 :            :          * rtt is high or nobody planned to ack (i.e. synflood).
     740                 :            :          * When server is a bit loaded, queue is populated with old
     741                 :            :          * open requests, reducing effective size of queue.
     742                 :            :          * When server is well loaded, queue size reduces to zero
     743                 :            :          * after several minutes of work. It is not synflood,
     744                 :            :          * it is normal operation. The solution is pruning
     745                 :            :          * too old entries overriding normal timeout, when
     746                 :            :          * situation becomes dangerous.
     747                 :            :          *
     748                 :            :          * Essentially, we reserve half of room for young
     749                 :            :          * embrions; and abort old ones without pity, if old
     750                 :            :          * ones are about to clog our table.
     751                 :            :          */
     752                 :            :         qlen = reqsk_queue_len(queue);
     753                 :          0 :         if ((qlen << 1) > max(8U, sk_listener->sk_max_ack_backlog)) {
     754                 :          0 :                 int young = reqsk_queue_len_young(queue) << 1;
     755                 :            : 
     756                 :          0 :                 while (thresh > 2) {
     757                 :          0 :                         if (qlen < young)
     758                 :            :                                 break;
     759                 :          0 :                         thresh--;
     760                 :          0 :                         young <<= 1;
     761                 :            :                 }
     762                 :            :         }
     763                 :            :         defer_accept = READ_ONCE(queue->rskq_defer_accept);
     764                 :          0 :         if (defer_accept)
     765                 :          0 :                 max_retries = defer_accept;
     766                 :          0 :         syn_ack_recalc(req, thresh, max_retries, defer_accept,
     767                 :            :                        &expire, &resend);
     768                 :          0 :         req->rsk_ops->syn_ack_timeout(req);
     769                 :          0 :         if (!expire &&
     770                 :          0 :             (!resend ||
     771                 :          0 :              !inet_rtx_syn_ack(sk_listener, req) ||
     772                 :            :              inet_rsk(req)->acked)) {
     773                 :            :                 unsigned long timeo;
     774                 :            : 
     775                 :          0 :                 if (req->num_timeout++ == 0)
     776                 :          0 :                         atomic_dec(&queue->young);
     777                 :          0 :                 timeo = min(TCP_TIMEOUT_INIT << req->num_timeout, TCP_RTO_MAX);
     778                 :          0 :                 mod_timer(&req->rsk_timer, jiffies + timeo);
     779                 :          0 :                 return;
     780                 :            :         }
     781                 :            : drop:
     782                 :            :         inet_csk_reqsk_queue_drop_and_put(sk_listener, req);
     783                 :            : }
     784                 :            : 
     785                 :          0 : static void reqsk_queue_hash_req(struct request_sock *req,
     786                 :            :                                  unsigned long timeout)
     787                 :            : {
     788                 :          0 :         timer_setup(&req->rsk_timer, reqsk_timer_handler, TIMER_PINNED);
     789                 :          0 :         mod_timer(&req->rsk_timer, jiffies + timeout);
     790                 :            : 
     791                 :          0 :         inet_ehash_insert(req_to_sk(req), NULL);
     792                 :            :         /* before letting lookups find us, make sure all req fields
     793                 :            :          * are committed to memory and refcnt initialized.
     794                 :            :          */
     795                 :          0 :         smp_wmb();
     796                 :            :         refcount_set(&req->rsk_refcnt, 2 + 1);
     797                 :          0 : }
     798                 :            : 
     799                 :          0 : void inet_csk_reqsk_queue_hash_add(struct sock *sk, struct request_sock *req,
     800                 :            :                                    unsigned long timeout)
     801                 :            : {
     802                 :          0 :         reqsk_queue_hash_req(req, timeout);
     803                 :            :         inet_csk_reqsk_queue_added(sk);
     804                 :          0 : }
     805                 :            : EXPORT_SYMBOL_GPL(inet_csk_reqsk_queue_hash_add);
     806                 :            : 
     807                 :            : /**
     808                 :            :  *      inet_csk_clone_lock - clone an inet socket, and lock its clone
     809                 :            :  *      @sk: the socket to clone
     810                 :            :  *      @req: request_sock
     811                 :            :  *      @priority: for allocation (%GFP_KERNEL, %GFP_ATOMIC, etc)
     812                 :            :  *
     813                 :            :  *      Caller must unlock socket even in error path (bh_unlock_sock(newsk))
     814                 :            :  */
     815                 :          0 : struct sock *inet_csk_clone_lock(const struct sock *sk,
     816                 :            :                                  const struct request_sock *req,
     817                 :            :                                  const gfp_t priority)
     818                 :            : {
     819                 :          0 :         struct sock *newsk = sk_clone_lock(sk, priority);
     820                 :            : 
     821                 :          0 :         if (newsk) {
     822                 :            :                 struct inet_connection_sock *newicsk = inet_csk(newsk);
     823                 :            : 
     824                 :          0 :                 inet_sk_set_state(newsk, TCP_SYN_RECV);
     825                 :          0 :                 newicsk->icsk_bind_hash = NULL;
     826                 :            : 
     827                 :          0 :                 inet_sk(newsk)->inet_dport = inet_rsk(req)->ir_rmt_port;
     828                 :          0 :                 inet_sk(newsk)->inet_num = inet_rsk(req)->ir_num;
     829                 :          0 :                 inet_sk(newsk)->inet_sport = htons(inet_rsk(req)->ir_num);
     830                 :            : 
     831                 :            :                 /* listeners have SOCK_RCU_FREE, not the children */
     832                 :            :                 sock_reset_flag(newsk, SOCK_RCU_FREE);
     833                 :            : 
     834                 :          0 :                 inet_sk(newsk)->mc_list = NULL;
     835                 :            : 
     836                 :          0 :                 newsk->sk_mark = inet_rsk(req)->ir_mark;
     837                 :          0 :                 atomic64_set(&newsk->sk_cookie,
     838                 :          0 :                              atomic64_read(&inet_rsk(req)->ir_cookie));
     839                 :            : 
     840                 :          0 :                 newicsk->icsk_retransmits = 0;
     841                 :          0 :                 newicsk->icsk_backoff          = 0;
     842                 :          0 :                 newicsk->icsk_probes_out  = 0;
     843                 :            : 
     844                 :            :                 /* Deinitialize accept_queue to trap illegal accesses. */
     845                 :          0 :                 memset(&newicsk->icsk_accept_queue, 0, sizeof(newicsk->icsk_accept_queue));
     846                 :            : 
     847                 :          0 :                 security_inet_csk_clone(newsk, req);
     848                 :            :         }
     849                 :          0 :         return newsk;
     850                 :            : }
     851                 :            : EXPORT_SYMBOL_GPL(inet_csk_clone_lock);
     852                 :            : 
     853                 :            : /*
     854                 :            :  * At this point, there should be no process reference to this
     855                 :            :  * socket, and thus no user references at all.  Therefore we
     856                 :            :  * can assume the socket waitqueue is inactive and nobody will
     857                 :            :  * try to jump onto it.
     858                 :            :  */
     859                 :          1 : void inet_csk_destroy_sock(struct sock *sk)
     860                 :            : {
     861                 :          1 :         WARN_ON(sk->sk_state != TCP_CLOSE);
     862                 :          1 :         WARN_ON(!sock_flag(sk, SOCK_DEAD));
     863                 :            : 
     864                 :            :         /* It cannot be in hash table! */
     865                 :          1 :         WARN_ON(!sk_unhashed(sk));
     866                 :            : 
     867                 :            :         /* If it has not 0 inet_sk(sk)->inet_num, it must be bound */
     868                 :          1 :         WARN_ON(inet_sk(sk)->inet_num && !inet_csk(sk)->icsk_bind_hash);
     869                 :            : 
     870                 :          1 :         sk->sk_prot->destroy(sk);
     871                 :            : 
     872                 :          1 :         sk_stream_kill_queues(sk);
     873                 :            : 
     874                 :          1 :         xfrm_sk_free_policy(sk);
     875                 :            : 
     876                 :            :         sk_refcnt_debug_release(sk);
     877                 :            : 
     878                 :          1 :         percpu_counter_dec(sk->sk_prot->orphan_count);
     879                 :            : 
     880                 :          1 :         sock_put(sk);
     881                 :          1 : }
     882                 :            : EXPORT_SYMBOL(inet_csk_destroy_sock);
     883                 :            : 
     884                 :            : /* This function allows to force a closure of a socket after the call to
     885                 :            :  * tcp/dccp_create_openreq_child().
     886                 :            :  */
     887                 :          0 : void inet_csk_prepare_forced_close(struct sock *sk)
     888                 :            :         __releases(&sk->sk_lock.slock)
     889                 :            : {
     890                 :            :         /* sk_clone_lock locked the socket and set refcnt to 2 */
     891                 :            :         bh_unlock_sock(sk);
     892                 :          0 :         sock_put(sk);
     893                 :            : 
     894                 :            :         /* The below has to be done to allow calling inet_csk_destroy_sock */
     895                 :            :         sock_set_flag(sk, SOCK_DEAD);
     896                 :          0 :         percpu_counter_inc(sk->sk_prot->orphan_count);
     897                 :          0 :         inet_sk(sk)->inet_num = 0;
     898                 :          0 : }
     899                 :            : EXPORT_SYMBOL(inet_csk_prepare_forced_close);
     900                 :            : 
     901                 :          0 : int inet_csk_listen_start(struct sock *sk, int backlog)
     902                 :            : {
     903                 :            :         struct inet_connection_sock *icsk = inet_csk(sk);
     904                 :            :         struct inet_sock *inet = inet_sk(sk);
     905                 :            :         int err = -EADDRINUSE;
     906                 :            : 
     907                 :          0 :         reqsk_queue_alloc(&icsk->icsk_accept_queue);
     908                 :            : 
     909                 :          0 :         sk->sk_ack_backlog = 0;
     910                 :            :         inet_csk_delack_init(sk);
     911                 :            : 
     912                 :            :         /* There is race window here: we announce ourselves listening,
     913                 :            :          * but this transition is still not validated by get_port().
     914                 :            :          * It is OK, because this socket enters to hash table only
     915                 :            :          * after validation is complete.
     916                 :            :          */
     917                 :          0 :         inet_sk_state_store(sk, TCP_LISTEN);
     918                 :          0 :         if (!sk->sk_prot->get_port(sk, inet->inet_num)) {
     919                 :          0 :                 inet->inet_sport = htons(inet->inet_num);
     920                 :            : 
     921                 :            :                 sk_dst_reset(sk);
     922                 :          0 :                 err = sk->sk_prot->hash(sk);
     923                 :            : 
     924                 :          0 :                 if (likely(!err))
     925                 :            :                         return 0;
     926                 :            :         }
     927                 :            : 
     928                 :          0 :         inet_sk_set_state(sk, TCP_CLOSE);
     929                 :          0 :         return err;
     930                 :            : }
     931                 :            : EXPORT_SYMBOL_GPL(inet_csk_listen_start);
     932                 :            : 
     933                 :          0 : static void inet_child_forget(struct sock *sk, struct request_sock *req,
     934                 :            :                               struct sock *child)
     935                 :            : {
     936                 :          0 :         sk->sk_prot->disconnect(child, O_NONBLOCK);
     937                 :            : 
     938                 :          0 :         sock_orphan(child);
     939                 :            : 
     940                 :          0 :         percpu_counter_inc(sk->sk_prot->orphan_count);
     941                 :            : 
     942                 :          0 :         if (sk->sk_protocol == IPPROTO_TCP && tcp_rsk(req)->tfo_listener) {
     943                 :          0 :                 BUG_ON(rcu_access_pointer(tcp_sk(child)->fastopen_rsk) != req);
     944                 :          0 :                 BUG_ON(sk != req->rsk_listener);
     945                 :            : 
     946                 :            :                 /* Paranoid, to prevent race condition if
     947                 :            :                  * an inbound pkt destined for child is
     948                 :            :                  * blocked by sock lock in tcp_v4_rcv().
     949                 :            :                  * Also to satisfy an assertion in
     950                 :            :                  * tcp_v4_destroy_sock().
     951                 :            :                  */
     952                 :            :                 RCU_INIT_POINTER(tcp_sk(child)->fastopen_rsk, NULL);
     953                 :            :         }
     954                 :          0 :         inet_csk_destroy_sock(child);
     955                 :          0 : }
     956                 :            : 
     957                 :          0 : struct sock *inet_csk_reqsk_queue_add(struct sock *sk,
     958                 :            :                                       struct request_sock *req,
     959                 :            :                                       struct sock *child)
     960                 :            : {
     961                 :            :         struct request_sock_queue *queue = &inet_csk(sk)->icsk_accept_queue;
     962                 :            : 
     963                 :            :         spin_lock(&queue->rskq_lock);
     964                 :          0 :         if (unlikely(sk->sk_state != TCP_LISTEN)) {
     965                 :          0 :                 inet_child_forget(sk, req, child);
     966                 :            :                 child = NULL;
     967                 :            :         } else {
     968                 :          0 :                 req->sk = child;
     969                 :          0 :                 req->dl_next = NULL;
     970                 :          0 :                 if (queue->rskq_accept_head == NULL)
     971                 :          0 :                         WRITE_ONCE(queue->rskq_accept_head, req);
     972                 :            :                 else
     973                 :          0 :                         queue->rskq_accept_tail->dl_next = req;
     974                 :          0 :                 queue->rskq_accept_tail = req;
     975                 :            :                 sk_acceptq_added(sk);
     976                 :            :         }
     977                 :            :         spin_unlock(&queue->rskq_lock);
     978                 :          0 :         return child;
     979                 :            : }
     980                 :            : EXPORT_SYMBOL(inet_csk_reqsk_queue_add);
     981                 :            : 
     982                 :          0 : struct sock *inet_csk_complete_hashdance(struct sock *sk, struct sock *child,
     983                 :            :                                          struct request_sock *req, bool own_req)
     984                 :            : {
     985                 :          0 :         if (own_req) {
     986                 :          0 :                 inet_csk_reqsk_queue_drop(sk, req);
     987                 :          0 :                 reqsk_queue_removed(&inet_csk(sk)->icsk_accept_queue, req);
     988                 :          0 :                 if (inet_csk_reqsk_queue_add(sk, req, child))
     989                 :            :                         return child;
     990                 :            :         }
     991                 :            :         /* Too bad, another child took ownership of the request, undo. */
     992                 :            :         bh_unlock_sock(child);
     993                 :          0 :         sock_put(child);
     994                 :          0 :         return NULL;
     995                 :            : }
     996                 :            : EXPORT_SYMBOL(inet_csk_complete_hashdance);
     997                 :            : 
     998                 :            : /*
     999                 :            :  *      This routine closes sockets which have been at least partially
    1000                 :            :  *      opened, but not yet accepted.
    1001                 :            :  */
    1002                 :          0 : void inet_csk_listen_stop(struct sock *sk)
    1003                 :            : {
    1004                 :            :         struct inet_connection_sock *icsk = inet_csk(sk);
    1005                 :          0 :         struct request_sock_queue *queue = &icsk->icsk_accept_queue;
    1006                 :            :         struct request_sock *next, *req;
    1007                 :            : 
    1008                 :            :         /* Following specs, it would be better either to send FIN
    1009                 :            :          * (and enter FIN-WAIT-1, it is normal close)
    1010                 :            :          * or to send active reset (abort).
    1011                 :            :          * Certainly, it is pretty dangerous while synflood, but it is
    1012                 :            :          * bad justification for our negligence 8)
    1013                 :            :          * To be honest, we are not able to make either
    1014                 :            :          * of the variants now.                 --ANK
    1015                 :            :          */
    1016                 :          0 :         while ((req = reqsk_queue_remove(queue, sk)) != NULL) {
    1017                 :          0 :                 struct sock *child = req->sk;
    1018                 :            : 
    1019                 :            :                 local_bh_disable();
    1020                 :            :                 bh_lock_sock(child);
    1021                 :          0 :                 WARN_ON(sock_owned_by_user(child));
    1022                 :            :                 sock_hold(child);
    1023                 :            : 
    1024                 :          0 :                 inet_child_forget(sk, req, child);
    1025                 :          0 :                 reqsk_put(req);
    1026                 :            :                 bh_unlock_sock(child);
    1027                 :            :                 local_bh_enable();
    1028                 :          0 :                 sock_put(child);
    1029                 :            : 
    1030                 :          0 :                 cond_resched();
    1031                 :            :         }
    1032                 :          0 :         if (queue->fastopenq.rskq_rst_head) {
    1033                 :            :                 /* Free all the reqs queued in rskq_rst_head. */
    1034                 :            :                 spin_lock_bh(&queue->fastopenq.lock);
    1035                 :          0 :                 req = queue->fastopenq.rskq_rst_head;
    1036                 :          0 :                 queue->fastopenq.rskq_rst_head = NULL;
    1037                 :            :                 spin_unlock_bh(&queue->fastopenq.lock);
    1038                 :          0 :                 while (req != NULL) {
    1039                 :          0 :                         next = req->dl_next;
    1040                 :          0 :                         reqsk_put(req);
    1041                 :            :                         req = next;
    1042                 :            :                 }
    1043                 :            :         }
    1044                 :          0 :         WARN_ON_ONCE(sk->sk_ack_backlog);
    1045                 :          0 : }
    1046                 :            : EXPORT_SYMBOL_GPL(inet_csk_listen_stop);
    1047                 :            : 
    1048                 :          0 : void inet_csk_addr2sockaddr(struct sock *sk, struct sockaddr *uaddr)
    1049                 :            : {
    1050                 :            :         struct sockaddr_in *sin = (struct sockaddr_in *)uaddr;
    1051                 :            :         const struct inet_sock *inet = inet_sk(sk);
    1052                 :            : 
    1053                 :          0 :         sin->sin_family              = AF_INET;
    1054                 :          0 :         sin->sin_addr.s_addr = inet->inet_daddr;
    1055                 :          0 :         sin->sin_port                = inet->inet_dport;
    1056                 :          0 : }
    1057                 :            : EXPORT_SYMBOL_GPL(inet_csk_addr2sockaddr);
    1058                 :            : 
    1059                 :            : #ifdef CONFIG_COMPAT
    1060                 :            : int inet_csk_compat_getsockopt(struct sock *sk, int level, int optname,
    1061                 :            :                                char __user *optval, int __user *optlen)
    1062                 :            : {
    1063                 :            :         const struct inet_connection_sock *icsk = inet_csk(sk);
    1064                 :            : 
    1065                 :            :         if (icsk->icsk_af_ops->compat_getsockopt)
    1066                 :            :                 return icsk->icsk_af_ops->compat_getsockopt(sk, level, optname,
    1067                 :            :                                                             optval, optlen);
    1068                 :            :         return icsk->icsk_af_ops->getsockopt(sk, level, optname,
    1069                 :            :                                              optval, optlen);
    1070                 :            : }
    1071                 :            : EXPORT_SYMBOL_GPL(inet_csk_compat_getsockopt);
    1072                 :            : 
    1073                 :            : int inet_csk_compat_setsockopt(struct sock *sk, int level, int optname,
    1074                 :            :                                char __user *optval, unsigned int optlen)
    1075                 :            : {
    1076                 :            :         const struct inet_connection_sock *icsk = inet_csk(sk);
    1077                 :            : 
    1078                 :            :         if (icsk->icsk_af_ops->compat_setsockopt)
    1079                 :            :                 return icsk->icsk_af_ops->compat_setsockopt(sk, level, optname,
    1080                 :            :                                                             optval, optlen);
    1081                 :            :         return icsk->icsk_af_ops->setsockopt(sk, level, optname,
    1082                 :            :                                              optval, optlen);
    1083                 :            : }
    1084                 :            : EXPORT_SYMBOL_GPL(inet_csk_compat_setsockopt);
    1085                 :            : #endif
    1086                 :            : 
    1087                 :          0 : static struct dst_entry *inet_csk_rebuild_route(struct sock *sk, struct flowi *fl)
    1088                 :            : {
    1089                 :            :         const struct inet_sock *inet = inet_sk(sk);
    1090                 :            :         const struct ip_options_rcu *inet_opt;
    1091                 :          0 :         __be32 daddr = inet->inet_daddr;
    1092                 :            :         struct flowi4 *fl4;
    1093                 :            :         struct rtable *rt;
    1094                 :            : 
    1095                 :            :         rcu_read_lock();
    1096                 :          0 :         inet_opt = rcu_dereference(inet->inet_opt);
    1097                 :          0 :         if (inet_opt && inet_opt->opt.srr)
    1098                 :          0 :                 daddr = inet_opt->opt.faddr;
    1099                 :          0 :         fl4 = &fl->u.ip4;
    1100                 :          0 :         rt = ip_route_output_ports(sock_net(sk), fl4, sk, daddr,
    1101                 :            :                                    inet->inet_saddr, inet->inet_dport,
    1102                 :            :                                    inet->inet_sport, sk->sk_protocol,
    1103                 :          0 :                                    RT_CONN_FLAGS(sk), sk->sk_bound_dev_if);
    1104                 :          0 :         if (IS_ERR(rt))
    1105                 :            :                 rt = NULL;
    1106                 :          0 :         if (rt)
    1107                 :          0 :                 sk_setup_caps(sk, &rt->dst);
    1108                 :            :         rcu_read_unlock();
    1109                 :            : 
    1110                 :          0 :         return &rt->dst;
    1111                 :            : }
    1112                 :            : 
    1113                 :          0 : struct dst_entry *inet_csk_update_pmtu(struct sock *sk, u32 mtu)
    1114                 :            : {
    1115                 :          0 :         struct dst_entry *dst = __sk_dst_check(sk, 0);
    1116                 :            :         struct inet_sock *inet = inet_sk(sk);
    1117                 :            : 
    1118                 :          0 :         if (!dst) {
    1119                 :          0 :                 dst = inet_csk_rebuild_route(sk, &inet->cork.fl);
    1120                 :          0 :                 if (!dst)
    1121                 :            :                         goto out;
    1122                 :            :         }
    1123                 :          0 :         dst->ops->update_pmtu(dst, sk, NULL, mtu, true);
    1124                 :            : 
    1125                 :          0 :         dst = __sk_dst_check(sk, 0);
    1126                 :          0 :         if (!dst)
    1127                 :          0 :                 dst = inet_csk_rebuild_route(sk, &inet->cork.fl);
    1128                 :            : out:
    1129                 :          0 :         return dst;
    1130                 :            : }
    1131                 :            : EXPORT_SYMBOL_GPL(inet_csk_update_pmtu);
    

Generated by: LCOV version 1.14