 @@ -1,14 +1,14 @@
-/*	$NetBSD: vfs_quotactl.c,v 1.23 2012/01/29 07:02:06 dholland Exp $	*/
+/*	$NetBSD: vfs_quotactl.c,v 1.24 2012/01/29 07:05:12 dholland Exp $	*/
 /*
  * Copyright (c) 1991, 1993, 1994
  *	The Regents of the University of California.  All rights reserved.
  * (c) UNIX System Laboratories, Inc.
  * All or some portions of this file are derived from material licensed
  * to the University of California by American Telephone and Telegraph
  * Co. or Unix System Laboratories, Inc. and are reproduced herein with
  * the permission of UNIX System Laboratories, Inc.
+ *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions
  * are met:
 @@ -70,27 +70,27 @@
+ *
  *	@(#)ufs_quota.c	8.5 (Berkeley) 5/20/95
  *	From NetBSD: ufs_quota.c,v 1.70 2011/03/24 17:05:46 bouyer Exp
  */
 /*
  * Note that both of the copyrights above are moderately spurious;
  * this code should almost certainly have the Copyright 2010 Manuel
  * Bouyer notice and license found in e.g. sys/ufs/ufs/quota2_subr.c.
  * However, they're what was on the files this code was sliced out of.
  */
 #include <sys/cdefs.h>
-__KERNEL_RCSID(0, "$NetBSD: vfs_quotactl.c,v 1.23 2012/01/29 07:02:06 dholland Exp $");
+__KERNEL_RCSID(0, "$NetBSD: vfs_quotactl.c,v 1.24 2012/01/29 07:05:12 dholland Exp $");
 #include <sys/malloc.h> /* XXX: temporary */
 #include <sys/mount.h>
 #include <sys/quota.h>
 #include <sys/quotactl.h>
 #include <quota/quotaprop.h>
 static int
 vfs_quotactl_getversion(struct mount *mp,
 			prop_dictionary_t cmddict, int q2type,
 			prop_array_t datas)
+{
 	prop_array_t replies;
 @@ -516,26 +516,28 @@ vfs_quotactl_getall(struct mount *mp,
 	int skip = 0;
 	KASSERT(prop_object_type(cmddict) == PROP_TYPE_DICTIONARY);
 	args.qc_type = QCT_CURSOROPEN;
 	args.u.cursoropen.qc_cursor = &cursor;
 	error = VFS_QUOTACTL(mp, QUOTACTL_CURSOROPEN, &args);
 	if (error) {
 		return error;
+	}
 	result.qr_keys = NULL;
 	result.qr_vals = NULL;
 	result.qr_num = 0;
 	result.qr_max = 0x7fffffff; /* XXX bogus; but temporary */
 	args.qc_type = QCT_GETALL;
 	args.u.getall.qc_cursor = &cursor;
 	args.u.getall.qc_idtype = q2type;
 	args.u.getall.qc_result = &result;
 	error = VFS_QUOTACTL(mp, QUOTACTL_GETALL, &args);
 	/*
 	 * XXX this is bogus but up until now *all* errors
 	 * from inside quotactl_getall were suppressed by the
 	 * dispatching code in ufs_quota.c. Fixing that causes
 	 * repquota to break in an undesirable way; this is a
 	 * workaround.
 	 */

 @@ -1,14 +1,14 @@
-/* $NetBSD: ufs_quota2.c,v 1.20 2012/01/29 07:04:21 dholland Exp $ */
+/* $NetBSD: ufs_quota2.c,v 1.21 2012/01/29 07:05:12 dholland Exp $ */
 /*-
   * Copyright (c) 2010 Manuel Bouyer
   * All rights reserved.
+  *
   * 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.
+  *
 @@ -16,27 +16,27 @@
   * ``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 FOUNDATION 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.
   */
 #include <sys/cdefs.h>
-__KERNEL_RCSID(0, "$NetBSD: ufs_quota2.c,v 1.20 2012/01/29 07:04:21 dholland Exp $");
+__KERNEL_RCSID(0, "$NetBSD: ufs_quota2.c,v 1.21 2012/01/29 07:05:12 dholland Exp $");
 #include <sys/buf.h>
 #include <sys/param.h>
 #include <sys/kernel.h>
 #include <sys/systm.h>
 #include <sys/malloc.h>
 #include <sys/namei.h>
 #include <sys/file.h>
 #include <sys/proc.h>
 #include <sys/vnode.h>
 #include <sys/mount.h>
 #include <sys/fstrans.h>
 #include <sys/kauth.h>
 @@ -819,27 +819,28 @@ quota2_handle_cmd_delete(struct ufsmount
 out_dqlock:
 	mutex_exit(&dqlock);
 out_wapbl:
 	UFS_WAPBL_END(ump->um_mountp);
 out_il:
 	mutex_exit(&dq->dq_interlock);
 out_dq:
 	dqrele(NULLVP, dq);
 	return error;
+}
 static int
 quota2_result_add_q2e(struct ufsmount *ump, int idtype,
-    int id, struct quota_getall_result *result, unsigned pos)
+    int id, struct quota_getall_result *result, unsigned pos,
     int skipfirst, int skiplast)
+{
 	struct dquot *dq;
 	int error;
 	struct quota2_entry *q2ep, q2e;
 	struct buf  *bp;
 	const int needswap = UFS_MPNEEDSWAP(ump);
 	error = dqget(NULLVP, id, ump, idtype, &dq);
 	if (error)
 		return error;
 	mutex_enter(&dq->dq_interlock);
 	if (dq->dq2_lblkno == 0 && dq->dq2_blkoff == 0) {
 @@ -849,35 +850,41 @@ quota2_result_add_q2e(struct ufsmount *u
+	}
 	error = getq2e(ump, idtype, dq->dq2_lblkno, dq->dq2_blkoff,
 	    &bp, &q2ep, 0);
 	if (error) {
 		mutex_exit(&dq->dq_interlock);
 		dqrele(NULLVP, dq);
 		return error;
+	}
 	quota2_ufs_rwq2e(q2ep, &q2e, needswap);
 	brelse(bp, 0);
 	mutex_exit(&dq->dq_interlock);
 	dqrele(NULLVP, dq);
-	result->qr_keys[pos].qk_idtype = idtype;
+	if (skipfirst == 0) {
-	result->qr_keys[pos].qk_objtype = QUOTA_OBJTYPE_BLOCKS;
+		result->qr_keys[pos].qk_idtype = idtype;
-	q2e_to_quotaval(&q2e, 0, &result->qr_keys[pos].qk_id,
+		result->qr_keys[pos].qk_objtype = QUOTA_OBJTYPE_BLOCKS;
-			QL_BLOCK, &result->qr_vals[pos]);
+		q2e_to_quotaval(&q2e, 0, &result->qr_keys[pos].qk_id,
 				QL_BLOCK, &result->qr_vals[pos]);
-	result->qr_keys[pos+1].qk_idtype = idtype;
+		pos++;
 	result->qr_keys[pos+1].qk_objtype = QUOTA_OBJTYPE_FILES;
 	q2e_to_quotaval(&q2e, 0, &result->qr_keys[pos+1].qk_id,
-			QL_FILE, &result->qr_vals[pos+1]);
+	if (skiplast == 0) {
 		result->qr_keys[pos].qk_idtype = idtype;
 		result->qr_keys[pos].qk_objtype = QUOTA_OBJTYPE_FILES;
 		q2e_to_quotaval(&q2e, 0, &result->qr_keys[pos].qk_id,
 				QL_FILE, &result->qr_vals[pos]);
 		pos++;
+	}
 	return 0;
+}
 static int
 quota2_fetch_q2e(struct ufsmount *ump, const struct quotakey *qk,
     struct quotaval *ret)
+{
 	struct dquot *dq;
 	int error;
 	struct quota2_entry *q2ep, q2e;
 	struct buf  *bp;
 	const int needswap = UFS_MPNEEDSWAP(ump);
 @@ -987,26 +994,27 @@ q2cursor_check(struct ufsq2_cursor *curs
+	}
 	if (cursor->q2c_blocks_done != 0 && cursor->q2c_blocks_done != 1) {
 		return EINVAL;
+	}
 	return 0;
+}
 struct getuids {
 	long nuids; /* number of uids in array */
 	long size;  /* size of array */
 	uid_t *uids; /* array of uids, dynamically allocated */
 	long skip;
 	long seen;
 	long limit;
 };
 static int
 quota2_getuids_callback(struct ufsmount *ump, uint64_t *offp,
     struct quota2_entry *q2ep, uint64_t off, void *v)
+{
 	struct getuids *gu = v;
 	uid_t *newuids;
 #ifdef FFS_EI
 	const int needswap = UFS_MPNEEDSWAP(ump);
 #endif
 	if (gu->skip > 0) {
 @@ -1016,46 +1024,50 @@ quota2_getuids_callback(struct ufsmount
 	if (gu->nuids == gu->size) {
 		newuids = realloc(gu->uids, gu->size + PAGE_SIZE, M_TEMP,
 		    M_WAITOK);
 		if (newuids == NULL) {
 			free(gu->uids, M_TEMP);
 			return ENOMEM;
+		}
 		gu->uids = newuids;
 		gu->size += (PAGE_SIZE / sizeof(uid_t));
+	}
 	gu->uids[gu->nuids] = ufs_rw32(q2ep->q2e_uid, needswap);
 	gu->nuids++;
 	gu->seen++;
 	if (gu->nuids == gu->limit) {
 		return Q2WL_ABORT;
+	}
 	return 0;
+}
 int
 quota2_handle_cmd_getall(struct ufsmount *ump, struct quotakcursor *qkc,
     int idtype, struct quota_getall_result *result)
+{
 	int error;
 	struct ufsq2_cursor *cursor;
 	struct quota2_header *q2h;
 	struct quota2_entry  q2e;
 	struct buf *hbp;
 	uint64_t offset;
 	int i, j;
 	int quota2_hash_size;
 	const int needswap = UFS_MPNEEDSWAP(ump);
 	struct getuids gu;
 	id_t junkid;
 	struct quotaval qv;
 	unsigned num, maxnum;
 	int skipfirst, skiplast;
 	cursor = Q2CURSOR(qkc);
 	error = q2cursor_check(cursor);
 	if (error) {
 		return error;
+	}
 	if (ump->um_quotas[idtype] == NULLVP) {
 		return ENODEV;
+	}
 	mutex_enter(&dqlock);
 	error = getq2h(ump, idtype, &hbp, &q2h, 0);
 @@ -1088,64 +1100,87 @@ quota2_handle_cmd_getall(struct ufsmount
 	memset(&gu, 0, sizeof(gu));
 	quota2_hash_size = ufs_rw16(q2h->q2h_hash_size, needswap);
 	/* if the table size has changed, make the caller start over */
 	if (cursor->q2c_hashsize == 0) {
 		cursor->q2c_hashsize = quota2_hash_size;
 	} else if (cursor->q2c_hashsize != quota2_hash_size) {
 		error = EDEADLK;
 		goto fail;
+	}
 	gu.skip = cursor->q2c_uidpos;
 	gu.seen = 0;
 	gu.limit = result->qr_max / 2;
 	if (gu.limit == 0 && result->qr_max > 0) {
 		gu.limit = 1;
+	}
 	for (i = cursor->q2c_hashpos; i < quota2_hash_size ; i++) {
 		offset = q2h->q2h_entries[i];
 		gu.seen = 0;
 		error = quota2_walk_list(ump, hbp, idtype, &offset, 0, &gu,
 		    quota2_getuids_callback);
 		if (error == Q2WL_ABORT) {
 			/* got enough uids for now */
 			error = 0;
 			break;
+		}
 		if (error) {
 			if (gu.uids != NULL)
 				free(gu.uids, M_TEMP);
 			break;
+		}
+	}
 	cursor->q2c_hashpos = i;
 	cursor->q2c_uidpos = gu.seen;
 fail:
 	mutex_exit(&dqlock);
 	brelse(hbp, 0);
 	if (error)
 		return error;
 	maxnum = gu.nuids*2;
 	result->qr_keys = malloc(maxnum * sizeof(result->qr_keys[0]),
 				 M_TEMP, M_WAITOK);
 	result->qr_vals = malloc(maxnum * sizeof(result->qr_vals[0]),
 				 M_TEMP, M_WAITOK);
 	/*
 	 * If we've already sent back the blocks value for the first id,
 	 * don't send it again (skipfirst).
+	 *
 	 * If we have an odd number of available result slots and we
 	 * aren't going to skip the first result entry, we need to
 	 * leave off the last result entry (skiplast).
 	 */
 	skipfirst = (cursor->q2c_blocks_done != 0);
 	skiplast = skipfirst == 0 && (result->qr_max < maxnum);
 	num = 0;
 	for (j = 0; j < gu.nuids; j++) {
 		error = quota2_result_add_q2e(ump, idtype,
-		    gu.uids[j], result, j*2);
+		    gu.uids[j], result, j*2,
 		    j == 0 && skipfirst,
 		    j + 1 == gu.nuids && skiplast);
 		if (error == ENOENT)
 			continue;
 		if (error)
 			break;
 		num += 2;
+	}
 	result->qr_num = num;
 	cursor->q2c_blocks_done = skiplast;
 	free(gu.uids, M_TEMP);
 	return error;
+}
 int
 quota2_handle_cmd_cursoropen(struct ufsmount *ump, struct quotakcursor *qkc)
+{
 	struct ufsq2_cursor *cursor;
 	CTASSERT(sizeof(*cursor) <= sizeof(qkc->u.qkc_space));
 	cursor = Q2CURSOR(qkc);
 	cursor->q2c_magic = Q2C_MAGIC;

 @@ -1,14 +1,14 @@
-/*	$NetBSD: quotactl.h,v 1.21 2012/01/29 07:02:06 dholland Exp $	*/
+/*	$NetBSD: quotactl.h,v 1.22 2012/01/29 07:05:12 dholland Exp $	*/
 /*-
  * Copyright (c) 2011 The NetBSD Foundation, Inc.
  * All rights reserved.
+ *
  * This code is derived from software contributed to The NetBSD Foundation
  * by David A. Holland.
+ *
  * 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
 @@ -102,19 +102,20 @@ struct vfs_quotactl_args {
 		} cursoropen;
 		struct {
 			struct quotakcursor *qc_cursor;
 		} cursorclose;
 		struct {
 			struct quotakcursor *qc_cursor;
 			int qc_idtype;
 			struct quota_getall_result {
 				struct quotaval qr_defblocks;
 				struct quotaval qr_deffiles;
 				struct quotakey *qr_keys;
 				struct quotaval *qr_vals;
 				unsigned qr_num;
 				unsigned qr_max;
 			} *qc_result;
 		} getall;
 	} u;
 };
 #endif /* _SYS_QUOTACTL_H_ */