Home | History | Annotate | Line # | Download | only in kern
vfs_quotactl.c revision 1.29
      1  1.29  dholland /*	$NetBSD: vfs_quotactl.c,v 1.29 2012/01/29 07:09:52 dholland Exp $	*/
      2   1.3  dholland 
      3   1.3  dholland /*
      4   1.3  dholland  * Copyright (c) 1991, 1993, 1994
      5   1.3  dholland  *	The Regents of the University of California.  All rights reserved.
      6   1.3  dholland  * (c) UNIX System Laboratories, Inc.
      7   1.3  dholland  * All or some portions of this file are derived from material licensed
      8   1.3  dholland  * to the University of California by American Telephone and Telegraph
      9   1.3  dholland  * Co. or Unix System Laboratories, Inc. and are reproduced herein with
     10   1.3  dholland  * the permission of UNIX System Laboratories, Inc.
     11   1.3  dholland  *
     12   1.3  dholland  * Redistribution and use in source and binary forms, with or without
     13   1.3  dholland  * modification, are permitted provided that the following conditions
     14   1.3  dholland  * are met:
     15   1.3  dholland  * 1. Redistributions of source code must retain the above copyright
     16   1.3  dholland  *    notice, this list of conditions and the following disclaimer.
     17   1.3  dholland  * 2. Redistributions in binary form must reproduce the above copyright
     18   1.3  dholland  *    notice, this list of conditions and the following disclaimer in the
     19   1.3  dholland  *    documentation and/or other materials provided with the distribution.
     20   1.3  dholland  * 3. Neither the name of the University nor the names of its contributors
     21   1.3  dholland  *    may be used to endorse or promote products derived from this software
     22   1.3  dholland  *    without specific prior written permission.
     23   1.3  dholland  *
     24   1.3  dholland  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
     25   1.3  dholland  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
     26   1.3  dholland  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
     27   1.3  dholland  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
     28   1.3  dholland  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
     29   1.3  dholland  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
     30   1.3  dholland  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
     31   1.3  dholland  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
     32   1.3  dholland  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
     33   1.3  dholland  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
     34   1.3  dholland  * SUCH DAMAGE.
     35   1.3  dholland  *
     36   1.3  dholland  *	@(#)ufs_vfsops.c	8.8 (Berkeley) 5/20/95
     37   1.3  dholland  *	From NetBSD: ufs_vfsops.c,v 1.42 2011/03/24 17:05:46 bouyer Exp
     38   1.3  dholland  */
     39   1.1  dholland 
     40   1.1  dholland /*
     41   1.1  dholland  * Copyright (c) 1982, 1986, 1990, 1993, 1995
     42   1.1  dholland  *	The Regents of the University of California.  All rights reserved.
     43   1.1  dholland  *
     44   1.1  dholland  * This code is derived from software contributed to Berkeley by
     45   1.1  dholland  * Robert Elz at The University of Melbourne.
     46   1.1  dholland  *
     47   1.1  dholland  * Redistribution and use in source and binary forms, with or without
     48   1.1  dholland  * modification, are permitted provided that the following conditions
     49   1.1  dholland  * are met:
     50   1.1  dholland  * 1. Redistributions of source code must retain the above copyright
     51   1.1  dholland  *    notice, this list of conditions and the following disclaimer.
     52   1.1  dholland  * 2. Redistributions in binary form must reproduce the above copyright
     53   1.1  dholland  *    notice, this list of conditions and the following disclaimer in the
     54   1.1  dholland  *    documentation and/or other materials provided with the distribution.
     55   1.1  dholland  * 3. Neither the name of the University nor the names of its contributors
     56   1.1  dholland  *    may be used to endorse or promote products derived from this software
     57   1.1  dholland  *    without specific prior written permission.
     58   1.1  dholland  *
     59   1.1  dholland  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
     60   1.1  dholland  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
     61   1.1  dholland  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
     62   1.1  dholland  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
     63   1.1  dholland  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
     64   1.1  dholland  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
     65   1.1  dholland  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
     66   1.1  dholland  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
     67   1.1  dholland  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
     68   1.1  dholland  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
     69   1.1  dholland  * SUCH DAMAGE.
     70   1.1  dholland  *
     71   1.1  dholland  *	@(#)ufs_quota.c	8.5 (Berkeley) 5/20/95
     72   1.3  dholland  *	From NetBSD: ufs_quota.c,v 1.70 2011/03/24 17:05:46 bouyer Exp
     73   1.3  dholland  */
     74   1.3  dholland 
     75   1.3  dholland /*
     76   1.3  dholland  * Note that both of the copyrights above are moderately spurious;
     77   1.3  dholland  * this code should almost certainly have the Copyright 2010 Manuel
     78   1.3  dholland  * Bouyer notice and license found in e.g. sys/ufs/ufs/quota2_subr.c.
     79   1.3  dholland  * However, they're what was on the files this code was sliced out of.
     80   1.1  dholland  */
     81   1.1  dholland 
     82   1.1  dholland #include <sys/cdefs.h>
     83  1.29  dholland __KERNEL_RCSID(0, "$NetBSD: vfs_quotactl.c,v 1.29 2012/01/29 07:09:52 dholland Exp $");
     84   1.1  dholland 
     85  1.23  dholland #include <sys/malloc.h> /* XXX: temporary */
     86   1.2  dholland #include <sys/mount.h>
     87   1.8  dholland #include <sys/quota.h>
     88   1.4  dholland #include <sys/quotactl.h>
     89   1.3  dholland #include <quota/quotaprop.h>
     90   1.2  dholland 
     91   1.4  dholland static int
     92   1.5  dholland vfs_quotactl_getversion(struct mount *mp,
     93   1.5  dholland 			prop_dictionary_t cmddict, int q2type,
     94   1.5  dholland 			prop_array_t datas)
     95   1.5  dholland {
     96   1.6  dholland 	prop_array_t replies;
     97   1.6  dholland 	prop_dictionary_t data;
     98   1.6  dholland 	int q2version;
     99   1.5  dholland 	struct vfs_quotactl_args args;
    100   1.6  dholland 	int error;
    101   1.6  dholland 
    102   1.6  dholland 	KASSERT(prop_object_type(cmddict) == PROP_TYPE_DICTIONARY);
    103   1.6  dholland 	KASSERT(prop_object_type(datas) == PROP_TYPE_ARRAY);
    104   1.6  dholland 
    105   1.6  dholland 	args.qc_type = QCT_GETVERSION;
    106   1.6  dholland 	args.u.getversion.qc_version_ret = &q2version;
    107   1.6  dholland 	error = VFS_QUOTACTL(mp, QUOTACTL_GETVERSION, &args);
    108   1.6  dholland 	if (error) {
    109   1.6  dholland 		return error;
    110   1.6  dholland 	}
    111   1.6  dholland 
    112   1.6  dholland 	data = prop_dictionary_create();
    113   1.6  dholland 	if (data == NULL) {
    114   1.6  dholland 		return ENOMEM;
    115   1.6  dholland 	}
    116   1.6  dholland 
    117   1.6  dholland 	if (!prop_dictionary_set_int8(data, "version", q2version)) {
    118   1.6  dholland 		prop_object_release(data);
    119   1.6  dholland 		return ENOMEM;
    120   1.6  dholland 	}
    121   1.6  dholland 
    122   1.6  dholland 	replies = prop_array_create();
    123   1.6  dholland 	if (replies == NULL) {
    124   1.6  dholland 		prop_object_release(data);
    125   1.6  dholland 		return ENOMEM;
    126   1.6  dholland 	}
    127   1.5  dholland 
    128   1.6  dholland 	if (!prop_array_add_and_rel(replies, data)) {
    129   1.6  dholland 		prop_object_release(data);
    130   1.6  dholland 		prop_object_release(replies);
    131   1.6  dholland 		return ENOMEM;
    132   1.6  dholland 	}
    133   1.6  dholland 
    134   1.6  dholland 	if (!prop_dictionary_set_and_rel(cmddict, "data", replies)) {
    135   1.6  dholland 		prop_object_release(replies);
    136   1.6  dholland 		return ENOMEM;
    137   1.6  dholland 	}
    138   1.6  dholland 
    139   1.6  dholland 	return error;
    140   1.5  dholland }
    141   1.5  dholland 
    142   1.5  dholland static int
    143   1.5  dholland vfs_quotactl_quotaon(struct mount *mp,
    144   1.5  dholland 		     prop_dictionary_t cmddict, int q2type,
    145   1.5  dholland 		     prop_array_t datas)
    146   1.5  dholland {
    147   1.5  dholland 	struct vfs_quotactl_args args;
    148   1.5  dholland 
    149   1.5  dholland 	args.qc_type = QCT_PROPLIB;
    150   1.5  dholland 	args.u.proplib.qc_cmddict = cmddict;
    151   1.5  dholland 	args.u.proplib.qc_q2type = q2type;
    152   1.5  dholland 	args.u.proplib.qc_datas = datas;
    153   1.5  dholland 	return VFS_QUOTACTL(mp, QUOTACTL_QUOTAON, &args);
    154   1.5  dholland }
    155   1.5  dholland 
    156   1.5  dholland static int
    157   1.5  dholland vfs_quotactl_quotaoff(struct mount *mp,
    158   1.5  dholland 			prop_dictionary_t cmddict, int q2type,
    159   1.5  dholland 			prop_array_t datas)
    160   1.5  dholland {
    161   1.5  dholland 	struct vfs_quotactl_args args;
    162   1.5  dholland 
    163   1.5  dholland 	args.qc_type = QCT_PROPLIB;
    164   1.5  dholland 	args.u.proplib.qc_cmddict = cmddict;
    165   1.5  dholland 	args.u.proplib.qc_q2type = q2type;
    166   1.5  dholland 	args.u.proplib.qc_datas = datas;
    167   1.5  dholland 	return VFS_QUOTACTL(mp, QUOTACTL_QUOTAOFF, &args);
    168   1.5  dholland }
    169   1.5  dholland 
    170   1.5  dholland static int
    171  1.10  dholland vfs_quotactl_get_addreply(const struct quotakey *qk,
    172   1.8  dholland 			  const struct quotaval *blocks,
    173   1.8  dholland 			  const struct quotaval *files,
    174   1.8  dholland 			  prop_array_t replies)
    175   1.8  dholland {
    176   1.8  dholland 	prop_dictionary_t dict;
    177  1.10  dholland 	id_t id;
    178  1.10  dholland 	int defaultq;
    179  1.10  dholland 	uint64_t *valuesp[QUOTA_NLIMITS];
    180   1.8  dholland 
    181   1.8  dholland 	/* XXX illegal casts */
    182   1.8  dholland 	valuesp[QUOTA_LIMIT_BLOCK] = (void *)(intptr_t)&blocks->qv_hardlimit;
    183   1.8  dholland 	valuesp[QUOTA_LIMIT_FILE] =  (void *)(intptr_t)&files->qv_hardlimit;
    184   1.8  dholland 
    185  1.10  dholland 	if (qk->qk_id == QUOTA_DEFAULTID) {
    186  1.10  dholland 		id = 0;
    187  1.10  dholland 		defaultq = 1;
    188  1.10  dholland 	} else {
    189  1.10  dholland 		id = qk->qk_id;
    190  1.10  dholland 		defaultq = 0;
    191  1.10  dholland 	}
    192  1.10  dholland 
    193   1.8  dholland 	dict = quota64toprop(id, defaultq, valuesp,
    194   1.8  dholland 	    ufs_quota_entry_names, UFS_QUOTA_NENTRIES,
    195   1.8  dholland 	    ufs_quota_limit_names, QUOTA_NLIMITS);
    196   1.8  dholland 	if (dict == NULL)
    197   1.8  dholland 		return ENOMEM;
    198   1.8  dholland 	if (!prop_array_add_and_rel(replies, dict)) {
    199   1.8  dholland 		prop_object_release(dict);
    200   1.8  dholland 		return ENOMEM;
    201   1.8  dholland 	}
    202   1.8  dholland 
    203   1.8  dholland 	return 0;
    204   1.8  dholland }
    205   1.8  dholland 
    206   1.8  dholland static int
    207   1.5  dholland vfs_quotactl_get(struct mount *mp,
    208  1.10  dholland 			prop_dictionary_t cmddict, int idtype,
    209   1.5  dholland 			prop_array_t datas)
    210   1.5  dholland {
    211   1.7  dholland 	prop_object_iterator_t iter;
    212   1.7  dholland 	prop_dictionary_t data;
    213   1.8  dholland 	prop_array_t replies;
    214   1.7  dholland 	uint32_t id;
    215   1.7  dholland 	const char *idstr;
    216   1.5  dholland 	struct vfs_quotactl_args args;
    217  1.10  dholland 	struct quotakey qk;
    218   1.8  dholland 	struct quotaval blocks, files;
    219   1.7  dholland 	int error;
    220   1.7  dholland 
    221   1.7  dholland 	KASSERT(prop_object_type(cmddict) == PROP_TYPE_DICTIONARY);
    222   1.7  dholland 	KASSERT(prop_object_type(datas) == PROP_TYPE_ARRAY);
    223   1.7  dholland 
    224   1.7  dholland 	replies = prop_array_create();
    225   1.7  dholland 	if (replies == NULL) {
    226   1.7  dholland 		return ENOMEM;
    227   1.7  dholland 	}
    228   1.7  dholland 
    229   1.7  dholland 	iter = prop_array_iterator(datas);
    230   1.7  dholland 	if (iter == NULL) {
    231   1.7  dholland 		prop_object_release(replies);
    232   1.7  dholland 		return ENOMEM;
    233   1.7  dholland 	}
    234   1.7  dholland 
    235   1.7  dholland 	while ((data = prop_object_iterator_next(iter)) != NULL) {
    236  1.10  dholland 		qk.qk_idtype = idtype;
    237  1.10  dholland 
    238   1.7  dholland 		if (!prop_dictionary_get_uint32(data, "id", &id)) {
    239   1.7  dholland 			if (!prop_dictionary_get_cstring_nocopy(data, "id",
    240   1.7  dholland 			    &idstr))
    241   1.7  dholland 				continue;
    242   1.7  dholland 			if (strcmp(idstr, "default")) {
    243   1.7  dholland 				error = EINVAL;
    244   1.7  dholland 				goto fail;
    245   1.7  dholland 			}
    246  1.10  dholland 			qk.qk_id = QUOTA_DEFAULTID;
    247   1.7  dholland 		} else {
    248  1.10  dholland 			qk.qk_id = id;
    249   1.7  dholland 		}
    250   1.5  dholland 
    251  1.10  dholland 		qk.qk_objtype = QUOTA_OBJTYPE_BLOCKS;
    252  1.10  dholland 
    253   1.7  dholland 		args.qc_type = QCT_GET;
    254  1.10  dholland 		args.u.get.qc_key = &qk;
    255   1.9  dholland 		args.u.get.qc_ret = &blocks;
    256   1.9  dholland 		error = VFS_QUOTACTL(mp, QUOTACTL_GET, &args);
    257   1.9  dholland 		if (error == EPERM) {
    258   1.9  dholland 			/* XXX does this make sense? */
    259   1.9  dholland 			continue;
    260   1.9  dholland 		} else if (error == ENOENT) {
    261   1.9  dholland 			/* XXX does *this* make sense? */
    262   1.9  dholland 			continue;
    263   1.9  dholland 		} else if (error) {
    264   1.9  dholland 			goto fail;
    265   1.9  dholland 		}
    266   1.9  dholland 
    267  1.10  dholland 		qk.qk_objtype = QUOTA_OBJTYPE_FILES;
    268  1.10  dholland 
    269   1.9  dholland 		args.qc_type = QCT_GET;
    270  1.10  dholland 		args.u.get.qc_key = &qk;
    271   1.9  dholland 		args.u.get.qc_ret = &files;
    272   1.7  dholland 		error = VFS_QUOTACTL(mp, QUOTACTL_GET, &args);
    273   1.7  dholland 		if (error == EPERM) {
    274   1.7  dholland 			/* XXX does this make sense? */
    275   1.7  dholland 			continue;
    276   1.7  dholland 		} else if (error == ENOENT) {
    277   1.7  dholland 			/* XXX does *this* make sense? */
    278   1.7  dholland 			continue;
    279   1.7  dholland 		} else if (error) {
    280   1.7  dholland 			goto fail;
    281   1.7  dholland 		}
    282   1.8  dholland 
    283  1.10  dholland 		error = vfs_quotactl_get_addreply(&qk, &blocks, &files,
    284   1.8  dholland 						  replies);
    285   1.7  dholland 	}
    286   1.7  dholland 
    287   1.7  dholland 	prop_object_iterator_release(iter);
    288   1.7  dholland 	if (!prop_dictionary_set_and_rel(cmddict, "data", replies)) {
    289   1.7  dholland 		error = ENOMEM;
    290   1.7  dholland 	} else {
    291   1.7  dholland 		error = 0;
    292   1.7  dholland 	}
    293   1.7  dholland 
    294   1.7  dholland 	return error;
    295   1.7  dholland 
    296   1.7  dholland  fail:
    297   1.7  dholland 	prop_object_iterator_release(iter);
    298   1.7  dholland 	prop_object_release(replies);
    299   1.7  dholland 	return error;
    300   1.5  dholland }
    301   1.5  dholland 
    302   1.5  dholland static int
    303  1.15  dholland vfs_quotactl_put_extractinfo(prop_dictionary_t data,
    304  1.12  dholland 			struct quotaval *blocks, struct quotaval *files)
    305  1.12  dholland {
    306  1.12  dholland 	/*
    307  1.12  dholland 	 * So, the way proptoquota64 works is that you pass it an
    308  1.12  dholland 	 * array of pointers to uint64. Each of these pointers is
    309  1.12  dholland 	 * supposed to point to 5 (UFS_QUOTA_NENTRIES) uint64s. This
    310  1.12  dholland 	 * array of pointers is the second argument. The third and
    311  1.12  dholland 	 * forth argument are the names of the five values to extract,
    312  1.12  dholland 	 * and UFS_QUOTA_NENTRIES. The last two arguments are the
    313  1.12  dholland 	 * names assocated with the pointers (QUOTATYPE_LDICT_BLOCK,
    314  1.12  dholland 	 * QUOTADICT_LTYPE_FILE) and the number of pointers. Most of
    315  1.12  dholland 	 * the existing code was unsafely casting struct quotaval
    316  1.12  dholland 	 * (formerly struct ufs_quota_entry) to (uint64_t *) and using
    317  1.12  dholland 	 * that as the block of 5 uint64s. Or worse, pointing to
    318  1.12  dholland 	 * subregions of that and reducing the number of uint64s to
    319  1.12  dholland 	 * pull "adjacent" values. Demons fly out of your nose!
    320  1.12  dholland 	 */
    321  1.12  dholland 
    322  1.12  dholland 	uint64_t bvals[UFS_QUOTA_NENTRIES];
    323  1.12  dholland 	uint64_t fvals[UFS_QUOTA_NENTRIES];
    324  1.12  dholland 	uint64_t *valptrs[QUOTA_NLIMITS];
    325  1.12  dholland 	int error;
    326  1.12  dholland 
    327  1.12  dholland 	valptrs[QUOTA_LIMIT_BLOCK] = bvals;
    328  1.12  dholland 	valptrs[QUOTA_LIMIT_FILE] = fvals;
    329  1.12  dholland 	error = proptoquota64(data, valptrs,
    330  1.12  dholland 			      ufs_quota_entry_names, UFS_QUOTA_NENTRIES,
    331  1.12  dholland 			      ufs_quota_limit_names, QUOTA_NLIMITS);
    332  1.12  dholland 	if (error) {
    333  1.12  dholland 		return error;
    334  1.12  dholland 	}
    335  1.12  dholland 
    336  1.12  dholland 	/*
    337  1.12  dholland 	 * There are no symbolic constants for these indexes!
    338  1.12  dholland 	 */
    339  1.12  dholland 
    340  1.12  dholland 	blocks->qv_hardlimit = bvals[0];
    341  1.12  dholland 	blocks->qv_softlimit = bvals[1];
    342  1.12  dholland 	blocks->qv_usage = bvals[2];
    343  1.12  dholland 	blocks->qv_expiretime = bvals[3];
    344  1.12  dholland 	blocks->qv_grace = bvals[4];
    345  1.12  dholland 	files->qv_hardlimit = fvals[0];
    346  1.12  dholland 	files->qv_softlimit = fvals[1];
    347  1.12  dholland 	files->qv_usage = fvals[2];
    348  1.12  dholland 	files->qv_expiretime = fvals[3];
    349  1.12  dholland 	files->qv_grace = fvals[4];
    350  1.12  dholland 
    351  1.12  dholland 	return 0;
    352  1.12  dholland }
    353  1.12  dholland 
    354  1.12  dholland static int
    355  1.15  dholland vfs_quotactl_put(struct mount *mp,
    356   1.5  dholland 			prop_dictionary_t cmddict, int q2type,
    357   1.5  dholland 			prop_array_t datas)
    358   1.5  dholland {
    359  1.11  dholland 	prop_array_t replies;
    360  1.11  dholland 	prop_object_iterator_t iter;
    361  1.11  dholland 	prop_dictionary_t data;
    362  1.11  dholland 	int defaultq;
    363  1.11  dholland 	uint32_t id;
    364  1.11  dholland 	const char *idstr;
    365  1.14  dholland 	struct quotakey qk;
    366  1.12  dholland 	struct quotaval blocks, files;
    367   1.5  dholland 	struct vfs_quotactl_args args;
    368  1.11  dholland 	int error;
    369  1.11  dholland 
    370  1.11  dholland 	KASSERT(prop_object_type(cmddict) == PROP_TYPE_DICTIONARY);
    371  1.11  dholland 	KASSERT(prop_object_type(datas) == PROP_TYPE_ARRAY);
    372   1.5  dholland 
    373  1.11  dholland 	replies = prop_array_create();
    374  1.11  dholland 	if (replies == NULL)
    375  1.11  dholland 		return ENOMEM;
    376  1.11  dholland 
    377  1.11  dholland 	iter = prop_array_iterator(datas);
    378  1.11  dholland 	if (iter == NULL) {
    379  1.11  dholland 		prop_object_release(replies);
    380  1.11  dholland 		return ENOMEM;
    381  1.11  dholland 	}
    382  1.11  dholland 
    383  1.11  dholland 	while ((data = prop_object_iterator_next(iter)) != NULL) {
    384  1.13  dholland 
    385  1.13  dholland 		KASSERT(prop_object_type(data) == PROP_TYPE_DICTIONARY);
    386  1.13  dholland 
    387  1.11  dholland 		if (!prop_dictionary_get_uint32(data, "id", &id)) {
    388  1.11  dholland 			if (!prop_dictionary_get_cstring_nocopy(data, "id",
    389  1.11  dholland 			    &idstr))
    390  1.11  dholland 				continue;
    391  1.11  dholland 			if (strcmp(idstr, "default"))
    392  1.11  dholland 				continue;
    393  1.11  dholland 			id = 0;
    394  1.11  dholland 			defaultq = 1;
    395  1.11  dholland 		} else {
    396  1.11  dholland 			defaultq = 0;
    397  1.11  dholland 		}
    398  1.11  dholland 
    399  1.15  dholland 		error = vfs_quotactl_put_extractinfo(data, &blocks, &files);
    400  1.12  dholland 		if (error) {
    401  1.12  dholland 			goto err;
    402  1.12  dholland 		}
    403  1.12  dholland 
    404  1.14  dholland 		qk.qk_idtype = q2type;
    405  1.14  dholland 		qk.qk_id = defaultq ? QUOTA_DEFAULTID : id;
    406  1.14  dholland 		qk.qk_objtype = QUOTA_OBJTYPE_BLOCKS;
    407  1.14  dholland 
    408  1.15  dholland 		args.qc_type = QCT_PUT;
    409  1.15  dholland 		args.u.put.qc_key = &qk;
    410  1.15  dholland 		args.u.put.qc_val = &blocks;
    411  1.15  dholland 		error = VFS_QUOTACTL(mp, QUOTACTL_PUT, &args);
    412  1.13  dholland 		if (error) {
    413  1.13  dholland 			goto err;
    414  1.13  dholland 		}
    415  1.13  dholland 
    416  1.14  dholland 		qk.qk_idtype = q2type;
    417  1.14  dholland 		qk.qk_id = defaultq ? QUOTA_DEFAULTID : id;
    418  1.14  dholland 		qk.qk_objtype = QUOTA_OBJTYPE_FILES;
    419  1.14  dholland 
    420  1.15  dholland 		args.qc_type = QCT_PUT;
    421  1.15  dholland 		args.u.put.qc_key = &qk;
    422  1.15  dholland 		args.u.put.qc_val = &files;
    423  1.15  dholland 		error = VFS_QUOTACTL(mp, QUOTACTL_PUT, &args);
    424  1.11  dholland 		if (error) {
    425  1.11  dholland 			goto err;
    426  1.11  dholland 		}
    427  1.11  dholland 	}
    428  1.11  dholland 	prop_object_iterator_release(iter);
    429  1.11  dholland 	if (!prop_dictionary_set_and_rel(cmddict, "data", replies)) {
    430  1.11  dholland 		error = ENOMEM;
    431  1.11  dholland 	} else {
    432  1.11  dholland 		error = 0;
    433  1.11  dholland 	}
    434  1.11  dholland 	return error;
    435  1.11  dholland err:
    436  1.11  dholland 	prop_object_iterator_release(iter);
    437  1.11  dholland 	prop_object_release(replies);
    438  1.11  dholland 	return error;
    439   1.5  dholland }
    440   1.5  dholland 
    441  1.23  dholland static prop_dictionary_t
    442  1.26  dholland vfs_quotactl_getall_makereply(const struct quotakey *key)
    443  1.26  dholland {
    444  1.26  dholland 	prop_dictionary_t dict;
    445  1.26  dholland 	id_t id;
    446  1.26  dholland 	int defaultq;
    447  1.26  dholland 
    448  1.26  dholland 	dict = prop_dictionary_create();
    449  1.26  dholland 	if (dict == NULL)
    450  1.26  dholland 		return NULL;
    451  1.26  dholland 
    452  1.26  dholland 	id = key->qk_id;
    453  1.26  dholland 	if (id == QUOTA_DEFAULTID) {
    454  1.26  dholland 		id = 0;
    455  1.26  dholland 		defaultq = 1;
    456  1.26  dholland 	} else {
    457  1.26  dholland 		defaultq = 0;
    458  1.26  dholland 	}
    459  1.26  dholland 
    460  1.26  dholland 	if (defaultq) {
    461  1.26  dholland 		if (!prop_dictionary_set_cstring_nocopy(dict, "id",
    462  1.26  dholland 		    "default")) {
    463  1.26  dholland 			goto err;
    464  1.26  dholland 		}
    465  1.26  dholland 	} else {
    466  1.26  dholland 		if (!prop_dictionary_set_uint32(dict, "id", id)) {
    467  1.26  dholland 			goto err;
    468  1.26  dholland 		}
    469  1.26  dholland 	}
    470  1.26  dholland 
    471  1.26  dholland 	return dict;
    472  1.26  dholland 
    473  1.26  dholland err:
    474  1.26  dholland 	prop_object_release(dict);
    475  1.26  dholland 	return NULL;
    476  1.26  dholland }
    477  1.26  dholland 
    478  1.26  dholland static int
    479  1.26  dholland vfs_quotactl_getall_addreply(prop_dictionary_t thisreply,
    480  1.26  dholland     const struct quotakey *key, const struct quotaval *val)
    481  1.23  dholland {
    482  1.23  dholland #define INITQVNAMES_ALL { \
    483  1.23  dholland     QUOTADICT_LIMIT_HARD, \
    484  1.23  dholland     QUOTADICT_LIMIT_SOFT, \
    485  1.23  dholland     QUOTADICT_LIMIT_USAGE, \
    486  1.23  dholland     QUOTADICT_LIMIT_ETIME, \
    487  1.23  dholland     QUOTADICT_LIMIT_GTIME \
    488  1.23  dholland     }
    489  1.23  dholland #define N_QV 5
    490  1.23  dholland 
    491  1.23  dholland 	const char *val_names[] = INITQVNAMES_ALL;
    492  1.23  dholland 	uint64_t vals[N_QV];
    493  1.23  dholland 	prop_dictionary_t dict2;
    494  1.26  dholland 	const char *objtypename;
    495  1.23  dholland 
    496  1.26  dholland 	switch (key->qk_objtype) {
    497  1.26  dholland 	    case QUOTA_OBJTYPE_BLOCKS:
    498  1.26  dholland 		objtypename = QUOTADICT_LTYPE_BLOCK;
    499  1.26  dholland 		break;
    500  1.26  dholland 	    case QUOTA_OBJTYPE_FILES:
    501  1.26  dholland 		objtypename = QUOTADICT_LTYPE_FILE;
    502  1.26  dholland 		break;
    503  1.26  dholland 	    default:
    504  1.26  dholland 		return EINVAL;
    505  1.23  dholland 	}
    506  1.23  dholland 
    507  1.26  dholland 	vals[0] = val->qv_hardlimit;
    508  1.26  dholland 	vals[1] = val->qv_softlimit;
    509  1.26  dholland 	vals[2] = val->qv_usage;
    510  1.26  dholland 	vals[3] = val->qv_expiretime;
    511  1.26  dholland 	vals[4] = val->qv_grace;
    512  1.23  dholland 	dict2 = limits64toprop(vals, val_names, N_QV);
    513  1.23  dholland 	if (dict2 == NULL)
    514  1.26  dholland 		return ENOMEM;
    515  1.23  dholland 
    516  1.26  dholland 	if (!prop_dictionary_set_and_rel(thisreply, objtypename, dict2))
    517  1.26  dholland 		return ENOMEM;
    518  1.23  dholland 
    519  1.26  dholland 	return 0;
    520  1.23  dholland }
    521  1.23  dholland 
    522   1.5  dholland static int
    523   1.5  dholland vfs_quotactl_getall(struct mount *mp,
    524   1.5  dholland 			prop_dictionary_t cmddict, int q2type,
    525   1.5  dholland 			prop_array_t datas)
    526   1.5  dholland {
    527  1.21  dholland 	struct quotakcursor cursor;
    528  1.27  dholland 	struct quotakey *keys;
    529  1.27  dholland 	struct quotaval *vals;
    530  1.27  dholland 	unsigned loopmax = 8;
    531  1.27  dholland 	unsigned loopnum;
    532   1.5  dholland 	struct vfs_quotactl_args args;
    533  1.23  dholland 	prop_array_t replies;
    534  1.26  dholland 	struct quotakey *key;
    535  1.26  dholland 	struct quotaval *val;
    536  1.26  dholland 	id_t lastid;
    537  1.27  dholland 	prop_dictionary_t thisreply;
    538  1.23  dholland 	unsigned i;
    539  1.21  dholland 	int error, error2;
    540  1.23  dholland 
    541  1.23  dholland 	KASSERT(prop_object_type(cmddict) == PROP_TYPE_DICTIONARY);
    542  1.21  dholland 
    543  1.21  dholland 	args.qc_type = QCT_CURSOROPEN;
    544  1.21  dholland 	args.u.cursoropen.qc_cursor = &cursor;
    545  1.21  dholland 	error = VFS_QUOTACTL(mp, QUOTACTL_CURSOROPEN, &args);
    546  1.21  dholland 	if (error) {
    547  1.21  dholland 		return error;
    548  1.21  dholland 	}
    549   1.5  dholland 
    550  1.27  dholland 	keys = malloc(loopmax * sizeof(keys[0]), M_TEMP, M_WAITOK);
    551  1.27  dholland 	vals = malloc(loopmax * sizeof(vals[0]), M_TEMP, M_WAITOK);
    552  1.23  dholland 
    553  1.23  dholland 	replies = prop_array_create();
    554  1.23  dholland 	if (replies == NULL) {
    555  1.23  dholland 		error = ENOMEM;
    556  1.23  dholland 		goto err;
    557  1.23  dholland 	}
    558  1.23  dholland 
    559  1.26  dholland 	thisreply = NULL;
    560  1.26  dholland 	lastid = 0; /* value not actually referenced */
    561  1.27  dholland 
    562  1.27  dholland 	while (1) {
    563  1.29  dholland 		args.qc_type = QCT_CURSORGET;
    564  1.29  dholland 		args.u.cursorget.qc_cursor = &cursor;
    565  1.29  dholland 		args.u.cursorget.qc_keys = keys;
    566  1.29  dholland 		args.u.cursorget.qc_vals = vals;
    567  1.29  dholland 		args.u.cursorget.qc_maxnum = loopmax;
    568  1.29  dholland 		args.u.cursorget.qc_ret = &loopnum;
    569  1.27  dholland 
    570  1.29  dholland 		error = VFS_QUOTACTL(mp, QUOTACTL_CURSORGET, &args);
    571  1.26  dholland 		if (error) {
    572  1.23  dholland 			goto err;
    573  1.23  dholland 		}
    574  1.27  dholland 
    575  1.27  dholland 		if (loopnum == 0) {
    576  1.27  dholland 			/* end of iteration */
    577  1.27  dholland 			break;
    578  1.27  dholland 		}
    579  1.27  dholland 
    580  1.27  dholland 		for (i = 0; i < loopnum; i++) {
    581  1.27  dholland 			key = &keys[i];
    582  1.27  dholland 			val = &vals[i];
    583  1.27  dholland 
    584  1.28  dholland 			if (key->qk_idtype != q2type) {
    585  1.28  dholland 				/* don't want this result */
    586  1.28  dholland 				continue;
    587  1.28  dholland 			}
    588  1.28  dholland 
    589  1.27  dholland 			if (thisreply == NULL || key->qk_id != lastid) {
    590  1.27  dholland 				lastid = key->qk_id;
    591  1.27  dholland 				thisreply = vfs_quotactl_getall_makereply(key);
    592  1.27  dholland 				if (thisreply == NULL) {
    593  1.27  dholland 					error = ENOMEM;
    594  1.27  dholland 					goto err;
    595  1.27  dholland 				}
    596  1.27  dholland 				/*
    597  1.27  dholland 				 * Note: while we release our reference to
    598  1.27  dholland 				 * thisreply here, we can (and do) continue to
    599  1.27  dholland 				 * use the pointer in the loop because the
    600  1.27  dholland 				 * copy attached to the replies array is not
    601  1.27  dholland 				 * going away.
    602  1.27  dholland 				 */
    603  1.27  dholland 				if (!prop_array_add_and_rel(replies,
    604  1.27  dholland 							    thisreply)) {
    605  1.27  dholland 					error = ENOMEM;
    606  1.27  dholland 					goto err;
    607  1.27  dholland 				}
    608  1.27  dholland 			}
    609  1.27  dholland 
    610  1.27  dholland 			error = vfs_quotactl_getall_addreply(thisreply,
    611  1.27  dholland 							     key, val);
    612  1.27  dholland 			if (error) {
    613  1.27  dholland 				goto err;
    614  1.27  dholland 			}
    615  1.27  dholland 		}
    616  1.23  dholland 	}
    617  1.23  dholland 
    618  1.23  dholland 	if (!prop_dictionary_set_and_rel(cmddict, "data", replies)) {
    619  1.23  dholland 		error = ENOMEM;
    620  1.23  dholland 		goto err;
    621  1.23  dholland 	}
    622  1.23  dholland 
    623  1.23  dholland 	error = 0;
    624  1.23  dholland  err:
    625  1.27  dholland 	free(keys, M_TEMP);
    626  1.27  dholland 	free(vals, M_TEMP);
    627  1.21  dholland 
    628  1.21  dholland 	args.qc_type = QCT_CURSORCLOSE;
    629  1.21  dholland 	args.u.cursorclose.qc_cursor = &cursor;
    630  1.21  dholland 	error2 = VFS_QUOTACTL(mp, QUOTACTL_CURSORCLOSE, &args);
    631  1.21  dholland 
    632  1.21  dholland 	if (error) {
    633  1.21  dholland 		return error;
    634  1.21  dholland 	}
    635  1.21  dholland 	error = error2;
    636  1.21  dholland 	return error;
    637   1.5  dholland }
    638   1.5  dholland 
    639   1.5  dholland static int
    640   1.5  dholland vfs_quotactl_clear(struct mount *mp,
    641   1.5  dholland 			prop_dictionary_t cmddict, int q2type,
    642   1.5  dholland 			prop_array_t datas)
    643   1.5  dholland {
    644  1.16  dholland 	prop_array_t replies;
    645  1.16  dholland 	prop_object_iterator_t iter;
    646  1.16  dholland 	prop_dictionary_t data;
    647  1.16  dholland 	uint32_t id;
    648  1.16  dholland 	int defaultq;
    649  1.16  dholland 	const char *idstr;
    650  1.20  dholland 	struct quotakey qk;
    651   1.5  dholland 	struct vfs_quotactl_args args;
    652  1.16  dholland 	int error;
    653  1.16  dholland 
    654  1.16  dholland 	KASSERT(prop_object_type(cmddict) == PROP_TYPE_DICTIONARY);
    655  1.16  dholland 	KASSERT(prop_object_type(datas) == PROP_TYPE_ARRAY);
    656  1.16  dholland 
    657  1.16  dholland 	replies = prop_array_create();
    658  1.16  dholland 	if (replies == NULL)
    659  1.16  dholland 		return ENOMEM;
    660   1.5  dholland 
    661  1.16  dholland 	iter = prop_array_iterator(datas);
    662  1.16  dholland 	if (iter == NULL) {
    663  1.16  dholland 		prop_object_release(replies);
    664  1.16  dholland 		return ENOMEM;
    665  1.16  dholland 	}
    666  1.16  dholland 
    667  1.16  dholland 	while ((data = prop_object_iterator_next(iter)) != NULL) {
    668  1.16  dholland 		if (!prop_dictionary_get_uint32(data, "id", &id)) {
    669  1.16  dholland 			if (!prop_dictionary_get_cstring_nocopy(data, "id",
    670  1.16  dholland 			    &idstr))
    671  1.16  dholland 				continue;
    672  1.16  dholland 			if (strcmp(idstr, "default"))
    673  1.16  dholland 				continue;
    674  1.16  dholland 			id = 0;
    675  1.16  dholland 			defaultq = 1;
    676  1.16  dholland 		} else {
    677  1.16  dholland 			defaultq = 0;
    678  1.16  dholland 		}
    679  1.16  dholland 
    680  1.20  dholland 		qk.qk_idtype = q2type;
    681  1.20  dholland 		qk.qk_id = defaultq ? QUOTA_DEFAULTID : id;
    682  1.20  dholland 		qk.qk_objtype = QUOTA_OBJTYPE_BLOCKS;
    683  1.20  dholland 
    684  1.19  dholland 		args.qc_type = QCT_DELETE;
    685  1.20  dholland 		args.u.delete.qc_key = &qk;
    686  1.19  dholland 		error = VFS_QUOTACTL(mp, QUOTACTL_DELETE, &args);
    687  1.18  dholland 		if (error) {
    688  1.18  dholland 			goto err;
    689  1.18  dholland 		}
    690  1.18  dholland 
    691  1.20  dholland 		qk.qk_idtype = q2type;
    692  1.20  dholland 		qk.qk_id = defaultq ? QUOTA_DEFAULTID : id;
    693  1.20  dholland 		qk.qk_objtype = QUOTA_OBJTYPE_FILES;
    694  1.20  dholland 
    695  1.19  dholland 		args.qc_type = QCT_DELETE;
    696  1.20  dholland 		args.u.delete.qc_key = &qk;
    697  1.19  dholland 		error = VFS_QUOTACTL(mp, QUOTACTL_DELETE, &args);
    698  1.16  dholland 		if (error) {
    699  1.16  dholland 			goto err;
    700  1.16  dholland 		}
    701  1.16  dholland 	}
    702  1.16  dholland 
    703  1.16  dholland 	prop_object_iterator_release(iter);
    704  1.16  dholland 	if (!prop_dictionary_set_and_rel(cmddict, "data", replies)) {
    705  1.16  dholland 		error = ENOMEM;
    706  1.16  dholland 	} else {
    707  1.16  dholland 		error = 0;
    708  1.16  dholland 	}
    709  1.16  dholland 	return error;
    710  1.16  dholland err:
    711  1.16  dholland 	prop_object_iterator_release(iter);
    712  1.16  dholland 	prop_object_release(replies);
    713  1.16  dholland 	return error;
    714   1.5  dholland }
    715   1.5  dholland 
    716   1.5  dholland static int
    717   1.4  dholland vfs_quotactl_cmd(struct mount *mp, prop_dictionary_t cmddict)
    718   1.4  dholland {
    719   1.4  dholland 	int error;
    720   1.4  dholland 	const char *cmd, *type;
    721   1.4  dholland 	prop_array_t datas;
    722   1.4  dholland 	int q2type;
    723   1.4  dholland 
    724   1.4  dholland 	if (!prop_dictionary_get_cstring_nocopy(cmddict, "command", &cmd))
    725   1.4  dholland 		return EINVAL;
    726   1.4  dholland 	if (!prop_dictionary_get_cstring_nocopy(cmddict, "type", &type))
    727   1.4  dholland 		return EINVAL;
    728   1.4  dholland 
    729   1.4  dholland 	if (!strcmp(type, QUOTADICT_CLASS_USER)) {
    730   1.4  dholland 		q2type = QUOTA_CLASS_USER;
    731   1.4  dholland 	} else if (!strcmp(type, QUOTADICT_CLASS_GROUP)) {
    732   1.4  dholland 		q2type = QUOTA_CLASS_GROUP;
    733   1.4  dholland 	} else {
    734   1.4  dholland 		/* XXX this is a bad errno for this case */
    735   1.4  dholland 		return EOPNOTSUPP;
    736   1.4  dholland 	}
    737   1.4  dholland 
    738   1.4  dholland 	datas = prop_dictionary_get(cmddict, "data");
    739   1.4  dholland 	if (datas == NULL || prop_object_type(datas) != PROP_TYPE_ARRAY)
    740   1.4  dholland 		return EINVAL;
    741   1.4  dholland 
    742   1.4  dholland 	prop_object_retain(datas);
    743   1.4  dholland 	prop_dictionary_remove(cmddict, "data"); /* prepare for return */
    744   1.4  dholland 
    745   1.4  dholland 	if (strcmp(cmd, "get version") == 0) {
    746   1.5  dholland 		error = vfs_quotactl_getversion(mp, cmddict, q2type, datas);
    747   1.4  dholland 	} else if (strcmp(cmd, "quotaon") == 0) {
    748   1.5  dholland 		error = vfs_quotactl_quotaon(mp, cmddict, q2type, datas);
    749   1.4  dholland 	} else if (strcmp(cmd, "quotaoff") == 0) {
    750   1.5  dholland 		error = vfs_quotactl_quotaoff(mp, cmddict, q2type, datas);
    751   1.4  dholland 	} else if (strcmp(cmd, "get") == 0) {
    752   1.5  dholland 		error = vfs_quotactl_get(mp, cmddict, q2type, datas);
    753   1.4  dholland 	} else if (strcmp(cmd, "set") == 0) {
    754  1.15  dholland 		error = vfs_quotactl_put(mp, cmddict, q2type, datas);
    755   1.4  dholland 	} else if (strcmp(cmd, "getall") == 0) {
    756   1.5  dholland 		error = vfs_quotactl_getall(mp, cmddict, q2type, datas);
    757   1.4  dholland 	} else if (strcmp(cmd, "clear") == 0) {
    758   1.5  dholland 		error = vfs_quotactl_clear(mp, cmddict, q2type, datas);
    759   1.4  dholland 	} else {
    760   1.4  dholland 		/* XXX this a bad errno for this case */
    761   1.4  dholland 		error = EOPNOTSUPP;
    762   1.4  dholland 	}
    763   1.4  dholland 
    764   1.4  dholland 	error = (prop_dictionary_set_int8(cmddict, "return",
    765   1.4  dholland 	    error) ? 0 : ENOMEM);
    766   1.4  dholland 	prop_object_release(datas);
    767   1.4  dholland 
    768   1.4  dholland 	return error;
    769   1.4  dholland }
    770   1.4  dholland 
    771   1.2  dholland int
    772   1.2  dholland vfs_quotactl(struct mount *mp, prop_dictionary_t dict)
    773   1.2  dholland {
    774   1.3  dholland 	prop_dictionary_t cmddict;
    775   1.3  dholland 	prop_array_t commands;
    776   1.3  dholland 	prop_object_iterator_t iter;
    777   1.3  dholland 	int error;
    778   1.3  dholland 
    779   1.3  dholland 	error = quota_get_cmds(dict, &commands);
    780   1.3  dholland 	if (error) {
    781   1.3  dholland 		return error;
    782   1.3  dholland 	}
    783   1.3  dholland 
    784   1.3  dholland 	iter = prop_array_iterator(commands);
    785   1.3  dholland 	if (iter == NULL) {
    786   1.3  dholland 		return ENOMEM;
    787   1.3  dholland 	}
    788   1.3  dholland 
    789   1.3  dholland 	while ((cmddict = prop_object_iterator_next(iter)) != NULL) {
    790   1.3  dholland 		if (prop_object_type(cmddict) != PROP_TYPE_DICTIONARY) {
    791   1.3  dholland 			/* XXX shouldn't this be an error? */
    792   1.3  dholland 			continue;
    793   1.3  dholland 		}
    794   1.4  dholland 		error = vfs_quotactl_cmd(mp, cmddict);
    795   1.3  dholland 		if (error) {
    796   1.3  dholland 			break;
    797   1.3  dholland 		}
    798   1.3  dholland 	}
    799   1.3  dholland 	prop_object_iterator_release(iter);
    800   1.3  dholland 	return error;
    801   1.2  dholland }
    802