xref: /linux/fs/bcachefs/quota.c (revision 6e7fd890f1d6ac83805409e9c346240de2705584)
1 // SPDX-License-Identifier: GPL-2.0
2 #include "bcachefs.h"
3 #include "btree_update.h"
4 #include "errcode.h"
5 #include "error.h"
6 #include "inode.h"
7 #include "quota.h"
8 #include "snapshot.h"
9 #include "super-io.h"
10 
11 static const char * const bch2_quota_types[] = {
12 	"user",
13 	"group",
14 	"project",
15 };
16 
17 static const char * const bch2_quota_counters[] = {
18 	"space",
19 	"inodes",
20 };
21 
22 static int bch2_sb_quota_validate(struct bch_sb *sb, struct bch_sb_field *f,
23 				  enum bch_validate_flags flags, struct printbuf *err)
24 {
25 	struct bch_sb_field_quota *q = field_to_type(f, quota);
26 
27 	if (vstruct_bytes(&q->field) < sizeof(*q)) {
28 		prt_printf(err, "wrong size (got %zu should be %zu)",
29 		       vstruct_bytes(&q->field), sizeof(*q));
30 		return -BCH_ERR_invalid_sb_quota;
31 	}
32 
33 	return 0;
34 }
35 
36 static void bch2_sb_quota_to_text(struct printbuf *out, struct bch_sb *sb,
37 				  struct bch_sb_field *f)
38 {
39 	struct bch_sb_field_quota *q = field_to_type(f, quota);
40 	unsigned qtyp, counter;
41 
42 	for (qtyp = 0; qtyp < ARRAY_SIZE(q->q); qtyp++) {
43 		prt_printf(out, "%s: flags %llx",
44 		       bch2_quota_types[qtyp],
45 		       le64_to_cpu(q->q[qtyp].flags));
46 
47 		for (counter = 0; counter < Q_COUNTERS; counter++)
48 			prt_printf(out, " %s timelimit %u warnlimit %u",
49 			       bch2_quota_counters[counter],
50 			       le32_to_cpu(q->q[qtyp].c[counter].timelimit),
51 			       le32_to_cpu(q->q[qtyp].c[counter].warnlimit));
52 
53 		prt_newline(out);
54 	}
55 }
56 
57 const struct bch_sb_field_ops bch_sb_field_ops_quota = {
58 	.validate	= bch2_sb_quota_validate,
59 	.to_text	= bch2_sb_quota_to_text,
60 };
61 
62 int bch2_quota_validate(struct bch_fs *c, struct bkey_s_c k,
63 			enum bch_validate_flags flags)
64 {
65 	int ret = 0;
66 
67 	bkey_fsck_err_on(k.k->p.inode >= QTYP_NR,
68 			 c, quota_type_invalid,
69 			 "invalid quota type (%llu >= %u)",
70 			 k.k->p.inode, QTYP_NR);
71 fsck_err:
72 	return ret;
73 }
74 
75 void bch2_quota_to_text(struct printbuf *out, struct bch_fs *c,
76 			struct bkey_s_c k)
77 {
78 	struct bkey_s_c_quota dq = bkey_s_c_to_quota(k);
79 	unsigned i;
80 
81 	for (i = 0; i < Q_COUNTERS; i++)
82 		prt_printf(out, "%s hardlimit %llu softlimit %llu",
83 		       bch2_quota_counters[i],
84 		       le64_to_cpu(dq.v->c[i].hardlimit),
85 		       le64_to_cpu(dq.v->c[i].softlimit));
86 }
87 
88 #ifdef CONFIG_BCACHEFS_QUOTA
89 
90 #include <linux/cred.h>
91 #include <linux/fs.h>
92 #include <linux/quota.h>
93 
94 static void qc_info_to_text(struct printbuf *out, struct qc_info *i)
95 {
96 	printbuf_tabstops_reset(out);
97 	printbuf_tabstop_push(out, 20);
98 
99 	prt_printf(out, "i_fieldmask\t%x\n",		i->i_fieldmask);
100 	prt_printf(out, "i_flags\t%u\n",		i->i_flags);
101 	prt_printf(out, "i_spc_timelimit\t%u\n",	i->i_spc_timelimit);
102 	prt_printf(out, "i_ino_timelimit\t%u\n",	i->i_ino_timelimit);
103 	prt_printf(out, "i_rt_spc_timelimit\t%u\n",	i->i_rt_spc_timelimit);
104 	prt_printf(out, "i_spc_warnlimit\t%u\n",	i->i_spc_warnlimit);
105 	prt_printf(out, "i_ino_warnlimit\t%u\n",	i->i_ino_warnlimit);
106 	prt_printf(out, "i_rt_spc_warnlimit\t%u\n",	i->i_rt_spc_warnlimit);
107 }
108 
109 static void qc_dqblk_to_text(struct printbuf *out, struct qc_dqblk *q)
110 {
111 	printbuf_tabstops_reset(out);
112 	printbuf_tabstop_push(out, 20);
113 
114 	prt_printf(out, "d_fieldmask\t%x\n",		q->d_fieldmask);
115 	prt_printf(out, "d_spc_hardlimit\t%llu\n",	q->d_spc_hardlimit);
116 	prt_printf(out, "d_spc_softlimit\t%llu\n",	q->d_spc_softlimit);
117 	prt_printf(out, "d_ino_hardlimit\%llu\n",	q->d_ino_hardlimit);
118 	prt_printf(out, "d_ino_softlimit\t%llu\n",	q->d_ino_softlimit);
119 	prt_printf(out, "d_space\t%llu\n",		q->d_space);
120 	prt_printf(out, "d_ino_count\t%llu\n",		q->d_ino_count);
121 	prt_printf(out, "d_ino_timer\t%llu\n",		q->d_ino_timer);
122 	prt_printf(out, "d_spc_timer\t%llu\n",		q->d_spc_timer);
123 	prt_printf(out, "d_ino_warns\t%i\n",		q->d_ino_warns);
124 	prt_printf(out, "d_spc_warns\t%i\n",		q->d_spc_warns);
125 }
126 
127 static inline unsigned __next_qtype(unsigned i, unsigned qtypes)
128 {
129 	qtypes >>= i;
130 	return qtypes ? i + __ffs(qtypes) : QTYP_NR;
131 }
132 
133 #define for_each_set_qtype(_c, _i, _q, _qtypes)				\
134 	for (_i = 0;							\
135 	     (_i = __next_qtype(_i, _qtypes),				\
136 	      _q = &(_c)->quotas[_i],					\
137 	      _i < QTYP_NR);						\
138 	     _i++)
139 
140 static bool ignore_hardlimit(struct bch_memquota_type *q)
141 {
142 	if (capable(CAP_SYS_RESOURCE))
143 		return true;
144 #if 0
145 	struct mem_dqinfo *info = &sb_dqopt(dquot->dq_sb)->info[dquot->dq_id.type];
146 
147 	return capable(CAP_SYS_RESOURCE) &&
148 	       (info->dqi_format->qf_fmt_id != QFMT_VFS_OLD ||
149 		!(info->dqi_flags & DQF_ROOT_SQUASH));
150 #endif
151 	return false;
152 }
153 
154 enum quota_msg {
155 	SOFTWARN,	/* Softlimit reached */
156 	SOFTLONGWARN,	/* Grace time expired */
157 	HARDWARN,	/* Hardlimit reached */
158 
159 	HARDBELOW,	/* Usage got below inode hardlimit */
160 	SOFTBELOW,	/* Usage got below inode softlimit */
161 };
162 
163 static int quota_nl[][Q_COUNTERS] = {
164 	[HARDWARN][Q_SPC]	= QUOTA_NL_BHARDWARN,
165 	[SOFTLONGWARN][Q_SPC]	= QUOTA_NL_BSOFTLONGWARN,
166 	[SOFTWARN][Q_SPC]	= QUOTA_NL_BSOFTWARN,
167 	[HARDBELOW][Q_SPC]	= QUOTA_NL_BHARDBELOW,
168 	[SOFTBELOW][Q_SPC]	= QUOTA_NL_BSOFTBELOW,
169 
170 	[HARDWARN][Q_INO]	= QUOTA_NL_IHARDWARN,
171 	[SOFTLONGWARN][Q_INO]	= QUOTA_NL_ISOFTLONGWARN,
172 	[SOFTWARN][Q_INO]	= QUOTA_NL_ISOFTWARN,
173 	[HARDBELOW][Q_INO]	= QUOTA_NL_IHARDBELOW,
174 	[SOFTBELOW][Q_INO]	= QUOTA_NL_ISOFTBELOW,
175 };
176 
177 struct quota_msgs {
178 	u8		nr;
179 	struct {
180 		u8	qtype;
181 		u8	msg;
182 	}		m[QTYP_NR * Q_COUNTERS];
183 };
184 
185 static void prepare_msg(unsigned qtype,
186 			enum quota_counters counter,
187 			struct quota_msgs *msgs,
188 			enum quota_msg msg_type)
189 {
190 	BUG_ON(msgs->nr >= ARRAY_SIZE(msgs->m));
191 
192 	msgs->m[msgs->nr].qtype	= qtype;
193 	msgs->m[msgs->nr].msg	= quota_nl[msg_type][counter];
194 	msgs->nr++;
195 }
196 
197 static void prepare_warning(struct memquota_counter *qc,
198 			    unsigned qtype,
199 			    enum quota_counters counter,
200 			    struct quota_msgs *msgs,
201 			    enum quota_msg msg_type)
202 {
203 	if (qc->warning_issued & (1 << msg_type))
204 		return;
205 
206 	prepare_msg(qtype, counter, msgs, msg_type);
207 }
208 
209 static void flush_warnings(struct bch_qid qid,
210 			   struct super_block *sb,
211 			   struct quota_msgs *msgs)
212 {
213 	unsigned i;
214 
215 	for (i = 0; i < msgs->nr; i++)
216 		quota_send_warning(make_kqid(&init_user_ns, msgs->m[i].qtype, qid.q[i]),
217 				   sb->s_dev, msgs->m[i].msg);
218 }
219 
220 static int bch2_quota_check_limit(struct bch_fs *c,
221 				  unsigned qtype,
222 				  struct bch_memquota *mq,
223 				  struct quota_msgs *msgs,
224 				  enum quota_counters counter,
225 				  s64 v,
226 				  enum quota_acct_mode mode)
227 {
228 	struct bch_memquota_type *q = &c->quotas[qtype];
229 	struct memquota_counter *qc = &mq->c[counter];
230 	u64 n = qc->v + v;
231 
232 	BUG_ON((s64) n < 0);
233 
234 	if (mode == KEY_TYPE_QUOTA_NOCHECK)
235 		return 0;
236 
237 	if (v <= 0) {
238 		if (n < qc->hardlimit &&
239 		    (qc->warning_issued & (1 << HARDWARN))) {
240 			qc->warning_issued &= ~(1 << HARDWARN);
241 			prepare_msg(qtype, counter, msgs, HARDBELOW);
242 		}
243 
244 		if (n < qc->softlimit &&
245 		    (qc->warning_issued & (1 << SOFTWARN))) {
246 			qc->warning_issued &= ~(1 << SOFTWARN);
247 			prepare_msg(qtype, counter, msgs, SOFTBELOW);
248 		}
249 
250 		qc->warning_issued = 0;
251 		return 0;
252 	}
253 
254 	if (qc->hardlimit &&
255 	    qc->hardlimit < n &&
256 	    !ignore_hardlimit(q)) {
257 		prepare_warning(qc, qtype, counter, msgs, HARDWARN);
258 		return -EDQUOT;
259 	}
260 
261 	if (qc->softlimit &&
262 	    qc->softlimit < n) {
263 		if (qc->timer == 0) {
264 			qc->timer = ktime_get_real_seconds() + q->limits[counter].timelimit;
265 			prepare_warning(qc, qtype, counter, msgs, SOFTWARN);
266 		} else if (ktime_get_real_seconds() >= qc->timer &&
267 			   !ignore_hardlimit(q)) {
268 			prepare_warning(qc, qtype, counter, msgs, SOFTLONGWARN);
269 			return -EDQUOT;
270 		}
271 	}
272 
273 	return 0;
274 }
275 
276 int bch2_quota_acct(struct bch_fs *c, struct bch_qid qid,
277 		    enum quota_counters counter, s64 v,
278 		    enum quota_acct_mode mode)
279 {
280 	unsigned qtypes = enabled_qtypes(c);
281 	struct bch_memquota_type *q;
282 	struct bch_memquota *mq[QTYP_NR];
283 	struct quota_msgs msgs;
284 	unsigned i;
285 	int ret = 0;
286 
287 	memset(&msgs, 0, sizeof(msgs));
288 
289 	for_each_set_qtype(c, i, q, qtypes) {
290 		mq[i] = genradix_ptr_alloc(&q->table, qid.q[i], GFP_KERNEL);
291 		if (!mq[i])
292 			return -ENOMEM;
293 	}
294 
295 	for_each_set_qtype(c, i, q, qtypes)
296 		mutex_lock_nested(&q->lock, i);
297 
298 	for_each_set_qtype(c, i, q, qtypes) {
299 		ret = bch2_quota_check_limit(c, i, mq[i], &msgs, counter, v, mode);
300 		if (ret)
301 			goto err;
302 	}
303 
304 	for_each_set_qtype(c, i, q, qtypes)
305 		mq[i]->c[counter].v += v;
306 err:
307 	for_each_set_qtype(c, i, q, qtypes)
308 		mutex_unlock(&q->lock);
309 
310 	flush_warnings(qid, c->vfs_sb, &msgs);
311 
312 	return ret;
313 }
314 
315 static void __bch2_quota_transfer(struct bch_memquota *src_q,
316 				  struct bch_memquota *dst_q,
317 				  enum quota_counters counter, s64 v)
318 {
319 	BUG_ON(v > src_q->c[counter].v);
320 	BUG_ON(v + dst_q->c[counter].v < v);
321 
322 	src_q->c[counter].v -= v;
323 	dst_q->c[counter].v += v;
324 }
325 
326 int bch2_quota_transfer(struct bch_fs *c, unsigned qtypes,
327 			struct bch_qid dst,
328 			struct bch_qid src, u64 space,
329 			enum quota_acct_mode mode)
330 {
331 	struct bch_memquota_type *q;
332 	struct bch_memquota *src_q[3], *dst_q[3];
333 	struct quota_msgs msgs;
334 	unsigned i;
335 	int ret = 0;
336 
337 	qtypes &= enabled_qtypes(c);
338 
339 	memset(&msgs, 0, sizeof(msgs));
340 
341 	for_each_set_qtype(c, i, q, qtypes) {
342 		src_q[i] = genradix_ptr_alloc(&q->table, src.q[i], GFP_KERNEL);
343 		dst_q[i] = genradix_ptr_alloc(&q->table, dst.q[i], GFP_KERNEL);
344 		if (!src_q[i] || !dst_q[i])
345 			return -ENOMEM;
346 	}
347 
348 	for_each_set_qtype(c, i, q, qtypes)
349 		mutex_lock_nested(&q->lock, i);
350 
351 	for_each_set_qtype(c, i, q, qtypes) {
352 		ret = bch2_quota_check_limit(c, i, dst_q[i], &msgs, Q_SPC,
353 					     dst_q[i]->c[Q_SPC].v + space,
354 					     mode);
355 		if (ret)
356 			goto err;
357 
358 		ret = bch2_quota_check_limit(c, i, dst_q[i], &msgs, Q_INO,
359 					     dst_q[i]->c[Q_INO].v + 1,
360 					     mode);
361 		if (ret)
362 			goto err;
363 	}
364 
365 	for_each_set_qtype(c, i, q, qtypes) {
366 		__bch2_quota_transfer(src_q[i], dst_q[i], Q_SPC, space);
367 		__bch2_quota_transfer(src_q[i], dst_q[i], Q_INO, 1);
368 	}
369 
370 err:
371 	for_each_set_qtype(c, i, q, qtypes)
372 		mutex_unlock(&q->lock);
373 
374 	flush_warnings(dst, c->vfs_sb, &msgs);
375 
376 	return ret;
377 }
378 
379 static int __bch2_quota_set(struct bch_fs *c, struct bkey_s_c k,
380 			    struct qc_dqblk *qdq)
381 {
382 	struct bkey_s_c_quota dq;
383 	struct bch_memquota_type *q;
384 	struct bch_memquota *mq;
385 	unsigned i;
386 
387 	BUG_ON(k.k->p.inode >= QTYP_NR);
388 
389 	if (!((1U << k.k->p.inode) & enabled_qtypes(c)))
390 		return 0;
391 
392 	switch (k.k->type) {
393 	case KEY_TYPE_quota:
394 		dq = bkey_s_c_to_quota(k);
395 		q = &c->quotas[k.k->p.inode];
396 
397 		mutex_lock(&q->lock);
398 		mq = genradix_ptr_alloc(&q->table, k.k->p.offset, GFP_KERNEL);
399 		if (!mq) {
400 			mutex_unlock(&q->lock);
401 			return -ENOMEM;
402 		}
403 
404 		for (i = 0; i < Q_COUNTERS; i++) {
405 			mq->c[i].hardlimit = le64_to_cpu(dq.v->c[i].hardlimit);
406 			mq->c[i].softlimit = le64_to_cpu(dq.v->c[i].softlimit);
407 		}
408 
409 		if (qdq && qdq->d_fieldmask & QC_SPC_TIMER)
410 			mq->c[Q_SPC].timer	= qdq->d_spc_timer;
411 		if (qdq && qdq->d_fieldmask & QC_SPC_WARNS)
412 			mq->c[Q_SPC].warns	= qdq->d_spc_warns;
413 		if (qdq && qdq->d_fieldmask & QC_INO_TIMER)
414 			mq->c[Q_INO].timer	= qdq->d_ino_timer;
415 		if (qdq && qdq->d_fieldmask & QC_INO_WARNS)
416 			mq->c[Q_INO].warns	= qdq->d_ino_warns;
417 
418 		mutex_unlock(&q->lock);
419 	}
420 
421 	return 0;
422 }
423 
424 void bch2_fs_quota_exit(struct bch_fs *c)
425 {
426 	unsigned i;
427 
428 	for (i = 0; i < ARRAY_SIZE(c->quotas); i++)
429 		genradix_free(&c->quotas[i].table);
430 }
431 
432 void bch2_fs_quota_init(struct bch_fs *c)
433 {
434 	unsigned i;
435 
436 	for (i = 0; i < ARRAY_SIZE(c->quotas); i++)
437 		mutex_init(&c->quotas[i].lock);
438 }
439 
440 static struct bch_sb_field_quota *bch2_sb_get_or_create_quota(struct bch_sb_handle *sb)
441 {
442 	struct bch_sb_field_quota *sb_quota = bch2_sb_field_get(sb->sb, quota);
443 
444 	if (sb_quota)
445 		return sb_quota;
446 
447 	sb_quota = bch2_sb_field_resize(sb, quota, sizeof(*sb_quota) / sizeof(u64));
448 	if (sb_quota) {
449 		unsigned qtype, qc;
450 
451 		for (qtype = 0; qtype < QTYP_NR; qtype++)
452 			for (qc = 0; qc < Q_COUNTERS; qc++)
453 				sb_quota->q[qtype].c[qc].timelimit =
454 					cpu_to_le32(7 * 24 * 60 * 60);
455 	}
456 
457 	return sb_quota;
458 }
459 
460 static void bch2_sb_quota_read(struct bch_fs *c)
461 {
462 	struct bch_sb_field_quota *sb_quota;
463 	unsigned i, j;
464 
465 	sb_quota = bch2_sb_field_get(c->disk_sb.sb, quota);
466 	if (!sb_quota)
467 		return;
468 
469 	for (i = 0; i < QTYP_NR; i++) {
470 		struct bch_memquota_type *q = &c->quotas[i];
471 
472 		for (j = 0; j < Q_COUNTERS; j++) {
473 			q->limits[j].timelimit =
474 				le32_to_cpu(sb_quota->q[i].c[j].timelimit);
475 			q->limits[j].warnlimit =
476 				le32_to_cpu(sb_quota->q[i].c[j].warnlimit);
477 		}
478 	}
479 }
480 
481 static int bch2_fs_quota_read_inode(struct btree_trans *trans,
482 				    struct btree_iter *iter,
483 				    struct bkey_s_c k)
484 {
485 	struct bch_fs *c = trans->c;
486 	struct bch_inode_unpacked u;
487 	struct bch_snapshot_tree s_t;
488 	u32 tree = bch2_snapshot_tree(c, k.k->p.snapshot);
489 
490 	int ret = bch2_snapshot_tree_lookup(trans, tree, &s_t);
491 	bch2_fs_inconsistent_on(bch2_err_matches(ret, ENOENT), c,
492 			"%s: snapshot tree %u not found", __func__, tree);
493 	if (ret)
494 		return ret;
495 
496 	if (!s_t.master_subvol)
497 		goto advance;
498 
499 	ret = bch2_inode_find_by_inum_nowarn_trans(trans,
500 				(subvol_inum) {
501 					le32_to_cpu(s_t.master_subvol),
502 					k.k->p.offset,
503 				}, &u);
504 	/*
505 	 * Inode might be deleted in this snapshot - the easiest way to handle
506 	 * that is to just skip it here:
507 	 */
508 	if (bch2_err_matches(ret, ENOENT))
509 		goto advance;
510 
511 	if (ret)
512 		return ret;
513 
514 	bch2_quota_acct(c, bch_qid(&u), Q_SPC, u.bi_sectors,
515 			KEY_TYPE_QUOTA_NOCHECK);
516 	bch2_quota_acct(c, bch_qid(&u), Q_INO, 1,
517 			KEY_TYPE_QUOTA_NOCHECK);
518 advance:
519 	bch2_btree_iter_set_pos(iter, bpos_nosnap_successor(iter->pos));
520 	return 0;
521 }
522 
523 int bch2_fs_quota_read(struct bch_fs *c)
524 {
525 
526 	mutex_lock(&c->sb_lock);
527 	struct bch_sb_field_quota *sb_quota = bch2_sb_get_or_create_quota(&c->disk_sb);
528 	if (!sb_quota) {
529 		mutex_unlock(&c->sb_lock);
530 		return -BCH_ERR_ENOSPC_sb_quota;
531 	}
532 
533 	bch2_sb_quota_read(c);
534 	mutex_unlock(&c->sb_lock);
535 
536 	int ret = bch2_trans_run(c,
537 		for_each_btree_key(trans, iter, BTREE_ID_quotas, POS_MIN,
538 				   BTREE_ITER_prefetch, k,
539 			__bch2_quota_set(c, k, NULL)) ?:
540 		for_each_btree_key(trans, iter, BTREE_ID_inodes, POS_MIN,
541 				   BTREE_ITER_prefetch|BTREE_ITER_all_snapshots, k,
542 			bch2_fs_quota_read_inode(trans, &iter, k)));
543 	bch_err_fn(c, ret);
544 	return ret;
545 }
546 
547 /* Enable/disable/delete quotas for an entire filesystem: */
548 
549 static int bch2_quota_enable(struct super_block	*sb, unsigned uflags)
550 {
551 	struct bch_fs *c = sb->s_fs_info;
552 	struct bch_sb_field_quota *sb_quota;
553 	int ret = 0;
554 
555 	if (sb->s_flags & SB_RDONLY)
556 		return -EROFS;
557 
558 	/* Accounting must be enabled at mount time: */
559 	if (uflags & (FS_QUOTA_UDQ_ACCT|FS_QUOTA_GDQ_ACCT|FS_QUOTA_PDQ_ACCT))
560 		return -EINVAL;
561 
562 	/* Can't enable enforcement without accounting: */
563 	if ((uflags & FS_QUOTA_UDQ_ENFD) && !c->opts.usrquota)
564 		return -EINVAL;
565 
566 	if ((uflags & FS_QUOTA_GDQ_ENFD) && !c->opts.grpquota)
567 		return -EINVAL;
568 
569 	if (uflags & FS_QUOTA_PDQ_ENFD && !c->opts.prjquota)
570 		return -EINVAL;
571 
572 	mutex_lock(&c->sb_lock);
573 	sb_quota = bch2_sb_get_or_create_quota(&c->disk_sb);
574 	if (!sb_quota) {
575 		ret = -BCH_ERR_ENOSPC_sb_quota;
576 		goto unlock;
577 	}
578 
579 	if (uflags & FS_QUOTA_UDQ_ENFD)
580 		SET_BCH_SB_USRQUOTA(c->disk_sb.sb, true);
581 
582 	if (uflags & FS_QUOTA_GDQ_ENFD)
583 		SET_BCH_SB_GRPQUOTA(c->disk_sb.sb, true);
584 
585 	if (uflags & FS_QUOTA_PDQ_ENFD)
586 		SET_BCH_SB_PRJQUOTA(c->disk_sb.sb, true);
587 
588 	bch2_write_super(c);
589 unlock:
590 	mutex_unlock(&c->sb_lock);
591 
592 	return bch2_err_class(ret);
593 }
594 
595 static int bch2_quota_disable(struct super_block *sb, unsigned uflags)
596 {
597 	struct bch_fs *c = sb->s_fs_info;
598 
599 	if (sb->s_flags & SB_RDONLY)
600 		return -EROFS;
601 
602 	mutex_lock(&c->sb_lock);
603 	if (uflags & FS_QUOTA_UDQ_ENFD)
604 		SET_BCH_SB_USRQUOTA(c->disk_sb.sb, false);
605 
606 	if (uflags & FS_QUOTA_GDQ_ENFD)
607 		SET_BCH_SB_GRPQUOTA(c->disk_sb.sb, false);
608 
609 	if (uflags & FS_QUOTA_PDQ_ENFD)
610 		SET_BCH_SB_PRJQUOTA(c->disk_sb.sb, false);
611 
612 	bch2_write_super(c);
613 	mutex_unlock(&c->sb_lock);
614 
615 	return 0;
616 }
617 
618 static int bch2_quota_remove(struct super_block *sb, unsigned uflags)
619 {
620 	struct bch_fs *c = sb->s_fs_info;
621 	int ret;
622 
623 	if (sb->s_flags & SB_RDONLY)
624 		return -EROFS;
625 
626 	if (uflags & FS_USER_QUOTA) {
627 		if (c->opts.usrquota)
628 			return -EINVAL;
629 
630 		ret = bch2_btree_delete_range(c, BTREE_ID_quotas,
631 					      POS(QTYP_USR, 0),
632 					      POS(QTYP_USR, U64_MAX),
633 					      0, NULL);
634 		if (ret)
635 			return ret;
636 	}
637 
638 	if (uflags & FS_GROUP_QUOTA) {
639 		if (c->opts.grpquota)
640 			return -EINVAL;
641 
642 		ret = bch2_btree_delete_range(c, BTREE_ID_quotas,
643 					      POS(QTYP_GRP, 0),
644 					      POS(QTYP_GRP, U64_MAX),
645 					      0, NULL);
646 		if (ret)
647 			return ret;
648 	}
649 
650 	if (uflags & FS_PROJ_QUOTA) {
651 		if (c->opts.prjquota)
652 			return -EINVAL;
653 
654 		ret = bch2_btree_delete_range(c, BTREE_ID_quotas,
655 					      POS(QTYP_PRJ, 0),
656 					      POS(QTYP_PRJ, U64_MAX),
657 					      0, NULL);
658 		if (ret)
659 			return ret;
660 	}
661 
662 	return 0;
663 }
664 
665 /*
666  * Return quota status information, such as enforcements, quota file inode
667  * numbers etc.
668  */
669 static int bch2_quota_get_state(struct super_block *sb, struct qc_state *state)
670 {
671 	struct bch_fs *c = sb->s_fs_info;
672 	unsigned qtypes = enabled_qtypes(c);
673 	unsigned i;
674 
675 	memset(state, 0, sizeof(*state));
676 
677 	for (i = 0; i < QTYP_NR; i++) {
678 		state->s_state[i].flags |= QCI_SYSFILE;
679 
680 		if (!(qtypes & (1 << i)))
681 			continue;
682 
683 		state->s_state[i].flags |= QCI_ACCT_ENABLED;
684 
685 		state->s_state[i].spc_timelimit = c->quotas[i].limits[Q_SPC].timelimit;
686 		state->s_state[i].spc_warnlimit = c->quotas[i].limits[Q_SPC].warnlimit;
687 
688 		state->s_state[i].ino_timelimit = c->quotas[i].limits[Q_INO].timelimit;
689 		state->s_state[i].ino_warnlimit = c->quotas[i].limits[Q_INO].warnlimit;
690 	}
691 
692 	return 0;
693 }
694 
695 /*
696  * Adjust quota timers & warnings
697  */
698 static int bch2_quota_set_info(struct super_block *sb, int type,
699 			       struct qc_info *info)
700 {
701 	struct bch_fs *c = sb->s_fs_info;
702 	struct bch_sb_field_quota *sb_quota;
703 	int ret = 0;
704 
705 	if (0) {
706 		struct printbuf buf = PRINTBUF;
707 
708 		qc_info_to_text(&buf, info);
709 		pr_info("setting:\n%s", buf.buf);
710 		printbuf_exit(&buf);
711 	}
712 
713 	if (sb->s_flags & SB_RDONLY)
714 		return -EROFS;
715 
716 	if (type >= QTYP_NR)
717 		return -EINVAL;
718 
719 	if (!((1 << type) & enabled_qtypes(c)))
720 		return -ESRCH;
721 
722 	if (info->i_fieldmask &
723 	    ~(QC_SPC_TIMER|QC_INO_TIMER|QC_SPC_WARNS|QC_INO_WARNS))
724 		return -EINVAL;
725 
726 	mutex_lock(&c->sb_lock);
727 	sb_quota = bch2_sb_get_or_create_quota(&c->disk_sb);
728 	if (!sb_quota) {
729 		ret = -BCH_ERR_ENOSPC_sb_quota;
730 		goto unlock;
731 	}
732 
733 	if (info->i_fieldmask & QC_SPC_TIMER)
734 		sb_quota->q[type].c[Q_SPC].timelimit =
735 			cpu_to_le32(info->i_spc_timelimit);
736 
737 	if (info->i_fieldmask & QC_SPC_WARNS)
738 		sb_quota->q[type].c[Q_SPC].warnlimit =
739 			cpu_to_le32(info->i_spc_warnlimit);
740 
741 	if (info->i_fieldmask & QC_INO_TIMER)
742 		sb_quota->q[type].c[Q_INO].timelimit =
743 			cpu_to_le32(info->i_ino_timelimit);
744 
745 	if (info->i_fieldmask & QC_INO_WARNS)
746 		sb_quota->q[type].c[Q_INO].warnlimit =
747 			cpu_to_le32(info->i_ino_warnlimit);
748 
749 	bch2_sb_quota_read(c);
750 
751 	bch2_write_super(c);
752 unlock:
753 	mutex_unlock(&c->sb_lock);
754 
755 	return bch2_err_class(ret);
756 }
757 
758 /* Get/set individual quotas: */
759 
760 static void __bch2_quota_get(struct qc_dqblk *dst, struct bch_memquota *src)
761 {
762 	dst->d_space		= src->c[Q_SPC].v << 9;
763 	dst->d_spc_hardlimit	= src->c[Q_SPC].hardlimit << 9;
764 	dst->d_spc_softlimit	= src->c[Q_SPC].softlimit << 9;
765 	dst->d_spc_timer	= src->c[Q_SPC].timer;
766 	dst->d_spc_warns	= src->c[Q_SPC].warns;
767 
768 	dst->d_ino_count	= src->c[Q_INO].v;
769 	dst->d_ino_hardlimit	= src->c[Q_INO].hardlimit;
770 	dst->d_ino_softlimit	= src->c[Q_INO].softlimit;
771 	dst->d_ino_timer	= src->c[Q_INO].timer;
772 	dst->d_ino_warns	= src->c[Q_INO].warns;
773 }
774 
775 static int bch2_get_quota(struct super_block *sb, struct kqid kqid,
776 			  struct qc_dqblk *qdq)
777 {
778 	struct bch_fs *c		= sb->s_fs_info;
779 	struct bch_memquota_type *q	= &c->quotas[kqid.type];
780 	qid_t qid			= from_kqid(&init_user_ns, kqid);
781 	struct bch_memquota *mq;
782 
783 	memset(qdq, 0, sizeof(*qdq));
784 
785 	mutex_lock(&q->lock);
786 	mq = genradix_ptr(&q->table, qid);
787 	if (mq)
788 		__bch2_quota_get(qdq, mq);
789 	mutex_unlock(&q->lock);
790 
791 	return 0;
792 }
793 
794 static int bch2_get_next_quota(struct super_block *sb, struct kqid *kqid,
795 			       struct qc_dqblk *qdq)
796 {
797 	struct bch_fs *c		= sb->s_fs_info;
798 	struct bch_memquota_type *q	= &c->quotas[kqid->type];
799 	qid_t qid			= from_kqid(&init_user_ns, *kqid);
800 	struct genradix_iter iter;
801 	struct bch_memquota *mq;
802 	int ret = 0;
803 
804 	mutex_lock(&q->lock);
805 
806 	genradix_for_each_from(&q->table, iter, mq, qid)
807 		if (memcmp(mq, page_address(ZERO_PAGE(0)), sizeof(*mq))) {
808 			__bch2_quota_get(qdq, mq);
809 			*kqid = make_kqid(current_user_ns(), kqid->type, iter.pos);
810 			goto found;
811 		}
812 
813 	ret = -ENOENT;
814 found:
815 	mutex_unlock(&q->lock);
816 	return bch2_err_class(ret);
817 }
818 
819 static int bch2_set_quota_trans(struct btree_trans *trans,
820 				struct bkey_i_quota *new_quota,
821 				struct qc_dqblk *qdq)
822 {
823 	struct btree_iter iter;
824 	struct bkey_s_c k;
825 	int ret;
826 
827 	k = bch2_bkey_get_iter(trans, &iter, BTREE_ID_quotas, new_quota->k.p,
828 			       BTREE_ITER_slots|BTREE_ITER_intent);
829 	ret = bkey_err(k);
830 	if (unlikely(ret))
831 		return ret;
832 
833 	if (k.k->type == KEY_TYPE_quota)
834 		new_quota->v = *bkey_s_c_to_quota(k).v;
835 
836 	if (qdq->d_fieldmask & QC_SPC_SOFT)
837 		new_quota->v.c[Q_SPC].softlimit = cpu_to_le64(qdq->d_spc_softlimit >> 9);
838 	if (qdq->d_fieldmask & QC_SPC_HARD)
839 		new_quota->v.c[Q_SPC].hardlimit = cpu_to_le64(qdq->d_spc_hardlimit >> 9);
840 
841 	if (qdq->d_fieldmask & QC_INO_SOFT)
842 		new_quota->v.c[Q_INO].softlimit = cpu_to_le64(qdq->d_ino_softlimit);
843 	if (qdq->d_fieldmask & QC_INO_HARD)
844 		new_quota->v.c[Q_INO].hardlimit = cpu_to_le64(qdq->d_ino_hardlimit);
845 
846 	ret = bch2_trans_update(trans, &iter, &new_quota->k_i, 0);
847 	bch2_trans_iter_exit(trans, &iter);
848 	return ret;
849 }
850 
851 static int bch2_set_quota(struct super_block *sb, struct kqid qid,
852 			  struct qc_dqblk *qdq)
853 {
854 	struct bch_fs *c = sb->s_fs_info;
855 	struct bkey_i_quota new_quota;
856 	int ret;
857 
858 	if (0) {
859 		struct printbuf buf = PRINTBUF;
860 
861 		qc_dqblk_to_text(&buf, qdq);
862 		pr_info("setting:\n%s", buf.buf);
863 		printbuf_exit(&buf);
864 	}
865 
866 	if (sb->s_flags & SB_RDONLY)
867 		return -EROFS;
868 
869 	bkey_quota_init(&new_quota.k_i);
870 	new_quota.k.p = POS(qid.type, from_kqid(&init_user_ns, qid));
871 
872 	ret = bch2_trans_do(c, NULL, NULL, 0,
873 			    bch2_set_quota_trans(trans, &new_quota, qdq)) ?:
874 		__bch2_quota_set(c, bkey_i_to_s_c(&new_quota.k_i), qdq);
875 
876 	return bch2_err_class(ret);
877 }
878 
879 const struct quotactl_ops bch2_quotactl_operations = {
880 	.quota_enable		= bch2_quota_enable,
881 	.quota_disable		= bch2_quota_disable,
882 	.rm_xquota		= bch2_quota_remove,
883 
884 	.get_state		= bch2_quota_get_state,
885 	.set_info		= bch2_quota_set_info,
886 
887 	.get_dqblk		= bch2_get_quota,
888 	.get_nextdqblk		= bch2_get_next_quota,
889 	.set_dqblk		= bch2_set_quota,
890 };
891 
892 #endif /* CONFIG_BCACHEFS_QUOTA */
893