]> git.saurik.com Git - apple/xnu.git/blame - bsd/netinet/udp_usrreq.c
xnu-6153.41.3.tar.gz
[apple/xnu.git] / bsd / netinet / udp_usrreq.c
CommitLineData
1c79356b 1/*
0a7de745 2 * Copyright (c) 2000-2019 Apple Inc. All rights reserved.
5d5c5d0d 3 *
2d21ac55 4 * @APPLE_OSREFERENCE_LICENSE_HEADER_START@
39037602 5 *
2d21ac55
A
6 * This file contains Original Code and/or Modifications of Original Code
7 * as defined in and that are subject to the Apple Public Source License
8 * Version 2.0 (the 'License'). You may not use this file except in
9 * compliance with the License. The rights granted to you under the License
10 * may not be used to create, or enable the creation or redistribution of,
11 * unlawful or unlicensed copies of an Apple operating system, or to
12 * circumvent, violate, or enable the circumvention or violation of, any
13 * terms of an Apple operating system software license agreement.
39037602 14 *
2d21ac55
A
15 * Please obtain a copy of the License at
16 * http://www.opensource.apple.com/apsl/ and read it before using this file.
39037602 17 *
2d21ac55
A
18 * The Original Code and all software distributed under the License are
19 * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
8f6c56a5
A
20 * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
21 * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
2d21ac55
A
22 * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
23 * Please see the License for the specific language governing rights and
24 * limitations under the License.
39037602 25 *
2d21ac55 26 * @APPLE_OSREFERENCE_LICENSE_HEADER_END@
1c79356b
A
27 */
28/*
29 * Copyright (c) 1982, 1986, 1988, 1990, 1993, 1995
30 * The Regents of the University of California. All rights reserved.
31 *
32 * Redistribution and use in source and binary forms, with or without
33 * modification, are permitted provided that the following conditions
34 * are met:
35 * 1. Redistributions of source code must retain the above copyright
36 * notice, this list of conditions and the following disclaimer.
37 * 2. Redistributions in binary form must reproduce the above copyright
38 * notice, this list of conditions and the following disclaimer in the
39 * documentation and/or other materials provided with the distribution.
40 * 3. All advertising materials mentioning features or use of this software
41 * must display the following acknowledgement:
42 * This product includes software developed by the University of
43 * California, Berkeley and its contributors.
44 * 4. Neither the name of the University nor the names of its contributors
45 * may be used to endorse or promote products derived from this software
46 * without specific prior written permission.
47 *
48 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
49 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
50 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
51 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
52 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
53 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
54 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
55 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
56 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
57 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
58 * SUCH DAMAGE.
59 *
60 * @(#)udp_usrreq.c 8.6 (Berkeley) 5/23/95
61 */
62
63#include <sys/param.h>
64#include <sys/systm.h>
65#include <sys/kernel.h>
66#include <sys/malloc.h>
67#include <sys/mbuf.h>
1c79356b 68#include <sys/domain.h>
1c79356b
A
69#include <sys/protosw.h>
70#include <sys/socket.h>
71#include <sys/socketvar.h>
72#include <sys/sysctl.h>
73#include <sys/syslog.h>
6d2010ae
A
74#include <sys/mcache.h>
75#include <net/ntstat.h>
1c79356b 76
2d21ac55 77#include <kern/zalloc.h>
39236c6e 78#include <mach/boolean.h>
2d21ac55 79
1c79356b 80#include <net/if.h>
91447636 81#include <net/if_types.h>
1c79356b 82#include <net/route.h>
39236c6e 83#include <net/dlil.h>
5ba3f43e 84#include <net/net_api_stats.h>
1c79356b
A
85
86#include <netinet/in.h>
87#include <netinet/in_systm.h>
39037602 88#include <netinet/in_tclass.h>
1c79356b 89#include <netinet/ip.h>
9bccf70c
A
90#if INET6
91#include <netinet/ip6.h>
39236c6e 92#endif /* INET6 */
1c79356b
A
93#include <netinet/in_pcb.h>
94#include <netinet/in_var.h>
95#include <netinet/ip_var.h>
1c79356b 96#if INET6
b0d623f7 97#include <netinet6/in6_pcb.h>
1c79356b 98#include <netinet6/ip6_var.h>
39236c6e
A
99#include <netinet6/udp6_var.h>
100#endif /* INET6 */
9bccf70c
A
101#include <netinet/ip_icmp.h>
102#include <netinet/icmp_var.h>
1c79356b
A
103#include <netinet/udp.h>
104#include <netinet/udp_var.h>
105#include <sys/kdebug.h>
106
107#if IPSEC
108#include <netinet6/ipsec.h>
2d21ac55 109#include <netinet6/esp.h>
94ff46dc 110#include <netkey/key.h>
9bccf70c 111extern int ipsec_bypass;
39236c6e
A
112extern int esp_udp_encap_port;
113#endif /* IPSEC */
1c79356b 114
fe8ab488
A
115#if NECP
116#include <net/necp.h>
117#endif /* NECP */
118
3e170ce0
A
119#if FLOW_DIVERT
120#include <netinet/flow_divert.h>
121#endif /* FLOW_DIVERT */
122
d9a64523
A
123#if CONTENT_FILTER
124#include <net/content_filter.h>
125#endif /* CONTENT_FILTER */
126
0a7de745
A
127#define DBG_LAYER_IN_BEG NETDBG_CODE(DBG_NETUDP, 0)
128#define DBG_LAYER_IN_END NETDBG_CODE(DBG_NETUDP, 2)
129#define DBG_LAYER_OUT_BEG NETDBG_CODE(DBG_NETUDP, 1)
130#define DBG_LAYER_OUT_END NETDBG_CODE(DBG_NETUDP, 3)
131#define DBG_FNC_UDP_INPUT NETDBG_CODE(DBG_NETUDP, (5 << 8))
132#define DBG_FNC_UDP_OUTPUT NETDBG_CODE(DBG_NETUDP, (6 << 8) | 1)
1c79356b 133
1c79356b
A
134/*
135 * UDP protocol implementation.
136 * Per RFC 768, August, 1980.
137 */
0a7de745 138#ifndef COMPAT_42
39236c6e 139static int udpcksum = 1;
1c79356b 140#else
0a7de745 141static int udpcksum = 0; /* XXX */
1c79356b 142#endif
39236c6e
A
143SYSCTL_INT(_net_inet_udp, UDPCTL_CHECKSUM, checksum,
144 CTLFLAG_RW | CTLFLAG_LOCKED, &udpcksum, 0, "");
145
146int udp_log_in_vain = 0;
6d2010ae 147SYSCTL_INT(_net_inet_udp, OID_AUTO, log_in_vain, CTLFLAG_RW | CTLFLAG_LOCKED,
0a7de745 148 &udp_log_in_vain, 0, "Log all incoming UDP packets");
9bccf70c 149
39236c6e 150static int blackhole = 0;
6d2010ae 151SYSCTL_INT(_net_inet_udp, OID_AUTO, blackhole, CTLFLAG_RW | CTLFLAG_LOCKED,
0a7de745 152 &blackhole, 0, "Do not send port unreachables for refused connects");
1c79356b 153
0a7de745
A
154struct inpcbhead udb; /* from udp_var.h */
155#define udb6 udb /* for KAME src sync over BSD*'s */
39236c6e 156struct inpcbinfo udbinfo;
1c79356b
A
157
158#ifndef UDBHASHSIZE
0a7de745 159#define UDBHASHSIZE 16
1c79356b
A
160#endif
161
39236c6e
A
162/* Garbage collection performed during most recent udp_gc() run */
163static boolean_t udp_gc_done = FALSE;
91447636 164
2d21ac55 165#if IPFIREWALL
39236c6e 166extern int fw_verbose;
39037602 167extern void ipfwsyslog(int level, const char *format, ...);
39236c6e
A
168extern void ipfw_stealth_stats_incr_udp(void);
169
170/* Apple logging, log to ipfw.log */
0a7de745
A
171#define log_in_vain_log(a) { \
172 if ((udp_log_in_vain == 3) && (fw_verbose == 2)) { \
173 ipfwsyslog a; \
39236c6e 174 } else if ((udp_log_in_vain == 4) && (fw_verbose == 2)) { \
0a7de745
A
175 ipfw_stealth_stats_incr_udp(); \
176 } else { \
177 log a; \
178 } \
91447636 179}
39236c6e 180#else /* !IPFIREWALL */
0a7de745 181#define log_in_vain_log(a) { log a; }
39236c6e 182#endif /* !IPFIREWALL */
91447636 183
39236c6e 184static int udp_getstat SYSCTL_HANDLER_ARGS;
0a7de745 185struct udpstat udpstat; /* from udp_var.h */
fe8ab488 186SYSCTL_PROC(_net_inet_udp, UDPCTL_STATS, stats,
0a7de745
A
187 CTLTYPE_STRUCT | CTLFLAG_RD | CTLFLAG_LOCKED,
188 0, 0, udp_getstat, "S,udpstat",
189 "UDP statistics (struct udpstat, netinet/udp_var.h)");
39236c6e
A
190
191SYSCTL_INT(_net_inet_udp, OID_AUTO, pcbcount,
0a7de745
A
192 CTLFLAG_RD | CTLFLAG_LOCKED, &udbinfo.ipi_count, 0,
193 "Number of active PCBs");
1c79356b 194
b0d623f7 195__private_extern__ int udp_use_randomport = 1;
39236c6e 196SYSCTL_INT(_net_inet_udp, OID_AUTO, randomize_ports,
0a7de745
A
197 CTLFLAG_RW | CTLFLAG_LOCKED, &udp_use_randomport, 0,
198 "Randomize UDP port numbers");
b0d623f7 199
1c79356b
A
200#if INET6
201struct udp_in6 {
0a7de745
A
202 struct sockaddr_in6 uin6_sin;
203 u_char uin6_init_done : 1;
1c79356b
A
204};
205struct udp_ip6 {
0a7de745
A
206 struct ip6_hdr uip6_ip6;
207 u_char uip6_init_done : 1;
2d21ac55 208};
39236c6e 209
5ba3f43e
A
210int udp_abort(struct socket *);
211int udp_attach(struct socket *, int, struct proc *);
212int udp_bind(struct socket *, struct sockaddr *, struct proc *);
213int udp_connect(struct socket *, struct sockaddr *, struct proc *);
214int udp_connectx(struct socket *, struct sockaddr *,
813fb2f6 215 struct sockaddr *, struct proc *, uint32_t, sae_associd_t,
3e170ce0 216 sae_connid_t *, uint32_t, void *, uint32_t, struct uio *, user_ssize_t *);
5ba3f43e
A
217int udp_detach(struct socket *);
218int udp_disconnect(struct socket *);
219int udp_disconnectx(struct socket *, sae_associd_t, sae_connid_t);
220int udp_send(struct socket *, int, struct mbuf *, struct sockaddr *,
39236c6e
A
221 struct mbuf *, struct proc *);
222static void udp_append(struct inpcb *, struct ip *, struct mbuf *, int,
223 struct sockaddr_in *, struct udp_in6 *, struct udp_ip6 *, struct ifnet *);
224#else /* !INET6 */
225static void udp_append(struct inpcb *, struct ip *, struct mbuf *, int,
226 struct sockaddr_in *, struct ifnet *);
227#endif /* !INET6 */
228static int udp_input_checksum(struct mbuf *, struct udphdr *, int, int);
5ba3f43e 229int udp_output(struct inpcb *, struct mbuf *, struct sockaddr *,
39236c6e 230 struct mbuf *, struct proc *);
91447636 231static void ip_2_ip6_hdr(struct ip6_hdr *ip6, struct ip *ip);
39236c6e 232static void udp_gc(struct inpcbinfo *);
1c79356b 233
39236c6e 234struct pr_usrreqs udp_usrreqs = {
0a7de745
A
235 .pru_abort = udp_abort,
236 .pru_attach = udp_attach,
237 .pru_bind = udp_bind,
238 .pru_connect = udp_connect,
239 .pru_connectx = udp_connectx,
240 .pru_control = in_control,
241 .pru_detach = udp_detach,
242 .pru_disconnect = udp_disconnect,
243 .pru_disconnectx = udp_disconnectx,
244 .pru_peeraddr = in_getpeeraddr,
245 .pru_send = udp_send,
246 .pru_shutdown = udp_shutdown,
247 .pru_sockaddr = in_getsockaddr,
248 .pru_sosend = sosend,
249 .pru_soreceive = soreceive,
250 .pru_soreceive_list = soreceive_list,
39236c6e 251};
1c79356b
A
252
253void
39236c6e 254udp_init(struct protosw *pp, struct domain *dp)
1c79356b 255{
39236c6e
A
256#pragma unused(dp)
257 static int udp_initialized = 0;
0a7de745
A
258 vm_size_t str_size;
259 struct inpcbinfo *pcbinfo;
39236c6e 260
0a7de745 261 VERIFY((pp->pr_flags & (PR_INITIALIZED | PR_ATTACHED)) == PR_ATTACHED);
39236c6e 262
0a7de745 263 if (udp_initialized) {
39236c6e 264 return;
0a7de745 265 }
39236c6e 266 udp_initialized = 1;
d9a64523
A
267 uint32_t pool_size = (nmbclusters << MCLSHIFT) >> MBSHIFT;
268 if (pool_size >= 96) {
269 /* Improves 10GbE UDP performance. */
270 udp_recvspace = 786896;
271 }
1c79356b 272 LIST_INIT(&udb);
39236c6e
A
273 udbinfo.ipi_listhead = &udb;
274 udbinfo.ipi_hashbase = hashinit(UDBHASHSIZE, M_PCB,
275 &udbinfo.ipi_hashmask);
276 udbinfo.ipi_porthashbase = hashinit(UDBHASHSIZE, M_PCB,
277 &udbinfo.ipi_porthashmask);
0a7de745
A
278 str_size = (vm_size_t) sizeof(struct inpcb);
279 udbinfo.ipi_zone = zinit(str_size, 80000 * str_size, 8192, "udpcb");
39236c6e
A
280
281 pcbinfo = &udbinfo;
91447636
A
282 /*
283 * allocate lock group attribute and group for udp pcb mutexes
284 */
39236c6e
A
285 pcbinfo->ipi_lock_grp_attr = lck_grp_attr_alloc_init();
286 pcbinfo->ipi_lock_grp = lck_grp_alloc_init("udppcb",
287 pcbinfo->ipi_lock_grp_attr);
288 pcbinfo->ipi_lock_attr = lck_attr_alloc_init();
289 if ((pcbinfo->ipi_lock = lck_rw_alloc_init(pcbinfo->ipi_lock_grp,
290 pcbinfo->ipi_lock_attr)) == NULL) {
291 panic("%s: unable to allocate PCB lock\n", __func__);
292 /* NOTREACHED */
293 }
91447636 294
39236c6e
A
295 udbinfo.ipi_gc = udp_gc;
296 in_pcbinfo_attach(&udbinfo);
1c79356b
A
297}
298
299void
39236c6e 300udp_input(struct mbuf *m, int iphlen)
1c79356b 301{
39236c6e
A
302 struct ip *ip;
303 struct udphdr *uh;
304 struct inpcb *inp;
305 struct mbuf *opts = NULL;
6d2010ae 306 int len, isbroadcast;
1c79356b
A
307 struct ip save_ip;
308 struct sockaddr *append_sa;
91447636 309 struct inpcbinfo *pcbinfo = &udbinfo;
39236c6e 310 struct sockaddr_in udp_in;
6d2010ae
A
311 struct ip_moptions *imo = NULL;
312 int foundmembership = 0, ret = 0;
2d21ac55 313#if INET6
39236c6e 314 struct udp_in6 udp_in6;
2d21ac55
A
315 struct udp_ip6 udp_ip6;
316#endif /* INET6 */
39236c6e
A
317 struct ifnet *ifp = m->m_pkthdr.rcvif;
318 boolean_t cell = IFNET_IS_CELLULAR(ifp);
319 boolean_t wifi = (!cell && IFNET_IS_WIFI(ifp));
fe8ab488 320 boolean_t wired = (!wifi && IFNET_IS_WIRED(ifp));
39236c6e 321
0a7de745
A
322 bzero(&udp_in, sizeof(udp_in));
323 udp_in.sin_len = sizeof(struct sockaddr_in);
39236c6e
A
324 udp_in.sin_family = AF_INET;
325#if INET6
0a7de745
A
326 bzero(&udp_in6, sizeof(udp_in6));
327 udp_in6.uin6_sin.sin6_len = sizeof(struct sockaddr_in6);
39236c6e
A
328 udp_in6.uin6_sin.sin6_family = AF_INET6;
329#endif /* INET6 */
1c79356b
A
330
331 udpstat.udps_ipackets++;
1c79356b 332
39037602 333 KERNEL_DEBUG(DBG_FNC_UDP_INPUT | DBG_FUNC_START, 0, 0, 0, 0, 0);
1c79356b 334
316670eb
A
335 /* Expect 32-bit aligned data pointer on strict-align platforms */
336 MBUF_STRICT_DATA_ALIGNMENT_CHECK_32(m);
337
1c79356b
A
338 /*
339 * Strip IP options, if any; should skip this,
340 * make available to user, and use on returned packets,
341 * but we don't yet have a way to check the checksum
342 * with options still present.
343 */
0a7de745 344 if (iphlen > sizeof(struct ip)) {
5ba3f43e 345 ip_stripoptions(m);
0a7de745 346 iphlen = sizeof(struct ip);
1c79356b
A
347 }
348
349 /*
350 * Get IP and UDP header together in first mbuf.
351 */
352 ip = mtod(m, struct ip *);
0a7de745
A
353 if (m->m_len < iphlen + sizeof(struct udphdr)) {
354 m = m_pullup(m, iphlen + sizeof(struct udphdr));
39236c6e 355 if (m == NULL) {
1c79356b 356 udpstat.udps_hdrops++;
39236c6e 357 KERNEL_DEBUG(DBG_FNC_UDP_INPUT | DBG_FUNC_END,
39037602 358 0, 0, 0, 0, 0);
1c79356b
A
359 return;
360 }
361 ip = mtod(m, struct ip *);
362 }
316670eb 363 uh = (struct udphdr *)(void *)((caddr_t)ip + iphlen);
1c79356b 364
9bccf70c 365 /* destination port of 0 is illegal, based on RFC768. */
316670eb 366 if (uh->uh_dport == 0) {
39236c6e 367 IF_UDP_STATINC(ifp, port0);
9bccf70c 368 goto bad;
316670eb 369 }
9bccf70c 370
1c79356b 371 KERNEL_DEBUG(DBG_LAYER_IN_BEG, uh->uh_dport, uh->uh_sport,
39236c6e 372 ip->ip_src.s_addr, ip->ip_dst.s_addr, uh->uh_ulen);
1c79356b
A
373
374 /*
375 * Make mbuf data length reflect UDP length.
376 * If not enough data to reflect UDP length, drop.
377 */
378 len = ntohs((u_short)uh->uh_ulen);
379 if (ip->ip_len != len) {
0a7de745 380 if (len > ip->ip_len || len < sizeof(struct udphdr)) {
1c79356b 381 udpstat.udps_badlen++;
39236c6e 382 IF_UDP_STATINC(ifp, badlength);
1c79356b
A
383 goto bad;
384 }
385 m_adj(m, len - ip->ip_len);
386 /* ip->ip_len = len; */
387 }
388 /*
389 * Save a copy of the IP header in case we want restore it
390 * for sending an ICMP error message in response.
391 */
392 save_ip = *ip;
393
394 /*
395 * Checksum extended UDP header and data.
396 */
0a7de745 397 if (udp_input_checksum(m, uh, iphlen, len)) {
39236c6e 398 goto bad;
0a7de745 399 }
1c79356b 400
316670eb 401 isbroadcast = in_broadcast(ip->ip_dst, ifp);
6d2010ae
A
402
403 if (IN_MULTICAST(ntohl(ip->ip_dst.s_addr)) || isbroadcast) {
2d21ac55 404 int reuse_sock = 0, mcast_delivered = 0;
2d21ac55 405
39236c6e 406 lck_rw_lock_shared(pcbinfo->ipi_lock);
1c79356b
A
407 /*
408 * Deliver a multicast or broadcast datagram to *all* sockets
409 * for which the local and remote addresses and ports match
410 * those of the incoming datagram. This allows more than
411 * one process to receive multi/broadcasts on the same port.
412 * (This really ought to be done for unicast datagrams as
413 * well, but that would cause problems with existing
414 * applications that open both address-specific sockets and
415 * a wildcard socket listening to the same port -- they would
416 * end up receiving duplicates of every unicast datagram.
417 * Those applications open the multiple sockets to overcome an
418 * inadequacy of the UDP socket interface, but for backwards
419 * compatibility we avoid the problem here rather than
420 * fixing the interface. Maybe 4.5BSD will remedy this?)
421 */
422
423 /*
424 * Construct sockaddr format source address.
425 */
426 udp_in.sin_port = uh->uh_sport;
427 udp_in.sin_addr = ip->ip_src;
428 /*
429 * Locate pcb(s) for datagram.
430 * (Algorithm copied from raw_intr().)
431 */
1c79356b
A
432#if INET6
433 udp_in6.uin6_init_done = udp_ip6.uip6_init_done = 0;
39236c6e 434#endif /* INET6 */
1c79356b 435 LIST_FOREACH(inp, &udb, inp_list) {
39236c6e
A
436#if IPSEC
437 int skipit;
438#endif /* IPSEC */
439
0a7de745 440 if (inp->inp_socket == NULL) {
1c79356b 441 continue;
0a7de745 442 }
39236c6e
A
443 if (inp != sotoinpcb(inp->inp_socket)) {
444 panic("%s: bad so back ptr inp=%p\n",
445 __func__, inp);
446 /* NOTREACHED */
447 }
91447636 448#if INET6
0a7de745 449 if ((inp->inp_vflag & INP_IPV4) == 0) {
39037602 450 continue;
0a7de745 451 }
39236c6e 452#endif /* INET6 */
0a7de745 453 if (inp_restricted_recv(inp, ifp)) {
6d2010ae 454 continue;
0a7de745 455 }
6d2010ae 456
39236c6e
A
457 if ((inp->inp_moptions == NULL) &&
458 (ntohl(ip->ip_dst.s_addr) !=
0a7de745 459 INADDR_ALLHOSTS_GROUP) && (isbroadcast == 0)) {
39236c6e 460 continue;
0a7de745 461 }
2d21ac55 462
39236c6e 463 if (in_pcb_checkstate(inp, WNT_ACQUIRE, 0) ==
0a7de745 464 WNT_STOPUSING) {
91447636 465 continue;
0a7de745 466 }
91447636 467
39236c6e
A
468 udp_lock(inp->inp_socket, 1, 0);
469
470 if (in_pcb_checkstate(inp, WNT_RELEASE, 1) ==
471 WNT_STOPUSING) {
91447636 472 udp_unlock(inp->inp_socket, 1, 0);
1c79356b 473 continue;
91447636
A
474 }
475
476 if (inp->inp_lport != uh->uh_dport) {
477 udp_unlock(inp->inp_socket, 1, 0);
478 continue;
479 }
1c79356b
A
480 if (inp->inp_laddr.s_addr != INADDR_ANY) {
481 if (inp->inp_laddr.s_addr !=
91447636
A
482 ip->ip_dst.s_addr) {
483 udp_unlock(inp->inp_socket, 1, 0);
1c79356b 484 continue;
91447636 485 }
1c79356b
A
486 }
487 if (inp->inp_faddr.s_addr != INADDR_ANY) {
488 if (inp->inp_faddr.s_addr !=
489 ip->ip_src.s_addr ||
91447636
A
490 inp->inp_fport != uh->uh_sport) {
491 udp_unlock(inp->inp_socket, 1, 0);
1c79356b 492 continue;
91447636 493 }
1c79356b
A
494 }
495
39236c6e
A
496 if (isbroadcast == 0 && (ntohl(ip->ip_dst.s_addr) !=
497 INADDR_ALLHOSTS_GROUP)) {
498 struct sockaddr_in group;
499 int blocked;
500
501 if ((imo = inp->inp_moptions) == NULL) {
6d2010ae
A
502 udp_unlock(inp->inp_socket, 1, 0);
503 continue;
6d2010ae 504 }
39236c6e
A
505 IMO_LOCK(imo);
506
0a7de745
A
507 bzero(&group, sizeof(struct sockaddr_in));
508 group.sin_len = sizeof(struct sockaddr_in);
39236c6e
A
509 group.sin_family = AF_INET;
510 group.sin_addr = ip->ip_dst;
511
512 blocked = imo_multi_filter(imo, ifp,
5ba3f43e 513 &group, &udp_in);
0a7de745 514 if (blocked == MCAST_PASS) {
39236c6e 515 foundmembership = 1;
0a7de745 516 }
39236c6e
A
517
518 IMO_UNLOCK(imo);
519 if (!foundmembership) {
520 udp_unlock(inp->inp_socket, 1, 0);
521 if (blocked == MCAST_NOTSMEMBER ||
0a7de745 522 blocked == MCAST_MUTED) {
39236c6e 523 udpstat.udps_filtermcast++;
0a7de745 524 }
39236c6e
A
525 continue;
526 }
527 foundmembership = 0;
6d2010ae 528 }
39236c6e
A
529
530 reuse_sock = (inp->inp_socket->so_options &
0a7de745 531 (SO_REUSEPORT | SO_REUSEADDR));
39236c6e 532
fe8ab488 533#if NECP
39236c6e 534 skipit = 0;
39037602 535 if (!necp_socket_is_allowed_to_send_recv_v4(inp,
3e170ce0 536 uh->uh_dport, uh->uh_sport, &ip->ip_dst,
d9a64523 537 &ip->ip_src, ifp, NULL, NULL, NULL)) {
39236c6e
A
538 /* do not inject data to pcb */
539 skipit = 1;
540 }
541 if (skipit == 0)
fe8ab488 542#endif /* NECP */
39236c6e
A
543 {
544 struct mbuf *n = NULL;
b0d623f7 545
0a7de745 546 if (reuse_sock) {
39236c6e 547 n = m_copy(m, 0, M_COPYALL);
0a7de745 548 }
39236c6e
A
549#if INET6
550 udp_append(inp, ip, m,
0a7de745 551 iphlen + sizeof(struct udphdr),
39236c6e
A
552 &udp_in, &udp_in6, &udp_ip6, ifp);
553#else /* !INET6 */
554 udp_append(inp, ip, m,
0a7de745 555 iphlen + sizeof(struct udphdr),
39236c6e
A
556 &udp_in, ifp);
557#endif /* !INET6 */
558 mcast_delivered++;
559
560 m = n;
1c79356b 561 }
39236c6e
A
562 udp_unlock(inp->inp_socket, 1, 0);
563
1c79356b
A
564 /*
565 * Don't look for additional matches if this one does
566 * not have either the SO_REUSEPORT or SO_REUSEADDR
567 * socket options set. This heuristic avoids searching
568 * through all pcbs in the common case of a non-shared
2d21ac55 569 * port. It assumes that an application will never
1c79356b
A
570 * clear these options after setting them.
571 */
0a7de745 572 if (reuse_sock == 0 || m == NULL) {
1c79356b 573 break;
0a7de745 574 }
316670eb
A
575
576 /*
577 * Expect 32-bit aligned data pointer on strict-align
578 * platforms.
579 */
580 MBUF_STRICT_DATA_ALIGNMENT_CHECK_32(m);
b0d623f7
A
581 /*
582 * Recompute IP and UDP header pointers for new mbuf
583 */
584 ip = mtod(m, struct ip *);
316670eb 585 uh = (struct udphdr *)(void *)((caddr_t)ip + iphlen);
1c79356b 586 }
39236c6e 587 lck_rw_done(pcbinfo->ipi_lock);
1c79356b 588
2d21ac55 589 if (mcast_delivered == 0) {
1c79356b
A
590 /*
591 * No matching pcb found; discard datagram.
592 * (No need to send an ICMP Port Unreachable
593 * for a broadcast or multicast datgram.)
594 */
595 udpstat.udps_noportbcast++;
39236c6e 596 IF_UDP_STATINC(ifp, port_unreach);
1c79356b
A
597 goto bad;
598 }
2d21ac55 599
cb323159 600 /* free the extra copy of mbuf or skipped by IPsec */
0a7de745 601 if (m != NULL) {
2d21ac55 602 m_freem(m);
0a7de745 603 }
39037602 604 KERNEL_DEBUG(DBG_FNC_UDP_INPUT | DBG_FUNC_END, 0, 0, 0, 0, 0);
1c79356b
A
605 return;
606 }
55e303ae 607
91447636 608#if IPSEC
55e303ae
A
609 /*
610 * UDP to port 4500 with a payload where the first four bytes are
cb323159 611 * not zero is a UDP encapsulated IPsec packet. Packets where
55e303ae 612 * the payload is one byte and that byte is 0xFF are NAT keepalive
cb323159 613 * packets. Decapsulate the ESP packet and carry on with IPsec input
55e303ae
A
614 * or discard the NAT keep-alive.
615 */
616 if (ipsec_bypass == 0 && (esp_udp_encap_port & 0xFFFF) != 0 &&
cb323159
A
617 (uh->uh_dport == ntohs((u_short)esp_udp_encap_port) ||
618 uh->uh_sport == ntohs((u_short)esp_udp_encap_port))) {
94ff46dc
A
619 /*
620 * Check if ESP or keepalive:
621 * 1. If the destination port of the incoming packet is 4500.
622 * 2. If the source port of the incoming packet is 4500,
623 * then check the SADB to match IP address and port.
624 */
625 bool check_esp = true;
626 if (uh->uh_dport != ntohs((u_short)esp_udp_encap_port)) {
627 check_esp = key_checksa_present(AF_INET, (caddr_t)&ip->ip_dst,
628 (caddr_t)&ip->ip_src, uh->uh_dport,
629 uh->uh_sport);
630 }
631
632 if (check_esp) {
633 int payload_len = len - sizeof(struct udphdr) > 4 ? 4 :
634 len - sizeof(struct udphdr);
635
636 if (m->m_len < iphlen + sizeof(struct udphdr) + payload_len) {
637 if ((m = m_pullup(m, iphlen + sizeof(struct udphdr) +
638 payload_len)) == NULL) {
639 udpstat.udps_hdrops++;
640 KERNEL_DEBUG(DBG_FNC_UDP_INPUT | DBG_FUNC_END,
641 0, 0, 0, 0, 0);
642 return;
643 }
644 /*
645 * Expect 32-bit aligned data pointer on strict-align
646 * platforms.
647 */
648 MBUF_STRICT_DATA_ALIGNMENT_CHECK_32(m);
39236c6e 649
94ff46dc
A
650 ip = mtod(m, struct ip *);
651 uh = (struct udphdr *)(void *)((caddr_t)ip + iphlen);
652 }
653 /* Check for NAT keepalive packet */
654 if (payload_len == 1 && *(u_int8_t *)
655 ((caddr_t)uh + sizeof(struct udphdr)) == 0xFF) {
656 m_freem(m);
39236c6e 657 KERNEL_DEBUG(DBG_FNC_UDP_INPUT | DBG_FUNC_END,
39037602 658 0, 0, 0, 0, 0);
55e303ae 659 return;
94ff46dc
A
660 } else if (payload_len == 4 && *(u_int32_t *)(void *)
661 ((caddr_t)uh + sizeof(struct udphdr)) != 0) {
662 /* UDP encapsulated IPsec packet to pass through NAT */
663 KERNEL_DEBUG(DBG_FNC_UDP_INPUT | DBG_FUNC_END,
664 0, 0, 0, 0, 0);
665 /* preserve the udp header */
666 esp4_input(m, iphlen + sizeof(struct udphdr));
667 return;
55e303ae 668 }
55e303ae
A
669 }
670 }
39236c6e 671#endif /* IPSEC */
55e303ae 672
1c79356b
A
673 /*
674 * Locate pcb for datagram.
675 */
676 inp = in_pcblookup_hash(&udbinfo, ip->ip_src, uh->uh_sport,
316670eb 677 ip->ip_dst, uh->uh_dport, 1, ifp);
1c79356b 678 if (inp == NULL) {
39236c6e
A
679 IF_UDP_STATINC(ifp, port_unreach);
680
681 if (udp_log_in_vain) {
91447636
A
682 char buf[MAX_IPv4_STR_LEN];
683 char buf2[MAX_IPv4_STR_LEN];
684
685 /* check src and dst address */
39236c6e
A
686 if (udp_log_in_vain < 3) {
687 log(LOG_INFO, "Connection attempt to "
688 "UDP %s:%d from %s:%d\n", inet_ntop(AF_INET,
0a7de745
A
689 &ip->ip_dst, buf, sizeof(buf)),
690 ntohs(uh->uh_dport), inet_ntop(AF_INET,
691 &ip->ip_src, buf2, sizeof(buf2)),
692 ntohs(uh->uh_sport));
39236c6e
A
693 } else if (!(m->m_flags & (M_BCAST | M_MCAST)) &&
694 ip->ip_dst.s_addr != ip->ip_src.s_addr) {
91447636 695 log_in_vain_log((LOG_INFO,
39236c6e
A
696 "Stealth Mode connection attempt to "
697 "UDP %s:%d from %s:%d\n", inet_ntop(AF_INET,
0a7de745 698 &ip->ip_dst, buf, sizeof(buf)),
39236c6e 699 ntohs(uh->uh_dport), inet_ntop(AF_INET,
0a7de745 700 &ip->ip_src, buf2, sizeof(buf2)),
39236c6e
A
701 ntohs(uh->uh_sport)))
702 }
1c79356b
A
703 }
704 udpstat.udps_noport++;
705 if (m->m_flags & (M_BCAST | M_MCAST)) {
706 udpstat.udps_noportbcast++;
707 goto bad;
708 }
1c79356b 709#if ICMP_BANDLIM
0a7de745 710 if (badport_bandlim(BANDLIM_ICMP_UNREACH) < 0) {
1c79356b 711 goto bad;
0a7de745 712 }
39236c6e 713#endif /* ICMP_BANDLIM */
0a7de745
A
714 if (blackhole) {
715 if (ifp && ifp->if_type != IFT_LOOP) {
91447636 716 goto bad;
0a7de745
A
717 }
718 }
9bccf70c
A
719 *ip = save_ip;
720 ip->ip_len += iphlen;
1c79356b 721 icmp_error(m, ICMP_UNREACH, ICMP_UNREACH_PORT, 0, 0);
39037602 722 KERNEL_DEBUG(DBG_FNC_UDP_INPUT | DBG_FUNC_END, 0, 0, 0, 0, 0);
1c79356b
A
723 return;
724 }
91447636
A
725 udp_lock(inp->inp_socket, 1, 0);
726
727 if (in_pcb_checkstate(inp, WNT_RELEASE, 1) == WNT_STOPUSING) {
728 udp_unlock(inp->inp_socket, 1, 0);
39236c6e 729 IF_UDP_STATINC(ifp, cleanup);
1c79356b
A
730 goto bad;
731 }
fe8ab488 732#if NECP
3e170ce0 733 if (!necp_socket_is_allowed_to_send_recv_v4(inp, uh->uh_dport,
d9a64523 734 uh->uh_sport, &ip->ip_dst, &ip->ip_src, ifp, NULL, NULL, NULL)) {
fe8ab488
A
735 udp_unlock(inp->inp_socket, 1, 0);
736 IF_UDP_STATINC(ifp, badipsec);
737 goto bad;
91447636 738 }
fe8ab488 739#endif /* NECP */
1c79356b
A
740
741 /*
742 * Construct sockaddr format source address.
743 * Stuff source address and datagram in user buffer.
744 */
745 udp_in.sin_port = uh->uh_sport;
746 udp_in.sin_addr = ip->ip_src;
39236c6e
A
747 if ((inp->inp_flags & INP_CONTROLOPTS) != 0 ||
748 (inp->inp_socket->so_options & SO_TIMESTAMP) != 0 ||
d9a64523 749 (inp->inp_socket->so_options & SO_TIMESTAMP_MONOTONIC) != 0 ||
0a7de745 750 (inp->inp_socket->so_options & SO_TIMESTAMP_CONTINUOUS) != 0) {
1c79356b
A
751#if INET6
752 if (inp->inp_vflag & INP_IPV6) {
753 int savedflags;
754
755 ip_2_ip6_hdr(&udp_ip6.uip6_ip6, ip);
756 savedflags = inp->inp_flags;
757 inp->inp_flags &= ~INP_UNMAPPABLEOPTS;
6d2010ae 758 ret = ip6_savecontrol(inp, m, &opts);
1c79356b
A
759 inp->inp_flags = savedflags;
760 } else
39236c6e 761#endif /* INET6 */
6d2010ae
A
762 {
763 ret = ip_savecontrol(inp, &opts, ip, m);
764 }
765 if (ret != 0) {
766 udp_unlock(inp->inp_socket, 1, 0);
767 goto bad;
768 }
1c79356b 769 }
0a7de745 770 m_adj(m, iphlen + sizeof(struct udphdr));
1c79356b
A
771
772 KERNEL_DEBUG(DBG_LAYER_IN_END, uh->uh_dport, uh->uh_sport,
39236c6e 773 save_ip.ip_src.s_addr, save_ip.ip_dst.s_addr, uh->uh_ulen);
1c79356b
A
774
775#if INET6
776 if (inp->inp_vflag & INP_IPV6) {
777 in6_sin_2_v4mapsin6(&udp_in, &udp_in6.uin6_sin);
2d21ac55 778 append_sa = (struct sockaddr *)&udp_in6.uin6_sin;
1c79356b 779 } else
39236c6e
A
780#endif /* INET6 */
781 {
782 append_sa = (struct sockaddr *)&udp_in;
783 }
6d2010ae 784 if (nstat_collect) {
fe8ab488
A
785 INP_ADD_STAT(inp, cell, wifi, wired, rxpackets, 1);
786 INP_ADD_STAT(inp, cell, wifi, wired, rxbytes, m->m_pkthdr.len);
5ba3f43e 787 inp_set_activity_bitmap(inp);
6d2010ae
A
788 }
789 so_recv_data_stat(inp->inp_socket, m, 0);
39236c6e
A
790 if (sbappendaddr(&inp->inp_socket->so_rcv, append_sa,
791 m, opts, NULL) == 0) {
1c79356b 792 udpstat.udps_fullsock++;
6d2010ae 793 } else {
91447636
A
794 sorwakeup(inp->inp_socket);
795 }
796 udp_unlock(inp->inp_socket, 1, 0);
39037602 797 KERNEL_DEBUG(DBG_FNC_UDP_INPUT | DBG_FUNC_END, 0, 0, 0, 0, 0);
1c79356b
A
798 return;
799bad:
800 m_freem(m);
0a7de745 801 if (opts) {
1c79356b 802 m_freem(opts);
0a7de745 803 }
39037602 804 KERNEL_DEBUG(DBG_FNC_UDP_INPUT | DBG_FUNC_END, 0, 0, 0, 0, 0);
1c79356b
A
805}
806
807#if INET6
808static void
39236c6e 809ip_2_ip6_hdr(struct ip6_hdr *ip6, struct ip *ip)
1c79356b 810{
0a7de745 811 bzero(ip6, sizeof(*ip6));
1c79356b
A
812
813 ip6->ip6_vfc = IPV6_VERSION;
814 ip6->ip6_plen = ip->ip_len;
815 ip6->ip6_nxt = ip->ip_p;
816 ip6->ip6_hlim = ip->ip_ttl;
316670eb
A
817 if (ip->ip_src.s_addr) {
818 ip6->ip6_src.s6_addr32[2] = IPV6_ADDR_INT32_SMP;
819 ip6->ip6_src.s6_addr32[3] = ip->ip_src.s_addr;
820 }
821 if (ip->ip_dst.s_addr) {
822 ip6->ip6_dst.s6_addr32[2] = IPV6_ADDR_INT32_SMP;
823 ip6->ip6_dst.s6_addr32[3] = ip->ip_dst.s_addr;
824 }
1c79356b 825}
39236c6e 826#endif /* INET6 */
1c79356b
A
827
828/*
829 * subroutine of udp_input(), mainly for source code readability.
1c79356b
A
830 */
831static void
2d21ac55
A
832#if INET6
833udp_append(struct inpcb *last, struct ip *ip, struct mbuf *n, int off,
834 struct sockaddr_in *pudp_in, struct udp_in6 *pudp_in6,
39236c6e
A
835 struct udp_ip6 *pudp_ip6, struct ifnet *ifp)
836#else /* !INET6 */
2d21ac55 837udp_append(struct inpcb *last, struct ip *ip, struct mbuf *n, int off,
39236c6e
A
838 struct sockaddr_in *pudp_in, struct ifnet *ifp)
839#endif /* !INET6 */
1c79356b
A
840{
841 struct sockaddr *append_sa;
842 struct mbuf *opts = 0;
39236c6e
A
843 boolean_t cell = IFNET_IS_CELLULAR(ifp);
844 boolean_t wifi = (!cell && IFNET_IS_WIFI(ifp));
fe8ab488 845 boolean_t wired = (!wifi && IFNET_IS_WIRED(ifp));
6d2010ae 846 int ret = 0;
1c79356b 847
2d21ac55
A
848#if CONFIG_MACF_NET
849 if (mac_inpcb_check_deliver(last, n, AF_INET, SOCK_DGRAM) != 0) {
850 m_freem(n);
851 return;
852 }
39236c6e 853#endif /* CONFIG_MACF_NET */
6d2010ae
A
854 if ((last->inp_flags & INP_CONTROLOPTS) != 0 ||
855 (last->inp_socket->so_options & SO_TIMESTAMP) != 0 ||
d9a64523 856 (last->inp_socket->so_options & SO_TIMESTAMP_MONOTONIC) != 0 ||
0a7de745 857 (last->inp_socket->so_options & SO_TIMESTAMP_CONTINUOUS) != 0) {
1c79356b
A
858#if INET6
859 if (last->inp_vflag & INP_IPV6) {
860 int savedflags;
861
2d21ac55
A
862 if (pudp_ip6->uip6_init_done == 0) {
863 ip_2_ip6_hdr(&pudp_ip6->uip6_ip6, ip);
864 pudp_ip6->uip6_init_done = 1;
1c79356b
A
865 }
866 savedflags = last->inp_flags;
867 last->inp_flags &= ~INP_UNMAPPABLEOPTS;
6d2010ae
A
868 ret = ip6_savecontrol(last, n, &opts);
869 if (ret != 0) {
870 last->inp_flags = savedflags;
871 goto error;
872 }
1c79356b
A
873 last->inp_flags = savedflags;
874 } else
39236c6e 875#endif /* INET6 */
6d2010ae
A
876 {
877 ret = ip_savecontrol(last, &opts, ip, n);
878 if (ret != 0) {
879 goto error;
880 }
881 }
1c79356b
A
882 }
883#if INET6
884 if (last->inp_vflag & INP_IPV6) {
2d21ac55
A
885 if (pudp_in6->uin6_init_done == 0) {
886 in6_sin_2_v4mapsin6(pudp_in, &pudp_in6->uin6_sin);
887 pudp_in6->uin6_init_done = 1;
1c79356b 888 }
2d21ac55 889 append_sa = (struct sockaddr *)&pudp_in6->uin6_sin;
1c79356b 890 } else
39236c6e 891#endif /* INET6 */
2d21ac55 892 append_sa = (struct sockaddr *)pudp_in;
6d2010ae 893 if (nstat_collect) {
fe8ab488 894 INP_ADD_STAT(last, cell, wifi, wired, rxpackets, 1);
39037602 895 INP_ADD_STAT(last, cell, wifi, wired, rxbytes,
fe8ab488 896 n->m_pkthdr.len);
5ba3f43e 897 inp_set_activity_bitmap(last);
6d2010ae
A
898 }
899 so_recv_data_stat(last->inp_socket, n, 0);
1c79356b 900 m_adj(n, off);
39236c6e
A
901 if (sbappendaddr(&last->inp_socket->so_rcv, append_sa,
902 n, opts, NULL) == 0) {
1c79356b 903 udpstat.udps_fullsock++;
6d2010ae 904 } else {
1c79356b 905 sorwakeup(last->inp_socket);
6d2010ae
A
906 }
907 return;
908error:
909 m_freem(n);
910 m_freem(opts);
1c79356b
A
911}
912
1c79356b
A
913/*
914 * Notify a udp user of an asynchronous error;
915 * just wake up so that he can collect error status.
916 */
917void
39236c6e 918udp_notify(struct inpcb *inp, int errno)
1c79356b
A
919{
920 inp->inp_socket->so_error = errno;
921 sorwakeup(inp->inp_socket);
922 sowwakeup(inp->inp_socket);
923}
924
925void
5ba3f43e 926udp_ctlinput(int cmd, struct sockaddr *sa, void *vip, __unused struct ifnet * ifp)
1c79356b 927{
9bccf70c 928 struct ip *ip = vip;
91447636 929 void (*notify)(struct inpcb *, int) = udp_notify;
39037602 930 struct in_addr faddr;
5ba3f43e 931 struct inpcb *inp = NULL;
9bccf70c 932
316670eb 933 faddr = ((struct sockaddr_in *)(void *)sa)->sin_addr;
0a7de745 934 if (sa->sa_family != AF_INET || faddr.s_addr == INADDR_ANY) {
316670eb 935 return;
0a7de745 936 }
1c79356b 937
9bccf70c
A
938 if (PRC_IS_REDIRECT(cmd)) {
939 ip = 0;
940 notify = in_rtchange;
39236c6e 941 } else if (cmd == PRC_HOSTDEAD) {
9bccf70c 942 ip = 0;
39236c6e 943 } else if ((unsigned)cmd >= PRC_NCMDS || inetctlerrmap[cmd] == 0) {
1c79356b 944 return;
39236c6e 945 }
1c79356b 946 if (ip) {
316670eb
A
947 struct udphdr uh;
948
0a7de745 949 bcopy(((caddr_t)ip + (ip->ip_hl << 2)), &uh, sizeof(uh));
316670eb 950 inp = in_pcblookup_hash(&udbinfo, faddr, uh.uh_dport,
39037602 951 ip->ip_src, uh.uh_sport, 0, NULL);
91447636
A
952 if (inp != NULL && inp->inp_socket != NULL) {
953 udp_lock(inp->inp_socket, 1, 0);
316670eb 954 if (in_pcb_checkstate(inp, WNT_RELEASE, 1) ==
0a7de745 955 WNT_STOPUSING) {
91447636
A
956 udp_unlock(inp->inp_socket, 1, 0);
957 return;
958 }
9bccf70c 959 (*notify)(inp, inetctlerrmap[cmd]);
91447636
A
960 udp_unlock(inp->inp_socket, 1, 0);
961 }
39236c6e 962 } else {
91447636 963 in_pcbnotifyall(&udbinfo, faddr, inetctlerrmap[cmd], notify);
39236c6e 964 }
1c79356b
A
965}
966
2d21ac55
A
967int
968udp_ctloutput(struct socket *so, struct sockopt *sopt)
969{
0a7de745
A
970 int error = 0, optval = 0;
971 struct inpcb *inp;
2d21ac55 972
316670eb
A
973 /* Allow <SOL_SOCKET,SO_FLUSH> at this level */
974 if (sopt->sopt_level != IPPROTO_UDP &&
0a7de745
A
975 !(sopt->sopt_level == SOL_SOCKET && sopt->sopt_name == SO_FLUSH)) {
976 return ip_ctloutput(so, sopt);
977 }
2d21ac55 978
2d21ac55
A
979 inp = sotoinpcb(so);
980
981 switch (sopt->sopt_dir) {
982 case SOPT_SET:
983 switch (sopt->sopt_name) {
984 case UDP_NOCKSUM:
985 /* This option is settable only for UDP over IPv4 */
986 if (!(inp->inp_vflag & INP_IPV4)) {
987 error = EINVAL;
988 break;
989 }
990
0a7de745
A
991 if ((error = sooptcopyin(sopt, &optval, sizeof(optval),
992 sizeof(optval))) != 0) {
2d21ac55 993 break;
0a7de745 994 }
2d21ac55 995
0a7de745 996 if (optval != 0) {
2d21ac55 997 inp->inp_flags |= INP_UDP_NOCKSUM;
0a7de745 998 } else {
2d21ac55 999 inp->inp_flags &= ~INP_UDP_NOCKSUM;
0a7de745 1000 }
2d21ac55 1001 break;
3e170ce0
A
1002 case UDP_KEEPALIVE_OFFLOAD:
1003 {
1004 struct udp_keepalive_offload ka;
1005 /*
1006 * If the socket is not connected, the stack will
1007 * not know the destination address to put in the
1008 * keepalive datagram. Return an error now instead
1009 * of failing later.
1010 */
1011 if (!(so->so_state & SS_ISCONNECTED)) {
1012 error = EINVAL;
1013 break;
1014 }
1015 if (sopt->sopt_valsize != sizeof(ka)) {
1016 error = EINVAL;
1017 break;
1018 }
1019 if ((error = sooptcopyin(sopt, &ka, sizeof(ka),
0a7de745 1020 sizeof(ka))) != 0) {
3e170ce0 1021 break;
0a7de745 1022 }
3e170ce0
A
1023
1024 /* application should specify the type */
0a7de745
A
1025 if (ka.ka_type == 0) {
1026 return EINVAL;
1027 }
3e170ce0
A
1028
1029 if (ka.ka_interval == 0) {
1030 /*
1031 * if interval is 0, disable the offload
1032 * mechanism
1033 */
0a7de745 1034 if (inp->inp_keepalive_data != NULL) {
3e170ce0
A
1035 FREE(inp->inp_keepalive_data,
1036 M_TEMP);
0a7de745 1037 }
3e170ce0
A
1038 inp->inp_keepalive_data = NULL;
1039 inp->inp_keepalive_datalen = 0;
1040 inp->inp_keepalive_interval = 0;
1041 inp->inp_keepalive_type = 0;
1042 inp->inp_flags2 &= ~INP2_KEEPALIVE_OFFLOAD;
1043 } else {
1044 if (inp->inp_keepalive_data != NULL) {
1045 FREE(inp->inp_keepalive_data,
1046 M_TEMP);
1047 inp->inp_keepalive_data = NULL;
1048 }
2d21ac55 1049
3e170ce0 1050 inp->inp_keepalive_datalen = min(
0a7de745
A
1051 ka.ka_data_len,
1052 UDP_KEEPALIVE_OFFLOAD_DATA_SIZE);
3e170ce0
A
1053 if (inp->inp_keepalive_datalen > 0) {
1054 MALLOC(inp->inp_keepalive_data,
39037602 1055 u_int8_t *,
3e170ce0
A
1056 inp->inp_keepalive_datalen,
1057 M_TEMP, M_WAITOK);
1058 if (inp->inp_keepalive_data == NULL) {
1059 inp->inp_keepalive_datalen = 0;
1060 error = ENOMEM;
1061 break;
1062 }
1063 bcopy(ka.ka_data,
1064 inp->inp_keepalive_data,
1065 inp->inp_keepalive_datalen);
1066 } else {
1067 inp->inp_keepalive_datalen = 0;
1068 }
1069 inp->inp_keepalive_interval =
1070 min(UDP_KEEPALIVE_INTERVAL_MAX_SECONDS,
1071 ka.ka_interval);
1072 inp->inp_keepalive_type = ka.ka_type;
1073 inp->inp_flags2 |= INP2_KEEPALIVE_OFFLOAD;
1074 }
1075 break;
1076 }
316670eb 1077 case SO_FLUSH:
0a7de745
A
1078 if ((error = sooptcopyin(sopt, &optval, sizeof(optval),
1079 sizeof(optval))) != 0) {
316670eb 1080 break;
0a7de745 1081 }
316670eb
A
1082
1083 error = inp_flush(inp, optval);
1084 break;
1085
2d21ac55
A
1086 default:
1087 error = ENOPROTOOPT;
1088 break;
1089 }
1090 break;
1091
1092 case SOPT_GET:
1093 switch (sopt->sopt_name) {
1094 case UDP_NOCKSUM:
1095 optval = inp->inp_flags & INP_UDP_NOCKSUM;
1096 break;
1097
1098 default:
1099 error = ENOPROTOOPT;
1100 break;
1101 }
0a7de745
A
1102 if (error == 0) {
1103 error = sooptcopyout(sopt, &optval, sizeof(optval));
1104 }
2d21ac55
A
1105 break;
1106 }
0a7de745 1107 return error;
2d21ac55
A
1108}
1109
1c79356b
A
1110static int
1111udp_pcblist SYSCTL_HANDLER_ARGS
1112{
2d21ac55 1113#pragma unused(oidp, arg1, arg2)
91447636 1114 int error, i, n;
1c79356b
A
1115 struct inpcb *inp, **inp_list;
1116 inp_gen_t gencnt;
1117 struct xinpgen xig;
1118
1119 /*
1120 * The process of preparing the TCB list is too time-consuming and
1121 * resource-intensive to repeat twice on every request.
1122 */
39236c6e 1123 lck_rw_lock_exclusive(udbinfo.ipi_lock);
91447636 1124 if (req->oldptr == USER_ADDR_NULL) {
1c79356b 1125 n = udbinfo.ipi_count;
0a7de745
A
1126 req->oldidx = 2 * (sizeof(xig))
1127 + (n + n / 8) * sizeof(struct xinpcb);
39236c6e 1128 lck_rw_done(udbinfo.ipi_lock);
0a7de745 1129 return 0;
1c79356b
A
1130 }
1131
91447636 1132 if (req->newptr != USER_ADDR_NULL) {
39236c6e 1133 lck_rw_done(udbinfo.ipi_lock);
0a7de745 1134 return EPERM;
91447636 1135 }
1c79356b
A
1136
1137 /*
1138 * OK, now we're committed to doing something.
1139 */
1c79356b
A
1140 gencnt = udbinfo.ipi_gencnt;
1141 n = udbinfo.ipi_count;
1c79356b 1142
0a7de745
A
1143 bzero(&xig, sizeof(xig));
1144 xig.xig_len = sizeof(xig);
1c79356b
A
1145 xig.xig_count = n;
1146 xig.xig_gen = gencnt;
1147 xig.xig_sogen = so_gencnt;
0a7de745 1148 error = SYSCTL_OUT(req, &xig, sizeof(xig));
91447636 1149 if (error) {
39236c6e 1150 lck_rw_done(udbinfo.ipi_lock);
0a7de745 1151 return error;
91447636 1152 }
39236c6e
A
1153 /*
1154 * We are done if there is no pcb
1155 */
1156 if (n == 0) {
1157 lck_rw_done(udbinfo.ipi_lock);
0a7de745 1158 return 0;
39236c6e
A
1159 }
1160
0a7de745 1161 inp_list = _MALLOC(n * sizeof(*inp_list), M_TEMP, M_WAITOK);
1c79356b 1162 if (inp_list == 0) {
39236c6e 1163 lck_rw_done(udbinfo.ipi_lock);
0a7de745 1164 return ENOMEM;
1c79356b 1165 }
39236c6e
A
1166
1167 for (inp = LIST_FIRST(udbinfo.ipi_listhead), i = 0; inp && i < n;
39037602 1168 inp = LIST_NEXT(inp, inp_list)) {
39236c6e 1169 if (inp->inp_gencnt <= gencnt &&
0a7de745 1170 inp->inp_state != INPCB_STATE_DEAD) {
1c79356b 1171 inp_list[i++] = inp;
0a7de745 1172 }
1c79356b 1173 }
1c79356b
A
1174 n = i;
1175
1176 error = 0;
1177 for (i = 0; i < n; i++) {
39037602
A
1178 struct xinpcb xi;
1179
1c79356b 1180 inp = inp_list[i];
39037602 1181
0a7de745 1182 if (in_pcb_checkstate(inp, WNT_ACQUIRE, 0) == WNT_STOPUSING) {
39037602 1183 continue;
0a7de745 1184 }
39037602
A
1185 udp_lock(inp->inp_socket, 1, 0);
1186 if (in_pcb_checkstate(inp, WNT_RELEASE, 1) == WNT_STOPUSING) {
1187 udp_unlock(inp->inp_socket, 1, 0);
1188 continue;
1189 }
1190 if (inp->inp_gencnt > gencnt) {
1191 udp_unlock(inp->inp_socket, 1, 0);
1192 continue;
1c79356b 1193 }
39037602 1194
0a7de745
A
1195 bzero(&xi, sizeof(xi));
1196 xi.xi_len = sizeof(xi);
39037602
A
1197 /* XXX should avoid extra copy */
1198 inpcb_to_compat(inp, &xi.xi_inp);
0a7de745 1199 if (inp->inp_socket) {
39037602 1200 sotoxsocket(inp->inp_socket, &xi.xi_socket);
0a7de745 1201 }
39037602
A
1202
1203 udp_unlock(inp->inp_socket, 1, 0);
1204
0a7de745 1205 error = SYSCTL_OUT(req, &xi, sizeof(xi));
1c79356b
A
1206 }
1207 if (!error) {
1208 /*
1209 * Give the user an updated idea of our state.
1210 * If the generation differs from what we told
1211 * her before, she knows that something happened
1212 * while we were processing this request, and it
1213 * might be necessary to retry.
1214 */
0a7de745
A
1215 bzero(&xig, sizeof(xig));
1216 xig.xig_len = sizeof(xig);
1c79356b
A
1217 xig.xig_gen = udbinfo.ipi_gencnt;
1218 xig.xig_sogen = so_gencnt;
1219 xig.xig_count = udbinfo.ipi_count;
0a7de745 1220 error = SYSCTL_OUT(req, &xig, sizeof(xig));
1c79356b
A
1221 }
1222 FREE(inp_list, M_TEMP);
39236c6e 1223 lck_rw_done(udbinfo.ipi_lock);
0a7de745 1224 return error;
1c79356b
A
1225}
1226
39236c6e 1227SYSCTL_PROC(_net_inet_udp, UDPCTL_PCBLIST, pcblist,
0a7de745
A
1228 CTLTYPE_STRUCT | CTLFLAG_RD | CTLFLAG_LOCKED, 0, 0, udp_pcblist,
1229 "S,xinpcb", "List of active UDP sockets");
1c79356b 1230
5ba3f43e 1231#if !CONFIG_EMBEDDED
1c79356b 1232
b0d623f7
A
1233static int
1234udp_pcblist64 SYSCTL_HANDLER_ARGS
1235{
1236#pragma unused(oidp, arg1, arg2)
39037602
A
1237 int error, i, n;
1238 struct inpcb *inp, **inp_list;
1239 inp_gen_t gencnt;
1240 struct xinpgen xig;
1241
1242 /*
1243 * The process of preparing the TCB list is too time-consuming and
1244 * resource-intensive to repeat twice on every request.
1245 */
1246 lck_rw_lock_shared(udbinfo.ipi_lock);
1247 if (req->oldptr == USER_ADDR_NULL) {
1248 n = udbinfo.ipi_count;
1249 req->oldidx =
0a7de745 1250 2 * (sizeof(xig)) + (n + n / 8) * sizeof(struct xinpcb64);
39037602 1251 lck_rw_done(udbinfo.ipi_lock);
0a7de745 1252 return 0;
39037602
A
1253 }
1254
1255 if (req->newptr != USER_ADDR_NULL) {
1256 lck_rw_done(udbinfo.ipi_lock);
0a7de745 1257 return EPERM;
39037602
A
1258 }
1259
1260 /*
1261 * OK, now we're committed to doing something.
1262 */
1263 gencnt = udbinfo.ipi_gencnt;
1264 n = udbinfo.ipi_count;
1265
0a7de745
A
1266 bzero(&xig, sizeof(xig));
1267 xig.xig_len = sizeof(xig);
39037602
A
1268 xig.xig_count = n;
1269 xig.xig_gen = gencnt;
1270 xig.xig_sogen = so_gencnt;
0a7de745 1271 error = SYSCTL_OUT(req, &xig, sizeof(xig));
39037602
A
1272 if (error) {
1273 lck_rw_done(udbinfo.ipi_lock);
0a7de745 1274 return error;
39037602 1275 }
39236c6e
A
1276 /*
1277 * We are done if there is no pcb
1278 */
1279 if (n == 0) {
1280 lck_rw_done(udbinfo.ipi_lock);
0a7de745 1281 return 0;
39236c6e
A
1282 }
1283
0a7de745 1284 inp_list = _MALLOC(n * sizeof(*inp_list), M_TEMP, M_WAITOK);
39037602
A
1285 if (inp_list == 0) {
1286 lck_rw_done(udbinfo.ipi_lock);
0a7de745 1287 return ENOMEM;
39037602 1288 }
b0d623f7 1289
39037602
A
1290 for (inp = LIST_FIRST(udbinfo.ipi_listhead), i = 0; inp && i < n;
1291 inp = LIST_NEXT(inp, inp_list)) {
1292 if (inp->inp_gencnt <= gencnt &&
0a7de745 1293 inp->inp_state != INPCB_STATE_DEAD) {
39037602 1294 inp_list[i++] = inp;
0a7de745 1295 }
39037602
A
1296 }
1297 n = i;
1298
1299 error = 0;
1300 for (i = 0; i < n; i++) {
1301 struct xinpcb64 xi;
1302
1303 inp = inp_list[i];
1304
0a7de745 1305 if (in_pcb_checkstate(inp, WNT_ACQUIRE, 0) == WNT_STOPUSING) {
39037602 1306 continue;
0a7de745 1307 }
39037602
A
1308 udp_lock(inp->inp_socket, 1, 0);
1309 if (in_pcb_checkstate(inp, WNT_RELEASE, 1) == WNT_STOPUSING) {
1310 udp_unlock(inp->inp_socket, 1, 0);
1311 continue;
1312 }
1313 if (inp->inp_gencnt > gencnt) {
1314 udp_unlock(inp->inp_socket, 1, 0);
1315 continue;
1316 }
1317
0a7de745
A
1318 bzero(&xi, sizeof(xi));
1319 xi.xi_len = sizeof(xi);
39037602 1320 inpcb_to_xinpcb64(inp, &xi);
0a7de745 1321 if (inp->inp_socket) {
39037602 1322 sotoxsocket64(inp->inp_socket, &xi.xi_socket);
0a7de745 1323 }
39037602
A
1324
1325 udp_unlock(inp->inp_socket, 1, 0);
1326
0a7de745 1327 error = SYSCTL_OUT(req, &xi, sizeof(xi));
39037602
A
1328 }
1329 if (!error) {
1330 /*
1331 * Give the user an updated idea of our state.
1332 * If the generation differs from what we told
1333 * her before, she knows that something happened
1334 * while we were processing this request, and it
1335 * might be necessary to retry.
1336 */
0a7de745
A
1337 bzero(&xig, sizeof(xig));
1338 xig.xig_len = sizeof(xig);
39037602
A
1339 xig.xig_gen = udbinfo.ipi_gencnt;
1340 xig.xig_sogen = so_gencnt;
1341 xig.xig_count = udbinfo.ipi_count;
0a7de745 1342 error = SYSCTL_OUT(req, &xig, sizeof(xig));
39037602
A
1343 }
1344 FREE(inp_list, M_TEMP);
1345 lck_rw_done(udbinfo.ipi_lock);
0a7de745 1346 return error;
b0d623f7
A
1347}
1348
39236c6e 1349SYSCTL_PROC(_net_inet_udp, OID_AUTO, pcblist64,
0a7de745
A
1350 CTLTYPE_STRUCT | CTLFLAG_RD | CTLFLAG_LOCKED, 0, 0, udp_pcblist64,
1351 "S,xinpcb64", "List of active UDP sockets");
b0d623f7 1352
5ba3f43e 1353#endif /* !CONFIG_EMBEDDED */
1c79356b 1354
6d2010ae
A
1355static int
1356udp_pcblist_n SYSCTL_HANDLER_ARGS
1357{
1358#pragma unused(oidp, arg1, arg2)
0a7de745 1359 return get_pcblist_n(IPPROTO_UDP, req, &udbinfo);
6d2010ae
A
1360}
1361
39236c6e 1362SYSCTL_PROC(_net_inet_udp, OID_AUTO, pcblist_n,
0a7de745
A
1363 CTLTYPE_STRUCT | CTLFLAG_RD | CTLFLAG_LOCKED, 0, 0, udp_pcblist_n,
1364 "S,xinpcb_n", "List of active UDP sockets");
6d2010ae 1365
316670eb 1366__private_extern__ void
fe8ab488 1367udp_get_ports_used(uint32_t ifindex, int protocol, uint32_t flags,
39236c6e 1368 bitstr_t *bitfield)
316670eb 1369{
d9a64523
A
1370 inpcb_get_ports_used(ifindex, protocol, flags, bitfield,
1371 &udbinfo);
1372}
316670eb
A
1373
1374__private_extern__ uint32_t
1375udp_count_opportunistic(unsigned int ifindex, u_int32_t flags)
1376{
0a7de745 1377 return inpcb_count_opportunistic(ifindex, &udbinfo, flags);
316670eb
A
1378}
1379
39236c6e
A
1380__private_extern__ uint32_t
1381udp_find_anypcb_byaddr(struct ifaddr *ifa)
91447636 1382{
0a7de745 1383 return inpcb_find_anypcb_byaddr(ifa, &udbinfo);
91447636
A
1384}
1385
6d2010ae 1386static int
39236c6e
A
1387udp_check_pktinfo(struct mbuf *control, struct ifnet **outif,
1388 struct in_addr *laddr)
6d2010ae
A
1389{
1390 struct cmsghdr *cm = 0;
316670eb 1391 struct in_pktinfo *pktinfo;
6d2010ae
A
1392 struct ifnet *ifp;
1393
0a7de745 1394 if (outif != NULL) {
39236c6e 1395 *outif = NULL;
0a7de745 1396 }
39236c6e 1397
6d2010ae
A
1398 /*
1399 * XXX: Currently, we assume all the optional information is stored
1400 * in a single mbuf.
1401 */
0a7de745
A
1402 if (control->m_next) {
1403 return EINVAL;
1404 }
6d2010ae 1405
0a7de745
A
1406 if (control->m_len < CMSG_LEN(0)) {
1407 return EINVAL;
1408 }
6d2010ae 1409
0a7de745
A
1410 for (cm = M_FIRST_CMSGHDR(control);
1411 is_cmsg_valid(control, cm);
39236c6e 1412 cm = M_NXT_CMSGHDR(control, cm)) {
0a7de745
A
1413 if (cm->cmsg_level != IPPROTO_IP ||
1414 cm->cmsg_type != IP_PKTINFO) {
6d2010ae 1415 continue;
0a7de745 1416 }
6d2010ae 1417
0a7de745
A
1418 if (cm->cmsg_len != CMSG_LEN(sizeof(struct in_pktinfo))) {
1419 return EINVAL;
1420 }
6d2010ae 1421
316670eb 1422 pktinfo = (struct in_pktinfo *)(void *)CMSG_DATA(cm);
6d2010ae
A
1423
1424 /* Check for a valid ifindex in pktinfo */
1425 ifnet_head_lock_shared();
1426
1427 if (pktinfo->ipi_ifindex > if_index) {
1428 ifnet_head_done();
0a7de745 1429 return ENXIO;
6d2010ae
A
1430 }
1431
39236c6e
A
1432 /*
1433 * If ipi_ifindex is specified it takes precedence
1434 * over ipi_spec_dst.
1435 */
6d2010ae
A
1436 if (pktinfo->ipi_ifindex) {
1437 ifp = ifindex2ifnet[pktinfo->ipi_ifindex];
1438 if (ifp == NULL) {
1439 ifnet_head_done();
0a7de745 1440 return ENXIO;
6d2010ae 1441 }
39236c6e
A
1442 if (outif != NULL) {
1443 ifnet_reference(ifp);
316670eb 1444 *outif = ifp;
39236c6e
A
1445 }
1446 ifnet_head_done();
6d2010ae
A
1447 laddr->s_addr = INADDR_ANY;
1448 break;
1449 }
1450
1451 ifnet_head_done();
1452
39236c6e
A
1453 /*
1454 * Use the provided ipi_spec_dst address for temp
1455 * source address.
1456 */
6d2010ae
A
1457 *laddr = pktinfo->ipi_spec_dst;
1458 break;
1459 }
0a7de745 1460 return 0;
6d2010ae
A
1461}
1462
5ba3f43e 1463int
39236c6e
A
1464udp_output(struct inpcb *inp, struct mbuf *m, struct sockaddr *addr,
1465 struct mbuf *control, struct proc *p)
1c79356b 1466{
39236c6e
A
1467 struct udpiphdr *ui;
1468 int len = m->m_pkthdr.len;
2d21ac55 1469 struct sockaddr_in *sin;
6d2010ae 1470 struct in_addr origladdr, laddr, faddr, pi_laddr;
91447636 1471 u_short lport, fport;
6d2010ae 1472 int error = 0, udp_dodisconnect = 0, pktinfo = 0;
2d21ac55 1473 struct socket *so = inp->inp_socket;
c910b4d9 1474 int soopts = 0;
2d21ac55
A
1475 struct mbuf *inpopts;
1476 struct ip_moptions *mopts;
1477 struct route ro;
a39ff7e2 1478 struct ip_out_args ipoa;
d9a64523
A
1479#if CONTENT_FILTER
1480 struct m_tag *cfil_tag = NULL;
1481 bool cfil_faddr_use = false;
1482 uint32_t cfil_so_state_change_cnt = 0;
1483 short cfil_so_options = 0;
1484 struct sockaddr *cfil_faddr = NULL;
1485#endif
a39ff7e2
A
1486
1487 bzero(&ipoa, sizeof(ipoa));
1488 ipoa.ipoa_boundif = IFSCOPE_NONE;
1489 ipoa.ipoa_flags = IPOAF_SELECT_SRCIF;
1490
316670eb
A
1491 struct ifnet *outif = NULL;
1492 struct flowadv *adv = &ipoa.ipoa_flowadv;
39037602
A
1493 int sotc = SO_TC_UNSPEC;
1494 int netsvctype = _NET_SERVICE_TYPE_UNSPEC;
fe8ab488 1495 struct ifnet *origoutifp = NULL;
316670eb 1496 int flowadv = 0;
94ff46dc 1497 int tos = IPTOS_UNSPEC;
316670eb
A
1498
1499 /* Enable flow advisory only when connected */
1500 flowadv = (so->so_state & SS_ISCONNECTED) ? 1 : 0;
6d2010ae 1501 pi_laddr.s_addr = INADDR_ANY;
1c79356b 1502
39037602 1503 KERNEL_DEBUG(DBG_FNC_UDP_OUTPUT | DBG_FUNC_START, 0, 0, 0, 0, 0);
1c79356b 1504
5ba3f43e 1505 socket_lock_assert_owned(so);
d9a64523
A
1506
1507#if CONTENT_FILTER
1508 /*
1509 * If socket is subject to UDP Content Filter and no addr is passed in,
1510 * retrieve CFIL saved state from mbuf and use it if necessary.
1511 */
1512 if (so->so_cfil_db && !addr) {
1513 cfil_tag = cfil_udp_get_socket_state(m, &cfil_so_state_change_cnt, &cfil_so_options, &cfil_faddr);
1514 if (cfil_tag) {
1515 sin = (struct sockaddr_in *)(void *)cfil_faddr;
1516 if (inp && inp->inp_faddr.s_addr == INADDR_ANY) {
1517 /*
1518 * Socket is unconnected, simply use the saved faddr as 'addr' to go through
1519 * the connect/disconnect logic.
1520 */
1521 addr = (struct sockaddr *)cfil_faddr;
1522 } else if ((so->so_state_change_cnt != cfil_so_state_change_cnt) &&
0a7de745
A
1523 (inp->inp_fport != sin->sin_port ||
1524 inp->inp_faddr.s_addr != sin->sin_addr.s_addr)) {
d9a64523
A
1525 /*
1526 * Socket is connected but socket state and dest addr/port changed.
1527 * We need to use the saved faddr info.
1528 */
1529 cfil_faddr_use = true;
1530 }
1531 }
1532 }
1533#endif
1534
d41d1dae 1535 if (control != NULL) {
94ff46dc 1536 tos = so_tos_from_control(control);
39037602 1537 sotc = so_tc_from_control(control, &netsvctype);
39236c6e 1538 VERIFY(outif == NULL);
316670eb 1539 error = udp_check_pktinfo(control, &outif, &pi_laddr);
d41d1dae 1540 m_freem(control);
39236c6e 1541 control = NULL;
0a7de745 1542 if (error) {
6d2010ae 1543 goto release;
0a7de745 1544 }
6d2010ae 1545 pktinfo++;
0a7de745 1546 if (outif != NULL) {
316670eb 1547 ipoa.ipoa_boundif = outif->if_index;
0a7de745 1548 }
d41d1dae 1549 }
39037602
A
1550 if (sotc == SO_TC_UNSPEC) {
1551 sotc = so->so_traffic_class;
1552 netsvctype = so->so_netsvctype;
1553 }
6d2010ae 1554
1c79356b 1555 KERNEL_DEBUG(DBG_LAYER_OUT_BEG, inp->inp_fport, inp->inp_lport,
39236c6e 1556 inp->inp_laddr.s_addr, inp->inp_faddr.s_addr,
0a7de745 1557 (htons((u_short)len + sizeof(struct udphdr))));
1c79356b 1558
0a7de745 1559 if (len + sizeof(struct udpiphdr) > IP_MAXPACKET) {
1c79356b
A
1560 error = EMSGSIZE;
1561 goto release;
1562 }
1563
316670eb
A
1564 if (flowadv && INP_WAIT_FOR_IF_FEEDBACK(inp)) {
1565 /*
1566 * The socket is flow-controlled, drop the packets
1567 * until the inp is not flow controlled
1568 */
1569 error = ENOBUFS;
1570 goto release;
1571 }
6d2010ae
A
1572 /*
1573 * If socket was bound to an ifindex, tell ip_output about it.
1574 * If the ancillary IP_PKTINFO option contains an interface index,
1575 * it takes precedence over the one specified by IP_BOUND_IF.
1576 */
39236c6e
A
1577 if (ipoa.ipoa_boundif == IFSCOPE_NONE &&
1578 (inp->inp_flags & INP_BOUND_IF)) {
1579 VERIFY(inp->inp_boundifp != NULL);
0a7de745
A
1580 ifnet_reference(inp->inp_boundifp); /* for this routine */
1581 if (outif != NULL) {
39236c6e 1582 ifnet_release(outif);
0a7de745 1583 }
316670eb
A
1584 outif = inp->inp_boundifp;
1585 ipoa.ipoa_boundif = outif->if_index;
1586 }
0a7de745 1587 if (INP_NO_CELLULAR(inp)) {
316670eb 1588 ipoa.ipoa_flags |= IPOAF_NO_CELLULAR;
0a7de745
A
1589 }
1590 if (INP_NO_EXPENSIVE(inp)) {
fe8ab488 1591 ipoa.ipoa_flags |= IPOAF_NO_EXPENSIVE;
0a7de745 1592 }
cb323159
A
1593 if (INP_NO_CONSTRAINED(inp)) {
1594 ipoa.ipoa_flags |= IPOAF_NO_CONSTRAINED;
1595 }
0a7de745 1596 if (INP_AWDL_UNRESTRICTED(inp)) {
fe8ab488 1597 ipoa.ipoa_flags |= IPOAF_AWDL_UNRESTRICTED;
0a7de745 1598 }
39037602
A
1599 ipoa.ipoa_sotc = sotc;
1600 ipoa.ipoa_netsvctype = netsvctype;
c910b4d9
A
1601 soopts |= IP_OUTARGS;
1602
39236c6e
A
1603 /*
1604 * If there was a routing change, discard cached route and check
1605 * that we have a valid source address. Reacquire a new source
1606 * address if INADDR_ANY was specified.
d9a64523
A
1607 *
1608 * If we are using cfil saved state, go through this cache cleanup
1609 * so that we can get a new route.
55e303ae 1610 */
d9a64523
A
1611 if (ROUTE_UNUSABLE(&inp->inp_route)
1612#if CONTENT_FILTER
0a7de745 1613 || cfil_faddr_use
d9a64523 1614#endif
0a7de745 1615 ) {
39236c6e
A
1616 struct in_ifaddr *ia = NULL;
1617
1618 ROUTE_RELEASE(&inp->inp_route);
b0d623f7
A
1619
1620 /* src address is gone? */
39236c6e
A
1621 if (inp->inp_laddr.s_addr != INADDR_ANY &&
1622 (ia = ifa_foraddr(inp->inp_laddr.s_addr)) == NULL) {
1623 if (!(inp->inp_flags & INP_INADDR_ANY) ||
316670eb 1624 (so->so_state & SS_ISCONNECTED)) {
39236c6e
A
1625 /*
1626 * Rdar://5448998
1627 * If the source address is gone, return an
1628 * error if:
6d2010ae
A
1629 * - the source was specified
1630 * - the socket was already connected
1631 */
39236c6e 1632 soevent(so, (SO_FILT_HINT_LOCKED |
316670eb 1633 SO_FILT_HINT_NOSRCADDR));
55e303ae
A
1634 error = EADDRNOTAVAIL;
1635 goto release;
6d2010ae
A
1636 } else {
1637 /* new src will be set later */
1638 inp->inp_laddr.s_addr = INADDR_ANY;
316670eb 1639 inp->inp_last_outifp = NULL;
55e303ae
A
1640 }
1641 }
0a7de745 1642 if (ia != NULL) {
6d2010ae 1643 IFA_REMREF(&ia->ia_ifa);
0a7de745 1644 }
55e303ae
A
1645 }
1646
39236c6e
A
1647 /*
1648 * IP_PKTINFO option check. If a temporary scope or src address
1649 * is provided, use it for this packet only and make sure we forget
1650 * it after sending this datagram.
6d2010ae 1651 */
6d2010ae
A
1652 if (pi_laddr.s_addr != INADDR_ANY ||
1653 (ipoa.ipoa_boundif != IFSCOPE_NONE && pktinfo)) {
39236c6e
A
1654 /* temp src address for this datagram only */
1655 laddr = pi_laddr;
6d2010ae 1656 origladdr.s_addr = INADDR_ANY;
39236c6e
A
1657 /* we don't want to keep the laddr or route */
1658 udp_dodisconnect = 1;
39037602 1659 /* remember we don't care about src addr */
39236c6e 1660 inp->inp_flags |= INP_INADDR_ANY;
6d2010ae
A
1661 } else {
1662 origladdr = laddr = inp->inp_laddr;
1663 }
1664
316670eb 1665 origoutifp = inp->inp_last_outifp;
91447636
A
1666 faddr = inp->inp_faddr;
1667 lport = inp->inp_lport;
1668 fport = inp->inp_fport;
1669
d9a64523 1670#if CONTENT_FILTER
0a7de745 1671 if (cfil_faddr_use) {
d9a64523
A
1672 faddr = ((struct sockaddr_in *)(void *)cfil_faddr)->sin_addr;
1673 fport = ((struct sockaddr_in *)(void *)cfil_faddr)->sin_port;
1674 }
1675#endif
1676
1c79356b 1677 if (addr) {
316670eb 1678 sin = (struct sockaddr_in *)(void *)addr;
91447636 1679 if (faddr.s_addr != INADDR_ANY) {
1c79356b
A
1680 error = EISCONN;
1681 goto release;
1682 }
91447636
A
1683 if (lport == 0) {
1684 /*
39236c6e
A
1685 * In case we don't have a local port set, go through
1686 * the full connect. We don't have a local port yet
1687 * (i.e., we can't be looked up), so it's not an issue
1688 * if the input runs at the same time we do this.
6d2010ae 1689 */
39236c6e 1690 /* if we have a source address specified, use that */
0a7de745 1691 if (pi_laddr.s_addr != INADDR_ANY) {
6d2010ae 1692 inp->inp_laddr = pi_laddr;
0a7de745 1693 }
39236c6e
A
1694 /*
1695 * If a scope is specified, use it. Scope from
1696 * IP_PKTINFO takes precendence over the the scope
1697 * set via INP_BOUND_IF.
1698 */
1699 error = in_pcbconnect(inp, addr, p, ipoa.ipoa_boundif,
1700 &outif);
0a7de745 1701 if (error) {
91447636 1702 goto release;
0a7de745 1703 }
39236c6e 1704
91447636
A
1705 laddr = inp->inp_laddr;
1706 lport = inp->inp_lport;
1707 faddr = inp->inp_faddr;
1708 fport = inp->inp_fport;
1709 udp_dodisconnect = 1;
39236c6e
A
1710
1711 /* synch up in case in_pcbladdr() overrides */
0a7de745 1712 if (outif != NULL && ipoa.ipoa_boundif != IFSCOPE_NONE) {
39236c6e 1713 ipoa.ipoa_boundif = outif->if_index;
0a7de745 1714 }
39037602 1715 } else {
39236c6e
A
1716 /*
1717 * Fast path case
1718 *
1719 * We have a full address and a local port; use those
1720 * info to build the packet without changing the pcb
1721 * and interfering with the input path. See 3851370.
1722 *
1723 * Scope from IP_PKTINFO takes precendence over the
1724 * the scope set via INP_BOUND_IF.
91447636
A
1725 */
1726 if (laddr.s_addr == INADDR_ANY) {
39236c6e 1727 if ((error = in_pcbladdr(inp, addr, &laddr,
0a7de745 1728 ipoa.ipoa_boundif, &outif, 0)) != 0) {
316670eb 1729 goto release;
0a7de745 1730 }
39236c6e
A
1731 /*
1732 * from pcbconnect: remember we don't
1733 * care about src addr.
1734 */
1735 inp->inp_flags |= INP_INADDR_ANY;
1736
1737 /* synch up in case in_pcbladdr() overrides */
1738 if (outif != NULL &&
0a7de745 1739 ipoa.ipoa_boundif != IFSCOPE_NONE) {
39236c6e 1740 ipoa.ipoa_boundif = outif->if_index;
0a7de745 1741 }
91447636 1742 }
6d2010ae 1743
91447636
A
1744 faddr = sin->sin_addr;
1745 fport = sin->sin_port;
1c79356b
A
1746 }
1747 } else {
91447636 1748 if (faddr.s_addr == INADDR_ANY) {
1c79356b
A
1749 error = ENOTCONN;
1750 goto release;
1751 }
1752 }
55e303ae 1753
2d21ac55
A
1754#if CONFIG_MACF_NET
1755 mac_mbuf_label_associate_inpcb(inp, m);
39236c6e
A
1756#endif /* CONFIG_MACF_NET */
1757
0a7de745 1758 if (inp->inp_flowhash == 0) {
316670eb 1759 inp->inp_flowhash = inp_calc_flowhash(inp);
0a7de745 1760 }
55e303ae 1761
5ba3f43e 1762 if (fport == htons(53) && !(so->so_flags1 & SOF1_DNS_COUNTED)) {
0a7de745 1763 so->so_flags1 |= SOF1_DNS_COUNTED;
5ba3f43e
A
1764 INC_ATOMIC_INT64_LIM(net_api_stats.nas_socket_inet_dgram_dns);
1765 }
1766
1c79356b
A
1767 /*
1768 * Calculate data length and get a mbuf
1769 * for UDP and IP headers.
1770 */
0a7de745 1771 M_PREPEND(m, sizeof(struct udpiphdr), M_DONTWAIT, 1);
1c79356b
A
1772 if (m == 0) {
1773 error = ENOBUFS;
55e303ae 1774 goto abort;
1c79356b
A
1775 }
1776
1777 /*
1778 * Fill in mbuf with extended UDP header
1779 * and addresses and length put into network format.
1780 */
1781 ui = mtod(m, struct udpiphdr *);
0a7de745 1782 bzero(ui->ui_x1, sizeof(ui->ui_x1)); /* XXX still needed? */
1c79356b 1783 ui->ui_pr = IPPROTO_UDP;
91447636
A
1784 ui->ui_src = laddr;
1785 ui->ui_dst = faddr;
1786 ui->ui_sport = lport;
1787 ui->ui_dport = fport;
0a7de745 1788 ui->ui_ulen = htons((u_short)len + sizeof(struct udphdr));
1c79356b
A
1789
1790 /*
d9a64523
A
1791 * Set up checksum to pseudo header checksum and output datagram.
1792 *
1793 * Treat flows to be CLAT46'd as IPv6 flow and compute checksum
1794 * no matter what, as IPv6 mandates checksum for UDP.
1795 *
1796 * Here we only compute the one's complement sum of the pseudo header.
1797 * The payload computation and final complement is delayed to much later
1798 * in IP processing to decide if remaining computation needs to be done
1799 * through offload.
1800 *
1801 * That is communicated by setting CSUM_UDP in csum_flags.
1802 * The offset of checksum from the start of ULP header is communicated
1803 * through csum_data.
1804 *
1805 * Note since this already contains the pseudo checksum header, any
1806 * later operation at IP layer that modify the values used here must
1807 * update the checksum as well (for example NAT etc).
1c79356b 1808 */
d9a64523
A
1809 if ((inp->inp_flags2 & INP2_CLAT46_FLOW) ||
1810 (udpcksum && !(inp->inp_flags & INP_UDP_NOCKSUM))) {
39236c6e 1811 ui->ui_sum = in_pseudo(ui->ui_src.s_addr, ui->ui_dst.s_addr,
0a7de745
A
1812 htons((u_short)len + sizeof(struct udphdr) + IPPROTO_UDP));
1813 m->m_pkthdr.csum_flags = (CSUM_UDP | CSUM_ZERO_INVERT);
9bccf70c
A
1814 m->m_pkthdr.csum_data = offsetof(struct udphdr, uh_sum);
1815 } else {
0b4e3aa0 1816 ui->ui_sum = 0;
9bccf70c 1817 }
0a7de745
A
1818 ((struct ip *)ui)->ip_len = sizeof(struct udpiphdr) + len;
1819 ((struct ip *)ui)->ip_ttl = inp->inp_ip_ttl; /* XXX */
94ff46dc
A
1820 if (tos != IPTOS_UNSPEC) {
1821 ((struct ip *)ui)->ip_tos = (uint8_t)(tos & IPTOS_MASK);
1822 } else {
1823 ((struct ip *)ui)->ip_tos = inp->inp_ip_tos; /* XXX */
1824 }
1c79356b
A
1825 udpstat.udps_opackets++;
1826
1827 KERNEL_DEBUG(DBG_LAYER_OUT_END, ui->ui_dport, ui->ui_sport,
39037602 1828 ui->ui_src.s_addr, ui->ui_dst.s_addr, ui->ui_ulen);
3e170ce0 1829
fe8ab488
A
1830#if NECP
1831 {
1832 necp_kernel_policy_id policy_id;
d9a64523 1833 necp_kernel_policy_id skip_policy_id;
3e170ce0 1834 u_int32_t route_rule_id;
39037602
A
1835
1836 /*
1837 * We need a route to perform NECP route rule checks
1838 */
1839 if (net_qos_policy_restricted != 0 &&
1840 ROUTE_UNUSABLE(&inp->inp_route)) {
1841 struct sockaddr_in to;
1842 struct sockaddr_in from;
1843
1844 ROUTE_RELEASE(&inp->inp_route);
1845
1846 bzero(&from, sizeof(struct sockaddr_in));
1847 from.sin_family = AF_INET;
1848 from.sin_len = sizeof(struct sockaddr_in);
1849 from.sin_addr = laddr;
1850
1851 bzero(&to, sizeof(struct sockaddr_in));
1852 to.sin_family = AF_INET;
1853 to.sin_len = sizeof(struct sockaddr_in);
1854 to.sin_addr = faddr;
1855
1856 inp->inp_route.ro_dst.sa_family = AF_INET;
1857 inp->inp_route.ro_dst.sa_len = sizeof(struct sockaddr_in);
1858 ((struct sockaddr_in *)(void *)&inp->inp_route.ro_dst)->sin_addr =
0a7de745 1859 faddr;
39037602
A
1860
1861 rtalloc_scoped(&inp->inp_route, ipoa.ipoa_boundif);
1862
1863 inp_update_necp_policy(inp, (struct sockaddr *)&from,
1864 (struct sockaddr *)&to, ipoa.ipoa_boundif);
1865 inp->inp_policyresult.results.qos_marking_gencount = 0;
1866 }
1867
3e170ce0 1868 if (!necp_socket_is_allowed_to_send_recv_v4(inp, lport, fport,
d9a64523 1869 &laddr, &faddr, NULL, &policy_id, &route_rule_id, &skip_policy_id)) {
fe8ab488
A
1870 error = EHOSTUNREACH;
1871 goto abort;
1872 }
1c79356b 1873
d9a64523 1874 necp_mark_packet_from_socket(m, inp, policy_id, route_rule_id, skip_policy_id);
39037602
A
1875
1876 if (net_qos_policy_restricted != 0) {
1877 necp_socket_update_qos_marking(inp,
1878 inp->inp_route.ro_rt, NULL, route_rule_id);
1879 }
fe8ab488
A
1880 }
1881#endif /* NECP */
0a7de745 1882 if ((so->so_flags1 & SOF1_QOSMARKING_ALLOWED)) {
39037602 1883 ipoa.ipoa_flags |= IPOAF_QOSMARKING_ALLOWED;
0a7de745 1884 }
3e170ce0 1885
1c79356b 1886#if IPSEC
fe8ab488 1887 if (inp->inp_sp != NULL && ipsec_setsocket(m, inp->inp_socket) != 0) {
9bccf70c 1888 error = ENOBUFS;
55e303ae 1889 goto abort;
9bccf70c 1890 }
39236c6e 1891#endif /* IPSEC */
2d21ac55
A
1892
1893 inpopts = inp->inp_options;
d9a64523 1894#if CONTENT_FILTER
0a7de745 1895 if (cfil_tag && (inp->inp_socket->so_options != cfil_so_options)) {
d9a64523 1896 soopts |= (cfil_so_options & (SO_DONTROUTE | SO_BROADCAST));
0a7de745 1897 } else
d9a64523 1898#endif
0a7de745 1899 soopts |= (inp->inp_socket->so_options & (SO_DONTROUTE | SO_BROADCAST));
d9a64523 1900
2d21ac55 1901 mopts = inp->inp_moptions;
316670eb
A
1902 if (mopts != NULL) {
1903 IMO_LOCK(mopts);
1904 IMO_ADDREF_LOCKED(mopts);
1905 if (IN_MULTICAST(ntohl(ui->ui_dst.s_addr)) &&
39236c6e
A
1906 mopts->imo_multicast_ifp != NULL) {
1907 /* no reference needed */
316670eb
A
1908 inp->inp_last_outifp = mopts->imo_multicast_ifp;
1909 }
1910 IMO_UNLOCK(mopts);
1911 }
2d21ac55 1912
b0d623f7
A
1913 /* Copy the cached route and take an extra reference */
1914 inp_route_copyout(inp, &ro);
2d21ac55 1915
39037602 1916 set_packet_service_class(m, so, sotc, 0);
39236c6e
A
1917 m->m_pkthdr.pkt_flowsrc = FLOWSRC_INPCB;
1918 m->m_pkthdr.pkt_flowid = inp->inp_flowhash;
1919 m->m_pkthdr.pkt_proto = IPPROTO_UDP;
1920 m->m_pkthdr.pkt_flags |= (PKTF_FLOW_ID | PKTF_FLOW_LOCALSRC);
0a7de745 1921 if (flowadv) {
39236c6e 1922 m->m_pkthdr.pkt_flags |= PKTF_FLOW_ADV;
0a7de745 1923 }
d9a64523 1924 m->m_pkthdr.tx_udp_pid = so->last_pid;
0a7de745 1925 if (so->so_flags & SOF_DELEGATED) {
d9a64523 1926 m->m_pkthdr.tx_udp_e_pid = so->e_pid;
0a7de745 1927 } else {
d9a64523 1928 m->m_pkthdr.tx_udp_e_pid = 0;
0a7de745 1929 }
316670eb 1930
0a7de745 1931 if (ipoa.ipoa_boundif != IFSCOPE_NONE) {
316670eb 1932 ipoa.ipoa_flags |= IPOAF_BOUND_IF;
0a7de745 1933 }
316670eb 1934
0a7de745 1935 if (laddr.s_addr != INADDR_ANY) {
316670eb 1936 ipoa.ipoa_flags |= IPOAF_BOUND_SRCADDR;
0a7de745 1937 }
316670eb
A
1938
1939 inp->inp_sndinprog_cnt++;
d1ecb069 1940
2d21ac55 1941 socket_unlock(so, 0);
39236c6e 1942 error = ip_output(m, inpopts, &ro, soopts, mopts, &ipoa);
6d2010ae 1943 m = NULL;
2d21ac55 1944 socket_lock(so, 0);
0a7de745 1945 if (mopts != NULL) {
6d2010ae 1946 IMO_REMREF(mopts);
0a7de745 1947 }
1c79356b 1948
6d2010ae 1949 if (error == 0 && nstat_collect) {
fe8ab488 1950 boolean_t cell, wifi, wired;
39236c6e
A
1951
1952 if (ro.ro_rt != NULL) {
1953 cell = IFNET_IS_CELLULAR(ro.ro_rt->rt_ifp);
1954 wifi = (!cell && IFNET_IS_WIFI(ro.ro_rt->rt_ifp));
fe8ab488 1955 wired = (!wifi && IFNET_IS_WIRED(ro.ro_rt->rt_ifp));
39236c6e 1956 } else {
fe8ab488 1957 cell = wifi = wired = FALSE;
39236c6e 1958 }
fe8ab488
A
1959 INP_ADD_STAT(inp, cell, wifi, wired, txpackets, 1);
1960 INP_ADD_STAT(inp, cell, wifi, wired, txbytes, len);
5ba3f43e 1961 inp_set_activity_bitmap(inp);
6d2010ae 1962 }
316670eb
A
1963
1964 if (flowadv && (adv->code == FADV_FLOW_CONTROLLED ||
1965 adv->code == FADV_SUSPENDED)) {
39037602
A
1966 /*
1967 * return a hint to the application that
316670eb
A
1968 * the packet has been dropped
1969 */
1970 error = ENOBUFS;
1971 inp_set_fc_state(inp, adv->code);
1972 }
1973
1974 VERIFY(inp->inp_sndinprog_cnt > 0);
0a7de745 1975 if (--inp->inp_sndinprog_cnt == 0) {
316670eb 1976 inp->inp_flags &= ~(INP_FC_FEEDBACK);
cb323159
A
1977 if (inp->inp_sndingprog_waiters > 0) {
1978 wakeup(&inp->inp_sndinprog_cnt);
1979 }
0a7de745 1980 }
316670eb 1981
b0d623f7
A
1982 /* Synchronize PCB cached route */
1983 inp_route_copyin(inp, &ro);
1984
6d2010ae 1985abort:
91447636 1986 if (udp_dodisconnect) {
d1ecb069 1987 /* Always discard the cached route for unconnected socket */
39236c6e 1988 ROUTE_RELEASE(&inp->inp_route);
1c79356b 1989 in_pcbdisconnect(inp);
0a7de745 1990 inp->inp_laddr = origladdr; /* XXX rehash? */
39236c6e 1991 /* no reference needed */
316670eb 1992 inp->inp_last_outifp = origoutifp;
6d2010ae
A
1993 } else if (inp->inp_route.ro_rt != NULL) {
1994 struct rtentry *rt = inp->inp_route.ro_rt;
316670eb 1995 struct ifnet *outifp;
d1ecb069 1996
0a7de745
A
1997 if (rt->rt_flags & (RTF_MULTICAST | RTF_BROADCAST)) {
1998 rt = NULL; /* unusable */
1999 }
d9a64523
A
2000#if CONTENT_FILTER
2001 /*
2002 * Discard temporary route for cfil case
2003 */
0a7de745
A
2004 if (cfil_faddr_use) {
2005 rt = NULL; /* unusable */
2006 }
d9a64523
A
2007#endif
2008
6d2010ae
A
2009 /*
2010 * Always discard if it is a multicast or broadcast route.
2011 */
0a7de745 2012 if (rt == NULL) {
39236c6e 2013 ROUTE_RELEASE(&inp->inp_route);
0a7de745 2014 }
39236c6e 2015
6d2010ae 2016 /*
316670eb
A
2017 * If the destination route is unicast, update outifp with
2018 * that of the route interface used by IP.
6d2010ae 2019 */
3e170ce0
A
2020 if (rt != NULL &&
2021 (outifp = rt->rt_ifp) != inp->inp_last_outifp) {
2022 inp->inp_last_outifp = outifp; /* no reference needed */
2023
2024 so->so_pktheadroom = P2ROUNDUP(
0a7de745
A
2025 sizeof(struct udphdr) +
2026 sizeof(struct ip) +
2027 ifnet_hdrlen(outifp) +
2028 ifnet_mbuf_packetpreamblelen(outifp),
2029 sizeof(u_int32_t));
3e170ce0 2030 }
39236c6e
A
2031 } else {
2032 ROUTE_RELEASE(&inp->inp_route);
d1ecb069 2033 }
55e303ae 2034
39236c6e 2035 /*
fe8ab488
A
2036 * If output interface was cellular/expensive, and this socket is
2037 * denied access to it, generate an event.
39236c6e
A
2038 */
2039 if (error != 0 && (ipoa.ipoa_retflags & IPOARF_IFDENIED) &&
cb323159 2040 (INP_NO_CELLULAR(inp) || INP_NO_EXPENSIVE(inp) || INP_NO_CONSTRAINED(inp))) {
0a7de745
A
2041 soevent(so, (SO_FILT_HINT_LOCKED | SO_FILT_HINT_IFDENIED));
2042 }
39236c6e 2043
1c79356b 2044release:
39236c6e
A
2045 KERNEL_DEBUG(DBG_FNC_UDP_OUTPUT | DBG_FUNC_END, error, 0, 0, 0, 0);
2046
0a7de745 2047 if (m != NULL) {
6d2010ae 2048 m_freem(m);
0a7de745 2049 }
39236c6e 2050
0a7de745 2051 if (outif != NULL) {
39236c6e 2052 ifnet_release(outif);
0a7de745 2053 }
39236c6e 2054
d9a64523 2055#if CONTENT_FILTER
0a7de745 2056 if (cfil_tag) {
d9a64523 2057 m_tag_free(cfil_tag);
0a7de745 2058 }
d9a64523
A
2059#endif
2060
0a7de745 2061 return error;
1c79356b
A
2062}
2063
0a7de745 2064u_int32_t udp_sendspace = 9216; /* really max datagram size */
316670eb 2065/* 187 1K datagrams (approx 192 KB) */
0a7de745 2066u_int32_t udp_recvspace = 187 * (1024 +
1c79356b 2067#if INET6
0a7de745 2068 sizeof(struct sockaddr_in6)
39236c6e 2069#else /* !INET6 */
0a7de745 2070 sizeof(struct sockaddr_in)
39236c6e 2071#endif /* !INET6 */
0a7de745 2072 );
b0d623f7
A
2073
2074/* Check that the values of udp send and recv space do not exceed sb_max */
2075static int
39236c6e
A
2076sysctl_udp_sospace(struct sysctl_oid *oidp, void *arg1, int arg2,
2077 struct sysctl_req *req)
2078{
2079#pragma unused(arg1, arg2)
b0d623f7
A
2080 u_int32_t new_value = 0, *space_p = NULL;
2081 int changed = 0, error = 0;
0a7de745 2082 u_quad_t sb_effective_max = (sb_max / (MSIZE + MCLBYTES)) * MCLBYTES;
b0d623f7
A
2083
2084 switch (oidp->oid_number) {
39236c6e
A
2085 case UDPCTL_RECVSPACE:
2086 space_p = &udp_recvspace;
2087 break;
2088 case UDPCTL_MAXDGRAM:
2089 space_p = &udp_sendspace;
2090 break;
2091 default:
0a7de745 2092 return EINVAL;
b0d623f7 2093 }
0a7de745 2094 error = sysctl_io_number(req, *space_p, sizeof(u_int32_t),
39236c6e 2095 &new_value, &changed);
39037602 2096 if (changed) {
0a7de745 2097 if (new_value > 0 && new_value <= sb_effective_max) {
39037602 2098 *space_p = new_value;
0a7de745 2099 } else {
39037602 2100 error = ERANGE;
0a7de745 2101 }
39037602 2102 }
0a7de745 2103 return error;
b0d623f7
A
2104}
2105
39236c6e 2106SYSCTL_PROC(_net_inet_udp, UDPCTL_RECVSPACE, recvspace,
0a7de745
A
2107 CTLTYPE_INT | CTLFLAG_RW | CTLFLAG_LOCKED, &udp_recvspace, 0,
2108 &sysctl_udp_sospace, "IU", "Maximum incoming UDP datagram size");
b0d623f7 2109
39236c6e 2110SYSCTL_PROC(_net_inet_udp, UDPCTL_MAXDGRAM, maxdgram,
0a7de745
A
2111 CTLTYPE_INT | CTLFLAG_RW | CTLFLAG_LOCKED, &udp_sendspace, 0,
2112 &sysctl_udp_sospace, "IU", "Maximum outgoing UDP datagram size");
1c79356b 2113
5ba3f43e 2114int
1c79356b
A
2115udp_abort(struct socket *so)
2116{
2117 struct inpcb *inp;
1c79356b
A
2118
2119 inp = sotoinpcb(so);
39236c6e
A
2120 if (inp == NULL) {
2121 panic("%s: so=%p null inp\n", __func__, so);
2122 /* NOTREACHED */
2123 }
1c79356b 2124 soisdisconnected(so);
1c79356b 2125 in_pcbdetach(inp);
0a7de745 2126 return 0;
1c79356b
A
2127}
2128
5ba3f43e 2129int
39236c6e 2130udp_attach(struct socket *so, int proto, struct proc *p)
1c79356b 2131{
39236c6e 2132#pragma unused(proto)
1c79356b 2133 struct inpcb *inp;
91447636 2134 int error;
1c79356b 2135
9bccf70c 2136 inp = sotoinpcb(so);
39236c6e 2137 if (inp != NULL) {
39037602 2138 panic("%s so=%p inp=%p\n", __func__, so, inp);
39236c6e
A
2139 /* NOTREACHED */
2140 }
1c79356b 2141 error = in_pcballoc(so, &udbinfo, p);
0a7de745
A
2142 if (error != 0) {
2143 return error;
2144 }
91447636 2145 error = soreserve(so, udp_sendspace, udp_recvspace);
0a7de745
A
2146 if (error != 0) {
2147 return error;
2148 }
1c79356b
A
2149 inp = (struct inpcb *)so->so_pcb;
2150 inp->inp_vflag |= INP_IPV4;
2151 inp->inp_ip_ttl = ip_defttl;
0a7de745 2152 if (nstat_collect) {
316670eb 2153 nstat_udp_new_pcb(inp);
0a7de745
A
2154 }
2155 return 0;
1c79356b
A
2156}
2157
5ba3f43e 2158int
1c79356b
A
2159udp_bind(struct socket *so, struct sockaddr *nam, struct proc *p)
2160{
2161 struct inpcb *inp;
91447636 2162 int error;
1c79356b 2163
39236c6e 2164 if (nam->sa_family != 0 && nam->sa_family != AF_INET &&
0a7de745
A
2165 nam->sa_family != AF_INET6) {
2166 return EAFNOSUPPORT;
2167 }
39236c6e 2168
1c79356b 2169 inp = sotoinpcb(so);
0a7de745
A
2170 if (inp == NULL) {
2171 return EINVAL;
2172 }
1c79356b 2173 error = in_pcbbind(inp, nam, p);
5ba3f43e
A
2174
2175#if NECP
2176 /* Update NECP client with bind result if not in middle of connect */
2177 if (error == 0 &&
0a7de745
A
2178 (inp->inp_flags2 & INP2_CONNECT_IN_PROGRESS) &&
2179 !uuid_is_null(inp->necp_client_uuid)) {
5ba3f43e
A
2180 socket_unlock(so, 0);
2181 necp_client_assign_from_socket(so->last_pid, inp->necp_client_uuid, inp);
2182 socket_lock(so, 0);
2183 }
2184#endif /* NECP */
2185
0a7de745 2186 return error;
1c79356b
A
2187}
2188
5ba3f43e 2189int
1c79356b
A
2190udp_connect(struct socket *so, struct sockaddr *nam, struct proc *p)
2191{
2192 struct inpcb *inp;
91447636 2193 int error;
1c79356b
A
2194
2195 inp = sotoinpcb(so);
0a7de745
A
2196 if (inp == NULL) {
2197 return EINVAL;
2198 }
2199 if (inp->inp_faddr.s_addr != INADDR_ANY) {
2200 return EISCONN;
2201 }
3e170ce0 2202
5ba3f43e
A
2203 if (!(so->so_flags1 & SOF1_CONNECT_COUNTED)) {
2204 so->so_flags1 |= SOF1_CONNECT_COUNTED;
2205 INC_ATOMIC_INT64_LIM(net_api_stats.nas_socket_inet_dgram_connected);
2206 }
2207
3e170ce0
A
2208#if NECP
2209#if FLOW_DIVERT
2210 if (necp_socket_should_use_flow_divert(inp)) {
2211 uint32_t fd_ctl_unit =
2212 necp_socket_get_flow_divert_control_unit(inp);
2213 if (fd_ctl_unit > 0) {
2214 error = flow_divert_pcb_init(so, fd_ctl_unit);
2215 if (error == 0) {
2216 error = flow_divert_connect_out(so, nam, p);
2217 }
2218 } else {
2219 error = ENETDOWN;
2220 }
0a7de745 2221 return error;
3e170ce0
A
2222 }
2223#endif /* FLOW_DIVERT */
2224#endif /* NECP */
2225
39236c6e 2226 error = in_pcbconnect(inp, nam, p, IFSCOPE_NONE, NULL);
316670eb 2227 if (error == 0) {
5ba3f43e
A
2228#if NECP
2229 /* Update NECP client with connected five-tuple */
2230 if (!uuid_is_null(inp->necp_client_uuid)) {
2231 socket_unlock(so, 0);
2232 necp_client_assign_from_socket(so->last_pid, inp->necp_client_uuid, inp);
2233 socket_lock(so, 0);
2234 }
2235#endif /* NECP */
2236
1c79356b 2237 soisconnected(so);
0a7de745 2238 if (inp->inp_flowhash == 0) {
316670eb 2239 inp->inp_flowhash = inp_calc_flowhash(inp);
0a7de745 2240 }
316670eb 2241 }
0a7de745 2242 return error;
39236c6e
A
2243}
2244
2245int
813fb2f6 2246udp_connectx_common(struct socket *so, int af, struct sockaddr *src, struct sockaddr *dst,
3e170ce0
A
2247 struct proc *p, uint32_t ifscope, sae_associd_t aid, sae_connid_t *pcid,
2248 uint32_t flags, void *arg, uint32_t arglen,
2249 struct uio *uio, user_ssize_t *bytes_written)
39236c6e
A
2250{
2251#pragma unused(aid, flags, arg, arglen)
39236c6e 2252 struct inpcb *inp = sotoinpcb(so);
5ba3f43e 2253 int error = 0;
3e170ce0 2254 user_ssize_t datalen = 0;
39236c6e 2255
0a7de745
A
2256 if (inp == NULL) {
2257 return EINVAL;
2258 }
39236c6e 2259
813fb2f6 2260 VERIFY(dst != NULL);
39236c6e 2261
5ba3f43e
A
2262 ASSERT(!(inp->inp_flags2 & INP2_CONNECT_IN_PROGRESS));
2263 inp->inp_flags2 |= INP2_CONNECT_IN_PROGRESS;
2264
fe8ab488 2265#if NECP
813fb2f6 2266 inp_update_necp_policy(inp, src, dst, ifscope);
fe8ab488 2267#endif /* NECP */
39037602 2268
39236c6e
A
2269 /* bind socket to the specified interface, if requested */
2270 if (ifscope != IFSCOPE_NONE &&
0a7de745 2271 (error = inp_bindif(inp, ifscope, NULL)) != 0) {
5ba3f43e
A
2272 goto done;
2273 }
39236c6e
A
2274
2275 /* if source address and/or port is specified, bind to it */
813fb2f6 2276 if (src != NULL) {
0a7de745 2277 error = sobindlock(so, src, 0); /* already locked */
5ba3f43e
A
2278 if (error != 0) {
2279 goto done;
2280 }
39236c6e
A
2281 }
2282
2283 switch (af) {
2284 case AF_INET:
813fb2f6 2285 error = udp_connect(so, dst, p);
39236c6e
A
2286 break;
2287#if INET6
2288 case AF_INET6:
813fb2f6 2289 error = udp6_connect(so, dst, p);
39236c6e
A
2290 break;
2291#endif /* INET6 */
2292 default:
2293 VERIFY(0);
2294 /* NOTREACHED */
2295 }
2296
5ba3f43e
A
2297 if (error != 0) {
2298 goto done;
2299 }
3e170ce0
A
2300
2301 /*
2302 * If there is data, copy it. DATA_IDEMPOTENT is ignored.
39037602 2303 * CONNECT_RESUME_ON_READ_WRITE is ignored.
3e170ce0
A
2304 */
2305 if (uio != NULL) {
2306 socket_unlock(so, 0);
2307
2308 VERIFY(bytes_written != NULL);
2309
2310 datalen = uio_resid(uio);
2311 error = so->so_proto->pr_usrreqs->pru_sosend(so, NULL,
39037602 2312 (uio_t)uio, NULL, NULL, 0);
3e170ce0
A
2313 socket_lock(so, 0);
2314
2315 /* If error returned is EMSGSIZE, for example, disconnect */
0a7de745 2316 if (error == 0 || error == EWOULDBLOCK) {
3e170ce0 2317 *bytes_written = datalen - uio_resid(uio);
0a7de745 2318 } else {
39037602 2319 (void) so->so_proto->pr_usrreqs->pru_disconnectx(so,
3e170ce0 2320 SAE_ASSOCID_ANY, SAE_CONNID_ANY);
0a7de745 2321 }
3e170ce0
A
2322 /*
2323 * mask the EWOULDBLOCK error so that the caller
2324 * knows that atleast the connect was successful.
2325 */
0a7de745 2326 if (error == EWOULDBLOCK) {
3e170ce0 2327 error = 0;
0a7de745 2328 }
3e170ce0
A
2329 }
2330
0a7de745
A
2331 if (error == 0 && pcid != NULL) {
2332 *pcid = 1; /* there is only 1 connection for UDP */
2333 }
5ba3f43e
A
2334done:
2335 inp->inp_flags2 &= ~INP2_CONNECT_IN_PROGRESS;
0a7de745 2336 return error;
39236c6e
A
2337}
2338
5ba3f43e 2339int
813fb2f6
A
2340udp_connectx(struct socket *so, struct sockaddr *src,
2341 struct sockaddr *dst, struct proc *p, uint32_t ifscope,
3e170ce0
A
2342 sae_associd_t aid, sae_connid_t *pcid, uint32_t flags, void *arg,
2343 uint32_t arglen, struct uio *uio, user_ssize_t *bytes_written)
39236c6e 2344{
0a7de745
A
2345 return udp_connectx_common(so, AF_INET, src, dst,
2346 p, ifscope, aid, pcid, flags, arg, arglen, uio, bytes_written);
1c79356b
A
2347}
2348
5ba3f43e 2349int
1c79356b
A
2350udp_detach(struct socket *so)
2351{
2352 struct inpcb *inp;
1c79356b
A
2353
2354 inp = sotoinpcb(so);
39236c6e
A
2355 if (inp == NULL) {
2356 panic("%s: so=%p null inp\n", __func__, so);
2357 /* NOTREACHED */
2358 }
fe8ab488
A
2359
2360 /*
2361 * If this is a socket that does not want to wakeup the device
39037602
A
2362 * for it's traffic, the application might be waiting for
2363 * close to complete before going to sleep. Send a notification
fe8ab488
A
2364 * for this kind of sockets
2365 */
0a7de745 2366 if (so->so_options & SO_NOWAKEFROMSLEEP) {
fe8ab488 2367 socket_post_kev_msg_closed(so);
0a7de745 2368 }
fe8ab488 2369
1c79356b 2370 in_pcbdetach(inp);
91447636 2371 inp->inp_state = INPCB_STATE_DEAD;
0a7de745 2372 return 0;
1c79356b
A
2373}
2374
5ba3f43e 2375int
1c79356b
A
2376udp_disconnect(struct socket *so)
2377{
2378 struct inpcb *inp;
1c79356b
A
2379
2380 inp = sotoinpcb(so);
fe8ab488
A
2381 if (inp == NULL
2382#if NECP
0a7de745 2383 || (necp_socket_should_use_flow_divert(inp))
fe8ab488 2384#endif /* NECP */
0a7de745
A
2385 ) {
2386 return inp == NULL ? EINVAL : EPROTOTYPE;
2387 }
2388 if (inp->inp_faddr.s_addr == INADDR_ANY) {
2389 return ENOTCONN;
2390 }
1c79356b 2391
1c79356b 2392 in_pcbdisconnect(inp);
39236c6e 2393
316670eb
A
2394 /* reset flow controlled state, just in case */
2395 inp_reset_fc_state(inp);
2396
1c79356b 2397 inp->inp_laddr.s_addr = INADDR_ANY;
0a7de745 2398 so->so_state &= ~SS_ISCONNECTED; /* XXX */
316670eb 2399 inp->inp_last_outifp = NULL;
5ba3f43e 2400
0a7de745 2401 return 0;
1c79356b
A
2402}
2403
5ba3f43e 2404int
3e170ce0 2405udp_disconnectx(struct socket *so, sae_associd_t aid, sae_connid_t cid)
1c79356b 2406{
39236c6e 2407#pragma unused(cid)
0a7de745
A
2408 if (aid != SAE_ASSOCID_ANY && aid != SAE_ASSOCID_ALL) {
2409 return EINVAL;
2410 }
39236c6e 2411
0a7de745 2412 return udp_disconnect(so);
39236c6e
A
2413}
2414
5ba3f43e 2415int
39236c6e
A
2416udp_send(struct socket *so, int flags, struct mbuf *m,
2417 struct sockaddr *addr, struct mbuf *control, struct proc *p)
2418{
3e170ce0 2419#ifndef FLOW_DIVERT
39236c6e 2420#pragma unused(flags)
3e170ce0 2421#endif /* !(FLOW_DIVERT) */
1c79356b
A
2422 struct inpcb *inp;
2423
2424 inp = sotoinpcb(so);
3e170ce0 2425 if (inp == NULL) {
0a7de745 2426 if (m != NULL) {
39236c6e 2427 m_freem(m);
0a7de745
A
2428 }
2429 if (control != NULL) {
39236c6e 2430 m_freem(control);
0a7de745
A
2431 }
2432 return EINVAL;
1c79356b 2433 }
39236c6e 2434
3e170ce0
A
2435#if NECP
2436#if FLOW_DIVERT
2437 if (necp_socket_should_use_flow_divert(inp)) {
2438 /* Implicit connect */
0a7de745
A
2439 return flow_divert_implicit_data_out(so, flags, m, addr,
2440 control, p);
3e170ce0
A
2441 }
2442#endif /* FLOW_DIVERT */
2443#endif /* NECP */
2444
0a7de745 2445 return udp_output(inp, m, addr, control, p);
1c79356b
A
2446}
2447
2448int
2449udp_shutdown(struct socket *so)
2450{
2451 struct inpcb *inp;
2452
2453 inp = sotoinpcb(so);
0a7de745
A
2454 if (inp == NULL) {
2455 return EINVAL;
2456 }
1c79356b 2457 socantsendmore(so);
0a7de745 2458 return 0;
1c79356b
A
2459}
2460
91447636 2461int
b0d623f7 2462udp_lock(struct socket *so, int refcount, void *debug)
91447636 2463{
b0d623f7
A
2464 void *lr_saved;
2465
0a7de745 2466 if (debug == NULL) {
b0d623f7 2467 lr_saved = __builtin_return_address(0);
0a7de745 2468 } else {
b0d623f7 2469 lr_saved = debug;
0a7de745 2470 }
91447636 2471
39236c6e 2472 if (so->so_pcb != NULL) {
5ba3f43e 2473 LCK_MTX_ASSERT(&((struct inpcb *)so->so_pcb)->inpcb_mtx,
b0d623f7 2474 LCK_MTX_ASSERT_NOTOWNED);
6d2010ae 2475 lck_mtx_lock(&((struct inpcb *)so->so_pcb)->inpcb_mtx);
b0d623f7 2476 } else {
39236c6e 2477 panic("%s: so=%p NO PCB! lr=%p lrh= %s\n", __func__,
b0d623f7
A
2478 so, lr_saved, solockhistory_nr(so));
2479 /* NOTREACHED */
91447636 2480 }
0a7de745 2481 if (refcount) {
91447636 2482 so->so_usecount++;
0a7de745 2483 }
91447636 2484
b0d623f7 2485 so->lock_lr[so->next_lock_lr] = lr_saved;
0a7de745
A
2486 so->next_lock_lr = (so->next_lock_lr + 1) % SO_LCKDBG_MAX;
2487 return 0;
91447636
A
2488}
2489
2490int
b0d623f7 2491udp_unlock(struct socket *so, int refcount, void *debug)
91447636 2492{
b0d623f7 2493 void *lr_saved;
0c530ab8 2494
0a7de745 2495 if (debug == NULL) {
b0d623f7 2496 lr_saved = __builtin_return_address(0);
0a7de745 2497 } else {
b0d623f7 2498 lr_saved = debug;
0a7de745 2499 }
0c530ab8 2500
d190cdc3
A
2501 if (refcount) {
2502 VERIFY(so->so_usecount > 0);
91447636 2503 so->so_usecount--;
d190cdc3 2504 }
b0d623f7 2505 if (so->so_pcb == NULL) {
39236c6e 2506 panic("%s: so=%p NO PCB! lr=%p lrh= %s\n", __func__,
b0d623f7
A
2507 so, lr_saved, solockhistory_nr(so));
2508 /* NOTREACHED */
2509 } else {
5ba3f43e 2510 LCK_MTX_ASSERT(&((struct inpcb *)so->so_pcb)->inpcb_mtx,
b0d623f7
A
2511 LCK_MTX_ASSERT_OWNED);
2512 so->unlock_lr[so->next_unlock_lr] = lr_saved;
0a7de745 2513 so->next_unlock_lr = (so->next_unlock_lr + 1) % SO_LCKDBG_MAX;
6d2010ae 2514 lck_mtx_unlock(&((struct inpcb *)so->so_pcb)->inpcb_mtx);
91447636 2515 }
0a7de745 2516 return 0;
91447636
A
2517}
2518
2519lck_mtx_t *
5ba3f43e 2520udp_getlock(struct socket *so, int flags)
91447636 2521{
5ba3f43e 2522#pragma unused(flags)
91447636
A
2523 struct inpcb *inp = sotoinpcb(so);
2524
39236c6e
A
2525 if (so->so_pcb == NULL) {
2526 panic("%s: so=%p NULL so_pcb lrh= %s\n", __func__,
2527 so, solockhistory_nr(so));
2528 /* NOTREACHED */
91447636 2529 }
0a7de745 2530 return &inp->inpcb_mtx;
91447636
A
2531}
2532
39236c6e
A
2533/*
2534 * UDP garbage collector callback (inpcb_timer_func_t).
2535 *
2536 * Returns > 0 to keep timer active.
2537 */
2538static void
2539udp_gc(struct inpcbinfo *ipi)
91447636
A
2540{
2541 struct inpcb *inp, *inpnxt;
2542 struct socket *so;
91447636 2543
39236c6e 2544 if (lck_rw_try_lock_exclusive(ipi->ipi_lock) == FALSE) {
2d21ac55
A
2545 if (udp_gc_done == TRUE) {
2546 udp_gc_done = FALSE;
39236c6e
A
2547 /* couldn't get the lock, must lock next time */
2548 atomic_add_32(&ipi->ipi_gc_req.intimer_fast, 1);
2549 return;
2d21ac55 2550 }
39236c6e 2551 lck_rw_lock_exclusive(ipi->ipi_lock);
2d21ac55
A
2552 }
2553
2554 udp_gc_done = TRUE;
91447636
A
2555
2556 for (inp = udb.lh_first; inp != NULL; inp = inpnxt) {
2557 inpnxt = inp->inp_list.le_next;
91447636 2558
39236c6e
A
2559 /*
2560 * Skip unless it's STOPUSING; garbage collector will
2561 * be triggered by in_pcb_checkstate() upon setting
2562 * wantcnt to that value. If the PCB is already dead,
2563 * keep gc active to anticipate wantcnt changing.
2564 */
0a7de745 2565 if (inp->inp_wantcnt != WNT_STOPUSING) {
91447636 2566 continue;
0a7de745 2567 }
91447636 2568
39236c6e
A
2569 /*
2570 * Skip if busy, no hurry for cleanup. Keep gc active
2571 * and try the lock again during next round.
2572 */
5ba3f43e 2573 if (!socket_try_lock(inp->inp_socket)) {
39236c6e 2574 atomic_add_32(&ipi->ipi_gc_req.intimer_fast, 1);
91447636 2575 continue;
39236c6e 2576 }
91447636 2577
39236c6e
A
2578 /*
2579 * Keep gc active unless usecount is 0.
2580 */
2581 so = inp->inp_socket;
b0d623f7
A
2582 if (so->so_usecount == 0) {
2583 if (inp->inp_state != INPCB_STATE_DEAD) {
2584#if INET6
0a7de745 2585 if (SOCK_CHECK_DOM(so, PF_INET6)) {
b0d623f7 2586 in6_pcbdetach(inp);
0a7de745 2587 } else
b0d623f7 2588#endif /* INET6 */
0a7de745 2589 in_pcbdetach(inp);
b0d623f7 2590 }
91447636 2591 in_pcbdispose(inp);
b0d623f7 2592 } else {
5ba3f43e 2593 socket_unlock(so, 0);
39236c6e 2594 atomic_add_32(&ipi->ipi_gc_req.intimer_fast, 1);
b0d623f7 2595 }
91447636 2596 }
39236c6e 2597 lck_rw_done(ipi->ipi_lock);
91447636
A
2598}
2599
39236c6e
A
2600static int
2601udp_getstat SYSCTL_HANDLER_ARGS
91447636 2602{
39236c6e 2603#pragma unused(oidp, arg1, arg2)
0a7de745
A
2604 if (req->oldptr == USER_ADDR_NULL) {
2605 req->oldlen = (size_t)sizeof(struct udpstat);
2606 }
39236c6e 2607
0a7de745 2608 return SYSCTL_OUT(req, &udpstat, MIN(sizeof(udpstat), req->oldlen));
91447636
A
2609}
2610
2d21ac55
A
2611void
2612udp_in_cksum_stats(u_int32_t len)
2613{
39236c6e
A
2614 udpstat.udps_rcv_swcsum++;
2615 udpstat.udps_rcv_swcsum_bytes += len;
2d21ac55
A
2616}
2617
2618void
2619udp_out_cksum_stats(u_int32_t len)
2620{
39236c6e
A
2621 udpstat.udps_snd_swcsum++;
2622 udpstat.udps_snd_swcsum_bytes += len;
2623}
2624
2625#if INET6
2626void
2627udp_in6_cksum_stats(u_int32_t len)
2628{
2629 udpstat.udps_rcv6_swcsum++;
2630 udpstat.udps_rcv6_swcsum_bytes += len;
2631}
2632
2633void
2634udp_out6_cksum_stats(u_int32_t len)
2635{
2636 udpstat.udps_snd6_swcsum++;
2637 udpstat.udps_snd6_swcsum_bytes += len;
2638}
2639#endif /* INET6 */
2640
2641/*
2642 * Checksum extended UDP header and data.
2643 */
2644static int
2645udp_input_checksum(struct mbuf *m, struct udphdr *uh, int off, int ulen)
2646{
2647 struct ifnet *ifp = m->m_pkthdr.rcvif;
2648 struct ip *ip = mtod(m, struct ip *);
2649 struct ipovly *ipov = (struct ipovly *)ip;
2650
2651 if (uh->uh_sum == 0) {
2652 udpstat.udps_nosum++;
0a7de745 2653 return 0;
39236c6e
A
2654 }
2655
5ba3f43e 2656 /* ip_stripoptions() must have been called before we get here */
0a7de745 2657 ASSERT((ip->ip_hl << 2) == sizeof(*ip));
5ba3f43e 2658
39236c6e
A
2659 if ((hwcksum_rx || (ifp->if_flags & IFF_LOOPBACK) ||
2660 (m->m_pkthdr.pkt_flags & PKTF_LOOP)) &&
2661 (m->m_pkthdr.csum_flags & CSUM_DATA_VALID)) {
2662 if (m->m_pkthdr.csum_flags & CSUM_PSEUDO_HDR) {
2663 uh->uh_sum = m->m_pkthdr.csum_rx_val;
2664 } else {
5ba3f43e
A
2665 uint32_t sum = m->m_pkthdr.csum_rx_val;
2666 uint32_t start = m->m_pkthdr.csum_rx_start;
2667 int32_t trailer = (m_pktlen(m) - (off + ulen));
39236c6e
A
2668
2669 /*
2670 * Perform 1's complement adjustment of octets
2671 * that got included/excluded in the hardware-
2672 * calculated checksum value. Ignore cases
5ba3f43e 2673 * where the value already includes the entire
39236c6e 2674 * IP header span, as the sum for those octets
5ba3f43e
A
2675 * would already be 0 by the time we get here;
2676 * IP has already performed its header checksum
2677 * checks. If we do need to adjust, restore
2678 * the original fields in the IP header when
2679 * computing the adjustment value. Also take
2680 * care of any trailing bytes and subtract out
2681 * their partial sum.
39236c6e 2682 */
5ba3f43e 2683 ASSERT(trailer >= 0);
39236c6e 2684 if ((m->m_pkthdr.csum_flags & CSUM_PARTIAL) &&
5ba3f43e
A
2685 ((start != 0 && start != off) || trailer != 0)) {
2686 uint32_t swbytes = (uint32_t)trailer;
2687
39236c6e 2688 if (start < off) {
0a7de745 2689 ip->ip_len += sizeof(*ip);
5ba3f43e 2690#if BYTE_ORDER != BIG_ENDIAN
39236c6e
A
2691 HTONS(ip->ip_len);
2692 HTONS(ip->ip_off);
39236c6e 2693#endif /* BYTE_ORDER != BIG_ENDIAN */
5ba3f43e 2694 }
39236c6e 2695 /* callee folds in sum */
5ba3f43e 2696 sum = m_adj_sum16(m, start, off, ulen, sum);
0a7de745 2697 if (off > start) {
5ba3f43e 2698 swbytes += (off - start);
0a7de745 2699 } else {
5ba3f43e 2700 swbytes += (start - off);
0a7de745 2701 }
5ba3f43e 2702
39236c6e 2703 if (start < off) {
5ba3f43e 2704#if BYTE_ORDER != BIG_ENDIAN
39236c6e
A
2705 NTOHS(ip->ip_off);
2706 NTOHS(ip->ip_len);
39236c6e 2707#endif /* BYTE_ORDER != BIG_ENDIAN */
0a7de745 2708 ip->ip_len -= sizeof(*ip);
5ba3f43e
A
2709 }
2710
0a7de745 2711 if (swbytes != 0) {
5ba3f43e 2712 udp_in_cksum_stats(swbytes);
0a7de745
A
2713 }
2714 if (trailer != 0) {
5ba3f43e 2715 m_adj(m, -trailer);
0a7de745 2716 }
39236c6e
A
2717 }
2718
2719 /* callee folds in sum */
2720 uh->uh_sum = in_pseudo(ip->ip_src.s_addr,
2721 ip->ip_dst.s_addr, sum + htonl(ulen + IPPROTO_UDP));
2722 }
2723 uh->uh_sum ^= 0xffff;
2724 } else {
2725 uint16_t ip_sum;
2726 char b[9];
2727
0a7de745
A
2728 bcopy(ipov->ih_x1, b, sizeof(ipov->ih_x1));
2729 bzero(ipov->ih_x1, sizeof(ipov->ih_x1));
39236c6e
A
2730 ip_sum = ipov->ih_len;
2731 ipov->ih_len = uh->uh_ulen;
0a7de745
A
2732 uh->uh_sum = in_cksum(m, ulen + sizeof(struct ip));
2733 bcopy(b, ipov->ih_x1, sizeof(ipov->ih_x1));
39236c6e
A
2734 ipov->ih_len = ip_sum;
2735
2736 udp_in_cksum_stats(ulen);
2737 }
2738
2739 if (uh->uh_sum != 0) {
2740 udpstat.udps_badsum++;
2741 IF_UDP_STATINC(ifp, badchksum);
0a7de745 2742 return -1;
39236c6e
A
2743 }
2744
0a7de745 2745 return 0;
2d21ac55 2746}
3e170ce0 2747
3e170ce0
A
2748void
2749udp_fill_keepalive_offload_frames(ifnet_t ifp,
2750 struct ifnet_keepalive_offload_frame *frames_array,
2751 u_int32_t frames_array_count, size_t frame_data_offset,
2752 u_int32_t *used_frames_count)
2753{
2754 struct inpcb *inp;
2755 inp_gen_t gencnt;
2756 u_int32_t frame_index = *used_frames_count;
2757
2758 if (ifp == NULL || frames_array == NULL ||
2759 frames_array_count == 0 ||
2760 frame_index >= frames_array_count ||
0a7de745 2761 frame_data_offset >= IFNET_KEEPALIVE_OFFLOAD_FRAME_DATA_SIZE) {
3e170ce0 2762 return;
0a7de745 2763 }
3e170ce0
A
2764
2765 lck_rw_lock_shared(udbinfo.ipi_lock);
2766 gencnt = udbinfo.ipi_gencnt;
2767 LIST_FOREACH(inp, udbinfo.ipi_listhead, inp_list) {
2768 struct socket *so;
2769 u_int8_t *data;
2770 struct ifnet_keepalive_offload_frame *frame;
2771 struct mbuf *m = NULL;
2772
0a7de745 2773 if (frame_index >= frames_array_count) {
3e170ce0 2774 break;
0a7de745 2775 }
3e170ce0
A
2776
2777 if (inp->inp_gencnt > gencnt ||
0a7de745 2778 inp->inp_state == INPCB_STATE_DEAD) {
3e170ce0 2779 continue;
0a7de745 2780 }
3e170ce0
A
2781
2782 if ((so = inp->inp_socket) == NULL ||
0a7de745 2783 (so->so_state & SS_DEFUNCT)) {
3e170ce0 2784 continue;
0a7de745 2785 }
3e170ce0
A
2786 /*
2787 * check for keepalive offload flag without socket
2788 * lock to avoid a deadlock
2789 */
2790 if (!(inp->inp_flags2 & INP2_KEEPALIVE_OFFLOAD)) {
2791 continue;
2792 }
2793
2794 udp_lock(so, 1, 0);
2795 if (!(inp->inp_vflag & (INP_IPV4 | INP_IPV6))) {
2796 udp_unlock(so, 1, 0);
2797 continue;
2798 }
2799 if ((inp->inp_vflag & INP_IPV4) &&
2800 (inp->inp_laddr.s_addr == INADDR_ANY ||
2801 inp->inp_faddr.s_addr == INADDR_ANY)) {
2802 udp_unlock(so, 1, 0);
2803 continue;
2804 }
2805 if ((inp->inp_vflag & INP_IPV6) &&
2806 (IN6_IS_ADDR_UNSPECIFIED(&inp->in6p_laddr) ||
2807 IN6_IS_ADDR_UNSPECIFIED(&inp->in6p_faddr))) {
2808 udp_unlock(so, 1, 0);
2809 continue;
2810 }
2811 if (inp->inp_lport == 0 || inp->inp_fport == 0) {
2812 udp_unlock(so, 1, 0);
2813 continue;
2814 }
2815 if (inp->inp_last_outifp == NULL ||
2816 inp->inp_last_outifp->if_index != ifp->if_index) {
2817 udp_unlock(so, 1, 0);
2818 continue;
2819 }
2820 if ((inp->inp_vflag & INP_IPV4)) {
39037602 2821 if ((frame_data_offset + sizeof(struct udpiphdr) +
3e170ce0
A
2822 inp->inp_keepalive_datalen) >
2823 IFNET_KEEPALIVE_OFFLOAD_FRAME_DATA_SIZE) {
2824 udp_unlock(so, 1, 0);
2825 continue;
2826 }
2827 if ((sizeof(struct udpiphdr) +
2828 inp->inp_keepalive_datalen) > _MHLEN) {
2829 udp_unlock(so, 1, 0);
2830 continue;
2831 }
2832 } else {
2833 if ((frame_data_offset + sizeof(struct ip6_hdr) +
2834 sizeof(struct udphdr) +
2835 inp->inp_keepalive_datalen) >
2836 IFNET_KEEPALIVE_OFFLOAD_FRAME_DATA_SIZE) {
2837 udp_unlock(so, 1, 0);
2838 continue;
2839 }
2840 if ((sizeof(struct ip6_hdr) + sizeof(struct udphdr) +
2841 inp->inp_keepalive_datalen) > _MHLEN) {
2842 udp_unlock(so, 1, 0);
2843 continue;
2844 }
2845 }
2846 MGETHDR(m, M_WAIT, MT_HEADER);
2847 if (m == NULL) {
2848 udp_unlock(so, 1, 0);
2849 continue;
2850 }
2851 /*
2852 * This inp has all the information that is needed to
2853 * generate an offload frame.
2854 */
2855 if (inp->inp_vflag & INP_IPV4) {
2856 struct ip *ip;
2857 struct udphdr *udp;
2858
2859 frame = &frames_array[frame_index];
2860 frame->length = frame_data_offset +
2861 sizeof(struct udpiphdr) +
2862 inp->inp_keepalive_datalen;
2863 frame->ether_type =
2864 IFNET_KEEPALIVE_OFFLOAD_FRAME_ETHERTYPE_IPV4;
2865 frame->interval = inp->inp_keepalive_interval;
2866 switch (inp->inp_keepalive_type) {
2867 case UDP_KEEPALIVE_OFFLOAD_TYPE_AIRPLAY:
2868 frame->type =
2869 IFNET_KEEPALIVE_OFFLOAD_FRAME_AIRPLAY;
2870 break;
2871 default:
2872 break;
2873 }
2874 data = mtod(m, u_int8_t *);
2875 bzero(data, sizeof(struct udpiphdr));
2876 ip = (__typeof__(ip))(void *)data;
2877 udp = (__typeof__(udp))(void *) (data +
2878 sizeof(struct ip));
2879 m->m_len = sizeof(struct udpiphdr);
2880 data = data + sizeof(struct udpiphdr);
2881 if (inp->inp_keepalive_datalen > 0 &&
2882 inp->inp_keepalive_data != NULL) {
2883 bcopy(inp->inp_keepalive_data, data,
2884 inp->inp_keepalive_datalen);
2885 m->m_len += inp->inp_keepalive_datalen;
2886 }
2887 m->m_pkthdr.len = m->m_len;
2888
2889 ip->ip_v = IPVERSION;
2890 ip->ip_hl = (sizeof(struct ip) >> 2);
2891 ip->ip_p = IPPROTO_UDP;
2892 ip->ip_len = htons(sizeof(struct udpiphdr) +
2893 (u_short)inp->inp_keepalive_datalen);
2894 ip->ip_ttl = inp->inp_ip_ttl;
813fb2f6 2895 ip->ip_tos |= (inp->inp_ip_tos & ~IPTOS_ECN_MASK);
3e170ce0
A
2896 ip->ip_src = inp->inp_laddr;
2897 ip->ip_dst = inp->inp_faddr;
2898 ip->ip_sum = in_cksum_hdr_opt(ip);
2899
2900 udp->uh_sport = inp->inp_lport;
2901 udp->uh_dport = inp->inp_fport;
2902 udp->uh_ulen = htons(sizeof(struct udphdr) +
2903 (u_short)inp->inp_keepalive_datalen);
2904
2905 if (!(inp->inp_flags & INP_UDP_NOCKSUM)) {
2906 udp->uh_sum = in_pseudo(ip->ip_src.s_addr,
2907 ip->ip_dst.s_addr,
2908 htons(sizeof(struct udphdr) +
2909 (u_short)inp->inp_keepalive_datalen +
2910 IPPROTO_UDP));
5ba3f43e 2911 m->m_pkthdr.csum_flags =
0a7de745 2912 (CSUM_UDP | CSUM_ZERO_INVERT);
3e170ce0
A
2913 m->m_pkthdr.csum_data = offsetof(struct udphdr,
2914 uh_sum);
2915 }
2916 m->m_pkthdr.pkt_proto = IPPROTO_UDP;
2917 in_delayed_cksum(m);
2918 bcopy(m->m_data, frame->data + frame_data_offset,
2919 m->m_len);
2920 } else {
2921 struct ip6_hdr *ip6;
2922 struct udphdr *udp6;
2923
2924 VERIFY(inp->inp_vflag & INP_IPV6);
2925 frame = &frames_array[frame_index];
2926 frame->length = frame_data_offset +
2927 sizeof(struct ip6_hdr) +
2928 sizeof(struct udphdr) +
2929 inp->inp_keepalive_datalen;
2930 frame->ether_type =
2931 IFNET_KEEPALIVE_OFFLOAD_FRAME_ETHERTYPE_IPV6;
2932 frame->interval = inp->inp_keepalive_interval;
2933 switch (inp->inp_keepalive_type) {
2934 case UDP_KEEPALIVE_OFFLOAD_TYPE_AIRPLAY:
2935 frame->type =
2936 IFNET_KEEPALIVE_OFFLOAD_FRAME_AIRPLAY;
2937 break;
2938 default:
2939 break;
2940 }
2941 data = mtod(m, u_int8_t *);
2942 bzero(data, sizeof(struct ip6_hdr) + sizeof(struct udphdr));
2943 ip6 = (__typeof__(ip6))(void *)data;
2944 udp6 = (__typeof__(udp6))(void *)(data +
2945 sizeof(struct ip6_hdr));
2946 m->m_len = sizeof(struct ip6_hdr) +
2947 sizeof(struct udphdr);
2948 data = data + (sizeof(struct ip6_hdr) +
2949 sizeof(struct udphdr));
2950 if (inp->inp_keepalive_datalen > 0 &&
2951 inp->inp_keepalive_data != NULL) {
2952 bcopy(inp->inp_keepalive_data, data,
2953 inp->inp_keepalive_datalen);
2954 m->m_len += inp->inp_keepalive_datalen;
2955 }
2956 m->m_pkthdr.len = m->m_len;
2957 ip6->ip6_flow = inp->inp_flow & IPV6_FLOWINFO_MASK;
813fb2f6 2958 ip6->ip6_flow = ip6->ip6_flow & ~IPV6_FLOW_ECN_MASK;
3e170ce0
A
2959 ip6->ip6_vfc &= ~IPV6_VERSION_MASK;
2960 ip6->ip6_vfc |= IPV6_VERSION;
2961 ip6->ip6_nxt = IPPROTO_UDP;
2962 ip6->ip6_hlim = ip6_defhlim;
2963 ip6->ip6_plen = htons(sizeof(struct udphdr) +
2964 (u_short)inp->inp_keepalive_datalen);
2965 ip6->ip6_src = inp->in6p_laddr;
0a7de745 2966 if (IN6_IS_SCOPE_EMBED(&ip6->ip6_src)) {
3e170ce0 2967 ip6->ip6_src.s6_addr16[1] = 0;
0a7de745 2968 }
3e170ce0
A
2969
2970 ip6->ip6_dst = inp->in6p_faddr;
0a7de745 2971 if (IN6_IS_SCOPE_EMBED(&ip6->ip6_dst)) {
3e170ce0 2972 ip6->ip6_dst.s6_addr16[1] = 0;
0a7de745 2973 }
3e170ce0
A
2974
2975 udp6->uh_sport = inp->in6p_lport;
2976 udp6->uh_dport = inp->in6p_fport;
2977 udp6->uh_ulen = htons(sizeof(struct udphdr) +
2978 (u_short)inp->inp_keepalive_datalen);
2979 if (!(inp->inp_flags & INP_UDP_NOCKSUM)) {
2980 udp6->uh_sum = in6_pseudo(&ip6->ip6_src,
2981 &ip6->ip6_dst,
2982 htonl(sizeof(struct udphdr) +
2983 (u_short)inp->inp_keepalive_datalen +
2984 IPPROTO_UDP));
5ba3f43e 2985 m->m_pkthdr.csum_flags =
0a7de745 2986 (CSUM_UDPIPV6 | CSUM_ZERO_INVERT);
3e170ce0
A
2987 m->m_pkthdr.csum_data = offsetof(struct udphdr,
2988 uh_sum);
2989 }
2990 m->m_pkthdr.pkt_proto = IPPROTO_UDP;
2991 in6_delayed_cksum(m);
2992 bcopy(m->m_data, frame->data + frame_data_offset,
2993 m->m_len);
2994 }
2995 if (m != NULL) {
2996 m_freem(m);
2997 m = NULL;
2998 }
2999 frame_index++;
3000 udp_unlock(so, 1, 0);
3001 }
3002 lck_rw_done(udbinfo.ipi_lock);
3003 *used_frames_count = frame_index;
3004}