Home | History | Annotate | Line # | Download | only in kern
uipc_socket.c revision 1.200.2.1
      1  1.200.2.1  uebayasi /*	$NetBSD: uipc_socket.c,v 1.200.2.1 2010/10/22 07:22:30 uebayasi Exp $	*/
      2       1.64   thorpej 
      3       1.64   thorpej /*-
      4      1.188        ad  * Copyright (c) 2002, 2007, 2008, 2009 The NetBSD Foundation, Inc.
      5       1.64   thorpej  * All rights reserved.
      6       1.64   thorpej  *
      7       1.64   thorpej  * This code is derived from software contributed to The NetBSD Foundation
      8      1.188        ad  * by Jason R. Thorpe of Wasabi Systems, Inc, and by Andrew Doran.
      9       1.64   thorpej  *
     10       1.64   thorpej  * Redistribution and use in source and binary forms, with or without
     11       1.64   thorpej  * modification, are permitted provided that the following conditions
     12       1.64   thorpej  * are met:
     13       1.64   thorpej  * 1. Redistributions of source code must retain the above copyright
     14       1.64   thorpej  *    notice, this list of conditions and the following disclaimer.
     15       1.64   thorpej  * 2. Redistributions in binary form must reproduce the above copyright
     16       1.64   thorpej  *    notice, this list of conditions and the following disclaimer in the
     17       1.64   thorpej  *    documentation and/or other materials provided with the distribution.
     18       1.64   thorpej  *
     19       1.64   thorpej  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
     20       1.64   thorpej  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
     21       1.64   thorpej  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
     22       1.64   thorpej  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
     23       1.64   thorpej  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
     24       1.64   thorpej  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
     25       1.64   thorpej  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
     26       1.64   thorpej  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
     27       1.64   thorpej  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
     28       1.64   thorpej  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
     29       1.64   thorpej  * POSSIBILITY OF SUCH DAMAGE.
     30       1.64   thorpej  */
     31       1.16       cgd 
     32        1.1       cgd /*
     33      1.159        ad  * Copyright (c) 2004 The FreeBSD Foundation
     34      1.159        ad  * Copyright (c) 2004 Robert Watson
     35       1.15   mycroft  * Copyright (c) 1982, 1986, 1988, 1990, 1993
     36       1.15   mycroft  *	The Regents of the University of California.  All rights reserved.
     37        1.1       cgd  *
     38        1.1       cgd  * Redistribution and use in source and binary forms, with or without
     39        1.1       cgd  * modification, are permitted provided that the following conditions
     40        1.1       cgd  * are met:
     41        1.1       cgd  * 1. Redistributions of source code must retain the above copyright
     42        1.1       cgd  *    notice, this list of conditions and the following disclaimer.
     43        1.1       cgd  * 2. Redistributions in binary form must reproduce the above copyright
     44        1.1       cgd  *    notice, this list of conditions and the following disclaimer in the
     45        1.1       cgd  *    documentation and/or other materials provided with the distribution.
     46       1.85       agc  * 3. Neither the name of the University nor the names of its contributors
     47        1.1       cgd  *    may be used to endorse or promote products derived from this software
     48        1.1       cgd  *    without specific prior written permission.
     49        1.1       cgd  *
     50        1.1       cgd  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
     51        1.1       cgd  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
     52        1.1       cgd  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
     53        1.1       cgd  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
     54        1.1       cgd  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
     55        1.1       cgd  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
     56        1.1       cgd  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
     57        1.1       cgd  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
     58        1.1       cgd  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
     59        1.1       cgd  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
     60        1.1       cgd  * SUCH DAMAGE.
     61        1.1       cgd  *
     62       1.32      fvdl  *	@(#)uipc_socket.c	8.6 (Berkeley) 5/2/95
     63        1.1       cgd  */
     64       1.59     lukem 
     65       1.59     lukem #include <sys/cdefs.h>
     66  1.200.2.1  uebayasi __KERNEL_RCSID(0, "$NetBSD: uipc_socket.c,v 1.200.2.1 2010/10/22 07:22:30 uebayasi Exp $");
     67       1.64   thorpej 
     68      1.179  christos #include "opt_compat_netbsd.h"
     69       1.64   thorpej #include "opt_sock_counters.h"
     70       1.64   thorpej #include "opt_sosend_loan.h"
     71       1.81    martin #include "opt_mbuftrace.h"
     72       1.84     ragge #include "opt_somaxkva.h"
     73      1.167        ad #include "opt_multiprocessor.h"	/* XXX */
     74        1.1       cgd 
     75        1.9   mycroft #include <sys/param.h>
     76        1.9   mycroft #include <sys/systm.h>
     77        1.9   mycroft #include <sys/proc.h>
     78        1.9   mycroft #include <sys/file.h>
     79      1.142    dyoung #include <sys/filedesc.h>
     80      1.173    plunky #include <sys/kmem.h>
     81        1.9   mycroft #include <sys/mbuf.h>
     82        1.9   mycroft #include <sys/domain.h>
     83        1.9   mycroft #include <sys/kernel.h>
     84        1.9   mycroft #include <sys/protosw.h>
     85        1.9   mycroft #include <sys/socket.h>
     86        1.9   mycroft #include <sys/socketvar.h>
     87       1.21  christos #include <sys/signalvar.h>
     88        1.9   mycroft #include <sys/resourcevar.h>
     89      1.174     pooka #include <sys/uidinfo.h>
     90       1.72  jdolecek #include <sys/event.h>
     91       1.89  christos #include <sys/poll.h>
     92      1.118      elad #include <sys/kauth.h>
     93      1.136        ad #include <sys/mutex.h>
     94      1.136        ad #include <sys/condvar.h>
     95       1.37   thorpej 
     96      1.179  christos #ifdef COMPAT_50
     97      1.179  christos #include <compat/sys/time.h>
     98      1.184  christos #include <compat/sys/socket.h>
     99      1.179  christos #endif
    100      1.179  christos 
    101       1.64   thorpej #include <uvm/uvm.h>
    102       1.64   thorpej 
    103       1.77   thorpej MALLOC_DEFINE(M_SOOPTS, "soopts", "socket options");
    104       1.77   thorpej MALLOC_DEFINE(M_SONAME, "soname", "socket name");
    105       1.37   thorpej 
    106      1.142    dyoung extern const struct fileops socketops;
    107      1.142    dyoung 
    108       1.54     lukem extern int	somaxconn;			/* patchable (XXX sysctl) */
    109       1.54     lukem int		somaxconn = SOMAXCONN;
    110      1.160        ad kmutex_t	*softnet_lock;
    111       1.49  jonathan 
    112       1.64   thorpej #ifdef SOSEND_COUNTERS
    113       1.64   thorpej #include <sys/device.h>
    114       1.64   thorpej 
    115      1.113   thorpej static struct evcnt sosend_loan_big = EVCNT_INITIALIZER(EVCNT_TYPE_MISC,
    116       1.64   thorpej     NULL, "sosend", "loan big");
    117      1.113   thorpej static struct evcnt sosend_copy_big = EVCNT_INITIALIZER(EVCNT_TYPE_MISC,
    118       1.64   thorpej     NULL, "sosend", "copy big");
    119      1.113   thorpej static struct evcnt sosend_copy_small = EVCNT_INITIALIZER(EVCNT_TYPE_MISC,
    120       1.64   thorpej     NULL, "sosend", "copy small");
    121      1.113   thorpej static struct evcnt sosend_kvalimit = EVCNT_INITIALIZER(EVCNT_TYPE_MISC,
    122       1.64   thorpej     NULL, "sosend", "kva limit");
    123       1.64   thorpej 
    124       1.64   thorpej #define	SOSEND_COUNTER_INCR(ev)		(ev)->ev_count++
    125       1.64   thorpej 
    126      1.101      matt EVCNT_ATTACH_STATIC(sosend_loan_big);
    127      1.101      matt EVCNT_ATTACH_STATIC(sosend_copy_big);
    128      1.101      matt EVCNT_ATTACH_STATIC(sosend_copy_small);
    129      1.101      matt EVCNT_ATTACH_STATIC(sosend_kvalimit);
    130       1.64   thorpej #else
    131       1.64   thorpej 
    132       1.64   thorpej #define	SOSEND_COUNTER_INCR(ev)		/* nothing */
    133       1.64   thorpej 
    134       1.64   thorpej #endif /* SOSEND_COUNTERS */
    135       1.64   thorpej 
    136      1.119      yamt static struct callback_entry sokva_reclaimerentry;
    137        1.1       cgd 
    138      1.167        ad #if defined(SOSEND_NO_LOAN) || defined(MULTIPROCESSOR)
    139      1.121      yamt int sock_loan_thresh = -1;
    140       1.71   thorpej #else
    141      1.121      yamt int sock_loan_thresh = 4096;
    142       1.65   thorpej #endif
    143       1.64   thorpej 
    144      1.136        ad static kmutex_t so_pendfree_lock;
    145      1.113   thorpej static struct mbuf *so_pendfree;
    146       1.64   thorpej 
    147       1.84     ragge #ifndef SOMAXKVA
    148       1.84     ragge #define	SOMAXKVA (16 * 1024 * 1024)
    149       1.84     ragge #endif
    150       1.84     ragge int somaxkva = SOMAXKVA;
    151      1.113   thorpej static int socurkva;
    152      1.136        ad static kcondvar_t socurkva_cv;
    153       1.64   thorpej 
    154      1.191      elad static kauth_listener_t socket_listener;
    155      1.191      elad 
    156       1.64   thorpej #define	SOCK_LOAN_CHUNK		65536
    157       1.64   thorpej 
    158      1.117      yamt static size_t sodopendfree(void);
    159      1.117      yamt static size_t sodopendfreel(void);
    160       1.93      yamt 
    161      1.178     pooka static void sysctl_kern_somaxkva_setup(void);
    162      1.178     pooka static struct sysctllog *socket_sysctllog;
    163      1.178     pooka 
    164      1.113   thorpej static vsize_t
    165      1.129      yamt sokvareserve(struct socket *so, vsize_t len)
    166       1.80      yamt {
    167       1.98  christos 	int error;
    168       1.80      yamt 
    169      1.136        ad 	mutex_enter(&so_pendfree_lock);
    170       1.80      yamt 	while (socurkva + len > somaxkva) {
    171       1.93      yamt 		size_t freed;
    172       1.93      yamt 
    173       1.93      yamt 		/*
    174       1.93      yamt 		 * try to do pendfree.
    175       1.93      yamt 		 */
    176       1.93      yamt 
    177      1.117      yamt 		freed = sodopendfreel();
    178       1.93      yamt 
    179       1.93      yamt 		/*
    180       1.93      yamt 		 * if some kva was freed, try again.
    181       1.93      yamt 		 */
    182       1.93      yamt 
    183       1.93      yamt 		if (freed)
    184       1.80      yamt 			continue;
    185       1.93      yamt 
    186       1.80      yamt 		SOSEND_COUNTER_INCR(&sosend_kvalimit);
    187      1.136        ad 		error = cv_wait_sig(&socurkva_cv, &so_pendfree_lock);
    188       1.98  christos 		if (error) {
    189       1.98  christos 			len = 0;
    190       1.98  christos 			break;
    191       1.98  christos 		}
    192       1.80      yamt 	}
    193       1.93      yamt 	socurkva += len;
    194      1.136        ad 	mutex_exit(&so_pendfree_lock);
    195       1.98  christos 	return len;
    196       1.95      yamt }
    197       1.95      yamt 
    198      1.113   thorpej static void
    199       1.95      yamt sokvaunreserve(vsize_t len)
    200       1.95      yamt {
    201       1.95      yamt 
    202      1.136        ad 	mutex_enter(&so_pendfree_lock);
    203       1.95      yamt 	socurkva -= len;
    204      1.136        ad 	cv_broadcast(&socurkva_cv);
    205      1.136        ad 	mutex_exit(&so_pendfree_lock);
    206       1.95      yamt }
    207       1.95      yamt 
    208       1.95      yamt /*
    209       1.95      yamt  * sokvaalloc: allocate kva for loan.
    210       1.95      yamt  */
    211       1.95      yamt 
    212       1.95      yamt vaddr_t
    213       1.95      yamt sokvaalloc(vsize_t len, struct socket *so)
    214       1.95      yamt {
    215       1.95      yamt 	vaddr_t lva;
    216       1.95      yamt 
    217       1.95      yamt 	/*
    218       1.95      yamt 	 * reserve kva.
    219       1.95      yamt 	 */
    220       1.95      yamt 
    221       1.98  christos 	if (sokvareserve(so, len) == 0)
    222       1.98  christos 		return 0;
    223       1.93      yamt 
    224       1.93      yamt 	/*
    225       1.93      yamt 	 * allocate kva.
    226       1.93      yamt 	 */
    227       1.80      yamt 
    228      1.109      yamt 	lva = uvm_km_alloc(kernel_map, len, 0, UVM_KMF_VAONLY | UVM_KMF_WAITVA);
    229       1.95      yamt 	if (lva == 0) {
    230       1.95      yamt 		sokvaunreserve(len);
    231       1.80      yamt 		return (0);
    232       1.95      yamt 	}
    233       1.80      yamt 
    234       1.80      yamt 	return lva;
    235       1.80      yamt }
    236       1.80      yamt 
    237       1.93      yamt /*
    238       1.93      yamt  * sokvafree: free kva for loan.
    239       1.93      yamt  */
    240       1.93      yamt 
    241       1.80      yamt void
    242       1.80      yamt sokvafree(vaddr_t sva, vsize_t len)
    243       1.80      yamt {
    244       1.93      yamt 
    245       1.93      yamt 	/*
    246       1.93      yamt 	 * free kva.
    247       1.93      yamt 	 */
    248       1.80      yamt 
    249      1.109      yamt 	uvm_km_free(kernel_map, sva, len, UVM_KMF_VAONLY);
    250       1.93      yamt 
    251       1.93      yamt 	/*
    252       1.93      yamt 	 * unreserve kva.
    253       1.93      yamt 	 */
    254       1.93      yamt 
    255       1.95      yamt 	sokvaunreserve(len);
    256       1.80      yamt }
    257       1.80      yamt 
    258       1.64   thorpej static void
    259      1.134  christos sodoloanfree(struct vm_page **pgs, void *buf, size_t size)
    260       1.64   thorpej {
    261      1.156      yamt 	vaddr_t sva, eva;
    262       1.64   thorpej 	vsize_t len;
    263      1.156      yamt 	int npgs;
    264      1.156      yamt 
    265      1.156      yamt 	KASSERT(pgs != NULL);
    266       1.64   thorpej 
    267       1.64   thorpej 	eva = round_page((vaddr_t) buf + size);
    268       1.64   thorpej 	sva = trunc_page((vaddr_t) buf);
    269       1.64   thorpej 	len = eva - sva;
    270       1.64   thorpej 	npgs = len >> PAGE_SHIFT;
    271       1.64   thorpej 
    272       1.64   thorpej 	pmap_kremove(sva, len);
    273       1.64   thorpej 	pmap_update(pmap_kernel());
    274       1.64   thorpej 	uvm_unloan(pgs, npgs, UVM_LOAN_TOPAGE);
    275       1.80      yamt 	sokvafree(sva, len);
    276       1.64   thorpej }
    277       1.64   thorpej 
    278       1.64   thorpej static size_t
    279      1.152      matt sodopendfree(void)
    280       1.64   thorpej {
    281       1.93      yamt 	size_t rv;
    282       1.64   thorpej 
    283      1.160        ad 	if (__predict_true(so_pendfree == NULL))
    284      1.160        ad 		return 0;
    285      1.160        ad 
    286      1.136        ad 	mutex_enter(&so_pendfree_lock);
    287      1.117      yamt 	rv = sodopendfreel();
    288      1.136        ad 	mutex_exit(&so_pendfree_lock);
    289       1.93      yamt 
    290       1.93      yamt 	return rv;
    291       1.93      yamt }
    292       1.93      yamt 
    293       1.93      yamt /*
    294       1.93      yamt  * sodopendfreel: free mbufs on "pendfree" list.
    295      1.136        ad  * unlock and relock so_pendfree_lock when freeing mbufs.
    296       1.93      yamt  *
    297      1.136        ad  * => called with so_pendfree_lock held.
    298       1.93      yamt  */
    299       1.93      yamt 
    300       1.93      yamt static size_t
    301      1.152      matt sodopendfreel(void)
    302       1.93      yamt {
    303      1.137        ad 	struct mbuf *m, *next;
    304       1.93      yamt 	size_t rv = 0;
    305       1.93      yamt 
    306      1.136        ad 	KASSERT(mutex_owned(&so_pendfree_lock));
    307       1.64   thorpej 
    308      1.137        ad 	while (so_pendfree != NULL) {
    309       1.64   thorpej 		m = so_pendfree;
    310       1.93      yamt 		so_pendfree = NULL;
    311      1.136        ad 		mutex_exit(&so_pendfree_lock);
    312       1.93      yamt 
    313       1.93      yamt 		for (; m != NULL; m = next) {
    314       1.93      yamt 			next = m->m_next;
    315      1.156      yamt 			KASSERT((~m->m_flags & (M_EXT|M_EXT_PAGES)) == 0);
    316      1.156      yamt 			KASSERT(m->m_ext.ext_refcnt == 0);
    317       1.93      yamt 
    318       1.93      yamt 			rv += m->m_ext.ext_size;
    319      1.156      yamt 			sodoloanfree(m->m_ext.ext_pgs, m->m_ext.ext_buf,
    320       1.93      yamt 			    m->m_ext.ext_size);
    321      1.145        ad 			pool_cache_put(mb_cache, m);
    322       1.93      yamt 		}
    323       1.64   thorpej 
    324      1.136        ad 		mutex_enter(&so_pendfree_lock);
    325       1.64   thorpej 	}
    326       1.64   thorpej 
    327       1.64   thorpej 	return (rv);
    328       1.64   thorpej }
    329       1.64   thorpej 
    330       1.80      yamt void
    331      1.134  christos soloanfree(struct mbuf *m, void *buf, size_t size, void *arg)
    332       1.64   thorpej {
    333       1.64   thorpej 
    334      1.156      yamt 	KASSERT(m != NULL);
    335       1.64   thorpej 
    336       1.93      yamt 	/*
    337       1.93      yamt 	 * postpone freeing mbuf.
    338       1.93      yamt 	 *
    339       1.93      yamt 	 * we can't do it in interrupt context
    340       1.93      yamt 	 * because we need to put kva back to kernel_map.
    341       1.93      yamt 	 */
    342       1.93      yamt 
    343      1.136        ad 	mutex_enter(&so_pendfree_lock);
    344       1.92      yamt 	m->m_next = so_pendfree;
    345       1.92      yamt 	so_pendfree = m;
    346      1.136        ad 	cv_broadcast(&socurkva_cv);
    347      1.136        ad 	mutex_exit(&so_pendfree_lock);
    348       1.64   thorpej }
    349       1.64   thorpej 
    350       1.64   thorpej static long
    351       1.64   thorpej sosend_loan(struct socket *so, struct uio *uio, struct mbuf *m, long space)
    352       1.64   thorpej {
    353       1.64   thorpej 	struct iovec *iov = uio->uio_iov;
    354       1.64   thorpej 	vaddr_t sva, eva;
    355       1.64   thorpej 	vsize_t len;
    356      1.156      yamt 	vaddr_t lva;
    357      1.156      yamt 	int npgs, error;
    358      1.156      yamt 	vaddr_t va;
    359      1.156      yamt 	int i;
    360       1.64   thorpej 
    361      1.116      yamt 	if (VMSPACE_IS_KERNEL_P(uio->uio_vmspace))
    362       1.64   thorpej 		return (0);
    363       1.64   thorpej 
    364       1.64   thorpej 	if (iov->iov_len < (size_t) space)
    365       1.64   thorpej 		space = iov->iov_len;
    366       1.64   thorpej 	if (space > SOCK_LOAN_CHUNK)
    367       1.64   thorpej 		space = SOCK_LOAN_CHUNK;
    368       1.64   thorpej 
    369       1.64   thorpej 	eva = round_page((vaddr_t) iov->iov_base + space);
    370       1.64   thorpej 	sva = trunc_page((vaddr_t) iov->iov_base);
    371       1.64   thorpej 	len = eva - sva;
    372       1.64   thorpej 	npgs = len >> PAGE_SHIFT;
    373       1.64   thorpej 
    374       1.79   thorpej 	KASSERT(npgs <= M_EXT_MAXPAGES);
    375       1.79   thorpej 
    376       1.80      yamt 	lva = sokvaalloc(len, so);
    377       1.64   thorpej 	if (lva == 0)
    378       1.80      yamt 		return 0;
    379       1.64   thorpej 
    380      1.116      yamt 	error = uvm_loan(&uio->uio_vmspace->vm_map, sva, len,
    381       1.79   thorpej 	    m->m_ext.ext_pgs, UVM_LOAN_TOPAGE);
    382       1.64   thorpej 	if (error) {
    383       1.80      yamt 		sokvafree(lva, len);
    384       1.64   thorpej 		return (0);
    385       1.64   thorpej 	}
    386       1.64   thorpej 
    387       1.64   thorpej 	for (i = 0, va = lva; i < npgs; i++, va += PAGE_SIZE)
    388       1.79   thorpej 		pmap_kenter_pa(va, VM_PAGE_TO_PHYS(m->m_ext.ext_pgs[i]),
    389      1.194    cegger 		    VM_PROT_READ, 0);
    390       1.64   thorpej 	pmap_update(pmap_kernel());
    391       1.64   thorpej 
    392       1.64   thorpej 	lva += (vaddr_t) iov->iov_base & PAGE_MASK;
    393       1.64   thorpej 
    394      1.134  christos 	MEXTADD(m, (void *) lva, space, M_MBUF, soloanfree, so);
    395       1.79   thorpej 	m->m_flags |= M_EXT_PAGES | M_EXT_ROMAP;
    396       1.64   thorpej 
    397       1.64   thorpej 	uio->uio_resid -= space;
    398       1.64   thorpej 	/* uio_offset not updated, not set/used for write(2) */
    399      1.134  christos 	uio->uio_iov->iov_base = (char *)uio->uio_iov->iov_base + space;
    400       1.64   thorpej 	uio->uio_iov->iov_len -= space;
    401       1.64   thorpej 	if (uio->uio_iov->iov_len == 0) {
    402       1.64   thorpej 		uio->uio_iov++;
    403       1.64   thorpej 		uio->uio_iovcnt--;
    404       1.64   thorpej 	}
    405       1.64   thorpej 
    406       1.64   thorpej 	return (space);
    407       1.64   thorpej }
    408       1.64   thorpej 
    409      1.119      yamt static int
    410      1.129      yamt sokva_reclaim_callback(struct callback_entry *ce, void *obj, void *arg)
    411      1.119      yamt {
    412      1.119      yamt 
    413      1.119      yamt 	KASSERT(ce == &sokva_reclaimerentry);
    414      1.119      yamt 	KASSERT(obj == NULL);
    415      1.119      yamt 
    416      1.119      yamt 	sodopendfree();
    417      1.119      yamt 	if (!vm_map_starved_p(kernel_map)) {
    418      1.119      yamt 		return CALLBACK_CHAIN_ABORT;
    419      1.119      yamt 	}
    420      1.119      yamt 	return CALLBACK_CHAIN_CONTINUE;
    421      1.119      yamt }
    422      1.119      yamt 
    423      1.142    dyoung struct mbuf *
    424      1.147    dyoung getsombuf(struct socket *so, int type)
    425      1.142    dyoung {
    426      1.142    dyoung 	struct mbuf *m;
    427      1.142    dyoung 
    428      1.147    dyoung 	m = m_get(M_WAIT, type);
    429      1.142    dyoung 	MCLAIM(m, so->so_mowner);
    430      1.142    dyoung 	return m;
    431      1.142    dyoung }
    432      1.142    dyoung 
    433      1.191      elad static int
    434      1.191      elad socket_listener_cb(kauth_cred_t cred, kauth_action_t action, void *cookie,
    435      1.191      elad     void *arg0, void *arg1, void *arg2, void *arg3)
    436      1.191      elad {
    437      1.191      elad 	int result;
    438      1.191      elad 	enum kauth_network_req req;
    439      1.191      elad 
    440      1.191      elad 	result = KAUTH_RESULT_DEFER;
    441      1.191      elad 	req = (enum kauth_network_req)arg0;
    442      1.191      elad 
    443      1.193      elad 	if ((action != KAUTH_NETWORK_SOCKET) &&
    444      1.193      elad 	    (action != KAUTH_NETWORK_BIND))
    445      1.191      elad 		return result;
    446      1.191      elad 
    447      1.191      elad 	switch (req) {
    448      1.193      elad 	case KAUTH_REQ_NETWORK_BIND_PORT:
    449      1.193      elad 		result = KAUTH_RESULT_ALLOW;
    450      1.193      elad 		break;
    451      1.193      elad 
    452      1.191      elad 	case KAUTH_REQ_NETWORK_SOCKET_DROP: {
    453      1.191      elad 		/* Normal users can only drop their own connections. */
    454      1.191      elad 		struct socket *so = (struct socket *)arg1;
    455      1.191      elad 
    456      1.199      elad 		if (proc_uidmatch(cred, so->so_cred))
    457      1.191      elad 			result = KAUTH_RESULT_ALLOW;
    458      1.191      elad 
    459      1.191      elad 		break;
    460      1.191      elad 		}
    461      1.191      elad 
    462      1.191      elad 	case KAUTH_REQ_NETWORK_SOCKET_OPEN:
    463      1.191      elad 		/* We allow "raw" routing/bluetooth sockets to anyone. */
    464      1.191      elad 		if ((u_long)arg1 == PF_ROUTE || (u_long)arg1 == PF_BLUETOOTH)
    465      1.191      elad 			result = KAUTH_RESULT_ALLOW;
    466      1.191      elad 		else {
    467      1.191      elad 			/* Privileged, let secmodel handle this. */
    468      1.191      elad 			if ((u_long)arg2 == SOCK_RAW)
    469      1.191      elad 				break;
    470      1.191      elad 		}
    471      1.191      elad 
    472      1.191      elad 		result = KAUTH_RESULT_ALLOW;
    473      1.191      elad 
    474      1.191      elad 		break;
    475      1.191      elad 
    476      1.192      elad 	case KAUTH_REQ_NETWORK_SOCKET_CANSEE:
    477      1.192      elad 		result = KAUTH_RESULT_ALLOW;
    478      1.192      elad 
    479      1.192      elad 		break;
    480      1.192      elad 
    481      1.191      elad 	default:
    482      1.191      elad 		break;
    483      1.191      elad 	}
    484      1.191      elad 
    485      1.191      elad 	return result;
    486      1.191      elad }
    487      1.191      elad 
    488      1.119      yamt void
    489      1.119      yamt soinit(void)
    490      1.119      yamt {
    491      1.119      yamt 
    492      1.178     pooka 	sysctl_kern_somaxkva_setup();
    493      1.178     pooka 
    494      1.148        ad 	mutex_init(&so_pendfree_lock, MUTEX_DEFAULT, IPL_VM);
    495      1.160        ad 	softnet_lock = mutex_obj_alloc(MUTEX_DEFAULT, IPL_NONE);
    496      1.136        ad 	cv_init(&socurkva_cv, "sokva");
    497      1.166        ad 	soinit2();
    498      1.136        ad 
    499      1.119      yamt 	/* Set the initial adjusted socket buffer size. */
    500      1.119      yamt 	if (sb_max_set(sb_max))
    501      1.119      yamt 		panic("bad initial sb_max value: %lu", sb_max);
    502      1.119      yamt 
    503      1.119      yamt 	callback_register(&vm_map_to_kernel(kernel_map)->vmk_reclaim_callback,
    504      1.119      yamt 	    &sokva_reclaimerentry, NULL, sokva_reclaim_callback);
    505      1.191      elad 
    506      1.191      elad 	socket_listener = kauth_listen_scope(KAUTH_SCOPE_NETWORK,
    507      1.191      elad 	    socket_listener_cb, NULL);
    508      1.119      yamt }
    509      1.119      yamt 
    510        1.1       cgd /*
    511        1.1       cgd  * Socket operation routines.
    512        1.1       cgd  * These routines are called by the routines in
    513        1.1       cgd  * sys_socket.c or from a system process, and
    514        1.1       cgd  * implement the semantics of socket operations by
    515        1.1       cgd  * switching out to the protocol specific routines.
    516        1.1       cgd  */
    517        1.1       cgd /*ARGSUSED*/
    518        1.3    andrew int
    519      1.160        ad socreate(int dom, struct socket **aso, int type, int proto, struct lwp *l,
    520      1.160        ad 	 struct socket *lockso)
    521        1.1       cgd {
    522       1.99      matt 	const struct protosw	*prp;
    523       1.54     lukem 	struct socket	*so;
    524      1.115      yamt 	uid_t		uid;
    525      1.160        ad 	int		error;
    526      1.160        ad 	kmutex_t	*lock;
    527        1.1       cgd 
    528      1.132      elad 	error = kauth_authorize_network(l->l_cred, KAUTH_NETWORK_SOCKET,
    529      1.132      elad 	    KAUTH_REQ_NETWORK_SOCKET_OPEN, KAUTH_ARG(dom), KAUTH_ARG(type),
    530      1.132      elad 	    KAUTH_ARG(proto));
    531      1.140    dyoung 	if (error != 0)
    532      1.140    dyoung 		return error;
    533      1.127      elad 
    534        1.1       cgd 	if (proto)
    535        1.1       cgd 		prp = pffindproto(dom, proto, type);
    536        1.1       cgd 	else
    537        1.1       cgd 		prp = pffindtype(dom, type);
    538      1.140    dyoung 	if (prp == NULL) {
    539      1.120  ginsbach 		/* no support for domain */
    540      1.120  ginsbach 		if (pffinddomain(dom) == 0)
    541      1.140    dyoung 			return EAFNOSUPPORT;
    542      1.120  ginsbach 		/* no support for socket type */
    543      1.120  ginsbach 		if (proto == 0 && type != 0)
    544      1.140    dyoung 			return EPROTOTYPE;
    545      1.140    dyoung 		return EPROTONOSUPPORT;
    546      1.120  ginsbach 	}
    547      1.140    dyoung 	if (prp->pr_usrreq == NULL)
    548      1.140    dyoung 		return EPROTONOSUPPORT;
    549        1.1       cgd 	if (prp->pr_type != type)
    550      1.140    dyoung 		return EPROTOTYPE;
    551      1.160        ad 
    552      1.160        ad 	so = soget(true);
    553        1.1       cgd 	so->so_type = type;
    554        1.1       cgd 	so->so_proto = prp;
    555       1.33      matt 	so->so_send = sosend;
    556       1.33      matt 	so->so_receive = soreceive;
    557       1.78      matt #ifdef MBUFTRACE
    558       1.78      matt 	so->so_rcv.sb_mowner = &prp->pr_domain->dom_mowner;
    559       1.78      matt 	so->so_snd.sb_mowner = &prp->pr_domain->dom_mowner;
    560       1.78      matt 	so->so_mowner = &prp->pr_domain->dom_mowner;
    561       1.78      matt #endif
    562      1.138     rmind 	uid = kauth_cred_geteuid(l->l_cred);
    563      1.115      yamt 	so->so_uidinfo = uid_find(uid);
    564      1.168      yamt 	so->so_cpid = l->l_proc->p_pid;
    565      1.160        ad 	if (lockso != NULL) {
    566      1.160        ad 		/* Caller wants us to share a lock. */
    567      1.160        ad 		lock = lockso->so_lock;
    568      1.160        ad 		so->so_lock = lock;
    569      1.160        ad 		mutex_obj_hold(lock);
    570      1.160        ad 		mutex_enter(lock);
    571      1.160        ad 	} else {
    572      1.160        ad 		/* Lock assigned and taken during PRU_ATTACH. */
    573      1.160        ad 	}
    574      1.140    dyoung 	error = (*prp->pr_usrreq)(so, PRU_ATTACH, NULL,
    575      1.140    dyoung 	    (struct mbuf *)(long)proto, NULL, l);
    576      1.160        ad 	KASSERT(solocked(so));
    577      1.140    dyoung 	if (error != 0) {
    578        1.1       cgd 		so->so_state |= SS_NOFDREF;
    579        1.1       cgd 		sofree(so);
    580      1.140    dyoung 		return error;
    581        1.1       cgd 	}
    582      1.198      elad 	so->so_cred = kauth_cred_dup(l->l_cred);
    583      1.160        ad 	sounlock(so);
    584        1.1       cgd 	*aso = so;
    585      1.140    dyoung 	return 0;
    586        1.1       cgd }
    587        1.1       cgd 
    588      1.142    dyoung /* On success, write file descriptor to fdout and return zero.  On
    589      1.142    dyoung  * failure, return non-zero; *fdout will be undefined.
    590      1.142    dyoung  */
    591      1.142    dyoung int
    592      1.142    dyoung fsocreate(int domain, struct socket **sop, int type, int protocol,
    593      1.142    dyoung     struct lwp *l, int *fdout)
    594      1.142    dyoung {
    595      1.142    dyoung 	struct socket	*so;
    596      1.142    dyoung 	struct file	*fp;
    597      1.142    dyoung 	int		fd, error;
    598      1.142    dyoung 
    599      1.155        ad 	if ((error = fd_allocfile(&fp, &fd)) != 0)
    600      1.142    dyoung 		return (error);
    601      1.142    dyoung 	fp->f_flag = FREAD|FWRITE;
    602      1.142    dyoung 	fp->f_type = DTYPE_SOCKET;
    603      1.142    dyoung 	fp->f_ops = &socketops;
    604      1.160        ad 	error = socreate(domain, &so, type, protocol, l, NULL);
    605      1.142    dyoung 	if (error != 0) {
    606      1.155        ad 		fd_abort(curproc, fp, fd);
    607      1.142    dyoung 	} else {
    608      1.142    dyoung 		if (sop != NULL)
    609      1.142    dyoung 			*sop = so;
    610      1.142    dyoung 		fp->f_data = so;
    611      1.155        ad 		fd_affix(curproc, fp, fd);
    612      1.142    dyoung 		*fdout = fd;
    613      1.142    dyoung 	}
    614      1.142    dyoung 	return error;
    615      1.142    dyoung }
    616      1.142    dyoung 
    617        1.3    andrew int
    618      1.190    dyoung sofamily(const struct socket *so)
    619      1.190    dyoung {
    620      1.190    dyoung 	const struct protosw *pr;
    621      1.190    dyoung 	const struct domain *dom;
    622      1.190    dyoung 
    623      1.190    dyoung 	if ((pr = so->so_proto) == NULL)
    624      1.190    dyoung 		return AF_UNSPEC;
    625      1.190    dyoung 	if ((dom = pr->pr_domain) == NULL)
    626      1.190    dyoung 		return AF_UNSPEC;
    627      1.190    dyoung 	return dom->dom_family;
    628      1.190    dyoung }
    629      1.190    dyoung 
    630      1.190    dyoung int
    631      1.114  christos sobind(struct socket *so, struct mbuf *nam, struct lwp *l)
    632        1.1       cgd {
    633      1.160        ad 	int	error;
    634        1.1       cgd 
    635      1.160        ad 	solock(so);
    636      1.140    dyoung 	error = (*so->so_proto->pr_usrreq)(so, PRU_BIND, NULL, nam, NULL, l);
    637      1.160        ad 	sounlock(so);
    638      1.140    dyoung 	return error;
    639        1.1       cgd }
    640        1.1       cgd 
    641        1.3    andrew int
    642      1.150      elad solisten(struct socket *so, int backlog, struct lwp *l)
    643        1.1       cgd {
    644      1.160        ad 	int	error;
    645        1.1       cgd 
    646      1.160        ad 	solock(so);
    647      1.158        ad 	if ((so->so_state & (SS_ISCONNECTED | SS_ISCONNECTING |
    648      1.163        ad 	    SS_ISDISCONNECTING)) != 0) {
    649      1.163        ad 	    	sounlock(so);
    650      1.158        ad 		return (EOPNOTSUPP);
    651      1.163        ad 	}
    652      1.140    dyoung 	error = (*so->so_proto->pr_usrreq)(so, PRU_LISTEN, NULL,
    653      1.150      elad 	    NULL, NULL, l);
    654      1.140    dyoung 	if (error != 0) {
    655      1.160        ad 		sounlock(so);
    656      1.140    dyoung 		return error;
    657        1.1       cgd 	}
    658       1.63      matt 	if (TAILQ_EMPTY(&so->so_q))
    659        1.1       cgd 		so->so_options |= SO_ACCEPTCONN;
    660        1.1       cgd 	if (backlog < 0)
    661        1.1       cgd 		backlog = 0;
    662       1.49  jonathan 	so->so_qlimit = min(backlog, somaxconn);
    663      1.160        ad 	sounlock(so);
    664      1.140    dyoung 	return 0;
    665        1.1       cgd }
    666        1.1       cgd 
    667       1.21  christos void
    668       1.54     lukem sofree(struct socket *so)
    669        1.1       cgd {
    670      1.161        ad 	u_int refs;
    671        1.1       cgd 
    672      1.160        ad 	KASSERT(solocked(so));
    673      1.160        ad 
    674      1.160        ad 	if (so->so_pcb || (so->so_state & SS_NOFDREF) == 0) {
    675      1.160        ad 		sounlock(so);
    676        1.1       cgd 		return;
    677      1.160        ad 	}
    678       1.43   mycroft 	if (so->so_head) {
    679       1.43   mycroft 		/*
    680       1.43   mycroft 		 * We must not decommission a socket that's on the accept(2)
    681       1.43   mycroft 		 * queue.  If we do, then accept(2) may hang after select(2)
    682       1.43   mycroft 		 * indicated that the listening socket was ready.
    683       1.43   mycroft 		 */
    684      1.160        ad 		if (!soqremque(so, 0)) {
    685      1.160        ad 			sounlock(so);
    686       1.43   mycroft 			return;
    687      1.160        ad 		}
    688       1.43   mycroft 	}
    689       1.98  christos 	if (so->so_rcv.sb_hiwat)
    690      1.110  christos 		(void)chgsbsize(so->so_uidinfo, &so->so_rcv.sb_hiwat, 0,
    691       1.98  christos 		    RLIM_INFINITY);
    692       1.98  christos 	if (so->so_snd.sb_hiwat)
    693      1.110  christos 		(void)chgsbsize(so->so_uidinfo, &so->so_snd.sb_hiwat, 0,
    694       1.98  christos 		    RLIM_INFINITY);
    695       1.98  christos 	sbrelease(&so->so_snd, so);
    696      1.160        ad 	KASSERT(!cv_has_waiters(&so->so_cv));
    697      1.160        ad 	KASSERT(!cv_has_waiters(&so->so_rcv.sb_cv));
    698      1.160        ad 	KASSERT(!cv_has_waiters(&so->so_snd.sb_cv));
    699        1.1       cgd 	sorflush(so);
    700      1.161        ad 	refs = so->so_aborting;	/* XXX */
    701      1.177        ad 	/* Remove acccept filter if one is present. */
    702      1.170       tls 	if (so->so_accf != NULL)
    703      1.177        ad 		(void)accept_filt_clear(so);
    704      1.160        ad 	sounlock(so);
    705      1.161        ad 	if (refs == 0)		/* XXX */
    706      1.161        ad 		soput(so);
    707        1.1       cgd }
    708        1.1       cgd 
    709        1.1       cgd /*
    710        1.1       cgd  * Close a socket on last file table reference removal.
    711        1.1       cgd  * Initiate disconnect if connected.
    712        1.1       cgd  * Free socket when disconnect complete.
    713        1.1       cgd  */
    714        1.3    andrew int
    715       1.54     lukem soclose(struct socket *so)
    716        1.1       cgd {
    717       1.54     lukem 	struct socket	*so2;
    718      1.160        ad 	int		error;
    719      1.160        ad 	int		error2;
    720        1.1       cgd 
    721       1.54     lukem 	error = 0;
    722      1.160        ad 	solock(so);
    723        1.1       cgd 	if (so->so_options & SO_ACCEPTCONN) {
    724      1.172        ad 		for (;;) {
    725      1.172        ad 			if ((so2 = TAILQ_FIRST(&so->so_q0)) != 0) {
    726      1.160        ad 				KASSERT(solocked2(so, so2));
    727      1.160        ad 				(void) soqremque(so2, 0);
    728      1.160        ad 				/* soabort drops the lock. */
    729      1.160        ad 				(void) soabort(so2);
    730      1.160        ad 				solock(so);
    731      1.172        ad 				continue;
    732      1.160        ad 			}
    733      1.172        ad 			if ((so2 = TAILQ_FIRST(&so->so_q)) != 0) {
    734      1.160        ad 				KASSERT(solocked2(so, so2));
    735      1.160        ad 				(void) soqremque(so2, 1);
    736      1.160        ad 				/* soabort drops the lock. */
    737      1.160        ad 				(void) soabort(so2);
    738      1.160        ad 				solock(so);
    739      1.172        ad 				continue;
    740      1.160        ad 			}
    741      1.172        ad 			break;
    742      1.172        ad 		}
    743        1.1       cgd 	}
    744        1.1       cgd 	if (so->so_pcb == 0)
    745        1.1       cgd 		goto discard;
    746        1.1       cgd 	if (so->so_state & SS_ISCONNECTED) {
    747        1.1       cgd 		if ((so->so_state & SS_ISDISCONNECTING) == 0) {
    748        1.1       cgd 			error = sodisconnect(so);
    749        1.1       cgd 			if (error)
    750        1.1       cgd 				goto drop;
    751        1.1       cgd 		}
    752        1.1       cgd 		if (so->so_options & SO_LINGER) {
    753      1.151        ad 			if ((so->so_state & SS_ISDISCONNECTING) && so->so_nbio)
    754        1.1       cgd 				goto drop;
    755       1.21  christos 			while (so->so_state & SS_ISCONNECTED) {
    756      1.185      yamt 				error = sowait(so, true, so->so_linger * hz);
    757       1.21  christos 				if (error)
    758        1.1       cgd 					break;
    759       1.21  christos 			}
    760        1.1       cgd 		}
    761        1.1       cgd 	}
    762       1.54     lukem  drop:
    763        1.1       cgd 	if (so->so_pcb) {
    764      1.160        ad 		error2 = (*so->so_proto->pr_usrreq)(so, PRU_DETACH,
    765      1.140    dyoung 		    NULL, NULL, NULL, NULL);
    766        1.1       cgd 		if (error == 0)
    767        1.1       cgd 			error = error2;
    768        1.1       cgd 	}
    769       1.54     lukem  discard:
    770        1.1       cgd 	if (so->so_state & SS_NOFDREF)
    771        1.1       cgd 		panic("soclose: NOFDREF");
    772      1.198      elad 	kauth_cred_free(so->so_cred);
    773        1.1       cgd 	so->so_state |= SS_NOFDREF;
    774        1.1       cgd 	sofree(so);
    775        1.1       cgd 	return (error);
    776        1.1       cgd }
    777        1.1       cgd 
    778        1.1       cgd /*
    779      1.160        ad  * Must be called with the socket locked..  Will return with it unlocked.
    780        1.1       cgd  */
    781        1.3    andrew int
    782       1.54     lukem soabort(struct socket *so)
    783        1.1       cgd {
    784      1.161        ad 	u_int refs;
    785      1.139      yamt 	int error;
    786      1.160        ad 
    787      1.160        ad 	KASSERT(solocked(so));
    788      1.160        ad 	KASSERT(so->so_head == NULL);
    789        1.1       cgd 
    790      1.161        ad 	so->so_aborting++;		/* XXX */
    791      1.140    dyoung 	error = (*so->so_proto->pr_usrreq)(so, PRU_ABORT, NULL,
    792      1.140    dyoung 	    NULL, NULL, NULL);
    793      1.161        ad 	refs = --so->so_aborting;	/* XXX */
    794      1.164  drochner 	if (error || (refs == 0)) {
    795      1.139      yamt 		sofree(so);
    796      1.160        ad 	} else {
    797      1.160        ad 		sounlock(so);
    798      1.139      yamt 	}
    799      1.139      yamt 	return error;
    800        1.1       cgd }
    801        1.1       cgd 
    802        1.3    andrew int
    803       1.54     lukem soaccept(struct socket *so, struct mbuf *nam)
    804        1.1       cgd {
    805      1.160        ad 	int	error;
    806      1.160        ad 
    807      1.160        ad 	KASSERT(solocked(so));
    808        1.1       cgd 
    809       1.54     lukem 	error = 0;
    810        1.1       cgd 	if ((so->so_state & SS_NOFDREF) == 0)
    811        1.1       cgd 		panic("soaccept: !NOFDREF");
    812        1.1       cgd 	so->so_state &= ~SS_NOFDREF;
    813       1.55   thorpej 	if ((so->so_state & SS_ISDISCONNECTED) == 0 ||
    814       1.55   thorpej 	    (so->so_proto->pr_flags & PR_ABRTACPTDIS) == 0)
    815       1.41   mycroft 		error = (*so->so_proto->pr_usrreq)(so, PRU_ACCEPT,
    816      1.140    dyoung 		    NULL, nam, NULL, NULL);
    817       1.41   mycroft 	else
    818       1.53    itojun 		error = ECONNABORTED;
    819       1.52    itojun 
    820        1.1       cgd 	return (error);
    821        1.1       cgd }
    822        1.1       cgd 
    823        1.3    andrew int
    824      1.114  christos soconnect(struct socket *so, struct mbuf *nam, struct lwp *l)
    825        1.1       cgd {
    826      1.160        ad 	int		error;
    827      1.160        ad 
    828      1.160        ad 	KASSERT(solocked(so));
    829        1.1       cgd 
    830        1.1       cgd 	if (so->so_options & SO_ACCEPTCONN)
    831        1.1       cgd 		return (EOPNOTSUPP);
    832        1.1       cgd 	/*
    833        1.1       cgd 	 * If protocol is connection-based, can only connect once.
    834        1.1       cgd 	 * Otherwise, if connected, try to disconnect first.
    835        1.1       cgd 	 * This allows user to disconnect by connecting to, e.g.,
    836        1.1       cgd 	 * a null address.
    837        1.1       cgd 	 */
    838        1.1       cgd 	if (so->so_state & (SS_ISCONNECTED|SS_ISCONNECTING) &&
    839        1.1       cgd 	    ((so->so_proto->pr_flags & PR_CONNREQUIRED) ||
    840        1.1       cgd 	    (error = sodisconnect(so))))
    841        1.1       cgd 		error = EISCONN;
    842        1.1       cgd 	else
    843        1.1       cgd 		error = (*so->so_proto->pr_usrreq)(so, PRU_CONNECT,
    844      1.140    dyoung 		    NULL, nam, NULL, l);
    845        1.1       cgd 	return (error);
    846        1.1       cgd }
    847        1.1       cgd 
    848        1.3    andrew int
    849       1.54     lukem soconnect2(struct socket *so1, struct socket *so2)
    850        1.1       cgd {
    851      1.160        ad 	int	error;
    852      1.160        ad 
    853      1.160        ad 	KASSERT(solocked2(so1, so2));
    854        1.1       cgd 
    855       1.22   mycroft 	error = (*so1->so_proto->pr_usrreq)(so1, PRU_CONNECT2,
    856      1.140    dyoung 	    NULL, (struct mbuf *)so2, NULL, NULL);
    857        1.1       cgd 	return (error);
    858        1.1       cgd }
    859        1.1       cgd 
    860        1.3    andrew int
    861       1.54     lukem sodisconnect(struct socket *so)
    862        1.1       cgd {
    863      1.160        ad 	int	error;
    864      1.160        ad 
    865      1.160        ad 	KASSERT(solocked(so));
    866        1.1       cgd 
    867        1.1       cgd 	if ((so->so_state & SS_ISCONNECTED) == 0) {
    868        1.1       cgd 		error = ENOTCONN;
    869      1.160        ad 	} else if (so->so_state & SS_ISDISCONNECTING) {
    870        1.1       cgd 		error = EALREADY;
    871      1.160        ad 	} else {
    872      1.160        ad 		error = (*so->so_proto->pr_usrreq)(so, PRU_DISCONNECT,
    873      1.160        ad 		    NULL, NULL, NULL, NULL);
    874        1.1       cgd 	}
    875      1.117      yamt 	sodopendfree();
    876        1.1       cgd 	return (error);
    877        1.1       cgd }
    878        1.1       cgd 
    879       1.15   mycroft #define	SBLOCKWAIT(f)	(((f) & MSG_DONTWAIT) ? M_NOWAIT : M_WAITOK)
    880        1.1       cgd /*
    881        1.1       cgd  * Send on a socket.
    882        1.1       cgd  * If send must go all at once and message is larger than
    883        1.1       cgd  * send buffering, then hard error.
    884        1.1       cgd  * Lock against other senders.
    885        1.1       cgd  * If must go all at once and not enough room now, then
    886        1.1       cgd  * inform user that this would block and do nothing.
    887        1.1       cgd  * Otherwise, if nonblocking, send as much as possible.
    888        1.1       cgd  * The data to be sent is described by "uio" if nonzero,
    889        1.1       cgd  * otherwise by the mbuf chain "top" (which must be null
    890        1.1       cgd  * if uio is not).  Data provided in mbuf chain must be small
    891        1.1       cgd  * enough to send all at once.
    892        1.1       cgd  *
    893        1.1       cgd  * Returns nonzero on error, timeout or signal; callers
    894        1.1       cgd  * must check for short counts if EINTR/ERESTART are returned.
    895        1.1       cgd  * Data and control buffers are freed on return.
    896        1.1       cgd  */
    897        1.3    andrew int
    898       1.54     lukem sosend(struct socket *so, struct mbuf *addr, struct uio *uio, struct mbuf *top,
    899      1.114  christos 	struct mbuf *control, int flags, struct lwp *l)
    900        1.1       cgd {
    901       1.54     lukem 	struct mbuf	**mp, *m;
    902      1.114  christos 	struct proc	*p;
    903       1.58  jdolecek 	long		space, len, resid, clen, mlen;
    904       1.58  jdolecek 	int		error, s, dontroute, atomic;
    905      1.196       dsl 	short		wakeup_state = 0;
    906       1.54     lukem 
    907      1.114  christos 	p = l->l_proc;
    908      1.117      yamt 	sodopendfree();
    909      1.160        ad 	clen = 0;
    910       1.64   thorpej 
    911      1.160        ad 	/*
    912      1.160        ad 	 * solock() provides atomicity of access.  splsoftnet() prevents
    913      1.160        ad 	 * protocol processing soft interrupts from interrupting us and
    914      1.160        ad 	 * blocking (expensive).
    915      1.160        ad 	 */
    916      1.160        ad 	s = splsoftnet();
    917      1.160        ad 	solock(so);
    918       1.54     lukem 	atomic = sosendallatonce(so) || top;
    919        1.1       cgd 	if (uio)
    920        1.1       cgd 		resid = uio->uio_resid;
    921        1.1       cgd 	else
    922        1.1       cgd 		resid = top->m_pkthdr.len;
    923        1.7       cgd 	/*
    924        1.7       cgd 	 * In theory resid should be unsigned.
    925        1.7       cgd 	 * However, space must be signed, as it might be less than 0
    926        1.7       cgd 	 * if we over-committed, and we must use a signed comparison
    927        1.7       cgd 	 * of space and resid.  On the other hand, a negative resid
    928        1.7       cgd 	 * causes us to loop sending 0-length segments to the protocol.
    929        1.7       cgd 	 */
    930       1.29   mycroft 	if (resid < 0) {
    931       1.29   mycroft 		error = EINVAL;
    932       1.29   mycroft 		goto out;
    933       1.29   mycroft 	}
    934        1.1       cgd 	dontroute =
    935        1.1       cgd 	    (flags & MSG_DONTROUTE) && (so->so_options & SO_DONTROUTE) == 0 &&
    936        1.1       cgd 	    (so->so_proto->pr_flags & PR_ATOMIC);
    937      1.165  christos 	l->l_ru.ru_msgsnd++;
    938        1.1       cgd 	if (control)
    939        1.1       cgd 		clen = control->m_len;
    940       1.54     lukem  restart:
    941       1.21  christos 	if ((error = sblock(&so->so_snd, SBLOCKWAIT(flags))) != 0)
    942        1.1       cgd 		goto out;
    943        1.1       cgd 	do {
    944      1.160        ad 		if (so->so_state & SS_CANTSENDMORE) {
    945      1.160        ad 			error = EPIPE;
    946      1.160        ad 			goto release;
    947      1.160        ad 		}
    948       1.48   thorpej 		if (so->so_error) {
    949       1.48   thorpej 			error = so->so_error;
    950       1.48   thorpej 			so->so_error = 0;
    951       1.48   thorpej 			goto release;
    952       1.48   thorpej 		}
    953        1.1       cgd 		if ((so->so_state & SS_ISCONNECTED) == 0) {
    954        1.1       cgd 			if (so->so_proto->pr_flags & PR_CONNREQUIRED) {
    955        1.1       cgd 				if ((so->so_state & SS_ISCONFIRMING) == 0 &&
    956      1.160        ad 				    !(resid == 0 && clen != 0)) {
    957      1.160        ad 					error = ENOTCONN;
    958      1.160        ad 					goto release;
    959      1.160        ad 				}
    960      1.160        ad 			} else if (addr == 0) {
    961      1.160        ad 				error = EDESTADDRREQ;
    962      1.160        ad 				goto release;
    963      1.160        ad 			}
    964        1.1       cgd 		}
    965        1.1       cgd 		space = sbspace(&so->so_snd);
    966        1.1       cgd 		if (flags & MSG_OOB)
    967        1.1       cgd 			space += 1024;
    968       1.21  christos 		if ((atomic && resid > so->so_snd.sb_hiwat) ||
    969      1.160        ad 		    clen > so->so_snd.sb_hiwat) {
    970      1.160        ad 			error = EMSGSIZE;
    971      1.160        ad 			goto release;
    972      1.160        ad 		}
    973       1.96   mycroft 		if (space < resid + clen &&
    974        1.1       cgd 		    (atomic || space < so->so_snd.sb_lowat || space < clen)) {
    975      1.160        ad 			if (so->so_nbio) {
    976      1.160        ad 				error = EWOULDBLOCK;
    977      1.160        ad 				goto release;
    978      1.160        ad 			}
    979        1.1       cgd 			sbunlock(&so->so_snd);
    980      1.196       dsl 			if (wakeup_state & SS_RESTARTSYS) {
    981      1.196       dsl 				error = ERESTART;
    982      1.196       dsl 				goto out;
    983      1.196       dsl 			}
    984        1.1       cgd 			error = sbwait(&so->so_snd);
    985        1.1       cgd 			if (error)
    986        1.1       cgd 				goto out;
    987      1.196       dsl 			wakeup_state = so->so_state;
    988        1.1       cgd 			goto restart;
    989        1.1       cgd 		}
    990      1.196       dsl 		wakeup_state = 0;
    991        1.1       cgd 		mp = &top;
    992        1.1       cgd 		space -= clen;
    993        1.1       cgd 		do {
    994       1.45        tv 			if (uio == NULL) {
    995       1.45        tv 				/*
    996       1.45        tv 				 * Data is prepackaged in "top".
    997       1.45        tv 				 */
    998       1.45        tv 				resid = 0;
    999       1.45        tv 				if (flags & MSG_EOR)
   1000       1.45        tv 					top->m_flags |= M_EOR;
   1001       1.45        tv 			} else do {
   1002      1.160        ad 				sounlock(so);
   1003      1.160        ad 				splx(s);
   1004      1.144    dyoung 				if (top == NULL) {
   1005       1.78      matt 					m = m_gethdr(M_WAIT, MT_DATA);
   1006       1.45        tv 					mlen = MHLEN;
   1007       1.45        tv 					m->m_pkthdr.len = 0;
   1008      1.140    dyoung 					m->m_pkthdr.rcvif = NULL;
   1009       1.45        tv 				} else {
   1010       1.78      matt 					m = m_get(M_WAIT, MT_DATA);
   1011       1.45        tv 					mlen = MLEN;
   1012       1.45        tv 				}
   1013       1.78      matt 				MCLAIM(m, so->so_snd.sb_mowner);
   1014      1.121      yamt 				if (sock_loan_thresh >= 0 &&
   1015      1.121      yamt 				    uio->uio_iov->iov_len >= sock_loan_thresh &&
   1016      1.121      yamt 				    space >= sock_loan_thresh &&
   1017       1.64   thorpej 				    (len = sosend_loan(so, uio, m,
   1018       1.64   thorpej 						       space)) != 0) {
   1019       1.64   thorpej 					SOSEND_COUNTER_INCR(&sosend_loan_big);
   1020       1.64   thorpej 					space -= len;
   1021       1.64   thorpej 					goto have_data;
   1022       1.64   thorpej 				}
   1023       1.45        tv 				if (resid >= MINCLSIZE && space >= MCLBYTES) {
   1024       1.64   thorpej 					SOSEND_COUNTER_INCR(&sosend_copy_big);
   1025  1.200.2.1  uebayasi 					m_clget(m, M_DONTWAIT);
   1026       1.45        tv 					if ((m->m_flags & M_EXT) == 0)
   1027       1.45        tv 						goto nopages;
   1028       1.45        tv 					mlen = MCLBYTES;
   1029       1.45        tv 					if (atomic && top == 0) {
   1030       1.58  jdolecek 						len = lmin(MCLBYTES - max_hdr,
   1031       1.54     lukem 						    resid);
   1032       1.45        tv 						m->m_data += max_hdr;
   1033       1.45        tv 					} else
   1034       1.58  jdolecek 						len = lmin(MCLBYTES, resid);
   1035       1.45        tv 					space -= len;
   1036       1.45        tv 				} else {
   1037       1.64   thorpej  nopages:
   1038       1.64   thorpej 					SOSEND_COUNTER_INCR(&sosend_copy_small);
   1039       1.58  jdolecek 					len = lmin(lmin(mlen, resid), space);
   1040       1.45        tv 					space -= len;
   1041       1.45        tv 					/*
   1042       1.45        tv 					 * For datagram protocols, leave room
   1043       1.45        tv 					 * for protocol headers in first mbuf.
   1044       1.45        tv 					 */
   1045       1.45        tv 					if (atomic && top == 0 && len < mlen)
   1046       1.45        tv 						MH_ALIGN(m, len);
   1047       1.45        tv 				}
   1048      1.144    dyoung 				error = uiomove(mtod(m, void *), (int)len, uio);
   1049       1.64   thorpej  have_data:
   1050       1.45        tv 				resid = uio->uio_resid;
   1051       1.45        tv 				m->m_len = len;
   1052       1.45        tv 				*mp = m;
   1053       1.45        tv 				top->m_pkthdr.len += len;
   1054      1.160        ad 				s = splsoftnet();
   1055      1.160        ad 				solock(so);
   1056      1.144    dyoung 				if (error != 0)
   1057       1.45        tv 					goto release;
   1058       1.45        tv 				mp = &m->m_next;
   1059       1.45        tv 				if (resid <= 0) {
   1060       1.45        tv 					if (flags & MSG_EOR)
   1061       1.45        tv 						top->m_flags |= M_EOR;
   1062       1.45        tv 					break;
   1063       1.45        tv 				}
   1064       1.45        tv 			} while (space > 0 && atomic);
   1065      1.108     perry 
   1066      1.160        ad 			if (so->so_state & SS_CANTSENDMORE) {
   1067      1.160        ad 				error = EPIPE;
   1068      1.160        ad 				goto release;
   1069      1.160        ad 			}
   1070       1.45        tv 			if (dontroute)
   1071       1.45        tv 				so->so_options |= SO_DONTROUTE;
   1072       1.45        tv 			if (resid > 0)
   1073       1.45        tv 				so->so_state |= SS_MORETOCOME;
   1074       1.46  sommerfe 			error = (*so->so_proto->pr_usrreq)(so,
   1075       1.46  sommerfe 			    (flags & MSG_OOB) ? PRU_SENDOOB : PRU_SEND,
   1076      1.160        ad 			    top, addr, control, curlwp);
   1077       1.45        tv 			if (dontroute)
   1078       1.45        tv 				so->so_options &= ~SO_DONTROUTE;
   1079       1.45        tv 			if (resid > 0)
   1080       1.45        tv 				so->so_state &= ~SS_MORETOCOME;
   1081       1.45        tv 			clen = 0;
   1082      1.144    dyoung 			control = NULL;
   1083      1.144    dyoung 			top = NULL;
   1084       1.45        tv 			mp = &top;
   1085      1.144    dyoung 			if (error != 0)
   1086        1.1       cgd 				goto release;
   1087        1.1       cgd 		} while (resid && space > 0);
   1088        1.1       cgd 	} while (resid);
   1089        1.1       cgd 
   1090       1.54     lukem  release:
   1091        1.1       cgd 	sbunlock(&so->so_snd);
   1092       1.54     lukem  out:
   1093      1.160        ad 	sounlock(so);
   1094      1.160        ad 	splx(s);
   1095        1.1       cgd 	if (top)
   1096        1.1       cgd 		m_freem(top);
   1097        1.1       cgd 	if (control)
   1098        1.1       cgd 		m_freem(control);
   1099        1.1       cgd 	return (error);
   1100        1.1       cgd }
   1101        1.1       cgd 
   1102        1.1       cgd /*
   1103      1.159        ad  * Following replacement or removal of the first mbuf on the first
   1104      1.159        ad  * mbuf chain of a socket buffer, push necessary state changes back
   1105      1.159        ad  * into the socket buffer so that other consumers see the values
   1106      1.159        ad  * consistently.  'nextrecord' is the callers locally stored value of
   1107      1.159        ad  * the original value of sb->sb_mb->m_nextpkt which must be restored
   1108      1.159        ad  * when the lead mbuf changes.  NOTE: 'nextrecord' may be NULL.
   1109      1.159        ad  */
   1110      1.159        ad static void
   1111      1.159        ad sbsync(struct sockbuf *sb, struct mbuf *nextrecord)
   1112      1.159        ad {
   1113      1.159        ad 
   1114      1.160        ad 	KASSERT(solocked(sb->sb_so));
   1115      1.160        ad 
   1116      1.159        ad 	/*
   1117      1.159        ad 	 * First, update for the new value of nextrecord.  If necessary,
   1118      1.159        ad 	 * make it the first record.
   1119      1.159        ad 	 */
   1120      1.159        ad 	if (sb->sb_mb != NULL)
   1121      1.159        ad 		sb->sb_mb->m_nextpkt = nextrecord;
   1122      1.159        ad 	else
   1123      1.159        ad 		sb->sb_mb = nextrecord;
   1124      1.159        ad 
   1125      1.159        ad         /*
   1126      1.159        ad          * Now update any dependent socket buffer fields to reflect
   1127      1.159        ad          * the new state.  This is an inline of SB_EMPTY_FIXUP, with
   1128      1.159        ad          * the addition of a second clause that takes care of the
   1129      1.159        ad          * case where sb_mb has been updated, but remains the last
   1130      1.159        ad          * record.
   1131      1.159        ad          */
   1132      1.159        ad         if (sb->sb_mb == NULL) {
   1133      1.159        ad                 sb->sb_mbtail = NULL;
   1134      1.159        ad                 sb->sb_lastrecord = NULL;
   1135      1.159        ad         } else if (sb->sb_mb->m_nextpkt == NULL)
   1136      1.159        ad                 sb->sb_lastrecord = sb->sb_mb;
   1137      1.159        ad }
   1138      1.159        ad 
   1139      1.159        ad /*
   1140        1.1       cgd  * Implement receive operations on a socket.
   1141        1.1       cgd  * We depend on the way that records are added to the sockbuf
   1142        1.1       cgd  * by sbappend*.  In particular, each record (mbufs linked through m_next)
   1143        1.1       cgd  * must begin with an address if the protocol so specifies,
   1144        1.1       cgd  * followed by an optional mbuf or mbufs containing ancillary data,
   1145        1.1       cgd  * and then zero or more mbufs of data.
   1146        1.1       cgd  * In order to avoid blocking network interrupts for the entire time here,
   1147        1.1       cgd  * we splx() while doing the actual copy to user space.
   1148        1.1       cgd  * Although the sockbuf is locked, new data may still be appended,
   1149        1.1       cgd  * and thus we must maintain consistency of the sockbuf during that time.
   1150        1.1       cgd  *
   1151        1.1       cgd  * The caller may receive the data as a single mbuf chain by supplying
   1152        1.1       cgd  * an mbuf **mp0 for use in returning the chain.  The uio is then used
   1153        1.1       cgd  * only for the count in uio_resid.
   1154        1.1       cgd  */
   1155        1.3    andrew int
   1156       1.54     lukem soreceive(struct socket *so, struct mbuf **paddr, struct uio *uio,
   1157       1.54     lukem 	struct mbuf **mp0, struct mbuf **controlp, int *flagsp)
   1158        1.1       cgd {
   1159      1.116      yamt 	struct lwp *l = curlwp;
   1160      1.160        ad 	struct mbuf	*m, **mp, *mt;
   1161      1.146    dyoung 	int atomic, flags, len, error, s, offset, moff, type, orig_resid;
   1162       1.99      matt 	const struct protosw	*pr;
   1163       1.54     lukem 	struct mbuf	*nextrecord;
   1164       1.67        he 	int		mbuf_removed = 0;
   1165      1.146    dyoung 	const struct domain *dom;
   1166      1.196       dsl 	short		wakeup_state = 0;
   1167       1.64   thorpej 
   1168       1.54     lukem 	pr = so->so_proto;
   1169      1.146    dyoung 	atomic = pr->pr_flags & PR_ATOMIC;
   1170      1.146    dyoung 	dom = pr->pr_domain;
   1171        1.1       cgd 	mp = mp0;
   1172       1.54     lukem 	type = 0;
   1173       1.54     lukem 	orig_resid = uio->uio_resid;
   1174      1.102  jonathan 
   1175      1.144    dyoung 	if (paddr != NULL)
   1176      1.144    dyoung 		*paddr = NULL;
   1177      1.144    dyoung 	if (controlp != NULL)
   1178      1.144    dyoung 		*controlp = NULL;
   1179      1.144    dyoung 	if (flagsp != NULL)
   1180        1.1       cgd 		flags = *flagsp &~ MSG_EOR;
   1181        1.1       cgd 	else
   1182        1.1       cgd 		flags = 0;
   1183       1.66     enami 
   1184       1.66     enami 	if ((flags & MSG_DONTWAIT) == 0)
   1185      1.117      yamt 		sodopendfree();
   1186       1.66     enami 
   1187        1.1       cgd 	if (flags & MSG_OOB) {
   1188        1.1       cgd 		m = m_get(M_WAIT, MT_DATA);
   1189      1.160        ad 		solock(so);
   1190       1.17       cgd 		error = (*pr->pr_usrreq)(so, PRU_RCVOOB, m,
   1191      1.140    dyoung 		    (struct mbuf *)(long)(flags & MSG_PEEK), NULL, l);
   1192      1.160        ad 		sounlock(so);
   1193        1.1       cgd 		if (error)
   1194        1.1       cgd 			goto bad;
   1195        1.1       cgd 		do {
   1196      1.134  christos 			error = uiomove(mtod(m, void *),
   1197        1.1       cgd 			    (int) min(uio->uio_resid, m->m_len), uio);
   1198        1.1       cgd 			m = m_free(m);
   1199      1.144    dyoung 		} while (uio->uio_resid > 0 && error == 0 && m);
   1200       1.54     lukem  bad:
   1201      1.144    dyoung 		if (m != NULL)
   1202        1.1       cgd 			m_freem(m);
   1203      1.144    dyoung 		return error;
   1204        1.1       cgd 	}
   1205      1.144    dyoung 	if (mp != NULL)
   1206      1.140    dyoung 		*mp = NULL;
   1207      1.160        ad 
   1208      1.160        ad 	/*
   1209      1.160        ad 	 * solock() provides atomicity of access.  splsoftnet() prevents
   1210      1.160        ad 	 * protocol processing soft interrupts from interrupting us and
   1211      1.160        ad 	 * blocking (expensive).
   1212      1.160        ad 	 */
   1213      1.160        ad 	s = splsoftnet();
   1214      1.160        ad 	solock(so);
   1215        1.1       cgd 	if (so->so_state & SS_ISCONFIRMING && uio->uio_resid)
   1216      1.140    dyoung 		(*pr->pr_usrreq)(so, PRU_RCVD, NULL, NULL, NULL, l);
   1217        1.1       cgd 
   1218       1.54     lukem  restart:
   1219      1.160        ad 	if ((error = sblock(&so->so_rcv, SBLOCKWAIT(flags))) != 0) {
   1220      1.160        ad 		sounlock(so);
   1221      1.160        ad 		splx(s);
   1222      1.144    dyoung 		return error;
   1223      1.160        ad 	}
   1224        1.1       cgd 
   1225        1.1       cgd 	m = so->so_rcv.sb_mb;
   1226        1.1       cgd 	/*
   1227        1.1       cgd 	 * If we have less data than requested, block awaiting more
   1228        1.1       cgd 	 * (subject to any timeout) if:
   1229       1.15   mycroft 	 *   1. the current count is less than the low water mark,
   1230        1.1       cgd 	 *   2. MSG_WAITALL is set, and it is possible to do the entire
   1231       1.15   mycroft 	 *	receive operation at once if we block (resid <= hiwat), or
   1232       1.15   mycroft 	 *   3. MSG_DONTWAIT is not set.
   1233        1.1       cgd 	 * If MSG_WAITALL is set but resid is larger than the receive buffer,
   1234        1.1       cgd 	 * we have to do the receive in sections, and thus risk returning
   1235        1.1       cgd 	 * a short count if a timeout or signal occurs after we start.
   1236        1.1       cgd 	 */
   1237      1.144    dyoung 	if (m == NULL ||
   1238      1.144    dyoung 	    ((flags & MSG_DONTWAIT) == 0 &&
   1239      1.144    dyoung 	     so->so_rcv.sb_cc < uio->uio_resid &&
   1240      1.144    dyoung 	     (so->so_rcv.sb_cc < so->so_rcv.sb_lowat ||
   1241      1.144    dyoung 	      ((flags & MSG_WAITALL) &&
   1242      1.144    dyoung 	       uio->uio_resid <= so->so_rcv.sb_hiwat)) &&
   1243      1.146    dyoung 	     m->m_nextpkt == NULL && !atomic)) {
   1244        1.1       cgd #ifdef DIAGNOSTIC
   1245      1.144    dyoung 		if (m == NULL && so->so_rcv.sb_cc)
   1246        1.1       cgd 			panic("receive 1");
   1247        1.1       cgd #endif
   1248        1.1       cgd 		if (so->so_error) {
   1249      1.144    dyoung 			if (m != NULL)
   1250       1.15   mycroft 				goto dontblock;
   1251        1.1       cgd 			error = so->so_error;
   1252        1.1       cgd 			if ((flags & MSG_PEEK) == 0)
   1253        1.1       cgd 				so->so_error = 0;
   1254        1.1       cgd 			goto release;
   1255        1.1       cgd 		}
   1256        1.1       cgd 		if (so->so_state & SS_CANTRCVMORE) {
   1257      1.144    dyoung 			if (m != NULL)
   1258       1.15   mycroft 				goto dontblock;
   1259        1.1       cgd 			else
   1260        1.1       cgd 				goto release;
   1261        1.1       cgd 		}
   1262      1.144    dyoung 		for (; m != NULL; m = m->m_next)
   1263        1.1       cgd 			if (m->m_type == MT_OOBDATA  || (m->m_flags & M_EOR)) {
   1264        1.1       cgd 				m = so->so_rcv.sb_mb;
   1265        1.1       cgd 				goto dontblock;
   1266        1.1       cgd 			}
   1267        1.1       cgd 		if ((so->so_state & (SS_ISCONNECTED|SS_ISCONNECTING)) == 0 &&
   1268        1.1       cgd 		    (so->so_proto->pr_flags & PR_CONNREQUIRED)) {
   1269        1.1       cgd 			error = ENOTCONN;
   1270        1.1       cgd 			goto release;
   1271        1.1       cgd 		}
   1272        1.1       cgd 		if (uio->uio_resid == 0)
   1273        1.1       cgd 			goto release;
   1274      1.151        ad 		if (so->so_nbio || (flags & MSG_DONTWAIT)) {
   1275        1.1       cgd 			error = EWOULDBLOCK;
   1276        1.1       cgd 			goto release;
   1277        1.1       cgd 		}
   1278       1.69   thorpej 		SBLASTRECORDCHK(&so->so_rcv, "soreceive sbwait 1");
   1279       1.69   thorpej 		SBLASTMBUFCHK(&so->so_rcv, "soreceive sbwait 1");
   1280        1.1       cgd 		sbunlock(&so->so_rcv);
   1281      1.196       dsl 		if (wakeup_state & SS_RESTARTSYS)
   1282      1.196       dsl 			error = ERESTART;
   1283      1.196       dsl 		else
   1284      1.196       dsl 			error = sbwait(&so->so_rcv);
   1285      1.160        ad 		if (error != 0) {
   1286      1.160        ad 			sounlock(so);
   1287      1.160        ad 			splx(s);
   1288      1.144    dyoung 			return error;
   1289      1.160        ad 		}
   1290      1.196       dsl 		wakeup_state = so->so_state;
   1291        1.1       cgd 		goto restart;
   1292        1.1       cgd 	}
   1293       1.54     lukem  dontblock:
   1294       1.69   thorpej 	/*
   1295       1.69   thorpej 	 * On entry here, m points to the first record of the socket buffer.
   1296      1.159        ad 	 * From this point onward, we maintain 'nextrecord' as a cache of the
   1297      1.159        ad 	 * pointer to the next record in the socket buffer.  We must keep the
   1298      1.159        ad 	 * various socket buffer pointers and local stack versions of the
   1299      1.159        ad 	 * pointers in sync, pushing out modifications before dropping the
   1300      1.160        ad 	 * socket lock, and re-reading them when picking it up.
   1301      1.159        ad 	 *
   1302      1.159        ad 	 * Otherwise, we will race with the network stack appending new data
   1303      1.159        ad 	 * or records onto the socket buffer by using inconsistent/stale
   1304      1.159        ad 	 * versions of the field, possibly resulting in socket buffer
   1305      1.159        ad 	 * corruption.
   1306      1.159        ad 	 *
   1307      1.159        ad 	 * By holding the high-level sblock(), we prevent simultaneous
   1308      1.159        ad 	 * readers from pulling off the front of the socket buffer.
   1309       1.69   thorpej 	 */
   1310      1.144    dyoung 	if (l != NULL)
   1311      1.157        ad 		l->l_ru.ru_msgrcv++;
   1312       1.69   thorpej 	KASSERT(m == so->so_rcv.sb_mb);
   1313       1.69   thorpej 	SBLASTRECORDCHK(&so->so_rcv, "soreceive 1");
   1314       1.69   thorpej 	SBLASTMBUFCHK(&so->so_rcv, "soreceive 1");
   1315        1.1       cgd 	nextrecord = m->m_nextpkt;
   1316        1.1       cgd 	if (pr->pr_flags & PR_ADDR) {
   1317        1.1       cgd #ifdef DIAGNOSTIC
   1318        1.1       cgd 		if (m->m_type != MT_SONAME)
   1319        1.1       cgd 			panic("receive 1a");
   1320        1.1       cgd #endif
   1321        1.3    andrew 		orig_resid = 0;
   1322        1.1       cgd 		if (flags & MSG_PEEK) {
   1323        1.1       cgd 			if (paddr)
   1324        1.1       cgd 				*paddr = m_copy(m, 0, m->m_len);
   1325        1.1       cgd 			m = m->m_next;
   1326        1.1       cgd 		} else {
   1327        1.1       cgd 			sbfree(&so->so_rcv, m);
   1328       1.67        he 			mbuf_removed = 1;
   1329      1.144    dyoung 			if (paddr != NULL) {
   1330        1.1       cgd 				*paddr = m;
   1331        1.1       cgd 				so->so_rcv.sb_mb = m->m_next;
   1332      1.144    dyoung 				m->m_next = NULL;
   1333        1.1       cgd 				m = so->so_rcv.sb_mb;
   1334        1.1       cgd 			} else {
   1335        1.1       cgd 				MFREE(m, so->so_rcv.sb_mb);
   1336        1.1       cgd 				m = so->so_rcv.sb_mb;
   1337        1.1       cgd 			}
   1338      1.159        ad 			sbsync(&so->so_rcv, nextrecord);
   1339        1.1       cgd 		}
   1340        1.1       cgd 	}
   1341      1.159        ad 
   1342      1.159        ad 	/*
   1343      1.159        ad 	 * Process one or more MT_CONTROL mbufs present before any data mbufs
   1344      1.159        ad 	 * in the first mbuf chain on the socket buffer.  If MSG_PEEK, we
   1345      1.159        ad 	 * just copy the data; if !MSG_PEEK, we call into the protocol to
   1346      1.159        ad 	 * perform externalization (or freeing if controlp == NULL).
   1347      1.159        ad 	 */
   1348      1.159        ad 	if (__predict_false(m != NULL && m->m_type == MT_CONTROL)) {
   1349      1.159        ad 		struct mbuf *cm = NULL, *cmn;
   1350      1.159        ad 		struct mbuf **cme = &cm;
   1351      1.159        ad 
   1352      1.159        ad 		do {
   1353      1.159        ad 			if (flags & MSG_PEEK) {
   1354      1.159        ad 				if (controlp != NULL) {
   1355      1.159        ad 					*controlp = m_copy(m, 0, m->m_len);
   1356      1.159        ad 					controlp = &(*controlp)->m_next;
   1357      1.159        ad 				}
   1358      1.159        ad 				m = m->m_next;
   1359      1.159        ad 			} else {
   1360      1.159        ad 				sbfree(&so->so_rcv, m);
   1361        1.1       cgd 				so->so_rcv.sb_mb = m->m_next;
   1362      1.144    dyoung 				m->m_next = NULL;
   1363      1.159        ad 				*cme = m;
   1364      1.159        ad 				cme = &(*cme)->m_next;
   1365        1.1       cgd 				m = so->so_rcv.sb_mb;
   1366      1.159        ad 			}
   1367      1.159        ad 		} while (m != NULL && m->m_type == MT_CONTROL);
   1368      1.159        ad 		if ((flags & MSG_PEEK) == 0)
   1369      1.159        ad 			sbsync(&so->so_rcv, nextrecord);
   1370      1.159        ad 		for (; cm != NULL; cm = cmn) {
   1371      1.159        ad 			cmn = cm->m_next;
   1372      1.159        ad 			cm->m_next = NULL;
   1373      1.159        ad 			type = mtod(cm, struct cmsghdr *)->cmsg_type;
   1374      1.159        ad 			if (controlp != NULL) {
   1375      1.159        ad 				if (dom->dom_externalize != NULL &&
   1376      1.159        ad 				    type == SCM_RIGHTS) {
   1377      1.160        ad 					sounlock(so);
   1378      1.159        ad 					splx(s);
   1379      1.159        ad 					error = (*dom->dom_externalize)(cm, l);
   1380      1.159        ad 					s = splsoftnet();
   1381      1.160        ad 					solock(so);
   1382      1.159        ad 				}
   1383      1.159        ad 				*controlp = cm;
   1384      1.159        ad 				while (*controlp != NULL)
   1385      1.159        ad 					controlp = &(*controlp)->m_next;
   1386        1.1       cgd 			} else {
   1387      1.106    itojun 				/*
   1388      1.106    itojun 				 * Dispose of any SCM_RIGHTS message that went
   1389      1.106    itojun 				 * through the read path rather than recv.
   1390      1.106    itojun 				 */
   1391      1.159        ad 				if (dom->dom_dispose != NULL &&
   1392      1.159        ad 				    type == SCM_RIGHTS) {
   1393      1.160        ad 				    	sounlock(so);
   1394      1.159        ad 					(*dom->dom_dispose)(cm);
   1395      1.160        ad 					solock(so);
   1396      1.159        ad 				}
   1397      1.159        ad 				m_freem(cm);
   1398        1.1       cgd 			}
   1399        1.1       cgd 		}
   1400      1.159        ad 		if (m != NULL)
   1401      1.159        ad 			nextrecord = so->so_rcv.sb_mb->m_nextpkt;
   1402      1.159        ad 		else
   1403      1.159        ad 			nextrecord = so->so_rcv.sb_mb;
   1404      1.159        ad 		orig_resid = 0;
   1405        1.1       cgd 	}
   1406       1.69   thorpej 
   1407      1.159        ad 	/* If m is non-NULL, we have some data to read. */
   1408      1.159        ad 	if (__predict_true(m != NULL)) {
   1409        1.1       cgd 		type = m->m_type;
   1410        1.1       cgd 		if (type == MT_OOBDATA)
   1411        1.1       cgd 			flags |= MSG_OOB;
   1412        1.1       cgd 	}
   1413       1.69   thorpej 	SBLASTRECORDCHK(&so->so_rcv, "soreceive 2");
   1414       1.69   thorpej 	SBLASTMBUFCHK(&so->so_rcv, "soreceive 2");
   1415       1.69   thorpej 
   1416        1.1       cgd 	moff = 0;
   1417        1.1       cgd 	offset = 0;
   1418      1.144    dyoung 	while (m != NULL && uio->uio_resid > 0 && error == 0) {
   1419        1.1       cgd 		if (m->m_type == MT_OOBDATA) {
   1420        1.1       cgd 			if (type != MT_OOBDATA)
   1421        1.1       cgd 				break;
   1422        1.1       cgd 		} else if (type == MT_OOBDATA)
   1423        1.1       cgd 			break;
   1424        1.1       cgd #ifdef DIAGNOSTIC
   1425        1.1       cgd 		else if (m->m_type != MT_DATA && m->m_type != MT_HEADER)
   1426        1.1       cgd 			panic("receive 3");
   1427        1.1       cgd #endif
   1428        1.1       cgd 		so->so_state &= ~SS_RCVATMARK;
   1429      1.196       dsl 		wakeup_state = 0;
   1430        1.1       cgd 		len = uio->uio_resid;
   1431        1.1       cgd 		if (so->so_oobmark && len > so->so_oobmark - offset)
   1432        1.1       cgd 			len = so->so_oobmark - offset;
   1433        1.1       cgd 		if (len > m->m_len - moff)
   1434        1.1       cgd 			len = m->m_len - moff;
   1435        1.1       cgd 		/*
   1436        1.1       cgd 		 * If mp is set, just pass back the mbufs.
   1437        1.1       cgd 		 * Otherwise copy them out via the uio, then free.
   1438        1.1       cgd 		 * Sockbuf must be consistent here (points to current mbuf,
   1439        1.1       cgd 		 * it points to next record) when we drop priority;
   1440        1.1       cgd 		 * we must note any additions to the sockbuf when we
   1441        1.1       cgd 		 * block interrupts again.
   1442        1.1       cgd 		 */
   1443      1.144    dyoung 		if (mp == NULL) {
   1444       1.69   thorpej 			SBLASTRECORDCHK(&so->so_rcv, "soreceive uiomove");
   1445       1.69   thorpej 			SBLASTMBUFCHK(&so->so_rcv, "soreceive uiomove");
   1446      1.160        ad 			sounlock(so);
   1447        1.1       cgd 			splx(s);
   1448      1.134  christos 			error = uiomove(mtod(m, char *) + moff, (int)len, uio);
   1449       1.20   mycroft 			s = splsoftnet();
   1450      1.160        ad 			solock(so);
   1451      1.144    dyoung 			if (error != 0) {
   1452       1.67        he 				/*
   1453       1.67        he 				 * If any part of the record has been removed
   1454       1.67        he 				 * (such as the MT_SONAME mbuf, which will
   1455       1.67        he 				 * happen when PR_ADDR, and thus also
   1456       1.67        he 				 * PR_ATOMIC, is set), then drop the entire
   1457       1.67        he 				 * record to maintain the atomicity of the
   1458       1.67        he 				 * receive operation.
   1459       1.67        he 				 *
   1460       1.67        he 				 * This avoids a later panic("receive 1a")
   1461       1.67        he 				 * when compiled with DIAGNOSTIC.
   1462       1.67        he 				 */
   1463      1.146    dyoung 				if (m && mbuf_removed && atomic)
   1464       1.67        he 					(void) sbdroprecord(&so->so_rcv);
   1465       1.67        he 
   1466       1.57  jdolecek 				goto release;
   1467       1.67        he 			}
   1468        1.1       cgd 		} else
   1469        1.1       cgd 			uio->uio_resid -= len;
   1470        1.1       cgd 		if (len == m->m_len - moff) {
   1471        1.1       cgd 			if (m->m_flags & M_EOR)
   1472        1.1       cgd 				flags |= MSG_EOR;
   1473        1.1       cgd 			if (flags & MSG_PEEK) {
   1474        1.1       cgd 				m = m->m_next;
   1475        1.1       cgd 				moff = 0;
   1476        1.1       cgd 			} else {
   1477        1.1       cgd 				nextrecord = m->m_nextpkt;
   1478        1.1       cgd 				sbfree(&so->so_rcv, m);
   1479        1.1       cgd 				if (mp) {
   1480        1.1       cgd 					*mp = m;
   1481        1.1       cgd 					mp = &m->m_next;
   1482        1.1       cgd 					so->so_rcv.sb_mb = m = m->m_next;
   1483      1.140    dyoung 					*mp = NULL;
   1484        1.1       cgd 				} else {
   1485        1.1       cgd 					MFREE(m, so->so_rcv.sb_mb);
   1486        1.1       cgd 					m = so->so_rcv.sb_mb;
   1487        1.1       cgd 				}
   1488       1.69   thorpej 				/*
   1489       1.69   thorpej 				 * If m != NULL, we also know that
   1490       1.69   thorpej 				 * so->so_rcv.sb_mb != NULL.
   1491       1.69   thorpej 				 */
   1492       1.69   thorpej 				KASSERT(so->so_rcv.sb_mb == m);
   1493       1.69   thorpej 				if (m) {
   1494        1.1       cgd 					m->m_nextpkt = nextrecord;
   1495       1.69   thorpej 					if (nextrecord == NULL)
   1496       1.69   thorpej 						so->so_rcv.sb_lastrecord = m;
   1497       1.69   thorpej 				} else {
   1498       1.69   thorpej 					so->so_rcv.sb_mb = nextrecord;
   1499       1.70   thorpej 					SB_EMPTY_FIXUP(&so->so_rcv);
   1500       1.69   thorpej 				}
   1501       1.69   thorpej 				SBLASTRECORDCHK(&so->so_rcv, "soreceive 3");
   1502       1.69   thorpej 				SBLASTMBUFCHK(&so->so_rcv, "soreceive 3");
   1503        1.1       cgd 			}
   1504      1.144    dyoung 		} else if (flags & MSG_PEEK)
   1505      1.144    dyoung 			moff += len;
   1506      1.144    dyoung 		else {
   1507      1.160        ad 			if (mp != NULL) {
   1508      1.160        ad 				mt = m_copym(m, 0, len, M_NOWAIT);
   1509      1.160        ad 				if (__predict_false(mt == NULL)) {
   1510      1.160        ad 					sounlock(so);
   1511      1.160        ad 					mt = m_copym(m, 0, len, M_WAIT);
   1512      1.160        ad 					solock(so);
   1513      1.160        ad 				}
   1514      1.160        ad 				*mp = mt;
   1515      1.160        ad 			}
   1516      1.144    dyoung 			m->m_data += len;
   1517      1.144    dyoung 			m->m_len -= len;
   1518      1.144    dyoung 			so->so_rcv.sb_cc -= len;
   1519        1.1       cgd 		}
   1520        1.1       cgd 		if (so->so_oobmark) {
   1521        1.1       cgd 			if ((flags & MSG_PEEK) == 0) {
   1522        1.1       cgd 				so->so_oobmark -= len;
   1523        1.1       cgd 				if (so->so_oobmark == 0) {
   1524        1.1       cgd 					so->so_state |= SS_RCVATMARK;
   1525        1.1       cgd 					break;
   1526        1.1       cgd 				}
   1527        1.7       cgd 			} else {
   1528        1.1       cgd 				offset += len;
   1529        1.7       cgd 				if (offset == so->so_oobmark)
   1530        1.7       cgd 					break;
   1531        1.7       cgd 			}
   1532        1.1       cgd 		}
   1533        1.1       cgd 		if (flags & MSG_EOR)
   1534        1.1       cgd 			break;
   1535        1.1       cgd 		/*
   1536        1.1       cgd 		 * If the MSG_WAITALL flag is set (for non-atomic socket),
   1537        1.1       cgd 		 * we must not quit until "uio->uio_resid == 0" or an error
   1538        1.1       cgd 		 * termination.  If a signal/timeout occurs, return
   1539        1.1       cgd 		 * with a short count but without error.
   1540        1.1       cgd 		 * Keep sockbuf locked against other readers.
   1541        1.1       cgd 		 */
   1542      1.144    dyoung 		while (flags & MSG_WAITALL && m == NULL && uio->uio_resid > 0 &&
   1543        1.3    andrew 		    !sosendallatonce(so) && !nextrecord) {
   1544        1.1       cgd 			if (so->so_error || so->so_state & SS_CANTRCVMORE)
   1545        1.1       cgd 				break;
   1546       1.68      matt 			/*
   1547       1.68      matt 			 * If we are peeking and the socket receive buffer is
   1548       1.68      matt 			 * full, stop since we can't get more data to peek at.
   1549       1.68      matt 			 */
   1550       1.68      matt 			if ((flags & MSG_PEEK) && sbspace(&so->so_rcv) <= 0)
   1551       1.68      matt 				break;
   1552       1.68      matt 			/*
   1553       1.68      matt 			 * If we've drained the socket buffer, tell the
   1554       1.68      matt 			 * protocol in case it needs to do something to
   1555       1.68      matt 			 * get it filled again.
   1556       1.68      matt 			 */
   1557       1.68      matt 			if ((pr->pr_flags & PR_WANTRCVD) && so->so_pcb)
   1558       1.68      matt 				(*pr->pr_usrreq)(so, PRU_RCVD,
   1559      1.140    dyoung 				    NULL, (struct mbuf *)(long)flags, NULL, l);
   1560       1.69   thorpej 			SBLASTRECORDCHK(&so->so_rcv, "soreceive sbwait 2");
   1561       1.69   thorpej 			SBLASTMBUFCHK(&so->so_rcv, "soreceive sbwait 2");
   1562      1.196       dsl 			if (wakeup_state & SS_RESTARTSYS)
   1563      1.196       dsl 				error = ERESTART;
   1564      1.196       dsl 			else
   1565      1.196       dsl 				error = sbwait(&so->so_rcv);
   1566      1.144    dyoung 			if (error != 0) {
   1567        1.1       cgd 				sbunlock(&so->so_rcv);
   1568      1.160        ad 				sounlock(so);
   1569        1.1       cgd 				splx(s);
   1570      1.144    dyoung 				return 0;
   1571        1.1       cgd 			}
   1572       1.21  christos 			if ((m = so->so_rcv.sb_mb) != NULL)
   1573        1.1       cgd 				nextrecord = m->m_nextpkt;
   1574      1.196       dsl 			wakeup_state = so->so_state;
   1575        1.1       cgd 		}
   1576        1.1       cgd 	}
   1577        1.3    andrew 
   1578      1.146    dyoung 	if (m && atomic) {
   1579        1.3    andrew 		flags |= MSG_TRUNC;
   1580        1.3    andrew 		if ((flags & MSG_PEEK) == 0)
   1581        1.3    andrew 			(void) sbdroprecord(&so->so_rcv);
   1582        1.3    andrew 	}
   1583        1.1       cgd 	if ((flags & MSG_PEEK) == 0) {
   1584      1.144    dyoung 		if (m == NULL) {
   1585       1.69   thorpej 			/*
   1586       1.70   thorpej 			 * First part is an inline SB_EMPTY_FIXUP().  Second
   1587       1.69   thorpej 			 * part makes sure sb_lastrecord is up-to-date if
   1588       1.69   thorpej 			 * there is still data in the socket buffer.
   1589       1.69   thorpej 			 */
   1590        1.1       cgd 			so->so_rcv.sb_mb = nextrecord;
   1591       1.69   thorpej 			if (so->so_rcv.sb_mb == NULL) {
   1592       1.69   thorpej 				so->so_rcv.sb_mbtail = NULL;
   1593       1.69   thorpej 				so->so_rcv.sb_lastrecord = NULL;
   1594       1.69   thorpej 			} else if (nextrecord->m_nextpkt == NULL)
   1595       1.69   thorpej 				so->so_rcv.sb_lastrecord = nextrecord;
   1596       1.69   thorpej 		}
   1597       1.69   thorpej 		SBLASTRECORDCHK(&so->so_rcv, "soreceive 4");
   1598       1.69   thorpej 		SBLASTMBUFCHK(&so->so_rcv, "soreceive 4");
   1599        1.1       cgd 		if (pr->pr_flags & PR_WANTRCVD && so->so_pcb)
   1600      1.140    dyoung 			(*pr->pr_usrreq)(so, PRU_RCVD, NULL,
   1601      1.140    dyoung 			    (struct mbuf *)(long)flags, NULL, l);
   1602        1.1       cgd 	}
   1603        1.3    andrew 	if (orig_resid == uio->uio_resid && orig_resid &&
   1604        1.3    andrew 	    (flags & MSG_EOR) == 0 && (so->so_state & SS_CANTRCVMORE) == 0) {
   1605        1.3    andrew 		sbunlock(&so->so_rcv);
   1606        1.3    andrew 		goto restart;
   1607        1.3    andrew 	}
   1608      1.108     perry 
   1609      1.144    dyoung 	if (flagsp != NULL)
   1610        1.1       cgd 		*flagsp |= flags;
   1611       1.54     lukem  release:
   1612        1.1       cgd 	sbunlock(&so->so_rcv);
   1613      1.160        ad 	sounlock(so);
   1614        1.1       cgd 	splx(s);
   1615      1.144    dyoung 	return error;
   1616        1.1       cgd }
   1617        1.1       cgd 
   1618       1.14   mycroft int
   1619       1.54     lukem soshutdown(struct socket *so, int how)
   1620        1.1       cgd {
   1621       1.99      matt 	const struct protosw	*pr;
   1622      1.160        ad 	int	error;
   1623      1.160        ad 
   1624      1.160        ad 	KASSERT(solocked(so));
   1625       1.34    kleink 
   1626       1.54     lukem 	pr = so->so_proto;
   1627       1.34    kleink 	if (!(how == SHUT_RD || how == SHUT_WR || how == SHUT_RDWR))
   1628       1.34    kleink 		return (EINVAL);
   1629        1.1       cgd 
   1630      1.160        ad 	if (how == SHUT_RD || how == SHUT_RDWR) {
   1631        1.1       cgd 		sorflush(so);
   1632      1.160        ad 		error = 0;
   1633      1.160        ad 	}
   1634       1.34    kleink 	if (how == SHUT_WR || how == SHUT_RDWR)
   1635      1.160        ad 		error = (*pr->pr_usrreq)(so, PRU_SHUTDOWN, NULL,
   1636      1.140    dyoung 		    NULL, NULL, NULL);
   1637      1.160        ad 
   1638      1.160        ad 	return error;
   1639        1.1       cgd }
   1640        1.1       cgd 
   1641      1.195       dsl void
   1642      1.196       dsl sorestart(struct socket *so)
   1643      1.188        ad {
   1644      1.196       dsl 	/*
   1645      1.196       dsl 	 * An application has called close() on an fd on which another
   1646      1.196       dsl 	 * of its threads has called a socket system call.
   1647      1.196       dsl 	 * Mark this and wake everyone up, and code that would block again
   1648      1.196       dsl 	 * instead returns ERESTART.
   1649      1.196       dsl 	 * On system call re-entry the fd is validated and EBADF returned.
   1650      1.196       dsl 	 * Any other fd will block again on the 2nd syscall.
   1651      1.196       dsl 	 */
   1652      1.188        ad 	solock(so);
   1653      1.196       dsl 	so->so_state |= SS_RESTARTSYS;
   1654      1.188        ad 	cv_broadcast(&so->so_cv);
   1655      1.196       dsl 	cv_broadcast(&so->so_snd.sb_cv);
   1656      1.196       dsl 	cv_broadcast(&so->so_rcv.sb_cv);
   1657      1.188        ad 	sounlock(so);
   1658      1.188        ad }
   1659      1.188        ad 
   1660       1.14   mycroft void
   1661       1.54     lukem sorflush(struct socket *so)
   1662        1.1       cgd {
   1663       1.54     lukem 	struct sockbuf	*sb, asb;
   1664       1.99      matt 	const struct protosw	*pr;
   1665      1.160        ad 
   1666      1.160        ad 	KASSERT(solocked(so));
   1667        1.1       cgd 
   1668       1.54     lukem 	sb = &so->so_rcv;
   1669       1.54     lukem 	pr = so->so_proto;
   1670      1.160        ad 	socantrcvmore(so);
   1671        1.1       cgd 	sb->sb_flags |= SB_NOINTR;
   1672      1.160        ad 	(void )sblock(sb, M_WAITOK);
   1673        1.1       cgd 	sbunlock(sb);
   1674        1.1       cgd 	asb = *sb;
   1675       1.86  wrstuden 	/*
   1676       1.86  wrstuden 	 * Clear most of the sockbuf structure, but leave some of the
   1677       1.86  wrstuden 	 * fields valid.
   1678       1.86  wrstuden 	 */
   1679       1.86  wrstuden 	memset(&sb->sb_startzero, 0,
   1680       1.86  wrstuden 	    sizeof(*sb) - offsetof(struct sockbuf, sb_startzero));
   1681      1.160        ad 	if (pr->pr_flags & PR_RIGHTS && pr->pr_domain->dom_dispose) {
   1682      1.160        ad 		sounlock(so);
   1683        1.1       cgd 		(*pr->pr_domain->dom_dispose)(asb.sb_mb);
   1684      1.160        ad 		solock(so);
   1685      1.160        ad 	}
   1686       1.98  christos 	sbrelease(&asb, so);
   1687        1.1       cgd }
   1688        1.1       cgd 
   1689      1.171    plunky /*
   1690      1.171    plunky  * internal set SOL_SOCKET options
   1691      1.171    plunky  */
   1692      1.142    dyoung static int
   1693      1.171    plunky sosetopt1(struct socket *so, const struct sockopt *sopt)
   1694        1.1       cgd {
   1695      1.182  christos 	int error = EINVAL, optval, opt;
   1696      1.171    plunky 	struct linger l;
   1697      1.171    plunky 	struct timeval tv;
   1698      1.142    dyoung 
   1699      1.179  christos 	switch ((opt = sopt->sopt_name)) {
   1700      1.142    dyoung 
   1701      1.170       tls 	case SO_ACCEPTFILTER:
   1702      1.177        ad 		error = accept_filt_setopt(so, sopt);
   1703      1.177        ad 		KASSERT(solocked(so));
   1704      1.170       tls 		break;
   1705      1.170       tls 
   1706      1.171    plunky   	case SO_LINGER:
   1707      1.171    plunky  		error = sockopt_get(sopt, &l, sizeof(l));
   1708      1.177        ad 		solock(so);
   1709      1.171    plunky  		if (error)
   1710      1.177        ad  			break;
   1711      1.171    plunky  		if (l.l_linger < 0 || l.l_linger > USHRT_MAX ||
   1712      1.177        ad  		    l.l_linger > (INT_MAX / hz)) {
   1713      1.177        ad 			error = EDOM;
   1714      1.177        ad 			break;
   1715      1.177        ad 		}
   1716      1.171    plunky  		so->so_linger = l.l_linger;
   1717      1.171    plunky  		if (l.l_onoff)
   1718      1.171    plunky  			so->so_options |= SO_LINGER;
   1719      1.171    plunky  		else
   1720      1.171    plunky  			so->so_options &= ~SO_LINGER;
   1721      1.177        ad    		break;
   1722        1.1       cgd 
   1723      1.142    dyoung 	case SO_DEBUG:
   1724      1.142    dyoung 	case SO_KEEPALIVE:
   1725      1.142    dyoung 	case SO_DONTROUTE:
   1726      1.142    dyoung 	case SO_USELOOPBACK:
   1727      1.142    dyoung 	case SO_BROADCAST:
   1728      1.142    dyoung 	case SO_REUSEADDR:
   1729      1.142    dyoung 	case SO_REUSEPORT:
   1730      1.142    dyoung 	case SO_OOBINLINE:
   1731      1.142    dyoung 	case SO_TIMESTAMP:
   1732      1.184  christos #ifdef SO_OTIMESTAMP
   1733      1.184  christos 	case SO_OTIMESTAMP:
   1734      1.184  christos #endif
   1735      1.171    plunky 		error = sockopt_getint(sopt, &optval);
   1736      1.177        ad 		solock(so);
   1737      1.171    plunky 		if (error)
   1738      1.177        ad 			break;
   1739      1.171    plunky 		if (optval)
   1740      1.179  christos 			so->so_options |= opt;
   1741      1.142    dyoung 		else
   1742      1.179  christos 			so->so_options &= ~opt;
   1743      1.142    dyoung 		break;
   1744      1.142    dyoung 
   1745      1.142    dyoung 	case SO_SNDBUF:
   1746      1.142    dyoung 	case SO_RCVBUF:
   1747      1.142    dyoung 	case SO_SNDLOWAT:
   1748      1.142    dyoung 	case SO_RCVLOWAT:
   1749      1.171    plunky 		error = sockopt_getint(sopt, &optval);
   1750      1.177        ad 		solock(so);
   1751      1.171    plunky 		if (error)
   1752      1.177        ad 			break;
   1753        1.1       cgd 
   1754      1.142    dyoung 		/*
   1755      1.142    dyoung 		 * Values < 1 make no sense for any of these
   1756      1.142    dyoung 		 * options, so disallow them.
   1757      1.142    dyoung 		 */
   1758      1.177        ad 		if (optval < 1) {
   1759      1.177        ad 			error = EINVAL;
   1760      1.177        ad 			break;
   1761      1.177        ad 		}
   1762        1.1       cgd 
   1763      1.179  christos 		switch (opt) {
   1764      1.171    plunky 		case SO_SNDBUF:
   1765      1.177        ad 			if (sbreserve(&so->so_snd, (u_long)optval, so) == 0) {
   1766      1.177        ad 				error = ENOBUFS;
   1767      1.177        ad 				break;
   1768      1.177        ad 			}
   1769      1.171    plunky 			so->so_snd.sb_flags &= ~SB_AUTOSIZE;
   1770      1.171    plunky 			break;
   1771        1.1       cgd 
   1772        1.1       cgd 		case SO_RCVBUF:
   1773      1.177        ad 			if (sbreserve(&so->so_rcv, (u_long)optval, so) == 0) {
   1774      1.177        ad 				error = ENOBUFS;
   1775      1.177        ad 				break;
   1776      1.177        ad 			}
   1777      1.171    plunky 			so->so_rcv.sb_flags &= ~SB_AUTOSIZE;
   1778      1.142    dyoung 			break;
   1779      1.142    dyoung 
   1780      1.142    dyoung 		/*
   1781      1.142    dyoung 		 * Make sure the low-water is never greater than
   1782      1.142    dyoung 		 * the high-water.
   1783      1.142    dyoung 		 */
   1784        1.1       cgd 		case SO_SNDLOWAT:
   1785      1.171    plunky 			if (optval > so->so_snd.sb_hiwat)
   1786      1.171    plunky 				optval = so->so_snd.sb_hiwat;
   1787      1.171    plunky 
   1788      1.171    plunky 			so->so_snd.sb_lowat = optval;
   1789      1.142    dyoung 			break;
   1790      1.171    plunky 
   1791        1.1       cgd 		case SO_RCVLOWAT:
   1792      1.171    plunky 			if (optval > so->so_rcv.sb_hiwat)
   1793      1.171    plunky 				optval = so->so_rcv.sb_hiwat;
   1794      1.171    plunky 
   1795      1.171    plunky 			so->so_rcv.sb_lowat = optval;
   1796      1.142    dyoung 			break;
   1797      1.142    dyoung 		}
   1798      1.142    dyoung 		break;
   1799       1.28   thorpej 
   1800      1.179  christos #ifdef COMPAT_50
   1801      1.179  christos 	case SO_OSNDTIMEO:
   1802      1.179  christos 	case SO_ORCVTIMEO: {
   1803      1.179  christos 		struct timeval50 otv;
   1804      1.179  christos 		error = sockopt_get(sopt, &otv, sizeof(otv));
   1805      1.186     pooka 		if (error) {
   1806      1.186     pooka 			solock(so);
   1807      1.183  christos 			break;
   1808      1.186     pooka 		}
   1809      1.179  christos 		timeval50_to_timeval(&otv, &tv);
   1810      1.179  christos 		opt = opt == SO_OSNDTIMEO ? SO_SNDTIMEO : SO_RCVTIMEO;
   1811      1.182  christos 		error = 0;
   1812      1.179  christos 		/*FALLTHROUGH*/
   1813      1.179  christos 	}
   1814      1.179  christos #endif /* COMPAT_50 */
   1815      1.179  christos 
   1816      1.142    dyoung 	case SO_SNDTIMEO:
   1817      1.142    dyoung 	case SO_RCVTIMEO:
   1818      1.182  christos 		if (error)
   1819      1.179  christos 			error = sockopt_get(sopt, &tv, sizeof(tv));
   1820      1.177        ad 		solock(so);
   1821      1.171    plunky 		if (error)
   1822      1.177        ad 			break;
   1823      1.171    plunky 
   1824      1.177        ad 		if (tv.tv_sec > (INT_MAX - tv.tv_usec / tick) / hz) {
   1825      1.177        ad 			error = EDOM;
   1826      1.177        ad 			break;
   1827      1.177        ad 		}
   1828       1.28   thorpej 
   1829      1.171    plunky 		optval = tv.tv_sec * hz + tv.tv_usec / tick;
   1830      1.171    plunky 		if (optval == 0 && tv.tv_usec != 0)
   1831      1.171    plunky 			optval = 1;
   1832       1.28   thorpej 
   1833      1.179  christos 		switch (opt) {
   1834      1.142    dyoung 		case SO_SNDTIMEO:
   1835      1.171    plunky 			so->so_snd.sb_timeo = optval;
   1836        1.1       cgd 			break;
   1837        1.1       cgd 		case SO_RCVTIMEO:
   1838      1.171    plunky 			so->so_rcv.sb_timeo = optval;
   1839      1.142    dyoung 			break;
   1840      1.142    dyoung 		}
   1841      1.142    dyoung 		break;
   1842        1.1       cgd 
   1843      1.142    dyoung 	default:
   1844      1.177        ad 		solock(so);
   1845      1.177        ad 		error = ENOPROTOOPT;
   1846      1.177        ad 		break;
   1847      1.142    dyoung 	}
   1848      1.177        ad 	KASSERT(solocked(so));
   1849      1.177        ad 	return error;
   1850      1.142    dyoung }
   1851        1.1       cgd 
   1852      1.142    dyoung int
   1853      1.171    plunky sosetopt(struct socket *so, struct sockopt *sopt)
   1854      1.142    dyoung {
   1855      1.142    dyoung 	int error, prerr;
   1856        1.1       cgd 
   1857      1.177        ad 	if (sopt->sopt_level == SOL_SOCKET) {
   1858      1.171    plunky 		error = sosetopt1(so, sopt);
   1859      1.177        ad 		KASSERT(solocked(so));
   1860      1.177        ad 	} else {
   1861      1.142    dyoung 		error = ENOPROTOOPT;
   1862      1.177        ad 		solock(so);
   1863      1.177        ad 	}
   1864        1.1       cgd 
   1865      1.142    dyoung 	if ((error == 0 || error == ENOPROTOOPT) &&
   1866      1.142    dyoung 	    so->so_proto != NULL && so->so_proto->pr_ctloutput != NULL) {
   1867      1.142    dyoung 		/* give the protocol stack a shot */
   1868      1.171    plunky 		prerr = (*so->so_proto->pr_ctloutput)(PRCO_SETOPT, so, sopt);
   1869      1.142    dyoung 		if (prerr == 0)
   1870      1.142    dyoung 			error = 0;
   1871      1.142    dyoung 		else if (prerr != ENOPROTOOPT)
   1872      1.142    dyoung 			error = prerr;
   1873      1.171    plunky 	}
   1874      1.160        ad 	sounlock(so);
   1875      1.142    dyoung 	return error;
   1876        1.1       cgd }
   1877        1.1       cgd 
   1878      1.171    plunky /*
   1879      1.171    plunky  * so_setsockopt() is a wrapper providing a sockopt structure for sosetopt()
   1880      1.171    plunky  */
   1881      1.171    plunky int
   1882      1.171    plunky so_setsockopt(struct lwp *l, struct socket *so, int level, int name,
   1883      1.171    plunky     const void *val, size_t valsize)
   1884      1.171    plunky {
   1885      1.171    plunky 	struct sockopt sopt;
   1886      1.171    plunky 	int error;
   1887      1.171    plunky 
   1888      1.171    plunky 	KASSERT(valsize == 0 || val != NULL);
   1889      1.171    plunky 
   1890      1.171    plunky 	sockopt_init(&sopt, level, name, valsize);
   1891      1.171    plunky 	sockopt_set(&sopt, val, valsize);
   1892      1.171    plunky 
   1893      1.171    plunky 	error = sosetopt(so, &sopt);
   1894      1.171    plunky 
   1895      1.171    plunky 	sockopt_destroy(&sopt);
   1896      1.171    plunky 
   1897      1.171    plunky 	return error;
   1898      1.171    plunky }
   1899      1.171    plunky 
   1900      1.171    plunky /*
   1901      1.171    plunky  * internal get SOL_SOCKET options
   1902      1.171    plunky  */
   1903      1.171    plunky static int
   1904      1.171    plunky sogetopt1(struct socket *so, struct sockopt *sopt)
   1905      1.171    plunky {
   1906      1.179  christos 	int error, optval, opt;
   1907      1.171    plunky 	struct linger l;
   1908      1.171    plunky 	struct timeval tv;
   1909      1.171    plunky 
   1910      1.179  christos 	switch ((opt = sopt->sopt_name)) {
   1911      1.171    plunky 
   1912      1.171    plunky 	case SO_ACCEPTFILTER:
   1913      1.177        ad 		error = accept_filt_getopt(so, sopt);
   1914      1.171    plunky 		break;
   1915      1.171    plunky 
   1916      1.171    plunky 	case SO_LINGER:
   1917      1.171    plunky 		l.l_onoff = (so->so_options & SO_LINGER) ? 1 : 0;
   1918      1.171    plunky 		l.l_linger = so->so_linger;
   1919      1.171    plunky 
   1920      1.171    plunky 		error = sockopt_set(sopt, &l, sizeof(l));
   1921      1.171    plunky 		break;
   1922      1.171    plunky 
   1923      1.171    plunky 	case SO_USELOOPBACK:
   1924      1.171    plunky 	case SO_DONTROUTE:
   1925      1.171    plunky 	case SO_DEBUG:
   1926      1.171    plunky 	case SO_KEEPALIVE:
   1927      1.171    plunky 	case SO_REUSEADDR:
   1928      1.171    plunky 	case SO_REUSEPORT:
   1929      1.171    plunky 	case SO_BROADCAST:
   1930      1.171    plunky 	case SO_OOBINLINE:
   1931      1.171    plunky 	case SO_TIMESTAMP:
   1932      1.184  christos #ifdef SO_OTIMESTAMP
   1933      1.184  christos 	case SO_OTIMESTAMP:
   1934      1.184  christos #endif
   1935      1.179  christos 		error = sockopt_setint(sopt, (so->so_options & opt) ? 1 : 0);
   1936      1.171    plunky 		break;
   1937      1.171    plunky 
   1938      1.171    plunky 	case SO_TYPE:
   1939      1.171    plunky 		error = sockopt_setint(sopt, so->so_type);
   1940      1.171    plunky 		break;
   1941      1.171    plunky 
   1942      1.171    plunky 	case SO_ERROR:
   1943      1.171    plunky 		error = sockopt_setint(sopt, so->so_error);
   1944      1.171    plunky 		so->so_error = 0;
   1945      1.171    plunky 		break;
   1946      1.171    plunky 
   1947      1.171    plunky 	case SO_SNDBUF:
   1948      1.171    plunky 		error = sockopt_setint(sopt, so->so_snd.sb_hiwat);
   1949      1.171    plunky 		break;
   1950      1.171    plunky 
   1951      1.171    plunky 	case SO_RCVBUF:
   1952      1.171    plunky 		error = sockopt_setint(sopt, so->so_rcv.sb_hiwat);
   1953      1.171    plunky 		break;
   1954      1.171    plunky 
   1955      1.171    plunky 	case SO_SNDLOWAT:
   1956      1.171    plunky 		error = sockopt_setint(sopt, so->so_snd.sb_lowat);
   1957      1.171    plunky 		break;
   1958      1.171    plunky 
   1959      1.171    plunky 	case SO_RCVLOWAT:
   1960      1.171    plunky 		error = sockopt_setint(sopt, so->so_rcv.sb_lowat);
   1961      1.171    plunky 		break;
   1962      1.171    plunky 
   1963      1.179  christos #ifdef COMPAT_50
   1964      1.179  christos 	case SO_OSNDTIMEO:
   1965      1.179  christos 	case SO_ORCVTIMEO: {
   1966      1.179  christos 		struct timeval50 otv;
   1967      1.179  christos 
   1968      1.179  christos 		optval = (opt == SO_OSNDTIMEO ?
   1969      1.179  christos 		     so->so_snd.sb_timeo : so->so_rcv.sb_timeo);
   1970      1.179  christos 
   1971      1.179  christos 		otv.tv_sec = optval / hz;
   1972      1.179  christos 		otv.tv_usec = (optval % hz) * tick;
   1973      1.179  christos 
   1974      1.179  christos 		error = sockopt_set(sopt, &otv, sizeof(otv));
   1975      1.179  christos 		break;
   1976      1.179  christos 	}
   1977      1.179  christos #endif /* COMPAT_50 */
   1978      1.179  christos 
   1979      1.171    plunky 	case SO_SNDTIMEO:
   1980      1.171    plunky 	case SO_RCVTIMEO:
   1981      1.179  christos 		optval = (opt == SO_SNDTIMEO ?
   1982      1.171    plunky 		     so->so_snd.sb_timeo : so->so_rcv.sb_timeo);
   1983      1.171    plunky 
   1984      1.171    plunky 		tv.tv_sec = optval / hz;
   1985      1.171    plunky 		tv.tv_usec = (optval % hz) * tick;
   1986      1.171    plunky 
   1987      1.171    plunky 		error = sockopt_set(sopt, &tv, sizeof(tv));
   1988      1.171    plunky 		break;
   1989      1.171    plunky 
   1990      1.171    plunky 	case SO_OVERFLOWED:
   1991      1.171    plunky 		error = sockopt_setint(sopt, so->so_rcv.sb_overflowed);
   1992      1.171    plunky 		break;
   1993      1.171    plunky 
   1994      1.171    plunky 	default:
   1995      1.171    plunky 		error = ENOPROTOOPT;
   1996      1.171    plunky 		break;
   1997      1.171    plunky 	}
   1998      1.171    plunky 
   1999      1.171    plunky 	return (error);
   2000      1.171    plunky }
   2001      1.171    plunky 
   2002       1.14   mycroft int
   2003      1.171    plunky sogetopt(struct socket *so, struct sockopt *sopt)
   2004        1.1       cgd {
   2005      1.160        ad 	int		error;
   2006        1.1       cgd 
   2007      1.160        ad 	solock(so);
   2008      1.171    plunky 	if (sopt->sopt_level != SOL_SOCKET) {
   2009        1.1       cgd 		if (so->so_proto && so->so_proto->pr_ctloutput) {
   2010      1.160        ad 			error = ((*so->so_proto->pr_ctloutput)
   2011      1.171    plunky 			    (PRCO_GETOPT, so, sopt));
   2012        1.1       cgd 		} else
   2013      1.160        ad 			error = (ENOPROTOOPT);
   2014        1.1       cgd 	} else {
   2015      1.171    plunky 		error = sogetopt1(so, sopt);
   2016      1.171    plunky 	}
   2017      1.171    plunky 	sounlock(so);
   2018      1.171    plunky 	return (error);
   2019      1.171    plunky }
   2020      1.171    plunky 
   2021      1.171    plunky /*
   2022      1.171    plunky  * alloc sockopt data buffer buffer
   2023      1.171    plunky  *	- will be released at destroy
   2024      1.171    plunky  */
   2025      1.176    plunky static int
   2026      1.176    plunky sockopt_alloc(struct sockopt *sopt, size_t len, km_flag_t kmflag)
   2027      1.171    plunky {
   2028      1.171    plunky 
   2029      1.171    plunky 	KASSERT(sopt->sopt_size == 0);
   2030      1.171    plunky 
   2031      1.176    plunky 	if (len > sizeof(sopt->sopt_buf)) {
   2032      1.176    plunky 		sopt->sopt_data = kmem_zalloc(len, kmflag);
   2033      1.176    plunky 		if (sopt->sopt_data == NULL)
   2034      1.176    plunky 			return ENOMEM;
   2035      1.176    plunky 	} else
   2036      1.171    plunky 		sopt->sopt_data = sopt->sopt_buf;
   2037      1.171    plunky 
   2038      1.171    plunky 	sopt->sopt_size = len;
   2039      1.176    plunky 	return 0;
   2040      1.171    plunky }
   2041      1.171    plunky 
   2042      1.171    plunky /*
   2043      1.171    plunky  * initialise sockopt storage
   2044      1.176    plunky  *	- MAY sleep during allocation
   2045      1.171    plunky  */
   2046      1.171    plunky void
   2047      1.171    plunky sockopt_init(struct sockopt *sopt, int level, int name, size_t size)
   2048      1.171    plunky {
   2049        1.1       cgd 
   2050      1.171    plunky 	memset(sopt, 0, sizeof(*sopt));
   2051        1.1       cgd 
   2052      1.171    plunky 	sopt->sopt_level = level;
   2053      1.171    plunky 	sopt->sopt_name = name;
   2054      1.176    plunky 	(void)sockopt_alloc(sopt, size, KM_SLEEP);
   2055      1.171    plunky }
   2056      1.171    plunky 
   2057      1.171    plunky /*
   2058      1.171    plunky  * destroy sockopt storage
   2059      1.171    plunky  *	- will release any held memory references
   2060      1.171    plunky  */
   2061      1.171    plunky void
   2062      1.171    plunky sockopt_destroy(struct sockopt *sopt)
   2063      1.171    plunky {
   2064      1.171    plunky 
   2065      1.171    plunky 	if (sopt->sopt_data != sopt->sopt_buf)
   2066      1.173    plunky 		kmem_free(sopt->sopt_data, sopt->sopt_size);
   2067      1.171    plunky 
   2068      1.171    plunky 	memset(sopt, 0, sizeof(*sopt));
   2069      1.171    plunky }
   2070      1.171    plunky 
   2071      1.171    plunky /*
   2072      1.171    plunky  * set sockopt value
   2073      1.171    plunky  *	- value is copied into sockopt
   2074      1.176    plunky  * 	- memory is allocated when necessary, will not sleep
   2075      1.171    plunky  */
   2076      1.171    plunky int
   2077      1.171    plunky sockopt_set(struct sockopt *sopt, const void *buf, size_t len)
   2078      1.171    plunky {
   2079      1.176    plunky 	int error;
   2080      1.171    plunky 
   2081      1.176    plunky 	if (sopt->sopt_size == 0) {
   2082      1.176    plunky 		error = sockopt_alloc(sopt, len, KM_NOSLEEP);
   2083      1.176    plunky 		if (error)
   2084      1.176    plunky 			return error;
   2085      1.176    plunky 	}
   2086      1.171    plunky 
   2087      1.171    plunky 	KASSERT(sopt->sopt_size == len);
   2088      1.171    plunky 	memcpy(sopt->sopt_data, buf, len);
   2089      1.171    plunky 	return 0;
   2090      1.171    plunky }
   2091      1.171    plunky 
   2092      1.171    plunky /*
   2093      1.171    plunky  * common case of set sockopt integer value
   2094      1.171    plunky  */
   2095      1.171    plunky int
   2096      1.171    plunky sockopt_setint(struct sockopt *sopt, int val)
   2097      1.171    plunky {
   2098      1.171    plunky 
   2099      1.171    plunky 	return sockopt_set(sopt, &val, sizeof(int));
   2100      1.171    plunky }
   2101      1.171    plunky 
   2102      1.171    plunky /*
   2103      1.171    plunky  * get sockopt value
   2104      1.171    plunky  *	- correct size must be given
   2105      1.171    plunky  */
   2106      1.171    plunky int
   2107      1.171    plunky sockopt_get(const struct sockopt *sopt, void *buf, size_t len)
   2108      1.171    plunky {
   2109      1.170       tls 
   2110      1.171    plunky 	if (sopt->sopt_size != len)
   2111      1.171    plunky 		return EINVAL;
   2112        1.1       cgd 
   2113      1.171    plunky 	memcpy(buf, sopt->sopt_data, len);
   2114      1.171    plunky 	return 0;
   2115      1.171    plunky }
   2116        1.1       cgd 
   2117      1.171    plunky /*
   2118      1.171    plunky  * common case of get sockopt integer value
   2119      1.171    plunky  */
   2120      1.171    plunky int
   2121      1.171    plunky sockopt_getint(const struct sockopt *sopt, int *valp)
   2122      1.171    plunky {
   2123        1.1       cgd 
   2124      1.171    plunky 	return sockopt_get(sopt, valp, sizeof(int));
   2125      1.171    plunky }
   2126        1.1       cgd 
   2127      1.171    plunky /*
   2128      1.171    plunky  * set sockopt value from mbuf
   2129      1.171    plunky  *	- ONLY for legacy code
   2130      1.171    plunky  *	- mbuf is released by sockopt
   2131      1.176    plunky  *	- will not sleep
   2132      1.171    plunky  */
   2133      1.171    plunky int
   2134      1.171    plunky sockopt_setmbuf(struct sockopt *sopt, struct mbuf *m)
   2135      1.171    plunky {
   2136      1.171    plunky 	size_t len;
   2137      1.176    plunky 	int error;
   2138        1.1       cgd 
   2139      1.171    plunky 	len = m_length(m);
   2140        1.1       cgd 
   2141      1.176    plunky 	if (sopt->sopt_size == 0) {
   2142      1.176    plunky 		error = sockopt_alloc(sopt, len, KM_NOSLEEP);
   2143      1.176    plunky 		if (error)
   2144      1.176    plunky 			return error;
   2145      1.176    plunky 	}
   2146        1.1       cgd 
   2147      1.171    plunky 	KASSERT(sopt->sopt_size == len);
   2148      1.171    plunky 	m_copydata(m, 0, len, sopt->sopt_data);
   2149      1.171    plunky 	m_freem(m);
   2150        1.1       cgd 
   2151      1.171    plunky 	return 0;
   2152      1.171    plunky }
   2153        1.1       cgd 
   2154      1.171    plunky /*
   2155      1.171    plunky  * get sockopt value into mbuf
   2156      1.171    plunky  *	- ONLY for legacy code
   2157      1.171    plunky  *	- mbuf to be released by the caller
   2158      1.176    plunky  *	- will not sleep
   2159      1.171    plunky  */
   2160      1.171    plunky struct mbuf *
   2161      1.171    plunky sockopt_getmbuf(const struct sockopt *sopt)
   2162      1.171    plunky {
   2163      1.171    plunky 	struct mbuf *m;
   2164      1.107   darrenr 
   2165      1.176    plunky 	if (sopt->sopt_size > MCLBYTES)
   2166      1.176    plunky 		return NULL;
   2167      1.176    plunky 
   2168      1.176    plunky 	m = m_get(M_DONTWAIT, MT_SOOPTS);
   2169      1.171    plunky 	if (m == NULL)
   2170      1.171    plunky 		return NULL;
   2171      1.171    plunky 
   2172      1.176    plunky 	if (sopt->sopt_size > MLEN) {
   2173      1.176    plunky 		MCLGET(m, M_DONTWAIT);
   2174      1.176    plunky 		if ((m->m_flags & M_EXT) == 0) {
   2175      1.176    plunky 			m_free(m);
   2176      1.176    plunky 			return NULL;
   2177      1.176    plunky 		}
   2178        1.1       cgd 	}
   2179      1.176    plunky 
   2180      1.176    plunky 	memcpy(mtod(m, void *), sopt->sopt_data, sopt->sopt_size);
   2181      1.176    plunky 	m->m_len = sopt->sopt_size;
   2182      1.160        ad 
   2183      1.171    plunky 	return m;
   2184        1.1       cgd }
   2185        1.1       cgd 
   2186       1.14   mycroft void
   2187       1.54     lukem sohasoutofband(struct socket *so)
   2188        1.1       cgd {
   2189      1.153     rmind 
   2190       1.90  christos 	fownsignal(so->so_pgid, SIGURG, POLL_PRI, POLLPRI|POLLRDBAND, so);
   2191      1.189        ad 	selnotify(&so->so_rcv.sb_sel, POLLPRI | POLLRDBAND, NOTE_SUBMIT);
   2192        1.1       cgd }
   2193       1.72  jdolecek 
   2194       1.72  jdolecek static void
   2195       1.72  jdolecek filt_sordetach(struct knote *kn)
   2196       1.72  jdolecek {
   2197       1.72  jdolecek 	struct socket	*so;
   2198       1.72  jdolecek 
   2199      1.155        ad 	so = ((file_t *)kn->kn_obj)->f_data;
   2200      1.160        ad 	solock(so);
   2201       1.73  christos 	SLIST_REMOVE(&so->so_rcv.sb_sel.sel_klist, kn, knote, kn_selnext);
   2202       1.73  christos 	if (SLIST_EMPTY(&so->so_rcv.sb_sel.sel_klist))
   2203       1.72  jdolecek 		so->so_rcv.sb_flags &= ~SB_KNOTE;
   2204      1.160        ad 	sounlock(so);
   2205       1.72  jdolecek }
   2206       1.72  jdolecek 
   2207       1.72  jdolecek /*ARGSUSED*/
   2208       1.72  jdolecek static int
   2209      1.129      yamt filt_soread(struct knote *kn, long hint)
   2210       1.72  jdolecek {
   2211       1.72  jdolecek 	struct socket	*so;
   2212      1.160        ad 	int rv;
   2213       1.72  jdolecek 
   2214      1.155        ad 	so = ((file_t *)kn->kn_obj)->f_data;
   2215      1.160        ad 	if (hint != NOTE_SUBMIT)
   2216      1.160        ad 		solock(so);
   2217       1.72  jdolecek 	kn->kn_data = so->so_rcv.sb_cc;
   2218       1.72  jdolecek 	if (so->so_state & SS_CANTRCVMORE) {
   2219      1.108     perry 		kn->kn_flags |= EV_EOF;
   2220       1.72  jdolecek 		kn->kn_fflags = so->so_error;
   2221      1.160        ad 		rv = 1;
   2222      1.160        ad 	} else if (so->so_error)	/* temporary udp error */
   2223      1.160        ad 		rv = 1;
   2224      1.160        ad 	else if (kn->kn_sfflags & NOTE_LOWAT)
   2225      1.160        ad 		rv = (kn->kn_data >= kn->kn_sdata);
   2226      1.160        ad 	else
   2227      1.160        ad 		rv = (kn->kn_data >= so->so_rcv.sb_lowat);
   2228      1.160        ad 	if (hint != NOTE_SUBMIT)
   2229      1.160        ad 		sounlock(so);
   2230      1.160        ad 	return rv;
   2231       1.72  jdolecek }
   2232       1.72  jdolecek 
   2233       1.72  jdolecek static void
   2234       1.72  jdolecek filt_sowdetach(struct knote *kn)
   2235       1.72  jdolecek {
   2236       1.72  jdolecek 	struct socket	*so;
   2237       1.72  jdolecek 
   2238      1.155        ad 	so = ((file_t *)kn->kn_obj)->f_data;
   2239      1.160        ad 	solock(so);
   2240       1.73  christos 	SLIST_REMOVE(&so->so_snd.sb_sel.sel_klist, kn, knote, kn_selnext);
   2241       1.73  christos 	if (SLIST_EMPTY(&so->so_snd.sb_sel.sel_klist))
   2242       1.72  jdolecek 		so->so_snd.sb_flags &= ~SB_KNOTE;
   2243      1.160        ad 	sounlock(so);
   2244       1.72  jdolecek }
   2245       1.72  jdolecek 
   2246       1.72  jdolecek /*ARGSUSED*/
   2247       1.72  jdolecek static int
   2248      1.129      yamt filt_sowrite(struct knote *kn, long hint)
   2249       1.72  jdolecek {
   2250       1.72  jdolecek 	struct socket	*so;
   2251      1.160        ad 	int rv;
   2252       1.72  jdolecek 
   2253      1.155        ad 	so = ((file_t *)kn->kn_obj)->f_data;
   2254      1.160        ad 	if (hint != NOTE_SUBMIT)
   2255      1.160        ad 		solock(so);
   2256       1.72  jdolecek 	kn->kn_data = sbspace(&so->so_snd);
   2257       1.72  jdolecek 	if (so->so_state & SS_CANTSENDMORE) {
   2258      1.108     perry 		kn->kn_flags |= EV_EOF;
   2259       1.72  jdolecek 		kn->kn_fflags = so->so_error;
   2260      1.160        ad 		rv = 1;
   2261      1.160        ad 	} else if (so->so_error)	/* temporary udp error */
   2262      1.160        ad 		rv = 1;
   2263      1.160        ad 	else if (((so->so_state & SS_ISCONNECTED) == 0) &&
   2264       1.72  jdolecek 	    (so->so_proto->pr_flags & PR_CONNREQUIRED))
   2265      1.160        ad 		rv = 0;
   2266      1.160        ad 	else if (kn->kn_sfflags & NOTE_LOWAT)
   2267      1.160        ad 		rv = (kn->kn_data >= kn->kn_sdata);
   2268      1.160        ad 	else
   2269      1.160        ad 		rv = (kn->kn_data >= so->so_snd.sb_lowat);
   2270      1.160        ad 	if (hint != NOTE_SUBMIT)
   2271      1.160        ad 		sounlock(so);
   2272      1.160        ad 	return rv;
   2273       1.72  jdolecek }
   2274       1.72  jdolecek 
   2275       1.72  jdolecek /*ARGSUSED*/
   2276       1.72  jdolecek static int
   2277      1.129      yamt filt_solisten(struct knote *kn, long hint)
   2278       1.72  jdolecek {
   2279       1.72  jdolecek 	struct socket	*so;
   2280      1.160        ad 	int rv;
   2281       1.72  jdolecek 
   2282      1.155        ad 	so = ((file_t *)kn->kn_obj)->f_data;
   2283       1.72  jdolecek 
   2284       1.72  jdolecek 	/*
   2285       1.72  jdolecek 	 * Set kn_data to number of incoming connections, not
   2286       1.72  jdolecek 	 * counting partial (incomplete) connections.
   2287      1.108     perry 	 */
   2288      1.160        ad 	if (hint != NOTE_SUBMIT)
   2289      1.160        ad 		solock(so);
   2290       1.72  jdolecek 	kn->kn_data = so->so_qlen;
   2291      1.160        ad 	rv = (kn->kn_data > 0);
   2292      1.160        ad 	if (hint != NOTE_SUBMIT)
   2293      1.160        ad 		sounlock(so);
   2294      1.160        ad 	return rv;
   2295       1.72  jdolecek }
   2296       1.72  jdolecek 
   2297       1.72  jdolecek static const struct filterops solisten_filtops =
   2298       1.72  jdolecek 	{ 1, NULL, filt_sordetach, filt_solisten };
   2299       1.72  jdolecek static const struct filterops soread_filtops =
   2300       1.72  jdolecek 	{ 1, NULL, filt_sordetach, filt_soread };
   2301       1.72  jdolecek static const struct filterops sowrite_filtops =
   2302       1.72  jdolecek 	{ 1, NULL, filt_sowdetach, filt_sowrite };
   2303       1.72  jdolecek 
   2304       1.72  jdolecek int
   2305      1.129      yamt soo_kqfilter(struct file *fp, struct knote *kn)
   2306       1.72  jdolecek {
   2307       1.72  jdolecek 	struct socket	*so;
   2308       1.72  jdolecek 	struct sockbuf	*sb;
   2309       1.72  jdolecek 
   2310      1.155        ad 	so = ((file_t *)kn->kn_obj)->f_data;
   2311      1.160        ad 	solock(so);
   2312       1.72  jdolecek 	switch (kn->kn_filter) {
   2313       1.72  jdolecek 	case EVFILT_READ:
   2314       1.72  jdolecek 		if (so->so_options & SO_ACCEPTCONN)
   2315       1.72  jdolecek 			kn->kn_fop = &solisten_filtops;
   2316       1.72  jdolecek 		else
   2317       1.72  jdolecek 			kn->kn_fop = &soread_filtops;
   2318       1.72  jdolecek 		sb = &so->so_rcv;
   2319       1.72  jdolecek 		break;
   2320       1.72  jdolecek 	case EVFILT_WRITE:
   2321       1.72  jdolecek 		kn->kn_fop = &sowrite_filtops;
   2322       1.72  jdolecek 		sb = &so->so_snd;
   2323       1.72  jdolecek 		break;
   2324       1.72  jdolecek 	default:
   2325      1.160        ad 		sounlock(so);
   2326      1.149     pooka 		return (EINVAL);
   2327       1.72  jdolecek 	}
   2328       1.73  christos 	SLIST_INSERT_HEAD(&sb->sb_sel.sel_klist, kn, kn_selnext);
   2329       1.72  jdolecek 	sb->sb_flags |= SB_KNOTE;
   2330      1.160        ad 	sounlock(so);
   2331       1.72  jdolecek 	return (0);
   2332       1.72  jdolecek }
   2333       1.72  jdolecek 
   2334      1.154        ad static int
   2335      1.154        ad sodopoll(struct socket *so, int events)
   2336      1.154        ad {
   2337      1.154        ad 	int revents;
   2338      1.154        ad 
   2339      1.154        ad 	revents = 0;
   2340      1.154        ad 
   2341      1.154        ad 	if (events & (POLLIN | POLLRDNORM))
   2342      1.154        ad 		if (soreadable(so))
   2343      1.154        ad 			revents |= events & (POLLIN | POLLRDNORM);
   2344      1.154        ad 
   2345      1.154        ad 	if (events & (POLLOUT | POLLWRNORM))
   2346      1.154        ad 		if (sowritable(so))
   2347      1.154        ad 			revents |= events & (POLLOUT | POLLWRNORM);
   2348      1.154        ad 
   2349      1.154        ad 	if (events & (POLLPRI | POLLRDBAND))
   2350      1.154        ad 		if (so->so_oobmark || (so->so_state & SS_RCVATMARK))
   2351      1.154        ad 			revents |= events & (POLLPRI | POLLRDBAND);
   2352      1.154        ad 
   2353      1.154        ad 	return revents;
   2354      1.154        ad }
   2355      1.154        ad 
   2356      1.154        ad int
   2357      1.154        ad sopoll(struct socket *so, int events)
   2358      1.154        ad {
   2359      1.154        ad 	int revents = 0;
   2360      1.154        ad 
   2361      1.160        ad #ifndef DIAGNOSTIC
   2362      1.160        ad 	/*
   2363      1.160        ad 	 * Do a quick, unlocked check in expectation that the socket
   2364      1.160        ad 	 * will be ready for I/O.  Don't do this check if DIAGNOSTIC,
   2365      1.160        ad 	 * as the solocked() assertions will fail.
   2366      1.160        ad 	 */
   2367      1.154        ad 	if ((revents = sodopoll(so, events)) != 0)
   2368      1.154        ad 		return revents;
   2369      1.160        ad #endif
   2370      1.154        ad 
   2371      1.160        ad 	solock(so);
   2372      1.154        ad 	if ((revents = sodopoll(so, events)) == 0) {
   2373      1.154        ad 		if (events & (POLLIN | POLLPRI | POLLRDNORM | POLLRDBAND)) {
   2374      1.154        ad 			selrecord(curlwp, &so->so_rcv.sb_sel);
   2375      1.160        ad 			so->so_rcv.sb_flags |= SB_NOTIFY;
   2376      1.154        ad 		}
   2377      1.154        ad 
   2378      1.154        ad 		if (events & (POLLOUT | POLLWRNORM)) {
   2379      1.154        ad 			selrecord(curlwp, &so->so_snd.sb_sel);
   2380      1.160        ad 			so->so_snd.sb_flags |= SB_NOTIFY;
   2381      1.154        ad 		}
   2382      1.154        ad 	}
   2383      1.160        ad 	sounlock(so);
   2384      1.154        ad 
   2385      1.154        ad 	return revents;
   2386      1.154        ad }
   2387      1.154        ad 
   2388      1.154        ad 
   2389       1.94      yamt #include <sys/sysctl.h>
   2390       1.94      yamt 
   2391       1.94      yamt static int sysctl_kern_somaxkva(SYSCTLFN_PROTO);
   2392       1.94      yamt 
   2393       1.94      yamt /*
   2394       1.94      yamt  * sysctl helper routine for kern.somaxkva.  ensures that the given
   2395       1.94      yamt  * value is not too small.
   2396       1.94      yamt  * (XXX should we maybe make sure it's not too large as well?)
   2397       1.94      yamt  */
   2398       1.94      yamt static int
   2399       1.94      yamt sysctl_kern_somaxkva(SYSCTLFN_ARGS)
   2400       1.94      yamt {
   2401       1.94      yamt 	int error, new_somaxkva;
   2402       1.94      yamt 	struct sysctlnode node;
   2403       1.94      yamt 
   2404       1.94      yamt 	new_somaxkva = somaxkva;
   2405       1.94      yamt 	node = *rnode;
   2406       1.94      yamt 	node.sysctl_data = &new_somaxkva;
   2407       1.94      yamt 	error = sysctl_lookup(SYSCTLFN_CALL(&node));
   2408       1.94      yamt 	if (error || newp == NULL)
   2409       1.94      yamt 		return (error);
   2410       1.94      yamt 
   2411       1.94      yamt 	if (new_somaxkva < (16 * 1024 * 1024)) /* sanity */
   2412       1.94      yamt 		return (EINVAL);
   2413       1.94      yamt 
   2414      1.136        ad 	mutex_enter(&so_pendfree_lock);
   2415       1.94      yamt 	somaxkva = new_somaxkva;
   2416      1.136        ad 	cv_broadcast(&socurkva_cv);
   2417      1.136        ad 	mutex_exit(&so_pendfree_lock);
   2418       1.94      yamt 
   2419       1.94      yamt 	return (error);
   2420       1.94      yamt }
   2421       1.94      yamt 
   2422      1.178     pooka static void
   2423      1.187    cegger sysctl_kern_somaxkva_setup(void)
   2424       1.94      yamt {
   2425       1.94      yamt 
   2426      1.178     pooka 	KASSERT(socket_sysctllog == NULL);
   2427      1.178     pooka 	sysctl_createv(&socket_sysctllog, 0, NULL, NULL,
   2428       1.97    atatat 		       CTLFLAG_PERMANENT,
   2429       1.97    atatat 		       CTLTYPE_NODE, "kern", NULL,
   2430       1.97    atatat 		       NULL, 0, NULL, 0,
   2431       1.97    atatat 		       CTL_KERN, CTL_EOL);
   2432       1.97    atatat 
   2433      1.178     pooka 	sysctl_createv(&socket_sysctllog, 0, NULL, NULL,
   2434       1.97    atatat 		       CTLFLAG_PERMANENT|CTLFLAG_READWRITE,
   2435      1.103    atatat 		       CTLTYPE_INT, "somaxkva",
   2436      1.103    atatat 		       SYSCTL_DESCR("Maximum amount of kernel memory to be "
   2437      1.103    atatat 				    "used for socket buffers"),
   2438       1.94      yamt 		       sysctl_kern_somaxkva, 0, NULL, 0,
   2439       1.94      yamt 		       CTL_KERN, KERN_SOMAXKVA, CTL_EOL);
   2440       1.94      yamt }
   2441