fs/xfs/xfs_alloc_btree.c

   1 /*
   2  * Copyright (c) 2000-2001,2005 Silicon Graphics, Inc.
   3  * All Rights Reserved.
   4  *
   5  * This program is free software; you can redistribute it and/or
   6  * modify it under the terms of the GNU General Public License as
   7  * published by the Free Software Foundation.
   8  *
   9  * This program is distributed in the hope that it would be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12  * GNU General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU General Public License
  15  * along with this program; if not, write the Free Software Foundation,
  16  * Inc.,  51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
  17  */
  18 #include "xfs.h"
  19 #include "xfs_fs.h"
  20 #include "xfs_types.h"
  21 #include "xfs_bit.h"
  22 #include "xfs_log.h"
  23 #include "xfs_inum.h"
  24 #include "xfs_trans.h"
  25 #include "xfs_sb.h"
  26 #include "xfs_ag.h"
  27 #include "xfs_dir2.h"
  28 #include "xfs_dmapi.h"
  29 #include "xfs_mount.h"
  30 #include "xfs_bmap_btree.h"
  31 #include "xfs_alloc_btree.h"
  32 #include "xfs_ialloc_btree.h"
  33 #include "xfs_dir2_sf.h"
  34 #include "xfs_attr_sf.h"
  35 #include "xfs_dinode.h"
  36 #include "xfs_inode.h"
  37 #include "xfs_btree.h"
  38 #include "xfs_ialloc.h"
  39 #include "xfs_alloc.h"
  40 #include "xfs_error.h"
  41
  42 /*
  43  * Prototypes for internal functions.
  44  */
  45
  46 STATIC void xfs_alloc_log_block(xfs_trans_t *, xfs_buf_t *, int);
  47 STATIC void xfs_alloc_log_keys(xfs_btree_cur_t *, xfs_buf_t *, int, int);
  48 STATIC void xfs_alloc_log_ptrs(xfs_btree_cur_t *, xfs_buf_t *, int, int);
  49 STATIC void xfs_alloc_log_recs(xfs_btree_cur_t *, xfs_buf_t *, int, int);
  50 STATIC int xfs_alloc_lshift(xfs_btree_cur_t *, int, int *);
  51 STATIC int xfs_alloc_newroot(xfs_btree_cur_t *, int *);
  52 STATIC int xfs_alloc_rshift(xfs_btree_cur_t *, int, int *);
  53 STATIC int xfs_alloc_split(xfs_btree_cur_t *, int, xfs_agblock_t *,
  54                 xfs_alloc_key_t *, xfs_btree_cur_t **, int *);
  55 STATIC int xfs_alloc_updkey(xfs_btree_cur_t *, xfs_alloc_key_t *, int);
  56
  57 /*
  58  * Internal functions.
  59  */
  60
  61 /*
  62  * Single level of the xfs_alloc_delete record deletion routine.
  63  * Delete record pointed to by cur/level.
  64  * Remove the record from its block then rebalance the tree.
  65  * Return 0 for error, 1 for done, 2 to go on to the next level.
  66  */
  67 STATIC int                              /* error */
  68 xfs_alloc_delrec(
  69         xfs_btree_cur_t         *cur,   /* btree cursor */
  70         int                     level,  /* level removing record from */
  71         int                     *stat)  /* fail/done/go-on */
  72 {
  73         xfs_agf_t               *agf;   /* allocation group freelist header */
  74         xfs_alloc_block_t       *block; /* btree block record/key lives in */
  75         xfs_agblock_t           bno;    /* btree block number */
  76         xfs_buf_t               *bp;    /* buffer for block */
  77         int                     error;  /* error return value */
  78         int                     i;      /* loop index */
  79         xfs_alloc_key_t         key;    /* kp points here if block is level 0 */
  80         xfs_agblock_t           lbno;   /* left block's block number */
  81         xfs_buf_t               *lbp;   /* left block's buffer pointer */
  82         xfs_alloc_block_t       *left;  /* left btree block */
  83         xfs_alloc_key_t         *lkp=NULL;      /* left block key pointer */
  84         xfs_alloc_ptr_t         *lpp=NULL;      /* left block address pointer */
  85         int                     lrecs=0;        /* number of records in left block */
  86         xfs_alloc_rec_t         *lrp;   /* left block record pointer */
  87         xfs_mount_t             *mp;    /* mount structure */
  88         int                     ptr;    /* index in btree block for this rec */
  89         xfs_agblock_t           rbno;   /* right block's block number */
  90         xfs_buf_t               *rbp;   /* right block's buffer pointer */
  91         xfs_alloc_block_t       *right; /* right btree block */
  92         xfs_alloc_key_t         *rkp;   /* right block key pointer */
  93         xfs_alloc_ptr_t         *rpp;   /* right block address pointer */
  94         int                     rrecs=0;        /* number of records in right block */
  95         int                     numrecs;
  96         xfs_alloc_rec_t         *rrp;   /* right block record pointer */
  97         xfs_btree_cur_t         *tcur;  /* temporary btree cursor */
  98
  99         /*
 100          * Get the index of the entry being deleted, check for nothing there.
 101          */
 102         ptr = cur->bc_ptrs[level];
 103         if (ptr == 0) {
 104                 *stat = 0;
 105                 return 0;
 106         }
 107         /*
 108          * Get the buffer & block containing the record or key/ptr.
 109          */
 110         bp = cur->bc_bufs[level];
 111         block = XFS_BUF_TO_ALLOC_BLOCK(bp);
 112 #ifdef DEBUG
 113         if ((error = xfs_btree_check_sblock(cur, block, level, bp)))
 114                 return error;
 115 #endif
 116         /*
 117          * Fail if we're off the end of the block.
 118          */
 119         numrecs = be16_to_cpu(block->bb_numrecs);
 120         if (ptr > numrecs) {
 121                 *stat = 0;
 122                 return 0;
 123         }
 124         XFS_STATS_INC(xs_abt_delrec);
 125         /*
 126          * It's a nonleaf.  Excise the key and ptr being deleted, by
 127          * sliding the entries past them down one.
 128          * Log the changed areas of the block.
 129          */
 130         if (level > 0) {
 131                 lkp = XFS_ALLOC_KEY_ADDR(block, 1, cur);
 132                 lpp = XFS_ALLOC_PTR_ADDR(block, 1, cur);
 133 #ifdef DEBUG
 134                 for (i = ptr; i < numrecs; i++) {
 135                         if ((error = xfs_btree_check_sptr(cur, be32_to_cpu(lpp[i]), level)))
 136                                 return error;
 137                 }
 138 #endif
 139                 if (ptr < numrecs) {
 140                         memmove(&lkp[ptr - 1], &lkp[ptr],
 141                                 (numrecs - ptr) * sizeof(*lkp));
 142                         memmove(&lpp[ptr - 1], &lpp[ptr],
 143                                 (numrecs - ptr) * sizeof(*lpp));
 144                         xfs_alloc_log_ptrs(cur, bp, ptr, numrecs - 1);
 145                         xfs_alloc_log_keys(cur, bp, ptr, numrecs - 1);
 146                 }
 147         }
 148         /*
 149          * It's a leaf.  Excise the record being deleted, by sliding the
 150          * entries past it down one.  Log the changed areas of the block.
 151          */
 152         else {
 153                 lrp = XFS_ALLOC_REC_ADDR(block, 1, cur);
 154                 if (ptr < numrecs) {
 155                         memmove(&lrp[ptr - 1], &lrp[ptr],
 156                                 (numrecs - ptr) * sizeof(*lrp));
 157                         xfs_alloc_log_recs(cur, bp, ptr, numrecs - 1);
 158                 }
 159                 /*
 160                  * If it's the first record in the block, we'll need a key
 161                  * structure to pass up to the next level (updkey).
 162                  */
 163                 if (ptr == 1) {
 164                         key.ar_startblock = lrp->ar_startblock;
 165                         key.ar_blockcount = lrp->ar_blockcount;
 166                         lkp = &key;
 167                 }
 168         }
 169         /*
 170          * Decrement and log the number of entries in the block.
 171          */
 172         numrecs--;
 173         block->bb_numrecs = cpu_to_be16(numrecs);
 174         xfs_alloc_log_block(cur->bc_tp, bp, XFS_BB_NUMRECS);
 175         /*
 176          * See if the longest free extent in the allocation group was
 177          * changed by this operation.  True if it's the by-size btree, and
 178          * this is the leaf level, and there is no right sibling block,
 179          * and this was the last record.
 180          */
 181         agf = XFS_BUF_TO_AGF(cur->bc_private.a.agbp);
 182         mp = cur->bc_mp;
 183
 184         if (level == 0 &&
 185             cur->bc_btnum == XFS_BTNUM_CNT &&
 186             be32_to_cpu(block->bb_rightsib) == NULLAGBLOCK &&
 187             ptr > numrecs) {
 188                 ASSERT(ptr == numrecs + 1);
 189                 /*
 190                  * There are still records in the block.  Grab the size
 191                  * from the last one.
 192                  */
 193                 if (numrecs) {
 194                         rrp = XFS_ALLOC_REC_ADDR(block, numrecs, cur);
 195                         agf->agf_longest = rrp->ar_blockcount;
 196                 }
 197                 /*
 198                  * No free extents left.
 199                  */
 200                 else
 201                         agf->agf_longest = 0;
 202                 mp->m_perag[be32_to_cpu(agf->agf_seqno)].pagf_longest =
 203                         be32_to_cpu(agf->agf_longest);
 204                 xfs_alloc_log_agf(cur->bc_tp, cur->bc_private.a.agbp,
 205                         XFS_AGF_LONGEST);
 206         }
 207         /*
 208          * Is this the root level?  If so, we're almost done.
 209          */
 210         if (level == cur->bc_nlevels - 1) {
 211                 /*
 212                  * If this is the root level,
 213                  * and there's only one entry left,
 214                  * and it's NOT the leaf level,
 215                  * then we can get rid of this level.
 216                  */
 217                 if (numrecs == 1 && level > 0) {
 218                         /*
 219                          * lpp is still set to the first pointer in the block.
 220                          * Make it the new root of the btree.
 221                          */
 222                         bno = be32_to_cpu(agf->agf_roots[cur->bc_btnum]);
 223                         agf->agf_roots[cur->bc_btnum] = *lpp;
 224                         be32_add(&agf->agf_levels[cur->bc_btnum], -1);
 225                         mp->m_perag[be32_to_cpu(agf->agf_seqno)].pagf_levels[cur->bc_btnum]--;
 226                         /*
 227                          * Put this buffer/block on the ag's freelist.
 228                          */
 229                         if ((error = xfs_alloc_put_freelist(cur->bc_tp,
 230                                         cur->bc_private.a.agbp, NULL, bno)))
 231                                 return error;
 232                         /*
 233                          * Since blocks move to the free list without the
 234                          * coordination used in xfs_bmap_finish, we can't allow
 235                          * block to be available for reallocation and
 236                          * non-transaction writing (user data) until we know
 237                          * that the transaction that moved it to the free list
 238                          * is permanently on disk. We track the blocks by
 239                          * declaring these blocks as "busy"; the busy list is
 240                          * maintained on a per-ag basis and each transaction
 241                          * records which entries should be removed when the
 242                          * iclog commits to disk. If a busy block is
 243                          * allocated, the iclog is pushed up to the LSN
 244                          * that freed the block.
 245                          */
 246                         xfs_alloc_mark_busy(cur->bc_tp,
 247                                 be32_to_cpu(agf->agf_seqno), bno, 1);
 248
 249                         xfs_trans_agbtree_delta(cur->bc_tp, -1);
 250                         xfs_alloc_log_agf(cur->bc_tp, cur->bc_private.a.agbp,
 251                                 XFS_AGF_ROOTS | XFS_AGF_LEVELS);
 252                         /*
 253                          * Update the cursor so there's one fewer level.
 254                          */
 255                         xfs_btree_setbuf(cur, level, NULL);
 256                         cur->bc_nlevels--;
 257                 } else if (level > 0 &&
 258                            (error = xfs_alloc_decrement(cur, level, &i)))
 259                         return error;
 260                 *stat = 1;
 261                 return 0;
 262         }
 263         /*
 264          * If we deleted the leftmost entry in the block, update the
 265          * key values above us in the tree.
 266          */
 267         if (ptr == 1 && (error = xfs_alloc_updkey(cur, lkp, level + 1)))
 268                 return error;
 269         /*
 270          * If the number of records remaining in the block is at least
 271          * the minimum, we're done.
 272          */
 273         if (numrecs >= XFS_ALLOC_BLOCK_MINRECS(level, cur)) {
 274                 if (level > 0 && (error = xfs_alloc_decrement(cur, level, &i)))
 275                         return error;
 276                 *stat = 1;
 277                 return 0;
 278         }
 279         /*
 280          * Otherwise, we have to move some records around to keep the
 281          * tree balanced.  Look at the left and right sibling blocks to
 282          * see if we can re-balance by moving only one record.
 283          */
 284         rbno = be32_to_cpu(block->bb_rightsib);
 285         lbno = be32_to_cpu(block->bb_leftsib);
 286         bno = NULLAGBLOCK;
 287         ASSERT(rbno != NULLAGBLOCK || lbno != NULLAGBLOCK);
 288         /*
 289          * Duplicate the cursor so our btree manipulations here won't
 290          * disrupt the next level up.
 291          */
 292         if ((error = xfs_btree_dup_cursor(cur, &tcur)))
 293                 return error;
 294         /*
 295          * If there's a right sibling, see if it's ok to shift an entry
 296          * out of it.
 297          */
 298         if (rbno != NULLAGBLOCK) {
 299                 /*
 300                  * Move the temp cursor to the last entry in the next block.
 301                  * Actually any entry but the first would suffice.
 302                  */
 303                 i = xfs_btree_lastrec(tcur, level);
 304                 XFS_WANT_CORRUPTED_GOTO(i == 1, error0);
 305                 if ((error = xfs_alloc_increment(tcur, level, &i)))
 306                         goto error0;
 307                 XFS_WANT_CORRUPTED_GOTO(i == 1, error0);
 308                 i = xfs_btree_lastrec(tcur, level);
 309                 XFS_WANT_CORRUPTED_GOTO(i == 1, error0);
 310                 /*
 311                  * Grab a pointer to the block.
 312                  */
 313                 rbp = tcur->bc_bufs[level];
 314                 right = XFS_BUF_TO_ALLOC_BLOCK(rbp);
 315 #ifdef DEBUG
 316                 if ((error = xfs_btree_check_sblock(cur, right, level, rbp)))
 317                         goto error0;
 318 #endif
 319                 /*
 320                  * Grab the current block number, for future use.
 321                  */
 322                 bno = be32_to_cpu(right->bb_leftsib);
 323                 /*
 324                  * If right block is full enough so that removing one entry
 325                  * won't make it too empty, and left-shifting an entry out
 326                  * of right to us works, we're done.
 327                  */
 328                 if (be16_to_cpu(right->bb_numrecs) - 1 >=
 329                      XFS_ALLOC_BLOCK_MINRECS(level, cur)) {
 330                         if ((error = xfs_alloc_lshift(tcur, level, &i)))
 331                                 goto error0;
 332                         if (i) {
 333                                 ASSERT(be16_to_cpu(block->bb_numrecs) >=
 334                                        XFS_ALLOC_BLOCK_MINRECS(level, cur));
 335                                 xfs_btree_del_cursor(tcur,
 336                                                      XFS_BTREE_NOERROR);
 337                                 if (level > 0 &&
 338                                     (error = xfs_alloc_decrement(cur, level,
 339                                             &i)))
 340                                         return error;
 341                                 *stat = 1;
 342                                 return 0;
 343                         }
 344                 }
 345                 /*
 346                  * Otherwise, grab the number of records in right for
 347                  * future reference, and fix up the temp cursor to point
 348                  * to our block again (last record).
 349                  */
 350                 rrecs = be16_to_cpu(right->bb_numrecs);
 351                 if (lbno != NULLAGBLOCK) {
 352                         i = xfs_btree_firstrec(tcur, level);
 353                         XFS_WANT_CORRUPTED_GOTO(i == 1, error0);
 354                         if ((error = xfs_alloc_decrement(tcur, level, &i)))
 355                                 goto error0;
 356                         XFS_WANT_CORRUPTED_GOTO(i == 1, error0);
 357                 }
 358         }
 359         /*
 360          * If there's a left sibling, see if it's ok to shift an entry
 361          * out of it.
 362          */
 363         if (lbno != NULLAGBLOCK) {
 364                 /*
 365                  * Move the temp cursor to the first entry in the
 366                  * previous block.
 367                  */
 368                 i = xfs_btree_firstrec(tcur, level);
 369                 XFS_WANT_CORRUPTED_GOTO(i == 1, error0);
 370                 if ((error = xfs_alloc_decrement(tcur, level, &i)))
 371                         goto error0;
 372                 XFS_WANT_CORRUPTED_GOTO(i == 1, error0);
 373                 xfs_btree_firstrec(tcur, level);
 374                 /*
 375                  * Grab a pointer to the block.
 376                  */
 377                 lbp = tcur->bc_bufs[level];
 378                 left = XFS_BUF_TO_ALLOC_BLOCK(lbp);
 379 #ifdef DEBUG
 380                 if ((error = xfs_btree_check_sblock(cur, left, level, lbp)))
 381                         goto error0;
 382 #endif
 383                 /*
 384                  * Grab the current block number, for future use.
 385                  */
 386                 bno = be32_to_cpu(left->bb_rightsib);
 387                 /*
 388                  * If left block is full enough so that removing one entry
 389                  * won't make it too empty, and right-shifting an entry out
 390                  * of left to us works, we're done.
 391                  */
 392                 if (be16_to_cpu(left->bb_numrecs) - 1 >=
 393                      XFS_ALLOC_BLOCK_MINRECS(level, cur)) {
 394                         if ((error = xfs_alloc_rshift(tcur, level, &i)))
 395                                 goto error0;
 396                         if (i) {
 397                                 ASSERT(be16_to_cpu(block->bb_numrecs) >=
 398                                        XFS_ALLOC_BLOCK_MINRECS(level, cur));
 399                                 xfs_btree_del_cursor(tcur,
 400                                                      XFS_BTREE_NOERROR);
 401                                 if (level == 0)
 402                                         cur->bc_ptrs[0]++;
 403                                 *stat = 1;
 404                                 return 0;
 405                         }
 406                 }
 407                 /*
 408                  * Otherwise, grab the number of records in right for
 409                  * future reference.
 410                  */
 411                 lrecs = be16_to_cpu(left->bb_numrecs);
 412         }
 413         /*
 414          * Delete the temp cursor, we're done with it.
 415          */
 416         xfs_btree_del_cursor(tcur, XFS_BTREE_NOERROR);
 417         /*
 418          * If here, we need to do a join to keep the tree balanced.
 419          */
 420         ASSERT(bno != NULLAGBLOCK);
 421         /*
 422          * See if we can join with the left neighbor block.
 423          */
 424         if (lbno != NULLAGBLOCK &&
 425             lrecs + numrecs <= XFS_ALLOC_BLOCK_MAXRECS(level, cur)) {
 426                 /*
 427                  * Set "right" to be the starting block,
 428                  * "left" to be the left neighbor.
 429                  */
 430                 rbno = bno;
 431                 right = block;
 432                 rrecs = be16_to_cpu(right->bb_numrecs);
 433                 rbp = bp;
 434                 if ((error = xfs_btree_read_bufs(mp, cur->bc_tp,
 435                                 cur->bc_private.a.agno, lbno, 0, &lbp,
 436                                 XFS_ALLOC_BTREE_REF)))
 437                         return error;
 438                 left = XFS_BUF_TO_ALLOC_BLOCK(lbp);
 439                 lrecs = be16_to_cpu(left->bb_numrecs);
 440                 if ((error = xfs_btree_check_sblock(cur, left, level, lbp)))
 441                         return error;
 442         }
 443         /*
 444          * If that won't work, see if we can join with the right neighbor block.
 445          */
 446         else if (rbno != NULLAGBLOCK &&
 447                  rrecs + numrecs <= XFS_ALLOC_BLOCK_MAXRECS(level, cur)) {
 448                 /*
 449                  * Set "left" to be the starting block,
 450                  * "right" to be the right neighbor.
 451                  */
 452                 lbno = bno;
 453                 left = block;
 454                 lrecs = be16_to_cpu(left->bb_numrecs);
 455                 lbp = bp;
 456                 if ((error = xfs_btree_read_bufs(mp, cur->bc_tp,
 457                                 cur->bc_private.a.agno, rbno, 0, &rbp,
 458                                 XFS_ALLOC_BTREE_REF)))
 459                         return error;
 460                 right = XFS_BUF_TO_ALLOC_BLOCK(rbp);
 461                 rrecs = be16_to_cpu(right->bb_numrecs);
 462                 if ((error = xfs_btree_check_sblock(cur, right, level, rbp)))
 463                         return error;
 464         }
 465         /*
 466          * Otherwise, we can't fix the imbalance.
 467          * Just return.  This is probably a logic error, but it's not fatal.
 468          */
 469         else {
 470                 if (level > 0 && (error = xfs_alloc_decrement(cur, level, &i)))
 471                         return error;
 472                 *stat = 1;
 473                 return 0;
 474         }
 475         /*
 476          * We're now going to join "left" and "right" by moving all the stuff
 477          * in "right" to "left" and deleting "right".
 478          */
 479         if (level > 0) {
 480                 /*
 481                  * It's a non-leaf.  Move keys and pointers.
 482                  */
 483                 lkp = XFS_ALLOC_KEY_ADDR(left, lrecs + 1, cur);
 484                 lpp = XFS_ALLOC_PTR_ADDR(left, lrecs + 1, cur);
 485                 rkp = XFS_ALLOC_KEY_ADDR(right, 1, cur);
 486                 rpp = XFS_ALLOC_PTR_ADDR(right, 1, cur);
 487 #ifdef DEBUG
 488                 for (i = 0; i < rrecs; i++) {
 489                         if ((error = xfs_btree_check_sptr(cur, be32_to_cpu(rpp[i]), level)))
 490                                 return error;
 491                 }
 492 #endif
 493                 memcpy(lkp, rkp, rrecs * sizeof(*lkp));
 494                 memcpy(lpp, rpp, rrecs * sizeof(*lpp));
 495                 xfs_alloc_log_keys(cur, lbp, lrecs + 1, lrecs + rrecs);
 496                 xfs_alloc_log_ptrs(cur, lbp, lrecs + 1, lrecs + rrecs);
 497         } else {
 498                 /*
 499                  * It's a leaf.  Move records.
 500                  */
 501                 lrp = XFS_ALLOC_REC_ADDR(left, lrecs + 1, cur);
 502                 rrp = XFS_ALLOC_REC_ADDR(right, 1, cur);
 503                 memcpy(lrp, rrp, rrecs * sizeof(*lrp));
 504                 xfs_alloc_log_recs(cur, lbp, lrecs + 1, lrecs + rrecs);
 505         }
 506         /*
 507          * If we joined with the left neighbor, set the buffer in the
 508          * cursor to the left block, and fix up the index.
 509          */
 510         if (bp != lbp) {
 511                 xfs_btree_setbuf(cur, level, lbp);
 512                 cur->bc_ptrs[level] += lrecs;
 513         }
 514         /*
 515          * If we joined with the right neighbor and there's a level above
 516          * us, increment the cursor at that level.
 517          */
 518         else if (level + 1 < cur->bc_nlevels &&
 519                  (error = xfs_alloc_increment(cur, level + 1, &i)))
 520                 return error;
 521         /*
 522          * Fix up the number of records in the surviving block.
 523          */
 524         lrecs += rrecs;
 525         left->bb_numrecs = cpu_to_be16(lrecs);
 526         /*
 527          * Fix up the right block pointer in the surviving block, and log it.
 528          */
 529         left->bb_rightsib = right->bb_rightsib;
 530         xfs_alloc_log_block(cur->bc_tp, lbp, XFS_BB_NUMRECS | XFS_BB_RIGHTSIB);
 531         /*
 532          * If there is a right sibling now, make it point to the
 533          * remaining block.
 534          */
 535         if (be32_to_cpu(left->bb_rightsib) != NULLAGBLOCK) {
 536                 xfs_alloc_block_t       *rrblock;
 537                 xfs_buf_t               *rrbp;
 538
 539                 if ((error = xfs_btree_read_bufs(mp, cur->bc_tp,
 540                                 cur->bc_private.a.agno, be32_to_cpu(left->bb_rightsib), 0,
 541                                 &rrbp, XFS_ALLOC_BTREE_REF)))
 542                         return error;
 543                 rrblock = XFS_BUF_TO_ALLOC_BLOCK(rrbp);
 544                 if ((error = xfs_btree_check_sblock(cur, rrblock, level, rrbp)))
 545                         return error;
 546                 rrblock->bb_leftsib = cpu_to_be32(lbno);
 547                 xfs_alloc_log_block(cur->bc_tp, rrbp, XFS_BB_LEFTSIB);
 548         }
 549         /*
 550          * Free the deleting block by putting it on the freelist.
 551          */
 552         if ((error = xfs_alloc_put_freelist(cur->bc_tp, cur->bc_private.a.agbp,
 553                         NULL, rbno)))
 554                 return error;
 555         /*
 556          * Since blocks move to the free list without the coordination
 557          * used in xfs_bmap_finish, we can't allow block to be available
 558          * for reallocation and non-transaction writing (user data)
 559          * until we know that the transaction that moved it to the free
 560          * list is permanently on disk. We track the blocks by declaring
 561          * these blocks as "busy"; the busy list is maintained on a
 562          * per-ag basis and each transaction records which entries
 563          * should be removed when the iclog commits to disk. If a
 564          * busy block is allocated, the iclog is pushed up to the
 565          * LSN that freed the block.
 566          */
 567         xfs_alloc_mark_busy(cur->bc_tp, be32_to_cpu(agf->agf_seqno), bno, 1);
 568         xfs_trans_agbtree_delta(cur->bc_tp, -1);
 569
 570         /*
 571          * Adjust the current level's cursor so that we're left referring
 572          * to the right node, after we're done.
 573          * If this leaves the ptr value 0 our caller will fix it up.
 574          */
 575         if (level > 0)
 576                 cur->bc_ptrs[level]--;
 577         /*
 578          * Return value means the next level up has something to do.
 579          */
 580         *stat = 2;
 581         return 0;
 582
 583 error0:
 584         xfs_btree_del_cursor(tcur, XFS_BTREE_ERROR);
 585         return error;
 586 }
 587
 588 /*
 589  * Insert one record/level.  Return information to the caller
 590  * allowing the next level up to proceed if necessary.
 591  */
 592 STATIC int                              /* error */
 593 xfs_alloc_insrec(
 594         xfs_btree_cur_t         *cur,   /* btree cursor */
 595         int                     level,  /* level to insert record at */
 596         xfs_agblock_t           *bnop,  /* i/o: block number inserted */
 597         xfs_alloc_rec_t         *recp,  /* i/o: record data inserted */
 598         xfs_btree_cur_t         **curp, /* output: new cursor replacing cur */
 599         int                     *stat)  /* output: success/failure */
 600 {
 601         xfs_agf_t               *agf;   /* allocation group freelist header */
 602         xfs_alloc_block_t       *block; /* btree block record/key lives in */
 603         xfs_buf_t               *bp;    /* buffer for block */
 604         int                     error;  /* error return value */
 605         int                     i;      /* loop index */
 606         xfs_alloc_key_t         key;    /* key value being inserted */
 607         xfs_alloc_key_t         *kp;    /* pointer to btree keys */
 608         xfs_agblock_t           nbno;   /* block number of allocated block */
 609         xfs_btree_cur_t         *ncur;  /* new cursor to be used at next lvl */
 610         xfs_alloc_key_t         nkey;   /* new key value, from split */
 611         xfs_alloc_rec_t         nrec;   /* new record value, for caller */
 612         int                     numrecs;
 613         int                     optr;   /* old ptr value */
 614         xfs_alloc_ptr_t         *pp;    /* pointer to btree addresses */
 615         int                     ptr;    /* index in btree block for this rec */
 616         xfs_alloc_rec_t         *rp;    /* pointer to btree records */
 617
 618         ASSERT(be32_to_cpu(recp->ar_blockcount) > 0);
 619
 620         /*
 621          * GCC doesn't understand the (arguably complex) control flow in
 622          * this function and complains about uninitialized structure fields
 623          * without this.
 624          */
 625         memset(&nrec, 0, sizeof(nrec));
 626
 627         /*
 628          * If we made it to the root level, allocate a new root block
 629          * and we're done.
 630          */
 631         if (level >= cur->bc_nlevels) {
 632                 XFS_STATS_INC(xs_abt_insrec);
 633                 if ((error = xfs_alloc_newroot(cur, &i)))
 634                         return error;
 635                 *bnop = NULLAGBLOCK;
 636                 *stat = i;
 637                 return 0;
 638         }
 639         /*
 640          * Make a key out of the record data to be inserted, and save it.
 641          */
 642         key.ar_startblock = recp->ar_startblock;
 643         key.ar_blockcount = recp->ar_blockcount;
 644         optr = ptr = cur->bc_ptrs[level];
 645         /*
 646          * If we're off the left edge, return failure.
 647          */
 648         if (ptr == 0) {
 649                 *stat = 0;
 650                 return 0;
 651         }
 652         XFS_STATS_INC(xs_abt_insrec);
 653         /*
 654          * Get pointers to the btree buffer and block.
 655          */
 656         bp = cur->bc_bufs[level];
 657         block = XFS_BUF_TO_ALLOC_BLOCK(bp);
 658         numrecs = be16_to_cpu(block->bb_numrecs);
 659 #ifdef DEBUG
 660         if ((error = xfs_btree_check_sblock(cur, block, level, bp)))
 661                 return error;
 662         /*
 663          * Check that the new entry is being inserted in the right place.
 664          */
 665         if (ptr <= numrecs) {
 666                 if (level == 0) {
 667                         rp = XFS_ALLOC_REC_ADDR(block, ptr, cur);
 668                         xfs_btree_check_rec(cur->bc_btnum, recp, rp);
 669                 } else {
 670                         kp = XFS_ALLOC_KEY_ADDR(block, ptr, cur);
 671                         xfs_btree_check_key(cur->bc_btnum, &key, kp);
 672                 }
 673         }
 674 #endif
 675         nbno = NULLAGBLOCK;
 676         ncur = NULL;
 677         /*
 678          * If the block is full, we can't insert the new entry until we
 679          * make the block un-full.
 680          */
 681         if (numrecs == XFS_ALLOC_BLOCK_MAXRECS(level, cur)) {
 682                 /*
 683                  * First, try shifting an entry to the right neighbor.
 684                  */
 685                 if ((error = xfs_alloc_rshift(cur, level, &i)))
 686                         return error;
 687                 if (i) {
 688                         /* nothing */
 689                 }
 690                 /*
 691                  * Next, try shifting an entry to the left neighbor.
 692                  */
 693                 else {
 694                         if ((error = xfs_alloc_lshift(cur, level, &i)))
 695                                 return error;
 696                         if (i)
 697                                 optr = ptr = cur->bc_ptrs[level];
 698                         else {
 699                                 /*
 700                                  * Next, try splitting the current block in
 701                                  * half. If this works we have to re-set our
 702                                  * variables because we could be in a
 703                                  * different block now.
 704                                  */
 705                                 if ((error = xfs_alloc_split(cur, level, &nbno,
 706                                                 &nkey, &ncur, &i)))
 707                                         return error;
 708                                 if (i) {
 709                                         bp = cur->bc_bufs[level];
 710                                         block = XFS_BUF_TO_ALLOC_BLOCK(bp);
 711 #ifdef DEBUG
 712                                         if ((error =
 713                                                 xfs_btree_check_sblock(cur,
 714                                                         block, level, bp)))
 715                                                 return error;
 716 #endif
 717                                         ptr = cur->bc_ptrs[level];
 718                                         nrec.ar_startblock = nkey.ar_startblock;
 719                                         nrec.ar_blockcount = nkey.ar_blockcount;
 720                                 }
 721                                 /*
 722                                  * Otherwise the insert fails.
 723                                  */
 724                                 else {
 725                                         *stat = 0;
 726                                         return 0;
 727                                 }
 728                         }
 729                 }
 730         }
 731         /*
 732          * At this point we know there's room for our new entry in the block
 733          * we're pointing at.
 734          */
 735         numrecs = be16_to_cpu(block->bb_numrecs);
 736         if (level > 0) {
 737                 /*
 738                  * It's a non-leaf entry.  Make a hole for the new data
 739                  * in the key and ptr regions of the block.
 740                  */
 741                 kp = XFS_ALLOC_KEY_ADDR(block, 1, cur);
 742                 pp = XFS_ALLOC_PTR_ADDR(block, 1, cur);
 743 #ifdef DEBUG
 744                 for (i = numrecs; i >= ptr; i--) {
 745                         if ((error = xfs_btree_check_sptr(cur, be32_to_cpu(pp[i - 1]), level)))
 746                                 return error;
 747                 }
 748 #endif
 749                 memmove(&kp[ptr], &kp[ptr - 1],
 750                         (numrecs - ptr + 1) * sizeof(*kp));
 751                 memmove(&pp[ptr], &pp[ptr - 1],
 752                         (numrecs - ptr + 1) * sizeof(*pp));
 753 #ifdef DEBUG
 754                 if ((error = xfs_btree_check_sptr(cur, *bnop, level)))
 755                         return error;
 756 #endif
 757                 /*
 758                  * Now stuff the new data in, bump numrecs and log the new data.
 759                  */
 760                 kp[ptr - 1] = key;
 761                 pp[ptr - 1] = cpu_to_be32(*bnop);
 762                 numrecs++;
 763                 block->bb_numrecs = cpu_to_be16(numrecs);
 764                 xfs_alloc_log_keys(cur, bp, ptr, numrecs);
 765                 xfs_alloc_log_ptrs(cur, bp, ptr, numrecs);
 766 #ifdef DEBUG
 767                 if (ptr < numrecs)
 768                         xfs_btree_check_key(cur->bc_btnum, kp + ptr - 1,
 769                                 kp + ptr);
 770 #endif
 771         } else {
 772                 /*
 773                  * It's a leaf entry.  Make a hole for the new record.
 774                  */
 775                 rp = XFS_ALLOC_REC_ADDR(block, 1, cur);
 776                 memmove(&rp[ptr], &rp[ptr - 1],
 777                         (numrecs - ptr + 1) * sizeof(*rp));
 778                 /*
 779                  * Now stuff the new record in, bump numrecs
 780                  * and log the new data.
 781                  */
 782                 rp[ptr - 1] = *recp;
 783                 numrecs++;
 784                 block->bb_numrecs = cpu_to_be16(numrecs);
 785                 xfs_alloc_log_recs(cur, bp, ptr, numrecs);
 786 #ifdef DEBUG
 787                 if (ptr < numrecs)
 788                         xfs_btree_check_rec(cur->bc_btnum, rp + ptr - 1,
 789                                 rp + ptr);
 790 #endif
 791         }
 792         /*
 793          * Log the new number of records in the btree header.
 794          */
 795         xfs_alloc_log_block(cur->bc_tp, bp, XFS_BB_NUMRECS);
 796         /*
 797          * If we inserted at the start of a block, update the parents' keys.
 798          */
 799         if (optr == 1 && (error = xfs_alloc_updkey(cur, &key, level + 1)))
 800                 return error;
 801         /*
 802          * Look to see if the longest extent in the allocation group
 803          * needs to be updated.
 804          */
 805
 806         agf = XFS_BUF_TO_AGF(cur->bc_private.a.agbp);
 807         if (level == 0 &&
 808             cur->bc_btnum == XFS_BTNUM_CNT &&
 809             be32_to_cpu(block->bb_rightsib) == NULLAGBLOCK &&
 810             be32_to_cpu(recp->ar_blockcount) > be32_to_cpu(agf->agf_longest)) {
 811                 /*
 812                  * If this is a leaf in the by-size btree and there
 813                  * is no right sibling block and this block is bigger
 814                  * than the previous longest block, update it.
 815                  */
 816                 agf->agf_longest = recp->ar_blockcount;
 817                 cur->bc_mp->m_perag[be32_to_cpu(agf->agf_seqno)].pagf_longest
 818                         = be32_to_cpu(recp->ar_blockcount);
 819                 xfs_alloc_log_agf(cur->bc_tp, cur->bc_private.a.agbp,
 820                         XFS_AGF_LONGEST);
 821         }
 822         /*
 823          * Return the new block number, if any.
 824          * If there is one, give back a record value and a cursor too.
 825          */
 826         *bnop = nbno;
 827         if (nbno != NULLAGBLOCK) {
 828                 *recp = nrec;
 829                 *curp = ncur;
 830         }
 831         *stat = 1;
 832         return 0;
 833 }
 834
 835 /*
 836  * Log header fields from a btree block.
 837  */
 838 STATIC void
 839 xfs_alloc_log_block(
 840         xfs_trans_t             *tp,    /* transaction pointer */
 841         xfs_buf_t               *bp,    /* buffer containing btree block */
 842         int                     fields) /* mask of fields: XFS_BB_... */
 843 {
 844         int                     first;  /* first byte offset logged */
 845         int                     last;   /* last byte offset logged */
 846         static const short      offsets[] = {   /* table of offsets */
 847                 offsetof(xfs_alloc_block_t, bb_magic),
 848                 offsetof(xfs_alloc_block_t, bb_level),
 849                 offsetof(xfs_alloc_block_t, bb_numrecs),
 850                 offsetof(xfs_alloc_block_t, bb_leftsib),
 851                 offsetof(xfs_alloc_block_t, bb_rightsib),
 852                 sizeof(xfs_alloc_block_t)
 853         };
 854
 855         xfs_btree_offsets(fields, offsets, XFS_BB_NUM_BITS, &first, &last);
 856         xfs_trans_log_buf(tp, bp, first, last);
 857 }
 858
 859 /*
 860  * Log keys from a btree block (nonleaf).
 861  */
 862 STATIC void
 863 xfs_alloc_log_keys(
 864         xfs_btree_cur_t         *cur,   /* btree cursor */
 865         xfs_buf_t               *bp,    /* buffer containing btree block */
 866         int                     kfirst, /* index of first key to log */
 867         int                     klast)  /* index of last key to log */
 868 {
 869         xfs_alloc_block_t       *block; /* btree block to log from */
 870         int                     first;  /* first byte offset logged */
 871         xfs_alloc_key_t         *kp;    /* key pointer in btree block */
 872         int                     last;   /* last byte offset logged */
 873
 874         block = XFS_BUF_TO_ALLOC_BLOCK(bp);
 875         kp = XFS_ALLOC_KEY_ADDR(block, 1, cur);
 876         first = (int)((xfs_caddr_t)&kp[kfirst - 1] - (xfs_caddr_t)block);
 877         last = (int)(((xfs_caddr_t)&kp[klast] - 1) - (xfs_caddr_t)block);
 878         xfs_trans_log_buf(cur->bc_tp, bp, first, last);
 879 }
 880
 881 /*
 882  * Log block pointer fields from a btree block (nonleaf).
 883  */
 884 STATIC void
 885 xfs_alloc_log_ptrs(
 886         xfs_btree_cur_t         *cur,   /* btree cursor */
 887         xfs_buf_t               *bp,    /* buffer containing btree block */
 888         int                     pfirst, /* index of first pointer to log */
 889         int                     plast)  /* index of last pointer to log */
 890 {
 891         xfs_alloc_block_t       *block; /* btree block to log from */
 892         int                     first;  /* first byte offset logged */
 893         int                     last;   /* last byte offset logged */
 894         xfs_alloc_ptr_t         *pp;    /* block-pointer pointer in btree blk */
 895
 896         block = XFS_BUF_TO_ALLOC_BLOCK(bp);
 897         pp = XFS_ALLOC_PTR_ADDR(block, 1, cur);
 898         first = (int)((xfs_caddr_t)&pp[pfirst - 1] - (xfs_caddr_t)block);
 899         last = (int)(((xfs_caddr_t)&pp[plast] - 1) - (xfs_caddr_t)block);
 900         xfs_trans_log_buf(cur->bc_tp, bp, first, last);
 901 }
 902
 903 /*
 904  * Log records from a btree block (leaf).
 905  */
 906 STATIC void
 907 xfs_alloc_log_recs(
 908         xfs_btree_cur_t         *cur,   /* btree cursor */
 909         xfs_buf_t               *bp,    /* buffer containing btree block */
 910         int                     rfirst, /* index of first record to log */
 911         int                     rlast)  /* index of last record to log */
 912 {
 913         xfs_alloc_block_t       *block; /* btree block to log from */
 914         int                     first;  /* first byte offset logged */
 915         int                     last;   /* last byte offset logged */
 916         xfs_alloc_rec_t         *rp;    /* record pointer for btree block */
 917
 918
 919         block = XFS_BUF_TO_ALLOC_BLOCK(bp);
 920         rp = XFS_ALLOC_REC_ADDR(block, 1, cur);
 921 #ifdef DEBUG
 922         {
 923                 xfs_agf_t       *agf;
 924                 xfs_alloc_rec_t *p;
 925
 926                 agf = XFS_BUF_TO_AGF(cur->bc_private.a.agbp);
 927                 for (p = &rp[rfirst - 1]; p <= &rp[rlast - 1]; p++)
 928                         ASSERT(be32_to_cpu(p->ar_startblock) +
 929                                be32_to_cpu(p->ar_blockcount) <=
 930                                be32_to_cpu(agf->agf_length));
 931         }
 932 #endif
 933         first = (int)((xfs_caddr_t)&rp[rfirst - 1] - (xfs_caddr_t)block);
 934         last = (int)(((xfs_caddr_t)&rp[rlast] - 1) - (xfs_caddr_t)block);
 935         xfs_trans_log_buf(cur->bc_tp, bp, first, last);
 936 }
 937
 938 /*
 939  * Lookup the record.  The cursor is made to point to it, based on dir.
 940  * Return 0 if can't find any such record, 1 for success.
 941  */
 942 STATIC int                              /* error */
 943 xfs_alloc_lookup(
 944         xfs_btree_cur_t         *cur,   /* btree cursor */
 945         xfs_lookup_t            dir,    /* <=, ==, or >= */
 946         int                     *stat)  /* success/failure */
 947 {
 948         xfs_agblock_t           agbno;  /* a.g. relative btree block number */
 949         xfs_agnumber_t          agno;   /* allocation group number */
 950         xfs_alloc_block_t       *block=NULL;    /* current btree block */
 951         int                     diff;   /* difference for the current key */
 952         int                     error;  /* error return value */
 953         int                     keyno=0;        /* current key number */
 954         int                     level;  /* level in the btree */
 955         xfs_mount_t             *mp;    /* file system mount point */
 956
 957         XFS_STATS_INC(xs_abt_lookup);
 958         /*
 959          * Get the allocation group header, and the root block number.
 960          */
 961         mp = cur->bc_mp;
 962
 963         {
 964                 xfs_agf_t       *agf;   /* a.g. freespace header */
 965
 966                 agf = XFS_BUF_TO_AGF(cur->bc_private.a.agbp);
 967                 agno = be32_to_cpu(agf->agf_seqno);
 968                 agbno = be32_to_cpu(agf->agf_roots[cur->bc_btnum]);
 969         }
 970         /*
 971          * Iterate over each level in the btree, starting at the root.
 972          * For each level above the leaves, find the key we need, based
 973          * on the lookup record, then follow the corresponding block
 974          * pointer down to the next level.
 975          */
 976         for (level = cur->bc_nlevels - 1, diff = 1; level >= 0; level--) {
 977                 xfs_buf_t       *bp;    /* buffer pointer for btree block */
 978                 xfs_daddr_t     d;      /* disk address of btree block */
 979
 980                 /*
 981                  * Get the disk address we're looking for.
 982                  */
 983                 d = XFS_AGB_TO_DADDR(mp, agno, agbno);
 984                 /*
 985                  * If the old buffer at this level is for a different block,
 986                  * throw it away, otherwise just use it.
 987                  */
 988                 bp = cur->bc_bufs[level];
 989                 if (bp && XFS_BUF_ADDR(bp) != d)
 990                         bp = NULL;
 991                 if (!bp) {
 992                         /*
 993                          * Need to get a new buffer.  Read it, then
 994                          * set it in the cursor, releasing the old one.
 995                          */
 996                         if ((error = xfs_btree_read_bufs(mp, cur->bc_tp, agno,
 997                                         agbno, 0, &bp, XFS_ALLOC_BTREE_REF)))
 998                                 return error;
 999                         xfs_btree_setbuf(cur, level, bp);
1000                         /*
1001                          * Point to the btree block, now that we have the buffer
1002                          */
1003                         block = XFS_BUF_TO_ALLOC_BLOCK(bp);
1004                         if ((error = xfs_btree_check_sblock(cur, block, level,
1005                                         bp)))
1006                                 return error;
1007                 } else
1008                         block = XFS_BUF_TO_ALLOC_BLOCK(bp);
1009                 /*
1010                  * If we already had a key match at a higher level, we know
1011                  * we need to use the first entry in this block.
1012                  */
1013                 if (diff == 0)
1014                         keyno = 1;
1015                 /*
1016                  * Otherwise we need to search this block.  Do a binary search.
1017                  */
1018                 else {
1019                         int             high;   /* high entry number */
1020                         xfs_alloc_key_t *kkbase=NULL;/* base of keys in block */
1021                         xfs_alloc_rec_t *krbase=NULL;/* base of records in block */
1022                         int             low;    /* low entry number */
1023
1024                         /*
1025                          * Get a pointer to keys or records.
1026                          */
1027                         if (level > 0)
1028                                 kkbase = XFS_ALLOC_KEY_ADDR(block, 1, cur);
1029                         else
1030                                 krbase = XFS_ALLOC_REC_ADDR(block, 1, cur);
1031                         /*
1032                          * Set low and high entry numbers, 1-based.
1033                          */
1034                         low = 1;
1035                         if (!(high = be16_to_cpu(block->bb_numrecs))) {
1036                                 /*
1037                                  * If the block is empty, the tree must
1038                                  * be an empty leaf.
1039                                  */
1040                                 ASSERT(level == 0 && cur->bc_nlevels == 1);
1041                                 cur->bc_ptrs[0] = dir != XFS_LOOKUP_LE;
1042                                 *stat = 0;
1043                                 return 0;
1044                         }
1045                         /*
1046                          * Binary search the block.
1047                          */
1048                         while (low <= high) {
1049                                 xfs_extlen_t    blockcount;     /* key value */
1050                                 xfs_agblock_t   startblock;     /* key value */
1051
1052                                 XFS_STATS_INC(xs_abt_compare);
1053                                 /*
1054                                  * keyno is average of low and high.
1055                                  */
1056                                 keyno = (low + high) >> 1;
1057                                 /*
1058                                  * Get startblock & blockcount.
1059                                  */
1060                                 if (level > 0) {
1061                                         xfs_alloc_key_t *kkp;
1062
1063                                         kkp = kkbase + keyno - 1;
1064                                         startblock = be32_to_cpu(kkp->ar_startblock);
1065                                         blockcount = be32_to_cpu(kkp->ar_blockcount);
1066                                 } else {
1067                                         xfs_alloc_rec_t *krp;
1068
1069                                         krp = krbase + keyno - 1;
1070                                         startblock = be32_to_cpu(krp->ar_startblock);
1071                                         blockcount = be32_to_cpu(krp->ar_blockcount);
1072                                 }
1073                                 /*
1074                                  * Compute difference to get next direction.
1075                                  */
1076                                 if (cur->bc_btnum == XFS_BTNUM_BNO)
1077                                         diff = (int)startblock -
1078                                                (int)cur->bc_rec.a.ar_startblock;
1079                                 else if (!(diff = (int)blockcount -
1080                                             (int)cur->bc_rec.a.ar_blockcount))
1081                                         diff = (int)startblock -
1082                                             (int)cur->bc_rec.a.ar_startblock;
1083                                 /*
1084                                  * Less than, move right.
1085                                  */
1086                                 if (diff < 0)
1087                                         low = keyno + 1;
1088                                 /*
1089                                  * Greater than, move left.
1090                                  */
1091                                 else if (diff > 0)
1092                                         high = keyno - 1;
1093                                 /*
1094                                  * Equal, we're done.
1095                                  */
1096                                 else
1097                                         break;
1098                         }
1099                 }
1100                 /*
1101                  * If there are more levels, set up for the next level
1102                  * by getting the block number and filling in the cursor.
1103                  */
1104                 if (level > 0) {
1105                         /*
1106                          * If we moved left, need the previous key number,
1107                          * unless there isn't one.
1108                          */
1109                         if (diff > 0 && --keyno < 1)
1110                                 keyno = 1;
1111                         agbno = be32_to_cpu(*XFS_ALLOC_PTR_ADDR(block, keyno, cur));
1112 #ifdef DEBUG
1113                         if ((error = xfs_btree_check_sptr(cur, agbno, level)))
1114                                 return error;
1115 #endif
1116                         cur->bc_ptrs[level] = keyno;
1117                 }
1118         }
1119         /*
1120          * Done with the search.
1121          * See if we need to adjust the results.
1122          */
1123         if (dir != XFS_LOOKUP_LE && diff < 0) {
1124                 keyno++;
1125                 /*
1126                  * If ge search and we went off the end of the block, but it's
1127                  * not the last block, we're in the wrong block.
1128                  */
1129                 if (dir == XFS_LOOKUP_GE &&
1130                     keyno > be16_to_cpu(block->bb_numrecs) &&
1131                     be32_to_cpu(block->bb_rightsib) != NULLAGBLOCK) {
1132                         int     i;
1133
1134                         cur->bc_ptrs[0] = keyno;
1135                         if ((error = xfs_alloc_increment(cur, 0, &i)))
1136                                 return error;
1137                         XFS_WANT_CORRUPTED_RETURN(i == 1);
1138                         *stat = 1;
1139                         return 0;
1140                 }
1141         }
1142         else if (dir == XFS_LOOKUP_LE && diff > 0)
1143                 keyno--;
1144         cur->bc_ptrs[0] = keyno;
1145         /*
1146          * Return if we succeeded or not.
1147          */
1148         if (keyno == 0 || keyno > be16_to_cpu(block->bb_numrecs))
1149                 *stat = 0;
1150         else
1151                 *stat = ((dir != XFS_LOOKUP_EQ) || (diff == 0));
1152         return 0;
1153 }
1154
1155 /*
1156  * Move 1 record left from cur/level if possible.
1157  * Update cur to reflect the new path.
1158  */
1159 STATIC int                              /* error */
1160 xfs_alloc_lshift(
1161         xfs_btree_cur_t         *cur,   /* btree cursor */
1162         int                     level,  /* level to shift record on */
1163         int                     *stat)  /* success/failure */
1164 {
1165         int                     error;  /* error return value */
1166 #ifdef DEBUG
1167         int                     i;      /* loop index */
1168 #endif
1169         xfs_alloc_key_t         key;    /* key value for leaf level upward */
1170         xfs_buf_t               *lbp;   /* buffer for left neighbor block */
1171         xfs_alloc_block_t       *left;  /* left neighbor btree block */
1172         int                     nrec;   /* new number of left block entries */
1173         xfs_buf_t               *rbp;   /* buffer for right (current) block */
1174         xfs_alloc_block_t       *right; /* right (current) btree block */
1175         xfs_alloc_key_t         *rkp=NULL;      /* key pointer for right block */
1176         xfs_alloc_ptr_t         *rpp=NULL;      /* address pointer for right block */
1177         xfs_alloc_rec_t         *rrp=NULL;      /* record pointer for right block */
1178
1179         /*
1180          * Set up variables for this block as "right".
1181          */
1182         rbp = cur->bc_bufs[level];
1183         right = XFS_BUF_TO_ALLOC_BLOCK(rbp);
1184 #ifdef DEBUG
1185         if ((error = xfs_btree_check_sblock(cur, right, level, rbp)))
1186                 return error;
1187 #endif
1188         /*
1189          * If we've got no left sibling then we can't shift an entry left.
1190          */
1191         if (be32_to_cpu(right->bb_leftsib) == NULLAGBLOCK) {
1192                 *stat = 0;
1193                 return 0;
1194         }
1195         /*
1196          * If the cursor entry is the one that would be moved, don't
1197          * do it... it's too complicated.
1198          */
1199         if (cur->bc_ptrs[level] <= 1) {
1200                 *stat = 0;
1201                 return 0;
1202         }
1203         /*
1204          * Set up the left neighbor as "left".
1205          */
1206         if ((error = xfs_btree_read_bufs(cur->bc_mp, cur->bc_tp,
1207                         cur->bc_private.a.agno, be32_to_cpu(right->bb_leftsib),
1208                         0, &lbp, XFS_ALLOC_BTREE_REF)))
1209                 return error;
1210         left = XFS_BUF_TO_ALLOC_BLOCK(lbp);
1211         if ((error = xfs_btree_check_sblock(cur, left, level, lbp)))
1212                 return error;
1213         /*
1214          * If it's full, it can't take another entry.
1215          */
1216         if (be16_to_cpu(left->bb_numrecs) == XFS_ALLOC_BLOCK_MAXRECS(level, cur)) {
1217                 *stat = 0;
1218                 return 0;
1219         }
1220         nrec = be16_to_cpu(left->bb_numrecs) + 1;
1221         /*
1222          * If non-leaf, copy a key and a ptr to the left block.
1223          */
1224         if (level > 0) {
1225                 xfs_alloc_key_t *lkp;   /* key pointer for left block */
1226                 xfs_alloc_ptr_t *lpp;   /* address pointer for left block */
1227
1228                 lkp = XFS_ALLOC_KEY_ADDR(left, nrec, cur);
1229                 rkp = XFS_ALLOC_KEY_ADDR(right, 1, cur);
1230                 *lkp = *rkp;
1231                 xfs_alloc_log_keys(cur, lbp, nrec, nrec);
1232                 lpp = XFS_ALLOC_PTR_ADDR(left, nrec, cur);
1233                 rpp = XFS_ALLOC_PTR_ADDR(right, 1, cur);
1234 #ifdef DEBUG
1235                 if ((error = xfs_btree_check_sptr(cur, be32_to_cpu(*rpp), level)))
1236                         return error;
1237 #endif
1238                 *lpp = *rpp;
1239                 xfs_alloc_log_ptrs(cur, lbp, nrec, nrec);
1240                 xfs_btree_check_key(cur->bc_btnum, lkp - 1, lkp);
1241         }
1242         /*
1243          * If leaf, copy a record to the left block.
1244          */
1245         else {
1246                 xfs_alloc_rec_t *lrp;   /* record pointer for left block */
1247
1248                 lrp = XFS_ALLOC_REC_ADDR(left, nrec, cur);
1249                 rrp = XFS_ALLOC_REC_ADDR(right, 1, cur);
1250                 *lrp = *rrp;
1251                 xfs_alloc_log_recs(cur, lbp, nrec, nrec);
1252                 xfs_btree_check_rec(cur->bc_btnum, lrp - 1, lrp);
1253         }
1254         /*
1255          * Bump and log left's numrecs, decrement and log right's numrecs.
1256          */
1257         be16_add(&left->bb_numrecs, 1);
1258         xfs_alloc_log_block(cur->bc_tp, lbp, XFS_BB_NUMRECS);
1259         be16_add(&right->bb_numrecs, -1);
1260         xfs_alloc_log_block(cur->bc_tp, rbp, XFS_BB_NUMRECS);
1261         /*
1262          * Slide the contents of right down one entry.
1263          */
1264         if (level > 0) {
1265 #ifdef DEBUG
1266                 for (i = 0; i < be16_to_cpu(right->bb_numrecs); i++) {
1267                         if ((error = xfs_btree_check_sptr(cur, be32_to_cpu(rpp[i + 1]),
1268                                         level)))
1269                                 return error;
1270                 }
1271 #endif
1272                 memmove(rkp, rkp + 1, be16_to_cpu(right->bb_numrecs) * sizeof(*rkp));
1273                 memmove(rpp, rpp + 1, be16_to_cpu(right->bb_numrecs) * sizeof(*rpp));
1274                 xfs_alloc_log_keys(cur, rbp, 1, be16_to_cpu(right->bb_numrecs));
1275                 xfs_alloc_log_ptrs(cur, rbp, 1, be16_to_cpu(right->bb_numrecs));
1276         } else {
1277                 memmove(rrp, rrp + 1, be16_to_cpu(right->bb_numrecs) * sizeof(*rrp));
1278                 xfs_alloc_log_recs(cur, rbp, 1, be16_to_cpu(right->bb_numrecs));
1279                 key.ar_startblock = rrp->ar_startblock;
1280                 key.ar_blockcount = rrp->ar_blockcount;
1281                 rkp = &key;
1282         }
1283         /*
1284          * Update the parent key values of right.
1285          */
1286         if ((error = xfs_alloc_updkey(cur, rkp, level + 1)))
1287                 return error;
1288         /*
1289          * Slide the cursor value left one.
1290          */
1291         cur->bc_ptrs[level]--;
1292         *stat = 1;
1293         return 0;
1294 }
1295
1296 /*
1297  * Allocate a new root block, fill it in.
1298  */
1299 STATIC int                              /* error */
1300 xfs_alloc_newroot(
1301         xfs_btree_cur_t         *cur,   /* btree cursor */
1302         int                     *stat)  /* success/failure */
1303 {
1304         int                     error;  /* error return value */
1305         xfs_agblock_t           lbno;   /* left block number */
1306         xfs_buf_t               *lbp;   /* left btree buffer */
1307         xfs_alloc_block_t       *left;  /* left btree block */
1308         xfs_mount_t             *mp;    /* mount structure */
1309         xfs_agblock_t           nbno;   /* new block number */
1310         xfs_buf_t               *nbp;   /* new (root) buffer */
1311         xfs_alloc_block_t       *new;   /* new (root) btree block */
1312         int                     nptr;   /* new value for key index, 1 or 2 */
1313         xfs_agblock_t           rbno;   /* right block number */
1314         xfs_buf_t               *rbp;   /* right btree buffer */
1315         xfs_alloc_block_t       *right; /* right btree block */
1316
1317         mp = cur->bc_mp;
1318
1319         ASSERT(cur->bc_nlevels < XFS_AG_MAXLEVELS(mp));
1320         /*
1321          * Get a buffer from the freelist blocks, for the new root.
1322          */
1323         if ((error = xfs_alloc_get_freelist(cur->bc_tp, cur->bc_private.a.agbp,
1324                         &nbno)))
1325                 return error;
1326         /*
1327          * None available, we fail.
1328          */
1329         if (nbno == NULLAGBLOCK) {
1330                 *stat = 0;
1331                 return 0;
1332         }
1333         xfs_trans_agbtree_delta(cur->bc_tp, 1);
1334         nbp = xfs_btree_get_bufs(mp, cur->bc_tp, cur->bc_private.a.agno, nbno,
1335                 0);
1336         new = XFS_BUF_TO_ALLOC_BLOCK(nbp);
1337         /*
1338          * Set the root data in the a.g. freespace structure.
1339          */
1340         {
1341                 xfs_agf_t       *agf;   /* a.g. freespace header */
1342                 xfs_agnumber_t  seqno;
1343
1344                 agf = XFS_BUF_TO_AGF(cur->bc_private.a.agbp);
1345                 agf->agf_roots[cur->bc_btnum] = cpu_to_be32(nbno);
1346                 be32_add(&agf->agf_levels[cur->bc_btnum], 1);
1347                 seqno = be32_to_cpu(agf->agf_seqno);
1348                 mp->m_perag[seqno].pagf_levels[cur->bc_btnum]++;
1349                 xfs_alloc_log_agf(cur->bc_tp, cur->bc_private.a.agbp,
1350                         XFS_AGF_ROOTS | XFS_AGF_LEVELS);
1351         }
1352         /*
1353          * At the previous root level there are now two blocks: the old
1354          * root, and the new block generated when it was split.
1355          * We don't know which one the cursor is pointing at, so we
1356          * set up variables "left" and "right" for each case.
1357          */
1358         lbp = cur->bc_bufs[cur->bc_nlevels - 1];
1359         left = XFS_BUF_TO_ALLOC_BLOCK(lbp);
1360 #ifdef DEBUG
1361         if ((error = xfs_btree_check_sblock(cur, left, cur->bc_nlevels - 1, lbp)))
1362                 return error;
1363 #endif
1364         if (be32_to_cpu(left->bb_rightsib) != NULLAGBLOCK) {
1365                 /*
1366                  * Our block is left, pick up the right block.
1367                  */
1368                 lbno = XFS_DADDR_TO_AGBNO(mp, XFS_BUF_ADDR(lbp));
1369                 rbno = be32_to_cpu(left->bb_rightsib);
1370                 if ((error = xfs_btree_read_bufs(mp, cur->bc_tp,
1371                                 cur->bc_private.a.agno, rbno, 0, &rbp,
1372                                 XFS_ALLOC_BTREE_REF)))
1373                         return error;
1374                 right = XFS_BUF_TO_ALLOC_BLOCK(rbp);
1375                 if ((error = xfs_btree_check_sblock(cur, right,
1376                                 cur->bc_nlevels - 1, rbp)))
1377                         return error;
1378                 nptr = 1;
1379         } else {
1380                 /*
1381                  * Our block is right, pick up the left block.
1382                  */
1383                 rbp = lbp;
1384                 right = left;
1385                 rbno = XFS_DADDR_TO_AGBNO(mp, XFS_BUF_ADDR(rbp));
1386                 lbno = be32_to_cpu(right->bb_leftsib);
1387                 if ((error = xfs_btree_read_bufs(mp, cur->bc_tp,
1388                                 cur->bc_private.a.agno, lbno, 0, &lbp,
1389                                 XFS_ALLOC_BTREE_REF)))
1390                         return error;
1391                 left = XFS_BUF_TO_ALLOC_BLOCK(lbp);
1392                 if ((error = xfs_btree_check_sblock(cur, left,
1393                                 cur->bc_nlevels - 1, lbp)))
1394                         return error;
1395                 nptr = 2;
1396         }
1397         /*
1398          * Fill in the new block's btree header and log it.
1399          */
1400         new->bb_magic = cpu_to_be32(xfs_magics[cur->bc_btnum]);
1401         new->bb_level = cpu_to_be16(cur->bc_nlevels);
1402         new->bb_numrecs = cpu_to_be16(2);
1403         new->bb_leftsib = cpu_to_be32(NULLAGBLOCK);
1404         new->bb_rightsib = cpu_to_be32(NULLAGBLOCK);
1405         xfs_alloc_log_block(cur->bc_tp, nbp, XFS_BB_ALL_BITS);
1406         ASSERT(lbno != NULLAGBLOCK && rbno != NULLAGBLOCK);
1407         /*
1408          * Fill in the key data in the new root.
1409          */
1410         {
1411                 xfs_alloc_key_t         *kp;    /* btree key pointer */
1412
1413                 kp = XFS_ALLOC_KEY_ADDR(new, 1, cur);
1414                 if (be16_to_cpu(left->bb_level) > 0) {
1415                         kp[0] = *XFS_ALLOC_KEY_ADDR(left, 1, cur);
1416                         kp[1] = *XFS_ALLOC_KEY_ADDR(right, 1, cur);
1417                 } else {
1418                         xfs_alloc_rec_t *rp;    /* btree record pointer */
1419
1420                         rp = XFS_ALLOC_REC_ADDR(left, 1, cur);
1421                         kp[0].ar_startblock = rp->ar_startblock;
1422                         kp[0].ar_blockcount = rp->ar_blockcount;
1423                         rp = XFS_ALLOC_REC_ADDR(right, 1, cur);
1424                         kp[1].ar_startblock = rp->ar_startblock;
1425                         kp[1].ar_blockcount = rp->ar_blockcount;
1426                 }
1427         }
1428         xfs_alloc_log_keys(cur, nbp, 1, 2);
1429         /*
1430          * Fill in the pointer data in the new root.
1431          */
1432         {
1433                 xfs_alloc_ptr_t         *pp;    /* btree address pointer */
1434
1435                 pp = XFS_ALLOC_PTR_ADDR(new, 1, cur);
1436                 pp[0] = cpu_to_be32(lbno);
1437                 pp[1] = cpu_to_be32(rbno);
1438         }
1439         xfs_alloc_log_ptrs(cur, nbp, 1, 2);
1440         /*
1441          * Fix up the cursor.
1442          */
1443         xfs_btree_setbuf(cur, cur->bc_nlevels, nbp);
1444         cur->bc_ptrs[cur->bc_nlevels] = nptr;
1445         cur->bc_nlevels++;
1446         *stat = 1;
1447         return 0;
1448 }
1449
1450 /*
1451  * Move 1 record right from cur/level if possible.
1452  * Update cur to reflect the new path.
1453  */
1454 STATIC int                              /* error */
1455 xfs_alloc_rshift(
1456         xfs_btree_cur_t         *cur,   /* btree cursor */
1457         int                     level,  /* level to shift record on */
1458         int                     *stat)  /* success/failure */
1459 {
1460         int                     error;  /* error return value */
1461         int                     i;      /* loop index */
1462         xfs_alloc_key_t         key;    /* key value for leaf level upward */
1463         xfs_buf_t               *lbp;   /* buffer for left (current) block */
1464         xfs_alloc_block_t       *left;  /* left (current) btree block */
1465         xfs_buf_t               *rbp;   /* buffer for right neighbor block */
1466         xfs_alloc_block_t       *right; /* right neighbor btree block */
1467         xfs_alloc_key_t         *rkp;   /* key pointer for right block */
1468         xfs_btree_cur_t         *tcur;  /* temporary cursor */
1469
1470         /*
1471          * Set up variables for this block as "left".
1472          */
1473         lbp = cur->bc_bufs[level];
1474         left = XFS_BUF_TO_ALLOC_BLOCK(lbp);
1475 #ifdef DEBUG
1476         if ((error = xfs_btree_check_sblock(cur, left, level, lbp)))
1477                 return error;
1478 #endif
1479         /*
1480          * If we've got no right sibling then we can't shift an entry right.
1481          */
1482         if (be32_to_cpu(left->bb_rightsib) == NULLAGBLOCK) {
1483                 *stat = 0;
1484                 return 0;
1485         }
1486         /*
1487          * If the cursor entry is the one that would be moved, don't
1488          * do it... it's too complicated.
1489          */
1490         if (cur->bc_ptrs[level] >= be16_to_cpu(left->bb_numrecs)) {
1491                 *stat = 0;
1492                 return 0;
1493         }
1494         /*
1495          * Set up the right neighbor as "right".
1496          */
1497         if ((error = xfs_btree_read_bufs(cur->bc_mp, cur->bc_tp,
1498                         cur->bc_private.a.agno, be32_to_cpu(left->bb_rightsib),
1499                         0, &rbp, XFS_ALLOC_BTREE_REF)))
1500                 return error;
1501         right = XFS_BUF_TO_ALLOC_BLOCK(rbp);
1502         if ((error = xfs_btree_check_sblock(cur, right, level, rbp)))
1503                 return error;
1504         /*
1505          * If it's full, it can't take another entry.
1506          */
1507         if (be16_to_cpu(right->bb_numrecs) == XFS_ALLOC_BLOCK_MAXRECS(level, cur)) {
1508                 *stat = 0;
1509                 return 0;
1510         }
1511         /*
1512          * Make a hole at the start of the right neighbor block, then
1513          * copy the last left block entry to the hole.
1514          */
1515         if (level > 0) {
1516                 xfs_alloc_key_t *lkp;   /* key pointer for left block */
1517                 xfs_alloc_ptr_t *lpp;   /* address pointer for left block */
1518                 xfs_alloc_ptr_t *rpp;   /* address pointer for right block */
1519
1520                 lkp = XFS_ALLOC_KEY_ADDR(left, be16_to_cpu(left->bb_numrecs), cur);
1521                 lpp = XFS_ALLOC_PTR_ADDR(left, be16_to_cpu(left->bb_numrecs), cur);
1522                 rkp = XFS_ALLOC_KEY_ADDR(right, 1, cur);
1523                 rpp = XFS_ALLOC_PTR_ADDR(right, 1, cur);
1524 #ifdef DEBUG
1525                 for (i = be16_to_cpu(right->bb_numrecs) - 1; i >= 0; i--) {
1526                         if ((error = xfs_btree_check_sptr(cur, be32_to_cpu(rpp[i]), level)))
1527                                 return error;
1528                 }
1529 #endif
1530                 memmove(rkp + 1, rkp, be16_to_cpu(right->bb_numrecs) * sizeof(*rkp));
1531                 memmove(rpp + 1, rpp, be16_to_cpu(right->bb_numrecs) * sizeof(*rpp));
1532 #ifdef DEBUG
1533                 if ((error = xfs_btree_check_sptr(cur, be32_to_cpu(*lpp), level)))
1534                         return error;
1535 #endif
1536                 *rkp = *lkp;
1537                 *rpp = *lpp;
1538                 xfs_alloc_log_keys(cur, rbp, 1, be16_to_cpu(right->bb_numrecs) + 1);
1539                 xfs_alloc_log_ptrs(cur, rbp, 1, be16_to_cpu(right->bb_numrecs) + 1);
1540                 xfs_btree_check_key(cur->bc_btnum, rkp, rkp + 1);
1541         } else {
1542                 xfs_alloc_rec_t *lrp;   /* record pointer for left block */
1543                 xfs_alloc_rec_t *rrp;   /* record pointer for right block */
1544
1545                 lrp = XFS_ALLOC_REC_ADDR(left, be16_to_cpu(left->bb_numrecs), cur);
1546                 rrp = XFS_ALLOC_REC_ADDR(right, 1, cur);
1547                 memmove(rrp + 1, rrp, be16_to_cpu(right->bb_numrecs) * sizeof(*rrp));
1548                 *rrp = *lrp;
1549                 xfs_alloc_log_recs(cur, rbp, 1, be16_to_cpu(right->bb_numrecs) + 1);
1550                 key.ar_startblock = rrp->ar_startblock;
1551                 key.ar_blockcount = rrp->ar_blockcount;
1552                 rkp = &key;
1553                 xfs_btree_check_rec(cur->bc_btnum, rrp, rrp + 1);
1554         }
1555         /*
1556          * Decrement and log left's numrecs, bump and log right's numrecs.
1557          */
1558         be16_add(&left->bb_numrecs, -1);
1559         xfs_alloc_log_block(cur->bc_tp, lbp, XFS_BB_NUMRECS);
1560         be16_add(&right->bb_numrecs, 1);
1561         xfs_alloc_log_block(cur->bc_tp, rbp, XFS_BB_NUMRECS);
1562         /*
1563          * Using a temporary cursor, update the parent key values of the
1564          * block on the right.
1565          */
1566         if ((error = xfs_btree_dup_cursor(cur, &tcur)))
1567                 return error;
1568         i = xfs_btree_lastrec(tcur, level);
1569         XFS_WANT_CORRUPTED_GOTO(i == 1, error0);
1570         if ((error = xfs_alloc_increment(tcur, level, &i)) ||
1571             (error = xfs_alloc_updkey(tcur, rkp, level + 1)))
1572                 goto error0;
1573         xfs_btree_del_cursor(tcur, XFS_BTREE_NOERROR);
1574         *stat = 1;
1575         return 0;
1576 error0:
1577         xfs_btree_del_cursor(tcur, XFS_BTREE_ERROR);
1578         return error;
1579 }
1580
1581 /*
1582  * Split cur/level block in half.
1583  * Return new block number and its first record (to be inserted into parent).
1584  */
1585 STATIC int                              /* error */
1586 xfs_alloc_split(
1587         xfs_btree_cur_t         *cur,   /* btree cursor */
1588         int                     level,  /* level to split */
1589         xfs_agblock_t           *bnop,  /* output: block number allocated */
1590         xfs_alloc_key_t         *keyp,  /* output: first key of new block */
1591         xfs_btree_cur_t         **curp, /* output: new cursor */
1592         int                     *stat)  /* success/failure */
1593 {
1594         int                     error;  /* error return value */
1595         int                     i;      /* loop index/record number */
1596         xfs_agblock_t           lbno;   /* left (current) block number */
1597         xfs_buf_t               *lbp;   /* buffer for left block */
1598         xfs_alloc_block_t       *left;  /* left (current) btree block */
1599         xfs_agblock_t           rbno;   /* right (new) block number */
1600         xfs_buf_t               *rbp;   /* buffer for right block */
1601         xfs_alloc_block_t       *right; /* right (new) btree block */
1602
1603         /*
1604          * Allocate the new block from the freelist.
1605          * If we can't do it, we're toast.  Give up.
1606          */
1607         if ((error = xfs_alloc_get_freelist(cur->bc_tp, cur->bc_private.a.agbp,
1608                         &rbno)))
1609                 return error;
1610         if (rbno == NULLAGBLOCK) {
1611                 *stat = 0;
1612                 return 0;
1613         }
1614         xfs_trans_agbtree_delta(cur->bc_tp, 1);
1615         rbp = xfs_btree_get_bufs(cur->bc_mp, cur->bc_tp, cur->bc_private.a.agno,
1616                 rbno, 0);
1617         /*
1618          * Set up the new block as "right".
1619          */
1620         right = XFS_BUF_TO_ALLOC_BLOCK(rbp);
1621         /*
1622          * "Left" is the current (according to the cursor) block.
1623          */
1624         lbp = cur->bc_bufs[level];
1625         left = XFS_BUF_TO_ALLOC_BLOCK(lbp);
1626 #ifdef DEBUG
1627         if ((error = xfs_btree_check_sblock(cur, left, level, lbp)))
1628                 return error;
1629 #endif
1630         /*
1631          * Fill in the btree header for the new block.
1632          */
1633         right->bb_magic = cpu_to_be32(xfs_magics[cur->bc_btnum]);
1634         right->bb_level = left->bb_level;
1635         right->bb_numrecs = cpu_to_be16(be16_to_cpu(left->bb_numrecs) / 2);
1636         /*
1637          * Make sure that if there's an odd number of entries now, that
1638          * each new block will have the same number of entries.
1639          */
1640         if ((be16_to_cpu(left->bb_numrecs) & 1) &&
1641             cur->bc_ptrs[level] <= be16_to_cpu(right->bb_numrecs) + 1)
1642                 be16_add(&right->bb_numrecs, 1);
1643         i = be16_to_cpu(left->bb_numrecs) - be16_to_cpu(right->bb_numrecs) + 1;
1644         /*
1645          * For non-leaf blocks, copy keys and addresses over to the new block.
1646          */
1647         if (level > 0) {
1648                 xfs_alloc_key_t *lkp;   /* left btree key pointer */
1649                 xfs_alloc_ptr_t *lpp;   /* left btree address pointer */
1650                 xfs_alloc_key_t *rkp;   /* right btree key pointer */
1651                 xfs_alloc_ptr_t *rpp;   /* right btree address pointer */
1652
1653                 lkp = XFS_ALLOC_KEY_ADDR(left, i, cur);
1654                 lpp = XFS_ALLOC_PTR_ADDR(left, i, cur);
1655                 rkp = XFS_ALLOC_KEY_ADDR(right, 1, cur);
1656                 rpp = XFS_ALLOC_PTR_ADDR(right, 1, cur);
1657 #ifdef DEBUG
1658                 for (i = 0; i < be16_to_cpu(right->bb_numrecs); i++) {
1659                         if ((error = xfs_btree_check_sptr(cur, be32_to_cpu(lpp[i]), level)))
1660                                 return error;
1661                 }
1662 #endif
1663                 memcpy(rkp, lkp, be16_to_cpu(right->bb_numrecs) * sizeof(*rkp));
1664                 memcpy(rpp, lpp, be16_to_cpu(right->bb_numrecs) * sizeof(*rpp));
1665                 xfs_alloc_log_keys(cur, rbp, 1, be16_to_cpu(right->bb_numrecs));
1666                 xfs_alloc_log_ptrs(cur, rbp, 1, be16_to_cpu(right->bb_numrecs));
1667                 *keyp = *rkp;
1668         }
1669         /*
1670          * For leaf blocks, copy records over to the new block.
1671          */
1672         else {
1673                 xfs_alloc_rec_t *lrp;   /* left btree record pointer */
1674                 xfs_alloc_rec_t *rrp;   /* right btree record pointer */
1675
1676                 lrp = XFS_ALLOC_REC_ADDR(left, i, cur);
1677                 rrp = XFS_ALLOC_REC_ADDR(right, 1, cur);
1678                 memcpy(rrp, lrp, be16_to_cpu(right->bb_numrecs) * sizeof(*rrp));
1679                 xfs_alloc_log_recs(cur, rbp, 1, be16_to_cpu(right->bb_numrecs));
1680                 keyp->ar_startblock = rrp->ar_startblock;
1681                 keyp->ar_blockcount = rrp->ar_blockcount;
1682         }
1683         /*
1684          * Find the left block number by looking in the buffer.
1685          * Adjust numrecs, sibling pointers.
1686          */
1687         lbno = XFS_DADDR_TO_AGBNO(cur->bc_mp, XFS_BUF_ADDR(lbp));
1688         be16_add(&left->bb_numrecs, -(be16_to_cpu(right->bb_numrecs)));
1689         right->bb_rightsib = left->bb_rightsib;
1690         left->bb_rightsib = cpu_to_be32(rbno);
1691         right->bb_leftsib = cpu_to_be32(lbno);
1692         xfs_alloc_log_block(cur->bc_tp, rbp, XFS_BB_ALL_BITS);
1693         xfs_alloc_log_block(cur->bc_tp, lbp, XFS_BB_NUMRECS | XFS_BB_RIGHTSIB);
1694         /*
1695          * If there's a block to the new block's right, make that block
1696          * point back to right instead of to left.
1697          */
1698         if (be32_to_cpu(right->bb_rightsib) != NULLAGBLOCK) {
1699                 xfs_alloc_block_t       *rrblock;       /* rr btree block */
1700                 xfs_buf_t               *rrbp;          /* buffer for rrblock */
1701
1702                 if ((error = xfs_btree_read_bufs(cur->bc_mp, cur->bc_tp,
1703                                 cur->bc_private.a.agno, be32_to_cpu(right->bb_rightsib), 0,
1704                                 &rrbp, XFS_ALLOC_BTREE_REF)))
1705                         return error;
1706                 rrblock = XFS_BUF_TO_ALLOC_BLOCK(rrbp);
1707                 if ((error = xfs_btree_check_sblock(cur, rrblock, level, rrbp)))
1708                         return error;
1709                 rrblock->bb_leftsib = cpu_to_be32(rbno);
1710                 xfs_alloc_log_block(cur->bc_tp, rrbp, XFS_BB_LEFTSIB);
1711         }
1712         /*
1713          * If the cursor is really in the right block, move it there.
1714          * If it's just pointing past the last entry in left, then we'll
1715          * insert there, so don't change anything in that case.
1716          */
1717         if (cur->bc_ptrs[level] > be16_to_cpu(left->bb_numrecs) + 1) {
1718                 xfs_btree_setbuf(cur, level, rbp);
1719                 cur->bc_ptrs[level] -= be16_to_cpu(left->bb_numrecs);
1720         }
1721         /*
1722          * If there are more levels, we'll need another cursor which refers to
1723          * the right block, no matter where this cursor was.
1724          */
1725         if (level + 1 < cur->bc_nlevels) {
1726                 if ((error = xfs_btree_dup_cursor(cur, curp)))
1727                         return error;
1728                 (*curp)->bc_ptrs[level + 1]++;
1729         }
1730         *bnop = rbno;
1731         *stat = 1;
1732         return 0;
1733 }
1734
1735 /*
1736  * Update keys at all levels from here to the root along the cursor's path.
1737  */
1738 STATIC int                              /* error */
1739 xfs_alloc_updkey(
1740         xfs_btree_cur_t         *cur,   /* btree cursor */
1741         xfs_alloc_key_t         *keyp,  /* new key value to update to */
1742         int                     level)  /* starting level for update */
1743 {
1744         int                     ptr;    /* index of key in block */
1745
1746         /*
1747          * Go up the tree from this level toward the root.
1748          * At each level, update the key value to the value input.
1749          * Stop when we reach a level where the cursor isn't pointing
1750          * at the first entry in the block.
1751          */
1752         for (ptr = 1; ptr == 1 && level < cur->bc_nlevels; level++) {
1753                 xfs_alloc_block_t       *block; /* btree block */
1754                 xfs_buf_t               *bp;    /* buffer for block */
1755 #ifdef DEBUG
1756                 int                     error;  /* error return value */
1757 #endif
1758                 xfs_alloc_key_t         *kp;    /* ptr to btree block keys */
1759
1760                 bp = cur->bc_bufs[level];
1761                 block = XFS_BUF_TO_ALLOC_BLOCK(bp);
1762 #ifdef DEBUG
1763                 if ((error = xfs_btree_check_sblock(cur, block, level, bp)))
1764                         return error;
1765 #endif
1766                 ptr = cur->bc_ptrs[level];
1767                 kp = XFS_ALLOC_KEY_ADDR(block, ptr, cur);
1768                 *kp = *keyp;
1769                 xfs_alloc_log_keys(cur, bp, ptr, ptr);
1770         }
1771         return 0;
1772 }
1773
1774 /*
1775  * Externally visible routines.
1776  */
1777
1778 /*
1779  * Decrement cursor by one record at the level.
1780  * For nonzero levels the leaf-ward information is untouched.
1781  */
1782 int                                     /* error */
1783 xfs_alloc_decrement(
1784         xfs_btree_cur_t         *cur,   /* btree cursor */
1785         int                     level,  /* level in btree, 0 is leaf */
1786         int                     *stat)  /* success/failure */
1787 {
1788         xfs_alloc_block_t       *block; /* btree block */
1789         int                     error;  /* error return value */
1790         int                     lev;    /* btree level */
1791
1792         ASSERT(level < cur->bc_nlevels);
1793         /*
1794          * Read-ahead to the left at this level.
1795          */
1796         xfs_btree_readahead(cur, level, XFS_BTCUR_LEFTRA);
1797         /*
1798          * Decrement the ptr at this level.  If we're still in the block
1799          * then we're done.
1800          */
1801         if (--cur->bc_ptrs[level] > 0) {
1802                 *stat = 1;
1803                 return 0;
1804         }
1805         /*
1806          * Get a pointer to the btree block.
1807          */
1808         block = XFS_BUF_TO_ALLOC_BLOCK(cur->bc_bufs[level]);
1809 #ifdef DEBUG
1810         if ((error = xfs_btree_check_sblock(cur, block, level,
1811                         cur->bc_bufs[level])))
1812                 return error;
1813 #endif
1814         /*
1815          * If we just went off the left edge of the tree, return failure.
1816          */
1817         if (be32_to_cpu(block->bb_leftsib) == NULLAGBLOCK) {
1818                 *stat = 0;
1819                 return 0;
1820         }
1821         /*
1822          * March up the tree decrementing pointers.
1823          * Stop when we don't go off the left edge of a block.
1824          */
1825         for (lev = level + 1; lev < cur->bc_nlevels; lev++) {
1826                 if (--cur->bc_ptrs[lev] > 0)
1827                         break;
1828                 /*
1829                  * Read-ahead the left block, we're going to read it
1830                  * in the next loop.
1831                  */
1832                 xfs_btree_readahead(cur, lev, XFS_BTCUR_LEFTRA);
1833         }
1834         /*
1835          * If we went off the root then we are seriously confused.
1836          */
1837         ASSERT(lev < cur->bc_nlevels);
1838         /*
1839          * Now walk back down the tree, fixing up the cursor's buffer
1840          * pointers and key numbers.
1841          */
1842         for (block = XFS_BUF_TO_ALLOC_BLOCK(cur->bc_bufs[lev]); lev > level; ) {
1843                 xfs_agblock_t   agbno;  /* block number of btree block */
1844                 xfs_buf_t       *bp;    /* buffer pointer for block */
1845
1846                 agbno = be32_to_cpu(*XFS_ALLOC_PTR_ADDR(block, cur->bc_ptrs[lev], cur));
1847                 if ((error = xfs_btree_read_bufs(cur->bc_mp, cur->bc_tp,
1848                                 cur->bc_private.a.agno, agbno, 0, &bp,
1849                                 XFS_ALLOC_BTREE_REF)))
1850                         return error;
1851                 lev--;
1852                 xfs_btree_setbuf(cur, lev, bp);
1853                 block = XFS_BUF_TO_ALLOC_BLOCK(bp);
1854                 if ((error = xfs_btree_check_sblock(cur, block, lev, bp)))
1855                         return error;
1856                 cur->bc_ptrs[lev] = be16_to_cpu(block->bb_numrecs);
1857         }
1858         *stat = 1;
1859         return 0;
1860 }
1861
1862 /*
1863  * Delete the record pointed to by cur.
1864  * The cursor refers to the place where the record was (could be inserted)
1865  * when the operation returns.
1866  */
1867 int                                     /* error */
1868 xfs_alloc_delete(
1869         xfs_btree_cur_t *cur,           /* btree cursor */
1870         int             *stat)          /* success/failure */
1871 {
1872         int             error;          /* error return value */
1873         int             i;              /* result code */
1874         int             level;          /* btree level */
1875
1876         /*
1877          * Go up the tree, starting at leaf level.
1878          * If 2 is returned then a join was done; go to the next level.
1879          * Otherwise we are done.
1880          */
1881         for (level = 0, i = 2; i == 2; level++) {
1882                 if ((error = xfs_alloc_delrec(cur, level, &i)))
1883                         return error;
1884         }
1885         if (i == 0) {
1886                 for (level = 1; level < cur->bc_nlevels; level++) {
1887                         if (cur->bc_ptrs[level] == 0) {
1888                                 if ((error = xfs_alloc_decrement(cur, level, &i)))
1889                                         return error;
1890                                 break;
1891                         }
1892                 }
1893         }
1894         *stat = i;
1895         return 0;
1896 }
1897
1898 /*
1899  * Get the data from the pointed-to record.
1900  */
1901 int                                     /* error */
1902 xfs_alloc_get_rec(
1903         xfs_btree_cur_t         *cur,   /* btree cursor */
1904         xfs_agblock_t           *bno,   /* output: starting block of extent */
1905         xfs_extlen_t            *len,   /* output: length of extent */
1906         int                     *stat)  /* output: success/failure */
1907 {
1908         xfs_alloc_block_t       *block; /* btree block */
1909 #ifdef DEBUG
1910         int                     error;  /* error return value */
1911 #endif
1912         int                     ptr;    /* record number */
1913
1914         ptr = cur->bc_ptrs[0];
1915         block = XFS_BUF_TO_ALLOC_BLOCK(cur->bc_bufs[0]);
1916 #ifdef DEBUG
1917         if ((error = xfs_btree_check_sblock(cur, block, 0, cur->bc_bufs[0])))
1918                 return error;
1919 #endif
1920         /*
1921          * Off the right end or left end, return failure.
1922          */
1923         if (ptr > be16_to_cpu(block->bb_numrecs) || ptr <= 0) {
1924                 *stat = 0;
1925                 return 0;
1926         }
1927         /*
1928          * Point to the record and extract its data.
1929          */
1930         {
1931                 xfs_alloc_rec_t         *rec;   /* record data */
1932
1933                 rec = XFS_ALLOC_REC_ADDR(block, ptr, cur);
1934                 *bno = be32_to_cpu(rec->ar_startblock);
1935                 *len = be32_to_cpu(rec->ar_blockcount);
1936         }
1937         *stat = 1;
1938         return 0;
1939 }
1940
1941 /*
1942  * Increment cursor by one record at the level.
1943  * For nonzero levels the leaf-ward information is untouched.
1944  */
1945 int                                     /* error */
1946 xfs_alloc_increment(
1947         xfs_btree_cur_t         *cur,   /* btree cursor */
1948         int                     level,  /* level in btree, 0 is leaf */
1949         int                     *stat)  /* success/failure */
1950 {
1951         xfs_alloc_block_t       *block; /* btree block */
1952         xfs_buf_t               *bp;    /* tree block buffer */
1953         int                     error;  /* error return value */
1954         int                     lev;    /* btree level */
1955
1956         ASSERT(level < cur->bc_nlevels);
1957         /*
1958          * Read-ahead to the right at this level.
1959          */
1960         xfs_btree_readahead(cur, level, XFS_BTCUR_RIGHTRA);
1961         /*
1962          * Get a pointer to the btree block.
1963          */
1964         bp = cur->bc_bufs[level];
1965         block = XFS_BUF_TO_ALLOC_BLOCK(bp);
1966 #ifdef DEBUG
1967         if ((error = xfs_btree_check_sblock(cur, block, level, bp)))
1968                 return error;
1969 #endif
1970         /*
1971          * Increment the ptr at this level.  If we're still in the block
1972          * then we're done.
1973          */
1974         if (++cur->bc_ptrs[level] <= be16_to_cpu(block->bb_numrecs)) {
1975                 *stat = 1;
1976                 return 0;
1977         }
1978         /*
1979          * If we just went off the right edge of the tree, return failure.
1980          */
1981         if (be32_to_cpu(block->bb_rightsib) == NULLAGBLOCK) {
1982                 *stat = 0;
1983                 return 0;
1984         }
1985         /*
1986          * March up the tree incrementing pointers.
1987          * Stop when we don't go off the right edge of a block.
1988          */
1989         for (lev = level + 1; lev < cur->bc_nlevels; lev++) {
1990                 bp = cur->bc_bufs[lev];
1991                 block = XFS_BUF_TO_ALLOC_BLOCK(bp);
1992 #ifdef DEBUG
1993                 if ((error = xfs_btree_check_sblock(cur, block, lev, bp)))
1994                         return error;
1995 #endif
1996                 if (++cur->bc_ptrs[lev] <= be16_to_cpu(block->bb_numrecs))
1997                         break;
1998                 /*
1999                  * Read-ahead the right block, we're going to read it
2000                  * in the next loop.
2001                  */
2002                 xfs_btree_readahead(cur, lev, XFS_BTCUR_RIGHTRA);
2003         }
2004         /*
2005          * If we went off the root then we are seriously confused.
2006          */
2007         ASSERT(lev < cur->bc_nlevels);
2008         /*
2009          * Now walk back down the tree, fixing up the cursor's buffer
2010          * pointers and key numbers.
2011          */
2012         for (bp = cur->bc_bufs[lev], block = XFS_BUF_TO_ALLOC_BLOCK(bp);
2013              lev > level; ) {
2014                 xfs_agblock_t   agbno;  /* block number of btree block */
2015
2016                 agbno = be32_to_cpu(*XFS_ALLOC_PTR_ADDR(block, cur->bc_ptrs[lev], cur));
2017                 if ((error = xfs_btree_read_bufs(cur->bc_mp, cur->bc_tp,
2018                                 cur->bc_private.a.agno, agbno, 0, &bp,
2019                                 XFS_ALLOC_BTREE_REF)))
2020                         return error;
2021                 lev--;
2022                 xfs_btree_setbuf(cur, lev, bp);
2023                 block = XFS_BUF_TO_ALLOC_BLOCK(bp);
2024                 if ((error = xfs_btree_check_sblock(cur, block, lev, bp)))
2025                         return error;
2026                 cur->bc_ptrs[lev] = 1;
2027         }
2028         *stat = 1;
2029         return 0;
2030 }
2031
2032 /*
2033  * Insert the current record at the point referenced by cur.
2034  * The cursor may be inconsistent on return if splits have been done.
2035  */
2036 int                                     /* error */
2037 xfs_alloc_insert(
2038         xfs_btree_cur_t *cur,           /* btree cursor */
2039         int             *stat)          /* success/failure */
2040 {
2041         int             error;          /* error return value */
2042         int             i;              /* result value, 0 for failure */
2043         int             level;          /* current level number in btree */
2044         xfs_agblock_t   nbno;           /* new block number (split result) */
2045         xfs_btree_cur_t *ncur;          /* new cursor (split result) */
2046         xfs_alloc_rec_t nrec;           /* record being inserted this level */
2047         xfs_btree_cur_t *pcur;          /* previous level's cursor */
2048
2049         level = 0;
2050         nbno = NULLAGBLOCK;
2051         nrec.ar_startblock = cpu_to_be32(cur->bc_rec.a.ar_startblock);
2052         nrec.ar_blockcount = cpu_to_be32(cur->bc_rec.a.ar_blockcount);
2053         ncur = NULL;
2054         pcur = cur;
2055         /*
2056          * Loop going up the tree, starting at the leaf level.
2057          * Stop when we don't get a split block, that must mean that
2058          * the insert is finished with this level.
2059          */
2060         do {
2061                 /*
2062                  * Insert nrec/nbno into this level of the tree.
2063                  * Note if we fail, nbno will be null.
2064                  */
2065                 if ((error = xfs_alloc_insrec(pcur, level++, &nbno, &nrec, &ncur,
2066                                 &i))) {
2067                         if (pcur != cur)
2068                                 xfs_btree_del_cursor(pcur, XFS_BTREE_ERROR);
2069                         return error;
2070                 }
2071                 /*
2072                  * See if the cursor we just used is trash.
2073                  * Can't trash the caller's cursor, but otherwise we should
2074                  * if ncur is a new cursor or we're about to be done.
2075                  */
2076                 if (pcur != cur && (ncur || nbno == NULLAGBLOCK)) {
2077                         cur->bc_nlevels = pcur->bc_nlevels;
2078                         xfs_btree_del_cursor(pcur, XFS_BTREE_NOERROR);
2079                 }
2080                 /*
2081                  * If we got a new cursor, switch to it.
2082                  */
2083                 if (ncur) {
2084                         pcur = ncur;
2085                         ncur = NULL;
2086                 }
2087         } while (nbno != NULLAGBLOCK);
2088         *stat = i;
2089         return 0;
2090 }
2091
2092 /*
2093  * Lookup the record equal to [bno, len] in the btree given by cur.
2094  */
2095 int                                     /* error */
2096 xfs_alloc_lookup_eq(
2097         xfs_btree_cur_t *cur,           /* btree cursor */
2098         xfs_agblock_t   bno,            /* starting block of extent */
2099         xfs_extlen_t    len,            /* length of extent */
2100         int             *stat)          /* success/failure */
2101 {
2102         cur->bc_rec.a.ar_startblock = bno;
2103         cur->bc_rec.a.ar_blockcount = len;
2104         return xfs_alloc_lookup(cur, XFS_LOOKUP_EQ, stat);
2105 }
2106
2107 /*
2108  * Lookup the first record greater than or equal to [bno, len]
2109  * in the btree given by cur.
2110  */
2111 int                                     /* error */
2112 xfs_alloc_lookup_ge(
2113         xfs_btree_cur_t *cur,           /* btree cursor */
2114         xfs_agblock_t   bno,            /* starting block of extent */
2115         xfs_extlen_t    len,            /* length of extent */
2116         int             *stat)          /* success/failure */
2117 {
2118         cur->bc_rec.a.ar_startblock = bno;
2119         cur->bc_rec.a.ar_blockcount = len;
2120         return xfs_alloc_lookup(cur, XFS_LOOKUP_GE, stat);
2121 }
2122
2123 /*
2124  * Lookup the first record less than or equal to [bno, len]
2125  * in the btree given by cur.
2126  */
2127 int                                     /* error */
2128 xfs_alloc_lookup_le(
2129         xfs_btree_cur_t *cur,           /* btree cursor */
2130         xfs_agblock_t   bno,            /* starting block of extent */
2131         xfs_extlen_t    len,            /* length of extent */
2132         int             *stat)          /* success/failure */
2133 {
2134         cur->bc_rec.a.ar_startblock = bno;
2135         cur->bc_rec.a.ar_blockcount = len;
2136         return xfs_alloc_lookup(cur, XFS_LOOKUP_LE, stat);
2137 }
2138
2139 /*
2140  * Update the record referred to by cur, to the value given by [bno, len].
2141  * This either works (return 0) or gets an EFSCORRUPTED error.
2142  */
2143 int                                     /* error */
2144 xfs_alloc_update(
2145         xfs_btree_cur_t         *cur,   /* btree cursor */
2146         xfs_agblock_t           bno,    /* starting block of extent */
2147         xfs_extlen_t            len)    /* length of extent */
2148 {
2149         xfs_alloc_block_t       *block; /* btree block to update */
2150         int                     error;  /* error return value */
2151         int                     ptr;    /* current record number (updating) */
2152
2153         ASSERT(len > 0);
2154         /*
2155          * Pick up the a.g. freelist struct and the current block.
2156          */
2157         block = XFS_BUF_TO_ALLOC_BLOCK(cur->bc_bufs[0]);
2158 #ifdef DEBUG
2159         if ((error = xfs_btree_check_sblock(cur, block, 0, cur->bc_bufs[0])))
2160                 return error;
2161 #endif
2162         /*
2163          * Get the address of the rec to be updated.
2164          */
2165         ptr = cur->bc_ptrs[0];
2166         {
2167                 xfs_alloc_rec_t         *rp;    /* pointer to updated record */
2168
2169                 rp = XFS_ALLOC_REC_ADDR(block, ptr, cur);
2170                 /*
2171                  * Fill in the new contents and log them.
2172                  */
2173                 rp->ar_startblock = cpu_to_be32(bno);
2174                 rp->ar_blockcount = cpu_to_be32(len);
2175                 xfs_alloc_log_recs(cur, cur->bc_bufs[0], ptr, ptr);
2176         }
2177         /*
2178          * If it's the by-size btree and it's the last leaf block and
2179          * it's the last record... then update the size of the longest
2180          * extent in the a.g., which we cache in the a.g. freelist header.
2181          */
2182         if (cur->bc_btnum == XFS_BTNUM_CNT &&
2183             be32_to_cpu(block->bb_rightsib) == NULLAGBLOCK &&
2184             ptr == be16_to_cpu(block->bb_numrecs)) {
2185                 xfs_agf_t       *agf;   /* a.g. freespace header */
2186                 xfs_agnumber_t  seqno;
2187
2188                 agf = XFS_BUF_TO_AGF(cur->bc_private.a.agbp);
2189                 seqno = be32_to_cpu(agf->agf_seqno);
2190                 cur->bc_mp->m_perag[seqno].pagf_longest = len;
2191                 agf->agf_longest = cpu_to_be32(len);
2192                 xfs_alloc_log_agf(cur->bc_tp, cur->bc_private.a.agbp,
2193                         XFS_AGF_LONGEST);
2194         }
2195         /*
2196          * Updating first record in leaf. Pass new key value up to our parent.
2197          */
2198         if (ptr == 1) {
2199                 xfs_alloc_key_t key;    /* key containing [bno, len] */
2200
2201                 key.ar_startblock = cpu_to_be32(bno);
2202                 key.ar_blockcount = cpu_to_be32(len);
2203                 if ((error = xfs_alloc_updkey(cur, &key, 1)))
2204                         return error;
2205         }
2206         return 0;
2207 }