hc
2024-05-09 b9d5c334faa47a75f1f28e72d203fc0334e8471d
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
// SPDX-License-Identifier: GPL-2.0-or-later
/*
 *    common UDP/RAW code
 *    Linux INET implementation
 *
 * Authors:
 *     Hideaki YOSHIFUJI <yoshfuji@linux-ipv6.org>
 */
 
#include <linux/types.h>
#include <linux/module.h>
#include <linux/ip.h>
#include <linux/in.h>
#include <net/ip.h>
#include <net/sock.h>
#include <net/route.h>
#include <net/tcp_states.h>
#include <net/sock_reuseport.h>
 
int __ip4_datagram_connect(struct sock *sk, struct sockaddr *uaddr, int addr_len)
{
   struct inet_sock *inet = inet_sk(sk);
   struct sockaddr_in *usin = (struct sockaddr_in *) uaddr;
   struct flowi4 *fl4;
   struct rtable *rt;
   __be32 saddr;
   int oif;
   int err;
 
 
   if (addr_len < sizeof(*usin))
       return -EINVAL;
 
   if (usin->sin_family != AF_INET)
       return -EAFNOSUPPORT;
 
   sk_dst_reset(sk);
 
   oif = sk->sk_bound_dev_if;
   saddr = inet->inet_saddr;
   if (ipv4_is_multicast(usin->sin_addr.s_addr)) {
       if (!oif || netif_index_is_l3_master(sock_net(sk), oif))
           oif = inet->mc_index;
       if (!saddr)
           saddr = inet->mc_addr;
   }
   fl4 = &inet->cork.fl.u.ip4;
   rt = ip_route_connect(fl4, usin->sin_addr.s_addr, saddr,
                 RT_CONN_FLAGS(sk), oif,
                 sk->sk_protocol,
                 inet->inet_sport, usin->sin_port, sk);
   if (IS_ERR(rt)) {
       err = PTR_ERR(rt);
       if (err == -ENETUNREACH)
           IP_INC_STATS(sock_net(sk), IPSTATS_MIB_OUTNOROUTES);
       goto out;
   }
 
   if ((rt->rt_flags & RTCF_BROADCAST) && !sock_flag(sk, SOCK_BROADCAST)) {
       ip_rt_put(rt);
       err = -EACCES;
       goto out;
   }
   if (!inet->inet_saddr)
       inet->inet_saddr = fl4->saddr;    /* Update source address */
   if (!inet->inet_rcv_saddr) {
       inet->inet_rcv_saddr = fl4->saddr;
       if (sk->sk_prot->rehash)
           sk->sk_prot->rehash(sk);
   }
   inet->inet_daddr = fl4->daddr;
   inet->inet_dport = usin->sin_port;
   reuseport_has_conns_set(sk);
   sk->sk_state = TCP_ESTABLISHED;
   sk_set_txhash(sk);
   inet->inet_id = prandom_u32();
 
   sk_dst_set(sk, &rt->dst);
   err = 0;
out:
   return err;
}
EXPORT_SYMBOL(__ip4_datagram_connect);
 
int ip4_datagram_connect(struct sock *sk, struct sockaddr *uaddr, int addr_len)
{
   int res;
 
   lock_sock(sk);
   res = __ip4_datagram_connect(sk, uaddr, addr_len);
   release_sock(sk);
   return res;
}
EXPORT_SYMBOL(ip4_datagram_connect);
 
/* Because UDP xmit path can manipulate sk_dst_cache without holding
 * socket lock, we need to use sk_dst_set() here,
 * even if we own the socket lock.
 */
void ip4_datagram_release_cb(struct sock *sk)
{
   const struct inet_sock *inet = inet_sk(sk);
   const struct ip_options_rcu *inet_opt;
   __be32 daddr = inet->inet_daddr;
   struct dst_entry *dst;
   struct flowi4 fl4;
   struct rtable *rt;
 
   rcu_read_lock();
 
   dst = __sk_dst_get(sk);
   if (!dst || !dst->obsolete || dst->ops->check(dst, 0)) {
       rcu_read_unlock();
       return;
   }
   inet_opt = rcu_dereference(inet->inet_opt);
   if (inet_opt && inet_opt->opt.srr)
       daddr = inet_opt->opt.faddr;
   rt = ip_route_output_ports(sock_net(sk), &fl4, sk, daddr,
                  inet->inet_saddr, inet->inet_dport,
                  inet->inet_sport, sk->sk_protocol,
                  RT_CONN_FLAGS(sk), sk->sk_bound_dev_if);
 
   dst = !IS_ERR(rt) ? &rt->dst : NULL;
   sk_dst_set(sk, dst);
 
   rcu_read_unlock();
}
EXPORT_SYMBOL_GPL(ip4_datagram_release_cb);