sys_select.c revision 1.5.2.1 1 /* $NetBSD: sys_select.c,v 1.5.2.1 2008/05/16 02:25:27 yamt Exp $ */
2
3 /*-
4 * Copyright (c) 2007, 2008 The NetBSD Foundation, Inc.
5 * All rights reserved.
6 *
7 * This code is derived from software contributed to The NetBSD Foundation
8 * by Andrew Doran.
9 *
10 * Redistribution and use in source and binary forms, with or without
11 * modification, are permitted provided that the following conditions
12 * are met:
13 * 1. Redistributions of source code must retain the above copyright
14 * notice, this list of conditions and the following disclaimer.
15 * 2. Redistributions in binary form must reproduce the above copyright
16 * notice, this list of conditions and the following disclaimer in the
17 * documentation and/or other materials provided with the distribution.
18 *
19 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
20 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
21 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
22 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
23 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
24 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
25 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
26 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
27 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
28 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
29 * POSSIBILITY OF SUCH DAMAGE.
30 */
31
32 /*
33 * Copyright (c) 1982, 1986, 1989, 1993
34 * The Regents of the University of California. All rights reserved.
35 * (c) UNIX System Laboratories, Inc.
36 * All or some portions of this file are derived from material licensed
37 * to the University of California by American Telephone and Telegraph
38 * Co. or Unix System Laboratories, Inc. and are reproduced herein with
39 * the permission of UNIX System Laboratories, Inc.
40 *
41 * Redistribution and use in source and binary forms, with or without
42 * modification, are permitted provided that the following conditions
43 * are met:
44 * 1. Redistributions of source code must retain the above copyright
45 * notice, this list of conditions and the following disclaimer.
46 * 2. Redistributions in binary form must reproduce the above copyright
47 * notice, this list of conditions and the following disclaimer in the
48 * documentation and/or other materials provided with the distribution.
49 * 3. Neither the name of the University nor the names of its contributors
50 * may be used to endorse or promote products derived from this software
51 * without specific prior written permission.
52 *
53 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
54 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
55 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
56 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
57 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
58 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
59 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
60 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
61 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
62 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
63 * SUCH DAMAGE.
64 *
65 * @(#)sys_generic.c 8.9 (Berkeley) 2/14/95
66 */
67
68 /*
69 * System calls relating to files.
70 */
71
72 #include <sys/cdefs.h>
73 __KERNEL_RCSID(0, "$NetBSD: sys_select.c,v 1.5.2.1 2008/05/16 02:25:27 yamt Exp $");
74
75 #include <sys/param.h>
76 #include <sys/systm.h>
77 #include <sys/filedesc.h>
78 #include <sys/ioctl.h>
79 #include <sys/file.h>
80 #include <sys/proc.h>
81 #include <sys/socketvar.h>
82 #include <sys/signalvar.h>
83 #include <sys/uio.h>
84 #include <sys/kernel.h>
85 #include <sys/stat.h>
86 #include <sys/poll.h>
87 #include <sys/vnode.h>
88 #include <sys/mount.h>
89 #include <sys/syscallargs.h>
90 #include <sys/cpu.h>
91 #include <sys/atomic.h>
92 #include <sys/socketvar.h>
93 #include <sys/sleepq.h>
94
95 /* Flags for lwp::l_selflag. */
96 #define SEL_RESET 0 /* awoken, interrupted, or not yet polling */
97 #define SEL_SCANNING 1 /* polling descriptors */
98 #define SEL_BLOCKING 2 /* about to block on select_cv */
99
100 /* Per-CPU state for select()/poll(). */
101 #if MAXCPUS > 32
102 #error adjust this code
103 #endif
104 typedef struct selcpu {
105 kmutex_t sc_lock;
106 sleepq_t sc_sleepq;
107 int sc_ncoll;
108 uint32_t sc_mask;
109 } selcpu_t;
110
111 static int selscan(lwp_t *, fd_mask *, fd_mask *, int, register_t *);
112 static int pollscan(lwp_t *, struct pollfd *, int, register_t *);
113 static void selclear(void);
114
115 static syncobj_t select_sobj = {
116 SOBJ_SLEEPQ_FIFO,
117 sleepq_unsleep,
118 sleepq_changepri,
119 sleepq_lendpri,
120 syncobj_noowner,
121 };
122
123 /*
124 * Select system call.
125 */
126 int
127 sys_pselect(struct lwp *l, const struct sys_pselect_args *uap, register_t *retval)
128 {
129 /* {
130 syscallarg(int) nd;
131 syscallarg(fd_set *) in;
132 syscallarg(fd_set *) ou;
133 syscallarg(fd_set *) ex;
134 syscallarg(const struct timespec *) ts;
135 syscallarg(sigset_t *) mask;
136 } */
137 struct timespec ats;
138 struct timeval atv, *tv = NULL;
139 sigset_t amask, *mask = NULL;
140 int error;
141
142 if (SCARG(uap, ts)) {
143 error = copyin(SCARG(uap, ts), &ats, sizeof(ats));
144 if (error)
145 return error;
146 atv.tv_sec = ats.tv_sec;
147 atv.tv_usec = ats.tv_nsec / 1000;
148 tv = &atv;
149 }
150 if (SCARG(uap, mask) != NULL) {
151 error = copyin(SCARG(uap, mask), &amask, sizeof(amask));
152 if (error)
153 return error;
154 mask = &amask;
155 }
156
157 return selcommon(l, retval, SCARG(uap, nd), SCARG(uap, in),
158 SCARG(uap, ou), SCARG(uap, ex), tv, mask);
159 }
160
161 int
162 inittimeleft(struct timeval *tv, struct timeval *sleeptv)
163 {
164 if (itimerfix(tv))
165 return -1;
166 getmicrouptime(sleeptv);
167 return 0;
168 }
169
170 int
171 gettimeleft(struct timeval *tv, struct timeval *sleeptv)
172 {
173 /*
174 * We have to recalculate the timeout on every retry.
175 */
176 struct timeval slepttv;
177 /*
178 * reduce tv by elapsed time
179 * based on monotonic time scale
180 */
181 getmicrouptime(&slepttv);
182 timeradd(tv, sleeptv, tv);
183 timersub(tv, &slepttv, tv);
184 *sleeptv = slepttv;
185 return tvtohz(tv);
186 }
187
188 int
189 sys_select(struct lwp *l, const struct sys_select_args *uap, register_t *retval)
190 {
191 /* {
192 syscallarg(int) nd;
193 syscallarg(fd_set *) in;
194 syscallarg(fd_set *) ou;
195 syscallarg(fd_set *) ex;
196 syscallarg(struct timeval *) tv;
197 } */
198 struct timeval atv, *tv = NULL;
199 int error;
200
201 if (SCARG(uap, tv)) {
202 error = copyin(SCARG(uap, tv), (void *)&atv,
203 sizeof(atv));
204 if (error)
205 return error;
206 tv = &atv;
207 }
208
209 return selcommon(l, retval, SCARG(uap, nd), SCARG(uap, in),
210 SCARG(uap, ou), SCARG(uap, ex), tv, NULL);
211 }
212
213 int
214 selcommon(lwp_t *l, register_t *retval, int nd, fd_set *u_in,
215 fd_set *u_ou, fd_set *u_ex, struct timeval *tv, sigset_t *mask)
216 {
217 char smallbits[howmany(FD_SETSIZE, NFDBITS) *
218 sizeof(fd_mask) * 6];
219 proc_t * const p = l->l_proc;
220 char *bits;
221 int ncoll, error, timo;
222 size_t ni;
223 sigset_t oldmask;
224 struct timeval sleeptv;
225 selcpu_t *sc;
226
227 error = 0;
228 if (nd < 0)
229 return (EINVAL);
230 if (nd > p->p_fd->fd_nfiles) {
231 /* forgiving; slightly wrong */
232 nd = p->p_fd->fd_nfiles;
233 }
234 ni = howmany(nd, NFDBITS) * sizeof(fd_mask);
235 if (ni * 6 > sizeof(smallbits))
236 bits = kmem_alloc(ni * 6, KM_SLEEP);
237 else
238 bits = smallbits;
239
240 #define getbits(name, x) \
241 if (u_ ## name) { \
242 error = copyin(u_ ## name, bits + ni * x, ni); \
243 if (error) \
244 goto done; \
245 } else \
246 memset(bits + ni * x, 0, ni);
247 getbits(in, 0);
248 getbits(ou, 1);
249 getbits(ex, 2);
250 #undef getbits
251
252 timo = 0;
253 if (tv && inittimeleft(tv, &sleeptv) == -1) {
254 error = EINVAL;
255 goto done;
256 }
257
258 if (mask) {
259 sigminusset(&sigcantmask, mask);
260 mutex_enter(p->p_lock);
261 oldmask = l->l_sigmask;
262 l->l_sigmask = *mask;
263 mutex_exit(p->p_lock);
264 } else
265 oldmask = l->l_sigmask; /* XXXgcc */
266
267 sc = curcpu()->ci_data.cpu_selcpu;
268 l->l_selcpu = sc;
269 SLIST_INIT(&l->l_selwait);
270 for (;;) {
271 /*
272 * No need to lock. If this is overwritten by another
273 * value while scanning, we will retry below. We only
274 * need to see exact state from the descriptors that
275 * we are about to poll, and lock activity resulting
276 * from fo_poll is enough to provide an up to date value
277 * for new polling activity.
278 */
279 l->l_selflag = SEL_SCANNING;
280 ncoll = sc->sc_ncoll;
281
282 error = selscan(l, (fd_mask *)(bits + ni * 0),
283 (fd_mask *)(bits + ni * 3), nd, retval);
284
285 if (error || *retval)
286 break;
287 if (tv && (timo = gettimeleft(tv, &sleeptv)) <= 0)
288 break;
289 mutex_spin_enter(&sc->sc_lock);
290 if (l->l_selflag != SEL_SCANNING || sc->sc_ncoll != ncoll) {
291 mutex_spin_exit(&sc->sc_lock);
292 continue;
293 }
294 l->l_selflag = SEL_BLOCKING;
295 l->l_kpriority = true;
296 lwp_lock(l);
297 lwp_unlock_to(l, &sc->sc_lock);
298 sleepq_enqueue(&sc->sc_sleepq, sc, "select", &select_sobj);
299 KERNEL_UNLOCK_ALL(NULL, &l->l_biglocks); /* XXX */
300 error = sleepq_block(timo, true);
301 if (error != 0)
302 break;
303 }
304 selclear();
305
306 if (mask) {
307 mutex_enter(p->p_lock);
308 l->l_sigmask = oldmask;
309 mutex_exit(p->p_lock);
310 }
311
312 done:
313 /* select is not restarted after signals... */
314 if (error == ERESTART)
315 error = EINTR;
316 if (error == EWOULDBLOCK)
317 error = 0;
318 if (error == 0 && u_in != NULL)
319 error = copyout(bits + ni * 3, u_in, ni);
320 if (error == 0 && u_ou != NULL)
321 error = copyout(bits + ni * 4, u_ou, ni);
322 if (error == 0 && u_ex != NULL)
323 error = copyout(bits + ni * 5, u_ex, ni);
324 if (bits != smallbits)
325 kmem_free(bits, ni * 6);
326 return (error);
327 }
328
329 int
330 selscan(lwp_t *l, fd_mask *ibitp, fd_mask *obitp, int nfd,
331 register_t *retval)
332 {
333 static const int flag[3] = { POLLRDNORM | POLLHUP | POLLERR,
334 POLLWRNORM | POLLHUP | POLLERR,
335 POLLRDBAND };
336 int msk, i, j, fd, n;
337 fd_mask ibits, obits;
338 file_t *fp;
339
340 n = 0;
341 for (msk = 0; msk < 3; msk++) {
342 for (i = 0; i < nfd; i += NFDBITS) {
343 ibits = *ibitp++;
344 obits = 0;
345 while ((j = ffs(ibits)) && (fd = i + --j) < nfd) {
346 ibits &= ~(1 << j);
347 if ((fp = fd_getfile(fd)) == NULL)
348 return (EBADF);
349 if ((*fp->f_ops->fo_poll)(fp, flag[msk])) {
350 obits |= (1 << j);
351 n++;
352 }
353 fd_putfile(fd);
354 }
355 *obitp++ = obits;
356 }
357 }
358 *retval = n;
359 return (0);
360 }
361
362 /*
363 * Poll system call.
364 */
365 int
366 sys_poll(struct lwp *l, const struct sys_poll_args *uap, register_t *retval)
367 {
368 /* {
369 syscallarg(struct pollfd *) fds;
370 syscallarg(u_int) nfds;
371 syscallarg(int) timeout;
372 } */
373 struct timeval atv, *tv = NULL;
374
375 if (SCARG(uap, timeout) != INFTIM) {
376 atv.tv_sec = SCARG(uap, timeout) / 1000;
377 atv.tv_usec = (SCARG(uap, timeout) % 1000) * 1000;
378 tv = &atv;
379 }
380
381 return pollcommon(l, retval, SCARG(uap, fds), SCARG(uap, nfds),
382 tv, NULL);
383 }
384
385 /*
386 * Poll system call.
387 */
388 int
389 sys_pollts(struct lwp *l, const struct sys_pollts_args *uap, register_t *retval)
390 {
391 /* {
392 syscallarg(struct pollfd *) fds;
393 syscallarg(u_int) nfds;
394 syscallarg(const struct timespec *) ts;
395 syscallarg(const sigset_t *) mask;
396 } */
397 struct timespec ats;
398 struct timeval atv, *tv = NULL;
399 sigset_t amask, *mask = NULL;
400 int error;
401
402 if (SCARG(uap, ts)) {
403 error = copyin(SCARG(uap, ts), &ats, sizeof(ats));
404 if (error)
405 return error;
406 atv.tv_sec = ats.tv_sec;
407 atv.tv_usec = ats.tv_nsec / 1000;
408 tv = &atv;
409 }
410 if (SCARG(uap, mask)) {
411 error = copyin(SCARG(uap, mask), &amask, sizeof(amask));
412 if (error)
413 return error;
414 mask = &amask;
415 }
416
417 return pollcommon(l, retval, SCARG(uap, fds), SCARG(uap, nfds),
418 tv, mask);
419 }
420
421 int
422 pollcommon(lwp_t *l, register_t *retval,
423 struct pollfd *u_fds, u_int nfds,
424 struct timeval *tv, sigset_t *mask)
425 {
426 char smallbits[32 * sizeof(struct pollfd)];
427 proc_t * const p = l->l_proc;
428 void * bits;
429 sigset_t oldmask;
430 int ncoll, error, timo;
431 size_t ni;
432 struct timeval sleeptv;
433 selcpu_t *sc;
434
435 if (nfds > p->p_fd->fd_nfiles) {
436 /* forgiving; slightly wrong */
437 nfds = p->p_fd->fd_nfiles;
438 }
439 ni = nfds * sizeof(struct pollfd);
440 if (ni > sizeof(smallbits))
441 bits = kmem_alloc(ni, KM_SLEEP);
442 else
443 bits = smallbits;
444
445 error = copyin(u_fds, bits, ni);
446 if (error)
447 goto done;
448
449 timo = 0;
450 if (tv && inittimeleft(tv, &sleeptv) == -1) {
451 error = EINVAL;
452 goto done;
453 }
454
455 if (mask) {
456 sigminusset(&sigcantmask, mask);
457 mutex_enter(p->p_lock);
458 oldmask = l->l_sigmask;
459 l->l_sigmask = *mask;
460 mutex_exit(p->p_lock);
461 } else
462 oldmask = l->l_sigmask; /* XXXgcc */
463
464 sc = curcpu()->ci_data.cpu_selcpu;
465 l->l_selcpu = sc;
466 SLIST_INIT(&l->l_selwait);
467 for (;;) {
468 /*
469 * No need to lock. If this is overwritten by another
470 * value while scanning, we will retry below. We only
471 * need to see exact state from the descriptors that
472 * we are about to poll, and lock activity resulting
473 * from fo_poll is enough to provide an up to date value
474 * for new polling activity.
475 */
476 ncoll = sc->sc_ncoll;
477 l->l_selflag = SEL_SCANNING;
478
479 error = pollscan(l, (struct pollfd *)bits, nfds, retval);
480
481 if (error || *retval)
482 break;
483 if (tv && (timo = gettimeleft(tv, &sleeptv)) <= 0)
484 break;
485 mutex_spin_enter(&sc->sc_lock);
486 if (l->l_selflag != SEL_SCANNING || sc->sc_ncoll != ncoll) {
487 mutex_spin_exit(&sc->sc_lock);
488 continue;
489 }
490 l->l_selflag = SEL_BLOCKING;
491 l->l_kpriority = true;
492 lwp_lock(l);
493 lwp_unlock_to(l, &sc->sc_lock);
494 sleepq_enqueue(&sc->sc_sleepq, sc, "select", &select_sobj);
495 KERNEL_UNLOCK_ALL(NULL, &l->l_biglocks); /* XXX */
496 error = sleepq_block(timo, true);
497 if (error != 0)
498 break;
499 }
500 selclear();
501
502 if (mask) {
503 mutex_enter(p->p_lock);
504 l->l_sigmask = oldmask;
505 mutex_exit(p->p_lock);
506 }
507 done:
508 /* poll is not restarted after signals... */
509 if (error == ERESTART)
510 error = EINTR;
511 if (error == EWOULDBLOCK)
512 error = 0;
513 if (error == 0)
514 error = copyout(bits, u_fds, ni);
515 if (bits != smallbits)
516 kmem_free(bits, ni);
517 return (error);
518 }
519
520 int
521 pollscan(lwp_t *l, struct pollfd *fds, int nfd, register_t *retval)
522 {
523 int i, n;
524 file_t *fp;
525
526 n = 0;
527 for (i = 0; i < nfd; i++, fds++) {
528 if (fds->fd < 0) {
529 fds->revents = 0;
530 } else if ((fp = fd_getfile(fds->fd)) == NULL) {
531 fds->revents = POLLNVAL;
532 n++;
533 } else {
534 fds->revents = (*fp->f_ops->fo_poll)(fp,
535 fds->events | POLLERR | POLLHUP);
536 if (fds->revents != 0)
537 n++;
538 fd_putfile(fds->fd);
539 }
540 }
541 *retval = n;
542 return (0);
543 }
544
545 /*ARGSUSED*/
546 int
547 seltrue(dev_t dev, int events, lwp_t *l)
548 {
549
550 return (events & (POLLIN | POLLOUT | POLLRDNORM | POLLWRNORM));
551 }
552
553 /*
554 * Record a select request. Concurrency issues:
555 *
556 * The caller holds the same lock across calls to selrecord() and
557 * selnotify(), so we don't need to consider a concurrent wakeup
558 * while in this routine.
559 *
560 * The only activity we need to guard against is selclear(), called by
561 * another thread that is exiting selcommon() or pollcommon().
562 * `sel_lwp' can only become non-NULL while the caller's lock is held,
563 * so it cannot become non-NULL due to a change made by another thread
564 * while we are in this routine. It can only become _NULL_ due to a
565 * call to selclear().
566 *
567 * If it is non-NULL and != selector there is the potential for
568 * selclear() to be called by another thread. If either of those
569 * conditions are true, we're not interested in touching the `named
570 * waiter' part of the selinfo record because we need to record a
571 * collision. Hence there is no need for additional locking in this
572 * routine.
573 */
574 void
575 selrecord(lwp_t *selector, struct selinfo *sip)
576 {
577 selcpu_t *sc;
578 lwp_t *other;
579
580 KASSERT(selector == curlwp);
581
582 sc = selector->l_selcpu;
583 other = sip->sel_lwp;
584
585 if (other == selector) {
586 /* `selector' has already claimed it. */
587 KASSERT(sip->sel_cpu = sc);
588 } else if (other == NULL) {
589 /*
590 * First named waiter, although there may be unnamed
591 * waiters (collisions). Issue a memory barrier to
592 * ensure that we access sel_lwp (above) before other
593 * fields - this guards against a call to selclear().
594 */
595 membar_enter();
596 sip->sel_lwp = selector;
597 SLIST_INSERT_HEAD(&selector->l_selwait, sip, sel_chain);
598 /* Replace selinfo's lock with our chosen CPU's lock. */
599 sip->sel_cpu = sc;
600 } else {
601 /* Multiple waiters: record a collision. */
602 sip->sel_collision |= sc->sc_mask;
603 KASSERT(sip->sel_cpu != NULL);
604 }
605 }
606
607 /*
608 * Do a wakeup when a selectable event occurs. Concurrency issues:
609 *
610 * As per selrecord(), the caller's object lock is held. If there
611 * is a named waiter, we must acquire the associated selcpu's lock
612 * in order to synchronize with selclear() and pollers going to sleep
613 * in selcommon() and/or pollcommon().
614 *
615 * sip->sel_cpu cannot change at this point, as it is only changed
616 * in selrecord(), and concurrent calls to selrecord() are locked
617 * out by the caller.
618 */
619 void
620 selnotify(struct selinfo *sip, int events, long knhint)
621 {
622 selcpu_t *sc;
623 uint32_t mask;
624 int index, oflag, swapin;
625 lwp_t *l;
626
627 KNOTE(&sip->sel_klist, knhint);
628
629 if (sip->sel_lwp != NULL) {
630 /* One named LWP is waiting. */
631 swapin = 0;
632 sc = sip->sel_cpu;
633 mutex_spin_enter(&sc->sc_lock);
634 /* Still there? */
635 if (sip->sel_lwp != NULL) {
636 l = sip->sel_lwp;
637 /*
638 * If thread is sleeping, wake it up. If it's not
639 * yet asleep, it will notice the change in state
640 * and will re-poll the descriptors.
641 */
642 oflag = l->l_selflag;
643 l->l_selflag = SEL_RESET;
644 if (oflag == SEL_BLOCKING &&
645 l->l_mutex == &sc->sc_lock) {
646 KASSERT(l->l_wchan == sc);
647 swapin = sleepq_unsleep(l, false);
648 }
649 }
650 mutex_spin_exit(&sc->sc_lock);
651 if (swapin)
652 uvm_kick_scheduler();
653 }
654
655 if ((mask = sip->sel_collision) != 0) {
656 /*
657 * There was a collision (multiple waiters): we must
658 * inform all potentially interested waiters.
659 */
660 sip->sel_collision = 0;
661 do {
662 index = ffs(mask) - 1;
663 mask &= ~(1 << index);
664 sc = cpu_lookup_byindex(index)->ci_data.cpu_selcpu;
665 mutex_spin_enter(&sc->sc_lock);
666 sc->sc_ncoll++;
667 sleepq_wake(&sc->sc_sleepq, sc, (u_int)-1);
668 } while (__predict_false(mask != 0));
669 }
670 }
671
672 /*
673 * Remove an LWP from all objects that it is waiting for. Concurrency
674 * issues:
675 *
676 * The object owner's (e.g. device driver) lock is not held here. Calls
677 * can be made to selrecord() and we do not synchronize against those
678 * directly using locks. However, we use `sel_lwp' to lock out changes.
679 * Before clearing it we must use memory barriers to ensure that we can
680 * safely traverse the list of selinfo records.
681 */
682 static void
683 selclear(void)
684 {
685 struct selinfo *sip, *next;
686 selcpu_t *sc;
687 lwp_t *l;
688
689 l = curlwp;
690 sc = l->l_selcpu;
691
692 mutex_spin_enter(&sc->sc_lock);
693 for (sip = SLIST_FIRST(&l->l_selwait); sip != NULL; sip = next) {
694 KASSERT(sip->sel_lwp == l);
695 KASSERT(sip->sel_cpu == l->l_selcpu);
696 /*
697 * Read link to next selinfo record, if any.
698 * It's no longer safe to touch `sip' after clearing
699 * `sel_lwp', so ensure that the read of `sel_chain'
700 * completes before the clearing of sel_lwp becomes
701 * globally visible.
702 */
703 next = SLIST_NEXT(sip, sel_chain);
704 membar_exit();
705 /* Release the record for another named waiter to use. */
706 sip->sel_lwp = NULL;
707 }
708 mutex_spin_exit(&sc->sc_lock);
709 }
710
711 /*
712 * Initialize the select/poll system calls. Called once for each
713 * CPU in the system, as they are attached.
714 */
715 void
716 selsysinit(struct cpu_info *ci)
717 {
718 selcpu_t *sc;
719
720 sc = kmem_alloc(roundup2(sizeof(selcpu_t), coherency_unit) +
721 coherency_unit, KM_SLEEP);
722 sc = (void *)roundup2((uintptr_t)sc, coherency_unit);
723 mutex_init(&sc->sc_lock, MUTEX_DEFAULT, IPL_SCHED);
724 sleepq_init(&sc->sc_sleepq, &sc->sc_lock);
725 sc->sc_ncoll = 0;
726 sc->sc_mask = (1 << cpu_index(ci));
727 ci->ci_data.cpu_selcpu = sc;
728 }
729
730 /*
731 * Initialize a selinfo record.
732 */
733 void
734 selinit(struct selinfo *sip)
735 {
736
737 memset(sip, 0, sizeof(*sip));
738 }
739
740 /*
741 * Destroy a selinfo record. The owning object must not gain new
742 * references while this is in progress: all activity on the record
743 * must be stopped.
744 *
745 * Concurrency issues: we only need guard against a call to selclear()
746 * by a thread exiting selcommon() and/or pollcommon(). The caller has
747 * prevented further references being made to the selinfo record via
748 * selrecord(), and it won't call selwakeup() again.
749 */
750 void
751 seldestroy(struct selinfo *sip)
752 {
753 selcpu_t *sc;
754 lwp_t *l;
755
756 if (sip->sel_lwp == NULL)
757 return;
758
759 /*
760 * Lock out selclear(). The selcpu pointer can't change while
761 * we are here since it is only ever changed in selrecord(),
762 * and that will not be entered again for this record because
763 * it is dying.
764 */
765 KASSERT(sip->sel_cpu != NULL);
766 sc = sip->sel_cpu;
767 mutex_spin_enter(&sc->sc_lock);
768 if ((l = sip->sel_lwp) != NULL) {
769 /*
770 * This should rarely happen, so although SLIST_REMOVE()
771 * is slow, using it here is not a problem.
772 */
773 KASSERT(l->l_selcpu == sc);
774 SLIST_REMOVE(&l->l_selwait, sip, selinfo, sel_chain);
775 sip->sel_lwp = NULL;
776 }
777 mutex_spin_exit(&sc->sc_lock);
778 }
779
780 int
781 pollsock(struct socket *so, const struct timeval *tvp, int events)
782 {
783 int ncoll, error, timo;
784 struct timeval sleeptv, tv;
785 selcpu_t *sc;
786 lwp_t *l;
787
788 timo = 0;
789 if (tvp != NULL) {
790 tv = *tvp;
791 if (inittimeleft(&tv, &sleeptv) == -1)
792 return EINVAL;
793 }
794
795 l = curlwp;
796 sc = l->l_cpu->ci_data.cpu_selcpu;
797 l->l_selcpu = sc;
798 SLIST_INIT(&l->l_selwait);
799 error = 0;
800 for (;;) {
801 /*
802 * No need to lock. If this is overwritten by another
803 * value while scanning, we will retry below. We only
804 * need to see exact state from the descriptors that
805 * we are about to poll, and lock activity resulting
806 * from fo_poll is enough to provide an up to date value
807 * for new polling activity.
808 */
809 ncoll = sc->sc_ncoll;
810 l->l_selflag = SEL_SCANNING;
811 if (sopoll(so, events) != 0)
812 break;
813 if (tvp && (timo = gettimeleft(&tv, &sleeptv)) <= 0)
814 break;
815 mutex_spin_enter(&sc->sc_lock);
816 if (l->l_selflag != SEL_SCANNING || sc->sc_ncoll != ncoll) {
817 mutex_spin_exit(&sc->sc_lock);
818 continue;
819 }
820 l->l_selflag = SEL_BLOCKING;
821 lwp_lock(l);
822 lwp_unlock_to(l, &sc->sc_lock);
823 sleepq_enqueue(&sc->sc_sleepq, sc, "pollsock", &select_sobj);
824 KERNEL_UNLOCK_ALL(NULL, &l->l_biglocks); /* XXX */
825 error = sleepq_block(timo, true);
826 if (error != 0)
827 break;
828 }
829 selclear();
830 /* poll is not restarted after signals... */
831 if (error == ERESTART)
832 error = EINTR;
833 if (error == EWOULDBLOCK)
834 error = 0;
835 return (error);
836 }
837