xref: /freebsd/sys/netinet/udp_usrreq.c (revision fcf596178b5f2be36424ecbc1b6a3224b29c91d2)
1c398230bSWarner Losh /*-
26dfab5b1SGarrett Wollman  * Copyright (c) 1982, 1986, 1988, 1990, 1993, 1995
33329b236SRobert Watson  *	The Regents of the University of California.
43144b7d3SRobert Watson  * Copyright (c) 2008 Robert N. M. Watson
5fa046d87SRobert Watson  * Copyright (c) 2010-2011 Juniper Networks, Inc.
6e06e816fSKevin Lo  * Copyright (c) 2014 Kevin Lo
73329b236SRobert Watson  * All rights reserved.
8df8bae1dSRodney W. Grimes  *
9fa046d87SRobert Watson  * Portions of this software were developed by Robert N. M. Watson under
10fa046d87SRobert Watson  * contract to Juniper Networks, Inc.
11fa046d87SRobert Watson  *
12df8bae1dSRodney W. Grimes  * Redistribution and use in source and binary forms, with or without
13df8bae1dSRodney W. Grimes  * modification, are permitted provided that the following conditions
14df8bae1dSRodney W. Grimes  * are met:
15df8bae1dSRodney W. Grimes  * 1. Redistributions of source code must retain the above copyright
16df8bae1dSRodney W. Grimes  *    notice, this list of conditions and the following disclaimer.
17df8bae1dSRodney W. Grimes  * 2. Redistributions in binary form must reproduce the above copyright
18df8bae1dSRodney W. Grimes  *    notice, this list of conditions and the following disclaimer in the
19df8bae1dSRodney W. Grimes  *    documentation and/or other materials provided with the distribution.
20df8bae1dSRodney W. Grimes  * 4. Neither the name of the University nor the names of its contributors
21df8bae1dSRodney W. Grimes  *    may be used to endorse or promote products derived from this software
22df8bae1dSRodney W. Grimes  *    without specific prior written permission.
23df8bae1dSRodney W. Grimes  *
24df8bae1dSRodney W. Grimes  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
25df8bae1dSRodney W. Grimes  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
26df8bae1dSRodney W. Grimes  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
27df8bae1dSRodney W. Grimes  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
28df8bae1dSRodney W. Grimes  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
29df8bae1dSRodney W. Grimes  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
30df8bae1dSRodney W. Grimes  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
31df8bae1dSRodney W. Grimes  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
32df8bae1dSRodney W. Grimes  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
33df8bae1dSRodney W. Grimes  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
34df8bae1dSRodney W. Grimes  * SUCH DAMAGE.
35df8bae1dSRodney W. Grimes  *
366dfab5b1SGarrett Wollman  *	@(#)udp_usrreq.c	8.6 (Berkeley) 5/23/95
37df8bae1dSRodney W. Grimes  */
38df8bae1dSRodney W. Grimes 
394b421e2dSMike Silbersack #include <sys/cdefs.h>
404b421e2dSMike Silbersack __FBSDID("$FreeBSD$");
414b421e2dSMike Silbersack 
4279288c11SBjoern A. Zeeb #include "opt_inet.h"
43cfa1ca9dSYoshinobu Inoue #include "opt_inet6.h"
44f5514f08SRobert Watson #include "opt_ipsec.h"
459d3ddf43SAdrian Chadd #include "opt_rss.h"
46cfa1ca9dSYoshinobu Inoue 
47df8bae1dSRodney W. Grimes #include <sys/param.h>
48960ed29cSSeigo Tanimura #include <sys/domain.h>
494f590175SPaul Saab #include <sys/eventhandler.h>
50960ed29cSSeigo Tanimura #include <sys/jail.h>
51b110a8a2SGarrett Wollman #include <sys/kernel.h>
52960ed29cSSeigo Tanimura #include <sys/lock.h>
53df8bae1dSRodney W. Grimes #include <sys/malloc.h>
54df8bae1dSRodney W. Grimes #include <sys/mbuf.h>
55acd3428bSRobert Watson #include <sys/priv.h>
56490d50b6SBrian Feldman #include <sys/proc.h>
57df8bae1dSRodney W. Grimes #include <sys/protosw.h>
5857f60867SMark Johnston #include <sys/sdt.h>
59960ed29cSSeigo Tanimura #include <sys/signalvar.h>
60df8bae1dSRodney W. Grimes #include <sys/socket.h>
61df8bae1dSRodney W. Grimes #include <sys/socketvar.h>
62960ed29cSSeigo Tanimura #include <sys/sx.h>
63b5e8ce9fSBruce Evans #include <sys/sysctl.h>
64816a3d83SPoul-Henning Kamp #include <sys/syslog.h>
65f5514f08SRobert Watson #include <sys/systm.h>
668781d8e9SBruce Evans 
6769c2d429SJeff Roberson #include <vm/uma.h>
68df8bae1dSRodney W. Grimes 
69df8bae1dSRodney W. Grimes #include <net/if.h>
7076039bc8SGleb Smirnoff #include <net/if_var.h>
71df8bae1dSRodney W. Grimes #include <net/route.h>
72b2bdc62aSAdrian Chadd #include <net/rss_config.h>
73df8bae1dSRodney W. Grimes 
74df8bae1dSRodney W. Grimes #include <netinet/in.h>
7557f60867SMark Johnston #include <netinet/in_kdtrace.h>
76960ed29cSSeigo Tanimura #include <netinet/in_pcb.h>
77f5514f08SRobert Watson #include <netinet/in_systm.h>
78960ed29cSSeigo Tanimura #include <netinet/in_var.h>
79df8bae1dSRodney W. Grimes #include <netinet/ip.h>
80cfa1ca9dSYoshinobu Inoue #ifdef INET6
81cfa1ca9dSYoshinobu Inoue #include <netinet/ip6.h>
82cfa1ca9dSYoshinobu Inoue #endif
83960ed29cSSeigo Tanimura #include <netinet/ip_icmp.h>
84960ed29cSSeigo Tanimura #include <netinet/icmp_var.h>
85df8bae1dSRodney W. Grimes #include <netinet/ip_var.h>
86ef39adf0SAndre Oppermann #include <netinet/ip_options.h>
87cfa1ca9dSYoshinobu Inoue #ifdef INET6
88cfa1ca9dSYoshinobu Inoue #include <netinet6/ip6_var.h>
89cfa1ca9dSYoshinobu Inoue #endif
90df8bae1dSRodney W. Grimes #include <netinet/udp.h>
91df8bae1dSRodney W. Grimes #include <netinet/udp_var.h>
92e06e816fSKevin Lo #include <netinet/udplite.h>
938ad1a83bSAdrian Chadd #include <netinet/in_rss.h>
94df8bae1dSRodney W. Grimes 
95*fcf59617SAndrey V. Elsukov #include <netipsec/ipsec_support.h>
96b9234fafSSam Leffler 
97db4f9cc7SJonathan Lemon #include <machine/in_cksum.h>
98db4f9cc7SJonathan Lemon 
99aed55708SRobert Watson #include <security/mac/mac_framework.h>
100aed55708SRobert Watson 
101df8bae1dSRodney W. Grimes /*
102e06e816fSKevin Lo  * UDP and UDP-Lite protocols implementation.
103df8bae1dSRodney W. Grimes  * Per RFC 768, August, 1980.
104e06e816fSKevin Lo  * Per RFC 3828, July, 2004.
105df8bae1dSRodney W. Grimes  */
10674eb3236SWarner Losh 
10774eb3236SWarner Losh /*
1083329b236SRobert Watson  * BSD 4.2 defaulted the udp checksum to be off.  Turning off udp checksums
1093329b236SRobert Watson  * removes the only data integrity mechanism for packets and malformed
110f5514f08SRobert Watson  * packets that would otherwise be discarded due to bad checksums, and may
111f5514f08SRobert Watson  * cause problems (especially for NFS data blocks).
11274eb3236SWarner Losh  */
11340b676beSBjoern A. Zeeb VNET_DEFINE(int, udp_cksum) = 1;
1146df8a710SGleb Smirnoff SYSCTL_INT(_net_inet_udp, UDPCTL_CHECKSUM, checksum, CTLFLAG_VNET | CTLFLAG_RW,
11540b676beSBjoern A. Zeeb     &VNET_NAME(udp_cksum), 0, "compute udp checksum");
116df8bae1dSRodney W. Grimes 
117afdb4274SRobert Watson int	udp_log_in_vain = 0;
118816a3d83SPoul-Henning Kamp SYSCTL_INT(_net_inet_udp, OID_AUTO, log_in_vain, CTLFLAG_RW,
119afdb4274SRobert Watson     &udp_log_in_vain, 0, "Log all incoming UDP packets");
120816a3d83SPoul-Henning Kamp 
12182cea7e6SBjoern A. Zeeb VNET_DEFINE(int, udp_blackhole) = 0;
1226df8a710SGleb Smirnoff SYSCTL_INT(_net_inet_udp, OID_AUTO, blackhole, CTLFLAG_VNET | CTLFLAG_RW,
123eddfbb76SRobert Watson     &VNET_NAME(udp_blackhole), 0,
1243329b236SRobert Watson     "Do not send port unreachables for refused connects");
12516f7f31fSGeoff Rehmet 
12623424a20SRyan Stone static VNET_DEFINE(int, udp_require_l2_bcast) = 0;
1279da85a91SMarko Zec #define	V_udp_require_l2_bcast		VNET(udp_require_l2_bcast)
12841029db1SRyan Stone SYSCTL_INT(_net_inet_udp, OID_AUTO, require_l2_bcast, CTLFLAG_VNET | CTLFLAG_RW,
12941029db1SRyan Stone     &VNET_NAME(udp_require_l2_bcast), 0,
13041029db1SRyan Stone     "Only treat packets sent to an L2 broadcast address as broadcast packets");
13141029db1SRyan Stone 
13243bbb6aaSRobert Watson u_long	udp_sendspace = 9216;		/* really max datagram size */
13343bbb6aaSRobert Watson SYSCTL_ULONG(_net_inet_udp, UDPCTL_MAXDGRAM, maxdgram, CTLFLAG_RW,
13443bbb6aaSRobert Watson     &udp_sendspace, 0, "Maximum outgoing UDP datagram size");
13543bbb6aaSRobert Watson 
13643bbb6aaSRobert Watson u_long	udp_recvspace = 40 * (1024 +
13743bbb6aaSRobert Watson #ifdef INET6
13843bbb6aaSRobert Watson 				      sizeof(struct sockaddr_in6)
13943bbb6aaSRobert Watson #else
14043bbb6aaSRobert Watson 				      sizeof(struct sockaddr_in)
14143bbb6aaSRobert Watson #endif
142e62b9bcaSSergey Kandaurov 				      );	/* 40 1K datagrams */
14343bbb6aaSRobert Watson 
14443bbb6aaSRobert Watson SYSCTL_ULONG(_net_inet_udp, UDPCTL_RECVSPACE, recvspace, CTLFLAG_RW,
14543bbb6aaSRobert Watson     &udp_recvspace, 0, "Maximum space for incoming UDP datagrams");
14643bbb6aaSRobert Watson 
147eddfbb76SRobert Watson VNET_DEFINE(struct inpcbhead, udb);		/* from udp_var.h */
148eddfbb76SRobert Watson VNET_DEFINE(struct inpcbinfo, udbinfo);
149e06e816fSKevin Lo VNET_DEFINE(struct inpcbhead, ulitecb);
150e06e816fSKevin Lo VNET_DEFINE(struct inpcbinfo, ulitecbinfo);
1513e288e62SDimitry Andric static VNET_DEFINE(uma_zone_t, udpcb_zone);
1521e77c105SRobert Watson #define	V_udpcb_zone			VNET(udpcb_zone)
15315bd2b43SDavid Greenman 
15415bd2b43SDavid Greenman #ifndef UDBHASHSIZE
155e2ed8f35SAlexander Motin #define	UDBHASHSIZE	128
15615bd2b43SDavid Greenman #endif
15715bd2b43SDavid Greenman 
1585b7cb97cSAndrey V. Elsukov VNET_PCPUSTAT_DEFINE(struct udpstat, udpstat);		/* from udp_var.h */
1595b7cb97cSAndrey V. Elsukov VNET_PCPUSTAT_SYSINIT(udpstat);
1605b7cb97cSAndrey V. Elsukov SYSCTL_VNET_PCPUSTAT(_net_inet_udp, UDPCTL_STATS, stats, struct udpstat,
1615b7cb97cSAndrey V. Elsukov     udpstat, "UDP statistics (struct udpstat, netinet/udp_var.h)");
162f2ea20e6SGarrett Wollman 
1635b7cb97cSAndrey V. Elsukov #ifdef VIMAGE
1645b7cb97cSAndrey V. Elsukov VNET_PCPUSTAT_SYSUNINIT(udpstat);
1655b7cb97cSAndrey V. Elsukov #endif /* VIMAGE */
16679288c11SBjoern A. Zeeb #ifdef INET
167bc725eafSRobert Watson static void	udp_detach(struct socket *so);
1684d77a549SAlfred Perlstein static int	udp_output(struct inpcb *, struct mbuf *, struct sockaddr *,
1694d77a549SAlfred Perlstein 		    struct mbuf *, struct thread *);
17079288c11SBjoern A. Zeeb #endif
17179288c11SBjoern A. Zeeb 
1724f590175SPaul Saab static void
1734f590175SPaul Saab udp_zone_change(void *tag)
1744f590175SPaul Saab {
1754f590175SPaul Saab 
176603724d3SBjoern A. Zeeb 	uma_zone_set_max(V_udbinfo.ipi_zone, maxsockets);
1776a9148feSBjoern A. Zeeb 	uma_zone_set_max(V_udpcb_zone, maxsockets);
1784f590175SPaul Saab }
1794f590175SPaul Saab 
180d915b280SStephan Uphoff static int
181d915b280SStephan Uphoff udp_inpcb_init(void *mem, int size, int flags)
182d915b280SStephan Uphoff {
183af1ee11dSRobert Watson 	struct inpcb *inp;
18408651e1fSJohn Baldwin 
185af1ee11dSRobert Watson 	inp = mem;
186d915b280SStephan Uphoff 	INP_LOCK_INIT(inp, "inp", "udpinp");
187d915b280SStephan Uphoff 	return (0);
188d915b280SStephan Uphoff }
189d915b280SStephan Uphoff 
190e06e816fSKevin Lo static int
191e06e816fSKevin Lo udplite_inpcb_init(void *mem, int size, int flags)
192e06e816fSKevin Lo {
193e06e816fSKevin Lo 	struct inpcb *inp;
194e06e816fSKevin Lo 
195e06e816fSKevin Lo 	inp = mem;
196e06e816fSKevin Lo 	INP_LOCK_INIT(inp, "inp", "udpliteinp");
197e06e816fSKevin Lo 	return (0);
198e06e816fSKevin Lo }
199e06e816fSKevin Lo 
200df8bae1dSRodney W. Grimes void
201af1ee11dSRobert Watson udp_init(void)
202df8bae1dSRodney W. Grimes {
203af1ee11dSRobert Watson 
2048ad1a83bSAdrian Chadd 	/*
2058ad1a83bSAdrian Chadd 	 * For now default to 2-tuple UDP hashing - until the fragment
2068ad1a83bSAdrian Chadd 	 * reassembly code can also update the flowid.
2078ad1a83bSAdrian Chadd 	 *
2088ad1a83bSAdrian Chadd 	 * Once we can calculate the flowid that way and re-establish
2098ad1a83bSAdrian Chadd 	 * a 4-tuple, flip this to 4-tuple.
2108ad1a83bSAdrian Chadd 	 */
2119bcd427bSRobert Watson 	in_pcbinfo_init(&V_udbinfo, "udp", &V_udb, UDBHASHSIZE, UDBHASHSIZE,
212a8da5dd6SCraig Rodrigues 	    "udp_inpcb", udp_inpcb_init, NULL, 0,
21352cd27cbSRobert Watson 	    IPI_HASHFIELDS_2TUPLE);
2146a9148feSBjoern A. Zeeb 	V_udpcb_zone = uma_zcreate("udpcb", sizeof(struct udpcb),
215a8da5dd6SCraig Rodrigues 	    NULL, NULL, NULL, NULL, UMA_ALIGN_PTR, 0);
2166a9148feSBjoern A. Zeeb 	uma_zone_set_max(V_udpcb_zone, maxsockets);
2176acd596eSPawel Jakub Dawidek 	uma_zone_set_warning(V_udpcb_zone, "kern.ipc.maxsockets limit reached");
2184f590175SPaul Saab 	EVENTHANDLER_REGISTER(maxsockets_change, udp_zone_change, NULL,
2194f590175SPaul Saab 	    EVENTHANDLER_PRI_ANY);
220df8bae1dSRodney W. Grimes }
221df8bae1dSRodney W. Grimes 
222e06e816fSKevin Lo void
223e06e816fSKevin Lo udplite_init(void)
224e06e816fSKevin Lo {
225e06e816fSKevin Lo 
226e06e816fSKevin Lo 	in_pcbinfo_init(&V_ulitecbinfo, "udplite", &V_ulitecb, UDBHASHSIZE,
227e06e816fSKevin Lo 	    UDBHASHSIZE, "udplite_inpcb", udplite_inpcb_init, NULL,
228a8da5dd6SCraig Rodrigues 	    0, IPI_HASHFIELDS_2TUPLE);
229e06e816fSKevin Lo }
230e06e816fSKevin Lo 
231315e3e38SRobert Watson /*
232315e3e38SRobert Watson  * Kernel module interface for updating udpstat.  The argument is an index
233315e3e38SRobert Watson  * into udpstat treated as an array of u_long.  While this encodes the
234315e3e38SRobert Watson  * general layout of udpstat into the caller, it doesn't encode its location,
235315e3e38SRobert Watson  * so that future changes to add, for example, per-CPU stats support won't
236315e3e38SRobert Watson  * cause binary compatibility problems for kernel modules.
237315e3e38SRobert Watson  */
238315e3e38SRobert Watson void
239315e3e38SRobert Watson kmod_udpstat_inc(int statnum)
240315e3e38SRobert Watson {
241315e3e38SRobert Watson 
2425b7cb97cSAndrey V. Elsukov 	counter_u64_add(VNET(udpstat)[statnum], 1);
243315e3e38SRobert Watson }
244315e3e38SRobert Watson 
2456a9148feSBjoern A. Zeeb int
2466a9148feSBjoern A. Zeeb udp_newudpcb(struct inpcb *inp)
2476a9148feSBjoern A. Zeeb {
2486a9148feSBjoern A. Zeeb 	struct udpcb *up;
2496a9148feSBjoern A. Zeeb 
2506a9148feSBjoern A. Zeeb 	up = uma_zalloc(V_udpcb_zone, M_NOWAIT | M_ZERO);
2516a9148feSBjoern A. Zeeb 	if (up == NULL)
2526a9148feSBjoern A. Zeeb 		return (ENOBUFS);
2536a9148feSBjoern A. Zeeb 	inp->inp_ppcb = up;
2546a9148feSBjoern A. Zeeb 	return (0);
2556a9148feSBjoern A. Zeeb }
2566a9148feSBjoern A. Zeeb 
2576a9148feSBjoern A. Zeeb void
2586a9148feSBjoern A. Zeeb udp_discardcb(struct udpcb *up)
2596a9148feSBjoern A. Zeeb {
2606a9148feSBjoern A. Zeeb 
2616a9148feSBjoern A. Zeeb 	uma_zfree(V_udpcb_zone, up);
2626a9148feSBjoern A. Zeeb }
2636a9148feSBjoern A. Zeeb 
264bc29160dSMarko Zec #ifdef VIMAGE
2653f58662dSBjoern A. Zeeb static void
2663f58662dSBjoern A. Zeeb udp_destroy(void *unused __unused)
267bc29160dSMarko Zec {
268bc29160dSMarko Zec 
2699bcd427bSRobert Watson 	in_pcbinfo_destroy(&V_udbinfo);
270391dab1cSBjoern A. Zeeb 	uma_zdestroy(V_udpcb_zone);
271bc29160dSMarko Zec }
2723f58662dSBjoern A. Zeeb VNET_SYSUNINIT(udp, SI_SUB_PROTO_DOMAIN, SI_ORDER_FOURTH, udp_destroy, NULL);
273e06e816fSKevin Lo 
2743f58662dSBjoern A. Zeeb static void
2753f58662dSBjoern A. Zeeb udplite_destroy(void *unused __unused)
276e06e816fSKevin Lo {
277e06e816fSKevin Lo 
278e06e816fSKevin Lo 	in_pcbinfo_destroy(&V_ulitecbinfo);
279e06e816fSKevin Lo }
2803f58662dSBjoern A. Zeeb VNET_SYSUNINIT(udplite, SI_SUB_PROTO_DOMAIN, SI_ORDER_FOURTH, udplite_destroy,
2813f58662dSBjoern A. Zeeb     NULL);
282bc29160dSMarko Zec #endif
283bc29160dSMarko Zec 
28479288c11SBjoern A. Zeeb #ifdef INET
28543bbb6aaSRobert Watson /*
28643bbb6aaSRobert Watson  * Subroutine of udp_input(), which appends the provided mbuf chain to the
28743bbb6aaSRobert Watson  * passed pcb/socket.  The caller must provide a sockaddr_in via udp_in that
28843bbb6aaSRobert Watson  * contains the source address.  If the socket ends up being an IPv6 socket,
28943bbb6aaSRobert Watson  * udp_append() will convert to a sockaddr_in6 before passing the address
29043bbb6aaSRobert Watson  * into the socket code.
291c0d1be08SRandall Stewart  *
292c0d1be08SRandall Stewart  * In the normal case udp_append() will return 0, indicating that you
293c0d1be08SRandall Stewart  * must unlock the inp. However if a tunneling protocol is in place we increment
294c0d1be08SRandall Stewart  * the inpcb refcnt and unlock the inp, on return from the tunneling protocol we
295c0d1be08SRandall Stewart  * then decrement the reference count. If the inp_rele returns 1, indicating the
296c0d1be08SRandall Stewart  * inp is gone, we return that to the caller to tell them *not* to unlock
297c0d1be08SRandall Stewart  * the inp. In the case of multi-cast this will cause the distribution
298c0d1be08SRandall Stewart  * to stop (though most tunneling protocols known currently do *not* use
299c0d1be08SRandall Stewart  * multicast).
30043bbb6aaSRobert Watson  */
301c0d1be08SRandall Stewart static int
30243bbb6aaSRobert Watson udp_append(struct inpcb *inp, struct ip *ip, struct mbuf *n, int off,
30343bbb6aaSRobert Watson     struct sockaddr_in *udp_in)
30443bbb6aaSRobert Watson {
30543bbb6aaSRobert Watson 	struct sockaddr *append_sa;
30643bbb6aaSRobert Watson 	struct socket *so;
30799d628d5SPedro F. Giffuni 	struct mbuf *opts = NULL;
30843bbb6aaSRobert Watson #ifdef INET6
30943bbb6aaSRobert Watson 	struct sockaddr_in6 udp_in6;
31043bbb6aaSRobert Watson #endif
3117b495c44SVANHULLEBUS Yvan 	struct udpcb *up;
31243bbb6aaSRobert Watson 
313fa046d87SRobert Watson 	INP_LOCK_ASSERT(inp);
31443bbb6aaSRobert Watson 
31579bb84fbSEdward Tomasz Napierala 	/*
31679bb84fbSEdward Tomasz Napierala 	 * Engage the tunneling protocol.
31779bb84fbSEdward Tomasz Napierala 	 */
31879bb84fbSEdward Tomasz Napierala 	up = intoudpcb(inp);
31979bb84fbSEdward Tomasz Napierala 	if (up->u_tun_func != NULL) {
320c0d1be08SRandall Stewart 		in_pcbref(inp);
321c0d1be08SRandall Stewart 		INP_RUNLOCK(inp);
32281d3ec17SBryan Venteicher 		(*up->u_tun_func)(n, off, inp, (struct sockaddr *)udp_in,
32381d3ec17SBryan Venteicher 		    up->u_tun_ctx);
324c0d1be08SRandall Stewart 		INP_RLOCK(inp);
325c0d1be08SRandall Stewart 		return (in_pcbrele_rlocked(inp));
32679bb84fbSEdward Tomasz Napierala 	}
32779bb84fbSEdward Tomasz Napierala 
32879bb84fbSEdward Tomasz Napierala 	off += sizeof(struct udphdr);
32979bb84fbSEdward Tomasz Napierala 
330*fcf59617SAndrey V. Elsukov #if defined(IPSEC) || defined(IPSEC_SUPPORT)
33143bbb6aaSRobert Watson 	/* Check AH/ESP integrity. */
332*fcf59617SAndrey V. Elsukov 	if (IPSEC_ENABLED(ipv4) &&
333*fcf59617SAndrey V. Elsukov 	    IPSEC_CHECK_POLICY(ipv4, n, inp) != 0) {
33443bbb6aaSRobert Watson 		m_freem(n);
335c0d1be08SRandall Stewart 		return (0);
33643bbb6aaSRobert Watson 	}
337*fcf59617SAndrey V. Elsukov 	if (up->u_flags & UF_ESPINUDP) {/* IPSec UDP encaps. */
338*fcf59617SAndrey V. Elsukov 		if (IPSEC_ENABLED(ipv4) &&
339*fcf59617SAndrey V. Elsukov 		    UDPENCAP_INPUT(n, off, AF_INET) != 0)
340*fcf59617SAndrey V. Elsukov 			return (0);	/* Consumed. */
3417b495c44SVANHULLEBUS Yvan 	}
34243bbb6aaSRobert Watson #endif /* IPSEC */
34343bbb6aaSRobert Watson #ifdef MAC
34430d239bcSRobert Watson 	if (mac_inpcb_check_deliver(inp, n) != 0) {
34543bbb6aaSRobert Watson 		m_freem(n);
346c0d1be08SRandall Stewart 		return (0);
34743bbb6aaSRobert Watson 	}
34879288c11SBjoern A. Zeeb #endif /* MAC */
34943bbb6aaSRobert Watson 	if (inp->inp_flags & INP_CONTROLOPTS ||
35043bbb6aaSRobert Watson 	    inp->inp_socket->so_options & (SO_TIMESTAMP | SO_BINTIME)) {
35143bbb6aaSRobert Watson #ifdef INET6
3529a38ba81SBjoern A. Zeeb 		if (inp->inp_vflag & INP_IPV6)
35348d48eb9SBjoern A. Zeeb 			(void)ip6_savecontrol_v4(inp, n, &opts, NULL);
3549a38ba81SBjoern A. Zeeb 		else
35579288c11SBjoern A. Zeeb #endif /* INET6 */
35643bbb6aaSRobert Watson 			ip_savecontrol(inp, &opts, ip, n);
35743bbb6aaSRobert Watson 	}
35843bbb6aaSRobert Watson #ifdef INET6
35943bbb6aaSRobert Watson 	if (inp->inp_vflag & INP_IPV6) {
36043bbb6aaSRobert Watson 		bzero(&udp_in6, sizeof(udp_in6));
36143bbb6aaSRobert Watson 		udp_in6.sin6_len = sizeof(udp_in6);
36243bbb6aaSRobert Watson 		udp_in6.sin6_family = AF_INET6;
36343bbb6aaSRobert Watson 		in6_sin_2_v4mapsin6(udp_in, &udp_in6);
36443bbb6aaSRobert Watson 		append_sa = (struct sockaddr *)&udp_in6;
36543bbb6aaSRobert Watson 	} else
36679288c11SBjoern A. Zeeb #endif /* INET6 */
36743bbb6aaSRobert Watson 		append_sa = (struct sockaddr *)udp_in;
36843bbb6aaSRobert Watson 	m_adj(n, off);
36943bbb6aaSRobert Watson 
37043bbb6aaSRobert Watson 	so = inp->inp_socket;
37143bbb6aaSRobert Watson 	SOCKBUF_LOCK(&so->so_rcv);
37243bbb6aaSRobert Watson 	if (sbappendaddr_locked(&so->so_rcv, append_sa, n, opts) == 0) {
37343bbb6aaSRobert Watson 		SOCKBUF_UNLOCK(&so->so_rcv);
37443bbb6aaSRobert Watson 		m_freem(n);
37543bbb6aaSRobert Watson 		if (opts)
37643bbb6aaSRobert Watson 			m_freem(opts);
377026decb8SRobert Watson 		UDPSTAT_INC(udps_fullsock);
37843bbb6aaSRobert Watson 	} else
37943bbb6aaSRobert Watson 		sorwakeup_locked(so);
380c0d1be08SRandall Stewart 	return (0);
38143bbb6aaSRobert Watson }
38243bbb6aaSRobert Watson 
3838f5a8818SKevin Lo int
3848f5a8818SKevin Lo udp_input(struct mbuf **mp, int *offp, int proto)
385df8bae1dSRodney W. Grimes {
3863329b236SRobert Watson 	struct ip *ip;
3873329b236SRobert Watson 	struct udphdr *uh;
38871498f30SBruce M Simpson 	struct ifnet *ifp;
3893329b236SRobert Watson 	struct inpcb *inp;
3908f134647SGleb Smirnoff 	uint16_t len, ip_len;
391e06e816fSKevin Lo 	struct inpcbinfo *pcbinfo;
392df8bae1dSRodney W. Grimes 	struct ip save_ip;
393d4b509bdSRobert Watson 	struct sockaddr_in udp_in;
3948f5a8818SKevin Lo 	struct mbuf *m;
3950b4ae859SGleb Smirnoff 	struct m_tag *fwd_tag;
3968f5a8818SKevin Lo 	int cscov_partial, iphlen;
397df8bae1dSRodney W. Grimes 
3988f5a8818SKevin Lo 	m = *mp;
3998f5a8818SKevin Lo 	iphlen = *offp;
40071498f30SBruce M Simpson 	ifp = m->m_pkthdr.rcvif;
4018f5a8818SKevin Lo 	*mp = NULL;
402026decb8SRobert Watson 	UDPSTAT_INC(udps_ipackets);
403df8bae1dSRodney W. Grimes 
404df8bae1dSRodney W. Grimes 	/*
4053329b236SRobert Watson 	 * Strip IP options, if any; should skip this, make available to
4063329b236SRobert Watson 	 * user, and use on returned packets, but we don't yet have a way to
4073329b236SRobert Watson 	 * check the checksum with options still present.
408df8bae1dSRodney W. Grimes 	 */
409df8bae1dSRodney W. Grimes 	if (iphlen > sizeof (struct ip)) {
410105bd211SGleb Smirnoff 		ip_stripoptions(m);
411df8bae1dSRodney W. Grimes 		iphlen = sizeof(struct ip);
412df8bae1dSRodney W. Grimes 	}
413df8bae1dSRodney W. Grimes 
414df8bae1dSRodney W. Grimes 	/*
415df8bae1dSRodney W. Grimes 	 * Get IP and UDP header together in first mbuf.
416df8bae1dSRodney W. Grimes 	 */
417df8bae1dSRodney W. Grimes 	ip = mtod(m, struct ip *);
418df8bae1dSRodney W. Grimes 	if (m->m_len < iphlen + sizeof(struct udphdr)) {
419d1b18731SKevin Lo 		if ((m = m_pullup(m, iphlen + sizeof(struct udphdr))) == NULL) {
420026decb8SRobert Watson 			UDPSTAT_INC(udps_hdrops);
4218f5a8818SKevin Lo 			return (IPPROTO_DONE);
422df8bae1dSRodney W. Grimes 		}
423df8bae1dSRodney W. Grimes 		ip = mtod(m, struct ip *);
424df8bae1dSRodney W. Grimes 	}
425df8bae1dSRodney W. Grimes 	uh = (struct udphdr *)((caddr_t)ip + iphlen);
4268f5a8818SKevin Lo 	cscov_partial = (proto == IPPROTO_UDPLITE) ? 1 : 0;
427df8bae1dSRodney W. Grimes 
4283329b236SRobert Watson 	/*
4293329b236SRobert Watson 	 * Destination port of 0 is illegal, based on RFC768.
4303329b236SRobert Watson 	 */
431686cdd19SJun-ichiro itojun Hagino 	if (uh->uh_dport == 0)
432f76fcf6dSJeffrey Hsu 		goto badunlocked;
433686cdd19SJun-ichiro itojun Hagino 
434df8bae1dSRodney W. Grimes 	/*
4353329b236SRobert Watson 	 * Construct sockaddr format source address.  Stuff source address
4363329b236SRobert Watson 	 * and datagram in user buffer.
437b9234fafSSam Leffler 	 */
438d4b509bdSRobert Watson 	bzero(&udp_in, sizeof(udp_in));
439d4b509bdSRobert Watson 	udp_in.sin_len = sizeof(udp_in);
440d4b509bdSRobert Watson 	udp_in.sin_family = AF_INET;
441b9234fafSSam Leffler 	udp_in.sin_port = uh->uh_sport;
442b9234fafSSam Leffler 	udp_in.sin_addr = ip->ip_src;
443b9234fafSSam Leffler 
444b9234fafSSam Leffler 	/*
445af1ee11dSRobert Watson 	 * Make mbuf data length reflect UDP length.  If not enough data to
446af1ee11dSRobert Watson 	 * reflect UDP length, drop.
447df8bae1dSRodney W. Grimes 	 */
448df8bae1dSRodney W. Grimes 	len = ntohs((u_short)uh->uh_ulen);
4498ad458a4SGleb Smirnoff 	ip_len = ntohs(ip->ip_len) - iphlen;
4500f4a0366SMichael Tuexen 	if (proto == IPPROTO_UDPLITE && (len == 0 || len == ip_len)) {
451e06e816fSKevin Lo 		/* Zero means checksum over the complete packet. */
4520f4a0366SMichael Tuexen 		if (len == 0)
453e06e816fSKevin Lo 			len = ip_len;
454e06e816fSKevin Lo 		cscov_partial = 0;
455e06e816fSKevin Lo 	}
4568f134647SGleb Smirnoff 	if (ip_len != len) {
4578f134647SGleb Smirnoff 		if (len > ip_len || len < sizeof(struct udphdr)) {
458026decb8SRobert Watson 			UDPSTAT_INC(udps_badlen);
459f76fcf6dSJeffrey Hsu 			goto badunlocked;
460df8bae1dSRodney W. Grimes 		}
4618f5a8818SKevin Lo 		if (proto == IPPROTO_UDP)
4628f134647SGleb Smirnoff 			m_adj(m, len - ip_len);
463df8bae1dSRodney W. Grimes 	}
4643329b236SRobert Watson 
465df8bae1dSRodney W. Grimes 	/*
4663329b236SRobert Watson 	 * Save a copy of the IP header in case we want restore it for
4673329b236SRobert Watson 	 * sending an ICMP error message in response.
468df8bae1dSRodney W. Grimes 	 */
469603724d3SBjoern A. Zeeb 	if (!V_udp_blackhole)
470df8bae1dSRodney W. Grimes 		save_ip = *ip;
471cce418d3SMatt Jacob 	else
472cce418d3SMatt Jacob 		memset(&save_ip, 0, sizeof(save_ip));
473df8bae1dSRodney W. Grimes 
474df8bae1dSRodney W. Grimes 	/*
475df8bae1dSRodney W. Grimes 	 * Checksum extended UDP header and data.
476df8bae1dSRodney W. Grimes 	 */
4776dfab5b1SGarrett Wollman 	if (uh->uh_sum) {
47839629c92SDavid Malone 		u_short uh_sum;
47939629c92SDavid Malone 
480e06e816fSKevin Lo 		if ((m->m_pkthdr.csum_flags & CSUM_DATA_VALID) &&
481e06e816fSKevin Lo 		    !cscov_partial) {
482db4f9cc7SJonathan Lemon 			if (m->m_pkthdr.csum_flags & CSUM_PSEUDO_HDR)
48339629c92SDavid Malone 				uh_sum = m->m_pkthdr.csum_data;
484db4f9cc7SJonathan Lemon 			else
48539629c92SDavid Malone 				uh_sum = in_pseudo(ip->ip_src.s_addr,
486506f4949SRuslan Ermilov 				    ip->ip_dst.s_addr, htonl((u_short)len +
4878f5a8818SKevin Lo 				    m->m_pkthdr.csum_data + proto));
48839629c92SDavid Malone 			uh_sum ^= 0xffff;
489db4f9cc7SJonathan Lemon 		} else {
490cb342100SHajimu UMEMOTO 			char b[9];
491af1ee11dSRobert Watson 
492cb342100SHajimu UMEMOTO 			bcopy(((struct ipovly *)ip)->ih_x1, b, 9);
4936effc713SDoug Rabson 			bzero(((struct ipovly *)ip)->ih_x1, 9);
4948f5a8818SKevin Lo 			((struct ipovly *)ip)->ih_len = (proto == IPPROTO_UDP) ?
495e06e816fSKevin Lo 			    uh->uh_ulen : htons(ip_len);
49639629c92SDavid Malone 			uh_sum = in_cksum(m, len + sizeof (struct ip));
497cb342100SHajimu UMEMOTO 			bcopy(b, ((struct ipovly *)ip)->ih_x1, 9);
498db4f9cc7SJonathan Lemon 		}
49939629c92SDavid Malone 		if (uh_sum) {
500026decb8SRobert Watson 			UDPSTAT_INC(udps_badsum);
501df8bae1dSRodney W. Grimes 			m_freem(m);
5028f5a8818SKevin Lo 			return (IPPROTO_DONE);
503df8bae1dSRodney W. Grimes 		}
504c6d81a34SMichael Tuexen 	} else {
505c6d81a34SMichael Tuexen 		if (proto == IPPROTO_UDP) {
506026decb8SRobert Watson 			UDPSTAT_INC(udps_nosum);
507c6d81a34SMichael Tuexen 		} else {
508c6d81a34SMichael Tuexen 			/* UDPLite requires a checksum */
509c6d81a34SMichael Tuexen 			/* XXX: What is the right UDPLite MIB counter here? */
510c6d81a34SMichael Tuexen 			m_freem(m);
511c6d81a34SMichael Tuexen 			return (IPPROTO_DONE);
512c6d81a34SMichael Tuexen 		}
513c6d81a34SMichael Tuexen 	}
514df8bae1dSRodney W. Grimes 
515a86e5c96SBjoern A. Zeeb 	pcbinfo = udp_get_inpcbinfo(proto);
516df8bae1dSRodney W. Grimes 	if (IN_MULTICAST(ntohl(ip->ip_dst.s_addr)) ||
5179da85a91SMarko Zec 	    ((!V_udp_require_l2_bcast || m->m_flags & M_BCAST) &&
51841029db1SRyan Stone 	    in_broadcast(ip->ip_dst, ifp))) {
51982c23ebaSBill Fenner 		struct inpcb *last;
520e06e816fSKevin Lo 		struct inpcbhead *pcblist;
52171498f30SBruce M Simpson 		struct ip_moptions *imo;
5223329b236SRobert Watson 
523e06e816fSKevin Lo 		INP_INFO_RLOCK(pcbinfo);
524a86e5c96SBjoern A. Zeeb 		pcblist = udp_get_pcblist(proto);
525df8bae1dSRodney W. Grimes 		last = NULL;
526e06e816fSKevin Lo 		LIST_FOREACH(inp, pcblist, inp_list) {
5279c1df695SRobert Watson 			if (inp->inp_lport != uh->uh_dport)
528f76fcf6dSJeffrey Hsu 				continue;
529cfa1ca9dSYoshinobu Inoue #ifdef INET6
530369dc8ceSEivind Eklund 			if ((inp->inp_vflag & INP_IPV4) == 0)
5319c1df695SRobert Watson 				continue;
532cfa1ca9dSYoshinobu Inoue #endif
53371498f30SBruce M Simpson 			if (inp->inp_laddr.s_addr != INADDR_ANY &&
53471498f30SBruce M Simpson 			    inp->inp_laddr.s_addr != ip->ip_dst.s_addr)
5359c1df695SRobert Watson 				continue;
53671498f30SBruce M Simpson 			if (inp->inp_faddr.s_addr != INADDR_ANY &&
53771498f30SBruce M Simpson 			    inp->inp_faddr.s_addr != ip->ip_src.s_addr)
53871498f30SBruce M Simpson 				continue;
53971498f30SBruce M Simpson 			if (inp->inp_fport != 0 &&
540df8bae1dSRodney W. Grimes 			    inp->inp_fport != uh->uh_sport)
5419c1df695SRobert Watson 				continue;
54271498f30SBruce M Simpson 
543119d85f6SRobert Watson 			INP_RLOCK(inp);
544df8bae1dSRodney W. Grimes 
54583453a06SBruce M Simpson 			/*
546fa046d87SRobert Watson 			 * XXXRW: Because we weren't holding either the inpcb
547fa046d87SRobert Watson 			 * or the hash lock when we checked for a match
548fa046d87SRobert Watson 			 * before, we should probably recheck now that the
549fa046d87SRobert Watson 			 * inpcb lock is held.
550fa046d87SRobert Watson 			 */
551fa046d87SRobert Watson 
552fa046d87SRobert Watson 			/*
55371498f30SBruce M Simpson 			 * Handle socket delivery policy for any-source
55471498f30SBruce M Simpson 			 * and source-specific multicast. [RFC3678]
55583453a06SBruce M Simpson 			 */
55671498f30SBruce M Simpson 			imo = inp->inp_moptions;
557a38b1c8cSRandall Stewart 			if (IN_MULTICAST(ntohl(ip->ip_dst.s_addr))) {
558d10910e6SBruce M Simpson 				struct sockaddr_in	 group;
559d10910e6SBruce M Simpson 				int			 blocked;
560a38b1c8cSRandall Stewart 				if (imo == NULL) {
561a38b1c8cSRandall Stewart 					INP_RUNLOCK(inp);
562a38b1c8cSRandall Stewart 					continue;
563a38b1c8cSRandall Stewart 				}
564d10910e6SBruce M Simpson 				bzero(&group, sizeof(struct sockaddr_in));
565d10910e6SBruce M Simpson 				group.sin_len = sizeof(struct sockaddr_in);
566d10910e6SBruce M Simpson 				group.sin_family = AF_INET;
567d10910e6SBruce M Simpson 				group.sin_addr = ip->ip_dst;
56871498f30SBruce M Simpson 
569d10910e6SBruce M Simpson 				blocked = imo_multi_filter(imo, ifp,
570d10910e6SBruce M Simpson 					(struct sockaddr *)&group,
57171498f30SBruce M Simpson 					(struct sockaddr *)&udp_in);
572d10910e6SBruce M Simpson 				if (blocked != MCAST_PASS) {
573d10910e6SBruce M Simpson 					if (blocked == MCAST_NOTGMEMBER)
57486425c62SRobert Watson 						IPSTAT_INC(ips_notmember);
575d10910e6SBruce M Simpson 					if (blocked == MCAST_NOTSMEMBER ||
576d10910e6SBruce M Simpson 					    blocked == MCAST_MUTED)
577026decb8SRobert Watson 						UDPSTAT_INC(udps_filtermcast);
578119d85f6SRobert Watson 					INP_RUNLOCK(inp);
5799c1df695SRobert Watson 					continue;
5809c1df695SRobert Watson 				}
58183453a06SBruce M Simpson 			}
582df8bae1dSRodney W. Grimes 			if (last != NULL) {
583df8bae1dSRodney W. Grimes 				struct mbuf *n;
584df8bae1dSRodney W. Grimes 
585c3bef61eSKevin Lo 				if ((n = m_copym(m, 0, M_COPYALL, M_NOWAIT)) !=
586c3bef61eSKevin Lo 				    NULL) {
587a0a9e1b5SBryan Venteicher 					UDP_PROBE(receive, NULL, last, ip,
588a0a9e1b5SBryan Venteicher 					    last, uh);
589c0d1be08SRandall Stewart 					if (udp_append(last, ip, n, iphlen,
590c0d1be08SRandall Stewart 						&udp_in)) {
591c0d1be08SRandall Stewart 						goto inp_lost;
592c0d1be08SRandall Stewart 					}
593c19f98ebSBryan Venteicher 				}
5946a9148feSBjoern A. Zeeb 				INP_RUNLOCK(last);
595df8bae1dSRodney W. Grimes 			}
59682c23ebaSBill Fenner 			last = inp;
597df8bae1dSRodney W. Grimes 			/*
598df8bae1dSRodney W. Grimes 			 * Don't look for additional matches if this one does
599df8bae1dSRodney W. Grimes 			 * not have either the SO_REUSEPORT or SO_REUSEADDR
6003329b236SRobert Watson 			 * socket options set.  This heuristic avoids
6013329b236SRobert Watson 			 * searching through all pcbs in the common case of a
6023329b236SRobert Watson 			 * non-shared port.  It assumes that an application
6033329b236SRobert Watson 			 * will never clear these options after setting them.
604df8bae1dSRodney W. Grimes 			 */
6053329b236SRobert Watson 			if ((last->inp_socket->so_options &
6063329b236SRobert Watson 			    (SO_REUSEPORT|SO_REUSEADDR)) == 0)
607df8bae1dSRodney W. Grimes 				break;
608df8bae1dSRodney W. Grimes 		}
609df8bae1dSRodney W. Grimes 
610df8bae1dSRodney W. Grimes 		if (last == NULL) {
611df8bae1dSRodney W. Grimes 			/*
6123329b236SRobert Watson 			 * No matching pcb found; discard datagram.  (No need
6133329b236SRobert Watson 			 * to send an ICMP Port Unreachable for a broadcast
6143329b236SRobert Watson 			 * or multicast datgram.)
615df8bae1dSRodney W. Grimes 			 */
616026decb8SRobert Watson 			UDPSTAT_INC(udps_noportbcast);
617fa046d87SRobert Watson 			if (inp)
618fa046d87SRobert Watson 				INP_RUNLOCK(inp);
619e06e816fSKevin Lo 			INP_INFO_RUNLOCK(pcbinfo);
620fa046d87SRobert Watson 			goto badunlocked;
621df8bae1dSRodney W. Grimes 		}
622a0a9e1b5SBryan Venteicher 		UDP_PROBE(receive, NULL, last, ip, last, uh);
623c0d1be08SRandall Stewart 		if (udp_append(last, ip, m, iphlen, &udp_in) == 0)
624c7c7ea4bSRandall Stewart 			INP_RUNLOCK(last);
625c0d1be08SRandall Stewart 	inp_lost:
626e06e816fSKevin Lo 		INP_INFO_RUNLOCK(pcbinfo);
6278f5a8818SKevin Lo 		return (IPPROTO_DONE);
628df8bae1dSRodney W. Grimes 	}
6293329b236SRobert Watson 
630df8bae1dSRodney W. Grimes 	/*
6316d6a026bSDavid Greenman 	 * Locate pcb for datagram.
632df8bae1dSRodney W. Grimes 	 */
633c1de64a4SAndrey V. Elsukov 
6348a006adbSBjoern A. Zeeb 	/*
6358a006adbSBjoern A. Zeeb 	 * Grab info from PACKET_TAG_IPFORWARD tag prepended to the chain.
6368a006adbSBjoern A. Zeeb 	 */
637ffdbf9daSAndrey V. Elsukov 	if ((m->m_flags & M_IP_NEXTHOP) &&
638c1de64a4SAndrey V. Elsukov 	    (fwd_tag = m_tag_find(m, PACKET_TAG_IPFORWARD, NULL)) != NULL) {
6398a006adbSBjoern A. Zeeb 		struct sockaddr_in *next_hop;
6408a006adbSBjoern A. Zeeb 
6418a006adbSBjoern A. Zeeb 		next_hop = (struct sockaddr_in *)(fwd_tag + 1);
6428a006adbSBjoern A. Zeeb 
6438a006adbSBjoern A. Zeeb 		/*
6448a006adbSBjoern A. Zeeb 		 * Transparently forwarded. Pretend to be the destination.
6458a006adbSBjoern A. Zeeb 		 * Already got one like this?
6468a006adbSBjoern A. Zeeb 		 */
647e06e816fSKevin Lo 		inp = in_pcblookup_mbuf(pcbinfo, ip->ip_src, uh->uh_sport,
6488a006adbSBjoern A. Zeeb 		    ip->ip_dst, uh->uh_dport, INPLOOKUP_RLOCKPCB, ifp, m);
6498a006adbSBjoern A. Zeeb 		if (!inp) {
6508a006adbSBjoern A. Zeeb 			/*
6518a006adbSBjoern A. Zeeb 			 * It's new.  Try to find the ambushing socket.
6528a006adbSBjoern A. Zeeb 			 * Because we've rewritten the destination address,
6538a006adbSBjoern A. Zeeb 			 * any hardware-generated hash is ignored.
6548a006adbSBjoern A. Zeeb 			 */
655e06e816fSKevin Lo 			inp = in_pcblookup(pcbinfo, ip->ip_src,
6568a006adbSBjoern A. Zeeb 			    uh->uh_sport, next_hop->sin_addr,
6578a006adbSBjoern A. Zeeb 			    next_hop->sin_port ? htons(next_hop->sin_port) :
6588a006adbSBjoern A. Zeeb 			    uh->uh_dport, INPLOOKUP_WILDCARD |
6598a006adbSBjoern A. Zeeb 			    INPLOOKUP_RLOCKPCB, ifp);
6608a006adbSBjoern A. Zeeb 		}
6618a006adbSBjoern A. Zeeb 		/* Remove the tag from the packet. We don't need it anymore. */
6628a006adbSBjoern A. Zeeb 		m_tag_delete(m, fwd_tag);
663ffdbf9daSAndrey V. Elsukov 		m->m_flags &= ~M_IP_NEXTHOP;
6648a006adbSBjoern A. Zeeb 	} else
665e06e816fSKevin Lo 		inp = in_pcblookup_mbuf(pcbinfo, ip->ip_src, uh->uh_sport,
6668a006adbSBjoern A. Zeeb 		    ip->ip_dst, uh->uh_dport, INPLOOKUP_WILDCARD |
6678a006adbSBjoern A. Zeeb 		    INPLOOKUP_RLOCKPCB, ifp, m);
66815bd2b43SDavid Greenman 	if (inp == NULL) {
669afdb4274SRobert Watson 		if (udp_log_in_vain) {
670df5c0b8aSBill Fenner 			char buf[4*sizeof "123"];
67175cfc95fSAndrey A. Chernov 
67275cfc95fSAndrey A. Chernov 			strcpy(buf, inet_ntoa(ip->ip_dst));
673592071e8SBruce Evans 			log(LOG_INFO,
674592071e8SBruce Evans 			    "Connection attempt to UDP %s:%d from %s:%d\n",
675592071e8SBruce Evans 			    buf, ntohs(uh->uh_dport), inet_ntoa(ip->ip_src),
676592071e8SBruce Evans 			    ntohs(uh->uh_sport));
67775cfc95fSAndrey A. Chernov 		}
678026decb8SRobert Watson 		UDPSTAT_INC(udps_noport);
679df8bae1dSRodney W. Grimes 		if (m->m_flags & (M_BCAST | M_MCAST)) {
680026decb8SRobert Watson 			UDPSTAT_INC(udps_noportbcast);
681fa046d87SRobert Watson 			goto badunlocked;
682df8bae1dSRodney W. Grimes 		}
683603724d3SBjoern A. Zeeb 		if (V_udp_blackhole)
684fa046d87SRobert Watson 			goto badunlocked;
6851cbd978eSLuigi Rizzo 		if (badport_bandlim(BANDLIM_ICMP_UNREACH) < 0)
686fa046d87SRobert Watson 			goto badunlocked;
68704287599SRuslan Ermilov 		*ip = save_ip;
688582a7760SBruce Evans 		icmp_error(m, ICMP_UNREACH, ICMP_UNREACH_PORT, 0, 0);
6898f5a8818SKevin Lo 		return (IPPROTO_DONE);
690df8bae1dSRodney W. Grimes 	}
6913329b236SRobert Watson 
6923329b236SRobert Watson 	/*
6933329b236SRobert Watson 	 * Check the minimum TTL for socket.
6943329b236SRobert Watson 	 */
695fa046d87SRobert Watson 	INP_RLOCK_ASSERT(inp);
69610cc62b7SRobert Watson 	if (inp->inp_ip_minttl && inp->inp_ip_minttl > ip->ip_ttl) {
69710cc62b7SRobert Watson 		INP_RUNLOCK(inp);
698fa046d87SRobert Watson 		m_freem(m);
6998f5a8818SKevin Lo 		return (IPPROTO_DONE);
70010cc62b7SRobert Watson 	}
701e06e816fSKevin Lo 	if (cscov_partial) {
702e06e816fSKevin Lo 		struct udpcb *up;
703e06e816fSKevin Lo 
704e06e816fSKevin Lo 		up = intoudpcb(inp);
70583e95fb3SMichael Tuexen 		if (up->u_rxcslen == 0 || up->u_rxcslen > len) {
706e06e816fSKevin Lo 			INP_RUNLOCK(inp);
707e06e816fSKevin Lo 			m_freem(m);
7088f5a8818SKevin Lo 			return (IPPROTO_DONE);
709e06e816fSKevin Lo 		}
710e06e816fSKevin Lo 	}
71157f60867SMark Johnston 
7121ad19fb6SMark Johnston 	UDP_PROBE(receive, NULL, inp, ip, inp, uh);
713c0d1be08SRandall Stewart 	if (udp_append(inp, ip, m, iphlen, &udp_in) == 0)
714119d85f6SRobert Watson 		INP_RUNLOCK(inp);
7158f5a8818SKevin Lo 	return (IPPROTO_DONE);
71661ffc0b1SJeffrey Hsu 
717f76fcf6dSJeffrey Hsu badunlocked:
718df8bae1dSRodney W. Grimes 	m_freem(m);
7198f5a8818SKevin Lo 	return (IPPROTO_DONE);
720cfa1ca9dSYoshinobu Inoue }
72179288c11SBjoern A. Zeeb #endif /* INET */
722cfa1ca9dSYoshinobu Inoue 
723cfa1ca9dSYoshinobu Inoue /*
7243329b236SRobert Watson  * Notify a udp user of an asynchronous error; just wake up so that they can
7253329b236SRobert Watson  * collect error status.
726df8bae1dSRodney W. Grimes  */
7273ce144eaSJeffrey Hsu struct inpcb *
7283329b236SRobert Watson udp_notify(struct inpcb *inp, int errno)
729df8bae1dSRodney W. Grimes {
7303329b236SRobert Watson 
731ac9ae279SRobert Watson 	/*
732ac9ae279SRobert Watson 	 * While udp_ctlinput() always calls udp_notify() with a read lock
733ac9ae279SRobert Watson 	 * when invoking it directly, in_pcbnotifyall() currently uses write
734ac9ae279SRobert Watson 	 * locks due to sharing code with TCP.  For now, accept either a read
735ac9ae279SRobert Watson 	 * or a write lock, but a read lock is sufficient.
736ac9ae279SRobert Watson 	 */
737ac9ae279SRobert Watson 	INP_LOCK_ASSERT(inp);
73884cc0778SGeorge V. Neville-Neil 	if ((errno == EHOSTUNREACH || errno == ENETUNREACH ||
73984cc0778SGeorge V. Neville-Neil 	     errno == EHOSTDOWN) && inp->inp_route.ro_rt) {
74084cc0778SGeorge V. Neville-Neil 		RTFREE(inp->inp_route.ro_rt);
74184cc0778SGeorge V. Neville-Neil 		inp->inp_route.ro_rt = (struct rtentry *)NULL;
74284cc0778SGeorge V. Neville-Neil 	}
7438501a69cSRobert Watson 
744df8bae1dSRodney W. Grimes 	inp->inp_socket->so_error = errno;
745df8bae1dSRodney W. Grimes 	sorwakeup(inp->inp_socket);
746df8bae1dSRodney W. Grimes 	sowwakeup(inp->inp_socket);
7473329b236SRobert Watson 	return (inp);
748df8bae1dSRodney W. Grimes }
749df8bae1dSRodney W. Grimes 
75079288c11SBjoern A. Zeeb #ifdef INET
751e06e816fSKevin Lo static void
752e06e816fSKevin Lo udp_common_ctlinput(int cmd, struct sockaddr *sa, void *vip,
753e06e816fSKevin Lo     struct inpcbinfo *pcbinfo)
754df8bae1dSRodney W. Grimes {
755c693a045SJonathan Lemon 	struct ip *ip = vip;
756c693a045SJonathan Lemon 	struct udphdr *uh;
757c693a045SJonathan Lemon 	struct in_addr faddr;
758c693a045SJonathan Lemon 	struct inpcb *inp;
759c693a045SJonathan Lemon 
760c693a045SJonathan Lemon 	faddr = ((struct sockaddr_in *)sa)->sin_addr;
761c693a045SJonathan Lemon 	if (sa->sa_family != AF_INET || faddr.s_addr == INADDR_ANY)
762c693a045SJonathan Lemon 		return;
763df8bae1dSRodney W. Grimes 
76484cc0778SGeorge V. Neville-Neil 	if (PRC_IS_REDIRECT(cmd)) {
76584cc0778SGeorge V. Neville-Neil 		/* signal EHOSTDOWN, as it flushes the cached route */
7664f321dbdSBjoern A. Zeeb 		in_pcbnotifyall(&V_udbinfo, faddr, EHOSTDOWN, udp_notify);
76797d8d152SAndre Oppermann 		return;
76884cc0778SGeorge V. Neville-Neil 	}
7693329b236SRobert Watson 
77097d8d152SAndre Oppermann 	/*
77197d8d152SAndre Oppermann 	 * Hostdead is ugly because it goes linearly through all PCBs.
7723329b236SRobert Watson 	 *
7733329b236SRobert Watson 	 * XXX: We never get this from ICMP, otherwise it makes an excellent
7743329b236SRobert Watson 	 * DoS attack on machines with many connections.
77597d8d152SAndre Oppermann 	 */
77697d8d152SAndre Oppermann 	if (cmd == PRC_HOSTDEAD)
777af1ee11dSRobert Watson 		ip = NULL;
778d1c54148SJesper Skriver 	else if ((unsigned)cmd >= PRC_NCMDS || inetctlerrmap[cmd] == 0)
779df8bae1dSRodney W. Grimes 		return;
780af1ee11dSRobert Watson 	if (ip != NULL) {
781df8bae1dSRodney W. Grimes 		uh = (struct udphdr *)((caddr_t)ip + (ip->ip_hl << 2));
782e06e816fSKevin Lo 		inp = in_pcblookup(pcbinfo, faddr, uh->uh_dport,
783fa046d87SRobert Watson 		    ip->ip_src, uh->uh_sport, INPLOOKUP_RLOCKPCB, NULL);
784f76fcf6dSJeffrey Hsu 		if (inp != NULL) {
785fa046d87SRobert Watson 			INP_RLOCK_ASSERT(inp);
786f76fcf6dSJeffrey Hsu 			if (inp->inp_socket != NULL) {
787f5514f08SRobert Watson 				udp_notify(inp, inetctlerrmap[cmd]);
788f76fcf6dSJeffrey Hsu 			}
789ac9ae279SRobert Watson 			INP_RUNLOCK(inp);
790abb901c5SRandall Stewart 		} else {
791abb901c5SRandall Stewart 			inp = in_pcblookup(pcbinfo, faddr, uh->uh_dport,
792abb901c5SRandall Stewart 					   ip->ip_src, uh->uh_sport,
793abb901c5SRandall Stewart 					   INPLOOKUP_WILDCARD | INPLOOKUP_RLOCKPCB, NULL);
794abb901c5SRandall Stewart 			if (inp != NULL) {
795abb901c5SRandall Stewart 				struct udpcb *up;
796abb901c5SRandall Stewart 
797abb901c5SRandall Stewart 				up = intoudpcb(inp);
798abb901c5SRandall Stewart 				if (up->u_icmp_func != NULL) {
799abb901c5SRandall Stewart 					INP_RUNLOCK(inp);
800abb901c5SRandall Stewart 					(*up->u_icmp_func)(cmd, sa, vip, up->u_tun_ctx);
801abb901c5SRandall Stewart 				} else {
802abb901c5SRandall Stewart 					INP_RUNLOCK(inp);
803abb901c5SRandall Stewart 				}
804abb901c5SRandall Stewart 			}
805f76fcf6dSJeffrey Hsu 		}
806df8bae1dSRodney W. Grimes 	} else
807e06e816fSKevin Lo 		in_pcbnotifyall(pcbinfo, faddr, inetctlerrmap[cmd],
808f5514f08SRobert Watson 		    udp_notify);
809df8bae1dSRodney W. Grimes }
810e06e816fSKevin Lo void
811e06e816fSKevin Lo udp_ctlinput(int cmd, struct sockaddr *sa, void *vip)
812e06e816fSKevin Lo {
813e06e816fSKevin Lo 
814e06e816fSKevin Lo 	return (udp_common_ctlinput(cmd, sa, vip, &V_udbinfo));
815e06e816fSKevin Lo }
816e06e816fSKevin Lo 
817e06e816fSKevin Lo void
818e06e816fSKevin Lo udplite_ctlinput(int cmd, struct sockaddr *sa, void *vip)
819e06e816fSKevin Lo {
820e06e816fSKevin Lo 
821e06e816fSKevin Lo 	return (udp_common_ctlinput(cmd, sa, vip, &V_ulitecbinfo));
822e06e816fSKevin Lo }
82379288c11SBjoern A. Zeeb #endif /* INET */
824df8bae1dSRodney W. Grimes 
8250312fbe9SPoul-Henning Kamp static int
82682d9ae4eSPoul-Henning Kamp udp_pcblist(SYSCTL_HANDLER_ARGS)
82798271db4SGarrett Wollman {
828277afaffSRobert Watson 	int error, i, n;
82998271db4SGarrett Wollman 	struct inpcb *inp, **inp_list;
83098271db4SGarrett Wollman 	inp_gen_t gencnt;
83198271db4SGarrett Wollman 	struct xinpgen xig;
83298271db4SGarrett Wollman 
83398271db4SGarrett Wollman 	/*
834f5514f08SRobert Watson 	 * The process of preparing the PCB list is too time-consuming and
83598271db4SGarrett Wollman 	 * resource-intensive to repeat twice on every request.
83698271db4SGarrett Wollman 	 */
83798271db4SGarrett Wollman 	if (req->oldptr == 0) {
838603724d3SBjoern A. Zeeb 		n = V_udbinfo.ipi_count;
839c007b96aSJohn Baldwin 		n += imax(n / 8, 10);
840c007b96aSJohn Baldwin 		req->oldidx = 2 * (sizeof xig) + n * sizeof(struct xinpcb);
8413329b236SRobert Watson 		return (0);
84298271db4SGarrett Wollman 	}
84398271db4SGarrett Wollman 
84498271db4SGarrett Wollman 	if (req->newptr != 0)
8453329b236SRobert Watson 		return (EPERM);
84698271db4SGarrett Wollman 
84798271db4SGarrett Wollman 	/*
84898271db4SGarrett Wollman 	 * OK, now we're committed to doing something.
84998271db4SGarrett Wollman 	 */
850603724d3SBjoern A. Zeeb 	INP_INFO_RLOCK(&V_udbinfo);
851603724d3SBjoern A. Zeeb 	gencnt = V_udbinfo.ipi_gencnt;
852603724d3SBjoern A. Zeeb 	n = V_udbinfo.ipi_count;
853603724d3SBjoern A. Zeeb 	INP_INFO_RUNLOCK(&V_udbinfo);
85498271db4SGarrett Wollman 
85547934cefSDon Lewis 	error = sysctl_wire_old_buffer(req, 2 * (sizeof xig)
8565c38b6dbSDon Lewis 		+ n * sizeof(struct xinpcb));
85747934cefSDon Lewis 	if (error != 0)
85847934cefSDon Lewis 		return (error);
8595c38b6dbSDon Lewis 
86098271db4SGarrett Wollman 	xig.xig_len = sizeof xig;
86198271db4SGarrett Wollman 	xig.xig_count = n;
86298271db4SGarrett Wollman 	xig.xig_gen = gencnt;
86398271db4SGarrett Wollman 	xig.xig_sogen = so_gencnt;
86498271db4SGarrett Wollman 	error = SYSCTL_OUT(req, &xig, sizeof xig);
86598271db4SGarrett Wollman 	if (error)
8663329b236SRobert Watson 		return (error);
86798271db4SGarrett Wollman 
868a163d034SWarner Losh 	inp_list = malloc(n * sizeof *inp_list, M_TEMP, M_WAITOK);
86999d628d5SPedro F. Giffuni 	if (inp_list == NULL)
8703329b236SRobert Watson 		return (ENOMEM);
87198271db4SGarrett Wollman 
872603724d3SBjoern A. Zeeb 	INP_INFO_RLOCK(&V_udbinfo);
873603724d3SBjoern A. Zeeb 	for (inp = LIST_FIRST(V_udbinfo.ipi_listhead), i = 0; inp && i < n;
874fc2ffbe6SPoul-Henning Kamp 	     inp = LIST_NEXT(inp, inp_list)) {
875d0e157f6SBjoern A. Zeeb 		INP_WLOCK(inp);
8762ded288cSJeffrey Hsu 		if (inp->inp_gencnt <= gencnt &&
877d0e157f6SBjoern A. Zeeb 		    cr_canseeinpcb(req->td->td_ucred, inp) == 0) {
878d0e157f6SBjoern A. Zeeb 			in_pcbref(inp);
87998271db4SGarrett Wollman 			inp_list[i++] = inp;
880d0e157f6SBjoern A. Zeeb 		}
881d0e157f6SBjoern A. Zeeb 		INP_WUNLOCK(inp);
8824787fd37SPaul Saab 	}
883603724d3SBjoern A. Zeeb 	INP_INFO_RUNLOCK(&V_udbinfo);
88498271db4SGarrett Wollman 	n = i;
88598271db4SGarrett Wollman 
88698271db4SGarrett Wollman 	error = 0;
88798271db4SGarrett Wollman 	for (i = 0; i < n; i++) {
88898271db4SGarrett Wollman 		inp = inp_list[i];
8899622e84fSRobert Watson 		INP_RLOCK(inp);
89098271db4SGarrett Wollman 		if (inp->inp_gencnt <= gencnt) {
89198271db4SGarrett Wollman 			struct xinpcb xi;
892d0e157f6SBjoern A. Zeeb 
893fd94099eSColin Percival 			bzero(&xi, sizeof(xi));
89498271db4SGarrett Wollman 			xi.xi_len = sizeof xi;
89598271db4SGarrett Wollman 			/* XXX should avoid extra copy */
89698271db4SGarrett Wollman 			bcopy(inp, &xi.xi_inp, sizeof *inp);
89798271db4SGarrett Wollman 			if (inp->inp_socket)
89898271db4SGarrett Wollman 				sotoxsocket(inp->inp_socket, &xi.xi_socket);
8994b40c56cSJeffrey Hsu 			xi.xi_inp.inp_gencnt = inp->inp_gencnt;
9009622e84fSRobert Watson 			INP_RUNLOCK(inp);
90198271db4SGarrett Wollman 			error = SYSCTL_OUT(req, &xi, sizeof xi);
902d915b280SStephan Uphoff 		} else
9039622e84fSRobert Watson 			INP_RUNLOCK(inp);
90498271db4SGarrett Wollman 	}
905d0e157f6SBjoern A. Zeeb 	INP_INFO_WLOCK(&V_udbinfo);
906d0e157f6SBjoern A. Zeeb 	for (i = 0; i < n; i++) {
907d0e157f6SBjoern A. Zeeb 		inp = inp_list[i];
908fa046d87SRobert Watson 		INP_RLOCK(inp);
909fa046d87SRobert Watson 		if (!in_pcbrele_rlocked(inp))
910fa046d87SRobert Watson 			INP_RUNLOCK(inp);
911d0e157f6SBjoern A. Zeeb 	}
912d0e157f6SBjoern A. Zeeb 	INP_INFO_WUNLOCK(&V_udbinfo);
913d0e157f6SBjoern A. Zeeb 
91498271db4SGarrett Wollman 	if (!error) {
91598271db4SGarrett Wollman 		/*
9163329b236SRobert Watson 		 * Give the user an updated idea of our state.  If the
9173329b236SRobert Watson 		 * generation differs from what we told her before, she knows
9183329b236SRobert Watson 		 * that something happened while we were processing this
9193329b236SRobert Watson 		 * request, and it might be necessary to retry.
92098271db4SGarrett Wollman 		 */
921603724d3SBjoern A. Zeeb 		INP_INFO_RLOCK(&V_udbinfo);
922603724d3SBjoern A. Zeeb 		xig.xig_gen = V_udbinfo.ipi_gencnt;
92398271db4SGarrett Wollman 		xig.xig_sogen = so_gencnt;
924603724d3SBjoern A. Zeeb 		xig.xig_count = V_udbinfo.ipi_count;
925603724d3SBjoern A. Zeeb 		INP_INFO_RUNLOCK(&V_udbinfo);
92698271db4SGarrett Wollman 		error = SYSCTL_OUT(req, &xig, sizeof xig);
92798271db4SGarrett Wollman 	}
92898271db4SGarrett Wollman 	free(inp_list, M_TEMP);
9293329b236SRobert Watson 	return (error);
93098271db4SGarrett Wollman }
93198271db4SGarrett Wollman 
93279c3d51bSMatthew D Fleming SYSCTL_PROC(_net_inet_udp, UDPCTL_PCBLIST, pcblist,
93379c3d51bSMatthew D Fleming     CTLTYPE_OPAQUE | CTLFLAG_RD, NULL, 0,
93498271db4SGarrett Wollman     udp_pcblist, "S,xinpcb", "List of active UDP sockets");
93598271db4SGarrett Wollman 
93679288c11SBjoern A. Zeeb #ifdef INET
93798271db4SGarrett Wollman static int
93882d9ae4eSPoul-Henning Kamp udp_getcred(SYSCTL_HANDLER_ARGS)
939490d50b6SBrian Feldman {
940c0511d3bSBrian Feldman 	struct xucred xuc;
941490d50b6SBrian Feldman 	struct sockaddr_in addrs[2];
942490d50b6SBrian Feldman 	struct inpcb *inp;
943277afaffSRobert Watson 	int error;
944490d50b6SBrian Feldman 
94532f9753cSRobert Watson 	error = priv_check(req->td, PRIV_NETINET_GETCRED);
946490d50b6SBrian Feldman 	if (error)
947490d50b6SBrian Feldman 		return (error);
948490d50b6SBrian Feldman 	error = SYSCTL_IN(req, addrs, sizeof(addrs));
949490d50b6SBrian Feldman 	if (error)
950490d50b6SBrian Feldman 		return (error);
951fa046d87SRobert Watson 	inp = in_pcblookup(&V_udbinfo, addrs[1].sin_addr, addrs[1].sin_port,
952fa046d87SRobert Watson 	    addrs[0].sin_addr, addrs[0].sin_port,
953fa046d87SRobert Watson 	    INPLOOKUP_WILDCARD | INPLOOKUP_RLOCKPCB, NULL);
9549622e84fSRobert Watson 	if (inp != NULL) {
955fa046d87SRobert Watson 		INP_RLOCK_ASSERT(inp);
9569622e84fSRobert Watson 		if (inp->inp_socket == NULL)
9579622e84fSRobert Watson 			error = ENOENT;
9589622e84fSRobert Watson 		if (error == 0)
959f08ef6c5SBjoern A. Zeeb 			error = cr_canseeinpcb(req->td->td_ucred, inp);
9609622e84fSRobert Watson 		if (error == 0)
96186d02c5cSBjoern A. Zeeb 			cru2x(inp->inp_cred, &xuc);
9629622e84fSRobert Watson 		INP_RUNLOCK(inp);
963fa046d87SRobert Watson 	} else
9649622e84fSRobert Watson 		error = ENOENT;
9650e1eebb8SDon Lewis 	if (error == 0)
9660e1eebb8SDon Lewis 		error = SYSCTL_OUT(req, &xuc, sizeof(struct xucred));
967490d50b6SBrian Feldman 	return (error);
968490d50b6SBrian Feldman }
969490d50b6SBrian Feldman 
9707ce87f12SDavid Malone SYSCTL_PROC(_net_inet_udp, OID_AUTO, getcred,
9717ce87f12SDavid Malone     CTLTYPE_OPAQUE|CTLFLAG_RW|CTLFLAG_PRISON, 0, 0,
9727ce87f12SDavid Malone     udp_getcred, "S,xucred", "Get the xucred of a UDP connection");
97379288c11SBjoern A. Zeeb #endif /* INET */
974490d50b6SBrian Feldman 
9757b495c44SVANHULLEBUS Yvan int
9767b495c44SVANHULLEBUS Yvan udp_ctloutput(struct socket *so, struct sockopt *sopt)
9777b495c44SVANHULLEBUS Yvan {
9787b495c44SVANHULLEBUS Yvan 	struct inpcb *inp;
9797b495c44SVANHULLEBUS Yvan 	struct udpcb *up;
980e06e816fSKevin Lo 	int isudplite, error, optval;
9817b495c44SVANHULLEBUS Yvan 
982e06e816fSKevin Lo 	error = 0;
983e06e816fSKevin Lo 	isudplite = (so->so_proto->pr_protocol == IPPROTO_UDPLITE) ? 1 : 0;
9847b495c44SVANHULLEBUS Yvan 	inp = sotoinpcb(so);
9857b495c44SVANHULLEBUS Yvan 	KASSERT(inp != NULL, ("%s: inp == NULL", __func__));
9867b495c44SVANHULLEBUS Yvan 	INP_WLOCK(inp);
987e06e816fSKevin Lo 	if (sopt->sopt_level != so->so_proto->pr_protocol) {
9887b495c44SVANHULLEBUS Yvan #ifdef INET6
9897b495c44SVANHULLEBUS Yvan 		if (INP_CHECK_SOCKAF(so, AF_INET6)) {
9907b495c44SVANHULLEBUS Yvan 			INP_WUNLOCK(inp);
9917b495c44SVANHULLEBUS Yvan 			error = ip6_ctloutput(so, sopt);
99279288c11SBjoern A. Zeeb 		}
9937b495c44SVANHULLEBUS Yvan #endif
99479288c11SBjoern A. Zeeb #if defined(INET) && defined(INET6)
99579288c11SBjoern A. Zeeb 		else
99679288c11SBjoern A. Zeeb #endif
99779288c11SBjoern A. Zeeb #ifdef INET
99879288c11SBjoern A. Zeeb 		{
9997b495c44SVANHULLEBUS Yvan 			INP_WUNLOCK(inp);
10007b495c44SVANHULLEBUS Yvan 			error = ip_ctloutput(so, sopt);
10017b495c44SVANHULLEBUS Yvan 		}
10027b495c44SVANHULLEBUS Yvan #endif
10037b495c44SVANHULLEBUS Yvan 		return (error);
10047b495c44SVANHULLEBUS Yvan 	}
10057b495c44SVANHULLEBUS Yvan 
10067b495c44SVANHULLEBUS Yvan 	switch (sopt->sopt_dir) {
10077b495c44SVANHULLEBUS Yvan 	case SOPT_SET:
10087b495c44SVANHULLEBUS Yvan 		switch (sopt->sopt_name) {
1009*fcf59617SAndrey V. Elsukov #if defined(IPSEC) || defined(IPSEC_SUPPORT)
1010*fcf59617SAndrey V. Elsukov #ifdef INET
10117b495c44SVANHULLEBUS Yvan 		case UDP_ENCAP:
1012*fcf59617SAndrey V. Elsukov 			if (!IPSEC_ENABLED(ipv4)) {
10137b495c44SVANHULLEBUS Yvan 				INP_WUNLOCK(inp);
1014*fcf59617SAndrey V. Elsukov 				return (ENOPROTOOPT);
10157b495c44SVANHULLEBUS Yvan 			}
1016*fcf59617SAndrey V. Elsukov 			error = UDPENCAP_PCBCTL(inp, sopt);
10177b495c44SVANHULLEBUS Yvan 			break;
1018*fcf59617SAndrey V. Elsukov #endif /* INET */
1019*fcf59617SAndrey V. Elsukov #endif /* IPSEC */
1020e06e816fSKevin Lo 		case UDPLITE_SEND_CSCOV:
1021e06e816fSKevin Lo 		case UDPLITE_RECV_CSCOV:
1022e06e816fSKevin Lo 			if (!isudplite) {
1023e06e816fSKevin Lo 				INP_WUNLOCK(inp);
1024e06e816fSKevin Lo 				error = ENOPROTOOPT;
1025e06e816fSKevin Lo 				break;
1026e06e816fSKevin Lo 			}
1027e06e816fSKevin Lo 			INP_WUNLOCK(inp);
1028e06e816fSKevin Lo 			error = sooptcopyin(sopt, &optval, sizeof(optval),
1029e06e816fSKevin Lo 			    sizeof(optval));
1030e06e816fSKevin Lo 			if (error != 0)
1031e06e816fSKevin Lo 				break;
1032e06e816fSKevin Lo 			inp = sotoinpcb(so);
1033e06e816fSKevin Lo 			KASSERT(inp != NULL, ("%s: inp == NULL", __func__));
1034e06e816fSKevin Lo 			INP_WLOCK(inp);
1035e06e816fSKevin Lo 			up = intoudpcb(inp);
1036e06e816fSKevin Lo 			KASSERT(up != NULL, ("%s: up == NULL", __func__));
103703f90784SMichael Tuexen 			if ((optval != 0 && optval < 8) || (optval > 65535)) {
1038e06e816fSKevin Lo 				INP_WUNLOCK(inp);
1039e06e816fSKevin Lo 				error = EINVAL;
1040e06e816fSKevin Lo 				break;
1041e06e816fSKevin Lo 			}
1042e06e816fSKevin Lo 			if (sopt->sopt_name == UDPLITE_SEND_CSCOV)
1043e06e816fSKevin Lo 				up->u_txcslen = optval;
1044e06e816fSKevin Lo 			else
1045e06e816fSKevin Lo 				up->u_rxcslen = optval;
1046e06e816fSKevin Lo 			INP_WUNLOCK(inp);
1047e06e816fSKevin Lo 			break;
10487b495c44SVANHULLEBUS Yvan 		default:
10497b495c44SVANHULLEBUS Yvan 			INP_WUNLOCK(inp);
10507b495c44SVANHULLEBUS Yvan 			error = ENOPROTOOPT;
10517b495c44SVANHULLEBUS Yvan 			break;
10527b495c44SVANHULLEBUS Yvan 		}
10537b495c44SVANHULLEBUS Yvan 		break;
10547b495c44SVANHULLEBUS Yvan 	case SOPT_GET:
10557b495c44SVANHULLEBUS Yvan 		switch (sopt->sopt_name) {
1056*fcf59617SAndrey V. Elsukov #if defined(IPSEC) || defined(IPSEC_SUPPORT)
1057*fcf59617SAndrey V. Elsukov #ifdef INET
10587b495c44SVANHULLEBUS Yvan 		case UDP_ENCAP:
1059*fcf59617SAndrey V. Elsukov 			if (!IPSEC_ENABLED(ipv4)) {
10607b495c44SVANHULLEBUS Yvan 				INP_WUNLOCK(inp);
1061*fcf59617SAndrey V. Elsukov 				return (ENOPROTOOPT);
1062*fcf59617SAndrey V. Elsukov 			}
1063*fcf59617SAndrey V. Elsukov 			error = UDPENCAP_PCBCTL(inp, sopt);
10647b495c44SVANHULLEBUS Yvan 			break;
1065*fcf59617SAndrey V. Elsukov #endif /* INET */
1066*fcf59617SAndrey V. Elsukov #endif /* IPSEC */
1067e06e816fSKevin Lo 		case UDPLITE_SEND_CSCOV:
1068e06e816fSKevin Lo 		case UDPLITE_RECV_CSCOV:
1069e06e816fSKevin Lo 			if (!isudplite) {
1070e06e816fSKevin Lo 				INP_WUNLOCK(inp);
1071e06e816fSKevin Lo 				error = ENOPROTOOPT;
1072e06e816fSKevin Lo 				break;
1073e06e816fSKevin Lo 			}
1074e06e816fSKevin Lo 			up = intoudpcb(inp);
1075e06e816fSKevin Lo 			KASSERT(up != NULL, ("%s: up == NULL", __func__));
1076e06e816fSKevin Lo 			if (sopt->sopt_name == UDPLITE_SEND_CSCOV)
1077e06e816fSKevin Lo 				optval = up->u_txcslen;
1078e06e816fSKevin Lo 			else
1079e06e816fSKevin Lo 				optval = up->u_rxcslen;
1080e06e816fSKevin Lo 			INP_WUNLOCK(inp);
1081e06e816fSKevin Lo 			error = sooptcopyout(sopt, &optval, sizeof(optval));
1082e06e816fSKevin Lo 			break;
10837b495c44SVANHULLEBUS Yvan 		default:
10847b495c44SVANHULLEBUS Yvan 			INP_WUNLOCK(inp);
10857b495c44SVANHULLEBUS Yvan 			error = ENOPROTOOPT;
10867b495c44SVANHULLEBUS Yvan 			break;
10877b495c44SVANHULLEBUS Yvan 		}
10887b495c44SVANHULLEBUS Yvan 		break;
10897b495c44SVANHULLEBUS Yvan 	}
10907b495c44SVANHULLEBUS Yvan 	return (error);
10917b495c44SVANHULLEBUS Yvan }
10927b495c44SVANHULLEBUS Yvan 
109379288c11SBjoern A. Zeeb #ifdef INET
1094fa046d87SRobert Watson #define	UH_WLOCKED	2
1095fa046d87SRobert Watson #define	UH_RLOCKED	1
1096fa046d87SRobert Watson #define	UH_UNLOCKED	0
1097490d50b6SBrian Feldman static int
10983329b236SRobert Watson udp_output(struct inpcb *inp, struct mbuf *m, struct sockaddr *addr,
10993329b236SRobert Watson     struct mbuf *control, struct thread *td)
1100df8bae1dSRodney W. Grimes {
11013329b236SRobert Watson 	struct udpiphdr *ui;
11023329b236SRobert Watson 	int len = m->m_pkthdr.len;
110390162a4eSIan Dowse 	struct in_addr faddr, laddr;
1104c557ae16SIan Dowse 	struct cmsghdr *cm;
1105e06e816fSKevin Lo 	struct inpcbinfo *pcbinfo;
1106c557ae16SIan Dowse 	struct sockaddr_in *sin, src;
1107e06e816fSKevin Lo 	int cscov_partial = 0;
110890162a4eSIan Dowse 	int error = 0;
11098afa2304SBruce M Simpson 	int ipflags;
111090162a4eSIan Dowse 	u_short fport, lport;
111184cc0778SGeorge V. Neville-Neil 	int unlock_udbinfo, unlock_inp;
1112f584d74bSMichael Tuexen 	u_char tos;
1113e06e816fSKevin Lo 	uint8_t pr;
1114e06e816fSKevin Lo 	uint16_t cscov = 0;
11159d3ddf43SAdrian Chadd 	uint32_t flowid = 0;
1116c2529042SHans Petter Selasky 	uint8_t flowtype = M_HASHTYPE_NONE;
1117df8bae1dSRodney W. Grimes 
11185c32ea65SRobert Watson 	/*
11195c32ea65SRobert Watson 	 * udp_output() may need to temporarily bind or connect the current
1120f5514f08SRobert Watson 	 * inpcb.  As such, we don't know up front whether we will need the
1121f5514f08SRobert Watson 	 * pcbinfo lock or not.  Do any work to decide what is needed up
1122f5514f08SRobert Watson 	 * front before acquiring any locks.
11235c32ea65SRobert Watson 	 */
1124430d30d8SBill Fenner 	if (len + sizeof(struct udpiphdr) > IP_MAXPACKET) {
1125c557ae16SIan Dowse 		if (control)
1126c557ae16SIan Dowse 			m_freem(control);
11275c32ea65SRobert Watson 		m_freem(m);
11283329b236SRobert Watson 		return (EMSGSIZE);
1129430d30d8SBill Fenner 	}
1130430d30d8SBill Fenner 
11311b7f0384SBruce M Simpson 	src.sin_family = 0;
113284cc0778SGeorge V. Neville-Neil 	sin = (struct sockaddr_in *)addr;
113384cc0778SGeorge V. Neville-Neil 	if (sin == NULL ||
113484cc0778SGeorge V. Neville-Neil 	    (inp->inp_laddr.s_addr == INADDR_ANY && inp->inp_lport == 0)) {
113584cc0778SGeorge V. Neville-Neil 		INP_WLOCK(inp);
113684cc0778SGeorge V. Neville-Neil 		unlock_inp = UH_WLOCKED;
113784cc0778SGeorge V. Neville-Neil 	} else {
11380cfdff24SBjoern A. Zeeb 		INP_RLOCK(inp);
113984cc0778SGeorge V. Neville-Neil 		unlock_inp = UH_RLOCKED;
114084cc0778SGeorge V. Neville-Neil 	}
1141f584d74bSMichael Tuexen 	tos = inp->inp_ip_tos;
1142c557ae16SIan Dowse 	if (control != NULL) {
1143c557ae16SIan Dowse 		/*
11443329b236SRobert Watson 		 * XXX: Currently, we assume all the optional information is
11453329b236SRobert Watson 		 * stored in a single mbuf.
1146c557ae16SIan Dowse 		 */
1147c557ae16SIan Dowse 		if (control->m_next) {
114884cc0778SGeorge V. Neville-Neil 			if (unlock_inp == UH_WLOCKED)
114984cc0778SGeorge V. Neville-Neil 				INP_WUNLOCK(inp);
115084cc0778SGeorge V. Neville-Neil 			else
11510cfdff24SBjoern A. Zeeb 				INP_RUNLOCK(inp);
1152c557ae16SIan Dowse 			m_freem(control);
11535c32ea65SRobert Watson 			m_freem(m);
11543329b236SRobert Watson 			return (EINVAL);
1155c557ae16SIan Dowse 		}
1156c557ae16SIan Dowse 		for (; control->m_len > 0;
1157c557ae16SIan Dowse 		    control->m_data += CMSG_ALIGN(cm->cmsg_len),
1158c557ae16SIan Dowse 		    control->m_len -= CMSG_ALIGN(cm->cmsg_len)) {
1159c557ae16SIan Dowse 			cm = mtod(control, struct cmsghdr *);
1160af1ee11dSRobert Watson 			if (control->m_len < sizeof(*cm) || cm->cmsg_len == 0
1161af1ee11dSRobert Watson 			    || cm->cmsg_len > control->m_len) {
1162c557ae16SIan Dowse 				error = EINVAL;
1163c557ae16SIan Dowse 				break;
1164c557ae16SIan Dowse 			}
1165c557ae16SIan Dowse 			if (cm->cmsg_level != IPPROTO_IP)
1166c557ae16SIan Dowse 				continue;
1167c557ae16SIan Dowse 
1168c557ae16SIan Dowse 			switch (cm->cmsg_type) {
1169c557ae16SIan Dowse 			case IP_SENDSRCADDR:
1170c557ae16SIan Dowse 				if (cm->cmsg_len !=
1171c557ae16SIan Dowse 				    CMSG_LEN(sizeof(struct in_addr))) {
1172c557ae16SIan Dowse 					error = EINVAL;
1173c557ae16SIan Dowse 					break;
1174c557ae16SIan Dowse 				}
1175c557ae16SIan Dowse 				bzero(&src, sizeof(src));
1176c557ae16SIan Dowse 				src.sin_family = AF_INET;
1177c557ae16SIan Dowse 				src.sin_len = sizeof(src);
1178c557ae16SIan Dowse 				src.sin_port = inp->inp_lport;
1179af1ee11dSRobert Watson 				src.sin_addr =
1180af1ee11dSRobert Watson 				    *(struct in_addr *)CMSG_DATA(cm);
1181c557ae16SIan Dowse 				break;
1182af1ee11dSRobert Watson 
1183f584d74bSMichael Tuexen 			case IP_TOS:
1184f584d74bSMichael Tuexen 				if (cm->cmsg_len != CMSG_LEN(sizeof(u_char))) {
1185f584d74bSMichael Tuexen 					error = EINVAL;
1186f584d74bSMichael Tuexen 					break;
1187f584d74bSMichael Tuexen 				}
1188f584d74bSMichael Tuexen 				tos = *(u_char *)CMSG_DATA(cm);
1189f584d74bSMichael Tuexen 				break;
1190f584d74bSMichael Tuexen 
11919d3ddf43SAdrian Chadd 			case IP_FLOWID:
11929d3ddf43SAdrian Chadd 				if (cm->cmsg_len != CMSG_LEN(sizeof(uint32_t))) {
11939d3ddf43SAdrian Chadd 					error = EINVAL;
11949d3ddf43SAdrian Chadd 					break;
11959d3ddf43SAdrian Chadd 				}
11969d3ddf43SAdrian Chadd 				flowid = *(uint32_t *) CMSG_DATA(cm);
11979d3ddf43SAdrian Chadd 				break;
11989d3ddf43SAdrian Chadd 
11999d3ddf43SAdrian Chadd 			case IP_FLOWTYPE:
12009d3ddf43SAdrian Chadd 				if (cm->cmsg_len != CMSG_LEN(sizeof(uint32_t))) {
12019d3ddf43SAdrian Chadd 					error = EINVAL;
12029d3ddf43SAdrian Chadd 					break;
12039d3ddf43SAdrian Chadd 				}
1204c2529042SHans Petter Selasky 				flowtype = *(uint32_t *) CMSG_DATA(cm);
12059d3ddf43SAdrian Chadd 				break;
12069d3ddf43SAdrian Chadd 
12079d3ddf43SAdrian Chadd #ifdef	RSS
12089d3ddf43SAdrian Chadd 			case IP_RSSBUCKETID:
12099d3ddf43SAdrian Chadd 				if (cm->cmsg_len != CMSG_LEN(sizeof(uint32_t))) {
12109d3ddf43SAdrian Chadd 					error = EINVAL;
12119d3ddf43SAdrian Chadd 					break;
12129d3ddf43SAdrian Chadd 				}
12139d3ddf43SAdrian Chadd 				/* This is just a placeholder for now */
12149d3ddf43SAdrian Chadd 				break;
12159d3ddf43SAdrian Chadd #endif	/* RSS */
1216c557ae16SIan Dowse 			default:
1217c557ae16SIan Dowse 				error = ENOPROTOOPT;
1218c557ae16SIan Dowse 				break;
1219c557ae16SIan Dowse 			}
1220c557ae16SIan Dowse 			if (error)
1221c557ae16SIan Dowse 				break;
1222c557ae16SIan Dowse 		}
1223c557ae16SIan Dowse 		m_freem(control);
1224c557ae16SIan Dowse 	}
12255c32ea65SRobert Watson 	if (error) {
122684cc0778SGeorge V. Neville-Neil 		if (unlock_inp == UH_WLOCKED)
122784cc0778SGeorge V. Neville-Neil 			INP_WUNLOCK(inp);
122884cc0778SGeorge V. Neville-Neil 		else
12290cfdff24SBjoern A. Zeeb 			INP_RUNLOCK(inp);
12305c32ea65SRobert Watson 		m_freem(m);
12313329b236SRobert Watson 		return (error);
12325c32ea65SRobert Watson 	}
12335c32ea65SRobert Watson 
123443cc0bc1SRobert Watson 	/*
123543cc0bc1SRobert Watson 	 * Depending on whether or not the application has bound or connected
1236ca528788SRobert Watson 	 * the socket, we may have to do varying levels of work.  The optimal
1237ca528788SRobert Watson 	 * case is for a connected UDP socket, as a global lock isn't
1238ca528788SRobert Watson 	 * required at all.
123943cc0bc1SRobert Watson 	 *
124043cc0bc1SRobert Watson 	 * In order to decide which we need, we require stability of the
124143cc0bc1SRobert Watson 	 * inpcb binding, which we ensure by acquiring a read lock on the
124243cc0bc1SRobert Watson 	 * inpcb.  This doesn't strictly follow the lock order, so we play
124343cc0bc1SRobert Watson 	 * the trylock and retry game; note that we may end up with more
124443cc0bc1SRobert Watson 	 * conservative locks than required the second time around, so later
124543cc0bc1SRobert Watson 	 * assertions have to accept that.  Further analysis of the number of
124643cc0bc1SRobert Watson 	 * misses under contention is required.
1247fa046d87SRobert Watson 	 *
1248fa046d87SRobert Watson 	 * XXXRW: Check that hash locking update here is correct.
124943cc0bc1SRobert Watson 	 */
1250e06e816fSKevin Lo 	pr = inp->inp_socket->so_proto->pr_protocol;
1251a86e5c96SBjoern A. Zeeb 	pcbinfo = udp_get_inpcbinfo(pr);
125243cc0bc1SRobert Watson 	sin = (struct sockaddr_in *)addr;
125343cc0bc1SRobert Watson 	if (sin != NULL &&
125443cc0bc1SRobert Watson 	    (inp->inp_laddr.s_addr == INADDR_ANY && inp->inp_lport == 0)) {
1255e06e816fSKevin Lo 		INP_HASH_WLOCK(pcbinfo);
1256fa046d87SRobert Watson 		unlock_udbinfo = UH_WLOCKED;
125743cc0bc1SRobert Watson 	} else if ((sin != NULL && (
125843cc0bc1SRobert Watson 	    (sin->sin_addr.s_addr == INADDR_ANY) ||
125943cc0bc1SRobert Watson 	    (sin->sin_addr.s_addr == INADDR_BROADCAST) ||
126043cc0bc1SRobert Watson 	    (inp->inp_laddr.s_addr == INADDR_ANY) ||
126143cc0bc1SRobert Watson 	    (inp->inp_lport == 0))) ||
126243cc0bc1SRobert Watson 	    (src.sin_family == AF_INET)) {
1263e06e816fSKevin Lo 		INP_HASH_RLOCK(pcbinfo);
1264fa046d87SRobert Watson 		unlock_udbinfo = UH_RLOCKED;
126543cc0bc1SRobert Watson 	} else
1266fa046d87SRobert Watson 		unlock_udbinfo = UH_UNLOCKED;
12675c32ea65SRobert Watson 
12681b7f0384SBruce M Simpson 	/*
12691b7f0384SBruce M Simpson 	 * If the IP_SENDSRCADDR control message was specified, override the
12701b7f0384SBruce M Simpson 	 * source address for this datagram.  Its use is invalidated if the
12711b7f0384SBruce M Simpson 	 * address thus specified is incomplete or clobbers other inpcbs.
12721b7f0384SBruce M Simpson 	 */
127390162a4eSIan Dowse 	laddr = inp->inp_laddr;
127490162a4eSIan Dowse 	lport = inp->inp_lport;
12751b7f0384SBruce M Simpson 	if (src.sin_family == AF_INET) {
1276e06e816fSKevin Lo 		INP_HASH_LOCK_ASSERT(pcbinfo);
12771b7f0384SBruce M Simpson 		if ((lport == 0) ||
12781b7f0384SBruce M Simpson 		    (laddr.s_addr == INADDR_ANY &&
12791b7f0384SBruce M Simpson 		     src.sin_addr.s_addr == INADDR_ANY)) {
1280c557ae16SIan Dowse 			error = EINVAL;
1281c557ae16SIan Dowse 			goto release;
1282c557ae16SIan Dowse 		}
1283c557ae16SIan Dowse 		error = in_pcbbind_setup(inp, (struct sockaddr *)&src,
1284b0330ed9SPawel Jakub Dawidek 		    &laddr.s_addr, &lport, td->td_ucred);
1285c557ae16SIan Dowse 		if (error)
1286c557ae16SIan Dowse 			goto release;
1287c557ae16SIan Dowse 	}
1288c557ae16SIan Dowse 
12893144b7d3SRobert Watson 	/*
12903144b7d3SRobert Watson 	 * If a UDP socket has been connected, then a local address/port will
12913144b7d3SRobert Watson 	 * have been selected and bound.
12923144b7d3SRobert Watson 	 *
129343cc0bc1SRobert Watson 	 * If a UDP socket has not been connected to, then an explicit
12943144b7d3SRobert Watson 	 * destination address must be used, in which case a local
12953144b7d3SRobert Watson 	 * address/port may not have been selected and bound.
12963144b7d3SRobert Watson 	 */
129743cc0bc1SRobert Watson 	if (sin != NULL) {
1298c4d585aeSRobert Watson 		INP_LOCK_ASSERT(inp);
1299df8bae1dSRodney W. Grimes 		if (inp->inp_faddr.s_addr != INADDR_ANY) {
1300df8bae1dSRodney W. Grimes 			error = EISCONN;
1301df8bae1dSRodney W. Grimes 			goto release;
1302df8bae1dSRodney W. Grimes 		}
13033144b7d3SRobert Watson 
13043144b7d3SRobert Watson 		/*
13053144b7d3SRobert Watson 		 * Jail may rewrite the destination address, so let it do
13063144b7d3SRobert Watson 		 * that before we use it.
13073144b7d3SRobert Watson 		 */
1308b89e82ddSJamie Gritton 		error = prison_remote_ip4(td->td_ucred, &sin->sin_addr);
1309b89e82ddSJamie Gritton 		if (error)
1310413628a7SBjoern A. Zeeb 			goto release;
13113144b7d3SRobert Watson 
13123144b7d3SRobert Watson 		/*
131343cc0bc1SRobert Watson 		 * If a local address or port hasn't yet been selected, or if
131443cc0bc1SRobert Watson 		 * the destination address needs to be rewritten due to using
131543cc0bc1SRobert Watson 		 * a special INADDR_ constant, invoke in_pcbconnect_setup()
131643cc0bc1SRobert Watson 		 * to do the heavy lifting.  Once a port is selected, we
131743cc0bc1SRobert Watson 		 * commit the binding back to the socket; we also commit the
131843cc0bc1SRobert Watson 		 * binding of the address if in jail.
131943cc0bc1SRobert Watson 		 *
132043cc0bc1SRobert Watson 		 * If we already have a valid binding and we're not
132143cc0bc1SRobert Watson 		 * requesting a destination address rewrite, use a fast path.
13223144b7d3SRobert Watson 		 */
132343cc0bc1SRobert Watson 		if (inp->inp_laddr.s_addr == INADDR_ANY ||
132443cc0bc1SRobert Watson 		    inp->inp_lport == 0 ||
132543cc0bc1SRobert Watson 		    sin->sin_addr.s_addr == INADDR_ANY ||
132643cc0bc1SRobert Watson 		    sin->sin_addr.s_addr == INADDR_BROADCAST) {
1327e06e816fSKevin Lo 			INP_HASH_LOCK_ASSERT(pcbinfo);
132843cc0bc1SRobert Watson 			error = in_pcbconnect_setup(inp, addr, &laddr.s_addr,
132943cc0bc1SRobert Watson 			    &lport, &faddr.s_addr, &fport, NULL,
133043cc0bc1SRobert Watson 			    td->td_ucred);
133190162a4eSIan Dowse 			if (error)
133290162a4eSIan Dowse 				goto release;
133390162a4eSIan Dowse 
133443cc0bc1SRobert Watson 			/*
133543cc0bc1SRobert Watson 			 * XXXRW: Why not commit the port if the address is
133643cc0bc1SRobert Watson 			 * !INADDR_ANY?
133743cc0bc1SRobert Watson 			 */
133890162a4eSIan Dowse 			/* Commit the local port if newly assigned. */
133990162a4eSIan Dowse 			if (inp->inp_laddr.s_addr == INADDR_ANY &&
134090162a4eSIan Dowse 			    inp->inp_lport == 0) {
1341c4d585aeSRobert Watson 				INP_WLOCK_ASSERT(inp);
1342e06e816fSKevin Lo 				INP_HASH_WLOCK_ASSERT(pcbinfo);
13433a1757b9SGleb Smirnoff 				/*
134443cc0bc1SRobert Watson 				 * Remember addr if jailed, to prevent
134543cc0bc1SRobert Watson 				 * rebinding.
13463a1757b9SGleb Smirnoff 				 */
13470304c731SJamie Gritton 				if (prison_flag(td->td_ucred, PR_IP4))
13483a1757b9SGleb Smirnoff 					inp->inp_laddr = laddr;
134990162a4eSIan Dowse 				inp->inp_lport = lport;
135090162a4eSIan Dowse 				if (in_pcbinshash(inp) != 0) {
135190162a4eSIan Dowse 					inp->inp_lport = 0;
135290162a4eSIan Dowse 					error = EAGAIN;
1353df8bae1dSRodney W. Grimes 					goto release;
1354df8bae1dSRodney W. Grimes 				}
135590162a4eSIan Dowse 				inp->inp_flags |= INP_ANONPORT;
135690162a4eSIan Dowse 			}
1357df8bae1dSRodney W. Grimes 		} else {
135843cc0bc1SRobert Watson 			faddr = sin->sin_addr;
135943cc0bc1SRobert Watson 			fport = sin->sin_port;
136043cc0bc1SRobert Watson 		}
136143cc0bc1SRobert Watson 	} else {
1362c4d585aeSRobert Watson 		INP_LOCK_ASSERT(inp);
136390162a4eSIan Dowse 		faddr = inp->inp_faddr;
136490162a4eSIan Dowse 		fport = inp->inp_fport;
136590162a4eSIan Dowse 		if (faddr.s_addr == INADDR_ANY) {
1366df8bae1dSRodney W. Grimes 			error = ENOTCONN;
1367df8bae1dSRodney W. Grimes 			goto release;
1368df8bae1dSRodney W. Grimes 		}
1369df8bae1dSRodney W. Grimes 	}
1370e6ccd709SRobert Watson 
1371df8bae1dSRodney W. Grimes 	/*
1372e6ccd709SRobert Watson 	 * Calculate data length and get a mbuf for UDP, IP, and possible
1373392e8407SRobert Watson 	 * link-layer headers.  Immediate slide the data pointer back forward
1374392e8407SRobert Watson 	 * since we won't use that space at this layer.
1375df8bae1dSRodney W. Grimes 	 */
1376eb1b1807SGleb Smirnoff 	M_PREPEND(m, sizeof(struct udpiphdr) + max_linkhdr, M_NOWAIT);
1377e6ccd709SRobert Watson 	if (m == NULL) {
1378df8bae1dSRodney W. Grimes 		error = ENOBUFS;
137949b19bfcSBruce M Simpson 		goto release;
1380df8bae1dSRodney W. Grimes 	}
1381e6ccd709SRobert Watson 	m->m_data += max_linkhdr;
1382e6ccd709SRobert Watson 	m->m_len -= max_linkhdr;
1383392e8407SRobert Watson 	m->m_pkthdr.len -= max_linkhdr;
1384df8bae1dSRodney W. Grimes 
1385df8bae1dSRodney W. Grimes 	/*
13863329b236SRobert Watson 	 * Fill in mbuf with extended UDP header and addresses and length put
13873329b236SRobert Watson 	 * into network format.
1388df8bae1dSRodney W. Grimes 	 */
1389df8bae1dSRodney W. Grimes 	ui = mtod(m, struct udpiphdr *);
1390db4f9cc7SJonathan Lemon 	bzero(ui->ui_x1, sizeof(ui->ui_x1));	/* XXX still needed? */
1391e06e816fSKevin Lo 	ui->ui_pr = pr;
139290162a4eSIan Dowse 	ui->ui_src = laddr;
139390162a4eSIan Dowse 	ui->ui_dst = faddr;
139490162a4eSIan Dowse 	ui->ui_sport = lport;
139590162a4eSIan Dowse 	ui->ui_dport = fport;
1396db4f9cc7SJonathan Lemon 	ui->ui_ulen = htons((u_short)len + sizeof(struct udphdr));
1397e06e816fSKevin Lo 	if (pr == IPPROTO_UDPLITE) {
1398e06e816fSKevin Lo 		struct udpcb *up;
1399e06e816fSKevin Lo 		uint16_t plen;
1400e06e816fSKevin Lo 
1401e06e816fSKevin Lo 		up = intoudpcb(inp);
1402e06e816fSKevin Lo 		cscov = up->u_txcslen;
1403e06e816fSKevin Lo 		plen = (u_short)len + sizeof(struct udphdr);
1404e06e816fSKevin Lo 		if (cscov >= plen)
1405e06e816fSKevin Lo 			cscov = 0;
1406e06e816fSKevin Lo 		ui->ui_len = htons(plen);
1407e06e816fSKevin Lo 		ui->ui_ulen = htons(cscov);
1408e06e816fSKevin Lo 		/*
1409e06e816fSKevin Lo 		 * For UDP-Lite, checksum coverage length of zero means
1410e06e816fSKevin Lo 		 * the entire UDPLite packet is covered by the checksum.
1411e06e816fSKevin Lo 		 */
1412e06e816fSKevin Lo 		cscov_partial = (cscov == 0) ? 0 : 1;
1413e06e816fSKevin Lo 	} else
1414e06e816fSKevin Lo 		ui->ui_v = IPVERSION << 4;
1415df8bae1dSRodney W. Grimes 
1416b2828ad2SAndre Oppermann 	/*
1417b2828ad2SAndre Oppermann 	 * Set the Don't Fragment bit in the IP header.
1418b2828ad2SAndre Oppermann 	 */
1419b2828ad2SAndre Oppermann 	if (inp->inp_flags & INP_DONTFRAG) {
1420b2828ad2SAndre Oppermann 		struct ip *ip;
14213329b236SRobert Watson 
1422b2828ad2SAndre Oppermann 		ip = (struct ip *)&ui->ui_i;
14238f134647SGleb Smirnoff 		ip->ip_off |= htons(IP_DF);
1424b2828ad2SAndre Oppermann 	}
1425b2828ad2SAndre Oppermann 
1426b5d47ff5SJohn-Mark Gurney 	ipflags = 0;
1427b5d47ff5SJohn-Mark Gurney 	if (inp->inp_socket->so_options & SO_DONTROUTE)
1428b5d47ff5SJohn-Mark Gurney 		ipflags |= IP_ROUTETOIF;
1429b5d47ff5SJohn-Mark Gurney 	if (inp->inp_socket->so_options & SO_BROADCAST)
1430b5d47ff5SJohn-Mark Gurney 		ipflags |= IP_ALLOWBROADCAST;
14316fbfd582SAndre Oppermann 	if (inp->inp_flags & INP_ONESBCAST)
14328afa2304SBruce M Simpson 		ipflags |= IP_SENDONES;
14338afa2304SBruce M Simpson 
14341175d9d5SRobert Watson #ifdef MAC
14351175d9d5SRobert Watson 	mac_inpcb_create_mbuf(inp, m);
14361175d9d5SRobert Watson #endif
14371175d9d5SRobert Watson 
1438df8bae1dSRodney W. Grimes 	/*
1439db4f9cc7SJonathan Lemon 	 * Set up checksum and output datagram.
1440df8bae1dSRodney W. Grimes 	 */
1441e06e816fSKevin Lo 	ui->ui_sum = 0;
1442a485f139SMichael Tuexen 	if (pr == IPPROTO_UDPLITE) {
1443e06e816fSKevin Lo 		if (inp->inp_flags & INP_ONESBCAST)
1444e06e816fSKevin Lo 			faddr.s_addr = INADDR_BROADCAST;
1445a485f139SMichael Tuexen 		if (cscov_partial) {
1446e06e816fSKevin Lo 			if ((ui->ui_sum = in_cksum(m, sizeof(struct ip) + cscov)) == 0)
1447e06e816fSKevin Lo 				ui->ui_sum = 0xffff;
1448a485f139SMichael Tuexen 		} else {
1449a485f139SMichael Tuexen 			if ((ui->ui_sum = in_cksum(m, sizeof(struct udpiphdr) + len)) == 0)
1450a485f139SMichael Tuexen 				ui->ui_sum = 0xffff;
1451a485f139SMichael Tuexen 		}
1452a485f139SMichael Tuexen 	} else if (V_udp_cksum) {
14536fbfd582SAndre Oppermann 		if (inp->inp_flags & INP_ONESBCAST)
14548a538743SBruce M Simpson 			faddr.s_addr = INADDR_BROADCAST;
14558a538743SBruce M Simpson 		ui->ui_sum = in_pseudo(ui->ui_src.s_addr, faddr.s_addr,
1456e06e816fSKevin Lo 		    htons((u_short)len + sizeof(struct udphdr) + pr));
1457db4f9cc7SJonathan Lemon 		m->m_pkthdr.csum_flags = CSUM_UDP;
1458db4f9cc7SJonathan Lemon 		m->m_pkthdr.csum_data = offsetof(struct udphdr, uh_sum);
1459e06e816fSKevin Lo 	}
14608f134647SGleb Smirnoff 	((struct ip *)ui)->ip_len = htons(sizeof(struct udpiphdr) + len);
1461ca98b82cSDavid Greenman 	((struct ip *)ui)->ip_ttl = inp->inp_ip_ttl;	/* XXX */
1462f584d74bSMichael Tuexen 	((struct ip *)ui)->ip_tos = tos;		/* XXX */
1463026decb8SRobert Watson 	UDPSTAT_INC(udps_opackets);
1464cfa1ca9dSYoshinobu Inoue 
14659d3ddf43SAdrian Chadd 	/*
14669d3ddf43SAdrian Chadd 	 * Setup flowid / RSS information for outbound socket.
14679d3ddf43SAdrian Chadd 	 *
14689d3ddf43SAdrian Chadd 	 * Once the UDP code decides to set a flowid some other way,
14699d3ddf43SAdrian Chadd 	 * this allows the flowid to be overridden by userland.
14709d3ddf43SAdrian Chadd 	 */
1471c2529042SHans Petter Selasky 	if (flowtype != M_HASHTYPE_NONE) {
14729d3ddf43SAdrian Chadd 		m->m_pkthdr.flowid = flowid;
1473c2529042SHans Petter Selasky 		M_HASHTYPE_SET(m, flowtype);
14748ad1a83bSAdrian Chadd #ifdef	RSS
14758ad1a83bSAdrian Chadd 	} else {
14768ad1a83bSAdrian Chadd 		uint32_t hash_val, hash_type;
14778ad1a83bSAdrian Chadd 		/*
14788ad1a83bSAdrian Chadd 		 * Calculate an appropriate RSS hash for UDP and
14798ad1a83bSAdrian Chadd 		 * UDP Lite.
14808ad1a83bSAdrian Chadd 		 *
14818ad1a83bSAdrian Chadd 		 * The called function will take care of figuring out
14828ad1a83bSAdrian Chadd 		 * whether a 2-tuple or 4-tuple hash is required based
14838ad1a83bSAdrian Chadd 		 * on the currently configured scheme.
14848ad1a83bSAdrian Chadd 		 *
14858ad1a83bSAdrian Chadd 		 * Later later on connected socket values should be
14868ad1a83bSAdrian Chadd 		 * cached in the inpcb and reused, rather than constantly
14878ad1a83bSAdrian Chadd 		 * re-calculating it.
14888ad1a83bSAdrian Chadd 		 *
14898ad1a83bSAdrian Chadd 		 * UDP Lite is a different protocol number and will
14908ad1a83bSAdrian Chadd 		 * likely end up being hashed as a 2-tuple until
14918ad1a83bSAdrian Chadd 		 * RSS / NICs grow UDP Lite protocol awareness.
14928ad1a83bSAdrian Chadd 		 */
14938ad1a83bSAdrian Chadd 		if (rss_proto_software_hash_v4(faddr, laddr, fport, lport,
14948ad1a83bSAdrian Chadd 		    pr, &hash_val, &hash_type) == 0) {
14958ad1a83bSAdrian Chadd 			m->m_pkthdr.flowid = hash_val;
14968ad1a83bSAdrian Chadd 			M_HASHTYPE_SET(m, hash_type);
14978ad1a83bSAdrian Chadd 		}
14988ad1a83bSAdrian Chadd #endif
14999d3ddf43SAdrian Chadd 	}
15009d3ddf43SAdrian Chadd 
15019d3ddf43SAdrian Chadd #ifdef	RSS
15028ad1a83bSAdrian Chadd 	/*
15038ad1a83bSAdrian Chadd 	 * Don't override with the inp cached flowid value.
15048ad1a83bSAdrian Chadd 	 *
15058ad1a83bSAdrian Chadd 	 * Depending upon the kind of send being done, the inp
15068ad1a83bSAdrian Chadd 	 * flowid/flowtype values may actually not be appropriate
15078ad1a83bSAdrian Chadd 	 * for this particular socket send.
15088ad1a83bSAdrian Chadd 	 *
15098ad1a83bSAdrian Chadd 	 * We should either leave the flowid at zero (which is what is
15108ad1a83bSAdrian Chadd 	 * currently done) or set it to some software generated
15118ad1a83bSAdrian Chadd 	 * hash value based on the packet contents.
15128ad1a83bSAdrian Chadd 	 */
15139d3ddf43SAdrian Chadd 	ipflags |= IP_NODEFAULTFLOWID;
15149d3ddf43SAdrian Chadd #endif	/* RSS */
15159d3ddf43SAdrian Chadd 
1516fa046d87SRobert Watson 	if (unlock_udbinfo == UH_WLOCKED)
1517e06e816fSKevin Lo 		INP_HASH_WUNLOCK(pcbinfo);
1518fa046d87SRobert Watson 	else if (unlock_udbinfo == UH_RLOCKED)
1519e06e816fSKevin Lo 		INP_HASH_RUNLOCK(pcbinfo);
152057f60867SMark Johnston 	UDP_PROBE(send, NULL, inp, &ui->ui_i, inp, &ui->ui_u);
152184cc0778SGeorge V. Neville-Neil 	error = ip_output(m, inp->inp_options,
152284cc0778SGeorge V. Neville-Neil 	    (unlock_inp == UH_WLOCKED ? &inp->inp_route : NULL), ipflags,
15235d846453SSam Leffler 	    inp->inp_moptions, inp);
152484cc0778SGeorge V. Neville-Neil 	if (unlock_inp == UH_WLOCKED)
15258501a69cSRobert Watson 		INP_WUNLOCK(inp);
1526948d0fc9SRobert Watson 	else
1527948d0fc9SRobert Watson 		INP_RUNLOCK(inp);
1528df8bae1dSRodney W. Grimes 	return (error);
1529df8bae1dSRodney W. Grimes 
1530df8bae1dSRodney W. Grimes release:
1531fa046d87SRobert Watson 	if (unlock_udbinfo == UH_WLOCKED) {
153200b460ffSRick Macklem 		KASSERT(unlock_inp == UH_WLOCKED,
153300b460ffSRick Macklem 		    ("%s: excl udbinfo lock, shared inp lock", __func__));
1534e06e816fSKevin Lo 		INP_HASH_WUNLOCK(pcbinfo);
1535948d0fc9SRobert Watson 		INP_WUNLOCK(inp);
1536fa046d87SRobert Watson 	} else if (unlock_udbinfo == UH_RLOCKED) {
153700b460ffSRick Macklem 		KASSERT(unlock_inp == UH_RLOCKED,
153800b460ffSRick Macklem 		    ("%s: shared udbinfo lock, excl inp lock", __func__));
1539e06e816fSKevin Lo 		INP_HASH_RUNLOCK(pcbinfo);
154043cc0bc1SRobert Watson 		INP_RUNLOCK(inp);
154100b460ffSRick Macklem 	} else if (unlock_inp == UH_WLOCKED)
154200b460ffSRick Macklem 		INP_WUNLOCK(inp);
154300b460ffSRick Macklem 	else
1544948d0fc9SRobert Watson 		INP_RUNLOCK(inp);
1545df8bae1dSRodney W. Grimes 	m_freem(m);
1546df8bae1dSRodney W. Grimes 	return (error);
1547df8bae1dSRodney W. Grimes }
1548df8bae1dSRodney W. Grimes 
1549ac45e92fSRobert Watson static void
1550d0390e05SGarrett Wollman udp_abort(struct socket *so)
1551df8bae1dSRodney W. Grimes {
1552d0390e05SGarrett Wollman 	struct inpcb *inp;
1553e06e816fSKevin Lo 	struct inpcbinfo *pcbinfo;
1554df8bae1dSRodney W. Grimes 
1555a86e5c96SBjoern A. Zeeb 	pcbinfo = udp_get_inpcbinfo(so->so_proto->pr_protocol);
1556d0390e05SGarrett Wollman 	inp = sotoinpcb(so);
155714ba8addSRobert Watson 	KASSERT(inp != NULL, ("udp_abort: inp == NULL"));
15588501a69cSRobert Watson 	INP_WLOCK(inp);
1559a152f8a3SRobert Watson 	if (inp->inp_faddr.s_addr != INADDR_ANY) {
1560e06e816fSKevin Lo 		INP_HASH_WLOCK(pcbinfo);
1561a152f8a3SRobert Watson 		in_pcbdisconnect(inp);
1562a152f8a3SRobert Watson 		inp->inp_laddr.s_addr = INADDR_ANY;
1563e06e816fSKevin Lo 		INP_HASH_WUNLOCK(pcbinfo);
1564d0390e05SGarrett Wollman 		soisdisconnected(so);
1565a152f8a3SRobert Watson 	}
15668501a69cSRobert Watson 	INP_WUNLOCK(inp);
1567df8bae1dSRodney W. Grimes }
1568df8bae1dSRodney W. Grimes 
1569d0390e05SGarrett Wollman static int
1570b40ce416SJulian Elischer udp_attach(struct socket *so, int proto, struct thread *td)
1571d0390e05SGarrett Wollman {
1572d0390e05SGarrett Wollman 	struct inpcb *inp;
1573e06e816fSKevin Lo 	struct inpcbinfo *pcbinfo;
1574277afaffSRobert Watson 	int error;
1575d0390e05SGarrett Wollman 
1576a86e5c96SBjoern A. Zeeb 	pcbinfo = udp_get_inpcbinfo(so->so_proto->pr_protocol);
1577d0390e05SGarrett Wollman 	inp = sotoinpcb(so);
157814ba8addSRobert Watson 	KASSERT(inp == NULL, ("udp_attach: inp != NULL"));
1579cfa1ca9dSYoshinobu Inoue 	error = soreserve(so, udp_sendspace, udp_recvspace);
1580f24618aaSRobert Watson 	if (error)
15813329b236SRobert Watson 		return (error);
1582e06e816fSKevin Lo 	INP_INFO_WLOCK(pcbinfo);
1583e06e816fSKevin Lo 	error = in_pcballoc(so, pcbinfo);
158453b57cd1SSam Leffler 	if (error) {
1585e06e816fSKevin Lo 		INP_INFO_WUNLOCK(pcbinfo);
15863329b236SRobert Watson 		return (error);
158753b57cd1SSam Leffler 	}
1588cfa1ca9dSYoshinobu Inoue 
158968b5629bSRobert Watson 	inp = sotoinpcb(so);
1590cfa1ca9dSYoshinobu Inoue 	inp->inp_vflag |= INP_IPV4;
1591603724d3SBjoern A. Zeeb 	inp->inp_ip_ttl = V_ip_defttl;
15926a9148feSBjoern A. Zeeb 
15936a9148feSBjoern A. Zeeb 	error = udp_newudpcb(inp);
15946a9148feSBjoern A. Zeeb 	if (error) {
15956a9148feSBjoern A. Zeeb 		in_pcbdetach(inp);
15966a9148feSBjoern A. Zeeb 		in_pcbfree(inp);
1597e06e816fSKevin Lo 		INP_INFO_WUNLOCK(pcbinfo);
15986a9148feSBjoern A. Zeeb 		return (error);
15996a9148feSBjoern A. Zeeb 	}
16006a9148feSBjoern A. Zeeb 
1601c7c7ea4bSRandall Stewart 	INP_WUNLOCK(inp);
1602e06e816fSKevin Lo 	INP_INFO_WUNLOCK(pcbinfo);
1603c7c7ea4bSRandall Stewart 	return (0);
1604c7c7ea4bSRandall Stewart }
160579288c11SBjoern A. Zeeb #endif /* INET */
1606c7c7ea4bSRandall Stewart 
1607c7c7ea4bSRandall Stewart int
1608abb901c5SRandall Stewart udp_set_kernel_tunneling(struct socket *so, udp_tun_func_t f, udp_tun_icmp_t i, void *ctx)
1609c7c7ea4bSRandall Stewart {
1610c7c7ea4bSRandall Stewart 	struct inpcb *inp;
16116a9148feSBjoern A. Zeeb 	struct udpcb *up;
1612c7c7ea4bSRandall Stewart 
161368b5629bSRobert Watson 	KASSERT(so->so_type == SOCK_DGRAM,
161468b5629bSRobert Watson 	    ("udp_set_kernel_tunneling: !dgram"));
161568b5629bSRobert Watson 	inp = sotoinpcb(so);
161668b5629bSRobert Watson 	KASSERT(inp != NULL, ("udp_set_kernel_tunneling: inp == NULL"));
1617c7c7ea4bSRandall Stewart 	INP_WLOCK(inp);
16186a9148feSBjoern A. Zeeb 	up = intoudpcb(inp);
1619abb901c5SRandall Stewart 	if ((up->u_tun_func != NULL) ||
1620abb901c5SRandall Stewart 	    (up->u_icmp_func != NULL)) {
1621bbb0e3d9SRandall Stewart 		INP_WUNLOCK(inp);
1622bbb0e3d9SRandall Stewart 		return (EBUSY);
1623bbb0e3d9SRandall Stewart 	}
16246a9148feSBjoern A. Zeeb 	up->u_tun_func = f;
1625abb901c5SRandall Stewart 	up->u_icmp_func = i;
162681d3ec17SBryan Venteicher 	up->u_tun_ctx = ctx;
16278501a69cSRobert Watson 	INP_WUNLOCK(inp);
16283329b236SRobert Watson 	return (0);
1629df8bae1dSRodney W. Grimes }
1630d0390e05SGarrett Wollman 
163179288c11SBjoern A. Zeeb #ifdef INET
1632d0390e05SGarrett Wollman static int
1633b40ce416SJulian Elischer udp_bind(struct socket *so, struct sockaddr *nam, struct thread *td)
1634d0390e05SGarrett Wollman {
1635d0390e05SGarrett Wollman 	struct inpcb *inp;
1636e06e816fSKevin Lo 	struct inpcbinfo *pcbinfo;
1637277afaffSRobert Watson 	int error;
1638d0390e05SGarrett Wollman 
1639a86e5c96SBjoern A. Zeeb 	pcbinfo = udp_get_inpcbinfo(so->so_proto->pr_protocol);
1640d0390e05SGarrett Wollman 	inp = sotoinpcb(so);
164114ba8addSRobert Watson 	KASSERT(inp != NULL, ("udp_bind: inp == NULL"));
16428501a69cSRobert Watson 	INP_WLOCK(inp);
1643e06e816fSKevin Lo 	INP_HASH_WLOCK(pcbinfo);
1644b0330ed9SPawel Jakub Dawidek 	error = in_pcbbind(inp, nam, td->td_ucred);
1645e06e816fSKevin Lo 	INP_HASH_WUNLOCK(pcbinfo);
16468501a69cSRobert Watson 	INP_WUNLOCK(inp);
16473329b236SRobert Watson 	return (error);
1648d0390e05SGarrett Wollman }
1649d0390e05SGarrett Wollman 
1650a152f8a3SRobert Watson static void
1651a152f8a3SRobert Watson udp_close(struct socket *so)
1652a152f8a3SRobert Watson {
1653a152f8a3SRobert Watson 	struct inpcb *inp;
1654e06e816fSKevin Lo 	struct inpcbinfo *pcbinfo;
1655a152f8a3SRobert Watson 
1656a86e5c96SBjoern A. Zeeb 	pcbinfo = udp_get_inpcbinfo(so->so_proto->pr_protocol);
1657a152f8a3SRobert Watson 	inp = sotoinpcb(so);
1658a152f8a3SRobert Watson 	KASSERT(inp != NULL, ("udp_close: inp == NULL"));
16598501a69cSRobert Watson 	INP_WLOCK(inp);
1660a152f8a3SRobert Watson 	if (inp->inp_faddr.s_addr != INADDR_ANY) {
1661e06e816fSKevin Lo 		INP_HASH_WLOCK(pcbinfo);
1662a152f8a3SRobert Watson 		in_pcbdisconnect(inp);
1663a152f8a3SRobert Watson 		inp->inp_laddr.s_addr = INADDR_ANY;
1664e06e816fSKevin Lo 		INP_HASH_WUNLOCK(pcbinfo);
1665a152f8a3SRobert Watson 		soisdisconnected(so);
1666a152f8a3SRobert Watson 	}
16678501a69cSRobert Watson 	INP_WUNLOCK(inp);
1668a152f8a3SRobert Watson }
1669a152f8a3SRobert Watson 
1670d0390e05SGarrett Wollman static int
1671b40ce416SJulian Elischer udp_connect(struct socket *so, struct sockaddr *nam, struct thread *td)
1672d0390e05SGarrett Wollman {
1673d0390e05SGarrett Wollman 	struct inpcb *inp;
1674e06e816fSKevin Lo 	struct inpcbinfo *pcbinfo;
167575c13541SPoul-Henning Kamp 	struct sockaddr_in *sin;
1676e06e816fSKevin Lo 	int error;
1677d0390e05SGarrett Wollman 
1678a86e5c96SBjoern A. Zeeb 	pcbinfo = udp_get_inpcbinfo(so->so_proto->pr_protocol);
1679d0390e05SGarrett Wollman 	inp = sotoinpcb(so);
168014ba8addSRobert Watson 	KASSERT(inp != NULL, ("udp_connect: inp == NULL"));
16818501a69cSRobert Watson 	INP_WLOCK(inp);
1682f76fcf6dSJeffrey Hsu 	if (inp->inp_faddr.s_addr != INADDR_ANY) {
16838501a69cSRobert Watson 		INP_WUNLOCK(inp);
16843329b236SRobert Watson 		return (EISCONN);
1685f76fcf6dSJeffrey Hsu 	}
168675c13541SPoul-Henning Kamp 	sin = (struct sockaddr_in *)nam;
1687b89e82ddSJamie Gritton 	error = prison_remote_ip4(td->td_ucred, &sin->sin_addr);
1688b89e82ddSJamie Gritton 	if (error != 0) {
1689413628a7SBjoern A. Zeeb 		INP_WUNLOCK(inp);
1690b89e82ddSJamie Gritton 		return (error);
1691413628a7SBjoern A. Zeeb 	}
1692e06e816fSKevin Lo 	INP_HASH_WLOCK(pcbinfo);
1693b0330ed9SPawel Jakub Dawidek 	error = in_pcbconnect(inp, nam, td->td_ucred);
1694e06e816fSKevin Lo 	INP_HASH_WUNLOCK(pcbinfo);
16954cc20ab1SSeigo Tanimura 	if (error == 0)
1696df8bae1dSRodney W. Grimes 		soisconnected(so);
16978501a69cSRobert Watson 	INP_WUNLOCK(inp);
16983329b236SRobert Watson 	return (error);
1699df8bae1dSRodney W. Grimes }
1700d0390e05SGarrett Wollman 
1701bc725eafSRobert Watson static void
1702d0390e05SGarrett Wollman udp_detach(struct socket *so)
1703d0390e05SGarrett Wollman {
1704d0390e05SGarrett Wollman 	struct inpcb *inp;
1705e06e816fSKevin Lo 	struct inpcbinfo *pcbinfo;
17066a9148feSBjoern A. Zeeb 	struct udpcb *up;
1707d0390e05SGarrett Wollman 
1708a86e5c96SBjoern A. Zeeb 	pcbinfo = udp_get_inpcbinfo(so->so_proto->pr_protocol);
1709d0390e05SGarrett Wollman 	inp = sotoinpcb(so);
171014ba8addSRobert Watson 	KASSERT(inp != NULL, ("udp_detach: inp == NULL"));
1711a152f8a3SRobert Watson 	KASSERT(inp->inp_faddr.s_addr == INADDR_ANY,
1712a152f8a3SRobert Watson 	    ("udp_detach: not disconnected"));
1713e06e816fSKevin Lo 	INP_INFO_WLOCK(pcbinfo);
17148501a69cSRobert Watson 	INP_WLOCK(inp);
17156a9148feSBjoern A. Zeeb 	up = intoudpcb(inp);
17166a9148feSBjoern A. Zeeb 	KASSERT(up != NULL, ("%s: up == NULL", __func__));
17176a9148feSBjoern A. Zeeb 	inp->inp_ppcb = NULL;
1718d0390e05SGarrett Wollman 	in_pcbdetach(inp);
171914ba8addSRobert Watson 	in_pcbfree(inp);
1720e06e816fSKevin Lo 	INP_INFO_WUNLOCK(pcbinfo);
17216a9148feSBjoern A. Zeeb 	udp_discardcb(up);
1722d0390e05SGarrett Wollman }
1723d0390e05SGarrett Wollman 
1724d0390e05SGarrett Wollman static int
1725d0390e05SGarrett Wollman udp_disconnect(struct socket *so)
1726d0390e05SGarrett Wollman {
1727d0390e05SGarrett Wollman 	struct inpcb *inp;
1728e06e816fSKevin Lo 	struct inpcbinfo *pcbinfo;
1729d0390e05SGarrett Wollman 
1730a86e5c96SBjoern A. Zeeb 	pcbinfo = udp_get_inpcbinfo(so->so_proto->pr_protocol);
1731d0390e05SGarrett Wollman 	inp = sotoinpcb(so);
173214ba8addSRobert Watson 	KASSERT(inp != NULL, ("udp_disconnect: inp == NULL"));
17338501a69cSRobert Watson 	INP_WLOCK(inp);
1734f76fcf6dSJeffrey Hsu 	if (inp->inp_faddr.s_addr == INADDR_ANY) {
17358501a69cSRobert Watson 		INP_WUNLOCK(inp);
17363329b236SRobert Watson 		return (ENOTCONN);
1737f76fcf6dSJeffrey Hsu 	}
1738e06e816fSKevin Lo 	INP_HASH_WLOCK(pcbinfo);
1739df8bae1dSRodney W. Grimes 	in_pcbdisconnect(inp);
1740df8bae1dSRodney W. Grimes 	inp->inp_laddr.s_addr = INADDR_ANY;
1741e06e816fSKevin Lo 	INP_HASH_WUNLOCK(pcbinfo);
1742d45e4f99SMaxim Konovalov 	SOCK_LOCK(so);
1743d45e4f99SMaxim Konovalov 	so->so_state &= ~SS_ISCONNECTED;		/* XXX */
1744d45e4f99SMaxim Konovalov 	SOCK_UNLOCK(so);
17458501a69cSRobert Watson 	INP_WUNLOCK(inp);
17463329b236SRobert Watson 	return (0);
1747df8bae1dSRodney W. Grimes }
1748df8bae1dSRodney W. Grimes 
1749d0390e05SGarrett Wollman static int
175057bf258eSGarrett Wollman udp_send(struct socket *so, int flags, struct mbuf *m, struct sockaddr *addr,
1751b40ce416SJulian Elischer     struct mbuf *control, struct thread *td)
1752d0390e05SGarrett Wollman {
1753d0390e05SGarrett Wollman 	struct inpcb *inp;
1754d0390e05SGarrett Wollman 
1755d0390e05SGarrett Wollman 	inp = sotoinpcb(so);
175614ba8addSRobert Watson 	KASSERT(inp != NULL, ("udp_send: inp == NULL"));
17573329b236SRobert Watson 	return (udp_output(inp, m, addr, control, td));
1758d0390e05SGarrett Wollman }
175979288c11SBjoern A. Zeeb #endif /* INET */
1760d0390e05SGarrett Wollman 
176176429de4SYoshinobu Inoue int
1762d0390e05SGarrett Wollman udp_shutdown(struct socket *so)
1763d0390e05SGarrett Wollman {
1764d0390e05SGarrett Wollman 	struct inpcb *inp;
1765d0390e05SGarrett Wollman 
1766d0390e05SGarrett Wollman 	inp = sotoinpcb(so);
176714ba8addSRobert Watson 	KASSERT(inp != NULL, ("udp_shutdown: inp == NULL"));
17688501a69cSRobert Watson 	INP_WLOCK(inp);
1769d0390e05SGarrett Wollman 	socantsendmore(so);
17708501a69cSRobert Watson 	INP_WUNLOCK(inp);
17713329b236SRobert Watson 	return (0);
1772d0390e05SGarrett Wollman }
1773d0390e05SGarrett Wollman 
177479288c11SBjoern A. Zeeb #ifdef INET
1775d0390e05SGarrett Wollman struct pr_usrreqs udp_usrreqs = {
1776756d52a1SPoul-Henning Kamp 	.pru_abort =		udp_abort,
1777756d52a1SPoul-Henning Kamp 	.pru_attach =		udp_attach,
1778756d52a1SPoul-Henning Kamp 	.pru_bind =		udp_bind,
1779756d52a1SPoul-Henning Kamp 	.pru_connect =		udp_connect,
1780756d52a1SPoul-Henning Kamp 	.pru_control =		in_control,
1781756d52a1SPoul-Henning Kamp 	.pru_detach =		udp_detach,
1782756d52a1SPoul-Henning Kamp 	.pru_disconnect =	udp_disconnect,
178354d642bbSRobert Watson 	.pru_peeraddr =		in_getpeeraddr,
1784756d52a1SPoul-Henning Kamp 	.pru_send =		udp_send,
17855df3e839SRobert Watson 	.pru_soreceive =	soreceive_dgram,
178659b8854eSRobert Watson 	.pru_sosend =		sosend_dgram,
1787756d52a1SPoul-Henning Kamp 	.pru_shutdown =		udp_shutdown,
178854d642bbSRobert Watson 	.pru_sockaddr =		in_getsockaddr,
1789a152f8a3SRobert Watson 	.pru_sosetlabel =	in_pcbsosetlabel,
1790a152f8a3SRobert Watson 	.pru_close =		udp_close,
1791d0390e05SGarrett Wollman };
179279288c11SBjoern A. Zeeb #endif /* INET */
1793