]> git.saurik.com Git - apple/xnu.git/blame - bsd/hfs/hfs_quota.c
xnu-1228.tar.gz
[apple/xnu.git] / bsd / hfs / hfs_quota.c
CommitLineData
9bccf70c 1/*
2d21ac55 2 * Copyright (c) 2002-2005 Apple Computer, Inc. All rights reserved.
9bccf70c 3 *
2d21ac55 4 * @APPLE_OSREFERENCE_LICENSE_HEADER_START@
9bccf70c 5 *
2d21ac55
A
6 * This file contains Original Code and/or Modifications of Original Code
7 * as defined in and that are subject to the Apple Public Source License
8 * Version 2.0 (the 'License'). You may not use this file except in
9 * compliance with the License. The rights granted to you under the License
10 * may not be used to create, or enable the creation or redistribution of,
11 * unlawful or unlicensed copies of an Apple operating system, or to
12 * circumvent, violate, or enable the circumvention or violation of, any
13 * terms of an Apple operating system software license agreement.
8f6c56a5 14 *
2d21ac55
A
15 * Please obtain a copy of the License at
16 * http://www.opensource.apple.com/apsl/ and read it before using this file.
17 *
18 * The Original Code and all software distributed under the License are
19 * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
8f6c56a5
A
20 * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
21 * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
2d21ac55
A
22 * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
23 * Please see the License for the specific language governing rights and
24 * limitations under the License.
8f6c56a5 25 *
2d21ac55 26 * @APPLE_OSREFERENCE_LICENSE_HEADER_END@
9bccf70c
A
27 */
28/*
29 * Copyright (c) 1982, 1986, 1990, 1993, 1995
30 * The Regents of the University of California. All rights reserved.
31 *
32 * This code is derived from software contributed to Berkeley by
33 * Robert Elz at The University of Melbourne.
34 *
35 * Redistribution and use in source and binary forms, with or without
36 * modification, are permitted provided that the following conditions
37 * are met:
38 * 1. Redistributions of source code must retain the above copyright
39 * notice, this list of conditions and the following disclaimer.
40 * 2. Redistributions in binary form must reproduce the above copyright
41 * notice, this list of conditions and the following disclaimer in the
42 * documentation and/or other materials provided with the distribution.
43 * 3. All advertising materials mentioning features or use of this software
44 * must display the following acknowledgement:
45 * This product includes software developed by the University of
46 * California, Berkeley and its contributors.
47 * 4. Neither the name of the University nor the names of its contributors
48 * may be used to endorse or promote products derived from this software
49 * without specific prior written permission.
50 *
51 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
52 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
53 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
54 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
55 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
56 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
57 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
58 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
59 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
60 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
61 * SUCH DAMAGE.
62 *
63 * @(#)hfs_quota.c
64 * derived from @(#)ufs_quota.c 8.5 (Berkeley) 5/20/95
65 */
66
67#include <sys/param.h>
68#include <sys/kernel.h>
69#include <sys/systm.h>
70#include <sys/mount.h>
9bccf70c
A
71#include <sys/malloc.h>
72#include <sys/file.h>
73#include <sys/proc.h>
91447636 74#include <sys/kauth.h>
9bccf70c 75#include <sys/vnode.h>
2d21ac55 76#include <sys/vnode_internal.h>
9bccf70c 77#include <sys/quota.h>
91447636
A
78#include <sys/proc_internal.h>
79#include <kern/kalloc.h>
9bccf70c
A
80
81#include <hfs/hfs.h>
82#include <hfs/hfs_cnode.h>
83#include <hfs/hfs_quota.h>
84#include <hfs/hfs_mount.h>
85
2d21ac55 86
9bccf70c
A
87/*
88 * Quota name to error message mapping.
89 */
91447636 90#if 0
9bccf70c 91static char *quotatypes[] = INITQFNAMES;
91447636 92#endif
9bccf70c
A
93
94/*
95 * Set up the quotas for a cnode.
96 *
97 * This routine completely defines the semantics of quotas.
98 * If other criterion want to be used to establish quotas, the
99 * MAXQUOTAS value in quotas.h should be increased, and the
100 * additional dquots set up here.
101 */
102int
103hfs_getinoquota(cp)
104 register struct cnode *cp;
105{
106 struct hfsmount *hfsmp;
107 struct vnode *vp;
108 int error;
109
110 vp = cp->c_vp ? cp->c_vp : cp->c_rsrc_vp;
91447636 111 hfsmp = VTOHFS(vp);
9bccf70c
A
112 /*
113 * Set up the user quota based on file uid.
114 * EINVAL means that quotas are not enabled.
115 */
116 if (cp->c_dquot[USRQUOTA] == NODQUOT &&
117 (error =
91447636 118 dqget(cp->c_uid, &hfsmp->hfs_qfiles[USRQUOTA], USRQUOTA, &cp->c_dquot[USRQUOTA])) &&
9bccf70c
A
119 error != EINVAL)
120 return (error);
121 /*
122 * Set up the group quota based on file gid.
123 * EINVAL means that quotas are not enabled.
124 */
125 if (cp->c_dquot[GRPQUOTA] == NODQUOT &&
126 (error =
91447636 127 dqget(cp->c_gid, &hfsmp->hfs_qfiles[GRPQUOTA], GRPQUOTA, &cp->c_dquot[GRPQUOTA])) &&
9bccf70c
A
128 error != EINVAL)
129 return (error);
130 return (0);
131}
132
133/*
134 * Update disk usage, and take corrective action.
135 */
136int
137hfs_chkdq(cp, change, cred, flags)
138 register struct cnode *cp;
139 int64_t change;
91447636 140 kauth_cred_t cred;
9bccf70c
A
141 int flags;
142{
143 register struct dquot *dq;
144 register int i;
145 int64_t ncurbytes;
d7e50217 146 int error=0;
9bccf70c
A
147 struct proc *p;
148
149#if DIAGNOSTIC
150 if ((flags & CHOWN) == 0)
151 hfs_chkdquot(cp);
152#endif
153 if (change == 0)
154 return (0);
155 if (change < 0) {
156 for (i = 0; i < MAXQUOTAS; i++) {
157 if ((dq = cp->c_dquot[i]) == NODQUOT)
158 continue;
91447636
A
159 dqlock(dq);
160
9bccf70c
A
161 ncurbytes = dq->dq_curbytes + change;
162 if (ncurbytes >= 0)
163 dq->dq_curbytes = ncurbytes;
164 else
165 dq->dq_curbytes = 0;
166 dq->dq_flags &= ~DQ_BLKS;
167 dq->dq_flags |= DQ_MOD;
91447636
A
168
169 dqunlock(dq);
9bccf70c
A
170 }
171 return (0);
172 }
173 p = current_proc();
0c530ab8
A
174 /*
175 * This use of proc_ucred() is safe because kernproc credential never
176 * changes.
177 */
178 if (!IS_VALID_CRED(cred))
91447636
A
179 cred = proc_ucred(kernproc);
180 if (suser(cred, NULL) || proc_forcequota(p)) {
9bccf70c
A
181 for (i = 0; i < MAXQUOTAS; i++) {
182 if ((dq = cp->c_dquot[i]) == NODQUOT)
183 continue;
d7e50217
A
184 error = hfs_chkdqchg(cp, change, cred, i);
185 if (error) {
186 break;
187 }
9bccf70c
A
188 }
189 }
d7e50217
A
190 if ((flags & FORCE) || error == 0) {
191 for (i = 0; i < MAXQUOTAS; i++) {
192 if ((dq = cp->c_dquot[i]) == NODQUOT)
193 continue;
91447636
A
194 dqlock(dq);
195
d7e50217
A
196 dq->dq_curbytes += change;
197 dq->dq_flags |= DQ_MOD;
91447636
A
198
199 dqunlock(dq);
9bccf70c 200 }
9bccf70c 201 }
d7e50217 202 return (error);
9bccf70c
A
203}
204
205/*
206 * Check for a valid change to a users allocation.
207 * Issue an error message if appropriate.
208 */
209int
210hfs_chkdqchg(cp, change, cred, type)
211 struct cnode *cp;
212 int64_t change;
91447636 213 kauth_cred_t cred;
9bccf70c
A
214 int type;
215{
216 register struct dquot *dq = cp->c_dquot[type];
91447636 217 u_int64_t ncurbytes;
9bccf70c 218 struct vnode *vp = cp->c_vp ? cp->c_vp : cp->c_rsrc_vp;
91447636
A
219
220 dqlock(dq);
221
222 ncurbytes = dq->dq_curbytes + change;
9bccf70c
A
223 /*
224 * If user would exceed their hard limit, disallow space allocation.
225 */
226 if (ncurbytes >= dq->dq_bhardlimit && dq->dq_bhardlimit) {
227 if ((dq->dq_flags & DQ_BLKS) == 0 &&
91447636 228 cp->c_uid == kauth_cred_getuid(cred)) {
d7e50217 229#if 0
91447636 230 printf("\nwrite failed, %s disk limit reached\n",
9bccf70c
A
231 quotatypes[type]);
232#endif
233 dq->dq_flags |= DQ_BLKS;
234 }
91447636
A
235 dqunlock(dq);
236
9bccf70c
A
237 return (EDQUOT);
238 }
239 /*
240 * If user is over their soft limit for too long, disallow space
241 * allocation. Reset time limit as they cross their soft limit.
242 */
243 if (ncurbytes >= dq->dq_bsoftlimit && dq->dq_bsoftlimit) {
91447636
A
244 struct timeval tv;
245
246 microuptime(&tv);
9bccf70c 247 if (dq->dq_curbytes < dq->dq_bsoftlimit) {
91447636
A
248 dq->dq_btime = tv.tv_sec +
249 VTOHFS(vp)->hfs_qfiles[type].qf_btime;
d7e50217 250#if 0
91447636
A
251 if (cp->c_uid == kauth_cred_getuid(cred))
252 printf("\nwarning, %s %s\n",
9bccf70c
A
253 quotatypes[type], "disk quota exceeded");
254#endif
91447636
A
255 dqunlock(dq);
256
9bccf70c
A
257 return (0);
258 }
91447636 259 if (tv.tv_sec > dq->dq_btime) {
9bccf70c 260 if ((dq->dq_flags & DQ_BLKS) == 0 &&
91447636 261 cp->c_uid == kauth_cred_getuid(cred)) {
d7e50217 262#if 0
91447636 263 printf("\nwrite failed, %s %s\n",
9bccf70c
A
264 quotatypes[type],
265 "disk quota exceeded for too long");
266#endif
267 dq->dq_flags |= DQ_BLKS;
268 }
91447636
A
269 dqunlock(dq);
270
9bccf70c
A
271 return (EDQUOT);
272 }
273 }
91447636
A
274 dqunlock(dq);
275
9bccf70c
A
276 return (0);
277}
278
279/*
280 * Check the inode limit, applying corrective action.
281 */
282int
283hfs_chkiq(cp, change, cred, flags)
284 register struct cnode *cp;
285 long change;
91447636 286 kauth_cred_t cred;
9bccf70c
A
287 int flags;
288{
289 register struct dquot *dq;
290 register int i;
d7e50217 291 int ncurinodes, error=0;
9bccf70c
A
292 struct proc *p;
293
294#if DIAGNOSTIC
295 if ((flags & CHOWN) == 0)
296 hfs_chkdquot(cp);
297#endif
298 if (change == 0)
299 return (0);
300 if (change < 0) {
301 for (i = 0; i < MAXQUOTAS; i++) {
302 if ((dq = cp->c_dquot[i]) == NODQUOT)
303 continue;
91447636
A
304 dqlock(dq);
305
9bccf70c
A
306 ncurinodes = dq->dq_curinodes + change;
307 if (ncurinodes >= 0)
308 dq->dq_curinodes = ncurinodes;
309 else
310 dq->dq_curinodes = 0;
311 dq->dq_flags &= ~DQ_INODS;
312 dq->dq_flags |= DQ_MOD;
91447636
A
313
314 dqunlock(dq);
9bccf70c
A
315 }
316 return (0);
317 }
318 p = current_proc();
0c530ab8
A
319 /*
320 * This use of proc_ucred() is safe because kernproc credential never
321 * changes.
322 */
323 if (!IS_VALID_CRED(cred))
91447636
A
324 cred = proc_ucred(kernproc);
325 if (suser(cred, NULL) || proc_forcequota(p)) {
9bccf70c
A
326 for (i = 0; i < MAXQUOTAS; i++) {
327 if ((dq = cp->c_dquot[i]) == NODQUOT)
328 continue;
d7e50217
A
329 error = hfs_chkiqchg(cp, change, cred, i);
330 if (error) {
331 break;
332 }
9bccf70c
A
333 }
334 }
d7e50217
A
335 if ((flags & FORCE) || error == 0) {
336 for (i = 0; i < MAXQUOTAS; i++) {
337 if ((dq = cp->c_dquot[i]) == NODQUOT)
338 continue;
91447636
A
339 dqlock(dq);
340
d7e50217
A
341 dq->dq_curinodes += change;
342 dq->dq_flags |= DQ_MOD;
91447636
A
343
344 dqunlock(dq);
9bccf70c 345 }
9bccf70c 346 }
d7e50217 347 return (error);
9bccf70c
A
348}
349
350/*
351 * Check for a valid change to a users allocation.
352 * Issue an error message if appropriate.
353 */
354int
355hfs_chkiqchg(cp, change, cred, type)
356 struct cnode *cp;
357 long change;
91447636 358 kauth_cred_t cred;
9bccf70c
A
359 int type;
360{
361 register struct dquot *dq = cp->c_dquot[type];
2d21ac55 362 unsigned long ncurinodes;
9bccf70c
A
363 struct vnode *vp = cp->c_vp ? cp->c_vp : cp->c_rsrc_vp;
364
91447636
A
365 dqlock(dq);
366
367 ncurinodes = dq->dq_curinodes + change;
9bccf70c
A
368 /*
369 * If user would exceed their hard limit, disallow cnode allocation.
370 */
371 if (ncurinodes >= dq->dq_ihardlimit && dq->dq_ihardlimit) {
372 if ((dq->dq_flags & DQ_INODS) == 0 &&
91447636 373 cp->c_uid == kauth_cred_getuid(cred)) {
d7e50217 374#if 0
91447636 375 printf("\nwrite failed, %s cnode limit reached\n",
9bccf70c
A
376 quotatypes[type]);
377#endif
378 dq->dq_flags |= DQ_INODS;
379 }
91447636
A
380 dqunlock(dq);
381
9bccf70c
A
382 return (EDQUOT);
383 }
384 /*
385 * If user is over their soft limit for too long, disallow cnode
386 * allocation. Reset time limit as they cross their soft limit.
387 */
388 if (ncurinodes >= dq->dq_isoftlimit && dq->dq_isoftlimit) {
91447636
A
389 struct timeval tv;
390
391 microuptime(&tv);
9bccf70c 392 if (dq->dq_curinodes < dq->dq_isoftlimit) {
91447636
A
393 dq->dq_itime = tv.tv_sec +
394 VTOHFS(vp)->hfs_qfiles[type].qf_itime;
d7e50217 395#if 0
91447636
A
396 if (cp->c_uid == kauth_cred_getuid(cred))
397 printf("\nwarning, %s %s\n",
9bccf70c
A
398 quotatypes[type], "cnode quota exceeded");
399#endif
91447636
A
400 dqunlock(dq);
401
9bccf70c
A
402 return (0);
403 }
91447636 404 if (tv.tv_sec > dq->dq_itime) {
9bccf70c 405 if ((dq->dq_flags & DQ_INODS) == 0 &&
91447636 406 cp->c_uid == kauth_cred_getuid(cred)) {
d7e50217 407#if 0
91447636 408 printf("\nwrite failed, %s %s\n",
9bccf70c
A
409 quotatypes[type],
410 "cnode quota exceeded for too long");
411#endif
412 dq->dq_flags |= DQ_INODS;
413 }
91447636
A
414 dqunlock(dq);
415
9bccf70c
A
416 return (EDQUOT);
417 }
418 }
91447636
A
419 dqunlock(dq);
420
9bccf70c
A
421 return (0);
422}
423
424#if DIAGNOSTIC
425/*
426 * On filesystems with quotas enabled, it is an error for a file to change
427 * size and not to have a dquot structure associated with it.
428 */
429void
430hfs_chkdquot(cp)
431 register struct cnode *cp;
432{
433 struct vnode *vp = cp->c_vp ? cp->c_vp : cp->c_rsrc_vp;
91447636 434 struct hfsmount *hfsmp = VTOHFS(vp);
9bccf70c
A
435 register int i;
436
437 for (i = 0; i < MAXQUOTAS; i++) {
91447636 438 if (hfsmp->hfs_qfiles[i].qf_vp == NULLVP)
9bccf70c
A
439 continue;
440 if (cp->c_dquot[i] == NODQUOT) {
441 vprint("chkdquot: missing dquot", vp);
442 panic("missing dquot");
443 }
444 }
445}
446#endif
447
448/*
449 * Code to process quotactl commands.
450 */
451
452/*
453 * Q_QUOTAON - set up a quota file for a particular file system.
454 */
91447636
A
455struct hfs_quotaon_cargs {
456 int error;
457};
458
459static int
460hfs_quotaon_callback(struct vnode *vp, void *cargs)
461{
462 struct hfs_quotaon_cargs *args;
463
464 args = (struct hfs_quotaon_cargs *)cargs;
465
466 args->error = hfs_getinoquota(VTOC(vp));
467 if (args->error)
468 return (VNODE_RETURNED_DONE);
469
470 return (VNODE_RETURNED);
471}
472
9bccf70c 473int
91447636 474hfs_quotaon(p, mp, type, fnamep)
9bccf70c
A
475 struct proc *p;
476 struct mount *mp;
477 register int type;
91447636 478 caddr_t fnamep;
9bccf70c
A
479{
480 struct hfsmount *hfsmp = VFSTOHFS(mp);
91447636
A
481 struct quotafile *qfp;
482 struct vnode *vp;
483 int error = 0;
484 struct hfs_quotaon_cargs args;
9bccf70c 485
2d21ac55
A
486 /* Finish setting up quota structures. */
487 dqhashinit();
488
91447636
A
489 qfp = &hfsmp->hfs_qfiles[type];
490
491 if ( (qf_get(qfp, QTF_OPENING)) )
492 return (0);
493
494 error = vnode_open(fnamep, FREAD|FWRITE, 0, 0, &vp, NULL);
495 if (error) {
496 goto out;
9bccf70c 497 }
91447636
A
498 if (!vnode_isreg(vp)) {
499 (void) vnode_close(vp, FREAD|FWRITE, NULL);
500 error = EACCES;
501 goto out;
502 }
2d21ac55
A
503 vfs_setflags(mp, (u_int64_t)((unsigned int)MNT_QUOTA));
504 HFS_MOUNT_LOCK(hfsmp, TRUE)
505 hfsmp->hfs_flags |= HFS_QUOTAS;
506 HFS_MOUNT_UNLOCK(hfsmp, TRUE);
91447636 507 vnode_setnoflush(vp);
9bccf70c
A
508 /*
509 * Save the credential of the process that turned on quotas.
510 */
91447636
A
511 qfp->qf_cred = kauth_cred_proc_ref(p);
512 qfp->qf_vp = vp;
513 /*
514 * Finish initializing the quota file
515 */
516 error = dqfileopen(qfp, type);
517 if (error) {
518 (void) vnode_close(vp, FREAD|FWRITE, NULL);
519
0c530ab8
A
520 if (IS_VALID_CRED(qfp->qf_cred))
521 kauth_cred_unref(&qfp->qf_cred);
91447636
A
522 qfp->qf_vp = NULLVP;
523 goto out;
524 }
525 qf_put(qfp, QTF_OPENING);
526
9bccf70c
A
527 /*
528 * Search vnodes associated with this mount point,
529 * adding references to quota file being opened.
530 * NB: only need to add dquot's for cnodes being modified.
91447636
A
531 *
532 * hfs_quota_callback will be called for each vnode open for
533 * 'write' (VNODE_WRITEABLE) hung off of this mount point
534 * the vnode will be in an 'unbusy' state (VNODE_WAIT) and
535 * properly referenced and unreferenced around the callback
9bccf70c 536 */
91447636
A
537 args.error = 0;
538
539 vnode_iterate(mp, VNODE_WRITEABLE | VNODE_WAIT, hfs_quotaon_callback, (void *)&args);
540
541 error = args.error;
542
543 if (error) {
9bccf70c 544 hfs_quotaoff(p, mp, type);
91447636
A
545 }
546 return (error);
547
548out:
549 qf_put(qfp, QTF_OPENING);
550
9bccf70c
A
551 return (error);
552}
553
91447636 554
9bccf70c
A
555/*
556 * Q_QUOTAOFF - turn off disk quotas for a filesystem.
557 */
91447636
A
558struct hfs_quotaoff_cargs {
559 int type;
560};
561
562static int
563hfs_quotaoff_callback(struct vnode *vp, void *cargs)
564{
565 struct hfs_quotaoff_cargs *args;
566 struct cnode *cp;
567 struct dquot *dq;
568
569 args = (struct hfs_quotaoff_cargs *)cargs;
570
571 cp = VTOC(vp);
572
573 dq = cp->c_dquot[args->type];
574 cp->c_dquot[args->type] = NODQUOT;
575
576 dqrele(dq);
577
578 return (VNODE_RETURNED);
579}
580
9bccf70c 581int
91447636 582hfs_quotaoff(__unused struct proc *p, struct mount *mp, register int type)
9bccf70c 583{
91447636 584 struct vnode *qvp;
9bccf70c 585 struct hfsmount *hfsmp = VFSTOHFS(mp);
91447636 586 struct quotafile *qfp;
9bccf70c 587 int error;
91447636
A
588 struct hfs_quotaoff_cargs args;
589
2d21ac55
A
590 /*
591 * If quotas haven't been initialized, there's no work to be done.
592 */
593 if (!dqisinitialized())
594 return (0);
595
91447636 596 qfp = &hfsmp->hfs_qfiles[type];
9bccf70c 597
91447636
A
598 if ( (qf_get(qfp, QTF_CLOSING)) )
599 return (0);
600 qvp = qfp->qf_vp;
d7e50217
A
601
602 /*
603 * Sync out any orpaned dirty dquot entries.
604 */
91447636 605 dqsync_orphans(qfp);
d7e50217 606
9bccf70c
A
607 /*
608 * Search vnodes associated with this mount point,
609 * deleting any references to quota file being closed.
91447636
A
610 *
611 * hfs_quotaoff_callback will be called for each vnode
612 * hung off of this mount point
613 * the vnode will be in an 'unbusy' state (VNODE_WAIT) and
614 * properly referenced and unreferenced around the callback
9bccf70c 615 */
91447636
A
616 args.type = type;
617
618 vnode_iterate(mp, VNODE_WAIT, hfs_quotaoff_callback, (void *)&args);
619
9bccf70c
A
620 dqflush(qvp);
621 /* Finish tearing down the quota file */
91447636
A
622 dqfileclose(qfp, type);
623
624 vnode_clearnoflush(qvp);
625 error = vnode_close(qvp, FREAD|FWRITE, NULL);
626
627 qfp->qf_vp = NULLVP;
0c530ab8
A
628
629 if (IS_VALID_CRED(qfp->qf_cred))
630 kauth_cred_unref(&qfp->qf_cred);
9bccf70c
A
631 for (type = 0; type < MAXQUOTAS; type++)
632 if (hfsmp->hfs_qfiles[type].qf_vp != NULLVP)
633 break;
2d21ac55
A
634 if (type == MAXQUOTAS) {
635 vfs_clearflags(mp, (u_int64_t)((unsigned int)MNT_QUOTA));
636 HFS_MOUNT_LOCK(hfsmp, TRUE)
637 hfsmp->hfs_flags &= ~HFS_QUOTAS;
638 HFS_MOUNT_UNLOCK(hfsmp, TRUE);
639 }
91447636
A
640
641 qf_put(qfp, QTF_CLOSING);
642
9bccf70c
A
643 return (error);
644}
645
646/*
647 * Q_GETQUOTA - return current values in a dqblk structure.
648 */
649int
91447636 650hfs_getquota(mp, id, type, datap)
9bccf70c
A
651 struct mount *mp;
652 u_long id;
653 int type;
91447636 654 caddr_t datap;
9bccf70c
A
655{
656 struct dquot *dq;
657 int error;
658
91447636
A
659 error = dqget(id, &VFSTOHFS(mp)->hfs_qfiles[type], type, &dq);
660 if (error)
9bccf70c 661 return (error);
91447636
A
662 dqlock(dq);
663
664 bcopy(&dq->dq_dqb, datap, sizeof(dq->dq_dqb));
665
666 dqunlock(dq);
667 dqrele(dq);
668
9bccf70c
A
669 return (error);
670}
671
672/*
673 * Q_SETQUOTA - assign an entire dqblk structure.
674 */
675int
91447636 676hfs_setquota(mp, id, type, datap)
9bccf70c
A
677 struct mount *mp;
678 u_long id;
679 int type;
91447636 680 caddr_t datap;
9bccf70c 681{
91447636 682 struct dquot *dq;
9bccf70c 683 struct hfsmount *hfsmp = VFSTOHFS(mp);
91447636
A
684 struct dqblk * newlimp = (struct dqblk *) datap;
685 struct timeval tv;
9bccf70c
A
686 int error;
687
91447636
A
688 error = dqget(id, &hfsmp->hfs_qfiles[type], type, &dq);
689 if (error)
9bccf70c 690 return (error);
91447636
A
691 dqlock(dq);
692
9bccf70c
A
693 /*
694 * Copy all but the current values.
695 * Reset time limit if previously had no soft limit or were
696 * under it, but now have a soft limit and are over it.
697 */
91447636
A
698 newlimp->dqb_curbytes = dq->dq_curbytes;
699 newlimp->dqb_curinodes = dq->dq_curinodes;
9bccf70c 700 if (dq->dq_id != 0) {
91447636
A
701 newlimp->dqb_btime = dq->dq_btime;
702 newlimp->dqb_itime = dq->dq_itime;
9bccf70c 703 }
91447636
A
704 if (newlimp->dqb_bsoftlimit &&
705 dq->dq_curbytes >= newlimp->dqb_bsoftlimit &&
706 (dq->dq_bsoftlimit == 0 || dq->dq_curbytes < dq->dq_bsoftlimit)) {
707 microuptime(&tv);
708 newlimp->dqb_btime = tv.tv_sec + hfsmp->hfs_qfiles[type].qf_btime;
709 }
710 if (newlimp->dqb_isoftlimit &&
711 dq->dq_curinodes >= newlimp->dqb_isoftlimit &&
712 (dq->dq_isoftlimit == 0 || dq->dq_curinodes < dq->dq_isoftlimit)) {
713 microuptime(&tv);
714 newlimp->dqb_itime = tv.tv_sec + hfsmp->hfs_qfiles[type].qf_itime;
715 }
716 bcopy(newlimp, &dq->dq_dqb, sizeof(dq->dq_dqb));
9bccf70c
A
717 if (dq->dq_curbytes < dq->dq_bsoftlimit)
718 dq->dq_flags &= ~DQ_BLKS;
719 if (dq->dq_curinodes < dq->dq_isoftlimit)
720 dq->dq_flags &= ~DQ_INODS;
721 if (dq->dq_isoftlimit == 0 && dq->dq_bsoftlimit == 0 &&
722 dq->dq_ihardlimit == 0 && dq->dq_bhardlimit == 0)
723 dq->dq_flags |= DQ_FAKE;
724 else
725 dq->dq_flags &= ~DQ_FAKE;
726 dq->dq_flags |= DQ_MOD;
91447636
A
727
728 dqunlock(dq);
729 dqrele(dq);
730
9bccf70c
A
731 return (0);
732}
733
734/*
735 * Q_SETUSE - set current cnode and byte usage.
736 */
737int
91447636 738hfs_setuse(mp, id, type, datap)
9bccf70c
A
739 struct mount *mp;
740 u_long id;
741 int type;
91447636 742 caddr_t datap;
9bccf70c 743{
9bccf70c 744 struct hfsmount *hfsmp = VFSTOHFS(mp);
91447636
A
745 struct dquot *dq;
746 struct timeval tv;
9bccf70c 747 int error;
91447636
A
748 struct dqblk *quotablkp = (struct dqblk *) datap;
749
750 error = dqget(id, &hfsmp->hfs_qfiles[type], type, &dq);
751 if (error)
752 return (error);
753 dqlock(dq);
9bccf70c 754
9bccf70c
A
755 /*
756 * Reset time limit if have a soft limit and were
757 * previously under it, but are now over it.
758 */
759 if (dq->dq_bsoftlimit && dq->dq_curbytes < dq->dq_bsoftlimit &&
91447636
A
760 quotablkp->dqb_curbytes >= dq->dq_bsoftlimit) {
761 microuptime(&tv);
762 dq->dq_btime = tv.tv_sec + hfsmp->hfs_qfiles[type].qf_btime;
763 }
9bccf70c 764 if (dq->dq_isoftlimit && dq->dq_curinodes < dq->dq_isoftlimit &&
91447636
A
765 quotablkp->dqb_curinodes >= dq->dq_isoftlimit) {
766 microuptime(&tv);
767 dq->dq_itime = tv.tv_sec + hfsmp->hfs_qfiles[type].qf_itime;
768 }
769 dq->dq_curbytes = quotablkp->dqb_curbytes;
770 dq->dq_curinodes = quotablkp->dqb_curinodes;
9bccf70c
A
771 if (dq->dq_curbytes < dq->dq_bsoftlimit)
772 dq->dq_flags &= ~DQ_BLKS;
773 if (dq->dq_curinodes < dq->dq_isoftlimit)
774 dq->dq_flags &= ~DQ_INODS;
775 dq->dq_flags |= DQ_MOD;
91447636
A
776
777 dqunlock(dq);
778 dqrele(dq);
779
9bccf70c
A
780 return (0);
781}
782
91447636 783
9bccf70c
A
784/*
785 * Q_SYNC - sync quota files to disk.
786 */
91447636
A
787static int
788hfs_qsync_callback(struct vnode *vp, __unused void *cargs)
789{
790 struct cnode *cp;
791 struct dquot *dq;
792 int i;
793
794 cp = VTOC(vp);
795
796 for (i = 0; i < MAXQUOTAS; i++) {
797 dq = cp->c_dquot[i];
798 if (dq != NODQUOT && (dq->dq_flags & DQ_MOD))
799 dqsync(dq);
800 }
801 return (VNODE_RETURNED);
802}
803
9bccf70c
A
804int
805hfs_qsync(mp)
806 struct mount *mp;
807{
808 struct hfsmount *hfsmp = VFSTOHFS(mp);
91447636 809 int i;
9bccf70c 810
2d21ac55
A
811 if (!dqisinitialized())
812 return (0);
813
9bccf70c
A
814 /*
815 * Check if the mount point has any quotas.
816 * If not, simply return.
817 */
818 for (i = 0; i < MAXQUOTAS; i++)
819 if (hfsmp->hfs_qfiles[i].qf_vp != NULLVP)
820 break;
821 if (i == MAXQUOTAS)
822 return (0);
d7e50217
A
823
824 /*
825 * Sync out any orpaned dirty dquot entries.
826 */
827 for (i = 0; i < MAXQUOTAS; i++)
828 if (hfsmp->hfs_qfiles[i].qf_vp != NULLVP)
829 dqsync_orphans(&hfsmp->hfs_qfiles[i]);
830
9bccf70c
A
831 /*
832 * Search vnodes associated with this mount point,
833 * synchronizing any modified dquot structures.
91447636
A
834 *
835 * hfs_qsync_callback will be called for each vnode
836 * hung off of this mount point
837 * the vnode will be
838 * properly referenced and unreferenced around the callback
9bccf70c 839 */
91447636 840 vnode_iterate(mp, 0, hfs_qsync_callback, (void *)NULL);
d7e50217 841
9bccf70c
A
842 return (0);
843}
844
845/*
846 * Q_QUOTASTAT - get quota on/off status
847 */
848int
91447636 849hfs_quotastat(mp, type, datap)
9bccf70c
A
850 struct mount *mp;
851 register int type;
91447636 852 caddr_t datap;
9bccf70c
A
853{
854 struct hfsmount *hfsmp = VFSTOHFS(mp);
855 int error = 0;
856 int qstat;
857
91447636 858 if ((((unsigned int)vfs_flags(mp)) & MNT_QUOTA) && (hfsmp->hfs_qfiles[type].qf_vp != NULLVP))
9bccf70c
A
859 qstat = 1; /* quotas are on for this type */
860 else
861 qstat = 0; /* quotas are off for this type */
862
91447636 863 *((int *)datap) = qstat;
9bccf70c
A
864 return (error);
865}
866