fs/ocfs2/suballoc.c

   1 /* -*- mode: c; c-basic-offset: 8; -*-
   2  * vim: noexpandtab sw=8 ts=8 sts=0:
   3  *
   4  * suballoc.c
   5  *
   6  * metadata alloc and free
   7  * Inspired by ext3 block groups.
   8  *
   9  * Copyright (C) 2002, 2004 Oracle.  All rights reserved.
  10  *
  11  * This program is free software; you can redistribute it and/or
  12  * modify it under the terms of the GNU General Public
  13  * License as published by the Free Software Foundation; either
  14  * version 2 of the License, or (at your option) any later version.
  15  *
  16  * This program is distributed in the hope that it will be useful,
  17  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  18  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  19  * General Public License for more details.
  20  *
  21  * You should have received a copy of the GNU General Public
  22  * License along with this program; if not, write to the
  23  * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  24  * Boston, MA 021110-1307, USA.
  25  */
  26
  27 #include <linux/fs.h>
  28 #include <linux/types.h>
  29 #include <linux/slab.h>
  30 #include <linux/highmem.h>
  31
  32 #include <cluster/masklog.h>
  33
  34 #include "ocfs2.h"
  35
  36 #include "alloc.h"
  37 #include "blockcheck.h"
  38 #include "dlmglue.h"
  39 #include "inode.h"
  40 #include "journal.h"
  41 #include "localalloc.h"
  42 #include "suballoc.h"
  43 #include "super.h"
  44 #include "sysfile.h"
  45 #include "uptodate.h"
  46 #include "ocfs2_trace.h"
  47
  48 #include "buffer_head_io.h"
  49
  50 #define NOT_ALLOC_NEW_GROUP             0
  51 #define ALLOC_NEW_GROUP                 0x1
  52 #define ALLOC_GROUPS_FROM_GLOBAL        0x2
  53
  54 #define OCFS2_MAX_TO_STEAL              1024
  55
  56 struct ocfs2_suballoc_result {
  57         u64             sr_bg_blkno;    /* The bg we allocated from.  Set
  58                                            to 0 when a block group is
  59                                            contiguous. */
  60         u64             sr_bg_stable_blkno; /*
  61                                              * Doesn't change, always
  62                                              * set to target block
  63                                              * group descriptor
  64                                              * block.
  65                                              */
  66         u64             sr_blkno;       /* The first allocated block */
  67         unsigned int    sr_bit_offset;  /* The bit in the bg */
  68         unsigned int    sr_bits;        /* How many bits we claimed */
  69 };
  70
  71 static u64 ocfs2_group_from_res(struct ocfs2_suballoc_result *res)
  72 {
  73         if (res->sr_blkno == 0)
  74                 return 0;
  75
  76         if (res->sr_bg_blkno)
  77                 return res->sr_bg_blkno;
  78
  79         return ocfs2_which_suballoc_group(res->sr_blkno, res->sr_bit_offset);
  80 }
  81
  82 static inline void ocfs2_debug_bg(struct ocfs2_group_desc *bg);
  83 static inline void ocfs2_debug_suballoc_inode(struct ocfs2_dinode *fe);
  84 static inline u16 ocfs2_find_victim_chain(struct ocfs2_chain_list *cl);
  85 static int ocfs2_block_group_fill(handle_t *handle,
  86                                   struct inode *alloc_inode,
  87                                   struct buffer_head *bg_bh,
  88                                   u64 group_blkno,
  89                                   unsigned int group_clusters,
  90                                   u16 my_chain,
  91                                   struct ocfs2_chain_list *cl);
  92 static int ocfs2_block_group_alloc(struct ocfs2_super *osb,
  93                                    struct inode *alloc_inode,
  94                                    struct buffer_head *bh,
  95                                    u64 max_block,
  96                                    u64 *last_alloc_group,
  97                                    int flags);
  98
  99 static int ocfs2_cluster_group_search(struct inode *inode,
 100                                       struct buffer_head *group_bh,
 101                                       u32 bits_wanted, u32 min_bits,
 102                                       u64 max_block,
 103                                       struct ocfs2_suballoc_result *res);
 104 static int ocfs2_block_group_search(struct inode *inode,
 105                                     struct buffer_head *group_bh,
 106                                     u32 bits_wanted, u32 min_bits,
 107                                     u64 max_block,
 108                                     struct ocfs2_suballoc_result *res);
 109 static int ocfs2_claim_suballoc_bits(struct ocfs2_alloc_context *ac,
 110                                      handle_t *handle,
 111                                      u32 bits_wanted,
 112                                      u32 min_bits,
 113                                      struct ocfs2_suballoc_result *res);
 114 static int ocfs2_test_bg_bit_allocatable(struct buffer_head *bg_bh,
 115                                          int nr);
 116 static inline int ocfs2_block_group_set_bits(handle_t *handle,
 117                                              struct inode *alloc_inode,
 118                                              struct ocfs2_group_desc *bg,
 119                                              struct buffer_head *group_bh,
 120                                              unsigned int bit_off,
 121                                              unsigned int num_bits);
 122 static int ocfs2_relink_block_group(handle_t *handle,
 123                                     struct inode *alloc_inode,
 124                                     struct buffer_head *fe_bh,
 125                                     struct buffer_head *bg_bh,
 126                                     struct buffer_head *prev_bg_bh,
 127                                     u16 chain);
 128 static inline int ocfs2_block_group_reasonably_empty(struct ocfs2_group_desc *bg,
 129                                                      u32 wanted);
 130 static inline u32 ocfs2_desc_bitmap_to_cluster_off(struct inode *inode,
 131                                                    u64 bg_blkno,
 132                                                    u16 bg_bit_off);
 133 static inline void ocfs2_block_to_cluster_group(struct inode *inode,
 134                                                 u64 data_blkno,
 135                                                 u64 *bg_blkno,
 136                                                 u16 *bg_bit_off);
 137 static int ocfs2_reserve_clusters_with_limit(struct ocfs2_super *osb,
 138                                              u32 bits_wanted, u64 max_block,
 139                                              int flags,
 140                                              struct ocfs2_alloc_context **ac);
 141
 142 void ocfs2_free_ac_resource(struct ocfs2_alloc_context *ac)
 143 {
 144         struct inode *inode = ac->ac_inode;
 145
 146         if (inode) {
 147                 if (ac->ac_which != OCFS2_AC_USE_LOCAL)
 148                         ocfs2_inode_unlock(inode, 1);
 149
 150                 mutex_unlock(&inode->i_mutex);
 151
 152                 iput(inode);
 153                 ac->ac_inode = NULL;
 154         }
 155         brelse(ac->ac_bh);
 156         ac->ac_bh = NULL;
 157         ac->ac_resv = NULL;
 158         if (ac->ac_find_loc_priv) {
 159                 kfree(ac->ac_find_loc_priv);
 160                 ac->ac_find_loc_priv = NULL;
 161         }
 162 }
 163
 164 void ocfs2_free_alloc_context(struct ocfs2_alloc_context *ac)
 165 {
 166         ocfs2_free_ac_resource(ac);
 167         kfree(ac);
 168 }
 169
 170 static u32 ocfs2_bits_per_group(struct ocfs2_chain_list *cl)
 171 {
 172         return (u32)le16_to_cpu(cl->cl_cpg) * (u32)le16_to_cpu(cl->cl_bpc);
 173 }
 174
 175 #define do_error(fmt, ...)                                              \
 176         do{                                                             \
 177                 if (resize)                                     \
 178                         mlog(ML_ERROR, fmt "\n", ##__VA_ARGS__);        \
 179                 else                                                    \
 180                         ocfs2_error(sb, fmt, ##__VA_ARGS__);            \
 181         } while (0)
 182
 183 static int ocfs2_validate_gd_self(struct super_block *sb,
 184                                   struct buffer_head *bh,
 185                                   int resize)
 186 {
 187         struct ocfs2_group_desc *gd = (struct ocfs2_group_desc *)bh->b_data;
 188
 189         if (!OCFS2_IS_VALID_GROUP_DESC(gd)) {
 190                 do_error("Group descriptor #%llu has bad signature %.*s",
 191                          (unsigned long long)bh->b_blocknr, 7,
 192                          gd->bg_signature);
 193                 return -EINVAL;
 194         }
 195
 196         if (le64_to_cpu(gd->bg_blkno) != bh->b_blocknr) {
 197                 do_error("Group descriptor #%llu has an invalid bg_blkno "
 198                          "of %llu",
 199                          (unsigned long long)bh->b_blocknr,
 200                          (unsigned long long)le64_to_cpu(gd->bg_blkno));
 201                 return -EINVAL;
 202         }
 203
 204         if (le32_to_cpu(gd->bg_generation) != OCFS2_SB(sb)->fs_generation) {
 205                 do_error("Group descriptor #%llu has an invalid "
 206                          "fs_generation of #%u",
 207                          (unsigned long long)bh->b_blocknr,
 208                          le32_to_cpu(gd->bg_generation));
 209                 return -EINVAL;
 210         }
 211
 212         if (le16_to_cpu(gd->bg_free_bits_count) > le16_to_cpu(gd->bg_bits)) {
 213                 do_error("Group descriptor #%llu has bit count %u but "
 214                          "claims that %u are free",
 215                          (unsigned long long)bh->b_blocknr,
 216                          le16_to_cpu(gd->bg_bits),
 217                          le16_to_cpu(gd->bg_free_bits_count));
 218                 return -EINVAL;
 219         }
 220
 221         if (le16_to_cpu(gd->bg_bits) > (8 * le16_to_cpu(gd->bg_size))) {
 222                 do_error("Group descriptor #%llu has bit count %u but "
 223                          "max bitmap bits of %u",
 224                          (unsigned long long)bh->b_blocknr,
 225                          le16_to_cpu(gd->bg_bits),
 226                          8 * le16_to_cpu(gd->bg_size));
 227                 return -EINVAL;
 228         }
 229
 230         return 0;
 231 }
 232
 233 static int ocfs2_validate_gd_parent(struct super_block *sb,
 234                                     struct ocfs2_dinode *di,
 235                                     struct buffer_head *bh,
 236                                     int resize)
 237 {
 238         unsigned int max_bits;
 239         struct ocfs2_group_desc *gd = (struct ocfs2_group_desc *)bh->b_data;
 240
 241         if (di->i_blkno != gd->bg_parent_dinode) {
 242                 do_error("Group descriptor #%llu has bad parent "
 243                          "pointer (%llu, expected %llu)",
 244                          (unsigned long long)bh->b_blocknr,
 245                          (unsigned long long)le64_to_cpu(gd->bg_parent_dinode),
 246                          (unsigned long long)le64_to_cpu(di->i_blkno));
 247                 return -EINVAL;
 248         }
 249
 250         max_bits = le16_to_cpu(di->id2.i_chain.cl_cpg) * le16_to_cpu(di->id2.i_chain.cl_bpc);
 251         if (le16_to_cpu(gd->bg_bits) > max_bits) {
 252                 do_error("Group descriptor #%llu has bit count of %u",
 253                          (unsigned long long)bh->b_blocknr,
 254                          le16_to_cpu(gd->bg_bits));
 255                 return -EINVAL;
 256         }
 257
 258         /* In resize, we may meet the case bg_chain == cl_next_free_rec. */
 259         if ((le16_to_cpu(gd->bg_chain) >
 260              le16_to_cpu(di->id2.i_chain.cl_next_free_rec)) ||
 261             ((le16_to_cpu(gd->bg_chain) ==
 262              le16_to_cpu(di->id2.i_chain.cl_next_free_rec)) && !resize)) {
 263                 do_error("Group descriptor #%llu has bad chain %u",
 264                          (unsigned long long)bh->b_blocknr,
 265                          le16_to_cpu(gd->bg_chain));
 266                 return -EINVAL;
 267         }
 268
 269         return 0;
 270 }
 271
 272 #undef do_error
 273
 274 /*
 275  * This version only prints errors.  It does not fail the filesystem, and
 276  * exists only for resize.
 277  */
 278 int ocfs2_check_group_descriptor(struct super_block *sb,
 279                                  struct ocfs2_dinode *di,
 280                                  struct buffer_head *bh)
 281 {
 282         int rc;
 283         struct ocfs2_group_desc *gd = (struct ocfs2_group_desc *)bh->b_data;
 284
 285         BUG_ON(!buffer_uptodate(bh));
 286
 287         /*
 288          * If the ecc fails, we return the error but otherwise
 289          * leave the filesystem running.  We know any error is
 290          * local to this block.
 291          */
 292         rc = ocfs2_validate_meta_ecc(sb, bh->b_data, &gd->bg_check);
 293         if (rc) {
 294                 mlog(ML_ERROR,
 295                      "Checksum failed for group descriptor %llu\n",
 296                      (unsigned long long)bh->b_blocknr);
 297         } else
 298                 rc = ocfs2_validate_gd_self(sb, bh, 1);
 299         if (!rc)
 300                 rc = ocfs2_validate_gd_parent(sb, di, bh, 1);
 301
 302         return rc;
 303 }
 304
 305 static int ocfs2_validate_group_descriptor(struct super_block *sb,
 306                                            struct buffer_head *bh)
 307 {
 308         int rc;
 309         struct ocfs2_group_desc *gd = (struct ocfs2_group_desc *)bh->b_data;
 310
 311         trace_ocfs2_validate_group_descriptor(
 312                                         (unsigned long long)bh->b_blocknr);
 313
 314         BUG_ON(!buffer_uptodate(bh));
 315
 316         /*
 317          * If the ecc fails, we return the error but otherwise
 318          * leave the filesystem running.  We know any error is
 319          * local to this block.
 320          */
 321         rc = ocfs2_validate_meta_ecc(sb, bh->b_data, &gd->bg_check);
 322         if (rc)
 323                 return rc;
 324
 325         /*
 326          * Errors after here are fatal.
 327          */
 328
 329         return ocfs2_validate_gd_self(sb, bh, 0);
 330 }
 331
 332 int ocfs2_read_group_descriptor(struct inode *inode, struct ocfs2_dinode *di,
 333                                 u64 gd_blkno, struct buffer_head **bh)
 334 {
 335         int rc;
 336         struct buffer_head *tmp = *bh;
 337
 338         rc = ocfs2_read_block(INODE_CACHE(inode), gd_blkno, &tmp,
 339                               ocfs2_validate_group_descriptor);
 340         if (rc)
 341                 goto out;
 342
 343         rc = ocfs2_validate_gd_parent(inode->i_sb, di, tmp, 0);
 344         if (rc) {
 345                 brelse(tmp);
 346                 goto out;
 347         }
 348
 349         /* If ocfs2_read_block() got us a new bh, pass it up. */
 350         if (!*bh)
 351                 *bh = tmp;
 352
 353 out:
 354         return rc;
 355 }
 356
 357 static void ocfs2_bg_discontig_add_extent(struct ocfs2_super *osb,
 358                                           struct ocfs2_group_desc *bg,
 359                                           struct ocfs2_chain_list *cl,
 360                                           u64 p_blkno, unsigned int clusters)
 361 {
 362         struct ocfs2_extent_list *el = &bg->bg_list;
 363         struct ocfs2_extent_rec *rec;
 364
 365         BUG_ON(!ocfs2_supports_discontig_bg(osb));
 366         if (!el->l_next_free_rec)
 367                 el->l_count = cpu_to_le16(ocfs2_extent_recs_per_gd(osb->sb));
 368         rec = &el->l_recs[le16_to_cpu(el->l_next_free_rec)];
 369         rec->e_blkno = cpu_to_le64(p_blkno);
 370         rec->e_cpos = cpu_to_le32(le16_to_cpu(bg->bg_bits) /
 371                                   le16_to_cpu(cl->cl_bpc));
 372         rec->e_leaf_clusters = cpu_to_le16(clusters);
 373         le16_add_cpu(&bg->bg_bits, clusters * le16_to_cpu(cl->cl_bpc));
 374         le16_add_cpu(&bg->bg_free_bits_count,
 375                      clusters * le16_to_cpu(cl->cl_bpc));
 376         le16_add_cpu(&el->l_next_free_rec, 1);
 377 }
 378
 379 static int ocfs2_block_group_fill(handle_t *handle,
 380                                   struct inode *alloc_inode,
 381                                   struct buffer_head *bg_bh,
 382                                   u64 group_blkno,
 383                                   unsigned int group_clusters,
 384                                   u16 my_chain,
 385                                   struct ocfs2_chain_list *cl)
 386 {
 387         int status = 0;
 388         struct ocfs2_super *osb = OCFS2_SB(alloc_inode->i_sb);
 389         struct ocfs2_group_desc *bg = (struct ocfs2_group_desc *) bg_bh->b_data;
 390         struct super_block * sb = alloc_inode->i_sb;
 391
 392         if (((unsigned long long) bg_bh->b_blocknr) != group_blkno) {
 393                 ocfs2_error(alloc_inode->i_sb, "group block (%llu) != "
 394                             "b_blocknr (%llu)",
 395                             (unsigned long long)group_blkno,
 396                             (unsigned long long) bg_bh->b_blocknr);
 397                 status = -EIO;
 398                 goto bail;
 399         }
 400
 401         status = ocfs2_journal_access_gd(handle,
 402                                          INODE_CACHE(alloc_inode),
 403                                          bg_bh,
 404                                          OCFS2_JOURNAL_ACCESS_CREATE);
 405         if (status < 0) {
 406                 mlog_errno(status);
 407                 goto bail;
 408         }
 409
 410         memset(bg, 0, sb->s_blocksize);
 411         strcpy(bg->bg_signature, OCFS2_GROUP_DESC_SIGNATURE);
 412         bg->bg_generation = cpu_to_le32(OCFS2_SB(sb)->fs_generation);
 413         bg->bg_size = cpu_to_le16(ocfs2_group_bitmap_size(sb, 1,
 414                                                 osb->s_feature_incompat));
 415         bg->bg_chain = cpu_to_le16(my_chain);
 416         bg->bg_next_group = cl->cl_recs[my_chain].c_blkno;
 417         bg->bg_parent_dinode = cpu_to_le64(OCFS2_I(alloc_inode)->ip_blkno);
 418         bg->bg_blkno = cpu_to_le64(group_blkno);
 419         if (group_clusters == le16_to_cpu(cl->cl_cpg))
 420                 bg->bg_bits = cpu_to_le16(ocfs2_bits_per_group(cl));
 421         else
 422                 ocfs2_bg_discontig_add_extent(osb, bg, cl, group_blkno,
 423                                               group_clusters);
 424
 425         /* set the 1st bit in the bitmap to account for the descriptor block */
 426         ocfs2_set_bit(0, (unsigned long *)bg->bg_bitmap);
 427         bg->bg_free_bits_count = cpu_to_le16(le16_to_cpu(bg->bg_bits) - 1);
 428
 429         ocfs2_journal_dirty(handle, bg_bh);
 430
 431         /* There is no need to zero out or otherwise initialize the
 432          * other blocks in a group - All valid FS metadata in a block
 433          * group stores the superblock fs_generation value at
 434          * allocation time. */
 435
 436 bail:
 437         if (status)
 438                 mlog_errno(status);
 439         return status;
 440 }
 441
 442 static inline u16 ocfs2_find_smallest_chain(struct ocfs2_chain_list *cl)
 443 {
 444         u16 curr, best;
 445
 446         best = curr = 0;
 447         while (curr < le16_to_cpu(cl->cl_count)) {
 448                 if (le32_to_cpu(cl->cl_recs[best].c_total) >
 449                     le32_to_cpu(cl->cl_recs[curr].c_total))
 450                         best = curr;
 451                 curr++;
 452         }
 453         return best;
 454 }
 455
 456 static struct buffer_head *
 457 ocfs2_block_group_alloc_contig(struct ocfs2_super *osb, handle_t *handle,
 458                                struct inode *alloc_inode,
 459                                struct ocfs2_alloc_context *ac,
 460                                struct ocfs2_chain_list *cl)
 461 {
 462         int status;
 463         u32 bit_off, num_bits;
 464         u64 bg_blkno;
 465         struct buffer_head *bg_bh;
 466         unsigned int alloc_rec = ocfs2_find_smallest_chain(cl);
 467
 468         status = ocfs2_claim_clusters(handle, ac,
 469                                       le16_to_cpu(cl->cl_cpg), &bit_off,
 470                                       &num_bits);
 471         if (status < 0) {
 472                 if (status != -ENOSPC)
 473                         mlog_errno(status);
 474                 goto bail;
 475         }
 476
 477         /* setup the group */
 478         bg_blkno = ocfs2_clusters_to_blocks(osb->sb, bit_off);
 479         trace_ocfs2_block_group_alloc_contig(
 480              (unsigned long long)bg_blkno, alloc_rec);
 481
 482         bg_bh = sb_getblk(osb->sb, bg_blkno);
 483         if (!bg_bh) {
 484                 status = -EIO;
 485                 mlog_errno(status);
 486                 goto bail;
 487         }
 488         ocfs2_set_new_buffer_uptodate(INODE_CACHE(alloc_inode), bg_bh);
 489
 490         status = ocfs2_block_group_fill(handle, alloc_inode, bg_bh,
 491                                         bg_blkno, num_bits, alloc_rec, cl);
 492         if (status < 0) {
 493                 brelse(bg_bh);
 494                 mlog_errno(status);
 495         }
 496
 497 bail:
 498         return status ? ERR_PTR(status) : bg_bh;
 499 }
 500
 501 static int ocfs2_block_group_claim_bits(struct ocfs2_super *osb,
 502                                         handle_t *handle,
 503                                         struct ocfs2_alloc_context *ac,
 504                                         unsigned int min_bits,
 505                                         u32 *bit_off, u32 *num_bits)
 506 {
 507         int status = 0;
 508
 509         while (min_bits) {
 510                 status = ocfs2_claim_clusters(handle, ac, min_bits,
 511                                               bit_off, num_bits);
 512                 if (status != -ENOSPC)
 513                         break;
 514
 515                 min_bits >>= 1;
 516         }
 517
 518         return status;
 519 }
 520
 521 static int ocfs2_block_group_grow_discontig(handle_t *handle,
 522                                             struct inode *alloc_inode,
 523                                             struct buffer_head *bg_bh,
 524                                             struct ocfs2_alloc_context *ac,
 525                                             struct ocfs2_chain_list *cl,
 526                                             unsigned int min_bits)
 527 {
 528         int status;
 529         struct ocfs2_super *osb = OCFS2_SB(alloc_inode->i_sb);
 530         struct ocfs2_group_desc *bg =
 531                 (struct ocfs2_group_desc *)bg_bh->b_data;
 532         unsigned int needed = le16_to_cpu(cl->cl_cpg) -
 533                          le16_to_cpu(bg->bg_bits) / le16_to_cpu(cl->cl_bpc);
 534         u32 p_cpos, clusters;
 535         u64 p_blkno;
 536         struct ocfs2_extent_list *el = &bg->bg_list;
 537
 538         status = ocfs2_journal_access_gd(handle,
 539                                          INODE_CACHE(alloc_inode),
 540                                          bg_bh,
 541                                          OCFS2_JOURNAL_ACCESS_CREATE);
 542         if (status < 0) {
 543                 mlog_errno(status);
 544                 goto bail;
 545         }
 546
 547         while ((needed > 0) && (le16_to_cpu(el->l_next_free_rec) <
 548                                 le16_to_cpu(el->l_count))) {
 549                 if (min_bits > needed)
 550                         min_bits = needed;
 551                 status = ocfs2_block_group_claim_bits(osb, handle, ac,
 552                                                       min_bits, &p_cpos,
 553                                                       &clusters);
 554                 if (status < 0) {
 555                         if (status != -ENOSPC)
 556                                 mlog_errno(status);
 557                         goto bail;
 558                 }
 559                 p_blkno = ocfs2_clusters_to_blocks(osb->sb, p_cpos);
 560                 ocfs2_bg_discontig_add_extent(osb, bg, cl, p_blkno,
 561                                               clusters);
 562
 563                 min_bits = clusters;
 564                 needed = le16_to_cpu(cl->cl_cpg) -
 565                          le16_to_cpu(bg->bg_bits) / le16_to_cpu(cl->cl_bpc);
 566         }
 567
 568         if (needed > 0) {
 569                 /*
 570                  * We have used up all the extent rec but can't fill up
 571                  * the cpg. So bail out.
 572                  */
 573                 status = -ENOSPC;
 574                 goto bail;
 575         }
 576
 577         ocfs2_journal_dirty(handle, bg_bh);
 578
 579 bail:
 580         return status;
 581 }
 582
 583 static void ocfs2_bg_alloc_cleanup(handle_t *handle,
 584                                    struct ocfs2_alloc_context *cluster_ac,
 585                                    struct inode *alloc_inode,
 586                                    struct buffer_head *bg_bh)
 587 {
 588         int i, ret;
 589         struct ocfs2_group_desc *bg;
 590         struct ocfs2_extent_list *el;
 591         struct ocfs2_extent_rec *rec;
 592
 593         if (!bg_bh)
 594                 return;
 595
 596         bg = (struct ocfs2_group_desc *)bg_bh->b_data;
 597         el = &bg->bg_list;
 598         for (i = 0; i < le16_to_cpu(el->l_next_free_rec); i++) {
 599                 rec = &el->l_recs[i];
 600                 ret = ocfs2_free_clusters(handle, cluster_ac->ac_inode,
 601                                           cluster_ac->ac_bh,
 602                                           le64_to_cpu(rec->e_blkno),
 603                                           le16_to_cpu(rec->e_leaf_clusters));
 604                 if (ret)
 605                         mlog_errno(ret);
 606                 /* Try all the clusters to free */
 607         }
 608
 609         ocfs2_remove_from_cache(INODE_CACHE(alloc_inode), bg_bh);
 610         brelse(bg_bh);
 611 }
 612
 613 static struct buffer_head *
 614 ocfs2_block_group_alloc_discontig(handle_t *handle,
 615                                   struct inode *alloc_inode,
 616                                   struct ocfs2_alloc_context *ac,
 617                                   struct ocfs2_chain_list *cl)
 618 {
 619         int status;
 620         u32 bit_off, num_bits;
 621         u64 bg_blkno;
 622         unsigned int min_bits = le16_to_cpu(cl->cl_cpg) >> 1;
 623         struct buffer_head *bg_bh = NULL;
 624         unsigned int alloc_rec = ocfs2_find_smallest_chain(cl);
 625         struct ocfs2_super *osb = OCFS2_SB(alloc_inode->i_sb);
 626
 627         if (!ocfs2_supports_discontig_bg(osb)) {
 628                 status = -ENOSPC;
 629                 goto bail;
 630         }
 631
 632         status = ocfs2_extend_trans(handle,
 633                                     ocfs2_calc_bg_discontig_credits(osb->sb));
 634         if (status) {
 635                 mlog_errno(status);
 636                 goto bail;
 637         }
 638
 639         /*
 640          * We're going to be grabbing from multiple cluster groups.
 641          * We don't have enough credits to relink them all, and the
 642          * cluster groups will be staying in cache for the duration of
 643          * this operation.
 644          */
 645         ac->ac_allow_chain_relink = 0;
 646
 647         /* Claim the first region */
 648         status = ocfs2_block_group_claim_bits(osb, handle, ac, min_bits,
 649                                               &bit_off, &num_bits);
 650         if (status < 0) {
 651                 if (status != -ENOSPC)
 652                         mlog_errno(status);
 653                 goto bail;
 654         }
 655         min_bits = num_bits;
 656
 657         /* setup the group */
 658         bg_blkno = ocfs2_clusters_to_blocks(osb->sb, bit_off);
 659         trace_ocfs2_block_group_alloc_discontig(
 660                                 (unsigned long long)bg_blkno, alloc_rec);
 661
 662         bg_bh = sb_getblk(osb->sb, bg_blkno);
 663         if (!bg_bh) {
 664                 status = -EIO;
 665                 mlog_errno(status);
 666                 goto bail;
 667         }
 668         ocfs2_set_new_buffer_uptodate(INODE_CACHE(alloc_inode), bg_bh);
 669
 670         status = ocfs2_block_group_fill(handle, alloc_inode, bg_bh,
 671                                         bg_blkno, num_bits, alloc_rec, cl);
 672         if (status < 0) {
 673                 mlog_errno(status);
 674                 goto bail;
 675         }
 676
 677         status = ocfs2_block_group_grow_discontig(handle, alloc_inode,
 678                                                   bg_bh, ac, cl, min_bits);
 679         if (status)
 680                 mlog_errno(status);
 681
 682 bail:
 683         if (status)
 684                 ocfs2_bg_alloc_cleanup(handle, ac, alloc_inode, bg_bh);
 685         return status ? ERR_PTR(status) : bg_bh;
 686 }
 687
 688 /*
 689  * We expect the block group allocator to already be locked.
 690  */
 691 static int ocfs2_block_group_alloc(struct ocfs2_super *osb,
 692                                    struct inode *alloc_inode,
 693                                    struct buffer_head *bh,
 694                                    u64 max_block,
 695                                    u64 *last_alloc_group,
 696                                    int flags)
 697 {
 698         int status, credits;
 699         struct ocfs2_dinode *fe = (struct ocfs2_dinode *) bh->b_data;
 700         struct ocfs2_chain_list *cl;
 701         struct ocfs2_alloc_context *ac = NULL;
 702         handle_t *handle = NULL;
 703         u16 alloc_rec;
 704         struct buffer_head *bg_bh = NULL;
 705         struct ocfs2_group_desc *bg;
 706
 707         BUG_ON(ocfs2_is_cluster_bitmap(alloc_inode));
 708
 709         cl = &fe->id2.i_chain;
 710         status = ocfs2_reserve_clusters_with_limit(osb,
 711                                                    le16_to_cpu(cl->cl_cpg),
 712                                                    max_block, flags, &ac);
 713         if (status < 0) {
 714                 if (status != -ENOSPC)
 715                         mlog_errno(status);
 716                 goto bail;
 717         }
 718
 719         credits = ocfs2_calc_group_alloc_credits(osb->sb,
 720                                                  le16_to_cpu(cl->cl_cpg));
 721         handle = ocfs2_start_trans(osb, credits);
 722         if (IS_ERR(handle)) {
 723                 status = PTR_ERR(handle);
 724                 handle = NULL;
 725                 mlog_errno(status);
 726                 goto bail;
 727         }
 728
 729         if (last_alloc_group && *last_alloc_group != 0) {
 730                 trace_ocfs2_block_group_alloc(
 731                                 (unsigned long long)*last_alloc_group);
 732                 ac->ac_last_group = *last_alloc_group;
 733         }
 734
 735         bg_bh = ocfs2_block_group_alloc_contig(osb, handle, alloc_inode,
 736                                                ac, cl);
 737         if (IS_ERR(bg_bh) && (PTR_ERR(bg_bh) == -ENOSPC))
 738                 bg_bh = ocfs2_block_group_alloc_discontig(handle,
 739                                                           alloc_inode,
 740                                                           ac, cl);
 741         if (IS_ERR(bg_bh)) {
 742                 status = PTR_ERR(bg_bh);
 743                 bg_bh = NULL;
 744                 if (status != -ENOSPC)
 745                         mlog_errno(status);
 746                 goto bail;
 747         }
 748         bg = (struct ocfs2_group_desc *) bg_bh->b_data;
 749
 750         status = ocfs2_journal_access_di(handle, INODE_CACHE(alloc_inode),
 751                                          bh, OCFS2_JOURNAL_ACCESS_WRITE);
 752         if (status < 0) {
 753                 mlog_errno(status);
 754                 goto bail;
 755         }
 756
 757         alloc_rec = le16_to_cpu(bg->bg_chain);
 758         le32_add_cpu(&cl->cl_recs[alloc_rec].c_free,
 759                      le16_to_cpu(bg->bg_free_bits_count));
 760         le32_add_cpu(&cl->cl_recs[alloc_rec].c_total,
 761                      le16_to_cpu(bg->bg_bits));
 762         cl->cl_recs[alloc_rec].c_blkno = bg->bg_blkno;
 763         if (le16_to_cpu(cl->cl_next_free_rec) < le16_to_cpu(cl->cl_count))
 764                 le16_add_cpu(&cl->cl_next_free_rec, 1);
 765
 766         le32_add_cpu(&fe->id1.bitmap1.i_used, le16_to_cpu(bg->bg_bits) -
 767                                         le16_to_cpu(bg->bg_free_bits_count));
 768         le32_add_cpu(&fe->id1.bitmap1.i_total, le16_to_cpu(bg->bg_bits));
 769         le32_add_cpu(&fe->i_clusters, le16_to_cpu(cl->cl_cpg));
 770
 771         ocfs2_journal_dirty(handle, bh);
 772
 773         spin_lock(&OCFS2_I(alloc_inode)->ip_lock);
 774         OCFS2_I(alloc_inode)->ip_clusters = le32_to_cpu(fe->i_clusters);
 775         fe->i_size = cpu_to_le64(ocfs2_clusters_to_bytes(alloc_inode->i_sb,
 776                                              le32_to_cpu(fe->i_clusters)));
 777         spin_unlock(&OCFS2_I(alloc_inode)->ip_lock);
 778         i_size_write(alloc_inode, le64_to_cpu(fe->i_size));
 779         alloc_inode->i_blocks = ocfs2_inode_sector_count(alloc_inode);
 780
 781         status = 0;
 782
 783         /* save the new last alloc group so that the caller can cache it. */
 784         if (last_alloc_group)
 785                 *last_alloc_group = ac->ac_last_group;
 786
 787 bail:
 788         if (handle)
 789                 ocfs2_commit_trans(osb, handle);
 790
 791         if (ac)
 792                 ocfs2_free_alloc_context(ac);
 793
 794         brelse(bg_bh);
 795
 796         if (status)
 797                 mlog_errno(status);
 798         return status;
 799 }
 800
 801 static int ocfs2_reserve_suballoc_bits(struct ocfs2_super *osb,
 802                                        struct ocfs2_alloc_context *ac,
 803                                        int type,
 804                                        u32 slot,
 805                                        u64 *last_alloc_group,
 806                                        int flags)
 807 {
 808         int status;
 809         u32 bits_wanted = ac->ac_bits_wanted;
 810         struct inode *alloc_inode;
 811         struct buffer_head *bh = NULL;
 812         struct ocfs2_dinode *fe;
 813         u32 free_bits;
 814
 815         alloc_inode = ocfs2_get_system_file_inode(osb, type, slot);
 816         if (!alloc_inode) {
 817                 mlog_errno(-EINVAL);
 818                 return -EINVAL;
 819         }
 820
 821         mutex_lock(&alloc_inode->i_mutex);
 822
 823         status = ocfs2_inode_lock(alloc_inode, &bh, 1);
 824         if (status < 0) {
 825                 mutex_unlock(&alloc_inode->i_mutex);
 826                 iput(alloc_inode);
 827
 828                 mlog_errno(status);
 829                 return status;
 830         }
 831
 832         ac->ac_inode = alloc_inode;
 833         ac->ac_alloc_slot = slot;
 834
 835         fe = (struct ocfs2_dinode *) bh->b_data;
 836
 837         /* The bh was validated by the inode read inside
 838          * ocfs2_inode_lock().  Any corruption is a code bug. */
 839         BUG_ON(!OCFS2_IS_VALID_DINODE(fe));
 840
 841         if (!(fe->i_flags & cpu_to_le32(OCFS2_CHAIN_FL))) {
 842                 ocfs2_error(alloc_inode->i_sb, "Invalid chain allocator %llu",
 843                             (unsigned long long)le64_to_cpu(fe->i_blkno));
 844                 status = -EIO;
 845                 goto bail;
 846         }
 847
 848         free_bits = le32_to_cpu(fe->id1.bitmap1.i_total) -
 849                 le32_to_cpu(fe->id1.bitmap1.i_used);
 850
 851         if (bits_wanted > free_bits) {
 852                 /* cluster bitmap never grows */
 853                 if (ocfs2_is_cluster_bitmap(alloc_inode)) {
 854                         trace_ocfs2_reserve_suballoc_bits_nospc(bits_wanted,
 855                                                                 free_bits);
 856                         status = -ENOSPC;
 857                         goto bail;
 858                 }
 859
 860                 if (!(flags & ALLOC_NEW_GROUP)) {
 861                         trace_ocfs2_reserve_suballoc_bits_no_new_group(
 862                                                 slot, bits_wanted, free_bits);
 863                         status = -ENOSPC;
 864                         goto bail;
 865                 }
 866
 867                 status = ocfs2_block_group_alloc(osb, alloc_inode, bh,
 868                                                  ac->ac_max_block,
 869                                                  last_alloc_group, flags);
 870                 if (status < 0) {
 871                         if (status != -ENOSPC)
 872                                 mlog_errno(status);
 873                         goto bail;
 874                 }
 875                 atomic_inc(&osb->alloc_stats.bg_extends);
 876
 877                 /* You should never ask for this much metadata */
 878                 BUG_ON(bits_wanted >
 879                        (le32_to_cpu(fe->id1.bitmap1.i_total)
 880                         - le32_to_cpu(fe->id1.bitmap1.i_used)));
 881         }
 882
 883         get_bh(bh);
 884         ac->ac_bh = bh;
 885 bail:
 886         brelse(bh);
 887
 888         if (status)
 889                 mlog_errno(status);
 890         return status;
 891 }
 892
 893 static void ocfs2_init_inode_steal_slot(struct ocfs2_super *osb)
 894 {
 895         spin_lock(&osb->osb_lock);
 896         osb->s_inode_steal_slot = OCFS2_INVALID_SLOT;
 897         spin_unlock(&osb->osb_lock);
 898         atomic_set(&osb->s_num_inodes_stolen, 0);
 899 }
 900
 901 static void ocfs2_init_meta_steal_slot(struct ocfs2_super *osb)
 902 {
 903         spin_lock(&osb->osb_lock);
 904         osb->s_meta_steal_slot = OCFS2_INVALID_SLOT;
 905         spin_unlock(&osb->osb_lock);
 906         atomic_set(&osb->s_num_meta_stolen, 0);
 907 }
 908
 909 void ocfs2_init_steal_slots(struct ocfs2_super *osb)
 910 {
 911         ocfs2_init_inode_steal_slot(osb);
 912         ocfs2_init_meta_steal_slot(osb);
 913 }
 914
 915 static void __ocfs2_set_steal_slot(struct ocfs2_super *osb, int slot, int type)
 916 {
 917         spin_lock(&osb->osb_lock);
 918         if (type == INODE_ALLOC_SYSTEM_INODE)
 919                 osb->s_inode_steal_slot = slot;
 920         else if (type == EXTENT_ALLOC_SYSTEM_INODE)
 921                 osb->s_meta_steal_slot = slot;
 922         spin_unlock(&osb->osb_lock);
 923 }
 924
 925 static int __ocfs2_get_steal_slot(struct ocfs2_super *osb, int type)
 926 {
 927         int slot = OCFS2_INVALID_SLOT;
 928
 929         spin_lock(&osb->osb_lock);
 930         if (type == INODE_ALLOC_SYSTEM_INODE)
 931                 slot = osb->s_inode_steal_slot;
 932         else if (type == EXTENT_ALLOC_SYSTEM_INODE)
 933                 slot = osb->s_meta_steal_slot;
 934         spin_unlock(&osb->osb_lock);
 935
 936         return slot;
 937 }
 938
 939 static int ocfs2_get_inode_steal_slot(struct ocfs2_super *osb)
 940 {
 941         return __ocfs2_get_steal_slot(osb, INODE_ALLOC_SYSTEM_INODE);
 942 }
 943
 944 static int ocfs2_get_meta_steal_slot(struct ocfs2_super *osb)
 945 {
 946         return __ocfs2_get_steal_slot(osb, EXTENT_ALLOC_SYSTEM_INODE);
 947 }
 948
 949 static int ocfs2_steal_resource(struct ocfs2_super *osb,
 950                                 struct ocfs2_alloc_context *ac,
 951                                 int type)
 952 {
 953         int i, status = -ENOSPC;
 954         int slot = __ocfs2_get_steal_slot(osb, type);
 955
 956         /* Start to steal resource from the first slot after ours. */
 957         if (slot == OCFS2_INVALID_SLOT)
 958                 slot = osb->slot_num + 1;
 959
 960         for (i = 0; i < osb->max_slots; i++, slot++) {
 961                 if (slot == osb->max_slots)
 962                         slot = 0;
 963
 964                 if (slot == osb->slot_num)
 965                         continue;
 966
 967                 status = ocfs2_reserve_suballoc_bits(osb, ac,
 968                                                      type,
 969                                                      (u32)slot, NULL,
 970                                                      NOT_ALLOC_NEW_GROUP);
 971                 if (status >= 0) {
 972                         __ocfs2_set_steal_slot(osb, slot, type);
 973                         break;
 974                 }
 975
 976                 ocfs2_free_ac_resource(ac);
 977         }
 978
 979         return status;
 980 }
 981
 982 static int ocfs2_steal_inode(struct ocfs2_super *osb,
 983                              struct ocfs2_alloc_context *ac)
 984 {
 985         return ocfs2_steal_resource(osb, ac, INODE_ALLOC_SYSTEM_INODE);
 986 }
 987
 988 static int ocfs2_steal_meta(struct ocfs2_super *osb,
 989                             struct ocfs2_alloc_context *ac)
 990 {
 991         return ocfs2_steal_resource(osb, ac, EXTENT_ALLOC_SYSTEM_INODE);
 992 }
 993
 994 int ocfs2_reserve_new_metadata_blocks(struct ocfs2_super *osb,
 995                                       int blocks,
 996                                       struct ocfs2_alloc_context **ac)
 997 {
 998         int status;
 999         int slot = ocfs2_get_meta_steal_slot(osb);
1000
1001         *ac = kzalloc(sizeof(struct ocfs2_alloc_context), GFP_KERNEL);
1002         if (!(*ac)) {
1003                 status = -ENOMEM;
1004                 mlog_errno(status);
1005                 goto bail;
1006         }
1007
1008         (*ac)->ac_bits_wanted = blocks;
1009         (*ac)->ac_which = OCFS2_AC_USE_META;
1010         (*ac)->ac_group_search = ocfs2_block_group_search;
1011
1012         if (slot != OCFS2_INVALID_SLOT &&
1013                 atomic_read(&osb->s_num_meta_stolen) < OCFS2_MAX_TO_STEAL)
1014                 goto extent_steal;
1015
1016         atomic_set(&osb->s_num_meta_stolen, 0);
1017         status = ocfs2_reserve_suballoc_bits(osb, (*ac),
1018                                              EXTENT_ALLOC_SYSTEM_INODE,
1019                                              (u32)osb->slot_num, NULL,
1020                                              ALLOC_GROUPS_FROM_GLOBAL|ALLOC_NEW_GROUP);
1021
1022
1023         if (status >= 0) {
1024                 status = 0;
1025                 if (slot != OCFS2_INVALID_SLOT)
1026                         ocfs2_init_meta_steal_slot(osb);
1027                 goto bail;
1028         } else if (status < 0 && status != -ENOSPC) {
1029                 mlog_errno(status);
1030                 goto bail;
1031         }
1032
1033         ocfs2_free_ac_resource(*ac);
1034
1035 extent_steal:
1036         status = ocfs2_steal_meta(osb, *ac);
1037         atomic_inc(&osb->s_num_meta_stolen);
1038         if (status < 0) {
1039                 if (status != -ENOSPC)
1040                         mlog_errno(status);
1041                 goto bail;
1042         }
1043
1044         status = 0;
1045 bail:
1046         if ((status < 0) && *ac) {
1047                 ocfs2_free_alloc_context(*ac);
1048                 *ac = NULL;
1049         }
1050
1051         if (status)
1052                 mlog_errno(status);
1053         return status;
1054 }
1055
1056 int ocfs2_reserve_new_metadata(struct ocfs2_super *osb,
1057                                struct ocfs2_extent_list *root_el,
1058                                struct ocfs2_alloc_context **ac)
1059 {
1060         return ocfs2_reserve_new_metadata_blocks(osb,
1061                                         ocfs2_extend_meta_needed(root_el),
1062                                         ac);
1063 }
1064
1065 int ocfs2_reserve_new_inode(struct ocfs2_super *osb,
1066                             struct ocfs2_alloc_context **ac)
1067 {
1068         int status;
1069         int slot = ocfs2_get_inode_steal_slot(osb);
1070         u64 alloc_group;
1071
1072         *ac = kzalloc(sizeof(struct ocfs2_alloc_context), GFP_KERNEL);
1073         if (!(*ac)) {
1074                 status = -ENOMEM;
1075                 mlog_errno(status);
1076                 goto bail;
1077         }
1078
1079         (*ac)->ac_bits_wanted = 1;
1080         (*ac)->ac_which = OCFS2_AC_USE_INODE;
1081
1082         (*ac)->ac_group_search = ocfs2_block_group_search;
1083
1084         /*
1085          * stat(2) can't handle i_ino > 32bits, so we tell the
1086          * lower levels not to allocate us a block group past that
1087          * limit.  The 'inode64' mount option avoids this behavior.
1088          */
1089         if (!(osb->s_mount_opt & OCFS2_MOUNT_INODE64))
1090                 (*ac)->ac_max_block = (u32)~0U;
1091
1092         /*
1093          * slot is set when we successfully steal inode from other nodes.
1094          * It is reset in 3 places:
1095          * 1. when we flush the truncate log
1096          * 2. when we complete local alloc recovery.
1097          * 3. when we successfully allocate from our own slot.
1098          * After it is set, we will go on stealing inodes until we find the
1099          * need to check our slots to see whether there is some space for us.
1100          */
1101         if (slot != OCFS2_INVALID_SLOT &&
1102             atomic_read(&osb->s_num_inodes_stolen) < OCFS2_MAX_TO_STEAL)
1103                 goto inode_steal;
1104
1105         atomic_set(&osb->s_num_inodes_stolen, 0);
1106         alloc_group = osb->osb_inode_alloc_group;
1107         status = ocfs2_reserve_suballoc_bits(osb, *ac,
1108                                              INODE_ALLOC_SYSTEM_INODE,
1109                                              (u32)osb->slot_num,
1110                                              &alloc_group,
1111                                              ALLOC_NEW_GROUP |
1112                                              ALLOC_GROUPS_FROM_GLOBAL);
1113         if (status >= 0) {
1114                 status = 0;
1115
1116                 spin_lock(&osb->osb_lock);
1117                 osb->osb_inode_alloc_group = alloc_group;
1118                 spin_unlock(&osb->osb_lock);
1119                 trace_ocfs2_reserve_new_inode_new_group(
1120                         (unsigned long long)alloc_group);
1121
1122                 /*
1123                  * Some inodes must be freed by us, so try to allocate
1124                  * from our own next time.
1125                  */
1126                 if (slot != OCFS2_INVALID_SLOT)
1127                         ocfs2_init_inode_steal_slot(osb);
1128                 goto bail;
1129         } else if (status < 0 && status != -ENOSPC) {
1130                 mlog_errno(status);
1131                 goto bail;
1132         }
1133
1134         ocfs2_free_ac_resource(*ac);
1135
1136 inode_steal:
1137         status = ocfs2_steal_inode(osb, *ac);
1138         atomic_inc(&osb->s_num_inodes_stolen);
1139         if (status < 0) {
1140                 if (status != -ENOSPC)
1141                         mlog_errno(status);
1142                 goto bail;
1143         }
1144
1145         status = 0;
1146 bail:
1147         if ((status < 0) && *ac) {
1148                 ocfs2_free_alloc_context(*ac);
1149                 *ac = NULL;
1150         }
1151
1152         if (status)
1153                 mlog_errno(status);
1154         return status;
1155 }
1156
1157 /* local alloc code has to do the same thing, so rather than do this
1158  * twice.. */
1159 int ocfs2_reserve_cluster_bitmap_bits(struct ocfs2_super *osb,
1160                                       struct ocfs2_alloc_context *ac)
1161 {
1162         int status;
1163
1164         ac->ac_which = OCFS2_AC_USE_MAIN;
1165         ac->ac_group_search = ocfs2_cluster_group_search;
1166
1167         status = ocfs2_reserve_suballoc_bits(osb, ac,
1168                                              GLOBAL_BITMAP_SYSTEM_INODE,
1169                                              OCFS2_INVALID_SLOT, NULL,
1170                                              ALLOC_NEW_GROUP);
1171         if (status < 0 && status != -ENOSPC) {
1172                 mlog_errno(status);
1173                 goto bail;
1174         }
1175
1176 bail:
1177         return status;
1178 }
1179
1180 /* Callers don't need to care which bitmap (local alloc or main) to
1181  * use so we figure it out for them, but unfortunately this clutters
1182  * things a bit. */
1183 static int ocfs2_reserve_clusters_with_limit(struct ocfs2_super *osb,
1184                                              u32 bits_wanted, u64 max_block,
1185                                              int flags,
1186                                              struct ocfs2_alloc_context **ac)
1187 {
1188         int status;
1189
1190         *ac = kzalloc(sizeof(struct ocfs2_alloc_context), GFP_KERNEL);
1191         if (!(*ac)) {
1192                 status = -ENOMEM;
1193                 mlog_errno(status);
1194                 goto bail;
1195         }
1196
1197         (*ac)->ac_bits_wanted = bits_wanted;
1198         (*ac)->ac_max_block = max_block;
1199
1200         status = -ENOSPC;
1201         if (!(flags & ALLOC_GROUPS_FROM_GLOBAL) &&
1202             ocfs2_alloc_should_use_local(osb, bits_wanted)) {
1203                 status = ocfs2_reserve_local_alloc_bits(osb,
1204                                                         bits_wanted,
1205                                                         *ac);
1206                 if ((status < 0) && (status != -ENOSPC)) {
1207                         mlog_errno(status);
1208                         goto bail;
1209                 }
1210         }
1211
1212         if (status == -ENOSPC) {
1213                 status = ocfs2_reserve_cluster_bitmap_bits(osb, *ac);
1214                 if (status < 0) {
1215                         if (status != -ENOSPC)
1216                                 mlog_errno(status);
1217                         goto bail;
1218                 }
1219         }
1220
1221         status = 0;
1222 bail:
1223         if ((status < 0) && *ac) {
1224                 ocfs2_free_alloc_context(*ac);
1225                 *ac = NULL;
1226         }
1227
1228         if (status)
1229                 mlog_errno(status);
1230         return status;
1231 }
1232
1233 int ocfs2_reserve_clusters(struct ocfs2_super *osb,
1234                            u32 bits_wanted,
1235                            struct ocfs2_alloc_context **ac)
1236 {
1237         return ocfs2_reserve_clusters_with_limit(osb, bits_wanted, 0,
1238                                                  ALLOC_NEW_GROUP, ac);
1239 }
1240
1241 /*
1242  * More or less lifted from ext3. I'll leave their description below:
1243  *
1244  * "For ext3 allocations, we must not reuse any blocks which are
1245  * allocated in the bitmap buffer's "last committed data" copy.  This
1246  * prevents deletes from freeing up the page for reuse until we have
1247  * committed the delete transaction.
1248  *
1249  * If we didn't do this, then deleting something and reallocating it as
1250  * data would allow the old block to be overwritten before the
1251  * transaction committed (because we force data to disk before commit).
1252  * This would lead to corruption if we crashed between overwriting the
1253  * data and committing the delete.
1254  *
1255  * @@@ We may want to make this allocation behaviour conditional on
1256  * data-writes at some point, and disable it for metadata allocations or
1257  * sync-data inodes."
1258  *
1259  * Note: OCFS2 already does this differently for metadata vs data
1260  * allocations, as those bitmaps are separate and undo access is never
1261  * called on a metadata group descriptor.
1262  */
1263 static int ocfs2_test_bg_bit_allocatable(struct buffer_head *bg_bh,
1264                                          int nr)
1265 {
1266         struct ocfs2_group_desc *bg = (struct ocfs2_group_desc *) bg_bh->b_data;
1267         int ret;
1268
1269         if (ocfs2_test_bit(nr, (unsigned long *)bg->bg_bitmap))
1270                 return 0;
1271
1272         if (!buffer_jbd(bg_bh))
1273                 return 1;
1274
1275         jbd_lock_bh_state(bg_bh);
1276         bg = (struct ocfs2_group_desc *) bh2jh(bg_bh)->b_committed_data;
1277         if (bg)
1278                 ret = !ocfs2_test_bit(nr, (unsigned long *)bg->bg_bitmap);
1279         else
1280                 ret = 1;
1281         jbd_unlock_bh_state(bg_bh);
1282
1283         return ret;
1284 }
1285
1286 static int ocfs2_block_group_find_clear_bits(struct ocfs2_super *osb,
1287                                              struct buffer_head *bg_bh,
1288                                              unsigned int bits_wanted,
1289                                              unsigned int total_bits,
1290                                              struct ocfs2_suballoc_result *res)
1291 {
1292         void *bitmap;
1293         u16 best_offset, best_size;
1294         int offset, start, found, status = 0;
1295         struct ocfs2_group_desc *bg = (struct ocfs2_group_desc *) bg_bh->b_data;
1296
1297         /* Callers got this descriptor from
1298          * ocfs2_read_group_descriptor().  Any corruption is a code bug. */
1299         BUG_ON(!OCFS2_IS_VALID_GROUP_DESC(bg));
1300
1301         found = start = best_offset = best_size = 0;
1302         bitmap = bg->bg_bitmap;
1303
1304         while((offset = ocfs2_find_next_zero_bit(bitmap, total_bits, start)) != -1) {
1305                 if (offset == total_bits)
1306                         break;
1307
1308                 if (!ocfs2_test_bg_bit_allocatable(bg_bh, offset)) {
1309                         /* We found a zero, but we can't use it as it
1310                          * hasn't been put to disk yet! */
1311                         found = 0;
1312                         start = offset + 1;
1313                 } else if (offset == start) {
1314                         /* we found a zero */
1315                         found++;
1316                         /* move start to the next bit to test */
1317                         start++;
1318                 } else {
1319                         /* got a zero after some ones */
1320                         found = 1;
1321                         start = offset + 1;
1322                 }
1323                 if (found > best_size) {
1324                         best_size = found;
1325                         best_offset = start - found;
1326                 }
1327                 /* we got everything we needed */
1328                 if (found == bits_wanted) {
1329                         /* mlog(0, "Found it all!\n"); */
1330                         break;
1331                 }
1332         }
1333
1334         if (best_size) {
1335                 res->sr_bit_offset = best_offset;
1336                 res->sr_bits = best_size;
1337         } else {
1338                 status = -ENOSPC;
1339                 /* No error log here -- see the comment above
1340                  * ocfs2_test_bg_bit_allocatable */
1341         }
1342
1343         return status;
1344 }
1345
1346 static inline int ocfs2_block_group_set_bits(handle_t *handle,
1347                                              struct inode *alloc_inode,
1348                                              struct ocfs2_group_desc *bg,
1349                                              struct buffer_head *group_bh,
1350                                              unsigned int bit_off,
1351                                              unsigned int num_bits)
1352 {
1353         int status;
1354         void *bitmap = bg->bg_bitmap;
1355         int journal_type = OCFS2_JOURNAL_ACCESS_WRITE;
1356
1357         /* All callers get the descriptor via
1358          * ocfs2_read_group_descriptor().  Any corruption is a code bug. */
1359         BUG_ON(!OCFS2_IS_VALID_GROUP_DESC(bg));
1360         BUG_ON(le16_to_cpu(bg->bg_free_bits_count) < num_bits);
1361
1362         trace_ocfs2_block_group_set_bits(bit_off, num_bits);
1363
1364         if (ocfs2_is_cluster_bitmap(alloc_inode))
1365                 journal_type = OCFS2_JOURNAL_ACCESS_UNDO;
1366
1367         status = ocfs2_journal_access_gd(handle,
1368                                          INODE_CACHE(alloc_inode),
1369                                          group_bh,
1370                                          journal_type);
1371         if (status < 0) {
1372                 mlog_errno(status);
1373                 goto bail;
1374         }
1375
1376         le16_add_cpu(&bg->bg_free_bits_count, -num_bits);
1377         if (le16_to_cpu(bg->bg_free_bits_count) > le16_to_cpu(bg->bg_bits)) {
1378                 ocfs2_error(alloc_inode->i_sb, "Group descriptor # %llu has bit"
1379                             " count %u but claims %u are freed. num_bits %d",
1380                             (unsigned long long)le64_to_cpu(bg->bg_blkno),
1381                             le16_to_cpu(bg->bg_bits),
1382                             le16_to_cpu(bg->bg_free_bits_count), num_bits);
1383                 return -EROFS;
1384         }
1385         while(num_bits--)
1386                 ocfs2_set_bit(bit_off++, bitmap);
1387
1388         ocfs2_journal_dirty(handle, group_bh);
1389
1390 bail:
1391         if (status)
1392                 mlog_errno(status);
1393         return status;
1394 }
1395
1396 /* find the one with the most empty bits */
1397 static inline u16 ocfs2_find_victim_chain(struct ocfs2_chain_list *cl)
1398 {
1399         u16 curr, best;
1400
1401         BUG_ON(!cl->cl_next_free_rec);
1402
1403         best = curr = 0;
1404         while (curr < le16_to_cpu(cl->cl_next_free_rec)) {
1405                 if (le32_to_cpu(cl->cl_recs[curr].c_free) >
1406                     le32_to_cpu(cl->cl_recs[best].c_free))
1407                         best = curr;
1408                 curr++;
1409         }
1410
1411         BUG_ON(best >= le16_to_cpu(cl->cl_next_free_rec));
1412         return best;
1413 }
1414
1415 static int ocfs2_relink_block_group(handle_t *handle,
1416                                     struct inode *alloc_inode,
1417                                     struct buffer_head *fe_bh,
1418                                     struct buffer_head *bg_bh,
1419                                     struct buffer_head *prev_bg_bh,
1420                                     u16 chain)
1421 {
1422         int status;
1423         /* there is a really tiny chance the journal calls could fail,
1424          * but we wouldn't want inconsistent blocks in *any* case. */
1425         u64 fe_ptr, bg_ptr, prev_bg_ptr;
1426         struct ocfs2_dinode *fe = (struct ocfs2_dinode *) fe_bh->b_data;
1427         struct ocfs2_group_desc *bg = (struct ocfs2_group_desc *) bg_bh->b_data;
1428         struct ocfs2_group_desc *prev_bg = (struct ocfs2_group_desc *) prev_bg_bh->b_data;
1429
1430         /* The caller got these descriptors from
1431          * ocfs2_read_group_descriptor().  Any corruption is a code bug. */
1432         BUG_ON(!OCFS2_IS_VALID_GROUP_DESC(bg));
1433         BUG_ON(!OCFS2_IS_VALID_GROUP_DESC(prev_bg));
1434
1435         trace_ocfs2_relink_block_group(
1436                 (unsigned long long)le64_to_cpu(fe->i_blkno), chain,
1437                 (unsigned long long)le64_to_cpu(bg->bg_blkno),
1438                 (unsigned long long)le64_to_cpu(prev_bg->bg_blkno));
1439
1440         fe_ptr = le64_to_cpu(fe->id2.i_chain.cl_recs[chain].c_blkno);
1441         bg_ptr = le64_to_cpu(bg->bg_next_group);
1442         prev_bg_ptr = le64_to_cpu(prev_bg->bg_next_group);
1443
1444         status = ocfs2_journal_access_gd(handle, INODE_CACHE(alloc_inode),
1445                                          prev_bg_bh,
1446                                          OCFS2_JOURNAL_ACCESS_WRITE);
1447         if (status < 0) {
1448                 mlog_errno(status);
1449                 goto out_rollback;
1450         }
1451
1452         prev_bg->bg_next_group = bg->bg_next_group;
1453         ocfs2_journal_dirty(handle, prev_bg_bh);
1454
1455         status = ocfs2_journal_access_gd(handle, INODE_CACHE(alloc_inode),
1456                                          bg_bh, OCFS2_JOURNAL_ACCESS_WRITE);
1457         if (status < 0) {
1458                 mlog_errno(status);
1459                 goto out_rollback;
1460         }
1461
1462         bg->bg_next_group = fe->id2.i_chain.cl_recs[chain].c_blkno;
1463         ocfs2_journal_dirty(handle, bg_bh);
1464
1465         status = ocfs2_journal_access_di(handle, INODE_CACHE(alloc_inode),
1466                                          fe_bh, OCFS2_JOURNAL_ACCESS_WRITE);
1467         if (status < 0) {
1468                 mlog_errno(status);
1469                 goto out_rollback;
1470         }
1471
1472         fe->id2.i_chain.cl_recs[chain].c_blkno = bg->bg_blkno;
1473         ocfs2_journal_dirty(handle, fe_bh);
1474
1475 out_rollback:
1476         if (status < 0) {
1477                 fe->id2.i_chain.cl_recs[chain].c_blkno = cpu_to_le64(fe_ptr);
1478                 bg->bg_next_group = cpu_to_le64(bg_ptr);
1479                 prev_bg->bg_next_group = cpu_to_le64(prev_bg_ptr);
1480         }
1481
1482         if (status)
1483                 mlog_errno(status);
1484         return status;
1485 }
1486
1487 static inline int ocfs2_block_group_reasonably_empty(struct ocfs2_group_desc *bg,
1488                                                      u32 wanted)
1489 {
1490         return le16_to_cpu(bg->bg_free_bits_count) > wanted;
1491 }
1492
1493 /* return 0 on success, -ENOSPC to keep searching and any other < 0
1494  * value on error. */
1495 static int ocfs2_cluster_group_search(struct inode *inode,
1496                                       struct buffer_head *group_bh,
1497                                       u32 bits_wanted, u32 min_bits,
1498                                       u64 max_block,
1499                                       struct ocfs2_suballoc_result *res)
1500 {
1501         int search = -ENOSPC;
1502         int ret;
1503         u64 blkoff;
1504         struct ocfs2_group_desc *gd = (struct ocfs2_group_desc *) group_bh->b_data;
1505         struct ocfs2_super *osb = OCFS2_SB(inode->i_sb);
1506         unsigned int max_bits, gd_cluster_off;
1507
1508         BUG_ON(!ocfs2_is_cluster_bitmap(inode));
1509
1510         if (gd->bg_free_bits_count) {
1511                 max_bits = le16_to_cpu(gd->bg_bits);
1512
1513                 /* Tail groups in cluster bitmaps which aren't cpg
1514                  * aligned are prone to partial extension by a failed
1515                  * fs resize. If the file system resize never got to
1516                  * update the dinode cluster count, then we don't want
1517                  * to trust any clusters past it, regardless of what
1518                  * the group descriptor says. */
1519                 gd_cluster_off = ocfs2_blocks_to_clusters(inode->i_sb,
1520                                                           le64_to_cpu(gd->bg_blkno));
1521                 if ((gd_cluster_off + max_bits) >
1522                     OCFS2_I(inode)->ip_clusters) {
1523                         max_bits = OCFS2_I(inode)->ip_clusters - gd_cluster_off;
1524                         trace_ocfs2_cluster_group_search_wrong_max_bits(
1525                                 (unsigned long long)le64_to_cpu(gd->bg_blkno),
1526                                 le16_to_cpu(gd->bg_bits),
1527                                 OCFS2_I(inode)->ip_clusters, max_bits);
1528                 }
1529
1530                 ret = ocfs2_block_group_find_clear_bits(OCFS2_SB(inode->i_sb),
1531                                                         group_bh, bits_wanted,
1532                                                         max_bits, res);
1533                 if (ret)
1534                         return ret;
1535
1536                 if (max_block) {
1537                         blkoff = ocfs2_clusters_to_blocks(inode->i_sb,
1538                                                           gd_cluster_off +
1539                                                           res->sr_bit_offset +
1540                                                           res->sr_bits);
1541                         trace_ocfs2_cluster_group_search_max_block(
1542                                 (unsigned long long)blkoff,
1543                                 (unsigned long long)max_block);
1544                         if (blkoff > max_block)
1545                                 return -ENOSPC;
1546                 }
1547
1548                 /* ocfs2_block_group_find_clear_bits() might
1549                  * return success, but we still want to return
1550                  * -ENOSPC unless it found the minimum number
1551                  * of bits. */
1552                 if (min_bits <= res->sr_bits)
1553                         search = 0; /* success */
1554                 else if (res->sr_bits) {
1555                         /*
1556                          * Don't show bits which we'll be returning
1557                          * for allocation to the local alloc bitmap.
1558                          */
1559                         ocfs2_local_alloc_seen_free_bits(osb, res->sr_bits);
1560                 }
1561         }
1562
1563         return search;
1564 }
1565
1566 static int ocfs2_block_group_search(struct inode *inode,
1567                                     struct buffer_head *group_bh,
1568                                     u32 bits_wanted, u32 min_bits,
1569                                     u64 max_block,
1570                                     struct ocfs2_suballoc_result *res)
1571 {
1572         int ret = -ENOSPC;
1573         u64 blkoff;
1574         struct ocfs2_group_desc *bg = (struct ocfs2_group_desc *) group_bh->b_data;
1575
1576         BUG_ON(min_bits != 1);
1577         BUG_ON(ocfs2_is_cluster_bitmap(inode));
1578
1579         if (bg->bg_free_bits_count) {
1580                 ret = ocfs2_block_group_find_clear_bits(OCFS2_SB(inode->i_sb),
1581                                                         group_bh, bits_wanted,
1582                                                         le16_to_cpu(bg->bg_bits),
1583                                                         res);
1584                 if (!ret && max_block) {
1585                         blkoff = le64_to_cpu(bg->bg_blkno) +
1586                                 res->sr_bit_offset + res->sr_bits;
1587                         trace_ocfs2_block_group_search_max_block(
1588                                 (unsigned long long)blkoff,
1589                                 (unsigned long long)max_block);
1590                         if (blkoff > max_block)
1591                                 ret = -ENOSPC;
1592                 }
1593         }
1594
1595         return ret;
1596 }
1597
1598 static int ocfs2_alloc_dinode_update_counts(struct inode *inode,
1599                                        handle_t *handle,
1600                                        struct buffer_head *di_bh,
1601                                        u32 num_bits,
1602                                        u16 chain)
1603 {
1604         int ret;
1605         u32 tmp_used;
1606         struct ocfs2_dinode *di = (struct ocfs2_dinode *) di_bh->b_data;
1607         struct ocfs2_chain_list *cl = (struct ocfs2_chain_list *) &di->id2.i_chain;
1608
1609         ret = ocfs2_journal_access_di(handle, INODE_CACHE(inode), di_bh,
1610                                       OCFS2_JOURNAL_ACCESS_WRITE);
1611         if (ret < 0) {
1612                 mlog_errno(ret);
1613                 goto out;
1614         }
1615
1616         tmp_used = le32_to_cpu(di->id1.bitmap1.i_used);
1617         di->id1.bitmap1.i_used = cpu_to_le32(num_bits + tmp_used);
1618         le32_add_cpu(&cl->cl_recs[chain].c_free, -num_bits);
1619         ocfs2_journal_dirty(handle, di_bh);
1620
1621 out:
1622         return ret;
1623 }
1624
1625 static int ocfs2_bg_discontig_fix_by_rec(struct ocfs2_suballoc_result *res,
1626                                          struct ocfs2_extent_rec *rec,
1627                                          struct ocfs2_chain_list *cl)
1628 {
1629         unsigned int bpc = le16_to_cpu(cl->cl_bpc);
1630         unsigned int bitoff = le32_to_cpu(rec->e_cpos) * bpc;
1631         unsigned int bitcount = le16_to_cpu(rec->e_leaf_clusters) * bpc;
1632
1633         if (res->sr_bit_offset < bitoff)
1634                 return 0;
1635         if (res->sr_bit_offset >= (bitoff + bitcount))
1636                 return 0;
1637         res->sr_blkno = le64_to_cpu(rec->e_blkno) +
1638                 (res->sr_bit_offset - bitoff);
1639         if ((res->sr_bit_offset + res->sr_bits) > (bitoff + bitcount))
1640                 res->sr_bits = (bitoff + bitcount) - res->sr_bit_offset;
1641         return 1;
1642 }
1643
1644 static void ocfs2_bg_discontig_fix_result(struct ocfs2_alloc_context *ac,
1645                                           struct ocfs2_group_desc *bg,
1646                                           struct ocfs2_suballoc_result *res)
1647 {
1648         int i;
1649         u64 bg_blkno = res->sr_bg_blkno;  /* Save off */
1650         struct ocfs2_extent_rec *rec;
1651         struct ocfs2_dinode *di = (struct ocfs2_dinode *)ac->ac_bh->b_data;
1652         struct ocfs2_chain_list *cl = &di->id2.i_chain;
1653
1654         if (ocfs2_is_cluster_bitmap(ac->ac_inode)) {
1655                 res->sr_blkno = 0;
1656                 return;
1657         }
1658
1659         res->sr_blkno = res->sr_bg_blkno + res->sr_bit_offset;
1660         res->sr_bg_blkno = 0;  /* Clear it for contig block groups */
1661         if (!ocfs2_supports_discontig_bg(OCFS2_SB(ac->ac_inode->i_sb)) ||
1662             !bg->bg_list.l_next_free_rec)
1663                 return;
1664
1665         for (i = 0; i < le16_to_cpu(bg->bg_list.l_next_free_rec); i++) {
1666                 rec = &bg->bg_list.l_recs[i];
1667                 if (ocfs2_bg_discontig_fix_by_rec(res, rec, cl)) {
1668                         res->sr_bg_blkno = bg_blkno;  /* Restore */
1669                         break;
1670                 }
1671         }
1672 }
1673
1674 static int ocfs2_search_one_group(struct ocfs2_alloc_context *ac,
1675                                   handle_t *handle,
1676                                   u32 bits_wanted,
1677                                   u32 min_bits,
1678                                   struct ocfs2_suballoc_result *res,
1679                                   u16 *bits_left)
1680 {
1681         int ret;
1682         struct buffer_head *group_bh = NULL;
1683         struct ocfs2_group_desc *gd;
1684         struct ocfs2_dinode *di = (struct ocfs2_dinode *)ac->ac_bh->b_data;
1685         struct inode *alloc_inode = ac->ac_inode;
1686
1687         ret = ocfs2_read_group_descriptor(alloc_inode, di,
1688                                           res->sr_bg_blkno, &group_bh);
1689         if (ret < 0) {
1690                 mlog_errno(ret);
1691                 return ret;
1692         }
1693
1694         gd = (struct ocfs2_group_desc *) group_bh->b_data;
1695         ret = ac->ac_group_search(alloc_inode, group_bh, bits_wanted, min_bits,
1696                                   ac->ac_max_block, res);
1697         if (ret < 0) {
1698                 if (ret != -ENOSPC)
1699                         mlog_errno(ret);
1700                 goto out;
1701         }
1702
1703         if (!ret)
1704                 ocfs2_bg_discontig_fix_result(ac, gd, res);
1705
1706         /*
1707          * sr_bg_blkno might have been changed by
1708          * ocfs2_bg_discontig_fix_result
1709          */
1710         res->sr_bg_stable_blkno = group_bh->b_blocknr;
1711
1712         if (ac->ac_find_loc_only)
1713                 goto out_loc_only;
1714
1715         ret = ocfs2_alloc_dinode_update_counts(alloc_inode, handle, ac->ac_bh,
1716                                                res->sr_bits,
1717                                                le16_to_cpu(gd->bg_chain));
1718         if (ret < 0) {
1719                 mlog_errno(ret);
1720                 goto out;
1721         }
1722
1723         ret = ocfs2_block_group_set_bits(handle, alloc_inode, gd, group_bh,
1724                                          res->sr_bit_offset, res->sr_bits);
1725         if (ret < 0)
1726                 mlog_errno(ret);
1727
1728 out_loc_only:
1729         *bits_left = le16_to_cpu(gd->bg_free_bits_count);
1730
1731 out:
1732         brelse(group_bh);
1733
1734         return ret;
1735 }
1736
1737 static int ocfs2_search_chain(struct ocfs2_alloc_context *ac,
1738                               handle_t *handle,
1739                               u32 bits_wanted,
1740                               u32 min_bits,
1741                               struct ocfs2_suballoc_result *res,
1742                               u16 *bits_left)
1743 {
1744         int status;
1745         u16 chain;
1746         u64 next_group;
1747         struct inode *alloc_inode = ac->ac_inode;
1748         struct buffer_head *group_bh = NULL;
1749         struct buffer_head *prev_group_bh = NULL;
1750         struct ocfs2_dinode *fe = (struct ocfs2_dinode *) ac->ac_bh->b_data;
1751         struct ocfs2_chain_list *cl = (struct ocfs2_chain_list *) &fe->id2.i_chain;
1752         struct ocfs2_group_desc *bg;
1753
1754         chain = ac->ac_chain;
1755         trace_ocfs2_search_chain_begin(
1756                 (unsigned long long)OCFS2_I(alloc_inode)->ip_blkno,
1757                 bits_wanted, chain);
1758
1759         status = ocfs2_read_group_descriptor(alloc_inode, fe,
1760                                              le64_to_cpu(cl->cl_recs[chain].c_blkno),
1761                                              &group_bh);
1762         if (status < 0) {
1763                 mlog_errno(status);
1764                 goto bail;
1765         }
1766         bg = (struct ocfs2_group_desc *) group_bh->b_data;
1767
1768         status = -ENOSPC;
1769         /* for now, the chain search is a bit simplistic. We just use
1770          * the 1st group with any empty bits. */
1771         while ((status = ac->ac_group_search(alloc_inode, group_bh,
1772                                              bits_wanted, min_bits,
1773                                              ac->ac_max_block,
1774                                              res)) == -ENOSPC) {
1775                 if (!bg->bg_next_group)
1776                         break;
1777
1778                 brelse(prev_group_bh);
1779                 prev_group_bh = NULL;
1780
1781                 next_group = le64_to_cpu(bg->bg_next_group);
1782                 prev_group_bh = group_bh;
1783                 group_bh = NULL;
1784                 status = ocfs2_read_group_descriptor(alloc_inode, fe,
1785                                                      next_group, &group_bh);
1786                 if (status < 0) {
1787                         mlog_errno(status);
1788                         goto bail;
1789                 }
1790                 bg = (struct ocfs2_group_desc *) group_bh->b_data;
1791         }
1792         if (status < 0) {
1793                 if (status != -ENOSPC)
1794                         mlog_errno(status);
1795                 goto bail;
1796         }
1797
1798         trace_ocfs2_search_chain_succ(
1799                 (unsigned long long)le64_to_cpu(bg->bg_blkno), res->sr_bits);
1800
1801         res->sr_bg_blkno = le64_to_cpu(bg->bg_blkno);
1802
1803         BUG_ON(res->sr_bits == 0);
1804         if (!status)
1805                 ocfs2_bg_discontig_fix_result(ac, bg, res);
1806
1807         /*
1808          * sr_bg_blkno might have been changed by
1809          * ocfs2_bg_discontig_fix_result
1810          */
1811         res->sr_bg_stable_blkno = group_bh->b_blocknr;
1812
1813         /*
1814          * Keep track of previous block descriptor read. When
1815          * we find a target, if we have read more than X
1816          * number of descriptors, and the target is reasonably
1817          * empty, relink him to top of his chain.
1818          *
1819          * We've read 0 extra blocks and only send one more to
1820          * the transaction, yet the next guy to search has a
1821          * much easier time.
1822          *
1823          * Do this *after* figuring out how many bits we're taking out
1824          * of our target group.
1825          */
1826         if (ac->ac_allow_chain_relink &&
1827             (prev_group_bh) &&
1828             (ocfs2_block_group_reasonably_empty(bg, res->sr_bits))) {
1829                 status = ocfs2_relink_block_group(handle, alloc_inode,
1830                                                   ac->ac_bh, group_bh,
1831                                                   prev_group_bh, chain);
1832                 if (status < 0) {
1833                         mlog_errno(status);
1834                         goto bail;
1835                 }
1836         }
1837
1838         if (ac->ac_find_loc_only)
1839                 goto out_loc_only;
1840
1841         status = ocfs2_alloc_dinode_update_counts(alloc_inode, handle,
1842                                                   ac->ac_bh, res->sr_bits,
1843                                                   chain);
1844         if (status) {
1845                 mlog_errno(status);
1846                 goto bail;
1847         }
1848
1849         status = ocfs2_block_group_set_bits(handle,
1850                                             alloc_inode,
1851                                             bg,
1852                                             group_bh,
1853                                             res->sr_bit_offset,
1854                                             res->sr_bits);
1855         if (status < 0) {
1856                 mlog_errno(status);
1857                 goto bail;
1858         }
1859
1860         trace_ocfs2_search_chain_end(
1861                         (unsigned long long)le64_to_cpu(fe->i_blkno),
1862                         res->sr_bits);
1863
1864 out_loc_only:
1865         *bits_left = le16_to_cpu(bg->bg_free_bits_count);
1866 bail:
1867         brelse(group_bh);
1868         brelse(prev_group_bh);
1869
1870         if (status)
1871                 mlog_errno(status);
1872         return status;
1873 }
1874
1875 /* will give out up to bits_wanted contiguous bits. */
1876 static int ocfs2_claim_suballoc_bits(struct ocfs2_alloc_context *ac,
1877                                      handle_t *handle,
1878                                      u32 bits_wanted,
1879                                      u32 min_bits,
1880                                      struct ocfs2_suballoc_result *res)
1881 {
1882         int status;
1883         u16 victim, i;
1884         u16 bits_left = 0;
1885         u64 hint = ac->ac_last_group;
1886         struct ocfs2_chain_list *cl;
1887         struct ocfs2_dinode *fe;
1888
1889         BUG_ON(ac->ac_bits_given >= ac->ac_bits_wanted);
1890         BUG_ON(bits_wanted > (ac->ac_bits_wanted - ac->ac_bits_given));
1891         BUG_ON(!ac->ac_bh);
1892
1893         fe = (struct ocfs2_dinode *) ac->ac_bh->b_data;
1894
1895         /* The bh was validated by the inode read during
1896          * ocfs2_reserve_suballoc_bits().  Any corruption is a code bug. */
1897         BUG_ON(!OCFS2_IS_VALID_DINODE(fe));
1898
1899         if (le32_to_cpu(fe->id1.bitmap1.i_used) >=
1900             le32_to_cpu(fe->id1.bitmap1.i_total)) {
1901                 ocfs2_error(ac->ac_inode->i_sb,
1902                             "Chain allocator dinode %llu has %u used "
1903                             "bits but only %u total.",
1904                             (unsigned long long)le64_to_cpu(fe->i_blkno),
1905                             le32_to_cpu(fe->id1.bitmap1.i_used),
1906                             le32_to_cpu(fe->id1.bitmap1.i_total));
1907                 status = -EIO;
1908                 goto bail;
1909         }
1910
1911         res->sr_bg_blkno = hint;
1912         if (res->sr_bg_blkno) {
1913                 /* Attempt to short-circuit the usual search mechanism
1914                  * by jumping straight to the most recently used
1915                  * allocation group. This helps us maintain some
1916                  * contiguousness across allocations. */
1917                 status = ocfs2_search_one_group(ac, handle, bits_wanted,
1918                                                 min_bits, res, &bits_left);
1919                 if (!status)
1920                         goto set_hint;
1921                 if (status < 0 && status != -ENOSPC) {
1922                         mlog_errno(status);
1923                         goto bail;
1924                 }
1925         }
1926
1927         cl = (struct ocfs2_chain_list *) &fe->id2.i_chain;
1928
1929         victim = ocfs2_find_victim_chain(cl);
1930         ac->ac_chain = victim;
1931         ac->ac_allow_chain_relink = 1;
1932
1933         status = ocfs2_search_chain(ac, handle, bits_wanted, min_bits,
1934                                     res, &bits_left);
1935         if (!status) {
1936                 hint = ocfs2_group_from_res(res);
1937                 goto set_hint;
1938         }
1939         if (status < 0 && status != -ENOSPC) {
1940                 mlog_errno(status);
1941                 goto bail;
1942         }
1943
1944         trace_ocfs2_claim_suballoc_bits(victim);
1945
1946         /* If we didn't pick a good victim, then just default to
1947          * searching each chain in order. Don't allow chain relinking
1948          * because we only calculate enough journal credits for one
1949          * relink per alloc. */
1950         ac->ac_allow_chain_relink = 0;
1951         for (i = 0; i < le16_to_cpu(cl->cl_next_free_rec); i ++) {
1952                 if (i == victim)
1953                         continue;
1954                 if (!cl->cl_recs[i].c_free)
1955                         continue;
1956
1957                 ac->ac_chain = i;
1958                 status = ocfs2_search_chain(ac, handle, bits_wanted, min_bits,
1959                                             res, &bits_left);
1960                 if (!status) {
1961                         hint = ocfs2_group_from_res(res);
1962                         break;
1963                 }
1964                 if (status < 0 && status != -ENOSPC) {
1965                         mlog_errno(status);
1966                         goto bail;
1967                 }
1968         }
1969
1970 set_hint:
1971         if (status != -ENOSPC) {
1972                 /* If the next search of this group is not likely to
1973                  * yield a suitable extent, then we reset the last
1974                  * group hint so as to not waste a disk read */
1975                 if (bits_left < min_bits)
1976                         ac->ac_last_group = 0;
1977                 else
1978                         ac->ac_last_group = hint;
1979         }
1980
1981 bail:
1982         if (status)
1983                 mlog_errno(status);
1984         return status;
1985 }
1986
1987 int ocfs2_claim_metadata(handle_t *handle,
1988                          struct ocfs2_alloc_context *ac,
1989                          u32 bits_wanted,
1990                          u64 *suballoc_loc,
1991                          u16 *suballoc_bit_start,
1992                          unsigned int *num_bits,
1993                          u64 *blkno_start)
1994 {
1995         int status;
1996         struct ocfs2_suballoc_result res = { .sr_blkno = 0, };
1997
1998         BUG_ON(!ac);
1999         BUG_ON(ac->ac_bits_wanted < (ac->ac_bits_given + bits_wanted));
2000         BUG_ON(ac->ac_which != OCFS2_AC_USE_META);
2001
2002         status = ocfs2_claim_suballoc_bits(ac,
2003                                            handle,
2004                                            bits_wanted,
2005                                            1,
2006                                            &res);
2007         if (status < 0) {
2008                 mlog_errno(status);
2009                 goto bail;
2010         }
2011         atomic_inc(&OCFS2_SB(ac->ac_inode->i_sb)->alloc_stats.bg_allocs);
2012
2013         *suballoc_loc = res.sr_bg_blkno;
2014         *suballoc_bit_start = res.sr_bit_offset;
2015         *blkno_start = res.sr_blkno;
2016         ac->ac_bits_given += res.sr_bits;
2017         *num_bits = res.sr_bits;
2018         status = 0;
2019 bail:
2020         if (status)
2021                 mlog_errno(status);
2022         return status;
2023 }
2024
2025 static void ocfs2_init_inode_ac_group(struct inode *dir,
2026                                       struct buffer_head *parent_di_bh,
2027                                       struct ocfs2_alloc_context *ac)
2028 {
2029         struct ocfs2_dinode *di = (struct ocfs2_dinode *)parent_di_bh->b_data;
2030         /*
2031          * Try to allocate inodes from some specific group.
2032          *
2033          * If the parent dir has recorded the last group used in allocation,
2034          * cool, use it. Otherwise if we try to allocate new inode from the
2035          * same slot the parent dir belongs to, use the same chunk.
2036          *
2037          * We are very careful here to avoid the mistake of setting
2038          * ac_last_group to a group descriptor from a different (unlocked) slot.
2039          */
2040         if (OCFS2_I(dir)->ip_last_used_group &&
2041             OCFS2_I(dir)->ip_last_used_slot == ac->ac_alloc_slot)
2042                 ac->ac_last_group = OCFS2_I(dir)->ip_last_used_group;
2043         else if (le16_to_cpu(di->i_suballoc_slot) == ac->ac_alloc_slot) {
2044                 if (di->i_suballoc_loc)
2045                         ac->ac_last_group = le64_to_cpu(di->i_suballoc_loc);
2046                 else
2047                         ac->ac_last_group = ocfs2_which_suballoc_group(
2048                                         le64_to_cpu(di->i_blkno),
2049                                         le16_to_cpu(di->i_suballoc_bit));
2050         }
2051 }
2052
2053 static inline void ocfs2_save_inode_ac_group(struct inode *dir,
2054                                              struct ocfs2_alloc_context *ac)
2055 {
2056         OCFS2_I(dir)->ip_last_used_group = ac->ac_last_group;
2057         OCFS2_I(dir)->ip_last_used_slot = ac->ac_alloc_slot;
2058 }
2059
2060 int ocfs2_find_new_inode_loc(struct inode *dir,
2061                              struct buffer_head *parent_fe_bh,
2062                              struct ocfs2_alloc_context *ac,
2063                              u64 *fe_blkno)
2064 {
2065         int ret;
2066         handle_t *handle = NULL;
2067         struct ocfs2_suballoc_result *res;
2068
2069         BUG_ON(!ac);
2070         BUG_ON(ac->ac_bits_given != 0);
2071         BUG_ON(ac->ac_bits_wanted != 1);
2072         BUG_ON(ac->ac_which != OCFS2_AC_USE_INODE);
2073
2074         res = kzalloc(sizeof(*res), GFP_NOFS);
2075         if (res == NULL) {
2076                 ret = -ENOMEM;
2077                 mlog_errno(ret);
2078                 goto out;
2079         }
2080
2081         ocfs2_init_inode_ac_group(dir, parent_fe_bh, ac);
2082
2083         /*
2084          * The handle started here is for chain relink. Alternatively,
2085          * we could just disable relink for these calls.
2086          */
2087         handle = ocfs2_start_trans(OCFS2_SB(dir->i_sb), OCFS2_SUBALLOC_ALLOC);
2088         if (IS_ERR(handle)) {
2089                 ret = PTR_ERR(handle);
2090                 handle = NULL;
2091                 mlog_errno(ret);
2092                 goto out;
2093         }
2094
2095         /*
2096          * This will instruct ocfs2_claim_suballoc_bits and
2097          * ocfs2_search_one_group to search but save actual allocation
2098          * for later.
2099          */
2100         ac->ac_find_loc_only = 1;
2101
2102         ret = ocfs2_claim_suballoc_bits(ac, handle, 1, 1, res);
2103         if (ret < 0) {
2104                 mlog_errno(ret);
2105                 goto out;
2106         }
2107
2108         ac->ac_find_loc_priv = res;
2109         *fe_blkno = res->sr_blkno;
2110
2111 out:
2112         if (handle)
2113                 ocfs2_commit_trans(OCFS2_SB(dir->i_sb), handle);
2114
2115         if (ret)
2116                 kfree(res);
2117
2118         return ret;
2119 }
2120
2121 int ocfs2_claim_new_inode_at_loc(handle_t *handle,
2122                                  struct inode *dir,
2123                                  struct ocfs2_alloc_context *ac,
2124                                  u64 *suballoc_loc,
2125                                  u16 *suballoc_bit,
2126                                  u64 di_blkno)
2127 {
2128         int ret;
2129         u16 chain;
2130         struct ocfs2_suballoc_result *res = ac->ac_find_loc_priv;
2131         struct buffer_head *bg_bh = NULL;
2132         struct ocfs2_group_desc *bg;
2133         struct ocfs2_dinode *di = (struct ocfs2_dinode *) ac->ac_bh->b_data;
2134
2135         /*
2136          * Since di_blkno is being passed back in, we check for any
2137          * inconsistencies which may have happened between
2138          * calls. These are code bugs as di_blkno is not expected to
2139          * change once returned from ocfs2_find_new_inode_loc()
2140          */
2141         BUG_ON(res->sr_blkno != di_blkno);
2142
2143         ret = ocfs2_read_group_descriptor(ac->ac_inode, di,
2144                                           res->sr_bg_stable_blkno, &bg_bh);
2145         if (ret) {
2146                 mlog_errno(ret);
2147                 goto out;
2148         }
2149
2150         bg = (struct ocfs2_group_desc *) bg_bh->b_data;
2151         chain = le16_to_cpu(bg->bg_chain);
2152
2153         ret = ocfs2_alloc_dinode_update_counts(ac->ac_inode, handle,
2154                                                ac->ac_bh, res->sr_bits,
2155                                                chain);
2156         if (ret) {
2157                 mlog_errno(ret);
2158                 goto out;
2159         }
2160
2161         ret = ocfs2_block_group_set_bits(handle,
2162                                          ac->ac_inode,
2163                                          bg,
2164                                          bg_bh,
2165                                          res->sr_bit_offset,
2166                                          res->sr_bits);
2167         if (ret < 0) {
2168                 mlog_errno(ret);
2169                 goto out;
2170         }
2171
2172         trace_ocfs2_claim_new_inode_at_loc((unsigned long long)di_blkno,
2173                                            res->sr_bits);
2174
2175         atomic_inc(&OCFS2_SB(ac->ac_inode->i_sb)->alloc_stats.bg_allocs);
2176
2177         BUG_ON(res->sr_bits != 1);
2178
2179         *suballoc_loc = res->sr_bg_blkno;
2180         *suballoc_bit = res->sr_bit_offset;
2181         ac->ac_bits_given++;
2182         ocfs2_save_inode_ac_group(dir, ac);
2183
2184 out:
2185         brelse(bg_bh);
2186
2187         return ret;
2188 }
2189
2190 int ocfs2_claim_new_inode(handle_t *handle,
2191                           struct inode *dir,
2192                           struct buffer_head *parent_fe_bh,
2193                           struct ocfs2_alloc_context *ac,
2194                           u64 *suballoc_loc,
2195                           u16 *suballoc_bit,
2196                           u64 *fe_blkno)
2197 {
2198         int status;
2199         struct ocfs2_suballoc_result res;
2200
2201         BUG_ON(!ac);
2202         BUG_ON(ac->ac_bits_given != 0);
2203         BUG_ON(ac->ac_bits_wanted != 1);
2204         BUG_ON(ac->ac_which != OCFS2_AC_USE_INODE);
2205
2206         ocfs2_init_inode_ac_group(dir, parent_fe_bh, ac);
2207
2208         status = ocfs2_claim_suballoc_bits(ac,
2209                                            handle,
2210                                            1,
2211                                            1,
2212                                            &res);
2213         if (status < 0) {
2214                 mlog_errno(status);
2215                 goto bail;
2216         }
2217         atomic_inc(&OCFS2_SB(ac->ac_inode->i_sb)->alloc_stats.bg_allocs);
2218
2219         BUG_ON(res.sr_bits != 1);
2220
2221         *suballoc_loc = res.sr_bg_blkno;
2222         *suballoc_bit = res.sr_bit_offset;
2223         *fe_blkno = res.sr_blkno;
2224         ac->ac_bits_given++;
2225         ocfs2_save_inode_ac_group(dir, ac);
2226         status = 0;
2227 bail:
2228         if (status)
2229                 mlog_errno(status);
2230         return status;
2231 }
2232
2233 /* translate a group desc. blkno and it's bitmap offset into
2234  * disk cluster offset. */
2235 static inline u32 ocfs2_desc_bitmap_to_cluster_off(struct inode *inode,
2236                                                    u64 bg_blkno,
2237                                                    u16 bg_bit_off)
2238 {
2239         struct ocfs2_super *osb = OCFS2_SB(inode->i_sb);
2240         u32 cluster = 0;
2241
2242         BUG_ON(!ocfs2_is_cluster_bitmap(inode));
2243
2244         if (bg_blkno != osb->first_cluster_group_blkno)
2245                 cluster = ocfs2_blocks_to_clusters(inode->i_sb, bg_blkno);
2246         cluster += (u32) bg_bit_off;
2247         return cluster;
2248 }
2249
2250 /* given a cluster offset, calculate which block group it belongs to
2251  * and return that block offset. */
2252 u64 ocfs2_which_cluster_group(struct inode *inode, u32 cluster)
2253 {
2254         struct ocfs2_super *osb = OCFS2_SB(inode->i_sb);
2255         u32 group_no;
2256
2257         BUG_ON(!ocfs2_is_cluster_bitmap(inode));
2258
2259         group_no = cluster / osb->bitmap_cpg;
2260         if (!group_no)
2261                 return osb->first_cluster_group_blkno;
2262         return ocfs2_clusters_to_blocks(inode->i_sb,
2263                                         group_no * osb->bitmap_cpg);
2264 }
2265
2266 /* given the block number of a cluster start, calculate which cluster
2267  * group and descriptor bitmap offset that corresponds to. */
2268 static inline void ocfs2_block_to_cluster_group(struct inode *inode,
2269                                                 u64 data_blkno,
2270                                                 u64 *bg_blkno,
2271                                                 u16 *bg_bit_off)
2272 {
2273         struct ocfs2_super *osb = OCFS2_SB(inode->i_sb);
2274         u32 data_cluster = ocfs2_blocks_to_clusters(osb->sb, data_blkno);
2275
2276         BUG_ON(!ocfs2_is_cluster_bitmap(inode));
2277
2278         *bg_blkno = ocfs2_which_cluster_group(inode,
2279                                               data_cluster);
2280
2281         if (*bg_blkno == osb->first_cluster_group_blkno)
2282                 *bg_bit_off = (u16) data_cluster;
2283         else
2284                 *bg_bit_off = (u16) ocfs2_blocks_to_clusters(osb->sb,
2285                                                              data_blkno - *bg_blkno);
2286 }
2287
2288 /*
2289  * min_bits - minimum contiguous chunk from this total allocation we
2290  * can handle. set to what we asked for originally for a full
2291  * contig. allocation, set to '1' to indicate we can deal with extents
2292  * of any size.
2293  */
2294 int __ocfs2_claim_clusters(handle_t *handle,
2295                            struct ocfs2_alloc_context *ac,
2296                            u32 min_clusters,
2297                            u32 max_clusters,
2298                            u32 *cluster_start,
2299                            u32 *num_clusters)
2300 {
2301         int status;
2302         unsigned int bits_wanted = max_clusters;
2303         struct ocfs2_suballoc_result res = { .sr_blkno = 0, };
2304         struct ocfs2_super *osb = OCFS2_SB(ac->ac_inode->i_sb);
2305
2306         BUG_ON(ac->ac_bits_given >= ac->ac_bits_wanted);
2307
2308         BUG_ON(ac->ac_which != OCFS2_AC_USE_LOCAL
2309                && ac->ac_which != OCFS2_AC_USE_MAIN);
2310
2311         if (ac->ac_which == OCFS2_AC_USE_LOCAL) {
2312                 WARN_ON(min_clusters > 1);
2313
2314                 status = ocfs2_claim_local_alloc_bits(osb,
2315                                                       handle,
2316                                                       ac,
2317                                                       bits_wanted,
2318                                                       cluster_start,
2319                                                       num_clusters);
2320                 if (!status)
2321                         atomic_inc(&osb->alloc_stats.local_data);
2322         } else {
2323                 if (min_clusters > (osb->bitmap_cpg - 1)) {
2324                         /* The only paths asking for contiguousness
2325                          * should know about this already. */
2326                         mlog(ML_ERROR, "minimum allocation requested %u exceeds "
2327                              "group bitmap size %u!\n", min_clusters,
2328                              osb->bitmap_cpg);
2329                         status = -ENOSPC;
2330                         goto bail;
2331                 }
2332                 /* clamp the current request down to a realistic size. */
2333                 if (bits_wanted > (osb->bitmap_cpg - 1))
2334                         bits_wanted = osb->bitmap_cpg - 1;
2335
2336                 status = ocfs2_claim_suballoc_bits(ac,
2337                                                    handle,
2338                                                    bits_wanted,
2339                                                    min_clusters,
2340                                                    &res);
2341                 if (!status) {
2342                         BUG_ON(res.sr_blkno); /* cluster alloc can't set */
2343                         *cluster_start =
2344                                 ocfs2_desc_bitmap_to_cluster_off(ac->ac_inode,
2345                                                                  res.sr_bg_blkno,
2346                                                                  res.sr_bit_offset);
2347                         atomic_inc(&osb->alloc_stats.bitmap_data);
2348                         *num_clusters = res.sr_bits;
2349                 }
2350         }
2351         if (status < 0) {
2352                 if (status != -ENOSPC)
2353                         mlog_errno(status);
2354                 goto bail;
2355         }
2356
2357         ac->ac_bits_given += *num_clusters;
2358
2359 bail:
2360         if (status)
2361                 mlog_errno(status);
2362         return status;
2363 }
2364
2365 int ocfs2_claim_clusters(handle_t *handle,
2366                          struct ocfs2_alloc_context *ac,
2367                          u32 min_clusters,
2368                          u32 *cluster_start,
2369                          u32 *num_clusters)
2370 {
2371         unsigned int bits_wanted = ac->ac_bits_wanted - ac->ac_bits_given;
2372
2373         return __ocfs2_claim_clusters(handle, ac, min_clusters,
2374                                       bits_wanted, cluster_start, num_clusters);
2375 }
2376
2377 static int ocfs2_block_group_clear_bits(handle_t *handle,
2378                                         struct inode *alloc_inode,
2379                                         struct ocfs2_group_desc *bg,
2380                                         struct buffer_head *group_bh,
2381                                         unsigned int bit_off,
2382                                         unsigned int num_bits,
2383                                         void (*undo_fn)(unsigned int bit,
2384                                                         unsigned long *bmap))
2385 {
2386         int status;
2387         unsigned int tmp;
2388         struct ocfs2_group_desc *undo_bg = NULL;
2389
2390         /* The caller got this descriptor from
2391          * ocfs2_read_group_descriptor().  Any corruption is a code bug. */
2392         BUG_ON(!OCFS2_IS_VALID_GROUP_DESC(bg));
2393
2394         trace_ocfs2_block_group_clear_bits(bit_off, num_bits);
2395
2396         BUG_ON(undo_fn && !ocfs2_is_cluster_bitmap(alloc_inode));
2397         status = ocfs2_journal_access_gd(handle, INODE_CACHE(alloc_inode),
2398                                          group_bh,
2399                                          undo_fn ?
2400                                          OCFS2_JOURNAL_ACCESS_UNDO :
2401                                          OCFS2_JOURNAL_ACCESS_WRITE);
2402         if (status < 0) {
2403                 mlog_errno(status);
2404                 goto bail;
2405         }
2406
2407         if (undo_fn) {
2408                 jbd_lock_bh_state(group_bh);
2409                 undo_bg = (struct ocfs2_group_desc *)
2410                                         bh2jh(group_bh)->b_committed_data;
2411                 BUG_ON(!undo_bg);
2412         }
2413
2414         tmp = num_bits;
2415         while(tmp--) {
2416                 ocfs2_clear_bit((bit_off + tmp),
2417                                 (unsigned long *) bg->bg_bitmap);
2418                 if (undo_fn)
2419                         undo_fn(bit_off + tmp,
2420                                 (unsigned long *) undo_bg->bg_bitmap);
2421         }
2422         le16_add_cpu(&bg->bg_free_bits_count, num_bits);
2423         if (le16_to_cpu(bg->bg_free_bits_count) > le16_to_cpu(bg->bg_bits)) {
2424                 ocfs2_error(alloc_inode->i_sb, "Group descriptor # %llu has bit"
2425                             " count %u but claims %u are freed. num_bits %d",
2426                             (unsigned long long)le64_to_cpu(bg->bg_blkno),
2427                             le16_to_cpu(bg->bg_bits),
2428                             le16_to_cpu(bg->bg_free_bits_count), num_bits);
2429                 return -EROFS;
2430         }
2431
2432         if (undo_fn)
2433                 jbd_unlock_bh_state(group_bh);
2434
2435         ocfs2_journal_dirty(handle, group_bh);
2436 bail:
2437         return status;
2438 }
2439
2440 /*
2441  * expects the suballoc inode to already be locked.
2442  */
2443 static int _ocfs2_free_suballoc_bits(handle_t *handle,
2444                                      struct inode *alloc_inode,
2445                                      struct buffer_head *alloc_bh,
2446                                      unsigned int start_bit,
2447                                      u64 bg_blkno,
2448                                      unsigned int count,
2449                                      void (*undo_fn)(unsigned int bit,
2450                                                      unsigned long *bitmap))
2451 {
2452         int status = 0;
2453         u32 tmp_used;
2454         struct ocfs2_dinode *fe = (struct ocfs2_dinode *) alloc_bh->b_data;
2455         struct ocfs2_chain_list *cl = &fe->id2.i_chain;
2456         struct buffer_head *group_bh = NULL;
2457         struct ocfs2_group_desc *group;
2458
2459         /* The alloc_bh comes from ocfs2_free_dinode() or
2460          * ocfs2_free_clusters().  The callers have all locked the
2461          * allocator and gotten alloc_bh from the lock call.  This
2462          * validates the dinode buffer.  Any corruption that has happened
2463          * is a code bug. */
2464         BUG_ON(!OCFS2_IS_VALID_DINODE(fe));
2465         BUG_ON((count + start_bit) > ocfs2_bits_per_group(cl));
2466
2467         trace_ocfs2_free_suballoc_bits(
2468                 (unsigned long long)OCFS2_I(alloc_inode)->ip_blkno,
2469                 (unsigned long long)bg_blkno,
2470                 start_bit, count);
2471
2472         status = ocfs2_read_group_descriptor(alloc_inode, fe, bg_blkno,
2473                                              &group_bh);
2474         if (status < 0) {
2475                 mlog_errno(status);
2476                 goto bail;
2477         }
2478         group = (struct ocfs2_group_desc *) group_bh->b_data;
2479
2480         BUG_ON((count + start_bit) > le16_to_cpu(group->bg_bits));
2481
2482         status = ocfs2_block_group_clear_bits(handle, alloc_inode,
2483                                               group, group_bh,
2484                                               start_bit, count, undo_fn);
2485         if (status < 0) {
2486                 mlog_errno(status);
2487                 goto bail;
2488         }
2489
2490         status = ocfs2_journal_access_di(handle, INODE_CACHE(alloc_inode),
2491                                          alloc_bh, OCFS2_JOURNAL_ACCESS_WRITE);
2492         if (status < 0) {
2493                 mlog_errno(status);
2494                 goto bail;
2495         }
2496
2497         le32_add_cpu(&cl->cl_recs[le16_to_cpu(group->bg_chain)].c_free,
2498                      count);
2499         tmp_used = le32_to_cpu(fe->id1.bitmap1.i_used);
2500         fe->id1.bitmap1.i_used = cpu_to_le32(tmp_used - count);
2501         ocfs2_journal_dirty(handle, alloc_bh);
2502
2503 bail:
2504         brelse(group_bh);
2505
2506         if (status)
2507                 mlog_errno(status);
2508         return status;
2509 }
2510
2511 int ocfs2_free_suballoc_bits(handle_t *handle,
2512                              struct inode *alloc_inode,
2513                              struct buffer_head *alloc_bh,
2514                              unsigned int start_bit,
2515                              u64 bg_blkno,
2516                              unsigned int count)
2517 {
2518         return _ocfs2_free_suballoc_bits(handle, alloc_inode, alloc_bh,
2519                                          start_bit, bg_blkno, count, NULL);
2520 }
2521
2522 int ocfs2_free_dinode(handle_t *handle,
2523                       struct inode *inode_alloc_inode,
2524                       struct buffer_head *inode_alloc_bh,
2525                       struct ocfs2_dinode *di)
2526 {
2527         u64 blk = le64_to_cpu(di->i_blkno);
2528         u16 bit = le16_to_cpu(di->i_suballoc_bit);
2529         u64 bg_blkno = ocfs2_which_suballoc_group(blk, bit);
2530
2531         if (di->i_suballoc_loc)
2532                 bg_blkno = le64_to_cpu(di->i_suballoc_loc);
2533         return ocfs2_free_suballoc_bits(handle, inode_alloc_inode,
2534                                         inode_alloc_bh, bit, bg_blkno, 1);
2535 }
2536
2537 static int _ocfs2_free_clusters(handle_t *handle,
2538                                 struct inode *bitmap_inode,
2539                                 struct buffer_head *bitmap_bh,
2540                                 u64 start_blk,
2541                                 unsigned int num_clusters,
2542                                 void (*undo_fn)(unsigned int bit,
2543                                                 unsigned long *bitmap))
2544 {
2545         int status;
2546         u16 bg_start_bit;
2547         u64 bg_blkno;
2548         struct ocfs2_dinode *fe;
2549
2550         /* You can't ever have a contiguous set of clusters
2551          * bigger than a block group bitmap so we never have to worry
2552          * about looping on them.
2553          * This is expensive. We can safely remove once this stuff has
2554          * gotten tested really well. */
2555         BUG_ON(start_blk != ocfs2_clusters_to_blocks(bitmap_inode->i_sb, ocfs2_blocks_to_clusters(bitmap_inode->i_sb, start_blk)));
2556
2557         fe = (struct ocfs2_dinode *) bitmap_bh->b_data;
2558
2559         ocfs2_block_to_cluster_group(bitmap_inode, start_blk, &bg_blkno,
2560                                      &bg_start_bit);
2561
2562         trace_ocfs2_free_clusters((unsigned long long)bg_blkno,
2563                         (unsigned long long)start_blk,
2564                         bg_start_bit, num_clusters);
2565
2566         status = _ocfs2_free_suballoc_bits(handle, bitmap_inode, bitmap_bh,
2567                                            bg_start_bit, bg_blkno,
2568                                            num_clusters, undo_fn);
2569         if (status < 0) {
2570                 mlog_errno(status);
2571                 goto out;
2572         }
2573
2574         ocfs2_local_alloc_seen_free_bits(OCFS2_SB(bitmap_inode->i_sb),
2575                                          num_clusters);
2576
2577 out:
2578         if (status)
2579                 mlog_errno(status);
2580         return status;
2581 }
2582
2583 int ocfs2_free_clusters(handle_t *handle,
2584                         struct inode *bitmap_inode,
2585                         struct buffer_head *bitmap_bh,
2586                         u64 start_blk,
2587                         unsigned int num_clusters)
2588 {
2589         return _ocfs2_free_clusters(handle, bitmap_inode, bitmap_bh,
2590                                     start_blk, num_clusters,
2591                                     _ocfs2_set_bit);
2592 }
2593
2594 /*
2595  * Give never-used clusters back to the global bitmap.  We don't need
2596  * to protect these bits in the undo buffer.
2597  */
2598 int ocfs2_release_clusters(handle_t *handle,
2599                            struct inode *bitmap_inode,
2600                            struct buffer_head *bitmap_bh,
2601                            u64 start_blk,
2602                            unsigned int num_clusters)
2603 {
2604         return _ocfs2_free_clusters(handle, bitmap_inode, bitmap_bh,
2605                                     start_blk, num_clusters,
2606                                     _ocfs2_clear_bit);
2607 }
2608
2609 static inline void ocfs2_debug_bg(struct ocfs2_group_desc *bg)
2610 {
2611         printk("Block Group:\n");
2612         printk("bg_signature:       %s\n", bg->bg_signature);
2613         printk("bg_size:            %u\n", bg->bg_size);
2614         printk("bg_bits:            %u\n", bg->bg_bits);
2615         printk("bg_free_bits_count: %u\n", bg->bg_free_bits_count);
2616         printk("bg_chain:           %u\n", bg->bg_chain);
2617         printk("bg_generation:      %u\n", le32_to_cpu(bg->bg_generation));
2618         printk("bg_next_group:      %llu\n",
2619                (unsigned long long)bg->bg_next_group);
2620         printk("bg_parent_dinode:   %llu\n",
2621                (unsigned long long)bg->bg_parent_dinode);
2622         printk("bg_blkno:           %llu\n",
2623                (unsigned long long)bg->bg_blkno);
2624 }
2625
2626 static inline void ocfs2_debug_suballoc_inode(struct ocfs2_dinode *fe)
2627 {
2628         int i;
2629
2630         printk("Suballoc Inode %llu:\n", (unsigned long long)fe->i_blkno);
2631         printk("i_signature:                  %s\n", fe->i_signature);
2632         printk("i_size:                       %llu\n",
2633                (unsigned long long)fe->i_size);
2634         printk("i_clusters:                   %u\n", fe->i_clusters);
2635         printk("i_generation:                 %u\n",
2636                le32_to_cpu(fe->i_generation));
2637         printk("id1.bitmap1.i_used:           %u\n",
2638                le32_to_cpu(fe->id1.bitmap1.i_used));
2639         printk("id1.bitmap1.i_total:          %u\n",
2640                le32_to_cpu(fe->id1.bitmap1.i_total));
2641         printk("id2.i_chain.cl_cpg:           %u\n", fe->id2.i_chain.cl_cpg);
2642         printk("id2.i_chain.cl_bpc:           %u\n", fe->id2.i_chain.cl_bpc);
2643         printk("id2.i_chain.cl_count:         %u\n", fe->id2.i_chain.cl_count);
2644         printk("id2.i_chain.cl_next_free_rec: %u\n",
2645                fe->id2.i_chain.cl_next_free_rec);
2646         for(i = 0; i < fe->id2.i_chain.cl_next_free_rec; i++) {
2647                 printk("fe->id2.i_chain.cl_recs[%d].c_free:  %u\n", i,
2648                        fe->id2.i_chain.cl_recs[i].c_free);
2649                 printk("fe->id2.i_chain.cl_recs[%d].c_total: %u\n", i,
2650                        fe->id2.i_chain.cl_recs[i].c_total);
2651                 printk("fe->id2.i_chain.cl_recs[%d].c_blkno: %llu\n", i,
2652                        (unsigned long long)fe->id2.i_chain.cl_recs[i].c_blkno);
2653         }
2654 }
2655
2656 /*
2657  * For a given allocation, determine which allocators will need to be
2658  * accessed, and lock them, reserving the appropriate number of bits.
2659  *
2660  * Sparse file systems call this from ocfs2_write_begin_nolock()
2661  * and ocfs2_allocate_unwritten_extents().
2662  *
2663  * File systems which don't support holes call this from
2664  * ocfs2_extend_allocation().
2665  */
2666 int ocfs2_lock_allocators(struct inode *inode,
2667                           struct ocfs2_extent_tree *et,
2668                           u32 clusters_to_add, u32 extents_to_split,
2669                           struct ocfs2_alloc_context **data_ac,
2670                           struct ocfs2_alloc_context **meta_ac)
2671 {
2672         int ret = 0, num_free_extents;
2673         unsigned int max_recs_needed = clusters_to_add + 2 * extents_to_split;
2674         struct ocfs2_super *osb = OCFS2_SB(inode->i_sb);
2675
2676         *meta_ac = NULL;
2677         if (data_ac)
2678                 *data_ac = NULL;
2679
2680         BUG_ON(clusters_to_add != 0 && data_ac == NULL);
2681
2682         num_free_extents = ocfs2_num_free_extents(osb, et);
2683         if (num_free_extents < 0) {
2684                 ret = num_free_extents;
2685                 mlog_errno(ret);
2686                 goto out;
2687         }
2688
2689         /*
2690          * Sparse allocation file systems need to be more conservative
2691          * with reserving room for expansion - the actual allocation
2692          * happens while we've got a journal handle open so re-taking
2693          * a cluster lock (because we ran out of room for another
2694          * extent) will violate ordering rules.
2695          *
2696          * Most of the time we'll only be seeing this 1 cluster at a time
2697          * anyway.
2698          *
2699          * Always lock for any unwritten extents - we might want to
2700          * add blocks during a split.
2701          */
2702         if (!num_free_extents ||
2703             (ocfs2_sparse_alloc(osb) && num_free_extents < max_recs_needed)) {
2704                 ret = ocfs2_reserve_new_metadata(osb, et->et_root_el, meta_ac);
2705                 if (ret < 0) {
2706                         if (ret != -ENOSPC)
2707                                 mlog_errno(ret);
2708                         goto out;
2709                 }
2710         }
2711
2712         if (clusters_to_add == 0)
2713                 goto out;
2714
2715         ret = ocfs2_reserve_clusters(osb, clusters_to_add, data_ac);
2716         if (ret < 0) {
2717                 if (ret != -ENOSPC)
2718                         mlog_errno(ret);
2719                 goto out;
2720         }
2721
2722 out:
2723         if (ret) {
2724                 if (*meta_ac) {
2725                         ocfs2_free_alloc_context(*meta_ac);
2726                         *meta_ac = NULL;
2727                 }
2728
2729                 /*
2730                  * We cannot have an error and a non null *data_ac.
2731                  */
2732         }
2733
2734         return ret;
2735 }
2736
2737 /*
2738  * Read the inode specified by blkno to get suballoc_slot and
2739  * suballoc_bit.
2740  */
2741 static int ocfs2_get_suballoc_slot_bit(struct ocfs2_super *osb, u64 blkno,
2742                                        u16 *suballoc_slot, u64 *group_blkno,
2743                                        u16 *suballoc_bit)
2744 {
2745         int status;
2746         struct buffer_head *inode_bh = NULL;
2747         struct ocfs2_dinode *inode_fe;
2748
2749         trace_ocfs2_get_suballoc_slot_bit((unsigned long long)blkno);
2750
2751         /* dirty read disk */
2752         status = ocfs2_read_blocks_sync(osb, blkno, 1, &inode_bh);
2753         if (status < 0) {
2754                 mlog(ML_ERROR, "read block %llu failed %d\n",
2755                      (unsigned long long)blkno, status);
2756                 goto bail;
2757         }
2758
2759         inode_fe = (struct ocfs2_dinode *) inode_bh->b_data;
2760         if (!OCFS2_IS_VALID_DINODE(inode_fe)) {
2761                 mlog(ML_ERROR, "invalid inode %llu requested\n",
2762                      (unsigned long long)blkno);
2763                 status = -EINVAL;
2764                 goto bail;
2765         }
2766
2767         if (le16_to_cpu(inode_fe->i_suballoc_slot) != (u16)OCFS2_INVALID_SLOT &&
2768             (u32)le16_to_cpu(inode_fe->i_suballoc_slot) > osb->max_slots - 1) {
2769                 mlog(ML_ERROR, "inode %llu has invalid suballoc slot %u\n",
2770                      (unsigned long long)blkno,
2771                      (u32)le16_to_cpu(inode_fe->i_suballoc_slot));
2772                 status = -EINVAL;
2773                 goto bail;
2774         }
2775
2776         if (suballoc_slot)
2777                 *suballoc_slot = le16_to_cpu(inode_fe->i_suballoc_slot);
2778         if (suballoc_bit)
2779                 *suballoc_bit = le16_to_cpu(inode_fe->i_suballoc_bit);
2780         if (group_blkno)
2781                 *group_blkno = le64_to_cpu(inode_fe->i_suballoc_loc);
2782
2783 bail:
2784         brelse(inode_bh);
2785
2786         if (status)
2787                 mlog_errno(status);
2788         return status;
2789 }
2790
2791 /*
2792  * test whether bit is SET in allocator bitmap or not.  on success, 0
2793  * is returned and *res is 1 for SET; 0 otherwise.  when fails, errno
2794  * is returned and *res is meaningless.  Call this after you have
2795  * cluster locked against suballoc, or you may get a result based on
2796  * non-up2date contents
2797  */
2798 static int ocfs2_test_suballoc_bit(struct ocfs2_super *osb,
2799                                    struct inode *suballoc,
2800                                    struct buffer_head *alloc_bh,
2801                                    u64 group_blkno, u64 blkno,
2802                                    u16 bit, int *res)
2803 {
2804         struct ocfs2_dinode *alloc_di;
2805         struct ocfs2_group_desc *group;
2806         struct buffer_head *group_bh = NULL;
2807         u64 bg_blkno;
2808         int status;
2809
2810         trace_ocfs2_test_suballoc_bit((unsigned long long)blkno,
2811                                       (unsigned int)bit);
2812
2813         alloc_di = (struct ocfs2_dinode *)alloc_bh->b_data;
2814         if ((bit + 1) > ocfs2_bits_per_group(&alloc_di->id2.i_chain)) {
2815                 mlog(ML_ERROR, "suballoc bit %u out of range of %u\n",
2816                      (unsigned int)bit,
2817                      ocfs2_bits_per_group(&alloc_di->id2.i_chain));
2818                 status = -EINVAL;
2819                 goto bail;
2820         }
2821
2822         bg_blkno = group_blkno ? group_blkno :
2823                    ocfs2_which_suballoc_group(blkno, bit);
2824         status = ocfs2_read_group_descriptor(suballoc, alloc_di, bg_blkno,
2825                                              &group_bh);
2826         if (status < 0) {
2827                 mlog(ML_ERROR, "read group %llu failed %d\n",
2828                      (unsigned long long)bg_blkno, status);
2829                 goto bail;
2830         }
2831
2832         group = (struct ocfs2_group_desc *) group_bh->b_data;
2833         *res = ocfs2_test_bit(bit, (unsigned long *)group->bg_bitmap);
2834
2835 bail:
2836         brelse(group_bh);
2837
2838         if (status)
2839                 mlog_errno(status);
2840         return status;
2841 }
2842
2843 /*
2844  * Test if the bit representing this inode (blkno) is set in the
2845  * suballocator.
2846  *
2847  * On success, 0 is returned and *res is 1 for SET; 0 otherwise.
2848  *
2849  * In the event of failure, a negative value is returned and *res is
2850  * meaningless.
2851  *
2852  * Callers must make sure to hold nfs_sync_lock to prevent
2853  * ocfs2_delete_inode() on another node from accessing the same
2854  * suballocator concurrently.
2855  */
2856 int ocfs2_test_inode_bit(struct ocfs2_super *osb, u64 blkno, int *res)
2857 {
2858         int status;
2859         u64 group_blkno = 0;
2860         u16 suballoc_bit = 0, suballoc_slot = 0;
2861         struct inode *inode_alloc_inode;
2862         struct buffer_head *alloc_bh = NULL;
2863
2864         trace_ocfs2_test_inode_bit((unsigned long long)blkno);
2865
2866         status = ocfs2_get_suballoc_slot_bit(osb, blkno, &suballoc_slot,
2867                                              &group_blkno, &suballoc_bit);
2868         if (status < 0) {
2869                 mlog(ML_ERROR, "get alloc slot and bit failed %d\n", status);
2870                 goto bail;
2871         }
2872
2873         inode_alloc_inode =
2874                 ocfs2_get_system_file_inode(osb, INODE_ALLOC_SYSTEM_INODE,
2875                                             suballoc_slot);
2876         if (!inode_alloc_inode) {
2877                 /* the error code could be inaccurate, but we are not able to
2878                  * get the correct one. */
2879                 status = -EINVAL;
2880                 mlog(ML_ERROR, "unable to get alloc inode in slot %u\n",
2881                      (u32)suballoc_slot);
2882                 goto bail;
2883         }
2884
2885         mutex_lock(&inode_alloc_inode->i_mutex);
2886         status = ocfs2_inode_lock(inode_alloc_inode, &alloc_bh, 0);
2887         if (status < 0) {
2888                 mutex_unlock(&inode_alloc_inode->i_mutex);
2889                 mlog(ML_ERROR, "lock on alloc inode on slot %u failed %d\n",
2890                      (u32)suballoc_slot, status);
2891                 goto bail;
2892         }
2893
2894         status = ocfs2_test_suballoc_bit(osb, inode_alloc_inode, alloc_bh,
2895                                          group_blkno, blkno, suballoc_bit, res);
2896         if (status < 0)
2897                 mlog(ML_ERROR, "test suballoc bit failed %d\n", status);
2898
2899         ocfs2_inode_unlock(inode_alloc_inode, 0);
2900         mutex_unlock(&inode_alloc_inode->i_mutex);
2901
2902         iput(inode_alloc_inode);
2903         brelse(alloc_bh);
2904 bail:
2905         if (status)
2906                 mlog_errno(status);
2907         return status;
2908 }