fs/xfs/xfs_bmap_btree.c

   1 /*
   2  * Copyright (c) 2000-2003,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_inode_item.h"
  38 #include "xfs_alloc.h"
  39 #include "xfs_btree.h"
  40 #include "xfs_btree_trace.h"
  41 #include "xfs_ialloc.h"
  42 #include "xfs_itable.h"
  43 #include "xfs_bmap.h"
  44 #include "xfs_error.h"
  45 #include "xfs_quota.h"
  46
  47 /*
  48  * Determine the extent state.
  49  */
  50 /* ARGSUSED */
  51 STATIC xfs_exntst_t
  52 xfs_extent_state(
  53         xfs_filblks_t           blks,
  54         int                     extent_flag)
  55 {
  56         if (extent_flag) {
  57                 ASSERT(blks != 0);      /* saved for DMIG */
  58                 return XFS_EXT_UNWRITTEN;
  59         }
  60         return XFS_EXT_NORM;
  61 }
  62
  63 /*
  64  * Convert on-disk form of btree root to in-memory form.
  65  */
  66 void
  67 xfs_bmdr_to_bmbt(
  68         struct xfs_mount        *mp,
  69         xfs_bmdr_block_t        *dblock,
  70         int                     dblocklen,
  71         struct xfs_btree_block  *rblock,
  72         int                     rblocklen)
  73 {
  74         int                     dmxr;
  75         xfs_bmbt_key_t          *fkp;
  76         __be64                  *fpp;
  77         xfs_bmbt_key_t          *tkp;
  78         __be64                  *tpp;
  79
  80         rblock->bb_magic = cpu_to_be32(XFS_BMAP_MAGIC);
  81         rblock->bb_level = dblock->bb_level;
  82         ASSERT(be16_to_cpu(rblock->bb_level) > 0);
  83         rblock->bb_numrecs = dblock->bb_numrecs;
  84         rblock->bb_u.l.bb_leftsib = cpu_to_be64(NULLDFSBNO);
  85         rblock->bb_u.l.bb_rightsib = cpu_to_be64(NULLDFSBNO);
  86         dmxr = xfs_bmdr_maxrecs(mp, dblocklen, 0);
  87         fkp = XFS_BMDR_KEY_ADDR(dblock, 1);
  88         tkp = XFS_BMBT_KEY_ADDR(mp, rblock, 1);
  89         fpp = XFS_BMDR_PTR_ADDR(dblock, 1, dmxr);
  90         tpp = XFS_BMAP_BROOT_PTR_ADDR(mp, rblock, 1, rblocklen);
  91         dmxr = be16_to_cpu(dblock->bb_numrecs);
  92         memcpy(tkp, fkp, sizeof(*fkp) * dmxr);
  93         memcpy(tpp, fpp, sizeof(*fpp) * dmxr);
  94 }
  95
  96 /*
  97  * Convert a compressed bmap extent record to an uncompressed form.
  98  * This code must be in sync with the routines xfs_bmbt_get_startoff,
  99  * xfs_bmbt_get_startblock, xfs_bmbt_get_blockcount and xfs_bmbt_get_state.
 100  */
 101
 102 STATIC_INLINE void
 103 __xfs_bmbt_get_all(
 104                 __uint64_t l0,
 105                 __uint64_t l1,
 106                 xfs_bmbt_irec_t *s)
 107 {
 108         int     ext_flag;
 109         xfs_exntst_t st;
 110
 111         ext_flag = (int)(l0 >> (64 - BMBT_EXNTFLAG_BITLEN));
 112         s->br_startoff = ((xfs_fileoff_t)l0 &
 113                            xfs_mask64lo(64 - BMBT_EXNTFLAG_BITLEN)) >> 9;
 114 #if XFS_BIG_BLKNOS
 115         s->br_startblock = (((xfs_fsblock_t)l0 & xfs_mask64lo(9)) << 43) |
 116                            (((xfs_fsblock_t)l1) >> 21);
 117 #else
 118 #ifdef DEBUG
 119         {
 120                 xfs_dfsbno_t    b;
 121
 122                 b = (((xfs_dfsbno_t)l0 & xfs_mask64lo(9)) << 43) |
 123                     (((xfs_dfsbno_t)l1) >> 21);
 124                 ASSERT((b >> 32) == 0 || isnulldstartblock(b));
 125                 s->br_startblock = (xfs_fsblock_t)b;
 126         }
 127 #else   /* !DEBUG */
 128         s->br_startblock = (xfs_fsblock_t)(((xfs_dfsbno_t)l1) >> 21);
 129 #endif  /* DEBUG */
 130 #endif  /* XFS_BIG_BLKNOS */
 131         s->br_blockcount = (xfs_filblks_t)(l1 & xfs_mask64lo(21));
 132         /* This is xfs_extent_state() in-line */
 133         if (ext_flag) {
 134                 ASSERT(s->br_blockcount != 0);  /* saved for DMIG */
 135                 st = XFS_EXT_UNWRITTEN;
 136         } else
 137                 st = XFS_EXT_NORM;
 138         s->br_state = st;
 139 }
 140
 141 void
 142 xfs_bmbt_get_all(
 143         xfs_bmbt_rec_host_t *r,
 144         xfs_bmbt_irec_t *s)
 145 {
 146         __xfs_bmbt_get_all(r->l0, r->l1, s);
 147 }
 148
 149 /*
 150  * Extract the blockcount field from an in memory bmap extent record.
 151  */
 152 xfs_filblks_t
 153 xfs_bmbt_get_blockcount(
 154         xfs_bmbt_rec_host_t     *r)
 155 {
 156         return (xfs_filblks_t)(r->l1 & xfs_mask64lo(21));
 157 }
 158
 159 /*
 160  * Extract the startblock field from an in memory bmap extent record.
 161  */
 162 xfs_fsblock_t
 163 xfs_bmbt_get_startblock(
 164         xfs_bmbt_rec_host_t     *r)
 165 {
 166 #if XFS_BIG_BLKNOS
 167         return (((xfs_fsblock_t)r->l0 & xfs_mask64lo(9)) << 43) |
 168                (((xfs_fsblock_t)r->l1) >> 21);
 169 #else
 170 #ifdef DEBUG
 171         xfs_dfsbno_t    b;
 172
 173         b = (((xfs_dfsbno_t)r->l0 & xfs_mask64lo(9)) << 43) |
 174             (((xfs_dfsbno_t)r->l1) >> 21);
 175         ASSERT((b >> 32) == 0 || isnulldstartblock(b));
 176         return (xfs_fsblock_t)b;
 177 #else   /* !DEBUG */
 178         return (xfs_fsblock_t)(((xfs_dfsbno_t)r->l1) >> 21);
 179 #endif  /* DEBUG */
 180 #endif  /* XFS_BIG_BLKNOS */
 181 }
 182
 183 /*
 184  * Extract the startoff field from an in memory bmap extent record.
 185  */
 186 xfs_fileoff_t
 187 xfs_bmbt_get_startoff(
 188         xfs_bmbt_rec_host_t     *r)
 189 {
 190         return ((xfs_fileoff_t)r->l0 &
 191                  xfs_mask64lo(64 - BMBT_EXNTFLAG_BITLEN)) >> 9;
 192 }
 193
 194 xfs_exntst_t
 195 xfs_bmbt_get_state(
 196         xfs_bmbt_rec_host_t     *r)
 197 {
 198         int     ext_flag;
 199
 200         ext_flag = (int)((r->l0) >> (64 - BMBT_EXNTFLAG_BITLEN));
 201         return xfs_extent_state(xfs_bmbt_get_blockcount(r),
 202                                 ext_flag);
 203 }
 204
 205 /*
 206  * Extract the blockcount field from an on disk bmap extent record.
 207  */
 208 xfs_filblks_t
 209 xfs_bmbt_disk_get_blockcount(
 210         xfs_bmbt_rec_t  *r)
 211 {
 212         return (xfs_filblks_t)(be64_to_cpu(r->l1) & xfs_mask64lo(21));
 213 }
 214
 215 /*
 216  * Extract the startoff field from a disk format bmap extent record.
 217  */
 218 xfs_fileoff_t
 219 xfs_bmbt_disk_get_startoff(
 220         xfs_bmbt_rec_t  *r)
 221 {
 222         return ((xfs_fileoff_t)be64_to_cpu(r->l0) &
 223                  xfs_mask64lo(64 - BMBT_EXNTFLAG_BITLEN)) >> 9;
 224 }
 225
 226
 227 /*
 228  * Set all the fields in a bmap extent record from the arguments.
 229  */
 230 void
 231 xfs_bmbt_set_allf(
 232         xfs_bmbt_rec_host_t     *r,
 233         xfs_fileoff_t           startoff,
 234         xfs_fsblock_t           startblock,
 235         xfs_filblks_t           blockcount,
 236         xfs_exntst_t            state)
 237 {
 238         int             extent_flag = (state == XFS_EXT_NORM) ? 0 : 1;
 239
 240         ASSERT(state == XFS_EXT_NORM || state == XFS_EXT_UNWRITTEN);
 241         ASSERT((startoff & xfs_mask64hi(64-BMBT_STARTOFF_BITLEN)) == 0);
 242         ASSERT((blockcount & xfs_mask64hi(64-BMBT_BLOCKCOUNT_BITLEN)) == 0);
 243
 244 #if XFS_BIG_BLKNOS
 245         ASSERT((startblock & xfs_mask64hi(64-BMBT_STARTBLOCK_BITLEN)) == 0);
 246
 247         r->l0 = ((xfs_bmbt_rec_base_t)extent_flag << 63) |
 248                 ((xfs_bmbt_rec_base_t)startoff << 9) |
 249                 ((xfs_bmbt_rec_base_t)startblock >> 43);
 250         r->l1 = ((xfs_bmbt_rec_base_t)startblock << 21) |
 251                 ((xfs_bmbt_rec_base_t)blockcount &
 252                 (xfs_bmbt_rec_base_t)xfs_mask64lo(21));
 253 #else   /* !XFS_BIG_BLKNOS */
 254         if (isnullstartblock(startblock)) {
 255                 r->l0 = ((xfs_bmbt_rec_base_t)extent_flag << 63) |
 256                         ((xfs_bmbt_rec_base_t)startoff << 9) |
 257                          (xfs_bmbt_rec_base_t)xfs_mask64lo(9);
 258                 r->l1 = xfs_mask64hi(11) |
 259                           ((xfs_bmbt_rec_base_t)startblock << 21) |
 260                           ((xfs_bmbt_rec_base_t)blockcount &
 261                            (xfs_bmbt_rec_base_t)xfs_mask64lo(21));
 262         } else {
 263                 r->l0 = ((xfs_bmbt_rec_base_t)extent_flag << 63) |
 264                         ((xfs_bmbt_rec_base_t)startoff << 9);
 265                 r->l1 = ((xfs_bmbt_rec_base_t)startblock << 21) |
 266                          ((xfs_bmbt_rec_base_t)blockcount &
 267                          (xfs_bmbt_rec_base_t)xfs_mask64lo(21));
 268         }
 269 #endif  /* XFS_BIG_BLKNOS */
 270 }
 271
 272 /*
 273  * Set all the fields in a bmap extent record from the uncompressed form.
 274  */
 275 void
 276 xfs_bmbt_set_all(
 277         xfs_bmbt_rec_host_t *r,
 278         xfs_bmbt_irec_t *s)
 279 {
 280         xfs_bmbt_set_allf(r, s->br_startoff, s->br_startblock,
 281                              s->br_blockcount, s->br_state);
 282 }
 283
 284
 285 /*
 286  * Set all the fields in a disk format bmap extent record from the arguments.
 287  */
 288 void
 289 xfs_bmbt_disk_set_allf(
 290         xfs_bmbt_rec_t          *r,
 291         xfs_fileoff_t           startoff,
 292         xfs_fsblock_t           startblock,
 293         xfs_filblks_t           blockcount,
 294         xfs_exntst_t            state)
 295 {
 296         int                     extent_flag = (state == XFS_EXT_NORM) ? 0 : 1;
 297
 298         ASSERT(state == XFS_EXT_NORM || state == XFS_EXT_UNWRITTEN);
 299         ASSERT((startoff & xfs_mask64hi(64-BMBT_STARTOFF_BITLEN)) == 0);
 300         ASSERT((blockcount & xfs_mask64hi(64-BMBT_BLOCKCOUNT_BITLEN)) == 0);
 301
 302 #if XFS_BIG_BLKNOS
 303         ASSERT((startblock & xfs_mask64hi(64-BMBT_STARTBLOCK_BITLEN)) == 0);
 304
 305         r->l0 = cpu_to_be64(
 306                 ((xfs_bmbt_rec_base_t)extent_flag << 63) |
 307                  ((xfs_bmbt_rec_base_t)startoff << 9) |
 308                  ((xfs_bmbt_rec_base_t)startblock >> 43));
 309         r->l1 = cpu_to_be64(
 310                 ((xfs_bmbt_rec_base_t)startblock << 21) |
 311                  ((xfs_bmbt_rec_base_t)blockcount &
 312                   (xfs_bmbt_rec_base_t)xfs_mask64lo(21)));
 313 #else   /* !XFS_BIG_BLKNOS */
 314         if (isnullstartblock(startblock)) {
 315                 r->l0 = cpu_to_be64(
 316                         ((xfs_bmbt_rec_base_t)extent_flag << 63) |
 317                          ((xfs_bmbt_rec_base_t)startoff << 9) |
 318                           (xfs_bmbt_rec_base_t)xfs_mask64lo(9));
 319                 r->l1 = cpu_to_be64(xfs_mask64hi(11) |
 320                           ((xfs_bmbt_rec_base_t)startblock << 21) |
 321                           ((xfs_bmbt_rec_base_t)blockcount &
 322                            (xfs_bmbt_rec_base_t)xfs_mask64lo(21)));
 323         } else {
 324                 r->l0 = cpu_to_be64(
 325                         ((xfs_bmbt_rec_base_t)extent_flag << 63) |
 326                          ((xfs_bmbt_rec_base_t)startoff << 9));
 327                 r->l1 = cpu_to_be64(
 328                         ((xfs_bmbt_rec_base_t)startblock << 21) |
 329                          ((xfs_bmbt_rec_base_t)blockcount &
 330                           (xfs_bmbt_rec_base_t)xfs_mask64lo(21)));
 331         }
 332 #endif  /* XFS_BIG_BLKNOS */
 333 }
 334
 335 /*
 336  * Set all the fields in a bmap extent record from the uncompressed form.
 337  */
 338 void
 339 xfs_bmbt_disk_set_all(
 340         xfs_bmbt_rec_t  *r,
 341         xfs_bmbt_irec_t *s)
 342 {
 343         xfs_bmbt_disk_set_allf(r, s->br_startoff, s->br_startblock,
 344                                   s->br_blockcount, s->br_state);
 345 }
 346
 347 /*
 348  * Set the blockcount field in a bmap extent record.
 349  */
 350 void
 351 xfs_bmbt_set_blockcount(
 352         xfs_bmbt_rec_host_t *r,
 353         xfs_filblks_t   v)
 354 {
 355         ASSERT((v & xfs_mask64hi(43)) == 0);
 356         r->l1 = (r->l1 & (xfs_bmbt_rec_base_t)xfs_mask64hi(43)) |
 357                   (xfs_bmbt_rec_base_t)(v & xfs_mask64lo(21));
 358 }
 359
 360 /*
 361  * Set the startblock field in a bmap extent record.
 362  */
 363 void
 364 xfs_bmbt_set_startblock(
 365         xfs_bmbt_rec_host_t *r,
 366         xfs_fsblock_t   v)
 367 {
 368 #if XFS_BIG_BLKNOS
 369         ASSERT((v & xfs_mask64hi(12)) == 0);
 370         r->l0 = (r->l0 & (xfs_bmbt_rec_base_t)xfs_mask64hi(55)) |
 371                   (xfs_bmbt_rec_base_t)(v >> 43);
 372         r->l1 = (r->l1 & (xfs_bmbt_rec_base_t)xfs_mask64lo(21)) |
 373                   (xfs_bmbt_rec_base_t)(v << 21);
 374 #else   /* !XFS_BIG_BLKNOS */
 375         if (isnullstartblock(v)) {
 376                 r->l0 |= (xfs_bmbt_rec_base_t)xfs_mask64lo(9);
 377                 r->l1 = (xfs_bmbt_rec_base_t)xfs_mask64hi(11) |
 378                           ((xfs_bmbt_rec_base_t)v << 21) |
 379                           (r->l1 & (xfs_bmbt_rec_base_t)xfs_mask64lo(21));
 380         } else {
 381                 r->l0 &= ~(xfs_bmbt_rec_base_t)xfs_mask64lo(9);
 382                 r->l1 = ((xfs_bmbt_rec_base_t)v << 21) |
 383                           (r->l1 & (xfs_bmbt_rec_base_t)xfs_mask64lo(21));
 384         }
 385 #endif  /* XFS_BIG_BLKNOS */
 386 }
 387
 388 /*
 389  * Set the startoff field in a bmap extent record.
 390  */
 391 void
 392 xfs_bmbt_set_startoff(
 393         xfs_bmbt_rec_host_t *r,
 394         xfs_fileoff_t   v)
 395 {
 396         ASSERT((v & xfs_mask64hi(9)) == 0);
 397         r->l0 = (r->l0 & (xfs_bmbt_rec_base_t) xfs_mask64hi(1)) |
 398                 ((xfs_bmbt_rec_base_t)v << 9) |
 399                   (r->l0 & (xfs_bmbt_rec_base_t)xfs_mask64lo(9));
 400 }
 401
 402 /*
 403  * Set the extent state field in a bmap extent record.
 404  */
 405 void
 406 xfs_bmbt_set_state(
 407         xfs_bmbt_rec_host_t *r,
 408         xfs_exntst_t    v)
 409 {
 410         ASSERT(v == XFS_EXT_NORM || v == XFS_EXT_UNWRITTEN);
 411         if (v == XFS_EXT_NORM)
 412                 r->l0 &= xfs_mask64lo(64 - BMBT_EXNTFLAG_BITLEN);
 413         else
 414                 r->l0 |= xfs_mask64hi(BMBT_EXNTFLAG_BITLEN);
 415 }
 416
 417 /*
 418  * Convert in-memory form of btree root to on-disk form.
 419  */
 420 void
 421 xfs_bmbt_to_bmdr(
 422         struct xfs_mount        *mp,
 423         struct xfs_btree_block  *rblock,
 424         int                     rblocklen,
 425         xfs_bmdr_block_t        *dblock,
 426         int                     dblocklen)
 427 {
 428         int                     dmxr;
 429         xfs_bmbt_key_t          *fkp;
 430         __be64                  *fpp;
 431         xfs_bmbt_key_t          *tkp;
 432         __be64                  *tpp;
 433
 434         ASSERT(be32_to_cpu(rblock->bb_magic) == XFS_BMAP_MAGIC);
 435         ASSERT(be64_to_cpu(rblock->bb_u.l.bb_leftsib) == NULLDFSBNO);
 436         ASSERT(be64_to_cpu(rblock->bb_u.l.bb_rightsib) == NULLDFSBNO);
 437         ASSERT(be16_to_cpu(rblock->bb_level) > 0);
 438         dblock->bb_level = rblock->bb_level;
 439         dblock->bb_numrecs = rblock->bb_numrecs;
 440         dmxr = xfs_bmdr_maxrecs(mp, dblocklen, 0);
 441         fkp = XFS_BMBT_KEY_ADDR(mp, rblock, 1);
 442         tkp = XFS_BMDR_KEY_ADDR(dblock, 1);
 443         fpp = XFS_BMAP_BROOT_PTR_ADDR(mp, rblock, 1, rblocklen);
 444         tpp = XFS_BMDR_PTR_ADDR(dblock, 1, dmxr);
 445         dmxr = be16_to_cpu(dblock->bb_numrecs);
 446         memcpy(tkp, fkp, sizeof(*fkp) * dmxr);
 447         memcpy(tpp, fpp, sizeof(*fpp) * dmxr);
 448 }
 449
 450 /*
 451  * Check extent records, which have just been read, for
 452  * any bit in the extent flag field. ASSERT on debug
 453  * kernels, as this condition should not occur.
 454  * Return an error condition (1) if any flags found,
 455  * otherwise return 0.
 456  */
 457
 458 int
 459 xfs_check_nostate_extents(
 460         xfs_ifork_t             *ifp,
 461         xfs_extnum_t            idx,
 462         xfs_extnum_t            num)
 463 {
 464         for (; num > 0; num--, idx++) {
 465                 xfs_bmbt_rec_host_t *ep = xfs_iext_get_ext(ifp, idx);
 466                 if ((ep->l0 >>
 467                      (64 - BMBT_EXNTFLAG_BITLEN)) != 0) {
 468                         ASSERT(0);
 469                         return 1;
 470                 }
 471         }
 472         return 0;
 473 }
 474
 475
 476 STATIC struct xfs_btree_cur *
 477 xfs_bmbt_dup_cursor(
 478         struct xfs_btree_cur    *cur)
 479 {
 480         struct xfs_btree_cur    *new;
 481
 482         new = xfs_bmbt_init_cursor(cur->bc_mp, cur->bc_tp,
 483                         cur->bc_private.b.ip, cur->bc_private.b.whichfork);
 484
 485         /*
 486          * Copy the firstblock, flist, and flags values,
 487          * since init cursor doesn't get them.
 488          */
 489         new->bc_private.b.firstblock = cur->bc_private.b.firstblock;
 490         new->bc_private.b.flist = cur->bc_private.b.flist;
 491         new->bc_private.b.flags = cur->bc_private.b.flags;
 492
 493         return new;
 494 }
 495
 496 STATIC void
 497 xfs_bmbt_update_cursor(
 498         struct xfs_btree_cur    *src,
 499         struct xfs_btree_cur    *dst)
 500 {
 501         ASSERT((dst->bc_private.b.firstblock != NULLFSBLOCK) ||
 502                (dst->bc_private.b.ip->i_d.di_flags & XFS_DIFLAG_REALTIME));
 503         ASSERT(dst->bc_private.b.flist == src->bc_private.b.flist);
 504
 505         dst->bc_private.b.allocated += src->bc_private.b.allocated;
 506         dst->bc_private.b.firstblock = src->bc_private.b.firstblock;
 507
 508         src->bc_private.b.allocated = 0;
 509 }
 510
 511 STATIC int
 512 xfs_bmbt_alloc_block(
 513         struct xfs_btree_cur    *cur,
 514         union xfs_btree_ptr     *start,
 515         union xfs_btree_ptr     *new,
 516         int                     length,
 517         int                     *stat)
 518 {
 519         xfs_alloc_arg_t         args;           /* block allocation args */
 520         int                     error;          /* error return value */
 521
 522         memset(&args, 0, sizeof(args));
 523         args.tp = cur->bc_tp;
 524         args.mp = cur->bc_mp;
 525         args.fsbno = cur->bc_private.b.firstblock;
 526         args.firstblock = args.fsbno;
 527
 528         if (args.fsbno == NULLFSBLOCK) {
 529                 args.fsbno = be64_to_cpu(start->l);
 530                 args.type = XFS_ALLOCTYPE_START_BNO;
 531                 /*
 532                  * Make sure there is sufficient room left in the AG to
 533                  * complete a full tree split for an extent insert.  If
 534                  * we are converting the middle part of an extent then
 535                  * we may need space for two tree splits.
 536                  *
 537                  * We are relying on the caller to make the correct block
 538                  * reservation for this operation to succeed.  If the
 539                  * reservation amount is insufficient then we may fail a
 540                  * block allocation here and corrupt the filesystem.
 541                  */
 542                 args.minleft = xfs_trans_get_block_res(args.tp);
 543         } else if (cur->bc_private.b.flist->xbf_low) {
 544                 args.type = XFS_ALLOCTYPE_START_BNO;
 545         } else {
 546                 args.type = XFS_ALLOCTYPE_NEAR_BNO;
 547         }
 548
 549         args.minlen = args.maxlen = args.prod = 1;
 550         args.wasdel = cur->bc_private.b.flags & XFS_BTCUR_BPRV_WASDEL;
 551         if (!args.wasdel && xfs_trans_get_block_res(args.tp) == 0) {
 552                 error = XFS_ERROR(ENOSPC);
 553                 goto error0;
 554         }
 555         error = xfs_alloc_vextent(&args);
 556         if (error)
 557                 goto error0;
 558
 559         if (args.fsbno == NULLFSBLOCK && args.minleft) {
 560                 /*
 561                  * Could not find an AG with enough free space to satisfy
 562                  * a full btree split.  Try again without minleft and if
 563                  * successful activate the lowspace algorithm.
 564                  */
 565                 args.fsbno = 0;
 566                 args.type = XFS_ALLOCTYPE_FIRST_AG;
 567                 args.minleft = 0;
 568                 error = xfs_alloc_vextent(&args);
 569                 if (error)
 570                         goto error0;
 571                 cur->bc_private.b.flist->xbf_low = 1;
 572         }
 573         if (args.fsbno == NULLFSBLOCK) {
 574                 XFS_BTREE_TRACE_CURSOR(cur, XBT_EXIT);
 575                 *stat = 0;
 576                 return 0;
 577         }
 578         ASSERT(args.len == 1);
 579         cur->bc_private.b.firstblock = args.fsbno;
 580         cur->bc_private.b.allocated++;
 581         cur->bc_private.b.ip->i_d.di_nblocks++;
 582         xfs_trans_log_inode(args.tp, cur->bc_private.b.ip, XFS_ILOG_CORE);
 583         xfs_trans_mod_dquot_byino(args.tp, cur->bc_private.b.ip,
 584                         XFS_TRANS_DQ_BCOUNT, 1L);
 585
 586         new->l = cpu_to_be64(args.fsbno);
 587
 588         XFS_BTREE_TRACE_CURSOR(cur, XBT_EXIT);
 589         *stat = 1;
 590         return 0;
 591
 592  error0:
 593         XFS_BTREE_TRACE_CURSOR(cur, XBT_ERROR);
 594         return error;
 595 }
 596
 597 STATIC int
 598 xfs_bmbt_free_block(
 599         struct xfs_btree_cur    *cur,
 600         struct xfs_buf          *bp)
 601 {
 602         struct xfs_mount        *mp = cur->bc_mp;
 603         struct xfs_inode        *ip = cur->bc_private.b.ip;
 604         struct xfs_trans        *tp = cur->bc_tp;
 605         xfs_fsblock_t           fsbno = XFS_DADDR_TO_FSB(mp, XFS_BUF_ADDR(bp));
 606
 607         xfs_bmap_add_free(fsbno, 1, cur->bc_private.b.flist, mp);
 608         ip->i_d.di_nblocks--;
 609
 610         xfs_trans_log_inode(tp, ip, XFS_ILOG_CORE);
 611         xfs_trans_mod_dquot_byino(tp, ip, XFS_TRANS_DQ_BCOUNT, -1L);
 612         xfs_trans_binval(tp, bp);
 613         return 0;
 614 }
 615
 616 STATIC int
 617 xfs_bmbt_get_minrecs(
 618         struct xfs_btree_cur    *cur,
 619         int                     level)
 620 {
 621         if (level == cur->bc_nlevels - 1) {
 622                 struct xfs_ifork        *ifp;
 623
 624                 ifp = XFS_IFORK_PTR(cur->bc_private.b.ip,
 625                                     cur->bc_private.b.whichfork);
 626
 627                 return xfs_bmbt_maxrecs(cur->bc_mp,
 628                                         ifp->if_broot_bytes, level == 0) / 2;
 629         }
 630
 631         return cur->bc_mp->m_bmap_dmnr[level != 0];
 632 }
 633
 634 int
 635 xfs_bmbt_get_maxrecs(
 636         struct xfs_btree_cur    *cur,
 637         int                     level)
 638 {
 639         if (level == cur->bc_nlevels - 1) {
 640                 struct xfs_ifork        *ifp;
 641
 642                 ifp = XFS_IFORK_PTR(cur->bc_private.b.ip,
 643                                     cur->bc_private.b.whichfork);
 644
 645                 return xfs_bmbt_maxrecs(cur->bc_mp,
 646                                         ifp->if_broot_bytes, level == 0);
 647         }
 648
 649         return cur->bc_mp->m_bmap_dmxr[level != 0];
 650
 651 }
 652
 653 /*
 654  * Get the maximum records we could store in the on-disk format.
 655  *
 656  * For non-root nodes this is equivalent to xfs_bmbt_get_maxrecs, but
 657  * for the root node this checks the available space in the dinode fork
 658  * so that we can resize the in-memory buffer to match it.  After a
 659  * resize to the maximum size this function returns the same value
 660  * as xfs_bmbt_get_maxrecs for the root node, too.
 661  */
 662 STATIC int
 663 xfs_bmbt_get_dmaxrecs(
 664         struct xfs_btree_cur    *cur,
 665         int                     level)
 666 {
 667         if (level != cur->bc_nlevels - 1)
 668                 return cur->bc_mp->m_bmap_dmxr[level != 0];
 669         return xfs_bmdr_maxrecs(cur->bc_mp, cur->bc_private.b.forksize,
 670                                 level == 0);
 671 }
 672
 673 STATIC void
 674 xfs_bmbt_init_key_from_rec(
 675         union xfs_btree_key     *key,
 676         union xfs_btree_rec     *rec)
 677 {
 678         key->bmbt.br_startoff =
 679                 cpu_to_be64(xfs_bmbt_disk_get_startoff(&rec->bmbt));
 680 }
 681
 682 STATIC void
 683 xfs_bmbt_init_rec_from_key(
 684         union xfs_btree_key     *key,
 685         union xfs_btree_rec     *rec)
 686 {
 687         ASSERT(key->bmbt.br_startoff != 0);
 688
 689         xfs_bmbt_disk_set_allf(&rec->bmbt, be64_to_cpu(key->bmbt.br_startoff),
 690                                0, 0, XFS_EXT_NORM);
 691 }
 692
 693 STATIC void
 694 xfs_bmbt_init_rec_from_cur(
 695         struct xfs_btree_cur    *cur,
 696         union xfs_btree_rec     *rec)
 697 {
 698         xfs_bmbt_disk_set_all(&rec->bmbt, &cur->bc_rec.b);
 699 }
 700
 701 STATIC void
 702 xfs_bmbt_init_ptr_from_cur(
 703         struct xfs_btree_cur    *cur,
 704         union xfs_btree_ptr     *ptr)
 705 {
 706         ptr->l = 0;
 707 }
 708
 709 STATIC __int64_t
 710 xfs_bmbt_key_diff(
 711         struct xfs_btree_cur    *cur,
 712         union xfs_btree_key     *key)
 713 {
 714         return (__int64_t)be64_to_cpu(key->bmbt.br_startoff) -
 715                                       cur->bc_rec.b.br_startoff;
 716 }
 717
 718 #ifdef DEBUG
 719 STATIC int
 720 xfs_bmbt_keys_inorder(
 721         struct xfs_btree_cur    *cur,
 722         union xfs_btree_key     *k1,
 723         union xfs_btree_key     *k2)
 724 {
 725         return be64_to_cpu(k1->bmbt.br_startoff) <
 726                 be64_to_cpu(k2->bmbt.br_startoff);
 727 }
 728
 729 STATIC int
 730 xfs_bmbt_recs_inorder(
 731         struct xfs_btree_cur    *cur,
 732         union xfs_btree_rec     *r1,
 733         union xfs_btree_rec     *r2)
 734 {
 735         return xfs_bmbt_disk_get_startoff(&r1->bmbt) +
 736                 xfs_bmbt_disk_get_blockcount(&r1->bmbt) <=
 737                 xfs_bmbt_disk_get_startoff(&r2->bmbt);
 738 }
 739 #endif  /* DEBUG */
 740
 741 #ifdef XFS_BTREE_TRACE
 742 ktrace_t        *xfs_bmbt_trace_buf;
 743
 744 STATIC void
 745 xfs_bmbt_trace_enter(
 746         struct xfs_btree_cur    *cur,
 747         const char              *func,
 748         char                    *s,
 749         int                     type,
 750         int                     line,
 751         __psunsigned_t          a0,
 752         __psunsigned_t          a1,
 753         __psunsigned_t          a2,
 754         __psunsigned_t          a3,
 755         __psunsigned_t          a4,
 756         __psunsigned_t          a5,
 757         __psunsigned_t          a6,
 758         __psunsigned_t          a7,
 759         __psunsigned_t          a8,
 760         __psunsigned_t          a9,
 761         __psunsigned_t          a10)
 762 {
 763         struct xfs_inode        *ip = cur->bc_private.b.ip;
 764         int                     whichfork = cur->bc_private.b.whichfork;
 765
 766         ktrace_enter(xfs_bmbt_trace_buf,
 767                 (void *)((__psint_t)type | (whichfork << 8) | (line << 16)),
 768                 (void *)func, (void *)s, (void *)ip, (void *)cur,
 769                 (void *)a0, (void *)a1, (void *)a2, (void *)a3,
 770                 (void *)a4, (void *)a5, (void *)a6, (void *)a7,
 771                 (void *)a8, (void *)a9, (void *)a10);
 772         ktrace_enter(ip->i_btrace,
 773                 (void *)((__psint_t)type | (whichfork << 8) | (line << 16)),
 774                 (void *)func, (void *)s, (void *)ip, (void *)cur,
 775                 (void *)a0, (void *)a1, (void *)a2, (void *)a3,
 776                 (void *)a4, (void *)a5, (void *)a6, (void *)a7,
 777                 (void *)a8, (void *)a9, (void *)a10);
 778 }
 779
 780 STATIC void
 781 xfs_bmbt_trace_cursor(
 782         struct xfs_btree_cur    *cur,
 783         __uint32_t              *s0,
 784         __uint64_t              *l0,
 785         __uint64_t              *l1)
 786 {
 787         struct xfs_bmbt_rec_host r;
 788
 789         xfs_bmbt_set_all(&r, &cur->bc_rec.b);
 790
 791         *s0 = (cur->bc_nlevels << 24) |
 792               (cur->bc_private.b.flags << 16) |
 793                cur->bc_private.b.allocated;
 794         *l0 = r.l0;
 795         *l1 = r.l1;
 796 }
 797
 798 STATIC void
 799 xfs_bmbt_trace_key(
 800         struct xfs_btree_cur    *cur,
 801         union xfs_btree_key     *key,
 802         __uint64_t              *l0,
 803         __uint64_t              *l1)
 804 {
 805         *l0 = be64_to_cpu(key->bmbt.br_startoff);
 806         *l1 = 0;
 807 }
 808
 809 /* Endian flipping versions of the bmbt extraction functions */
 810 STATIC void
 811 xfs_bmbt_disk_get_all(
 812         xfs_bmbt_rec_t  *r,
 813         xfs_bmbt_irec_t *s)
 814 {
 815         __xfs_bmbt_get_all(get_unaligned_be64(&r->l0),
 816                                 get_unaligned_be64(&r->l1), s);
 817 }
 818
 819 STATIC void
 820 xfs_bmbt_trace_record(
 821         struct xfs_btree_cur    *cur,
 822         union xfs_btree_rec     *rec,
 823         __uint64_t              *l0,
 824         __uint64_t              *l1,
 825         __uint64_t              *l2)
 826 {
 827         struct xfs_bmbt_irec    irec;
 828
 829         xfs_bmbt_disk_get_all(&rec->bmbt, &irec);
 830         *l0 = irec.br_startoff;
 831         *l1 = irec.br_startblock;
 832         *l2 = irec.br_blockcount;
 833 }
 834 #endif /* XFS_BTREE_TRACE */
 835
 836 static const struct xfs_btree_ops xfs_bmbt_ops = {
 837         .rec_len                = sizeof(xfs_bmbt_rec_t),
 838         .key_len                = sizeof(xfs_bmbt_key_t),
 839
 840         .dup_cursor             = xfs_bmbt_dup_cursor,
 841         .update_cursor          = xfs_bmbt_update_cursor,
 842         .alloc_block            = xfs_bmbt_alloc_block,
 843         .free_block             = xfs_bmbt_free_block,
 844         .get_maxrecs            = xfs_bmbt_get_maxrecs,
 845         .get_minrecs            = xfs_bmbt_get_minrecs,
 846         .get_dmaxrecs           = xfs_bmbt_get_dmaxrecs,
 847         .init_key_from_rec      = xfs_bmbt_init_key_from_rec,
 848         .init_rec_from_key      = xfs_bmbt_init_rec_from_key,
 849         .init_rec_from_cur      = xfs_bmbt_init_rec_from_cur,
 850         .init_ptr_from_cur      = xfs_bmbt_init_ptr_from_cur,
 851         .key_diff               = xfs_bmbt_key_diff,
 852
 853 #ifdef DEBUG
 854         .keys_inorder           = xfs_bmbt_keys_inorder,
 855         .recs_inorder           = xfs_bmbt_recs_inorder,
 856 #endif
 857
 858 #ifdef XFS_BTREE_TRACE
 859         .trace_enter            = xfs_bmbt_trace_enter,
 860         .trace_cursor           = xfs_bmbt_trace_cursor,
 861         .trace_key              = xfs_bmbt_trace_key,
 862         .trace_record           = xfs_bmbt_trace_record,
 863 #endif
 864 };
 865
 866 /*
 867  * Allocate a new bmap btree cursor.
 868  */
 869 struct xfs_btree_cur *                          /* new bmap btree cursor */
 870 xfs_bmbt_init_cursor(
 871         struct xfs_mount        *mp,            /* file system mount point */
 872         struct xfs_trans        *tp,            /* transaction pointer */
 873         struct xfs_inode        *ip,            /* inode owning the btree */
 874         int                     whichfork)      /* data or attr fork */
 875 {
 876         struct xfs_ifork        *ifp = XFS_IFORK_PTR(ip, whichfork);
 877         struct xfs_btree_cur    *cur;
 878
 879         cur = kmem_zone_zalloc(xfs_btree_cur_zone, KM_SLEEP);
 880
 881         cur->bc_tp = tp;
 882         cur->bc_mp = mp;
 883         cur->bc_nlevels = be16_to_cpu(ifp->if_broot->bb_level) + 1;
 884         cur->bc_btnum = XFS_BTNUM_BMAP;
 885         cur->bc_blocklog = mp->m_sb.sb_blocklog;
 886
 887         cur->bc_ops = &xfs_bmbt_ops;
 888         cur->bc_flags = XFS_BTREE_LONG_PTRS | XFS_BTREE_ROOT_IN_INODE;
 889
 890         cur->bc_private.b.forksize = XFS_IFORK_SIZE(ip, whichfork);
 891         cur->bc_private.b.ip = ip;
 892         cur->bc_private.b.firstblock = NULLFSBLOCK;
 893         cur->bc_private.b.flist = NULL;
 894         cur->bc_private.b.allocated = 0;
 895         cur->bc_private.b.flags = 0;
 896         cur->bc_private.b.whichfork = whichfork;
 897
 898         return cur;
 899 }
 900
 901 /*
 902  * Calculate number of records in a bmap btree block.
 903  */
 904 int
 905 xfs_bmbt_maxrecs(
 906         struct xfs_mount        *mp,
 907         int                     blocklen,
 908         int                     leaf)
 909 {
 910         blocklen -= XFS_BMBT_BLOCK_LEN(mp);
 911
 912         if (leaf)
 913                 return blocklen / sizeof(xfs_bmbt_rec_t);
 914         return blocklen / (sizeof(xfs_bmbt_key_t) + sizeof(xfs_bmbt_ptr_t));
 915 }
 916
 917 /*
 918  * Calculate number of records in a bmap btree inode root.
 919  */
 920 int
 921 xfs_bmdr_maxrecs(
 922         struct xfs_mount        *mp,
 923         int                     blocklen,
 924         int                     leaf)
 925 {
 926         blocklen -= sizeof(xfs_bmdr_block_t);
 927
 928         if (leaf)
 929                 return blocklen / sizeof(xfs_bmdr_rec_t);
 930         return blocklen / (sizeof(xfs_bmdr_key_t) + sizeof(xfs_bmdr_ptr_t));
 931 }