Home | History | Annotate | Line # | Download | only in ufs
ufs_quota.c revision 1.73
      1  1.73  dholland /*	$NetBSD: ufs_quota.c,v 1.73 2012/01/29 06:36:51 dholland Exp $	*/
      2   1.2       cgd 
      3   1.1   mycroft /*
      4  1.11      fvdl  * Copyright (c) 1982, 1986, 1990, 1993, 1995
      5   1.1   mycroft  *	The Regents of the University of California.  All rights reserved.
      6   1.1   mycroft  *
      7   1.1   mycroft  * This code is derived from software contributed to Berkeley by
      8   1.1   mycroft  * Robert Elz at The University of Melbourne.
      9   1.1   mycroft  *
     10   1.1   mycroft  * Redistribution and use in source and binary forms, with or without
     11   1.1   mycroft  * modification, are permitted provided that the following conditions
     12   1.1   mycroft  * are met:
     13   1.1   mycroft  * 1. Redistributions of source code must retain the above copyright
     14   1.1   mycroft  *    notice, this list of conditions and the following disclaimer.
     15   1.1   mycroft  * 2. Redistributions in binary form must reproduce the above copyright
     16   1.1   mycroft  *    notice, this list of conditions and the following disclaimer in the
     17   1.1   mycroft  *    documentation and/or other materials provided with the distribution.
     18  1.28       agc  * 3. Neither the name of the University nor the names of its contributors
     19   1.1   mycroft  *    may be used to endorse or promote products derived from this software
     20   1.1   mycroft  *    without specific prior written permission.
     21   1.1   mycroft  *
     22   1.1   mycroft  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
     23   1.1   mycroft  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
     24   1.1   mycroft  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
     25   1.1   mycroft  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
     26   1.1   mycroft  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
     27   1.1   mycroft  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
     28   1.1   mycroft  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
     29   1.1   mycroft  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
     30   1.1   mycroft  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
     31   1.1   mycroft  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
     32   1.1   mycroft  * SUCH DAMAGE.
     33   1.1   mycroft  *
     34  1.11      fvdl  *	@(#)ufs_quota.c	8.5 (Berkeley) 5/20/95
     35   1.1   mycroft  */
     36  1.22     lukem 
     37  1.22     lukem #include <sys/cdefs.h>
     38  1.73  dholland __KERNEL_RCSID(0, "$NetBSD: ufs_quota.c,v 1.73 2012/01/29 06:36:51 dholland Exp $");
     39  1.22     lukem 
     40  1.69    bouyer #if defined(_KERNEL_OPT)
     41  1.69    bouyer #include "opt_quota.h"
     42  1.69    bouyer #endif
     43   1.1   mycroft #include <sys/param.h>
     44   1.1   mycroft #include <sys/kernel.h>
     45   1.1   mycroft #include <sys/systm.h>
     46   1.1   mycroft #include <sys/namei.h>
     47   1.1   mycroft #include <sys/file.h>
     48   1.1   mycroft #include <sys/proc.h>
     49   1.1   mycroft #include <sys/vnode.h>
     50   1.1   mycroft #include <sys/mount.h>
     51  1.39      elad #include <sys/kauth.h>
     52   1.1   mycroft 
     53  1.71  dholland #include <sys/quotactl.h>
     54   1.1   mycroft #include <ufs/ufs/quota.h>
     55   1.1   mycroft #include <ufs/ufs/inode.h>
     56   1.1   mycroft #include <ufs/ufs/ufsmount.h>
     57   1.1   mycroft #include <ufs/ufs/ufs_extern.h>
     58  1.69    bouyer #include <ufs/ufs/ufs_quota.h>
     59  1.70    bouyer #include <quota/quotaprop.h>
     60   1.1   mycroft 
     61  1.69    bouyer kmutex_t dqlock;
     62  1.69    bouyer kcondvar_t dqcv;
     63  1.48   hannken 
     64  1.48   hannken /*
     65  1.69    bouyer  * Code pertaining to management of the in-core dquot data structures.
     66   1.1   mycroft  */
     67  1.69    bouyer #define DQHASH(dqvp, id) \
     68  1.69    bouyer 	(((((long)(dqvp)) >> 8) + id) & dqhash)
     69  1.69    bouyer static LIST_HEAD(dqhashhead, dquot) *dqhashtbl;
     70  1.69    bouyer static u_long dqhash;
     71  1.69    bouyer static pool_cache_t dquot_cache;
     72   1.1   mycroft 
     73   1.1   mycroft 
     74  1.69    bouyer static int quota_handle_cmd_get_version(struct mount *, struct lwp *,
     75  1.72  dholland     struct vfs_quotactl_args *args);
     76  1.69    bouyer static int quota_handle_cmd_get(struct mount *, struct lwp *,
     77  1.72  dholland     struct vfs_quotactl_args *args);
     78  1.69    bouyer static int quota_handle_cmd_set(struct mount *, struct lwp *,
     79  1.72  dholland     struct vfs_quotactl_args *args);
     80  1.69    bouyer static int quota_handle_cmd_getall(struct mount *, struct lwp *,
     81  1.72  dholland     struct vfs_quotactl_args *args);
     82  1.69    bouyer static int quota_handle_cmd_clear(struct mount *, struct lwp *,
     83  1.72  dholland     struct vfs_quotactl_args *args);
     84  1.69    bouyer static int quota_handle_cmd_quotaon(struct mount *, struct lwp *,
     85  1.72  dholland     struct vfs_quotactl_args *args);
     86  1.69    bouyer static int quota_handle_cmd_quotaoff(struct mount *, struct lwp *,
     87  1.72  dholland     struct vfs_quotactl_args *args);
     88  1.72  dholland 
     89   1.1   mycroft /*
     90  1.48   hannken  * Initialize the quota fields of an inode.
     91  1.48   hannken  */
     92  1.48   hannken void
     93  1.48   hannken ufsquota_init(struct inode *ip)
     94  1.48   hannken {
     95  1.48   hannken 	int i;
     96  1.48   hannken 
     97  1.48   hannken 	for (i = 0; i < MAXQUOTAS; i++)
     98  1.48   hannken 		ip->i_dquot[i] = NODQUOT;
     99  1.48   hannken }
    100  1.48   hannken 
    101  1.48   hannken /*
    102  1.48   hannken  * Release the quota fields from an inode.
    103  1.48   hannken  */
    104  1.48   hannken void
    105  1.48   hannken ufsquota_free(struct inode *ip)
    106  1.48   hannken {
    107  1.48   hannken 	int i;
    108  1.48   hannken 
    109  1.48   hannken 	for (i = 0; i < MAXQUOTAS; i++) {
    110  1.48   hannken 		dqrele(ITOV(ip), ip->i_dquot[i]);
    111  1.48   hannken 		ip->i_dquot[i] = NODQUOT;
    112  1.48   hannken 	}
    113  1.48   hannken }
    114  1.48   hannken 
    115  1.48   hannken /*
    116   1.1   mycroft  * Update disk usage, and take corrective action.
    117   1.1   mycroft  */
    118   1.1   mycroft int
    119  1.39      elad chkdq(struct inode *ip, int64_t change, kauth_cred_t cred, int flags)
    120   1.1   mycroft {
    121  1.69    bouyer 	/* do not track snapshot usage, or we will deadlock */
    122  1.69    bouyer 	if ((ip->i_flags & SF_SNAPSHOT) != 0)
    123  1.69    bouyer 		return 0;
    124   1.1   mycroft 
    125  1.69    bouyer #ifdef QUOTA
    126  1.69    bouyer 	if (ip->i_ump->um_flags & UFS_QUOTA)
    127  1.69    bouyer 		return chkdq1(ip, change, cred, flags);
    128  1.69    bouyer #endif
    129  1.69    bouyer #ifdef QUOTA2
    130  1.69    bouyer 	if (ip->i_ump->um_flags & UFS_QUOTA2)
    131  1.69    bouyer 		return chkdq2(ip, change, cred, flags);
    132  1.69    bouyer #endif
    133  1.69    bouyer 	return 0;
    134   1.1   mycroft }
    135   1.1   mycroft 
    136   1.1   mycroft /*
    137  1.69    bouyer  * Check the inode limit, applying corrective action.
    138   1.1   mycroft  */
    139  1.69    bouyer int
    140  1.69    bouyer chkiq(struct inode *ip, int32_t change, kauth_cred_t cred, int flags)
    141   1.1   mycroft {
    142  1.69    bouyer 	/* do not track snapshot usage, or we will deadlock */
    143  1.69    bouyer 	if ((ip->i_flags & SF_SNAPSHOT) != 0)
    144  1.69    bouyer 		return 0;
    145  1.69    bouyer #ifdef QUOTA
    146  1.69    bouyer 	if (ip->i_ump->um_flags & UFS_QUOTA)
    147  1.69    bouyer 		return chkiq1(ip, change, cred, flags);
    148  1.69    bouyer #endif
    149  1.69    bouyer #ifdef QUOTA2
    150  1.69    bouyer 	if (ip->i_ump->um_flags & UFS_QUOTA2)
    151  1.69    bouyer 		return chkiq2(ip, change, cred, flags);
    152  1.69    bouyer #endif
    153  1.69    bouyer 	return 0;
    154   1.1   mycroft }
    155   1.1   mycroft 
    156   1.1   mycroft int
    157  1.71  dholland quota_handle_cmd(struct mount *mp, struct lwp *l, int op,
    158  1.72  dholland 		 struct vfs_quotactl_args *args)
    159  1.69    bouyer {
    160  1.69    bouyer 	int error = 0;
    161  1.69    bouyer 
    162  1.71  dholland 	switch (op) {
    163  1.71  dholland 	    case QUOTACTL_GETVERSION:
    164  1.72  dholland 		error = quota_handle_cmd_get_version(mp, l, args);
    165  1.71  dholland 		break;
    166  1.71  dholland 	    case QUOTACTL_QUOTAON:
    167  1.72  dholland 		error = quota_handle_cmd_quotaon(mp, l, args);
    168  1.71  dholland 		break;
    169  1.71  dholland 	    case QUOTACTL_QUOTAOFF:
    170  1.72  dholland 		error = quota_handle_cmd_quotaoff(mp, l, args);
    171  1.71  dholland 		break;
    172  1.71  dholland 	    case QUOTACTL_GET:
    173  1.72  dholland 		error = quota_handle_cmd_get(mp, l, args);
    174  1.71  dholland 		break;
    175  1.71  dholland 	    case QUOTACTL_SET:
    176  1.72  dholland 		error = quota_handle_cmd_set(mp, l, args);
    177  1.71  dholland 		break;
    178  1.71  dholland 	    case QUOTACTL_GETALL:
    179  1.72  dholland 		error = quota_handle_cmd_getall(mp, l, args);
    180  1.71  dholland 		break;
    181  1.71  dholland 	    case QUOTACTL_CLEAR:
    182  1.72  dholland 		error = quota_handle_cmd_clear(mp, l, args);
    183  1.71  dholland 		break;
    184  1.71  dholland 	    default:
    185  1.71  dholland 		panic("Invalid quotactl operation %d\n", op);
    186  1.69    bouyer 	}
    187  1.71  dholland 
    188  1.69    bouyer 	return error;
    189  1.69    bouyer }
    190  1.69    bouyer 
    191  1.69    bouyer static int
    192  1.69    bouyer quota_handle_cmd_get_version(struct mount *mp, struct lwp *l,
    193  1.72  dholland     struct vfs_quotactl_args *args)
    194   1.1   mycroft {
    195  1.69    bouyer 	struct ufsmount *ump = VFSTOUFS(mp);
    196  1.73  dholland 	int *version_ret;
    197  1.72  dholland 
    198  1.73  dholland 	KASSERT(args->qc_type == QCT_GETVERSION);
    199  1.73  dholland 	version_ret = args->u.getversion.qc_version_ret;
    200  1.69    bouyer 
    201  1.69    bouyer 	if ((ump->um_flags & (UFS_QUOTA|UFS_QUOTA2)) == 0)
    202  1.69    bouyer 		return EOPNOTSUPP;
    203  1.69    bouyer 
    204  1.69    bouyer #ifdef QUOTA
    205  1.69    bouyer 	if (ump->um_flags & UFS_QUOTA) {
    206  1.73  dholland 		*version_ret = 1;
    207  1.69    bouyer 	} else
    208  1.69    bouyer #endif
    209  1.69    bouyer #ifdef QUOTA2
    210  1.69    bouyer 	if (ump->um_flags & UFS_QUOTA2) {
    211  1.73  dholland 		*version_ret = 2;
    212  1.69    bouyer 	} else
    213  1.69    bouyer #endif
    214  1.73  dholland 		return EOPNOTSUPP;
    215  1.73  dholland 
    216  1.73  dholland 	return 0;
    217   1.1   mycroft }
    218   1.1   mycroft 
    219  1.69    bouyer /* XXX shouldn't all this be in kauth ? */
    220  1.48   hannken static int
    221  1.69    bouyer quota_get_auth(struct mount *mp, struct lwp *l, uid_t id) {
    222  1.69    bouyer 	/* The user can always query about his own quota. */
    223  1.69    bouyer 	if (id == kauth_cred_getuid(l->l_cred))
    224  1.69    bouyer 		return 0;
    225  1.69    bouyer 	return kauth_authorize_system(l->l_cred, KAUTH_SYSTEM_FS_QUOTA,
    226  1.69    bouyer 	    KAUTH_REQ_SYSTEM_FS_QUOTA_GET, mp, KAUTH_ARG(id), NULL);
    227  1.69    bouyer }
    228  1.69    bouyer 
    229  1.69    bouyer static int
    230  1.69    bouyer quota_handle_cmd_get(struct mount *mp, struct lwp *l,
    231  1.72  dholland     struct vfs_quotactl_args *args)
    232  1.69    bouyer {
    233  1.69    bouyer 	prop_array_t replies;
    234  1.69    bouyer 	prop_object_iterator_t iter;
    235  1.69    bouyer 	prop_dictionary_t data;
    236  1.69    bouyer 	uint32_t id;
    237  1.69    bouyer 	struct ufsmount *ump = VFSTOUFS(mp);
    238  1.69    bouyer 	int error, defaultq = 0;
    239  1.69    bouyer 	const char *idstr;
    240  1.72  dholland 	prop_dictionary_t cmddict;
    241  1.72  dholland 	int q2type;
    242  1.72  dholland 	prop_array_t datas;
    243  1.72  dholland 
    244  1.72  dholland 	KASSERT(args->qc_type == QCT_PROPLIB);
    245  1.72  dholland 	cmddict = args->u.proplib.qc_cmddict;
    246  1.72  dholland 	q2type = args->u.proplib.qc_q2type;
    247  1.72  dholland 	datas = args->u.proplib.qc_datas;
    248  1.72  dholland 
    249  1.72  dholland 	KASSERT(prop_object_type(cmddict) == PROP_TYPE_DICTIONARY);
    250  1.72  dholland 	KASSERT(prop_object_type(datas) == PROP_TYPE_ARRAY);
    251  1.69    bouyer 
    252  1.69    bouyer 	if ((ump->um_flags & (UFS_QUOTA|UFS_QUOTA2)) == 0)
    253  1.69    bouyer 		return EOPNOTSUPP;
    254  1.69    bouyer 
    255  1.69    bouyer 	replies = prop_array_create();
    256  1.69    bouyer 	if (replies == NULL)
    257  1.69    bouyer 		return ENOMEM;
    258   1.1   mycroft 
    259  1.69    bouyer 	iter = prop_array_iterator(datas);
    260  1.69    bouyer 	if (iter == NULL) {
    261  1.69    bouyer 		prop_object_release(replies);
    262  1.69    bouyer 		return ENOMEM;
    263   1.1   mycroft 	}
    264  1.69    bouyer 	while ((data = prop_object_iterator_next(iter)) != NULL) {
    265  1.69    bouyer 		if (!prop_dictionary_get_uint32(data, "id", &id)) {
    266  1.69    bouyer 			if (!prop_dictionary_get_cstring_nocopy(data, "id",
    267  1.69    bouyer 			    &idstr))
    268  1.69    bouyer 				continue;
    269  1.69    bouyer 			if (strcmp(idstr, "default")) {
    270  1.69    bouyer 				error = EINVAL;
    271  1.69    bouyer 				goto err;
    272   1.1   mycroft 			}
    273  1.69    bouyer 			id = 0;
    274  1.69    bouyer 			defaultq = 1;
    275  1.69    bouyer 		} else {
    276  1.69    bouyer 			defaultq = 0;
    277   1.1   mycroft 		}
    278  1.69    bouyer 		error = quota_get_auth(mp, l, id);
    279  1.69    bouyer 		if (error == EPERM)
    280  1.69    bouyer 			continue;
    281  1.69    bouyer 		if (error != 0)
    282  1.69    bouyer 			goto err;
    283  1.69    bouyer #ifdef QUOTA
    284  1.69    bouyer 		if (ump->um_flags & UFS_QUOTA)
    285  1.72  dholland 			error = quota1_handle_cmd_get(ump, q2type, id, defaultq,
    286  1.69    bouyer 			    replies);
    287  1.69    bouyer 		else
    288  1.69    bouyer #endif
    289  1.69    bouyer #ifdef QUOTA2
    290  1.69    bouyer 		if (ump->um_flags & UFS_QUOTA2) {
    291  1.72  dholland 			error = quota2_handle_cmd_get(ump, q2type, id, defaultq,
    292  1.69    bouyer 			    replies);
    293  1.69    bouyer 		} else
    294  1.69    bouyer #endif
    295  1.69    bouyer 			panic("quota_handle_cmd_get: no support ?");
    296  1.69    bouyer 
    297  1.69    bouyer 		if (error == ENOENT)
    298  1.69    bouyer 			continue;
    299  1.69    bouyer 		if (error != 0)
    300  1.69    bouyer 			goto err;
    301  1.69    bouyer 	}
    302  1.69    bouyer 	prop_object_iterator_release(iter);
    303  1.69    bouyer 	if (!prop_dictionary_set_and_rel(cmddict, "data", replies)) {
    304  1.69    bouyer 		error = ENOMEM;
    305  1.69    bouyer 	} else {
    306  1.69    bouyer 		error = 0;
    307   1.1   mycroft 	}
    308  1.69    bouyer 	return error;
    309  1.69    bouyer err:
    310  1.69    bouyer 	prop_object_iterator_release(iter);
    311  1.69    bouyer 	prop_object_release(replies);
    312  1.69    bouyer 	return error;
    313  1.69    bouyer }
    314  1.69    bouyer 
    315  1.69    bouyer static int
    316  1.69    bouyer quota_handle_cmd_set(struct mount *mp, struct lwp *l,
    317  1.72  dholland     struct vfs_quotactl_args *args)
    318  1.69    bouyer {
    319  1.69    bouyer 	prop_array_t replies;
    320  1.69    bouyer 	prop_object_iterator_t iter;
    321  1.69    bouyer 	prop_dictionary_t data;
    322  1.69    bouyer 	uint32_t id;
    323  1.11      fvdl 	struct ufsmount *ump = VFSTOUFS(mp);
    324  1.69    bouyer 	int error, defaultq = 0;
    325  1.69    bouyer 	const char *idstr;
    326  1.72  dholland 	prop_dictionary_t cmddict;
    327  1.72  dholland 	int q2type;
    328  1.72  dholland 	prop_array_t datas;
    329  1.72  dholland 
    330  1.72  dholland 	KASSERT(args->qc_type == QCT_PROPLIB);
    331  1.72  dholland 	cmddict = args->u.proplib.qc_cmddict;
    332  1.72  dholland 	q2type = args->u.proplib.qc_q2type;
    333  1.72  dholland 	datas = args->u.proplib.qc_datas;
    334  1.72  dholland 
    335  1.72  dholland 	KASSERT(prop_object_type(cmddict) == PROP_TYPE_DICTIONARY);
    336  1.72  dholland 	KASSERT(prop_object_type(datas) == PROP_TYPE_ARRAY);
    337   1.1   mycroft 
    338  1.69    bouyer 	if ((ump->um_flags & (UFS_QUOTA|UFS_QUOTA2)) == 0)
    339  1.69    bouyer 		return EOPNOTSUPP;
    340  1.69    bouyer 
    341  1.69    bouyer 	replies = prop_array_create();
    342  1.69    bouyer 	if (replies == NULL)
    343  1.69    bouyer 		return ENOMEM;
    344  1.61        ad 
    345  1.69    bouyer 	iter = prop_array_iterator(datas);
    346  1.69    bouyer 	if (iter == NULL) {
    347  1.69    bouyer 		prop_object_release(replies);
    348  1.69    bouyer 		return ENOMEM;
    349  1.68  dholland 	}
    350  1.69    bouyer 	while ((data = prop_object_iterator_next(iter)) != NULL) {
    351  1.69    bouyer 		if (!prop_dictionary_get_uint32(data, "id", &id)) {
    352  1.69    bouyer 			if (!prop_dictionary_get_cstring_nocopy(data, "id",
    353  1.69    bouyer 			    &idstr))
    354  1.69    bouyer 				continue;
    355  1.69    bouyer 			if (strcmp(idstr, "default"))
    356  1.69    bouyer 				continue;
    357  1.69    bouyer 			id = 0;
    358  1.69    bouyer 			defaultq = 1;
    359  1.69    bouyer 		} else {
    360  1.69    bouyer 			defaultq = 0;
    361  1.69    bouyer 		}
    362  1.69    bouyer 		error = kauth_authorize_system(l->l_cred, KAUTH_SYSTEM_FS_QUOTA,
    363  1.69    bouyer 		    KAUTH_REQ_SYSTEM_FS_QUOTA_MANAGE, mp, KAUTH_ARG(id), NULL);
    364  1.69    bouyer 		if (error != 0)
    365  1.69    bouyer 			goto err;
    366  1.69    bouyer #ifdef QUOTA
    367  1.69    bouyer 		if (ump->um_flags & UFS_QUOTA)
    368  1.72  dholland 			error = quota1_handle_cmd_set(ump, q2type, id, defaultq,
    369  1.69    bouyer 			    data);
    370  1.69    bouyer 		else
    371  1.69    bouyer #endif
    372  1.69    bouyer #ifdef QUOTA2
    373  1.69    bouyer 		if (ump->um_flags & UFS_QUOTA2) {
    374  1.72  dholland 			error = quota2_handle_cmd_set(ump, q2type, id, defaultq,
    375  1.69    bouyer 			    data);
    376  1.69    bouyer 		} else
    377  1.69    bouyer #endif
    378  1.69    bouyer 			panic("quota_handle_cmd_get: no support ?");
    379  1.69    bouyer 
    380  1.69    bouyer 		if (error && error != ENOENT)
    381  1.69    bouyer 			goto err;
    382   1.1   mycroft 	}
    383  1.69    bouyer 	prop_object_iterator_release(iter);
    384  1.69    bouyer 	if (!prop_dictionary_set_and_rel(cmddict, "data", replies)) {
    385  1.53        ad 		error = ENOMEM;
    386  1.69    bouyer 	} else {
    387  1.69    bouyer 		error = 0;
    388   1.1   mycroft 	}
    389  1.69    bouyer 	return error;
    390  1.69    bouyer err:
    391  1.69    bouyer 	prop_object_iterator_release(iter);
    392  1.69    bouyer 	prop_object_release(replies);
    393  1.69    bouyer 	return error;
    394  1.69    bouyer }
    395  1.69    bouyer 
    396  1.69    bouyer static int
    397  1.69    bouyer quota_handle_cmd_clear(struct mount *mp, struct lwp *l,
    398  1.72  dholland     struct vfs_quotactl_args *args)
    399  1.69    bouyer {
    400  1.69    bouyer 	prop_array_t replies;
    401  1.69    bouyer 	prop_object_iterator_t iter;
    402  1.69    bouyer 	prop_dictionary_t data;
    403  1.69    bouyer 	uint32_t id;
    404   1.1   mycroft 	struct ufsmount *ump = VFSTOUFS(mp);
    405  1.69    bouyer 	int error, defaultq = 0;
    406  1.69    bouyer 	const char *idstr;
    407  1.72  dholland 	prop_dictionary_t cmddict;
    408  1.72  dholland 	int q2type;
    409  1.72  dholland 	prop_array_t datas;
    410  1.72  dholland 
    411  1.72  dholland 	KASSERT(args->qc_type == QCT_PROPLIB);
    412  1.72  dholland 	cmddict = args->u.proplib.qc_cmddict;
    413  1.72  dholland 	q2type = args->u.proplib.qc_q2type;
    414  1.72  dholland 	datas = args->u.proplib.qc_datas;
    415  1.72  dholland 
    416  1.72  dholland 	KASSERT(prop_object_type(cmddict) == PROP_TYPE_DICTIONARY);
    417  1.72  dholland 	KASSERT(prop_object_type(datas) == PROP_TYPE_ARRAY);
    418  1.33     perry 
    419  1.69    bouyer 	if ((ump->um_flags & UFS_QUOTA2) == 0)
    420  1.69    bouyer 		return EOPNOTSUPP;
    421  1.69    bouyer 
    422  1.69    bouyer 	replies = prop_array_create();
    423  1.69    bouyer 	if (replies == NULL)
    424  1.53        ad 		return ENOMEM;
    425  1.53        ad 
    426  1.69    bouyer 	iter = prop_array_iterator(datas);
    427  1.69    bouyer 	if (iter == NULL) {
    428  1.69    bouyer 		prop_object_release(replies);
    429  1.69    bouyer 		return ENOMEM;
    430  1.69    bouyer 	}
    431  1.69    bouyer 	while ((data = prop_object_iterator_next(iter)) != NULL) {
    432  1.69    bouyer 		if (!prop_dictionary_get_uint32(data, "id", &id)) {
    433  1.69    bouyer 			if (!prop_dictionary_get_cstring_nocopy(data, "id",
    434  1.69    bouyer 			    &idstr))
    435  1.69    bouyer 				continue;
    436  1.69    bouyer 			if (strcmp(idstr, "default"))
    437  1.69    bouyer 				continue;
    438  1.69    bouyer 			id = 0;
    439  1.69    bouyer 			defaultq = 1;
    440  1.69    bouyer 		} else {
    441  1.69    bouyer 			defaultq = 0;
    442  1.69    bouyer 		}
    443  1.69    bouyer 		error = kauth_authorize_system(l->l_cred, KAUTH_SYSTEM_FS_QUOTA,
    444  1.69    bouyer 		    KAUTH_REQ_SYSTEM_FS_QUOTA_MANAGE, mp, KAUTH_ARG(id), NULL);
    445  1.69    bouyer 		if (error != 0)
    446  1.69    bouyer 			goto err;
    447  1.69    bouyer #ifdef QUOTA2
    448  1.69    bouyer 		if (ump->um_flags & UFS_QUOTA2) {
    449  1.72  dholland 			error = quota2_handle_cmd_clear(ump, q2type, id, defaultq,
    450  1.69    bouyer 			    data);
    451  1.69    bouyer 		} else
    452  1.69    bouyer #endif
    453  1.69    bouyer 			panic("quota_handle_cmd_get: no support ?");
    454  1.69    bouyer 
    455  1.69    bouyer 		if (error && error != ENOENT)
    456  1.69    bouyer 			goto err;
    457  1.50   hannken 	}
    458  1.69    bouyer 	prop_object_iterator_release(iter);
    459  1.69    bouyer 	if (!prop_dictionary_set_and_rel(cmddict, "data", replies)) {
    460  1.69    bouyer 		error = ENOMEM;
    461  1.69    bouyer 	} else {
    462  1.69    bouyer 		error = 0;
    463   1.1   mycroft 	}
    464  1.69    bouyer 	return error;
    465  1.69    bouyer err:
    466  1.69    bouyer 	prop_object_iterator_release(iter);
    467  1.69    bouyer 	prop_object_release(replies);
    468  1.69    bouyer 	return error;
    469   1.1   mycroft }
    470   1.1   mycroft 
    471  1.69    bouyer static int
    472  1.69    bouyer quota_handle_cmd_getall(struct mount *mp, struct lwp *l,
    473  1.72  dholland     struct vfs_quotactl_args *args)
    474   1.1   mycroft {
    475  1.69    bouyer 	prop_array_t replies;
    476   1.1   mycroft 	struct ufsmount *ump = VFSTOUFS(mp);
    477   1.1   mycroft 	int error;
    478  1.72  dholland 	prop_dictionary_t cmddict;
    479  1.72  dholland 	int q2type;
    480  1.72  dholland 	prop_array_t datas;
    481  1.72  dholland 
    482  1.72  dholland 	KASSERT(args->qc_type == QCT_PROPLIB);
    483  1.72  dholland 	cmddict = args->u.proplib.qc_cmddict;
    484  1.72  dholland 	q2type = args->u.proplib.qc_q2type;
    485  1.72  dholland 	datas = args->u.proplib.qc_datas;
    486  1.72  dholland 
    487  1.72  dholland 	KASSERT(prop_object_type(cmddict) == PROP_TYPE_DICTIONARY);
    488  1.72  dholland 	KASSERT(prop_object_type(datas) == PROP_TYPE_ARRAY);
    489   1.1   mycroft 
    490  1.69    bouyer 	if ((ump->um_flags & UFS_QUOTA2) == 0)
    491  1.69    bouyer 		return EOPNOTSUPP;
    492  1.69    bouyer 
    493  1.69    bouyer 	error = kauth_authorize_system(l->l_cred, KAUTH_SYSTEM_FS_QUOTA,
    494  1.69    bouyer 	    KAUTH_REQ_SYSTEM_FS_QUOTA_GET, mp, NULL, NULL);
    495   1.8  christos 	if (error)
    496  1.69    bouyer 		return error;
    497  1.69    bouyer 
    498  1.69    bouyer 	replies = prop_array_create();
    499  1.69    bouyer 	if (replies == NULL)
    500  1.69    bouyer 		return ENOMEM;
    501  1.69    bouyer 
    502  1.69    bouyer #ifdef QUOTA2
    503  1.69    bouyer 	if (ump->um_flags & UFS_QUOTA2) {
    504  1.72  dholland 		error = quota2_handle_cmd_getall(ump, q2type, replies);
    505  1.69    bouyer 	} else
    506  1.69    bouyer #endif
    507  1.69    bouyer 		panic("quota_handle_cmd_getall: no support ?");
    508  1.69    bouyer 	if (!prop_dictionary_set_and_rel(cmddict, "data", replies)) {
    509  1.69    bouyer 		error = ENOMEM;
    510  1.69    bouyer 	} else {
    511  1.69    bouyer 		error = 0;
    512  1.69    bouyer 	}
    513  1.69    bouyer 	return error;
    514   1.1   mycroft }
    515   1.1   mycroft 
    516  1.69    bouyer static int
    517  1.69    bouyer quota_handle_cmd_quotaon(struct mount *mp, struct lwp *l,
    518  1.72  dholland     struct vfs_quotactl_args *args)
    519   1.1   mycroft {
    520  1.69    bouyer 	prop_dictionary_t data;
    521   1.1   mycroft 	struct ufsmount *ump = VFSTOUFS(mp);
    522   1.1   mycroft 	int error;
    523  1.69    bouyer 	const char *qfile;
    524  1.72  dholland 	prop_dictionary_t cmddict;
    525  1.72  dholland 	int q2type;
    526  1.72  dholland 	prop_array_t datas;
    527  1.72  dholland 
    528  1.72  dholland 	KASSERT(args->qc_type == QCT_PROPLIB);
    529  1.72  dholland 	cmddict = args->u.proplib.qc_cmddict;
    530  1.72  dholland 	q2type = args->u.proplib.qc_q2type;
    531  1.72  dholland 	datas = args->u.proplib.qc_datas;
    532  1.72  dholland 
    533  1.72  dholland 	KASSERT(prop_object_type(cmddict) == PROP_TYPE_DICTIONARY);
    534  1.72  dholland 	KASSERT(prop_object_type(datas) == PROP_TYPE_ARRAY);
    535  1.69    bouyer 
    536  1.69    bouyer 	if ((ump->um_flags & UFS_QUOTA2) != 0)
    537  1.69    bouyer 		return EBUSY;
    538  1.69    bouyer 
    539  1.69    bouyer 	if (prop_array_count(datas) != 1)
    540  1.69    bouyer 		return EINVAL;
    541   1.1   mycroft 
    542  1.69    bouyer 	data = prop_array_get(datas, 0);
    543  1.69    bouyer 	if (data == NULL)
    544  1.69    bouyer 		return ENOMEM;
    545  1.69    bouyer 	if (!prop_dictionary_get_cstring_nocopy(data, "quotafile",
    546  1.69    bouyer 	    &qfile))
    547  1.69    bouyer 		return EINVAL;
    548  1.69    bouyer 
    549  1.69    bouyer 	error = kauth_authorize_system(l->l_cred, KAUTH_SYSTEM_FS_QUOTA,
    550  1.69    bouyer 	    KAUTH_REQ_SYSTEM_FS_QUOTA_ONOFF, mp, NULL, NULL);
    551  1.69    bouyer 	if (error != 0) {
    552  1.69    bouyer 		return error;
    553  1.69    bouyer 	}
    554  1.69    bouyer #ifdef QUOTA
    555  1.72  dholland 	error = quota1_handle_cmd_quotaon(l, ump, q2type, qfile);
    556  1.69    bouyer #else
    557  1.69    bouyer 	error = EOPNOTSUPP;
    558  1.69    bouyer #endif
    559  1.69    bouyer 
    560  1.69    bouyer 	return error;
    561   1.1   mycroft }
    562   1.1   mycroft 
    563  1.69    bouyer static int
    564  1.69    bouyer quota_handle_cmd_quotaoff(struct mount *mp, struct lwp *l,
    565  1.72  dholland     struct vfs_quotactl_args *args)
    566   1.1   mycroft {
    567   1.1   mycroft 	struct ufsmount *ump = VFSTOUFS(mp);
    568  1.69    bouyer 	int error;
    569  1.72  dholland 	prop_dictionary_t cmddict;
    570  1.72  dholland 	int q2type;
    571  1.72  dholland 	prop_array_t datas;
    572  1.72  dholland 
    573  1.72  dholland 	KASSERT(args->qc_type == QCT_PROPLIB);
    574  1.72  dholland 	cmddict = args->u.proplib.qc_cmddict;
    575  1.72  dholland 	q2type = args->u.proplib.qc_q2type;
    576  1.72  dholland 	datas = args->u.proplib.qc_datas;
    577  1.72  dholland 
    578  1.72  dholland 	KASSERT(prop_object_type(cmddict) == PROP_TYPE_DICTIONARY);
    579  1.72  dholland 	KASSERT(prop_object_type(datas) == PROP_TYPE_ARRAY);
    580   1.1   mycroft 
    581  1.69    bouyer 	if ((ump->um_flags & UFS_QUOTA2) != 0)
    582  1.69    bouyer 		return EOPNOTSUPP;
    583  1.69    bouyer 
    584  1.69    bouyer 	if (prop_array_count(datas) != 0)
    585  1.69    bouyer 		return EINVAL;
    586  1.69    bouyer 
    587  1.69    bouyer 	error = kauth_authorize_system(l->l_cred, KAUTH_SYSTEM_FS_QUOTA,
    588  1.69    bouyer 	    KAUTH_REQ_SYSTEM_FS_QUOTA_ONOFF, mp, NULL, NULL);
    589  1.69    bouyer 	if (error != 0) {
    590  1.69    bouyer 		return error;
    591   1.1   mycroft 	}
    592  1.69    bouyer #ifdef QUOTA
    593  1.72  dholland 	error = quota1_handle_cmd_quotaoff(l, ump, q2type);
    594  1.69    bouyer #else
    595  1.69    bouyer 	error = EOPNOTSUPP;
    596  1.69    bouyer #endif
    597  1.69    bouyer 
    598  1.69    bouyer 	return error;
    599   1.1   mycroft }
    600   1.1   mycroft 
    601   1.1   mycroft /*
    602   1.1   mycroft  * Initialize the quota system.
    603   1.1   mycroft  */
    604   1.1   mycroft void
    605  1.35   thorpej dqinit(void)
    606   1.1   mycroft {
    607  1.47     pooka 
    608  1.50   hannken 	mutex_init(&dqlock, MUTEX_DEFAULT, IPL_NONE);
    609  1.50   hannken 	cv_init(&dqcv, "quota");
    610  1.60        ad 	dqhashtbl = hashinit(desiredvnodes, HASH_LIST, true, &dqhash);
    611  1.55        ad 	dquot_cache = pool_cache_init(sizeof(struct dquot), 0, 0, 0, "ufsdq",
    612  1.55        ad 	    NULL, IPL_NONE, NULL, NULL, NULL);
    613  1.14  jdolecek }
    614  1.14  jdolecek 
    615  1.21       chs void
    616  1.35   thorpej dqreinit(void)
    617  1.21       chs {
    618  1.21       chs 	struct dquot *dq;
    619  1.21       chs 	struct dqhashhead *oldhash, *hash;
    620  1.21       chs 	struct vnode *dqvp;
    621  1.21       chs 	u_long oldmask, mask, hashval;
    622  1.21       chs 	int i;
    623  1.21       chs 
    624  1.60        ad 	hash = hashinit(desiredvnodes, HASH_LIST, true, &mask);
    625  1.50   hannken 	mutex_enter(&dqlock);
    626  1.21       chs 	oldhash = dqhashtbl;
    627  1.21       chs 	oldmask = dqhash;
    628  1.21       chs 	dqhashtbl = hash;
    629  1.21       chs 	dqhash = mask;
    630  1.21       chs 	for (i = 0; i <= oldmask; i++) {
    631  1.21       chs 		while ((dq = LIST_FIRST(&oldhash[i])) != NULL) {
    632  1.21       chs 			dqvp = dq->dq_ump->um_quotas[dq->dq_type];
    633  1.21       chs 			LIST_REMOVE(dq, dq_hash);
    634  1.21       chs 			hashval = DQHASH(dqvp, dq->dq_id);
    635  1.21       chs 			LIST_INSERT_HEAD(&dqhashtbl[hashval], dq, dq_hash);
    636  1.21       chs 		}
    637  1.21       chs 	}
    638  1.50   hannken 	mutex_exit(&dqlock);
    639  1.60        ad 	hashdone(oldhash, HASH_LIST, oldmask);
    640  1.21       chs }
    641  1.21       chs 
    642  1.14  jdolecek /*
    643  1.14  jdolecek  * Free resources held by quota system.
    644  1.14  jdolecek  */
    645  1.14  jdolecek void
    646  1.35   thorpej dqdone(void)
    647  1.14  jdolecek {
    648  1.47     pooka 
    649  1.55        ad 	pool_cache_destroy(dquot_cache);
    650  1.60        ad 	hashdone(dqhashtbl, HASH_LIST, dqhash);
    651  1.50   hannken 	cv_destroy(&dqcv);
    652  1.50   hannken 	mutex_destroy(&dqlock);
    653   1.1   mycroft }
    654   1.1   mycroft 
    655   1.1   mycroft /*
    656  1.69    bouyer  * Set up the quotas for an inode.
    657  1.69    bouyer  *
    658  1.69    bouyer  * This routine completely defines the semantics of quotas.
    659  1.69    bouyer  * If other criterion want to be used to establish quotas, the
    660  1.69    bouyer  * MAXQUOTAS value in quotas.h should be increased, and the
    661  1.69    bouyer  * additional dquots set up here.
    662  1.69    bouyer  */
    663  1.69    bouyer int
    664  1.69    bouyer getinoquota(struct inode *ip)
    665  1.69    bouyer {
    666  1.69    bouyer 	struct ufsmount *ump = ip->i_ump;
    667  1.69    bouyer 	struct vnode *vp = ITOV(ip);
    668  1.69    bouyer 	int i, error;
    669  1.69    bouyer 	u_int32_t ino_ids[MAXQUOTAS];
    670  1.69    bouyer 
    671  1.69    bouyer 	/*
    672  1.69    bouyer 	 * To avoid deadlocks never update quotas for quota files
    673  1.69    bouyer 	 * on the same file system
    674  1.69    bouyer 	 */
    675  1.69    bouyer 	for (i = 0; i < MAXQUOTAS; i++)
    676  1.69    bouyer 		if (vp == ump->um_quotas[i])
    677  1.69    bouyer 			return 0;
    678  1.69    bouyer 
    679  1.69    bouyer 	ino_ids[USRQUOTA] = ip->i_uid;
    680  1.69    bouyer 	ino_ids[GRPQUOTA] = ip->i_gid;
    681  1.69    bouyer 	for (i = 0; i < MAXQUOTAS; i++) {
    682  1.69    bouyer 		/*
    683  1.69    bouyer 		 * If the file id changed the quota needs update.
    684  1.69    bouyer 		 */
    685  1.69    bouyer 		if (ip->i_dquot[i] != NODQUOT &&
    686  1.69    bouyer 		    ip->i_dquot[i]->dq_id != ino_ids[i]) {
    687  1.69    bouyer 			dqrele(ITOV(ip), ip->i_dquot[i]);
    688  1.69    bouyer 			ip->i_dquot[i] = NODQUOT;
    689  1.69    bouyer 		}
    690  1.69    bouyer 		/*
    691  1.69    bouyer 		 * Set up the quota based on file id.
    692  1.69    bouyer 		 * ENODEV means that quotas are not enabled.
    693  1.69    bouyer 		 */
    694  1.69    bouyer 		if (ip->i_dquot[i] == NODQUOT &&
    695  1.69    bouyer 		    (error = dqget(vp, ino_ids[i], ump, i, &ip->i_dquot[i])) &&
    696  1.69    bouyer 		    error != ENODEV)
    697  1.69    bouyer 			return (error);
    698  1.69    bouyer 	}
    699  1.69    bouyer 	return 0;
    700  1.69    bouyer }
    701  1.69    bouyer 
    702  1.69    bouyer /*
    703   1.1   mycroft  * Obtain a dquot structure for the specified identifier and quota file
    704   1.1   mycroft  * reading the information from the file if necessary.
    705   1.1   mycroft  */
    706  1.69    bouyer int
    707  1.35   thorpej dqget(struct vnode *vp, u_long id, struct ufsmount *ump, int type,
    708  1.35   thorpej     struct dquot **dqp)
    709   1.1   mycroft {
    710  1.50   hannken 	struct dquot *dq, *ndq;
    711  1.21       chs 	struct dqhashhead *dqh;
    712  1.11      fvdl 	struct vnode *dqvp;
    713  1.69    bouyer 	int error = 0; /* XXX gcc */
    714   1.1   mycroft 
    715  1.50   hannken 	/* Lock to see an up to date value for QTF_CLOSING. */
    716  1.50   hannken 	mutex_enter(&dqlock);
    717  1.69    bouyer 	if ((ump->um_flags & (UFS_QUOTA|UFS_QUOTA2)) == 0) {
    718  1.50   hannken 		mutex_exit(&dqlock);
    719   1.1   mycroft 		*dqp = NODQUOT;
    720  1.69    bouyer 		return (ENODEV);
    721  1.69    bouyer 	}
    722  1.69    bouyer 	dqvp = ump->um_quotas[type];
    723  1.69    bouyer #ifdef QUOTA
    724  1.69    bouyer 	if (ump->um_flags & UFS_QUOTA) {
    725  1.69    bouyer 		if (dqvp == NULLVP || (ump->umq1_qflags[type] & QTF_CLOSING)) {
    726  1.69    bouyer 			mutex_exit(&dqlock);
    727  1.69    bouyer 			*dqp = NODQUOT;
    728  1.69    bouyer 			return (ENODEV);
    729  1.69    bouyer 		}
    730   1.1   mycroft 	}
    731  1.69    bouyer #endif
    732  1.69    bouyer #ifdef QUOTA2
    733  1.69    bouyer 	if (ump->um_flags & UFS_QUOTA2) {
    734  1.69    bouyer 		if (dqvp == NULLVP) {
    735  1.69    bouyer 			mutex_exit(&dqlock);
    736  1.69    bouyer 			*dqp = NODQUOT;
    737  1.69    bouyer 			return (ENODEV);
    738  1.69    bouyer 		}
    739  1.69    bouyer 	}
    740  1.69    bouyer #endif
    741  1.50   hannken 	KASSERT(dqvp != vp);
    742   1.1   mycroft 	/*
    743   1.1   mycroft 	 * Check the cache first.
    744   1.1   mycroft 	 */
    745  1.21       chs 	dqh = &dqhashtbl[DQHASH(dqvp, id)];
    746  1.21       chs 	LIST_FOREACH(dq, dqh, dq_hash) {
    747   1.1   mycroft 		if (dq->dq_id != id ||
    748   1.1   mycroft 		    dq->dq_ump->um_quotas[dq->dq_type] != dqvp)
    749   1.1   mycroft 			continue;
    750  1.50   hannken 		KASSERT(dq->dq_cnt > 0);
    751  1.37       chs 		dqref(dq);
    752  1.50   hannken 		mutex_exit(&dqlock);
    753   1.1   mycroft 		*dqp = dq;
    754   1.1   mycroft 		return (0);
    755   1.1   mycroft 	}
    756   1.1   mycroft 	/*
    757   1.1   mycroft 	 * Not in cache, allocate a new one.
    758   1.1   mycroft 	 */
    759  1.50   hannken 	mutex_exit(&dqlock);
    760  1.55        ad 	ndq = pool_cache_get(dquot_cache, PR_WAITOK);
    761   1.1   mycroft 	/*
    762   1.1   mycroft 	 * Initialize the contents of the dquot structure.
    763   1.1   mycroft 	 */
    764  1.50   hannken 	memset((char *)ndq, 0, sizeof *ndq);
    765  1.50   hannken 	ndq->dq_flags = 0;
    766  1.50   hannken 	ndq->dq_id = id;
    767  1.50   hannken 	ndq->dq_ump = ump;
    768  1.50   hannken 	ndq->dq_type = type;
    769  1.50   hannken 	mutex_init(&ndq->dq_interlock, MUTEX_DEFAULT, IPL_NONE);
    770  1.50   hannken 	mutex_enter(&dqlock);
    771  1.49   hannken 	dqh = &dqhashtbl[DQHASH(dqvp, id)];
    772  1.50   hannken 	LIST_FOREACH(dq, dqh, dq_hash) {
    773  1.50   hannken 		if (dq->dq_id != id ||
    774  1.50   hannken 		    dq->dq_ump->um_quotas[dq->dq_type] != dqvp)
    775  1.50   hannken 			continue;
    776  1.50   hannken 		/*
    777  1.50   hannken 		 * Another thread beat us allocating this dquot.
    778  1.50   hannken 		 */
    779  1.50   hannken 		KASSERT(dq->dq_cnt > 0);
    780  1.50   hannken 		dqref(dq);
    781  1.50   hannken 		mutex_exit(&dqlock);
    782  1.64    bouyer 		mutex_destroy(&ndq->dq_interlock);
    783  1.55        ad 		pool_cache_put(dquot_cache, ndq);
    784  1.50   hannken 		*dqp = dq;
    785  1.50   hannken 		return 0;
    786  1.50   hannken 	}
    787  1.50   hannken 	dq = ndq;
    788   1.5   mycroft 	LIST_INSERT_HEAD(dqh, dq, dq_hash);
    789  1.37       chs 	dqref(dq);
    790  1.50   hannken 	mutex_enter(&dq->dq_interlock);
    791  1.50   hannken 	mutex_exit(&dqlock);
    792  1.69    bouyer #ifdef QUOTA
    793  1.69    bouyer 	if (ump->um_flags & UFS_QUOTA)
    794  1.69    bouyer 		error = dq1get(dqvp, id, ump, type, dq);
    795  1.69    bouyer #endif
    796  1.69    bouyer #ifdef QUOTA2
    797  1.69    bouyer 	if (ump->um_flags & UFS_QUOTA2)
    798  1.69    bouyer 		error = dq2get(dqvp, id, ump, type, dq);
    799  1.69    bouyer #endif
    800   1.1   mycroft 	/*
    801   1.1   mycroft 	 * I/O error in reading quota file, release
    802   1.1   mycroft 	 * quota structure and reflect problem to caller.
    803   1.1   mycroft 	 */
    804   1.1   mycroft 	if (error) {
    805  1.50   hannken 		mutex_enter(&dqlock);
    806   1.5   mycroft 		LIST_REMOVE(dq, dq_hash);
    807  1.50   hannken 		mutex_exit(&dqlock);
    808  1.49   hannken 		mutex_exit(&dq->dq_interlock);
    809   1.1   mycroft 		dqrele(vp, dq);
    810   1.1   mycroft 		*dqp = NODQUOT;
    811   1.1   mycroft 		return (error);
    812   1.1   mycroft 	}
    813  1.49   hannken 	mutex_exit(&dq->dq_interlock);
    814   1.1   mycroft 	*dqp = dq;
    815   1.1   mycroft 	return (0);
    816   1.1   mycroft }
    817   1.1   mycroft 
    818   1.1   mycroft /*
    819   1.1   mycroft  * Obtain a reference to a dquot.
    820   1.1   mycroft  */
    821  1.69    bouyer void
    822  1.35   thorpej dqref(struct dquot *dq)
    823   1.1   mycroft {
    824   1.1   mycroft 
    825  1.50   hannken 	KASSERT(mutex_owned(&dqlock));
    826   1.1   mycroft 	dq->dq_cnt++;
    827  1.46   hannken 	KASSERT(dq->dq_cnt > 0);
    828   1.1   mycroft }
    829   1.1   mycroft 
    830   1.1   mycroft /*
    831   1.1   mycroft  * Release a reference to a dquot.
    832   1.1   mycroft  */
    833  1.69    bouyer void
    834  1.35   thorpej dqrele(struct vnode *vp, struct dquot *dq)
    835   1.1   mycroft {
    836   1.1   mycroft 
    837   1.1   mycroft 	if (dq == NODQUOT)
    838   1.1   mycroft 		return;
    839  1.50   hannken 	mutex_enter(&dq->dq_interlock);
    840  1.50   hannken 	for (;;) {
    841  1.50   hannken 		mutex_enter(&dqlock);
    842  1.50   hannken 		if (dq->dq_cnt > 1) {
    843  1.50   hannken 			dq->dq_cnt--;
    844  1.50   hannken 			mutex_exit(&dqlock);
    845  1.50   hannken 			mutex_exit(&dq->dq_interlock);
    846  1.50   hannken 			return;
    847  1.50   hannken 		}
    848  1.50   hannken 		if ((dq->dq_flags & DQ_MOD) == 0)
    849  1.50   hannken 			break;
    850  1.50   hannken 		mutex_exit(&dqlock);
    851  1.69    bouyer #ifdef QUOTA
    852  1.69    bouyer 		if (dq->dq_ump->um_flags & UFS_QUOTA)
    853  1.69    bouyer 			(void) dq1sync(vp, dq);
    854  1.69    bouyer #endif
    855  1.69    bouyer #ifdef QUOTA2
    856  1.69    bouyer 		if (dq->dq_ump->um_flags & UFS_QUOTA2)
    857  1.69    bouyer 			(void) dq2sync(vp, dq);
    858  1.69    bouyer #endif
    859   1.1   mycroft 	}
    860  1.50   hannken 	KASSERT(dq->dq_cnt == 1 && (dq->dq_flags & DQ_MOD) == 0);
    861  1.50   hannken 	LIST_REMOVE(dq, dq_hash);
    862  1.50   hannken 	mutex_exit(&dqlock);
    863  1.50   hannken 	mutex_exit(&dq->dq_interlock);
    864  1.50   hannken 	mutex_destroy(&dq->dq_interlock);
    865  1.55        ad 	pool_cache_put(dquot_cache, dq);
    866   1.1   mycroft }
    867   1.1   mycroft 
    868  1.69    bouyer int
    869  1.69    bouyer qsync(struct mount *mp)
    870   1.1   mycroft {
    871  1.69    bouyer 	struct ufsmount *ump = VFSTOUFS(mp);
    872  1.69    bouyer #ifdef QUOTA
    873  1.69    bouyer 	if (ump->um_flags & UFS_QUOTA)
    874  1.69    bouyer 		return q1sync(mp);
    875  1.69    bouyer #endif
    876  1.69    bouyer #ifdef QUOTA2
    877  1.69    bouyer 	if (ump->um_flags & UFS_QUOTA2)
    878  1.69    bouyer 		return q2sync(mp);
    879  1.69    bouyer #endif
    880  1.69    bouyer 	return 0;
    881   1.1   mycroft }
    882   1.1   mycroft 
    883  1.50   hannken #ifdef DIAGNOSTIC
    884   1.1   mycroft /*
    885  1.50   hannken  * Check the hash chains for stray dquot's.
    886   1.1   mycroft  */
    887  1.69    bouyer void
    888  1.35   thorpej dqflush(struct vnode *vp)
    889   1.1   mycroft {
    890  1.50   hannken 	struct dquot *dq;
    891  1.50   hannken 	int i;
    892   1.1   mycroft 
    893  1.50   hannken 	mutex_enter(&dqlock);
    894  1.50   hannken 	for (i = 0; i <= dqhash; i++)
    895  1.50   hannken 		LIST_FOREACH(dq, &dqhashtbl[i], dq_hash)
    896  1.50   hannken 			KASSERT(dq->dq_ump->um_quotas[dq->dq_type] != vp);
    897  1.50   hannken 	mutex_exit(&dqlock);
    898   1.1   mycroft }
    899  1.50   hannken #endif
    900