xref: /freebsd/sys/netinet/udp_usrreq.c (revision db0ac6ded61105caab4700aeac255328d4238dc4)
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 
1734f590175SPaul Saab static void
1744f590175SPaul Saab udp_zone_change(void *tag)
1754f590175SPaul Saab {
1764f590175SPaul Saab 
177603724d3SBjoern A. Zeeb 	uma_zone_set_max(V_udbinfo.ipi_zone, maxsockets);
1786a9148feSBjoern A. Zeeb 	uma_zone_set_max(V_udpcb_zone, maxsockets);
1794f590175SPaul Saab }
1804f590175SPaul Saab 
181d915b280SStephan Uphoff static int
182d915b280SStephan Uphoff udp_inpcb_init(void *mem, int size, int flags)
183d915b280SStephan Uphoff {
184af1ee11dSRobert Watson 	struct inpcb *inp;
18508651e1fSJohn Baldwin 
186af1ee11dSRobert Watson 	inp = mem;
187d915b280SStephan Uphoff 	INP_LOCK_INIT(inp, "inp", "udpinp");
188d915b280SStephan Uphoff 	return (0);
189d915b280SStephan Uphoff }
190d915b280SStephan Uphoff 
191e06e816fSKevin Lo static int
192e06e816fSKevin Lo udplite_inpcb_init(void *mem, int size, int flags)
193e06e816fSKevin Lo {
194e06e816fSKevin Lo 	struct inpcb *inp;
195e06e816fSKevin Lo 
196e06e816fSKevin Lo 	inp = mem;
197e06e816fSKevin Lo 	INP_LOCK_INIT(inp, "inp", "udpliteinp");
198e06e816fSKevin Lo 	return (0);
199e06e816fSKevin Lo }
200e06e816fSKevin Lo 
201df8bae1dSRodney W. Grimes void
202af1ee11dSRobert Watson udp_init(void)
203df8bae1dSRodney W. Grimes {
204af1ee11dSRobert Watson 
2058ad1a83bSAdrian Chadd 	/*
2068ad1a83bSAdrian Chadd 	 * For now default to 2-tuple UDP hashing - until the fragment
2078ad1a83bSAdrian Chadd 	 * reassembly code can also update the flowid.
2088ad1a83bSAdrian Chadd 	 *
2098ad1a83bSAdrian Chadd 	 * Once we can calculate the flowid that way and re-establish
2108ad1a83bSAdrian Chadd 	 * a 4-tuple, flip this to 4-tuple.
2118ad1a83bSAdrian Chadd 	 */
212*db0ac6deSCy Schubert 	in_pcbinfo_init(&V_udbinfo, "udp", UDBHASHSIZE, UDBHASHSIZE,
213*db0ac6deSCy Schubert 	    "udp_inpcb", udp_inpcb_init);
2146a9148feSBjoern A. Zeeb 	V_udpcb_zone = uma_zcreate("udpcb", sizeof(struct udpcb),
215a8da5dd6SCraig Rodrigues 	    NULL, NULL, NULL, NULL, UMA_ALIGN_PTR, 0);
2166a9148feSBjoern A. Zeeb 	uma_zone_set_max(V_udpcb_zone, maxsockets);
2176acd596eSPawel Jakub Dawidek 	uma_zone_set_warning(V_udpcb_zone, "kern.ipc.maxsockets limit reached");
2184f590175SPaul Saab 	EVENTHANDLER_REGISTER(maxsockets_change, udp_zone_change, NULL,
2194f590175SPaul Saab 	    EVENTHANDLER_PRI_ANY);
220df8bae1dSRodney W. Grimes }
221df8bae1dSRodney W. Grimes 
222e06e816fSKevin Lo void
223e06e816fSKevin Lo udplite_init(void)
224e06e816fSKevin Lo {
225e06e816fSKevin Lo 
226*db0ac6deSCy Schubert 	in_pcbinfo_init(&V_ulitecbinfo, "udplite", UDBHASHSIZE,
227*db0ac6deSCy Schubert 	    UDBHASHSIZE, "udplite_inpcb", udplite_inpcb_init);
228e06e816fSKevin Lo }
229e06e816fSKevin Lo 
230315e3e38SRobert Watson /*
231315e3e38SRobert Watson  * Kernel module interface for updating udpstat.  The argument is an index
232315e3e38SRobert Watson  * into udpstat treated as an array of u_long.  While this encodes the
233315e3e38SRobert Watson  * general layout of udpstat into the caller, it doesn't encode its location,
234315e3e38SRobert Watson  * so that future changes to add, for example, per-CPU stats support won't
235315e3e38SRobert Watson  * cause binary compatibility problems for kernel modules.
236315e3e38SRobert Watson  */
237315e3e38SRobert Watson void
238315e3e38SRobert Watson kmod_udpstat_inc(int statnum)
239315e3e38SRobert Watson {
240315e3e38SRobert Watson 
2415b7cb97cSAndrey V. Elsukov 	counter_u64_add(VNET(udpstat)[statnum], 1);
242315e3e38SRobert Watson }
243315e3e38SRobert Watson 
2446a9148feSBjoern A. Zeeb int
2456a9148feSBjoern A. Zeeb udp_newudpcb(struct inpcb *inp)
2466a9148feSBjoern A. Zeeb {
2476a9148feSBjoern A. Zeeb 	struct udpcb *up;
2486a9148feSBjoern A. Zeeb 
2496a9148feSBjoern A. Zeeb 	up = uma_zalloc(V_udpcb_zone, M_NOWAIT | M_ZERO);
2506a9148feSBjoern A. Zeeb 	if (up == NULL)
2516a9148feSBjoern A. Zeeb 		return (ENOBUFS);
2526a9148feSBjoern A. Zeeb 	inp->inp_ppcb = up;
2536a9148feSBjoern A. Zeeb 	return (0);
2546a9148feSBjoern A. Zeeb }
2556a9148feSBjoern A. Zeeb 
2566a9148feSBjoern A. Zeeb void
2576a9148feSBjoern A. Zeeb udp_discardcb(struct udpcb *up)
2586a9148feSBjoern A. Zeeb {
2596a9148feSBjoern A. Zeeb 
2606a9148feSBjoern A. Zeeb 	uma_zfree(V_udpcb_zone, up);
2616a9148feSBjoern A. Zeeb }
2626a9148feSBjoern A. Zeeb 
263bc29160dSMarko Zec #ifdef VIMAGE
2643f58662dSBjoern A. Zeeb static void
2653f58662dSBjoern A. Zeeb udp_destroy(void *unused __unused)
266bc29160dSMarko Zec {
267bc29160dSMarko Zec 
2689bcd427bSRobert Watson 	in_pcbinfo_destroy(&V_udbinfo);
269391dab1cSBjoern A. Zeeb 	uma_zdestroy(V_udpcb_zone);
270bc29160dSMarko Zec }
2713f58662dSBjoern A. Zeeb VNET_SYSUNINIT(udp, SI_SUB_PROTO_DOMAIN, SI_ORDER_FOURTH, udp_destroy, NULL);
272e06e816fSKevin Lo 
2733f58662dSBjoern A. Zeeb static void
2743f58662dSBjoern A. Zeeb udplite_destroy(void *unused __unused)
275e06e816fSKevin Lo {
276e06e816fSKevin Lo 
277e06e816fSKevin Lo 	in_pcbinfo_destroy(&V_ulitecbinfo);
278e06e816fSKevin Lo }
2793f58662dSBjoern A. Zeeb VNET_SYSUNINIT(udplite, SI_SUB_PROTO_DOMAIN, SI_ORDER_FOURTH, udplite_destroy,
2803f58662dSBjoern A. Zeeb     NULL);
281bc29160dSMarko Zec #endif
282bc29160dSMarko Zec 
28379288c11SBjoern A. Zeeb #ifdef INET
28443bbb6aaSRobert Watson /*
28543bbb6aaSRobert Watson  * Subroutine of udp_input(), which appends the provided mbuf chain to the
28643bbb6aaSRobert Watson  * passed pcb/socket.  The caller must provide a sockaddr_in via udp_in that
28743bbb6aaSRobert Watson  * contains the source address.  If the socket ends up being an IPv6 socket,
28843bbb6aaSRobert Watson  * udp_append() will convert to a sockaddr_in6 before passing the address
28943bbb6aaSRobert Watson  * into the socket code.
290c0d1be08SRandall Stewart  *
291c0d1be08SRandall Stewart  * In the normal case udp_append() will return 0, indicating that you
292c0d1be08SRandall Stewart  * must unlock the inp. However if a tunneling protocol is in place we increment
293c0d1be08SRandall Stewart  * the inpcb refcnt and unlock the inp, on return from the tunneling protocol we
294c0d1be08SRandall Stewart  * then decrement the reference count. If the inp_rele returns 1, indicating the
295c0d1be08SRandall Stewart  * inp is gone, we return that to the caller to tell them *not* to unlock
296c0d1be08SRandall Stewart  * the inp. In the case of multi-cast this will cause the distribution
297c0d1be08SRandall Stewart  * to stop (though most tunneling protocols known currently do *not* use
298c0d1be08SRandall Stewart  * multicast).
29943bbb6aaSRobert Watson  */
300c0d1be08SRandall Stewart static int
30143bbb6aaSRobert Watson udp_append(struct inpcb *inp, struct ip *ip, struct mbuf *n, int off,
30243bbb6aaSRobert Watson     struct sockaddr_in *udp_in)
30343bbb6aaSRobert Watson {
30443bbb6aaSRobert Watson 	struct sockaddr *append_sa;
30543bbb6aaSRobert Watson 	struct socket *so;
306dce33a45SErmal Luçi 	struct mbuf *tmpopts, *opts = NULL;
30743bbb6aaSRobert Watson #ifdef INET6
30843bbb6aaSRobert Watson 	struct sockaddr_in6 udp_in6;
30943bbb6aaSRobert Watson #endif
3107b495c44SVANHULLEBUS Yvan 	struct udpcb *up;
31143bbb6aaSRobert Watson 
312fa046d87SRobert Watson 	INP_LOCK_ASSERT(inp);
31343bbb6aaSRobert Watson 
31479bb84fbSEdward Tomasz Napierala 	/*
31579bb84fbSEdward Tomasz Napierala 	 * Engage the tunneling protocol.
31679bb84fbSEdward Tomasz Napierala 	 */
31779bb84fbSEdward Tomasz Napierala 	up = intoudpcb(inp);
31879bb84fbSEdward Tomasz Napierala 	if (up->u_tun_func != NULL) {
319c0d1be08SRandall Stewart 		in_pcbref(inp);
320c0d1be08SRandall Stewart 		INP_RUNLOCK(inp);
321dce33a45SErmal Luçi 		(*up->u_tun_func)(n, off, inp, (struct sockaddr *)&udp_in[0],
32281d3ec17SBryan Venteicher 		    up->u_tun_ctx);
323c0d1be08SRandall Stewart 		INP_RLOCK(inp);
324c0d1be08SRandall Stewart 		return (in_pcbrele_rlocked(inp));
32579bb84fbSEdward Tomasz Napierala 	}
32679bb84fbSEdward Tomasz Napierala 
32779bb84fbSEdward Tomasz Napierala 	off += sizeof(struct udphdr);
32879bb84fbSEdward Tomasz Napierala 
329fcf59617SAndrey V. Elsukov #if defined(IPSEC) || defined(IPSEC_SUPPORT)
33043bbb6aaSRobert Watson 	/* Check AH/ESP integrity. */
331fcf59617SAndrey V. Elsukov 	if (IPSEC_ENABLED(ipv4) &&
332fcf59617SAndrey V. Elsukov 	    IPSEC_CHECK_POLICY(ipv4, n, inp) != 0) {
33343bbb6aaSRobert Watson 		m_freem(n);
334c0d1be08SRandall Stewart 		return (0);
33543bbb6aaSRobert Watson 	}
336fcf59617SAndrey V. Elsukov 	if (up->u_flags & UF_ESPINUDP) {/* IPSec UDP encaps. */
337fcf59617SAndrey V. Elsukov 		if (IPSEC_ENABLED(ipv4) &&
338fcf59617SAndrey V. Elsukov 		    UDPENCAP_INPUT(n, off, AF_INET) != 0)
339fcf59617SAndrey V. Elsukov 			return (0);	/* Consumed. */
3407b495c44SVANHULLEBUS Yvan 	}
34143bbb6aaSRobert Watson #endif /* IPSEC */
34243bbb6aaSRobert Watson #ifdef MAC
34330d239bcSRobert Watson 	if (mac_inpcb_check_deliver(inp, n) != 0) {
34443bbb6aaSRobert Watson 		m_freem(n);
345c0d1be08SRandall Stewart 		return (0);
34643bbb6aaSRobert Watson 	}
34779288c11SBjoern A. Zeeb #endif /* MAC */
34843bbb6aaSRobert Watson 	if (inp->inp_flags & INP_CONTROLOPTS ||
34943bbb6aaSRobert Watson 	    inp->inp_socket->so_options & (SO_TIMESTAMP | SO_BINTIME)) {
35043bbb6aaSRobert Watson #ifdef INET6
3519a38ba81SBjoern A. Zeeb 		if (inp->inp_vflag & INP_IPV6)
35248d48eb9SBjoern A. Zeeb 			(void)ip6_savecontrol_v4(inp, n, &opts, NULL);
3539a38ba81SBjoern A. Zeeb 		else
35479288c11SBjoern A. Zeeb #endif /* INET6 */
35543bbb6aaSRobert Watson 			ip_savecontrol(inp, &opts, ip, n);
35643bbb6aaSRobert Watson 	}
357dce33a45SErmal Luçi 	if ((inp->inp_vflag & INP_IPV4) && (inp->inp_flags2 & INP_ORIGDSTADDR)) {
358dce33a45SErmal Luçi 		tmpopts = sbcreatecontrol((caddr_t)&udp_in[1],
359dce33a45SErmal Luçi 			sizeof(struct sockaddr_in), IP_ORIGDSTADDR, IPPROTO_IP);
360dce33a45SErmal Luçi 		if (tmpopts) {
361dce33a45SErmal Luçi 			if (opts) {
362dce33a45SErmal Luçi 				tmpopts->m_next = opts;
363dce33a45SErmal Luçi 				opts = tmpopts;
364dce33a45SErmal Luçi 			} else
365dce33a45SErmal Luçi 				opts = tmpopts;
366dce33a45SErmal Luçi 		}
367dce33a45SErmal Luçi 	}
36843bbb6aaSRobert Watson #ifdef INET6
36943bbb6aaSRobert Watson 	if (inp->inp_vflag & INP_IPV6) {
37043bbb6aaSRobert Watson 		bzero(&udp_in6, sizeof(udp_in6));
37143bbb6aaSRobert Watson 		udp_in6.sin6_len = sizeof(udp_in6);
37243bbb6aaSRobert Watson 		udp_in6.sin6_family = AF_INET6;
373dce33a45SErmal Luçi 		in6_sin_2_v4mapsin6(&udp_in[0], &udp_in6);
37443bbb6aaSRobert Watson 		append_sa = (struct sockaddr *)&udp_in6;
37543bbb6aaSRobert Watson 	} else
37679288c11SBjoern A. Zeeb #endif /* INET6 */
377dce33a45SErmal Luçi 		append_sa = (struct sockaddr *)&udp_in[0];
37843bbb6aaSRobert Watson 	m_adj(n, off);
37943bbb6aaSRobert Watson 
38043bbb6aaSRobert Watson 	so = inp->inp_socket;
38143bbb6aaSRobert Watson 	SOCKBUF_LOCK(&so->so_rcv);
38243bbb6aaSRobert Watson 	if (sbappendaddr_locked(&so->so_rcv, append_sa, n, opts) == 0) {
383a61c24ddSKonstantin Kukushkin 		soroverflow_locked(so);
38443bbb6aaSRobert Watson 		m_freem(n);
38543bbb6aaSRobert Watson 		if (opts)
38643bbb6aaSRobert Watson 			m_freem(opts);
387026decb8SRobert Watson 		UDPSTAT_INC(udps_fullsock);
38843bbb6aaSRobert Watson 	} else
38943bbb6aaSRobert Watson 		sorwakeup_locked(so);
390c0d1be08SRandall Stewart 	return (0);
39143bbb6aaSRobert Watson }
39243bbb6aaSRobert Watson 
393*db0ac6deSCy Schubert static bool
394*db0ac6deSCy Schubert udp_multi_match(const struct inpcb *inp, void *v)
395*db0ac6deSCy Schubert {
396*db0ac6deSCy Schubert 	struct ip *ip = v;
397*db0ac6deSCy Schubert 	struct udphdr *uh = (struct udphdr *)(ip + 1);
398*db0ac6deSCy Schubert 
399*db0ac6deSCy Schubert 	if (inp->inp_lport != uh->uh_dport)
400*db0ac6deSCy Schubert 		return (false);
401*db0ac6deSCy Schubert #ifdef INET6
402*db0ac6deSCy Schubert 	if ((inp->inp_vflag & INP_IPV4) == 0)
403*db0ac6deSCy Schubert 		return (false);
404*db0ac6deSCy Schubert #endif
405*db0ac6deSCy Schubert 	if (inp->inp_laddr.s_addr != INADDR_ANY &&
406*db0ac6deSCy Schubert 	    inp->inp_laddr.s_addr != ip->ip_dst.s_addr)
407*db0ac6deSCy Schubert 		return (false);
408*db0ac6deSCy Schubert 	if (inp->inp_faddr.s_addr != INADDR_ANY &&
409*db0ac6deSCy Schubert 	    inp->inp_faddr.s_addr != ip->ip_src.s_addr)
410*db0ac6deSCy Schubert 		return (false);
411*db0ac6deSCy Schubert 	if (inp->inp_fport != 0 &&
412*db0ac6deSCy Schubert 	    inp->inp_fport != uh->uh_sport)
413*db0ac6deSCy Schubert 		return (false);
414*db0ac6deSCy Schubert 
415*db0ac6deSCy Schubert 	return (true);
416*db0ac6deSCy Schubert }
417*db0ac6deSCy Schubert 
418*db0ac6deSCy Schubert static int
419*db0ac6deSCy Schubert udp_multi_input(struct mbuf *m, int proto, struct sockaddr_in *udp_in)
420*db0ac6deSCy Schubert {
421*db0ac6deSCy Schubert 	struct ip *ip = mtod(m, struct ip *);
422*db0ac6deSCy Schubert 	struct inpcb_iterator inpi = INP_ITERATOR(udp_get_inpcbinfo(proto),
423*db0ac6deSCy Schubert 	    INPLOOKUP_RLOCKPCB, udp_multi_match, ip);
424*db0ac6deSCy Schubert 	struct udphdr *uh = (struct udphdr *)(ip + 1);
425*db0ac6deSCy Schubert 	struct inpcb *inp;
426*db0ac6deSCy Schubert 	struct mbuf *n;
427*db0ac6deSCy Schubert 	int appends = 0;
428*db0ac6deSCy Schubert 
429*db0ac6deSCy Schubert 	MPASS(ip->ip_hl == sizeof(struct ip) >> 2);
430*db0ac6deSCy Schubert 
431*db0ac6deSCy Schubert 	while ((inp = inp_next(&inpi)) != NULL) {
432*db0ac6deSCy Schubert 		/*
433*db0ac6deSCy Schubert 		 * XXXRW: Because we weren't holding either the inpcb
434*db0ac6deSCy Schubert 		 * or the hash lock when we checked for a match
435*db0ac6deSCy Schubert 		 * before, we should probably recheck now that the
436*db0ac6deSCy Schubert 		 * inpcb lock is held.
437*db0ac6deSCy Schubert 		 */
438*db0ac6deSCy Schubert 		/*
439*db0ac6deSCy Schubert 		 * Handle socket delivery policy for any-source
440*db0ac6deSCy Schubert 		 * and source-specific multicast. [RFC3678]
441*db0ac6deSCy Schubert 		 */
442*db0ac6deSCy Schubert 		if (IN_MULTICAST(ntohl(ip->ip_dst.s_addr))) {
443*db0ac6deSCy Schubert 			struct ip_moptions	*imo;
444*db0ac6deSCy Schubert 			struct sockaddr_in	 group;
445*db0ac6deSCy Schubert 			int			 blocked;
446*db0ac6deSCy Schubert 
447*db0ac6deSCy Schubert 			imo = inp->inp_moptions;
448*db0ac6deSCy Schubert 			if (imo == NULL)
449*db0ac6deSCy Schubert 				continue;
450*db0ac6deSCy Schubert 			bzero(&group, sizeof(struct sockaddr_in));
451*db0ac6deSCy Schubert 			group.sin_len = sizeof(struct sockaddr_in);
452*db0ac6deSCy Schubert 			group.sin_family = AF_INET;
453*db0ac6deSCy Schubert 			group.sin_addr = ip->ip_dst;
454*db0ac6deSCy Schubert 
455*db0ac6deSCy Schubert 			blocked = imo_multi_filter(imo, m->m_pkthdr.rcvif,
456*db0ac6deSCy Schubert 				(struct sockaddr *)&group,
457*db0ac6deSCy Schubert 				(struct sockaddr *)&udp_in[0]);
458*db0ac6deSCy Schubert 			if (blocked != MCAST_PASS) {
459*db0ac6deSCy Schubert 				if (blocked == MCAST_NOTGMEMBER)
460*db0ac6deSCy Schubert 					IPSTAT_INC(ips_notmember);
461*db0ac6deSCy Schubert 				if (blocked == MCAST_NOTSMEMBER ||
462*db0ac6deSCy Schubert 				    blocked == MCAST_MUTED)
463*db0ac6deSCy Schubert 					UDPSTAT_INC(udps_filtermcast);
464*db0ac6deSCy Schubert 				continue;
465*db0ac6deSCy Schubert 			}
466*db0ac6deSCy Schubert 		}
467*db0ac6deSCy Schubert 		if ((n = m_copym(m, 0, M_COPYALL, M_NOWAIT)) != NULL) {
468*db0ac6deSCy Schubert 			if (proto == IPPROTO_UDPLITE)
469*db0ac6deSCy Schubert 				UDPLITE_PROBE(receive, NULL, inp, ip, inp, uh);
470*db0ac6deSCy Schubert 			else
471*db0ac6deSCy Schubert 				UDP_PROBE(receive, NULL, inp, ip, inp, uh);
472*db0ac6deSCy Schubert 			if (udp_append(inp, ip, n, sizeof(struct ip), udp_in)) {
473*db0ac6deSCy Schubert 				INP_RUNLOCK(inp);
474*db0ac6deSCy Schubert 				break;
475*db0ac6deSCy Schubert 			} else
476*db0ac6deSCy Schubert 				appends++;
477*db0ac6deSCy Schubert 		}
478*db0ac6deSCy Schubert 		/*
479*db0ac6deSCy Schubert 		 * Don't look for additional matches if this one does
480*db0ac6deSCy Schubert 		 * not have either the SO_REUSEPORT or SO_REUSEADDR
481*db0ac6deSCy Schubert 		 * socket options set.  This heuristic avoids
482*db0ac6deSCy Schubert 		 * searching through all pcbs in the common case of a
483*db0ac6deSCy Schubert 		 * non-shared port.  It assumes that an application
484*db0ac6deSCy Schubert 		 * will never clear these options after setting them.
485*db0ac6deSCy Schubert 		 */
486*db0ac6deSCy Schubert 		if ((inp->inp_socket->so_options &
487*db0ac6deSCy Schubert 		    (SO_REUSEPORT|SO_REUSEPORT_LB|SO_REUSEADDR)) == 0) {
488*db0ac6deSCy Schubert 			INP_RUNLOCK(inp);
489*db0ac6deSCy Schubert 			break;
490*db0ac6deSCy Schubert 		}
491*db0ac6deSCy Schubert 	}
492*db0ac6deSCy Schubert 	m_freem(m);
493*db0ac6deSCy Schubert 
494*db0ac6deSCy Schubert 	if (appends == 0) {
495*db0ac6deSCy Schubert 		/*
496*db0ac6deSCy Schubert 		 * No matching pcb found; discard datagram.  (No need
497*db0ac6deSCy Schubert 		 * to send an ICMP Port Unreachable for a broadcast
498*db0ac6deSCy Schubert 		 * or multicast datgram.)
499*db0ac6deSCy Schubert 		 */
500*db0ac6deSCy Schubert 		UDPSTAT_INC(udps_noport);
501*db0ac6deSCy Schubert 		if (IN_MULTICAST(ntohl(ip->ip_dst.s_addr)))
502*db0ac6deSCy Schubert 			UDPSTAT_INC(udps_noportmcast);
503*db0ac6deSCy Schubert 		else
504*db0ac6deSCy Schubert 			UDPSTAT_INC(udps_noportbcast);
505*db0ac6deSCy Schubert 	}
506*db0ac6deSCy Schubert 
507*db0ac6deSCy Schubert 	return (IPPROTO_DONE);
508*db0ac6deSCy Schubert }
509*db0ac6deSCy Schubert 
5108f5a8818SKevin Lo int
5118f5a8818SKevin Lo udp_input(struct mbuf **mp, int *offp, int proto)
512df8bae1dSRodney W. Grimes {
5133329b236SRobert Watson 	struct ip *ip;
5143329b236SRobert Watson 	struct udphdr *uh;
51571498f30SBruce M Simpson 	struct ifnet *ifp;
5163329b236SRobert Watson 	struct inpcb *inp;
5178f134647SGleb Smirnoff 	uint16_t len, ip_len;
518e06e816fSKevin Lo 	struct inpcbinfo *pcbinfo;
519dce33a45SErmal Luçi 	struct sockaddr_in udp_in[2];
5208f5a8818SKevin Lo 	struct mbuf *m;
5210b4ae859SGleb Smirnoff 	struct m_tag *fwd_tag;
5228f5a8818SKevin Lo 	int cscov_partial, iphlen;
523df8bae1dSRodney W. Grimes 
5248f5a8818SKevin Lo 	m = *mp;
5258f5a8818SKevin Lo 	iphlen = *offp;
52671498f30SBruce M Simpson 	ifp = m->m_pkthdr.rcvif;
5278f5a8818SKevin Lo 	*mp = NULL;
528026decb8SRobert Watson 	UDPSTAT_INC(udps_ipackets);
529df8bae1dSRodney W. Grimes 
530df8bae1dSRodney W. Grimes 	/*
5313329b236SRobert Watson 	 * Strip IP options, if any; should skip this, make available to
5323329b236SRobert Watson 	 * user, and use on returned packets, but we don't yet have a way to
5333329b236SRobert Watson 	 * check the checksum with options still present.
534df8bae1dSRodney W. Grimes 	 */
535df8bae1dSRodney W. Grimes 	if (iphlen > sizeof (struct ip)) {
536105bd211SGleb Smirnoff 		ip_stripoptions(m);
537df8bae1dSRodney W. Grimes 		iphlen = sizeof(struct ip);
538df8bae1dSRodney W. Grimes 	}
539df8bae1dSRodney W. Grimes 
540df8bae1dSRodney W. Grimes 	/*
541df8bae1dSRodney W. Grimes 	 * Get IP and UDP header together in first mbuf.
542df8bae1dSRodney W. Grimes 	 */
543df8bae1dSRodney W. Grimes 	if (m->m_len < iphlen + sizeof(struct udphdr)) {
544d1b18731SKevin Lo 		if ((m = m_pullup(m, iphlen + sizeof(struct udphdr))) == NULL) {
545026decb8SRobert Watson 			UDPSTAT_INC(udps_hdrops);
5468f5a8818SKevin Lo 			return (IPPROTO_DONE);
547df8bae1dSRodney W. Grimes 		}
548df8bae1dSRodney W. Grimes 	}
549503f4e47SBjoern A. Zeeb 	ip = mtod(m, struct ip *);
550df8bae1dSRodney W. Grimes 	uh = (struct udphdr *)((caddr_t)ip + iphlen);
5518f5a8818SKevin Lo 	cscov_partial = (proto == IPPROTO_UDPLITE) ? 1 : 0;
552df8bae1dSRodney W. Grimes 
5533329b236SRobert Watson 	/*
5543329b236SRobert Watson 	 * Destination port of 0 is illegal, based on RFC768.
5553329b236SRobert Watson 	 */
556686cdd19SJun-ichiro itojun Hagino 	if (uh->uh_dport == 0)
557f76fcf6dSJeffrey Hsu 		goto badunlocked;
558686cdd19SJun-ichiro itojun Hagino 
559df8bae1dSRodney W. Grimes 	/*
5603329b236SRobert Watson 	 * Construct sockaddr format source address.  Stuff source address
5613329b236SRobert Watson 	 * and datagram in user buffer.
562b9234fafSSam Leffler 	 */
563dce33a45SErmal Luçi 	bzero(&udp_in[0], sizeof(struct sockaddr_in) * 2);
564dce33a45SErmal Luçi 	udp_in[0].sin_len = sizeof(struct sockaddr_in);
565dce33a45SErmal Luçi 	udp_in[0].sin_family = AF_INET;
566dce33a45SErmal Luçi 	udp_in[0].sin_port = uh->uh_sport;
567dce33a45SErmal Luçi 	udp_in[0].sin_addr = ip->ip_src;
568dce33a45SErmal Luçi 	udp_in[1].sin_len = sizeof(struct sockaddr_in);
569dce33a45SErmal Luçi 	udp_in[1].sin_family = AF_INET;
570dce33a45SErmal Luçi 	udp_in[1].sin_port = uh->uh_dport;
571dce33a45SErmal Luçi 	udp_in[1].sin_addr = ip->ip_dst;
572b9234fafSSam Leffler 
573b9234fafSSam Leffler 	/*
574af1ee11dSRobert Watson 	 * Make mbuf data length reflect UDP length.  If not enough data to
575af1ee11dSRobert Watson 	 * reflect UDP length, drop.
576df8bae1dSRodney W. Grimes 	 */
577df8bae1dSRodney W. Grimes 	len = ntohs((u_short)uh->uh_ulen);
5788ad458a4SGleb Smirnoff 	ip_len = ntohs(ip->ip_len) - iphlen;
5790f4a0366SMichael Tuexen 	if (proto == IPPROTO_UDPLITE && (len == 0 || len == ip_len)) {
580e06e816fSKevin Lo 		/* Zero means checksum over the complete packet. */
5810f4a0366SMichael Tuexen 		if (len == 0)
582e06e816fSKevin Lo 			len = ip_len;
583e06e816fSKevin Lo 		cscov_partial = 0;
584e06e816fSKevin Lo 	}
5858f134647SGleb Smirnoff 	if (ip_len != len) {
5868f134647SGleb Smirnoff 		if (len > ip_len || len < sizeof(struct udphdr)) {
587026decb8SRobert Watson 			UDPSTAT_INC(udps_badlen);
588f76fcf6dSJeffrey Hsu 			goto badunlocked;
589df8bae1dSRodney W. Grimes 		}
5908f5a8818SKevin Lo 		if (proto == IPPROTO_UDP)
5918f134647SGleb Smirnoff 			m_adj(m, len - ip_len);
592df8bae1dSRodney W. Grimes 	}
5933329b236SRobert Watson 
594df8bae1dSRodney W. Grimes 	/*
595df8bae1dSRodney W. Grimes 	 * Checksum extended UDP header and data.
596df8bae1dSRodney W. Grimes 	 */
5976dfab5b1SGarrett Wollman 	if (uh->uh_sum) {
59839629c92SDavid Malone 		u_short uh_sum;
59939629c92SDavid Malone 
600e06e816fSKevin Lo 		if ((m->m_pkthdr.csum_flags & CSUM_DATA_VALID) &&
601e06e816fSKevin Lo 		    !cscov_partial) {
602db4f9cc7SJonathan Lemon 			if (m->m_pkthdr.csum_flags & CSUM_PSEUDO_HDR)
60339629c92SDavid Malone 				uh_sum = m->m_pkthdr.csum_data;
604db4f9cc7SJonathan Lemon 			else
60539629c92SDavid Malone 				uh_sum = in_pseudo(ip->ip_src.s_addr,
606506f4949SRuslan Ermilov 				    ip->ip_dst.s_addr, htonl((u_short)len +
6078f5a8818SKevin Lo 				    m->m_pkthdr.csum_data + proto));
60839629c92SDavid Malone 			uh_sum ^= 0xffff;
609db4f9cc7SJonathan Lemon 		} else {
6103358df29SGleb Smirnoff 			char b[offsetof(struct ipovly, ih_src)];
6113358df29SGleb Smirnoff 			struct ipovly *ipov = (struct ipovly *)ip;
612af1ee11dSRobert Watson 
6133358df29SGleb Smirnoff 			bcopy(ipov, b, sizeof(b));
6143358df29SGleb Smirnoff 			bzero(ipov, sizeof(ipov->ih_x1));
6153358df29SGleb Smirnoff 			ipov->ih_len = (proto == IPPROTO_UDP) ?
616e06e816fSKevin Lo 			    uh->uh_ulen : htons(ip_len);
61739629c92SDavid Malone 			uh_sum = in_cksum(m, len + sizeof (struct ip));
6183358df29SGleb Smirnoff 			bcopy(b, ipov, sizeof(b));
619db4f9cc7SJonathan Lemon 		}
62039629c92SDavid Malone 		if (uh_sum) {
621026decb8SRobert Watson 			UDPSTAT_INC(udps_badsum);
622df8bae1dSRodney W. Grimes 			m_freem(m);
6238f5a8818SKevin Lo 			return (IPPROTO_DONE);
624df8bae1dSRodney W. Grimes 		}
625c6d81a34SMichael Tuexen 	} else {
626c6d81a34SMichael Tuexen 		if (proto == IPPROTO_UDP) {
627026decb8SRobert Watson 			UDPSTAT_INC(udps_nosum);
628c6d81a34SMichael Tuexen 		} else {
629c6d81a34SMichael Tuexen 			/* UDPLite requires a checksum */
630c6d81a34SMichael Tuexen 			/* XXX: What is the right UDPLite MIB counter here? */
631c6d81a34SMichael Tuexen 			m_freem(m);
632c6d81a34SMichael Tuexen 			return (IPPROTO_DONE);
633c6d81a34SMichael Tuexen 		}
634c6d81a34SMichael Tuexen 	}
635df8bae1dSRodney W. Grimes 
636266f97b5SCy Schubert 	if (IN_MULTICAST(ntohl(ip->ip_dst.s_addr)) ||
637*db0ac6deSCy Schubert 	    in_broadcast(ip->ip_dst, ifp))
638*db0ac6deSCy Schubert 			return (udp_multi_input(m, proto, udp_in));
639266f97b5SCy Schubert 
640*db0ac6deSCy Schubert 	pcbinfo = udp_get_inpcbinfo(proto);
6413329b236SRobert Watson 
642df8bae1dSRodney W. Grimes 	/*
6436d6a026bSDavid Greenman 	 * Locate pcb for datagram.
644*db0ac6deSCy Schubert 	 *
6458a006adbSBjoern A. Zeeb 	 * Grab info from PACKET_TAG_IPFORWARD tag prepended to the chain.
6468a006adbSBjoern A. Zeeb 	 */
647ffdbf9daSAndrey V. Elsukov 	if ((m->m_flags & M_IP_NEXTHOP) &&
648c1de64a4SAndrey V. Elsukov 	    (fwd_tag = m_tag_find(m, PACKET_TAG_IPFORWARD, NULL)) != NULL) {
6498a006adbSBjoern A. Zeeb 		struct sockaddr_in *next_hop;
6508a006adbSBjoern A. Zeeb 
6518a006adbSBjoern A. Zeeb 		next_hop = (struct sockaddr_in *)(fwd_tag + 1);
6528a006adbSBjoern A. Zeeb 
6538a006adbSBjoern A. Zeeb 		/*
6548a006adbSBjoern A. Zeeb 		 * Transparently forwarded. Pretend to be the destination.
6558a006adbSBjoern A. Zeeb 		 * Already got one like this?
6568a006adbSBjoern A. Zeeb 		 */
657e06e816fSKevin Lo 		inp = in_pcblookup_mbuf(pcbinfo, ip->ip_src, uh->uh_sport,
6588a006adbSBjoern A. Zeeb 		    ip->ip_dst, uh->uh_dport, INPLOOKUP_RLOCKPCB, ifp, m);
6598a006adbSBjoern A. Zeeb 		if (!inp) {
6608a006adbSBjoern A. Zeeb 			/*
6618a006adbSBjoern A. Zeeb 			 * It's new.  Try to find the ambushing socket.
6628a006adbSBjoern A. Zeeb 			 * Because we've rewritten the destination address,
6638a006adbSBjoern A. Zeeb 			 * any hardware-generated hash is ignored.
6648a006adbSBjoern A. Zeeb 			 */
665e06e816fSKevin Lo 			inp = in_pcblookup(pcbinfo, ip->ip_src,
6668a006adbSBjoern A. Zeeb 			    uh->uh_sport, next_hop->sin_addr,
6678a006adbSBjoern A. Zeeb 			    next_hop->sin_port ? htons(next_hop->sin_port) :
6688a006adbSBjoern A. Zeeb 			    uh->uh_dport, INPLOOKUP_WILDCARD |
6698a006adbSBjoern A. Zeeb 			    INPLOOKUP_RLOCKPCB, ifp);
6708a006adbSBjoern A. Zeeb 		}
6718a006adbSBjoern A. Zeeb 		/* Remove the tag from the packet. We don't need it anymore. */
6728a006adbSBjoern A. Zeeb 		m_tag_delete(m, fwd_tag);
673ffdbf9daSAndrey V. Elsukov 		m->m_flags &= ~M_IP_NEXTHOP;
6748a006adbSBjoern A. Zeeb 	} else
675e06e816fSKevin Lo 		inp = in_pcblookup_mbuf(pcbinfo, ip->ip_src, uh->uh_sport,
6768a006adbSBjoern A. Zeeb 		    ip->ip_dst, uh->uh_dport, INPLOOKUP_WILDCARD |
6778a006adbSBjoern A. Zeeb 		    INPLOOKUP_RLOCKPCB, ifp, m);
67815bd2b43SDavid Greenman 	if (inp == NULL) {
679334fc582SBjoern A. Zeeb 		if (V_udp_log_in_vain) {
680edf0313bSEric van Gyzen 			char src[INET_ADDRSTRLEN];
681edf0313bSEric van Gyzen 			char dst[INET_ADDRSTRLEN];
68275cfc95fSAndrey A. Chernov 
683592071e8SBruce Evans 			log(LOG_INFO,
684592071e8SBruce Evans 			    "Connection attempt to UDP %s:%d from %s:%d\n",
685edf0313bSEric van Gyzen 			    inet_ntoa_r(ip->ip_dst, dst), ntohs(uh->uh_dport),
686edf0313bSEric van Gyzen 			    inet_ntoa_r(ip->ip_src, src), ntohs(uh->uh_sport));
68775cfc95fSAndrey A. Chernov 		}
6887bda9663SMichael Tuexen 		if (proto == IPPROTO_UDPLITE)
6897bda9663SMichael Tuexen 			UDPLITE_PROBE(receive, NULL, NULL, ip, NULL, uh);
6907bda9663SMichael Tuexen 		else
691e1526d5aSMichael Tuexen 			UDP_PROBE(receive, NULL, NULL, ip, NULL, uh);
692026decb8SRobert Watson 		UDPSTAT_INC(udps_noport);
693df8bae1dSRodney W. Grimes 		if (m->m_flags & (M_BCAST | M_MCAST)) {
694026decb8SRobert Watson 			UDPSTAT_INC(udps_noportbcast);
695fa046d87SRobert Watson 			goto badunlocked;
696df8bae1dSRodney W. Grimes 		}
6973ea9a7cfSGleb Smirnoff 		if (V_udp_blackhole && (V_udp_blackhole_local ||
6983ea9a7cfSGleb Smirnoff 		    !in_localip(ip->ip_src)))
699fa046d87SRobert Watson 			goto badunlocked;
7001cbd978eSLuigi Rizzo 		if (badport_bandlim(BANDLIM_ICMP_UNREACH) < 0)
701fa046d87SRobert Watson 			goto badunlocked;
702582a7760SBruce Evans 		icmp_error(m, ICMP_UNREACH, ICMP_UNREACH_PORT, 0, 0);
7038f5a8818SKevin Lo 		return (IPPROTO_DONE);
704df8bae1dSRodney W. Grimes 	}
7053329b236SRobert Watson 
7063329b236SRobert Watson 	/*
7073329b236SRobert Watson 	 * Check the minimum TTL for socket.
7083329b236SRobert Watson 	 */
709fa046d87SRobert Watson 	INP_RLOCK_ASSERT(inp);
71010cc62b7SRobert Watson 	if (inp->inp_ip_minttl && inp->inp_ip_minttl > ip->ip_ttl) {
7117bda9663SMichael Tuexen 		if (proto == IPPROTO_UDPLITE)
7127bda9663SMichael Tuexen 			UDPLITE_PROBE(receive, NULL, inp, ip, inp, uh);
7137bda9663SMichael Tuexen 		else
714e1526d5aSMichael Tuexen 			UDP_PROBE(receive, NULL, inp, ip, inp, uh);
71510cc62b7SRobert Watson 		INP_RUNLOCK(inp);
716fa046d87SRobert Watson 		m_freem(m);
7178f5a8818SKevin Lo 		return (IPPROTO_DONE);
71810cc62b7SRobert Watson 	}
719e06e816fSKevin Lo 	if (cscov_partial) {
720e06e816fSKevin Lo 		struct udpcb *up;
721e06e816fSKevin Lo 
722e06e816fSKevin Lo 		up = intoudpcb(inp);
72383e95fb3SMichael Tuexen 		if (up->u_rxcslen == 0 || up->u_rxcslen > len) {
724e06e816fSKevin Lo 			INP_RUNLOCK(inp);
725e06e816fSKevin Lo 			m_freem(m);
7268f5a8818SKevin Lo 			return (IPPROTO_DONE);
727e06e816fSKevin Lo 		}
728e06e816fSKevin Lo 	}
72957f60867SMark Johnston 
7307bda9663SMichael Tuexen 	if (proto == IPPROTO_UDPLITE)
7317bda9663SMichael Tuexen 		UDPLITE_PROBE(receive, NULL, inp, ip, inp, uh);
7327bda9663SMichael Tuexen 	else
7331ad19fb6SMark Johnston 		UDP_PROBE(receive, NULL, inp, ip, inp, uh);
734dce33a45SErmal Luçi 	if (udp_append(inp, ip, m, iphlen, udp_in) == 0)
735119d85f6SRobert Watson 		INP_RUNLOCK(inp);
7368f5a8818SKevin Lo 	return (IPPROTO_DONE);
73761ffc0b1SJeffrey Hsu 
738f76fcf6dSJeffrey Hsu badunlocked:
739df8bae1dSRodney W. Grimes 	m_freem(m);
7408f5a8818SKevin Lo 	return (IPPROTO_DONE);
741cfa1ca9dSYoshinobu Inoue }
74279288c11SBjoern A. Zeeb #endif /* INET */
743cfa1ca9dSYoshinobu Inoue 
744cfa1ca9dSYoshinobu Inoue /*
7453329b236SRobert Watson  * Notify a udp user of an asynchronous error; just wake up so that they can
7463329b236SRobert Watson  * collect error status.
747df8bae1dSRodney W. Grimes  */
7483ce144eaSJeffrey Hsu struct inpcb *
7493329b236SRobert Watson udp_notify(struct inpcb *inp, int errno)
750df8bae1dSRodney W. Grimes {
7513329b236SRobert Watson 
752083a010cSRyan Stone 	INP_WLOCK_ASSERT(inp);
75384cc0778SGeorge V. Neville-Neil 	if ((errno == EHOSTUNREACH || errno == ENETUNREACH ||
754983066f0SAlexander V. Chernikov 	     errno == EHOSTDOWN) && inp->inp_route.ro_nh) {
755983066f0SAlexander V. Chernikov 		NH_FREE(inp->inp_route.ro_nh);
756983066f0SAlexander V. Chernikov 		inp->inp_route.ro_nh = (struct nhop_object *)NULL;
75784cc0778SGeorge V. Neville-Neil 	}
7588501a69cSRobert Watson 
759df8bae1dSRodney W. Grimes 	inp->inp_socket->so_error = errno;
760df8bae1dSRodney W. Grimes 	sorwakeup(inp->inp_socket);
761df8bae1dSRodney W. Grimes 	sowwakeup(inp->inp_socket);
7623329b236SRobert Watson 	return (inp);
763df8bae1dSRodney W. Grimes }
764df8bae1dSRodney W. Grimes 
76579288c11SBjoern A. Zeeb #ifdef INET
766e06e816fSKevin Lo static void
767e06e816fSKevin Lo udp_common_ctlinput(int cmd, struct sockaddr *sa, void *vip,
768e06e816fSKevin Lo     struct inpcbinfo *pcbinfo)
769df8bae1dSRodney W. Grimes {
770c693a045SJonathan Lemon 	struct ip *ip = vip;
771c693a045SJonathan Lemon 	struct udphdr *uh;
772c693a045SJonathan Lemon 	struct in_addr faddr;
773c693a045SJonathan Lemon 	struct inpcb *inp;
774c693a045SJonathan Lemon 
775c693a045SJonathan Lemon 	faddr = ((struct sockaddr_in *)sa)->sin_addr;
776c693a045SJonathan Lemon 	if (sa->sa_family != AF_INET || faddr.s_addr == INADDR_ANY)
777c693a045SJonathan Lemon 		return;
778df8bae1dSRodney W. Grimes 
77984cc0778SGeorge V. Neville-Neil 	if (PRC_IS_REDIRECT(cmd)) {
78084cc0778SGeorge V. Neville-Neil 		/* signal EHOSTDOWN, as it flushes the cached route */
7814f321dbdSBjoern A. Zeeb 		in_pcbnotifyall(&V_udbinfo, faddr, EHOSTDOWN, udp_notify);
78297d8d152SAndre Oppermann 		return;
78384cc0778SGeorge V. Neville-Neil 	}
7843329b236SRobert Watson 
78597d8d152SAndre Oppermann 	/*
78697d8d152SAndre Oppermann 	 * Hostdead is ugly because it goes linearly through all PCBs.
7873329b236SRobert Watson 	 *
7883329b236SRobert Watson 	 * XXX: We never get this from ICMP, otherwise it makes an excellent
7893329b236SRobert Watson 	 * DoS attack on machines with many connections.
79097d8d152SAndre Oppermann 	 */
79197d8d152SAndre Oppermann 	if (cmd == PRC_HOSTDEAD)
792af1ee11dSRobert Watson 		ip = NULL;
793d1c54148SJesper Skriver 	else if ((unsigned)cmd >= PRC_NCMDS || inetctlerrmap[cmd] == 0)
794df8bae1dSRodney W. Grimes 		return;
795af1ee11dSRobert Watson 	if (ip != NULL) {
796df8bae1dSRodney W. Grimes 		uh = (struct udphdr *)((caddr_t)ip + (ip->ip_hl << 2));
797e06e816fSKevin Lo 		inp = in_pcblookup(pcbinfo, faddr, uh->uh_dport,
798083a010cSRyan Stone 		    ip->ip_src, uh->uh_sport, INPLOOKUP_WLOCKPCB, NULL);
799f76fcf6dSJeffrey Hsu 		if (inp != NULL) {
800083a010cSRyan Stone 			INP_WLOCK_ASSERT(inp);
801f76fcf6dSJeffrey Hsu 			if (inp->inp_socket != NULL) {
802f5514f08SRobert Watson 				udp_notify(inp, inetctlerrmap[cmd]);
803f76fcf6dSJeffrey Hsu 			}
804083a010cSRyan Stone 			INP_WUNLOCK(inp);
805abb901c5SRandall Stewart 		} else {
806abb901c5SRandall Stewart 			inp = in_pcblookup(pcbinfo, faddr, uh->uh_dport,
807abb901c5SRandall Stewart 					   ip->ip_src, uh->uh_sport,
808abb901c5SRandall Stewart 					   INPLOOKUP_WILDCARD | INPLOOKUP_RLOCKPCB, NULL);
809abb901c5SRandall Stewart 			if (inp != NULL) {
810abb901c5SRandall Stewart 				struct udpcb *up;
81146374cbfSMatt Macy 				void *ctx;
81246374cbfSMatt Macy 				udp_tun_icmp_t func;
813abb901c5SRandall Stewart 
814abb901c5SRandall Stewart 				up = intoudpcb(inp);
81546374cbfSMatt Macy 				ctx = up->u_tun_ctx;
81646374cbfSMatt Macy 				func = up->u_icmp_func;
817abb901c5SRandall Stewart 				INP_RUNLOCK(inp);
81846374cbfSMatt Macy 				if (func != NULL)
81946374cbfSMatt Macy 					(*func)(cmd, sa, vip, ctx);
820abb901c5SRandall Stewart 			}
821f76fcf6dSJeffrey Hsu 		}
822df8bae1dSRodney W. Grimes 	} else
823e06e816fSKevin Lo 		in_pcbnotifyall(pcbinfo, faddr, inetctlerrmap[cmd],
824f5514f08SRobert Watson 		    udp_notify);
825df8bae1dSRodney W. Grimes }
826e06e816fSKevin Lo void
827e06e816fSKevin Lo udp_ctlinput(int cmd, struct sockaddr *sa, void *vip)
828e06e816fSKevin Lo {
829e06e816fSKevin Lo 
830e06e816fSKevin Lo 	return (udp_common_ctlinput(cmd, sa, vip, &V_udbinfo));
831e06e816fSKevin Lo }
832e06e816fSKevin Lo 
833e06e816fSKevin Lo void
834e06e816fSKevin Lo udplite_ctlinput(int cmd, struct sockaddr *sa, void *vip)
835e06e816fSKevin Lo {
836e06e816fSKevin Lo 
837e06e816fSKevin Lo 	return (udp_common_ctlinput(cmd, sa, vip, &V_ulitecbinfo));
838e06e816fSKevin Lo }
83979288c11SBjoern A. Zeeb #endif /* INET */
840df8bae1dSRodney W. Grimes 
8410312fbe9SPoul-Henning Kamp static int
84282d9ae4eSPoul-Henning Kamp udp_pcblist(SYSCTL_HANDLER_ARGS)
84398271db4SGarrett Wollman {
844*db0ac6deSCy Schubert 	struct inpcb_iterator inpi = INP_ALL_ITERATOR(&V_udbinfo,
845*db0ac6deSCy Schubert 	    INPLOOKUP_RLOCKPCB);
84698271db4SGarrett Wollman 	struct xinpgen xig;
847032677ceSGleb Smirnoff 	struct inpcb *inp;
848032677ceSGleb Smirnoff 	int error;
84998271db4SGarrett Wollman 
850032677ceSGleb Smirnoff 	if (req->newptr != 0)
851032677ceSGleb Smirnoff 		return (EPERM);
852032677ceSGleb Smirnoff 
85398271db4SGarrett Wollman 	if (req->oldptr == 0) {
854032677ceSGleb Smirnoff 		int n;
855032677ceSGleb Smirnoff 
856603724d3SBjoern A. Zeeb 		n = V_udbinfo.ipi_count;
857c007b96aSJohn Baldwin 		n += imax(n / 8, 10);
858c007b96aSJohn Baldwin 		req->oldidx = 2 * (sizeof xig) + n * sizeof(struct xinpcb);
8593329b236SRobert Watson 		return (0);
86098271db4SGarrett Wollman 	}
86198271db4SGarrett Wollman 
862032677ceSGleb Smirnoff 	if ((error = sysctl_wire_old_buffer(req, 0)) != 0)
86347934cefSDon Lewis 		return (error);
8645c38b6dbSDon Lewis 
86579db6fe7SMark Johnston 	bzero(&xig, sizeof(xig));
86698271db4SGarrett Wollman 	xig.xig_len = sizeof xig;
867032677ceSGleb Smirnoff 	xig.xig_count = V_udbinfo.ipi_count;
868032677ceSGleb Smirnoff 	xig.xig_gen = V_udbinfo.ipi_gencnt;
86998271db4SGarrett Wollman 	xig.xig_sogen = so_gencnt;
87098271db4SGarrett Wollman 	error = SYSCTL_OUT(req, &xig, sizeof xig);
87198271db4SGarrett Wollman 	if (error)
8723329b236SRobert Watson 		return (error);
87399208b82SMatt Macy 
874*db0ac6deSCy Schubert 	while ((inp = inp_next(&inpi)) != NULL) {
875032677ceSGleb Smirnoff 		if (inp->inp_gencnt <= xig.xig_gen &&
876032677ceSGleb Smirnoff 		    cr_canseeinpcb(req->td->td_ucred, inp) == 0) {
87798271db4SGarrett Wollman 			struct xinpcb xi;
878d0e157f6SBjoern A. Zeeb 
879cc65eb4eSGleb Smirnoff 			in_pcbtoxinpcb(inp, &xi);
880266f97b5SCy Schubert 			error = SYSCTL_OUT(req, &xi, sizeof xi);
881*db0ac6deSCy Schubert 			if (error) {
882266f97b5SCy Schubert 				INP_RUNLOCK(inp);
883*db0ac6deSCy Schubert 				break;
88498271db4SGarrett Wollman 			}
885*db0ac6deSCy Schubert 		}
886*db0ac6deSCy Schubert 	}
887d0e157f6SBjoern A. Zeeb 
88898271db4SGarrett Wollman 	if (!error) {
88998271db4SGarrett Wollman 		/*
8903329b236SRobert Watson 		 * Give the user an updated idea of our state.  If the
8913329b236SRobert Watson 		 * generation differs from what we told her before, she knows
8923329b236SRobert Watson 		 * that something happened while we were processing this
8933329b236SRobert Watson 		 * request, and it might be necessary to retry.
89498271db4SGarrett Wollman 		 */
895603724d3SBjoern A. Zeeb 		xig.xig_gen = V_udbinfo.ipi_gencnt;
89698271db4SGarrett Wollman 		xig.xig_sogen = so_gencnt;
897603724d3SBjoern A. Zeeb 		xig.xig_count = V_udbinfo.ipi_count;
89898271db4SGarrett Wollman 		error = SYSCTL_OUT(req, &xig, sizeof xig);
89998271db4SGarrett Wollman 	}
900032677ceSGleb Smirnoff 
9013329b236SRobert Watson 	return (error);
90298271db4SGarrett Wollman }
90398271db4SGarrett Wollman 
90479c3d51bSMatthew D Fleming SYSCTL_PROC(_net_inet_udp, UDPCTL_PCBLIST, pcblist,
9057029da5cSPawel Biernacki     CTLTYPE_OPAQUE | CTLFLAG_RD | CTLFLAG_MPSAFE, NULL, 0,
9067029da5cSPawel Biernacki     udp_pcblist, "S,xinpcb",
9077029da5cSPawel Biernacki     "List of active UDP sockets");
90898271db4SGarrett Wollman 
90979288c11SBjoern A. Zeeb #ifdef INET
91098271db4SGarrett Wollman static int
91182d9ae4eSPoul-Henning Kamp udp_getcred(SYSCTL_HANDLER_ARGS)
912490d50b6SBrian Feldman {
913c0511d3bSBrian Feldman 	struct xucred xuc;
914490d50b6SBrian Feldman 	struct sockaddr_in addrs[2];
915d797164aSGleb Smirnoff 	struct epoch_tracker et;
916490d50b6SBrian Feldman 	struct inpcb *inp;
917277afaffSRobert Watson 	int error;
918490d50b6SBrian Feldman 
91932f9753cSRobert Watson 	error = priv_check(req->td, PRIV_NETINET_GETCRED);
920490d50b6SBrian Feldman 	if (error)
921490d50b6SBrian Feldman 		return (error);
922490d50b6SBrian Feldman 	error = SYSCTL_IN(req, addrs, sizeof(addrs));
923490d50b6SBrian Feldman 	if (error)
924490d50b6SBrian Feldman 		return (error);
925d797164aSGleb Smirnoff 	NET_EPOCH_ENTER(et);
926fa046d87SRobert Watson 	inp = in_pcblookup(&V_udbinfo, addrs[1].sin_addr, addrs[1].sin_port,
927fa046d87SRobert Watson 	    addrs[0].sin_addr, addrs[0].sin_port,
928fa046d87SRobert Watson 	    INPLOOKUP_WILDCARD | INPLOOKUP_RLOCKPCB, NULL);
929d797164aSGleb Smirnoff 	NET_EPOCH_EXIT(et);
9309622e84fSRobert Watson 	if (inp != NULL) {
931fa046d87SRobert Watson 		INP_RLOCK_ASSERT(inp);
9329622e84fSRobert Watson 		if (inp->inp_socket == NULL)
9339622e84fSRobert Watson 			error = ENOENT;
9349622e84fSRobert Watson 		if (error == 0)
935f08ef6c5SBjoern A. Zeeb 			error = cr_canseeinpcb(req->td->td_ucred, inp);
9369622e84fSRobert Watson 		if (error == 0)
93786d02c5cSBjoern A. Zeeb 			cru2x(inp->inp_cred, &xuc);
9389622e84fSRobert Watson 		INP_RUNLOCK(inp);
939fa046d87SRobert Watson 	} else
9409622e84fSRobert Watson 		error = ENOENT;
9410e1eebb8SDon Lewis 	if (error == 0)
9420e1eebb8SDon Lewis 		error = SYSCTL_OUT(req, &xuc, sizeof(struct xucred));
943490d50b6SBrian Feldman 	return (error);
944490d50b6SBrian Feldman }
945490d50b6SBrian Feldman 
9467ce87f12SDavid Malone SYSCTL_PROC(_net_inet_udp, OID_AUTO, getcred,
9477029da5cSPawel Biernacki     CTLTYPE_OPAQUE | CTLFLAG_RW | CTLFLAG_PRISON | CTLFLAG_MPSAFE,
9487029da5cSPawel Biernacki     0, 0, udp_getcred, "S,xucred",
9497029da5cSPawel Biernacki     "Get the xucred of a UDP connection");
95079288c11SBjoern A. Zeeb #endif /* INET */
951490d50b6SBrian Feldman 
9527b495c44SVANHULLEBUS Yvan int
9537b495c44SVANHULLEBUS Yvan udp_ctloutput(struct socket *so, struct sockopt *sopt)
9547b495c44SVANHULLEBUS Yvan {
9557b495c44SVANHULLEBUS Yvan 	struct inpcb *inp;
9567b495c44SVANHULLEBUS Yvan 	struct udpcb *up;
957e06e816fSKevin Lo 	int isudplite, error, optval;
9587b495c44SVANHULLEBUS Yvan 
959e06e816fSKevin Lo 	error = 0;
960e06e816fSKevin Lo 	isudplite = (so->so_proto->pr_protocol == IPPROTO_UDPLITE) ? 1 : 0;
9617b495c44SVANHULLEBUS Yvan 	inp = sotoinpcb(so);
9627b495c44SVANHULLEBUS Yvan 	KASSERT(inp != NULL, ("%s: inp == NULL", __func__));
9637b495c44SVANHULLEBUS Yvan 	INP_WLOCK(inp);
964e06e816fSKevin Lo 	if (sopt->sopt_level != so->so_proto->pr_protocol) {
9657b495c44SVANHULLEBUS Yvan #ifdef INET6
9667b495c44SVANHULLEBUS Yvan 		if (INP_CHECK_SOCKAF(so, AF_INET6)) {
9677b495c44SVANHULLEBUS Yvan 			INP_WUNLOCK(inp);
9687b495c44SVANHULLEBUS Yvan 			error = ip6_ctloutput(so, sopt);
96979288c11SBjoern A. Zeeb 		}
9707b495c44SVANHULLEBUS Yvan #endif
97179288c11SBjoern A. Zeeb #if defined(INET) && defined(INET6)
97279288c11SBjoern A. Zeeb 		else
97379288c11SBjoern A. Zeeb #endif
97479288c11SBjoern A. Zeeb #ifdef INET
97579288c11SBjoern A. Zeeb 		{
9767b495c44SVANHULLEBUS Yvan 			INP_WUNLOCK(inp);
9777b495c44SVANHULLEBUS Yvan 			error = ip_ctloutput(so, sopt);
9787b495c44SVANHULLEBUS Yvan 		}
9797b495c44SVANHULLEBUS Yvan #endif
9807b495c44SVANHULLEBUS Yvan 		return (error);
9817b495c44SVANHULLEBUS Yvan 	}
9827b495c44SVANHULLEBUS Yvan 
9837b495c44SVANHULLEBUS Yvan 	switch (sopt->sopt_dir) {
9847b495c44SVANHULLEBUS Yvan 	case SOPT_SET:
9857b495c44SVANHULLEBUS Yvan 		switch (sopt->sopt_name) {
986fcf59617SAndrey V. Elsukov #if defined(IPSEC) || defined(IPSEC_SUPPORT)
987fcf59617SAndrey V. Elsukov #ifdef INET
9887b495c44SVANHULLEBUS Yvan 		case UDP_ENCAP:
989fcf59617SAndrey V. Elsukov 			if (!IPSEC_ENABLED(ipv4)) {
9907b495c44SVANHULLEBUS Yvan 				INP_WUNLOCK(inp);
991fcf59617SAndrey V. Elsukov 				return (ENOPROTOOPT);
9927b495c44SVANHULLEBUS Yvan 			}
993fcf59617SAndrey V. Elsukov 			error = UDPENCAP_PCBCTL(inp, sopt);
9947b495c44SVANHULLEBUS Yvan 			break;
995fcf59617SAndrey V. Elsukov #endif /* INET */
996fcf59617SAndrey V. Elsukov #endif /* IPSEC */
997e06e816fSKevin Lo 		case UDPLITE_SEND_CSCOV:
998e06e816fSKevin Lo 		case UDPLITE_RECV_CSCOV:
999e06e816fSKevin Lo 			if (!isudplite) {
1000e06e816fSKevin Lo 				INP_WUNLOCK(inp);
1001e06e816fSKevin Lo 				error = ENOPROTOOPT;
1002e06e816fSKevin Lo 				break;
1003e06e816fSKevin Lo 			}
1004e06e816fSKevin Lo 			INP_WUNLOCK(inp);
1005e06e816fSKevin Lo 			error = sooptcopyin(sopt, &optval, sizeof(optval),
1006e06e816fSKevin Lo 			    sizeof(optval));
1007e06e816fSKevin Lo 			if (error != 0)
1008e06e816fSKevin Lo 				break;
1009e06e816fSKevin Lo 			inp = sotoinpcb(so);
1010e06e816fSKevin Lo 			KASSERT(inp != NULL, ("%s: inp == NULL", __func__));
1011e06e816fSKevin Lo 			INP_WLOCK(inp);
1012e06e816fSKevin Lo 			up = intoudpcb(inp);
1013e06e816fSKevin Lo 			KASSERT(up != NULL, ("%s: up == NULL", __func__));
101403f90784SMichael Tuexen 			if ((optval != 0 && optval < 8) || (optval > 65535)) {
1015e06e816fSKevin Lo 				INP_WUNLOCK(inp);
1016e06e816fSKevin Lo 				error = EINVAL;
1017e06e816fSKevin Lo 				break;
1018e06e816fSKevin Lo 			}
1019e06e816fSKevin Lo 			if (sopt->sopt_name == UDPLITE_SEND_CSCOV)
1020e06e816fSKevin Lo 				up->u_txcslen = optval;
1021e06e816fSKevin Lo 			else
1022e06e816fSKevin Lo 				up->u_rxcslen = optval;
1023e06e816fSKevin Lo 			INP_WUNLOCK(inp);
1024e06e816fSKevin Lo 			break;
10257b495c44SVANHULLEBUS Yvan 		default:
10267b495c44SVANHULLEBUS Yvan 			INP_WUNLOCK(inp);
10277b495c44SVANHULLEBUS Yvan 			error = ENOPROTOOPT;
10287b495c44SVANHULLEBUS Yvan 			break;
10297b495c44SVANHULLEBUS Yvan 		}
10307b495c44SVANHULLEBUS Yvan 		break;
10317b495c44SVANHULLEBUS Yvan 	case SOPT_GET:
10327b495c44SVANHULLEBUS Yvan 		switch (sopt->sopt_name) {
1033fcf59617SAndrey V. Elsukov #if defined(IPSEC) || defined(IPSEC_SUPPORT)
1034fcf59617SAndrey V. Elsukov #ifdef INET
10357b495c44SVANHULLEBUS Yvan 		case UDP_ENCAP:
1036fcf59617SAndrey V. Elsukov 			if (!IPSEC_ENABLED(ipv4)) {
10377b495c44SVANHULLEBUS Yvan 				INP_WUNLOCK(inp);
1038fcf59617SAndrey V. Elsukov 				return (ENOPROTOOPT);
1039fcf59617SAndrey V. Elsukov 			}
1040fcf59617SAndrey V. Elsukov 			error = UDPENCAP_PCBCTL(inp, sopt);
10417b495c44SVANHULLEBUS Yvan 			break;
1042fcf59617SAndrey V. Elsukov #endif /* INET */
1043fcf59617SAndrey V. Elsukov #endif /* IPSEC */
1044e06e816fSKevin Lo 		case UDPLITE_SEND_CSCOV:
1045e06e816fSKevin Lo 		case UDPLITE_RECV_CSCOV:
1046e06e816fSKevin Lo 			if (!isudplite) {
1047e06e816fSKevin Lo 				INP_WUNLOCK(inp);
1048e06e816fSKevin Lo 				error = ENOPROTOOPT;
1049e06e816fSKevin Lo 				break;
1050e06e816fSKevin Lo 			}
1051e06e816fSKevin Lo 			up = intoudpcb(inp);
1052e06e816fSKevin Lo 			KASSERT(up != NULL, ("%s: up == NULL", __func__));
1053e06e816fSKevin Lo 			if (sopt->sopt_name == UDPLITE_SEND_CSCOV)
1054e06e816fSKevin Lo 				optval = up->u_txcslen;
1055e06e816fSKevin Lo 			else
1056e06e816fSKevin Lo 				optval = up->u_rxcslen;
1057e06e816fSKevin Lo 			INP_WUNLOCK(inp);
1058e06e816fSKevin Lo 			error = sooptcopyout(sopt, &optval, sizeof(optval));
1059e06e816fSKevin Lo 			break;
10607b495c44SVANHULLEBUS Yvan 		default:
10617b495c44SVANHULLEBUS Yvan 			INP_WUNLOCK(inp);
10627b495c44SVANHULLEBUS Yvan 			error = ENOPROTOOPT;
10637b495c44SVANHULLEBUS Yvan 			break;
10647b495c44SVANHULLEBUS Yvan 		}
10657b495c44SVANHULLEBUS Yvan 		break;
10667b495c44SVANHULLEBUS Yvan 	}
10677b495c44SVANHULLEBUS Yvan 	return (error);
10687b495c44SVANHULLEBUS Yvan }
10697b495c44SVANHULLEBUS Yvan 
107079288c11SBjoern A. Zeeb #ifdef INET
1071a9839c4aSBjoern A. Zeeb #ifdef INET6
1072a9839c4aSBjoern A. Zeeb /* The logic here is derived from ip6_setpktopt(). See comments there. */
1073a9839c4aSBjoern A. Zeeb static int
1074a9839c4aSBjoern A. Zeeb udp_v4mapped_pktinfo(struct cmsghdr *cm, struct sockaddr_in * src,
1075a9839c4aSBjoern A. Zeeb     struct inpcb *inp, int flags)
1076a9839c4aSBjoern A. Zeeb {
1077a9839c4aSBjoern A. Zeeb 	struct ifnet *ifp;
1078a9839c4aSBjoern A. Zeeb 	struct in6_pktinfo *pktinfo;
1079a9839c4aSBjoern A. Zeeb 	struct in_addr ia;
1080a9839c4aSBjoern A. Zeeb 
1081a9839c4aSBjoern A. Zeeb 	if ((flags & PRUS_IPV6) == 0)
1082a9839c4aSBjoern A. Zeeb 		return (0);
1083a9839c4aSBjoern A. Zeeb 
1084a9839c4aSBjoern A. Zeeb 	if (cm->cmsg_level != IPPROTO_IPV6)
1085a9839c4aSBjoern A. Zeeb 		return (0);
1086a9839c4aSBjoern A. Zeeb 
1087a9839c4aSBjoern A. Zeeb 	if  (cm->cmsg_type != IPV6_2292PKTINFO &&
1088a9839c4aSBjoern A. Zeeb 	    cm->cmsg_type != IPV6_PKTINFO)
1089a9839c4aSBjoern A. Zeeb 		return (0);
1090a9839c4aSBjoern A. Zeeb 
1091a9839c4aSBjoern A. Zeeb 	if (cm->cmsg_len !=
1092a9839c4aSBjoern A. Zeeb 	    CMSG_LEN(sizeof(struct in6_pktinfo)))
1093a9839c4aSBjoern A. Zeeb 		return (EINVAL);
1094a9839c4aSBjoern A. Zeeb 
1095a9839c4aSBjoern A. Zeeb 	pktinfo = (struct in6_pktinfo *)CMSG_DATA(cm);
1096a9839c4aSBjoern A. Zeeb 	if (!IN6_IS_ADDR_V4MAPPED(&pktinfo->ipi6_addr) &&
1097a9839c4aSBjoern A. Zeeb 	    !IN6_IS_ADDR_UNSPECIFIED(&pktinfo->ipi6_addr))
1098a9839c4aSBjoern A. Zeeb 		return (EINVAL);
1099a9839c4aSBjoern A. Zeeb 
1100a9839c4aSBjoern A. Zeeb 	/* Validate the interface index if specified. */
1101a9839c4aSBjoern A. Zeeb 	if (pktinfo->ipi6_ifindex > V_if_index)
1102a9839c4aSBjoern A. Zeeb 		return (ENXIO);
1103a9839c4aSBjoern A. Zeeb 
1104a9839c4aSBjoern A. Zeeb 	ifp = NULL;
1105a9839c4aSBjoern A. Zeeb 	if (pktinfo->ipi6_ifindex) {
1106a9839c4aSBjoern A. Zeeb 		ifp = ifnet_byindex(pktinfo->ipi6_ifindex);
1107a9839c4aSBjoern A. Zeeb 		if (ifp == NULL)
1108a9839c4aSBjoern A. Zeeb 			return (ENXIO);
1109a9839c4aSBjoern A. Zeeb 	}
1110a9839c4aSBjoern A. Zeeb 	if (ifp != NULL && !IN6_IS_ADDR_UNSPECIFIED(&pktinfo->ipi6_addr)) {
1111a9839c4aSBjoern A. Zeeb 		ia.s_addr = pktinfo->ipi6_addr.s6_addr32[3];
1112a9839c4aSBjoern A. Zeeb 		if (in_ifhasaddr(ifp, ia) == 0)
1113a9839c4aSBjoern A. Zeeb 			return (EADDRNOTAVAIL);
1114a9839c4aSBjoern A. Zeeb 	}
1115a9839c4aSBjoern A. Zeeb 
1116a9839c4aSBjoern A. Zeeb 	bzero(src, sizeof(*src));
1117a9839c4aSBjoern A. Zeeb 	src->sin_family = AF_INET;
1118a9839c4aSBjoern A. Zeeb 	src->sin_len = sizeof(*src);
1119a9839c4aSBjoern A. Zeeb 	src->sin_port = inp->inp_lport;
1120a9839c4aSBjoern A. Zeeb 	src->sin_addr.s_addr = pktinfo->ipi6_addr.s6_addr32[3];
1121a9839c4aSBjoern A. Zeeb 
1122a9839c4aSBjoern A. Zeeb 	return (0);
1123a9839c4aSBjoern A. Zeeb }
1124a9839c4aSBjoern A. Zeeb #endif
1125a9839c4aSBjoern A. Zeeb 
1126490d50b6SBrian Feldman static int
11273329b236SRobert Watson udp_output(struct inpcb *inp, struct mbuf *m, struct sockaddr *addr,
1128a9839c4aSBjoern A. Zeeb     struct mbuf *control, struct thread *td, int flags)
1129df8bae1dSRodney W. Grimes {
11303329b236SRobert Watson 	struct udpiphdr *ui;
11313329b236SRobert Watson 	int len = m->m_pkthdr.len;
113290162a4eSIan Dowse 	struct in_addr faddr, laddr;
1133c557ae16SIan Dowse 	struct cmsghdr *cm;
1134e06e816fSKevin Lo 	struct inpcbinfo *pcbinfo;
1135c557ae16SIan Dowse 	struct sockaddr_in *sin, src;
11366573d758SMatt Macy 	struct epoch_tracker et;
1137e06e816fSKevin Lo 	int cscov_partial = 0;
113890162a4eSIan Dowse 	int error = 0;
1139374ce248SMitchell Horne 	int ipflags = 0;
114090162a4eSIan Dowse 	u_short fport, lport;
1141f584d74bSMichael Tuexen 	u_char tos;
1142e06e816fSKevin Lo 	uint8_t pr;
1143e06e816fSKevin Lo 	uint16_t cscov = 0;
11449d3ddf43SAdrian Chadd 	uint32_t flowid = 0;
1145c2529042SHans Petter Selasky 	uint8_t flowtype = M_HASHTYPE_NONE;
1146df8bae1dSRodney W. Grimes 
1147430d30d8SBill Fenner 	if (len + sizeof(struct udpiphdr) > IP_MAXPACKET) {
1148c557ae16SIan Dowse 		if (control)
1149c557ae16SIan Dowse 			m_freem(control);
11505c32ea65SRobert Watson 		m_freem(m);
11513329b236SRobert Watson 		return (EMSGSIZE);
1152430d30d8SBill Fenner 	}
1153430d30d8SBill Fenner 
11541b7f0384SBruce M Simpson 	src.sin_family = 0;
115584cc0778SGeorge V. Neville-Neil 	sin = (struct sockaddr_in *)addr;
11562435e507SGleb Smirnoff 
1157eafaa1bcSBjoern A. Zeeb 	/*
11582435e507SGleb Smirnoff 	 * udp_output() may need to temporarily bind or connect the current
11592435e507SGleb Smirnoff 	 * inpcb.  As such, we don't know up front whether we will need the
11602435e507SGleb Smirnoff 	 * pcbinfo lock or not.  Do any work to decide what is needed up
11612435e507SGleb Smirnoff 	 * front before acquiring any locks.
11622435e507SGleb Smirnoff 	 *
11632435e507SGleb Smirnoff 	 * We will need network epoch in either case, to safely lookup into
11642435e507SGleb Smirnoff 	 * pcb hash.
1165eafaa1bcSBjoern A. Zeeb 	 */
11662435e507SGleb Smirnoff 	if (sin == NULL ||
11672435e507SGleb Smirnoff 	    (inp->inp_laddr.s_addr == INADDR_ANY && inp->inp_lport == 0))
11682435e507SGleb Smirnoff 		INP_WLOCK(inp);
11692435e507SGleb Smirnoff 	else
11700cfdff24SBjoern A. Zeeb 		INP_RLOCK(inp);
11712435e507SGleb Smirnoff 	NET_EPOCH_ENTER(et);
1172f584d74bSMichael Tuexen 	tos = inp->inp_ip_tos;
1173c557ae16SIan Dowse 	if (control != NULL) {
1174c557ae16SIan Dowse 		/*
11753329b236SRobert Watson 		 * XXX: Currently, we assume all the optional information is
11763329b236SRobert Watson 		 * stored in a single mbuf.
1177c557ae16SIan Dowse 		 */
1178c557ae16SIan Dowse 		if (control->m_next) {
1179c557ae16SIan Dowse 			m_freem(control);
11802435e507SGleb Smirnoff 			error = EINVAL;
11812435e507SGleb Smirnoff 			goto release;
1182c557ae16SIan Dowse 		}
1183c557ae16SIan Dowse 		for (; control->m_len > 0;
1184c557ae16SIan Dowse 		    control->m_data += CMSG_ALIGN(cm->cmsg_len),
1185c557ae16SIan Dowse 		    control->m_len -= CMSG_ALIGN(cm->cmsg_len)) {
1186c557ae16SIan Dowse 			cm = mtod(control, struct cmsghdr *);
1187af1ee11dSRobert Watson 			if (control->m_len < sizeof(*cm) || cm->cmsg_len == 0
1188af1ee11dSRobert Watson 			    || cm->cmsg_len > control->m_len) {
1189c557ae16SIan Dowse 				error = EINVAL;
1190c557ae16SIan Dowse 				break;
1191c557ae16SIan Dowse 			}
1192a9839c4aSBjoern A. Zeeb #ifdef INET6
1193a9839c4aSBjoern A. Zeeb 			error = udp_v4mapped_pktinfo(cm, &src, inp, flags);
1194a9839c4aSBjoern A. Zeeb 			if (error != 0)
1195a9839c4aSBjoern A. Zeeb 				break;
1196a9839c4aSBjoern A. Zeeb #endif
1197c557ae16SIan Dowse 			if (cm->cmsg_level != IPPROTO_IP)
1198c557ae16SIan Dowse 				continue;
1199c557ae16SIan Dowse 
1200c557ae16SIan Dowse 			switch (cm->cmsg_type) {
1201c557ae16SIan Dowse 			case IP_SENDSRCADDR:
1202c557ae16SIan Dowse 				if (cm->cmsg_len !=
1203c557ae16SIan Dowse 				    CMSG_LEN(sizeof(struct in_addr))) {
1204c557ae16SIan Dowse 					error = EINVAL;
1205c557ae16SIan Dowse 					break;
1206c557ae16SIan Dowse 				}
1207c557ae16SIan Dowse 				bzero(&src, sizeof(src));
1208c557ae16SIan Dowse 				src.sin_family = AF_INET;
1209c557ae16SIan Dowse 				src.sin_len = sizeof(src);
1210c557ae16SIan Dowse 				src.sin_port = inp->inp_lport;
1211af1ee11dSRobert Watson 				src.sin_addr =
1212af1ee11dSRobert Watson 				    *(struct in_addr *)CMSG_DATA(cm);
1213c557ae16SIan Dowse 				break;
1214af1ee11dSRobert Watson 
1215f584d74bSMichael Tuexen 			case IP_TOS:
1216f584d74bSMichael Tuexen 				if (cm->cmsg_len != CMSG_LEN(sizeof(u_char))) {
1217f584d74bSMichael Tuexen 					error = EINVAL;
1218f584d74bSMichael Tuexen 					break;
1219f584d74bSMichael Tuexen 				}
1220f584d74bSMichael Tuexen 				tos = *(u_char *)CMSG_DATA(cm);
1221f584d74bSMichael Tuexen 				break;
1222f584d74bSMichael Tuexen 
12239d3ddf43SAdrian Chadd 			case IP_FLOWID:
12249d3ddf43SAdrian Chadd 				if (cm->cmsg_len != CMSG_LEN(sizeof(uint32_t))) {
12259d3ddf43SAdrian Chadd 					error = EINVAL;
12269d3ddf43SAdrian Chadd 					break;
12279d3ddf43SAdrian Chadd 				}
12289d3ddf43SAdrian Chadd 				flowid = *(uint32_t *) CMSG_DATA(cm);
12299d3ddf43SAdrian Chadd 				break;
12309d3ddf43SAdrian Chadd 
12319d3ddf43SAdrian Chadd 			case IP_FLOWTYPE:
12329d3ddf43SAdrian Chadd 				if (cm->cmsg_len != CMSG_LEN(sizeof(uint32_t))) {
12339d3ddf43SAdrian Chadd 					error = EINVAL;
12349d3ddf43SAdrian Chadd 					break;
12359d3ddf43SAdrian Chadd 				}
1236c2529042SHans Petter Selasky 				flowtype = *(uint32_t *) CMSG_DATA(cm);
12379d3ddf43SAdrian Chadd 				break;
12389d3ddf43SAdrian Chadd 
12399d3ddf43SAdrian Chadd #ifdef	RSS
12409d3ddf43SAdrian Chadd 			case IP_RSSBUCKETID:
12419d3ddf43SAdrian Chadd 				if (cm->cmsg_len != CMSG_LEN(sizeof(uint32_t))) {
12429d3ddf43SAdrian Chadd 					error = EINVAL;
12439d3ddf43SAdrian Chadd 					break;
12449d3ddf43SAdrian Chadd 				}
12459d3ddf43SAdrian Chadd 				/* This is just a placeholder for now */
12469d3ddf43SAdrian Chadd 				break;
12479d3ddf43SAdrian Chadd #endif	/* RSS */
1248c557ae16SIan Dowse 			default:
1249c557ae16SIan Dowse 				error = ENOPROTOOPT;
1250c557ae16SIan Dowse 				break;
1251c557ae16SIan Dowse 			}
1252c557ae16SIan Dowse 			if (error)
1253c557ae16SIan Dowse 				break;
1254c557ae16SIan Dowse 		}
1255c557ae16SIan Dowse 		m_freem(control);
1256d8acd268SMark Johnston 		control = NULL;
1257c557ae16SIan Dowse 	}
12582435e507SGleb Smirnoff 	if (error)
12592435e507SGleb Smirnoff 		goto release;
12605c32ea65SRobert Watson 
1261e06e816fSKevin Lo 	pr = inp->inp_socket->so_proto->pr_protocol;
1262a86e5c96SBjoern A. Zeeb 	pcbinfo = udp_get_inpcbinfo(pr);
12635c32ea65SRobert Watson 
12641b7f0384SBruce M Simpson 	/*
12651b7f0384SBruce M Simpson 	 * If the IP_SENDSRCADDR control message was specified, override the
12661b7f0384SBruce M Simpson 	 * source address for this datagram.  Its use is invalidated if the
12671b7f0384SBruce M Simpson 	 * address thus specified is incomplete or clobbers other inpcbs.
12681b7f0384SBruce M Simpson 	 */
126990162a4eSIan Dowse 	laddr = inp->inp_laddr;
127090162a4eSIan Dowse 	lport = inp->inp_lport;
12711b7f0384SBruce M Simpson 	if (src.sin_family == AF_INET) {
12721b7f0384SBruce M Simpson 		if ((lport == 0) ||
12731b7f0384SBruce M Simpson 		    (laddr.s_addr == INADDR_ANY &&
12741b7f0384SBruce M Simpson 		     src.sin_addr.s_addr == INADDR_ANY)) {
1275c557ae16SIan Dowse 			error = EINVAL;
1276c557ae16SIan Dowse 			goto release;
1277c557ae16SIan Dowse 		}
1278*db0ac6deSCy Schubert 		INP_HASH_WLOCK(pcbinfo);
1279c557ae16SIan Dowse 		error = in_pcbbind_setup(inp, (struct sockaddr *)&src,
1280b0330ed9SPawel Jakub Dawidek 		    &laddr.s_addr, &lport, td->td_ucred);
1281*db0ac6deSCy Schubert 		INP_HASH_WUNLOCK(pcbinfo);
1282c557ae16SIan Dowse 		if (error)
1283c557ae16SIan Dowse 			goto release;
1284c557ae16SIan Dowse 	}
1285c557ae16SIan Dowse 
12863144b7d3SRobert Watson 	/*
12873144b7d3SRobert Watson 	 * If a UDP socket has been connected, then a local address/port will
12883144b7d3SRobert Watson 	 * have been selected and bound.
12893144b7d3SRobert Watson 	 *
129043cc0bc1SRobert Watson 	 * If a UDP socket has not been connected to, then an explicit
12913144b7d3SRobert Watson 	 * destination address must be used, in which case a local
12923144b7d3SRobert Watson 	 * address/port may not have been selected and bound.
12933144b7d3SRobert Watson 	 */
129443cc0bc1SRobert Watson 	if (sin != NULL) {
1295c4d585aeSRobert Watson 		INP_LOCK_ASSERT(inp);
1296df8bae1dSRodney W. Grimes 		if (inp->inp_faddr.s_addr != INADDR_ANY) {
1297df8bae1dSRodney W. Grimes 			error = EISCONN;
1298df8bae1dSRodney W. Grimes 			goto release;
1299df8bae1dSRodney W. Grimes 		}
13003144b7d3SRobert Watson 
13013144b7d3SRobert Watson 		/*
13023144b7d3SRobert Watson 		 * Jail may rewrite the destination address, so let it do
13033144b7d3SRobert Watson 		 * that before we use it.
13043144b7d3SRobert Watson 		 */
1305b89e82ddSJamie Gritton 		error = prison_remote_ip4(td->td_ucred, &sin->sin_addr);
1306b89e82ddSJamie Gritton 		if (error)
1307413628a7SBjoern A. Zeeb 			goto release;
13083144b7d3SRobert Watson 
13093144b7d3SRobert Watson 		/*
131043cc0bc1SRobert Watson 		 * If a local address or port hasn't yet been selected, or if
131143cc0bc1SRobert Watson 		 * the destination address needs to be rewritten due to using
131243cc0bc1SRobert Watson 		 * a special INADDR_ constant, invoke in_pcbconnect_setup()
131343cc0bc1SRobert Watson 		 * to do the heavy lifting.  Once a port is selected, we
131443cc0bc1SRobert Watson 		 * commit the binding back to the socket; we also commit the
131543cc0bc1SRobert Watson 		 * binding of the address if in jail.
131643cc0bc1SRobert Watson 		 *
131743cc0bc1SRobert Watson 		 * If we already have a valid binding and we're not
131843cc0bc1SRobert Watson 		 * requesting a destination address rewrite, use a fast path.
13193144b7d3SRobert Watson 		 */
132043cc0bc1SRobert Watson 		if (inp->inp_laddr.s_addr == INADDR_ANY ||
132143cc0bc1SRobert Watson 		    inp->inp_lport == 0 ||
132243cc0bc1SRobert Watson 		    sin->sin_addr.s_addr == INADDR_ANY ||
132343cc0bc1SRobert Watson 		    sin->sin_addr.s_addr == INADDR_BROADCAST) {
1324*db0ac6deSCy Schubert 			INP_HASH_WLOCK(pcbinfo);
132543cc0bc1SRobert Watson 			error = in_pcbconnect_setup(inp, addr, &laddr.s_addr,
132643cc0bc1SRobert Watson 			    &lport, &faddr.s_addr, &fport, NULL,
132743cc0bc1SRobert Watson 			    td->td_ucred);
1328*db0ac6deSCy Schubert 			if (error) {
1329*db0ac6deSCy Schubert 				INP_HASH_WUNLOCK(pcbinfo);
133090162a4eSIan Dowse 				goto release;
1331*db0ac6deSCy Schubert 			}
133290162a4eSIan Dowse 
133343cc0bc1SRobert Watson 			/*
133443cc0bc1SRobert Watson 			 * XXXRW: Why not commit the port if the address is
133543cc0bc1SRobert Watson 			 * !INADDR_ANY?
133643cc0bc1SRobert Watson 			 */
133790162a4eSIan Dowse 			/* Commit the local port if newly assigned. */
133890162a4eSIan Dowse 			if (inp->inp_laddr.s_addr == INADDR_ANY &&
133990162a4eSIan Dowse 			    inp->inp_lport == 0) {
1340c4d585aeSRobert Watson 				INP_WLOCK_ASSERT(inp);
13413a1757b9SGleb Smirnoff 				/*
134243cc0bc1SRobert Watson 				 * Remember addr if jailed, to prevent
134343cc0bc1SRobert Watson 				 * rebinding.
13443a1757b9SGleb Smirnoff 				 */
13450304c731SJamie Gritton 				if (prison_flag(td->td_ucred, PR_IP4))
13463a1757b9SGleb Smirnoff 					inp->inp_laddr = laddr;
134790162a4eSIan Dowse 				inp->inp_lport = lport;
13482435e507SGleb Smirnoff 				error = in_pcbinshash(inp);
13492435e507SGleb Smirnoff 				INP_HASH_WUNLOCK(pcbinfo);
13502435e507SGleb Smirnoff 				if (error != 0) {
135190162a4eSIan Dowse 					inp->inp_lport = 0;
135290162a4eSIan Dowse 					error = EAGAIN;
1353df8bae1dSRodney W. Grimes 					goto release;
1354df8bae1dSRodney W. Grimes 				}
135590162a4eSIan Dowse 				inp->inp_flags |= INP_ANONPORT;
1356*db0ac6deSCy Schubert 			} else
1357*db0ac6deSCy Schubert 				INP_HASH_WUNLOCK(pcbinfo);
1358df8bae1dSRodney W. Grimes 		} else {
135943cc0bc1SRobert Watson 			faddr = sin->sin_addr;
136043cc0bc1SRobert Watson 			fport = sin->sin_port;
136143cc0bc1SRobert Watson 		}
136243cc0bc1SRobert Watson 	} else {
1363c4d585aeSRobert Watson 		INP_LOCK_ASSERT(inp);
136490162a4eSIan Dowse 		faddr = inp->inp_faddr;
136590162a4eSIan Dowse 		fport = inp->inp_fport;
136690162a4eSIan Dowse 		if (faddr.s_addr == INADDR_ANY) {
1367df8bae1dSRodney W. Grimes 			error = ENOTCONN;
1368df8bae1dSRodney W. Grimes 			goto release;
1369df8bae1dSRodney W. Grimes 		}
1370df8bae1dSRodney W. Grimes 	}
1371e6ccd709SRobert Watson 
1372df8bae1dSRodney W. Grimes 	/*
1373e6ccd709SRobert Watson 	 * Calculate data length and get a mbuf for UDP, IP, and possible
1374392e8407SRobert Watson 	 * link-layer headers.  Immediate slide the data pointer back forward
1375392e8407SRobert Watson 	 * since we won't use that space at this layer.
1376df8bae1dSRodney W. Grimes 	 */
1377eb1b1807SGleb Smirnoff 	M_PREPEND(m, sizeof(struct udpiphdr) + max_linkhdr, M_NOWAIT);
1378e6ccd709SRobert Watson 	if (m == NULL) {
1379df8bae1dSRodney W. Grimes 		error = ENOBUFS;
138049b19bfcSBruce M Simpson 		goto release;
1381df8bae1dSRodney W. Grimes 	}
1382e6ccd709SRobert Watson 	m->m_data += max_linkhdr;
1383e6ccd709SRobert Watson 	m->m_len -= max_linkhdr;
1384392e8407SRobert Watson 	m->m_pkthdr.len -= max_linkhdr;
1385df8bae1dSRodney W. Grimes 
1386df8bae1dSRodney W. Grimes 	/*
13873329b236SRobert Watson 	 * Fill in mbuf with extended UDP header and addresses and length put
13883329b236SRobert Watson 	 * into network format.
1389df8bae1dSRodney W. Grimes 	 */
1390df8bae1dSRodney W. Grimes 	ui = mtod(m, struct udpiphdr *);
1391db4f9cc7SJonathan Lemon 	bzero(ui->ui_x1, sizeof(ui->ui_x1));	/* XXX still needed? */
139234fc9072SMichael Tuexen 	ui->ui_v = IPVERSION << 4;
1393e06e816fSKevin Lo 	ui->ui_pr = pr;
139490162a4eSIan Dowse 	ui->ui_src = laddr;
139590162a4eSIan Dowse 	ui->ui_dst = faddr;
139690162a4eSIan Dowse 	ui->ui_sport = lport;
139790162a4eSIan Dowse 	ui->ui_dport = fport;
1398db4f9cc7SJonathan Lemon 	ui->ui_ulen = htons((u_short)len + sizeof(struct udphdr));
1399e06e816fSKevin Lo 	if (pr == IPPROTO_UDPLITE) {
1400e06e816fSKevin Lo 		struct udpcb *up;
1401e06e816fSKevin Lo 		uint16_t plen;
1402e06e816fSKevin Lo 
1403e06e816fSKevin Lo 		up = intoudpcb(inp);
1404e06e816fSKevin Lo 		cscov = up->u_txcslen;
1405e06e816fSKevin Lo 		plen = (u_short)len + sizeof(struct udphdr);
1406e06e816fSKevin Lo 		if (cscov >= plen)
1407e06e816fSKevin Lo 			cscov = 0;
1408e06e816fSKevin Lo 		ui->ui_len = htons(plen);
1409e06e816fSKevin Lo 		ui->ui_ulen = htons(cscov);
1410e06e816fSKevin Lo 		/*
1411e06e816fSKevin Lo 		 * For UDP-Lite, checksum coverage length of zero means
1412e06e816fSKevin Lo 		 * the entire UDPLite packet is covered by the checksum.
1413e06e816fSKevin Lo 		 */
1414e06e816fSKevin Lo 		cscov_partial = (cscov == 0) ? 0 : 1;
141534fc9072SMichael Tuexen 	}
1416df8bae1dSRodney W. Grimes 
1417b2828ad2SAndre Oppermann 	/*
1418b2828ad2SAndre Oppermann 	 * Set the Don't Fragment bit in the IP header.
1419b2828ad2SAndre Oppermann 	 */
1420b2828ad2SAndre Oppermann 	if (inp->inp_flags & INP_DONTFRAG) {
1421b2828ad2SAndre Oppermann 		struct ip *ip;
14223329b236SRobert Watson 
1423b2828ad2SAndre Oppermann 		ip = (struct ip *)&ui->ui_i;
14248f134647SGleb Smirnoff 		ip->ip_off |= htons(IP_DF);
1425b2828ad2SAndre Oppermann 	}
1426b2828ad2SAndre Oppermann 
1427b5d47ff5SJohn-Mark Gurney 	if (inp->inp_socket->so_options & SO_DONTROUTE)
1428b5d47ff5SJohn-Mark Gurney 		ipflags |= IP_ROUTETOIF;
1429b5d47ff5SJohn-Mark Gurney 	if (inp->inp_socket->so_options & SO_BROADCAST)
1430b5d47ff5SJohn-Mark Gurney 		ipflags |= IP_ALLOWBROADCAST;
14316fbfd582SAndre Oppermann 	if (inp->inp_flags & INP_ONESBCAST)
14328afa2304SBruce M Simpson 		ipflags |= IP_SENDONES;
14338afa2304SBruce M Simpson 
14341175d9d5SRobert Watson #ifdef MAC
14351175d9d5SRobert Watson 	mac_inpcb_create_mbuf(inp, m);
14361175d9d5SRobert Watson #endif
14371175d9d5SRobert Watson 
1438df8bae1dSRodney W. Grimes 	/*
1439db4f9cc7SJonathan Lemon 	 * Set up checksum and output datagram.
1440df8bae1dSRodney W. Grimes 	 */
1441e06e816fSKevin Lo 	ui->ui_sum = 0;
1442a485f139SMichael Tuexen 	if (pr == IPPROTO_UDPLITE) {
1443e06e816fSKevin Lo 		if (inp->inp_flags & INP_ONESBCAST)
1444e06e816fSKevin Lo 			faddr.s_addr = INADDR_BROADCAST;
1445a485f139SMichael Tuexen 		if (cscov_partial) {
1446e06e816fSKevin Lo 			if ((ui->ui_sum = in_cksum(m, sizeof(struct ip) + cscov)) == 0)
1447e06e816fSKevin Lo 				ui->ui_sum = 0xffff;
1448a485f139SMichael Tuexen 		} else {
1449a485f139SMichael Tuexen 			if ((ui->ui_sum = in_cksum(m, sizeof(struct udpiphdr) + len)) == 0)
1450a485f139SMichael Tuexen 				ui->ui_sum = 0xffff;
1451a485f139SMichael Tuexen 		}
1452a485f139SMichael Tuexen 	} else if (V_udp_cksum) {
14536fbfd582SAndre Oppermann 		if (inp->inp_flags & INP_ONESBCAST)
14548a538743SBruce M Simpson 			faddr.s_addr = INADDR_BROADCAST;
14558a538743SBruce M Simpson 		ui->ui_sum = in_pseudo(ui->ui_src.s_addr, faddr.s_addr,
1456e06e816fSKevin Lo 		    htons((u_short)len + sizeof(struct udphdr) + pr));
1457db4f9cc7SJonathan Lemon 		m->m_pkthdr.csum_flags = CSUM_UDP;
1458db4f9cc7SJonathan Lemon 		m->m_pkthdr.csum_data = offsetof(struct udphdr, uh_sum);
1459e06e816fSKevin Lo 	}
14608f134647SGleb Smirnoff 	((struct ip *)ui)->ip_len = htons(sizeof(struct udpiphdr) + len);
1461ca98b82cSDavid Greenman 	((struct ip *)ui)->ip_ttl = inp->inp_ip_ttl;	/* XXX */
1462f584d74bSMichael Tuexen 	((struct ip *)ui)->ip_tos = tos;		/* XXX */
1463026decb8SRobert Watson 	UDPSTAT_INC(udps_opackets);
1464cfa1ca9dSYoshinobu Inoue 
14659d3ddf43SAdrian Chadd 	/*
14669d3ddf43SAdrian Chadd 	 * Setup flowid / RSS information for outbound socket.
14679d3ddf43SAdrian Chadd 	 *
14689d3ddf43SAdrian Chadd 	 * Once the UDP code decides to set a flowid some other way,
14699d3ddf43SAdrian Chadd 	 * this allows the flowid to be overridden by userland.
14709d3ddf43SAdrian Chadd 	 */
1471c2529042SHans Petter Selasky 	if (flowtype != M_HASHTYPE_NONE) {
14729d3ddf43SAdrian Chadd 		m->m_pkthdr.flowid = flowid;
1473c2529042SHans Petter Selasky 		M_HASHTYPE_SET(m, flowtype);
1474f9a6e8d7SBjoern A. Zeeb 	}
14750c325f53SAlexander V. Chernikov #if defined(ROUTE_MPATH) || defined(RSS)
14760c325f53SAlexander V. Chernikov 	else if (CALC_FLOWID_OUTBOUND_SENDTO) {
14778ad1a83bSAdrian Chadd 		uint32_t hash_val, hash_type;
14780c325f53SAlexander V. Chernikov 
14790c325f53SAlexander V. Chernikov 		hash_val = fib4_calc_packet_hash(laddr, faddr,
14800c325f53SAlexander V. Chernikov 		    lport, fport, pr, &hash_type);
14818ad1a83bSAdrian Chadd 		m->m_pkthdr.flowid = hash_val;
14828ad1a83bSAdrian Chadd 		M_HASHTYPE_SET(m, hash_type);
14838ad1a83bSAdrian Chadd 	}
14849d3ddf43SAdrian Chadd 
14858ad1a83bSAdrian Chadd 	/*
14868ad1a83bSAdrian Chadd 	 * Don't override with the inp cached flowid value.
14878ad1a83bSAdrian Chadd 	 *
14888ad1a83bSAdrian Chadd 	 * Depending upon the kind of send being done, the inp
14898ad1a83bSAdrian Chadd 	 * flowid/flowtype values may actually not be appropriate
14908ad1a83bSAdrian Chadd 	 * for this particular socket send.
14918ad1a83bSAdrian Chadd 	 *
14928ad1a83bSAdrian Chadd 	 * We should either leave the flowid at zero (which is what is
14938ad1a83bSAdrian Chadd 	 * currently done) or set it to some software generated
14948ad1a83bSAdrian Chadd 	 * hash value based on the packet contents.
14958ad1a83bSAdrian Chadd 	 */
14969d3ddf43SAdrian Chadd 	ipflags |= IP_NODEFAULTFLOWID;
14979d3ddf43SAdrian Chadd #endif	/* RSS */
14989d3ddf43SAdrian Chadd 
14997bda9663SMichael Tuexen 	if (pr == IPPROTO_UDPLITE)
15007bda9663SMichael Tuexen 		UDPLITE_PROBE(send, NULL, inp, &ui->ui_i, inp, &ui->ui_u);
15017bda9663SMichael Tuexen 	else
150257f60867SMark Johnston 		UDP_PROBE(send, NULL, inp, &ui->ui_i, inp, &ui->ui_u);
150384cc0778SGeorge V. Neville-Neil 	error = ip_output(m, inp->inp_options,
15042435e507SGleb Smirnoff 	    INP_WLOCKED(inp) ? &inp->inp_route : NULL, ipflags,
15055d846453SSam Leffler 	    inp->inp_moptions, inp);
15062435e507SGleb Smirnoff 	INP_UNLOCK(inp);
15072435e507SGleb Smirnoff 	NET_EPOCH_EXIT(et);
1508df8bae1dSRodney W. Grimes 	return (error);
1509df8bae1dSRodney W. Grimes 
1510df8bae1dSRodney W. Grimes release:
15112435e507SGleb Smirnoff 	INP_UNLOCK(inp);
15122435e507SGleb Smirnoff 	NET_EPOCH_EXIT(et);
1513df8bae1dSRodney W. Grimes 	m_freem(m);
1514df8bae1dSRodney W. Grimes 	return (error);
1515df8bae1dSRodney W. Grimes }
1516df8bae1dSRodney W. Grimes 
1517ac45e92fSRobert Watson static void
1518d0390e05SGarrett Wollman udp_abort(struct socket *so)
1519df8bae1dSRodney W. Grimes {
1520d0390e05SGarrett Wollman 	struct inpcb *inp;
1521e06e816fSKevin Lo 	struct inpcbinfo *pcbinfo;
1522df8bae1dSRodney W. Grimes 
1523a86e5c96SBjoern A. Zeeb 	pcbinfo = udp_get_inpcbinfo(so->so_proto->pr_protocol);
1524d0390e05SGarrett Wollman 	inp = sotoinpcb(so);
152514ba8addSRobert Watson 	KASSERT(inp != NULL, ("udp_abort: inp == NULL"));
15268501a69cSRobert Watson 	INP_WLOCK(inp);
1527a152f8a3SRobert Watson 	if (inp->inp_faddr.s_addr != INADDR_ANY) {
1528e06e816fSKevin Lo 		INP_HASH_WLOCK(pcbinfo);
1529a152f8a3SRobert Watson 		in_pcbdisconnect(inp);
1530a152f8a3SRobert Watson 		inp->inp_laddr.s_addr = INADDR_ANY;
1531e06e816fSKevin Lo 		INP_HASH_WUNLOCK(pcbinfo);
1532d0390e05SGarrett Wollman 		soisdisconnected(so);
1533a152f8a3SRobert Watson 	}
15348501a69cSRobert Watson 	INP_WUNLOCK(inp);
1535df8bae1dSRodney W. Grimes }
1536df8bae1dSRodney W. Grimes 
1537d0390e05SGarrett Wollman static int
1538b40ce416SJulian Elischer udp_attach(struct socket *so, int proto, struct thread *td)
1539d0390e05SGarrett Wollman {
1540630ba2c5SMatt Macy 	static uint32_t udp_flowid;
1541d0390e05SGarrett Wollman 	struct inpcb *inp;
1542e06e816fSKevin Lo 	struct inpcbinfo *pcbinfo;
1543277afaffSRobert Watson 	int error;
1544d0390e05SGarrett Wollman 
1545a86e5c96SBjoern A. Zeeb 	pcbinfo = udp_get_inpcbinfo(so->so_proto->pr_protocol);
1546d0390e05SGarrett Wollman 	inp = sotoinpcb(so);
154714ba8addSRobert Watson 	KASSERT(inp == NULL, ("udp_attach: inp != NULL"));
1548cfa1ca9dSYoshinobu Inoue 	error = soreserve(so, udp_sendspace, udp_recvspace);
1549f24618aaSRobert Watson 	if (error)
15503329b236SRobert Watson 		return (error);
1551e06e816fSKevin Lo 	error = in_pcballoc(so, pcbinfo);
1552*db0ac6deSCy Schubert 	if (error)
15533329b236SRobert Watson 		return (error);
1554cfa1ca9dSYoshinobu Inoue 
155568b5629bSRobert Watson 	inp = sotoinpcb(so);
1556cfa1ca9dSYoshinobu Inoue 	inp->inp_vflag |= INP_IPV4;
1557603724d3SBjoern A. Zeeb 	inp->inp_ip_ttl = V_ip_defttl;
1558630ba2c5SMatt Macy 	inp->inp_flowid = atomic_fetchadd_int(&udp_flowid, 1);
1559630ba2c5SMatt Macy 	inp->inp_flowtype = M_HASHTYPE_OPAQUE;
15606a9148feSBjoern A. Zeeb 
15616a9148feSBjoern A. Zeeb 	error = udp_newudpcb(inp);
15626a9148feSBjoern A. Zeeb 	if (error) {
15636a9148feSBjoern A. Zeeb 		in_pcbdetach(inp);
15646a9148feSBjoern A. Zeeb 		in_pcbfree(inp);
15656a9148feSBjoern A. Zeeb 		return (error);
15666a9148feSBjoern A. Zeeb 	}
1567266f97b5SCy Schubert 	INP_WUNLOCK(inp);
1568*db0ac6deSCy Schubert 
1569c7c7ea4bSRandall Stewart 	return (0);
1570c7c7ea4bSRandall Stewart }
157179288c11SBjoern A. Zeeb #endif /* INET */
1572c7c7ea4bSRandall Stewart 
1573c7c7ea4bSRandall Stewart int
1574abb901c5SRandall Stewart udp_set_kernel_tunneling(struct socket *so, udp_tun_func_t f, udp_tun_icmp_t i, void *ctx)
1575c7c7ea4bSRandall Stewart {
1576c7c7ea4bSRandall Stewart 	struct inpcb *inp;
15776a9148feSBjoern A. Zeeb 	struct udpcb *up;
1578c7c7ea4bSRandall Stewart 
157968b5629bSRobert Watson 	KASSERT(so->so_type == SOCK_DGRAM,
158068b5629bSRobert Watson 	    ("udp_set_kernel_tunneling: !dgram"));
158168b5629bSRobert Watson 	inp = sotoinpcb(so);
158268b5629bSRobert Watson 	KASSERT(inp != NULL, ("udp_set_kernel_tunneling: inp == NULL"));
1583c7c7ea4bSRandall Stewart 	INP_WLOCK(inp);
15846a9148feSBjoern A. Zeeb 	up = intoudpcb(inp);
1585abb901c5SRandall Stewart 	if ((up->u_tun_func != NULL) ||
1586abb901c5SRandall Stewart 	    (up->u_icmp_func != NULL)) {
1587bbb0e3d9SRandall Stewart 		INP_WUNLOCK(inp);
1588bbb0e3d9SRandall Stewart 		return (EBUSY);
1589bbb0e3d9SRandall Stewart 	}
15906a9148feSBjoern A. Zeeb 	up->u_tun_func = f;
1591abb901c5SRandall Stewart 	up->u_icmp_func = i;
159281d3ec17SBryan Venteicher 	up->u_tun_ctx = ctx;
15938501a69cSRobert Watson 	INP_WUNLOCK(inp);
15943329b236SRobert Watson 	return (0);
1595df8bae1dSRodney W. Grimes }
1596d0390e05SGarrett Wollman 
159779288c11SBjoern A. Zeeb #ifdef INET
1598d0390e05SGarrett Wollman static int
1599b40ce416SJulian Elischer udp_bind(struct socket *so, struct sockaddr *nam, struct thread *td)
1600d0390e05SGarrett Wollman {
1601d0390e05SGarrett Wollman 	struct inpcb *inp;
1602e06e816fSKevin Lo 	struct inpcbinfo *pcbinfo;
1603f96603b5SMark Johnston 	struct sockaddr_in *sinp;
1604277afaffSRobert Watson 	int error;
1605d0390e05SGarrett Wollman 
1606a86e5c96SBjoern A. Zeeb 	pcbinfo = udp_get_inpcbinfo(so->so_proto->pr_protocol);
1607d0390e05SGarrett Wollman 	inp = sotoinpcb(so);
160814ba8addSRobert Watson 	KASSERT(inp != NULL, ("udp_bind: inp == NULL"));
1609f161d294SMark Johnston 
1610f96603b5SMark Johnston 	sinp = (struct sockaddr_in *)nam;
1611f96603b5SMark Johnston 	if (nam->sa_family != AF_INET) {
1612f96603b5SMark Johnston 		/*
1613f96603b5SMark Johnston 		 * Preserve compatibility with old programs.
1614f96603b5SMark Johnston 		 */
1615f96603b5SMark Johnston 		if (nam->sa_family != AF_UNSPEC ||
16163f1f6b6eSMichael Tuexen 		    nam->sa_len < offsetof(struct sockaddr_in, sin_zero) ||
1617f96603b5SMark Johnston 		    sinp->sin_addr.s_addr != INADDR_ANY)
1618f161d294SMark Johnston 			return (EAFNOSUPPORT);
1619f96603b5SMark Johnston 		nam->sa_family = AF_INET;
1620f96603b5SMark Johnston 	}
1621f161d294SMark Johnston 	if (nam->sa_len != sizeof(struct sockaddr_in))
1622f161d294SMark Johnston 		return (EINVAL);
1623f161d294SMark Johnston 
16248501a69cSRobert Watson 	INP_WLOCK(inp);
1625e06e816fSKevin Lo 	INP_HASH_WLOCK(pcbinfo);
1626b0330ed9SPawel Jakub Dawidek 	error = in_pcbbind(inp, nam, td->td_ucred);
1627e06e816fSKevin Lo 	INP_HASH_WUNLOCK(pcbinfo);
16288501a69cSRobert Watson 	INP_WUNLOCK(inp);
16293329b236SRobert Watson 	return (error);
1630d0390e05SGarrett Wollman }
1631d0390e05SGarrett Wollman 
1632a152f8a3SRobert Watson static void
1633a152f8a3SRobert Watson udp_close(struct socket *so)
1634a152f8a3SRobert Watson {
1635a152f8a3SRobert Watson 	struct inpcb *inp;
1636e06e816fSKevin Lo 	struct inpcbinfo *pcbinfo;
1637a152f8a3SRobert Watson 
1638a86e5c96SBjoern A. Zeeb 	pcbinfo = udp_get_inpcbinfo(so->so_proto->pr_protocol);
1639a152f8a3SRobert Watson 	inp = sotoinpcb(so);
1640a152f8a3SRobert Watson 	KASSERT(inp != NULL, ("udp_close: inp == NULL"));
16418501a69cSRobert Watson 	INP_WLOCK(inp);
1642a152f8a3SRobert Watson 	if (inp->inp_faddr.s_addr != INADDR_ANY) {
1643e06e816fSKevin Lo 		INP_HASH_WLOCK(pcbinfo);
1644a152f8a3SRobert Watson 		in_pcbdisconnect(inp);
1645a152f8a3SRobert Watson 		inp->inp_laddr.s_addr = INADDR_ANY;
1646e06e816fSKevin Lo 		INP_HASH_WUNLOCK(pcbinfo);
1647a152f8a3SRobert Watson 		soisdisconnected(so);
1648a152f8a3SRobert Watson 	}
16498501a69cSRobert Watson 	INP_WUNLOCK(inp);
1650a152f8a3SRobert Watson }
1651a152f8a3SRobert Watson 
1652d0390e05SGarrett Wollman static int
1653b40ce416SJulian Elischer udp_connect(struct socket *so, struct sockaddr *nam, struct thread *td)
1654d0390e05SGarrett Wollman {
1655c1604fe4SGleb Smirnoff 	struct epoch_tracker et;
1656d0390e05SGarrett Wollman 	struct inpcb *inp;
1657e06e816fSKevin Lo 	struct inpcbinfo *pcbinfo;
165875c13541SPoul-Henning Kamp 	struct sockaddr_in *sin;
1659e06e816fSKevin Lo 	int error;
1660d0390e05SGarrett Wollman 
1661a86e5c96SBjoern A. Zeeb 	pcbinfo = udp_get_inpcbinfo(so->so_proto->pr_protocol);
1662d0390e05SGarrett Wollman 	inp = sotoinpcb(so);
166314ba8addSRobert Watson 	KASSERT(inp != NULL, ("udp_connect: inp == NULL"));
1664f161d294SMark Johnston 
1665f161d294SMark Johnston 	sin = (struct sockaddr_in *)nam;
1666f161d294SMark Johnston 	if (sin->sin_family != AF_INET)
1667f161d294SMark Johnston 		return (EAFNOSUPPORT);
1668f161d294SMark Johnston 	if (sin->sin_len != sizeof(*sin))
1669f161d294SMark Johnston 		return (EINVAL);
1670f161d294SMark Johnston 
16718501a69cSRobert Watson 	INP_WLOCK(inp);
1672f76fcf6dSJeffrey Hsu 	if (inp->inp_faddr.s_addr != INADDR_ANY) {
16738501a69cSRobert Watson 		INP_WUNLOCK(inp);
16743329b236SRobert Watson 		return (EISCONN);
1675f76fcf6dSJeffrey Hsu 	}
1676b89e82ddSJamie Gritton 	error = prison_remote_ip4(td->td_ucred, &sin->sin_addr);
1677b89e82ddSJamie Gritton 	if (error != 0) {
1678413628a7SBjoern A. Zeeb 		INP_WUNLOCK(inp);
1679b89e82ddSJamie Gritton 		return (error);
1680413628a7SBjoern A. Zeeb 	}
1681c1604fe4SGleb Smirnoff 	NET_EPOCH_ENTER(et);
1682e06e816fSKevin Lo 	INP_HASH_WLOCK(pcbinfo);
1683*db0ac6deSCy Schubert 	error = in_pcbconnect(inp, nam, td->td_ucred, true);
1684e06e816fSKevin Lo 	INP_HASH_WUNLOCK(pcbinfo);
1685c1604fe4SGleb Smirnoff 	NET_EPOCH_EXIT(et);
16864cc20ab1SSeigo Tanimura 	if (error == 0)
1687df8bae1dSRodney W. Grimes 		soisconnected(so);
16888501a69cSRobert Watson 	INP_WUNLOCK(inp);
16893329b236SRobert Watson 	return (error);
1690df8bae1dSRodney W. Grimes }
1691d0390e05SGarrett Wollman 
1692bc725eafSRobert Watson static void
1693d0390e05SGarrett Wollman udp_detach(struct socket *so)
1694d0390e05SGarrett Wollman {
1695d0390e05SGarrett Wollman 	struct inpcb *inp;
1696e06e816fSKevin Lo 	struct inpcbinfo *pcbinfo;
16976a9148feSBjoern A. Zeeb 	struct udpcb *up;
1698d0390e05SGarrett Wollman 
1699a86e5c96SBjoern A. Zeeb 	pcbinfo = udp_get_inpcbinfo(so->so_proto->pr_protocol);
1700d0390e05SGarrett Wollman 	inp = sotoinpcb(so);
170114ba8addSRobert Watson 	KASSERT(inp != NULL, ("udp_detach: inp == NULL"));
1702a152f8a3SRobert Watson 	KASSERT(inp->inp_faddr.s_addr == INADDR_ANY,
1703a152f8a3SRobert Watson 	    ("udp_detach: not disconnected"));
17048501a69cSRobert Watson 	INP_WLOCK(inp);
17056a9148feSBjoern A. Zeeb 	up = intoudpcb(inp);
17066a9148feSBjoern A. Zeeb 	KASSERT(up != NULL, ("%s: up == NULL", __func__));
17076a9148feSBjoern A. Zeeb 	inp->inp_ppcb = NULL;
1708d0390e05SGarrett Wollman 	in_pcbdetach(inp);
170914ba8addSRobert Watson 	in_pcbfree(inp);
17106a9148feSBjoern A. Zeeb 	udp_discardcb(up);
1711d0390e05SGarrett Wollman }
1712d0390e05SGarrett Wollman 
1713d0390e05SGarrett Wollman static int
1714d0390e05SGarrett Wollman udp_disconnect(struct socket *so)
1715d0390e05SGarrett Wollman {
1716d0390e05SGarrett Wollman 	struct inpcb *inp;
1717e06e816fSKevin Lo 	struct inpcbinfo *pcbinfo;
1718d0390e05SGarrett Wollman 
1719a86e5c96SBjoern A. Zeeb 	pcbinfo = udp_get_inpcbinfo(so->so_proto->pr_protocol);
1720d0390e05SGarrett Wollman 	inp = sotoinpcb(so);
172114ba8addSRobert Watson 	KASSERT(inp != NULL, ("udp_disconnect: inp == NULL"));
17228501a69cSRobert Watson 	INP_WLOCK(inp);
1723f76fcf6dSJeffrey Hsu 	if (inp->inp_faddr.s_addr == INADDR_ANY) {
17248501a69cSRobert Watson 		INP_WUNLOCK(inp);
17253329b236SRobert Watson 		return (ENOTCONN);
1726f76fcf6dSJeffrey Hsu 	}
1727e06e816fSKevin Lo 	INP_HASH_WLOCK(pcbinfo);
1728df8bae1dSRodney W. Grimes 	in_pcbdisconnect(inp);
1729df8bae1dSRodney W. Grimes 	inp->inp_laddr.s_addr = INADDR_ANY;
1730e06e816fSKevin Lo 	INP_HASH_WUNLOCK(pcbinfo);
1731d45e4f99SMaxim Konovalov 	SOCK_LOCK(so);
1732d45e4f99SMaxim Konovalov 	so->so_state &= ~SS_ISCONNECTED;		/* XXX */
1733d45e4f99SMaxim Konovalov 	SOCK_UNLOCK(so);
17348501a69cSRobert Watson 	INP_WUNLOCK(inp);
17353329b236SRobert Watson 	return (0);
1736df8bae1dSRodney W. Grimes }
1737df8bae1dSRodney W. Grimes 
1738d0390e05SGarrett Wollman static int
173957bf258eSGarrett Wollman udp_send(struct socket *so, int flags, struct mbuf *m, struct sockaddr *addr,
1740b40ce416SJulian Elischer     struct mbuf *control, struct thread *td)
1741d0390e05SGarrett Wollman {
1742d0390e05SGarrett Wollman 	struct inpcb *inp;
1743f161d294SMark Johnston 	int error;
1744d0390e05SGarrett Wollman 
1745d0390e05SGarrett Wollman 	inp = sotoinpcb(so);
174614ba8addSRobert Watson 	KASSERT(inp != NULL, ("udp_send: inp == NULL"));
1747f161d294SMark Johnston 
1748f161d294SMark Johnston 	if (addr != NULL) {
1749f161d294SMark Johnston 		error = 0;
1750f161d294SMark Johnston 		if (addr->sa_family != AF_INET)
1751f161d294SMark Johnston 			error = EAFNOSUPPORT;
1752f161d294SMark Johnston 		else if (addr->sa_len != sizeof(struct sockaddr_in))
1753f161d294SMark Johnston 			error = EINVAL;
1754f161d294SMark Johnston 		if (__predict_false(error != 0)) {
1755f161d294SMark Johnston 			m_freem(control);
1756f161d294SMark Johnston 			m_freem(m);
1757f161d294SMark Johnston 			return (error);
1758f161d294SMark Johnston 		}
1759f161d294SMark Johnston 	}
1760a9839c4aSBjoern A. Zeeb 	return (udp_output(inp, m, addr, control, td, flags));
1761d0390e05SGarrett Wollman }
176279288c11SBjoern A. Zeeb #endif /* INET */
1763d0390e05SGarrett Wollman 
176476429de4SYoshinobu Inoue int
1765d0390e05SGarrett Wollman udp_shutdown(struct socket *so)
1766d0390e05SGarrett Wollman {
1767d0390e05SGarrett Wollman 	struct inpcb *inp;
1768d0390e05SGarrett Wollman 
1769d0390e05SGarrett Wollman 	inp = sotoinpcb(so);
177014ba8addSRobert Watson 	KASSERT(inp != NULL, ("udp_shutdown: inp == NULL"));
17718501a69cSRobert Watson 	INP_WLOCK(inp);
1772d0390e05SGarrett Wollman 	socantsendmore(so);
17738501a69cSRobert Watson 	INP_WUNLOCK(inp);
17743329b236SRobert Watson 	return (0);
1775d0390e05SGarrett Wollman }
1776d0390e05SGarrett Wollman 
177779288c11SBjoern A. Zeeb #ifdef INET
1778d0390e05SGarrett Wollman struct pr_usrreqs udp_usrreqs = {
1779756d52a1SPoul-Henning Kamp 	.pru_abort =		udp_abort,
1780756d52a1SPoul-Henning Kamp 	.pru_attach =		udp_attach,
1781756d52a1SPoul-Henning Kamp 	.pru_bind =		udp_bind,
1782756d52a1SPoul-Henning Kamp 	.pru_connect =		udp_connect,
1783756d52a1SPoul-Henning Kamp 	.pru_control =		in_control,
1784756d52a1SPoul-Henning Kamp 	.pru_detach =		udp_detach,
1785756d52a1SPoul-Henning Kamp 	.pru_disconnect =	udp_disconnect,
178654d642bbSRobert Watson 	.pru_peeraddr =		in_getpeeraddr,
1787756d52a1SPoul-Henning Kamp 	.pru_send =		udp_send,
17885df3e839SRobert Watson 	.pru_soreceive =	soreceive_dgram,
178959b8854eSRobert Watson 	.pru_sosend =		sosend_dgram,
1790756d52a1SPoul-Henning Kamp 	.pru_shutdown =		udp_shutdown,
179154d642bbSRobert Watson 	.pru_sockaddr =		in_getsockaddr,
1792a152f8a3SRobert Watson 	.pru_sosetlabel =	in_pcbsosetlabel,
1793a152f8a3SRobert Watson 	.pru_close =		udp_close,
1794d0390e05SGarrett Wollman };
179579288c11SBjoern A. Zeeb #endif /* INET */
1796