xref: /freebsd/sys/netinet/udp_usrreq.c (revision 742e7210d00b359d81b9c778ab520003704e9b6c)
1c398230bSWarner Losh /*-
251369649SPedro F. Giffuni  * SPDX-License-Identifier: BSD-3-Clause
351369649SPedro F. Giffuni  *
46dfab5b1SGarrett Wollman  * Copyright (c) 1982, 1986, 1988, 1990, 1993, 1995
53329b236SRobert Watson  *	The Regents of the University of California.
63144b7d3SRobert Watson  * Copyright (c) 2008 Robert N. M. Watson
7fa046d87SRobert Watson  * Copyright (c) 2010-2011 Juniper Networks, Inc.
8e06e816fSKevin Lo  * Copyright (c) 2014 Kevin Lo
93329b236SRobert Watson  * All rights reserved.
10df8bae1dSRodney W. Grimes  *
11fa046d87SRobert Watson  * Portions of this software were developed by Robert N. M. Watson under
12fa046d87SRobert Watson  * contract to Juniper Networks, Inc.
13fa046d87SRobert Watson  *
14df8bae1dSRodney W. Grimes  * Redistribution and use in source and binary forms, with or without
15df8bae1dSRodney W. Grimes  * modification, are permitted provided that the following conditions
16df8bae1dSRodney W. Grimes  * are met:
17df8bae1dSRodney W. Grimes  * 1. Redistributions of source code must retain the above copyright
18df8bae1dSRodney W. Grimes  *    notice, this list of conditions and the following disclaimer.
19df8bae1dSRodney W. Grimes  * 2. Redistributions in binary form must reproduce the above copyright
20df8bae1dSRodney W. Grimes  *    notice, this list of conditions and the following disclaimer in the
21df8bae1dSRodney W. Grimes  *    documentation and/or other materials provided with the distribution.
22fbbd9655SWarner Losh  * 3. Neither the name of the University nor the names of its contributors
23df8bae1dSRodney W. Grimes  *    may be used to endorse or promote products derived from this software
24df8bae1dSRodney W. Grimes  *    without specific prior written permission.
25df8bae1dSRodney W. Grimes  *
26df8bae1dSRodney W. Grimes  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
27df8bae1dSRodney W. Grimes  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
28df8bae1dSRodney W. Grimes  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
29df8bae1dSRodney W. Grimes  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
30df8bae1dSRodney W. Grimes  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
31df8bae1dSRodney W. Grimes  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
32df8bae1dSRodney W. Grimes  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
33df8bae1dSRodney W. Grimes  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
34df8bae1dSRodney W. Grimes  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
35df8bae1dSRodney W. Grimes  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
36df8bae1dSRodney W. Grimes  * SUCH DAMAGE.
37df8bae1dSRodney W. Grimes  *
386dfab5b1SGarrett Wollman  *	@(#)udp_usrreq.c	8.6 (Berkeley) 5/23/95
39df8bae1dSRodney W. Grimes  */
40df8bae1dSRodney W. Grimes 
414b421e2dSMike Silbersack #include <sys/cdefs.h>
424b421e2dSMike Silbersack __FBSDID("$FreeBSD$");
434b421e2dSMike Silbersack 
4479288c11SBjoern A. Zeeb #include "opt_inet.h"
45cfa1ca9dSYoshinobu Inoue #include "opt_inet6.h"
46f5514f08SRobert Watson #include "opt_ipsec.h"
470c325f53SAlexander V. Chernikov #include "opt_route.h"
489d3ddf43SAdrian Chadd #include "opt_rss.h"
49cfa1ca9dSYoshinobu Inoue 
50df8bae1dSRodney W. Grimes #include <sys/param.h>
51960ed29cSSeigo Tanimura #include <sys/domain.h>
524f590175SPaul Saab #include <sys/eventhandler.h>
53960ed29cSSeigo Tanimura #include <sys/jail.h>
54b110a8a2SGarrett Wollman #include <sys/kernel.h>
55960ed29cSSeigo Tanimura #include <sys/lock.h>
56df8bae1dSRodney W. Grimes #include <sys/malloc.h>
57df8bae1dSRodney W. Grimes #include <sys/mbuf.h>
58acd3428bSRobert Watson #include <sys/priv.h>
59490d50b6SBrian Feldman #include <sys/proc.h>
60df8bae1dSRodney W. Grimes #include <sys/protosw.h>
6157f60867SMark Johnston #include <sys/sdt.h>
62960ed29cSSeigo Tanimura #include <sys/signalvar.h>
63df8bae1dSRodney W. Grimes #include <sys/socket.h>
64df8bae1dSRodney W. Grimes #include <sys/socketvar.h>
65960ed29cSSeigo Tanimura #include <sys/sx.h>
66b5e8ce9fSBruce Evans #include <sys/sysctl.h>
67816a3d83SPoul-Henning Kamp #include <sys/syslog.h>
68f5514f08SRobert Watson #include <sys/systm.h>
698781d8e9SBruce Evans 
7069c2d429SJeff Roberson #include <vm/uma.h>
71df8bae1dSRodney W. Grimes 
72df8bae1dSRodney W. Grimes #include <net/if.h>
7376039bc8SGleb Smirnoff #include <net/if_var.h>
74df8bae1dSRodney W. Grimes #include <net/route.h>
75983066f0SAlexander V. Chernikov #include <net/route/nhop.h>
76b2bdc62aSAdrian Chadd #include <net/rss_config.h>
77df8bae1dSRodney W. Grimes 
78df8bae1dSRodney W. Grimes #include <netinet/in.h>
7957f60867SMark Johnston #include <netinet/in_kdtrace.h>
800c325f53SAlexander V. Chernikov #include <netinet/in_fib.h>
81960ed29cSSeigo Tanimura #include <netinet/in_pcb.h>
82f5514f08SRobert Watson #include <netinet/in_systm.h>
83960ed29cSSeigo Tanimura #include <netinet/in_var.h>
84df8bae1dSRodney W. Grimes #include <netinet/ip.h>
85cfa1ca9dSYoshinobu Inoue #ifdef INET6
86cfa1ca9dSYoshinobu Inoue #include <netinet/ip6.h>
87cfa1ca9dSYoshinobu Inoue #endif
88960ed29cSSeigo Tanimura #include <netinet/ip_icmp.h>
89960ed29cSSeigo Tanimura #include <netinet/icmp_var.h>
90df8bae1dSRodney W. Grimes #include <netinet/ip_var.h>
91ef39adf0SAndre Oppermann #include <netinet/ip_options.h>
92cfa1ca9dSYoshinobu Inoue #ifdef INET6
93cfa1ca9dSYoshinobu Inoue #include <netinet6/ip6_var.h>
94cfa1ca9dSYoshinobu Inoue #endif
95df8bae1dSRodney W. Grimes #include <netinet/udp.h>
96df8bae1dSRodney W. Grimes #include <netinet/udp_var.h>
97e06e816fSKevin Lo #include <netinet/udplite.h>
988ad1a83bSAdrian Chadd #include <netinet/in_rss.h>
99df8bae1dSRodney W. Grimes 
100fcf59617SAndrey V. Elsukov #include <netipsec/ipsec_support.h>
101b9234fafSSam Leffler 
102db4f9cc7SJonathan Lemon #include <machine/in_cksum.h>
103db4f9cc7SJonathan Lemon 
104aed55708SRobert Watson #include <security/mac/mac_framework.h>
105aed55708SRobert Watson 
106df8bae1dSRodney W. Grimes /*
107e06e816fSKevin Lo  * UDP and UDP-Lite protocols implementation.
108df8bae1dSRodney W. Grimes  * Per RFC 768, August, 1980.
109e06e816fSKevin Lo  * Per RFC 3828, July, 2004.
110df8bae1dSRodney W. Grimes  */
11174eb3236SWarner Losh 
11274eb3236SWarner Losh /*
1133329b236SRobert Watson  * BSD 4.2 defaulted the udp checksum to be off.  Turning off udp checksums
1143329b236SRobert Watson  * removes the only data integrity mechanism for packets and malformed
115f5514f08SRobert Watson  * packets that would otherwise be discarded due to bad checksums, and may
116f5514f08SRobert Watson  * cause problems (especially for NFS data blocks).
11774eb3236SWarner Losh  */
11840b676beSBjoern A. Zeeb VNET_DEFINE(int, udp_cksum) = 1;
1196df8a710SGleb Smirnoff SYSCTL_INT(_net_inet_udp, UDPCTL_CHECKSUM, checksum, CTLFLAG_VNET | CTLFLAG_RW,
12040b676beSBjoern A. Zeeb     &VNET_NAME(udp_cksum), 0, "compute udp checksum");
121df8bae1dSRodney W. Grimes 
122334fc582SBjoern A. Zeeb VNET_DEFINE(int, udp_log_in_vain) = 0;
123334fc582SBjoern A. Zeeb SYSCTL_INT(_net_inet_udp, OID_AUTO, log_in_vain, CTLFLAG_VNET | CTLFLAG_RW,
124334fc582SBjoern A. Zeeb     &VNET_NAME(udp_log_in_vain), 0, "Log all incoming UDP packets");
125816a3d83SPoul-Henning Kamp 
12682cea7e6SBjoern A. Zeeb VNET_DEFINE(int, udp_blackhole) = 0;
1276df8a710SGleb Smirnoff SYSCTL_INT(_net_inet_udp, OID_AUTO, blackhole, CTLFLAG_VNET | CTLFLAG_RW,
128eddfbb76SRobert Watson     &VNET_NAME(udp_blackhole), 0,
1293329b236SRobert Watson     "Do not send port unreachables for refused connects");
1303ea9a7cfSGleb Smirnoff VNET_DEFINE(bool, udp_blackhole_local) = false;
1313ea9a7cfSGleb Smirnoff SYSCTL_BOOL(_net_inet_udp, OID_AUTO, blackhole_local, CTLFLAG_VNET |
1323ea9a7cfSGleb Smirnoff     CTLFLAG_RW, &VNET_NAME(udp_blackhole_local), false,
1333ea9a7cfSGleb Smirnoff     "Enforce net.inet.udp.blackhole for locally originated packets");
13416f7f31fSGeoff Rehmet 
13543bbb6aaSRobert Watson u_long	udp_sendspace = 9216;		/* really max datagram size */
13643bbb6aaSRobert Watson SYSCTL_ULONG(_net_inet_udp, UDPCTL_MAXDGRAM, maxdgram, CTLFLAG_RW,
13743bbb6aaSRobert Watson     &udp_sendspace, 0, "Maximum outgoing UDP datagram size");
13843bbb6aaSRobert Watson 
13943bbb6aaSRobert Watson u_long	udp_recvspace = 40 * (1024 +
14043bbb6aaSRobert Watson #ifdef INET6
14143bbb6aaSRobert Watson 				      sizeof(struct sockaddr_in6)
14243bbb6aaSRobert Watson #else
14343bbb6aaSRobert Watson 				      sizeof(struct sockaddr_in)
14443bbb6aaSRobert Watson #endif
145e62b9bcaSSergey Kandaurov 				      );	/* 40 1K datagrams */
14643bbb6aaSRobert Watson 
14743bbb6aaSRobert Watson SYSCTL_ULONG(_net_inet_udp, UDPCTL_RECVSPACE, recvspace, CTLFLAG_RW,
14843bbb6aaSRobert Watson     &udp_recvspace, 0, "Maximum space for incoming UDP datagrams");
14943bbb6aaSRobert Watson 
150eddfbb76SRobert Watson VNET_DEFINE(struct inpcbinfo, udbinfo);
151e06e816fSKevin Lo VNET_DEFINE(struct inpcbinfo, ulitecbinfo);
1525f901c92SAndrew Turner VNET_DEFINE_STATIC(uma_zone_t, udpcb_zone);
1531e77c105SRobert Watson #define	V_udpcb_zone			VNET(udpcb_zone)
15415bd2b43SDavid Greenman 
15515bd2b43SDavid Greenman #ifndef UDBHASHSIZE
156e2ed8f35SAlexander Motin #define	UDBHASHSIZE	128
15715bd2b43SDavid Greenman #endif
15815bd2b43SDavid Greenman 
1595b7cb97cSAndrey V. Elsukov VNET_PCPUSTAT_DEFINE(struct udpstat, udpstat);		/* from udp_var.h */
1605b7cb97cSAndrey V. Elsukov VNET_PCPUSTAT_SYSINIT(udpstat);
1615b7cb97cSAndrey V. Elsukov SYSCTL_VNET_PCPUSTAT(_net_inet_udp, UDPCTL_STATS, stats, struct udpstat,
1625b7cb97cSAndrey V. Elsukov     udpstat, "UDP statistics (struct udpstat, netinet/udp_var.h)");
163f2ea20e6SGarrett Wollman 
1645b7cb97cSAndrey V. Elsukov #ifdef VIMAGE
1655b7cb97cSAndrey V. Elsukov VNET_PCPUSTAT_SYSUNINIT(udpstat);
1665b7cb97cSAndrey V. Elsukov #endif /* VIMAGE */
16779288c11SBjoern A. Zeeb #ifdef INET
168bc725eafSRobert Watson static void	udp_detach(struct socket *so);
1694d77a549SAlfred Perlstein static int	udp_output(struct inpcb *, struct mbuf *, struct sockaddr *,
170a9839c4aSBjoern A. Zeeb 		    struct mbuf *, struct thread *, int);
17179288c11SBjoern A. Zeeb #endif
17279288c11SBjoern A. Zeeb 
173fec8a8c7SGleb Smirnoff INPCBSTORAGE_DEFINE(udpcbstor, "udpinp", "udp_inpcb", "udp", "udphash");
174fec8a8c7SGleb Smirnoff INPCBSTORAGE_DEFINE(udplitecbstor, "udpliteinp", "udplite_inpcb", "udplite",
175fec8a8c7SGleb Smirnoff     "udplitehash");
176e06e816fSKevin Lo 
17789128ff3SGleb Smirnoff static void
17889128ff3SGleb Smirnoff udp_init(void *arg __unused)
179df8bae1dSRodney W. Grimes {
180af1ee11dSRobert Watson 
1818ad1a83bSAdrian Chadd 	/*
1828ad1a83bSAdrian Chadd 	 * For now default to 2-tuple UDP hashing - until the fragment
1838ad1a83bSAdrian Chadd 	 * reassembly code can also update the flowid.
1848ad1a83bSAdrian Chadd 	 *
1858ad1a83bSAdrian Chadd 	 * Once we can calculate the flowid that way and re-establish
1868ad1a83bSAdrian Chadd 	 * a 4-tuple, flip this to 4-tuple.
1878ad1a83bSAdrian Chadd 	 */
188fec8a8c7SGleb Smirnoff 	in_pcbinfo_init(&V_udbinfo, &udpcbstor, UDBHASHSIZE, UDBHASHSIZE);
1896a9148feSBjoern A. Zeeb 	V_udpcb_zone = uma_zcreate("udpcb", sizeof(struct udpcb),
190a8da5dd6SCraig Rodrigues 	    NULL, NULL, NULL, NULL, UMA_ALIGN_PTR, 0);
1916a9148feSBjoern A. Zeeb 	uma_zone_set_max(V_udpcb_zone, maxsockets);
1926acd596eSPawel Jakub Dawidek 	uma_zone_set_warning(V_udpcb_zone, "kern.ipc.maxsockets limit reached");
193df8bae1dSRodney W. Grimes 
19489128ff3SGleb Smirnoff 	/* Additional pcbinfo for UDP-Lite */
195fec8a8c7SGleb Smirnoff 	in_pcbinfo_init(&V_ulitecbinfo, &udplitecbstor, UDBHASHSIZE,
196fec8a8c7SGleb Smirnoff 	    UDBHASHSIZE);
197e06e816fSKevin Lo }
19889128ff3SGleb Smirnoff VNET_SYSINIT(udp_init, SI_SUB_PROTO_DOMAIN, SI_ORDER_THIRD, udp_init, NULL);
199e06e816fSKevin Lo 
200315e3e38SRobert Watson /*
201315e3e38SRobert Watson  * Kernel module interface for updating udpstat.  The argument is an index
202315e3e38SRobert Watson  * into udpstat treated as an array of u_long.  While this encodes the
203315e3e38SRobert Watson  * general layout of udpstat into the caller, it doesn't encode its location,
204315e3e38SRobert Watson  * so that future changes to add, for example, per-CPU stats support won't
205315e3e38SRobert Watson  * cause binary compatibility problems for kernel modules.
206315e3e38SRobert Watson  */
207315e3e38SRobert Watson void
208315e3e38SRobert Watson kmod_udpstat_inc(int statnum)
209315e3e38SRobert Watson {
210315e3e38SRobert Watson 
2115b7cb97cSAndrey V. Elsukov 	counter_u64_add(VNET(udpstat)[statnum], 1);
212315e3e38SRobert Watson }
213315e3e38SRobert Watson 
2146a9148feSBjoern A. Zeeb int
2156a9148feSBjoern A. Zeeb udp_newudpcb(struct inpcb *inp)
2166a9148feSBjoern A. Zeeb {
2176a9148feSBjoern A. Zeeb 	struct udpcb *up;
2186a9148feSBjoern A. Zeeb 
2196a9148feSBjoern A. Zeeb 	up = uma_zalloc(V_udpcb_zone, M_NOWAIT | M_ZERO);
2206a9148feSBjoern A. Zeeb 	if (up == NULL)
2216a9148feSBjoern A. Zeeb 		return (ENOBUFS);
2226a9148feSBjoern A. Zeeb 	inp->inp_ppcb = up;
2236a9148feSBjoern A. Zeeb 	return (0);
2246a9148feSBjoern A. Zeeb }
2256a9148feSBjoern A. Zeeb 
2266a9148feSBjoern A. Zeeb void
2276a9148feSBjoern A. Zeeb udp_discardcb(struct udpcb *up)
2286a9148feSBjoern A. Zeeb {
2296a9148feSBjoern A. Zeeb 
2306a9148feSBjoern A. Zeeb 	uma_zfree(V_udpcb_zone, up);
2316a9148feSBjoern A. Zeeb }
2326a9148feSBjoern A. Zeeb 
233bc29160dSMarko Zec #ifdef VIMAGE
2343f58662dSBjoern A. Zeeb static void
2353f58662dSBjoern A. Zeeb udp_destroy(void *unused __unused)
236bc29160dSMarko Zec {
237bc29160dSMarko Zec 
2389bcd427bSRobert Watson 	in_pcbinfo_destroy(&V_udbinfo);
239391dab1cSBjoern A. Zeeb 	uma_zdestroy(V_udpcb_zone);
240bc29160dSMarko Zec }
2413f58662dSBjoern A. Zeeb VNET_SYSUNINIT(udp, SI_SUB_PROTO_DOMAIN, SI_ORDER_FOURTH, udp_destroy, NULL);
242e06e816fSKevin Lo 
2433f58662dSBjoern A. Zeeb static void
2443f58662dSBjoern A. Zeeb udplite_destroy(void *unused __unused)
245e06e816fSKevin Lo {
246e06e816fSKevin Lo 
247e06e816fSKevin Lo 	in_pcbinfo_destroy(&V_ulitecbinfo);
248e06e816fSKevin Lo }
2493f58662dSBjoern A. Zeeb VNET_SYSUNINIT(udplite, SI_SUB_PROTO_DOMAIN, SI_ORDER_FOURTH, udplite_destroy,
2503f58662dSBjoern A. Zeeb     NULL);
251bc29160dSMarko Zec #endif
252bc29160dSMarko Zec 
25379288c11SBjoern A. Zeeb #ifdef INET
25443bbb6aaSRobert Watson /*
25543bbb6aaSRobert Watson  * Subroutine of udp_input(), which appends the provided mbuf chain to the
25643bbb6aaSRobert Watson  * passed pcb/socket.  The caller must provide a sockaddr_in via udp_in that
25743bbb6aaSRobert Watson  * contains the source address.  If the socket ends up being an IPv6 socket,
25843bbb6aaSRobert Watson  * udp_append() will convert to a sockaddr_in6 before passing the address
25943bbb6aaSRobert Watson  * into the socket code.
260c0d1be08SRandall Stewart  *
261c0d1be08SRandall Stewart  * In the normal case udp_append() will return 0, indicating that you
262c0d1be08SRandall Stewart  * must unlock the inp. However if a tunneling protocol is in place we increment
263c0d1be08SRandall Stewart  * the inpcb refcnt and unlock the inp, on return from the tunneling protocol we
264c0d1be08SRandall Stewart  * then decrement the reference count. If the inp_rele returns 1, indicating the
265c0d1be08SRandall Stewart  * inp is gone, we return that to the caller to tell them *not* to unlock
266c0d1be08SRandall Stewart  * the inp. In the case of multi-cast this will cause the distribution
267c0d1be08SRandall Stewart  * to stop (though most tunneling protocols known currently do *not* use
268c0d1be08SRandall Stewart  * multicast).
26943bbb6aaSRobert Watson  */
270c0d1be08SRandall Stewart static int
27143bbb6aaSRobert Watson udp_append(struct inpcb *inp, struct ip *ip, struct mbuf *n, int off,
27243bbb6aaSRobert Watson     struct sockaddr_in *udp_in)
27343bbb6aaSRobert Watson {
27443bbb6aaSRobert Watson 	struct sockaddr *append_sa;
27543bbb6aaSRobert Watson 	struct socket *so;
276dce33a45SErmal Luçi 	struct mbuf *tmpopts, *opts = NULL;
27743bbb6aaSRobert Watson #ifdef INET6
27843bbb6aaSRobert Watson 	struct sockaddr_in6 udp_in6;
27943bbb6aaSRobert Watson #endif
2807b495c44SVANHULLEBUS Yvan 	struct udpcb *up;
281*742e7210SKristof Provost 	bool filtered;
28243bbb6aaSRobert Watson 
283fa046d87SRobert Watson 	INP_LOCK_ASSERT(inp);
28443bbb6aaSRobert Watson 
28579bb84fbSEdward Tomasz Napierala 	/*
28679bb84fbSEdward Tomasz Napierala 	 * Engage the tunneling protocol.
28779bb84fbSEdward Tomasz Napierala 	 */
28879bb84fbSEdward Tomasz Napierala 	up = intoudpcb(inp);
28979bb84fbSEdward Tomasz Napierala 	if (up->u_tun_func != NULL) {
290c0d1be08SRandall Stewart 		in_pcbref(inp);
291c0d1be08SRandall Stewart 		INP_RUNLOCK(inp);
292*742e7210SKristof Provost 		filtered = (*up->u_tun_func)(n, off, inp, (struct sockaddr *)&udp_in[0],
29381d3ec17SBryan Venteicher 		    up->u_tun_ctx);
294c0d1be08SRandall Stewart 		INP_RLOCK(inp);
295*742e7210SKristof Provost 		if (filtered)
296c0d1be08SRandall Stewart 			return (in_pcbrele_rlocked(inp));
29779bb84fbSEdward Tomasz Napierala 	}
29879bb84fbSEdward Tomasz Napierala 
29979bb84fbSEdward Tomasz Napierala 	off += sizeof(struct udphdr);
30079bb84fbSEdward Tomasz Napierala 
301fcf59617SAndrey V. Elsukov #if defined(IPSEC) || defined(IPSEC_SUPPORT)
30243bbb6aaSRobert Watson 	/* Check AH/ESP integrity. */
303fcf59617SAndrey V. Elsukov 	if (IPSEC_ENABLED(ipv4) &&
304fcf59617SAndrey V. Elsukov 	    IPSEC_CHECK_POLICY(ipv4, n, inp) != 0) {
30543bbb6aaSRobert Watson 		m_freem(n);
306c0d1be08SRandall Stewart 		return (0);
30743bbb6aaSRobert Watson 	}
308fcf59617SAndrey V. Elsukov 	if (up->u_flags & UF_ESPINUDP) {/* IPSec UDP encaps. */
309fcf59617SAndrey V. Elsukov 		if (IPSEC_ENABLED(ipv4) &&
310fcf59617SAndrey V. Elsukov 		    UDPENCAP_INPUT(n, off, AF_INET) != 0)
311fcf59617SAndrey V. Elsukov 			return (0);	/* Consumed. */
3127b495c44SVANHULLEBUS Yvan 	}
31343bbb6aaSRobert Watson #endif /* IPSEC */
31443bbb6aaSRobert Watson #ifdef MAC
31530d239bcSRobert Watson 	if (mac_inpcb_check_deliver(inp, n) != 0) {
31643bbb6aaSRobert Watson 		m_freem(n);
317c0d1be08SRandall Stewart 		return (0);
31843bbb6aaSRobert Watson 	}
31979288c11SBjoern A. Zeeb #endif /* MAC */
32043bbb6aaSRobert Watson 	if (inp->inp_flags & INP_CONTROLOPTS ||
32143bbb6aaSRobert Watson 	    inp->inp_socket->so_options & (SO_TIMESTAMP | SO_BINTIME)) {
32243bbb6aaSRobert Watson #ifdef INET6
3239a38ba81SBjoern A. Zeeb 		if (inp->inp_vflag & INP_IPV6)
32448d48eb9SBjoern A. Zeeb 			(void)ip6_savecontrol_v4(inp, n, &opts, NULL);
3259a38ba81SBjoern A. Zeeb 		else
32679288c11SBjoern A. Zeeb #endif /* INET6 */
32743bbb6aaSRobert Watson 			ip_savecontrol(inp, &opts, ip, n);
32843bbb6aaSRobert Watson 	}
329dce33a45SErmal Luçi 	if ((inp->inp_vflag & INP_IPV4) && (inp->inp_flags2 & INP_ORIGDSTADDR)) {
330dce33a45SErmal Luçi 		tmpopts = sbcreatecontrol((caddr_t)&udp_in[1],
331dce33a45SErmal Luçi 			sizeof(struct sockaddr_in), IP_ORIGDSTADDR, IPPROTO_IP);
332dce33a45SErmal Luçi 		if (tmpopts) {
333dce33a45SErmal Luçi 			if (opts) {
334dce33a45SErmal Luçi 				tmpopts->m_next = opts;
335dce33a45SErmal Luçi 				opts = tmpopts;
336dce33a45SErmal Luçi 			} else
337dce33a45SErmal Luçi 				opts = tmpopts;
338dce33a45SErmal Luçi 		}
339dce33a45SErmal Luçi 	}
34043bbb6aaSRobert Watson #ifdef INET6
34143bbb6aaSRobert Watson 	if (inp->inp_vflag & INP_IPV6) {
34243bbb6aaSRobert Watson 		bzero(&udp_in6, sizeof(udp_in6));
34343bbb6aaSRobert Watson 		udp_in6.sin6_len = sizeof(udp_in6);
34443bbb6aaSRobert Watson 		udp_in6.sin6_family = AF_INET6;
345dce33a45SErmal Luçi 		in6_sin_2_v4mapsin6(&udp_in[0], &udp_in6);
34643bbb6aaSRobert Watson 		append_sa = (struct sockaddr *)&udp_in6;
34743bbb6aaSRobert Watson 	} else
34879288c11SBjoern A. Zeeb #endif /* INET6 */
349dce33a45SErmal Luçi 		append_sa = (struct sockaddr *)&udp_in[0];
35043bbb6aaSRobert Watson 	m_adj(n, off);
35143bbb6aaSRobert Watson 
35243bbb6aaSRobert Watson 	so = inp->inp_socket;
35343bbb6aaSRobert Watson 	SOCKBUF_LOCK(&so->so_rcv);
35443bbb6aaSRobert Watson 	if (sbappendaddr_locked(&so->so_rcv, append_sa, n, opts) == 0) {
355a61c24ddSKonstantin Kukushkin 		soroverflow_locked(so);
35643bbb6aaSRobert Watson 		m_freem(n);
35743bbb6aaSRobert Watson 		if (opts)
35843bbb6aaSRobert Watson 			m_freem(opts);
359026decb8SRobert Watson 		UDPSTAT_INC(udps_fullsock);
36043bbb6aaSRobert Watson 	} else
36143bbb6aaSRobert Watson 		sorwakeup_locked(so);
362c0d1be08SRandall Stewart 	return (0);
36343bbb6aaSRobert Watson }
36443bbb6aaSRobert Watson 
365db0ac6deSCy Schubert static bool
366db0ac6deSCy Schubert udp_multi_match(const struct inpcb *inp, void *v)
367db0ac6deSCy Schubert {
368db0ac6deSCy Schubert 	struct ip *ip = v;
369db0ac6deSCy Schubert 	struct udphdr *uh = (struct udphdr *)(ip + 1);
370db0ac6deSCy Schubert 
371db0ac6deSCy Schubert 	if (inp->inp_lport != uh->uh_dport)
372db0ac6deSCy Schubert 		return (false);
373db0ac6deSCy Schubert #ifdef INET6
374db0ac6deSCy Schubert 	if ((inp->inp_vflag & INP_IPV4) == 0)
375db0ac6deSCy Schubert 		return (false);
376db0ac6deSCy Schubert #endif
377db0ac6deSCy Schubert 	if (inp->inp_laddr.s_addr != INADDR_ANY &&
378db0ac6deSCy Schubert 	    inp->inp_laddr.s_addr != ip->ip_dst.s_addr)
379db0ac6deSCy Schubert 		return (false);
380db0ac6deSCy Schubert 	if (inp->inp_faddr.s_addr != INADDR_ANY &&
381db0ac6deSCy Schubert 	    inp->inp_faddr.s_addr != ip->ip_src.s_addr)
382db0ac6deSCy Schubert 		return (false);
383db0ac6deSCy Schubert 	if (inp->inp_fport != 0 &&
384db0ac6deSCy Schubert 	    inp->inp_fport != uh->uh_sport)
385db0ac6deSCy Schubert 		return (false);
386db0ac6deSCy Schubert 
387db0ac6deSCy Schubert 	return (true);
388db0ac6deSCy Schubert }
389db0ac6deSCy Schubert 
390db0ac6deSCy Schubert static int
391db0ac6deSCy Schubert udp_multi_input(struct mbuf *m, int proto, struct sockaddr_in *udp_in)
392db0ac6deSCy Schubert {
393db0ac6deSCy Schubert 	struct ip *ip = mtod(m, struct ip *);
394db0ac6deSCy Schubert 	struct inpcb_iterator inpi = INP_ITERATOR(udp_get_inpcbinfo(proto),
395db0ac6deSCy Schubert 	    INPLOOKUP_RLOCKPCB, udp_multi_match, ip);
396bd1d0850SGleb Smirnoff #ifdef KDTRACE_HOOKS
397db0ac6deSCy Schubert 	struct udphdr *uh = (struct udphdr *)(ip + 1);
398bd1d0850SGleb Smirnoff #endif
399db0ac6deSCy Schubert 	struct inpcb *inp;
400db0ac6deSCy Schubert 	struct mbuf *n;
401db0ac6deSCy Schubert 	int appends = 0;
402db0ac6deSCy Schubert 
403db0ac6deSCy Schubert 	MPASS(ip->ip_hl == sizeof(struct ip) >> 2);
404db0ac6deSCy Schubert 
405db0ac6deSCy Schubert 	while ((inp = inp_next(&inpi)) != NULL) {
406db0ac6deSCy Schubert 		/*
407db0ac6deSCy Schubert 		 * XXXRW: Because we weren't holding either the inpcb
408db0ac6deSCy Schubert 		 * or the hash lock when we checked for a match
409db0ac6deSCy Schubert 		 * before, we should probably recheck now that the
410db0ac6deSCy Schubert 		 * inpcb lock is held.
411db0ac6deSCy Schubert 		 */
412db0ac6deSCy Schubert 		/*
413db0ac6deSCy Schubert 		 * Handle socket delivery policy for any-source
414db0ac6deSCy Schubert 		 * and source-specific multicast. [RFC3678]
415db0ac6deSCy Schubert 		 */
416db0ac6deSCy Schubert 		if (IN_MULTICAST(ntohl(ip->ip_dst.s_addr))) {
417db0ac6deSCy Schubert 			struct ip_moptions	*imo;
418db0ac6deSCy Schubert 			struct sockaddr_in	 group;
419db0ac6deSCy Schubert 			int			 blocked;
420db0ac6deSCy Schubert 
421db0ac6deSCy Schubert 			imo = inp->inp_moptions;
422db0ac6deSCy Schubert 			if (imo == NULL)
423db0ac6deSCy Schubert 				continue;
424db0ac6deSCy Schubert 			bzero(&group, sizeof(struct sockaddr_in));
425db0ac6deSCy Schubert 			group.sin_len = sizeof(struct sockaddr_in);
426db0ac6deSCy Schubert 			group.sin_family = AF_INET;
427db0ac6deSCy Schubert 			group.sin_addr = ip->ip_dst;
428db0ac6deSCy Schubert 
429db0ac6deSCy Schubert 			blocked = imo_multi_filter(imo, m->m_pkthdr.rcvif,
430db0ac6deSCy Schubert 				(struct sockaddr *)&group,
431db0ac6deSCy Schubert 				(struct sockaddr *)&udp_in[0]);
432db0ac6deSCy Schubert 			if (blocked != MCAST_PASS) {
433db0ac6deSCy Schubert 				if (blocked == MCAST_NOTGMEMBER)
434db0ac6deSCy Schubert 					IPSTAT_INC(ips_notmember);
435db0ac6deSCy Schubert 				if (blocked == MCAST_NOTSMEMBER ||
436db0ac6deSCy Schubert 				    blocked == MCAST_MUTED)
437db0ac6deSCy Schubert 					UDPSTAT_INC(udps_filtermcast);
438db0ac6deSCy Schubert 				continue;
439db0ac6deSCy Schubert 			}
440db0ac6deSCy Schubert 		}
441db0ac6deSCy Schubert 		if ((n = m_copym(m, 0, M_COPYALL, M_NOWAIT)) != NULL) {
442db0ac6deSCy Schubert 			if (proto == IPPROTO_UDPLITE)
443db0ac6deSCy Schubert 				UDPLITE_PROBE(receive, NULL, inp, ip, inp, uh);
444db0ac6deSCy Schubert 			else
445db0ac6deSCy Schubert 				UDP_PROBE(receive, NULL, inp, ip, inp, uh);
446db0ac6deSCy Schubert 			if (udp_append(inp, ip, n, sizeof(struct ip), udp_in)) {
447db0ac6deSCy Schubert 				INP_RUNLOCK(inp);
448db0ac6deSCy Schubert 				break;
449db0ac6deSCy Schubert 			} else
450db0ac6deSCy Schubert 				appends++;
451db0ac6deSCy Schubert 		}
452db0ac6deSCy Schubert 		/*
453db0ac6deSCy Schubert 		 * Don't look for additional matches if this one does
454db0ac6deSCy Schubert 		 * not have either the SO_REUSEPORT or SO_REUSEADDR
455db0ac6deSCy Schubert 		 * socket options set.  This heuristic avoids
456db0ac6deSCy Schubert 		 * searching through all pcbs in the common case of a
457db0ac6deSCy Schubert 		 * non-shared port.  It assumes that an application
458db0ac6deSCy Schubert 		 * will never clear these options after setting them.
459db0ac6deSCy Schubert 		 */
460db0ac6deSCy Schubert 		if ((inp->inp_socket->so_options &
461db0ac6deSCy Schubert 		    (SO_REUSEPORT|SO_REUSEPORT_LB|SO_REUSEADDR)) == 0) {
462db0ac6deSCy Schubert 			INP_RUNLOCK(inp);
463db0ac6deSCy Schubert 			break;
464db0ac6deSCy Schubert 		}
465db0ac6deSCy Schubert 	}
466db0ac6deSCy Schubert 
467db0ac6deSCy Schubert 	if (appends == 0) {
468db0ac6deSCy Schubert 		/*
469db0ac6deSCy Schubert 		 * No matching pcb found; discard datagram.  (No need
470db0ac6deSCy Schubert 		 * to send an ICMP Port Unreachable for a broadcast
471db0ac6deSCy Schubert 		 * or multicast datgram.)
472db0ac6deSCy Schubert 		 */
473db0ac6deSCy Schubert 		UDPSTAT_INC(udps_noport);
474db0ac6deSCy Schubert 		if (IN_MULTICAST(ntohl(ip->ip_dst.s_addr)))
475db0ac6deSCy Schubert 			UDPSTAT_INC(udps_noportmcast);
476db0ac6deSCy Schubert 		else
477db0ac6deSCy Schubert 			UDPSTAT_INC(udps_noportbcast);
478db0ac6deSCy Schubert 	}
479014f98b1SMark Johnston 	m_freem(m);
480db0ac6deSCy Schubert 
481db0ac6deSCy Schubert 	return (IPPROTO_DONE);
482db0ac6deSCy Schubert }
483db0ac6deSCy Schubert 
4848f5a8818SKevin Lo int
4858f5a8818SKevin Lo udp_input(struct mbuf **mp, int *offp, int proto)
486df8bae1dSRodney W. Grimes {
4873329b236SRobert Watson 	struct ip *ip;
4883329b236SRobert Watson 	struct udphdr *uh;
48971498f30SBruce M Simpson 	struct ifnet *ifp;
4903329b236SRobert Watson 	struct inpcb *inp;
4918f134647SGleb Smirnoff 	uint16_t len, ip_len;
492e06e816fSKevin Lo 	struct inpcbinfo *pcbinfo;
493dce33a45SErmal Luçi 	struct sockaddr_in udp_in[2];
4948f5a8818SKevin Lo 	struct mbuf *m;
4950b4ae859SGleb Smirnoff 	struct m_tag *fwd_tag;
4968f5a8818SKevin Lo 	int cscov_partial, iphlen;
497df8bae1dSRodney W. Grimes 
4988f5a8818SKevin Lo 	m = *mp;
4998f5a8818SKevin Lo 	iphlen = *offp;
50071498f30SBruce M Simpson 	ifp = m->m_pkthdr.rcvif;
5018f5a8818SKevin Lo 	*mp = NULL;
502026decb8SRobert Watson 	UDPSTAT_INC(udps_ipackets);
503df8bae1dSRodney W. Grimes 
504df8bae1dSRodney W. Grimes 	/*
5053329b236SRobert Watson 	 * Strip IP options, if any; should skip this, make available to
5063329b236SRobert Watson 	 * user, and use on returned packets, but we don't yet have a way to
5073329b236SRobert Watson 	 * check the checksum with options still present.
508df8bae1dSRodney W. Grimes 	 */
509df8bae1dSRodney W. Grimes 	if (iphlen > sizeof (struct ip)) {
510105bd211SGleb Smirnoff 		ip_stripoptions(m);
511df8bae1dSRodney W. Grimes 		iphlen = sizeof(struct ip);
512df8bae1dSRodney W. Grimes 	}
513df8bae1dSRodney W. Grimes 
514df8bae1dSRodney W. Grimes 	/*
515df8bae1dSRodney W. Grimes 	 * Get IP and UDP header together in first mbuf.
516df8bae1dSRodney W. Grimes 	 */
517df8bae1dSRodney W. Grimes 	if (m->m_len < iphlen + sizeof(struct udphdr)) {
518d1b18731SKevin Lo 		if ((m = m_pullup(m, iphlen + sizeof(struct udphdr))) == NULL) {
519026decb8SRobert Watson 			UDPSTAT_INC(udps_hdrops);
5208f5a8818SKevin Lo 			return (IPPROTO_DONE);
521df8bae1dSRodney W. Grimes 		}
522df8bae1dSRodney W. Grimes 	}
523503f4e47SBjoern A. Zeeb 	ip = mtod(m, struct ip *);
524df8bae1dSRodney W. Grimes 	uh = (struct udphdr *)((caddr_t)ip + iphlen);
5258f5a8818SKevin Lo 	cscov_partial = (proto == IPPROTO_UDPLITE) ? 1 : 0;
526df8bae1dSRodney W. Grimes 
5273329b236SRobert Watson 	/*
5283329b236SRobert Watson 	 * Destination port of 0 is illegal, based on RFC768.
5293329b236SRobert Watson 	 */
530686cdd19SJun-ichiro itojun Hagino 	if (uh->uh_dport == 0)
531f76fcf6dSJeffrey Hsu 		goto badunlocked;
532686cdd19SJun-ichiro itojun Hagino 
533df8bae1dSRodney W. Grimes 	/*
5343329b236SRobert Watson 	 * Construct sockaddr format source address.  Stuff source address
5353329b236SRobert Watson 	 * and datagram in user buffer.
536b9234fafSSam Leffler 	 */
537dce33a45SErmal Luçi 	bzero(&udp_in[0], sizeof(struct sockaddr_in) * 2);
538dce33a45SErmal Luçi 	udp_in[0].sin_len = sizeof(struct sockaddr_in);
539dce33a45SErmal Luçi 	udp_in[0].sin_family = AF_INET;
540dce33a45SErmal Luçi 	udp_in[0].sin_port = uh->uh_sport;
541dce33a45SErmal Luçi 	udp_in[0].sin_addr = ip->ip_src;
542dce33a45SErmal Luçi 	udp_in[1].sin_len = sizeof(struct sockaddr_in);
543dce33a45SErmal Luçi 	udp_in[1].sin_family = AF_INET;
544dce33a45SErmal Luçi 	udp_in[1].sin_port = uh->uh_dport;
545dce33a45SErmal Luçi 	udp_in[1].sin_addr = ip->ip_dst;
546b9234fafSSam Leffler 
547b9234fafSSam Leffler 	/*
548af1ee11dSRobert Watson 	 * Make mbuf data length reflect UDP length.  If not enough data to
549af1ee11dSRobert Watson 	 * reflect UDP length, drop.
550df8bae1dSRodney W. Grimes 	 */
551df8bae1dSRodney W. Grimes 	len = ntohs((u_short)uh->uh_ulen);
5528ad458a4SGleb Smirnoff 	ip_len = ntohs(ip->ip_len) - iphlen;
5530f4a0366SMichael Tuexen 	if (proto == IPPROTO_UDPLITE && (len == 0 || len == ip_len)) {
554e06e816fSKevin Lo 		/* Zero means checksum over the complete packet. */
5550f4a0366SMichael Tuexen 		if (len == 0)
556e06e816fSKevin Lo 			len = ip_len;
557e06e816fSKevin Lo 		cscov_partial = 0;
558e06e816fSKevin Lo 	}
5598f134647SGleb Smirnoff 	if (ip_len != len) {
5608f134647SGleb Smirnoff 		if (len > ip_len || len < sizeof(struct udphdr)) {
561026decb8SRobert Watson 			UDPSTAT_INC(udps_badlen);
562f76fcf6dSJeffrey Hsu 			goto badunlocked;
563df8bae1dSRodney W. Grimes 		}
5648f5a8818SKevin Lo 		if (proto == IPPROTO_UDP)
5658f134647SGleb Smirnoff 			m_adj(m, len - ip_len);
566df8bae1dSRodney W. Grimes 	}
5673329b236SRobert Watson 
568df8bae1dSRodney W. Grimes 	/*
569df8bae1dSRodney W. Grimes 	 * Checksum extended UDP header and data.
570df8bae1dSRodney W. Grimes 	 */
5716dfab5b1SGarrett Wollman 	if (uh->uh_sum) {
57239629c92SDavid Malone 		u_short uh_sum;
57339629c92SDavid Malone 
574e06e816fSKevin Lo 		if ((m->m_pkthdr.csum_flags & CSUM_DATA_VALID) &&
575e06e816fSKevin Lo 		    !cscov_partial) {
576db4f9cc7SJonathan Lemon 			if (m->m_pkthdr.csum_flags & CSUM_PSEUDO_HDR)
57739629c92SDavid Malone 				uh_sum = m->m_pkthdr.csum_data;
578db4f9cc7SJonathan Lemon 			else
57939629c92SDavid Malone 				uh_sum = in_pseudo(ip->ip_src.s_addr,
580506f4949SRuslan Ermilov 				    ip->ip_dst.s_addr, htonl((u_short)len +
5818f5a8818SKevin Lo 				    m->m_pkthdr.csum_data + proto));
58239629c92SDavid Malone 			uh_sum ^= 0xffff;
583db4f9cc7SJonathan Lemon 		} else {
5843358df29SGleb Smirnoff 			char b[offsetof(struct ipovly, ih_src)];
5853358df29SGleb Smirnoff 			struct ipovly *ipov = (struct ipovly *)ip;
586af1ee11dSRobert Watson 
5873358df29SGleb Smirnoff 			bcopy(ipov, b, sizeof(b));
5883358df29SGleb Smirnoff 			bzero(ipov, sizeof(ipov->ih_x1));
5893358df29SGleb Smirnoff 			ipov->ih_len = (proto == IPPROTO_UDP) ?
590e06e816fSKevin Lo 			    uh->uh_ulen : htons(ip_len);
59139629c92SDavid Malone 			uh_sum = in_cksum(m, len + sizeof (struct ip));
5923358df29SGleb Smirnoff 			bcopy(b, ipov, sizeof(b));
593db4f9cc7SJonathan Lemon 		}
59439629c92SDavid Malone 		if (uh_sum) {
595026decb8SRobert Watson 			UDPSTAT_INC(udps_badsum);
596df8bae1dSRodney W. Grimes 			m_freem(m);
5978f5a8818SKevin Lo 			return (IPPROTO_DONE);
598df8bae1dSRodney W. Grimes 		}
599c6d81a34SMichael Tuexen 	} else {
600c6d81a34SMichael Tuexen 		if (proto == IPPROTO_UDP) {
601026decb8SRobert Watson 			UDPSTAT_INC(udps_nosum);
602c6d81a34SMichael Tuexen 		} else {
603c6d81a34SMichael Tuexen 			/* UDPLite requires a checksum */
604c6d81a34SMichael Tuexen 			/* XXX: What is the right UDPLite MIB counter here? */
605c6d81a34SMichael Tuexen 			m_freem(m);
606c6d81a34SMichael Tuexen 			return (IPPROTO_DONE);
607c6d81a34SMichael Tuexen 		}
608c6d81a34SMichael Tuexen 	}
609df8bae1dSRodney W. Grimes 
610266f97b5SCy Schubert 	if (IN_MULTICAST(ntohl(ip->ip_dst.s_addr)) ||
611db0ac6deSCy Schubert 	    in_broadcast(ip->ip_dst, ifp))
612db0ac6deSCy Schubert 		return (udp_multi_input(m, proto, udp_in));
613266f97b5SCy Schubert 
614db0ac6deSCy Schubert 	pcbinfo = udp_get_inpcbinfo(proto);
6153329b236SRobert Watson 
616df8bae1dSRodney W. Grimes 	/*
6176d6a026bSDavid Greenman 	 * Locate pcb for datagram.
618db0ac6deSCy Schubert 	 *
6198a006adbSBjoern A. Zeeb 	 * Grab info from PACKET_TAG_IPFORWARD tag prepended to the chain.
6208a006adbSBjoern A. Zeeb 	 */
621ffdbf9daSAndrey V. Elsukov 	if ((m->m_flags & M_IP_NEXTHOP) &&
622c1de64a4SAndrey V. Elsukov 	    (fwd_tag = m_tag_find(m, PACKET_TAG_IPFORWARD, NULL)) != NULL) {
6238a006adbSBjoern A. Zeeb 		struct sockaddr_in *next_hop;
6248a006adbSBjoern A. Zeeb 
6258a006adbSBjoern A. Zeeb 		next_hop = (struct sockaddr_in *)(fwd_tag + 1);
6268a006adbSBjoern A. Zeeb 
6278a006adbSBjoern A. Zeeb 		/*
6288a006adbSBjoern A. Zeeb 		 * Transparently forwarded. Pretend to be the destination.
6298a006adbSBjoern A. Zeeb 		 * Already got one like this?
6308a006adbSBjoern A. Zeeb 		 */
631e06e816fSKevin Lo 		inp = in_pcblookup_mbuf(pcbinfo, ip->ip_src, uh->uh_sport,
6328a006adbSBjoern A. Zeeb 		    ip->ip_dst, uh->uh_dport, INPLOOKUP_RLOCKPCB, ifp, m);
6338a006adbSBjoern A. Zeeb 		if (!inp) {
6348a006adbSBjoern A. Zeeb 			/*
6358a006adbSBjoern A. Zeeb 			 * It's new.  Try to find the ambushing socket.
6368a006adbSBjoern A. Zeeb 			 * Because we've rewritten the destination address,
6378a006adbSBjoern A. Zeeb 			 * any hardware-generated hash is ignored.
6388a006adbSBjoern A. Zeeb 			 */
639e06e816fSKevin Lo 			inp = in_pcblookup(pcbinfo, ip->ip_src,
6408a006adbSBjoern A. Zeeb 			    uh->uh_sport, next_hop->sin_addr,
6418a006adbSBjoern A. Zeeb 			    next_hop->sin_port ? htons(next_hop->sin_port) :
6428a006adbSBjoern A. Zeeb 			    uh->uh_dport, INPLOOKUP_WILDCARD |
6438a006adbSBjoern A. Zeeb 			    INPLOOKUP_RLOCKPCB, ifp);
6448a006adbSBjoern A. Zeeb 		}
6458a006adbSBjoern A. Zeeb 		/* Remove the tag from the packet. We don't need it anymore. */
6468a006adbSBjoern A. Zeeb 		m_tag_delete(m, fwd_tag);
647ffdbf9daSAndrey V. Elsukov 		m->m_flags &= ~M_IP_NEXTHOP;
6488a006adbSBjoern A. Zeeb 	} else
649e06e816fSKevin Lo 		inp = in_pcblookup_mbuf(pcbinfo, ip->ip_src, uh->uh_sport,
6508a006adbSBjoern A. Zeeb 		    ip->ip_dst, uh->uh_dport, INPLOOKUP_WILDCARD |
6518a006adbSBjoern A. Zeeb 		    INPLOOKUP_RLOCKPCB, ifp, m);
65215bd2b43SDavid Greenman 	if (inp == NULL) {
653334fc582SBjoern A. Zeeb 		if (V_udp_log_in_vain) {
654edf0313bSEric van Gyzen 			char src[INET_ADDRSTRLEN];
655edf0313bSEric van Gyzen 			char dst[INET_ADDRSTRLEN];
65675cfc95fSAndrey A. Chernov 
657592071e8SBruce Evans 			log(LOG_INFO,
658592071e8SBruce Evans 			    "Connection attempt to UDP %s:%d from %s:%d\n",
659edf0313bSEric van Gyzen 			    inet_ntoa_r(ip->ip_dst, dst), ntohs(uh->uh_dport),
660edf0313bSEric van Gyzen 			    inet_ntoa_r(ip->ip_src, src), ntohs(uh->uh_sport));
66175cfc95fSAndrey A. Chernov 		}
6627bda9663SMichael Tuexen 		if (proto == IPPROTO_UDPLITE)
6637bda9663SMichael Tuexen 			UDPLITE_PROBE(receive, NULL, NULL, ip, NULL, uh);
6647bda9663SMichael Tuexen 		else
665e1526d5aSMichael Tuexen 			UDP_PROBE(receive, NULL, NULL, ip, NULL, uh);
666026decb8SRobert Watson 		UDPSTAT_INC(udps_noport);
667df8bae1dSRodney W. Grimes 		if (m->m_flags & (M_BCAST | M_MCAST)) {
668026decb8SRobert Watson 			UDPSTAT_INC(udps_noportbcast);
669fa046d87SRobert Watson 			goto badunlocked;
670df8bae1dSRodney W. Grimes 		}
6713ea9a7cfSGleb Smirnoff 		if (V_udp_blackhole && (V_udp_blackhole_local ||
6723ea9a7cfSGleb Smirnoff 		    !in_localip(ip->ip_src)))
673fa046d87SRobert Watson 			goto badunlocked;
6741cbd978eSLuigi Rizzo 		if (badport_bandlim(BANDLIM_ICMP_UNREACH) < 0)
675fa046d87SRobert Watson 			goto badunlocked;
676582a7760SBruce Evans 		icmp_error(m, ICMP_UNREACH, ICMP_UNREACH_PORT, 0, 0);
6778f5a8818SKevin Lo 		return (IPPROTO_DONE);
678df8bae1dSRodney W. Grimes 	}
6793329b236SRobert Watson 
6803329b236SRobert Watson 	/*
6813329b236SRobert Watson 	 * Check the minimum TTL for socket.
6823329b236SRobert Watson 	 */
683fa046d87SRobert Watson 	INP_RLOCK_ASSERT(inp);
68410cc62b7SRobert Watson 	if (inp->inp_ip_minttl && inp->inp_ip_minttl > ip->ip_ttl) {
6857bda9663SMichael Tuexen 		if (proto == IPPROTO_UDPLITE)
6867bda9663SMichael Tuexen 			UDPLITE_PROBE(receive, NULL, inp, ip, inp, uh);
6877bda9663SMichael Tuexen 		else
688e1526d5aSMichael Tuexen 			UDP_PROBE(receive, NULL, inp, ip, inp, uh);
68910cc62b7SRobert Watson 		INP_RUNLOCK(inp);
690fa046d87SRobert Watson 		m_freem(m);
6918f5a8818SKevin Lo 		return (IPPROTO_DONE);
69210cc62b7SRobert Watson 	}
693e06e816fSKevin Lo 	if (cscov_partial) {
694e06e816fSKevin Lo 		struct udpcb *up;
695e06e816fSKevin Lo 
696e06e816fSKevin Lo 		up = intoudpcb(inp);
69783e95fb3SMichael Tuexen 		if (up->u_rxcslen == 0 || up->u_rxcslen > len) {
698e06e816fSKevin Lo 			INP_RUNLOCK(inp);
699e06e816fSKevin Lo 			m_freem(m);
7008f5a8818SKevin Lo 			return (IPPROTO_DONE);
701e06e816fSKevin Lo 		}
702e06e816fSKevin Lo 	}
70357f60867SMark Johnston 
7047bda9663SMichael Tuexen 	if (proto == IPPROTO_UDPLITE)
7057bda9663SMichael Tuexen 		UDPLITE_PROBE(receive, NULL, inp, ip, inp, uh);
7067bda9663SMichael Tuexen 	else
7071ad19fb6SMark Johnston 		UDP_PROBE(receive, NULL, inp, ip, inp, uh);
708dce33a45SErmal Luçi 	if (udp_append(inp, ip, m, iphlen, udp_in) == 0)
709119d85f6SRobert Watson 		INP_RUNLOCK(inp);
7108f5a8818SKevin Lo 	return (IPPROTO_DONE);
71161ffc0b1SJeffrey Hsu 
712f76fcf6dSJeffrey Hsu badunlocked:
713df8bae1dSRodney W. Grimes 	m_freem(m);
7148f5a8818SKevin Lo 	return (IPPROTO_DONE);
715cfa1ca9dSYoshinobu Inoue }
71679288c11SBjoern A. Zeeb #endif /* INET */
717cfa1ca9dSYoshinobu Inoue 
718cfa1ca9dSYoshinobu Inoue /*
7193329b236SRobert Watson  * Notify a udp user of an asynchronous error; just wake up so that they can
7203329b236SRobert Watson  * collect error status.
721df8bae1dSRodney W. Grimes  */
7223ce144eaSJeffrey Hsu struct inpcb *
7233329b236SRobert Watson udp_notify(struct inpcb *inp, int errno)
724df8bae1dSRodney W. Grimes {
7253329b236SRobert Watson 
726083a010cSRyan Stone 	INP_WLOCK_ASSERT(inp);
72784cc0778SGeorge V. Neville-Neil 	if ((errno == EHOSTUNREACH || errno == ENETUNREACH ||
728983066f0SAlexander V. Chernikov 	     errno == EHOSTDOWN) && inp->inp_route.ro_nh) {
729983066f0SAlexander V. Chernikov 		NH_FREE(inp->inp_route.ro_nh);
730983066f0SAlexander V. Chernikov 		inp->inp_route.ro_nh = (struct nhop_object *)NULL;
73184cc0778SGeorge V. Neville-Neil 	}
7328501a69cSRobert Watson 
733df8bae1dSRodney W. Grimes 	inp->inp_socket->so_error = errno;
734df8bae1dSRodney W. Grimes 	sorwakeup(inp->inp_socket);
735df8bae1dSRodney W. Grimes 	sowwakeup(inp->inp_socket);
7363329b236SRobert Watson 	return (inp);
737df8bae1dSRodney W. Grimes }
738df8bae1dSRodney W. Grimes 
73979288c11SBjoern A. Zeeb #ifdef INET
740e06e816fSKevin Lo static void
741e06e816fSKevin Lo udp_common_ctlinput(int cmd, struct sockaddr *sa, void *vip,
742e06e816fSKevin Lo     struct inpcbinfo *pcbinfo)
743df8bae1dSRodney W. Grimes {
744c693a045SJonathan Lemon 	struct ip *ip = vip;
745c693a045SJonathan Lemon 	struct udphdr *uh;
746c693a045SJonathan Lemon 	struct in_addr faddr;
747c693a045SJonathan Lemon 	struct inpcb *inp;
748c693a045SJonathan Lemon 
749c693a045SJonathan Lemon 	faddr = ((struct sockaddr_in *)sa)->sin_addr;
750c693a045SJonathan Lemon 	if (sa->sa_family != AF_INET || faddr.s_addr == INADDR_ANY)
751c693a045SJonathan Lemon 		return;
752df8bae1dSRodney W. Grimes 
75384cc0778SGeorge V. Neville-Neil 	if (PRC_IS_REDIRECT(cmd)) {
75484cc0778SGeorge V. Neville-Neil 		/* signal EHOSTDOWN, as it flushes the cached route */
7554760956eSMichael Tuexen 		in_pcbnotifyall(pcbinfo, faddr, EHOSTDOWN, udp_notify);
75697d8d152SAndre Oppermann 		return;
75784cc0778SGeorge V. Neville-Neil 	}
7583329b236SRobert Watson 
75997d8d152SAndre Oppermann 	/*
76097d8d152SAndre Oppermann 	 * Hostdead is ugly because it goes linearly through all PCBs.
7613329b236SRobert Watson 	 *
7623329b236SRobert Watson 	 * XXX: We never get this from ICMP, otherwise it makes an excellent
7633329b236SRobert Watson 	 * DoS attack on machines with many connections.
76497d8d152SAndre Oppermann 	 */
76597d8d152SAndre Oppermann 	if (cmd == PRC_HOSTDEAD)
766af1ee11dSRobert Watson 		ip = NULL;
767d1c54148SJesper Skriver 	else if ((unsigned)cmd >= PRC_NCMDS || inetctlerrmap[cmd] == 0)
768df8bae1dSRodney W. Grimes 		return;
769af1ee11dSRobert Watson 	if (ip != NULL) {
770df8bae1dSRodney W. Grimes 		uh = (struct udphdr *)((caddr_t)ip + (ip->ip_hl << 2));
771e06e816fSKevin Lo 		inp = in_pcblookup(pcbinfo, faddr, uh->uh_dport,
772083a010cSRyan Stone 		    ip->ip_src, uh->uh_sport, INPLOOKUP_WLOCKPCB, NULL);
773f76fcf6dSJeffrey Hsu 		if (inp != NULL) {
774083a010cSRyan Stone 			INP_WLOCK_ASSERT(inp);
775f76fcf6dSJeffrey Hsu 			if (inp->inp_socket != NULL) {
776f5514f08SRobert Watson 				udp_notify(inp, inetctlerrmap[cmd]);
777f76fcf6dSJeffrey Hsu 			}
778083a010cSRyan Stone 			INP_WUNLOCK(inp);
779abb901c5SRandall Stewart 		} else {
780abb901c5SRandall Stewart 			inp = in_pcblookup(pcbinfo, faddr, uh->uh_dport,
781abb901c5SRandall Stewart 					   ip->ip_src, uh->uh_sport,
782abb901c5SRandall Stewart 					   INPLOOKUP_WILDCARD | INPLOOKUP_RLOCKPCB, NULL);
783abb901c5SRandall Stewart 			if (inp != NULL) {
784abb901c5SRandall Stewart 				struct udpcb *up;
78546374cbfSMatt Macy 				void *ctx;
78646374cbfSMatt Macy 				udp_tun_icmp_t func;
787abb901c5SRandall Stewart 
788abb901c5SRandall Stewart 				up = intoudpcb(inp);
78946374cbfSMatt Macy 				ctx = up->u_tun_ctx;
79046374cbfSMatt Macy 				func = up->u_icmp_func;
791abb901c5SRandall Stewart 				INP_RUNLOCK(inp);
79246374cbfSMatt Macy 				if (func != NULL)
79346374cbfSMatt Macy 					(*func)(cmd, sa, vip, ctx);
794abb901c5SRandall Stewart 			}
795f76fcf6dSJeffrey Hsu 		}
796df8bae1dSRodney W. Grimes 	} else
797e06e816fSKevin Lo 		in_pcbnotifyall(pcbinfo, faddr, inetctlerrmap[cmd],
798f5514f08SRobert Watson 		    udp_notify);
799df8bae1dSRodney W. Grimes }
800e06e816fSKevin Lo void
801e06e816fSKevin Lo udp_ctlinput(int cmd, struct sockaddr *sa, void *vip)
802e06e816fSKevin Lo {
803e06e816fSKevin Lo 
804e06e816fSKevin Lo 	return (udp_common_ctlinput(cmd, sa, vip, &V_udbinfo));
805e06e816fSKevin Lo }
806e06e816fSKevin Lo 
807e06e816fSKevin Lo void
808e06e816fSKevin Lo udplite_ctlinput(int cmd, struct sockaddr *sa, void *vip)
809e06e816fSKevin Lo {
810e06e816fSKevin Lo 
811e06e816fSKevin Lo 	return (udp_common_ctlinput(cmd, sa, vip, &V_ulitecbinfo));
812e06e816fSKevin Lo }
81379288c11SBjoern A. Zeeb #endif /* INET */
814df8bae1dSRodney W. Grimes 
8150312fbe9SPoul-Henning Kamp static int
81682d9ae4eSPoul-Henning Kamp udp_pcblist(SYSCTL_HANDLER_ARGS)
81798271db4SGarrett Wollman {
818db0ac6deSCy Schubert 	struct inpcb_iterator inpi = INP_ALL_ITERATOR(&V_udbinfo,
819db0ac6deSCy Schubert 	    INPLOOKUP_RLOCKPCB);
82098271db4SGarrett Wollman 	struct xinpgen xig;
821032677ceSGleb Smirnoff 	struct inpcb *inp;
822032677ceSGleb Smirnoff 	int error;
82398271db4SGarrett Wollman 
824032677ceSGleb Smirnoff 	if (req->newptr != 0)
825032677ceSGleb Smirnoff 		return (EPERM);
826032677ceSGleb Smirnoff 
82798271db4SGarrett Wollman 	if (req->oldptr == 0) {
828032677ceSGleb Smirnoff 		int n;
829032677ceSGleb Smirnoff 
830603724d3SBjoern A. Zeeb 		n = V_udbinfo.ipi_count;
831c007b96aSJohn Baldwin 		n += imax(n / 8, 10);
832c007b96aSJohn Baldwin 		req->oldidx = 2 * (sizeof xig) + n * sizeof(struct xinpcb);
8333329b236SRobert Watson 		return (0);
83498271db4SGarrett Wollman 	}
83598271db4SGarrett Wollman 
836032677ceSGleb Smirnoff 	if ((error = sysctl_wire_old_buffer(req, 0)) != 0)
83747934cefSDon Lewis 		return (error);
8385c38b6dbSDon Lewis 
83979db6fe7SMark Johnston 	bzero(&xig, sizeof(xig));
84098271db4SGarrett Wollman 	xig.xig_len = sizeof xig;
841032677ceSGleb Smirnoff 	xig.xig_count = V_udbinfo.ipi_count;
842032677ceSGleb Smirnoff 	xig.xig_gen = V_udbinfo.ipi_gencnt;
84398271db4SGarrett Wollman 	xig.xig_sogen = so_gencnt;
84498271db4SGarrett Wollman 	error = SYSCTL_OUT(req, &xig, sizeof xig);
84598271db4SGarrett Wollman 	if (error)
8463329b236SRobert Watson 		return (error);
84799208b82SMatt Macy 
848db0ac6deSCy Schubert 	while ((inp = inp_next(&inpi)) != NULL) {
849032677ceSGleb Smirnoff 		if (inp->inp_gencnt <= xig.xig_gen &&
850032677ceSGleb Smirnoff 		    cr_canseeinpcb(req->td->td_ucred, inp) == 0) {
85198271db4SGarrett Wollman 			struct xinpcb xi;
852d0e157f6SBjoern A. Zeeb 
853cc65eb4eSGleb Smirnoff 			in_pcbtoxinpcb(inp, &xi);
854266f97b5SCy Schubert 			error = SYSCTL_OUT(req, &xi, sizeof xi);
855db0ac6deSCy Schubert 			if (error) {
856266f97b5SCy Schubert 				INP_RUNLOCK(inp);
857db0ac6deSCy Schubert 				break;
85898271db4SGarrett Wollman 			}
859db0ac6deSCy Schubert 		}
860db0ac6deSCy Schubert 	}
861d0e157f6SBjoern A. Zeeb 
86298271db4SGarrett Wollman 	if (!error) {
86398271db4SGarrett Wollman 		/*
8643329b236SRobert Watson 		 * Give the user an updated idea of our state.  If the
8653329b236SRobert Watson 		 * generation differs from what we told her before, she knows
8663329b236SRobert Watson 		 * that something happened while we were processing this
8673329b236SRobert Watson 		 * request, and it might be necessary to retry.
86898271db4SGarrett Wollman 		 */
869603724d3SBjoern A. Zeeb 		xig.xig_gen = V_udbinfo.ipi_gencnt;
87098271db4SGarrett Wollman 		xig.xig_sogen = so_gencnt;
871603724d3SBjoern A. Zeeb 		xig.xig_count = V_udbinfo.ipi_count;
87298271db4SGarrett Wollman 		error = SYSCTL_OUT(req, &xig, sizeof xig);
87398271db4SGarrett Wollman 	}
874032677ceSGleb Smirnoff 
8753329b236SRobert Watson 	return (error);
87698271db4SGarrett Wollman }
87798271db4SGarrett Wollman 
87879c3d51bSMatthew D Fleming SYSCTL_PROC(_net_inet_udp, UDPCTL_PCBLIST, pcblist,
8797029da5cSPawel Biernacki     CTLTYPE_OPAQUE | CTLFLAG_RD | CTLFLAG_MPSAFE, NULL, 0,
8807029da5cSPawel Biernacki     udp_pcblist, "S,xinpcb",
8817029da5cSPawel Biernacki     "List of active UDP sockets");
88298271db4SGarrett Wollman 
88379288c11SBjoern A. Zeeb #ifdef INET
88498271db4SGarrett Wollman static int
88582d9ae4eSPoul-Henning Kamp udp_getcred(SYSCTL_HANDLER_ARGS)
886490d50b6SBrian Feldman {
887c0511d3bSBrian Feldman 	struct xucred xuc;
888490d50b6SBrian Feldman 	struct sockaddr_in addrs[2];
889d797164aSGleb Smirnoff 	struct epoch_tracker et;
890490d50b6SBrian Feldman 	struct inpcb *inp;
891277afaffSRobert Watson 	int error;
892490d50b6SBrian Feldman 
89332f9753cSRobert Watson 	error = priv_check(req->td, PRIV_NETINET_GETCRED);
894490d50b6SBrian Feldman 	if (error)
895490d50b6SBrian Feldman 		return (error);
896490d50b6SBrian Feldman 	error = SYSCTL_IN(req, addrs, sizeof(addrs));
897490d50b6SBrian Feldman 	if (error)
898490d50b6SBrian Feldman 		return (error);
899d797164aSGleb Smirnoff 	NET_EPOCH_ENTER(et);
900fa046d87SRobert Watson 	inp = in_pcblookup(&V_udbinfo, addrs[1].sin_addr, addrs[1].sin_port,
901fa046d87SRobert Watson 	    addrs[0].sin_addr, addrs[0].sin_port,
902fa046d87SRobert Watson 	    INPLOOKUP_WILDCARD | INPLOOKUP_RLOCKPCB, NULL);
903d797164aSGleb Smirnoff 	NET_EPOCH_EXIT(et);
9049622e84fSRobert Watson 	if (inp != NULL) {
905fa046d87SRobert Watson 		INP_RLOCK_ASSERT(inp);
9069622e84fSRobert Watson 		if (inp->inp_socket == NULL)
9079622e84fSRobert Watson 			error = ENOENT;
9089622e84fSRobert Watson 		if (error == 0)
909f08ef6c5SBjoern A. Zeeb 			error = cr_canseeinpcb(req->td->td_ucred, inp);
9109622e84fSRobert Watson 		if (error == 0)
91186d02c5cSBjoern A. Zeeb 			cru2x(inp->inp_cred, &xuc);
9129622e84fSRobert Watson 		INP_RUNLOCK(inp);
913fa046d87SRobert Watson 	} else
9149622e84fSRobert Watson 		error = ENOENT;
9150e1eebb8SDon Lewis 	if (error == 0)
9160e1eebb8SDon Lewis 		error = SYSCTL_OUT(req, &xuc, sizeof(struct xucred));
917490d50b6SBrian Feldman 	return (error);
918490d50b6SBrian Feldman }
919490d50b6SBrian Feldman 
9207ce87f12SDavid Malone SYSCTL_PROC(_net_inet_udp, OID_AUTO, getcred,
9217029da5cSPawel Biernacki     CTLTYPE_OPAQUE | CTLFLAG_RW | CTLFLAG_PRISON | CTLFLAG_MPSAFE,
9227029da5cSPawel Biernacki     0, 0, udp_getcred, "S,xucred",
9237029da5cSPawel Biernacki     "Get the xucred of a UDP connection");
92479288c11SBjoern A. Zeeb #endif /* INET */
925490d50b6SBrian Feldman 
9267b495c44SVANHULLEBUS Yvan int
9277b495c44SVANHULLEBUS Yvan udp_ctloutput(struct socket *so, struct sockopt *sopt)
9287b495c44SVANHULLEBUS Yvan {
9297b495c44SVANHULLEBUS Yvan 	struct inpcb *inp;
9307b495c44SVANHULLEBUS Yvan 	struct udpcb *up;
931e06e816fSKevin Lo 	int isudplite, error, optval;
9327b495c44SVANHULLEBUS Yvan 
933e06e816fSKevin Lo 	error = 0;
934e06e816fSKevin Lo 	isudplite = (so->so_proto->pr_protocol == IPPROTO_UDPLITE) ? 1 : 0;
9357b495c44SVANHULLEBUS Yvan 	inp = sotoinpcb(so);
9367b495c44SVANHULLEBUS Yvan 	KASSERT(inp != NULL, ("%s: inp == NULL", __func__));
9377b495c44SVANHULLEBUS Yvan 	INP_WLOCK(inp);
938e06e816fSKevin Lo 	if (sopt->sopt_level != so->so_proto->pr_protocol) {
9397b495c44SVANHULLEBUS Yvan #ifdef INET6
9407b495c44SVANHULLEBUS Yvan 		if (INP_CHECK_SOCKAF(so, AF_INET6)) {
9417b495c44SVANHULLEBUS Yvan 			INP_WUNLOCK(inp);
9427b495c44SVANHULLEBUS Yvan 			error = ip6_ctloutput(so, sopt);
94379288c11SBjoern A. Zeeb 		}
9447b495c44SVANHULLEBUS Yvan #endif
94579288c11SBjoern A. Zeeb #if defined(INET) && defined(INET6)
94679288c11SBjoern A. Zeeb 		else
94779288c11SBjoern A. Zeeb #endif
94879288c11SBjoern A. Zeeb #ifdef INET
94979288c11SBjoern A. Zeeb 		{
9507b495c44SVANHULLEBUS Yvan 			INP_WUNLOCK(inp);
9517b495c44SVANHULLEBUS Yvan 			error = ip_ctloutput(so, sopt);
9527b495c44SVANHULLEBUS Yvan 		}
9537b495c44SVANHULLEBUS Yvan #endif
9547b495c44SVANHULLEBUS Yvan 		return (error);
9557b495c44SVANHULLEBUS Yvan 	}
9567b495c44SVANHULLEBUS Yvan 
9577b495c44SVANHULLEBUS Yvan 	switch (sopt->sopt_dir) {
9587b495c44SVANHULLEBUS Yvan 	case SOPT_SET:
9597b495c44SVANHULLEBUS Yvan 		switch (sopt->sopt_name) {
960fcf59617SAndrey V. Elsukov #if defined(IPSEC) || defined(IPSEC_SUPPORT)
961fcf59617SAndrey V. Elsukov #ifdef INET
9627b495c44SVANHULLEBUS Yvan 		case UDP_ENCAP:
963fcf59617SAndrey V. Elsukov 			if (!IPSEC_ENABLED(ipv4)) {
9647b495c44SVANHULLEBUS Yvan 				INP_WUNLOCK(inp);
965fcf59617SAndrey V. Elsukov 				return (ENOPROTOOPT);
9667b495c44SVANHULLEBUS Yvan 			}
967fcf59617SAndrey V. Elsukov 			error = UDPENCAP_PCBCTL(inp, sopt);
9687b495c44SVANHULLEBUS Yvan 			break;
969fcf59617SAndrey V. Elsukov #endif /* INET */
970fcf59617SAndrey V. Elsukov #endif /* IPSEC */
971e06e816fSKevin Lo 		case UDPLITE_SEND_CSCOV:
972e06e816fSKevin Lo 		case UDPLITE_RECV_CSCOV:
973e06e816fSKevin Lo 			if (!isudplite) {
974e06e816fSKevin Lo 				INP_WUNLOCK(inp);
975e06e816fSKevin Lo 				error = ENOPROTOOPT;
976e06e816fSKevin Lo 				break;
977e06e816fSKevin Lo 			}
978e06e816fSKevin Lo 			INP_WUNLOCK(inp);
979e06e816fSKevin Lo 			error = sooptcopyin(sopt, &optval, sizeof(optval),
980e06e816fSKevin Lo 			    sizeof(optval));
981e06e816fSKevin Lo 			if (error != 0)
982e06e816fSKevin Lo 				break;
983e06e816fSKevin Lo 			inp = sotoinpcb(so);
984e06e816fSKevin Lo 			KASSERT(inp != NULL, ("%s: inp == NULL", __func__));
985e06e816fSKevin Lo 			INP_WLOCK(inp);
986e06e816fSKevin Lo 			up = intoudpcb(inp);
987e06e816fSKevin Lo 			KASSERT(up != NULL, ("%s: up == NULL", __func__));
98803f90784SMichael Tuexen 			if ((optval != 0 && optval < 8) || (optval > 65535)) {
989e06e816fSKevin Lo 				INP_WUNLOCK(inp);
990e06e816fSKevin Lo 				error = EINVAL;
991e06e816fSKevin Lo 				break;
992e06e816fSKevin Lo 			}
993e06e816fSKevin Lo 			if (sopt->sopt_name == UDPLITE_SEND_CSCOV)
994e06e816fSKevin Lo 				up->u_txcslen = optval;
995e06e816fSKevin Lo 			else
996e06e816fSKevin Lo 				up->u_rxcslen = optval;
997e06e816fSKevin Lo 			INP_WUNLOCK(inp);
998e06e816fSKevin Lo 			break;
9997b495c44SVANHULLEBUS Yvan 		default:
10007b495c44SVANHULLEBUS Yvan 			INP_WUNLOCK(inp);
10017b495c44SVANHULLEBUS Yvan 			error = ENOPROTOOPT;
10027b495c44SVANHULLEBUS Yvan 			break;
10037b495c44SVANHULLEBUS Yvan 		}
10047b495c44SVANHULLEBUS Yvan 		break;
10057b495c44SVANHULLEBUS Yvan 	case SOPT_GET:
10067b495c44SVANHULLEBUS Yvan 		switch (sopt->sopt_name) {
1007fcf59617SAndrey V. Elsukov #if defined(IPSEC) || defined(IPSEC_SUPPORT)
1008fcf59617SAndrey V. Elsukov #ifdef INET
10097b495c44SVANHULLEBUS Yvan 		case UDP_ENCAP:
1010fcf59617SAndrey V. Elsukov 			if (!IPSEC_ENABLED(ipv4)) {
10117b495c44SVANHULLEBUS Yvan 				INP_WUNLOCK(inp);
1012fcf59617SAndrey V. Elsukov 				return (ENOPROTOOPT);
1013fcf59617SAndrey V. Elsukov 			}
1014fcf59617SAndrey V. Elsukov 			error = UDPENCAP_PCBCTL(inp, sopt);
10157b495c44SVANHULLEBUS Yvan 			break;
1016fcf59617SAndrey V. Elsukov #endif /* INET */
1017fcf59617SAndrey V. Elsukov #endif /* IPSEC */
1018e06e816fSKevin Lo 		case UDPLITE_SEND_CSCOV:
1019e06e816fSKevin Lo 		case UDPLITE_RECV_CSCOV:
1020e06e816fSKevin Lo 			if (!isudplite) {
1021e06e816fSKevin Lo 				INP_WUNLOCK(inp);
1022e06e816fSKevin Lo 				error = ENOPROTOOPT;
1023e06e816fSKevin Lo 				break;
1024e06e816fSKevin Lo 			}
1025e06e816fSKevin Lo 			up = intoudpcb(inp);
1026e06e816fSKevin Lo 			KASSERT(up != NULL, ("%s: up == NULL", __func__));
1027e06e816fSKevin Lo 			if (sopt->sopt_name == UDPLITE_SEND_CSCOV)
1028e06e816fSKevin Lo 				optval = up->u_txcslen;
1029e06e816fSKevin Lo 			else
1030e06e816fSKevin Lo 				optval = up->u_rxcslen;
1031e06e816fSKevin Lo 			INP_WUNLOCK(inp);
1032e06e816fSKevin Lo 			error = sooptcopyout(sopt, &optval, sizeof(optval));
1033e06e816fSKevin Lo 			break;
10347b495c44SVANHULLEBUS Yvan 		default:
10357b495c44SVANHULLEBUS Yvan 			INP_WUNLOCK(inp);
10367b495c44SVANHULLEBUS Yvan 			error = ENOPROTOOPT;
10377b495c44SVANHULLEBUS Yvan 			break;
10387b495c44SVANHULLEBUS Yvan 		}
10397b495c44SVANHULLEBUS Yvan 		break;
10407b495c44SVANHULLEBUS Yvan 	}
10417b495c44SVANHULLEBUS Yvan 	return (error);
10427b495c44SVANHULLEBUS Yvan }
10437b495c44SVANHULLEBUS Yvan 
104479288c11SBjoern A. Zeeb #ifdef INET
1045a9839c4aSBjoern A. Zeeb #ifdef INET6
1046a9839c4aSBjoern A. Zeeb /* The logic here is derived from ip6_setpktopt(). See comments there. */
1047a9839c4aSBjoern A. Zeeb static int
1048a9839c4aSBjoern A. Zeeb udp_v4mapped_pktinfo(struct cmsghdr *cm, struct sockaddr_in * src,
1049a9839c4aSBjoern A. Zeeb     struct inpcb *inp, int flags)
1050a9839c4aSBjoern A. Zeeb {
1051a9839c4aSBjoern A. Zeeb 	struct ifnet *ifp;
1052a9839c4aSBjoern A. Zeeb 	struct in6_pktinfo *pktinfo;
1053a9839c4aSBjoern A. Zeeb 	struct in_addr ia;
1054a9839c4aSBjoern A. Zeeb 
1055a9839c4aSBjoern A. Zeeb 	if ((flags & PRUS_IPV6) == 0)
1056a9839c4aSBjoern A. Zeeb 		return (0);
1057a9839c4aSBjoern A. Zeeb 
1058a9839c4aSBjoern A. Zeeb 	if (cm->cmsg_level != IPPROTO_IPV6)
1059a9839c4aSBjoern A. Zeeb 		return (0);
1060a9839c4aSBjoern A. Zeeb 
1061a9839c4aSBjoern A. Zeeb 	if  (cm->cmsg_type != IPV6_2292PKTINFO &&
1062a9839c4aSBjoern A. Zeeb 	    cm->cmsg_type != IPV6_PKTINFO)
1063a9839c4aSBjoern A. Zeeb 		return (0);
1064a9839c4aSBjoern A. Zeeb 
1065a9839c4aSBjoern A. Zeeb 	if (cm->cmsg_len !=
1066a9839c4aSBjoern A. Zeeb 	    CMSG_LEN(sizeof(struct in6_pktinfo)))
1067a9839c4aSBjoern A. Zeeb 		return (EINVAL);
1068a9839c4aSBjoern A. Zeeb 
1069a9839c4aSBjoern A. Zeeb 	pktinfo = (struct in6_pktinfo *)CMSG_DATA(cm);
1070a9839c4aSBjoern A. Zeeb 	if (!IN6_IS_ADDR_V4MAPPED(&pktinfo->ipi6_addr) &&
1071a9839c4aSBjoern A. Zeeb 	    !IN6_IS_ADDR_UNSPECIFIED(&pktinfo->ipi6_addr))
1072a9839c4aSBjoern A. Zeeb 		return (EINVAL);
1073a9839c4aSBjoern A. Zeeb 
1074a9839c4aSBjoern A. Zeeb 	/* Validate the interface index if specified. */
1075a9839c4aSBjoern A. Zeeb 	if (pktinfo->ipi6_ifindex) {
1076d74b7baeSGleb Smirnoff 		struct epoch_tracker et;
1077d74b7baeSGleb Smirnoff 
1078d74b7baeSGleb Smirnoff 		NET_EPOCH_ENTER(et);
1079a9839c4aSBjoern A. Zeeb 		ifp = ifnet_byindex(pktinfo->ipi6_ifindex);
1080d74b7baeSGleb Smirnoff 		NET_EPOCH_EXIT(et);	/* XXXGL: unsafe ifp */
1081a9839c4aSBjoern A. Zeeb 		if (ifp == NULL)
1082a9839c4aSBjoern A. Zeeb 			return (ENXIO);
1083d74b7baeSGleb Smirnoff 	} else
1084d74b7baeSGleb Smirnoff 		ifp = NULL;
1085a9839c4aSBjoern A. Zeeb 	if (ifp != NULL && !IN6_IS_ADDR_UNSPECIFIED(&pktinfo->ipi6_addr)) {
1086a9839c4aSBjoern A. Zeeb 		ia.s_addr = pktinfo->ipi6_addr.s6_addr32[3];
1087a9839c4aSBjoern A. Zeeb 		if (in_ifhasaddr(ifp, ia) == 0)
1088a9839c4aSBjoern A. Zeeb 			return (EADDRNOTAVAIL);
1089a9839c4aSBjoern A. Zeeb 	}
1090a9839c4aSBjoern A. Zeeb 
1091a9839c4aSBjoern A. Zeeb 	bzero(src, sizeof(*src));
1092a9839c4aSBjoern A. Zeeb 	src->sin_family = AF_INET;
1093a9839c4aSBjoern A. Zeeb 	src->sin_len = sizeof(*src);
1094a9839c4aSBjoern A. Zeeb 	src->sin_port = inp->inp_lport;
1095a9839c4aSBjoern A. Zeeb 	src->sin_addr.s_addr = pktinfo->ipi6_addr.s6_addr32[3];
1096a9839c4aSBjoern A. Zeeb 
1097a9839c4aSBjoern A. Zeeb 	return (0);
1098a9839c4aSBjoern A. Zeeb }
1099a9839c4aSBjoern A. Zeeb #endif
1100a9839c4aSBjoern A. Zeeb 
1101490d50b6SBrian Feldman static int
11023329b236SRobert Watson udp_output(struct inpcb *inp, struct mbuf *m, struct sockaddr *addr,
1103a9839c4aSBjoern A. Zeeb     struct mbuf *control, struct thread *td, int flags)
1104df8bae1dSRodney W. Grimes {
11053329b236SRobert Watson 	struct udpiphdr *ui;
11063329b236SRobert Watson 	int len = m->m_pkthdr.len;
110790162a4eSIan Dowse 	struct in_addr faddr, laddr;
1108c557ae16SIan Dowse 	struct cmsghdr *cm;
1109e06e816fSKevin Lo 	struct inpcbinfo *pcbinfo;
1110c557ae16SIan Dowse 	struct sockaddr_in *sin, src;
11116573d758SMatt Macy 	struct epoch_tracker et;
1112e06e816fSKevin Lo 	int cscov_partial = 0;
111390162a4eSIan Dowse 	int error = 0;
1114374ce248SMitchell Horne 	int ipflags = 0;
111590162a4eSIan Dowse 	u_short fport, lport;
1116f584d74bSMichael Tuexen 	u_char tos;
1117e06e816fSKevin Lo 	uint8_t pr;
1118e06e816fSKevin Lo 	uint16_t cscov = 0;
11199d3ddf43SAdrian Chadd 	uint32_t flowid = 0;
1120c2529042SHans Petter Selasky 	uint8_t flowtype = M_HASHTYPE_NONE;
1121df8bae1dSRodney W. Grimes 
1122430d30d8SBill Fenner 	if (len + sizeof(struct udpiphdr) > IP_MAXPACKET) {
1123c557ae16SIan Dowse 		if (control)
1124c557ae16SIan Dowse 			m_freem(control);
11255c32ea65SRobert Watson 		m_freem(m);
11263329b236SRobert Watson 		return (EMSGSIZE);
1127430d30d8SBill Fenner 	}
1128430d30d8SBill Fenner 
11291b7f0384SBruce M Simpson 	src.sin_family = 0;
113084cc0778SGeorge V. Neville-Neil 	sin = (struct sockaddr_in *)addr;
11312435e507SGleb Smirnoff 
1132eafaa1bcSBjoern A. Zeeb 	/*
11332435e507SGleb Smirnoff 	 * udp_output() may need to temporarily bind or connect the current
11342435e507SGleb Smirnoff 	 * inpcb.  As such, we don't know up front whether we will need the
11352435e507SGleb Smirnoff 	 * pcbinfo lock or not.  Do any work to decide what is needed up
11362435e507SGleb Smirnoff 	 * front before acquiring any locks.
11372435e507SGleb Smirnoff 	 *
11382435e507SGleb Smirnoff 	 * We will need network epoch in either case, to safely lookup into
11392435e507SGleb Smirnoff 	 * pcb hash.
1140eafaa1bcSBjoern A. Zeeb 	 */
11412435e507SGleb Smirnoff 	if (sin == NULL ||
11422435e507SGleb Smirnoff 	    (inp->inp_laddr.s_addr == INADDR_ANY && inp->inp_lport == 0))
11432435e507SGleb Smirnoff 		INP_WLOCK(inp);
11442435e507SGleb Smirnoff 	else
11450cfdff24SBjoern A. Zeeb 		INP_RLOCK(inp);
11462435e507SGleb Smirnoff 	NET_EPOCH_ENTER(et);
1147f584d74bSMichael Tuexen 	tos = inp->inp_ip_tos;
1148c557ae16SIan Dowse 	if (control != NULL) {
1149c557ae16SIan Dowse 		/*
11503329b236SRobert Watson 		 * XXX: Currently, we assume all the optional information is
11513329b236SRobert Watson 		 * stored in a single mbuf.
1152c557ae16SIan Dowse 		 */
1153c557ae16SIan Dowse 		if (control->m_next) {
1154c557ae16SIan Dowse 			m_freem(control);
11552435e507SGleb Smirnoff 			error = EINVAL;
11562435e507SGleb Smirnoff 			goto release;
1157c557ae16SIan Dowse 		}
1158c557ae16SIan Dowse 		for (; control->m_len > 0;
1159c557ae16SIan Dowse 		    control->m_data += CMSG_ALIGN(cm->cmsg_len),
1160c557ae16SIan Dowse 		    control->m_len -= CMSG_ALIGN(cm->cmsg_len)) {
1161c557ae16SIan Dowse 			cm = mtod(control, struct cmsghdr *);
1162af1ee11dSRobert Watson 			if (control->m_len < sizeof(*cm) || cm->cmsg_len == 0
1163af1ee11dSRobert Watson 			    || cm->cmsg_len > control->m_len) {
1164c557ae16SIan Dowse 				error = EINVAL;
1165c557ae16SIan Dowse 				break;
1166c557ae16SIan Dowse 			}
1167a9839c4aSBjoern A. Zeeb #ifdef INET6
1168a9839c4aSBjoern A. Zeeb 			error = udp_v4mapped_pktinfo(cm, &src, inp, flags);
1169a9839c4aSBjoern A. Zeeb 			if (error != 0)
1170a9839c4aSBjoern A. Zeeb 				break;
1171a9839c4aSBjoern A. Zeeb #endif
1172c557ae16SIan Dowse 			if (cm->cmsg_level != IPPROTO_IP)
1173c557ae16SIan Dowse 				continue;
1174c557ae16SIan Dowse 
1175c557ae16SIan Dowse 			switch (cm->cmsg_type) {
1176c557ae16SIan Dowse 			case IP_SENDSRCADDR:
1177c557ae16SIan Dowse 				if (cm->cmsg_len !=
1178c557ae16SIan Dowse 				    CMSG_LEN(sizeof(struct in_addr))) {
1179c557ae16SIan Dowse 					error = EINVAL;
1180c557ae16SIan Dowse 					break;
1181c557ae16SIan Dowse 				}
1182c557ae16SIan Dowse 				bzero(&src, sizeof(src));
1183c557ae16SIan Dowse 				src.sin_family = AF_INET;
1184c557ae16SIan Dowse 				src.sin_len = sizeof(src);
1185c557ae16SIan Dowse 				src.sin_port = inp->inp_lport;
1186af1ee11dSRobert Watson 				src.sin_addr =
1187af1ee11dSRobert Watson 				    *(struct in_addr *)CMSG_DATA(cm);
1188c557ae16SIan Dowse 				break;
1189af1ee11dSRobert Watson 
1190f584d74bSMichael Tuexen 			case IP_TOS:
1191f584d74bSMichael Tuexen 				if (cm->cmsg_len != CMSG_LEN(sizeof(u_char))) {
1192f584d74bSMichael Tuexen 					error = EINVAL;
1193f584d74bSMichael Tuexen 					break;
1194f584d74bSMichael Tuexen 				}
1195f584d74bSMichael Tuexen 				tos = *(u_char *)CMSG_DATA(cm);
1196f584d74bSMichael Tuexen 				break;
1197f584d74bSMichael Tuexen 
11989d3ddf43SAdrian Chadd 			case IP_FLOWID:
11999d3ddf43SAdrian Chadd 				if (cm->cmsg_len != CMSG_LEN(sizeof(uint32_t))) {
12009d3ddf43SAdrian Chadd 					error = EINVAL;
12019d3ddf43SAdrian Chadd 					break;
12029d3ddf43SAdrian Chadd 				}
12039d3ddf43SAdrian Chadd 				flowid = *(uint32_t *) CMSG_DATA(cm);
12049d3ddf43SAdrian Chadd 				break;
12059d3ddf43SAdrian Chadd 
12069d3ddf43SAdrian Chadd 			case IP_FLOWTYPE:
12079d3ddf43SAdrian Chadd 				if (cm->cmsg_len != CMSG_LEN(sizeof(uint32_t))) {
12089d3ddf43SAdrian Chadd 					error = EINVAL;
12099d3ddf43SAdrian Chadd 					break;
12109d3ddf43SAdrian Chadd 				}
1211c2529042SHans Petter Selasky 				flowtype = *(uint32_t *) CMSG_DATA(cm);
12129d3ddf43SAdrian Chadd 				break;
12139d3ddf43SAdrian Chadd 
12149d3ddf43SAdrian Chadd #ifdef	RSS
12159d3ddf43SAdrian Chadd 			case IP_RSSBUCKETID:
12169d3ddf43SAdrian Chadd 				if (cm->cmsg_len != CMSG_LEN(sizeof(uint32_t))) {
12179d3ddf43SAdrian Chadd 					error = EINVAL;
12189d3ddf43SAdrian Chadd 					break;
12199d3ddf43SAdrian Chadd 				}
12209d3ddf43SAdrian Chadd 				/* This is just a placeholder for now */
12219d3ddf43SAdrian Chadd 				break;
12229d3ddf43SAdrian Chadd #endif	/* RSS */
1223c557ae16SIan Dowse 			default:
1224c557ae16SIan Dowse 				error = ENOPROTOOPT;
1225c557ae16SIan Dowse 				break;
1226c557ae16SIan Dowse 			}
1227c557ae16SIan Dowse 			if (error)
1228c557ae16SIan Dowse 				break;
1229c557ae16SIan Dowse 		}
1230c557ae16SIan Dowse 		m_freem(control);
1231d8acd268SMark Johnston 		control = NULL;
1232c557ae16SIan Dowse 	}
12332435e507SGleb Smirnoff 	if (error)
12342435e507SGleb Smirnoff 		goto release;
12355c32ea65SRobert Watson 
1236e06e816fSKevin Lo 	pr = inp->inp_socket->so_proto->pr_protocol;
1237a86e5c96SBjoern A. Zeeb 	pcbinfo = udp_get_inpcbinfo(pr);
12385c32ea65SRobert Watson 
12391b7f0384SBruce M Simpson 	/*
12401b7f0384SBruce M Simpson 	 * If the IP_SENDSRCADDR control message was specified, override the
12411b7f0384SBruce M Simpson 	 * source address for this datagram.  Its use is invalidated if the
12421b7f0384SBruce M Simpson 	 * address thus specified is incomplete or clobbers other inpcbs.
12431b7f0384SBruce M Simpson 	 */
124490162a4eSIan Dowse 	laddr = inp->inp_laddr;
124590162a4eSIan Dowse 	lport = inp->inp_lport;
12461b7f0384SBruce M Simpson 	if (src.sin_family == AF_INET) {
12471b7f0384SBruce M Simpson 		if ((lport == 0) ||
12481b7f0384SBruce M Simpson 		    (laddr.s_addr == INADDR_ANY &&
12491b7f0384SBruce M Simpson 		     src.sin_addr.s_addr == INADDR_ANY)) {
1250c557ae16SIan Dowse 			error = EINVAL;
1251c557ae16SIan Dowse 			goto release;
1252c557ae16SIan Dowse 		}
1253db0ac6deSCy Schubert 		INP_HASH_WLOCK(pcbinfo);
1254c557ae16SIan Dowse 		error = in_pcbbind_setup(inp, (struct sockaddr *)&src,
1255b0330ed9SPawel Jakub Dawidek 		    &laddr.s_addr, &lport, td->td_ucred);
1256db0ac6deSCy Schubert 		INP_HASH_WUNLOCK(pcbinfo);
1257c557ae16SIan Dowse 		if (error)
1258c557ae16SIan Dowse 			goto release;
1259c557ae16SIan Dowse 	}
1260c557ae16SIan Dowse 
12613144b7d3SRobert Watson 	/*
12623144b7d3SRobert Watson 	 * If a UDP socket has been connected, then a local address/port will
12633144b7d3SRobert Watson 	 * have been selected and bound.
12643144b7d3SRobert Watson 	 *
126543cc0bc1SRobert Watson 	 * If a UDP socket has not been connected to, then an explicit
12663144b7d3SRobert Watson 	 * destination address must be used, in which case a local
12673144b7d3SRobert Watson 	 * address/port may not have been selected and bound.
12683144b7d3SRobert Watson 	 */
126943cc0bc1SRobert Watson 	if (sin != NULL) {
1270c4d585aeSRobert Watson 		INP_LOCK_ASSERT(inp);
1271df8bae1dSRodney W. Grimes 		if (inp->inp_faddr.s_addr != INADDR_ANY) {
1272df8bae1dSRodney W. Grimes 			error = EISCONN;
1273df8bae1dSRodney W. Grimes 			goto release;
1274df8bae1dSRodney W. Grimes 		}
12753144b7d3SRobert Watson 
12763144b7d3SRobert Watson 		/*
12773144b7d3SRobert Watson 		 * Jail may rewrite the destination address, so let it do
12783144b7d3SRobert Watson 		 * that before we use it.
12793144b7d3SRobert Watson 		 */
1280b89e82ddSJamie Gritton 		error = prison_remote_ip4(td->td_ucred, &sin->sin_addr);
1281b89e82ddSJamie Gritton 		if (error)
1282413628a7SBjoern A. Zeeb 			goto release;
12833144b7d3SRobert Watson 
12843144b7d3SRobert Watson 		/*
128543cc0bc1SRobert Watson 		 * If a local address or port hasn't yet been selected, or if
128643cc0bc1SRobert Watson 		 * the destination address needs to be rewritten due to using
128743cc0bc1SRobert Watson 		 * a special INADDR_ constant, invoke in_pcbconnect_setup()
128843cc0bc1SRobert Watson 		 * to do the heavy lifting.  Once a port is selected, we
128943cc0bc1SRobert Watson 		 * commit the binding back to the socket; we also commit the
129043cc0bc1SRobert Watson 		 * binding of the address if in jail.
129143cc0bc1SRobert Watson 		 *
129243cc0bc1SRobert Watson 		 * If we already have a valid binding and we're not
129343cc0bc1SRobert Watson 		 * requesting a destination address rewrite, use a fast path.
12943144b7d3SRobert Watson 		 */
129543cc0bc1SRobert Watson 		if (inp->inp_laddr.s_addr == INADDR_ANY ||
129643cc0bc1SRobert Watson 		    inp->inp_lport == 0 ||
129743cc0bc1SRobert Watson 		    sin->sin_addr.s_addr == INADDR_ANY ||
129843cc0bc1SRobert Watson 		    sin->sin_addr.s_addr == INADDR_BROADCAST) {
1299db0ac6deSCy Schubert 			INP_HASH_WLOCK(pcbinfo);
130043cc0bc1SRobert Watson 			error = in_pcbconnect_setup(inp, addr, &laddr.s_addr,
130143cc0bc1SRobert Watson 			    &lport, &faddr.s_addr, &fport, NULL,
130243cc0bc1SRobert Watson 			    td->td_ucred);
1303db0ac6deSCy Schubert 			if (error) {
1304db0ac6deSCy Schubert 				INP_HASH_WUNLOCK(pcbinfo);
130590162a4eSIan Dowse 				goto release;
1306db0ac6deSCy Schubert 			}
130790162a4eSIan Dowse 
130843cc0bc1SRobert Watson 			/*
130943cc0bc1SRobert Watson 			 * XXXRW: Why not commit the port if the address is
131043cc0bc1SRobert Watson 			 * !INADDR_ANY?
131143cc0bc1SRobert Watson 			 */
131290162a4eSIan Dowse 			/* Commit the local port if newly assigned. */
131390162a4eSIan Dowse 			if (inp->inp_laddr.s_addr == INADDR_ANY &&
131490162a4eSIan Dowse 			    inp->inp_lport == 0) {
1315c4d585aeSRobert Watson 				INP_WLOCK_ASSERT(inp);
13163a1757b9SGleb Smirnoff 				/*
131743cc0bc1SRobert Watson 				 * Remember addr if jailed, to prevent
131843cc0bc1SRobert Watson 				 * rebinding.
13193a1757b9SGleb Smirnoff 				 */
13200304c731SJamie Gritton 				if (prison_flag(td->td_ucred, PR_IP4))
13213a1757b9SGleb Smirnoff 					inp->inp_laddr = laddr;
132290162a4eSIan Dowse 				inp->inp_lport = lport;
13232435e507SGleb Smirnoff 				error = in_pcbinshash(inp);
13242435e507SGleb Smirnoff 				INP_HASH_WUNLOCK(pcbinfo);
13252435e507SGleb Smirnoff 				if (error != 0) {
132690162a4eSIan Dowse 					inp->inp_lport = 0;
132790162a4eSIan Dowse 					error = EAGAIN;
1328df8bae1dSRodney W. Grimes 					goto release;
1329df8bae1dSRodney W. Grimes 				}
133090162a4eSIan Dowse 				inp->inp_flags |= INP_ANONPORT;
1331db0ac6deSCy Schubert 			} else
1332db0ac6deSCy Schubert 				INP_HASH_WUNLOCK(pcbinfo);
1333df8bae1dSRodney W. Grimes 		} else {
133443cc0bc1SRobert Watson 			faddr = sin->sin_addr;
133543cc0bc1SRobert Watson 			fport = sin->sin_port;
133643cc0bc1SRobert Watson 		}
133743cc0bc1SRobert Watson 	} else {
1338c4d585aeSRobert Watson 		INP_LOCK_ASSERT(inp);
133990162a4eSIan Dowse 		faddr = inp->inp_faddr;
134090162a4eSIan Dowse 		fport = inp->inp_fport;
134190162a4eSIan Dowse 		if (faddr.s_addr == INADDR_ANY) {
1342df8bae1dSRodney W. Grimes 			error = ENOTCONN;
1343df8bae1dSRodney W. Grimes 			goto release;
1344df8bae1dSRodney W. Grimes 		}
1345df8bae1dSRodney W. Grimes 	}
1346e6ccd709SRobert Watson 
1347df8bae1dSRodney W. Grimes 	/*
1348e6ccd709SRobert Watson 	 * Calculate data length and get a mbuf for UDP, IP, and possible
1349392e8407SRobert Watson 	 * link-layer headers.  Immediate slide the data pointer back forward
1350392e8407SRobert Watson 	 * since we won't use that space at this layer.
1351df8bae1dSRodney W. Grimes 	 */
1352eb1b1807SGleb Smirnoff 	M_PREPEND(m, sizeof(struct udpiphdr) + max_linkhdr, M_NOWAIT);
1353e6ccd709SRobert Watson 	if (m == NULL) {
1354df8bae1dSRodney W. Grimes 		error = ENOBUFS;
135549b19bfcSBruce M Simpson 		goto release;
1356df8bae1dSRodney W. Grimes 	}
1357e6ccd709SRobert Watson 	m->m_data += max_linkhdr;
1358e6ccd709SRobert Watson 	m->m_len -= max_linkhdr;
1359392e8407SRobert Watson 	m->m_pkthdr.len -= max_linkhdr;
1360df8bae1dSRodney W. Grimes 
1361df8bae1dSRodney W. Grimes 	/*
13623329b236SRobert Watson 	 * Fill in mbuf with extended UDP header and addresses and length put
13633329b236SRobert Watson 	 * into network format.
1364df8bae1dSRodney W. Grimes 	 */
1365df8bae1dSRodney W. Grimes 	ui = mtod(m, struct udpiphdr *);
1366db4f9cc7SJonathan Lemon 	bzero(ui->ui_x1, sizeof(ui->ui_x1));	/* XXX still needed? */
136734fc9072SMichael Tuexen 	ui->ui_v = IPVERSION << 4;
1368e06e816fSKevin Lo 	ui->ui_pr = pr;
136990162a4eSIan Dowse 	ui->ui_src = laddr;
137090162a4eSIan Dowse 	ui->ui_dst = faddr;
137190162a4eSIan Dowse 	ui->ui_sport = lport;
137290162a4eSIan Dowse 	ui->ui_dport = fport;
1373db4f9cc7SJonathan Lemon 	ui->ui_ulen = htons((u_short)len + sizeof(struct udphdr));
1374e06e816fSKevin Lo 	if (pr == IPPROTO_UDPLITE) {
1375e06e816fSKevin Lo 		struct udpcb *up;
1376e06e816fSKevin Lo 		uint16_t plen;
1377e06e816fSKevin Lo 
1378e06e816fSKevin Lo 		up = intoudpcb(inp);
1379e06e816fSKevin Lo 		cscov = up->u_txcslen;
1380e06e816fSKevin Lo 		plen = (u_short)len + sizeof(struct udphdr);
1381e06e816fSKevin Lo 		if (cscov >= plen)
1382e06e816fSKevin Lo 			cscov = 0;
1383e06e816fSKevin Lo 		ui->ui_len = htons(plen);
1384e06e816fSKevin Lo 		ui->ui_ulen = htons(cscov);
1385e06e816fSKevin Lo 		/*
1386e06e816fSKevin Lo 		 * For UDP-Lite, checksum coverage length of zero means
1387e06e816fSKevin Lo 		 * the entire UDPLite packet is covered by the checksum.
1388e06e816fSKevin Lo 		 */
1389e06e816fSKevin Lo 		cscov_partial = (cscov == 0) ? 0 : 1;
139034fc9072SMichael Tuexen 	}
1391df8bae1dSRodney W. Grimes 
1392b2828ad2SAndre Oppermann 	/*
1393b2828ad2SAndre Oppermann 	 * Set the Don't Fragment bit in the IP header.
1394b2828ad2SAndre Oppermann 	 */
1395b2828ad2SAndre Oppermann 	if (inp->inp_flags & INP_DONTFRAG) {
1396b2828ad2SAndre Oppermann 		struct ip *ip;
13973329b236SRobert Watson 
1398b2828ad2SAndre Oppermann 		ip = (struct ip *)&ui->ui_i;
13998f134647SGleb Smirnoff 		ip->ip_off |= htons(IP_DF);
1400b2828ad2SAndre Oppermann 	}
1401b2828ad2SAndre Oppermann 
1402b5d47ff5SJohn-Mark Gurney 	if (inp->inp_socket->so_options & SO_DONTROUTE)
1403b5d47ff5SJohn-Mark Gurney 		ipflags |= IP_ROUTETOIF;
1404b5d47ff5SJohn-Mark Gurney 	if (inp->inp_socket->so_options & SO_BROADCAST)
1405b5d47ff5SJohn-Mark Gurney 		ipflags |= IP_ALLOWBROADCAST;
14066fbfd582SAndre Oppermann 	if (inp->inp_flags & INP_ONESBCAST)
14078afa2304SBruce M Simpson 		ipflags |= IP_SENDONES;
14088afa2304SBruce M Simpson 
14091175d9d5SRobert Watson #ifdef MAC
14101175d9d5SRobert Watson 	mac_inpcb_create_mbuf(inp, m);
14111175d9d5SRobert Watson #endif
14121175d9d5SRobert Watson 
1413df8bae1dSRodney W. Grimes 	/*
1414db4f9cc7SJonathan Lemon 	 * Set up checksum and output datagram.
1415df8bae1dSRodney W. Grimes 	 */
1416e06e816fSKevin Lo 	ui->ui_sum = 0;
1417a485f139SMichael Tuexen 	if (pr == IPPROTO_UDPLITE) {
1418e06e816fSKevin Lo 		if (inp->inp_flags & INP_ONESBCAST)
1419e06e816fSKevin Lo 			faddr.s_addr = INADDR_BROADCAST;
1420a485f139SMichael Tuexen 		if (cscov_partial) {
1421e06e816fSKevin Lo 			if ((ui->ui_sum = in_cksum(m, sizeof(struct ip) + cscov)) == 0)
1422e06e816fSKevin Lo 				ui->ui_sum = 0xffff;
1423a485f139SMichael Tuexen 		} else {
1424a485f139SMichael Tuexen 			if ((ui->ui_sum = in_cksum(m, sizeof(struct udpiphdr) + len)) == 0)
1425a485f139SMichael Tuexen 				ui->ui_sum = 0xffff;
1426a485f139SMichael Tuexen 		}
1427a485f139SMichael Tuexen 	} else if (V_udp_cksum) {
14286fbfd582SAndre Oppermann 		if (inp->inp_flags & INP_ONESBCAST)
14298a538743SBruce M Simpson 			faddr.s_addr = INADDR_BROADCAST;
14308a538743SBruce M Simpson 		ui->ui_sum = in_pseudo(ui->ui_src.s_addr, faddr.s_addr,
1431e06e816fSKevin Lo 		    htons((u_short)len + sizeof(struct udphdr) + pr));
1432db4f9cc7SJonathan Lemon 		m->m_pkthdr.csum_flags = CSUM_UDP;
1433db4f9cc7SJonathan Lemon 		m->m_pkthdr.csum_data = offsetof(struct udphdr, uh_sum);
1434e06e816fSKevin Lo 	}
14358f134647SGleb Smirnoff 	((struct ip *)ui)->ip_len = htons(sizeof(struct udpiphdr) + len);
1436ca98b82cSDavid Greenman 	((struct ip *)ui)->ip_ttl = inp->inp_ip_ttl;	/* XXX */
1437f584d74bSMichael Tuexen 	((struct ip *)ui)->ip_tos = tos;		/* XXX */
1438026decb8SRobert Watson 	UDPSTAT_INC(udps_opackets);
1439cfa1ca9dSYoshinobu Inoue 
14409d3ddf43SAdrian Chadd 	/*
14419d3ddf43SAdrian Chadd 	 * Setup flowid / RSS information for outbound socket.
14429d3ddf43SAdrian Chadd 	 *
14439d3ddf43SAdrian Chadd 	 * Once the UDP code decides to set a flowid some other way,
14449d3ddf43SAdrian Chadd 	 * this allows the flowid to be overridden by userland.
14459d3ddf43SAdrian Chadd 	 */
1446c2529042SHans Petter Selasky 	if (flowtype != M_HASHTYPE_NONE) {
14479d3ddf43SAdrian Chadd 		m->m_pkthdr.flowid = flowid;
1448c2529042SHans Petter Selasky 		M_HASHTYPE_SET(m, flowtype);
1449f9a6e8d7SBjoern A. Zeeb 	}
14500c325f53SAlexander V. Chernikov #if defined(ROUTE_MPATH) || defined(RSS)
14510c325f53SAlexander V. Chernikov 	else if (CALC_FLOWID_OUTBOUND_SENDTO) {
14528ad1a83bSAdrian Chadd 		uint32_t hash_val, hash_type;
14530c325f53SAlexander V. Chernikov 
14540c325f53SAlexander V. Chernikov 		hash_val = fib4_calc_packet_hash(laddr, faddr,
14550c325f53SAlexander V. Chernikov 		    lport, fport, pr, &hash_type);
14568ad1a83bSAdrian Chadd 		m->m_pkthdr.flowid = hash_val;
14578ad1a83bSAdrian Chadd 		M_HASHTYPE_SET(m, hash_type);
14588ad1a83bSAdrian Chadd 	}
14599d3ddf43SAdrian Chadd 
14608ad1a83bSAdrian Chadd 	/*
14618ad1a83bSAdrian Chadd 	 * Don't override with the inp cached flowid value.
14628ad1a83bSAdrian Chadd 	 *
14638ad1a83bSAdrian Chadd 	 * Depending upon the kind of send being done, the inp
14648ad1a83bSAdrian Chadd 	 * flowid/flowtype values may actually not be appropriate
14658ad1a83bSAdrian Chadd 	 * for this particular socket send.
14668ad1a83bSAdrian Chadd 	 *
14678ad1a83bSAdrian Chadd 	 * We should either leave the flowid at zero (which is what is
14688ad1a83bSAdrian Chadd 	 * currently done) or set it to some software generated
14698ad1a83bSAdrian Chadd 	 * hash value based on the packet contents.
14708ad1a83bSAdrian Chadd 	 */
14719d3ddf43SAdrian Chadd 	ipflags |= IP_NODEFAULTFLOWID;
14729d3ddf43SAdrian Chadd #endif	/* RSS */
14739d3ddf43SAdrian Chadd 
14747bda9663SMichael Tuexen 	if (pr == IPPROTO_UDPLITE)
14757bda9663SMichael Tuexen 		UDPLITE_PROBE(send, NULL, inp, &ui->ui_i, inp, &ui->ui_u);
14767bda9663SMichael Tuexen 	else
147757f60867SMark Johnston 		UDP_PROBE(send, NULL, inp, &ui->ui_i, inp, &ui->ui_u);
147884cc0778SGeorge V. Neville-Neil 	error = ip_output(m, inp->inp_options,
14792435e507SGleb Smirnoff 	    INP_WLOCKED(inp) ? &inp->inp_route : NULL, ipflags,
14805d846453SSam Leffler 	    inp->inp_moptions, inp);
14812435e507SGleb Smirnoff 	INP_UNLOCK(inp);
14822435e507SGleb Smirnoff 	NET_EPOCH_EXIT(et);
1483df8bae1dSRodney W. Grimes 	return (error);
1484df8bae1dSRodney W. Grimes 
1485df8bae1dSRodney W. Grimes release:
14862435e507SGleb Smirnoff 	INP_UNLOCK(inp);
14872435e507SGleb Smirnoff 	NET_EPOCH_EXIT(et);
1488df8bae1dSRodney W. Grimes 	m_freem(m);
1489df8bae1dSRodney W. Grimes 	return (error);
1490df8bae1dSRodney W. Grimes }
1491df8bae1dSRodney W. Grimes 
1492ac45e92fSRobert Watson static void
1493d0390e05SGarrett Wollman udp_abort(struct socket *so)
1494df8bae1dSRodney W. Grimes {
1495d0390e05SGarrett Wollman 	struct inpcb *inp;
1496e06e816fSKevin Lo 	struct inpcbinfo *pcbinfo;
1497df8bae1dSRodney W. Grimes 
1498a86e5c96SBjoern A. Zeeb 	pcbinfo = udp_get_inpcbinfo(so->so_proto->pr_protocol);
1499d0390e05SGarrett Wollman 	inp = sotoinpcb(so);
150014ba8addSRobert Watson 	KASSERT(inp != NULL, ("udp_abort: inp == NULL"));
15018501a69cSRobert Watson 	INP_WLOCK(inp);
1502a152f8a3SRobert Watson 	if (inp->inp_faddr.s_addr != INADDR_ANY) {
1503e06e816fSKevin Lo 		INP_HASH_WLOCK(pcbinfo);
1504a152f8a3SRobert Watson 		in_pcbdisconnect(inp);
1505a152f8a3SRobert Watson 		inp->inp_laddr.s_addr = INADDR_ANY;
1506e06e816fSKevin Lo 		INP_HASH_WUNLOCK(pcbinfo);
1507d0390e05SGarrett Wollman 		soisdisconnected(so);
1508a152f8a3SRobert Watson 	}
15098501a69cSRobert Watson 	INP_WUNLOCK(inp);
1510df8bae1dSRodney W. Grimes }
1511df8bae1dSRodney W. Grimes 
1512d0390e05SGarrett Wollman static int
1513b40ce416SJulian Elischer udp_attach(struct socket *so, int proto, struct thread *td)
1514d0390e05SGarrett Wollman {
1515630ba2c5SMatt Macy 	static uint32_t udp_flowid;
1516d0390e05SGarrett Wollman 	struct inpcb *inp;
1517e06e816fSKevin Lo 	struct inpcbinfo *pcbinfo;
1518277afaffSRobert Watson 	int error;
1519d0390e05SGarrett Wollman 
1520a86e5c96SBjoern A. Zeeb 	pcbinfo = udp_get_inpcbinfo(so->so_proto->pr_protocol);
1521d0390e05SGarrett Wollman 	inp = sotoinpcb(so);
152214ba8addSRobert Watson 	KASSERT(inp == NULL, ("udp_attach: inp != NULL"));
1523cfa1ca9dSYoshinobu Inoue 	error = soreserve(so, udp_sendspace, udp_recvspace);
1524f24618aaSRobert Watson 	if (error)
15253329b236SRobert Watson 		return (error);
1526e06e816fSKevin Lo 	error = in_pcballoc(so, pcbinfo);
1527db0ac6deSCy Schubert 	if (error)
15283329b236SRobert Watson 		return (error);
1529cfa1ca9dSYoshinobu Inoue 
153068b5629bSRobert Watson 	inp = sotoinpcb(so);
1531cfa1ca9dSYoshinobu Inoue 	inp->inp_vflag |= INP_IPV4;
1532603724d3SBjoern A. Zeeb 	inp->inp_ip_ttl = V_ip_defttl;
1533630ba2c5SMatt Macy 	inp->inp_flowid = atomic_fetchadd_int(&udp_flowid, 1);
1534630ba2c5SMatt Macy 	inp->inp_flowtype = M_HASHTYPE_OPAQUE;
15356a9148feSBjoern A. Zeeb 
15366a9148feSBjoern A. Zeeb 	error = udp_newudpcb(inp);
15376a9148feSBjoern A. Zeeb 	if (error) {
15386a9148feSBjoern A. Zeeb 		in_pcbdetach(inp);
15396a9148feSBjoern A. Zeeb 		in_pcbfree(inp);
15406a9148feSBjoern A. Zeeb 		return (error);
15416a9148feSBjoern A. Zeeb 	}
1542266f97b5SCy Schubert 	INP_WUNLOCK(inp);
1543db0ac6deSCy Schubert 
1544c7c7ea4bSRandall Stewart 	return (0);
1545c7c7ea4bSRandall Stewart }
154679288c11SBjoern A. Zeeb #endif /* INET */
1547c7c7ea4bSRandall Stewart 
1548c7c7ea4bSRandall Stewart int
1549abb901c5SRandall Stewart udp_set_kernel_tunneling(struct socket *so, udp_tun_func_t f, udp_tun_icmp_t i, void *ctx)
1550c7c7ea4bSRandall Stewart {
1551c7c7ea4bSRandall Stewart 	struct inpcb *inp;
15526a9148feSBjoern A. Zeeb 	struct udpcb *up;
1553c7c7ea4bSRandall Stewart 
155468b5629bSRobert Watson 	KASSERT(so->so_type == SOCK_DGRAM,
155568b5629bSRobert Watson 	    ("udp_set_kernel_tunneling: !dgram"));
155668b5629bSRobert Watson 	inp = sotoinpcb(so);
155768b5629bSRobert Watson 	KASSERT(inp != NULL, ("udp_set_kernel_tunneling: inp == NULL"));
1558c7c7ea4bSRandall Stewart 	INP_WLOCK(inp);
15596a9148feSBjoern A. Zeeb 	up = intoudpcb(inp);
1560995cba5aSKristof Provost 	if ((f != NULL || i != NULL) && ((up->u_tun_func != NULL) ||
1561995cba5aSKristof Provost 	    (up->u_icmp_func != NULL))) {
1562bbb0e3d9SRandall Stewart 		INP_WUNLOCK(inp);
1563bbb0e3d9SRandall Stewart 		return (EBUSY);
1564bbb0e3d9SRandall Stewart 	}
15656a9148feSBjoern A. Zeeb 	up->u_tun_func = f;
1566abb901c5SRandall Stewart 	up->u_icmp_func = i;
156781d3ec17SBryan Venteicher 	up->u_tun_ctx = ctx;
15688501a69cSRobert Watson 	INP_WUNLOCK(inp);
15693329b236SRobert Watson 	return (0);
1570df8bae1dSRodney W. Grimes }
1571d0390e05SGarrett Wollman 
157279288c11SBjoern A. Zeeb #ifdef INET
1573d0390e05SGarrett Wollman static int
1574b40ce416SJulian Elischer udp_bind(struct socket *so, struct sockaddr *nam, struct thread *td)
1575d0390e05SGarrett Wollman {
1576d0390e05SGarrett Wollman 	struct inpcb *inp;
1577e06e816fSKevin Lo 	struct inpcbinfo *pcbinfo;
1578f96603b5SMark Johnston 	struct sockaddr_in *sinp;
1579277afaffSRobert Watson 	int error;
1580d0390e05SGarrett Wollman 
1581a86e5c96SBjoern A. Zeeb 	pcbinfo = udp_get_inpcbinfo(so->so_proto->pr_protocol);
1582d0390e05SGarrett Wollman 	inp = sotoinpcb(so);
158314ba8addSRobert Watson 	KASSERT(inp != NULL, ("udp_bind: inp == NULL"));
1584f161d294SMark Johnston 
1585f96603b5SMark Johnston 	sinp = (struct sockaddr_in *)nam;
1586f96603b5SMark Johnston 	if (nam->sa_family != AF_INET) {
1587f96603b5SMark Johnston 		/*
1588f96603b5SMark Johnston 		 * Preserve compatibility with old programs.
1589f96603b5SMark Johnston 		 */
1590f96603b5SMark Johnston 		if (nam->sa_family != AF_UNSPEC ||
15913f1f6b6eSMichael Tuexen 		    nam->sa_len < offsetof(struct sockaddr_in, sin_zero) ||
1592f96603b5SMark Johnston 		    sinp->sin_addr.s_addr != INADDR_ANY)
1593f161d294SMark Johnston 			return (EAFNOSUPPORT);
1594f96603b5SMark Johnston 		nam->sa_family = AF_INET;
1595f96603b5SMark Johnston 	}
1596f161d294SMark Johnston 	if (nam->sa_len != sizeof(struct sockaddr_in))
1597f161d294SMark Johnston 		return (EINVAL);
1598f161d294SMark Johnston 
15998501a69cSRobert Watson 	INP_WLOCK(inp);
1600e06e816fSKevin Lo 	INP_HASH_WLOCK(pcbinfo);
1601b0330ed9SPawel Jakub Dawidek 	error = in_pcbbind(inp, nam, td->td_ucred);
1602e06e816fSKevin Lo 	INP_HASH_WUNLOCK(pcbinfo);
16038501a69cSRobert Watson 	INP_WUNLOCK(inp);
16043329b236SRobert Watson 	return (error);
1605d0390e05SGarrett Wollman }
1606d0390e05SGarrett Wollman 
1607a152f8a3SRobert Watson static void
1608a152f8a3SRobert Watson udp_close(struct socket *so)
1609a152f8a3SRobert Watson {
1610a152f8a3SRobert Watson 	struct inpcb *inp;
1611e06e816fSKevin Lo 	struct inpcbinfo *pcbinfo;
1612a152f8a3SRobert Watson 
1613a86e5c96SBjoern A. Zeeb 	pcbinfo = udp_get_inpcbinfo(so->so_proto->pr_protocol);
1614a152f8a3SRobert Watson 	inp = sotoinpcb(so);
1615a152f8a3SRobert Watson 	KASSERT(inp != NULL, ("udp_close: inp == NULL"));
16168501a69cSRobert Watson 	INP_WLOCK(inp);
1617a152f8a3SRobert Watson 	if (inp->inp_faddr.s_addr != INADDR_ANY) {
1618e06e816fSKevin Lo 		INP_HASH_WLOCK(pcbinfo);
1619a152f8a3SRobert Watson 		in_pcbdisconnect(inp);
1620a152f8a3SRobert Watson 		inp->inp_laddr.s_addr = INADDR_ANY;
1621e06e816fSKevin Lo 		INP_HASH_WUNLOCK(pcbinfo);
1622a152f8a3SRobert Watson 		soisdisconnected(so);
1623a152f8a3SRobert Watson 	}
16248501a69cSRobert Watson 	INP_WUNLOCK(inp);
1625a152f8a3SRobert Watson }
1626a152f8a3SRobert Watson 
1627d0390e05SGarrett Wollman static int
1628b40ce416SJulian Elischer udp_connect(struct socket *so, struct sockaddr *nam, struct thread *td)
1629d0390e05SGarrett Wollman {
1630c1604fe4SGleb Smirnoff 	struct epoch_tracker et;
1631d0390e05SGarrett Wollman 	struct inpcb *inp;
1632e06e816fSKevin Lo 	struct inpcbinfo *pcbinfo;
163375c13541SPoul-Henning Kamp 	struct sockaddr_in *sin;
1634e06e816fSKevin Lo 	int error;
1635d0390e05SGarrett Wollman 
1636a86e5c96SBjoern A. Zeeb 	pcbinfo = udp_get_inpcbinfo(so->so_proto->pr_protocol);
1637d0390e05SGarrett Wollman 	inp = sotoinpcb(so);
163814ba8addSRobert Watson 	KASSERT(inp != NULL, ("udp_connect: inp == NULL"));
1639f161d294SMark Johnston 
1640f161d294SMark Johnston 	sin = (struct sockaddr_in *)nam;
1641f161d294SMark Johnston 	if (sin->sin_family != AF_INET)
1642f161d294SMark Johnston 		return (EAFNOSUPPORT);
1643f161d294SMark Johnston 	if (sin->sin_len != sizeof(*sin))
1644f161d294SMark Johnston 		return (EINVAL);
1645f161d294SMark Johnston 
16468501a69cSRobert Watson 	INP_WLOCK(inp);
1647f76fcf6dSJeffrey Hsu 	if (inp->inp_faddr.s_addr != INADDR_ANY) {
16488501a69cSRobert Watson 		INP_WUNLOCK(inp);
16493329b236SRobert Watson 		return (EISCONN);
1650f76fcf6dSJeffrey Hsu 	}
1651b89e82ddSJamie Gritton 	error = prison_remote_ip4(td->td_ucred, &sin->sin_addr);
1652b89e82ddSJamie Gritton 	if (error != 0) {
1653413628a7SBjoern A. Zeeb 		INP_WUNLOCK(inp);
1654b89e82ddSJamie Gritton 		return (error);
1655413628a7SBjoern A. Zeeb 	}
1656c1604fe4SGleb Smirnoff 	NET_EPOCH_ENTER(et);
1657e06e816fSKevin Lo 	INP_HASH_WLOCK(pcbinfo);
1658db0ac6deSCy Schubert 	error = in_pcbconnect(inp, nam, td->td_ucred, true);
1659e06e816fSKevin Lo 	INP_HASH_WUNLOCK(pcbinfo);
1660c1604fe4SGleb Smirnoff 	NET_EPOCH_EXIT(et);
16614cc20ab1SSeigo Tanimura 	if (error == 0)
1662df8bae1dSRodney W. Grimes 		soisconnected(so);
16638501a69cSRobert Watson 	INP_WUNLOCK(inp);
16643329b236SRobert Watson 	return (error);
1665df8bae1dSRodney W. Grimes }
1666d0390e05SGarrett Wollman 
1667bc725eafSRobert Watson static void
1668d0390e05SGarrett Wollman udp_detach(struct socket *so)
1669d0390e05SGarrett Wollman {
1670d0390e05SGarrett Wollman 	struct inpcb *inp;
16716a9148feSBjoern A. Zeeb 	struct udpcb *up;
1672d0390e05SGarrett Wollman 
1673d0390e05SGarrett Wollman 	inp = sotoinpcb(so);
167414ba8addSRobert Watson 	KASSERT(inp != NULL, ("udp_detach: inp == NULL"));
1675a152f8a3SRobert Watson 	KASSERT(inp->inp_faddr.s_addr == INADDR_ANY,
1676a152f8a3SRobert Watson 	    ("udp_detach: not disconnected"));
16778501a69cSRobert Watson 	INP_WLOCK(inp);
16786a9148feSBjoern A. Zeeb 	up = intoudpcb(inp);
16796a9148feSBjoern A. Zeeb 	KASSERT(up != NULL, ("%s: up == NULL", __func__));
16806a9148feSBjoern A. Zeeb 	inp->inp_ppcb = NULL;
1681d0390e05SGarrett Wollman 	in_pcbdetach(inp);
168214ba8addSRobert Watson 	in_pcbfree(inp);
16836a9148feSBjoern A. Zeeb 	udp_discardcb(up);
1684d0390e05SGarrett Wollman }
1685d0390e05SGarrett Wollman 
1686d0390e05SGarrett Wollman static int
1687d0390e05SGarrett Wollman udp_disconnect(struct socket *so)
1688d0390e05SGarrett Wollman {
1689d0390e05SGarrett Wollman 	struct inpcb *inp;
1690e06e816fSKevin Lo 	struct inpcbinfo *pcbinfo;
1691d0390e05SGarrett Wollman 
1692a86e5c96SBjoern A. Zeeb 	pcbinfo = udp_get_inpcbinfo(so->so_proto->pr_protocol);
1693d0390e05SGarrett Wollman 	inp = sotoinpcb(so);
169414ba8addSRobert Watson 	KASSERT(inp != NULL, ("udp_disconnect: inp == NULL"));
16958501a69cSRobert Watson 	INP_WLOCK(inp);
1696f76fcf6dSJeffrey Hsu 	if (inp->inp_faddr.s_addr == INADDR_ANY) {
16978501a69cSRobert Watson 		INP_WUNLOCK(inp);
16983329b236SRobert Watson 		return (ENOTCONN);
1699f76fcf6dSJeffrey Hsu 	}
1700e06e816fSKevin Lo 	INP_HASH_WLOCK(pcbinfo);
1701df8bae1dSRodney W. Grimes 	in_pcbdisconnect(inp);
1702df8bae1dSRodney W. Grimes 	inp->inp_laddr.s_addr = INADDR_ANY;
1703e06e816fSKevin Lo 	INP_HASH_WUNLOCK(pcbinfo);
1704d45e4f99SMaxim Konovalov 	SOCK_LOCK(so);
1705d45e4f99SMaxim Konovalov 	so->so_state &= ~SS_ISCONNECTED;		/* XXX */
1706d45e4f99SMaxim Konovalov 	SOCK_UNLOCK(so);
17078501a69cSRobert Watson 	INP_WUNLOCK(inp);
17083329b236SRobert Watson 	return (0);
1709df8bae1dSRodney W. Grimes }
1710df8bae1dSRodney W. Grimes 
1711d0390e05SGarrett Wollman static int
171257bf258eSGarrett Wollman udp_send(struct socket *so, int flags, struct mbuf *m, struct sockaddr *addr,
1713b40ce416SJulian Elischer     struct mbuf *control, struct thread *td)
1714d0390e05SGarrett Wollman {
1715d0390e05SGarrett Wollman 	struct inpcb *inp;
1716f161d294SMark Johnston 	int error;
1717d0390e05SGarrett Wollman 
1718d0390e05SGarrett Wollman 	inp = sotoinpcb(so);
171914ba8addSRobert Watson 	KASSERT(inp != NULL, ("udp_send: inp == NULL"));
1720f161d294SMark Johnston 
1721f161d294SMark Johnston 	if (addr != NULL) {
1722f161d294SMark Johnston 		error = 0;
1723f161d294SMark Johnston 		if (addr->sa_family != AF_INET)
1724f161d294SMark Johnston 			error = EAFNOSUPPORT;
1725f161d294SMark Johnston 		else if (addr->sa_len != sizeof(struct sockaddr_in))
1726f161d294SMark Johnston 			error = EINVAL;
1727f161d294SMark Johnston 		if (__predict_false(error != 0)) {
1728f161d294SMark Johnston 			m_freem(control);
1729f161d294SMark Johnston 			m_freem(m);
1730f161d294SMark Johnston 			return (error);
1731f161d294SMark Johnston 		}
1732f161d294SMark Johnston 	}
1733a9839c4aSBjoern A. Zeeb 	return (udp_output(inp, m, addr, control, td, flags));
1734d0390e05SGarrett Wollman }
173579288c11SBjoern A. Zeeb #endif /* INET */
1736d0390e05SGarrett Wollman 
173776429de4SYoshinobu Inoue int
1738d0390e05SGarrett Wollman udp_shutdown(struct socket *so)
1739d0390e05SGarrett Wollman {
1740d0390e05SGarrett Wollman 	struct inpcb *inp;
1741d0390e05SGarrett Wollman 
1742d0390e05SGarrett Wollman 	inp = sotoinpcb(so);
174314ba8addSRobert Watson 	KASSERT(inp != NULL, ("udp_shutdown: inp == NULL"));
17448501a69cSRobert Watson 	INP_WLOCK(inp);
1745d0390e05SGarrett Wollman 	socantsendmore(so);
17468501a69cSRobert Watson 	INP_WUNLOCK(inp);
17473329b236SRobert Watson 	return (0);
1748d0390e05SGarrett Wollman }
1749d0390e05SGarrett Wollman 
175079288c11SBjoern A. Zeeb #ifdef INET
1751d0390e05SGarrett Wollman struct pr_usrreqs udp_usrreqs = {
1752756d52a1SPoul-Henning Kamp 	.pru_abort =		udp_abort,
1753756d52a1SPoul-Henning Kamp 	.pru_attach =		udp_attach,
1754756d52a1SPoul-Henning Kamp 	.pru_bind =		udp_bind,
1755756d52a1SPoul-Henning Kamp 	.pru_connect =		udp_connect,
1756756d52a1SPoul-Henning Kamp 	.pru_control =		in_control,
1757756d52a1SPoul-Henning Kamp 	.pru_detach =		udp_detach,
1758756d52a1SPoul-Henning Kamp 	.pru_disconnect =	udp_disconnect,
175954d642bbSRobert Watson 	.pru_peeraddr =		in_getpeeraddr,
1760756d52a1SPoul-Henning Kamp 	.pru_send =		udp_send,
17615df3e839SRobert Watson 	.pru_soreceive =	soreceive_dgram,
176259b8854eSRobert Watson 	.pru_sosend =		sosend_dgram,
1763756d52a1SPoul-Henning Kamp 	.pru_shutdown =		udp_shutdown,
176454d642bbSRobert Watson 	.pru_sockaddr =		in_getsockaddr,
1765a152f8a3SRobert Watson 	.pru_sosetlabel =	in_pcbsosetlabel,
1766a152f8a3SRobert Watson 	.pru_close =		udp_close,
1767d0390e05SGarrett Wollman };
176879288c11SBjoern A. Zeeb #endif /* INET */
1769