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