fs/bcachefs/quota.c

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