vfs_quotactl.c revision 1.32 1 1.32 dholland /* $NetBSD: vfs_quotactl.c,v 1.32 2012/01/29 07:11:55 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.32 dholland __KERNEL_RCSID(0, "$NetBSD: vfs_quotactl.c,v 1.32 2012/01/29 07:11:55 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.31 dholland prop_dictionary_t data;
148 1.31 dholland const char *qfile;
149 1.5 dholland struct vfs_quotactl_args args;
150 1.5 dholland
151 1.31 dholland KASSERT(prop_object_type(cmddict) == PROP_TYPE_DICTIONARY);
152 1.31 dholland KASSERT(prop_object_type(datas) == PROP_TYPE_ARRAY);
153 1.31 dholland
154 1.31 dholland if (prop_array_count(datas) != 1)
155 1.31 dholland return EINVAL;
156 1.31 dholland
157 1.31 dholland data = prop_array_get(datas, 0);
158 1.31 dholland if (data == NULL)
159 1.31 dholland return ENOMEM;
160 1.31 dholland if (!prop_dictionary_get_cstring_nocopy(data, "quotafile",
161 1.31 dholland &qfile))
162 1.31 dholland return EINVAL;
163 1.31 dholland
164 1.31 dholland args.qc_type = QCT_QUOTAON;
165 1.31 dholland args.u.quotaon.qc_idtype = q2type;
166 1.31 dholland args.u.quotaon.qc_quotafile = qfile;
167 1.5 dholland return VFS_QUOTACTL(mp, QUOTACTL_QUOTAON, &args);
168 1.5 dholland }
169 1.5 dholland
170 1.5 dholland static int
171 1.5 dholland vfs_quotactl_quotaoff(struct mount *mp,
172 1.5 dholland prop_dictionary_t cmddict, int q2type,
173 1.5 dholland prop_array_t datas)
174 1.5 dholland {
175 1.5 dholland struct vfs_quotactl_args args;
176 1.5 dholland
177 1.32 dholland KASSERT(prop_object_type(cmddict) == PROP_TYPE_DICTIONARY);
178 1.32 dholland KASSERT(prop_object_type(datas) == PROP_TYPE_ARRAY);
179 1.32 dholland
180 1.32 dholland if (prop_array_count(datas) != 0)
181 1.32 dholland return EINVAL;
182 1.32 dholland
183 1.32 dholland args.qc_type = QCT_QUOTAOFF;
184 1.32 dholland args.u.quotaoff.qc_idtype = q2type;
185 1.5 dholland return VFS_QUOTACTL(mp, QUOTACTL_QUOTAOFF, &args);
186 1.5 dholland }
187 1.5 dholland
188 1.5 dholland static int
189 1.10 dholland vfs_quotactl_get_addreply(const struct quotakey *qk,
190 1.8 dholland const struct quotaval *blocks,
191 1.8 dholland const struct quotaval *files,
192 1.8 dholland prop_array_t replies)
193 1.8 dholland {
194 1.8 dholland prop_dictionary_t dict;
195 1.10 dholland id_t id;
196 1.10 dholland int defaultq;
197 1.10 dholland uint64_t *valuesp[QUOTA_NLIMITS];
198 1.8 dholland
199 1.8 dholland /* XXX illegal casts */
200 1.8 dholland valuesp[QUOTA_LIMIT_BLOCK] = (void *)(intptr_t)&blocks->qv_hardlimit;
201 1.8 dholland valuesp[QUOTA_LIMIT_FILE] = (void *)(intptr_t)&files->qv_hardlimit;
202 1.8 dholland
203 1.10 dholland if (qk->qk_id == QUOTA_DEFAULTID) {
204 1.10 dholland id = 0;
205 1.10 dholland defaultq = 1;
206 1.10 dholland } else {
207 1.10 dholland id = qk->qk_id;
208 1.10 dholland defaultq = 0;
209 1.10 dholland }
210 1.10 dholland
211 1.8 dholland dict = quota64toprop(id, defaultq, valuesp,
212 1.8 dholland ufs_quota_entry_names, UFS_QUOTA_NENTRIES,
213 1.8 dholland ufs_quota_limit_names, QUOTA_NLIMITS);
214 1.8 dholland if (dict == NULL)
215 1.8 dholland return ENOMEM;
216 1.8 dholland if (!prop_array_add_and_rel(replies, dict)) {
217 1.8 dholland prop_object_release(dict);
218 1.8 dholland return ENOMEM;
219 1.8 dholland }
220 1.8 dholland
221 1.8 dholland return 0;
222 1.8 dholland }
223 1.8 dholland
224 1.8 dholland static int
225 1.5 dholland vfs_quotactl_get(struct mount *mp,
226 1.10 dholland prop_dictionary_t cmddict, int idtype,
227 1.5 dholland prop_array_t datas)
228 1.5 dholland {
229 1.7 dholland prop_object_iterator_t iter;
230 1.7 dholland prop_dictionary_t data;
231 1.8 dholland prop_array_t replies;
232 1.7 dholland uint32_t id;
233 1.7 dholland const char *idstr;
234 1.5 dholland struct vfs_quotactl_args args;
235 1.10 dholland struct quotakey qk;
236 1.8 dholland struct quotaval blocks, files;
237 1.7 dholland int error;
238 1.7 dholland
239 1.7 dholland KASSERT(prop_object_type(cmddict) == PROP_TYPE_DICTIONARY);
240 1.7 dholland KASSERT(prop_object_type(datas) == PROP_TYPE_ARRAY);
241 1.7 dholland
242 1.7 dholland replies = prop_array_create();
243 1.7 dholland if (replies == NULL) {
244 1.7 dholland return ENOMEM;
245 1.7 dholland }
246 1.7 dholland
247 1.7 dholland iter = prop_array_iterator(datas);
248 1.7 dholland if (iter == NULL) {
249 1.7 dholland prop_object_release(replies);
250 1.7 dholland return ENOMEM;
251 1.7 dholland }
252 1.7 dholland
253 1.7 dholland while ((data = prop_object_iterator_next(iter)) != NULL) {
254 1.10 dholland qk.qk_idtype = idtype;
255 1.10 dholland
256 1.7 dholland if (!prop_dictionary_get_uint32(data, "id", &id)) {
257 1.7 dholland if (!prop_dictionary_get_cstring_nocopy(data, "id",
258 1.7 dholland &idstr))
259 1.7 dholland continue;
260 1.7 dholland if (strcmp(idstr, "default")) {
261 1.7 dholland error = EINVAL;
262 1.7 dholland goto fail;
263 1.7 dholland }
264 1.10 dholland qk.qk_id = QUOTA_DEFAULTID;
265 1.7 dholland } else {
266 1.10 dholland qk.qk_id = id;
267 1.7 dholland }
268 1.5 dholland
269 1.10 dholland qk.qk_objtype = QUOTA_OBJTYPE_BLOCKS;
270 1.10 dholland
271 1.7 dholland args.qc_type = QCT_GET;
272 1.10 dholland args.u.get.qc_key = &qk;
273 1.9 dholland args.u.get.qc_ret = &blocks;
274 1.9 dholland error = VFS_QUOTACTL(mp, QUOTACTL_GET, &args);
275 1.9 dholland if (error == EPERM) {
276 1.9 dholland /* XXX does this make sense? */
277 1.9 dholland continue;
278 1.9 dholland } else if (error == ENOENT) {
279 1.9 dholland /* XXX does *this* make sense? */
280 1.9 dholland continue;
281 1.9 dholland } else if (error) {
282 1.9 dholland goto fail;
283 1.9 dholland }
284 1.9 dholland
285 1.10 dholland qk.qk_objtype = QUOTA_OBJTYPE_FILES;
286 1.10 dholland
287 1.9 dholland args.qc_type = QCT_GET;
288 1.10 dholland args.u.get.qc_key = &qk;
289 1.9 dholland args.u.get.qc_ret = &files;
290 1.7 dholland error = VFS_QUOTACTL(mp, QUOTACTL_GET, &args);
291 1.7 dholland if (error == EPERM) {
292 1.7 dholland /* XXX does this make sense? */
293 1.7 dholland continue;
294 1.7 dholland } else if (error == ENOENT) {
295 1.7 dholland /* XXX does *this* make sense? */
296 1.7 dholland continue;
297 1.7 dholland } else if (error) {
298 1.7 dholland goto fail;
299 1.7 dholland }
300 1.8 dholland
301 1.10 dholland error = vfs_quotactl_get_addreply(&qk, &blocks, &files,
302 1.8 dholland replies);
303 1.7 dholland }
304 1.7 dholland
305 1.7 dholland prop_object_iterator_release(iter);
306 1.7 dholland if (!prop_dictionary_set_and_rel(cmddict, "data", replies)) {
307 1.7 dholland error = ENOMEM;
308 1.7 dholland } else {
309 1.7 dholland error = 0;
310 1.7 dholland }
311 1.7 dholland
312 1.7 dholland return error;
313 1.7 dholland
314 1.7 dholland fail:
315 1.7 dholland prop_object_iterator_release(iter);
316 1.7 dholland prop_object_release(replies);
317 1.7 dholland return error;
318 1.5 dholland }
319 1.5 dholland
320 1.5 dholland static int
321 1.15 dholland vfs_quotactl_put_extractinfo(prop_dictionary_t data,
322 1.12 dholland struct quotaval *blocks, struct quotaval *files)
323 1.12 dholland {
324 1.12 dholland /*
325 1.12 dholland * So, the way proptoquota64 works is that you pass it an
326 1.12 dholland * array of pointers to uint64. Each of these pointers is
327 1.12 dholland * supposed to point to 5 (UFS_QUOTA_NENTRIES) uint64s. This
328 1.12 dholland * array of pointers is the second argument. The third and
329 1.12 dholland * forth argument are the names of the five values to extract,
330 1.12 dholland * and UFS_QUOTA_NENTRIES. The last two arguments are the
331 1.12 dholland * names assocated with the pointers (QUOTATYPE_LDICT_BLOCK,
332 1.12 dholland * QUOTADICT_LTYPE_FILE) and the number of pointers. Most of
333 1.12 dholland * the existing code was unsafely casting struct quotaval
334 1.12 dholland * (formerly struct ufs_quota_entry) to (uint64_t *) and using
335 1.12 dholland * that as the block of 5 uint64s. Or worse, pointing to
336 1.12 dholland * subregions of that and reducing the number of uint64s to
337 1.12 dholland * pull "adjacent" values. Demons fly out of your nose!
338 1.12 dholland */
339 1.12 dholland
340 1.12 dholland uint64_t bvals[UFS_QUOTA_NENTRIES];
341 1.12 dholland uint64_t fvals[UFS_QUOTA_NENTRIES];
342 1.12 dholland uint64_t *valptrs[QUOTA_NLIMITS];
343 1.12 dholland int error;
344 1.12 dholland
345 1.12 dholland valptrs[QUOTA_LIMIT_BLOCK] = bvals;
346 1.12 dholland valptrs[QUOTA_LIMIT_FILE] = fvals;
347 1.12 dholland error = proptoquota64(data, valptrs,
348 1.12 dholland ufs_quota_entry_names, UFS_QUOTA_NENTRIES,
349 1.12 dholland ufs_quota_limit_names, QUOTA_NLIMITS);
350 1.12 dholland if (error) {
351 1.12 dholland return error;
352 1.12 dholland }
353 1.12 dholland
354 1.12 dholland /*
355 1.12 dholland * There are no symbolic constants for these indexes!
356 1.12 dholland */
357 1.12 dholland
358 1.12 dholland blocks->qv_hardlimit = bvals[0];
359 1.12 dholland blocks->qv_softlimit = bvals[1];
360 1.12 dholland blocks->qv_usage = bvals[2];
361 1.12 dholland blocks->qv_expiretime = bvals[3];
362 1.12 dholland blocks->qv_grace = bvals[4];
363 1.12 dholland files->qv_hardlimit = fvals[0];
364 1.12 dholland files->qv_softlimit = fvals[1];
365 1.12 dholland files->qv_usage = fvals[2];
366 1.12 dholland files->qv_expiretime = fvals[3];
367 1.12 dholland files->qv_grace = fvals[4];
368 1.12 dholland
369 1.12 dholland return 0;
370 1.12 dholland }
371 1.12 dholland
372 1.12 dholland static int
373 1.15 dholland vfs_quotactl_put(struct mount *mp,
374 1.5 dholland prop_dictionary_t cmddict, int q2type,
375 1.5 dholland prop_array_t datas)
376 1.5 dholland {
377 1.11 dholland prop_array_t replies;
378 1.11 dholland prop_object_iterator_t iter;
379 1.11 dholland prop_dictionary_t data;
380 1.11 dholland int defaultq;
381 1.11 dholland uint32_t id;
382 1.11 dholland const char *idstr;
383 1.14 dholland struct quotakey qk;
384 1.12 dholland struct quotaval blocks, files;
385 1.5 dholland struct vfs_quotactl_args args;
386 1.11 dholland int error;
387 1.11 dholland
388 1.11 dholland KASSERT(prop_object_type(cmddict) == PROP_TYPE_DICTIONARY);
389 1.11 dholland KASSERT(prop_object_type(datas) == PROP_TYPE_ARRAY);
390 1.5 dholland
391 1.11 dholland replies = prop_array_create();
392 1.11 dholland if (replies == NULL)
393 1.11 dholland return ENOMEM;
394 1.11 dholland
395 1.11 dholland iter = prop_array_iterator(datas);
396 1.11 dholland if (iter == NULL) {
397 1.11 dholland prop_object_release(replies);
398 1.11 dholland return ENOMEM;
399 1.11 dholland }
400 1.11 dholland
401 1.11 dholland while ((data = prop_object_iterator_next(iter)) != NULL) {
402 1.13 dholland
403 1.13 dholland KASSERT(prop_object_type(data) == PROP_TYPE_DICTIONARY);
404 1.13 dholland
405 1.11 dholland if (!prop_dictionary_get_uint32(data, "id", &id)) {
406 1.11 dholland if (!prop_dictionary_get_cstring_nocopy(data, "id",
407 1.11 dholland &idstr))
408 1.11 dholland continue;
409 1.11 dholland if (strcmp(idstr, "default"))
410 1.11 dholland continue;
411 1.11 dholland id = 0;
412 1.11 dholland defaultq = 1;
413 1.11 dholland } else {
414 1.11 dholland defaultq = 0;
415 1.11 dholland }
416 1.11 dholland
417 1.15 dholland error = vfs_quotactl_put_extractinfo(data, &blocks, &files);
418 1.12 dholland if (error) {
419 1.12 dholland goto err;
420 1.12 dholland }
421 1.12 dholland
422 1.14 dholland qk.qk_idtype = q2type;
423 1.14 dholland qk.qk_id = defaultq ? QUOTA_DEFAULTID : id;
424 1.14 dholland qk.qk_objtype = QUOTA_OBJTYPE_BLOCKS;
425 1.14 dholland
426 1.15 dholland args.qc_type = QCT_PUT;
427 1.15 dholland args.u.put.qc_key = &qk;
428 1.15 dholland args.u.put.qc_val = &blocks;
429 1.15 dholland error = VFS_QUOTACTL(mp, QUOTACTL_PUT, &args);
430 1.13 dholland if (error) {
431 1.13 dholland goto err;
432 1.13 dholland }
433 1.13 dholland
434 1.14 dholland qk.qk_idtype = q2type;
435 1.14 dholland qk.qk_id = defaultq ? QUOTA_DEFAULTID : id;
436 1.14 dholland qk.qk_objtype = QUOTA_OBJTYPE_FILES;
437 1.14 dholland
438 1.15 dholland args.qc_type = QCT_PUT;
439 1.15 dholland args.u.put.qc_key = &qk;
440 1.15 dholland args.u.put.qc_val = &files;
441 1.15 dholland error = VFS_QUOTACTL(mp, QUOTACTL_PUT, &args);
442 1.11 dholland if (error) {
443 1.11 dholland goto err;
444 1.11 dholland }
445 1.11 dholland }
446 1.11 dholland prop_object_iterator_release(iter);
447 1.11 dholland if (!prop_dictionary_set_and_rel(cmddict, "data", replies)) {
448 1.11 dholland error = ENOMEM;
449 1.11 dholland } else {
450 1.11 dholland error = 0;
451 1.11 dholland }
452 1.11 dholland return error;
453 1.11 dholland err:
454 1.11 dholland prop_object_iterator_release(iter);
455 1.11 dholland prop_object_release(replies);
456 1.11 dholland return error;
457 1.5 dholland }
458 1.5 dholland
459 1.23 dholland static prop_dictionary_t
460 1.26 dholland vfs_quotactl_getall_makereply(const struct quotakey *key)
461 1.26 dholland {
462 1.26 dholland prop_dictionary_t dict;
463 1.26 dholland id_t id;
464 1.26 dholland int defaultq;
465 1.26 dholland
466 1.26 dholland dict = prop_dictionary_create();
467 1.26 dholland if (dict == NULL)
468 1.26 dholland return NULL;
469 1.26 dholland
470 1.26 dholland id = key->qk_id;
471 1.26 dholland if (id == QUOTA_DEFAULTID) {
472 1.26 dholland id = 0;
473 1.26 dholland defaultq = 1;
474 1.26 dholland } else {
475 1.26 dholland defaultq = 0;
476 1.26 dholland }
477 1.26 dholland
478 1.26 dholland if (defaultq) {
479 1.26 dholland if (!prop_dictionary_set_cstring_nocopy(dict, "id",
480 1.26 dholland "default")) {
481 1.26 dholland goto err;
482 1.26 dholland }
483 1.26 dholland } else {
484 1.26 dholland if (!prop_dictionary_set_uint32(dict, "id", id)) {
485 1.26 dholland goto err;
486 1.26 dholland }
487 1.26 dholland }
488 1.26 dholland
489 1.26 dholland return dict;
490 1.26 dholland
491 1.26 dholland err:
492 1.26 dholland prop_object_release(dict);
493 1.26 dholland return NULL;
494 1.26 dholland }
495 1.26 dholland
496 1.26 dholland static int
497 1.26 dholland vfs_quotactl_getall_addreply(prop_dictionary_t thisreply,
498 1.26 dholland const struct quotakey *key, const struct quotaval *val)
499 1.23 dholland {
500 1.23 dholland #define INITQVNAMES_ALL { \
501 1.23 dholland QUOTADICT_LIMIT_HARD, \
502 1.23 dholland QUOTADICT_LIMIT_SOFT, \
503 1.23 dholland QUOTADICT_LIMIT_USAGE, \
504 1.23 dholland QUOTADICT_LIMIT_ETIME, \
505 1.23 dholland QUOTADICT_LIMIT_GTIME \
506 1.23 dholland }
507 1.23 dholland #define N_QV 5
508 1.23 dholland
509 1.23 dholland const char *val_names[] = INITQVNAMES_ALL;
510 1.23 dholland uint64_t vals[N_QV];
511 1.23 dholland prop_dictionary_t dict2;
512 1.26 dholland const char *objtypename;
513 1.23 dholland
514 1.26 dholland switch (key->qk_objtype) {
515 1.26 dholland case QUOTA_OBJTYPE_BLOCKS:
516 1.26 dholland objtypename = QUOTADICT_LTYPE_BLOCK;
517 1.26 dholland break;
518 1.26 dholland case QUOTA_OBJTYPE_FILES:
519 1.26 dholland objtypename = QUOTADICT_LTYPE_FILE;
520 1.26 dholland break;
521 1.26 dholland default:
522 1.26 dholland return EINVAL;
523 1.23 dholland }
524 1.23 dholland
525 1.26 dholland vals[0] = val->qv_hardlimit;
526 1.26 dholland vals[1] = val->qv_softlimit;
527 1.26 dholland vals[2] = val->qv_usage;
528 1.26 dholland vals[3] = val->qv_expiretime;
529 1.26 dholland vals[4] = val->qv_grace;
530 1.23 dholland dict2 = limits64toprop(vals, val_names, N_QV);
531 1.23 dholland if (dict2 == NULL)
532 1.26 dholland return ENOMEM;
533 1.23 dholland
534 1.26 dholland if (!prop_dictionary_set_and_rel(thisreply, objtypename, dict2))
535 1.26 dholland return ENOMEM;
536 1.23 dholland
537 1.26 dholland return 0;
538 1.23 dholland }
539 1.23 dholland
540 1.5 dholland static int
541 1.5 dholland vfs_quotactl_getall(struct mount *mp,
542 1.5 dholland prop_dictionary_t cmddict, int q2type,
543 1.5 dholland prop_array_t datas)
544 1.5 dholland {
545 1.21 dholland struct quotakcursor cursor;
546 1.27 dholland struct quotakey *keys;
547 1.27 dholland struct quotaval *vals;
548 1.27 dholland unsigned loopmax = 8;
549 1.27 dholland unsigned loopnum;
550 1.30 dholland int skipidtype;
551 1.5 dholland struct vfs_quotactl_args args;
552 1.23 dholland prop_array_t replies;
553 1.30 dholland int atend, atzero;
554 1.26 dholland struct quotakey *key;
555 1.26 dholland struct quotaval *val;
556 1.26 dholland id_t lastid;
557 1.27 dholland prop_dictionary_t thisreply;
558 1.23 dholland unsigned i;
559 1.21 dholland int error, error2;
560 1.23 dholland
561 1.23 dholland KASSERT(prop_object_type(cmddict) == PROP_TYPE_DICTIONARY);
562 1.21 dholland
563 1.21 dholland args.qc_type = QCT_CURSOROPEN;
564 1.21 dholland args.u.cursoropen.qc_cursor = &cursor;
565 1.21 dholland error = VFS_QUOTACTL(mp, QUOTACTL_CURSOROPEN, &args);
566 1.21 dholland if (error) {
567 1.21 dholland return error;
568 1.21 dholland }
569 1.5 dholland
570 1.27 dholland keys = malloc(loopmax * sizeof(keys[0]), M_TEMP, M_WAITOK);
571 1.27 dholland vals = malloc(loopmax * sizeof(vals[0]), M_TEMP, M_WAITOK);
572 1.23 dholland
573 1.30 dholland skipidtype = (q2type == QUOTA_IDTYPE_USER ?
574 1.30 dholland QUOTA_IDTYPE_GROUP : QUOTA_IDTYPE_USER);
575 1.30 dholland args.qc_type = QCT_CURSORSKIPIDTYPE;
576 1.30 dholland args.u.cursorskipidtype.qc_cursor = &cursor;
577 1.30 dholland args.u.cursorskipidtype.qc_idtype = skipidtype;
578 1.30 dholland error = VFS_QUOTACTL(mp, QUOTACTL_CURSORSKIPIDTYPE, &args);
579 1.30 dholland /* ignore if it fails */
580 1.30 dholland (void)error;
581 1.30 dholland
582 1.23 dholland replies = prop_array_create();
583 1.23 dholland if (replies == NULL) {
584 1.23 dholland error = ENOMEM;
585 1.23 dholland goto err;
586 1.23 dholland }
587 1.23 dholland
588 1.26 dholland thisreply = NULL;
589 1.26 dholland lastid = 0; /* value not actually referenced */
590 1.30 dholland atzero = 0;
591 1.27 dholland
592 1.27 dholland while (1) {
593 1.30 dholland args.qc_type = QCT_CURSORATEND;
594 1.30 dholland args.u.cursoratend.qc_cursor = &cursor;
595 1.30 dholland args.u.cursoratend.qc_ret = &atend;
596 1.30 dholland error = VFS_QUOTACTL(mp, QUOTACTL_CURSORATEND, &args);
597 1.30 dholland if (error) {
598 1.30 dholland goto err;
599 1.30 dholland }
600 1.30 dholland if (atend) {
601 1.30 dholland break;
602 1.30 dholland }
603 1.30 dholland
604 1.29 dholland args.qc_type = QCT_CURSORGET;
605 1.29 dholland args.u.cursorget.qc_cursor = &cursor;
606 1.29 dholland args.u.cursorget.qc_keys = keys;
607 1.29 dholland args.u.cursorget.qc_vals = vals;
608 1.29 dholland args.u.cursorget.qc_maxnum = loopmax;
609 1.29 dholland args.u.cursorget.qc_ret = &loopnum;
610 1.27 dholland
611 1.29 dholland error = VFS_QUOTACTL(mp, QUOTACTL_CURSORGET, &args);
612 1.30 dholland if (error == EDEADLK) {
613 1.30 dholland /*
614 1.30 dholland * transaction abort, start over
615 1.30 dholland */
616 1.30 dholland
617 1.30 dholland args.qc_type = QCT_CURSORREWIND;
618 1.30 dholland args.u.cursorrewind.qc_cursor = &cursor;
619 1.30 dholland error = VFS_QUOTACTL(mp, QUOTACTL_CURSORREWIND, &args);
620 1.30 dholland if (error) {
621 1.30 dholland goto err;
622 1.30 dholland }
623 1.30 dholland
624 1.30 dholland args.qc_type = QCT_CURSORSKIPIDTYPE;
625 1.30 dholland args.u.cursorskipidtype.qc_cursor = &cursor;
626 1.30 dholland args.u.cursorskipidtype.qc_idtype = skipidtype;
627 1.30 dholland error = VFS_QUOTACTL(mp, QUOTACTL_CURSORSKIPIDTYPE,
628 1.30 dholland &args);
629 1.30 dholland /* ignore if it fails */
630 1.30 dholland (void)error;
631 1.30 dholland
632 1.30 dholland prop_object_release(replies);
633 1.30 dholland replies = prop_array_create();
634 1.30 dholland if (replies == NULL) {
635 1.30 dholland error = ENOMEM;
636 1.30 dholland goto err;
637 1.30 dholland }
638 1.30 dholland
639 1.30 dholland thisreply = NULL;
640 1.30 dholland lastid = 0;
641 1.30 dholland atzero = 0;
642 1.30 dholland
643 1.30 dholland continue;
644 1.30 dholland }
645 1.26 dholland if (error) {
646 1.23 dholland goto err;
647 1.23 dholland }
648 1.27 dholland
649 1.27 dholland if (loopnum == 0) {
650 1.30 dholland /*
651 1.30 dholland * This is not supposed to happen. However,
652 1.30 dholland * allow a return of zero items once as long
653 1.30 dholland * as something happens (including an atend
654 1.30 dholland * indication) on the next pass. If it happens
655 1.30 dholland * twice, warn and assume end of iteration.
656 1.30 dholland */
657 1.30 dholland if (atzero) {
658 1.30 dholland printf("vfs_quotactl: zero items returned\n");
659 1.30 dholland break;
660 1.30 dholland }
661 1.30 dholland atzero = 1;
662 1.30 dholland } else {
663 1.30 dholland atzero = 0;
664 1.27 dholland }
665 1.27 dholland
666 1.27 dholland for (i = 0; i < loopnum; i++) {
667 1.27 dholland key = &keys[i];
668 1.27 dholland val = &vals[i];
669 1.27 dholland
670 1.28 dholland if (key->qk_idtype != q2type) {
671 1.28 dholland /* don't want this result */
672 1.28 dholland continue;
673 1.28 dholland }
674 1.28 dholland
675 1.27 dholland if (thisreply == NULL || key->qk_id != lastid) {
676 1.27 dholland lastid = key->qk_id;
677 1.27 dholland thisreply = vfs_quotactl_getall_makereply(key);
678 1.27 dholland if (thisreply == NULL) {
679 1.27 dholland error = ENOMEM;
680 1.27 dholland goto err;
681 1.27 dholland }
682 1.27 dholland /*
683 1.27 dholland * Note: while we release our reference to
684 1.27 dholland * thisreply here, we can (and do) continue to
685 1.27 dholland * use the pointer in the loop because the
686 1.27 dholland * copy attached to the replies array is not
687 1.27 dholland * going away.
688 1.27 dholland */
689 1.27 dholland if (!prop_array_add_and_rel(replies,
690 1.27 dholland thisreply)) {
691 1.27 dholland error = ENOMEM;
692 1.27 dholland goto err;
693 1.27 dholland }
694 1.27 dholland }
695 1.27 dholland
696 1.27 dholland error = vfs_quotactl_getall_addreply(thisreply,
697 1.27 dholland key, val);
698 1.27 dholland if (error) {
699 1.27 dholland goto err;
700 1.27 dholland }
701 1.27 dholland }
702 1.23 dholland }
703 1.23 dholland
704 1.23 dholland if (!prop_dictionary_set_and_rel(cmddict, "data", replies)) {
705 1.30 dholland replies = NULL;
706 1.23 dholland error = ENOMEM;
707 1.23 dholland goto err;
708 1.23 dholland }
709 1.30 dholland replies = NULL;
710 1.30 dholland error = 0;
711 1.23 dholland
712 1.23 dholland err:
713 1.27 dholland free(keys, M_TEMP);
714 1.27 dholland free(vals, M_TEMP);
715 1.21 dholland
716 1.30 dholland if (replies != NULL) {
717 1.30 dholland prop_object_release(replies);
718 1.30 dholland }
719 1.30 dholland
720 1.21 dholland args.qc_type = QCT_CURSORCLOSE;
721 1.21 dholland args.u.cursorclose.qc_cursor = &cursor;
722 1.21 dholland error2 = VFS_QUOTACTL(mp, QUOTACTL_CURSORCLOSE, &args);
723 1.21 dholland
724 1.21 dholland if (error) {
725 1.21 dholland return error;
726 1.21 dholland }
727 1.21 dholland error = error2;
728 1.21 dholland return error;
729 1.5 dholland }
730 1.5 dholland
731 1.5 dholland static int
732 1.5 dholland vfs_quotactl_clear(struct mount *mp,
733 1.5 dholland prop_dictionary_t cmddict, int q2type,
734 1.5 dholland prop_array_t datas)
735 1.5 dholland {
736 1.16 dholland prop_array_t replies;
737 1.16 dholland prop_object_iterator_t iter;
738 1.16 dholland prop_dictionary_t data;
739 1.16 dholland uint32_t id;
740 1.16 dholland int defaultq;
741 1.16 dholland const char *idstr;
742 1.20 dholland struct quotakey qk;
743 1.5 dholland struct vfs_quotactl_args args;
744 1.16 dholland int error;
745 1.16 dholland
746 1.16 dholland KASSERT(prop_object_type(cmddict) == PROP_TYPE_DICTIONARY);
747 1.16 dholland KASSERT(prop_object_type(datas) == PROP_TYPE_ARRAY);
748 1.16 dholland
749 1.16 dholland replies = prop_array_create();
750 1.16 dholland if (replies == NULL)
751 1.16 dholland return ENOMEM;
752 1.5 dholland
753 1.16 dholland iter = prop_array_iterator(datas);
754 1.16 dholland if (iter == NULL) {
755 1.16 dholland prop_object_release(replies);
756 1.16 dholland return ENOMEM;
757 1.16 dholland }
758 1.16 dholland
759 1.16 dholland while ((data = prop_object_iterator_next(iter)) != NULL) {
760 1.16 dholland if (!prop_dictionary_get_uint32(data, "id", &id)) {
761 1.16 dholland if (!prop_dictionary_get_cstring_nocopy(data, "id",
762 1.16 dholland &idstr))
763 1.16 dholland continue;
764 1.16 dholland if (strcmp(idstr, "default"))
765 1.16 dholland continue;
766 1.16 dholland id = 0;
767 1.16 dholland defaultq = 1;
768 1.16 dholland } else {
769 1.16 dholland defaultq = 0;
770 1.16 dholland }
771 1.16 dholland
772 1.20 dholland qk.qk_idtype = q2type;
773 1.20 dholland qk.qk_id = defaultq ? QUOTA_DEFAULTID : id;
774 1.20 dholland qk.qk_objtype = QUOTA_OBJTYPE_BLOCKS;
775 1.20 dholland
776 1.19 dholland args.qc_type = QCT_DELETE;
777 1.20 dholland args.u.delete.qc_key = &qk;
778 1.19 dholland error = VFS_QUOTACTL(mp, QUOTACTL_DELETE, &args);
779 1.18 dholland if (error) {
780 1.18 dholland goto err;
781 1.18 dholland }
782 1.18 dholland
783 1.20 dholland qk.qk_idtype = q2type;
784 1.20 dholland qk.qk_id = defaultq ? QUOTA_DEFAULTID : id;
785 1.20 dholland qk.qk_objtype = QUOTA_OBJTYPE_FILES;
786 1.20 dholland
787 1.19 dholland args.qc_type = QCT_DELETE;
788 1.20 dholland args.u.delete.qc_key = &qk;
789 1.19 dholland error = VFS_QUOTACTL(mp, QUOTACTL_DELETE, &args);
790 1.16 dholland if (error) {
791 1.16 dholland goto err;
792 1.16 dholland }
793 1.16 dholland }
794 1.16 dholland
795 1.16 dholland prop_object_iterator_release(iter);
796 1.16 dholland if (!prop_dictionary_set_and_rel(cmddict, "data", replies)) {
797 1.16 dholland error = ENOMEM;
798 1.16 dholland } else {
799 1.16 dholland error = 0;
800 1.16 dholland }
801 1.16 dholland return error;
802 1.16 dholland err:
803 1.16 dholland prop_object_iterator_release(iter);
804 1.16 dholland prop_object_release(replies);
805 1.16 dholland return error;
806 1.5 dholland }
807 1.5 dholland
808 1.5 dholland static int
809 1.4 dholland vfs_quotactl_cmd(struct mount *mp, prop_dictionary_t cmddict)
810 1.4 dholland {
811 1.4 dholland int error;
812 1.4 dholland const char *cmd, *type;
813 1.4 dholland prop_array_t datas;
814 1.4 dholland int q2type;
815 1.4 dholland
816 1.4 dholland if (!prop_dictionary_get_cstring_nocopy(cmddict, "command", &cmd))
817 1.4 dholland return EINVAL;
818 1.4 dholland if (!prop_dictionary_get_cstring_nocopy(cmddict, "type", &type))
819 1.4 dholland return EINVAL;
820 1.4 dholland
821 1.4 dholland if (!strcmp(type, QUOTADICT_CLASS_USER)) {
822 1.4 dholland q2type = QUOTA_CLASS_USER;
823 1.4 dholland } else if (!strcmp(type, QUOTADICT_CLASS_GROUP)) {
824 1.4 dholland q2type = QUOTA_CLASS_GROUP;
825 1.4 dholland } else {
826 1.4 dholland /* XXX this is a bad errno for this case */
827 1.4 dholland return EOPNOTSUPP;
828 1.4 dholland }
829 1.4 dholland
830 1.4 dholland datas = prop_dictionary_get(cmddict, "data");
831 1.4 dholland if (datas == NULL || prop_object_type(datas) != PROP_TYPE_ARRAY)
832 1.4 dholland return EINVAL;
833 1.4 dholland
834 1.4 dholland prop_object_retain(datas);
835 1.4 dholland prop_dictionary_remove(cmddict, "data"); /* prepare for return */
836 1.4 dholland
837 1.4 dholland if (strcmp(cmd, "get version") == 0) {
838 1.5 dholland error = vfs_quotactl_getversion(mp, cmddict, q2type, datas);
839 1.4 dholland } else if (strcmp(cmd, "quotaon") == 0) {
840 1.5 dholland error = vfs_quotactl_quotaon(mp, cmddict, q2type, datas);
841 1.4 dholland } else if (strcmp(cmd, "quotaoff") == 0) {
842 1.5 dholland error = vfs_quotactl_quotaoff(mp, cmddict, q2type, datas);
843 1.4 dholland } else if (strcmp(cmd, "get") == 0) {
844 1.5 dholland error = vfs_quotactl_get(mp, cmddict, q2type, datas);
845 1.4 dholland } else if (strcmp(cmd, "set") == 0) {
846 1.15 dholland error = vfs_quotactl_put(mp, cmddict, q2type, datas);
847 1.4 dholland } else if (strcmp(cmd, "getall") == 0) {
848 1.5 dholland error = vfs_quotactl_getall(mp, cmddict, q2type, datas);
849 1.4 dholland } else if (strcmp(cmd, "clear") == 0) {
850 1.5 dholland error = vfs_quotactl_clear(mp, cmddict, q2type, datas);
851 1.4 dholland } else {
852 1.4 dholland /* XXX this a bad errno for this case */
853 1.4 dholland error = EOPNOTSUPP;
854 1.4 dholland }
855 1.4 dholland
856 1.4 dholland error = (prop_dictionary_set_int8(cmddict, "return",
857 1.4 dholland error) ? 0 : ENOMEM);
858 1.4 dholland prop_object_release(datas);
859 1.4 dholland
860 1.4 dholland return error;
861 1.4 dholland }
862 1.4 dholland
863 1.2 dholland int
864 1.2 dholland vfs_quotactl(struct mount *mp, prop_dictionary_t dict)
865 1.2 dholland {
866 1.3 dholland prop_dictionary_t cmddict;
867 1.3 dholland prop_array_t commands;
868 1.3 dholland prop_object_iterator_t iter;
869 1.3 dholland int error;
870 1.3 dholland
871 1.3 dholland error = quota_get_cmds(dict, &commands);
872 1.3 dholland if (error) {
873 1.3 dholland return error;
874 1.3 dholland }
875 1.3 dholland
876 1.3 dholland iter = prop_array_iterator(commands);
877 1.3 dholland if (iter == NULL) {
878 1.3 dholland return ENOMEM;
879 1.3 dholland }
880 1.3 dholland
881 1.3 dholland while ((cmddict = prop_object_iterator_next(iter)) != NULL) {
882 1.3 dholland if (prop_object_type(cmddict) != PROP_TYPE_DICTIONARY) {
883 1.3 dholland /* XXX shouldn't this be an error? */
884 1.3 dholland continue;
885 1.3 dholland }
886 1.4 dholland error = vfs_quotactl_cmd(mp, cmddict);
887 1.3 dholland if (error) {
888 1.3 dholland break;
889 1.3 dholland }
890 1.3 dholland }
891 1.3 dholland prop_object_iterator_release(iter);
892 1.3 dholland return error;
893 1.2 dholland }
894