Lines Matching +full:conntrack +full:- +full:related
1 // SPDX-License-Identifier: GPL-2.0-or-later
33 * Fred Van Kempen : Net2e support for sk->broadcast.
63 * bound-to-device socket
77 #include <linux/bpf-cgroup.h>
138 return sk->sk_prot->h.udp_table ? : sock_net(sk)->ipv4.udp_table; in udp_get_table_prot()
149 sk_for_each(sk2, &hslot->head) { in udp_lib_lport_inuse()
152 (bitmap || udp_sk(sk2)->udp_port_hash == num) && in udp_lib_lport_inuse()
153 (!sk2->sk_reuse || !sk->sk_reuse) && in udp_lib_lport_inuse()
154 (!sk2->sk_bound_dev_if || !sk->sk_bound_dev_if || in udp_lib_lport_inuse()
155 sk2->sk_bound_dev_if == sk->sk_bound_dev_if) && in udp_lib_lport_inuse()
157 if (sk2->sk_reuseport && sk->sk_reuseport && in udp_lib_lport_inuse()
158 !rcu_access_pointer(sk->sk_reuseport_cb) && in udp_lib_lport_inuse()
165 __set_bit(udp_sk(sk2)->udp_port_hash >> log, in udp_lib_lport_inuse()
185 spin_lock(&hslot2->lock); in udp_lib_lport_inuse2()
186 udp_portaddr_for_each_entry(sk2, &hslot2->head) { in udp_lib_lport_inuse2()
189 (udp_sk(sk2)->udp_port_hash == num) && in udp_lib_lport_inuse2()
190 (!sk2->sk_reuse || !sk->sk_reuse) && in udp_lib_lport_inuse2()
191 (!sk2->sk_bound_dev_if || !sk->sk_bound_dev_if || in udp_lib_lport_inuse2()
192 sk2->sk_bound_dev_if == sk->sk_bound_dev_if) && in udp_lib_lport_inuse2()
194 if (sk2->sk_reuseport && sk->sk_reuseport && in udp_lib_lport_inuse2()
195 !rcu_access_pointer(sk->sk_reuseport_cb) && in udp_lib_lport_inuse2()
204 spin_unlock(&hslot2->lock); in udp_lib_lport_inuse2()
214 sk_for_each(sk2, &hslot->head) { in udp_reuseport_add_sock()
217 sk2->sk_family == sk->sk_family && in udp_reuseport_add_sock()
219 (udp_sk(sk2)->udp_port_hash == udp_sk(sk)->udp_port_hash) && in udp_reuseport_add_sock()
220 (sk2->sk_bound_dev_if == sk->sk_bound_dev_if) && in udp_reuseport_add_sock()
221 sk2->sk_reuseport && uid_eq(uid, sock_i_uid(sk2)) && in udp_reuseport_add_sock()
232 * udp_lib_get_port - UDP/-Lite port lookup for IPv4 and IPv6
236 * @hash2_nulladdr: AF-dependent hash value in secondary hash chains,
245 int error = -EADDRINUSE; in udp_lib_get_port()
254 remaining = (high - low) + 1; in udp_lib_get_port()
261 rand = (rand | 1) * (udptable->mask + 1); in udp_lib_get_port()
262 last = first + udptable->mask + 1; in udp_lib_get_port()
266 spin_lock_bh(&hslot->lock); in udp_lib_get_port()
268 udptable->log); in udp_lib_get_port()
278 !test_bit(snum >> udptable->log, bitmap) && in udp_lib_get_port()
283 spin_unlock_bh(&hslot->lock); in udp_lib_get_port()
289 spin_lock_bh(&hslot->lock); in udp_lib_get_port()
290 if (hslot->count > 10) { in udp_lib_get_port()
292 unsigned int slot2 = udp_sk(sk)->udp_portaddr_hash ^ snum; in udp_lib_get_port()
294 slot2 &= udptable->mask; in udp_lib_get_port()
295 hash2_nulladdr &= udptable->mask; in udp_lib_get_port()
298 if (hslot->count < hslot2->count) in udp_lib_get_port()
317 inet_sk(sk)->inet_num = snum; in udp_lib_get_port()
318 udp_sk(sk)->udp_port_hash = snum; in udp_lib_get_port()
319 udp_sk(sk)->udp_portaddr_hash ^= snum; in udp_lib_get_port()
321 if (sk->sk_reuseport && in udp_lib_get_port()
323 inet_sk(sk)->inet_num = 0; in udp_lib_get_port()
324 udp_sk(sk)->udp_port_hash = 0; in udp_lib_get_port()
325 udp_sk(sk)->udp_portaddr_hash ^= snum; in udp_lib_get_port()
331 sk_add_node_rcu(sk, &hslot->head); in udp_lib_get_port()
332 hslot->count++; in udp_lib_get_port()
333 sock_prot_inuse_add(sock_net(sk), sk->sk_prot, 1); in udp_lib_get_port()
335 hslot2 = udp_hashslot2(udptable, udp_sk(sk)->udp_portaddr_hash); in udp_lib_get_port()
336 spin_lock(&hslot2->lock); in udp_lib_get_port()
337 if (IS_ENABLED(CONFIG_IPV6) && sk->sk_reuseport && in udp_lib_get_port()
338 sk->sk_family == AF_INET6) in udp_lib_get_port()
339 hlist_add_tail_rcu(&udp_sk(sk)->udp_portaddr_node, in udp_lib_get_port()
340 &hslot2->head); in udp_lib_get_port()
342 hlist_add_head_rcu(&udp_sk(sk)->udp_portaddr_node, in udp_lib_get_port()
343 &hslot2->head); in udp_lib_get_port()
344 hslot2->count++; in udp_lib_get_port()
345 spin_unlock(&hslot2->lock); in udp_lib_get_port()
350 spin_unlock_bh(&hslot->lock); in udp_lib_get_port()
361 ipv4_portaddr_hash(sock_net(sk), inet_sk(sk)->inet_rcv_saddr, 0); in udp_v4_get_port()
364 udp_sk(sk)->udp_portaddr_hash = hash2_partial; in udp_v4_get_port()
378 udp_sk(sk)->udp_port_hash != hnum || in compute_score()
380 return -1; in compute_score()
382 if (sk->sk_rcv_saddr != daddr) in compute_score()
383 return -1; in compute_score()
385 score = (sk->sk_family == PF_INET) ? 2 : 1; in compute_score()
388 if (inet->inet_daddr) { in compute_score()
389 if (inet->inet_daddr != saddr) in compute_score()
390 return -1; in compute_score()
394 if (inet->inet_dport) { in compute_score()
395 if (inet->inet_dport != sport) in compute_score()
396 return -1; in compute_score()
400 dev_match = udp_sk_bound_dev_eq(net, sk->sk_bound_dev_if, in compute_score()
403 return -1; in compute_score()
404 if (sk->sk_bound_dev_if) in compute_score()
407 if (READ_ONCE(sk->sk_incoming_cpu) == raw_smp_processor_id()) in compute_score()
423 * udp4_lib_lookup1() - Simplified lookup using primary hash (destination port)
448 unsigned int slot = udp_hashfn(net, hnum, udptable->mask); in udp4_lib_lookup1()
449 struct udp_hslot *hslot = &udptable->hash[slot]; in udp4_lib_lookup1()
453 sk_for_each_rcu(sk, &hslot->head) { in udp4_lib_lookup1()
479 udp_portaddr_for_each_entry_rcu(sk, &hslot2->head) { in udp4_lib_lookup2()
490 if (sk->sk_state == TCP_ESTABLISHED) { in udp4_lib_lookup2()
556 slot = hash4 & udptable->mask; in udp4_lib_lookup4()
557 hslot4 = &udptable->hash4[slot]; in udp4_lib_lookup4()
562 udp_lrpa_for_each_entry_rcu(up, node, &hslot4->nulls_head) { in udp4_lib_lookup4()
584 hslot4 = udp_hashslot4(udptable, udp_sk(sk)->udp_lrpa_hash); in udp_rehash4()
586 udp_sk(sk)->udp_lrpa_hash = newhash4; in udp_rehash4()
589 spin_lock_bh(&hslot4->lock); in udp_rehash4()
590 hlist_nulls_del_init_rcu(&udp_sk(sk)->udp_lrpa_node); in udp_rehash4()
591 hslot4->count--; in udp_rehash4()
592 spin_unlock_bh(&hslot4->lock); in udp_rehash4()
594 spin_lock_bh(&nhslot4->lock); in udp_rehash4()
595 hlist_nulls_add_head_rcu(&udp_sk(sk)->udp_lrpa_node, in udp_rehash4()
596 &nhslot4->nulls_head); in udp_rehash4()
597 nhslot4->count++; in udp_rehash4()
598 spin_unlock_bh(&nhslot4->lock); in udp_rehash4()
607 hslot2 = udp_hashslot2(udptable, udp_sk(sk)->udp_portaddr_hash); in udp_unhash4()
608 hslot4 = udp_hashslot4(udptable, udp_sk(sk)->udp_lrpa_hash); in udp_unhash4()
610 spin_lock(&hslot4->lock); in udp_unhash4()
611 hlist_nulls_del_init_rcu(&udp_sk(sk)->udp_lrpa_node); in udp_unhash4()
612 hslot4->count--; in udp_unhash4()
613 spin_unlock(&hslot4->lock); in udp_unhash4()
615 spin_lock(&hslot2->lock); in udp_unhash4()
617 spin_unlock(&hslot2->lock); in udp_unhash4()
627 /* Connected udp socket can re-connect to another remote address, which in udp_lib_hash4()
633 udptable = net->ipv4.udp_table; in udp_lib_hash4()
634 hslot = udp_hashslot(udptable, net, udp_sk(sk)->udp_port_hash); in udp_lib_hash4()
635 hslot2 = udp_hashslot2(udptable, udp_sk(sk)->udp_portaddr_hash); in udp_lib_hash4()
637 udp_sk(sk)->udp_lrpa_hash = hash; in udp_lib_hash4()
639 spin_lock_bh(&hslot->lock); in udp_lib_hash4()
640 if (rcu_access_pointer(sk->sk_reuseport_cb)) in udp_lib_hash4()
643 spin_lock(&hslot4->lock); in udp_lib_hash4()
644 hlist_nulls_add_head_rcu(&udp_sk(sk)->udp_lrpa_node, in udp_lib_hash4()
645 &hslot4->nulls_head); in udp_lib_hash4()
646 hslot4->count++; in udp_lib_hash4()
647 spin_unlock(&hslot4->lock); in udp_lib_hash4()
649 spin_lock(&hslot2->lock); in udp_lib_hash4()
651 spin_unlock(&hslot2->lock); in udp_lib_hash4()
653 spin_unlock_bh(&hslot->lock); in udp_lib_hash4()
663 if (sk_unhashed(sk) || sk->sk_rcv_saddr == htonl(INADDR_ANY)) in udp4_hash4()
666 hash = udp_ehashfn(net, sk->sk_rcv_saddr, sk->sk_num, in udp4_hash4()
667 sk->sk_daddr, sk->sk_dport); in udp4_hash4()
675 * harder than this. -DaveM
696 /* Lookup connected or non-wildcard socket */ in __udp4_lib_lookup()
700 if (!IS_ERR_OR_NULL(result) && result->sk_state == TCP_ESTABLISHED) in __udp4_lib_lookup()
705 udptable == net->ipv4.udp_table) { in __udp4_lib_lookup()
715 /* Got non-wildcard socket or error on first lookup */ in __udp4_lib_lookup()
732 * 3. rehash operation updating _secondary and four-tuple_ hashes in __udp4_lib_lookup()
752 return __udp4_lib_lookup(dev_net(skb->dev), iph->saddr, sport, in __udp4_lib_lookup_skb()
753 iph->daddr, dport, inet_iif(skb), in __udp4_lib_lookup_skb()
760 const u16 offset = NAPI_GRO_CB(skb)->network_offsets[skb->encapsulation]; in udp4_lib_lookup_skb()
761 const struct iphdr *iph = (struct iphdr *)(skb->data + offset); in udp4_lib_lookup_skb()
762 struct net *net = dev_net(skb->dev); in udp4_lib_lookup_skb()
767 return __udp4_lib_lookup(net, iph->saddr, sport, in udp4_lib_lookup_skb()
768 iph->daddr, dport, iif, in udp4_lib_lookup_skb()
769 sdif, net->ipv4.udp_table, NULL); in udp4_lib_lookup_skb()
782 dif, 0, net->ipv4.udp_table, NULL); in udp4_lib_lookup()
783 if (sk && !refcount_inc_not_zero(&sk->sk_refcnt)) in udp4_lib_lookup()
798 udp_sk(sk)->udp_port_hash != hnum || in __udp_is_mcast_sock()
799 (inet->inet_daddr && inet->inet_daddr != rmt_addr) || in __udp_is_mcast_sock()
800 (inet->inet_dport != rmt_port && inet->inet_dport) || in __udp_is_mcast_sock()
801 (inet->inet_rcv_saddr && inet->inet_rcv_saddr != loc_addr) || in __udp_is_mcast_sock()
803 !udp_sk_bound_dev_eq(net, sk->sk_bound_dev_if, dif, sdif)) in __udp_is_mcast_sock()
844 handler = encap->err_handler; in __udp4_lib_err_encap_no_sk()
849 return -ENOENT; in __udp4_lib_err_encap_no_sk()
887 skb_set_transport_header(skb, iph->ihl << 2); in __udp4_lib_err_encap()
892 lookup = READ_ONCE(up->encap_err_lookup); in __udp4_lib_err_encap()
899 sk = __udp4_lib_lookup(net, iph->daddr, uh->source, in __udp4_lib_err_encap()
900 iph->saddr, uh->dest, skb->dev->ifindex, 0, in __udp4_lib_err_encap()
905 lookup = READ_ONCE(up->encap_err_lookup); in __udp4_lib_err_encap()
934 const struct iphdr *iph = (const struct iphdr *)skb->data; in __udp4_lib_err()
935 struct udphdr *uh = (struct udphdr *)(skb->data+(iph->ihl<<2)); in __udp4_lib_err()
936 const int type = icmp_hdr(skb)->type; in __udp4_lib_err()
937 const int code = icmp_hdr(skb)->code; in __udp4_lib_err()
942 struct net *net = dev_net(skb->dev); in __udp4_lib_err()
944 sk = __udp4_lib_lookup(net, iph->daddr, uh->dest, in __udp4_lib_err()
945 iph->saddr, uh->source, skb->dev->ifindex, in __udp4_lib_err()
948 if (!sk || READ_ONCE(udp_sk(sk)->encap_type)) { in __udp4_lib_err()
956 sk = ERR_PTR(-ENOENT); in __udp4_lib_err()
984 if (READ_ONCE(inet->pmtudisc) != IP_PMTUDISC_DONT) { in __udp4_lib_err()
1008 if (udp_sk(sk)->encap_err_rcv) in __udp4_lib_err()
1009 udp_sk(sk)->encap_err_rcv(sk, skb, err, uh->dest, info, in __udp4_lib_err()
1014 if (!harderr || sk->sk_state != TCP_ESTABLISHED) in __udp4_lib_err()
1017 ip_icmp_error(sk, skb, err, uh->dest, info, (u8 *)(uh+1)); in __udp4_lib_err()
1019 sk->sk_err = err; in __udp4_lib_err()
1027 return __udp4_lib_err(skb, info, dev_net(skb->dev)->ipv4.udp_table); in udp_err()
1037 if (up->pending) { in udp_flush_pending_frames()
1038 up->len = 0; in udp_flush_pending_frames()
1039 WRITE_ONCE(up->pending, 0); in udp_flush_pending_frames()
1046 * udp4_hwcsum - handle outgoing HW checksumming
1047 * @skb: sk_buff containing the filled-in UDP header
1056 int len = skb->len - offset; in udp4_hwcsum()
1064 skb->csum_start = skb_transport_header(skb) - skb->head; in udp4_hwcsum()
1065 skb->csum_offset = offsetof(struct udphdr, check); in udp4_hwcsum()
1066 uh->check = ~csum_tcpudp_magic(src, dst, len, in udp4_hwcsum()
1072 * HW-checksum won't work as there are two or more in udp4_hwcsum()
1077 csum = csum_add(csum, frags->csum); in udp4_hwcsum()
1078 hlen -= frags->len; in udp4_hwcsum()
1082 skb->ip_summed = CHECKSUM_NONE; in udp4_hwcsum()
1084 uh->check = csum_tcpudp_magic(src, dst, len, IPPROTO_UDP, csum); in udp4_hwcsum()
1085 if (uh->check == 0) in udp4_hwcsum()
1086 uh->check = CSUM_MANGLED_0; in udp4_hwcsum()
1100 uh->check = 0; in udp_set_csum()
1102 uh->check = ~udp_v4_check(len, saddr, daddr, 0); in udp_set_csum()
1103 } else if (skb->ip_summed == CHECKSUM_PARTIAL) { in udp_set_csum()
1104 uh->check = 0; in udp_set_csum()
1105 uh->check = udp_v4_check(len, saddr, daddr, lco_csum(skb)); in udp_set_csum()
1106 if (uh->check == 0) in udp_set_csum()
1107 uh->check = CSUM_MANGLED_0; in udp_set_csum()
1109 skb->ip_summed = CHECKSUM_PARTIAL; in udp_set_csum()
1110 skb->csum_start = skb_transport_header(skb) - skb->head; in udp_set_csum()
1111 skb->csum_offset = offsetof(struct udphdr, check); in udp_set_csum()
1112 uh->check = ~udp_v4_check(len, saddr, daddr, 0); in udp_set_csum()
1120 struct sock *sk = skb->sk; in udp_send_skb()
1126 int len = skb->len - offset; in udp_send_skb()
1127 int datalen = len - sizeof(*uh); in udp_send_skb()
1134 uh->source = inet->inet_sport; in udp_send_skb()
1135 uh->dest = fl4->fl4_dport; in udp_send_skb()
1136 uh->len = htons(len); in udp_send_skb()
1137 uh->check = 0; in udp_send_skb()
1139 if (cork->gso_size) { in udp_send_skb()
1143 if (hlen + min(datalen, cork->gso_size) > cork->fragsize) { in udp_send_skb()
1145 return -EMSGSIZE; in udp_send_skb()
1147 if (datalen > cork->gso_size * UDP_MAX_SEGMENTS) { in udp_send_skb()
1149 return -EINVAL; in udp_send_skb()
1151 if (sk->sk_no_check_tx) { in udp_send_skb()
1153 return -EINVAL; in udp_send_skb()
1157 return -EIO; in udp_send_skb()
1160 if (datalen > cork->gso_size) { in udp_send_skb()
1161 skb_shinfo(skb)->gso_size = cork->gso_size; in udp_send_skb()
1162 skb_shinfo(skb)->gso_type = SKB_GSO_UDP_L4; in udp_send_skb()
1163 skb_shinfo(skb)->gso_segs = DIV_ROUND_UP(datalen, in udp_send_skb()
1164 cork->gso_size); in udp_send_skb()
1171 if (is_udplite) /* UDP-Lite */ in udp_send_skb()
1174 else if (sk->sk_no_check_tx) { /* UDP csum off */ in udp_send_skb()
1176 skb->ip_summed = CHECKSUM_NONE; in udp_send_skb()
1179 } else if (skb->ip_summed == CHECKSUM_PARTIAL) { /* UDP hardware csum */ in udp_send_skb()
1182 udp4_hwcsum(skb, fl4->saddr, fl4->daddr); in udp_send_skb()
1188 /* add protocol-dependent pseudo-header */ in udp_send_skb()
1189 uh->check = csum_tcpudp_magic(fl4->saddr, fl4->daddr, len, in udp_send_skb()
1190 sk->sk_protocol, csum); in udp_send_skb()
1191 if (uh->check == 0) in udp_send_skb()
1192 uh->check = CSUM_MANGLED_0; in udp_send_skb()
1197 if (err == -ENOBUFS && in udp_send_skb()
1216 struct flowi4 *fl4 = &inet->cork.fl.u.ip4; in udp_push_pending_frames()
1224 err = udp_send_skb(skb, fl4, &inet->cork.base); in udp_push_pending_frames()
1227 up->len = 0; in udp_push_pending_frames()
1228 WRITE_ONCE(up->pending, 0); in udp_push_pending_frames()
1235 switch (cmsg->cmsg_type) { in __udp_cmsg_send()
1237 if (cmsg->cmsg_len != CMSG_LEN(sizeof(__u16))) in __udp_cmsg_send()
1238 return -EINVAL; in __udp_cmsg_send()
1242 return -EINVAL; in __udp_cmsg_send()
1254 return -EINVAL; in udp_cmsg_send()
1256 if (cmsg->cmsg_level != SOL_UDP) { in udp_cmsg_send()
1274 DECLARE_SOCKADDR(struct sockaddr_in *, usin, msg->msg_name); in udp_sendmsg()
1286 int corkreq = udp_test_bit(CORK, sk) || msg->msg_flags & MSG_MORE; in udp_sendmsg()
1293 return -EMSGSIZE; in udp_sendmsg()
1299 if (msg->msg_flags & MSG_OOB) /* Mirror BSD error message compatibility */ in udp_sendmsg()
1300 return -EOPNOTSUPP; in udp_sendmsg()
1304 fl4 = &inet->cork.fl.u.ip4; in udp_sendmsg()
1305 if (READ_ONCE(up->pending)) { in udp_sendmsg()
1311 if (likely(up->pending)) { in udp_sendmsg()
1312 if (unlikely(up->pending != AF_INET)) { in udp_sendmsg()
1314 return -EINVAL; in udp_sendmsg()
1326 if (msg->msg_namelen < sizeof(*usin)) in udp_sendmsg()
1327 return -EINVAL; in udp_sendmsg()
1328 if (usin->sin_family != AF_INET) { in udp_sendmsg()
1329 if (usin->sin_family != AF_UNSPEC) in udp_sendmsg()
1330 return -EAFNOSUPPORT; in udp_sendmsg()
1333 daddr = usin->sin_addr.s_addr; in udp_sendmsg()
1334 dport = usin->sin_port; in udp_sendmsg()
1336 return -EINVAL; in udp_sendmsg()
1338 if (sk->sk_state != TCP_ESTABLISHED) in udp_sendmsg()
1339 return -EDESTADDRREQ; in udp_sendmsg()
1340 daddr = inet->inet_daddr; in udp_sendmsg()
1341 dport = inet->inet_dport; in udp_sendmsg()
1349 ipc.gso_size = READ_ONCE(up->gso_size); in udp_sendmsg()
1351 if (msg->msg_controllen) { in udp_sendmsg()
1355 sk->sk_family == AF_INET6); in udp_sendmsg()
1369 inet_opt = rcu_dereference(inet->inet_opt); in udp_sendmsg()
1372 sizeof(*inet_opt) + inet_opt->opt.optlen); in udp_sendmsg()
1381 &msg->msg_namelen, in udp_sendmsg()
1386 if (usin->sin_port == 0) { in udp_sendmsg()
1388 err = -EINVAL; in udp_sendmsg()
1391 daddr = usin->sin_addr.s_addr; in udp_sendmsg()
1392 dport = usin->sin_port; in udp_sendmsg()
1399 if (ipc.opt && ipc.opt->opt.srr) { in udp_sendmsg()
1401 err = -EINVAL; in udp_sendmsg()
1404 faddr = ipc.opt->opt.faddr; in udp_sendmsg()
1411 uc_index = READ_ONCE(inet->uc_index); in udp_sendmsg()
1414 ipc.oif = READ_ONCE(inet->mc_index); in udp_sendmsg()
1416 saddr = READ_ONCE(inet->mc_addr); in udp_sendmsg()
1445 sk->sk_protocol, flow_flags, faddr, saddr, in udp_sendmsg()
1446 dport, inet->inet_sport, sk->sk_uid); in udp_sendmsg()
1453 if (err == -ENETUNREACH) in udp_sendmsg()
1458 err = -EACCES; in udp_sendmsg()
1459 if ((rt->rt_flags & RTCF_BROADCAST) && in udp_sendmsg()
1463 sk_dst_set(sk, dst_clone(&rt->dst)); in udp_sendmsg()
1466 if (msg->msg_flags&MSG_CONFIRM) in udp_sendmsg()
1470 saddr = fl4->saddr; in udp_sendmsg()
1472 daddr = ipc.addr = fl4->daddr; in udp_sendmsg()
1474 /* Lockless fast path for the non-corking case. */ in udp_sendmsg()
1480 &cork, msg->msg_flags); in udp_sendmsg()
1488 if (unlikely(up->pending)) { in udp_sendmsg()
1490 /* ... which is an evident application bug. --ANK */ in udp_sendmsg()
1494 err = -EINVAL; in udp_sendmsg()
1500 fl4 = &inet->cork.fl.u.ip4; in udp_sendmsg()
1501 fl4->daddr = daddr; in udp_sendmsg()
1502 fl4->saddr = saddr; in udp_sendmsg()
1503 fl4->fl4_dport = dport; in udp_sendmsg()
1504 fl4->fl4_sport = inet->inet_sport; in udp_sendmsg()
1505 WRITE_ONCE(up->pending, AF_INET); in udp_sendmsg()
1508 up->len += ulen; in udp_sendmsg()
1511 corkreq ? msg->msg_flags|MSG_MORE : msg->msg_flags); in udp_sendmsg()
1516 else if (unlikely(skb_queue_empty(&sk->sk_write_queue))) in udp_sendmsg()
1517 WRITE_ONCE(up->pending, 0); in udp_sendmsg()
1534 if (err == -ENOBUFS || test_bit(SOCK_NOSPACE, &sk->sk_socket->flags)) { in udp_sendmsg()
1541 if (msg->msg_flags & MSG_PROBE) in udp_sendmsg()
1542 dst_confirm_neigh(&rt->dst, &fl4->daddr); in udp_sendmsg()
1543 if (!(msg->msg_flags&MSG_PROBE) || len) in udp_sendmsg()
1552 struct sock *sk = sock->sk; in udp_splice_eof()
1555 if (!READ_ONCE(up->pending) || udp_test_bit(CORK, sk)) in udp_splice_eof()
1559 if (up->pending && !udp_test_bit(CORK, sk)) in udp_splice_eof()
1567 /* all head states (dst, sk, nf conntrack) except skb extensions are
1593 scratch->_tsize_state = skb->truesize; in udp_set_dev_scratch()
1595 scratch->len = skb->len; in udp_set_dev_scratch()
1596 scratch->csum_unnecessary = !!skb_csum_unnecessary(skb); in udp_set_dev_scratch()
1597 scratch->is_linear = !skb_is_nonlinear(skb); in udp_set_dev_scratch()
1600 scratch->_tsize_state |= UDP_SKB_IS_STATELESS; in udp_set_dev_scratch()
1607 * set skb->csum_valid to 1. in udp_skb_csum_unnecessary_set()
1613 udp_skb_scratch(skb)->csum_unnecessary = true; in udp_skb_csum_unnecessary_set()
1619 return udp_skb_scratch(skb)->_tsize_state & ~UDP_SKB_IS_STATELESS; in udp_skb_truesize()
1624 return !(udp_skb_scratch(skb)->_tsize_state & UDP_SKB_IS_STATELESS); in udp_skb_has_head_state()
1636 up->forward_deficit += size; in udp_rmem_release()
1637 size = up->forward_deficit; in udp_rmem_release()
1638 if (size < READ_ONCE(up->forward_threshold) && in udp_rmem_release()
1639 !skb_queue_empty(&up->reader_queue)) in udp_rmem_release()
1642 size += up->forward_deficit; in udp_rmem_release()
1644 up->forward_deficit = 0; in udp_rmem_release()
1649 sk_queue = &sk->sk_receive_queue; in udp_rmem_release()
1651 spin_lock(&sk_queue->lock); in udp_rmem_release()
1653 amt = (size + sk->sk_forward_alloc - partial) & ~(PAGE_SIZE - 1); in udp_rmem_release()
1654 sk_forward_alloc_add(sk, size - amt); in udp_rmem_release()
1659 atomic_sub(size, &sk->sk_rmem_alloc); in udp_rmem_release()
1662 skb_queue_splice_tail_init(sk_queue, &up->reader_queue); in udp_rmem_release()
1665 spin_unlock(&sk_queue->lock); in udp_rmem_release()
1669 * Instead of using skb->truesize here, find a copy of it in skb->dev_scratch
1675 prefetch(&skb->data); in udp_skb_destructor()
1683 prefetch(&skb->data); in udp_skb_dtor_locked()
1716 delta = size - sk->sk_forward_alloc; in udp_rmem_schedule()
1718 return -ENOBUFS; in udp_rmem_schedule()
1725 struct sk_buff_head *list = &sk->sk_receive_queue; in __udp_enqueue_schedule_skb()
1728 int size, err = -ENOMEM; in __udp_enqueue_schedule_skb()
1730 rmem = atomic_read(&sk->sk_rmem_alloc); in __udp_enqueue_schedule_skb()
1731 rcvbuf = READ_ONCE(sk->sk_rcvbuf); in __udp_enqueue_schedule_skb()
1732 size = skb->truesize; in __udp_enqueue_schedule_skb()
1748 * - Reduce memory overhead and thus increase receive queue capacity in __udp_enqueue_schedule_skb()
1749 * - Less cache line misses at copyout() time in __udp_enqueue_schedule_skb()
1750 * - Less work at consume_skb() (less alien page frag freeing) in __udp_enqueue_schedule_skb()
1754 size = skb->truesize; in __udp_enqueue_schedule_skb()
1760 atomic_add(size, &sk->sk_rmem_alloc); in __udp_enqueue_schedule_skb()
1762 spin_lock(&list->lock); in __udp_enqueue_schedule_skb()
1765 spin_unlock(&list->lock); in __udp_enqueue_schedule_skb()
1769 sk_forward_alloc_add(sk, -size); in __udp_enqueue_schedule_skb()
1777 spin_unlock(&list->lock); in __udp_enqueue_schedule_skb()
1780 INDIRECT_CALL_1(sk->sk_data_ready, sock_def_readable, sk); in __udp_enqueue_schedule_skb()
1786 atomic_sub(skb->truesize, &sk->sk_rmem_alloc); in __udp_enqueue_schedule_skb()
1789 atomic_inc(&sk->sk_drops); in __udp_enqueue_schedule_skb()
1802 skb_queue_splice_tail_init(&sk->sk_receive_queue, &up->reader_queue); in udp_destruct_common()
1803 while ((skb = __skb_dequeue(&up->reader_queue)) != NULL) { in udp_destruct_common()
1804 total += skb->truesize; in udp_destruct_common()
1820 sk->sk_destruct = udp_destruct_sock; in udp_init_sock()
1821 set_bit(SOCK_SUPPORT_ZC, &sk->sk_socket->flags); in udp_init_sock()
1827 if (unlikely(READ_ONCE(udp_sk(sk)->peeking_with_offset))) in skb_consume_udp()
1854 atomic_inc(&sk->sk_drops); in __first_packet_length()
1856 *total += skb->truesize; in __first_packet_length()
1867 * first_packet_length - return length of first packet in receive queue
1871 * Returns the length of found skb, or -1 if none is found.
1875 struct sk_buff_head *rcvq = &udp_sk(sk)->reader_queue; in first_packet_length()
1876 struct sk_buff_head *sk_queue = &sk->sk_receive_queue; in first_packet_length()
1881 spin_lock_bh(&rcvq->lock); in first_packet_length()
1884 spin_lock(&sk_queue->lock); in first_packet_length()
1886 spin_unlock(&sk_queue->lock); in first_packet_length()
1890 res = skb ? skb->len : -1; in first_packet_length()
1893 spin_unlock_bh(&rcvq->lock); in first_packet_length()
1917 return -ENOIOCTLCMD; in udp_ioctl()
1927 struct sk_buff_head *sk_queue = &sk->sk_receive_queue; in __skb_recv_udp()
1933 queue = &udp_sk(sk)->reader_queue; in __skb_recv_udp()
1942 error = -EAGAIN; in __skb_recv_udp()
1944 spin_lock_bh(&queue->lock); in __skb_recv_udp()
1950 spin_unlock_bh(&queue->lock); in __skb_recv_udp()
1955 spin_unlock_bh(&queue->lock); in __skb_recv_udp()
1960 * keep both queues locked to avoid re-acquiring in __skb_recv_udp()
1964 spin_lock(&sk_queue->lock); in __skb_recv_udp()
1971 spin_unlock(&sk_queue->lock); in __skb_recv_udp()
1972 spin_unlock_bh(&queue->lock); in __skb_recv_udp()
1985 !__skb_wait_for_more_packets(sk, &sk->sk_receive_queue, in __skb_recv_udp()
2010 atomic_inc(&sk->sk_drops); in udp_read_skb()
2029 DECLARE_SOCKADDR(struct sockaddr_in *, sin, msg->msg_name); in udp_recvmsg()
2047 if (copied > ulen - off) in udp_recvmsg()
2048 copied = ulen - off; in udp_recvmsg()
2050 msg->msg_flags |= MSG_TRUNC; in udp_recvmsg()
2055 * coverage checksum (UDP-Lite), do it before the copy. in udp_recvmsg()
2059 (is_udplite && UDP_SKB_CB(skb)->partial_cov)) { in udp_recvmsg()
2068 err = copy_linear_skb(skb, copied, off, &msg->msg_iter); in udp_recvmsg()
2074 if (err == -EINVAL) in udp_recvmsg()
2080 atomic_inc(&sk->sk_drops); in udp_recvmsg()
2096 sin->sin_family = AF_INET; in udp_recvmsg()
2097 sin->sin_port = udp_hdr(skb)->source; in udp_recvmsg()
2098 sin->sin_addr.s_addr = ip_hdr(skb)->saddr; in udp_recvmsg()
2099 memset(sin->sin_zero, 0, sizeof(sin->sin_zero)); in udp_recvmsg()
2117 skb_consume_udp(sk, skb, peeking ? -err : err); in udp_recvmsg()
2121 if (!__sk_queue_drop_skb(sk, &udp_sk(sk)->reader_queue, skb, flags, in udp_recvmsg()
2130 msg->msg_flags &= ~MSG_TRUNC; in udp_recvmsg()
2141 return -EINVAL; in udp_pre_connect()
2163 * 1003.1g - break association. in __udp_disconnect()
2166 sk->sk_state = TCP_CLOSE; in __udp_disconnect()
2167 inet->inet_daddr = 0; in __udp_disconnect()
2168 inet->inet_dport = 0; in __udp_disconnect()
2170 sk->sk_bound_dev_if = 0; in __udp_disconnect()
2171 if (!(sk->sk_userlocks & SOCK_BINDADDR_LOCK)) { in __udp_disconnect()
2173 if (sk->sk_prot->rehash && in __udp_disconnect()
2174 (sk->sk_userlocks & SOCK_BINDPORT_LOCK)) in __udp_disconnect()
2175 sk->sk_prot->rehash(sk); in __udp_disconnect()
2178 if (!(sk->sk_userlocks & SOCK_BINDPORT_LOCK)) { in __udp_disconnect()
2179 sk->sk_prot->unhash(sk); in __udp_disconnect()
2180 inet->inet_sport = 0; in __udp_disconnect()
2203 udp_sk(sk)->udp_port_hash); in udp_lib_unhash()
2204 hslot2 = udp_hashslot2(udptable, udp_sk(sk)->udp_portaddr_hash); in udp_lib_unhash()
2206 spin_lock_bh(&hslot->lock); in udp_lib_unhash()
2207 if (rcu_access_pointer(sk->sk_reuseport_cb)) in udp_lib_unhash()
2210 hslot->count--; in udp_lib_unhash()
2211 inet_sk(sk)->inet_num = 0; in udp_lib_unhash()
2212 sock_prot_inuse_add(sock_net(sk), sk->sk_prot, -1); in udp_lib_unhash()
2214 spin_lock(&hslot2->lock); in udp_lib_unhash()
2215 hlist_del_init_rcu(&udp_sk(sk)->udp_portaddr_node); in udp_lib_unhash()
2216 hslot2->count--; in udp_lib_unhash()
2217 spin_unlock(&hslot2->lock); in udp_lib_unhash()
2221 spin_unlock_bh(&hslot->lock); in udp_lib_unhash()
2236 udp_sk(sk)->udp_port_hash); in udp_lib_rehash()
2237 hslot2 = udp_hashslot2(udptable, udp_sk(sk)->udp_portaddr_hash); in udp_lib_rehash()
2239 udp_sk(sk)->udp_portaddr_hash = newhash; in udp_lib_rehash()
2242 rcu_access_pointer(sk->sk_reuseport_cb)) { in udp_lib_rehash()
2244 spin_lock_bh(&hslot->lock); in udp_lib_rehash()
2245 if (rcu_access_pointer(sk->sk_reuseport_cb)) in udp_lib_rehash()
2249 spin_lock(&hslot2->lock); in udp_lib_rehash()
2250 hlist_del_init_rcu(&udp_sk(sk)->udp_portaddr_node); in udp_lib_rehash()
2251 hslot2->count--; in udp_lib_rehash()
2252 spin_unlock(&hslot2->lock); in udp_lib_rehash()
2254 spin_lock(&nhslot2->lock); in udp_lib_rehash()
2255 hlist_add_head_rcu(&udp_sk(sk)->udp_portaddr_node, in udp_lib_rehash()
2256 &nhslot2->head); in udp_lib_rehash()
2257 nhslot2->count++; in udp_lib_rehash()
2258 spin_unlock(&nhslot2->lock); in udp_lib_rehash()
2261 spin_unlock_bh(&hslot->lock); in udp_lib_rehash()
2266 * (2) update hslot2->hash4_cnt. in udp_lib_rehash()
2271 spin_lock_bh(&hslot->lock); in udp_lib_rehash()
2275 spin_lock(&hslot2->lock); in udp_lib_rehash()
2277 spin_unlock(&hslot2->lock); in udp_lib_rehash()
2279 spin_lock(&nhslot2->lock); in udp_lib_rehash()
2281 spin_unlock(&nhslot2->lock); in udp_lib_rehash()
2284 spin_unlock_bh(&hslot->lock); in udp_lib_rehash()
2293 inet_sk(sk)->inet_rcv_saddr, in udp_v4_rehash()
2294 inet_sk(sk)->inet_num); in udp_v4_rehash()
2296 sk->sk_rcv_saddr, sk->sk_num, in udp_v4_rehash()
2297 sk->sk_daddr, sk->sk_dport); in udp_v4_rehash()
2306 if (inet_sk(sk)->inet_daddr) { in __udp_queue_rcv_skb()
2320 if (rc == -ENOMEM) { in __udp_queue_rcv_skb()
2332 return -1; in __udp_queue_rcv_skb()
2339 * -1: error
2362 READ_ONCE(up->encap_type)) { in udp_queue_rcv_one_skb()
2369 * up->encap_rcv() returns the following value: in udp_queue_rcv_one_skb()
2373 * <0 if skb should be resubmitted as proto -N in udp_queue_rcv_one_skb()
2377 encap_rcv = READ_ONCE(up->encap_rcv); in udp_queue_rcv_one_skb()
2390 return -ret; in udp_queue_rcv_one_skb()
2394 /* FALLTHROUGH -- it's a UDP Packet */ in udp_queue_rcv_one_skb()
2398 * UDP-Lite specific tests, ignored on UDP sockets in udp_queue_rcv_one_skb()
2400 if (udp_test_bit(UDPLITE_RECV_CC, sk) && UDP_SKB_CB(skb)->partial_cov) { in udp_queue_rcv_one_skb()
2401 u16 pcrlen = READ_ONCE(up->pcrlen); in udp_queue_rcv_one_skb()
2416 UDP_SKB_CB(skb)->cscov, skb->len); in udp_queue_rcv_one_skb()
2423 * Therefore the above ...()->partial_cov statement is essential. in udp_queue_rcv_one_skb()
2425 if (UDP_SKB_CB(skb)->cscov < pcrlen) { in udp_queue_rcv_one_skb()
2427 UDP_SKB_CB(skb)->cscov, pcrlen); in udp_queue_rcv_one_skb()
2432 prefetch(&sk->sk_rmem_alloc); in udp_queue_rcv_one_skb()
2433 if (rcu_access_pointer(sk->sk_filter) && in udp_queue_rcv_one_skb()
2452 atomic_inc(&sk->sk_drops); in udp_queue_rcv_one_skb()
2454 return -1; in udp_queue_rcv_one_skb()
2466 __skb_push(skb, -skb_mac_offset(skb)); in udp_queue_rcv_skb()
2474 ip_protocol_deliver_rcu(dev_net(skb->dev), skb, ret); in udp_queue_rcv_skb()
2487 old = unrcu_pointer(xchg(&sk->sk_rx_dst, RCU_INITIALIZER(dst))); in udp_sk_rx_dst_set()
2507 unsigned short hnum = ntohs(uh->dest); in __udp4_lib_mcast_deliver()
2509 unsigned int hash2 = 0, hash2_any = 0, use_hash2 = (hslot->count > 10); in __udp4_lib_mcast_deliver()
2511 int dif = skb->dev->ifindex; in __udp4_lib_mcast_deliver()
2518 udptable->mask; in __udp4_lib_mcast_deliver()
2519 hash2 = ipv4_portaddr_hash(net, daddr, hnum) & udptable->mask; in __udp4_lib_mcast_deliver()
2521 hslot = &udptable->hash2[hash2].hslot; in __udp4_lib_mcast_deliver()
2525 sk_for_each_entry_offset_rcu(sk, node, &hslot->head, offset) { in __udp4_lib_mcast_deliver()
2526 if (!__udp_is_mcast_sock(net, sk, uh->dest, daddr, in __udp4_lib_mcast_deliver()
2527 uh->source, saddr, dif, sdif, hnum)) in __udp4_lib_mcast_deliver()
2537 atomic_inc(&sk->sk_drops); in __udp4_lib_mcast_deliver()
2568 * including udp header and folding it to skb->csum.
2575 UDP_SKB_CB(skb)->partial_cov = 0; in udp4_csum_init()
2576 UDP_SKB_CB(skb)->cscov = skb->len; in udp4_csum_init()
2583 if (UDP_SKB_CB(skb)->partial_cov) { in udp4_csum_init()
2584 skb->csum = inet_compute_pseudo(skb, proto); in udp4_csum_init()
2592 err = (__force int)skb_checksum_init_zero_check(skb, proto, uh->check, in udp4_csum_init()
2597 if (skb->ip_summed == CHECKSUM_COMPLETE && !skb->csum_valid) { in udp4_csum_init()
2599 if (skb->csum_complete_sw) in udp4_csum_init()
2603 * skb->csum is no longer the full packet checksum, in udp4_csum_init()
2620 if (inet_get_convert_csum(sk) && uh->check && !IS_UDPLITE(sk)) in udp_unicast_rcv_skb()
2626 * it wants the return to be -protocol, or 0 in udp_unicast_rcv_skb()
2629 return -ret; in udp_unicast_rcv_skb()
2645 struct net *net = dev_net(skb->dev); in __udp4_lib_rcv()
2658 ulen = ntohs(uh->len); in __udp4_lib_rcv()
2659 saddr = ip_hdr(skb)->saddr; in __udp4_lib_rcv()
2660 daddr = ip_hdr(skb)->daddr; in __udp4_lib_rcv()
2662 if (ulen > skb->len) in __udp4_lib_rcv()
2675 sk = inet_steal_sock(net, skb, sizeof(struct udphdr), saddr, uh->source, daddr, uh->dest, in __udp4_lib_rcv()
2684 if (unlikely(rcu_dereference(sk->sk_rx_dst) != dst)) in __udp4_lib_rcv()
2693 if (rt->rt_flags & (RTCF_BROADCAST|RTCF_MULTICAST)) in __udp4_lib_rcv()
2697 sk = __udp4_lib_lookup_skb(skb, uh->source, uh->dest, udptable); in __udp4_lib_rcv()
2724 &saddr, ntohs(uh->source), in __udp4_lib_rcv()
2725 ulen, skb->len, in __udp4_lib_rcv()
2726 &daddr, ntohs(uh->dest)); in __udp4_lib_rcv()
2737 &saddr, ntohs(uh->source), &daddr, ntohs(uh->dest), in __udp4_lib_rcv()
2754 struct udp_table *udptable = net->ipv4.udp_table; in __udp4_lib_mcast_demux_lookup()
2760 slot = udp_hashfn(net, hnum, udptable->mask); in __udp4_lib_mcast_demux_lookup()
2761 hslot = &udptable->hash[slot]; in __udp4_lib_mcast_demux_lookup()
2764 if (hslot->count > 10) in __udp4_lib_mcast_demux_lookup()
2768 sk_for_each_rcu(sk, &hslot->head) { in __udp4_lib_mcast_demux_lookup()
2789 struct udp_table *udptable = net->ipv4.udp_table; in __udp4_lib_demux_lookup()
2801 udp_portaddr_for_each_entry_rcu(sk, &hslot2->head) { in __udp4_lib_demux_lookup()
2812 struct net *net = dev_net(skb->dev); in udp_v4_early_demux()
2818 int dif = skb->dev->ifindex; in udp_v4_early_demux()
2829 if (skb->pkt_type == PACKET_MULTICAST) { in udp_v4_early_demux()
2830 in_dev = __in_dev_get_rcu(skb->dev); in udp_v4_early_demux()
2835 ours = ip_check_mc_rcu(in_dev, iph->daddr, iph->saddr, in udp_v4_early_demux()
2836 iph->protocol); in udp_v4_early_demux()
2840 sk = __udp4_lib_mcast_demux_lookup(net, uh->dest, iph->daddr, in udp_v4_early_demux()
2841 uh->source, iph->saddr, in udp_v4_early_demux()
2843 } else if (skb->pkt_type == PACKET_HOST) { in udp_v4_early_demux()
2844 sk = __udp4_lib_demux_lookup(net, uh->dest, iph->daddr, in udp_v4_early_demux()
2845 uh->source, iph->saddr, dif, sdif); in udp_v4_early_demux()
2851 skb->sk = sk; in udp_v4_early_demux()
2853 skb->destructor = sock_pfree; in udp_v4_early_demux()
2854 dst = rcu_dereference(sk->sk_rx_dst); in udp_v4_early_demux()
2870 if (!inet_sk(sk)->inet_daddr && in_dev) in udp_v4_early_demux()
2871 return ip_mc_validate_source(skb, iph->daddr, in udp_v4_early_demux()
2872 iph->saddr, in udp_v4_early_demux()
2874 skb->dev, in_dev, &itag); in udp_v4_early_demux()
2881 return __udp4_lib_rcv(skb, dev_net(skb->dev)->ipv4.udp_table, IPPROTO_UDP); in udp_rcv()
2894 if (up->encap_type) { in udp_destroy_sock()
2896 encap_destroy = READ_ONCE(up->encap_destroy); in udp_destroy_sock()
2911 WRITE_ONCE(udp_sk(sk)->gro_receive, xfrm4_gro_udp_encap_rcv); in set_xfrm_gro_udp_encap_rcv()
2913 WRITE_ONCE(udp_sk(sk)->gro_receive, ipv6_stub->xfrm6_gro_udp_encap_rcv); in set_xfrm_gro_udp_encap_rcv()
2936 WRITE_ONCE(up->forward_threshold, sk->sk_rcvbuf >> 2); in udp_lib_setsockopt()
2943 return -EINVAL; in udp_lib_setsockopt()
2946 return -EFAULT; in udp_lib_setsockopt()
2967 set_xfrm_gro_udp_encap_rcv(val, sk->sk_family, sk); in udp_lib_setsockopt()
2969 if (sk->sk_family == AF_INET6) in udp_lib_setsockopt()
2970 WRITE_ONCE(up->encap_rcv, in udp_lib_setsockopt()
2971 ipv6_stub->xfrm6_udp_encap_rcv); in udp_lib_setsockopt()
2974 WRITE_ONCE(up->encap_rcv, in udp_lib_setsockopt()
2979 WRITE_ONCE(up->encap_type, val); in udp_lib_setsockopt()
2983 err = -ENOPROTOOPT; in udp_lib_setsockopt()
2998 return -EINVAL; in udp_lib_setsockopt()
2999 WRITE_ONCE(up->gso_size, val); in udp_lib_setsockopt()
3004 /* when enabling GRO, accept the related GSO packet type */ in udp_lib_setsockopt()
3009 set_xfrm_gro_udp_encap_rcv(up->encap_type, sk->sk_family, sk); in udp_lib_setsockopt()
3013 * UDP-Lite's partial checksum coverage (RFC 3828). in udp_lib_setsockopt()
3019 return -ENOPROTOOPT; in udp_lib_setsockopt()
3024 WRITE_ONCE(up->pcslen, val); in udp_lib_setsockopt()
3033 return -ENOPROTOOPT; in udp_lib_setsockopt()
3038 WRITE_ONCE(up->pcrlen, val); in udp_lib_setsockopt()
3043 err = -ENOPROTOOPT; in udp_lib_setsockopt()
3068 return -EFAULT; in udp_lib_getsockopt()
3071 return -EINVAL; in udp_lib_getsockopt()
3081 val = READ_ONCE(up->encap_type); in udp_lib_getsockopt()
3093 val = READ_ONCE(up->gso_size); in udp_lib_getsockopt()
3103 val = READ_ONCE(up->pcslen); in udp_lib_getsockopt()
3107 val = READ_ONCE(up->pcrlen); in udp_lib_getsockopt()
3111 return -ENOPROTOOPT; in udp_lib_getsockopt()
3115 return -EFAULT; in udp_lib_getsockopt()
3117 return -EFAULT; in udp_lib_getsockopt()
3131 * udp_poll - wait for a UDP event.
3132 * @file: - file struct
3133 * @sock: - socket
3134 * @wait: - poll table
3146 struct sock *sk = sock->sk; in udp_poll()
3148 if (!skb_queue_empty_lockless(&udp_sk(sk)->reader_queue)) in udp_poll()
3152 if ((mask & EPOLLRDNORM) && !(file->f_flags & O_NONBLOCK) && in udp_poll()
3153 !(sk->sk_shutdown & RCV_SHUTDOWN) && first_packet_length(sk) == -1) in udp_poll()
3175 sk->sk_err = err; in udp_abort()
3223 /* ------------------------------------------------------------------------ */
3232 return ((family == AF_UNSPEC || family == sk->sk_family) && in seq_sk_match()
3245 if (seq->op == &bpf_iter_udp_seq_ops) in udp_get_table_seq()
3246 return net->ipv4.udp_table; in udp_get_table_seq()
3249 afinfo = pde_data(file_inode(seq->file)); in udp_get_table_seq()
3250 return afinfo->udp_table ? : net->ipv4.udp_table; in udp_get_table_seq()
3255 struct udp_iter_state *state = seq->private; in udp_get_first()
3262 for (state->bucket = start; state->bucket <= udptable->mask; in udp_get_first()
3263 ++state->bucket) { in udp_get_first()
3264 struct udp_hslot *hslot = &udptable->hash[state->bucket]; in udp_get_first()
3266 if (hlist_empty(&hslot->head)) in udp_get_first()
3269 spin_lock_bh(&hslot->lock); in udp_get_first()
3270 sk_for_each(sk, &hslot->head) { in udp_get_first()
3274 spin_unlock_bh(&hslot->lock); in udp_get_first()
3283 struct udp_iter_state *state = seq->private; in udp_get_next()
3294 if (state->bucket <= udptable->mask) in udp_get_next()
3295 spin_unlock_bh(&udptable->hash[state->bucket].lock); in udp_get_next()
3297 return udp_get_first(seq, state->bucket + 1); in udp_get_next()
3308 --pos; in udp_get_idx()
3314 struct udp_iter_state *state = seq->private; in udp_seq_start()
3315 state->bucket = MAX_UDP_PORTS; in udp_seq_start()
3317 return *pos ? udp_get_idx(seq, *pos-1) : SEQ_START_TOKEN; in udp_seq_start()
3337 struct udp_iter_state *state = seq->private; in udp_seq_stop()
3342 if (state->bucket <= udptable->mask) in udp_seq_stop()
3343 spin_unlock_bh(&udptable->hash[state->bucket].lock); in udp_seq_stop()
3347 /* ------------------------------------------------------------------------ */
3352 __be32 dest = inet->inet_daddr; in udp4_format_sock()
3353 __be32 src = inet->inet_rcv_saddr; in udp4_format_sock()
3354 __u16 destp = ntohs(inet->inet_dport); in udp4_format_sock()
3355 __u16 srcp = ntohs(inet->inet_sport); in udp4_format_sock()
3359 bucket, src, srcp, dest, destp, sp->sk_state, in udp4_format_sock()
3365 refcount_read(&sp->sk_refcnt), sp, in udp4_format_sock()
3366 atomic_read(&sp->sk_drops)); in udp4_format_sock()
3374 "rx_queue tr tm->when retrnsmt uid timeout " in udp4_seq_show()
3377 struct udp_iter_state *state = seq->private; in udp4_seq_show()
3379 udp4_format_sock(v, seq, state->bucket); in udp4_seq_show()
3407 struct bpf_udp_iter_state *iter = seq->private; in bpf_iter_udp_batch()
3408 struct udp_iter_state *state = &iter->state; in bpf_iter_udp_batch()
3416 resume_bucket = state->bucket; in bpf_iter_udp_batch()
3417 resume_offset = iter->offset; in bpf_iter_udp_batch()
3420 if (iter->st_bucket_done) in bpf_iter_udp_batch()
3421 state->bucket++; in bpf_iter_udp_batch()
3433 iter->cur_sk = 0; in bpf_iter_udp_batch()
3434 iter->end_sk = 0; in bpf_iter_udp_batch()
3435 iter->st_bucket_done = false; in bpf_iter_udp_batch()
3438 for (; state->bucket <= udptable->mask; state->bucket++) { in bpf_iter_udp_batch()
3439 struct udp_hslot *hslot2 = &udptable->hash2[state->bucket].hslot; in bpf_iter_udp_batch()
3441 if (hlist_empty(&hslot2->head)) in bpf_iter_udp_batch()
3444 iter->offset = 0; in bpf_iter_udp_batch()
3445 spin_lock_bh(&hslot2->lock); in bpf_iter_udp_batch()
3446 udp_portaddr_for_each_entry(sk, &hslot2->head) { in bpf_iter_udp_batch()
3451 if (state->bucket == resume_bucket && in bpf_iter_udp_batch()
3452 iter->offset < resume_offset) { in bpf_iter_udp_batch()
3453 ++iter->offset; in bpf_iter_udp_batch()
3456 if (iter->end_sk < iter->max_sk) { in bpf_iter_udp_batch()
3458 iter->batch[iter->end_sk++] = sk; in bpf_iter_udp_batch()
3463 spin_unlock_bh(&hslot2->lock); in bpf_iter_udp_batch()
3465 if (iter->end_sk) in bpf_iter_udp_batch()
3470 if (!iter->end_sk) in bpf_iter_udp_batch()
3473 if (iter->end_sk == batch_sks) { in bpf_iter_udp_batch()
3477 iter->st_bucket_done = true; in bpf_iter_udp_batch()
3488 return iter->batch[0]; in bpf_iter_udp_batch()
3493 struct bpf_udp_iter_state *iter = seq->private; in bpf_iter_udp_seq_next()
3496 /* Whenever seq_next() is called, the iter->cur_sk is in bpf_iter_udp_seq_next()
3497 * done with seq_show(), so unref the iter->cur_sk. in bpf_iter_udp_seq_next()
3499 if (iter->cur_sk < iter->end_sk) { in bpf_iter_udp_seq_next()
3500 sock_put(iter->batch[iter->cur_sk++]); in bpf_iter_udp_seq_next()
3501 ++iter->offset; in bpf_iter_udp_seq_next()
3504 /* After updating iter->cur_sk, check if there are more sockets in bpf_iter_udp_seq_next()
3507 if (iter->cur_sk < iter->end_sk) in bpf_iter_udp_seq_next()
3508 sk = iter->batch[iter->cur_sk]; in bpf_iter_udp_seq_next()
3520 * continue from where it was stop()-ped. in bpf_iter_udp_seq_start()
3533 meta->seq_num--; /* skip SEQ_START_TOKEN */ in udp_prog_seq_show()
3543 struct udp_iter_state *state = seq->private; in bpf_iter_udp_seq_show()
3563 ret = udp_prog_seq_show(prog, &meta, v, uid, state->bucket); in bpf_iter_udp_seq_show()
3572 while (iter->cur_sk < iter->end_sk) in bpf_iter_udp_put_batch()
3573 sock_put(iter->batch[iter->cur_sk++]); in bpf_iter_udp_put_batch()
3578 struct bpf_udp_iter_state *iter = seq->private; in bpf_iter_udp_seq_stop()
3589 if (iter->cur_sk < iter->end_sk) { in bpf_iter_udp_seq_stop()
3591 iter->st_bucket_done = false; in bpf_iter_udp_seq_stop()
3609 if (seq->op == &bpf_iter_udp_seq_ops) in seq_file_family()
3614 afinfo = pde_data(file_inode(seq->file)); in seq_file_family()
3615 return afinfo->family; in seq_file_family()
3633 if (!proc_create_net_data("udp", 0444, net->proc_net, &udp_seq_ops, in udp4_proc_init_net()
3635 return -ENOMEM; in udp4_proc_init_net()
3641 remove_proc_entry("udp", net->proc_net); in udp4_proc_exit_net()
3684 table->hash = alloc_large_system_hash(name, in udp_table_init()
3689 &table->log, in udp_table_init()
3690 &table->mask, in udp_table_init()
3694 table->hash2 = (void *)(table->hash + (table->mask + 1)); in udp_table_init()
3695 for (i = 0; i <= table->mask; i++) { in udp_table_init()
3696 INIT_HLIST_HEAD(&table->hash[i].head); in udp_table_init()
3697 table->hash[i].count = 0; in udp_table_init()
3698 spin_lock_init(&table->hash[i].lock); in udp_table_init()
3700 for (i = 0; i <= table->mask; i++) { in udp_table_init()
3701 INIT_HLIST_HEAD(&table->hash2[i].hslot.head); in udp_table_init()
3702 table->hash2[i].hslot.count = 0; in udp_table_init()
3703 spin_lock_init(&table->hash2[i].hslot.lock); in udp_table_init()
3720 net->ipv4.sysctl_udp_rmem_min = PAGE_SIZE; in udp_sysctl_init()
3721 net->ipv4.sysctl_udp_wmem_min = PAGE_SIZE; in udp_sysctl_init()
3724 net->ipv4.sysctl_udp_l3mdev_accept = 0; in udp_sysctl_init()
3740 udptable->hash = vmalloc_huge(hash_entries * slot_size, in udp_pernet_table_alloc()
3742 if (!udptable->hash) in udp_pernet_table_alloc()
3745 udptable->hash2 = (void *)(udptable->hash + hash_entries); in udp_pernet_table_alloc()
3746 udptable->mask = hash_entries - 1; in udp_pernet_table_alloc()
3747 udptable->log = ilog2(hash_entries); in udp_pernet_table_alloc()
3750 INIT_HLIST_HEAD(&udptable->hash[i].head); in udp_pernet_table_alloc()
3751 udptable->hash[i].count = 0; in udp_pernet_table_alloc()
3752 spin_lock_init(&udptable->hash[i].lock); in udp_pernet_table_alloc()
3754 INIT_HLIST_HEAD(&udptable->hash2[i].hslot.head); in udp_pernet_table_alloc()
3755 udptable->hash2[i].hslot.count = 0; in udp_pernet_table_alloc()
3756 spin_lock_init(&udptable->hash2[i].hslot.lock); in udp_pernet_table_alloc()
3770 struct udp_table *udptable = net->ipv4.udp_table; in udp_pernet_table_free()
3775 kvfree(udptable->hash); in udp_pernet_table_free()
3788 old_net = current->nsproxy->net_ns; in udp_set_table()
3789 hash_entries = READ_ONCE(old_net->ipv4.sysctl_udp_child_hash_entries); in udp_set_table()
3801 net->ipv4.udp_table = udptable; in udp_set_table()
3807 net->ipv4.udp_table = &udp_table; in udp_set_table()
3841 return -ENOMEM; in DEFINE_BPF_ITER_FUNC()
3844 kvfree(iter->batch); in DEFINE_BPF_ITER_FUNC()
3845 iter->batch = new_batch; in DEFINE_BPF_ITER_FUNC()
3846 iter->max_sk = new_batch_sz; in DEFINE_BPF_ITER_FUNC()
3874 kvfree(iter->batch); in bpf_iter_fini_udp()