Remove the thread argument from all mount->vfs_* function vectors,
[dragonfly.git] / sys / vfs / gnu / ext2fs / ext2_quota.c
CommitLineData
1f1db49f
MD
1/*
2 * Copyright (c) 1982, 1986, 1990, 1993, 1995
3 * The Regents of the University of California. All rights reserved.
4 *
5 * This code is derived from software contributed to Berkeley by
6 * Robert Elz at The University of Melbourne.
7 *
8 * Redistribution and use in source and binary forms, with or without
9 * modification, are permitted provided that the following conditions
10 * are met:
11 * 1. Redistributions of source code must retain the above copyright
12 * notice, this list of conditions and the following disclaimer.
13 * 2. Redistributions in binary form must reproduce the above copyright
14 * notice, this list of conditions and the following disclaimer in the
15 * documentation and/or other materials provided with the distribution.
16 * 3. All advertising materials mentioning features or use of this software
17 * must display the following acknowledgement:
18 * This product includes software developed by the University of
19 * California, Berkeley and its contributors.
20 * 4. Neither the name of the University nor the names of its contributors
21 * may be used to endorse or promote products derived from this software
22 * without specific prior written permission.
23 *
24 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
25 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
26 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
27 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
28 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
29 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
30 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
31 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
32 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
33 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
34 * SUCH DAMAGE.
35 *
36 * @(#)ufs_quota.c 8.5 (Berkeley) 5/20/95
37 * $FreeBSD: src/sys/ufs/ufs/ufs_quota.c,v 1.27.2.3 2002/01/15 10:33:32 phk Exp $
acde96db 38 * $DragonFly: src/sys/vfs/gnu/ext2fs/ext2_quota.c,v 1.5 2006/05/06 18:48:52 dillon Exp $
1f1db49f
MD
39 */
40
41#include <sys/param.h>
42#include <sys/systm.h>
43#include <sys/kernel.h>
44#include <sys/malloc.h>
45#include <sys/fcntl.h>
46#include <sys/proc.h>
47#include <sys/nlookup.h>
48#include <sys/vnode.h>
49#include <sys/mount.h>
50#include <vm/vm_zone.h>
51
52#include "quota.h"
53#include "dinode.h"
54#include "inode.h"
55#include "ext2mount.h"
56#include "ext2_extern.h"
57
58static MALLOC_DEFINE(M_EXT2DQUOT, "EXT2 quota", "EXT2 quota entries");
59
60/*
61 * Quota name to error message mapping.
62 */
63static char *quotatypes[] = INITQFNAMES;
64
65static int ext2_chkdqchg (struct inode *, long, struct ucred *, int);
66static int ext2_chkiqchg (struct inode *, long, struct ucred *, int);
67static int ext2_dqget (struct vnode *,
68 u_long, struct ext2mount *, int, struct ext2_dquot **);
69static int ext2_dqsync (struct vnode *, struct ext2_dquot *);
70static void ext2_dqflush (struct vnode *);
71
72#ifdef DIAGNOSTIC
73static void ext2_dqref (struct ext2_dquot *);
74static void ext2_chkdquot (struct inode *);
75#endif
76
77/*
78 * Set up the quotas for an inode.
79 *
80 * This routine completely defines the semantics of quotas.
81 * If other criterion want to be used to establish quotas, the
82 * MAXQUOTAS value in quotas.h should be increased, and the
83 * additional dquots set up here.
84 */
85int
86ext2_getinoquota(struct inode *ip)
87{
88 struct ext2mount *ump;
89 struct vnode *vp = ITOV(ip);
90 int error;
91
92 ump = VFSTOEXT2(vp->v_mount);
93 /*
94 * Set up the user quota based on file uid.
95 * EINVAL means that quotas are not enabled.
96 */
97 if (ip->i_dquot[USRQUOTA] == NODQUOT &&
98 (error = ext2_dqget(vp, ip->i_uid, ump, USRQUOTA, &ip->i_dquot[USRQUOTA])) &&
99 error != EINVAL)
100 return (error);
101 /*
102 * Set up the group quota based on file gid.
103 * EINVAL means that quotas are not enabled.
104 */
105 if (ip->i_dquot[GRPQUOTA] == NODQUOT &&
106 (error = ext2_dqget(vp, ip->i_gid, ump, GRPQUOTA, &ip->i_dquot[GRPQUOTA])) &&
107 error != EINVAL)
108 return (error);
109 return (0);
110}
111
112/*
113 * Update disk usage, and take corrective action.
114 */
115int
116ext2_chkdq(struct inode *ip, long change, struct ucred *cred, int flags)
117{
118 struct ext2_dquot *dq;
119 int i;
120 int ncurblocks, error;
121
122#ifdef DIAGNOSTIC
123 if ((flags & CHOWN) == 0)
124 ext2_chkdquot(ip);
125#endif
126 if (change == 0)
127 return (0);
128 if (change < 0) {
129 for (i = 0; i < MAXQUOTAS; i++) {
130 if ((dq = ip->i_dquot[i]) == NODQUOT)
131 continue;
132 while (dq->dq_flags & DQ_LOCK) {
133 dq->dq_flags |= DQ_WANT;
134 (void) tsleep((caddr_t)dq, 0, "chkdq1", 0);
135 }
136 ncurblocks = dq->dq_curblocks + change;
137 if (ncurblocks >= 0)
138 dq->dq_curblocks = ncurblocks;
139 else
140 dq->dq_curblocks = 0;
141 dq->dq_flags &= ~DQ_BLKS;
142 dq->dq_flags |= DQ_MOD;
143 }
144 return (0);
145 }
146 if ((flags & FORCE) == 0 && cred->cr_uid != 0) {
147 for (i = 0; i < MAXQUOTAS; i++) {
148 if ((dq = ip->i_dquot[i]) == NODQUOT)
149 continue;
150 error = ext2_chkdqchg(ip, change, cred, i);
151 if (error)
152 return (error);
153 }
154 }
155 for (i = 0; i < MAXQUOTAS; i++) {
156 if ((dq = ip->i_dquot[i]) == NODQUOT)
157 continue;
158 while (dq->dq_flags & DQ_LOCK) {
159 dq->dq_flags |= DQ_WANT;
160 (void) tsleep((caddr_t)dq, 0, "chkdq2", 0);
161 }
162 /* Reset timer when crossing soft limit */
163 if (dq->dq_curblocks + change >= dq->dq_bsoftlimit &&
164 dq->dq_curblocks < dq->dq_bsoftlimit)
165 dq->dq_btime = time_second +
166 VFSTOEXT2(ITOV(ip)->v_mount)->um_btime[i];
167 dq->dq_curblocks += change;
168 dq->dq_flags |= DQ_MOD;
169 }
170 return (0);
171}
172
173/*
174 * Check for a valid change to a users allocation.
175 * Issue an error message if appropriate.
176 */
177static int
178ext2_chkdqchg(struct inode *ip, long change, struct ucred *cred, int type)
179{
180 struct ext2_dquot *dq = ip->i_dquot[type];
181 long ncurblocks = dq->dq_curblocks + change;
182
183 /*
184 * If user would exceed their hard limit, disallow space allocation.
185 */
186 if (ncurblocks >= dq->dq_bhardlimit && dq->dq_bhardlimit) {
187 if ((dq->dq_flags & DQ_BLKS) == 0 &&
188 ip->i_uid == cred->cr_uid) {
189 uprintf("\n%s: write failed, %s disk limit reached\n",
190 ITOV(ip)->v_mount->mnt_stat.f_mntfromname,
191 quotatypes[type]);
192 dq->dq_flags |= DQ_BLKS;
193 }
194 return (EDQUOT);
195 }
196 /*
197 * If user is over their soft limit for too long, disallow space
198 * allocation. Reset time limit as they cross their soft limit.
199 */
200 if (ncurblocks >= dq->dq_bsoftlimit && dq->dq_bsoftlimit) {
201 if (dq->dq_curblocks < dq->dq_bsoftlimit) {
202 dq->dq_btime = time_second +
203 VFSTOEXT2(ITOV(ip)->v_mount)->um_btime[type];
204 if (ip->i_uid == cred->cr_uid)
205 uprintf("\n%s: warning, %s %s\n",
206 ITOV(ip)->v_mount->mnt_stat.f_mntfromname,
207 quotatypes[type], "disk quota exceeded");
208 return (0);
209 }
210 if (time_second > dq->dq_btime) {
211 if ((dq->dq_flags & DQ_BLKS) == 0 &&
212 ip->i_uid == cred->cr_uid) {
213 uprintf("\n%s: write failed, %s %s\n",
214 ITOV(ip)->v_mount->mnt_stat.f_mntfromname,
215 quotatypes[type],
216 "disk quota exceeded for too long");
217 dq->dq_flags |= DQ_BLKS;
218 }
219 return (EDQUOT);
220 }
221 }
222 return (0);
223}
224
225/*
226 * Check the inode limit, applying corrective action.
227 */
228int
229ext2_chkiq(struct inode *ip, long change, struct ucred *cred, int flags)
230{
231 struct ext2_dquot *dq;
232 int i;
233 int ncurinodes, error;
234
235#ifdef DIAGNOSTIC
236 if ((flags & CHOWN) == 0)
237 ext2_chkdquot(ip);
238#endif
239 if (change == 0)
240 return (0);
241 if (change < 0) {
242 for (i = 0; i < MAXQUOTAS; i++) {
243 if ((dq = ip->i_dquot[i]) == NODQUOT)
244 continue;
245 while (dq->dq_flags & DQ_LOCK) {
246 dq->dq_flags |= DQ_WANT;
247 (void) tsleep((caddr_t)dq, 0, "chkiq1", 0);
248 }
249 ncurinodes = dq->dq_curinodes + change;
250 if (ncurinodes >= 0)
251 dq->dq_curinodes = ncurinodes;
252 else
253 dq->dq_curinodes = 0;
254 dq->dq_flags &= ~DQ_INODS;
255 dq->dq_flags |= DQ_MOD;
256 }
257 return (0);
258 }
259 if ((flags & FORCE) == 0 && cred->cr_uid != 0) {
260 for (i = 0; i < MAXQUOTAS; i++) {
261 if ((dq = ip->i_dquot[i]) == NODQUOT)
262 continue;
263 error = ext2_chkiqchg(ip, change, cred, i);
264 if (error)
265 return (error);
266 }
267 }
268 for (i = 0; i < MAXQUOTAS; i++) {
269 if ((dq = ip->i_dquot[i]) == NODQUOT)
270 continue;
271 while (dq->dq_flags & DQ_LOCK) {
272 dq->dq_flags |= DQ_WANT;
273 (void) tsleep((caddr_t)dq, 0, "chkiq2", 0);
274 }
275 /* Reset timer when crossing soft limit */
276 if (dq->dq_curinodes + change >= dq->dq_isoftlimit &&
277 dq->dq_curinodes < dq->dq_isoftlimit)
278 dq->dq_itime = time_second +
279 VFSTOEXT2(ITOV(ip)->v_mount)->um_itime[i];
280 dq->dq_curinodes += change;
281 dq->dq_flags |= DQ_MOD;
282 }
283 return (0);
284}
285
286/*
287 * Check for a valid change to a users allocation.
288 * Issue an error message if appropriate.
289 */
290static int
291ext2_chkiqchg(struct inode *ip, long change, struct ucred *cred, int type)
292{
293 struct ext2_dquot *dq = ip->i_dquot[type];
294 long ncurinodes = dq->dq_curinodes + change;
295
296 /*
297 * If user would exceed their hard limit, disallow inode allocation.
298 */
299 if (ncurinodes >= dq->dq_ihardlimit && dq->dq_ihardlimit) {
300 if ((dq->dq_flags & DQ_INODS) == 0 &&
301 ip->i_uid == cred->cr_uid) {
302 uprintf("\n%s: write failed, %s inode limit reached\n",
303 ITOV(ip)->v_mount->mnt_stat.f_mntfromname,
304 quotatypes[type]);
305 dq->dq_flags |= DQ_INODS;
306 }
307 return (EDQUOT);
308 }
309 /*
310 * If user is over their soft limit for too long, disallow inode
311 * allocation. Reset time limit as they cross their soft limit.
312 */
313 if (ncurinodes >= dq->dq_isoftlimit && dq->dq_isoftlimit) {
314 if (dq->dq_curinodes < dq->dq_isoftlimit) {
315 dq->dq_itime = time_second +
316 VFSTOEXT2(ITOV(ip)->v_mount)->um_itime[type];
317 if (ip->i_uid == cred->cr_uid)
318 uprintf("\n%s: warning, %s %s\n",
319 ITOV(ip)->v_mount->mnt_stat.f_mntfromname,
320 quotatypes[type], "inode quota exceeded");
321 return (0);
322 }
323 if (time_second > dq->dq_itime) {
324 if ((dq->dq_flags & DQ_INODS) == 0 &&
325 ip->i_uid == cred->cr_uid) {
326 uprintf("\n%s: write failed, %s %s\n",
327 ITOV(ip)->v_mount->mnt_stat.f_mntfromname,
328 quotatypes[type],
329 "inode quota exceeded for too long");
330 dq->dq_flags |= DQ_INODS;
331 }
332 return (EDQUOT);
333 }
334 }
335 return (0);
336}
337
338#ifdef DIAGNOSTIC
339/*
340 * On filesystems with quotas enabled, it is an error for a file to change
341 * size and not to have a dquot structure associated with it.
342 */
343static void
344ext2_chkdquot(struct inode *ip)
345{
346 struct ext2mount *ump = VFSTOEXT2(ITOV(ip)->v_mount);
347 int i;
348
349 for (i = 0; i < MAXQUOTAS; i++) {
350 if (ump->um_quotas[i] == NULLVP ||
351 (ump->um_qflags[i] & (QTF_OPENING|QTF_CLOSING)))
352 continue;
353 if (ip->i_dquot[i] == NODQUOT) {
354 vprint("chkdquot: missing dquot", ITOV(ip));
355 panic("chkdquot: missing dquot");
356 }
357 }
358}
359#endif
360
361/*
362 * Code to process quotactl commands.
363 */
364
365struct scaninfo {
1f1db49f
MD
366 int rescan;
367 int type;
368};
369
370/*
371 * Q_QUOTAON - set up a quota file for a particular filesystem.
372 */
373static int ext2_quotaon_scan(struct mount *mp, struct vnode *vp, void *data);
374
375int
acde96db 376ext2_quotaon(struct ucred *cred, struct mount *mp, int type, caddr_t fname)
1f1db49f
MD
377{
378 struct ext2mount *ump = VFSTOEXT2(mp);
379 struct vnode *vp, **vpp;
380 struct ext2_dquot *dq;
381 int error;
382 struct nlookupdata nd;
1f1db49f
MD
383 struct scaninfo scaninfo;
384
1f1db49f
MD
385 vpp = &ump->um_quotas[type];
386 error = nlookup_init(&nd, fname, UIO_USERSPACE, NLC_FOLLOW|NLC_LOCKVP);
387 if (error == 0)
388 error = vn_open(&nd, NULL, FREAD|FWRITE, 0);
389 if (error == 0 && nd.nl_open_vp->v_type != VREG)
390 error = EACCES;
391 if (error) {
392 nlookup_done(&nd);
393 return (error);
394 }
395 vp = nd.nl_open_vp;
396 nd.nl_open_vp = NULL;
397 nlookup_done(&nd);
398
ca466bae 399 VOP_UNLOCK(vp, 0);
1f1db49f 400 if (*vpp != vp)
f94bf21c 401 ext2_quotaoff(mp, type);
1f1db49f
MD
402 ump->um_qflags[type] |= QTF_OPENING;
403 mp->mnt_flag |= MNT_QUOTA;
404 vp->v_flag |= VSYSTEM;
405 *vpp = vp;
406 /* XXX release duplicate vp if *vpp == vp? */
407 /*
408 * Save the credential of the process that turned on quotas.
409 * Set up the time limits for this quota.
410 */
411 ump->um_cred[type] = crhold(cred);
412 ump->um_btime[type] = MAX_DQ_TIME;
413 ump->um_itime[type] = MAX_IQ_TIME;
414 if (ext2_dqget(NULLVP, 0, ump, type, &dq) == 0) {
415 if (dq->dq_btime > 0)
416 ump->um_btime[type] = dq->dq_btime;
417 if (dq->dq_itime > 0)
418 ump->um_itime[type] = dq->dq_itime;
419 ext2_dqrele(NULLVP, dq);
420 }
421 /*
422 * Search vnodes associated with this mount point,
423 * adding references to quota file being opened.
424 * NB: only need to add dquot's for inodes being modified.
425 */
426 scaninfo.rescan = 1;
1f1db49f
MD
427 while (scaninfo.rescan) {
428 scaninfo.rescan = 0;
429 error = vmntvnodescan(mp, VMSC_GETVP,
430 NULL, ext2_quotaon_scan, &scaninfo);
431 if (error)
432 break;
433 }
434 ump->um_qflags[type] &= ~QTF_OPENING;
435 if (error)
f94bf21c 436 ext2_quotaoff(mp, type);
1f1db49f
MD
437 return (error);
438}
439
440static int
441ext2_quotaon_scan(struct mount *mp, struct vnode *vp, void *data)
442{
443 int error;
444 /*struct scaninfo *info = data;*/
445
446 if (vp->v_writecount == 0)
447 return(0);
448 error = ext2_getinoquota(VTOI(vp));
449 return(error);
450}
451
452/*
453 * Q_QUOTAOFF - turn off disk quotas for a filesystem.
454 */
455
456static int ext2_quotaoff_scan(struct mount *mp, struct vnode *vp, void *data);
457
458int
f94bf21c 459ext2_quotaoff(struct mount *mp, int type)
1f1db49f
MD
460{
461 struct vnode *qvp;
462 struct ext2mount *ump = VFSTOEXT2(mp);
1f1db49f
MD
463 int error;
464 struct scaninfo scaninfo;
465
1f1db49f
MD
466 if ((qvp = ump->um_quotas[type]) == NULLVP)
467 return (0);
468 ump->um_qflags[type] |= QTF_CLOSING;
469
470 /*
471 * Search vnodes associated with this mount point,
472 * deleting any references to quota file being closed.
473 */
474 scaninfo.rescan = 1;
1f1db49f
MD
475 scaninfo.type = type;
476 while (scaninfo.rescan) {
477 scaninfo.rescan = 0;
478 vmntvnodescan(mp, VMSC_GETVP, NULL, ext2_quotaoff_scan, &scaninfo);
479 }
480 ext2_dqflush(qvp);
481 qvp->v_flag &= ~VSYSTEM;
87de5057 482 error = vn_close(qvp, FREAD|FWRITE);
1f1db49f
MD
483 ump->um_quotas[type] = NULLVP;
484 crfree(ump->um_cred[type]);
485 ump->um_cred[type] = NOCRED;
486 ump->um_qflags[type] &= ~QTF_CLOSING;
487 for (type = 0; type < MAXQUOTAS; type++) {
488 if (ump->um_quotas[type] != NULLVP)
489 break;
490 }
491 if (type == MAXQUOTAS)
492 mp->mnt_flag &= ~MNT_QUOTA;
493 return (error);
494}
495
496static int
497ext2_quotaoff_scan(struct mount *mp, struct vnode *vp, void *data)
498{
499 struct scaninfo *info = data;
500 struct ext2_dquot *dq;
501 struct inode *ip;
502
503 if (vp->v_type == VNON) {
504 return(0);
505 }
506 ip = VTOI(vp);
507 dq = ip->i_dquot[info->type];
508 ip->i_dquot[info->type] = NODQUOT;
509 ext2_dqrele(vp, dq);
510 return(0);
511}
512
513/*
514 * Q_GETQUOTA - return current values in a dqblk structure.
515 */
516int
517ext2_getquota(struct mount *mp, u_long id, int type, caddr_t addr)
518{
519 struct ext2_dquot *dq;
520 int error;
521
522 error = ext2_dqget(NULLVP, id, VFSTOEXT2(mp), type, &dq);
523 if (error)
524 return (error);
525 error = copyout((caddr_t)&dq->dq_dqb, addr, sizeof (struct ext2_dqblk));
526 ext2_dqrele(NULLVP, dq);
527 return (error);
528}
529
530/*
531 * Q_SETQUOTA - assign an entire dqblk structure.
532 */
533int
534ext2_setquota(struct mount *mp, u_long id, int type, caddr_t addr)
535{
536 struct ext2_dquot *dq;
537 struct ext2_dquot *ndq;
538 struct ext2mount *ump = VFSTOEXT2(mp);
539 struct ext2_dqblk newlim;
540 int error;
541
542 error = copyin(addr, (caddr_t)&newlim, sizeof (struct ext2_dqblk));
543 if (error)
544 return (error);
545 error = ext2_dqget(NULLVP, id, ump, type, &ndq);
546 if (error)
547 return (error);
548 dq = ndq;
549 while (dq->dq_flags & DQ_LOCK) {
550 dq->dq_flags |= DQ_WANT;
551 (void) tsleep((caddr_t)dq, 0, "setqta", 0);
552 }
553 /*
554 * Copy all but the current values.
555 * Reset time limit if previously had no soft limit or were
556 * under it, but now have a soft limit and are over it.
557 */
558 newlim.dqb_curblocks = dq->dq_curblocks;
559 newlim.dqb_curinodes = dq->dq_curinodes;
560 if (dq->dq_id != 0) {
561 newlim.dqb_btime = dq->dq_btime;
562 newlim.dqb_itime = dq->dq_itime;
563 }
564 if (newlim.dqb_bsoftlimit &&
565 dq->dq_curblocks >= newlim.dqb_bsoftlimit &&
566 (dq->dq_bsoftlimit == 0 || dq->dq_curblocks < dq->dq_bsoftlimit))
567 newlim.dqb_btime = time_second + ump->um_btime[type];
568 if (newlim.dqb_isoftlimit &&
569 dq->dq_curinodes >= newlim.dqb_isoftlimit &&
570 (dq->dq_isoftlimit == 0 || dq->dq_curinodes < dq->dq_isoftlimit))
571 newlim.dqb_itime = time_second + ump->um_itime[type];
572 dq->dq_dqb = newlim;
573 if (dq->dq_curblocks < dq->dq_bsoftlimit)
574 dq->dq_flags &= ~DQ_BLKS;
575 if (dq->dq_curinodes < dq->dq_isoftlimit)
576 dq->dq_flags &= ~DQ_INODS;
577 if (dq->dq_isoftlimit == 0 && dq->dq_bsoftlimit == 0 &&
578 dq->dq_ihardlimit == 0 && dq->dq_bhardlimit == 0)
579 dq->dq_flags |= DQ_FAKE;
580 else
581 dq->dq_flags &= ~DQ_FAKE;
582 dq->dq_flags |= DQ_MOD;
583 ext2_dqrele(NULLVP, dq);
584 return (0);
585}
586
587/*
588 * Q_SETUSE - set current inode and block usage.
589 */
590int
591ext2_setuse(struct mount *mp, u_long id, int type, caddr_t addr)
592{
593 struct ext2_dquot *dq;
594 struct ext2mount *ump = VFSTOEXT2(mp);
595 struct ext2_dquot *ndq;
596 struct ext2_dqblk usage;
597 int error;
598
599 error = copyin(addr, (caddr_t)&usage, sizeof (struct ext2_dqblk));
600 if (error)
601 return (error);
602 error = ext2_dqget(NULLVP, id, ump, type, &ndq);
603 if (error)
604 return (error);
605 dq = ndq;
606 while (dq->dq_flags & DQ_LOCK) {
607 dq->dq_flags |= DQ_WANT;
608 (void) tsleep((caddr_t)dq, 0, "setuse", 0);
609 }
610 /*
611 * Reset time limit if have a soft limit and were
612 * previously under it, but are now over it.
613 */
614 if (dq->dq_bsoftlimit && dq->dq_curblocks < dq->dq_bsoftlimit &&
615 usage.dqb_curblocks >= dq->dq_bsoftlimit)
616 dq->dq_btime = time_second + ump->um_btime[type];
617 if (dq->dq_isoftlimit && dq->dq_curinodes < dq->dq_isoftlimit &&
618 usage.dqb_curinodes >= dq->dq_isoftlimit)
619 dq->dq_itime = time_second + ump->um_itime[type];
620 dq->dq_curblocks = usage.dqb_curblocks;
621 dq->dq_curinodes = usage.dqb_curinodes;
622 if (dq->dq_curblocks < dq->dq_bsoftlimit)
623 dq->dq_flags &= ~DQ_BLKS;
624 if (dq->dq_curinodes < dq->dq_isoftlimit)
625 dq->dq_flags &= ~DQ_INODS;
626 dq->dq_flags |= DQ_MOD;
627 ext2_dqrele(NULLVP, dq);
628 return (0);
629}
630
631/*
632 * Q_SYNC - sync quota files to disk.
633 */
634
635static int ext2_qsync_scan(struct mount *mp, struct vnode *vp, void *data);
636
637int
638ext2_qsync(struct mount *mp)
639{
640 struct ext2mount *ump = VFSTOEXT2(mp);
1f1db49f
MD
641 struct scaninfo scaninfo;
642 int i;
643
644 /*
645 * Check if the mount point has any quotas.
646 * If not, simply return.
647 */
648 for (i = 0; i < MAXQUOTAS; i++)
649 if (ump->um_quotas[i] != NULLVP)
650 break;
651 if (i == MAXQUOTAS)
652 return (0);
653 /*
654 * Search vnodes associated with this mount point,
655 * synchronizing any modified ext2_dquot structures.
656 */
657 scaninfo.rescan = 1;
1f1db49f
MD
658 while (scaninfo.rescan) {
659 scaninfo.rescan = 0;
660 vmntvnodescan(mp, VMSC_GETVP|VMSC_NOWAIT,
661 NULL, ext2_qsync_scan, &scaninfo);
662 }
663 return (0);
664}
665
666static int
667ext2_qsync_scan(struct mount *mp, struct vnode *vp, void *data)
668{
669 /*struct scaninfo *info = data;*/
670 struct ext2_dquot *dq;
671 /* int error;*/
672 int i;
673
674 for (i = 0; i < MAXQUOTAS; i++) {
675 dq = VTOI(vp)->i_dquot[i];
676 if (dq != NODQUOT && (dq->dq_flags & DQ_MOD))
677 ext2_dqsync(vp, dq);
678 }
679 return(0);
680}
681
682/*
683 * Code pertaining to management of the in-core dquot data structures.
684 */
685#define DQHASH(dqvp, id) \
686 (&ext2_dqhashtbl[((((intptr_t)(dqvp)) >> 8) + id) & ext2_dqhash])
687static LIST_HEAD(ext2_dqhash, ext2_dquot) *ext2_dqhashtbl;
688static u_long ext2_dqhash;
689
690/*
691 * Dquot free list.
692 */
693#define DQUOTINC 5 /* minimum free dquots desired */
694static TAILQ_HEAD(ext2_dqfreelist, ext2_dquot) ext2_dqfreelist;
695static long ext2_numdquot, ext2_desireddquot = DQUOTINC;
696
697/*
698 * Initialize the quota system.
699 */
700void
701ext2_dqinit(void)
702{
703 ext2_dqhashtbl = hashinit(desiredvnodes, M_EXT2DQUOT, &ext2_dqhash);
704 TAILQ_INIT(&ext2_dqfreelist);
705}
706
707/*
708 * Obtain a dquot structure for the specified identifier and quota file
709 * reading the information from the file if necessary.
710 */
711static int
712ext2_dqget(struct vnode *vp, u_long id, struct ext2mount *ump, int type,
713 struct ext2_dquot **dqp)
714{
1f1db49f
MD
715 struct ext2_dquot *dq;
716 struct ext2_dqhash *dqh;
717 struct vnode *dqvp;
718 struct iovec aiov;
719 struct uio auio;
720 int error;
721
722 dqvp = ump->um_quotas[type];
723 if (dqvp == NULLVP || (ump->um_qflags[type] & QTF_CLOSING)) {
724 *dqp = NODQUOT;
725 return (EINVAL);
726 }
727 /*
728 * Check the cache first.
729 */
730 dqh = DQHASH(dqvp, id);
731 for (dq = dqh->lh_first; dq; dq = dq->dq_hash.le_next) {
732 if (dq->dq_id != id ||
733 dq->dq_ump->um_quotas[dq->dq_type] != dqvp)
734 continue;
735 /*
736 * Cache hit with no references. Take
737 * the structure off the free list.
738 */
739 if (dq->dq_cnt == 0)
740 TAILQ_REMOVE(&ext2_dqfreelist, dq, dq_freelist);
741 DQREF(dq);
742 *dqp = dq;
743 return (0);
744 }
745 /*
746 * Not in cache, allocate a new one.
747 */
748 if (TAILQ_EMPTY(&ext2_dqfreelist) && ext2_numdquot < MAXQUOTAS * desiredvnodes)
749 ext2_desireddquot += DQUOTINC;
750 if (ext2_numdquot < ext2_desireddquot) {
751 dq = (struct ext2_dquot *)malloc(sizeof *dq, M_EXT2DQUOT, M_WAITOK);
752 bzero((char *)dq, sizeof *dq);
753 ext2_numdquot++;
754 } else {
755 if ((dq = TAILQ_FIRST(&ext2_dqfreelist)) == NULL) {
756 tablefull("dquot");
757 *dqp = NODQUOT;
758 return (EUSERS);
759 }
760 if (dq->dq_cnt || (dq->dq_flags & DQ_MOD))
761 panic("dqget: free dquot isn't");
762 TAILQ_REMOVE(&ext2_dqfreelist, dq, dq_freelist);
763 if (dq->dq_ump != NULL)
764 LIST_REMOVE(dq, dq_hash);
765 }
766 /*
767 * Initialize the contents of the dquot structure.
768 */
769 if (vp != dqvp)
ca466bae 770 vn_lock(dqvp, LK_EXCLUSIVE | LK_RETRY);
1f1db49f
MD
771 LIST_INSERT_HEAD(dqh, dq, dq_hash);
772 DQREF(dq);
773 dq->dq_flags = DQ_LOCK;
774 dq->dq_id = id;
775 dq->dq_ump = ump;
776 dq->dq_type = type;
777 auio.uio_iov = &aiov;
778 auio.uio_iovcnt = 1;
779 aiov.iov_base = (caddr_t)&dq->dq_dqb;
780 aiov.iov_len = sizeof (struct ext2_dqblk);
781 auio.uio_resid = sizeof (struct ext2_dqblk);
782 auio.uio_offset = (off_t)(id * sizeof (struct ext2_dqblk));
783 auio.uio_segflg = UIO_SYSSPACE;
784 auio.uio_rw = UIO_READ;
785 auio.uio_td = NULL;
786 error = VOP_READ(dqvp, &auio, 0, ump->um_cred[type]);
787 if (auio.uio_resid == sizeof(struct ext2_dqblk) && error == 0)
788 bzero((caddr_t)&dq->dq_dqb, sizeof(struct ext2_dqblk));
789 if (vp != dqvp)
ca466bae 790 VOP_UNLOCK(dqvp, 0);
1f1db49f
MD
791 if (dq->dq_flags & DQ_WANT)
792 wakeup((caddr_t)dq);
793 dq->dq_flags = 0;
794 /*
795 * I/O error in reading quota file, release
796 * quota structure and reflect problem to caller.
797 */
798 if (error) {
799 LIST_REMOVE(dq, dq_hash);
800 ext2_dqrele(vp, dq);
801 *dqp = NODQUOT;
802 return (error);
803 }
804 /*
805 * Check for no limit to enforce.
806 * Initialize time values if necessary.
807 */
808 if (dq->dq_isoftlimit == 0 && dq->dq_bsoftlimit == 0 &&
809 dq->dq_ihardlimit == 0 && dq->dq_bhardlimit == 0)
810 dq->dq_flags |= DQ_FAKE;
811 if (dq->dq_id != 0) {
812 if (dq->dq_btime == 0)
813 dq->dq_btime = time_second + ump->um_btime[type];
814 if (dq->dq_itime == 0)
815 dq->dq_itime = time_second + ump->um_itime[type];
816 }
817 *dqp = dq;
818 return (0);
819}
820
821#ifdef DIAGNOSTIC
822/*
823 * Obtain a reference to a dquot.
824 */
825static void
826ext2_dqref(struct ext2_dquot *dq)
827{
828 dq->dq_cnt++;
829}
830#endif
831
832/*
833 * Release a reference to a dquot.
834 */
835void
836ext2_dqrele(struct vnode *vp, struct ext2_dquot *dq)
837{
838 if (dq == NODQUOT)
839 return;
840 if (dq->dq_cnt > 1) {
841 dq->dq_cnt--;
842 return;
843 }
844 if (dq->dq_flags & DQ_MOD)
845 (void)ext2_dqsync(vp, dq);
846 if (--dq->dq_cnt > 0)
847 return;
848 TAILQ_INSERT_TAIL(&ext2_dqfreelist, dq, dq_freelist);
849}
850
851/*
852 * Update the disk quota in the quota file.
853 */
854static int
855ext2_dqsync(struct vnode *vp, struct ext2_dquot *dq)
856{
1f1db49f
MD
857 struct vnode *dqvp;
858 struct iovec aiov;
859 struct uio auio;
860 int error;
861
862 if (dq == NODQUOT)
863 panic("dqsync: dquot");
864 if ((dq->dq_flags & DQ_MOD) == 0)
865 return (0);
866 if ((dqvp = dq->dq_ump->um_quotas[dq->dq_type]) == NULLVP)
867 panic("dqsync: file");
868 if (vp != dqvp)
ca466bae 869 vn_lock(dqvp, LK_EXCLUSIVE | LK_RETRY);
1f1db49f
MD
870 while (dq->dq_flags & DQ_LOCK) {
871 dq->dq_flags |= DQ_WANT;
872 (void) tsleep((caddr_t)dq, 0, "dqsync", 0);
873 if ((dq->dq_flags & DQ_MOD) == 0) {
874 if (vp != dqvp)
ca466bae 875 VOP_UNLOCK(dqvp, 0);
1f1db49f
MD
876 return (0);
877 }
878 }
879 dq->dq_flags |= DQ_LOCK;
880 auio.uio_iov = &aiov;
881 auio.uio_iovcnt = 1;
882 aiov.iov_base = (caddr_t)&dq->dq_dqb;
883 aiov.iov_len = sizeof (struct ext2_dqblk);
884 auio.uio_resid = sizeof (struct ext2_dqblk);
885 auio.uio_offset = (off_t)(dq->dq_id * sizeof (struct ext2_dqblk));
886 auio.uio_segflg = UIO_SYSSPACE;
887 auio.uio_rw = UIO_WRITE;
888 auio.uio_td = NULL;
889 error = VOP_WRITE(dqvp, &auio, 0, dq->dq_ump->um_cred[dq->dq_type]);
890 if (auio.uio_resid && error == 0)
891 error = EIO;
892 if (dq->dq_flags & DQ_WANT)
893 wakeup((caddr_t)dq);
894 dq->dq_flags &= ~(DQ_MOD|DQ_LOCK|DQ_WANT);
895 if (vp != dqvp)
ca466bae 896 VOP_UNLOCK(dqvp, 0);
1f1db49f
MD
897 return (error);
898}
899
900/*
901 * Flush all entries from the cache for a particular vnode.
902 */
903static void
904ext2_dqflush(struct vnode *vp)
905{
906 struct ext2_dquot *dq, *nextdq;
907 struct ext2_dqhash *dqh;
908
909 /*
910 * Move all dquot's that used to refer to this quota
911 * file off their hash chains (they will eventually
912 * fall off the head of the free list and be re-used).
913 */
914 for (dqh = &ext2_dqhashtbl[ext2_dqhash]; dqh >= ext2_dqhashtbl; dqh--) {
915 for (dq = dqh->lh_first; dq; dq = nextdq) {
916 nextdq = dq->dq_hash.le_next;
917 if (dq->dq_ump->um_quotas[dq->dq_type] != vp)
918 continue;
919 if (dq->dq_cnt)
920 panic("dqflush: stray dquot");
921 LIST_REMOVE(dq, dq_hash);
922 dq->dq_ump = (struct ext2mount *)0;
923 }
924 }
925}