sys_generic.c revision 1.103 1 1.103 ad /* $NetBSD: sys_generic.c,v 1.103 2007/07/09 21:10:56 ad Exp $ */
2 1.103 ad
3 1.103 ad /*-
4 1.103 ad * Copyright (c) 2007 The NetBSD Foundation, Inc.
5 1.103 ad * All rights reserved.
6 1.103 ad *
7 1.103 ad * This code is derived from software contributed to The NetBSD Foundation
8 1.103 ad * by Andrew Doran.
9 1.103 ad *
10 1.103 ad * Redistribution and use in source and binary forms, with or without
11 1.103 ad * modification, are permitted provided that the following conditions
12 1.103 ad * are met:
13 1.103 ad * 1. Redistributions of source code must retain the above copyright
14 1.103 ad * notice, this list of conditions and the following disclaimer.
15 1.103 ad * 2. Redistributions in binary form must reproduce the above copyright
16 1.103 ad * notice, this list of conditions and the following disclaimer in the
17 1.103 ad * documentation and/or other materials provided with the distribution.
18 1.103 ad * 3. All advertising materials mentioning features or use of this software
19 1.103 ad * must display the following acknowledgement:
20 1.103 ad * This product includes software developed by the NetBSD
21 1.103 ad * Foundation, Inc. and its contributors.
22 1.103 ad * 4. Neither the name of The NetBSD Foundation nor the names of its
23 1.103 ad * contributors may be used to endorse or promote products derived
24 1.103 ad * from this software without specific prior written permission.
25 1.103 ad *
26 1.103 ad * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
27 1.103 ad * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
28 1.103 ad * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
29 1.103 ad * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
30 1.103 ad * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
31 1.103 ad * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
32 1.103 ad * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
33 1.103 ad * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
34 1.103 ad * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
35 1.103 ad * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
36 1.103 ad * POSSIBILITY OF SUCH DAMAGE.
37 1.103 ad */
38 1.15 cgd
39 1.15 cgd /*
40 1.15 cgd * Copyright (c) 1982, 1986, 1989, 1993
41 1.15 cgd * The Regents of the University of California. All rights reserved.
42 1.15 cgd * (c) UNIX System Laboratories, Inc.
43 1.15 cgd * All or some portions of this file are derived from material licensed
44 1.15 cgd * to the University of California by American Telephone and Telegraph
45 1.15 cgd * Co. or Unix System Laboratories, Inc. and are reproduced herein with
46 1.15 cgd * the permission of UNIX System Laboratories, Inc.
47 1.15 cgd *
48 1.15 cgd * Redistribution and use in source and binary forms, with or without
49 1.15 cgd * modification, are permitted provided that the following conditions
50 1.15 cgd * are met:
51 1.15 cgd * 1. Redistributions of source code must retain the above copyright
52 1.15 cgd * notice, this list of conditions and the following disclaimer.
53 1.15 cgd * 2. Redistributions in binary form must reproduce the above copyright
54 1.15 cgd * notice, this list of conditions and the following disclaimer in the
55 1.15 cgd * documentation and/or other materials provided with the distribution.
56 1.77 agc * 3. Neither the name of the University nor the names of its contributors
57 1.15 cgd * may be used to endorse or promote products derived from this software
58 1.15 cgd * without specific prior written permission.
59 1.15 cgd *
60 1.15 cgd * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
61 1.15 cgd * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
62 1.15 cgd * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
63 1.15 cgd * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
64 1.15 cgd * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
65 1.15 cgd * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
66 1.15 cgd * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
67 1.15 cgd * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
68 1.15 cgd * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
69 1.15 cgd * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
70 1.15 cgd * SUCH DAMAGE.
71 1.15 cgd *
72 1.36 fvdl * @(#)sys_generic.c 8.9 (Berkeley) 2/14/95
73 1.15 cgd */
74 1.59 lukem
75 1.103 ad /*
76 1.103 ad * System calls relating to files.
77 1.103 ad */
78 1.103 ad
79 1.59 lukem #include <sys/cdefs.h>
80 1.103 ad __KERNEL_RCSID(0, "$NetBSD: sys_generic.c,v 1.103 2007/07/09 21:10:56 ad Exp $");
81 1.37 thorpej
82 1.37 thorpej #include "opt_ktrace.h"
83 1.15 cgd
84 1.15 cgd #include <sys/param.h>
85 1.15 cgd #include <sys/systm.h>
86 1.15 cgd #include <sys/filedesc.h>
87 1.15 cgd #include <sys/ioctl.h>
88 1.15 cgd #include <sys/file.h>
89 1.15 cgd #include <sys/proc.h>
90 1.15 cgd #include <sys/socketvar.h>
91 1.22 christos #include <sys/signalvar.h>
92 1.15 cgd #include <sys/uio.h>
93 1.15 cgd #include <sys/kernel.h>
94 1.15 cgd #include <sys/stat.h>
95 1.103 ad #include <sys/kmem.h>
96 1.103 ad #include <sys/poll.h>
97 1.102 dsl #include <sys/vnode.h>
98 1.103 ad #include <sys/mount.h>
99 1.103 ad #include <sys/syscallargs.h>
100 1.15 cgd #ifdef KTRACE
101 1.15 cgd #include <sys/ktrace.h>
102 1.15 cgd #endif
103 1.15 cgd
104 1.85 yamt #include <uvm/uvm_extern.h>
105 1.85 yamt
106 1.103 ad /* Flags for lwp::l_selflag. */
107 1.103 ad #define SEL_RESET 0 /* awoken, interrupted, or not yet polling */
108 1.103 ad #define SEL_SCANNING 1 /* polling descriptors */
109 1.103 ad #define SEL_BLOCKING 2 /* about to block on select_cv */
110 1.103 ad
111 1.103 ad static int selscan(lwp_t *, fd_mask *, fd_mask *, int, register_t *);
112 1.103 ad static int pollscan(lwp_t *, struct pollfd *, int, register_t *);
113 1.103 ad static void selclear(void);
114 1.103 ad
115 1.103 ad /* Global state for select()/poll(). */
116 1.103 ad kmutex_t select_lock;
117 1.103 ad kcondvar_t select_cv;
118 1.103 ad int nselcoll;
119 1.82 matt
120 1.15 cgd /*
121 1.15 cgd * Read system call.
122 1.15 cgd */
123 1.15 cgd /* ARGSUSED */
124 1.22 christos int
125 1.103 ad sys_read(lwp_t *l, void *v, register_t *retval)
126 1.20 thorpej {
127 1.47 augustss struct sys_read_args /* {
128 1.53 lukem syscallarg(int) fd;
129 1.53 lukem syscallarg(void *) buf;
130 1.53 lukem syscallarg(size_t) nbyte;
131 1.20 thorpej } */ *uap = v;
132 1.53 lukem int fd;
133 1.53 lukem struct file *fp;
134 1.103 ad proc_t *p;
135 1.53 lukem struct filedesc *fdp;
136 1.39 thorpej
137 1.53 lukem fd = SCARG(uap, fd);
138 1.69 thorpej p = l->l_proc;
139 1.53 lukem fdp = p->p_fd;
140 1.56 thorpej
141 1.56 thorpej if ((fp = fd_getfile(fdp, fd)) == NULL)
142 1.56 thorpej return (EBADF);
143 1.56 thorpej
144 1.70 pk if ((fp->f_flag & FREAD) == 0) {
145 1.70 pk simple_unlock(&fp->f_slock);
146 1.39 thorpej return (EBADF);
147 1.70 pk }
148 1.39 thorpej
149 1.45 thorpej FILE_USE(fp);
150 1.45 thorpej
151 1.45 thorpej /* dofileread() will unuse the descriptor for us */
152 1.84 christos return (dofileread(l, fd, fp, SCARG(uap, buf), SCARG(uap, nbyte),
153 1.39 thorpej &fp->f_offset, FOF_UPDATE_OFFSET, retval));
154 1.39 thorpej }
155 1.39 thorpej
156 1.39 thorpej int
157 1.103 ad dofileread(lwp_t *l, int fd, struct file *fp, void *buf, size_t nbyte,
158 1.53 lukem off_t *offset, int flags, register_t *retval)
159 1.53 lukem {
160 1.84 christos struct iovec aiov;
161 1.84 christos struct uio auio;
162 1.103 ad proc_t *p;
163 1.85 yamt struct vmspace *vm;
164 1.84 christos size_t cnt;
165 1.84 christos int error;
166 1.15 cgd #ifdef KTRACE
167 1.101 dsl struct iovec ktriov;
168 1.15 cgd #endif
169 1.84 christos p = l->l_proc;
170 1.85 yamt
171 1.85 yamt error = proc_vmspace_getref(p, &vm);
172 1.85 yamt if (error) {
173 1.85 yamt goto out;
174 1.85 yamt }
175 1.15 cgd
176 1.100 christos aiov.iov_base = (void *)buf;
177 1.39 thorpej aiov.iov_len = nbyte;
178 1.15 cgd auio.uio_iov = &aiov;
179 1.15 cgd auio.uio_iovcnt = 1;
180 1.39 thorpej auio.uio_resid = nbyte;
181 1.15 cgd auio.uio_rw = UIO_READ;
182 1.85 yamt auio.uio_vmspace = vm;
183 1.40 thorpej
184 1.40 thorpej /*
185 1.40 thorpej * Reads return ssize_t because -1 is returned on error. Therefore
186 1.40 thorpej * we must restrict the length to SSIZE_MAX to avoid garbage return
187 1.40 thorpej * values.
188 1.40 thorpej */
189 1.45 thorpej if (auio.uio_resid > SSIZE_MAX) {
190 1.45 thorpej error = EINVAL;
191 1.45 thorpej goto out;
192 1.45 thorpej }
193 1.40 thorpej
194 1.15 cgd #ifdef KTRACE
195 1.101 dsl /* In case we are tracing, save a copy of iovec */
196 1.101 dsl ktriov = aiov;
197 1.15 cgd #endif
198 1.38 thorpej cnt = auio.uio_resid;
199 1.39 thorpej error = (*fp->f_ops->fo_read)(fp, offset, &auio, fp->f_cred, flags);
200 1.22 christos if (error)
201 1.15 cgd if (auio.uio_resid != cnt && (error == ERESTART ||
202 1.15 cgd error == EINTR || error == EWOULDBLOCK))
203 1.15 cgd error = 0;
204 1.15 cgd cnt -= auio.uio_resid;
205 1.15 cgd #ifdef KTRACE
206 1.15 cgd if (KTRPOINT(p, KTR_GENIO) && error == 0)
207 1.84 christos ktrgenio(l, fd, UIO_READ, &ktriov, cnt, error);
208 1.15 cgd #endif
209 1.15 cgd *retval = cnt;
210 1.45 thorpej out:
211 1.84 christos FILE_UNUSE(fp, l);
212 1.85 yamt uvmspace_free(vm);
213 1.15 cgd return (error);
214 1.15 cgd }
215 1.15 cgd
216 1.15 cgd /*
217 1.15 cgd * Scatter read system call.
218 1.15 cgd */
219 1.22 christos int
220 1.103 ad sys_readv(lwp_t *l, void *v, register_t *retval)
221 1.20 thorpej {
222 1.47 augustss struct sys_readv_args /* {
223 1.53 lukem syscallarg(int) fd;
224 1.53 lukem syscallarg(const struct iovec *) iovp;
225 1.53 lukem syscallarg(int) iovcnt;
226 1.20 thorpej } */ *uap = v;
227 1.102 dsl
228 1.102 dsl return do_filereadv(l, SCARG(uap, fd), SCARG(uap, iovp),
229 1.102 dsl SCARG(uap, iovcnt), NULL, FOF_UPDATE_OFFSET, retval);
230 1.102 dsl }
231 1.102 dsl
232 1.102 dsl int
233 1.102 dsl do_filereadv(struct lwp *l, int fd, const struct iovec *iovp, int iovcnt,
234 1.102 dsl off_t *offset, int flags, register_t *retval)
235 1.102 dsl {
236 1.102 dsl struct proc *p;
237 1.102 dsl struct uio auio;
238 1.102 dsl struct iovec *iov, *needfree = NULL, aiov[UIO_SMALLIOV];
239 1.102 dsl struct vmspace *vm;
240 1.102 dsl int i, error;
241 1.102 dsl size_t cnt;
242 1.102 dsl u_int iovlen;
243 1.102 dsl struct file *fp;
244 1.53 lukem struct filedesc *fdp;
245 1.102 dsl #ifdef KTRACE
246 1.102 dsl struct iovec *ktriov = NULL;
247 1.102 dsl #endif
248 1.102 dsl
249 1.102 dsl if (iovcnt == 0)
250 1.102 dsl return EINVAL;
251 1.39 thorpej
252 1.69 thorpej p = l->l_proc;
253 1.53 lukem fdp = p->p_fd;
254 1.56 thorpej
255 1.56 thorpej if ((fp = fd_getfile(fdp, fd)) == NULL)
256 1.102 dsl return EBADF;
257 1.56 thorpej
258 1.70 pk if ((fp->f_flag & FREAD) == 0) {
259 1.70 pk simple_unlock(&fp->f_slock);
260 1.102 dsl return EBADF;
261 1.70 pk }
262 1.39 thorpej
263 1.45 thorpej FILE_USE(fp);
264 1.45 thorpej
265 1.102 dsl if (offset == NULL)
266 1.102 dsl offset = &fp->f_offset;
267 1.102 dsl else {
268 1.102 dsl struct vnode *vp = fp->f_data;
269 1.102 dsl if (fp->f_type != DTYPE_VNODE || vp->v_type == VFIFO) {
270 1.102 dsl error = ESPIPE;
271 1.102 dsl goto out;
272 1.102 dsl }
273 1.102 dsl /*
274 1.102 dsl * Test that the device is seekable ?
275 1.102 dsl * XXX This works because no file systems actually
276 1.102 dsl * XXX take any action on the seek operation.
277 1.102 dsl */
278 1.102 dsl error = VOP_SEEK(vp, fp->f_offset, *offset, fp->f_cred);
279 1.102 dsl if (error != 0)
280 1.102 dsl goto out;
281 1.102 dsl }
282 1.15 cgd
283 1.85 yamt error = proc_vmspace_getref(p, &vm);
284 1.102 dsl if (error)
285 1.85 yamt goto out;
286 1.85 yamt
287 1.42 perry iovlen = iovcnt * sizeof(struct iovec);
288 1.102 dsl if (flags & FOF_IOV_SYSSPACE)
289 1.102 dsl iov = __UNCONST(iovp);
290 1.102 dsl else {
291 1.102 dsl iov = aiov;
292 1.102 dsl if ((u_int)iovcnt > UIO_SMALLIOV) {
293 1.102 dsl if ((u_int)iovcnt > IOV_MAX) {
294 1.102 dsl error = EINVAL;
295 1.102 dsl goto out;
296 1.102 dsl }
297 1.103 ad iov = kmem_alloc(iovlen, KM_SLEEP);
298 1.103 ad if (iov == NULL) {
299 1.103 ad error = ENOMEM;
300 1.103 ad goto out;
301 1.103 ad }
302 1.102 dsl needfree = iov;
303 1.45 thorpej }
304 1.102 dsl error = copyin(iovp, iov, iovlen);
305 1.102 dsl if (error)
306 1.102 dsl goto done;
307 1.45 thorpej }
308 1.41 kleink
309 1.15 cgd auio.uio_iov = iov;
310 1.34 mycroft auio.uio_iovcnt = iovcnt;
311 1.15 cgd auio.uio_rw = UIO_READ;
312 1.85 yamt auio.uio_vmspace = vm;
313 1.102 dsl
314 1.15 cgd auio.uio_resid = 0;
315 1.102 dsl for (i = 0; i < iovcnt; i++, iov++) {
316 1.15 cgd auio.uio_resid += iov->iov_len;
317 1.40 thorpej /*
318 1.40 thorpej * Reads return ssize_t because -1 is returned on error.
319 1.40 thorpej * Therefore we must restrict the length to SSIZE_MAX to
320 1.40 thorpej * avoid garbage return values.
321 1.40 thorpej */
322 1.40 thorpej if (iov->iov_len > SSIZE_MAX || auio.uio_resid > SSIZE_MAX) {
323 1.15 cgd error = EINVAL;
324 1.15 cgd goto done;
325 1.15 cgd }
326 1.15 cgd }
327 1.102 dsl
328 1.15 cgd #ifdef KTRACE
329 1.15 cgd /*
330 1.15 cgd * if tracing, save a copy of iovec
331 1.15 cgd */
332 1.15 cgd if (KTRPOINT(p, KTR_GENIO)) {
333 1.103 ad ktriov = kmem_alloc(iovlen, KM_SLEEP);
334 1.103 ad if (ktriov != NULL)
335 1.103 ad memcpy(ktriov, auio.uio_iov, iovlen);
336 1.15 cgd }
337 1.15 cgd #endif
338 1.102 dsl
339 1.15 cgd cnt = auio.uio_resid;
340 1.39 thorpej error = (*fp->f_ops->fo_read)(fp, offset, &auio, fp->f_cred, flags);
341 1.22 christos if (error)
342 1.15 cgd if (auio.uio_resid != cnt && (error == ERESTART ||
343 1.15 cgd error == EINTR || error == EWOULDBLOCK))
344 1.15 cgd error = 0;
345 1.15 cgd cnt -= auio.uio_resid;
346 1.102 dsl *retval = cnt;
347 1.102 dsl
348 1.15 cgd #ifdef KTRACE
349 1.58 itohy if (ktriov != NULL) {
350 1.78 drochner if (KTRPOINT(p, KTR_GENIO) && (error == 0))
351 1.84 christos ktrgenio(l, fd, UIO_READ, ktriov, cnt, error);
352 1.103 ad kmem_free(ktriov, iovlen);
353 1.15 cgd }
354 1.15 cgd #endif
355 1.102 dsl
356 1.45 thorpej done:
357 1.15 cgd if (needfree)
358 1.103 ad kmem_free(needfree, iovlen);
359 1.45 thorpej out:
360 1.84 christos FILE_UNUSE(fp, l);
361 1.85 yamt uvmspace_free(vm);
362 1.15 cgd return (error);
363 1.15 cgd }
364 1.15 cgd
365 1.15 cgd /*
366 1.15 cgd * Write system call
367 1.15 cgd */
368 1.22 christos int
369 1.103 ad sys_write(lwp_t *l, void *v, register_t *retval)
370 1.20 thorpej {
371 1.47 augustss struct sys_write_args /* {
372 1.53 lukem syscallarg(int) fd;
373 1.53 lukem syscallarg(const void *) buf;
374 1.53 lukem syscallarg(size_t) nbyte;
375 1.20 thorpej } */ *uap = v;
376 1.53 lukem int fd;
377 1.53 lukem struct file *fp;
378 1.103 ad proc_t *p;
379 1.53 lukem struct filedesc *fdp;
380 1.39 thorpej
381 1.53 lukem fd = SCARG(uap, fd);
382 1.69 thorpej p = l->l_proc;
383 1.53 lukem fdp = p->p_fd;
384 1.56 thorpej
385 1.56 thorpej if ((fp = fd_getfile(fdp, fd)) == NULL)
386 1.56 thorpej return (EBADF);
387 1.56 thorpej
388 1.70 pk if ((fp->f_flag & FWRITE) == 0) {
389 1.70 pk simple_unlock(&fp->f_slock);
390 1.39 thorpej return (EBADF);
391 1.70 pk }
392 1.39 thorpej
393 1.45 thorpej FILE_USE(fp);
394 1.45 thorpej
395 1.45 thorpej /* dofilewrite() will unuse the descriptor for us */
396 1.84 christos return (dofilewrite(l, fd, fp, SCARG(uap, buf), SCARG(uap, nbyte),
397 1.39 thorpej &fp->f_offset, FOF_UPDATE_OFFSET, retval));
398 1.39 thorpej }
399 1.39 thorpej
400 1.39 thorpej int
401 1.103 ad dofilewrite(lwp_t *l, int fd, struct file *fp, const void *buf,
402 1.53 lukem size_t nbyte, off_t *offset, int flags, register_t *retval)
403 1.53 lukem {
404 1.84 christos struct iovec aiov;
405 1.84 christos struct uio auio;
406 1.103 ad proc_t *p;
407 1.85 yamt struct vmspace *vm;
408 1.84 christos size_t cnt;
409 1.84 christos int error;
410 1.15 cgd #ifdef KTRACE
411 1.101 dsl struct iovec ktriov;
412 1.15 cgd #endif
413 1.15 cgd
414 1.84 christos p = l->l_proc;
415 1.85 yamt error = proc_vmspace_getref(p, &vm);
416 1.85 yamt if (error) {
417 1.85 yamt goto out;
418 1.85 yamt }
419 1.83 christos aiov.iov_base = __UNCONST(buf); /* XXXUNCONST kills const */
420 1.39 thorpej aiov.iov_len = nbyte;
421 1.15 cgd auio.uio_iov = &aiov;
422 1.15 cgd auio.uio_iovcnt = 1;
423 1.39 thorpej auio.uio_resid = nbyte;
424 1.15 cgd auio.uio_rw = UIO_WRITE;
425 1.85 yamt auio.uio_vmspace = vm;
426 1.40 thorpej
427 1.40 thorpej /*
428 1.40 thorpej * Writes return ssize_t because -1 is returned on error. Therefore
429 1.40 thorpej * we must restrict the length to SSIZE_MAX to avoid garbage return
430 1.40 thorpej * values.
431 1.40 thorpej */
432 1.45 thorpej if (auio.uio_resid > SSIZE_MAX) {
433 1.45 thorpej error = EINVAL;
434 1.45 thorpej goto out;
435 1.45 thorpej }
436 1.40 thorpej
437 1.15 cgd #ifdef KTRACE
438 1.101 dsl /* In case we are tracing, save a copy of iovec */
439 1.101 dsl ktriov = aiov;
440 1.15 cgd #endif
441 1.38 thorpej cnt = auio.uio_resid;
442 1.39 thorpej error = (*fp->f_ops->fo_write)(fp, offset, &auio, fp->f_cred, flags);
443 1.22 christos if (error) {
444 1.15 cgd if (auio.uio_resid != cnt && (error == ERESTART ||
445 1.15 cgd error == EINTR || error == EWOULDBLOCK))
446 1.15 cgd error = 0;
447 1.98 ad if (error == EPIPE) {
448 1.98 ad mutex_enter(&proclist_mutex);
449 1.15 cgd psignal(p, SIGPIPE);
450 1.98 ad mutex_exit(&proclist_mutex);
451 1.98 ad }
452 1.15 cgd }
453 1.15 cgd cnt -= auio.uio_resid;
454 1.15 cgd #ifdef KTRACE
455 1.15 cgd if (KTRPOINT(p, KTR_GENIO) && error == 0)
456 1.84 christos ktrgenio(l, fd, UIO_WRITE, &ktriov, cnt, error);
457 1.15 cgd #endif
458 1.15 cgd *retval = cnt;
459 1.45 thorpej out:
460 1.84 christos FILE_UNUSE(fp, l);
461 1.85 yamt uvmspace_free(vm);
462 1.15 cgd return (error);
463 1.15 cgd }
464 1.15 cgd
465 1.15 cgd /*
466 1.15 cgd * Gather write system call
467 1.15 cgd */
468 1.22 christos int
469 1.103 ad sys_writev(lwp_t *l, void *v, register_t *retval)
470 1.20 thorpej {
471 1.47 augustss struct sys_writev_args /* {
472 1.53 lukem syscallarg(int) fd;
473 1.53 lukem syscallarg(const struct iovec *) iovp;
474 1.53 lukem syscallarg(int) iovcnt;
475 1.20 thorpej } */ *uap = v;
476 1.102 dsl
477 1.102 dsl return do_filewritev(l, SCARG(uap, fd), SCARG(uap, iovp),
478 1.102 dsl SCARG(uap, iovcnt), NULL, FOF_UPDATE_OFFSET, retval);
479 1.102 dsl }
480 1.102 dsl
481 1.102 dsl int
482 1.102 dsl do_filewritev(struct lwp *l, int fd, const struct iovec *iovp, int iovcnt,
483 1.102 dsl off_t *offset, int flags, register_t *retval)
484 1.102 dsl {
485 1.102 dsl struct proc *p;
486 1.102 dsl struct uio auio;
487 1.102 dsl struct iovec *iov, *needfree = NULL, aiov[UIO_SMALLIOV];
488 1.102 dsl struct vmspace *vm;
489 1.102 dsl int i, error;
490 1.102 dsl size_t cnt;
491 1.102 dsl u_int iovlen;
492 1.53 lukem struct file *fp;
493 1.53 lukem struct filedesc *fdp;
494 1.102 dsl #ifdef KTRACE
495 1.102 dsl struct iovec *ktriov = NULL;
496 1.102 dsl #endif
497 1.102 dsl
498 1.102 dsl if (iovcnt == 0)
499 1.102 dsl return EINVAL;
500 1.39 thorpej
501 1.69 thorpej p = l->l_proc;
502 1.53 lukem fdp = p->p_fd;
503 1.56 thorpej
504 1.56 thorpej if ((fp = fd_getfile(fdp, fd)) == NULL)
505 1.102 dsl return EBADF;
506 1.56 thorpej
507 1.70 pk if ((fp->f_flag & FWRITE) == 0) {
508 1.70 pk simple_unlock(&fp->f_slock);
509 1.102 dsl return EBADF;
510 1.70 pk }
511 1.39 thorpej
512 1.45 thorpej FILE_USE(fp);
513 1.45 thorpej
514 1.102 dsl if (offset == NULL)
515 1.102 dsl offset = &fp->f_offset;
516 1.102 dsl else {
517 1.102 dsl struct vnode *vp = fp->f_data;
518 1.102 dsl if (fp->f_type != DTYPE_VNODE || vp->v_type == VFIFO) {
519 1.102 dsl error = ESPIPE;
520 1.102 dsl goto out;
521 1.102 dsl }
522 1.102 dsl /*
523 1.102 dsl * Test that the device is seekable ?
524 1.102 dsl * XXX This works because no file systems actually
525 1.102 dsl * XXX take any action on the seek operation.
526 1.102 dsl */
527 1.102 dsl error = VOP_SEEK(vp, fp->f_offset, *offset, fp->f_cred);
528 1.102 dsl if (error != 0)
529 1.102 dsl goto out;
530 1.102 dsl }
531 1.39 thorpej
532 1.85 yamt error = proc_vmspace_getref(p, &vm);
533 1.102 dsl if (error)
534 1.85 yamt goto out;
535 1.102 dsl
536 1.42 perry iovlen = iovcnt * sizeof(struct iovec);
537 1.102 dsl if (flags & FOF_IOV_SYSSPACE)
538 1.102 dsl iov = __UNCONST(iovp);
539 1.102 dsl else {
540 1.102 dsl iov = aiov;
541 1.102 dsl if ((u_int)iovcnt > UIO_SMALLIOV) {
542 1.102 dsl if ((u_int)iovcnt > IOV_MAX) {
543 1.102 dsl error = EINVAL;
544 1.102 dsl goto out;
545 1.102 dsl }
546 1.103 ad iov = kmem_alloc(iovlen, KM_SLEEP);
547 1.103 ad if (iov == NULL) {
548 1.103 ad error = ENOMEM;
549 1.103 ad goto out;
550 1.103 ad }
551 1.102 dsl needfree = iov;
552 1.62 jdolecek }
553 1.102 dsl error = copyin(iovp, iov, iovlen);
554 1.102 dsl if (error)
555 1.102 dsl goto done;
556 1.45 thorpej }
557 1.41 kleink
558 1.15 cgd auio.uio_iov = iov;
559 1.34 mycroft auio.uio_iovcnt = iovcnt;
560 1.15 cgd auio.uio_rw = UIO_WRITE;
561 1.85 yamt auio.uio_vmspace = vm;
562 1.102 dsl
563 1.15 cgd auio.uio_resid = 0;
564 1.102 dsl for (i = 0; i < iovcnt; i++, iov++) {
565 1.15 cgd auio.uio_resid += iov->iov_len;
566 1.40 thorpej /*
567 1.40 thorpej * Writes return ssize_t because -1 is returned on error.
568 1.40 thorpej * Therefore we must restrict the length to SSIZE_MAX to
569 1.40 thorpej * avoid garbage return values.
570 1.40 thorpej */
571 1.40 thorpej if (iov->iov_len > SSIZE_MAX || auio.uio_resid > SSIZE_MAX) {
572 1.15 cgd error = EINVAL;
573 1.15 cgd goto done;
574 1.15 cgd }
575 1.15 cgd }
576 1.102 dsl
577 1.15 cgd #ifdef KTRACE
578 1.15 cgd /*
579 1.15 cgd * if tracing, save a copy of iovec
580 1.15 cgd */
581 1.15 cgd if (KTRPOINT(p, KTR_GENIO)) {
582 1.103 ad ktriov = kmem_alloc(iovlen, KM_SLEEP);
583 1.103 ad if (ktriov != NULL)
584 1.103 ad memcpy(ktriov, auio.uio_iov, iovlen);
585 1.15 cgd }
586 1.15 cgd #endif
587 1.15 cgd cnt = auio.uio_resid;
588 1.39 thorpej error = (*fp->f_ops->fo_write)(fp, offset, &auio, fp->f_cred, flags);
589 1.22 christos if (error) {
590 1.15 cgd if (auio.uio_resid != cnt && (error == ERESTART ||
591 1.15 cgd error == EINTR || error == EWOULDBLOCK))
592 1.15 cgd error = 0;
593 1.98 ad if (error == EPIPE) {
594 1.98 ad mutex_enter(&proclist_mutex);
595 1.15 cgd psignal(p, SIGPIPE);
596 1.98 ad mutex_exit(&proclist_mutex);
597 1.98 ad }
598 1.15 cgd }
599 1.15 cgd cnt -= auio.uio_resid;
600 1.102 dsl *retval = cnt;
601 1.102 dsl
602 1.15 cgd #ifdef KTRACE
603 1.78 drochner if (ktriov != NULL) {
604 1.78 drochner if (KTRPOINT(p, KTR_GENIO) && (error == 0))
605 1.84 christos ktrgenio(l, fd, UIO_WRITE, ktriov, cnt, error);
606 1.103 ad kmem_free(ktriov, iovlen);
607 1.15 cgd }
608 1.15 cgd #endif
609 1.102 dsl
610 1.45 thorpej done:
611 1.15 cgd if (needfree)
612 1.103 ad kmem_free(needfree, iovlen);
613 1.45 thorpej out:
614 1.84 christos FILE_UNUSE(fp, l);
615 1.85 yamt uvmspace_free(vm);
616 1.15 cgd return (error);
617 1.15 cgd }
618 1.15 cgd
619 1.15 cgd /*
620 1.15 cgd * Ioctl system call
621 1.15 cgd */
622 1.15 cgd /* ARGSUSED */
623 1.22 christos int
624 1.96 yamt sys_ioctl(struct lwp *l, void *v, register_t *retval)
625 1.20 thorpej {
626 1.47 augustss struct sys_ioctl_args /* {
627 1.53 lukem syscallarg(int) fd;
628 1.53 lukem syscallarg(u_long) com;
629 1.100 christos syscallarg(void *) data;
630 1.20 thorpej } */ *uap = v;
631 1.53 lukem struct file *fp;
632 1.103 ad proc_t *p;
633 1.53 lukem struct filedesc *fdp;
634 1.53 lukem u_long com;
635 1.53 lukem int error;
636 1.53 lukem u_int size;
637 1.100 christos void *data, *memp;
638 1.53 lukem #define STK_PARAMS 128
639 1.53 lukem u_long stkbuf[STK_PARAMS/sizeof(u_long)];
640 1.15 cgd
641 1.53 lukem error = 0;
642 1.69 thorpej p = l->l_proc;
643 1.15 cgd fdp = p->p_fd;
644 1.56 thorpej
645 1.56 thorpej if ((fp = fd_getfile(fdp, SCARG(uap, fd))) == NULL)
646 1.15 cgd return (EBADF);
647 1.15 cgd
648 1.45 thorpej FILE_USE(fp);
649 1.45 thorpej
650 1.45 thorpej if ((fp->f_flag & (FREAD | FWRITE)) == 0) {
651 1.45 thorpej error = EBADF;
652 1.65 scw com = 0;
653 1.45 thorpej goto out;
654 1.45 thorpej }
655 1.15 cgd
656 1.16 cgd switch (com = SCARG(uap, com)) {
657 1.15 cgd case FIONCLEX:
658 1.16 cgd fdp->fd_ofileflags[SCARG(uap, fd)] &= ~UF_EXCLOSE;
659 1.45 thorpej goto out;
660 1.45 thorpej
661 1.15 cgd case FIOCLEX:
662 1.16 cgd fdp->fd_ofileflags[SCARG(uap, fd)] |= UF_EXCLOSE;
663 1.45 thorpej goto out;
664 1.15 cgd }
665 1.15 cgd
666 1.15 cgd /*
667 1.15 cgd * Interpret high order word to find amount of data to be
668 1.15 cgd * copied to/from the user's address space.
669 1.15 cgd */
670 1.15 cgd size = IOCPARM_LEN(com);
671 1.45 thorpej if (size > IOCPARM_MAX) {
672 1.45 thorpej error = ENOTTY;
673 1.45 thorpej goto out;
674 1.45 thorpej }
675 1.15 cgd memp = NULL;
676 1.42 perry if (size > sizeof(stkbuf)) {
677 1.103 ad memp = kmem_alloc(size, KM_SLEEP);
678 1.15 cgd data = memp;
679 1.15 cgd } else
680 1.100 christos data = (void *)stkbuf;
681 1.15 cgd if (com&IOC_IN) {
682 1.15 cgd if (size) {
683 1.31 cgd error = copyin(SCARG(uap, data), data, size);
684 1.15 cgd if (error) {
685 1.15 cgd if (memp)
686 1.103 ad kmem_free(memp, size);
687 1.45 thorpej goto out;
688 1.15 cgd }
689 1.73 dsl #ifdef KTRACE
690 1.73 dsl if (KTRPOINT(p, KTR_GENIO)) {
691 1.73 dsl struct iovec iov;
692 1.73 dsl iov.iov_base = SCARG(uap, data);
693 1.73 dsl iov.iov_len = size;
694 1.84 christos ktrgenio(l, SCARG(uap, fd), UIO_WRITE, &iov,
695 1.73 dsl size, 0);
696 1.73 dsl }
697 1.73 dsl #endif
698 1.15 cgd } else
699 1.100 christos *(void **)data = SCARG(uap, data);
700 1.15 cgd } else if ((com&IOC_OUT) && size)
701 1.15 cgd /*
702 1.15 cgd * Zero the buffer so the user always
703 1.15 cgd * gets back something deterministic.
704 1.15 cgd */
705 1.44 perry memset(data, 0, size);
706 1.15 cgd else if (com&IOC_VOID)
707 1.100 christos *(void **)data = SCARG(uap, data);
708 1.15 cgd
709 1.15 cgd switch (com) {
710 1.15 cgd
711 1.15 cgd case FIONBIO:
712 1.79 jdolecek if (*(int *)data != 0)
713 1.15 cgd fp->f_flag |= FNONBLOCK;
714 1.15 cgd else
715 1.15 cgd fp->f_flag &= ~FNONBLOCK;
716 1.84 christos error = (*fp->f_ops->fo_ioctl)(fp, FIONBIO, data, l);
717 1.15 cgd break;
718 1.15 cgd
719 1.15 cgd case FIOASYNC:
720 1.79 jdolecek if (*(int *)data != 0)
721 1.15 cgd fp->f_flag |= FASYNC;
722 1.15 cgd else
723 1.15 cgd fp->f_flag &= ~FASYNC;
724 1.84 christos error = (*fp->f_ops->fo_ioctl)(fp, FIOASYNC, data, l);
725 1.15 cgd break;
726 1.15 cgd
727 1.15 cgd default:
728 1.84 christos error = (*fp->f_ops->fo_ioctl)(fp, com, data, l);
729 1.15 cgd /*
730 1.15 cgd * Copy any data to user, size was
731 1.15 cgd * already set and checked above.
732 1.15 cgd */
733 1.73 dsl if (error == 0 && (com&IOC_OUT) && size) {
734 1.31 cgd error = copyout(data, SCARG(uap, data), size);
735 1.73 dsl #ifdef KTRACE
736 1.73 dsl if (KTRPOINT(p, KTR_GENIO)) {
737 1.73 dsl struct iovec iov;
738 1.73 dsl iov.iov_base = SCARG(uap, data);
739 1.73 dsl iov.iov_len = size;
740 1.84 christos ktrgenio(l, SCARG(uap, fd), UIO_READ, &iov,
741 1.73 dsl size, error);
742 1.73 dsl }
743 1.73 dsl #endif
744 1.73 dsl }
745 1.15 cgd break;
746 1.15 cgd }
747 1.15 cgd if (memp)
748 1.103 ad kmem_free(memp, size);
749 1.45 thorpej out:
750 1.84 christos FILE_UNUSE(fp, l);
751 1.61 atatat switch (error) {
752 1.61 atatat case -1:
753 1.61 atatat printf("sys_ioctl: _IO%s%s('%c', %lu, %lu) returned -1: "
754 1.61 atatat "pid=%d comm=%s\n",
755 1.61 atatat (com & IOC_IN) ? "W" : "", (com & IOC_OUT) ? "R" : "",
756 1.61 atatat (char)IOCGROUP(com), (com & 0xff), IOCPARM_LEN(com),
757 1.61 atatat p->p_pid, p->p_comm);
758 1.61 atatat /* FALLTHROUGH */
759 1.61 atatat case EPASSTHROUGH:
760 1.61 atatat error = ENOTTY;
761 1.61 atatat /* FALLTHROUGH */
762 1.61 atatat default:
763 1.61 atatat return (error);
764 1.61 atatat }
765 1.15 cgd }
766 1.15 cgd
767 1.15 cgd /*
768 1.15 cgd * Select system call.
769 1.15 cgd */
770 1.22 christos int
771 1.103 ad sys_pselect(lwp_t *l, void *v, register_t *retval)
772 1.82 matt {
773 1.82 matt struct sys_pselect_args /* {
774 1.82 matt syscallarg(int) nd;
775 1.82 matt syscallarg(fd_set *) in;
776 1.82 matt syscallarg(fd_set *) ou;
777 1.82 matt syscallarg(fd_set *) ex;
778 1.82 matt syscallarg(const struct timespec *) ts;
779 1.82 matt syscallarg(sigset_t *) mask;
780 1.82 matt } */ * const uap = v;
781 1.82 matt struct timespec ats;
782 1.82 matt struct timeval atv, *tv = NULL;
783 1.82 matt sigset_t amask, *mask = NULL;
784 1.82 matt int error;
785 1.82 matt
786 1.82 matt if (SCARG(uap, ts)) {
787 1.82 matt error = copyin(SCARG(uap, ts), &ats, sizeof(ats));
788 1.82 matt if (error)
789 1.82 matt return error;
790 1.82 matt atv.tv_sec = ats.tv_sec;
791 1.82 matt atv.tv_usec = ats.tv_nsec / 1000;
792 1.82 matt tv = &atv;
793 1.82 matt }
794 1.82 matt if (SCARG(uap, mask) != NULL) {
795 1.82 matt error = copyin(SCARG(uap, mask), &amask, sizeof(amask));
796 1.82 matt if (error)
797 1.82 matt return error;
798 1.82 matt mask = &amask;
799 1.82 matt }
800 1.82 matt
801 1.82 matt return selcommon(l, retval, SCARG(uap, nd), SCARG(uap, in),
802 1.82 matt SCARG(uap, ou), SCARG(uap, ex), tv, mask);
803 1.82 matt }
804 1.82 matt
805 1.91 kardel int
806 1.90 christos inittimeleft(struct timeval *tv, struct timeval *sleeptv)
807 1.90 christos {
808 1.90 christos if (itimerfix(tv))
809 1.90 christos return -1;
810 1.90 christos getmicrouptime(sleeptv);
811 1.90 christos return 0;
812 1.90 christos }
813 1.90 christos
814 1.91 kardel int
815 1.90 christos gettimeleft(struct timeval *tv, struct timeval *sleeptv)
816 1.90 christos {
817 1.90 christos /*
818 1.90 christos * We have to recalculate the timeout on every retry.
819 1.90 christos */
820 1.90 christos struct timeval slepttv;
821 1.90 christos /*
822 1.90 christos * reduce tv by elapsed time
823 1.90 christos * based on monotonic time scale
824 1.90 christos */
825 1.90 christos getmicrouptime(&slepttv);
826 1.90 christos timeradd(tv, sleeptv, tv);
827 1.90 christos timersub(tv, &slepttv, tv);
828 1.90 christos *sleeptv = slepttv;
829 1.90 christos return tvtohz(tv);
830 1.90 christos }
831 1.90 christos
832 1.82 matt int
833 1.103 ad sys_select(lwp_t *l, void *v, register_t *retval)
834 1.20 thorpej {
835 1.47 augustss struct sys_select_args /* {
836 1.53 lukem syscallarg(int) nd;
837 1.53 lukem syscallarg(fd_set *) in;
838 1.53 lukem syscallarg(fd_set *) ou;
839 1.53 lukem syscallarg(fd_set *) ex;
840 1.53 lukem syscallarg(struct timeval *) tv;
841 1.82 matt } */ * const uap = v;
842 1.82 matt struct timeval atv, *tv = NULL;
843 1.82 matt int error;
844 1.82 matt
845 1.82 matt if (SCARG(uap, tv)) {
846 1.100 christos error = copyin(SCARG(uap, tv), (void *)&atv,
847 1.82 matt sizeof(atv));
848 1.82 matt if (error)
849 1.82 matt return error;
850 1.82 matt tv = &atv;
851 1.82 matt }
852 1.82 matt
853 1.82 matt return selcommon(l, retval, SCARG(uap, nd), SCARG(uap, in),
854 1.82 matt SCARG(uap, ou), SCARG(uap, ex), tv, NULL);
855 1.82 matt }
856 1.82 matt
857 1.82 matt int
858 1.103 ad selcommon(lwp_t *l, register_t *retval, int nd, fd_set *u_in,
859 1.103 ad fd_set *u_ou, fd_set *u_ex, struct timeval *tv, sigset_t *mask)
860 1.82 matt {
861 1.86 kardel char smallbits[howmany(FD_SETSIZE, NFDBITS) *
862 1.86 kardel sizeof(fd_mask) * 6];
863 1.103 ad proc_t * const p = l->l_proc;
864 1.100 christos char *bits;
865 1.103 ad int ncoll, error, timo;
866 1.53 lukem size_t ni;
867 1.82 matt sigset_t oldmask;
868 1.89 christos struct timeval sleeptv;
869 1.15 cgd
870 1.53 lukem error = 0;
871 1.82 matt if (nd < 0)
872 1.35 thorpej return (EINVAL);
873 1.82 matt if (nd > p->p_fd->fd_nfiles) {
874 1.16 cgd /* forgiving; slightly wrong */
875 1.82 matt nd = p->p_fd->fd_nfiles;
876 1.16 cgd }
877 1.82 matt ni = howmany(nd, NFDBITS) * sizeof(fd_mask);
878 1.27 mycroft if (ni * 6 > sizeof(smallbits))
879 1.103 ad bits = kmem_alloc(ni * 6, KM_SLEEP);
880 1.25 mycroft else
881 1.26 cgd bits = smallbits;
882 1.15 cgd
883 1.53 lukem #define getbits(name, x) \
884 1.82 matt if (u_ ## name) { \
885 1.82 matt error = copyin(u_ ## name, bits + ni * x, ni); \
886 1.53 lukem if (error) \
887 1.53 lukem goto done; \
888 1.53 lukem } else \
889 1.44 perry memset(bits + ni * x, 0, ni);
890 1.15 cgd getbits(in, 0);
891 1.15 cgd getbits(ou, 1);
892 1.15 cgd getbits(ex, 2);
893 1.15 cgd #undef getbits
894 1.15 cgd
895 1.65 scw timo = 0;
896 1.90 christos if (tv && inittimeleft(tv, &sleeptv) == -1) {
897 1.90 christos error = EINVAL;
898 1.90 christos goto done;
899 1.65 scw }
900 1.89 christos
901 1.98 ad if (mask) {
902 1.98 ad sigminusset(&sigcantmask, mask);
903 1.98 ad mutex_enter(&p->p_smutex);
904 1.98 ad oldmask = l->l_sigmask;
905 1.98 ad l->l_sigmask = *mask;
906 1.98 ad mutex_exit(&p->p_smutex);
907 1.98 ad } else
908 1.98 ad oldmask = l->l_sigmask; /* XXXgcc */
909 1.65 scw
910 1.103 ad mutex_enter(&select_lock);
911 1.103 ad SLIST_INIT(&l->l_selwait);
912 1.103 ad for (;;) {
913 1.103 ad l->l_selflag = SEL_SCANNING;
914 1.103 ad ncoll = nselcoll;
915 1.103 ad mutex_exit(&select_lock);
916 1.103 ad
917 1.103 ad error = selscan(l, (fd_mask *)(bits + ni * 0),
918 1.103 ad (fd_mask *)(bits + ni * 3), nd, retval);
919 1.103 ad
920 1.103 ad mutex_enter(&select_lock);
921 1.103 ad if (error || *retval)
922 1.103 ad break;
923 1.103 ad if (tv && (timo = gettimeleft(tv, &sleeptv)) <= 0)
924 1.103 ad break;
925 1.103 ad if (l->l_selflag != SEL_SCANNING || ncoll != nselcoll)
926 1.103 ad continue;
927 1.103 ad l->l_selflag = SEL_BLOCKING;
928 1.103 ad error = cv_timedwait_sig(&select_cv, &select_lock, timo);
929 1.103 ad if (error != 0)
930 1.103 ad break;
931 1.103 ad }
932 1.103 ad selclear();
933 1.103 ad mutex_exit(&select_lock);
934 1.103 ad
935 1.98 ad if (mask) {
936 1.98 ad mutex_enter(&p->p_smutex);
937 1.98 ad l->l_sigmask = oldmask;
938 1.98 ad mutex_exit(&p->p_smutex);
939 1.98 ad }
940 1.103 ad
941 1.97 ad done:
942 1.15 cgd /* select is not restarted after signals... */
943 1.15 cgd if (error == ERESTART)
944 1.15 cgd error = EINTR;
945 1.15 cgd if (error == EWOULDBLOCK)
946 1.15 cgd error = 0;
947 1.103 ad if (error == 0 && u_in != NULL)
948 1.103 ad error = copyout(bits + ni * 3, u_in, ni);
949 1.103 ad if (error == 0 && u_ou != NULL)
950 1.103 ad error = copyout(bits + ni * 4, u_ou, ni);
951 1.103 ad if (error == 0 && u_ex != NULL)
952 1.103 ad error = copyout(bits + ni * 5, u_ex, ni);
953 1.103 ad if (bits != smallbits)
954 1.103 ad kmem_free(bits, ni * 6);
955 1.15 cgd return (error);
956 1.15 cgd }
957 1.15 cgd
958 1.22 christos int
959 1.103 ad selscan(lwp_t *l, fd_mask *ibitp, fd_mask *obitp, int nfd,
960 1.53 lukem register_t *retval)
961 1.53 lukem {
962 1.63 jdolecek static const int flag[3] = { POLLRDNORM | POLLHUP | POLLERR,
963 1.28 mycroft POLLWRNORM | POLLHUP | POLLERR,
964 1.28 mycroft POLLRDBAND };
965 1.103 ad proc_t *p = l->l_proc;
966 1.84 christos struct filedesc *fdp;
967 1.84 christos int msk, i, j, fd, n;
968 1.84 christos fd_mask ibits, obits;
969 1.84 christos struct file *fp;
970 1.15 cgd
971 1.53 lukem fdp = p->p_fd;
972 1.53 lukem n = 0;
973 1.15 cgd for (msk = 0; msk < 3; msk++) {
974 1.15 cgd for (i = 0; i < nfd; i += NFDBITS) {
975 1.25 mycroft ibits = *ibitp++;
976 1.25 mycroft obits = 0;
977 1.25 mycroft while ((j = ffs(ibits)) && (fd = i + --j) < nfd) {
978 1.25 mycroft ibits &= ~(1 << j);
979 1.56 thorpej if ((fp = fd_getfile(fdp, fd)) == NULL)
980 1.15 cgd return (EBADF);
981 1.45 thorpej FILE_USE(fp);
982 1.84 christos if ((*fp->f_ops->fo_poll)(fp, flag[msk], l)) {
983 1.25 mycroft obits |= (1 << j);
984 1.15 cgd n++;
985 1.15 cgd }
986 1.84 christos FILE_UNUSE(fp, l);
987 1.15 cgd }
988 1.25 mycroft *obitp++ = obits;
989 1.15 cgd }
990 1.15 cgd }
991 1.15 cgd *retval = n;
992 1.15 cgd return (0);
993 1.15 cgd }
994 1.15 cgd
995 1.28 mycroft /*
996 1.28 mycroft * Poll system call.
997 1.28 mycroft */
998 1.28 mycroft int
999 1.103 ad sys_poll(lwp_t *l, void *v, register_t *retval)
1000 1.28 mycroft {
1001 1.47 augustss struct sys_poll_args /* {
1002 1.53 lukem syscallarg(struct pollfd *) fds;
1003 1.53 lukem syscallarg(u_int) nfds;
1004 1.53 lukem syscallarg(int) timeout;
1005 1.82 matt } */ * const uap = v;
1006 1.82 matt struct timeval atv, *tv = NULL;
1007 1.82 matt
1008 1.82 matt if (SCARG(uap, timeout) != INFTIM) {
1009 1.82 matt atv.tv_sec = SCARG(uap, timeout) / 1000;
1010 1.82 matt atv.tv_usec = (SCARG(uap, timeout) % 1000) * 1000;
1011 1.82 matt tv = &atv;
1012 1.82 matt }
1013 1.82 matt
1014 1.82 matt return pollcommon(l, retval, SCARG(uap, fds), SCARG(uap, nfds),
1015 1.82 matt tv, NULL);
1016 1.82 matt }
1017 1.82 matt
1018 1.82 matt /*
1019 1.82 matt * Poll system call.
1020 1.82 matt */
1021 1.82 matt int
1022 1.103 ad sys_pollts(lwp_t *l, void *v, register_t *retval)
1023 1.82 matt {
1024 1.82 matt struct sys_pollts_args /* {
1025 1.82 matt syscallarg(struct pollfd *) fds;
1026 1.82 matt syscallarg(u_int) nfds;
1027 1.82 matt syscallarg(const struct timespec *) ts;
1028 1.82 matt syscallarg(const sigset_t *) mask;
1029 1.82 matt } */ * const uap = v;
1030 1.82 matt struct timespec ats;
1031 1.82 matt struct timeval atv, *tv = NULL;
1032 1.82 matt sigset_t amask, *mask = NULL;
1033 1.82 matt int error;
1034 1.82 matt
1035 1.82 matt if (SCARG(uap, ts)) {
1036 1.82 matt error = copyin(SCARG(uap, ts), &ats, sizeof(ats));
1037 1.82 matt if (error)
1038 1.82 matt return error;
1039 1.82 matt atv.tv_sec = ats.tv_sec;
1040 1.82 matt atv.tv_usec = ats.tv_nsec / 1000;
1041 1.82 matt tv = &atv;
1042 1.82 matt }
1043 1.82 matt if (SCARG(uap, mask)) {
1044 1.82 matt error = copyin(SCARG(uap, mask), &amask, sizeof(amask));
1045 1.82 matt if (error)
1046 1.82 matt return error;
1047 1.82 matt mask = &amask;
1048 1.82 matt }
1049 1.82 matt
1050 1.82 matt return pollcommon(l, retval, SCARG(uap, fds), SCARG(uap, nfds),
1051 1.82 matt tv, mask);
1052 1.82 matt }
1053 1.82 matt
1054 1.82 matt int
1055 1.103 ad pollcommon(lwp_t *l, register_t *retval,
1056 1.82 matt struct pollfd *u_fds, u_int nfds,
1057 1.82 matt struct timeval *tv, sigset_t *mask)
1058 1.82 matt {
1059 1.86 kardel char smallbits[32 * sizeof(struct pollfd)];
1060 1.103 ad proc_t * const p = l->l_proc;
1061 1.100 christos void * bits;
1062 1.82 matt sigset_t oldmask;
1063 1.103 ad int ncoll, error, timo;
1064 1.53 lukem size_t ni;
1065 1.89 christos struct timeval sleeptv;
1066 1.28 mycroft
1067 1.82 matt if (nfds > p->p_fd->fd_nfiles) {
1068 1.28 mycroft /* forgiving; slightly wrong */
1069 1.82 matt nfds = p->p_fd->fd_nfiles;
1070 1.28 mycroft }
1071 1.82 matt ni = nfds * sizeof(struct pollfd);
1072 1.28 mycroft if (ni > sizeof(smallbits))
1073 1.103 ad bits = kmem_alloc(ni, KM_SLEEP);
1074 1.28 mycroft else
1075 1.28 mycroft bits = smallbits;
1076 1.28 mycroft
1077 1.82 matt error = copyin(u_fds, bits, ni);
1078 1.28 mycroft if (error)
1079 1.28 mycroft goto done;
1080 1.28 mycroft
1081 1.65 scw timo = 0;
1082 1.90 christos if (tv && inittimeleft(tv, &sleeptv) == -1) {
1083 1.90 christos error = EINVAL;
1084 1.90 christos goto done;
1085 1.65 scw }
1086 1.89 christos
1087 1.98 ad if (mask) {
1088 1.98 ad sigminusset(&sigcantmask, mask);
1089 1.98 ad mutex_enter(&p->p_smutex);
1090 1.98 ad oldmask = l->l_sigmask;
1091 1.98 ad l->l_sigmask = *mask;
1092 1.98 ad mutex_exit(&p->p_smutex);
1093 1.98 ad } else
1094 1.98 ad oldmask = l->l_sigmask; /* XXXgcc */
1095 1.65 scw
1096 1.103 ad mutex_enter(&select_lock);
1097 1.103 ad SLIST_INIT(&l->l_selwait);
1098 1.103 ad for (;;) {
1099 1.103 ad ncoll = nselcoll;
1100 1.103 ad l->l_selflag = SEL_SCANNING;
1101 1.103 ad mutex_exit(&select_lock);
1102 1.103 ad
1103 1.103 ad error = pollscan(l, (struct pollfd *)bits, nfds, retval);
1104 1.103 ad
1105 1.103 ad mutex_enter(&select_lock);
1106 1.103 ad if (error || *retval)
1107 1.103 ad break;
1108 1.103 ad if (tv && (timo = gettimeleft(tv, &sleeptv)) <= 0)
1109 1.103 ad break;
1110 1.103 ad if (l->l_selflag != SEL_SCANNING || nselcoll != ncoll)
1111 1.103 ad continue;
1112 1.103 ad l->l_selflag = SEL_BLOCKING;
1113 1.103 ad error = cv_timedwait_sig(&select_cv, &select_lock, timo);
1114 1.103 ad if (error != 0)
1115 1.103 ad break;
1116 1.103 ad }
1117 1.103 ad selclear();
1118 1.103 ad mutex_exit(&select_lock);
1119 1.103 ad
1120 1.98 ad if (mask) {
1121 1.98 ad mutex_enter(&p->p_smutex);
1122 1.98 ad l->l_sigmask = oldmask;
1123 1.98 ad mutex_exit(&p->p_smutex);
1124 1.98 ad }
1125 1.97 ad done:
1126 1.28 mycroft /* poll is not restarted after signals... */
1127 1.28 mycroft if (error == ERESTART)
1128 1.28 mycroft error = EINTR;
1129 1.28 mycroft if (error == EWOULDBLOCK)
1130 1.28 mycroft error = 0;
1131 1.103 ad if (error == 0)
1132 1.82 matt error = copyout(bits, u_fds, ni);
1133 1.103 ad if (bits != smallbits)
1134 1.103 ad kmem_free(bits, ni);
1135 1.28 mycroft return (error);
1136 1.28 mycroft }
1137 1.28 mycroft
1138 1.28 mycroft int
1139 1.103 ad pollscan(lwp_t *l, struct pollfd *fds, int nfd, register_t *retval)
1140 1.53 lukem {
1141 1.103 ad proc_t *p = l->l_proc;
1142 1.53 lukem struct filedesc *fdp;
1143 1.53 lukem int i, n;
1144 1.53 lukem struct file *fp;
1145 1.28 mycroft
1146 1.53 lukem fdp = p->p_fd;
1147 1.54 lukem n = 0;
1148 1.28 mycroft for (i = 0; i < nfd; i++, fds++) {
1149 1.60 christos if (fds->fd >= fdp->fd_nfiles) {
1150 1.28 mycroft fds->revents = POLLNVAL;
1151 1.28 mycroft n++;
1152 1.60 christos } else if (fds->fd < 0) {
1153 1.60 christos fds->revents = 0;
1154 1.28 mycroft } else {
1155 1.56 thorpej if ((fp = fd_getfile(fdp, fds->fd)) == NULL) {
1156 1.32 mrg fds->revents = POLLNVAL;
1157 1.28 mycroft n++;
1158 1.32 mrg } else {
1159 1.45 thorpej FILE_USE(fp);
1160 1.32 mrg fds->revents = (*fp->f_ops->fo_poll)(fp,
1161 1.84 christos fds->events | POLLERR | POLLHUP, l);
1162 1.32 mrg if (fds->revents != 0)
1163 1.32 mrg n++;
1164 1.84 christos FILE_UNUSE(fp, l);
1165 1.32 mrg }
1166 1.28 mycroft }
1167 1.28 mycroft }
1168 1.28 mycroft *retval = n;
1169 1.28 mycroft return (0);
1170 1.28 mycroft }
1171 1.28 mycroft
1172 1.15 cgd /*ARGSUSED*/
1173 1.22 christos int
1174 1.103 ad seltrue(dev_t dev, int events, lwp_t *l)
1175 1.15 cgd {
1176 1.15 cgd
1177 1.28 mycroft return (events & (POLLIN | POLLOUT | POLLRDNORM | POLLWRNORM));
1178 1.15 cgd }
1179 1.15 cgd
1180 1.15 cgd /*
1181 1.15 cgd * Record a select request.
1182 1.15 cgd */
1183 1.15 cgd void
1184 1.103 ad selrecord(lwp_t *selector, struct selinfo *sip)
1185 1.15 cgd {
1186 1.15 cgd
1187 1.103 ad mutex_enter(&select_lock);
1188 1.103 ad if (sip->sel_lwp == NULL) {
1189 1.103 ad /* First named waiter, although there may be more. */
1190 1.103 ad sip->sel_lwp = selector;
1191 1.103 ad SLIST_INSERT_HEAD(&selector->l_selwait, sip, sel_chain);
1192 1.103 ad } else if (sip->sel_lwp != selector) {
1193 1.103 ad /* Multiple waiters. */
1194 1.103 ad sip->sel_collision = true;
1195 1.103 ad }
1196 1.103 ad mutex_exit(&select_lock);
1197 1.103 ad }
1198 1.98 ad
1199 1.103 ad /*
1200 1.103 ad * Do a wakeup when a selectable event occurs.
1201 1.103 ad */
1202 1.103 ad void
1203 1.103 ad selwakeup(struct selinfo *sip)
1204 1.103 ad {
1205 1.103 ad lwp_t *l;
1206 1.103 ad
1207 1.103 ad mutex_enter(&select_lock);
1208 1.103 ad if (sip->sel_collision) {
1209 1.103 ad /* Multiple waiters - just notify everybody. */
1210 1.103 ad nselcoll++;
1211 1.103 ad sip->sel_collision = false;
1212 1.103 ad cv_broadcast(&select_cv);
1213 1.103 ad } else if (sip->sel_lwp != NULL) {
1214 1.103 ad /* Only one LWP waiting. */
1215 1.103 ad l = sip->sel_lwp;
1216 1.103 ad if (l->l_selflag == SEL_BLOCKING) {
1217 1.103 ad /*
1218 1.103 ad * If it's sleeping, wake it up. If not, it's
1219 1.103 ad * already awake but hasn't yet removed itself
1220 1.103 ad * from the selector. We reset the state below
1221 1.103 ad * so that we only attempt to do this once.
1222 1.103 ad */
1223 1.98 ad lwp_lock(l);
1224 1.103 ad if (l->l_wchan == &select_cv) {
1225 1.103 ad /* lwp_unsleep() releases the LWP lock. */
1226 1.103 ad lwp_unsleep(l);
1227 1.103 ad } else
1228 1.98 ad lwp_unlock(l);
1229 1.103 ad } else {
1230 1.103 ad /*
1231 1.103 ad * Not yet asleep. Reset its state below so that
1232 1.103 ad * it will go around again.
1233 1.103 ad */
1234 1.69 thorpej }
1235 1.103 ad l->l_selflag = SEL_RESET;
1236 1.69 thorpej }
1237 1.103 ad mutex_exit(&select_lock);
1238 1.103 ad }
1239 1.103 ad
1240 1.103 ad void
1241 1.103 ad selnotify(struct selinfo *sip, long knhint)
1242 1.103 ad {
1243 1.69 thorpej
1244 1.103 ad selwakeup(sip);
1245 1.103 ad KNOTE(&sip->sel_klist, knhint);
1246 1.15 cgd }
1247 1.15 cgd
1248 1.15 cgd /*
1249 1.103 ad * Remove an LWP from all objects that it is waiting for.
1250 1.15 cgd */
1251 1.103 ad static void
1252 1.103 ad selclear(void)
1253 1.103 ad {
1254 1.47 augustss struct selinfo *sip;
1255 1.103 ad lwp_t *l = curlwp;
1256 1.103 ad
1257 1.103 ad KASSERT(mutex_owned(&select_lock));
1258 1.15 cgd
1259 1.103 ad SLIST_FOREACH(sip, &l->l_selwait, sel_chain) {
1260 1.103 ad KASSERT(sip->sel_lwp == l);
1261 1.103 ad sip->sel_lwp = NULL;
1262 1.15 cgd }
1263 1.103 ad }
1264 1.98 ad
1265 1.103 ad /*
1266 1.103 ad * Initialize the select/poll system calls.
1267 1.103 ad */
1268 1.103 ad void
1269 1.103 ad selsysinit(void)
1270 1.103 ad {
1271 1.98 ad
1272 1.103 ad mutex_init(&select_lock, MUTEX_DRIVER, IPL_VM);
1273 1.103 ad cv_init(&select_cv, "select");
1274 1.15 cgd }
1275