xref: /freebsd/sys/netinet/udp_usrreq.c (revision 89128ff3e42196c8b41a3c59f7a2d95c0a4ac0ee)
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 
201*89128ff3SGleb Smirnoff static void
202*89128ff3SGleb Smirnoff udp_init(void *arg __unused)
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 	 */
212db0ac6deSCy Schubert 	in_pcbinfo_init(&V_udbinfo, "udp", UDBHASHSIZE, UDBHASHSIZE,
213db0ac6deSCy 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 
221*89128ff3SGleb Smirnoff 	/* Additional pcbinfo for UDP-Lite */
222db0ac6deSCy Schubert 	in_pcbinfo_init(&V_ulitecbinfo, "udplite", UDBHASHSIZE,
223db0ac6deSCy Schubert 	    UDBHASHSIZE, "udplite_inpcb", udplite_inpcb_init);
224e06e816fSKevin Lo }
225*89128ff3SGleb Smirnoff VNET_SYSINIT(udp_init, SI_SUB_PROTO_DOMAIN, SI_ORDER_THIRD, udp_init, NULL);
226e06e816fSKevin Lo 
227315e3e38SRobert Watson /*
228315e3e38SRobert Watson  * Kernel module interface for updating udpstat.  The argument is an index
229315e3e38SRobert Watson  * into udpstat treated as an array of u_long.  While this encodes the
230315e3e38SRobert Watson  * general layout of udpstat into the caller, it doesn't encode its location,
231315e3e38SRobert Watson  * so that future changes to add, for example, per-CPU stats support won't
232315e3e38SRobert Watson  * cause binary compatibility problems for kernel modules.
233315e3e38SRobert Watson  */
234315e3e38SRobert Watson void
235315e3e38SRobert Watson kmod_udpstat_inc(int statnum)
236315e3e38SRobert Watson {
237315e3e38SRobert Watson 
2385b7cb97cSAndrey V. Elsukov 	counter_u64_add(VNET(udpstat)[statnum], 1);
239315e3e38SRobert Watson }
240315e3e38SRobert Watson 
2416a9148feSBjoern A. Zeeb int
2426a9148feSBjoern A. Zeeb udp_newudpcb(struct inpcb *inp)
2436a9148feSBjoern A. Zeeb {
2446a9148feSBjoern A. Zeeb 	struct udpcb *up;
2456a9148feSBjoern A. Zeeb 
2466a9148feSBjoern A. Zeeb 	up = uma_zalloc(V_udpcb_zone, M_NOWAIT | M_ZERO);
2476a9148feSBjoern A. Zeeb 	if (up == NULL)
2486a9148feSBjoern A. Zeeb 		return (ENOBUFS);
2496a9148feSBjoern A. Zeeb 	inp->inp_ppcb = up;
2506a9148feSBjoern A. Zeeb 	return (0);
2516a9148feSBjoern A. Zeeb }
2526a9148feSBjoern A. Zeeb 
2536a9148feSBjoern A. Zeeb void
2546a9148feSBjoern A. Zeeb udp_discardcb(struct udpcb *up)
2556a9148feSBjoern A. Zeeb {
2566a9148feSBjoern A. Zeeb 
2576a9148feSBjoern A. Zeeb 	uma_zfree(V_udpcb_zone, up);
2586a9148feSBjoern A. Zeeb }
2596a9148feSBjoern A. Zeeb 
260bc29160dSMarko Zec #ifdef VIMAGE
2613f58662dSBjoern A. Zeeb static void
2623f58662dSBjoern A. Zeeb udp_destroy(void *unused __unused)
263bc29160dSMarko Zec {
264bc29160dSMarko Zec 
2659bcd427bSRobert Watson 	in_pcbinfo_destroy(&V_udbinfo);
266391dab1cSBjoern A. Zeeb 	uma_zdestroy(V_udpcb_zone);
267bc29160dSMarko Zec }
2683f58662dSBjoern A. Zeeb VNET_SYSUNINIT(udp, SI_SUB_PROTO_DOMAIN, SI_ORDER_FOURTH, udp_destroy, NULL);
269e06e816fSKevin Lo 
2703f58662dSBjoern A. Zeeb static void
2713f58662dSBjoern A. Zeeb udplite_destroy(void *unused __unused)
272e06e816fSKevin Lo {
273e06e816fSKevin Lo 
274e06e816fSKevin Lo 	in_pcbinfo_destroy(&V_ulitecbinfo);
275e06e816fSKevin Lo }
2763f58662dSBjoern A. Zeeb VNET_SYSUNINIT(udplite, SI_SUB_PROTO_DOMAIN, SI_ORDER_FOURTH, udplite_destroy,
2773f58662dSBjoern A. Zeeb     NULL);
278bc29160dSMarko Zec #endif
279bc29160dSMarko Zec 
28079288c11SBjoern A. Zeeb #ifdef INET
28143bbb6aaSRobert Watson /*
28243bbb6aaSRobert Watson  * Subroutine of udp_input(), which appends the provided mbuf chain to the
28343bbb6aaSRobert Watson  * passed pcb/socket.  The caller must provide a sockaddr_in via udp_in that
28443bbb6aaSRobert Watson  * contains the source address.  If the socket ends up being an IPv6 socket,
28543bbb6aaSRobert Watson  * udp_append() will convert to a sockaddr_in6 before passing the address
28643bbb6aaSRobert Watson  * into the socket code.
287c0d1be08SRandall Stewart  *
288c0d1be08SRandall Stewart  * In the normal case udp_append() will return 0, indicating that you
289c0d1be08SRandall Stewart  * must unlock the inp. However if a tunneling protocol is in place we increment
290c0d1be08SRandall Stewart  * the inpcb refcnt and unlock the inp, on return from the tunneling protocol we
291c0d1be08SRandall Stewart  * then decrement the reference count. If the inp_rele returns 1, indicating the
292c0d1be08SRandall Stewart  * inp is gone, we return that to the caller to tell them *not* to unlock
293c0d1be08SRandall Stewart  * the inp. In the case of multi-cast this will cause the distribution
294c0d1be08SRandall Stewart  * to stop (though most tunneling protocols known currently do *not* use
295c0d1be08SRandall Stewart  * multicast).
29643bbb6aaSRobert Watson  */
297c0d1be08SRandall Stewart static int
29843bbb6aaSRobert Watson udp_append(struct inpcb *inp, struct ip *ip, struct mbuf *n, int off,
29943bbb6aaSRobert Watson     struct sockaddr_in *udp_in)
30043bbb6aaSRobert Watson {
30143bbb6aaSRobert Watson 	struct sockaddr *append_sa;
30243bbb6aaSRobert Watson 	struct socket *so;
303dce33a45SErmal Luçi 	struct mbuf *tmpopts, *opts = NULL;
30443bbb6aaSRobert Watson #ifdef INET6
30543bbb6aaSRobert Watson 	struct sockaddr_in6 udp_in6;
30643bbb6aaSRobert Watson #endif
3077b495c44SVANHULLEBUS Yvan 	struct udpcb *up;
30843bbb6aaSRobert Watson 
309fa046d87SRobert Watson 	INP_LOCK_ASSERT(inp);
31043bbb6aaSRobert Watson 
31179bb84fbSEdward Tomasz Napierala 	/*
31279bb84fbSEdward Tomasz Napierala 	 * Engage the tunneling protocol.
31379bb84fbSEdward Tomasz Napierala 	 */
31479bb84fbSEdward Tomasz Napierala 	up = intoudpcb(inp);
31579bb84fbSEdward Tomasz Napierala 	if (up->u_tun_func != NULL) {
316c0d1be08SRandall Stewart 		in_pcbref(inp);
317c0d1be08SRandall Stewart 		INP_RUNLOCK(inp);
318dce33a45SErmal Luçi 		(*up->u_tun_func)(n, off, inp, (struct sockaddr *)&udp_in[0],
31981d3ec17SBryan Venteicher 		    up->u_tun_ctx);
320c0d1be08SRandall Stewart 		INP_RLOCK(inp);
321c0d1be08SRandall Stewart 		return (in_pcbrele_rlocked(inp));
32279bb84fbSEdward Tomasz Napierala 	}
32379bb84fbSEdward Tomasz Napierala 
32479bb84fbSEdward Tomasz Napierala 	off += sizeof(struct udphdr);
32579bb84fbSEdward Tomasz Napierala 
326fcf59617SAndrey V. Elsukov #if defined(IPSEC) || defined(IPSEC_SUPPORT)
32743bbb6aaSRobert Watson 	/* Check AH/ESP integrity. */
328fcf59617SAndrey V. Elsukov 	if (IPSEC_ENABLED(ipv4) &&
329fcf59617SAndrey V. Elsukov 	    IPSEC_CHECK_POLICY(ipv4, n, inp) != 0) {
33043bbb6aaSRobert Watson 		m_freem(n);
331c0d1be08SRandall Stewart 		return (0);
33243bbb6aaSRobert Watson 	}
333fcf59617SAndrey V. Elsukov 	if (up->u_flags & UF_ESPINUDP) {/* IPSec UDP encaps. */
334fcf59617SAndrey V. Elsukov 		if (IPSEC_ENABLED(ipv4) &&
335fcf59617SAndrey V. Elsukov 		    UDPENCAP_INPUT(n, off, AF_INET) != 0)
336fcf59617SAndrey V. Elsukov 			return (0);	/* Consumed. */
3377b495c44SVANHULLEBUS Yvan 	}
33843bbb6aaSRobert Watson #endif /* IPSEC */
33943bbb6aaSRobert Watson #ifdef MAC
34030d239bcSRobert Watson 	if (mac_inpcb_check_deliver(inp, n) != 0) {
34143bbb6aaSRobert Watson 		m_freem(n);
342c0d1be08SRandall Stewart 		return (0);
34343bbb6aaSRobert Watson 	}
34479288c11SBjoern A. Zeeb #endif /* MAC */
34543bbb6aaSRobert Watson 	if (inp->inp_flags & INP_CONTROLOPTS ||
34643bbb6aaSRobert Watson 	    inp->inp_socket->so_options & (SO_TIMESTAMP | SO_BINTIME)) {
34743bbb6aaSRobert Watson #ifdef INET6
3489a38ba81SBjoern A. Zeeb 		if (inp->inp_vflag & INP_IPV6)
34948d48eb9SBjoern A. Zeeb 			(void)ip6_savecontrol_v4(inp, n, &opts, NULL);
3509a38ba81SBjoern A. Zeeb 		else
35179288c11SBjoern A. Zeeb #endif /* INET6 */
35243bbb6aaSRobert Watson 			ip_savecontrol(inp, &opts, ip, n);
35343bbb6aaSRobert Watson 	}
354dce33a45SErmal Luçi 	if ((inp->inp_vflag & INP_IPV4) && (inp->inp_flags2 & INP_ORIGDSTADDR)) {
355dce33a45SErmal Luçi 		tmpopts = sbcreatecontrol((caddr_t)&udp_in[1],
356dce33a45SErmal Luçi 			sizeof(struct sockaddr_in), IP_ORIGDSTADDR, IPPROTO_IP);
357dce33a45SErmal Luçi 		if (tmpopts) {
358dce33a45SErmal Luçi 			if (opts) {
359dce33a45SErmal Luçi 				tmpopts->m_next = opts;
360dce33a45SErmal Luçi 				opts = tmpopts;
361dce33a45SErmal Luçi 			} else
362dce33a45SErmal Luçi 				opts = tmpopts;
363dce33a45SErmal Luçi 		}
364dce33a45SErmal Luçi 	}
36543bbb6aaSRobert Watson #ifdef INET6
36643bbb6aaSRobert Watson 	if (inp->inp_vflag & INP_IPV6) {
36743bbb6aaSRobert Watson 		bzero(&udp_in6, sizeof(udp_in6));
36843bbb6aaSRobert Watson 		udp_in6.sin6_len = sizeof(udp_in6);
36943bbb6aaSRobert Watson 		udp_in6.sin6_family = AF_INET6;
370dce33a45SErmal Luçi 		in6_sin_2_v4mapsin6(&udp_in[0], &udp_in6);
37143bbb6aaSRobert Watson 		append_sa = (struct sockaddr *)&udp_in6;
37243bbb6aaSRobert Watson 	} else
37379288c11SBjoern A. Zeeb #endif /* INET6 */
374dce33a45SErmal Luçi 		append_sa = (struct sockaddr *)&udp_in[0];
37543bbb6aaSRobert Watson 	m_adj(n, off);
37643bbb6aaSRobert Watson 
37743bbb6aaSRobert Watson 	so = inp->inp_socket;
37843bbb6aaSRobert Watson 	SOCKBUF_LOCK(&so->so_rcv);
37943bbb6aaSRobert Watson 	if (sbappendaddr_locked(&so->so_rcv, append_sa, n, opts) == 0) {
380a61c24ddSKonstantin Kukushkin 		soroverflow_locked(so);
38143bbb6aaSRobert Watson 		m_freem(n);
38243bbb6aaSRobert Watson 		if (opts)
38343bbb6aaSRobert Watson 			m_freem(opts);
384026decb8SRobert Watson 		UDPSTAT_INC(udps_fullsock);
38543bbb6aaSRobert Watson 	} else
38643bbb6aaSRobert Watson 		sorwakeup_locked(so);
387c0d1be08SRandall Stewart 	return (0);
38843bbb6aaSRobert Watson }
38943bbb6aaSRobert Watson 
390db0ac6deSCy Schubert static bool
391db0ac6deSCy Schubert udp_multi_match(const struct inpcb *inp, void *v)
392db0ac6deSCy Schubert {
393db0ac6deSCy Schubert 	struct ip *ip = v;
394db0ac6deSCy Schubert 	struct udphdr *uh = (struct udphdr *)(ip + 1);
395db0ac6deSCy Schubert 
396db0ac6deSCy Schubert 	if (inp->inp_lport != uh->uh_dport)
397db0ac6deSCy Schubert 		return (false);
398db0ac6deSCy Schubert #ifdef INET6
399db0ac6deSCy Schubert 	if ((inp->inp_vflag & INP_IPV4) == 0)
400db0ac6deSCy Schubert 		return (false);
401db0ac6deSCy Schubert #endif
402db0ac6deSCy Schubert 	if (inp->inp_laddr.s_addr != INADDR_ANY &&
403db0ac6deSCy Schubert 	    inp->inp_laddr.s_addr != ip->ip_dst.s_addr)
404db0ac6deSCy Schubert 		return (false);
405db0ac6deSCy Schubert 	if (inp->inp_faddr.s_addr != INADDR_ANY &&
406db0ac6deSCy Schubert 	    inp->inp_faddr.s_addr != ip->ip_src.s_addr)
407db0ac6deSCy Schubert 		return (false);
408db0ac6deSCy Schubert 	if (inp->inp_fport != 0 &&
409db0ac6deSCy Schubert 	    inp->inp_fport != uh->uh_sport)
410db0ac6deSCy Schubert 		return (false);
411db0ac6deSCy Schubert 
412db0ac6deSCy Schubert 	return (true);
413db0ac6deSCy Schubert }
414db0ac6deSCy Schubert 
415db0ac6deSCy Schubert static int
416db0ac6deSCy Schubert udp_multi_input(struct mbuf *m, int proto, struct sockaddr_in *udp_in)
417db0ac6deSCy Schubert {
418db0ac6deSCy Schubert 	struct ip *ip = mtod(m, struct ip *);
419db0ac6deSCy Schubert 	struct inpcb_iterator inpi = INP_ITERATOR(udp_get_inpcbinfo(proto),
420db0ac6deSCy Schubert 	    INPLOOKUP_RLOCKPCB, udp_multi_match, ip);
421bd1d0850SGleb Smirnoff #ifdef KDTRACE_HOOKS
422db0ac6deSCy Schubert 	struct udphdr *uh = (struct udphdr *)(ip + 1);
423bd1d0850SGleb Smirnoff #endif
424db0ac6deSCy Schubert 	struct inpcb *inp;
425db0ac6deSCy Schubert 	struct mbuf *n;
426db0ac6deSCy Schubert 	int appends = 0;
427db0ac6deSCy Schubert 
428db0ac6deSCy Schubert 	MPASS(ip->ip_hl == sizeof(struct ip) >> 2);
429db0ac6deSCy Schubert 
430db0ac6deSCy Schubert 	while ((inp = inp_next(&inpi)) != NULL) {
431db0ac6deSCy Schubert 		/*
432db0ac6deSCy Schubert 		 * XXXRW: Because we weren't holding either the inpcb
433db0ac6deSCy Schubert 		 * or the hash lock when we checked for a match
434db0ac6deSCy Schubert 		 * before, we should probably recheck now that the
435db0ac6deSCy Schubert 		 * inpcb lock is held.
436db0ac6deSCy Schubert 		 */
437db0ac6deSCy Schubert 		/*
438db0ac6deSCy Schubert 		 * Handle socket delivery policy for any-source
439db0ac6deSCy Schubert 		 * and source-specific multicast. [RFC3678]
440db0ac6deSCy Schubert 		 */
441db0ac6deSCy Schubert 		if (IN_MULTICAST(ntohl(ip->ip_dst.s_addr))) {
442db0ac6deSCy Schubert 			struct ip_moptions	*imo;
443db0ac6deSCy Schubert 			struct sockaddr_in	 group;
444db0ac6deSCy Schubert 			int			 blocked;
445db0ac6deSCy Schubert 
446db0ac6deSCy Schubert 			imo = inp->inp_moptions;
447db0ac6deSCy Schubert 			if (imo == NULL)
448db0ac6deSCy Schubert 				continue;
449db0ac6deSCy Schubert 			bzero(&group, sizeof(struct sockaddr_in));
450db0ac6deSCy Schubert 			group.sin_len = sizeof(struct sockaddr_in);
451db0ac6deSCy Schubert 			group.sin_family = AF_INET;
452db0ac6deSCy Schubert 			group.sin_addr = ip->ip_dst;
453db0ac6deSCy Schubert 
454db0ac6deSCy Schubert 			blocked = imo_multi_filter(imo, m->m_pkthdr.rcvif,
455db0ac6deSCy Schubert 				(struct sockaddr *)&group,
456db0ac6deSCy Schubert 				(struct sockaddr *)&udp_in[0]);
457db0ac6deSCy Schubert 			if (blocked != MCAST_PASS) {
458db0ac6deSCy Schubert 				if (blocked == MCAST_NOTGMEMBER)
459db0ac6deSCy Schubert 					IPSTAT_INC(ips_notmember);
460db0ac6deSCy Schubert 				if (blocked == MCAST_NOTSMEMBER ||
461db0ac6deSCy Schubert 				    blocked == MCAST_MUTED)
462db0ac6deSCy Schubert 					UDPSTAT_INC(udps_filtermcast);
463db0ac6deSCy Schubert 				continue;
464db0ac6deSCy Schubert 			}
465db0ac6deSCy Schubert 		}
466db0ac6deSCy Schubert 		if ((n = m_copym(m, 0, M_COPYALL, M_NOWAIT)) != NULL) {
467db0ac6deSCy Schubert 			if (proto == IPPROTO_UDPLITE)
468db0ac6deSCy Schubert 				UDPLITE_PROBE(receive, NULL, inp, ip, inp, uh);
469db0ac6deSCy Schubert 			else
470db0ac6deSCy Schubert 				UDP_PROBE(receive, NULL, inp, ip, inp, uh);
471db0ac6deSCy Schubert 			if (udp_append(inp, ip, n, sizeof(struct ip), udp_in)) {
472db0ac6deSCy Schubert 				INP_RUNLOCK(inp);
473db0ac6deSCy Schubert 				break;
474db0ac6deSCy Schubert 			} else
475db0ac6deSCy Schubert 				appends++;
476db0ac6deSCy Schubert 		}
477db0ac6deSCy Schubert 		/*
478db0ac6deSCy Schubert 		 * Don't look for additional matches if this one does
479db0ac6deSCy Schubert 		 * not have either the SO_REUSEPORT or SO_REUSEADDR
480db0ac6deSCy Schubert 		 * socket options set.  This heuristic avoids
481db0ac6deSCy Schubert 		 * searching through all pcbs in the common case of a
482db0ac6deSCy Schubert 		 * non-shared port.  It assumes that an application
483db0ac6deSCy Schubert 		 * will never clear these options after setting them.
484db0ac6deSCy Schubert 		 */
485db0ac6deSCy Schubert 		if ((inp->inp_socket->so_options &
486db0ac6deSCy Schubert 		    (SO_REUSEPORT|SO_REUSEPORT_LB|SO_REUSEADDR)) == 0) {
487db0ac6deSCy Schubert 			INP_RUNLOCK(inp);
488db0ac6deSCy Schubert 			break;
489db0ac6deSCy Schubert 		}
490db0ac6deSCy Schubert 	}
491db0ac6deSCy Schubert 
492db0ac6deSCy Schubert 	if (appends == 0) {
493db0ac6deSCy Schubert 		/*
494db0ac6deSCy Schubert 		 * No matching pcb found; discard datagram.  (No need
495db0ac6deSCy Schubert 		 * to send an ICMP Port Unreachable for a broadcast
496db0ac6deSCy Schubert 		 * or multicast datgram.)
497db0ac6deSCy Schubert 		 */
498db0ac6deSCy Schubert 		UDPSTAT_INC(udps_noport);
499db0ac6deSCy Schubert 		if (IN_MULTICAST(ntohl(ip->ip_dst.s_addr)))
500db0ac6deSCy Schubert 			UDPSTAT_INC(udps_noportmcast);
501db0ac6deSCy Schubert 		else
502db0ac6deSCy Schubert 			UDPSTAT_INC(udps_noportbcast);
503db0ac6deSCy Schubert 	}
504014f98b1SMark Johnston 	m_freem(m);
505db0ac6deSCy Schubert 
506db0ac6deSCy Schubert 	return (IPPROTO_DONE);
507db0ac6deSCy Schubert }
508db0ac6deSCy Schubert 
5098f5a8818SKevin Lo int
5108f5a8818SKevin Lo udp_input(struct mbuf **mp, int *offp, int proto)
511df8bae1dSRodney W. Grimes {
5123329b236SRobert Watson 	struct ip *ip;
5133329b236SRobert Watson 	struct udphdr *uh;
51471498f30SBruce M Simpson 	struct ifnet *ifp;
5153329b236SRobert Watson 	struct inpcb *inp;
5168f134647SGleb Smirnoff 	uint16_t len, ip_len;
517e06e816fSKevin Lo 	struct inpcbinfo *pcbinfo;
518dce33a45SErmal Luçi 	struct sockaddr_in udp_in[2];
5198f5a8818SKevin Lo 	struct mbuf *m;
5200b4ae859SGleb Smirnoff 	struct m_tag *fwd_tag;
5218f5a8818SKevin Lo 	int cscov_partial, iphlen;
522df8bae1dSRodney W. Grimes 
5238f5a8818SKevin Lo 	m = *mp;
5248f5a8818SKevin Lo 	iphlen = *offp;
52571498f30SBruce M Simpson 	ifp = m->m_pkthdr.rcvif;
5268f5a8818SKevin Lo 	*mp = NULL;
527026decb8SRobert Watson 	UDPSTAT_INC(udps_ipackets);
528df8bae1dSRodney W. Grimes 
529df8bae1dSRodney W. Grimes 	/*
5303329b236SRobert Watson 	 * Strip IP options, if any; should skip this, make available to
5313329b236SRobert Watson 	 * user, and use on returned packets, but we don't yet have a way to
5323329b236SRobert Watson 	 * check the checksum with options still present.
533df8bae1dSRodney W. Grimes 	 */
534df8bae1dSRodney W. Grimes 	if (iphlen > sizeof (struct ip)) {
535105bd211SGleb Smirnoff 		ip_stripoptions(m);
536df8bae1dSRodney W. Grimes 		iphlen = sizeof(struct ip);
537df8bae1dSRodney W. Grimes 	}
538df8bae1dSRodney W. Grimes 
539df8bae1dSRodney W. Grimes 	/*
540df8bae1dSRodney W. Grimes 	 * Get IP and UDP header together in first mbuf.
541df8bae1dSRodney W. Grimes 	 */
542df8bae1dSRodney W. Grimes 	if (m->m_len < iphlen + sizeof(struct udphdr)) {
543d1b18731SKevin Lo 		if ((m = m_pullup(m, iphlen + sizeof(struct udphdr))) == NULL) {
544026decb8SRobert Watson 			UDPSTAT_INC(udps_hdrops);
5458f5a8818SKevin Lo 			return (IPPROTO_DONE);
546df8bae1dSRodney W. Grimes 		}
547df8bae1dSRodney W. Grimes 	}
548503f4e47SBjoern A. Zeeb 	ip = mtod(m, struct ip *);
549df8bae1dSRodney W. Grimes 	uh = (struct udphdr *)((caddr_t)ip + iphlen);
5508f5a8818SKevin Lo 	cscov_partial = (proto == IPPROTO_UDPLITE) ? 1 : 0;
551df8bae1dSRodney W. Grimes 
5523329b236SRobert Watson 	/*
5533329b236SRobert Watson 	 * Destination port of 0 is illegal, based on RFC768.
5543329b236SRobert Watson 	 */
555686cdd19SJun-ichiro itojun Hagino 	if (uh->uh_dport == 0)
556f76fcf6dSJeffrey Hsu 		goto badunlocked;
557686cdd19SJun-ichiro itojun Hagino 
558df8bae1dSRodney W. Grimes 	/*
5593329b236SRobert Watson 	 * Construct sockaddr format source address.  Stuff source address
5603329b236SRobert Watson 	 * and datagram in user buffer.
561b9234fafSSam Leffler 	 */
562dce33a45SErmal Luçi 	bzero(&udp_in[0], sizeof(struct sockaddr_in) * 2);
563dce33a45SErmal Luçi 	udp_in[0].sin_len = sizeof(struct sockaddr_in);
564dce33a45SErmal Luçi 	udp_in[0].sin_family = AF_INET;
565dce33a45SErmal Luçi 	udp_in[0].sin_port = uh->uh_sport;
566dce33a45SErmal Luçi 	udp_in[0].sin_addr = ip->ip_src;
567dce33a45SErmal Luçi 	udp_in[1].sin_len = sizeof(struct sockaddr_in);
568dce33a45SErmal Luçi 	udp_in[1].sin_family = AF_INET;
569dce33a45SErmal Luçi 	udp_in[1].sin_port = uh->uh_dport;
570dce33a45SErmal Luçi 	udp_in[1].sin_addr = ip->ip_dst;
571b9234fafSSam Leffler 
572b9234fafSSam Leffler 	/*
573af1ee11dSRobert Watson 	 * Make mbuf data length reflect UDP length.  If not enough data to
574af1ee11dSRobert Watson 	 * reflect UDP length, drop.
575df8bae1dSRodney W. Grimes 	 */
576df8bae1dSRodney W. Grimes 	len = ntohs((u_short)uh->uh_ulen);
5778ad458a4SGleb Smirnoff 	ip_len = ntohs(ip->ip_len) - iphlen;
5780f4a0366SMichael Tuexen 	if (proto == IPPROTO_UDPLITE && (len == 0 || len == ip_len)) {
579e06e816fSKevin Lo 		/* Zero means checksum over the complete packet. */
5800f4a0366SMichael Tuexen 		if (len == 0)
581e06e816fSKevin Lo 			len = ip_len;
582e06e816fSKevin Lo 		cscov_partial = 0;
583e06e816fSKevin Lo 	}
5848f134647SGleb Smirnoff 	if (ip_len != len) {
5858f134647SGleb Smirnoff 		if (len > ip_len || len < sizeof(struct udphdr)) {
586026decb8SRobert Watson 			UDPSTAT_INC(udps_badlen);
587f76fcf6dSJeffrey Hsu 			goto badunlocked;
588df8bae1dSRodney W. Grimes 		}
5898f5a8818SKevin Lo 		if (proto == IPPROTO_UDP)
5908f134647SGleb Smirnoff 			m_adj(m, len - ip_len);
591df8bae1dSRodney W. Grimes 	}
5923329b236SRobert Watson 
593df8bae1dSRodney W. Grimes 	/*
594df8bae1dSRodney W. Grimes 	 * Checksum extended UDP header and data.
595df8bae1dSRodney W. Grimes 	 */
5966dfab5b1SGarrett Wollman 	if (uh->uh_sum) {
59739629c92SDavid Malone 		u_short uh_sum;
59839629c92SDavid Malone 
599e06e816fSKevin Lo 		if ((m->m_pkthdr.csum_flags & CSUM_DATA_VALID) &&
600e06e816fSKevin Lo 		    !cscov_partial) {
601db4f9cc7SJonathan Lemon 			if (m->m_pkthdr.csum_flags & CSUM_PSEUDO_HDR)
60239629c92SDavid Malone 				uh_sum = m->m_pkthdr.csum_data;
603db4f9cc7SJonathan Lemon 			else
60439629c92SDavid Malone 				uh_sum = in_pseudo(ip->ip_src.s_addr,
605506f4949SRuslan Ermilov 				    ip->ip_dst.s_addr, htonl((u_short)len +
6068f5a8818SKevin Lo 				    m->m_pkthdr.csum_data + proto));
60739629c92SDavid Malone 			uh_sum ^= 0xffff;
608db4f9cc7SJonathan Lemon 		} else {
6093358df29SGleb Smirnoff 			char b[offsetof(struct ipovly, ih_src)];
6103358df29SGleb Smirnoff 			struct ipovly *ipov = (struct ipovly *)ip;
611af1ee11dSRobert Watson 
6123358df29SGleb Smirnoff 			bcopy(ipov, b, sizeof(b));
6133358df29SGleb Smirnoff 			bzero(ipov, sizeof(ipov->ih_x1));
6143358df29SGleb Smirnoff 			ipov->ih_len = (proto == IPPROTO_UDP) ?
615e06e816fSKevin Lo 			    uh->uh_ulen : htons(ip_len);
61639629c92SDavid Malone 			uh_sum = in_cksum(m, len + sizeof (struct ip));
6173358df29SGleb Smirnoff 			bcopy(b, ipov, sizeof(b));
618db4f9cc7SJonathan Lemon 		}
61939629c92SDavid Malone 		if (uh_sum) {
620026decb8SRobert Watson 			UDPSTAT_INC(udps_badsum);
621df8bae1dSRodney W. Grimes 			m_freem(m);
6228f5a8818SKevin Lo 			return (IPPROTO_DONE);
623df8bae1dSRodney W. Grimes 		}
624c6d81a34SMichael Tuexen 	} else {
625c6d81a34SMichael Tuexen 		if (proto == IPPROTO_UDP) {
626026decb8SRobert Watson 			UDPSTAT_INC(udps_nosum);
627c6d81a34SMichael Tuexen 		} else {
628c6d81a34SMichael Tuexen 			/* UDPLite requires a checksum */
629c6d81a34SMichael Tuexen 			/* XXX: What is the right UDPLite MIB counter here? */
630c6d81a34SMichael Tuexen 			m_freem(m);
631c6d81a34SMichael Tuexen 			return (IPPROTO_DONE);
632c6d81a34SMichael Tuexen 		}
633c6d81a34SMichael Tuexen 	}
634df8bae1dSRodney W. Grimes 
635266f97b5SCy Schubert 	if (IN_MULTICAST(ntohl(ip->ip_dst.s_addr)) ||
636db0ac6deSCy Schubert 	    in_broadcast(ip->ip_dst, ifp))
637db0ac6deSCy Schubert 		return (udp_multi_input(m, proto, udp_in));
638266f97b5SCy Schubert 
639db0ac6deSCy Schubert 	pcbinfo = udp_get_inpcbinfo(proto);
6403329b236SRobert Watson 
641df8bae1dSRodney W. Grimes 	/*
6426d6a026bSDavid Greenman 	 * Locate pcb for datagram.
643db0ac6deSCy Schubert 	 *
6448a006adbSBjoern A. Zeeb 	 * Grab info from PACKET_TAG_IPFORWARD tag prepended to the chain.
6458a006adbSBjoern A. Zeeb 	 */
646ffdbf9daSAndrey V. Elsukov 	if ((m->m_flags & M_IP_NEXTHOP) &&
647c1de64a4SAndrey V. Elsukov 	    (fwd_tag = m_tag_find(m, PACKET_TAG_IPFORWARD, NULL)) != NULL) {
6488a006adbSBjoern A. Zeeb 		struct sockaddr_in *next_hop;
6498a006adbSBjoern A. Zeeb 
6508a006adbSBjoern A. Zeeb 		next_hop = (struct sockaddr_in *)(fwd_tag + 1);
6518a006adbSBjoern A. Zeeb 
6528a006adbSBjoern A. Zeeb 		/*
6538a006adbSBjoern A. Zeeb 		 * Transparently forwarded. Pretend to be the destination.
6548a006adbSBjoern A. Zeeb 		 * Already got one like this?
6558a006adbSBjoern A. Zeeb 		 */
656e06e816fSKevin Lo 		inp = in_pcblookup_mbuf(pcbinfo, ip->ip_src, uh->uh_sport,
6578a006adbSBjoern A. Zeeb 		    ip->ip_dst, uh->uh_dport, INPLOOKUP_RLOCKPCB, ifp, m);
6588a006adbSBjoern A. Zeeb 		if (!inp) {
6598a006adbSBjoern A. Zeeb 			/*
6608a006adbSBjoern A. Zeeb 			 * It's new.  Try to find the ambushing socket.
6618a006adbSBjoern A. Zeeb 			 * Because we've rewritten the destination address,
6628a006adbSBjoern A. Zeeb 			 * any hardware-generated hash is ignored.
6638a006adbSBjoern A. Zeeb 			 */
664e06e816fSKevin Lo 			inp = in_pcblookup(pcbinfo, ip->ip_src,
6658a006adbSBjoern A. Zeeb 			    uh->uh_sport, next_hop->sin_addr,
6668a006adbSBjoern A. Zeeb 			    next_hop->sin_port ? htons(next_hop->sin_port) :
6678a006adbSBjoern A. Zeeb 			    uh->uh_dport, INPLOOKUP_WILDCARD |
6688a006adbSBjoern A. Zeeb 			    INPLOOKUP_RLOCKPCB, ifp);
6698a006adbSBjoern A. Zeeb 		}
6708a006adbSBjoern A. Zeeb 		/* Remove the tag from the packet. We don't need it anymore. */
6718a006adbSBjoern A. Zeeb 		m_tag_delete(m, fwd_tag);
672ffdbf9daSAndrey V. Elsukov 		m->m_flags &= ~M_IP_NEXTHOP;
6738a006adbSBjoern A. Zeeb 	} else
674e06e816fSKevin Lo 		inp = in_pcblookup_mbuf(pcbinfo, ip->ip_src, uh->uh_sport,
6758a006adbSBjoern A. Zeeb 		    ip->ip_dst, uh->uh_dport, INPLOOKUP_WILDCARD |
6768a006adbSBjoern A. Zeeb 		    INPLOOKUP_RLOCKPCB, ifp, m);
67715bd2b43SDavid Greenman 	if (inp == NULL) {
678334fc582SBjoern A. Zeeb 		if (V_udp_log_in_vain) {
679edf0313bSEric van Gyzen 			char src[INET_ADDRSTRLEN];
680edf0313bSEric van Gyzen 			char dst[INET_ADDRSTRLEN];
68175cfc95fSAndrey A. Chernov 
682592071e8SBruce Evans 			log(LOG_INFO,
683592071e8SBruce Evans 			    "Connection attempt to UDP %s:%d from %s:%d\n",
684edf0313bSEric van Gyzen 			    inet_ntoa_r(ip->ip_dst, dst), ntohs(uh->uh_dport),
685edf0313bSEric van Gyzen 			    inet_ntoa_r(ip->ip_src, src), ntohs(uh->uh_sport));
68675cfc95fSAndrey A. Chernov 		}
6877bda9663SMichael Tuexen 		if (proto == IPPROTO_UDPLITE)
6887bda9663SMichael Tuexen 			UDPLITE_PROBE(receive, NULL, NULL, ip, NULL, uh);
6897bda9663SMichael Tuexen 		else
690e1526d5aSMichael Tuexen 			UDP_PROBE(receive, NULL, NULL, ip, NULL, uh);
691026decb8SRobert Watson 		UDPSTAT_INC(udps_noport);
692df8bae1dSRodney W. Grimes 		if (m->m_flags & (M_BCAST | M_MCAST)) {
693026decb8SRobert Watson 			UDPSTAT_INC(udps_noportbcast);
694fa046d87SRobert Watson 			goto badunlocked;
695df8bae1dSRodney W. Grimes 		}
6963ea9a7cfSGleb Smirnoff 		if (V_udp_blackhole && (V_udp_blackhole_local ||
6973ea9a7cfSGleb Smirnoff 		    !in_localip(ip->ip_src)))
698fa046d87SRobert Watson 			goto badunlocked;
6991cbd978eSLuigi Rizzo 		if (badport_bandlim(BANDLIM_ICMP_UNREACH) < 0)
700fa046d87SRobert Watson 			goto badunlocked;
701582a7760SBruce Evans 		icmp_error(m, ICMP_UNREACH, ICMP_UNREACH_PORT, 0, 0);
7028f5a8818SKevin Lo 		return (IPPROTO_DONE);
703df8bae1dSRodney W. Grimes 	}
7043329b236SRobert Watson 
7053329b236SRobert Watson 	/*
7063329b236SRobert Watson 	 * Check the minimum TTL for socket.
7073329b236SRobert Watson 	 */
708fa046d87SRobert Watson 	INP_RLOCK_ASSERT(inp);
70910cc62b7SRobert Watson 	if (inp->inp_ip_minttl && inp->inp_ip_minttl > ip->ip_ttl) {
7107bda9663SMichael Tuexen 		if (proto == IPPROTO_UDPLITE)
7117bda9663SMichael Tuexen 			UDPLITE_PROBE(receive, NULL, inp, ip, inp, uh);
7127bda9663SMichael Tuexen 		else
713e1526d5aSMichael Tuexen 			UDP_PROBE(receive, NULL, inp, ip, inp, uh);
71410cc62b7SRobert Watson 		INP_RUNLOCK(inp);
715fa046d87SRobert Watson 		m_freem(m);
7168f5a8818SKevin Lo 		return (IPPROTO_DONE);
71710cc62b7SRobert Watson 	}
718e06e816fSKevin Lo 	if (cscov_partial) {
719e06e816fSKevin Lo 		struct udpcb *up;
720e06e816fSKevin Lo 
721e06e816fSKevin Lo 		up = intoudpcb(inp);
72283e95fb3SMichael Tuexen 		if (up->u_rxcslen == 0 || up->u_rxcslen > len) {
723e06e816fSKevin Lo 			INP_RUNLOCK(inp);
724e06e816fSKevin Lo 			m_freem(m);
7258f5a8818SKevin Lo 			return (IPPROTO_DONE);
726e06e816fSKevin Lo 		}
727e06e816fSKevin Lo 	}
72857f60867SMark Johnston 
7297bda9663SMichael Tuexen 	if (proto == IPPROTO_UDPLITE)
7307bda9663SMichael Tuexen 		UDPLITE_PROBE(receive, NULL, inp, ip, inp, uh);
7317bda9663SMichael Tuexen 	else
7321ad19fb6SMark Johnston 		UDP_PROBE(receive, NULL, inp, ip, inp, uh);
733dce33a45SErmal Luçi 	if (udp_append(inp, ip, m, iphlen, udp_in) == 0)
734119d85f6SRobert Watson 		INP_RUNLOCK(inp);
7358f5a8818SKevin Lo 	return (IPPROTO_DONE);
73661ffc0b1SJeffrey Hsu 
737f76fcf6dSJeffrey Hsu badunlocked:
738df8bae1dSRodney W. Grimes 	m_freem(m);
7398f5a8818SKevin Lo 	return (IPPROTO_DONE);
740cfa1ca9dSYoshinobu Inoue }
74179288c11SBjoern A. Zeeb #endif /* INET */
742cfa1ca9dSYoshinobu Inoue 
743cfa1ca9dSYoshinobu Inoue /*
7443329b236SRobert Watson  * Notify a udp user of an asynchronous error; just wake up so that they can
7453329b236SRobert Watson  * collect error status.
746df8bae1dSRodney W. Grimes  */
7473ce144eaSJeffrey Hsu struct inpcb *
7483329b236SRobert Watson udp_notify(struct inpcb *inp, int errno)
749df8bae1dSRodney W. Grimes {
7503329b236SRobert Watson 
751083a010cSRyan Stone 	INP_WLOCK_ASSERT(inp);
75284cc0778SGeorge V. Neville-Neil 	if ((errno == EHOSTUNREACH || errno == ENETUNREACH ||
753983066f0SAlexander V. Chernikov 	     errno == EHOSTDOWN) && inp->inp_route.ro_nh) {
754983066f0SAlexander V. Chernikov 		NH_FREE(inp->inp_route.ro_nh);
755983066f0SAlexander V. Chernikov 		inp->inp_route.ro_nh = (struct nhop_object *)NULL;
75684cc0778SGeorge V. Neville-Neil 	}
7578501a69cSRobert Watson 
758df8bae1dSRodney W. Grimes 	inp->inp_socket->so_error = errno;
759df8bae1dSRodney W. Grimes 	sorwakeup(inp->inp_socket);
760df8bae1dSRodney W. Grimes 	sowwakeup(inp->inp_socket);
7613329b236SRobert Watson 	return (inp);
762df8bae1dSRodney W. Grimes }
763df8bae1dSRodney W. Grimes 
76479288c11SBjoern A. Zeeb #ifdef INET
765e06e816fSKevin Lo static void
766e06e816fSKevin Lo udp_common_ctlinput(int cmd, struct sockaddr *sa, void *vip,
767e06e816fSKevin Lo     struct inpcbinfo *pcbinfo)
768df8bae1dSRodney W. Grimes {
769c693a045SJonathan Lemon 	struct ip *ip = vip;
770c693a045SJonathan Lemon 	struct udphdr *uh;
771c693a045SJonathan Lemon 	struct in_addr faddr;
772c693a045SJonathan Lemon 	struct inpcb *inp;
773c693a045SJonathan Lemon 
774c693a045SJonathan Lemon 	faddr = ((struct sockaddr_in *)sa)->sin_addr;
775c693a045SJonathan Lemon 	if (sa->sa_family != AF_INET || faddr.s_addr == INADDR_ANY)
776c693a045SJonathan Lemon 		return;
777df8bae1dSRodney W. Grimes 
77884cc0778SGeorge V. Neville-Neil 	if (PRC_IS_REDIRECT(cmd)) {
77984cc0778SGeorge V. Neville-Neil 		/* signal EHOSTDOWN, as it flushes the cached route */
7804760956eSMichael Tuexen 		in_pcbnotifyall(pcbinfo, faddr, EHOSTDOWN, udp_notify);
78197d8d152SAndre Oppermann 		return;
78284cc0778SGeorge V. Neville-Neil 	}
7833329b236SRobert Watson 
78497d8d152SAndre Oppermann 	/*
78597d8d152SAndre Oppermann 	 * Hostdead is ugly because it goes linearly through all PCBs.
7863329b236SRobert Watson 	 *
7873329b236SRobert Watson 	 * XXX: We never get this from ICMP, otherwise it makes an excellent
7883329b236SRobert Watson 	 * DoS attack on machines with many connections.
78997d8d152SAndre Oppermann 	 */
79097d8d152SAndre Oppermann 	if (cmd == PRC_HOSTDEAD)
791af1ee11dSRobert Watson 		ip = NULL;
792d1c54148SJesper Skriver 	else if ((unsigned)cmd >= PRC_NCMDS || inetctlerrmap[cmd] == 0)
793df8bae1dSRodney W. Grimes 		return;
794af1ee11dSRobert Watson 	if (ip != NULL) {
795df8bae1dSRodney W. Grimes 		uh = (struct udphdr *)((caddr_t)ip + (ip->ip_hl << 2));
796e06e816fSKevin Lo 		inp = in_pcblookup(pcbinfo, faddr, uh->uh_dport,
797083a010cSRyan Stone 		    ip->ip_src, uh->uh_sport, INPLOOKUP_WLOCKPCB, NULL);
798f76fcf6dSJeffrey Hsu 		if (inp != NULL) {
799083a010cSRyan Stone 			INP_WLOCK_ASSERT(inp);
800f76fcf6dSJeffrey Hsu 			if (inp->inp_socket != NULL) {
801f5514f08SRobert Watson 				udp_notify(inp, inetctlerrmap[cmd]);
802f76fcf6dSJeffrey Hsu 			}
803083a010cSRyan Stone 			INP_WUNLOCK(inp);
804abb901c5SRandall Stewart 		} else {
805abb901c5SRandall Stewart 			inp = in_pcblookup(pcbinfo, faddr, uh->uh_dport,
806abb901c5SRandall Stewart 					   ip->ip_src, uh->uh_sport,
807abb901c5SRandall Stewart 					   INPLOOKUP_WILDCARD | INPLOOKUP_RLOCKPCB, NULL);
808abb901c5SRandall Stewart 			if (inp != NULL) {
809abb901c5SRandall Stewart 				struct udpcb *up;
81046374cbfSMatt Macy 				void *ctx;
81146374cbfSMatt Macy 				udp_tun_icmp_t func;
812abb901c5SRandall Stewart 
813abb901c5SRandall Stewart 				up = intoudpcb(inp);
81446374cbfSMatt Macy 				ctx = up->u_tun_ctx;
81546374cbfSMatt Macy 				func = up->u_icmp_func;
816abb901c5SRandall Stewart 				INP_RUNLOCK(inp);
81746374cbfSMatt Macy 				if (func != NULL)
81846374cbfSMatt Macy 					(*func)(cmd, sa, vip, ctx);
819abb901c5SRandall Stewart 			}
820f76fcf6dSJeffrey Hsu 		}
821df8bae1dSRodney W. Grimes 	} else
822e06e816fSKevin Lo 		in_pcbnotifyall(pcbinfo, faddr, inetctlerrmap[cmd],
823f5514f08SRobert Watson 		    udp_notify);
824df8bae1dSRodney W. Grimes }
825e06e816fSKevin Lo void
826e06e816fSKevin Lo udp_ctlinput(int cmd, struct sockaddr *sa, void *vip)
827e06e816fSKevin Lo {
828e06e816fSKevin Lo 
829e06e816fSKevin Lo 	return (udp_common_ctlinput(cmd, sa, vip, &V_udbinfo));
830e06e816fSKevin Lo }
831e06e816fSKevin Lo 
832e06e816fSKevin Lo void
833e06e816fSKevin Lo udplite_ctlinput(int cmd, struct sockaddr *sa, void *vip)
834e06e816fSKevin Lo {
835e06e816fSKevin Lo 
836e06e816fSKevin Lo 	return (udp_common_ctlinput(cmd, sa, vip, &V_ulitecbinfo));
837e06e816fSKevin Lo }
83879288c11SBjoern A. Zeeb #endif /* INET */
839df8bae1dSRodney W. Grimes 
8400312fbe9SPoul-Henning Kamp static int
84182d9ae4eSPoul-Henning Kamp udp_pcblist(SYSCTL_HANDLER_ARGS)
84298271db4SGarrett Wollman {
843db0ac6deSCy Schubert 	struct inpcb_iterator inpi = INP_ALL_ITERATOR(&V_udbinfo,
844db0ac6deSCy Schubert 	    INPLOOKUP_RLOCKPCB);
84598271db4SGarrett Wollman 	struct xinpgen xig;
846032677ceSGleb Smirnoff 	struct inpcb *inp;
847032677ceSGleb Smirnoff 	int error;
84898271db4SGarrett Wollman 
849032677ceSGleb Smirnoff 	if (req->newptr != 0)
850032677ceSGleb Smirnoff 		return (EPERM);
851032677ceSGleb Smirnoff 
85298271db4SGarrett Wollman 	if (req->oldptr == 0) {
853032677ceSGleb Smirnoff 		int n;
854032677ceSGleb Smirnoff 
855603724d3SBjoern A. Zeeb 		n = V_udbinfo.ipi_count;
856c007b96aSJohn Baldwin 		n += imax(n / 8, 10);
857c007b96aSJohn Baldwin 		req->oldidx = 2 * (sizeof xig) + n * sizeof(struct xinpcb);
8583329b236SRobert Watson 		return (0);
85998271db4SGarrett Wollman 	}
86098271db4SGarrett Wollman 
861032677ceSGleb Smirnoff 	if ((error = sysctl_wire_old_buffer(req, 0)) != 0)
86247934cefSDon Lewis 		return (error);
8635c38b6dbSDon Lewis 
86479db6fe7SMark Johnston 	bzero(&xig, sizeof(xig));
86598271db4SGarrett Wollman 	xig.xig_len = sizeof xig;
866032677ceSGleb Smirnoff 	xig.xig_count = V_udbinfo.ipi_count;
867032677ceSGleb Smirnoff 	xig.xig_gen = V_udbinfo.ipi_gencnt;
86898271db4SGarrett Wollman 	xig.xig_sogen = so_gencnt;
86998271db4SGarrett Wollman 	error = SYSCTL_OUT(req, &xig, sizeof xig);
87098271db4SGarrett Wollman 	if (error)
8713329b236SRobert Watson 		return (error);
87299208b82SMatt Macy 
873db0ac6deSCy Schubert 	while ((inp = inp_next(&inpi)) != NULL) {
874032677ceSGleb Smirnoff 		if (inp->inp_gencnt <= xig.xig_gen &&
875032677ceSGleb Smirnoff 		    cr_canseeinpcb(req->td->td_ucred, inp) == 0) {
87698271db4SGarrett Wollman 			struct xinpcb xi;
877d0e157f6SBjoern A. Zeeb 
878cc65eb4eSGleb Smirnoff 			in_pcbtoxinpcb(inp, &xi);
879266f97b5SCy Schubert 			error = SYSCTL_OUT(req, &xi, sizeof xi);
880db0ac6deSCy Schubert 			if (error) {
881266f97b5SCy Schubert 				INP_RUNLOCK(inp);
882db0ac6deSCy Schubert 				break;
88398271db4SGarrett Wollman 			}
884db0ac6deSCy Schubert 		}
885db0ac6deSCy Schubert 	}
886d0e157f6SBjoern A. Zeeb 
88798271db4SGarrett Wollman 	if (!error) {
88898271db4SGarrett Wollman 		/*
8893329b236SRobert Watson 		 * Give the user an updated idea of our state.  If the
8903329b236SRobert Watson 		 * generation differs from what we told her before, she knows
8913329b236SRobert Watson 		 * that something happened while we were processing this
8923329b236SRobert Watson 		 * request, and it might be necessary to retry.
89398271db4SGarrett Wollman 		 */
894603724d3SBjoern A. Zeeb 		xig.xig_gen = V_udbinfo.ipi_gencnt;
89598271db4SGarrett Wollman 		xig.xig_sogen = so_gencnt;
896603724d3SBjoern A. Zeeb 		xig.xig_count = V_udbinfo.ipi_count;
89798271db4SGarrett Wollman 		error = SYSCTL_OUT(req, &xig, sizeof xig);
89898271db4SGarrett Wollman 	}
899032677ceSGleb Smirnoff 
9003329b236SRobert Watson 	return (error);
90198271db4SGarrett Wollman }
90298271db4SGarrett Wollman 
90379c3d51bSMatthew D Fleming SYSCTL_PROC(_net_inet_udp, UDPCTL_PCBLIST, pcblist,
9047029da5cSPawel Biernacki     CTLTYPE_OPAQUE | CTLFLAG_RD | CTLFLAG_MPSAFE, NULL, 0,
9057029da5cSPawel Biernacki     udp_pcblist, "S,xinpcb",
9067029da5cSPawel Biernacki     "List of active UDP sockets");
90798271db4SGarrett Wollman 
90879288c11SBjoern A. Zeeb #ifdef INET
90998271db4SGarrett Wollman static int
91082d9ae4eSPoul-Henning Kamp udp_getcred(SYSCTL_HANDLER_ARGS)
911490d50b6SBrian Feldman {
912c0511d3bSBrian Feldman 	struct xucred xuc;
913490d50b6SBrian Feldman 	struct sockaddr_in addrs[2];
914d797164aSGleb Smirnoff 	struct epoch_tracker et;
915490d50b6SBrian Feldman 	struct inpcb *inp;
916277afaffSRobert Watson 	int error;
917490d50b6SBrian Feldman 
91832f9753cSRobert Watson 	error = priv_check(req->td, PRIV_NETINET_GETCRED);
919490d50b6SBrian Feldman 	if (error)
920490d50b6SBrian Feldman 		return (error);
921490d50b6SBrian Feldman 	error = SYSCTL_IN(req, addrs, sizeof(addrs));
922490d50b6SBrian Feldman 	if (error)
923490d50b6SBrian Feldman 		return (error);
924d797164aSGleb Smirnoff 	NET_EPOCH_ENTER(et);
925fa046d87SRobert Watson 	inp = in_pcblookup(&V_udbinfo, addrs[1].sin_addr, addrs[1].sin_port,
926fa046d87SRobert Watson 	    addrs[0].sin_addr, addrs[0].sin_port,
927fa046d87SRobert Watson 	    INPLOOKUP_WILDCARD | INPLOOKUP_RLOCKPCB, NULL);
928d797164aSGleb Smirnoff 	NET_EPOCH_EXIT(et);
9299622e84fSRobert Watson 	if (inp != NULL) {
930fa046d87SRobert Watson 		INP_RLOCK_ASSERT(inp);
9319622e84fSRobert Watson 		if (inp->inp_socket == NULL)
9329622e84fSRobert Watson 			error = ENOENT;
9339622e84fSRobert Watson 		if (error == 0)
934f08ef6c5SBjoern A. Zeeb 			error = cr_canseeinpcb(req->td->td_ucred, inp);
9359622e84fSRobert Watson 		if (error == 0)
93686d02c5cSBjoern A. Zeeb 			cru2x(inp->inp_cred, &xuc);
9379622e84fSRobert Watson 		INP_RUNLOCK(inp);
938fa046d87SRobert Watson 	} else
9399622e84fSRobert Watson 		error = ENOENT;
9400e1eebb8SDon Lewis 	if (error == 0)
9410e1eebb8SDon Lewis 		error = SYSCTL_OUT(req, &xuc, sizeof(struct xucred));
942490d50b6SBrian Feldman 	return (error);
943490d50b6SBrian Feldman }
944490d50b6SBrian Feldman 
9457ce87f12SDavid Malone SYSCTL_PROC(_net_inet_udp, OID_AUTO, getcred,
9467029da5cSPawel Biernacki     CTLTYPE_OPAQUE | CTLFLAG_RW | CTLFLAG_PRISON | CTLFLAG_MPSAFE,
9477029da5cSPawel Biernacki     0, 0, udp_getcred, "S,xucred",
9487029da5cSPawel Biernacki     "Get the xucred of a UDP connection");
94979288c11SBjoern A. Zeeb #endif /* INET */
950490d50b6SBrian Feldman 
9517b495c44SVANHULLEBUS Yvan int
9527b495c44SVANHULLEBUS Yvan udp_ctloutput(struct socket *so, struct sockopt *sopt)
9537b495c44SVANHULLEBUS Yvan {
9547b495c44SVANHULLEBUS Yvan 	struct inpcb *inp;
9557b495c44SVANHULLEBUS Yvan 	struct udpcb *up;
956e06e816fSKevin Lo 	int isudplite, error, optval;
9577b495c44SVANHULLEBUS Yvan 
958e06e816fSKevin Lo 	error = 0;
959e06e816fSKevin Lo 	isudplite = (so->so_proto->pr_protocol == IPPROTO_UDPLITE) ? 1 : 0;
9607b495c44SVANHULLEBUS Yvan 	inp = sotoinpcb(so);
9617b495c44SVANHULLEBUS Yvan 	KASSERT(inp != NULL, ("%s: inp == NULL", __func__));
9627b495c44SVANHULLEBUS Yvan 	INP_WLOCK(inp);
963e06e816fSKevin Lo 	if (sopt->sopt_level != so->so_proto->pr_protocol) {
9647b495c44SVANHULLEBUS Yvan #ifdef INET6
9657b495c44SVANHULLEBUS Yvan 		if (INP_CHECK_SOCKAF(so, AF_INET6)) {
9667b495c44SVANHULLEBUS Yvan 			INP_WUNLOCK(inp);
9677b495c44SVANHULLEBUS Yvan 			error = ip6_ctloutput(so, sopt);
96879288c11SBjoern A. Zeeb 		}
9697b495c44SVANHULLEBUS Yvan #endif
97079288c11SBjoern A. Zeeb #if defined(INET) && defined(INET6)
97179288c11SBjoern A. Zeeb 		else
97279288c11SBjoern A. Zeeb #endif
97379288c11SBjoern A. Zeeb #ifdef INET
97479288c11SBjoern A. Zeeb 		{
9757b495c44SVANHULLEBUS Yvan 			INP_WUNLOCK(inp);
9767b495c44SVANHULLEBUS Yvan 			error = ip_ctloutput(so, sopt);
9777b495c44SVANHULLEBUS Yvan 		}
9787b495c44SVANHULLEBUS Yvan #endif
9797b495c44SVANHULLEBUS Yvan 		return (error);
9807b495c44SVANHULLEBUS Yvan 	}
9817b495c44SVANHULLEBUS Yvan 
9827b495c44SVANHULLEBUS Yvan 	switch (sopt->sopt_dir) {
9837b495c44SVANHULLEBUS Yvan 	case SOPT_SET:
9847b495c44SVANHULLEBUS Yvan 		switch (sopt->sopt_name) {
985fcf59617SAndrey V. Elsukov #if defined(IPSEC) || defined(IPSEC_SUPPORT)
986fcf59617SAndrey V. Elsukov #ifdef INET
9877b495c44SVANHULLEBUS Yvan 		case UDP_ENCAP:
988fcf59617SAndrey V. Elsukov 			if (!IPSEC_ENABLED(ipv4)) {
9897b495c44SVANHULLEBUS Yvan 				INP_WUNLOCK(inp);
990fcf59617SAndrey V. Elsukov 				return (ENOPROTOOPT);
9917b495c44SVANHULLEBUS Yvan 			}
992fcf59617SAndrey V. Elsukov 			error = UDPENCAP_PCBCTL(inp, sopt);
9937b495c44SVANHULLEBUS Yvan 			break;
994fcf59617SAndrey V. Elsukov #endif /* INET */
995fcf59617SAndrey V. Elsukov #endif /* IPSEC */
996e06e816fSKevin Lo 		case UDPLITE_SEND_CSCOV:
997e06e816fSKevin Lo 		case UDPLITE_RECV_CSCOV:
998e06e816fSKevin Lo 			if (!isudplite) {
999e06e816fSKevin Lo 				INP_WUNLOCK(inp);
1000e06e816fSKevin Lo 				error = ENOPROTOOPT;
1001e06e816fSKevin Lo 				break;
1002e06e816fSKevin Lo 			}
1003e06e816fSKevin Lo 			INP_WUNLOCK(inp);
1004e06e816fSKevin Lo 			error = sooptcopyin(sopt, &optval, sizeof(optval),
1005e06e816fSKevin Lo 			    sizeof(optval));
1006e06e816fSKevin Lo 			if (error != 0)
1007e06e816fSKevin Lo 				break;
1008e06e816fSKevin Lo 			inp = sotoinpcb(so);
1009e06e816fSKevin Lo 			KASSERT(inp != NULL, ("%s: inp == NULL", __func__));
1010e06e816fSKevin Lo 			INP_WLOCK(inp);
1011e06e816fSKevin Lo 			up = intoudpcb(inp);
1012e06e816fSKevin Lo 			KASSERT(up != NULL, ("%s: up == NULL", __func__));
101303f90784SMichael Tuexen 			if ((optval != 0 && optval < 8) || (optval > 65535)) {
1014e06e816fSKevin Lo 				INP_WUNLOCK(inp);
1015e06e816fSKevin Lo 				error = EINVAL;
1016e06e816fSKevin Lo 				break;
1017e06e816fSKevin Lo 			}
1018e06e816fSKevin Lo 			if (sopt->sopt_name == UDPLITE_SEND_CSCOV)
1019e06e816fSKevin Lo 				up->u_txcslen = optval;
1020e06e816fSKevin Lo 			else
1021e06e816fSKevin Lo 				up->u_rxcslen = optval;
1022e06e816fSKevin Lo 			INP_WUNLOCK(inp);
1023e06e816fSKevin Lo 			break;
10247b495c44SVANHULLEBUS Yvan 		default:
10257b495c44SVANHULLEBUS Yvan 			INP_WUNLOCK(inp);
10267b495c44SVANHULLEBUS Yvan 			error = ENOPROTOOPT;
10277b495c44SVANHULLEBUS Yvan 			break;
10287b495c44SVANHULLEBUS Yvan 		}
10297b495c44SVANHULLEBUS Yvan 		break;
10307b495c44SVANHULLEBUS Yvan 	case SOPT_GET:
10317b495c44SVANHULLEBUS Yvan 		switch (sopt->sopt_name) {
1032fcf59617SAndrey V. Elsukov #if defined(IPSEC) || defined(IPSEC_SUPPORT)
1033fcf59617SAndrey V. Elsukov #ifdef INET
10347b495c44SVANHULLEBUS Yvan 		case UDP_ENCAP:
1035fcf59617SAndrey V. Elsukov 			if (!IPSEC_ENABLED(ipv4)) {
10367b495c44SVANHULLEBUS Yvan 				INP_WUNLOCK(inp);
1037fcf59617SAndrey V. Elsukov 				return (ENOPROTOOPT);
1038fcf59617SAndrey V. Elsukov 			}
1039fcf59617SAndrey V. Elsukov 			error = UDPENCAP_PCBCTL(inp, sopt);
10407b495c44SVANHULLEBUS Yvan 			break;
1041fcf59617SAndrey V. Elsukov #endif /* INET */
1042fcf59617SAndrey V. Elsukov #endif /* IPSEC */
1043e06e816fSKevin Lo 		case UDPLITE_SEND_CSCOV:
1044e06e816fSKevin Lo 		case UDPLITE_RECV_CSCOV:
1045e06e816fSKevin Lo 			if (!isudplite) {
1046e06e816fSKevin Lo 				INP_WUNLOCK(inp);
1047e06e816fSKevin Lo 				error = ENOPROTOOPT;
1048e06e816fSKevin Lo 				break;
1049e06e816fSKevin Lo 			}
1050e06e816fSKevin Lo 			up = intoudpcb(inp);
1051e06e816fSKevin Lo 			KASSERT(up != NULL, ("%s: up == NULL", __func__));
1052e06e816fSKevin Lo 			if (sopt->sopt_name == UDPLITE_SEND_CSCOV)
1053e06e816fSKevin Lo 				optval = up->u_txcslen;
1054e06e816fSKevin Lo 			else
1055e06e816fSKevin Lo 				optval = up->u_rxcslen;
1056e06e816fSKevin Lo 			INP_WUNLOCK(inp);
1057e06e816fSKevin Lo 			error = sooptcopyout(sopt, &optval, sizeof(optval));
1058e06e816fSKevin Lo 			break;
10597b495c44SVANHULLEBUS Yvan 		default:
10607b495c44SVANHULLEBUS Yvan 			INP_WUNLOCK(inp);
10617b495c44SVANHULLEBUS Yvan 			error = ENOPROTOOPT;
10627b495c44SVANHULLEBUS Yvan 			break;
10637b495c44SVANHULLEBUS Yvan 		}
10647b495c44SVANHULLEBUS Yvan 		break;
10657b495c44SVANHULLEBUS Yvan 	}
10667b495c44SVANHULLEBUS Yvan 	return (error);
10677b495c44SVANHULLEBUS Yvan }
10687b495c44SVANHULLEBUS Yvan 
106979288c11SBjoern A. Zeeb #ifdef INET
1070a9839c4aSBjoern A. Zeeb #ifdef INET6
1071a9839c4aSBjoern A. Zeeb /* The logic here is derived from ip6_setpktopt(). See comments there. */
1072a9839c4aSBjoern A. Zeeb static int
1073a9839c4aSBjoern A. Zeeb udp_v4mapped_pktinfo(struct cmsghdr *cm, struct sockaddr_in * src,
1074a9839c4aSBjoern A. Zeeb     struct inpcb *inp, int flags)
1075a9839c4aSBjoern A. Zeeb {
1076a9839c4aSBjoern A. Zeeb 	struct ifnet *ifp;
1077a9839c4aSBjoern A. Zeeb 	struct in6_pktinfo *pktinfo;
1078a9839c4aSBjoern A. Zeeb 	struct in_addr ia;
1079a9839c4aSBjoern A. Zeeb 
1080a9839c4aSBjoern A. Zeeb 	if ((flags & PRUS_IPV6) == 0)
1081a9839c4aSBjoern A. Zeeb 		return (0);
1082a9839c4aSBjoern A. Zeeb 
1083a9839c4aSBjoern A. Zeeb 	if (cm->cmsg_level != IPPROTO_IPV6)
1084a9839c4aSBjoern A. Zeeb 		return (0);
1085a9839c4aSBjoern A. Zeeb 
1086a9839c4aSBjoern A. Zeeb 	if  (cm->cmsg_type != IPV6_2292PKTINFO &&
1087a9839c4aSBjoern A. Zeeb 	    cm->cmsg_type != IPV6_PKTINFO)
1088a9839c4aSBjoern A. Zeeb 		return (0);
1089a9839c4aSBjoern A. Zeeb 
1090a9839c4aSBjoern A. Zeeb 	if (cm->cmsg_len !=
1091a9839c4aSBjoern A. Zeeb 	    CMSG_LEN(sizeof(struct in6_pktinfo)))
1092a9839c4aSBjoern A. Zeeb 		return (EINVAL);
1093a9839c4aSBjoern A. Zeeb 
1094a9839c4aSBjoern A. Zeeb 	pktinfo = (struct in6_pktinfo *)CMSG_DATA(cm);
1095a9839c4aSBjoern A. Zeeb 	if (!IN6_IS_ADDR_V4MAPPED(&pktinfo->ipi6_addr) &&
1096a9839c4aSBjoern A. Zeeb 	    !IN6_IS_ADDR_UNSPECIFIED(&pktinfo->ipi6_addr))
1097a9839c4aSBjoern A. Zeeb 		return (EINVAL);
1098a9839c4aSBjoern A. Zeeb 
1099a9839c4aSBjoern A. Zeeb 	/* Validate the interface index if specified. */
1100a9839c4aSBjoern A. Zeeb 	if (pktinfo->ipi6_ifindex) {
1101d74b7baeSGleb Smirnoff 		struct epoch_tracker et;
1102d74b7baeSGleb Smirnoff 
1103d74b7baeSGleb Smirnoff 		NET_EPOCH_ENTER(et);
1104a9839c4aSBjoern A. Zeeb 		ifp = ifnet_byindex(pktinfo->ipi6_ifindex);
1105d74b7baeSGleb Smirnoff 		NET_EPOCH_EXIT(et);	/* XXXGL: unsafe ifp */
1106a9839c4aSBjoern A. Zeeb 		if (ifp == NULL)
1107a9839c4aSBjoern A. Zeeb 			return (ENXIO);
1108d74b7baeSGleb Smirnoff 	} else
1109d74b7baeSGleb Smirnoff 		ifp = NULL;
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 		}
1278db0ac6deSCy 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);
1281db0ac6deSCy 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) {
1324db0ac6deSCy 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);
1328db0ac6deSCy Schubert 			if (error) {
1329db0ac6deSCy Schubert 				INP_HASH_WUNLOCK(pcbinfo);
133090162a4eSIan Dowse 				goto release;
1331db0ac6deSCy 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;
1356db0ac6deSCy Schubert 			} else
1357db0ac6deSCy 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);
1552db0ac6deSCy 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);
1568db0ac6deSCy 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);
1683db0ac6deSCy 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;
16966a9148feSBjoern A. Zeeb 	struct udpcb *up;
1697d0390e05SGarrett Wollman 
1698d0390e05SGarrett Wollman 	inp = sotoinpcb(so);
169914ba8addSRobert Watson 	KASSERT(inp != NULL, ("udp_detach: inp == NULL"));
1700a152f8a3SRobert Watson 	KASSERT(inp->inp_faddr.s_addr == INADDR_ANY,
1701a152f8a3SRobert Watson 	    ("udp_detach: not disconnected"));
17028501a69cSRobert Watson 	INP_WLOCK(inp);
17036a9148feSBjoern A. Zeeb 	up = intoudpcb(inp);
17046a9148feSBjoern A. Zeeb 	KASSERT(up != NULL, ("%s: up == NULL", __func__));
17056a9148feSBjoern A. Zeeb 	inp->inp_ppcb = NULL;
1706d0390e05SGarrett Wollman 	in_pcbdetach(inp);
170714ba8addSRobert Watson 	in_pcbfree(inp);
17086a9148feSBjoern A. Zeeb 	udp_discardcb(up);
1709d0390e05SGarrett Wollman }
1710d0390e05SGarrett Wollman 
1711d0390e05SGarrett Wollman static int
1712d0390e05SGarrett Wollman udp_disconnect(struct socket *so)
1713d0390e05SGarrett Wollman {
1714d0390e05SGarrett Wollman 	struct inpcb *inp;
1715e06e816fSKevin Lo 	struct inpcbinfo *pcbinfo;
1716d0390e05SGarrett Wollman 
1717a86e5c96SBjoern A. Zeeb 	pcbinfo = udp_get_inpcbinfo(so->so_proto->pr_protocol);
1718d0390e05SGarrett Wollman 	inp = sotoinpcb(so);
171914ba8addSRobert Watson 	KASSERT(inp != NULL, ("udp_disconnect: inp == NULL"));
17208501a69cSRobert Watson 	INP_WLOCK(inp);
1721f76fcf6dSJeffrey Hsu 	if (inp->inp_faddr.s_addr == INADDR_ANY) {
17228501a69cSRobert Watson 		INP_WUNLOCK(inp);
17233329b236SRobert Watson 		return (ENOTCONN);
1724f76fcf6dSJeffrey Hsu 	}
1725e06e816fSKevin Lo 	INP_HASH_WLOCK(pcbinfo);
1726df8bae1dSRodney W. Grimes 	in_pcbdisconnect(inp);
1727df8bae1dSRodney W. Grimes 	inp->inp_laddr.s_addr = INADDR_ANY;
1728e06e816fSKevin Lo 	INP_HASH_WUNLOCK(pcbinfo);
1729d45e4f99SMaxim Konovalov 	SOCK_LOCK(so);
1730d45e4f99SMaxim Konovalov 	so->so_state &= ~SS_ISCONNECTED;		/* XXX */
1731d45e4f99SMaxim Konovalov 	SOCK_UNLOCK(so);
17328501a69cSRobert Watson 	INP_WUNLOCK(inp);
17333329b236SRobert Watson 	return (0);
1734df8bae1dSRodney W. Grimes }
1735df8bae1dSRodney W. Grimes 
1736d0390e05SGarrett Wollman static int
173757bf258eSGarrett Wollman udp_send(struct socket *so, int flags, struct mbuf *m, struct sockaddr *addr,
1738b40ce416SJulian Elischer     struct mbuf *control, struct thread *td)
1739d0390e05SGarrett Wollman {
1740d0390e05SGarrett Wollman 	struct inpcb *inp;
1741f161d294SMark Johnston 	int error;
1742d0390e05SGarrett Wollman 
1743d0390e05SGarrett Wollman 	inp = sotoinpcb(so);
174414ba8addSRobert Watson 	KASSERT(inp != NULL, ("udp_send: inp == NULL"));
1745f161d294SMark Johnston 
1746f161d294SMark Johnston 	if (addr != NULL) {
1747f161d294SMark Johnston 		error = 0;
1748f161d294SMark Johnston 		if (addr->sa_family != AF_INET)
1749f161d294SMark Johnston 			error = EAFNOSUPPORT;
1750f161d294SMark Johnston 		else if (addr->sa_len != sizeof(struct sockaddr_in))
1751f161d294SMark Johnston 			error = EINVAL;
1752f161d294SMark Johnston 		if (__predict_false(error != 0)) {
1753f161d294SMark Johnston 			m_freem(control);
1754f161d294SMark Johnston 			m_freem(m);
1755f161d294SMark Johnston 			return (error);
1756f161d294SMark Johnston 		}
1757f161d294SMark Johnston 	}
1758a9839c4aSBjoern A. Zeeb 	return (udp_output(inp, m, addr, control, td, flags));
1759d0390e05SGarrett Wollman }
176079288c11SBjoern A. Zeeb #endif /* INET */
1761d0390e05SGarrett Wollman 
176276429de4SYoshinobu Inoue int
1763d0390e05SGarrett Wollman udp_shutdown(struct socket *so)
1764d0390e05SGarrett Wollman {
1765d0390e05SGarrett Wollman 	struct inpcb *inp;
1766d0390e05SGarrett Wollman 
1767d0390e05SGarrett Wollman 	inp = sotoinpcb(so);
176814ba8addSRobert Watson 	KASSERT(inp != NULL, ("udp_shutdown: inp == NULL"));
17698501a69cSRobert Watson 	INP_WLOCK(inp);
1770d0390e05SGarrett Wollman 	socantsendmore(so);
17718501a69cSRobert Watson 	INP_WUNLOCK(inp);
17723329b236SRobert Watson 	return (0);
1773d0390e05SGarrett Wollman }
1774d0390e05SGarrett Wollman 
177579288c11SBjoern A. Zeeb #ifdef INET
1776d0390e05SGarrett Wollman struct pr_usrreqs udp_usrreqs = {
1777756d52a1SPoul-Henning Kamp 	.pru_abort =		udp_abort,
1778756d52a1SPoul-Henning Kamp 	.pru_attach =		udp_attach,
1779756d52a1SPoul-Henning Kamp 	.pru_bind =		udp_bind,
1780756d52a1SPoul-Henning Kamp 	.pru_connect =		udp_connect,
1781756d52a1SPoul-Henning Kamp 	.pru_control =		in_control,
1782756d52a1SPoul-Henning Kamp 	.pru_detach =		udp_detach,
1783756d52a1SPoul-Henning Kamp 	.pru_disconnect =	udp_disconnect,
178454d642bbSRobert Watson 	.pru_peeraddr =		in_getpeeraddr,
1785756d52a1SPoul-Henning Kamp 	.pru_send =		udp_send,
17865df3e839SRobert Watson 	.pru_soreceive =	soreceive_dgram,
178759b8854eSRobert Watson 	.pru_sosend =		sosend_dgram,
1788756d52a1SPoul-Henning Kamp 	.pru_shutdown =		udp_shutdown,
178954d642bbSRobert Watson 	.pru_sockaddr =		in_getsockaddr,
1790a152f8a3SRobert Watson 	.pru_sosetlabel =	in_pcbsosetlabel,
1791a152f8a3SRobert Watson 	.pru_close =		udp_close,
1792d0390e05SGarrett Wollman };
179379288c11SBjoern A. Zeeb #endif /* INET */
1794