xref: /freebsd/sys/netinet/udp_usrreq.c (revision 374ce2488ae6f84a6b902a0125569bae434fa375)
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"
479d3ddf43SAdrian Chadd #include "opt_rss.h"
48cfa1ca9dSYoshinobu Inoue 
49df8bae1dSRodney W. Grimes #include <sys/param.h>
50960ed29cSSeigo Tanimura #include <sys/domain.h>
514f590175SPaul Saab #include <sys/eventhandler.h>
52960ed29cSSeigo Tanimura #include <sys/jail.h>
53b110a8a2SGarrett Wollman #include <sys/kernel.h>
54960ed29cSSeigo Tanimura #include <sys/lock.h>
55df8bae1dSRodney W. Grimes #include <sys/malloc.h>
56df8bae1dSRodney W. Grimes #include <sys/mbuf.h>
57acd3428bSRobert Watson #include <sys/priv.h>
58490d50b6SBrian Feldman #include <sys/proc.h>
59df8bae1dSRodney W. Grimes #include <sys/protosw.h>
6057f60867SMark Johnston #include <sys/sdt.h>
61960ed29cSSeigo Tanimura #include <sys/signalvar.h>
62df8bae1dSRodney W. Grimes #include <sys/socket.h>
63df8bae1dSRodney W. Grimes #include <sys/socketvar.h>
64960ed29cSSeigo Tanimura #include <sys/sx.h>
65b5e8ce9fSBruce Evans #include <sys/sysctl.h>
66816a3d83SPoul-Henning Kamp #include <sys/syslog.h>
67f5514f08SRobert Watson #include <sys/systm.h>
688781d8e9SBruce Evans 
6969c2d429SJeff Roberson #include <vm/uma.h>
70df8bae1dSRodney W. Grimes 
71df8bae1dSRodney W. Grimes #include <net/if.h>
7276039bc8SGleb Smirnoff #include <net/if_var.h>
73df8bae1dSRodney W. Grimes #include <net/route.h>
74983066f0SAlexander V. Chernikov #include <net/route/nhop.h>
75b2bdc62aSAdrian Chadd #include <net/rss_config.h>
76df8bae1dSRodney W. Grimes 
77df8bae1dSRodney W. Grimes #include <netinet/in.h>
7857f60867SMark Johnston #include <netinet/in_kdtrace.h>
79960ed29cSSeigo Tanimura #include <netinet/in_pcb.h>
80f5514f08SRobert Watson #include <netinet/in_systm.h>
81960ed29cSSeigo Tanimura #include <netinet/in_var.h>
82df8bae1dSRodney W. Grimes #include <netinet/ip.h>
83cfa1ca9dSYoshinobu Inoue #ifdef INET6
84cfa1ca9dSYoshinobu Inoue #include <netinet/ip6.h>
85cfa1ca9dSYoshinobu Inoue #endif
86960ed29cSSeigo Tanimura #include <netinet/ip_icmp.h>
87960ed29cSSeigo Tanimura #include <netinet/icmp_var.h>
88df8bae1dSRodney W. Grimes #include <netinet/ip_var.h>
89ef39adf0SAndre Oppermann #include <netinet/ip_options.h>
90cfa1ca9dSYoshinobu Inoue #ifdef INET6
91cfa1ca9dSYoshinobu Inoue #include <netinet6/ip6_var.h>
92cfa1ca9dSYoshinobu Inoue #endif
93df8bae1dSRodney W. Grimes #include <netinet/udp.h>
94df8bae1dSRodney W. Grimes #include <netinet/udp_var.h>
95e06e816fSKevin Lo #include <netinet/udplite.h>
968ad1a83bSAdrian Chadd #include <netinet/in_rss.h>
97df8bae1dSRodney W. Grimes 
98fcf59617SAndrey V. Elsukov #include <netipsec/ipsec_support.h>
99b9234fafSSam Leffler 
100db4f9cc7SJonathan Lemon #include <machine/in_cksum.h>
101db4f9cc7SJonathan Lemon 
102aed55708SRobert Watson #include <security/mac/mac_framework.h>
103aed55708SRobert Watson 
104df8bae1dSRodney W. Grimes /*
105e06e816fSKevin Lo  * UDP and UDP-Lite protocols implementation.
106df8bae1dSRodney W. Grimes  * Per RFC 768, August, 1980.
107e06e816fSKevin Lo  * Per RFC 3828, July, 2004.
108df8bae1dSRodney W. Grimes  */
10974eb3236SWarner Losh 
11074eb3236SWarner Losh /*
1113329b236SRobert Watson  * BSD 4.2 defaulted the udp checksum to be off.  Turning off udp checksums
1123329b236SRobert Watson  * removes the only data integrity mechanism for packets and malformed
113f5514f08SRobert Watson  * packets that would otherwise be discarded due to bad checksums, and may
114f5514f08SRobert Watson  * cause problems (especially for NFS data blocks).
11574eb3236SWarner Losh  */
11640b676beSBjoern A. Zeeb VNET_DEFINE(int, udp_cksum) = 1;
1176df8a710SGleb Smirnoff SYSCTL_INT(_net_inet_udp, UDPCTL_CHECKSUM, checksum, CTLFLAG_VNET | CTLFLAG_RW,
11840b676beSBjoern A. Zeeb     &VNET_NAME(udp_cksum), 0, "compute udp checksum");
119df8bae1dSRodney W. Grimes 
120334fc582SBjoern A. Zeeb VNET_DEFINE(int, udp_log_in_vain) = 0;
121334fc582SBjoern A. Zeeb SYSCTL_INT(_net_inet_udp, OID_AUTO, log_in_vain, CTLFLAG_VNET | CTLFLAG_RW,
122334fc582SBjoern A. Zeeb     &VNET_NAME(udp_log_in_vain), 0, "Log all incoming UDP packets");
123816a3d83SPoul-Henning Kamp 
12482cea7e6SBjoern A. Zeeb VNET_DEFINE(int, udp_blackhole) = 0;
1256df8a710SGleb Smirnoff SYSCTL_INT(_net_inet_udp, OID_AUTO, blackhole, CTLFLAG_VNET | CTLFLAG_RW,
126eddfbb76SRobert Watson     &VNET_NAME(udp_blackhole), 0,
1273329b236SRobert Watson     "Do not send port unreachables for refused connects");
12816f7f31fSGeoff Rehmet 
12943bbb6aaSRobert Watson u_long	udp_sendspace = 9216;		/* really max datagram size */
13043bbb6aaSRobert Watson SYSCTL_ULONG(_net_inet_udp, UDPCTL_MAXDGRAM, maxdgram, CTLFLAG_RW,
13143bbb6aaSRobert Watson     &udp_sendspace, 0, "Maximum outgoing UDP datagram size");
13243bbb6aaSRobert Watson 
13343bbb6aaSRobert Watson u_long	udp_recvspace = 40 * (1024 +
13443bbb6aaSRobert Watson #ifdef INET6
13543bbb6aaSRobert Watson 				      sizeof(struct sockaddr_in6)
13643bbb6aaSRobert Watson #else
13743bbb6aaSRobert Watson 				      sizeof(struct sockaddr_in)
13843bbb6aaSRobert Watson #endif
139e62b9bcaSSergey Kandaurov 				      );	/* 40 1K datagrams */
14043bbb6aaSRobert Watson 
14143bbb6aaSRobert Watson SYSCTL_ULONG(_net_inet_udp, UDPCTL_RECVSPACE, recvspace, CTLFLAG_RW,
14243bbb6aaSRobert Watson     &udp_recvspace, 0, "Maximum space for incoming UDP datagrams");
14343bbb6aaSRobert Watson 
144eddfbb76SRobert Watson VNET_DEFINE(struct inpcbhead, udb);		/* from udp_var.h */
145eddfbb76SRobert Watson VNET_DEFINE(struct inpcbinfo, udbinfo);
146e06e816fSKevin Lo VNET_DEFINE(struct inpcbhead, ulitecb);
147e06e816fSKevin Lo VNET_DEFINE(struct inpcbinfo, ulitecbinfo);
1485f901c92SAndrew Turner VNET_DEFINE_STATIC(uma_zone_t, udpcb_zone);
1491e77c105SRobert Watson #define	V_udpcb_zone			VNET(udpcb_zone)
15015bd2b43SDavid Greenman 
15115bd2b43SDavid Greenman #ifndef UDBHASHSIZE
152e2ed8f35SAlexander Motin #define	UDBHASHSIZE	128
15315bd2b43SDavid Greenman #endif
15415bd2b43SDavid Greenman 
1555b7cb97cSAndrey V. Elsukov VNET_PCPUSTAT_DEFINE(struct udpstat, udpstat);		/* from udp_var.h */
1565b7cb97cSAndrey V. Elsukov VNET_PCPUSTAT_SYSINIT(udpstat);
1575b7cb97cSAndrey V. Elsukov SYSCTL_VNET_PCPUSTAT(_net_inet_udp, UDPCTL_STATS, stats, struct udpstat,
1585b7cb97cSAndrey V. Elsukov     udpstat, "UDP statistics (struct udpstat, netinet/udp_var.h)");
159f2ea20e6SGarrett Wollman 
1605b7cb97cSAndrey V. Elsukov #ifdef VIMAGE
1615b7cb97cSAndrey V. Elsukov VNET_PCPUSTAT_SYSUNINIT(udpstat);
1625b7cb97cSAndrey V. Elsukov #endif /* VIMAGE */
16379288c11SBjoern A. Zeeb #ifdef INET
164bc725eafSRobert Watson static void	udp_detach(struct socket *so);
1654d77a549SAlfred Perlstein static int	udp_output(struct inpcb *, struct mbuf *, struct sockaddr *,
166a9839c4aSBjoern A. Zeeb 		    struct mbuf *, struct thread *, int);
16779288c11SBjoern A. Zeeb #endif
16879288c11SBjoern A. Zeeb 
1694f590175SPaul Saab static void
1704f590175SPaul Saab udp_zone_change(void *tag)
1714f590175SPaul Saab {
1724f590175SPaul Saab 
173603724d3SBjoern A. Zeeb 	uma_zone_set_max(V_udbinfo.ipi_zone, maxsockets);
1746a9148feSBjoern A. Zeeb 	uma_zone_set_max(V_udpcb_zone, maxsockets);
1754f590175SPaul Saab }
1764f590175SPaul Saab 
177d915b280SStephan Uphoff static int
178d915b280SStephan Uphoff udp_inpcb_init(void *mem, int size, int flags)
179d915b280SStephan Uphoff {
180af1ee11dSRobert Watson 	struct inpcb *inp;
18108651e1fSJohn Baldwin 
182af1ee11dSRobert Watson 	inp = mem;
183d915b280SStephan Uphoff 	INP_LOCK_INIT(inp, "inp", "udpinp");
184d915b280SStephan Uphoff 	return (0);
185d915b280SStephan Uphoff }
186d915b280SStephan Uphoff 
187e06e816fSKevin Lo static int
188e06e816fSKevin Lo udplite_inpcb_init(void *mem, int size, int flags)
189e06e816fSKevin Lo {
190e06e816fSKevin Lo 	struct inpcb *inp;
191e06e816fSKevin Lo 
192e06e816fSKevin Lo 	inp = mem;
193e06e816fSKevin Lo 	INP_LOCK_INIT(inp, "inp", "udpliteinp");
194e06e816fSKevin Lo 	return (0);
195e06e816fSKevin Lo }
196e06e816fSKevin Lo 
197df8bae1dSRodney W. Grimes void
198af1ee11dSRobert Watson udp_init(void)
199df8bae1dSRodney W. Grimes {
200af1ee11dSRobert Watson 
2018ad1a83bSAdrian Chadd 	/*
2028ad1a83bSAdrian Chadd 	 * For now default to 2-tuple UDP hashing - until the fragment
2038ad1a83bSAdrian Chadd 	 * reassembly code can also update the flowid.
2048ad1a83bSAdrian Chadd 	 *
2058ad1a83bSAdrian Chadd 	 * Once we can calculate the flowid that way and re-establish
2068ad1a83bSAdrian Chadd 	 * a 4-tuple, flip this to 4-tuple.
2078ad1a83bSAdrian Chadd 	 */
2089bcd427bSRobert Watson 	in_pcbinfo_init(&V_udbinfo, "udp", &V_udb, UDBHASHSIZE, UDBHASHSIZE,
209cc487c16SGleb Smirnoff 	    "udp_inpcb", udp_inpcb_init, IPI_HASHFIELDS_2TUPLE);
2106a9148feSBjoern A. Zeeb 	V_udpcb_zone = uma_zcreate("udpcb", sizeof(struct udpcb),
211a8da5dd6SCraig Rodrigues 	    NULL, NULL, NULL, NULL, UMA_ALIGN_PTR, 0);
2126a9148feSBjoern A. Zeeb 	uma_zone_set_max(V_udpcb_zone, maxsockets);
2136acd596eSPawel Jakub Dawidek 	uma_zone_set_warning(V_udpcb_zone, "kern.ipc.maxsockets limit reached");
2144f590175SPaul Saab 	EVENTHANDLER_REGISTER(maxsockets_change, udp_zone_change, NULL,
2154f590175SPaul Saab 	    EVENTHANDLER_PRI_ANY);
216df8bae1dSRodney W. Grimes }
217df8bae1dSRodney W. Grimes 
218e06e816fSKevin Lo void
219e06e816fSKevin Lo udplite_init(void)
220e06e816fSKevin Lo {
221e06e816fSKevin Lo 
222e06e816fSKevin Lo 	in_pcbinfo_init(&V_ulitecbinfo, "udplite", &V_ulitecb, UDBHASHSIZE,
223cc487c16SGleb Smirnoff 	    UDBHASHSIZE, "udplite_inpcb", udplite_inpcb_init,
224cc487c16SGleb Smirnoff 	    IPI_HASHFIELDS_2TUPLE);
225e06e816fSKevin Lo }
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) {
38043bbb6aaSRobert Watson 		SOCKBUF_UNLOCK(&so->so_rcv);
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 
3908f5a8818SKevin Lo int
3918f5a8818SKevin Lo udp_input(struct mbuf **mp, int *offp, int proto)
392df8bae1dSRodney W. Grimes {
3933329b236SRobert Watson 	struct ip *ip;
3943329b236SRobert Watson 	struct udphdr *uh;
39571498f30SBruce M Simpson 	struct ifnet *ifp;
3963329b236SRobert Watson 	struct inpcb *inp;
3978f134647SGleb Smirnoff 	uint16_t len, ip_len;
398e06e816fSKevin Lo 	struct inpcbinfo *pcbinfo;
399df8bae1dSRodney W. Grimes 	struct ip save_ip;
400dce33a45SErmal Luçi 	struct sockaddr_in udp_in[2];
4018f5a8818SKevin Lo 	struct mbuf *m;
4020b4ae859SGleb Smirnoff 	struct m_tag *fwd_tag;
4038f5a8818SKevin Lo 	int cscov_partial, iphlen;
404df8bae1dSRodney W. Grimes 
4058f5a8818SKevin Lo 	m = *mp;
4068f5a8818SKevin Lo 	iphlen = *offp;
40771498f30SBruce M Simpson 	ifp = m->m_pkthdr.rcvif;
4088f5a8818SKevin Lo 	*mp = NULL;
409026decb8SRobert Watson 	UDPSTAT_INC(udps_ipackets);
410df8bae1dSRodney W. Grimes 
411df8bae1dSRodney W. Grimes 	/*
4123329b236SRobert Watson 	 * Strip IP options, if any; should skip this, make available to
4133329b236SRobert Watson 	 * user, and use on returned packets, but we don't yet have a way to
4143329b236SRobert Watson 	 * check the checksum with options still present.
415df8bae1dSRodney W. Grimes 	 */
416df8bae1dSRodney W. Grimes 	if (iphlen > sizeof (struct ip)) {
417105bd211SGleb Smirnoff 		ip_stripoptions(m);
418df8bae1dSRodney W. Grimes 		iphlen = sizeof(struct ip);
419df8bae1dSRodney W. Grimes 	}
420df8bae1dSRodney W. Grimes 
421df8bae1dSRodney W. Grimes 	/*
422df8bae1dSRodney W. Grimes 	 * Get IP and UDP header together in first mbuf.
423df8bae1dSRodney W. Grimes 	 */
424df8bae1dSRodney W. Grimes 	if (m->m_len < iphlen + sizeof(struct udphdr)) {
425d1b18731SKevin Lo 		if ((m = m_pullup(m, iphlen + sizeof(struct udphdr))) == NULL) {
426026decb8SRobert Watson 			UDPSTAT_INC(udps_hdrops);
4278f5a8818SKevin Lo 			return (IPPROTO_DONE);
428df8bae1dSRodney W. Grimes 		}
429df8bae1dSRodney W. Grimes 	}
430503f4e47SBjoern A. Zeeb 	ip = mtod(m, struct ip *);
431df8bae1dSRodney W. Grimes 	uh = (struct udphdr *)((caddr_t)ip + iphlen);
4328f5a8818SKevin Lo 	cscov_partial = (proto == IPPROTO_UDPLITE) ? 1 : 0;
433df8bae1dSRodney W. Grimes 
4343329b236SRobert Watson 	/*
4353329b236SRobert Watson 	 * Destination port of 0 is illegal, based on RFC768.
4363329b236SRobert Watson 	 */
437686cdd19SJun-ichiro itojun Hagino 	if (uh->uh_dport == 0)
438f76fcf6dSJeffrey Hsu 		goto badunlocked;
439686cdd19SJun-ichiro itojun Hagino 
440df8bae1dSRodney W. Grimes 	/*
4413329b236SRobert Watson 	 * Construct sockaddr format source address.  Stuff source address
4423329b236SRobert Watson 	 * and datagram in user buffer.
443b9234fafSSam Leffler 	 */
444dce33a45SErmal Luçi 	bzero(&udp_in[0], sizeof(struct sockaddr_in) * 2);
445dce33a45SErmal Luçi 	udp_in[0].sin_len = sizeof(struct sockaddr_in);
446dce33a45SErmal Luçi 	udp_in[0].sin_family = AF_INET;
447dce33a45SErmal Luçi 	udp_in[0].sin_port = uh->uh_sport;
448dce33a45SErmal Luçi 	udp_in[0].sin_addr = ip->ip_src;
449dce33a45SErmal Luçi 	udp_in[1].sin_len = sizeof(struct sockaddr_in);
450dce33a45SErmal Luçi 	udp_in[1].sin_family = AF_INET;
451dce33a45SErmal Luçi 	udp_in[1].sin_port = uh->uh_dport;
452dce33a45SErmal Luçi 	udp_in[1].sin_addr = ip->ip_dst;
453b9234fafSSam Leffler 
454b9234fafSSam Leffler 	/*
455af1ee11dSRobert Watson 	 * Make mbuf data length reflect UDP length.  If not enough data to
456af1ee11dSRobert Watson 	 * reflect UDP length, drop.
457df8bae1dSRodney W. Grimes 	 */
458df8bae1dSRodney W. Grimes 	len = ntohs((u_short)uh->uh_ulen);
4598ad458a4SGleb Smirnoff 	ip_len = ntohs(ip->ip_len) - iphlen;
4600f4a0366SMichael Tuexen 	if (proto == IPPROTO_UDPLITE && (len == 0 || len == ip_len)) {
461e06e816fSKevin Lo 		/* Zero means checksum over the complete packet. */
4620f4a0366SMichael Tuexen 		if (len == 0)
463e06e816fSKevin Lo 			len = ip_len;
464e06e816fSKevin Lo 		cscov_partial = 0;
465e06e816fSKevin Lo 	}
4668f134647SGleb Smirnoff 	if (ip_len != len) {
4678f134647SGleb Smirnoff 		if (len > ip_len || len < sizeof(struct udphdr)) {
468026decb8SRobert Watson 			UDPSTAT_INC(udps_badlen);
469f76fcf6dSJeffrey Hsu 			goto badunlocked;
470df8bae1dSRodney W. Grimes 		}
4718f5a8818SKevin Lo 		if (proto == IPPROTO_UDP)
4728f134647SGleb Smirnoff 			m_adj(m, len - ip_len);
473df8bae1dSRodney W. Grimes 	}
4743329b236SRobert Watson 
475df8bae1dSRodney W. Grimes 	/*
4763329b236SRobert Watson 	 * Save a copy of the IP header in case we want restore it for
4773329b236SRobert Watson 	 * sending an ICMP error message in response.
478df8bae1dSRodney W. Grimes 	 */
479603724d3SBjoern A. Zeeb 	if (!V_udp_blackhole)
480df8bae1dSRodney W. Grimes 		save_ip = *ip;
481cce418d3SMatt Jacob 	else
482cce418d3SMatt Jacob 		memset(&save_ip, 0, sizeof(save_ip));
483df8bae1dSRodney W. Grimes 
484df8bae1dSRodney W. Grimes 	/*
485df8bae1dSRodney W. Grimes 	 * Checksum extended UDP header and data.
486df8bae1dSRodney W. Grimes 	 */
4876dfab5b1SGarrett Wollman 	if (uh->uh_sum) {
48839629c92SDavid Malone 		u_short uh_sum;
48939629c92SDavid Malone 
490e06e816fSKevin Lo 		if ((m->m_pkthdr.csum_flags & CSUM_DATA_VALID) &&
491e06e816fSKevin Lo 		    !cscov_partial) {
492db4f9cc7SJonathan Lemon 			if (m->m_pkthdr.csum_flags & CSUM_PSEUDO_HDR)
49339629c92SDavid Malone 				uh_sum = m->m_pkthdr.csum_data;
494db4f9cc7SJonathan Lemon 			else
49539629c92SDavid Malone 				uh_sum = in_pseudo(ip->ip_src.s_addr,
496506f4949SRuslan Ermilov 				    ip->ip_dst.s_addr, htonl((u_short)len +
4978f5a8818SKevin Lo 				    m->m_pkthdr.csum_data + proto));
49839629c92SDavid Malone 			uh_sum ^= 0xffff;
499db4f9cc7SJonathan Lemon 		} else {
500cb342100SHajimu UMEMOTO 			char b[9];
501af1ee11dSRobert Watson 
502cb342100SHajimu UMEMOTO 			bcopy(((struct ipovly *)ip)->ih_x1, b, 9);
5036effc713SDoug Rabson 			bzero(((struct ipovly *)ip)->ih_x1, 9);
5048f5a8818SKevin Lo 			((struct ipovly *)ip)->ih_len = (proto == IPPROTO_UDP) ?
505e06e816fSKevin Lo 			    uh->uh_ulen : htons(ip_len);
50639629c92SDavid Malone 			uh_sum = in_cksum(m, len + sizeof (struct ip));
507cb342100SHajimu UMEMOTO 			bcopy(b, ((struct ipovly *)ip)->ih_x1, 9);
508db4f9cc7SJonathan Lemon 		}
50939629c92SDavid Malone 		if (uh_sum) {
510026decb8SRobert Watson 			UDPSTAT_INC(udps_badsum);
511df8bae1dSRodney W. Grimes 			m_freem(m);
5128f5a8818SKevin Lo 			return (IPPROTO_DONE);
513df8bae1dSRodney W. Grimes 		}
514c6d81a34SMichael Tuexen 	} else {
515c6d81a34SMichael Tuexen 		if (proto == IPPROTO_UDP) {
516026decb8SRobert Watson 			UDPSTAT_INC(udps_nosum);
517c6d81a34SMichael Tuexen 		} else {
518c6d81a34SMichael Tuexen 			/* UDPLite requires a checksum */
519c6d81a34SMichael Tuexen 			/* XXX: What is the right UDPLite MIB counter here? */
520c6d81a34SMichael Tuexen 			m_freem(m);
521c6d81a34SMichael Tuexen 			return (IPPROTO_DONE);
522c6d81a34SMichael Tuexen 		}
523c6d81a34SMichael Tuexen 	}
524df8bae1dSRodney W. Grimes 
525a86e5c96SBjoern A. Zeeb 	pcbinfo = udp_get_inpcbinfo(proto);
526df8bae1dSRodney W. Grimes 	if (IN_MULTICAST(ntohl(ip->ip_dst.s_addr)) ||
5274af540d1SRyan Stone 	    in_broadcast(ip->ip_dst, ifp)) {
52882c23ebaSBill Fenner 		struct inpcb *last;
529e06e816fSKevin Lo 		struct inpcbhead *pcblist;
5303329b236SRobert Watson 
53180577e55SGleb Smirnoff 		NET_EPOCH_ASSERT();
53280577e55SGleb Smirnoff 
533a86e5c96SBjoern A. Zeeb 		pcblist = udp_get_pcblist(proto);
534df8bae1dSRodney W. Grimes 		last = NULL;
535b872626dSMatt Macy 		CK_LIST_FOREACH(inp, pcblist, inp_list) {
5369c1df695SRobert Watson 			if (inp->inp_lport != uh->uh_dport)
537f76fcf6dSJeffrey Hsu 				continue;
538cfa1ca9dSYoshinobu Inoue #ifdef INET6
539369dc8ceSEivind Eklund 			if ((inp->inp_vflag & INP_IPV4) == 0)
5409c1df695SRobert Watson 				continue;
541cfa1ca9dSYoshinobu Inoue #endif
54271498f30SBruce M Simpson 			if (inp->inp_laddr.s_addr != INADDR_ANY &&
54371498f30SBruce M Simpson 			    inp->inp_laddr.s_addr != ip->ip_dst.s_addr)
5449c1df695SRobert Watson 				continue;
54571498f30SBruce M Simpson 			if (inp->inp_faddr.s_addr != INADDR_ANY &&
54671498f30SBruce M Simpson 			    inp->inp_faddr.s_addr != ip->ip_src.s_addr)
54771498f30SBruce M Simpson 				continue;
54871498f30SBruce M Simpson 			if (inp->inp_fport != 0 &&
549df8bae1dSRodney W. Grimes 			    inp->inp_fport != uh->uh_sport)
5509c1df695SRobert Watson 				continue;
55171498f30SBruce M Simpson 
552119d85f6SRobert Watson 			INP_RLOCK(inp);
553df8bae1dSRodney W. Grimes 
5544ba16a92SBjoern A. Zeeb 			if (__predict_false(inp->inp_flags2 & INP_FREED)) {
5554ba16a92SBjoern A. Zeeb 				INP_RUNLOCK(inp);
5564ba16a92SBjoern A. Zeeb 				continue;
5574ba16a92SBjoern A. Zeeb 			}
5584ba16a92SBjoern A. Zeeb 
55983453a06SBruce M Simpson 			/*
560fa046d87SRobert Watson 			 * XXXRW: Because we weren't holding either the inpcb
561fa046d87SRobert Watson 			 * or the hash lock when we checked for a match
562fa046d87SRobert Watson 			 * before, we should probably recheck now that the
563fa046d87SRobert Watson 			 * inpcb lock is held.
564fa046d87SRobert Watson 			 */
565fa046d87SRobert Watson 
566fa046d87SRobert Watson 			/*
56771498f30SBruce M Simpson 			 * Handle socket delivery policy for any-source
56871498f30SBruce M Simpson 			 * and source-specific multicast. [RFC3678]
56983453a06SBruce M Simpson 			 */
570a38b1c8cSRandall Stewart 			if (IN_MULTICAST(ntohl(ip->ip_dst.s_addr))) {
5713afdfcafSBjoern A. Zeeb 				struct ip_moptions	*imo;
572d10910e6SBruce M Simpson 				struct sockaddr_in	 group;
573d10910e6SBruce M Simpson 				int			 blocked;
5743afdfcafSBjoern A. Zeeb 
5753afdfcafSBjoern A. Zeeb 				imo = inp->inp_moptions;
576a38b1c8cSRandall Stewart 				if (imo == NULL) {
577a38b1c8cSRandall Stewart 					INP_RUNLOCK(inp);
578a38b1c8cSRandall Stewart 					continue;
579a38b1c8cSRandall Stewart 				}
580d10910e6SBruce M Simpson 				bzero(&group, sizeof(struct sockaddr_in));
581d10910e6SBruce M Simpson 				group.sin_len = sizeof(struct sockaddr_in);
582d10910e6SBruce M Simpson 				group.sin_family = AF_INET;
583d10910e6SBruce M Simpson 				group.sin_addr = ip->ip_dst;
58471498f30SBruce M Simpson 
585d10910e6SBruce M Simpson 				blocked = imo_multi_filter(imo, ifp,
586d10910e6SBruce M Simpson 					(struct sockaddr *)&group,
587dce33a45SErmal Luçi 					(struct sockaddr *)&udp_in[0]);
588d10910e6SBruce M Simpson 				if (blocked != MCAST_PASS) {
589d10910e6SBruce M Simpson 					if (blocked == MCAST_NOTGMEMBER)
59086425c62SRobert Watson 						IPSTAT_INC(ips_notmember);
591d10910e6SBruce M Simpson 					if (blocked == MCAST_NOTSMEMBER ||
592d10910e6SBruce M Simpson 					    blocked == MCAST_MUTED)
593026decb8SRobert Watson 						UDPSTAT_INC(udps_filtermcast);
594119d85f6SRobert Watson 					INP_RUNLOCK(inp);
5959c1df695SRobert Watson 					continue;
5969c1df695SRobert Watson 				}
59783453a06SBruce M Simpson 			}
598df8bae1dSRodney W. Grimes 			if (last != NULL) {
599df8bae1dSRodney W. Grimes 				struct mbuf *n;
600df8bae1dSRodney W. Grimes 
601c3bef61eSKevin Lo 				if ((n = m_copym(m, 0, M_COPYALL, M_NOWAIT)) !=
602c3bef61eSKevin Lo 				    NULL) {
6037bda9663SMichael Tuexen 					if (proto == IPPROTO_UDPLITE)
6047bda9663SMichael Tuexen 						UDPLITE_PROBE(receive, NULL, last, ip,
605a0a9e1b5SBryan Venteicher 						    last, uh);
6067bda9663SMichael Tuexen 					else
6077bda9663SMichael Tuexen 						UDP_PROBE(receive, NULL, last, ip, last,
6087bda9663SMichael Tuexen 						    uh);
609c0d1be08SRandall Stewart 					if (udp_append(last, ip, n, iphlen,
610dce33a45SErmal Luçi 						udp_in)) {
611c0d1be08SRandall Stewart 						goto inp_lost;
612c0d1be08SRandall Stewart 					}
613c19f98ebSBryan Venteicher 				}
6146a9148feSBjoern A. Zeeb 				INP_RUNLOCK(last);
615df8bae1dSRodney W. Grimes 			}
61682c23ebaSBill Fenner 			last = inp;
617df8bae1dSRodney W. Grimes 			/*
618df8bae1dSRodney W. Grimes 			 * Don't look for additional matches if this one does
619df8bae1dSRodney W. Grimes 			 * not have either the SO_REUSEPORT or SO_REUSEADDR
6203329b236SRobert Watson 			 * socket options set.  This heuristic avoids
6213329b236SRobert Watson 			 * searching through all pcbs in the common case of a
6223329b236SRobert Watson 			 * non-shared port.  It assumes that an application
6233329b236SRobert Watson 			 * will never clear these options after setting them.
624df8bae1dSRodney W. Grimes 			 */
6253329b236SRobert Watson 			if ((last->inp_socket->so_options &
6261a43cff9SSean Bruno 			    (SO_REUSEPORT|SO_REUSEPORT_LB|SO_REUSEADDR)) == 0)
627df8bae1dSRodney W. Grimes 				break;
628df8bae1dSRodney W. Grimes 		}
629df8bae1dSRodney W. Grimes 
630df8bae1dSRodney W. Grimes 		if (last == NULL) {
631df8bae1dSRodney W. Grimes 			/*
6323329b236SRobert Watson 			 * No matching pcb found; discard datagram.  (No need
6333329b236SRobert Watson 			 * to send an ICMP Port Unreachable for a broadcast
6343329b236SRobert Watson 			 * or multicast datgram.)
635df8bae1dSRodney W. Grimes 			 */
636026decb8SRobert Watson 			UDPSTAT_INC(udps_noportbcast);
637fa046d87SRobert Watson 			if (inp)
638fa046d87SRobert Watson 				INP_RUNLOCK(inp);
639fa046d87SRobert Watson 			goto badunlocked;
640df8bae1dSRodney W. Grimes 		}
6417bda9663SMichael Tuexen 		if (proto == IPPROTO_UDPLITE)
6427bda9663SMichael Tuexen 			UDPLITE_PROBE(receive, NULL, last, ip, last, uh);
6437bda9663SMichael Tuexen 		else
644a0a9e1b5SBryan Venteicher 			UDP_PROBE(receive, NULL, last, ip, last, uh);
645dce33a45SErmal Luçi 		if (udp_append(last, ip, m, iphlen, udp_in) == 0)
646c7c7ea4bSRandall Stewart 			INP_RUNLOCK(last);
647c0d1be08SRandall Stewart 	inp_lost:
6488f5a8818SKevin Lo 		return (IPPROTO_DONE);
649df8bae1dSRodney W. Grimes 	}
6503329b236SRobert Watson 
651df8bae1dSRodney W. Grimes 	/*
6526d6a026bSDavid Greenman 	 * Locate pcb for datagram.
653df8bae1dSRodney W. Grimes 	 */
654c1de64a4SAndrey V. Elsukov 
6558a006adbSBjoern A. Zeeb 	/*
6568a006adbSBjoern A. Zeeb 	 * Grab info from PACKET_TAG_IPFORWARD tag prepended to the chain.
6578a006adbSBjoern A. Zeeb 	 */
658ffdbf9daSAndrey V. Elsukov 	if ((m->m_flags & M_IP_NEXTHOP) &&
659c1de64a4SAndrey V. Elsukov 	    (fwd_tag = m_tag_find(m, PACKET_TAG_IPFORWARD, NULL)) != NULL) {
6608a006adbSBjoern A. Zeeb 		struct sockaddr_in *next_hop;
6618a006adbSBjoern A. Zeeb 
6628a006adbSBjoern A. Zeeb 		next_hop = (struct sockaddr_in *)(fwd_tag + 1);
6638a006adbSBjoern A. Zeeb 
6648a006adbSBjoern A. Zeeb 		/*
6658a006adbSBjoern A. Zeeb 		 * Transparently forwarded. Pretend to be the destination.
6668a006adbSBjoern A. Zeeb 		 * Already got one like this?
6678a006adbSBjoern A. Zeeb 		 */
668e06e816fSKevin Lo 		inp = in_pcblookup_mbuf(pcbinfo, ip->ip_src, uh->uh_sport,
6698a006adbSBjoern A. Zeeb 		    ip->ip_dst, uh->uh_dport, INPLOOKUP_RLOCKPCB, ifp, m);
6708a006adbSBjoern A. Zeeb 		if (!inp) {
6718a006adbSBjoern A. Zeeb 			/*
6728a006adbSBjoern A. Zeeb 			 * It's new.  Try to find the ambushing socket.
6738a006adbSBjoern A. Zeeb 			 * Because we've rewritten the destination address,
6748a006adbSBjoern A. Zeeb 			 * any hardware-generated hash is ignored.
6758a006adbSBjoern A. Zeeb 			 */
676e06e816fSKevin Lo 			inp = in_pcblookup(pcbinfo, ip->ip_src,
6778a006adbSBjoern A. Zeeb 			    uh->uh_sport, next_hop->sin_addr,
6788a006adbSBjoern A. Zeeb 			    next_hop->sin_port ? htons(next_hop->sin_port) :
6798a006adbSBjoern A. Zeeb 			    uh->uh_dport, INPLOOKUP_WILDCARD |
6808a006adbSBjoern A. Zeeb 			    INPLOOKUP_RLOCKPCB, ifp);
6818a006adbSBjoern A. Zeeb 		}
6828a006adbSBjoern A. Zeeb 		/* Remove the tag from the packet. We don't need it anymore. */
6838a006adbSBjoern A. Zeeb 		m_tag_delete(m, fwd_tag);
684ffdbf9daSAndrey V. Elsukov 		m->m_flags &= ~M_IP_NEXTHOP;
6858a006adbSBjoern A. Zeeb 	} else
686e06e816fSKevin Lo 		inp = in_pcblookup_mbuf(pcbinfo, ip->ip_src, uh->uh_sport,
6878a006adbSBjoern A. Zeeb 		    ip->ip_dst, uh->uh_dport, INPLOOKUP_WILDCARD |
6888a006adbSBjoern A. Zeeb 		    INPLOOKUP_RLOCKPCB, ifp, m);
68915bd2b43SDavid Greenman 	if (inp == NULL) {
690334fc582SBjoern A. Zeeb 		if (V_udp_log_in_vain) {
691edf0313bSEric van Gyzen 			char src[INET_ADDRSTRLEN];
692edf0313bSEric van Gyzen 			char dst[INET_ADDRSTRLEN];
69375cfc95fSAndrey A. Chernov 
694592071e8SBruce Evans 			log(LOG_INFO,
695592071e8SBruce Evans 			    "Connection attempt to UDP %s:%d from %s:%d\n",
696edf0313bSEric van Gyzen 			    inet_ntoa_r(ip->ip_dst, dst), ntohs(uh->uh_dport),
697edf0313bSEric van Gyzen 			    inet_ntoa_r(ip->ip_src, src), ntohs(uh->uh_sport));
69875cfc95fSAndrey A. Chernov 		}
6997bda9663SMichael Tuexen 		if (proto == IPPROTO_UDPLITE)
7007bda9663SMichael Tuexen 			UDPLITE_PROBE(receive, NULL, NULL, ip, NULL, uh);
7017bda9663SMichael Tuexen 		else
702e1526d5aSMichael Tuexen 			UDP_PROBE(receive, NULL, NULL, ip, NULL, uh);
703026decb8SRobert Watson 		UDPSTAT_INC(udps_noport);
704df8bae1dSRodney W. Grimes 		if (m->m_flags & (M_BCAST | M_MCAST)) {
705026decb8SRobert Watson 			UDPSTAT_INC(udps_noportbcast);
706fa046d87SRobert Watson 			goto badunlocked;
707df8bae1dSRodney W. Grimes 		}
708603724d3SBjoern A. Zeeb 		if (V_udp_blackhole)
709fa046d87SRobert Watson 			goto badunlocked;
7101cbd978eSLuigi Rizzo 		if (badport_bandlim(BANDLIM_ICMP_UNREACH) < 0)
711fa046d87SRobert Watson 			goto badunlocked;
71204287599SRuslan Ermilov 		*ip = save_ip;
713582a7760SBruce Evans 		icmp_error(m, ICMP_UNREACH, ICMP_UNREACH_PORT, 0, 0);
7148f5a8818SKevin Lo 		return (IPPROTO_DONE);
715df8bae1dSRodney W. Grimes 	}
7163329b236SRobert Watson 
7173329b236SRobert Watson 	/*
7183329b236SRobert Watson 	 * Check the minimum TTL for socket.
7193329b236SRobert Watson 	 */
720fa046d87SRobert Watson 	INP_RLOCK_ASSERT(inp);
72110cc62b7SRobert Watson 	if (inp->inp_ip_minttl && inp->inp_ip_minttl > ip->ip_ttl) {
7227bda9663SMichael Tuexen 		if (proto == IPPROTO_UDPLITE)
7237bda9663SMichael Tuexen 			UDPLITE_PROBE(receive, NULL, inp, ip, inp, uh);
7247bda9663SMichael Tuexen 		else
725e1526d5aSMichael Tuexen 			UDP_PROBE(receive, NULL, inp, ip, inp, uh);
72610cc62b7SRobert Watson 		INP_RUNLOCK(inp);
727fa046d87SRobert Watson 		m_freem(m);
7288f5a8818SKevin Lo 		return (IPPROTO_DONE);
72910cc62b7SRobert Watson 	}
730e06e816fSKevin Lo 	if (cscov_partial) {
731e06e816fSKevin Lo 		struct udpcb *up;
732e06e816fSKevin Lo 
733e06e816fSKevin Lo 		up = intoudpcb(inp);
73483e95fb3SMichael Tuexen 		if (up->u_rxcslen == 0 || up->u_rxcslen > len) {
735e06e816fSKevin Lo 			INP_RUNLOCK(inp);
736e06e816fSKevin Lo 			m_freem(m);
7378f5a8818SKevin Lo 			return (IPPROTO_DONE);
738e06e816fSKevin Lo 		}
739e06e816fSKevin Lo 	}
74057f60867SMark Johnston 
7417bda9663SMichael Tuexen 	if (proto == IPPROTO_UDPLITE)
7427bda9663SMichael Tuexen 		UDPLITE_PROBE(receive, NULL, inp, ip, inp, uh);
7437bda9663SMichael Tuexen 	else
7441ad19fb6SMark Johnston 		UDP_PROBE(receive, NULL, inp, ip, inp, uh);
745dce33a45SErmal Luçi 	if (udp_append(inp, ip, m, iphlen, udp_in) == 0)
746119d85f6SRobert Watson 		INP_RUNLOCK(inp);
7478f5a8818SKevin Lo 	return (IPPROTO_DONE);
74861ffc0b1SJeffrey Hsu 
749f76fcf6dSJeffrey Hsu badunlocked:
750df8bae1dSRodney W. Grimes 	m_freem(m);
7518f5a8818SKevin Lo 	return (IPPROTO_DONE);
752cfa1ca9dSYoshinobu Inoue }
75379288c11SBjoern A. Zeeb #endif /* INET */
754cfa1ca9dSYoshinobu Inoue 
755cfa1ca9dSYoshinobu Inoue /*
7563329b236SRobert Watson  * Notify a udp user of an asynchronous error; just wake up so that they can
7573329b236SRobert Watson  * collect error status.
758df8bae1dSRodney W. Grimes  */
7593ce144eaSJeffrey Hsu struct inpcb *
7603329b236SRobert Watson udp_notify(struct inpcb *inp, int errno)
761df8bae1dSRodney W. Grimes {
7623329b236SRobert Watson 
763083a010cSRyan Stone 	INP_WLOCK_ASSERT(inp);
76484cc0778SGeorge V. Neville-Neil 	if ((errno == EHOSTUNREACH || errno == ENETUNREACH ||
765983066f0SAlexander V. Chernikov 	     errno == EHOSTDOWN) && inp->inp_route.ro_nh) {
766983066f0SAlexander V. Chernikov 		NH_FREE(inp->inp_route.ro_nh);
767983066f0SAlexander V. Chernikov 		inp->inp_route.ro_nh = (struct nhop_object *)NULL;
76884cc0778SGeorge V. Neville-Neil 	}
7698501a69cSRobert Watson 
770df8bae1dSRodney W. Grimes 	inp->inp_socket->so_error = errno;
771df8bae1dSRodney W. Grimes 	sorwakeup(inp->inp_socket);
772df8bae1dSRodney W. Grimes 	sowwakeup(inp->inp_socket);
7733329b236SRobert Watson 	return (inp);
774df8bae1dSRodney W. Grimes }
775df8bae1dSRodney W. Grimes 
77679288c11SBjoern A. Zeeb #ifdef INET
777e06e816fSKevin Lo static void
778e06e816fSKevin Lo udp_common_ctlinput(int cmd, struct sockaddr *sa, void *vip,
779e06e816fSKevin Lo     struct inpcbinfo *pcbinfo)
780df8bae1dSRodney W. Grimes {
781c693a045SJonathan Lemon 	struct ip *ip = vip;
782c693a045SJonathan Lemon 	struct udphdr *uh;
783c693a045SJonathan Lemon 	struct in_addr faddr;
784c693a045SJonathan Lemon 	struct inpcb *inp;
785c693a045SJonathan Lemon 
786c693a045SJonathan Lemon 	faddr = ((struct sockaddr_in *)sa)->sin_addr;
787c693a045SJonathan Lemon 	if (sa->sa_family != AF_INET || faddr.s_addr == INADDR_ANY)
788c693a045SJonathan Lemon 		return;
789df8bae1dSRodney W. Grimes 
79084cc0778SGeorge V. Neville-Neil 	if (PRC_IS_REDIRECT(cmd)) {
79184cc0778SGeorge V. Neville-Neil 		/* signal EHOSTDOWN, as it flushes the cached route */
7924f321dbdSBjoern A. Zeeb 		in_pcbnotifyall(&V_udbinfo, faddr, EHOSTDOWN, udp_notify);
79397d8d152SAndre Oppermann 		return;
79484cc0778SGeorge V. Neville-Neil 	}
7953329b236SRobert Watson 
79697d8d152SAndre Oppermann 	/*
79797d8d152SAndre Oppermann 	 * Hostdead is ugly because it goes linearly through all PCBs.
7983329b236SRobert Watson 	 *
7993329b236SRobert Watson 	 * XXX: We never get this from ICMP, otherwise it makes an excellent
8003329b236SRobert Watson 	 * DoS attack on machines with many connections.
80197d8d152SAndre Oppermann 	 */
80297d8d152SAndre Oppermann 	if (cmd == PRC_HOSTDEAD)
803af1ee11dSRobert Watson 		ip = NULL;
804d1c54148SJesper Skriver 	else if ((unsigned)cmd >= PRC_NCMDS || inetctlerrmap[cmd] == 0)
805df8bae1dSRodney W. Grimes 		return;
806af1ee11dSRobert Watson 	if (ip != NULL) {
807df8bae1dSRodney W. Grimes 		uh = (struct udphdr *)((caddr_t)ip + (ip->ip_hl << 2));
808e06e816fSKevin Lo 		inp = in_pcblookup(pcbinfo, faddr, uh->uh_dport,
809083a010cSRyan Stone 		    ip->ip_src, uh->uh_sport, INPLOOKUP_WLOCKPCB, NULL);
810f76fcf6dSJeffrey Hsu 		if (inp != NULL) {
811083a010cSRyan Stone 			INP_WLOCK_ASSERT(inp);
812f76fcf6dSJeffrey Hsu 			if (inp->inp_socket != NULL) {
813f5514f08SRobert Watson 				udp_notify(inp, inetctlerrmap[cmd]);
814f76fcf6dSJeffrey Hsu 			}
815083a010cSRyan Stone 			INP_WUNLOCK(inp);
816abb901c5SRandall Stewart 		} else {
817abb901c5SRandall Stewart 			inp = in_pcblookup(pcbinfo, faddr, uh->uh_dport,
818abb901c5SRandall Stewart 					   ip->ip_src, uh->uh_sport,
819abb901c5SRandall Stewart 					   INPLOOKUP_WILDCARD | INPLOOKUP_RLOCKPCB, NULL);
820abb901c5SRandall Stewart 			if (inp != NULL) {
821abb901c5SRandall Stewart 				struct udpcb *up;
82246374cbfSMatt Macy 				void *ctx;
82346374cbfSMatt Macy 				udp_tun_icmp_t func;
824abb901c5SRandall Stewart 
825abb901c5SRandall Stewart 				up = intoudpcb(inp);
82646374cbfSMatt Macy 				ctx = up->u_tun_ctx;
82746374cbfSMatt Macy 				func = up->u_icmp_func;
828abb901c5SRandall Stewart 				INP_RUNLOCK(inp);
82946374cbfSMatt Macy 				if (func != NULL)
83046374cbfSMatt Macy 					(*func)(cmd, sa, vip, ctx);
831abb901c5SRandall Stewart 			}
832f76fcf6dSJeffrey Hsu 		}
833df8bae1dSRodney W. Grimes 	} else
834e06e816fSKevin Lo 		in_pcbnotifyall(pcbinfo, faddr, inetctlerrmap[cmd],
835f5514f08SRobert Watson 		    udp_notify);
836df8bae1dSRodney W. Grimes }
837e06e816fSKevin Lo void
838e06e816fSKevin Lo udp_ctlinput(int cmd, struct sockaddr *sa, void *vip)
839e06e816fSKevin Lo {
840e06e816fSKevin Lo 
841e06e816fSKevin Lo 	return (udp_common_ctlinput(cmd, sa, vip, &V_udbinfo));
842e06e816fSKevin Lo }
843e06e816fSKevin Lo 
844e06e816fSKevin Lo void
845e06e816fSKevin Lo udplite_ctlinput(int cmd, struct sockaddr *sa, void *vip)
846e06e816fSKevin Lo {
847e06e816fSKevin Lo 
848e06e816fSKevin Lo 	return (udp_common_ctlinput(cmd, sa, vip, &V_ulitecbinfo));
849e06e816fSKevin Lo }
85079288c11SBjoern A. Zeeb #endif /* INET */
851df8bae1dSRodney W. Grimes 
8520312fbe9SPoul-Henning Kamp static int
85382d9ae4eSPoul-Henning Kamp udp_pcblist(SYSCTL_HANDLER_ARGS)
85498271db4SGarrett Wollman {
85598271db4SGarrett Wollman 	struct xinpgen xig;
8566573d758SMatt Macy 	struct epoch_tracker et;
857032677ceSGleb Smirnoff 	struct inpcb *inp;
858032677ceSGleb Smirnoff 	int error;
85998271db4SGarrett Wollman 
860032677ceSGleb Smirnoff 	if (req->newptr != 0)
861032677ceSGleb Smirnoff 		return (EPERM);
862032677ceSGleb Smirnoff 
86398271db4SGarrett Wollman 	if (req->oldptr == 0) {
864032677ceSGleb Smirnoff 		int n;
865032677ceSGleb Smirnoff 
866603724d3SBjoern A. Zeeb 		n = V_udbinfo.ipi_count;
867c007b96aSJohn Baldwin 		n += imax(n / 8, 10);
868c007b96aSJohn Baldwin 		req->oldidx = 2 * (sizeof xig) + n * sizeof(struct xinpcb);
8693329b236SRobert Watson 		return (0);
87098271db4SGarrett Wollman 	}
87198271db4SGarrett Wollman 
872032677ceSGleb Smirnoff 	if ((error = sysctl_wire_old_buffer(req, 0)) != 0)
87347934cefSDon Lewis 		return (error);
8745c38b6dbSDon Lewis 
87579db6fe7SMark Johnston 	bzero(&xig, sizeof(xig));
87698271db4SGarrett Wollman 	xig.xig_len = sizeof xig;
877032677ceSGleb Smirnoff 	xig.xig_count = V_udbinfo.ipi_count;
878032677ceSGleb Smirnoff 	xig.xig_gen = V_udbinfo.ipi_gencnt;
87998271db4SGarrett Wollman 	xig.xig_sogen = so_gencnt;
88098271db4SGarrett Wollman 	error = SYSCTL_OUT(req, &xig, sizeof xig);
88198271db4SGarrett Wollman 	if (error)
8823329b236SRobert Watson 		return (error);
88399208b82SMatt Macy 
884032677ceSGleb Smirnoff 	NET_EPOCH_ENTER(et);
885032677ceSGleb Smirnoff 	for (inp = CK_LIST_FIRST(V_udbinfo.ipi_listhead);
886032677ceSGleb Smirnoff 	    inp != NULL;
887b872626dSMatt Macy 	    inp = CK_LIST_NEXT(inp, inp_list)) {
8889622e84fSRobert Watson 		INP_RLOCK(inp);
889032677ceSGleb Smirnoff 		if (inp->inp_gencnt <= xig.xig_gen &&
890032677ceSGleb Smirnoff 		    cr_canseeinpcb(req->td->td_ucred, inp) == 0) {
89198271db4SGarrett Wollman 			struct xinpcb xi;
892d0e157f6SBjoern A. Zeeb 
893cc65eb4eSGleb Smirnoff 			in_pcbtoxinpcb(inp, &xi);
8949622e84fSRobert Watson 			INP_RUNLOCK(inp);
89598271db4SGarrett Wollman 			error = SYSCTL_OUT(req, &xi, sizeof xi);
896032677ceSGleb Smirnoff 			if (error)
897032677ceSGleb Smirnoff 				break;
898d915b280SStephan Uphoff 		} else
8999622e84fSRobert Watson 			INP_RUNLOCK(inp);
90098271db4SGarrett Wollman 	}
901032677ceSGleb Smirnoff 	NET_EPOCH_EXIT(et);
902d0e157f6SBjoern A. Zeeb 
90398271db4SGarrett Wollman 	if (!error) {
90498271db4SGarrett Wollman 		/*
9053329b236SRobert Watson 		 * Give the user an updated idea of our state.  If the
9063329b236SRobert Watson 		 * generation differs from what we told her before, she knows
9073329b236SRobert Watson 		 * that something happened while we were processing this
9083329b236SRobert Watson 		 * request, and it might be necessary to retry.
90998271db4SGarrett Wollman 		 */
910603724d3SBjoern A. Zeeb 		xig.xig_gen = V_udbinfo.ipi_gencnt;
91198271db4SGarrett Wollman 		xig.xig_sogen = so_gencnt;
912603724d3SBjoern A. Zeeb 		xig.xig_count = V_udbinfo.ipi_count;
91398271db4SGarrett Wollman 		error = SYSCTL_OUT(req, &xig, sizeof xig);
91498271db4SGarrett Wollman 	}
915032677ceSGleb Smirnoff 
9163329b236SRobert Watson 	return (error);
91798271db4SGarrett Wollman }
91898271db4SGarrett Wollman 
91979c3d51bSMatthew D Fleming SYSCTL_PROC(_net_inet_udp, UDPCTL_PCBLIST, pcblist,
9207029da5cSPawel Biernacki     CTLTYPE_OPAQUE | CTLFLAG_RD | CTLFLAG_MPSAFE, NULL, 0,
9217029da5cSPawel Biernacki     udp_pcblist, "S,xinpcb",
9227029da5cSPawel Biernacki     "List of active UDP sockets");
92398271db4SGarrett Wollman 
92479288c11SBjoern A. Zeeb #ifdef INET
92598271db4SGarrett Wollman static int
92682d9ae4eSPoul-Henning Kamp udp_getcred(SYSCTL_HANDLER_ARGS)
927490d50b6SBrian Feldman {
928c0511d3bSBrian Feldman 	struct xucred xuc;
929490d50b6SBrian Feldman 	struct sockaddr_in addrs[2];
930d797164aSGleb Smirnoff 	struct epoch_tracker et;
931490d50b6SBrian Feldman 	struct inpcb *inp;
932277afaffSRobert Watson 	int error;
933490d50b6SBrian Feldman 
93432f9753cSRobert Watson 	error = priv_check(req->td, PRIV_NETINET_GETCRED);
935490d50b6SBrian Feldman 	if (error)
936490d50b6SBrian Feldman 		return (error);
937490d50b6SBrian Feldman 	error = SYSCTL_IN(req, addrs, sizeof(addrs));
938490d50b6SBrian Feldman 	if (error)
939490d50b6SBrian Feldman 		return (error);
940d797164aSGleb Smirnoff 	NET_EPOCH_ENTER(et);
941fa046d87SRobert Watson 	inp = in_pcblookup(&V_udbinfo, addrs[1].sin_addr, addrs[1].sin_port,
942fa046d87SRobert Watson 	    addrs[0].sin_addr, addrs[0].sin_port,
943fa046d87SRobert Watson 	    INPLOOKUP_WILDCARD | INPLOOKUP_RLOCKPCB, NULL);
944d797164aSGleb Smirnoff 	NET_EPOCH_EXIT(et);
9459622e84fSRobert Watson 	if (inp != NULL) {
946fa046d87SRobert Watson 		INP_RLOCK_ASSERT(inp);
9479622e84fSRobert Watson 		if (inp->inp_socket == NULL)
9489622e84fSRobert Watson 			error = ENOENT;
9499622e84fSRobert Watson 		if (error == 0)
950f08ef6c5SBjoern A. Zeeb 			error = cr_canseeinpcb(req->td->td_ucred, inp);
9519622e84fSRobert Watson 		if (error == 0)
95286d02c5cSBjoern A. Zeeb 			cru2x(inp->inp_cred, &xuc);
9539622e84fSRobert Watson 		INP_RUNLOCK(inp);
954fa046d87SRobert Watson 	} else
9559622e84fSRobert Watson 		error = ENOENT;
9560e1eebb8SDon Lewis 	if (error == 0)
9570e1eebb8SDon Lewis 		error = SYSCTL_OUT(req, &xuc, sizeof(struct xucred));
958490d50b6SBrian Feldman 	return (error);
959490d50b6SBrian Feldman }
960490d50b6SBrian Feldman 
9617ce87f12SDavid Malone SYSCTL_PROC(_net_inet_udp, OID_AUTO, getcred,
9627029da5cSPawel Biernacki     CTLTYPE_OPAQUE | CTLFLAG_RW | CTLFLAG_PRISON | CTLFLAG_MPSAFE,
9637029da5cSPawel Biernacki     0, 0, udp_getcred, "S,xucred",
9647029da5cSPawel Biernacki     "Get the xucred of a UDP connection");
96579288c11SBjoern A. Zeeb #endif /* INET */
966490d50b6SBrian Feldman 
9677b495c44SVANHULLEBUS Yvan int
9687b495c44SVANHULLEBUS Yvan udp_ctloutput(struct socket *so, struct sockopt *sopt)
9697b495c44SVANHULLEBUS Yvan {
9707b495c44SVANHULLEBUS Yvan 	struct inpcb *inp;
9717b495c44SVANHULLEBUS Yvan 	struct udpcb *up;
972e06e816fSKevin Lo 	int isudplite, error, optval;
9737b495c44SVANHULLEBUS Yvan 
974e06e816fSKevin Lo 	error = 0;
975e06e816fSKevin Lo 	isudplite = (so->so_proto->pr_protocol == IPPROTO_UDPLITE) ? 1 : 0;
9767b495c44SVANHULLEBUS Yvan 	inp = sotoinpcb(so);
9777b495c44SVANHULLEBUS Yvan 	KASSERT(inp != NULL, ("%s: inp == NULL", __func__));
9787b495c44SVANHULLEBUS Yvan 	INP_WLOCK(inp);
979e06e816fSKevin Lo 	if (sopt->sopt_level != so->so_proto->pr_protocol) {
9807b495c44SVANHULLEBUS Yvan #ifdef INET6
9817b495c44SVANHULLEBUS Yvan 		if (INP_CHECK_SOCKAF(so, AF_INET6)) {
9827b495c44SVANHULLEBUS Yvan 			INP_WUNLOCK(inp);
9837b495c44SVANHULLEBUS Yvan 			error = ip6_ctloutput(so, sopt);
98479288c11SBjoern A. Zeeb 		}
9857b495c44SVANHULLEBUS Yvan #endif
98679288c11SBjoern A. Zeeb #if defined(INET) && defined(INET6)
98779288c11SBjoern A. Zeeb 		else
98879288c11SBjoern A. Zeeb #endif
98979288c11SBjoern A. Zeeb #ifdef INET
99079288c11SBjoern A. Zeeb 		{
9917b495c44SVANHULLEBUS Yvan 			INP_WUNLOCK(inp);
9927b495c44SVANHULLEBUS Yvan 			error = ip_ctloutput(so, sopt);
9937b495c44SVANHULLEBUS Yvan 		}
9947b495c44SVANHULLEBUS Yvan #endif
9957b495c44SVANHULLEBUS Yvan 		return (error);
9967b495c44SVANHULLEBUS Yvan 	}
9977b495c44SVANHULLEBUS Yvan 
9987b495c44SVANHULLEBUS Yvan 	switch (sopt->sopt_dir) {
9997b495c44SVANHULLEBUS Yvan 	case SOPT_SET:
10007b495c44SVANHULLEBUS Yvan 		switch (sopt->sopt_name) {
1001fcf59617SAndrey V. Elsukov #if defined(IPSEC) || defined(IPSEC_SUPPORT)
1002fcf59617SAndrey V. Elsukov #ifdef INET
10037b495c44SVANHULLEBUS Yvan 		case UDP_ENCAP:
1004fcf59617SAndrey V. Elsukov 			if (!IPSEC_ENABLED(ipv4)) {
10057b495c44SVANHULLEBUS Yvan 				INP_WUNLOCK(inp);
1006fcf59617SAndrey V. Elsukov 				return (ENOPROTOOPT);
10077b495c44SVANHULLEBUS Yvan 			}
1008fcf59617SAndrey V. Elsukov 			error = UDPENCAP_PCBCTL(inp, sopt);
10097b495c44SVANHULLEBUS Yvan 			break;
1010fcf59617SAndrey V. Elsukov #endif /* INET */
1011fcf59617SAndrey V. Elsukov #endif /* IPSEC */
1012e06e816fSKevin Lo 		case UDPLITE_SEND_CSCOV:
1013e06e816fSKevin Lo 		case UDPLITE_RECV_CSCOV:
1014e06e816fSKevin Lo 			if (!isudplite) {
1015e06e816fSKevin Lo 				INP_WUNLOCK(inp);
1016e06e816fSKevin Lo 				error = ENOPROTOOPT;
1017e06e816fSKevin Lo 				break;
1018e06e816fSKevin Lo 			}
1019e06e816fSKevin Lo 			INP_WUNLOCK(inp);
1020e06e816fSKevin Lo 			error = sooptcopyin(sopt, &optval, sizeof(optval),
1021e06e816fSKevin Lo 			    sizeof(optval));
1022e06e816fSKevin Lo 			if (error != 0)
1023e06e816fSKevin Lo 				break;
1024e06e816fSKevin Lo 			inp = sotoinpcb(so);
1025e06e816fSKevin Lo 			KASSERT(inp != NULL, ("%s: inp == NULL", __func__));
1026e06e816fSKevin Lo 			INP_WLOCK(inp);
1027e06e816fSKevin Lo 			up = intoudpcb(inp);
1028e06e816fSKevin Lo 			KASSERT(up != NULL, ("%s: up == NULL", __func__));
102903f90784SMichael Tuexen 			if ((optval != 0 && optval < 8) || (optval > 65535)) {
1030e06e816fSKevin Lo 				INP_WUNLOCK(inp);
1031e06e816fSKevin Lo 				error = EINVAL;
1032e06e816fSKevin Lo 				break;
1033e06e816fSKevin Lo 			}
1034e06e816fSKevin Lo 			if (sopt->sopt_name == UDPLITE_SEND_CSCOV)
1035e06e816fSKevin Lo 				up->u_txcslen = optval;
1036e06e816fSKevin Lo 			else
1037e06e816fSKevin Lo 				up->u_rxcslen = optval;
1038e06e816fSKevin Lo 			INP_WUNLOCK(inp);
1039e06e816fSKevin Lo 			break;
10407b495c44SVANHULLEBUS Yvan 		default:
10417b495c44SVANHULLEBUS Yvan 			INP_WUNLOCK(inp);
10427b495c44SVANHULLEBUS Yvan 			error = ENOPROTOOPT;
10437b495c44SVANHULLEBUS Yvan 			break;
10447b495c44SVANHULLEBUS Yvan 		}
10457b495c44SVANHULLEBUS Yvan 		break;
10467b495c44SVANHULLEBUS Yvan 	case SOPT_GET:
10477b495c44SVANHULLEBUS Yvan 		switch (sopt->sopt_name) {
1048fcf59617SAndrey V. Elsukov #if defined(IPSEC) || defined(IPSEC_SUPPORT)
1049fcf59617SAndrey V. Elsukov #ifdef INET
10507b495c44SVANHULLEBUS Yvan 		case UDP_ENCAP:
1051fcf59617SAndrey V. Elsukov 			if (!IPSEC_ENABLED(ipv4)) {
10527b495c44SVANHULLEBUS Yvan 				INP_WUNLOCK(inp);
1053fcf59617SAndrey V. Elsukov 				return (ENOPROTOOPT);
1054fcf59617SAndrey V. Elsukov 			}
1055fcf59617SAndrey V. Elsukov 			error = UDPENCAP_PCBCTL(inp, sopt);
10567b495c44SVANHULLEBUS Yvan 			break;
1057fcf59617SAndrey V. Elsukov #endif /* INET */
1058fcf59617SAndrey V. Elsukov #endif /* IPSEC */
1059e06e816fSKevin Lo 		case UDPLITE_SEND_CSCOV:
1060e06e816fSKevin Lo 		case UDPLITE_RECV_CSCOV:
1061e06e816fSKevin Lo 			if (!isudplite) {
1062e06e816fSKevin Lo 				INP_WUNLOCK(inp);
1063e06e816fSKevin Lo 				error = ENOPROTOOPT;
1064e06e816fSKevin Lo 				break;
1065e06e816fSKevin Lo 			}
1066e06e816fSKevin Lo 			up = intoudpcb(inp);
1067e06e816fSKevin Lo 			KASSERT(up != NULL, ("%s: up == NULL", __func__));
1068e06e816fSKevin Lo 			if (sopt->sopt_name == UDPLITE_SEND_CSCOV)
1069e06e816fSKevin Lo 				optval = up->u_txcslen;
1070e06e816fSKevin Lo 			else
1071e06e816fSKevin Lo 				optval = up->u_rxcslen;
1072e06e816fSKevin Lo 			INP_WUNLOCK(inp);
1073e06e816fSKevin Lo 			error = sooptcopyout(sopt, &optval, sizeof(optval));
1074e06e816fSKevin Lo 			break;
10757b495c44SVANHULLEBUS Yvan 		default:
10767b495c44SVANHULLEBUS Yvan 			INP_WUNLOCK(inp);
10777b495c44SVANHULLEBUS Yvan 			error = ENOPROTOOPT;
10787b495c44SVANHULLEBUS Yvan 			break;
10797b495c44SVANHULLEBUS Yvan 		}
10807b495c44SVANHULLEBUS Yvan 		break;
10817b495c44SVANHULLEBUS Yvan 	}
10827b495c44SVANHULLEBUS Yvan 	return (error);
10837b495c44SVANHULLEBUS Yvan }
10847b495c44SVANHULLEBUS Yvan 
108579288c11SBjoern A. Zeeb #ifdef INET
1086a9839c4aSBjoern A. Zeeb #ifdef INET6
1087a9839c4aSBjoern A. Zeeb /* The logic here is derived from ip6_setpktopt(). See comments there. */
1088a9839c4aSBjoern A. Zeeb static int
1089a9839c4aSBjoern A. Zeeb udp_v4mapped_pktinfo(struct cmsghdr *cm, struct sockaddr_in * src,
1090a9839c4aSBjoern A. Zeeb     struct inpcb *inp, int flags)
1091a9839c4aSBjoern A. Zeeb {
1092a9839c4aSBjoern A. Zeeb 	struct ifnet *ifp;
1093a9839c4aSBjoern A. Zeeb 	struct in6_pktinfo *pktinfo;
1094a9839c4aSBjoern A. Zeeb 	struct in_addr ia;
1095a9839c4aSBjoern A. Zeeb 
1096a9839c4aSBjoern A. Zeeb 	if ((flags & PRUS_IPV6) == 0)
1097a9839c4aSBjoern A. Zeeb 		return (0);
1098a9839c4aSBjoern A. Zeeb 
1099a9839c4aSBjoern A. Zeeb 	if (cm->cmsg_level != IPPROTO_IPV6)
1100a9839c4aSBjoern A. Zeeb 		return (0);
1101a9839c4aSBjoern A. Zeeb 
1102a9839c4aSBjoern A. Zeeb 	if  (cm->cmsg_type != IPV6_2292PKTINFO &&
1103a9839c4aSBjoern A. Zeeb 	    cm->cmsg_type != IPV6_PKTINFO)
1104a9839c4aSBjoern A. Zeeb 		return (0);
1105a9839c4aSBjoern A. Zeeb 
1106a9839c4aSBjoern A. Zeeb 	if (cm->cmsg_len !=
1107a9839c4aSBjoern A. Zeeb 	    CMSG_LEN(sizeof(struct in6_pktinfo)))
1108a9839c4aSBjoern A. Zeeb 		return (EINVAL);
1109a9839c4aSBjoern A. Zeeb 
1110a9839c4aSBjoern A. Zeeb 	pktinfo = (struct in6_pktinfo *)CMSG_DATA(cm);
1111a9839c4aSBjoern A. Zeeb 	if (!IN6_IS_ADDR_V4MAPPED(&pktinfo->ipi6_addr) &&
1112a9839c4aSBjoern A. Zeeb 	    !IN6_IS_ADDR_UNSPECIFIED(&pktinfo->ipi6_addr))
1113a9839c4aSBjoern A. Zeeb 		return (EINVAL);
1114a9839c4aSBjoern A. Zeeb 
1115a9839c4aSBjoern A. Zeeb 	/* Validate the interface index if specified. */
1116a9839c4aSBjoern A. Zeeb 	if (pktinfo->ipi6_ifindex > V_if_index)
1117a9839c4aSBjoern A. Zeeb 		return (ENXIO);
1118a9839c4aSBjoern A. Zeeb 
1119a9839c4aSBjoern A. Zeeb 	ifp = NULL;
1120a9839c4aSBjoern A. Zeeb 	if (pktinfo->ipi6_ifindex) {
1121a9839c4aSBjoern A. Zeeb 		ifp = ifnet_byindex(pktinfo->ipi6_ifindex);
1122a9839c4aSBjoern A. Zeeb 		if (ifp == NULL)
1123a9839c4aSBjoern A. Zeeb 			return (ENXIO);
1124a9839c4aSBjoern A. Zeeb 	}
1125a9839c4aSBjoern A. Zeeb 	if (ifp != NULL && !IN6_IS_ADDR_UNSPECIFIED(&pktinfo->ipi6_addr)) {
1126a9839c4aSBjoern A. Zeeb 		ia.s_addr = pktinfo->ipi6_addr.s6_addr32[3];
1127a9839c4aSBjoern A. Zeeb 		if (in_ifhasaddr(ifp, ia) == 0)
1128a9839c4aSBjoern A. Zeeb 			return (EADDRNOTAVAIL);
1129a9839c4aSBjoern A. Zeeb 	}
1130a9839c4aSBjoern A. Zeeb 
1131a9839c4aSBjoern A. Zeeb 	bzero(src, sizeof(*src));
1132a9839c4aSBjoern A. Zeeb 	src->sin_family = AF_INET;
1133a9839c4aSBjoern A. Zeeb 	src->sin_len = sizeof(*src);
1134a9839c4aSBjoern A. Zeeb 	src->sin_port = inp->inp_lport;
1135a9839c4aSBjoern A. Zeeb 	src->sin_addr.s_addr = pktinfo->ipi6_addr.s6_addr32[3];
1136a9839c4aSBjoern A. Zeeb 
1137a9839c4aSBjoern A. Zeeb 	return (0);
1138a9839c4aSBjoern A. Zeeb }
1139a9839c4aSBjoern A. Zeeb #endif
1140a9839c4aSBjoern A. Zeeb 
1141490d50b6SBrian Feldman static int
11423329b236SRobert Watson udp_output(struct inpcb *inp, struct mbuf *m, struct sockaddr *addr,
1143a9839c4aSBjoern A. Zeeb     struct mbuf *control, struct thread *td, int flags)
1144df8bae1dSRodney W. Grimes {
11453329b236SRobert Watson 	struct udpiphdr *ui;
11463329b236SRobert Watson 	int len = m->m_pkthdr.len;
114790162a4eSIan Dowse 	struct in_addr faddr, laddr;
1148c557ae16SIan Dowse 	struct cmsghdr *cm;
1149e06e816fSKevin Lo 	struct inpcbinfo *pcbinfo;
1150c557ae16SIan Dowse 	struct sockaddr_in *sin, src;
11516573d758SMatt Macy 	struct epoch_tracker et;
1152e06e816fSKevin Lo 	int cscov_partial = 0;
115390162a4eSIan Dowse 	int error = 0;
1154*374ce248SMitchell Horne 	int ipflags = 0;
115590162a4eSIan Dowse 	u_short fport, lport;
1156f584d74bSMichael Tuexen 	u_char tos;
1157e06e816fSKevin Lo 	uint8_t pr;
1158e06e816fSKevin Lo 	uint16_t cscov = 0;
11599d3ddf43SAdrian Chadd 	uint32_t flowid = 0;
1160c2529042SHans Petter Selasky 	uint8_t flowtype = M_HASHTYPE_NONE;
1161df8bae1dSRodney W. Grimes 
1162430d30d8SBill Fenner 	if (len + sizeof(struct udpiphdr) > IP_MAXPACKET) {
1163c557ae16SIan Dowse 		if (control)
1164c557ae16SIan Dowse 			m_freem(control);
11655c32ea65SRobert Watson 		m_freem(m);
11663329b236SRobert Watson 		return (EMSGSIZE);
1167430d30d8SBill Fenner 	}
1168430d30d8SBill Fenner 
11691b7f0384SBruce M Simpson 	src.sin_family = 0;
117084cc0778SGeorge V. Neville-Neil 	sin = (struct sockaddr_in *)addr;
11712435e507SGleb Smirnoff 
1172eafaa1bcSBjoern A. Zeeb 	/*
11732435e507SGleb Smirnoff 	 * udp_output() may need to temporarily bind or connect the current
11742435e507SGleb Smirnoff 	 * inpcb.  As such, we don't know up front whether we will need the
11752435e507SGleb Smirnoff 	 * pcbinfo lock or not.  Do any work to decide what is needed up
11762435e507SGleb Smirnoff 	 * front before acquiring any locks.
11772435e507SGleb Smirnoff 	 *
11782435e507SGleb Smirnoff 	 * We will need network epoch in either case, to safely lookup into
11792435e507SGleb Smirnoff 	 * pcb hash.
1180eafaa1bcSBjoern A. Zeeb 	 */
11812435e507SGleb Smirnoff 	if (sin == NULL ||
11822435e507SGleb Smirnoff 	    (inp->inp_laddr.s_addr == INADDR_ANY && inp->inp_lport == 0))
11832435e507SGleb Smirnoff 		INP_WLOCK(inp);
11842435e507SGleb Smirnoff 	else
11850cfdff24SBjoern A. Zeeb 		INP_RLOCK(inp);
11862435e507SGleb Smirnoff 	NET_EPOCH_ENTER(et);
1187f584d74bSMichael Tuexen 	tos = inp->inp_ip_tos;
1188c557ae16SIan Dowse 	if (control != NULL) {
1189c557ae16SIan Dowse 		/*
11903329b236SRobert Watson 		 * XXX: Currently, we assume all the optional information is
11913329b236SRobert Watson 		 * stored in a single mbuf.
1192c557ae16SIan Dowse 		 */
1193c557ae16SIan Dowse 		if (control->m_next) {
1194c557ae16SIan Dowse 			m_freem(control);
11952435e507SGleb Smirnoff 			error = EINVAL;
11962435e507SGleb Smirnoff 			goto release;
1197c557ae16SIan Dowse 		}
1198c557ae16SIan Dowse 		for (; control->m_len > 0;
1199c557ae16SIan Dowse 		    control->m_data += CMSG_ALIGN(cm->cmsg_len),
1200c557ae16SIan Dowse 		    control->m_len -= CMSG_ALIGN(cm->cmsg_len)) {
1201c557ae16SIan Dowse 			cm = mtod(control, struct cmsghdr *);
1202af1ee11dSRobert Watson 			if (control->m_len < sizeof(*cm) || cm->cmsg_len == 0
1203af1ee11dSRobert Watson 			    || cm->cmsg_len > control->m_len) {
1204c557ae16SIan Dowse 				error = EINVAL;
1205c557ae16SIan Dowse 				break;
1206c557ae16SIan Dowse 			}
1207a9839c4aSBjoern A. Zeeb #ifdef INET6
1208a9839c4aSBjoern A. Zeeb 			error = udp_v4mapped_pktinfo(cm, &src, inp, flags);
1209a9839c4aSBjoern A. Zeeb 			if (error != 0)
1210a9839c4aSBjoern A. Zeeb 				break;
1211a9839c4aSBjoern A. Zeeb #endif
1212c557ae16SIan Dowse 			if (cm->cmsg_level != IPPROTO_IP)
1213c557ae16SIan Dowse 				continue;
1214c557ae16SIan Dowse 
1215c557ae16SIan Dowse 			switch (cm->cmsg_type) {
1216c557ae16SIan Dowse 			case IP_SENDSRCADDR:
1217c557ae16SIan Dowse 				if (cm->cmsg_len !=
1218c557ae16SIan Dowse 				    CMSG_LEN(sizeof(struct in_addr))) {
1219c557ae16SIan Dowse 					error = EINVAL;
1220c557ae16SIan Dowse 					break;
1221c557ae16SIan Dowse 				}
1222c557ae16SIan Dowse 				bzero(&src, sizeof(src));
1223c557ae16SIan Dowse 				src.sin_family = AF_INET;
1224c557ae16SIan Dowse 				src.sin_len = sizeof(src);
1225c557ae16SIan Dowse 				src.sin_port = inp->inp_lport;
1226af1ee11dSRobert Watson 				src.sin_addr =
1227af1ee11dSRobert Watson 				    *(struct in_addr *)CMSG_DATA(cm);
1228c557ae16SIan Dowse 				break;
1229af1ee11dSRobert Watson 
1230f584d74bSMichael Tuexen 			case IP_TOS:
1231f584d74bSMichael Tuexen 				if (cm->cmsg_len != CMSG_LEN(sizeof(u_char))) {
1232f584d74bSMichael Tuexen 					error = EINVAL;
1233f584d74bSMichael Tuexen 					break;
1234f584d74bSMichael Tuexen 				}
1235f584d74bSMichael Tuexen 				tos = *(u_char *)CMSG_DATA(cm);
1236f584d74bSMichael Tuexen 				break;
1237f584d74bSMichael Tuexen 
12389d3ddf43SAdrian Chadd 			case IP_FLOWID:
12399d3ddf43SAdrian Chadd 				if (cm->cmsg_len != CMSG_LEN(sizeof(uint32_t))) {
12409d3ddf43SAdrian Chadd 					error = EINVAL;
12419d3ddf43SAdrian Chadd 					break;
12429d3ddf43SAdrian Chadd 				}
12439d3ddf43SAdrian Chadd 				flowid = *(uint32_t *) CMSG_DATA(cm);
12449d3ddf43SAdrian Chadd 				break;
12459d3ddf43SAdrian Chadd 
12469d3ddf43SAdrian Chadd 			case IP_FLOWTYPE:
12479d3ddf43SAdrian Chadd 				if (cm->cmsg_len != CMSG_LEN(sizeof(uint32_t))) {
12489d3ddf43SAdrian Chadd 					error = EINVAL;
12499d3ddf43SAdrian Chadd 					break;
12509d3ddf43SAdrian Chadd 				}
1251c2529042SHans Petter Selasky 				flowtype = *(uint32_t *) CMSG_DATA(cm);
12529d3ddf43SAdrian Chadd 				break;
12539d3ddf43SAdrian Chadd 
12549d3ddf43SAdrian Chadd #ifdef	RSS
12559d3ddf43SAdrian Chadd 			case IP_RSSBUCKETID:
12569d3ddf43SAdrian Chadd 				if (cm->cmsg_len != CMSG_LEN(sizeof(uint32_t))) {
12579d3ddf43SAdrian Chadd 					error = EINVAL;
12589d3ddf43SAdrian Chadd 					break;
12599d3ddf43SAdrian Chadd 				}
12609d3ddf43SAdrian Chadd 				/* This is just a placeholder for now */
12619d3ddf43SAdrian Chadd 				break;
12629d3ddf43SAdrian Chadd #endif	/* RSS */
1263c557ae16SIan Dowse 			default:
1264c557ae16SIan Dowse 				error = ENOPROTOOPT;
1265c557ae16SIan Dowse 				break;
1266c557ae16SIan Dowse 			}
1267c557ae16SIan Dowse 			if (error)
1268c557ae16SIan Dowse 				break;
1269c557ae16SIan Dowse 		}
1270c557ae16SIan Dowse 		m_freem(control);
1271c557ae16SIan Dowse 	}
12722435e507SGleb Smirnoff 	if (error)
12732435e507SGleb Smirnoff 		goto release;
12745c32ea65SRobert Watson 
1275e06e816fSKevin Lo 	pr = inp->inp_socket->so_proto->pr_protocol;
1276a86e5c96SBjoern A. Zeeb 	pcbinfo = udp_get_inpcbinfo(pr);
12775c32ea65SRobert Watson 
12781b7f0384SBruce M Simpson 	/*
12791b7f0384SBruce M Simpson 	 * If the IP_SENDSRCADDR control message was specified, override the
12801b7f0384SBruce M Simpson 	 * source address for this datagram.  Its use is invalidated if the
12811b7f0384SBruce M Simpson 	 * address thus specified is incomplete or clobbers other inpcbs.
12821b7f0384SBruce M Simpson 	 */
128390162a4eSIan Dowse 	laddr = inp->inp_laddr;
128490162a4eSIan Dowse 	lport = inp->inp_lport;
12851b7f0384SBruce M Simpson 	if (src.sin_family == AF_INET) {
1286e06e816fSKevin Lo 		INP_HASH_LOCK_ASSERT(pcbinfo);
12871b7f0384SBruce M Simpson 		if ((lport == 0) ||
12881b7f0384SBruce M Simpson 		    (laddr.s_addr == INADDR_ANY &&
12891b7f0384SBruce M Simpson 		     src.sin_addr.s_addr == INADDR_ANY)) {
1290c557ae16SIan Dowse 			error = EINVAL;
1291c557ae16SIan Dowse 			goto release;
1292c557ae16SIan Dowse 		}
1293c557ae16SIan Dowse 		error = in_pcbbind_setup(inp, (struct sockaddr *)&src,
1294b0330ed9SPawel Jakub Dawidek 		    &laddr.s_addr, &lport, td->td_ucred);
1295c557ae16SIan Dowse 		if (error)
1296c557ae16SIan Dowse 			goto release;
1297c557ae16SIan Dowse 	}
1298c557ae16SIan Dowse 
12993144b7d3SRobert Watson 	/*
13003144b7d3SRobert Watson 	 * If a UDP socket has been connected, then a local address/port will
13013144b7d3SRobert Watson 	 * have been selected and bound.
13023144b7d3SRobert Watson 	 *
130343cc0bc1SRobert Watson 	 * If a UDP socket has not been connected to, then an explicit
13043144b7d3SRobert Watson 	 * destination address must be used, in which case a local
13053144b7d3SRobert Watson 	 * address/port may not have been selected and bound.
13063144b7d3SRobert Watson 	 */
130743cc0bc1SRobert Watson 	if (sin != NULL) {
1308c4d585aeSRobert Watson 		INP_LOCK_ASSERT(inp);
1309df8bae1dSRodney W. Grimes 		if (inp->inp_faddr.s_addr != INADDR_ANY) {
1310df8bae1dSRodney W. Grimes 			error = EISCONN;
1311df8bae1dSRodney W. Grimes 			goto release;
1312df8bae1dSRodney W. Grimes 		}
13133144b7d3SRobert Watson 
13143144b7d3SRobert Watson 		/*
13153144b7d3SRobert Watson 		 * Jail may rewrite the destination address, so let it do
13163144b7d3SRobert Watson 		 * that before we use it.
13173144b7d3SRobert Watson 		 */
1318b89e82ddSJamie Gritton 		error = prison_remote_ip4(td->td_ucred, &sin->sin_addr);
1319b89e82ddSJamie Gritton 		if (error)
1320413628a7SBjoern A. Zeeb 			goto release;
13213144b7d3SRobert Watson 
13223144b7d3SRobert Watson 		/*
132343cc0bc1SRobert Watson 		 * If a local address or port hasn't yet been selected, or if
132443cc0bc1SRobert Watson 		 * the destination address needs to be rewritten due to using
132543cc0bc1SRobert Watson 		 * a special INADDR_ constant, invoke in_pcbconnect_setup()
132643cc0bc1SRobert Watson 		 * to do the heavy lifting.  Once a port is selected, we
132743cc0bc1SRobert Watson 		 * commit the binding back to the socket; we also commit the
132843cc0bc1SRobert Watson 		 * binding of the address if in jail.
132943cc0bc1SRobert Watson 		 *
133043cc0bc1SRobert Watson 		 * If we already have a valid binding and we're not
133143cc0bc1SRobert Watson 		 * requesting a destination address rewrite, use a fast path.
13323144b7d3SRobert Watson 		 */
133343cc0bc1SRobert Watson 		if (inp->inp_laddr.s_addr == INADDR_ANY ||
133443cc0bc1SRobert Watson 		    inp->inp_lport == 0 ||
133543cc0bc1SRobert Watson 		    sin->sin_addr.s_addr == INADDR_ANY ||
133643cc0bc1SRobert Watson 		    sin->sin_addr.s_addr == INADDR_BROADCAST) {
1337e06e816fSKevin Lo 			INP_HASH_LOCK_ASSERT(pcbinfo);
133843cc0bc1SRobert Watson 			error = in_pcbconnect_setup(inp, addr, &laddr.s_addr,
133943cc0bc1SRobert Watson 			    &lport, &faddr.s_addr, &fport, NULL,
134043cc0bc1SRobert Watson 			    td->td_ucred);
134190162a4eSIan Dowse 			if (error)
134290162a4eSIan Dowse 				goto release;
134390162a4eSIan Dowse 
134443cc0bc1SRobert Watson 			/*
134543cc0bc1SRobert Watson 			 * XXXRW: Why not commit the port if the address is
134643cc0bc1SRobert Watson 			 * !INADDR_ANY?
134743cc0bc1SRobert Watson 			 */
134890162a4eSIan Dowse 			/* Commit the local port if newly assigned. */
134990162a4eSIan Dowse 			if (inp->inp_laddr.s_addr == INADDR_ANY &&
135090162a4eSIan Dowse 			    inp->inp_lport == 0) {
1351c4d585aeSRobert Watson 				INP_WLOCK_ASSERT(inp);
13523a1757b9SGleb Smirnoff 				/*
135343cc0bc1SRobert Watson 				 * Remember addr if jailed, to prevent
135443cc0bc1SRobert Watson 				 * rebinding.
13553a1757b9SGleb Smirnoff 				 */
13560304c731SJamie Gritton 				if (prison_flag(td->td_ucred, PR_IP4))
13573a1757b9SGleb Smirnoff 					inp->inp_laddr = laddr;
135890162a4eSIan Dowse 				inp->inp_lport = lport;
13592435e507SGleb Smirnoff 				INP_HASH_WLOCK(pcbinfo);
13602435e507SGleb Smirnoff 				error = in_pcbinshash(inp);
13612435e507SGleb Smirnoff 				INP_HASH_WUNLOCK(pcbinfo);
13622435e507SGleb Smirnoff 				if (error != 0) {
136390162a4eSIan Dowse 					inp->inp_lport = 0;
136490162a4eSIan Dowse 					error = EAGAIN;
1365df8bae1dSRodney W. Grimes 					goto release;
1366df8bae1dSRodney W. Grimes 				}
136790162a4eSIan Dowse 				inp->inp_flags |= INP_ANONPORT;
136890162a4eSIan Dowse 			}
1369df8bae1dSRodney W. Grimes 		} else {
137043cc0bc1SRobert Watson 			faddr = sin->sin_addr;
137143cc0bc1SRobert Watson 			fport = sin->sin_port;
137243cc0bc1SRobert Watson 		}
137343cc0bc1SRobert Watson 	} else {
1374c4d585aeSRobert Watson 		INP_LOCK_ASSERT(inp);
137590162a4eSIan Dowse 		faddr = inp->inp_faddr;
137690162a4eSIan Dowse 		fport = inp->inp_fport;
137790162a4eSIan Dowse 		if (faddr.s_addr == INADDR_ANY) {
1378df8bae1dSRodney W. Grimes 			error = ENOTCONN;
1379df8bae1dSRodney W. Grimes 			goto release;
1380df8bae1dSRodney W. Grimes 		}
1381df8bae1dSRodney W. Grimes 	}
1382e6ccd709SRobert Watson 
1383df8bae1dSRodney W. Grimes 	/*
1384e6ccd709SRobert Watson 	 * Calculate data length and get a mbuf for UDP, IP, and possible
1385392e8407SRobert Watson 	 * link-layer headers.  Immediate slide the data pointer back forward
1386392e8407SRobert Watson 	 * since we won't use that space at this layer.
1387df8bae1dSRodney W. Grimes 	 */
1388eb1b1807SGleb Smirnoff 	M_PREPEND(m, sizeof(struct udpiphdr) + max_linkhdr, M_NOWAIT);
1389e6ccd709SRobert Watson 	if (m == NULL) {
1390df8bae1dSRodney W. Grimes 		error = ENOBUFS;
139149b19bfcSBruce M Simpson 		goto release;
1392df8bae1dSRodney W. Grimes 	}
1393e6ccd709SRobert Watson 	m->m_data += max_linkhdr;
1394e6ccd709SRobert Watson 	m->m_len -= max_linkhdr;
1395392e8407SRobert Watson 	m->m_pkthdr.len -= max_linkhdr;
1396df8bae1dSRodney W. Grimes 
1397df8bae1dSRodney W. Grimes 	/*
13983329b236SRobert Watson 	 * Fill in mbuf with extended UDP header and addresses and length put
13993329b236SRobert Watson 	 * into network format.
1400df8bae1dSRodney W. Grimes 	 */
1401df8bae1dSRodney W. Grimes 	ui = mtod(m, struct udpiphdr *);
1402db4f9cc7SJonathan Lemon 	bzero(ui->ui_x1, sizeof(ui->ui_x1));	/* XXX still needed? */
140334fc9072SMichael Tuexen 	ui->ui_v = IPVERSION << 4;
1404e06e816fSKevin Lo 	ui->ui_pr = pr;
140590162a4eSIan Dowse 	ui->ui_src = laddr;
140690162a4eSIan Dowse 	ui->ui_dst = faddr;
140790162a4eSIan Dowse 	ui->ui_sport = lport;
140890162a4eSIan Dowse 	ui->ui_dport = fport;
1409db4f9cc7SJonathan Lemon 	ui->ui_ulen = htons((u_short)len + sizeof(struct udphdr));
1410e06e816fSKevin Lo 	if (pr == IPPROTO_UDPLITE) {
1411e06e816fSKevin Lo 		struct udpcb *up;
1412e06e816fSKevin Lo 		uint16_t plen;
1413e06e816fSKevin Lo 
1414e06e816fSKevin Lo 		up = intoudpcb(inp);
1415e06e816fSKevin Lo 		cscov = up->u_txcslen;
1416e06e816fSKevin Lo 		plen = (u_short)len + sizeof(struct udphdr);
1417e06e816fSKevin Lo 		if (cscov >= plen)
1418e06e816fSKevin Lo 			cscov = 0;
1419e06e816fSKevin Lo 		ui->ui_len = htons(plen);
1420e06e816fSKevin Lo 		ui->ui_ulen = htons(cscov);
1421e06e816fSKevin Lo 		/*
1422e06e816fSKevin Lo 		 * For UDP-Lite, checksum coverage length of zero means
1423e06e816fSKevin Lo 		 * the entire UDPLite packet is covered by the checksum.
1424e06e816fSKevin Lo 		 */
1425e06e816fSKevin Lo 		cscov_partial = (cscov == 0) ? 0 : 1;
142634fc9072SMichael Tuexen 	}
1427df8bae1dSRodney W. Grimes 
1428b2828ad2SAndre Oppermann 	/*
1429b2828ad2SAndre Oppermann 	 * Set the Don't Fragment bit in the IP header.
1430b2828ad2SAndre Oppermann 	 */
1431b2828ad2SAndre Oppermann 	if (inp->inp_flags & INP_DONTFRAG) {
1432b2828ad2SAndre Oppermann 		struct ip *ip;
14333329b236SRobert Watson 
1434b2828ad2SAndre Oppermann 		ip = (struct ip *)&ui->ui_i;
14358f134647SGleb Smirnoff 		ip->ip_off |= htons(IP_DF);
1436b2828ad2SAndre Oppermann 	}
1437b2828ad2SAndre Oppermann 
1438b5d47ff5SJohn-Mark Gurney 	if (inp->inp_socket->so_options & SO_DONTROUTE)
1439b5d47ff5SJohn-Mark Gurney 		ipflags |= IP_ROUTETOIF;
1440b5d47ff5SJohn-Mark Gurney 	if (inp->inp_socket->so_options & SO_BROADCAST)
1441b5d47ff5SJohn-Mark Gurney 		ipflags |= IP_ALLOWBROADCAST;
14426fbfd582SAndre Oppermann 	if (inp->inp_flags & INP_ONESBCAST)
14438afa2304SBruce M Simpson 		ipflags |= IP_SENDONES;
14448afa2304SBruce M Simpson 
14451175d9d5SRobert Watson #ifdef MAC
14461175d9d5SRobert Watson 	mac_inpcb_create_mbuf(inp, m);
14471175d9d5SRobert Watson #endif
14481175d9d5SRobert Watson 
1449df8bae1dSRodney W. Grimes 	/*
1450db4f9cc7SJonathan Lemon 	 * Set up checksum and output datagram.
1451df8bae1dSRodney W. Grimes 	 */
1452e06e816fSKevin Lo 	ui->ui_sum = 0;
1453a485f139SMichael Tuexen 	if (pr == IPPROTO_UDPLITE) {
1454e06e816fSKevin Lo 		if (inp->inp_flags & INP_ONESBCAST)
1455e06e816fSKevin Lo 			faddr.s_addr = INADDR_BROADCAST;
1456a485f139SMichael Tuexen 		if (cscov_partial) {
1457e06e816fSKevin Lo 			if ((ui->ui_sum = in_cksum(m, sizeof(struct ip) + cscov)) == 0)
1458e06e816fSKevin Lo 				ui->ui_sum = 0xffff;
1459a485f139SMichael Tuexen 		} else {
1460a485f139SMichael Tuexen 			if ((ui->ui_sum = in_cksum(m, sizeof(struct udpiphdr) + len)) == 0)
1461a485f139SMichael Tuexen 				ui->ui_sum = 0xffff;
1462a485f139SMichael Tuexen 		}
1463a485f139SMichael Tuexen 	} else if (V_udp_cksum) {
14646fbfd582SAndre Oppermann 		if (inp->inp_flags & INP_ONESBCAST)
14658a538743SBruce M Simpson 			faddr.s_addr = INADDR_BROADCAST;
14668a538743SBruce M Simpson 		ui->ui_sum = in_pseudo(ui->ui_src.s_addr, faddr.s_addr,
1467e06e816fSKevin Lo 		    htons((u_short)len + sizeof(struct udphdr) + pr));
1468db4f9cc7SJonathan Lemon 		m->m_pkthdr.csum_flags = CSUM_UDP;
1469db4f9cc7SJonathan Lemon 		m->m_pkthdr.csum_data = offsetof(struct udphdr, uh_sum);
1470e06e816fSKevin Lo 	}
14718f134647SGleb Smirnoff 	((struct ip *)ui)->ip_len = htons(sizeof(struct udpiphdr) + len);
1472ca98b82cSDavid Greenman 	((struct ip *)ui)->ip_ttl = inp->inp_ip_ttl;	/* XXX */
1473f584d74bSMichael Tuexen 	((struct ip *)ui)->ip_tos = tos;		/* XXX */
1474026decb8SRobert Watson 	UDPSTAT_INC(udps_opackets);
1475cfa1ca9dSYoshinobu Inoue 
14769d3ddf43SAdrian Chadd 	/*
14779d3ddf43SAdrian Chadd 	 * Setup flowid / RSS information for outbound socket.
14789d3ddf43SAdrian Chadd 	 *
14799d3ddf43SAdrian Chadd 	 * Once the UDP code decides to set a flowid some other way,
14809d3ddf43SAdrian Chadd 	 * this allows the flowid to be overridden by userland.
14819d3ddf43SAdrian Chadd 	 */
1482c2529042SHans Petter Selasky 	if (flowtype != M_HASHTYPE_NONE) {
14839d3ddf43SAdrian Chadd 		m->m_pkthdr.flowid = flowid;
1484c2529042SHans Petter Selasky 		M_HASHTYPE_SET(m, flowtype);
1485f9a6e8d7SBjoern A. Zeeb 	}
14868ad1a83bSAdrian Chadd #ifdef	RSS
1487f9a6e8d7SBjoern A. Zeeb 	else {
14888ad1a83bSAdrian Chadd 		uint32_t hash_val, hash_type;
14898ad1a83bSAdrian Chadd 		/*
14908ad1a83bSAdrian Chadd 		 * Calculate an appropriate RSS hash for UDP and
14918ad1a83bSAdrian Chadd 		 * UDP Lite.
14928ad1a83bSAdrian Chadd 		 *
14938ad1a83bSAdrian Chadd 		 * The called function will take care of figuring out
14948ad1a83bSAdrian Chadd 		 * whether a 2-tuple or 4-tuple hash is required based
14958ad1a83bSAdrian Chadd 		 * on the currently configured scheme.
14968ad1a83bSAdrian Chadd 		 *
14978ad1a83bSAdrian Chadd 		 * Later later on connected socket values should be
14988ad1a83bSAdrian Chadd 		 * cached in the inpcb and reused, rather than constantly
14998ad1a83bSAdrian Chadd 		 * re-calculating it.
15008ad1a83bSAdrian Chadd 		 *
15018ad1a83bSAdrian Chadd 		 * UDP Lite is a different protocol number and will
15028ad1a83bSAdrian Chadd 		 * likely end up being hashed as a 2-tuple until
15038ad1a83bSAdrian Chadd 		 * RSS / NICs grow UDP Lite protocol awareness.
15048ad1a83bSAdrian Chadd 		 */
15058ad1a83bSAdrian Chadd 		if (rss_proto_software_hash_v4(faddr, laddr, fport, lport,
15068ad1a83bSAdrian Chadd 		    pr, &hash_val, &hash_type) == 0) {
15078ad1a83bSAdrian Chadd 			m->m_pkthdr.flowid = hash_val;
15088ad1a83bSAdrian Chadd 			M_HASHTYPE_SET(m, hash_type);
15098ad1a83bSAdrian Chadd 		}
15109d3ddf43SAdrian Chadd 	}
15119d3ddf43SAdrian Chadd 
15128ad1a83bSAdrian Chadd 	/*
15138ad1a83bSAdrian Chadd 	 * Don't override with the inp cached flowid value.
15148ad1a83bSAdrian Chadd 	 *
15158ad1a83bSAdrian Chadd 	 * Depending upon the kind of send being done, the inp
15168ad1a83bSAdrian Chadd 	 * flowid/flowtype values may actually not be appropriate
15178ad1a83bSAdrian Chadd 	 * for this particular socket send.
15188ad1a83bSAdrian Chadd 	 *
15198ad1a83bSAdrian Chadd 	 * We should either leave the flowid at zero (which is what is
15208ad1a83bSAdrian Chadd 	 * currently done) or set it to some software generated
15218ad1a83bSAdrian Chadd 	 * hash value based on the packet contents.
15228ad1a83bSAdrian Chadd 	 */
15239d3ddf43SAdrian Chadd 	ipflags |= IP_NODEFAULTFLOWID;
15249d3ddf43SAdrian Chadd #endif	/* RSS */
15259d3ddf43SAdrian Chadd 
15267bda9663SMichael Tuexen 	if (pr == IPPROTO_UDPLITE)
15277bda9663SMichael Tuexen 		UDPLITE_PROBE(send, NULL, inp, &ui->ui_i, inp, &ui->ui_u);
15287bda9663SMichael Tuexen 	else
152957f60867SMark Johnston 		UDP_PROBE(send, NULL, inp, &ui->ui_i, inp, &ui->ui_u);
153084cc0778SGeorge V. Neville-Neil 	error = ip_output(m, inp->inp_options,
15312435e507SGleb Smirnoff 	    INP_WLOCKED(inp) ? &inp->inp_route : NULL, ipflags,
15325d846453SSam Leffler 	    inp->inp_moptions, inp);
15332435e507SGleb Smirnoff 	INP_UNLOCK(inp);
15342435e507SGleb Smirnoff 	NET_EPOCH_EXIT(et);
1535df8bae1dSRodney W. Grimes 	return (error);
1536df8bae1dSRodney W. Grimes 
1537df8bae1dSRodney W. Grimes release:
15382435e507SGleb Smirnoff 	INP_UNLOCK(inp);
15392435e507SGleb Smirnoff 	NET_EPOCH_EXIT(et);
1540df8bae1dSRodney W. Grimes 	m_freem(m);
1541df8bae1dSRodney W. Grimes 	return (error);
1542df8bae1dSRodney W. Grimes }
1543df8bae1dSRodney W. Grimes 
1544ac45e92fSRobert Watson static void
1545d0390e05SGarrett Wollman udp_abort(struct socket *so)
1546df8bae1dSRodney W. Grimes {
1547d0390e05SGarrett Wollman 	struct inpcb *inp;
1548e06e816fSKevin Lo 	struct inpcbinfo *pcbinfo;
1549df8bae1dSRodney W. Grimes 
1550a86e5c96SBjoern A. Zeeb 	pcbinfo = udp_get_inpcbinfo(so->so_proto->pr_protocol);
1551d0390e05SGarrett Wollman 	inp = sotoinpcb(so);
155214ba8addSRobert Watson 	KASSERT(inp != NULL, ("udp_abort: inp == NULL"));
15538501a69cSRobert Watson 	INP_WLOCK(inp);
1554a152f8a3SRobert Watson 	if (inp->inp_faddr.s_addr != INADDR_ANY) {
1555e06e816fSKevin Lo 		INP_HASH_WLOCK(pcbinfo);
1556a152f8a3SRobert Watson 		in_pcbdisconnect(inp);
1557a152f8a3SRobert Watson 		inp->inp_laddr.s_addr = INADDR_ANY;
1558e06e816fSKevin Lo 		INP_HASH_WUNLOCK(pcbinfo);
1559d0390e05SGarrett Wollman 		soisdisconnected(so);
1560a152f8a3SRobert Watson 	}
15618501a69cSRobert Watson 	INP_WUNLOCK(inp);
1562df8bae1dSRodney W. Grimes }
1563df8bae1dSRodney W. Grimes 
1564d0390e05SGarrett Wollman static int
1565b40ce416SJulian Elischer udp_attach(struct socket *so, int proto, struct thread *td)
1566d0390e05SGarrett Wollman {
1567630ba2c5SMatt Macy 	static uint32_t udp_flowid;
1568d0390e05SGarrett Wollman 	struct inpcb *inp;
1569e06e816fSKevin Lo 	struct inpcbinfo *pcbinfo;
1570277afaffSRobert Watson 	int error;
1571d0390e05SGarrett Wollman 
1572a86e5c96SBjoern A. Zeeb 	pcbinfo = udp_get_inpcbinfo(so->so_proto->pr_protocol);
1573d0390e05SGarrett Wollman 	inp = sotoinpcb(so);
157414ba8addSRobert Watson 	KASSERT(inp == NULL, ("udp_attach: inp != NULL"));
1575cfa1ca9dSYoshinobu Inoue 	error = soreserve(so, udp_sendspace, udp_recvspace);
1576f24618aaSRobert Watson 	if (error)
15773329b236SRobert Watson 		return (error);
1578e06e816fSKevin Lo 	INP_INFO_WLOCK(pcbinfo);
1579e06e816fSKevin Lo 	error = in_pcballoc(so, pcbinfo);
158053b57cd1SSam Leffler 	if (error) {
1581e06e816fSKevin Lo 		INP_INFO_WUNLOCK(pcbinfo);
15823329b236SRobert Watson 		return (error);
158353b57cd1SSam Leffler 	}
1584cfa1ca9dSYoshinobu Inoue 
158568b5629bSRobert Watson 	inp = sotoinpcb(so);
1586cfa1ca9dSYoshinobu Inoue 	inp->inp_vflag |= INP_IPV4;
1587603724d3SBjoern A. Zeeb 	inp->inp_ip_ttl = V_ip_defttl;
1588630ba2c5SMatt Macy 	inp->inp_flowid = atomic_fetchadd_int(&udp_flowid, 1);
1589630ba2c5SMatt Macy 	inp->inp_flowtype = M_HASHTYPE_OPAQUE;
15906a9148feSBjoern A. Zeeb 
15916a9148feSBjoern A. Zeeb 	error = udp_newudpcb(inp);
15926a9148feSBjoern A. Zeeb 	if (error) {
15936a9148feSBjoern A. Zeeb 		in_pcbdetach(inp);
15946a9148feSBjoern A. Zeeb 		in_pcbfree(inp);
1595e06e816fSKevin Lo 		INP_INFO_WUNLOCK(pcbinfo);
15966a9148feSBjoern A. Zeeb 		return (error);
15976a9148feSBjoern A. Zeeb 	}
15986a9148feSBjoern A. Zeeb 
1599c7c7ea4bSRandall Stewart 	INP_WUNLOCK(inp);
1600e06e816fSKevin Lo 	INP_INFO_WUNLOCK(pcbinfo);
1601c7c7ea4bSRandall Stewart 	return (0);
1602c7c7ea4bSRandall Stewart }
160379288c11SBjoern A. Zeeb #endif /* INET */
1604c7c7ea4bSRandall Stewart 
1605c7c7ea4bSRandall Stewart int
1606abb901c5SRandall Stewart udp_set_kernel_tunneling(struct socket *so, udp_tun_func_t f, udp_tun_icmp_t i, void *ctx)
1607c7c7ea4bSRandall Stewart {
1608c7c7ea4bSRandall Stewart 	struct inpcb *inp;
16096a9148feSBjoern A. Zeeb 	struct udpcb *up;
1610c7c7ea4bSRandall Stewart 
161168b5629bSRobert Watson 	KASSERT(so->so_type == SOCK_DGRAM,
161268b5629bSRobert Watson 	    ("udp_set_kernel_tunneling: !dgram"));
161368b5629bSRobert Watson 	inp = sotoinpcb(so);
161468b5629bSRobert Watson 	KASSERT(inp != NULL, ("udp_set_kernel_tunneling: inp == NULL"));
1615c7c7ea4bSRandall Stewart 	INP_WLOCK(inp);
16166a9148feSBjoern A. Zeeb 	up = intoudpcb(inp);
1617abb901c5SRandall Stewart 	if ((up->u_tun_func != NULL) ||
1618abb901c5SRandall Stewart 	    (up->u_icmp_func != NULL)) {
1619bbb0e3d9SRandall Stewart 		INP_WUNLOCK(inp);
1620bbb0e3d9SRandall Stewart 		return (EBUSY);
1621bbb0e3d9SRandall Stewart 	}
16226a9148feSBjoern A. Zeeb 	up->u_tun_func = f;
1623abb901c5SRandall Stewart 	up->u_icmp_func = i;
162481d3ec17SBryan Venteicher 	up->u_tun_ctx = ctx;
16258501a69cSRobert Watson 	INP_WUNLOCK(inp);
16263329b236SRobert Watson 	return (0);
1627df8bae1dSRodney W. Grimes }
1628d0390e05SGarrett Wollman 
162979288c11SBjoern A. Zeeb #ifdef INET
1630d0390e05SGarrett Wollman static int
1631b40ce416SJulian Elischer udp_bind(struct socket *so, struct sockaddr *nam, struct thread *td)
1632d0390e05SGarrett Wollman {
1633d0390e05SGarrett Wollman 	struct inpcb *inp;
1634e06e816fSKevin Lo 	struct inpcbinfo *pcbinfo;
1635277afaffSRobert Watson 	int error;
1636d0390e05SGarrett Wollman 
1637a86e5c96SBjoern A. Zeeb 	pcbinfo = udp_get_inpcbinfo(so->so_proto->pr_protocol);
1638d0390e05SGarrett Wollman 	inp = sotoinpcb(so);
163914ba8addSRobert Watson 	KASSERT(inp != NULL, ("udp_bind: inp == NULL"));
16408501a69cSRobert Watson 	INP_WLOCK(inp);
1641e06e816fSKevin Lo 	INP_HASH_WLOCK(pcbinfo);
1642b0330ed9SPawel Jakub Dawidek 	error = in_pcbbind(inp, nam, td->td_ucred);
1643e06e816fSKevin Lo 	INP_HASH_WUNLOCK(pcbinfo);
16448501a69cSRobert Watson 	INP_WUNLOCK(inp);
16453329b236SRobert Watson 	return (error);
1646d0390e05SGarrett Wollman }
1647d0390e05SGarrett Wollman 
1648a152f8a3SRobert Watson static void
1649a152f8a3SRobert Watson udp_close(struct socket *so)
1650a152f8a3SRobert Watson {
1651a152f8a3SRobert Watson 	struct inpcb *inp;
1652e06e816fSKevin Lo 	struct inpcbinfo *pcbinfo;
1653a152f8a3SRobert Watson 
1654a86e5c96SBjoern A. Zeeb 	pcbinfo = udp_get_inpcbinfo(so->so_proto->pr_protocol);
1655a152f8a3SRobert Watson 	inp = sotoinpcb(so);
1656a152f8a3SRobert Watson 	KASSERT(inp != NULL, ("udp_close: inp == NULL"));
16578501a69cSRobert Watson 	INP_WLOCK(inp);
1658a152f8a3SRobert Watson 	if (inp->inp_faddr.s_addr != INADDR_ANY) {
1659e06e816fSKevin Lo 		INP_HASH_WLOCK(pcbinfo);
1660a152f8a3SRobert Watson 		in_pcbdisconnect(inp);
1661a152f8a3SRobert Watson 		inp->inp_laddr.s_addr = INADDR_ANY;
1662e06e816fSKevin Lo 		INP_HASH_WUNLOCK(pcbinfo);
1663a152f8a3SRobert Watson 		soisdisconnected(so);
1664a152f8a3SRobert Watson 	}
16658501a69cSRobert Watson 	INP_WUNLOCK(inp);
1666a152f8a3SRobert Watson }
1667a152f8a3SRobert Watson 
1668d0390e05SGarrett Wollman static int
1669b40ce416SJulian Elischer udp_connect(struct socket *so, struct sockaddr *nam, struct thread *td)
1670d0390e05SGarrett Wollman {
1671c1604fe4SGleb Smirnoff 	struct epoch_tracker et;
1672d0390e05SGarrett Wollman 	struct inpcb *inp;
1673e06e816fSKevin Lo 	struct inpcbinfo *pcbinfo;
167475c13541SPoul-Henning Kamp 	struct sockaddr_in *sin;
1675e06e816fSKevin Lo 	int error;
1676d0390e05SGarrett Wollman 
1677a86e5c96SBjoern A. Zeeb 	pcbinfo = udp_get_inpcbinfo(so->so_proto->pr_protocol);
1678d0390e05SGarrett Wollman 	inp = sotoinpcb(so);
167914ba8addSRobert Watson 	KASSERT(inp != NULL, ("udp_connect: inp == NULL"));
16808501a69cSRobert Watson 	INP_WLOCK(inp);
1681f76fcf6dSJeffrey Hsu 	if (inp->inp_faddr.s_addr != INADDR_ANY) {
16828501a69cSRobert Watson 		INP_WUNLOCK(inp);
16833329b236SRobert Watson 		return (EISCONN);
1684f76fcf6dSJeffrey Hsu 	}
168575c13541SPoul-Henning Kamp 	sin = (struct sockaddr_in *)nam;
1686b89e82ddSJamie Gritton 	error = prison_remote_ip4(td->td_ucred, &sin->sin_addr);
1687b89e82ddSJamie Gritton 	if (error != 0) {
1688413628a7SBjoern A. Zeeb 		INP_WUNLOCK(inp);
1689b89e82ddSJamie Gritton 		return (error);
1690413628a7SBjoern A. Zeeb 	}
1691c1604fe4SGleb Smirnoff 	NET_EPOCH_ENTER(et);
1692e06e816fSKevin Lo 	INP_HASH_WLOCK(pcbinfo);
1693b0330ed9SPawel Jakub Dawidek 	error = in_pcbconnect(inp, nam, td->td_ucred);
1694e06e816fSKevin Lo 	INP_HASH_WUNLOCK(pcbinfo);
1695c1604fe4SGleb Smirnoff 	NET_EPOCH_EXIT(et);
16964cc20ab1SSeigo Tanimura 	if (error == 0)
1697df8bae1dSRodney W. Grimes 		soisconnected(so);
16988501a69cSRobert Watson 	INP_WUNLOCK(inp);
16993329b236SRobert Watson 	return (error);
1700df8bae1dSRodney W. Grimes }
1701d0390e05SGarrett Wollman 
1702bc725eafSRobert Watson static void
1703d0390e05SGarrett Wollman udp_detach(struct socket *so)
1704d0390e05SGarrett Wollman {
1705d0390e05SGarrett Wollman 	struct inpcb *inp;
1706e06e816fSKevin Lo 	struct inpcbinfo *pcbinfo;
17076a9148feSBjoern A. Zeeb 	struct udpcb *up;
1708d0390e05SGarrett Wollman 
1709a86e5c96SBjoern A. Zeeb 	pcbinfo = udp_get_inpcbinfo(so->so_proto->pr_protocol);
1710d0390e05SGarrett Wollman 	inp = sotoinpcb(so);
171114ba8addSRobert Watson 	KASSERT(inp != NULL, ("udp_detach: inp == NULL"));
1712a152f8a3SRobert Watson 	KASSERT(inp->inp_faddr.s_addr == INADDR_ANY,
1713a152f8a3SRobert Watson 	    ("udp_detach: not disconnected"));
1714e06e816fSKevin Lo 	INP_INFO_WLOCK(pcbinfo);
17158501a69cSRobert Watson 	INP_WLOCK(inp);
17166a9148feSBjoern A. Zeeb 	up = intoudpcb(inp);
17176a9148feSBjoern A. Zeeb 	KASSERT(up != NULL, ("%s: up == NULL", __func__));
17186a9148feSBjoern A. Zeeb 	inp->inp_ppcb = NULL;
1719d0390e05SGarrett Wollman 	in_pcbdetach(inp);
172014ba8addSRobert Watson 	in_pcbfree(inp);
1721e06e816fSKevin Lo 	INP_INFO_WUNLOCK(pcbinfo);
17226a9148feSBjoern A. Zeeb 	udp_discardcb(up);
1723d0390e05SGarrett Wollman }
1724d0390e05SGarrett Wollman 
1725d0390e05SGarrett Wollman static int
1726d0390e05SGarrett Wollman udp_disconnect(struct socket *so)
1727d0390e05SGarrett Wollman {
1728d0390e05SGarrett Wollman 	struct inpcb *inp;
1729e06e816fSKevin Lo 	struct inpcbinfo *pcbinfo;
1730d0390e05SGarrett Wollman 
1731a86e5c96SBjoern A. Zeeb 	pcbinfo = udp_get_inpcbinfo(so->so_proto->pr_protocol);
1732d0390e05SGarrett Wollman 	inp = sotoinpcb(so);
173314ba8addSRobert Watson 	KASSERT(inp != NULL, ("udp_disconnect: inp == NULL"));
17348501a69cSRobert Watson 	INP_WLOCK(inp);
1735f76fcf6dSJeffrey Hsu 	if (inp->inp_faddr.s_addr == INADDR_ANY) {
17368501a69cSRobert Watson 		INP_WUNLOCK(inp);
17373329b236SRobert Watson 		return (ENOTCONN);
1738f76fcf6dSJeffrey Hsu 	}
1739e06e816fSKevin Lo 	INP_HASH_WLOCK(pcbinfo);
1740df8bae1dSRodney W. Grimes 	in_pcbdisconnect(inp);
1741df8bae1dSRodney W. Grimes 	inp->inp_laddr.s_addr = INADDR_ANY;
1742e06e816fSKevin Lo 	INP_HASH_WUNLOCK(pcbinfo);
1743d45e4f99SMaxim Konovalov 	SOCK_LOCK(so);
1744d45e4f99SMaxim Konovalov 	so->so_state &= ~SS_ISCONNECTED;		/* XXX */
1745d45e4f99SMaxim Konovalov 	SOCK_UNLOCK(so);
17468501a69cSRobert Watson 	INP_WUNLOCK(inp);
17473329b236SRobert Watson 	return (0);
1748df8bae1dSRodney W. Grimes }
1749df8bae1dSRodney W. Grimes 
1750d0390e05SGarrett Wollman static int
175157bf258eSGarrett Wollman udp_send(struct socket *so, int flags, struct mbuf *m, struct sockaddr *addr,
1752b40ce416SJulian Elischer     struct mbuf *control, struct thread *td)
1753d0390e05SGarrett Wollman {
1754d0390e05SGarrett Wollman 	struct inpcb *inp;
1755d0390e05SGarrett Wollman 
1756d0390e05SGarrett Wollman 	inp = sotoinpcb(so);
175714ba8addSRobert Watson 	KASSERT(inp != NULL, ("udp_send: inp == NULL"));
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