Lines Matching +full:broken +full:- +full:prefetch +full:- +full:cmd
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>
139 return sk->sk_prot->h.udp_table ? : sock_net(sk)->ipv4.udp_table; in udp_get_table_prot()
150 sk_for_each(sk2, &hslot->head) { in udp_lib_lport_inuse()
153 (bitmap || udp_sk(sk2)->udp_port_hash == num) && in udp_lib_lport_inuse()
154 (!sk2->sk_reuse || !sk->sk_reuse) && in udp_lib_lport_inuse()
155 (!sk2->sk_bound_dev_if || !sk->sk_bound_dev_if || in udp_lib_lport_inuse()
156 sk2->sk_bound_dev_if == sk->sk_bound_dev_if) && in udp_lib_lport_inuse()
158 if (sk2->sk_reuseport && sk->sk_reuseport && in udp_lib_lport_inuse()
159 !rcu_access_pointer(sk->sk_reuseport_cb) && in udp_lib_lport_inuse()
166 __set_bit(udp_sk(sk2)->udp_port_hash >> log, in udp_lib_lport_inuse()
186 spin_lock(&hslot2->lock); in udp_lib_lport_inuse2()
187 udp_portaddr_for_each_entry(sk2, &hslot2->head) { in udp_lib_lport_inuse2()
190 (udp_sk(sk2)->udp_port_hash == num) && in udp_lib_lport_inuse2()
191 (!sk2->sk_reuse || !sk->sk_reuse) && in udp_lib_lport_inuse2()
192 (!sk2->sk_bound_dev_if || !sk->sk_bound_dev_if || in udp_lib_lport_inuse2()
193 sk2->sk_bound_dev_if == sk->sk_bound_dev_if) && in udp_lib_lport_inuse2()
195 if (sk2->sk_reuseport && sk->sk_reuseport && in udp_lib_lport_inuse2()
196 !rcu_access_pointer(sk->sk_reuseport_cb) && in udp_lib_lport_inuse2()
205 spin_unlock(&hslot2->lock); in udp_lib_lport_inuse2()
215 sk_for_each(sk2, &hslot->head) { in udp_reuseport_add_sock()
218 sk2->sk_family == sk->sk_family && in udp_reuseport_add_sock()
220 (udp_sk(sk2)->udp_port_hash == udp_sk(sk)->udp_port_hash) && in udp_reuseport_add_sock()
221 (sk2->sk_bound_dev_if == sk->sk_bound_dev_if) && in udp_reuseport_add_sock()
222 sk2->sk_reuseport && uid_eq(uid, sock_i_uid(sk2)) && in udp_reuseport_add_sock()
233 * udp_lib_get_port - UDP/-Lite port lookup for IPv4 and IPv6
237 * @hash2_nulladdr: AF-dependent hash value in secondary hash chains,
246 int error = -EADDRINUSE; in udp_lib_get_port()
255 remaining = (high - low) + 1; in udp_lib_get_port()
262 rand = (rand | 1) * (udptable->mask + 1); in udp_lib_get_port()
263 last = first + udptable->mask + 1; in udp_lib_get_port()
267 spin_lock_bh(&hslot->lock); in udp_lib_get_port()
269 udptable->log); in udp_lib_get_port()
279 !test_bit(snum >> udptable->log, bitmap) && in udp_lib_get_port()
284 spin_unlock_bh(&hslot->lock); in udp_lib_get_port()
290 spin_lock_bh(&hslot->lock); in udp_lib_get_port()
291 if (hslot->count > 10) { in udp_lib_get_port()
293 unsigned int slot2 = udp_sk(sk)->udp_portaddr_hash ^ snum; in udp_lib_get_port()
295 slot2 &= udptable->mask; in udp_lib_get_port()
296 hash2_nulladdr &= udptable->mask; in udp_lib_get_port()
299 if (hslot->count < hslot2->count) in udp_lib_get_port()
318 inet_sk(sk)->inet_num = snum; in udp_lib_get_port()
319 udp_sk(sk)->udp_port_hash = snum; in udp_lib_get_port()
320 udp_sk(sk)->udp_portaddr_hash ^= snum; in udp_lib_get_port()
322 if (sk->sk_reuseport && in udp_lib_get_port()
324 inet_sk(sk)->inet_num = 0; in udp_lib_get_port()
325 udp_sk(sk)->udp_port_hash = 0; in udp_lib_get_port()
326 udp_sk(sk)->udp_portaddr_hash ^= snum; in udp_lib_get_port()
332 sk_add_node_rcu(sk, &hslot->head); in udp_lib_get_port()
333 hslot->count++; in udp_lib_get_port()
334 sock_prot_inuse_add(sock_net(sk), sk->sk_prot, 1); in udp_lib_get_port()
336 hslot2 = udp_hashslot2(udptable, udp_sk(sk)->udp_portaddr_hash); in udp_lib_get_port()
337 spin_lock(&hslot2->lock); in udp_lib_get_port()
338 if (IS_ENABLED(CONFIG_IPV6) && sk->sk_reuseport && in udp_lib_get_port()
339 sk->sk_family == AF_INET6) in udp_lib_get_port()
340 hlist_add_tail_rcu(&udp_sk(sk)->udp_portaddr_node, in udp_lib_get_port()
341 &hslot2->head); in udp_lib_get_port()
343 hlist_add_head_rcu(&udp_sk(sk)->udp_portaddr_node, in udp_lib_get_port()
344 &hslot2->head); in udp_lib_get_port()
345 hslot2->count++; in udp_lib_get_port()
346 spin_unlock(&hslot2->lock); in udp_lib_get_port()
351 spin_unlock_bh(&hslot->lock); in udp_lib_get_port()
362 ipv4_portaddr_hash(sock_net(sk), inet_sk(sk)->inet_rcv_saddr, 0); in udp_v4_get_port()
365 udp_sk(sk)->udp_portaddr_hash = hash2_partial; in udp_v4_get_port()
379 udp_sk(sk)->udp_port_hash != hnum || in compute_score()
381 return -1; in compute_score()
383 if (sk->sk_rcv_saddr != daddr) in compute_score()
384 return -1; in compute_score()
386 score = (sk->sk_family == PF_INET) ? 2 : 1; in compute_score()
389 if (inet->inet_daddr) { in compute_score()
390 if (inet->inet_daddr != saddr) in compute_score()
391 return -1; in compute_score()
395 if (inet->inet_dport) { in compute_score()
396 if (inet->inet_dport != sport) in compute_score()
397 return -1; in compute_score()
401 dev_match = udp_sk_bound_dev_eq(net, sk->sk_bound_dev_if, in compute_score()
404 return -1; in compute_score()
405 if (sk->sk_bound_dev_if) in compute_score()
408 if (READ_ONCE(sk->sk_incoming_cpu) == raw_smp_processor_id()) in compute_score()
424 * udp4_lib_lookup1() - Simplified lookup using primary hash (destination port)
449 unsigned int slot = udp_hashfn(net, hnum, udptable->mask); in udp4_lib_lookup1()
450 struct udp_hslot *hslot = &udptable->hash[slot]; in udp4_lib_lookup1()
454 sk_for_each_rcu(sk, &hslot->head) { in udp4_lib_lookup1()
480 udp_portaddr_for_each_entry_rcu(sk, &hslot2->head) { in udp4_lib_lookup2()
491 if (sk->sk_state == TCP_ESTABLISHED) { in udp4_lib_lookup2()
557 slot = hash4 & udptable->mask; in udp4_lib_lookup4()
558 hslot4 = &udptable->hash4[slot]; in udp4_lib_lookup4()
563 udp_lrpa_for_each_entry_rcu(up, node, &hslot4->nulls_head) { in udp4_lib_lookup4()
585 hslot4 = udp_hashslot4(udptable, udp_sk(sk)->udp_lrpa_hash); in udp_rehash4()
587 udp_sk(sk)->udp_lrpa_hash = newhash4; in udp_rehash4()
590 spin_lock_bh(&hslot4->lock); in udp_rehash4()
591 hlist_nulls_del_init_rcu(&udp_sk(sk)->udp_lrpa_node); in udp_rehash4()
592 hslot4->count--; in udp_rehash4()
593 spin_unlock_bh(&hslot4->lock); in udp_rehash4()
595 spin_lock_bh(&nhslot4->lock); in udp_rehash4()
596 hlist_nulls_add_head_rcu(&udp_sk(sk)->udp_lrpa_node, in udp_rehash4()
597 &nhslot4->nulls_head); in udp_rehash4()
598 nhslot4->count++; in udp_rehash4()
599 spin_unlock_bh(&nhslot4->lock); in udp_rehash4()
608 hslot2 = udp_hashslot2(udptable, udp_sk(sk)->udp_portaddr_hash); in udp_unhash4()
609 hslot4 = udp_hashslot4(udptable, udp_sk(sk)->udp_lrpa_hash); in udp_unhash4()
611 spin_lock(&hslot4->lock); in udp_unhash4()
612 hlist_nulls_del_init_rcu(&udp_sk(sk)->udp_lrpa_node); in udp_unhash4()
613 hslot4->count--; in udp_unhash4()
614 spin_unlock(&hslot4->lock); in udp_unhash4()
616 spin_lock(&hslot2->lock); in udp_unhash4()
618 spin_unlock(&hslot2->lock); in udp_unhash4()
628 /* Connected udp socket can re-connect to another remote address, which in udp_lib_hash4()
634 udptable = net->ipv4.udp_table; in udp_lib_hash4()
635 hslot = udp_hashslot(udptable, net, udp_sk(sk)->udp_port_hash); in udp_lib_hash4()
636 hslot2 = udp_hashslot2(udptable, udp_sk(sk)->udp_portaddr_hash); in udp_lib_hash4()
638 udp_sk(sk)->udp_lrpa_hash = hash; in udp_lib_hash4()
640 spin_lock_bh(&hslot->lock); in udp_lib_hash4()
641 if (rcu_access_pointer(sk->sk_reuseport_cb)) in udp_lib_hash4()
644 spin_lock(&hslot4->lock); in udp_lib_hash4()
645 hlist_nulls_add_head_rcu(&udp_sk(sk)->udp_lrpa_node, in udp_lib_hash4()
646 &hslot4->nulls_head); in udp_lib_hash4()
647 hslot4->count++; in udp_lib_hash4()
648 spin_unlock(&hslot4->lock); in udp_lib_hash4()
650 spin_lock(&hslot2->lock); in udp_lib_hash4()
652 spin_unlock(&hslot2->lock); in udp_lib_hash4()
654 spin_unlock_bh(&hslot->lock); in udp_lib_hash4()
664 if (sk_unhashed(sk) || sk->sk_rcv_saddr == htonl(INADDR_ANY)) in udp4_hash4()
667 hash = udp_ehashfn(net, sk->sk_rcv_saddr, sk->sk_num, in udp4_hash4()
668 sk->sk_daddr, sk->sk_dport); in udp4_hash4()
676 * harder than this. -DaveM
697 /* Lookup connected or non-wildcard socket */ in __udp4_lib_lookup()
701 if (!IS_ERR_OR_NULL(result) && result->sk_state == TCP_ESTABLISHED) in __udp4_lib_lookup()
706 udptable == net->ipv4.udp_table) { in __udp4_lib_lookup()
716 /* Got non-wildcard socket or error on first lookup */ in __udp4_lib_lookup()
733 * 3. rehash operation updating _secondary and four-tuple_ hashes in __udp4_lib_lookup()
753 return __udp4_lib_lookup(dev_net(skb->dev), iph->saddr, sport, in __udp4_lib_lookup_skb()
754 iph->daddr, dport, inet_iif(skb), in __udp4_lib_lookup_skb()
761 const u16 offset = NAPI_GRO_CB(skb)->network_offsets[skb->encapsulation]; in udp4_lib_lookup_skb()
762 const struct iphdr *iph = (struct iphdr *)(skb->data + offset); in udp4_lib_lookup_skb()
763 struct net *net = dev_net(skb->dev); in udp4_lib_lookup_skb()
768 return __udp4_lib_lookup(net, iph->saddr, sport, in udp4_lib_lookup_skb()
769 iph->daddr, dport, iif, in udp4_lib_lookup_skb()
770 sdif, net->ipv4.udp_table, NULL); in udp4_lib_lookup_skb()
783 dif, 0, net->ipv4.udp_table, NULL); in udp4_lib_lookup()
784 if (sk && !refcount_inc_not_zero(&sk->sk_refcnt)) in udp4_lib_lookup()
799 udp_sk(sk)->udp_port_hash != hnum || in __udp_is_mcast_sock()
800 (inet->inet_daddr && inet->inet_daddr != rmt_addr) || in __udp_is_mcast_sock()
801 (inet->inet_dport != rmt_port && inet->inet_dport) || in __udp_is_mcast_sock()
802 (inet->inet_rcv_saddr && inet->inet_rcv_saddr != loc_addr) || in __udp_is_mcast_sock()
804 !udp_sk_bound_dev_eq(net, sk->sk_bound_dev_if, dif, sdif)) in __udp_is_mcast_sock()
845 handler = encap->err_handler; in __udp4_lib_err_encap_no_sk()
850 return -ENOENT; in __udp4_lib_err_encap_no_sk()
888 skb_set_transport_header(skb, iph->ihl << 2); in __udp4_lib_err_encap()
893 lookup = READ_ONCE(up->encap_err_lookup); in __udp4_lib_err_encap()
900 sk = __udp4_lib_lookup(net, iph->daddr, uh->source, in __udp4_lib_err_encap()
901 iph->saddr, uh->dest, skb->dev->ifindex, 0, in __udp4_lib_err_encap()
906 lookup = READ_ONCE(up->encap_err_lookup); in __udp4_lib_err_encap()
935 const struct iphdr *iph = (const struct iphdr *)skb->data; in __udp4_lib_err()
936 struct udphdr *uh = (struct udphdr *)(skb->data+(iph->ihl<<2)); in __udp4_lib_err()
937 const int type = icmp_hdr(skb)->type; in __udp4_lib_err()
938 const int code = icmp_hdr(skb)->code; in __udp4_lib_err()
943 struct net *net = dev_net(skb->dev); in __udp4_lib_err()
945 sk = __udp4_lib_lookup(net, iph->daddr, uh->dest, in __udp4_lib_err()
946 iph->saddr, uh->source, skb->dev->ifindex, in __udp4_lib_err()
949 if (!sk || READ_ONCE(udp_sk(sk)->encap_type)) { in __udp4_lib_err()
957 sk = ERR_PTR(-ENOENT); in __udp4_lib_err()
985 if (READ_ONCE(inet->pmtudisc) != IP_PMTUDISC_DONT) { in __udp4_lib_err()
1009 if (udp_sk(sk)->encap_err_rcv) in __udp4_lib_err()
1010 udp_sk(sk)->encap_err_rcv(sk, skb, err, uh->dest, info, in __udp4_lib_err()
1015 if (!harderr || sk->sk_state != TCP_ESTABLISHED) in __udp4_lib_err()
1018 ip_icmp_error(sk, skb, err, uh->dest, info, (u8 *)(uh+1)); in __udp4_lib_err()
1020 sk->sk_err = err; in __udp4_lib_err()
1028 return __udp4_lib_err(skb, info, dev_net(skb->dev)->ipv4.udp_table); in udp_err()
1038 if (up->pending) { in udp_flush_pending_frames()
1039 up->len = 0; in udp_flush_pending_frames()
1040 WRITE_ONCE(up->pending, 0); in udp_flush_pending_frames()
1047 * udp4_hwcsum - handle outgoing HW checksumming
1048 * @skb: sk_buff containing the filled-in UDP header
1057 int len = skb->len - offset; in udp4_hwcsum()
1065 skb->csum_start = skb_transport_header(skb) - skb->head; in udp4_hwcsum()
1066 skb->csum_offset = offsetof(struct udphdr, check); in udp4_hwcsum()
1067 uh->check = ~csum_tcpudp_magic(src, dst, len, in udp4_hwcsum()
1073 * HW-checksum won't work as there are two or more in udp4_hwcsum()
1078 csum = csum_add(csum, frags->csum); in udp4_hwcsum()
1079 hlen -= frags->len; in udp4_hwcsum()
1083 skb->ip_summed = CHECKSUM_NONE; in udp4_hwcsum()
1085 uh->check = csum_tcpudp_magic(src, dst, len, IPPROTO_UDP, csum); in udp4_hwcsum()
1086 if (uh->check == 0) in udp4_hwcsum()
1087 uh->check = CSUM_MANGLED_0; in udp4_hwcsum()
1101 uh->check = 0; in udp_set_csum()
1103 uh->check = ~udp_v4_check(len, saddr, daddr, 0); in udp_set_csum()
1104 } else if (skb->ip_summed == CHECKSUM_PARTIAL) { in udp_set_csum()
1105 uh->check = 0; in udp_set_csum()
1106 uh->check = udp_v4_check(len, saddr, daddr, lco_csum(skb)); in udp_set_csum()
1107 if (uh->check == 0) in udp_set_csum()
1108 uh->check = CSUM_MANGLED_0; in udp_set_csum()
1110 skb->ip_summed = CHECKSUM_PARTIAL; in udp_set_csum()
1111 skb->csum_start = skb_transport_header(skb) - skb->head; in udp_set_csum()
1112 skb->csum_offset = offsetof(struct udphdr, check); in udp_set_csum()
1113 uh->check = ~udp_v4_check(len, saddr, daddr, 0); in udp_set_csum()
1121 struct sock *sk = skb->sk; in udp_send_skb()
1127 int len = skb->len - offset; in udp_send_skb()
1128 int datalen = len - sizeof(*uh); in udp_send_skb()
1135 uh->source = inet->inet_sport; in udp_send_skb()
1136 uh->dest = fl4->fl4_dport; in udp_send_skb()
1137 uh->len = htons(len); in udp_send_skb()
1138 uh->check = 0; in udp_send_skb()
1140 if (cork->gso_size) { in udp_send_skb()
1144 if (hlen + min(datalen, cork->gso_size) > cork->fragsize) { in udp_send_skb()
1146 return -EMSGSIZE; in udp_send_skb()
1148 if (datalen > cork->gso_size * UDP_MAX_SEGMENTS) { in udp_send_skb()
1150 return -EINVAL; in udp_send_skb()
1152 if (sk->sk_no_check_tx) { in udp_send_skb()
1154 return -EINVAL; in udp_send_skb()
1158 return -EIO; in udp_send_skb()
1161 if (datalen > cork->gso_size) { in udp_send_skb()
1162 skb_shinfo(skb)->gso_size = cork->gso_size; in udp_send_skb()
1163 skb_shinfo(skb)->gso_type = SKB_GSO_UDP_L4; in udp_send_skb()
1164 skb_shinfo(skb)->gso_segs = DIV_ROUND_UP(datalen, in udp_send_skb()
1165 cork->gso_size); in udp_send_skb()
1172 if (is_udplite) /* UDP-Lite */ in udp_send_skb()
1175 else if (sk->sk_no_check_tx) { /* UDP csum off */ in udp_send_skb()
1177 skb->ip_summed = CHECKSUM_NONE; in udp_send_skb()
1180 } else if (skb->ip_summed == CHECKSUM_PARTIAL) { /* UDP hardware csum */ in udp_send_skb()
1183 udp4_hwcsum(skb, fl4->saddr, fl4->daddr); in udp_send_skb()
1189 /* add protocol-dependent pseudo-header */ in udp_send_skb()
1190 uh->check = csum_tcpudp_magic(fl4->saddr, fl4->daddr, len, in udp_send_skb()
1191 sk->sk_protocol, csum); in udp_send_skb()
1192 if (uh->check == 0) in udp_send_skb()
1193 uh->check = CSUM_MANGLED_0; in udp_send_skb()
1198 if (err == -ENOBUFS && in udp_send_skb()
1217 struct flowi4 *fl4 = &inet->cork.fl.u.ip4; in udp_push_pending_frames()
1225 err = udp_send_skb(skb, fl4, &inet->cork.base); in udp_push_pending_frames()
1228 up->len = 0; in udp_push_pending_frames()
1229 WRITE_ONCE(up->pending, 0); in udp_push_pending_frames()
1236 switch (cmsg->cmsg_type) { in __udp_cmsg_send()
1238 if (cmsg->cmsg_len != CMSG_LEN(sizeof(__u16))) in __udp_cmsg_send()
1239 return -EINVAL; in __udp_cmsg_send()
1243 return -EINVAL; in __udp_cmsg_send()
1255 return -EINVAL; in udp_cmsg_send()
1257 if (cmsg->cmsg_level != SOL_UDP) { in udp_cmsg_send()
1275 DECLARE_SOCKADDR(struct sockaddr_in *, usin, msg->msg_name); in udp_sendmsg()
1287 int corkreq = udp_test_bit(CORK, sk) || msg->msg_flags & MSG_MORE; in udp_sendmsg()
1294 return -EMSGSIZE; in udp_sendmsg()
1300 if (msg->msg_flags & MSG_OOB) /* Mirror BSD error message compatibility */ in udp_sendmsg()
1301 return -EOPNOTSUPP; in udp_sendmsg()
1305 fl4 = &inet->cork.fl.u.ip4; in udp_sendmsg()
1306 if (READ_ONCE(up->pending)) { in udp_sendmsg()
1312 if (likely(up->pending)) { in udp_sendmsg()
1313 if (unlikely(up->pending != AF_INET)) { in udp_sendmsg()
1315 return -EINVAL; in udp_sendmsg()
1327 if (msg->msg_namelen < sizeof(*usin)) in udp_sendmsg()
1328 return -EINVAL; in udp_sendmsg()
1329 if (usin->sin_family != AF_INET) { in udp_sendmsg()
1330 if (usin->sin_family != AF_UNSPEC) in udp_sendmsg()
1331 return -EAFNOSUPPORT; in udp_sendmsg()
1334 daddr = usin->sin_addr.s_addr; in udp_sendmsg()
1335 dport = usin->sin_port; in udp_sendmsg()
1337 return -EINVAL; in udp_sendmsg()
1339 if (sk->sk_state != TCP_ESTABLISHED) in udp_sendmsg()
1340 return -EDESTADDRREQ; in udp_sendmsg()
1341 daddr = inet->inet_daddr; in udp_sendmsg()
1342 dport = inet->inet_dport; in udp_sendmsg()
1350 ipc.gso_size = READ_ONCE(up->gso_size); in udp_sendmsg()
1352 if (msg->msg_controllen) { in udp_sendmsg()
1356 sk->sk_family == AF_INET6); in udp_sendmsg()
1370 inet_opt = rcu_dereference(inet->inet_opt); in udp_sendmsg()
1373 sizeof(*inet_opt) + inet_opt->opt.optlen); in udp_sendmsg()
1382 &msg->msg_namelen, in udp_sendmsg()
1387 if (usin->sin_port == 0) { in udp_sendmsg()
1389 err = -EINVAL; in udp_sendmsg()
1392 daddr = usin->sin_addr.s_addr; in udp_sendmsg()
1393 dport = usin->sin_port; in udp_sendmsg()
1400 if (ipc.opt && ipc.opt->opt.srr) { in udp_sendmsg()
1402 err = -EINVAL; in udp_sendmsg()
1405 faddr = ipc.opt->opt.faddr; in udp_sendmsg()
1413 uc_index = READ_ONCE(inet->uc_index); in udp_sendmsg()
1416 ipc.oif = READ_ONCE(inet->mc_index); in udp_sendmsg()
1418 saddr = READ_ONCE(inet->mc_addr); in udp_sendmsg()
1446 sk->sk_protocol, flow_flags, faddr, saddr, in udp_sendmsg()
1447 dport, inet->inet_sport, sk->sk_uid); in udp_sendmsg()
1454 if (err == -ENETUNREACH) in udp_sendmsg()
1459 err = -EACCES; in udp_sendmsg()
1460 if ((rt->rt_flags & RTCF_BROADCAST) && in udp_sendmsg()
1464 sk_dst_set(sk, dst_clone(&rt->dst)); in udp_sendmsg()
1467 if (msg->msg_flags&MSG_CONFIRM) in udp_sendmsg()
1471 saddr = fl4->saddr; in udp_sendmsg()
1473 daddr = ipc.addr = fl4->daddr; in udp_sendmsg()
1475 /* Lockless fast path for the non-corking case. */ in udp_sendmsg()
1481 &cork, msg->msg_flags); in udp_sendmsg()
1489 if (unlikely(up->pending)) { in udp_sendmsg()
1491 /* ... which is an evident application bug. --ANK */ in udp_sendmsg()
1495 err = -EINVAL; in udp_sendmsg()
1501 fl4 = &inet->cork.fl.u.ip4; in udp_sendmsg()
1502 fl4->daddr = daddr; in udp_sendmsg()
1503 fl4->saddr = saddr; in udp_sendmsg()
1504 fl4->fl4_dport = dport; in udp_sendmsg()
1505 fl4->fl4_sport = inet->inet_sport; in udp_sendmsg()
1506 WRITE_ONCE(up->pending, AF_INET); in udp_sendmsg()
1509 up->len += ulen; in udp_sendmsg()
1512 corkreq ? msg->msg_flags|MSG_MORE : msg->msg_flags); in udp_sendmsg()
1517 else if (unlikely(skb_queue_empty(&sk->sk_write_queue))) in udp_sendmsg()
1518 WRITE_ONCE(up->pending, 0); in udp_sendmsg()
1535 if (err == -ENOBUFS || test_bit(SOCK_NOSPACE, &sk->sk_socket->flags)) { in udp_sendmsg()
1542 if (msg->msg_flags & MSG_PROBE) in udp_sendmsg()
1543 dst_confirm_neigh(&rt->dst, &fl4->daddr); in udp_sendmsg()
1544 if (!(msg->msg_flags&MSG_PROBE) || len) in udp_sendmsg()
1553 struct sock *sk = sock->sk; in udp_splice_eof()
1556 if (!READ_ONCE(up->pending) || udp_test_bit(CORK, sk)) in udp_splice_eof()
1560 if (up->pending && !udp_test_bit(CORK, sk)) in udp_splice_eof()
1594 scratch->_tsize_state = skb->truesize; in udp_set_dev_scratch()
1596 scratch->len = skb->len; in udp_set_dev_scratch()
1597 scratch->csum_unnecessary = !!skb_csum_unnecessary(skb); in udp_set_dev_scratch()
1598 scratch->is_linear = !skb_is_nonlinear(skb); in udp_set_dev_scratch()
1601 scratch->_tsize_state |= UDP_SKB_IS_STATELESS; in udp_set_dev_scratch()
1608 * set skb->csum_valid to 1. in udp_skb_csum_unnecessary_set()
1614 udp_skb_scratch(skb)->csum_unnecessary = true; in udp_skb_csum_unnecessary_set()
1620 return udp_skb_scratch(skb)->_tsize_state & ~UDP_SKB_IS_STATELESS; in udp_skb_truesize()
1625 return !(udp_skb_scratch(skb)->_tsize_state & UDP_SKB_IS_STATELESS); in udp_skb_has_head_state()
1637 up->forward_deficit += size; in udp_rmem_release()
1638 size = up->forward_deficit; in udp_rmem_release()
1639 if (size < READ_ONCE(up->forward_threshold) && in udp_rmem_release()
1640 !skb_queue_empty(&up->reader_queue)) in udp_rmem_release()
1643 size += up->forward_deficit; in udp_rmem_release()
1645 up->forward_deficit = 0; in udp_rmem_release()
1650 sk_queue = &sk->sk_receive_queue; in udp_rmem_release()
1652 spin_lock(&sk_queue->lock); in udp_rmem_release()
1654 amt = (size + sk->sk_forward_alloc - partial) & ~(PAGE_SIZE - 1); in udp_rmem_release()
1655 sk_forward_alloc_add(sk, size - amt); in udp_rmem_release()
1660 atomic_sub(size, &sk->sk_rmem_alloc); in udp_rmem_release()
1663 skb_queue_splice_tail_init(sk_queue, &up->reader_queue); in udp_rmem_release()
1666 spin_unlock(&sk_queue->lock); in udp_rmem_release()
1670 * Instead of using skb->truesize here, find a copy of it in skb->dev_scratch
1676 prefetch(&skb->data); in udp_skb_destructor()
1684 prefetch(&skb->data); in udp_skb_dtor_locked()
1717 delta = size - sk->sk_forward_alloc; in udp_rmem_schedule()
1719 return -ENOBUFS; in udp_rmem_schedule()
1726 struct sk_buff_head *list = &sk->sk_receive_queue; in __udp_enqueue_schedule_skb()
1729 int size, err = -ENOMEM; in __udp_enqueue_schedule_skb()
1731 rmem = atomic_read(&sk->sk_rmem_alloc); in __udp_enqueue_schedule_skb()
1732 rcvbuf = READ_ONCE(sk->sk_rcvbuf); in __udp_enqueue_schedule_skb()
1733 size = skb->truesize; in __udp_enqueue_schedule_skb()
1749 * - Reduce memory overhead and thus increase receive queue capacity in __udp_enqueue_schedule_skb()
1750 * - Less cache line misses at copyout() time in __udp_enqueue_schedule_skb()
1751 * - Less work at consume_skb() (less alien page frag freeing) in __udp_enqueue_schedule_skb()
1755 size = skb->truesize; in __udp_enqueue_schedule_skb()
1761 atomic_add(size, &sk->sk_rmem_alloc); in __udp_enqueue_schedule_skb()
1763 spin_lock(&list->lock); in __udp_enqueue_schedule_skb()
1766 spin_unlock(&list->lock); in __udp_enqueue_schedule_skb()
1770 sk_forward_alloc_add(sk, -size); in __udp_enqueue_schedule_skb()
1778 spin_unlock(&list->lock); in __udp_enqueue_schedule_skb()
1781 INDIRECT_CALL_1(sk->sk_data_ready, sock_def_readable, sk); in __udp_enqueue_schedule_skb()
1787 atomic_sub(skb->truesize, &sk->sk_rmem_alloc); in __udp_enqueue_schedule_skb()
1790 atomic_inc(&sk->sk_drops); in __udp_enqueue_schedule_skb()
1803 skb_queue_splice_tail_init(&sk->sk_receive_queue, &up->reader_queue); in udp_destruct_common()
1804 while ((skb = __skb_dequeue(&up->reader_queue)) != NULL) { in udp_destruct_common()
1805 total += skb->truesize; in udp_destruct_common()
1821 sk->sk_destruct = udp_destruct_sock; in udp_init_sock()
1822 set_bit(SOCK_SUPPORT_ZC, &sk->sk_socket->flags); in udp_init_sock()
1828 if (unlikely(READ_ONCE(udp_sk(sk)->peeking_with_offset))) in skb_consume_udp()
1855 atomic_inc(&sk->sk_drops); in __first_packet_length()
1857 *total += skb->truesize; in __first_packet_length()
1868 * first_packet_length - return length of first packet in receive queue
1872 * Returns the length of found skb, or -1 if none is found.
1876 struct sk_buff_head *rcvq = &udp_sk(sk)->reader_queue; in first_packet_length()
1877 struct sk_buff_head *sk_queue = &sk->sk_receive_queue; in first_packet_length()
1882 spin_lock_bh(&rcvq->lock); in first_packet_length()
1885 spin_lock(&sk_queue->lock); in first_packet_length()
1887 spin_unlock(&sk_queue->lock); in first_packet_length()
1891 res = skb ? skb->len : -1; in first_packet_length()
1894 spin_unlock_bh(&rcvq->lock); in first_packet_length()
1902 int udp_ioctl(struct sock *sk, int cmd, int *karg) in udp_ioctl() argument
1904 switch (cmd) { in udp_ioctl()
1918 return -ENOIOCTLCMD; in udp_ioctl()
1928 struct sk_buff_head *sk_queue = &sk->sk_receive_queue; in __skb_recv_udp()
1934 queue = &udp_sk(sk)->reader_queue; in __skb_recv_udp()
1943 error = -EAGAIN; in __skb_recv_udp()
1945 spin_lock_bh(&queue->lock); in __skb_recv_udp()
1951 spin_unlock_bh(&queue->lock); in __skb_recv_udp()
1956 spin_unlock_bh(&queue->lock); in __skb_recv_udp()
1961 * keep both queues locked to avoid re-acquiring in __skb_recv_udp()
1965 spin_lock(&sk_queue->lock); in __skb_recv_udp()
1972 spin_unlock(&sk_queue->lock); in __skb_recv_udp()
1973 spin_unlock_bh(&queue->lock); in __skb_recv_udp()
1986 !__skb_wait_for_more_packets(sk, &sk->sk_receive_queue, in __skb_recv_udp()
2011 atomic_inc(&sk->sk_drops); in udp_read_skb()
2030 DECLARE_SOCKADDR(struct sockaddr_in *, sin, msg->msg_name); in udp_recvmsg()
2048 if (copied > ulen - off) in udp_recvmsg()
2049 copied = ulen - off; in udp_recvmsg()
2051 msg->msg_flags |= MSG_TRUNC; in udp_recvmsg()
2056 * coverage checksum (UDP-Lite), do it before the copy. in udp_recvmsg()
2060 (is_udplite && UDP_SKB_CB(skb)->partial_cov)) { in udp_recvmsg()
2069 err = copy_linear_skb(skb, copied, off, &msg->msg_iter); in udp_recvmsg()
2075 if (err == -EINVAL) in udp_recvmsg()
2081 atomic_inc(&sk->sk_drops); in udp_recvmsg()
2097 sin->sin_family = AF_INET; in udp_recvmsg()
2098 sin->sin_port = udp_hdr(skb)->source; in udp_recvmsg()
2099 sin->sin_addr.s_addr = ip_hdr(skb)->saddr; in udp_recvmsg()
2100 memset(sin->sin_zero, 0, sizeof(sin->sin_zero)); in udp_recvmsg()
2118 skb_consume_udp(sk, skb, peeking ? -err : err); in udp_recvmsg()
2122 if (!__sk_queue_drop_skb(sk, &udp_sk(sk)->reader_queue, skb, flags, in udp_recvmsg()
2131 msg->msg_flags &= ~MSG_TRUNC; in udp_recvmsg()
2142 return -EINVAL; in udp_pre_connect()
2164 * 1003.1g - break association. in __udp_disconnect()
2167 sk->sk_state = TCP_CLOSE; in __udp_disconnect()
2168 inet->inet_daddr = 0; in __udp_disconnect()
2169 inet->inet_dport = 0; in __udp_disconnect()
2171 sk->sk_bound_dev_if = 0; in __udp_disconnect()
2172 if (!(sk->sk_userlocks & SOCK_BINDADDR_LOCK)) { in __udp_disconnect()
2174 if (sk->sk_prot->rehash && in __udp_disconnect()
2175 (sk->sk_userlocks & SOCK_BINDPORT_LOCK)) in __udp_disconnect()
2176 sk->sk_prot->rehash(sk); in __udp_disconnect()
2179 if (!(sk->sk_userlocks & SOCK_BINDPORT_LOCK)) { in __udp_disconnect()
2180 sk->sk_prot->unhash(sk); in __udp_disconnect()
2181 inet->inet_sport = 0; in __udp_disconnect()
2204 udp_sk(sk)->udp_port_hash); in udp_lib_unhash()
2205 hslot2 = udp_hashslot2(udptable, udp_sk(sk)->udp_portaddr_hash); in udp_lib_unhash()
2207 spin_lock_bh(&hslot->lock); in udp_lib_unhash()
2208 if (rcu_access_pointer(sk->sk_reuseport_cb)) in udp_lib_unhash()
2211 hslot->count--; in udp_lib_unhash()
2212 inet_sk(sk)->inet_num = 0; in udp_lib_unhash()
2213 sock_prot_inuse_add(sock_net(sk), sk->sk_prot, -1); in udp_lib_unhash()
2215 spin_lock(&hslot2->lock); in udp_lib_unhash()
2216 hlist_del_init_rcu(&udp_sk(sk)->udp_portaddr_node); in udp_lib_unhash()
2217 hslot2->count--; in udp_lib_unhash()
2218 spin_unlock(&hslot2->lock); in udp_lib_unhash()
2222 spin_unlock_bh(&hslot->lock); in udp_lib_unhash()
2237 udp_sk(sk)->udp_port_hash); in udp_lib_rehash()
2238 hslot2 = udp_hashslot2(udptable, udp_sk(sk)->udp_portaddr_hash); in udp_lib_rehash()
2240 udp_sk(sk)->udp_portaddr_hash = newhash; in udp_lib_rehash()
2243 rcu_access_pointer(sk->sk_reuseport_cb)) { in udp_lib_rehash()
2245 spin_lock_bh(&hslot->lock); in udp_lib_rehash()
2246 if (rcu_access_pointer(sk->sk_reuseport_cb)) in udp_lib_rehash()
2250 spin_lock(&hslot2->lock); in udp_lib_rehash()
2251 hlist_del_init_rcu(&udp_sk(sk)->udp_portaddr_node); in udp_lib_rehash()
2252 hslot2->count--; in udp_lib_rehash()
2253 spin_unlock(&hslot2->lock); in udp_lib_rehash()
2255 spin_lock(&nhslot2->lock); in udp_lib_rehash()
2256 hlist_add_head_rcu(&udp_sk(sk)->udp_portaddr_node, in udp_lib_rehash()
2257 &nhslot2->head); in udp_lib_rehash()
2258 nhslot2->count++; in udp_lib_rehash()
2259 spin_unlock(&nhslot2->lock); in udp_lib_rehash()
2262 spin_unlock_bh(&hslot->lock); in udp_lib_rehash()
2267 * (2) update hslot2->hash4_cnt. in udp_lib_rehash()
2272 spin_lock_bh(&hslot->lock); in udp_lib_rehash()
2276 spin_lock(&hslot2->lock); in udp_lib_rehash()
2278 spin_unlock(&hslot2->lock); in udp_lib_rehash()
2280 spin_lock(&nhslot2->lock); in udp_lib_rehash()
2282 spin_unlock(&nhslot2->lock); in udp_lib_rehash()
2285 spin_unlock_bh(&hslot->lock); in udp_lib_rehash()
2294 inet_sk(sk)->inet_rcv_saddr, in udp_v4_rehash()
2295 inet_sk(sk)->inet_num); in udp_v4_rehash()
2297 sk->sk_rcv_saddr, sk->sk_num, in udp_v4_rehash()
2298 sk->sk_daddr, sk->sk_dport); in udp_v4_rehash()
2307 if (inet_sk(sk)->inet_daddr) { in __udp_queue_rcv_skb()
2321 if (rc == -ENOMEM) { in __udp_queue_rcv_skb()
2333 return -1; in __udp_queue_rcv_skb()
2340 * -1: error
2363 READ_ONCE(up->encap_type)) { in udp_queue_rcv_one_skb()
2370 * up->encap_rcv() returns the following value: in udp_queue_rcv_one_skb()
2374 * <0 if skb should be resubmitted as proto -N in udp_queue_rcv_one_skb()
2378 encap_rcv = READ_ONCE(up->encap_rcv); in udp_queue_rcv_one_skb()
2391 return -ret; in udp_queue_rcv_one_skb()
2395 /* FALLTHROUGH -- it's a UDP Packet */ in udp_queue_rcv_one_skb()
2399 * UDP-Lite specific tests, ignored on UDP sockets in udp_queue_rcv_one_skb()
2401 if (udp_test_bit(UDPLITE_RECV_CC, sk) && UDP_SKB_CB(skb)->partial_cov) { in udp_queue_rcv_one_skb()
2402 u16 pcrlen = READ_ONCE(up->pcrlen); in udp_queue_rcv_one_skb()
2417 UDP_SKB_CB(skb)->cscov, skb->len); in udp_queue_rcv_one_skb()
2424 * Therefore the above ...()->partial_cov statement is essential. in udp_queue_rcv_one_skb()
2426 if (UDP_SKB_CB(skb)->cscov < pcrlen) { in udp_queue_rcv_one_skb()
2428 UDP_SKB_CB(skb)->cscov, pcrlen); in udp_queue_rcv_one_skb()
2433 prefetch(&sk->sk_rmem_alloc); in udp_queue_rcv_one_skb()
2434 if (rcu_access_pointer(sk->sk_filter) && in udp_queue_rcv_one_skb()
2453 atomic_inc(&sk->sk_drops); in udp_queue_rcv_one_skb()
2455 return -1; in udp_queue_rcv_one_skb()
2467 __skb_push(skb, -skb_mac_offset(skb)); in udp_queue_rcv_skb()
2475 ip_protocol_deliver_rcu(dev_net(skb->dev), skb, ret); in udp_queue_rcv_skb()
2488 old = unrcu_pointer(xchg(&sk->sk_rx_dst, RCU_INITIALIZER(dst))); in udp_sk_rx_dst_set()
2508 unsigned short hnum = ntohs(uh->dest); in __udp4_lib_mcast_deliver()
2510 unsigned int hash2 = 0, hash2_any = 0, use_hash2 = (hslot->count > 10); in __udp4_lib_mcast_deliver()
2512 int dif = skb->dev->ifindex; in __udp4_lib_mcast_deliver()
2519 udptable->mask; in __udp4_lib_mcast_deliver()
2520 hash2 = ipv4_portaddr_hash(net, daddr, hnum) & udptable->mask; in __udp4_lib_mcast_deliver()
2522 hslot = &udptable->hash2[hash2].hslot; in __udp4_lib_mcast_deliver()
2526 sk_for_each_entry_offset_rcu(sk, node, &hslot->head, offset) { in __udp4_lib_mcast_deliver()
2527 if (!__udp_is_mcast_sock(net, sk, uh->dest, daddr, in __udp4_lib_mcast_deliver()
2528 uh->source, saddr, dif, sdif, hnum)) in __udp4_lib_mcast_deliver()
2538 atomic_inc(&sk->sk_drops); in __udp4_lib_mcast_deliver()
2569 * including udp header and folding it to skb->csum.
2576 UDP_SKB_CB(skb)->partial_cov = 0; in udp4_csum_init()
2577 UDP_SKB_CB(skb)->cscov = skb->len; in udp4_csum_init()
2584 if (UDP_SKB_CB(skb)->partial_cov) { in udp4_csum_init()
2585 skb->csum = inet_compute_pseudo(skb, proto); in udp4_csum_init()
2593 err = (__force int)skb_checksum_init_zero_check(skb, proto, uh->check, in udp4_csum_init()
2598 if (skb->ip_summed == CHECKSUM_COMPLETE && !skb->csum_valid) { in udp4_csum_init()
2600 if (skb->csum_complete_sw) in udp4_csum_init()
2604 * skb->csum is no longer the full packet checksum, in udp4_csum_init()
2621 if (inet_get_convert_csum(sk) && uh->check && !IS_UDPLITE(sk)) in udp_unicast_rcv_skb()
2627 * it wants the return to be -protocol, or 0 in udp_unicast_rcv_skb()
2630 return -ret; in udp_unicast_rcv_skb()
2646 struct net *net = dev_net(skb->dev); in __udp4_lib_rcv()
2659 ulen = ntohs(uh->len); in __udp4_lib_rcv()
2660 saddr = ip_hdr(skb)->saddr; in __udp4_lib_rcv()
2661 daddr = ip_hdr(skb)->daddr; in __udp4_lib_rcv()
2663 if (ulen > skb->len) in __udp4_lib_rcv()
2676 sk = inet_steal_sock(net, skb, sizeof(struct udphdr), saddr, uh->source, daddr, uh->dest, in __udp4_lib_rcv()
2685 if (unlikely(rcu_dereference(sk->sk_rx_dst) != dst)) in __udp4_lib_rcv()
2694 if (rt->rt_flags & (RTCF_BROADCAST|RTCF_MULTICAST)) in __udp4_lib_rcv()
2698 sk = __udp4_lib_lookup_skb(skb, uh->source, uh->dest, udptable); in __udp4_lib_rcv()
2725 &saddr, ntohs(uh->source), in __udp4_lib_rcv()
2726 ulen, skb->len, in __udp4_lib_rcv()
2727 &daddr, ntohs(uh->dest)); in __udp4_lib_rcv()
2738 &saddr, ntohs(uh->source), &daddr, ntohs(uh->dest), in __udp4_lib_rcv()
2755 struct udp_table *udptable = net->ipv4.udp_table; in __udp4_lib_mcast_demux_lookup()
2761 slot = udp_hashfn(net, hnum, udptable->mask); in __udp4_lib_mcast_demux_lookup()
2762 hslot = &udptable->hash[slot]; in __udp4_lib_mcast_demux_lookup()
2765 if (hslot->count > 10) in __udp4_lib_mcast_demux_lookup()
2769 sk_for_each_rcu(sk, &hslot->head) { in __udp4_lib_mcast_demux_lookup()
2790 struct udp_table *udptable = net->ipv4.udp_table; in __udp4_lib_demux_lookup()
2802 udp_portaddr_for_each_entry_rcu(sk, &hslot2->head) { in __udp4_lib_demux_lookup()
2813 struct net *net = dev_net(skb->dev); in udp_v4_early_demux()
2819 int dif = skb->dev->ifindex; in udp_v4_early_demux()
2830 if (skb->pkt_type == PACKET_MULTICAST) { in udp_v4_early_demux()
2831 in_dev = __in_dev_get_rcu(skb->dev); in udp_v4_early_demux()
2836 ours = ip_check_mc_rcu(in_dev, iph->daddr, iph->saddr, in udp_v4_early_demux()
2837 iph->protocol); in udp_v4_early_demux()
2841 sk = __udp4_lib_mcast_demux_lookup(net, uh->dest, iph->daddr, in udp_v4_early_demux()
2842 uh->source, iph->saddr, in udp_v4_early_demux()
2844 } else if (skb->pkt_type == PACKET_HOST) { in udp_v4_early_demux()
2845 sk = __udp4_lib_demux_lookup(net, uh->dest, iph->daddr, in udp_v4_early_demux()
2846 uh->source, iph->saddr, dif, sdif); in udp_v4_early_demux()
2852 skb->sk = sk; in udp_v4_early_demux()
2854 skb->destructor = sock_pfree; in udp_v4_early_demux()
2855 dst = rcu_dereference(sk->sk_rx_dst); in udp_v4_early_demux()
2871 if (!inet_sk(sk)->inet_daddr && in_dev) in udp_v4_early_demux()
2872 return ip_mc_validate_source(skb, iph->daddr, in udp_v4_early_demux()
2873 iph->saddr, in udp_v4_early_demux()
2875 skb->dev, in_dev, &itag); in udp_v4_early_demux()
2882 return __udp4_lib_rcv(skb, dev_net(skb->dev)->ipv4.udp_table, IPPROTO_UDP); in udp_rcv()
2895 if (up->encap_type) { in udp_destroy_sock()
2897 encap_destroy = READ_ONCE(up->encap_destroy); in udp_destroy_sock()
2912 WRITE_ONCE(udp_sk(sk)->gro_receive, xfrm4_gro_udp_encap_rcv); in set_xfrm_gro_udp_encap_rcv()
2914 WRITE_ONCE(udp_sk(sk)->gro_receive, ipv6_stub->xfrm6_gro_udp_encap_rcv); in set_xfrm_gro_udp_encap_rcv()
2937 WRITE_ONCE(up->forward_threshold, sk->sk_rcvbuf >> 2); in udp_lib_setsockopt()
2944 return -EINVAL; in udp_lib_setsockopt()
2947 return -EFAULT; in udp_lib_setsockopt()
2968 set_xfrm_gro_udp_encap_rcv(val, sk->sk_family, sk); in udp_lib_setsockopt()
2970 if (sk->sk_family == AF_INET6) in udp_lib_setsockopt()
2971 WRITE_ONCE(up->encap_rcv, in udp_lib_setsockopt()
2972 ipv6_stub->xfrm6_udp_encap_rcv); in udp_lib_setsockopt()
2975 WRITE_ONCE(up->encap_rcv, in udp_lib_setsockopt()
2980 WRITE_ONCE(up->encap_type, val); in udp_lib_setsockopt()
2984 err = -ENOPROTOOPT; in udp_lib_setsockopt()
2999 return -EINVAL; in udp_lib_setsockopt()
3000 WRITE_ONCE(up->gso_size, val); in udp_lib_setsockopt()
3010 set_xfrm_gro_udp_encap_rcv(up->encap_type, sk->sk_family, sk); in udp_lib_setsockopt()
3014 * UDP-Lite's partial checksum coverage (RFC 3828). in udp_lib_setsockopt()
3020 return -ENOPROTOOPT; in udp_lib_setsockopt()
3025 WRITE_ONCE(up->pcslen, val); in udp_lib_setsockopt()
3034 return -ENOPROTOOPT; in udp_lib_setsockopt()
3039 WRITE_ONCE(up->pcrlen, val); in udp_lib_setsockopt()
3044 err = -ENOPROTOOPT; in udp_lib_setsockopt()
3069 return -EFAULT; in udp_lib_getsockopt()
3072 return -EINVAL; in udp_lib_getsockopt()
3082 val = READ_ONCE(up->encap_type); in udp_lib_getsockopt()
3094 val = READ_ONCE(up->gso_size); in udp_lib_getsockopt()
3104 val = READ_ONCE(up->pcslen); in udp_lib_getsockopt()
3108 val = READ_ONCE(up->pcrlen); in udp_lib_getsockopt()
3112 return -ENOPROTOOPT; in udp_lib_getsockopt()
3116 return -EFAULT; in udp_lib_getsockopt()
3118 return -EFAULT; in udp_lib_getsockopt()
3132 * udp_poll - wait for a UDP event.
3133 * @file: - file struct
3134 * @sock: - socket
3135 * @wait: - poll table
3142 * to work around these arguably broken applications.
3147 struct sock *sk = sock->sk; in udp_poll()
3149 if (!skb_queue_empty_lockless(&udp_sk(sk)->reader_queue)) in udp_poll()
3153 if ((mask & EPOLLRDNORM) && !(file->f_flags & O_NONBLOCK) && in udp_poll()
3154 !(sk->sk_shutdown & RCV_SHUTDOWN) && first_packet_length(sk) == -1) in udp_poll()
3176 sk->sk_err = err; in udp_abort()
3224 /* ------------------------------------------------------------------------ */
3233 return ((family == AF_UNSPEC || family == sk->sk_family) && in seq_sk_match()
3246 if (seq->op == &bpf_iter_udp_seq_ops) in udp_get_table_seq()
3247 return net->ipv4.udp_table; in udp_get_table_seq()
3250 afinfo = pde_data(file_inode(seq->file)); in udp_get_table_seq()
3251 return afinfo->udp_table ? : net->ipv4.udp_table; in udp_get_table_seq()
3256 struct udp_iter_state *state = seq->private; in udp_get_first()
3263 for (state->bucket = start; state->bucket <= udptable->mask; in udp_get_first()
3264 ++state->bucket) { in udp_get_first()
3265 struct udp_hslot *hslot = &udptable->hash[state->bucket]; in udp_get_first()
3267 if (hlist_empty(&hslot->head)) in udp_get_first()
3270 spin_lock_bh(&hslot->lock); in udp_get_first()
3271 sk_for_each(sk, &hslot->head) { in udp_get_first()
3275 spin_unlock_bh(&hslot->lock); in udp_get_first()
3284 struct udp_iter_state *state = seq->private; in udp_get_next()
3295 if (state->bucket <= udptable->mask) in udp_get_next()
3296 spin_unlock_bh(&udptable->hash[state->bucket].lock); in udp_get_next()
3298 return udp_get_first(seq, state->bucket + 1); in udp_get_next()
3309 --pos; in udp_get_idx()
3315 struct udp_iter_state *state = seq->private; in udp_seq_start()
3316 state->bucket = MAX_UDP_PORTS; in udp_seq_start()
3318 return *pos ? udp_get_idx(seq, *pos-1) : SEQ_START_TOKEN; in udp_seq_start()
3338 struct udp_iter_state *state = seq->private; in udp_seq_stop()
3343 if (state->bucket <= udptable->mask) in udp_seq_stop()
3344 spin_unlock_bh(&udptable->hash[state->bucket].lock); in udp_seq_stop()
3348 /* ------------------------------------------------------------------------ */
3353 __be32 dest = inet->inet_daddr; in udp4_format_sock()
3354 __be32 src = inet->inet_rcv_saddr; in udp4_format_sock()
3355 __u16 destp = ntohs(inet->inet_dport); in udp4_format_sock()
3356 __u16 srcp = ntohs(inet->inet_sport); in udp4_format_sock()
3360 bucket, src, srcp, dest, destp, sp->sk_state, in udp4_format_sock()
3366 refcount_read(&sp->sk_refcnt), sp, in udp4_format_sock()
3367 atomic_read(&sp->sk_drops)); in udp4_format_sock()
3375 "rx_queue tr tm->when retrnsmt uid timeout " in udp4_seq_show()
3378 struct udp_iter_state *state = seq->private; in udp4_seq_show()
3380 udp4_format_sock(v, seq, state->bucket); in udp4_seq_show()
3408 struct bpf_udp_iter_state *iter = seq->private; in bpf_iter_udp_batch()
3409 struct udp_iter_state *state = &iter->state; in bpf_iter_udp_batch()
3417 resume_bucket = state->bucket; in bpf_iter_udp_batch()
3418 resume_offset = iter->offset; in bpf_iter_udp_batch()
3421 if (iter->st_bucket_done) in bpf_iter_udp_batch()
3422 state->bucket++; in bpf_iter_udp_batch()
3434 iter->cur_sk = 0; in bpf_iter_udp_batch()
3435 iter->end_sk = 0; in bpf_iter_udp_batch()
3436 iter->st_bucket_done = false; in bpf_iter_udp_batch()
3439 for (; state->bucket <= udptable->mask; state->bucket++) { in bpf_iter_udp_batch()
3440 struct udp_hslot *hslot2 = &udptable->hash2[state->bucket].hslot; in bpf_iter_udp_batch()
3442 if (hlist_empty(&hslot2->head)) in bpf_iter_udp_batch()
3445 iter->offset = 0; in bpf_iter_udp_batch()
3446 spin_lock_bh(&hslot2->lock); in bpf_iter_udp_batch()
3447 udp_portaddr_for_each_entry(sk, &hslot2->head) { in bpf_iter_udp_batch()
3452 if (state->bucket == resume_bucket && in bpf_iter_udp_batch()
3453 iter->offset < resume_offset) { in bpf_iter_udp_batch()
3454 ++iter->offset; in bpf_iter_udp_batch()
3457 if (iter->end_sk < iter->max_sk) { in bpf_iter_udp_batch()
3459 iter->batch[iter->end_sk++] = sk; in bpf_iter_udp_batch()
3464 spin_unlock_bh(&hslot2->lock); in bpf_iter_udp_batch()
3466 if (iter->end_sk) in bpf_iter_udp_batch()
3471 if (!iter->end_sk) in bpf_iter_udp_batch()
3474 if (iter->end_sk == batch_sks) { in bpf_iter_udp_batch()
3478 iter->st_bucket_done = true; in bpf_iter_udp_batch()
3489 return iter->batch[0]; in bpf_iter_udp_batch()
3494 struct bpf_udp_iter_state *iter = seq->private; in bpf_iter_udp_seq_next()
3497 /* Whenever seq_next() is called, the iter->cur_sk is in bpf_iter_udp_seq_next()
3498 * done with seq_show(), so unref the iter->cur_sk. in bpf_iter_udp_seq_next()
3500 if (iter->cur_sk < iter->end_sk) { in bpf_iter_udp_seq_next()
3501 sock_put(iter->batch[iter->cur_sk++]); in bpf_iter_udp_seq_next()
3502 ++iter->offset; in bpf_iter_udp_seq_next()
3505 /* After updating iter->cur_sk, check if there are more sockets in bpf_iter_udp_seq_next()
3508 if (iter->cur_sk < iter->end_sk) in bpf_iter_udp_seq_next()
3509 sk = iter->batch[iter->cur_sk]; in bpf_iter_udp_seq_next()
3521 * continue from where it was stop()-ped. in bpf_iter_udp_seq_start()
3534 meta->seq_num--; /* skip SEQ_START_TOKEN */ in udp_prog_seq_show()
3544 struct udp_iter_state *state = seq->private; in bpf_iter_udp_seq_show()
3564 ret = udp_prog_seq_show(prog, &meta, v, uid, state->bucket); in bpf_iter_udp_seq_show()
3573 while (iter->cur_sk < iter->end_sk) in bpf_iter_udp_put_batch()
3574 sock_put(iter->batch[iter->cur_sk++]); in bpf_iter_udp_put_batch()
3579 struct bpf_udp_iter_state *iter = seq->private; in bpf_iter_udp_seq_stop()
3590 if (iter->cur_sk < iter->end_sk) { in bpf_iter_udp_seq_stop()
3592 iter->st_bucket_done = false; in bpf_iter_udp_seq_stop()
3610 if (seq->op == &bpf_iter_udp_seq_ops) in seq_file_family()
3615 afinfo = pde_data(file_inode(seq->file)); in seq_file_family()
3616 return afinfo->family; in seq_file_family()
3634 if (!proc_create_net_data("udp", 0444, net->proc_net, &udp_seq_ops, in udp4_proc_init_net()
3636 return -ENOMEM; in udp4_proc_init_net()
3642 remove_proc_entry("udp", net->proc_net); in udp4_proc_exit_net()
3685 table->hash = alloc_large_system_hash(name, in udp_table_init()
3690 &table->log, in udp_table_init()
3691 &table->mask, in udp_table_init()
3695 table->hash2 = (void *)(table->hash + (table->mask + 1)); in udp_table_init()
3696 for (i = 0; i <= table->mask; i++) { in udp_table_init()
3697 INIT_HLIST_HEAD(&table->hash[i].head); in udp_table_init()
3698 table->hash[i].count = 0; in udp_table_init()
3699 spin_lock_init(&table->hash[i].lock); in udp_table_init()
3701 for (i = 0; i <= table->mask; i++) { in udp_table_init()
3702 INIT_HLIST_HEAD(&table->hash2[i].hslot.head); in udp_table_init()
3703 table->hash2[i].hslot.count = 0; in udp_table_init()
3704 spin_lock_init(&table->hash2[i].hslot.lock); in udp_table_init()
3721 net->ipv4.sysctl_udp_rmem_min = PAGE_SIZE; in udp_sysctl_init()
3722 net->ipv4.sysctl_udp_wmem_min = PAGE_SIZE; in udp_sysctl_init()
3725 net->ipv4.sysctl_udp_l3mdev_accept = 0; in udp_sysctl_init()
3741 udptable->hash = vmalloc_huge(hash_entries * slot_size, in udp_pernet_table_alloc()
3743 if (!udptable->hash) in udp_pernet_table_alloc()
3746 udptable->hash2 = (void *)(udptable->hash + hash_entries); in udp_pernet_table_alloc()
3747 udptable->mask = hash_entries - 1; in udp_pernet_table_alloc()
3748 udptable->log = ilog2(hash_entries); in udp_pernet_table_alloc()
3751 INIT_HLIST_HEAD(&udptable->hash[i].head); in udp_pernet_table_alloc()
3752 udptable->hash[i].count = 0; in udp_pernet_table_alloc()
3753 spin_lock_init(&udptable->hash[i].lock); in udp_pernet_table_alloc()
3755 INIT_HLIST_HEAD(&udptable->hash2[i].hslot.head); in udp_pernet_table_alloc()
3756 udptable->hash2[i].hslot.count = 0; in udp_pernet_table_alloc()
3757 spin_lock_init(&udptable->hash2[i].hslot.lock); in udp_pernet_table_alloc()
3771 struct udp_table *udptable = net->ipv4.udp_table; in udp_pernet_table_free()
3776 kvfree(udptable->hash); in udp_pernet_table_free()
3789 old_net = current->nsproxy->net_ns; in udp_set_table()
3790 hash_entries = READ_ONCE(old_net->ipv4.sysctl_udp_child_hash_entries); in udp_set_table()
3802 net->ipv4.udp_table = udptable; in udp_set_table()
3808 net->ipv4.udp_table = &udp_table; in udp_set_table()
3842 return -ENOMEM; in DEFINE_BPF_ITER_FUNC()
3845 kvfree(iter->batch); in DEFINE_BPF_ITER_FUNC()
3846 iter->batch = new_batch; in DEFINE_BPF_ITER_FUNC()
3847 iter->max_sk = new_batch_sz; in DEFINE_BPF_ITER_FUNC()
3875 kvfree(iter->batch); in bpf_iter_fini_udp()