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
11static const char * const bch2_quota_types[] = {
12	"user",
13	"group",
14	"project",
15};
16
17static const char * const bch2_quota_counters[] = {
18	"space",
19	"inodes",
20};
21
22static 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
36static 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
57const 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
62int bch2_quota_invalid(struct bch_fs *c, struct bkey_s_c k,
63		       enum bch_validate_flags flags, struct printbuf *err)
64{
65	int ret = 0;
66
67	bkey_fsck_err_on(k.k->p.inode >= QTYP_NR, c, err,
68			 quota_type_invalid,
69			 "invalid quota type (%llu >= %u)",
70			 k.k->p.inode, QTYP_NR);
71fsck_err:
72	return ret;
73}
74
75void 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
94static 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
109static 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
127static 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
140static 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
154enum 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
163static 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
177struct quota_msgs {
178	u8		nr;
179	struct {
180		u8	qtype;
181		u8	msg;
182	}		m[QTYP_NR * Q_COUNTERS];
183};
184
185static 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
197static 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
209static 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
220static 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
276int 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;
306err:
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
315static 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
326int 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
370err:
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
379static 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
424void 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
432void 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
440static 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
460static 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
481static 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);
518advance:
519	bch2_btree_iter_set_pos(iter, bpos_nosnap_successor(iter->pos));
520	return 0;
521}
522
523int 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
549static 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);
589unlock:
590	mutex_unlock(&c->sb_lock);
591
592	return bch2_err_class(ret);
593}
594
595static 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
618static 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 */
669static 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 */
698static 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);
752unlock:
753	mutex_unlock(&c->sb_lock);
754
755	return bch2_err_class(ret);
756}
757
758/* Get/set individual quotas: */
759
760static 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
775static 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
794static 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;
814found:
815	mutex_unlock(&q->lock);
816	return bch2_err_class(ret);
817}
818
819static 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
851static 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
879const 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