[apple/hfs.git] / core / hfs_quota.c

/*
 * Copyright (c) 2002-2015 Apple Inc. All rights reserved.
 *
 * @APPLE_OSREFERENCE_LICENSE_HEADER_START@
 * 
 * This file contains Original Code and/or Modifications of Original Code
 * as defined in and that are subject to the Apple Public Source License
 * Version 2.0 (the 'License'). You may not use this file except in
 * compliance with the License. The rights granted to you under the License
 * may not be used to create, or enable the creation or redistribution of,
 * unlawful or unlicensed copies of an Apple operating system, or to
 * circumvent, violate, or enable the circumvention or violation of, any
 * terms of an Apple operating system software license agreement.
 * 
 * Please obtain a copy of the License at
 * http://www.opensource.apple.com/apsl/ and read it before using this file.
 * 
 * The Original Code and all software distributed under the License are
 * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
 * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
 * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
 * Please see the License for the specific language governing rights and
 * limitations under the License.
 * 
 * @APPLE_OSREFERENCE_LICENSE_HEADER_END@
 */
/*
 * Copyright (c) 1982, 1986, 1990, 1993, 1995
 *      The Regents of the University of California.  All rights reserved.
 *
 * This code is derived from software contributed to Berkeley by
 * Robert Elz at The University of Melbourne.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 * 3. All advertising materials mentioning features or use of this software
 *    must display the following acknowledgement:
 *      This product includes software developed by the University of
 *      California, Berkeley and its contributors.
 * 4. Neither the name of the University nor the names of its contributors
 *    may be used to endorse or promote products derived from this software
 *    without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 *
 *      @(#)hfs_quota.c
 *      derived from @(#)ufs_quota.c    8.5 (Berkeley) 5/20/95
 */

#if QUOTA

#include <sys/param.h>
#include <sys/kernel.h>
#include <sys/systm.h>
#include <sys/mount.h>
#include <sys/malloc.h>
#include <sys/file.h>
#include <sys/proc.h>
#include <sys/kauth.h>
#include <sys/vnode.h>
#include <sys/quota.h>
#include <sys/vm.h>

#include "hfs.h"
#include "hfs_cnode.h"
#include "hfs_quota.h"
#include "hfs_mount.h"


/*
 * Quota name to error message mapping.
 */
#if 0
static char *quotatypes[] = INITQFNAMES;
#endif

/*
 * Set up the quotas for a cnode.
 *
 * This routine completely defines the semantics of quotas.
 * If other criterion want to be used to establish quotas, the
 * MAXQUOTAS value in quotas.h should be increased, and the
 * additional dquots set up here.
 */
int
hfs_getinoquota(cp)
        register struct cnode *cp;
{
        struct hfsmount *hfsmp;
        struct vnode *vp;
        int error;
        int drop_usrquota = false;

        vp = cp->c_vp ? cp->c_vp : cp->c_rsrc_vp;
        hfsmp = VTOHFS(vp);
        /*
         * Set up the user quota based on file uid.
         * EINVAL means that quotas are not enabled.
         */
        if (cp->c_dquot[USRQUOTA] == NODQUOT) {
                error = dqget(cp->c_uid, &hfsmp->hfs_qfiles[USRQUOTA], USRQUOTA, &cp->c_dquot[USRQUOTA]);
                if ((error != 0) && (error != EINVAL)) {
                        return error;
                } else if (error == 0) {
                        drop_usrquota = true;
                }
        }

        /*
         * Set up the group quota based on file gid.
         * EINVAL means that quotas are not enabled.
         */
        if (cp->c_dquot[GRPQUOTA] == NODQUOT) {
               error = dqget(cp->c_gid, &hfsmp->hfs_qfiles[GRPQUOTA], GRPQUOTA, &cp->c_dquot[GRPQUOTA]);
               if ((error != 0) && (error != EINVAL)) {
                       if (drop_usrquota == true) {
                               dqrele(cp->c_dquot[USRQUOTA]);
                               cp->c_dquot[USRQUOTA] = NODQUOT;
                       }
                       return error;
               }
        }

        return (0);
}

/*
 * Update disk usage, and take corrective action.
 */
int
hfs_chkdq(cp, change, cred, flags)
        register struct cnode *cp;
        int64_t change;
        kauth_cred_t cred;
        int flags;
{
        register struct dquot *dq;
        register int i;
        int64_t ncurbytes;
        int error=0;
        struct proc *p;

#if DIAGNOSTIC
        if ((flags & CHOWN) == 0)
                hfs_chkdquot(cp);
#endif
        if (change == 0)
                return (0);
        if (change < 0) {
                for (i = 0; i < MAXQUOTAS; i++) {
                        if ((dq = cp->c_dquot[i]) == NODQUOT)
                                continue;
                        dqlock(dq);

                        ncurbytes = dq->dq_curbytes + change;
                        if (ncurbytes >= 0)
                                dq->dq_curbytes = ncurbytes;
                        else
                                dq->dq_curbytes = 0;
                        dq->dq_flags &= ~DQ_BLKS;
                        dq->dq_flags |= DQ_MOD;

                        dqunlock(dq);
                }
                return (0);
        }
        p = current_proc();
        /*
         * This use of proc_ucred() is safe because kernproc credential never
         * changes.
         */
        if (!IS_VALID_CRED(cred))
                cred = proc_ucred(kernproc);
        if (suser(cred, NULL) || proc_forcequota(p)) {
                for (i = 0; i < MAXQUOTAS; i++) {
                        if ((dq = cp->c_dquot[i]) == NODQUOT)
                                continue;
                        error = hfs_chkdqchg(cp, change, cred, i);
                        if (error) {
                                break;
                        }
                }
        }
        if ((flags & FORCE) || error == 0) {
                for (i = 0; i < MAXQUOTAS; i++) {
                        if ((dq = cp->c_dquot[i]) == NODQUOT)
                                continue;
                        dqlock(dq);

                        dq->dq_curbytes += change;
                        dq->dq_flags |= DQ_MOD;

                        dqunlock(dq);
                }
        }
        return (error);
}

/*
 * Check for a valid change to a users allocation.
 * Issue an error message and vfs event if appropriate.
 */
int
hfs_chkdqchg(cp, change, cred, type)
        struct cnode *cp;
        int64_t change;
        kauth_cred_t cred;
        int type;
{
        register struct dquot *dq = cp->c_dquot[type];
        u_int64_t ncurbytes;
        struct vnode *vp = cp->c_vp ? cp->c_vp : cp->c_rsrc_vp;

        fsid_t fsid;
        fsid.val[0] = VTOHFS(vp)->hfs_raw_dev;
        fsid.val[1] = vfs_typenum(VTOVFS(vp));

        dqlock(dq);
        
        ncurbytes = dq->dq_curbytes + change;
        /*
         * If user would exceed their hard limit, disallow space allocation.
         */
        if (ncurbytes >= dq->dq_bhardlimit && dq->dq_bhardlimit) {
                if ((dq->dq_flags & DQ_BLKS) == 0 &&
                    cp->c_uid == kauth_cred_getuid(cred)) {
#if 0   
                        printf("\nhfs: write failed, %s disk limit reached\n",
                            quotatypes[type]);
#endif
                        dq->dq_flags |= DQ_BLKS;
                        vfs_event_signal(&fsid, VQ_QUOTA, (intptr_t)NULL);
                }
                dqunlock(dq);

                return (EDQUOT);
        }
        /*
         * If user is over their soft limit for too long, disallow space
         * allocation. Reset time limit as they cross their soft limit.
         */
        if (ncurbytes >= dq->dq_bsoftlimit && dq->dq_bsoftlimit) {
                struct timeval tv;

                microuptime(&tv);
                if (dq->dq_curbytes < dq->dq_bsoftlimit) {
                        dq->dq_btime = tv.tv_sec +
                            VTOHFS(vp)->hfs_qfiles[type].qf_btime;
#if 0
                        if (cp->c_uid == kauth_cred_getuid(cred))
                                printf("\nhfs: warning, %s %s\n",
                                    quotatypes[type], "disk quota exceeded");
#endif
                        vfs_event_signal(&fsid, VQ_QUOTA, (intptr_t)NULL);
                        dqunlock(dq);

                        return (0);
                }
                if (tv.tv_sec > (time_t)dq->dq_btime) {
                        if ((dq->dq_flags & DQ_BLKS) == 0 &&
                            cp->c_uid == kauth_cred_getuid(cred)) {
#if 0
                                printf("\nhfs: write failed, %s %s\n",
                                    quotatypes[type],
                                    "disk quota exceeded for too long");
#endif
                                dq->dq_flags |= DQ_BLKS;
                                vfs_event_signal(&fsid, VQ_QUOTA, (intptr_t)NULL);
                        }
                        dqunlock(dq);

                        return (EDQUOT);
                }
        }
        dqunlock(dq);

        return (0);
}

/*
 * Check the inode limit, applying corrective action.
 */
int
hfs_chkiq(cp, change, cred, flags)
        register struct cnode *cp;
        int32_t change;
        kauth_cred_t cred;
        int flags;
{
        register struct dquot *dq;
        register int i;
        int ncurinodes, error=0;
        struct proc *p;

#if DIAGNOSTIC
        if ((flags & CHOWN) == 0)
                hfs_chkdquot(cp);
#endif
        if (change == 0)
                return (0);
        if (change < 0) {
                for (i = 0; i < MAXQUOTAS; i++) {
                        if ((dq = cp->c_dquot[i]) == NODQUOT)
                                continue;
                        dqlock(dq);

                        ncurinodes = dq->dq_curinodes + change;
                        if (ncurinodes >= 0)
                                dq->dq_curinodes = ncurinodes;
                        else
                                dq->dq_curinodes = 0;
                        dq->dq_flags &= ~DQ_INODS;
                        dq->dq_flags |= DQ_MOD;

                        dqunlock(dq);
                }
                return (0);
        }
        p = current_proc();
        /*
         * This use of proc_ucred() is safe because kernproc credential never
         * changes.
         */
        if (!IS_VALID_CRED(cred))
                cred = proc_ucred(kernproc);
        if (suser(cred, NULL) || proc_forcequota(p)) {
                for (i = 0; i < MAXQUOTAS; i++) {
                        if ((dq = cp->c_dquot[i]) == NODQUOT)
                                continue;
                        error = hfs_chkiqchg(cp, change, cred, i);
                        if (error) {
                                break;
                        }
                }
        }
        if ((flags & FORCE) || error == 0) { 
                for (i = 0; i < MAXQUOTAS; i++) {
                        if ((dq = cp->c_dquot[i]) == NODQUOT)
                                continue;
                        dqlock(dq);

                        dq->dq_curinodes += change;
                        dq->dq_flags |= DQ_MOD;

                        dqunlock(dq);
                }
        }
        return (error);
}


/*
 * Check to see if a change to a user's allocation should be permitted or not.
 * Issue an error message if it should not be permitted.  Return 0 if 
 * it should be allowed.
 */
int hfs_isiqchg_allowed(dq, hfsmp, change, cred, type, uid)
        struct dquot* dq;
        struct hfsmount* hfsmp;
        int32_t change;
        kauth_cred_t cred;
        int type;
        uid_t uid;
{
        u_int32_t ncurinodes;

        fsid_t fsid;
        fsid.val[0] = hfsmp->hfs_raw_dev;
        fsid.val[1] = vfs_typenum(HFSTOVFS(hfsmp));

        dqlock(dq);

        ncurinodes = dq->dq_curinodes + change;
        /*
         * If user would exceed their hard limit, disallow cnode allocation.
         */
        if (ncurinodes >= dq->dq_ihardlimit && dq->dq_ihardlimit) {
                if ((dq->dq_flags & DQ_INODS) == 0 &&
                    uid == kauth_cred_getuid(cred)) {
                        dq->dq_flags |= DQ_INODS;
                        vfs_event_signal(&fsid, VQ_QUOTA, (intptr_t)NULL);
                }
                dqunlock(dq);

                return (EDQUOT);
        }
        /*
         * If user is over their soft limit for too long, disallow cnode
         * allocation. Reset time limit as they cross their soft limit.
         */
        if (ncurinodes >= dq->dq_isoftlimit && dq->dq_isoftlimit) {
                struct timeval tv;
                
                microuptime(&tv);
                if (dq->dq_curinodes < dq->dq_isoftlimit) {
                        dq->dq_itime = tv.tv_sec + hfsmp->hfs_qfiles[type].qf_itime;
                        vfs_event_signal(&fsid, VQ_QUOTA, (intptr_t)NULL);
                        dqunlock(dq);
                        return (0);
                }
                if (tv.tv_sec > (time_t)dq->dq_itime) {
                        if (((dq->dq_flags & DQ_INODS) == 0) &&
                            (uid == kauth_cred_getuid(cred))) {
                                dq->dq_flags |= DQ_INODS;
                                vfs_event_signal(&fsid, VQ_QUOTA, (intptr_t)NULL);
                        }
                        dqunlock(dq);

                        return (EDQUOT);
                }
        }
        dqunlock(dq);

        return (0);
}


/*
 * Check for a valid change to a users allocation.
 * Issue an error message if appropriate.
 */
int
hfs_chkiqchg(cp, change, cred, type)
        struct cnode *cp;
        int32_t change;
        kauth_cred_t cred;
        int type;
{
        register struct dquot *dq = cp->c_dquot[type];
        u_int32_t ncurinodes;
        struct vnode *vp = cp->c_vp ? cp->c_vp : cp->c_rsrc_vp;

        dqlock(dq);

        ncurinodes = dq->dq_curinodes + change;
        /*
         * If user would exceed their hard limit, disallow cnode allocation.
         */
        if (ncurinodes >= dq->dq_ihardlimit && dq->dq_ihardlimit) {
                if ((dq->dq_flags & DQ_INODS) == 0 &&
                    cp->c_uid == kauth_cred_getuid(cred)) {
#if 0
                        printf("\nhfs: write failed, %s cnode limit reached\n",
                            quotatypes[type]);
#endif
                        dq->dq_flags |= DQ_INODS;
                }
                dqunlock(dq);

                return (EDQUOT);
        }
        /*
         * If user is over their soft limit for too long, disallow cnode
         * allocation. Reset time limit as they cross their soft limit.
         */
        if (ncurinodes >= dq->dq_isoftlimit && dq->dq_isoftlimit) {
                struct timeval tv;
                
                microuptime(&tv);
                if (dq->dq_curinodes < dq->dq_isoftlimit) {
                        dq->dq_itime = tv.tv_sec +
                            VTOHFS(vp)->hfs_qfiles[type].qf_itime;
#if 0
                        if (cp->c_uid == kauth_cred_getuid(cred))
                                printf("\nhfs: warning, %s %s\n",
                                    quotatypes[type], "cnode quota exceeded");
#endif
                        dqunlock(dq);

                        return (0);
                }
                if (tv.tv_sec > (time_t)dq->dq_itime) {
                        if ((dq->dq_flags & DQ_INODS) == 0 &&
                            cp->c_uid == kauth_cred_getuid(cred)) {
#if 0
                                printf("\nhfs: write failed, %s %s\n",
                                    quotatypes[type],
                                    "cnode quota exceeded for too long");
#endif
                                dq->dq_flags |= DQ_INODS;
                        }
                        dqunlock(dq);

                        return (EDQUOT);
                }
        }
        dqunlock(dq);

        return (0);
}

#if DIAGNOSTIC
/*
 * On filesystems with quotas enabled, it is an error for a file to change
 * size and not to have a dquot structure associated with it.
 */
void
hfs_chkdquot(cp)
        register struct cnode *cp;
{
        struct vnode *vp = cp->c_vp ? cp->c_vp : cp->c_rsrc_vp;
        struct hfsmount *hfsmp = VTOHFS(vp);
        register int i;

        for (i = 0; i < MAXQUOTAS; i++) {
                if (hfsmp->hfs_qfiles[i].qf_vp == NULLVP)
                        continue;
                if (cp->c_dquot[i] == NODQUOT) {
                        vprint("chkdquot: missing dquot", vp);
                        panic("missing dquot");
                }
        }
}
#endif

/*
 * Code to process quotactl commands.
 */

/*
 * Q_QUOTAON - set up a quota file for a particular file system.
 */
struct hfs_quotaon_cargs {
        int     error;
};

static int
hfs_quotaon_callback(struct vnode *vp, void *cargs)
{
        struct hfs_quotaon_cargs *args;

        args = (struct hfs_quotaon_cargs *)cargs;

        args->error = hfs_getinoquota(VTOC(vp));
        if (args->error)
                return (VNODE_RETURNED_DONE);

        return (VNODE_RETURNED);
}

int
hfs_quotaon(p, mp, type, fnamep)
        struct proc *p;
        struct mount *mp;
        register int type;
        caddr_t fnamep;
{
        struct hfsmount *hfsmp = VFSTOHFS(mp);
        struct quotafile *qfp;
        struct vnode *vp;
        int error = 0;
        struct hfs_quotaon_cargs args;

        /* Finish setting up quota structures. */
        dqhashinit();

        qfp = &hfsmp->hfs_qfiles[type];

        if ( (qf_get(qfp, QTF_OPENING)) )
                return (0);

        error = vnode_open(fnamep, FREAD|FWRITE, 0, 0, &vp, NULL);
        if (error) {
                goto out;
        }
        if (!vnode_isreg(vp)) {
                (void) vnode_close(vp, FREAD|FWRITE, NULL);
                error = EACCES;
                goto out;
        }
        vfs_setflags(mp, (u_int64_t)((unsigned int)MNT_QUOTA));
        hfs_lock_mount (hfsmp);
        hfsmp->hfs_flags |= HFS_QUOTAS;
        hfs_unlock_mount (hfsmp);
        vnode_setnoflush(vp);
        /*
         * Save the credential of the process that turned on quotas.
         */
        qfp->qf_cred = kauth_cred_proc_ref(p);
        qfp->qf_vp = vp;
        /*
         * Finish initializing the quota file
         */
        error = dqfileopen(qfp, type);
        if (error) {
                (void) vnode_close(vp, FREAD|FWRITE, NULL);

                if (IS_VALID_CRED(qfp->qf_cred))
                        kauth_cred_unref(&qfp->qf_cred);
                qfp->qf_vp = NULLVP;
                goto out;
        }
        qf_put(qfp, QTF_OPENING);

        /*
         * Search vnodes associated with this mount point,
         * adding references to quota file being opened.
         * NB: only need to add dquot's for cnodes being modified.
         *
         * hfs_quota_callback will be called for each vnode open for
         * 'write' (VNODE_WRITEABLE) hung off of this mount point
         * the vnode will be in an 'unbusy' state (VNODE_WAIT) and 
         * properly referenced and unreferenced around the callback
         */
        args.error = 0;

        vnode_iterate(mp, VNODE_WRITEABLE | VNODE_WAIT, hfs_quotaon_callback, (void *)&args);
        
        error = args.error;

        if (error) {
                hfs_quotaoff(p, mp, type);
        }
        return (error);

out:
        qf_put(qfp, QTF_OPENING);

        return (error);
}


/*
 * Q_QUOTAOFF - turn off disk quotas for a filesystem.
 */
struct hfs_quotaoff_cargs {
        int     type;
};

static int
hfs_quotaoff_callback(struct vnode *vp, void *cargs)
{
        struct hfs_quotaoff_cargs *args;
        struct cnode *cp;
        struct dquot *dq;

        args = (struct hfs_quotaoff_cargs *)cargs;

        cp = VTOC(vp);

        dq = cp->c_dquot[args->type];
        cp->c_dquot[args->type] = NODQUOT;

        dqrele(dq);

        return (VNODE_RETURNED);
}

int
hfs_quotaoff(__unused struct proc *p, struct mount *mp, register int type)
{
        struct vnode *qvp;
        struct hfsmount *hfsmp = VFSTOHFS(mp);
        struct quotafile *qfp;
        int error;
        struct hfs_quotaoff_cargs args;

        /*
         * If quotas haven't been initialized, there's no work to be done.
         */
        if (!dqisinitialized())
                return (0);

        qfp = &hfsmp->hfs_qfiles[type];
        
        if ( (qf_get(qfp, QTF_CLOSING)) )
                return (0);
        qvp = qfp->qf_vp;

        /*
         * Sync out any orpaned dirty dquot entries.
         */
        dqsync_orphans(qfp);

        /*
         * Search vnodes associated with this mount point,
         * deleting any references to quota file being closed.
     *
         * hfs_quotaoff_callback will be called for each vnode
         * hung off of this mount point
         * the vnode will be in an 'unbusy' state (VNODE_WAIT) and 
         * properly referenced and unreferenced around the callback
         */
        args.type = type;

        vnode_iterate(mp, VNODE_WAIT, hfs_quotaoff_callback, (void *)&args);

        dqflush(qvp);
        /* Finish tearing down the quota file */
        dqfileclose(qfp, type);

        vnode_clearnoflush(qvp);
        error = vnode_close(qvp, FREAD|FWRITE, NULL);

        qfp->qf_vp = NULLVP;

        if (IS_VALID_CRED(qfp->qf_cred))
                kauth_cred_unref(&qfp->qf_cred);
        for (type = 0; type < MAXQUOTAS; type++)
                if (hfsmp->hfs_qfiles[type].qf_vp != NULLVP)
                        break;
        if (type == MAXQUOTAS) {
                vfs_clearflags(mp, (u_int64_t)((unsigned int)MNT_QUOTA));
                hfs_lock_mount (hfsmp);
                hfsmp->hfs_flags &= ~HFS_QUOTAS;
                hfs_unlock_mount (hfsmp);
        }

        qf_put(qfp, QTF_CLOSING);

        return (error);
}

/*
 * hfs_quotacheck - checks quotas mountwide for a hypothetical situation.  It probes
 * the quota data structures to see if adding an inode would be allowed or not.  If it
 * will be allowed, the change is made.  Otherwise, it reports an error back out so the
 * caller will know not to proceed with inode allocation in the HFS Catalog.
 * 
 * Note that this function ONLY tests for addition of inodes, not subtraction.
 */
int hfs_quotacheck(hfsmp, change, uid, gid, cred)
        struct hfsmount *hfsmp;
        int change;
        uid_t uid;
        gid_t gid;
        kauth_cred_t cred;
{
        struct dquot *dq = NULL;
        struct proc *p;
        int error = 0;
        int i;
        id_t id = uid;

        p = current_proc();
        if (!IS_VALID_CRED(cred)) {
                /* This use of proc_ucred() is safe because kernproc credential never changes */
                cred = proc_ucred(kernproc);
        }

        if (suser(cred, NULL) || proc_forcequota(p)) {
                for (i = 0; i < MAXQUOTAS; i++) {
                        /* Select if user or group id should be used */
                        if (i == USRQUOTA)
                                id = uid;
                        else if (i == GRPQUOTA)
                                id = gid;

                        error = dqget(id, &hfsmp->hfs_qfiles[i], i, &dq);
                        if (error && (error != EINVAL))
                                break;

                        error = 0;
                        if (dq == NODQUOT)
                                continue;

                        /* Check quota information */
                        error = hfs_isiqchg_allowed(dq, hfsmp, change, cred, i, id);
                        if (error) {
                                dqrele(dq);
                                break;
                        }
                        
                        dqlock(dq);
                        /* Update quota information */
                        dq->dq_curinodes += change;
                        dqunlock(dq);
                        dqrele(dq);
                }
        }

        return error;
}


/*
 * Q_GETQUOTA - return current values in a dqblk structure.
 */
int
hfs_getquota(mp, id, type, datap)
        struct mount *mp;
        u_int32_t id;
        int type;
        caddr_t datap;
{
        struct dquot *dq;
        int error;

        error = dqget(id, &VFSTOHFS(mp)->hfs_qfiles[type], type, &dq);
        if (error)
                return (error);
        dqlock(dq);

        bcopy(&dq->dq_dqb, datap, sizeof(dq->dq_dqb));
        
        dqunlock(dq);
        dqrele(dq);

        return (error);
}

/*
 * Q_SETQUOTA - assign an entire dqblk structure.
 */
int
hfs_setquota(mp, id, type, datap)
        struct mount *mp;
        u_int32_t id;
        int type;
        caddr_t datap;
{
        struct dquot *dq;
        struct hfsmount *hfsmp = VFSTOHFS(mp);
        struct dqblk * newlimp = (struct dqblk *) datap;
        struct timeval tv;
        int error;

        error = dqget(id, &hfsmp->hfs_qfiles[type], type, &dq);
        if (error)
                return (error);
        dqlock(dq);

        /*
         * Copy all but the current values.
         * Reset time limit if previously had no soft limit or were
         * under it, but now have a soft limit and are over it.
         */
        newlimp->dqb_curbytes = dq->dq_curbytes;
        newlimp->dqb_curinodes = dq->dq_curinodes;
        if (dq->dq_id != 0) {
                newlimp->dqb_btime = dq->dq_btime;
                newlimp->dqb_itime = dq->dq_itime;
        }
        if (newlimp->dqb_bsoftlimit &&
            dq->dq_curbytes >= newlimp->dqb_bsoftlimit &&
            (dq->dq_bsoftlimit == 0 || dq->dq_curbytes < dq->dq_bsoftlimit)) {
                microuptime(&tv);
                newlimp->dqb_btime = tv.tv_sec + hfsmp->hfs_qfiles[type].qf_btime;
        }
        if (newlimp->dqb_isoftlimit &&
            dq->dq_curinodes >= newlimp->dqb_isoftlimit &&
            (dq->dq_isoftlimit == 0 || dq->dq_curinodes < dq->dq_isoftlimit)) {
                microuptime(&tv);
                newlimp->dqb_itime = tv.tv_sec + hfsmp->hfs_qfiles[type].qf_itime;
        }
        bcopy(newlimp, &dq->dq_dqb, sizeof(dq->dq_dqb));
        if (dq->dq_curbytes < dq->dq_bsoftlimit)
                dq->dq_flags &= ~DQ_BLKS;
        if (dq->dq_curinodes < dq->dq_isoftlimit)
                dq->dq_flags &= ~DQ_INODS;
        if (dq->dq_isoftlimit == 0 && dq->dq_bsoftlimit == 0 &&
            dq->dq_ihardlimit == 0 && dq->dq_bhardlimit == 0)
                dq->dq_flags |= DQ_FAKE;
        else
                dq->dq_flags &= ~DQ_FAKE;
        dq->dq_flags |= DQ_MOD;

        dqunlock(dq);
        dqrele(dq);

        return (0);
}

/*
 * Q_SETUSE - set current cnode and byte usage.
 */
int
hfs_setuse(mp, id, type, datap)
        struct mount *mp;
        u_int32_t id;
        int type;
        caddr_t datap;
{
        struct hfsmount *hfsmp = VFSTOHFS(mp);
        struct dquot *dq;
        struct timeval tv;
        int error;
        struct dqblk *quotablkp = (struct dqblk *) datap;

        error = dqget(id, &hfsmp->hfs_qfiles[type], type, &dq);
        if (error)
                return (error);
        dqlock(dq);

        /*
         * Reset time limit if have a soft limit and were
         * previously under it, but are now over it.
         */
        if (dq->dq_bsoftlimit && dq->dq_curbytes < dq->dq_bsoftlimit &&
            quotablkp->dqb_curbytes >= dq->dq_bsoftlimit) {
                microuptime(&tv);
                dq->dq_btime = tv.tv_sec + hfsmp->hfs_qfiles[type].qf_btime;
        }
        if (dq->dq_isoftlimit && dq->dq_curinodes < dq->dq_isoftlimit &&
            quotablkp->dqb_curinodes >= dq->dq_isoftlimit) {
                microuptime(&tv);
                dq->dq_itime = tv.tv_sec + hfsmp->hfs_qfiles[type].qf_itime;
        }
        dq->dq_curbytes = quotablkp->dqb_curbytes;
        dq->dq_curinodes = quotablkp->dqb_curinodes;
        if (dq->dq_curbytes < dq->dq_bsoftlimit)
                dq->dq_flags &= ~DQ_BLKS;
        if (dq->dq_curinodes < dq->dq_isoftlimit)
                dq->dq_flags &= ~DQ_INODS;
        dq->dq_flags |= DQ_MOD;

        dqunlock(dq);
        dqrele(dq);

        return (0);
}


/*
 * Q_SYNC - sync quota files to disk.
 */
static int
hfs_qsync_callback(struct vnode *vp, __unused void *cargs)
{
        struct cnode *cp;
        struct dquot *dq;
        int     i;

        cp = VTOC(vp);
                    
        for (i = 0; i < MAXQUOTAS; i++) {
                dq = cp->c_dquot[i];
                if (dq != NODQUOT && (dq->dq_flags & DQ_MOD))
                        dqsync(dq);
        }
        return (VNODE_RETURNED);
}

int
hfs_qsync(mp)
        struct mount *mp;
{
        struct hfsmount *hfsmp = VFSTOHFS(mp);
        int i;

        if (!dqisinitialized())
                return (0);

        /*
         * Check if the mount point has any quotas.
         * If not, simply return.
         */
        for (i = 0; i < MAXQUOTAS; i++)
                if (hfsmp->hfs_qfiles[i].qf_vp != NULLVP)
                        break;
        if (i == MAXQUOTAS)
                return (0);

        /*
         * Sync out any orpaned dirty dquot entries.
         */
        for (i = 0; i < MAXQUOTAS; i++)
                if (hfsmp->hfs_qfiles[i].qf_vp != NULLVP)
                        dqsync_orphans(&hfsmp->hfs_qfiles[i]);

        /*
         * Search vnodes associated with this mount point,
         * synchronizing any modified dquot structures.
         *
         * hfs_qsync_callback will be called for each vnode
         * hung off of this mount point
         * the vnode will be
         * properly referenced and unreferenced around the callback
         */
        vnode_iterate(mp, 0, hfs_qsync_callback, (void *)NULL);

        return (0);
}

/*
 * Q_QUOTASTAT - get quota on/off status 
 */
int
hfs_quotastat(mp, type, datap)
        struct mount *mp;
        register int type;
        caddr_t datap;
{
        struct hfsmount *hfsmp = VFSTOHFS(mp);
        int error = 0;
        int qstat;

        if ((((unsigned int)vfs_flags(mp)) & MNT_QUOTA) && (hfsmp->hfs_qfiles[type].qf_vp != NULLVP))
          qstat = 1;   /* quotas are on for this type */
        else
          qstat = 0;   /* quotas are off for this type */
        
        *((int *)datap) = qstat;
        return (error);
}

#endif // QUOTA