git.sesse.net Git - bcachefs-tools-debian/blob - libbcachefs/backpointers.c

   1 // SPDX-License-Identifier: GPL-2.0
   2 #include "bcachefs.h"
   3 #include "alloc_background.h"
   4 #include "backpointers.h"
   5 #include "btree_cache.h"
   6 #include "btree_update.h"
   7 #include "error.h"
   8
   9 #define MAX_EXTENT_COMPRESS_RATIO_SHIFT         10
  10
  11 /*
  12  * Convert from pos in backpointer btree to pos of corresponding bucket in alloc
  13  * btree:
  14  */
  15 static inline struct bpos bp_pos_to_bucket(const struct bch_fs *c,
  16                                            struct bpos bp_pos)
  17 {
  18         struct bch_dev *ca = bch_dev_bkey_exists(c, bp_pos.inode);
  19         u64 bucket_sector = bp_pos.offset >> MAX_EXTENT_COMPRESS_RATIO_SHIFT;
  20
  21         return POS(bp_pos.inode, sector_to_bucket(ca, bucket_sector));
  22 }
  23
  24 /*
  25  * Convert from pos in alloc btree + bucket offset to pos in backpointer btree:
  26  */
  27 static inline struct bpos bucket_pos_to_bp(const struct bch_fs *c,
  28                                            struct bpos bucket,
  29                                            u64 bucket_offset)
  30 {
  31         struct bch_dev *ca = bch_dev_bkey_exists(c, bucket.inode);
  32
  33         return POS(bucket.inode,
  34                    (bucket_to_sector(ca, bucket.offset) <<
  35                     MAX_EXTENT_COMPRESS_RATIO_SHIFT) + bucket_offset);
  36 }
  37
  38 void bch2_extent_ptr_to_bp(struct bch_fs *c,
  39                            enum btree_id btree_id, unsigned level,
  40                            struct bkey_s_c k, struct extent_ptr_decoded p,
  41                            struct bpos *bucket_pos, struct bch_backpointer *bp)
  42 {
  43         enum bch_data_type data_type = level ? BCH_DATA_btree : BCH_DATA_user;
  44         s64 sectors = level ? btree_sectors(c) : k.k->size;
  45         u32 bucket_offset;
  46
  47         *bucket_pos = PTR_BUCKET_POS_OFFSET(c, &p.ptr, &bucket_offset);
  48         *bp = (struct bch_backpointer) {
  49                 .btree_id       = btree_id,
  50                 .level          = level,
  51                 .data_type      = data_type,
  52                 .bucket_offset  = ((u64) bucket_offset << MAX_EXTENT_COMPRESS_RATIO_SHIFT) +
  53                         p.crc.offset,
  54                 .bucket_len     = ptr_disk_sectors(sectors, p),
  55                 .pos            = k.k->p,
  56         };
  57 }
  58
  59 static bool extent_matches_bp(struct bch_fs *c,
  60                               enum btree_id btree_id, unsigned level,
  61                               struct bkey_s_c k,
  62                               struct bpos bucket,
  63                               struct bch_backpointer bp)
  64 {
  65         struct bkey_ptrs_c ptrs = bch2_bkey_ptrs_c(k);
  66         const union bch_extent_entry *entry;
  67         struct extent_ptr_decoded p;
  68
  69         bkey_for_each_ptr_decode(k.k, ptrs, p, entry) {
  70                 struct bpos bucket2;
  71                 struct bch_backpointer bp2;
  72
  73                 if (p.ptr.cached)
  74                         continue;
  75
  76                 bch2_extent_ptr_to_bp(c, btree_id, level, k, p,
  77                                       &bucket2, &bp2);
  78                 if (!bpos_cmp(bucket, bucket2) &&
  79                     !memcmp(&bp, &bp2, sizeof(bp)))
  80                         return true;
  81         }
  82
  83         return false;
  84 }
  85
  86 int bch2_backpointer_invalid(const struct bch_fs *c, struct bkey_s_c k,
  87                              int rw, struct printbuf *err)
  88 {
  89         struct bkey_s_c_backpointer bp = bkey_s_c_to_backpointer(k);
  90         struct bpos bucket = bp_pos_to_bucket(c, bp.k->p);
  91
  92         if (bkey_val_bytes(bp.k) < sizeof(*bp.v)) {
  93                 prt_str(err, "incorrect value size");
  94                 return -EINVAL;
  95         }
  96
  97         if (bpos_cmp(bp.k->p, bucket_pos_to_bp(c, bucket, bp.v->bucket_offset))) {
  98                 prt_str(err, "backpointer at wrong pos");
  99                 return -EINVAL;
 100         }
 101
 102         return 0;
 103 }
 104
 105 void bch2_backpointer_to_text(struct printbuf *out, const struct bch_backpointer *bp)
 106 {
 107         prt_printf(out, "btree=%s l=%u offset=%llu:%u len=%u pos=",
 108                bch2_btree_ids[bp->btree_id],
 109                bp->level,
 110                (u64) (bp->bucket_offset >> MAX_EXTENT_COMPRESS_RATIO_SHIFT),
 111                (u32) bp->bucket_offset & ~(~0U << MAX_EXTENT_COMPRESS_RATIO_SHIFT),
 112                bp->bucket_len);
 113         bch2_bpos_to_text(out, bp->pos);
 114 }
 115
 116 void bch2_backpointer_k_to_text(struct printbuf *out, struct bch_fs *c, struct bkey_s_c k)
 117 {
 118         bch2_backpointer_to_text(out, bkey_s_c_to_backpointer(k).v);
 119 }
 120
 121 void bch2_backpointer_swab(struct bkey_s k)
 122 {
 123         struct bkey_s_backpointer bp = bkey_s_to_backpointer(k);
 124
 125         bp.v->bucket_offset     = swab32(bp.v->bucket_offset);
 126         bp.v->bucket_len        = swab32(bp.v->bucket_len);
 127         bch2_bpos_swab(&bp.v->pos);
 128 }
 129
 130 #define BACKPOINTER_OFFSET_MAX  ((1ULL << 40) - 1)
 131
 132 static inline int backpointer_cmp(struct bch_backpointer l, struct bch_backpointer r)
 133 {
 134         return cmp_int(l.bucket_offset, r.bucket_offset);
 135 }
 136
 137 static int bch2_backpointer_del_by_offset(struct btree_trans *trans,
 138                                           struct bpos bucket,
 139                                           u64 bp_offset,
 140                                           struct bch_backpointer bp)
 141 {
 142         struct bch_fs *c = trans->c;
 143         struct btree_iter iter;
 144         struct bkey_s_c k;
 145         int ret;
 146
 147         if (bp_offset < BACKPOINTER_OFFSET_MAX) {
 148                 struct bch_backpointer *bps;
 149                 struct bkey_i_alloc_v4 *a;
 150                 unsigned i, nr;
 151
 152                 bch2_trans_iter_init(trans, &iter, BTREE_ID_alloc,
 153                                      bucket,
 154                                      BTREE_ITER_INTENT|
 155                                      BTREE_ITER_SLOTS|
 156                                      BTREE_ITER_WITH_UPDATES);
 157                 k = bch2_btree_iter_peek_slot(&iter);
 158                 ret = bkey_err(k);
 159                 if (ret)
 160                         goto err;
 161
 162                 if (k.k->type != KEY_TYPE_alloc_v4) {
 163                         ret = -ENOENT;
 164                         goto err;
 165                 }
 166
 167                 a = bch2_alloc_to_v4_mut(trans, k);
 168                 ret = PTR_ERR_OR_ZERO(a);
 169                 if (ret)
 170                         goto err;
 171                 bps = alloc_v4_backpointers(&a->v);
 172                 nr = BCH_ALLOC_V4_NR_BACKPOINTERS(&a->v);
 173
 174                 for (i = 0; i < nr; i++) {
 175                         if (bps[i].bucket_offset == bp_offset)
 176                                 goto found;
 177                         if (bps[i].bucket_offset > bp_offset)
 178                                 break;
 179                 }
 180
 181                 ret = -ENOENT;
 182                 goto err;
 183 found:
 184                 if (memcmp(&bps[i], &bp, sizeof(bp))) {
 185                         ret = -ENOENT;
 186                         goto err;
 187                 }
 188                 array_remove_item(bps, nr, i);
 189                 SET_BCH_ALLOC_V4_NR_BACKPOINTERS(&a->v, nr);
 190                 set_alloc_v4_u64s(a);
 191                 ret = bch2_trans_update(trans, &iter, &a->k_i, 0);
 192         } else {
 193                 bp_offset -= BACKPOINTER_OFFSET_MAX;
 194
 195                 bch2_trans_iter_init(trans, &iter, BTREE_ID_backpointers,
 196                                      bucket_pos_to_bp(c, bucket, bp_offset),
 197                                      BTREE_ITER_INTENT|
 198                                      BTREE_ITER_SLOTS|
 199                                      BTREE_ITER_WITH_UPDATES);
 200                 k = bch2_btree_iter_peek_slot(&iter);
 201                 ret = bkey_err(k);
 202                 if (ret)
 203                         goto err;
 204
 205                 if (k.k->type != KEY_TYPE_backpointer ||
 206                     memcmp(bkey_s_c_to_backpointer(k).v, &bp, sizeof(bp))) {
 207                         ret = -ENOENT;
 208                         goto err;
 209                 }
 210
 211                 ret = bch2_btree_delete_at(trans, &iter, 0);
 212         }
 213 err:
 214         bch2_trans_iter_exit(trans, &iter);
 215         return ret;
 216 }
 217
 218 int bch2_bucket_backpointer_del(struct btree_trans *trans,
 219                                 struct bkey_i_alloc_v4 *a,
 220                                 struct bch_backpointer bp,
 221                                 struct bkey_s_c orig_k)
 222 {
 223         struct bch_fs *c = trans->c;
 224         struct bch_backpointer *bps = alloc_v4_backpointers(&a->v);
 225         unsigned i, nr = BCH_ALLOC_V4_NR_BACKPOINTERS(&a->v);
 226         struct btree_iter bp_iter;
 227         struct bkey_s_c k;
 228         int ret;
 229
 230         for (i = 0; i < nr; i++) {
 231                 int cmp = backpointer_cmp(bps[i], bp) ?:
 232                         memcmp(&bps[i], &bp, sizeof(bp));
 233                 if (!cmp)
 234                         goto found;
 235                 if (cmp >= 0)
 236                         break;
 237         }
 238
 239         goto btree;
 240 found:
 241         array_remove_item(bps, nr, i);
 242         SET_BCH_ALLOC_V4_NR_BACKPOINTERS(&a->v, nr);
 243         set_alloc_v4_u64s(a);
 244         return 0;
 245 btree:
 246         bch2_trans_iter_init(trans, &bp_iter, BTREE_ID_backpointers,
 247                              bucket_pos_to_bp(c, a->k.p, bp.bucket_offset),
 248                              BTREE_ITER_INTENT|
 249                              BTREE_ITER_SLOTS|
 250                              BTREE_ITER_WITH_UPDATES);
 251         k = bch2_btree_iter_peek_slot(&bp_iter);
 252         ret = bkey_err(k);
 253         if (ret)
 254                 goto err;
 255
 256         if (k.k->type != KEY_TYPE_backpointer ||
 257             memcmp(bkey_s_c_to_backpointer(k).v, &bp, sizeof(bp))) {
 258                 struct printbuf buf = PRINTBUF;
 259
 260                 prt_printf(&buf, "backpointer not found when deleting");
 261                 prt_newline(&buf);
 262                 printbuf_indent_add(&buf, 2);
 263
 264                 prt_printf(&buf, "searching for ");
 265                 bch2_backpointer_to_text(&buf, &bp);
 266                 prt_newline(&buf);
 267
 268                 prt_printf(&buf, "got ");
 269                 bch2_bkey_val_to_text(&buf, c, k);
 270                 prt_newline(&buf);
 271
 272                 prt_str(&buf, "alloc ");
 273                 bch2_bkey_val_to_text(&buf, c, bkey_i_to_s_c(&a->k_i));
 274                 prt_newline(&buf);
 275
 276                 prt_printf(&buf, "for ");
 277                 bch2_bkey_val_to_text(&buf, c, orig_k);
 278
 279                 if (!test_bit(BCH_FS_CHECK_BACKPOINTERS_DONE, &c->flags)) {
 280                         bch_err(c, "%s", buf.buf);
 281                 } else {
 282                         ret = -EIO;
 283                         bch2_trans_inconsistent(trans, "%s", buf.buf);
 284                 }
 285                 printbuf_exit(&buf);
 286                 goto err;
 287         }
 288
 289         ret = bch2_btree_delete_at(trans, &bp_iter, 0);
 290 err:
 291         bch2_trans_iter_exit(trans, &bp_iter);
 292         return ret;
 293 }
 294
 295 int bch2_bucket_backpointer_add(struct btree_trans *trans,
 296                                 struct bkey_i_alloc_v4 *a,
 297                                 struct bch_backpointer bp,
 298                                 struct bkey_s_c orig_k)
 299 {
 300         struct bch_fs *c = trans->c;
 301         struct bch_dev *ca;
 302         struct bch_backpointer *bps = alloc_v4_backpointers(&a->v);
 303         unsigned i, nr = BCH_ALLOC_V4_NR_BACKPOINTERS(&a->v);
 304         struct bkey_i_backpointer *bp_k;
 305         struct btree_iter bp_iter;
 306         struct bkey_s_c k;
 307         int ret;
 308
 309         /* Check for duplicates: */
 310         for (i = 0; i < nr; i++) {
 311                 int cmp = backpointer_cmp(bps[i], bp);
 312                 if (cmp >= 0)
 313                         break;
 314         }
 315
 316         if ((i &&
 317              (bps[i - 1].bucket_offset +
 318               bps[i - 1].bucket_len > bp.bucket_offset)) ||
 319             (i < nr &&
 320              (bp.bucket_offset + bp.bucket_len > bps[i].bucket_offset))) {
 321                 struct printbuf buf = PRINTBUF;
 322
 323                 prt_printf(&buf, "overlapping backpointer found when inserting ");
 324                 bch2_backpointer_to_text(&buf, &bp);
 325                 prt_newline(&buf);
 326                 printbuf_indent_add(&buf, 2);
 327
 328                 prt_printf(&buf, "into ");
 329                 bch2_bkey_val_to_text(&buf, c, bkey_i_to_s_c(&a->k_i));
 330                 prt_newline(&buf);
 331
 332                 prt_printf(&buf, "for ");
 333                 bch2_bkey_val_to_text(&buf, c, orig_k);
 334
 335                 if (!test_bit(BCH_FS_CHECK_BACKPOINTERS_DONE, &c->flags))
 336                         bch_err(c, "%s", buf.buf);
 337                 else {
 338                         bch2_trans_inconsistent(trans, "%s", buf.buf);
 339                         printbuf_exit(&buf);
 340                         return -EIO;
 341                 }
 342         }
 343
 344         if (nr < BCH_ALLOC_V4_NR_BACKPOINTERS_MAX) {
 345                 array_insert_item(bps, nr, i, bp);
 346                 SET_BCH_ALLOC_V4_NR_BACKPOINTERS(&a->v, nr);
 347                 set_alloc_v4_u64s(a);
 348                 return 0;
 349         }
 350
 351         /* Overflow: use backpointer btree */
 352         bp_k = bch2_trans_kmalloc(trans, sizeof(*bp_k));
 353         ret = PTR_ERR_OR_ZERO(bp_k);
 354         if (ret)
 355                 return ret;
 356
 357         ca = bch_dev_bkey_exists(c, a->k.p.inode);
 358
 359         bkey_backpointer_init(&bp_k->k_i);
 360         bp_k->k.p = bucket_pos_to_bp(c, a->k.p, bp.bucket_offset);
 361         bp_k->v = bp;
 362
 363         bch2_trans_iter_init(trans, &bp_iter, BTREE_ID_backpointers, bp_k->k.p,
 364                              BTREE_ITER_INTENT|
 365                              BTREE_ITER_SLOTS|
 366                              BTREE_ITER_WITH_UPDATES);
 367         k = bch2_btree_iter_peek_slot(&bp_iter);
 368         ret = bkey_err(k);
 369         if (ret)
 370                 goto err;
 371
 372         if (k.k->type) {
 373                 struct printbuf buf = PRINTBUF;
 374
 375                 prt_printf(&buf, "existing btree backpointer key found when inserting ");
 376                 bch2_backpointer_to_text(&buf, &bp);
 377                 prt_newline(&buf);
 378                 printbuf_indent_add(&buf, 2);
 379
 380                 prt_printf(&buf, "found ");
 381                 bch2_bkey_val_to_text(&buf, c, k);
 382                 prt_newline(&buf);
 383
 384                 prt_printf(&buf, "for ");
 385                 bch2_bkey_val_to_text(&buf, c, orig_k);
 386
 387                 if (!test_bit(BCH_FS_CHECK_BACKPOINTERS_DONE, &c->flags))
 388                         bch_err(c, "%s", buf.buf);
 389                 else {
 390                         bch2_trans_inconsistent(trans, "%s", buf.buf);
 391                         printbuf_exit(&buf);
 392                         ret = -EIO;
 393                         goto err;
 394                 }
 395         }
 396
 397         ret = bch2_trans_update(trans, &bp_iter, &bp_k->k_i, 0);
 398 err:
 399         bch2_trans_iter_exit(trans, &bp_iter);
 400         return ret;
 401 }
 402
 403 /*
 404  * Find the next backpointer >= *bp_offset:
 405  */
 406 int bch2_get_next_backpointer(struct btree_trans *trans,
 407                               struct bpos bucket, int gen,
 408                               u64 *bp_offset,
 409                               struct bch_backpointer *dst)
 410 {
 411         struct bch_fs *c = trans->c;
 412         struct bpos bp_pos =
 413                 bucket_pos_to_bp(c, bucket,
 414                                 max(*bp_offset, BACKPOINTER_OFFSET_MAX) - BACKPOINTER_OFFSET_MAX);
 415         struct bpos bp_end_pos =
 416                 bucket_pos_to_bp(c, bpos_nosnap_successor(bucket), 0);
 417         struct btree_iter alloc_iter, bp_iter = { NULL };
 418         struct bkey_s_c k;
 419         struct bkey_s_c_alloc_v4 a;
 420         size_t i;
 421         int ret;
 422
 423         bch2_trans_iter_init(trans, &alloc_iter, BTREE_ID_alloc,
 424                              bucket, BTREE_ITER_CACHED);
 425         k = bch2_btree_iter_peek_slot(&alloc_iter);
 426         ret = bkey_err(k);
 427         if (ret)
 428                 goto out;
 429
 430         if (k.k->type != KEY_TYPE_alloc_v4)
 431                 goto done;
 432
 433         a = bkey_s_c_to_alloc_v4(k);
 434         if (gen >= 0 && a.v->gen != gen)
 435                 goto done;
 436
 437         for (i = 0; i < BCH_ALLOC_V4_NR_BACKPOINTERS(a.v); i++) {
 438                 if (alloc_v4_backpointers_c(a.v)[i].bucket_offset < *bp_offset)
 439                         continue;
 440
 441                 *dst = alloc_v4_backpointers_c(a.v)[i];
 442                 *bp_offset = dst->bucket_offset;
 443                 goto out;
 444         }
 445
 446         for_each_btree_key_norestart(trans, bp_iter, BTREE_ID_backpointers,
 447                                      bp_pos, 0, k, ret) {
 448                 if (bpos_cmp(k.k->p, bp_end_pos) >= 0)
 449                         break;
 450
 451                 if (k.k->type != KEY_TYPE_backpointer)
 452                         continue;
 453
 454                 *dst = *bkey_s_c_to_backpointer(k).v;
 455                 *bp_offset = dst->bucket_offset + BACKPOINTER_OFFSET_MAX;
 456                 goto out;
 457         }
 458 done:
 459         *bp_offset = U64_MAX;
 460 out:
 461         bch2_trans_iter_exit(trans, &bp_iter);
 462         bch2_trans_iter_exit(trans, &alloc_iter);
 463         return ret;
 464 }
 465
 466 static void backpointer_not_found(struct btree_trans *trans,
 467                                   struct bpos bucket,
 468                                   u64 bp_offset,
 469                                   struct bch_backpointer bp,
 470                                   struct bkey_s_c k,
 471                                   const char *thing_it_points_to)
 472 {
 473         struct bch_fs *c = trans->c;
 474         struct printbuf buf = PRINTBUF;
 475
 476         prt_printf(&buf, "backpointer doesn't match %s it points to:\n  ",
 477                    thing_it_points_to);
 478         prt_printf(&buf, "bucket: ");
 479         bch2_bpos_to_text(&buf, bucket);
 480         prt_printf(&buf, "\n  ");
 481
 482         if (bp_offset >= BACKPOINTER_OFFSET_MAX) {
 483                 struct bpos bp_pos =
 484                         bucket_pos_to_bp(c, bucket,
 485                                         bp_offset - BACKPOINTER_OFFSET_MAX);
 486                 prt_printf(&buf, "backpointer pos: ");
 487                 bch2_bpos_to_text(&buf, bp_pos);
 488                 prt_printf(&buf, "\n  ");
 489         }
 490
 491         bch2_backpointer_to_text(&buf, &bp);
 492         prt_printf(&buf, "\n  ");
 493         bch2_bkey_val_to_text(&buf, c, k);
 494         if (!test_bit(BCH_FS_CHECK_BACKPOINTERS_DONE, &c->flags))
 495                 bch_err_ratelimited(c, "%s", buf.buf);
 496         else
 497                 bch2_trans_inconsistent(trans, "%s", buf.buf);
 498
 499         printbuf_exit(&buf);
 500 }
 501
 502 struct bkey_s_c bch2_backpointer_get_key(struct btree_trans *trans,
 503                                          struct btree_iter *iter,
 504                                          struct bpos bucket,
 505                                          u64 bp_offset,
 506                                          struct bch_backpointer bp)
 507 {
 508         struct bch_fs *c = trans->c;
 509         struct bkey_s_c k;
 510
 511         bch2_trans_node_iter_init(trans, iter,
 512                                   bp.btree_id,
 513                                   bp.pos,
 514                                   0,
 515                                   min(bp.level, c->btree_roots[bp.btree_id].level),
 516                                   0);
 517         k = bch2_btree_iter_peek_slot(iter);
 518         if (bkey_err(k)) {
 519                 bch2_trans_iter_exit(trans, iter);
 520                 return k;
 521         }
 522
 523         if (bp.level == c->btree_roots[bp.btree_id].level + 1)
 524                 k = bkey_i_to_s_c(&c->btree_roots[bp.btree_id].key);
 525
 526         if (extent_matches_bp(c, bp.btree_id, bp.level, k, bucket, bp))
 527                 return k;
 528
 529         bch2_trans_iter_exit(trans, iter);
 530
 531         if (bp.level) {
 532                 struct btree *b;
 533
 534                 /*
 535                  * If a backpointer for a btree node wasn't found, it may be
 536                  * because it was overwritten by a new btree node that hasn't
 537                  * been written out yet - backpointer_get_node() checks for
 538                  * this:
 539                  */
 540                 b = bch2_backpointer_get_node(trans, iter, bucket, bp_offset, bp);
 541                 if (!IS_ERR_OR_NULL(b))
 542                         return bkey_i_to_s_c(&b->key);
 543
 544                 bch2_trans_iter_exit(trans, iter);
 545
 546                 if (IS_ERR(b))
 547                         return bkey_s_c_err(PTR_ERR(b));
 548                 return bkey_s_c_null;
 549         }
 550
 551         backpointer_not_found(trans, bucket, bp_offset, bp, k, "extent");
 552         return bkey_s_c_null;
 553 }
 554
 555 struct btree *bch2_backpointer_get_node(struct btree_trans *trans,
 556                                         struct btree_iter *iter,
 557                                         struct bpos bucket,
 558                                         u64 bp_offset,
 559                                         struct bch_backpointer bp)
 560 {
 561         struct bch_fs *c = trans->c;
 562         struct btree *b;
 563
 564         BUG_ON(!bp.level);
 565
 566         bch2_trans_node_iter_init(trans, iter,
 567                                   bp.btree_id,
 568                                   bp.pos,
 569                                   0,
 570                                   bp.level - 1,
 571                                   0);
 572         b = bch2_btree_iter_peek_node(iter);
 573         if (IS_ERR(b))
 574                 goto err;
 575
 576         if (extent_matches_bp(c, bp.btree_id, bp.level,
 577                               bkey_i_to_s_c(&b->key),
 578                               bucket, bp))
 579                 return b;
 580
 581         if (btree_node_will_make_reachable(b)) {
 582                 b = ERR_PTR(-BCH_ERR_backpointer_to_overwritten_btree_node);
 583         } else {
 584                 backpointer_not_found(trans, bucket, bp_offset, bp,
 585                                       bkey_i_to_s_c(&b->key), "btree node");
 586                 b = NULL;
 587         }
 588 err:
 589         bch2_trans_iter_exit(trans, iter);
 590         return b;
 591 }
 592
 593 static int bch2_check_btree_backpointer(struct btree_trans *trans, struct btree_iter *bp_iter,
 594                                         struct bkey_s_c k)
 595 {
 596         struct bch_fs *c = trans->c;
 597         struct btree_iter alloc_iter = { NULL };
 598         struct bch_dev *ca;
 599         struct bkey_s_c alloc_k;
 600         struct printbuf buf = PRINTBUF;
 601         int ret = 0;
 602
 603         if (fsck_err_on(!bch2_dev_exists2(c, k.k->p.inode), c,
 604                         "backpointer for mising device:\n%s",
 605                         (bch2_bkey_val_to_text(&buf, c, k), buf.buf))) {
 606                 ret = bch2_btree_delete_at(trans, bp_iter, 0);
 607                 goto out;
 608         }
 609
 610         ca = bch_dev_bkey_exists(c, k.k->p.inode);
 611
 612         bch2_trans_iter_init(trans, &alloc_iter, BTREE_ID_alloc,
 613                              bp_pos_to_bucket(c, k.k->p), 0);
 614
 615         alloc_k = bch2_btree_iter_peek_slot(&alloc_iter);
 616         ret = bkey_err(alloc_k);
 617         if (ret)
 618                 goto out;
 619
 620         if (fsck_err_on(alloc_k.k->type != KEY_TYPE_alloc_v4, c,
 621                         "backpointer for nonexistent alloc key: %llu:%llu:0\n%s",
 622                         alloc_iter.pos.inode, alloc_iter.pos.offset,
 623                         (bch2_bkey_val_to_text(&buf, c, alloc_k), buf.buf))) {
 624                 ret = bch2_btree_delete_at(trans, bp_iter, 0);
 625                 goto out;
 626         }
 627 out:
 628 fsck_err:
 629         bch2_trans_iter_exit(trans, &alloc_iter);
 630         printbuf_exit(&buf);
 631         return ret;
 632 }
 633
 634 /* verify that every backpointer has a corresponding alloc key */
 635 int bch2_check_btree_backpointers(struct bch_fs *c)
 636 {
 637         struct btree_iter iter;
 638         struct bkey_s_c k;
 639
 640         return bch2_trans_run(c,
 641                 for_each_btree_key_commit(&trans, iter,
 642                         BTREE_ID_backpointers, POS_MIN, 0, k,
 643                         NULL, NULL, BTREE_INSERT_LAZY_RW|BTREE_INSERT_NOFAIL,
 644                   bch2_check_btree_backpointer(&trans, &iter, k)));
 645 }
 646
 647 static int check_bp_exists(struct btree_trans *trans,
 648                            struct bpos bucket_pos,
 649                            struct bch_backpointer bp,
 650                            struct bkey_s_c orig_k)
 651 {
 652         struct bch_fs *c = trans->c;
 653         struct btree_iter alloc_iter, bp_iter = { NULL };
 654         struct printbuf buf = PRINTBUF;
 655         struct bkey_s_c alloc_k, bp_k;
 656         int ret;
 657
 658         bch2_trans_iter_init(trans, &alloc_iter, BTREE_ID_alloc, bucket_pos, 0);
 659         alloc_k = bch2_btree_iter_peek_slot(&alloc_iter);
 660         ret = bkey_err(alloc_k);
 661         if (ret)
 662                 goto err;
 663
 664         if (alloc_k.k->type == KEY_TYPE_alloc_v4) {
 665                 struct bkey_s_c_alloc_v4 a = bkey_s_c_to_alloc_v4(alloc_k);
 666                 const struct bch_backpointer *bps = alloc_v4_backpointers_c(a.v);
 667                 unsigned i, nr = BCH_ALLOC_V4_NR_BACKPOINTERS(a.v);
 668
 669                 for (i = 0; i < nr; i++) {
 670                         int cmp = backpointer_cmp(bps[i], bp) ?:
 671                                 memcmp(&bps[i], &bp, sizeof(bp));
 672                         if (!cmp)
 673                                 goto out;
 674                         if (cmp >= 0)
 675                                 break;
 676                 }
 677         } else {
 678                 goto missing;
 679         }
 680
 681         bch2_trans_iter_init(trans, &bp_iter, BTREE_ID_backpointers,
 682                              bucket_pos_to_bp(c, bucket_pos, bp.bucket_offset),
 683                              0);
 684         bp_k = bch2_btree_iter_peek_slot(&bp_iter);
 685         ret = bkey_err(bp_k);
 686         if (ret)
 687                 goto err;
 688
 689         if (bp_k.k->type != KEY_TYPE_backpointer ||
 690             memcmp(bkey_s_c_to_backpointer(bp_k).v, &bp, sizeof(bp)))
 691                 goto missing;
 692 out:
 693 err:
 694 fsck_err:
 695         bch2_trans_iter_exit(trans, &bp_iter);
 696         bch2_trans_iter_exit(trans, &alloc_iter);
 697         printbuf_exit(&buf);
 698         return ret;
 699 missing:
 700         prt_printf(&buf, "missing backpointer for btree=%s l=%u ",
 701                bch2_btree_ids[bp.btree_id], bp.level);
 702         bch2_bkey_val_to_text(&buf, c, orig_k);
 703         prt_printf(&buf, "\nin alloc key ");
 704         bch2_bkey_val_to_text(&buf, c, alloc_k);
 705
 706         if (c->sb.version < bcachefs_metadata_version_backpointers ||
 707             c->opts.reconstruct_alloc ||
 708             fsck_err(c, "%s", buf.buf)) {
 709                 struct bkey_i_alloc_v4 *a = bch2_alloc_to_v4_mut(trans, alloc_k);
 710
 711                 ret   = PTR_ERR_OR_ZERO(a) ?:
 712                         bch2_bucket_backpointer_add(trans, a, bp, orig_k) ?:
 713                         bch2_trans_update(trans, &alloc_iter, &a->k_i, 0);
 714         }
 715
 716         goto out;
 717 }
 718
 719 static int check_extent_to_backpointers(struct btree_trans *trans,
 720                                         struct btree_iter *iter)
 721 {
 722         struct bch_fs *c = trans->c;
 723         struct bkey_ptrs_c ptrs;
 724         const union bch_extent_entry *entry;
 725         struct extent_ptr_decoded p;
 726         struct bkey_s_c k;
 727         int ret;
 728
 729         k = bch2_btree_iter_peek_all_levels(iter);
 730         ret = bkey_err(k);
 731         if (ret)
 732                 return ret;
 733         if (!k.k)
 734                 return 0;
 735
 736         ptrs = bch2_bkey_ptrs_c(k);
 737         bkey_for_each_ptr_decode(k.k, ptrs, p, entry) {
 738                 struct bpos bucket_pos;
 739                 struct bch_backpointer bp;
 740
 741                 if (p.ptr.cached)
 742                         continue;
 743
 744                 bch2_extent_ptr_to_bp(c, iter->btree_id, iter->path->level,
 745                                       k, p, &bucket_pos, &bp);
 746
 747                 ret = check_bp_exists(trans, bucket_pos, bp, k);
 748                 if (ret)
 749                         return ret;
 750         }
 751
 752         return 0;
 753 }
 754
 755 static int check_btree_root_to_backpointers(struct btree_trans *trans,
 756                                             enum btree_id btree_id)
 757 {
 758         struct bch_fs *c = trans->c;
 759         struct btree_iter iter;
 760         struct btree *b;
 761         struct bkey_s_c k;
 762         struct bkey_ptrs_c ptrs;
 763         struct extent_ptr_decoded p;
 764         const union bch_extent_entry *entry;
 765         int ret;
 766
 767         bch2_trans_node_iter_init(trans, &iter, btree_id, POS_MIN, 0,
 768                                   c->btree_roots[btree_id].level, 0);
 769         b = bch2_btree_iter_peek_node(&iter);
 770         ret = PTR_ERR_OR_ZERO(b);
 771         if (ret)
 772                 goto err;
 773
 774         BUG_ON(b != btree_node_root(c, b));
 775
 776         k = bkey_i_to_s_c(&b->key);
 777         ptrs = bch2_bkey_ptrs_c(k);
 778         bkey_for_each_ptr_decode(k.k, ptrs, p, entry) {
 779                 struct bpos bucket_pos;
 780                 struct bch_backpointer bp;
 781
 782                 if (p.ptr.cached)
 783                         continue;
 784
 785                 bch2_extent_ptr_to_bp(c, iter.btree_id, iter.path->level + 1,
 786                                       k, p, &bucket_pos, &bp);
 787
 788                 ret = check_bp_exists(trans, bucket_pos, bp, k);
 789                 if (ret)
 790                         goto err;
 791         }
 792 err:
 793         bch2_trans_iter_exit(trans, &iter);
 794         return ret;
 795 }
 796
 797 int bch2_check_extents_to_backpointers(struct bch_fs *c)
 798 {
 799         struct btree_trans trans;
 800         struct btree_iter iter;
 801         enum btree_id btree_id;
 802         int ret = 0;
 803
 804         bch2_trans_init(&trans, c, 0, 0);
 805         for (btree_id = 0; btree_id < BTREE_ID_NR; btree_id++) {
 806                 bch2_trans_node_iter_init(&trans, &iter, btree_id, POS_MIN, 0,
 807                                           0,
 808                                           BTREE_ITER_ALL_LEVELS|
 809                                           BTREE_ITER_PREFETCH);
 810
 811                 do {
 812                         ret = commit_do(&trans, NULL, NULL,
 813                                               BTREE_INSERT_LAZY_RW|
 814                                               BTREE_INSERT_NOFAIL,
 815                                               check_extent_to_backpointers(&trans, &iter));
 816                         if (ret)
 817                                 break;
 818                 } while (!bch2_btree_iter_advance(&iter));
 819
 820                 bch2_trans_iter_exit(&trans, &iter);
 821
 822                 if (ret)
 823                         break;
 824
 825                 ret = commit_do(&trans, NULL, NULL,
 826                                       BTREE_INSERT_LAZY_RW|
 827                                       BTREE_INSERT_NOFAIL,
 828                                       check_btree_root_to_backpointers(&trans, btree_id));
 829                 if (ret)
 830                         break;
 831         }
 832         bch2_trans_exit(&trans);
 833         return ret;
 834 }
 835
 836 static int check_one_backpointer(struct btree_trans *trans,
 837                                  struct bpos bucket,
 838                                  u64 *bp_offset)
 839 {
 840         struct btree_iter iter;
 841         struct bch_backpointer bp;
 842         struct bkey_s_c k;
 843         struct printbuf buf = PRINTBUF;
 844         int ret;
 845
 846         ret = bch2_get_next_backpointer(trans, bucket, -1,
 847                                         bp_offset, &bp);
 848         if (ret || *bp_offset == U64_MAX)
 849                 return ret;
 850
 851         k = bch2_backpointer_get_key(trans, &iter, bucket, *bp_offset, bp);
 852         ret = bkey_err(k);
 853         if (ret == -BCH_ERR_backpointer_to_overwritten_btree_node)
 854                 return 0;
 855         if (ret)
 856                 return ret;
 857
 858         if (fsck_err_on(!k.k, trans->c,
 859                         "%s backpointer points to missing extent\n%s",
 860                         *bp_offset < BACKPOINTER_OFFSET_MAX ? "alloc" : "btree",
 861                         (bch2_backpointer_to_text(&buf, &bp), buf.buf))) {
 862                 ret = bch2_backpointer_del_by_offset(trans, bucket, *bp_offset, bp);
 863                 if (ret == -ENOENT)
 864                         bch_err(trans->c, "backpointer at %llu not found", *bp_offset);
 865         }
 866
 867         bch2_trans_iter_exit(trans, &iter);
 868 fsck_err:
 869         printbuf_exit(&buf);
 870         return ret;
 871 }
 872
 873 int bch2_check_backpointers_to_extents(struct bch_fs *c)
 874 {
 875         struct btree_trans trans;
 876         struct btree_iter iter;
 877         struct bkey_s_c k;
 878         int ret = 0;
 879
 880         bch2_trans_init(&trans, c, 0, 0);
 881         for_each_btree_key(&trans, iter, BTREE_ID_alloc, POS_MIN,
 882                            BTREE_ITER_PREFETCH, k, ret) {
 883                 u64 bp_offset = 0;
 884
 885                 while (!(ret = commit_do(&trans, NULL, NULL,
 886                                                BTREE_INSERT_LAZY_RW|
 887                                                BTREE_INSERT_NOFAIL,
 888                                 check_one_backpointer(&trans, iter.pos, &bp_offset))) &&
 889                        bp_offset < U64_MAX)
 890                         bp_offset++;
 891
 892                 if (ret)
 893                         break;
 894         }
 895         bch2_trans_iter_exit(&trans, &iter);
 896         bch2_trans_exit(&trans);
 897         return ret < 0 ? ret : 0;
 898 }