Home | History | Annotate | Line # | Download | only in lfs
ulfs_quota1.c revision 1.7.4.2
      1  1.7.4.2  yamt /*	$NetBSD: ulfs_quota1.c,v 1.7.4.2 2014/05/22 11:41:19 yamt Exp $	*/
      2  1.7.4.2  yamt /*  from NetBSD: ufs_quota1.c,v 1.18 2012/02/02 03:00:48 matt Exp  */
      3  1.7.4.2  yamt 
      4  1.7.4.2  yamt /*
      5  1.7.4.2  yamt  * Copyright (c) 1982, 1986, 1990, 1993, 1995
      6  1.7.4.2  yamt  *	The Regents of the University of California.  All rights reserved.
      7  1.7.4.2  yamt  *
      8  1.7.4.2  yamt  * This code is derived from software contributed to Berkeley by
      9  1.7.4.2  yamt  * Robert Elz at The University of Melbourne.
     10  1.7.4.2  yamt  *
     11  1.7.4.2  yamt  * Redistribution and use in source and binary forms, with or without
     12  1.7.4.2  yamt  * modification, are permitted provided that the following conditions
     13  1.7.4.2  yamt  * are met:
     14  1.7.4.2  yamt  * 1. Redistributions of source code must retain the above copyright
     15  1.7.4.2  yamt  *    notice, this list of conditions and the following disclaimer.
     16  1.7.4.2  yamt  * 2. Redistributions in binary form must reproduce the above copyright
     17  1.7.4.2  yamt  *    notice, this list of conditions and the following disclaimer in the
     18  1.7.4.2  yamt  *    documentation and/or other materials provided with the distribution.
     19  1.7.4.2  yamt  * 3. Neither the name of the University nor the names of its contributors
     20  1.7.4.2  yamt  *    may be used to endorse or promote products derived from this software
     21  1.7.4.2  yamt  *    without specific prior written permission.
     22  1.7.4.2  yamt  *
     23  1.7.4.2  yamt  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
     24  1.7.4.2  yamt  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
     25  1.7.4.2  yamt  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
     26  1.7.4.2  yamt  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
     27  1.7.4.2  yamt  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
     28  1.7.4.2  yamt  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
     29  1.7.4.2  yamt  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
     30  1.7.4.2  yamt  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
     31  1.7.4.2  yamt  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
     32  1.7.4.2  yamt  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
     33  1.7.4.2  yamt  * SUCH DAMAGE.
     34  1.7.4.2  yamt  *
     35  1.7.4.2  yamt  *	@(#)ufs_quota.c	8.5 (Berkeley) 5/20/95
     36  1.7.4.2  yamt  */
     37  1.7.4.2  yamt 
     38  1.7.4.2  yamt #include <sys/cdefs.h>
     39  1.7.4.2  yamt __KERNEL_RCSID(0, "$NetBSD: ulfs_quota1.c,v 1.7.4.2 2014/05/22 11:41:19 yamt Exp $");
     40  1.7.4.2  yamt 
     41  1.7.4.2  yamt #include <sys/param.h>
     42  1.7.4.2  yamt #include <sys/kernel.h>
     43  1.7.4.2  yamt #include <sys/systm.h>
     44  1.7.4.2  yamt #include <sys/namei.h>
     45  1.7.4.2  yamt #include <sys/file.h>
     46  1.7.4.2  yamt #include <sys/proc.h>
     47  1.7.4.2  yamt #include <sys/vnode.h>
     48  1.7.4.2  yamt #include <sys/mount.h>
     49  1.7.4.2  yamt #include <sys/kauth.h>
     50  1.7.4.2  yamt 
     51  1.7.4.2  yamt #include <ufs/lfs/ulfs_quota1.h>
     52  1.7.4.2  yamt #include <ufs/lfs/ulfs_inode.h>
     53  1.7.4.2  yamt #include <ufs/lfs/ulfsmount.h>
     54  1.7.4.2  yamt #include <ufs/lfs/ulfs_extern.h>
     55  1.7.4.2  yamt #include <ufs/lfs/ulfs_quota.h>
     56  1.7.4.2  yamt 
     57  1.7.4.2  yamt static int chkdqchg(struct inode *, int64_t, kauth_cred_t, int);
     58  1.7.4.2  yamt static int chkiqchg(struct inode *, int32_t, kauth_cred_t, int);
     59  1.7.4.2  yamt 
     60  1.7.4.2  yamt /*
     61  1.7.4.2  yamt  * Update disk usage, and take corrective action.
     62  1.7.4.2  yamt  */
     63  1.7.4.2  yamt int
     64  1.7.4.2  yamt lfs_chkdq1(struct inode *ip, int64_t change, kauth_cred_t cred, int flags)
     65  1.7.4.2  yamt {
     66  1.7.4.2  yamt 	struct dquot *dq;
     67  1.7.4.2  yamt 	int i;
     68  1.7.4.2  yamt 	int ncurblocks, error;
     69  1.7.4.2  yamt 
     70  1.7.4.2  yamt 	if ((error = lfs_getinoquota(ip)) != 0)
     71  1.7.4.2  yamt 		return error;
     72  1.7.4.2  yamt 	if (change == 0)
     73  1.7.4.2  yamt 		return (0);
     74  1.7.4.2  yamt 	if (change < 0) {
     75  1.7.4.2  yamt 		for (i = 0; i < ULFS_MAXQUOTAS; i++) {
     76  1.7.4.2  yamt 			if ((dq = ip->i_dquot[i]) == NODQUOT)
     77  1.7.4.2  yamt 				continue;
     78  1.7.4.2  yamt 			mutex_enter(&dq->dq_interlock);
     79  1.7.4.2  yamt 			ncurblocks = dq->dq_curblocks + change;
     80  1.7.4.2  yamt 			if (ncurblocks >= 0)
     81  1.7.4.2  yamt 				dq->dq_curblocks = ncurblocks;
     82  1.7.4.2  yamt 			else
     83  1.7.4.2  yamt 				dq->dq_curblocks = 0;
     84  1.7.4.2  yamt 			dq->dq_flags &= ~DQ_WARN(QL_BLOCK);
     85  1.7.4.2  yamt 			dq->dq_flags |= DQ_MOD;
     86  1.7.4.2  yamt 			mutex_exit(&dq->dq_interlock);
     87  1.7.4.2  yamt 		}
     88  1.7.4.2  yamt 		return (0);
     89  1.7.4.2  yamt 	}
     90  1.7.4.2  yamt 	for (i = 0; i < ULFS_MAXQUOTAS; i++) {
     91  1.7.4.2  yamt 		if ((dq = ip->i_dquot[i]) == NODQUOT)
     92  1.7.4.2  yamt 			continue;
     93  1.7.4.2  yamt 		if ((flags & FORCE) == 0 &&
     94  1.7.4.2  yamt 		    kauth_authorize_system(cred, KAUTH_SYSTEM_FS_QUOTA,
     95  1.7.4.2  yamt 		    KAUTH_REQ_SYSTEM_FS_QUOTA_NOLIMIT, KAUTH_ARG(i),
     96  1.7.4.2  yamt 		    KAUTH_ARG(QL_BLOCK), NULL) != 0) {
     97  1.7.4.2  yamt 			mutex_enter(&dq->dq_interlock);
     98  1.7.4.2  yamt 			error = chkdqchg(ip, change, cred, i);
     99  1.7.4.2  yamt 			mutex_exit(&dq->dq_interlock);
    100  1.7.4.2  yamt 			if (error != 0)
    101  1.7.4.2  yamt 				return (error);
    102  1.7.4.2  yamt 		}
    103  1.7.4.2  yamt 	}
    104  1.7.4.2  yamt 	for (i = 0; i < ULFS_MAXQUOTAS; i++) {
    105  1.7.4.2  yamt 		if ((dq = ip->i_dquot[i]) == NODQUOT)
    106  1.7.4.2  yamt 			continue;
    107  1.7.4.2  yamt 		mutex_enter(&dq->dq_interlock);
    108  1.7.4.2  yamt 		dq->dq_curblocks += change;
    109  1.7.4.2  yamt 		dq->dq_flags |= DQ_MOD;
    110  1.7.4.2  yamt 		mutex_exit(&dq->dq_interlock);
    111  1.7.4.2  yamt 	}
    112  1.7.4.2  yamt 	return (0);
    113  1.7.4.2  yamt }
    114  1.7.4.2  yamt 
    115  1.7.4.2  yamt /*
    116  1.7.4.2  yamt  * Check for a valid change to a users allocation.
    117  1.7.4.2  yamt  * Issue an error message if appropriate.
    118  1.7.4.2  yamt  */
    119  1.7.4.2  yamt static int
    120  1.7.4.2  yamt chkdqchg(struct inode *ip, int64_t change, kauth_cred_t cred, int type)
    121  1.7.4.2  yamt {
    122  1.7.4.2  yamt 	struct dquot *dq = ip->i_dquot[type];
    123  1.7.4.2  yamt 	long ncurblocks = dq->dq_curblocks + change;
    124  1.7.4.2  yamt 
    125  1.7.4.2  yamt 	KASSERT(mutex_owned(&dq->dq_interlock));
    126  1.7.4.2  yamt 	/*
    127  1.7.4.2  yamt 	 * If user would exceed their hard limit, disallow space allocation.
    128  1.7.4.2  yamt 	 */
    129  1.7.4.2  yamt 	if (ncurblocks >= dq->dq_bhardlimit && dq->dq_bhardlimit) {
    130  1.7.4.2  yamt 		if ((dq->dq_flags & DQ_WARN(QL_BLOCK)) == 0 &&
    131  1.7.4.2  yamt 		    ip->i_uid == kauth_cred_geteuid(cred)) {
    132  1.7.4.2  yamt 			uprintf("\n%s: write failed, %s disk limit reached\n",
    133  1.7.4.2  yamt 			    ITOV(ip)->v_mount->mnt_stat.f_mntonname,
    134  1.7.4.2  yamt 			    lfs_quotatypes[type]);
    135  1.7.4.2  yamt 			dq->dq_flags |= DQ_WARN(QL_BLOCK);
    136  1.7.4.2  yamt 		}
    137  1.7.4.2  yamt 		return (EDQUOT);
    138  1.7.4.2  yamt 	}
    139  1.7.4.2  yamt 	/*
    140  1.7.4.2  yamt 	 * If user is over their soft limit for too long, disallow space
    141  1.7.4.2  yamt 	 * allocation. Reset time limit as they cross their soft limit.
    142  1.7.4.2  yamt 	 */
    143  1.7.4.2  yamt 	if (ncurblocks >= dq->dq_bsoftlimit && dq->dq_bsoftlimit) {
    144  1.7.4.2  yamt 		if (dq->dq_curblocks < dq->dq_bsoftlimit) {
    145  1.7.4.2  yamt 			dq->dq_btime =
    146  1.7.4.2  yamt 			    time_second + ip->i_ump->umq1_btime[type];
    147  1.7.4.2  yamt 			if (ip->i_uid == kauth_cred_geteuid(cred))
    148  1.7.4.2  yamt 				uprintf("\n%s: warning, %s %s\n",
    149  1.7.4.2  yamt 				    ITOV(ip)->v_mount->mnt_stat.f_mntonname,
    150  1.7.4.2  yamt 				    lfs_quotatypes[type], "disk quota exceeded");
    151  1.7.4.2  yamt 			return (0);
    152  1.7.4.2  yamt 		}
    153  1.7.4.2  yamt 		if (time_second > dq->dq_btime) {
    154  1.7.4.2  yamt 			if ((dq->dq_flags & DQ_WARN(QL_BLOCK)) == 0 &&
    155  1.7.4.2  yamt 			    ip->i_uid == kauth_cred_geteuid(cred)) {
    156  1.7.4.2  yamt 				uprintf("\n%s: write failed, %s %s\n",
    157  1.7.4.2  yamt 				    ITOV(ip)->v_mount->mnt_stat.f_mntonname,
    158  1.7.4.2  yamt 				    lfs_quotatypes[type],
    159  1.7.4.2  yamt 				    "disk quota exceeded for too long");
    160  1.7.4.2  yamt 				dq->dq_flags |= DQ_WARN(QL_BLOCK);
    161  1.7.4.2  yamt 			}
    162  1.7.4.2  yamt 			return (EDQUOT);
    163  1.7.4.2  yamt 		}
    164  1.7.4.2  yamt 	}
    165  1.7.4.2  yamt 	return (0);
    166  1.7.4.2  yamt }
    167  1.7.4.2  yamt 
    168  1.7.4.2  yamt /*
    169  1.7.4.2  yamt  * Check the inode limit, applying corrective action.
    170  1.7.4.2  yamt  */
    171  1.7.4.2  yamt int
    172  1.7.4.2  yamt lfs_chkiq1(struct inode *ip, int32_t change, kauth_cred_t cred, int flags)
    173  1.7.4.2  yamt {
    174  1.7.4.2  yamt 	struct dquot *dq;
    175  1.7.4.2  yamt 	int i;
    176  1.7.4.2  yamt 	int ncurinodes, error;
    177  1.7.4.2  yamt 
    178  1.7.4.2  yamt 	if ((error = lfs_getinoquota(ip)) != 0)
    179  1.7.4.2  yamt 		return error;
    180  1.7.4.2  yamt 	if (change == 0)
    181  1.7.4.2  yamt 		return (0);
    182  1.7.4.2  yamt 	if (change < 0) {
    183  1.7.4.2  yamt 		for (i = 0; i < ULFS_MAXQUOTAS; i++) {
    184  1.7.4.2  yamt 			if ((dq = ip->i_dquot[i]) == NODQUOT)
    185  1.7.4.2  yamt 				continue;
    186  1.7.4.2  yamt 			mutex_enter(&dq->dq_interlock);
    187  1.7.4.2  yamt 			ncurinodes = dq->dq_curinodes + change;
    188  1.7.4.2  yamt 			if (ncurinodes >= 0)
    189  1.7.4.2  yamt 				dq->dq_curinodes = ncurinodes;
    190  1.7.4.2  yamt 			else
    191  1.7.4.2  yamt 				dq->dq_curinodes = 0;
    192  1.7.4.2  yamt 			dq->dq_flags &= ~DQ_WARN(QL_FILE);
    193  1.7.4.2  yamt 			dq->dq_flags |= DQ_MOD;
    194  1.7.4.2  yamt 			mutex_exit(&dq->dq_interlock);
    195  1.7.4.2  yamt 		}
    196  1.7.4.2  yamt 		return (0);
    197  1.7.4.2  yamt 	}
    198  1.7.4.2  yamt 	for (i = 0; i < ULFS_MAXQUOTAS; i++) {
    199  1.7.4.2  yamt 		if ((dq = ip->i_dquot[i]) == NODQUOT)
    200  1.7.4.2  yamt 			continue;
    201  1.7.4.2  yamt 		if ((flags & FORCE) == 0 && kauth_authorize_system(cred,
    202  1.7.4.2  yamt 		    KAUTH_SYSTEM_FS_QUOTA, KAUTH_REQ_SYSTEM_FS_QUOTA_NOLIMIT,
    203  1.7.4.2  yamt 		    KAUTH_ARG(i), KAUTH_ARG(QL_FILE), NULL) != 0) {
    204  1.7.4.2  yamt 			mutex_enter(&dq->dq_interlock);
    205  1.7.4.2  yamt 			error = chkiqchg(ip, change, cred, i);
    206  1.7.4.2  yamt 			mutex_exit(&dq->dq_interlock);
    207  1.7.4.2  yamt 			if (error != 0)
    208  1.7.4.2  yamt 				return (error);
    209  1.7.4.2  yamt 		}
    210  1.7.4.2  yamt 	}
    211  1.7.4.2  yamt 	for (i = 0; i < ULFS_MAXQUOTAS; i++) {
    212  1.7.4.2  yamt 		if ((dq = ip->i_dquot[i]) == NODQUOT)
    213  1.7.4.2  yamt 			continue;
    214  1.7.4.2  yamt 		mutex_enter(&dq->dq_interlock);
    215  1.7.4.2  yamt 		dq->dq_curinodes += change;
    216  1.7.4.2  yamt 		dq->dq_flags |= DQ_MOD;
    217  1.7.4.2  yamt 		mutex_exit(&dq->dq_interlock);
    218  1.7.4.2  yamt 	}
    219  1.7.4.2  yamt 	return (0);
    220  1.7.4.2  yamt }
    221  1.7.4.2  yamt 
    222  1.7.4.2  yamt /*
    223  1.7.4.2  yamt  * Check for a valid change to a users allocation.
    224  1.7.4.2  yamt  * Issue an error message if appropriate.
    225  1.7.4.2  yamt  */
    226  1.7.4.2  yamt static int
    227  1.7.4.2  yamt chkiqchg(struct inode *ip, int32_t change, kauth_cred_t cred, int type)
    228  1.7.4.2  yamt {
    229  1.7.4.2  yamt 	struct dquot *dq = ip->i_dquot[type];
    230  1.7.4.2  yamt 	long ncurinodes = dq->dq_curinodes + change;
    231  1.7.4.2  yamt 
    232  1.7.4.2  yamt 	KASSERT(mutex_owned(&dq->dq_interlock));
    233  1.7.4.2  yamt 	/*
    234  1.7.4.2  yamt 	 * If user would exceed their hard limit, disallow inode allocation.
    235  1.7.4.2  yamt 	 */
    236  1.7.4.2  yamt 	if (ncurinodes >= dq->dq_ihardlimit && dq->dq_ihardlimit) {
    237  1.7.4.2  yamt 		if ((dq->dq_flags & DQ_WARN(QL_FILE)) == 0 &&
    238  1.7.4.2  yamt 		    ip->i_uid == kauth_cred_geteuid(cred)) {
    239  1.7.4.2  yamt 			uprintf("\n%s: write failed, %s inode limit reached\n",
    240  1.7.4.2  yamt 			    ITOV(ip)->v_mount->mnt_stat.f_mntonname,
    241  1.7.4.2  yamt 			    lfs_quotatypes[type]);
    242  1.7.4.2  yamt 			dq->dq_flags |= DQ_WARN(QL_FILE);
    243  1.7.4.2  yamt 		}
    244  1.7.4.2  yamt 		return (EDQUOT);
    245  1.7.4.2  yamt 	}
    246  1.7.4.2  yamt 	/*
    247  1.7.4.2  yamt 	 * If user is over their soft limit for too long, disallow inode
    248  1.7.4.2  yamt 	 * allocation. Reset time limit as they cross their soft limit.
    249  1.7.4.2  yamt 	 */
    250  1.7.4.2  yamt 	if (ncurinodes >= dq->dq_isoftlimit && dq->dq_isoftlimit) {
    251  1.7.4.2  yamt 		if (dq->dq_curinodes < dq->dq_isoftlimit) {
    252  1.7.4.2  yamt 			dq->dq_itime =
    253  1.7.4.2  yamt 			    time_second + ip->i_ump->umq1_itime[type];
    254  1.7.4.2  yamt 			if (ip->i_uid == kauth_cred_geteuid(cred))
    255  1.7.4.2  yamt 				uprintf("\n%s: warning, %s %s\n",
    256  1.7.4.2  yamt 				    ITOV(ip)->v_mount->mnt_stat.f_mntonname,
    257  1.7.4.2  yamt 				    lfs_quotatypes[type], "inode quota exceeded");
    258  1.7.4.2  yamt 			return (0);
    259  1.7.4.2  yamt 		}
    260  1.7.4.2  yamt 		if (time_second > dq->dq_itime) {
    261  1.7.4.2  yamt 			if ((dq->dq_flags & DQ_WARN(QL_FILE)) == 0 &&
    262  1.7.4.2  yamt 			    ip->i_uid == kauth_cred_geteuid(cred)) {
    263  1.7.4.2  yamt 				uprintf("\n%s: write failed, %s %s\n",
    264  1.7.4.2  yamt 				    ITOV(ip)->v_mount->mnt_stat.f_mntonname,
    265  1.7.4.2  yamt 				    lfs_quotatypes[type],
    266  1.7.4.2  yamt 				    "inode quota exceeded for too long");
    267  1.7.4.2  yamt 				dq->dq_flags |= DQ_WARN(QL_FILE);
    268  1.7.4.2  yamt 			}
    269  1.7.4.2  yamt 			return (EDQUOT);
    270  1.7.4.2  yamt 		}
    271  1.7.4.2  yamt 	}
    272  1.7.4.2  yamt 	return (0);
    273  1.7.4.2  yamt }
    274  1.7.4.2  yamt 
    275  1.7.4.2  yamt int
    276  1.7.4.2  yamt lfsquota1_umount(struct mount *mp, int flags)
    277  1.7.4.2  yamt {
    278  1.7.4.2  yamt 	int i, error;
    279  1.7.4.2  yamt 	struct ulfsmount *ump = VFSTOULFS(mp);
    280  1.7.4.2  yamt 	struct lfs *fs = ump->um_lfs;
    281  1.7.4.2  yamt 	struct lwp *l = curlwp;
    282  1.7.4.2  yamt 
    283  1.7.4.2  yamt 	if ((fs->um_flags & ULFS_QUOTA) == 0)
    284  1.7.4.2  yamt 		return 0;
    285  1.7.4.2  yamt 
    286  1.7.4.2  yamt 	if ((error = vflush(mp, NULLVP, SKIPSYSTEM | flags)) != 0)
    287  1.7.4.2  yamt 		return (error);
    288  1.7.4.2  yamt 
    289  1.7.4.2  yamt 	for (i = 0; i < ULFS_MAXQUOTAS; i++) {
    290  1.7.4.2  yamt 		if (ump->um_quotas[i] != NULLVP) {
    291  1.7.4.2  yamt 			lfsquota1_handle_cmd_quotaoff(l, ump, i);
    292  1.7.4.2  yamt 		}
    293  1.7.4.2  yamt 	}
    294  1.7.4.2  yamt 	return 0;
    295  1.7.4.2  yamt }
    296  1.7.4.2  yamt 
    297  1.7.4.2  yamt /*
    298  1.7.4.2  yamt  * Code to process quotactl commands.
    299  1.7.4.2  yamt  */
    300  1.7.4.2  yamt 
    301  1.7.4.2  yamt /*
    302  1.7.4.2  yamt  * set up a quota file for a particular file system.
    303  1.7.4.2  yamt  */
    304  1.7.4.2  yamt int
    305  1.7.4.2  yamt lfsquota1_handle_cmd_quotaon(struct lwp *l, struct ulfsmount *ump, int type,
    306  1.7.4.2  yamt     const char *fname)
    307  1.7.4.2  yamt {
    308  1.7.4.2  yamt 	struct mount *mp = ump->um_mountp;
    309  1.7.4.2  yamt 	struct lfs *fs = ump->um_lfs;
    310  1.7.4.2  yamt 	struct vnode *vp, **vpp;
    311  1.7.4.2  yamt 	struct vnode_iterator *marker;
    312  1.7.4.2  yamt 	struct dquot *dq;
    313  1.7.4.2  yamt 	int error;
    314  1.7.4.2  yamt 	struct pathbuf *pb;
    315  1.7.4.2  yamt 	struct nameidata nd;
    316  1.7.4.2  yamt 
    317  1.7.4.2  yamt 	if (fs->um_flags & ULFS_QUOTA2) {
    318  1.7.4.2  yamt 		uprintf("%s: quotas v2 already enabled\n",
    319  1.7.4.2  yamt 		    mp->mnt_stat.f_mntonname);
    320  1.7.4.2  yamt 		return (EBUSY);
    321  1.7.4.2  yamt 	}
    322  1.7.4.2  yamt 
    323  1.7.4.2  yamt 	vpp = &ump->um_quotas[type];
    324  1.7.4.2  yamt 
    325  1.7.4.2  yamt 	pb = pathbuf_create(fname);
    326  1.7.4.2  yamt 	if (pb == NULL) {
    327  1.7.4.2  yamt 		return ENOMEM;
    328  1.7.4.2  yamt 	}
    329  1.7.4.2  yamt 	NDINIT(&nd, LOOKUP, FOLLOW, pb);
    330  1.7.4.2  yamt 	if ((error = vn_open(&nd, FREAD|FWRITE, 0)) != 0) {
    331  1.7.4.2  yamt 		pathbuf_destroy(pb);
    332  1.7.4.2  yamt 		return error;
    333  1.7.4.2  yamt 	}
    334  1.7.4.2  yamt 	vp = nd.ni_vp;
    335  1.7.4.2  yamt 	pathbuf_destroy(pb);
    336  1.7.4.2  yamt 
    337  1.7.4.2  yamt 	VOP_UNLOCK(vp);
    338  1.7.4.2  yamt 	if (vp->v_type != VREG) {
    339  1.7.4.2  yamt 		(void) vn_close(vp, FREAD|FWRITE, l->l_cred);
    340  1.7.4.2  yamt 		return (EACCES);
    341  1.7.4.2  yamt 	}
    342  1.7.4.2  yamt 	if (*vpp != vp)
    343  1.7.4.2  yamt 		lfsquota1_handle_cmd_quotaoff(l, ump, type);
    344  1.7.4.2  yamt 	mutex_enter(&lfs_dqlock);
    345  1.7.4.2  yamt 	while ((ump->umq1_qflags[type] & (QTF_CLOSING | QTF_OPENING)) != 0)
    346  1.7.4.2  yamt 		cv_wait(&lfs_dqcv, &lfs_dqlock);
    347  1.7.4.2  yamt 	ump->umq1_qflags[type] |= QTF_OPENING;
    348  1.7.4.2  yamt 	mutex_exit(&lfs_dqlock);
    349  1.7.4.2  yamt 	mp->mnt_flag |= MNT_QUOTA;
    350  1.7.4.2  yamt 	vp->v_vflag |= VV_SYSTEM;	/* XXXSMP */
    351  1.7.4.2  yamt 	*vpp = vp;
    352  1.7.4.2  yamt 	/*
    353  1.7.4.2  yamt 	 * Save the credential of the process that turned on quotas.
    354  1.7.4.2  yamt 	 * Set up the time limits for this quota.
    355  1.7.4.2  yamt 	 */
    356  1.7.4.2  yamt 	kauth_cred_hold(l->l_cred);
    357  1.7.4.2  yamt 	ump->um_cred[type] = l->l_cred;
    358  1.7.4.2  yamt 	ump->umq1_btime[type] = MAX_DQ_TIME;
    359  1.7.4.2  yamt 	ump->umq1_itime[type] = MAX_IQ_TIME;
    360  1.7.4.2  yamt 	if (lfs_dqget(NULLVP, 0, ump, type, &dq) == 0) {
    361  1.7.4.2  yamt 		if (dq->dq_btime > 0)
    362  1.7.4.2  yamt 			ump->umq1_btime[type] = dq->dq_btime;
    363  1.7.4.2  yamt 		if (dq->dq_itime > 0)
    364  1.7.4.2  yamt 			ump->umq1_itime[type] = dq->dq_itime;
    365  1.7.4.2  yamt 		lfs_dqrele(NULLVP, dq);
    366  1.7.4.2  yamt 	}
    367  1.7.4.2  yamt 	/*
    368  1.7.4.2  yamt 	 * Search vnodes associated with this mount point,
    369  1.7.4.2  yamt 	 * adding references to quota file being opened.
    370  1.7.4.2  yamt 	 * NB: only need to add dquot's for inodes being modified.
    371  1.7.4.2  yamt 	 */
    372  1.7.4.2  yamt 	vfs_vnode_iterator_init(mp, &marker);
    373  1.7.4.2  yamt 	while (vfs_vnode_iterator_next(marker, &vp)) {
    374  1.7.4.2  yamt 		error = vn_lock(vp, LK_EXCLUSIVE);
    375  1.7.4.2  yamt 		if (error) {
    376  1.7.4.2  yamt 			vrele(vp);
    377  1.7.4.2  yamt 			continue;
    378  1.7.4.2  yamt 		}
    379  1.7.4.2  yamt 		mutex_enter(vp->v_interlock);
    380  1.7.4.2  yamt 		if (VTOI(vp) == NULL || vp->v_type == VNON ||
    381  1.7.4.2  yamt 		    vp->v_writecount == 0) {
    382  1.7.4.2  yamt 			mutex_exit(vp->v_interlock);
    383  1.7.4.2  yamt 			vput(vp);
    384  1.7.4.2  yamt 			continue;
    385  1.7.4.2  yamt 		}
    386  1.7.4.2  yamt 		mutex_exit(vp->v_interlock);
    387  1.7.4.2  yamt 		if ((error = lfs_getinoquota(VTOI(vp))) != 0) {
    388  1.7.4.2  yamt 			vput(vp);
    389  1.7.4.2  yamt 			break;
    390  1.7.4.2  yamt 		}
    391  1.7.4.2  yamt 		vput(vp);
    392  1.7.4.2  yamt 	}
    393  1.7.4.2  yamt 	vfs_vnode_iterator_destroy(marker);
    394  1.7.4.2  yamt 
    395  1.7.4.2  yamt 	mutex_enter(&lfs_dqlock);
    396  1.7.4.2  yamt 	ump->umq1_qflags[type] &= ~QTF_OPENING;
    397  1.7.4.2  yamt 	cv_broadcast(&lfs_dqcv);
    398  1.7.4.2  yamt 	if (error == 0)
    399  1.7.4.2  yamt 		fs->um_flags |= ULFS_QUOTA;
    400  1.7.4.2  yamt 	mutex_exit(&lfs_dqlock);
    401  1.7.4.2  yamt 	if (error)
    402  1.7.4.2  yamt 		lfsquota1_handle_cmd_quotaoff(l, ump, type);
    403  1.7.4.2  yamt 	return (error);
    404  1.7.4.2  yamt }
    405  1.7.4.2  yamt 
    406  1.7.4.2  yamt /*
    407  1.7.4.2  yamt  * turn off disk quotas for a filesystem.
    408  1.7.4.2  yamt  */
    409  1.7.4.2  yamt int
    410  1.7.4.2  yamt lfsquota1_handle_cmd_quotaoff(struct lwp *l, struct ulfsmount *ump, int type)
    411  1.7.4.2  yamt {
    412  1.7.4.2  yamt 	struct mount *mp = ump->um_mountp;
    413  1.7.4.2  yamt 	struct lfs *fs = ump->um_lfs;
    414  1.7.4.2  yamt 	struct vnode *vp;
    415  1.7.4.2  yamt 	struct vnode *qvp;
    416  1.7.4.2  yamt 	struct vnode_iterator *marker;
    417  1.7.4.2  yamt 	struct dquot *dq;
    418  1.7.4.2  yamt 	struct inode *ip;
    419  1.7.4.2  yamt 	kauth_cred_t cred;
    420  1.7.4.2  yamt 	int i, error;
    421  1.7.4.2  yamt 
    422  1.7.4.2  yamt 	mutex_enter(&lfs_dqlock);
    423  1.7.4.2  yamt 	while ((ump->umq1_qflags[type] & (QTF_CLOSING | QTF_OPENING)) != 0)
    424  1.7.4.2  yamt 		cv_wait(&lfs_dqcv, &lfs_dqlock);
    425  1.7.4.2  yamt 	if ((qvp = ump->um_quotas[type]) == NULLVP) {
    426  1.7.4.2  yamt 		mutex_exit(&lfs_dqlock);
    427  1.7.4.2  yamt 		return (0);
    428  1.7.4.2  yamt 	}
    429  1.7.4.2  yamt 	ump->umq1_qflags[type] |= QTF_CLOSING;
    430  1.7.4.2  yamt 	fs->um_flags &= ~ULFS_QUOTA;
    431  1.7.4.2  yamt 	mutex_exit(&lfs_dqlock);
    432  1.7.4.2  yamt 	/*
    433  1.7.4.2  yamt 	 * Search vnodes associated with this mount point,
    434  1.7.4.2  yamt 	 * deleting any references to quota file being closed.
    435  1.7.4.2  yamt 	 */
    436  1.7.4.2  yamt 	vfs_vnode_iterator_init(mp, &marker);
    437  1.7.4.2  yamt 	while (vfs_vnode_iterator_next(marker, &vp)) {
    438  1.7.4.2  yamt 		error = vn_lock(vp, LK_EXCLUSIVE);
    439  1.7.4.2  yamt 		if (error) {
    440  1.7.4.2  yamt 			vrele(vp);
    441  1.7.4.2  yamt 			continue;
    442  1.7.4.2  yamt 		}
    443  1.7.4.2  yamt 		ip = VTOI(vp);
    444  1.7.4.2  yamt 		if (ip == NULL || vp->v_type == VNON) {
    445  1.7.4.2  yamt 			vput(vp);
    446  1.7.4.2  yamt 			continue;
    447  1.7.4.2  yamt 		}
    448  1.7.4.2  yamt 		dq = ip->i_dquot[type];
    449  1.7.4.2  yamt 		ip->i_dquot[type] = NODQUOT;
    450  1.7.4.2  yamt 		lfs_dqrele(vp, dq);
    451  1.7.4.2  yamt 		vput(vp);
    452  1.7.4.2  yamt 	}
    453  1.7.4.2  yamt 	vfs_vnode_iterator_destroy(marker);
    454  1.7.4.2  yamt #ifdef DIAGNOSTIC
    455  1.7.4.2  yamt 	lfs_dqflush(qvp);
    456  1.7.4.2  yamt #endif
    457  1.7.4.2  yamt 	qvp->v_vflag &= ~VV_SYSTEM;
    458  1.7.4.2  yamt 	error = vn_close(qvp, FREAD|FWRITE, l->l_cred);
    459  1.7.4.2  yamt 	mutex_enter(&lfs_dqlock);
    460  1.7.4.2  yamt 	ump->um_quotas[type] = NULLVP;
    461  1.7.4.2  yamt 	cred = ump->um_cred[type];
    462  1.7.4.2  yamt 	ump->um_cred[type] = NOCRED;
    463  1.7.4.2  yamt 	for (i = 0; i < ULFS_MAXQUOTAS; i++)
    464  1.7.4.2  yamt 		if (ump->um_quotas[i] != NULLVP)
    465  1.7.4.2  yamt 			break;
    466  1.7.4.2  yamt 	ump->umq1_qflags[type] &= ~QTF_CLOSING;
    467  1.7.4.2  yamt 	cv_broadcast(&lfs_dqcv);
    468  1.7.4.2  yamt 	mutex_exit(&lfs_dqlock);
    469  1.7.4.2  yamt 	kauth_cred_free(cred);
    470  1.7.4.2  yamt 	if (i == ULFS_MAXQUOTAS)
    471  1.7.4.2  yamt 		mp->mnt_flag &= ~MNT_QUOTA;
    472  1.7.4.2  yamt 	return (error);
    473  1.7.4.2  yamt }
    474  1.7.4.2  yamt 
    475  1.7.4.2  yamt int
    476  1.7.4.2  yamt lfsquota1_handle_cmd_get(struct ulfsmount *ump, const struct quotakey *qk,
    477  1.7.4.2  yamt     struct quotaval *qv)
    478  1.7.4.2  yamt {
    479  1.7.4.2  yamt 	struct dquot *dq;
    480  1.7.4.2  yamt 	int error;
    481  1.7.4.2  yamt 	struct quotaval blocks, files;
    482  1.7.4.2  yamt 	int idtype;
    483  1.7.4.2  yamt 	id_t id;
    484  1.7.4.2  yamt 
    485  1.7.4.2  yamt 	idtype = qk->qk_idtype;
    486  1.7.4.2  yamt 	id = qk->qk_id;
    487  1.7.4.2  yamt 
    488  1.7.4.2  yamt 	if (ump->um_quotas[idtype] == NULLVP)
    489  1.7.4.2  yamt 		return ENODEV;
    490  1.7.4.2  yamt 
    491  1.7.4.2  yamt 	if (id == QUOTA_DEFAULTID) { /* we want the grace period of id 0 */
    492  1.7.4.2  yamt 		if ((error = lfs_dqget(NULLVP, 0, ump, idtype, &dq)) != 0)
    493  1.7.4.2  yamt 			return error;
    494  1.7.4.2  yamt 
    495  1.7.4.2  yamt 	} else {
    496  1.7.4.2  yamt 		if ((error = lfs_dqget(NULLVP, id, ump, idtype, &dq)) != 0)
    497  1.7.4.2  yamt 			return error;
    498  1.7.4.2  yamt 	}
    499  1.7.4.2  yamt 	lfs_dqblk_to_quotavals(&dq->dq_un.dq1_dqb, &blocks, &files);
    500  1.7.4.2  yamt 	lfs_dqrele(NULLVP, dq);
    501  1.7.4.2  yamt 	if (id == QUOTA_DEFAULTID) {
    502  1.7.4.2  yamt 		if (blocks.qv_expiretime > 0)
    503  1.7.4.2  yamt 			blocks.qv_grace = blocks.qv_expiretime;
    504  1.7.4.2  yamt 		else
    505  1.7.4.2  yamt 			blocks.qv_grace = MAX_DQ_TIME;
    506  1.7.4.2  yamt 		if (files.qv_expiretime > 0)
    507  1.7.4.2  yamt 			files.qv_grace = files.qv_expiretime;
    508  1.7.4.2  yamt 		else
    509  1.7.4.2  yamt 			files.qv_grace = MAX_DQ_TIME;
    510  1.7.4.2  yamt 	}
    511  1.7.4.2  yamt 
    512  1.7.4.2  yamt 	switch (qk->qk_objtype) {
    513  1.7.4.2  yamt 	    case QUOTA_OBJTYPE_BLOCKS:
    514  1.7.4.2  yamt 		*qv = blocks;
    515  1.7.4.2  yamt 		break;
    516  1.7.4.2  yamt 	    case QUOTA_OBJTYPE_FILES:
    517  1.7.4.2  yamt 		*qv = files;
    518  1.7.4.2  yamt 		break;
    519  1.7.4.2  yamt 	    default:
    520  1.7.4.2  yamt 		return EINVAL;
    521  1.7.4.2  yamt 	}
    522  1.7.4.2  yamt 
    523  1.7.4.2  yamt 	return 0;
    524  1.7.4.2  yamt }
    525  1.7.4.2  yamt 
    526  1.7.4.2  yamt static uint32_t
    527  1.7.4.2  yamt quota1_encode_limit(uint64_t lim)
    528  1.7.4.2  yamt {
    529  1.7.4.2  yamt 	if (lim == QUOTA_NOLIMIT || lim >= 0xffffffff) {
    530  1.7.4.2  yamt 		return 0;
    531  1.7.4.2  yamt 	}
    532  1.7.4.2  yamt 	return lim;
    533  1.7.4.2  yamt }
    534  1.7.4.2  yamt 
    535  1.7.4.2  yamt int
    536  1.7.4.2  yamt lfsquota1_handle_cmd_put(struct ulfsmount *ump, const struct quotakey *key,
    537  1.7.4.2  yamt     const struct quotaval *val)
    538  1.7.4.2  yamt {
    539  1.7.4.2  yamt 	struct dquot *dq;
    540  1.7.4.2  yamt 	struct dqblk dqb;
    541  1.7.4.2  yamt 	int error;
    542  1.7.4.2  yamt 
    543  1.7.4.2  yamt 	switch (key->qk_idtype) {
    544  1.7.4.2  yamt 	    case QUOTA_IDTYPE_USER:
    545  1.7.4.2  yamt 	    case QUOTA_IDTYPE_GROUP:
    546  1.7.4.2  yamt 		break;
    547  1.7.4.2  yamt 	    default:
    548  1.7.4.2  yamt 		return EINVAL;
    549  1.7.4.2  yamt 	}
    550  1.7.4.2  yamt 
    551  1.7.4.2  yamt 	switch (key->qk_objtype) {
    552  1.7.4.2  yamt 	    case QUOTA_OBJTYPE_BLOCKS:
    553  1.7.4.2  yamt 	    case QUOTA_OBJTYPE_FILES:
    554  1.7.4.2  yamt 		break;
    555  1.7.4.2  yamt 	    default:
    556  1.7.4.2  yamt 		return EINVAL;
    557  1.7.4.2  yamt 	}
    558  1.7.4.2  yamt 
    559  1.7.4.2  yamt 	if (ump->um_quotas[key->qk_idtype] == NULLVP)
    560  1.7.4.2  yamt 		return ENODEV;
    561  1.7.4.2  yamt 
    562  1.7.4.2  yamt 	if (key->qk_id == QUOTA_DEFAULTID) {
    563  1.7.4.2  yamt 		/* just update grace times */
    564  1.7.4.2  yamt 		id_t id = 0;
    565  1.7.4.2  yamt 
    566  1.7.4.2  yamt 		if ((error = lfs_dqget(NULLVP, id, ump, key->qk_idtype, &dq)) != 0)
    567  1.7.4.2  yamt 			return error;
    568  1.7.4.2  yamt 		mutex_enter(&dq->dq_interlock);
    569  1.7.4.2  yamt 		if (val->qv_grace != QUOTA_NOTIME) {
    570  1.7.4.2  yamt 			if (key->qk_objtype == QUOTA_OBJTYPE_BLOCKS)
    571  1.7.4.2  yamt 				ump->umq1_btime[key->qk_idtype] = dq->dq_btime =
    572  1.7.4.2  yamt 					val->qv_grace;
    573  1.7.4.2  yamt 			if (key->qk_objtype == QUOTA_OBJTYPE_FILES)
    574  1.7.4.2  yamt 				ump->umq1_itime[key->qk_idtype] = dq->dq_itime =
    575  1.7.4.2  yamt 					val->qv_grace;
    576  1.7.4.2  yamt 		}
    577  1.7.4.2  yamt 		dq->dq_flags |= DQ_MOD;
    578  1.7.4.2  yamt 		mutex_exit(&dq->dq_interlock);
    579  1.7.4.2  yamt 		lfs_dqrele(NULLVP, dq);
    580  1.7.4.2  yamt 		return 0;
    581  1.7.4.2  yamt 	}
    582  1.7.4.2  yamt 
    583  1.7.4.2  yamt 	if ((error = lfs_dqget(NULLVP, key->qk_id, ump, key->qk_idtype, &dq)) != 0)
    584  1.7.4.2  yamt 		return (error);
    585  1.7.4.2  yamt 	mutex_enter(&dq->dq_interlock);
    586  1.7.4.2  yamt 	/*
    587  1.7.4.2  yamt 	 * Copy all but the current values.
    588  1.7.4.2  yamt 	 * Reset time limit if previously had no soft limit or were
    589  1.7.4.2  yamt 	 * under it, but now have a soft limit and are over it.
    590  1.7.4.2  yamt 	 */
    591  1.7.4.2  yamt 	dqb.dqb_curblocks = dq->dq_curblocks;
    592  1.7.4.2  yamt 	dqb.dqb_curinodes = dq->dq_curinodes;
    593  1.7.4.2  yamt 	dqb.dqb_btime = dq->dq_btime;
    594  1.7.4.2  yamt 	dqb.dqb_itime = dq->dq_itime;
    595  1.7.4.2  yamt 	if (key->qk_objtype == QUOTA_OBJTYPE_BLOCKS) {
    596  1.7.4.2  yamt 		dqb.dqb_bsoftlimit = quota1_encode_limit(val->qv_softlimit);
    597  1.7.4.2  yamt 		dqb.dqb_bhardlimit = quota1_encode_limit(val->qv_hardlimit);
    598  1.7.4.2  yamt 		dqb.dqb_isoftlimit = dq->dq_isoftlimit;
    599  1.7.4.2  yamt 		dqb.dqb_ihardlimit = dq->dq_ihardlimit;
    600  1.7.4.2  yamt 	} else {
    601  1.7.4.2  yamt 		KASSERT(key->qk_objtype == QUOTA_OBJTYPE_FILES);
    602  1.7.4.2  yamt 		dqb.dqb_bsoftlimit = dq->dq_bsoftlimit;
    603  1.7.4.2  yamt 		dqb.dqb_bhardlimit = dq->dq_bhardlimit;
    604  1.7.4.2  yamt 		dqb.dqb_isoftlimit = quota1_encode_limit(val->qv_softlimit);
    605  1.7.4.2  yamt 		dqb.dqb_ihardlimit = quota1_encode_limit(val->qv_hardlimit);
    606  1.7.4.2  yamt 	}
    607  1.7.4.2  yamt 	if (dq->dq_id == 0 && val->qv_grace != QUOTA_NOTIME) {
    608  1.7.4.2  yamt 		/* also update grace time if available */
    609  1.7.4.2  yamt 		if (key->qk_objtype == QUOTA_OBJTYPE_BLOCKS) {
    610  1.7.4.2  yamt 			ump->umq1_btime[key->qk_idtype] = dqb.dqb_btime =
    611  1.7.4.2  yamt 				val->qv_grace;
    612  1.7.4.2  yamt 		}
    613  1.7.4.2  yamt 		if (key->qk_objtype == QUOTA_OBJTYPE_FILES) {
    614  1.7.4.2  yamt 			ump->umq1_itime[key->qk_idtype] = dqb.dqb_itime =
    615  1.7.4.2  yamt 				val->qv_grace;
    616  1.7.4.2  yamt 		}
    617  1.7.4.2  yamt 	}
    618  1.7.4.2  yamt 	if (dqb.dqb_bsoftlimit &&
    619  1.7.4.2  yamt 	    dq->dq_curblocks >= dqb.dqb_bsoftlimit &&
    620  1.7.4.2  yamt 	    (dq->dq_bsoftlimit == 0 || dq->dq_curblocks < dq->dq_bsoftlimit))
    621  1.7.4.2  yamt 		dqb.dqb_btime = time_second + ump->umq1_btime[key->qk_idtype];
    622  1.7.4.2  yamt 	if (dqb.dqb_isoftlimit &&
    623  1.7.4.2  yamt 	    dq->dq_curinodes >= dqb.dqb_isoftlimit &&
    624  1.7.4.2  yamt 	    (dq->dq_isoftlimit == 0 || dq->dq_curinodes < dq->dq_isoftlimit))
    625  1.7.4.2  yamt 		dqb.dqb_itime = time_second + ump->umq1_itime[key->qk_idtype];
    626  1.7.4.2  yamt 	dq->dq_un.dq1_dqb = dqb;
    627  1.7.4.2  yamt 	if (dq->dq_curblocks < dq->dq_bsoftlimit)
    628  1.7.4.2  yamt 		dq->dq_flags &= ~DQ_WARN(QL_BLOCK);
    629  1.7.4.2  yamt 	if (dq->dq_curinodes < dq->dq_isoftlimit)
    630  1.7.4.2  yamt 		dq->dq_flags &= ~DQ_WARN(QL_FILE);
    631  1.7.4.2  yamt 	if (dq->dq_isoftlimit == 0 && dq->dq_bsoftlimit == 0 &&
    632  1.7.4.2  yamt 	    dq->dq_ihardlimit == 0 && dq->dq_bhardlimit == 0)
    633  1.7.4.2  yamt 		dq->dq_flags |= DQ_FAKE;
    634  1.7.4.2  yamt 	else
    635  1.7.4.2  yamt 		dq->dq_flags &= ~DQ_FAKE;
    636  1.7.4.2  yamt 	dq->dq_flags |= DQ_MOD;
    637  1.7.4.2  yamt 	mutex_exit(&dq->dq_interlock);
    638  1.7.4.2  yamt 	lfs_dqrele(NULLVP, dq);
    639  1.7.4.2  yamt 	return (0);
    640  1.7.4.2  yamt }
    641  1.7.4.2  yamt 
    642  1.7.4.2  yamt 
    643  1.7.4.2  yamt #if 0
    644  1.7.4.2  yamt /*
    645  1.7.4.2  yamt  * Q_SETQUOTA - assign an entire dqblk structure.
    646  1.7.4.2  yamt  */
    647  1.7.4.2  yamt int
    648  1.7.4.2  yamt setquota1(struct mount *mp, u_long id, int type, struct dqblk *dqb)
    649  1.7.4.2  yamt {
    650  1.7.4.2  yamt 	struct dquot *dq;
    651  1.7.4.2  yamt 	struct dquot *ndq;
    652  1.7.4.2  yamt 	struct ulfsmount *ump = VFSTOULFS(mp);
    653  1.7.4.2  yamt 
    654  1.7.4.2  yamt 
    655  1.7.4.2  yamt 	if ((error = lfs_dqget(NULLVP, id, ump, type, &ndq)) != 0)
    656  1.7.4.2  yamt 		return (error);
    657  1.7.4.2  yamt 	dq = ndq;
    658  1.7.4.2  yamt 	mutex_enter(&dq->dq_interlock);
    659  1.7.4.2  yamt 	/*
    660  1.7.4.2  yamt 	 * Copy all but the current values.
    661  1.7.4.2  yamt 	 * Reset time limit if previously had no soft limit or were
    662  1.7.4.2  yamt 	 * under it, but now have a soft limit and are over it.
    663  1.7.4.2  yamt 	 */
    664  1.7.4.2  yamt 	dqb->dqb_curblocks = dq->dq_curblocks;
    665  1.7.4.2  yamt 	dqb->dqb_curinodes = dq->dq_curinodes;
    666  1.7.4.2  yamt 	if (dq->dq_id != 0) {
    667  1.7.4.2  yamt 		dqb->dqb_btime = dq->dq_btime;
    668  1.7.4.2  yamt 		dqb->dqb_itime = dq->dq_itime;
    669  1.7.4.2  yamt 	}
    670  1.7.4.2  yamt 	if (dqb->dqb_bsoftlimit &&
    671  1.7.4.2  yamt 	    dq->dq_curblocks >= dqb->dqb_bsoftlimit &&
    672  1.7.4.2  yamt 	    (dq->dq_bsoftlimit == 0 || dq->dq_curblocks < dq->dq_bsoftlimit))
    673  1.7.4.2  yamt 		dqb->dqb_btime = time_second + ump->umq1_btime[type];
    674  1.7.4.2  yamt 	if (dqb->dqb_isoftlimit &&
    675  1.7.4.2  yamt 	    dq->dq_curinodes >= dqb->dqb_isoftlimit &&
    676  1.7.4.2  yamt 	    (dq->dq_isoftlimit == 0 || dq->dq_curinodes < dq->dq_isoftlimit))
    677  1.7.4.2  yamt 		dqb->dqb_itime = time_second + ump->umq1_itime[type];
    678  1.7.4.2  yamt 	dq->dq_un.dq1_dqb = *dqb;
    679  1.7.4.2  yamt 	if (dq->dq_curblocks < dq->dq_bsoftlimit)
    680  1.7.4.2  yamt 		dq->dq_flags &= ~DQ_WARN(QL_BLOCK);
    681  1.7.4.2  yamt 	if (dq->dq_curinodes < dq->dq_isoftlimit)
    682  1.7.4.2  yamt 		dq->dq_flags &= ~DQ_WARN(QL_FILE);
    683  1.7.4.2  yamt 	if (dq->dq_isoftlimit == 0 && dq->dq_bsoftlimit == 0 &&
    684  1.7.4.2  yamt 	    dq->dq_ihardlimit == 0 && dq->dq_bhardlimit == 0)
    685  1.7.4.2  yamt 		dq->dq_flags |= DQ_FAKE;
    686  1.7.4.2  yamt 	else
    687  1.7.4.2  yamt 		dq->dq_flags &= ~DQ_FAKE;
    688  1.7.4.2  yamt 	dq->dq_flags |= DQ_MOD;
    689  1.7.4.2  yamt 	mutex_exit(&dq->dq_interlock);
    690  1.7.4.2  yamt 	lfs_dqrele(NULLVP, dq);
    691  1.7.4.2  yamt 	return (0);
    692  1.7.4.2  yamt }
    693  1.7.4.2  yamt 
    694  1.7.4.2  yamt /*
    695  1.7.4.2  yamt  * Q_SETUSE - set current inode and block usage.
    696  1.7.4.2  yamt  */
    697  1.7.4.2  yamt int
    698  1.7.4.2  yamt setuse(struct mount *mp, u_long id, int type, void *addr)
    699  1.7.4.2  yamt {
    700  1.7.4.2  yamt 	struct dquot *dq;
    701  1.7.4.2  yamt 	struct ulfsmount *ump = VFSTOULFS(mp);
    702  1.7.4.2  yamt 	struct dquot *ndq;
    703  1.7.4.2  yamt 	struct dqblk usage;
    704  1.7.4.2  yamt 	int error;
    705  1.7.4.2  yamt 
    706  1.7.4.2  yamt 	error = copyin(addr, (void *)&usage, sizeof (struct dqblk));
    707  1.7.4.2  yamt 	if (error)
    708  1.7.4.2  yamt 		return (error);
    709  1.7.4.2  yamt 	if ((error = lfs_dqget(NULLVP, id, ump, type, &ndq)) != 0)
    710  1.7.4.2  yamt 		return (error);
    711  1.7.4.2  yamt 	dq = ndq;
    712  1.7.4.2  yamt 	mutex_enter(&dq->dq_interlock);
    713  1.7.4.2  yamt 	/*
    714  1.7.4.2  yamt 	 * Reset time limit if have a soft limit and were
    715  1.7.4.2  yamt 	 * previously under it, but are now over it.
    716  1.7.4.2  yamt 	 */
    717  1.7.4.2  yamt 	if (dq->dq_bsoftlimit && dq->dq_curblocks < dq->dq_bsoftlimit &&
    718  1.7.4.2  yamt 	    usage.dqb_curblocks >= dq->dq_bsoftlimit)
    719  1.7.4.2  yamt 		dq->dq_btime = time_second + ump->umq1_btime[type];
    720  1.7.4.2  yamt 	if (dq->dq_isoftlimit && dq->dq_curinodes < dq->dq_isoftlimit &&
    721  1.7.4.2  yamt 	    usage.dqb_curinodes >= dq->dq_isoftlimit)
    722  1.7.4.2  yamt 		dq->dq_itime = time_second + ump->umq1_itime[type];
    723  1.7.4.2  yamt 	dq->dq_curblocks = usage.dqb_curblocks;
    724  1.7.4.2  yamt 	dq->dq_curinodes = usage.dqb_curinodes;
    725  1.7.4.2  yamt 	if (dq->dq_curblocks < dq->dq_bsoftlimit)
    726  1.7.4.2  yamt 		dq->dq_flags &= ~DQ_WARN(QL_BLOCK);
    727  1.7.4.2  yamt 	if (dq->dq_curinodes < dq->dq_isoftlimit)
    728  1.7.4.2  yamt 		dq->dq_flags &= ~DQ_WARN(QL_FILE);
    729  1.7.4.2  yamt 	dq->dq_flags |= DQ_MOD;
    730  1.7.4.2  yamt 	mutex_exit(&dq->dq_interlock);
    731  1.7.4.2  yamt 	lfs_dqrele(NULLVP, dq);
    732  1.7.4.2  yamt 	return (0);
    733  1.7.4.2  yamt }
    734  1.7.4.2  yamt #endif
    735  1.7.4.2  yamt 
    736  1.7.4.2  yamt /*
    737  1.7.4.2  yamt  * Q_SYNC - sync quota files to disk.
    738  1.7.4.2  yamt  */
    739  1.7.4.2  yamt int
    740  1.7.4.2  yamt lfs_q1sync(struct mount *mp)
    741  1.7.4.2  yamt {
    742  1.7.4.2  yamt 	struct ulfsmount *ump = VFSTOULFS(mp);
    743  1.7.4.2  yamt 	struct vnode *vp;
    744  1.7.4.2  yamt 	struct vnode_iterator *marker;
    745  1.7.4.2  yamt 	struct dquot *dq;
    746  1.7.4.2  yamt 	int i, error;
    747  1.7.4.2  yamt 
    748  1.7.4.2  yamt 	/*
    749  1.7.4.2  yamt 	 * Check if the mount point has any quotas.
    750  1.7.4.2  yamt 	 * If not, simply return.
    751  1.7.4.2  yamt 	 */
    752  1.7.4.2  yamt 	for (i = 0; i < ULFS_MAXQUOTAS; i++)
    753  1.7.4.2  yamt 		if (ump->um_quotas[i] != NULLVP)
    754  1.7.4.2  yamt 			break;
    755  1.7.4.2  yamt 	if (i == ULFS_MAXQUOTAS)
    756  1.7.4.2  yamt 		return (0);
    757  1.7.4.2  yamt 
    758  1.7.4.2  yamt 	/*
    759  1.7.4.2  yamt 	 * Search vnodes associated with this mount point,
    760  1.7.4.2  yamt 	 * synchronizing any modified dquot structures.
    761  1.7.4.2  yamt 	 */
    762  1.7.4.2  yamt 	vfs_vnode_iterator_init(mp, &marker);
    763  1.7.4.2  yamt 	while (vfs_vnode_iterator_next(marker, &vp)) {
    764  1.7.4.2  yamt 		error = vn_lock(vp, LK_EXCLUSIVE);
    765  1.7.4.2  yamt 		if (error) {
    766  1.7.4.2  yamt 			vrele(vp);
    767  1.7.4.2  yamt 			continue;
    768  1.7.4.2  yamt 		}
    769  1.7.4.2  yamt 		if (VTOI(vp) == NULL || vp->v_type == VNON) {
    770  1.7.4.2  yamt 			vput(vp);
    771  1.7.4.2  yamt 			continue;
    772  1.7.4.2  yamt 		}
    773  1.7.4.2  yamt 		for (i = 0; i < ULFS_MAXQUOTAS; i++) {
    774  1.7.4.2  yamt 			dq = VTOI(vp)->i_dquot[i];
    775  1.7.4.2  yamt 			if (dq == NODQUOT)
    776  1.7.4.2  yamt 				continue;
    777  1.7.4.2  yamt 			mutex_enter(&dq->dq_interlock);
    778  1.7.4.2  yamt 			if (dq->dq_flags & DQ_MOD)
    779  1.7.4.2  yamt 				lfs_dq1sync(vp, dq);
    780  1.7.4.2  yamt 			mutex_exit(&dq->dq_interlock);
    781  1.7.4.2  yamt 		}
    782  1.7.4.2  yamt 		vput(vp);
    783  1.7.4.2  yamt 		mutex_enter(&mntvnode_lock);
    784  1.7.4.2  yamt 	}
    785  1.7.4.2  yamt 	vfs_vnode_iterator_destroy(marker);
    786  1.7.4.2  yamt 	return (0);
    787  1.7.4.2  yamt }
    788  1.7.4.2  yamt 
    789  1.7.4.2  yamt /*
    790  1.7.4.2  yamt  * Obtain a dquot structure for the specified identifier and quota file
    791  1.7.4.2  yamt  * reading the information from the file if necessary.
    792  1.7.4.2  yamt  */
    793  1.7.4.2  yamt int
    794  1.7.4.2  yamt lfs_dq1get(struct vnode *dqvp, u_long id, struct ulfsmount *ump, int type,
    795  1.7.4.2  yamt     struct dquot *dq)
    796  1.7.4.2  yamt {
    797  1.7.4.2  yamt 	struct iovec aiov;
    798  1.7.4.2  yamt 	struct uio auio;
    799  1.7.4.2  yamt 	int error;
    800  1.7.4.2  yamt 
    801  1.7.4.2  yamt 	KASSERT(mutex_owned(&dq->dq_interlock));
    802  1.7.4.2  yamt 	vn_lock(dqvp, LK_EXCLUSIVE | LK_RETRY);
    803  1.7.4.2  yamt 	auio.uio_iov = &aiov;
    804  1.7.4.2  yamt 	auio.uio_iovcnt = 1;
    805  1.7.4.2  yamt 	aiov.iov_base = (void *)&dq->dq_un.dq1_dqb;
    806  1.7.4.2  yamt 	aiov.iov_len = sizeof (struct dqblk);
    807  1.7.4.2  yamt 	auio.uio_resid = sizeof (struct dqblk);
    808  1.7.4.2  yamt 	auio.uio_offset = (off_t)(id * sizeof (struct dqblk));
    809  1.7.4.2  yamt 	auio.uio_rw = UIO_READ;
    810  1.7.4.2  yamt 	UIO_SETUP_SYSSPACE(&auio);
    811  1.7.4.2  yamt 	error = VOP_READ(dqvp, &auio, 0, ump->um_cred[type]);
    812  1.7.4.2  yamt 	if (auio.uio_resid == sizeof(struct dqblk) && error == 0)
    813  1.7.4.2  yamt 		memset((void *)&dq->dq_un.dq1_dqb, 0, sizeof(struct dqblk));
    814  1.7.4.2  yamt 	VOP_UNLOCK(dqvp);
    815  1.7.4.2  yamt 	/*
    816  1.7.4.2  yamt 	 * I/O error in reading quota file, release
    817  1.7.4.2  yamt 	 * quota structure and reflect problem to caller.
    818  1.7.4.2  yamt 	 */
    819  1.7.4.2  yamt 	if (error)
    820  1.7.4.2  yamt 		return (error);
    821  1.7.4.2  yamt 	/*
    822  1.7.4.2  yamt 	 * Check for no limit to enforce.
    823  1.7.4.2  yamt 	 * Initialize time values if necessary.
    824  1.7.4.2  yamt 	 */
    825  1.7.4.2  yamt 	if (dq->dq_isoftlimit == 0 && dq->dq_bsoftlimit == 0 &&
    826  1.7.4.2  yamt 	    dq->dq_ihardlimit == 0 && dq->dq_bhardlimit == 0)
    827  1.7.4.2  yamt 		dq->dq_flags |= DQ_FAKE;
    828  1.7.4.2  yamt 	if (dq->dq_id != 0) {
    829  1.7.4.2  yamt 		if (dq->dq_btime == 0)
    830  1.7.4.2  yamt 			dq->dq_btime = time_second + ump->umq1_btime[type];
    831  1.7.4.2  yamt 		if (dq->dq_itime == 0)
    832  1.7.4.2  yamt 			dq->dq_itime = time_second + ump->umq1_itime[type];
    833  1.7.4.2  yamt 	}
    834  1.7.4.2  yamt 	return (0);
    835  1.7.4.2  yamt }
    836  1.7.4.2  yamt 
    837  1.7.4.2  yamt /*
    838  1.7.4.2  yamt  * Update the disk quota in the quota file.
    839  1.7.4.2  yamt  */
    840  1.7.4.2  yamt int
    841  1.7.4.2  yamt lfs_dq1sync(struct vnode *vp, struct dquot *dq)
    842  1.7.4.2  yamt {
    843  1.7.4.2  yamt 	struct vnode *dqvp;
    844  1.7.4.2  yamt 	struct iovec aiov;
    845  1.7.4.2  yamt 	struct uio auio;
    846  1.7.4.2  yamt 	int error;
    847  1.7.4.2  yamt 
    848  1.7.4.2  yamt 	if (dq == NODQUOT)
    849  1.7.4.2  yamt 		panic("dq1sync: dquot");
    850  1.7.4.2  yamt 	KASSERT(mutex_owned(&dq->dq_interlock));
    851  1.7.4.2  yamt 	if ((dq->dq_flags & DQ_MOD) == 0)
    852  1.7.4.2  yamt 		return (0);
    853  1.7.4.2  yamt 	if ((dqvp = dq->dq_ump->um_quotas[dq->dq_type]) == NULLVP)
    854  1.7.4.2  yamt 		panic("dq1sync: file");
    855  1.7.4.2  yamt 	KASSERT(dqvp != vp);
    856  1.7.4.2  yamt 	vn_lock(dqvp, LK_EXCLUSIVE | LK_RETRY);
    857  1.7.4.2  yamt 	auio.uio_iov = &aiov;
    858  1.7.4.2  yamt 	auio.uio_iovcnt = 1;
    859  1.7.4.2  yamt 	aiov.iov_base = (void *)&dq->dq_un.dq1_dqb;
    860  1.7.4.2  yamt 	aiov.iov_len = sizeof (struct dqblk);
    861  1.7.4.2  yamt 	auio.uio_resid = sizeof (struct dqblk);
    862  1.7.4.2  yamt 	auio.uio_offset = (off_t)(dq->dq_id * sizeof (struct dqblk));
    863  1.7.4.2  yamt 	auio.uio_rw = UIO_WRITE;
    864  1.7.4.2  yamt 	UIO_SETUP_SYSSPACE(&auio);
    865  1.7.4.2  yamt 	error = VOP_WRITE(dqvp, &auio, 0, dq->dq_ump->um_cred[dq->dq_type]);
    866  1.7.4.2  yamt 	if (auio.uio_resid && error == 0)
    867  1.7.4.2  yamt 		error = EIO;
    868  1.7.4.2  yamt 	dq->dq_flags &= ~DQ_MOD;
    869  1.7.4.2  yamt 	VOP_UNLOCK(dqvp);
    870  1.7.4.2  yamt 	return (error);
    871  1.7.4.2  yamt }
    872