Home | History | Annotate | Line # | Download | only in netbsd32
netbsd32_select.c revision 1.5
      1  1.5  christos /*	$NetBSD: netbsd32_select.c,v 1.5 2003/10/26 19:12:50 christos Exp $	*/
      2  1.1       mrg 
      3  1.1       mrg /*
      4  1.1       mrg  * Copyright (c) 1998, 2001 Matthew R. Green
      5  1.1       mrg  * All rights reserved.
      6  1.1       mrg  *
      7  1.1       mrg  * Redistribution and use in source and binary forms, with or without
      8  1.1       mrg  * modification, are permitted provided that the following conditions
      9  1.1       mrg  * are met:
     10  1.1       mrg  * 1. Redistributions of source code must retain the above copyright
     11  1.1       mrg  *    notice, this list of conditions and the following disclaimer.
     12  1.1       mrg  * 2. Redistributions in binary form must reproduce the above copyright
     13  1.1       mrg  *    notice, this list of conditions and the following disclaimer in the
     14  1.1       mrg  *    documentation and/or other materials provided with the distribution.
     15  1.1       mrg  * 3. The name of the author may not be used to endorse or promote products
     16  1.1       mrg  *    derived from this software without specific prior written permission.
     17  1.1       mrg  *
     18  1.1       mrg  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
     19  1.1       mrg  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
     20  1.1       mrg  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
     21  1.1       mrg  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
     22  1.1       mrg  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
     23  1.1       mrg  * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
     24  1.1       mrg  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
     25  1.1       mrg  * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
     26  1.1       mrg  * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
     27  1.1       mrg  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
     28  1.1       mrg  * SUCH DAMAGE.
     29  1.1       mrg  */
     30  1.2     lukem 
     31  1.2     lukem #include <sys/cdefs.h>
     32  1.5  christos __KERNEL_RCSID(0, "$NetBSD: netbsd32_select.c,v 1.5 2003/10/26 19:12:50 christos Exp $");
     33  1.1       mrg 
     34  1.1       mrg #include <sys/param.h>
     35  1.1       mrg #include <sys/systm.h>
     36  1.1       mrg #include <sys/malloc.h>
     37  1.1       mrg #include <sys/mount.h>
     38  1.1       mrg #include <sys/time.h>
     39  1.1       mrg #include <sys/vnode.h>
     40  1.1       mrg #include <sys/file.h>
     41  1.1       mrg #include <sys/filedesc.h>
     42  1.1       mrg 
     43  1.1       mrg #include <sys/proc.h>
     44  1.1       mrg 
     45  1.1       mrg #include <net/if.h>
     46  1.1       mrg 
     47  1.1       mrg #include <compat/netbsd32/netbsd32.h>
     48  1.1       mrg #include <compat/netbsd32/netbsd32_syscall.h>
     49  1.1       mrg #include <compat/netbsd32/netbsd32_syscallargs.h>
     50  1.1       mrg #include <compat/netbsd32/netbsd32_conv.h>
     51  1.1       mrg 
     52  1.1       mrg int
     53  1.4   thorpej netbsd32_select(l, v, retval)
     54  1.4   thorpej 	struct lwp *l;
     55  1.1       mrg 	void *v;
     56  1.1       mrg 	register_t *retval;
     57  1.1       mrg {
     58  1.1       mrg 	struct netbsd32_select_args /* {
     59  1.1       mrg 		syscallarg(int) nd;
     60  1.1       mrg 		syscallarg(netbsd32_fd_setp_t) in;
     61  1.1       mrg 		syscallarg(netbsd32_fd_setp_t) ou;
     62  1.1       mrg 		syscallarg(netbsd32_fd_setp_t) ex;
     63  1.1       mrg 		syscallarg(netbsd32_timevalp_t) tv;
     64  1.1       mrg 	} */ *uap = v;
     65  1.1       mrg /* This one must be done in-line 'cause of the timeval */
     66  1.4   thorpej 	struct proc *p = l->l_proc;
     67  1.1       mrg 	struct netbsd32_timeval tv32;
     68  1.1       mrg 	caddr_t bits;
     69  1.1       mrg 	char smallbits[howmany(FD_SETSIZE, NFDBITS) * sizeof(fd_mask) * 6];
     70  1.1       mrg 	struct timeval atv;
     71  1.1       mrg 	int s, ncoll, error = 0, timo;
     72  1.1       mrg 	size_t ni;
     73  1.1       mrg 	extern int	selwait, nselcoll;
     74  1.1       mrg 	extern int selscan __P((struct proc *, fd_mask *, fd_mask *, int, register_t *));
     75  1.1       mrg 
     76  1.1       mrg 	if (SCARG(uap, nd) < 0)
     77  1.1       mrg 		return (EINVAL);
     78  1.1       mrg 	if (SCARG(uap, nd) > p->p_fd->fd_nfiles) {
     79  1.1       mrg 		/* forgiving; slightly wrong */
     80  1.1       mrg 		SCARG(uap, nd) = p->p_fd->fd_nfiles;
     81  1.1       mrg 	}
     82  1.1       mrg 	ni = howmany(SCARG(uap, nd), NFDBITS) * sizeof(fd_mask);
     83  1.1       mrg 	if (ni * 6 > sizeof(smallbits))
     84  1.1       mrg 		bits = malloc(ni * 6, M_TEMP, M_WAITOK);
     85  1.1       mrg 	else
     86  1.1       mrg 		bits = smallbits;
     87  1.1       mrg 
     88  1.1       mrg #define	getbits(name, x) \
     89  1.1       mrg 	if (SCARG(uap, name)) { \
     90  1.3       scw 		error = copyin((caddr_t)NETBSD32PTR64(SCARG(uap, name)), \
     91  1.3       scw 		    bits + ni * x, ni); \
     92  1.1       mrg 		if (error) \
     93  1.1       mrg 			goto done; \
     94  1.1       mrg 	} else \
     95  1.1       mrg 		memset(bits + ni * x, 0, ni);
     96  1.1       mrg 	getbits(in, 0);
     97  1.1       mrg 	getbits(ou, 1);
     98  1.1       mrg 	getbits(ex, 2);
     99  1.1       mrg #undef	getbits
    100  1.1       mrg 
    101  1.1       mrg 	if (SCARG(uap, tv)) {
    102  1.3       scw 		error = copyin((caddr_t)NETBSD32PTR64(SCARG(uap, tv)),
    103  1.3       scw 		    (caddr_t)&tv32, sizeof(tv32));
    104  1.1       mrg 		if (error)
    105  1.1       mrg 			goto done;
    106  1.1       mrg 		netbsd32_to_timeval(&tv32, &atv);
    107  1.1       mrg 		if (itimerfix(&atv)) {
    108  1.1       mrg 			error = EINVAL;
    109  1.1       mrg 			goto done;
    110  1.1       mrg 		}
    111  1.1       mrg 		s = splclock();
    112  1.1       mrg 		timeradd(&atv, &time, &atv);
    113  1.1       mrg 		splx(s);
    114  1.5  christos 	}
    115  1.1       mrg retry:
    116  1.1       mrg 	ncoll = nselcoll;
    117  1.4   thorpej 	l->l_flag |= L_SELECT;
    118  1.1       mrg 	error = selscan(p, (fd_mask *)(bits + ni * 0),
    119  1.1       mrg 			   (fd_mask *)(bits + ni * 3), SCARG(uap, nd), retval);
    120  1.1       mrg 	if (error || *retval)
    121  1.1       mrg 		goto done;
    122  1.1       mrg 	if (SCARG(uap, tv)) {
    123  1.1       mrg 		/*
    124  1.1       mrg 		 * We have to recalculate the timeout on every retry.
    125  1.1       mrg 		 */
    126  1.1       mrg 		timo = hzto(&atv);
    127  1.5  christos 	} else
    128  1.5  christos 		timo = 0;
    129  1.5  christos 	if (timo <= 0)
    130  1.5  christos 		goto done;
    131  1.1       mrg 	s = splhigh();
    132  1.4   thorpej 	if ((l->l_flag & L_SELECT) == 0 || nselcoll != ncoll) {
    133  1.1       mrg 		splx(s);
    134  1.1       mrg 		goto retry;
    135  1.1       mrg 	}
    136  1.4   thorpej 	l->l_flag &= ~L_SELECT;
    137  1.1       mrg 	error = tsleep((caddr_t)&selwait, PSOCK | PCATCH, "select", timo);
    138  1.1       mrg 	splx(s);
    139  1.1       mrg 	if (error == 0)
    140  1.1       mrg 		goto retry;
    141  1.1       mrg done:
    142  1.4   thorpej 	l->l_flag &= ~L_SELECT;
    143  1.1       mrg 	/* select is not restarted after signals... */
    144  1.1       mrg 	if (error == ERESTART)
    145  1.1       mrg 		error = EINTR;
    146  1.1       mrg 	if (error == EWOULDBLOCK)
    147  1.1       mrg 		error = 0;
    148  1.1       mrg 	if (error == 0) {
    149  1.1       mrg #define	putbits(name, x) \
    150  1.1       mrg 		if (SCARG(uap, name)) { \
    151  1.3       scw 			error = copyout(bits + ni * x, \
    152  1.3       scw 			    (caddr_t)NETBSD32PTR64(SCARG(uap, name)), ni); \
    153  1.1       mrg 			if (error) \
    154  1.1       mrg 				goto out; \
    155  1.1       mrg 		}
    156  1.1       mrg 		putbits(in, 3);
    157  1.1       mrg 		putbits(ou, 4);
    158  1.1       mrg 		putbits(ex, 5);
    159  1.1       mrg #undef putbits
    160  1.1       mrg 	}
    161  1.1       mrg out:
    162  1.1       mrg 	if (ni * 6 > sizeof(smallbits))
    163  1.1       mrg 		free(bits, M_TEMP);
    164  1.1       mrg 	return (error);
    165  1.1       mrg }
    166