ulfs_quota.c revision 1.6.2.2 1 1.6.2.2 tls /* $NetBSD: ulfs_quota.c,v 1.6.2.2 2013/06/23 06:18:39 tls Exp $ */
2 1.6.2.2 tls /* from NetBSD: ufs_quota.c,v 1.112 2012/09/09 04:27:49 manu Exp */
3 1.6.2.2 tls
4 1.6.2.2 tls /*
5 1.6.2.2 tls * Copyright (c) 1982, 1986, 1990, 1993, 1995
6 1.6.2.2 tls * The Regents of the University of California. All rights reserved.
7 1.6.2.2 tls *
8 1.6.2.2 tls * This code is derived from software contributed to Berkeley by
9 1.6.2.2 tls * Robert Elz at The University of Melbourne.
10 1.6.2.2 tls *
11 1.6.2.2 tls * Redistribution and use in source and binary forms, with or without
12 1.6.2.2 tls * modification, are permitted provided that the following conditions
13 1.6.2.2 tls * are met:
14 1.6.2.2 tls * 1. Redistributions of source code must retain the above copyright
15 1.6.2.2 tls * notice, this list of conditions and the following disclaimer.
16 1.6.2.2 tls * 2. Redistributions in binary form must reproduce the above copyright
17 1.6.2.2 tls * notice, this list of conditions and the following disclaimer in the
18 1.6.2.2 tls * documentation and/or other materials provided with the distribution.
19 1.6.2.2 tls * 3. Neither the name of the University nor the names of its contributors
20 1.6.2.2 tls * may be used to endorse or promote products derived from this software
21 1.6.2.2 tls * without specific prior written permission.
22 1.6.2.2 tls *
23 1.6.2.2 tls * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
24 1.6.2.2 tls * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
25 1.6.2.2 tls * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
26 1.6.2.2 tls * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
27 1.6.2.2 tls * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
28 1.6.2.2 tls * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
29 1.6.2.2 tls * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
30 1.6.2.2 tls * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
31 1.6.2.2 tls * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
32 1.6.2.2 tls * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
33 1.6.2.2 tls * SUCH DAMAGE.
34 1.6.2.2 tls *
35 1.6.2.2 tls * @(#)ufs_quota.c 8.5 (Berkeley) 5/20/95
36 1.6.2.2 tls */
37 1.6.2.2 tls
38 1.6.2.2 tls #include <sys/cdefs.h>
39 1.6.2.2 tls __KERNEL_RCSID(0, "$NetBSD: ulfs_quota.c,v 1.6.2.2 2013/06/23 06:18:39 tls Exp $");
40 1.6.2.2 tls
41 1.6.2.2 tls #if defined(_KERNEL_OPT)
42 1.6.2.2 tls #include "opt_quota.h"
43 1.6.2.2 tls #endif
44 1.6.2.2 tls #include <sys/param.h>
45 1.6.2.2 tls #include <sys/kernel.h>
46 1.6.2.2 tls #include <sys/systm.h>
47 1.6.2.2 tls #include <sys/namei.h>
48 1.6.2.2 tls #include <sys/file.h>
49 1.6.2.2 tls #include <sys/proc.h>
50 1.6.2.2 tls #include <sys/vnode.h>
51 1.6.2.2 tls #include <sys/mount.h>
52 1.6.2.2 tls #include <sys/kauth.h>
53 1.6.2.2 tls
54 1.6.2.2 tls #include <sys/quotactl.h>
55 1.6.2.2 tls #include <ufs/lfs/ulfs_quotacommon.h>
56 1.6.2.2 tls #include <ufs/lfs/ulfs_inode.h>
57 1.6.2.2 tls #include <ufs/lfs/ulfsmount.h>
58 1.6.2.2 tls #include <ufs/lfs/ulfs_extern.h>
59 1.6.2.2 tls #include <ufs/lfs/ulfs_quota.h>
60 1.6.2.2 tls
61 1.6.2.2 tls kmutex_t lfs_dqlock;
62 1.6.2.2 tls kcondvar_t lfs_dqcv;
63 1.6.2.2 tls const char *lfs_quotatypes[ULFS_MAXQUOTAS] = INITQFNAMES;
64 1.6.2.2 tls
65 1.6.2.2 tls /*
66 1.6.2.2 tls * Code pertaining to management of the in-core dquot data structures.
67 1.6.2.2 tls */
68 1.6.2.2 tls #define DQHASH(dqvp, id) \
69 1.6.2.2 tls (((((long)(dqvp)) >> 8) + id) & dqhash)
70 1.6.2.2 tls static LIST_HEAD(dqhashhead, dquot) *dqhashtbl;
71 1.6.2.2 tls static u_long dqhash;
72 1.6.2.2 tls static pool_cache_t dquot_cache;
73 1.6.2.2 tls
74 1.6.2.2 tls
75 1.6.2.2 tls static int quota_handle_cmd_stat(struct mount *, struct lwp *,
76 1.6.2.2 tls struct quotactl_args *args);
77 1.6.2.2 tls static int quota_handle_cmd_idtypestat(struct mount *, struct lwp *,
78 1.6.2.2 tls struct quotactl_args *args);
79 1.6.2.2 tls static int quota_handle_cmd_objtypestat(struct mount *, struct lwp *,
80 1.6.2.2 tls struct quotactl_args *args);
81 1.6.2.2 tls static int quota_handle_cmd_get(struct mount *, struct lwp *,
82 1.6.2.2 tls struct quotactl_args *args);
83 1.6.2.2 tls static int quota_handle_cmd_put(struct mount *, struct lwp *,
84 1.6.2.2 tls struct quotactl_args *args);
85 1.6.2.2 tls static int quota_handle_cmd_cursorget(struct mount *, struct lwp *,
86 1.6.2.2 tls struct quotactl_args *args);
87 1.6.2.2 tls static int quota_handle_cmd_delete(struct mount *, struct lwp *,
88 1.6.2.2 tls struct quotactl_args *args);
89 1.6.2.2 tls static int quota_handle_cmd_quotaon(struct mount *, struct lwp *,
90 1.6.2.2 tls struct quotactl_args *args);
91 1.6.2.2 tls static int quota_handle_cmd_quotaoff(struct mount *, struct lwp *,
92 1.6.2.2 tls struct quotactl_args *args);
93 1.6.2.2 tls static int quota_handle_cmd_cursoropen(struct mount *, struct lwp *,
94 1.6.2.2 tls struct quotactl_args *args);
95 1.6.2.2 tls static int quota_handle_cmd_cursorclose(struct mount *, struct lwp *,
96 1.6.2.2 tls struct quotactl_args *args);
97 1.6.2.2 tls static int quota_handle_cmd_cursorskipidtype(struct mount *, struct lwp *,
98 1.6.2.2 tls struct quotactl_args *args);
99 1.6.2.2 tls static int quota_handle_cmd_cursoratend(struct mount *, struct lwp *,
100 1.6.2.2 tls struct quotactl_args *args);
101 1.6.2.2 tls static int quota_handle_cmd_cursorrewind(struct mount *, struct lwp *,
102 1.6.2.2 tls struct quotactl_args *args);
103 1.6.2.2 tls
104 1.6.2.2 tls /*
105 1.6.2.2 tls * Initialize the quota fields of an inode.
106 1.6.2.2 tls */
107 1.6.2.2 tls void
108 1.6.2.2 tls ulfsquota_init(struct inode *ip)
109 1.6.2.2 tls {
110 1.6.2.2 tls int i;
111 1.6.2.2 tls
112 1.6.2.2 tls for (i = 0; i < ULFS_MAXQUOTAS; i++)
113 1.6.2.2 tls ip->i_dquot[i] = NODQUOT;
114 1.6.2.2 tls }
115 1.6.2.2 tls
116 1.6.2.2 tls /*
117 1.6.2.2 tls * Release the quota fields from an inode.
118 1.6.2.2 tls */
119 1.6.2.2 tls void
120 1.6.2.2 tls ulfsquota_free(struct inode *ip)
121 1.6.2.2 tls {
122 1.6.2.2 tls int i;
123 1.6.2.2 tls
124 1.6.2.2 tls for (i = 0; i < ULFS_MAXQUOTAS; i++) {
125 1.6.2.2 tls lfs_dqrele(ITOV(ip), ip->i_dquot[i]);
126 1.6.2.2 tls ip->i_dquot[i] = NODQUOT;
127 1.6.2.2 tls }
128 1.6.2.2 tls }
129 1.6.2.2 tls
130 1.6.2.2 tls /*
131 1.6.2.2 tls * Update disk usage, and take corrective action.
132 1.6.2.2 tls */
133 1.6.2.2 tls int
134 1.6.2.2 tls lfs_chkdq(struct inode *ip, int64_t change, kauth_cred_t cred, int flags)
135 1.6.2.2 tls {
136 1.6.2.2 tls /* do not track snapshot usage, or we will deadlock */
137 1.6.2.2 tls if ((ip->i_flags & SF_SNAPSHOT) != 0)
138 1.6.2.2 tls return 0;
139 1.6.2.2 tls
140 1.6.2.2 tls #ifdef LFS_QUOTA
141 1.6.2.2 tls if (ip->i_ump->um_flags & ULFS_QUOTA)
142 1.6.2.2 tls return lfs_chkdq1(ip, change, cred, flags);
143 1.6.2.2 tls #endif
144 1.6.2.2 tls #ifdef LFS_QUOTA2
145 1.6.2.2 tls if (ip->i_ump->um_flags & ULFS_QUOTA2)
146 1.6.2.2 tls return lfs_chkdq2(ip, change, cred, flags);
147 1.6.2.2 tls #endif
148 1.6.2.2 tls return 0;
149 1.6.2.2 tls }
150 1.6.2.2 tls
151 1.6.2.2 tls /*
152 1.6.2.2 tls * Check the inode limit, applying corrective action.
153 1.6.2.2 tls */
154 1.6.2.2 tls int
155 1.6.2.2 tls lfs_chkiq(struct inode *ip, int32_t change, kauth_cred_t cred, int flags)
156 1.6.2.2 tls {
157 1.6.2.2 tls /* do not track snapshot usage, or we will deadlock */
158 1.6.2.2 tls if ((ip->i_flags & SF_SNAPSHOT) != 0)
159 1.6.2.2 tls return 0;
160 1.6.2.2 tls #ifdef LFS_QUOTA
161 1.6.2.2 tls if (ip->i_ump->um_flags & ULFS_QUOTA)
162 1.6.2.2 tls return lfs_chkiq1(ip, change, cred, flags);
163 1.6.2.2 tls #endif
164 1.6.2.2 tls #ifdef LFS_QUOTA2
165 1.6.2.2 tls if (ip->i_ump->um_flags & ULFS_QUOTA2)
166 1.6.2.2 tls return lfs_chkiq2(ip, change, cred, flags);
167 1.6.2.2 tls #endif
168 1.6.2.2 tls return 0;
169 1.6.2.2 tls }
170 1.6.2.2 tls
171 1.6.2.2 tls int
172 1.6.2.2 tls lfsquota_handle_cmd(struct mount *mp, struct lwp *l,
173 1.6.2.2 tls struct quotactl_args *args)
174 1.6.2.2 tls {
175 1.6.2.2 tls int error = 0;
176 1.6.2.2 tls
177 1.6.2.2 tls switch (args->qc_op) {
178 1.6.2.2 tls case QUOTACTL_STAT:
179 1.6.2.2 tls error = quota_handle_cmd_stat(mp, l, args);
180 1.6.2.2 tls break;
181 1.6.2.2 tls case QUOTACTL_IDTYPESTAT:
182 1.6.2.2 tls error = quota_handle_cmd_idtypestat(mp, l, args);
183 1.6.2.2 tls break;
184 1.6.2.2 tls case QUOTACTL_OBJTYPESTAT:
185 1.6.2.2 tls error = quota_handle_cmd_objtypestat(mp, l, args);
186 1.6.2.2 tls break;
187 1.6.2.2 tls case QUOTACTL_QUOTAON:
188 1.6.2.2 tls error = quota_handle_cmd_quotaon(mp, l, args);
189 1.6.2.2 tls break;
190 1.6.2.2 tls case QUOTACTL_QUOTAOFF:
191 1.6.2.2 tls error = quota_handle_cmd_quotaoff(mp, l, args);
192 1.6.2.2 tls break;
193 1.6.2.2 tls case QUOTACTL_GET:
194 1.6.2.2 tls error = quota_handle_cmd_get(mp, l, args);
195 1.6.2.2 tls break;
196 1.6.2.2 tls case QUOTACTL_PUT:
197 1.6.2.2 tls error = quota_handle_cmd_put(mp, l, args);
198 1.6.2.2 tls break;
199 1.6.2.2 tls case QUOTACTL_CURSORGET:
200 1.6.2.2 tls error = quota_handle_cmd_cursorget(mp, l, args);
201 1.6.2.2 tls break;
202 1.6.2.2 tls case QUOTACTL_DELETE:
203 1.6.2.2 tls error = quota_handle_cmd_delete(mp, l, args);
204 1.6.2.2 tls break;
205 1.6.2.2 tls case QUOTACTL_CURSOROPEN:
206 1.6.2.2 tls error = quota_handle_cmd_cursoropen(mp, l, args);
207 1.6.2.2 tls break;
208 1.6.2.2 tls case QUOTACTL_CURSORCLOSE:
209 1.6.2.2 tls error = quota_handle_cmd_cursorclose(mp, l, args);
210 1.6.2.2 tls break;
211 1.6.2.2 tls case QUOTACTL_CURSORSKIPIDTYPE:
212 1.6.2.2 tls error = quota_handle_cmd_cursorskipidtype(mp, l, args);
213 1.6.2.2 tls break;
214 1.6.2.2 tls case QUOTACTL_CURSORATEND:
215 1.6.2.2 tls error = quota_handle_cmd_cursoratend(mp, l, args);
216 1.6.2.2 tls break;
217 1.6.2.2 tls case QUOTACTL_CURSORREWIND:
218 1.6.2.2 tls error = quota_handle_cmd_cursorrewind(mp, l, args);
219 1.6.2.2 tls break;
220 1.6.2.2 tls default:
221 1.6.2.2 tls panic("Invalid quotactl operation %d\n", args->qc_op);
222 1.6.2.2 tls }
223 1.6.2.2 tls
224 1.6.2.2 tls return error;
225 1.6.2.2 tls }
226 1.6.2.2 tls
227 1.6.2.2 tls static int
228 1.6.2.2 tls quota_handle_cmd_stat(struct mount *mp, struct lwp *l,
229 1.6.2.2 tls struct quotactl_args *args)
230 1.6.2.2 tls {
231 1.6.2.2 tls struct ulfsmount *ump = VFSTOULFS(mp);
232 1.6.2.2 tls struct quotastat *info;
233 1.6.2.2 tls
234 1.6.2.2 tls KASSERT(args->qc_op == QUOTACTL_STAT);
235 1.6.2.2 tls info = args->u.stat.qc_info;
236 1.6.2.2 tls
237 1.6.2.2 tls if ((ump->um_flags & (ULFS_QUOTA|ULFS_QUOTA2)) == 0)
238 1.6.2.2 tls return EOPNOTSUPP;
239 1.6.2.2 tls
240 1.6.2.2 tls #ifdef LFS_QUOTA
241 1.6.2.2 tls if (ump->um_flags & ULFS_QUOTA) {
242 1.6.2.2 tls strcpy(info->qs_implname, "lfs quota v1");
243 1.6.2.2 tls info->qs_numidtypes = ULFS_MAXQUOTAS;
244 1.6.2.2 tls /* XXX no define for this */
245 1.6.2.2 tls info->qs_numobjtypes = 2;
246 1.6.2.2 tls info->qs_restrictions = 0;
247 1.6.2.2 tls info->qs_restrictions |= QUOTA_RESTRICT_NEEDSQUOTACHECK;
248 1.6.2.2 tls info->qs_restrictions |= QUOTA_RESTRICT_UNIFORMGRACE;
249 1.6.2.2 tls info->qs_restrictions |= QUOTA_RESTRICT_32BIT;
250 1.6.2.2 tls } else
251 1.6.2.2 tls #endif
252 1.6.2.2 tls #ifdef LFS_QUOTA2
253 1.6.2.2 tls if (ump->um_flags & ULFS_QUOTA2) {
254 1.6.2.2 tls strcpy(info->qs_implname, "lfs quota v2");
255 1.6.2.2 tls info->qs_numidtypes = ULFS_MAXQUOTAS;
256 1.6.2.2 tls info->qs_numobjtypes = N_QL;
257 1.6.2.2 tls info->qs_restrictions = 0;
258 1.6.2.2 tls } else
259 1.6.2.2 tls #endif
260 1.6.2.2 tls return EOPNOTSUPP;
261 1.6.2.2 tls
262 1.6.2.2 tls return 0;
263 1.6.2.2 tls }
264 1.6.2.2 tls
265 1.6.2.2 tls static int
266 1.6.2.2 tls quota_handle_cmd_idtypestat(struct mount *mp, struct lwp *l,
267 1.6.2.2 tls struct quotactl_args *args)
268 1.6.2.2 tls {
269 1.6.2.2 tls struct ulfsmount *ump = VFSTOULFS(mp);
270 1.6.2.2 tls int idtype;
271 1.6.2.2 tls struct quotaidtypestat *info;
272 1.6.2.2 tls const char *name;
273 1.6.2.2 tls
274 1.6.2.2 tls KASSERT(args->qc_op == QUOTACTL_IDTYPESTAT);
275 1.6.2.2 tls idtype = args->u.idtypestat.qc_idtype;
276 1.6.2.2 tls info = args->u.idtypestat.qc_info;
277 1.6.2.2 tls
278 1.6.2.2 tls if ((ump->um_flags & (ULFS_QUOTA|ULFS_QUOTA2)) == 0)
279 1.6.2.2 tls return EOPNOTSUPP;
280 1.6.2.2 tls
281 1.6.2.2 tls /*
282 1.6.2.2 tls * These are the same for both QUOTA and QUOTA2.
283 1.6.2.2 tls */
284 1.6.2.2 tls switch (idtype) {
285 1.6.2.2 tls case QUOTA_IDTYPE_USER:
286 1.6.2.2 tls name = "user";
287 1.6.2.2 tls break;
288 1.6.2.2 tls case QUOTA_IDTYPE_GROUP:
289 1.6.2.2 tls name = "group";
290 1.6.2.2 tls break;
291 1.6.2.2 tls default:
292 1.6.2.2 tls return EINVAL;
293 1.6.2.2 tls }
294 1.6.2.2 tls strlcpy(info->qis_name, name, sizeof(info->qis_name));
295 1.6.2.2 tls return 0;
296 1.6.2.2 tls }
297 1.6.2.2 tls
298 1.6.2.2 tls static int
299 1.6.2.2 tls quota_handle_cmd_objtypestat(struct mount *mp, struct lwp *l,
300 1.6.2.2 tls struct quotactl_args *args)
301 1.6.2.2 tls {
302 1.6.2.2 tls struct ulfsmount *ump = VFSTOULFS(mp);
303 1.6.2.2 tls int objtype;
304 1.6.2.2 tls struct quotaobjtypestat *info;
305 1.6.2.2 tls const char *name;
306 1.6.2.2 tls int isbytes;
307 1.6.2.2 tls
308 1.6.2.2 tls KASSERT(args->qc_op == QUOTACTL_OBJTYPESTAT);
309 1.6.2.2 tls objtype = args->u.objtypestat.qc_objtype;
310 1.6.2.2 tls info = args->u.objtypestat.qc_info;
311 1.6.2.2 tls
312 1.6.2.2 tls if ((ump->um_flags & (ULFS_QUOTA|ULFS_QUOTA2)) == 0)
313 1.6.2.2 tls return EOPNOTSUPP;
314 1.6.2.2 tls
315 1.6.2.2 tls /*
316 1.6.2.2 tls * These are the same for both QUOTA and QUOTA2.
317 1.6.2.2 tls */
318 1.6.2.2 tls switch (objtype) {
319 1.6.2.2 tls case QUOTA_OBJTYPE_BLOCKS:
320 1.6.2.2 tls name = "block";
321 1.6.2.2 tls isbytes = 1;
322 1.6.2.2 tls break;
323 1.6.2.2 tls case QUOTA_OBJTYPE_FILES:
324 1.6.2.2 tls name = "file";
325 1.6.2.2 tls isbytes = 0;
326 1.6.2.2 tls break;
327 1.6.2.2 tls default:
328 1.6.2.2 tls return EINVAL;
329 1.6.2.2 tls }
330 1.6.2.2 tls strlcpy(info->qos_name, name, sizeof(info->qos_name));
331 1.6.2.2 tls info->qos_isbytes = isbytes;
332 1.6.2.2 tls return 0;
333 1.6.2.2 tls }
334 1.6.2.2 tls
335 1.6.2.2 tls /* XXX shouldn't all this be in kauth ? */
336 1.6.2.2 tls static int
337 1.6.2.2 tls quota_get_auth(struct mount *mp, struct lwp *l, uid_t id) {
338 1.6.2.2 tls /* The user can always query about his own quota. */
339 1.6.2.2 tls if (id == kauth_cred_geteuid(l->l_cred))
340 1.6.2.2 tls return 0;
341 1.6.2.2 tls return kauth_authorize_system(l->l_cred, KAUTH_SYSTEM_FS_QUOTA,
342 1.6.2.2 tls KAUTH_REQ_SYSTEM_FS_QUOTA_GET, mp, KAUTH_ARG(id), NULL);
343 1.6.2.2 tls }
344 1.6.2.2 tls
345 1.6.2.2 tls static int
346 1.6.2.2 tls quota_handle_cmd_get(struct mount *mp, struct lwp *l,
347 1.6.2.2 tls struct quotactl_args *args)
348 1.6.2.2 tls {
349 1.6.2.2 tls struct ulfsmount *ump = VFSTOULFS(mp);
350 1.6.2.2 tls int error;
351 1.6.2.2 tls const struct quotakey *qk;
352 1.6.2.2 tls struct quotaval *qv;
353 1.6.2.2 tls
354 1.6.2.2 tls KASSERT(args->qc_op == QUOTACTL_GET);
355 1.6.2.2 tls qk = args->u.get.qc_key;
356 1.6.2.2 tls qv = args->u.get.qc_val;
357 1.6.2.2 tls
358 1.6.2.2 tls if ((ump->um_flags & (ULFS_QUOTA|ULFS_QUOTA2)) == 0)
359 1.6.2.2 tls return EOPNOTSUPP;
360 1.6.2.2 tls
361 1.6.2.2 tls error = quota_get_auth(mp, l, qk->qk_id);
362 1.6.2.2 tls if (error != 0)
363 1.6.2.2 tls return error;
364 1.6.2.2 tls #ifdef LFS_QUOTA
365 1.6.2.2 tls if (ump->um_flags & ULFS_QUOTA) {
366 1.6.2.2 tls error = lfsquota1_handle_cmd_get(ump, qk, qv);
367 1.6.2.2 tls } else
368 1.6.2.2 tls #endif
369 1.6.2.2 tls #ifdef LFS_QUOTA2
370 1.6.2.2 tls if (ump->um_flags & ULFS_QUOTA2) {
371 1.6.2.2 tls error = lfsquota2_handle_cmd_get(ump, qk, qv);
372 1.6.2.2 tls } else
373 1.6.2.2 tls #endif
374 1.6.2.2 tls panic("quota_handle_cmd_get: no support ?");
375 1.6.2.2 tls
376 1.6.2.2 tls if (error != 0)
377 1.6.2.2 tls return error;
378 1.6.2.2 tls
379 1.6.2.2 tls return error;
380 1.6.2.2 tls }
381 1.6.2.2 tls
382 1.6.2.2 tls static int
383 1.6.2.2 tls quota_handle_cmd_put(struct mount *mp, struct lwp *l,
384 1.6.2.2 tls struct quotactl_args *args)
385 1.6.2.2 tls {
386 1.6.2.2 tls struct ulfsmount *ump = VFSTOULFS(mp);
387 1.6.2.2 tls const struct quotakey *qk;
388 1.6.2.2 tls const struct quotaval *qv;
389 1.6.2.2 tls id_t kauth_id;
390 1.6.2.2 tls int error;
391 1.6.2.2 tls
392 1.6.2.2 tls KASSERT(args->qc_op == QUOTACTL_PUT);
393 1.6.2.2 tls qk = args->u.put.qc_key;
394 1.6.2.2 tls qv = args->u.put.qc_val;
395 1.6.2.2 tls
396 1.6.2.2 tls if ((ump->um_flags & (ULFS_QUOTA|ULFS_QUOTA2)) == 0)
397 1.6.2.2 tls return EOPNOTSUPP;
398 1.6.2.2 tls
399 1.6.2.2 tls kauth_id = qk->qk_id;
400 1.6.2.2 tls if (kauth_id == QUOTA_DEFAULTID) {
401 1.6.2.2 tls kauth_id = 0;
402 1.6.2.2 tls }
403 1.6.2.2 tls
404 1.6.2.2 tls error = kauth_authorize_system(l->l_cred, KAUTH_SYSTEM_FS_QUOTA,
405 1.6.2.2 tls KAUTH_REQ_SYSTEM_FS_QUOTA_MANAGE, mp, KAUTH_ARG(kauth_id),
406 1.6.2.2 tls NULL);
407 1.6.2.2 tls if (error != 0) {
408 1.6.2.2 tls return error;
409 1.6.2.2 tls }
410 1.6.2.2 tls
411 1.6.2.2 tls #ifdef LFS_QUOTA
412 1.6.2.2 tls if (ump->um_flags & ULFS_QUOTA)
413 1.6.2.2 tls error = lfsquota1_handle_cmd_put(ump, qk, qv);
414 1.6.2.2 tls else
415 1.6.2.2 tls #endif
416 1.6.2.2 tls #ifdef LFS_QUOTA2
417 1.6.2.2 tls if (ump->um_flags & ULFS_QUOTA2) {
418 1.6.2.2 tls error = lfsquota2_handle_cmd_put(ump, qk, qv);
419 1.6.2.2 tls } else
420 1.6.2.2 tls #endif
421 1.6.2.2 tls panic("quota_handle_cmd_get: no support ?");
422 1.6.2.2 tls
423 1.6.2.2 tls if (error == ENOENT) {
424 1.6.2.2 tls error = 0;
425 1.6.2.2 tls }
426 1.6.2.2 tls
427 1.6.2.2 tls return error;
428 1.6.2.2 tls }
429 1.6.2.2 tls
430 1.6.2.2 tls static int
431 1.6.2.2 tls quota_handle_cmd_delete(struct mount *mp, struct lwp *l,
432 1.6.2.2 tls struct quotactl_args *args)
433 1.6.2.2 tls {
434 1.6.2.2 tls struct ulfsmount *ump = VFSTOULFS(mp);
435 1.6.2.2 tls const struct quotakey *qk;
436 1.6.2.2 tls id_t kauth_id;
437 1.6.2.2 tls int error;
438 1.6.2.2 tls
439 1.6.2.2 tls KASSERT(args->qc_op == QUOTACTL_DELETE);
440 1.6.2.2 tls qk = args->u.delete.qc_key;
441 1.6.2.2 tls
442 1.6.2.2 tls kauth_id = qk->qk_id;
443 1.6.2.2 tls if (kauth_id == QUOTA_DEFAULTID) {
444 1.6.2.2 tls kauth_id = 0;
445 1.6.2.2 tls }
446 1.6.2.2 tls
447 1.6.2.2 tls if ((ump->um_flags & ULFS_QUOTA2) == 0)
448 1.6.2.2 tls return EOPNOTSUPP;
449 1.6.2.2 tls
450 1.6.2.2 tls /* avoid whitespace changes */
451 1.6.2.2 tls {
452 1.6.2.2 tls error = kauth_authorize_system(l->l_cred, KAUTH_SYSTEM_FS_QUOTA,
453 1.6.2.2 tls KAUTH_REQ_SYSTEM_FS_QUOTA_MANAGE, mp, KAUTH_ARG(kauth_id),
454 1.6.2.2 tls NULL);
455 1.6.2.2 tls if (error != 0)
456 1.6.2.2 tls goto err;
457 1.6.2.2 tls #ifdef LFS_QUOTA2
458 1.6.2.2 tls if (ump->um_flags & ULFS_QUOTA2) {
459 1.6.2.2 tls error = lfsquota2_handle_cmd_delete(ump, qk);
460 1.6.2.2 tls } else
461 1.6.2.2 tls #endif
462 1.6.2.2 tls panic("quota_handle_cmd_get: no support ?");
463 1.6.2.2 tls
464 1.6.2.2 tls if (error && error != ENOENT)
465 1.6.2.2 tls goto err;
466 1.6.2.2 tls }
467 1.6.2.2 tls
468 1.6.2.2 tls return 0;
469 1.6.2.2 tls err:
470 1.6.2.2 tls return error;
471 1.6.2.2 tls }
472 1.6.2.2 tls
473 1.6.2.2 tls static int
474 1.6.2.2 tls quota_handle_cmd_cursorget(struct mount *mp, struct lwp *l,
475 1.6.2.2 tls struct quotactl_args *args)
476 1.6.2.2 tls {
477 1.6.2.2 tls struct ulfsmount *ump = VFSTOULFS(mp);
478 1.6.2.2 tls struct quotakcursor *cursor;
479 1.6.2.2 tls struct quotakey *keys;
480 1.6.2.2 tls struct quotaval *vals;
481 1.6.2.2 tls unsigned maxnum;
482 1.6.2.2 tls unsigned *ret;
483 1.6.2.2 tls int error;
484 1.6.2.2 tls
485 1.6.2.2 tls KASSERT(args->qc_op == QUOTACTL_CURSORGET);
486 1.6.2.2 tls cursor = args->u.cursorget.qc_cursor;
487 1.6.2.2 tls keys = args->u.cursorget.qc_keys;
488 1.6.2.2 tls vals = args->u.cursorget.qc_vals;
489 1.6.2.2 tls maxnum = args->u.cursorget.qc_maxnum;
490 1.6.2.2 tls ret = args->u.cursorget.qc_ret;
491 1.6.2.2 tls
492 1.6.2.2 tls if ((ump->um_flags & ULFS_QUOTA2) == 0)
493 1.6.2.2 tls return EOPNOTSUPP;
494 1.6.2.2 tls
495 1.6.2.2 tls error = kauth_authorize_system(l->l_cred, KAUTH_SYSTEM_FS_QUOTA,
496 1.6.2.2 tls KAUTH_REQ_SYSTEM_FS_QUOTA_GET, mp, NULL, NULL);
497 1.6.2.2 tls if (error)
498 1.6.2.2 tls return error;
499 1.6.2.2 tls
500 1.6.2.2 tls #ifdef LFS_QUOTA2
501 1.6.2.2 tls if (ump->um_flags & ULFS_QUOTA2) {
502 1.6.2.2 tls error = lfsquota2_handle_cmd_cursorget(ump, cursor, keys, vals,
503 1.6.2.2 tls maxnum, ret);
504 1.6.2.2 tls } else
505 1.6.2.2 tls #endif
506 1.6.2.2 tls panic("quota_handle_cmd_cursorget: no support ?");
507 1.6.2.2 tls
508 1.6.2.2 tls return error;
509 1.6.2.2 tls }
510 1.6.2.2 tls
511 1.6.2.2 tls static int
512 1.6.2.2 tls quota_handle_cmd_cursoropen(struct mount *mp, struct lwp *l,
513 1.6.2.2 tls struct quotactl_args *args)
514 1.6.2.2 tls {
515 1.6.2.2 tls #ifdef LFS_QUOTA2
516 1.6.2.2 tls struct ulfsmount *ump = VFSTOULFS(mp);
517 1.6.2.2 tls #endif
518 1.6.2.2 tls struct quotakcursor *cursor;
519 1.6.2.2 tls int error;
520 1.6.2.2 tls
521 1.6.2.2 tls KASSERT(args->qc_op == QUOTACTL_CURSOROPEN);
522 1.6.2.2 tls cursor = args->u.cursoropen.qc_cursor;
523 1.6.2.2 tls
524 1.6.2.2 tls error = kauth_authorize_system(l->l_cred, KAUTH_SYSTEM_FS_QUOTA,
525 1.6.2.2 tls KAUTH_REQ_SYSTEM_FS_QUOTA_GET, mp, NULL, NULL);
526 1.6.2.2 tls if (error)
527 1.6.2.2 tls return error;
528 1.6.2.2 tls
529 1.6.2.2 tls #ifdef LFS_QUOTA2
530 1.6.2.2 tls if (ump->um_flags & ULFS_QUOTA2) {
531 1.6.2.2 tls error = lfsquota2_handle_cmd_cursoropen(ump, cursor);
532 1.6.2.2 tls } else
533 1.6.2.2 tls #endif
534 1.6.2.2 tls error = EOPNOTSUPP;
535 1.6.2.2 tls
536 1.6.2.2 tls return error;
537 1.6.2.2 tls }
538 1.6.2.2 tls
539 1.6.2.2 tls static int
540 1.6.2.2 tls quota_handle_cmd_cursorclose(struct mount *mp, struct lwp *l,
541 1.6.2.2 tls struct quotactl_args *args)
542 1.6.2.2 tls {
543 1.6.2.2 tls #ifdef LFS_QUOTA2
544 1.6.2.2 tls struct ulfsmount *ump = VFSTOULFS(mp);
545 1.6.2.2 tls #endif
546 1.6.2.2 tls struct quotakcursor *cursor;
547 1.6.2.2 tls int error;
548 1.6.2.2 tls
549 1.6.2.2 tls KASSERT(args->qc_op == QUOTACTL_CURSORCLOSE);
550 1.6.2.2 tls cursor = args->u.cursorclose.qc_cursor;
551 1.6.2.2 tls
552 1.6.2.2 tls error = kauth_authorize_system(l->l_cred, KAUTH_SYSTEM_FS_QUOTA,
553 1.6.2.2 tls KAUTH_REQ_SYSTEM_FS_QUOTA_GET, mp, NULL, NULL);
554 1.6.2.2 tls if (error)
555 1.6.2.2 tls return error;
556 1.6.2.2 tls
557 1.6.2.2 tls #ifdef LFS_QUOTA2
558 1.6.2.2 tls if (ump->um_flags & ULFS_QUOTA2) {
559 1.6.2.2 tls error = lfsquota2_handle_cmd_cursorclose(ump, cursor);
560 1.6.2.2 tls } else
561 1.6.2.2 tls #endif
562 1.6.2.2 tls error = EOPNOTSUPP;
563 1.6.2.2 tls
564 1.6.2.2 tls return error;
565 1.6.2.2 tls }
566 1.6.2.2 tls
567 1.6.2.2 tls static int
568 1.6.2.2 tls quota_handle_cmd_cursorskipidtype(struct mount *mp, struct lwp *l,
569 1.6.2.2 tls struct quotactl_args *args)
570 1.6.2.2 tls {
571 1.6.2.2 tls #ifdef LFS_QUOTA2
572 1.6.2.2 tls struct ulfsmount *ump = VFSTOULFS(mp);
573 1.6.2.2 tls #endif
574 1.6.2.2 tls struct quotakcursor *cursor;
575 1.6.2.2 tls int idtype;
576 1.6.2.2 tls int error;
577 1.6.2.2 tls
578 1.6.2.2 tls KASSERT(args->qc_op == QUOTACTL_CURSORSKIPIDTYPE);
579 1.6.2.2 tls cursor = args->u.cursorskipidtype.qc_cursor;
580 1.6.2.2 tls idtype = args->u.cursorskipidtype.qc_idtype;
581 1.6.2.2 tls
582 1.6.2.2 tls #ifdef LFS_QUOTA2
583 1.6.2.2 tls if (ump->um_flags & ULFS_QUOTA2) {
584 1.6.2.2 tls error = lfsquota2_handle_cmd_cursorskipidtype(ump, cursor, idtype);
585 1.6.2.2 tls } else
586 1.6.2.2 tls #endif
587 1.6.2.2 tls error = EOPNOTSUPP;
588 1.6.2.2 tls
589 1.6.2.2 tls return error;
590 1.6.2.2 tls }
591 1.6.2.2 tls
592 1.6.2.2 tls static int
593 1.6.2.2 tls quota_handle_cmd_cursoratend(struct mount *mp, struct lwp *l,
594 1.6.2.2 tls struct quotactl_args *args)
595 1.6.2.2 tls {
596 1.6.2.2 tls #ifdef LFS_QUOTA2
597 1.6.2.2 tls struct ulfsmount *ump = VFSTOULFS(mp);
598 1.6.2.2 tls #endif
599 1.6.2.2 tls struct quotakcursor *cursor;
600 1.6.2.2 tls int *ret;
601 1.6.2.2 tls int error;
602 1.6.2.2 tls
603 1.6.2.2 tls KASSERT(args->qc_op == QUOTACTL_CURSORATEND);
604 1.6.2.2 tls cursor = args->u.cursoratend.qc_cursor;
605 1.6.2.2 tls ret = args->u.cursoratend.qc_ret;
606 1.6.2.2 tls
607 1.6.2.2 tls #ifdef LFS_QUOTA2
608 1.6.2.2 tls if (ump->um_flags & ULFS_QUOTA2) {
609 1.6.2.2 tls error = lfsquota2_handle_cmd_cursoratend(ump, cursor, ret);
610 1.6.2.2 tls } else
611 1.6.2.2 tls #endif
612 1.6.2.2 tls error = EOPNOTSUPP;
613 1.6.2.2 tls
614 1.6.2.2 tls return error;
615 1.6.2.2 tls }
616 1.6.2.2 tls
617 1.6.2.2 tls static int
618 1.6.2.2 tls quota_handle_cmd_cursorrewind(struct mount *mp, struct lwp *l,
619 1.6.2.2 tls struct quotactl_args *args)
620 1.6.2.2 tls {
621 1.6.2.2 tls #ifdef LFS_QUOTA2
622 1.6.2.2 tls struct ulfsmount *ump = VFSTOULFS(mp);
623 1.6.2.2 tls #endif
624 1.6.2.2 tls struct quotakcursor *cursor;
625 1.6.2.2 tls int error;
626 1.6.2.2 tls
627 1.6.2.2 tls KASSERT(args->qc_op == QUOTACTL_CURSORREWIND);
628 1.6.2.2 tls cursor = args->u.cursorrewind.qc_cursor;
629 1.6.2.2 tls
630 1.6.2.2 tls #ifdef LFS_QUOTA2
631 1.6.2.2 tls if (ump->um_flags & ULFS_QUOTA2) {
632 1.6.2.2 tls error = lfsquota2_handle_cmd_cursorrewind(ump, cursor);
633 1.6.2.2 tls } else
634 1.6.2.2 tls #endif
635 1.6.2.2 tls error = EOPNOTSUPP;
636 1.6.2.2 tls
637 1.6.2.2 tls return error;
638 1.6.2.2 tls }
639 1.6.2.2 tls
640 1.6.2.2 tls static int
641 1.6.2.2 tls quota_handle_cmd_quotaon(struct mount *mp, struct lwp *l,
642 1.6.2.2 tls struct quotactl_args *args)
643 1.6.2.2 tls {
644 1.6.2.2 tls struct ulfsmount *ump = VFSTOULFS(mp);
645 1.6.2.2 tls int idtype;
646 1.6.2.2 tls const char *qfile;
647 1.6.2.2 tls int error;
648 1.6.2.2 tls
649 1.6.2.2 tls KASSERT(args->qc_op == QUOTACTL_QUOTAON);
650 1.6.2.2 tls idtype = args->u.quotaon.qc_idtype;
651 1.6.2.2 tls qfile = args->u.quotaon.qc_quotafile;
652 1.6.2.2 tls
653 1.6.2.2 tls if ((ump->um_flags & ULFS_QUOTA2) != 0)
654 1.6.2.2 tls return EBUSY;
655 1.6.2.2 tls
656 1.6.2.2 tls error = kauth_authorize_system(l->l_cred, KAUTH_SYSTEM_FS_QUOTA,
657 1.6.2.2 tls KAUTH_REQ_SYSTEM_FS_QUOTA_ONOFF, mp, NULL, NULL);
658 1.6.2.2 tls if (error != 0) {
659 1.6.2.2 tls return error;
660 1.6.2.2 tls }
661 1.6.2.2 tls #ifdef LFS_QUOTA
662 1.6.2.2 tls error = lfsquota1_handle_cmd_quotaon(l, ump, idtype, qfile);
663 1.6.2.2 tls #else
664 1.6.2.2 tls error = EOPNOTSUPP;
665 1.6.2.2 tls #endif
666 1.6.2.2 tls
667 1.6.2.2 tls return error;
668 1.6.2.2 tls }
669 1.6.2.2 tls
670 1.6.2.2 tls static int
671 1.6.2.2 tls quota_handle_cmd_quotaoff(struct mount *mp, struct lwp *l,
672 1.6.2.2 tls struct quotactl_args *args)
673 1.6.2.2 tls {
674 1.6.2.2 tls struct ulfsmount *ump = VFSTOULFS(mp);
675 1.6.2.2 tls int idtype;
676 1.6.2.2 tls int error;
677 1.6.2.2 tls
678 1.6.2.2 tls KASSERT(args->qc_op == QUOTACTL_QUOTAOFF);
679 1.6.2.2 tls idtype = args->u.quotaoff.qc_idtype;
680 1.6.2.2 tls
681 1.6.2.2 tls if ((ump->um_flags & ULFS_QUOTA2) != 0)
682 1.6.2.2 tls return EOPNOTSUPP;
683 1.6.2.2 tls
684 1.6.2.2 tls error = kauth_authorize_system(l->l_cred, KAUTH_SYSTEM_FS_QUOTA,
685 1.6.2.2 tls KAUTH_REQ_SYSTEM_FS_QUOTA_ONOFF, mp, NULL, NULL);
686 1.6.2.2 tls if (error != 0) {
687 1.6.2.2 tls return error;
688 1.6.2.2 tls }
689 1.6.2.2 tls #ifdef LFS_QUOTA
690 1.6.2.2 tls error = lfsquota1_handle_cmd_quotaoff(l, ump, idtype);
691 1.6.2.2 tls #else
692 1.6.2.2 tls error = EOPNOTSUPP;
693 1.6.2.2 tls #endif
694 1.6.2.2 tls
695 1.6.2.2 tls return error;
696 1.6.2.2 tls }
697 1.6.2.2 tls
698 1.6.2.2 tls /*
699 1.6.2.2 tls * Initialize the quota system.
700 1.6.2.2 tls */
701 1.6.2.2 tls void
702 1.6.2.2 tls lfs_dqinit(void)
703 1.6.2.2 tls {
704 1.6.2.2 tls
705 1.6.2.2 tls mutex_init(&lfs_dqlock, MUTEX_DEFAULT, IPL_NONE);
706 1.6.2.2 tls cv_init(&lfs_dqcv, "quota");
707 1.6.2.2 tls dqhashtbl = hashinit(desiredvnodes, HASH_LIST, true, &dqhash);
708 1.6.2.2 tls dquot_cache = pool_cache_init(sizeof(struct dquot), 0, 0, 0, "lfsdq",
709 1.6.2.2 tls NULL, IPL_NONE, NULL, NULL, NULL);
710 1.6.2.2 tls }
711 1.6.2.2 tls
712 1.6.2.2 tls void
713 1.6.2.2 tls lfs_dqreinit(void)
714 1.6.2.2 tls {
715 1.6.2.2 tls struct dquot *dq;
716 1.6.2.2 tls struct dqhashhead *oldhash, *hash;
717 1.6.2.2 tls struct vnode *dqvp;
718 1.6.2.2 tls u_long oldmask, mask, hashval;
719 1.6.2.2 tls int i;
720 1.6.2.2 tls
721 1.6.2.2 tls hash = hashinit(desiredvnodes, HASH_LIST, true, &mask);
722 1.6.2.2 tls mutex_enter(&lfs_dqlock);
723 1.6.2.2 tls oldhash = dqhashtbl;
724 1.6.2.2 tls oldmask = dqhash;
725 1.6.2.2 tls dqhashtbl = hash;
726 1.6.2.2 tls dqhash = mask;
727 1.6.2.2 tls for (i = 0; i <= oldmask; i++) {
728 1.6.2.2 tls while ((dq = LIST_FIRST(&oldhash[i])) != NULL) {
729 1.6.2.2 tls dqvp = dq->dq_ump->um_quotas[dq->dq_type];
730 1.6.2.2 tls LIST_REMOVE(dq, dq_hash);
731 1.6.2.2 tls hashval = DQHASH(dqvp, dq->dq_id);
732 1.6.2.2 tls LIST_INSERT_HEAD(&dqhashtbl[hashval], dq, dq_hash);
733 1.6.2.2 tls }
734 1.6.2.2 tls }
735 1.6.2.2 tls mutex_exit(&lfs_dqlock);
736 1.6.2.2 tls hashdone(oldhash, HASH_LIST, oldmask);
737 1.6.2.2 tls }
738 1.6.2.2 tls
739 1.6.2.2 tls /*
740 1.6.2.2 tls * Free resources held by quota system.
741 1.6.2.2 tls */
742 1.6.2.2 tls void
743 1.6.2.2 tls lfs_dqdone(void)
744 1.6.2.2 tls {
745 1.6.2.2 tls
746 1.6.2.2 tls pool_cache_destroy(dquot_cache);
747 1.6.2.2 tls hashdone(dqhashtbl, HASH_LIST, dqhash);
748 1.6.2.2 tls cv_destroy(&lfs_dqcv);
749 1.6.2.2 tls mutex_destroy(&lfs_dqlock);
750 1.6.2.2 tls }
751 1.6.2.2 tls
752 1.6.2.2 tls /*
753 1.6.2.2 tls * Set up the quotas for an inode.
754 1.6.2.2 tls *
755 1.6.2.2 tls * This routine completely defines the semantics of quotas.
756 1.6.2.2 tls * If other criteria want to be used to establish quotas, the
757 1.6.2.2 tls * ULFS_MAXQUOTAS value in quotas.h should be increased, and the
758 1.6.2.2 tls * additional dquots set up here.
759 1.6.2.2 tls */
760 1.6.2.2 tls int
761 1.6.2.2 tls lfs_getinoquota(struct inode *ip)
762 1.6.2.2 tls {
763 1.6.2.2 tls struct ulfsmount *ump = ip->i_ump;
764 1.6.2.2 tls struct vnode *vp = ITOV(ip);
765 1.6.2.2 tls int i, error;
766 1.6.2.2 tls u_int32_t ino_ids[ULFS_MAXQUOTAS];
767 1.6.2.2 tls
768 1.6.2.2 tls /*
769 1.6.2.2 tls * To avoid deadlocks never update quotas for quota files
770 1.6.2.2 tls * on the same file system
771 1.6.2.2 tls */
772 1.6.2.2 tls for (i = 0; i < ULFS_MAXQUOTAS; i++)
773 1.6.2.2 tls if (vp == ump->um_quotas[i])
774 1.6.2.2 tls return 0;
775 1.6.2.2 tls
776 1.6.2.2 tls ino_ids[ULFS_USRQUOTA] = ip->i_uid;
777 1.6.2.2 tls ino_ids[ULFS_GRPQUOTA] = ip->i_gid;
778 1.6.2.2 tls for (i = 0; i < ULFS_MAXQUOTAS; i++) {
779 1.6.2.2 tls /*
780 1.6.2.2 tls * If the file id changed the quota needs update.
781 1.6.2.2 tls */
782 1.6.2.2 tls if (ip->i_dquot[i] != NODQUOT &&
783 1.6.2.2 tls ip->i_dquot[i]->dq_id != ino_ids[i]) {
784 1.6.2.2 tls lfs_dqrele(ITOV(ip), ip->i_dquot[i]);
785 1.6.2.2 tls ip->i_dquot[i] = NODQUOT;
786 1.6.2.2 tls }
787 1.6.2.2 tls /*
788 1.6.2.2 tls * Set up the quota based on file id.
789 1.6.2.2 tls * ENODEV means that quotas are not enabled.
790 1.6.2.2 tls */
791 1.6.2.2 tls if (ip->i_dquot[i] == NODQUOT &&
792 1.6.2.2 tls (error = lfs_dqget(vp, ino_ids[i], ump, i, &ip->i_dquot[i])) &&
793 1.6.2.2 tls error != ENODEV)
794 1.6.2.2 tls return (error);
795 1.6.2.2 tls }
796 1.6.2.2 tls return 0;
797 1.6.2.2 tls }
798 1.6.2.2 tls
799 1.6.2.2 tls /*
800 1.6.2.2 tls * Obtain a dquot structure for the specified identifier and quota file
801 1.6.2.2 tls * reading the information from the file if necessary.
802 1.6.2.2 tls */
803 1.6.2.2 tls int
804 1.6.2.2 tls lfs_dqget(struct vnode *vp, u_long id, struct ulfsmount *ump, int type,
805 1.6.2.2 tls struct dquot **dqp)
806 1.6.2.2 tls {
807 1.6.2.2 tls struct dquot *dq, *ndq;
808 1.6.2.2 tls struct dqhashhead *dqh;
809 1.6.2.2 tls struct vnode *dqvp;
810 1.6.2.2 tls int error = 0; /* XXX gcc */
811 1.6.2.2 tls
812 1.6.2.2 tls /* Lock to see an up to date value for QTF_CLOSING. */
813 1.6.2.2 tls mutex_enter(&lfs_dqlock);
814 1.6.2.2 tls if ((ump->um_flags & (ULFS_QUOTA|ULFS_QUOTA2)) == 0) {
815 1.6.2.2 tls mutex_exit(&lfs_dqlock);
816 1.6.2.2 tls *dqp = NODQUOT;
817 1.6.2.2 tls return (ENODEV);
818 1.6.2.2 tls }
819 1.6.2.2 tls dqvp = ump->um_quotas[type];
820 1.6.2.2 tls #ifdef LFS_QUOTA
821 1.6.2.2 tls if (ump->um_flags & ULFS_QUOTA) {
822 1.6.2.2 tls if (dqvp == NULLVP || (ump->umq1_qflags[type] & QTF_CLOSING)) {
823 1.6.2.2 tls mutex_exit(&lfs_dqlock);
824 1.6.2.2 tls *dqp = NODQUOT;
825 1.6.2.2 tls return (ENODEV);
826 1.6.2.2 tls }
827 1.6.2.2 tls }
828 1.6.2.2 tls #endif
829 1.6.2.2 tls #ifdef LFS_QUOTA2
830 1.6.2.2 tls if (ump->um_flags & ULFS_QUOTA2) {
831 1.6.2.2 tls if (dqvp == NULLVP) {
832 1.6.2.2 tls mutex_exit(&lfs_dqlock);
833 1.6.2.2 tls *dqp = NODQUOT;
834 1.6.2.2 tls return (ENODEV);
835 1.6.2.2 tls }
836 1.6.2.2 tls }
837 1.6.2.2 tls #endif
838 1.6.2.2 tls KASSERT(dqvp != vp);
839 1.6.2.2 tls /*
840 1.6.2.2 tls * Check the cache first.
841 1.6.2.2 tls */
842 1.6.2.2 tls dqh = &dqhashtbl[DQHASH(dqvp, id)];
843 1.6.2.2 tls LIST_FOREACH(dq, dqh, dq_hash) {
844 1.6.2.2 tls if (dq->dq_id != id ||
845 1.6.2.2 tls dq->dq_ump->um_quotas[dq->dq_type] != dqvp)
846 1.6.2.2 tls continue;
847 1.6.2.2 tls KASSERT(dq->dq_cnt > 0);
848 1.6.2.2 tls lfs_dqref(dq);
849 1.6.2.2 tls mutex_exit(&lfs_dqlock);
850 1.6.2.2 tls *dqp = dq;
851 1.6.2.2 tls return (0);
852 1.6.2.2 tls }
853 1.6.2.2 tls /*
854 1.6.2.2 tls * Not in cache, allocate a new one.
855 1.6.2.2 tls */
856 1.6.2.2 tls mutex_exit(&lfs_dqlock);
857 1.6.2.2 tls ndq = pool_cache_get(dquot_cache, PR_WAITOK);
858 1.6.2.2 tls /*
859 1.6.2.2 tls * Initialize the contents of the dquot structure.
860 1.6.2.2 tls */
861 1.6.2.2 tls memset((char *)ndq, 0, sizeof *ndq);
862 1.6.2.2 tls ndq->dq_flags = 0;
863 1.6.2.2 tls ndq->dq_id = id;
864 1.6.2.2 tls ndq->dq_ump = ump;
865 1.6.2.2 tls ndq->dq_type = type;
866 1.6.2.2 tls mutex_init(&ndq->dq_interlock, MUTEX_DEFAULT, IPL_NONE);
867 1.6.2.2 tls mutex_enter(&lfs_dqlock);
868 1.6.2.2 tls dqh = &dqhashtbl[DQHASH(dqvp, id)];
869 1.6.2.2 tls LIST_FOREACH(dq, dqh, dq_hash) {
870 1.6.2.2 tls if (dq->dq_id != id ||
871 1.6.2.2 tls dq->dq_ump->um_quotas[dq->dq_type] != dqvp)
872 1.6.2.2 tls continue;
873 1.6.2.2 tls /*
874 1.6.2.2 tls * Another thread beat us allocating this dquot.
875 1.6.2.2 tls */
876 1.6.2.2 tls KASSERT(dq->dq_cnt > 0);
877 1.6.2.2 tls lfs_dqref(dq);
878 1.6.2.2 tls mutex_exit(&lfs_dqlock);
879 1.6.2.2 tls mutex_destroy(&ndq->dq_interlock);
880 1.6.2.2 tls pool_cache_put(dquot_cache, ndq);
881 1.6.2.2 tls *dqp = dq;
882 1.6.2.2 tls return 0;
883 1.6.2.2 tls }
884 1.6.2.2 tls dq = ndq;
885 1.6.2.2 tls LIST_INSERT_HEAD(dqh, dq, dq_hash);
886 1.6.2.2 tls lfs_dqref(dq);
887 1.6.2.2 tls mutex_enter(&dq->dq_interlock);
888 1.6.2.2 tls mutex_exit(&lfs_dqlock);
889 1.6.2.2 tls #ifdef LFS_QUOTA
890 1.6.2.2 tls if (ump->um_flags & ULFS_QUOTA)
891 1.6.2.2 tls error = lfs_dq1get(dqvp, id, ump, type, dq);
892 1.6.2.2 tls #endif
893 1.6.2.2 tls #ifdef LFS_QUOTA2
894 1.6.2.2 tls if (ump->um_flags & ULFS_QUOTA2)
895 1.6.2.2 tls error = lfs_dq2get(dqvp, id, ump, type, dq);
896 1.6.2.2 tls #endif
897 1.6.2.2 tls /*
898 1.6.2.2 tls * I/O error in reading quota file, release
899 1.6.2.2 tls * quota structure and reflect problem to caller.
900 1.6.2.2 tls */
901 1.6.2.2 tls if (error) {
902 1.6.2.2 tls mutex_enter(&lfs_dqlock);
903 1.6.2.2 tls LIST_REMOVE(dq, dq_hash);
904 1.6.2.2 tls mutex_exit(&lfs_dqlock);
905 1.6.2.2 tls mutex_exit(&dq->dq_interlock);
906 1.6.2.2 tls lfs_dqrele(vp, dq);
907 1.6.2.2 tls *dqp = NODQUOT;
908 1.6.2.2 tls return (error);
909 1.6.2.2 tls }
910 1.6.2.2 tls mutex_exit(&dq->dq_interlock);
911 1.6.2.2 tls *dqp = dq;
912 1.6.2.2 tls return (0);
913 1.6.2.2 tls }
914 1.6.2.2 tls
915 1.6.2.2 tls /*
916 1.6.2.2 tls * Obtain a reference to a dquot.
917 1.6.2.2 tls */
918 1.6.2.2 tls void
919 1.6.2.2 tls lfs_dqref(struct dquot *dq)
920 1.6.2.2 tls {
921 1.6.2.2 tls
922 1.6.2.2 tls KASSERT(mutex_owned(&lfs_dqlock));
923 1.6.2.2 tls dq->dq_cnt++;
924 1.6.2.2 tls KASSERT(dq->dq_cnt > 0);
925 1.6.2.2 tls }
926 1.6.2.2 tls
927 1.6.2.2 tls /*
928 1.6.2.2 tls * Release a reference to a dquot.
929 1.6.2.2 tls */
930 1.6.2.2 tls void
931 1.6.2.2 tls lfs_dqrele(struct vnode *vp, struct dquot *dq)
932 1.6.2.2 tls {
933 1.6.2.2 tls
934 1.6.2.2 tls if (dq == NODQUOT)
935 1.6.2.2 tls return;
936 1.6.2.2 tls mutex_enter(&dq->dq_interlock);
937 1.6.2.2 tls for (;;) {
938 1.6.2.2 tls mutex_enter(&lfs_dqlock);
939 1.6.2.2 tls if (dq->dq_cnt > 1) {
940 1.6.2.2 tls dq->dq_cnt--;
941 1.6.2.2 tls mutex_exit(&lfs_dqlock);
942 1.6.2.2 tls mutex_exit(&dq->dq_interlock);
943 1.6.2.2 tls return;
944 1.6.2.2 tls }
945 1.6.2.2 tls if ((dq->dq_flags & DQ_MOD) == 0)
946 1.6.2.2 tls break;
947 1.6.2.2 tls mutex_exit(&lfs_dqlock);
948 1.6.2.2 tls #ifdef LFS_QUOTA
949 1.6.2.2 tls if (dq->dq_ump->um_flags & ULFS_QUOTA)
950 1.6.2.2 tls (void) lfs_dq1sync(vp, dq);
951 1.6.2.2 tls #endif
952 1.6.2.2 tls #ifdef LFS_QUOTA2
953 1.6.2.2 tls if (dq->dq_ump->um_flags & ULFS_QUOTA2)
954 1.6.2.2 tls (void) lfs_dq2sync(vp, dq);
955 1.6.2.2 tls #endif
956 1.6.2.2 tls }
957 1.6.2.2 tls KASSERT(dq->dq_cnt == 1 && (dq->dq_flags & DQ_MOD) == 0);
958 1.6.2.2 tls LIST_REMOVE(dq, dq_hash);
959 1.6.2.2 tls mutex_exit(&lfs_dqlock);
960 1.6.2.2 tls mutex_exit(&dq->dq_interlock);
961 1.6.2.2 tls mutex_destroy(&dq->dq_interlock);
962 1.6.2.2 tls pool_cache_put(dquot_cache, dq);
963 1.6.2.2 tls }
964 1.6.2.2 tls
965 1.6.2.2 tls int
966 1.6.2.2 tls lfs_qsync(struct mount *mp)
967 1.6.2.2 tls {
968 1.6.2.2 tls struct ulfsmount *ump = VFSTOULFS(mp);
969 1.6.2.2 tls
970 1.6.2.2 tls /* avoid compiler warning when quotas aren't enabled */
971 1.6.2.2 tls (void)ump;
972 1.6.2.2 tls
973 1.6.2.2 tls #ifdef LFS_QUOTA
974 1.6.2.2 tls if (ump->um_flags & ULFS_QUOTA)
975 1.6.2.2 tls return lfs_q1sync(mp);
976 1.6.2.2 tls #endif
977 1.6.2.2 tls #ifdef LFS_QUOTA2
978 1.6.2.2 tls if (ump->um_flags & ULFS_QUOTA2)
979 1.6.2.2 tls return lfs_q2sync(mp);
980 1.6.2.2 tls #endif
981 1.6.2.2 tls return 0;
982 1.6.2.2 tls }
983 1.6.2.2 tls
984 1.6.2.2 tls #ifdef DIAGNOSTIC
985 1.6.2.2 tls /*
986 1.6.2.2 tls * Check the hash chains for stray dquot's.
987 1.6.2.2 tls */
988 1.6.2.2 tls void
989 1.6.2.2 tls lfs_dqflush(struct vnode *vp)
990 1.6.2.2 tls {
991 1.6.2.2 tls struct dquot *dq;
992 1.6.2.2 tls int i;
993 1.6.2.2 tls
994 1.6.2.2 tls mutex_enter(&lfs_dqlock);
995 1.6.2.2 tls for (i = 0; i <= dqhash; i++)
996 1.6.2.2 tls LIST_FOREACH(dq, &dqhashtbl[i], dq_hash)
997 1.6.2.2 tls KASSERT(dq->dq_ump->um_quotas[dq->dq_type] != vp);
998 1.6.2.2 tls mutex_exit(&lfs_dqlock);
999 1.6.2.2 tls }
1000 1.6.2.2 tls #endif
1001