resize/resize2fs.c

   1 /*
   2  * resize2fs.c --- ext2 main routine
   3  *
   4  * Copyright (C) 1997, 1998 by Theodore Ts'o and
   5  *      PowerQuest, Inc.
   6  *
   7  * Copyright (C) 1999, 2000 by Theosore Ts'o
   8  *
   9  * %Begin-Header%
  10  * This file may be redistributed under the terms of the GNU Public
  11  * License.
  12  * %End-Header%
  13  */
  14
  15 /*
  16  * Resizing a filesystem consists of the following phases:
  17  *
  18  *      1.  Adjust superblock and write out new parts of the inode
  19  *              table
  20  *      2.  Determine blocks which need to be relocated, and copy the
  21  *              contents of blocks from their old locations to the new ones.
  22  *      3.  Scan the inode table, doing the following:
  23  *              a.  If blocks have been moved, update the block
  24  *                      pointers in the inodes and indirect blocks to
  25  *                      point at the new block locations.
  26  *              b.  If parts of the inode table need to be evacuated,
  27  *                      copy inodes from their old locations to their
  28  *                      new ones.
  29  *              c.  If (b) needs to be done, note which blocks contain
  30  *                      directory information, since we will need to
  31  *                      update the directory information.
  32  *      4.  Update the directory blocks with the new inode locations.
  33  *      5.  Move the inode tables, if necessary.
  34  */
  35
  36 #include "config.h"
  37 #include "resize2fs.h"
  38 #include <time.h>
  39
  40 #ifdef __linux__                        /* Kludge for debugging */
  41 #define RESIZE2FS_DEBUG
  42 #endif
  43
  44 static void fix_uninit_block_bitmaps(ext2_filsys fs);
  45 static errcode_t adjust_superblock(ext2_resize_t rfs, blk64_t new_size);
  46 static errcode_t blocks_to_move(ext2_resize_t rfs);
  47 static errcode_t block_mover(ext2_resize_t rfs);
  48 static errcode_t inode_scan_and_fix(ext2_resize_t rfs);
  49 static errcode_t inode_ref_fix(ext2_resize_t rfs);
  50 static errcode_t move_itables(ext2_resize_t rfs);
  51 static errcode_t fix_resize_inode(ext2_filsys fs);
  52 static errcode_t ext2fs_calculate_summary_stats(ext2_filsys fs);
  53 static errcode_t fix_sb_journal_backup(ext2_filsys fs);
  54 static errcode_t mark_table_blocks(ext2_filsys fs,
  55                                    ext2fs_block_bitmap bmap);
  56 static errcode_t clear_sparse_super2_last_group(ext2_resize_t rfs);
  57 static errcode_t reserve_sparse_super2_last_group(ext2_resize_t rfs,
  58                                                  ext2fs_block_bitmap meta_bmap);
  59 static errcode_t resize_group_descriptors(ext2_resize_t rfs, blk64_t new_size);
  60 static errcode_t move_bg_metadata(ext2_resize_t rfs);
  61 static errcode_t zero_high_bits_in_inodes(ext2_resize_t rfs);
  62
  63 /*
  64  * Some helper functions to check if a block is in a metadata area
  65  */
  66 static inline int is_block_bm(ext2_filsys fs, unsigned int grp, blk64_t blk)
  67 {
  68         return blk == ext2fs_block_bitmap_loc(fs, grp);
  69 }
  70
  71 static inline int is_inode_bm(ext2_filsys fs, unsigned int grp, blk64_t blk)
  72 {
  73         return blk == ext2fs_inode_bitmap_loc(fs, grp);
  74 }
  75
  76 static inline int is_inode_tb(ext2_filsys fs, unsigned int grp, blk64_t blk)
  77 {
  78         return blk >= ext2fs_inode_table_loc(fs, grp) &&
  79                blk < (ext2fs_inode_table_loc(fs, grp) +
  80                       fs->inode_blocks_per_group);
  81 }
  82
  83 /* Some bigalloc helper macros which are more succint... */
  84 #define B2C(x)  EXT2FS_B2C(fs, (x))
  85 #define C2B(x)  EXT2FS_C2B(fs, (x))
  86 #define EQ_CLSTR(x, y) (B2C(x) == B2C(y))
  87 #define LE_CLSTR(x, y) (B2C(x) <= B2C(y))
  88 #define LT_CLSTR(x, y) (B2C(x) <  B2C(y))
  89 #define GE_CLSTR(x, y) (B2C(x) >= B2C(y))
  90 #define GT_CLSTR(x, y) (B2C(x) >  B2C(y))
  91
  92 static int lazy_itable_init;
  93
  94 /*
  95  * This is the top-level routine which does the dirty deed....
  96  */
  97 errcode_t resize_fs(ext2_filsys fs, blk64_t *new_size, int flags,
  98             errcode_t (*progress)(ext2_resize_t rfs, int pass,
  99                                           unsigned long cur,
 100                                           unsigned long max_val))
 101 {
 102         ext2_resize_t   rfs;
 103         errcode_t       retval;
 104         struct resource_track   rtrack, overall_track;
 105
 106         /*
 107          * Create the data structure
 108          */
 109         retval = ext2fs_get_mem(sizeof(struct ext2_resize_struct), &rfs);
 110         if (retval)
 111                 return retval;
 112
 113         memset(rfs, 0, sizeof(struct ext2_resize_struct));
 114         fs->priv_data = rfs;
 115         rfs->old_fs = fs;
 116         rfs->flags = flags;
 117         rfs->itable_buf  = 0;
 118         rfs->progress = progress;
 119
 120         init_resource_track(&overall_track, "overall resize2fs", fs->io);
 121         init_resource_track(&rtrack, "read_bitmaps", fs->io);
 122         retval = ext2fs_read_bitmaps(fs);
 123         if (retval)
 124                 goto errout;
 125         print_resource_track(rfs, &rtrack, fs->io);
 126
 127         fs->super->s_state |= EXT2_ERROR_FS;
 128         ext2fs_mark_super_dirty(fs);
 129         ext2fs_flush(fs);
 130
 131         init_resource_track(&rtrack, "fix_uninit_block_bitmaps 1", fs->io);
 132         fix_uninit_block_bitmaps(fs);
 133         print_resource_track(rfs, &rtrack, fs->io);
 134         retval = ext2fs_dup_handle(fs, &rfs->new_fs);
 135         if (retval)
 136                 goto errout;
 137
 138         init_resource_track(&rtrack, "resize_group_descriptors", fs->io);
 139         retval = resize_group_descriptors(rfs, *new_size);
 140         if (retval)
 141                 goto errout;
 142         print_resource_track(rfs, &rtrack, fs->io);
 143
 144         init_resource_track(&rtrack, "move_bg_metadata", fs->io);
 145         retval = move_bg_metadata(rfs);
 146         if (retval)
 147                 goto errout;
 148         print_resource_track(rfs, &rtrack, fs->io);
 149
 150         init_resource_track(&rtrack, "zero_high_bits_in_metadata", fs->io);
 151         retval = zero_high_bits_in_inodes(rfs);
 152         if (retval)
 153                 goto errout;
 154         print_resource_track(rfs, &rtrack, fs->io);
 155
 156         init_resource_track(&rtrack, "adjust_superblock", fs->io);
 157         retval = adjust_superblock(rfs, *new_size);
 158         if (retval)
 159                 goto errout;
 160         print_resource_track(rfs, &rtrack, fs->io);
 161
 162         init_resource_track(&rtrack, "fix_uninit_block_bitmaps 2", fs->io);
 163         fix_uninit_block_bitmaps(rfs->new_fs);
 164         print_resource_track(rfs, &rtrack, fs->io);
 165         /* Clear the block bitmap uninit flag for the last block group */
 166         ext2fs_bg_flags_clear(rfs->new_fs, rfs->new_fs->group_desc_count - 1,
 167                              EXT2_BG_BLOCK_UNINIT);
 168
 169         *new_size = ext2fs_blocks_count(rfs->new_fs->super);
 170
 171         init_resource_track(&rtrack, "blocks_to_move", fs->io);
 172         retval = blocks_to_move(rfs);
 173         if (retval)
 174                 goto errout;
 175         print_resource_track(rfs, &rtrack, fs->io);
 176
 177 #ifdef RESIZE2FS_DEBUG
 178         if (rfs->flags & RESIZE_DEBUG_BMOVE)
 179                 printf("Number of free blocks: %llu/%llu, Needed: %llu\n",
 180                        ext2fs_free_blocks_count(rfs->old_fs->super),
 181                        ext2fs_free_blocks_count(rfs->new_fs->super),
 182                        rfs->needed_blocks);
 183 #endif
 184
 185         init_resource_track(&rtrack, "block_mover", fs->io);
 186         retval = block_mover(rfs);
 187         if (retval)
 188                 goto errout;
 189         print_resource_track(rfs, &rtrack, fs->io);
 190
 191         init_resource_track(&rtrack, "inode_scan_and_fix", fs->io);
 192         retval = inode_scan_and_fix(rfs);
 193         if (retval)
 194                 goto errout;
 195         print_resource_track(rfs, &rtrack, fs->io);
 196
 197         init_resource_track(&rtrack, "inode_ref_fix", fs->io);
 198         retval = inode_ref_fix(rfs);
 199         if (retval)
 200                 goto errout;
 201         print_resource_track(rfs, &rtrack, fs->io);
 202
 203         init_resource_track(&rtrack, "move_itables", fs->io);
 204         retval = move_itables(rfs);
 205         if (retval)
 206                 goto errout;
 207         print_resource_track(rfs, &rtrack, fs->io);
 208
 209         init_resource_track(&rtrack, "calculate_summary_stats", fs->io);
 210         retval = ext2fs_calculate_summary_stats(rfs->new_fs);
 211         if (retval)
 212                 goto errout;
 213         print_resource_track(rfs, &rtrack, fs->io);
 214
 215         init_resource_track(&rtrack, "fix_resize_inode", fs->io);
 216         retval = fix_resize_inode(rfs->new_fs);
 217         if (retval)
 218                 goto errout;
 219         print_resource_track(rfs, &rtrack, fs->io);
 220
 221         init_resource_track(&rtrack, "fix_sb_journal_backup", fs->io);
 222         retval = fix_sb_journal_backup(rfs->new_fs);
 223         if (retval)
 224                 goto errout;
 225         print_resource_track(rfs, &rtrack, fs->io);
 226
 227         retval = clear_sparse_super2_last_group(rfs);
 228         if (retval)
 229                 goto errout;
 230
 231         retval = ext2fs_set_gdt_csum(rfs->new_fs);
 232         if (retval)
 233                 goto errout;
 234
 235         rfs->new_fs->super->s_state &= ~EXT2_ERROR_FS;
 236         rfs->new_fs->flags &= ~EXT2_FLAG_MASTER_SB_ONLY;
 237
 238         print_resource_track(rfs, &overall_track, fs->io);
 239         retval = ext2fs_close_free(&rfs->new_fs);
 240         if (retval)
 241                 goto errout;
 242
 243         rfs->flags = flags;
 244
 245         ext2fs_free(rfs->old_fs);
 246         rfs->old_fs = NULL;
 247         if (rfs->itable_buf)
 248                 ext2fs_free_mem(&rfs->itable_buf);
 249         if (rfs->reserve_blocks)
 250                 ext2fs_free_block_bitmap(rfs->reserve_blocks);
 251         if (rfs->move_blocks)
 252                 ext2fs_free_block_bitmap(rfs->move_blocks);
 253         ext2fs_free_mem(&rfs);
 254
 255         return 0;
 256
 257 errout:
 258         if (rfs->new_fs) {
 259                 ext2fs_free(rfs->new_fs);
 260                 rfs->new_fs = NULL;
 261         }
 262         if (rfs->itable_buf)
 263                 ext2fs_free_mem(&rfs->itable_buf);
 264         ext2fs_free_mem(&rfs);
 265         return retval;
 266 }
 267
 268 /* Keep the size of the group descriptor region constant */
 269 static void adjust_reserved_gdt_blocks(ext2_filsys old_fs, ext2_filsys fs)
 270 {
 271         if (ext2fs_has_feature_resize_inode(fs->super) &&
 272             (old_fs->desc_blocks != fs->desc_blocks)) {
 273                 int new;
 274
 275                 new = ((int) fs->super->s_reserved_gdt_blocks) +
 276                         (old_fs->desc_blocks - fs->desc_blocks);
 277                 if (new < 0)
 278                         new = 0;
 279                 if (new > (int) fs->blocksize/4)
 280                         new = fs->blocksize/4;
 281                 fs->super->s_reserved_gdt_blocks = new;
 282         }
 283 }
 284
 285 /* Toggle 64bit mode */
 286 static errcode_t resize_group_descriptors(ext2_resize_t rfs, blk64_t new_size)
 287 {
 288         void *o, *n, *new_group_desc;
 289         dgrp_t i;
 290         int copy_size;
 291         errcode_t retval;
 292
 293         if (!(rfs->flags & (RESIZE_DISABLE_64BIT | RESIZE_ENABLE_64BIT)))
 294                 return 0;
 295
 296         if (new_size != ext2fs_blocks_count(rfs->new_fs->super) ||
 297             ext2fs_blocks_count(rfs->new_fs->super) >= (1ULL << 32) ||
 298             (rfs->flags & RESIZE_DISABLE_64BIT &&
 299              rfs->flags & RESIZE_ENABLE_64BIT))
 300                 return EXT2_ET_INVALID_ARGUMENT;
 301
 302         if (rfs->flags & RESIZE_DISABLE_64BIT) {
 303                 ext2fs_clear_feature_64bit(rfs->new_fs->super);
 304                 rfs->new_fs->super->s_desc_size = EXT2_MIN_DESC_SIZE;
 305         } else if (rfs->flags & RESIZE_ENABLE_64BIT) {
 306                 ext2fs_set_feature_64bit(rfs->new_fs->super);
 307                 rfs->new_fs->super->s_desc_size = EXT2_MIN_DESC_SIZE_64BIT;
 308         }
 309
 310         if (EXT2_DESC_SIZE(rfs->old_fs->super) ==
 311             EXT2_DESC_SIZE(rfs->new_fs->super))
 312                 return 0;
 313
 314         o = rfs->new_fs->group_desc;
 315         rfs->new_fs->desc_blocks = ext2fs_div_ceil(
 316                         rfs->old_fs->group_desc_count,
 317                         EXT2_DESC_PER_BLOCK(rfs->new_fs->super));
 318         retval = ext2fs_get_arrayzero(rfs->new_fs->desc_blocks,
 319                                       rfs->old_fs->blocksize, &new_group_desc);
 320         if (retval)
 321                 return retval;
 322
 323         n = new_group_desc;
 324
 325         if (EXT2_DESC_SIZE(rfs->old_fs->super) <=
 326             EXT2_DESC_SIZE(rfs->new_fs->super))
 327                 copy_size = EXT2_DESC_SIZE(rfs->old_fs->super);
 328         else
 329                 copy_size = EXT2_DESC_SIZE(rfs->new_fs->super);
 330         for (i = 0; i < rfs->old_fs->group_desc_count; i++) {
 331                 memcpy(n, o, copy_size);
 332                 n += EXT2_DESC_SIZE(rfs->new_fs->super);
 333                 o += EXT2_DESC_SIZE(rfs->old_fs->super);
 334         }
 335
 336         ext2fs_free_mem(&rfs->new_fs->group_desc);
 337         rfs->new_fs->group_desc = new_group_desc;
 338
 339         for (i = 0; i < rfs->old_fs->group_desc_count; i++)
 340                 ext2fs_group_desc_csum_set(rfs->new_fs, i);
 341
 342         adjust_reserved_gdt_blocks(rfs->old_fs, rfs->new_fs);
 343
 344         return 0;
 345 }
 346
 347 /* Move bitmaps/inode tables out of the way. */
 348 static errcode_t move_bg_metadata(ext2_resize_t rfs)
 349 {
 350         dgrp_t i;
 351         blk64_t b, c, d, old_desc_blocks, new_desc_blocks, j;
 352         ext2fs_block_bitmap old_map, new_map;
 353         int old, new;
 354         errcode_t retval;
 355         int cluster_ratio;
 356
 357         if (!(rfs->flags & (RESIZE_DISABLE_64BIT | RESIZE_ENABLE_64BIT)))
 358                 return 0;
 359
 360         retval = ext2fs_allocate_block_bitmap(rfs->old_fs, "oldfs", &old_map);
 361         if (retval)
 362                 return retval;
 363
 364         retval = ext2fs_allocate_block_bitmap(rfs->new_fs, "newfs", &new_map);
 365         if (retval)
 366                 goto out;
 367
 368         if (ext2fs_has_feature_meta_bg(rfs->old_fs->super)) {
 369                 old_desc_blocks = rfs->old_fs->super->s_first_meta_bg;
 370                 new_desc_blocks = rfs->new_fs->super->s_first_meta_bg;
 371         } else {
 372                 old_desc_blocks = rfs->old_fs->desc_blocks +
 373                                 rfs->old_fs->super->s_reserved_gdt_blocks;
 374                 new_desc_blocks = rfs->new_fs->desc_blocks +
 375                                 rfs->new_fs->super->s_reserved_gdt_blocks;
 376         }
 377
 378         /* Construct bitmaps of super/descriptor blocks in old and new fs */
 379         for (i = 0; i < rfs->old_fs->group_desc_count; i++) {
 380                 retval = ext2fs_super_and_bgd_loc2(rfs->old_fs, i, &b, &c, &d,
 381                                                    NULL);
 382                 if (retval)
 383                         goto out;
 384                 if (b)
 385                         ext2fs_mark_block_bitmap2(old_map, b);
 386                 for (j = 0; c != 0 && j < old_desc_blocks; j++)
 387                         ext2fs_mark_block_bitmap2(old_map, c + j);
 388                 if (d)
 389                         ext2fs_mark_block_bitmap2(old_map, d);
 390
 391                 retval = ext2fs_super_and_bgd_loc2(rfs->new_fs, i, &b, &c, &d,
 392                                                    NULL);
 393                 if (retval)
 394                         goto out;
 395                 if (b)
 396                         ext2fs_mark_block_bitmap2(new_map, b);
 397                 for (j = 0; c != 0 && j < new_desc_blocks; j++)
 398                         ext2fs_mark_block_bitmap2(new_map, c + j);
 399                 if (d)
 400                         ext2fs_mark_block_bitmap2(new_map, d);
 401         }
 402
 403         cluster_ratio = EXT2FS_CLUSTER_RATIO(rfs->new_fs);
 404
 405         /* Find changes in block allocations for bg metadata */
 406         for (b = EXT2FS_B2C(rfs->old_fs,
 407                             rfs->old_fs->super->s_first_data_block);
 408              b < ext2fs_blocks_count(rfs->new_fs->super);
 409              b += cluster_ratio) {
 410                 old = ext2fs_test_block_bitmap2(old_map, b);
 411                 new = ext2fs_test_block_bitmap2(new_map, b);
 412
 413                 if (old && !new) {
 414                         /* mark old_map, unmark new_map */
 415                         if (cluster_ratio == 1)
 416                                 ext2fs_unmark_block_bitmap2(
 417                                                 rfs->new_fs->block_map, b);
 418                 } else if (!old && new)
 419                         ; /* unmark old_map, mark new_map */
 420                 else {
 421                         ext2fs_unmark_block_bitmap2(old_map, b);
 422                         ext2fs_unmark_block_bitmap2(new_map, b);
 423                 }
 424         }
 425
 426         /*
 427          * new_map now shows blocks that have been newly allocated.
 428          * old_map now shows blocks that have been newly freed.
 429          */
 430
 431         /*
 432          * Move any conflicting bitmaps and inode tables.  Ensure that we
 433          * don't try to free clusters associated with bitmaps or tables.
 434          */
 435         for (i = 0; i < rfs->old_fs->group_desc_count; i++) {
 436                 b = ext2fs_block_bitmap_loc(rfs->new_fs, i);
 437                 if (ext2fs_test_block_bitmap2(new_map, b))
 438                         ext2fs_block_bitmap_loc_set(rfs->new_fs, i, 0);
 439                 else if (ext2fs_test_block_bitmap2(old_map, b))
 440                         ext2fs_unmark_block_bitmap2(old_map, b);
 441
 442                 b = ext2fs_inode_bitmap_loc(rfs->new_fs, i);
 443                 if (ext2fs_test_block_bitmap2(new_map, b))
 444                         ext2fs_inode_bitmap_loc_set(rfs->new_fs, i, 0);
 445                 else if (ext2fs_test_block_bitmap2(old_map, b))
 446                         ext2fs_unmark_block_bitmap2(old_map, b);
 447
 448                 c = ext2fs_inode_table_loc(rfs->new_fs, i);
 449                 for (b = 0;
 450                      b < rfs->new_fs->inode_blocks_per_group;
 451                      b++) {
 452                         if (ext2fs_test_block_bitmap2(new_map, b + c))
 453                                 ext2fs_inode_table_loc_set(rfs->new_fs, i, 0);
 454                         else if (ext2fs_test_block_bitmap2(old_map, b + c))
 455                                 ext2fs_unmark_block_bitmap2(old_map, b + c);
 456                 }
 457         }
 458
 459         /* Free unused clusters */
 460         for (b = 0;
 461              cluster_ratio > 1 && b < ext2fs_blocks_count(rfs->new_fs->super);
 462              b += cluster_ratio)
 463                 if (ext2fs_test_block_bitmap2(old_map, b))
 464                         ext2fs_unmark_block_bitmap2(rfs->new_fs->block_map, b);
 465 out:
 466         if (old_map)
 467                 ext2fs_free_block_bitmap(old_map);
 468         if (new_map)
 469                 ext2fs_free_block_bitmap(new_map);
 470         return retval;
 471 }
 472
 473 /* Zero out the high bits of extent fields */
 474 static errcode_t zero_high_bits_in_extents(ext2_filsys fs, ext2_ino_t ino,
 475                                  struct ext2_inode *inode)
 476 {
 477         ext2_extent_handle_t    handle;
 478         struct ext2fs_extent    extent;
 479         int                     op = EXT2_EXTENT_ROOT;
 480         errcode_t               errcode;
 481
 482         if (!(inode->i_flags & EXT4_EXTENTS_FL))
 483                 return 0;
 484
 485         errcode = ext2fs_extent_open(fs, ino, &handle);
 486         if (errcode)
 487                 return errcode;
 488
 489         while (1) {
 490                 errcode = ext2fs_extent_get(handle, op, &extent);
 491                 if (errcode)
 492                         break;
 493
 494                 op = EXT2_EXTENT_NEXT_SIB;
 495
 496                 if (extent.e_pblk > (1ULL << 32)) {
 497                         extent.e_pblk &= (1ULL << 32) - 1;
 498                         errcode = ext2fs_extent_replace(handle, 0, &extent);
 499                         if (errcode)
 500                                 break;
 501                 }
 502         }
 503
 504         /* Ok if we run off the end */
 505         if (errcode == EXT2_ET_EXTENT_NO_NEXT)
 506                 errcode = 0;
 507         ext2fs_extent_free(handle);
 508         return errcode;
 509 }
 510
 511 /* Zero out the high bits of inodes. */
 512 static errcode_t zero_high_bits_in_inodes(ext2_resize_t rfs)
 513 {
 514         ext2_filsys     fs = rfs->old_fs;
 515         int length = EXT2_INODE_SIZE(fs->super);
 516         struct ext2_inode *inode = NULL;
 517         ext2_inode_scan scan = NULL;
 518         errcode_t       retval;
 519         ext2_ino_t      ino;
 520
 521         if (!(rfs->flags & (RESIZE_DISABLE_64BIT | RESIZE_ENABLE_64BIT)))
 522                 return 0;
 523
 524         if (fs->super->s_creator_os != EXT2_OS_LINUX)
 525                 return 0;
 526
 527         retval = ext2fs_open_inode_scan(fs, 0, &scan);
 528         if (retval)
 529                 return retval;
 530
 531         retval = ext2fs_get_mem(length, &inode);
 532         if (retval)
 533                 goto out;
 534
 535         do {
 536                 retval = ext2fs_get_next_inode_full(scan, &ino, inode, length);
 537                 if (retval)
 538                         goto out;
 539                 if (!ino)
 540                         break;
 541                 if (!ext2fs_test_inode_bitmap2(fs->inode_map, ino))
 542                         continue;
 543
 544                 /*
 545                  * Here's how we deal with high block number fields:
 546                  *
 547                  *  - i_size_high has been been written out with i_size_lo
 548                  *    since the ext2 days, so no conversion is needed.
 549                  *
 550                  *  - i_blocks_hi is guarded by both the huge_file feature and
 551                  *    inode flags and has always been written out with
 552                  *    i_blocks_lo if the feature is set.  The field is only
 553                  *    ever read if both feature and inode flag are set, so
 554                  *    we don't need to zero it now.
 555                  *
 556                  *  - i_file_acl_high can be uninitialized, so zero it if
 557                  *    it isn't already.
 558                  */
 559                 if (inode->osd2.linux2.l_i_file_acl_high) {
 560                         inode->osd2.linux2.l_i_file_acl_high = 0;
 561                         retval = ext2fs_write_inode_full(fs, ino, inode,
 562                                                          length);
 563                         if (retval)
 564                                 goto out;
 565                 }
 566
 567                 retval = zero_high_bits_in_extents(fs, ino, inode);
 568                 if (retval)
 569                         goto out;
 570         } while (ino);
 571
 572 out:
 573         if (inode)
 574                 ext2fs_free_mem(&inode);
 575         if (scan)
 576                 ext2fs_close_inode_scan(scan);
 577         return retval;
 578 }
 579
 580 /*
 581  * Clean up the bitmaps for unitialized bitmaps
 582  */
 583 static void fix_uninit_block_bitmaps(ext2_filsys fs)
 584 {
 585         blk64_t         blk, lblk;
 586         dgrp_t          g;
 587         int             i;
 588
 589         if (!ext2fs_has_group_desc_csum(fs))
 590                 return;
 591
 592         for (g=0; g < fs->group_desc_count; g++) {
 593                 if (!(ext2fs_bg_flags_test(fs, g, EXT2_BG_BLOCK_UNINIT)))
 594                         continue;
 595
 596                 blk = ext2fs_group_first_block2(fs, g);
 597                 lblk = ext2fs_group_last_block2(fs, g);
 598                 ext2fs_unmark_block_bitmap_range2(fs->block_map, blk,
 599                                                   lblk - blk + 1);
 600
 601                 ext2fs_reserve_super_and_bgd(fs, g, fs->block_map);
 602                 ext2fs_mark_block_bitmap2(fs->block_map,
 603                                           ext2fs_block_bitmap_loc(fs, g));
 604                 ext2fs_mark_block_bitmap2(fs->block_map,
 605                                           ext2fs_inode_bitmap_loc(fs, g));
 606                 for (i = 0, blk = ext2fs_inode_table_loc(fs, g);
 607                      i < (unsigned int) fs->inode_blocks_per_group;
 608                      i++, blk++)
 609                         ext2fs_mark_block_bitmap2(fs->block_map, blk);
 610         }
 611 }
 612
 613 /* --------------------------------------------------------------------
 614  *
 615  * Resize processing, phase 1.
 616  *
 617  * In this phase we adjust the in-memory superblock information, and
 618  * initialize any new parts of the inode table.  The new parts of the
 619  * inode table are created in virgin disk space, so we can abort here
 620  * without any side effects.
 621  * --------------------------------------------------------------------
 622  */
 623
 624 /*
 625  * If the group descriptor's bitmap and inode table blocks are valid,
 626  * release them in the new filesystem data structure, and mark them as
 627  * reserved so the old inode table blocks don't get overwritten.
 628  */
 629 static errcode_t free_gdp_blocks(ext2_filsys fs,
 630                                  ext2fs_block_bitmap reserve_blocks,
 631                                  ext2_filsys old_fs,
 632                                  dgrp_t group)
 633 {
 634         blk64_t blk;
 635         int     j;
 636         dgrp_t  i;
 637         ext2fs_block_bitmap bg_map = NULL;
 638         errcode_t retval = 0;
 639         dgrp_t count = old_fs->group_desc_count - fs->group_desc_count;
 640
 641         /* If bigalloc, don't free metadata living in the same cluster */
 642         if (EXT2FS_CLUSTER_RATIO(fs) > 1) {
 643                 retval = ext2fs_allocate_block_bitmap(fs, "bgdata", &bg_map);
 644                 if (retval)
 645                         goto out;
 646
 647                 retval = mark_table_blocks(fs, bg_map);
 648                 if (retval)
 649                         goto out;
 650         }
 651
 652         for (i = group; i < group + count; i++) {
 653                 blk = ext2fs_block_bitmap_loc(old_fs, i);
 654                 if (blk &&
 655                     (blk < ext2fs_blocks_count(fs->super)) &&
 656                     !(bg_map && ext2fs_test_block_bitmap2(bg_map, blk))) {
 657                         ext2fs_block_alloc_stats2(fs, blk, -1);
 658                         ext2fs_mark_block_bitmap2(reserve_blocks, blk);
 659                 }
 660
 661                 blk = ext2fs_inode_bitmap_loc(old_fs, i);
 662                 if (blk &&
 663                     (blk < ext2fs_blocks_count(fs->super)) &&
 664                     !(bg_map && ext2fs_test_block_bitmap2(bg_map, blk))) {
 665                         ext2fs_block_alloc_stats2(fs, blk, -1);
 666                         ext2fs_mark_block_bitmap2(reserve_blocks, blk);
 667                 }
 668
 669                 blk = ext2fs_inode_table_loc(old_fs, i);
 670                 for (j = 0;
 671                      j < fs->inode_blocks_per_group; j++, blk++) {
 672                         if (blk >= ext2fs_blocks_count(fs->super) ||
 673                             (bg_map && ext2fs_test_block_bitmap2(bg_map, blk)))
 674                                 continue;
 675                         ext2fs_block_alloc_stats2(fs, blk, -1);
 676                         ext2fs_mark_block_bitmap2(reserve_blocks, blk);
 677                 }
 678         }
 679
 680 out:
 681         if (bg_map)
 682                 ext2fs_free_block_bitmap(bg_map);
 683         return retval;
 684 }
 685
 686 /*
 687  * This routine is shared by the online and offline resize routines.
 688  * All of the information which is adjusted in memory is done here.
 689  */
 690 errcode_t adjust_fs_info(ext2_filsys fs, ext2_filsys old_fs,
 691                          ext2fs_block_bitmap reserve_blocks, blk64_t new_size)
 692 {
 693         errcode_t       retval;
 694         blk64_t         overhead = 0;
 695         blk64_t         rem;
 696         blk64_t         blk, group_block;
 697         blk64_t         real_end;
 698         blk64_t         old_numblocks, numblocks, adjblocks;
 699         unsigned long   i, j, old_desc_blocks;
 700         unsigned int    meta_bg, meta_bg_size;
 701         int             has_super, csum_flag;
 702         unsigned long long new_inodes;  /* u64 to check for overflow */
 703         double          percent;
 704
 705         ext2fs_blocks_count_set(fs->super, new_size);
 706
 707 retry:
 708         fs->group_desc_count = ext2fs_div64_ceil(ext2fs_blocks_count(fs->super) -
 709                                        fs->super->s_first_data_block,
 710                                        EXT2_BLOCKS_PER_GROUP(fs->super));
 711         if (fs->group_desc_count == 0)
 712                 return EXT2_ET_TOOSMALL;
 713         fs->desc_blocks = ext2fs_div_ceil(fs->group_desc_count,
 714                                           EXT2_DESC_PER_BLOCK(fs->super));
 715
 716         /*
 717          * Overhead is the number of bookkeeping blocks per group.  It
 718          * includes the superblock backup, the group descriptor
 719          * backups, the inode bitmap, the block bitmap, and the inode
 720          * table.
 721          */
 722         overhead = (int) (2 + fs->inode_blocks_per_group);
 723
 724         if (ext2fs_bg_has_super(fs, fs->group_desc_count - 1))
 725                 overhead += 1 + fs->desc_blocks +
 726                         fs->super->s_reserved_gdt_blocks;
 727
 728         /*
 729          * See if the last group is big enough to support the
 730          * necessary data structures.  If not, we need to get rid of
 731          * it.
 732          */
 733         rem = (ext2fs_blocks_count(fs->super) - fs->super->s_first_data_block) %
 734                 fs->super->s_blocks_per_group;
 735         if ((fs->group_desc_count == 1) && rem && (rem < overhead))
 736                 return EXT2_ET_TOOSMALL;
 737         if ((fs->group_desc_count > 1) && rem && (rem < overhead+50)) {
 738                 ext2fs_blocks_count_set(fs->super,
 739                                         ext2fs_blocks_count(fs->super) - rem);
 740                 goto retry;
 741         }
 742         /*
 743          * Adjust the number of inodes
 744          */
 745         new_inodes =(unsigned long long) fs->super->s_inodes_per_group * fs->group_desc_count;
 746         if (new_inodes > ~0U) {
 747                 fprintf(stderr, _("inodes (%llu) must be less than %u"),
 748                                    new_inodes, ~0U);
 749                 return EXT2_ET_TOO_MANY_INODES;
 750         }
 751         fs->super->s_inodes_count = fs->super->s_inodes_per_group *
 752                 fs->group_desc_count;
 753
 754         /*
 755          * Adjust the number of free blocks
 756          */
 757         blk = ext2fs_blocks_count(old_fs->super);
 758         if (blk > ext2fs_blocks_count(fs->super))
 759                 ext2fs_free_blocks_count_set(fs->super,
 760                         ext2fs_free_blocks_count(fs->super) -
 761                         (blk - ext2fs_blocks_count(fs->super)));
 762         else
 763                 ext2fs_free_blocks_count_set(fs->super,
 764                         ext2fs_free_blocks_count(fs->super) +
 765                         (ext2fs_blocks_count(fs->super) - blk));
 766
 767         /*
 768          * Adjust the number of reserved blocks
 769          */
 770         percent = (ext2fs_r_blocks_count(old_fs->super) * 100.0) /
 771                 ext2fs_blocks_count(old_fs->super);
 772         ext2fs_r_blocks_count_set(fs->super,
 773                                   (percent * ext2fs_blocks_count(fs->super) /
 774                                    100.0));
 775
 776         /*
 777          * Adjust the bitmaps for size
 778          */
 779         retval = ext2fs_resize_inode_bitmap2(fs->super->s_inodes_count,
 780                                             fs->super->s_inodes_count,
 781                                             fs->inode_map);
 782         if (retval) goto errout;
 783
 784         real_end = EXT2_GROUPS_TO_BLOCKS(fs->super, fs->group_desc_count) - 1 +
 785                 fs->super->s_first_data_block;
 786         retval = ext2fs_resize_block_bitmap2(new_size - 1,
 787                                              real_end, fs->block_map);
 788         if (retval) goto errout;
 789
 790         /*
 791          * If we are growing the file system, also grow the size of
 792          * the reserve_blocks bitmap
 793          */
 794         if (reserve_blocks && new_size > ext2fs_blocks_count(old_fs->super)) {
 795                 retval = ext2fs_resize_block_bitmap2(new_size - 1,
 796                                                      real_end, reserve_blocks);
 797                 if (retval) goto errout;
 798         }
 799
 800         /*
 801          * Reallocate the group descriptors as necessary.
 802          */
 803         if (EXT2_DESC_SIZE(old_fs->super) == EXT2_DESC_SIZE(fs->super) &&
 804             old_fs->desc_blocks != fs->desc_blocks) {
 805                 retval = ext2fs_resize_mem(old_fs->desc_blocks *
 806                                            fs->blocksize,
 807                                            fs->desc_blocks * fs->blocksize,
 808                                            &fs->group_desc);
 809                 if (retval)
 810                         goto errout;
 811                 if (fs->desc_blocks > old_fs->desc_blocks)
 812                         memset((char *) fs->group_desc +
 813                                (old_fs->desc_blocks * fs->blocksize), 0,
 814                                (fs->desc_blocks - old_fs->desc_blocks) *
 815                                fs->blocksize);
 816         }
 817
 818         /*
 819          * If the resize_inode feature is set, and we are changing the
 820          * number of descriptor blocks, then adjust
 821          * s_reserved_gdt_blocks if possible to avoid needing to move
 822          * the inode table either now or in the future.
 823          *
 824          * Note: If we're converting to 64bit mode, we did this earlier.
 825          */
 826         if (EXT2_DESC_SIZE(old_fs->super) == EXT2_DESC_SIZE(fs->super))
 827                 adjust_reserved_gdt_blocks(old_fs, fs);
 828
 829         if (ext2fs_has_feature_meta_bg(fs->super) &&
 830             (fs->super->s_first_meta_bg > fs->desc_blocks)) {
 831                 ext2fs_clear_feature_meta_bg(fs->super);
 832                 fs->super->s_first_meta_bg = 0;
 833         }
 834
 835         /*
 836          * Update the location of the backup superblocks if the
 837          * sparse_super2 feature is enabled.
 838          */
 839         if (ext2fs_has_feature_sparse_super2(fs->super)) {
 840                 dgrp_t last_bg = fs->group_desc_count - 1;
 841                 dgrp_t old_last_bg = old_fs->group_desc_count - 1;
 842
 843                 if (last_bg > old_last_bg) {
 844                         if (old_fs->group_desc_count == 1)
 845                                 fs->super->s_backup_bgs[0] = 1;
 846                         if (old_fs->group_desc_count == 1 &&
 847                             fs->super->s_backup_bgs[0])
 848                                 fs->super->s_backup_bgs[0] = last_bg;
 849                         else if (fs->super->s_backup_bgs[1])
 850                                 fs->super->s_backup_bgs[1] = last_bg;
 851                 } else if (last_bg < old_last_bg) {
 852                         if (fs->super->s_backup_bgs[0] > last_bg)
 853                                 fs->super->s_backup_bgs[0] = 0;
 854                         if (fs->super->s_backup_bgs[1] > last_bg)
 855                                 fs->super->s_backup_bgs[1] = 0;
 856                         if (last_bg > 1 &&
 857                             old_fs->super->s_backup_bgs[1] == old_last_bg)
 858                                 fs->super->s_backup_bgs[1] = last_bg;
 859                 }
 860         }
 861
 862         /*
 863          * If we are shrinking the number of block groups, we're done
 864          * and can exit now.
 865          */
 866         if (old_fs->group_desc_count > fs->group_desc_count) {
 867                 /*
 868                  * Check the block groups that we are chopping off
 869                  * and free any blocks associated with their metadata
 870                  */
 871                 retval = free_gdp_blocks(fs, reserve_blocks, old_fs,
 872                                          fs->group_desc_count);
 873                 goto errout;
 874         }
 875
 876         /*
 877          * Fix the count of the last (old) block group
 878          */
 879         old_numblocks = (ext2fs_blocks_count(old_fs->super) -
 880                          old_fs->super->s_first_data_block) %
 881                                  old_fs->super->s_blocks_per_group;
 882         if (!old_numblocks)
 883                 old_numblocks = old_fs->super->s_blocks_per_group;
 884         if (old_fs->group_desc_count == fs->group_desc_count) {
 885                 numblocks = (ext2fs_blocks_count(fs->super) -
 886                              fs->super->s_first_data_block) %
 887                         fs->super->s_blocks_per_group;
 888                 if (!numblocks)
 889                         numblocks = fs->super->s_blocks_per_group;
 890         } else
 891                 numblocks = fs->super->s_blocks_per_group;
 892         i = old_fs->group_desc_count - 1;
 893         ext2fs_bg_free_blocks_count_set(fs, i, ext2fs_bg_free_blocks_count(fs, i) + (numblocks - old_numblocks));
 894         ext2fs_group_desc_csum_set(fs, i);
 895
 896         /*
 897          * If the number of block groups is staying the same, we're
 898          * done and can exit now.  (If the number block groups is
 899          * shrinking, we had exited earlier.)
 900          */
 901         if (old_fs->group_desc_count >= fs->group_desc_count) {
 902                 retval = 0;
 903                 goto errout;
 904         }
 905
 906         /*
 907          * Initialize the new block group descriptors
 908          */
 909         group_block = ext2fs_group_first_block2(fs,
 910                                                 old_fs->group_desc_count);
 911         csum_flag = ext2fs_has_group_desc_csum(fs);
 912         if (!getenv("RESIZE2FS_FORCE_ITABLE_INIT") &&
 913             access("/sys/fs/ext4/features/lazy_itable_init", F_OK) == 0)
 914                 lazy_itable_init = 1;
 915         if (ext2fs_has_feature_meta_bg(fs->super))
 916                 old_desc_blocks = fs->super->s_first_meta_bg;
 917         else
 918                 old_desc_blocks = fs->desc_blocks +
 919                         fs->super->s_reserved_gdt_blocks;
 920
 921         /*
 922          * If we changed the number of block_group descriptor blocks,
 923          * we need to make sure they are all marked as reserved in the
 924          * file systems's block allocation map.
 925          */
 926         for (i = 0; i < old_fs->group_desc_count; i++)
 927                 ext2fs_reserve_super_and_bgd(fs, i, fs->block_map);
 928
 929         for (i = old_fs->group_desc_count;
 930              i < fs->group_desc_count; i++) {
 931                 memset(ext2fs_group_desc(fs, fs->group_desc, i), 0,
 932                        sizeof(struct ext2_group_desc));
 933                 adjblocks = 0;
 934
 935                 ext2fs_bg_flags_zap(fs, i);
 936                 if (csum_flag) {
 937                         ext2fs_bg_flags_set(fs, i, EXT2_BG_INODE_UNINIT);
 938                         if (!lazy_itable_init)
 939                                 ext2fs_bg_flags_set(fs, i,
 940                                                     EXT2_BG_INODE_ZEROED);
 941                         ext2fs_bg_itable_unused_set(fs, i,
 942                                         fs->super->s_inodes_per_group);
 943                 }
 944
 945                 numblocks = ext2fs_group_blocks_count(fs, i);
 946                 if ((i < fs->group_desc_count - 1) && csum_flag)
 947                         ext2fs_bg_flags_set(fs, i, EXT2_BG_BLOCK_UNINIT);
 948
 949                 has_super = ext2fs_bg_has_super(fs, i);
 950                 if (has_super) {
 951                         ext2fs_block_alloc_stats2(fs, group_block, +1);
 952                         adjblocks++;
 953                 }
 954                 meta_bg_size = EXT2_DESC_PER_BLOCK(fs->super);
 955                 meta_bg = i / meta_bg_size;
 956                 if (!ext2fs_has_feature_meta_bg(fs->super) ||
 957                     (meta_bg < fs->super->s_first_meta_bg)) {
 958                         if (has_super) {
 959                                 for (j=0; j < old_desc_blocks; j++)
 960                                         ext2fs_block_alloc_stats2(fs,
 961                                                  group_block + 1 + j, +1);
 962                                 adjblocks += old_desc_blocks;
 963                         }
 964                 } else {
 965                         if (has_super)
 966                                 has_super = 1;
 967                         if (((i % meta_bg_size) == 0) ||
 968                             ((i % meta_bg_size) == 1) ||
 969                             ((i % meta_bg_size) == (meta_bg_size-1)))
 970                                 ext2fs_block_alloc_stats2(fs,
 971                                                  group_block + has_super, +1);
 972                 }
 973
 974                 adjblocks += 2 + fs->inode_blocks_per_group;
 975
 976                 numblocks -= adjblocks;
 977                 ext2fs_free_blocks_count_set(fs->super,
 978                              ext2fs_free_blocks_count(fs->super) - adjblocks);
 979                 fs->super->s_free_inodes_count +=
 980                         fs->super->s_inodes_per_group;
 981                 ext2fs_bg_free_blocks_count_set(fs, i, numblocks);
 982                 ext2fs_bg_free_inodes_count_set(fs, i,
 983                                                 fs->super->s_inodes_per_group);
 984                 ext2fs_bg_used_dirs_count_set(fs, i, 0);
 985                 ext2fs_group_desc_csum_set(fs, i);
 986
 987                 retval = ext2fs_allocate_group_table(fs, i, 0);
 988                 if (retval) goto errout;
 989
 990                 group_block += fs->super->s_blocks_per_group;
 991         }
 992         retval = 0;
 993
 994         /*
 995          * Mark all of the metadata blocks as reserved so they won't
 996          * get allocated by the call to ext2fs_allocate_group_table()
 997          * in blocks_to_move(), where we allocate new blocks to
 998          * replace those allocation bitmap and inode table blocks
 999          * which have to get relocated to make space for an increased
1000          * number of the block group descriptors.
1001          */
1002         if (reserve_blocks)
1003                 mark_table_blocks(fs, reserve_blocks);
1004
1005 errout:
1006         return (retval);
1007 }
1008
1009 /*
1010  * This routine adjusts the superblock and other data structures, both
1011  * in disk as well as in memory...
1012  */
1013 static errcode_t adjust_superblock(ext2_resize_t rfs, blk64_t new_size)
1014 {
1015         ext2_filsys     fs = rfs->new_fs;
1016         int             adj = 0;
1017         errcode_t       retval;
1018         blk64_t         group_block;
1019         unsigned long   i;
1020         unsigned long   max_group;
1021
1022         ext2fs_mark_super_dirty(fs);
1023         ext2fs_mark_bb_dirty(fs);
1024         ext2fs_mark_ib_dirty(fs);
1025
1026         retval = ext2fs_allocate_block_bitmap(fs, _("reserved blocks"),
1027                                               &rfs->reserve_blocks);
1028         if (retval)
1029                 return retval;
1030
1031         retval = adjust_fs_info(fs, rfs->old_fs, rfs->reserve_blocks, new_size);
1032         if (retval)
1033                 goto errout;
1034
1035         /*
1036          * Check to make sure there are enough inodes
1037          */
1038         if ((rfs->old_fs->super->s_inodes_count -
1039              rfs->old_fs->super->s_free_inodes_count) >
1040             rfs->new_fs->super->s_inodes_count) {
1041                 retval = ENOSPC;
1042                 goto errout;
1043         }
1044
1045         /*
1046          * If we are shrinking the number block groups, we're done and
1047          * can exit now.
1048          */
1049         if (rfs->old_fs->group_desc_count > fs->group_desc_count) {
1050                 retval = 0;
1051                 goto errout;
1052         }
1053
1054         /*
1055          * If the number of block groups is staying the same, we're
1056          * done and can exit now.  (If the number block groups is
1057          * shrinking, we had exited earlier.)
1058          */
1059         if (rfs->old_fs->group_desc_count >= fs->group_desc_count) {
1060                 retval = 0;
1061                 goto errout;
1062         }
1063
1064         /*
1065          * If we are using uninit_bg (aka GDT_CSUM) and the kernel
1066          * supports lazy inode initialization, we can skip
1067          * initializing the inode table.
1068          */
1069         if (lazy_itable_init && ext2fs_has_group_desc_csum(fs)) {
1070                 retval = 0;
1071                 goto errout;
1072         }
1073
1074         /*
1075          * Initialize the inode table
1076          */
1077         retval = ext2fs_get_array(fs->blocksize, fs->inode_blocks_per_group,
1078                                 &rfs->itable_buf);
1079         if (retval)
1080                 goto errout;
1081
1082         memset(rfs->itable_buf, 0, fs->blocksize * fs->inode_blocks_per_group);
1083         group_block = ext2fs_group_first_block2(fs,
1084                                                 rfs->old_fs->group_desc_count);
1085         adj = rfs->old_fs->group_desc_count;
1086         max_group = fs->group_desc_count - adj;
1087         if (rfs->progress) {
1088                 retval = rfs->progress(rfs, E2_RSZ_EXTEND_ITABLE_PASS,
1089                                        0, max_group);
1090                 if (retval)
1091                         goto errout;
1092         }
1093         for (i = rfs->old_fs->group_desc_count;
1094              i < fs->group_desc_count; i++) {
1095                 /*
1096                  * Write out the new inode table
1097                  */
1098                 retval = ext2fs_zero_blocks2(fs, ext2fs_inode_table_loc(fs, i),
1099                                              fs->inode_blocks_per_group, NULL,
1100                                              NULL);
1101                 if (retval)
1102                         goto errout;
1103
1104                 io_channel_flush(fs->io);
1105                 if (rfs->progress) {
1106                         retval = rfs->progress(rfs, E2_RSZ_EXTEND_ITABLE_PASS,
1107                                                i - adj + 1, max_group);
1108                         if (retval)
1109                                 goto errout;
1110                 }
1111                 group_block += fs->super->s_blocks_per_group;
1112         }
1113         io_channel_flush(fs->io);
1114         retval = 0;
1115
1116 errout:
1117         return retval;
1118 }
1119
1120 /* --------------------------------------------------------------------
1121  *
1122  * Resize processing, phase 2.
1123  *
1124  * In this phase we adjust determine which blocks need to be moved, in
1125  * blocks_to_move().  We then copy the blocks to their ultimate new
1126  * destinations using block_mover().  Since we are copying blocks to
1127  * their new locations, again during this pass we can abort without
1128  * any problems.
1129  * --------------------------------------------------------------------
1130  */
1131
1132 /*
1133  * This helper function creates a block bitmap with all of the
1134  * filesystem meta-data blocks.
1135  */
1136 static errcode_t mark_table_blocks(ext2_filsys fs,
1137                                    ext2fs_block_bitmap bmap)
1138 {
1139         dgrp_t                  i;
1140         blk64_t                 blk;
1141
1142         for (i = 0; i < fs->group_desc_count; i++) {
1143                 ext2fs_reserve_super_and_bgd(fs, i, bmap);
1144
1145                 /*
1146                  * Mark the blocks used for the inode table
1147                  */
1148                 blk = ext2fs_inode_table_loc(fs, i);
1149                 if (blk)
1150                         ext2fs_mark_block_bitmap_range2(bmap, blk,
1151                                                 fs->inode_blocks_per_group);
1152
1153                 /*
1154                  * Mark block used for the block bitmap
1155                  */
1156                 blk = ext2fs_block_bitmap_loc(fs, i);
1157                 if (blk)
1158                         ext2fs_mark_block_bitmap2(bmap, blk);
1159
1160                 /*
1161                  * Mark block used for the inode bitmap
1162                  */
1163                 blk = ext2fs_inode_bitmap_loc(fs, i);
1164                 if (blk)
1165                         ext2fs_mark_block_bitmap2(bmap, blk);
1166         }
1167         return 0;
1168 }
1169
1170 /*
1171  * This function checks to see if a particular block (either a
1172  * superblock or a block group descriptor) overlaps with an inode or
1173  * block bitmap block, or with the inode table.
1174  */
1175 static void mark_fs_metablock(ext2_resize_t rfs,
1176                               ext2fs_block_bitmap meta_bmap,
1177                               int group, blk64_t blk)
1178 {
1179         ext2_filsys fs = rfs->new_fs;
1180
1181         ext2fs_mark_block_bitmap2(rfs->reserve_blocks, blk);
1182         ext2fs_block_alloc_stats2(fs, blk, +1);
1183
1184         /*
1185          * Check to see if we overlap with the inode or block bitmap,
1186          * or the inode tables.  If not, and the block is in use, then
1187          * mark it as a block to be moved.
1188          */
1189         if (is_block_bm(fs, group, blk)) {
1190                 ext2fs_block_bitmap_loc_set(fs, group, 0);
1191                 rfs->needed_blocks++;
1192                 return;
1193         }
1194         if (is_inode_bm(fs, group, blk)) {
1195                 ext2fs_inode_bitmap_loc_set(fs, group, 0);
1196                 rfs->needed_blocks++;
1197                 return;
1198         }
1199         if (is_inode_tb(fs, group, blk)) {
1200                 ext2fs_inode_table_loc_set(fs, group, 0);
1201                 rfs->needed_blocks++;
1202                 return;
1203         }
1204         if (ext2fs_has_feature_flex_bg(fs->super)) {
1205                 dgrp_t i;
1206
1207                 for (i = 0; i < rfs->old_fs->group_desc_count; i++) {
1208                         if (is_block_bm(fs, i, blk)) {
1209                                 ext2fs_block_bitmap_loc_set(fs, i, 0);
1210                                 rfs->needed_blocks++;
1211                                 return;
1212                         }
1213                         if (is_inode_bm(fs, i, blk)) {
1214                                 ext2fs_inode_bitmap_loc_set(fs, i, 0);
1215                                 rfs->needed_blocks++;
1216                                 return;
1217                         }
1218                         if (is_inode_tb(fs, i, blk)) {
1219                                 ext2fs_inode_table_loc_set(fs, i, 0);
1220                                 rfs->needed_blocks++;
1221                                 return;
1222                         }
1223                 }
1224         }
1225
1226         if (ext2fs_has_group_desc_csum(fs) &&
1227             (ext2fs_bg_flags_test(fs, group, EXT2_BG_BLOCK_UNINIT))) {
1228                 /*
1229                  * If the block bitmap is uninitialized, which means
1230                  * nothing other than standard metadata in use.
1231                  */
1232                 return;
1233         } else if (ext2fs_test_block_bitmap2(rfs->old_fs->block_map, blk) &&
1234                    !ext2fs_test_block_bitmap2(meta_bmap, blk)) {
1235                 ext2fs_mark_block_bitmap2(rfs->move_blocks, blk);
1236                 rfs->needed_blocks++;
1237         }
1238 }
1239
1240
1241 /*
1242  * This routine marks and unmarks reserved blocks in the new block
1243  * bitmap.  It also determines which blocks need to be moved and
1244  * places this information into the move_blocks bitmap.
1245  */
1246 static errcode_t blocks_to_move(ext2_resize_t rfs)
1247 {
1248         int             j, has_super;
1249         dgrp_t          i, max_groups, g;
1250         blk64_t         blk, group_blk;
1251         blk64_t         old_blocks, new_blocks, group_end, cluster_freed;
1252         blk64_t         new_size;
1253         unsigned int    meta_bg, meta_bg_size;
1254         errcode_t       retval;
1255         ext2_filsys     fs, old_fs;
1256         ext2fs_block_bitmap     meta_bmap, new_meta_bmap = NULL;
1257         int             flex_bg;
1258
1259         fs = rfs->new_fs;
1260         old_fs = rfs->old_fs;
1261         if (ext2fs_blocks_count(old_fs->super) > ext2fs_blocks_count(fs->super))
1262                 fs = rfs->old_fs;
1263
1264         retval = ext2fs_allocate_block_bitmap(fs, _("blocks to be moved"),
1265                                               &rfs->move_blocks);
1266         if (retval)
1267                 return retval;
1268
1269         retval = ext2fs_allocate_block_bitmap(fs, _("meta-data blocks"),
1270                                               &meta_bmap);
1271         if (retval)
1272                 return retval;
1273
1274         retval = mark_table_blocks(old_fs, meta_bmap);
1275         if (retval)
1276                 return retval;
1277
1278         fs = rfs->new_fs;
1279
1280         /*
1281          * If we're shrinking the filesystem, we need to move any
1282          * group's metadata blocks (either allocation bitmaps or the
1283          * inode table) which are beyond the end of the new
1284          * filesystem.
1285          */
1286         new_size = ext2fs_blocks_count(fs->super);
1287         if (new_size < ext2fs_blocks_count(old_fs->super)) {
1288                 for (g = 0; g < fs->group_desc_count; g++) {
1289                         int realloc = 0;
1290                         /*
1291                          * ext2fs_allocate_group_table will re-allocate any
1292                          * metadata blocks whose location is set to zero.
1293                          */
1294                         if (ext2fs_block_bitmap_loc(fs, g) >= new_size) {
1295                                 ext2fs_block_bitmap_loc_set(fs, g, 0);
1296                                 realloc = 1;
1297                         }
1298                         if (ext2fs_inode_bitmap_loc(fs, g) >= new_size) {
1299                                 ext2fs_inode_bitmap_loc_set(fs, g, 0);
1300                                 realloc = 1;
1301                         }
1302                         if ((ext2fs_inode_table_loc(fs, g) +
1303                              fs->inode_blocks_per_group) > new_size) {
1304                                 ext2fs_inode_table_loc_set(fs, g, 0);
1305                                 realloc = 1;
1306                         }
1307
1308                         if (realloc) {
1309                                 retval = ext2fs_allocate_group_table(fs, g, 0);
1310                                 if (retval)
1311                                         return retval;
1312                         }
1313                 }
1314         }
1315
1316         /*
1317          * If we're shrinking the filesystem, we need to move all of
1318          * the blocks that don't fit any more
1319          */
1320         for (blk = ext2fs_blocks_count(fs->super);
1321              blk < ext2fs_blocks_count(old_fs->super); blk++) {
1322                 g = ext2fs_group_of_blk2(fs, blk);
1323                 if (ext2fs_has_group_desc_csum(fs) &&
1324                     ext2fs_bg_flags_test(old_fs, g, EXT2_BG_BLOCK_UNINIT)) {
1325                         /*
1326                          * The block bitmap is uninitialized, so skip
1327                          * to the next block group.
1328                          */
1329                         blk = ext2fs_group_first_block2(fs, g+1) - 1;
1330                         continue;
1331                 }
1332                 if (ext2fs_test_block_bitmap2(old_fs->block_map, blk) &&
1333                     !ext2fs_test_block_bitmap2(meta_bmap, blk)) {
1334                         ext2fs_mark_block_bitmap2(rfs->move_blocks, blk);
1335                         rfs->needed_blocks++;
1336                 }
1337                 ext2fs_mark_block_bitmap2(rfs->reserve_blocks, blk);
1338         }
1339
1340         if (ext2fs_has_feature_meta_bg(old_fs->super))
1341                 old_blocks = old_fs->super->s_first_meta_bg;
1342         else
1343                 old_blocks = old_fs->desc_blocks +
1344                         old_fs->super->s_reserved_gdt_blocks;
1345         if (ext2fs_has_feature_meta_bg(fs->super))
1346                 new_blocks = fs->super->s_first_meta_bg;
1347         else
1348                 new_blocks = fs->desc_blocks + fs->super->s_reserved_gdt_blocks;
1349
1350         retval = reserve_sparse_super2_last_group(rfs, meta_bmap);
1351         if (retval)
1352                 goto errout;
1353
1354         if (EXT2_DESC_SIZE(rfs->old_fs->super) ==
1355             EXT2_DESC_SIZE(rfs->new_fs->super) &&
1356             old_blocks == new_blocks) {
1357                 retval = 0;
1358                 goto errout;
1359         }
1360
1361         max_groups = fs->group_desc_count;
1362         if (max_groups > old_fs->group_desc_count)
1363                 max_groups = old_fs->group_desc_count;
1364         group_blk = old_fs->super->s_first_data_block;
1365         /*
1366          * If we're reducing the number of descriptor blocks, this
1367          * makes life easy.  :-)   We just have to mark some extra
1368          * blocks as free.
1369          */
1370         if (old_blocks > new_blocks) {
1371                 if (EXT2FS_CLUSTER_RATIO(fs) > 1) {
1372                         retval = ext2fs_allocate_block_bitmap(fs,
1373                                                         _("new meta blocks"),
1374                                                         &new_meta_bmap);
1375                         if (retval)
1376                                 goto errout;
1377
1378                         retval = mark_table_blocks(fs, new_meta_bmap);
1379                         if (retval)
1380                                 goto errout;
1381                 }
1382
1383                 for (i = 0; i < max_groups; i++) {
1384                         if (!ext2fs_bg_has_super(old_fs, i)) {
1385                                 group_blk += fs->super->s_blocks_per_group;
1386                                 continue;
1387                         }
1388                         group_end = group_blk + 1 + old_blocks;
1389                         for (blk = group_blk + 1 + new_blocks;
1390                              blk < group_end;) {
1391                                 if (new_meta_bmap == NULL ||
1392                                     !ext2fs_test_block_bitmap2(new_meta_bmap,
1393                                                                blk)) {
1394                                         cluster_freed =
1395                                                 EXT2FS_CLUSTER_RATIO(fs) -
1396                                                 (blk &
1397                                                  EXT2FS_CLUSTER_MASK(fs));
1398                                         if (cluster_freed > group_end - blk)
1399                                                 cluster_freed = group_end - blk;
1400                                         ext2fs_block_alloc_stats2(fs, blk, -1);
1401                                         blk += EXT2FS_CLUSTER_RATIO(fs);
1402                                         rfs->needed_blocks -= cluster_freed;
1403                                         continue;
1404                                 }
1405                                 rfs->needed_blocks--;
1406                                 blk++;
1407                         }
1408                         group_blk += fs->super->s_blocks_per_group;
1409                 }
1410                 retval = 0;
1411                 goto errout;
1412         }
1413         /*
1414          * If we're increasing the number of descriptor blocks, life
1415          * gets interesting....
1416          */
1417         meta_bg_size = EXT2_DESC_PER_BLOCK(fs->super);
1418         flex_bg = ext2fs_has_feature_flex_bg(fs->super);
1419         /* first reserve all of the existing fs meta blocks */
1420         for (i = 0; i < max_groups; i++) {
1421                 has_super = ext2fs_bg_has_super(fs, i);
1422                 if (has_super)
1423                         mark_fs_metablock(rfs, meta_bmap, i, group_blk);
1424
1425                 meta_bg = i / meta_bg_size;
1426                 if (!ext2fs_has_feature_meta_bg(fs->super) ||
1427                     (meta_bg < fs->super->s_first_meta_bg)) {
1428                         if (has_super) {
1429                                 for (blk = group_blk+1;
1430                                      blk < group_blk + 1 + new_blocks; blk++)
1431                                         mark_fs_metablock(rfs, meta_bmap,
1432                                                           i, blk);
1433                         }
1434                 } else {
1435                         if (has_super)
1436                                 has_super = 1;
1437                         if (((i % meta_bg_size) == 0) ||
1438                             ((i % meta_bg_size) == 1) ||
1439                             ((i % meta_bg_size) == (meta_bg_size-1)))
1440                                 mark_fs_metablock(rfs, meta_bmap, i,
1441                                                   group_blk + has_super);
1442                 }
1443
1444                 /*
1445                  * Reserve the existing meta blocks that we know
1446                  * aren't to be moved.
1447                  *
1448                  * For flex_bg file systems, in order to avoid
1449                  * overwriting fs metadata (especially inode table
1450                  * blocks) belonging to a different block group when
1451                  * we are relocating the inode tables, we need to
1452                  * reserve all existing fs metadata blocks.
1453                  */
1454                 if (ext2fs_block_bitmap_loc(fs, i))
1455                         ext2fs_mark_block_bitmap2(rfs->reserve_blocks,
1456                                  ext2fs_block_bitmap_loc(fs, i));
1457                 else if (flex_bg && i < old_fs->group_desc_count)
1458                         ext2fs_mark_block_bitmap2(rfs->reserve_blocks,
1459                                  ext2fs_block_bitmap_loc(old_fs, i));
1460
1461                 if (ext2fs_inode_bitmap_loc(fs, i))
1462                         ext2fs_mark_block_bitmap2(rfs->reserve_blocks,
1463                                  ext2fs_inode_bitmap_loc(fs, i));
1464                 else if (flex_bg && i < old_fs->group_desc_count)
1465                         ext2fs_mark_block_bitmap2(rfs->reserve_blocks,
1466                                  ext2fs_inode_bitmap_loc(old_fs, i));
1467
1468                 if (ext2fs_inode_table_loc(fs, i))
1469                         ext2fs_mark_block_bitmap_range2(rfs->reserve_blocks,
1470                                         ext2fs_inode_table_loc(fs, i),
1471                                         fs->inode_blocks_per_group);
1472                 else if (flex_bg && i < old_fs->group_desc_count)
1473                         ext2fs_mark_block_bitmap_range2(rfs->reserve_blocks,
1474                                         ext2fs_inode_table_loc(old_fs, i),
1475                                         old_fs->inode_blocks_per_group);
1476
1477                 group_blk += rfs->new_fs->super->s_blocks_per_group;
1478         }
1479
1480         /* Allocate the missing data structures */
1481         for (i = 0; i < max_groups; i++) {
1482                 if (ext2fs_inode_table_loc(fs, i) &&
1483                     ext2fs_inode_bitmap_loc(fs, i) &&
1484                     ext2fs_block_bitmap_loc(fs, i))
1485                         continue;
1486
1487                 retval = ext2fs_allocate_group_table(fs, i,
1488                                                      rfs->reserve_blocks);
1489                 if (retval)
1490                         goto errout;
1491
1492                 /*
1493                  * For those structures that have changed, we need to
1494                  * do bookkeepping.
1495                  */
1496                 if (ext2fs_block_bitmap_loc(old_fs, i) !=
1497                     (blk = ext2fs_block_bitmap_loc(fs, i))) {
1498                         ext2fs_block_alloc_stats2(fs, blk, +1);
1499                         if (ext2fs_test_block_bitmap2(old_fs->block_map, blk) &&
1500                             !ext2fs_test_block_bitmap2(meta_bmap, blk))
1501                                 ext2fs_mark_block_bitmap2(rfs->move_blocks,
1502                                                          blk);
1503                 }
1504                 if (ext2fs_inode_bitmap_loc(old_fs, i) !=
1505                     (blk = ext2fs_inode_bitmap_loc(fs, i))) {
1506                         ext2fs_block_alloc_stats2(fs, blk, +1);
1507                         if (ext2fs_test_block_bitmap2(old_fs->block_map, blk) &&
1508                             !ext2fs_test_block_bitmap2(meta_bmap, blk))
1509                                 ext2fs_mark_block_bitmap2(rfs->move_blocks,
1510                                                          blk);
1511                 }
1512
1513                 /*
1514                  * The inode table, if we need to relocate it, is
1515                  * handled specially.  We have to reserve the blocks
1516                  * for both the old and the new inode table, since we
1517                  * can't have the inode table be destroyed during the
1518                  * block relocation phase.
1519                  */
1520                 if (ext2fs_inode_table_loc(fs, i) == ext2fs_inode_table_loc(old_fs, i))
1521                         continue;       /* inode table not moved */
1522
1523                 rfs->needed_blocks += fs->inode_blocks_per_group;
1524
1525                 /*
1526                  * Mark the new inode table as in use in the new block
1527                  * allocation bitmap, and move any blocks that might
1528                  * be necessary.
1529                  */
1530                 for (blk = ext2fs_inode_table_loc(fs, i), j=0;
1531                      j < fs->inode_blocks_per_group ; j++, blk++) {
1532                         ext2fs_block_alloc_stats2(fs, blk, +1);
1533                         if (ext2fs_test_block_bitmap2(old_fs->block_map, blk) &&
1534                             !ext2fs_test_block_bitmap2(meta_bmap, blk))
1535                                 ext2fs_mark_block_bitmap2(rfs->move_blocks,
1536                                                          blk);
1537                 }
1538
1539                 /*
1540                  * Make sure the old inode table is reserved in the
1541                  * block reservation bitmap.
1542                  */
1543                 for (blk = ext2fs_inode_table_loc(rfs->old_fs, i), j=0;
1544                      j < fs->inode_blocks_per_group ; j++, blk++)
1545                         ext2fs_mark_block_bitmap2(rfs->reserve_blocks, blk);
1546         }
1547         retval = 0;
1548
1549 errout:
1550         if (new_meta_bmap)
1551                 ext2fs_free_block_bitmap(new_meta_bmap);
1552         if (meta_bmap)
1553                 ext2fs_free_block_bitmap(meta_bmap);
1554
1555         return retval;
1556 }
1557
1558 /*
1559  * This helper function tries to allocate a new block.  We try to
1560  * avoid hitting the original group descriptor blocks at least at
1561  * first, since we want to make it possible to recover from a badly
1562  * aborted resize operation as much as possible.
1563  *
1564  * In the future, I may further modify this routine to balance out
1565  * where we get the new blocks across the various block groups.
1566  * Ideally we would allocate blocks that corresponded with the block
1567  * group of the containing inode, and keep contiguous blocks
1568  * together.  However, this very difficult to do efficiently, since we
1569  * don't have the necessary information up front.
1570  */
1571
1572 #define AVOID_OLD       1
1573 #define DESPERATION     2
1574
1575 static void init_block_alloc(ext2_resize_t rfs)
1576 {
1577         rfs->alloc_state = AVOID_OLD;
1578         rfs->new_blk = rfs->new_fs->super->s_first_data_block;
1579 #if 0
1580         /* HACK for testing */
1581         if (ext2fs_blocks_count(rfs->new_fs->super) >
1582             ext2fs_blocks_count(rfs->old_fs->super))
1583                 rfs->new_blk = ext2fs_blocks_count(rfs->old_fs->super);
1584 #endif
1585 }
1586
1587 static blk64_t get_new_block(ext2_resize_t rfs)
1588 {
1589         ext2_filsys     fs = rfs->new_fs;
1590
1591         while (1) {
1592                 if (rfs->new_blk >= ext2fs_blocks_count(fs->super)) {
1593                         if (rfs->alloc_state == DESPERATION)
1594                                 return 0;
1595
1596 #ifdef RESIZE2FS_DEBUG
1597                         if (rfs->flags & RESIZE_DEBUG_BMOVE)
1598                                 printf("Going into desperation mode "
1599                                        "for block allocations\n");
1600 #endif
1601                         rfs->alloc_state = DESPERATION;
1602                         rfs->new_blk = fs->super->s_first_data_block;
1603                         continue;
1604                 }
1605                 if (ext2fs_test_block_bitmap2(fs->block_map, rfs->new_blk) ||
1606                     ext2fs_test_block_bitmap2(rfs->reserve_blocks,
1607                                              rfs->new_blk) ||
1608                     ((rfs->alloc_state == AVOID_OLD) &&
1609                      (rfs->new_blk < ext2fs_blocks_count(rfs->old_fs->super)) &&
1610                      ext2fs_test_block_bitmap2(rfs->old_fs->block_map,
1611                                               rfs->new_blk))) {
1612                         rfs->new_blk++;
1613                         continue;
1614                 }
1615                 return rfs->new_blk;
1616         }
1617 }
1618
1619 static errcode_t resize2fs_get_alloc_block(ext2_filsys fs,
1620                                            blk64_t goal EXT2FS_ATTR((unused)),
1621                                            blk64_t *ret)
1622 {
1623         ext2_resize_t rfs = (ext2_resize_t) fs->priv_data;
1624         blk64_t blk;
1625
1626         blk = get_new_block(rfs);
1627         if (!blk)
1628                 return ENOSPC;
1629
1630 #ifdef RESIZE2FS_DEBUG
1631         if (rfs->flags & 0xF)
1632                 printf("get_alloc_block allocating %llu\n", blk);
1633 #endif
1634
1635         ext2fs_mark_block_bitmap2(rfs->old_fs->block_map, blk);
1636         ext2fs_mark_block_bitmap2(rfs->new_fs->block_map, blk);
1637         *ret = (blk64_t) blk;
1638         return 0;
1639 }
1640
1641 static errcode_t block_mover(ext2_resize_t rfs)
1642 {
1643         blk64_t                 blk, old_blk, new_blk;
1644         ext2_filsys             fs = rfs->new_fs;
1645         ext2_filsys             old_fs = rfs->old_fs;
1646         errcode_t               retval;
1647         __u64                   size;
1648         int                     c;
1649         int                     to_move, moved;
1650         ext2_badblocks_list     badblock_list = 0;
1651         int                     bb_modified = 0;
1652
1653         fs->get_alloc_block = resize2fs_get_alloc_block;
1654         old_fs->get_alloc_block = resize2fs_get_alloc_block;
1655
1656         retval = ext2fs_read_bb_inode(old_fs, &badblock_list);
1657         if (retval)
1658                 return retval;
1659
1660         new_blk = fs->super->s_first_data_block;
1661         if (!rfs->itable_buf) {
1662                 retval = ext2fs_get_array(fs->blocksize,
1663                                         fs->inode_blocks_per_group,
1664                                         &rfs->itable_buf);
1665                 if (retval)
1666                         return retval;
1667         }
1668         retval = ext2fs_create_extent_table(&rfs->bmap, 0);
1669         if (retval)
1670                 return retval;
1671
1672         /*
1673          * The first step is to figure out where all of the blocks
1674          * will go.
1675          */
1676         to_move = moved = 0;
1677         init_block_alloc(rfs);
1678         for (blk = B2C(old_fs->super->s_first_data_block);
1679              blk < ext2fs_blocks_count(old_fs->super);
1680              blk += EXT2FS_CLUSTER_RATIO(fs)) {
1681                 if (!ext2fs_test_block_bitmap2(old_fs->block_map, blk))
1682                         continue;
1683                 if (!ext2fs_test_block_bitmap2(rfs->move_blocks, blk))
1684                         continue;
1685                 if (ext2fs_badblocks_list_test(badblock_list, blk)) {
1686                         ext2fs_badblocks_list_del(badblock_list, blk);
1687                         bb_modified++;
1688                         continue;
1689                 }
1690
1691                 new_blk = get_new_block(rfs);
1692                 if (!new_blk) {
1693                         retval = ENOSPC;
1694                         goto errout;
1695                 }
1696                 ext2fs_block_alloc_stats2(fs, new_blk, +1);
1697                 ext2fs_add_extent_entry(rfs->bmap, B2C(blk), B2C(new_blk));
1698                 to_move++;
1699         }
1700
1701         if (to_move == 0) {
1702                 if (rfs->bmap) {
1703                         ext2fs_free_extent_table(rfs->bmap);
1704                         rfs->bmap = 0;
1705                 }
1706                 retval = 0;
1707                 goto errout;
1708         }
1709
1710         /*
1711          * Step two is to actually move the blocks
1712          */
1713         retval =  ext2fs_iterate_extent(rfs->bmap, 0, 0, 0);
1714         if (retval) goto errout;
1715
1716         if (rfs->progress) {
1717                 retval = (rfs->progress)(rfs, E2_RSZ_BLOCK_RELOC_PASS,
1718                                          0, to_move);
1719                 if (retval)
1720                         goto errout;
1721         }
1722         while (1) {
1723                 retval = ext2fs_iterate_extent(rfs->bmap, &old_blk, &new_blk, &size);
1724                 if (retval) goto errout;
1725                 if (!size)
1726                         break;
1727                 old_blk = C2B(old_blk);
1728                 new_blk = C2B(new_blk);
1729                 size = C2B(size);
1730 #ifdef RESIZE2FS_DEBUG
1731                 if (rfs->flags & RESIZE_DEBUG_BMOVE)
1732                         printf("Moving %llu blocks %llu->%llu\n",
1733                                size, old_blk, new_blk);
1734 #endif
1735                 do {
1736                         c = size;
1737                         if (c > fs->inode_blocks_per_group)
1738                                 c = fs->inode_blocks_per_group;
1739                         retval = io_channel_read_blk64(fs->io, old_blk, c,
1740                                                        rfs->itable_buf);
1741                         if (retval) goto errout;
1742                         retval = io_channel_write_blk64(fs->io, new_blk, c,
1743                                                         rfs->itable_buf);
1744                         if (retval) goto errout;
1745                         size -= c;
1746                         new_blk += c;
1747                         old_blk += c;
1748                         moved += c;
1749                         if (rfs->progress) {
1750                                 io_channel_flush(fs->io);
1751                                 retval = (rfs->progress)(rfs,
1752                                                 E2_RSZ_BLOCK_RELOC_PASS,
1753                                                 moved, to_move);
1754                                 if (retval)
1755                                         goto errout;
1756                         }
1757                 } while (size > 0);
1758                 io_channel_flush(fs->io);
1759         }
1760
1761 errout:
1762         if (badblock_list) {
1763                 if (!retval && bb_modified)
1764                         retval = ext2fs_update_bb_inode(old_fs,
1765                                                         badblock_list);
1766                 ext2fs_badblocks_list_free(badblock_list);
1767         }
1768         return retval;
1769 }
1770
1771
1772 /* --------------------------------------------------------------------
1773  *
1774  * Resize processing, phase 3
1775  *
1776  * --------------------------------------------------------------------
1777  */
1778
1779
1780 /*
1781  * The extent translation table is stored in clusters so we need to
1782  * take special care when mapping a source block number to its
1783  * destination block number.
1784  */
1785 static __u64 extent_translate(ext2_filsys fs, ext2_extent extent, __u64 old_loc)
1786 {
1787         __u64 new_block = C2B(ext2fs_extent_translate(extent, B2C(old_loc)));
1788
1789         if (new_block != 0)
1790                 new_block += old_loc & (EXT2FS_CLUSTER_RATIO(fs) - 1);
1791         return new_block;
1792 }
1793
1794 struct process_block_struct {
1795         ext2_resize_t           rfs;
1796         ext2_ino_t              ino;
1797         ext2_ino_t              old_ino;
1798         struct ext2_inode *     inode;
1799         errcode_t               error;
1800         int                     is_dir;
1801         int                     changed;
1802         int                     has_extents;
1803 };
1804
1805 static int process_block(ext2_filsys fs, blk64_t        *block_nr,
1806                          e2_blkcnt_t blockcnt,
1807                          blk64_t ref_block EXT2FS_ATTR((unused)),
1808                          int ref_offset EXT2FS_ATTR((unused)), void *priv_data)
1809 {
1810         struct process_block_struct *pb;
1811         errcode_t       retval;
1812         blk64_t         block, new_block;
1813         int             ret = 0;
1814
1815         pb = (struct process_block_struct *) priv_data;
1816         block = *block_nr;
1817         if (pb->rfs->bmap) {
1818                 new_block = extent_translate(fs, pb->rfs->bmap, block);
1819                 if (new_block) {
1820                         *block_nr = new_block;
1821                         ret |= BLOCK_CHANGED;
1822                         pb->changed = 1;
1823 #ifdef RESIZE2FS_DEBUG
1824                         if (pb->rfs->flags & RESIZE_DEBUG_BMOVE)
1825                                 printf("ino=%u, blockcnt=%lld, %llu->%llu\n",
1826                                        pb->old_ino, blockcnt, block,
1827                                        new_block);
1828 #endif
1829                         block = new_block;
1830                 }
1831         }
1832
1833         if (pb->is_dir) {
1834                 retval = ext2fs_add_dir_block2(fs->dblist, pb->ino,
1835                                                block, (int) blockcnt);
1836                 if (retval) {
1837                         pb->error = retval;
1838                         ret |= BLOCK_ABORT;
1839                 }
1840         }
1841         return ret;
1842 }
1843
1844 /*
1845  * Progress callback
1846  */
1847 static errcode_t progress_callback(ext2_filsys fs,
1848                                    ext2_inode_scan scan EXT2FS_ATTR((unused)),
1849                                    dgrp_t group, void * priv_data)
1850 {
1851         ext2_resize_t rfs = (ext2_resize_t) priv_data;
1852         errcode_t               retval;
1853
1854         /*
1855          * This check is to protect against old ext2 libraries.  It
1856          * shouldn't be needed against new libraries.
1857          */
1858         if ((group+1) == 0)
1859                 return 0;
1860
1861         if (rfs->progress) {
1862                 io_channel_flush(fs->io);
1863                 retval = (rfs->progress)(rfs, E2_RSZ_INODE_SCAN_PASS,
1864                                          group+1, fs->group_desc_count);
1865                 if (retval)
1866                         return retval;
1867         }
1868
1869         return 0;
1870 }
1871
1872 static errcode_t migrate_ea_block(ext2_resize_t rfs, ext2_ino_t ino,
1873                                   struct ext2_inode *inode, int *changed)
1874 {
1875         char *buf = NULL;
1876         blk64_t new_block;
1877         errcode_t err = 0;
1878
1879         /* No EA block or no remapping?  Quit early. */
1880         if (ext2fs_file_acl_block(rfs->old_fs, inode) == 0 && !rfs->bmap)
1881                 return 0;
1882         new_block = extent_translate(rfs->old_fs, rfs->bmap,
1883                 ext2fs_file_acl_block(rfs->old_fs, inode));
1884         if (new_block == 0)
1885                 return 0;
1886
1887         /* Set the new ACL block */
1888         ext2fs_file_acl_block_set(rfs->old_fs, inode, new_block);
1889
1890         /* Update checksum */
1891         if (ext2fs_has_feature_metadata_csum(rfs->new_fs->super)) {
1892                 err = ext2fs_get_mem(rfs->old_fs->blocksize, &buf);
1893                 if (err)
1894                         return err;
1895                 rfs->old_fs->flags |= EXT2_FLAG_IGNORE_CSUM_ERRORS;
1896                 err = ext2fs_read_ext_attr3(rfs->old_fs, new_block, buf, ino);
1897                 rfs->old_fs->flags &= ~EXT2_FLAG_IGNORE_CSUM_ERRORS;
1898                 if (err)
1899                         goto out;
1900                 err = ext2fs_write_ext_attr3(rfs->old_fs, new_block, buf, ino);
1901                 if (err)
1902                         goto out;
1903         }
1904         *changed = 1;
1905
1906 out:
1907         ext2fs_free_mem(&buf);
1908         return err;
1909 }
1910
1911 /* Rewrite extents */
1912 static errcode_t rewrite_extents(ext2_filsys fs, ext2_ino_t ino)
1913 {
1914         ext2_extent_handle_t    handle;
1915         struct ext2fs_extent    extent;
1916         errcode_t               errcode;
1917         struct ext2_extent_info info;
1918
1919         errcode = ext2fs_extent_open(fs, ino, &handle);
1920         if (errcode)
1921                 return errcode;
1922
1923         errcode = ext2fs_extent_get(handle, EXT2_EXTENT_ROOT, &extent);
1924         if (errcode)
1925                 goto out;
1926
1927         do {
1928                 errcode = ext2fs_extent_get_info(handle, &info);
1929                 if (errcode)
1930                         break;
1931
1932                 /*
1933                  * If this is the first extent in an extent block that we
1934                  * haven't visited, rewrite the extent to force the ETB
1935                  * checksum to be rewritten.
1936                  */
1937                 if (info.curr_entry == 1 && info.curr_level != 0 &&
1938                     !(extent.e_flags & EXT2_EXTENT_FLAGS_SECOND_VISIT)) {
1939                         errcode = ext2fs_extent_replace(handle, 0, &extent);
1940                         if (errcode)
1941                                 break;
1942                 }
1943
1944                 /* Skip to the end of a block of leaf nodes */
1945                 if (extent.e_flags & EXT2_EXTENT_FLAGS_LEAF) {
1946                         errcode = ext2fs_extent_get(handle,
1947                                                     EXT2_EXTENT_LAST_SIB,
1948                                                     &extent);
1949                         if (errcode)
1950                                 break;
1951                 }
1952
1953                 errcode = ext2fs_extent_get(handle, EXT2_EXTENT_NEXT, &extent);
1954         } while (errcode == 0);
1955
1956 out:
1957         /* Ok if we run off the end */
1958         if (errcode == EXT2_ET_EXTENT_NO_NEXT)
1959                 errcode = 0;
1960         ext2fs_extent_free(handle);
1961         return errcode;
1962 }
1963
1964 static void quiet_com_err_proc(const char *whoami EXT2FS_ATTR((unused)),
1965                                errcode_t code EXT2FS_ATTR((unused)),
1966                                const char *fmt EXT2FS_ATTR((unused)),
1967                                va_list args EXT2FS_ATTR((unused)))
1968 {
1969 }
1970
1971 static errcode_t inode_scan_and_fix(ext2_resize_t rfs)
1972 {
1973         struct process_block_struct     pb;
1974         ext2_ino_t              ino, new_inode;
1975         struct ext2_inode       *inode = NULL;
1976         ext2_inode_scan         scan = NULL;
1977         errcode_t               retval;
1978         char                    *block_buf = 0;
1979         ext2_ino_t              start_to_move;
1980         int                     inode_size;
1981
1982         if ((rfs->old_fs->group_desc_count <=
1983              rfs->new_fs->group_desc_count) &&
1984             !rfs->bmap)
1985                 return 0;
1986
1987         set_com_err_hook(quiet_com_err_proc);
1988
1989         retval = ext2fs_open_inode_scan(rfs->old_fs, 0, &scan);
1990         if (retval) goto errout;
1991
1992         retval = ext2fs_init_dblist(rfs->old_fs, 0);
1993         if (retval) goto errout;
1994         retval = ext2fs_get_array(rfs->old_fs->blocksize, 3, &block_buf);
1995         if (retval) goto errout;
1996
1997         start_to_move = (rfs->new_fs->group_desc_count *
1998                          rfs->new_fs->super->s_inodes_per_group);
1999
2000         if (rfs->progress) {
2001                 retval = (rfs->progress)(rfs, E2_RSZ_INODE_SCAN_PASS,
2002                                          0, rfs->old_fs->group_desc_count);
2003                 if (retval)
2004                         goto errout;
2005         }
2006         ext2fs_set_inode_callback(scan, progress_callback, (void *) rfs);
2007         pb.rfs = rfs;
2008         pb.inode = inode;
2009         pb.error = 0;
2010         new_inode = EXT2_FIRST_INODE(rfs->new_fs->super);
2011         inode_size = EXT2_INODE_SIZE(rfs->new_fs->super);
2012         inode = malloc(inode_size);
2013         if (!inode) {
2014                 retval = ENOMEM;
2015                 goto errout;
2016         }
2017         /*
2018          * First, copy all of the inodes that need to be moved
2019          * elsewhere in the inode table
2020          */
2021         while (1) {
2022                 retval = ext2fs_get_next_inode_full(scan, &ino, inode, inode_size);
2023                 if (retval) goto errout;
2024                 if (!ino)
2025                         break;
2026
2027                 if (inode->i_links_count == 0 && ino != EXT2_RESIZE_INO)
2028                         continue; /* inode not in use */
2029
2030                 pb.is_dir = LINUX_S_ISDIR(inode->i_mode);
2031                 pb.changed = 0;
2032
2033                 /* Remap EA block */
2034                 retval = migrate_ea_block(rfs, ino, inode, &pb.changed);
2035                 if (retval)
2036                         goto errout;
2037
2038                 new_inode = ino;
2039                 if (ino <= start_to_move)
2040                         goto remap_blocks; /* Don't need to move inode. */
2041
2042                 /*
2043                  * Find a new inode.  Now that extents and directory blocks
2044                  * are tied to the inode number through the checksum, we must
2045                  * set up the new inode before we start rewriting blocks.
2046                  */
2047                 retval = ext2fs_new_inode(rfs->new_fs, 0, 0, 0, &new_inode);
2048                 if (retval)
2049                         goto errout;
2050
2051                 ext2fs_inode_alloc_stats2(rfs->new_fs, new_inode, +1,
2052                                           pb.is_dir);
2053                 inode->i_ctime = time(0);
2054                 retval = ext2fs_write_inode_full(rfs->old_fs, new_inode,
2055                                                 inode, inode_size);
2056                 if (retval)
2057                         goto errout;
2058                 pb.changed = 0;
2059
2060 #ifdef RESIZE2FS_DEBUG
2061                 if (rfs->flags & RESIZE_DEBUG_INODEMAP)
2062                         printf("Inode moved %u->%u\n", ino, new_inode);
2063 #endif
2064                 if (!rfs->imap) {
2065                         retval = ext2fs_create_extent_table(&rfs->imap, 0);
2066                         if (retval)
2067                                 goto errout;
2068                 }
2069                 ext2fs_add_extent_entry(rfs->imap, ino, new_inode);
2070
2071 remap_blocks:
2072                 if (pb.changed)
2073                         retval = ext2fs_write_inode_full(rfs->old_fs,
2074                                                          new_inode,
2075                                                          inode, inode_size);
2076                 if (retval)
2077                         goto errout;
2078
2079                 /* Rewrite extent block checksums with new inode number */
2080                 if (ext2fs_has_feature_metadata_csum(rfs->old_fs->super) &&
2081                     (inode->i_flags & EXT4_EXTENTS_FL)) {
2082                         rfs->old_fs->flags |= EXT2_FLAG_IGNORE_CSUM_ERRORS;
2083                         retval = rewrite_extents(rfs->old_fs, new_inode);
2084                         rfs->old_fs->flags &= ~EXT2_FLAG_IGNORE_CSUM_ERRORS;
2085                         if (retval)
2086                                 goto errout;
2087                 }
2088
2089                 /*
2090                  * Update inodes to point to new blocks; schedule directory
2091                  * blocks for inode remapping.  Need to write out dir blocks
2092                  * with new inode numbers if we have metadata_csum enabled.
2093                  */
2094                 if (ext2fs_inode_has_valid_blocks2(rfs->old_fs, inode) &&
2095                     (rfs->bmap || pb.is_dir)) {
2096                         pb.ino = new_inode;
2097                         pb.old_ino = ino;
2098                         pb.has_extents = inode->i_flags & EXT4_EXTENTS_FL;
2099                         rfs->old_fs->flags |= EXT2_FLAG_IGNORE_CSUM_ERRORS;
2100                         retval = ext2fs_block_iterate3(rfs->old_fs,
2101                                                        new_inode, 0, block_buf,
2102                                                        process_block, &pb);
2103                         rfs->old_fs->flags &= ~EXT2_FLAG_IGNORE_CSUM_ERRORS;
2104                         if (retval)
2105                                 goto errout;
2106                         if (pb.error) {
2107                                 retval = pb.error;
2108                                 goto errout;
2109                         }
2110                 } else if ((inode->i_flags & EXT4_INLINE_DATA_FL) &&
2111                            (rfs->bmap || pb.is_dir)) {
2112                         /* inline data dir; update it too */
2113                         retval = ext2fs_add_dir_block2(rfs->old_fs->dblist,
2114                                                        new_inode, 0, 0);
2115                         if (retval)
2116                                 goto errout;
2117                 }
2118         }
2119         io_channel_flush(rfs->old_fs->io);
2120
2121 errout:
2122         reset_com_err_hook();
2123         if (rfs->bmap) {
2124                 ext2fs_free_extent_table(rfs->bmap);
2125                 rfs->bmap = 0;
2126         }
2127         if (scan)
2128                 ext2fs_close_inode_scan(scan);
2129         if (block_buf)
2130                 ext2fs_free_mem(&block_buf);
2131         free(inode);
2132         return retval;
2133 }
2134
2135 /* --------------------------------------------------------------------
2136  *
2137  * Resize processing, phase 4.
2138  *
2139  * --------------------------------------------------------------------
2140  */
2141
2142 struct istruct {
2143         ext2_resize_t rfs;
2144         errcode_t       err;
2145         unsigned int    max_dirs;
2146         unsigned int    num;
2147 };
2148
2149 static int check_and_change_inodes(ext2_ino_t dir,
2150                                    int entry EXT2FS_ATTR((unused)),
2151                                    struct ext2_dir_entry *dirent, int offset,
2152                                    int  blocksize EXT2FS_ATTR((unused)),
2153                                    char *buf EXT2FS_ATTR((unused)),
2154                                    void *priv_data)
2155 {
2156         struct istruct *is = (struct istruct *) priv_data;
2157         struct ext2_inode       inode;
2158         ext2_ino_t              new_inode;
2159         errcode_t               retval;
2160         int                     ret = 0;
2161
2162         if (is->rfs->progress && offset == 0) {
2163                 io_channel_flush(is->rfs->old_fs->io);
2164                 is->err = (is->rfs->progress)(is->rfs,
2165                                               E2_RSZ_INODE_REF_UPD_PASS,
2166                                               ++is->num, is->max_dirs);
2167                 if (is->err)
2168                         return DIRENT_ABORT;
2169         }
2170
2171         /*
2172          * If we have checksums enabled and the inode wasn't present in the
2173          * old fs, then we must rewrite all dir blocks with new checksums.
2174          */
2175         if (ext2fs_has_feature_metadata_csum(is->rfs->old_fs->super) &&
2176             !ext2fs_test_inode_bitmap2(is->rfs->old_fs->inode_map, dir))
2177                 ret |= DIRENT_CHANGED;
2178
2179         if (!dirent->inode)
2180                 return ret;
2181
2182         new_inode = ext2fs_extent_translate(is->rfs->imap, dirent->inode);
2183
2184         if (!new_inode)
2185                 return ret;
2186 #ifdef RESIZE2FS_DEBUG
2187         if (is->rfs->flags & RESIZE_DEBUG_INODEMAP)
2188                 printf("Inode translate (dir=%u, name=%.*s, %u->%u)\n",
2189                        dir, ext2fs_dirent_name_len(dirent), dirent->name,
2190                        dirent->inode, new_inode);
2191 #endif
2192
2193         dirent->inode = new_inode;
2194
2195         /* Update the directory mtime and ctime */
2196         retval = ext2fs_read_inode(is->rfs->old_fs, dir, &inode);
2197         if (retval == 0) {
2198                 inode.i_mtime = inode.i_ctime = time(0);
2199                 is->err = ext2fs_write_inode(is->rfs->old_fs, dir, &inode);
2200                 if (is->err)
2201                         return ret | DIRENT_ABORT;
2202         }
2203
2204         return ret | DIRENT_CHANGED;
2205 }
2206
2207 static errcode_t inode_ref_fix(ext2_resize_t rfs)
2208 {
2209         errcode_t               retval;
2210         struct istruct          is;
2211
2212         if (!rfs->imap)
2213                 return 0;
2214
2215         /*
2216          * Now, we iterate over all of the directories to update the
2217          * inode references
2218          */
2219         is.num = 0;
2220         is.max_dirs = ext2fs_dblist_count2(rfs->old_fs->dblist);
2221         is.rfs = rfs;
2222         is.err = 0;
2223
2224         if (rfs->progress) {
2225                 retval = (rfs->progress)(rfs, E2_RSZ_INODE_REF_UPD_PASS,
2226                                          0, is.max_dirs);
2227                 if (retval)
2228                         goto errout;
2229         }
2230
2231         rfs->old_fs->flags |= EXT2_FLAG_IGNORE_CSUM_ERRORS;
2232         retval = ext2fs_dblist_dir_iterate(rfs->old_fs->dblist,
2233                                            DIRENT_FLAG_INCLUDE_EMPTY, 0,
2234                                            check_and_change_inodes, &is);
2235         rfs->old_fs->flags &= ~EXT2_FLAG_IGNORE_CSUM_ERRORS;
2236         if (retval)
2237                 goto errout;
2238         if (is.err) {
2239                 retval = is.err;
2240                 goto errout;
2241         }
2242
2243         if (rfs->progress && (is.num < is.max_dirs))
2244                 (rfs->progress)(rfs, E2_RSZ_INODE_REF_UPD_PASS,
2245                                 is.max_dirs, is.max_dirs);
2246
2247 errout:
2248         ext2fs_free_extent_table(rfs->imap);
2249         rfs->imap = 0;
2250         return retval;
2251 }
2252
2253
2254 /* --------------------------------------------------------------------
2255  *
2256  * Resize processing, phase 5.
2257  *
2258  * In this phase we actually move the inode table around, and then
2259  * update the summary statistics.  This is scary, since aborting here
2260  * will potentially scramble the filesystem.  (We are moving the
2261  * inode tables around in place, and so the potential for lost data,
2262  * or at the very least scrambling the mapping between filenames and
2263  * inode numbers is very high in case of a power failure here.)
2264  * --------------------------------------------------------------------
2265  */
2266
2267
2268 /*
2269  * A very scary routine --- this one moves the inode table around!!!
2270  *
2271  * After this you have to use the rfs->new_fs file handle to read and
2272  * write inodes.
2273  */
2274 static errcode_t move_itables(ext2_resize_t rfs)
2275 {
2276         int             n, num, size;
2277         long long       diff;
2278         dgrp_t          i, max_groups;
2279         ext2_filsys     fs = rfs->new_fs;
2280         char            *cp;
2281         blk64_t         old_blk, new_blk, blk, cluster_freed;
2282         errcode_t       retval;
2283         int             j, to_move, moved;
2284         ext2fs_block_bitmap     new_bmap = NULL;
2285
2286         max_groups = fs->group_desc_count;
2287         if (max_groups > rfs->old_fs->group_desc_count)
2288                 max_groups = rfs->old_fs->group_desc_count;
2289
2290         size = fs->blocksize * fs->inode_blocks_per_group;
2291         if (!rfs->itable_buf) {
2292                 retval = ext2fs_get_mem(size, &rfs->itable_buf);
2293                 if (retval)
2294                         return retval;
2295         }
2296
2297         if (EXT2FS_CLUSTER_RATIO(fs) > 1) {
2298                 retval = ext2fs_allocate_block_bitmap(fs, _("new meta blocks"),
2299                                                       &new_bmap);
2300                 if (retval)
2301                         return retval;
2302
2303                 retval = mark_table_blocks(fs, new_bmap);
2304                 if (retval)
2305                         goto errout;
2306         }
2307
2308         /*
2309          * Figure out how many inode tables we need to move
2310          */
2311         to_move = moved = 0;
2312         for (i=0; i < max_groups; i++)
2313                 if (ext2fs_inode_table_loc(rfs->old_fs, i) !=
2314                     ext2fs_inode_table_loc(fs, i))
2315                         to_move++;
2316
2317         if (to_move == 0) {
2318                 retval = 0;
2319                 goto errout;
2320         }
2321
2322         if (rfs->progress) {
2323                 retval = rfs->progress(rfs, E2_RSZ_MOVE_ITABLE_PASS,
2324                                        0, to_move);
2325                 if (retval)
2326                         goto errout;
2327         }
2328
2329         rfs->old_fs->flags |= EXT2_FLAG_MASTER_SB_ONLY;
2330
2331         for (i=0; i < max_groups; i++) {
2332                 old_blk = ext2fs_inode_table_loc(rfs->old_fs, i);
2333                 new_blk = ext2fs_inode_table_loc(fs, i);
2334                 diff = new_blk - old_blk;
2335
2336 #ifdef RESIZE2FS_DEBUG
2337                 if (rfs->flags & RESIZE_DEBUG_ITABLEMOVE)
2338                         printf("Itable move group %d block %llu->%llu (diff %lld)\n",
2339                                i, old_blk, new_blk, diff);
2340 #endif
2341
2342                 if (!diff)
2343                         continue;
2344                 if (diff < 0)
2345                         diff = 0;
2346
2347                 retval = io_channel_read_blk64(fs->io, old_blk,
2348                                                fs->inode_blocks_per_group,
2349                                                rfs->itable_buf);
2350                 if (retval)
2351                         goto errout;
2352                 /*
2353                  * The end of the inode table segment often contains
2354                  * all zeros, and we're often only moving the inode
2355                  * table down a block or two.  If so, we can optimize
2356                  * things by not rewriting blocks that we know to be zero
2357                  * already.
2358                  */
2359                 for (cp = rfs->itable_buf+size-1, n=0; n < size; n++, cp--)
2360                         if (*cp)
2361                                 break;
2362                 n = n >> EXT2_BLOCK_SIZE_BITS(fs->super);
2363 #ifdef RESIZE2FS_DEBUG
2364                 if (rfs->flags & RESIZE_DEBUG_ITABLEMOVE)
2365                         printf("%d blocks of zeros...\n", n);
2366 #endif
2367                 num = fs->inode_blocks_per_group;
2368                 if (n > diff)
2369                         num -= n;
2370
2371                 retval = io_channel_write_blk64(fs->io, new_blk,
2372                                                 num, rfs->itable_buf);
2373                 if (retval) {
2374                         io_channel_write_blk64(fs->io, old_blk,
2375                                                num, rfs->itable_buf);
2376                         goto errout;
2377                 }
2378                 if (n > diff) {
2379                         retval = io_channel_write_blk64(fs->io,
2380                               old_blk + fs->inode_blocks_per_group,
2381                               diff, (rfs->itable_buf +
2382                                      (fs->inode_blocks_per_group - diff) *
2383                                      fs->blocksize));
2384                         if (retval)
2385                                 goto errout;
2386                 }
2387
2388                 for (blk = ext2fs_inode_table_loc(rfs->old_fs, i), j=0;
2389                      j < fs->inode_blocks_per_group;) {
2390                         if (new_bmap == NULL ||
2391                             !ext2fs_test_block_bitmap2(new_bmap, blk)) {
2392                                 ext2fs_block_alloc_stats2(fs, blk, -1);
2393                                 cluster_freed = EXT2FS_CLUSTER_RATIO(fs) -
2394                                                 (blk & EXT2FS_CLUSTER_MASK(fs));
2395                                 blk += cluster_freed;
2396                                 j += cluster_freed;
2397                                 continue;
2398                         }
2399                         blk++;
2400                         j++;
2401                 }
2402
2403                 ext2fs_inode_table_loc_set(rfs->old_fs, i, new_blk);
2404                 ext2fs_group_desc_csum_set(rfs->old_fs, i);
2405                 ext2fs_mark_super_dirty(rfs->old_fs);
2406                 ext2fs_flush(rfs->old_fs);
2407
2408                 if (rfs->progress) {
2409                         retval = rfs->progress(rfs, E2_RSZ_MOVE_ITABLE_PASS,
2410                                                ++moved, to_move);
2411                         if (retval)
2412                                 goto errout;
2413                 }
2414         }
2415         mark_table_blocks(fs, fs->block_map);
2416         ext2fs_flush(fs);
2417 #ifdef RESIZE2FS_DEBUG
2418         if (rfs->flags & RESIZE_DEBUG_ITABLEMOVE)
2419                 printf("Inode table move finished.\n");
2420 #endif
2421         retval = 0;
2422
2423 errout:
2424         if (new_bmap)
2425                 ext2fs_free_block_bitmap(new_bmap);
2426         return retval;
2427 }
2428
2429 /*
2430  * This function is used when expanding a file system.  It frees the
2431  * superblock and block group descriptor blocks from the block group
2432  * which is no longer the last block group.
2433  */
2434 static errcode_t clear_sparse_super2_last_group(ext2_resize_t rfs)
2435 {
2436         ext2_filsys     fs = rfs->new_fs;
2437         ext2_filsys     old_fs = rfs->old_fs;
2438         errcode_t       retval;
2439         dgrp_t          old_last_bg = rfs->old_fs->group_desc_count - 1;
2440         dgrp_t          last_bg = fs->group_desc_count - 1;
2441         blk64_t         sb, old_desc;
2442         blk_t           num;
2443
2444         if (!ext2fs_has_feature_sparse_super2(fs->super))
2445                 return 0;
2446
2447         if (last_bg <= old_last_bg)
2448                 return 0;
2449
2450         if (fs->super->s_backup_bgs[0] == old_fs->super->s_backup_bgs[0] &&
2451             fs->super->s_backup_bgs[1] == old_fs->super->s_backup_bgs[1])
2452                 return 0;
2453
2454         if (old_fs->super->s_backup_bgs[0] != old_last_bg &&
2455             old_fs->super->s_backup_bgs[1] != old_last_bg)
2456                 return 0;
2457
2458         if (fs->super->s_backup_bgs[0] == old_last_bg ||
2459             fs->super->s_backup_bgs[1] == old_last_bg)
2460                 return 0;
2461
2462         retval = ext2fs_super_and_bgd_loc2(rfs->old_fs, old_last_bg,
2463                                            &sb, &old_desc, NULL, &num);
2464         if (retval)
2465                 return retval;
2466
2467         if (sb)
2468                 ext2fs_unmark_block_bitmap2(fs->block_map, sb);
2469         if (old_desc)
2470                 ext2fs_unmark_block_bitmap_range2(fs->block_map, old_desc, num);
2471         return 0;
2472 }
2473
2474 /*
2475  * This function is used when shrinking a file system.  We need to
2476  * utilize blocks from what will be the new last block group for the
2477  * backup superblock and block group descriptor blocks.
2478  * Unfortunately, those blocks may be used by other files or fs
2479  * metadata blocks.  We need to mark them as being in use.
2480  */
2481 static errcode_t reserve_sparse_super2_last_group(ext2_resize_t rfs,
2482                                                  ext2fs_block_bitmap meta_bmap)
2483 {
2484         ext2_filsys     fs = rfs->new_fs;
2485         ext2_filsys     old_fs = rfs->old_fs;
2486         errcode_t       retval;
2487         dgrp_t          old_last_bg = rfs->old_fs->group_desc_count - 1;
2488         dgrp_t          last_bg = fs->group_desc_count - 1;
2489         dgrp_t          g;
2490         blk64_t         blk, sb, old_desc;
2491         blk_t           i, num;
2492         int             realloc = 0;
2493
2494         if (!ext2fs_has_feature_sparse_super2(fs->super))
2495                 return 0;
2496
2497         if (last_bg >= old_last_bg)
2498                 return 0;
2499
2500         if (fs->super->s_backup_bgs[0] == old_fs->super->s_backup_bgs[0] &&
2501             fs->super->s_backup_bgs[1] == old_fs->super->s_backup_bgs[1])
2502                 return 0;
2503
2504         if (fs->super->s_backup_bgs[0] != last_bg &&
2505             fs->super->s_backup_bgs[1] != last_bg)
2506                 return 0;
2507
2508         if (old_fs->super->s_backup_bgs[0] == last_bg ||
2509             old_fs->super->s_backup_bgs[1] == last_bg)
2510                 return 0;
2511
2512         retval = ext2fs_super_and_bgd_loc2(rfs->new_fs, last_bg,
2513                                            &sb, &old_desc, NULL, &num);
2514         if (retval)
2515                 return retval;
2516
2517         if (last_bg && !sb) {
2518                 fputs(_("Should never happen!  No sb in last super_sparse bg?\n"),
2519                       stderr);
2520                 exit(1);
2521         }
2522         if (old_desc && old_desc != sb+1) {
2523                 fputs(_("Should never happen!  Unexpected old_desc in "
2524                         "super_sparse bg?\n"),
2525                       stderr);
2526                 exit(1);
2527         }
2528         num = (old_desc) ? num : 1;
2529
2530         /* Reserve the backup blocks */
2531         ext2fs_mark_block_bitmap_range2(fs->block_map, sb, num);
2532
2533         for (g = 0; g < fs->group_desc_count; g++) {
2534                 blk64_t mb;
2535
2536                 mb = ext2fs_block_bitmap_loc(fs, g);
2537                 if ((mb >= sb) && (mb < sb + num)) {
2538                         ext2fs_block_bitmap_loc_set(fs, g, 0);
2539                         realloc = 1;
2540                 }
2541                 mb = ext2fs_inode_bitmap_loc(fs, g);
2542                 if ((mb >= sb) && (mb < sb + num)) {
2543                         ext2fs_inode_bitmap_loc_set(fs, g, 0);
2544                         realloc = 1;
2545                 }
2546                 mb = ext2fs_inode_table_loc(fs, g);
2547                 if ((mb < sb + num) &&
2548                     (sb < mb + fs->inode_blocks_per_group)) {
2549                         ext2fs_inode_table_loc_set(fs, g, 0);
2550                         realloc = 1;
2551                 }
2552                 if (realloc) {
2553                         retval = ext2fs_allocate_group_table(fs, g, 0);
2554                         if (retval)
2555                                 return retval;
2556                 }
2557         }
2558
2559         for (blk = sb, i = 0; i < num; blk++, i++) {
2560                 if (ext2fs_test_block_bitmap2(old_fs->block_map, blk) &&
2561                     !ext2fs_test_block_bitmap2(meta_bmap, blk)) {
2562                         ext2fs_mark_block_bitmap2(rfs->move_blocks, blk);
2563                         rfs->needed_blocks++;
2564                 }
2565                 ext2fs_mark_block_bitmap2(rfs->reserve_blocks, blk);
2566         }
2567         return 0;
2568 }
2569
2570 /*
2571  * Fix the resize inode
2572  */
2573 static errcode_t fix_resize_inode(ext2_filsys fs)
2574 {
2575         struct ext2_inode       inode;
2576         errcode_t               retval;
2577
2578         if (!ext2fs_has_feature_resize_inode(fs->super))
2579                 return 0;
2580
2581         retval = ext2fs_read_inode(fs, EXT2_RESIZE_INO, &inode);
2582         if (retval) goto errout;
2583
2584         ext2fs_iblk_set(fs, &inode, 1);
2585
2586         retval = ext2fs_write_inode(fs, EXT2_RESIZE_INO, &inode);
2587         if (retval) goto errout;
2588
2589         if (!inode.i_block[EXT2_DIND_BLOCK]) {
2590                 /*
2591                  * Avoid zeroing out block #0; that's rude.  This
2592                  * should never happen anyway since the filesystem
2593                  * should be fsck'ed and we assume it is consistent.
2594                  */
2595                 fprintf(stderr, "%s",
2596                         _("Should never happen: resize inode corrupt!\n"));
2597                 exit(1);
2598         }
2599
2600         retval = ext2fs_zero_blocks2(fs, inode.i_block[EXT2_DIND_BLOCK], 1,
2601                                      NULL, NULL);
2602         if (retval)
2603                 goto errout;
2604
2605         retval = ext2fs_create_resize_inode(fs);
2606         if (retval)
2607                 goto errout;
2608
2609 errout:
2610         return retval;
2611 }
2612
2613 /*
2614  * Finally, recalculate the summary information
2615  */
2616 static errcode_t ext2fs_calculate_summary_stats(ext2_filsys fs)
2617 {
2618         blk64_t         blk;
2619         ext2_ino_t      ino;
2620         unsigned int    group = 0;
2621         unsigned int    count = 0;
2622         blk64_t         total_blocks_free = 0;
2623         int             total_inodes_free = 0;
2624         int             group_free = 0;
2625         int             uninit = 0;
2626         blk64_t         super_blk, old_desc_blk, new_desc_blk;
2627         int             old_desc_blocks;
2628
2629         /*
2630          * First calculate the block statistics
2631          */
2632         uninit = ext2fs_bg_flags_test(fs, group, EXT2_BG_BLOCK_UNINIT);
2633         ext2fs_super_and_bgd_loc2(fs, group, &super_blk, &old_desc_blk,
2634                                   &new_desc_blk, 0);
2635         if (ext2fs_has_feature_meta_bg(fs->super))
2636                 old_desc_blocks = fs->super->s_first_meta_bg;
2637         else
2638                 old_desc_blocks = fs->desc_blocks +
2639                         fs->super->s_reserved_gdt_blocks;
2640         for (blk = B2C(fs->super->s_first_data_block);
2641              blk < ext2fs_blocks_count(fs->super);
2642              blk += EXT2FS_CLUSTER_RATIO(fs)) {
2643                 if ((uninit &&
2644                      !(EQ_CLSTR(blk, super_blk) ||
2645                        ((old_desc_blk && old_desc_blocks &&
2646                          GE_CLSTR(blk, old_desc_blk) &&
2647                          LT_CLSTR(blk, old_desc_blk + old_desc_blocks))) ||
2648                        ((new_desc_blk && EQ_CLSTR(blk, new_desc_blk))) ||
2649                        EQ_CLSTR(blk, ext2fs_block_bitmap_loc(fs, group)) ||
2650                        EQ_CLSTR(blk, ext2fs_inode_bitmap_loc(fs, group)) ||
2651                        ((GE_CLSTR(blk, ext2fs_inode_table_loc(fs, group)) &&
2652                          LT_CLSTR(blk, ext2fs_inode_table_loc(fs, group)
2653                                   + fs->inode_blocks_per_group))))) ||
2654                     (!ext2fs_fast_test_block_bitmap2(fs->block_map, blk))) {
2655                         group_free++;
2656                         total_blocks_free++;
2657                 }
2658                 count++;
2659                 if ((count == fs->super->s_clusters_per_group) ||
2660                     EQ_CLSTR(blk, ext2fs_blocks_count(fs->super)-1)) {
2661                         ext2fs_bg_free_blocks_count_set(fs, group, group_free);
2662                         ext2fs_group_desc_csum_set(fs, group);
2663                         group++;
2664                         if (group >= fs->group_desc_count)
2665                                 break;
2666                         count = 0;
2667                         group_free = 0;
2668                         uninit = ext2fs_bg_flags_test(fs, group, EXT2_BG_BLOCK_UNINIT);
2669                         ext2fs_super_and_bgd_loc2(fs, group, &super_blk,
2670                                                   &old_desc_blk,
2671                                                   &new_desc_blk, 0);
2672                         if (ext2fs_has_feature_meta_bg(fs->super))
2673                                 old_desc_blocks = fs->super->s_first_meta_bg;
2674                         else
2675                                 old_desc_blocks = fs->desc_blocks +
2676                                         fs->super->s_reserved_gdt_blocks;
2677                 }
2678         }
2679         total_blocks_free = C2B(total_blocks_free);
2680         ext2fs_free_blocks_count_set(fs->super, total_blocks_free);
2681
2682         /*
2683          * Next, calculate the inode statistics
2684          */
2685         group_free = 0;
2686         count = 0;
2687         group = 0;
2688
2689         /* Protect loop from wrap-around if s_inodes_count maxed */
2690         uninit = ext2fs_bg_flags_test(fs, group, EXT2_BG_INODE_UNINIT);
2691         for (ino = 1; ino <= fs->super->s_inodes_count && ino > 0; ino++) {
2692                 if (uninit ||
2693                     !ext2fs_fast_test_inode_bitmap2(fs->inode_map, ino)) {
2694                         group_free++;
2695                         total_inodes_free++;
2696                 }
2697                 count++;
2698                 if ((count == fs->super->s_inodes_per_group) ||
2699                     (ino == fs->super->s_inodes_count)) {
2700                         ext2fs_bg_free_inodes_count_set(fs, group, group_free);
2701                         ext2fs_group_desc_csum_set(fs, group);
2702                         group++;
2703                         if (group >= fs->group_desc_count)
2704                                 break;
2705                         count = 0;
2706                         group_free = 0;
2707                         uninit = ext2fs_bg_flags_test(fs, group, EXT2_BG_INODE_UNINIT);
2708                 }
2709         }
2710         fs->super->s_free_inodes_count = total_inodes_free;
2711         ext2fs_mark_super_dirty(fs);
2712         return 0;
2713 }
2714
2715 /*
2716  *  Journal may have been relocated; update the backup journal blocks
2717  *  in the superblock.
2718  */
2719 static errcode_t fix_sb_journal_backup(ext2_filsys fs)
2720 {
2721         errcode_t         retval;
2722         struct ext2_inode inode;
2723
2724         if (!ext2fs_has_feature_journal(fs->super))
2725                 return 0;
2726
2727         /* External journal? Nothing to do. */
2728         if (fs->super->s_journal_dev && !fs->super->s_journal_inum)
2729                 return 0;
2730
2731         retval = ext2fs_read_inode(fs, fs->super->s_journal_inum, &inode);
2732         if (retval)
2733                 return retval;
2734         memcpy(fs->super->s_jnl_blocks, inode.i_block, EXT2_N_BLOCKS*4);
2735         fs->super->s_jnl_blocks[15] = inode.i_size_high;
2736         fs->super->s_jnl_blocks[16] = inode.i_size;
2737         fs->super->s_jnl_backup_type = EXT3_JNL_BACKUP_BLOCKS;
2738         ext2fs_mark_super_dirty(fs);
2739         return 0;
2740 }
2741
2742 static int calc_group_overhead(ext2_filsys fs, blk64_t grp,
2743                                int old_desc_blocks)
2744 {
2745         blk64_t super_blk, old_desc_blk, new_desc_blk;
2746         int overhead;
2747
2748         /* inode table blocks plus allocation bitmaps */
2749         overhead = fs->inode_blocks_per_group + 2;
2750
2751         ext2fs_super_and_bgd_loc2(fs, grp, &super_blk,
2752                                   &old_desc_blk, &new_desc_blk, 0);
2753         if ((grp == 0) || super_blk)
2754                 overhead++;
2755         if (old_desc_blk)
2756                 overhead += old_desc_blocks;
2757         else if (new_desc_blk)
2758                 overhead++;
2759         return overhead;
2760 }
2761
2762
2763 /*
2764  * calcluate the minimum number of blocks the given fs can be resized to
2765  */
2766 blk64_t calculate_minimum_resize_size(ext2_filsys fs, int flags)
2767 {
2768         ext2_ino_t inode_count;
2769         dgrp_t groups, flex_groups;
2770         blk64_t blks_needed, data_blocks;
2771         blk64_t grp, data_needed, last_start;
2772         blk64_t overhead = 0;
2773         int old_desc_blocks;
2774         int flexbg_size = 1 << fs->super->s_log_groups_per_flex;
2775
2776         /*
2777          * first figure out how many group descriptors we need to
2778          * handle the number of inodes we have
2779          */
2780         inode_count = fs->super->s_inodes_count -
2781                 fs->super->s_free_inodes_count;
2782         blks_needed = ext2fs_div_ceil(inode_count,
2783                                       fs->super->s_inodes_per_group) *
2784                 (blk64_t) EXT2_BLOCKS_PER_GROUP(fs->super);
2785         groups = ext2fs_div64_ceil(blks_needed,
2786                                    EXT2_BLOCKS_PER_GROUP(fs->super));
2787 #ifdef RESIZE2FS_DEBUG
2788         if (flags & RESIZE_DEBUG_MIN_CALC)
2789                 printf("fs has %d inodes, %d groups required.\n",
2790                        inode_count, groups);
2791 #endif
2792
2793         /*
2794          * number of old-style block group descriptor blocks
2795          */
2796         if (ext2fs_has_feature_meta_bg(fs->super))
2797                 old_desc_blocks = fs->super->s_first_meta_bg;
2798         else
2799                 old_desc_blocks = fs->desc_blocks +
2800                         fs->super->s_reserved_gdt_blocks;
2801
2802         /* calculate how many blocks are needed for data */
2803         data_needed = ext2fs_blocks_count(fs->super) -
2804                 ext2fs_free_blocks_count(fs->super);
2805
2806         for (grp = 0; grp < fs->group_desc_count; grp++)
2807                 data_needed -= calc_group_overhead(fs, grp, old_desc_blocks);
2808 #ifdef RESIZE2FS_DEBUG
2809         if (flags & RESIZE_DEBUG_MIN_CALC)
2810                 printf("fs requires %llu data blocks.\n", data_needed);
2811 #endif
2812
2813         /*
2814          * For ext4 we need to allow for up to a flex_bg worth of
2815          * inode tables of slack space so the resize operation can be
2816          * guaranteed to finish.
2817          */
2818         flex_groups = groups;
2819         if (ext2fs_has_feature_flex_bg(fs->super)) {
2820                 dgrp_t remainder = groups & (flexbg_size - 1);
2821
2822                 flex_groups += flexbg_size - remainder;
2823                 if (flex_groups > fs->group_desc_count)
2824                         flex_groups = fs->group_desc_count;
2825         }
2826
2827         /*
2828          * figure out how many data blocks we have given the number of groups
2829          * we need for our inodes
2830          */
2831         data_blocks = EXT2_GROUPS_TO_BLOCKS(fs->super, groups);
2832         last_start = 0;
2833         for (grp = 0; grp < flex_groups; grp++) {
2834                 overhead = calc_group_overhead(fs, grp, old_desc_blocks);
2835
2836                 /*
2837                  * we want to keep track of how much data we can store in
2838                  * the groups leading up to the last group so we can determine
2839                  * how big the last group needs to be
2840                  */
2841                 if (grp < (groups - 1))
2842                         last_start += EXT2_BLOCKS_PER_GROUP(fs->super) -
2843                                 overhead;
2844
2845                 if (data_blocks > overhead)
2846                         data_blocks -= overhead;
2847                 else
2848                         data_blocks = 0;
2849         }
2850 #ifdef RESIZE2FS_DEBUG
2851         if (flags & RESIZE_DEBUG_MIN_CALC)
2852                 printf("With %d group(s), we have %llu blocks available.\n",
2853                        groups, data_blocks);
2854 #endif
2855
2856         /*
2857          * if we need more group descriptors in order to accomodate our data
2858          * then we need to add them here
2859          */
2860         blks_needed = data_needed;
2861         while (blks_needed > data_blocks) {
2862                 blk64_t remainder = blks_needed - data_blocks;
2863                 dgrp_t extra_grps;
2864
2865                 /* figure out how many more groups we need for the data */
2866                 extra_grps = ext2fs_div64_ceil(remainder,
2867                                                EXT2_BLOCKS_PER_GROUP(fs->super));
2868
2869                 data_blocks += EXT2_GROUPS_TO_BLOCKS(fs->super, extra_grps);
2870
2871                 /* ok we have to account for the last group */
2872                 overhead = calc_group_overhead(fs, groups-1, old_desc_blocks);
2873                 last_start += EXT2_BLOCKS_PER_GROUP(fs->super) - overhead;
2874
2875                 grp = flex_groups;
2876                 groups += extra_grps;
2877                 if (!ext2fs_has_feature_flex_bg(fs->super))
2878                         flex_groups = groups;
2879                 else if (groups > flex_groups) {
2880                         dgrp_t r = groups & (flexbg_size - 1);
2881
2882                         flex_groups = groups + flexbg_size - r;
2883                         if (flex_groups > fs->group_desc_count)
2884                                 flex_groups = fs->group_desc_count;
2885                 }
2886
2887                 for (; grp < flex_groups; grp++) {
2888                         overhead = calc_group_overhead(fs, grp,
2889                                                        old_desc_blocks);
2890
2891                         /*
2892                          * again, we need to see how much data we cram into
2893                          * all of the groups leading up to the last group
2894                          */
2895                         if (grp < groups - 1)
2896                                 last_start += EXT2_BLOCKS_PER_GROUP(fs->super)
2897                                         - overhead;
2898
2899                         data_blocks -= overhead;
2900                 }
2901
2902 #ifdef RESIZE2FS_DEBUG
2903                 if (flags & RESIZE_DEBUG_MIN_CALC)
2904                         printf("Added %d extra group(s), "
2905                                "blks_needed %llu, data_blocks %llu, "
2906                                "last_start %llu\n", extra_grps, blks_needed,
2907                                data_blocks, last_start);
2908 #endif
2909         }
2910
2911         /* now for the fun voodoo */
2912         grp = groups - 1;
2913         if (ext2fs_has_feature_flex_bg(fs->super) &&
2914             (grp & ~(flexbg_size - 1)) == 0)
2915                 grp = grp & ~(flexbg_size - 1);
2916         overhead = 0;
2917         for (; grp < flex_groups; grp++)
2918                 overhead += calc_group_overhead(fs, grp, old_desc_blocks);
2919
2920 #ifdef RESIZE2FS_DEBUG
2921         if (flags & RESIZE_DEBUG_MIN_CALC)
2922                 printf("Last group's overhead is %llu\n", overhead);
2923 #endif
2924
2925         /*
2926          * if this is the case then the last group is going to have data in it
2927          * so we need to adjust the size of the last group accordingly
2928          */
2929         if (last_start < blks_needed) {
2930                 blk64_t remainder = blks_needed - last_start;
2931
2932 #ifdef RESIZE2FS_DEBUG
2933                 if (flags & RESIZE_DEBUG_MIN_CALC)
2934                         printf("Need %llu data blocks in last group\n",
2935                                remainder);
2936 #endif
2937                 /*
2938                  * 50 is a magic number that mkfs/resize uses to see if its
2939                  * even worth making/resizing the fs.  basically you need to
2940                  * have at least 50 blocks in addition to the blocks needed
2941                  * for the metadata in the last group
2942                  */
2943                 if (remainder > 50)
2944                         overhead += remainder;
2945                 else
2946                         overhead += 50;
2947         } else
2948                 overhead += 50;
2949
2950         overhead += fs->super->s_first_data_block;
2951 #ifdef RESIZE2FS_DEBUG
2952         if (flags & RESIZE_DEBUG_MIN_CALC)
2953                 printf("Final size of last group is %lld\n", overhead);
2954 #endif
2955
2956         /* Add extra slack for bigalloc file systems */
2957         if (EXT2FS_CLUSTER_RATIO(fs) > 1)
2958                 overhead += EXT2FS_CLUSTER_RATIO(fs) * 2;
2959
2960         /*
2961          * since our last group doesn't have to be BLOCKS_PER_GROUP
2962          * large, we only do groups-1, and then add the number of
2963          * blocks needed to handle the group descriptor metadata+data
2964          * that we need
2965          */
2966         blks_needed = EXT2_GROUPS_TO_BLOCKS(fs->super, groups - 1);
2967         blks_needed += overhead;
2968
2969         /*
2970          * Make sure blks_needed covers the end of the inode table in
2971          * the last block group.
2972          */
2973         overhead = ext2fs_inode_table_loc(fs, groups-1) +
2974                 fs->inode_blocks_per_group;
2975         if (blks_needed < overhead)
2976                 blks_needed = overhead;
2977
2978 #ifdef RESIZE2FS_DEBUG
2979         if (flags & RESIZE_DEBUG_MIN_CALC)
2980                 printf("Estimated blocks needed: %llu\n", blks_needed);
2981 #endif
2982
2983         /*
2984          * If at this point we've already added up more "needed" than
2985          * the current size, just return current size as minimum.
2986          */
2987         if (blks_needed >= ext2fs_blocks_count(fs->super))
2988                 return ext2fs_blocks_count(fs->super);
2989         /*
2990          * We need to reserve a few extra blocks if extents are
2991          * enabled, in case we need to grow the extent tree.  The more
2992          * we shrink the file system, the more space we need.
2993          *
2994          * The absolute worst case is every single data block is in
2995          * the part of the file system that needs to be evacuated,
2996          * with each data block needs to be in its own extent, and
2997          * with each inode needing at least one extent block.
2998          */
2999         if (ext2fs_has_feature_extents(fs->super)) {
3000                 blk64_t safe_margin = (ext2fs_blocks_count(fs->super) -
3001                                        blks_needed)/500;
3002                 unsigned int exts_per_blk = (fs->blocksize /
3003                                              sizeof(struct ext3_extent)) - 1;
3004                 blk64_t worst_case = ((data_needed + exts_per_blk - 1) /
3005                                       exts_per_blk);
3006
3007                 if (worst_case < inode_count)
3008                         worst_case = inode_count;
3009
3010                 if (safe_margin > worst_case)
3011                         safe_margin = worst_case;
3012
3013 #ifdef RESIZE2FS_DEBUG
3014                 if (flags & RESIZE_DEBUG_MIN_CALC)
3015                         printf("Extents safety margin: %llu\n", safe_margin);
3016 #endif
3017                 blks_needed += safe_margin;
3018         }
3019
3020         return blks_needed;
3021 }