xref: /titanic_50/usr/src/uts/common/io/devpoll.c (revision bf75909a55b2efbe96ab2fe820866fc77fab58c9)
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  * Copyright 2008 Sun Microsystems, Inc.  All rights reserved.
23  * Use is subject to license terms.
24  */
25 
26 /*
27  * Copyright (c) 2012 by Delphix. All rights reserved.
28  * Copyright 2016 Joyent, Inc.
29  */
30 
31 #include <sys/types.h>
32 #include <sys/devops.h>
33 #include <sys/conf.h>
34 #include <sys/modctl.h>
35 #include <sys/sunddi.h>
36 #include <sys/stat.h>
37 #include <sys/poll_impl.h>
38 #include <sys/errno.h>
39 #include <sys/kmem.h>
40 #include <sys/mkdev.h>
41 #include <sys/debug.h>
42 #include <sys/file.h>
43 #include <sys/sysmacros.h>
44 #include <sys/systm.h>
45 #include <sys/bitmap.h>
46 #include <sys/devpoll.h>
47 #include <sys/rctl.h>
48 #include <sys/resource.h>
49 #include <sys/schedctl.h>
50 #include <sys/epoll.h>
51 
52 #define	RESERVED	1
53 
54 /* local data struct */
55 static	dp_entry_t	**devpolltbl;	/* dev poll entries */
56 static	size_t		dptblsize;
57 
58 static	kmutex_t	devpoll_lock;	/* lock protecting dev tbl */
59 int			devpoll_init;	/* is /dev/poll initialized already */
60 
61 /* device local functions */
62 
63 static int dpopen(dev_t *devp, int flag, int otyp, cred_t *credp);
64 static int dpwrite(dev_t dev, struct uio *uiop, cred_t *credp);
65 static int dpioctl(dev_t dev, int cmd, intptr_t arg, int mode, cred_t *credp,
66     int *rvalp);
67 static int dppoll(dev_t dev, short events, int anyyet, short *reventsp,
68     struct pollhead **phpp);
69 static int dpclose(dev_t dev, int flag, int otyp, cred_t *credp);
70 static dev_info_t *dpdevi;
71 
72 
73 static struct cb_ops    dp_cb_ops = {
74 	dpopen,			/* open */
75 	dpclose,		/* close */
76 	nodev,			/* strategy */
77 	nodev,			/* print */
78 	nodev,			/* dump */
79 	nodev,			/* read */
80 	dpwrite,		/* write */
81 	dpioctl,		/* ioctl */
82 	nodev,			/* devmap */
83 	nodev,			/* mmap */
84 	nodev,			/* segmap */
85 	dppoll,			/* poll */
86 	ddi_prop_op,		/* prop_op */
87 	(struct streamtab *)0,	/* streamtab */
88 	D_MP,			/* flags */
89 	CB_REV,			/* cb_ops revision */
90 	nodev,			/* aread */
91 	nodev			/* awrite */
92 };
93 
94 static int dpattach(dev_info_t *, ddi_attach_cmd_t);
95 static int dpdetach(dev_info_t *, ddi_detach_cmd_t);
96 static int dpinfo(dev_info_t *, ddi_info_cmd_t, void *, void **);
97 
98 static struct dev_ops dp_ops = {
99 	DEVO_REV,		/* devo_rev */
100 	0,			/* refcnt */
101 	dpinfo,			/* info */
102 	nulldev,		/* identify */
103 	nulldev,		/* probe */
104 	dpattach,		/* attach */
105 	dpdetach,		/* detach */
106 	nodev,			/* reset */
107 	&dp_cb_ops,		/* driver operations */
108 	(struct bus_ops *)NULL, /* bus operations */
109 	nulldev,		/* power */
110 	ddi_quiesce_not_needed,		/* quiesce */
111 };
112 
113 
114 static struct modldrv modldrv = {
115 	&mod_driverops,		/* type of module - a driver */
116 	"/dev/poll driver",
117 	&dp_ops,
118 };
119 
120 static struct modlinkage modlinkage = {
121 	MODREV_1,
122 	(void *)&modldrv,
123 	NULL
124 };
125 
126 static void pcachelink_assoc(pollcache_t *, pollcache_t *);
127 static void pcachelink_mark_stale(pollcache_t *);
128 static void pcachelink_purge_stale(pollcache_t *);
129 static void pcachelink_purge_all(pollcache_t *);
130 
131 
132 /*
133  * Locking Design
134  *
135  * The /dev/poll driver shares most of its code with poll sys call whose
136  * code is in common/syscall/poll.c. In poll(2) design, the pollcache
137  * structure is per lwp. An implicit assumption is made there that some
138  * portion of pollcache will never be touched by other lwps. E.g., in
139  * poll(2) design, no lwp will ever need to grow bitmap of other lwp.
140  * This assumption is not true for /dev/poll; hence the need for extra
141  * locking.
142  *
143  * To allow more parallelism, each /dev/poll file descriptor (indexed by
144  * minor number) has its own lock. Since read (dpioctl) is a much more
145  * frequent operation than write, we want to allow multiple reads on same
146  * /dev/poll fd. However, we prevent writes from being starved by giving
147  * priority to write operation. Theoretically writes can starve reads as
148  * well. But in practical sense this is not important because (1) writes
149  * happens less often than reads, and (2) write operation defines the
150  * content of poll fd a cache set. If writes happens so often that they
151  * can starve reads, that means the cached set is very unstable. It may
152  * not make sense to read an unstable cache set anyway. Therefore, the
153  * writers starving readers case is not handled in this design.
154  */
155 
156 int
_init()157 _init()
158 {
159 	int	error;
160 
161 	dptblsize = DEVPOLLSIZE;
162 	devpolltbl = kmem_zalloc(sizeof (caddr_t) * dptblsize, KM_SLEEP);
163 	mutex_init(&devpoll_lock, NULL, MUTEX_DEFAULT, NULL);
164 	devpoll_init = 1;
165 	if ((error = mod_install(&modlinkage)) != 0) {
166 		kmem_free(devpolltbl, sizeof (caddr_t) * dptblsize);
167 		devpoll_init = 0;
168 	}
169 	return (error);
170 }
171 
172 int
_fini()173 _fini()
174 {
175 	int error;
176 
177 	if ((error = mod_remove(&modlinkage)) != 0) {
178 		return (error);
179 	}
180 	mutex_destroy(&devpoll_lock);
181 	kmem_free(devpolltbl, sizeof (caddr_t) * dptblsize);
182 	return (0);
183 }
184 
185 int
_info(struct modinfo * modinfop)186 _info(struct modinfo *modinfop)
187 {
188 	return (mod_info(&modlinkage, modinfop));
189 }
190 
191 /*ARGSUSED*/
192 static int
dpattach(dev_info_t * devi,ddi_attach_cmd_t cmd)193 dpattach(dev_info_t *devi, ddi_attach_cmd_t cmd)
194 {
195 	if (ddi_create_minor_node(devi, "poll", S_IFCHR, 0, DDI_PSEUDO, NULL)
196 	    == DDI_FAILURE) {
197 		ddi_remove_minor_node(devi, NULL);
198 		return (DDI_FAILURE);
199 	}
200 	dpdevi = devi;
201 	return (DDI_SUCCESS);
202 }
203 
204 static int
dpdetach(dev_info_t * devi,ddi_detach_cmd_t cmd)205 dpdetach(dev_info_t *devi, ddi_detach_cmd_t cmd)
206 {
207 	if (cmd != DDI_DETACH)
208 		return (DDI_FAILURE);
209 
210 	ddi_remove_minor_node(devi, NULL);
211 	return (DDI_SUCCESS);
212 }
213 
214 /* ARGSUSED */
215 static int
dpinfo(dev_info_t * dip,ddi_info_cmd_t infocmd,void * arg,void ** result)216 dpinfo(dev_info_t *dip, ddi_info_cmd_t infocmd, void *arg, void **result)
217 {
218 	int error;
219 
220 	switch (infocmd) {
221 	case DDI_INFO_DEVT2DEVINFO:
222 		*result = (void *)dpdevi;
223 		error = DDI_SUCCESS;
224 		break;
225 	case DDI_INFO_DEVT2INSTANCE:
226 		*result = (void *)0;
227 		error = DDI_SUCCESS;
228 		break;
229 	default:
230 		error = DDI_FAILURE;
231 	}
232 	return (error);
233 }
234 
235 /*
236  * dp_pcache_poll has similar logic to pcache_poll() in poll.c. The major
237  * differences are: (1) /dev/poll requires scanning the bitmap starting at
238  * where it was stopped last time, instead of always starting from 0,
239  * (2) since user may not have cleaned up the cached fds when they are
240  * closed, some polldats in cache may refer to closed or reused fds. We
241  * need to check for those cases.
242  *
243  * NOTE: Upon closing an fd, automatic poll cache cleanup is done for
244  *	 poll(2) caches but NOT for /dev/poll caches. So expect some
245  *	 stale entries!
246  */
247 static int
dp_pcache_poll(dp_entry_t * dpep,void * dpbuf,pollcache_t * pcp,nfds_t nfds,int * fdcntp)248 dp_pcache_poll(dp_entry_t *dpep, void *dpbuf,
249     pollcache_t *pcp, nfds_t nfds, int *fdcntp)
250 {
251 	int		start, ostart, end;
252 	int		fdcnt, fd;
253 	boolean_t	done;
254 	file_t		*fp;
255 	short		revent;
256 	boolean_t	no_wrap;
257 	pollhead_t	*php;
258 	polldat_t	*pdp;
259 	pollfd_t	*pfdp;
260 	epoll_event_t	*epoll;
261 	int		error = 0;
262 	short		mask = POLLRDHUP | POLLWRBAND;
263 	boolean_t	is_epoll = (dpep->dpe_flag & DP_ISEPOLLCOMPAT) != 0;
264 
265 	ASSERT(MUTEX_HELD(&pcp->pc_lock));
266 	if (pcp->pc_bitmap == NULL) {
267 		/*
268 		 * No Need to search because no poll fd
269 		 * has been cached.
270 		 */
271 		return (error);
272 	}
273 
274 	if (is_epoll) {
275 		pfdp = NULL;
276 		epoll = (epoll_event_t *)dpbuf;
277 	} else {
278 		pfdp = (pollfd_t *)dpbuf;
279 		epoll = NULL;
280 	}
281 retry:
282 	start = ostart = pcp->pc_mapstart;
283 	end = pcp->pc_mapend;
284 	php = NULL;
285 
286 	if (start == 0) {
287 		/*
288 		 * started from every begining, no need to wrap around.
289 		 */
290 		no_wrap = B_TRUE;
291 	} else {
292 		no_wrap = B_FALSE;
293 	}
294 	done = B_FALSE;
295 	fdcnt = 0;
296 	while ((fdcnt < nfds) && !done) {
297 		php = NULL;
298 		revent = 0;
299 		/*
300 		 * Examine the bit map in a circular fashion
301 		 * to avoid starvation. Always resume from
302 		 * last stop. Scan till end of the map. Then
303 		 * wrap around.
304 		 */
305 		fd = bt_getlowbit(pcp->pc_bitmap, start, end);
306 		ASSERT(fd <= end);
307 		if (fd >= 0) {
308 			if (fd == end) {
309 				if (no_wrap) {
310 					done = B_TRUE;
311 				} else {
312 					start = 0;
313 					end = ostart - 1;
314 					no_wrap = B_TRUE;
315 				}
316 			} else {
317 				start = fd + 1;
318 			}
319 			pdp = pcache_lookup_fd(pcp, fd);
320 repoll:
321 			ASSERT(pdp != NULL);
322 			ASSERT(pdp->pd_fd == fd);
323 			if (pdp->pd_fp == NULL) {
324 				/*
325 				 * The fd is POLLREMOVed. This fd is
326 				 * logically no longer cached. So move
327 				 * on to the next one.
328 				 */
329 				continue;
330 			}
331 			if ((fp = getf(fd)) == NULL) {
332 				/*
333 				 * The fd has been closed, but user has not
334 				 * done a POLLREMOVE on this fd yet. Instead
335 				 * of cleaning it here implicitly, we return
336 				 * POLLNVAL. This is consistent with poll(2)
337 				 * polling a closed fd. Hope this will remind
338 				 * user to do a POLLREMOVE.
339 				 */
340 				if (!is_epoll && pfdp != NULL) {
341 					pfdp[fdcnt].fd = fd;
342 					pfdp[fdcnt].revents = POLLNVAL;
343 					fdcnt++;
344 					continue;
345 				}
346 
347 				/*
348 				 * In the epoll compatibility case, we actually
349 				 * perform the implicit removal to remain
350 				 * closer to the epoll semantics.
351 				 */
352 				if (is_epoll) {
353 					pdp->pd_fp = NULL;
354 					pdp->pd_events = 0;
355 
356 					if (php != NULL) {
357 						pollhead_delete(php, pdp);
358 						pdp->pd_php = NULL;
359 					}
360 
361 					BT_CLEAR(pcp->pc_bitmap, fd);
362 					continue;
363 				}
364 			}
365 
366 			if (fp != pdp->pd_fp) {
367 				/*
368 				 * user is polling on a cached fd which was
369 				 * closed and then reused. Unfortunately
370 				 * there is no good way to inform user.
371 				 * If the file struct is also reused, we
372 				 * may not be able to detect the fd reuse
373 				 * at all.  As long as this does not
374 				 * cause system failure and/or memory leak,
375 				 * we will play along. Man page states if
376 				 * user does not clean up closed fds, polling
377 				 * results will be indeterministic.
378 				 *
379 				 * XXX - perhaps log the detection of fd
380 				 *	 reuse?
381 				 */
382 				pdp->pd_fp = fp;
383 			}
384 			/*
385 			 * XXX - pollrelock() logic needs to know which
386 			 * which pollcache lock to grab. It'd be a
387 			 * cleaner solution if we could pass pcp as
388 			 * an arguement in VOP_POLL interface instead
389 			 * of implicitly passing it using thread_t
390 			 * struct. On the other hand, changing VOP_POLL
391 			 * interface will require all driver/file system
392 			 * poll routine to change. May want to revisit
393 			 * the tradeoff later.
394 			 */
395 			curthread->t_pollcache = pcp;
396 			error = VOP_POLL(fp->f_vnode, pdp->pd_events, 0,
397 			    &revent, &php, NULL);
398 			curthread->t_pollcache = NULL;
399 			releasef(fd);
400 			if (error != 0) {
401 				break;
402 			}
403 
404 			/*
405 			 * layered devices (e.g. console driver)
406 			 * may change the vnode and thus the pollhead
407 			 * pointer out from underneath us.
408 			 */
409 			if (php != NULL && pdp->pd_php != NULL &&
410 			    php != pdp->pd_php) {
411 				pollhead_delete(pdp->pd_php, pdp);
412 				pdp->pd_php = php;
413 				pollhead_insert(php, pdp);
414 				/*
415 				 * The bit should still be set.
416 				 */
417 				ASSERT(BT_TEST(pcp->pc_bitmap, fd));
418 				goto retry;
419 			}
420 
421 			if (revent != 0) {
422 				if (pfdp != NULL) {
423 					pfdp[fdcnt].fd = fd;
424 					pfdp[fdcnt].events = pdp->pd_events;
425 					pfdp[fdcnt].revents = revent;
426 				} else if (epoll != NULL) {
427 					epoll_event_t *ep = &epoll[fdcnt];
428 
429 					ASSERT(epoll != NULL);
430 					ep->data.u64 = pdp->pd_epolldata;
431 
432 					/*
433 					 * If any of the event bits are set for
434 					 * which poll and epoll representations
435 					 * differ, swizzle in the native epoll
436 					 * values.
437 					 */
438 					if (revent & mask) {
439 						ep->events = (revent & ~mask) |
440 						    ((revent & POLLRDHUP) ?
441 						    EPOLLRDHUP : 0) |
442 						    ((revent & POLLWRBAND) ?
443 						    EPOLLWRBAND : 0);
444 					} else {
445 						ep->events = revent;
446 					}
447 
448 					/*
449 					 * We define POLLWRNORM to be POLLOUT,
450 					 * but epoll has separate definitions
451 					 * for them; if POLLOUT is set and the
452 					 * user has asked for EPOLLWRNORM, set
453 					 * that as well.
454 					 */
455 					if ((revent & POLLOUT) &&
456 					    (pdp->pd_events & EPOLLWRNORM)) {
457 						ep->events |= EPOLLWRNORM;
458 					}
459 				} else {
460 					pollstate_t *ps =
461 					    curthread->t_pollstate;
462 					/*
463 					 * The devpoll handle itself is being
464 					 * polled.  Notify the caller of any
465 					 * readable event(s), leaving as much
466 					 * state as possible untouched.
467 					 */
468 					VERIFY(fdcnt == 0);
469 					VERIFY(ps != NULL);
470 
471 					/*
472 					 * If a call to pollunlock() fails
473 					 * during VOP_POLL, skip over the fd
474 					 * and continue polling.
475 					 *
476 					 * Otherwise, report that there is an
477 					 * event pending.
478 					 */
479 					if ((ps->ps_flags & POLLSTATE_ULFAIL)
480 					    != 0) {
481 						ps->ps_flags &=
482 						    ~POLLSTATE_ULFAIL;
483 						continue;
484 					} else {
485 						fdcnt++;
486 						break;
487 					}
488 				}
489 
490 				/*
491 				 * If POLLET is set, clear the bit in the
492 				 * bitmap -- which effectively latches the
493 				 * edge on a pollwakeup() from the driver.
494 				 */
495 				if (pdp->pd_events & POLLET)
496 					BT_CLEAR(pcp->pc_bitmap, fd);
497 
498 				/*
499 				 * If POLLONESHOT is set, perform the implicit
500 				 * POLLREMOVE.
501 				 */
502 				if (pdp->pd_events & POLLONESHOT) {
503 					pdp->pd_fp = NULL;
504 					pdp->pd_events = 0;
505 
506 					if (php != NULL) {
507 						pollhead_delete(php, pdp);
508 						pdp->pd_php = NULL;
509 					}
510 
511 					BT_CLEAR(pcp->pc_bitmap, fd);
512 				}
513 
514 				fdcnt++;
515 			} else if (php != NULL) {
516 				/*
517 				 * We clear a bit or cache a poll fd if
518 				 * the driver returns a poll head ptr,
519 				 * which is expected in the case of 0
520 				 * revents. Some buggy driver may return
521 				 * NULL php pointer with 0 revents. In
522 				 * this case, we just treat the driver as
523 				 * "noncachable" and not clearing the bit
524 				 * in bitmap.
525 				 */
526 				if ((pdp->pd_php != NULL) &&
527 				    ((pcp->pc_flag & PC_POLLWAKE) == 0)) {
528 					BT_CLEAR(pcp->pc_bitmap, fd);
529 				}
530 				if (pdp->pd_php == NULL) {
531 					pollhead_insert(php, pdp);
532 					pdp->pd_php = php;
533 					/*
534 					 * An event of interest may have
535 					 * arrived between the VOP_POLL() and
536 					 * the pollhead_insert(); check again.
537 					 */
538 					goto repoll;
539 				}
540 			}
541 		} else {
542 			/*
543 			 * No bit set in the range. Check for wrap around.
544 			 */
545 			if (!no_wrap) {
546 				start = 0;
547 				end = ostart - 1;
548 				no_wrap = B_TRUE;
549 			} else {
550 				done = B_TRUE;
551 			}
552 		}
553 	}
554 
555 	if (!done) {
556 		pcp->pc_mapstart = start;
557 	}
558 	ASSERT(*fdcntp == 0);
559 	*fdcntp = fdcnt;
560 	return (error);
561 }
562 
563 /*ARGSUSED*/
564 static int
dpopen(dev_t * devp,int flag,int otyp,cred_t * credp)565 dpopen(dev_t *devp, int flag, int otyp, cred_t *credp)
566 {
567 	minor_t		minordev;
568 	dp_entry_t	*dpep;
569 	pollcache_t	*pcp;
570 
571 	ASSERT(devpoll_init);
572 	ASSERT(dptblsize <= MAXMIN);
573 	mutex_enter(&devpoll_lock);
574 	for (minordev = 0; minordev < dptblsize; minordev++) {
575 		if (devpolltbl[minordev] == NULL) {
576 			devpolltbl[minordev] = (dp_entry_t *)RESERVED;
577 			break;
578 		}
579 	}
580 	if (minordev == dptblsize) {
581 		dp_entry_t	**newtbl;
582 		size_t		oldsize;
583 
584 		/*
585 		 * Used up every entry in the existing devpoll table.
586 		 * Grow the table by DEVPOLLSIZE.
587 		 */
588 		if ((oldsize = dptblsize) >= MAXMIN) {
589 			mutex_exit(&devpoll_lock);
590 			return (ENXIO);
591 		}
592 		dptblsize += DEVPOLLSIZE;
593 		if (dptblsize > MAXMIN) {
594 			dptblsize = MAXMIN;
595 		}
596 		newtbl = kmem_zalloc(sizeof (caddr_t) * dptblsize, KM_SLEEP);
597 		bcopy(devpolltbl, newtbl, sizeof (caddr_t) * oldsize);
598 		kmem_free(devpolltbl, sizeof (caddr_t) * oldsize);
599 		devpolltbl = newtbl;
600 		devpolltbl[minordev] = (dp_entry_t *)RESERVED;
601 	}
602 	mutex_exit(&devpoll_lock);
603 
604 	dpep = kmem_zalloc(sizeof (dp_entry_t), KM_SLEEP);
605 	/*
606 	 * allocate a pollcache skeleton here. Delay allocating bitmap
607 	 * structures until dpwrite() time, since we don't know the
608 	 * optimal size yet.  We also delay setting the pid until either
609 	 * dpwrite() or attempt to poll on the instance, allowing parents
610 	 * to create instances of /dev/poll for their children.  (In the
611 	 * epoll compatibility case, this check isn't performed to maintain
612 	 * semantic compatibility.)
613 	 */
614 	pcp = pcache_alloc();
615 	dpep->dpe_pcache = pcp;
616 	pcp->pc_pid = -1;
617 	*devp = makedevice(getmajor(*devp), minordev);  /* clone the driver */
618 	mutex_enter(&devpoll_lock);
619 	ASSERT(minordev < dptblsize);
620 	ASSERT(devpolltbl[minordev] == (dp_entry_t *)RESERVED);
621 	devpolltbl[minordev] = dpep;
622 	mutex_exit(&devpoll_lock);
623 	return (0);
624 }
625 
626 /*
627  * Write to dev/poll add/remove fd's to/from a cached poll fd set,
628  * or change poll events for a watched fd.
629  */
630 /*ARGSUSED*/
631 static int
dpwrite(dev_t dev,struct uio * uiop,cred_t * credp)632 dpwrite(dev_t dev, struct uio *uiop, cred_t *credp)
633 {
634 	minor_t		minor;
635 	dp_entry_t	*dpep;
636 	pollcache_t	*pcp;
637 	pollfd_t	*pollfdp, *pfdp;
638 	dvpoll_epollfd_t *epfdp;
639 	uintptr_t	limit;
640 	int		error, size;
641 	ssize_t		uiosize;
642 	nfds_t		pollfdnum;
643 	struct pollhead	*php = NULL;
644 	polldat_t	*pdp;
645 	int		fd;
646 	file_t		*fp;
647 	boolean_t	is_epoll, fds_added = B_FALSE;
648 
649 	minor = getminor(dev);
650 
651 	mutex_enter(&devpoll_lock);
652 	ASSERT(minor < dptblsize);
653 	dpep = devpolltbl[minor];
654 	ASSERT(dpep != NULL);
655 	mutex_exit(&devpoll_lock);
656 
657 	mutex_enter(&dpep->dpe_lock);
658 	pcp = dpep->dpe_pcache;
659 	is_epoll = (dpep->dpe_flag & DP_ISEPOLLCOMPAT) != 0;
660 	size = (is_epoll) ? sizeof (dvpoll_epollfd_t) : sizeof (pollfd_t);
661 	mutex_exit(&dpep->dpe_lock);
662 
663 	if (!is_epoll && curproc->p_pid != pcp->pc_pid) {
664 		if (pcp->pc_pid != -1) {
665 			return (EACCES);
666 		}
667 
668 		pcp->pc_pid = curproc->p_pid;
669 	}
670 
671 	uiosize = uiop->uio_resid;
672 	pollfdnum = uiosize / size;
673 	mutex_enter(&curproc->p_lock);
674 	if (pollfdnum > (uint_t)rctl_enforced_value(
675 	    rctlproc_legacy[RLIMIT_NOFILE], curproc->p_rctls, curproc)) {
676 		(void) rctl_action(rctlproc_legacy[RLIMIT_NOFILE],
677 		    curproc->p_rctls, curproc, RCA_SAFE);
678 		mutex_exit(&curproc->p_lock);
679 		return (EINVAL);
680 	}
681 	mutex_exit(&curproc->p_lock);
682 	/*
683 	 * Copy in the pollfd array.  Walk through the array and add
684 	 * each polled fd to the cached set.
685 	 */
686 	pollfdp = kmem_alloc(uiosize, KM_SLEEP);
687 	limit = (uintptr_t)pollfdp + (pollfdnum * size);
688 
689 	/*
690 	 * Although /dev/poll uses the write(2) interface to cache fds, it's
691 	 * not supposed to function as a seekable device. To prevent offset
692 	 * from growing and eventually exceed the maximum, reset the offset
693 	 * here for every call.
694 	 */
695 	uiop->uio_loffset = 0;
696 	if ((error = uiomove((caddr_t)pollfdp, uiosize, UIO_WRITE, uiop))
697 	    != 0) {
698 		kmem_free(pollfdp, uiosize);
699 		return (error);
700 	}
701 	/*
702 	 * We are about to enter the core portion of dpwrite(). Make sure this
703 	 * write has exclusive access in this portion of the code, i.e., no
704 	 * other writers in this code.
705 	 *
706 	 * Waiting for all readers to drop their references to the dpe is
707 	 * unecessary since the pollcache itself is protected by pc_lock.
708 	 */
709 	mutex_enter(&dpep->dpe_lock);
710 	dpep->dpe_writerwait++;
711 	while ((dpep->dpe_flag & DP_WRITER_PRESENT) != 0) {
712 		ASSERT(dpep->dpe_refcnt != 0);
713 
714 		if (!cv_wait_sig_swap(&dpep->dpe_cv, &dpep->dpe_lock)) {
715 			dpep->dpe_writerwait--;
716 			mutex_exit(&dpep->dpe_lock);
717 			kmem_free(pollfdp, uiosize);
718 			return (EINTR);
719 		}
720 	}
721 	dpep->dpe_writerwait--;
722 	dpep->dpe_flag |= DP_WRITER_PRESENT;
723 	dpep->dpe_refcnt++;
724 
725 	if (!is_epoll && (dpep->dpe_flag & DP_ISEPOLLCOMPAT) != 0) {
726 		/*
727 		 * The epoll compat mode was enabled while we were waiting to
728 		 * establish write access. It is not safe to continue since
729 		 * state was prepared for non-epoll operation.
730 		 */
731 		error = EBUSY;
732 		goto bypass;
733 	}
734 	mutex_exit(&dpep->dpe_lock);
735 
736 	/*
737 	 * Since the dpwrite() may recursively walk an added /dev/poll handle,
738 	 * pollstate_enter() deadlock and loop detection must be used.
739 	 */
740 	(void) pollstate_create();
741 	VERIFY(pollstate_enter(pcp) == PSE_SUCCESS);
742 
743 	if (pcp->pc_bitmap == NULL) {
744 		pcache_create(pcp, pollfdnum);
745 	}
746 	for (pfdp = pollfdp; (uintptr_t)pfdp < limit;
747 	    pfdp = (pollfd_t *)((uintptr_t)pfdp + size)) {
748 		fd = pfdp->fd;
749 		if ((uint_t)fd >= P_FINFO(curproc)->fi_nfiles) {
750 			/*
751 			 * epoll semantics demand that we return EBADF if our
752 			 * specified fd is invalid.
753 			 */
754 			if (is_epoll) {
755 				error = EBADF;
756 				break;
757 			}
758 
759 			continue;
760 		}
761 
762 		pdp = pcache_lookup_fd(pcp, fd);
763 		if (pfdp->events != POLLREMOVE) {
764 
765 			fp = NULL;
766 
767 			if (pdp == NULL) {
768 				/*
769 				 * If we're in epoll compatibility mode, check
770 				 * that the fd is valid before allocating
771 				 * anything for it; epoll semantics demand that
772 				 * we return EBADF if our specified fd is
773 				 * invalid.
774 				 */
775 				if (is_epoll) {
776 					if ((fp = getf(fd)) == NULL) {
777 						error = EBADF;
778 						break;
779 					}
780 				}
781 
782 				pdp = pcache_alloc_fd(0);
783 				pdp->pd_fd = fd;
784 				pdp->pd_pcache = pcp;
785 				pcache_insert_fd(pcp, pdp, pollfdnum);
786 			} else {
787 				/*
788 				 * epoll semantics demand that we error out if
789 				 * a file descriptor is added twice, which we
790 				 * check (imperfectly) by checking if we both
791 				 * have the file descriptor cached and the
792 				 * file pointer that correponds to the file
793 				 * descriptor matches our cached value.  If
794 				 * there is a pointer mismatch, the file
795 				 * descriptor was closed without being removed.
796 				 * The converse is clearly not true, however,
797 				 * so to narrow the window by which a spurious
798 				 * EEXIST may be returned, we also check if
799 				 * this fp has been added to an epoll control
800 				 * descriptor in the past; if it hasn't, we
801 				 * know that this is due to fp reuse -- it's
802 				 * not a true EEXIST case.  (By performing this
803 				 * additional check, we limit the window of
804 				 * spurious EEXIST to situations where a single
805 				 * file descriptor is being used across two or
806 				 * more epoll control descriptors -- and even
807 				 * then, the file descriptor must be closed and
808 				 * reused in a relatively tight time span.)
809 				 */
810 				if (is_epoll) {
811 					if (pdp->pd_fp != NULL &&
812 					    (fp = getf(fd)) != NULL &&
813 					    fp == pdp->pd_fp &&
814 					    (fp->f_flag2 & FEPOLLED)) {
815 						error = EEXIST;
816 						releasef(fd);
817 						break;
818 					}
819 
820 					/*
821 					 * We have decided that the cached
822 					 * information was stale: it either
823 					 * didn't match, or the fp had never
824 					 * actually been epoll()'d on before.
825 					 * We need to now clear our pd_events
826 					 * to assure that we don't mistakenly
827 					 * operate on cached event disposition.
828 					 */
829 					pdp->pd_events = 0;
830 				}
831 			}
832 
833 			if (is_epoll) {
834 				epfdp = (dvpoll_epollfd_t *)pfdp;
835 				pdp->pd_epolldata = epfdp->dpep_data;
836 			}
837 
838 			ASSERT(pdp->pd_fd == fd);
839 			ASSERT(pdp->pd_pcache == pcp);
840 			if (fd >= pcp->pc_mapsize) {
841 				mutex_exit(&pcp->pc_lock);
842 				pcache_grow_map(pcp, fd);
843 				mutex_enter(&pcp->pc_lock);
844 			}
845 			if (fd > pcp->pc_mapend) {
846 				pcp->pc_mapend = fd;
847 			}
848 			if (fp == NULL && (fp = getf(fd)) == NULL) {
849 				/*
850 				 * The fd is not valid. Since we can't pass
851 				 * this error back in the write() call, set
852 				 * the bit in bitmap to force DP_POLL ioctl
853 				 * to examine it.
854 				 */
855 				BT_SET(pcp->pc_bitmap, fd);
856 				pdp->pd_events |= pfdp->events;
857 				continue;
858 			}
859 
860 			/*
861 			 * To (greatly) reduce EEXIST false positives, we
862 			 * denote that this fp has been epoll()'d.  We do this
863 			 * regardless of epoll compatibility mode, as the flag
864 			 * is harmless if not in epoll compatibility mode.
865 			 */
866 			fp->f_flag2 |= FEPOLLED;
867 
868 			/*
869 			 * Don't do VOP_POLL for an already cached fd with
870 			 * same poll events.
871 			 */
872 			if ((pdp->pd_events == pfdp->events) &&
873 			    (pdp->pd_fp == fp)) {
874 				/*
875 				 * the events are already cached
876 				 */
877 				releasef(fd);
878 				continue;
879 			}
880 
881 			/*
882 			 * do VOP_POLL and cache this poll fd.
883 			 */
884 			/*
885 			 * XXX - pollrelock() logic needs to know which
886 			 * which pollcache lock to grab. It'd be a
887 			 * cleaner solution if we could pass pcp as
888 			 * an arguement in VOP_POLL interface instead
889 			 * of implicitly passing it using thread_t
890 			 * struct. On the other hand, changing VOP_POLL
891 			 * interface will require all driver/file system
892 			 * poll routine to change. May want to revisit
893 			 * the tradeoff later.
894 			 */
895 			curthread->t_pollcache = pcp;
896 			error = VOP_POLL(fp->f_vnode, pfdp->events, 0,
897 			    &pfdp->revents, &php, NULL);
898 			curthread->t_pollcache = NULL;
899 			/*
900 			 * We always set the bit when this fd is cached;
901 			 * this forces the first DP_POLL to poll this fd.
902 			 * Real performance gain comes from subsequent
903 			 * DP_POLL.  We also attempt a pollhead_insert();
904 			 * if it's not possible, we'll do it in dpioctl().
905 			 */
906 			BT_SET(pcp->pc_bitmap, fd);
907 			if (error != 0) {
908 				releasef(fd);
909 				break;
910 			}
911 			pdp->pd_fp = fp;
912 			pdp->pd_events |= pfdp->events;
913 			if (php != NULL) {
914 				if (pdp->pd_php == NULL) {
915 					pollhead_insert(php, pdp);
916 					pdp->pd_php = php;
917 				} else {
918 					if (pdp->pd_php != php) {
919 						pollhead_delete(pdp->pd_php,
920 						    pdp);
921 						pollhead_insert(php, pdp);
922 						pdp->pd_php = php;
923 					}
924 				}
925 			}
926 			fds_added = B_TRUE;
927 			releasef(fd);
928 		} else {
929 			if (pdp == NULL || pdp->pd_fp == NULL) {
930 				if (is_epoll) {
931 					/*
932 					 * As with the add case (above), epoll
933 					 * semantics demand that we error out
934 					 * in this case.
935 					 */
936 					error = ENOENT;
937 					break;
938 				}
939 
940 				continue;
941 			}
942 			ASSERT(pdp->pd_fd == fd);
943 			pdp->pd_fp = NULL;
944 			pdp->pd_events = 0;
945 			ASSERT(pdp->pd_thread == NULL);
946 			if (pdp->pd_php != NULL) {
947 				pollhead_delete(pdp->pd_php, pdp);
948 				pdp->pd_php = NULL;
949 			}
950 			BT_CLEAR(pcp->pc_bitmap, fd);
951 		}
952 	}
953 	/*
954 	 * Wake any pollcache waiters so they can check the new descriptors.
955 	 *
956 	 * Any fds added to an recursive-capable pollcache could themselves be
957 	 * /dev/poll handles. To ensure that proper event propagation occurs,
958 	 * parent pollcaches are woken too, so that they can create any needed
959 	 * pollcache links.
960 	 */
961 	if (fds_added) {
962 		cv_broadcast(&pcp->pc_cv);
963 		pcache_wake_parents(pcp);
964 	}
965 	pollstate_exit(pcp);
966 	mutex_enter(&dpep->dpe_lock);
967 bypass:
968 	dpep->dpe_flag &= ~DP_WRITER_PRESENT;
969 	dpep->dpe_refcnt--;
970 	cv_broadcast(&dpep->dpe_cv);
971 	mutex_exit(&dpep->dpe_lock);
972 	kmem_free(pollfdp, uiosize);
973 	return (error);
974 }
975 
976 #define	DP_SIGMASK_RESTORE(ksetp) {					\
977 	if (ksetp != NULL) {						\
978 		mutex_enter(&p->p_lock);				\
979 		if (lwp->lwp_cursig == 0) {				\
980 			t->t_hold = lwp->lwp_sigoldmask;		\
981 			t->t_flag &= ~T_TOMASK;				\
982 		}							\
983 		mutex_exit(&p->p_lock);					\
984 	}								\
985 }
986 
987 /*ARGSUSED*/
988 static int
dpioctl(dev_t dev,int cmd,intptr_t arg,int mode,cred_t * credp,int * rvalp)989 dpioctl(dev_t dev, int cmd, intptr_t arg, int mode, cred_t *credp, int *rvalp)
990 {
991 	minor_t		minor;
992 	dp_entry_t	*dpep;
993 	pollcache_t	*pcp;
994 	hrtime_t	now;
995 	int		error = 0;
996 	boolean_t	is_epoll;
997 	STRUCT_DECL(dvpoll, dvpoll);
998 
999 	if (cmd == DP_POLL || cmd == DP_PPOLL) {
1000 		/* do this now, before we sleep on DP_WRITER_PRESENT */
1001 		now = gethrtime();
1002 	}
1003 
1004 	minor = getminor(dev);
1005 	mutex_enter(&devpoll_lock);
1006 	ASSERT(minor < dptblsize);
1007 	dpep = devpolltbl[minor];
1008 	mutex_exit(&devpoll_lock);
1009 	ASSERT(dpep != NULL);
1010 	pcp = dpep->dpe_pcache;
1011 
1012 	mutex_enter(&dpep->dpe_lock);
1013 	is_epoll = (dpep->dpe_flag & DP_ISEPOLLCOMPAT) != 0;
1014 
1015 	if (cmd == DP_EPOLLCOMPAT) {
1016 		if (dpep->dpe_refcnt != 0) {
1017 			/*
1018 			 * We can't turn on epoll compatibility while there
1019 			 * are outstanding operations.
1020 			 */
1021 			mutex_exit(&dpep->dpe_lock);
1022 			return (EBUSY);
1023 		}
1024 
1025 		/*
1026 		 * epoll compatibility is a one-way street: there's no way
1027 		 * to turn it off for a particular open.
1028 		 */
1029 		dpep->dpe_flag |= DP_ISEPOLLCOMPAT;
1030 		mutex_exit(&dpep->dpe_lock);
1031 
1032 		return (0);
1033 	}
1034 
1035 	if (!is_epoll && curproc->p_pid != pcp->pc_pid) {
1036 		if (pcp->pc_pid != -1) {
1037 			mutex_exit(&dpep->dpe_lock);
1038 			return (EACCES);
1039 		}
1040 
1041 		pcp->pc_pid = curproc->p_pid;
1042 	}
1043 
1044 	/* Wait until all writers have cleared the handle before continuing */
1045 	while ((dpep->dpe_flag & DP_WRITER_PRESENT) != 0 ||
1046 	    (dpep->dpe_writerwait != 0)) {
1047 		if (!cv_wait_sig_swap(&dpep->dpe_cv, &dpep->dpe_lock)) {
1048 			mutex_exit(&dpep->dpe_lock);
1049 			return (EINTR);
1050 		}
1051 	}
1052 	dpep->dpe_refcnt++;
1053 	mutex_exit(&dpep->dpe_lock);
1054 
1055 	switch (cmd) {
1056 	case	DP_POLL:
1057 	case	DP_PPOLL:
1058 	{
1059 		pollstate_t	*ps;
1060 		nfds_t		nfds;
1061 		int		fdcnt = 0;
1062 		size_t		size, fdsize, dpsize;
1063 		hrtime_t	deadline = 0;
1064 		k_sigset_t	*ksetp = NULL;
1065 		k_sigset_t	kset;
1066 		sigset_t	set;
1067 		kthread_t	*t = curthread;
1068 		klwp_t		*lwp = ttolwp(t);
1069 		struct proc	*p = ttoproc(curthread);
1070 
1071 		STRUCT_INIT(dvpoll, mode);
1072 
1073 		/*
1074 		 * The dp_setp member is only required/consumed for DP_PPOLL,
1075 		 * which otherwise uses the same structure as DP_POLL.
1076 		 */
1077 		if (cmd == DP_POLL) {
1078 			dpsize = (uintptr_t)STRUCT_FADDR(dvpoll, dp_setp) -
1079 			    (uintptr_t)STRUCT_FADDR(dvpoll, dp_fds);
1080 		} else {
1081 			ASSERT(cmd == DP_PPOLL);
1082 			dpsize = STRUCT_SIZE(dvpoll);
1083 		}
1084 
1085 		if ((mode & FKIOCTL) != 0) {
1086 			/* Kernel-internal ioctl call */
1087 			bcopy((caddr_t)arg, STRUCT_BUF(dvpoll), dpsize);
1088 			error = 0;
1089 		} else {
1090 			error = copyin((caddr_t)arg, STRUCT_BUF(dvpoll),
1091 			    dpsize);
1092 		}
1093 
1094 		if (error) {
1095 			DP_REFRELE(dpep);
1096 			return (EFAULT);
1097 		}
1098 
1099 		deadline = STRUCT_FGET(dvpoll, dp_timeout);
1100 		if (deadline > 0) {
1101 			/*
1102 			 * Convert the deadline from relative milliseconds
1103 			 * to absolute nanoseconds.  They must wait for at
1104 			 * least a tick.
1105 			 */
1106 			deadline = MSEC2NSEC(deadline);
1107 			deadline = MAX(deadline, nsec_per_tick);
1108 			deadline += now;
1109 		}
1110 
1111 		if (cmd == DP_PPOLL) {
1112 			void *setp = STRUCT_FGETP(dvpoll, dp_setp);
1113 
1114 			if (setp != NULL) {
1115 				if (copyin(setp, &set, sizeof (set))) {
1116 					DP_REFRELE(dpep);
1117 					return (EFAULT);
1118 				}
1119 
1120 				sigutok(&set, &kset);
1121 				ksetp = &kset;
1122 
1123 				mutex_enter(&p->p_lock);
1124 				schedctl_finish_sigblock(t);
1125 				lwp->lwp_sigoldmask = t->t_hold;
1126 				t->t_hold = *ksetp;
1127 				t->t_flag |= T_TOMASK;
1128 
1129 				/*
1130 				 * Like ppoll() with a non-NULL sigset, we'll
1131 				 * call cv_reltimedwait_sig() just to check for
1132 				 * signals.  This call will return immediately
1133 				 * with either 0 (signalled) or -1 (no signal).
1134 				 * There are some conditions whereby we can
1135 				 * get 0 from cv_reltimedwait_sig() without
1136 				 * a true signal (e.g., a directed stop), so
1137 				 * we restore our signal mask in the unlikely
1138 				 * event that lwp_cursig is 0.
1139 				 */
1140 				if (!cv_reltimedwait_sig(&t->t_delay_cv,
1141 				    &p->p_lock, 0, TR_CLOCK_TICK)) {
1142 					if (lwp->lwp_cursig == 0) {
1143 						t->t_hold = lwp->lwp_sigoldmask;
1144 						t->t_flag &= ~T_TOMASK;
1145 					}
1146 
1147 					mutex_exit(&p->p_lock);
1148 
1149 					DP_REFRELE(dpep);
1150 					return (EINTR);
1151 				}
1152 
1153 				mutex_exit(&p->p_lock);
1154 			}
1155 		}
1156 
1157 		if ((nfds = STRUCT_FGET(dvpoll, dp_nfds)) == 0) {
1158 			/*
1159 			 * We are just using DP_POLL to sleep, so
1160 			 * we don't any of the devpoll apparatus.
1161 			 * Do not check for signals if we have a zero timeout.
1162 			 */
1163 			DP_REFRELE(dpep);
1164 			if (deadline == 0) {
1165 				DP_SIGMASK_RESTORE(ksetp);
1166 				return (0);
1167 			}
1168 
1169 			mutex_enter(&curthread->t_delay_lock);
1170 			while ((error =
1171 			    cv_timedwait_sig_hrtime(&curthread->t_delay_cv,
1172 			    &curthread->t_delay_lock, deadline)) > 0)
1173 				continue;
1174 			mutex_exit(&curthread->t_delay_lock);
1175 
1176 			DP_SIGMASK_RESTORE(ksetp);
1177 
1178 			return (error == 0 ? EINTR : 0);
1179 		}
1180 
1181 		if (is_epoll) {
1182 			size = nfds * (fdsize = sizeof (epoll_event_t));
1183 		} else {
1184 			size = nfds * (fdsize = sizeof (pollfd_t));
1185 		}
1186 
1187 		/*
1188 		 * XXX It would be nice not to have to alloc each time, but it
1189 		 * requires another per thread structure hook. This can be
1190 		 * implemented later if data suggests that it's necessary.
1191 		 */
1192 		ps = pollstate_create();
1193 
1194 		if (ps->ps_dpbufsize < size) {
1195 			/*
1196 			 * If nfds is larger than twice the current maximum
1197 			 * open file count, we'll silently clamp it.  This
1198 			 * only limits our exposure to allocating an
1199 			 * inordinate amount of kernel memory; it doesn't
1200 			 * otherwise affect the semantics.  (We have this
1201 			 * check at twice the maximum instead of merely the
1202 			 * maximum because some applications pass an nfds that
1203 			 * is only slightly larger than their limit.)
1204 			 */
1205 			mutex_enter(&p->p_lock);
1206 			if ((nfds >> 1) > p->p_fno_ctl) {
1207 				nfds = p->p_fno_ctl;
1208 				size = nfds * fdsize;
1209 			}
1210 			mutex_exit(&p->p_lock);
1211 
1212 			if (ps->ps_dpbufsize < size) {
1213 				kmem_free(ps->ps_dpbuf, ps->ps_dpbufsize);
1214 				ps->ps_dpbuf = kmem_zalloc(size, KM_SLEEP);
1215 				ps->ps_dpbufsize = size;
1216 			}
1217 		}
1218 
1219 		VERIFY(pollstate_enter(pcp) == PSE_SUCCESS);
1220 		for (;;) {
1221 			pcp->pc_flag &= ~PC_POLLWAKE;
1222 
1223 			/*
1224 			 * Mark all child pcachelinks as stale.
1225 			 * Those which are still part of the tree will be
1226 			 * marked as valid during the poll.
1227 			 */
1228 			pcachelink_mark_stale(pcp);
1229 
1230 			error = dp_pcache_poll(dpep, ps->ps_dpbuf,
1231 			    pcp, nfds, &fdcnt);
1232 			if (fdcnt > 0 || error != 0)
1233 				break;
1234 
1235 			/* Purge still-stale child pcachelinks */
1236 			pcachelink_purge_stale(pcp);
1237 
1238 			/*
1239 			 * A pollwake has happened since we polled cache.
1240 			 */
1241 			if (pcp->pc_flag & PC_POLLWAKE)
1242 				continue;
1243 
1244 			/*
1245 			 * Sleep until we are notified, signaled, or timed out.
1246 			 */
1247 			if (deadline == 0) {
1248 				/* immediate timeout; do not check signals */
1249 				break;
1250 			}
1251 
1252 			error = cv_timedwait_sig_hrtime(&pcp->pc_cv,
1253 			    &pcp->pc_lock, deadline);
1254 
1255 			/*
1256 			 * If we were awakened by a signal or timeout then
1257 			 * break the loop, else poll again.
1258 			 */
1259 			if (error <= 0) {
1260 				error = (error == 0) ? EINTR : 0;
1261 				break;
1262 			} else {
1263 				error = 0;
1264 			}
1265 		}
1266 		pollstate_exit(pcp);
1267 
1268 		DP_SIGMASK_RESTORE(ksetp);
1269 
1270 		if (error == 0 && fdcnt > 0) {
1271 			if (copyout(ps->ps_dpbuf,
1272 			    STRUCT_FGETP(dvpoll, dp_fds), fdcnt * fdsize)) {
1273 				DP_REFRELE(dpep);
1274 				return (EFAULT);
1275 			}
1276 			*rvalp = fdcnt;
1277 		}
1278 		break;
1279 	}
1280 
1281 	case	DP_ISPOLLED:
1282 	{
1283 		pollfd_t	pollfd;
1284 		polldat_t	*pdp;
1285 
1286 		STRUCT_INIT(dvpoll, mode);
1287 		error = copyin((caddr_t)arg, &pollfd, sizeof (pollfd_t));
1288 		if (error) {
1289 			DP_REFRELE(dpep);
1290 			return (EFAULT);
1291 		}
1292 		mutex_enter(&pcp->pc_lock);
1293 		if (pcp->pc_hash == NULL) {
1294 			/*
1295 			 * No Need to search because no poll fd
1296 			 * has been cached.
1297 			 */
1298 			mutex_exit(&pcp->pc_lock);
1299 			DP_REFRELE(dpep);
1300 			return (0);
1301 		}
1302 		if (pollfd.fd < 0) {
1303 			mutex_exit(&pcp->pc_lock);
1304 			break;
1305 		}
1306 		pdp = pcache_lookup_fd(pcp, pollfd.fd);
1307 		if ((pdp != NULL) && (pdp->pd_fd == pollfd.fd) &&
1308 		    (pdp->pd_fp != NULL)) {
1309 			pollfd.revents = pdp->pd_events;
1310 			if (copyout(&pollfd, (caddr_t)arg, sizeof (pollfd_t))) {
1311 				mutex_exit(&pcp->pc_lock);
1312 				DP_REFRELE(dpep);
1313 				return (EFAULT);
1314 			}
1315 			*rvalp = 1;
1316 		}
1317 		mutex_exit(&pcp->pc_lock);
1318 		break;
1319 	}
1320 
1321 	default:
1322 		DP_REFRELE(dpep);
1323 		return (EINVAL);
1324 	}
1325 	DP_REFRELE(dpep);
1326 	return (error);
1327 }
1328 
1329 /*
1330  * Overview of Recursive Polling
1331  *
1332  * It is possible for /dev/poll to poll for events on file descriptors which
1333  * themselves are /dev/poll handles.  Pending events in the child handle are
1334  * represented as readable data via the POLLIN flag.  To limit surface area,
1335  * this recursion is presently allowed on only /dev/poll handles which have
1336  * been placed in epoll mode via the DP_EPOLLCOMPAT ioctl.  Recursion depth is
1337  * limited to 5 in order to be consistent with Linux epoll.
1338  *
1339  * Extending dppoll() for VOP_POLL:
1340  *
1341  * The recursive /dev/poll implementation begins by extending dppoll() to
1342  * report when resources contained in the pollcache have relevant event state.
1343  * At the highest level, it means calling dp_pcache_poll() so it indicates if
1344  * fd events are present without consuming them or altering the pollcache
1345  * bitmap.  This ensures that a subsequent DP_POLL operation on the bitmap will
1346  * yield the initiating event.  Additionally, the VOP_POLL should return in
1347  * such a way that dp_pcache_poll() does not clear the parent bitmap entry
1348  * which corresponds to the child /dev/poll fd.  This means that child
1349  * pollcaches will be checked during every poll which facilitates wake-up
1350  * behavior detailed below.
1351  *
1352  * Pollcache Links and Wake Events:
1353  *
1354  * Recursive /dev/poll avoids complicated pollcache locking constraints during
1355  * pollwakeup events by eschewing the traditional pollhead mechanism in favor
1356  * of a different approach.  For each pollcache at the root of a recursive
1357  * /dev/poll "tree", pcachelink_t structures are established to all child
1358  * /dev/poll pollcaches.  During pollnotify() in a child pollcache, the
1359  * linked list of pcachelink_t entries is walked, where those marked as valid
1360  * incur a cv_broadcast to their parent pollcache.  Most notably, these
1361  * pcachelink_t cv wakeups are performed without acquiring pc_lock on the
1362  * parent pollcache (which would require careful deadlock avoidance).  This
1363  * still allows the woken poll on the parent to discover the pertinent events
1364  * due to the fact that bitmap entires for the child pollcache are always
1365  * maintained by the dppoll() logic above.
1366  *
1367  * Depth Limiting and Loop Prevention:
1368  *
1369  * As each pollcache is encountered (either via DP_POLL or dppoll()), depth and
1370  * loop constraints are enforced via pollstate_enter().  The pollcache_t
1371  * pointer is compared against any existing entries in ps_pc_stack and is added
1372  * to the end if no match (and therefore loop) is found.  Once poll operations
1373  * for a given pollcache_t are complete, pollstate_exit() clears the pointer
1374  * from the list.  The pollstate_enter() and pollstate_exit() functions are
1375  * responsible for acquiring and releasing pc_lock, respectively.
1376  *
1377  * Deadlock Safety:
1378  *
1379  * Descending through a tree of recursive /dev/poll handles involves the tricky
1380  * business of sequentially entering multiple pollcache locks.  This tree
1381  * topology cannot define a lock acquisition order in such a way that it is
1382  * immune to deadlocks between threads.  The pollstate_enter() and
1383  * pollstate_exit() functions provide an interface for recursive /dev/poll
1384  * operations to safely lock pollcaches while failing gracefully in the face of
1385  * deadlocking topologies. (See pollstate_contend() for more detail about how
1386  * deadlocks are detected and resolved.)
1387  */
1388 
1389 /*ARGSUSED*/
1390 static int
dppoll(dev_t dev,short events,int anyyet,short * reventsp,struct pollhead ** phpp)1391 dppoll(dev_t dev, short events, int anyyet, short *reventsp,
1392     struct pollhead **phpp)
1393 {
1394 	minor_t		minor;
1395 	dp_entry_t	*dpep;
1396 	pollcache_t	*pcp;
1397 	int		res, rc = 0;
1398 
1399 	minor = getminor(dev);
1400 	mutex_enter(&devpoll_lock);
1401 	ASSERT(minor < dptblsize);
1402 	dpep = devpolltbl[minor];
1403 	ASSERT(dpep != NULL);
1404 	mutex_exit(&devpoll_lock);
1405 
1406 	mutex_enter(&dpep->dpe_lock);
1407 	if ((dpep->dpe_flag & DP_ISEPOLLCOMPAT) == 0) {
1408 		/* Poll recursion is not yet supported for non-epoll handles */
1409 		*reventsp = POLLERR;
1410 		mutex_exit(&dpep->dpe_lock);
1411 		return (0);
1412 	} else {
1413 		dpep->dpe_refcnt++;
1414 		pcp = dpep->dpe_pcache;
1415 		mutex_exit(&dpep->dpe_lock);
1416 	}
1417 
1418 	res = pollstate_enter(pcp);
1419 	if (res == PSE_SUCCESS) {
1420 		nfds_t		nfds = 1;
1421 		int		fdcnt = 0;
1422 		pollstate_t	*ps = curthread->t_pollstate;
1423 
1424 		rc = dp_pcache_poll(dpep, NULL, pcp, nfds, &fdcnt);
1425 		if (rc == 0) {
1426 			*reventsp = (fdcnt > 0) ? POLLIN : 0;
1427 		}
1428 		pcachelink_assoc(pcp, ps->ps_pc_stack[0]);
1429 		pollstate_exit(pcp);
1430 	} else {
1431 		switch (res) {
1432 		case PSE_FAIL_DEPTH:
1433 			rc = EINVAL;
1434 			break;
1435 		case PSE_FAIL_LOOP:
1436 		case PSE_FAIL_DEADLOCK:
1437 			rc = ELOOP;
1438 			break;
1439 		default:
1440 			/*
1441 			 * If anything else has gone awry, such as being polled
1442 			 * from an unexpected context, fall back to the
1443 			 * recursion-intolerant response.
1444 			 */
1445 			*reventsp = POLLERR;
1446 			rc = 0;
1447 			break;
1448 		}
1449 	}
1450 
1451 	DP_REFRELE(dpep);
1452 	return (rc);
1453 }
1454 
1455 /*
1456  * devpoll close should do enough clean up before the pollcache is deleted,
1457  * i.e., it should ensure no one still references the pollcache later.
1458  * There is no "permission" check in here. Any process having the last
1459  * reference of this /dev/poll fd can close.
1460  */
1461 /*ARGSUSED*/
1462 static int
dpclose(dev_t dev,int flag,int otyp,cred_t * credp)1463 dpclose(dev_t dev, int flag, int otyp, cred_t *credp)
1464 {
1465 	minor_t		minor;
1466 	dp_entry_t	*dpep;
1467 	pollcache_t	*pcp;
1468 	int		i;
1469 	polldat_t	**hashtbl;
1470 	polldat_t	*pdp;
1471 
1472 	minor = getminor(dev);
1473 
1474 	mutex_enter(&devpoll_lock);
1475 	dpep = devpolltbl[minor];
1476 	ASSERT(dpep != NULL);
1477 	devpolltbl[minor] = NULL;
1478 	mutex_exit(&devpoll_lock);
1479 	pcp = dpep->dpe_pcache;
1480 	ASSERT(pcp != NULL);
1481 	/*
1482 	 * At this point, no other lwp can access this pollcache via the
1483 	 * /dev/poll fd. This pollcache is going away, so do the clean
1484 	 * up without the pc_lock.
1485 	 */
1486 	hashtbl = pcp->pc_hash;
1487 	for (i = 0; i < pcp->pc_hashsize; i++) {
1488 		for (pdp = hashtbl[i]; pdp; pdp = pdp->pd_hashnext) {
1489 			if (pdp->pd_php != NULL) {
1490 				pollhead_delete(pdp->pd_php, pdp);
1491 				pdp->pd_php = NULL;
1492 				pdp->pd_fp = NULL;
1493 			}
1494 		}
1495 	}
1496 	/*
1497 	 * pollwakeup() may still interact with this pollcache. Wait until
1498 	 * it is done.
1499 	 */
1500 	mutex_enter(&pcp->pc_no_exit);
1501 	ASSERT(pcp->pc_busy >= 0);
1502 	while (pcp->pc_busy > 0)
1503 		cv_wait(&pcp->pc_busy_cv, &pcp->pc_no_exit);
1504 	mutex_exit(&pcp->pc_no_exit);
1505 
1506 	/* Clean up any pollcache links created via recursive /dev/poll */
1507 	if (pcp->pc_parents != NULL || pcp->pc_children != NULL) {
1508 		/*
1509 		 * Because of the locking rules for pcachelink manipulation,
1510 		 * acquring pc_lock is required for this step.
1511 		 */
1512 		mutex_enter(&pcp->pc_lock);
1513 		pcachelink_purge_all(pcp);
1514 		mutex_exit(&pcp->pc_lock);
1515 	}
1516 
1517 	pcache_destroy(pcp);
1518 	ASSERT(dpep->dpe_refcnt == 0);
1519 	kmem_free(dpep, sizeof (dp_entry_t));
1520 	return (0);
1521 }
1522 
1523 static void
pcachelink_locked_rele(pcachelink_t * pl)1524 pcachelink_locked_rele(pcachelink_t *pl)
1525 {
1526 	ASSERT(MUTEX_HELD(&pl->pcl_lock));
1527 	VERIFY(pl->pcl_refcnt >= 1);
1528 
1529 	pl->pcl_refcnt--;
1530 	if (pl->pcl_refcnt == 0) {
1531 		VERIFY(pl->pcl_state == PCL_INVALID);
1532 		ASSERT(pl->pcl_parent_pc == NULL);
1533 		ASSERT(pl->pcl_child_pc == NULL);
1534 		ASSERT(pl->pcl_parent_next == NULL);
1535 		ASSERT(pl->pcl_child_next == NULL);
1536 
1537 		pl->pcl_state = PCL_FREE;
1538 		mutex_destroy(&pl->pcl_lock);
1539 		kmem_free(pl, sizeof (pcachelink_t));
1540 	} else {
1541 		mutex_exit(&pl->pcl_lock);
1542 	}
1543 }
1544 
1545 /*
1546  * Associate parent and child pollcaches via a pcachelink_t.  If an existing
1547  * link (stale or valid) between the two is found, it will be reused.  If a
1548  * suitable link is not found for reuse, a new one will be allocated.
1549  */
1550 static void
pcachelink_assoc(pollcache_t * child,pollcache_t * parent)1551 pcachelink_assoc(pollcache_t *child, pollcache_t *parent)
1552 {
1553 	pcachelink_t	*pl, **plpn;
1554 
1555 	ASSERT(MUTEX_HELD(&child->pc_lock));
1556 	ASSERT(MUTEX_HELD(&parent->pc_lock));
1557 
1558 	/* Search for an existing link we can reuse. */
1559 	plpn = &child->pc_parents;
1560 	for (pl = child->pc_parents; pl != NULL; pl = *plpn) {
1561 		mutex_enter(&pl->pcl_lock);
1562 		if (pl->pcl_state == PCL_INVALID) {
1563 			/* Clean any invalid links while walking the list */
1564 			*plpn = pl->pcl_parent_next;
1565 			pl->pcl_child_pc = NULL;
1566 			pl->pcl_parent_next = NULL;
1567 			pcachelink_locked_rele(pl);
1568 		} else if (pl->pcl_parent_pc == parent) {
1569 			/* Successfully found parent link */
1570 			ASSERT(pl->pcl_state == PCL_VALID ||
1571 			    pl->pcl_state == PCL_STALE);
1572 			pl->pcl_state = PCL_VALID;
1573 			mutex_exit(&pl->pcl_lock);
1574 			return;
1575 		} else {
1576 			plpn = &pl->pcl_parent_next;
1577 			mutex_exit(&pl->pcl_lock);
1578 		}
1579 	}
1580 
1581 	/* No existing link to the parent was found.  Create a fresh one. */
1582 	pl = kmem_zalloc(sizeof (pcachelink_t), KM_SLEEP);
1583 	mutex_init(&pl->pcl_lock,  NULL, MUTEX_DEFAULT, NULL);
1584 
1585 	pl->pcl_parent_pc = parent;
1586 	pl->pcl_child_next = parent->pc_children;
1587 	parent->pc_children = pl;
1588 	pl->pcl_refcnt++;
1589 
1590 	pl->pcl_child_pc = child;
1591 	pl->pcl_parent_next = child->pc_parents;
1592 	child->pc_parents = pl;
1593 	pl->pcl_refcnt++;
1594 
1595 	pl->pcl_state = PCL_VALID;
1596 }
1597 
1598 /*
1599  * Mark all child links in a pollcache as stale.  Any invalid child links found
1600  * during iteration are purged.
1601  */
1602 static void
pcachelink_mark_stale(pollcache_t * pcp)1603 pcachelink_mark_stale(pollcache_t *pcp)
1604 {
1605 	pcachelink_t	*pl, **plpn;
1606 
1607 	ASSERT(MUTEX_HELD(&pcp->pc_lock));
1608 
1609 	plpn = &pcp->pc_children;
1610 	for (pl = pcp->pc_children; pl != NULL; pl = *plpn) {
1611 		mutex_enter(&pl->pcl_lock);
1612 		if (pl->pcl_state == PCL_INVALID) {
1613 			/*
1614 			 * Remove any invalid links while we are going to the
1615 			 * trouble of walking the list.
1616 			 */
1617 			*plpn = pl->pcl_child_next;
1618 			pl->pcl_parent_pc = NULL;
1619 			pl->pcl_child_next = NULL;
1620 			pcachelink_locked_rele(pl);
1621 		} else {
1622 			pl->pcl_state = PCL_STALE;
1623 			plpn = &pl->pcl_child_next;
1624 			mutex_exit(&pl->pcl_lock);
1625 		}
1626 	}
1627 }
1628 
1629 /*
1630  * Purge all stale (or invalid) child links from a pollcache.
1631  */
1632 static void
pcachelink_purge_stale(pollcache_t * pcp)1633 pcachelink_purge_stale(pollcache_t *pcp)
1634 {
1635 	pcachelink_t	*pl, **plpn;
1636 
1637 	ASSERT(MUTEX_HELD(&pcp->pc_lock));
1638 
1639 	plpn = &pcp->pc_children;
1640 	for (pl = pcp->pc_children; pl != NULL; pl = *plpn) {
1641 		mutex_enter(&pl->pcl_lock);
1642 		switch (pl->pcl_state) {
1643 		case PCL_STALE:
1644 			pl->pcl_state = PCL_INVALID;
1645 			/* FALLTHROUGH */
1646 		case PCL_INVALID:
1647 			*plpn = pl->pcl_child_next;
1648 			pl->pcl_parent_pc = NULL;
1649 			pl->pcl_child_next = NULL;
1650 			pcachelink_locked_rele(pl);
1651 			break;
1652 		default:
1653 			plpn = &pl->pcl_child_next;
1654 			mutex_exit(&pl->pcl_lock);
1655 		}
1656 	}
1657 }
1658 
1659 /*
1660  * Purge all child and parent links from a pollcache, regardless of status.
1661  */
1662 static void
pcachelink_purge_all(pollcache_t * pcp)1663 pcachelink_purge_all(pollcache_t *pcp)
1664 {
1665 	pcachelink_t	*pl, **plpn;
1666 
1667 	ASSERT(MUTEX_HELD(&pcp->pc_lock));
1668 
1669 	plpn = &pcp->pc_parents;
1670 	for (pl = pcp->pc_parents; pl != NULL; pl = *plpn) {
1671 		mutex_enter(&pl->pcl_lock);
1672 		pl->pcl_state = PCL_INVALID;
1673 		*plpn = pl->pcl_parent_next;
1674 		pl->pcl_child_pc = NULL;
1675 		pl->pcl_parent_next = NULL;
1676 		pcachelink_locked_rele(pl);
1677 	}
1678 
1679 	plpn = &pcp->pc_children;
1680 	for (pl = pcp->pc_children; pl != NULL; pl = *plpn) {
1681 		mutex_enter(&pl->pcl_lock);
1682 		pl->pcl_state = PCL_INVALID;
1683 		*plpn = pl->pcl_child_next;
1684 		pl->pcl_parent_pc = NULL;
1685 		pl->pcl_child_next = NULL;
1686 		pcachelink_locked_rele(pl);
1687 	}
1688 
1689 	ASSERT(pcp->pc_parents == NULL);
1690 	ASSERT(pcp->pc_children == NULL);
1691 }
1692