Home | History | Annotate | Line # | Download | only in kern
uipc_socket.c revision 1.97.2.1.2.3
      1  1.97.2.1.2.3       jmc /*	$NetBSD: uipc_socket.c,v 1.97.2.1.2.3 2005/10/28 19:26:22 jmc Exp $	*/
      2          1.64   thorpej 
      3          1.64   thorpej /*-
      4          1.64   thorpej  * Copyright (c) 2002 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.64   thorpej  * by Jason R. Thorpe of Wasabi Systems, Inc.
      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  * 3. All advertising materials mentioning features or use of this software
     19          1.64   thorpej  *    must display the following acknowledgement:
     20          1.64   thorpej  *	This product includes software developed by the NetBSD
     21          1.64   thorpej  *	Foundation, Inc. and its contributors.
     22          1.64   thorpej  * 4. Neither the name of The NetBSD Foundation nor the names of its
     23          1.64   thorpej  *    contributors may be used to endorse or promote products derived
     24          1.64   thorpej  *    from this software without specific prior written permission.
     25          1.64   thorpej  *
     26          1.64   thorpej  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
     27          1.64   thorpej  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
     28          1.64   thorpej  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
     29          1.64   thorpej  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
     30          1.64   thorpej  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
     31          1.64   thorpej  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
     32          1.64   thorpej  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
     33          1.64   thorpej  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
     34          1.64   thorpej  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
     35          1.64   thorpej  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
     36          1.64   thorpej  * POSSIBILITY OF SUCH DAMAGE.
     37          1.64   thorpej  */
     38          1.16       cgd 
     39           1.1       cgd /*
     40          1.15   mycroft  * Copyright (c) 1982, 1986, 1988, 1990, 1993
     41          1.15   mycroft  *	The Regents of the University of California.  All rights reserved.
     42           1.1       cgd  *
     43           1.1       cgd  * Redistribution and use in source and binary forms, with or without
     44           1.1       cgd  * modification, are permitted provided that the following conditions
     45           1.1       cgd  * are met:
     46           1.1       cgd  * 1. Redistributions of source code must retain the above copyright
     47           1.1       cgd  *    notice, this list of conditions and the following disclaimer.
     48           1.1       cgd  * 2. Redistributions in binary form must reproduce the above copyright
     49           1.1       cgd  *    notice, this list of conditions and the following disclaimer in the
     50           1.1       cgd  *    documentation and/or other materials provided with the distribution.
     51          1.85       agc  * 3. Neither the name of the University nor the names of its contributors
     52           1.1       cgd  *    may be used to endorse or promote products derived from this software
     53           1.1       cgd  *    without specific prior written permission.
     54           1.1       cgd  *
     55           1.1       cgd  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
     56           1.1       cgd  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
     57           1.1       cgd  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
     58           1.1       cgd  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
     59           1.1       cgd  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
     60           1.1       cgd  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
     61           1.1       cgd  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
     62           1.1       cgd  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
     63           1.1       cgd  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
     64           1.1       cgd  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
     65           1.1       cgd  * SUCH DAMAGE.
     66           1.1       cgd  *
     67          1.32      fvdl  *	@(#)uipc_socket.c	8.6 (Berkeley) 5/2/95
     68           1.1       cgd  */
     69          1.59     lukem 
     70          1.59     lukem #include <sys/cdefs.h>
     71  1.97.2.1.2.3       jmc __KERNEL_RCSID(0, "$NetBSD: uipc_socket.c,v 1.97.2.1.2.3 2005/10/28 19:26:22 jmc Exp $");
     72          1.64   thorpej 
     73          1.64   thorpej #include "opt_sock_counters.h"
     74          1.64   thorpej #include "opt_sosend_loan.h"
     75          1.81    martin #include "opt_mbuftrace.h"
     76          1.84     ragge #include "opt_somaxkva.h"
     77           1.1       cgd 
     78           1.9   mycroft #include <sys/param.h>
     79           1.9   mycroft #include <sys/systm.h>
     80           1.9   mycroft #include <sys/proc.h>
     81           1.9   mycroft #include <sys/file.h>
     82           1.9   mycroft #include <sys/malloc.h>
     83           1.9   mycroft #include <sys/mbuf.h>
     84           1.9   mycroft #include <sys/domain.h>
     85           1.9   mycroft #include <sys/kernel.h>
     86           1.9   mycroft #include <sys/protosw.h>
     87           1.9   mycroft #include <sys/socket.h>
     88           1.9   mycroft #include <sys/socketvar.h>
     89          1.21  christos #include <sys/signalvar.h>
     90           1.9   mycroft #include <sys/resourcevar.h>
     91          1.37   thorpej #include <sys/pool.h>
     92          1.72  jdolecek #include <sys/event.h>
     93          1.89  christos #include <sys/poll.h>
     94          1.37   thorpej 
     95          1.64   thorpej #include <uvm/uvm.h>
     96          1.64   thorpej 
     97          1.54     lukem struct pool	socket_pool;
     98          1.77   thorpej 
     99          1.77   thorpej MALLOC_DEFINE(M_SOOPTS, "soopts", "socket options");
    100          1.77   thorpej MALLOC_DEFINE(M_SONAME, "soname", "socket name");
    101          1.37   thorpej 
    102          1.54     lukem extern int	somaxconn;			/* patchable (XXX sysctl) */
    103          1.54     lukem int		somaxconn = SOMAXCONN;
    104          1.49  jonathan 
    105          1.64   thorpej #ifdef SOSEND_COUNTERS
    106          1.64   thorpej #include <sys/device.h>
    107          1.64   thorpej 
    108          1.64   thorpej struct evcnt sosend_loan_big = EVCNT_INITIALIZER(EVCNT_TYPE_MISC,
    109          1.64   thorpej     NULL, "sosend", "loan big");
    110          1.64   thorpej struct evcnt sosend_copy_big = EVCNT_INITIALIZER(EVCNT_TYPE_MISC,
    111          1.64   thorpej     NULL, "sosend", "copy big");
    112          1.64   thorpej struct evcnt sosend_copy_small = EVCNT_INITIALIZER(EVCNT_TYPE_MISC,
    113          1.64   thorpej     NULL, "sosend", "copy small");
    114          1.64   thorpej struct evcnt sosend_kvalimit = EVCNT_INITIALIZER(EVCNT_TYPE_MISC,
    115          1.64   thorpej     NULL, "sosend", "kva limit");
    116          1.64   thorpej 
    117          1.64   thorpej #define	SOSEND_COUNTER_INCR(ev)		(ev)->ev_count++
    118          1.64   thorpej 
    119          1.64   thorpej #else
    120          1.64   thorpej 
    121          1.64   thorpej #define	SOSEND_COUNTER_INCR(ev)		/* nothing */
    122          1.64   thorpej 
    123          1.64   thorpej #endif /* SOSEND_COUNTERS */
    124          1.64   thorpej 
    125          1.37   thorpej void
    126          1.54     lukem soinit(void)
    127          1.37   thorpej {
    128          1.91   thorpej 
    129          1.91   thorpej 	/* Set the initial adjusted socket buffer size. */
    130          1.91   thorpej 	if (sb_max_set(sb_max))
    131          1.91   thorpej 		panic("bad initial sb_max value: %lu\n", sb_max);
    132          1.37   thorpej 
    133          1.37   thorpej 	pool_init(&socket_pool, sizeof(struct socket), 0, 0, 0,
    134          1.62   thorpej 	    "sockpl", NULL);
    135          1.64   thorpej 
    136          1.64   thorpej #ifdef SOSEND_COUNTERS
    137          1.64   thorpej 	evcnt_attach_static(&sosend_loan_big);
    138          1.64   thorpej 	evcnt_attach_static(&sosend_copy_big);
    139          1.64   thorpej 	evcnt_attach_static(&sosend_copy_small);
    140          1.64   thorpej 	evcnt_attach_static(&sosend_kvalimit);
    141          1.64   thorpej #endif /* SOSEND_COUNTERS */
    142          1.37   thorpej }
    143           1.1       cgd 
    144          1.71   thorpej #ifdef SOSEND_NO_LOAN
    145          1.71   thorpej int use_sosend_loan = 0;
    146          1.71   thorpej #else
    147          1.65   thorpej int use_sosend_loan = 1;
    148          1.65   thorpej #endif
    149          1.64   thorpej 
    150          1.93      yamt struct simplelock so_pendfree_slock = SIMPLELOCK_INITIALIZER;
    151          1.64   thorpej struct mbuf *so_pendfree;
    152          1.64   thorpej 
    153          1.84     ragge #ifndef SOMAXKVA
    154          1.84     ragge #define	SOMAXKVA (16 * 1024 * 1024)
    155          1.84     ragge #endif
    156          1.84     ragge int somaxkva = SOMAXKVA;
    157          1.64   thorpej int socurkva;
    158          1.64   thorpej int sokvawaiters;
    159          1.64   thorpej 
    160          1.64   thorpej #define	SOCK_LOAN_THRESH	4096
    161          1.64   thorpej #define	SOCK_LOAN_CHUNK		65536
    162          1.64   thorpej 
    163          1.80      yamt static size_t sodopendfree(struct socket *);
    164          1.93      yamt static size_t sodopendfreel(struct socket *);
    165          1.95      yamt static __inline void sokvareserve(struct socket *, vsize_t);
    166          1.95      yamt static __inline void sokvaunreserve(vsize_t);
    167          1.93      yamt 
    168          1.95      yamt static __inline void
    169          1.95      yamt sokvareserve(struct socket *so, vsize_t len)
    170          1.80      yamt {
    171          1.80      yamt 	int s;
    172          1.80      yamt 
    173          1.93      yamt 	s = splvm();
    174          1.93      yamt 	simple_lock(&so_pendfree_slock);
    175          1.80      yamt 	while (socurkva + len > somaxkva) {
    176          1.93      yamt 		size_t freed;
    177          1.93      yamt 
    178          1.93      yamt 		/*
    179          1.93      yamt 		 * try to do pendfree.
    180          1.93      yamt 		 */
    181          1.93      yamt 
    182          1.93      yamt 		freed = sodopendfreel(so);
    183          1.93      yamt 
    184          1.93      yamt 		/*
    185          1.93      yamt 		 * if some kva was freed, try again.
    186          1.93      yamt 		 */
    187          1.93      yamt 
    188          1.93      yamt 		if (freed)
    189          1.80      yamt 			continue;
    190          1.93      yamt 
    191          1.80      yamt 		SOSEND_COUNTER_INCR(&sosend_kvalimit);
    192          1.80      yamt 		sokvawaiters++;
    193          1.93      yamt 		(void) ltsleep(&socurkva, PVM, "sokva", 0, &so_pendfree_slock);
    194          1.80      yamt 		sokvawaiters--;
    195          1.80      yamt 	}
    196          1.93      yamt 	socurkva += len;
    197          1.93      yamt 	simple_unlock(&so_pendfree_slock);
    198          1.93      yamt 	splx(s);
    199          1.95      yamt }
    200          1.95      yamt 
    201          1.95      yamt static __inline void
    202          1.95      yamt sokvaunreserve(vsize_t len)
    203          1.95      yamt {
    204          1.95      yamt 	int s;
    205          1.95      yamt 
    206          1.95      yamt 	s = splvm();
    207          1.95      yamt 	simple_lock(&so_pendfree_slock);
    208          1.95      yamt 	socurkva -= len;
    209          1.95      yamt 	if (sokvawaiters)
    210          1.95      yamt 		wakeup(&socurkva);
    211          1.95      yamt 	simple_unlock(&so_pendfree_slock);
    212          1.95      yamt 	splx(s);
    213          1.95      yamt }
    214          1.95      yamt 
    215          1.95      yamt /*
    216          1.95      yamt  * sokvaalloc: allocate kva for loan.
    217          1.95      yamt  */
    218          1.95      yamt 
    219          1.95      yamt vaddr_t
    220          1.95      yamt sokvaalloc(vsize_t len, struct socket *so)
    221          1.95      yamt {
    222          1.95      yamt 	vaddr_t lva;
    223          1.95      yamt 
    224          1.95      yamt 	/*
    225          1.95      yamt 	 * reserve kva.
    226          1.95      yamt 	 */
    227          1.95      yamt 
    228          1.95      yamt 	sokvareserve(so, len);
    229          1.93      yamt 
    230          1.93      yamt 	/*
    231          1.93      yamt 	 * allocate kva.
    232          1.93      yamt 	 */
    233          1.80      yamt 
    234          1.80      yamt 	lva = uvm_km_valloc_wait(kernel_map, len);
    235          1.95      yamt 	if (lva == 0) {
    236          1.95      yamt 		sokvaunreserve(len);
    237          1.80      yamt 		return (0);
    238          1.95      yamt 	}
    239          1.80      yamt 
    240          1.80      yamt 	return lva;
    241          1.80      yamt }
    242          1.80      yamt 
    243          1.93      yamt /*
    244          1.93      yamt  * sokvafree: free kva for loan.
    245          1.93      yamt  */
    246          1.93      yamt 
    247          1.80      yamt void
    248          1.80      yamt sokvafree(vaddr_t sva, vsize_t len)
    249          1.80      yamt {
    250          1.93      yamt 
    251          1.93      yamt 	/*
    252          1.93      yamt 	 * free kva.
    253          1.93      yamt 	 */
    254          1.80      yamt 
    255          1.80      yamt 	uvm_km_free(kernel_map, sva, len);
    256          1.93      yamt 
    257          1.93      yamt 	/*
    258          1.93      yamt 	 * unreserve kva.
    259          1.93      yamt 	 */
    260          1.93      yamt 
    261          1.95      yamt 	sokvaunreserve(len);
    262          1.80      yamt }
    263          1.80      yamt 
    264          1.64   thorpej static void
    265          1.79   thorpej sodoloanfree(struct vm_page **pgs, caddr_t buf, size_t size)
    266          1.64   thorpej {
    267          1.64   thorpej 	vaddr_t va, sva, eva;
    268          1.64   thorpej 	vsize_t len;
    269          1.64   thorpej 	paddr_t pa;
    270          1.64   thorpej 	int i, npgs;
    271          1.64   thorpej 
    272          1.64   thorpej 	eva = round_page((vaddr_t) buf + size);
    273          1.64   thorpej 	sva = trunc_page((vaddr_t) buf);
    274          1.64   thorpej 	len = eva - sva;
    275          1.64   thorpej 	npgs = len >> PAGE_SHIFT;
    276          1.64   thorpej 
    277          1.79   thorpej 	if (__predict_false(pgs == NULL)) {
    278          1.79   thorpej 		pgs = alloca(npgs * sizeof(*pgs));
    279          1.64   thorpej 
    280          1.79   thorpej 		for (i = 0, va = sva; va < eva; i++, va += PAGE_SIZE) {
    281          1.79   thorpej 			if (pmap_extract(pmap_kernel(), va, &pa) == FALSE)
    282          1.79   thorpej 				panic("sodoloanfree: va 0x%lx not mapped", va);
    283          1.79   thorpej 			pgs[i] = PHYS_TO_VM_PAGE(pa);
    284          1.79   thorpej 		}
    285          1.64   thorpej 	}
    286          1.64   thorpej 
    287          1.64   thorpej 	pmap_kremove(sva, len);
    288          1.64   thorpej 	pmap_update(pmap_kernel());
    289          1.64   thorpej 	uvm_unloan(pgs, npgs, UVM_LOAN_TOPAGE);
    290          1.80      yamt 	sokvafree(sva, len);
    291          1.64   thorpej }
    292          1.64   thorpej 
    293          1.64   thorpej static size_t
    294          1.64   thorpej sodopendfree(struct socket *so)
    295          1.64   thorpej {
    296          1.64   thorpej 	int s;
    297          1.93      yamt 	size_t rv;
    298          1.64   thorpej 
    299          1.64   thorpej 	s = splvm();
    300          1.93      yamt 	simple_lock(&so_pendfree_slock);
    301          1.93      yamt 	rv = sodopendfreel(so);
    302          1.93      yamt 	simple_unlock(&so_pendfree_slock);
    303          1.93      yamt 	splx(s);
    304          1.93      yamt 
    305          1.93      yamt 	return rv;
    306          1.93      yamt }
    307          1.93      yamt 
    308          1.93      yamt /*
    309          1.93      yamt  * sodopendfreel: free mbufs on "pendfree" list.
    310          1.93      yamt  * unlock and relock so_pendfree_slock when freeing mbufs.
    311          1.93      yamt  *
    312          1.93      yamt  * => called with so_pendfree_slock held.
    313          1.93      yamt  * => called at splvm.
    314          1.93      yamt  */
    315          1.93      yamt 
    316          1.93      yamt static size_t
    317          1.93      yamt sodopendfreel(struct socket *so)
    318          1.93      yamt {
    319          1.93      yamt 	size_t rv = 0;
    320          1.93      yamt 
    321          1.93      yamt 	LOCK_ASSERT(simple_lock_held(&so_pendfree_slock));
    322          1.64   thorpej 
    323          1.64   thorpej 	for (;;) {
    324          1.93      yamt 		struct mbuf *m;
    325          1.93      yamt 		struct mbuf *next;
    326          1.93      yamt 
    327          1.64   thorpej 		m = so_pendfree;
    328          1.64   thorpej 		if (m == NULL)
    329          1.64   thorpej 			break;
    330          1.93      yamt 		so_pendfree = NULL;
    331          1.93      yamt 		simple_unlock(&so_pendfree_slock);
    332          1.93      yamt 		/* XXX splx */
    333          1.93      yamt 
    334          1.93      yamt 		for (; m != NULL; m = next) {
    335          1.93      yamt 			next = m->m_next;
    336          1.93      yamt 
    337          1.93      yamt 			rv += m->m_ext.ext_size;
    338          1.93      yamt 			sodoloanfree((m->m_flags & M_EXT_PAGES) ?
    339          1.93      yamt 			    m->m_ext.ext_pgs : NULL, m->m_ext.ext_buf,
    340          1.93      yamt 			    m->m_ext.ext_size);
    341          1.93      yamt 			pool_cache_put(&mbpool_cache, m);
    342          1.93      yamt 		}
    343          1.64   thorpej 
    344          1.93      yamt 		/* XXX splvm */
    345          1.93      yamt 		simple_lock(&so_pendfree_slock);
    346          1.64   thorpej 	}
    347          1.64   thorpej 
    348          1.64   thorpej 	return (rv);
    349          1.64   thorpej }
    350          1.64   thorpej 
    351          1.80      yamt void
    352          1.76   thorpej soloanfree(struct mbuf *m, caddr_t buf, size_t size, void *arg)
    353          1.64   thorpej {
    354          1.64   thorpej 	int s;
    355          1.64   thorpej 
    356          1.64   thorpej 	if (m == NULL) {
    357          1.93      yamt 
    358          1.93      yamt 		/*
    359          1.93      yamt 		 * called from MEXTREMOVE.
    360          1.93      yamt 		 */
    361          1.93      yamt 
    362          1.79   thorpej 		sodoloanfree(NULL, buf, size);
    363          1.64   thorpej 		return;
    364          1.64   thorpej 	}
    365          1.64   thorpej 
    366          1.93      yamt 	/*
    367          1.93      yamt 	 * postpone freeing mbuf.
    368          1.93      yamt 	 *
    369          1.93      yamt 	 * we can't do it in interrupt context
    370          1.93      yamt 	 * because we need to put kva back to kernel_map.
    371          1.93      yamt 	 */
    372          1.93      yamt 
    373          1.64   thorpej 	s = splvm();
    374          1.93      yamt 	simple_lock(&so_pendfree_slock);
    375          1.92      yamt 	m->m_next = so_pendfree;
    376          1.92      yamt 	so_pendfree = m;
    377          1.64   thorpej 	if (sokvawaiters)
    378          1.64   thorpej 		wakeup(&socurkva);
    379          1.93      yamt 	simple_unlock(&so_pendfree_slock);
    380          1.93      yamt 	splx(s);
    381          1.64   thorpej }
    382          1.64   thorpej 
    383          1.64   thorpej static long
    384          1.64   thorpej sosend_loan(struct socket *so, struct uio *uio, struct mbuf *m, long space)
    385          1.64   thorpej {
    386          1.64   thorpej 	struct iovec *iov = uio->uio_iov;
    387          1.64   thorpej 	vaddr_t sva, eva;
    388          1.64   thorpej 	vsize_t len;
    389          1.64   thorpej 	vaddr_t lva, va;
    390          1.80      yamt 	int npgs, i, error;
    391          1.64   thorpej 
    392          1.64   thorpej 	if (uio->uio_segflg != UIO_USERSPACE)
    393          1.64   thorpej 		return (0);
    394          1.64   thorpej 
    395          1.64   thorpej 	if (iov->iov_len < (size_t) space)
    396          1.64   thorpej 		space = iov->iov_len;
    397          1.64   thorpej 	if (space > SOCK_LOAN_CHUNK)
    398          1.64   thorpej 		space = SOCK_LOAN_CHUNK;
    399          1.64   thorpej 
    400          1.64   thorpej 	eva = round_page((vaddr_t) iov->iov_base + space);
    401          1.64   thorpej 	sva = trunc_page((vaddr_t) iov->iov_base);
    402          1.64   thorpej 	len = eva - sva;
    403          1.64   thorpej 	npgs = len >> PAGE_SHIFT;
    404          1.64   thorpej 
    405          1.79   thorpej 	/* XXX KDASSERT */
    406          1.79   thorpej 	KASSERT(npgs <= M_EXT_MAXPAGES);
    407          1.79   thorpej 
    408          1.80      yamt 	lva = sokvaalloc(len, so);
    409          1.64   thorpej 	if (lva == 0)
    410          1.80      yamt 		return 0;
    411          1.64   thorpej 
    412          1.83      fvdl 	error = uvm_loan(&uio->uio_procp->p_vmspace->vm_map, sva, len,
    413          1.79   thorpej 	    m->m_ext.ext_pgs, UVM_LOAN_TOPAGE);
    414          1.64   thorpej 	if (error) {
    415          1.80      yamt 		sokvafree(lva, len);
    416          1.64   thorpej 		return (0);
    417          1.64   thorpej 	}
    418          1.64   thorpej 
    419          1.64   thorpej 	for (i = 0, va = lva; i < npgs; i++, va += PAGE_SIZE)
    420          1.79   thorpej 		pmap_kenter_pa(va, VM_PAGE_TO_PHYS(m->m_ext.ext_pgs[i]),
    421          1.79   thorpej 		    VM_PROT_READ);
    422          1.64   thorpej 	pmap_update(pmap_kernel());
    423          1.64   thorpej 
    424          1.64   thorpej 	lva += (vaddr_t) iov->iov_base & PAGE_MASK;
    425          1.64   thorpej 
    426          1.64   thorpej 	MEXTADD(m, (caddr_t) lva, space, M_MBUF, soloanfree, so);
    427          1.79   thorpej 	m->m_flags |= M_EXT_PAGES | M_EXT_ROMAP;
    428          1.64   thorpej 
    429          1.64   thorpej 	uio->uio_resid -= space;
    430          1.64   thorpej 	/* uio_offset not updated, not set/used for write(2) */
    431          1.64   thorpej 	uio->uio_iov->iov_base = (caddr_t) uio->uio_iov->iov_base + space;
    432          1.64   thorpej 	uio->uio_iov->iov_len -= space;
    433          1.64   thorpej 	if (uio->uio_iov->iov_len == 0) {
    434          1.64   thorpej 		uio->uio_iov++;
    435          1.64   thorpej 		uio->uio_iovcnt--;
    436          1.64   thorpej 	}
    437          1.64   thorpej 
    438          1.64   thorpej 	return (space);
    439          1.64   thorpej }
    440          1.64   thorpej 
    441           1.1       cgd /*
    442           1.1       cgd  * Socket operation routines.
    443           1.1       cgd  * These routines are called by the routines in
    444           1.1       cgd  * sys_socket.c or from a system process, and
    445           1.1       cgd  * implement the semantics of socket operations by
    446           1.1       cgd  * switching out to the protocol specific routines.
    447           1.1       cgd  */
    448           1.1       cgd /*ARGSUSED*/
    449           1.3    andrew int
    450          1.54     lukem socreate(int dom, struct socket **aso, int type, int proto)
    451           1.1       cgd {
    452          1.54     lukem 	struct proc	*p;
    453          1.54     lukem 	struct protosw	*prp;
    454          1.54     lukem 	struct socket	*so;
    455          1.54     lukem 	int		error, s;
    456           1.1       cgd 
    457          1.54     lukem 	p = curproc;		/* XXX */
    458           1.1       cgd 	if (proto)
    459           1.1       cgd 		prp = pffindproto(dom, proto, type);
    460           1.1       cgd 	else
    461           1.1       cgd 		prp = pffindtype(dom, type);
    462          1.15   mycroft 	if (prp == 0 || prp->pr_usrreq == 0)
    463           1.1       cgd 		return (EPROTONOSUPPORT);
    464           1.1       cgd 	if (prp->pr_type != type)
    465           1.1       cgd 		return (EPROTOTYPE);
    466          1.39      matt 	s = splsoftnet();
    467          1.37   thorpej 	so = pool_get(&socket_pool, PR_WAITOK);
    468          1.38     perry 	memset((caddr_t)so, 0, sizeof(*so));
    469          1.31   thorpej 	TAILQ_INIT(&so->so_q0);
    470          1.31   thorpej 	TAILQ_INIT(&so->so_q);
    471           1.1       cgd 	so->so_type = type;
    472           1.1       cgd 	so->so_proto = prp;
    473          1.33      matt 	so->so_send = sosend;
    474          1.33      matt 	so->so_receive = soreceive;
    475          1.78      matt #ifdef MBUFTRACE
    476          1.78      matt 	so->so_rcv.sb_mowner = &prp->pr_domain->dom_mowner;
    477          1.78      matt 	so->so_snd.sb_mowner = &prp->pr_domain->dom_mowner;
    478          1.78      matt 	so->so_mowner = &prp->pr_domain->dom_mowner;
    479          1.78      matt #endif
    480          1.44     lukem 	if (p != 0)
    481          1.44     lukem 		so->so_uid = p->p_ucred->cr_uid;
    482          1.22   mycroft 	error = (*prp->pr_usrreq)(so, PRU_ATTACH, (struct mbuf *)0,
    483          1.83      fvdl 	    (struct mbuf *)(long)proto, (struct mbuf *)0, p);
    484           1.1       cgd 	if (error) {
    485           1.1       cgd 		so->so_state |= SS_NOFDREF;
    486           1.1       cgd 		sofree(so);
    487          1.39      matt 		splx(s);
    488           1.1       cgd 		return (error);
    489           1.1       cgd 	}
    490          1.39      matt 	splx(s);
    491           1.1       cgd 	*aso = so;
    492           1.1       cgd 	return (0);
    493           1.1       cgd }
    494           1.1       cgd 
    495           1.3    andrew int
    496          1.83      fvdl sobind(struct socket *so, struct mbuf *nam, struct proc *p)
    497           1.1       cgd {
    498          1.54     lukem 	int	s, error;
    499           1.1       cgd 
    500          1.54     lukem 	s = splsoftnet();
    501          1.22   mycroft 	error = (*so->so_proto->pr_usrreq)(so, PRU_BIND, (struct mbuf *)0,
    502          1.83      fvdl 	    nam, (struct mbuf *)0, p);
    503           1.1       cgd 	splx(s);
    504           1.1       cgd 	return (error);
    505           1.1       cgd }
    506           1.1       cgd 
    507           1.3    andrew int
    508          1.54     lukem solisten(struct socket *so, int backlog)
    509           1.1       cgd {
    510          1.54     lukem 	int	s, error;
    511           1.1       cgd 
    512          1.54     lukem 	s = splsoftnet();
    513          1.22   mycroft 	error = (*so->so_proto->pr_usrreq)(so, PRU_LISTEN, (struct mbuf *)0,
    514          1.83      fvdl 	    (struct mbuf *)0, (struct mbuf *)0, (struct proc *)0);
    515           1.1       cgd 	if (error) {
    516           1.1       cgd 		splx(s);
    517           1.1       cgd 		return (error);
    518           1.1       cgd 	}
    519          1.63      matt 	if (TAILQ_EMPTY(&so->so_q))
    520           1.1       cgd 		so->so_options |= SO_ACCEPTCONN;
    521           1.1       cgd 	if (backlog < 0)
    522           1.1       cgd 		backlog = 0;
    523          1.49  jonathan 	so->so_qlimit = min(backlog, somaxconn);
    524           1.1       cgd 	splx(s);
    525           1.1       cgd 	return (0);
    526           1.1       cgd }
    527           1.1       cgd 
    528          1.21  christos void
    529          1.54     lukem sofree(struct socket *so)
    530           1.1       cgd {
    531           1.1       cgd 
    532          1.43   mycroft 	if (so->so_pcb || (so->so_state & SS_NOFDREF) == 0)
    533           1.1       cgd 		return;
    534          1.43   mycroft 	if (so->so_head) {
    535          1.43   mycroft 		/*
    536          1.43   mycroft 		 * We must not decommission a socket that's on the accept(2)
    537          1.43   mycroft 		 * queue.  If we do, then accept(2) may hang after select(2)
    538          1.43   mycroft 		 * indicated that the listening socket was ready.
    539          1.43   mycroft 		 */
    540          1.43   mycroft 		if (!soqremque(so, 0))
    541          1.43   mycroft 			return;
    542          1.43   mycroft 	}
    543           1.1       cgd 	sbrelease(&so->so_snd);
    544           1.1       cgd 	sorflush(so);
    545          1.37   thorpej 	pool_put(&socket_pool, so);
    546           1.1       cgd }
    547           1.1       cgd 
    548           1.1       cgd /*
    549           1.1       cgd  * Close a socket on last file table reference removal.
    550           1.1       cgd  * Initiate disconnect if connected.
    551           1.1       cgd  * Free socket when disconnect complete.
    552           1.1       cgd  */
    553           1.3    andrew int
    554          1.54     lukem soclose(struct socket *so)
    555           1.1       cgd {
    556          1.54     lukem 	struct socket	*so2;
    557          1.54     lukem 	int		s, error;
    558           1.1       cgd 
    559          1.54     lukem 	error = 0;
    560          1.54     lukem 	s = splsoftnet();		/* conservative */
    561           1.1       cgd 	if (so->so_options & SO_ACCEPTCONN) {
    562          1.63      matt 		while ((so2 = TAILQ_FIRST(&so->so_q0)) != 0) {
    563          1.42   mycroft 			(void) soqremque(so2, 0);
    564          1.41   mycroft 			(void) soabort(so2);
    565          1.41   mycroft 		}
    566          1.63      matt 		while ((so2 = TAILQ_FIRST(&so->so_q)) != 0) {
    567          1.42   mycroft 			(void) soqremque(so2, 1);
    568          1.41   mycroft 			(void) soabort(so2);
    569          1.41   mycroft 		}
    570           1.1       cgd 	}
    571           1.1       cgd 	if (so->so_pcb == 0)
    572           1.1       cgd 		goto discard;
    573           1.1       cgd 	if (so->so_state & SS_ISCONNECTED) {
    574           1.1       cgd 		if ((so->so_state & SS_ISDISCONNECTING) == 0) {
    575           1.1       cgd 			error = sodisconnect(so);
    576           1.1       cgd 			if (error)
    577           1.1       cgd 				goto drop;
    578           1.1       cgd 		}
    579           1.1       cgd 		if (so->so_options & SO_LINGER) {
    580           1.1       cgd 			if ((so->so_state & SS_ISDISCONNECTING) &&
    581           1.1       cgd 			    (so->so_state & SS_NBIO))
    582           1.1       cgd 				goto drop;
    583          1.21  christos 			while (so->so_state & SS_ISCONNECTED) {
    584          1.21  christos 				error = tsleep((caddr_t)&so->so_timeo,
    585          1.21  christos 					       PSOCK | PCATCH, netcls,
    586          1.30   thorpej 					       so->so_linger * hz);
    587          1.21  christos 				if (error)
    588           1.1       cgd 					break;
    589          1.21  christos 			}
    590           1.1       cgd 		}
    591           1.1       cgd 	}
    592          1.54     lukem  drop:
    593           1.1       cgd 	if (so->so_pcb) {
    594          1.22   mycroft 		int error2 = (*so->so_proto->pr_usrreq)(so, PRU_DETACH,
    595          1.22   mycroft 		    (struct mbuf *)0, (struct mbuf *)0, (struct mbuf *)0,
    596          1.83      fvdl 		    (struct proc *)0);
    597           1.1       cgd 		if (error == 0)
    598           1.1       cgd 			error = error2;
    599           1.1       cgd 	}
    600          1.54     lukem  discard:
    601           1.1       cgd 	if (so->so_state & SS_NOFDREF)
    602           1.1       cgd 		panic("soclose: NOFDREF");
    603           1.1       cgd 	so->so_state |= SS_NOFDREF;
    604           1.1       cgd 	sofree(so);
    605           1.1       cgd 	splx(s);
    606           1.1       cgd 	return (error);
    607           1.1       cgd }
    608           1.1       cgd 
    609           1.1       cgd /*
    610          1.20   mycroft  * Must be called at splsoftnet...
    611           1.1       cgd  */
    612           1.3    andrew int
    613          1.54     lukem soabort(struct socket *so)
    614           1.1       cgd {
    615           1.1       cgd 
    616          1.22   mycroft 	return (*so->so_proto->pr_usrreq)(so, PRU_ABORT, (struct mbuf *)0,
    617          1.83      fvdl 	    (struct mbuf *)0, (struct mbuf *)0, (struct proc *)0);
    618           1.1       cgd }
    619           1.1       cgd 
    620           1.3    andrew int
    621          1.54     lukem soaccept(struct socket *so, struct mbuf *nam)
    622           1.1       cgd {
    623          1.54     lukem 	int	s, error;
    624           1.1       cgd 
    625          1.54     lukem 	error = 0;
    626          1.54     lukem 	s = splsoftnet();
    627           1.1       cgd 	if ((so->so_state & SS_NOFDREF) == 0)
    628           1.1       cgd 		panic("soaccept: !NOFDREF");
    629           1.1       cgd 	so->so_state &= ~SS_NOFDREF;
    630          1.55   thorpej 	if ((so->so_state & SS_ISDISCONNECTED) == 0 ||
    631          1.55   thorpej 	    (so->so_proto->pr_flags & PR_ABRTACPTDIS) == 0)
    632          1.41   mycroft 		error = (*so->so_proto->pr_usrreq)(so, PRU_ACCEPT,
    633          1.83      fvdl 		    (struct mbuf *)0, nam, (struct mbuf *)0, (struct proc *)0);
    634          1.41   mycroft 	else
    635          1.53    itojun 		error = ECONNABORTED;
    636          1.52    itojun 
    637           1.1       cgd 	splx(s);
    638           1.1       cgd 	return (error);
    639           1.1       cgd }
    640           1.1       cgd 
    641           1.3    andrew int
    642          1.54     lukem soconnect(struct socket *so, struct mbuf *nam)
    643           1.1       cgd {
    644          1.83      fvdl 	struct proc	*p;
    645          1.54     lukem 	int		s, error;
    646           1.1       cgd 
    647          1.83      fvdl 	p = curproc;		/* XXX */
    648           1.1       cgd 	if (so->so_options & SO_ACCEPTCONN)
    649           1.1       cgd 		return (EOPNOTSUPP);
    650          1.20   mycroft 	s = splsoftnet();
    651           1.1       cgd 	/*
    652           1.1       cgd 	 * If protocol is connection-based, can only connect once.
    653           1.1       cgd 	 * Otherwise, if connected, try to disconnect first.
    654           1.1       cgd 	 * This allows user to disconnect by connecting to, e.g.,
    655           1.1       cgd 	 * a null address.
    656           1.1       cgd 	 */
    657           1.1       cgd 	if (so->so_state & (SS_ISCONNECTED|SS_ISCONNECTING) &&
    658           1.1       cgd 	    ((so->so_proto->pr_flags & PR_CONNREQUIRED) ||
    659           1.1       cgd 	    (error = sodisconnect(so))))
    660           1.1       cgd 		error = EISCONN;
    661           1.1       cgd 	else
    662           1.1       cgd 		error = (*so->so_proto->pr_usrreq)(so, PRU_CONNECT,
    663          1.83      fvdl 		    (struct mbuf *)0, nam, (struct mbuf *)0, p);
    664           1.1       cgd 	splx(s);
    665           1.1       cgd 	return (error);
    666           1.1       cgd }
    667           1.1       cgd 
    668           1.3    andrew int
    669          1.54     lukem soconnect2(struct socket *so1, struct socket *so2)
    670           1.1       cgd {
    671          1.54     lukem 	int	s, error;
    672           1.1       cgd 
    673          1.54     lukem 	s = splsoftnet();
    674          1.22   mycroft 	error = (*so1->so_proto->pr_usrreq)(so1, PRU_CONNECT2,
    675          1.22   mycroft 	    (struct mbuf *)0, (struct mbuf *)so2, (struct mbuf *)0,
    676          1.83      fvdl 	    (struct proc *)0);
    677           1.1       cgd 	splx(s);
    678           1.1       cgd 	return (error);
    679           1.1       cgd }
    680           1.1       cgd 
    681           1.3    andrew int
    682          1.54     lukem sodisconnect(struct socket *so)
    683           1.1       cgd {
    684          1.54     lukem 	int	s, error;
    685           1.1       cgd 
    686          1.54     lukem 	s = splsoftnet();
    687           1.1       cgd 	if ((so->so_state & SS_ISCONNECTED) == 0) {
    688           1.1       cgd 		error = ENOTCONN;
    689           1.1       cgd 		goto bad;
    690           1.1       cgd 	}
    691           1.1       cgd 	if (so->so_state & SS_ISDISCONNECTING) {
    692           1.1       cgd 		error = EALREADY;
    693           1.1       cgd 		goto bad;
    694           1.1       cgd 	}
    695          1.22   mycroft 	error = (*so->so_proto->pr_usrreq)(so, PRU_DISCONNECT,
    696          1.22   mycroft 	    (struct mbuf *)0, (struct mbuf *)0, (struct mbuf *)0,
    697          1.83      fvdl 	    (struct proc *)0);
    698          1.54     lukem  bad:
    699           1.1       cgd 	splx(s);
    700          1.64   thorpej 	sodopendfree(so);
    701           1.1       cgd 	return (error);
    702           1.1       cgd }
    703           1.1       cgd 
    704          1.15   mycroft #define	SBLOCKWAIT(f)	(((f) & MSG_DONTWAIT) ? M_NOWAIT : M_WAITOK)
    705           1.1       cgd /*
    706           1.1       cgd  * Send on a socket.
    707           1.1       cgd  * If send must go all at once and message is larger than
    708           1.1       cgd  * send buffering, then hard error.
    709           1.1       cgd  * Lock against other senders.
    710           1.1       cgd  * If must go all at once and not enough room now, then
    711           1.1       cgd  * inform user that this would block and do nothing.
    712           1.1       cgd  * Otherwise, if nonblocking, send as much as possible.
    713           1.1       cgd  * The data to be sent is described by "uio" if nonzero,
    714           1.1       cgd  * otherwise by the mbuf chain "top" (which must be null
    715           1.1       cgd  * if uio is not).  Data provided in mbuf chain must be small
    716           1.1       cgd  * enough to send all at once.
    717           1.1       cgd  *
    718           1.1       cgd  * Returns nonzero on error, timeout or signal; callers
    719           1.1       cgd  * must check for short counts if EINTR/ERESTART are returned.
    720           1.1       cgd  * Data and control buffers are freed on return.
    721           1.1       cgd  */
    722           1.3    andrew int
    723          1.54     lukem sosend(struct socket *so, struct mbuf *addr, struct uio *uio, struct mbuf *top,
    724          1.54     lukem 	struct mbuf *control, int flags)
    725           1.1       cgd {
    726          1.54     lukem 	struct proc	*p;
    727          1.54     lukem 	struct mbuf	**mp, *m;
    728          1.58  jdolecek 	long		space, len, resid, clen, mlen;
    729          1.58  jdolecek 	int		error, s, dontroute, atomic;
    730          1.54     lukem 
    731          1.64   thorpej 	sodopendfree(so);
    732          1.64   thorpej 
    733          1.83      fvdl 	p = curproc;		/* XXX */
    734          1.54     lukem 	clen = 0;
    735          1.54     lukem 	atomic = sosendallatonce(so) || top;
    736           1.1       cgd 	if (uio)
    737           1.1       cgd 		resid = uio->uio_resid;
    738           1.1       cgd 	else
    739           1.1       cgd 		resid = top->m_pkthdr.len;
    740           1.7       cgd 	/*
    741           1.7       cgd 	 * In theory resid should be unsigned.
    742           1.7       cgd 	 * However, space must be signed, as it might be less than 0
    743           1.7       cgd 	 * if we over-committed, and we must use a signed comparison
    744           1.7       cgd 	 * of space and resid.  On the other hand, a negative resid
    745           1.7       cgd 	 * causes us to loop sending 0-length segments to the protocol.
    746           1.7       cgd 	 */
    747          1.29   mycroft 	if (resid < 0) {
    748          1.29   mycroft 		error = EINVAL;
    749          1.29   mycroft 		goto out;
    750          1.29   mycroft 	}
    751           1.1       cgd 	dontroute =
    752           1.1       cgd 	    (flags & MSG_DONTROUTE) && (so->so_options & SO_DONTROUTE) == 0 &&
    753           1.1       cgd 	    (so->so_proto->pr_flags & PR_ATOMIC);
    754          1.12   mycroft 	p->p_stats->p_ru.ru_msgsnd++;
    755           1.1       cgd 	if (control)
    756           1.1       cgd 		clen = control->m_len;
    757           1.1       cgd #define	snderr(errno)	{ error = errno; splx(s); goto release; }
    758           1.1       cgd 
    759          1.54     lukem  restart:
    760          1.21  christos 	if ((error = sblock(&so->so_snd, SBLOCKWAIT(flags))) != 0)
    761           1.1       cgd 		goto out;
    762           1.1       cgd 	do {
    763          1.20   mycroft 		s = splsoftnet();
    764           1.1       cgd 		if (so->so_state & SS_CANTSENDMORE)
    765           1.1       cgd 			snderr(EPIPE);
    766          1.48   thorpej 		if (so->so_error) {
    767          1.48   thorpej 			error = so->so_error;
    768          1.48   thorpej 			so->so_error = 0;
    769          1.48   thorpej 			splx(s);
    770          1.48   thorpej 			goto release;
    771          1.48   thorpej 		}
    772           1.1       cgd 		if ((so->so_state & SS_ISCONNECTED) == 0) {
    773           1.1       cgd 			if (so->so_proto->pr_flags & PR_CONNREQUIRED) {
    774           1.1       cgd 				if ((so->so_state & SS_ISCONFIRMING) == 0 &&
    775           1.1       cgd 				    !(resid == 0 && clen != 0))
    776           1.1       cgd 					snderr(ENOTCONN);
    777           1.1       cgd 			} else if (addr == 0)
    778           1.1       cgd 				snderr(EDESTADDRREQ);
    779           1.1       cgd 		}
    780           1.1       cgd 		space = sbspace(&so->so_snd);
    781           1.1       cgd 		if (flags & MSG_OOB)
    782           1.1       cgd 			space += 1024;
    783          1.21  christos 		if ((atomic && resid > so->so_snd.sb_hiwat) ||
    784          1.11   mycroft 		    clen > so->so_snd.sb_hiwat)
    785          1.11   mycroft 			snderr(EMSGSIZE);
    786          1.96   mycroft 		if (space < resid + clen &&
    787           1.1       cgd 		    (atomic || space < so->so_snd.sb_lowat || space < clen)) {
    788           1.1       cgd 			if (so->so_state & SS_NBIO)
    789           1.1       cgd 				snderr(EWOULDBLOCK);
    790           1.1       cgd 			sbunlock(&so->so_snd);
    791           1.1       cgd 			error = sbwait(&so->so_snd);
    792           1.1       cgd 			splx(s);
    793           1.1       cgd 			if (error)
    794           1.1       cgd 				goto out;
    795           1.1       cgd 			goto restart;
    796           1.1       cgd 		}
    797           1.1       cgd 		splx(s);
    798           1.1       cgd 		mp = &top;
    799           1.1       cgd 		space -= clen;
    800           1.1       cgd 		do {
    801          1.45        tv 			if (uio == NULL) {
    802          1.45        tv 				/*
    803          1.45        tv 				 * Data is prepackaged in "top".
    804          1.45        tv 				 */
    805          1.45        tv 				resid = 0;
    806          1.45        tv 				if (flags & MSG_EOR)
    807          1.45        tv 					top->m_flags |= M_EOR;
    808          1.45        tv 			} else do {
    809          1.45        tv 				if (top == 0) {
    810          1.78      matt 					m = m_gethdr(M_WAIT, MT_DATA);
    811          1.45        tv 					mlen = MHLEN;
    812          1.45        tv 					m->m_pkthdr.len = 0;
    813          1.45        tv 					m->m_pkthdr.rcvif = (struct ifnet *)0;
    814          1.45        tv 				} else {
    815          1.78      matt 					m = m_get(M_WAIT, MT_DATA);
    816          1.45        tv 					mlen = MLEN;
    817          1.45        tv 				}
    818          1.78      matt 				MCLAIM(m, so->so_snd.sb_mowner);
    819          1.65   thorpej 				if (use_sosend_loan &&
    820          1.65   thorpej 				    uio->uio_iov->iov_len >= SOCK_LOAN_THRESH &&
    821          1.64   thorpej 				    space >= SOCK_LOAN_THRESH &&
    822          1.64   thorpej 				    (len = sosend_loan(so, uio, m,
    823          1.64   thorpej 						       space)) != 0) {
    824          1.64   thorpej 					SOSEND_COUNTER_INCR(&sosend_loan_big);
    825          1.64   thorpej 					space -= len;
    826          1.64   thorpej 					goto have_data;
    827          1.64   thorpej 				}
    828          1.45        tv 				if (resid >= MINCLSIZE && space >= MCLBYTES) {
    829          1.64   thorpej 					SOSEND_COUNTER_INCR(&sosend_copy_big);
    830          1.78      matt 					m_clget(m, M_WAIT);
    831          1.45        tv 					if ((m->m_flags & M_EXT) == 0)
    832          1.45        tv 						goto nopages;
    833          1.45        tv 					mlen = MCLBYTES;
    834          1.45        tv 					if (atomic && top == 0) {
    835          1.58  jdolecek 						len = lmin(MCLBYTES - max_hdr,
    836          1.54     lukem 						    resid);
    837          1.45        tv 						m->m_data += max_hdr;
    838          1.45        tv 					} else
    839          1.58  jdolecek 						len = lmin(MCLBYTES, resid);
    840          1.45        tv 					space -= len;
    841          1.45        tv 				} else {
    842          1.64   thorpej  nopages:
    843          1.64   thorpej 					SOSEND_COUNTER_INCR(&sosend_copy_small);
    844          1.58  jdolecek 					len = lmin(lmin(mlen, resid), space);
    845          1.45        tv 					space -= len;
    846          1.45        tv 					/*
    847          1.45        tv 					 * For datagram protocols, leave room
    848          1.45        tv 					 * for protocol headers in first mbuf.
    849          1.45        tv 					 */
    850          1.45        tv 					if (atomic && top == 0 && len < mlen)
    851          1.45        tv 						MH_ALIGN(m, len);
    852          1.45        tv 				}
    853          1.54     lukem 				error = uiomove(mtod(m, caddr_t), (int)len,
    854          1.54     lukem 				    uio);
    855          1.64   thorpej  have_data:
    856          1.45        tv 				resid = uio->uio_resid;
    857          1.45        tv 				m->m_len = len;
    858          1.45        tv 				*mp = m;
    859          1.45        tv 				top->m_pkthdr.len += len;
    860          1.45        tv 				if (error)
    861          1.45        tv 					goto release;
    862          1.45        tv 				mp = &m->m_next;
    863          1.45        tv 				if (resid <= 0) {
    864          1.45        tv 					if (flags & MSG_EOR)
    865          1.45        tv 						top->m_flags |= M_EOR;
    866          1.45        tv 					break;
    867          1.45        tv 				}
    868          1.45        tv 			} while (space > 0 && atomic);
    869          1.46  sommerfe 
    870          1.46  sommerfe 			s = splsoftnet();
    871          1.46  sommerfe 
    872          1.46  sommerfe 			if (so->so_state & SS_CANTSENDMORE)
    873          1.46  sommerfe 				snderr(EPIPE);
    874          1.45        tv 
    875          1.45        tv 			if (dontroute)
    876          1.45        tv 				so->so_options |= SO_DONTROUTE;
    877          1.45        tv 			if (resid > 0)
    878          1.45        tv 				so->so_state |= SS_MORETOCOME;
    879          1.46  sommerfe 			error = (*so->so_proto->pr_usrreq)(so,
    880          1.46  sommerfe 			    (flags & MSG_OOB) ? PRU_SENDOOB : PRU_SEND,
    881          1.83      fvdl 			    top, addr, control, p);
    882          1.45        tv 			if (dontroute)
    883          1.45        tv 				so->so_options &= ~SO_DONTROUTE;
    884          1.45        tv 			if (resid > 0)
    885          1.45        tv 				so->so_state &= ~SS_MORETOCOME;
    886          1.46  sommerfe 			splx(s);
    887          1.46  sommerfe 
    888          1.45        tv 			clen = 0;
    889          1.45        tv 			control = 0;
    890          1.45        tv 			top = 0;
    891          1.45        tv 			mp = &top;
    892           1.1       cgd 			if (error)
    893           1.1       cgd 				goto release;
    894           1.1       cgd 		} while (resid && space > 0);
    895           1.1       cgd 	} while (resid);
    896           1.1       cgd 
    897          1.54     lukem  release:
    898           1.1       cgd 	sbunlock(&so->so_snd);
    899          1.54     lukem  out:
    900           1.1       cgd 	if (top)
    901           1.1       cgd 		m_freem(top);
    902           1.1       cgd 	if (control)
    903           1.1       cgd 		m_freem(control);
    904           1.1       cgd 	return (error);
    905           1.1       cgd }
    906           1.1       cgd 
    907           1.1       cgd /*
    908           1.1       cgd  * Implement receive operations on a socket.
    909           1.1       cgd  * We depend on the way that records are added to the sockbuf
    910           1.1       cgd  * by sbappend*.  In particular, each record (mbufs linked through m_next)
    911           1.1       cgd  * must begin with an address if the protocol so specifies,
    912           1.1       cgd  * followed by an optional mbuf or mbufs containing ancillary data,
    913           1.1       cgd  * and then zero or more mbufs of data.
    914           1.1       cgd  * In order to avoid blocking network interrupts for the entire time here,
    915           1.1       cgd  * we splx() while doing the actual copy to user space.
    916           1.1       cgd  * Although the sockbuf is locked, new data may still be appended,
    917           1.1       cgd  * and thus we must maintain consistency of the sockbuf during that time.
    918           1.1       cgd  *
    919           1.1       cgd  * The caller may receive the data as a single mbuf chain by supplying
    920           1.1       cgd  * an mbuf **mp0 for use in returning the chain.  The uio is then used
    921           1.1       cgd  * only for the count in uio_resid.
    922           1.1       cgd  */
    923           1.3    andrew int
    924          1.54     lukem soreceive(struct socket *so, struct mbuf **paddr, struct uio *uio,
    925          1.54     lukem 	struct mbuf **mp0, struct mbuf **controlp, int *flagsp)
    926           1.1       cgd {
    927          1.54     lukem 	struct mbuf	*m, **mp;
    928          1.54     lukem 	int		flags, len, error, s, offset, moff, type, orig_resid;
    929          1.54     lukem 	struct protosw	*pr;
    930          1.54     lukem 	struct mbuf	*nextrecord;
    931          1.67        he 	int		mbuf_removed = 0;
    932          1.64   thorpej 
    933          1.54     lukem 	pr = so->so_proto;
    934           1.1       cgd 	mp = mp0;
    935          1.54     lukem 	type = 0;
    936          1.54     lukem 	orig_resid = uio->uio_resid;
    937           1.1       cgd 	if (paddr)
    938           1.1       cgd 		*paddr = 0;
    939           1.1       cgd 	if (controlp)
    940           1.1       cgd 		*controlp = 0;
    941           1.1       cgd 	if (flagsp)
    942           1.1       cgd 		flags = *flagsp &~ MSG_EOR;
    943           1.1       cgd 	else
    944           1.1       cgd 		flags = 0;
    945          1.66     enami 
    946          1.66     enami 	if ((flags & MSG_DONTWAIT) == 0)
    947          1.66     enami 		sodopendfree(so);
    948          1.66     enami 
    949           1.1       cgd 	if (flags & MSG_OOB) {
    950           1.1       cgd 		m = m_get(M_WAIT, MT_DATA);
    951          1.17       cgd 		error = (*pr->pr_usrreq)(so, PRU_RCVOOB, m,
    952          1.22   mycroft 		    (struct mbuf *)(long)(flags & MSG_PEEK), (struct mbuf *)0,
    953          1.83      fvdl 		    (struct proc *)0);
    954           1.1       cgd 		if (error)
    955           1.1       cgd 			goto bad;
    956           1.1       cgd 		do {
    957           1.1       cgd 			error = uiomove(mtod(m, caddr_t),
    958           1.1       cgd 			    (int) min(uio->uio_resid, m->m_len), uio);
    959           1.1       cgd 			m = m_free(m);
    960           1.1       cgd 		} while (uio->uio_resid && error == 0 && m);
    961          1.54     lukem  bad:
    962           1.1       cgd 		if (m)
    963           1.1       cgd 			m_freem(m);
    964           1.1       cgd 		return (error);
    965           1.1       cgd 	}
    966           1.1       cgd 	if (mp)
    967           1.1       cgd 		*mp = (struct mbuf *)0;
    968           1.1       cgd 	if (so->so_state & SS_ISCONFIRMING && uio->uio_resid)
    969          1.22   mycroft 		(*pr->pr_usrreq)(so, PRU_RCVD, (struct mbuf *)0,
    970          1.83      fvdl 		    (struct mbuf *)0, (struct mbuf *)0, (struct proc *)0);
    971           1.1       cgd 
    972          1.54     lukem  restart:
    973          1.21  christos 	if ((error = sblock(&so->so_rcv, SBLOCKWAIT(flags))) != 0)
    974           1.1       cgd 		return (error);
    975          1.20   mycroft 	s = splsoftnet();
    976           1.1       cgd 
    977           1.1       cgd 	m = so->so_rcv.sb_mb;
    978           1.1       cgd 	/*
    979           1.1       cgd 	 * If we have less data than requested, block awaiting more
    980           1.1       cgd 	 * (subject to any timeout) if:
    981          1.15   mycroft 	 *   1. the current count is less than the low water mark,
    982           1.1       cgd 	 *   2. MSG_WAITALL is set, and it is possible to do the entire
    983          1.15   mycroft 	 *	receive operation at once if we block (resid <= hiwat), or
    984          1.15   mycroft 	 *   3. MSG_DONTWAIT is not set.
    985           1.1       cgd 	 * If MSG_WAITALL is set but resid is larger than the receive buffer,
    986           1.1       cgd 	 * we have to do the receive in sections, and thus risk returning
    987           1.1       cgd 	 * a short count if a timeout or signal occurs after we start.
    988           1.1       cgd 	 */
    989          1.21  christos 	if (m == 0 || (((flags & MSG_DONTWAIT) == 0 &&
    990          1.15   mycroft 	    so->so_rcv.sb_cc < uio->uio_resid) &&
    991           1.1       cgd 	    (so->so_rcv.sb_cc < so->so_rcv.sb_lowat ||
    992           1.1       cgd 	    ((flags & MSG_WAITALL) && uio->uio_resid <= so->so_rcv.sb_hiwat)) &&
    993          1.21  christos 	    m->m_nextpkt == 0 && (pr->pr_flags & PR_ATOMIC) == 0)) {
    994           1.1       cgd #ifdef DIAGNOSTIC
    995           1.1       cgd 		if (m == 0 && so->so_rcv.sb_cc)
    996           1.1       cgd 			panic("receive 1");
    997           1.1       cgd #endif
    998           1.1       cgd 		if (so->so_error) {
    999           1.1       cgd 			if (m)
   1000          1.15   mycroft 				goto dontblock;
   1001           1.1       cgd 			error = so->so_error;
   1002           1.1       cgd 			if ((flags & MSG_PEEK) == 0)
   1003           1.1       cgd 				so->so_error = 0;
   1004           1.1       cgd 			goto release;
   1005           1.1       cgd 		}
   1006           1.1       cgd 		if (so->so_state & SS_CANTRCVMORE) {
   1007           1.1       cgd 			if (m)
   1008          1.15   mycroft 				goto dontblock;
   1009           1.1       cgd 			else
   1010           1.1       cgd 				goto release;
   1011           1.1       cgd 		}
   1012           1.1       cgd 		for (; m; m = m->m_next)
   1013           1.1       cgd 			if (m->m_type == MT_OOBDATA  || (m->m_flags & M_EOR)) {
   1014           1.1       cgd 				m = so->so_rcv.sb_mb;
   1015           1.1       cgd 				goto dontblock;
   1016           1.1       cgd 			}
   1017           1.1       cgd 		if ((so->so_state & (SS_ISCONNECTED|SS_ISCONNECTING)) == 0 &&
   1018           1.1       cgd 		    (so->so_proto->pr_flags & PR_CONNREQUIRED)) {
   1019           1.1       cgd 			error = ENOTCONN;
   1020           1.1       cgd 			goto release;
   1021           1.1       cgd 		}
   1022           1.1       cgd 		if (uio->uio_resid == 0)
   1023           1.1       cgd 			goto release;
   1024          1.15   mycroft 		if ((so->so_state & SS_NBIO) || (flags & MSG_DONTWAIT)) {
   1025           1.1       cgd 			error = EWOULDBLOCK;
   1026           1.1       cgd 			goto release;
   1027           1.1       cgd 		}
   1028          1.69   thorpej 		SBLASTRECORDCHK(&so->so_rcv, "soreceive sbwait 1");
   1029          1.69   thorpej 		SBLASTMBUFCHK(&so->so_rcv, "soreceive sbwait 1");
   1030           1.1       cgd 		sbunlock(&so->so_rcv);
   1031           1.1       cgd 		error = sbwait(&so->so_rcv);
   1032           1.1       cgd 		splx(s);
   1033           1.1       cgd 		if (error)
   1034           1.1       cgd 			return (error);
   1035           1.1       cgd 		goto restart;
   1036           1.1       cgd 	}
   1037          1.54     lukem  dontblock:
   1038          1.69   thorpej 	/*
   1039          1.69   thorpej 	 * On entry here, m points to the first record of the socket buffer.
   1040          1.69   thorpej 	 * While we process the initial mbufs containing address and control
   1041          1.69   thorpej 	 * info, we save a copy of m->m_nextpkt into nextrecord.
   1042          1.69   thorpej 	 */
   1043  1.97.2.1.2.2       riz #ifdef notyet /* XXXX */
   1044          1.83      fvdl 	if (uio->uio_procp)
   1045          1.83      fvdl 		uio->uio_procp->p_stats->p_ru.ru_msgrcv++;
   1046  1.97.2.1.2.2       riz #endif
   1047          1.69   thorpej 	KASSERT(m == so->so_rcv.sb_mb);
   1048          1.69   thorpej 	SBLASTRECORDCHK(&so->so_rcv, "soreceive 1");
   1049          1.69   thorpej 	SBLASTMBUFCHK(&so->so_rcv, "soreceive 1");
   1050           1.1       cgd 	nextrecord = m->m_nextpkt;
   1051           1.1       cgd 	if (pr->pr_flags & PR_ADDR) {
   1052           1.1       cgd #ifdef DIAGNOSTIC
   1053           1.1       cgd 		if (m->m_type != MT_SONAME)
   1054           1.1       cgd 			panic("receive 1a");
   1055           1.1       cgd #endif
   1056           1.3    andrew 		orig_resid = 0;
   1057           1.1       cgd 		if (flags & MSG_PEEK) {
   1058           1.1       cgd 			if (paddr)
   1059           1.1       cgd 				*paddr = m_copy(m, 0, m->m_len);
   1060           1.1       cgd 			m = m->m_next;
   1061           1.1       cgd 		} else {
   1062           1.1       cgd 			sbfree(&so->so_rcv, m);
   1063          1.67        he 			mbuf_removed = 1;
   1064           1.1       cgd 			if (paddr) {
   1065           1.1       cgd 				*paddr = m;
   1066           1.1       cgd 				so->so_rcv.sb_mb = m->m_next;
   1067           1.1       cgd 				m->m_next = 0;
   1068           1.1       cgd 				m = so->so_rcv.sb_mb;
   1069           1.1       cgd 			} else {
   1070           1.1       cgd 				MFREE(m, so->so_rcv.sb_mb);
   1071           1.1       cgd 				m = so->so_rcv.sb_mb;
   1072           1.1       cgd 			}
   1073           1.1       cgd 		}
   1074           1.1       cgd 	}
   1075           1.1       cgd 	while (m && m->m_type == MT_CONTROL && error == 0) {
   1076           1.1       cgd 		if (flags & MSG_PEEK) {
   1077           1.1       cgd 			if (controlp)
   1078           1.1       cgd 				*controlp = m_copy(m, 0, m->m_len);
   1079           1.1       cgd 			m = m->m_next;
   1080           1.1       cgd 		} else {
   1081           1.1       cgd 			sbfree(&so->so_rcv, m);
   1082          1.67        he 			mbuf_removed = 1;
   1083           1.1       cgd 			if (controlp) {
   1084           1.1       cgd 				if (pr->pr_domain->dom_externalize &&
   1085           1.1       cgd 				    mtod(m, struct cmsghdr *)->cmsg_type ==
   1086           1.1       cgd 				    SCM_RIGHTS)
   1087          1.45        tv 					error = (*pr->pr_domain->dom_externalize)(m);
   1088           1.1       cgd 				*controlp = m;
   1089           1.1       cgd 				so->so_rcv.sb_mb = m->m_next;
   1090           1.1       cgd 				m->m_next = 0;
   1091           1.1       cgd 				m = so->so_rcv.sb_mb;
   1092           1.1       cgd 			} else {
   1093           1.1       cgd 				MFREE(m, so->so_rcv.sb_mb);
   1094           1.1       cgd 				m = so->so_rcv.sb_mb;
   1095           1.1       cgd 			}
   1096           1.1       cgd 		}
   1097           1.3    andrew 		if (controlp) {
   1098           1.3    andrew 			orig_resid = 0;
   1099           1.1       cgd 			controlp = &(*controlp)->m_next;
   1100           1.3    andrew 		}
   1101           1.1       cgd 	}
   1102          1.69   thorpej 
   1103          1.69   thorpej 	/*
   1104          1.69   thorpej 	 * If m is non-NULL, we have some data to read.  From now on,
   1105          1.69   thorpej 	 * make sure to keep sb_lastrecord consistent when working on
   1106          1.69   thorpej 	 * the last packet on the chain (nextrecord == NULL) and we
   1107          1.69   thorpej 	 * change m->m_nextpkt.
   1108          1.69   thorpej 	 */
   1109           1.1       cgd 	if (m) {
   1110          1.69   thorpej 		if ((flags & MSG_PEEK) == 0) {
   1111           1.1       cgd 			m->m_nextpkt = nextrecord;
   1112          1.69   thorpej 			/*
   1113          1.69   thorpej 			 * If nextrecord == NULL (this is a single chain),
   1114          1.69   thorpej 			 * then sb_lastrecord may not be valid here if m
   1115          1.69   thorpej 			 * was changed earlier.
   1116          1.69   thorpej 			 */
   1117          1.69   thorpej 			if (nextrecord == NULL) {
   1118          1.69   thorpej 				KASSERT(so->so_rcv.sb_mb == m);
   1119          1.69   thorpej 				so->so_rcv.sb_lastrecord = m;
   1120          1.69   thorpej 			}
   1121          1.69   thorpej 		}
   1122           1.1       cgd 		type = m->m_type;
   1123           1.1       cgd 		if (type == MT_OOBDATA)
   1124           1.1       cgd 			flags |= MSG_OOB;
   1125          1.69   thorpej 	} else {
   1126          1.69   thorpej 		if ((flags & MSG_PEEK) == 0) {
   1127          1.69   thorpej 			KASSERT(so->so_rcv.sb_mb == m);
   1128          1.69   thorpej 			so->so_rcv.sb_mb = nextrecord;
   1129          1.70   thorpej 			SB_EMPTY_FIXUP(&so->so_rcv);
   1130          1.69   thorpej 		}
   1131           1.1       cgd 	}
   1132          1.69   thorpej 	SBLASTRECORDCHK(&so->so_rcv, "soreceive 2");
   1133          1.69   thorpej 	SBLASTMBUFCHK(&so->so_rcv, "soreceive 2");
   1134          1.69   thorpej 
   1135           1.1       cgd 	moff = 0;
   1136           1.1       cgd 	offset = 0;
   1137           1.1       cgd 	while (m && uio->uio_resid > 0 && error == 0) {
   1138           1.1       cgd 		if (m->m_type == MT_OOBDATA) {
   1139           1.1       cgd 			if (type != MT_OOBDATA)
   1140           1.1       cgd 				break;
   1141           1.1       cgd 		} else if (type == MT_OOBDATA)
   1142           1.1       cgd 			break;
   1143           1.1       cgd #ifdef DIAGNOSTIC
   1144           1.1       cgd 		else if (m->m_type != MT_DATA && m->m_type != MT_HEADER)
   1145           1.1       cgd 			panic("receive 3");
   1146           1.1       cgd #endif
   1147           1.1       cgd 		so->so_state &= ~SS_RCVATMARK;
   1148           1.1       cgd 		len = uio->uio_resid;
   1149           1.1       cgd 		if (so->so_oobmark && len > so->so_oobmark - offset)
   1150           1.1       cgd 			len = so->so_oobmark - offset;
   1151           1.1       cgd 		if (len > m->m_len - moff)
   1152           1.1       cgd 			len = m->m_len - moff;
   1153           1.1       cgd 		/*
   1154           1.1       cgd 		 * If mp is set, just pass back the mbufs.
   1155           1.1       cgd 		 * Otherwise copy them out via the uio, then free.
   1156           1.1       cgd 		 * Sockbuf must be consistent here (points to current mbuf,
   1157           1.1       cgd 		 * it points to next record) when we drop priority;
   1158           1.1       cgd 		 * we must note any additions to the sockbuf when we
   1159           1.1       cgd 		 * block interrupts again.
   1160           1.1       cgd 		 */
   1161           1.1       cgd 		if (mp == 0) {
   1162          1.69   thorpej 			SBLASTRECORDCHK(&so->so_rcv, "soreceive uiomove");
   1163          1.69   thorpej 			SBLASTMBUFCHK(&so->so_rcv, "soreceive uiomove");
   1164           1.1       cgd 			splx(s);
   1165           1.1       cgd 			error = uiomove(mtod(m, caddr_t) + moff, (int)len, uio);
   1166          1.20   mycroft 			s = splsoftnet();
   1167          1.67        he 			if (error) {
   1168          1.67        he 				/*
   1169          1.67        he 				 * If any part of the record has been removed
   1170          1.67        he 				 * (such as the MT_SONAME mbuf, which will
   1171          1.67        he 				 * happen when PR_ADDR, and thus also
   1172          1.67        he 				 * PR_ATOMIC, is set), then drop the entire
   1173          1.67        he 				 * record to maintain the atomicity of the
   1174          1.67        he 				 * receive operation.
   1175          1.67        he 				 *
   1176          1.67        he 				 * This avoids a later panic("receive 1a")
   1177          1.67        he 				 * when compiled with DIAGNOSTIC.
   1178          1.67        he 				 */
   1179          1.67        he 				if (m && mbuf_removed
   1180          1.67        he 				    && (pr->pr_flags & PR_ATOMIC))
   1181          1.67        he 					(void) sbdroprecord(&so->so_rcv);
   1182          1.67        he 
   1183          1.57  jdolecek 				goto release;
   1184          1.67        he 			}
   1185           1.1       cgd 		} else
   1186           1.1       cgd 			uio->uio_resid -= len;
   1187           1.1       cgd 		if (len == m->m_len - moff) {
   1188           1.1       cgd 			if (m->m_flags & M_EOR)
   1189           1.1       cgd 				flags |= MSG_EOR;
   1190           1.1       cgd 			if (flags & MSG_PEEK) {
   1191           1.1       cgd 				m = m->m_next;
   1192           1.1       cgd 				moff = 0;
   1193           1.1       cgd 			} else {
   1194           1.1       cgd 				nextrecord = m->m_nextpkt;
   1195           1.1       cgd 				sbfree(&so->so_rcv, m);
   1196           1.1       cgd 				if (mp) {
   1197           1.1       cgd 					*mp = m;
   1198           1.1       cgd 					mp = &m->m_next;
   1199           1.1       cgd 					so->so_rcv.sb_mb = m = m->m_next;
   1200           1.1       cgd 					*mp = (struct mbuf *)0;
   1201           1.1       cgd 				} else {
   1202           1.1       cgd 					MFREE(m, so->so_rcv.sb_mb);
   1203           1.1       cgd 					m = so->so_rcv.sb_mb;
   1204           1.1       cgd 				}
   1205          1.69   thorpej 				/*
   1206          1.69   thorpej 				 * If m != NULL, we also know that
   1207          1.69   thorpej 				 * so->so_rcv.sb_mb != NULL.
   1208          1.69   thorpej 				 */
   1209          1.69   thorpej 				KASSERT(so->so_rcv.sb_mb == m);
   1210          1.69   thorpej 				if (m) {
   1211           1.1       cgd 					m->m_nextpkt = nextrecord;
   1212          1.69   thorpej 					if (nextrecord == NULL)
   1213          1.69   thorpej 						so->so_rcv.sb_lastrecord = m;
   1214          1.69   thorpej 				} else {
   1215          1.69   thorpej 					so->so_rcv.sb_mb = nextrecord;
   1216          1.70   thorpej 					SB_EMPTY_FIXUP(&so->so_rcv);
   1217          1.69   thorpej 				}
   1218          1.69   thorpej 				SBLASTRECORDCHK(&so->so_rcv, "soreceive 3");
   1219          1.69   thorpej 				SBLASTMBUFCHK(&so->so_rcv, "soreceive 3");
   1220           1.1       cgd 			}
   1221           1.1       cgd 		} else {
   1222           1.1       cgd 			if (flags & MSG_PEEK)
   1223           1.1       cgd 				moff += len;
   1224           1.1       cgd 			else {
   1225           1.1       cgd 				if (mp)
   1226           1.1       cgd 					*mp = m_copym(m, 0, len, M_WAIT);
   1227           1.1       cgd 				m->m_data += len;
   1228           1.1       cgd 				m->m_len -= len;
   1229           1.1       cgd 				so->so_rcv.sb_cc -= len;
   1230           1.1       cgd 			}
   1231           1.1       cgd 		}
   1232           1.1       cgd 		if (so->so_oobmark) {
   1233           1.1       cgd 			if ((flags & MSG_PEEK) == 0) {
   1234           1.1       cgd 				so->so_oobmark -= len;
   1235           1.1       cgd 				if (so->so_oobmark == 0) {
   1236           1.1       cgd 					so->so_state |= SS_RCVATMARK;
   1237           1.1       cgd 					break;
   1238           1.1       cgd 				}
   1239           1.7       cgd 			} else {
   1240           1.1       cgd 				offset += len;
   1241           1.7       cgd 				if (offset == so->so_oobmark)
   1242           1.7       cgd 					break;
   1243           1.7       cgd 			}
   1244           1.1       cgd 		}
   1245           1.1       cgd 		if (flags & MSG_EOR)
   1246           1.1       cgd 			break;
   1247           1.1       cgd 		/*
   1248           1.1       cgd 		 * If the MSG_WAITALL flag is set (for non-atomic socket),
   1249           1.1       cgd 		 * we must not quit until "uio->uio_resid == 0" or an error
   1250           1.1       cgd 		 * termination.  If a signal/timeout occurs, return
   1251           1.1       cgd 		 * with a short count but without error.
   1252           1.1       cgd 		 * Keep sockbuf locked against other readers.
   1253           1.1       cgd 		 */
   1254           1.1       cgd 		while (flags & MSG_WAITALL && m == 0 && uio->uio_resid > 0 &&
   1255           1.3    andrew 		    !sosendallatonce(so) && !nextrecord) {
   1256           1.1       cgd 			if (so->so_error || so->so_state & SS_CANTRCVMORE)
   1257           1.1       cgd 				break;
   1258          1.68      matt 			/*
   1259          1.68      matt 			 * If we are peeking and the socket receive buffer is
   1260          1.68      matt 			 * full, stop since we can't get more data to peek at.
   1261          1.68      matt 			 */
   1262          1.68      matt 			if ((flags & MSG_PEEK) && sbspace(&so->so_rcv) <= 0)
   1263          1.68      matt 				break;
   1264          1.68      matt 			/*
   1265          1.68      matt 			 * If we've drained the socket buffer, tell the
   1266          1.68      matt 			 * protocol in case it needs to do something to
   1267          1.68      matt 			 * get it filled again.
   1268          1.68      matt 			 */
   1269          1.68      matt 			if ((pr->pr_flags & PR_WANTRCVD) && so->so_pcb)
   1270          1.68      matt 				(*pr->pr_usrreq)(so, PRU_RCVD,
   1271          1.68      matt 				    (struct mbuf *)0,
   1272          1.68      matt 				    (struct mbuf *)(long)flags,
   1273          1.68      matt 				    (struct mbuf *)0,
   1274          1.83      fvdl 				    (struct proc *)0);
   1275          1.69   thorpej 			SBLASTRECORDCHK(&so->so_rcv, "soreceive sbwait 2");
   1276          1.69   thorpej 			SBLASTMBUFCHK(&so->so_rcv, "soreceive sbwait 2");
   1277           1.1       cgd 			error = sbwait(&so->so_rcv);
   1278           1.1       cgd 			if (error) {
   1279           1.1       cgd 				sbunlock(&so->so_rcv);
   1280           1.1       cgd 				splx(s);
   1281           1.1       cgd 				return (0);
   1282           1.1       cgd 			}
   1283          1.21  christos 			if ((m = so->so_rcv.sb_mb) != NULL)
   1284           1.1       cgd 				nextrecord = m->m_nextpkt;
   1285           1.1       cgd 		}
   1286           1.1       cgd 	}
   1287           1.3    andrew 
   1288           1.3    andrew 	if (m && pr->pr_flags & PR_ATOMIC) {
   1289           1.3    andrew 		flags |= MSG_TRUNC;
   1290           1.3    andrew 		if ((flags & MSG_PEEK) == 0)
   1291           1.3    andrew 			(void) sbdroprecord(&so->so_rcv);
   1292           1.3    andrew 	}
   1293           1.1       cgd 	if ((flags & MSG_PEEK) == 0) {
   1294          1.69   thorpej 		if (m == 0) {
   1295          1.69   thorpej 			/*
   1296          1.70   thorpej 			 * First part is an inline SB_EMPTY_FIXUP().  Second
   1297          1.69   thorpej 			 * part makes sure sb_lastrecord is up-to-date if
   1298          1.69   thorpej 			 * there is still data in the socket buffer.
   1299          1.69   thorpej 			 */
   1300           1.1       cgd 			so->so_rcv.sb_mb = nextrecord;
   1301          1.69   thorpej 			if (so->so_rcv.sb_mb == NULL) {
   1302          1.69   thorpej 				so->so_rcv.sb_mbtail = NULL;
   1303          1.69   thorpej 				so->so_rcv.sb_lastrecord = NULL;
   1304          1.69   thorpej 			} else if (nextrecord->m_nextpkt == NULL)
   1305          1.69   thorpej 				so->so_rcv.sb_lastrecord = nextrecord;
   1306          1.69   thorpej 		}
   1307          1.69   thorpej 		SBLASTRECORDCHK(&so->so_rcv, "soreceive 4");
   1308          1.69   thorpej 		SBLASTMBUFCHK(&so->so_rcv, "soreceive 4");
   1309           1.1       cgd 		if (pr->pr_flags & PR_WANTRCVD && so->so_pcb)
   1310          1.22   mycroft 			(*pr->pr_usrreq)(so, PRU_RCVD, (struct mbuf *)0,
   1311          1.22   mycroft 			    (struct mbuf *)(long)flags, (struct mbuf *)0,
   1312          1.83      fvdl 			    (struct proc *)0);
   1313           1.1       cgd 	}
   1314           1.3    andrew 	if (orig_resid == uio->uio_resid && orig_resid &&
   1315           1.3    andrew 	    (flags & MSG_EOR) == 0 && (so->so_state & SS_CANTRCVMORE) == 0) {
   1316           1.3    andrew 		sbunlock(&so->so_rcv);
   1317           1.3    andrew 		splx(s);
   1318           1.3    andrew 		goto restart;
   1319           1.3    andrew 	}
   1320           1.3    andrew 
   1321           1.1       cgd 	if (flagsp)
   1322           1.1       cgd 		*flagsp |= flags;
   1323          1.54     lukem  release:
   1324           1.1       cgd 	sbunlock(&so->so_rcv);
   1325           1.1       cgd 	splx(s);
   1326           1.1       cgd 	return (error);
   1327           1.1       cgd }
   1328           1.1       cgd 
   1329          1.14   mycroft int
   1330          1.54     lukem soshutdown(struct socket *so, int how)
   1331           1.1       cgd {
   1332          1.54     lukem 	struct protosw	*pr;
   1333          1.34    kleink 
   1334          1.54     lukem 	pr = so->so_proto;
   1335          1.34    kleink 	if (!(how == SHUT_RD || how == SHUT_WR || how == SHUT_RDWR))
   1336          1.34    kleink 		return (EINVAL);
   1337           1.1       cgd 
   1338          1.34    kleink 	if (how == SHUT_RD || how == SHUT_RDWR)
   1339           1.1       cgd 		sorflush(so);
   1340          1.34    kleink 	if (how == SHUT_WR || how == SHUT_RDWR)
   1341          1.22   mycroft 		return (*pr->pr_usrreq)(so, PRU_SHUTDOWN, (struct mbuf *)0,
   1342          1.83      fvdl 		    (struct mbuf *)0, (struct mbuf *)0, (struct proc *)0);
   1343           1.1       cgd 	return (0);
   1344           1.1       cgd }
   1345           1.1       cgd 
   1346          1.14   mycroft void
   1347          1.54     lukem sorflush(struct socket *so)
   1348           1.1       cgd {
   1349          1.54     lukem 	struct sockbuf	*sb, asb;
   1350          1.54     lukem 	struct protosw	*pr;
   1351          1.54     lukem 	int		s;
   1352           1.1       cgd 
   1353          1.54     lukem 	sb = &so->so_rcv;
   1354          1.54     lukem 	pr = so->so_proto;
   1355           1.1       cgd 	sb->sb_flags |= SB_NOINTR;
   1356          1.15   mycroft 	(void) sblock(sb, M_WAITOK);
   1357          1.56   thorpej 	s = splnet();
   1358           1.1       cgd 	socantrcvmore(so);
   1359           1.1       cgd 	sbunlock(sb);
   1360           1.1       cgd 	asb = *sb;
   1361          1.86  wrstuden 	/*
   1362          1.86  wrstuden 	 * Clear most of the sockbuf structure, but leave some of the
   1363          1.86  wrstuden 	 * fields valid.
   1364          1.86  wrstuden 	 */
   1365          1.86  wrstuden 	memset(&sb->sb_startzero, 0,
   1366          1.86  wrstuden 	    sizeof(*sb) - offsetof(struct sockbuf, sb_startzero));
   1367           1.1       cgd 	splx(s);
   1368           1.1       cgd 	if (pr->pr_flags & PR_RIGHTS && pr->pr_domain->dom_dispose)
   1369           1.1       cgd 		(*pr->pr_domain->dom_dispose)(asb.sb_mb);
   1370           1.1       cgd 	sbrelease(&asb);
   1371           1.1       cgd }
   1372           1.1       cgd 
   1373          1.14   mycroft int
   1374          1.54     lukem sosetopt(struct socket *so, int level, int optname, struct mbuf *m0)
   1375           1.1       cgd {
   1376          1.54     lukem 	int		error;
   1377          1.54     lukem 	struct mbuf	*m;
   1378           1.1       cgd 
   1379          1.54     lukem 	error = 0;
   1380          1.54     lukem 	m = m0;
   1381           1.1       cgd 	if (level != SOL_SOCKET) {
   1382           1.1       cgd 		if (so->so_proto && so->so_proto->pr_ctloutput)
   1383           1.1       cgd 			return ((*so->so_proto->pr_ctloutput)
   1384           1.1       cgd 				  (PRCO_SETOPT, so, level, optname, &m0));
   1385           1.1       cgd 		error = ENOPROTOOPT;
   1386           1.1       cgd 	} else {
   1387           1.1       cgd 		switch (optname) {
   1388           1.1       cgd 
   1389           1.1       cgd 		case SO_LINGER:
   1390          1.36     perry 			if (m == NULL || m->m_len != sizeof(struct linger)) {
   1391           1.1       cgd 				error = EINVAL;
   1392           1.1       cgd 				goto bad;
   1393           1.1       cgd 			}
   1394  1.97.2.1.2.3       jmc 			if (mtod(m, struct linger *)->l_linger < 0 ||
   1395  1.97.2.1.2.3       jmc 			    mtod(m, struct linger *)->l_linger > (INT_MAX / hz)) {
   1396  1.97.2.1.2.3       jmc 				error = EDOM;
   1397  1.97.2.1.2.3       jmc 				goto bad;
   1398  1.97.2.1.2.3       jmc 			}
   1399           1.1       cgd 			so->so_linger = mtod(m, struct linger *)->l_linger;
   1400           1.1       cgd 			/* fall thru... */
   1401           1.1       cgd 
   1402           1.1       cgd 		case SO_DEBUG:
   1403           1.1       cgd 		case SO_KEEPALIVE:
   1404           1.1       cgd 		case SO_DONTROUTE:
   1405           1.1       cgd 		case SO_USELOOPBACK:
   1406           1.1       cgd 		case SO_BROADCAST:
   1407           1.1       cgd 		case SO_REUSEADDR:
   1408          1.15   mycroft 		case SO_REUSEPORT:
   1409           1.1       cgd 		case SO_OOBINLINE:
   1410          1.26   thorpej 		case SO_TIMESTAMP:
   1411          1.36     perry 			if (m == NULL || m->m_len < sizeof(int)) {
   1412           1.1       cgd 				error = EINVAL;
   1413           1.1       cgd 				goto bad;
   1414           1.1       cgd 			}
   1415           1.1       cgd 			if (*mtod(m, int *))
   1416           1.1       cgd 				so->so_options |= optname;
   1417           1.1       cgd 			else
   1418           1.1       cgd 				so->so_options &= ~optname;
   1419           1.1       cgd 			break;
   1420           1.1       cgd 
   1421           1.1       cgd 		case SO_SNDBUF:
   1422           1.1       cgd 		case SO_RCVBUF:
   1423           1.1       cgd 		case SO_SNDLOWAT:
   1424           1.1       cgd 		case SO_RCVLOWAT:
   1425          1.28   thorpej 		    {
   1426          1.28   thorpej 			int optval;
   1427          1.28   thorpej 
   1428          1.36     perry 			if (m == NULL || m->m_len < sizeof(int)) {
   1429           1.1       cgd 				error = EINVAL;
   1430           1.1       cgd 				goto bad;
   1431           1.1       cgd 			}
   1432          1.28   thorpej 
   1433          1.28   thorpej 			/*
   1434          1.28   thorpej 			 * Values < 1 make no sense for any of these
   1435          1.28   thorpej 			 * options, so disallow them.
   1436          1.28   thorpej 			 */
   1437          1.28   thorpej 			optval = *mtod(m, int *);
   1438          1.28   thorpej 			if (optval < 1) {
   1439          1.28   thorpej 				error = EINVAL;
   1440          1.28   thorpej 				goto bad;
   1441          1.28   thorpej 			}
   1442          1.28   thorpej 
   1443           1.1       cgd 			switch (optname) {
   1444           1.1       cgd 
   1445           1.1       cgd 			case SO_SNDBUF:
   1446           1.1       cgd 			case SO_RCVBUF:
   1447           1.1       cgd 				if (sbreserve(optname == SO_SNDBUF ?
   1448           1.1       cgd 				    &so->so_snd : &so->so_rcv,
   1449          1.28   thorpej 				    (u_long) optval) == 0) {
   1450           1.1       cgd 					error = ENOBUFS;
   1451           1.1       cgd 					goto bad;
   1452           1.1       cgd 				}
   1453           1.1       cgd 				break;
   1454           1.1       cgd 
   1455          1.28   thorpej 			/*
   1456          1.28   thorpej 			 * Make sure the low-water is never greater than
   1457          1.28   thorpej 			 * the high-water.
   1458          1.28   thorpej 			 */
   1459           1.1       cgd 			case SO_SNDLOWAT:
   1460          1.28   thorpej 				so->so_snd.sb_lowat =
   1461          1.28   thorpej 				    (optval > so->so_snd.sb_hiwat) ?
   1462          1.28   thorpej 				    so->so_snd.sb_hiwat : optval;
   1463           1.1       cgd 				break;
   1464           1.1       cgd 			case SO_RCVLOWAT:
   1465          1.28   thorpej 				so->so_rcv.sb_lowat =
   1466          1.28   thorpej 				    (optval > so->so_rcv.sb_hiwat) ?
   1467          1.28   thorpej 				    so->so_rcv.sb_hiwat : optval;
   1468           1.1       cgd 				break;
   1469           1.1       cgd 			}
   1470           1.1       cgd 			break;
   1471          1.28   thorpej 		    }
   1472           1.1       cgd 
   1473           1.1       cgd 		case SO_SNDTIMEO:
   1474           1.1       cgd 		case SO_RCVTIMEO:
   1475           1.1       cgd 		    {
   1476           1.1       cgd 			struct timeval *tv;
   1477           1.1       cgd 			short val;
   1478           1.1       cgd 
   1479          1.36     perry 			if (m == NULL || m->m_len < sizeof(*tv)) {
   1480           1.1       cgd 				error = EINVAL;
   1481           1.1       cgd 				goto bad;
   1482           1.1       cgd 			}
   1483           1.1       cgd 			tv = mtod(m, struct timeval *);
   1484          1.75    itojun 			if (tv->tv_sec > (SHRT_MAX - tv->tv_usec / tick) / hz) {
   1485           1.1       cgd 				error = EDOM;
   1486           1.1       cgd 				goto bad;
   1487           1.1       cgd 			}
   1488           1.1       cgd 			val = tv->tv_sec * hz + tv->tv_usec / tick;
   1489          1.74    itojun 			if (val == 0 && tv->tv_usec != 0)
   1490          1.74    itojun 				val = 1;
   1491           1.1       cgd 
   1492           1.1       cgd 			switch (optname) {
   1493           1.1       cgd 
   1494           1.1       cgd 			case SO_SNDTIMEO:
   1495           1.1       cgd 				so->so_snd.sb_timeo = val;
   1496           1.1       cgd 				break;
   1497           1.1       cgd 			case SO_RCVTIMEO:
   1498           1.1       cgd 				so->so_rcv.sb_timeo = val;
   1499           1.1       cgd 				break;
   1500           1.1       cgd 			}
   1501           1.1       cgd 			break;
   1502           1.1       cgd 		    }
   1503           1.1       cgd 
   1504           1.1       cgd 		default:
   1505           1.1       cgd 			error = ENOPROTOOPT;
   1506           1.1       cgd 			break;
   1507           1.1       cgd 		}
   1508          1.15   mycroft 		if (error == 0 && so->so_proto && so->so_proto->pr_ctloutput) {
   1509          1.15   mycroft 			(void) ((*so->so_proto->pr_ctloutput)
   1510          1.15   mycroft 				  (PRCO_SETOPT, so, level, optname, &m0));
   1511          1.15   mycroft 			m = NULL;	/* freed by protocol */
   1512          1.15   mycroft 		}
   1513           1.1       cgd 	}
   1514          1.54     lukem  bad:
   1515           1.1       cgd 	if (m)
   1516           1.1       cgd 		(void) m_free(m);
   1517           1.1       cgd 	return (error);
   1518           1.1       cgd }
   1519           1.1       cgd 
   1520          1.14   mycroft int
   1521          1.54     lukem sogetopt(struct socket *so, int level, int optname, struct mbuf **mp)
   1522           1.1       cgd {
   1523          1.54     lukem 	struct mbuf	*m;
   1524           1.1       cgd 
   1525           1.1       cgd 	if (level != SOL_SOCKET) {
   1526           1.1       cgd 		if (so->so_proto && so->so_proto->pr_ctloutput) {
   1527           1.1       cgd 			return ((*so->so_proto->pr_ctloutput)
   1528           1.1       cgd 				  (PRCO_GETOPT, so, level, optname, mp));
   1529           1.1       cgd 		} else
   1530           1.1       cgd 			return (ENOPROTOOPT);
   1531           1.1       cgd 	} else {
   1532           1.1       cgd 		m = m_get(M_WAIT, MT_SOOPTS);
   1533          1.36     perry 		m->m_len = sizeof(int);
   1534           1.1       cgd 
   1535           1.1       cgd 		switch (optname) {
   1536           1.1       cgd 
   1537           1.1       cgd 		case SO_LINGER:
   1538          1.36     perry 			m->m_len = sizeof(struct linger);
   1539           1.1       cgd 			mtod(m, struct linger *)->l_onoff =
   1540           1.1       cgd 				so->so_options & SO_LINGER;
   1541           1.1       cgd 			mtod(m, struct linger *)->l_linger = so->so_linger;
   1542           1.1       cgd 			break;
   1543           1.1       cgd 
   1544           1.1       cgd 		case SO_USELOOPBACK:
   1545           1.1       cgd 		case SO_DONTROUTE:
   1546           1.1       cgd 		case SO_DEBUG:
   1547           1.1       cgd 		case SO_KEEPALIVE:
   1548           1.1       cgd 		case SO_REUSEADDR:
   1549          1.15   mycroft 		case SO_REUSEPORT:
   1550           1.1       cgd 		case SO_BROADCAST:
   1551           1.1       cgd 		case SO_OOBINLINE:
   1552          1.26   thorpej 		case SO_TIMESTAMP:
   1553           1.1       cgd 			*mtod(m, int *) = so->so_options & optname;
   1554           1.1       cgd 			break;
   1555           1.1       cgd 
   1556           1.1       cgd 		case SO_TYPE:
   1557           1.1       cgd 			*mtod(m, int *) = so->so_type;
   1558           1.1       cgd 			break;
   1559           1.1       cgd 
   1560           1.1       cgd 		case SO_ERROR:
   1561           1.1       cgd 			*mtod(m, int *) = so->so_error;
   1562           1.1       cgd 			so->so_error = 0;
   1563           1.1       cgd 			break;
   1564           1.1       cgd 
   1565           1.1       cgd 		case SO_SNDBUF:
   1566           1.1       cgd 			*mtod(m, int *) = so->so_snd.sb_hiwat;
   1567           1.1       cgd 			break;
   1568           1.1       cgd 
   1569           1.1       cgd 		case SO_RCVBUF:
   1570           1.1       cgd 			*mtod(m, int *) = so->so_rcv.sb_hiwat;
   1571           1.1       cgd 			break;
   1572           1.1       cgd 
   1573           1.1       cgd 		case SO_SNDLOWAT:
   1574           1.1       cgd 			*mtod(m, int *) = so->so_snd.sb_lowat;
   1575           1.1       cgd 			break;
   1576           1.1       cgd 
   1577           1.1       cgd 		case SO_RCVLOWAT:
   1578           1.1       cgd 			*mtod(m, int *) = so->so_rcv.sb_lowat;
   1579           1.1       cgd 			break;
   1580           1.1       cgd 
   1581           1.1       cgd 		case SO_SNDTIMEO:
   1582           1.1       cgd 		case SO_RCVTIMEO:
   1583           1.1       cgd 		    {
   1584           1.1       cgd 			int val = (optname == SO_SNDTIMEO ?
   1585           1.1       cgd 			     so->so_snd.sb_timeo : so->so_rcv.sb_timeo);
   1586           1.1       cgd 
   1587           1.1       cgd 			m->m_len = sizeof(struct timeval);
   1588           1.1       cgd 			mtod(m, struct timeval *)->tv_sec = val / hz;
   1589           1.1       cgd 			mtod(m, struct timeval *)->tv_usec =
   1590          1.27    kleink 			    (val % hz) * tick;
   1591           1.1       cgd 			break;
   1592           1.1       cgd 		    }
   1593           1.1       cgd 
   1594           1.1       cgd 		default:
   1595           1.1       cgd 			(void)m_free(m);
   1596           1.1       cgd 			return (ENOPROTOOPT);
   1597           1.1       cgd 		}
   1598           1.1       cgd 		*mp = m;
   1599           1.1       cgd 		return (0);
   1600           1.1       cgd 	}
   1601           1.1       cgd }
   1602           1.1       cgd 
   1603          1.14   mycroft void
   1604          1.54     lukem sohasoutofband(struct socket *so)
   1605           1.1       cgd {
   1606          1.90  christos 	fownsignal(so->so_pgid, SIGURG, POLL_PRI, POLLPRI|POLLRDBAND, so);
   1607           1.2       cgd 	selwakeup(&so->so_rcv.sb_sel);
   1608           1.1       cgd }
   1609          1.72  jdolecek 
   1610          1.72  jdolecek static void
   1611          1.72  jdolecek filt_sordetach(struct knote *kn)
   1612          1.72  jdolecek {
   1613          1.72  jdolecek 	struct socket	*so;
   1614          1.72  jdolecek 
   1615          1.72  jdolecek 	so = (struct socket *)kn->kn_fp->f_data;
   1616          1.73  christos 	SLIST_REMOVE(&so->so_rcv.sb_sel.sel_klist, kn, knote, kn_selnext);
   1617          1.73  christos 	if (SLIST_EMPTY(&so->so_rcv.sb_sel.sel_klist))
   1618          1.72  jdolecek 		so->so_rcv.sb_flags &= ~SB_KNOTE;
   1619          1.72  jdolecek }
   1620          1.72  jdolecek 
   1621          1.72  jdolecek /*ARGSUSED*/
   1622          1.72  jdolecek static int
   1623          1.72  jdolecek filt_soread(struct knote *kn, long hint)
   1624          1.72  jdolecek {
   1625          1.72  jdolecek 	struct socket	*so;
   1626          1.72  jdolecek 
   1627          1.72  jdolecek 	so = (struct socket *)kn->kn_fp->f_data;
   1628          1.72  jdolecek 	kn->kn_data = so->so_rcv.sb_cc;
   1629          1.72  jdolecek 	if (so->so_state & SS_CANTRCVMORE) {
   1630          1.72  jdolecek 		kn->kn_flags |= EV_EOF;
   1631          1.72  jdolecek 		kn->kn_fflags = so->so_error;
   1632          1.72  jdolecek 		return (1);
   1633          1.72  jdolecek 	}
   1634          1.72  jdolecek 	if (so->so_error)	/* temporary udp error */
   1635          1.72  jdolecek 		return (1);
   1636          1.72  jdolecek 	if (kn->kn_sfflags & NOTE_LOWAT)
   1637          1.72  jdolecek 		return (kn->kn_data >= kn->kn_sdata);
   1638          1.72  jdolecek 	return (kn->kn_data >= so->so_rcv.sb_lowat);
   1639          1.72  jdolecek }
   1640          1.72  jdolecek 
   1641          1.72  jdolecek static void
   1642          1.72  jdolecek filt_sowdetach(struct knote *kn)
   1643          1.72  jdolecek {
   1644          1.72  jdolecek 	struct socket	*so;
   1645          1.72  jdolecek 
   1646          1.72  jdolecek 	so = (struct socket *)kn->kn_fp->f_data;
   1647          1.73  christos 	SLIST_REMOVE(&so->so_snd.sb_sel.sel_klist, kn, knote, kn_selnext);
   1648          1.73  christos 	if (SLIST_EMPTY(&so->so_snd.sb_sel.sel_klist))
   1649          1.72  jdolecek 		so->so_snd.sb_flags &= ~SB_KNOTE;
   1650          1.72  jdolecek }
   1651          1.72  jdolecek 
   1652          1.72  jdolecek /*ARGSUSED*/
   1653          1.72  jdolecek static int
   1654          1.72  jdolecek filt_sowrite(struct knote *kn, long hint)
   1655          1.72  jdolecek {
   1656          1.72  jdolecek 	struct socket	*so;
   1657          1.72  jdolecek 
   1658          1.72  jdolecek 	so = (struct socket *)kn->kn_fp->f_data;
   1659          1.72  jdolecek 	kn->kn_data = sbspace(&so->so_snd);
   1660          1.72  jdolecek 	if (so->so_state & SS_CANTSENDMORE) {
   1661          1.72  jdolecek 		kn->kn_flags |= EV_EOF;
   1662          1.72  jdolecek 		kn->kn_fflags = so->so_error;
   1663          1.72  jdolecek 		return (1);
   1664          1.72  jdolecek 	}
   1665          1.72  jdolecek 	if (so->so_error)	/* temporary udp error */
   1666          1.72  jdolecek 		return (1);
   1667          1.72  jdolecek 	if (((so->so_state & SS_ISCONNECTED) == 0) &&
   1668          1.72  jdolecek 	    (so->so_proto->pr_flags & PR_CONNREQUIRED))
   1669          1.72  jdolecek 		return (0);
   1670          1.72  jdolecek 	if (kn->kn_sfflags & NOTE_LOWAT)
   1671          1.72  jdolecek 		return (kn->kn_data >= kn->kn_sdata);
   1672          1.72  jdolecek 	return (kn->kn_data >= so->so_snd.sb_lowat);
   1673          1.72  jdolecek }
   1674          1.72  jdolecek 
   1675          1.72  jdolecek /*ARGSUSED*/
   1676          1.72  jdolecek static int
   1677          1.72  jdolecek filt_solisten(struct knote *kn, long hint)
   1678          1.72  jdolecek {
   1679          1.72  jdolecek 	struct socket	*so;
   1680          1.72  jdolecek 
   1681          1.72  jdolecek 	so = (struct socket *)kn->kn_fp->f_data;
   1682          1.72  jdolecek 
   1683          1.72  jdolecek 	/*
   1684          1.72  jdolecek 	 * Set kn_data to number of incoming connections, not
   1685          1.72  jdolecek 	 * counting partial (incomplete) connections.
   1686          1.72  jdolecek 	 */
   1687          1.72  jdolecek 	kn->kn_data = so->so_qlen;
   1688          1.72  jdolecek 	return (kn->kn_data > 0);
   1689          1.72  jdolecek }
   1690          1.72  jdolecek 
   1691          1.72  jdolecek static const struct filterops solisten_filtops =
   1692          1.72  jdolecek 	{ 1, NULL, filt_sordetach, filt_solisten };
   1693          1.72  jdolecek static const struct filterops soread_filtops =
   1694          1.72  jdolecek 	{ 1, NULL, filt_sordetach, filt_soread };
   1695          1.72  jdolecek static const struct filterops sowrite_filtops =
   1696          1.72  jdolecek 	{ 1, NULL, filt_sowdetach, filt_sowrite };
   1697          1.72  jdolecek 
   1698          1.72  jdolecek int
   1699          1.72  jdolecek soo_kqfilter(struct file *fp, struct knote *kn)
   1700          1.72  jdolecek {
   1701          1.72  jdolecek 	struct socket	*so;
   1702          1.72  jdolecek 	struct sockbuf	*sb;
   1703          1.72  jdolecek 
   1704          1.72  jdolecek 	so = (struct socket *)kn->kn_fp->f_data;
   1705          1.72  jdolecek 	switch (kn->kn_filter) {
   1706          1.72  jdolecek 	case EVFILT_READ:
   1707          1.72  jdolecek 		if (so->so_options & SO_ACCEPTCONN)
   1708          1.72  jdolecek 			kn->kn_fop = &solisten_filtops;
   1709          1.72  jdolecek 		else
   1710          1.72  jdolecek 			kn->kn_fop = &soread_filtops;
   1711          1.72  jdolecek 		sb = &so->so_rcv;
   1712          1.72  jdolecek 		break;
   1713          1.72  jdolecek 	case EVFILT_WRITE:
   1714          1.72  jdolecek 		kn->kn_fop = &sowrite_filtops;
   1715          1.72  jdolecek 		sb = &so->so_snd;
   1716          1.72  jdolecek 		break;
   1717          1.72  jdolecek 	default:
   1718          1.72  jdolecek 		return (1);
   1719          1.72  jdolecek 	}
   1720          1.73  christos 	SLIST_INSERT_HEAD(&sb->sb_sel.sel_klist, kn, kn_selnext);
   1721          1.72  jdolecek 	sb->sb_flags |= SB_KNOTE;
   1722          1.72  jdolecek 	return (0);
   1723          1.72  jdolecek }
   1724          1.72  jdolecek 
   1725          1.94      yamt #include <sys/sysctl.h>
   1726          1.94      yamt 
   1727          1.94      yamt static int sysctl_kern_somaxkva(SYSCTLFN_PROTO);
   1728          1.94      yamt 
   1729          1.94      yamt /*
   1730          1.94      yamt  * sysctl helper routine for kern.somaxkva.  ensures that the given
   1731          1.94      yamt  * value is not too small.
   1732          1.94      yamt  * (XXX should we maybe make sure it's not too large as well?)
   1733          1.94      yamt  */
   1734          1.94      yamt static int
   1735          1.94      yamt sysctl_kern_somaxkva(SYSCTLFN_ARGS)
   1736          1.94      yamt {
   1737          1.94      yamt 	int error, new_somaxkva;
   1738          1.94      yamt 	struct sysctlnode node;
   1739          1.94      yamt 	int s;
   1740          1.94      yamt 
   1741          1.94      yamt 	new_somaxkva = somaxkva;
   1742          1.94      yamt 	node = *rnode;
   1743          1.94      yamt 	node.sysctl_data = &new_somaxkva;
   1744          1.94      yamt 	error = sysctl_lookup(SYSCTLFN_CALL(&node));
   1745          1.94      yamt 	if (error || newp == NULL)
   1746          1.94      yamt 		return (error);
   1747          1.94      yamt 
   1748          1.94      yamt 	if (new_somaxkva < (16 * 1024 * 1024)) /* sanity */
   1749          1.94      yamt 		return (EINVAL);
   1750          1.94      yamt 
   1751          1.94      yamt 	s = splvm();
   1752          1.94      yamt 	simple_lock(&so_pendfree_slock);
   1753          1.94      yamt 	somaxkva = new_somaxkva;
   1754          1.94      yamt 	wakeup(&socurkva);
   1755          1.94      yamt 	simple_unlock(&so_pendfree_slock);
   1756          1.94      yamt 	splx(s);
   1757          1.94      yamt 
   1758          1.94      yamt 	return (error);
   1759          1.94      yamt }
   1760          1.94      yamt 
   1761          1.94      yamt SYSCTL_SETUP(sysctl_kern_somaxkva_setup, "sysctl kern.somaxkva setup")
   1762          1.94      yamt {
   1763          1.94      yamt 
   1764          1.97    atatat 	sysctl_createv(clog, 0, NULL, NULL,
   1765          1.97    atatat 		       CTLFLAG_PERMANENT,
   1766          1.97    atatat 		       CTLTYPE_NODE, "kern", NULL,
   1767          1.97    atatat 		       NULL, 0, NULL, 0,
   1768          1.97    atatat 		       CTL_KERN, CTL_EOL);
   1769          1.97    atatat 
   1770          1.97    atatat 	sysctl_createv(clog, 0, NULL, NULL,
   1771          1.97    atatat 		       CTLFLAG_PERMANENT|CTLFLAG_READWRITE,
   1772      1.97.2.1        he 		       CTLTYPE_INT, "somaxkva",
   1773      1.97.2.1        he 		       SYSCTL_DESCR("Maximum amount of kernel memory to be "
   1774      1.97.2.1        he 				    "used for socket buffers"),
   1775          1.94      yamt 		       sysctl_kern_somaxkva, 0, NULL, 0,
   1776          1.94      yamt 		       CTL_KERN, KERN_SOMAXKVA, CTL_EOL);
   1777          1.94      yamt }
   1778