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()
329 sk_add_node_rcu(sk, &hslot->head); in udp_lib_get_port()
330 hslot->count++; in udp_lib_get_port()
331 sock_prot_inuse_add(sock_net(sk), sk->sk_prot, 1); in udp_lib_get_port()
333 hslot2 = udp_hashslot2(udptable, udp_sk(sk)->udp_portaddr_hash); in udp_lib_get_port()
334 spin_lock(&hslot2->lock); in udp_lib_get_port()
335 if (IS_ENABLED(CONFIG_IPV6) && sk->sk_reuseport && in udp_lib_get_port()
336 sk->sk_family == AF_INET6) in udp_lib_get_port()
337 hlist_add_tail_rcu(&udp_sk(sk)->udp_portaddr_node, in udp_lib_get_port()
338 &hslot2->head); in udp_lib_get_port()
340 hlist_add_head_rcu(&udp_sk(sk)->udp_portaddr_node, in udp_lib_get_port()
341 &hslot2->head); in udp_lib_get_port()
342 hslot2->count++; in udp_lib_get_port()
343 spin_unlock(&hslot2->lock); in udp_lib_get_port()
348 spin_unlock_bh(&hslot->lock); in udp_lib_get_port()
359 ipv4_portaddr_hash(sock_net(sk), inet_sk(sk)->inet_rcv_saddr, 0); in udp_v4_get_port()
362 udp_sk(sk)->udp_portaddr_hash = hash2_partial; in udp_v4_get_port()
376 udp_sk(sk)->udp_port_hash != hnum || in compute_score()
378 return -1; in compute_score()
380 if (sk->sk_rcv_saddr != daddr) in compute_score()
381 return -1; in compute_score()
383 score = (sk->sk_family == PF_INET) ? 2 : 1; in compute_score()
386 if (inet->inet_daddr) { in compute_score()
387 if (inet->inet_daddr != saddr) in compute_score()
388 return -1; in compute_score()
392 if (inet->inet_dport) { in compute_score()
393 if (inet->inet_dport != sport) in compute_score()
394 return -1; in compute_score()
398 dev_match = udp_sk_bound_dev_eq(net, sk->sk_bound_dev_if, in compute_score()
401 return -1; in compute_score()
402 if (sk->sk_bound_dev_if) in compute_score()
405 if (READ_ONCE(sk->sk_incoming_cpu) == raw_smp_processor_id()) in compute_score()
435 udp_portaddr_for_each_entry_rcu(sk, &hslot2->head) { in udp4_lib_lookup2()
441 if (sk->sk_state == TCP_ESTABLISHED) { in udp4_lib_lookup2()
470 * harder than this. -DaveM
482 slot2 = hash2 & udptable->mask; in __udp4_lib_lookup()
483 hslot2 = &udptable->hash2[slot2]; in __udp4_lib_lookup()
485 /* Lookup connected or non-wildcard socket */ in __udp4_lib_lookup()
489 if (!IS_ERR_OR_NULL(result) && result->sk_state == TCP_ESTABLISHED) in __udp4_lib_lookup()
494 udptable == net->ipv4.udp_table) { in __udp4_lib_lookup()
504 /* Got non-wildcard socket or error on first lookup */ in __udp4_lib_lookup()
510 slot2 = hash2 & udptable->mask; in __udp4_lib_lookup()
511 hslot2 = &udptable->hash2[slot2]; in __udp4_lib_lookup()
529 return __udp4_lib_lookup(dev_net(skb->dev), iph->saddr, sport, in __udp4_lib_lookup_skb()
530 iph->daddr, dport, inet_iif(skb), in __udp4_lib_lookup_skb()
538 struct net *net = dev_net(skb->dev); in udp4_lib_lookup_skb()
543 return __udp4_lib_lookup(net, iph->saddr, sport, in udp4_lib_lookup_skb()
544 iph->daddr, dport, iif, in udp4_lib_lookup_skb()
545 sdif, net->ipv4.udp_table, NULL); in udp4_lib_lookup_skb()
558 dif, 0, net->ipv4.udp_table, NULL); in udp4_lib_lookup()
559 if (sk && !refcount_inc_not_zero(&sk->sk_refcnt)) in udp4_lib_lookup()
574 udp_sk(sk)->udp_port_hash != hnum || in __udp_is_mcast_sock()
575 (inet->inet_daddr && inet->inet_daddr != rmt_addr) || in __udp_is_mcast_sock()
576 (inet->inet_dport != rmt_port && inet->inet_dport) || in __udp_is_mcast_sock()
577 (inet->inet_rcv_saddr && inet->inet_rcv_saddr != loc_addr) || in __udp_is_mcast_sock()
579 !udp_sk_bound_dev_eq(net, sk->sk_bound_dev_if, dif, sdif)) in __udp_is_mcast_sock()
613 handler = encap->err_handler; in __udp4_lib_err_encap_no_sk()
618 return -ENOENT; in __udp4_lib_err_encap_no_sk()
656 skb_set_transport_header(skb, iph->ihl << 2); in __udp4_lib_err_encap()
661 lookup = READ_ONCE(up->encap_err_lookup); in __udp4_lib_err_encap()
668 sk = __udp4_lib_lookup(net, iph->daddr, uh->source, in __udp4_lib_err_encap()
669 iph->saddr, uh->dest, skb->dev->ifindex, 0, in __udp4_lib_err_encap()
674 lookup = READ_ONCE(up->encap_err_lookup); in __udp4_lib_err_encap()
703 const struct iphdr *iph = (const struct iphdr *)skb->data; in __udp4_lib_err()
704 struct udphdr *uh = (struct udphdr *)(skb->data+(iph->ihl<<2)); in __udp4_lib_err()
705 const int type = icmp_hdr(skb)->type; in __udp4_lib_err()
706 const int code = icmp_hdr(skb)->code; in __udp4_lib_err()
711 struct net *net = dev_net(skb->dev); in __udp4_lib_err()
713 sk = __udp4_lib_lookup(net, iph->daddr, uh->dest, in __udp4_lib_err()
714 iph->saddr, uh->source, skb->dev->ifindex, in __udp4_lib_err()
717 if (!sk || READ_ONCE(udp_sk(sk)->encap_type)) { in __udp4_lib_err()
725 sk = ERR_PTR(-ENOENT); in __udp4_lib_err()
753 if (READ_ONCE(inet->pmtudisc) != IP_PMTUDISC_DONT) { in __udp4_lib_err()
777 if (udp_sk(sk)->encap_err_rcv) in __udp4_lib_err()
778 udp_sk(sk)->encap_err_rcv(sk, skb, err, uh->dest, info, in __udp4_lib_err()
783 if (!harderr || sk->sk_state != TCP_ESTABLISHED) in __udp4_lib_err()
786 ip_icmp_error(sk, skb, err, uh->dest, info, (u8 *)(uh+1)); in __udp4_lib_err()
788 sk->sk_err = err; in __udp4_lib_err()
796 return __udp4_lib_err(skb, info, dev_net(skb->dev)->ipv4.udp_table); in udp_err()
806 if (up->pending) { in udp_flush_pending_frames()
807 up->len = 0; in udp_flush_pending_frames()
808 WRITE_ONCE(up->pending, 0); in udp_flush_pending_frames()
815 * udp4_hwcsum - handle outgoing HW checksumming
816 * @skb: sk_buff containing the filled-in UDP header
825 int len = skb->len - offset; in udp4_hwcsum()
833 skb->csum_start = skb_transport_header(skb) - skb->head; in udp4_hwcsum()
834 skb->csum_offset = offsetof(struct udphdr, check); in udp4_hwcsum()
835 uh->check = ~csum_tcpudp_magic(src, dst, len, in udp4_hwcsum()
841 * HW-checksum won't work as there are two or more in udp4_hwcsum()
846 csum = csum_add(csum, frags->csum); in udp4_hwcsum()
847 hlen -= frags->len; in udp4_hwcsum()
851 skb->ip_summed = CHECKSUM_NONE; in udp4_hwcsum()
853 uh->check = csum_tcpudp_magic(src, dst, len, IPPROTO_UDP, csum); in udp4_hwcsum()
854 if (uh->check == 0) in udp4_hwcsum()
855 uh->check = CSUM_MANGLED_0; in udp4_hwcsum()
869 uh->check = 0; in udp_set_csum()
871 uh->check = ~udp_v4_check(len, saddr, daddr, 0); in udp_set_csum()
872 } else if (skb->ip_summed == CHECKSUM_PARTIAL) { in udp_set_csum()
873 uh->check = 0; in udp_set_csum()
874 uh->check = udp_v4_check(len, saddr, daddr, lco_csum(skb)); in udp_set_csum()
875 if (uh->check == 0) in udp_set_csum()
876 uh->check = CSUM_MANGLED_0; in udp_set_csum()
878 skb->ip_summed = CHECKSUM_PARTIAL; in udp_set_csum()
879 skb->csum_start = skb_transport_header(skb) - skb->head; in udp_set_csum()
880 skb->csum_offset = offsetof(struct udphdr, check); in udp_set_csum()
881 uh->check = ~udp_v4_check(len, saddr, daddr, 0); in udp_set_csum()
889 struct sock *sk = skb->sk; in udp_send_skb()
895 int len = skb->len - offset; in udp_send_skb()
896 int datalen = len - sizeof(*uh); in udp_send_skb()
903 uh->source = inet->inet_sport; in udp_send_skb()
904 uh->dest = fl4->fl4_dport; in udp_send_skb()
905 uh->len = htons(len); in udp_send_skb()
906 uh->check = 0; in udp_send_skb()
908 if (cork->gso_size) { in udp_send_skb()
912 if (hlen + cork->gso_size > cork->fragsize) { in udp_send_skb()
914 return -EINVAL; in udp_send_skb()
916 if (datalen > cork->gso_size * UDP_MAX_SEGMENTS) { in udp_send_skb()
918 return -EINVAL; in udp_send_skb()
920 if (sk->sk_no_check_tx) { in udp_send_skb()
922 return -EINVAL; in udp_send_skb()
924 if (skb->ip_summed != CHECKSUM_PARTIAL || is_udplite || in udp_send_skb()
927 return -EIO; in udp_send_skb()
930 if (datalen > cork->gso_size) { in udp_send_skb()
931 skb_shinfo(skb)->gso_size = cork->gso_size; in udp_send_skb()
932 skb_shinfo(skb)->gso_type = SKB_GSO_UDP_L4; in udp_send_skb()
933 skb_shinfo(skb)->gso_segs = DIV_ROUND_UP(datalen, in udp_send_skb()
934 cork->gso_size); in udp_send_skb()
939 if (is_udplite) /* UDP-Lite */ in udp_send_skb()
942 else if (sk->sk_no_check_tx) { /* UDP csum off */ in udp_send_skb()
944 skb->ip_summed = CHECKSUM_NONE; in udp_send_skb()
947 } else if (skb->ip_summed == CHECKSUM_PARTIAL) { /* UDP hardware csum */ in udp_send_skb()
950 udp4_hwcsum(skb, fl4->saddr, fl4->daddr); in udp_send_skb()
956 /* add protocol-dependent pseudo-header */ in udp_send_skb()
957 uh->check = csum_tcpudp_magic(fl4->saddr, fl4->daddr, len, in udp_send_skb()
958 sk->sk_protocol, csum); in udp_send_skb()
959 if (uh->check == 0) in udp_send_skb()
960 uh->check = CSUM_MANGLED_0; in udp_send_skb()
965 if (err == -ENOBUFS && in udp_send_skb()
984 struct flowi4 *fl4 = &inet->cork.fl.u.ip4; in udp_push_pending_frames()
992 err = udp_send_skb(skb, fl4, &inet->cork.base); in udp_push_pending_frames()
995 up->len = 0; in udp_push_pending_frames()
996 WRITE_ONCE(up->pending, 0); in udp_push_pending_frames()
1003 switch (cmsg->cmsg_type) { in __udp_cmsg_send()
1005 if (cmsg->cmsg_len != CMSG_LEN(sizeof(__u16))) in __udp_cmsg_send()
1006 return -EINVAL; in __udp_cmsg_send()
1010 return -EINVAL; in __udp_cmsg_send()
1022 return -EINVAL; in udp_cmsg_send()
1024 if (cmsg->cmsg_level != SOL_UDP) { in udp_cmsg_send()
1042 DECLARE_SOCKADDR(struct sockaddr_in *, usin, msg->msg_name); in udp_sendmsg()
1054 int corkreq = udp_test_bit(CORK, sk) || msg->msg_flags & MSG_MORE; in udp_sendmsg()
1061 return -EMSGSIZE; in udp_sendmsg()
1067 if (msg->msg_flags & MSG_OOB) /* Mirror BSD error message compatibility */ in udp_sendmsg()
1068 return -EOPNOTSUPP; in udp_sendmsg()
1072 fl4 = &inet->cork.fl.u.ip4; in udp_sendmsg()
1073 if (READ_ONCE(up->pending)) { in udp_sendmsg()
1079 if (likely(up->pending)) { in udp_sendmsg()
1080 if (unlikely(up->pending != AF_INET)) { in udp_sendmsg()
1082 return -EINVAL; in udp_sendmsg()
1094 if (msg->msg_namelen < sizeof(*usin)) in udp_sendmsg()
1095 return -EINVAL; in udp_sendmsg()
1096 if (usin->sin_family != AF_INET) { in udp_sendmsg()
1097 if (usin->sin_family != AF_UNSPEC) in udp_sendmsg()
1098 return -EAFNOSUPPORT; in udp_sendmsg()
1101 daddr = usin->sin_addr.s_addr; in udp_sendmsg()
1102 dport = usin->sin_port; in udp_sendmsg()
1104 return -EINVAL; in udp_sendmsg()
1106 if (sk->sk_state != TCP_ESTABLISHED) in udp_sendmsg()
1107 return -EDESTADDRREQ; in udp_sendmsg()
1108 daddr = inet->inet_daddr; in udp_sendmsg()
1109 dport = inet->inet_dport; in udp_sendmsg()
1117 ipc.gso_size = READ_ONCE(up->gso_size); in udp_sendmsg()
1119 if (msg->msg_controllen) { in udp_sendmsg()
1123 sk->sk_family == AF_INET6); in udp_sendmsg()
1136 inet_opt = rcu_dereference(inet->inet_opt); in udp_sendmsg()
1139 sizeof(*inet_opt) + inet_opt->opt.optlen); in udp_sendmsg()
1148 &msg->msg_namelen, in udp_sendmsg()
1153 if (usin->sin_port == 0) { in udp_sendmsg()
1155 err = -EINVAL; in udp_sendmsg()
1158 daddr = usin->sin_addr.s_addr; in udp_sendmsg()
1159 dport = usin->sin_port; in udp_sendmsg()
1166 if (ipc.opt && ipc.opt->opt.srr) { in udp_sendmsg()
1168 err = -EINVAL; in udp_sendmsg()
1171 faddr = ipc.opt->opt.faddr; in udp_sendmsg()
1179 uc_index = READ_ONCE(inet->uc_index); in udp_sendmsg()
1182 ipc.oif = READ_ONCE(inet->mc_index); in udp_sendmsg()
1184 saddr = READ_ONCE(inet->mc_addr); in udp_sendmsg()
1212 sk->sk_protocol, flow_flags, faddr, saddr, in udp_sendmsg()
1213 dport, inet->inet_sport, sk->sk_uid); in udp_sendmsg()
1220 if (err == -ENETUNREACH) in udp_sendmsg()
1225 err = -EACCES; in udp_sendmsg()
1226 if ((rt->rt_flags & RTCF_BROADCAST) && in udp_sendmsg()
1230 sk_dst_set(sk, dst_clone(&rt->dst)); in udp_sendmsg()
1233 if (msg->msg_flags&MSG_CONFIRM) in udp_sendmsg()
1237 saddr = fl4->saddr; in udp_sendmsg()
1239 daddr = ipc.addr = fl4->daddr; in udp_sendmsg()
1241 /* Lockless fast path for the non-corking case. */ in udp_sendmsg()
1247 &cork, msg->msg_flags); in udp_sendmsg()
1255 if (unlikely(up->pending)) { in udp_sendmsg()
1257 /* ... which is an evident application bug. --ANK */ in udp_sendmsg()
1261 err = -EINVAL; in udp_sendmsg()
1267 fl4 = &inet->cork.fl.u.ip4; in udp_sendmsg()
1268 fl4->daddr = daddr; in udp_sendmsg()
1269 fl4->saddr = saddr; in udp_sendmsg()
1270 fl4->fl4_dport = dport; in udp_sendmsg()
1271 fl4->fl4_sport = inet->inet_sport; in udp_sendmsg()
1272 WRITE_ONCE(up->pending, AF_INET); in udp_sendmsg()
1275 up->len += ulen; in udp_sendmsg()
1278 corkreq ? msg->msg_flags|MSG_MORE : msg->msg_flags); in udp_sendmsg()
1283 else if (unlikely(skb_queue_empty(&sk->sk_write_queue))) in udp_sendmsg()
1284 WRITE_ONCE(up->pending, 0); in udp_sendmsg()
1301 if (err == -ENOBUFS || test_bit(SOCK_NOSPACE, &sk->sk_socket->flags)) { in udp_sendmsg()
1308 if (msg->msg_flags & MSG_PROBE) in udp_sendmsg()
1309 dst_confirm_neigh(&rt->dst, &fl4->daddr); in udp_sendmsg()
1310 if (!(msg->msg_flags&MSG_PROBE) || len) in udp_sendmsg()
1319 struct sock *sk = sock->sk; in udp_splice_eof()
1322 if (!READ_ONCE(up->pending) || udp_test_bit(CORK, sk)) in udp_splice_eof()
1326 if (up->pending && !udp_test_bit(CORK, sk)) in udp_splice_eof()
1334 /* all head states (dst, sk, nf conntrack) except skb extensions are
1360 scratch->_tsize_state = skb->truesize; in udp_set_dev_scratch()
1362 scratch->len = skb->len; in udp_set_dev_scratch()
1363 scratch->csum_unnecessary = !!skb_csum_unnecessary(skb); in udp_set_dev_scratch()
1364 scratch->is_linear = !skb_is_nonlinear(skb); in udp_set_dev_scratch()
1367 scratch->_tsize_state |= UDP_SKB_IS_STATELESS; in udp_set_dev_scratch()
1374 * set skb->csum_valid to 1. in udp_skb_csum_unnecessary_set()
1380 udp_skb_scratch(skb)->csum_unnecessary = true; in udp_skb_csum_unnecessary_set()
1386 return udp_skb_scratch(skb)->_tsize_state & ~UDP_SKB_IS_STATELESS; in udp_skb_truesize()
1391 return !(udp_skb_scratch(skb)->_tsize_state & UDP_SKB_IS_STATELESS); in udp_skb_has_head_state()
1403 up->forward_deficit += size; in udp_rmem_release()
1404 size = up->forward_deficit; in udp_rmem_release()
1405 if (size < READ_ONCE(up->forward_threshold) && in udp_rmem_release()
1406 !skb_queue_empty(&up->reader_queue)) in udp_rmem_release()
1409 size += up->forward_deficit; in udp_rmem_release()
1411 up->forward_deficit = 0; in udp_rmem_release()
1416 sk_queue = &sk->sk_receive_queue; in udp_rmem_release()
1418 spin_lock(&sk_queue->lock); in udp_rmem_release()
1422 amt = (sk->sk_forward_alloc - partial) & ~(PAGE_SIZE - 1); in udp_rmem_release()
1423 sk_forward_alloc_add(sk, -amt); in udp_rmem_release()
1428 atomic_sub(size, &sk->sk_rmem_alloc); in udp_rmem_release()
1431 skb_queue_splice_tail_init(sk_queue, &up->reader_queue); in udp_rmem_release()
1434 spin_unlock(&sk_queue->lock); in udp_rmem_release()
1438 * Instead of using skb->truesize here, find a copy of it in skb->dev_scratch
1444 prefetch(&skb->data); in udp_skb_destructor()
1452 prefetch(&skb->data); in udp_skb_dtor_locked()
1485 delta = size - sk->sk_forward_alloc; in udp_rmem_schedule()
1487 return -ENOBUFS; in udp_rmem_schedule()
1494 struct sk_buff_head *list = &sk->sk_receive_queue; in __udp_enqueue_schedule_skb()
1495 int rmem, err = -ENOMEM; in __udp_enqueue_schedule_skb()
1502 rmem = atomic_read(&sk->sk_rmem_alloc); in __udp_enqueue_schedule_skb()
1503 if (rmem > sk->sk_rcvbuf) in __udp_enqueue_schedule_skb()
1508 * - Reduce memory overhead and thus increase receive queue capacity in __udp_enqueue_schedule_skb()
1509 * - Less cache line misses at copyout() time in __udp_enqueue_schedule_skb()
1510 * - Less work at consume_skb() (less alien page frag freeing) in __udp_enqueue_schedule_skb()
1512 if (rmem > (sk->sk_rcvbuf >> 1)) { in __udp_enqueue_schedule_skb()
1517 size = skb->truesize; in __udp_enqueue_schedule_skb()
1523 rmem = atomic_add_return(size, &sk->sk_rmem_alloc); in __udp_enqueue_schedule_skb()
1524 if (rmem > (size + (unsigned int)sk->sk_rcvbuf)) in __udp_enqueue_schedule_skb()
1527 spin_lock(&list->lock); in __udp_enqueue_schedule_skb()
1530 spin_unlock(&list->lock); in __udp_enqueue_schedule_skb()
1534 sk_forward_alloc_add(sk, -size); in __udp_enqueue_schedule_skb()
1542 spin_unlock(&list->lock); in __udp_enqueue_schedule_skb()
1545 INDIRECT_CALL_1(sk->sk_data_ready, sock_def_readable, sk); in __udp_enqueue_schedule_skb()
1551 atomic_sub(skb->truesize, &sk->sk_rmem_alloc); in __udp_enqueue_schedule_skb()
1554 atomic_inc(&sk->sk_drops); in __udp_enqueue_schedule_skb()
1567 skb_queue_splice_tail_init(&sk->sk_receive_queue, &up->reader_queue); in udp_destruct_common()
1568 while ((skb = __skb_dequeue(&up->reader_queue)) != NULL) { in udp_destruct_common()
1569 total += skb->truesize; in udp_destruct_common()
1585 sk->sk_destruct = udp_destruct_sock; in udp_init_sock()
1586 set_bit(SOCK_SUPPORT_ZC, &sk->sk_socket->flags); in udp_init_sock()
1618 atomic_inc(&sk->sk_drops); in __first_packet_length()
1620 *total += skb->truesize; in __first_packet_length()
1631 * first_packet_length - return length of first packet in receive queue
1635 * Returns the length of found skb, or -1 if none is found.
1639 struct sk_buff_head *rcvq = &udp_sk(sk)->reader_queue; in first_packet_length()
1640 struct sk_buff_head *sk_queue = &sk->sk_receive_queue; in first_packet_length()
1645 spin_lock_bh(&rcvq->lock); in first_packet_length()
1648 spin_lock(&sk_queue->lock); in first_packet_length()
1650 spin_unlock(&sk_queue->lock); in first_packet_length()
1654 res = skb ? skb->len : -1; in first_packet_length()
1657 spin_unlock_bh(&rcvq->lock); in first_packet_length()
1681 return -ENOIOCTLCMD; in udp_ioctl()
1691 struct sk_buff_head *sk_queue = &sk->sk_receive_queue; in __skb_recv_udp()
1697 queue = &udp_sk(sk)->reader_queue; in __skb_recv_udp()
1706 error = -EAGAIN; in __skb_recv_udp()
1708 spin_lock_bh(&queue->lock); in __skb_recv_udp()
1714 spin_unlock_bh(&queue->lock); in __skb_recv_udp()
1719 spin_unlock_bh(&queue->lock); in __skb_recv_udp()
1724 * keep both queues locked to avoid re-acquiring in __skb_recv_udp()
1728 spin_lock(&sk_queue->lock); in __skb_recv_udp()
1735 spin_unlock(&sk_queue->lock); in __skb_recv_udp()
1736 spin_unlock_bh(&queue->lock); in __skb_recv_udp()
1749 !__skb_wait_for_more_packets(sk, &sk->sk_receive_queue, in __skb_recv_udp()
1774 atomic_inc(&sk->sk_drops); in udp_read_skb()
1793 DECLARE_SOCKADDR(struct sockaddr_in *, sin, msg->msg_name); in udp_recvmsg()
1811 if (copied > ulen - off) in udp_recvmsg()
1812 copied = ulen - off; in udp_recvmsg()
1814 msg->msg_flags |= MSG_TRUNC; in udp_recvmsg()
1819 * coverage checksum (UDP-Lite), do it before the copy. in udp_recvmsg()
1823 (is_udplite && UDP_SKB_CB(skb)->partial_cov)) { in udp_recvmsg()
1832 err = copy_linear_skb(skb, copied, off, &msg->msg_iter); in udp_recvmsg()
1838 if (err == -EINVAL) in udp_recvmsg()
1844 atomic_inc(&sk->sk_drops); in udp_recvmsg()
1860 sin->sin_family = AF_INET; in udp_recvmsg()
1861 sin->sin_port = udp_hdr(skb)->source; in udp_recvmsg()
1862 sin->sin_addr.s_addr = ip_hdr(skb)->saddr; in udp_recvmsg()
1863 memset(sin->sin_zero, 0, sizeof(sin->sin_zero)); in udp_recvmsg()
1881 skb_consume_udp(sk, skb, peeking ? -err : err); in udp_recvmsg()
1885 if (!__sk_queue_drop_skb(sk, &udp_sk(sk)->reader_queue, skb, flags, in udp_recvmsg()
1894 msg->msg_flags &= ~MSG_TRUNC; in udp_recvmsg()
1905 return -EINVAL; in udp_pre_connect()
1915 * 1003.1g - break association. in __udp_disconnect()
1918 sk->sk_state = TCP_CLOSE; in __udp_disconnect()
1919 inet->inet_daddr = 0; in __udp_disconnect()
1920 inet->inet_dport = 0; in __udp_disconnect()
1922 sk->sk_bound_dev_if = 0; in __udp_disconnect()
1923 if (!(sk->sk_userlocks & SOCK_BINDADDR_LOCK)) { in __udp_disconnect()
1925 if (sk->sk_prot->rehash && in __udp_disconnect()
1926 (sk->sk_userlocks & SOCK_BINDPORT_LOCK)) in __udp_disconnect()
1927 sk->sk_prot->rehash(sk); in __udp_disconnect()
1930 if (!(sk->sk_userlocks & SOCK_BINDPORT_LOCK)) { in __udp_disconnect()
1931 sk->sk_prot->unhash(sk); in __udp_disconnect()
1932 inet->inet_sport = 0; in __udp_disconnect()
1955 udp_sk(sk)->udp_port_hash); in udp_lib_unhash()
1956 hslot2 = udp_hashslot2(udptable, udp_sk(sk)->udp_portaddr_hash); in udp_lib_unhash()
1958 spin_lock_bh(&hslot->lock); in udp_lib_unhash()
1959 if (rcu_access_pointer(sk->sk_reuseport_cb)) in udp_lib_unhash()
1962 hslot->count--; in udp_lib_unhash()
1963 inet_sk(sk)->inet_num = 0; in udp_lib_unhash()
1964 sock_prot_inuse_add(sock_net(sk), sk->sk_prot, -1); in udp_lib_unhash()
1966 spin_lock(&hslot2->lock); in udp_lib_unhash()
1967 hlist_del_init_rcu(&udp_sk(sk)->udp_portaddr_node); in udp_lib_unhash()
1968 hslot2->count--; in udp_lib_unhash()
1969 spin_unlock(&hslot2->lock); in udp_lib_unhash()
1971 spin_unlock_bh(&hslot->lock); in udp_lib_unhash()
1985 hslot2 = udp_hashslot2(udptable, udp_sk(sk)->udp_portaddr_hash); in udp_lib_rehash()
1987 udp_sk(sk)->udp_portaddr_hash = newhash; in udp_lib_rehash()
1990 rcu_access_pointer(sk->sk_reuseport_cb)) { in udp_lib_rehash()
1992 udp_sk(sk)->udp_port_hash); in udp_lib_rehash()
1994 spin_lock_bh(&hslot->lock); in udp_lib_rehash()
1995 if (rcu_access_pointer(sk->sk_reuseport_cb)) in udp_lib_rehash()
1999 spin_lock(&hslot2->lock); in udp_lib_rehash()
2000 hlist_del_init_rcu(&udp_sk(sk)->udp_portaddr_node); in udp_lib_rehash()
2001 hslot2->count--; in udp_lib_rehash()
2002 spin_unlock(&hslot2->lock); in udp_lib_rehash()
2004 spin_lock(&nhslot2->lock); in udp_lib_rehash()
2005 hlist_add_head_rcu(&udp_sk(sk)->udp_portaddr_node, in udp_lib_rehash()
2006 &nhslot2->head); in udp_lib_rehash()
2007 nhslot2->count++; in udp_lib_rehash()
2008 spin_unlock(&nhslot2->lock); in udp_lib_rehash()
2011 spin_unlock_bh(&hslot->lock); in udp_lib_rehash()
2020 inet_sk(sk)->inet_rcv_saddr, in udp_v4_rehash()
2021 inet_sk(sk)->inet_num); in udp_v4_rehash()
2029 if (inet_sk(sk)->inet_daddr) { in __udp_queue_rcv_skb()
2043 if (rc == -ENOMEM) { in __udp_queue_rcv_skb()
2055 return -1; in __udp_queue_rcv_skb()
2062 * -1: error
2085 READ_ONCE(up->encap_type)) { in udp_queue_rcv_one_skb()
2092 * up->encap_rcv() returns the following value: in udp_queue_rcv_one_skb()
2096 * <0 if skb should be resubmitted as proto -N in udp_queue_rcv_one_skb()
2100 encap_rcv = READ_ONCE(up->encap_rcv); in udp_queue_rcv_one_skb()
2113 return -ret; in udp_queue_rcv_one_skb()
2117 /* FALLTHROUGH -- it's a UDP Packet */ in udp_queue_rcv_one_skb()
2121 * UDP-Lite specific tests, ignored on UDP sockets in udp_queue_rcv_one_skb()
2123 if (udp_test_bit(UDPLITE_RECV_CC, sk) && UDP_SKB_CB(skb)->partial_cov) { in udp_queue_rcv_one_skb()
2124 u16 pcrlen = READ_ONCE(up->pcrlen); in udp_queue_rcv_one_skb()
2139 UDP_SKB_CB(skb)->cscov, skb->len); in udp_queue_rcv_one_skb()
2146 * Therefore the above ...()->partial_cov statement is essential. in udp_queue_rcv_one_skb()
2148 if (UDP_SKB_CB(skb)->cscov < pcrlen) { in udp_queue_rcv_one_skb()
2150 UDP_SKB_CB(skb)->cscov, pcrlen); in udp_queue_rcv_one_skb()
2155 prefetch(&sk->sk_rmem_alloc); in udp_queue_rcv_one_skb()
2156 if (rcu_access_pointer(sk->sk_filter) && in udp_queue_rcv_one_skb()
2175 atomic_inc(&sk->sk_drops); in udp_queue_rcv_one_skb()
2177 return -1; in udp_queue_rcv_one_skb()
2189 __skb_push(skb, -skb_mac_offset(skb)); in udp_queue_rcv_skb()
2197 ip_protocol_deliver_rcu(dev_net(skb->dev), skb, ret); in udp_queue_rcv_skb()
2210 old = xchg((__force struct dst_entry **)&sk->sk_rx_dst, dst); in udp_sk_rx_dst_set()
2230 unsigned short hnum = ntohs(uh->dest); in __udp4_lib_mcast_deliver()
2232 unsigned int hash2 = 0, hash2_any = 0, use_hash2 = (hslot->count > 10); in __udp4_lib_mcast_deliver()
2234 int dif = skb->dev->ifindex; in __udp4_lib_mcast_deliver()
2241 udptable->mask; in __udp4_lib_mcast_deliver()
2242 hash2 = ipv4_portaddr_hash(net, daddr, hnum) & udptable->mask; in __udp4_lib_mcast_deliver()
2244 hslot = &udptable->hash2[hash2]; in __udp4_lib_mcast_deliver()
2248 sk_for_each_entry_offset_rcu(sk, node, &hslot->head, offset) { in __udp4_lib_mcast_deliver()
2249 if (!__udp_is_mcast_sock(net, sk, uh->dest, daddr, in __udp4_lib_mcast_deliver()
2250 uh->source, saddr, dif, sdif, hnum)) in __udp4_lib_mcast_deliver()
2260 atomic_inc(&sk->sk_drops); in __udp4_lib_mcast_deliver()
2291 * including udp header and folding it to skb->csum.
2298 UDP_SKB_CB(skb)->partial_cov = 0; in udp4_csum_init()
2299 UDP_SKB_CB(skb)->cscov = skb->len; in udp4_csum_init()
2306 if (UDP_SKB_CB(skb)->partial_cov) { in udp4_csum_init()
2307 skb->csum = inet_compute_pseudo(skb, proto); in udp4_csum_init()
2315 err = (__force int)skb_checksum_init_zero_check(skb, proto, uh->check, in udp4_csum_init()
2320 if (skb->ip_summed == CHECKSUM_COMPLETE && !skb->csum_valid) { in udp4_csum_init()
2322 if (skb->csum_complete_sw) in udp4_csum_init()
2326 * skb->csum is no longer the full packet checksum, in udp4_csum_init()
2343 if (inet_get_convert_csum(sk) && uh->check && !IS_UDPLITE(sk)) in udp_unicast_rcv_skb()
2349 * it wants the return to be -protocol, or 0 in udp_unicast_rcv_skb()
2352 return -ret; in udp_unicast_rcv_skb()
2368 struct net *net = dev_net(skb->dev); in __udp4_lib_rcv()
2381 ulen = ntohs(uh->len); in __udp4_lib_rcv()
2382 saddr = ip_hdr(skb)->saddr; in __udp4_lib_rcv()
2383 daddr = ip_hdr(skb)->daddr; in __udp4_lib_rcv()
2385 if (ulen > skb->len) in __udp4_lib_rcv()
2398 sk = inet_steal_sock(net, skb, sizeof(struct udphdr), saddr, uh->source, daddr, uh->dest, in __udp4_lib_rcv()
2407 if (unlikely(rcu_dereference(sk->sk_rx_dst) != dst)) in __udp4_lib_rcv()
2416 if (rt->rt_flags & (RTCF_BROADCAST|RTCF_MULTICAST)) in __udp4_lib_rcv()
2420 sk = __udp4_lib_lookup_skb(skb, uh->source, uh->dest, udptable); in __udp4_lib_rcv()
2447 &saddr, ntohs(uh->source), in __udp4_lib_rcv()
2448 ulen, skb->len, in __udp4_lib_rcv()
2449 &daddr, ntohs(uh->dest)); in __udp4_lib_rcv()
2460 &saddr, ntohs(uh->source), &daddr, ntohs(uh->dest), in __udp4_lib_rcv()
2477 struct udp_table *udptable = net->ipv4.udp_table; in __udp4_lib_mcast_demux_lookup()
2483 slot = udp_hashfn(net, hnum, udptable->mask); in __udp4_lib_mcast_demux_lookup()
2484 hslot = &udptable->hash[slot]; in __udp4_lib_mcast_demux_lookup()
2487 if (hslot->count > 10) in __udp4_lib_mcast_demux_lookup()
2491 sk_for_each_rcu(sk, &hslot->head) { in __udp4_lib_mcast_demux_lookup()
2512 struct udp_table *udptable = net->ipv4.udp_table; in __udp4_lib_demux_lookup()
2521 slot2 = hash2 & udptable->mask; in __udp4_lib_demux_lookup()
2522 hslot2 = &udptable->hash2[slot2]; in __udp4_lib_demux_lookup()
2525 udp_portaddr_for_each_entry_rcu(sk, &hslot2->head) { in __udp4_lib_demux_lookup()
2536 struct net *net = dev_net(skb->dev); in udp_v4_early_demux()
2542 int dif = skb->dev->ifindex; in udp_v4_early_demux()
2553 if (skb->pkt_type == PACKET_MULTICAST) { in udp_v4_early_demux()
2554 in_dev = __in_dev_get_rcu(skb->dev); in udp_v4_early_demux()
2559 ours = ip_check_mc_rcu(in_dev, iph->daddr, iph->saddr, in udp_v4_early_demux()
2560 iph->protocol); in udp_v4_early_demux()
2564 sk = __udp4_lib_mcast_demux_lookup(net, uh->dest, iph->daddr, in udp_v4_early_demux()
2565 uh->source, iph->saddr, in udp_v4_early_demux()
2567 } else if (skb->pkt_type == PACKET_HOST) { in udp_v4_early_demux()
2568 sk = __udp4_lib_demux_lookup(net, uh->dest, iph->daddr, in udp_v4_early_demux()
2569 uh->source, iph->saddr, dif, sdif); in udp_v4_early_demux()
2572 if (!sk || !refcount_inc_not_zero(&sk->sk_refcnt)) in udp_v4_early_demux()
2575 skb->sk = sk; in udp_v4_early_demux()
2576 skb->destructor = sock_efree; in udp_v4_early_demux()
2577 dst = rcu_dereference(sk->sk_rx_dst); in udp_v4_early_demux()
2593 if (!inet_sk(sk)->inet_daddr && in_dev) in udp_v4_early_demux()
2594 return ip_mc_validate_source(skb, iph->daddr, in udp_v4_early_demux()
2595 iph->saddr, in udp_v4_early_demux()
2596 iph->tos & IPTOS_RT_MASK, in udp_v4_early_demux()
2597 skb->dev, in_dev, &itag); in udp_v4_early_demux()
2604 return __udp4_lib_rcv(skb, dev_net(skb->dev)->ipv4.udp_table, IPPROTO_UDP); in udp_rcv()
2617 if (up->encap_type) { in udp_destroy_sock()
2619 encap_destroy = READ_ONCE(up->encap_destroy); in udp_destroy_sock()
2634 WRITE_ONCE(udp_sk(sk)->gro_receive, xfrm4_gro_udp_encap_rcv); in set_xfrm_gro_udp_encap_rcv()
2636 WRITE_ONCE(udp_sk(sk)->gro_receive, ipv6_stub->xfrm6_gro_udp_encap_rcv); in set_xfrm_gro_udp_encap_rcv()
2659 WRITE_ONCE(up->forward_threshold, sk->sk_rcvbuf >> 2); in udp_lib_setsockopt()
2666 return -EINVAL; in udp_lib_setsockopt()
2669 return -EFAULT; in udp_lib_setsockopt()
2690 set_xfrm_gro_udp_encap_rcv(val, sk->sk_family, sk); in udp_lib_setsockopt()
2694 if (sk->sk_family == AF_INET6) in udp_lib_setsockopt()
2695 WRITE_ONCE(up->encap_rcv, in udp_lib_setsockopt()
2696 ipv6_stub->xfrm6_udp_encap_rcv); in udp_lib_setsockopt()
2699 WRITE_ONCE(up->encap_rcv, in udp_lib_setsockopt()
2704 WRITE_ONCE(up->encap_type, val); in udp_lib_setsockopt()
2708 err = -ENOPROTOOPT; in udp_lib_setsockopt()
2723 return -EINVAL; in udp_lib_setsockopt()
2724 WRITE_ONCE(up->gso_size, val); in udp_lib_setsockopt()
2729 /* when enabling GRO, accept the related GSO packet type */ in udp_lib_setsockopt()
2734 set_xfrm_gro_udp_encap_rcv(up->encap_type, sk->sk_family, sk); in udp_lib_setsockopt()
2738 * UDP-Lite's partial checksum coverage (RFC 3828). in udp_lib_setsockopt()
2744 return -ENOPROTOOPT; in udp_lib_setsockopt()
2749 WRITE_ONCE(up->pcslen, val); in udp_lib_setsockopt()
2758 return -ENOPROTOOPT; in udp_lib_setsockopt()
2763 WRITE_ONCE(up->pcrlen, val); in udp_lib_setsockopt()
2768 err = -ENOPROTOOPT; in udp_lib_setsockopt()
2793 return -EFAULT; in udp_lib_getsockopt()
2798 return -EINVAL; in udp_lib_getsockopt()
2806 val = READ_ONCE(up->encap_type); in udp_lib_getsockopt()
2818 val = READ_ONCE(up->gso_size); in udp_lib_getsockopt()
2828 val = READ_ONCE(up->pcslen); in udp_lib_getsockopt()
2832 val = READ_ONCE(up->pcrlen); in udp_lib_getsockopt()
2836 return -ENOPROTOOPT; in udp_lib_getsockopt()
2840 return -EFAULT; in udp_lib_getsockopt()
2842 return -EFAULT; in udp_lib_getsockopt()
2856 * udp_poll - wait for a UDP event.
2857 * @file: - file struct
2858 * @sock: - socket
2859 * @wait: - poll table
2871 struct sock *sk = sock->sk; in udp_poll()
2873 if (!skb_queue_empty_lockless(&udp_sk(sk)->reader_queue)) in udp_poll()
2877 if ((mask & EPOLLRDNORM) && !(file->f_flags & O_NONBLOCK) && in udp_poll()
2878 !(sk->sk_shutdown & RCV_SHUTDOWN) && first_packet_length(sk) == -1) in udp_poll()
2900 sk->sk_err = err; in udp_abort()
2948 /* ------------------------------------------------------------------------ */
2957 return ((family == AF_UNSPEC || family == sk->sk_family) && in seq_sk_match()
2970 if (seq->op == &bpf_iter_udp_seq_ops) in udp_get_table_seq()
2971 return net->ipv4.udp_table; in udp_get_table_seq()
2974 afinfo = pde_data(file_inode(seq->file)); in udp_get_table_seq()
2975 return afinfo->udp_table ? : net->ipv4.udp_table; in udp_get_table_seq()
2980 struct udp_iter_state *state = seq->private; in udp_get_first()
2987 for (state->bucket = start; state->bucket <= udptable->mask; in udp_get_first()
2988 ++state->bucket) { in udp_get_first()
2989 struct udp_hslot *hslot = &udptable->hash[state->bucket]; in udp_get_first()
2991 if (hlist_empty(&hslot->head)) in udp_get_first()
2994 spin_lock_bh(&hslot->lock); in udp_get_first()
2995 sk_for_each(sk, &hslot->head) { in udp_get_first()
2999 spin_unlock_bh(&hslot->lock); in udp_get_first()
3008 struct udp_iter_state *state = seq->private; in udp_get_next()
3019 if (state->bucket <= udptable->mask) in udp_get_next()
3020 spin_unlock_bh(&udptable->hash[state->bucket].lock); in udp_get_next()
3022 return udp_get_first(seq, state->bucket + 1); in udp_get_next()
3033 --pos; in udp_get_idx()
3039 struct udp_iter_state *state = seq->private; in udp_seq_start()
3040 state->bucket = MAX_UDP_PORTS; in udp_seq_start()
3042 return *pos ? udp_get_idx(seq, *pos-1) : SEQ_START_TOKEN; in udp_seq_start()
3062 struct udp_iter_state *state = seq->private; in udp_seq_stop()
3067 if (state->bucket <= udptable->mask) in udp_seq_stop()
3068 spin_unlock_bh(&udptable->hash[state->bucket].lock); in udp_seq_stop()
3072 /* ------------------------------------------------------------------------ */
3077 __be32 dest = inet->inet_daddr; in udp4_format_sock()
3078 __be32 src = inet->inet_rcv_saddr; in udp4_format_sock()
3079 __u16 destp = ntohs(inet->inet_dport); in udp4_format_sock()
3080 __u16 srcp = ntohs(inet->inet_sport); in udp4_format_sock()
3084 bucket, src, srcp, dest, destp, sp->sk_state, in udp4_format_sock()
3090 refcount_read(&sp->sk_refcnt), sp, in udp4_format_sock()
3091 atomic_read(&sp->sk_drops)); in udp4_format_sock()
3099 "rx_queue tr tm->when retrnsmt uid timeout " in udp4_seq_show()
3102 struct udp_iter_state *state = seq->private; in udp4_seq_show()
3104 udp4_format_sock(v, seq, state->bucket); in udp4_seq_show()
3132 struct bpf_udp_iter_state *iter = seq->private; in bpf_iter_udp_batch()
3133 struct udp_iter_state *state = &iter->state; in bpf_iter_udp_batch()
3141 resume_bucket = state->bucket; in bpf_iter_udp_batch()
3142 resume_offset = iter->offset; in bpf_iter_udp_batch()
3145 if (iter->st_bucket_done) in bpf_iter_udp_batch()
3146 state->bucket++; in bpf_iter_udp_batch()
3158 iter->cur_sk = 0; in bpf_iter_udp_batch()
3159 iter->end_sk = 0; in bpf_iter_udp_batch()
3160 iter->st_bucket_done = false; in bpf_iter_udp_batch()
3163 for (; state->bucket <= udptable->mask; state->bucket++) { in bpf_iter_udp_batch()
3164 struct udp_hslot *hslot2 = &udptable->hash2[state->bucket]; in bpf_iter_udp_batch()
3166 if (hlist_empty(&hslot2->head)) in bpf_iter_udp_batch()
3169 iter->offset = 0; in bpf_iter_udp_batch()
3170 spin_lock_bh(&hslot2->lock); in bpf_iter_udp_batch()
3171 udp_portaddr_for_each_entry(sk, &hslot2->head) { in bpf_iter_udp_batch()
3176 if (state->bucket == resume_bucket && in bpf_iter_udp_batch()
3177 iter->offset < resume_offset) { in bpf_iter_udp_batch()
3178 ++iter->offset; in bpf_iter_udp_batch()
3181 if (iter->end_sk < iter->max_sk) { in bpf_iter_udp_batch()
3183 iter->batch[iter->end_sk++] = sk; in bpf_iter_udp_batch()
3188 spin_unlock_bh(&hslot2->lock); in bpf_iter_udp_batch()
3190 if (iter->end_sk) in bpf_iter_udp_batch()
3195 if (!iter->end_sk) in bpf_iter_udp_batch()
3198 if (iter->end_sk == batch_sks) { in bpf_iter_udp_batch()
3202 iter->st_bucket_done = true; in bpf_iter_udp_batch()
3213 return iter->batch[0]; in bpf_iter_udp_batch()
3218 struct bpf_udp_iter_state *iter = seq->private; in bpf_iter_udp_seq_next()
3221 /* Whenever seq_next() is called, the iter->cur_sk is in bpf_iter_udp_seq_next()
3222 * done with seq_show(), so unref the iter->cur_sk. in bpf_iter_udp_seq_next()
3224 if (iter->cur_sk < iter->end_sk) { in bpf_iter_udp_seq_next()
3225 sock_put(iter->batch[iter->cur_sk++]); in bpf_iter_udp_seq_next()
3226 ++iter->offset; in bpf_iter_udp_seq_next()
3229 /* After updating iter->cur_sk, check if there are more sockets in bpf_iter_udp_seq_next()
3232 if (iter->cur_sk < iter->end_sk) in bpf_iter_udp_seq_next()
3233 sk = iter->batch[iter->cur_sk]; in bpf_iter_udp_seq_next()
3245 * continue from where it was stop()-ped. in bpf_iter_udp_seq_start()
3258 meta->seq_num--; /* skip SEQ_START_TOKEN */ in udp_prog_seq_show()
3268 struct udp_iter_state *state = seq->private; in bpf_iter_udp_seq_show()
3288 ret = udp_prog_seq_show(prog, &meta, v, uid, state->bucket); in bpf_iter_udp_seq_show()
3297 while (iter->cur_sk < iter->end_sk) in bpf_iter_udp_put_batch()
3298 sock_put(iter->batch[iter->cur_sk++]); in bpf_iter_udp_put_batch()
3303 struct bpf_udp_iter_state *iter = seq->private; in bpf_iter_udp_seq_stop()
3314 if (iter->cur_sk < iter->end_sk) { in bpf_iter_udp_seq_stop()
3316 iter->st_bucket_done = false; in bpf_iter_udp_seq_stop()
3334 if (seq->op == &bpf_iter_udp_seq_ops) in seq_file_family()
3339 afinfo = pde_data(file_inode(seq->file)); in seq_file_family()
3340 return afinfo->family; in seq_file_family()
3358 if (!proc_create_net_data("udp", 0444, net->proc_net, &udp_seq_ops, in udp4_proc_init_net()
3360 return -ENOMEM; in udp4_proc_init_net()
3366 remove_proc_entry("udp", net->proc_net); in udp4_proc_exit_net()
3407 table->hash = alloc_large_system_hash(name, in udp_table_init()
3412 &table->log, in udp_table_init()
3413 &table->mask, in udp_table_init()
3417 table->hash2 = table->hash + (table->mask + 1); in udp_table_init()
3418 for (i = 0; i <= table->mask; i++) { in udp_table_init()
3419 INIT_HLIST_HEAD(&table->hash[i].head); in udp_table_init()
3420 table->hash[i].count = 0; in udp_table_init()
3421 spin_lock_init(&table->hash[i].lock); in udp_table_init()
3423 for (i = 0; i <= table->mask; i++) { in udp_table_init()
3424 INIT_HLIST_HEAD(&table->hash2[i].head); in udp_table_init()
3425 table->hash2[i].count = 0; in udp_table_init()
3426 spin_lock_init(&table->hash2[i].lock); in udp_table_init()
3442 net->ipv4.sysctl_udp_rmem_min = PAGE_SIZE; in udp_sysctl_init()
3443 net->ipv4.sysctl_udp_wmem_min = PAGE_SIZE; in udp_sysctl_init()
3446 net->ipv4.sysctl_udp_l3mdev_accept = 0; in udp_sysctl_init()
3459 udptable->hash = vmalloc_huge(hash_entries * 2 * sizeof(struct udp_hslot), in udp_pernet_table_alloc()
3461 if (!udptable->hash) in udp_pernet_table_alloc()
3464 udptable->hash2 = udptable->hash + hash_entries; in udp_pernet_table_alloc()
3465 udptable->mask = hash_entries - 1; in udp_pernet_table_alloc()
3466 udptable->log = ilog2(hash_entries); in udp_pernet_table_alloc()
3469 INIT_HLIST_HEAD(&udptable->hash[i].head); in udp_pernet_table_alloc()
3470 udptable->hash[i].count = 0; in udp_pernet_table_alloc()
3471 spin_lock_init(&udptable->hash[i].lock); in udp_pernet_table_alloc()
3473 INIT_HLIST_HEAD(&udptable->hash2[i].head); in udp_pernet_table_alloc()
3474 udptable->hash2[i].count = 0; in udp_pernet_table_alloc()
3475 spin_lock_init(&udptable->hash2[i].lock); in udp_pernet_table_alloc()
3488 struct udp_table *udptable = net->ipv4.udp_table; in udp_pernet_table_free()
3493 kvfree(udptable->hash); in udp_pernet_table_free()
3506 old_net = current->nsproxy->net_ns; in udp_set_table()
3507 hash_entries = READ_ONCE(old_net->ipv4.sysctl_udp_child_hash_entries); in udp_set_table()
3519 net->ipv4.udp_table = udptable; in udp_set_table()
3525 net->ipv4.udp_table = &udp_table; in udp_set_table()
3559 return -ENOMEM; in DEFINE_BPF_ITER_FUNC()
3562 kvfree(iter->batch); in DEFINE_BPF_ITER_FUNC()
3563 iter->batch = new_batch; in DEFINE_BPF_ITER_FUNC()
3564 iter->max_sk = new_batch_sz; in DEFINE_BPF_ITER_FUNC()
3592 kvfree(iter->batch); in bpf_iter_fini_udp()