]> git.proxmox.com Git - mirror_ubuntu-jammy-kernel.git/blame - fs/quota/quota_tree.c
UBUNTU: Ubuntu-5.15.0-39.42
[mirror_ubuntu-jammy-kernel.git] / fs / quota / quota_tree.c
CommitLineData
09c434b8 1// SPDX-License-Identifier: GPL-2.0-only
1ccd14b9
JK
2/*
3 * vfsv0 quota IO operations on file
4 */
5
6#include <linux/errno.h>
7#include <linux/fs.h>
8#include <linux/mount.h>
9#include <linux/dqblk_v2.h>
10#include <linux/kernel.h>
11#include <linux/init.h>
12#include <linux/module.h>
13#include <linux/slab.h>
14#include <linux/quotaops.h>
15
16#include <asm/byteorder.h>
17
18#include "quota_tree.h"
19
20MODULE_AUTHOR("Jan Kara");
21MODULE_DESCRIPTION("Quota trie support");
22MODULE_LICENSE("GPL");
23
24#define __QUOTA_QT_PARANOIA
25
0066373d 26static int __get_index(struct qtree_mem_dqinfo *info, qid_t id, int depth)
1ccd14b9
JK
27{
28 unsigned int epb = info->dqi_usable_bs >> 2;
29
30 depth = info->dqi_qtree_depth - depth - 1;
31 while (depth--)
32 id /= epb;
33 return id % epb;
34}
35
0066373d
JK
36static int get_index(struct qtree_mem_dqinfo *info, struct kqid qid, int depth)
37{
38 qid_t id = from_kqid(&init_user_ns, qid);
39
40 return __get_index(info, id, depth);
41}
42
1ccd14b9 43/* Number of entries in one blocks */
7a2435d8 44static int qtree_dqstr_in_blk(struct qtree_mem_dqinfo *info)
1ccd14b9
JK
45{
46 return (info->dqi_usable_bs - sizeof(struct qt_disk_dqdbheader))
47 / info->dqi_entry_size;
48}
49
7a2435d8 50static ssize_t read_blk(struct qtree_mem_dqinfo *info, uint blk, char *buf)
1ccd14b9
JK
51{
52 struct super_block *sb = info->dqi_sb;
53
54 memset(buf, 0, info->dqi_usable_bs);
d26ac1a8 55 return sb->s_op->quota_read(sb, info->dqi_type, buf,
10f04d40 56 info->dqi_usable_bs, (loff_t)blk << info->dqi_blocksize_bits);
1ccd14b9
JK
57}
58
7a2435d8 59static ssize_t write_blk(struct qtree_mem_dqinfo *info, uint blk, char *buf)
1ccd14b9
JK
60{
61 struct super_block *sb = info->dqi_sb;
1907131b 62 ssize_t ret;
1ccd14b9 63
1907131b 64 ret = sb->s_op->quota_write(sb, info->dqi_type, buf,
10f04d40 65 info->dqi_usable_bs, (loff_t)blk << info->dqi_blocksize_bits);
1907131b 66 if (ret != info->dqi_usable_bs) {
fb5ffb0e 67 quota_error(sb, "dquota write failed");
1907131b
JZ
68 if (ret >= 0)
69 ret = -EIO;
70 }
71 return ret;
1ccd14b9
JK
72}
73
74/* Remove empty block from list and return it */
75static int get_free_dqblk(struct qtree_mem_dqinfo *info)
76{
ce1b06c5 77 char *buf = kmalloc(info->dqi_usable_bs, GFP_NOFS);
1ccd14b9
JK
78 struct qt_disk_dqdbheader *dh = (struct qt_disk_dqdbheader *)buf;
79 int ret, blk;
80
81 if (!buf)
82 return -ENOMEM;
83 if (info->dqi_free_blk) {
84 blk = info->dqi_free_blk;
85 ret = read_blk(info, blk, buf);
86 if (ret < 0)
87 goto out_buf;
88 info->dqi_free_blk = le32_to_cpu(dh->dqdh_next_free);
89 }
90 else {
91 memset(buf, 0, info->dqi_usable_bs);
92 /* Assure block allocation... */
93 ret = write_blk(info, info->dqi_blocks, buf);
94 if (ret < 0)
95 goto out_buf;
96 blk = info->dqi_blocks++;
97 }
98 mark_info_dirty(info->dqi_sb, info->dqi_type);
99 ret = blk;
100out_buf:
d26ac1a8 101 kfree(buf);
1ccd14b9
JK
102 return ret;
103}
104
105/* Insert empty block to the list */
d26ac1a8 106static int put_free_dqblk(struct qtree_mem_dqinfo *info, char *buf, uint blk)
1ccd14b9
JK
107{
108 struct qt_disk_dqdbheader *dh = (struct qt_disk_dqdbheader *)buf;
109 int err;
110
111 dh->dqdh_next_free = cpu_to_le32(info->dqi_free_blk);
112 dh->dqdh_prev_free = cpu_to_le32(0);
113 dh->dqdh_entries = cpu_to_le16(0);
114 err = write_blk(info, blk, buf);
115 if (err < 0)
116 return err;
117 info->dqi_free_blk = blk;
118 mark_info_dirty(info->dqi_sb, info->dqi_type);
119 return 0;
120}
121
122/* Remove given block from the list of blocks with free entries */
268157ba
JK
123static int remove_free_dqentry(struct qtree_mem_dqinfo *info, char *buf,
124 uint blk)
1ccd14b9 125{
ce1b06c5 126 char *tmpbuf = kmalloc(info->dqi_usable_bs, GFP_NOFS);
1ccd14b9
JK
127 struct qt_disk_dqdbheader *dh = (struct qt_disk_dqdbheader *)buf;
128 uint nextblk = le32_to_cpu(dh->dqdh_next_free);
129 uint prevblk = le32_to_cpu(dh->dqdh_prev_free);
130 int err;
131
132 if (!tmpbuf)
133 return -ENOMEM;
134 if (nextblk) {
135 err = read_blk(info, nextblk, tmpbuf);
136 if (err < 0)
137 goto out_buf;
138 ((struct qt_disk_dqdbheader *)tmpbuf)->dqdh_prev_free =
139 dh->dqdh_prev_free;
140 err = write_blk(info, nextblk, tmpbuf);
141 if (err < 0)
142 goto out_buf;
143 }
144 if (prevblk) {
145 err = read_blk(info, prevblk, tmpbuf);
146 if (err < 0)
147 goto out_buf;
148 ((struct qt_disk_dqdbheader *)tmpbuf)->dqdh_next_free =
149 dh->dqdh_next_free;
150 err = write_blk(info, prevblk, tmpbuf);
151 if (err < 0)
152 goto out_buf;
153 } else {
154 info->dqi_free_entry = nextblk;
155 mark_info_dirty(info->dqi_sb, info->dqi_type);
156 }
d26ac1a8 157 kfree(tmpbuf);
1ccd14b9
JK
158 dh->dqdh_next_free = dh->dqdh_prev_free = cpu_to_le32(0);
159 /* No matter whether write succeeds block is out of list */
160 if (write_blk(info, blk, buf) < 0)
fb5ffb0e
JZ
161 quota_error(info->dqi_sb, "Can't write block (%u) "
162 "with free entries", blk);
1ccd14b9
JK
163 return 0;
164out_buf:
d26ac1a8 165 kfree(tmpbuf);
1ccd14b9
JK
166 return err;
167}
168
169/* Insert given block to the beginning of list with free entries */
268157ba
JK
170static int insert_free_dqentry(struct qtree_mem_dqinfo *info, char *buf,
171 uint blk)
1ccd14b9 172{
ce1b06c5 173 char *tmpbuf = kmalloc(info->dqi_usable_bs, GFP_NOFS);
1ccd14b9
JK
174 struct qt_disk_dqdbheader *dh = (struct qt_disk_dqdbheader *)buf;
175 int err;
176
177 if (!tmpbuf)
178 return -ENOMEM;
179 dh->dqdh_next_free = cpu_to_le32(info->dqi_free_entry);
180 dh->dqdh_prev_free = cpu_to_le32(0);
181 err = write_blk(info, blk, buf);
182 if (err < 0)
183 goto out_buf;
184 if (info->dqi_free_entry) {
185 err = read_blk(info, info->dqi_free_entry, tmpbuf);
186 if (err < 0)
187 goto out_buf;
188 ((struct qt_disk_dqdbheader *)tmpbuf)->dqdh_prev_free =
189 cpu_to_le32(blk);
190 err = write_blk(info, info->dqi_free_entry, tmpbuf);
191 if (err < 0)
192 goto out_buf;
193 }
d26ac1a8 194 kfree(tmpbuf);
1ccd14b9
JK
195 info->dqi_free_entry = blk;
196 mark_info_dirty(info->dqi_sb, info->dqi_type);
197 return 0;
198out_buf:
d26ac1a8 199 kfree(tmpbuf);
1ccd14b9
JK
200 return err;
201}
202
203/* Is the entry in the block free? */
204int qtree_entry_unused(struct qtree_mem_dqinfo *info, char *disk)
205{
206 int i;
207
208 for (i = 0; i < info->dqi_entry_size; i++)
209 if (disk[i])
210 return 0;
211 return 1;
212}
213EXPORT_SYMBOL(qtree_entry_unused);
214
215/* Find space for dquot */
216static uint find_free_dqentry(struct qtree_mem_dqinfo *info,
217 struct dquot *dquot, int *err)
218{
219 uint blk, i;
220 struct qt_disk_dqdbheader *dh;
ce1b06c5 221 char *buf = kmalloc(info->dqi_usable_bs, GFP_NOFS);
1ccd14b9
JK
222 char *ddquot;
223
224 *err = 0;
225 if (!buf) {
226 *err = -ENOMEM;
227 return 0;
228 }
229 dh = (struct qt_disk_dqdbheader *)buf;
230 if (info->dqi_free_entry) {
231 blk = info->dqi_free_entry;
232 *err = read_blk(info, blk, buf);
233 if (*err < 0)
234 goto out_buf;
235 } else {
236 blk = get_free_dqblk(info);
237 if ((int)blk < 0) {
238 *err = blk;
d26ac1a8 239 kfree(buf);
1ccd14b9
JK
240 return 0;
241 }
242 memset(buf, 0, info->dqi_usable_bs);
268157ba
JK
243 /* This is enough as the block is already zeroed and the entry
244 * list is empty... */
1ccd14b9 245 info->dqi_free_entry = blk;
4c376dca 246 mark_info_dirty(dquot->dq_sb, dquot->dq_id.type);
1ccd14b9
JK
247 }
248 /* Block will be full? */
249 if (le16_to_cpu(dh->dqdh_entries) + 1 >= qtree_dqstr_in_blk(info)) {
250 *err = remove_free_dqentry(info, buf, blk);
251 if (*err < 0) {
fb5ffb0e
JZ
252 quota_error(dquot->dq_sb, "Can't remove block (%u) "
253 "from entry free list", blk);
1ccd14b9
JK
254 goto out_buf;
255 }
256 }
257 le16_add_cpu(&dh->dqdh_entries, 1);
258 /* Find free structure in block */
268157ba
JK
259 ddquot = buf + sizeof(struct qt_disk_dqdbheader);
260 for (i = 0; i < qtree_dqstr_in_blk(info); i++) {
261 if (qtree_entry_unused(info, ddquot))
262 break;
263 ddquot += info->dqi_entry_size;
264 }
1ccd14b9
JK
265#ifdef __QUOTA_QT_PARANOIA
266 if (i == qtree_dqstr_in_blk(info)) {
fb5ffb0e 267 quota_error(dquot->dq_sb, "Data block full but it shouldn't");
1ccd14b9
JK
268 *err = -EIO;
269 goto out_buf;
270 }
271#endif
272 *err = write_blk(info, blk, buf);
273 if (*err < 0) {
fb5ffb0e
JZ
274 quota_error(dquot->dq_sb, "Can't write quota data block %u",
275 blk);
1ccd14b9
JK
276 goto out_buf;
277 }
10f04d40 278 dquot->dq_off = ((loff_t)blk << info->dqi_blocksize_bits) +
1ccd14b9
JK
279 sizeof(struct qt_disk_dqdbheader) +
280 i * info->dqi_entry_size;
d26ac1a8 281 kfree(buf);
1ccd14b9
JK
282 return blk;
283out_buf:
d26ac1a8 284 kfree(buf);
1ccd14b9
JK
285 return 0;
286}
287
288/* Insert reference to structure into the trie */
289static int do_insert_tree(struct qtree_mem_dqinfo *info, struct dquot *dquot,
290 uint *treeblk, int depth)
291{
ce1b06c5 292 char *buf = kmalloc(info->dqi_usable_bs, GFP_NOFS);
1ccd14b9
JK
293 int ret = 0, newson = 0, newact = 0;
294 __le32 *ref;
295 uint newblk;
296
297 if (!buf)
298 return -ENOMEM;
299 if (!*treeblk) {
300 ret = get_free_dqblk(info);
301 if (ret < 0)
302 goto out_buf;
303 *treeblk = ret;
304 memset(buf, 0, info->dqi_usable_bs);
305 newact = 1;
306 } else {
307 ret = read_blk(info, *treeblk, buf);
308 if (ret < 0) {
fb5ffb0e
JZ
309 quota_error(dquot->dq_sb, "Can't read tree quota "
310 "block %u", *treeblk);
1ccd14b9
JK
311 goto out_buf;
312 }
313 }
314 ref = (__le32 *)buf;
315 newblk = le32_to_cpu(ref[get_index(info, dquot->dq_id, depth)]);
316 if (!newblk)
317 newson = 1;
318 if (depth == info->dqi_qtree_depth - 1) {
319#ifdef __QUOTA_QT_PARANOIA
320 if (newblk) {
fb5ffb0e
JZ
321 quota_error(dquot->dq_sb, "Inserting already present "
322 "quota entry (block %u)",
323 le32_to_cpu(ref[get_index(info,
1ccd14b9
JK
324 dquot->dq_id, depth)]));
325 ret = -EIO;
326 goto out_buf;
327 }
328#endif
329 newblk = find_free_dqentry(info, dquot, &ret);
330 } else {
331 ret = do_insert_tree(info, dquot, &newblk, depth+1);
332 }
333 if (newson && ret >= 0) {
334 ref[get_index(info, dquot->dq_id, depth)] =
335 cpu_to_le32(newblk);
336 ret = write_blk(info, *treeblk, buf);
337 } else if (newact && ret < 0) {
338 put_free_dqblk(info, buf, *treeblk);
339 }
340out_buf:
d26ac1a8 341 kfree(buf);
1ccd14b9
JK
342 return ret;
343}
344
345/* Wrapper for inserting quota structure into tree */
346static inline int dq_insert_tree(struct qtree_mem_dqinfo *info,
347 struct dquot *dquot)
348{
349 int tmp = QT_TREEOFF;
69a25ee2
KK
350
351#ifdef __QUOTA_QT_PARANOIA
352 if (info->dqi_blocks <= QT_TREEOFF) {
353 quota_error(dquot->dq_sb, "Quota tree root isn't allocated!");
354 return -EIO;
355 }
356#endif
1ccd14b9
JK
357 return do_insert_tree(info, dquot, &tmp, 0);
358}
359
360/*
268157ba
JK
361 * We don't have to be afraid of deadlocks as we never have quotas on quota
362 * files...
1ccd14b9
JK
363 */
364int qtree_write_dquot(struct qtree_mem_dqinfo *info, struct dquot *dquot)
365{
4c376dca 366 int type = dquot->dq_id.type;
1ccd14b9
JK
367 struct super_block *sb = dquot->dq_sb;
368 ssize_t ret;
ce1b06c5 369 char *ddquot = kmalloc(info->dqi_entry_size, GFP_NOFS);
1ccd14b9
JK
370
371 if (!ddquot)
372 return -ENOMEM;
373
bc8230ee 374 /* dq_off is guarded by dqio_sem */
1ccd14b9
JK
375 if (!dquot->dq_off) {
376 ret = dq_insert_tree(info, dquot);
377 if (ret < 0) {
fb5ffb0e
JZ
378 quota_error(sb, "Error %zd occurred while creating "
379 "quota", ret);
d26ac1a8 380 kfree(ddquot);
1ccd14b9
JK
381 return ret;
382 }
383 }
7b9ca4c6 384 spin_lock(&dquot->dq_dqb_lock);
1ccd14b9 385 info->dqi_ops->mem2disk_dqblk(ddquot, dquot);
7b9ca4c6 386 spin_unlock(&dquot->dq_dqb_lock);
d26ac1a8
JK
387 ret = sb->s_op->quota_write(sb, type, ddquot, info->dqi_entry_size,
388 dquot->dq_off);
1ccd14b9 389 if (ret != info->dqi_entry_size) {
fb5ffb0e 390 quota_error(sb, "dquota write failed");
1ccd14b9
JK
391 if (ret >= 0)
392 ret = -ENOSPC;
393 } else {
394 ret = 0;
395 }
dde95888 396 dqstats_inc(DQST_WRITES);
d26ac1a8 397 kfree(ddquot);
1ccd14b9
JK
398
399 return ret;
400}
401EXPORT_SYMBOL(qtree_write_dquot);
402
403/* Free dquot entry in data block */
404static int free_dqentry(struct qtree_mem_dqinfo *info, struct dquot *dquot,
405 uint blk)
406{
407 struct qt_disk_dqdbheader *dh;
ce1b06c5 408 char *buf = kmalloc(info->dqi_usable_bs, GFP_NOFS);
1ccd14b9
JK
409 int ret = 0;
410
411 if (!buf)
412 return -ENOMEM;
413 if (dquot->dq_off >> info->dqi_blocksize_bits != blk) {
fb5ffb0e
JZ
414 quota_error(dquot->dq_sb, "Quota structure has offset to "
415 "other block (%u) than it should (%u)", blk,
416 (uint)(dquot->dq_off >> info->dqi_blocksize_bits));
aef5800a 417 ret = -EIO;
1ccd14b9
JK
418 goto out_buf;
419 }
420 ret = read_blk(info, blk, buf);
421 if (ret < 0) {
fb5ffb0e
JZ
422 quota_error(dquot->dq_sb, "Can't read quota data block %u",
423 blk);
1ccd14b9
JK
424 goto out_buf;
425 }
426 dh = (struct qt_disk_dqdbheader *)buf;
427 le16_add_cpu(&dh->dqdh_entries, -1);
428 if (!le16_to_cpu(dh->dqdh_entries)) { /* Block got free? */
429 ret = remove_free_dqentry(info, buf, blk);
430 if (ret >= 0)
431 ret = put_free_dqblk(info, buf, blk);
432 if (ret < 0) {
fb5ffb0e
JZ
433 quota_error(dquot->dq_sb, "Can't move quota data block "
434 "(%u) to free list", blk);
1ccd14b9
JK
435 goto out_buf;
436 }
437 } else {
438 memset(buf +
439 (dquot->dq_off & ((1 << info->dqi_blocksize_bits) - 1)),
440 0, info->dqi_entry_size);
441 if (le16_to_cpu(dh->dqdh_entries) ==
442 qtree_dqstr_in_blk(info) - 1) {
443 /* Insert will write block itself */
444 ret = insert_free_dqentry(info, buf, blk);
445 if (ret < 0) {
fb5ffb0e
JZ
446 quota_error(dquot->dq_sb, "Can't insert quota "
447 "data block (%u) to free entry list", blk);
1ccd14b9
JK
448 goto out_buf;
449 }
450 } else {
451 ret = write_blk(info, blk, buf);
452 if (ret < 0) {
fb5ffb0e
JZ
453 quota_error(dquot->dq_sb, "Can't write quota "
454 "data block %u", blk);
1ccd14b9
JK
455 goto out_buf;
456 }
457 }
458 }
459 dquot->dq_off = 0; /* Quota is now unattached */
460out_buf:
d26ac1a8 461 kfree(buf);
1ccd14b9
JK
462 return ret;
463}
464
465/* Remove reference to dquot from tree */
466static int remove_tree(struct qtree_mem_dqinfo *info, struct dquot *dquot,
467 uint *blk, int depth)
468{
ce1b06c5 469 char *buf = kmalloc(info->dqi_usable_bs, GFP_NOFS);
1ccd14b9
JK
470 int ret = 0;
471 uint newblk;
472 __le32 *ref = (__le32 *)buf;
473
474 if (!buf)
475 return -ENOMEM;
476 ret = read_blk(info, *blk, buf);
477 if (ret < 0) {
055adcbd
JP
478 quota_error(dquot->dq_sb, "Can't read quota data block %u",
479 *blk);
1ccd14b9
JK
480 goto out_buf;
481 }
482 newblk = le32_to_cpu(ref[get_index(info, dquot->dq_id, depth)]);
6c81fad0
ZY
483 if (newblk < QT_TREEOFF || newblk >= info->dqi_blocks) {
484 quota_error(dquot->dq_sb, "Getting block too big (%u >= %u)",
485 newblk, info->dqi_blocks);
486 ret = -EUCLEAN;
487 goto out_buf;
488 }
489
1ccd14b9
JK
490 if (depth == info->dqi_qtree_depth - 1) {
491 ret = free_dqentry(info, dquot, newblk);
492 newblk = 0;
493 } else {
494 ret = remove_tree(info, dquot, &newblk, depth+1);
495 }
496 if (ret >= 0 && !newblk) {
497 int i;
498 ref[get_index(info, dquot->dq_id, depth)] = cpu_to_le32(0);
499 /* Block got empty? */
d26ac1a8
JK
500 for (i = 0; i < (info->dqi_usable_bs >> 2) && !ref[i]; i++)
501 ;
1ccd14b9
JK
502 /* Don't put the root block into the free block list */
503 if (i == (info->dqi_usable_bs >> 2)
504 && *blk != QT_TREEOFF) {
505 put_free_dqblk(info, buf, *blk);
506 *blk = 0;
507 } else {
508 ret = write_blk(info, *blk, buf);
509 if (ret < 0)
055adcbd
JP
510 quota_error(dquot->dq_sb,
511 "Can't write quota tree block %u",
512 *blk);
1ccd14b9
JK
513 }
514 }
515out_buf:
d26ac1a8 516 kfree(buf);
1ccd14b9
JK
517 return ret;
518}
519
520/* Delete dquot from tree */
521int qtree_delete_dquot(struct qtree_mem_dqinfo *info, struct dquot *dquot)
522{
523 uint tmp = QT_TREEOFF;
524
525 if (!dquot->dq_off) /* Even not allocated? */
526 return 0;
527 return remove_tree(info, dquot, &tmp, 0);
528}
529EXPORT_SYMBOL(qtree_delete_dquot);
530
531/* Find entry in block */
532static loff_t find_block_dqentry(struct qtree_mem_dqinfo *info,
533 struct dquot *dquot, uint blk)
534{
ce1b06c5 535 char *buf = kmalloc(info->dqi_usable_bs, GFP_NOFS);
1ccd14b9
JK
536 loff_t ret = 0;
537 int i;
538 char *ddquot;
539
540 if (!buf)
541 return -ENOMEM;
542 ret = read_blk(info, blk, buf);
543 if (ret < 0) {
fb5ffb0e
JZ
544 quota_error(dquot->dq_sb, "Can't read quota tree "
545 "block %u", blk);
1ccd14b9
JK
546 goto out_buf;
547 }
268157ba
JK
548 ddquot = buf + sizeof(struct qt_disk_dqdbheader);
549 for (i = 0; i < qtree_dqstr_in_blk(info); i++) {
550 if (info->dqi_ops->is_id(ddquot, dquot))
551 break;
552 ddquot += info->dqi_entry_size;
553 }
1ccd14b9 554 if (i == qtree_dqstr_in_blk(info)) {
4c376dca
EB
555 quota_error(dquot->dq_sb,
556 "Quota for id %u referenced but not present",
557 from_kqid(&init_user_ns, dquot->dq_id));
1ccd14b9
JK
558 ret = -EIO;
559 goto out_buf;
560 } else {
10f04d40 561 ret = ((loff_t)blk << info->dqi_blocksize_bits) + sizeof(struct
1ccd14b9
JK
562 qt_disk_dqdbheader) + i * info->dqi_entry_size;
563 }
564out_buf:
d26ac1a8 565 kfree(buf);
1ccd14b9
JK
566 return ret;
567}
568
569/* Find entry for given id in the tree */
570static loff_t find_tree_dqentry(struct qtree_mem_dqinfo *info,
571 struct dquot *dquot, uint blk, int depth)
572{
ce1b06c5 573 char *buf = kmalloc(info->dqi_usable_bs, GFP_NOFS);
1ccd14b9
JK
574 loff_t ret = 0;
575 __le32 *ref = (__le32 *)buf;
576
577 if (!buf)
578 return -ENOMEM;
579 ret = read_blk(info, blk, buf);
580 if (ret < 0) {
fb5ffb0e
JZ
581 quota_error(dquot->dq_sb, "Can't read quota tree block %u",
582 blk);
1ccd14b9
JK
583 goto out_buf;
584 }
585 ret = 0;
586 blk = le32_to_cpu(ref[get_index(info, dquot->dq_id, depth)]);
587 if (!blk) /* No reference? */
588 goto out_buf;
6c81fad0
ZY
589 if (blk < QT_TREEOFF || blk >= info->dqi_blocks) {
590 quota_error(dquot->dq_sb, "Getting block too big (%u >= %u)",
591 blk, info->dqi_blocks);
592 ret = -EUCLEAN;
593 goto out_buf;
594 }
595
1ccd14b9
JK
596 if (depth < info->dqi_qtree_depth - 1)
597 ret = find_tree_dqentry(info, dquot, blk, depth+1);
598 else
599 ret = find_block_dqentry(info, dquot, blk);
600out_buf:
d26ac1a8 601 kfree(buf);
1ccd14b9
JK
602 return ret;
603}
604
605/* Find entry for given id in the tree - wrapper function */
606static inline loff_t find_dqentry(struct qtree_mem_dqinfo *info,
607 struct dquot *dquot)
608{
609 return find_tree_dqentry(info, dquot, QT_TREEOFF, 0);
610}
611
612int qtree_read_dquot(struct qtree_mem_dqinfo *info, struct dquot *dquot)
613{
4c376dca 614 int type = dquot->dq_id.type;
1ccd14b9
JK
615 struct super_block *sb = dquot->dq_sb;
616 loff_t offset;
d26ac1a8 617 char *ddquot;
1ccd14b9
JK
618 int ret = 0;
619
620#ifdef __QUOTA_QT_PARANOIA
621 /* Invalidated quota? */
622 if (!sb_dqopt(dquot->dq_sb)->files[type]) {
fb5ffb0e 623 quota_error(sb, "Quota invalidated while reading!");
1ccd14b9
JK
624 return -EIO;
625 }
626#endif
627 /* Do we know offset of the dquot entry in the quota file? */
628 if (!dquot->dq_off) {
629 offset = find_dqentry(info, dquot);
630 if (offset <= 0) { /* Entry not present? */
631 if (offset < 0)
4c376dca
EB
632 quota_error(sb,"Can't read quota structure "
633 "for id %u",
634 from_kqid(&init_user_ns,
635 dquot->dq_id));
1ccd14b9
JK
636 dquot->dq_off = 0;
637 set_bit(DQ_FAKE_B, &dquot->dq_flags);
638 memset(&dquot->dq_dqb, 0, sizeof(struct mem_dqblk));
639 ret = offset;
640 goto out;
641 }
642 dquot->dq_off = offset;
643 }
ce1b06c5 644 ddquot = kmalloc(info->dqi_entry_size, GFP_NOFS);
1ccd14b9
JK
645 if (!ddquot)
646 return -ENOMEM;
d26ac1a8
JK
647 ret = sb->s_op->quota_read(sb, type, ddquot, info->dqi_entry_size,
648 dquot->dq_off);
1ccd14b9
JK
649 if (ret != info->dqi_entry_size) {
650 if (ret >= 0)
651 ret = -EIO;
fb5ffb0e 652 quota_error(sb, "Error while reading quota structure for id %u",
4c376dca 653 from_kqid(&init_user_ns, dquot->dq_id));
1ccd14b9
JK
654 set_bit(DQ_FAKE_B, &dquot->dq_flags);
655 memset(&dquot->dq_dqb, 0, sizeof(struct mem_dqblk));
d26ac1a8 656 kfree(ddquot);
1ccd14b9
JK
657 goto out;
658 }
7b9ca4c6 659 spin_lock(&dquot->dq_dqb_lock);
1ccd14b9
JK
660 info->dqi_ops->disk2mem_dqblk(dquot, ddquot);
661 if (!dquot->dq_dqb.dqb_bhardlimit &&
662 !dquot->dq_dqb.dqb_bsoftlimit &&
663 !dquot->dq_dqb.dqb_ihardlimit &&
664 !dquot->dq_dqb.dqb_isoftlimit)
665 set_bit(DQ_FAKE_B, &dquot->dq_flags);
7b9ca4c6 666 spin_unlock(&dquot->dq_dqb_lock);
d26ac1a8 667 kfree(ddquot);
1ccd14b9 668out:
dde95888 669 dqstats_inc(DQST_READS);
1ccd14b9
JK
670 return ret;
671}
672EXPORT_SYMBOL(qtree_read_dquot);
673
674/* Check whether dquot should not be deleted. We know we are
675 * the only one operating on dquot (thanks to dq_lock) */
676int qtree_release_dquot(struct qtree_mem_dqinfo *info, struct dquot *dquot)
677{
268157ba
JK
678 if (test_bit(DQ_FAKE_B, &dquot->dq_flags) &&
679 !(dquot->dq_dqb.dqb_curinodes | dquot->dq_dqb.dqb_curspace))
1ccd14b9
JK
680 return qtree_delete_dquot(info, dquot);
681 return 0;
682}
683EXPORT_SYMBOL(qtree_release_dquot);
0066373d
JK
684
685static int find_next_id(struct qtree_mem_dqinfo *info, qid_t *id,
686 unsigned int blk, int depth)
687{
ce1b06c5 688 char *buf = kmalloc(info->dqi_usable_bs, GFP_NOFS);
0066373d
JK
689 __le32 *ref = (__le32 *)buf;
690 ssize_t ret;
691 unsigned int epb = info->dqi_usable_bs >> 2;
692 unsigned int level_inc = 1;
693 int i;
694
695 if (!buf)
696 return -ENOMEM;
697
698 for (i = depth; i < info->dqi_qtree_depth - 1; i++)
699 level_inc *= epb;
700
701 ret = read_blk(info, blk, buf);
702 if (ret < 0) {
703 quota_error(info->dqi_sb,
704 "Can't read quota tree block %u", blk);
705 goto out_buf;
706 }
707 for (i = __get_index(info, *id, depth); i < epb; i++) {
708 if (ref[i] == cpu_to_le32(0)) {
709 *id += level_inc;
710 continue;
711 }
712 if (depth == info->dqi_qtree_depth - 1) {
713 ret = 0;
714 goto out_buf;
715 }
716 ret = find_next_id(info, id, le32_to_cpu(ref[i]), depth + 1);
717 if (ret != -ENOENT)
718 break;
719 }
720 if (i == epb) {
721 ret = -ENOENT;
722 goto out_buf;
723 }
724out_buf:
725 kfree(buf);
726 return ret;
727}
728
729int qtree_get_next_id(struct qtree_mem_dqinfo *info, struct kqid *qid)
730{
731 qid_t id = from_kqid(&init_user_ns, *qid);
732 int ret;
733
734 ret = find_next_id(info, &id, QT_TREEOFF, 0);
735 if (ret < 0)
736 return ret;
737 *qid = make_kqid(&init_user_ns, qid->type, id);
738 return 0;
739}
740EXPORT_SYMBOL(qtree_get_next_id);