xref: /freebsd/sys/netinet/udp_usrreq.c (revision e7d02be19d40063783d6b8f1ff2bc4c7170fd434)
1c398230bSWarner Losh /*-
251369649SPedro F. Giffuni  * SPDX-License-Identifier: BSD-3-Clause
351369649SPedro F. Giffuni  *
46dfab5b1SGarrett Wollman  * Copyright (c) 1982, 1986, 1988, 1990, 1993, 1995
53329b236SRobert Watson  *	The Regents of the University of California.
63144b7d3SRobert Watson  * Copyright (c) 2008 Robert N. M. Watson
7fa046d87SRobert Watson  * Copyright (c) 2010-2011 Juniper Networks, Inc.
8e06e816fSKevin Lo  * Copyright (c) 2014 Kevin Lo
93329b236SRobert Watson  * All rights reserved.
10df8bae1dSRodney W. Grimes  *
11fa046d87SRobert Watson  * Portions of this software were developed by Robert N. M. Watson under
12fa046d87SRobert Watson  * contract to Juniper Networks, Inc.
13fa046d87SRobert Watson  *
14df8bae1dSRodney W. Grimes  * Redistribution and use in source and binary forms, with or without
15df8bae1dSRodney W. Grimes  * modification, are permitted provided that the following conditions
16df8bae1dSRodney W. Grimes  * are met:
17df8bae1dSRodney W. Grimes  * 1. Redistributions of source code must retain the above copyright
18df8bae1dSRodney W. Grimes  *    notice, this list of conditions and the following disclaimer.
19df8bae1dSRodney W. Grimes  * 2. Redistributions in binary form must reproduce the above copyright
20df8bae1dSRodney W. Grimes  *    notice, this list of conditions and the following disclaimer in the
21df8bae1dSRodney W. Grimes  *    documentation and/or other materials provided with the distribution.
22fbbd9655SWarner Losh  * 3. Neither the name of the University nor the names of its contributors
23df8bae1dSRodney W. Grimes  *    may be used to endorse or promote products derived from this software
24df8bae1dSRodney W. Grimes  *    without specific prior written permission.
25df8bae1dSRodney W. Grimes  *
26df8bae1dSRodney W. Grimes  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
27df8bae1dSRodney W. Grimes  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
28df8bae1dSRodney W. Grimes  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
29df8bae1dSRodney W. Grimes  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
30df8bae1dSRodney W. Grimes  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
31df8bae1dSRodney W. Grimes  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
32df8bae1dSRodney W. Grimes  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
33df8bae1dSRodney W. Grimes  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
34df8bae1dSRodney W. Grimes  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
35df8bae1dSRodney W. Grimes  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
36df8bae1dSRodney W. Grimes  * SUCH DAMAGE.
37df8bae1dSRodney W. Grimes  *
386dfab5b1SGarrett Wollman  *	@(#)udp_usrreq.c	8.6 (Berkeley) 5/23/95
39df8bae1dSRodney W. Grimes  */
40df8bae1dSRodney W. Grimes 
414b421e2dSMike Silbersack #include <sys/cdefs.h>
424b421e2dSMike Silbersack __FBSDID("$FreeBSD$");
434b421e2dSMike Silbersack 
4479288c11SBjoern A. Zeeb #include "opt_inet.h"
45cfa1ca9dSYoshinobu Inoue #include "opt_inet6.h"
46f5514f08SRobert Watson #include "opt_ipsec.h"
470c325f53SAlexander V. Chernikov #include "opt_route.h"
489d3ddf43SAdrian Chadd #include "opt_rss.h"
49cfa1ca9dSYoshinobu Inoue 
50df8bae1dSRodney W. Grimes #include <sys/param.h>
51960ed29cSSeigo Tanimura #include <sys/domain.h>
524f590175SPaul Saab #include <sys/eventhandler.h>
53960ed29cSSeigo Tanimura #include <sys/jail.h>
54b110a8a2SGarrett Wollman #include <sys/kernel.h>
55960ed29cSSeigo Tanimura #include <sys/lock.h>
56df8bae1dSRodney W. Grimes #include <sys/malloc.h>
57df8bae1dSRodney W. Grimes #include <sys/mbuf.h>
58acd3428bSRobert Watson #include <sys/priv.h>
59490d50b6SBrian Feldman #include <sys/proc.h>
60df8bae1dSRodney W. Grimes #include <sys/protosw.h>
6157f60867SMark Johnston #include <sys/sdt.h>
62960ed29cSSeigo Tanimura #include <sys/signalvar.h>
63df8bae1dSRodney W. Grimes #include <sys/socket.h>
64df8bae1dSRodney W. Grimes #include <sys/socketvar.h>
65960ed29cSSeigo Tanimura #include <sys/sx.h>
66b5e8ce9fSBruce Evans #include <sys/sysctl.h>
67816a3d83SPoul-Henning Kamp #include <sys/syslog.h>
68f5514f08SRobert Watson #include <sys/systm.h>
698781d8e9SBruce Evans 
7069c2d429SJeff Roberson #include <vm/uma.h>
71df8bae1dSRodney W. Grimes 
72df8bae1dSRodney W. Grimes #include <net/if.h>
7376039bc8SGleb Smirnoff #include <net/if_var.h>
74df8bae1dSRodney W. Grimes #include <net/route.h>
75983066f0SAlexander V. Chernikov #include <net/route/nhop.h>
76b2bdc62aSAdrian Chadd #include <net/rss_config.h>
77df8bae1dSRodney W. Grimes 
78df8bae1dSRodney W. Grimes #include <netinet/in.h>
7957f60867SMark Johnston #include <netinet/in_kdtrace.h>
800c325f53SAlexander V. Chernikov #include <netinet/in_fib.h>
81960ed29cSSeigo Tanimura #include <netinet/in_pcb.h>
82f5514f08SRobert Watson #include <netinet/in_systm.h>
83960ed29cSSeigo Tanimura #include <netinet/in_var.h>
84df8bae1dSRodney W. Grimes #include <netinet/ip.h>
85cfa1ca9dSYoshinobu Inoue #ifdef INET6
86cfa1ca9dSYoshinobu Inoue #include <netinet/ip6.h>
87cfa1ca9dSYoshinobu Inoue #endif
88960ed29cSSeigo Tanimura #include <netinet/ip_icmp.h>
89960ed29cSSeigo Tanimura #include <netinet/icmp_var.h>
90df8bae1dSRodney W. Grimes #include <netinet/ip_var.h>
91ef39adf0SAndre Oppermann #include <netinet/ip_options.h>
92cfa1ca9dSYoshinobu Inoue #ifdef INET6
93cfa1ca9dSYoshinobu Inoue #include <netinet6/ip6_var.h>
94cfa1ca9dSYoshinobu Inoue #endif
95df8bae1dSRodney W. Grimes #include <netinet/udp.h>
96df8bae1dSRodney W. Grimes #include <netinet/udp_var.h>
97e06e816fSKevin Lo #include <netinet/udplite.h>
988ad1a83bSAdrian Chadd #include <netinet/in_rss.h>
99df8bae1dSRodney W. Grimes 
100fcf59617SAndrey V. Elsukov #include <netipsec/ipsec_support.h>
101b9234fafSSam Leffler 
102db4f9cc7SJonathan Lemon #include <machine/in_cksum.h>
103db4f9cc7SJonathan Lemon 
104aed55708SRobert Watson #include <security/mac/mac_framework.h>
105aed55708SRobert Watson 
106df8bae1dSRodney W. Grimes /*
107e06e816fSKevin Lo  * UDP and UDP-Lite protocols implementation.
108df8bae1dSRodney W. Grimes  * Per RFC 768, August, 1980.
109e06e816fSKevin Lo  * Per RFC 3828, July, 2004.
110df8bae1dSRodney W. Grimes  */
11174eb3236SWarner Losh 
11274eb3236SWarner Losh /*
1133329b236SRobert Watson  * BSD 4.2 defaulted the udp checksum to be off.  Turning off udp checksums
1143329b236SRobert Watson  * removes the only data integrity mechanism for packets and malformed
115f5514f08SRobert Watson  * packets that would otherwise be discarded due to bad checksums, and may
116f5514f08SRobert Watson  * cause problems (especially for NFS data blocks).
11774eb3236SWarner Losh  */
11840b676beSBjoern A. Zeeb VNET_DEFINE(int, udp_cksum) = 1;
1196df8a710SGleb Smirnoff SYSCTL_INT(_net_inet_udp, UDPCTL_CHECKSUM, checksum, CTLFLAG_VNET | CTLFLAG_RW,
12040b676beSBjoern A. Zeeb     &VNET_NAME(udp_cksum), 0, "compute udp checksum");
121df8bae1dSRodney W. Grimes 
122334fc582SBjoern A. Zeeb VNET_DEFINE(int, udp_log_in_vain) = 0;
123334fc582SBjoern A. Zeeb SYSCTL_INT(_net_inet_udp, OID_AUTO, log_in_vain, CTLFLAG_VNET | CTLFLAG_RW,
124334fc582SBjoern A. Zeeb     &VNET_NAME(udp_log_in_vain), 0, "Log all incoming UDP packets");
125816a3d83SPoul-Henning Kamp 
12682cea7e6SBjoern A. Zeeb VNET_DEFINE(int, udp_blackhole) = 0;
1276df8a710SGleb Smirnoff SYSCTL_INT(_net_inet_udp, OID_AUTO, blackhole, CTLFLAG_VNET | CTLFLAG_RW,
128eddfbb76SRobert Watson     &VNET_NAME(udp_blackhole), 0,
1293329b236SRobert Watson     "Do not send port unreachables for refused connects");
1303ea9a7cfSGleb Smirnoff VNET_DEFINE(bool, udp_blackhole_local) = false;
1313ea9a7cfSGleb Smirnoff SYSCTL_BOOL(_net_inet_udp, OID_AUTO, blackhole_local, CTLFLAG_VNET |
1323ea9a7cfSGleb Smirnoff     CTLFLAG_RW, &VNET_NAME(udp_blackhole_local), false,
1333ea9a7cfSGleb Smirnoff     "Enforce net.inet.udp.blackhole for locally originated packets");
13416f7f31fSGeoff Rehmet 
13543bbb6aaSRobert Watson u_long	udp_sendspace = 9216;		/* really max datagram size */
13643bbb6aaSRobert Watson SYSCTL_ULONG(_net_inet_udp, UDPCTL_MAXDGRAM, maxdgram, CTLFLAG_RW,
13743bbb6aaSRobert Watson     &udp_sendspace, 0, "Maximum outgoing UDP datagram size");
13843bbb6aaSRobert Watson 
13943bbb6aaSRobert Watson u_long	udp_recvspace = 40 * (1024 +
14043bbb6aaSRobert Watson #ifdef INET6
14143bbb6aaSRobert Watson 				      sizeof(struct sockaddr_in6)
14243bbb6aaSRobert Watson #else
14343bbb6aaSRobert Watson 				      sizeof(struct sockaddr_in)
14443bbb6aaSRobert Watson #endif
145e62b9bcaSSergey Kandaurov 				      );	/* 40 1K datagrams */
14643bbb6aaSRobert Watson 
14743bbb6aaSRobert Watson SYSCTL_ULONG(_net_inet_udp, UDPCTL_RECVSPACE, recvspace, CTLFLAG_RW,
14843bbb6aaSRobert Watson     &udp_recvspace, 0, "Maximum space for incoming UDP datagrams");
14943bbb6aaSRobert Watson 
150eddfbb76SRobert Watson VNET_DEFINE(struct inpcbinfo, udbinfo);
151e06e816fSKevin Lo VNET_DEFINE(struct inpcbinfo, ulitecbinfo);
1525f901c92SAndrew Turner VNET_DEFINE_STATIC(uma_zone_t, udpcb_zone);
1531e77c105SRobert Watson #define	V_udpcb_zone			VNET(udpcb_zone)
15415bd2b43SDavid Greenman 
15515bd2b43SDavid Greenman #ifndef UDBHASHSIZE
156e2ed8f35SAlexander Motin #define	UDBHASHSIZE	128
15715bd2b43SDavid Greenman #endif
15815bd2b43SDavid Greenman 
1595b7cb97cSAndrey V. Elsukov VNET_PCPUSTAT_DEFINE(struct udpstat, udpstat);		/* from udp_var.h */
1605b7cb97cSAndrey V. Elsukov VNET_PCPUSTAT_SYSINIT(udpstat);
1615b7cb97cSAndrey V. Elsukov SYSCTL_VNET_PCPUSTAT(_net_inet_udp, UDPCTL_STATS, stats, struct udpstat,
1625b7cb97cSAndrey V. Elsukov     udpstat, "UDP statistics (struct udpstat, netinet/udp_var.h)");
163f2ea20e6SGarrett Wollman 
1645b7cb97cSAndrey V. Elsukov #ifdef VIMAGE
1655b7cb97cSAndrey V. Elsukov VNET_PCPUSTAT_SYSUNINIT(udpstat);
1665b7cb97cSAndrey V. Elsukov #endif /* VIMAGE */
16779288c11SBjoern A. Zeeb #ifdef INET
168bc725eafSRobert Watson static void	udp_detach(struct socket *so);
1694d77a549SAlfred Perlstein static int	udp_output(struct inpcb *, struct mbuf *, struct sockaddr *,
170a9839c4aSBjoern A. Zeeb 		    struct mbuf *, struct thread *, int);
17179288c11SBjoern A. Zeeb #endif
17279288c11SBjoern A. Zeeb 
173fec8a8c7SGleb Smirnoff INPCBSTORAGE_DEFINE(udpcbstor, "udpinp", "udp_inpcb", "udp", "udphash");
174fec8a8c7SGleb Smirnoff INPCBSTORAGE_DEFINE(udplitecbstor, "udpliteinp", "udplite_inpcb", "udplite",
175fec8a8c7SGleb Smirnoff     "udplitehash");
176e06e816fSKevin Lo 
17789128ff3SGleb Smirnoff static void
17878b1fc05SGleb Smirnoff udp_vnet_init(void *arg __unused)
179df8bae1dSRodney W. Grimes {
180af1ee11dSRobert Watson 
1818ad1a83bSAdrian Chadd 	/*
1828ad1a83bSAdrian Chadd 	 * For now default to 2-tuple UDP hashing - until the fragment
1838ad1a83bSAdrian Chadd 	 * reassembly code can also update the flowid.
1848ad1a83bSAdrian Chadd 	 *
1858ad1a83bSAdrian Chadd 	 * Once we can calculate the flowid that way and re-establish
1868ad1a83bSAdrian Chadd 	 * a 4-tuple, flip this to 4-tuple.
1878ad1a83bSAdrian Chadd 	 */
188fec8a8c7SGleb Smirnoff 	in_pcbinfo_init(&V_udbinfo, &udpcbstor, UDBHASHSIZE, UDBHASHSIZE);
1896a9148feSBjoern A. Zeeb 	V_udpcb_zone = uma_zcreate("udpcb", sizeof(struct udpcb),
190a8da5dd6SCraig Rodrigues 	    NULL, NULL, NULL, NULL, UMA_ALIGN_PTR, 0);
1916a9148feSBjoern A. Zeeb 	uma_zone_set_max(V_udpcb_zone, maxsockets);
1926acd596eSPawel Jakub Dawidek 	uma_zone_set_warning(V_udpcb_zone, "kern.ipc.maxsockets limit reached");
193df8bae1dSRodney W. Grimes 
19489128ff3SGleb Smirnoff 	/* Additional pcbinfo for UDP-Lite */
195fec8a8c7SGleb Smirnoff 	in_pcbinfo_init(&V_ulitecbinfo, &udplitecbstor, UDBHASHSIZE,
196fec8a8c7SGleb Smirnoff 	    UDBHASHSIZE);
197e06e816fSKevin Lo }
19878b1fc05SGleb Smirnoff VNET_SYSINIT(udp_vnet_init, SI_SUB_PROTO_DOMAIN, SI_ORDER_FOURTH,
19978b1fc05SGleb Smirnoff     udp_vnet_init, NULL);
200e06e816fSKevin Lo 
201315e3e38SRobert Watson /*
202315e3e38SRobert Watson  * Kernel module interface for updating udpstat.  The argument is an index
203315e3e38SRobert Watson  * into udpstat treated as an array of u_long.  While this encodes the
204315e3e38SRobert Watson  * general layout of udpstat into the caller, it doesn't encode its location,
205315e3e38SRobert Watson  * so that future changes to add, for example, per-CPU stats support won't
206315e3e38SRobert Watson  * cause binary compatibility problems for kernel modules.
207315e3e38SRobert Watson  */
208315e3e38SRobert Watson void
209315e3e38SRobert Watson kmod_udpstat_inc(int statnum)
210315e3e38SRobert Watson {
211315e3e38SRobert Watson 
2125b7cb97cSAndrey V. Elsukov 	counter_u64_add(VNET(udpstat)[statnum], 1);
213315e3e38SRobert Watson }
214315e3e38SRobert Watson 
2156a9148feSBjoern A. Zeeb int
2166a9148feSBjoern A. Zeeb udp_newudpcb(struct inpcb *inp)
2176a9148feSBjoern A. Zeeb {
2186a9148feSBjoern A. Zeeb 	struct udpcb *up;
2196a9148feSBjoern A. Zeeb 
2206a9148feSBjoern A. Zeeb 	up = uma_zalloc(V_udpcb_zone, M_NOWAIT | M_ZERO);
2216a9148feSBjoern A. Zeeb 	if (up == NULL)
2226a9148feSBjoern A. Zeeb 		return (ENOBUFS);
2236a9148feSBjoern A. Zeeb 	inp->inp_ppcb = up;
2246a9148feSBjoern A. Zeeb 	return (0);
2256a9148feSBjoern A. Zeeb }
2266a9148feSBjoern A. Zeeb 
2276a9148feSBjoern A. Zeeb void
2286a9148feSBjoern A. Zeeb udp_discardcb(struct udpcb *up)
2296a9148feSBjoern A. Zeeb {
2306a9148feSBjoern A. Zeeb 
2316a9148feSBjoern A. Zeeb 	uma_zfree(V_udpcb_zone, up);
2326a9148feSBjoern A. Zeeb }
2336a9148feSBjoern A. Zeeb 
234bc29160dSMarko Zec #ifdef VIMAGE
2353f58662dSBjoern A. Zeeb static void
2363f58662dSBjoern A. Zeeb udp_destroy(void *unused __unused)
237bc29160dSMarko Zec {
238bc29160dSMarko Zec 
2399bcd427bSRobert Watson 	in_pcbinfo_destroy(&V_udbinfo);
240391dab1cSBjoern A. Zeeb 	uma_zdestroy(V_udpcb_zone);
241bc29160dSMarko Zec }
2423f58662dSBjoern A. Zeeb VNET_SYSUNINIT(udp, SI_SUB_PROTO_DOMAIN, SI_ORDER_FOURTH, udp_destroy, NULL);
243e06e816fSKevin Lo 
2443f58662dSBjoern A. Zeeb static void
2453f58662dSBjoern A. Zeeb udplite_destroy(void *unused __unused)
246e06e816fSKevin Lo {
247e06e816fSKevin Lo 
248e06e816fSKevin Lo 	in_pcbinfo_destroy(&V_ulitecbinfo);
249e06e816fSKevin Lo }
2503f58662dSBjoern A. Zeeb VNET_SYSUNINIT(udplite, SI_SUB_PROTO_DOMAIN, SI_ORDER_FOURTH, udplite_destroy,
2513f58662dSBjoern A. Zeeb     NULL);
252bc29160dSMarko Zec #endif
253bc29160dSMarko Zec 
25479288c11SBjoern A. Zeeb #ifdef INET
25543bbb6aaSRobert Watson /*
25643bbb6aaSRobert Watson  * Subroutine of udp_input(), which appends the provided mbuf chain to the
25743bbb6aaSRobert Watson  * passed pcb/socket.  The caller must provide a sockaddr_in via udp_in that
25843bbb6aaSRobert Watson  * contains the source address.  If the socket ends up being an IPv6 socket,
25943bbb6aaSRobert Watson  * udp_append() will convert to a sockaddr_in6 before passing the address
26043bbb6aaSRobert Watson  * into the socket code.
261c0d1be08SRandall Stewart  *
262c0d1be08SRandall Stewart  * In the normal case udp_append() will return 0, indicating that you
263c0d1be08SRandall Stewart  * must unlock the inp. However if a tunneling protocol is in place we increment
264c0d1be08SRandall Stewart  * the inpcb refcnt and unlock the inp, on return from the tunneling protocol we
265c0d1be08SRandall Stewart  * then decrement the reference count. If the inp_rele returns 1, indicating the
266c0d1be08SRandall Stewart  * inp is gone, we return that to the caller to tell them *not* to unlock
267c0d1be08SRandall Stewart  * the inp. In the case of multi-cast this will cause the distribution
268c0d1be08SRandall Stewart  * to stop (though most tunneling protocols known currently do *not* use
269c0d1be08SRandall Stewart  * multicast).
27043bbb6aaSRobert Watson  */
271c0d1be08SRandall Stewart static int
27243bbb6aaSRobert Watson udp_append(struct inpcb *inp, struct ip *ip, struct mbuf *n, int off,
27343bbb6aaSRobert Watson     struct sockaddr_in *udp_in)
27443bbb6aaSRobert Watson {
27543bbb6aaSRobert Watson 	struct sockaddr *append_sa;
27643bbb6aaSRobert Watson 	struct socket *so;
277dce33a45SErmal Luçi 	struct mbuf *tmpopts, *opts = NULL;
27843bbb6aaSRobert Watson #ifdef INET6
27943bbb6aaSRobert Watson 	struct sockaddr_in6 udp_in6;
28043bbb6aaSRobert Watson #endif
2817b495c44SVANHULLEBUS Yvan 	struct udpcb *up;
282742e7210SKristof Provost 	bool filtered;
28343bbb6aaSRobert Watson 
284fa046d87SRobert Watson 	INP_LOCK_ASSERT(inp);
28543bbb6aaSRobert Watson 
28679bb84fbSEdward Tomasz Napierala 	/*
28779bb84fbSEdward Tomasz Napierala 	 * Engage the tunneling protocol.
28879bb84fbSEdward Tomasz Napierala 	 */
28979bb84fbSEdward Tomasz Napierala 	up = intoudpcb(inp);
29079bb84fbSEdward Tomasz Napierala 	if (up->u_tun_func != NULL) {
291c0d1be08SRandall Stewart 		in_pcbref(inp);
292c0d1be08SRandall Stewart 		INP_RUNLOCK(inp);
293742e7210SKristof Provost 		filtered = (*up->u_tun_func)(n, off, inp, (struct sockaddr *)&udp_in[0],
29481d3ec17SBryan Venteicher 		    up->u_tun_ctx);
295c0d1be08SRandall Stewart 		INP_RLOCK(inp);
296742e7210SKristof Provost 		if (filtered)
297c0d1be08SRandall Stewart 			return (in_pcbrele_rlocked(inp));
29879bb84fbSEdward Tomasz Napierala 	}
29979bb84fbSEdward Tomasz Napierala 
30079bb84fbSEdward Tomasz Napierala 	off += sizeof(struct udphdr);
30179bb84fbSEdward Tomasz Napierala 
302fcf59617SAndrey V. Elsukov #if defined(IPSEC) || defined(IPSEC_SUPPORT)
30343bbb6aaSRobert Watson 	/* Check AH/ESP integrity. */
304fcf59617SAndrey V. Elsukov 	if (IPSEC_ENABLED(ipv4) &&
305fcf59617SAndrey V. Elsukov 	    IPSEC_CHECK_POLICY(ipv4, n, inp) != 0) {
30643bbb6aaSRobert Watson 		m_freem(n);
307c0d1be08SRandall Stewart 		return (0);
30843bbb6aaSRobert Watson 	}
309fcf59617SAndrey V. Elsukov 	if (up->u_flags & UF_ESPINUDP) {/* IPSec UDP encaps. */
310fcf59617SAndrey V. Elsukov 		if (IPSEC_ENABLED(ipv4) &&
311fcf59617SAndrey V. Elsukov 		    UDPENCAP_INPUT(n, off, AF_INET) != 0)
312fcf59617SAndrey V. Elsukov 			return (0);	/* Consumed. */
3137b495c44SVANHULLEBUS Yvan 	}
31443bbb6aaSRobert Watson #endif /* IPSEC */
31543bbb6aaSRobert Watson #ifdef MAC
31630d239bcSRobert Watson 	if (mac_inpcb_check_deliver(inp, n) != 0) {
31743bbb6aaSRobert Watson 		m_freem(n);
318c0d1be08SRandall Stewart 		return (0);
31943bbb6aaSRobert Watson 	}
32079288c11SBjoern A. Zeeb #endif /* MAC */
32143bbb6aaSRobert Watson 	if (inp->inp_flags & INP_CONTROLOPTS ||
32243bbb6aaSRobert Watson 	    inp->inp_socket->so_options & (SO_TIMESTAMP | SO_BINTIME)) {
32343bbb6aaSRobert Watson #ifdef INET6
3249a38ba81SBjoern A. Zeeb 		if (inp->inp_vflag & INP_IPV6)
32548d48eb9SBjoern A. Zeeb 			(void)ip6_savecontrol_v4(inp, n, &opts, NULL);
3269a38ba81SBjoern A. Zeeb 		else
32779288c11SBjoern A. Zeeb #endif /* INET6 */
32843bbb6aaSRobert Watson 			ip_savecontrol(inp, &opts, ip, n);
32943bbb6aaSRobert Watson 	}
330dce33a45SErmal Luçi 	if ((inp->inp_vflag & INP_IPV4) && (inp->inp_flags2 & INP_ORIGDSTADDR)) {
331b46667c6SGleb Smirnoff 		tmpopts = sbcreatecontrol(&udp_in[1],
332b46667c6SGleb Smirnoff 		    sizeof(struct sockaddr_in), IP_ORIGDSTADDR, IPPROTO_IP,
333b46667c6SGleb Smirnoff 		    M_NOWAIT);
334dce33a45SErmal Luçi 		if (tmpopts) {
335dce33a45SErmal Luçi 			if (opts) {
336dce33a45SErmal Luçi 				tmpopts->m_next = opts;
337dce33a45SErmal Luçi 				opts = tmpopts;
338dce33a45SErmal Luçi 			} else
339dce33a45SErmal Luçi 				opts = tmpopts;
340dce33a45SErmal Luçi 		}
341dce33a45SErmal Luçi 	}
34243bbb6aaSRobert Watson #ifdef INET6
34343bbb6aaSRobert Watson 	if (inp->inp_vflag & INP_IPV6) {
34443bbb6aaSRobert Watson 		bzero(&udp_in6, sizeof(udp_in6));
34543bbb6aaSRobert Watson 		udp_in6.sin6_len = sizeof(udp_in6);
34643bbb6aaSRobert Watson 		udp_in6.sin6_family = AF_INET6;
347dce33a45SErmal Luçi 		in6_sin_2_v4mapsin6(&udp_in[0], &udp_in6);
34843bbb6aaSRobert Watson 		append_sa = (struct sockaddr *)&udp_in6;
34943bbb6aaSRobert Watson 	} else
35079288c11SBjoern A. Zeeb #endif /* INET6 */
351dce33a45SErmal Luçi 		append_sa = (struct sockaddr *)&udp_in[0];
35243bbb6aaSRobert Watson 	m_adj(n, off);
35343bbb6aaSRobert Watson 
35443bbb6aaSRobert Watson 	so = inp->inp_socket;
35543bbb6aaSRobert Watson 	SOCKBUF_LOCK(&so->so_rcv);
35643bbb6aaSRobert Watson 	if (sbappendaddr_locked(&so->so_rcv, append_sa, n, opts) == 0) {
357a61c24ddSKonstantin Kukushkin 		soroverflow_locked(so);
35843bbb6aaSRobert Watson 		m_freem(n);
35943bbb6aaSRobert Watson 		if (opts)
36043bbb6aaSRobert Watson 			m_freem(opts);
361026decb8SRobert Watson 		UDPSTAT_INC(udps_fullsock);
36243bbb6aaSRobert Watson 	} else
36343bbb6aaSRobert Watson 		sorwakeup_locked(so);
364c0d1be08SRandall Stewart 	return (0);
36543bbb6aaSRobert Watson }
36643bbb6aaSRobert Watson 
367db0ac6deSCy Schubert static bool
368db0ac6deSCy Schubert udp_multi_match(const struct inpcb *inp, void *v)
369db0ac6deSCy Schubert {
370db0ac6deSCy Schubert 	struct ip *ip = v;
371db0ac6deSCy Schubert 	struct udphdr *uh = (struct udphdr *)(ip + 1);
372db0ac6deSCy Schubert 
373db0ac6deSCy Schubert 	if (inp->inp_lport != uh->uh_dport)
374db0ac6deSCy Schubert 		return (false);
375db0ac6deSCy Schubert #ifdef INET6
376db0ac6deSCy Schubert 	if ((inp->inp_vflag & INP_IPV4) == 0)
377db0ac6deSCy Schubert 		return (false);
378db0ac6deSCy Schubert #endif
379db0ac6deSCy Schubert 	if (inp->inp_laddr.s_addr != INADDR_ANY &&
380db0ac6deSCy Schubert 	    inp->inp_laddr.s_addr != ip->ip_dst.s_addr)
381db0ac6deSCy Schubert 		return (false);
382db0ac6deSCy Schubert 	if (inp->inp_faddr.s_addr != INADDR_ANY &&
383db0ac6deSCy Schubert 	    inp->inp_faddr.s_addr != ip->ip_src.s_addr)
384db0ac6deSCy Schubert 		return (false);
385db0ac6deSCy Schubert 	if (inp->inp_fport != 0 &&
386db0ac6deSCy Schubert 	    inp->inp_fport != uh->uh_sport)
387db0ac6deSCy Schubert 		return (false);
388db0ac6deSCy Schubert 
389db0ac6deSCy Schubert 	return (true);
390db0ac6deSCy Schubert }
391db0ac6deSCy Schubert 
392db0ac6deSCy Schubert static int
393db0ac6deSCy Schubert udp_multi_input(struct mbuf *m, int proto, struct sockaddr_in *udp_in)
394db0ac6deSCy Schubert {
395db0ac6deSCy Schubert 	struct ip *ip = mtod(m, struct ip *);
396db0ac6deSCy Schubert 	struct inpcb_iterator inpi = INP_ITERATOR(udp_get_inpcbinfo(proto),
397db0ac6deSCy Schubert 	    INPLOOKUP_RLOCKPCB, udp_multi_match, ip);
398bd1d0850SGleb Smirnoff #ifdef KDTRACE_HOOKS
399db0ac6deSCy Schubert 	struct udphdr *uh = (struct udphdr *)(ip + 1);
400bd1d0850SGleb Smirnoff #endif
401db0ac6deSCy Schubert 	struct inpcb *inp;
402db0ac6deSCy Schubert 	struct mbuf *n;
403db0ac6deSCy Schubert 	int appends = 0;
404db0ac6deSCy Schubert 
405db0ac6deSCy Schubert 	MPASS(ip->ip_hl == sizeof(struct ip) >> 2);
406db0ac6deSCy Schubert 
407db0ac6deSCy Schubert 	while ((inp = inp_next(&inpi)) != NULL) {
408db0ac6deSCy Schubert 		/*
409db0ac6deSCy Schubert 		 * XXXRW: Because we weren't holding either the inpcb
410db0ac6deSCy Schubert 		 * or the hash lock when we checked for a match
411db0ac6deSCy Schubert 		 * before, we should probably recheck now that the
412db0ac6deSCy Schubert 		 * inpcb lock is held.
413db0ac6deSCy Schubert 		 */
414db0ac6deSCy Schubert 		/*
415db0ac6deSCy Schubert 		 * Handle socket delivery policy for any-source
416db0ac6deSCy Schubert 		 * and source-specific multicast. [RFC3678]
417db0ac6deSCy Schubert 		 */
418db0ac6deSCy Schubert 		if (IN_MULTICAST(ntohl(ip->ip_dst.s_addr))) {
419db0ac6deSCy Schubert 			struct ip_moptions	*imo;
420db0ac6deSCy Schubert 			struct sockaddr_in	 group;
421db0ac6deSCy Schubert 			int			 blocked;
422db0ac6deSCy Schubert 
423db0ac6deSCy Schubert 			imo = inp->inp_moptions;
424db0ac6deSCy Schubert 			if (imo == NULL)
425db0ac6deSCy Schubert 				continue;
426db0ac6deSCy Schubert 			bzero(&group, sizeof(struct sockaddr_in));
427db0ac6deSCy Schubert 			group.sin_len = sizeof(struct sockaddr_in);
428db0ac6deSCy Schubert 			group.sin_family = AF_INET;
429db0ac6deSCy Schubert 			group.sin_addr = ip->ip_dst;
430db0ac6deSCy Schubert 
431db0ac6deSCy Schubert 			blocked = imo_multi_filter(imo, m->m_pkthdr.rcvif,
432db0ac6deSCy Schubert 				(struct sockaddr *)&group,
433db0ac6deSCy Schubert 				(struct sockaddr *)&udp_in[0]);
434db0ac6deSCy Schubert 			if (blocked != MCAST_PASS) {
435db0ac6deSCy Schubert 				if (blocked == MCAST_NOTGMEMBER)
436db0ac6deSCy Schubert 					IPSTAT_INC(ips_notmember);
437db0ac6deSCy Schubert 				if (blocked == MCAST_NOTSMEMBER ||
438db0ac6deSCy Schubert 				    blocked == MCAST_MUTED)
439db0ac6deSCy Schubert 					UDPSTAT_INC(udps_filtermcast);
440db0ac6deSCy Schubert 				continue;
441db0ac6deSCy Schubert 			}
442db0ac6deSCy Schubert 		}
443db0ac6deSCy Schubert 		if ((n = m_copym(m, 0, M_COPYALL, M_NOWAIT)) != NULL) {
444db0ac6deSCy Schubert 			if (proto == IPPROTO_UDPLITE)
445db0ac6deSCy Schubert 				UDPLITE_PROBE(receive, NULL, inp, ip, inp, uh);
446db0ac6deSCy Schubert 			else
447db0ac6deSCy Schubert 				UDP_PROBE(receive, NULL, inp, ip, inp, uh);
448db0ac6deSCy Schubert 			if (udp_append(inp, ip, n, sizeof(struct ip), udp_in)) {
449db0ac6deSCy Schubert 				INP_RUNLOCK(inp);
450db0ac6deSCy Schubert 				break;
451db0ac6deSCy Schubert 			} else
452db0ac6deSCy Schubert 				appends++;
453db0ac6deSCy Schubert 		}
454db0ac6deSCy Schubert 		/*
455db0ac6deSCy Schubert 		 * Don't look for additional matches if this one does
456db0ac6deSCy Schubert 		 * not have either the SO_REUSEPORT or SO_REUSEADDR
457db0ac6deSCy Schubert 		 * socket options set.  This heuristic avoids
458db0ac6deSCy Schubert 		 * searching through all pcbs in the common case of a
459db0ac6deSCy Schubert 		 * non-shared port.  It assumes that an application
460db0ac6deSCy Schubert 		 * will never clear these options after setting them.
461db0ac6deSCy Schubert 		 */
462db0ac6deSCy Schubert 		if ((inp->inp_socket->so_options &
463db0ac6deSCy Schubert 		    (SO_REUSEPORT|SO_REUSEPORT_LB|SO_REUSEADDR)) == 0) {
464db0ac6deSCy Schubert 			INP_RUNLOCK(inp);
465db0ac6deSCy Schubert 			break;
466db0ac6deSCy Schubert 		}
467db0ac6deSCy Schubert 	}
468db0ac6deSCy Schubert 
469db0ac6deSCy Schubert 	if (appends == 0) {
470db0ac6deSCy Schubert 		/*
471db0ac6deSCy Schubert 		 * No matching pcb found; discard datagram.  (No need
472db0ac6deSCy Schubert 		 * to send an ICMP Port Unreachable for a broadcast
473db0ac6deSCy Schubert 		 * or multicast datgram.)
474db0ac6deSCy Schubert 		 */
475db0ac6deSCy Schubert 		UDPSTAT_INC(udps_noport);
476db0ac6deSCy Schubert 		if (IN_MULTICAST(ntohl(ip->ip_dst.s_addr)))
477db0ac6deSCy Schubert 			UDPSTAT_INC(udps_noportmcast);
478db0ac6deSCy Schubert 		else
479db0ac6deSCy Schubert 			UDPSTAT_INC(udps_noportbcast);
480db0ac6deSCy Schubert 	}
481014f98b1SMark Johnston 	m_freem(m);
482db0ac6deSCy Schubert 
483db0ac6deSCy Schubert 	return (IPPROTO_DONE);
484db0ac6deSCy Schubert }
485db0ac6deSCy Schubert 
48678b1fc05SGleb Smirnoff static int
4878f5a8818SKevin Lo udp_input(struct mbuf **mp, int *offp, int proto)
488df8bae1dSRodney W. Grimes {
4893329b236SRobert Watson 	struct ip *ip;
4903329b236SRobert Watson 	struct udphdr *uh;
49171498f30SBruce M Simpson 	struct ifnet *ifp;
4923329b236SRobert Watson 	struct inpcb *inp;
4938f134647SGleb Smirnoff 	uint16_t len, ip_len;
494e06e816fSKevin Lo 	struct inpcbinfo *pcbinfo;
495dce33a45SErmal Luçi 	struct sockaddr_in udp_in[2];
4968f5a8818SKevin Lo 	struct mbuf *m;
4970b4ae859SGleb Smirnoff 	struct m_tag *fwd_tag;
4988f5a8818SKevin Lo 	int cscov_partial, iphlen;
499df8bae1dSRodney W. Grimes 
5008f5a8818SKevin Lo 	m = *mp;
5018f5a8818SKevin Lo 	iphlen = *offp;
50271498f30SBruce M Simpson 	ifp = m->m_pkthdr.rcvif;
5038f5a8818SKevin Lo 	*mp = NULL;
504026decb8SRobert Watson 	UDPSTAT_INC(udps_ipackets);
505df8bae1dSRodney W. Grimes 
506df8bae1dSRodney W. Grimes 	/*
5073329b236SRobert Watson 	 * Strip IP options, if any; should skip this, make available to
5083329b236SRobert Watson 	 * user, and use on returned packets, but we don't yet have a way to
5093329b236SRobert Watson 	 * check the checksum with options still present.
510df8bae1dSRodney W. Grimes 	 */
511df8bae1dSRodney W. Grimes 	if (iphlen > sizeof (struct ip)) {
512105bd211SGleb Smirnoff 		ip_stripoptions(m);
513df8bae1dSRodney W. Grimes 		iphlen = sizeof(struct ip);
514df8bae1dSRodney W. Grimes 	}
515df8bae1dSRodney W. Grimes 
516df8bae1dSRodney W. Grimes 	/*
517df8bae1dSRodney W. Grimes 	 * Get IP and UDP header together in first mbuf.
518df8bae1dSRodney W. Grimes 	 */
519df8bae1dSRodney W. Grimes 	if (m->m_len < iphlen + sizeof(struct udphdr)) {
520d1b18731SKevin Lo 		if ((m = m_pullup(m, iphlen + sizeof(struct udphdr))) == NULL) {
521026decb8SRobert Watson 			UDPSTAT_INC(udps_hdrops);
5228f5a8818SKevin Lo 			return (IPPROTO_DONE);
523df8bae1dSRodney W. Grimes 		}
524df8bae1dSRodney W. Grimes 	}
525503f4e47SBjoern A. Zeeb 	ip = mtod(m, struct ip *);
526df8bae1dSRodney W. Grimes 	uh = (struct udphdr *)((caddr_t)ip + iphlen);
5278f5a8818SKevin Lo 	cscov_partial = (proto == IPPROTO_UDPLITE) ? 1 : 0;
528df8bae1dSRodney W. Grimes 
5293329b236SRobert Watson 	/*
5303329b236SRobert Watson 	 * Destination port of 0 is illegal, based on RFC768.
5313329b236SRobert Watson 	 */
532686cdd19SJun-ichiro itojun Hagino 	if (uh->uh_dport == 0)
533f76fcf6dSJeffrey Hsu 		goto badunlocked;
534686cdd19SJun-ichiro itojun Hagino 
535df8bae1dSRodney W. Grimes 	/*
5363329b236SRobert Watson 	 * Construct sockaddr format source address.  Stuff source address
5373329b236SRobert Watson 	 * and datagram in user buffer.
538b9234fafSSam Leffler 	 */
539dce33a45SErmal Luçi 	bzero(&udp_in[0], sizeof(struct sockaddr_in) * 2);
540dce33a45SErmal Luçi 	udp_in[0].sin_len = sizeof(struct sockaddr_in);
541dce33a45SErmal Luçi 	udp_in[0].sin_family = AF_INET;
542dce33a45SErmal Luçi 	udp_in[0].sin_port = uh->uh_sport;
543dce33a45SErmal Luçi 	udp_in[0].sin_addr = ip->ip_src;
544dce33a45SErmal Luçi 	udp_in[1].sin_len = sizeof(struct sockaddr_in);
545dce33a45SErmal Luçi 	udp_in[1].sin_family = AF_INET;
546dce33a45SErmal Luçi 	udp_in[1].sin_port = uh->uh_dport;
547dce33a45SErmal Luçi 	udp_in[1].sin_addr = ip->ip_dst;
548b9234fafSSam Leffler 
549b9234fafSSam Leffler 	/*
550af1ee11dSRobert Watson 	 * Make mbuf data length reflect UDP length.  If not enough data to
551af1ee11dSRobert Watson 	 * reflect UDP length, drop.
552df8bae1dSRodney W. Grimes 	 */
553df8bae1dSRodney W. Grimes 	len = ntohs((u_short)uh->uh_ulen);
5548ad458a4SGleb Smirnoff 	ip_len = ntohs(ip->ip_len) - iphlen;
5550f4a0366SMichael Tuexen 	if (proto == IPPROTO_UDPLITE && (len == 0 || len == ip_len)) {
556e06e816fSKevin Lo 		/* Zero means checksum over the complete packet. */
5570f4a0366SMichael Tuexen 		if (len == 0)
558e06e816fSKevin Lo 			len = ip_len;
559e06e816fSKevin Lo 		cscov_partial = 0;
560e06e816fSKevin Lo 	}
5618f134647SGleb Smirnoff 	if (ip_len != len) {
5628f134647SGleb Smirnoff 		if (len > ip_len || len < sizeof(struct udphdr)) {
563026decb8SRobert Watson 			UDPSTAT_INC(udps_badlen);
564f76fcf6dSJeffrey Hsu 			goto badunlocked;
565df8bae1dSRodney W. Grimes 		}
5668f5a8818SKevin Lo 		if (proto == IPPROTO_UDP)
5678f134647SGleb Smirnoff 			m_adj(m, len - ip_len);
568df8bae1dSRodney W. Grimes 	}
5693329b236SRobert Watson 
570df8bae1dSRodney W. Grimes 	/*
571df8bae1dSRodney W. Grimes 	 * Checksum extended UDP header and data.
572df8bae1dSRodney W. Grimes 	 */
5736dfab5b1SGarrett Wollman 	if (uh->uh_sum) {
57439629c92SDavid Malone 		u_short uh_sum;
57539629c92SDavid Malone 
576e06e816fSKevin Lo 		if ((m->m_pkthdr.csum_flags & CSUM_DATA_VALID) &&
577e06e816fSKevin Lo 		    !cscov_partial) {
578db4f9cc7SJonathan Lemon 			if (m->m_pkthdr.csum_flags & CSUM_PSEUDO_HDR)
57939629c92SDavid Malone 				uh_sum = m->m_pkthdr.csum_data;
580db4f9cc7SJonathan Lemon 			else
58139629c92SDavid Malone 				uh_sum = in_pseudo(ip->ip_src.s_addr,
582506f4949SRuslan Ermilov 				    ip->ip_dst.s_addr, htonl((u_short)len +
5838f5a8818SKevin Lo 				    m->m_pkthdr.csum_data + proto));
58439629c92SDavid Malone 			uh_sum ^= 0xffff;
585db4f9cc7SJonathan Lemon 		} else {
5863358df29SGleb Smirnoff 			char b[offsetof(struct ipovly, ih_src)];
5873358df29SGleb Smirnoff 			struct ipovly *ipov = (struct ipovly *)ip;
588af1ee11dSRobert Watson 
5893358df29SGleb Smirnoff 			bcopy(ipov, b, sizeof(b));
5903358df29SGleb Smirnoff 			bzero(ipov, sizeof(ipov->ih_x1));
5913358df29SGleb Smirnoff 			ipov->ih_len = (proto == IPPROTO_UDP) ?
592e06e816fSKevin Lo 			    uh->uh_ulen : htons(ip_len);
59339629c92SDavid Malone 			uh_sum = in_cksum(m, len + sizeof (struct ip));
5943358df29SGleb Smirnoff 			bcopy(b, ipov, sizeof(b));
595db4f9cc7SJonathan Lemon 		}
59639629c92SDavid Malone 		if (uh_sum) {
597026decb8SRobert Watson 			UDPSTAT_INC(udps_badsum);
598df8bae1dSRodney W. Grimes 			m_freem(m);
5998f5a8818SKevin Lo 			return (IPPROTO_DONE);
600df8bae1dSRodney W. Grimes 		}
601c6d81a34SMichael Tuexen 	} else {
602c6d81a34SMichael Tuexen 		if (proto == IPPROTO_UDP) {
603026decb8SRobert Watson 			UDPSTAT_INC(udps_nosum);
604c6d81a34SMichael Tuexen 		} else {
605c6d81a34SMichael Tuexen 			/* UDPLite requires a checksum */
606c6d81a34SMichael Tuexen 			/* XXX: What is the right UDPLite MIB counter here? */
607c6d81a34SMichael Tuexen 			m_freem(m);
608c6d81a34SMichael Tuexen 			return (IPPROTO_DONE);
609c6d81a34SMichael Tuexen 		}
610c6d81a34SMichael Tuexen 	}
611df8bae1dSRodney W. Grimes 
612266f97b5SCy Schubert 	if (IN_MULTICAST(ntohl(ip->ip_dst.s_addr)) ||
613db0ac6deSCy Schubert 	    in_broadcast(ip->ip_dst, ifp))
614db0ac6deSCy Schubert 		return (udp_multi_input(m, proto, udp_in));
615266f97b5SCy Schubert 
616db0ac6deSCy Schubert 	pcbinfo = udp_get_inpcbinfo(proto);
6173329b236SRobert Watson 
618df8bae1dSRodney W. Grimes 	/*
6196d6a026bSDavid Greenman 	 * Locate pcb for datagram.
620db0ac6deSCy Schubert 	 *
6218a006adbSBjoern A. Zeeb 	 * Grab info from PACKET_TAG_IPFORWARD tag prepended to the chain.
6228a006adbSBjoern A. Zeeb 	 */
623ffdbf9daSAndrey V. Elsukov 	if ((m->m_flags & M_IP_NEXTHOP) &&
624c1de64a4SAndrey V. Elsukov 	    (fwd_tag = m_tag_find(m, PACKET_TAG_IPFORWARD, NULL)) != NULL) {
6258a006adbSBjoern A. Zeeb 		struct sockaddr_in *next_hop;
6268a006adbSBjoern A. Zeeb 
6278a006adbSBjoern A. Zeeb 		next_hop = (struct sockaddr_in *)(fwd_tag + 1);
6288a006adbSBjoern A. Zeeb 
6298a006adbSBjoern A. Zeeb 		/*
6308a006adbSBjoern A. Zeeb 		 * Transparently forwarded. Pretend to be the destination.
6318a006adbSBjoern A. Zeeb 		 * Already got one like this?
6328a006adbSBjoern A. Zeeb 		 */
633e06e816fSKevin Lo 		inp = in_pcblookup_mbuf(pcbinfo, ip->ip_src, uh->uh_sport,
6348a006adbSBjoern A. Zeeb 		    ip->ip_dst, uh->uh_dport, INPLOOKUP_RLOCKPCB, ifp, m);
6358a006adbSBjoern A. Zeeb 		if (!inp) {
6368a006adbSBjoern A. Zeeb 			/*
6378a006adbSBjoern A. Zeeb 			 * It's new.  Try to find the ambushing socket.
6388a006adbSBjoern A. Zeeb 			 * Because we've rewritten the destination address,
6398a006adbSBjoern A. Zeeb 			 * any hardware-generated hash is ignored.
6408a006adbSBjoern A. Zeeb 			 */
641e06e816fSKevin Lo 			inp = in_pcblookup(pcbinfo, ip->ip_src,
6428a006adbSBjoern A. Zeeb 			    uh->uh_sport, next_hop->sin_addr,
6438a006adbSBjoern A. Zeeb 			    next_hop->sin_port ? htons(next_hop->sin_port) :
6448a006adbSBjoern A. Zeeb 			    uh->uh_dport, INPLOOKUP_WILDCARD |
6458a006adbSBjoern A. Zeeb 			    INPLOOKUP_RLOCKPCB, ifp);
6468a006adbSBjoern A. Zeeb 		}
6478a006adbSBjoern A. Zeeb 		/* Remove the tag from the packet. We don't need it anymore. */
6488a006adbSBjoern A. Zeeb 		m_tag_delete(m, fwd_tag);
649ffdbf9daSAndrey V. Elsukov 		m->m_flags &= ~M_IP_NEXTHOP;
6508a006adbSBjoern A. Zeeb 	} else
651e06e816fSKevin Lo 		inp = in_pcblookup_mbuf(pcbinfo, ip->ip_src, uh->uh_sport,
6528a006adbSBjoern A. Zeeb 		    ip->ip_dst, uh->uh_dport, INPLOOKUP_WILDCARD |
6538a006adbSBjoern A. Zeeb 		    INPLOOKUP_RLOCKPCB, ifp, m);
65415bd2b43SDavid Greenman 	if (inp == NULL) {
655334fc582SBjoern A. Zeeb 		if (V_udp_log_in_vain) {
656edf0313bSEric van Gyzen 			char src[INET_ADDRSTRLEN];
657edf0313bSEric van Gyzen 			char dst[INET_ADDRSTRLEN];
65875cfc95fSAndrey A. Chernov 
659592071e8SBruce Evans 			log(LOG_INFO,
660592071e8SBruce Evans 			    "Connection attempt to UDP %s:%d from %s:%d\n",
661edf0313bSEric van Gyzen 			    inet_ntoa_r(ip->ip_dst, dst), ntohs(uh->uh_dport),
662edf0313bSEric van Gyzen 			    inet_ntoa_r(ip->ip_src, src), ntohs(uh->uh_sport));
66375cfc95fSAndrey A. Chernov 		}
6647bda9663SMichael Tuexen 		if (proto == IPPROTO_UDPLITE)
6657bda9663SMichael Tuexen 			UDPLITE_PROBE(receive, NULL, NULL, ip, NULL, uh);
6667bda9663SMichael Tuexen 		else
667e1526d5aSMichael Tuexen 			UDP_PROBE(receive, NULL, NULL, ip, NULL, uh);
668026decb8SRobert Watson 		UDPSTAT_INC(udps_noport);
669df8bae1dSRodney W. Grimes 		if (m->m_flags & (M_BCAST | M_MCAST)) {
670026decb8SRobert Watson 			UDPSTAT_INC(udps_noportbcast);
671fa046d87SRobert Watson 			goto badunlocked;
672df8bae1dSRodney W. Grimes 		}
6733ea9a7cfSGleb Smirnoff 		if (V_udp_blackhole && (V_udp_blackhole_local ||
6743ea9a7cfSGleb Smirnoff 		    !in_localip(ip->ip_src)))
675fa046d87SRobert Watson 			goto badunlocked;
6761cbd978eSLuigi Rizzo 		if (badport_bandlim(BANDLIM_ICMP_UNREACH) < 0)
677fa046d87SRobert Watson 			goto badunlocked;
678582a7760SBruce Evans 		icmp_error(m, ICMP_UNREACH, ICMP_UNREACH_PORT, 0, 0);
6798f5a8818SKevin Lo 		return (IPPROTO_DONE);
680df8bae1dSRodney W. Grimes 	}
6813329b236SRobert Watson 
6823329b236SRobert Watson 	/*
6833329b236SRobert Watson 	 * Check the minimum TTL for socket.
6843329b236SRobert Watson 	 */
685fa046d87SRobert Watson 	INP_RLOCK_ASSERT(inp);
68610cc62b7SRobert Watson 	if (inp->inp_ip_minttl && inp->inp_ip_minttl > ip->ip_ttl) {
6877bda9663SMichael Tuexen 		if (proto == IPPROTO_UDPLITE)
6887bda9663SMichael Tuexen 			UDPLITE_PROBE(receive, NULL, inp, ip, inp, uh);
6897bda9663SMichael Tuexen 		else
690e1526d5aSMichael Tuexen 			UDP_PROBE(receive, NULL, inp, ip, inp, uh);
69110cc62b7SRobert Watson 		INP_RUNLOCK(inp);
692fa046d87SRobert Watson 		m_freem(m);
6938f5a8818SKevin Lo 		return (IPPROTO_DONE);
69410cc62b7SRobert Watson 	}
695e06e816fSKevin Lo 	if (cscov_partial) {
696e06e816fSKevin Lo 		struct udpcb *up;
697e06e816fSKevin Lo 
698e06e816fSKevin Lo 		up = intoudpcb(inp);
69983e95fb3SMichael Tuexen 		if (up->u_rxcslen == 0 || up->u_rxcslen > len) {
700e06e816fSKevin Lo 			INP_RUNLOCK(inp);
701e06e816fSKevin Lo 			m_freem(m);
7028f5a8818SKevin Lo 			return (IPPROTO_DONE);
703e06e816fSKevin Lo 		}
704e06e816fSKevin Lo 	}
70557f60867SMark Johnston 
7067bda9663SMichael Tuexen 	if (proto == IPPROTO_UDPLITE)
7077bda9663SMichael Tuexen 		UDPLITE_PROBE(receive, NULL, inp, ip, inp, uh);
7087bda9663SMichael Tuexen 	else
7091ad19fb6SMark Johnston 		UDP_PROBE(receive, NULL, inp, ip, inp, uh);
710dce33a45SErmal Luçi 	if (udp_append(inp, ip, m, iphlen, udp_in) == 0)
711119d85f6SRobert Watson 		INP_RUNLOCK(inp);
7128f5a8818SKevin Lo 	return (IPPROTO_DONE);
71361ffc0b1SJeffrey Hsu 
714f76fcf6dSJeffrey Hsu badunlocked:
715df8bae1dSRodney W. Grimes 	m_freem(m);
7168f5a8818SKevin Lo 	return (IPPROTO_DONE);
717cfa1ca9dSYoshinobu Inoue }
71879288c11SBjoern A. Zeeb #endif /* INET */
719cfa1ca9dSYoshinobu Inoue 
720cfa1ca9dSYoshinobu Inoue /*
7213329b236SRobert Watson  * Notify a udp user of an asynchronous error; just wake up so that they can
7223329b236SRobert Watson  * collect error status.
723df8bae1dSRodney W. Grimes  */
7243ce144eaSJeffrey Hsu struct inpcb *
7253329b236SRobert Watson udp_notify(struct inpcb *inp, int errno)
726df8bae1dSRodney W. Grimes {
7273329b236SRobert Watson 
728083a010cSRyan Stone 	INP_WLOCK_ASSERT(inp);
72984cc0778SGeorge V. Neville-Neil 	if ((errno == EHOSTUNREACH || errno == ENETUNREACH ||
730983066f0SAlexander V. Chernikov 	     errno == EHOSTDOWN) && inp->inp_route.ro_nh) {
731983066f0SAlexander V. Chernikov 		NH_FREE(inp->inp_route.ro_nh);
732983066f0SAlexander V. Chernikov 		inp->inp_route.ro_nh = (struct nhop_object *)NULL;
73384cc0778SGeorge V. Neville-Neil 	}
7348501a69cSRobert Watson 
735df8bae1dSRodney W. Grimes 	inp->inp_socket->so_error = errno;
736df8bae1dSRodney W. Grimes 	sorwakeup(inp->inp_socket);
737df8bae1dSRodney W. Grimes 	sowwakeup(inp->inp_socket);
7383329b236SRobert Watson 	return (inp);
739df8bae1dSRodney W. Grimes }
740df8bae1dSRodney W. Grimes 
74179288c11SBjoern A. Zeeb #ifdef INET
742e06e816fSKevin Lo static void
743e06e816fSKevin Lo udp_common_ctlinput(int cmd, struct sockaddr *sa, void *vip,
744e06e816fSKevin Lo     struct inpcbinfo *pcbinfo)
745df8bae1dSRodney W. Grimes {
746c693a045SJonathan Lemon 	struct ip *ip = vip;
747c693a045SJonathan Lemon 	struct udphdr *uh;
748c693a045SJonathan Lemon 	struct in_addr faddr;
749c693a045SJonathan Lemon 	struct inpcb *inp;
750c693a045SJonathan Lemon 
751c693a045SJonathan Lemon 	faddr = ((struct sockaddr_in *)sa)->sin_addr;
752c693a045SJonathan Lemon 	if (sa->sa_family != AF_INET || faddr.s_addr == INADDR_ANY)
753c693a045SJonathan Lemon 		return;
754df8bae1dSRodney W. Grimes 
75584cc0778SGeorge V. Neville-Neil 	if (PRC_IS_REDIRECT(cmd)) {
75684cc0778SGeorge V. Neville-Neil 		/* signal EHOSTDOWN, as it flushes the cached route */
7574760956eSMichael Tuexen 		in_pcbnotifyall(pcbinfo, faddr, EHOSTDOWN, udp_notify);
75897d8d152SAndre Oppermann 		return;
75984cc0778SGeorge V. Neville-Neil 	}
7603329b236SRobert Watson 
76197d8d152SAndre Oppermann 	/*
76297d8d152SAndre Oppermann 	 * Hostdead is ugly because it goes linearly through all PCBs.
7633329b236SRobert Watson 	 *
7643329b236SRobert Watson 	 * XXX: We never get this from ICMP, otherwise it makes an excellent
7653329b236SRobert Watson 	 * DoS attack on machines with many connections.
76697d8d152SAndre Oppermann 	 */
76797d8d152SAndre Oppermann 	if (cmd == PRC_HOSTDEAD)
768af1ee11dSRobert Watson 		ip = NULL;
769d1c54148SJesper Skriver 	else if ((unsigned)cmd >= PRC_NCMDS || inetctlerrmap[cmd] == 0)
770df8bae1dSRodney W. Grimes 		return;
771af1ee11dSRobert Watson 	if (ip != NULL) {
772df8bae1dSRodney W. Grimes 		uh = (struct udphdr *)((caddr_t)ip + (ip->ip_hl << 2));
773e06e816fSKevin Lo 		inp = in_pcblookup(pcbinfo, faddr, uh->uh_dport,
774083a010cSRyan Stone 		    ip->ip_src, uh->uh_sport, INPLOOKUP_WLOCKPCB, NULL);
775f76fcf6dSJeffrey Hsu 		if (inp != NULL) {
776083a010cSRyan Stone 			INP_WLOCK_ASSERT(inp);
777f76fcf6dSJeffrey Hsu 			if (inp->inp_socket != NULL) {
778f5514f08SRobert Watson 				udp_notify(inp, inetctlerrmap[cmd]);
779f76fcf6dSJeffrey Hsu 			}
780083a010cSRyan Stone 			INP_WUNLOCK(inp);
781abb901c5SRandall Stewart 		} else {
782abb901c5SRandall Stewart 			inp = in_pcblookup(pcbinfo, faddr, uh->uh_dport,
783abb901c5SRandall Stewart 					   ip->ip_src, uh->uh_sport,
784abb901c5SRandall Stewart 					   INPLOOKUP_WILDCARD | INPLOOKUP_RLOCKPCB, NULL);
785abb901c5SRandall Stewart 			if (inp != NULL) {
786abb901c5SRandall Stewart 				struct udpcb *up;
78746374cbfSMatt Macy 				void *ctx;
78846374cbfSMatt Macy 				udp_tun_icmp_t func;
789abb901c5SRandall Stewart 
790abb901c5SRandall Stewart 				up = intoudpcb(inp);
79146374cbfSMatt Macy 				ctx = up->u_tun_ctx;
79246374cbfSMatt Macy 				func = up->u_icmp_func;
793abb901c5SRandall Stewart 				INP_RUNLOCK(inp);
79446374cbfSMatt Macy 				if (func != NULL)
79546374cbfSMatt Macy 					(*func)(cmd, sa, vip, ctx);
796abb901c5SRandall Stewart 			}
797f76fcf6dSJeffrey Hsu 		}
798df8bae1dSRodney W. Grimes 	} else
799e06e816fSKevin Lo 		in_pcbnotifyall(pcbinfo, faddr, inetctlerrmap[cmd],
800f5514f08SRobert Watson 		    udp_notify);
801df8bae1dSRodney W. Grimes }
80278b1fc05SGleb Smirnoff 
80378b1fc05SGleb Smirnoff static void
804e06e816fSKevin Lo udp_ctlinput(int cmd, struct sockaddr *sa, void *vip)
805e06e816fSKevin Lo {
806e06e816fSKevin Lo 
807e06e816fSKevin Lo 	return (udp_common_ctlinput(cmd, sa, vip, &V_udbinfo));
808e06e816fSKevin Lo }
809e06e816fSKevin Lo 
81078b1fc05SGleb Smirnoff static void
811e06e816fSKevin Lo udplite_ctlinput(int cmd, struct sockaddr *sa, void *vip)
812e06e816fSKevin Lo {
813e06e816fSKevin Lo 
814e06e816fSKevin Lo 	return (udp_common_ctlinput(cmd, sa, vip, &V_ulitecbinfo));
815e06e816fSKevin Lo }
81679288c11SBjoern A. Zeeb #endif /* INET */
817df8bae1dSRodney W. Grimes 
8180312fbe9SPoul-Henning Kamp static int
81982d9ae4eSPoul-Henning Kamp udp_pcblist(SYSCTL_HANDLER_ARGS)
82098271db4SGarrett Wollman {
821db0ac6deSCy Schubert 	struct inpcb_iterator inpi = INP_ALL_ITERATOR(&V_udbinfo,
822db0ac6deSCy Schubert 	    INPLOOKUP_RLOCKPCB);
82398271db4SGarrett Wollman 	struct xinpgen xig;
824032677ceSGleb Smirnoff 	struct inpcb *inp;
825032677ceSGleb Smirnoff 	int error;
82698271db4SGarrett Wollman 
827032677ceSGleb Smirnoff 	if (req->newptr != 0)
828032677ceSGleb Smirnoff 		return (EPERM);
829032677ceSGleb Smirnoff 
83098271db4SGarrett Wollman 	if (req->oldptr == 0) {
831032677ceSGleb Smirnoff 		int n;
832032677ceSGleb Smirnoff 
833603724d3SBjoern A. Zeeb 		n = V_udbinfo.ipi_count;
834c007b96aSJohn Baldwin 		n += imax(n / 8, 10);
835c007b96aSJohn Baldwin 		req->oldidx = 2 * (sizeof xig) + n * sizeof(struct xinpcb);
8363329b236SRobert Watson 		return (0);
83798271db4SGarrett Wollman 	}
83898271db4SGarrett Wollman 
839032677ceSGleb Smirnoff 	if ((error = sysctl_wire_old_buffer(req, 0)) != 0)
84047934cefSDon Lewis 		return (error);
8415c38b6dbSDon Lewis 
84279db6fe7SMark Johnston 	bzero(&xig, sizeof(xig));
84398271db4SGarrett Wollman 	xig.xig_len = sizeof xig;
844032677ceSGleb Smirnoff 	xig.xig_count = V_udbinfo.ipi_count;
845032677ceSGleb Smirnoff 	xig.xig_gen = V_udbinfo.ipi_gencnt;
84698271db4SGarrett Wollman 	xig.xig_sogen = so_gencnt;
84798271db4SGarrett Wollman 	error = SYSCTL_OUT(req, &xig, sizeof xig);
84898271db4SGarrett Wollman 	if (error)
8493329b236SRobert Watson 		return (error);
85099208b82SMatt Macy 
851db0ac6deSCy Schubert 	while ((inp = inp_next(&inpi)) != NULL) {
852032677ceSGleb Smirnoff 		if (inp->inp_gencnt <= xig.xig_gen &&
853032677ceSGleb Smirnoff 		    cr_canseeinpcb(req->td->td_ucred, inp) == 0) {
85498271db4SGarrett Wollman 			struct xinpcb xi;
855d0e157f6SBjoern A. Zeeb 
856cc65eb4eSGleb Smirnoff 			in_pcbtoxinpcb(inp, &xi);
857266f97b5SCy Schubert 			error = SYSCTL_OUT(req, &xi, sizeof xi);
858db0ac6deSCy Schubert 			if (error) {
859266f97b5SCy Schubert 				INP_RUNLOCK(inp);
860db0ac6deSCy Schubert 				break;
86198271db4SGarrett Wollman 			}
862db0ac6deSCy Schubert 		}
863db0ac6deSCy Schubert 	}
864d0e157f6SBjoern A. Zeeb 
86598271db4SGarrett Wollman 	if (!error) {
86698271db4SGarrett Wollman 		/*
8673329b236SRobert Watson 		 * Give the user an updated idea of our state.  If the
8683329b236SRobert Watson 		 * generation differs from what we told her before, she knows
8693329b236SRobert Watson 		 * that something happened while we were processing this
8703329b236SRobert Watson 		 * request, and it might be necessary to retry.
87198271db4SGarrett Wollman 		 */
872603724d3SBjoern A. Zeeb 		xig.xig_gen = V_udbinfo.ipi_gencnt;
87398271db4SGarrett Wollman 		xig.xig_sogen = so_gencnt;
874603724d3SBjoern A. Zeeb 		xig.xig_count = V_udbinfo.ipi_count;
87598271db4SGarrett Wollman 		error = SYSCTL_OUT(req, &xig, sizeof xig);
87698271db4SGarrett Wollman 	}
877032677ceSGleb Smirnoff 
8783329b236SRobert Watson 	return (error);
87998271db4SGarrett Wollman }
88098271db4SGarrett Wollman 
88179c3d51bSMatthew D Fleming SYSCTL_PROC(_net_inet_udp, UDPCTL_PCBLIST, pcblist,
8827029da5cSPawel Biernacki     CTLTYPE_OPAQUE | CTLFLAG_RD | CTLFLAG_MPSAFE, NULL, 0,
8837029da5cSPawel Biernacki     udp_pcblist, "S,xinpcb",
8847029da5cSPawel Biernacki     "List of active UDP sockets");
88598271db4SGarrett Wollman 
88679288c11SBjoern A. Zeeb #ifdef INET
88798271db4SGarrett Wollman static int
88882d9ae4eSPoul-Henning Kamp udp_getcred(SYSCTL_HANDLER_ARGS)
889490d50b6SBrian Feldman {
890c0511d3bSBrian Feldman 	struct xucred xuc;
891490d50b6SBrian Feldman 	struct sockaddr_in addrs[2];
892d797164aSGleb Smirnoff 	struct epoch_tracker et;
893490d50b6SBrian Feldman 	struct inpcb *inp;
894277afaffSRobert Watson 	int error;
895490d50b6SBrian Feldman 
89632f9753cSRobert Watson 	error = priv_check(req->td, PRIV_NETINET_GETCRED);
897490d50b6SBrian Feldman 	if (error)
898490d50b6SBrian Feldman 		return (error);
899490d50b6SBrian Feldman 	error = SYSCTL_IN(req, addrs, sizeof(addrs));
900490d50b6SBrian Feldman 	if (error)
901490d50b6SBrian Feldman 		return (error);
902d797164aSGleb Smirnoff 	NET_EPOCH_ENTER(et);
903fa046d87SRobert Watson 	inp = in_pcblookup(&V_udbinfo, addrs[1].sin_addr, addrs[1].sin_port,
904fa046d87SRobert Watson 	    addrs[0].sin_addr, addrs[0].sin_port,
905fa046d87SRobert Watson 	    INPLOOKUP_WILDCARD | INPLOOKUP_RLOCKPCB, NULL);
906d797164aSGleb Smirnoff 	NET_EPOCH_EXIT(et);
9079622e84fSRobert Watson 	if (inp != NULL) {
908fa046d87SRobert Watson 		INP_RLOCK_ASSERT(inp);
9099622e84fSRobert Watson 		if (inp->inp_socket == NULL)
9109622e84fSRobert Watson 			error = ENOENT;
9119622e84fSRobert Watson 		if (error == 0)
912f08ef6c5SBjoern A. Zeeb 			error = cr_canseeinpcb(req->td->td_ucred, inp);
9139622e84fSRobert Watson 		if (error == 0)
91486d02c5cSBjoern A. Zeeb 			cru2x(inp->inp_cred, &xuc);
9159622e84fSRobert Watson 		INP_RUNLOCK(inp);
916fa046d87SRobert Watson 	} else
9179622e84fSRobert Watson 		error = ENOENT;
9180e1eebb8SDon Lewis 	if (error == 0)
9190e1eebb8SDon Lewis 		error = SYSCTL_OUT(req, &xuc, sizeof(struct xucred));
920490d50b6SBrian Feldman 	return (error);
921490d50b6SBrian Feldman }
922490d50b6SBrian Feldman 
9237ce87f12SDavid Malone SYSCTL_PROC(_net_inet_udp, OID_AUTO, getcred,
9247029da5cSPawel Biernacki     CTLTYPE_OPAQUE | CTLFLAG_RW | CTLFLAG_PRISON | CTLFLAG_MPSAFE,
9257029da5cSPawel Biernacki     0, 0, udp_getcred, "S,xucred",
9267029da5cSPawel Biernacki     "Get the xucred of a UDP connection");
92779288c11SBjoern A. Zeeb #endif /* INET */
928490d50b6SBrian Feldman 
9297b495c44SVANHULLEBUS Yvan int
9307b495c44SVANHULLEBUS Yvan udp_ctloutput(struct socket *so, struct sockopt *sopt)
9317b495c44SVANHULLEBUS Yvan {
9327b495c44SVANHULLEBUS Yvan 	struct inpcb *inp;
9337b495c44SVANHULLEBUS Yvan 	struct udpcb *up;
934e06e816fSKevin Lo 	int isudplite, error, optval;
9357b495c44SVANHULLEBUS Yvan 
936e06e816fSKevin Lo 	error = 0;
937e06e816fSKevin Lo 	isudplite = (so->so_proto->pr_protocol == IPPROTO_UDPLITE) ? 1 : 0;
9387b495c44SVANHULLEBUS Yvan 	inp = sotoinpcb(so);
9397b495c44SVANHULLEBUS Yvan 	KASSERT(inp != NULL, ("%s: inp == NULL", __func__));
9407b495c44SVANHULLEBUS Yvan 	INP_WLOCK(inp);
941e06e816fSKevin Lo 	if (sopt->sopt_level != so->so_proto->pr_protocol) {
9427b495c44SVANHULLEBUS Yvan #ifdef INET6
9437b495c44SVANHULLEBUS Yvan 		if (INP_CHECK_SOCKAF(so, AF_INET6)) {
9447b495c44SVANHULLEBUS Yvan 			INP_WUNLOCK(inp);
9457b495c44SVANHULLEBUS Yvan 			error = ip6_ctloutput(so, sopt);
94679288c11SBjoern A. Zeeb 		}
9477b495c44SVANHULLEBUS Yvan #endif
94879288c11SBjoern A. Zeeb #if defined(INET) && defined(INET6)
94979288c11SBjoern A. Zeeb 		else
95079288c11SBjoern A. Zeeb #endif
95179288c11SBjoern A. Zeeb #ifdef INET
95279288c11SBjoern A. Zeeb 		{
9537b495c44SVANHULLEBUS Yvan 			INP_WUNLOCK(inp);
9547b495c44SVANHULLEBUS Yvan 			error = ip_ctloutput(so, sopt);
9557b495c44SVANHULLEBUS Yvan 		}
9567b495c44SVANHULLEBUS Yvan #endif
9577b495c44SVANHULLEBUS Yvan 		return (error);
9587b495c44SVANHULLEBUS Yvan 	}
9597b495c44SVANHULLEBUS Yvan 
9607b495c44SVANHULLEBUS Yvan 	switch (sopt->sopt_dir) {
9617b495c44SVANHULLEBUS Yvan 	case SOPT_SET:
9627b495c44SVANHULLEBUS Yvan 		switch (sopt->sopt_name) {
963fcf59617SAndrey V. Elsukov #if defined(IPSEC) || defined(IPSEC_SUPPORT)
964fcf59617SAndrey V. Elsukov #ifdef INET
9657b495c44SVANHULLEBUS Yvan 		case UDP_ENCAP:
966fcf59617SAndrey V. Elsukov 			if (!IPSEC_ENABLED(ipv4)) {
9677b495c44SVANHULLEBUS Yvan 				INP_WUNLOCK(inp);
968fcf59617SAndrey V. Elsukov 				return (ENOPROTOOPT);
9697b495c44SVANHULLEBUS Yvan 			}
970fcf59617SAndrey V. Elsukov 			error = UDPENCAP_PCBCTL(inp, sopt);
9717b495c44SVANHULLEBUS Yvan 			break;
972fcf59617SAndrey V. Elsukov #endif /* INET */
973fcf59617SAndrey V. Elsukov #endif /* IPSEC */
974e06e816fSKevin Lo 		case UDPLITE_SEND_CSCOV:
975e06e816fSKevin Lo 		case UDPLITE_RECV_CSCOV:
976e06e816fSKevin Lo 			if (!isudplite) {
977e06e816fSKevin Lo 				INP_WUNLOCK(inp);
978e06e816fSKevin Lo 				error = ENOPROTOOPT;
979e06e816fSKevin Lo 				break;
980e06e816fSKevin Lo 			}
981e06e816fSKevin Lo 			INP_WUNLOCK(inp);
982e06e816fSKevin Lo 			error = sooptcopyin(sopt, &optval, sizeof(optval),
983e06e816fSKevin Lo 			    sizeof(optval));
984e06e816fSKevin Lo 			if (error != 0)
985e06e816fSKevin Lo 				break;
986e06e816fSKevin Lo 			inp = sotoinpcb(so);
987e06e816fSKevin Lo 			KASSERT(inp != NULL, ("%s: inp == NULL", __func__));
988e06e816fSKevin Lo 			INP_WLOCK(inp);
989e06e816fSKevin Lo 			up = intoudpcb(inp);
990e06e816fSKevin Lo 			KASSERT(up != NULL, ("%s: up == NULL", __func__));
99103f90784SMichael Tuexen 			if ((optval != 0 && optval < 8) || (optval > 65535)) {
992e06e816fSKevin Lo 				INP_WUNLOCK(inp);
993e06e816fSKevin Lo 				error = EINVAL;
994e06e816fSKevin Lo 				break;
995e06e816fSKevin Lo 			}
996e06e816fSKevin Lo 			if (sopt->sopt_name == UDPLITE_SEND_CSCOV)
997e06e816fSKevin Lo 				up->u_txcslen = optval;
998e06e816fSKevin Lo 			else
999e06e816fSKevin Lo 				up->u_rxcslen = optval;
1000e06e816fSKevin Lo 			INP_WUNLOCK(inp);
1001e06e816fSKevin Lo 			break;
10027b495c44SVANHULLEBUS Yvan 		default:
10037b495c44SVANHULLEBUS Yvan 			INP_WUNLOCK(inp);
10047b495c44SVANHULLEBUS Yvan 			error = ENOPROTOOPT;
10057b495c44SVANHULLEBUS Yvan 			break;
10067b495c44SVANHULLEBUS Yvan 		}
10077b495c44SVANHULLEBUS Yvan 		break;
10087b495c44SVANHULLEBUS Yvan 	case SOPT_GET:
10097b495c44SVANHULLEBUS Yvan 		switch (sopt->sopt_name) {
1010fcf59617SAndrey V. Elsukov #if defined(IPSEC) || defined(IPSEC_SUPPORT)
1011fcf59617SAndrey V. Elsukov #ifdef INET
10127b495c44SVANHULLEBUS Yvan 		case UDP_ENCAP:
1013fcf59617SAndrey V. Elsukov 			if (!IPSEC_ENABLED(ipv4)) {
10147b495c44SVANHULLEBUS Yvan 				INP_WUNLOCK(inp);
1015fcf59617SAndrey V. Elsukov 				return (ENOPROTOOPT);
1016fcf59617SAndrey V. Elsukov 			}
1017fcf59617SAndrey V. Elsukov 			error = UDPENCAP_PCBCTL(inp, sopt);
10187b495c44SVANHULLEBUS Yvan 			break;
1019fcf59617SAndrey V. Elsukov #endif /* INET */
1020fcf59617SAndrey V. Elsukov #endif /* IPSEC */
1021e06e816fSKevin Lo 		case UDPLITE_SEND_CSCOV:
1022e06e816fSKevin Lo 		case UDPLITE_RECV_CSCOV:
1023e06e816fSKevin Lo 			if (!isudplite) {
1024e06e816fSKevin Lo 				INP_WUNLOCK(inp);
1025e06e816fSKevin Lo 				error = ENOPROTOOPT;
1026e06e816fSKevin Lo 				break;
1027e06e816fSKevin Lo 			}
1028e06e816fSKevin Lo 			up = intoudpcb(inp);
1029e06e816fSKevin Lo 			KASSERT(up != NULL, ("%s: up == NULL", __func__));
1030e06e816fSKevin Lo 			if (sopt->sopt_name == UDPLITE_SEND_CSCOV)
1031e06e816fSKevin Lo 				optval = up->u_txcslen;
1032e06e816fSKevin Lo 			else
1033e06e816fSKevin Lo 				optval = up->u_rxcslen;
1034e06e816fSKevin Lo 			INP_WUNLOCK(inp);
1035e06e816fSKevin Lo 			error = sooptcopyout(sopt, &optval, sizeof(optval));
1036e06e816fSKevin Lo 			break;
10377b495c44SVANHULLEBUS Yvan 		default:
10387b495c44SVANHULLEBUS Yvan 			INP_WUNLOCK(inp);
10397b495c44SVANHULLEBUS Yvan 			error = ENOPROTOOPT;
10407b495c44SVANHULLEBUS Yvan 			break;
10417b495c44SVANHULLEBUS Yvan 		}
10427b495c44SVANHULLEBUS Yvan 		break;
10437b495c44SVANHULLEBUS Yvan 	}
10447b495c44SVANHULLEBUS Yvan 	return (error);
10457b495c44SVANHULLEBUS Yvan }
10467b495c44SVANHULLEBUS Yvan 
104779288c11SBjoern A. Zeeb #ifdef INET
1048a9839c4aSBjoern A. Zeeb #ifdef INET6
1049a9839c4aSBjoern A. Zeeb /* The logic here is derived from ip6_setpktopt(). See comments there. */
1050a9839c4aSBjoern A. Zeeb static int
1051a9839c4aSBjoern A. Zeeb udp_v4mapped_pktinfo(struct cmsghdr *cm, struct sockaddr_in * src,
1052a9839c4aSBjoern A. Zeeb     struct inpcb *inp, int flags)
1053a9839c4aSBjoern A. Zeeb {
1054a9839c4aSBjoern A. Zeeb 	struct ifnet *ifp;
1055a9839c4aSBjoern A. Zeeb 	struct in6_pktinfo *pktinfo;
1056a9839c4aSBjoern A. Zeeb 	struct in_addr ia;
1057a9839c4aSBjoern A. Zeeb 
1058a9839c4aSBjoern A. Zeeb 	if ((flags & PRUS_IPV6) == 0)
1059a9839c4aSBjoern A. Zeeb 		return (0);
1060a9839c4aSBjoern A. Zeeb 
1061a9839c4aSBjoern A. Zeeb 	if (cm->cmsg_level != IPPROTO_IPV6)
1062a9839c4aSBjoern A. Zeeb 		return (0);
1063a9839c4aSBjoern A. Zeeb 
1064a9839c4aSBjoern A. Zeeb 	if  (cm->cmsg_type != IPV6_2292PKTINFO &&
1065a9839c4aSBjoern A. Zeeb 	    cm->cmsg_type != IPV6_PKTINFO)
1066a9839c4aSBjoern A. Zeeb 		return (0);
1067a9839c4aSBjoern A. Zeeb 
1068a9839c4aSBjoern A. Zeeb 	if (cm->cmsg_len !=
1069a9839c4aSBjoern A. Zeeb 	    CMSG_LEN(sizeof(struct in6_pktinfo)))
1070a9839c4aSBjoern A. Zeeb 		return (EINVAL);
1071a9839c4aSBjoern A. Zeeb 
1072a9839c4aSBjoern A. Zeeb 	pktinfo = (struct in6_pktinfo *)CMSG_DATA(cm);
1073a9839c4aSBjoern A. Zeeb 	if (!IN6_IS_ADDR_V4MAPPED(&pktinfo->ipi6_addr) &&
1074a9839c4aSBjoern A. Zeeb 	    !IN6_IS_ADDR_UNSPECIFIED(&pktinfo->ipi6_addr))
1075a9839c4aSBjoern A. Zeeb 		return (EINVAL);
1076a9839c4aSBjoern A. Zeeb 
1077a9839c4aSBjoern A. Zeeb 	/* Validate the interface index if specified. */
1078a9839c4aSBjoern A. Zeeb 	if (pktinfo->ipi6_ifindex) {
1079d74b7baeSGleb Smirnoff 		struct epoch_tracker et;
1080d74b7baeSGleb Smirnoff 
1081d74b7baeSGleb Smirnoff 		NET_EPOCH_ENTER(et);
1082a9839c4aSBjoern A. Zeeb 		ifp = ifnet_byindex(pktinfo->ipi6_ifindex);
1083d74b7baeSGleb Smirnoff 		NET_EPOCH_EXIT(et);	/* XXXGL: unsafe ifp */
1084a9839c4aSBjoern A. Zeeb 		if (ifp == NULL)
1085a9839c4aSBjoern A. Zeeb 			return (ENXIO);
1086d74b7baeSGleb Smirnoff 	} else
1087d74b7baeSGleb Smirnoff 		ifp = NULL;
1088a9839c4aSBjoern A. Zeeb 	if (ifp != NULL && !IN6_IS_ADDR_UNSPECIFIED(&pktinfo->ipi6_addr)) {
1089a9839c4aSBjoern A. Zeeb 		ia.s_addr = pktinfo->ipi6_addr.s6_addr32[3];
1090a9839c4aSBjoern A. Zeeb 		if (in_ifhasaddr(ifp, ia) == 0)
1091a9839c4aSBjoern A. Zeeb 			return (EADDRNOTAVAIL);
1092a9839c4aSBjoern A. Zeeb 	}
1093a9839c4aSBjoern A. Zeeb 
1094a9839c4aSBjoern A. Zeeb 	bzero(src, sizeof(*src));
1095a9839c4aSBjoern A. Zeeb 	src->sin_family = AF_INET;
1096a9839c4aSBjoern A. Zeeb 	src->sin_len = sizeof(*src);
1097a9839c4aSBjoern A. Zeeb 	src->sin_port = inp->inp_lport;
1098a9839c4aSBjoern A. Zeeb 	src->sin_addr.s_addr = pktinfo->ipi6_addr.s6_addr32[3];
1099a9839c4aSBjoern A. Zeeb 
1100a9839c4aSBjoern A. Zeeb 	return (0);
1101a9839c4aSBjoern A. Zeeb }
1102a9839c4aSBjoern A. Zeeb #endif
1103a9839c4aSBjoern A. Zeeb 
1104490d50b6SBrian Feldman static int
11053329b236SRobert Watson udp_output(struct inpcb *inp, struct mbuf *m, struct sockaddr *addr,
1106a9839c4aSBjoern A. Zeeb     struct mbuf *control, struct thread *td, int flags)
1107df8bae1dSRodney W. Grimes {
11083329b236SRobert Watson 	struct udpiphdr *ui;
11093329b236SRobert Watson 	int len = m->m_pkthdr.len;
111090162a4eSIan Dowse 	struct in_addr faddr, laddr;
1111c557ae16SIan Dowse 	struct cmsghdr *cm;
1112e06e816fSKevin Lo 	struct inpcbinfo *pcbinfo;
1113c557ae16SIan Dowse 	struct sockaddr_in *sin, src;
11146573d758SMatt Macy 	struct epoch_tracker et;
1115e06e816fSKevin Lo 	int cscov_partial = 0;
111690162a4eSIan Dowse 	int error = 0;
1117374ce248SMitchell Horne 	int ipflags = 0;
111890162a4eSIan Dowse 	u_short fport, lport;
1119f584d74bSMichael Tuexen 	u_char tos;
1120e06e816fSKevin Lo 	uint8_t pr;
1121e06e816fSKevin Lo 	uint16_t cscov = 0;
11229d3ddf43SAdrian Chadd 	uint32_t flowid = 0;
1123c2529042SHans Petter Selasky 	uint8_t flowtype = M_HASHTYPE_NONE;
1124df8bae1dSRodney W. Grimes 
1125430d30d8SBill Fenner 	if (len + sizeof(struct udpiphdr) > IP_MAXPACKET) {
1126c557ae16SIan Dowse 		if (control)
1127c557ae16SIan Dowse 			m_freem(control);
11285c32ea65SRobert Watson 		m_freem(m);
11293329b236SRobert Watson 		return (EMSGSIZE);
1130430d30d8SBill Fenner 	}
1131430d30d8SBill Fenner 
11321b7f0384SBruce M Simpson 	src.sin_family = 0;
113384cc0778SGeorge V. Neville-Neil 	sin = (struct sockaddr_in *)addr;
11342435e507SGleb Smirnoff 
1135eafaa1bcSBjoern A. Zeeb 	/*
11362435e507SGleb Smirnoff 	 * udp_output() may need to temporarily bind or connect the current
11372435e507SGleb Smirnoff 	 * inpcb.  As such, we don't know up front whether we will need the
11382435e507SGleb Smirnoff 	 * pcbinfo lock or not.  Do any work to decide what is needed up
11392435e507SGleb Smirnoff 	 * front before acquiring any locks.
11402435e507SGleb Smirnoff 	 *
11412435e507SGleb Smirnoff 	 * We will need network epoch in either case, to safely lookup into
11422435e507SGleb Smirnoff 	 * pcb hash.
1143eafaa1bcSBjoern A. Zeeb 	 */
11442435e507SGleb Smirnoff 	if (sin == NULL ||
11452435e507SGleb Smirnoff 	    (inp->inp_laddr.s_addr == INADDR_ANY && inp->inp_lport == 0))
11462435e507SGleb Smirnoff 		INP_WLOCK(inp);
11472435e507SGleb Smirnoff 	else
11480cfdff24SBjoern A. Zeeb 		INP_RLOCK(inp);
11492435e507SGleb Smirnoff 	NET_EPOCH_ENTER(et);
1150f584d74bSMichael Tuexen 	tos = inp->inp_ip_tos;
1151c557ae16SIan Dowse 	if (control != NULL) {
1152c557ae16SIan Dowse 		/*
11533329b236SRobert Watson 		 * XXX: Currently, we assume all the optional information is
11543329b236SRobert Watson 		 * stored in a single mbuf.
1155c557ae16SIan Dowse 		 */
1156c557ae16SIan Dowse 		if (control->m_next) {
1157c557ae16SIan Dowse 			m_freem(control);
11582435e507SGleb Smirnoff 			error = EINVAL;
11592435e507SGleb Smirnoff 			goto release;
1160c557ae16SIan Dowse 		}
1161c557ae16SIan Dowse 		for (; control->m_len > 0;
1162c557ae16SIan Dowse 		    control->m_data += CMSG_ALIGN(cm->cmsg_len),
1163c557ae16SIan Dowse 		    control->m_len -= CMSG_ALIGN(cm->cmsg_len)) {
1164c557ae16SIan Dowse 			cm = mtod(control, struct cmsghdr *);
1165af1ee11dSRobert Watson 			if (control->m_len < sizeof(*cm) || cm->cmsg_len == 0
1166af1ee11dSRobert Watson 			    || cm->cmsg_len > control->m_len) {
1167c557ae16SIan Dowse 				error = EINVAL;
1168c557ae16SIan Dowse 				break;
1169c557ae16SIan Dowse 			}
1170a9839c4aSBjoern A. Zeeb #ifdef INET6
1171a9839c4aSBjoern A. Zeeb 			error = udp_v4mapped_pktinfo(cm, &src, inp, flags);
1172a9839c4aSBjoern A. Zeeb 			if (error != 0)
1173a9839c4aSBjoern A. Zeeb 				break;
1174a9839c4aSBjoern A. Zeeb #endif
1175c557ae16SIan Dowse 			if (cm->cmsg_level != IPPROTO_IP)
1176c557ae16SIan Dowse 				continue;
1177c557ae16SIan Dowse 
1178c557ae16SIan Dowse 			switch (cm->cmsg_type) {
1179c557ae16SIan Dowse 			case IP_SENDSRCADDR:
1180c557ae16SIan Dowse 				if (cm->cmsg_len !=
1181c557ae16SIan Dowse 				    CMSG_LEN(sizeof(struct in_addr))) {
1182c557ae16SIan Dowse 					error = EINVAL;
1183c557ae16SIan Dowse 					break;
1184c557ae16SIan Dowse 				}
1185c557ae16SIan Dowse 				bzero(&src, sizeof(src));
1186c557ae16SIan Dowse 				src.sin_family = AF_INET;
1187c557ae16SIan Dowse 				src.sin_len = sizeof(src);
1188c557ae16SIan Dowse 				src.sin_port = inp->inp_lport;
1189af1ee11dSRobert Watson 				src.sin_addr =
1190af1ee11dSRobert Watson 				    *(struct in_addr *)CMSG_DATA(cm);
1191c557ae16SIan Dowse 				break;
1192af1ee11dSRobert Watson 
1193f584d74bSMichael Tuexen 			case IP_TOS:
1194f584d74bSMichael Tuexen 				if (cm->cmsg_len != CMSG_LEN(sizeof(u_char))) {
1195f584d74bSMichael Tuexen 					error = EINVAL;
1196f584d74bSMichael Tuexen 					break;
1197f584d74bSMichael Tuexen 				}
1198f584d74bSMichael Tuexen 				tos = *(u_char *)CMSG_DATA(cm);
1199f584d74bSMichael Tuexen 				break;
1200f584d74bSMichael Tuexen 
12019d3ddf43SAdrian Chadd 			case IP_FLOWID:
12029d3ddf43SAdrian Chadd 				if (cm->cmsg_len != CMSG_LEN(sizeof(uint32_t))) {
12039d3ddf43SAdrian Chadd 					error = EINVAL;
12049d3ddf43SAdrian Chadd 					break;
12059d3ddf43SAdrian Chadd 				}
12069d3ddf43SAdrian Chadd 				flowid = *(uint32_t *) CMSG_DATA(cm);
12079d3ddf43SAdrian Chadd 				break;
12089d3ddf43SAdrian Chadd 
12099d3ddf43SAdrian Chadd 			case IP_FLOWTYPE:
12109d3ddf43SAdrian Chadd 				if (cm->cmsg_len != CMSG_LEN(sizeof(uint32_t))) {
12119d3ddf43SAdrian Chadd 					error = EINVAL;
12129d3ddf43SAdrian Chadd 					break;
12139d3ddf43SAdrian Chadd 				}
1214c2529042SHans Petter Selasky 				flowtype = *(uint32_t *) CMSG_DATA(cm);
12159d3ddf43SAdrian Chadd 				break;
12169d3ddf43SAdrian Chadd 
12179d3ddf43SAdrian Chadd #ifdef	RSS
12189d3ddf43SAdrian Chadd 			case IP_RSSBUCKETID:
12199d3ddf43SAdrian Chadd 				if (cm->cmsg_len != CMSG_LEN(sizeof(uint32_t))) {
12209d3ddf43SAdrian Chadd 					error = EINVAL;
12219d3ddf43SAdrian Chadd 					break;
12229d3ddf43SAdrian Chadd 				}
12239d3ddf43SAdrian Chadd 				/* This is just a placeholder for now */
12249d3ddf43SAdrian Chadd 				break;
12259d3ddf43SAdrian Chadd #endif	/* RSS */
1226c557ae16SIan Dowse 			default:
1227c557ae16SIan Dowse 				error = ENOPROTOOPT;
1228c557ae16SIan Dowse 				break;
1229c557ae16SIan Dowse 			}
1230c557ae16SIan Dowse 			if (error)
1231c557ae16SIan Dowse 				break;
1232c557ae16SIan Dowse 		}
1233c557ae16SIan Dowse 		m_freem(control);
1234d8acd268SMark Johnston 		control = NULL;
1235c557ae16SIan Dowse 	}
12362435e507SGleb Smirnoff 	if (error)
12372435e507SGleb Smirnoff 		goto release;
12385c32ea65SRobert Watson 
1239e06e816fSKevin Lo 	pr = inp->inp_socket->so_proto->pr_protocol;
1240a86e5c96SBjoern A. Zeeb 	pcbinfo = udp_get_inpcbinfo(pr);
12415c32ea65SRobert Watson 
12421b7f0384SBruce M Simpson 	/*
12431b7f0384SBruce M Simpson 	 * If the IP_SENDSRCADDR control message was specified, override the
12441b7f0384SBruce M Simpson 	 * source address for this datagram.  Its use is invalidated if the
12451b7f0384SBruce M Simpson 	 * address thus specified is incomplete or clobbers other inpcbs.
12461b7f0384SBruce M Simpson 	 */
124790162a4eSIan Dowse 	laddr = inp->inp_laddr;
124890162a4eSIan Dowse 	lport = inp->inp_lport;
12491b7f0384SBruce M Simpson 	if (src.sin_family == AF_INET) {
12501b7f0384SBruce M Simpson 		if ((lport == 0) ||
12511b7f0384SBruce M Simpson 		    (laddr.s_addr == INADDR_ANY &&
12521b7f0384SBruce M Simpson 		     src.sin_addr.s_addr == INADDR_ANY)) {
1253c557ae16SIan Dowse 			error = EINVAL;
1254c557ae16SIan Dowse 			goto release;
1255c557ae16SIan Dowse 		}
1256db0ac6deSCy Schubert 		INP_HASH_WLOCK(pcbinfo);
1257c557ae16SIan Dowse 		error = in_pcbbind_setup(inp, (struct sockaddr *)&src,
1258b0330ed9SPawel Jakub Dawidek 		    &laddr.s_addr, &lport, td->td_ucred);
1259db0ac6deSCy Schubert 		INP_HASH_WUNLOCK(pcbinfo);
1260c557ae16SIan Dowse 		if (error)
1261c557ae16SIan Dowse 			goto release;
1262c557ae16SIan Dowse 	}
1263c557ae16SIan Dowse 
12643144b7d3SRobert Watson 	/*
12653144b7d3SRobert Watson 	 * If a UDP socket has been connected, then a local address/port will
12663144b7d3SRobert Watson 	 * have been selected and bound.
12673144b7d3SRobert Watson 	 *
126843cc0bc1SRobert Watson 	 * If a UDP socket has not been connected to, then an explicit
12693144b7d3SRobert Watson 	 * destination address must be used, in which case a local
12703144b7d3SRobert Watson 	 * address/port may not have been selected and bound.
12713144b7d3SRobert Watson 	 */
127243cc0bc1SRobert Watson 	if (sin != NULL) {
1273c4d585aeSRobert Watson 		INP_LOCK_ASSERT(inp);
1274df8bae1dSRodney W. Grimes 		if (inp->inp_faddr.s_addr != INADDR_ANY) {
1275df8bae1dSRodney W. Grimes 			error = EISCONN;
1276df8bae1dSRodney W. Grimes 			goto release;
1277df8bae1dSRodney W. Grimes 		}
12783144b7d3SRobert Watson 
12793144b7d3SRobert Watson 		/*
12803144b7d3SRobert Watson 		 * Jail may rewrite the destination address, so let it do
12813144b7d3SRobert Watson 		 * that before we use it.
12823144b7d3SRobert Watson 		 */
1283b89e82ddSJamie Gritton 		error = prison_remote_ip4(td->td_ucred, &sin->sin_addr);
1284b89e82ddSJamie Gritton 		if (error)
1285413628a7SBjoern A. Zeeb 			goto release;
12863144b7d3SRobert Watson 
12873144b7d3SRobert Watson 		/*
128843cc0bc1SRobert Watson 		 * If a local address or port hasn't yet been selected, or if
128943cc0bc1SRobert Watson 		 * the destination address needs to be rewritten due to using
129043cc0bc1SRobert Watson 		 * a special INADDR_ constant, invoke in_pcbconnect_setup()
129143cc0bc1SRobert Watson 		 * to do the heavy lifting.  Once a port is selected, we
129243cc0bc1SRobert Watson 		 * commit the binding back to the socket; we also commit the
129343cc0bc1SRobert Watson 		 * binding of the address if in jail.
129443cc0bc1SRobert Watson 		 *
129543cc0bc1SRobert Watson 		 * If we already have a valid binding and we're not
129643cc0bc1SRobert Watson 		 * requesting a destination address rewrite, use a fast path.
12973144b7d3SRobert Watson 		 */
129843cc0bc1SRobert Watson 		if (inp->inp_laddr.s_addr == INADDR_ANY ||
129943cc0bc1SRobert Watson 		    inp->inp_lport == 0 ||
130043cc0bc1SRobert Watson 		    sin->sin_addr.s_addr == INADDR_ANY ||
130143cc0bc1SRobert Watson 		    sin->sin_addr.s_addr == INADDR_BROADCAST) {
1302db0ac6deSCy Schubert 			INP_HASH_WLOCK(pcbinfo);
130343cc0bc1SRobert Watson 			error = in_pcbconnect_setup(inp, addr, &laddr.s_addr,
130443cc0bc1SRobert Watson 			    &lport, &faddr.s_addr, &fport, NULL,
130543cc0bc1SRobert Watson 			    td->td_ucred);
1306db0ac6deSCy Schubert 			if (error) {
1307db0ac6deSCy Schubert 				INP_HASH_WUNLOCK(pcbinfo);
130890162a4eSIan Dowse 				goto release;
1309db0ac6deSCy Schubert 			}
131090162a4eSIan Dowse 
131143cc0bc1SRobert Watson 			/*
131243cc0bc1SRobert Watson 			 * XXXRW: Why not commit the port if the address is
131343cc0bc1SRobert Watson 			 * !INADDR_ANY?
131443cc0bc1SRobert Watson 			 */
131590162a4eSIan Dowse 			/* Commit the local port if newly assigned. */
131690162a4eSIan Dowse 			if (inp->inp_laddr.s_addr == INADDR_ANY &&
131790162a4eSIan Dowse 			    inp->inp_lport == 0) {
1318c4d585aeSRobert Watson 				INP_WLOCK_ASSERT(inp);
13193a1757b9SGleb Smirnoff 				/*
132043cc0bc1SRobert Watson 				 * Remember addr if jailed, to prevent
132143cc0bc1SRobert Watson 				 * rebinding.
13223a1757b9SGleb Smirnoff 				 */
13230304c731SJamie Gritton 				if (prison_flag(td->td_ucred, PR_IP4))
13243a1757b9SGleb Smirnoff 					inp->inp_laddr = laddr;
132590162a4eSIan Dowse 				inp->inp_lport = lport;
13262435e507SGleb Smirnoff 				error = in_pcbinshash(inp);
13272435e507SGleb Smirnoff 				INP_HASH_WUNLOCK(pcbinfo);
13282435e507SGleb Smirnoff 				if (error != 0) {
132990162a4eSIan Dowse 					inp->inp_lport = 0;
133090162a4eSIan Dowse 					error = EAGAIN;
1331df8bae1dSRodney W. Grimes 					goto release;
1332df8bae1dSRodney W. Grimes 				}
133390162a4eSIan Dowse 				inp->inp_flags |= INP_ANONPORT;
1334db0ac6deSCy Schubert 			} else
1335db0ac6deSCy Schubert 				INP_HASH_WUNLOCK(pcbinfo);
1336df8bae1dSRodney W. Grimes 		} else {
133743cc0bc1SRobert Watson 			faddr = sin->sin_addr;
133843cc0bc1SRobert Watson 			fport = sin->sin_port;
133943cc0bc1SRobert Watson 		}
134043cc0bc1SRobert Watson 	} else {
1341c4d585aeSRobert Watson 		INP_LOCK_ASSERT(inp);
134290162a4eSIan Dowse 		faddr = inp->inp_faddr;
134390162a4eSIan Dowse 		fport = inp->inp_fport;
134490162a4eSIan Dowse 		if (faddr.s_addr == INADDR_ANY) {
1345df8bae1dSRodney W. Grimes 			error = ENOTCONN;
1346df8bae1dSRodney W. Grimes 			goto release;
1347df8bae1dSRodney W. Grimes 		}
1348df8bae1dSRodney W. Grimes 	}
1349e6ccd709SRobert Watson 
1350df8bae1dSRodney W. Grimes 	/*
1351e6ccd709SRobert Watson 	 * Calculate data length and get a mbuf for UDP, IP, and possible
1352392e8407SRobert Watson 	 * link-layer headers.  Immediate slide the data pointer back forward
1353392e8407SRobert Watson 	 * since we won't use that space at this layer.
1354df8bae1dSRodney W. Grimes 	 */
1355eb1b1807SGleb Smirnoff 	M_PREPEND(m, sizeof(struct udpiphdr) + max_linkhdr, M_NOWAIT);
1356e6ccd709SRobert Watson 	if (m == NULL) {
1357df8bae1dSRodney W. Grimes 		error = ENOBUFS;
135849b19bfcSBruce M Simpson 		goto release;
1359df8bae1dSRodney W. Grimes 	}
1360e6ccd709SRobert Watson 	m->m_data += max_linkhdr;
1361e6ccd709SRobert Watson 	m->m_len -= max_linkhdr;
1362392e8407SRobert Watson 	m->m_pkthdr.len -= max_linkhdr;
1363df8bae1dSRodney W. Grimes 
1364df8bae1dSRodney W. Grimes 	/*
13653329b236SRobert Watson 	 * Fill in mbuf with extended UDP header and addresses and length put
13663329b236SRobert Watson 	 * into network format.
1367df8bae1dSRodney W. Grimes 	 */
1368df8bae1dSRodney W. Grimes 	ui = mtod(m, struct udpiphdr *);
1369db4f9cc7SJonathan Lemon 	bzero(ui->ui_x1, sizeof(ui->ui_x1));	/* XXX still needed? */
137034fc9072SMichael Tuexen 	ui->ui_v = IPVERSION << 4;
1371e06e816fSKevin Lo 	ui->ui_pr = pr;
137290162a4eSIan Dowse 	ui->ui_src = laddr;
137390162a4eSIan Dowse 	ui->ui_dst = faddr;
137490162a4eSIan Dowse 	ui->ui_sport = lport;
137590162a4eSIan Dowse 	ui->ui_dport = fport;
1376db4f9cc7SJonathan Lemon 	ui->ui_ulen = htons((u_short)len + sizeof(struct udphdr));
1377e06e816fSKevin Lo 	if (pr == IPPROTO_UDPLITE) {
1378e06e816fSKevin Lo 		struct udpcb *up;
1379e06e816fSKevin Lo 		uint16_t plen;
1380e06e816fSKevin Lo 
1381e06e816fSKevin Lo 		up = intoudpcb(inp);
1382e06e816fSKevin Lo 		cscov = up->u_txcslen;
1383e06e816fSKevin Lo 		plen = (u_short)len + sizeof(struct udphdr);
1384e06e816fSKevin Lo 		if (cscov >= plen)
1385e06e816fSKevin Lo 			cscov = 0;
1386e06e816fSKevin Lo 		ui->ui_len = htons(plen);
1387e06e816fSKevin Lo 		ui->ui_ulen = htons(cscov);
1388e06e816fSKevin Lo 		/*
1389e06e816fSKevin Lo 		 * For UDP-Lite, checksum coverage length of zero means
1390e06e816fSKevin Lo 		 * the entire UDPLite packet is covered by the checksum.
1391e06e816fSKevin Lo 		 */
1392e06e816fSKevin Lo 		cscov_partial = (cscov == 0) ? 0 : 1;
139334fc9072SMichael Tuexen 	}
1394df8bae1dSRodney W. Grimes 
1395b2828ad2SAndre Oppermann 	/*
1396b2828ad2SAndre Oppermann 	 * Set the Don't Fragment bit in the IP header.
1397b2828ad2SAndre Oppermann 	 */
1398b2828ad2SAndre Oppermann 	if (inp->inp_flags & INP_DONTFRAG) {
1399b2828ad2SAndre Oppermann 		struct ip *ip;
14003329b236SRobert Watson 
1401b2828ad2SAndre Oppermann 		ip = (struct ip *)&ui->ui_i;
14028f134647SGleb Smirnoff 		ip->ip_off |= htons(IP_DF);
1403b2828ad2SAndre Oppermann 	}
1404b2828ad2SAndre Oppermann 
1405b5d47ff5SJohn-Mark Gurney 	if (inp->inp_socket->so_options & SO_DONTROUTE)
1406b5d47ff5SJohn-Mark Gurney 		ipflags |= IP_ROUTETOIF;
1407b5d47ff5SJohn-Mark Gurney 	if (inp->inp_socket->so_options & SO_BROADCAST)
1408b5d47ff5SJohn-Mark Gurney 		ipflags |= IP_ALLOWBROADCAST;
14096fbfd582SAndre Oppermann 	if (inp->inp_flags & INP_ONESBCAST)
14108afa2304SBruce M Simpson 		ipflags |= IP_SENDONES;
14118afa2304SBruce M Simpson 
14121175d9d5SRobert Watson #ifdef MAC
14131175d9d5SRobert Watson 	mac_inpcb_create_mbuf(inp, m);
14141175d9d5SRobert Watson #endif
14151175d9d5SRobert Watson 
1416df8bae1dSRodney W. Grimes 	/*
1417db4f9cc7SJonathan Lemon 	 * Set up checksum and output datagram.
1418df8bae1dSRodney W. Grimes 	 */
1419e06e816fSKevin Lo 	ui->ui_sum = 0;
1420a485f139SMichael Tuexen 	if (pr == IPPROTO_UDPLITE) {
1421e06e816fSKevin Lo 		if (inp->inp_flags & INP_ONESBCAST)
1422e06e816fSKevin Lo 			faddr.s_addr = INADDR_BROADCAST;
1423a485f139SMichael Tuexen 		if (cscov_partial) {
1424e06e816fSKevin Lo 			if ((ui->ui_sum = in_cksum(m, sizeof(struct ip) + cscov)) == 0)
1425e06e816fSKevin Lo 				ui->ui_sum = 0xffff;
1426a485f139SMichael Tuexen 		} else {
1427a485f139SMichael Tuexen 			if ((ui->ui_sum = in_cksum(m, sizeof(struct udpiphdr) + len)) == 0)
1428a485f139SMichael Tuexen 				ui->ui_sum = 0xffff;
1429a485f139SMichael Tuexen 		}
1430a485f139SMichael Tuexen 	} else if (V_udp_cksum) {
14316fbfd582SAndre Oppermann 		if (inp->inp_flags & INP_ONESBCAST)
14328a538743SBruce M Simpson 			faddr.s_addr = INADDR_BROADCAST;
14338a538743SBruce M Simpson 		ui->ui_sum = in_pseudo(ui->ui_src.s_addr, faddr.s_addr,
1434e06e816fSKevin Lo 		    htons((u_short)len + sizeof(struct udphdr) + pr));
1435db4f9cc7SJonathan Lemon 		m->m_pkthdr.csum_flags = CSUM_UDP;
1436db4f9cc7SJonathan Lemon 		m->m_pkthdr.csum_data = offsetof(struct udphdr, uh_sum);
1437e06e816fSKevin Lo 	}
14388f134647SGleb Smirnoff 	((struct ip *)ui)->ip_len = htons(sizeof(struct udpiphdr) + len);
1439ca98b82cSDavid Greenman 	((struct ip *)ui)->ip_ttl = inp->inp_ip_ttl;	/* XXX */
1440f584d74bSMichael Tuexen 	((struct ip *)ui)->ip_tos = tos;		/* XXX */
1441026decb8SRobert Watson 	UDPSTAT_INC(udps_opackets);
1442cfa1ca9dSYoshinobu Inoue 
14439d3ddf43SAdrian Chadd 	/*
14449d3ddf43SAdrian Chadd 	 * Setup flowid / RSS information for outbound socket.
14459d3ddf43SAdrian Chadd 	 *
14469d3ddf43SAdrian Chadd 	 * Once the UDP code decides to set a flowid some other way,
14479d3ddf43SAdrian Chadd 	 * this allows the flowid to be overridden by userland.
14489d3ddf43SAdrian Chadd 	 */
1449c2529042SHans Petter Selasky 	if (flowtype != M_HASHTYPE_NONE) {
14509d3ddf43SAdrian Chadd 		m->m_pkthdr.flowid = flowid;
1451c2529042SHans Petter Selasky 		M_HASHTYPE_SET(m, flowtype);
1452f9a6e8d7SBjoern A. Zeeb 	}
14530c325f53SAlexander V. Chernikov #if defined(ROUTE_MPATH) || defined(RSS)
14540c325f53SAlexander V. Chernikov 	else if (CALC_FLOWID_OUTBOUND_SENDTO) {
14558ad1a83bSAdrian Chadd 		uint32_t hash_val, hash_type;
14560c325f53SAlexander V. Chernikov 
14570c325f53SAlexander V. Chernikov 		hash_val = fib4_calc_packet_hash(laddr, faddr,
14580c325f53SAlexander V. Chernikov 		    lport, fport, pr, &hash_type);
14598ad1a83bSAdrian Chadd 		m->m_pkthdr.flowid = hash_val;
14608ad1a83bSAdrian Chadd 		M_HASHTYPE_SET(m, hash_type);
14618ad1a83bSAdrian Chadd 	}
14629d3ddf43SAdrian Chadd 
14638ad1a83bSAdrian Chadd 	/*
14648ad1a83bSAdrian Chadd 	 * Don't override with the inp cached flowid value.
14658ad1a83bSAdrian Chadd 	 *
14668ad1a83bSAdrian Chadd 	 * Depending upon the kind of send being done, the inp
14678ad1a83bSAdrian Chadd 	 * flowid/flowtype values may actually not be appropriate
14688ad1a83bSAdrian Chadd 	 * for this particular socket send.
14698ad1a83bSAdrian Chadd 	 *
14708ad1a83bSAdrian Chadd 	 * We should either leave the flowid at zero (which is what is
14718ad1a83bSAdrian Chadd 	 * currently done) or set it to some software generated
14728ad1a83bSAdrian Chadd 	 * hash value based on the packet contents.
14738ad1a83bSAdrian Chadd 	 */
14749d3ddf43SAdrian Chadd 	ipflags |= IP_NODEFAULTFLOWID;
14759d3ddf43SAdrian Chadd #endif	/* RSS */
14769d3ddf43SAdrian Chadd 
14777bda9663SMichael Tuexen 	if (pr == IPPROTO_UDPLITE)
14787bda9663SMichael Tuexen 		UDPLITE_PROBE(send, NULL, inp, &ui->ui_i, inp, &ui->ui_u);
14797bda9663SMichael Tuexen 	else
148057f60867SMark Johnston 		UDP_PROBE(send, NULL, inp, &ui->ui_i, inp, &ui->ui_u);
148184cc0778SGeorge V. Neville-Neil 	error = ip_output(m, inp->inp_options,
14822435e507SGleb Smirnoff 	    INP_WLOCKED(inp) ? &inp->inp_route : NULL, ipflags,
14835d846453SSam Leffler 	    inp->inp_moptions, inp);
14842435e507SGleb Smirnoff 	INP_UNLOCK(inp);
14852435e507SGleb Smirnoff 	NET_EPOCH_EXIT(et);
1486df8bae1dSRodney W. Grimes 	return (error);
1487df8bae1dSRodney W. Grimes 
1488df8bae1dSRodney W. Grimes release:
14892435e507SGleb Smirnoff 	INP_UNLOCK(inp);
14902435e507SGleb Smirnoff 	NET_EPOCH_EXIT(et);
1491df8bae1dSRodney W. Grimes 	m_freem(m);
1492df8bae1dSRodney W. Grimes 	return (error);
1493df8bae1dSRodney W. Grimes }
1494df8bae1dSRodney W. Grimes 
1495c93db4abSGleb Smirnoff pr_abort_t udp_abort;			/* shared with udp6_usrreq.c */
1496c93db4abSGleb Smirnoff void
1497d0390e05SGarrett Wollman udp_abort(struct socket *so)
1498df8bae1dSRodney W. Grimes {
1499d0390e05SGarrett Wollman 	struct inpcb *inp;
1500e06e816fSKevin Lo 	struct inpcbinfo *pcbinfo;
1501df8bae1dSRodney W. Grimes 
1502a86e5c96SBjoern A. Zeeb 	pcbinfo = udp_get_inpcbinfo(so->so_proto->pr_protocol);
1503d0390e05SGarrett Wollman 	inp = sotoinpcb(so);
150414ba8addSRobert Watson 	KASSERT(inp != NULL, ("udp_abort: inp == NULL"));
15058501a69cSRobert Watson 	INP_WLOCK(inp);
1506a152f8a3SRobert Watson 	if (inp->inp_faddr.s_addr != INADDR_ANY) {
1507e06e816fSKevin Lo 		INP_HASH_WLOCK(pcbinfo);
1508a152f8a3SRobert Watson 		in_pcbdisconnect(inp);
1509a152f8a3SRobert Watson 		inp->inp_laddr.s_addr = INADDR_ANY;
1510e06e816fSKevin Lo 		INP_HASH_WUNLOCK(pcbinfo);
1511d0390e05SGarrett Wollman 		soisdisconnected(so);
1512a152f8a3SRobert Watson 	}
15138501a69cSRobert Watson 	INP_WUNLOCK(inp);
1514df8bae1dSRodney W. Grimes }
1515df8bae1dSRodney W. Grimes 
1516d0390e05SGarrett Wollman static int
1517b40ce416SJulian Elischer udp_attach(struct socket *so, int proto, struct thread *td)
1518d0390e05SGarrett Wollman {
1519630ba2c5SMatt Macy 	static uint32_t udp_flowid;
1520d0390e05SGarrett Wollman 	struct inpcb *inp;
1521e06e816fSKevin Lo 	struct inpcbinfo *pcbinfo;
1522277afaffSRobert Watson 	int error;
1523d0390e05SGarrett Wollman 
1524a86e5c96SBjoern A. Zeeb 	pcbinfo = udp_get_inpcbinfo(so->so_proto->pr_protocol);
1525d0390e05SGarrett Wollman 	inp = sotoinpcb(so);
152614ba8addSRobert Watson 	KASSERT(inp == NULL, ("udp_attach: inp != NULL"));
1527cfa1ca9dSYoshinobu Inoue 	error = soreserve(so, udp_sendspace, udp_recvspace);
1528f24618aaSRobert Watson 	if (error)
15293329b236SRobert Watson 		return (error);
1530e06e816fSKevin Lo 	error = in_pcballoc(so, pcbinfo);
1531db0ac6deSCy Schubert 	if (error)
15323329b236SRobert Watson 		return (error);
1533cfa1ca9dSYoshinobu Inoue 
153468b5629bSRobert Watson 	inp = sotoinpcb(so);
1535603724d3SBjoern A. Zeeb 	inp->inp_ip_ttl = V_ip_defttl;
1536630ba2c5SMatt Macy 	inp->inp_flowid = atomic_fetchadd_int(&udp_flowid, 1);
1537630ba2c5SMatt Macy 	inp->inp_flowtype = M_HASHTYPE_OPAQUE;
15386a9148feSBjoern A. Zeeb 
15396a9148feSBjoern A. Zeeb 	error = udp_newudpcb(inp);
15406a9148feSBjoern A. Zeeb 	if (error) {
15416a9148feSBjoern A. Zeeb 		in_pcbdetach(inp);
15426a9148feSBjoern A. Zeeb 		in_pcbfree(inp);
15436a9148feSBjoern A. Zeeb 		return (error);
15446a9148feSBjoern A. Zeeb 	}
1545266f97b5SCy Schubert 	INP_WUNLOCK(inp);
1546db0ac6deSCy Schubert 
1547c7c7ea4bSRandall Stewart 	return (0);
1548c7c7ea4bSRandall Stewart }
154979288c11SBjoern A. Zeeb #endif /* INET */
1550c7c7ea4bSRandall Stewart 
1551c7c7ea4bSRandall Stewart int
1552abb901c5SRandall Stewart udp_set_kernel_tunneling(struct socket *so, udp_tun_func_t f, udp_tun_icmp_t i, void *ctx)
1553c7c7ea4bSRandall Stewart {
1554c7c7ea4bSRandall Stewart 	struct inpcb *inp;
15556a9148feSBjoern A. Zeeb 	struct udpcb *up;
1556c7c7ea4bSRandall Stewart 
155768b5629bSRobert Watson 	KASSERT(so->so_type == SOCK_DGRAM,
155868b5629bSRobert Watson 	    ("udp_set_kernel_tunneling: !dgram"));
155968b5629bSRobert Watson 	inp = sotoinpcb(so);
156068b5629bSRobert Watson 	KASSERT(inp != NULL, ("udp_set_kernel_tunneling: inp == NULL"));
1561c7c7ea4bSRandall Stewart 	INP_WLOCK(inp);
15626a9148feSBjoern A. Zeeb 	up = intoudpcb(inp);
1563995cba5aSKristof Provost 	if ((f != NULL || i != NULL) && ((up->u_tun_func != NULL) ||
1564995cba5aSKristof Provost 	    (up->u_icmp_func != NULL))) {
1565bbb0e3d9SRandall Stewart 		INP_WUNLOCK(inp);
1566bbb0e3d9SRandall Stewart 		return (EBUSY);
1567bbb0e3d9SRandall Stewart 	}
15686a9148feSBjoern A. Zeeb 	up->u_tun_func = f;
1569abb901c5SRandall Stewart 	up->u_icmp_func = i;
157081d3ec17SBryan Venteicher 	up->u_tun_ctx = ctx;
15718501a69cSRobert Watson 	INP_WUNLOCK(inp);
15723329b236SRobert Watson 	return (0);
1573df8bae1dSRodney W. Grimes }
1574d0390e05SGarrett Wollman 
157579288c11SBjoern A. Zeeb #ifdef INET
1576d0390e05SGarrett Wollman static int
1577b40ce416SJulian Elischer udp_bind(struct socket *so, struct sockaddr *nam, struct thread *td)
1578d0390e05SGarrett Wollman {
1579d0390e05SGarrett Wollman 	struct inpcb *inp;
1580e06e816fSKevin Lo 	struct inpcbinfo *pcbinfo;
1581f96603b5SMark Johnston 	struct sockaddr_in *sinp;
1582277afaffSRobert Watson 	int error;
1583d0390e05SGarrett Wollman 
1584a86e5c96SBjoern A. Zeeb 	pcbinfo = udp_get_inpcbinfo(so->so_proto->pr_protocol);
1585d0390e05SGarrett Wollman 	inp = sotoinpcb(so);
158614ba8addSRobert Watson 	KASSERT(inp != NULL, ("udp_bind: inp == NULL"));
1587f161d294SMark Johnston 
1588f96603b5SMark Johnston 	sinp = (struct sockaddr_in *)nam;
1589f96603b5SMark Johnston 	if (nam->sa_family != AF_INET) {
1590f96603b5SMark Johnston 		/*
1591f96603b5SMark Johnston 		 * Preserve compatibility with old programs.
1592f96603b5SMark Johnston 		 */
1593f96603b5SMark Johnston 		if (nam->sa_family != AF_UNSPEC ||
15943f1f6b6eSMichael Tuexen 		    nam->sa_len < offsetof(struct sockaddr_in, sin_zero) ||
1595f96603b5SMark Johnston 		    sinp->sin_addr.s_addr != INADDR_ANY)
1596f161d294SMark Johnston 			return (EAFNOSUPPORT);
1597f96603b5SMark Johnston 		nam->sa_family = AF_INET;
1598f96603b5SMark Johnston 	}
1599f161d294SMark Johnston 	if (nam->sa_len != sizeof(struct sockaddr_in))
1600f161d294SMark Johnston 		return (EINVAL);
1601f161d294SMark Johnston 
16028501a69cSRobert Watson 	INP_WLOCK(inp);
1603e06e816fSKevin Lo 	INP_HASH_WLOCK(pcbinfo);
1604b0330ed9SPawel Jakub Dawidek 	error = in_pcbbind(inp, nam, td->td_ucred);
1605e06e816fSKevin Lo 	INP_HASH_WUNLOCK(pcbinfo);
16068501a69cSRobert Watson 	INP_WUNLOCK(inp);
16073329b236SRobert Watson 	return (error);
1608d0390e05SGarrett Wollman }
1609d0390e05SGarrett Wollman 
1610a152f8a3SRobert Watson static void
1611a152f8a3SRobert Watson udp_close(struct socket *so)
1612a152f8a3SRobert Watson {
1613a152f8a3SRobert Watson 	struct inpcb *inp;
1614e06e816fSKevin Lo 	struct inpcbinfo *pcbinfo;
1615a152f8a3SRobert Watson 
1616a86e5c96SBjoern A. Zeeb 	pcbinfo = udp_get_inpcbinfo(so->so_proto->pr_protocol);
1617a152f8a3SRobert Watson 	inp = sotoinpcb(so);
1618a152f8a3SRobert Watson 	KASSERT(inp != NULL, ("udp_close: inp == NULL"));
16198501a69cSRobert Watson 	INP_WLOCK(inp);
1620a152f8a3SRobert Watson 	if (inp->inp_faddr.s_addr != INADDR_ANY) {
1621e06e816fSKevin Lo 		INP_HASH_WLOCK(pcbinfo);
1622a152f8a3SRobert Watson 		in_pcbdisconnect(inp);
1623a152f8a3SRobert Watson 		inp->inp_laddr.s_addr = INADDR_ANY;
1624e06e816fSKevin Lo 		INP_HASH_WUNLOCK(pcbinfo);
1625a152f8a3SRobert Watson 		soisdisconnected(so);
1626a152f8a3SRobert Watson 	}
16278501a69cSRobert Watson 	INP_WUNLOCK(inp);
1628a152f8a3SRobert Watson }
1629a152f8a3SRobert Watson 
1630d0390e05SGarrett Wollman static int
1631b40ce416SJulian Elischer udp_connect(struct socket *so, struct sockaddr *nam, struct thread *td)
1632d0390e05SGarrett Wollman {
1633c1604fe4SGleb Smirnoff 	struct epoch_tracker et;
1634d0390e05SGarrett Wollman 	struct inpcb *inp;
1635e06e816fSKevin Lo 	struct inpcbinfo *pcbinfo;
163675c13541SPoul-Henning Kamp 	struct sockaddr_in *sin;
1637e06e816fSKevin Lo 	int error;
1638d0390e05SGarrett Wollman 
1639a86e5c96SBjoern A. Zeeb 	pcbinfo = udp_get_inpcbinfo(so->so_proto->pr_protocol);
1640d0390e05SGarrett Wollman 	inp = sotoinpcb(so);
164114ba8addSRobert Watson 	KASSERT(inp != NULL, ("udp_connect: inp == NULL"));
1642f161d294SMark Johnston 
1643f161d294SMark Johnston 	sin = (struct sockaddr_in *)nam;
1644f161d294SMark Johnston 	if (sin->sin_family != AF_INET)
1645f161d294SMark Johnston 		return (EAFNOSUPPORT);
1646f161d294SMark Johnston 	if (sin->sin_len != sizeof(*sin))
1647f161d294SMark Johnston 		return (EINVAL);
1648f161d294SMark Johnston 
16498501a69cSRobert Watson 	INP_WLOCK(inp);
1650f76fcf6dSJeffrey Hsu 	if (inp->inp_faddr.s_addr != INADDR_ANY) {
16518501a69cSRobert Watson 		INP_WUNLOCK(inp);
16523329b236SRobert Watson 		return (EISCONN);
1653f76fcf6dSJeffrey Hsu 	}
1654b89e82ddSJamie Gritton 	error = prison_remote_ip4(td->td_ucred, &sin->sin_addr);
1655b89e82ddSJamie Gritton 	if (error != 0) {
1656413628a7SBjoern A. Zeeb 		INP_WUNLOCK(inp);
1657b89e82ddSJamie Gritton 		return (error);
1658413628a7SBjoern A. Zeeb 	}
1659c1604fe4SGleb Smirnoff 	NET_EPOCH_ENTER(et);
1660e06e816fSKevin Lo 	INP_HASH_WLOCK(pcbinfo);
1661db0ac6deSCy Schubert 	error = in_pcbconnect(inp, nam, td->td_ucred, true);
1662e06e816fSKevin Lo 	INP_HASH_WUNLOCK(pcbinfo);
1663c1604fe4SGleb Smirnoff 	NET_EPOCH_EXIT(et);
16644cc20ab1SSeigo Tanimura 	if (error == 0)
1665df8bae1dSRodney W. Grimes 		soisconnected(so);
16668501a69cSRobert Watson 	INP_WUNLOCK(inp);
16673329b236SRobert Watson 	return (error);
1668df8bae1dSRodney W. Grimes }
1669d0390e05SGarrett Wollman 
1670bc725eafSRobert Watson static void
1671d0390e05SGarrett Wollman udp_detach(struct socket *so)
1672d0390e05SGarrett Wollman {
1673d0390e05SGarrett Wollman 	struct inpcb *inp;
16746a9148feSBjoern A. Zeeb 	struct udpcb *up;
1675d0390e05SGarrett Wollman 
1676d0390e05SGarrett Wollman 	inp = sotoinpcb(so);
167714ba8addSRobert Watson 	KASSERT(inp != NULL, ("udp_detach: inp == NULL"));
1678a152f8a3SRobert Watson 	KASSERT(inp->inp_faddr.s_addr == INADDR_ANY,
1679a152f8a3SRobert Watson 	    ("udp_detach: not disconnected"));
16808501a69cSRobert Watson 	INP_WLOCK(inp);
16816a9148feSBjoern A. Zeeb 	up = intoudpcb(inp);
16826a9148feSBjoern A. Zeeb 	KASSERT(up != NULL, ("%s: up == NULL", __func__));
16836a9148feSBjoern A. Zeeb 	inp->inp_ppcb = NULL;
1684d0390e05SGarrett Wollman 	in_pcbdetach(inp);
168514ba8addSRobert Watson 	in_pcbfree(inp);
16866a9148feSBjoern A. Zeeb 	udp_discardcb(up);
1687d0390e05SGarrett Wollman }
1688d0390e05SGarrett Wollman 
1689c93db4abSGleb Smirnoff pr_disconnect_t udp_disconnect;		/* shared with udp6_usrreq.c */
1690c93db4abSGleb Smirnoff int
1691d0390e05SGarrett Wollman udp_disconnect(struct socket *so)
1692d0390e05SGarrett Wollman {
1693d0390e05SGarrett Wollman 	struct inpcb *inp;
1694e06e816fSKevin Lo 	struct inpcbinfo *pcbinfo;
1695d0390e05SGarrett Wollman 
1696a86e5c96SBjoern A. Zeeb 	pcbinfo = udp_get_inpcbinfo(so->so_proto->pr_protocol);
1697d0390e05SGarrett Wollman 	inp = sotoinpcb(so);
169814ba8addSRobert Watson 	KASSERT(inp != NULL, ("udp_disconnect: inp == NULL"));
16998501a69cSRobert Watson 	INP_WLOCK(inp);
1700f76fcf6dSJeffrey Hsu 	if (inp->inp_faddr.s_addr == INADDR_ANY) {
17018501a69cSRobert Watson 		INP_WUNLOCK(inp);
17023329b236SRobert Watson 		return (ENOTCONN);
1703f76fcf6dSJeffrey Hsu 	}
1704e06e816fSKevin Lo 	INP_HASH_WLOCK(pcbinfo);
1705df8bae1dSRodney W. Grimes 	in_pcbdisconnect(inp);
1706df8bae1dSRodney W. Grimes 	inp->inp_laddr.s_addr = INADDR_ANY;
1707e06e816fSKevin Lo 	INP_HASH_WUNLOCK(pcbinfo);
1708d45e4f99SMaxim Konovalov 	SOCK_LOCK(so);
1709d45e4f99SMaxim Konovalov 	so->so_state &= ~SS_ISCONNECTED;		/* XXX */
1710d45e4f99SMaxim Konovalov 	SOCK_UNLOCK(so);
17118501a69cSRobert Watson 	INP_WUNLOCK(inp);
17123329b236SRobert Watson 	return (0);
1713df8bae1dSRodney W. Grimes }
1714df8bae1dSRodney W. Grimes 
1715c93db4abSGleb Smirnoff pr_send_t udp_send;			/* shared with udp6_usrreq.c */
1716c93db4abSGleb Smirnoff int
171757bf258eSGarrett Wollman udp_send(struct socket *so, int flags, struct mbuf *m, struct sockaddr *addr,
1718b40ce416SJulian Elischer     struct mbuf *control, struct thread *td)
1719d0390e05SGarrett Wollman {
1720d0390e05SGarrett Wollman 	struct inpcb *inp;
1721f161d294SMark Johnston 	int error;
1722d0390e05SGarrett Wollman 
1723d0390e05SGarrett Wollman 	inp = sotoinpcb(so);
172414ba8addSRobert Watson 	KASSERT(inp != NULL, ("udp_send: inp == NULL"));
1725f161d294SMark Johnston 
1726f161d294SMark Johnston 	if (addr != NULL) {
1727f161d294SMark Johnston 		error = 0;
1728f161d294SMark Johnston 		if (addr->sa_family != AF_INET)
1729f161d294SMark Johnston 			error = EAFNOSUPPORT;
1730f161d294SMark Johnston 		else if (addr->sa_len != sizeof(struct sockaddr_in))
1731f161d294SMark Johnston 			error = EINVAL;
1732f161d294SMark Johnston 		if (__predict_false(error != 0)) {
1733f161d294SMark Johnston 			m_freem(control);
1734f161d294SMark Johnston 			m_freem(m);
1735f161d294SMark Johnston 			return (error);
1736f161d294SMark Johnston 		}
1737f161d294SMark Johnston 	}
1738a9839c4aSBjoern A. Zeeb 	return (udp_output(inp, m, addr, control, td, flags));
1739d0390e05SGarrett Wollman }
174079288c11SBjoern A. Zeeb #endif /* INET */
1741d0390e05SGarrett Wollman 
174276429de4SYoshinobu Inoue int
1743d0390e05SGarrett Wollman udp_shutdown(struct socket *so)
1744d0390e05SGarrett Wollman {
1745d0390e05SGarrett Wollman 	struct inpcb *inp;
1746d0390e05SGarrett Wollman 
1747d0390e05SGarrett Wollman 	inp = sotoinpcb(so);
174814ba8addSRobert Watson 	KASSERT(inp != NULL, ("udp_shutdown: inp == NULL"));
17498501a69cSRobert Watson 	INP_WLOCK(inp);
1750d0390e05SGarrett Wollman 	socantsendmore(so);
17518501a69cSRobert Watson 	INP_WUNLOCK(inp);
17523329b236SRobert Watson 	return (0);
1753d0390e05SGarrett Wollman }
1754d0390e05SGarrett Wollman 
175579288c11SBjoern A. Zeeb #ifdef INET
1756*e7d02be1SGleb Smirnoff #define	UDP_PROTOSW							\
1757*e7d02be1SGleb Smirnoff 	.pr_type =		SOCK_DGRAM,				\
1758*e7d02be1SGleb Smirnoff 	.pr_flags =		PR_ATOMIC | PR_ADDR | PR_CAPATTACH,	\
1759*e7d02be1SGleb Smirnoff 	.pr_ctloutput =		udp_ctloutput,				\
1760*e7d02be1SGleb Smirnoff 	.pr_abort =		udp_abort,				\
1761*e7d02be1SGleb Smirnoff 	.pr_attach =		udp_attach,				\
1762*e7d02be1SGleb Smirnoff 	.pr_bind =		udp_bind,				\
1763*e7d02be1SGleb Smirnoff 	.pr_connect =		udp_connect,				\
1764*e7d02be1SGleb Smirnoff 	.pr_control =		in_control,				\
1765*e7d02be1SGleb Smirnoff 	.pr_detach =		udp_detach,				\
1766*e7d02be1SGleb Smirnoff 	.pr_disconnect =	udp_disconnect,				\
1767*e7d02be1SGleb Smirnoff 	.pr_peeraddr =		in_getpeeraddr,				\
1768*e7d02be1SGleb Smirnoff 	.pr_send =		udp_send,				\
1769*e7d02be1SGleb Smirnoff 	.pr_soreceive =		soreceive_dgram,			\
1770*e7d02be1SGleb Smirnoff 	.pr_sosend =		sosend_dgram,				\
1771*e7d02be1SGleb Smirnoff 	.pr_shutdown =		udp_shutdown,				\
1772*e7d02be1SGleb Smirnoff 	.pr_sockaddr =		in_getsockaddr,				\
1773*e7d02be1SGleb Smirnoff 	.pr_sosetlabel =	in_pcbsosetlabel,			\
1774*e7d02be1SGleb Smirnoff 	.pr_close =		udp_close
1775*e7d02be1SGleb Smirnoff 
1776*e7d02be1SGleb Smirnoff struct protosw udp_protosw = {
1777*e7d02be1SGleb Smirnoff 	.pr_protocol =		IPPROTO_UDP,
1778*e7d02be1SGleb Smirnoff 	UDP_PROTOSW
1779*e7d02be1SGleb Smirnoff };
1780*e7d02be1SGleb Smirnoff 
1781*e7d02be1SGleb Smirnoff struct protosw udplite_protosw = {
1782*e7d02be1SGleb Smirnoff 	.pr_protocol =		IPPROTO_UDPLITE,
1783*e7d02be1SGleb Smirnoff 	UDP_PROTOSW
1784d0390e05SGarrett Wollman };
178578b1fc05SGleb Smirnoff 
178678b1fc05SGleb Smirnoff static void
178778b1fc05SGleb Smirnoff udp_init(void *arg __unused)
178878b1fc05SGleb Smirnoff {
178978b1fc05SGleb Smirnoff 
179078b1fc05SGleb Smirnoff 	IPPROTO_REGISTER(IPPROTO_UDP, udp_input, udp_ctlinput);
179178b1fc05SGleb Smirnoff 	IPPROTO_REGISTER(IPPROTO_UDPLITE, udp_input, udplite_ctlinput);
179278b1fc05SGleb Smirnoff }
179378b1fc05SGleb Smirnoff SYSINIT(udp_init, SI_SUB_PROTO_DOMAIN, SI_ORDER_THIRD, udp_init, NULL);
179479288c11SBjoern A. Zeeb #endif /* INET */
1795