1 /* 2 * CDDL HEADER START 3 * 4 * The contents of this file are subject to the terms of the 5 * Common Development and Distribution License (the "License"). 6 * You may not use this file except in compliance with the License. 7 * 8 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE 9 * or http://www.opensolaris.org/os/licensing. 10 * See the License for the specific language governing permissions 11 * and limitations under the License. 12 * 13 * When distributing Covered Code, include this CDDL HEADER in each 14 * file and include the License file at usr/src/OPENSOLARIS.LICENSE. 15 * If applicable, add the following below this CDDL HEADER, with the 16 * fields enclosed by brackets "[]" replaced with your own identifying 17 * information: Portions Copyright [yyyy] [name of copyright owner] 18 * 19 * CDDL HEADER END 20 */ 21 22 /* 23 * Copyright 2007 Sun Microsystems, Inc. All rights reserved. 24 * Use is subject to license terms. 25 */ 26 27 /* Copyright (c) 1983, 1984, 1985, 1986, 1987, 1988, 1989 AT&T */ 28 /* All Rights Reserved */ 29 30 /* 31 * Portions of this source code were derived from Berkeley 4.3 BSD 32 * under license from the Regents of the University of California. 33 */ 34 35 #pragma ident "%Z%%M% %I% %E% SMI" 36 37 #include <sys/param.h> 38 #include <sys/isa_defs.h> 39 #include <sys/types.h> 40 #include <sys/sysmacros.h> 41 #include <sys/user.h> 42 #include <sys/systm.h> 43 #include <sys/errno.h> 44 #include <sys/time.h> 45 #include <sys/vnode.h> 46 #include <sys/file.h> 47 #include <sys/mode.h> 48 #include <sys/proc.h> 49 #include <sys/uio.h> 50 #include <sys/poll_impl.h> 51 #include <sys/kmem.h> 52 #include <sys/cmn_err.h> 53 #include <sys/debug.h> 54 #include <sys/bitmap.h> 55 #include <sys/kstat.h> 56 #include <sys/rctl.h> 57 #include <sys/port_impl.h> 58 #include <sys/schedctl.h> 59 60 #define NPHLOCKS 64 /* Number of locks; must be power of 2 */ 61 #define PHLOCKADDR(php) &plocks[(((uintptr_t)(php)) >> 8) & (NPHLOCKS - 1)] 62 #define PHLOCK(php) PHLOCKADDR(php).pp_lock 63 #define PH_ENTER(php) mutex_enter(PHLOCK(php)) 64 #define PH_EXIT(php) mutex_exit(PHLOCK(php)) 65 #define VALID_POLL_EVENTS (POLLIN | POLLPRI | POLLOUT | POLLRDNORM \ 66 | POLLRDBAND | POLLWRBAND | POLLHUP | POLLERR | POLLNVAL) 67 68 /* 69 * global counters to collect some stats 70 */ 71 static struct { 72 kstat_named_t polllistmiss; /* failed to find a cached poll list */ 73 kstat_named_t pollcachehit; /* list matched 100% w/ cached one */ 74 kstat_named_t pollcachephit; /* list matched < 100% w/ cached one */ 75 kstat_named_t pollcachemiss; /* every list entry is dif from cache */ 76 } pollstats = { 77 { "polllistmiss", KSTAT_DATA_UINT64 }, 78 { "pollcachehit", KSTAT_DATA_UINT64 }, 79 { "pollcachephit", KSTAT_DATA_UINT64 }, 80 { "pollcachemiss", KSTAT_DATA_UINT64 } 81 }; 82 83 kstat_named_t *pollstats_ptr = (kstat_named_t *)&pollstats; 84 uint_t pollstats_ndata = sizeof (pollstats) / sizeof (kstat_named_t); 85 86 struct pplock { 87 kmutex_t pp_lock; 88 short pp_flag; 89 kcondvar_t pp_wait_cv; 90 int32_t pp_pad; /* to a nice round 16 bytes */ 91 }; 92 93 static struct pplock plocks[NPHLOCKS]; /* Hash array of pollhead locks */ 94 95 #ifdef DEBUG 96 static int pollchecksanity(pollstate_t *, nfds_t); 97 static int pollcheckxref(pollstate_t *, int); 98 static void pollcheckphlist(void); 99 static int pollcheckrevents(pollstate_t *, int, int, int); 100 static void checkpolldat(pollstate_t *); 101 #endif /* DEBUG */ 102 static int plist_chkdupfd(file_t *, polldat_t *, pollstate_t *, pollfd_t *, int, 103 int *); 104 105 /* 106 * Data structure overview: 107 * The per-thread poll state consists of 108 * one pollstate_t 109 * one pollcache_t 110 * one bitmap with one event bit per fd 111 * a (two-dimensional) hashed array of polldat_t structures - one entry 112 * per fd 113 * 114 * This conglomerate of data structures interact with 115 * the pollhead which is used by VOP_POLL and pollwakeup 116 * (protected by the PHLOCK, cached array of plocks), and 117 * the fpollinfo list hanging off the fi_list which is used to notify 118 * poll when a cached fd is closed. This is protected by uf_lock. 119 * 120 * Invariants: 121 * pd_php (pollhead pointer) is set iff (if and only if) the polldat 122 * is on that pollhead. This is modified atomically under pc_lock. 123 * 124 * pd_fp (file_t pointer) is set iff the thread is on the fpollinfo 125 * list for that open file. 126 * This is modified atomically under pc_lock. 127 * 128 * pd_count is the sum (over all values of i) of pd_ref[i].xf_refcnt. 129 * Iff pd_ref[i].xf_refcnt >= 1 then 130 * ps_pcacheset[i].pcs_pollfd[pd_ref[i].xf_position].fd == pd_fd 131 * Iff pd_ref[i].xf_refcnt > 1 then 132 * In ps_pcacheset[i].pcs_pollfd between index 133 * pd_ref[i].xf_position] and the end of the list 134 * there are xf_refcnt entries with .fd == pd_fd 135 * 136 * Locking design: 137 * Whenever possible the design relies on the fact that the poll cache state 138 * is per thread thus for both poll and exit it is self-synchronizing. 139 * Thus the key interactions where other threads access the state are: 140 * pollwakeup (and polltime), and 141 * close cleaning up the cached references to an open file 142 * 143 * The two key locks in poll proper is ps_lock and pc_lock. 144 * 145 * The ps_lock is used for synchronization between poll, (lwp_)exit and close 146 * to ensure that modifications to pollcacheset structure are serialized. 147 * This lock is held through most of poll() except where poll sleeps 148 * since there is little need to handle closes concurrently with the execution 149 * of poll. 150 * The pc_lock protects most of the fields in pollcache structure and polldat 151 * structures (which are accessed by poll, pollwakeup, and polltime) 152 * with the exception of fields that are only modified when only one thread 153 * can access this per-thread state. 154 * Those exceptions occur in poll when first allocating the per-thread state, 155 * when poll grows the number of polldat (never shrinks), and when 156 * exit/pollcleanup has ensured that there are no references from either 157 * pollheads or fpollinfo to the threads poll state. 158 * 159 * Poll(2) system call is the only path which ps_lock and pc_lock are both 160 * held, in that order. It needs ps_lock to synchronize with close and 161 * lwp_exit; and pc_lock with pollwakeup. 162 * 163 * The locking interaction between pc_lock and PHLOCK take into account 164 * that poll acquires these locks in the order of pc_lock and then PHLOCK 165 * while pollwakeup does it in the reverse order. Thus pollwakeup implements 166 * deadlock avoidance by dropping the locks and reacquiring them in the 167 * reverse order. For this to work pollwakeup needs to prevent the thread 168 * from exiting and freeing all of the poll related state. Thus is done 169 * using 170 * the pc_no_exit lock 171 * the pc_busy counter 172 * the pc_busy_cv condition variable 173 * 174 * The locking interaction between pc_lock and uf_lock has similar 175 * issues. Poll holds ps_lock and/or pc_lock across calls to getf/releasef 176 * which acquire uf_lock. The poll cleanup in close needs to hold uf_lock 177 * to prevent poll or exit from doing a delfpollinfo after which the thread 178 * might exit. But the cleanup needs to acquire pc_lock when modifying 179 * the poll cache state. The solution is to use pc_busy and do the close 180 * cleanup in two phases: 181 * First close calls pollblockexit which increments pc_busy. 182 * This prevents the per-thread poll related state from being freed. 183 * Then close drops uf_lock and calls pollcacheclean. 184 * This routine can then acquire pc_lock and remove any references 185 * to the closing fd (as well as recording that it has been closed 186 * so that a POLLNVAL can be generated even if the fd is reused before 187 * poll has been woken up and checked getf() again). 188 * 189 * When removing a polled fd from poll cache, the fd is always removed 190 * from pollhead list first and then from fpollinfo list, i.e., 191 * pollhead_delete() is called before delfpollinfo(). 192 * 193 * 194 * Locking hierarchy: 195 * pc_no_exit is a leaf level lock. 196 * ps_lock is held when acquiring pc_lock (except when pollwakeup 197 * acquires pc_lock). 198 * pc_lock might be held when acquiring PHLOCK (pollhead_insert/ 199 * pollhead_delete) 200 * pc_lock is always held (but this is not required) 201 * when acquiring PHLOCK (in polladd/pollhead_delete and pollwakeup called 202 * from pcache_clean_entry). 203 * pc_lock is held across addfpollinfo/delfpollinfo which acquire 204 * uf_lock. 205 * pc_lock is held across getf/releasef which acquire uf_lock. 206 * ps_lock might be held across getf/releasef which acquire uf_lock. 207 * pollwakeup tries to acquire pc_lock while holding PHLOCK 208 * but drops the locks and reacquire them in reverse order to avoid 209 * deadlock. 210 * 211 * Note also that there is deadlock avoidance support for VOP_POLL routines 212 * and pollwakeup involving a file system or driver lock. 213 * See below. 214 */ 215 216 /* 217 * Deadlock avoidance support for VOP_POLL() routines. This is 218 * sometimes necessary to prevent deadlock between polling threads 219 * (which hold poll locks on entry to xx_poll(), then acquire foo) 220 * and pollwakeup() threads (which hold foo, then acquire poll locks). 221 * 222 * pollunlock(void) releases whatever poll locks the current thread holds, 223 * returning a cookie for use by pollrelock(); 224 * 225 * pollrelock(cookie) reacquires previously dropped poll locks; 226 * 227 * polllock(php, mutex) does the common case: pollunlock(), 228 * acquire the problematic mutex, pollrelock(). 229 */ 230 int 231 pollunlock(void) 232 { 233 pollcache_t *pcp; 234 int lockstate = 0; 235 236 /* 237 * t_pollcache is set by /dev/poll and event ports (port_fd.c). 238 * If the pollrelock/pollunlock is called as a result of poll(2), 239 * the t_pollcache should be NULL. 240 */ 241 if (curthread->t_pollcache == NULL) 242 pcp = curthread->t_pollstate->ps_pcache; 243 else 244 pcp = curthread->t_pollcache; 245 246 if (mutex_owned(&pcp->pc_lock)) { 247 lockstate = 1; 248 mutex_exit(&pcp->pc_lock); 249 } 250 return (lockstate); 251 } 252 253 void 254 pollrelock(int lockstate) 255 { 256 pollcache_t *pcp; 257 258 /* 259 * t_pollcache is set by /dev/poll and event ports (port_fd.c). 260 * If the pollrelock/pollunlock is called as a result of poll(2), 261 * the t_pollcache should be NULL. 262 */ 263 if (curthread->t_pollcache == NULL) 264 pcp = curthread->t_pollstate->ps_pcache; 265 else 266 pcp = curthread->t_pollcache; 267 268 if (lockstate > 0) 269 mutex_enter(&pcp->pc_lock); 270 } 271 272 /* ARGSUSED */ 273 void 274 polllock(pollhead_t *php, kmutex_t *lp) 275 { 276 if (!mutex_tryenter(lp)) { 277 int lockstate = pollunlock(); 278 mutex_enter(lp); 279 pollrelock(lockstate); 280 } 281 } 282 283 static int 284 poll_common(pollfd_t *fds, nfds_t nfds, timespec_t *tsp, k_sigset_t *ksetp) 285 { 286 kthread_t *t = curthread; 287 klwp_t *lwp = ttolwp(t); 288 proc_t *p = ttoproc(t); 289 int fdcnt = 0; 290 int rval; 291 int i; 292 timespec_t *rqtp = NULL; 293 int timecheck = 0; 294 int imm_timeout = 0; 295 pollfd_t *pollfdp; 296 pollstate_t *ps; 297 pollcache_t *pcp; 298 int error = 0; 299 nfds_t old_nfds; 300 int cacheindex = 0; /* which cache set is used */ 301 302 /* 303 * Determine the precise future time of the requested timeout, if any. 304 */ 305 if (tsp != NULL) { 306 if (tsp->tv_sec == 0 && tsp->tv_nsec == 0) 307 imm_timeout = 1; 308 else { 309 timespec_t now; 310 timecheck = timechanged; 311 gethrestime(&now); 312 rqtp = tsp; 313 timespecadd(rqtp, &now); 314 } 315 } 316 317 /* 318 * Reset our signal mask, if requested. 319 */ 320 if (ksetp != NULL) { 321 mutex_enter(&p->p_lock); 322 schedctl_finish_sigblock(t); 323 lwp->lwp_sigoldmask = t->t_hold; 324 t->t_hold = *ksetp; 325 t->t_flag |= T_TOMASK; 326 /* 327 * Call cv_timedwait_sig() just to check for signals. 328 * We will return immediately with either 0 or -1. 329 */ 330 if (!cv_timedwait_sig(&t->t_delay_cv, &p->p_lock, lbolt)) { 331 mutex_exit(&p->p_lock); 332 error = EINTR; 333 goto pollout; 334 } 335 mutex_exit(&p->p_lock); 336 } 337 338 /* 339 * Check to see if this guy just wants to use poll() as a timeout. 340 * If yes then bypass all the other stuff and make him sleep. 341 */ 342 if (nfds == 0) { 343 /* 344 * Sleep until we have passed the requested future 345 * time or until interrupted by a signal. 346 * Do not check for signals if we have a zero timeout. 347 */ 348 if (!imm_timeout) { 349 mutex_enter(&t->t_delay_lock); 350 while ((rval = cv_waituntil_sig(&t->t_delay_cv, 351 &t->t_delay_lock, rqtp, timecheck)) > 0) 352 continue; 353 mutex_exit(&t->t_delay_lock); 354 if (rval == 0) 355 error = EINTR; 356 } 357 goto pollout; 358 } 359 360 if (nfds > p->p_fno_ctl) { 361 mutex_enter(&p->p_lock); 362 (void) rctl_action(rctlproc_legacy[RLIMIT_NOFILE], 363 p->p_rctls, p, RCA_SAFE); 364 mutex_exit(&p->p_lock); 365 error = EINVAL; 366 goto pollout; 367 } 368 369 /* 370 * Need to allocate memory for pollstate before anything because 371 * the mutex and cv are created in this space 372 */ 373 if ((ps = t->t_pollstate) == NULL) { 374 t->t_pollstate = pollstate_create(); 375 ps = t->t_pollstate; 376 } 377 378 if (ps->ps_pcache == NULL) 379 ps->ps_pcache = pcache_alloc(); 380 pcp = ps->ps_pcache; 381 382 /* 383 * NOTE: for performance, buffers are saved across poll() calls. 384 * The theory is that if a process polls heavily, it tends to poll 385 * on the same set of descriptors. Therefore, we only reallocate 386 * buffers when nfds changes. There is no hysteresis control, 387 * because there is no data to suggest that this is necessary; 388 * the penalty of reallocating is not *that* great in any event. 389 */ 390 old_nfds = ps->ps_nfds; 391 if (nfds != old_nfds) { 392 393 kmem_free(ps->ps_pollfd, old_nfds * sizeof (pollfd_t)); 394 pollfdp = kmem_alloc(nfds * sizeof (pollfd_t), KM_SLEEP); 395 ps->ps_pollfd = pollfdp; 396 ps->ps_nfds = nfds; 397 } 398 399 pollfdp = ps->ps_pollfd; 400 if (copyin(fds, pollfdp, nfds * sizeof (pollfd_t))) { 401 error = EFAULT; 402 goto pollout; 403 } 404 405 if (fds == NULL) { 406 /* 407 * If the process has page 0 mapped, then the copyin() above 408 * will succeed even if fds is NULL. However, our cached 409 * poll lists are keyed by the address of the passed-in fds 410 * structure, and we use the value NULL to indicate an unused 411 * poll cache list entry. As such, we elect not to support 412 * NULL as a valid (user) memory address and fail the poll() 413 * call. 414 */ 415 error = EINVAL; 416 goto pollout; 417 } 418 419 /* 420 * If this thread polls for the first time, allocate ALL poll 421 * cache data structures and cache the poll fd list. This 422 * allocation is delayed till now because lwp's polling 0 fd 423 * (i.e. using poll as timeout()) don't need this memory. 424 */ 425 mutex_enter(&ps->ps_lock); 426 pcp = ps->ps_pcache; 427 ASSERT(pcp != NULL); 428 if (pcp->pc_bitmap == NULL) { 429 pcache_create(pcp, nfds); 430 /* 431 * poll and cache this poll fd list in ps_pcacheset[0]. 432 */ 433 error = pcacheset_cache_list(ps, fds, &fdcnt, cacheindex); 434 if (fdcnt || error) { 435 mutex_exit(&ps->ps_lock); 436 goto pollout; 437 } 438 } else { 439 pollcacheset_t *pcset = ps->ps_pcacheset; 440 441 /* 442 * Not first time polling. Select a cached poll list by 443 * matching user pollfd list buffer address. 444 */ 445 for (cacheindex = 0; cacheindex < ps->ps_nsets; cacheindex++) { 446 if (pcset[cacheindex].pcs_usradr == (uintptr_t)fds) { 447 if ((++pcset[cacheindex].pcs_count) == 0) { 448 /* 449 * counter is wrapping around. 450 */ 451 pcacheset_reset_count(ps, cacheindex); 452 } 453 /* 454 * examine and resolve possible 455 * difference of the current poll 456 * list and previously cached one. 457 * If there is an error during resolve(), 458 * the callee will guarantee the consistency 459 * of cached poll list and cache content. 460 */ 461 error = pcacheset_resolve(ps, nfds, &fdcnt, 462 cacheindex); 463 if (error) { 464 mutex_exit(&ps->ps_lock); 465 goto pollout; 466 } 467 break; 468 } 469 470 /* 471 * Note that pcs_usradr field of an used entry won't be 472 * NULL because it stores the address of passed-in fds, 473 * and NULL fds will not be cached (Then it is either 474 * the special timeout case when nfds is 0 or it returns 475 * failure directly). 476 */ 477 if (pcset[cacheindex].pcs_usradr == NULL) { 478 /* 479 * found an unused entry. Use it to cache 480 * this poll list. 481 */ 482 error = pcacheset_cache_list(ps, fds, &fdcnt, 483 cacheindex); 484 if (fdcnt || error) { 485 mutex_exit(&ps->ps_lock); 486 goto pollout; 487 } 488 break; 489 } 490 } 491 if (cacheindex == ps->ps_nsets) { 492 /* 493 * We failed to find a matching cached poll fd list. 494 * replace an old list. 495 */ 496 pollstats.polllistmiss.value.ui64++; 497 cacheindex = pcacheset_replace(ps); 498 ASSERT(cacheindex < ps->ps_nsets); 499 pcset[cacheindex].pcs_usradr = (uintptr_t)fds; 500 error = pcacheset_resolve(ps, nfds, &fdcnt, cacheindex); 501 if (error) { 502 mutex_exit(&ps->ps_lock); 503 goto pollout; 504 } 505 } 506 } 507 508 /* 509 * Always scan the bitmap with the lock on the pollcache held. 510 * This is to make sure that a wakeup does not come undetected. 511 * If the lock is not held, a pollwakeup could have come for an 512 * fd we already checked but before this thread sleeps, in which 513 * case the wakeup is missed. Now we hold the pcache lock and 514 * check the bitmap again. This will prevent wakeup from happening 515 * while we hold pcache lock since pollwakeup() will also lock 516 * the pcache before updating poll bitmap. 517 */ 518 mutex_enter(&pcp->pc_lock); 519 for (;;) { 520 pcp->pc_flag = 0; 521 error = pcache_poll(pollfdp, ps, nfds, &fdcnt, cacheindex); 522 if (fdcnt || error) { 523 mutex_exit(&pcp->pc_lock); 524 mutex_exit(&ps->ps_lock); 525 break; 526 } 527 528 /* 529 * If T_POLLWAKE is set, a pollwakeup() was performed on 530 * one of the file descriptors. This can happen only if 531 * one of the VOP_POLL() functions dropped pcp->pc_lock. 532 * The only current cases of this is in procfs (prpoll()) 533 * and STREAMS (strpoll()). 534 */ 535 if (pcp->pc_flag & T_POLLWAKE) 536 continue; 537 538 /* 539 * If you get here, the poll of fds was unsuccessful. 540 * Wait until some fd becomes readable, writable, or gets 541 * an exception, or until a signal or a timeout occurs. 542 * Do not check for signals if we have a zero timeout. 543 */ 544 mutex_exit(&ps->ps_lock); 545 if (imm_timeout) 546 rval = -1; 547 else 548 rval = cv_waituntil_sig(&pcp->pc_cv, &pcp->pc_lock, 549 rqtp, timecheck); 550 mutex_exit(&pcp->pc_lock); 551 /* 552 * If we have received a signal or timed out 553 * then break out and return. 554 */ 555 if (rval <= 0) { 556 if (rval == 0) 557 error = EINTR; 558 break; 559 } 560 /* 561 * We have not received a signal or timed out. 562 * Continue around and poll fds again. 563 */ 564 mutex_enter(&ps->ps_lock); 565 mutex_enter(&pcp->pc_lock); 566 } 567 568 pollout: 569 /* 570 * If we changed the signal mask but we received 571 * no signal then restore the signal mask. 572 * Otherwise psig() will deal with the signal mask. 573 */ 574 if (ksetp != NULL) { 575 mutex_enter(&p->p_lock); 576 if (lwp->lwp_cursig == 0) { 577 t->t_hold = lwp->lwp_sigoldmask; 578 t->t_flag &= ~T_TOMASK; 579 } 580 mutex_exit(&p->p_lock); 581 } 582 583 if (error) 584 return (set_errno(error)); 585 586 /* 587 * Copy out the events and return the fdcnt to the user. 588 */ 589 if (nfds != 0 && 590 copyout(pollfdp, fds, nfds * sizeof (pollfd_t))) 591 return (set_errno(EFAULT)); 592 593 #ifdef DEBUG 594 /* 595 * Another sanity check: 596 */ 597 if (fdcnt) { 598 int reventcnt = 0; 599 600 for (i = 0; i < nfds; i++) { 601 if (pollfdp[i].fd < 0) { 602 ASSERT(pollfdp[i].revents == 0); 603 continue; 604 } 605 if (pollfdp[i].revents) { 606 reventcnt++; 607 } 608 } 609 ASSERT(fdcnt == reventcnt); 610 } else { 611 for (i = 0; i < nfds; i++) { 612 ASSERT(pollfdp[i].revents == 0); 613 } 614 } 615 #endif /* DEBUG */ 616 617 return (fdcnt); 618 } 619 620 /* 621 * This system call trap exists solely for binary compatibility with 622 * old statically-linked applications. It is not called from libc. 623 * It should be removed in the next release. 624 */ 625 int 626 poll(pollfd_t *fds, nfds_t nfds, int time_out) 627 { 628 timespec_t ts; 629 timespec_t *tsp; 630 631 if (time_out < 0) 632 tsp = NULL; 633 else { 634 ts.tv_sec = time_out / MILLISEC; 635 ts.tv_nsec = (time_out % MILLISEC) * MICROSEC; 636 tsp = &ts; 637 } 638 639 return (poll_common(fds, nfds, tsp, NULL)); 640 } 641 642 /* 643 * This is the system call trap that poll(), 644 * select() and pselect() are built upon. 645 * It is a private interface between libc and the kernel. 646 */ 647 int 648 pollsys(pollfd_t *fds, nfds_t nfds, timespec_t *timeoutp, sigset_t *setp) 649 { 650 timespec_t ts; 651 timespec_t *tsp; 652 sigset_t set; 653 k_sigset_t kset; 654 k_sigset_t *ksetp; 655 model_t datamodel = get_udatamodel(); 656 657 if (timeoutp == NULL) 658 tsp = NULL; 659 else { 660 if (datamodel == DATAMODEL_NATIVE) { 661 if (copyin(timeoutp, &ts, sizeof (ts))) 662 return (set_errno(EFAULT)); 663 } else { 664 timespec32_t ts32; 665 666 if (copyin(timeoutp, &ts32, sizeof (ts32))) 667 return (set_errno(EFAULT)); 668 TIMESPEC32_TO_TIMESPEC(&ts, &ts32) 669 } 670 671 if (itimerspecfix(&ts)) 672 return (set_errno(EINVAL)); 673 tsp = &ts; 674 } 675 676 if (setp == NULL) 677 ksetp = NULL; 678 else { 679 if (copyin(setp, &set, sizeof (set))) 680 return (set_errno(EFAULT)); 681 sigutok(&set, &kset); 682 ksetp = &kset; 683 } 684 685 return (poll_common(fds, nfds, tsp, ksetp)); 686 } 687 688 /* 689 * Clean up any state left around by poll(2). Called when a thread exits. 690 */ 691 void 692 pollcleanup() 693 { 694 pollstate_t *ps = curthread->t_pollstate; 695 pollcache_t *pcp; 696 697 if (ps == NULL) 698 return; 699 pcp = ps->ps_pcache; 700 /* 701 * free up all cached poll fds 702 */ 703 if (pcp == NULL) { 704 /* this pollstate is used by /dev/poll */ 705 goto pollcleanout; 706 } 707 708 if (pcp->pc_bitmap != NULL) { 709 ASSERT(MUTEX_NOT_HELD(&ps->ps_lock)); 710 /* 711 * a close lwp can race with us when cleaning up a polldat 712 * entry. We hold the ps_lock when cleaning hash table. 713 * Since this pollcache is going away anyway, there is no 714 * need to hold the pc_lock. 715 */ 716 mutex_enter(&ps->ps_lock); 717 pcache_clean(pcp); 718 mutex_exit(&ps->ps_lock); 719 #ifdef DEBUG 720 /* 721 * At this point, all fds cached by this lwp should be 722 * cleaned up. There should be no fd in fi_list still 723 * reference this thread. 724 */ 725 checkfpollinfo(); /* sanity check */ 726 pollcheckphlist(); /* sanity check */ 727 #endif /* DEBUG */ 728 } 729 /* 730 * Be sure no one is referencing thread before exiting 731 */ 732 mutex_enter(&pcp->pc_no_exit); 733 ASSERT(pcp->pc_busy >= 0); 734 while (pcp->pc_busy > 0) 735 cv_wait(&pcp->pc_busy_cv, &pcp->pc_no_exit); 736 mutex_exit(&pcp->pc_no_exit); 737 pollcleanout: 738 pollstate_destroy(ps); 739 curthread->t_pollstate = NULL; 740 } 741 742 /* 743 * pollwakeup() - poke threads waiting in poll() for some event 744 * on a particular object. 745 * 746 * The threads hanging off of the specified pollhead structure are scanned. 747 * If their event mask matches the specified event(s), then pollnotify() is 748 * called to poke the thread. 749 * 750 * Multiple events may be specified. When POLLHUP or POLLERR are specified, 751 * all waiting threads are poked. 752 * 753 * It is important that pollnotify() not drop the lock protecting the list 754 * of threads. 755 */ 756 void 757 pollwakeup(pollhead_t *php, short events_arg) 758 { 759 polldat_t *pdp; 760 int events = (ushort_t)events_arg; 761 struct plist { 762 port_t *pp; 763 int pevents; 764 struct plist *next; 765 }; 766 struct plist *plhead = NULL, *pltail = NULL; 767 768 retry: 769 PH_ENTER(php); 770 771 for (pdp = php->ph_list; pdp; pdp = pdp->pd_next) { 772 if ((pdp->pd_events & events) || 773 (events & (POLLHUP | POLLERR))) { 774 775 pollcache_t *pcp; 776 777 if (pdp->pd_portev != NULL) { 778 port_kevent_t *pkevp = pdp->pd_portev; 779 /* 780 * Object (fd) is associated with an event port, 781 * => send event notification to the port. 782 */ 783 ASSERT(pkevp->portkev_source == PORT_SOURCE_FD); 784 mutex_enter(&pkevp->portkev_lock); 785 if (pkevp->portkev_flags & PORT_KEV_VALID) { 786 int pevents; 787 788 pkevp->portkev_flags &= ~PORT_KEV_VALID; 789 pkevp->portkev_events |= events & 790 (pdp->pd_events | POLLHUP | 791 POLLERR); 792 /* 793 * portkev_lock mutex will be released 794 * by port_send_event(). 795 */ 796 port_send_event(pkevp); 797 798 /* 799 * If we have some thread polling the 800 * port's fd, add it to the list. They 801 * will be notified later. 802 * The port_pollwkup() will flag the 803 * port_t so that it will not disappear 804 * till port_pollwkdone() is called. 805 */ 806 pevents = 807 port_pollwkup(pkevp->portkev_port); 808 if (pevents) { 809 struct plist *t; 810 t = kmem_zalloc( 811 sizeof (struct plist), 812 KM_SLEEP); 813 t->pp = pkevp->portkev_port; 814 t->pevents = pevents; 815 if (plhead == NULL) { 816 plhead = t; 817 } else { 818 pltail->next = t; 819 } 820 pltail = t; 821 } 822 } else { 823 mutex_exit(&pkevp->portkev_lock); 824 } 825 continue; 826 } 827 828 pcp = pdp->pd_pcache; 829 830 /* 831 * Try to grab the lock for this thread. If 832 * we don't get it then we may deadlock so 833 * back out and restart all over again. Note 834 * that the failure rate is very very low. 835 */ 836 if (mutex_tryenter(&pcp->pc_lock)) { 837 pollnotify(pcp, pdp->pd_fd); 838 mutex_exit(&pcp->pc_lock); 839 } else { 840 /* 841 * We are here because: 842 * 1) This thread has been woke up 843 * and is trying to get out of poll(). 844 * 2) Some other thread is also here 845 * but with a different pollhead lock. 846 * 847 * So, we need to drop the lock on pollhead 848 * because of (1) but we want to prevent 849 * that thread from doing lwp_exit() or 850 * devpoll close. We want to ensure that 851 * the pollcache pointer is still invalid. 852 * 853 * Solution: Grab the pcp->pc_no_exit lock, 854 * increment the pc_busy counter, drop every 855 * lock in sight. Get out of the way and wait 856 * for type (2) threads to finish. 857 */ 858 859 mutex_enter(&pcp->pc_no_exit); 860 pcp->pc_busy++; /* prevents exit()'s */ 861 mutex_exit(&pcp->pc_no_exit); 862 863 PH_EXIT(php); 864 mutex_enter(&pcp->pc_lock); 865 mutex_exit(&pcp->pc_lock); 866 mutex_enter(&pcp->pc_no_exit); 867 pcp->pc_busy--; 868 if (pcp->pc_busy == 0) { 869 /* 870 * Wakeup the thread waiting in 871 * thread_exit(). 872 */ 873 cv_signal(&pcp->pc_busy_cv); 874 } 875 mutex_exit(&pcp->pc_no_exit); 876 goto retry; 877 } 878 } 879 } 880 881 882 /* 883 * Event ports - If this php is of the port on the list, 884 * call port_pollwkdone() to release it. The port_pollwkdone() 885 * needs to be called before dropping the PH lock so that any new 886 * thread attempting to poll this port are blocked. There can be 887 * only one thread here in pollwakeup notifying this port's fd. 888 */ 889 if (plhead != NULL && &plhead->pp->port_pollhd == php) { 890 struct plist *t; 891 port_pollwkdone(plhead->pp); 892 t = plhead; 893 plhead = plhead->next; 894 kmem_free(t, sizeof (struct plist)); 895 } 896 PH_EXIT(php); 897 898 /* 899 * Event ports - Notify threads polling the event port's fd. 900 * This is normally done in port_send_event() where it calls 901 * pollwakeup() on the port. But, for PORT_SOURCE_FD source alone, 902 * we do it here in pollwakeup() to avoid a recursive call. 903 */ 904 if (plhead != NULL) { 905 php = &plhead->pp->port_pollhd; 906 events = plhead->pevents; 907 goto retry; 908 } 909 } 910 911 /* 912 * This function is called to inform a thread that 913 * an event being polled for has occurred. 914 * The pollstate lock on the thread should be held on entry. 915 */ 916 void 917 pollnotify(pollcache_t *pcp, int fd) 918 { 919 ASSERT(fd < pcp->pc_mapsize); 920 ASSERT(MUTEX_HELD(&pcp->pc_lock)); 921 BT_SET(pcp->pc_bitmap, fd); 922 pcp->pc_flag |= T_POLLWAKE; 923 cv_signal(&pcp->pc_cv); 924 } 925 926 /* 927 * add a polldat entry to pollhead ph_list. The polldat struct is used 928 * by pollwakeup to wake sleeping pollers when polled events has happened. 929 */ 930 void 931 pollhead_insert(pollhead_t *php, polldat_t *pdp) 932 { 933 PH_ENTER(php); 934 ASSERT(pdp->pd_next == NULL); 935 #ifdef DEBUG 936 { 937 /* 938 * the polldat should not be already on the list 939 */ 940 polldat_t *wp; 941 for (wp = php->ph_list; wp; wp = wp->pd_next) { 942 ASSERT(wp != pdp); 943 } 944 } 945 #endif /* DEBUG */ 946 pdp->pd_next = php->ph_list; 947 php->ph_list = pdp; 948 PH_EXIT(php); 949 } 950 951 /* 952 * Delete the polldat entry from ph_list. 953 */ 954 void 955 pollhead_delete(pollhead_t *php, polldat_t *pdp) 956 { 957 polldat_t *wp; 958 polldat_t **wpp; 959 960 PH_ENTER(php); 961 for (wpp = &php->ph_list; (wp = *wpp) != NULL; wpp = &wp->pd_next) { 962 if (wp == pdp) { 963 *wpp = pdp->pd_next; 964 pdp->pd_next = NULL; 965 break; 966 } 967 } 968 #ifdef DEBUG 969 /* assert that pdp is no longer in the list */ 970 for (wp = *wpp; wp; wp = wp->pd_next) { 971 ASSERT(wp != pdp); 972 } 973 #endif /* DEBUG */ 974 PH_EXIT(php); 975 } 976 977 /* 978 * walk through the poll fd lists to see if they are identical. This is an 979 * expensive operation and should not be done more than once for each poll() 980 * call. 981 * 982 * As an optimization (i.e., not having to go through the lists more than 983 * once), this routine also clear the revents field of pollfd in 'current'. 984 * Zeroing out the revents field of each entry in current poll list is 985 * required by poll man page. 986 * 987 * Since the events field of cached list has illegal poll events filtered 988 * out, the current list applies the same filtering before comparison. 989 * 990 * The routine stops when it detects a meaningful difference, or when it 991 * exhausts the lists. 992 */ 993 int 994 pcacheset_cmp(pollfd_t *current, pollfd_t *cached, pollfd_t *newlist, int n) 995 { 996 int ix; 997 998 for (ix = 0; ix < n; ix++) { 999 if (current[ix].fd == cached[ix].fd) { 1000 /* 1001 * Filter out invalid poll events while we are in 1002 * inside the loop. 1003 */ 1004 if (current[ix].events & ~VALID_POLL_EVENTS) { 1005 current[ix].events &= VALID_POLL_EVENTS; 1006 if (newlist != NULL) 1007 newlist[ix].events = current[ix].events; 1008 } 1009 if (current[ix].events == cached[ix].events) { 1010 current[ix].revents = 0; 1011 continue; 1012 } 1013 } 1014 if ((current[ix].fd < 0) && (cached[ix].fd < 0)) { 1015 current[ix].revents = 0; 1016 continue; 1017 } 1018 return (ix); 1019 } 1020 return (ix); 1021 } 1022 1023 /* 1024 * This routine returns a pointer to a cached poll fd entry, or NULL if it 1025 * does not find it in the hash table. 1026 */ 1027 polldat_t * 1028 pcache_lookup_fd(pollcache_t *pcp, int fd) 1029 { 1030 int hashindex; 1031 polldat_t *pdp; 1032 1033 hashindex = POLLHASH(pcp->pc_hashsize, fd); 1034 pdp = pcp->pc_hash[hashindex]; 1035 while (pdp != NULL) { 1036 if (pdp->pd_fd == fd) 1037 break; 1038 pdp = pdp->pd_hashnext; 1039 } 1040 return (pdp); 1041 } 1042 1043 polldat_t * 1044 pcache_alloc_fd(int nsets) 1045 { 1046 polldat_t *pdp; 1047 1048 pdp = kmem_zalloc(sizeof (polldat_t), KM_SLEEP); 1049 if (nsets > 0) { 1050 pdp->pd_ref = kmem_zalloc(sizeof (xref_t) * nsets, KM_SLEEP); 1051 pdp->pd_nsets = nsets; 1052 } 1053 return (pdp); 1054 } 1055 1056 /* 1057 * This routine inserts a polldat into the pollcache's hash table. It 1058 * may be necessary to grow the size of the hash table. 1059 */ 1060 void 1061 pcache_insert_fd(pollcache_t *pcp, polldat_t *pdp, nfds_t nfds) 1062 { 1063 int hashindex; 1064 int fd; 1065 1066 if ((pcp->pc_fdcount > pcp->pc_hashsize * POLLHASHTHRESHOLD) || 1067 (nfds > pcp->pc_hashsize * POLLHASHTHRESHOLD)) { 1068 pcache_grow_hashtbl(pcp, nfds); 1069 } 1070 fd = pdp->pd_fd; 1071 hashindex = POLLHASH(pcp->pc_hashsize, fd); 1072 pdp->pd_hashnext = pcp->pc_hash[hashindex]; 1073 pcp->pc_hash[hashindex] = pdp; 1074 pcp->pc_fdcount++; 1075 1076 #ifdef DEBUG 1077 { 1078 /* 1079 * same fd should not appear on a hash list twice 1080 */ 1081 polldat_t *pdp1; 1082 for (pdp1 = pdp->pd_hashnext; pdp1; pdp1 = pdp1->pd_hashnext) { 1083 ASSERT(pdp->pd_fd != pdp1->pd_fd); 1084 } 1085 } 1086 #endif /* DEBUG */ 1087 } 1088 1089 /* 1090 * Grow the hash table -- either double the table size or round it to the 1091 * nearest multiples of POLLHASHCHUNKSZ, whichever is bigger. Rehash all the 1092 * elements on the hash table. 1093 */ 1094 void 1095 pcache_grow_hashtbl(pollcache_t *pcp, nfds_t nfds) 1096 { 1097 int oldsize; 1098 polldat_t **oldtbl; 1099 polldat_t *pdp, *pdp1; 1100 int i; 1101 #ifdef DEBUG 1102 int count = 0; 1103 #endif 1104 1105 ASSERT(pcp->pc_hashsize % POLLHASHCHUNKSZ == 0); 1106 oldsize = pcp->pc_hashsize; 1107 oldtbl = pcp->pc_hash; 1108 if (nfds > pcp->pc_hashsize * POLLHASHINC) { 1109 pcp->pc_hashsize = (nfds + POLLHASHCHUNKSZ - 1) & 1110 ~(POLLHASHCHUNKSZ - 1); 1111 } else { 1112 pcp->pc_hashsize = pcp->pc_hashsize * POLLHASHINC; 1113 } 1114 pcp->pc_hash = kmem_zalloc(pcp->pc_hashsize * sizeof (polldat_t *), 1115 KM_SLEEP); 1116 /* 1117 * rehash existing elements 1118 */ 1119 pcp->pc_fdcount = 0; 1120 for (i = 0; i < oldsize; i++) { 1121 pdp = oldtbl[i]; 1122 while (pdp != NULL) { 1123 pdp1 = pdp->pd_hashnext; 1124 pcache_insert_fd(pcp, pdp, nfds); 1125 pdp = pdp1; 1126 #ifdef DEBUG 1127 count++; 1128 #endif 1129 } 1130 } 1131 kmem_free(oldtbl, oldsize * sizeof (polldat_t *)); 1132 ASSERT(pcp->pc_fdcount == count); 1133 } 1134 1135 void 1136 pcache_grow_map(pollcache_t *pcp, int fd) 1137 { 1138 int newsize; 1139 ulong_t *newmap; 1140 1141 /* 1142 * grow to nearest multiple of POLLMAPCHUNK, assuming POLLMAPCHUNK is 1143 * power of 2. 1144 */ 1145 newsize = (fd + POLLMAPCHUNK) & ~(POLLMAPCHUNK - 1); 1146 newmap = kmem_zalloc((newsize / BT_NBIPUL) * sizeof (ulong_t), 1147 KM_SLEEP); 1148 /* 1149 * don't want pollwakeup to set a bit while growing the bitmap. 1150 */ 1151 ASSERT(mutex_owned(&pcp->pc_lock) == 0); 1152 mutex_enter(&pcp->pc_lock); 1153 bcopy(pcp->pc_bitmap, newmap, 1154 (pcp->pc_mapsize / BT_NBIPUL) * sizeof (ulong_t)); 1155 kmem_free(pcp->pc_bitmap, 1156 (pcp->pc_mapsize /BT_NBIPUL) * sizeof (ulong_t)); 1157 pcp->pc_bitmap = newmap; 1158 pcp->pc_mapsize = newsize; 1159 mutex_exit(&pcp->pc_lock); 1160 } 1161 1162 /* 1163 * remove all the reference from pollhead list and fpollinfo lists. 1164 */ 1165 void 1166 pcache_clean(pollcache_t *pcp) 1167 { 1168 int i; 1169 polldat_t **hashtbl; 1170 polldat_t *pdp; 1171 1172 ASSERT(MUTEX_HELD(&curthread->t_pollstate->ps_lock)); 1173 hashtbl = pcp->pc_hash; 1174 for (i = 0; i < pcp->pc_hashsize; i++) { 1175 for (pdp = hashtbl[i]; pdp; pdp = pdp->pd_hashnext) { 1176 if (pdp->pd_php != NULL) { 1177 pollhead_delete(pdp->pd_php, pdp); 1178 pdp->pd_php = NULL; 1179 } 1180 if (pdp->pd_fp != NULL) { 1181 delfpollinfo(pdp->pd_fd); 1182 pdp->pd_fp = NULL; 1183 } 1184 } 1185 } 1186 } 1187 1188 void 1189 pcacheset_invalidate(pollstate_t *ps, polldat_t *pdp) 1190 { 1191 int i; 1192 int fd = pdp->pd_fd; 1193 1194 /* 1195 * we come here because an earlier close() on this cached poll fd. 1196 */ 1197 ASSERT(pdp->pd_fp == NULL); 1198 ASSERT(MUTEX_HELD(&ps->ps_lock)); 1199 pdp->pd_events = 0; 1200 for (i = 0; i < ps->ps_nsets; i++) { 1201 xref_t *refp; 1202 pollcacheset_t *pcsp; 1203 1204 ASSERT(pdp->pd_ref != NULL); 1205 refp = &pdp->pd_ref[i]; 1206 if (refp->xf_refcnt) { 1207 ASSERT(refp->xf_position >= 0); 1208 pcsp = &ps->ps_pcacheset[i]; 1209 if (refp->xf_refcnt == 1) { 1210 pcsp->pcs_pollfd[refp->xf_position].fd = -1; 1211 refp->xf_refcnt = 0; 1212 pdp->pd_count--; 1213 } else if (refp->xf_refcnt > 1) { 1214 int j; 1215 1216 /* 1217 * turn off every appearance in pcs_pollfd list 1218 */ 1219 for (j = refp->xf_position; 1220 j < pcsp->pcs_nfds; j++) { 1221 if (pcsp->pcs_pollfd[j].fd == fd) { 1222 pcsp->pcs_pollfd[j].fd = -1; 1223 refp->xf_refcnt--; 1224 pdp->pd_count--; 1225 } 1226 } 1227 } 1228 ASSERT(refp->xf_refcnt == 0); 1229 refp->xf_position = POLLPOSINVAL; 1230 } 1231 } 1232 ASSERT(pdp->pd_count == 0); 1233 } 1234 1235 /* 1236 * Insert poll fd into the pollcache, and add poll registration. 1237 * This routine is called after getf() and before releasef(). So the vnode 1238 * can not disappear even if we block here. 1239 * If there is an error, the polled fd is not cached. 1240 */ 1241 int 1242 pcache_insert(pollstate_t *ps, file_t *fp, pollfd_t *pollfdp, int *fdcntp, 1243 ssize_t pos, int which) 1244 { 1245 pollcache_t *pcp = ps->ps_pcache; 1246 polldat_t *pdp; 1247 int error; 1248 int fd; 1249 pollhead_t *memphp = NULL; 1250 xref_t *refp; 1251 int newpollfd = 0; 1252 1253 ASSERT(MUTEX_HELD(&ps->ps_lock)); 1254 /* 1255 * The poll caching uses the existing VOP_POLL interface. If there 1256 * is no polled events, we want the polled device to set its "some 1257 * one is sleeping in poll" flag. When the polled events happen 1258 * later, the driver will call pollwakeup(). We achieve this by 1259 * always passing 0 in the third parameter ("anyyet") when calling 1260 * VOP_POLL. This parameter is not looked at by drivers when the 1261 * polled events exist. If a driver chooses to ignore this parameter 1262 * and call pollwakeup whenever the polled events happen, that will 1263 * be OK too. 1264 */ 1265 ASSERT(curthread->t_pollcache == NULL); 1266 error = VOP_POLL(fp->f_vnode, pollfdp->events, 0, &pollfdp->revents, 1267 &memphp, NULL); 1268 if (error) { 1269 return (error); 1270 } 1271 if (pollfdp->revents) { 1272 (*fdcntp)++; 1273 } 1274 /* 1275 * polling the underlying device succeeded. Now we can cache it. 1276 * A close can't come in here because we have not done a releasef() 1277 * yet. 1278 */ 1279 fd = pollfdp->fd; 1280 pdp = pcache_lookup_fd(pcp, fd); 1281 if (pdp == NULL) { 1282 ASSERT(ps->ps_nsets > 0); 1283 pdp = pcache_alloc_fd(ps->ps_nsets); 1284 newpollfd = 1; 1285 } 1286 /* 1287 * If this entry was used to cache a poll fd which was closed, and 1288 * this entry has not been cleaned, do it now. 1289 */ 1290 if ((pdp->pd_count > 0) && (pdp->pd_fp == NULL)) { 1291 pcacheset_invalidate(ps, pdp); 1292 ASSERT(pdp->pd_next == NULL); 1293 } 1294 if (pdp->pd_count == 0) { 1295 pdp->pd_fd = fd; 1296 pdp->pd_fp = fp; 1297 addfpollinfo(fd); 1298 pdp->pd_thread = curthread; 1299 pdp->pd_pcache = pcp; 1300 /* 1301 * the entry is never used or cleared by removing a cached 1302 * pollfd (pcache_delete_fd). So all the fields should be clear. 1303 */ 1304 ASSERT(pdp->pd_next == NULL); 1305 } 1306 1307 /* 1308 * A polled fd is considered cached. So there should be a fpollinfo 1309 * entry on uf_fpollinfo list. 1310 */ 1311 ASSERT(infpollinfo(fd)); 1312 /* 1313 * If there is an inconsistency, we want to know it here. 1314 */ 1315 ASSERT(pdp->pd_fp == fp); 1316 1317 /* 1318 * XXX pd_events is a union of all polled events on this fd, possibly 1319 * by different threads. Unless this is a new first poll(), pd_events 1320 * never shrinks. If an event is no longer polled by a process, there 1321 * is no way to cancel that event. In that case, poll degrade to its 1322 * old form -- polling on this fd every time poll() is called. The 1323 * assumption is an app always polls the same type of events. 1324 */ 1325 pdp->pd_events |= pollfdp->events; 1326 1327 pdp->pd_count++; 1328 /* 1329 * There is not much special handling for multiple appearances of 1330 * same fd other than xf_position always recording the first 1331 * appearance in poll list. If this is called from pcacheset_cache_list, 1332 * a VOP_POLL is called on every pollfd entry; therefore each 1333 * revents and fdcnt should be set correctly. If this is called from 1334 * pcacheset_resolve, we don't care about fdcnt here. Pollreadmap will 1335 * pick up the right count and handle revents field of each pollfd 1336 * entry. 1337 */ 1338 ASSERT(pdp->pd_ref != NULL); 1339 refp = &pdp->pd_ref[which]; 1340 if (refp->xf_refcnt == 0) { 1341 refp->xf_position = pos; 1342 } else { 1343 /* 1344 * xf_position records the fd's first appearance in poll list 1345 */ 1346 if (pos < refp->xf_position) { 1347 refp->xf_position = pos; 1348 } 1349 } 1350 ASSERT(pollfdp->fd == ps->ps_pollfd[refp->xf_position].fd); 1351 refp->xf_refcnt++; 1352 if (fd >= pcp->pc_mapsize) { 1353 pcache_grow_map(pcp, fd); 1354 } 1355 if (fd > pcp->pc_mapend) { 1356 pcp->pc_mapend = fd; 1357 } 1358 if (newpollfd != 0) { 1359 pcache_insert_fd(ps->ps_pcache, pdp, ps->ps_nfds); 1360 } 1361 if (memphp) { 1362 if (pdp->pd_php == NULL) { 1363 pollhead_insert(memphp, pdp); 1364 pdp->pd_php = memphp; 1365 } else { 1366 if (memphp != pdp->pd_php) { 1367 /* 1368 * layered devices (e.g. console driver) 1369 * may change the vnode and thus the pollhead 1370 * pointer out from underneath us. 1371 */ 1372 pollhead_delete(pdp->pd_php, pdp); 1373 pollhead_insert(memphp, pdp); 1374 pdp->pd_php = memphp; 1375 } 1376 } 1377 } 1378 /* 1379 * Since there is a considerable window between VOP_POLL and when 1380 * we actually put the polldat struct on the pollhead list, we could 1381 * miss a pollwakeup. In the case of polling additional events, we 1382 * don't update the events until after VOP_POLL. So we could miss 1383 * pollwakeup there too. So we always set the bit here just to be 1384 * safe. The real performance gain is in subsequent pcache_poll. 1385 */ 1386 mutex_enter(&pcp->pc_lock); 1387 BT_SET(pcp->pc_bitmap, fd); 1388 mutex_exit(&pcp->pc_lock); 1389 return (0); 1390 } 1391 1392 /* 1393 * The entry is not really deleted. The fields are cleared so that the 1394 * entry is no longer useful, but it will remain in the hash table for reuse 1395 * later. It will be freed when the polling lwp exits. 1396 */ 1397 int 1398 pcache_delete_fd(pollstate_t *ps, int fd, size_t pos, int which, uint_t cevent) 1399 { 1400 pollcache_t *pcp = ps->ps_pcache; 1401 polldat_t *pdp; 1402 xref_t *refp; 1403 1404 ASSERT(fd < pcp->pc_mapsize); 1405 ASSERT(MUTEX_HELD(&ps->ps_lock)); 1406 1407 pdp = pcache_lookup_fd(pcp, fd); 1408 ASSERT(pdp != NULL); 1409 ASSERT(pdp->pd_count > 0); 1410 ASSERT(pdp->pd_ref != NULL); 1411 refp = &pdp->pd_ref[which]; 1412 if (pdp->pd_count == 1) { 1413 pdp->pd_events = 0; 1414 refp->xf_position = POLLPOSINVAL; 1415 ASSERT(refp->xf_refcnt == 1); 1416 refp->xf_refcnt = 0; 1417 if (pdp->pd_php) { 1418 /* 1419 * It is possible for a wakeup thread to get ahead 1420 * of the following pollhead_delete and set the bit in 1421 * bitmap. It is OK because the bit will be cleared 1422 * here anyway. 1423 */ 1424 pollhead_delete(pdp->pd_php, pdp); 1425 pdp->pd_php = NULL; 1426 } 1427 pdp->pd_count = 0; 1428 if (pdp->pd_fp != NULL) { 1429 pdp->pd_fp = NULL; 1430 delfpollinfo(fd); 1431 } 1432 mutex_enter(&pcp->pc_lock); 1433 BT_CLEAR(pcp->pc_bitmap, fd); 1434 mutex_exit(&pcp->pc_lock); 1435 return (0); 1436 } 1437 if ((cevent & POLLCLOSED) == POLLCLOSED) { 1438 /* 1439 * fd cached here has been closed. This is the first 1440 * pcache_delete_fd called after the close. Clean up the 1441 * entire entry. 1442 */ 1443 pcacheset_invalidate(ps, pdp); 1444 ASSERT(pdp->pd_php == NULL); 1445 mutex_enter(&pcp->pc_lock); 1446 BT_CLEAR(pcp->pc_bitmap, fd); 1447 mutex_exit(&pcp->pc_lock); 1448 return (0); 1449 } 1450 #ifdef DEBUG 1451 if (getf(fd) != NULL) { 1452 ASSERT(infpollinfo(fd)); 1453 releasef(fd); 1454 } 1455 #endif /* DEBUG */ 1456 pdp->pd_count--; 1457 ASSERT(refp->xf_refcnt > 0); 1458 if (--refp->xf_refcnt == 0) { 1459 refp->xf_position = POLLPOSINVAL; 1460 } else { 1461 ASSERT(pos >= refp->xf_position); 1462 if (pos == refp->xf_position) { 1463 /* 1464 * The xref position is no longer valid. 1465 * Reset it to a special value and let 1466 * caller know it needs to updatexref() 1467 * with a new xf_position value. 1468 */ 1469 refp->xf_position = POLLPOSTRANS; 1470 return (1); 1471 } 1472 } 1473 return (0); 1474 } 1475 1476 void 1477 pcache_update_xref(pollcache_t *pcp, int fd, ssize_t pos, int which) 1478 { 1479 polldat_t *pdp; 1480 1481 pdp = pcache_lookup_fd(pcp, fd); 1482 ASSERT(pdp != NULL); 1483 ASSERT(pdp->pd_ref != NULL); 1484 pdp->pd_ref[which].xf_position = pos; 1485 } 1486 1487 #ifdef DEBUG 1488 /* 1489 * For each polled fd, it's either in the bitmap or cached in 1490 * pcache hash table. If this routine returns 0, something is wrong. 1491 */ 1492 static int 1493 pollchecksanity(pollstate_t *ps, nfds_t nfds) 1494 { 1495 int i; 1496 int fd; 1497 pollcache_t *pcp = ps->ps_pcache; 1498 polldat_t *pdp; 1499 pollfd_t *pollfdp = ps->ps_pollfd; 1500 file_t *fp; 1501 1502 ASSERT(MUTEX_HELD(&ps->ps_lock)); 1503 for (i = 0; i < nfds; i++) { 1504 fd = pollfdp[i].fd; 1505 if (fd < 0) { 1506 ASSERT(pollfdp[i].revents == 0); 1507 continue; 1508 } 1509 if (pollfdp[i].revents == POLLNVAL) 1510 continue; 1511 if ((fp = getf(fd)) == NULL) 1512 continue; 1513 pdp = pcache_lookup_fd(pcp, fd); 1514 ASSERT(pdp != NULL); 1515 ASSERT(infpollinfo(fd)); 1516 ASSERT(pdp->pd_fp == fp); 1517 releasef(fd); 1518 if (BT_TEST(pcp->pc_bitmap, fd)) 1519 continue; 1520 if (pdp->pd_php == NULL) 1521 return (0); 1522 } 1523 return (1); 1524 } 1525 #endif /* DEBUG */ 1526 1527 /* 1528 * resolve the difference between the current poll list and a cached one. 1529 */ 1530 int 1531 pcacheset_resolve(pollstate_t *ps, nfds_t nfds, int *fdcntp, int which) 1532 { 1533 int i; 1534 pollcache_t *pcp = ps->ps_pcache; 1535 pollfd_t *newlist = NULL; 1536 pollfd_t *current = ps->ps_pollfd; 1537 pollfd_t *cached; 1538 pollcacheset_t *pcsp; 1539 int common; 1540 int count = 0; 1541 int offset; 1542 int remain; 1543 int fd; 1544 file_t *fp; 1545 int fdcnt = 0; 1546 int cnt = 0; 1547 nfds_t old_nfds; 1548 int error = 0; 1549 int mismatch = 0; 1550 1551 ASSERT(MUTEX_HELD(&ps->ps_lock)); 1552 #ifdef DEBUG 1553 checkpolldat(ps); 1554 #endif 1555 pcsp = &ps->ps_pcacheset[which]; 1556 old_nfds = pcsp->pcs_nfds; 1557 common = (nfds > old_nfds) ? old_nfds : nfds; 1558 if (nfds != old_nfds) { 1559 /* 1560 * the length of poll list has changed. allocate a new 1561 * pollfd list. 1562 */ 1563 newlist = kmem_alloc(nfds * sizeof (pollfd_t), KM_SLEEP); 1564 bcopy(current, newlist, sizeof (pollfd_t) * nfds); 1565 } 1566 /* 1567 * Compare the overlapping part of the current fd list with the 1568 * cached one. Whenever a difference is found, resolve it. 1569 * The comparison is done on the current poll list and the 1570 * cached list. But we may be setting up the newlist to be the 1571 * cached list for next poll. 1572 */ 1573 cached = pcsp->pcs_pollfd; 1574 remain = common; 1575 1576 while (count < common) { 1577 int tmpfd; 1578 pollfd_t *np; 1579 1580 np = (newlist != NULL) ? &newlist[count] : NULL; 1581 offset = pcacheset_cmp(¤t[count], &cached[count], np, 1582 remain); 1583 /* 1584 * Collect stats. If lists are completed the first time, 1585 * it's a hit. Otherwise, it's a partial hit or miss. 1586 */ 1587 if ((count == 0) && (offset == common)) { 1588 pollstats.pollcachehit.value.ui64++; 1589 } else { 1590 mismatch++; 1591 } 1592 count += offset; 1593 if (offset < remain) { 1594 ASSERT(count < common); 1595 ASSERT((current[count].fd != cached[count].fd) || 1596 (current[count].events != cached[count].events)); 1597 /* 1598 * Filter out invalid events. 1599 */ 1600 if (current[count].events & ~VALID_POLL_EVENTS) { 1601 if (newlist != NULL) { 1602 newlist[count].events = 1603 current[count].events &= 1604 VALID_POLL_EVENTS; 1605 } else { 1606 current[count].events &= 1607 VALID_POLL_EVENTS; 1608 } 1609 } 1610 /* 1611 * when resolving a difference, we always remove the 1612 * fd from cache before inserting one into cache. 1613 */ 1614 if (cached[count].fd >= 0) { 1615 tmpfd = cached[count].fd; 1616 if (pcache_delete_fd(ps, tmpfd, count, which, 1617 (uint_t)cached[count].events)) { 1618 /* 1619 * This should be rare but needed for 1620 * correctness. 1621 * 1622 * The first appearance in cached list 1623 * is being "turned off". The same fd 1624 * appear more than once in the cached 1625 * poll list. Find the next one on the 1626 * list and update the cached 1627 * xf_position field. 1628 */ 1629 for (i = count + 1; i < old_nfds; i++) { 1630 if (cached[i].fd == tmpfd) { 1631 pcache_update_xref(pcp, 1632 tmpfd, (ssize_t)i, 1633 which); 1634 break; 1635 } 1636 } 1637 ASSERT(i <= old_nfds); 1638 } 1639 /* 1640 * In case a new cache list is allocated, 1641 * need to keep both cache lists in sync 1642 * b/c the new one can be freed if we have 1643 * an error later. 1644 */ 1645 cached[count].fd = -1; 1646 if (newlist != NULL) { 1647 newlist[count].fd = -1; 1648 } 1649 } 1650 if ((tmpfd = current[count].fd) >= 0) { 1651 /* 1652 * add to the cached fd tbl and bitmap. 1653 */ 1654 if ((fp = getf(tmpfd)) == NULL) { 1655 current[count].revents = POLLNVAL; 1656 if (newlist != NULL) { 1657 newlist[count].fd = -1; 1658 } 1659 cached[count].fd = -1; 1660 fdcnt++; 1661 } else { 1662 /* 1663 * Here we don't care about the 1664 * fdcnt. We will examine the bitmap 1665 * later and pick up the correct 1666 * fdcnt there. So we never bother 1667 * to check value of 'cnt'. 1668 */ 1669 error = pcache_insert(ps, fp, 1670 ¤t[count], &cnt, 1671 (ssize_t)count, which); 1672 /* 1673 * if no error, we want to do releasef 1674 * after we updated cache poll list 1675 * entry so that close() won't race 1676 * us. 1677 */ 1678 if (error) { 1679 /* 1680 * If we encountered an error, 1681 * we have invalidated an 1682 * entry in cached poll list 1683 * (in pcache_delete_fd() above) 1684 * but failed to add one here. 1685 * This is OK b/c what's in the 1686 * cached list is consistent 1687 * with content of cache. 1688 * It will not have any ill 1689 * effect on next poll(). 1690 */ 1691 releasef(tmpfd); 1692 if (newlist != NULL) { 1693 kmem_free(newlist, 1694 nfds * 1695 sizeof (pollfd_t)); 1696 } 1697 return (error); 1698 } 1699 /* 1700 * If we have allocated a new(temp) 1701 * cache list, we need to keep both 1702 * in sync b/c the new one can be freed 1703 * if we have an error later. 1704 */ 1705 if (newlist != NULL) { 1706 newlist[count].fd = 1707 current[count].fd; 1708 newlist[count].events = 1709 current[count].events; 1710 } 1711 cached[count].fd = current[count].fd; 1712 cached[count].events = 1713 current[count].events; 1714 releasef(tmpfd); 1715 } 1716 } else { 1717 current[count].revents = 0; 1718 } 1719 count++; 1720 remain = common - count; 1721 } 1722 } 1723 if (mismatch != 0) { 1724 if (mismatch == common) { 1725 pollstats.pollcachemiss.value.ui64++; 1726 } else { 1727 pollstats.pollcachephit.value.ui64++; 1728 } 1729 } 1730 /* 1731 * take care of the non overlapping part of a list 1732 */ 1733 if (nfds > old_nfds) { 1734 ASSERT(newlist != NULL); 1735 for (i = old_nfds; i < nfds; i++) { 1736 /* filter out invalid events */ 1737 if (current[i].events & ~VALID_POLL_EVENTS) { 1738 newlist[i].events = current[i].events = 1739 current[i].events & VALID_POLL_EVENTS; 1740 } 1741 if ((fd = current[i].fd) < 0) { 1742 current[i].revents = 0; 1743 continue; 1744 } 1745 /* 1746 * add to the cached fd tbl and bitmap. 1747 */ 1748 if ((fp = getf(fd)) == NULL) { 1749 current[i].revents = POLLNVAL; 1750 newlist[i].fd = -1; 1751 fdcnt++; 1752 continue; 1753 } 1754 /* 1755 * Here we don't care about the 1756 * fdcnt. We will examine the bitmap 1757 * later and pick up the correct 1758 * fdcnt there. So we never bother to 1759 * check 'cnt'. 1760 */ 1761 error = pcache_insert(ps, fp, ¤t[i], &cnt, 1762 (ssize_t)i, which); 1763 releasef(fd); 1764 if (error) { 1765 /* 1766 * Here we are half way through adding newly 1767 * polled fd. Undo enough to keep the cache 1768 * list consistent with the cache content. 1769 */ 1770 pcacheset_remove_list(ps, current, old_nfds, 1771 i, which, 0); 1772 kmem_free(newlist, nfds * sizeof (pollfd_t)); 1773 return (error); 1774 } 1775 } 1776 } 1777 if (old_nfds > nfds) { 1778 /* 1779 * remove the fd's which are no longer polled. 1780 */ 1781 pcacheset_remove_list(ps, pcsp->pcs_pollfd, nfds, old_nfds, 1782 which, 1); 1783 } 1784 /* 1785 * set difference resolved. update nfds and cachedlist 1786 * in pollstate struct. 1787 */ 1788 if (newlist != NULL) { 1789 kmem_free(pcsp->pcs_pollfd, old_nfds * sizeof (pollfd_t)); 1790 /* 1791 * By now, the pollfd.revents field should 1792 * all be zeroed. 1793 */ 1794 pcsp->pcs_pollfd = newlist; 1795 pcsp->pcs_nfds = nfds; 1796 } 1797 ASSERT(*fdcntp == 0); 1798 *fdcntp = fdcnt; 1799 /* 1800 * By now for every fd in pollfdp, one of the following should be 1801 * true. Otherwise we will miss a polled event. 1802 * 1803 * 1. the bit corresponding to the fd in bitmap is set. So VOP_POLL 1804 * will be called on this fd in next poll. 1805 * 2. the fd is cached in the pcache (i.e. pd_php is set). So 1806 * pollnotify will happen. 1807 */ 1808 ASSERT(pollchecksanity(ps, nfds)); 1809 /* 1810 * make sure cross reference between cached poll lists and cached 1811 * poll fds are correct. 1812 */ 1813 ASSERT(pollcheckxref(ps, which)); 1814 /* 1815 * ensure each polldat in pollcache reference a polled fd in 1816 * pollcacheset. 1817 */ 1818 #ifdef DEBUG 1819 checkpolldat(ps); 1820 #endif 1821 return (0); 1822 } 1823 1824 #ifdef DEBUG 1825 static int 1826 pollscanrevents(pollcache_t *pcp, pollfd_t *pollfdp, nfds_t nfds) 1827 { 1828 int i; 1829 int reventcnt = 0; 1830 1831 for (i = 0; i < nfds; i++) { 1832 if (pollfdp[i].fd < 0) { 1833 ASSERT(pollfdp[i].revents == 0); 1834 continue; 1835 } 1836 if (pollfdp[i].revents) { 1837 reventcnt++; 1838 } 1839 if (pollfdp[i].revents && (pollfdp[i].revents != POLLNVAL)) { 1840 ASSERT(BT_TEST(pcp->pc_bitmap, pollfdp[i].fd)); 1841 } 1842 } 1843 return (reventcnt); 1844 } 1845 #endif /* DEBUG */ 1846 1847 /* 1848 * read the bitmap and poll on fds corresponding to the '1' bits. The ps_lock 1849 * is held upon entry. 1850 */ 1851 int 1852 pcache_poll(pollfd_t *pollfdp, pollstate_t *ps, nfds_t nfds, int *fdcntp, 1853 int which) 1854 { 1855 int i; 1856 pollcache_t *pcp; 1857 int fd; 1858 int begin, end, done; 1859 pollhead_t *php; 1860 int fdcnt; 1861 int error = 0; 1862 file_t *fp; 1863 polldat_t *pdp; 1864 xref_t *refp; 1865 int entry; 1866 1867 pcp = ps->ps_pcache; 1868 ASSERT(MUTEX_HELD(&ps->ps_lock)); 1869 ASSERT(MUTEX_HELD(&pcp->pc_lock)); 1870 retry: 1871 done = 0; 1872 begin = 0; 1873 fdcnt = 0; 1874 end = pcp->pc_mapend; 1875 while ((fdcnt < nfds) && !done) { 1876 php = NULL; 1877 /* 1878 * only poll fds which may have events 1879 */ 1880 fd = bt_getlowbit(pcp->pc_bitmap, begin, end); 1881 ASSERT(fd <= end); 1882 if (fd >= 0) { 1883 ASSERT(pollcheckrevents(ps, begin, fd, which)); 1884 /* 1885 * adjust map pointers for next round 1886 */ 1887 if (fd == end) { 1888 done = 1; 1889 } else { 1890 begin = fd + 1; 1891 } 1892 /* 1893 * A bitmap caches poll state information of 1894 * multiple poll lists. Call VOP_POLL only if 1895 * the bit corresponds to an fd in this poll 1896 * list. 1897 */ 1898 pdp = pcache_lookup_fd(pcp, fd); 1899 ASSERT(pdp != NULL); 1900 ASSERT(pdp->pd_ref != NULL); 1901 refp = &pdp->pd_ref[which]; 1902 if (refp->xf_refcnt == 0) 1903 continue; 1904 entry = refp->xf_position; 1905 ASSERT((entry >= 0) && (entry < nfds)); 1906 ASSERT(pollfdp[entry].fd == fd); 1907 /* 1908 * we are in this routine implies that we have 1909 * successfully polled this fd in the past. 1910 * Check to see this fd is closed while we are 1911 * blocked in poll. This ensures that we don't 1912 * miss a close on the fd in the case this fd is 1913 * reused. 1914 */ 1915 if (pdp->pd_fp == NULL) { 1916 ASSERT(pdp->pd_count > 0); 1917 pollfdp[entry].revents = POLLNVAL; 1918 fdcnt++; 1919 if (refp->xf_refcnt > 1) { 1920 /* 1921 * this fd appeared multiple time 1922 * in the poll list. Find all of them. 1923 */ 1924 for (i = entry + 1; i < nfds; i++) { 1925 if (pollfdp[i].fd == fd) { 1926 pollfdp[i].revents = 1927 POLLNVAL; 1928 fdcnt++; 1929 } 1930 } 1931 } 1932 pcacheset_invalidate(ps, pdp); 1933 continue; 1934 } 1935 /* 1936 * We can be here polling a device that is being 1937 * closed (i.e. the file pointer is set to NULL, 1938 * but pollcacheclean has not happened yet). 1939 */ 1940 if ((fp = getf(fd)) == NULL) { 1941 pollfdp[entry].revents = POLLNVAL; 1942 fdcnt++; 1943 if (refp->xf_refcnt > 1) { 1944 /* 1945 * this fd appeared multiple time 1946 * in the poll list. Find all of them. 1947 */ 1948 for (i = entry + 1; i < nfds; i++) { 1949 if (pollfdp[i].fd == fd) { 1950 pollfdp[i].revents = 1951 POLLNVAL; 1952 fdcnt++; 1953 } 1954 } 1955 } 1956 continue; 1957 } 1958 ASSERT(pdp->pd_fp == fp); 1959 ASSERT(infpollinfo(fd)); 1960 /* 1961 * Since we no longer hold poll head lock across 1962 * VOP_POLL, pollunlock logic can be simplifed. 1963 */ 1964 ASSERT(pdp->pd_php == NULL || 1965 MUTEX_NOT_HELD(PHLOCK(pdp->pd_php))); 1966 /* 1967 * underlying file systems may set a "pollpending" 1968 * flag when it sees the poll may block. Pollwakeup() 1969 * is called by wakeup thread if pollpending is set. 1970 * Pass a 0 fdcnt so that the underlying file system 1971 * will set the "pollpending" flag set when there is 1972 * no polled events. 1973 * 1974 * Use pollfdp[].events for actual polling because 1975 * the pd_events is union of all cached poll events 1976 * on this fd. The events parameter also affects 1977 * how the polled device sets the "poll pending" 1978 * flag. 1979 */ 1980 ASSERT(curthread->t_pollcache == NULL); 1981 error = VOP_POLL(fp->f_vnode, pollfdp[entry].events, 0, 1982 &pollfdp[entry].revents, &php, NULL); 1983 /* 1984 * releasef after completely done with this cached 1985 * poll entry. To prevent close() coming in to clear 1986 * this entry. 1987 */ 1988 if (error) { 1989 releasef(fd); 1990 break; 1991 } 1992 /* 1993 * layered devices (e.g. console driver) 1994 * may change the vnode and thus the pollhead 1995 * pointer out from underneath us. 1996 */ 1997 if (php != NULL && pdp->pd_php != NULL && 1998 php != pdp->pd_php) { 1999 releasef(fd); 2000 pollhead_delete(pdp->pd_php, pdp); 2001 pdp->pd_php = php; 2002 pollhead_insert(php, pdp); 2003 /* 2004 * We could have missed a wakeup on the new 2005 * target device. Make sure the new target 2006 * gets polled once. 2007 */ 2008 BT_SET(pcp->pc_bitmap, fd); 2009 goto retry; 2010 } 2011 2012 if (pollfdp[entry].revents) { 2013 ASSERT(refp->xf_refcnt >= 1); 2014 fdcnt++; 2015 if (refp->xf_refcnt > 1) { 2016 /* 2017 * this fd appeared multiple time 2018 * in the poll list. This is rare but 2019 * we have to look at all of them for 2020 * correctness. 2021 */ 2022 error = plist_chkdupfd(fp, pdp, ps, 2023 pollfdp, entry, &fdcnt); 2024 if (error > 0) { 2025 releasef(fd); 2026 break; 2027 } 2028 if (error < 0) { 2029 goto retry; 2030 } 2031 } 2032 releasef(fd); 2033 } else { 2034 /* 2035 * VOP_POLL didn't return any revents. We can 2036 * clear the bit in bitmap only if we have the 2037 * pollhead ptr cached and no other cached 2038 * entry is polling different events on this fd. 2039 * VOP_POLL may have dropped the ps_lock. Make 2040 * sure pollwakeup has not happened before clear 2041 * the bit. 2042 */ 2043 if ((pdp->pd_php != NULL) && 2044 (pollfdp[entry].events == pdp->pd_events) && 2045 ((pcp->pc_flag & T_POLLWAKE) == 0)) { 2046 BT_CLEAR(pcp->pc_bitmap, fd); 2047 } 2048 /* 2049 * if the fd can be cached now but not before, 2050 * do it now. 2051 */ 2052 if ((pdp->pd_php == NULL) && (php != NULL)) { 2053 pdp->pd_php = php; 2054 pollhead_insert(php, pdp); 2055 /* 2056 * We are inserting a polldat struct for 2057 * the first time. We may have missed a 2058 * wakeup on this device. Re-poll once. 2059 * This should be a rare event. 2060 */ 2061 releasef(fd); 2062 goto retry; 2063 } 2064 if (refp->xf_refcnt > 1) { 2065 /* 2066 * this fd appeared multiple time 2067 * in the poll list. This is rare but 2068 * we have to look at all of them for 2069 * correctness. 2070 */ 2071 error = plist_chkdupfd(fp, pdp, ps, 2072 pollfdp, entry, &fdcnt); 2073 if (error > 0) { 2074 releasef(fd); 2075 break; 2076 } 2077 if (error < 0) { 2078 goto retry; 2079 } 2080 } 2081 releasef(fd); 2082 } 2083 } else { 2084 done = 1; 2085 ASSERT(pollcheckrevents(ps, begin, end + 1, which)); 2086 } 2087 } 2088 if (!error) { 2089 ASSERT(*fdcntp + fdcnt == pollscanrevents(pcp, pollfdp, nfds)); 2090 *fdcntp += fdcnt; 2091 } 2092 return (error); 2093 } 2094 2095 /* 2096 * Going through the poll list without much locking. Poll all fds and 2097 * cache all valid fds in the pollcache. 2098 */ 2099 int 2100 pcacheset_cache_list(pollstate_t *ps, pollfd_t *fds, int *fdcntp, int which) 2101 { 2102 pollfd_t *pollfdp = ps->ps_pollfd; 2103 pollcacheset_t *pcacheset = ps->ps_pcacheset; 2104 pollfd_t *newfdlist; 2105 int i; 2106 int fd; 2107 file_t *fp; 2108 int error = 0; 2109 2110 ASSERT(MUTEX_HELD(&ps->ps_lock)); 2111 ASSERT(which < ps->ps_nsets); 2112 ASSERT(pcacheset != NULL); 2113 ASSERT(pcacheset[which].pcs_pollfd == NULL); 2114 newfdlist = kmem_alloc(ps->ps_nfds * sizeof (pollfd_t), KM_SLEEP); 2115 /* 2116 * cache the new poll list in pollcachset. 2117 */ 2118 bcopy(pollfdp, newfdlist, sizeof (pollfd_t) * ps->ps_nfds); 2119 2120 pcacheset[which].pcs_pollfd = newfdlist; 2121 pcacheset[which].pcs_nfds = ps->ps_nfds; 2122 pcacheset[which].pcs_usradr = (uintptr_t)fds; 2123 2124 /* 2125 * We have saved a copy of current poll fd list in one pollcacheset. 2126 * The 'revents' field of the new list is not yet set to 0. Loop 2127 * through the new list just to do that is expensive. We do that 2128 * while polling the list. 2129 */ 2130 for (i = 0; i < ps->ps_nfds; i++) { 2131 fd = pollfdp[i].fd; 2132 /* 2133 * We also filter out the illegal poll events in the event 2134 * field for the cached poll list/set. 2135 */ 2136 if (pollfdp[i].events & ~VALID_POLL_EVENTS) { 2137 newfdlist[i].events = pollfdp[i].events = 2138 pollfdp[i].events & VALID_POLL_EVENTS; 2139 } 2140 if (fd < 0) { 2141 pollfdp[i].revents = 0; 2142 continue; 2143 } 2144 if ((fp = getf(fd)) == NULL) { 2145 pollfdp[i].revents = POLLNVAL; 2146 /* 2147 * invalidate this cache entry in the cached poll list 2148 */ 2149 newfdlist[i].fd = -1; 2150 (*fdcntp)++; 2151 continue; 2152 } 2153 /* 2154 * cache this fd. 2155 */ 2156 error = pcache_insert(ps, fp, &pollfdp[i], fdcntp, (ssize_t)i, 2157 which); 2158 releasef(fd); 2159 if (error) { 2160 /* 2161 * Here we are half way through caching a new 2162 * poll list. Undo every thing. 2163 */ 2164 pcacheset_remove_list(ps, pollfdp, 0, i, which, 0); 2165 kmem_free(newfdlist, ps->ps_nfds * sizeof (pollfd_t)); 2166 pcacheset[which].pcs_pollfd = NULL; 2167 pcacheset[which].pcs_usradr = NULL; 2168 break; 2169 } 2170 } 2171 return (error); 2172 } 2173 2174 /* 2175 * called by pollcacheclean() to set the fp NULL. It also sets polled events 2176 * in pcacheset entries to a special events 'POLLCLOSED'. Do a pollwakeup to 2177 * wake any sleeping poller, then remove the polldat from the driver. 2178 * The routine is called with ps_pcachelock held. 2179 */ 2180 void 2181 pcache_clean_entry(pollstate_t *ps, int fd) 2182 { 2183 pollcache_t *pcp; 2184 polldat_t *pdp; 2185 int i; 2186 2187 ASSERT(ps != NULL); 2188 ASSERT(MUTEX_HELD(&ps->ps_lock)); 2189 pcp = ps->ps_pcache; 2190 ASSERT(pcp); 2191 pdp = pcache_lookup_fd(pcp, fd); 2192 ASSERT(pdp != NULL); 2193 /* 2194 * the corresponding fpollinfo in fi_list has been removed by 2195 * a close on this fd. Reset the cached fp ptr here. 2196 */ 2197 pdp->pd_fp = NULL; 2198 /* 2199 * XXX - This routine also touches data in pcacheset struct. 2200 * 2201 * set the event in cached poll lists to POLLCLOSED. This invalidate 2202 * the cached poll fd entry in that poll list, which will force a 2203 * removal of this cached entry in next poll(). The cleanup is done 2204 * at the removal time. 2205 */ 2206 ASSERT(pdp->pd_ref != NULL); 2207 for (i = 0; i < ps->ps_nsets; i++) { 2208 xref_t *refp; 2209 pollcacheset_t *pcsp; 2210 2211 refp = &pdp->pd_ref[i]; 2212 if (refp->xf_refcnt) { 2213 ASSERT(refp->xf_position >= 0); 2214 pcsp = &ps->ps_pcacheset[i]; 2215 if (refp->xf_refcnt == 1) { 2216 pcsp->pcs_pollfd[refp->xf_position].events = 2217 (short)POLLCLOSED; 2218 } 2219 if (refp->xf_refcnt > 1) { 2220 int j; 2221 /* 2222 * mark every matching entry in pcs_pollfd 2223 */ 2224 for (j = refp->xf_position; 2225 j < pcsp->pcs_nfds; j++) { 2226 if (pcsp->pcs_pollfd[j].fd == fd) { 2227 pcsp->pcs_pollfd[j].events = 2228 (short)POLLCLOSED; 2229 } 2230 } 2231 } 2232 } 2233 } 2234 if (pdp->pd_php) { 2235 pollwakeup(pdp->pd_php, POLLHUP); 2236 pollhead_delete(pdp->pd_php, pdp); 2237 pdp->pd_php = NULL; 2238 } 2239 } 2240 2241 /* 2242 * This is the first time this thread has ever polled, 2243 * so we have to create its pollstate structure. 2244 * This will persist for the life of the thread, 2245 * until it calls pollcleanup(). 2246 */ 2247 pollstate_t * 2248 pollstate_create(void) 2249 { 2250 pollstate_t *ps; 2251 2252 ps = kmem_zalloc(sizeof (pollstate_t), KM_SLEEP); 2253 ps->ps_nsets = POLLFDSETS; 2254 ps->ps_pcacheset = pcacheset_create(ps->ps_nsets); 2255 return (ps); 2256 } 2257 2258 void 2259 pollstate_destroy(pollstate_t *ps) 2260 { 2261 if (ps->ps_pollfd != NULL) { 2262 kmem_free(ps->ps_pollfd, ps->ps_nfds * sizeof (pollfd_t)); 2263 ps->ps_pollfd = NULL; 2264 } 2265 if (ps->ps_pcache != NULL) { 2266 pcache_destroy(ps->ps_pcache); 2267 ps->ps_pcache = NULL; 2268 } 2269 pcacheset_destroy(ps->ps_pcacheset, ps->ps_nsets); 2270 ps->ps_pcacheset = NULL; 2271 if (ps->ps_dpbuf != NULL) { 2272 kmem_free(ps->ps_dpbuf, ps->ps_dpbufsize * sizeof (pollfd_t)); 2273 ps->ps_dpbuf = NULL; 2274 } 2275 mutex_destroy(&ps->ps_lock); 2276 kmem_free(ps, sizeof (pollstate_t)); 2277 } 2278 2279 /* 2280 * We are holding the appropriate uf_lock entering this routine. 2281 * Bump up the ps_busy count to prevent the thread from exiting. 2282 */ 2283 void 2284 pollblockexit(fpollinfo_t *fpip) 2285 { 2286 for (; fpip; fpip = fpip->fp_next) { 2287 pollcache_t *pcp = fpip->fp_thread->t_pollstate->ps_pcache; 2288 2289 mutex_enter(&pcp->pc_no_exit); 2290 pcp->pc_busy++; /* prevents exit()'s */ 2291 mutex_exit(&pcp->pc_no_exit); 2292 } 2293 } 2294 2295 /* 2296 * Complete phase 2 of cached poll fd cleanup. Call pcache_clean_entry to mark 2297 * the pcacheset events field POLLCLOSED to force the next poll() to remove 2298 * this cache entry. We can't clean the polldat entry clean up here because 2299 * lwp block in poll() needs the info to return. Wakeup anyone blocked in 2300 * poll and let exiting lwp go. No lock is help upon entry. So it's OK for 2301 * pcache_clean_entry to call pollwakeup(). 2302 */ 2303 void 2304 pollcacheclean(fpollinfo_t *fip, int fd) 2305 { 2306 struct fpollinfo *fpip, *fpip2; 2307 2308 fpip = fip; 2309 while (fpip) { 2310 pollstate_t *ps = fpip->fp_thread->t_pollstate; 2311 pollcache_t *pcp = ps->ps_pcache; 2312 2313 mutex_enter(&ps->ps_lock); 2314 pcache_clean_entry(ps, fd); 2315 mutex_exit(&ps->ps_lock); 2316 mutex_enter(&pcp->pc_no_exit); 2317 pcp->pc_busy--; 2318 if (pcp->pc_busy == 0) { 2319 /* 2320 * Wakeup the thread waiting in 2321 * thread_exit(). 2322 */ 2323 cv_signal(&pcp->pc_busy_cv); 2324 } 2325 mutex_exit(&pcp->pc_no_exit); 2326 2327 fpip2 = fpip; 2328 fpip = fpip->fp_next; 2329 kmem_free(fpip2, sizeof (fpollinfo_t)); 2330 } 2331 } 2332 2333 /* 2334 * one of the cache line's counter is wrapping around. Reset all cache line 2335 * counters to zero except one. This is simplistic, but probably works 2336 * effectively. 2337 */ 2338 void 2339 pcacheset_reset_count(pollstate_t *ps, int index) 2340 { 2341 int i; 2342 2343 ASSERT(MUTEX_HELD(&ps->ps_lock)); 2344 for (i = 0; i < ps->ps_nsets; i++) { 2345 if (ps->ps_pcacheset[i].pcs_pollfd != NULL) { 2346 ps->ps_pcacheset[i].pcs_count = 0; 2347 } 2348 } 2349 ps->ps_pcacheset[index].pcs_count = 1; 2350 } 2351 2352 /* 2353 * this routine implements poll cache list replacement policy. 2354 * It is currently choose the "least used". 2355 */ 2356 int 2357 pcacheset_replace(pollstate_t *ps) 2358 { 2359 int i; 2360 int index = 0; 2361 2362 ASSERT(MUTEX_HELD(&ps->ps_lock)); 2363 for (i = 1; i < ps->ps_nsets; i++) { 2364 if (ps->ps_pcacheset[index].pcs_count > 2365 ps->ps_pcacheset[i].pcs_count) { 2366 index = i; 2367 } 2368 } 2369 ps->ps_pcacheset[index].pcs_count = 0; 2370 return (index); 2371 } 2372 2373 /* 2374 * this routine is called by strclose to remove remaining polldat struct on 2375 * the pollhead list of the device being closed. There are two reasons as why 2376 * the polldat structures still remain on the pollhead list: 2377 * 2378 * (1) The layered device(e.g.the console driver). 2379 * In this case, the existence of a polldat implies that the thread putting 2380 * the polldat on this list has not exited yet. Before the thread exits, it 2381 * will have to hold this pollhead lock to remove the polldat. So holding the 2382 * pollhead lock here effectively prevents the thread which put the polldat 2383 * on this list from exiting. 2384 * 2385 * (2) /dev/poll. 2386 * When a polled fd is cached in /dev/poll, its polldat will remain on the 2387 * pollhead list if the process has not done a POLLREMOVE before closing the 2388 * polled fd. We just unlink it here. 2389 */ 2390 void 2391 pollhead_clean(pollhead_t *php) 2392 { 2393 polldat_t *pdp; 2394 2395 /* 2396 * In case(1), while we must prevent the thread in question from 2397 * exiting, we must also obey the proper locking order, i.e. 2398 * (ps_lock -> phlock). 2399 */ 2400 PH_ENTER(php); 2401 while (php->ph_list != NULL) { 2402 pollstate_t *ps; 2403 pollcache_t *pcp; 2404 2405 pdp = php->ph_list; 2406 ASSERT(pdp->pd_php == php); 2407 if (pdp->pd_thread == NULL) { 2408 /* 2409 * This is case(2). Since the ph_lock is sufficient 2410 * to synchronize this lwp with any other /dev/poll 2411 * lwp, just unlink the polldat. 2412 */ 2413 php->ph_list = pdp->pd_next; 2414 pdp->pd_php = NULL; 2415 pdp->pd_next = NULL; 2416 continue; 2417 } 2418 ps = pdp->pd_thread->t_pollstate; 2419 ASSERT(ps != NULL); 2420 pcp = pdp->pd_pcache; 2421 ASSERT(pcp != NULL); 2422 mutex_enter(&pcp->pc_no_exit); 2423 pcp->pc_busy++; /* prevents exit()'s */ 2424 mutex_exit(&pcp->pc_no_exit); 2425 /* 2426 * Now get the locks in proper order to avoid deadlock. 2427 */ 2428 PH_EXIT(php); 2429 mutex_enter(&ps->ps_lock); 2430 /* 2431 * while we dropped the pollhead lock, the element could be 2432 * taken off the list already. 2433 */ 2434 PH_ENTER(php); 2435 if (pdp->pd_php == php) { 2436 ASSERT(pdp == php->ph_list); 2437 php->ph_list = pdp->pd_next; 2438 pdp->pd_php = NULL; 2439 pdp->pd_next = NULL; 2440 } 2441 PH_EXIT(php); 2442 mutex_exit(&ps->ps_lock); 2443 mutex_enter(&pcp->pc_no_exit); 2444 pcp->pc_busy--; 2445 if (pcp->pc_busy == 0) { 2446 /* 2447 * Wakeup the thread waiting in 2448 * thread_exit(). 2449 */ 2450 cv_signal(&pcp->pc_busy_cv); 2451 } 2452 mutex_exit(&pcp->pc_no_exit); 2453 PH_ENTER(php); 2454 } 2455 PH_EXIT(php); 2456 } 2457 2458 /* 2459 * The remove_list is called to cleanup a partially cached 'current' list or 2460 * to remove a partial list which is no longer cached. The flag value of 1 2461 * indicates the second case. 2462 */ 2463 void 2464 pcacheset_remove_list(pollstate_t *ps, pollfd_t *pollfdp, int start, int end, 2465 int cacheindex, int flag) 2466 { 2467 int i; 2468 2469 ASSERT(MUTEX_HELD(&ps->ps_lock)); 2470 for (i = start; i < end; i++) { 2471 if ((pollfdp[i].fd >= 0) && 2472 (flag || !(pollfdp[i].revents & POLLNVAL))) { 2473 if (pcache_delete_fd(ps, pollfdp[i].fd, i, cacheindex, 2474 (uint_t)pollfdp[i].events)) { 2475 int j; 2476 int fd = pollfdp[i].fd; 2477 2478 for (j = i + 1; j < end; j++) { 2479 if (pollfdp[j].fd == fd) { 2480 pcache_update_xref( 2481 ps->ps_pcache, fd, 2482 (ssize_t)j, cacheindex); 2483 break; 2484 } 2485 } 2486 ASSERT(j <= end); 2487 } 2488 } 2489 } 2490 } 2491 2492 #ifdef DEBUG 2493 2494 #include<sys/strsubr.h> 2495 /* 2496 * make sure curthread is not on anyone's pollhead list any more. 2497 */ 2498 static void 2499 pollcheckphlist() 2500 { 2501 int i; 2502 file_t *fp; 2503 uf_entry_t *ufp; 2504 uf_info_t *fip = P_FINFO(curproc); 2505 struct stdata *stp; 2506 polldat_t *pdp; 2507 2508 mutex_enter(&fip->fi_lock); 2509 for (i = 0; i < fip->fi_nfiles; i++) { 2510 UF_ENTER(ufp, fip, i); 2511 if ((fp = ufp->uf_file) != NULL) { 2512 if ((stp = fp->f_vnode->v_stream) != NULL) { 2513 PH_ENTER(&stp->sd_pollist); 2514 pdp = stp->sd_pollist.ph_list; 2515 while (pdp) { 2516 ASSERT(pdp->pd_thread != curthread); 2517 pdp = pdp->pd_next; 2518 } 2519 PH_EXIT(&stp->sd_pollist); 2520 } 2521 } 2522 UF_EXIT(ufp); 2523 } 2524 mutex_exit(&fip->fi_lock); 2525 } 2526 2527 /* 2528 * for resolved set poll list, the xref info in the pcache should be 2529 * consistent with this poll list. 2530 */ 2531 static int 2532 pollcheckxref(pollstate_t *ps, int cacheindex) 2533 { 2534 pollfd_t *pollfdp = ps->ps_pcacheset[cacheindex].pcs_pollfd; 2535 pollcache_t *pcp = ps->ps_pcache; 2536 polldat_t *pdp; 2537 int i; 2538 xref_t *refp; 2539 2540 for (i = 0; i < ps->ps_pcacheset[cacheindex].pcs_nfds; i++) { 2541 if (pollfdp[i].fd < 0) { 2542 continue; 2543 } 2544 pdp = pcache_lookup_fd(pcp, pollfdp[i].fd); 2545 ASSERT(pdp != NULL); 2546 ASSERT(pdp->pd_ref != NULL); 2547 refp = &pdp->pd_ref[cacheindex]; 2548 if (refp->xf_position >= 0) { 2549 ASSERT(refp->xf_refcnt >= 1); 2550 ASSERT(pollfdp[refp->xf_position].fd == pdp->pd_fd); 2551 if (refp->xf_refcnt > 1) { 2552 int j; 2553 int count = 0; 2554 2555 for (j = refp->xf_position; 2556 j < ps->ps_pcacheset[cacheindex].pcs_nfds; 2557 j++) { 2558 if (pollfdp[j].fd == pdp->pd_fd) { 2559 count++; 2560 } 2561 } 2562 ASSERT(count == refp->xf_refcnt); 2563 } 2564 } 2565 } 2566 return (1); 2567 } 2568 2569 /* 2570 * For every cached pollfd, its polldat struct should be consistent with 2571 * what is in the pcacheset lists. 2572 */ 2573 static void 2574 checkpolldat(pollstate_t *ps) 2575 { 2576 pollcache_t *pcp = ps->ps_pcache; 2577 polldat_t **hashtbl; 2578 int i; 2579 2580 hashtbl = pcp->pc_hash; 2581 for (i = 0; i < pcp->pc_hashsize; i++) { 2582 polldat_t *pdp; 2583 2584 for (pdp = hashtbl[i]; pdp; pdp = pdp->pd_hashnext) { 2585 ASSERT(pdp->pd_ref != NULL); 2586 if (pdp->pd_count > 0) { 2587 xref_t *refp; 2588 int j; 2589 pollcacheset_t *pcsp; 2590 pollfd_t *pollfd; 2591 2592 for (j = 0; j < ps->ps_nsets; j++) { 2593 refp = &pdp->pd_ref[j]; 2594 if (refp->xf_refcnt > 0) { 2595 pcsp = &ps->ps_pcacheset[j]; 2596 ASSERT(refp->xf_position < pcsp->pcs_nfds); 2597 pollfd = pcsp->pcs_pollfd; 2598 ASSERT(pdp->pd_fd == pollfd[refp->xf_position].fd); 2599 } 2600 } 2601 } 2602 } 2603 } 2604 } 2605 2606 /* 2607 * every wfd element on ph_list must have a corresponding fpollinfo on the 2608 * uf_fpollinfo list. This is a variation of infpollinfo() w/o holding locks. 2609 */ 2610 void 2611 checkwfdlist(vnode_t *vp, fpollinfo_t *fpip) 2612 { 2613 stdata_t *stp; 2614 polldat_t *pdp; 2615 fpollinfo_t *fpip2; 2616 2617 if ((stp = vp->v_stream) == NULL) { 2618 return; 2619 } 2620 PH_ENTER(&stp->sd_pollist); 2621 for (pdp = stp->sd_pollist.ph_list; pdp; pdp = pdp->pd_next) { 2622 if (pdp->pd_thread->t_procp == curthread->t_procp) { 2623 for (fpip2 = fpip; fpip2; fpip2 = fpip2->fp_next) { 2624 if (pdp->pd_thread == fpip2->fp_thread) { 2625 break; 2626 } 2627 } 2628 ASSERT(fpip2 != NULL); 2629 } 2630 } 2631 PH_EXIT(&stp->sd_pollist); 2632 } 2633 2634 /* 2635 * For each cached fd whose bit is not set in bitmap, its revents field in 2636 * current poll list should be 0. 2637 */ 2638 static int 2639 pollcheckrevents(pollstate_t *ps, int begin, int end, int cacheindex) 2640 { 2641 pollcache_t *pcp = ps->ps_pcache; 2642 pollfd_t *pollfdp = ps->ps_pollfd; 2643 int i; 2644 2645 for (i = begin; i < end; i++) { 2646 polldat_t *pdp; 2647 2648 ASSERT(!BT_TEST(pcp->pc_bitmap, i)); 2649 pdp = pcache_lookup_fd(pcp, i); 2650 if (pdp && pdp->pd_fp != NULL) { 2651 xref_t *refp; 2652 int entry; 2653 2654 ASSERT(pdp->pd_ref != NULL); 2655 refp = &pdp->pd_ref[cacheindex]; 2656 if (refp->xf_refcnt == 0) { 2657 continue; 2658 } 2659 entry = refp->xf_position; 2660 ASSERT(entry >= 0); 2661 ASSERT(pollfdp[entry].revents == 0); 2662 if (refp->xf_refcnt > 1) { 2663 int j; 2664 2665 for (j = entry + 1; j < ps->ps_nfds; j++) { 2666 if (pollfdp[j].fd == i) { 2667 ASSERT(pollfdp[j].revents == 0); 2668 } 2669 } 2670 } 2671 } 2672 } 2673 return (1); 2674 } 2675 2676 #endif /* DEBUG */ 2677 2678 pollcache_t * 2679 pcache_alloc() 2680 { 2681 return (kmem_zalloc(sizeof (pollcache_t), KM_SLEEP)); 2682 } 2683 2684 void 2685 pcache_create(pollcache_t *pcp, nfds_t nfds) 2686 { 2687 size_t mapsize; 2688 2689 /* 2690 * allocate enough bits for the poll fd list 2691 */ 2692 if ((mapsize = POLLMAPCHUNK) <= nfds) { 2693 mapsize = (nfds + POLLMAPCHUNK - 1) & ~(POLLMAPCHUNK - 1); 2694 } 2695 pcp->pc_bitmap = kmem_zalloc((mapsize / BT_NBIPUL) * sizeof (ulong_t), 2696 KM_SLEEP); 2697 pcp->pc_mapsize = mapsize; 2698 /* 2699 * The hash size is at least POLLHASHCHUNKSZ. If user polls a large 2700 * number of fd to start with, allocate a bigger hash table (to the 2701 * nearest multiple of POLLHASHCHUNKSZ) because dynamically growing a 2702 * hash table is expensive. 2703 */ 2704 if (nfds < POLLHASHCHUNKSZ) { 2705 pcp->pc_hashsize = POLLHASHCHUNKSZ; 2706 } else { 2707 pcp->pc_hashsize = (nfds + POLLHASHCHUNKSZ - 1) & 2708 ~(POLLHASHCHUNKSZ - 1); 2709 } 2710 pcp->pc_hash = kmem_zalloc(pcp->pc_hashsize * sizeof (polldat_t *), 2711 KM_SLEEP); 2712 } 2713 2714 void 2715 pcache_destroy(pollcache_t *pcp) 2716 { 2717 polldat_t **hashtbl; 2718 int i; 2719 2720 hashtbl = pcp->pc_hash; 2721 for (i = 0; i < pcp->pc_hashsize; i++) { 2722 if (hashtbl[i] != NULL) { 2723 polldat_t *pdp, *pdp2; 2724 2725 pdp = hashtbl[i]; 2726 while (pdp != NULL) { 2727 pdp2 = pdp->pd_hashnext; 2728 if (pdp->pd_ref != NULL) { 2729 kmem_free(pdp->pd_ref, sizeof (xref_t) * 2730 pdp->pd_nsets); 2731 } 2732 kmem_free(pdp, sizeof (polldat_t)); 2733 pdp = pdp2; 2734 pcp->pc_fdcount--; 2735 } 2736 } 2737 } 2738 ASSERT(pcp->pc_fdcount == 0); 2739 kmem_free(pcp->pc_hash, sizeof (polldat_t *) * pcp->pc_hashsize); 2740 kmem_free(pcp->pc_bitmap, 2741 sizeof (ulong_t) * (pcp->pc_mapsize/BT_NBIPUL)); 2742 mutex_destroy(&pcp->pc_no_exit); 2743 mutex_destroy(&pcp->pc_lock); 2744 cv_destroy(&pcp->pc_cv); 2745 cv_destroy(&pcp->pc_busy_cv); 2746 kmem_free(pcp, sizeof (pollcache_t)); 2747 } 2748 2749 pollcacheset_t * 2750 pcacheset_create(int nsets) 2751 { 2752 return (kmem_zalloc(sizeof (pollcacheset_t) * nsets, KM_SLEEP)); 2753 } 2754 2755 void 2756 pcacheset_destroy(pollcacheset_t *pcsp, int nsets) 2757 { 2758 int i; 2759 2760 for (i = 0; i < nsets; i++) { 2761 if (pcsp[i].pcs_pollfd != NULL) { 2762 kmem_free(pcsp[i].pcs_pollfd, pcsp[i].pcs_nfds * 2763 sizeof (pollfd_t)); 2764 } 2765 } 2766 kmem_free(pcsp, sizeof (pollcacheset_t) * nsets); 2767 } 2768 2769 /* 2770 * Check each duplicated poll fd in the poll list. It may be necessary to 2771 * VOP_POLL the same fd again using different poll events. getf() has been 2772 * done by caller. This routine returns 0 if it can sucessfully process the 2773 * entire poll fd list. It returns -1 if underlying vnode has changed during 2774 * a VOP_POLL, in which case the caller has to repoll. It returns a positive 2775 * value if VOP_POLL failed. 2776 */ 2777 static int 2778 plist_chkdupfd(file_t *fp, polldat_t *pdp, pollstate_t *psp, pollfd_t *pollfdp, 2779 int entry, int *fdcntp) 2780 { 2781 int i; 2782 int fd; 2783 nfds_t nfds = psp->ps_nfds; 2784 2785 fd = pollfdp[entry].fd; 2786 for (i = entry + 1; i < nfds; i++) { 2787 if (pollfdp[i].fd == fd) { 2788 if (pollfdp[i].events == pollfdp[entry].events) { 2789 if ((pollfdp[i].revents = 2790 pollfdp[entry].revents) != 0) { 2791 (*fdcntp)++; 2792 } 2793 } else { 2794 2795 int error; 2796 pollhead_t *php; 2797 pollcache_t *pcp = psp->ps_pcache; 2798 2799 /* 2800 * the events are different. VOP_POLL on this 2801 * fd so that we don't miss any revents. 2802 */ 2803 php = NULL; 2804 ASSERT(curthread->t_pollcache == NULL); 2805 error = VOP_POLL(fp->f_vnode, 2806 pollfdp[i].events, 0, 2807 &pollfdp[i].revents, &php, NULL); 2808 if (error) { 2809 return (error); 2810 } 2811 /* 2812 * layered devices(e.g. console driver) 2813 * may change the vnode and thus the pollhead 2814 * pointer out from underneath us. 2815 */ 2816 if (php != NULL && pdp->pd_php != NULL && 2817 php != pdp->pd_php) { 2818 pollhead_delete(pdp->pd_php, pdp); 2819 pdp->pd_php = php; 2820 pollhead_insert(php, pdp); 2821 /* 2822 * We could have missed a wakeup on the 2823 * new target device. Make sure the new 2824 * target gets polled once. 2825 */ 2826 BT_SET(pcp->pc_bitmap, fd); 2827 return (-1); 2828 } 2829 if (pollfdp[i].revents) { 2830 (*fdcntp)++; 2831 } 2832 } 2833 } 2834 } 2835 return (0); 2836 } 2837