sys_generic.c revision 1.36 1 1.36 fvdl /* $NetBSD: sys_generic.c,v 1.36 1998/03/01 02:22:32 fvdl Exp $ */
2 1.15 cgd
3 1.15 cgd /*
4 1.15 cgd * Copyright (c) 1982, 1986, 1989, 1993
5 1.15 cgd * The Regents of the University of California. All rights reserved.
6 1.15 cgd * (c) UNIX System Laboratories, Inc.
7 1.15 cgd * All or some portions of this file are derived from material licensed
8 1.15 cgd * to the University of California by American Telephone and Telegraph
9 1.15 cgd * Co. or Unix System Laboratories, Inc. and are reproduced herein with
10 1.15 cgd * the permission of UNIX System Laboratories, Inc.
11 1.15 cgd *
12 1.15 cgd * Redistribution and use in source and binary forms, with or without
13 1.15 cgd * modification, are permitted provided that the following conditions
14 1.15 cgd * are met:
15 1.15 cgd * 1. Redistributions of source code must retain the above copyright
16 1.15 cgd * notice, this list of conditions and the following disclaimer.
17 1.15 cgd * 2. Redistributions in binary form must reproduce the above copyright
18 1.15 cgd * notice, this list of conditions and the following disclaimer in the
19 1.15 cgd * documentation and/or other materials provided with the distribution.
20 1.15 cgd * 3. All advertising materials mentioning features or use of this software
21 1.15 cgd * must display the following acknowledgement:
22 1.15 cgd * This product includes software developed by the University of
23 1.15 cgd * California, Berkeley and its contributors.
24 1.15 cgd * 4. Neither the name of the University nor the names of its contributors
25 1.15 cgd * may be used to endorse or promote products derived from this software
26 1.15 cgd * without specific prior written permission.
27 1.15 cgd *
28 1.15 cgd * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
29 1.15 cgd * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
30 1.15 cgd * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
31 1.15 cgd * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
32 1.15 cgd * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
33 1.15 cgd * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
34 1.15 cgd * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
35 1.15 cgd * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
36 1.15 cgd * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
37 1.15 cgd * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
38 1.15 cgd * SUCH DAMAGE.
39 1.15 cgd *
40 1.36 fvdl * @(#)sys_generic.c 8.9 (Berkeley) 2/14/95
41 1.15 cgd */
42 1.15 cgd
43 1.15 cgd #include <sys/param.h>
44 1.15 cgd #include <sys/systm.h>
45 1.15 cgd #include <sys/filedesc.h>
46 1.15 cgd #include <sys/ioctl.h>
47 1.15 cgd #include <sys/file.h>
48 1.15 cgd #include <sys/proc.h>
49 1.15 cgd #include <sys/socketvar.h>
50 1.22 christos #include <sys/signalvar.h>
51 1.15 cgd #include <sys/uio.h>
52 1.15 cgd #include <sys/kernel.h>
53 1.15 cgd #include <sys/stat.h>
54 1.15 cgd #include <sys/malloc.h>
55 1.28 mycroft #include <sys/poll.h>
56 1.15 cgd #ifdef KTRACE
57 1.15 cgd #include <sys/ktrace.h>
58 1.15 cgd #endif
59 1.15 cgd
60 1.16 cgd #include <sys/mount.h>
61 1.16 cgd #include <sys/syscallargs.h>
62 1.22 christos
63 1.25 mycroft int selscan __P((struct proc *, fd_mask *, fd_mask *, int, register_t *));
64 1.28 mycroft int pollscan __P((struct proc *, struct pollfd *, int, register_t *));
65 1.22 christos
66 1.15 cgd /*
67 1.15 cgd * Read system call.
68 1.15 cgd */
69 1.15 cgd /* ARGSUSED */
70 1.22 christos int
71 1.21 mycroft sys_read(p, v, retval)
72 1.15 cgd struct proc *p;
73 1.20 thorpej void *v;
74 1.20 thorpej register_t *retval;
75 1.20 thorpej {
76 1.21 mycroft register struct sys_read_args /* {
77 1.16 cgd syscallarg(int) fd;
78 1.31 cgd syscallarg(void *) buf;
79 1.31 cgd syscallarg(size_t) nbyte;
80 1.20 thorpej } */ *uap = v;
81 1.34 mycroft int fd = SCARG(uap, fd);
82 1.15 cgd register struct file *fp;
83 1.15 cgd register struct filedesc *fdp = p->p_fd;
84 1.15 cgd struct uio auio;
85 1.15 cgd struct iovec aiov;
86 1.15 cgd long cnt, error = 0;
87 1.15 cgd #ifdef KTRACE
88 1.15 cgd struct iovec ktriov;
89 1.15 cgd #endif
90 1.15 cgd
91 1.34 mycroft if ((u_int)fd >= fdp->fd_nfiles ||
92 1.34 mycroft (fp = fdp->fd_ofiles[fd]) == NULL ||
93 1.15 cgd (fp->f_flag & FREAD) == 0)
94 1.15 cgd return (EBADF);
95 1.16 cgd aiov.iov_base = (caddr_t)SCARG(uap, buf);
96 1.16 cgd aiov.iov_len = SCARG(uap, nbyte);
97 1.15 cgd auio.uio_iov = &aiov;
98 1.15 cgd auio.uio_iovcnt = 1;
99 1.16 cgd auio.uio_resid = SCARG(uap, nbyte);
100 1.15 cgd auio.uio_rw = UIO_READ;
101 1.15 cgd auio.uio_segflg = UIO_USERSPACE;
102 1.15 cgd auio.uio_procp = p;
103 1.15 cgd if (auio.uio_resid < 0)
104 1.15 cgd return EINVAL;
105 1.15 cgd #ifdef KTRACE
106 1.15 cgd /*
107 1.15 cgd * if tracing, save a copy of iovec
108 1.15 cgd */
109 1.15 cgd if (KTRPOINT(p, KTR_GENIO))
110 1.15 cgd ktriov = aiov;
111 1.15 cgd #endif
112 1.16 cgd cnt = SCARG(uap, nbyte);
113 1.22 christos error = (*fp->f_ops->fo_read)(fp, &auio, fp->f_cred);
114 1.22 christos if (error)
115 1.15 cgd if (auio.uio_resid != cnt && (error == ERESTART ||
116 1.15 cgd error == EINTR || error == EWOULDBLOCK))
117 1.15 cgd error = 0;
118 1.15 cgd cnt -= auio.uio_resid;
119 1.15 cgd #ifdef KTRACE
120 1.15 cgd if (KTRPOINT(p, KTR_GENIO) && error == 0)
121 1.34 mycroft ktrgenio(p->p_tracep, fd, UIO_READ, &ktriov, cnt, error);
122 1.15 cgd #endif
123 1.15 cgd *retval = cnt;
124 1.15 cgd return (error);
125 1.15 cgd }
126 1.15 cgd
127 1.15 cgd /*
128 1.15 cgd * Scatter read system call.
129 1.15 cgd */
130 1.22 christos int
131 1.21 mycroft sys_readv(p, v, retval)
132 1.15 cgd struct proc *p;
133 1.20 thorpej void *v;
134 1.20 thorpej register_t *retval;
135 1.20 thorpej {
136 1.21 mycroft register struct sys_readv_args /* {
137 1.16 cgd syscallarg(int) fd;
138 1.31 cgd syscallarg(const struct iovec *) iovp;
139 1.34 mycroft syscallarg(int) iovcnt;
140 1.20 thorpej } */ *uap = v;
141 1.34 mycroft int fd = SCARG(uap, fd);
142 1.34 mycroft int iovcnt = SCARG(uap, iovcnt);
143 1.15 cgd register struct file *fp;
144 1.15 cgd register struct filedesc *fdp = p->p_fd;
145 1.15 cgd struct uio auio;
146 1.15 cgd register struct iovec *iov;
147 1.15 cgd struct iovec *needfree;
148 1.15 cgd struct iovec aiov[UIO_SMALLIOV];
149 1.15 cgd long i, cnt, error = 0;
150 1.15 cgd u_int iovlen;
151 1.15 cgd #ifdef KTRACE
152 1.15 cgd struct iovec *ktriov = NULL;
153 1.15 cgd #endif
154 1.15 cgd
155 1.34 mycroft if ((u_int)fd >= fdp->fd_nfiles ||
156 1.34 mycroft (fp = fdp->fd_ofiles[fd]) == NULL ||
157 1.15 cgd (fp->f_flag & FREAD) == 0)
158 1.15 cgd return (EBADF);
159 1.15 cgd /* note: can't use iovlen until iovcnt is validated */
160 1.34 mycroft iovlen = iovcnt * sizeof (struct iovec);
161 1.34 mycroft if ((u_int)iovcnt > UIO_SMALLIOV) {
162 1.34 mycroft if ((u_int)iovcnt > UIO_MAXIOV)
163 1.15 cgd return (EINVAL);
164 1.15 cgd MALLOC(iov, struct iovec *, iovlen, M_IOV, M_WAITOK);
165 1.15 cgd needfree = iov;
166 1.15 cgd } else {
167 1.15 cgd iov = aiov;
168 1.15 cgd needfree = NULL;
169 1.15 cgd }
170 1.15 cgd auio.uio_iov = iov;
171 1.34 mycroft auio.uio_iovcnt = iovcnt;
172 1.15 cgd auio.uio_rw = UIO_READ;
173 1.15 cgd auio.uio_segflg = UIO_USERSPACE;
174 1.15 cgd auio.uio_procp = p;
175 1.31 cgd error = copyin(SCARG(uap, iovp), iov, iovlen);
176 1.22 christos if (error)
177 1.15 cgd goto done;
178 1.15 cgd auio.uio_resid = 0;
179 1.34 mycroft for (i = 0; i < iovcnt; i++) {
180 1.15 cgd auio.uio_resid += iov->iov_len;
181 1.15 cgd if (auio.uio_resid < 0) {
182 1.15 cgd error = EINVAL;
183 1.15 cgd goto done;
184 1.15 cgd }
185 1.15 cgd iov++;
186 1.15 cgd }
187 1.15 cgd #ifdef KTRACE
188 1.15 cgd /*
189 1.15 cgd * if tracing, save a copy of iovec
190 1.15 cgd */
191 1.15 cgd if (KTRPOINT(p, KTR_GENIO)) {
192 1.15 cgd MALLOC(ktriov, struct iovec *, iovlen, M_TEMP, M_WAITOK);
193 1.15 cgd bcopy((caddr_t)auio.uio_iov, (caddr_t)ktriov, iovlen);
194 1.15 cgd }
195 1.15 cgd #endif
196 1.15 cgd cnt = auio.uio_resid;
197 1.22 christos error = (*fp->f_ops->fo_read)(fp, &auio, fp->f_cred);
198 1.22 christos if (error)
199 1.15 cgd if (auio.uio_resid != cnt && (error == ERESTART ||
200 1.15 cgd error == EINTR || error == EWOULDBLOCK))
201 1.15 cgd error = 0;
202 1.15 cgd cnt -= auio.uio_resid;
203 1.15 cgd #ifdef KTRACE
204 1.34 mycroft if (KTRPOINT(p, KTR_GENIO))
205 1.34 mycroft if (error == 0) {
206 1.34 mycroft ktrgenio(p->p_tracep, fd, UIO_READ, ktriov, cnt,
207 1.34 mycroft error);
208 1.15 cgd FREE(ktriov, M_TEMP);
209 1.15 cgd }
210 1.15 cgd #endif
211 1.15 cgd *retval = cnt;
212 1.15 cgd done:
213 1.15 cgd if (needfree)
214 1.15 cgd FREE(needfree, M_IOV);
215 1.15 cgd return (error);
216 1.15 cgd }
217 1.15 cgd
218 1.15 cgd /*
219 1.15 cgd * Write system call
220 1.15 cgd */
221 1.22 christos int
222 1.21 mycroft sys_write(p, v, retval)
223 1.15 cgd struct proc *p;
224 1.20 thorpej void *v;
225 1.20 thorpej register_t *retval;
226 1.20 thorpej {
227 1.21 mycroft register struct sys_write_args /* {
228 1.16 cgd syscallarg(int) fd;
229 1.31 cgd syscallarg(const void *) buf;
230 1.31 cgd syscallarg(size_t) nbyte;
231 1.20 thorpej } */ *uap = v;
232 1.34 mycroft int fd = SCARG(uap, fd);
233 1.15 cgd register struct file *fp;
234 1.15 cgd register struct filedesc *fdp = p->p_fd;
235 1.15 cgd struct uio auio;
236 1.15 cgd struct iovec aiov;
237 1.15 cgd long cnt, error = 0;
238 1.15 cgd #ifdef KTRACE
239 1.15 cgd struct iovec ktriov;
240 1.15 cgd #endif
241 1.15 cgd
242 1.34 mycroft if ((u_int)fd >= fdp->fd_nfiles ||
243 1.34 mycroft (fp = fdp->fd_ofiles[fd]) == NULL ||
244 1.15 cgd (fp->f_flag & FWRITE) == 0)
245 1.15 cgd return (EBADF);
246 1.31 cgd aiov.iov_base = (char *)SCARG(uap, buf); /* XXX kills const */
247 1.16 cgd aiov.iov_len = SCARG(uap, nbyte);
248 1.15 cgd auio.uio_iov = &aiov;
249 1.15 cgd auio.uio_iovcnt = 1;
250 1.16 cgd auio.uio_resid = SCARG(uap, nbyte);
251 1.15 cgd auio.uio_rw = UIO_WRITE;
252 1.15 cgd auio.uio_segflg = UIO_USERSPACE;
253 1.15 cgd auio.uio_procp = p;
254 1.15 cgd if (auio.uio_resid < 0)
255 1.15 cgd return EINVAL;
256 1.15 cgd #ifdef KTRACE
257 1.15 cgd /*
258 1.15 cgd * if tracing, save a copy of iovec
259 1.15 cgd */
260 1.15 cgd if (KTRPOINT(p, KTR_GENIO))
261 1.15 cgd ktriov = aiov;
262 1.15 cgd #endif
263 1.16 cgd cnt = SCARG(uap, nbyte);
264 1.22 christos error = (*fp->f_ops->fo_write)(fp, &auio, fp->f_cred);
265 1.22 christos if (error) {
266 1.15 cgd if (auio.uio_resid != cnt && (error == ERESTART ||
267 1.15 cgd error == EINTR || error == EWOULDBLOCK))
268 1.15 cgd error = 0;
269 1.15 cgd if (error == EPIPE)
270 1.15 cgd psignal(p, SIGPIPE);
271 1.15 cgd }
272 1.15 cgd cnt -= auio.uio_resid;
273 1.15 cgd #ifdef KTRACE
274 1.15 cgd if (KTRPOINT(p, KTR_GENIO) && error == 0)
275 1.34 mycroft ktrgenio(p->p_tracep, fd, UIO_WRITE, &ktriov, cnt, error);
276 1.15 cgd #endif
277 1.15 cgd *retval = cnt;
278 1.15 cgd return (error);
279 1.15 cgd }
280 1.15 cgd
281 1.15 cgd /*
282 1.15 cgd * Gather write system call
283 1.15 cgd */
284 1.22 christos int
285 1.21 mycroft sys_writev(p, v, retval)
286 1.15 cgd struct proc *p;
287 1.20 thorpej void *v;
288 1.20 thorpej register_t *retval;
289 1.20 thorpej {
290 1.21 mycroft register struct sys_writev_args /* {
291 1.16 cgd syscallarg(int) fd;
292 1.31 cgd syscallarg(const struct iovec *) iovp;
293 1.16 cgd syscallarg(u_int) iovcnt;
294 1.20 thorpej } */ *uap = v;
295 1.34 mycroft int fd = SCARG(uap, fd);
296 1.34 mycroft int iovcnt = SCARG(uap, iovcnt);
297 1.15 cgd register struct file *fp;
298 1.15 cgd register struct filedesc *fdp = p->p_fd;
299 1.15 cgd struct uio auio;
300 1.15 cgd register struct iovec *iov;
301 1.15 cgd struct iovec *needfree;
302 1.15 cgd struct iovec aiov[UIO_SMALLIOV];
303 1.15 cgd long i, cnt, error = 0;
304 1.15 cgd u_int iovlen;
305 1.15 cgd #ifdef KTRACE
306 1.15 cgd struct iovec *ktriov = NULL;
307 1.15 cgd #endif
308 1.15 cgd
309 1.34 mycroft if ((u_int)fd >= fdp->fd_nfiles ||
310 1.34 mycroft (fp = fdp->fd_ofiles[fd]) == NULL ||
311 1.15 cgd (fp->f_flag & FWRITE) == 0)
312 1.15 cgd return (EBADF);
313 1.15 cgd /* note: can't use iovlen until iovcnt is validated */
314 1.34 mycroft iovlen = iovcnt * sizeof (struct iovec);
315 1.34 mycroft if ((u_int)iovcnt > UIO_SMALLIOV) {
316 1.34 mycroft if ((u_int)iovcnt > UIO_MAXIOV)
317 1.15 cgd return (EINVAL);
318 1.15 cgd MALLOC(iov, struct iovec *, iovlen, M_IOV, M_WAITOK);
319 1.15 cgd needfree = iov;
320 1.15 cgd } else {
321 1.15 cgd iov = aiov;
322 1.15 cgd needfree = NULL;
323 1.15 cgd }
324 1.15 cgd auio.uio_iov = iov;
325 1.34 mycroft auio.uio_iovcnt = iovcnt;
326 1.15 cgd auio.uio_rw = UIO_WRITE;
327 1.15 cgd auio.uio_segflg = UIO_USERSPACE;
328 1.15 cgd auio.uio_procp = p;
329 1.31 cgd error = copyin(SCARG(uap, iovp), iov, iovlen);
330 1.22 christos if (error)
331 1.15 cgd goto done;
332 1.15 cgd auio.uio_resid = 0;
333 1.34 mycroft for (i = 0; i < iovcnt; i++) {
334 1.15 cgd auio.uio_resid += iov->iov_len;
335 1.15 cgd if (auio.uio_resid < 0) {
336 1.15 cgd error = EINVAL;
337 1.15 cgd goto done;
338 1.15 cgd }
339 1.15 cgd iov++;
340 1.15 cgd }
341 1.15 cgd #ifdef KTRACE
342 1.15 cgd /*
343 1.15 cgd * if tracing, save a copy of iovec
344 1.15 cgd */
345 1.15 cgd if (KTRPOINT(p, KTR_GENIO)) {
346 1.15 cgd MALLOC(ktriov, struct iovec *, iovlen, M_TEMP, M_WAITOK);
347 1.15 cgd bcopy((caddr_t)auio.uio_iov, (caddr_t)ktriov, iovlen);
348 1.15 cgd }
349 1.15 cgd #endif
350 1.15 cgd cnt = auio.uio_resid;
351 1.22 christos error = (*fp->f_ops->fo_write)(fp, &auio, fp->f_cred);
352 1.22 christos if (error) {
353 1.15 cgd if (auio.uio_resid != cnt && (error == ERESTART ||
354 1.15 cgd error == EINTR || error == EWOULDBLOCK))
355 1.15 cgd error = 0;
356 1.15 cgd if (error == EPIPE)
357 1.15 cgd psignal(p, SIGPIPE);
358 1.15 cgd }
359 1.15 cgd cnt -= auio.uio_resid;
360 1.15 cgd #ifdef KTRACE
361 1.34 mycroft if (KTRPOINT(p, KTR_GENIO))
362 1.34 mycroft if (error == 0) {
363 1.34 mycroft ktrgenio(p->p_tracep, fd, UIO_WRITE, ktriov, cnt,
364 1.34 mycroft error);
365 1.15 cgd FREE(ktriov, M_TEMP);
366 1.15 cgd }
367 1.15 cgd #endif
368 1.15 cgd *retval = cnt;
369 1.15 cgd done:
370 1.15 cgd if (needfree)
371 1.15 cgd FREE(needfree, M_IOV);
372 1.15 cgd return (error);
373 1.15 cgd }
374 1.15 cgd
375 1.15 cgd /*
376 1.15 cgd * Ioctl system call
377 1.15 cgd */
378 1.15 cgd /* ARGSUSED */
379 1.22 christos int
380 1.21 mycroft sys_ioctl(p, v, retval)
381 1.15 cgd struct proc *p;
382 1.20 thorpej void *v;
383 1.20 thorpej register_t *retval;
384 1.20 thorpej {
385 1.21 mycroft register struct sys_ioctl_args /* {
386 1.16 cgd syscallarg(int) fd;
387 1.17 cgd syscallarg(u_long) com;
388 1.16 cgd syscallarg(caddr_t) data;
389 1.20 thorpej } */ *uap = v;
390 1.15 cgd register struct file *fp;
391 1.15 cgd register struct filedesc *fdp;
392 1.17 cgd register u_long com;
393 1.17 cgd register int error;
394 1.15 cgd register u_int size;
395 1.15 cgd caddr_t data, memp;
396 1.15 cgd int tmp;
397 1.15 cgd #define STK_PARAMS 128
398 1.15 cgd char stkbuf[STK_PARAMS];
399 1.15 cgd
400 1.15 cgd fdp = p->p_fd;
401 1.16 cgd if ((u_int)SCARG(uap, fd) >= fdp->fd_nfiles ||
402 1.16 cgd (fp = fdp->fd_ofiles[SCARG(uap, fd)]) == NULL)
403 1.15 cgd return (EBADF);
404 1.15 cgd
405 1.15 cgd if ((fp->f_flag & (FREAD | FWRITE)) == 0)
406 1.15 cgd return (EBADF);
407 1.15 cgd
408 1.16 cgd switch (com = SCARG(uap, com)) {
409 1.15 cgd case FIONCLEX:
410 1.16 cgd fdp->fd_ofileflags[SCARG(uap, fd)] &= ~UF_EXCLOSE;
411 1.15 cgd return (0);
412 1.15 cgd case FIOCLEX:
413 1.16 cgd fdp->fd_ofileflags[SCARG(uap, fd)] |= UF_EXCLOSE;
414 1.15 cgd return (0);
415 1.15 cgd }
416 1.15 cgd
417 1.15 cgd /*
418 1.15 cgd * Interpret high order word to find amount of data to be
419 1.15 cgd * copied to/from the user's address space.
420 1.15 cgd */
421 1.15 cgd size = IOCPARM_LEN(com);
422 1.15 cgd if (size > IOCPARM_MAX)
423 1.15 cgd return (ENOTTY);
424 1.15 cgd memp = NULL;
425 1.15 cgd if (size > sizeof (stkbuf)) {
426 1.15 cgd memp = (caddr_t)malloc((u_long)size, M_IOCTLOPS, M_WAITOK);
427 1.15 cgd data = memp;
428 1.15 cgd } else
429 1.15 cgd data = stkbuf;
430 1.15 cgd if (com&IOC_IN) {
431 1.15 cgd if (size) {
432 1.31 cgd error = copyin(SCARG(uap, data), data, size);
433 1.15 cgd if (error) {
434 1.15 cgd if (memp)
435 1.15 cgd free(memp, M_IOCTLOPS);
436 1.15 cgd return (error);
437 1.15 cgd }
438 1.15 cgd } else
439 1.16 cgd *(caddr_t *)data = SCARG(uap, data);
440 1.15 cgd } else if ((com&IOC_OUT) && size)
441 1.15 cgd /*
442 1.15 cgd * Zero the buffer so the user always
443 1.15 cgd * gets back something deterministic.
444 1.15 cgd */
445 1.15 cgd bzero(data, size);
446 1.15 cgd else if (com&IOC_VOID)
447 1.16 cgd *(caddr_t *)data = SCARG(uap, data);
448 1.15 cgd
449 1.15 cgd switch (com) {
450 1.15 cgd
451 1.15 cgd case FIONBIO:
452 1.22 christos if ((tmp = *(int *)data) != 0)
453 1.15 cgd fp->f_flag |= FNONBLOCK;
454 1.15 cgd else
455 1.15 cgd fp->f_flag &= ~FNONBLOCK;
456 1.15 cgd error = (*fp->f_ops->fo_ioctl)(fp, FIONBIO, (caddr_t)&tmp, p);
457 1.15 cgd break;
458 1.15 cgd
459 1.15 cgd case FIOASYNC:
460 1.22 christos if ((tmp = *(int *)data) != 0)
461 1.15 cgd fp->f_flag |= FASYNC;
462 1.15 cgd else
463 1.15 cgd fp->f_flag &= ~FASYNC;
464 1.15 cgd error = (*fp->f_ops->fo_ioctl)(fp, FIOASYNC, (caddr_t)&tmp, p);
465 1.15 cgd break;
466 1.15 cgd
467 1.15 cgd case FIOSETOWN:
468 1.15 cgd tmp = *(int *)data;
469 1.15 cgd if (fp->f_type == DTYPE_SOCKET) {
470 1.15 cgd ((struct socket *)fp->f_data)->so_pgid = tmp;
471 1.15 cgd error = 0;
472 1.15 cgd break;
473 1.15 cgd }
474 1.15 cgd if (tmp <= 0) {
475 1.15 cgd tmp = -tmp;
476 1.15 cgd } else {
477 1.15 cgd struct proc *p1 = pfind(tmp);
478 1.15 cgd if (p1 == 0) {
479 1.15 cgd error = ESRCH;
480 1.15 cgd break;
481 1.15 cgd }
482 1.15 cgd tmp = p1->p_pgrp->pg_id;
483 1.15 cgd }
484 1.15 cgd error = (*fp->f_ops->fo_ioctl)
485 1.24 cgd (fp, TIOCSPGRP, (caddr_t)&tmp, p);
486 1.15 cgd break;
487 1.15 cgd
488 1.15 cgd case FIOGETOWN:
489 1.15 cgd if (fp->f_type == DTYPE_SOCKET) {
490 1.15 cgd error = 0;
491 1.15 cgd *(int *)data = ((struct socket *)fp->f_data)->so_pgid;
492 1.15 cgd break;
493 1.15 cgd }
494 1.17 cgd error = (*fp->f_ops->fo_ioctl)(fp, TIOCGPGRP, data, p);
495 1.15 cgd *(int *)data = -*(int *)data;
496 1.15 cgd break;
497 1.15 cgd
498 1.15 cgd default:
499 1.15 cgd error = (*fp->f_ops->fo_ioctl)(fp, com, data, p);
500 1.15 cgd /*
501 1.15 cgd * Copy any data to user, size was
502 1.15 cgd * already set and checked above.
503 1.15 cgd */
504 1.15 cgd if (error == 0 && (com&IOC_OUT) && size)
505 1.31 cgd error = copyout(data, SCARG(uap, data), size);
506 1.15 cgd break;
507 1.15 cgd }
508 1.15 cgd if (memp)
509 1.15 cgd free(memp, M_IOCTLOPS);
510 1.15 cgd return (error);
511 1.15 cgd }
512 1.15 cgd
513 1.15 cgd int selwait, nselcoll;
514 1.15 cgd
515 1.15 cgd /*
516 1.15 cgd * Select system call.
517 1.15 cgd */
518 1.22 christos int
519 1.21 mycroft sys_select(p, v, retval)
520 1.15 cgd register struct proc *p;
521 1.20 thorpej void *v;
522 1.20 thorpej register_t *retval;
523 1.20 thorpej {
524 1.21 mycroft register struct sys_select_args /* {
525 1.35 thorpej syscallarg(int) nd;
526 1.16 cgd syscallarg(fd_set *) in;
527 1.16 cgd syscallarg(fd_set *) ou;
528 1.16 cgd syscallarg(fd_set *) ex;
529 1.16 cgd syscallarg(struct timeval *) tv;
530 1.20 thorpej } */ *uap = v;
531 1.25 mycroft caddr_t bits;
532 1.26 cgd char smallbits[howmany(FD_SETSIZE, NFDBITS) * sizeof(fd_mask) * 6];
533 1.15 cgd struct timeval atv;
534 1.15 cgd int s, ncoll, error = 0, timo;
535 1.27 mycroft size_t ni;
536 1.15 cgd
537 1.35 thorpej if (SCARG(uap, nd) < 0)
538 1.35 thorpej return (EINVAL);
539 1.16 cgd if (SCARG(uap, nd) > p->p_fd->fd_nfiles) {
540 1.16 cgd /* forgiving; slightly wrong */
541 1.16 cgd SCARG(uap, nd) = p->p_fd->fd_nfiles;
542 1.16 cgd }
543 1.16 cgd ni = howmany(SCARG(uap, nd), NFDBITS) * sizeof(fd_mask);
544 1.27 mycroft if (ni * 6 > sizeof(smallbits))
545 1.25 mycroft bits = malloc(ni * 6, M_TEMP, M_WAITOK);
546 1.25 mycroft else
547 1.26 cgd bits = smallbits;
548 1.15 cgd
549 1.15 cgd #define getbits(name, x) \
550 1.25 mycroft if (SCARG(uap, name)) { \
551 1.31 cgd error = copyin(SCARG(uap, name), bits + ni * x, ni); \
552 1.25 mycroft if (error) \
553 1.25 mycroft goto done; \
554 1.25 mycroft } else \
555 1.25 mycroft bzero(bits + ni * x, ni);
556 1.15 cgd getbits(in, 0);
557 1.15 cgd getbits(ou, 1);
558 1.15 cgd getbits(ex, 2);
559 1.15 cgd #undef getbits
560 1.15 cgd
561 1.16 cgd if (SCARG(uap, tv)) {
562 1.31 cgd error = copyin(SCARG(uap, tv), (caddr_t)&atv,
563 1.15 cgd sizeof (atv));
564 1.15 cgd if (error)
565 1.15 cgd goto done;
566 1.15 cgd if (itimerfix(&atv)) {
567 1.15 cgd error = EINVAL;
568 1.15 cgd goto done;
569 1.15 cgd }
570 1.15 cgd s = splclock();
571 1.19 mycroft timeradd(&atv, &time, &atv);
572 1.15 cgd timo = hzto(&atv);
573 1.15 cgd /*
574 1.15 cgd * Avoid inadvertently sleeping forever.
575 1.15 cgd */
576 1.15 cgd if (timo == 0)
577 1.15 cgd timo = 1;
578 1.15 cgd splx(s);
579 1.15 cgd } else
580 1.15 cgd timo = 0;
581 1.15 cgd retry:
582 1.15 cgd ncoll = nselcoll;
583 1.15 cgd p->p_flag |= P_SELECT;
584 1.25 mycroft error = selscan(p, (fd_mask *)(bits + ni * 0),
585 1.25 mycroft (fd_mask *)(bits + ni * 3), SCARG(uap, nd), retval);
586 1.15 cgd if (error || *retval)
587 1.15 cgd goto done;
588 1.15 cgd s = splhigh();
589 1.27 mycroft if (timo && timercmp(&time, &atv, >=)) {
590 1.15 cgd splx(s);
591 1.15 cgd goto done;
592 1.15 cgd }
593 1.15 cgd if ((p->p_flag & P_SELECT) == 0 || nselcoll != ncoll) {
594 1.15 cgd splx(s);
595 1.15 cgd goto retry;
596 1.15 cgd }
597 1.15 cgd p->p_flag &= ~P_SELECT;
598 1.15 cgd error = tsleep((caddr_t)&selwait, PSOCK | PCATCH, "select", timo);
599 1.15 cgd splx(s);
600 1.15 cgd if (error == 0)
601 1.15 cgd goto retry;
602 1.15 cgd done:
603 1.15 cgd p->p_flag &= ~P_SELECT;
604 1.15 cgd /* select is not restarted after signals... */
605 1.15 cgd if (error == ERESTART)
606 1.15 cgd error = EINTR;
607 1.15 cgd if (error == EWOULDBLOCK)
608 1.15 cgd error = 0;
609 1.27 mycroft if (error == 0) {
610 1.15 cgd #define putbits(name, x) \
611 1.27 mycroft if (SCARG(uap, name)) { \
612 1.31 cgd error = copyout(bits + ni * x, SCARG(uap, name), ni); \
613 1.27 mycroft if (error) \
614 1.27 mycroft goto out; \
615 1.27 mycroft }
616 1.25 mycroft putbits(in, 3);
617 1.25 mycroft putbits(ou, 4);
618 1.25 mycroft putbits(ex, 5);
619 1.15 cgd #undef putbits
620 1.15 cgd }
621 1.27 mycroft out:
622 1.27 mycroft if (ni * 6 > sizeof(smallbits))
623 1.25 mycroft free(bits, M_TEMP);
624 1.15 cgd return (error);
625 1.15 cgd }
626 1.15 cgd
627 1.22 christos int
628 1.25 mycroft selscan(p, ibitp, obitp, nfd, retval)
629 1.15 cgd struct proc *p;
630 1.25 mycroft fd_mask *ibitp, *obitp;
631 1.16 cgd int nfd;
632 1.16 cgd register_t *retval;
633 1.15 cgd {
634 1.15 cgd register struct filedesc *fdp = p->p_fd;
635 1.15 cgd register int msk, i, j, fd;
636 1.25 mycroft register fd_mask ibits, obits;
637 1.15 cgd struct file *fp;
638 1.15 cgd int n = 0;
639 1.28 mycroft static int flag[3] = { POLLRDNORM | POLLHUP | POLLERR,
640 1.28 mycroft POLLWRNORM | POLLHUP | POLLERR,
641 1.28 mycroft POLLRDBAND };
642 1.15 cgd
643 1.15 cgd for (msk = 0; msk < 3; msk++) {
644 1.15 cgd for (i = 0; i < nfd; i += NFDBITS) {
645 1.25 mycroft ibits = *ibitp++;
646 1.25 mycroft obits = 0;
647 1.25 mycroft while ((j = ffs(ibits)) && (fd = i + --j) < nfd) {
648 1.25 mycroft ibits &= ~(1 << j);
649 1.15 cgd fp = fdp->fd_ofiles[fd];
650 1.15 cgd if (fp == NULL)
651 1.15 cgd return (EBADF);
652 1.28 mycroft if ((*fp->f_ops->fo_poll)(fp, flag[msk], p)) {
653 1.25 mycroft obits |= (1 << j);
654 1.15 cgd n++;
655 1.15 cgd }
656 1.15 cgd }
657 1.25 mycroft *obitp++ = obits;
658 1.15 cgd }
659 1.15 cgd }
660 1.15 cgd *retval = n;
661 1.15 cgd return (0);
662 1.15 cgd }
663 1.15 cgd
664 1.28 mycroft /*
665 1.28 mycroft * Poll system call.
666 1.28 mycroft */
667 1.28 mycroft int
668 1.28 mycroft sys_poll(p, v, retval)
669 1.28 mycroft register struct proc *p;
670 1.28 mycroft void *v;
671 1.28 mycroft register_t *retval;
672 1.28 mycroft {
673 1.28 mycroft register struct sys_poll_args /* {
674 1.28 mycroft syscallarg(struct pollfd *) fds;
675 1.29 mycroft syscallarg(u_int) nfds;
676 1.28 mycroft syscallarg(int) timeout;
677 1.28 mycroft } */ *uap = v;
678 1.28 mycroft caddr_t bits;
679 1.28 mycroft char smallbits[32 * sizeof(struct pollfd)];
680 1.28 mycroft struct timeval atv;
681 1.28 mycroft int s, ncoll, error = 0, timo;
682 1.28 mycroft size_t ni;
683 1.28 mycroft
684 1.28 mycroft if (SCARG(uap, nfds) > p->p_fd->fd_nfiles) {
685 1.28 mycroft /* forgiving; slightly wrong */
686 1.28 mycroft SCARG(uap, nfds) = p->p_fd->fd_nfiles;
687 1.28 mycroft }
688 1.28 mycroft ni = SCARG(uap, nfds) * sizeof(struct pollfd);
689 1.28 mycroft if (ni > sizeof(smallbits))
690 1.28 mycroft bits = malloc(ni, M_TEMP, M_WAITOK);
691 1.28 mycroft else
692 1.28 mycroft bits = smallbits;
693 1.28 mycroft
694 1.31 cgd error = copyin(SCARG(uap, fds), bits, ni);
695 1.28 mycroft if (error)
696 1.28 mycroft goto done;
697 1.28 mycroft
698 1.30 mycroft if (SCARG(uap, timeout) != INFTIM) {
699 1.28 mycroft atv.tv_sec = SCARG(uap, timeout) / 1000;
700 1.28 mycroft atv.tv_usec = (SCARG(uap, timeout) % 1000) * 1000;
701 1.28 mycroft if (itimerfix(&atv)) {
702 1.28 mycroft error = EINVAL;
703 1.28 mycroft goto done;
704 1.28 mycroft }
705 1.28 mycroft s = splclock();
706 1.28 mycroft timeradd(&atv, &time, &atv);
707 1.28 mycroft timo = hzto(&atv);
708 1.28 mycroft /*
709 1.28 mycroft * Avoid inadvertently sleeping forever.
710 1.28 mycroft */
711 1.28 mycroft if (timo == 0)
712 1.28 mycroft timo = 1;
713 1.28 mycroft splx(s);
714 1.28 mycroft } else
715 1.28 mycroft timo = 0;
716 1.28 mycroft retry:
717 1.28 mycroft ncoll = nselcoll;
718 1.28 mycroft p->p_flag |= P_SELECT;
719 1.28 mycroft error = pollscan(p, (struct pollfd *)bits, SCARG(uap, nfds), retval);
720 1.28 mycroft if (error || *retval)
721 1.28 mycroft goto done;
722 1.28 mycroft s = splhigh();
723 1.28 mycroft if (timo && timercmp(&time, &atv, >=)) {
724 1.28 mycroft splx(s);
725 1.28 mycroft goto done;
726 1.28 mycroft }
727 1.28 mycroft if ((p->p_flag & P_SELECT) == 0 || nselcoll != ncoll) {
728 1.28 mycroft splx(s);
729 1.28 mycroft goto retry;
730 1.28 mycroft }
731 1.28 mycroft p->p_flag &= ~P_SELECT;
732 1.28 mycroft error = tsleep((caddr_t)&selwait, PSOCK | PCATCH, "select", timo);
733 1.28 mycroft splx(s);
734 1.28 mycroft if (error == 0)
735 1.28 mycroft goto retry;
736 1.28 mycroft done:
737 1.28 mycroft p->p_flag &= ~P_SELECT;
738 1.28 mycroft /* poll is not restarted after signals... */
739 1.28 mycroft if (error == ERESTART)
740 1.28 mycroft error = EINTR;
741 1.28 mycroft if (error == EWOULDBLOCK)
742 1.28 mycroft error = 0;
743 1.28 mycroft if (error == 0) {
744 1.31 cgd error = copyout(bits, SCARG(uap, fds), ni);
745 1.28 mycroft if (error)
746 1.28 mycroft goto out;
747 1.28 mycroft }
748 1.28 mycroft out:
749 1.28 mycroft if (ni > sizeof(smallbits))
750 1.28 mycroft free(bits, M_TEMP);
751 1.28 mycroft return (error);
752 1.28 mycroft }
753 1.28 mycroft
754 1.28 mycroft int
755 1.28 mycroft pollscan(p, fds, nfd, retval)
756 1.28 mycroft struct proc *p;
757 1.28 mycroft struct pollfd *fds;
758 1.28 mycroft int nfd;
759 1.28 mycroft register_t *retval;
760 1.28 mycroft {
761 1.28 mycroft register struct filedesc *fdp = p->p_fd;
762 1.28 mycroft int i;
763 1.28 mycroft struct file *fp;
764 1.28 mycroft int n = 0;
765 1.28 mycroft
766 1.28 mycroft for (i = 0; i < nfd; i++, fds++) {
767 1.33 mrg if ((u_int)fds->fd >= fdp->fd_nfiles) {
768 1.28 mycroft fds->revents = POLLNVAL;
769 1.28 mycroft n++;
770 1.28 mycroft } else {
771 1.32 mrg fp = fdp->fd_ofiles[fds->fd];
772 1.32 mrg if (fp == 0) {
773 1.32 mrg fds->revents = POLLNVAL;
774 1.28 mycroft n++;
775 1.32 mrg } else {
776 1.32 mrg fds->revents = (*fp->f_ops->fo_poll)(fp,
777 1.32 mrg fds->events | POLLERR | POLLHUP, p);
778 1.32 mrg if (fds->revents != 0)
779 1.32 mrg n++;
780 1.32 mrg }
781 1.28 mycroft }
782 1.28 mycroft }
783 1.28 mycroft *retval = n;
784 1.28 mycroft return (0);
785 1.28 mycroft }
786 1.28 mycroft
787 1.15 cgd /*ARGSUSED*/
788 1.22 christos int
789 1.28 mycroft seltrue(dev, events, p)
790 1.15 cgd dev_t dev;
791 1.28 mycroft int events;
792 1.15 cgd struct proc *p;
793 1.15 cgd {
794 1.15 cgd
795 1.28 mycroft return (events & (POLLIN | POLLOUT | POLLRDNORM | POLLWRNORM));
796 1.15 cgd }
797 1.15 cgd
798 1.15 cgd /*
799 1.15 cgd * Record a select request.
800 1.15 cgd */
801 1.15 cgd void
802 1.15 cgd selrecord(selector, sip)
803 1.15 cgd struct proc *selector;
804 1.15 cgd struct selinfo *sip;
805 1.15 cgd {
806 1.15 cgd struct proc *p;
807 1.15 cgd pid_t mypid;
808 1.15 cgd
809 1.15 cgd mypid = selector->p_pid;
810 1.15 cgd if (sip->si_pid == mypid)
811 1.15 cgd return;
812 1.15 cgd if (sip->si_pid && (p = pfind(sip->si_pid)) &&
813 1.15 cgd p->p_wchan == (caddr_t)&selwait)
814 1.15 cgd sip->si_flags |= SI_COLL;
815 1.15 cgd else
816 1.15 cgd sip->si_pid = mypid;
817 1.15 cgd }
818 1.15 cgd
819 1.15 cgd /*
820 1.15 cgd * Do a wakeup when a selectable event occurs.
821 1.15 cgd */
822 1.15 cgd void
823 1.15 cgd selwakeup(sip)
824 1.15 cgd register struct selinfo *sip;
825 1.15 cgd {
826 1.15 cgd register struct proc *p;
827 1.15 cgd int s;
828 1.15 cgd
829 1.15 cgd if (sip->si_pid == 0)
830 1.15 cgd return;
831 1.15 cgd if (sip->si_flags & SI_COLL) {
832 1.15 cgd nselcoll++;
833 1.15 cgd sip->si_flags &= ~SI_COLL;
834 1.15 cgd wakeup((caddr_t)&selwait);
835 1.15 cgd }
836 1.15 cgd p = pfind(sip->si_pid);
837 1.15 cgd sip->si_pid = 0;
838 1.15 cgd if (p != NULL) {
839 1.15 cgd s = splhigh();
840 1.15 cgd if (p->p_wchan == (caddr_t)&selwait) {
841 1.15 cgd if (p->p_stat == SSLEEP)
842 1.15 cgd setrunnable(p);
843 1.15 cgd else
844 1.15 cgd unsleep(p);
845 1.15 cgd } else if (p->p_flag & P_SELECT)
846 1.15 cgd p->p_flag &= ~P_SELECT;
847 1.15 cgd splx(s);
848 1.15 cgd }
849 1.15 cgd }
850