xref: /illumos-gate/usr/src/uts/common/os/fio.c (revision 03100a6332bd4edc7a53091fcf7c9a7131bcdaa7)
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 (c) 1984, 1986, 1987, 1988, 1989 AT&T	*/
23 /*	All Rights Reserved */
24 
25 /*
26  * Copyright 2007 Sun Microsystems, Inc.  All rights reserved.
27  * Use is subject to license terms.
28  */
29 
30 #pragma ident	"%Z%%M%	%I%	%E% SMI"
31 
32 #include <sys/types.h>
33 #include <sys/sysmacros.h>
34 #include <sys/param.h>
35 #include <sys/systm.h>
36 #include <sys/errno.h>
37 #include <sys/signal.h>
38 #include <sys/cred.h>
39 #include <sys/user.h>
40 #include <sys/conf.h>
41 #include <sys/vfs.h>
42 #include <sys/vnode.h>
43 #include <sys/pathname.h>
44 #include <sys/file.h>
45 #include <sys/proc.h>
46 #include <sys/var.h>
47 #include <sys/cpuvar.h>
48 #include <sys/open.h>
49 #include <sys/cmn_err.h>
50 #include <sys/priocntl.h>
51 #include <sys/procset.h>
52 #include <sys/prsystm.h>
53 #include <sys/debug.h>
54 #include <sys/kmem.h>
55 #include <sys/atomic.h>
56 #include <sys/fcntl.h>
57 #include <sys/poll.h>
58 #include <sys/rctl.h>
59 #include <sys/port_impl.h>
60 
61 #include <c2/audit.h>
62 #include <sys/nbmlock.h>
63 
64 #ifdef DEBUG
65 
66 static uint32_t afd_maxfd;	/* # of entries in maximum allocated array */
67 static uint32_t afd_alloc;	/* count of kmem_alloc()s */
68 static uint32_t afd_free;	/* count of kmem_free()s */
69 static uint32_t afd_wait;	/* count of waits on non-zero ref count */
70 #define	MAXFD(x)	(afd_maxfd = ((afd_maxfd >= (x))? afd_maxfd : (x)))
71 #define	COUNT(x)	atomic_add_32(&x, 1)
72 
73 #else	/* DEBUG */
74 
75 #define	MAXFD(x)
76 #define	COUNT(x)
77 
78 #endif	/* DEBUG */
79 
80 kmem_cache_t *file_cache;
81 static int vpsetattr(vnode_t *, vattr_t *, int);
82 
83 static void port_close_fd(portfd_t *);
84 
85 /*
86  * File descriptor allocation.
87  *
88  * fd_find(fip, minfd) finds the first available descriptor >= minfd.
89  * The most common case is open(2), in which minfd = 0, but we must also
90  * support fcntl(fd, F_DUPFD, minfd).
91  *
92  * The algorithm is as follows: we keep all file descriptors in an infix
93  * binary tree in which each node records the number of descriptors
94  * allocated in its right subtree, including itself.  Starting at minfd,
95  * we ascend the tree until we find a non-fully allocated right subtree.
96  * We then descend that subtree in a binary search for the smallest fd.
97  * Finally, we ascend the tree again to increment the allocation count
98  * of every subtree containing the newly-allocated fd.  Freeing an fd
99  * requires only the last step: we ascend the tree to decrement allocation
100  * counts.  Each of these three steps (ascent to find non-full subtree,
101  * descent to find lowest fd, ascent to update allocation counts) is
102  * O(log n), thus the algorithm as a whole is O(log n).
103  *
104  * We don't implement the fd tree using the customary left/right/parent
105  * pointers, but instead take advantage of the glorious mathematics of
106  * full infix binary trees.  For reference, here's an illustration of the
107  * logical structure of such a tree, rooted at 4 (binary 100), covering
108  * the range 1-7 (binary 001-111).  Our canonical trees do not include
109  * fd 0; we'll deal with that later.
110  *
111  *	      100
112  *	     /	 \
113  *	    /	  \
114  *	  010	  110
115  *	  / \	  / \
116  *	001 011 101 111
117  *
118  * We make the following observations, all of which are easily proven by
119  * induction on the depth of the tree:
120  *
121  * (T1) The least-significant bit (LSB) of any node is equal to its level
122  *      in the tree.  In our example, nodes 001, 011, 101 and 111 are at
123  *      level 0; nodes 010 and 110 are at level 1; and node 100 is at level 2.
124  *
125  * (T2) The child size (CSIZE) of node N -- that is, the total number of
126  *	right-branch descendants in a child of node N, including itself -- is
127  *	given by clearing all but the least significant bit of N.  This
128  *	follows immediately from (T1).  Applying this rule to our example, we
129  *	see that CSIZE(100) = 100, CSIZE(x10) = 10, and CSIZE(xx1) = 1.
130  *
131  * (T3) The nearest left ancestor (LPARENT) of node N -- that is, the nearest
132  *	ancestor containing node N in its right child -- is given by clearing
133  *	the LSB of N.  For example, LPARENT(111) = 110 and LPARENT(110) = 100.
134  *	Clearing the LSB of nodes 001, 010 or 100 yields zero, reflecting
135  *	the fact that these are leftmost nodes.  Note that this algorithm
136  *	automatically skips generations as necessary.  For example, the parent
137  *      of node 101 is 110, which is a *right* ancestor (not what we want);
138  *      but its grandparent is 100, which is a left ancestor. Clearing the LSB
139  *      of 101 gets us to 100 directly, skipping right past the uninteresting
140  *      generation (110).
141  *
142  *      Note that since LPARENT clears the LSB, whereas CSIZE clears all *but*
143  *	the LSB, we can express LPARENT() nicely in terms of CSIZE():
144  *
145  *	LPARENT(N) = N - CSIZE(N)
146  *
147  * (T4) The nearest right ancestor (RPARENT) of node N is given by:
148  *
149  *	RPARENT(N) = N + CSIZE(N)
150  *
151  * (T5) For every interior node, the children differ from their parent by
152  *	CSIZE(parent) / 2.  In our example, CSIZE(100) / 2 = 2 = 10 binary,
153  *      and indeed, the children of 100 are 100 +/- 10 = 010 and 110.
154  *
155  * Next, we'll need a few two's-complement math tricks.  Suppose a number,
156  * N, has the following form:
157  *
158  *		N = xxxx10...0
159  *
160  * That is, the binary representation of N consists of some string of bits,
161  * then a 1, then all zeroes.  This amounts to nothing more than saying that
162  * N has a least-significant bit, which is true for any N != 0.  If we look
163  * at N and N - 1 together, we see that we can combine them in useful ways:
164  *
165  *		  N = xxxx10...0
166  *	      N - 1 = xxxx01...1
167  *	------------------------
168  *	N & (N - 1) = xxxx000000
169  *	N | (N - 1) = xxxx111111
170  *	N ^ (N - 1) =     111111
171  *
172  * In particular, this suggests several easy ways to clear all but the LSB,
173  * which by (T2) is exactly what we need to determine CSIZE(N) = 10...0.
174  * We'll opt for this formulation:
175  *
176  *	(C1) CSIZE(N) = (N - 1) ^ (N | (N - 1))
177  *
178  * Similarly, we have an easy way to determine LPARENT(N), which requires
179  * that we clear the LSB of N:
180  *
181  *	(L1) LPARENT(N) = N & (N - 1)
182  *
183  * We note in the above relations that (N | (N - 1)) - N = CSIZE(N) - 1.
184  * When combined with (T4), this yields an easy way to compute RPARENT(N):
185  *
186  *	(R1) RPARENT(N) = (N | (N - 1)) + 1
187  *
188  * Finally, to accommodate fd 0 we must adjust all of our results by +/-1 to
189  * move the fd range from [1, 2^n) to [0, 2^n - 1).  This is straightforward,
190  * so there's no need to belabor the algebra; the revised relations become:
191  *
192  *	(C1a) CSIZE(N) = N ^ (N | (N + 1))
193  *
194  *	(L1a) LPARENT(N) = (N & (N + 1)) - 1
195  *
196  *	(R1a) RPARENT(N) = N | (N + 1)
197  *
198  * This completes the mathematical framework.  We now have all the tools
199  * we need to implement fd_find() and fd_reserve().
200  *
201  * fd_find(fip, minfd) finds the smallest available file descriptor >= minfd.
202  * It does not actually allocate the descriptor; that's done by fd_reserve().
203  * fd_find() proceeds in two steps:
204  *
205  * (1) Find the leftmost subtree that contains a descriptor >= minfd.
206  *     We start at the right subtree rooted at minfd.  If this subtree is
207  *     not full -- if fip->fi_list[minfd].uf_alloc != CSIZE(minfd) -- then
208  *     step 1 is done.  Otherwise, we know that all fds in this subtree
209  *     are taken, so we ascend to RPARENT(minfd) using (R1a).  We repeat
210  *     this process until we either find a candidate subtree or exceed
211  *     fip->fi_nfiles.  We use (C1a) to compute CSIZE().
212  *
213  * (2) Find the smallest fd in the subtree discovered by step 1.
214  *     Starting at the root of this subtree, we descend to find the
215  *     smallest available fd.  Since the left children have the smaller
216  *     fds, we will descend rightward only when the left child is full.
217  *
218  *     We begin by comparing the number of allocated fds in the root
219  *     to the number of allocated fds in its right child; if they differ
220  *     by exactly CSIZE(child), we know the left subtree is full, so we
221  *     descend right; that is, the right child becomes the search root.
222  *     Otherwise we leave the root alone and start following the right
223  *     child's left children.  As fortune would have it, this is very
224  *     simple computationally: by (T5), the right child of fd is just
225  *     fd + size, where size = CSIZE(fd) / 2.  Applying (T5) again,
226  *     we find that the right child's left child is fd + size - (size / 2) =
227  *     fd + (size / 2); *its* left child is fd + (size / 2) - (size / 4) =
228  *     fd + (size / 4), and so on.  In general, fd's right child's
229  *     leftmost nth descendant is fd + (size >> n).  Thus, to follow
230  *     the right child's left descendants, we just halve the size in
231  *     each iteration of the search.
232  *
233  *     When we descend leftward, we must keep track of the number of fds
234  *     that were allocated in all the right subtrees we rejected, so we
235  *     know how many of the root fd's allocations are in the remaining
236  *     (as yet unexplored) leftmost part of its right subtree.  When we
237  *     encounter a fully-allocated left child -- that is, when we find
238  *     that fip->fi_list[fd].uf_alloc == ralloc + size -- we descend right
239  *     (as described earlier), resetting ralloc to zero.
240  *
241  * fd_reserve(fip, fd, incr) either allocates or frees fd, depending
242  * on whether incr is 1 or -1.  Starting at fd, fd_reserve() ascends
243  * the leftmost ancestors (see (T3)) and updates the allocation counts.
244  * At each step we use (L1a) to compute LPARENT(), the next left ancestor.
245  *
246  * flist_minsize() finds the minimal tree that still covers all
247  * used fds; as long as the allocation count of a root node is zero, we
248  * don't need that node or its right subtree.
249  *
250  * flist_nalloc() counts the number of allocated fds in the tree, by starting
251  * at the top of the tree and summing the right-subtree allocation counts as
252  * it descends leftwards.
253  *
254  * Note: we assume that flist_grow() will keep fip->fi_nfiles of the form
255  * 2^n - 1.  This ensures that the fd trees are always full, which saves
256  * quite a bit of boundary checking.
257  */
258 static int
259 fd_find(uf_info_t *fip, int minfd)
260 {
261 	int size, ralloc, fd;
262 
263 	ASSERT(MUTEX_HELD(&fip->fi_lock));
264 	ASSERT((fip->fi_nfiles & (fip->fi_nfiles + 1)) == 0);
265 
266 	for (fd = minfd; (uint_t)fd < fip->fi_nfiles; fd |= fd + 1) {
267 		size = fd ^ (fd | (fd + 1));
268 		if (fip->fi_list[fd].uf_alloc == size)
269 			continue;
270 		for (ralloc = 0, size >>= 1; size != 0; size >>= 1) {
271 			ralloc += fip->fi_list[fd + size].uf_alloc;
272 			if (fip->fi_list[fd].uf_alloc == ralloc + size) {
273 				fd += size;
274 				ralloc = 0;
275 			}
276 		}
277 		return (fd);
278 	}
279 	return (-1);
280 }
281 
282 static void
283 fd_reserve(uf_info_t *fip, int fd, int incr)
284 {
285 	int pfd;
286 	uf_entry_t *ufp = &fip->fi_list[fd];
287 
288 	ASSERT((uint_t)fd < fip->fi_nfiles);
289 	ASSERT((ufp->uf_busy == 0 && incr == 1) ||
290 	    (ufp->uf_busy == 1 && incr == -1));
291 	ASSERT(MUTEX_HELD(&ufp->uf_lock));
292 	ASSERT(MUTEX_HELD(&fip->fi_lock));
293 
294 	for (pfd = fd; pfd >= 0; pfd = (pfd & (pfd + 1)) - 1)
295 		fip->fi_list[pfd].uf_alloc += incr;
296 
297 	ufp->uf_busy += incr;
298 }
299 
300 static int
301 flist_minsize(uf_info_t *fip)
302 {
303 	int fd;
304 
305 	/*
306 	 * We'd like to ASSERT(MUTEX_HELD(&fip->fi_lock)), but we're called
307 	 * by flist_fork(), which relies on other mechanisms for mutual
308 	 * exclusion.
309 	 */
310 	ASSERT((fip->fi_nfiles & (fip->fi_nfiles + 1)) == 0);
311 
312 	for (fd = fip->fi_nfiles; fd != 0; fd >>= 1)
313 		if (fip->fi_list[fd >> 1].uf_alloc != 0)
314 			break;
315 
316 	return (fd);
317 }
318 
319 static int
320 flist_nalloc(uf_info_t *fip)
321 {
322 	int fd;
323 	int nalloc = 0;
324 
325 	ASSERT(MUTEX_HELD(&fip->fi_lock));
326 	ASSERT((fip->fi_nfiles & (fip->fi_nfiles + 1)) == 0);
327 
328 	for (fd = fip->fi_nfiles; fd != 0; fd >>= 1)
329 		nalloc += fip->fi_list[fd >> 1].uf_alloc;
330 
331 	return (nalloc);
332 }
333 
334 /*
335  * Increase size of the fi_list array to accommodate at least maxfd.
336  * We keep the size of the form 2^n - 1 for benefit of fd_find().
337  */
338 static void
339 flist_grow(int maxfd)
340 {
341 	uf_info_t *fip = P_FINFO(curproc);
342 	int newcnt, oldcnt;
343 	uf_entry_t *src, *dst, *newlist, *oldlist, *newend, *oldend;
344 	uf_rlist_t *urp;
345 
346 	for (newcnt = 1; newcnt <= maxfd; newcnt = (newcnt << 1) | 1)
347 		continue;
348 
349 	newlist = kmem_zalloc(newcnt * sizeof (uf_entry_t), KM_SLEEP);
350 
351 	mutex_enter(&fip->fi_lock);
352 	oldcnt = fip->fi_nfiles;
353 	if (newcnt <= oldcnt) {
354 		mutex_exit(&fip->fi_lock);
355 		kmem_free(newlist, newcnt * sizeof (uf_entry_t));
356 		return;
357 	}
358 	ASSERT((newcnt & (newcnt + 1)) == 0);
359 	oldlist = fip->fi_list;
360 	oldend = oldlist + oldcnt;
361 	newend = newlist + oldcnt;	/* no need to lock beyond old end */
362 
363 	/*
364 	 * fi_list and fi_nfiles cannot change while any uf_lock is held,
365 	 * so we must grab all the old locks *and* the new locks up to oldcnt.
366 	 * (Locks beyond the end of oldcnt aren't visible until we store
367 	 * the new fi_nfiles, which is the last thing we do before dropping
368 	 * all the locks, so there's no need to acquire these locks).
369 	 * Holding the new locks is necessary because when fi_list changes
370 	 * to point to the new list, fi_nfiles won't have been stored yet.
371 	 * If we *didn't* hold the new locks, someone doing a UF_ENTER()
372 	 * could see the new fi_list, grab the new uf_lock, and then see
373 	 * fi_nfiles change while the lock is held -- in violation of
374 	 * UF_ENTER() semantics.
375 	 */
376 	for (src = oldlist; src < oldend; src++)
377 		mutex_enter(&src->uf_lock);
378 
379 	for (dst = newlist; dst < newend; dst++)
380 		mutex_enter(&dst->uf_lock);
381 
382 	for (src = oldlist, dst = newlist; src < oldend; src++, dst++) {
383 		dst->uf_file = src->uf_file;
384 		dst->uf_fpollinfo = src->uf_fpollinfo;
385 		dst->uf_refcnt = src->uf_refcnt;
386 		dst->uf_alloc = src->uf_alloc;
387 		dst->uf_flag = src->uf_flag;
388 		dst->uf_busy = src->uf_busy;
389 		dst->uf_portfd = src->uf_portfd;
390 	}
391 
392 	/*
393 	 * As soon as we store the new flist, future locking operations
394 	 * will use it.  Therefore, we must ensure that all the state
395 	 * we've just established reaches global visibility before the
396 	 * new flist does.
397 	 */
398 	membar_producer();
399 	fip->fi_list = newlist;
400 
401 	/*
402 	 * Routines like getf() make an optimistic check on the validity
403 	 * of the supplied file descriptor: if it's less than the current
404 	 * value of fi_nfiles -- examined without any locks -- then it's
405 	 * safe to attempt a UF_ENTER() on that fd (which is a valid
406 	 * assumption because fi_nfiles only increases).  Therefore, it
407 	 * is critical that the new value of fi_nfiles not reach global
408 	 * visibility until after the new fi_list: if it happened the
409 	 * other way around, getf() could see the new fi_nfiles and attempt
410 	 * a UF_ENTER() on the old fi_list, which would write beyond its
411 	 * end if the fd exceeded the old fi_nfiles.
412 	 */
413 	membar_producer();
414 	fip->fi_nfiles = newcnt;
415 
416 	/*
417 	 * The new state is consistent now, so we can drop all the locks.
418 	 */
419 	for (dst = newlist; dst < newend; dst++)
420 		mutex_exit(&dst->uf_lock);
421 
422 	for (src = oldlist; src < oldend; src++) {
423 		/*
424 		 * If any threads are blocked on the old cvs, wake them.
425 		 * This will force them to wake up, discover that fi_list
426 		 * has changed, and go back to sleep on the new cvs.
427 		 */
428 		cv_broadcast(&src->uf_wanted_cv);
429 		cv_broadcast(&src->uf_closing_cv);
430 		mutex_exit(&src->uf_lock);
431 	}
432 
433 	mutex_exit(&fip->fi_lock);
434 
435 	/*
436 	 * Retire the old flist.  We can't actually kmem_free() it now
437 	 * because someone may still have a pointer to it.  Instead,
438 	 * we link it onto a list of retired flists.  The new flist
439 	 * is at least double the size of the previous flist, so the
440 	 * total size of all retired flists will be less than the size
441 	 * of the current one (to prove, consider the sum of a geometric
442 	 * series in powers of 2).  exit() frees the retired flists.
443 	 */
444 	urp = kmem_zalloc(sizeof (uf_rlist_t), KM_SLEEP);
445 	urp->ur_list = oldlist;
446 	urp->ur_nfiles = oldcnt;
447 
448 	mutex_enter(&fip->fi_lock);
449 	urp->ur_next = fip->fi_rlist;
450 	fip->fi_rlist = urp;
451 	mutex_exit(&fip->fi_lock);
452 }
453 
454 /*
455  * Utility functions for keeping track of the active file descriptors.
456  */
457 void
458 clear_stale_fd()		/* called from post_syscall() */
459 {
460 	afd_t *afd = &curthread->t_activefd;
461 	int i;
462 
463 	/* uninitialized is ok here, a_nfd is then zero */
464 	for (i = 0; i < afd->a_nfd; i++) {
465 		/* assert that this should not be necessary */
466 		ASSERT(afd->a_fd[i] == -1);
467 		afd->a_fd[i] = -1;
468 	}
469 	afd->a_stale = 0;
470 }
471 
472 void
473 free_afd(afd_t *afd)		/* called below and from thread_free() */
474 {
475 	int i;
476 
477 	/* free the buffer if it was kmem_alloc()ed */
478 	if (afd->a_nfd > sizeof (afd->a_buf) / sizeof (afd->a_buf[0])) {
479 		COUNT(afd_free);
480 		kmem_free(afd->a_fd, afd->a_nfd * sizeof (afd->a_fd[0]));
481 	}
482 
483 	/* (re)initialize the structure */
484 	afd->a_fd = &afd->a_buf[0];
485 	afd->a_nfd = sizeof (afd->a_buf) / sizeof (afd->a_buf[0]);
486 	afd->a_stale = 0;
487 	for (i = 0; i < afd->a_nfd; i++)
488 		afd->a_fd[i] = -1;
489 }
490 
491 static void
492 set_active_fd(int fd)
493 {
494 	afd_t *afd = &curthread->t_activefd;
495 	int i;
496 	int *old_fd;
497 	int old_nfd;
498 
499 	if (afd->a_nfd == 0)	/* first time initialization */
500 		free_afd(afd);
501 
502 	/* insert fd into vacant slot, if any */
503 	for (i = 0; i < afd->a_nfd; i++) {
504 		if (afd->a_fd[i] == -1) {
505 			afd->a_fd[i] = fd;
506 			return;
507 		}
508 	}
509 
510 	/*
511 	 * Reallocate the a_fd[] array to add one more slot.
512 	 */
513 	old_fd = afd->a_fd;
514 	old_nfd = afd->a_nfd;
515 	afd->a_nfd = old_nfd + 1;
516 	MAXFD(afd->a_nfd);
517 	COUNT(afd_alloc);
518 	afd->a_fd = kmem_alloc(afd->a_nfd * sizeof (afd->a_fd[0]), KM_SLEEP);
519 	for (i = 0; i < old_nfd; i++)
520 		afd->a_fd[i] = old_fd[i];
521 	afd->a_fd[i] = fd;
522 
523 	if (old_nfd > sizeof (afd->a_buf) / sizeof (afd->a_buf[0])) {
524 		COUNT(afd_free);
525 		kmem_free(old_fd, old_nfd * sizeof (afd->a_fd[0]));
526 	}
527 }
528 
529 void
530 clear_active_fd(int fd)		/* called below and from aio.c */
531 {
532 	afd_t *afd = &curthread->t_activefd;
533 	int i;
534 
535 	for (i = 0; i < afd->a_nfd; i++) {
536 		if (afd->a_fd[i] == fd) {
537 			afd->a_fd[i] = -1;
538 			break;
539 		}
540 	}
541 	ASSERT(i < afd->a_nfd);		/* not found is not ok */
542 }
543 
544 /*
545  * Does this thread have this fd active?
546  */
547 static int
548 is_active_fd(kthread_t *t, int fd)
549 {
550 	afd_t *afd = &t->t_activefd;
551 	int i;
552 
553 	/* uninitialized is ok here, a_nfd is then zero */
554 	for (i = 0; i < afd->a_nfd; i++) {
555 		if (afd->a_fd[i] == fd)
556 			return (1);
557 	}
558 	return (0);
559 }
560 
561 /*
562  * Convert a user supplied file descriptor into a pointer to a file
563  * structure.  Only task is to check range of the descriptor (soft
564  * resource limit was enforced at open time and shouldn't be checked
565  * here).
566  */
567 file_t *
568 getf(int fd)
569 {
570 	uf_info_t *fip = P_FINFO(curproc);
571 	uf_entry_t *ufp;
572 	file_t *fp;
573 
574 	if ((uint_t)fd >= fip->fi_nfiles)
575 		return (NULL);
576 
577 	UF_ENTER(ufp, fip, fd);
578 	if ((fp = ufp->uf_file) == NULL) {
579 		UF_EXIT(ufp);
580 
581 		if (fd == fip->fi_badfd && fip->fi_action > 0)
582 			tsignal(curthread, fip->fi_action);
583 
584 		return (NULL);
585 	}
586 	ufp->uf_refcnt++;
587 
588 #ifdef C2_AUDIT
589 	/*
590 	 * archive per file audit data
591 	 */
592 	if (audit_active)
593 		(void) audit_getf(fd);
594 #endif
595 	UF_EXIT(ufp);
596 
597 	set_active_fd(fd);	/* record the active file descriptor */
598 
599 	return (fp);
600 }
601 
602 /*
603  * Close whatever file currently occupies the file descriptor slot
604  * and install the new file, usually NULL, in the file descriptor slot.
605  * The close must complete before we release the file descriptor slot.
606  * If newfp != NULL we only return an error if we can't allocate the
607  * slot so the caller knows that it needs to free the filep;
608  * in the other cases we return the error number from closef().
609  */
610 int
611 closeandsetf(int fd, file_t *newfp)
612 {
613 	proc_t *p = curproc;
614 	uf_info_t *fip = P_FINFO(p);
615 	uf_entry_t *ufp;
616 	file_t *fp;
617 	fpollinfo_t *fpip;
618 	portfd_t *pfd;
619 	int error;
620 
621 	if ((uint_t)fd >= fip->fi_nfiles) {
622 		if (newfp == NULL)
623 			return (EBADF);
624 		flist_grow(fd);
625 	}
626 
627 	if (newfp != NULL) {
628 		/*
629 		 * If ufp is reserved but has no file pointer, it's in the
630 		 * transition between ufalloc() and setf().  We must wait
631 		 * for this transition to complete before assigning the
632 		 * new non-NULL file pointer.
633 		 */
634 		mutex_enter(&fip->fi_lock);
635 		if (fd == fip->fi_badfd) {
636 			mutex_exit(&fip->fi_lock);
637 			if (fip->fi_action > 0)
638 				tsignal(curthread, fip->fi_action);
639 			return (EBADF);
640 		}
641 		UF_ENTER(ufp, fip, fd);
642 		while (ufp->uf_busy && ufp->uf_file == NULL) {
643 			mutex_exit(&fip->fi_lock);
644 			cv_wait_stop(&ufp->uf_wanted_cv, &ufp->uf_lock, 250);
645 			UF_EXIT(ufp);
646 			mutex_enter(&fip->fi_lock);
647 			UF_ENTER(ufp, fip, fd);
648 		}
649 		if ((fp = ufp->uf_file) == NULL) {
650 			ASSERT(ufp->uf_fpollinfo == NULL);
651 			ASSERT(ufp->uf_flag == 0);
652 			fd_reserve(fip, fd, 1);
653 			ufp->uf_file = newfp;
654 			UF_EXIT(ufp);
655 			mutex_exit(&fip->fi_lock);
656 			return (0);
657 		}
658 		mutex_exit(&fip->fi_lock);
659 	} else {
660 		UF_ENTER(ufp, fip, fd);
661 		if ((fp = ufp->uf_file) == NULL) {
662 			UF_EXIT(ufp);
663 			return (EBADF);
664 		}
665 	}
666 
667 #ifdef C2_AUDIT
668 	/*
669 	 * archive per file audit data
670 	 */
671 	if (audit_active)
672 		(void) audit_getf(fd);
673 #endif
674 	ASSERT(ufp->uf_busy);
675 	ufp->uf_file = NULL;
676 	ufp->uf_flag = 0;
677 
678 	/*
679 	 * If the file descriptor reference count is non-zero, then
680 	 * some other lwp in the process is performing system call
681 	 * activity on the file.  To avoid blocking here for a long
682 	 * time (the other lwp might be in a long term sleep in its
683 	 * system call), we stop all other lwps in the process and
684 	 * scan them to find the ones with this fd as one of their
685 	 * active fds and set their a_stale flag so they will emerge
686 	 * from their system calls immediately.  post_syscall() will
687 	 * test the a_stale flag and set errno to EBADF.
688 	 */
689 	ASSERT(ufp->uf_refcnt == 0 || p->p_lwpcnt > 1);
690 	if (ufp->uf_refcnt > 0) {
691 		UF_EXIT(ufp);
692 		COUNT(afd_wait);
693 
694 		/*
695 		 * Make all other lwps hold in place, as if doing fork1().
696 		 * holdlwps(SHOLDFORK1) fails only if another lwp wants to
697 		 * perform a forkall() or the process is exiting.  In either
698 		 * case, all other lwps are either returning from their
699 		 * system calls (because of SHOLDFORK) or calling lwp_exit()
700 		 * (because of SEXITLWPS) so we don't need to scan them.
701 		 */
702 		if (holdlwps(SHOLDFORK1)) {
703 			kthread_t *t;
704 
705 			mutex_enter(&p->p_lock);
706 			for (t = curthread->t_forw; t != curthread;
707 			    t = t->t_forw) {
708 				if (is_active_fd(t, fd)) {
709 					t->t_activefd.a_stale = 1;
710 					t->t_post_sys = 1;
711 				}
712 			}
713 			continuelwps(p);
714 			mutex_exit(&p->p_lock);
715 		}
716 		UF_ENTER(ufp, fip, fd);
717 		ASSERT(ufp->uf_file == NULL);
718 	}
719 
720 	/*
721 	 * Wait for other lwps to stop using this file descriptor.
722 	 */
723 	while (ufp->uf_refcnt > 0) {
724 		cv_wait_stop(&ufp->uf_closing_cv, &ufp->uf_lock, 250);
725 		/*
726 		 * cv_wait_stop() drops ufp->uf_lock, so the file list
727 		 * can change.  Drop the lock on our (possibly) stale
728 		 * ufp and let UF_ENTER() find and lock the current ufp.
729 		 */
730 		UF_EXIT(ufp);
731 		UF_ENTER(ufp, fip, fd);
732 	}
733 
734 #ifdef DEBUG
735 	/*
736 	 * catch a watchfd on device's pollhead list but not on fpollinfo list
737 	 */
738 	if (ufp->uf_fpollinfo != NULL)
739 		checkwfdlist(fp->f_vnode, ufp->uf_fpollinfo);
740 #endif	/* DEBUG */
741 
742 	/*
743 	 * We may need to cleanup some cached poll states in t_pollstate
744 	 * before the fd can be reused. It is important that we don't
745 	 * access a stale thread structure. We will do the cleanup in two
746 	 * phases to avoid deadlock and holding uf_lock for too long.
747 	 * In phase 1, hold the uf_lock and call pollblockexit() to set
748 	 * state in t_pollstate struct so that a thread does not exit on
749 	 * us. In phase 2, we drop the uf_lock and call pollcacheclean().
750 	 */
751 	pfd = ufp->uf_portfd;
752 	ufp->uf_portfd = NULL;
753 	fpip = ufp->uf_fpollinfo;
754 	ufp->uf_fpollinfo = NULL;
755 	if (fpip != NULL)
756 		pollblockexit(fpip);
757 	UF_EXIT(ufp);
758 	if (fpip != NULL)
759 		pollcacheclean(fpip, fd);
760 	if (pfd)
761 		port_close_fd(pfd);
762 
763 	/*
764 	 * Keep the file descriptor entry reserved across the closef().
765 	 */
766 	error = closef(fp);
767 
768 	setf(fd, newfp);
769 
770 	/* Only return closef() error when closing is all we do */
771 	return (newfp == NULL ? error : 0);
772 }
773 
774 /*
775  * Decrement uf_refcnt; wakeup anyone waiting to close the file.
776  */
777 void
778 releasef(int fd)
779 {
780 	uf_info_t *fip = P_FINFO(curproc);
781 	uf_entry_t *ufp;
782 
783 	clear_active_fd(fd);	/* clear the active file descriptor */
784 
785 	UF_ENTER(ufp, fip, fd);
786 	ASSERT(ufp->uf_refcnt > 0);
787 	if (--ufp->uf_refcnt == 0)
788 		cv_broadcast(&ufp->uf_closing_cv);
789 	UF_EXIT(ufp);
790 }
791 
792 /*
793  * Identical to releasef() but can be called from another process.
794  */
795 void
796 areleasef(int fd, uf_info_t *fip)
797 {
798 	uf_entry_t *ufp;
799 
800 	UF_ENTER(ufp, fip, fd);
801 	ASSERT(ufp->uf_refcnt > 0);
802 	if (--ufp->uf_refcnt == 0)
803 		cv_broadcast(&ufp->uf_closing_cv);
804 	UF_EXIT(ufp);
805 }
806 
807 /*
808  * Duplicate all file descriptors across a fork.
809  */
810 void
811 flist_fork(uf_info_t *pfip, uf_info_t *cfip)
812 {
813 	int fd, nfiles;
814 	uf_entry_t *pufp, *cufp;
815 
816 	mutex_init(&cfip->fi_lock, NULL, MUTEX_DEFAULT, NULL);
817 	cfip->fi_rlist = NULL;
818 
819 	/*
820 	 * We don't need to hold fi_lock because all other lwp's in the
821 	 * parent have been held.
822 	 */
823 	cfip->fi_nfiles = nfiles = flist_minsize(pfip);
824 
825 	cfip->fi_list = kmem_zalloc(nfiles * sizeof (uf_entry_t), KM_SLEEP);
826 
827 	for (fd = 0, pufp = pfip->fi_list, cufp = cfip->fi_list; fd < nfiles;
828 	    fd++, pufp++, cufp++) {
829 		cufp->uf_file = pufp->uf_file;
830 		cufp->uf_alloc = pufp->uf_alloc;
831 		cufp->uf_flag = pufp->uf_flag;
832 		cufp->uf_busy = pufp->uf_busy;
833 		if (pufp->uf_file == NULL) {
834 			ASSERT(pufp->uf_flag == 0);
835 			if (pufp->uf_busy) {
836 				/*
837 				 * Grab locks to appease ASSERTs in fd_reserve
838 				 */
839 				mutex_enter(&cfip->fi_lock);
840 				mutex_enter(&cufp->uf_lock);
841 				fd_reserve(cfip, fd, -1);
842 				mutex_exit(&cufp->uf_lock);
843 				mutex_exit(&cfip->fi_lock);
844 			}
845 		}
846 	}
847 }
848 
849 /*
850  * Close all open file descriptors for the current process.
851  * This is only called from exit(), which is single-threaded,
852  * so we don't need any locking.
853  */
854 void
855 closeall(uf_info_t *fip)
856 {
857 	int fd;
858 	file_t *fp;
859 	uf_entry_t *ufp;
860 
861 	ufp = fip->fi_list;
862 	for (fd = 0; fd < fip->fi_nfiles; fd++, ufp++) {
863 		if ((fp = ufp->uf_file) != NULL) {
864 			ufp->uf_file = NULL;
865 			if (ufp->uf_portfd != NULL) {
866 				portfd_t *pfd;
867 				/* remove event port association */
868 				pfd = ufp->uf_portfd;
869 				ufp->uf_portfd = NULL;
870 				port_close_fd(pfd);
871 			}
872 			ASSERT(ufp->uf_fpollinfo == NULL);
873 			(void) closef(fp);
874 		}
875 	}
876 
877 	kmem_free(fip->fi_list, fip->fi_nfiles * sizeof (uf_entry_t));
878 	fip->fi_list = NULL;
879 	fip->fi_nfiles = 0;
880 	while (fip->fi_rlist != NULL) {
881 		uf_rlist_t *urp = fip->fi_rlist;
882 		fip->fi_rlist = urp->ur_next;
883 		kmem_free(urp->ur_list, urp->ur_nfiles * sizeof (uf_entry_t));
884 		kmem_free(urp, sizeof (uf_rlist_t));
885 	}
886 }
887 
888 /*
889  * Internal form of close.  Decrement reference count on file
890  * structure.  Decrement reference count on the vnode following
891  * removal of the referencing file structure.
892  */
893 int
894 closef(file_t *fp)
895 {
896 	vnode_t *vp;
897 	int error;
898 	int count;
899 	int flag;
900 	offset_t offset;
901 
902 #ifdef C2_AUDIT
903 	/*
904 	 * audit close of file (may be exit)
905 	 */
906 	if (audit_active)
907 		audit_closef(fp);
908 #endif
909 	ASSERT(MUTEX_NOT_HELD(&P_FINFO(curproc)->fi_lock));
910 
911 	mutex_enter(&fp->f_tlock);
912 
913 	ASSERT(fp->f_count > 0);
914 
915 	count = fp->f_count--;
916 	flag = fp->f_flag;
917 	offset = fp->f_offset;
918 
919 	vp = fp->f_vnode;
920 
921 	error = VOP_CLOSE(vp, flag, count, offset, fp->f_cred, NULL);
922 
923 	if (count > 1) {
924 		mutex_exit(&fp->f_tlock);
925 		return (error);
926 	}
927 	ASSERT(fp->f_count == 0);
928 	mutex_exit(&fp->f_tlock);
929 
930 	VN_RELE(vp);
931 #ifdef C2_AUDIT
932 	/*
933 	 * deallocate resources to audit_data
934 	 */
935 	if (audit_active)
936 		audit_unfalloc(fp);
937 #endif
938 	crfree(fp->f_cred);
939 	kmem_cache_free(file_cache, fp);
940 	return (error);
941 }
942 
943 /*
944  * This is a combination of ufalloc() and setf().
945  */
946 int
947 ufalloc_file(int start, file_t *fp)
948 {
949 	proc_t *p = curproc;
950 	uf_info_t *fip = P_FINFO(p);
951 	int filelimit;
952 	uf_entry_t *ufp;
953 	int nfiles;
954 	int fd;
955 
956 	/*
957 	 * Assertion is to convince the correctness of the following
958 	 * assignment for filelimit after casting to int.
959 	 */
960 	ASSERT(p->p_fno_ctl <= INT_MAX);
961 	filelimit = (int)p->p_fno_ctl;
962 
963 	for (;;) {
964 		mutex_enter(&fip->fi_lock);
965 		fd = fd_find(fip, start);
966 		if (fd >= 0 && fd == fip->fi_badfd) {
967 			start = fd + 1;
968 			mutex_exit(&fip->fi_lock);
969 			continue;
970 		}
971 		if ((uint_t)fd < filelimit)
972 			break;
973 		if (fd >= filelimit) {
974 			mutex_exit(&fip->fi_lock);
975 			mutex_enter(&p->p_lock);
976 			(void) rctl_action(rctlproc_legacy[RLIMIT_NOFILE],
977 			    p->p_rctls, p, RCA_SAFE);
978 			mutex_exit(&p->p_lock);
979 			return (-1);
980 		}
981 		/* fd_find() returned -1 */
982 		nfiles = fip->fi_nfiles;
983 		mutex_exit(&fip->fi_lock);
984 		flist_grow(MAX(start, nfiles));
985 	}
986 
987 	UF_ENTER(ufp, fip, fd);
988 	fd_reserve(fip, fd, 1);
989 	ASSERT(ufp->uf_file == NULL);
990 	ufp->uf_file = fp;
991 	UF_EXIT(ufp);
992 	mutex_exit(&fip->fi_lock);
993 	return (fd);
994 }
995 
996 /*
997  * Allocate a user file descriptor greater than or equal to "start".
998  */
999 int
1000 ufalloc(int start)
1001 {
1002 	return (ufalloc_file(start, NULL));
1003 }
1004 
1005 /*
1006  * Check that a future allocation of count fds on proc p has a good
1007  * chance of succeeding.  If not, do rctl processing as if we'd failed
1008  * the allocation.
1009  *
1010  * Our caller must guarantee that p cannot disappear underneath us.
1011  */
1012 int
1013 ufcanalloc(proc_t *p, uint_t count)
1014 {
1015 	uf_info_t *fip = P_FINFO(p);
1016 	int filelimit;
1017 	int current;
1018 
1019 	if (count == 0)
1020 		return (1);
1021 
1022 	ASSERT(p->p_fno_ctl <= INT_MAX);
1023 	filelimit = (int)p->p_fno_ctl;
1024 
1025 	mutex_enter(&fip->fi_lock);
1026 	current = flist_nalloc(fip);		/* # of in-use descriptors */
1027 	mutex_exit(&fip->fi_lock);
1028 
1029 	/*
1030 	 * If count is a positive integer, the worst that can happen is
1031 	 * an overflow to a negative value, which is caught by the >= 0 check.
1032 	 */
1033 	current += count;
1034 	if (count <= INT_MAX && current >= 0 && current <= filelimit)
1035 		return (1);
1036 
1037 	mutex_enter(&p->p_lock);
1038 	(void) rctl_action(rctlproc_legacy[RLIMIT_NOFILE],
1039 	    p->p_rctls, p, RCA_SAFE);
1040 	mutex_exit(&p->p_lock);
1041 	return (0);
1042 }
1043 
1044 /*
1045  * Allocate a user file descriptor and a file structure.
1046  * Initialize the descriptor to point at the file structure.
1047  * If fdp is NULL, the user file descriptor will not be allocated.
1048  */
1049 int
1050 falloc(vnode_t *vp, int flag, file_t **fpp, int *fdp)
1051 {
1052 	file_t *fp;
1053 	int fd;
1054 
1055 	if (fdp) {
1056 		if ((fd = ufalloc(0)) == -1)
1057 			return (EMFILE);
1058 	}
1059 	fp = kmem_cache_alloc(file_cache, KM_SLEEP);
1060 	/*
1061 	 * Note: falloc returns the fp locked
1062 	 */
1063 	mutex_enter(&fp->f_tlock);
1064 	fp->f_count = 1;
1065 	fp->f_flag = (ushort_t)flag;
1066 	fp->f_vnode = vp;
1067 	fp->f_offset = 0;
1068 	fp->f_audit_data = 0;
1069 	crhold(fp->f_cred = CRED());
1070 #ifdef C2_AUDIT
1071 	/*
1072 	 * allocate resources to audit_data
1073 	 */
1074 	if (audit_active)
1075 		audit_falloc(fp);
1076 #endif
1077 	*fpp = fp;
1078 	if (fdp)
1079 		*fdp = fd;
1080 	return (0);
1081 }
1082 
1083 /*ARGSUSED*/
1084 static int
1085 file_cache_constructor(void *buf, void *cdrarg, int kmflags)
1086 {
1087 	file_t *fp = buf;
1088 
1089 	mutex_init(&fp->f_tlock, NULL, MUTEX_DEFAULT, NULL);
1090 	return (0);
1091 }
1092 
1093 /*ARGSUSED*/
1094 static void
1095 file_cache_destructor(void *buf, void *cdrarg)
1096 {
1097 	file_t *fp = buf;
1098 
1099 	mutex_destroy(&fp->f_tlock);
1100 }
1101 
1102 void
1103 finit()
1104 {
1105 	file_cache = kmem_cache_create("file_cache", sizeof (file_t), 0,
1106 	    file_cache_constructor, file_cache_destructor, NULL, NULL, NULL, 0);
1107 }
1108 
1109 void
1110 unfalloc(file_t *fp)
1111 {
1112 	ASSERT(MUTEX_HELD(&fp->f_tlock));
1113 	if (--fp->f_count <= 0) {
1114 #ifdef C2_AUDIT
1115 		/*
1116 		 * deallocate resources to audit_data
1117 		 */
1118 		if (audit_active)
1119 			audit_unfalloc(fp);
1120 #endif
1121 		crfree(fp->f_cred);
1122 		mutex_exit(&fp->f_tlock);
1123 		kmem_cache_free(file_cache, fp);
1124 	} else
1125 		mutex_exit(&fp->f_tlock);
1126 }
1127 
1128 /*
1129  * Given a file descriptor, set the user's
1130  * file pointer to the given parameter.
1131  */
1132 void
1133 setf(int fd, file_t *fp)
1134 {
1135 	uf_info_t *fip = P_FINFO(curproc);
1136 	uf_entry_t *ufp;
1137 
1138 #ifdef C2_AUDIT
1139 	if (audit_active)
1140 		audit_setf(fp, fd);
1141 #endif /* C2_AUDIT */
1142 
1143 	if (fp == NULL) {
1144 		mutex_enter(&fip->fi_lock);
1145 		UF_ENTER(ufp, fip, fd);
1146 		fd_reserve(fip, fd, -1);
1147 		mutex_exit(&fip->fi_lock);
1148 	} else {
1149 		UF_ENTER(ufp, fip, fd);
1150 		ASSERT(ufp->uf_busy);
1151 	}
1152 	ASSERT(ufp->uf_fpollinfo == NULL);
1153 	ASSERT(ufp->uf_flag == 0);
1154 	ufp->uf_file = fp;
1155 	cv_broadcast(&ufp->uf_wanted_cv);
1156 	UF_EXIT(ufp);
1157 }
1158 
1159 /*
1160  * Given a file descriptor, return the file table flags, plus,
1161  * if this is a socket in asynchronous mode, the FASYNC flag.
1162  * getf() may or may not have been called before calling f_getfl().
1163  */
1164 int
1165 f_getfl(int fd, int *flagp)
1166 {
1167 	uf_info_t *fip = P_FINFO(curproc);
1168 	uf_entry_t *ufp;
1169 	file_t *fp;
1170 	int error;
1171 
1172 	if ((uint_t)fd >= fip->fi_nfiles)
1173 		error = EBADF;
1174 	else {
1175 		UF_ENTER(ufp, fip, fd);
1176 		if ((fp = ufp->uf_file) == NULL)
1177 			error = EBADF;
1178 		else {
1179 			vnode_t *vp = fp->f_vnode;
1180 			int flag = fp->f_flag;
1181 
1182 			/*
1183 			 * BSD fcntl() FASYNC compatibility.
1184 			 *
1185 			 * SCTP doesn't have an associated stream and thus
1186 			 * doesn't store flags on it.
1187 			 */
1188 			if ((vp->v_type == VSOCK) && (vp->v_stream != NULL))
1189 				flag |= sock_getfasync(vp);
1190 			*flagp = flag;
1191 			error = 0;
1192 		}
1193 		UF_EXIT(ufp);
1194 	}
1195 
1196 	return (error);
1197 }
1198 
1199 /*
1200  * Given a file descriptor, return the user's file flags.
1201  * Force the FD_CLOEXEC flag for writable self-open /proc files.
1202  * getf() may or may not have been called before calling f_getfd_error().
1203  */
1204 int
1205 f_getfd_error(int fd, int *flagp)
1206 {
1207 	uf_info_t *fip = P_FINFO(curproc);
1208 	uf_entry_t *ufp;
1209 	file_t *fp;
1210 	int flag;
1211 	int error;
1212 
1213 	if ((uint_t)fd >= fip->fi_nfiles)
1214 		error = EBADF;
1215 	else {
1216 		UF_ENTER(ufp, fip, fd);
1217 		if ((fp = ufp->uf_file) == NULL)
1218 			error = EBADF;
1219 		else {
1220 			flag = ufp->uf_flag;
1221 			if ((fp->f_flag & FWRITE) && pr_isself(fp->f_vnode))
1222 				flag |= FD_CLOEXEC;
1223 			*flagp = flag;
1224 			error = 0;
1225 		}
1226 		UF_EXIT(ufp);
1227 	}
1228 
1229 	return (error);
1230 }
1231 
1232 /*
1233  * getf() must have been called before calling f_getfd().
1234  */
1235 char
1236 f_getfd(int fd)
1237 {
1238 	int flag = 0;
1239 	(void) f_getfd_error(fd, &flag);
1240 	return ((char)flag);
1241 }
1242 
1243 /*
1244  * Given a file descriptor and file flags, set the user's file flags.
1245  * At present, the only valid flag is FD_CLOEXEC.
1246  * getf() may or may not have been called before calling f_setfd_error().
1247  */
1248 int
1249 f_setfd_error(int fd, int flags)
1250 {
1251 	uf_info_t *fip = P_FINFO(curproc);
1252 	uf_entry_t *ufp;
1253 	int error;
1254 
1255 	if ((uint_t)fd >= fip->fi_nfiles)
1256 		error = EBADF;
1257 	else {
1258 		UF_ENTER(ufp, fip, fd);
1259 		if (ufp->uf_file == NULL)
1260 			error = EBADF;
1261 		else {
1262 			ufp->uf_flag = flags & FD_CLOEXEC;
1263 			error = 0;
1264 		}
1265 		UF_EXIT(ufp);
1266 	}
1267 	return (error);
1268 }
1269 
1270 void
1271 f_setfd(int fd, char flags)
1272 {
1273 	(void) f_setfd_error(fd, flags);
1274 }
1275 
1276 #define	BADFD_MIN	3
1277 #define	BADFD_MAX	255
1278 
1279 /*
1280  * Attempt to allocate a file descriptor which is bad and which
1281  * is "poison" to the application.  It cannot be closed (except
1282  * on exec), allocated for a different use, etc.
1283  */
1284 int
1285 f_badfd(int start, int *fdp, int action)
1286 {
1287 	int fdr;
1288 	int badfd;
1289 	uf_info_t *fip = P_FINFO(curproc);
1290 
1291 #ifdef _LP64
1292 	/* No restrictions on 64 bit _file */
1293 	if (get_udatamodel() != DATAMODEL_ILP32)
1294 		return (EINVAL);
1295 #endif
1296 
1297 	if (start > BADFD_MAX || start < BADFD_MIN)
1298 		return (EINVAL);
1299 
1300 	if (action >= NSIG || action < 0)
1301 		return (EINVAL);
1302 
1303 	mutex_enter(&fip->fi_lock);
1304 	badfd = fip->fi_badfd;
1305 	mutex_exit(&fip->fi_lock);
1306 
1307 	if (badfd != -1)
1308 		return (EAGAIN);
1309 
1310 	fdr = ufalloc(start);
1311 
1312 	if (fdr > BADFD_MAX) {
1313 		setf(fdr, NULL);
1314 		return (EMFILE);
1315 	}
1316 	if (fdr < 0)
1317 		return (EMFILE);
1318 
1319 	mutex_enter(&fip->fi_lock);
1320 	if (fip->fi_badfd != -1) {
1321 		/* Lost race */
1322 		mutex_exit(&fip->fi_lock);
1323 		setf(fdr, NULL);
1324 		return (EAGAIN);
1325 	}
1326 	fip->fi_action = action;
1327 	fip->fi_badfd = fdr;
1328 	mutex_exit(&fip->fi_lock);
1329 	setf(fdr, NULL);
1330 
1331 	*fdp = fdr;
1332 
1333 	return (0);
1334 }
1335 
1336 /*
1337  * Allocate a file descriptor and assign it to the vnode "*vpp",
1338  * performing the usual open protocol upon it and returning the
1339  * file descriptor allocated.  It is the responsibility of the
1340  * caller to dispose of "*vpp" if any error occurs.
1341  */
1342 int
1343 fassign(vnode_t **vpp, int mode, int *fdp)
1344 {
1345 	file_t *fp;
1346 	int error;
1347 	int fd;
1348 
1349 	if (error = falloc((vnode_t *)NULL, mode, &fp, &fd))
1350 		return (error);
1351 	if (error = VOP_OPEN(vpp, mode, fp->f_cred, NULL)) {
1352 		setf(fd, NULL);
1353 		unfalloc(fp);
1354 		return (error);
1355 	}
1356 	fp->f_vnode = *vpp;
1357 	mutex_exit(&fp->f_tlock);
1358 	/*
1359 	 * Fill in the slot falloc reserved.
1360 	 */
1361 	setf(fd, fp);
1362 	*fdp = fd;
1363 	return (0);
1364 }
1365 
1366 /*
1367  * When a process forks it must increment the f_count of all file pointers
1368  * since there is a new process pointing at them.  fcnt_add(fip, 1) does this.
1369  * Since we are called when there is only 1 active lwp we don't need to
1370  * hold fi_lock or any uf_lock.  If the fork fails, fork_fail() calls
1371  * fcnt_add(fip, -1) to restore the counts.
1372  */
1373 void
1374 fcnt_add(uf_info_t *fip, int incr)
1375 {
1376 	int i;
1377 	uf_entry_t *ufp;
1378 	file_t *fp;
1379 
1380 	ufp = fip->fi_list;
1381 	for (i = 0; i < fip->fi_nfiles; i++, ufp++) {
1382 		if ((fp = ufp->uf_file) != NULL) {
1383 			mutex_enter(&fp->f_tlock);
1384 			ASSERT((incr == 1 && fp->f_count >= 1) ||
1385 			    (incr == -1 && fp->f_count >= 2));
1386 			fp->f_count += incr;
1387 			mutex_exit(&fp->f_tlock);
1388 		}
1389 	}
1390 }
1391 
1392 /*
1393  * This is called from exec to close all fd's that have the FD_CLOEXEC flag
1394  * set and also to close all self-open for write /proc file descriptors.
1395  */
1396 void
1397 close_exec(uf_info_t *fip)
1398 {
1399 	int fd;
1400 	file_t *fp;
1401 	fpollinfo_t *fpip;
1402 	uf_entry_t *ufp;
1403 	portfd_t *pfd;
1404 
1405 	ufp = fip->fi_list;
1406 	for (fd = 0; fd < fip->fi_nfiles; fd++, ufp++) {
1407 		if ((fp = ufp->uf_file) != NULL &&
1408 		    ((ufp->uf_flag & FD_CLOEXEC) ||
1409 		    ((fp->f_flag & FWRITE) && pr_isself(fp->f_vnode)))) {
1410 			fpip = ufp->uf_fpollinfo;
1411 			mutex_enter(&fip->fi_lock);
1412 			mutex_enter(&ufp->uf_lock);
1413 			fd_reserve(fip, fd, -1);
1414 			mutex_exit(&fip->fi_lock);
1415 			ufp->uf_file = NULL;
1416 			ufp->uf_fpollinfo = NULL;
1417 			ufp->uf_flag = 0;
1418 			/*
1419 			 * We may need to cleanup some cached poll states
1420 			 * in t_pollstate before the fd can be reused. It
1421 			 * is important that we don't access a stale thread
1422 			 * structure. We will do the cleanup in two
1423 			 * phases to avoid deadlock and holding uf_lock for
1424 			 * too long. In phase 1, hold the uf_lock and call
1425 			 * pollblockexit() to set state in t_pollstate struct
1426 			 * so that a thread does not exit on us. In phase 2,
1427 			 * we drop the uf_lock and call pollcacheclean().
1428 			 */
1429 			pfd = ufp->uf_portfd;
1430 			ufp->uf_portfd = NULL;
1431 			if (fpip != NULL)
1432 				pollblockexit(fpip);
1433 			mutex_exit(&ufp->uf_lock);
1434 			if (fpip != NULL)
1435 				pollcacheclean(fpip, fd);
1436 			if (pfd)
1437 				port_close_fd(pfd);
1438 			(void) closef(fp);
1439 		}
1440 	}
1441 
1442 	/* Reset bad fd */
1443 	fip->fi_badfd = -1;
1444 	fip->fi_action = -1;
1445 }
1446 
1447 /*
1448  * Common routine for modifying attributes of named files.
1449  */
1450 int
1451 namesetattr(char *fnamep, enum symfollow followlink, vattr_t *vap, int flags)
1452 {
1453 	vnode_t *vp;
1454 	int error = 0;
1455 
1456 	if (error = lookupname(fnamep, UIO_USERSPACE, followlink, NULLVPP, &vp))
1457 		return (set_errno(error));
1458 	if (error = vpsetattr(vp, vap, flags))
1459 		(void) set_errno(error);
1460 	VN_RELE(vp);
1461 	return (error);
1462 }
1463 
1464 /*
1465  * Common routine for modifying attributes of files referenced
1466  * by descriptor.
1467  */
1468 int
1469 fdsetattr(int fd, vattr_t *vap)
1470 {
1471 	file_t *fp;
1472 	vnode_t *vp;
1473 	int error = 0;
1474 
1475 	if ((fp = getf(fd)) != NULL) {
1476 		vp = fp->f_vnode;
1477 		if (error = vpsetattr(vp, vap, 0)) {
1478 			(void) set_errno(error);
1479 		}
1480 		releasef(fd);
1481 	} else
1482 		error = set_errno(EBADF);
1483 	return (error);
1484 }
1485 
1486 /*
1487  * Common routine to set the attributes for the given vnode.
1488  * If the vnode is a file and the filesize is being manipulated,
1489  * this makes sure that there are no conflicting non-blocking
1490  * mandatory locks in that region.
1491  */
1492 static int
1493 vpsetattr(vnode_t *vp, vattr_t *vap, int flags)
1494 {
1495 	int error = 0;
1496 	int in_crit = 0;
1497 	u_offset_t	begin;
1498 	vattr_t	vattr;
1499 	ssize_t	length;
1500 
1501 	if (vn_is_readonly(vp)) {
1502 		error = EROFS;
1503 	}
1504 	if (!error && (vap->va_mask & AT_SIZE) &&
1505 	    nbl_need_check(vp)) {
1506 		nbl_start_crit(vp, RW_READER);
1507 		in_crit = 1;
1508 		vattr.va_mask = AT_SIZE;
1509 		if (!(error = VOP_GETATTR(vp, &vattr, 0, CRED(), NULL))) {
1510 			begin = vap->va_size > vattr.va_size ?
1511 					vattr.va_size : vap->va_size;
1512 			length = vattr.va_size > vap->va_size ?
1513 					vattr.va_size - vap->va_size :
1514 					vap->va_size - vattr.va_size;
1515 
1516 			if (nbl_conflict(vp, NBL_WRITE, begin, length, 0,
1517 			    NULL)) {
1518 				error = EACCES;
1519 			}
1520 		}
1521 	}
1522 	if (!error)
1523 		error = VOP_SETATTR(vp, vap, flags, CRED(), NULL);
1524 
1525 	if (in_crit)
1526 		nbl_end_crit(vp);
1527 
1528 	return (error);
1529 }
1530 
1531 /*
1532  * Return true if the given vnode is referenced by any
1533  * entry in the current process's file descriptor table.
1534  */
1535 int
1536 fisopen(vnode_t *vp)
1537 {
1538 	int fd;
1539 	file_t *fp;
1540 	vnode_t *ovp;
1541 	uf_info_t *fip = P_FINFO(curproc);
1542 	uf_entry_t *ufp;
1543 
1544 	mutex_enter(&fip->fi_lock);
1545 	for (fd = 0; fd < fip->fi_nfiles; fd++) {
1546 		UF_ENTER(ufp, fip, fd);
1547 		if ((fp = ufp->uf_file) != NULL &&
1548 		    (ovp = fp->f_vnode) != NULL && VN_CMP(vp, ovp)) {
1549 			UF_EXIT(ufp);
1550 			mutex_exit(&fip->fi_lock);
1551 			return (1);
1552 		}
1553 		UF_EXIT(ufp);
1554 	}
1555 	mutex_exit(&fip->fi_lock);
1556 	return (0);
1557 }
1558 
1559 /*
1560  * Return zero if at least one file currently open (by curproc) shouldn't be
1561  * allowed to change zones.
1562  */
1563 int
1564 files_can_change_zones(void)
1565 {
1566 	int fd;
1567 	file_t *fp;
1568 	uf_info_t *fip = P_FINFO(curproc);
1569 	uf_entry_t *ufp;
1570 
1571 	mutex_enter(&fip->fi_lock);
1572 	for (fd = 0; fd < fip->fi_nfiles; fd++) {
1573 		UF_ENTER(ufp, fip, fd);
1574 		if ((fp = ufp->uf_file) != NULL &&
1575 		    !vn_can_change_zones(fp->f_vnode)) {
1576 			UF_EXIT(ufp);
1577 			mutex_exit(&fip->fi_lock);
1578 			return (0);
1579 		}
1580 		UF_EXIT(ufp);
1581 	}
1582 	mutex_exit(&fip->fi_lock);
1583 	return (1);
1584 }
1585 
1586 #ifdef DEBUG
1587 
1588 /*
1589  * The following functions are only used in ASSERT()s elsewhere.
1590  * They do not modify the state of the system.
1591  */
1592 
1593 /*
1594  * Return true (1) if the current thread is in the fpollinfo
1595  * list for this file descriptor, else false (0).
1596  */
1597 static int
1598 curthread_in_plist(uf_entry_t *ufp)
1599 {
1600 	fpollinfo_t *fpip;
1601 
1602 	ASSERT(MUTEX_HELD(&ufp->uf_lock));
1603 	for (fpip = ufp->uf_fpollinfo; fpip; fpip = fpip->fp_next)
1604 		if (fpip->fp_thread == curthread)
1605 			return (1);
1606 	return (0);
1607 }
1608 
1609 /*
1610  * Sanity check to make sure that after lwp_exit(),
1611  * curthread does not appear on any fd's fpollinfo list.
1612  */
1613 void
1614 checkfpollinfo(void)
1615 {
1616 	int fd;
1617 	uf_info_t *fip = P_FINFO(curproc);
1618 	uf_entry_t *ufp;
1619 
1620 	mutex_enter(&fip->fi_lock);
1621 	for (fd = 0; fd < fip->fi_nfiles; fd++) {
1622 		UF_ENTER(ufp, fip, fd);
1623 		ASSERT(!curthread_in_plist(ufp));
1624 		UF_EXIT(ufp);
1625 	}
1626 	mutex_exit(&fip->fi_lock);
1627 }
1628 
1629 /*
1630  * Return true (1) if the current thread is in the fpollinfo
1631  * list for this file descriptor, else false (0).
1632  * This is the same as curthread_in_plist(),
1633  * but is called w/o holding uf_lock.
1634  */
1635 int
1636 infpollinfo(int fd)
1637 {
1638 	uf_info_t *fip = P_FINFO(curproc);
1639 	uf_entry_t *ufp;
1640 	int rc;
1641 
1642 	UF_ENTER(ufp, fip, fd);
1643 	rc = curthread_in_plist(ufp);
1644 	UF_EXIT(ufp);
1645 	return (rc);
1646 }
1647 
1648 #endif	/* DEBUG */
1649 
1650 /*
1651  * Add the curthread to fpollinfo list, meaning this fd is currently in the
1652  * thread's poll cache. Each lwp polling this file descriptor should call
1653  * this routine once.
1654  */
1655 void
1656 addfpollinfo(int fd)
1657 {
1658 	struct uf_entry *ufp;
1659 	fpollinfo_t *fpip;
1660 	uf_info_t *fip = P_FINFO(curproc);
1661 
1662 	fpip = kmem_zalloc(sizeof (fpollinfo_t), KM_SLEEP);
1663 	fpip->fp_thread = curthread;
1664 	UF_ENTER(ufp, fip, fd);
1665 	/*
1666 	 * Assert we are not already on the list, that is, that
1667 	 * this lwp did not call addfpollinfo twice for the same fd.
1668 	 */
1669 	ASSERT(!curthread_in_plist(ufp));
1670 	/*
1671 	 * addfpollinfo is always done inside the getf/releasef pair.
1672 	 */
1673 	ASSERT(ufp->uf_refcnt >= 1);
1674 	fpip->fp_next = ufp->uf_fpollinfo;
1675 	ufp->uf_fpollinfo = fpip;
1676 	UF_EXIT(ufp);
1677 }
1678 
1679 /*
1680  * delete curthread from fpollinfo list.
1681  */
1682 /*ARGSUSED*/
1683 void
1684 delfpollinfo(int fd)
1685 {
1686 	struct uf_entry *ufp;
1687 	struct fpollinfo *fpip;
1688 	struct fpollinfo **fpipp;
1689 	uf_info_t *fip = P_FINFO(curproc);
1690 
1691 	UF_ENTER(ufp, fip, fd);
1692 	if (ufp->uf_fpollinfo == NULL) {
1693 		UF_EXIT(ufp);
1694 		return;
1695 	}
1696 	ASSERT(ufp->uf_busy);
1697 	/*
1698 	 * Find and delete curthread from the list.
1699 	 */
1700 	fpipp = &ufp->uf_fpollinfo;
1701 	while ((fpip = *fpipp)->fp_thread != curthread)
1702 		fpipp = &fpip->fp_next;
1703 	*fpipp = fpip->fp_next;
1704 	kmem_free(fpip, sizeof (fpollinfo_t));
1705 	/*
1706 	 * Assert that we are not still on the list, that is, that
1707 	 * this lwp did not call addfpollinfo twice for the same fd.
1708 	 */
1709 	ASSERT(!curthread_in_plist(ufp));
1710 	UF_EXIT(ufp);
1711 }
1712 
1713 /*
1714  * fd is associated with a port. pfd is a pointer to the fd entry in the
1715  * cache of the port.
1716  */
1717 
1718 void
1719 addfd_port(int fd, portfd_t *pfd)
1720 {
1721 	struct uf_entry *ufp;
1722 	uf_info_t *fip = P_FINFO(curproc);
1723 
1724 	UF_ENTER(ufp, fip, fd);
1725 	/*
1726 	 * addfd_port is always done inside the getf/releasef pair.
1727 	 */
1728 	ASSERT(ufp->uf_refcnt >= 1);
1729 	if (ufp->uf_portfd == NULL) {
1730 		/* first entry */
1731 		ufp->uf_portfd = pfd;
1732 		pfd->pfd_next = NULL;
1733 	} else {
1734 		pfd->pfd_next = ufp->uf_portfd;
1735 		ufp->uf_portfd = pfd;
1736 		pfd->pfd_next->pfd_prev = pfd;
1737 	}
1738 	UF_EXIT(ufp);
1739 }
1740 
1741 void
1742 delfd_port(int fd, portfd_t *pfd)
1743 {
1744 	struct uf_entry *ufp;
1745 	uf_info_t *fip = P_FINFO(curproc);
1746 
1747 	UF_ENTER(ufp, fip, fd);
1748 	/*
1749 	 * delfd_port is always done inside the getf/releasef pair.
1750 	 */
1751 	ASSERT(ufp->uf_refcnt >= 1);
1752 	if (ufp->uf_portfd == pfd) {
1753 		/* remove first entry */
1754 		ufp->uf_portfd = pfd->pfd_next;
1755 	} else {
1756 		pfd->pfd_prev->pfd_next = pfd->pfd_next;
1757 		if (pfd->pfd_next != NULL)
1758 			pfd->pfd_next->pfd_prev = pfd->pfd_prev;
1759 	}
1760 	UF_EXIT(ufp);
1761 }
1762 
1763 static void
1764 port_close_fd(portfd_t *pfd)
1765 {
1766 	portfd_t	*pfdn;
1767 
1768 	/*
1769 	 * At this point, no other thread should access
1770 	 * the portfd_t list for this fd. The uf_file, uf_portfd
1771 	 * pointers in the uf_entry_t struct for this fd would
1772 	 * be set to NULL.
1773 	 */
1774 	for (; pfd != NULL; pfd = pfdn) {
1775 		pfdn = pfd->pfd_next;
1776 		port_close_pfd(pfd);
1777 	}
1778 }
1779