1 /*- 2 * Copyright (c) 1994 Jan-Simon Pendry 3 * Copyright (c) 1994 4 * The Regents of the University of California. All rights reserved. 5 * Copyright (c) 2005, 2006, 2012 Masanori Ozawa <ozawa@ongs.co.jp>, ONGS Inc. 6 * Copyright (c) 2006, 2012 Daichi Goto <daichi@freebsd.org> 7 * 8 * This code is derived from software contributed to Berkeley by 9 * Jan-Simon Pendry. 10 * 11 * Redistribution and use in source and binary forms, with or without 12 * modification, are permitted provided that the following conditions 13 * are met: 14 * 1. Redistributions of source code must retain the above copyright 15 * notice, this list of conditions and the following disclaimer. 16 * 2. Redistributions in binary form must reproduce the above copyright 17 * notice, this list of conditions and the following disclaimer in the 18 * documentation and/or other materials provided with the distribution. 19 * 4. Neither the name of the University nor the names of its contributors 20 * may be used to endorse or promote products derived from this software 21 * without specific prior written permission. 22 * 23 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND 24 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 25 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 26 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE 27 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 28 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 29 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 30 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 31 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 32 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 33 * SUCH DAMAGE. 34 * 35 * @(#)union_subr.c 8.20 (Berkeley) 5/20/95 36 * $FreeBSD$ 37 */ 38 39 #include <sys/param.h> 40 #include <sys/systm.h> 41 #include <sys/kernel.h> 42 #include <sys/lock.h> 43 #include <sys/mutex.h> 44 #include <sys/malloc.h> 45 #include <sys/mount.h> 46 #include <sys/namei.h> 47 #include <sys/proc.h> 48 #include <sys/vnode.h> 49 #include <sys/dirent.h> 50 #include <sys/fcntl.h> 51 #include <sys/filedesc.h> 52 #include <sys/stat.h> 53 #include <sys/resourcevar.h> 54 55 #include <security/mac/mac_framework.h> 56 57 #include <vm/uma.h> 58 59 #include <fs/unionfs/union.h> 60 61 #define NUNIONFSNODECACHE 16 62 63 static MALLOC_DEFINE(M_UNIONFSHASH, "UNIONFS hash", "UNIONFS hash table"); 64 MALLOC_DEFINE(M_UNIONFSNODE, "UNIONFS node", "UNIONFS vnode private part"); 65 MALLOC_DEFINE(M_UNIONFSPATH, "UNIONFS path", "UNIONFS path private part"); 66 67 /* 68 * Initialize 69 */ 70 int 71 unionfs_init(struct vfsconf *vfsp) 72 { 73 UNIONFSDEBUG("unionfs_init\n"); /* printed during system boot */ 74 return (0); 75 } 76 77 /* 78 * Uninitialize 79 */ 80 int 81 unionfs_uninit(struct vfsconf *vfsp) 82 { 83 return (0); 84 } 85 86 static struct unionfs_node_hashhead * 87 unionfs_get_hashhead(struct vnode *dvp, char *path) 88 { 89 int count; 90 char hash; 91 struct unionfs_node *unp; 92 93 hash = 0; 94 unp = VTOUNIONFS(dvp); 95 if (path != NULL) { 96 for (count = 0; path[count]; count++) 97 hash += path[count]; 98 } 99 100 return (&(unp->un_hashtbl[hash & (unp->un_hashmask)])); 101 } 102 103 /* 104 * Get the cached vnode. 105 */ 106 static struct vnode * 107 unionfs_get_cached_vnode(struct vnode *uvp, struct vnode *lvp, 108 struct vnode *dvp, char *path) 109 { 110 struct unionfs_node_hashhead *hd; 111 struct unionfs_node *unp; 112 struct vnode *vp; 113 114 KASSERT((uvp == NULLVP || uvp->v_type == VDIR), 115 ("unionfs_get_cached_vnode: v_type != VDIR")); 116 KASSERT((lvp == NULLVP || lvp->v_type == VDIR), 117 ("unionfs_get_cached_vnode: v_type != VDIR")); 118 119 VI_LOCK(dvp); 120 hd = unionfs_get_hashhead(dvp, path); 121 LIST_FOREACH(unp, hd, un_hash) { 122 if (!strcmp(unp->un_path, path)) { 123 vp = UNIONFSTOV(unp); 124 VI_LOCK_FLAGS(vp, MTX_DUPOK); 125 VI_UNLOCK(dvp); 126 vp->v_iflag &= ~VI_OWEINACT; 127 if ((vp->v_iflag & (VI_DOOMED | VI_DOINGINACT)) != 0) { 128 VI_UNLOCK(vp); 129 vp = NULLVP; 130 } else 131 VI_UNLOCK(vp); 132 return (vp); 133 } 134 } 135 VI_UNLOCK(dvp); 136 137 return (NULLVP); 138 } 139 140 /* 141 * Add the new vnode into cache. 142 */ 143 static struct vnode * 144 unionfs_ins_cached_vnode(struct unionfs_node *uncp, 145 struct vnode *dvp, char *path) 146 { 147 struct unionfs_node_hashhead *hd; 148 struct unionfs_node *unp; 149 struct vnode *vp; 150 151 KASSERT((uncp->un_uppervp==NULLVP || uncp->un_uppervp->v_type==VDIR), 152 ("unionfs_ins_cached_vnode: v_type != VDIR")); 153 KASSERT((uncp->un_lowervp==NULLVP || uncp->un_lowervp->v_type==VDIR), 154 ("unionfs_ins_cached_vnode: v_type != VDIR")); 155 156 VI_LOCK(dvp); 157 hd = unionfs_get_hashhead(dvp, path); 158 LIST_FOREACH(unp, hd, un_hash) { 159 if (!strcmp(unp->un_path, path)) { 160 vp = UNIONFSTOV(unp); 161 VI_LOCK_FLAGS(vp, MTX_DUPOK); 162 vp->v_iflag &= ~VI_OWEINACT; 163 if ((vp->v_iflag & (VI_DOOMED | VI_DOINGINACT)) != 0) { 164 LIST_INSERT_HEAD(hd, uncp, un_hash); 165 VI_UNLOCK(vp); 166 vp = NULLVP; 167 } else 168 VI_UNLOCK(vp); 169 VI_UNLOCK(dvp); 170 return (vp); 171 } 172 } 173 174 LIST_INSERT_HEAD(hd, uncp, un_hash); 175 VI_UNLOCK(dvp); 176 177 return (NULLVP); 178 } 179 180 /* 181 * Remove the vnode. 182 */ 183 static void 184 unionfs_rem_cached_vnode(struct unionfs_node *unp, struct vnode *dvp) 185 { 186 KASSERT((unp != NULL), ("unionfs_rem_cached_vnode: null node")); 187 KASSERT((dvp != NULLVP), 188 ("unionfs_rem_cached_vnode: null parent vnode")); 189 KASSERT((unp->un_hash.le_prev != NULL), 190 ("unionfs_rem_cached_vnode: null hash")); 191 192 VI_LOCK(dvp); 193 LIST_REMOVE(unp, un_hash); 194 unp->un_hash.le_next = NULL; 195 unp->un_hash.le_prev = NULL; 196 VI_UNLOCK(dvp); 197 } 198 199 /* 200 * Make a new or get existing unionfs node. 201 * 202 * uppervp and lowervp should be unlocked. Because if new unionfs vnode is 203 * locked, uppervp or lowervp is locked too. In order to prevent dead lock, 204 * you should not lock plurality simultaneously. 205 */ 206 int 207 unionfs_nodeget(struct mount *mp, struct vnode *uppervp, 208 struct vnode *lowervp, struct vnode *dvp, 209 struct vnode **vpp, struct componentname *cnp, 210 struct thread *td) 211 { 212 struct unionfs_mount *ump; 213 struct unionfs_node *unp; 214 struct vnode *vp; 215 int error; 216 int lkflags; 217 enum vtype vt; 218 char *path; 219 220 ump = MOUNTTOUNIONFSMOUNT(mp); 221 lkflags = (cnp ? cnp->cn_lkflags : 0); 222 path = (cnp ? cnp->cn_nameptr : NULL); 223 *vpp = NULLVP; 224 225 if (uppervp == NULLVP && lowervp == NULLVP) 226 panic("unionfs_nodeget: upper and lower is null"); 227 228 vt = (uppervp != NULLVP ? uppervp->v_type : lowervp->v_type); 229 230 /* If it has no ISLASTCN flag, path check is skipped. */ 231 if (cnp && !(cnp->cn_flags & ISLASTCN)) 232 path = NULL; 233 234 /* check the cache */ 235 if (path != NULL && dvp != NULLVP && vt == VDIR) { 236 vp = unionfs_get_cached_vnode(uppervp, lowervp, dvp, path); 237 if (vp != NULLVP) { 238 vref(vp); 239 *vpp = vp; 240 goto unionfs_nodeget_out; 241 } 242 } 243 244 if ((uppervp == NULLVP || ump->um_uppervp != uppervp) || 245 (lowervp == NULLVP || ump->um_lowervp != lowervp)) { 246 /* dvp will be NULLVP only in case of root vnode. */ 247 if (dvp == NULLVP) 248 return (EINVAL); 249 } 250 unp = malloc(sizeof(struct unionfs_node), 251 M_UNIONFSNODE, M_WAITOK | M_ZERO); 252 253 error = getnewvnode("unionfs", mp, &unionfs_vnodeops, &vp); 254 if (error != 0) { 255 free(unp, M_UNIONFSNODE); 256 return (error); 257 } 258 error = insmntque(vp, mp); /* XXX: Too early for mpsafe fs */ 259 if (error != 0) { 260 free(unp, M_UNIONFSNODE); 261 return (error); 262 } 263 if (dvp != NULLVP) 264 vref(dvp); 265 if (uppervp != NULLVP) 266 vref(uppervp); 267 if (lowervp != NULLVP) 268 vref(lowervp); 269 270 if (vt == VDIR) 271 unp->un_hashtbl = hashinit(NUNIONFSNODECACHE, M_UNIONFSHASH, 272 &(unp->un_hashmask)); 273 274 unp->un_vnode = vp; 275 unp->un_uppervp = uppervp; 276 unp->un_lowervp = lowervp; 277 unp->un_dvp = dvp; 278 if (uppervp != NULLVP) 279 vp->v_vnlock = uppervp->v_vnlock; 280 else 281 vp->v_vnlock = lowervp->v_vnlock; 282 283 if (path != NULL) { 284 unp->un_path = (char *) 285 malloc(cnp->cn_namelen +1, M_UNIONFSPATH, M_WAITOK|M_ZERO); 286 bcopy(cnp->cn_nameptr, unp->un_path, cnp->cn_namelen); 287 unp->un_path[cnp->cn_namelen] = '\0'; 288 } 289 vp->v_type = vt; 290 vp->v_data = unp; 291 292 if ((uppervp != NULLVP && ump->um_uppervp == uppervp) && 293 (lowervp != NULLVP && ump->um_lowervp == lowervp)) 294 vp->v_vflag |= VV_ROOT; 295 296 if (path != NULL && dvp != NULLVP && vt == VDIR) 297 *vpp = unionfs_ins_cached_vnode(unp, dvp, path); 298 if ((*vpp) != NULLVP) { 299 if (dvp != NULLVP) 300 vrele(dvp); 301 if (uppervp != NULLVP) 302 vrele(uppervp); 303 if (lowervp != NULLVP) 304 vrele(lowervp); 305 306 unp->un_uppervp = NULLVP; 307 unp->un_lowervp = NULLVP; 308 unp->un_dvp = NULLVP; 309 vrele(vp); 310 vp = *vpp; 311 vref(vp); 312 } else 313 *vpp = vp; 314 315 unionfs_nodeget_out: 316 if (lkflags & LK_TYPE_MASK) 317 vn_lock(vp, lkflags | LK_RETRY); 318 319 return (0); 320 } 321 322 /* 323 * Clean up the unionfs node. 324 */ 325 void 326 unionfs_noderem(struct vnode *vp, struct thread *td) 327 { 328 int count; 329 struct unionfs_node *unp, *unp_t1, *unp_t2; 330 struct unionfs_node_hashhead *hd; 331 struct unionfs_node_status *unsp, *unsp_tmp; 332 struct vnode *lvp; 333 struct vnode *uvp; 334 struct vnode *dvp; 335 336 /* 337 * Use the interlock to protect the clearing of v_data to 338 * prevent faults in unionfs_lock(). 339 */ 340 VI_LOCK(vp); 341 unp = VTOUNIONFS(vp); 342 lvp = unp->un_lowervp; 343 uvp = unp->un_uppervp; 344 dvp = unp->un_dvp; 345 unp->un_lowervp = unp->un_uppervp = NULLVP; 346 vp->v_vnlock = &(vp->v_lock); 347 vp->v_data = NULL; 348 vp->v_object = NULL; 349 VI_UNLOCK(vp); 350 351 if (lvp != NULLVP) 352 VOP_UNLOCK(lvp, LK_RELEASE); 353 if (uvp != NULLVP) 354 VOP_UNLOCK(uvp, LK_RELEASE); 355 356 if (dvp != NULLVP && unp->un_hash.le_prev != NULL) 357 unionfs_rem_cached_vnode(unp, dvp); 358 359 if (lockmgr(vp->v_vnlock, LK_EXCLUSIVE, VI_MTX(vp)) != 0) 360 panic("the lock for deletion is unacquirable."); 361 362 if (lvp != NULLVP) 363 vrele(lvp); 364 if (uvp != NULLVP) 365 vrele(uvp); 366 if (dvp != NULLVP) { 367 vrele(dvp); 368 unp->un_dvp = NULLVP; 369 } 370 if (unp->un_path != NULL) { 371 free(unp->un_path, M_UNIONFSPATH); 372 unp->un_path = NULL; 373 } 374 375 if (unp->un_hashtbl != NULL) { 376 for (count = 0; count <= unp->un_hashmask; count++) { 377 hd = unp->un_hashtbl + count; 378 LIST_FOREACH_SAFE(unp_t1, hd, un_hash, unp_t2) { 379 LIST_REMOVE(unp_t1, un_hash); 380 unp_t1->un_hash.le_next = NULL; 381 unp_t1->un_hash.le_prev = NULL; 382 } 383 } 384 hashdestroy(unp->un_hashtbl, M_UNIONFSHASH, unp->un_hashmask); 385 } 386 387 LIST_FOREACH_SAFE(unsp, &(unp->un_unshead), uns_list, unsp_tmp) { 388 LIST_REMOVE(unsp, uns_list); 389 free(unsp, M_TEMP); 390 } 391 free(unp, M_UNIONFSNODE); 392 } 393 394 /* 395 * Get the unionfs node status. 396 * You need exclusive lock this vnode. 397 */ 398 void 399 unionfs_get_node_status(struct unionfs_node *unp, struct thread *td, 400 struct unionfs_node_status **unspp) 401 { 402 struct unionfs_node_status *unsp; 403 pid_t pid = td->td_proc->p_pid; 404 405 KASSERT(NULL != unspp, ("null pointer")); 406 ASSERT_VOP_ELOCKED(UNIONFSTOV(unp), "unionfs_get_node_status"); 407 408 LIST_FOREACH(unsp, &(unp->un_unshead), uns_list) { 409 if (unsp->uns_pid == pid) { 410 *unspp = unsp; 411 return; 412 } 413 } 414 415 /* create a new unionfs node status */ 416 unsp = malloc(sizeof(struct unionfs_node_status), 417 M_TEMP, M_WAITOK | M_ZERO); 418 419 unsp->uns_pid = pid; 420 LIST_INSERT_HEAD(&(unp->un_unshead), unsp, uns_list); 421 422 *unspp = unsp; 423 } 424 425 /* 426 * Remove the unionfs node status, if you can. 427 * You need exclusive lock this vnode. 428 */ 429 void 430 unionfs_tryrem_node_status(struct unionfs_node *unp, 431 struct unionfs_node_status *unsp) 432 { 433 KASSERT(NULL != unsp, ("null pointer")); 434 ASSERT_VOP_ELOCKED(UNIONFSTOV(unp), "unionfs_get_node_status"); 435 436 if (0 < unsp->uns_lower_opencnt || 0 < unsp->uns_upper_opencnt) 437 return; 438 439 LIST_REMOVE(unsp, uns_list); 440 free(unsp, M_TEMP); 441 } 442 443 /* 444 * Create upper node attr. 445 */ 446 void 447 unionfs_create_uppervattr_core(struct unionfs_mount *ump, 448 struct vattr *lva, 449 struct vattr *uva, 450 struct thread *td) 451 { 452 VATTR_NULL(uva); 453 uva->va_type = lva->va_type; 454 uva->va_atime = lva->va_atime; 455 uva->va_mtime = lva->va_mtime; 456 uva->va_ctime = lva->va_ctime; 457 458 switch (ump->um_copymode) { 459 case UNIONFS_TRANSPARENT: 460 uva->va_mode = lva->va_mode; 461 uva->va_uid = lva->va_uid; 462 uva->va_gid = lva->va_gid; 463 break; 464 case UNIONFS_MASQUERADE: 465 if (ump->um_uid == lva->va_uid) { 466 uva->va_mode = lva->va_mode & 077077; 467 uva->va_mode |= (lva->va_type == VDIR ? ump->um_udir : ump->um_ufile) & 0700; 468 uva->va_uid = lva->va_uid; 469 uva->va_gid = lva->va_gid; 470 } else { 471 uva->va_mode = (lva->va_type == VDIR ? ump->um_udir : ump->um_ufile); 472 uva->va_uid = ump->um_uid; 473 uva->va_gid = ump->um_gid; 474 } 475 break; 476 default: /* UNIONFS_TRADITIONAL */ 477 uva->va_mode = 0777 & ~td->td_proc->p_fd->fd_cmask; 478 uva->va_uid = ump->um_uid; 479 uva->va_gid = ump->um_gid; 480 break; 481 } 482 } 483 484 /* 485 * Create upper node attr. 486 */ 487 int 488 unionfs_create_uppervattr(struct unionfs_mount *ump, 489 struct vnode *lvp, 490 struct vattr *uva, 491 struct ucred *cred, 492 struct thread *td) 493 { 494 int error; 495 struct vattr lva; 496 497 if ((error = VOP_GETATTR(lvp, &lva, cred))) 498 return (error); 499 500 unionfs_create_uppervattr_core(ump, &lva, uva, td); 501 502 return (error); 503 } 504 505 /* 506 * relookup 507 * 508 * dvp should be locked on entry and will be locked on return. 509 * 510 * If an error is returned, *vpp will be invalid, otherwise it will hold a 511 * locked, referenced vnode. If *vpp == dvp then remember that only one 512 * LK_EXCLUSIVE lock is held. 513 */ 514 int 515 unionfs_relookup(struct vnode *dvp, struct vnode **vpp, 516 struct componentname *cnp, struct componentname *cn, 517 struct thread *td, char *path, int pathlen, u_long nameiop) 518 { 519 int error; 520 521 cn->cn_namelen = pathlen; 522 cn->cn_pnbuf = uma_zalloc(namei_zone, M_WAITOK); 523 bcopy(path, cn->cn_pnbuf, pathlen); 524 cn->cn_pnbuf[pathlen] = '\0'; 525 526 cn->cn_nameiop = nameiop; 527 cn->cn_flags = (LOCKPARENT | LOCKLEAF | HASBUF | SAVENAME | ISLASTCN); 528 cn->cn_lkflags = LK_EXCLUSIVE; 529 cn->cn_thread = td; 530 cn->cn_cred = cnp->cn_cred; 531 532 cn->cn_nameptr = cn->cn_pnbuf; 533 cn->cn_consume = cnp->cn_consume; 534 535 if (nameiop == DELETE) 536 cn->cn_flags |= (cnp->cn_flags & (DOWHITEOUT | SAVESTART)); 537 else if (RENAME == nameiop) 538 cn->cn_flags |= (cnp->cn_flags & SAVESTART); 539 540 vref(dvp); 541 VOP_UNLOCK(dvp, LK_RELEASE); 542 543 if ((error = relookup(dvp, vpp, cn))) { 544 uma_zfree(namei_zone, cn->cn_pnbuf); 545 cn->cn_flags &= ~HASBUF; 546 vn_lock(dvp, LK_EXCLUSIVE | LK_RETRY); 547 } else 548 vrele(dvp); 549 550 return (error); 551 } 552 553 /* 554 * relookup for CREATE namei operation. 555 * 556 * dvp is unionfs vnode. dvp should be locked. 557 * 558 * If it called 'unionfs_copyfile' function by unionfs_link etc, 559 * VOP_LOOKUP information is broken. 560 * So it need relookup in order to create link etc. 561 */ 562 int 563 unionfs_relookup_for_create(struct vnode *dvp, struct componentname *cnp, 564 struct thread *td) 565 { 566 int error; 567 struct vnode *udvp; 568 struct vnode *vp; 569 struct componentname cn; 570 571 udvp = UNIONFSVPTOUPPERVP(dvp); 572 vp = NULLVP; 573 574 error = unionfs_relookup(udvp, &vp, cnp, &cn, td, cnp->cn_nameptr, 575 strlen(cnp->cn_nameptr), CREATE); 576 if (error) 577 return (error); 578 579 if (vp != NULLVP) { 580 if (udvp == vp) 581 vrele(vp); 582 else 583 vput(vp); 584 585 error = EEXIST; 586 } 587 588 if (cn.cn_flags & HASBUF) { 589 uma_zfree(namei_zone, cn.cn_pnbuf); 590 cn.cn_flags &= ~HASBUF; 591 } 592 593 if (!error) { 594 cn.cn_flags |= (cnp->cn_flags & HASBUF); 595 cnp->cn_flags = cn.cn_flags; 596 } 597 598 return (error); 599 } 600 601 /* 602 * relookup for DELETE namei operation. 603 * 604 * dvp is unionfs vnode. dvp should be locked. 605 */ 606 int 607 unionfs_relookup_for_delete(struct vnode *dvp, struct componentname *cnp, 608 struct thread *td) 609 { 610 int error; 611 struct vnode *udvp; 612 struct vnode *vp; 613 struct componentname cn; 614 615 udvp = UNIONFSVPTOUPPERVP(dvp); 616 vp = NULLVP; 617 618 error = unionfs_relookup(udvp, &vp, cnp, &cn, td, cnp->cn_nameptr, 619 strlen(cnp->cn_nameptr), DELETE); 620 if (error) 621 return (error); 622 623 if (vp == NULLVP) 624 error = ENOENT; 625 else { 626 if (udvp == vp) 627 vrele(vp); 628 else 629 vput(vp); 630 } 631 632 if (cn.cn_flags & HASBUF) { 633 uma_zfree(namei_zone, cn.cn_pnbuf); 634 cn.cn_flags &= ~HASBUF; 635 } 636 637 if (!error) { 638 cn.cn_flags |= (cnp->cn_flags & HASBUF); 639 cnp->cn_flags = cn.cn_flags; 640 } 641 642 return (error); 643 } 644 645 /* 646 * relookup for RENAME namei operation. 647 * 648 * dvp is unionfs vnode. dvp should be locked. 649 */ 650 int 651 unionfs_relookup_for_rename(struct vnode *dvp, struct componentname *cnp, 652 struct thread *td) 653 { 654 int error; 655 struct vnode *udvp; 656 struct vnode *vp; 657 struct componentname cn; 658 659 udvp = UNIONFSVPTOUPPERVP(dvp); 660 vp = NULLVP; 661 662 error = unionfs_relookup(udvp, &vp, cnp, &cn, td, cnp->cn_nameptr, 663 strlen(cnp->cn_nameptr), RENAME); 664 if (error) 665 return (error); 666 667 if (vp != NULLVP) { 668 if (udvp == vp) 669 vrele(vp); 670 else 671 vput(vp); 672 } 673 674 if (cn.cn_flags & HASBUF) { 675 uma_zfree(namei_zone, cn.cn_pnbuf); 676 cn.cn_flags &= ~HASBUF; 677 } 678 679 if (!error) { 680 cn.cn_flags |= (cnp->cn_flags & HASBUF); 681 cnp->cn_flags = cn.cn_flags; 682 } 683 684 return (error); 685 686 } 687 688 /* 689 * Update the unionfs_node. 690 * 691 * uvp is new locked upper vnode. unionfs vnode's lock will be exchanged to the 692 * uvp's lock and lower's lock will be unlocked. 693 */ 694 static void 695 unionfs_node_update(struct unionfs_node *unp, struct vnode *uvp, 696 struct thread *td) 697 { 698 unsigned count, lockrec; 699 struct vnode *vp; 700 struct vnode *lvp; 701 struct vnode *dvp; 702 703 vp = UNIONFSTOV(unp); 704 lvp = unp->un_lowervp; 705 ASSERT_VOP_ELOCKED(lvp, "unionfs_node_update"); 706 dvp = unp->un_dvp; 707 708 /* 709 * lock update 710 */ 711 VI_LOCK(vp); 712 unp->un_uppervp = uvp; 713 vp->v_vnlock = uvp->v_vnlock; 714 VI_UNLOCK(vp); 715 lockrec = lvp->v_vnlock->lk_recurse; 716 for (count = 0; count < lockrec; count++) 717 vn_lock(uvp, LK_EXCLUSIVE | LK_CANRECURSE | LK_RETRY); 718 719 /* 720 * cache update 721 */ 722 if (unp->un_path != NULL && dvp != NULLVP && vp->v_type == VDIR) { 723 static struct unionfs_node_hashhead *hd; 724 725 VI_LOCK(dvp); 726 hd = unionfs_get_hashhead(dvp, unp->un_path); 727 LIST_REMOVE(unp, un_hash); 728 LIST_INSERT_HEAD(hd, unp, un_hash); 729 VI_UNLOCK(dvp); 730 } 731 } 732 733 /* 734 * Create a new shadow dir. 735 * 736 * udvp should be locked on entry and will be locked on return. 737 * 738 * If no error returned, unp will be updated. 739 */ 740 int 741 unionfs_mkshadowdir(struct unionfs_mount *ump, struct vnode *udvp, 742 struct unionfs_node *unp, struct componentname *cnp, 743 struct thread *td) 744 { 745 int error; 746 struct vnode *lvp; 747 struct vnode *uvp; 748 struct vattr va; 749 struct vattr lva; 750 struct componentname cn; 751 struct mount *mp; 752 struct ucred *cred; 753 struct ucred *credbk; 754 struct uidinfo *rootinfo; 755 756 if (unp->un_uppervp != NULLVP) 757 return (EEXIST); 758 759 lvp = unp->un_lowervp; 760 uvp = NULLVP; 761 credbk = cnp->cn_cred; 762 763 /* Authority change to root */ 764 rootinfo = uifind((uid_t)0); 765 cred = crdup(cnp->cn_cred); 766 /* 767 * The calls to chgproccnt() are needed to compensate for change_ruid() 768 * calling chgproccnt(). 769 */ 770 chgproccnt(cred->cr_ruidinfo, 1, 0); 771 change_euid(cred, rootinfo); 772 change_ruid(cred, rootinfo); 773 change_svuid(cred, (uid_t)0); 774 uifree(rootinfo); 775 cnp->cn_cred = cred; 776 777 memset(&cn, 0, sizeof(cn)); 778 779 if ((error = VOP_GETATTR(lvp, &lva, cnp->cn_cred))) 780 goto unionfs_mkshadowdir_abort; 781 782 if ((error = unionfs_relookup(udvp, &uvp, cnp, &cn, td, cnp->cn_nameptr, cnp->cn_namelen, CREATE))) 783 goto unionfs_mkshadowdir_abort; 784 if (uvp != NULLVP) { 785 if (udvp == uvp) 786 vrele(uvp); 787 else 788 vput(uvp); 789 790 error = EEXIST; 791 goto unionfs_mkshadowdir_free_out; 792 } 793 794 if ((error = vn_start_write(udvp, &mp, V_WAIT | PCATCH))) 795 goto unionfs_mkshadowdir_free_out; 796 unionfs_create_uppervattr_core(ump, &lva, &va, td); 797 798 error = VOP_MKDIR(udvp, &uvp, &cn, &va); 799 800 if (!error) { 801 unionfs_node_update(unp, uvp, td); 802 803 /* 804 * XXX The bug which cannot set uid/gid was corrected. 805 * Ignore errors. 806 */ 807 va.va_type = VNON; 808 VOP_SETATTR(uvp, &va, cn.cn_cred); 809 } 810 vn_finished_write(mp); 811 812 unionfs_mkshadowdir_free_out: 813 if (cn.cn_flags & HASBUF) { 814 uma_zfree(namei_zone, cn.cn_pnbuf); 815 cn.cn_flags &= ~HASBUF; 816 } 817 818 unionfs_mkshadowdir_abort: 819 cnp->cn_cred = credbk; 820 chgproccnt(cred->cr_ruidinfo, -1, 0); 821 crfree(cred); 822 823 return (error); 824 } 825 826 /* 827 * Create a new whiteout. 828 * 829 * dvp should be locked on entry and will be locked on return. 830 */ 831 int 832 unionfs_mkwhiteout(struct vnode *dvp, struct componentname *cnp, 833 struct thread *td, char *path) 834 { 835 int error; 836 struct vnode *wvp; 837 struct componentname cn; 838 struct mount *mp; 839 840 if (path == NULL) 841 path = cnp->cn_nameptr; 842 843 wvp = NULLVP; 844 if ((error = unionfs_relookup(dvp, &wvp, cnp, &cn, td, path, strlen(path), CREATE))) 845 return (error); 846 if (wvp != NULLVP) { 847 if (cn.cn_flags & HASBUF) { 848 uma_zfree(namei_zone, cn.cn_pnbuf); 849 cn.cn_flags &= ~HASBUF; 850 } 851 if (dvp == wvp) 852 vrele(wvp); 853 else 854 vput(wvp); 855 856 return (EEXIST); 857 } 858 859 if ((error = vn_start_write(dvp, &mp, V_WAIT | PCATCH))) 860 goto unionfs_mkwhiteout_free_out; 861 error = VOP_WHITEOUT(dvp, &cn, CREATE); 862 863 vn_finished_write(mp); 864 865 unionfs_mkwhiteout_free_out: 866 if (cn.cn_flags & HASBUF) { 867 uma_zfree(namei_zone, cn.cn_pnbuf); 868 cn.cn_flags &= ~HASBUF; 869 } 870 871 return (error); 872 } 873 874 /* 875 * Create a new vnode for create a new shadow file. 876 * 877 * If an error is returned, *vpp will be invalid, otherwise it will hold a 878 * locked, referenced and opened vnode. 879 * 880 * unp is never updated. 881 */ 882 static int 883 unionfs_vn_create_on_upper(struct vnode **vpp, struct vnode *udvp, 884 struct unionfs_node *unp, struct vattr *uvap, 885 struct thread *td) 886 { 887 struct unionfs_mount *ump; 888 struct vnode *vp; 889 struct vnode *lvp; 890 struct ucred *cred; 891 struct vattr lva; 892 int fmode; 893 int error; 894 struct componentname cn; 895 896 ump = MOUNTTOUNIONFSMOUNT(UNIONFSTOV(unp)->v_mount); 897 vp = NULLVP; 898 lvp = unp->un_lowervp; 899 cred = td->td_ucred; 900 fmode = FFLAGS(O_WRONLY | O_CREAT | O_TRUNC | O_EXCL); 901 error = 0; 902 903 if ((error = VOP_GETATTR(lvp, &lva, cred)) != 0) 904 return (error); 905 unionfs_create_uppervattr_core(ump, &lva, uvap, td); 906 907 if (unp->un_path == NULL) 908 panic("unionfs: un_path is null"); 909 910 cn.cn_namelen = strlen(unp->un_path); 911 cn.cn_pnbuf = uma_zalloc(namei_zone, M_WAITOK); 912 bcopy(unp->un_path, cn.cn_pnbuf, cn.cn_namelen + 1); 913 cn.cn_nameiop = CREATE; 914 cn.cn_flags = (LOCKPARENT | LOCKLEAF | HASBUF | SAVENAME | ISLASTCN); 915 cn.cn_lkflags = LK_EXCLUSIVE; 916 cn.cn_thread = td; 917 cn.cn_cred = cred; 918 cn.cn_nameptr = cn.cn_pnbuf; 919 cn.cn_consume = 0; 920 921 vref(udvp); 922 if ((error = relookup(udvp, &vp, &cn)) != 0) 923 goto unionfs_vn_create_on_upper_free_out2; 924 vrele(udvp); 925 926 if (vp != NULLVP) { 927 if (vp == udvp) 928 vrele(vp); 929 else 930 vput(vp); 931 error = EEXIST; 932 goto unionfs_vn_create_on_upper_free_out1; 933 } 934 935 if ((error = VOP_CREATE(udvp, &vp, &cn, uvap)) != 0) 936 goto unionfs_vn_create_on_upper_free_out1; 937 938 if ((error = VOP_OPEN(vp, fmode, cred, td, NULL)) != 0) { 939 vput(vp); 940 goto unionfs_vn_create_on_upper_free_out1; 941 } 942 VOP_ADD_WRITECOUNT(vp, 1); 943 CTR3(KTR_VFS, "%s: vp %p v_writecount increased to %d", __func__, vp, 944 vp->v_writecount); 945 *vpp = vp; 946 947 unionfs_vn_create_on_upper_free_out1: 948 VOP_UNLOCK(udvp, LK_RELEASE); 949 950 unionfs_vn_create_on_upper_free_out2: 951 if (cn.cn_flags & HASBUF) { 952 uma_zfree(namei_zone, cn.cn_pnbuf); 953 cn.cn_flags &= ~HASBUF; 954 } 955 956 return (error); 957 } 958 959 /* 960 * Copy from lvp to uvp. 961 * 962 * lvp and uvp should be locked and opened on entry and will be locked and 963 * opened on return. 964 */ 965 static int 966 unionfs_copyfile_core(struct vnode *lvp, struct vnode *uvp, 967 struct ucred *cred, struct thread *td) 968 { 969 int error; 970 off_t offset; 971 int count; 972 int bufoffset; 973 char *buf; 974 struct uio uio; 975 struct iovec iov; 976 977 error = 0; 978 memset(&uio, 0, sizeof(uio)); 979 980 uio.uio_td = td; 981 uio.uio_segflg = UIO_SYSSPACE; 982 uio.uio_offset = 0; 983 984 buf = malloc(MAXBSIZE, M_TEMP, M_WAITOK); 985 986 while (error == 0) { 987 offset = uio.uio_offset; 988 989 uio.uio_iov = &iov; 990 uio.uio_iovcnt = 1; 991 iov.iov_base = buf; 992 iov.iov_len = MAXBSIZE; 993 uio.uio_resid = iov.iov_len; 994 uio.uio_rw = UIO_READ; 995 996 if ((error = VOP_READ(lvp, &uio, 0, cred)) != 0) 997 break; 998 if ((count = MAXBSIZE - uio.uio_resid) == 0) 999 break; 1000 1001 bufoffset = 0; 1002 while (bufoffset < count) { 1003 uio.uio_iov = &iov; 1004 uio.uio_iovcnt = 1; 1005 iov.iov_base = buf + bufoffset; 1006 iov.iov_len = count - bufoffset; 1007 uio.uio_offset = offset + bufoffset; 1008 uio.uio_resid = iov.iov_len; 1009 uio.uio_rw = UIO_WRITE; 1010 1011 if ((error = VOP_WRITE(uvp, &uio, 0, cred)) != 0) 1012 break; 1013 1014 bufoffset += (count - bufoffset) - uio.uio_resid; 1015 } 1016 1017 uio.uio_offset = offset + bufoffset; 1018 } 1019 1020 free(buf, M_TEMP); 1021 1022 return (error); 1023 } 1024 1025 /* 1026 * Copy file from lower to upper. 1027 * 1028 * If you need copy of the contents, set 1 to docopy. Otherwise, set 0 to 1029 * docopy. 1030 * 1031 * If no error returned, unp will be updated. 1032 */ 1033 int 1034 unionfs_copyfile(struct unionfs_node *unp, int docopy, struct ucred *cred, 1035 struct thread *td) 1036 { 1037 int error; 1038 struct mount *mp; 1039 struct vnode *udvp; 1040 struct vnode *lvp; 1041 struct vnode *uvp; 1042 struct vattr uva; 1043 1044 lvp = unp->un_lowervp; 1045 uvp = NULLVP; 1046 1047 if ((UNIONFSTOV(unp)->v_mount->mnt_flag & MNT_RDONLY)) 1048 return (EROFS); 1049 if (unp->un_dvp == NULLVP) 1050 return (EINVAL); 1051 if (unp->un_uppervp != NULLVP) 1052 return (EEXIST); 1053 udvp = VTOUNIONFS(unp->un_dvp)->un_uppervp; 1054 if (udvp == NULLVP) 1055 return (EROFS); 1056 if ((udvp->v_mount->mnt_flag & MNT_RDONLY)) 1057 return (EROFS); 1058 1059 error = VOP_ACCESS(lvp, VREAD, cred, td); 1060 if (error != 0) 1061 return (error); 1062 1063 if ((error = vn_start_write(udvp, &mp, V_WAIT | PCATCH)) != 0) 1064 return (error); 1065 error = unionfs_vn_create_on_upper(&uvp, udvp, unp, &uva, td); 1066 if (error != 0) { 1067 vn_finished_write(mp); 1068 return (error); 1069 } 1070 1071 if (docopy != 0) { 1072 error = VOP_OPEN(lvp, FREAD, cred, td, NULL); 1073 if (error == 0) { 1074 error = unionfs_copyfile_core(lvp, uvp, cred, td); 1075 VOP_CLOSE(lvp, FREAD, cred, td); 1076 } 1077 } 1078 VOP_CLOSE(uvp, FWRITE, cred, td); 1079 VOP_ADD_WRITECOUNT(uvp, -1); 1080 CTR3(KTR_VFS, "%s: vp %p v_writecount decreased to %d", __func__, uvp, 1081 uvp->v_writecount); 1082 1083 vn_finished_write(mp); 1084 1085 if (error == 0) { 1086 /* Reset the attributes. Ignore errors. */ 1087 uva.va_type = VNON; 1088 VOP_SETATTR(uvp, &uva, cred); 1089 } 1090 1091 unionfs_node_update(unp, uvp, td); 1092 1093 return (error); 1094 } 1095 1096 /* 1097 * It checks whether vp can rmdir. (check empty) 1098 * 1099 * vp is unionfs vnode. 1100 * vp should be locked. 1101 */ 1102 int 1103 unionfs_check_rmdir(struct vnode *vp, struct ucred *cred, struct thread *td) 1104 { 1105 int error; 1106 int eofflag; 1107 int lookuperr; 1108 struct vnode *uvp; 1109 struct vnode *lvp; 1110 struct vnode *tvp; 1111 struct vattr va; 1112 struct componentname cn; 1113 /* 1114 * The size of buf needs to be larger than DIRBLKSIZ. 1115 */ 1116 char buf[256 * 6]; 1117 struct dirent *dp; 1118 struct dirent *edp; 1119 struct uio uio; 1120 struct iovec iov; 1121 1122 ASSERT_VOP_ELOCKED(vp, "unionfs_check_rmdir"); 1123 1124 eofflag = 0; 1125 uvp = UNIONFSVPTOUPPERVP(vp); 1126 lvp = UNIONFSVPTOLOWERVP(vp); 1127 1128 /* check opaque */ 1129 if ((error = VOP_GETATTR(uvp, &va, cred)) != 0) 1130 return (error); 1131 if (va.va_flags & OPAQUE) 1132 return (0); 1133 1134 /* open vnode */ 1135 #ifdef MAC 1136 if ((error = mac_vnode_check_open(cred, vp, VEXEC|VREAD)) != 0) 1137 return (error); 1138 #endif 1139 if ((error = VOP_ACCESS(vp, VEXEC|VREAD, cred, td)) != 0) 1140 return (error); 1141 if ((error = VOP_OPEN(vp, FREAD, cred, td, NULL)) != 0) 1142 return (error); 1143 1144 uio.uio_rw = UIO_READ; 1145 uio.uio_segflg = UIO_SYSSPACE; 1146 uio.uio_td = td; 1147 uio.uio_offset = 0; 1148 1149 #ifdef MAC 1150 error = mac_vnode_check_readdir(td->td_ucred, lvp); 1151 #endif 1152 while (!error && !eofflag) { 1153 iov.iov_base = buf; 1154 iov.iov_len = sizeof(buf); 1155 uio.uio_iov = &iov; 1156 uio.uio_iovcnt = 1; 1157 uio.uio_resid = iov.iov_len; 1158 1159 error = VOP_READDIR(lvp, &uio, cred, &eofflag, NULL, NULL); 1160 if (error != 0) 1161 break; 1162 if (eofflag == 0 && uio.uio_resid == sizeof(buf)) { 1163 #ifdef DIAGNOSTIC 1164 panic("bad readdir response from lower FS."); 1165 #endif 1166 break; 1167 } 1168 1169 edp = (struct dirent*)&buf[sizeof(buf) - uio.uio_resid]; 1170 for (dp = (struct dirent*)buf; !error && dp < edp; 1171 dp = (struct dirent*)((caddr_t)dp + dp->d_reclen)) { 1172 if (dp->d_type == DT_WHT || dp->d_fileno == 0 || 1173 (dp->d_namlen == 1 && dp->d_name[0] == '.') || 1174 (dp->d_namlen == 2 && !bcmp(dp->d_name, "..", 2))) 1175 continue; 1176 1177 cn.cn_namelen = dp->d_namlen; 1178 cn.cn_pnbuf = NULL; 1179 cn.cn_nameptr = dp->d_name; 1180 cn.cn_nameiop = LOOKUP; 1181 cn.cn_flags = (LOCKPARENT | LOCKLEAF | SAVENAME | RDONLY | ISLASTCN); 1182 cn.cn_lkflags = LK_EXCLUSIVE; 1183 cn.cn_thread = td; 1184 cn.cn_cred = cred; 1185 cn.cn_consume = 0; 1186 1187 /* 1188 * check entry in lower. 1189 * Sometimes, readdir function returns 1190 * wrong entry. 1191 */ 1192 lookuperr = VOP_LOOKUP(lvp, &tvp, &cn); 1193 1194 if (!lookuperr) 1195 vput(tvp); 1196 else 1197 continue; /* skip entry */ 1198 1199 /* 1200 * check entry 1201 * If it has no exist/whiteout entry in upper, 1202 * directory is not empty. 1203 */ 1204 cn.cn_flags = (LOCKPARENT | LOCKLEAF | SAVENAME | RDONLY | ISLASTCN); 1205 lookuperr = VOP_LOOKUP(uvp, &tvp, &cn); 1206 1207 if (!lookuperr) 1208 vput(tvp); 1209 1210 /* ignore exist or whiteout entry */ 1211 if (!lookuperr || 1212 (lookuperr == ENOENT && (cn.cn_flags & ISWHITEOUT))) 1213 continue; 1214 1215 error = ENOTEMPTY; 1216 } 1217 } 1218 1219 /* close vnode */ 1220 VOP_CLOSE(vp, FREAD, cred, td); 1221 1222 return (error); 1223 } 1224 1225 #ifdef DIAGNOSTIC 1226 1227 struct vnode * 1228 unionfs_checkuppervp(struct vnode *vp, char *fil, int lno) 1229 { 1230 struct unionfs_node *unp; 1231 1232 unp = VTOUNIONFS(vp); 1233 1234 #ifdef notyet 1235 if (vp->v_op != unionfs_vnodeop_p) { 1236 printf("unionfs_checkuppervp: on non-unionfs-node.\n"); 1237 #ifdef KDB 1238 kdb_enter(KDB_WHY_UNIONFS, 1239 "unionfs_checkuppervp: on non-unionfs-node.\n"); 1240 #endif 1241 panic("unionfs_checkuppervp"); 1242 }; 1243 #endif 1244 return (unp->un_uppervp); 1245 } 1246 1247 struct vnode * 1248 unionfs_checklowervp(struct vnode *vp, char *fil, int lno) 1249 { 1250 struct unionfs_node *unp; 1251 1252 unp = VTOUNIONFS(vp); 1253 1254 #ifdef notyet 1255 if (vp->v_op != unionfs_vnodeop_p) { 1256 printf("unionfs_checklowervp: on non-unionfs-node.\n"); 1257 #ifdef KDB 1258 kdb_enter(KDB_WHY_UNIONFS, 1259 "unionfs_checklowervp: on non-unionfs-node.\n"); 1260 #endif 1261 panic("unionfs_checklowervp"); 1262 }; 1263 #endif 1264 return (unp->un_lowervp); 1265 } 1266 #endif 1267