Home | History | Annotate | Line # | Download | only in netbsd32
netbsd32_select.c revision 1.1
      1  1.1  mrg /*	$NetBSD: netbsd32_select.c,v 1.1 2001/02/08 13:19:34 mrg 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.1  mrg 
     31  1.1  mrg #include <sys/param.h>
     32  1.1  mrg #include <sys/systm.h>
     33  1.1  mrg #include <sys/malloc.h>
     34  1.1  mrg #include <sys/mount.h>
     35  1.1  mrg #include <sys/time.h>
     36  1.1  mrg #include <sys/vnode.h>
     37  1.1  mrg #include <sys/file.h>
     38  1.1  mrg #include <sys/filedesc.h>
     39  1.1  mrg 
     40  1.1  mrg #include <sys/proc.h>
     41  1.1  mrg 
     42  1.1  mrg #include <net/if.h>
     43  1.1  mrg 
     44  1.1  mrg #include <compat/netbsd32/netbsd32.h>
     45  1.1  mrg #include <compat/netbsd32/netbsd32_syscall.h>
     46  1.1  mrg #include <compat/netbsd32/netbsd32_syscallargs.h>
     47  1.1  mrg #include <compat/netbsd32/netbsd32_conv.h>
     48  1.1  mrg 
     49  1.1  mrg int
     50  1.1  mrg netbsd32_select(p, v, retval)
     51  1.1  mrg 	struct proc *p;
     52  1.1  mrg 	void *v;
     53  1.1  mrg 	register_t *retval;
     54  1.1  mrg {
     55  1.1  mrg 	struct netbsd32_select_args /* {
     56  1.1  mrg 		syscallarg(int) nd;
     57  1.1  mrg 		syscallarg(netbsd32_fd_setp_t) in;
     58  1.1  mrg 		syscallarg(netbsd32_fd_setp_t) ou;
     59  1.1  mrg 		syscallarg(netbsd32_fd_setp_t) ex;
     60  1.1  mrg 		syscallarg(netbsd32_timevalp_t) tv;
     61  1.1  mrg 	} */ *uap = v;
     62  1.1  mrg /* This one must be done in-line 'cause of the timeval */
     63  1.1  mrg 	struct netbsd32_timeval tv32;
     64  1.1  mrg 	caddr_t bits;
     65  1.1  mrg 	char smallbits[howmany(FD_SETSIZE, NFDBITS) * sizeof(fd_mask) * 6];
     66  1.1  mrg 	struct timeval atv;
     67  1.1  mrg 	int s, ncoll, error = 0, timo;
     68  1.1  mrg 	size_t ni;
     69  1.1  mrg 	extern int	selwait, nselcoll;
     70  1.1  mrg 	extern int selscan __P((struct proc *, fd_mask *, fd_mask *, int, register_t *));
     71  1.1  mrg 
     72  1.1  mrg 	if (SCARG(uap, nd) < 0)
     73  1.1  mrg 		return (EINVAL);
     74  1.1  mrg 	if (SCARG(uap, nd) > p->p_fd->fd_nfiles) {
     75  1.1  mrg 		/* forgiving; slightly wrong */
     76  1.1  mrg 		SCARG(uap, nd) = p->p_fd->fd_nfiles;
     77  1.1  mrg 	}
     78  1.1  mrg 	ni = howmany(SCARG(uap, nd), NFDBITS) * sizeof(fd_mask);
     79  1.1  mrg 	if (ni * 6 > sizeof(smallbits))
     80  1.1  mrg 		bits = malloc(ni * 6, M_TEMP, M_WAITOK);
     81  1.1  mrg 	else
     82  1.1  mrg 		bits = smallbits;
     83  1.1  mrg 
     84  1.1  mrg #define	getbits(name, x) \
     85  1.1  mrg 	if (SCARG(uap, name)) { \
     86  1.1  mrg 		error = copyin((caddr_t)(u_long)SCARG(uap, name), bits + ni * x, ni); \
     87  1.1  mrg 		if (error) \
     88  1.1  mrg 			goto done; \
     89  1.1  mrg 	} else \
     90  1.1  mrg 		memset(bits + ni * x, 0, ni);
     91  1.1  mrg 	getbits(in, 0);
     92  1.1  mrg 	getbits(ou, 1);
     93  1.1  mrg 	getbits(ex, 2);
     94  1.1  mrg #undef	getbits
     95  1.1  mrg 
     96  1.1  mrg 	if (SCARG(uap, tv)) {
     97  1.1  mrg 		error = copyin((caddr_t)(u_long)SCARG(uap, tv), (caddr_t)&tv32,
     98  1.1  mrg 			sizeof(tv32));
     99  1.1  mrg 		if (error)
    100  1.1  mrg 			goto done;
    101  1.1  mrg 		netbsd32_to_timeval(&tv32, &atv);
    102  1.1  mrg 		if (itimerfix(&atv)) {
    103  1.1  mrg 			error = EINVAL;
    104  1.1  mrg 			goto done;
    105  1.1  mrg 		}
    106  1.1  mrg 		s = splclock();
    107  1.1  mrg 		timeradd(&atv, &time, &atv);
    108  1.1  mrg 		splx(s);
    109  1.1  mrg 	} else
    110  1.1  mrg 		timo = 0;
    111  1.1  mrg retry:
    112  1.1  mrg 	ncoll = nselcoll;
    113  1.1  mrg 	p->p_flag |= P_SELECT;
    114  1.1  mrg 	error = selscan(p, (fd_mask *)(bits + ni * 0),
    115  1.1  mrg 			   (fd_mask *)(bits + ni * 3), SCARG(uap, nd), retval);
    116  1.1  mrg 	if (error || *retval)
    117  1.1  mrg 		goto done;
    118  1.1  mrg 	if (SCARG(uap, tv)) {
    119  1.1  mrg 		/*
    120  1.1  mrg 		 * We have to recalculate the timeout on every retry.
    121  1.1  mrg 		 */
    122  1.1  mrg 		timo = hzto(&atv);
    123  1.1  mrg 		if (timo <= 0)
    124  1.1  mrg 			goto done;
    125  1.1  mrg 	}
    126  1.1  mrg 	s = splhigh();
    127  1.1  mrg 	if ((p->p_flag & P_SELECT) == 0 || nselcoll != ncoll) {
    128  1.1  mrg 		splx(s);
    129  1.1  mrg 		goto retry;
    130  1.1  mrg 	}
    131  1.1  mrg 	p->p_flag &= ~P_SELECT;
    132  1.1  mrg 	error = tsleep((caddr_t)&selwait, PSOCK | PCATCH, "select", timo);
    133  1.1  mrg 	splx(s);
    134  1.1  mrg 	if (error == 0)
    135  1.1  mrg 		goto retry;
    136  1.1  mrg done:
    137  1.1  mrg 	p->p_flag &= ~P_SELECT;
    138  1.1  mrg 	/* select is not restarted after signals... */
    139  1.1  mrg 	if (error == ERESTART)
    140  1.1  mrg 		error = EINTR;
    141  1.1  mrg 	if (error == EWOULDBLOCK)
    142  1.1  mrg 		error = 0;
    143  1.1  mrg 	if (error == 0) {
    144  1.1  mrg #define	putbits(name, x) \
    145  1.1  mrg 		if (SCARG(uap, name)) { \
    146  1.1  mrg 			error = copyout(bits + ni * x, (caddr_t)(u_long)SCARG(uap, name), ni); \
    147  1.1  mrg 			if (error) \
    148  1.1  mrg 				goto out; \
    149  1.1  mrg 		}
    150  1.1  mrg 		putbits(in, 3);
    151  1.1  mrg 		putbits(ou, 4);
    152  1.1  mrg 		putbits(ex, 5);
    153  1.1  mrg #undef putbits
    154  1.1  mrg 	}
    155  1.1  mrg out:
    156  1.1  mrg 	if (ni * 6 > sizeof(smallbits))
    157  1.1  mrg 		free(bits, M_TEMP);
    158  1.1  mrg 	return (error);
    159  1.1  mrg }
    160