netbsd32_select.c revision 1.2 1 1.2 lukem /* $NetBSD: netbsd32_select.c,v 1.2 2001/11/13 02:09:07 lukem 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.2 lukem __KERNEL_RCSID(0, "$NetBSD: netbsd32_select.c,v 1.2 2001/11/13 02:09:07 lukem 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.1 mrg netbsd32_select(p, v, retval)
54 1.1 mrg struct proc *p;
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.1 mrg struct netbsd32_timeval tv32;
67 1.1 mrg caddr_t bits;
68 1.1 mrg char smallbits[howmany(FD_SETSIZE, NFDBITS) * sizeof(fd_mask) * 6];
69 1.1 mrg struct timeval atv;
70 1.1 mrg int s, ncoll, error = 0, timo;
71 1.1 mrg size_t ni;
72 1.1 mrg extern int selwait, nselcoll;
73 1.1 mrg extern int selscan __P((struct proc *, fd_mask *, fd_mask *, int, register_t *));
74 1.1 mrg
75 1.1 mrg if (SCARG(uap, nd) < 0)
76 1.1 mrg return (EINVAL);
77 1.1 mrg if (SCARG(uap, nd) > p->p_fd->fd_nfiles) {
78 1.1 mrg /* forgiving; slightly wrong */
79 1.1 mrg SCARG(uap, nd) = p->p_fd->fd_nfiles;
80 1.1 mrg }
81 1.1 mrg ni = howmany(SCARG(uap, nd), NFDBITS) * sizeof(fd_mask);
82 1.1 mrg if (ni * 6 > sizeof(smallbits))
83 1.1 mrg bits = malloc(ni * 6, M_TEMP, M_WAITOK);
84 1.1 mrg else
85 1.1 mrg bits = smallbits;
86 1.1 mrg
87 1.1 mrg #define getbits(name, x) \
88 1.1 mrg if (SCARG(uap, name)) { \
89 1.1 mrg error = copyin((caddr_t)(u_long)SCARG(uap, name), bits + ni * x, ni); \
90 1.1 mrg if (error) \
91 1.1 mrg goto done; \
92 1.1 mrg } else \
93 1.1 mrg memset(bits + ni * x, 0, ni);
94 1.1 mrg getbits(in, 0);
95 1.1 mrg getbits(ou, 1);
96 1.1 mrg getbits(ex, 2);
97 1.1 mrg #undef getbits
98 1.1 mrg
99 1.1 mrg if (SCARG(uap, tv)) {
100 1.1 mrg error = copyin((caddr_t)(u_long)SCARG(uap, tv), (caddr_t)&tv32,
101 1.1 mrg sizeof(tv32));
102 1.1 mrg if (error)
103 1.1 mrg goto done;
104 1.1 mrg netbsd32_to_timeval(&tv32, &atv);
105 1.1 mrg if (itimerfix(&atv)) {
106 1.1 mrg error = EINVAL;
107 1.1 mrg goto done;
108 1.1 mrg }
109 1.1 mrg s = splclock();
110 1.1 mrg timeradd(&atv, &time, &atv);
111 1.1 mrg splx(s);
112 1.1 mrg } else
113 1.1 mrg timo = 0;
114 1.1 mrg retry:
115 1.1 mrg ncoll = nselcoll;
116 1.1 mrg p->p_flag |= P_SELECT;
117 1.1 mrg error = selscan(p, (fd_mask *)(bits + ni * 0),
118 1.1 mrg (fd_mask *)(bits + ni * 3), SCARG(uap, nd), retval);
119 1.1 mrg if (error || *retval)
120 1.1 mrg goto done;
121 1.1 mrg if (SCARG(uap, tv)) {
122 1.1 mrg /*
123 1.1 mrg * We have to recalculate the timeout on every retry.
124 1.1 mrg */
125 1.1 mrg timo = hzto(&atv);
126 1.1 mrg if (timo <= 0)
127 1.1 mrg goto done;
128 1.1 mrg }
129 1.1 mrg s = splhigh();
130 1.1 mrg if ((p->p_flag & P_SELECT) == 0 || nselcoll != ncoll) {
131 1.1 mrg splx(s);
132 1.1 mrg goto retry;
133 1.1 mrg }
134 1.1 mrg p->p_flag &= ~P_SELECT;
135 1.1 mrg error = tsleep((caddr_t)&selwait, PSOCK | PCATCH, "select", timo);
136 1.1 mrg splx(s);
137 1.1 mrg if (error == 0)
138 1.1 mrg goto retry;
139 1.1 mrg done:
140 1.1 mrg p->p_flag &= ~P_SELECT;
141 1.1 mrg /* select is not restarted after signals... */
142 1.1 mrg if (error == ERESTART)
143 1.1 mrg error = EINTR;
144 1.1 mrg if (error == EWOULDBLOCK)
145 1.1 mrg error = 0;
146 1.1 mrg if (error == 0) {
147 1.1 mrg #define putbits(name, x) \
148 1.1 mrg if (SCARG(uap, name)) { \
149 1.1 mrg error = copyout(bits + ni * x, (caddr_t)(u_long)SCARG(uap, name), ni); \
150 1.1 mrg if (error) \
151 1.1 mrg goto out; \
152 1.1 mrg }
153 1.1 mrg putbits(in, 3);
154 1.1 mrg putbits(ou, 4);
155 1.1 mrg putbits(ex, 5);
156 1.1 mrg #undef putbits
157 1.1 mrg }
158 1.1 mrg out:
159 1.1 mrg if (ni * 6 > sizeof(smallbits))
160 1.1 mrg free(bits, M_TEMP);
161 1.1 mrg return (error);
162 1.1 mrg }
163