#include <sys/param.h>
#include <sys/systm.h>
#include <sys/proc_internal.h>
#include <sys/kauth.h>
#include <sys/time.h>
#include <sys/kernel.h>
#include <sys/vnode_internal.h>
#include <sys/namei.h>
#include <sys/malloc.h>
#include <sys/file_internal.h>
#include <sys/filedesc.h>
#include <sys/queue.h>
#include <sys/mount_internal.h>
#include <sys/stat.h>
#include <sys/ubc.h>
#include <sys/uio_internal.h>
#include <miscfs/union/union.h>
#include <sys/lock.h>
#include <sys/kdebug.h>
#if CONFIG_MACF
#include <security/mac_framework.h>
#endif
static int union_vn_close(struct vnode *vp, int fmode, vfs_context_t ctx);
#define NHASH 32
#define UNION_HASH(u, l) \
(((((unsigned long) (u)) + ((unsigned long) l)) >> 8) & (NHASH-1))
static LIST_HEAD(unhead, union_node) unhead[NHASH];
static int unvplock[NHASH];
static lck_grp_t * union_lck_grp;
static lck_grp_attr_t * union_lck_grp_attr;
static lck_attr_t * union_lck_attr;
static lck_mtx_t * union_mtxp;
static int union_dircheck(struct vnode **, struct fileproc *, vfs_context_t ctx);
static void union_newlower(struct union_node *, struct vnode *);
static void union_newupper(struct union_node *, struct vnode *);
int
union_init(__unused struct vfsconf *vfsp)
{
int i;
union_lck_grp_attr= lck_grp_attr_alloc_init();
#if DIAGNOSTIC
lck_grp_attr_setstat(union_lck_grp_attr);
#endif
union_lck_grp = lck_grp_alloc_init("union", union_lck_grp_attr);
union_lck_attr = lck_attr_alloc_init();
#if DIAGNOSTIC
lck_attr_setdebug(union_lck_attr);
#endif
union_mtxp = lck_mtx_alloc_init(union_lck_grp, union_lck_attr);
for (i = 0; i < NHASH; i++)
LIST_INIT(&unhead[i]);
bzero((caddr_t) unvplock, sizeof(unvplock));
union_dircheckp = union_dircheck;
return (0);
}
void
union_lock()
{
lck_mtx_lock(union_mtxp);
}
void
union_unlock()
{
lck_mtx_unlock(union_mtxp);
}
static int
union_list_lock(int ix)
{
if (unvplock[ix] & UNVP_LOCKED) {
unvplock[ix] |= UNVP_WANT;
msleep((caddr_t) &unvplock[ix], union_mtxp, PINOD, "union_list_lock", NULL);
return (1);
}
unvplock[ix] |= UNVP_LOCKED;
return (0);
}
static void
union_list_unlock(int ix)
{
unvplock[ix] &= ~UNVP_LOCKED;
if (unvplock[ix] & UNVP_WANT) {
unvplock[ix] &= ~UNVP_WANT;
wakeup((caddr_t) &unvplock[ix]);
}
}
void
union_updatevp(struct union_node *un, struct vnode *uppervp,
struct vnode *lowervp)
{
int ohash = UNION_HASH(un->un_uppervp, un->un_lowervp);
int nhash = UNION_HASH(uppervp, lowervp);
int docache = (lowervp != NULLVP || uppervp != NULLVP);
int lhash, uhash;
vnode_t freevp;
vnode_t freedirvp;
caddr_t freepath;
if (nhash < ohash) {
lhash = nhash;
uhash = ohash;
} else {
lhash = ohash;
uhash = nhash;
}
if (lhash != uhash) {
while (union_list_lock(lhash))
continue;
}
while (union_list_lock(uhash))
continue;
if (ohash != nhash || !docache) {
if (un->un_flags & UN_CACHED) {
un->un_flags &= ~UN_CACHED;
LIST_REMOVE(un, un_cache);
}
}
if (ohash != nhash)
union_list_unlock(ohash);
if (un->un_lowervp != lowervp) {
freevp = freedirvp = NULLVP;
freepath = (caddr_t)0;
if (un->un_lowervp) {
freevp = un->un_lowervp;
un->un_lowervp = lowervp;
if (un->un_path) {
freepath = un->un_path;
un->un_path = 0;
}
if (un->un_dirvp) {
freedirvp = un->un_dirvp;
un->un_dirvp = NULLVP;
}
union_unlock();
if (freevp)
vnode_put(freevp);
if (freedirvp)
vnode_put(freedirvp);
if (freepath)
_FREE(un->un_path, M_TEMP);
union_lock();
} else
un->un_lowervp = lowervp;
if (lowervp != NULLVP)
un->un_lowervid = vnode_vid(lowervp);
un->un_lowersz = VNOVAL;
}
if (un->un_uppervp != uppervp) {
freevp = NULLVP;
if (un->un_uppervp) {
freevp = un->un_uppervp;
}
un->un_uppervp = uppervp;
if (uppervp != NULLVP)
un->un_uppervid = vnode_vid(uppervp);
un->un_uppersz = VNOVAL;
union_unlock();
if (freevp)
vnode_put(freevp);
union_lock();
}
if (docache && (ohash != nhash)) {
LIST_INSERT_HEAD(&unhead[nhash], un, un_cache);
un->un_flags |= UN_CACHED;
}
union_list_unlock(nhash);
}
static void
union_newlower(un, lowervp)
struct union_node *un;
struct vnode *lowervp;
{
union_updatevp(un, un->un_uppervp, lowervp);
}
static void
union_newupper(un, uppervp)
struct union_node *un;
struct vnode *uppervp;
{
union_updatevp(un, uppervp, un->un_lowervp);
}
void
union_newsize(vp, uppersz, lowersz)
struct vnode *vp;
off_t uppersz, lowersz;
{
struct union_node *un;
off_t sz;
if (vp->v_type != VREG)
return;
un = VTOUNION(vp);
sz = VNOVAL;
if ((uppersz != VNOVAL) && (un->un_uppersz != uppersz)) {
un->un_uppersz = uppersz;
if (sz == VNOVAL)
sz = un->un_uppersz;
}
if ((lowersz != VNOVAL) && (un->un_lowersz != lowersz)) {
un->un_lowersz = lowersz;
if (sz == VNOVAL)
sz = un->un_lowersz;
}
if (sz != VNOVAL) {
#ifdef UNION_DIAGNOSTIC
printf("union: %s size now %ld\n",
uppersz != VNOVAL ? "upper" : "lower", (long) sz);
#endif
union_unlock();
ubc_setsize(vp, sz);
union_lock();
}
}
int
union_allocvp(struct vnode **vpp,
struct mount *mp,
struct vnode *undvp,
struct vnode *dvp,
struct componentname *cnp,
struct vnode *uppervp,
struct vnode *lowervp,
int docache)
{
int error;
struct union_node *un = NULL;
struct union_node *unp;
struct vnode *xlowervp = NULLVP;
struct union_mount *um = MOUNTTOUNIONMOUNT(mp);
int hash = 0;
int markroot;
int try;
struct vnode_fsparam vfsp;
enum vtype vtype;
if (uppervp == NULLVP && lowervp == NULLVP)
panic("union: unidentifiable allocation");
if (uppervp && lowervp && (uppervp->v_type != lowervp->v_type)) {
xlowervp = lowervp;
lowervp = NULLVP;
}
markroot = 0;
if ((uppervp == um->um_uppervp) &&
((lowervp == NULLVP) || lowervp == um->um_lowervp)) {
if (lowervp == NULLVP) {
lowervp = um->um_lowervp;
if (lowervp != NULLVP) {
union_unlock();
vnode_get(lowervp);
union_lock();
}
}
markroot = VROOT;
}
loop:
if (!docache) {
un = NULL;
} else for (try = 0; try < 3; try++) {
switch (try) {
case 0:
if (lowervp == NULLVP)
continue;
hash = UNION_HASH(uppervp, lowervp);
break;
case 1:
if (uppervp == NULLVP)
continue;
hash = UNION_HASH(uppervp, NULLVP);
break;
case 2:
if (lowervp == NULLVP)
continue;
hash = UNION_HASH(NULLVP, lowervp);
break;
}
while (union_list_lock(hash))
continue;
for (un = unhead[hash].lh_first; un != 0;
un = un->un_cache.le_next) {
if ((un->un_lowervp == lowervp ||
un->un_lowervp == NULLVP) &&
(un->un_uppervp == uppervp ||
un->un_uppervp == NULLVP) &&
(un->un_mount == mp)) {
break;
}
}
union_list_unlock(hash);
if (un)
break;
}
if (un) {
if (un->un_flags & UN_LOCKED) {
un->un_flags |= UN_WANT;
msleep((caddr_t) &un->un_flags, union_mtxp, PINOD, "union node locked", 0);
goto loop;
}
un->un_flags |= UN_LOCKED;
union_unlock();
if (UNIONTOV(un) == NULLVP)
panic("null vnode in union node\n");
if (vnode_get(UNIONTOV(un))) {
union_lock();
un->un_flags &= ~UN_LOCKED;
if ((un->un_flags & UN_WANT) == UN_WANT) {
un->un_flags &= ~UN_LOCKED;
wakeup(&un->un_flags);
}
goto loop;
}
union_lock();
if (uppervp != un->un_uppervp) {
union_newupper(un, uppervp);
} else if (uppervp) {
union_unlock();
vnode_put(uppervp);
union_lock();
}
if (lowervp != un->un_lowervp) {
union_newlower(un, lowervp);
if (cnp && (lowervp != NULLVP)) {
un->un_hash = cnp->cn_hash;
union_unlock();
MALLOC(un->un_path, caddr_t, cnp->cn_namelen+1,
M_TEMP, M_WAITOK);
bcopy(cnp->cn_nameptr, un->un_path,
cnp->cn_namelen);
vnode_get(dvp);
union_lock();
un->un_path[cnp->cn_namelen] = '\0';
un->un_dirvp = dvp;
}
} else if (lowervp) {
union_unlock();
vnode_put(lowervp);
union_lock();
}
*vpp = UNIONTOV(un);
un->un_flags &= ~UN_LOCKED;
if ((un->un_flags & UN_WANT) == UN_WANT) {
un->un_flags &= ~UN_WANT;
wakeup(&un->un_flags);
}
return (0);
}
if (docache) {
hash = UNION_HASH(uppervp, lowervp);
if (union_list_lock(hash))
goto loop;
}
union_unlock();
MALLOC(unp, void *, sizeof(struct union_node), M_TEMP, M_WAITOK);
union_lock();
bzero(unp, sizeof(struct union_node));
un = unp;
un->un_uppervp = uppervp;
if (uppervp != NULLVP)
un->un_uppervid = vnode_vid(uppervp);
un->un_uppersz = VNOVAL;
un->un_lowervp = lowervp;
if (lowervp != NULLVP)
un->un_lowervid = vnode_vid(lowervp);
un->un_lowersz = VNOVAL;
un->un_pvp = undvp;
if (undvp != NULLVP)
vnode_get(undvp);
un->un_dircache = 0;
un->un_openl = 0;
un->un_mount = mp;
un->un_flags = UN_LOCKED;
#ifdef FAULTFS
if (UNION_FAULTIN(um))
un->un_flags |= UN_FAULTFS;
#endif
if (docache) {
LIST_INSERT_HEAD(&unhead[hash], un, un_cache);
un->un_flags |= UN_CACHED;
union_list_unlock(hash);
}
union_unlock();
if (uppervp)
vtype = uppervp->v_type;
else
vtype = lowervp->v_type;
bzero(&vfsp, sizeof(struct vnode_fsparam));
vfsp.vnfs_mp = mp;
vfsp.vnfs_vtype = vtype;
vfsp.vnfs_str = "unionfs";
vfsp.vnfs_dvp = undvp;
vfsp.vnfs_fsnode = unp;
vfsp.vnfs_cnp = cnp;
vfsp.vnfs_vops = union_vnodeop_p;
vfsp.vnfs_rdev = 0;
vfsp.vnfs_filesize = 0;
vfsp.vnfs_flags = VNFS_NOCACHE | VNFS_CANTCACHE;
vfsp.vnfs_marksystem = 0;
vfsp.vnfs_markroot = markroot;
error = vnode_create(VNCREATE_FLAVOR, VCREATESIZE, &vfsp, vpp);
if (error) {
if (uppervp) {
vnode_put(uppervp);
}
if (lowervp)
vnode_put(lowervp);
union_lock();
if (un->un_flags & UN_CACHED) {
un->un_flags &= ~UN_CACHED;
LIST_REMOVE(un, un_cache);
}
if (docache)
union_list_unlock(hash);
FREE(unp, M_TEMP);
return (error);
}
if (cnp && (lowervp != NULLVP)) {
un->un_hash = cnp->cn_hash;
un->un_path = _MALLOC(cnp->cn_namelen+1, M_TEMP, M_WAITOK);
bcopy(cnp->cn_nameptr, un->un_path, cnp->cn_namelen);
un->un_path[cnp->cn_namelen] = '\0';
vnode_get(dvp);
un->un_dirvp = dvp;
} else {
un->un_hash = 0;
un->un_path = 0;
un->un_dirvp = 0;
}
if (xlowervp)
vnode_put(xlowervp);
union_lock();
vnode_settag(*vpp, VT_UNION);
un->un_vnode = *vpp;
if (un->un_vnode->v_type == VDIR) {
if (un->un_uppervp == NULLVP) {
panic("faulting fs and no upper vp for dir?");
}
}
un->un_flags &= ~UN_LOCKED;
if ((un->un_flags & UN_WANT) == UN_WANT) {
un->un_flags &= ~UN_WANT;
wakeup(&un->un_flags);
}
return(error);
}
int
union_freevp(struct vnode *vp)
{
struct union_node *un = VTOUNION(vp);
if (un->un_flags & UN_CACHED) {
un->un_flags &= ~UN_CACHED;
LIST_REMOVE(un, un_cache);
}
union_unlock();
if (un->un_pvp != NULLVP)
vnode_put(un->un_pvp);
if (un->un_uppervp != NULLVP)
vnode_put(un->un_uppervp);
if (un->un_lowervp != NULLVP)
vnode_put(un->un_lowervp);
if (un->un_dirvp != NULLVP)
vnode_put(un->un_dirvp);
if (un->un_path)
_FREE(un->un_path, M_TEMP);
FREE(vp->v_data, M_TEMP);
vp->v_data = 0;
union_lock();
return (0);
}
int
union_copyfile(struct vnode *fvp, struct vnode *tvp, vfs_context_t context)
{
char *bufp;
struct uio uio;
struct iovec_32 iov;
int error = 0;
#if 1
uio.uio_segflg = UIO_SYSSPACE;
#else
uio.uio_segflg = UIO_SYSSPACE32;
#endif
uio.uio_offset = 0;
bufp = _MALLOC(MAXPHYSIO, M_TEMP, M_WAITOK);
do {
off_t offset = uio.uio_offset;
uio.uio_iovs.iov32p = &iov;
uio.uio_iovcnt = 1;
iov.iov_base = (uintptr_t)bufp;
iov.iov_len = MAXPHYSIO;
uio_setresid(&uio, iov.iov_len);
uio.uio_rw = UIO_READ;
error = VNOP_READ(fvp, &uio, 0, context);
if (error == 0) {
uio.uio_iovs.iov32p = &iov;
uio.uio_iovcnt = 1;
iov.iov_base = (uintptr_t)bufp;
iov.iov_len = MAXPHYSIO - uio_resid(&uio);
uio.uio_offset = offset;
uio.uio_rw = UIO_WRITE;
uio_setresid(&uio, iov.iov_len);
if (uio_resid(&uio) == 0)
break;
do {
error = VNOP_WRITE(tvp, &uio, 0, context);
} while ((uio_resid(&uio) > 0) && (error == 0));
}
} while (error == 0);
_FREE(bufp, M_TEMP);
return (error);
}
int
union_copyup(struct union_node *un, int docopy, vfs_context_t context)
{
int error;
struct vnode *lvp, *uvp;
struct vnode_attr vattr;
mode_t cmode = 0;
lvp = un->un_lowervp;
union_unlock();
if (UNNODE_FAULTIN(un)) {
VATTR_INIT(&vattr);
VATTR_WANTED(&vattr, va_flags);
if (vnode_getattr(lvp, &vattr, context) == 0 )
cmode = vattr.va_mode;
}
error = union_vn_create(&uvp, un, cmode, context);
if (error) {
union_lock();
if (error == EEXIST) {
if (uvp != NULLVP) {
union_newupper(un, uvp);
error = 0;
}
}
return (error);
}
union_lock();
union_newupper(un, uvp);
union_unlock();
if (docopy) {
error = VNOP_OPEN(lvp, FREAD, context);
if (error == 0) {
error = union_copyfile(lvp, uvp, context);
(void) VNOP_CLOSE(lvp, FREAD, context);
}
#ifdef UNION_DIAGNOSTIC
if (error == 0)
uprintf("union: copied up %s\n", un->un_path);
#endif
}
union_vn_close(uvp, FWRITE, context);
if (error == 0) {
int i;
for (i = 0; i < un->un_openl; i++) {
(void) VNOP_CLOSE(lvp, FREAD, context);
(void) VNOP_OPEN(uvp, FREAD, context);
}
un->un_openl = 0;
}
union_lock();
return (error);
}
int
union_faultin_copyup(struct vnode **vpp, vnode_t udvp, vnode_t lvp, struct componentname * cnp, vfs_context_t context)
{
int error;
struct vnode *uvp;
struct vnode_attr vattr;
struct vnode_attr *vap;
mode_t cmode = 0;
int fmode = FFLAGS(O_WRONLY|O_CREAT|O_TRUNC|O_EXCL);
struct proc * p = vfs_context_proc(context);
struct componentname cn;
vap = &vattr;
VATTR_INIT(vap);
VATTR_WANTED(vap, va_flags);
if (vnode_getattr(lvp, vap, context) == 0 )
cmode = vattr.va_mode;
*vpp = NULLVP;
if (cmode == (mode_t)0)
cmode = UN_FILEMODE & ~p->p_fd->fd_cmask;
else
cmode = cmode & ~p->p_fd->fd_cmask;
bzero(&cn, sizeof(struct componentname));
cn.cn_namelen = cnp->cn_namelen;
cn.cn_pnbuf = (caddr_t) _MALLOC_ZONE(cn.cn_namelen+1,
M_NAMEI, M_WAITOK);
cn.cn_pnlen = cn.cn_namelen+1;
bcopy(cnp->cn_nameptr, cn.cn_pnbuf, cn.cn_namelen+1);
cn.cn_nameiop = CREATE;
cn.cn_flags = (HASBUF|SAVENAME|SAVESTART|ISLASTCN|UNIONCREATED);
cn.cn_context = context;
cn.cn_nameptr = cn.cn_pnbuf;
cn.cn_hash = 0;
cn.cn_consume = 0;
if ((error = relookup(udvp, &uvp, &cn)) != 0) {
goto out;
}
if (uvp) {
*vpp = uvp;
error = EEXIST;
goto out;
}
VATTR_INIT(vap);
VATTR_SET(vap, va_type, VREG);
VATTR_SET(vap, va_mode, cmode);
cn.cn_flags |= (UNIONCREATED);
if ((error = vn_create(udvp, &uvp, &cn, vap, 0, context)) != 0) {
goto out;
}
if ((error = VNOP_OPEN(uvp, fmode, context)) != 0) {
vn_clearunionwait(uvp, 0);
vnode_recycle(uvp);
vnode_put(uvp);
goto out;
}
error = vnode_ref_ext(uvp, fmode);
if (error ) {
vn_clearunionwait(uvp, 0);
VNOP_CLOSE(uvp, fmode, context);
vnode_recycle(uvp);
vnode_put(uvp);
goto out;
}
error = VNOP_OPEN(lvp, FREAD, context);
if (error == 0) {
error = union_copyfile(lvp, uvp, context);
(void) VNOP_CLOSE(lvp, FREAD, context);
}
VNOP_CLOSE(uvp, fmode, context);
vnode_rele_ext(uvp, fmode, 0);
vn_clearunionwait(uvp, 0);
*vpp = uvp;
out:
if ((cn.cn_flags & HASBUF) == HASBUF) {
FREE_ZONE(cn.cn_pnbuf, cn.cn_pnlen, M_NAMEI);
cn.cn_flags &= ~HASBUF;
}
return (error);
}
static int
union_relookup(
#ifdef XXX_HELP_ME
struct union_mount *um,
#else
__unused struct union_mount *um,
#endif
struct vnode *dvp,
struct vnode **vpp,
struct componentname *cnp,
struct componentname *cn,
char *path,
int pathlen)
{
int error;
cn->cn_namelen = pathlen;
cn->cn_pnbuf = _MALLOC_ZONE(cn->cn_namelen+1, M_NAMEI, M_WAITOK);
cn->cn_pnlen = cn->cn_namelen+1;
bcopy(path, cn->cn_pnbuf, cn->cn_namelen);
cn->cn_pnbuf[cn->cn_namelen] = '\0';
cn->cn_nameiop = CREATE;
cn->cn_flags = (HASBUF|SAVENAME|SAVESTART|ISLASTCN );
#ifdef XXX_HELP_ME
cn->cn_proc = cnp->cn_proc;
if (um->um_op == UNMNT_ABOVE)
cn->cn_cred = cnp->cn_cred;
else
cn->cn_cred = um->um_cred;
#endif
cn->cn_context = cnp->cn_context;
cn->cn_nameptr = cn->cn_pnbuf;
cn->cn_hash = 0;
cn->cn_consume = cnp->cn_consume;
vnode_get(dvp);
error = relookup(dvp, vpp, cn);
vnode_put(dvp);
return (error);
}
int
union_mkshadow(um, dvp, cnp, vpp)
struct union_mount *um;
struct vnode *dvp;
struct componentname *cnp;
struct vnode **vpp;
{
int error;
struct vnode_attr va;
struct componentname cn;
bzero(&cn, sizeof(struct componentname));
error = union_relookup(um, dvp, vpp, cnp, &cn,
cnp->cn_nameptr, cnp->cn_namelen);
if (error)
goto out;
if (*vpp) {
error = EEXIST;
goto out;
}
VATTR_INIT(&va);
VATTR_SET(&va, va_type, VDIR);
VATTR_SET(&va, va_mode, um->um_cmode);
error = vn_create(dvp, vpp, &cn, &va, 0, cnp->cn_context);
out:
if ((cn.cn_flags & HASBUF) == HASBUF) {
FREE_ZONE(cn.cn_pnbuf, cn.cn_pnlen, M_NAMEI);
cn.cn_flags &= ~HASBUF;
}
return (error);
}
int
union_mkwhiteout(um, dvp, cnp, path)
struct union_mount *um;
struct vnode *dvp;
struct componentname *cnp;
char *path;
{
int error;
struct vnode *wvp;
struct componentname cn;
bzero(&cn, sizeof(struct componentname));
error = union_relookup(um, dvp, &wvp, cnp, &cn, path, strlen(path));
if (error) {
goto out;
}
if (wvp) {
error = EEXIST;
goto out;
}
error = VNOP_WHITEOUT(dvp, &cn, CREATE, cnp->cn_context);
out:
if ((cn.cn_flags & HASBUF) == HASBUF) {
FREE_ZONE(cn.cn_pnbuf, cn.cn_pnlen, M_NAMEI);
cn.cn_flags &= ~HASBUF;
}
return (error);
}
int
union_vn_create(struct vnode **vpp, struct union_node *un, mode_t cmode, vfs_context_t context)
{
struct vnode *vp;
struct vnode_attr vat;
struct vnode_attr *vap = &vat;
int fmode = FFLAGS(O_WRONLY|O_CREAT|O_TRUNC|O_EXCL);
int error;
struct proc * p = vfs_context_proc(context);
struct componentname cn;
bzero(&cn, sizeof(struct componentname));
*vpp = NULLVP;
if (cmode == (mode_t)0)
cmode = UN_FILEMODE & ~p->p_fd->fd_cmask;
else
cmode = cmode & ~p->p_fd->fd_cmask;
cn.cn_namelen = strlen(un->un_path);
cn.cn_pnbuf = (caddr_t) _MALLOC_ZONE(cn.cn_namelen+1,
M_NAMEI, M_WAITOK);
cn.cn_pnlen = cn.cn_namelen+1;
bcopy(un->un_path, cn.cn_pnbuf, cn.cn_namelen+1);
cn.cn_nameiop = CREATE;
if (UNNODE_FAULTIN(un))
cn.cn_flags = (HASBUF|SAVENAME|SAVESTART|ISLASTCN|UNIONCREATED);
else
cn.cn_flags = (HASBUF|SAVENAME|SAVESTART|ISLASTCN);
cn.cn_context = context;
cn.cn_nameptr = cn.cn_pnbuf;
cn.cn_hash = un->un_hash;
cn.cn_consume = 0;
vnode_get(un->un_dirvp);
if ((error = relookup(un->un_dirvp, &vp, &cn)) != 0) {
vnode_put(un->un_dirvp);
goto out;
}
vnode_put(un->un_dirvp);
if (vp) {
*vpp = vp;
error = EEXIST;
goto out;
}
VATTR_INIT(vap);
VATTR_SET(vap, va_type, VREG);
VATTR_SET(vap, va_mode, cmode);
if ((error = vn_create(un->un_dirvp, &vp, &cn, vap, 0, context)) != 0) {
goto out;
}
if ((error = VNOP_OPEN(vp, fmode, context)) != 0) {
vnode_put(vp);
goto out;
}
vnode_lock(vp);
if (++vp->v_writecount <= 0)
panic("union: v_writecount");
vnode_unlock(vp);
*vpp = vp;
error = 0;
out:
if ((cn.cn_flags & HASBUF) == HASBUF) {
FREE_ZONE(cn.cn_pnbuf, cn.cn_pnlen, M_NAMEI);
cn.cn_flags &= ~HASBUF;
}
return(error);
}
static int
union_vn_close(struct vnode *vp, int fmode, vfs_context_t context)
{
if (fmode & FWRITE) {
vnode_lock(vp);
--vp->v_writecount;
vnode_unlock(vp);
}
return (VNOP_CLOSE(vp, fmode, context));
}
void
union_removed_upper(un)
struct union_node *un;
{
union_newupper(un, NULLVP);
if (un->un_flags & UN_CACHED) {
un->un_flags &= ~UN_CACHED;
LIST_REMOVE(un, un_cache);
}
}
#if 0
struct vnode *
union_lowervp(vp)
struct vnode *vp;
{
struct union_node *un = VTOUNION(vp);
if ((un->un_lowervp != NULLVP) &&
(vp->v_type == un->un_lowervp->v_type)) {
if (vnode_get(un->un_lowervp) == 0)
return (un->un_lowervp);
}
return (NULLVP);
}
#endif
int
union_dowhiteout(struct union_node *un, vfs_context_t ctx)
{
struct vnode_attr va;
if (UNNODE_FAULTIN(un))
return(0);
if ((un->un_lowervp != NULLVP) )
return (1);
VATTR_INIT(&va);
VATTR_WANTED(&va, va_flags);
if (vnode_getattr(un->un_uppervp, &va, ctx) == 0 &&
(va.va_flags & OPAQUE))
return (1);
return (0);
}
static void
union_dircache_r(struct vnode *vp, struct vnode ***vppp, int *cntp)
{
struct union_node *un;
if (vp->v_op != union_vnodeop_p) {
if (vppp) {
vnode_get(vp);
*(*vppp)++ = vp;
if (--(*cntp) == 0)
panic("union: dircache table too small");
} else {
(*cntp)++;
}
return;
}
un = VTOUNION(vp);
if (un->un_uppervp != NULLVP)
union_dircache_r(un->un_uppervp, vppp, cntp);
if (un->un_lowervp != NULLVP)
union_dircache_r(un->un_lowervp, vppp, cntp);
}
struct vnode *
union_dircache(struct vnode *vp, __unused vfs_context_t context)
{
int count;
struct vnode *nvp, *lvp;
struct vnode **vpp;
struct vnode **dircache, **newdircache;
struct union_node *un;
int error;
int alloced = 0;
union_lock();
newdircache = NULL;
nvp = NULLVP;
un = VTOUNION(vp);
dircache = un->un_dircache;
if (dircache == 0) {
union_unlock();
count = 0;
union_dircache_r(vp, 0, &count);
count++;
#if 0
#if MAC_XXX
panic("MAC Framework doesn't support unionfs (yet)\n");
#endif
#endif
dircache = (struct vnode **)
_MALLOC(count * sizeof(struct vnode *),
M_TEMP, M_WAITOK);
newdircache = dircache;
alloced = 1;
vpp = dircache;
union_dircache_r(vp, &vpp, &count);
*vpp = NULLVP;
vpp = dircache + 1;
union_lock();
} else {
vpp = dircache;
do {
if (*vpp++ == un->un_uppervp)
break;
} while (*vpp != NULLVP);
}
lvp = *vpp;
union_unlock();
if (lvp == NULLVP) {
goto out;
}
vnode_get(lvp);
union_lock();
error = union_allocvp(&nvp, vp->v_mount, NULLVP, NULLVP, 0, lvp, NULLVP, 0);
if (error) {
union_unlock();
vnode_put(lvp);
goto out;
}
un->un_dircache = 0;
un = VTOUNION(nvp);
#if 0
if ((alloced != 0) && (un->un_dircache != 0)) {
union_unlock();
for (vpp = newdircache; *vpp != NULLVP; vpp++)
vnode_put(*vpp);
_FREE(newdircache, M_TEMP);
newdircache = NULL;
union_lock();
if (nvp != NULLVP)
union_freevp(nvp);
goto loop;
}
#endif
un->un_dircache = dircache;
un->un_flags |= UN_DIRENVN;
newdircache = NULL;
union_unlock();
return (nvp);
out:
if (newdircache) {
for (vpp = newdircache; *vpp != NULLVP; vpp++)
vnode_put(*vpp);
_FREE(newdircache, M_TEMP);
}
return (NULLVP);
}
static int
union_dircheck(struct vnode **vpp, struct fileproc *fp, vfs_context_t ctx)
{
int error = 0;
vnode_t vp = *vpp;
if (vp->v_op == union_vnodeop_p) {
struct vnode *lvp;
lvp = union_dircache(vp, ctx);
if (lvp != NULLVP) {
struct vnode_attr va;
VATTR_INIT(&va);
VATTR_WANTED(&va, va_flags);
error = vnode_getattr(vp, &va, ctx);
if (va.va_flags & OPAQUE) {
vnode_put(lvp);
lvp = NULL;
}
}
if (lvp != NULLVP) {
#if CONFIG_MACF
error = mac_vnode_check_open(ctx, lvp, FREAD);
if (error) {
vnode_put(lvp);
return(error);
}
#endif
error = VNOP_OPEN(lvp, FREAD, ctx);
if (error) {
vnode_put(lvp);
return(error);
}
vnode_ref(lvp);
fp->f_fglob->fg_data = (caddr_t) lvp;
fp->f_fglob->fg_offset = 0;
error = VNOP_CLOSE(vp, FREAD, ctx);
vnode_rele(vp);
vnode_put(vp);
if (error)
return(error);
*vpp = lvp;
return -1;
}
}
return error;
}
void
union_dircache_free(struct union_node *un)
{
struct vnode **vpp;
vpp = un->un_dircache;
un->un_dircache = NULL;
union_unlock();
for (; *vpp != NULLVP; vpp++)
vnode_put(*vpp);
_FREE(un->un_dircache, M_TEMP);
union_lock();
}