proc->thread stage 3.5: Add an IO_CORE flag so coda doesn't have to dig
[dragonfly.git] / sys / vfs / ufs / ufs_quota.c
CommitLineData
984263bc
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 $
1de703da 38 * $DragonFly: src/sys/vfs/ufs/ufs_quota.c,v 1.2 2003/06/17 04:29:00 dillon Exp $
984263bc
MD
39 */
40
41#include <sys/param.h>
42#include <sys/systm.h>
43#include <sys/kernel.h>
44#include <sys/namei.h>
45#include <sys/malloc.h>
46#include <sys/fcntl.h>
47#include <sys/proc.h>
48#include <sys/vnode.h>
49#include <sys/mount.h>
50#include <vm/vm_zone.h>
51
52#include <ufs/ufs/quota.h>
53#include <ufs/ufs/inode.h>
54#include <ufs/ufs/ufsmount.h>
55
56static MALLOC_DEFINE(M_DQUOT, "UFS quota", "UFS quota entries");
57
58/*
59 * Quota name to error message mapping.
60 */
61static char *quotatypes[] = INITQFNAMES;
62
63static int chkdqchg __P((struct inode *, long, struct ucred *, int));
64static int chkiqchg __P((struct inode *, long, struct ucred *, int));
65static int dqget __P((struct vnode *,
66 u_long, struct ufsmount *, int, struct dquot **));
67static int dqsync __P((struct vnode *, struct dquot *));
68static void dqflush __P((struct vnode *));
69
70#ifdef DIAGNOSTIC
71static void dqref __P((struct dquot *));
72static void chkdquot __P((struct inode *));
73#endif
74
75/*
76 * Set up the quotas for an inode.
77 *
78 * This routine completely defines the semantics of quotas.
79 * If other criterion want to be used to establish quotas, the
80 * MAXQUOTAS value in quotas.h should be increased, and the
81 * additional dquots set up here.
82 */
83int
84getinoquota(ip)
85 register struct inode *ip;
86{
87 struct ufsmount *ump;
88 struct vnode *vp = ITOV(ip);
89 int error;
90
91 ump = VFSTOUFS(vp->v_mount);
92 /*
93 * Set up the user quota based on file uid.
94 * EINVAL means that quotas are not enabled.
95 */
96 if (ip->i_dquot[USRQUOTA] == NODQUOT &&
97 (error =
98 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 =
107 dqget(vp, ip->i_gid, ump, GRPQUOTA, &ip->i_dquot[GRPQUOTA])) &&
108 error != EINVAL)
109 return (error);
110 return (0);
111}
112
113/*
114 * Update disk usage, and take corrective action.
115 */
116int
117chkdq(ip, change, cred, flags)
118 register struct inode *ip;
119 long change;
120 struct ucred *cred;
121 int flags;
122{
123 register struct dquot *dq;
124 register int i;
125 int ncurblocks, error;
126
127#ifdef DIAGNOSTIC
128 if ((flags & CHOWN) == 0)
129 chkdquot(ip);
130#endif
131 if (change == 0)
132 return (0);
133 if (change < 0) {
134 for (i = 0; i < MAXQUOTAS; i++) {
135 if ((dq = ip->i_dquot[i]) == NODQUOT)
136 continue;
137 while (dq->dq_flags & DQ_LOCK) {
138 dq->dq_flags |= DQ_WANT;
139 (void) tsleep((caddr_t)dq, PINOD+1, "chkdq1", 0);
140 }
141 ncurblocks = dq->dq_curblocks + change;
142 if (ncurblocks >= 0)
143 dq->dq_curblocks = ncurblocks;
144 else
145 dq->dq_curblocks = 0;
146 dq->dq_flags &= ~DQ_BLKS;
147 dq->dq_flags |= DQ_MOD;
148 }
149 return (0);
150 }
151 if ((flags & FORCE) == 0 && cred->cr_uid != 0) {
152 for (i = 0; i < MAXQUOTAS; i++) {
153 if ((dq = ip->i_dquot[i]) == NODQUOT)
154 continue;
155 error = chkdqchg(ip, change, cred, i);
156 if (error)
157 return (error);
158 }
159 }
160 for (i = 0; i < MAXQUOTAS; i++) {
161 if ((dq = ip->i_dquot[i]) == NODQUOT)
162 continue;
163 while (dq->dq_flags & DQ_LOCK) {
164 dq->dq_flags |= DQ_WANT;
165 (void) tsleep((caddr_t)dq, PINOD+1, "chkdq2", 0);
166 }
167 /* Reset timer when crossing soft limit */
168 if (dq->dq_curblocks + change >= dq->dq_bsoftlimit &&
169 dq->dq_curblocks < dq->dq_bsoftlimit)
170 dq->dq_btime = time_second +
171 VFSTOUFS(ITOV(ip)->v_mount)->um_btime[i];
172 dq->dq_curblocks += change;
173 dq->dq_flags |= DQ_MOD;
174 }
175 return (0);
176}
177
178/*
179 * Check for a valid change to a users allocation.
180 * Issue an error message if appropriate.
181 */
182static int
183chkdqchg(ip, change, cred, type)
184 struct inode *ip;
185 long change;
186 struct ucred *cred;
187 int type;
188{
189 register struct dquot *dq = ip->i_dquot[type];
190 long ncurblocks = dq->dq_curblocks + change;
191
192 /*
193 * If user would exceed their hard limit, disallow space allocation.
194 */
195 if (ncurblocks >= dq->dq_bhardlimit && dq->dq_bhardlimit) {
196 if ((dq->dq_flags & DQ_BLKS) == 0 &&
197 ip->i_uid == cred->cr_uid) {
198 uprintf("\n%s: write failed, %s disk limit reached\n",
199 ITOV(ip)->v_mount->mnt_stat.f_mntonname,
200 quotatypes[type]);
201 dq->dq_flags |= DQ_BLKS;
202 }
203 return (EDQUOT);
204 }
205 /*
206 * If user is over their soft limit for too long, disallow space
207 * allocation. Reset time limit as they cross their soft limit.
208 */
209 if (ncurblocks >= dq->dq_bsoftlimit && dq->dq_bsoftlimit) {
210 if (dq->dq_curblocks < dq->dq_bsoftlimit) {
211 dq->dq_btime = time_second +
212 VFSTOUFS(ITOV(ip)->v_mount)->um_btime[type];
213 if (ip->i_uid == cred->cr_uid)
214 uprintf("\n%s: warning, %s %s\n",
215 ITOV(ip)->v_mount->mnt_stat.f_mntonname,
216 quotatypes[type], "disk quota exceeded");
217 return (0);
218 }
219 if (time_second > dq->dq_btime) {
220 if ((dq->dq_flags & DQ_BLKS) == 0 &&
221 ip->i_uid == cred->cr_uid) {
222 uprintf("\n%s: write failed, %s %s\n",
223 ITOV(ip)->v_mount->mnt_stat.f_mntonname,
224 quotatypes[type],
225 "disk quota exceeded for too long");
226 dq->dq_flags |= DQ_BLKS;
227 }
228 return (EDQUOT);
229 }
230 }
231 return (0);
232}
233
234/*
235 * Check the inode limit, applying corrective action.
236 */
237int
238chkiq(ip, change, cred, flags)
239 register struct inode *ip;
240 long change;
241 struct ucred *cred;
242 int flags;
243{
244 register struct dquot *dq;
245 register int i;
246 int ncurinodes, error;
247
248#ifdef DIAGNOSTIC
249 if ((flags & CHOWN) == 0)
250 chkdquot(ip);
251#endif
252 if (change == 0)
253 return (0);
254 if (change < 0) {
255 for (i = 0; i < MAXQUOTAS; i++) {
256 if ((dq = ip->i_dquot[i]) == NODQUOT)
257 continue;
258 while (dq->dq_flags & DQ_LOCK) {
259 dq->dq_flags |= DQ_WANT;
260 (void) tsleep((caddr_t)dq, PINOD+1, "chkiq1", 0);
261 }
262 ncurinodes = dq->dq_curinodes + change;
263 if (ncurinodes >= 0)
264 dq->dq_curinodes = ncurinodes;
265 else
266 dq->dq_curinodes = 0;
267 dq->dq_flags &= ~DQ_INODS;
268 dq->dq_flags |= DQ_MOD;
269 }
270 return (0);
271 }
272 if ((flags & FORCE) == 0 && cred->cr_uid != 0) {
273 for (i = 0; i < MAXQUOTAS; i++) {
274 if ((dq = ip->i_dquot[i]) == NODQUOT)
275 continue;
276 error = chkiqchg(ip, change, cred, i);
277 if (error)
278 return (error);
279 }
280 }
281 for (i = 0; i < MAXQUOTAS; i++) {
282 if ((dq = ip->i_dquot[i]) == NODQUOT)
283 continue;
284 while (dq->dq_flags & DQ_LOCK) {
285 dq->dq_flags |= DQ_WANT;
286 (void) tsleep((caddr_t)dq, PINOD+1, "chkiq2", 0);
287 }
288 /* Reset timer when crossing soft limit */
289 if (dq->dq_curinodes + change >= dq->dq_isoftlimit &&
290 dq->dq_curinodes < dq->dq_isoftlimit)
291 dq->dq_itime = time_second +
292 VFSTOUFS(ITOV(ip)->v_mount)->um_itime[i];
293 dq->dq_curinodes += change;
294 dq->dq_flags |= DQ_MOD;
295 }
296 return (0);
297}
298
299/*
300 * Check for a valid change to a users allocation.
301 * Issue an error message if appropriate.
302 */
303static int
304chkiqchg(ip, change, cred, type)
305 struct inode *ip;
306 long change;
307 struct ucred *cred;
308 int type;
309{
310 register struct dquot *dq = ip->i_dquot[type];
311 long ncurinodes = dq->dq_curinodes + change;
312
313 /*
314 * If user would exceed their hard limit, disallow inode allocation.
315 */
316 if (ncurinodes >= dq->dq_ihardlimit && dq->dq_ihardlimit) {
317 if ((dq->dq_flags & DQ_INODS) == 0 &&
318 ip->i_uid == cred->cr_uid) {
319 uprintf("\n%s: write failed, %s inode limit reached\n",
320 ITOV(ip)->v_mount->mnt_stat.f_mntonname,
321 quotatypes[type]);
322 dq->dq_flags |= DQ_INODS;
323 }
324 return (EDQUOT);
325 }
326 /*
327 * If user is over their soft limit for too long, disallow inode
328 * allocation. Reset time limit as they cross their soft limit.
329 */
330 if (ncurinodes >= dq->dq_isoftlimit && dq->dq_isoftlimit) {
331 if (dq->dq_curinodes < dq->dq_isoftlimit) {
332 dq->dq_itime = time_second +
333 VFSTOUFS(ITOV(ip)->v_mount)->um_itime[type];
334 if (ip->i_uid == cred->cr_uid)
335 uprintf("\n%s: warning, %s %s\n",
336 ITOV(ip)->v_mount->mnt_stat.f_mntonname,
337 quotatypes[type], "inode quota exceeded");
338 return (0);
339 }
340 if (time_second > dq->dq_itime) {
341 if ((dq->dq_flags & DQ_INODS) == 0 &&
342 ip->i_uid == cred->cr_uid) {
343 uprintf("\n%s: write failed, %s %s\n",
344 ITOV(ip)->v_mount->mnt_stat.f_mntonname,
345 quotatypes[type],
346 "inode quota exceeded for too long");
347 dq->dq_flags |= DQ_INODS;
348 }
349 return (EDQUOT);
350 }
351 }
352 return (0);
353}
354
355#ifdef DIAGNOSTIC
356/*
357 * On filesystems with quotas enabled, it is an error for a file to change
358 * size and not to have a dquot structure associated with it.
359 */
360static void
361chkdquot(ip)
362 register struct inode *ip;
363{
364 struct ufsmount *ump = VFSTOUFS(ITOV(ip)->v_mount);
365 register int i;
366
367 for (i = 0; i < MAXQUOTAS; i++) {
368 if (ump->um_quotas[i] == NULLVP ||
369 (ump->um_qflags[i] & (QTF_OPENING|QTF_CLOSING)))
370 continue;
371 if (ip->i_dquot[i] == NODQUOT) {
372 vprint("chkdquot: missing dquot", ITOV(ip));
373 panic("chkdquot: missing dquot");
374 }
375 }
376}
377#endif
378
379/*
380 * Code to process quotactl commands.
381 */
382
383/*
384 * Q_QUOTAON - set up a quota file for a particular file system.
385 */
386int
387quotaon(p, mp, type, fname)
388 struct proc *p;
389 struct mount *mp;
390 register int type;
391 caddr_t fname;
392{
393 struct ufsmount *ump = VFSTOUFS(mp);
394 struct vnode *vp, **vpp;
395 struct vnode *nextvp;
396 struct dquot *dq;
397 int error;
398 struct nameidata nd;
399
400 vpp = &ump->um_quotas[type];
401 NDINIT(&nd, LOOKUP, FOLLOW, UIO_USERSPACE, fname, p);
402 error = vn_open(&nd, FREAD|FWRITE, 0);
403 if (error)
404 return (error);
405 NDFREE(&nd, NDF_ONLY_PNBUF);
406 vp = nd.ni_vp;
407 VOP_UNLOCK(vp, 0, p);
408 if (vp->v_type != VREG) {
409 (void) vn_close(vp, FREAD|FWRITE, p->p_ucred, p);
410 return (EACCES);
411 }
412 if (*vpp != vp)
413 quotaoff(p, mp, type);
414 ump->um_qflags[type] |= QTF_OPENING;
415 mp->mnt_flag |= MNT_QUOTA;
416 vp->v_flag |= VSYSTEM;
417 *vpp = vp;
418 /*
419 * Save the credential of the process that turned on quotas.
420 * Set up the time limits for this quota.
421 */
422 crhold(p->p_ucred);
423 ump->um_cred[type] = p->p_ucred;
424 ump->um_btime[type] = MAX_DQ_TIME;
425 ump->um_itime[type] = MAX_IQ_TIME;
426 if (dqget(NULLVP, 0, ump, type, &dq) == 0) {
427 if (dq->dq_btime > 0)
428 ump->um_btime[type] = dq->dq_btime;
429 if (dq->dq_itime > 0)
430 ump->um_itime[type] = dq->dq_itime;
431 dqrele(NULLVP, dq);
432 }
433 /*
434 * Search vnodes associated with this mount point,
435 * adding references to quota file being opened.
436 * NB: only need to add dquot's for inodes being modified.
437 */
438again:
439 for (vp = TAILQ_FIRST(&mp->mnt_nvnodelist); vp != NULL; vp = nextvp) {
440 nextvp = TAILQ_NEXT(vp, v_nmntvnodes);
441 if (vp->v_type == VNON || vp->v_writecount == 0)
442 continue;
443 if (vget(vp, LK_EXCLUSIVE, p))
444 goto again;
445 error = getinoquota(VTOI(vp));
446 if (error) {
447 vput(vp);
448 break;
449 }
450 vput(vp);
451 if (TAILQ_NEXT(vp, v_nmntvnodes) != nextvp || vp->v_mount != mp)
452 goto again;
453 }
454 ump->um_qflags[type] &= ~QTF_OPENING;
455 if (error)
456 quotaoff(p, mp, type);
457 return (error);
458}
459
460/*
461 * Q_QUOTAOFF - turn off disk quotas for a filesystem.
462 */
463int
464quotaoff(p, mp, type)
465 struct proc *p;
466 struct mount *mp;
467 register int type;
468{
469 struct vnode *vp;
470 struct vnode *qvp, *nextvp;
471 struct ufsmount *ump = VFSTOUFS(mp);
472 struct dquot *dq;
473 struct inode *ip;
474 int error;
475
476 if ((qvp = ump->um_quotas[type]) == NULLVP)
477 return (0);
478 ump->um_qflags[type] |= QTF_CLOSING;
479 /*
480 * Search vnodes associated with this mount point,
481 * deleting any references to quota file being closed.
482 */
483again:
484 for (vp = TAILQ_FIRST(&mp->mnt_nvnodelist); vp != NULL; vp = nextvp) {
485 nextvp = TAILQ_NEXT(vp, v_nmntvnodes);
486 if (vp->v_type == VNON)
487 continue;
488 if (vget(vp, LK_EXCLUSIVE, p))
489 goto again;
490 ip = VTOI(vp);
491 dq = ip->i_dquot[type];
492 ip->i_dquot[type] = NODQUOT;
493 dqrele(vp, dq);
494 vput(vp);
495 if (TAILQ_NEXT(vp, v_nmntvnodes) != nextvp || vp->v_mount != mp)
496 goto again;
497 }
498 dqflush(qvp);
499 qvp->v_flag &= ~VSYSTEM;
500 error = vn_close(qvp, FREAD|FWRITE, p->p_ucred, p);
501 ump->um_quotas[type] = NULLVP;
502 crfree(ump->um_cred[type]);
503 ump->um_cred[type] = NOCRED;
504 ump->um_qflags[type] &= ~QTF_CLOSING;
505 for (type = 0; type < MAXQUOTAS; type++)
506 if (ump->um_quotas[type] != NULLVP)
507 break;
508 if (type == MAXQUOTAS)
509 mp->mnt_flag &= ~MNT_QUOTA;
510 return (error);
511}
512
513/*
514 * Q_GETQUOTA - return current values in a dqblk structure.
515 */
516int
517getquota(mp, id, type, addr)
518 struct mount *mp;
519 u_long id;
520 int type;
521 caddr_t addr;
522{
523 struct dquot *dq;
524 int error;
525
526 error = dqget(NULLVP, id, VFSTOUFS(mp), type, &dq);
527 if (error)
528 return (error);
529 error = copyout((caddr_t)&dq->dq_dqb, addr, sizeof (struct dqblk));
530 dqrele(NULLVP, dq);
531 return (error);
532}
533
534/*
535 * Q_SETQUOTA - assign an entire dqblk structure.
536 */
537int
538setquota(mp, id, type, addr)
539 struct mount *mp;
540 u_long id;
541 int type;
542 caddr_t addr;
543{
544 register struct dquot *dq;
545 struct dquot *ndq;
546 struct ufsmount *ump = VFSTOUFS(mp);
547 struct dqblk newlim;
548 int error;
549
550 error = copyin(addr, (caddr_t)&newlim, sizeof (struct dqblk));
551 if (error)
552 return (error);
553 error = dqget(NULLVP, id, ump, type, &ndq);
554 if (error)
555 return (error);
556 dq = ndq;
557 while (dq->dq_flags & DQ_LOCK) {
558 dq->dq_flags |= DQ_WANT;
559 (void) tsleep((caddr_t)dq, PINOD+1, "setqta", 0);
560 }
561 /*
562 * Copy all but the current values.
563 * Reset time limit if previously had no soft limit or were
564 * under it, but now have a soft limit and are over it.
565 */
566 newlim.dqb_curblocks = dq->dq_curblocks;
567 newlim.dqb_curinodes = dq->dq_curinodes;
568 if (dq->dq_id != 0) {
569 newlim.dqb_btime = dq->dq_btime;
570 newlim.dqb_itime = dq->dq_itime;
571 }
572 if (newlim.dqb_bsoftlimit &&
573 dq->dq_curblocks >= newlim.dqb_bsoftlimit &&
574 (dq->dq_bsoftlimit == 0 || dq->dq_curblocks < dq->dq_bsoftlimit))
575 newlim.dqb_btime = time_second + ump->um_btime[type];
576 if (newlim.dqb_isoftlimit &&
577 dq->dq_curinodes >= newlim.dqb_isoftlimit &&
578 (dq->dq_isoftlimit == 0 || dq->dq_curinodes < dq->dq_isoftlimit))
579 newlim.dqb_itime = time_second + ump->um_itime[type];
580 dq->dq_dqb = newlim;
581 if (dq->dq_curblocks < dq->dq_bsoftlimit)
582 dq->dq_flags &= ~DQ_BLKS;
583 if (dq->dq_curinodes < dq->dq_isoftlimit)
584 dq->dq_flags &= ~DQ_INODS;
585 if (dq->dq_isoftlimit == 0 && dq->dq_bsoftlimit == 0 &&
586 dq->dq_ihardlimit == 0 && dq->dq_bhardlimit == 0)
587 dq->dq_flags |= DQ_FAKE;
588 else
589 dq->dq_flags &= ~DQ_FAKE;
590 dq->dq_flags |= DQ_MOD;
591 dqrele(NULLVP, dq);
592 return (0);
593}
594
595/*
596 * Q_SETUSE - set current inode and block usage.
597 */
598int
599setuse(mp, id, type, addr)
600 struct mount *mp;
601 u_long id;
602 int type;
603 caddr_t addr;
604{
605 register struct dquot *dq;
606 struct ufsmount *ump = VFSTOUFS(mp);
607 struct dquot *ndq;
608 struct dqblk usage;
609 int error;
610
611 error = copyin(addr, (caddr_t)&usage, sizeof (struct dqblk));
612 if (error)
613 return (error);
614 error = dqget(NULLVP, id, ump, type, &ndq);
615 if (error)
616 return (error);
617 dq = ndq;
618 while (dq->dq_flags & DQ_LOCK) {
619 dq->dq_flags |= DQ_WANT;
620 (void) tsleep((caddr_t)dq, PINOD+1, "setuse", 0);
621 }
622 /*
623 * Reset time limit if have a soft limit and were
624 * previously under it, but are now over it.
625 */
626 if (dq->dq_bsoftlimit && dq->dq_curblocks < dq->dq_bsoftlimit &&
627 usage.dqb_curblocks >= dq->dq_bsoftlimit)
628 dq->dq_btime = time_second + ump->um_btime[type];
629 if (dq->dq_isoftlimit && dq->dq_curinodes < dq->dq_isoftlimit &&
630 usage.dqb_curinodes >= dq->dq_isoftlimit)
631 dq->dq_itime = time_second + ump->um_itime[type];
632 dq->dq_curblocks = usage.dqb_curblocks;
633 dq->dq_curinodes = usage.dqb_curinodes;
634 if (dq->dq_curblocks < dq->dq_bsoftlimit)
635 dq->dq_flags &= ~DQ_BLKS;
636 if (dq->dq_curinodes < dq->dq_isoftlimit)
637 dq->dq_flags &= ~DQ_INODS;
638 dq->dq_flags |= DQ_MOD;
639 dqrele(NULLVP, dq);
640 return (0);
641}
642
643/*
644 * Q_SYNC - sync quota files to disk.
645 */
646int
647qsync(mp)
648 struct mount *mp;
649{
650 struct ufsmount *ump = VFSTOUFS(mp);
651 struct proc *p = curproc; /* XXX */
652 struct vnode *vp, *nextvp;
653 struct dquot *dq;
654 int i, error;
655
656 /*
657 * Check if the mount point has any quotas.
658 * If not, simply return.
659 */
660 for (i = 0; i < MAXQUOTAS; i++)
661 if (ump->um_quotas[i] != NULLVP)
662 break;
663 if (i == MAXQUOTAS)
664 return (0);
665 /*
666 * Search vnodes associated with this mount point,
667 * synchronizing any modified dquot structures.
668 */
669 simple_lock(&mntvnode_slock);
670again:
671 for (vp = TAILQ_FIRST(&mp->mnt_nvnodelist); vp != NULL; vp = nextvp) {
672 if (vp->v_mount != mp)
673 goto again;
674 nextvp = TAILQ_NEXT(vp, v_nmntvnodes);
675 if (vp->v_type == VNON)
676 continue;
677 simple_lock(&vp->v_interlock);
678 simple_unlock(&mntvnode_slock);
679 error = vget(vp, LK_EXCLUSIVE | LK_NOWAIT | LK_INTERLOCK, p);
680 if (error) {
681 simple_lock(&mntvnode_slock);
682 if (error == ENOENT)
683 goto again;
684 continue;
685 }
686 for (i = 0; i < MAXQUOTAS; i++) {
687 dq = VTOI(vp)->i_dquot[i];
688 if (dq != NODQUOT && (dq->dq_flags & DQ_MOD))
689 dqsync(vp, dq);
690 }
691 vput(vp);
692 simple_lock(&mntvnode_slock);
693 if (TAILQ_NEXT(vp, v_nmntvnodes) != nextvp)
694 goto again;
695 }
696 simple_unlock(&mntvnode_slock);
697 return (0);
698}
699
700/*
701 * Code pertaining to management of the in-core dquot data structures.
702 */
703#define DQHASH(dqvp, id) \
704 (&dqhashtbl[((((intptr_t)(dqvp)) >> 8) + id) & dqhash])
705static LIST_HEAD(dqhash, dquot) *dqhashtbl;
706static u_long dqhash;
707
708/*
709 * Dquot free list.
710 */
711#define DQUOTINC 5 /* minimum free dquots desired */
712static TAILQ_HEAD(dqfreelist, dquot) dqfreelist;
713static long numdquot, desireddquot = DQUOTINC;
714
715/*
716 * Initialize the quota system.
717 */
718void
719dqinit()
720{
721
722 dqhashtbl = hashinit(desiredvnodes, M_DQUOT, &dqhash);
723 TAILQ_INIT(&dqfreelist);
724}
725
726/*
727 * Obtain a dquot structure for the specified identifier and quota file
728 * reading the information from the file if necessary.
729 */
730static int
731dqget(vp, id, ump, type, dqp)
732 struct vnode *vp;
733 u_long id;
734 register struct ufsmount *ump;
735 register int type;
736 struct dquot **dqp;
737{
738 struct proc *p = curproc; /* XXX */
739 struct dquot *dq;
740 struct dqhash *dqh;
741 struct vnode *dqvp;
742 struct iovec aiov;
743 struct uio auio;
744 int error;
745
746 dqvp = ump->um_quotas[type];
747 if (dqvp == NULLVP || (ump->um_qflags[type] & QTF_CLOSING)) {
748 *dqp = NODQUOT;
749 return (EINVAL);
750 }
751 /*
752 * Check the cache first.
753 */
754 dqh = DQHASH(dqvp, id);
755 for (dq = dqh->lh_first; dq; dq = dq->dq_hash.le_next) {
756 if (dq->dq_id != id ||
757 dq->dq_ump->um_quotas[dq->dq_type] != dqvp)
758 continue;
759 /*
760 * Cache hit with no references. Take
761 * the structure off the free list.
762 */
763 if (dq->dq_cnt == 0)
764 TAILQ_REMOVE(&dqfreelist, dq, dq_freelist);
765 DQREF(dq);
766 *dqp = dq;
767 return (0);
768 }
769 /*
770 * Not in cache, allocate a new one.
771 */
772 if (dqfreelist.tqh_first == NODQUOT &&
773 numdquot < MAXQUOTAS * desiredvnodes)
774 desireddquot += DQUOTINC;
775 if (numdquot < desireddquot) {
776 dq = (struct dquot *)malloc(sizeof *dq, M_DQUOT, M_WAITOK);
777 bzero((char *)dq, sizeof *dq);
778 numdquot++;
779 } else {
780 if ((dq = dqfreelist.tqh_first) == NULL) {
781 tablefull("dquot");
782 *dqp = NODQUOT;
783 return (EUSERS);
784 }
785 if (dq->dq_cnt || (dq->dq_flags & DQ_MOD))
786 panic("dqget: free dquot isn't");
787 TAILQ_REMOVE(&dqfreelist, dq, dq_freelist);
788 if (dq->dq_ump != NULL)
789 LIST_REMOVE(dq, dq_hash);
790 }
791 /*
792 * Initialize the contents of the dquot structure.
793 */
794 if (vp != dqvp)
795 vn_lock(dqvp, LK_EXCLUSIVE | LK_RETRY, p);
796 LIST_INSERT_HEAD(dqh, dq, dq_hash);
797 DQREF(dq);
798 dq->dq_flags = DQ_LOCK;
799 dq->dq_id = id;
800 dq->dq_ump = ump;
801 dq->dq_type = type;
802 auio.uio_iov = &aiov;
803 auio.uio_iovcnt = 1;
804 aiov.iov_base = (caddr_t)&dq->dq_dqb;
805 aiov.iov_len = sizeof (struct dqblk);
806 auio.uio_resid = sizeof (struct dqblk);
807 auio.uio_offset = (off_t)(id * sizeof (struct dqblk));
808 auio.uio_segflg = UIO_SYSSPACE;
809 auio.uio_rw = UIO_READ;
810 auio.uio_procp = (struct proc *)0;
811 error = VOP_READ(dqvp, &auio, 0, ump->um_cred[type]);
812 if (auio.uio_resid == sizeof(struct dqblk) && error == 0)
813 bzero((caddr_t)&dq->dq_dqb, sizeof(struct dqblk));
814 if (vp != dqvp)
815 VOP_UNLOCK(dqvp, 0, p);
816 if (dq->dq_flags & DQ_WANT)
817 wakeup((caddr_t)dq);
818 dq->dq_flags = 0;
819 /*
820 * I/O error in reading quota file, release
821 * quota structure and reflect problem to caller.
822 */
823 if (error) {
824 LIST_REMOVE(dq, dq_hash);
825 dqrele(vp, dq);
826 *dqp = NODQUOT;
827 return (error);
828 }
829 /*
830 * Check for no limit to enforce.
831 * Initialize time values if necessary.
832 */
833 if (dq->dq_isoftlimit == 0 && dq->dq_bsoftlimit == 0 &&
834 dq->dq_ihardlimit == 0 && dq->dq_bhardlimit == 0)
835 dq->dq_flags |= DQ_FAKE;
836 if (dq->dq_id != 0) {
837 if (dq->dq_btime == 0)
838 dq->dq_btime = time_second + ump->um_btime[type];
839 if (dq->dq_itime == 0)
840 dq->dq_itime = time_second + ump->um_itime[type];
841 }
842 *dqp = dq;
843 return (0);
844}
845
846#ifdef DIAGNOSTIC
847/*
848 * Obtain a reference to a dquot.
849 */
850static void
851dqref(dq)
852 struct dquot *dq;
853{
854
855 dq->dq_cnt++;
856}
857#endif
858
859/*
860 * Release a reference to a dquot.
861 */
862void
863dqrele(vp, dq)
864 struct vnode *vp;
865 register struct dquot *dq;
866{
867
868 if (dq == NODQUOT)
869 return;
870 if (dq->dq_cnt > 1) {
871 dq->dq_cnt--;
872 return;
873 }
874 if (dq->dq_flags & DQ_MOD)
875 (void) dqsync(vp, dq);
876 if (--dq->dq_cnt > 0)
877 return;
878 TAILQ_INSERT_TAIL(&dqfreelist, dq, dq_freelist);
879}
880
881/*
882 * Update the disk quota in the quota file.
883 */
884static int
885dqsync(vp, dq)
886 struct vnode *vp;
887 struct dquot *dq;
888{
889 struct proc *p = curproc; /* XXX */
890 struct vnode *dqvp;
891 struct iovec aiov;
892 struct uio auio;
893 int error;
894
895 if (dq == NODQUOT)
896 panic("dqsync: dquot");
897 if ((dq->dq_flags & DQ_MOD) == 0)
898 return (0);
899 if ((dqvp = dq->dq_ump->um_quotas[dq->dq_type]) == NULLVP)
900 panic("dqsync: file");
901 if (vp != dqvp)
902 vn_lock(dqvp, LK_EXCLUSIVE | LK_RETRY, p);
903 while (dq->dq_flags & DQ_LOCK) {
904 dq->dq_flags |= DQ_WANT;
905 (void) tsleep((caddr_t)dq, PINOD+2, "dqsync", 0);
906 if ((dq->dq_flags & DQ_MOD) == 0) {
907 if (vp != dqvp)
908 VOP_UNLOCK(dqvp, 0, p);
909 return (0);
910 }
911 }
912 dq->dq_flags |= DQ_LOCK;
913 auio.uio_iov = &aiov;
914 auio.uio_iovcnt = 1;
915 aiov.iov_base = (caddr_t)&dq->dq_dqb;
916 aiov.iov_len = sizeof (struct dqblk);
917 auio.uio_resid = sizeof (struct dqblk);
918 auio.uio_offset = (off_t)(dq->dq_id * sizeof (struct dqblk));
919 auio.uio_segflg = UIO_SYSSPACE;
920 auio.uio_rw = UIO_WRITE;
921 auio.uio_procp = (struct proc *)0;
922 error = VOP_WRITE(dqvp, &auio, 0, dq->dq_ump->um_cred[dq->dq_type]);
923 if (auio.uio_resid && error == 0)
924 error = EIO;
925 if (dq->dq_flags & DQ_WANT)
926 wakeup((caddr_t)dq);
927 dq->dq_flags &= ~(DQ_MOD|DQ_LOCK|DQ_WANT);
928 if (vp != dqvp)
929 VOP_UNLOCK(dqvp, 0, p);
930 return (error);
931}
932
933/*
934 * Flush all entries from the cache for a particular vnode.
935 */
936static void
937dqflush(vp)
938 register struct vnode *vp;
939{
940 register struct dquot *dq, *nextdq;
941 struct dqhash *dqh;
942
943 /*
944 * Move all dquot's that used to refer to this quota
945 * file off their hash chains (they will eventually
946 * fall off the head of the free list and be re-used).
947 */
948 for (dqh = &dqhashtbl[dqhash]; dqh >= dqhashtbl; dqh--) {
949 for (dq = dqh->lh_first; dq; dq = nextdq) {
950 nextdq = dq->dq_hash.le_next;
951 if (dq->dq_ump->um_quotas[dq->dq_type] != vp)
952 continue;
953 if (dq->dq_cnt)
954 panic("dqflush: stray dquot");
955 LIST_REMOVE(dq, dq_hash);
956 dq->dq_ump = (struct ufsmount *)0;
957 }
958 }
959}