vfs_quotactl.c revision 1.30 1 1.30 dholland /* $NetBSD: vfs_quotactl.c,v 1.30 2012/01/29 07:10:24 dholland Exp $ */
2 1.3 dholland
3 1.3 dholland /*
4 1.3 dholland * Copyright (c) 1991, 1993, 1994
5 1.3 dholland * The Regents of the University of California. All rights reserved.
6 1.3 dholland * (c) UNIX System Laboratories, Inc.
7 1.3 dholland * All or some portions of this file are derived from material licensed
8 1.3 dholland * to the University of California by American Telephone and Telegraph
9 1.3 dholland * Co. or Unix System Laboratories, Inc. and are reproduced herein with
10 1.3 dholland * the permission of UNIX System Laboratories, Inc.
11 1.3 dholland *
12 1.3 dholland * Redistribution and use in source and binary forms, with or without
13 1.3 dholland * modification, are permitted provided that the following conditions
14 1.3 dholland * are met:
15 1.3 dholland * 1. Redistributions of source code must retain the above copyright
16 1.3 dholland * notice, this list of conditions and the following disclaimer.
17 1.3 dholland * 2. Redistributions in binary form must reproduce the above copyright
18 1.3 dholland * notice, this list of conditions and the following disclaimer in the
19 1.3 dholland * documentation and/or other materials provided with the distribution.
20 1.3 dholland * 3. Neither the name of the University nor the names of its contributors
21 1.3 dholland * may be used to endorse or promote products derived from this software
22 1.3 dholland * without specific prior written permission.
23 1.3 dholland *
24 1.3 dholland * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
25 1.3 dholland * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
26 1.3 dholland * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
27 1.3 dholland * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
28 1.3 dholland * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
29 1.3 dholland * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
30 1.3 dholland * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
31 1.3 dholland * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
32 1.3 dholland * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
33 1.3 dholland * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
34 1.3 dholland * SUCH DAMAGE.
35 1.3 dholland *
36 1.3 dholland * @(#)ufs_vfsops.c 8.8 (Berkeley) 5/20/95
37 1.3 dholland * From NetBSD: ufs_vfsops.c,v 1.42 2011/03/24 17:05:46 bouyer Exp
38 1.3 dholland */
39 1.1 dholland
40 1.1 dholland /*
41 1.1 dholland * Copyright (c) 1982, 1986, 1990, 1993, 1995
42 1.1 dholland * The Regents of the University of California. All rights reserved.
43 1.1 dholland *
44 1.1 dholland * This code is derived from software contributed to Berkeley by
45 1.1 dholland * Robert Elz at The University of Melbourne.
46 1.1 dholland *
47 1.1 dholland * Redistribution and use in source and binary forms, with or without
48 1.1 dholland * modification, are permitted provided that the following conditions
49 1.1 dholland * are met:
50 1.1 dholland * 1. Redistributions of source code must retain the above copyright
51 1.1 dholland * notice, this list of conditions and the following disclaimer.
52 1.1 dholland * 2. Redistributions in binary form must reproduce the above copyright
53 1.1 dholland * notice, this list of conditions and the following disclaimer in the
54 1.1 dholland * documentation and/or other materials provided with the distribution.
55 1.1 dholland * 3. Neither the name of the University nor the names of its contributors
56 1.1 dholland * may be used to endorse or promote products derived from this software
57 1.1 dholland * without specific prior written permission.
58 1.1 dholland *
59 1.1 dholland * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
60 1.1 dholland * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
61 1.1 dholland * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
62 1.1 dholland * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
63 1.1 dholland * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
64 1.1 dholland * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
65 1.1 dholland * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
66 1.1 dholland * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
67 1.1 dholland * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
68 1.1 dholland * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
69 1.1 dholland * SUCH DAMAGE.
70 1.1 dholland *
71 1.1 dholland * @(#)ufs_quota.c 8.5 (Berkeley) 5/20/95
72 1.3 dholland * From NetBSD: ufs_quota.c,v 1.70 2011/03/24 17:05:46 bouyer Exp
73 1.3 dholland */
74 1.3 dholland
75 1.3 dholland /*
76 1.3 dholland * Note that both of the copyrights above are moderately spurious;
77 1.3 dholland * this code should almost certainly have the Copyright 2010 Manuel
78 1.3 dholland * Bouyer notice and license found in e.g. sys/ufs/ufs/quota2_subr.c.
79 1.3 dholland * However, they're what was on the files this code was sliced out of.
80 1.1 dholland */
81 1.1 dholland
82 1.1 dholland #include <sys/cdefs.h>
83 1.30 dholland __KERNEL_RCSID(0, "$NetBSD: vfs_quotactl.c,v 1.30 2012/01/29 07:10:24 dholland Exp $");
84 1.1 dholland
85 1.23 dholland #include <sys/malloc.h> /* XXX: temporary */
86 1.2 dholland #include <sys/mount.h>
87 1.8 dholland #include <sys/quota.h>
88 1.4 dholland #include <sys/quotactl.h>
89 1.3 dholland #include <quota/quotaprop.h>
90 1.2 dholland
91 1.4 dholland static int
92 1.5 dholland vfs_quotactl_getversion(struct mount *mp,
93 1.5 dholland prop_dictionary_t cmddict, int q2type,
94 1.5 dholland prop_array_t datas)
95 1.5 dholland {
96 1.6 dholland prop_array_t replies;
97 1.6 dholland prop_dictionary_t data;
98 1.6 dholland int q2version;
99 1.5 dholland struct vfs_quotactl_args args;
100 1.6 dholland int error;
101 1.6 dholland
102 1.6 dholland KASSERT(prop_object_type(cmddict) == PROP_TYPE_DICTIONARY);
103 1.6 dholland KASSERT(prop_object_type(datas) == PROP_TYPE_ARRAY);
104 1.6 dholland
105 1.6 dholland args.qc_type = QCT_GETVERSION;
106 1.6 dholland args.u.getversion.qc_version_ret = &q2version;
107 1.6 dholland error = VFS_QUOTACTL(mp, QUOTACTL_GETVERSION, &args);
108 1.6 dholland if (error) {
109 1.6 dholland return error;
110 1.6 dholland }
111 1.6 dholland
112 1.6 dholland data = prop_dictionary_create();
113 1.6 dholland if (data == NULL) {
114 1.6 dholland return ENOMEM;
115 1.6 dholland }
116 1.6 dholland
117 1.6 dholland if (!prop_dictionary_set_int8(data, "version", q2version)) {
118 1.6 dholland prop_object_release(data);
119 1.6 dholland return ENOMEM;
120 1.6 dholland }
121 1.6 dholland
122 1.6 dholland replies = prop_array_create();
123 1.6 dholland if (replies == NULL) {
124 1.6 dholland prop_object_release(data);
125 1.6 dholland return ENOMEM;
126 1.6 dholland }
127 1.5 dholland
128 1.6 dholland if (!prop_array_add_and_rel(replies, data)) {
129 1.6 dholland prop_object_release(data);
130 1.6 dholland prop_object_release(replies);
131 1.6 dholland return ENOMEM;
132 1.6 dholland }
133 1.6 dholland
134 1.6 dholland if (!prop_dictionary_set_and_rel(cmddict, "data", replies)) {
135 1.6 dholland prop_object_release(replies);
136 1.6 dholland return ENOMEM;
137 1.6 dholland }
138 1.6 dholland
139 1.6 dholland return error;
140 1.5 dholland }
141 1.5 dholland
142 1.5 dholland static int
143 1.5 dholland vfs_quotactl_quotaon(struct mount *mp,
144 1.5 dholland prop_dictionary_t cmddict, int q2type,
145 1.5 dholland prop_array_t datas)
146 1.5 dholland {
147 1.5 dholland struct vfs_quotactl_args args;
148 1.5 dholland
149 1.5 dholland args.qc_type = QCT_PROPLIB;
150 1.5 dholland args.u.proplib.qc_cmddict = cmddict;
151 1.5 dholland args.u.proplib.qc_q2type = q2type;
152 1.5 dholland args.u.proplib.qc_datas = datas;
153 1.5 dholland return VFS_QUOTACTL(mp, QUOTACTL_QUOTAON, &args);
154 1.5 dholland }
155 1.5 dholland
156 1.5 dholland static int
157 1.5 dholland vfs_quotactl_quotaoff(struct mount *mp,
158 1.5 dholland prop_dictionary_t cmddict, int q2type,
159 1.5 dholland prop_array_t datas)
160 1.5 dholland {
161 1.5 dholland struct vfs_quotactl_args args;
162 1.5 dholland
163 1.5 dholland args.qc_type = QCT_PROPLIB;
164 1.5 dholland args.u.proplib.qc_cmddict = cmddict;
165 1.5 dholland args.u.proplib.qc_q2type = q2type;
166 1.5 dholland args.u.proplib.qc_datas = datas;
167 1.5 dholland return VFS_QUOTACTL(mp, QUOTACTL_QUOTAOFF, &args);
168 1.5 dholland }
169 1.5 dholland
170 1.5 dholland static int
171 1.10 dholland vfs_quotactl_get_addreply(const struct quotakey *qk,
172 1.8 dholland const struct quotaval *blocks,
173 1.8 dholland const struct quotaval *files,
174 1.8 dholland prop_array_t replies)
175 1.8 dholland {
176 1.8 dholland prop_dictionary_t dict;
177 1.10 dholland id_t id;
178 1.10 dholland int defaultq;
179 1.10 dholland uint64_t *valuesp[QUOTA_NLIMITS];
180 1.8 dholland
181 1.8 dholland /* XXX illegal casts */
182 1.8 dholland valuesp[QUOTA_LIMIT_BLOCK] = (void *)(intptr_t)&blocks->qv_hardlimit;
183 1.8 dholland valuesp[QUOTA_LIMIT_FILE] = (void *)(intptr_t)&files->qv_hardlimit;
184 1.8 dholland
185 1.10 dholland if (qk->qk_id == QUOTA_DEFAULTID) {
186 1.10 dholland id = 0;
187 1.10 dholland defaultq = 1;
188 1.10 dholland } else {
189 1.10 dholland id = qk->qk_id;
190 1.10 dholland defaultq = 0;
191 1.10 dholland }
192 1.10 dholland
193 1.8 dholland dict = quota64toprop(id, defaultq, valuesp,
194 1.8 dholland ufs_quota_entry_names, UFS_QUOTA_NENTRIES,
195 1.8 dholland ufs_quota_limit_names, QUOTA_NLIMITS);
196 1.8 dholland if (dict == NULL)
197 1.8 dholland return ENOMEM;
198 1.8 dholland if (!prop_array_add_and_rel(replies, dict)) {
199 1.8 dholland prop_object_release(dict);
200 1.8 dholland return ENOMEM;
201 1.8 dholland }
202 1.8 dholland
203 1.8 dholland return 0;
204 1.8 dholland }
205 1.8 dholland
206 1.8 dholland static int
207 1.5 dholland vfs_quotactl_get(struct mount *mp,
208 1.10 dholland prop_dictionary_t cmddict, int idtype,
209 1.5 dholland prop_array_t datas)
210 1.5 dholland {
211 1.7 dholland prop_object_iterator_t iter;
212 1.7 dholland prop_dictionary_t data;
213 1.8 dholland prop_array_t replies;
214 1.7 dholland uint32_t id;
215 1.7 dholland const char *idstr;
216 1.5 dholland struct vfs_quotactl_args args;
217 1.10 dholland struct quotakey qk;
218 1.8 dholland struct quotaval blocks, files;
219 1.7 dholland int error;
220 1.7 dholland
221 1.7 dholland KASSERT(prop_object_type(cmddict) == PROP_TYPE_DICTIONARY);
222 1.7 dholland KASSERT(prop_object_type(datas) == PROP_TYPE_ARRAY);
223 1.7 dholland
224 1.7 dholland replies = prop_array_create();
225 1.7 dholland if (replies == NULL) {
226 1.7 dholland return ENOMEM;
227 1.7 dholland }
228 1.7 dholland
229 1.7 dholland iter = prop_array_iterator(datas);
230 1.7 dholland if (iter == NULL) {
231 1.7 dholland prop_object_release(replies);
232 1.7 dholland return ENOMEM;
233 1.7 dholland }
234 1.7 dholland
235 1.7 dholland while ((data = prop_object_iterator_next(iter)) != NULL) {
236 1.10 dholland qk.qk_idtype = idtype;
237 1.10 dholland
238 1.7 dholland if (!prop_dictionary_get_uint32(data, "id", &id)) {
239 1.7 dholland if (!prop_dictionary_get_cstring_nocopy(data, "id",
240 1.7 dholland &idstr))
241 1.7 dholland continue;
242 1.7 dholland if (strcmp(idstr, "default")) {
243 1.7 dholland error = EINVAL;
244 1.7 dholland goto fail;
245 1.7 dholland }
246 1.10 dholland qk.qk_id = QUOTA_DEFAULTID;
247 1.7 dholland } else {
248 1.10 dholland qk.qk_id = id;
249 1.7 dholland }
250 1.5 dholland
251 1.10 dholland qk.qk_objtype = QUOTA_OBJTYPE_BLOCKS;
252 1.10 dholland
253 1.7 dholland args.qc_type = QCT_GET;
254 1.10 dholland args.u.get.qc_key = &qk;
255 1.9 dholland args.u.get.qc_ret = &blocks;
256 1.9 dholland error = VFS_QUOTACTL(mp, QUOTACTL_GET, &args);
257 1.9 dholland if (error == EPERM) {
258 1.9 dholland /* XXX does this make sense? */
259 1.9 dholland continue;
260 1.9 dholland } else if (error == ENOENT) {
261 1.9 dholland /* XXX does *this* make sense? */
262 1.9 dholland continue;
263 1.9 dholland } else if (error) {
264 1.9 dholland goto fail;
265 1.9 dholland }
266 1.9 dholland
267 1.10 dholland qk.qk_objtype = QUOTA_OBJTYPE_FILES;
268 1.10 dholland
269 1.9 dholland args.qc_type = QCT_GET;
270 1.10 dholland args.u.get.qc_key = &qk;
271 1.9 dholland args.u.get.qc_ret = &files;
272 1.7 dholland error = VFS_QUOTACTL(mp, QUOTACTL_GET, &args);
273 1.7 dholland if (error == EPERM) {
274 1.7 dholland /* XXX does this make sense? */
275 1.7 dholland continue;
276 1.7 dholland } else if (error == ENOENT) {
277 1.7 dholland /* XXX does *this* make sense? */
278 1.7 dholland continue;
279 1.7 dholland } else if (error) {
280 1.7 dholland goto fail;
281 1.7 dholland }
282 1.8 dholland
283 1.10 dholland error = vfs_quotactl_get_addreply(&qk, &blocks, &files,
284 1.8 dholland replies);
285 1.7 dholland }
286 1.7 dholland
287 1.7 dholland prop_object_iterator_release(iter);
288 1.7 dholland if (!prop_dictionary_set_and_rel(cmddict, "data", replies)) {
289 1.7 dholland error = ENOMEM;
290 1.7 dholland } else {
291 1.7 dholland error = 0;
292 1.7 dholland }
293 1.7 dholland
294 1.7 dholland return error;
295 1.7 dholland
296 1.7 dholland fail:
297 1.7 dholland prop_object_iterator_release(iter);
298 1.7 dholland prop_object_release(replies);
299 1.7 dholland return error;
300 1.5 dholland }
301 1.5 dholland
302 1.5 dholland static int
303 1.15 dholland vfs_quotactl_put_extractinfo(prop_dictionary_t data,
304 1.12 dholland struct quotaval *blocks, struct quotaval *files)
305 1.12 dholland {
306 1.12 dholland /*
307 1.12 dholland * So, the way proptoquota64 works is that you pass it an
308 1.12 dholland * array of pointers to uint64. Each of these pointers is
309 1.12 dholland * supposed to point to 5 (UFS_QUOTA_NENTRIES) uint64s. This
310 1.12 dholland * array of pointers is the second argument. The third and
311 1.12 dholland * forth argument are the names of the five values to extract,
312 1.12 dholland * and UFS_QUOTA_NENTRIES. The last two arguments are the
313 1.12 dholland * names assocated with the pointers (QUOTATYPE_LDICT_BLOCK,
314 1.12 dholland * QUOTADICT_LTYPE_FILE) and the number of pointers. Most of
315 1.12 dholland * the existing code was unsafely casting struct quotaval
316 1.12 dholland * (formerly struct ufs_quota_entry) to (uint64_t *) and using
317 1.12 dholland * that as the block of 5 uint64s. Or worse, pointing to
318 1.12 dholland * subregions of that and reducing the number of uint64s to
319 1.12 dholland * pull "adjacent" values. Demons fly out of your nose!
320 1.12 dholland */
321 1.12 dholland
322 1.12 dholland uint64_t bvals[UFS_QUOTA_NENTRIES];
323 1.12 dholland uint64_t fvals[UFS_QUOTA_NENTRIES];
324 1.12 dholland uint64_t *valptrs[QUOTA_NLIMITS];
325 1.12 dholland int error;
326 1.12 dholland
327 1.12 dholland valptrs[QUOTA_LIMIT_BLOCK] = bvals;
328 1.12 dholland valptrs[QUOTA_LIMIT_FILE] = fvals;
329 1.12 dholland error = proptoquota64(data, valptrs,
330 1.12 dholland ufs_quota_entry_names, UFS_QUOTA_NENTRIES,
331 1.12 dholland ufs_quota_limit_names, QUOTA_NLIMITS);
332 1.12 dholland if (error) {
333 1.12 dholland return error;
334 1.12 dholland }
335 1.12 dholland
336 1.12 dholland /*
337 1.12 dholland * There are no symbolic constants for these indexes!
338 1.12 dholland */
339 1.12 dholland
340 1.12 dholland blocks->qv_hardlimit = bvals[0];
341 1.12 dholland blocks->qv_softlimit = bvals[1];
342 1.12 dholland blocks->qv_usage = bvals[2];
343 1.12 dholland blocks->qv_expiretime = bvals[3];
344 1.12 dholland blocks->qv_grace = bvals[4];
345 1.12 dholland files->qv_hardlimit = fvals[0];
346 1.12 dholland files->qv_softlimit = fvals[1];
347 1.12 dholland files->qv_usage = fvals[2];
348 1.12 dholland files->qv_expiretime = fvals[3];
349 1.12 dholland files->qv_grace = fvals[4];
350 1.12 dholland
351 1.12 dholland return 0;
352 1.12 dholland }
353 1.12 dholland
354 1.12 dholland static int
355 1.15 dholland vfs_quotactl_put(struct mount *mp,
356 1.5 dholland prop_dictionary_t cmddict, int q2type,
357 1.5 dholland prop_array_t datas)
358 1.5 dholland {
359 1.11 dholland prop_array_t replies;
360 1.11 dholland prop_object_iterator_t iter;
361 1.11 dholland prop_dictionary_t data;
362 1.11 dholland int defaultq;
363 1.11 dholland uint32_t id;
364 1.11 dholland const char *idstr;
365 1.14 dholland struct quotakey qk;
366 1.12 dholland struct quotaval blocks, files;
367 1.5 dholland struct vfs_quotactl_args args;
368 1.11 dholland int error;
369 1.11 dholland
370 1.11 dholland KASSERT(prop_object_type(cmddict) == PROP_TYPE_DICTIONARY);
371 1.11 dholland KASSERT(prop_object_type(datas) == PROP_TYPE_ARRAY);
372 1.5 dholland
373 1.11 dholland replies = prop_array_create();
374 1.11 dholland if (replies == NULL)
375 1.11 dholland return ENOMEM;
376 1.11 dholland
377 1.11 dholland iter = prop_array_iterator(datas);
378 1.11 dholland if (iter == NULL) {
379 1.11 dholland prop_object_release(replies);
380 1.11 dholland return ENOMEM;
381 1.11 dholland }
382 1.11 dholland
383 1.11 dholland while ((data = prop_object_iterator_next(iter)) != NULL) {
384 1.13 dholland
385 1.13 dholland KASSERT(prop_object_type(data) == PROP_TYPE_DICTIONARY);
386 1.13 dholland
387 1.11 dholland if (!prop_dictionary_get_uint32(data, "id", &id)) {
388 1.11 dholland if (!prop_dictionary_get_cstring_nocopy(data, "id",
389 1.11 dholland &idstr))
390 1.11 dholland continue;
391 1.11 dholland if (strcmp(idstr, "default"))
392 1.11 dholland continue;
393 1.11 dholland id = 0;
394 1.11 dholland defaultq = 1;
395 1.11 dholland } else {
396 1.11 dholland defaultq = 0;
397 1.11 dholland }
398 1.11 dholland
399 1.15 dholland error = vfs_quotactl_put_extractinfo(data, &blocks, &files);
400 1.12 dholland if (error) {
401 1.12 dholland goto err;
402 1.12 dholland }
403 1.12 dholland
404 1.14 dholland qk.qk_idtype = q2type;
405 1.14 dholland qk.qk_id = defaultq ? QUOTA_DEFAULTID : id;
406 1.14 dholland qk.qk_objtype = QUOTA_OBJTYPE_BLOCKS;
407 1.14 dholland
408 1.15 dholland args.qc_type = QCT_PUT;
409 1.15 dholland args.u.put.qc_key = &qk;
410 1.15 dholland args.u.put.qc_val = &blocks;
411 1.15 dholland error = VFS_QUOTACTL(mp, QUOTACTL_PUT, &args);
412 1.13 dholland if (error) {
413 1.13 dholland goto err;
414 1.13 dholland }
415 1.13 dholland
416 1.14 dholland qk.qk_idtype = q2type;
417 1.14 dholland qk.qk_id = defaultq ? QUOTA_DEFAULTID : id;
418 1.14 dholland qk.qk_objtype = QUOTA_OBJTYPE_FILES;
419 1.14 dholland
420 1.15 dholland args.qc_type = QCT_PUT;
421 1.15 dholland args.u.put.qc_key = &qk;
422 1.15 dholland args.u.put.qc_val = &files;
423 1.15 dholland error = VFS_QUOTACTL(mp, QUOTACTL_PUT, &args);
424 1.11 dholland if (error) {
425 1.11 dholland goto err;
426 1.11 dholland }
427 1.11 dholland }
428 1.11 dholland prop_object_iterator_release(iter);
429 1.11 dholland if (!prop_dictionary_set_and_rel(cmddict, "data", replies)) {
430 1.11 dholland error = ENOMEM;
431 1.11 dholland } else {
432 1.11 dholland error = 0;
433 1.11 dholland }
434 1.11 dholland return error;
435 1.11 dholland err:
436 1.11 dholland prop_object_iterator_release(iter);
437 1.11 dholland prop_object_release(replies);
438 1.11 dholland return error;
439 1.5 dholland }
440 1.5 dholland
441 1.23 dholland static prop_dictionary_t
442 1.26 dholland vfs_quotactl_getall_makereply(const struct quotakey *key)
443 1.26 dholland {
444 1.26 dholland prop_dictionary_t dict;
445 1.26 dholland id_t id;
446 1.26 dholland int defaultq;
447 1.26 dholland
448 1.26 dholland dict = prop_dictionary_create();
449 1.26 dholland if (dict == NULL)
450 1.26 dholland return NULL;
451 1.26 dholland
452 1.26 dholland id = key->qk_id;
453 1.26 dholland if (id == QUOTA_DEFAULTID) {
454 1.26 dholland id = 0;
455 1.26 dholland defaultq = 1;
456 1.26 dholland } else {
457 1.26 dholland defaultq = 0;
458 1.26 dholland }
459 1.26 dholland
460 1.26 dholland if (defaultq) {
461 1.26 dholland if (!prop_dictionary_set_cstring_nocopy(dict, "id",
462 1.26 dholland "default")) {
463 1.26 dholland goto err;
464 1.26 dholland }
465 1.26 dholland } else {
466 1.26 dholland if (!prop_dictionary_set_uint32(dict, "id", id)) {
467 1.26 dholland goto err;
468 1.26 dholland }
469 1.26 dholland }
470 1.26 dholland
471 1.26 dholland return dict;
472 1.26 dholland
473 1.26 dholland err:
474 1.26 dholland prop_object_release(dict);
475 1.26 dholland return NULL;
476 1.26 dholland }
477 1.26 dholland
478 1.26 dholland static int
479 1.26 dholland vfs_quotactl_getall_addreply(prop_dictionary_t thisreply,
480 1.26 dholland const struct quotakey *key, const struct quotaval *val)
481 1.23 dholland {
482 1.23 dholland #define INITQVNAMES_ALL { \
483 1.23 dholland QUOTADICT_LIMIT_HARD, \
484 1.23 dholland QUOTADICT_LIMIT_SOFT, \
485 1.23 dholland QUOTADICT_LIMIT_USAGE, \
486 1.23 dholland QUOTADICT_LIMIT_ETIME, \
487 1.23 dholland QUOTADICT_LIMIT_GTIME \
488 1.23 dholland }
489 1.23 dholland #define N_QV 5
490 1.23 dholland
491 1.23 dholland const char *val_names[] = INITQVNAMES_ALL;
492 1.23 dholland uint64_t vals[N_QV];
493 1.23 dholland prop_dictionary_t dict2;
494 1.26 dholland const char *objtypename;
495 1.23 dholland
496 1.26 dholland switch (key->qk_objtype) {
497 1.26 dholland case QUOTA_OBJTYPE_BLOCKS:
498 1.26 dholland objtypename = QUOTADICT_LTYPE_BLOCK;
499 1.26 dholland break;
500 1.26 dholland case QUOTA_OBJTYPE_FILES:
501 1.26 dholland objtypename = QUOTADICT_LTYPE_FILE;
502 1.26 dholland break;
503 1.26 dholland default:
504 1.26 dholland return EINVAL;
505 1.23 dholland }
506 1.23 dholland
507 1.26 dholland vals[0] = val->qv_hardlimit;
508 1.26 dholland vals[1] = val->qv_softlimit;
509 1.26 dholland vals[2] = val->qv_usage;
510 1.26 dholland vals[3] = val->qv_expiretime;
511 1.26 dholland vals[4] = val->qv_grace;
512 1.23 dholland dict2 = limits64toprop(vals, val_names, N_QV);
513 1.23 dholland if (dict2 == NULL)
514 1.26 dholland return ENOMEM;
515 1.23 dholland
516 1.26 dholland if (!prop_dictionary_set_and_rel(thisreply, objtypename, dict2))
517 1.26 dholland return ENOMEM;
518 1.23 dholland
519 1.26 dholland return 0;
520 1.23 dholland }
521 1.23 dholland
522 1.5 dholland static int
523 1.5 dholland vfs_quotactl_getall(struct mount *mp,
524 1.5 dholland prop_dictionary_t cmddict, int q2type,
525 1.5 dholland prop_array_t datas)
526 1.5 dholland {
527 1.21 dholland struct quotakcursor cursor;
528 1.27 dholland struct quotakey *keys;
529 1.27 dholland struct quotaval *vals;
530 1.27 dholland unsigned loopmax = 8;
531 1.27 dholland unsigned loopnum;
532 1.30 dholland int skipidtype;
533 1.5 dholland struct vfs_quotactl_args args;
534 1.23 dholland prop_array_t replies;
535 1.30 dholland int atend, atzero;
536 1.26 dholland struct quotakey *key;
537 1.26 dholland struct quotaval *val;
538 1.26 dholland id_t lastid;
539 1.27 dholland prop_dictionary_t thisreply;
540 1.23 dholland unsigned i;
541 1.21 dholland int error, error2;
542 1.23 dholland
543 1.23 dholland KASSERT(prop_object_type(cmddict) == PROP_TYPE_DICTIONARY);
544 1.21 dholland
545 1.21 dholland args.qc_type = QCT_CURSOROPEN;
546 1.21 dholland args.u.cursoropen.qc_cursor = &cursor;
547 1.21 dholland error = VFS_QUOTACTL(mp, QUOTACTL_CURSOROPEN, &args);
548 1.21 dholland if (error) {
549 1.21 dholland return error;
550 1.21 dholland }
551 1.5 dholland
552 1.27 dholland keys = malloc(loopmax * sizeof(keys[0]), M_TEMP, M_WAITOK);
553 1.27 dholland vals = malloc(loopmax * sizeof(vals[0]), M_TEMP, M_WAITOK);
554 1.23 dholland
555 1.30 dholland skipidtype = (q2type == QUOTA_IDTYPE_USER ?
556 1.30 dholland QUOTA_IDTYPE_GROUP : QUOTA_IDTYPE_USER);
557 1.30 dholland args.qc_type = QCT_CURSORSKIPIDTYPE;
558 1.30 dholland args.u.cursorskipidtype.qc_cursor = &cursor;
559 1.30 dholland args.u.cursorskipidtype.qc_idtype = skipidtype;
560 1.30 dholland error = VFS_QUOTACTL(mp, QUOTACTL_CURSORSKIPIDTYPE, &args);
561 1.30 dholland /* ignore if it fails */
562 1.30 dholland (void)error;
563 1.30 dholland
564 1.23 dholland replies = prop_array_create();
565 1.23 dholland if (replies == NULL) {
566 1.23 dholland error = ENOMEM;
567 1.23 dholland goto err;
568 1.23 dholland }
569 1.23 dholland
570 1.26 dholland thisreply = NULL;
571 1.26 dholland lastid = 0; /* value not actually referenced */
572 1.30 dholland atzero = 0;
573 1.27 dholland
574 1.27 dholland while (1) {
575 1.30 dholland args.qc_type = QCT_CURSORATEND;
576 1.30 dholland args.u.cursoratend.qc_cursor = &cursor;
577 1.30 dholland args.u.cursoratend.qc_ret = &atend;
578 1.30 dholland error = VFS_QUOTACTL(mp, QUOTACTL_CURSORATEND, &args);
579 1.30 dholland if (error) {
580 1.30 dholland goto err;
581 1.30 dholland }
582 1.30 dholland if (atend) {
583 1.30 dholland break;
584 1.30 dholland }
585 1.30 dholland
586 1.29 dholland args.qc_type = QCT_CURSORGET;
587 1.29 dholland args.u.cursorget.qc_cursor = &cursor;
588 1.29 dholland args.u.cursorget.qc_keys = keys;
589 1.29 dholland args.u.cursorget.qc_vals = vals;
590 1.29 dholland args.u.cursorget.qc_maxnum = loopmax;
591 1.29 dholland args.u.cursorget.qc_ret = &loopnum;
592 1.27 dholland
593 1.29 dholland error = VFS_QUOTACTL(mp, QUOTACTL_CURSORGET, &args);
594 1.30 dholland if (error == EDEADLK) {
595 1.30 dholland /*
596 1.30 dholland * transaction abort, start over
597 1.30 dholland */
598 1.30 dholland
599 1.30 dholland args.qc_type = QCT_CURSORREWIND;
600 1.30 dholland args.u.cursorrewind.qc_cursor = &cursor;
601 1.30 dholland error = VFS_QUOTACTL(mp, QUOTACTL_CURSORREWIND, &args);
602 1.30 dholland if (error) {
603 1.30 dholland goto err;
604 1.30 dholland }
605 1.30 dholland
606 1.30 dholland args.qc_type = QCT_CURSORSKIPIDTYPE;
607 1.30 dholland args.u.cursorskipidtype.qc_cursor = &cursor;
608 1.30 dholland args.u.cursorskipidtype.qc_idtype = skipidtype;
609 1.30 dholland error = VFS_QUOTACTL(mp, QUOTACTL_CURSORSKIPIDTYPE,
610 1.30 dholland &args);
611 1.30 dholland /* ignore if it fails */
612 1.30 dholland (void)error;
613 1.30 dholland
614 1.30 dholland prop_object_release(replies);
615 1.30 dholland replies = prop_array_create();
616 1.30 dholland if (replies == NULL) {
617 1.30 dholland error = ENOMEM;
618 1.30 dholland goto err;
619 1.30 dholland }
620 1.30 dholland
621 1.30 dholland thisreply = NULL;
622 1.30 dholland lastid = 0;
623 1.30 dholland atzero = 0;
624 1.30 dholland
625 1.30 dholland continue;
626 1.30 dholland }
627 1.26 dholland if (error) {
628 1.23 dholland goto err;
629 1.23 dholland }
630 1.27 dholland
631 1.27 dholland if (loopnum == 0) {
632 1.30 dholland /*
633 1.30 dholland * This is not supposed to happen. However,
634 1.30 dholland * allow a return of zero items once as long
635 1.30 dholland * as something happens (including an atend
636 1.30 dholland * indication) on the next pass. If it happens
637 1.30 dholland * twice, warn and assume end of iteration.
638 1.30 dholland */
639 1.30 dholland if (atzero) {
640 1.30 dholland printf("vfs_quotactl: zero items returned\n");
641 1.30 dholland break;
642 1.30 dholland }
643 1.30 dholland atzero = 1;
644 1.30 dholland } else {
645 1.30 dholland atzero = 0;
646 1.27 dholland }
647 1.27 dholland
648 1.27 dholland for (i = 0; i < loopnum; i++) {
649 1.27 dholland key = &keys[i];
650 1.27 dholland val = &vals[i];
651 1.27 dholland
652 1.28 dholland if (key->qk_idtype != q2type) {
653 1.28 dholland /* don't want this result */
654 1.28 dholland continue;
655 1.28 dholland }
656 1.28 dholland
657 1.27 dholland if (thisreply == NULL || key->qk_id != lastid) {
658 1.27 dholland lastid = key->qk_id;
659 1.27 dholland thisreply = vfs_quotactl_getall_makereply(key);
660 1.27 dholland if (thisreply == NULL) {
661 1.27 dholland error = ENOMEM;
662 1.27 dholland goto err;
663 1.27 dholland }
664 1.27 dholland /*
665 1.27 dholland * Note: while we release our reference to
666 1.27 dholland * thisreply here, we can (and do) continue to
667 1.27 dholland * use the pointer in the loop because the
668 1.27 dholland * copy attached to the replies array is not
669 1.27 dholland * going away.
670 1.27 dholland */
671 1.27 dholland if (!prop_array_add_and_rel(replies,
672 1.27 dholland thisreply)) {
673 1.27 dholland error = ENOMEM;
674 1.27 dholland goto err;
675 1.27 dholland }
676 1.27 dholland }
677 1.27 dholland
678 1.27 dholland error = vfs_quotactl_getall_addreply(thisreply,
679 1.27 dholland key, val);
680 1.27 dholland if (error) {
681 1.27 dholland goto err;
682 1.27 dholland }
683 1.27 dholland }
684 1.23 dholland }
685 1.23 dholland
686 1.23 dholland if (!prop_dictionary_set_and_rel(cmddict, "data", replies)) {
687 1.30 dholland replies = NULL;
688 1.23 dholland error = ENOMEM;
689 1.23 dholland goto err;
690 1.23 dholland }
691 1.30 dholland replies = NULL;
692 1.30 dholland error = 0;
693 1.23 dholland
694 1.23 dholland err:
695 1.27 dholland free(keys, M_TEMP);
696 1.27 dholland free(vals, M_TEMP);
697 1.21 dholland
698 1.30 dholland if (replies != NULL) {
699 1.30 dholland prop_object_release(replies);
700 1.30 dholland }
701 1.30 dholland
702 1.21 dholland args.qc_type = QCT_CURSORCLOSE;
703 1.21 dholland args.u.cursorclose.qc_cursor = &cursor;
704 1.21 dholland error2 = VFS_QUOTACTL(mp, QUOTACTL_CURSORCLOSE, &args);
705 1.21 dholland
706 1.21 dholland if (error) {
707 1.21 dholland return error;
708 1.21 dholland }
709 1.21 dholland error = error2;
710 1.21 dholland return error;
711 1.5 dholland }
712 1.5 dholland
713 1.5 dholland static int
714 1.5 dholland vfs_quotactl_clear(struct mount *mp,
715 1.5 dholland prop_dictionary_t cmddict, int q2type,
716 1.5 dholland prop_array_t datas)
717 1.5 dholland {
718 1.16 dholland prop_array_t replies;
719 1.16 dholland prop_object_iterator_t iter;
720 1.16 dholland prop_dictionary_t data;
721 1.16 dholland uint32_t id;
722 1.16 dholland int defaultq;
723 1.16 dholland const char *idstr;
724 1.20 dholland struct quotakey qk;
725 1.5 dholland struct vfs_quotactl_args args;
726 1.16 dholland int error;
727 1.16 dholland
728 1.16 dholland KASSERT(prop_object_type(cmddict) == PROP_TYPE_DICTIONARY);
729 1.16 dholland KASSERT(prop_object_type(datas) == PROP_TYPE_ARRAY);
730 1.16 dholland
731 1.16 dholland replies = prop_array_create();
732 1.16 dholland if (replies == NULL)
733 1.16 dholland return ENOMEM;
734 1.5 dholland
735 1.16 dholland iter = prop_array_iterator(datas);
736 1.16 dholland if (iter == NULL) {
737 1.16 dholland prop_object_release(replies);
738 1.16 dholland return ENOMEM;
739 1.16 dholland }
740 1.16 dholland
741 1.16 dholland while ((data = prop_object_iterator_next(iter)) != NULL) {
742 1.16 dholland if (!prop_dictionary_get_uint32(data, "id", &id)) {
743 1.16 dholland if (!prop_dictionary_get_cstring_nocopy(data, "id",
744 1.16 dholland &idstr))
745 1.16 dholland continue;
746 1.16 dholland if (strcmp(idstr, "default"))
747 1.16 dholland continue;
748 1.16 dholland id = 0;
749 1.16 dholland defaultq = 1;
750 1.16 dholland } else {
751 1.16 dholland defaultq = 0;
752 1.16 dholland }
753 1.16 dholland
754 1.20 dholland qk.qk_idtype = q2type;
755 1.20 dholland qk.qk_id = defaultq ? QUOTA_DEFAULTID : id;
756 1.20 dholland qk.qk_objtype = QUOTA_OBJTYPE_BLOCKS;
757 1.20 dholland
758 1.19 dholland args.qc_type = QCT_DELETE;
759 1.20 dholland args.u.delete.qc_key = &qk;
760 1.19 dholland error = VFS_QUOTACTL(mp, QUOTACTL_DELETE, &args);
761 1.18 dholland if (error) {
762 1.18 dholland goto err;
763 1.18 dholland }
764 1.18 dholland
765 1.20 dholland qk.qk_idtype = q2type;
766 1.20 dholland qk.qk_id = defaultq ? QUOTA_DEFAULTID : id;
767 1.20 dholland qk.qk_objtype = QUOTA_OBJTYPE_FILES;
768 1.20 dholland
769 1.19 dholland args.qc_type = QCT_DELETE;
770 1.20 dholland args.u.delete.qc_key = &qk;
771 1.19 dholland error = VFS_QUOTACTL(mp, QUOTACTL_DELETE, &args);
772 1.16 dholland if (error) {
773 1.16 dholland goto err;
774 1.16 dholland }
775 1.16 dholland }
776 1.16 dholland
777 1.16 dholland prop_object_iterator_release(iter);
778 1.16 dholland if (!prop_dictionary_set_and_rel(cmddict, "data", replies)) {
779 1.16 dholland error = ENOMEM;
780 1.16 dholland } else {
781 1.16 dholland error = 0;
782 1.16 dholland }
783 1.16 dholland return error;
784 1.16 dholland err:
785 1.16 dholland prop_object_iterator_release(iter);
786 1.16 dholland prop_object_release(replies);
787 1.16 dholland return error;
788 1.5 dholland }
789 1.5 dholland
790 1.5 dholland static int
791 1.4 dholland vfs_quotactl_cmd(struct mount *mp, prop_dictionary_t cmddict)
792 1.4 dholland {
793 1.4 dholland int error;
794 1.4 dholland const char *cmd, *type;
795 1.4 dholland prop_array_t datas;
796 1.4 dholland int q2type;
797 1.4 dholland
798 1.4 dholland if (!prop_dictionary_get_cstring_nocopy(cmddict, "command", &cmd))
799 1.4 dholland return EINVAL;
800 1.4 dholland if (!prop_dictionary_get_cstring_nocopy(cmddict, "type", &type))
801 1.4 dholland return EINVAL;
802 1.4 dholland
803 1.4 dholland if (!strcmp(type, QUOTADICT_CLASS_USER)) {
804 1.4 dholland q2type = QUOTA_CLASS_USER;
805 1.4 dholland } else if (!strcmp(type, QUOTADICT_CLASS_GROUP)) {
806 1.4 dholland q2type = QUOTA_CLASS_GROUP;
807 1.4 dholland } else {
808 1.4 dholland /* XXX this is a bad errno for this case */
809 1.4 dholland return EOPNOTSUPP;
810 1.4 dholland }
811 1.4 dholland
812 1.4 dholland datas = prop_dictionary_get(cmddict, "data");
813 1.4 dholland if (datas == NULL || prop_object_type(datas) != PROP_TYPE_ARRAY)
814 1.4 dholland return EINVAL;
815 1.4 dholland
816 1.4 dholland prop_object_retain(datas);
817 1.4 dholland prop_dictionary_remove(cmddict, "data"); /* prepare for return */
818 1.4 dholland
819 1.4 dholland if (strcmp(cmd, "get version") == 0) {
820 1.5 dholland error = vfs_quotactl_getversion(mp, cmddict, q2type, datas);
821 1.4 dholland } else if (strcmp(cmd, "quotaon") == 0) {
822 1.5 dholland error = vfs_quotactl_quotaon(mp, cmddict, q2type, datas);
823 1.4 dholland } else if (strcmp(cmd, "quotaoff") == 0) {
824 1.5 dholland error = vfs_quotactl_quotaoff(mp, cmddict, q2type, datas);
825 1.4 dholland } else if (strcmp(cmd, "get") == 0) {
826 1.5 dholland error = vfs_quotactl_get(mp, cmddict, q2type, datas);
827 1.4 dholland } else if (strcmp(cmd, "set") == 0) {
828 1.15 dholland error = vfs_quotactl_put(mp, cmddict, q2type, datas);
829 1.4 dholland } else if (strcmp(cmd, "getall") == 0) {
830 1.5 dholland error = vfs_quotactl_getall(mp, cmddict, q2type, datas);
831 1.4 dholland } else if (strcmp(cmd, "clear") == 0) {
832 1.5 dholland error = vfs_quotactl_clear(mp, cmddict, q2type, datas);
833 1.4 dholland } else {
834 1.4 dholland /* XXX this a bad errno for this case */
835 1.4 dholland error = EOPNOTSUPP;
836 1.4 dholland }
837 1.4 dholland
838 1.4 dholland error = (prop_dictionary_set_int8(cmddict, "return",
839 1.4 dholland error) ? 0 : ENOMEM);
840 1.4 dholland prop_object_release(datas);
841 1.4 dholland
842 1.4 dholland return error;
843 1.4 dholland }
844 1.4 dholland
845 1.2 dholland int
846 1.2 dholland vfs_quotactl(struct mount *mp, prop_dictionary_t dict)
847 1.2 dholland {
848 1.3 dholland prop_dictionary_t cmddict;
849 1.3 dholland prop_array_t commands;
850 1.3 dholland prop_object_iterator_t iter;
851 1.3 dholland int error;
852 1.3 dholland
853 1.3 dholland error = quota_get_cmds(dict, &commands);
854 1.3 dholland if (error) {
855 1.3 dholland return error;
856 1.3 dholland }
857 1.3 dholland
858 1.3 dholland iter = prop_array_iterator(commands);
859 1.3 dholland if (iter == NULL) {
860 1.3 dholland return ENOMEM;
861 1.3 dholland }
862 1.3 dholland
863 1.3 dholland while ((cmddict = prop_object_iterator_next(iter)) != NULL) {
864 1.3 dholland if (prop_object_type(cmddict) != PROP_TYPE_DICTIONARY) {
865 1.3 dholland /* XXX shouldn't this be an error? */
866 1.3 dholland continue;
867 1.3 dholland }
868 1.4 dholland error = vfs_quotactl_cmd(mp, cmddict);
869 1.3 dholland if (error) {
870 1.3 dholland break;
871 1.3 dholland }
872 1.3 dholland }
873 1.3 dholland prop_object_iterator_release(iter);
874 1.3 dholland return error;
875 1.2 dholland }
876