ufs_quota.c revision 1.73 1 1.73 dholland /* $NetBSD: ufs_quota.c,v 1.73 2012/01/29 06:36:51 dholland Exp $ */
2 1.2 cgd
3 1.1 mycroft /*
4 1.11 fvdl * Copyright (c) 1982, 1986, 1990, 1993, 1995
5 1.1 mycroft * The Regents of the University of California. All rights reserved.
6 1.1 mycroft *
7 1.1 mycroft * This code is derived from software contributed to Berkeley by
8 1.1 mycroft * Robert Elz at The University of Melbourne.
9 1.1 mycroft *
10 1.1 mycroft * Redistribution and use in source and binary forms, with or without
11 1.1 mycroft * modification, are permitted provided that the following conditions
12 1.1 mycroft * are met:
13 1.1 mycroft * 1. Redistributions of source code must retain the above copyright
14 1.1 mycroft * notice, this list of conditions and the following disclaimer.
15 1.1 mycroft * 2. Redistributions in binary form must reproduce the above copyright
16 1.1 mycroft * notice, this list of conditions and the following disclaimer in the
17 1.1 mycroft * documentation and/or other materials provided with the distribution.
18 1.28 agc * 3. Neither the name of the University nor the names of its contributors
19 1.1 mycroft * may be used to endorse or promote products derived from this software
20 1.1 mycroft * without specific prior written permission.
21 1.1 mycroft *
22 1.1 mycroft * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
23 1.1 mycroft * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
24 1.1 mycroft * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
25 1.1 mycroft * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
26 1.1 mycroft * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
27 1.1 mycroft * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
28 1.1 mycroft * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
29 1.1 mycroft * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
30 1.1 mycroft * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
31 1.1 mycroft * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
32 1.1 mycroft * SUCH DAMAGE.
33 1.1 mycroft *
34 1.11 fvdl * @(#)ufs_quota.c 8.5 (Berkeley) 5/20/95
35 1.1 mycroft */
36 1.22 lukem
37 1.22 lukem #include <sys/cdefs.h>
38 1.73 dholland __KERNEL_RCSID(0, "$NetBSD: ufs_quota.c,v 1.73 2012/01/29 06:36:51 dholland Exp $");
39 1.22 lukem
40 1.69 bouyer #if defined(_KERNEL_OPT)
41 1.69 bouyer #include "opt_quota.h"
42 1.69 bouyer #endif
43 1.1 mycroft #include <sys/param.h>
44 1.1 mycroft #include <sys/kernel.h>
45 1.1 mycroft #include <sys/systm.h>
46 1.1 mycroft #include <sys/namei.h>
47 1.1 mycroft #include <sys/file.h>
48 1.1 mycroft #include <sys/proc.h>
49 1.1 mycroft #include <sys/vnode.h>
50 1.1 mycroft #include <sys/mount.h>
51 1.39 elad #include <sys/kauth.h>
52 1.1 mycroft
53 1.71 dholland #include <sys/quotactl.h>
54 1.1 mycroft #include <ufs/ufs/quota.h>
55 1.1 mycroft #include <ufs/ufs/inode.h>
56 1.1 mycroft #include <ufs/ufs/ufsmount.h>
57 1.1 mycroft #include <ufs/ufs/ufs_extern.h>
58 1.69 bouyer #include <ufs/ufs/ufs_quota.h>
59 1.70 bouyer #include <quota/quotaprop.h>
60 1.1 mycroft
61 1.69 bouyer kmutex_t dqlock;
62 1.69 bouyer kcondvar_t dqcv;
63 1.48 hannken
64 1.48 hannken /*
65 1.69 bouyer * Code pertaining to management of the in-core dquot data structures.
66 1.1 mycroft */
67 1.69 bouyer #define DQHASH(dqvp, id) \
68 1.69 bouyer (((((long)(dqvp)) >> 8) + id) & dqhash)
69 1.69 bouyer static LIST_HEAD(dqhashhead, dquot) *dqhashtbl;
70 1.69 bouyer static u_long dqhash;
71 1.69 bouyer static pool_cache_t dquot_cache;
72 1.1 mycroft
73 1.1 mycroft
74 1.69 bouyer static int quota_handle_cmd_get_version(struct mount *, struct lwp *,
75 1.72 dholland struct vfs_quotactl_args *args);
76 1.69 bouyer static int quota_handle_cmd_get(struct mount *, struct lwp *,
77 1.72 dholland struct vfs_quotactl_args *args);
78 1.69 bouyer static int quota_handle_cmd_set(struct mount *, struct lwp *,
79 1.72 dholland struct vfs_quotactl_args *args);
80 1.69 bouyer static int quota_handle_cmd_getall(struct mount *, struct lwp *,
81 1.72 dholland struct vfs_quotactl_args *args);
82 1.69 bouyer static int quota_handle_cmd_clear(struct mount *, struct lwp *,
83 1.72 dholland struct vfs_quotactl_args *args);
84 1.69 bouyer static int quota_handle_cmd_quotaon(struct mount *, struct lwp *,
85 1.72 dholland struct vfs_quotactl_args *args);
86 1.69 bouyer static int quota_handle_cmd_quotaoff(struct mount *, struct lwp *,
87 1.72 dholland struct vfs_quotactl_args *args);
88 1.72 dholland
89 1.1 mycroft /*
90 1.48 hannken * Initialize the quota fields of an inode.
91 1.48 hannken */
92 1.48 hannken void
93 1.48 hannken ufsquota_init(struct inode *ip)
94 1.48 hannken {
95 1.48 hannken int i;
96 1.48 hannken
97 1.48 hannken for (i = 0; i < MAXQUOTAS; i++)
98 1.48 hannken ip->i_dquot[i] = NODQUOT;
99 1.48 hannken }
100 1.48 hannken
101 1.48 hannken /*
102 1.48 hannken * Release the quota fields from an inode.
103 1.48 hannken */
104 1.48 hannken void
105 1.48 hannken ufsquota_free(struct inode *ip)
106 1.48 hannken {
107 1.48 hannken int i;
108 1.48 hannken
109 1.48 hannken for (i = 0; i < MAXQUOTAS; i++) {
110 1.48 hannken dqrele(ITOV(ip), ip->i_dquot[i]);
111 1.48 hannken ip->i_dquot[i] = NODQUOT;
112 1.48 hannken }
113 1.48 hannken }
114 1.48 hannken
115 1.48 hannken /*
116 1.1 mycroft * Update disk usage, and take corrective action.
117 1.1 mycroft */
118 1.1 mycroft int
119 1.39 elad chkdq(struct inode *ip, int64_t change, kauth_cred_t cred, int flags)
120 1.1 mycroft {
121 1.69 bouyer /* do not track snapshot usage, or we will deadlock */
122 1.69 bouyer if ((ip->i_flags & SF_SNAPSHOT) != 0)
123 1.69 bouyer return 0;
124 1.1 mycroft
125 1.69 bouyer #ifdef QUOTA
126 1.69 bouyer if (ip->i_ump->um_flags & UFS_QUOTA)
127 1.69 bouyer return chkdq1(ip, change, cred, flags);
128 1.69 bouyer #endif
129 1.69 bouyer #ifdef QUOTA2
130 1.69 bouyer if (ip->i_ump->um_flags & UFS_QUOTA2)
131 1.69 bouyer return chkdq2(ip, change, cred, flags);
132 1.69 bouyer #endif
133 1.69 bouyer return 0;
134 1.1 mycroft }
135 1.1 mycroft
136 1.1 mycroft /*
137 1.69 bouyer * Check the inode limit, applying corrective action.
138 1.1 mycroft */
139 1.69 bouyer int
140 1.69 bouyer chkiq(struct inode *ip, int32_t change, kauth_cred_t cred, int flags)
141 1.1 mycroft {
142 1.69 bouyer /* do not track snapshot usage, or we will deadlock */
143 1.69 bouyer if ((ip->i_flags & SF_SNAPSHOT) != 0)
144 1.69 bouyer return 0;
145 1.69 bouyer #ifdef QUOTA
146 1.69 bouyer if (ip->i_ump->um_flags & UFS_QUOTA)
147 1.69 bouyer return chkiq1(ip, change, cred, flags);
148 1.69 bouyer #endif
149 1.69 bouyer #ifdef QUOTA2
150 1.69 bouyer if (ip->i_ump->um_flags & UFS_QUOTA2)
151 1.69 bouyer return chkiq2(ip, change, cred, flags);
152 1.69 bouyer #endif
153 1.69 bouyer return 0;
154 1.1 mycroft }
155 1.1 mycroft
156 1.1 mycroft int
157 1.71 dholland quota_handle_cmd(struct mount *mp, struct lwp *l, int op,
158 1.72 dholland struct vfs_quotactl_args *args)
159 1.69 bouyer {
160 1.69 bouyer int error = 0;
161 1.69 bouyer
162 1.71 dholland switch (op) {
163 1.71 dholland case QUOTACTL_GETVERSION:
164 1.72 dholland error = quota_handle_cmd_get_version(mp, l, args);
165 1.71 dholland break;
166 1.71 dholland case QUOTACTL_QUOTAON:
167 1.72 dholland error = quota_handle_cmd_quotaon(mp, l, args);
168 1.71 dholland break;
169 1.71 dholland case QUOTACTL_QUOTAOFF:
170 1.72 dholland error = quota_handle_cmd_quotaoff(mp, l, args);
171 1.71 dholland break;
172 1.71 dholland case QUOTACTL_GET:
173 1.72 dholland error = quota_handle_cmd_get(mp, l, args);
174 1.71 dholland break;
175 1.71 dholland case QUOTACTL_SET:
176 1.72 dholland error = quota_handle_cmd_set(mp, l, args);
177 1.71 dholland break;
178 1.71 dholland case QUOTACTL_GETALL:
179 1.72 dholland error = quota_handle_cmd_getall(mp, l, args);
180 1.71 dholland break;
181 1.71 dholland case QUOTACTL_CLEAR:
182 1.72 dholland error = quota_handle_cmd_clear(mp, l, args);
183 1.71 dholland break;
184 1.71 dholland default:
185 1.71 dholland panic("Invalid quotactl operation %d\n", op);
186 1.69 bouyer }
187 1.71 dholland
188 1.69 bouyer return error;
189 1.69 bouyer }
190 1.69 bouyer
191 1.69 bouyer static int
192 1.69 bouyer quota_handle_cmd_get_version(struct mount *mp, struct lwp *l,
193 1.72 dholland struct vfs_quotactl_args *args)
194 1.1 mycroft {
195 1.69 bouyer struct ufsmount *ump = VFSTOUFS(mp);
196 1.73 dholland int *version_ret;
197 1.72 dholland
198 1.73 dholland KASSERT(args->qc_type == QCT_GETVERSION);
199 1.73 dholland version_ret = args->u.getversion.qc_version_ret;
200 1.69 bouyer
201 1.69 bouyer if ((ump->um_flags & (UFS_QUOTA|UFS_QUOTA2)) == 0)
202 1.69 bouyer return EOPNOTSUPP;
203 1.69 bouyer
204 1.69 bouyer #ifdef QUOTA
205 1.69 bouyer if (ump->um_flags & UFS_QUOTA) {
206 1.73 dholland *version_ret = 1;
207 1.69 bouyer } else
208 1.69 bouyer #endif
209 1.69 bouyer #ifdef QUOTA2
210 1.69 bouyer if (ump->um_flags & UFS_QUOTA2) {
211 1.73 dholland *version_ret = 2;
212 1.69 bouyer } else
213 1.69 bouyer #endif
214 1.73 dholland return EOPNOTSUPP;
215 1.73 dholland
216 1.73 dholland return 0;
217 1.1 mycroft }
218 1.1 mycroft
219 1.69 bouyer /* XXX shouldn't all this be in kauth ? */
220 1.48 hannken static int
221 1.69 bouyer quota_get_auth(struct mount *mp, struct lwp *l, uid_t id) {
222 1.69 bouyer /* The user can always query about his own quota. */
223 1.69 bouyer if (id == kauth_cred_getuid(l->l_cred))
224 1.69 bouyer return 0;
225 1.69 bouyer return kauth_authorize_system(l->l_cred, KAUTH_SYSTEM_FS_QUOTA,
226 1.69 bouyer KAUTH_REQ_SYSTEM_FS_QUOTA_GET, mp, KAUTH_ARG(id), NULL);
227 1.69 bouyer }
228 1.69 bouyer
229 1.69 bouyer static int
230 1.69 bouyer quota_handle_cmd_get(struct mount *mp, struct lwp *l,
231 1.72 dholland struct vfs_quotactl_args *args)
232 1.69 bouyer {
233 1.69 bouyer prop_array_t replies;
234 1.69 bouyer prop_object_iterator_t iter;
235 1.69 bouyer prop_dictionary_t data;
236 1.69 bouyer uint32_t id;
237 1.69 bouyer struct ufsmount *ump = VFSTOUFS(mp);
238 1.69 bouyer int error, defaultq = 0;
239 1.69 bouyer const char *idstr;
240 1.72 dholland prop_dictionary_t cmddict;
241 1.72 dholland int q2type;
242 1.72 dholland prop_array_t datas;
243 1.72 dholland
244 1.72 dholland KASSERT(args->qc_type == QCT_PROPLIB);
245 1.72 dholland cmddict = args->u.proplib.qc_cmddict;
246 1.72 dholland q2type = args->u.proplib.qc_q2type;
247 1.72 dholland datas = args->u.proplib.qc_datas;
248 1.72 dholland
249 1.72 dholland KASSERT(prop_object_type(cmddict) == PROP_TYPE_DICTIONARY);
250 1.72 dholland KASSERT(prop_object_type(datas) == PROP_TYPE_ARRAY);
251 1.69 bouyer
252 1.69 bouyer if ((ump->um_flags & (UFS_QUOTA|UFS_QUOTA2)) == 0)
253 1.69 bouyer return EOPNOTSUPP;
254 1.69 bouyer
255 1.69 bouyer replies = prop_array_create();
256 1.69 bouyer if (replies == NULL)
257 1.69 bouyer return ENOMEM;
258 1.1 mycroft
259 1.69 bouyer iter = prop_array_iterator(datas);
260 1.69 bouyer if (iter == NULL) {
261 1.69 bouyer prop_object_release(replies);
262 1.69 bouyer return ENOMEM;
263 1.1 mycroft }
264 1.69 bouyer while ((data = prop_object_iterator_next(iter)) != NULL) {
265 1.69 bouyer if (!prop_dictionary_get_uint32(data, "id", &id)) {
266 1.69 bouyer if (!prop_dictionary_get_cstring_nocopy(data, "id",
267 1.69 bouyer &idstr))
268 1.69 bouyer continue;
269 1.69 bouyer if (strcmp(idstr, "default")) {
270 1.69 bouyer error = EINVAL;
271 1.69 bouyer goto err;
272 1.1 mycroft }
273 1.69 bouyer id = 0;
274 1.69 bouyer defaultq = 1;
275 1.69 bouyer } else {
276 1.69 bouyer defaultq = 0;
277 1.1 mycroft }
278 1.69 bouyer error = quota_get_auth(mp, l, id);
279 1.69 bouyer if (error == EPERM)
280 1.69 bouyer continue;
281 1.69 bouyer if (error != 0)
282 1.69 bouyer goto err;
283 1.69 bouyer #ifdef QUOTA
284 1.69 bouyer if (ump->um_flags & UFS_QUOTA)
285 1.72 dholland error = quota1_handle_cmd_get(ump, q2type, id, defaultq,
286 1.69 bouyer replies);
287 1.69 bouyer else
288 1.69 bouyer #endif
289 1.69 bouyer #ifdef QUOTA2
290 1.69 bouyer if (ump->um_flags & UFS_QUOTA2) {
291 1.72 dholland error = quota2_handle_cmd_get(ump, q2type, id, defaultq,
292 1.69 bouyer replies);
293 1.69 bouyer } else
294 1.69 bouyer #endif
295 1.69 bouyer panic("quota_handle_cmd_get: no support ?");
296 1.69 bouyer
297 1.69 bouyer if (error == ENOENT)
298 1.69 bouyer continue;
299 1.69 bouyer if (error != 0)
300 1.69 bouyer goto err;
301 1.69 bouyer }
302 1.69 bouyer prop_object_iterator_release(iter);
303 1.69 bouyer if (!prop_dictionary_set_and_rel(cmddict, "data", replies)) {
304 1.69 bouyer error = ENOMEM;
305 1.69 bouyer } else {
306 1.69 bouyer error = 0;
307 1.1 mycroft }
308 1.69 bouyer return error;
309 1.69 bouyer err:
310 1.69 bouyer prop_object_iterator_release(iter);
311 1.69 bouyer prop_object_release(replies);
312 1.69 bouyer return error;
313 1.69 bouyer }
314 1.69 bouyer
315 1.69 bouyer static int
316 1.69 bouyer quota_handle_cmd_set(struct mount *mp, struct lwp *l,
317 1.72 dholland struct vfs_quotactl_args *args)
318 1.69 bouyer {
319 1.69 bouyer prop_array_t replies;
320 1.69 bouyer prop_object_iterator_t iter;
321 1.69 bouyer prop_dictionary_t data;
322 1.69 bouyer uint32_t id;
323 1.11 fvdl struct ufsmount *ump = VFSTOUFS(mp);
324 1.69 bouyer int error, defaultq = 0;
325 1.69 bouyer const char *idstr;
326 1.72 dholland prop_dictionary_t cmddict;
327 1.72 dholland int q2type;
328 1.72 dholland prop_array_t datas;
329 1.72 dholland
330 1.72 dholland KASSERT(args->qc_type == QCT_PROPLIB);
331 1.72 dholland cmddict = args->u.proplib.qc_cmddict;
332 1.72 dholland q2type = args->u.proplib.qc_q2type;
333 1.72 dholland datas = args->u.proplib.qc_datas;
334 1.72 dholland
335 1.72 dholland KASSERT(prop_object_type(cmddict) == PROP_TYPE_DICTIONARY);
336 1.72 dholland KASSERT(prop_object_type(datas) == PROP_TYPE_ARRAY);
337 1.1 mycroft
338 1.69 bouyer if ((ump->um_flags & (UFS_QUOTA|UFS_QUOTA2)) == 0)
339 1.69 bouyer return EOPNOTSUPP;
340 1.69 bouyer
341 1.69 bouyer replies = prop_array_create();
342 1.69 bouyer if (replies == NULL)
343 1.69 bouyer return ENOMEM;
344 1.61 ad
345 1.69 bouyer iter = prop_array_iterator(datas);
346 1.69 bouyer if (iter == NULL) {
347 1.69 bouyer prop_object_release(replies);
348 1.69 bouyer return ENOMEM;
349 1.68 dholland }
350 1.69 bouyer while ((data = prop_object_iterator_next(iter)) != NULL) {
351 1.69 bouyer if (!prop_dictionary_get_uint32(data, "id", &id)) {
352 1.69 bouyer if (!prop_dictionary_get_cstring_nocopy(data, "id",
353 1.69 bouyer &idstr))
354 1.69 bouyer continue;
355 1.69 bouyer if (strcmp(idstr, "default"))
356 1.69 bouyer continue;
357 1.69 bouyer id = 0;
358 1.69 bouyer defaultq = 1;
359 1.69 bouyer } else {
360 1.69 bouyer defaultq = 0;
361 1.69 bouyer }
362 1.69 bouyer error = kauth_authorize_system(l->l_cred, KAUTH_SYSTEM_FS_QUOTA,
363 1.69 bouyer KAUTH_REQ_SYSTEM_FS_QUOTA_MANAGE, mp, KAUTH_ARG(id), NULL);
364 1.69 bouyer if (error != 0)
365 1.69 bouyer goto err;
366 1.69 bouyer #ifdef QUOTA
367 1.69 bouyer if (ump->um_flags & UFS_QUOTA)
368 1.72 dholland error = quota1_handle_cmd_set(ump, q2type, id, defaultq,
369 1.69 bouyer data);
370 1.69 bouyer else
371 1.69 bouyer #endif
372 1.69 bouyer #ifdef QUOTA2
373 1.69 bouyer if (ump->um_flags & UFS_QUOTA2) {
374 1.72 dholland error = quota2_handle_cmd_set(ump, q2type, id, defaultq,
375 1.69 bouyer data);
376 1.69 bouyer } else
377 1.69 bouyer #endif
378 1.69 bouyer panic("quota_handle_cmd_get: no support ?");
379 1.69 bouyer
380 1.69 bouyer if (error && error != ENOENT)
381 1.69 bouyer goto err;
382 1.1 mycroft }
383 1.69 bouyer prop_object_iterator_release(iter);
384 1.69 bouyer if (!prop_dictionary_set_and_rel(cmddict, "data", replies)) {
385 1.53 ad error = ENOMEM;
386 1.69 bouyer } else {
387 1.69 bouyer error = 0;
388 1.1 mycroft }
389 1.69 bouyer return error;
390 1.69 bouyer err:
391 1.69 bouyer prop_object_iterator_release(iter);
392 1.69 bouyer prop_object_release(replies);
393 1.69 bouyer return error;
394 1.69 bouyer }
395 1.69 bouyer
396 1.69 bouyer static int
397 1.69 bouyer quota_handle_cmd_clear(struct mount *mp, struct lwp *l,
398 1.72 dholland struct vfs_quotactl_args *args)
399 1.69 bouyer {
400 1.69 bouyer prop_array_t replies;
401 1.69 bouyer prop_object_iterator_t iter;
402 1.69 bouyer prop_dictionary_t data;
403 1.69 bouyer uint32_t id;
404 1.1 mycroft struct ufsmount *ump = VFSTOUFS(mp);
405 1.69 bouyer int error, defaultq = 0;
406 1.69 bouyer const char *idstr;
407 1.72 dholland prop_dictionary_t cmddict;
408 1.72 dholland int q2type;
409 1.72 dholland prop_array_t datas;
410 1.72 dholland
411 1.72 dholland KASSERT(args->qc_type == QCT_PROPLIB);
412 1.72 dholland cmddict = args->u.proplib.qc_cmddict;
413 1.72 dholland q2type = args->u.proplib.qc_q2type;
414 1.72 dholland datas = args->u.proplib.qc_datas;
415 1.72 dholland
416 1.72 dholland KASSERT(prop_object_type(cmddict) == PROP_TYPE_DICTIONARY);
417 1.72 dholland KASSERT(prop_object_type(datas) == PROP_TYPE_ARRAY);
418 1.33 perry
419 1.69 bouyer if ((ump->um_flags & UFS_QUOTA2) == 0)
420 1.69 bouyer return EOPNOTSUPP;
421 1.69 bouyer
422 1.69 bouyer replies = prop_array_create();
423 1.69 bouyer if (replies == NULL)
424 1.53 ad return ENOMEM;
425 1.53 ad
426 1.69 bouyer iter = prop_array_iterator(datas);
427 1.69 bouyer if (iter == NULL) {
428 1.69 bouyer prop_object_release(replies);
429 1.69 bouyer return ENOMEM;
430 1.69 bouyer }
431 1.69 bouyer while ((data = prop_object_iterator_next(iter)) != NULL) {
432 1.69 bouyer if (!prop_dictionary_get_uint32(data, "id", &id)) {
433 1.69 bouyer if (!prop_dictionary_get_cstring_nocopy(data, "id",
434 1.69 bouyer &idstr))
435 1.69 bouyer continue;
436 1.69 bouyer if (strcmp(idstr, "default"))
437 1.69 bouyer continue;
438 1.69 bouyer id = 0;
439 1.69 bouyer defaultq = 1;
440 1.69 bouyer } else {
441 1.69 bouyer defaultq = 0;
442 1.69 bouyer }
443 1.69 bouyer error = kauth_authorize_system(l->l_cred, KAUTH_SYSTEM_FS_QUOTA,
444 1.69 bouyer KAUTH_REQ_SYSTEM_FS_QUOTA_MANAGE, mp, KAUTH_ARG(id), NULL);
445 1.69 bouyer if (error != 0)
446 1.69 bouyer goto err;
447 1.69 bouyer #ifdef QUOTA2
448 1.69 bouyer if (ump->um_flags & UFS_QUOTA2) {
449 1.72 dholland error = quota2_handle_cmd_clear(ump, q2type, id, defaultq,
450 1.69 bouyer data);
451 1.69 bouyer } else
452 1.69 bouyer #endif
453 1.69 bouyer panic("quota_handle_cmd_get: no support ?");
454 1.69 bouyer
455 1.69 bouyer if (error && error != ENOENT)
456 1.69 bouyer goto err;
457 1.50 hannken }
458 1.69 bouyer prop_object_iterator_release(iter);
459 1.69 bouyer if (!prop_dictionary_set_and_rel(cmddict, "data", replies)) {
460 1.69 bouyer error = ENOMEM;
461 1.69 bouyer } else {
462 1.69 bouyer error = 0;
463 1.1 mycroft }
464 1.69 bouyer return error;
465 1.69 bouyer err:
466 1.69 bouyer prop_object_iterator_release(iter);
467 1.69 bouyer prop_object_release(replies);
468 1.69 bouyer return error;
469 1.1 mycroft }
470 1.1 mycroft
471 1.69 bouyer static int
472 1.69 bouyer quota_handle_cmd_getall(struct mount *mp, struct lwp *l,
473 1.72 dholland struct vfs_quotactl_args *args)
474 1.1 mycroft {
475 1.69 bouyer prop_array_t replies;
476 1.1 mycroft struct ufsmount *ump = VFSTOUFS(mp);
477 1.1 mycroft int error;
478 1.72 dholland prop_dictionary_t cmddict;
479 1.72 dholland int q2type;
480 1.72 dholland prop_array_t datas;
481 1.72 dholland
482 1.72 dholland KASSERT(args->qc_type == QCT_PROPLIB);
483 1.72 dholland cmddict = args->u.proplib.qc_cmddict;
484 1.72 dholland q2type = args->u.proplib.qc_q2type;
485 1.72 dholland datas = args->u.proplib.qc_datas;
486 1.72 dholland
487 1.72 dholland KASSERT(prop_object_type(cmddict) == PROP_TYPE_DICTIONARY);
488 1.72 dholland KASSERT(prop_object_type(datas) == PROP_TYPE_ARRAY);
489 1.1 mycroft
490 1.69 bouyer if ((ump->um_flags & UFS_QUOTA2) == 0)
491 1.69 bouyer return EOPNOTSUPP;
492 1.69 bouyer
493 1.69 bouyer error = kauth_authorize_system(l->l_cred, KAUTH_SYSTEM_FS_QUOTA,
494 1.69 bouyer KAUTH_REQ_SYSTEM_FS_QUOTA_GET, mp, NULL, NULL);
495 1.8 christos if (error)
496 1.69 bouyer return error;
497 1.69 bouyer
498 1.69 bouyer replies = prop_array_create();
499 1.69 bouyer if (replies == NULL)
500 1.69 bouyer return ENOMEM;
501 1.69 bouyer
502 1.69 bouyer #ifdef QUOTA2
503 1.69 bouyer if (ump->um_flags & UFS_QUOTA2) {
504 1.72 dholland error = quota2_handle_cmd_getall(ump, q2type, replies);
505 1.69 bouyer } else
506 1.69 bouyer #endif
507 1.69 bouyer panic("quota_handle_cmd_getall: no support ?");
508 1.69 bouyer if (!prop_dictionary_set_and_rel(cmddict, "data", replies)) {
509 1.69 bouyer error = ENOMEM;
510 1.69 bouyer } else {
511 1.69 bouyer error = 0;
512 1.69 bouyer }
513 1.69 bouyer return error;
514 1.1 mycroft }
515 1.1 mycroft
516 1.69 bouyer static int
517 1.69 bouyer quota_handle_cmd_quotaon(struct mount *mp, struct lwp *l,
518 1.72 dholland struct vfs_quotactl_args *args)
519 1.1 mycroft {
520 1.69 bouyer prop_dictionary_t data;
521 1.1 mycroft struct ufsmount *ump = VFSTOUFS(mp);
522 1.1 mycroft int error;
523 1.69 bouyer const char *qfile;
524 1.72 dholland prop_dictionary_t cmddict;
525 1.72 dholland int q2type;
526 1.72 dholland prop_array_t datas;
527 1.72 dholland
528 1.72 dholland KASSERT(args->qc_type == QCT_PROPLIB);
529 1.72 dholland cmddict = args->u.proplib.qc_cmddict;
530 1.72 dholland q2type = args->u.proplib.qc_q2type;
531 1.72 dholland datas = args->u.proplib.qc_datas;
532 1.72 dholland
533 1.72 dholland KASSERT(prop_object_type(cmddict) == PROP_TYPE_DICTIONARY);
534 1.72 dholland KASSERT(prop_object_type(datas) == PROP_TYPE_ARRAY);
535 1.69 bouyer
536 1.69 bouyer if ((ump->um_flags & UFS_QUOTA2) != 0)
537 1.69 bouyer return EBUSY;
538 1.69 bouyer
539 1.69 bouyer if (prop_array_count(datas) != 1)
540 1.69 bouyer return EINVAL;
541 1.1 mycroft
542 1.69 bouyer data = prop_array_get(datas, 0);
543 1.69 bouyer if (data == NULL)
544 1.69 bouyer return ENOMEM;
545 1.69 bouyer if (!prop_dictionary_get_cstring_nocopy(data, "quotafile",
546 1.69 bouyer &qfile))
547 1.69 bouyer return EINVAL;
548 1.69 bouyer
549 1.69 bouyer error = kauth_authorize_system(l->l_cred, KAUTH_SYSTEM_FS_QUOTA,
550 1.69 bouyer KAUTH_REQ_SYSTEM_FS_QUOTA_ONOFF, mp, NULL, NULL);
551 1.69 bouyer if (error != 0) {
552 1.69 bouyer return error;
553 1.69 bouyer }
554 1.69 bouyer #ifdef QUOTA
555 1.72 dholland error = quota1_handle_cmd_quotaon(l, ump, q2type, qfile);
556 1.69 bouyer #else
557 1.69 bouyer error = EOPNOTSUPP;
558 1.69 bouyer #endif
559 1.69 bouyer
560 1.69 bouyer return error;
561 1.1 mycroft }
562 1.1 mycroft
563 1.69 bouyer static int
564 1.69 bouyer quota_handle_cmd_quotaoff(struct mount *mp, struct lwp *l,
565 1.72 dholland struct vfs_quotactl_args *args)
566 1.1 mycroft {
567 1.1 mycroft struct ufsmount *ump = VFSTOUFS(mp);
568 1.69 bouyer int error;
569 1.72 dholland prop_dictionary_t cmddict;
570 1.72 dholland int q2type;
571 1.72 dholland prop_array_t datas;
572 1.72 dholland
573 1.72 dholland KASSERT(args->qc_type == QCT_PROPLIB);
574 1.72 dholland cmddict = args->u.proplib.qc_cmddict;
575 1.72 dholland q2type = args->u.proplib.qc_q2type;
576 1.72 dholland datas = args->u.proplib.qc_datas;
577 1.72 dholland
578 1.72 dholland KASSERT(prop_object_type(cmddict) == PROP_TYPE_DICTIONARY);
579 1.72 dholland KASSERT(prop_object_type(datas) == PROP_TYPE_ARRAY);
580 1.1 mycroft
581 1.69 bouyer if ((ump->um_flags & UFS_QUOTA2) != 0)
582 1.69 bouyer return EOPNOTSUPP;
583 1.69 bouyer
584 1.69 bouyer if (prop_array_count(datas) != 0)
585 1.69 bouyer return EINVAL;
586 1.69 bouyer
587 1.69 bouyer error = kauth_authorize_system(l->l_cred, KAUTH_SYSTEM_FS_QUOTA,
588 1.69 bouyer KAUTH_REQ_SYSTEM_FS_QUOTA_ONOFF, mp, NULL, NULL);
589 1.69 bouyer if (error != 0) {
590 1.69 bouyer return error;
591 1.1 mycroft }
592 1.69 bouyer #ifdef QUOTA
593 1.72 dholland error = quota1_handle_cmd_quotaoff(l, ump, q2type);
594 1.69 bouyer #else
595 1.69 bouyer error = EOPNOTSUPP;
596 1.69 bouyer #endif
597 1.69 bouyer
598 1.69 bouyer return error;
599 1.1 mycroft }
600 1.1 mycroft
601 1.1 mycroft /*
602 1.1 mycroft * Initialize the quota system.
603 1.1 mycroft */
604 1.1 mycroft void
605 1.35 thorpej dqinit(void)
606 1.1 mycroft {
607 1.47 pooka
608 1.50 hannken mutex_init(&dqlock, MUTEX_DEFAULT, IPL_NONE);
609 1.50 hannken cv_init(&dqcv, "quota");
610 1.60 ad dqhashtbl = hashinit(desiredvnodes, HASH_LIST, true, &dqhash);
611 1.55 ad dquot_cache = pool_cache_init(sizeof(struct dquot), 0, 0, 0, "ufsdq",
612 1.55 ad NULL, IPL_NONE, NULL, NULL, NULL);
613 1.14 jdolecek }
614 1.14 jdolecek
615 1.21 chs void
616 1.35 thorpej dqreinit(void)
617 1.21 chs {
618 1.21 chs struct dquot *dq;
619 1.21 chs struct dqhashhead *oldhash, *hash;
620 1.21 chs struct vnode *dqvp;
621 1.21 chs u_long oldmask, mask, hashval;
622 1.21 chs int i;
623 1.21 chs
624 1.60 ad hash = hashinit(desiredvnodes, HASH_LIST, true, &mask);
625 1.50 hannken mutex_enter(&dqlock);
626 1.21 chs oldhash = dqhashtbl;
627 1.21 chs oldmask = dqhash;
628 1.21 chs dqhashtbl = hash;
629 1.21 chs dqhash = mask;
630 1.21 chs for (i = 0; i <= oldmask; i++) {
631 1.21 chs while ((dq = LIST_FIRST(&oldhash[i])) != NULL) {
632 1.21 chs dqvp = dq->dq_ump->um_quotas[dq->dq_type];
633 1.21 chs LIST_REMOVE(dq, dq_hash);
634 1.21 chs hashval = DQHASH(dqvp, dq->dq_id);
635 1.21 chs LIST_INSERT_HEAD(&dqhashtbl[hashval], dq, dq_hash);
636 1.21 chs }
637 1.21 chs }
638 1.50 hannken mutex_exit(&dqlock);
639 1.60 ad hashdone(oldhash, HASH_LIST, oldmask);
640 1.21 chs }
641 1.21 chs
642 1.14 jdolecek /*
643 1.14 jdolecek * Free resources held by quota system.
644 1.14 jdolecek */
645 1.14 jdolecek void
646 1.35 thorpej dqdone(void)
647 1.14 jdolecek {
648 1.47 pooka
649 1.55 ad pool_cache_destroy(dquot_cache);
650 1.60 ad hashdone(dqhashtbl, HASH_LIST, dqhash);
651 1.50 hannken cv_destroy(&dqcv);
652 1.50 hannken mutex_destroy(&dqlock);
653 1.1 mycroft }
654 1.1 mycroft
655 1.1 mycroft /*
656 1.69 bouyer * Set up the quotas for an inode.
657 1.69 bouyer *
658 1.69 bouyer * This routine completely defines the semantics of quotas.
659 1.69 bouyer * If other criterion want to be used to establish quotas, the
660 1.69 bouyer * MAXQUOTAS value in quotas.h should be increased, and the
661 1.69 bouyer * additional dquots set up here.
662 1.69 bouyer */
663 1.69 bouyer int
664 1.69 bouyer getinoquota(struct inode *ip)
665 1.69 bouyer {
666 1.69 bouyer struct ufsmount *ump = ip->i_ump;
667 1.69 bouyer struct vnode *vp = ITOV(ip);
668 1.69 bouyer int i, error;
669 1.69 bouyer u_int32_t ino_ids[MAXQUOTAS];
670 1.69 bouyer
671 1.69 bouyer /*
672 1.69 bouyer * To avoid deadlocks never update quotas for quota files
673 1.69 bouyer * on the same file system
674 1.69 bouyer */
675 1.69 bouyer for (i = 0; i < MAXQUOTAS; i++)
676 1.69 bouyer if (vp == ump->um_quotas[i])
677 1.69 bouyer return 0;
678 1.69 bouyer
679 1.69 bouyer ino_ids[USRQUOTA] = ip->i_uid;
680 1.69 bouyer ino_ids[GRPQUOTA] = ip->i_gid;
681 1.69 bouyer for (i = 0; i < MAXQUOTAS; i++) {
682 1.69 bouyer /*
683 1.69 bouyer * If the file id changed the quota needs update.
684 1.69 bouyer */
685 1.69 bouyer if (ip->i_dquot[i] != NODQUOT &&
686 1.69 bouyer ip->i_dquot[i]->dq_id != ino_ids[i]) {
687 1.69 bouyer dqrele(ITOV(ip), ip->i_dquot[i]);
688 1.69 bouyer ip->i_dquot[i] = NODQUOT;
689 1.69 bouyer }
690 1.69 bouyer /*
691 1.69 bouyer * Set up the quota based on file id.
692 1.69 bouyer * ENODEV means that quotas are not enabled.
693 1.69 bouyer */
694 1.69 bouyer if (ip->i_dquot[i] == NODQUOT &&
695 1.69 bouyer (error = dqget(vp, ino_ids[i], ump, i, &ip->i_dquot[i])) &&
696 1.69 bouyer error != ENODEV)
697 1.69 bouyer return (error);
698 1.69 bouyer }
699 1.69 bouyer return 0;
700 1.69 bouyer }
701 1.69 bouyer
702 1.69 bouyer /*
703 1.1 mycroft * Obtain a dquot structure for the specified identifier and quota file
704 1.1 mycroft * reading the information from the file if necessary.
705 1.1 mycroft */
706 1.69 bouyer int
707 1.35 thorpej dqget(struct vnode *vp, u_long id, struct ufsmount *ump, int type,
708 1.35 thorpej struct dquot **dqp)
709 1.1 mycroft {
710 1.50 hannken struct dquot *dq, *ndq;
711 1.21 chs struct dqhashhead *dqh;
712 1.11 fvdl struct vnode *dqvp;
713 1.69 bouyer int error = 0; /* XXX gcc */
714 1.1 mycroft
715 1.50 hannken /* Lock to see an up to date value for QTF_CLOSING. */
716 1.50 hannken mutex_enter(&dqlock);
717 1.69 bouyer if ((ump->um_flags & (UFS_QUOTA|UFS_QUOTA2)) == 0) {
718 1.50 hannken mutex_exit(&dqlock);
719 1.1 mycroft *dqp = NODQUOT;
720 1.69 bouyer return (ENODEV);
721 1.69 bouyer }
722 1.69 bouyer dqvp = ump->um_quotas[type];
723 1.69 bouyer #ifdef QUOTA
724 1.69 bouyer if (ump->um_flags & UFS_QUOTA) {
725 1.69 bouyer if (dqvp == NULLVP || (ump->umq1_qflags[type] & QTF_CLOSING)) {
726 1.69 bouyer mutex_exit(&dqlock);
727 1.69 bouyer *dqp = NODQUOT;
728 1.69 bouyer return (ENODEV);
729 1.69 bouyer }
730 1.1 mycroft }
731 1.69 bouyer #endif
732 1.69 bouyer #ifdef QUOTA2
733 1.69 bouyer if (ump->um_flags & UFS_QUOTA2) {
734 1.69 bouyer if (dqvp == NULLVP) {
735 1.69 bouyer mutex_exit(&dqlock);
736 1.69 bouyer *dqp = NODQUOT;
737 1.69 bouyer return (ENODEV);
738 1.69 bouyer }
739 1.69 bouyer }
740 1.69 bouyer #endif
741 1.50 hannken KASSERT(dqvp != vp);
742 1.1 mycroft /*
743 1.1 mycroft * Check the cache first.
744 1.1 mycroft */
745 1.21 chs dqh = &dqhashtbl[DQHASH(dqvp, id)];
746 1.21 chs LIST_FOREACH(dq, dqh, dq_hash) {
747 1.1 mycroft if (dq->dq_id != id ||
748 1.1 mycroft dq->dq_ump->um_quotas[dq->dq_type] != dqvp)
749 1.1 mycroft continue;
750 1.50 hannken KASSERT(dq->dq_cnt > 0);
751 1.37 chs dqref(dq);
752 1.50 hannken mutex_exit(&dqlock);
753 1.1 mycroft *dqp = dq;
754 1.1 mycroft return (0);
755 1.1 mycroft }
756 1.1 mycroft /*
757 1.1 mycroft * Not in cache, allocate a new one.
758 1.1 mycroft */
759 1.50 hannken mutex_exit(&dqlock);
760 1.55 ad ndq = pool_cache_get(dquot_cache, PR_WAITOK);
761 1.1 mycroft /*
762 1.1 mycroft * Initialize the contents of the dquot structure.
763 1.1 mycroft */
764 1.50 hannken memset((char *)ndq, 0, sizeof *ndq);
765 1.50 hannken ndq->dq_flags = 0;
766 1.50 hannken ndq->dq_id = id;
767 1.50 hannken ndq->dq_ump = ump;
768 1.50 hannken ndq->dq_type = type;
769 1.50 hannken mutex_init(&ndq->dq_interlock, MUTEX_DEFAULT, IPL_NONE);
770 1.50 hannken mutex_enter(&dqlock);
771 1.49 hannken dqh = &dqhashtbl[DQHASH(dqvp, id)];
772 1.50 hannken LIST_FOREACH(dq, dqh, dq_hash) {
773 1.50 hannken if (dq->dq_id != id ||
774 1.50 hannken dq->dq_ump->um_quotas[dq->dq_type] != dqvp)
775 1.50 hannken continue;
776 1.50 hannken /*
777 1.50 hannken * Another thread beat us allocating this dquot.
778 1.50 hannken */
779 1.50 hannken KASSERT(dq->dq_cnt > 0);
780 1.50 hannken dqref(dq);
781 1.50 hannken mutex_exit(&dqlock);
782 1.64 bouyer mutex_destroy(&ndq->dq_interlock);
783 1.55 ad pool_cache_put(dquot_cache, ndq);
784 1.50 hannken *dqp = dq;
785 1.50 hannken return 0;
786 1.50 hannken }
787 1.50 hannken dq = ndq;
788 1.5 mycroft LIST_INSERT_HEAD(dqh, dq, dq_hash);
789 1.37 chs dqref(dq);
790 1.50 hannken mutex_enter(&dq->dq_interlock);
791 1.50 hannken mutex_exit(&dqlock);
792 1.69 bouyer #ifdef QUOTA
793 1.69 bouyer if (ump->um_flags & UFS_QUOTA)
794 1.69 bouyer error = dq1get(dqvp, id, ump, type, dq);
795 1.69 bouyer #endif
796 1.69 bouyer #ifdef QUOTA2
797 1.69 bouyer if (ump->um_flags & UFS_QUOTA2)
798 1.69 bouyer error = dq2get(dqvp, id, ump, type, dq);
799 1.69 bouyer #endif
800 1.1 mycroft /*
801 1.1 mycroft * I/O error in reading quota file, release
802 1.1 mycroft * quota structure and reflect problem to caller.
803 1.1 mycroft */
804 1.1 mycroft if (error) {
805 1.50 hannken mutex_enter(&dqlock);
806 1.5 mycroft LIST_REMOVE(dq, dq_hash);
807 1.50 hannken mutex_exit(&dqlock);
808 1.49 hannken mutex_exit(&dq->dq_interlock);
809 1.1 mycroft dqrele(vp, dq);
810 1.1 mycroft *dqp = NODQUOT;
811 1.1 mycroft return (error);
812 1.1 mycroft }
813 1.49 hannken mutex_exit(&dq->dq_interlock);
814 1.1 mycroft *dqp = dq;
815 1.1 mycroft return (0);
816 1.1 mycroft }
817 1.1 mycroft
818 1.1 mycroft /*
819 1.1 mycroft * Obtain a reference to a dquot.
820 1.1 mycroft */
821 1.69 bouyer void
822 1.35 thorpej dqref(struct dquot *dq)
823 1.1 mycroft {
824 1.1 mycroft
825 1.50 hannken KASSERT(mutex_owned(&dqlock));
826 1.1 mycroft dq->dq_cnt++;
827 1.46 hannken KASSERT(dq->dq_cnt > 0);
828 1.1 mycroft }
829 1.1 mycroft
830 1.1 mycroft /*
831 1.1 mycroft * Release a reference to a dquot.
832 1.1 mycroft */
833 1.69 bouyer void
834 1.35 thorpej dqrele(struct vnode *vp, struct dquot *dq)
835 1.1 mycroft {
836 1.1 mycroft
837 1.1 mycroft if (dq == NODQUOT)
838 1.1 mycroft return;
839 1.50 hannken mutex_enter(&dq->dq_interlock);
840 1.50 hannken for (;;) {
841 1.50 hannken mutex_enter(&dqlock);
842 1.50 hannken if (dq->dq_cnt > 1) {
843 1.50 hannken dq->dq_cnt--;
844 1.50 hannken mutex_exit(&dqlock);
845 1.50 hannken mutex_exit(&dq->dq_interlock);
846 1.50 hannken return;
847 1.50 hannken }
848 1.50 hannken if ((dq->dq_flags & DQ_MOD) == 0)
849 1.50 hannken break;
850 1.50 hannken mutex_exit(&dqlock);
851 1.69 bouyer #ifdef QUOTA
852 1.69 bouyer if (dq->dq_ump->um_flags & UFS_QUOTA)
853 1.69 bouyer (void) dq1sync(vp, dq);
854 1.69 bouyer #endif
855 1.69 bouyer #ifdef QUOTA2
856 1.69 bouyer if (dq->dq_ump->um_flags & UFS_QUOTA2)
857 1.69 bouyer (void) dq2sync(vp, dq);
858 1.69 bouyer #endif
859 1.1 mycroft }
860 1.50 hannken KASSERT(dq->dq_cnt == 1 && (dq->dq_flags & DQ_MOD) == 0);
861 1.50 hannken LIST_REMOVE(dq, dq_hash);
862 1.50 hannken mutex_exit(&dqlock);
863 1.50 hannken mutex_exit(&dq->dq_interlock);
864 1.50 hannken mutex_destroy(&dq->dq_interlock);
865 1.55 ad pool_cache_put(dquot_cache, dq);
866 1.1 mycroft }
867 1.1 mycroft
868 1.69 bouyer int
869 1.69 bouyer qsync(struct mount *mp)
870 1.1 mycroft {
871 1.69 bouyer struct ufsmount *ump = VFSTOUFS(mp);
872 1.69 bouyer #ifdef QUOTA
873 1.69 bouyer if (ump->um_flags & UFS_QUOTA)
874 1.69 bouyer return q1sync(mp);
875 1.69 bouyer #endif
876 1.69 bouyer #ifdef QUOTA2
877 1.69 bouyer if (ump->um_flags & UFS_QUOTA2)
878 1.69 bouyer return q2sync(mp);
879 1.69 bouyer #endif
880 1.69 bouyer return 0;
881 1.1 mycroft }
882 1.1 mycroft
883 1.50 hannken #ifdef DIAGNOSTIC
884 1.1 mycroft /*
885 1.50 hannken * Check the hash chains for stray dquot's.
886 1.1 mycroft */
887 1.69 bouyer void
888 1.35 thorpej dqflush(struct vnode *vp)
889 1.1 mycroft {
890 1.50 hannken struct dquot *dq;
891 1.50 hannken int i;
892 1.1 mycroft
893 1.50 hannken mutex_enter(&dqlock);
894 1.50 hannken for (i = 0; i <= dqhash; i++)
895 1.50 hannken LIST_FOREACH(dq, &dqhashtbl[i], dq_hash)
896 1.50 hannken KASSERT(dq->dq_ump->um_quotas[dq->dq_type] != vp);
897 1.50 hannken mutex_exit(&dqlock);
898 1.1 mycroft }
899 1.50 hannken #endif
900