fs/xfs/libxfs/xfs_ialloc_btree.c

   1 // SPDX-License-Identifier: GPL-2.0
   2 /*
   3  * Copyright (c) 2000-2001,2005 Silicon Graphics, Inc.
   4  * All Rights Reserved.
   5  */
   6 #include "xfs.h"
   7 #include "xfs_fs.h"
   8 #include "xfs_shared.h"
   9 #include "xfs_format.h"
  10 #include "xfs_log_format.h"
  11 #include "xfs_trans_resv.h"
  12 #include "xfs_bit.h"
  13 #include "xfs_mount.h"
  14 #include "xfs_btree.h"
  15 #include "xfs_btree_staging.h"
  16 #include "xfs_ialloc.h"
  17 #include "xfs_ialloc_btree.h"
  18 #include "xfs_alloc.h"
  19 #include "xfs_error.h"
  20 #include "xfs_trace.h"
  21 #include "xfs_trans.h"
  22 #include "xfs_rmap.h"
  23
  24 STATIC int
  25 xfs_inobt_get_minrecs(
  26         struct xfs_btree_cur    *cur,
  27         int                     level)
  28 {
  29         return M_IGEO(cur->bc_mp)->inobt_mnr[level != 0];
  30 }
  31
  32 STATIC struct xfs_btree_cur *
  33 xfs_inobt_dup_cursor(
  34         struct xfs_btree_cur    *cur)
  35 {
  36         return xfs_inobt_init_cursor(cur->bc_mp, cur->bc_tp,
  37                         cur->bc_ag.agbp, cur->bc_ag.agno,
  38                         cur->bc_btnum);
  39 }
  40
  41 STATIC void
  42 xfs_inobt_set_root(
  43         struct xfs_btree_cur    *cur,
  44         union xfs_btree_ptr     *nptr,
  45         int                     inc)    /* level change */
  46 {
  47         struct xfs_buf          *agbp = cur->bc_ag.agbp;
  48         struct xfs_agi          *agi = agbp->b_addr;
  49
  50         agi->agi_root = nptr->s;
  51         be32_add_cpu(&agi->agi_level, inc);
  52         xfs_ialloc_log_agi(cur->bc_tp, agbp, XFS_AGI_ROOT | XFS_AGI_LEVEL);
  53 }
  54
  55 STATIC void
  56 xfs_finobt_set_root(
  57         struct xfs_btree_cur    *cur,
  58         union xfs_btree_ptr     *nptr,
  59         int                     inc)    /* level change */
  60 {
  61         struct xfs_buf          *agbp = cur->bc_ag.agbp;
  62         struct xfs_agi          *agi = agbp->b_addr;
  63
  64         agi->agi_free_root = nptr->s;
  65         be32_add_cpu(&agi->agi_free_level, inc);
  66         xfs_ialloc_log_agi(cur->bc_tp, agbp,
  67                            XFS_AGI_FREE_ROOT | XFS_AGI_FREE_LEVEL);
  68 }
  69
  70 STATIC int
  71 __xfs_inobt_alloc_block(
  72         struct xfs_btree_cur    *cur,
  73         union xfs_btree_ptr     *start,
  74         union xfs_btree_ptr     *new,
  75         int                     *stat,
  76         enum xfs_ag_resv_type   resv)
  77 {
  78         xfs_alloc_arg_t         args;           /* block allocation args */
  79         int                     error;          /* error return value */
  80         xfs_agblock_t           sbno = be32_to_cpu(start->s);
  81
  82         memset(&args, 0, sizeof(args));
  83         args.tp = cur->bc_tp;
  84         args.mp = cur->bc_mp;
  85         args.oinfo = XFS_RMAP_OINFO_INOBT;
  86         args.fsbno = XFS_AGB_TO_FSB(args.mp, cur->bc_ag.agno, sbno);
  87         args.minlen = 1;
  88         args.maxlen = 1;
  89         args.prod = 1;
  90         args.type = XFS_ALLOCTYPE_NEAR_BNO;
  91         args.resv = resv;
  92
  93         error = xfs_alloc_vextent(&args);
  94         if (error)
  95                 return error;
  96
  97         if (args.fsbno == NULLFSBLOCK) {
  98                 *stat = 0;
  99                 return 0;
 100         }
 101         ASSERT(args.len == 1);
 102
 103         new->s = cpu_to_be32(XFS_FSB_TO_AGBNO(args.mp, args.fsbno));
 104         *stat = 1;
 105         return 0;
 106 }
 107
 108 STATIC int
 109 xfs_inobt_alloc_block(
 110         struct xfs_btree_cur    *cur,
 111         union xfs_btree_ptr     *start,
 112         union xfs_btree_ptr     *new,
 113         int                     *stat)
 114 {
 115         return __xfs_inobt_alloc_block(cur, start, new, stat, XFS_AG_RESV_NONE);
 116 }
 117
 118 STATIC int
 119 xfs_finobt_alloc_block(
 120         struct xfs_btree_cur    *cur,
 121         union xfs_btree_ptr     *start,
 122         union xfs_btree_ptr     *new,
 123         int                     *stat)
 124 {
 125         if (cur->bc_mp->m_finobt_nores)
 126                 return xfs_inobt_alloc_block(cur, start, new, stat);
 127         return __xfs_inobt_alloc_block(cur, start, new, stat,
 128                         XFS_AG_RESV_METADATA);
 129 }
 130
 131 STATIC int
 132 __xfs_inobt_free_block(
 133         struct xfs_btree_cur    *cur,
 134         struct xfs_buf          *bp,
 135         enum xfs_ag_resv_type   resv)
 136 {
 137         return xfs_free_extent(cur->bc_tp,
 138                         XFS_DADDR_TO_FSB(cur->bc_mp, XFS_BUF_ADDR(bp)), 1,
 139                         &XFS_RMAP_OINFO_INOBT, resv);
 140 }
 141
 142 STATIC int
 143 xfs_inobt_free_block(
 144         struct xfs_btree_cur    *cur,
 145         struct xfs_buf          *bp)
 146 {
 147         return __xfs_inobt_free_block(cur, bp, XFS_AG_RESV_NONE);
 148 }
 149
 150 STATIC int
 151 xfs_finobt_free_block(
 152         struct xfs_btree_cur    *cur,
 153         struct xfs_buf          *bp)
 154 {
 155         if (cur->bc_mp->m_finobt_nores)
 156                 return xfs_inobt_free_block(cur, bp);
 157         return __xfs_inobt_free_block(cur, bp, XFS_AG_RESV_METADATA);
 158 }
 159
 160 STATIC int
 161 xfs_inobt_get_maxrecs(
 162         struct xfs_btree_cur    *cur,
 163         int                     level)
 164 {
 165         return M_IGEO(cur->bc_mp)->inobt_mxr[level != 0];
 166 }
 167
 168 STATIC void
 169 xfs_inobt_init_key_from_rec(
 170         union xfs_btree_key     *key,
 171         union xfs_btree_rec     *rec)
 172 {
 173         key->inobt.ir_startino = rec->inobt.ir_startino;
 174 }
 175
 176 STATIC void
 177 xfs_inobt_init_high_key_from_rec(
 178         union xfs_btree_key     *key,
 179         union xfs_btree_rec     *rec)
 180 {
 181         __u32                   x;
 182
 183         x = be32_to_cpu(rec->inobt.ir_startino);
 184         x += XFS_INODES_PER_CHUNK - 1;
 185         key->inobt.ir_startino = cpu_to_be32(x);
 186 }
 187
 188 STATIC void
 189 xfs_inobt_init_rec_from_cur(
 190         struct xfs_btree_cur    *cur,
 191         union xfs_btree_rec     *rec)
 192 {
 193         rec->inobt.ir_startino = cpu_to_be32(cur->bc_rec.i.ir_startino);
 194         if (xfs_sb_version_hassparseinodes(&cur->bc_mp->m_sb)) {
 195                 rec->inobt.ir_u.sp.ir_holemask =
 196                                         cpu_to_be16(cur->bc_rec.i.ir_holemask);
 197                 rec->inobt.ir_u.sp.ir_count = cur->bc_rec.i.ir_count;
 198                 rec->inobt.ir_u.sp.ir_freecount = cur->bc_rec.i.ir_freecount;
 199         } else {
 200                 /* ir_holemask/ir_count not supported on-disk */
 201                 rec->inobt.ir_u.f.ir_freecount =
 202                                         cpu_to_be32(cur->bc_rec.i.ir_freecount);
 203         }
 204         rec->inobt.ir_free = cpu_to_be64(cur->bc_rec.i.ir_free);
 205 }
 206
 207 /*
 208  * initial value of ptr for lookup
 209  */
 210 STATIC void
 211 xfs_inobt_init_ptr_from_cur(
 212         struct xfs_btree_cur    *cur,
 213         union xfs_btree_ptr     *ptr)
 214 {
 215         struct xfs_agi          *agi = cur->bc_ag.agbp->b_addr;
 216
 217         ASSERT(cur->bc_ag.agno == be32_to_cpu(agi->agi_seqno));
 218
 219         ptr->s = agi->agi_root;
 220 }
 221
 222 STATIC void
 223 xfs_finobt_init_ptr_from_cur(
 224         struct xfs_btree_cur    *cur,
 225         union xfs_btree_ptr     *ptr)
 226 {
 227         struct xfs_agi          *agi = cur->bc_ag.agbp->b_addr;
 228
 229         ASSERT(cur->bc_ag.agno == be32_to_cpu(agi->agi_seqno));
 230         ptr->s = agi->agi_free_root;
 231 }
 232
 233 STATIC int64_t
 234 xfs_inobt_key_diff(
 235         struct xfs_btree_cur    *cur,
 236         union xfs_btree_key     *key)
 237 {
 238         return (int64_t)be32_to_cpu(key->inobt.ir_startino) -
 239                           cur->bc_rec.i.ir_startino;
 240 }
 241
 242 STATIC int64_t
 243 xfs_inobt_diff_two_keys(
 244         struct xfs_btree_cur    *cur,
 245         union xfs_btree_key     *k1,
 246         union xfs_btree_key     *k2)
 247 {
 248         return (int64_t)be32_to_cpu(k1->inobt.ir_startino) -
 249                           be32_to_cpu(k2->inobt.ir_startino);
 250 }
 251
 252 static xfs_failaddr_t
 253 xfs_inobt_verify(
 254         struct xfs_buf          *bp)
 255 {
 256         struct xfs_mount        *mp = bp->b_mount;
 257         struct xfs_btree_block  *block = XFS_BUF_TO_BLOCK(bp);
 258         xfs_failaddr_t          fa;
 259         unsigned int            level;
 260
 261         if (!xfs_verify_magic(bp, block->bb_magic))
 262                 return __this_address;
 263
 264         /*
 265          * During growfs operations, we can't verify the exact owner as the
 266          * perag is not fully initialised and hence not attached to the buffer.
 267          *
 268          * Similarly, during log recovery we will have a perag structure
 269          * attached, but the agi information will not yet have been initialised
 270          * from the on disk AGI. We don't currently use any of this information,
 271          * but beware of the landmine (i.e. need to check pag->pagi_init) if we
 272          * ever do.
 273          */
 274         if (xfs_sb_version_hascrc(&mp->m_sb)) {
 275                 fa = xfs_btree_sblock_v5hdr_verify(bp);
 276                 if (fa)
 277                         return fa;
 278         }
 279
 280         /* level verification */
 281         level = be16_to_cpu(block->bb_level);
 282         if (level >= M_IGEO(mp)->inobt_maxlevels)
 283                 return __this_address;
 284
 285         return xfs_btree_sblock_verify(bp,
 286                         M_IGEO(mp)->inobt_mxr[level != 0]);
 287 }
 288
 289 static void
 290 xfs_inobt_read_verify(
 291         struct xfs_buf  *bp)
 292 {
 293         xfs_failaddr_t  fa;
 294
 295         if (!xfs_btree_sblock_verify_crc(bp))
 296                 xfs_verifier_error(bp, -EFSBADCRC, __this_address);
 297         else {
 298                 fa = xfs_inobt_verify(bp);
 299                 if (fa)
 300                         xfs_verifier_error(bp, -EFSCORRUPTED, fa);
 301         }
 302
 303         if (bp->b_error)
 304                 trace_xfs_btree_corrupt(bp, _RET_IP_);
 305 }
 306
 307 static void
 308 xfs_inobt_write_verify(
 309         struct xfs_buf  *bp)
 310 {
 311         xfs_failaddr_t  fa;
 312
 313         fa = xfs_inobt_verify(bp);
 314         if (fa) {
 315                 trace_xfs_btree_corrupt(bp, _RET_IP_);
 316                 xfs_verifier_error(bp, -EFSCORRUPTED, fa);
 317                 return;
 318         }
 319         xfs_btree_sblock_calc_crc(bp);
 320
 321 }
 322
 323 const struct xfs_buf_ops xfs_inobt_buf_ops = {
 324         .name = "xfs_inobt",
 325         .magic = { cpu_to_be32(XFS_IBT_MAGIC), cpu_to_be32(XFS_IBT_CRC_MAGIC) },
 326         .verify_read = xfs_inobt_read_verify,
 327         .verify_write = xfs_inobt_write_verify,
 328         .verify_struct = xfs_inobt_verify,
 329 };
 330
 331 const struct xfs_buf_ops xfs_finobt_buf_ops = {
 332         .name = "xfs_finobt",
 333         .magic = { cpu_to_be32(XFS_FIBT_MAGIC),
 334                    cpu_to_be32(XFS_FIBT_CRC_MAGIC) },
 335         .verify_read = xfs_inobt_read_verify,
 336         .verify_write = xfs_inobt_write_verify,
 337         .verify_struct = xfs_inobt_verify,
 338 };
 339
 340 STATIC int
 341 xfs_inobt_keys_inorder(
 342         struct xfs_btree_cur    *cur,
 343         union xfs_btree_key     *k1,
 344         union xfs_btree_key     *k2)
 345 {
 346         return be32_to_cpu(k1->inobt.ir_startino) <
 347                 be32_to_cpu(k2->inobt.ir_startino);
 348 }
 349
 350 STATIC int
 351 xfs_inobt_recs_inorder(
 352         struct xfs_btree_cur    *cur,
 353         union xfs_btree_rec     *r1,
 354         union xfs_btree_rec     *r2)
 355 {
 356         return be32_to_cpu(r1->inobt.ir_startino) + XFS_INODES_PER_CHUNK <=
 357                 be32_to_cpu(r2->inobt.ir_startino);
 358 }
 359
 360 static const struct xfs_btree_ops xfs_inobt_ops = {
 361         .rec_len                = sizeof(xfs_inobt_rec_t),
 362         .key_len                = sizeof(xfs_inobt_key_t),
 363
 364         .dup_cursor             = xfs_inobt_dup_cursor,
 365         .set_root               = xfs_inobt_set_root,
 366         .alloc_block            = xfs_inobt_alloc_block,
 367         .free_block             = xfs_inobt_free_block,
 368         .get_minrecs            = xfs_inobt_get_minrecs,
 369         .get_maxrecs            = xfs_inobt_get_maxrecs,
 370         .init_key_from_rec      = xfs_inobt_init_key_from_rec,
 371         .init_high_key_from_rec = xfs_inobt_init_high_key_from_rec,
 372         .init_rec_from_cur      = xfs_inobt_init_rec_from_cur,
 373         .init_ptr_from_cur      = xfs_inobt_init_ptr_from_cur,
 374         .key_diff               = xfs_inobt_key_diff,
 375         .buf_ops                = &xfs_inobt_buf_ops,
 376         .diff_two_keys          = xfs_inobt_diff_two_keys,
 377         .keys_inorder           = xfs_inobt_keys_inorder,
 378         .recs_inorder           = xfs_inobt_recs_inorder,
 379 };
 380
 381 static const struct xfs_btree_ops xfs_finobt_ops = {
 382         .rec_len                = sizeof(xfs_inobt_rec_t),
 383         .key_len                = sizeof(xfs_inobt_key_t),
 384
 385         .dup_cursor             = xfs_inobt_dup_cursor,
 386         .set_root               = xfs_finobt_set_root,
 387         .alloc_block            = xfs_finobt_alloc_block,
 388         .free_block             = xfs_finobt_free_block,
 389         .get_minrecs            = xfs_inobt_get_minrecs,
 390         .get_maxrecs            = xfs_inobt_get_maxrecs,
 391         .init_key_from_rec      = xfs_inobt_init_key_from_rec,
 392         .init_high_key_from_rec = xfs_inobt_init_high_key_from_rec,
 393         .init_rec_from_cur      = xfs_inobt_init_rec_from_cur,
 394         .init_ptr_from_cur      = xfs_finobt_init_ptr_from_cur,
 395         .key_diff               = xfs_inobt_key_diff,
 396         .buf_ops                = &xfs_finobt_buf_ops,
 397         .diff_two_keys          = xfs_inobt_diff_two_keys,
 398         .keys_inorder           = xfs_inobt_keys_inorder,
 399         .recs_inorder           = xfs_inobt_recs_inorder,
 400 };
 401
 402 /*
 403  * Initialize a new inode btree cursor.
 404  */
 405 static struct xfs_btree_cur *
 406 xfs_inobt_init_common(
 407         struct xfs_mount        *mp,            /* file system mount point */
 408         struct xfs_trans        *tp,            /* transaction pointer */
 409         xfs_agnumber_t          agno,           /* allocation group number */
 410         xfs_btnum_t             btnum)          /* ialloc or free ino btree */
 411 {
 412         struct xfs_btree_cur    *cur;
 413
 414         cur = kmem_zone_zalloc(xfs_btree_cur_zone, KM_NOFS);
 415         cur->bc_tp = tp;
 416         cur->bc_mp = mp;
 417         cur->bc_btnum = btnum;
 418         if (btnum == XFS_BTNUM_INO) {
 419                 cur->bc_statoff = XFS_STATS_CALC_INDEX(xs_ibt_2);
 420                 cur->bc_ops = &xfs_inobt_ops;
 421         } else {
 422                 cur->bc_statoff = XFS_STATS_CALC_INDEX(xs_fibt_2);
 423                 cur->bc_ops = &xfs_finobt_ops;
 424         }
 425
 426         cur->bc_blocklog = mp->m_sb.sb_blocklog;
 427
 428         if (xfs_sb_version_hascrc(&mp->m_sb))
 429                 cur->bc_flags |= XFS_BTREE_CRC_BLOCKS;
 430
 431         cur->bc_ag.agno = agno;
 432         return cur;
 433 }
 434
 435 /* Create an inode btree cursor. */
 436 struct xfs_btree_cur *
 437 xfs_inobt_init_cursor(
 438         struct xfs_mount        *mp,
 439         struct xfs_trans        *tp,
 440         struct xfs_buf          *agbp,
 441         xfs_agnumber_t          agno,
 442         xfs_btnum_t             btnum)
 443 {
 444         struct xfs_btree_cur    *cur;
 445         struct xfs_agi          *agi = agbp->b_addr;
 446
 447         cur = xfs_inobt_init_common(mp, tp, agno, btnum);
 448         if (btnum == XFS_BTNUM_INO)
 449                 cur->bc_nlevels = be32_to_cpu(agi->agi_level);
 450         else
 451                 cur->bc_nlevels = be32_to_cpu(agi->agi_free_level);
 452         cur->bc_ag.agbp = agbp;
 453         return cur;
 454 }
 455
 456 /* Create an inode btree cursor with a fake root for staging. */
 457 struct xfs_btree_cur *
 458 xfs_inobt_stage_cursor(
 459         struct xfs_mount        *mp,
 460         struct xbtree_afakeroot *afake,
 461         xfs_agnumber_t          agno,
 462         xfs_btnum_t             btnum)
 463 {
 464         struct xfs_btree_cur    *cur;
 465
 466         cur = xfs_inobt_init_common(mp, NULL, agno, btnum);
 467         xfs_btree_stage_afakeroot(cur, afake);
 468         return cur;
 469 }
 470
 471 /*
 472  * Install a new inobt btree root.  Caller is responsible for invalidating
 473  * and freeing the old btree blocks.
 474  */
 475 void
 476 xfs_inobt_commit_staged_btree(
 477         struct xfs_btree_cur    *cur,
 478         struct xfs_trans        *tp,
 479         struct xfs_buf          *agbp)
 480 {
 481         struct xfs_agi          *agi = agbp->b_addr;
 482         struct xbtree_afakeroot *afake = cur->bc_ag.afake;
 483
 484         ASSERT(cur->bc_flags & XFS_BTREE_STAGING);
 485
 486         if (cur->bc_btnum == XFS_BTNUM_INO) {
 487                 agi->agi_root = cpu_to_be32(afake->af_root);
 488                 agi->agi_level = cpu_to_be32(afake->af_levels);
 489                 xfs_ialloc_log_agi(tp, agbp, XFS_AGI_ROOT | XFS_AGI_LEVEL);
 490                 xfs_btree_commit_afakeroot(cur, tp, agbp, &xfs_inobt_ops);
 491         } else {
 492                 agi->agi_free_root = cpu_to_be32(afake->af_root);
 493                 agi->agi_free_level = cpu_to_be32(afake->af_levels);
 494                 xfs_ialloc_log_agi(tp, agbp, XFS_AGI_FREE_ROOT |
 495                                              XFS_AGI_FREE_LEVEL);
 496                 xfs_btree_commit_afakeroot(cur, tp, agbp, &xfs_finobt_ops);
 497         }
 498 }
 499
 500 /*
 501  * Calculate number of records in an inobt btree block.
 502  */
 503 int
 504 xfs_inobt_maxrecs(
 505         struct xfs_mount        *mp,
 506         int                     blocklen,
 507         int                     leaf)
 508 {
 509         blocklen -= XFS_INOBT_BLOCK_LEN(mp);
 510
 511         if (leaf)
 512                 return blocklen / sizeof(xfs_inobt_rec_t);
 513         return blocklen / (sizeof(xfs_inobt_key_t) + sizeof(xfs_inobt_ptr_t));
 514 }
 515
 516 /*
 517  * Convert the inode record holemask to an inode allocation bitmap. The inode
 518  * allocation bitmap is inode granularity and specifies whether an inode is
 519  * physically allocated on disk (not whether the inode is considered allocated
 520  * or free by the fs).
 521  *
 522  * A bit value of 1 means the inode is allocated, a value of 0 means it is free.
 523  */
 524 uint64_t
 525 xfs_inobt_irec_to_allocmask(
 526         struct xfs_inobt_rec_incore     *rec)
 527 {
 528         uint64_t                        bitmap = 0;
 529         uint64_t                        inodespbit;
 530         int                             nextbit;
 531         uint                            allocbitmap;
 532
 533         /*
 534          * The holemask has 16-bits for a 64 inode record. Therefore each
 535          * holemask bit represents multiple inodes. Create a mask of bits to set
 536          * in the allocmask for each holemask bit.
 537          */
 538         inodespbit = (1 << XFS_INODES_PER_HOLEMASK_BIT) - 1;
 539
 540         /*
 541          * Allocated inodes are represented by 0 bits in holemask. Invert the 0
 542          * bits to 1 and convert to a uint so we can use xfs_next_bit(). Mask
 543          * anything beyond the 16 holemask bits since this casts to a larger
 544          * type.
 545          */
 546         allocbitmap = ~rec->ir_holemask & ((1 << XFS_INOBT_HOLEMASK_BITS) - 1);
 547
 548         /*
 549          * allocbitmap is the inverted holemask so every set bit represents
 550          * allocated inodes. To expand from 16-bit holemask granularity to
 551          * 64-bit (e.g., bit-per-inode), set inodespbit bits in the target
 552          * bitmap for every holemask bit.
 553          */
 554         nextbit = xfs_next_bit(&allocbitmap, 1, 0);
 555         while (nextbit != -1) {
 556                 ASSERT(nextbit < (sizeof(rec->ir_holemask) * NBBY));
 557
 558                 bitmap |= (inodespbit <<
 559                            (nextbit * XFS_INODES_PER_HOLEMASK_BIT));
 560
 561                 nextbit = xfs_next_bit(&allocbitmap, 1, nextbit + 1);
 562         }
 563
 564         return bitmap;
 565 }
 566
 567 #if defined(DEBUG) || defined(XFS_WARN)
 568 /*
 569  * Verify that an in-core inode record has a valid inode count.
 570  */
 571 int
 572 xfs_inobt_rec_check_count(
 573         struct xfs_mount                *mp,
 574         struct xfs_inobt_rec_incore     *rec)
 575 {
 576         int                             inocount = 0;
 577         int                             nextbit = 0;
 578         uint64_t                        allocbmap;
 579         int                             wordsz;
 580
 581         wordsz = sizeof(allocbmap) / sizeof(unsigned int);
 582         allocbmap = xfs_inobt_irec_to_allocmask(rec);
 583
 584         nextbit = xfs_next_bit((uint *) &allocbmap, wordsz, nextbit);
 585         while (nextbit != -1) {
 586                 inocount++;
 587                 nextbit = xfs_next_bit((uint *) &allocbmap, wordsz,
 588                                        nextbit + 1);
 589         }
 590
 591         if (inocount != rec->ir_count)
 592                 return -EFSCORRUPTED;
 593
 594         return 0;
 595 }
 596 #endif  /* DEBUG */
 597
 598 static xfs_extlen_t
 599 xfs_inobt_max_size(
 600         struct xfs_mount        *mp,
 601         xfs_agnumber_t          agno)
 602 {
 603         xfs_agblock_t           agblocks = xfs_ag_block_count(mp, agno);
 604
 605         /* Bail out if we're uninitialized, which can happen in mkfs. */
 606         if (M_IGEO(mp)->inobt_mxr[0] == 0)
 607                 return 0;
 608
 609         /*
 610          * The log is permanently allocated, so the space it occupies will
 611          * never be available for the kinds of things that would require btree
 612          * expansion.  We therefore can pretend the space isn't there.
 613          */
 614         if (mp->m_sb.sb_logstart &&
 615             XFS_FSB_TO_AGNO(mp, mp->m_sb.sb_logstart) == agno)
 616                 agblocks -= mp->m_sb.sb_logblocks;
 617
 618         return xfs_btree_calc_size(M_IGEO(mp)->inobt_mnr,
 619                                 (uint64_t)agblocks * mp->m_sb.sb_inopblock /
 620                                         XFS_INODES_PER_CHUNK);
 621 }
 622
 623 /* Read AGI and create inobt cursor. */
 624 int
 625 xfs_inobt_cur(
 626         struct xfs_mount        *mp,
 627         struct xfs_trans        *tp,
 628         xfs_agnumber_t          agno,
 629         xfs_btnum_t             which,
 630         struct xfs_btree_cur    **curpp,
 631         struct xfs_buf          **agi_bpp)
 632 {
 633         struct xfs_btree_cur    *cur;
 634         int                     error;
 635
 636         ASSERT(*agi_bpp == NULL);
 637         ASSERT(*curpp == NULL);
 638
 639         error = xfs_ialloc_read_agi(mp, tp, agno, agi_bpp);
 640         if (error)
 641                 return error;
 642
 643         cur = xfs_inobt_init_cursor(mp, tp, *agi_bpp, agno, which);
 644         if (!cur) {
 645                 xfs_trans_brelse(tp, *agi_bpp);
 646                 *agi_bpp = NULL;
 647                 return -ENOMEM;
 648         }
 649         *curpp = cur;
 650         return 0;
 651 }
 652
 653 static int
 654 xfs_inobt_count_blocks(
 655         struct xfs_mount        *mp,
 656         struct xfs_trans        *tp,
 657         xfs_agnumber_t          agno,
 658         xfs_btnum_t             btnum,
 659         xfs_extlen_t            *tree_blocks)
 660 {
 661         struct xfs_buf          *agbp = NULL;
 662         struct xfs_btree_cur    *cur = NULL;
 663         int                     error;
 664
 665         error = xfs_inobt_cur(mp, tp, agno, btnum, &cur, &agbp);
 666         if (error)
 667                 return error;
 668
 669         error = xfs_btree_count_blocks(cur, tree_blocks);
 670         xfs_btree_del_cursor(cur, error);
 671         xfs_trans_brelse(tp, agbp);
 672
 673         return error;
 674 }
 675
 676 /*
 677  * Figure out how many blocks to reserve and how many are used by this btree.
 678  */
 679 int
 680 xfs_finobt_calc_reserves(
 681         struct xfs_mount        *mp,
 682         struct xfs_trans        *tp,
 683         xfs_agnumber_t          agno,
 684         xfs_extlen_t            *ask,
 685         xfs_extlen_t            *used)
 686 {
 687         xfs_extlen_t            tree_len = 0;
 688         int                     error;
 689
 690         if (!xfs_sb_version_hasfinobt(&mp->m_sb))
 691                 return 0;
 692
 693         error = xfs_inobt_count_blocks(mp, tp, agno, XFS_BTNUM_FINO, &tree_len);
 694         if (error)
 695                 return error;
 696
 697         *ask += xfs_inobt_max_size(mp, agno);
 698         *used += tree_len;
 699         return 0;
 700 }
 701
 702 /* Calculate the inobt btree size for some records. */
 703 xfs_extlen_t
 704 xfs_iallocbt_calc_size(
 705         struct xfs_mount        *mp,
 706         unsigned long long      len)
 707 {
 708         return xfs_btree_calc_size(M_IGEO(mp)->inobt_mnr, len);
 709 }