xref: /netbsd-src/sys/fs/union/union_subr.c (revision 404fbe5fb94ca1e054339640cabb2801ce52dd30)
1 /*	$NetBSD: union_subr.c,v 1.34 2008/12/17 20:51:35 cegger Exp $	*/
2 
3 /*
4  * Copyright (c) 1994
5  *	The Regents of the University of California.  All rights reserved.
6  *
7  * This code is derived from software contributed to Berkeley by
8  * Jan-Simon Pendry.
9  *
10  * Redistribution and use in source and binary forms, with or without
11  * modification, are permitted provided that the following conditions
12  * are met:
13  * 1. Redistributions of source code must retain the above copyright
14  *    notice, this list of conditions and the following disclaimer.
15  * 2. Redistributions in binary form must reproduce the above copyright
16  *    notice, this list of conditions and the following disclaimer in the
17  *    documentation and/or other materials provided with the distribution.
18  * 3. Neither the name of the University nor the names of its contributors
19  *    may be used to endorse or promote products derived from this software
20  *    without specific prior written permission.
21  *
22  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
23  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
24  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
25  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
26  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
27  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
28  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
29  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
30  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
31  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
32  * SUCH DAMAGE.
33  *
34  *	@(#)union_subr.c	8.20 (Berkeley) 5/20/95
35  */
36 
37 /*
38  * Copyright (c) 1994 Jan-Simon Pendry
39  *
40  * This code is derived from software contributed to Berkeley by
41  * Jan-Simon Pendry.
42  *
43  * Redistribution and use in source and binary forms, with or without
44  * modification, are permitted provided that the following conditions
45  * are met:
46  * 1. Redistributions of source code must retain the above copyright
47  *    notice, this list of conditions and the following disclaimer.
48  * 2. Redistributions in binary form must reproduce the above copyright
49  *    notice, this list of conditions and the following disclaimer in the
50  *    documentation and/or other materials provided with the distribution.
51  * 3. All advertising materials mentioning features or use of this software
52  *    must display the following acknowledgement:
53  *	This product includes software developed by the University of
54  *	California, Berkeley and its contributors.
55  * 4. Neither the name of the University nor the names of its contributors
56  *    may be used to endorse or promote products derived from this software
57  *    without specific prior written permission.
58  *
59  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
60  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
61  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
62  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
63  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
64  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
65  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
66  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
67  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
68  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
69  * SUCH DAMAGE.
70  *
71  *	@(#)union_subr.c	8.20 (Berkeley) 5/20/95
72  */
73 
74 #include <sys/cdefs.h>
75 __KERNEL_RCSID(0, "$NetBSD: union_subr.c,v 1.34 2008/12/17 20:51:35 cegger Exp $");
76 
77 #include <sys/param.h>
78 #include <sys/systm.h>
79 #include <sys/proc.h>
80 #include <sys/time.h>
81 #include <sys/kernel.h>
82 #include <sys/vnode.h>
83 #include <sys/namei.h>
84 #include <sys/malloc.h>
85 #include <sys/file.h>
86 #include <sys/filedesc.h>
87 #include <sys/queue.h>
88 #include <sys/mount.h>
89 #include <sys/stat.h>
90 #include <sys/kauth.h>
91 
92 #include <uvm/uvm_extern.h>
93 
94 #include <fs/union/union.h>
95 
96 /* must be power of two, otherwise change UNION_HASH() */
97 #define NHASH 32
98 
99 /* unsigned int ... */
100 #define UNION_HASH(u, l) \
101 	(((((unsigned long) (u)) + ((unsigned long) l)) >> 8) & (NHASH-1))
102 
103 static LIST_HEAD(unhead, union_node) unhead[NHASH];
104 static int unvplock[NHASH];
105 
106 static int union_list_lock(int);
107 static void union_list_unlock(int);
108 void union_updatevp(struct union_node *, struct vnode *, struct vnode *);
109 static int union_relookup(struct union_mount *, struct vnode *,
110 			       struct vnode **, struct componentname *,
111 			       struct componentname *, const char *, int);
112 int union_vn_close(struct vnode *, int, kauth_cred_t, struct lwp *);
113 static void union_dircache_r(struct vnode *, struct vnode ***, int *);
114 struct vnode *union_dircache(struct vnode *, struct lwp *);
115 
116 void
117 union_init(void)
118 {
119 	int i;
120 
121 	for (i = 0; i < NHASH; i++)
122 		LIST_INIT(&unhead[i]);
123 	memset(unvplock, 0, sizeof(unvplock));
124 }
125 
126 /*
127  * Free global unionfs resources.
128  */
129 void
130 union_done(void)
131 {
132 
133 	/* Make sure to unset the readdir hook. */
134 	vn_union_readdir_hook = NULL;
135 }
136 
137 static int
138 union_list_lock(int ix)
139 {
140 
141 	if (unvplock[ix] & UN_LOCKED) {
142 		unvplock[ix] |= UN_WANTED;
143 		(void) tsleep(&unvplock[ix], PINOD, "unionlk", 0);
144 		return (1);
145 	}
146 
147 	unvplock[ix] |= UN_LOCKED;
148 
149 	return (0);
150 }
151 
152 static void
153 union_list_unlock(int ix)
154 {
155 
156 	unvplock[ix] &= ~UN_LOCKED;
157 
158 	if (unvplock[ix] & UN_WANTED) {
159 		unvplock[ix] &= ~UN_WANTED;
160 		wakeup(&unvplock[ix]);
161 	}
162 }
163 
164 void
165 union_updatevp(struct union_node *un, struct vnode *uppervp,
166 	struct vnode *lowervp)
167 {
168 	int ohash = UNION_HASH(un->un_uppervp, un->un_lowervp);
169 	int nhash = UNION_HASH(uppervp, lowervp);
170 	int docache = (lowervp != NULLVP || uppervp != NULLVP);
171 	int lhash, uhash;
172 
173 	/*
174 	 * Ensure locking is ordered from lower to higher
175 	 * to avoid deadlocks.
176 	 */
177 	if (nhash < ohash) {
178 		lhash = nhash;
179 		uhash = ohash;
180 	} else {
181 		lhash = ohash;
182 		uhash = nhash;
183 	}
184 
185 	if (lhash != uhash)
186 		while (union_list_lock(lhash))
187 			continue;
188 
189 	while (union_list_lock(uhash))
190 		continue;
191 
192 	if (ohash != nhash || !docache) {
193 		if (un->un_flags & UN_CACHED) {
194 			un->un_flags &= ~UN_CACHED;
195 			LIST_REMOVE(un, un_cache);
196 		}
197 	}
198 
199 	if (ohash != nhash)
200 		union_list_unlock(ohash);
201 
202 	if (un->un_lowervp != lowervp) {
203 		if (un->un_lowervp) {
204 			vrele(un->un_lowervp);
205 			if (un->un_path) {
206 				free(un->un_path, M_TEMP);
207 				un->un_path = 0;
208 			}
209 			if (un->un_dirvp) {
210 				vrele(un->un_dirvp);
211 				un->un_dirvp = NULLVP;
212 			}
213 		}
214 		un->un_lowervp = lowervp;
215 		un->un_lowersz = VNOVAL;
216 	}
217 
218 	if (un->un_uppervp != uppervp) {
219 		if (un->un_uppervp)
220 			vrele(un->un_uppervp);
221 
222 		un->un_uppervp = uppervp;
223 		un->un_uppersz = VNOVAL;
224 	}
225 
226 	if (docache && (ohash != nhash)) {
227 		LIST_INSERT_HEAD(&unhead[nhash], un, un_cache);
228 		un->un_flags |= UN_CACHED;
229 	}
230 
231 	union_list_unlock(nhash);
232 }
233 
234 void
235 union_newlower(struct union_node *un, struct vnode *lowervp)
236 {
237 
238 	union_updatevp(un, un->un_uppervp, lowervp);
239 }
240 
241 void
242 union_newupper(struct union_node *un, struct vnode *uppervp)
243 {
244 
245 	union_updatevp(un, uppervp, un->un_lowervp);
246 }
247 
248 /*
249  * Keep track of size changes in the underlying vnodes.
250  * If the size changes, then callback to the vm layer
251  * giving priority to the upper layer size.
252  */
253 void
254 union_newsize(struct vnode *vp, off_t uppersz, off_t lowersz)
255 {
256 	struct union_node *un;
257 	off_t sz;
258 
259 	/* only interested in regular files */
260 	if (vp->v_type != VREG) {
261 		uvm_vnp_setsize(vp, 0);
262 		return;
263 	}
264 
265 	un = VTOUNION(vp);
266 	sz = VNOVAL;
267 
268 	if ((uppersz != VNOVAL) && (un->un_uppersz != uppersz)) {
269 		un->un_uppersz = uppersz;
270 		if (sz == VNOVAL)
271 			sz = un->un_uppersz;
272 	}
273 
274 	if ((lowersz != VNOVAL) && (un->un_lowersz != lowersz)) {
275 		un->un_lowersz = lowersz;
276 		if (sz == VNOVAL)
277 			sz = un->un_lowersz;
278 	}
279 
280 	if (sz != VNOVAL) {
281 #ifdef UNION_DIAGNOSTIC
282 		printf("union: %s size now %qd\n",
283 		    uppersz != VNOVAL ? "upper" : "lower", sz);
284 #endif
285 		uvm_vnp_setsize(vp, sz);
286 	}
287 }
288 
289 /*
290  * allocate a union_node/vnode pair.  the vnode is
291  * referenced and locked.  the new vnode is returned
292  * via (vpp).  (mp) is the mountpoint of the union filesystem,
293  * (dvp) is the parent directory where the upper layer object
294  * should exist (but doesn't) and (cnp) is the componentname
295  * information which is partially copied to allow the upper
296  * layer object to be created at a later time.  (uppervp)
297  * and (lowervp) reference the upper and lower layer objects
298  * being mapped.  either, but not both, can be nil.
299  * if supplied, (uppervp) is locked.
300  * the reference is either maintained in the new union_node
301  * object which is allocated, or they are vrele'd.
302  *
303  * all union_nodes are maintained on a singly-linked
304  * list.  new nodes are only allocated when they cannot
305  * be found on this list.  entries on the list are
306  * removed when the vfs reclaim entry is called.
307  *
308  * a single lock is kept for the entire list.  this is
309  * needed because the getnewvnode() function can block
310  * waiting for a vnode to become free, in which case there
311  * may be more than one process trying to get the same
312  * vnode.  this lock is only taken if we are going to
313  * call getnewvnode, since the kernel itself is single-threaded.
314  *
315  * if an entry is found on the list, then call vget() to
316  * take a reference.  this is done because there may be
317  * zero references to it and so it needs to removed from
318  * the vnode free list.
319  */
320 int
321 union_allocvp(
322 	struct vnode **vpp,
323 	struct mount *mp,
324 	struct vnode *undvp,		/* parent union vnode */
325 	struct vnode *dvp,		/* may be null */
326 	struct componentname *cnp,	/* may be null */
327 	struct vnode *uppervp,		/* may be null */
328 	struct vnode *lowervp,		/* may be null */
329 	int docache)
330 {
331 	int error;
332 	struct vattr va;
333 	struct union_node *un = NULL;
334 	struct vnode *xlowervp = NULLVP;
335 	struct union_mount *um = MOUNTTOUNIONMOUNT(mp);
336 	voff_t uppersz, lowersz;
337 	int hash = 0;
338 	int vflag, iflag;
339 	int try;
340 
341 	if (uppervp == NULLVP && lowervp == NULLVP)
342 		panic("union: unidentifiable allocation");
343 
344 	if (uppervp && lowervp && (uppervp->v_type != lowervp->v_type)) {
345 		xlowervp = lowervp;
346 		lowervp = NULLVP;
347 	}
348 
349 	/* detect the root vnode (and aliases) */
350 	iflag = VI_LAYER;
351 	vflag = 0;
352 	if ((uppervp == um->um_uppervp) &&
353 	    ((lowervp == NULLVP) || lowervp == um->um_lowervp)) {
354 		if (lowervp == NULLVP) {
355 			lowervp = um->um_lowervp;
356 			if (lowervp != NULLVP)
357 				VREF(lowervp);
358 		}
359 		iflag = 0;
360 		vflag = VV_ROOT;
361 	}
362 
363 loop:
364 	if (!docache) {
365 		un = 0;
366 	} else for (try = 0; try < 3; try++) {
367 		switch (try) {
368 		case 0:
369 			if (lowervp == NULLVP)
370 				continue;
371 			hash = UNION_HASH(uppervp, lowervp);
372 			break;
373 
374 		case 1:
375 			if (uppervp == NULLVP)
376 				continue;
377 			hash = UNION_HASH(uppervp, NULLVP);
378 			break;
379 
380 		case 2:
381 			if (lowervp == NULLVP)
382 				continue;
383 			hash = UNION_HASH(NULLVP, lowervp);
384 			break;
385 		}
386 
387 		while (union_list_lock(hash))
388 			continue;
389 
390 		for (un = unhead[hash].lh_first; un != 0;
391 					un = un->un_cache.le_next) {
392 			if ((un->un_lowervp == lowervp ||
393 			     un->un_lowervp == NULLVP) &&
394 			    (un->un_uppervp == uppervp ||
395 			     un->un_uppervp == NULLVP) &&
396 			    (UNIONTOV(un)->v_mount == mp)) {
397 				if (vget(UNIONTOV(un), 0)) {
398 					union_list_unlock(hash);
399 					goto loop;
400 				}
401 				break;
402 			}
403 		}
404 
405 		union_list_unlock(hash);
406 
407 		if (un)
408 			break;
409 	}
410 
411 	if (un) {
412 		/*
413 		 * Obtain a lock on the union_node.
414 		 * uppervp is locked, though un->un_uppervp
415 		 * may not be.  this doesn't break the locking
416 		 * hierarchy since in the case that un->un_uppervp
417 		 * is not yet locked it will be vrele'd and replaced
418 		 * with uppervp.
419 		 */
420 
421 		if ((dvp != NULLVP) && (uppervp == dvp)) {
422 			/*
423 			 * Access ``.'', so (un) will already
424 			 * be locked.  Since this process has
425 			 * the lock on (uppervp) no other
426 			 * process can hold the lock on (un).
427 			 */
428 #ifdef DIAGNOSTIC
429 			if ((un->un_flags & UN_LOCKED) == 0)
430 				panic("union: . not locked");
431 			else if (curproc && un->un_pid != curproc->p_pid &&
432 				    un->un_pid > -1 && curproc->p_pid > -1)
433 				panic("union: allocvp not lock owner");
434 #endif
435 		} else {
436 			if (un->un_flags & UN_LOCKED) {
437 				vrele(UNIONTOV(un));
438 				un->un_flags |= UN_WANTED;
439 				(void) tsleep(&un->un_flags, PINOD,
440 				    "unionalloc", 0);
441 				goto loop;
442 			}
443 			un->un_flags |= UN_LOCKED;
444 
445 #ifdef DIAGNOSTIC
446 			if (curproc)
447 				un->un_pid = curproc->p_pid;
448 			else
449 				un->un_pid = -1;
450 #endif
451 		}
452 
453 		/*
454 		 * At this point, the union_node is locked,
455 		 * un->un_uppervp may not be locked, and uppervp
456 		 * is locked or nil.
457 		 */
458 
459 		/*
460 		 * Save information about the upper layer.
461 		 */
462 		if (uppervp != un->un_uppervp) {
463 			union_newupper(un, uppervp);
464 		} else if (uppervp) {
465 			vrele(uppervp);
466 		}
467 
468 		if (un->un_uppervp) {
469 			un->un_flags |= UN_ULOCK;
470 			un->un_flags &= ~UN_KLOCK;
471 		}
472 
473 		/*
474 		 * Save information about the lower layer.
475 		 * This needs to keep track of pathname
476 		 * and directory information which union_vn_create
477 		 * might need.
478 		 */
479 		if (lowervp != un->un_lowervp) {
480 			union_newlower(un, lowervp);
481 			if (cnp && (lowervp != NULLVP)) {
482 				un->un_hash = cnp->cn_hash;
483 				un->un_path = malloc(cnp->cn_namelen+1,
484 						M_TEMP, M_WAITOK);
485 				memcpy(un->un_path, cnp->cn_nameptr,
486 						cnp->cn_namelen);
487 				un->un_path[cnp->cn_namelen] = '\0';
488 				VREF(dvp);
489 				un->un_dirvp = dvp;
490 			}
491 		} else if (lowervp) {
492 			vrele(lowervp);
493 		}
494 		*vpp = UNIONTOV(un);
495 		return (0);
496 	}
497 
498 	uppersz = lowersz = VNOVAL;
499 	if (uppervp != NULLVP)
500 		if (VOP_GETATTR(uppervp, &va, FSCRED) == 0)
501 			uppersz = va.va_size;
502 	if (lowervp != NULLVP)
503 		if (VOP_GETATTR(lowervp, &va, FSCRED) == 0)
504 			lowersz = va.va_size;
505 
506 	if (docache) {
507 		/*
508 		 * otherwise lock the vp list while we call getnewvnode
509 		 * since that can block.
510 		 */
511 		hash = UNION_HASH(uppervp, lowervp);
512 
513 		if (union_list_lock(hash))
514 			goto loop;
515 	}
516 
517 	error = getnewvnode(VT_UNION, mp, union_vnodeop_p, vpp);
518 	if (error) {
519 		if (uppervp) {
520 			if (dvp == uppervp)
521 				vrele(uppervp);
522 			else
523 				vput(uppervp);
524 		}
525 		if (lowervp)
526 			vrele(lowervp);
527 
528 		goto out;
529 	}
530 
531 	(*vpp)->v_data = malloc(sizeof(struct union_node), M_TEMP, M_WAITOK);
532 
533 	(*vpp)->v_vflag |= vflag;
534 	(*vpp)->v_iflag |= iflag;
535 	(*vpp)->v_vnlock = NULL;	/* Make upper layers call VOP_LOCK */
536 	if (uppervp)
537 		(*vpp)->v_type = uppervp->v_type;
538 	else
539 		(*vpp)->v_type = lowervp->v_type;
540 	un = VTOUNION(*vpp);
541 	un->un_vnode = *vpp;
542 	un->un_uppervp = uppervp;
543 	un->un_lowervp = lowervp;
544 	un->un_pvp = undvp;
545 	if (undvp != NULLVP)
546 		VREF(undvp);
547 	un->un_dircache = 0;
548 	un->un_openl = 0;
549 	un->un_flags = UN_LOCKED;
550 
551 	un->un_uppersz = VNOVAL;
552 	un->un_lowersz = VNOVAL;
553 	union_newsize(*vpp, uppersz, lowersz);
554 
555 	if (un->un_uppervp)
556 		un->un_flags |= UN_ULOCK;
557 #ifdef DIAGNOSTIC
558 	if (curproc)
559 		un->un_pid = curproc->p_pid;
560 	else
561 		un->un_pid = -1;
562 #endif
563 	if (dvp && cnp && (lowervp != NULLVP)) {
564 		un->un_hash = cnp->cn_hash;
565 		un->un_path = malloc(cnp->cn_namelen+1, M_TEMP, M_WAITOK);
566 		memcpy(un->un_path, cnp->cn_nameptr, cnp->cn_namelen);
567 		un->un_path[cnp->cn_namelen] = '\0';
568 		VREF(dvp);
569 		un->un_dirvp = dvp;
570 	} else {
571 		un->un_hash = 0;
572 		un->un_path = 0;
573 		un->un_dirvp = 0;
574 	}
575 
576 	if (docache) {
577 		LIST_INSERT_HEAD(&unhead[hash], un, un_cache);
578 		un->un_flags |= UN_CACHED;
579 	}
580 
581 	if (xlowervp)
582 		vrele(xlowervp);
583 
584 out:
585 	if (docache)
586 		union_list_unlock(hash);
587 
588 	return (error);
589 }
590 
591 int
592 union_freevp(struct vnode *vp)
593 {
594 	struct union_node *un = VTOUNION(vp);
595 
596 	if (un->un_flags & UN_CACHED) {
597 		un->un_flags &= ~UN_CACHED;
598 		LIST_REMOVE(un, un_cache);
599 	}
600 
601 	if (un->un_pvp != NULLVP)
602 		vrele(un->un_pvp);
603 	if (un->un_uppervp != NULLVP)
604 		vrele(un->un_uppervp);
605 	if (un->un_lowervp != NULLVP)
606 		vrele(un->un_lowervp);
607 	if (un->un_dirvp != NULLVP)
608 		vrele(un->un_dirvp);
609 	if (un->un_path)
610 		free(un->un_path, M_TEMP);
611 
612 	free(vp->v_data, M_TEMP);
613 	vp->v_data = NULL;
614 
615 	return (0);
616 }
617 
618 /*
619  * copyfile.  copy the vnode (fvp) to the vnode (tvp)
620  * using a sequence of reads and writes.  both (fvp)
621  * and (tvp) are locked on entry and exit.
622  */
623 int
624 union_copyfile(struct vnode *fvp, struct vnode *tvp, kauth_cred_t cred,
625 	struct lwp *l)
626 {
627 	char *tbuf;
628 	struct uio uio;
629 	struct iovec iov;
630 	int error = 0;
631 
632 	/*
633 	 * strategy:
634 	 * allocate a buffer of size MAXBSIZE.
635 	 * loop doing reads and writes, keeping track
636 	 * of the current uio offset.
637 	 * give up at the first sign of trouble.
638 	 */
639 
640 	uio.uio_offset = 0;
641 	UIO_SETUP_SYSSPACE(&uio);
642 
643 	VOP_UNLOCK(fvp, 0);			/* XXX */
644 	vn_lock(fvp, LK_EXCLUSIVE | LK_RETRY);	/* XXX */
645 	VOP_UNLOCK(tvp, 0);			/* XXX */
646 	vn_lock(tvp, LK_EXCLUSIVE | LK_RETRY);	/* XXX */
647 
648 	tbuf = malloc(MAXBSIZE, M_TEMP, M_WAITOK);
649 
650 	/* ugly loop follows... */
651 	do {
652 		off_t offset = uio.uio_offset;
653 
654 		uio.uio_iov = &iov;
655 		uio.uio_iovcnt = 1;
656 		iov.iov_base = tbuf;
657 		iov.iov_len = MAXBSIZE;
658 		uio.uio_resid = iov.iov_len;
659 		uio.uio_rw = UIO_READ;
660 		error = VOP_READ(fvp, &uio, 0, cred);
661 
662 		if (error == 0) {
663 			uio.uio_iov = &iov;
664 			uio.uio_iovcnt = 1;
665 			iov.iov_base = tbuf;
666 			iov.iov_len = MAXBSIZE - uio.uio_resid;
667 			uio.uio_offset = offset;
668 			uio.uio_rw = UIO_WRITE;
669 			uio.uio_resid = iov.iov_len;
670 
671 			if (uio.uio_resid == 0)
672 				break;
673 
674 			do {
675 				error = VOP_WRITE(tvp, &uio, 0, cred);
676 			} while ((uio.uio_resid > 0) && (error == 0));
677 		}
678 
679 	} while (error == 0);
680 
681 	free(tbuf, M_TEMP);
682 	return (error);
683 }
684 
685 /*
686  * (un) is assumed to be locked on entry and remains
687  * locked on exit.
688  */
689 int
690 union_copyup(struct union_node *un, int docopy, kauth_cred_t cred,
691 	struct lwp *l)
692 {
693 	int error;
694 	struct vnode *lvp, *uvp;
695 	struct vattr lvattr, uvattr;
696 
697 	error = union_vn_create(&uvp, un, l);
698 	if (error)
699 		return (error);
700 
701 	/* at this point, uppervp is locked */
702 	union_newupper(un, uvp);
703 	un->un_flags |= UN_ULOCK;
704 
705 	lvp = un->un_lowervp;
706 
707 	if (docopy) {
708 		/*
709 		 * XX - should not ignore errors
710 		 * from VOP_CLOSE
711 		 */
712 		vn_lock(lvp, LK_EXCLUSIVE | LK_RETRY);
713 
714         	error = VOP_GETATTR(lvp, &lvattr, cred);
715 		if (error == 0)
716 			error = VOP_OPEN(lvp, FREAD, cred);
717 		if (error == 0) {
718 			error = union_copyfile(lvp, uvp, cred, l);
719 			(void) VOP_CLOSE(lvp, FREAD, cred);
720 		}
721 		if (error == 0) {
722 			/* Copy permissions up too */
723 			VATTR_NULL(&uvattr);
724 			uvattr.va_mode = lvattr.va_mode;
725 			uvattr.va_flags = lvattr.va_flags;
726         		error = VOP_SETATTR(uvp, &uvattr, cred);
727 		}
728 		VOP_UNLOCK(lvp, 0);
729 #ifdef UNION_DIAGNOSTIC
730 		if (error == 0)
731 			uprintf("union: copied up %s\n", un->un_path);
732 #endif
733 
734 	}
735 	union_vn_close(uvp, FWRITE, cred, l);
736 
737 	/*
738 	 * Subsequent IOs will go to the top layer, so
739 	 * call close on the lower vnode and open on the
740 	 * upper vnode to ensure that the filesystem keeps
741 	 * its references counts right.  This doesn't do
742 	 * the right thing with (cred) and (FREAD) though.
743 	 * Ignoring error returns is not right, either.
744 	 */
745 	if (error == 0) {
746 		int i;
747 
748 		vn_lock(lvp, LK_EXCLUSIVE | LK_RETRY);
749 		for (i = 0; i < un->un_openl; i++) {
750 			(void) VOP_CLOSE(lvp, FREAD, cred);
751 			(void) VOP_OPEN(uvp, FREAD, cred);
752 		}
753 		un->un_openl = 0;
754 		VOP_UNLOCK(lvp, 0);
755 	}
756 
757 	return (error);
758 
759 }
760 
761 static int
762 union_relookup(
763 	struct union_mount *um,
764 	struct vnode *dvp,
765 	struct vnode **vpp,
766 	struct componentname *cnp,
767 	struct componentname *cn,
768 	const char *path,
769 	int pathlen)
770 {
771 	int error;
772 
773 	/*
774 	 * A new componentname structure must be faked up because
775 	 * there is no way to know where the upper level cnp came
776 	 * from or what it is being used for.  This must duplicate
777 	 * some of the work done by NDINIT, some of the work done
778 	 * by namei, some of the work done by lookup and some of
779 	 * the work done by VOP_LOOKUP when given a CREATE flag.
780 	 * Conclusion: Horrible.
781 	 *
782 	 * The pathname buffer will be PNBUF_PUT'd by VOP_MKDIR.
783 	 */
784 	cn->cn_namelen = pathlen;
785 	if ((cn->cn_namelen + 1) > MAXPATHLEN)
786 		return (ENAMETOOLONG);
787 	cn->cn_pnbuf = PNBUF_GET();
788 	memcpy(cn->cn_pnbuf, path, cn->cn_namelen);
789 	cn->cn_pnbuf[cn->cn_namelen] = '\0';
790 
791 	cn->cn_nameiop = CREATE;
792 	cn->cn_flags = (LOCKPARENT|HASBUF|SAVENAME|ISLASTCN);
793 	if (um->um_op == UNMNT_ABOVE)
794 		cn->cn_cred = cnp->cn_cred;
795 	else
796 		cn->cn_cred = um->um_cred;
797 	cn->cn_nameptr = cn->cn_pnbuf;
798 	cn->cn_hash = cnp->cn_hash;
799 	cn->cn_consume = cnp->cn_consume;
800 
801 	error = relookup(dvp, vpp, cn);
802 	if (error) {
803 		PNBUF_PUT(cn->cn_pnbuf);
804 		cn->cn_pnbuf = 0;
805 	}
806 
807 	return (error);
808 }
809 
810 /*
811  * Create a shadow directory in the upper layer.
812  * The new vnode is returned locked.
813  *
814  * (um) points to the union mount structure for access to the
815  * the mounting process's credentials.
816  * (dvp) is the directory in which to create the shadow directory.
817  * it is unlocked on entry and exit.
818  * (cnp) is the componentname to be created.
819  * (vpp) is the returned newly created shadow directory, which
820  * is returned locked.
821  *
822  * N.B. We still attempt to create shadow directories even if the union
823  * is mounted read-only, which is a little nonintuitive.
824  */
825 int
826 union_mkshadow(struct union_mount *um, struct vnode *dvp,
827 	struct componentname *cnp, struct vnode **vpp)
828 {
829 	int error;
830 	struct vattr va;
831 	struct componentname cn;
832 
833 	vn_lock(dvp, LK_EXCLUSIVE | LK_RETRY);
834 	error = union_relookup(um, dvp, vpp, cnp, &cn,
835 			cnp->cn_nameptr, cnp->cn_namelen);
836 	if (error) {
837 		VOP_UNLOCK(dvp, 0);
838 		return (error);
839 	}
840 
841 	if (*vpp) {
842 		VOP_ABORTOP(dvp, &cn);
843 		if (dvp != *vpp)
844 			VOP_UNLOCK(dvp, 0);
845 		vput(*vpp);
846 		*vpp = NULLVP;
847 		return (EEXIST);
848 	}
849 
850 	/*
851 	 * policy: when creating the shadow directory in the
852 	 * upper layer, create it owned by the user who did
853 	 * the mount, group from parent directory, and mode
854 	 * 777 modified by umask (ie mostly identical to the
855 	 * mkdir syscall).  (jsp, kb)
856 	 */
857 
858 	VATTR_NULL(&va);
859 	va.va_type = VDIR;
860 	va.va_mode = um->um_cmode;
861 
862 	vref(dvp);
863 	error = VOP_MKDIR(dvp, vpp, &cn, &va);
864 	return (error);
865 }
866 
867 /*
868  * Create a whiteout entry in the upper layer.
869  *
870  * (um) points to the union mount structure for access to the
871  * the mounting process's credentials.
872  * (dvp) is the directory in which to create the whiteout.
873  * it is locked on entry and exit.
874  * (cnp) is the componentname to be created.
875  */
876 int
877 union_mkwhiteout(struct union_mount *um, struct vnode *dvp,
878 	struct componentname *cnp, char *path)
879 {
880 	int error;
881 	struct vnode *wvp;
882 	struct componentname cn;
883 
884 	VOP_UNLOCK(dvp, 0);
885 	vn_lock(dvp, LK_EXCLUSIVE | LK_RETRY);
886 	error = union_relookup(um, dvp, &wvp, cnp, &cn, path, strlen(path));
887 	if (error)
888 		return (error);
889 
890 	if (wvp) {
891 		VOP_ABORTOP(dvp, &cn);
892 		if (dvp != wvp)
893 			VOP_UNLOCK(dvp, 0);
894 		vput(wvp);
895 		return (EEXIST);
896 	}
897 
898 	error = VOP_WHITEOUT(dvp, &cn, CREATE);
899 	if (error)
900 		VOP_ABORTOP(dvp, &cn);
901 
902 	return (error);
903 }
904 
905 /*
906  * union_vn_create: creates and opens a new shadow file
907  * on the upper union layer.  this function is similar
908  * in spirit to calling vn_open but it avoids calling namei().
909  * the problem with calling namei is that a) it locks too many
910  * things, and b) it doesn't start at the "right" directory,
911  * whereas relookup is told where to start.
912  */
913 int
914 union_vn_create(struct vnode **vpp, struct union_node *un, struct lwp *l)
915 {
916 	struct vnode *vp;
917 	kauth_cred_t cred = l->l_cred;
918 	struct vattr vat;
919 	struct vattr *vap = &vat;
920 	int fmode = FFLAGS(O_WRONLY|O_CREAT|O_TRUNC|O_EXCL);
921 	int error;
922 	int cmode = UN_FILEMODE & ~l->l_proc->p_cwdi->cwdi_cmask;
923 	struct componentname cn;
924 
925 	*vpp = NULLVP;
926 
927 	/*
928 	 * Build a new componentname structure (for the same
929 	 * reasons outlines in union_mkshadow).
930 	 * The difference here is that the file is owned by
931 	 * the current user, rather than by the person who
932 	 * did the mount, since the current user needs to be
933 	 * able to write the file (that's why it is being
934 	 * copied in the first place).
935 	 */
936 	cn.cn_namelen = strlen(un->un_path);
937 	if ((cn.cn_namelen + 1) > MAXPATHLEN)
938 		return (ENAMETOOLONG);
939 	cn.cn_pnbuf = PNBUF_GET();
940 	memcpy(cn.cn_pnbuf, un->un_path, cn.cn_namelen+1);
941 	cn.cn_nameiop = CREATE;
942 	cn.cn_flags = (LOCKPARENT|HASBUF|SAVENAME|ISLASTCN);
943 	cn.cn_cred = l->l_cred;
944 	cn.cn_nameptr = cn.cn_pnbuf;
945 	cn.cn_hash = un->un_hash;
946 	cn.cn_consume = 0;
947 
948 	vn_lock(un->un_dirvp, LK_EXCLUSIVE | LK_RETRY);
949 	error = relookup(un->un_dirvp, &vp, &cn);
950 	if (error) {
951 		VOP_UNLOCK(un->un_dirvp, 0);
952 		return (error);
953 	}
954 
955 	if (vp) {
956 		VOP_ABORTOP(un->un_dirvp, &cn);
957 		if (un->un_dirvp != vp)
958 			VOP_UNLOCK(un->un_dirvp, 0);
959 		vput(vp);
960 		return (EEXIST);
961 	}
962 
963 	/*
964 	 * Good - there was no race to create the file
965 	 * so go ahead and create it.  The permissions
966 	 * on the file will be 0666 modified by the
967 	 * current user's umask.  Access to the file, while
968 	 * it is unioned, will require access to the top *and*
969 	 * bottom files.  Access when not unioned will simply
970 	 * require access to the top-level file.
971 	 * TODO: confirm choice of access permissions.
972 	 */
973 	VATTR_NULL(vap);
974 	vap->va_type = VREG;
975 	vap->va_mode = cmode;
976 	vref(un->un_dirvp);
977 	if ((error = VOP_CREATE(un->un_dirvp, &vp, &cn, vap)) != 0)
978 		return (error);
979 
980 	if ((error = VOP_OPEN(vp, fmode, cred)) != 0) {
981 		vput(vp);
982 		return (error);
983 	}
984 
985 	vp->v_writecount++;
986 	*vpp = vp;
987 	return (0);
988 }
989 
990 int
991 union_vn_close(struct vnode *vp, int fmode, kauth_cred_t cred, struct lwp *l)
992 {
993 
994 	if (fmode & FWRITE)
995 		--vp->v_writecount;
996 	return (VOP_CLOSE(vp, fmode, cred));
997 }
998 
999 void
1000 union_removed_upper(struct union_node *un)
1001 {
1002 #if 1
1003 	/*
1004 	 * We do not set the uppervp to NULLVP here, because lowervp
1005 	 * may also be NULLVP, so this routine would end up creating
1006 	 * a bogus union node with no upper or lower VP (that causes
1007 	 * pain in many places that assume at least one VP exists).
1008 	 * Since we've removed this node from the cache hash chains,
1009 	 * it won't be found again.  When all current holders
1010 	 * release it, union_inactive() will vgone() it.
1011 	 */
1012 	union_diruncache(un);
1013 #else
1014 	union_newupper(un, NULLVP);
1015 #endif
1016 
1017 	if (un->un_flags & UN_CACHED) {
1018 		un->un_flags &= ~UN_CACHED;
1019 		LIST_REMOVE(un, un_cache);
1020 	}
1021 
1022 	if (un->un_flags & UN_ULOCK) {
1023 		un->un_flags &= ~UN_ULOCK;
1024 		VOP_UNLOCK(un->un_uppervp, 0);
1025 	}
1026 }
1027 
1028 #if 0
1029 struct vnode *
1030 union_lowervp(struct vnode *vp)
1031 {
1032 	struct union_node *un = VTOUNION(vp);
1033 
1034 	if ((un->un_lowervp != NULLVP) &&
1035 	    (vp->v_type == un->un_lowervp->v_type)) {
1036 		if (vget(un->un_lowervp, 0) == 0)
1037 			return (un->un_lowervp);
1038 	}
1039 
1040 	return (NULLVP);
1041 }
1042 #endif
1043 
1044 /*
1045  * determine whether a whiteout is needed
1046  * during a remove/rmdir operation.
1047  */
1048 int
1049 union_dowhiteout(struct union_node *un, kauth_cred_t cred)
1050 {
1051 	struct vattr va;
1052 
1053 	if (un->un_lowervp != NULLVP)
1054 		return (1);
1055 
1056 	if (VOP_GETATTR(un->un_uppervp, &va, cred) == 0 &&
1057 	    (va.va_flags & OPAQUE))
1058 		return (1);
1059 
1060 	return (0);
1061 }
1062 
1063 static void
1064 union_dircache_r(struct vnode *vp, struct vnode ***vppp, int *cntp)
1065 {
1066 	struct union_node *un;
1067 
1068 	if (vp->v_op != union_vnodeop_p) {
1069 		if (vppp) {
1070 			VREF(vp);
1071 			*(*vppp)++ = vp;
1072 			if (--(*cntp) == 0)
1073 				panic("union: dircache table too small");
1074 		} else {
1075 			(*cntp)++;
1076 		}
1077 
1078 		return;
1079 	}
1080 
1081 	un = VTOUNION(vp);
1082 	if (un->un_uppervp != NULLVP)
1083 		union_dircache_r(un->un_uppervp, vppp, cntp);
1084 	if (un->un_lowervp != NULLVP)
1085 		union_dircache_r(un->un_lowervp, vppp, cntp);
1086 }
1087 
1088 struct vnode *
1089 union_dircache(struct vnode *vp, struct lwp *l)
1090 {
1091 	int cnt;
1092 	struct vnode *nvp = NULLVP;
1093 	struct vnode **vpp;
1094 	struct vnode **dircache;
1095 	int error;
1096 
1097 	vn_lock(vp, LK_EXCLUSIVE | LK_RETRY);
1098 	dircache = VTOUNION(vp)->un_dircache;
1099 
1100 	nvp = NULLVP;
1101 
1102 	if (dircache == 0) {
1103 		cnt = 0;
1104 		union_dircache_r(vp, 0, &cnt);
1105 		cnt++;
1106 		dircache = (struct vnode **)
1107 				malloc(cnt * sizeof(struct vnode *),
1108 					M_TEMP, M_WAITOK);
1109 		vpp = dircache;
1110 		union_dircache_r(vp, &vpp, &cnt);
1111 		VTOUNION(vp)->un_dircache = dircache;
1112 		*vpp = NULLVP;
1113 		vpp = dircache + 1;
1114 	} else {
1115 		vpp = dircache;
1116 		do {
1117 			if (*vpp++ == VTOUNION(vp)->un_uppervp)
1118 				break;
1119 		} while (*vpp != NULLVP);
1120 	}
1121 
1122 	if (*vpp == NULLVP)
1123 		goto out;
1124 
1125 	vn_lock(*vpp, LK_EXCLUSIVE | LK_RETRY);
1126 	VREF(*vpp);
1127 	error = union_allocvp(&nvp, vp->v_mount, NULLVP, NULLVP, 0, *vpp, NULLVP, 0);
1128 	if (!error) {
1129 		VTOUNION(vp)->un_dircache = 0;
1130 		VTOUNION(nvp)->un_dircache = dircache;
1131 	}
1132 
1133 out:
1134 	VOP_UNLOCK(vp, 0);
1135 	return (nvp);
1136 }
1137 
1138 void
1139 union_diruncache(struct union_node *un)
1140 {
1141 	struct vnode **vpp;
1142 
1143 	if (un->un_dircache != 0) {
1144 		for (vpp = un->un_dircache; *vpp != NULLVP; vpp++)
1145 			vrele(*vpp);
1146 		free(un->un_dircache, M_TEMP);
1147 		un->un_dircache = 0;
1148 	}
1149 }
1150 
1151 /*
1152  * This hook is called from vn_readdir() to switch to lower directory
1153  * entry after the upper directory is read.
1154  */
1155 int
1156 union_readdirhook(struct vnode **vpp, struct file *fp, struct lwp *l)
1157 {
1158 	struct vnode *vp = *vpp, *lvp;
1159 	struct vattr va;
1160 	int error;
1161 
1162 	if (vp->v_op != union_vnodeop_p)
1163 		return (0);
1164 
1165 	if ((lvp = union_dircache(vp, l)) == NULLVP)
1166 		return (0);
1167 
1168 	/*
1169 	 * If the directory is opaque,
1170 	 * then don't show lower entries
1171 	 */
1172 	error = VOP_GETATTR(vp, &va, fp->f_cred);
1173 	if (error || (va.va_flags & OPAQUE)) {
1174 		vput(lvp);
1175 		return (error);
1176 	}
1177 
1178 	error = VOP_OPEN(lvp, FREAD, fp->f_cred);
1179 	if (error) {
1180 		vput(lvp);
1181 		return (error);
1182 	}
1183 	VOP_UNLOCK(lvp, 0);
1184 	fp->f_data = lvp;
1185 	fp->f_offset = 0;
1186 	error = vn_close(vp, FREAD, fp->f_cred);
1187 	if (error)
1188 		return (error);
1189 	*vpp = lvp;
1190 	return (0);
1191 }
1192