e2fsck/pass1.c

   1 /*
   2  * pass1.c -- pass #1 of e2fsck: sequential scan of the inode table
   3  *
   4  * Copyright (C) 1993, 1994, 1995, 1996, 1997 Theodore Ts'o.
   5  *
   6  * %Begin-Header%
   7  * This file may be redistributed under the terms of the GNU Public
   8  * License.
   9  * %End-Header%
  10  *
  11  * Pass 1 of e2fsck iterates over all the inodes in the filesystems,
  12  * and applies the following tests to each inode:
  13  *
  14  *      - The mode field of the inode must be legal.
  15  *      - The size and block count fields of the inode are correct.
  16  *      - A data block must not be used by another inode
  17  *
  18  * Pass 1 also gathers the collects the following information:
  19  *
  20  *      - A bitmap of which inodes are in use.          (inode_used_map)
  21  *      - A bitmap of which inodes are directories.     (inode_dir_map)
  22  *      - A bitmap of which inodes are regular files.   (inode_reg_map)
  23  *      - A bitmap of which inodes have bad fields.     (inode_bad_map)
  24  *      - A bitmap of which inodes are in bad blocks.   (inode_bb_map)
  25  *      - A bitmap of which inodes are imagic inodes.   (inode_imagic_map)
  26  *      - A bitmap of which inodes are casefolded.      (inode_casefold_map)
  27  *      - A bitmap of which blocks are in use.          (block_found_map)
  28  *      - A bitmap of which blocks are in use by two inodes     (block_dup_map)
  29  *      - The data blocks of the directory inodes.      (dir_map)
  30  *      - Ref counts for ea_inodes.                     (ea_inode_refs)
  31  *      - The encryption policy ID of each encrypted inode. (encrypted_files)
  32  *
  33  * Pass 1 is designed to stash away enough information so that the
  34  * other passes should not need to read in the inode information
  35  * during the normal course of a filesystem check.  (Although if an
  36  * inconsistency is detected, other passes may need to read in an
  37  * inode to fix it.)
  38  *
  39  * Note that pass 1B will be invoked if there are any duplicate blocks
  40  * found.
  41  */
  42
  43 #define _GNU_SOURCE 1 /* get strnlen() */
  44 #include "config.h"
  45 #include <string.h>
  46 #include <time.h>
  47 #ifdef HAVE_ERRNO_H
  48 #include <errno.h>
  49 #endif
  50
  51 #include "e2fsck.h"
  52 #include <ext2fs/ext2_ext_attr.h>
  53 /* todo remove this finally */
  54 #include <ext2fs/ext2fsP.h>
  55 #include <e2p/e2p.h>
  56
  57 #include "problem.h"
  58
  59 #ifdef NO_INLINE_FUNCS
  60 #define _INLINE_
  61 #else
  62 #define _INLINE_ inline
  63 #endif
  64
  65 #undef DEBUG
  66
  67 struct ea_quota {
  68         blk64_t blocks;
  69         __u64 inodes;
  70 };
  71
  72 static int process_block(ext2_filsys fs, blk64_t        *blocknr,
  73                          e2_blkcnt_t blockcnt, blk64_t ref_blk,
  74                          int ref_offset, void *priv_data);
  75 static int process_bad_block(ext2_filsys fs, blk64_t *block_nr,
  76                              e2_blkcnt_t blockcnt, blk64_t ref_blk,
  77                              int ref_offset, void *priv_data);
  78 static void check_blocks(e2fsck_t ctx, struct problem_context *pctx,
  79                          char *block_buf,
  80                          const struct ea_quota *ea_ibody_quota);
  81 static void mark_table_blocks(e2fsck_t ctx);
  82 static void alloc_bb_map(e2fsck_t ctx);
  83 static void alloc_imagic_map(e2fsck_t ctx);
  84 static void mark_inode_bad(e2fsck_t ctx, ext2_ino_t ino);
  85 static void add_casefolded_dir(e2fsck_t ctx, ext2_ino_t ino);
  86 static void handle_fs_bad_blocks(e2fsck_t ctx);
  87 static void process_inodes(e2fsck_t ctx, char *block_buf);
  88 static EXT2_QSORT_TYPE process_inode_cmp(const void *a, const void *b);
  89 static errcode_t scan_callback(ext2_filsys fs, ext2_inode_scan scan,
  90                                   dgrp_t group, void * priv_data);
  91 static void adjust_extattr_refcount(e2fsck_t ctx, ext2_refcount_t refcount,
  92                                     char *block_buf, int adjust_sign);
  93 /* static char *describe_illegal_block(ext2_filsys fs, blk64_t block); */
  94
  95 struct process_block_struct {
  96         ext2_ino_t      ino;
  97         unsigned        is_dir:1, is_reg:1, clear:1, suppress:1,
  98                                 fragmented:1, compressed:1, bbcheck:1,
  99                                 inode_modified:1;
 100         blk64_t         num_blocks;
 101         blk64_t         max_blocks;
 102         blk64_t         last_block;
 103         e2_blkcnt_t     last_init_lblock;
 104         e2_blkcnt_t     last_db_block;
 105         int             num_illegal_blocks;
 106         blk64_t         previous_block;
 107         struct ext2_inode *inode;
 108         struct problem_context *pctx;
 109         ext2fs_block_bitmap fs_meta_blocks;
 110         e2fsck_t        ctx;
 111         blk64_t         next_lblock;
 112         struct extent_tree_info eti;
 113 };
 114
 115 struct process_inode_block {
 116         ext2_ino_t ino;
 117         struct ea_quota ea_ibody_quota;
 118         struct ext2_inode_large inode;
 119 };
 120
 121 struct scan_callback_struct {
 122         e2fsck_t        ctx;
 123         char            *block_buf;
 124 };
 125
 126 /*
 127  * For the inodes to process list.
 128  */
 129 static struct process_inode_block *inodes_to_process;
 130 static int process_inode_count;
 131
 132 static __u64 ext2_max_sizes[EXT2_MAX_BLOCK_LOG_SIZE -
 133                             EXT2_MIN_BLOCK_LOG_SIZE + 1];
 134
 135 /*
 136  * Check to make sure a device inode is real.  Returns 1 if the device
 137  * checks out, 0 if not.
 138  *
 139  * Note: this routine is now also used to check FIFO's and Sockets,
 140  * since they have the same requirement; the i_block fields should be
 141  * zero.
 142  */
 143 int e2fsck_pass1_check_device_inode(ext2_filsys fs EXT2FS_ATTR((unused)),
 144                                     struct ext2_inode *inode)
 145 {
 146         int     i;
 147
 148         /*
 149          * If the index or extents flag is set, then this is a bogus
 150          * device/fifo/socket
 151          */
 152         if (inode->i_flags & (EXT2_INDEX_FL | EXT4_EXTENTS_FL))
 153                 return 0;
 154
 155         /*
 156          * We should be able to do the test below all the time, but
 157          * because the kernel doesn't forcibly clear the device
 158          * inode's additional i_block fields, there are some rare
 159          * occasions when a legitimate device inode will have non-zero
 160          * additional i_block fields.  So for now, we only complain
 161          * when the immutable flag is set, which should never happen
 162          * for devices.  (And that's when the problem is caused, since
 163          * you can't set or clear immutable flags for devices.)  Once
 164          * the kernel has been fixed we can change this...
 165          */
 166         if (inode->i_flags & (EXT2_IMMUTABLE_FL | EXT2_APPEND_FL)) {
 167                 for (i=4; i < EXT2_N_BLOCKS; i++)
 168                         if (inode->i_block[i])
 169                                 return 0;
 170         }
 171         return 1;
 172 }
 173
 174 /*
 175  * Check to make sure a symlink inode is real.  Returns 1 if the symlink
 176  * checks out, 0 if not.
 177  */
 178 int e2fsck_pass1_check_symlink(ext2_filsys fs, ext2_ino_t ino,
 179                                struct ext2_inode *inode, char *buf)
 180 {
 181         unsigned int buflen;
 182         unsigned int len;
 183
 184         if ((inode->i_size_high || inode->i_size == 0) ||
 185             (inode->i_flags & EXT2_INDEX_FL))
 186                 return 0;
 187
 188         if (inode->i_flags & EXT4_INLINE_DATA_FL) {
 189                 size_t inline_size;
 190
 191                 if (inode->i_flags & EXT4_EXTENTS_FL)
 192                         return 0;
 193                 if (ext2fs_inline_data_size(fs, ino, &inline_size))
 194                         return 0;
 195                 if (inode->i_size != inline_size)
 196                         return 0;
 197
 198                 return 1;
 199         }
 200
 201         if (ext2fs_is_fast_symlink(inode)) {
 202                 if (inode->i_flags & EXT4_EXTENTS_FL)
 203                         return 0;
 204                 buf = (char *)inode->i_block;
 205                 buflen = sizeof(inode->i_block);
 206         } else {
 207                 ext2_extent_handle_t    handle;
 208                 struct ext2_extent_info info;
 209                 struct ext2fs_extent    extent;
 210                 blk64_t blk;
 211                 int i;
 212
 213                 if (inode->i_flags & EXT4_EXTENTS_FL) {
 214                         if (ext2fs_extent_open2(fs, ino, inode, &handle))
 215                                 return 0;
 216                         if (ext2fs_extent_get_info(handle, &info) ||
 217                             (info.num_entries != 1) ||
 218                             (info.max_depth != 0)) {
 219                                 ext2fs_extent_free(handle);
 220                                 return 0;
 221                         }
 222                         if (ext2fs_extent_get(handle, EXT2_EXTENT_ROOT,
 223                                               &extent) ||
 224                             (extent.e_lblk != 0) ||
 225                             (extent.e_len != 1)) {
 226                                 ext2fs_extent_free(handle);
 227                                 return 0;
 228                         }
 229                         blk = extent.e_pblk;
 230                         ext2fs_extent_free(handle);
 231                 } else {
 232                         blk = inode->i_block[0];
 233
 234                         for (i = 1; i < EXT2_N_BLOCKS; i++)
 235                                 if (inode->i_block[i])
 236                                         return 0;
 237                 }
 238
 239                 if (blk < fs->super->s_first_data_block ||
 240                     blk >= ext2fs_blocks_count(fs->super))
 241                         return 0;
 242
 243                 if (io_channel_read_blk64(fs->io, blk, 1, buf))
 244                         return 0;
 245
 246                 buflen = fs->blocksize;
 247         }
 248
 249         if (inode->i_flags & EXT4_ENCRYPT_FL)
 250                 len = ext2fs_le16_to_cpu(*(__u16 *)buf) + 2;
 251         else
 252                 len = strnlen(buf, buflen);
 253
 254         if (len >= buflen)
 255                 return 0;
 256
 257         if (len != inode->i_size)
 258                 return 0;
 259         return 1;
 260 }
 261
 262 /*
 263  * If the extents or inlinedata flags are set on the inode, offer to clear 'em.
 264  */
 265 #define BAD_SPECIAL_FLAGS (EXT4_EXTENTS_FL | EXT4_INLINE_DATA_FL)
 266 static void check_extents_inlinedata(e2fsck_t ctx,
 267                                      struct problem_context *pctx)
 268 {
 269         if (!(pctx->inode->i_flags & BAD_SPECIAL_FLAGS))
 270                 return;
 271
 272         if (!fix_problem(ctx, PR_1_SPECIAL_EXTENTS_IDATA, pctx))
 273                 return;
 274
 275         pctx->inode->i_flags &= ~BAD_SPECIAL_FLAGS;
 276         e2fsck_write_inode(ctx, pctx->ino, pctx->inode, "pass1");
 277 }
 278 #undef BAD_SPECIAL_FLAGS
 279
 280 /*
 281  * If the immutable (or append-only) flag is set on the inode, offer
 282  * to clear it.
 283  */
 284 #define BAD_SPECIAL_FLAGS (EXT2_IMMUTABLE_FL | EXT2_APPEND_FL)
 285 static void check_immutable(e2fsck_t ctx, struct problem_context *pctx)
 286 {
 287         if (!(pctx->inode->i_flags & BAD_SPECIAL_FLAGS))
 288                 return;
 289
 290         if (!fix_problem(ctx, PR_1_SET_IMMUTABLE, pctx))
 291                 return;
 292
 293         pctx->inode->i_flags &= ~BAD_SPECIAL_FLAGS;
 294         e2fsck_write_inode(ctx, pctx->ino, pctx->inode, "pass1");
 295 }
 296
 297 /*
 298  * If device, fifo or socket, check size is zero -- if not offer to
 299  * clear it
 300  */
 301 static void check_size(e2fsck_t ctx, struct problem_context *pctx)
 302 {
 303         struct ext2_inode *inode = pctx->inode;
 304
 305         if (EXT2_I_SIZE(inode) == 0)
 306                 return;
 307
 308         if (!fix_problem(ctx, PR_1_SET_NONZSIZE, pctx))
 309                 return;
 310
 311         ext2fs_inode_size_set(ctx->fs, inode, 0);
 312         e2fsck_write_inode(ctx, pctx->ino, pctx->inode, "pass1");
 313 }
 314
 315 /*
 316  * For a given size, calculate how many blocks would be charged towards quota.
 317  */
 318 static blk64_t size_to_quota_blocks(ext2_filsys fs, size_t size)
 319 {
 320         blk64_t clusters;
 321
 322         clusters = DIV_ROUND_UP(size, fs->blocksize << fs->cluster_ratio_bits);
 323         return EXT2FS_C2B(fs, clusters);
 324 }
 325
 326 /*
 327  * Check validity of EA inode. Return 0 if EA inode is valid, otherwise return
 328  * the problem code.
 329  */
 330 static problem_t check_large_ea_inode(e2fsck_t ctx,
 331                                       struct ext2_ext_attr_entry *entry,
 332                                       struct problem_context *pctx,
 333                                       blk64_t *quota_blocks)
 334 {
 335         struct ext2_inode inode;
 336         __u32 hash, signed_hash;
 337         errcode_t retval;
 338
 339         /* Check if inode is within valid range */
 340         if ((entry->e_value_inum < EXT2_FIRST_INODE(ctx->fs->super)) ||
 341             (entry->e_value_inum > ctx->fs->super->s_inodes_count)) {
 342                 pctx->num = entry->e_value_inum;
 343                 return PR_1_ATTR_VALUE_EA_INODE;
 344         }
 345
 346         e2fsck_read_inode(ctx, entry->e_value_inum, &inode, "pass1");
 347
 348         retval = ext2fs_ext_attr_hash_entry3(ctx->fs, entry, NULL, &hash,
 349                                              &signed_hash);
 350         if (retval) {
 351                 com_err("check_large_ea_inode", retval,
 352                         _("while hashing entry with e_value_inum = %u"),
 353                         entry->e_value_inum);
 354                 fatal_error(ctx, 0);
 355         }
 356
 357         if ((hash == entry->e_hash) || (signed_hash == entry->e_hash)) {
 358                 *quota_blocks = size_to_quota_blocks(ctx->fs,
 359                                                      entry->e_value_size);
 360         } else {
 361                 /* This might be an old Lustre-style ea_inode reference. */
 362                 if (inode.i_mtime == pctx->ino &&
 363                     inode.i_generation == pctx->inode->i_generation) {
 364                         *quota_blocks = 0;
 365                 } else {
 366                         /* If target inode is also missing EA_INODE flag,
 367                          * this is likely to be a bad reference.
 368                          */
 369                         if (!(inode.i_flags & EXT4_EA_INODE_FL)) {
 370                                 pctx->num = entry->e_value_inum;
 371                                 return PR_1_ATTR_VALUE_EA_INODE;
 372                         } else {
 373                                 pctx->num = entry->e_hash;
 374                                 return PR_1_ATTR_HASH;
 375                         }
 376                 }
 377         }
 378
 379         if (!(inode.i_flags & EXT4_EA_INODE_FL)) {
 380                 pctx->num = entry->e_value_inum;
 381                 if (fix_problem(ctx, PR_1_ATTR_SET_EA_INODE_FL, pctx)) {
 382                         inode.i_flags |= EXT4_EA_INODE_FL;
 383                         ext2fs_write_inode(ctx->fs, entry->e_value_inum,
 384                                            &inode);
 385                 } else {
 386                         return PR_1_ATTR_NO_EA_INODE_FL;
 387                 }
 388         }
 389         return 0;
 390 }
 391
 392 static int alloc_ea_inode_refs(e2fsck_t ctx, struct problem_context *pctx)
 393 {
 394         pctx->errcode = ea_refcount_create(0, &ctx->ea_inode_refs);
 395         if (pctx->errcode) {
 396                 pctx->num = 4;
 397                 fix_problem(ctx, PR_1_ALLOCATE_REFCOUNT, pctx);
 398                 ctx->flags |= E2F_FLAG_ABORT;
 399                 return 0;
 400         }
 401         return 1;
 402 }
 403
 404 static void inc_ea_inode_refs(e2fsck_t ctx, struct problem_context *pctx,
 405                               struct ext2_ext_attr_entry *first, void *end)
 406 {
 407         struct ext2_ext_attr_entry *entry = first;
 408         struct ext2_ext_attr_entry *np = EXT2_EXT_ATTR_NEXT(entry);
 409         ea_value_t refs;
 410
 411         while ((void *) entry < end && (void *) np < end &&
 412                !EXT2_EXT_IS_LAST_ENTRY(entry)) {
 413                 if (!entry->e_value_inum)
 414                         goto next;
 415                 if (!ctx->ea_inode_refs && !alloc_ea_inode_refs(ctx, pctx))
 416                         return;
 417                 ea_refcount_fetch(ctx->ea_inode_refs, entry->e_value_inum,
 418                                   &refs);
 419                 if (refs == EA_INODE_NO_REFS)
 420                         refs = 1;
 421                 else
 422                         refs += 1;
 423                 ea_refcount_store(ctx->ea_inode_refs, entry->e_value_inum, refs);
 424         next:
 425                 entry = np;
 426                 np = EXT2_EXT_ATTR_NEXT(entry);
 427         }
 428 }
 429
 430 /*
 431  * Make sure inode is tracked as EA inode. We use special EA_INODE_NO_REFS
 432  * value if we didn't find any xattrs referencing this inode yet.
 433  */
 434 static int track_ea_inode(e2fsck_t ctx, struct problem_context *pctx,
 435                           ext2_ino_t ino)
 436 {
 437         ea_value_t refs;
 438
 439         if (!ctx->ea_inode_refs && !alloc_ea_inode_refs(ctx, pctx))
 440                 return 0;
 441
 442         ea_refcount_fetch(ctx->ea_inode_refs, ino, &refs);
 443         if (refs > 0)
 444                 return 1;
 445
 446         pctx->errcode = ea_refcount_store(ctx->ea_inode_refs, ino,
 447                                           EA_INODE_NO_REFS);
 448         if (pctx->errcode) {
 449                 pctx->num = 5;
 450                 fix_problem(ctx, PR_1_ALLOCATE_REFCOUNT, pctx);
 451                 ctx->flags |= E2F_FLAG_ABORT;
 452                 return 0;
 453         }
 454         return 1;
 455 }
 456
 457 static void check_ea_in_inode(e2fsck_t ctx, struct problem_context *pctx,
 458                               struct ea_quota *ea_ibody_quota)
 459 {
 460         struct ext2_super_block *sb = ctx->fs->super;
 461         struct ext2_inode_large *inode;
 462         struct ext2_ext_attr_entry *entry;
 463         char *start, *header, *end;
 464         unsigned int storage_size, remain;
 465         problem_t problem = 0;
 466         region_t region = 0;
 467
 468         ea_ibody_quota->blocks = 0;
 469         ea_ibody_quota->inodes = 0;
 470
 471         inode = (struct ext2_inode_large *) pctx->inode;
 472         storage_size = EXT2_INODE_SIZE(ctx->fs->super) - EXT2_GOOD_OLD_INODE_SIZE -
 473                 inode->i_extra_isize;
 474         header = ((char *) inode) + EXT2_GOOD_OLD_INODE_SIZE +
 475                  inode->i_extra_isize;
 476         end = header + storage_size;
 477         start = header + sizeof(__u32);
 478         entry = (struct ext2_ext_attr_entry *) start;
 479
 480         /* scan all entry's headers first */
 481
 482         /* take finish entry 0UL into account */
 483         remain = storage_size - sizeof(__u32);
 484
 485         region = region_create(0, storage_size);
 486         if (!region) {
 487                 fix_problem(ctx, PR_1_EA_ALLOC_REGION_ABORT, pctx);
 488                 problem = 0;
 489                 ctx->flags |= E2F_FLAG_ABORT;
 490                 return;
 491         }
 492         if (region_allocate(region, 0, sizeof(__u32))) {
 493                 problem = PR_1_INODE_EA_ALLOC_COLLISION;
 494                 goto fix;
 495         }
 496
 497         while (remain >= sizeof(struct ext2_ext_attr_entry) &&
 498                !EXT2_EXT_IS_LAST_ENTRY(entry)) {
 499                 __u32 hash;
 500
 501                 if (region_allocate(region, (char *)entry - (char *)header,
 502                                     EXT2_EXT_ATTR_LEN(entry->e_name_len))) {
 503                         problem = PR_1_INODE_EA_ALLOC_COLLISION;
 504                         goto fix;
 505                 }
 506
 507                 /* header eats this space */
 508                 remain -= sizeof(struct ext2_ext_attr_entry);
 509
 510                 /* is attribute name valid? */
 511                 if (EXT2_EXT_ATTR_SIZE(entry->e_name_len) > remain) {
 512                         pctx->num = entry->e_name_len;
 513                         problem = PR_1_ATTR_NAME_LEN;
 514                         goto fix;
 515                 }
 516
 517                 /* attribute len eats this space */
 518                 remain -= EXT2_EXT_ATTR_SIZE(entry->e_name_len);
 519
 520                 if (entry->e_value_inum == 0) {
 521                         /* check value size */
 522                         if (entry->e_value_size > remain) {
 523                                 pctx->num = entry->e_value_size;
 524                                 problem = PR_1_ATTR_VALUE_SIZE;
 525                                 goto fix;
 526                         }
 527
 528                         if (entry->e_value_size &&
 529                             region_allocate(region,
 530                                             sizeof(__u32) + entry->e_value_offs,
 531                                             EXT2_EXT_ATTR_SIZE(
 532                                                 entry->e_value_size))) {
 533                                 problem = PR_1_INODE_EA_ALLOC_COLLISION;
 534                                 goto fix;
 535                         }
 536
 537                         hash = ext2fs_ext_attr_hash_entry(entry,
 538                                                 start + entry->e_value_offs);
 539                         if (entry->e_hash != 0 && entry->e_hash != hash)
 540                                 hash = ext2fs_ext_attr_hash_entry_signed(entry,
 541                                                 start + entry->e_value_offs);
 542
 543                         /* e_hash may be 0 in older inode's ea */
 544                         if (entry->e_hash != 0 && entry->e_hash != hash) {
 545                                 pctx->num = entry->e_hash;
 546                                 problem = PR_1_ATTR_HASH;
 547                                 goto fix;
 548                         }
 549                 } else {
 550                         blk64_t quota_blocks;
 551
 552                         if (!ext2fs_has_feature_ea_inode(sb) &&
 553                             fix_problem(ctx, PR_1_EA_INODE_FEATURE, pctx)) {
 554                                 ext2fs_set_feature_ea_inode(sb);
 555                                 ext2fs_mark_super_dirty(ctx->fs);
 556                         }
 557
 558                         problem = check_large_ea_inode(ctx, entry, pctx,
 559                                                        &quota_blocks);
 560                         if (problem != 0)
 561                                 goto fix;
 562
 563                         ea_ibody_quota->blocks += quota_blocks;
 564                         ea_ibody_quota->inodes++;
 565                 }
 566
 567                 /* If EA value is stored in external inode then it does not
 568                  * consume space here */
 569                 if (entry->e_value_inum == 0)
 570                         remain -= entry->e_value_size;
 571
 572                 entry = EXT2_EXT_ATTR_NEXT(entry);
 573         }
 574
 575         if (region_allocate(region, (char *)entry - (char *)header,
 576                             sizeof(__u32))) {
 577                 problem = PR_1_INODE_EA_ALLOC_COLLISION;
 578                 goto fix;
 579         }
 580 fix:
 581         if (region)
 582                 region_free(region);
 583         /*
 584          * it seems like a corruption. it's very unlikely we could repair
 585          * EA(s) in automatic fashion -bzzz
 586          */
 587         if (problem == 0 || !fix_problem(ctx, problem, pctx)) {
 588                 inc_ea_inode_refs(ctx, pctx,
 589                                   (struct ext2_ext_attr_entry *)start, end);
 590                 return;
 591         }
 592
 593         /* simply remove all possible EA(s) */
 594         *((__u32 *)header) = 0UL;
 595         e2fsck_write_inode_full(ctx, pctx->ino, pctx->inode,
 596                                 EXT2_INODE_SIZE(sb), "pass1");
 597         ea_ibody_quota->blocks = 0;
 598         ea_ibody_quota->inodes = 0;
 599 }
 600
 601 static int check_inode_extra_negative_epoch(__u32 xtime, __u32 extra) {
 602         return (xtime & (1U << 31)) != 0 &&
 603                 (extra & EXT4_EPOCH_MASK) == EXT4_EPOCH_MASK;
 604 }
 605
 606 #define CHECK_INODE_EXTRA_NEGATIVE_EPOCH(inode, xtime) \
 607         check_inode_extra_negative_epoch(inode->i_##xtime, \
 608                                          inode->i_##xtime##_extra)
 609
 610 /* When today's date is earlier than 2242, we assume that atimes,
 611  * ctimes, crtimes, and mtimes with years in the range 2310..2378 are
 612  * actually pre-1970 dates mis-encoded.
 613  */
 614 #define EXT4_EXTRA_NEGATIVE_DATE_CUTOFF 2 * (1LL << 32)
 615
 616 static void check_inode_extra_space(e2fsck_t ctx, struct problem_context *pctx,
 617                                     struct ea_quota *ea_ibody_quota)
 618 {
 619         struct ext2_super_block *sb = ctx->fs->super;
 620         struct ext2_inode_large *inode;
 621         __u32 *eamagic;
 622         int min, max;
 623
 624         ea_ibody_quota->blocks = 0;
 625         ea_ibody_quota->inodes = 0;
 626
 627         inode = (struct ext2_inode_large *) pctx->inode;
 628         if (EXT2_INODE_SIZE(sb) == EXT2_GOOD_OLD_INODE_SIZE) {
 629                 /* this isn't large inode. so, nothing to check */
 630                 return;
 631         }
 632
 633 #if 0
 634         printf("inode #%u, i_extra_size %d\n", pctx->ino,
 635                         inode->i_extra_isize);
 636 #endif
 637         /* i_extra_isize must cover i_extra_isize + i_checksum_hi at least */
 638         min = sizeof(inode->i_extra_isize) + sizeof(inode->i_checksum_hi);
 639         max = EXT2_INODE_SIZE(sb) - EXT2_GOOD_OLD_INODE_SIZE;
 640         /*
 641          * For now we will allow i_extra_isize to be 0, but really
 642          * implementations should never allow i_extra_isize to be 0
 643          */
 644         if (inode->i_extra_isize &&
 645             (inode->i_extra_isize < min || inode->i_extra_isize > max ||
 646              inode->i_extra_isize & 3)) {
 647                 if (!fix_problem(ctx, PR_1_EXTRA_ISIZE, pctx))
 648                         return;
 649                 if (inode->i_extra_isize < min || inode->i_extra_isize > max)
 650                         inode->i_extra_isize = sb->s_want_extra_isize;
 651                 else
 652                         inode->i_extra_isize = (inode->i_extra_isize + 3) & ~3;
 653                 e2fsck_write_inode_full(ctx, pctx->ino, pctx->inode,
 654                                         EXT2_INODE_SIZE(sb), "pass1");
 655         }
 656
 657         /* check if there is no place for an EA header */
 658         if (inode->i_extra_isize >= max - sizeof(__u32))
 659                 return;
 660
 661         eamagic = (__u32 *) (((char *) inode) + EXT2_GOOD_OLD_INODE_SIZE +
 662                         inode->i_extra_isize);
 663         if (*eamagic == EXT2_EXT_ATTR_MAGIC) {
 664                 /* it seems inode has an extended attribute(s) in body */
 665                 check_ea_in_inode(ctx, pctx, ea_ibody_quota);
 666         }
 667
 668         /*
 669          * If the inode's extended atime (ctime, crtime, mtime) is stored in
 670          * the old, invalid format, repair it.
 671          */
 672         if (((sizeof(time_t) <= 4) ||
 673              (((sizeof(time_t) > 4) &&
 674                ctx->now < EXT4_EXTRA_NEGATIVE_DATE_CUTOFF))) &&
 675             (CHECK_INODE_EXTRA_NEGATIVE_EPOCH(inode, atime) ||
 676              CHECK_INODE_EXTRA_NEGATIVE_EPOCH(inode, ctime) ||
 677              CHECK_INODE_EXTRA_NEGATIVE_EPOCH(inode, crtime) ||
 678              CHECK_INODE_EXTRA_NEGATIVE_EPOCH(inode, mtime))) {
 679
 680                 if (!fix_problem(ctx, PR_1_EA_TIME_OUT_OF_RANGE, pctx))
 681                         return;
 682
 683                 if (CHECK_INODE_EXTRA_NEGATIVE_EPOCH(inode, atime))
 684                         inode->i_atime_extra &= ~EXT4_EPOCH_MASK;
 685                 if (CHECK_INODE_EXTRA_NEGATIVE_EPOCH(inode, ctime))
 686                         inode->i_ctime_extra &= ~EXT4_EPOCH_MASK;
 687                 if (CHECK_INODE_EXTRA_NEGATIVE_EPOCH(inode, crtime))
 688                         inode->i_crtime_extra &= ~EXT4_EPOCH_MASK;
 689                 if (CHECK_INODE_EXTRA_NEGATIVE_EPOCH(inode, mtime))
 690                         inode->i_mtime_extra &= ~EXT4_EPOCH_MASK;
 691                 e2fsck_write_inode_full(ctx, pctx->ino, pctx->inode,
 692                                         EXT2_INODE_SIZE(sb), "pass1");
 693         }
 694
 695 }
 696
 697 /*
 698  * Check to see if the inode might really be a directory, despite i_mode
 699  *
 700  * This is a lot of complexity for something for which I'm not really
 701  * convinced happens frequently in the wild.  If for any reason this
 702  * causes any problems, take this code out.
 703  * [tytso:20070331.0827EDT]
 704  */
 705 static void check_is_really_dir(e2fsck_t ctx, struct problem_context *pctx,
 706                                 char *buf)
 707 {
 708         struct ext2_inode *inode = pctx->inode;
 709         struct ext2_dir_entry   *dirent;
 710         errcode_t               retval;
 711         blk64_t                 blk;
 712         unsigned int            i, rec_len, not_device = 0;
 713         int                     extent_fs;
 714         int                     inlinedata_fs;
 715
 716         /*
 717          * If the mode looks OK, we believe it.  If the first block in
 718          * the i_block array is 0, this cannot be a directory. If the
 719          * inode is extent-mapped, it is still the case that the latter
 720          * cannot be 0 - the magic number in the extent header would make
 721          * it nonzero.
 722          */
 723         if (LINUX_S_ISDIR(inode->i_mode) || LINUX_S_ISREG(inode->i_mode) ||
 724             LINUX_S_ISLNK(inode->i_mode) || inode->i_block[0] == 0)
 725                 return;
 726
 727         /*
 728          * Check the block numbers in the i_block array for validity:
 729          * zero blocks are skipped (but the first one cannot be zero -
 730          * see above), other blocks are checked against the first and
 731          * max data blocks (from the the superblock) and against the
 732          * block bitmap. Any invalid block found means this cannot be
 733          * a directory.
 734          *
 735          * If there are non-zero blocks past the fourth entry, then
 736          * this cannot be a device file: we remember that for the next
 737          * check.
 738          *
 739          * For extent mapped files, we don't do any sanity checking:
 740          * just try to get the phys block of logical block 0 and run
 741          * with it.
 742          *
 743          * For inline data files, we just try to get the size of inline
 744          * data.  If it's true, we will treat it as a directory.
 745          */
 746
 747         extent_fs = ext2fs_has_feature_extents(ctx->fs->super);
 748         inlinedata_fs = ext2fs_has_feature_inline_data(ctx->fs->super);
 749         if (inlinedata_fs && (inode->i_flags & EXT4_INLINE_DATA_FL)) {
 750                 size_t size;
 751                 __u32 dotdot;
 752                 unsigned int rec_len2;
 753                 struct ext2_dir_entry de;
 754
 755                 if (ext2fs_inline_data_size(ctx->fs, pctx->ino, &size))
 756                         return;
 757                 /*
 758                  * If the size isn't a multiple of 4, it's probably not a
 759                  * directory??
 760                  */
 761                 if (size & 3)
 762                         return;
 763                 /*
 764                  * If the first 10 bytes don't look like a directory entry,
 765                  * it's probably not a directory.
 766                  */
 767                 memcpy(&dotdot, inode->i_block, sizeof(dotdot));
 768                 memcpy(&de, ((char *)inode->i_block) + EXT4_INLINE_DATA_DOTDOT_SIZE,
 769                        EXT2_DIR_REC_LEN(0));
 770                 dotdot = ext2fs_le32_to_cpu(dotdot);
 771                 de.inode = ext2fs_le32_to_cpu(de.inode);
 772                 de.rec_len = ext2fs_le16_to_cpu(de.rec_len);
 773                 ext2fs_get_rec_len(ctx->fs, &de, &rec_len2);
 774                 if (dotdot >= ctx->fs->super->s_inodes_count ||
 775                     (dotdot < EXT2_FIRST_INO(ctx->fs->super) &&
 776                      dotdot != EXT2_ROOT_INO) ||
 777                     de.inode >= ctx->fs->super->s_inodes_count ||
 778                     (de.inode < EXT2_FIRST_INO(ctx->fs->super) &&
 779                      de.inode != 0) ||
 780                     rec_len2 > EXT4_MIN_INLINE_DATA_SIZE -
 781                               EXT4_INLINE_DATA_DOTDOT_SIZE)
 782                         return;
 783                 /* device files never have a "system.data" entry */
 784                 goto isdir;
 785         } else if (extent_fs && (inode->i_flags & EXT4_EXTENTS_FL)) {
 786                 /* extent mapped */
 787                 if  (ext2fs_bmap2(ctx->fs, pctx->ino, inode, 0, 0, 0, 0,
 788                                  &blk))
 789                         return;
 790                 /* device files are never extent mapped */
 791                 not_device++;
 792         } else {
 793                 for (i=0; i < EXT2_N_BLOCKS; i++) {
 794                         blk = inode->i_block[i];
 795                         if (!blk)
 796                                 continue;
 797                         if (i >= 4)
 798                                 not_device++;
 799
 800                         if (blk < ctx->fs->super->s_first_data_block ||
 801                             blk >= ext2fs_blocks_count(ctx->fs->super) ||
 802                             ext2fs_fast_test_block_bitmap2(ctx->block_found_map,
 803                                                            blk))
 804                                 return; /* Invalid block, can't be dir */
 805                 }
 806                 blk = inode->i_block[0];
 807         }
 808
 809         /*
 810          * If the mode says this is a device file and the i_links_count field
 811          * is sane and we have not ruled it out as a device file previously,
 812          * we declare it a device file, not a directory.
 813          */
 814         if ((LINUX_S_ISCHR(inode->i_mode) || LINUX_S_ISBLK(inode->i_mode)) &&
 815             (inode->i_links_count == 1) && !not_device)
 816                 return;
 817
 818         /* read the first block */
 819         ehandler_operation(_("reading directory block"));
 820         retval = ext2fs_read_dir_block4(ctx->fs, blk, buf, 0, pctx->ino);
 821         ehandler_operation(0);
 822         if (retval)
 823                 return;
 824
 825         dirent = (struct ext2_dir_entry *) buf;
 826         retval = ext2fs_get_rec_len(ctx->fs, dirent, &rec_len);
 827         if (retval)
 828                 return;
 829         if ((ext2fs_dirent_name_len(dirent) != 1) ||
 830             (dirent->name[0] != '.') ||
 831             (dirent->inode != pctx->ino) ||
 832             (rec_len < 12) ||
 833             (rec_len % 4) ||
 834             (rec_len >= ctx->fs->blocksize - 12))
 835                 return;
 836
 837         dirent = (struct ext2_dir_entry *) (buf + rec_len);
 838         retval = ext2fs_get_rec_len(ctx->fs, dirent, &rec_len);
 839         if (retval)
 840                 return;
 841         if ((ext2fs_dirent_name_len(dirent) != 2) ||
 842             (dirent->name[0] != '.') ||
 843             (dirent->name[1] != '.') ||
 844             (rec_len < 12) ||
 845             (rec_len % 4))
 846                 return;
 847
 848 isdir:
 849         if (fix_problem(ctx, PR_1_TREAT_AS_DIRECTORY, pctx)) {
 850                 inode->i_mode = (inode->i_mode & 07777) | LINUX_S_IFDIR;
 851                 e2fsck_write_inode_full(ctx, pctx->ino, inode,
 852                                         EXT2_INODE_SIZE(ctx->fs->super),
 853                                         "check_is_really_dir");
 854         }
 855 }
 856
 857 extern errcode_t e2fsck_setup_icount(e2fsck_t ctx, const char *icount_name,
 858                                      int flags, ext2_icount_t hint,
 859                                      ext2_icount_t *ret)
 860 {
 861         unsigned int            threshold;
 862         unsigned int            save_type;
 863         ext2_ino_t              num_dirs;
 864         errcode_t               retval;
 865         char                    *tdb_dir;
 866         int                     enable;
 867
 868         *ret = 0;
 869
 870         profile_get_string(ctx->profile, "scratch_files", "directory", 0, 0,
 871                            &tdb_dir);
 872         profile_get_uint(ctx->profile, "scratch_files",
 873                          "numdirs_threshold", 0, 0, &threshold);
 874         profile_get_boolean(ctx->profile, "scratch_files",
 875                             "icount", 0, 1, &enable);
 876
 877         retval = ext2fs_get_num_dirs(ctx->fs, &num_dirs);
 878         if (retval)
 879                 num_dirs = 1024;        /* Guess */
 880
 881         if (enable && tdb_dir && !access(tdb_dir, W_OK) &&
 882             (!threshold || num_dirs > threshold)) {
 883                 retval = ext2fs_create_icount_tdb(ctx->fs, tdb_dir,
 884                                                   flags, ret);
 885                 if (retval == 0)
 886                         return 0;
 887         }
 888         e2fsck_set_bitmap_type(ctx->fs, EXT2FS_BMAP64_RBTREE, icount_name,
 889                                &save_type);
 890         if (ctx->options & E2F_OPT_ICOUNT_FULLMAP)
 891                 flags |= EXT2_ICOUNT_OPT_FULLMAP;
 892         retval = ext2fs_create_icount2(ctx->fs, flags, 0, hint, ret);
 893         ctx->fs->default_bitmap_type = save_type;
 894         return retval;
 895 }
 896
 897 static errcode_t recheck_bad_inode_checksum(ext2_filsys fs, ext2_ino_t ino,
 898                                             e2fsck_t ctx,
 899                                             struct problem_context *pctx)
 900 {
 901         errcode_t retval;
 902         struct ext2_inode_large inode;
 903
 904         /*
 905          * Reread inode.  If we don't see checksum error, then this inode
 906          * has been fixed elsewhere.
 907          */
 908         ctx->stashed_ino = 0;
 909         retval = ext2fs_read_inode_full(fs, ino, (struct ext2_inode *)&inode,
 910                                         sizeof(inode));
 911         if (retval && retval != EXT2_ET_INODE_CSUM_INVALID)
 912                 return retval;
 913         if (!retval)
 914                 return 0;
 915
 916         /*
 917          * Checksum still doesn't match.  That implies that the inode passes
 918          * all the sanity checks, so maybe the checksum is simply corrupt.
 919          * See if the user will go for fixing that.
 920          */
 921         if (!fix_problem(ctx, PR_1_INODE_ONLY_CSUM_INVALID, pctx))
 922                 return 0;
 923
 924         retval = ext2fs_write_inode_full(fs, ino, (struct ext2_inode *)&inode,
 925                                          sizeof(inode));
 926         return retval;
 927 }
 928
 929 static void reserve_block_for_root_repair(e2fsck_t ctx)
 930 {
 931         blk64_t         blk = 0;
 932         errcode_t       err;
 933         ext2_filsys     fs = ctx->fs;
 934
 935         ctx->root_repair_block = 0;
 936         if (ext2fs_test_inode_bitmap2(ctx->inode_used_map, EXT2_ROOT_INO))
 937                 return;
 938
 939         err = ext2fs_new_block2(fs, 0, ctx->block_found_map, &blk);
 940         if (err)
 941                 return;
 942         ext2fs_mark_block_bitmap2(ctx->block_found_map, blk);
 943         ctx->root_repair_block = blk;
 944 }
 945
 946 static void reserve_block_for_lnf_repair(e2fsck_t ctx)
 947 {
 948         blk64_t         blk = 0;
 949         errcode_t       err;
 950         ext2_filsys     fs = ctx->fs;
 951         static const char name[] = "lost+found";
 952         ext2_ino_t      ino;
 953
 954         ctx->lnf_repair_block = 0;
 955         if (!ext2fs_lookup(fs, EXT2_ROOT_INO, name, sizeof(name)-1, 0, &ino))
 956                 return;
 957
 958         err = ext2fs_new_block2(fs, 0, ctx->block_found_map, &blk);
 959         if (err)
 960                 return;
 961         ext2fs_mark_block_bitmap2(ctx->block_found_map, blk);
 962         ctx->lnf_repair_block = blk;
 963 }
 964
 965 static errcode_t get_inline_data_ea_size(ext2_filsys fs, ext2_ino_t ino,
 966                                          struct ext2_inode *inode,
 967                                          size_t *sz)
 968 {
 969         void *p;
 970         struct ext2_xattr_handle *handle;
 971         errcode_t retval;
 972
 973         retval = ext2fs_xattrs_open(fs, ino, &handle);
 974         if (retval)
 975                 return retval;
 976
 977         retval = ext2fs_xattrs_read_inode(handle,
 978                                           (struct ext2_inode_large *)inode);
 979         if (retval)
 980                 goto err;
 981
 982         retval = ext2fs_xattr_get(handle, "system.data", &p, sz);
 983         if (retval)
 984                 goto err;
 985         ext2fs_free_mem(&p);
 986 err:
 987         (void) ext2fs_xattrs_close(&handle);
 988         return retval;
 989 }
 990
 991 static void finish_processing_inode(e2fsck_t ctx, ext2_ino_t ino,
 992                                     struct problem_context *pctx,
 993                                     int failed_csum)
 994 {
 995         if (!failed_csum)
 996                 return;
 997
 998         /*
 999          * If the inode failed the checksum and the user didn't
1000          * clear the inode, test the checksum again -- if it still
1001          * fails, ask the user if the checksum should be corrected.
1002          */
1003         pctx->errcode = recheck_bad_inode_checksum(ctx->fs, ino, ctx, pctx);
1004         if (pctx->errcode)
1005                 ctx->flags |= E2F_FLAG_ABORT;
1006 }
1007 #define FINISH_INODE_LOOP(ctx, ino, pctx, failed_csum) \
1008         do { \
1009                 finish_processing_inode((ctx), (ino), (pctx), (failed_csum)); \
1010                 if ((ctx)->flags & E2F_FLAG_ABORT) \
1011                         return; \
1012         } while (0)
1013
1014 static int could_be_block_map(ext2_filsys fs, struct ext2_inode *inode)
1015 {
1016         __u32 x;
1017         int i;
1018
1019         for (i = 0; i < EXT2_N_BLOCKS; i++) {
1020                 x = inode->i_block[i];
1021 #ifdef WORDS_BIGENDIAN
1022                 x = ext2fs_swab32(x);
1023 #endif
1024                 if (x >= ext2fs_blocks_count(fs->super))
1025                         return 0;
1026         }
1027
1028         return 1;
1029 }
1030
1031 /*
1032  * Figure out what to do with an inode that has both extents and inline data
1033  * inode flags set.  Returns -1 if we decide to erase the inode, 0 otherwise.
1034  */
1035 static int fix_inline_data_extents_file(e2fsck_t ctx,
1036                                         ext2_ino_t ino,
1037                                         struct ext2_inode *inode,
1038                                         int inode_size,
1039                                         struct problem_context *pctx)
1040 {
1041         size_t max_inline_ea_size;
1042         ext2_filsys fs = ctx->fs;
1043         int dirty = 0;
1044
1045         /* Both feature flags not set?  Just run the regular checks */
1046         if (!ext2fs_has_feature_extents(fs->super) &&
1047             !ext2fs_has_feature_inline_data(fs->super))
1048                 return 0;
1049
1050         /* Clear both flags if it's a special file */
1051         if (LINUX_S_ISCHR(inode->i_mode) ||
1052             LINUX_S_ISBLK(inode->i_mode) ||
1053             LINUX_S_ISFIFO(inode->i_mode) ||
1054             LINUX_S_ISSOCK(inode->i_mode)) {
1055                 check_extents_inlinedata(ctx, pctx);
1056                 return 0;
1057         }
1058
1059         /* If it looks like an extent tree, try to clear inlinedata */
1060         if (ext2fs_extent_header_verify(inode->i_block,
1061                                  sizeof(inode->i_block)) == 0 &&
1062             fix_problem(ctx, PR_1_CLEAR_INLINE_DATA_FOR_EXTENT, pctx)) {
1063                 inode->i_flags &= ~EXT4_INLINE_DATA_FL;
1064                 dirty = 1;
1065                 goto out;
1066         }
1067
1068         /* If it looks short enough to be inline data, try to clear extents */
1069         if (inode_size > EXT2_GOOD_OLD_INODE_SIZE)
1070                 max_inline_ea_size = inode_size -
1071                                      (EXT2_GOOD_OLD_INODE_SIZE +
1072                                       ((struct ext2_inode_large *)inode)->i_extra_isize);
1073         else
1074                 max_inline_ea_size = 0;
1075         if (EXT2_I_SIZE(inode) <
1076             EXT4_MIN_INLINE_DATA_SIZE + max_inline_ea_size &&
1077             fix_problem(ctx, PR_1_CLEAR_EXTENT_FOR_INLINE_DATA, pctx)) {
1078                 inode->i_flags &= ~EXT4_EXTENTS_FL;
1079                 dirty = 1;
1080                 goto out;
1081         }
1082
1083         /*
1084          * Too big for inline data, but no evidence of extent tree -
1085          * maybe it's a block map file?  If the mappings all look valid?
1086          */
1087         if (could_be_block_map(fs, inode) &&
1088             fix_problem(ctx, PR_1_CLEAR_EXTENT_INLINE_DATA_FLAGS, pctx)) {
1089 #ifdef WORDS_BIGENDIAN
1090                 int i;
1091
1092                 for (i = 0; i < EXT2_N_BLOCKS; i++)
1093                         inode->i_block[i] = ext2fs_swab32(inode->i_block[i]);
1094 #endif
1095
1096                 inode->i_flags &= ~(EXT4_EXTENTS_FL | EXT4_INLINE_DATA_FL);
1097                 dirty = 1;
1098                 goto out;
1099         }
1100
1101         /* Oh well, just clear the busted inode. */
1102         if (fix_problem(ctx, PR_1_CLEAR_EXTENT_INLINE_DATA_INODE, pctx)) {
1103                 e2fsck_clear_inode(ctx, ino, inode, 0, "pass1");
1104                 return -1;
1105         }
1106
1107 out:
1108         if (dirty)
1109                 e2fsck_write_inode(ctx, ino, inode, "pass1");
1110
1111         return 0;
1112 }
1113
1114 static void pass1_readahead(e2fsck_t ctx, dgrp_t *group, ext2_ino_t *next_ino)
1115 {
1116         ext2_ino_t inodes_in_group = 0, inodes_per_block, inodes_per_buffer;
1117         dgrp_t start = *group, grp;
1118         blk64_t blocks_to_read = 0;
1119         errcode_t err = EXT2_ET_INVALID_ARGUMENT;
1120
1121         if (ctx->readahead_kb == 0)
1122                 goto out;
1123
1124         /* Keep iterating groups until we have enough to readahead */
1125         inodes_per_block = EXT2_INODES_PER_BLOCK(ctx->fs->super);
1126         for (grp = start; grp < ctx->fs->group_desc_count; grp++) {
1127                 if (ext2fs_bg_flags_test(ctx->fs, grp, EXT2_BG_INODE_UNINIT))
1128                         continue;
1129                 inodes_in_group = ctx->fs->super->s_inodes_per_group -
1130                                         ext2fs_bg_itable_unused(ctx->fs, grp);
1131                 blocks_to_read += (inodes_in_group + inodes_per_block - 1) /
1132                                         inodes_per_block;
1133                 if (blocks_to_read * ctx->fs->blocksize >
1134                     ctx->readahead_kb * 1024)
1135                         break;
1136         }
1137
1138         err = e2fsck_readahead(ctx->fs, E2FSCK_READA_ITABLE, start,
1139                                grp - start + 1);
1140         if (err == EAGAIN) {
1141                 ctx->readahead_kb /= 2;
1142                 err = 0;
1143         }
1144
1145 out:
1146         if (err) {
1147                 /* Error; disable itable readahead */
1148                 *group = ctx->fs->group_desc_count;
1149                 *next_ino = ctx->fs->super->s_inodes_count;
1150         } else {
1151                 /*
1152                  * Don't do more readahead until we've reached the first inode
1153                  * of the last inode scan buffer block for the last group.
1154                  */
1155                 *group = grp + 1;
1156                 inodes_per_buffer = (ctx->inode_buffer_blocks ?
1157                                      ctx->inode_buffer_blocks :
1158                                      EXT2_INODE_SCAN_DEFAULT_BUFFER_BLOCKS) *
1159                                     ctx->fs->blocksize /
1160                                     EXT2_INODE_SIZE(ctx->fs->super);
1161                 inodes_in_group--;
1162                 *next_ino = inodes_in_group -
1163                             (inodes_in_group % inodes_per_buffer) + 1 +
1164                             (grp * ctx->fs->super->s_inodes_per_group);
1165         }
1166 }
1167
1168 /*
1169  * Check if the passed ino is one of the used superblock quota inodes.
1170  *
1171  * Before the quota inodes were journaled, older superblock quota inodes
1172  * were just regular files in the filesystem and not reserved inodes.  This
1173  * checks if the passed ino is one of the s_*_quota_inum superblock fields,
1174  * which may not always be the same as the EXT4_*_QUOTA_INO fields.
1175  */
1176 static int quota_inum_is_super(struct ext2_super_block *sb, ext2_ino_t ino)
1177 {
1178         enum quota_type qtype;
1179
1180         for (qtype = 0; qtype < MAXQUOTAS; qtype++)
1181                 if (*quota_sb_inump(sb, qtype) == ino)
1182                         return 1;
1183
1184         return 0;
1185 }
1186
1187 /*
1188  * Check if the passed ino is one of the reserved quota inodes.
1189  * This checks if the inode number is one of the reserved EXT4_*_QUOTA_INO
1190  * inodes.  These inodes may or may not be in use by the quota feature.
1191  */
1192 static int quota_inum_is_reserved(ext2_filsys fs, ext2_ino_t ino)
1193 {
1194         enum quota_type qtype;
1195
1196         for (qtype = 0; qtype < MAXQUOTAS; qtype++)
1197                 if (quota_type2inum(qtype, fs->super) == ino)
1198                         return 1;
1199
1200         return 0;
1201 }
1202
1203 static int e2fsck_should_abort(e2fsck_t ctx)
1204 {
1205         e2fsck_t global_ctx;
1206
1207         if (ctx->flags & E2F_FLAG_SIGNAL_MASK)
1208                 return 1;
1209
1210         if (ctx->global_ctx) {
1211                 global_ctx = ctx->global_ctx;
1212                 if (global_ctx->flags & E2F_FLAG_SIGNAL_MASK)
1213                         return 1;
1214         }
1215         return 0;
1216 }
1217
1218 void e2fsck_pass1_thread(e2fsck_t ctx)
1219 {
1220         int     i;
1221         __u64   max_sizes;
1222         ext2_filsys fs = ctx->fs;
1223         ext2_ino_t      ino = 0;
1224         struct ext2_inode *inode = NULL;
1225         ext2_inode_scan scan = NULL;
1226         char            *block_buf = NULL;
1227 #ifdef RESOURCE_TRACK
1228         struct resource_track   rtrack;
1229 #endif
1230         unsigned char   frag, fsize;
1231         struct          problem_context pctx;
1232         struct          scan_callback_struct scan_struct;
1233         struct ext2_super_block *sb = ctx->fs->super;
1234         const char      *old_op;
1235         const char      *eop_next_inode = _("getting next inode from scan");
1236         int             imagic_fs, extent_fs, inlinedata_fs, casefold_fs;
1237         int             low_dtime_check = 1;
1238         unsigned int    inode_size = EXT2_INODE_SIZE(fs->super);
1239         unsigned int    bufsize;
1240         int             failed_csum = 0;
1241         ext2_ino_t      ino_threshold = 0;
1242         dgrp_t          ra_group = 0;
1243         struct ea_quota ea_ibody_quota;
1244         time_t          tm;
1245
1246         init_resource_track(&rtrack, ctx->fs->io);
1247         clear_problem_context(&pctx);
1248
1249         /* If we can do readahead, figure out how many groups to pull in. */
1250         if (!e2fsck_can_readahead(ctx->fs))
1251                 ctx->readahead_kb = 0;
1252         else if (ctx->readahead_kb == ~0ULL)
1253                 ctx->readahead_kb = e2fsck_guess_readahead(ctx->fs);
1254         pass1_readahead(ctx, &ra_group, &ino_threshold);
1255
1256         if (!(ctx->options & E2F_OPT_PREEN))
1257                 fix_problem(ctx, PR_1_PASS_HEADER, &pctx);
1258
1259         if (ext2fs_has_feature_dir_index(fs->super) &&
1260             !(ctx->options & E2F_OPT_NO)) {
1261                 if (ext2fs_u32_list_create(&ctx->dirs_to_hash, 50))
1262                         ctx->dirs_to_hash = 0;
1263         }
1264
1265 #ifdef MTRACE
1266         mtrace_print("Pass 1");
1267 #endif
1268
1269 #define EXT2_BPP(bits) (1ULL << ((bits) - 2))
1270
1271         for (i = EXT2_MIN_BLOCK_LOG_SIZE; i <= EXT2_MAX_BLOCK_LOG_SIZE; i++) {
1272                 max_sizes = EXT2_NDIR_BLOCKS + EXT2_BPP(i);
1273                 max_sizes = max_sizes + EXT2_BPP(i) * EXT2_BPP(i);
1274                 max_sizes = max_sizes + EXT2_BPP(i) * EXT2_BPP(i) * EXT2_BPP(i);
1275                 max_sizes = (max_sizes * (1UL << i));
1276                 ext2_max_sizes[i - EXT2_MIN_BLOCK_LOG_SIZE] = max_sizes;
1277         }
1278 #undef EXT2_BPP
1279
1280         imagic_fs = ext2fs_has_feature_imagic_inodes(sb);
1281         extent_fs = ext2fs_has_feature_extents(sb);
1282         inlinedata_fs = ext2fs_has_feature_inline_data(sb);
1283         casefold_fs = ext2fs_has_feature_casefold(sb);
1284
1285         /*
1286          * Allocate bitmaps structures
1287          */
1288         pctx.errcode = e2fsck_allocate_inode_bitmap(fs, _("in-use inode map"),
1289                                                     EXT2FS_BMAP64_RBTREE,
1290                                                     "inode_used_map",
1291                                                     &ctx->inode_used_map);
1292         if (pctx.errcode) {
1293                 pctx.num = 1;
1294                 fix_problem(ctx, PR_1_ALLOCATE_IBITMAP_ERROR, &pctx);
1295                 ctx->flags |= E2F_FLAG_ABORT;
1296                 return;
1297         }
1298         pctx.errcode = e2fsck_allocate_inode_bitmap(fs,
1299                         _("directory inode map"),
1300                         EXT2FS_BMAP64_AUTODIR,
1301                         "inode_dir_map", &ctx->inode_dir_map);
1302         if (pctx.errcode) {
1303                 pctx.num = 2;
1304                 fix_problem(ctx, PR_1_ALLOCATE_IBITMAP_ERROR, &pctx);
1305                 ctx->flags |= E2F_FLAG_ABORT;
1306                 return;
1307         }
1308         pctx.errcode = e2fsck_allocate_inode_bitmap(fs,
1309                         _("regular file inode map"), EXT2FS_BMAP64_RBTREE,
1310                         "inode_reg_map", &ctx->inode_reg_map);
1311         if (pctx.errcode) {
1312                 pctx.num = 6;
1313                 fix_problem(ctx, PR_1_ALLOCATE_IBITMAP_ERROR, &pctx);
1314                 ctx->flags |= E2F_FLAG_ABORT;
1315                 return;
1316         }
1317         pctx.errcode = e2fsck_allocate_subcluster_bitmap(fs,
1318                         _("in-use block map"), EXT2FS_BMAP64_RBTREE,
1319                         "block_found_map", &ctx->block_found_map);
1320         if (pctx.errcode) {
1321                 pctx.num = 1;
1322                 fix_problem(ctx, PR_1_ALLOCATE_BBITMAP_ERROR, &pctx);
1323                 ctx->flags |= E2F_FLAG_ABORT;
1324                 return;
1325         }
1326         pctx.errcode = e2fsck_allocate_block_bitmap(fs,
1327                         _("metadata block map"), EXT2FS_BMAP64_RBTREE,
1328                         "block_metadata_map", &ctx->block_metadata_map);
1329         if (pctx.errcode) {
1330                 pctx.num = 1;
1331                 fix_problem(ctx, PR_1_ALLOCATE_BBITMAP_ERROR, &pctx);
1332                 ctx->flags |= E2F_FLAG_ABORT;
1333                 return;
1334         }
1335         if (casefold_fs) {
1336                 pctx.errcode =
1337                         e2fsck_allocate_inode_bitmap(fs,
1338                                                      _("inode casefold map"),
1339                                                      EXT2FS_BMAP64_RBTREE,
1340                                                      "inode_casefold_map",
1341                                                      &ctx->inode_casefold_map);
1342                 if (pctx.errcode) {
1343                         pctx.num = 1;
1344                         fix_problem(ctx, PR_1_ALLOCATE_IBITMAP_ERROR, &pctx);
1345                         ctx->flags |= E2F_FLAG_ABORT;
1346                         return;
1347                 }
1348         }
1349         pctx.errcode = e2fsck_setup_icount(ctx, "inode_link_info", 0, NULL,
1350                                            &ctx->inode_link_info);
1351         if (pctx.errcode) {
1352                 fix_problem(ctx, PR_1_ALLOCATE_ICOUNT, &pctx);
1353                 ctx->flags |= E2F_FLAG_ABORT;
1354                 return;
1355         }
1356         bufsize = inode_size;
1357         if (bufsize < sizeof(struct ext2_inode_large))
1358                 bufsize = sizeof(struct ext2_inode_large);
1359         inode = (struct ext2_inode *)
1360                 e2fsck_allocate_memory(ctx, bufsize, "scratch inode");
1361
1362         inodes_to_process = (struct process_inode_block *)
1363                 e2fsck_allocate_memory(ctx,
1364                                        (ctx->process_inode_size *
1365                                         sizeof(struct process_inode_block)),
1366                                        "array of inodes to process");
1367         process_inode_count = 0;
1368
1369         pctx.errcode = ext2fs_init_dblist(fs, 0);
1370         if (pctx.errcode) {
1371                 fix_problem(ctx, PR_1_ALLOCATE_DBCOUNT, &pctx);
1372                 ctx->flags |= E2F_FLAG_ABORT;
1373                 goto endit;
1374         }
1375
1376         /*
1377          * If the last orphan field is set, clear it, since the pass1
1378          * processing will automatically find and clear the orphans.
1379          * In the future, we may want to try using the last_orphan
1380          * linked list ourselves, but for now, we clear it so that the
1381          * ext3 mount code won't get confused.
1382          */
1383         if (!(ctx->options & E2F_OPT_READONLY)) {
1384                 if (fs->super->s_last_orphan) {
1385                         fs->super->s_last_orphan = 0;
1386                         ext2fs_mark_super_dirty(fs);
1387                 }
1388         }
1389
1390         mark_table_blocks(ctx);
1391         pctx.errcode = ext2fs_convert_subcluster_bitmap(fs,
1392                                                 &ctx->block_found_map);
1393         if (pctx.errcode) {
1394                 fix_problem(ctx, PR_1_CONVERT_SUBCLUSTER, &pctx);
1395                 ctx->flags |= E2F_FLAG_ABORT;
1396                 goto endit;
1397         }
1398         block_buf = (char *) e2fsck_allocate_memory(ctx, fs->blocksize * 3,
1399                                                     "block iterate buffer");
1400         if (EXT2_INODE_SIZE(fs->super) == EXT2_GOOD_OLD_INODE_SIZE)
1401                 e2fsck_use_inode_shortcuts(ctx, 1);
1402         e2fsck_intercept_block_allocations(ctx);
1403         old_op = ehandler_operation(_("opening inode scan"));
1404         pctx.errcode = ext2fs_open_inode_scan(fs, ctx->inode_buffer_blocks,
1405                                               &scan);
1406         ehandler_operation(old_op);
1407         if (pctx.errcode) {
1408                 fix_problem(ctx, PR_1_ISCAN_ERROR, &pctx);
1409                 ctx->flags |= E2F_FLAG_ABORT;
1410                 goto endit;
1411         }
1412         ext2fs_inode_scan_flags(scan, EXT2_SF_SKIP_MISSING_ITABLE |
1413                                       EXT2_SF_WARN_GARBAGE_INODES, 0);
1414         ctx->stashed_inode = inode;
1415         scan_struct.ctx = ctx;
1416         scan_struct.block_buf = block_buf;
1417         ext2fs_set_inode_callback(scan, scan_callback, &scan_struct);
1418         if (ctx->progress && ((ctx->progress)(ctx, 1, 0,
1419                                               ctx->fs->group_desc_count)))
1420                 goto endit;
1421
1422         if (((tm = ext2fs_get_tstamp(fs->super, s_wtime)) &&
1423              tm < fs->super->s_inodes_count) ||
1424             ((tm = ext2fs_get_tstamp(fs->super, s_mtime)) &&
1425              tm < fs->super->s_inodes_count) ||
1426             ((tm = ext2fs_get_tstamp(fs->super, s_mkfs_time)) &&
1427              tm < fs->super->s_inodes_count))
1428                 low_dtime_check = 0;
1429
1430         if (ext2fs_has_feature_mmp(fs->super) &&
1431             fs->super->s_mmp_block > fs->super->s_first_data_block &&
1432             fs->super->s_mmp_block < ext2fs_blocks_count(fs->super))
1433                 ext2fs_mark_block_bitmap2(ctx->block_found_map,
1434                                           fs->super->s_mmp_block);
1435
1436         /* Set up ctx->lost_and_found if possible */
1437         (void) e2fsck_get_lost_and_found(ctx, 0);
1438
1439         while (1) {
1440                 if (ino % (fs->super->s_inodes_per_group * 4) == 1) {
1441                         if (e2fsck_mmp_update(fs))
1442                                 fatal_error(ctx, 0);
1443                 }
1444                 old_op = ehandler_operation(eop_next_inode);
1445                 pctx.errcode = ext2fs_get_next_inode_full(scan, &ino,
1446                                                           inode, inode_size);
1447                 if (ino > ino_threshold)
1448                         pass1_readahead(ctx, &ra_group, &ino_threshold);
1449                 ehandler_operation(old_op);
1450                 if (e2fsck_should_abort(ctx))
1451                         goto endit;
1452                 if (pctx.errcode == EXT2_ET_BAD_BLOCK_IN_INODE_TABLE) {
1453                         /*
1454                          * If badblocks says badblocks is bad, offer to clear
1455                          * the list, update the in-core bb list, and restart
1456                          * the inode scan.
1457                          */
1458                         if (ino == EXT2_BAD_INO &&
1459                             fix_problem(ctx, PR_1_BADBLOCKS_IN_BADBLOCKS,
1460                                         &pctx)) {
1461                                 errcode_t err;
1462
1463                                 e2fsck_clear_inode(ctx, ino, inode, 0, "pass1");
1464                                 ext2fs_badblocks_list_free(ctx->fs->badblocks);
1465                                 ctx->fs->badblocks = NULL;
1466                                 err = ext2fs_read_bb_inode(ctx->fs,
1467                                                         &ctx->fs->badblocks);
1468                                 if (err) {
1469                                         fix_problem(ctx, PR_1_ISCAN_ERROR,
1470                                                     &pctx);
1471                                         ctx->flags |= E2F_FLAG_ABORT;
1472                                 } else
1473                                         ctx->flags |= E2F_FLAG_RESTART;
1474                                 goto endit;
1475                         }
1476                         if (!ctx->inode_bb_map)
1477                                 alloc_bb_map(ctx);
1478                         ext2fs_mark_inode_bitmap2(ctx->inode_bb_map, ino);
1479                         ext2fs_mark_inode_bitmap2(ctx->inode_used_map, ino);
1480                         continue;
1481                 }
1482                 if (pctx.errcode &&
1483                     pctx.errcode != EXT2_ET_INODE_CSUM_INVALID &&
1484                     pctx.errcode != EXT2_ET_INODE_IS_GARBAGE) {
1485                         fix_problem(ctx, PR_1_ISCAN_ERROR, &pctx);
1486                         ctx->flags |= E2F_FLAG_ABORT;
1487                         goto endit;
1488                 }
1489                 if (!ino)
1490                         break;
1491                 pctx.ino = ino;
1492                 pctx.inode = inode;
1493                 ctx->stashed_ino = ino;
1494
1495                 /* Clear trashed inode? */
1496                 if (pctx.errcode == EXT2_ET_INODE_IS_GARBAGE &&
1497                     inode->i_links_count > 0 &&
1498                     fix_problem(ctx, PR_1_INODE_IS_GARBAGE, &pctx)) {
1499                         pctx.errcode = 0;
1500                         e2fsck_clear_inode(ctx, ino, inode, 0, "pass1");
1501                 }
1502                 failed_csum = pctx.errcode != 0;
1503
1504                 /*
1505                  * Check for inodes who might have been part of the
1506                  * orphaned list linked list.  They should have gotten
1507                  * dealt with by now, unless the list had somehow been
1508                  * corrupted.
1509                  *
1510                  * FIXME: In the future, inodes which are still in use
1511                  * (and which are therefore) pending truncation should
1512                  * be handled specially.  Right now we just clear the
1513                  * dtime field, and the normal e2fsck handling of
1514                  * inodes where i_size and the inode blocks are
1515                  * inconsistent is to fix i_size, instead of releasing
1516                  * the extra blocks.  This won't catch the inodes that
1517                  * was at the end of the orphan list, but it's better
1518                  * than nothing.  The right answer is that there
1519                  * shouldn't be any bugs in the orphan list handling.  :-)
1520                  */
1521                 if (inode->i_dtime && low_dtime_check &&
1522                     inode->i_dtime < ctx->fs->super->s_inodes_count) {
1523                         if (fix_problem(ctx, PR_1_LOW_DTIME, &pctx)) {
1524                                 inode->i_dtime = inode->i_links_count ?
1525                                         0 : ctx->now;
1526                                 e2fsck_write_inode(ctx, ino, inode,
1527                                                    "pass1");
1528                                 failed_csum = 0;
1529                         }
1530                 }
1531
1532                 if (inode->i_links_count) {
1533                         pctx.errcode = ext2fs_icount_store(ctx->inode_link_info,
1534                                            ino, inode->i_links_count);
1535                         if (pctx.errcode) {
1536                                 pctx.num = inode->i_links_count;
1537                                 fix_problem(ctx, PR_1_ICOUNT_STORE, &pctx);
1538                                 ctx->flags |= E2F_FLAG_ABORT;
1539                                 goto endit;
1540                         }
1541                 } else if ((ino >= EXT2_FIRST_INODE(fs->super)) &&
1542                            !quota_inum_is_reserved(fs, ino)) {
1543                         if (!inode->i_dtime && inode->i_mode) {
1544                                 if (fix_problem(ctx,
1545                                             PR_1_ZERO_DTIME, &pctx)) {
1546                                         ext2fs_set_dtime(fs, inode);
1547                                         e2fsck_write_inode(ctx, ino, inode,
1548                                                            "pass1");
1549                                         failed_csum = 0;
1550                                 }
1551                         }
1552                         FINISH_INODE_LOOP(ctx, ino, &pctx, failed_csum);
1553                         continue;
1554                 }
1555
1556                 if ((inode->i_flags & EXT4_CASEFOLD_FL) &&
1557                     ((!LINUX_S_ISDIR(inode->i_mode) &&
1558                       fix_problem(ctx, PR_1_CASEFOLD_NONDIR, &pctx)) ||
1559                      (!casefold_fs &&
1560                       fix_problem(ctx, PR_1_CASEFOLD_FEATURE, &pctx)))) {
1561                         inode->i_flags &= ~EXT4_CASEFOLD_FL;
1562                         e2fsck_write_inode(ctx, ino, inode, "pass1");
1563                 }
1564
1565                 if (inode->i_flags & EXT4_EA_INODE_FL) {
1566                         if (!LINUX_S_ISREG(inode->i_mode) &&
1567                             fix_problem(ctx, PR_1_EA_INODE_NONREG, &pctx)) {
1568                                 inode->i_flags &= ~EXT4_EA_INODE_FL;
1569                                 e2fsck_write_inode(ctx, ino, inode, "pass1");
1570                         }
1571                         if (inode->i_flags & EXT4_EA_INODE_FL)
1572                                 if (!track_ea_inode(ctx, &pctx, ino))
1573                                         continue;
1574                 }
1575
1576                 /* Conflicting inlinedata/extents inode flags? */
1577                 if ((inode->i_flags & EXT4_INLINE_DATA_FL) &&
1578                     (inode->i_flags & EXT4_EXTENTS_FL)) {
1579                         int res = fix_inline_data_extents_file(ctx, ino, inode,
1580                                                                inode_size,
1581                                                                &pctx);
1582                         if (res < 0) {
1583                                 /* skip FINISH_INODE_LOOP */
1584                                 continue;
1585                         }
1586                 }
1587
1588                 /* Test for incorrect inline_data flags settings. */
1589                 if ((inode->i_flags & EXT4_INLINE_DATA_FL) && !inlinedata_fs &&
1590                     (ino >= EXT2_FIRST_INODE(fs->super))) {
1591                         size_t size = 0;
1592
1593                         pctx.errcode = get_inline_data_ea_size(fs, ino, inode,
1594                                                                &size);
1595                         if (!pctx.errcode &&
1596                             fix_problem(ctx, PR_1_INLINE_DATA_FEATURE, &pctx)) {
1597                                 ext2fs_set_feature_inline_data(sb);
1598                                 ext2fs_mark_super_dirty(fs);
1599                                 inlinedata_fs = 1;
1600                         } else if (fix_problem(ctx, PR_1_INLINE_DATA_SET, &pctx)) {
1601                                 e2fsck_clear_inode(ctx, ino, inode, 0, "pass1");
1602                                 /* skip FINISH_INODE_LOOP */
1603                                 continue;
1604                         }
1605                 }
1606
1607                 /* Test for inline data flag but no attr */
1608                 if ((inode->i_flags & EXT4_INLINE_DATA_FL) && inlinedata_fs &&
1609                     (ino >= EXT2_FIRST_INODE(fs->super))) {
1610                         size_t size = 0;
1611                         errcode_t err;
1612                         int flags;
1613
1614                         flags = fs->flags;
1615                         if (failed_csum)
1616                                 fs->flags |= EXT2_FLAG_IGNORE_CSUM_ERRORS;
1617                         err = get_inline_data_ea_size(fs, ino, inode, &size);
1618                         fs->flags = (flags & EXT2_FLAG_IGNORE_CSUM_ERRORS) |
1619                                     (fs->flags & ~EXT2_FLAG_IGNORE_CSUM_ERRORS);
1620
1621                         switch (err) {
1622                         case 0:
1623                                 /* Everything is awesome... */
1624                                 break;
1625                         case EXT2_ET_BAD_EA_BLOCK_NUM:
1626                         case EXT2_ET_BAD_EA_HASH:
1627                         case EXT2_ET_BAD_EA_HEADER:
1628                         case EXT2_ET_EA_BAD_NAME_LEN:
1629                         case EXT2_ET_EA_BAD_VALUE_SIZE:
1630                         case EXT2_ET_EA_KEY_NOT_FOUND:
1631                         case EXT2_ET_EA_NO_SPACE:
1632                         case EXT2_ET_MISSING_EA_FEATURE:
1633                         case EXT2_ET_INLINE_DATA_CANT_ITERATE:
1634                         case EXT2_ET_INLINE_DATA_NO_BLOCK:
1635                         case EXT2_ET_INLINE_DATA_NO_SPACE:
1636                         case EXT2_ET_NO_INLINE_DATA:
1637                         case EXT2_ET_EXT_ATTR_CSUM_INVALID:
1638                         case EXT2_ET_EA_BAD_VALUE_OFFSET:
1639                         case EXT2_ET_EA_INODE_CORRUPTED:
1640                                 /* broken EA or no system.data EA; truncate */
1641                                 if (fix_problem(ctx, PR_1_INLINE_DATA_NO_ATTR,
1642                                                 &pctx)) {
1643                                         err = ext2fs_inode_size_set(fs, inode, 0);
1644                                         if (err) {
1645                                                 pctx.errcode = err;
1646                                                 ctx->flags |= E2F_FLAG_ABORT;
1647                                                 goto endit;
1648                                         }
1649                                         inode->i_flags &= ~EXT4_INLINE_DATA_FL;
1650                                         memset(&inode->i_block, 0,
1651                                                sizeof(inode->i_block));
1652                                         e2fsck_write_inode(ctx, ino, inode,
1653                                                            "pass1");
1654                                         failed_csum = 0;
1655                                 }
1656                                 break;
1657                         default:
1658                                 /* Some other kind of non-xattr error? */
1659                                 pctx.errcode = err;
1660                                 ctx->flags |= E2F_FLAG_ABORT;
1661                                 goto endit;
1662                         }
1663                 }
1664
1665                 /*
1666                  * Test for incorrect extent flag settings.
1667                  *
1668                  * On big-endian machines we must be careful:
1669                  * When the inode is read, the i_block array is not swapped
1670                  * if the extent flag is set.  Therefore if we are testing
1671                  * for or fixing a wrongly-set flag, we must potentially
1672                  * (un)swap before testing, or after fixing.
1673                  */
1674
1675                 /*
1676                  * In this case the extents flag was set when read, so
1677                  * extent_header_verify is ok.  If the inode is cleared,
1678                  * no need to swap... so no extra swapping here.
1679                  */
1680                 if ((inode->i_flags & EXT4_EXTENTS_FL) && !extent_fs &&
1681                     (inode->i_links_count || (ino == EXT2_BAD_INO) ||
1682                      (ino == EXT2_ROOT_INO) || (ino == EXT2_JOURNAL_INO))) {
1683                         if ((ext2fs_extent_header_verify(inode->i_block,
1684                                                  sizeof(inode->i_block)) == 0) &&
1685                             fix_problem(ctx, PR_1_EXTENT_FEATURE, &pctx)) {
1686                                 ext2fs_set_feature_extents(sb);
1687                                 ext2fs_mark_super_dirty(fs);
1688                                 extent_fs = 1;
1689                         } else if (fix_problem(ctx, PR_1_EXTENTS_SET, &pctx)) {
1690                         clear_inode:
1691                                 e2fsck_clear_inode(ctx, ino, inode, 0, "pass1");
1692                                 if (ino == EXT2_BAD_INO)
1693                                         ext2fs_mark_inode_bitmap2(ctx->inode_used_map,
1694                                                                  ino);
1695                                 /* skip FINISH_INODE_LOOP */
1696                                 continue;
1697                         }
1698                 }
1699
1700                 /*
1701                  * For big-endian machines:
1702                  * If the inode didn't have the extents flag set when it
1703                  * was read, then the i_blocks array was swapped.  To test
1704                  * as an extents header, we must swap it back first.
1705                  * IF we then set the extents flag, the entire i_block
1706                  * array must be un/re-swapped to make it proper extents data.
1707                  */
1708                 if (extent_fs && !(inode->i_flags & EXT4_EXTENTS_FL) &&
1709                     (inode->i_links_count || (ino == EXT2_BAD_INO) ||
1710                      (ino == EXT2_ROOT_INO) || (ino == EXT2_JOURNAL_INO)) &&
1711                     (LINUX_S_ISREG(inode->i_mode) ||
1712                      LINUX_S_ISDIR(inode->i_mode))) {
1713                         void *ehp;
1714 #ifdef WORDS_BIGENDIAN
1715                         __u32 tmp_block[EXT2_N_BLOCKS];
1716
1717                         for (i = 0; i < EXT2_N_BLOCKS; i++)
1718                                 tmp_block[i] = ext2fs_swab32(inode->i_block[i]);
1719                         ehp = tmp_block;
1720 #else
1721                         ehp = inode->i_block;
1722 #endif
1723                         if ((ext2fs_extent_header_verify(ehp,
1724                                          sizeof(inode->i_block)) == 0) &&
1725                             (fix_problem(ctx, PR_1_UNSET_EXTENT_FL, &pctx))) {
1726                                 inode->i_flags |= EXT4_EXTENTS_FL;
1727 #ifdef WORDS_BIGENDIAN
1728                                 memcpy(inode->i_block, tmp_block,
1729                                        sizeof(inode->i_block));
1730 #endif
1731                                 e2fsck_write_inode(ctx, ino, inode, "pass1");
1732                                 failed_csum = 0;
1733                         }
1734                 }
1735
1736                 if (ino == EXT2_BAD_INO) {
1737                         struct process_block_struct pb;
1738
1739                         if ((failed_csum || inode->i_mode || inode->i_uid ||
1740                              inode->i_gid || inode->i_links_count ||
1741                              (inode->i_flags & EXT4_INLINE_DATA_FL) ||
1742                              inode->i_file_acl) &&
1743                             fix_problem(ctx, PR_1_INVALID_BAD_INODE, &pctx)) {
1744                                 memset(inode, 0, sizeof(struct ext2_inode));
1745                                 e2fsck_write_inode(ctx, ino, inode,
1746                                                    "clear bad inode");
1747                                 failed_csum = 0;
1748                         }
1749
1750                         pctx.errcode = ext2fs_copy_bitmap(ctx->block_found_map,
1751                                                           &pb.fs_meta_blocks);
1752                         if (pctx.errcode) {
1753                                 pctx.num = 4;
1754                                 fix_problem(ctx, PR_1_ALLOCATE_BBITMAP_ERROR, &pctx);
1755                                 ctx->flags |= E2F_FLAG_ABORT;
1756                                 goto endit;
1757                         }
1758                         pb.ino = EXT2_BAD_INO;
1759                         pb.num_blocks = pb.last_block = 0;
1760                         pb.last_db_block = -1;
1761                         pb.num_illegal_blocks = 0;
1762                         pb.suppress = 0; pb.clear = 0; pb.is_dir = 0;
1763                         pb.is_reg = 0; pb.fragmented = 0; pb.bbcheck = 0;
1764                         pb.inode = inode;
1765                         pb.pctx = &pctx;
1766                         pb.ctx = ctx;
1767                         pctx.errcode = ext2fs_block_iterate3(fs, ino, 0,
1768                                      block_buf, process_bad_block, &pb);
1769                         ext2fs_free_block_bitmap(pb.fs_meta_blocks);
1770                         if (pctx.errcode) {
1771                                 fix_problem(ctx, PR_1_BLOCK_ITERATE, &pctx);
1772                                 ctx->flags |= E2F_FLAG_ABORT;
1773                                 goto endit;
1774                         }
1775                         if (pb.bbcheck)
1776                                 if (!fix_problem(ctx, PR_1_BBINODE_BAD_METABLOCK_PROMPT, &pctx)) {
1777                                 ctx->flags |= E2F_FLAG_ABORT;
1778                                 goto endit;
1779                         }
1780                         ext2fs_mark_inode_bitmap2(ctx->inode_used_map, ino);
1781                         clear_problem_context(&pctx);
1782                         FINISH_INODE_LOOP(ctx, ino, &pctx, failed_csum);
1783                         continue;
1784                 } else if (ino == EXT2_ROOT_INO) {
1785                         /*
1786                          * Make sure the root inode is a directory; if
1787                          * not, offer to clear it.  It will be
1788                          * regenerated in pass #3.
1789                          */
1790                         if (!LINUX_S_ISDIR(inode->i_mode)) {
1791                                 if (fix_problem(ctx, PR_1_ROOT_NO_DIR, &pctx))
1792                                         goto clear_inode;
1793                         }
1794                         /*
1795                          * If dtime is set, offer to clear it.  mke2fs
1796                          * version 0.2b created filesystems with the
1797                          * dtime field set for the root and lost+found
1798                          * directories.  We won't worry about
1799                          * /lost+found, since that can be regenerated
1800                          * easily.  But we will fix the root directory
1801                          * as a special case.
1802                          */
1803                         if (inode->i_dtime && inode->i_links_count) {
1804                                 if (fix_problem(ctx, PR_1_ROOT_DTIME, &pctx)) {
1805                                         inode->i_dtime = 0;
1806                                         e2fsck_write_inode(ctx, ino, inode,
1807                                                            "pass1");
1808                                         failed_csum = 0;
1809                                 }
1810                         }
1811                 } else if (ino == EXT2_JOURNAL_INO) {
1812                         ext2fs_mark_inode_bitmap2(ctx->inode_used_map, ino);
1813                         if (fs->super->s_journal_inum == EXT2_JOURNAL_INO) {
1814                                 if (!LINUX_S_ISREG(inode->i_mode) &&
1815                                     fix_problem(ctx, PR_1_JOURNAL_BAD_MODE,
1816                                                 &pctx)) {
1817                                         inode->i_mode = LINUX_S_IFREG;
1818                                         e2fsck_write_inode(ctx, ino, inode,
1819                                                            "pass1");
1820                                         failed_csum = 0;
1821                                 }
1822                                 check_blocks(ctx, &pctx, block_buf, NULL);
1823                                 FINISH_INODE_LOOP(ctx, ino, &pctx, failed_csum);
1824                                 continue;
1825                         }
1826                         if ((inode->i_links_count ||
1827                              inode->i_blocks || inode->i_block[0]) &&
1828                             fix_problem(ctx, PR_1_JOURNAL_INODE_NOT_CLEAR,
1829                                         &pctx)) {
1830                                 memset(inode, 0, inode_size);
1831                                 ext2fs_icount_store(ctx->inode_link_info,
1832                                                     ino, 0);
1833                                 e2fsck_write_inode_full(ctx, ino, inode,
1834                                                         inode_size, "pass1");
1835                                 failed_csum = 0;
1836                         }
1837                 } else if (quota_inum_is_reserved(fs, ino)) {
1838                         ext2fs_mark_inode_bitmap2(ctx->inode_used_map, ino);
1839                         if (ext2fs_has_feature_quota(fs->super) &&
1840                             quota_inum_is_super(fs->super, ino)) {
1841                                 if (!LINUX_S_ISREG(inode->i_mode) &&
1842                                     fix_problem(ctx, PR_1_QUOTA_BAD_MODE,
1843                                                         &pctx)) {
1844                                         inode->i_mode = LINUX_S_IFREG;
1845                                         e2fsck_write_inode(ctx, ino, inode,
1846                                                         "pass1");
1847                                         failed_csum = 0;
1848                                 }
1849                                 check_blocks(ctx, &pctx, block_buf, NULL);
1850                                 FINISH_INODE_LOOP(ctx, ino, &pctx, failed_csum);
1851                                 continue;
1852                         }
1853                         if ((inode->i_links_count ||
1854                              inode->i_blocks || inode->i_block[0]) &&
1855                             fix_problem(ctx, PR_1_QUOTA_INODE_NOT_CLEAR,
1856                                         &pctx)) {
1857                                 memset(inode, 0, inode_size);
1858                                 ext2fs_icount_store(ctx->inode_link_info,
1859                                                     ino, 0);
1860                                 e2fsck_write_inode_full(ctx, ino, inode,
1861                                                         inode_size, "pass1");
1862                                 failed_csum = 0;
1863                         }
1864                 } else if (ino == fs->super->s_orphan_file_inum) {
1865                         ext2fs_mark_inode_bitmap2(ctx->inode_used_map, ino);
1866                         if (ext2fs_has_feature_orphan_file(fs->super)) {
1867                                 if (!LINUX_S_ISREG(inode->i_mode) &&
1868                                     fix_problem(ctx, PR_1_ORPHAN_FILE_BAD_MODE,
1869                                                 &pctx)) {
1870                                         inode->i_mode = LINUX_S_IFREG;
1871                                         e2fsck_write_inode(ctx, ino, inode,
1872                                                            "pass1");
1873                                         failed_csum = 0;
1874                                 }
1875                                 check_blocks(ctx, &pctx, block_buf, NULL);
1876                                 FINISH_INODE_LOOP(ctx, ino, &pctx, failed_csum);
1877                                 continue;
1878                         }
1879                         if ((inode->i_links_count ||
1880                              inode->i_blocks || inode->i_block[0]) &&
1881                             fix_problem(ctx, PR_1_ORPHAN_FILE_NOT_CLEAR,
1882                                         &pctx)) {
1883                                 memset(inode, 0, inode_size);
1884                                 ext2fs_icount_store(ctx->inode_link_info, ino,
1885                                                     0);
1886                                 e2fsck_write_inode_full(ctx, ino, inode,
1887                                                         inode_size, "pass1");
1888                                 failed_csum = 0;
1889                         }
1890                 } else if (ino < EXT2_FIRST_INODE(fs->super)) {
1891                         problem_t problem = 0;
1892
1893                         ext2fs_mark_inode_bitmap2(ctx->inode_used_map, ino);
1894                         if (ino == EXT2_BOOT_LOADER_INO) {
1895                                 if (LINUX_S_ISDIR(inode->i_mode))
1896                                         problem = PR_1_RESERVED_BAD_MODE;
1897                         } else if (ino == EXT2_RESIZE_INO) {
1898                                 if (inode->i_mode &&
1899                                     !LINUX_S_ISREG(inode->i_mode))
1900                                         problem = PR_1_RESERVED_BAD_MODE;
1901                         } else {
1902                                 if (inode->i_mode != 0)
1903                                         problem = PR_1_RESERVED_BAD_MODE;
1904                         }
1905                         if (problem) {
1906                                 if (fix_problem(ctx, problem, &pctx)) {
1907                                         inode->i_mode = 0;
1908                                         e2fsck_write_inode(ctx, ino, inode,
1909                                                            "pass1");
1910                                         failed_csum = 0;
1911                                 }
1912                         }
1913                         check_blocks(ctx, &pctx, block_buf, NULL);
1914                         FINISH_INODE_LOOP(ctx, ino, &pctx, failed_csum);
1915                         continue;
1916                 }
1917
1918                 if (!inode->i_links_count) {
1919                         FINISH_INODE_LOOP(ctx, ino, &pctx, failed_csum);
1920                         continue;
1921                 }
1922                 /*
1923                  * n.b.  0.3c ext2fs code didn't clear i_links_count for
1924                  * deleted files.  Oops.
1925                  *
1926                  * Since all new ext2 implementations get this right,
1927                  * we now assume that the case of non-zero
1928                  * i_links_count and non-zero dtime means that we
1929                  * should keep the file, not delete it.
1930                  *
1931                  */
1932                 if (inode->i_dtime) {
1933                         if (fix_problem(ctx, PR_1_SET_DTIME, &pctx)) {
1934                                 inode->i_dtime = 0;
1935                                 e2fsck_write_inode(ctx, ino, inode, "pass1");
1936                                 failed_csum = 0;
1937                         }
1938                 }
1939
1940                 ext2fs_mark_inode_bitmap2(ctx->inode_used_map, ino);
1941                 switch (fs->super->s_creator_os) {
1942                     case EXT2_OS_HURD:
1943                         frag = inode->osd2.hurd2.h_i_frag;
1944                         fsize = inode->osd2.hurd2.h_i_fsize;
1945                         break;
1946                     default:
1947                         frag = fsize = 0;
1948                 }
1949
1950                 if (inode->i_faddr || frag || fsize ||
1951                     (!ext2fs_has_feature_largedir(fs->super) &&
1952                     (LINUX_S_ISDIR(inode->i_mode) && inode->i_size_high)))
1953                         mark_inode_bad(ctx, ino);
1954                 if ((fs->super->s_creator_os != EXT2_OS_HURD) &&
1955                     !ext2fs_has_feature_64bit(fs->super) &&
1956                     inode->osd2.linux2.l_i_file_acl_high != 0)
1957                         mark_inode_bad(ctx, ino);
1958                 if ((fs->super->s_creator_os != EXT2_OS_HURD) &&
1959                     !ext2fs_has_feature_huge_file(fs->super) &&
1960                     (inode->osd2.linux2.l_i_blocks_hi != 0))
1961                         mark_inode_bad(ctx, ino);
1962                 if (inode->i_flags & EXT2_IMAGIC_FL) {
1963                         if (imagic_fs) {
1964                                 if (!ctx->inode_imagic_map)
1965                                         alloc_imagic_map(ctx);
1966                                 ext2fs_mark_inode_bitmap2(ctx->inode_imagic_map,
1967                                                          ino);
1968                         } else {
1969                                 if (fix_problem(ctx, PR_1_SET_IMAGIC, &pctx)) {
1970                                         inode->i_flags &= ~EXT2_IMAGIC_FL;
1971                                         e2fsck_write_inode(ctx, ino,
1972                                                            inode, "pass1");
1973                                         failed_csum = 0;
1974                                 }
1975                         }
1976                 }
1977
1978                 check_inode_extra_space(ctx, &pctx, &ea_ibody_quota);
1979                 check_is_really_dir(ctx, &pctx, block_buf);
1980
1981                 /*
1982                  * ext2fs_inode_has_valid_blocks2 does not actually look
1983                  * at i_block[] values, so not endian-sensitive here.
1984                  */
1985                 if (extent_fs && (inode->i_flags & EXT4_EXTENTS_FL) &&
1986                     LINUX_S_ISLNK(inode->i_mode) &&
1987                     !ext2fs_inode_has_valid_blocks2(fs, inode) &&
1988                     fix_problem(ctx, PR_1_FAST_SYMLINK_EXTENT_FL, &pctx)) {
1989                         inode->i_flags &= ~EXT4_EXTENTS_FL;
1990                         e2fsck_write_inode(ctx, ino, inode, "pass1");
1991                         failed_csum = 0;
1992                 }
1993
1994                 if ((inode->i_flags & EXT4_ENCRYPT_FL) &&
1995                     add_encrypted_file(ctx, &pctx) < 0)
1996                         goto clear_inode;
1997
1998                 if (casefold_fs && inode->i_flags & EXT4_CASEFOLD_FL)
1999                         ext2fs_mark_inode_bitmap2(ctx->inode_casefold_map, ino);
2000
2001                 if (LINUX_S_ISDIR(inode->i_mode)) {
2002                         ext2fs_mark_inode_bitmap2(ctx->inode_dir_map, ino);
2003                         e2fsck_add_dir_info(ctx, ino, 0);
2004                         ctx->fs_directory_count++;
2005                         if (inode->i_flags & EXT4_CASEFOLD_FL)
2006                                 add_casefolded_dir(ctx, ino);
2007                 } else if (LINUX_S_ISREG (inode->i_mode)) {
2008                         ext2fs_mark_inode_bitmap2(ctx->inode_reg_map, ino);
2009                         ctx->fs_regular_count++;
2010                 } else if (LINUX_S_ISCHR (inode->i_mode) &&
2011                            e2fsck_pass1_check_device_inode(fs, inode)) {
2012                         check_extents_inlinedata(ctx, &pctx);
2013                         check_immutable(ctx, &pctx);
2014                         check_size(ctx, &pctx);
2015                         ctx->fs_chardev_count++;
2016                 } else if (LINUX_S_ISBLK (inode->i_mode) &&
2017                            e2fsck_pass1_check_device_inode(fs, inode)) {
2018                         check_extents_inlinedata(ctx, &pctx);
2019                         check_immutable(ctx, &pctx);
2020                         check_size(ctx, &pctx);
2021                         ctx->fs_blockdev_count++;
2022                 } else if (LINUX_S_ISLNK (inode->i_mode) &&
2023                            e2fsck_pass1_check_symlink(fs, ino, inode,
2024                                                       block_buf)) {
2025                         check_immutable(ctx, &pctx);
2026                         ctx->fs_symlinks_count++;
2027                         if (inode->i_flags & EXT4_INLINE_DATA_FL) {
2028                                 FINISH_INODE_LOOP(ctx, ino, &pctx, failed_csum);
2029                                 continue;
2030                         } else if (ext2fs_is_fast_symlink(inode)) {
2031                                 ctx->fs_fast_symlinks_count++;
2032                                 check_blocks(ctx, &pctx, block_buf,
2033                                              &ea_ibody_quota);
2034                                 FINISH_INODE_LOOP(ctx, ino, &pctx, failed_csum);
2035                                 continue;
2036                         }
2037                 }
2038                 else if (LINUX_S_ISFIFO (inode->i_mode) &&
2039                          e2fsck_pass1_check_device_inode(fs, inode)) {
2040                         check_extents_inlinedata(ctx, &pctx);
2041                         check_immutable(ctx, &pctx);
2042                         check_size(ctx, &pctx);
2043                         ctx->fs_fifo_count++;
2044                 } else if ((LINUX_S_ISSOCK (inode->i_mode)) &&
2045                            e2fsck_pass1_check_device_inode(fs, inode)) {
2046                         check_extents_inlinedata(ctx, &pctx);
2047                         check_immutable(ctx, &pctx);
2048                         check_size(ctx, &pctx);
2049                         ctx->fs_sockets_count++;
2050                 } else
2051                         mark_inode_bad(ctx, ino);
2052                 if (!(inode->i_flags & EXT4_EXTENTS_FL) &&
2053                     !(inode->i_flags & EXT4_INLINE_DATA_FL)) {
2054                         if (inode->i_block[EXT2_IND_BLOCK])
2055                                 ctx->fs_ind_count++;
2056                         if (inode->i_block[EXT2_DIND_BLOCK])
2057                                 ctx->fs_dind_count++;
2058                         if (inode->i_block[EXT2_TIND_BLOCK])
2059                                 ctx->fs_tind_count++;
2060                 }
2061                 if (!(inode->i_flags & EXT4_EXTENTS_FL) &&
2062                     !(inode->i_flags & EXT4_INLINE_DATA_FL) &&
2063                     (inode->i_block[EXT2_IND_BLOCK] ||
2064                      inode->i_block[EXT2_DIND_BLOCK] ||
2065                      inode->i_block[EXT2_TIND_BLOCK] ||
2066                      ext2fs_file_acl_block(fs, inode))) {
2067                         struct process_inode_block *itp;
2068
2069                         itp = &inodes_to_process[process_inode_count];
2070                         itp->ino = ino;
2071                         itp->ea_ibody_quota = ea_ibody_quota;
2072                         if (inode_size < sizeof(struct ext2_inode_large))
2073                                 memcpy(&itp->inode, inode, inode_size);
2074                         else
2075                                 memcpy(&itp->inode, inode, sizeof(itp->inode));
2076                         process_inode_count++;
2077                 } else
2078                         check_blocks(ctx, &pctx, block_buf, &ea_ibody_quota);
2079
2080                 FINISH_INODE_LOOP(ctx, ino, &pctx, failed_csum);
2081
2082                 if (ctx->flags & E2F_FLAG_SIGNAL_MASK)
2083                         goto endit;
2084
2085                 if (process_inode_count >= ctx->process_inode_size) {
2086                         process_inodes(ctx, block_buf);
2087
2088                         if (e2fsck_should_abort(ctx))
2089                                 goto endit;
2090                 }
2091         }
2092         process_inodes(ctx, block_buf);
2093         ext2fs_close_inode_scan(scan);
2094         scan = NULL;
2095
2096         reserve_block_for_root_repair(ctx);
2097         reserve_block_for_lnf_repair(ctx);
2098
2099         /*
2100          * If any extended attribute blocks' reference counts need to
2101          * be adjusted, either up (ctx->refcount_extra), or down
2102          * (ctx->refcount), then fix them.
2103          */
2104         if (ctx->refcount) {
2105                 adjust_extattr_refcount(ctx, ctx->refcount, block_buf, -1);
2106                 ea_refcount_free(ctx->refcount);
2107                 ctx->refcount = 0;
2108         }
2109         if (ctx->refcount_extra) {
2110                 adjust_extattr_refcount(ctx, ctx->refcount_extra,
2111                                         block_buf, +1);
2112                 ea_refcount_free(ctx->refcount_extra);
2113                 ctx->refcount_extra = 0;
2114         }
2115
2116         if (ctx->ea_block_quota_blocks) {
2117                 ea_refcount_free(ctx->ea_block_quota_blocks);
2118                 ctx->ea_block_quota_blocks = 0;
2119         }
2120
2121         if (ctx->ea_block_quota_inodes) {
2122                 ea_refcount_free(ctx->ea_block_quota_inodes);
2123                 ctx->ea_block_quota_inodes = 0;
2124         }
2125
2126         if (ctx->invalid_bitmaps)
2127                 handle_fs_bad_blocks(ctx);
2128
2129         /* We don't need the block_ea_map any more */
2130         if (ctx->block_ea_map) {
2131                 ext2fs_free_block_bitmap(ctx->block_ea_map);
2132                 ctx->block_ea_map = 0;
2133         }
2134
2135         /* We don't need the encryption policy => ID map any more */
2136         destroy_encryption_policy_map(ctx);
2137
2138         if (ctx->flags & E2F_FLAG_RESIZE_INODE) {
2139                 clear_problem_context(&pctx);
2140                 pctx.errcode = ext2fs_create_resize_inode(fs);
2141                 if (pctx.errcode) {
2142                         if (!fix_problem(ctx, PR_1_RESIZE_INODE_CREATE,
2143                                          &pctx)) {
2144                                 ctx->flags |= E2F_FLAG_ABORT;
2145                                 goto endit;
2146                         }
2147                         pctx.errcode = 0;
2148                 }
2149                 if (!pctx.errcode) {
2150                         e2fsck_read_inode(ctx, EXT2_RESIZE_INO, inode,
2151                                           "recreate inode");
2152                         ext2fs_inode_xtime_set(inode, i_mtime, ctx->now);
2153                         e2fsck_write_inode(ctx, EXT2_RESIZE_INO, inode,
2154                                            "recreate inode");
2155                 }
2156                 ctx->flags &= ~E2F_FLAG_RESIZE_INODE;
2157         }
2158
2159         if (ctx->flags & E2F_FLAG_RESTART) {
2160                 /*
2161                  * Only the master copy of the superblock and block
2162                  * group descriptors are going to be written during a
2163                  * restart, so set the superblock to be used to be the
2164                  * master superblock.
2165                  */
2166                 ctx->use_superblock = 0;
2167                 goto endit;
2168         }
2169
2170         if (ctx->large_dirs && !ext2fs_has_feature_largedir(fs->super)) {
2171                 if (fix_problem(ctx, PR_2_FEATURE_LARGE_DIRS, &pctx)) {
2172                         ext2fs_set_feature_largedir(fs->super);
2173                         fs->flags &= ~EXT2_FLAG_MASTER_SB_ONLY;
2174                         ext2fs_mark_super_dirty(fs);
2175                 }
2176                 if (fs->super->s_rev_level == EXT2_GOOD_OLD_REV &&
2177                     fix_problem(ctx, PR_1_FS_REV_LEVEL, &pctx)) {
2178                         ext2fs_update_dynamic_rev(fs);
2179                         ext2fs_mark_super_dirty(fs);
2180                 }
2181         }
2182
2183         if (ctx->block_dup_map) {
2184                 if (ctx->options & E2F_OPT_PREEN) {
2185                         clear_problem_context(&pctx);
2186                         fix_problem(ctx, PR_1_DUP_BLOCKS_PREENSTOP, &pctx);
2187                 }
2188                 e2fsck_pass1_dupblocks(ctx, block_buf);
2189         }
2190         ctx->flags |= E2F_FLAG_ALLOC_OK;
2191 endit:
2192         e2fsck_use_inode_shortcuts(ctx, 0);
2193         ext2fs_free_mem(&inodes_to_process);
2194         inodes_to_process = 0;
2195
2196         if (scan)
2197                 ext2fs_close_inode_scan(scan);
2198         if (block_buf)
2199                 ext2fs_free_mem(&block_buf);
2200         if (inode)
2201                 ext2fs_free_mem(&inode);
2202
2203         /*
2204          * The l+f inode may have been cleared, so zap it now and
2205          * later passes will recalculate it if necessary
2206          */
2207         ctx->lost_and_found = 0;
2208
2209         if ((ctx->flags & E2F_FLAG_SIGNAL_MASK) == 0)
2210                 print_resource_track(ctx, _("Pass 1"), &rtrack, ctx->fs->io);
2211         else
2212                 ctx->invalid_bitmaps++;
2213 }
2214
2215 static void e2fsck_pass1_copy_fs(ext2_filsys dest, ext2_filsys src)
2216 {
2217         memcpy(dest, src, sizeof(struct struct_ext2_filsys));
2218         if (dest->dblist)
2219                 dest->dblist->fs = dest;
2220         if (dest->inode_map)
2221                 dest->inode_map->fs = dest;
2222         if (dest->block_map)
2223                 dest->block_map->fs = dest;
2224 }
2225
2226 static errcode_t e2fsck_pass1_thread_prepare(e2fsck_t global_ctx, e2fsck_t *thread_ctx)
2227 {
2228         errcode_t       retval;
2229         e2fsck_t        thread_context;
2230         ext2_filsys     thread_fs;
2231         ext2_filsys     global_fs = global_ctx->fs;
2232
2233         retval = ext2fs_get_mem(sizeof(struct e2fsck_struct), &thread_context);
2234         if (retval) {
2235                 com_err(global_ctx->program_name, retval, "while allocating memory");
2236                 return retval;
2237         }
2238         memcpy(thread_context, global_ctx, sizeof(struct e2fsck_struct));
2239         thread_context->global_ctx = global_ctx;
2240
2241         retval = ext2fs_get_mem(sizeof(struct struct_ext2_filsys), &thread_fs);
2242         if (retval) {
2243                 com_err(global_ctx->program_name, retval, "while allocating memory");
2244                 goto out_context;
2245         }
2246
2247         e2fsck_pass1_copy_fs(thread_fs, global_fs);
2248         thread_fs->priv_data = thread_context;
2249
2250         thread_context->fs = thread_fs;
2251         *thread_ctx = thread_context;
2252         return 0;
2253 out_context:
2254         ext2fs_free_mem(&thread_context);
2255         return retval;
2256 }
2257
2258 static int e2fsck_pass1_thread_join(e2fsck_t global_ctx, e2fsck_t thread_ctx)
2259 {
2260         int             flags = global_ctx->flags;
2261         ext2_filsys     thread_fs = thread_ctx->fs;
2262         ext2_filsys     global_fs = global_ctx->fs;
2263 #ifdef HAVE_SETJMP_H
2264         jmp_buf         old_jmp;
2265
2266         memcpy(old_jmp, global_ctx->abort_loc, sizeof(jmp_buf));
2267 #endif
2268         memcpy(global_ctx, thread_ctx, sizeof(struct e2fsck_struct));
2269 #ifdef HAVE_SETJMP_H
2270         memcpy(global_ctx->abort_loc, old_jmp, sizeof(jmp_buf));
2271 #endif
2272         /* Keep the global singal flags*/
2273         global_ctx->flags |= (flags & E2F_FLAG_SIGNAL_MASK) |
2274                              (global_ctx->flags & E2F_FLAG_SIGNAL_MASK);
2275
2276         e2fsck_pass1_copy_fs(global_fs, thread_fs);
2277         global_fs->priv_data = global_ctx;
2278         global_ctx->fs = global_fs;
2279
2280         ext2fs_free_mem(&thread_ctx->fs);
2281         ext2fs_free_mem(&thread_ctx);
2282         return 0;
2283 }
2284
2285 void e2fsck_pass1_multithread(e2fsck_t ctx)
2286 {
2287         errcode_t       retval;
2288         e2fsck_t        thread_ctx;
2289
2290         retval = e2fsck_pass1_thread_prepare(ctx, &thread_ctx);
2291         if (retval) {
2292                 com_err(ctx->program_name, 0,
2293                         _("while preparing pass1 thread\n"));
2294                 ctx->flags |= E2F_FLAG_ABORT;
2295                 return;
2296         }
2297
2298 #ifdef HAVE_SETJMP_H
2299         /*
2300          * When fatal_error() happens, jump to here. The thread
2301          * context's flags will be saved, but its abort_loc will
2302          * be overwritten by original jump buffer for the later
2303          * tests.
2304          */
2305         if (setjmp(thread_ctx->abort_loc)) {
2306                 thread_ctx->flags &= ~E2F_FLAG_SETJMP_OK;
2307                 e2fsck_pass1_thread_join(ctx, thread_ctx);
2308                 return;
2309         }
2310         thread_ctx->flags |= E2F_FLAG_SETJMP_OK;
2311 #endif
2312
2313         e2fsck_pass1_thread(thread_ctx);
2314         retval = e2fsck_pass1_thread_join(ctx, thread_ctx);
2315         if (retval) {
2316                 com_err(ctx->program_name, 0,
2317                         _("while joining pass1 thread\n"));
2318                 ctx->flags |= E2F_FLAG_ABORT;
2319                 return;
2320         }
2321 }
2322
2323 void e2fsck_pass1(e2fsck_t ctx)
2324 {
2325         e2fsck_pass1_multithread(ctx);
2326 }
2327
2328 #undef FINISH_INODE_LOOP
2329
2330 /*
2331  * When the inode_scan routines call this callback at the end of the
2332  * glock group, call process_inodes.
2333  */
2334 static errcode_t scan_callback(ext2_filsys fs,
2335                                ext2_inode_scan scan EXT2FS_ATTR((unused)),
2336                                dgrp_t group, void * priv_data)
2337 {
2338         struct scan_callback_struct *scan_struct;
2339         e2fsck_t ctx;
2340
2341         scan_struct = (struct scan_callback_struct *) priv_data;
2342         ctx = scan_struct->ctx;
2343
2344         process_inodes((e2fsck_t) fs->priv_data, scan_struct->block_buf);
2345
2346         if (ctx->progress)
2347                 if ((ctx->progress)(ctx, 1, group+1,
2348                                     ctx->fs->group_desc_count))
2349                         return EXT2_ET_CANCEL_REQUESTED;
2350
2351         return 0;
2352 }
2353
2354 /*
2355  * Process the inodes in the "inodes to process" list.
2356  */
2357 static void process_inodes(e2fsck_t ctx, char *block_buf)
2358 {
2359         int                     i;
2360         struct ext2_inode       *old_stashed_inode;
2361         ext2_ino_t              old_stashed_ino;
2362         const char              *old_operation;
2363         char                    buf[80];
2364         struct problem_context  pctx;
2365
2366 #if 0
2367         printf("begin process_inodes: ");
2368 #endif
2369         if (process_inode_count == 0)
2370                 return;
2371         old_operation = ehandler_operation(0);
2372         old_stashed_inode = ctx->stashed_inode;
2373         old_stashed_ino = ctx->stashed_ino;
2374         qsort(inodes_to_process, process_inode_count,
2375                       sizeof(struct process_inode_block), process_inode_cmp);
2376         clear_problem_context(&pctx);
2377         for (i=0; i < process_inode_count; i++) {
2378                 pctx.inode = ctx->stashed_inode =
2379                         (struct ext2_inode *) &inodes_to_process[i].inode;
2380                 pctx.ino = ctx->stashed_ino = inodes_to_process[i].ino;
2381
2382 #if 0
2383                 printf("%u ", pctx.ino);
2384 #endif
2385                 sprintf(buf, _("reading indirect blocks of inode %u"),
2386                         pctx.ino);
2387                 ehandler_operation(buf);
2388                 check_blocks(ctx, &pctx, block_buf,
2389                              &inodes_to_process[i].ea_ibody_quota);
2390                 if (e2fsck_should_abort(ctx))
2391                         break;
2392         }
2393         ctx->stashed_inode = old_stashed_inode;
2394         ctx->stashed_ino = old_stashed_ino;
2395         process_inode_count = 0;
2396 #if 0
2397         printf("end process inodes\n");
2398 #endif
2399         ehandler_operation(old_operation);
2400 }
2401
2402 static EXT2_QSORT_TYPE process_inode_cmp(const void *a, const void *b)
2403 {
2404         const struct process_inode_block *ib_a =
2405                 (const struct process_inode_block *) a;
2406         const struct process_inode_block *ib_b =
2407                 (const struct process_inode_block *) b;
2408         int     ret;
2409
2410         ret = (ib_a->inode.i_block[EXT2_IND_BLOCK] -
2411                ib_b->inode.i_block[EXT2_IND_BLOCK]);
2412         if (ret == 0)
2413                 /*
2414                  * We only call process_inodes() for non-extent
2415                  * inodes, so it's OK to pass NULL to
2416                  * ext2fs_file_acl_block() here.
2417                  */
2418                 ret = ext2fs_file_acl_block(0, ext2fs_const_inode(&ib_a->inode)) -
2419                         ext2fs_file_acl_block(0, ext2fs_const_inode(&ib_b->inode));
2420         if (ret == 0)
2421                 ret = ib_a->ino - ib_b->ino;
2422         return ret;
2423 }
2424
2425 /*
2426  * Mark an inode as being bad in some what
2427  */
2428 static void mark_inode_bad(e2fsck_t ctx, ext2_ino_t ino)
2429 {
2430         struct          problem_context pctx;
2431
2432         if (!ctx->inode_bad_map) {
2433                 clear_problem_context(&pctx);
2434
2435                 pctx.errcode = e2fsck_allocate_inode_bitmap(ctx->fs,
2436                                 _("bad inode map"), EXT2FS_BMAP64_RBTREE,
2437                                 "inode_bad_map", &ctx->inode_bad_map);
2438                 if (pctx.errcode) {
2439                         pctx.num = 3;
2440                         fix_problem(ctx, PR_1_ALLOCATE_IBITMAP_ERROR, &pctx);
2441                         /* Should never get here */
2442                         ctx->flags |= E2F_FLAG_ABORT;
2443                         return;
2444                 }
2445         }
2446         ext2fs_mark_inode_bitmap2(ctx->inode_bad_map, ino);
2447 }
2448
2449 static void add_casefolded_dir(e2fsck_t ctx, ext2_ino_t ino)
2450 {
2451         struct          problem_context pctx;
2452
2453         if (!ctx->casefolded_dirs) {
2454                 pctx.errcode = ext2fs_u32_list_create(&ctx->casefolded_dirs, 0);
2455                 if (pctx.errcode)
2456                         goto error;
2457         }
2458         pctx.errcode = ext2fs_u32_list_add(ctx->casefolded_dirs, ino);
2459         if (pctx.errcode == 0)
2460                 return;
2461 error:
2462         fix_problem(ctx, PR_1_ALLOCATE_CASEFOLDED_DIRLIST, &pctx);
2463         /* Should never get here */
2464         ctx->flags |= E2F_FLAG_ABORT;
2465 }
2466
2467 /*
2468  * This procedure will allocate the inode "bb" (badblock) map table
2469  */
2470 static void alloc_bb_map(e2fsck_t ctx)
2471 {
2472         struct          problem_context pctx;
2473
2474         clear_problem_context(&pctx);
2475         pctx.errcode = e2fsck_allocate_inode_bitmap(ctx->fs,
2476                         _("inode in bad block map"), EXT2FS_BMAP64_RBTREE,
2477                         "inode_bb_map", &ctx->inode_bb_map);
2478         if (pctx.errcode) {
2479                 pctx.num = 4;
2480                 fix_problem(ctx, PR_1_ALLOCATE_IBITMAP_ERROR, &pctx);
2481                 /* Should never get here */
2482                 ctx->flags |= E2F_FLAG_ABORT;
2483                 return;
2484         }
2485 }
2486
2487 /*
2488  * This procedure will allocate the inode imagic table
2489  */
2490 static void alloc_imagic_map(e2fsck_t ctx)
2491 {
2492         struct          problem_context pctx;
2493
2494         clear_problem_context(&pctx);
2495         pctx.errcode = e2fsck_allocate_inode_bitmap(ctx->fs,
2496                         _("imagic inode map"), EXT2FS_BMAP64_RBTREE,
2497                         "inode_imagic_map", &ctx->inode_imagic_map);
2498         if (pctx.errcode) {
2499                 pctx.num = 5;
2500                 fix_problem(ctx, PR_1_ALLOCATE_IBITMAP_ERROR, &pctx);
2501                 /* Should never get here */
2502                 ctx->flags |= E2F_FLAG_ABORT;
2503                 return;
2504         }
2505 }
2506
2507 /*
2508  * Marks a block as in use, setting the dup_map if it's been set
2509  * already.  Called by process_block and process_bad_block.
2510  *
2511  * WARNING: Assumes checks have already been done to make sure block
2512  * is valid.  This is true in both process_block and process_bad_block.
2513  */
2514 static _INLINE_ void mark_block_used(e2fsck_t ctx, blk64_t block)
2515 {
2516         struct          problem_context pctx;
2517
2518         clear_problem_context(&pctx);
2519
2520         if (ext2fs_fast_test_block_bitmap2(ctx->block_found_map, block)) {
2521                 if (ext2fs_has_feature_shared_blocks(ctx->fs->super) &&
2522                     !(ctx->options & E2F_OPT_UNSHARE_BLOCKS)) {
2523                         return;
2524                 }
2525                 if (!ctx->block_dup_map) {
2526                         pctx.errcode = e2fsck_allocate_block_bitmap(ctx->fs,
2527                                         _("multiply claimed block map"),
2528                                         EXT2FS_BMAP64_RBTREE, "block_dup_map",
2529                                         &ctx->block_dup_map);
2530                         if (pctx.errcode) {
2531                                 pctx.num = 3;
2532                                 fix_problem(ctx, PR_1_ALLOCATE_BBITMAP_ERROR,
2533                                             &pctx);
2534                                 /* Should never get here */
2535                                 ctx->flags |= E2F_FLAG_ABORT;
2536                                 return;
2537                         }
2538                 }
2539                 ext2fs_fast_mark_block_bitmap2(ctx->block_dup_map, block);
2540         } else {
2541                 ext2fs_fast_mark_block_bitmap2(ctx->block_found_map, block);
2542         }
2543 }
2544
2545 /*
2546  * When cluster size is greater than one block, it is caller's responsibility
2547  * to make sure block parameter starts at a cluster boundary.
2548  */
2549 static _INLINE_ void mark_blocks_used(e2fsck_t ctx, blk64_t block,
2550                                       unsigned int num)
2551 {
2552         if (ext2fs_test_block_bitmap_range2(ctx->block_found_map, block, num))
2553                 ext2fs_mark_block_bitmap_range2(ctx->block_found_map, block, num);
2554         else {
2555                 unsigned int i;
2556
2557                 for (i = 0; i < num; i += EXT2FS_CLUSTER_RATIO(ctx->fs))
2558                         mark_block_used(ctx, block + i);
2559         }
2560 }
2561
2562 /*
2563  * Adjust the extended attribute block's reference counts at the end
2564  * of pass 1, either by subtracting out references for EA blocks that
2565  * are still referenced in ctx->refcount, or by adding references for
2566  * EA blocks that had extra references as accounted for in
2567  * ctx->refcount_extra.
2568  */
2569 static void adjust_extattr_refcount(e2fsck_t ctx, ext2_refcount_t refcount,
2570                                     char *block_buf, int adjust_sign)
2571 {
2572         struct ext2_ext_attr_header     *header;
2573         struct problem_context          pctx;
2574         ext2_filsys                     fs = ctx->fs;
2575         blk64_t                         blk;
2576         __u32                           should_be;
2577         ea_value_t                      count;
2578
2579         clear_problem_context(&pctx);
2580
2581         ea_refcount_intr_begin(refcount);
2582         while (1) {
2583                 if ((blk = ea_refcount_intr_next(refcount, &count)) == 0)
2584                         break;
2585                 pctx.blk = blk;
2586                 pctx.errcode = ext2fs_read_ext_attr3(fs, blk, block_buf,
2587                                                      pctx.ino);
2588                 if (pctx.errcode) {
2589                         fix_problem(ctx, PR_1_EXTATTR_READ_ABORT, &pctx);
2590                         return;
2591                 }
2592                 header = (struct ext2_ext_attr_header *) block_buf;
2593                 pctx.blkcount = header->h_refcount;
2594                 should_be = header->h_refcount + adjust_sign * (int)count;
2595                 pctx.num = should_be;
2596                 if (fix_problem(ctx, PR_1_EXTATTR_REFCOUNT, &pctx)) {
2597                         header->h_refcount = should_be;
2598                         pctx.errcode = ext2fs_write_ext_attr3(fs, blk,
2599                                                              block_buf,
2600                                                              pctx.ino);
2601                         if (pctx.errcode) {
2602                                 fix_problem(ctx, PR_1_EXTATTR_WRITE_ABORT,
2603                                             &pctx);
2604                                 continue;
2605                         }
2606                 }
2607         }
2608 }
2609
2610 /*
2611  * Handle processing the extended attribute blocks
2612  */
2613 static int check_ext_attr(e2fsck_t ctx, struct problem_context *pctx,
2614                            char *block_buf, struct ea_quota *ea_block_quota)
2615 {
2616         ext2_filsys fs = ctx->fs;
2617         ext2_ino_t      ino = pctx->ino;
2618         struct ext2_inode *inode = pctx->inode;
2619         blk64_t         blk;
2620         char *          end;
2621         struct ext2_ext_attr_header *header;
2622         struct ext2_ext_attr_entry *first, *entry;
2623         blk64_t         quota_blocks = EXT2FS_C2B(fs, 1);
2624         __u64           quota_inodes = 0;
2625         region_t        region = 0;
2626         int             failed_csum = 0;
2627
2628         ea_block_quota->blocks = 0;
2629         ea_block_quota->inodes = 0;
2630
2631         blk = ext2fs_file_acl_block(fs, inode);
2632         if (blk == 0)
2633                 return 0;
2634
2635         /*
2636          * If the Extended attribute flag isn't set, then a non-zero
2637          * file acl means that the inode is corrupted.
2638          *
2639          * Or if the extended attribute block is an invalid block,
2640          * then the inode is also corrupted.
2641          */
2642         if (!ext2fs_has_feature_xattr(fs->super) ||
2643             (blk < fs->super->s_first_data_block) ||
2644             (blk >= ext2fs_blocks_count(fs->super))) {
2645                 mark_inode_bad(ctx, ino);
2646                 return 0;
2647         }
2648
2649         /* If ea bitmap hasn't been allocated, create it */
2650         if (!ctx->block_ea_map) {
2651                 pctx->errcode = e2fsck_allocate_block_bitmap(fs,
2652                                         _("ext attr block map"),
2653                                         EXT2FS_BMAP64_RBTREE, "block_ea_map",
2654                                         &ctx->block_ea_map);
2655                 if (pctx->errcode) {
2656                         pctx->num = 2;
2657                         fix_problem(ctx, PR_1_ALLOCATE_BBITMAP_ERROR, pctx);
2658                         ctx->flags |= E2F_FLAG_ABORT;
2659                         return 0;
2660                 }
2661         }
2662
2663         /* Create the EA refcount structure if necessary */
2664         if (!ctx->refcount) {
2665                 pctx->errcode = ea_refcount_create(0, &ctx->refcount);
2666                 if (pctx->errcode) {
2667                         pctx->num = 1;
2668                         fix_problem(ctx, PR_1_ALLOCATE_REFCOUNT, pctx);
2669                         ctx->flags |= E2F_FLAG_ABORT;
2670                         return 0;
2671                 }
2672         }
2673
2674 #if 0
2675         /* Debugging text */
2676         printf("Inode %u has EA block %u\n", ino, blk);
2677 #endif
2678
2679         /* Have we seen this EA block before? */
2680         if (ext2fs_fast_test_block_bitmap2(ctx->block_ea_map, blk)) {
2681                 ea_block_quota->blocks = EXT2FS_C2B(fs, 1);
2682                 ea_block_quota->inodes = 0;
2683
2684                 if (ctx->ea_block_quota_blocks) {
2685                         ea_refcount_fetch(ctx->ea_block_quota_blocks, blk,
2686                                           &quota_blocks);
2687                         if (quota_blocks)
2688                                 ea_block_quota->blocks = quota_blocks;
2689                 }
2690
2691                 if (ctx->ea_block_quota_inodes)
2692                         ea_refcount_fetch(ctx->ea_block_quota_inodes, blk,
2693                                           &ea_block_quota->inodes);
2694
2695                 if (ea_refcount_decrement(ctx->refcount, blk, 0) == 0)
2696                         return 1;
2697                 /* Ooops, this EA was referenced more than it stated */
2698                 if (!ctx->refcount_extra) {
2699                         pctx->errcode = ea_refcount_create(0,
2700                                            &ctx->refcount_extra);
2701                         if (pctx->errcode) {
2702                                 pctx->num = 2;
2703                                 fix_problem(ctx, PR_1_ALLOCATE_REFCOUNT, pctx);
2704                                 ctx->flags |= E2F_FLAG_ABORT;
2705                                 return 0;
2706                         }
2707                 }
2708                 ea_refcount_increment(ctx->refcount_extra, blk, 0);
2709                 return 1;
2710         }
2711
2712         /*
2713          * OK, we haven't seen this EA block yet.  So we need to
2714          * validate it
2715          */
2716         pctx->blk = blk;
2717         pctx->errcode = ext2fs_read_ext_attr3(fs, blk, block_buf, pctx->ino);
2718         if (pctx->errcode == EXT2_ET_EXT_ATTR_CSUM_INVALID) {
2719                 pctx->errcode = 0;
2720                 failed_csum = 1;
2721         } else if (pctx->errcode == EXT2_ET_BAD_EA_HEADER)
2722                 pctx->errcode = 0;
2723
2724         if (pctx->errcode &&
2725             fix_problem(ctx, PR_1_READ_EA_BLOCK, pctx)) {
2726                 pctx->errcode = 0;
2727                 goto clear_extattr;
2728         }
2729         header = (struct ext2_ext_attr_header *) block_buf;
2730         pctx->blk = ext2fs_file_acl_block(fs, inode);
2731         if (((ctx->ext_attr_ver == 1) &&
2732              (header->h_magic != EXT2_EXT_ATTR_MAGIC_v1)) ||
2733             ((ctx->ext_attr_ver == 2) &&
2734              (header->h_magic != EXT2_EXT_ATTR_MAGIC))) {
2735                 if (fix_problem(ctx, PR_1_BAD_EA_BLOCK, pctx))
2736                         goto clear_extattr;
2737         }
2738
2739         if (header->h_blocks != 1) {
2740                 if (fix_problem(ctx, PR_1_EA_MULTI_BLOCK, pctx))
2741                         goto clear_extattr;
2742         }
2743
2744         if (pctx->errcode && fix_problem(ctx, PR_1_READ_EA_BLOCK, pctx))
2745                 goto clear_extattr;
2746
2747         region = region_create(0, fs->blocksize);
2748         if (!region) {
2749                 fix_problem(ctx, PR_1_EA_ALLOC_REGION_ABORT, pctx);
2750                 ctx->flags |= E2F_FLAG_ABORT;
2751                 return 0;
2752         }
2753         if (region_allocate(region, 0, sizeof(struct ext2_ext_attr_header))) {
2754                 if (fix_problem(ctx, PR_1_EA_ALLOC_COLLISION, pctx))
2755                         goto clear_extattr;
2756         }
2757
2758         first = (struct ext2_ext_attr_entry *)(header+1);
2759         end = block_buf + fs->blocksize;
2760         entry = first;
2761         while ((char *)entry < end && *(__u32 *)entry) {
2762                 __u32 hash;
2763
2764                 if (region_allocate(region, (char *)entry - (char *)header,
2765                                    EXT2_EXT_ATTR_LEN(entry->e_name_len))) {
2766                         if (fix_problem(ctx, PR_1_EA_ALLOC_COLLISION, pctx))
2767                                 goto clear_extattr;
2768                         break;
2769                 }
2770                 if ((ctx->ext_attr_ver == 1 &&
2771                      (entry->e_name_len == 0 || entry->e_name_index != 0)) ||
2772                     (ctx->ext_attr_ver == 2 &&
2773                      entry->e_name_index == 0)) {
2774                         if (fix_problem(ctx, PR_1_EA_BAD_NAME, pctx))
2775                                 goto clear_extattr;
2776                         break;
2777                 }
2778                 if (entry->e_value_inum == 0) {
2779                         if (entry->e_value_size > EXT2_XATTR_SIZE_MAX ||
2780                             (entry->e_value_offs + entry->e_value_size >
2781                              fs->blocksize)) {
2782                                 if (fix_problem(ctx, PR_1_EA_BAD_VALUE, pctx))
2783                                         goto clear_extattr;
2784                                 break;
2785                         }
2786                         if (entry->e_value_size &&
2787                             region_allocate(region, entry->e_value_offs,
2788                                             EXT2_EXT_ATTR_SIZE(entry->e_value_size))) {
2789                                 if (fix_problem(ctx, PR_1_EA_ALLOC_COLLISION,
2790                                                 pctx))
2791                                         goto clear_extattr;
2792                         }
2793
2794                         hash = ext2fs_ext_attr_hash_entry(entry, block_buf +
2795                                                           entry->e_value_offs);
2796                         if (entry->e_hash != hash)
2797                                 hash = ext2fs_ext_attr_hash_entry_signed(entry,
2798                                         block_buf + entry->e_value_offs);
2799
2800                         if (entry->e_hash != hash) {
2801                                 pctx->num = entry->e_hash;
2802                                 if (fix_problem(ctx, PR_1_ATTR_HASH, pctx))
2803                                         goto clear_extattr;
2804                                 entry->e_hash = hash;
2805                         }
2806                 } else {
2807                         problem_t problem;
2808                         blk64_t entry_quota_blocks;
2809
2810                         if (!ext2fs_has_feature_ea_inode(fs->super) &&
2811                             fix_problem(ctx, PR_1_EA_INODE_FEATURE, pctx)) {
2812                                 ext2fs_set_feature_ea_inode(fs->super);
2813                                 ext2fs_mark_super_dirty(fs);
2814                         }
2815
2816                         problem = check_large_ea_inode(ctx, entry, pctx,
2817                                                        &entry_quota_blocks);
2818                         if (problem && fix_problem(ctx, problem, pctx))
2819                                 goto clear_extattr;
2820
2821                         quota_blocks += entry_quota_blocks;
2822                         quota_inodes++;
2823                 }
2824
2825                 entry = EXT2_EXT_ATTR_NEXT(entry);
2826         }
2827         if (region_allocate(region, (char *)entry - (char *)header, 4)) {
2828                 if (fix_problem(ctx, PR_1_EA_ALLOC_COLLISION, pctx))
2829                         goto clear_extattr;
2830         }
2831         region_free(region);
2832
2833         /*
2834          * We only get here if there was no other errors that were fixed.
2835          * If there was a checksum fail, ask to correct it.
2836          */
2837         if (failed_csum &&
2838             fix_problem(ctx, PR_1_EA_BLOCK_ONLY_CSUM_INVALID, pctx)) {
2839                 pctx->errcode = ext2fs_write_ext_attr3(fs, blk, block_buf,
2840                                                        pctx->ino);
2841                 if (pctx->errcode)
2842                         return 0;
2843         }
2844
2845         if (quota_blocks != EXT2FS_C2B(fs, 1U)) {
2846                 if (!ctx->ea_block_quota_blocks) {
2847                         pctx->errcode = ea_refcount_create(0,
2848                                                 &ctx->ea_block_quota_blocks);
2849                         if (pctx->errcode) {
2850                                 pctx->num = 3;
2851                                 goto refcount_fail;
2852                         }
2853                 }
2854                 ea_refcount_store(ctx->ea_block_quota_blocks, blk,
2855                                   quota_blocks);
2856         }
2857
2858         if (quota_inodes) {
2859                 if (!ctx->ea_block_quota_inodes) {
2860                         pctx->errcode = ea_refcount_create(0,
2861                                                 &ctx->ea_block_quota_inodes);
2862                         if (pctx->errcode) {
2863                                 pctx->num = 4;
2864 refcount_fail:
2865                                 fix_problem(ctx, PR_1_ALLOCATE_REFCOUNT, pctx);
2866                                 ctx->flags |= E2F_FLAG_ABORT;
2867                                 return 0;
2868                         }
2869                 }
2870
2871                 ea_refcount_store(ctx->ea_block_quota_inodes, blk,
2872                                   quota_inodes);
2873         }
2874         ea_block_quota->blocks = quota_blocks;
2875         ea_block_quota->inodes = quota_inodes;
2876
2877         inc_ea_inode_refs(ctx, pctx, first, end);
2878         ea_refcount_store(ctx->refcount, blk, header->h_refcount - 1);
2879         mark_block_used(ctx, blk);
2880         ext2fs_fast_mark_block_bitmap2(ctx->block_ea_map, blk);
2881         return 1;
2882
2883 clear_extattr:
2884         if (region)
2885                 region_free(region);
2886         ext2fs_file_acl_block_set(fs, inode, 0);
2887         e2fsck_write_inode(ctx, ino, inode, "check_ext_attr");
2888         return 0;
2889 }
2890
2891 /* Returns 1 if bad htree, 0 if OK */
2892 static int handle_htree(e2fsck_t ctx, struct problem_context *pctx,
2893                         ext2_ino_t ino, struct ext2_inode *inode,
2894                         char *block_buf)
2895 {
2896         struct ext2_dx_root_info        *root;
2897         ext2_filsys                     fs = ctx->fs;
2898         errcode_t                       retval;
2899         blk64_t                         blk;
2900
2901         if ((!LINUX_S_ISDIR(inode->i_mode) &&
2902              fix_problem(ctx, PR_1_HTREE_NODIR, pctx)) ||
2903             (!ext2fs_has_feature_dir_index(fs->super) &&
2904              fix_problem(ctx, PR_1_HTREE_SET, pctx)))
2905                 return 1;
2906
2907         pctx->errcode = ext2fs_bmap2(fs, ino, inode, 0, 0, 0, 0, &blk);
2908
2909         if ((pctx->errcode) ||
2910             (blk == 0) ||
2911             (blk < fs->super->s_first_data_block) ||
2912             (blk >= ext2fs_blocks_count(fs->super))) {
2913                 if (fix_problem(ctx, PR_1_HTREE_BADROOT, pctx))
2914                         return 1;
2915                 else
2916                         return 0;
2917         }
2918
2919         retval = io_channel_read_blk64(fs->io, blk, 1, block_buf);
2920         if (retval && fix_problem(ctx, PR_1_HTREE_BADROOT, pctx))
2921                 return 1;
2922
2923         /* XXX should check that beginning matches a directory */
2924         root = (struct ext2_dx_root_info *) (block_buf + 24);
2925
2926         if ((root->reserved_zero || root->info_length < 8) &&
2927             fix_problem(ctx, PR_1_HTREE_BADROOT, pctx))
2928                 return 1;
2929
2930         pctx->num = root->hash_version;
2931         if ((root->hash_version != EXT2_HASH_LEGACY) &&
2932             (root->hash_version != EXT2_HASH_HALF_MD4) &&
2933             (root->hash_version != EXT2_HASH_TEA) &&
2934             (root->hash_version != EXT2_HASH_SIPHASH) &&
2935             fix_problem(ctx, PR_1_HTREE_HASHV, pctx))
2936                 return 1;
2937
2938         if (ext4_hash_in_dirent(inode)) {
2939                 if (root->hash_version != EXT2_HASH_SIPHASH &&
2940                     fix_problem(ctx, PR_1_HTREE_NEEDS_SIPHASH, pctx))
2941                         return 1;
2942         } else {
2943                 if (root->hash_version == EXT2_HASH_SIPHASH &&
2944                    fix_problem(ctx, PR_1_HTREE_CANNOT_SIPHASH, pctx))
2945                         return 1;
2946         }
2947
2948         if ((root->unused_flags & EXT2_HASH_FLAG_INCOMPAT) &&
2949             fix_problem(ctx, PR_1_HTREE_INCOMPAT, pctx))
2950                 return 1;
2951
2952         pctx->num = root->indirect_levels;
2953         /* if htree level is clearly too high, consider it to be broken */
2954         if (root->indirect_levels > EXT4_HTREE_LEVEL &&
2955             fix_problem(ctx, PR_1_HTREE_DEPTH, pctx))
2956                 return 1;
2957
2958         /* if level is only maybe too high, LARGE_DIR feature could be unset */
2959         if (root->indirect_levels > ext2_dir_htree_level(fs) &&
2960             !ext2fs_has_feature_largedir(fs->super)) {
2961                 int blockbits = EXT2_BLOCK_SIZE_BITS(fs->super) + 10;
2962                 unsigned idx_pb = 1 << (blockbits - 3);
2963
2964                 /* compare inode size/blocks vs. max-sized 2-level htree */
2965                 if (EXT2_I_SIZE(pctx->inode) <
2966                     (idx_pb - 1) * (idx_pb - 2) << blockbits &&
2967                     pctx->inode->i_blocks <
2968                     (idx_pb - 1) * (idx_pb - 2) << (blockbits - 9) &&
2969                     fix_problem(ctx, PR_1_HTREE_DEPTH, pctx))
2970                         return 1;
2971         }
2972
2973         if (root->indirect_levels > EXT4_HTREE_LEVEL_COMPAT ||
2974             ext2fs_needs_large_file_feature(EXT2_I_SIZE(inode)))
2975                 ctx->large_dirs++;
2976
2977         return 0;
2978 }
2979
2980 void e2fsck_clear_inode(e2fsck_t ctx, ext2_ino_t ino,
2981                         struct ext2_inode *inode, int restart_flag,
2982                         const char *source)
2983 {
2984         inode->i_flags = 0;
2985         inode->i_links_count = 0;
2986         ext2fs_icount_store(ctx->inode_link_info, ino, 0);
2987         ext2fs_set_dtime(ctx->fs, inode);
2988
2989         /*
2990          * If a special inode has such rotten block mappings that we
2991          * want to clear the whole inode, be sure to actually zap
2992          * the block maps because i_links_count isn't checked for
2993          * special inodes, and we'll end up right back here the next
2994          * time we run fsck.
2995          */
2996         if (ino < EXT2_FIRST_INODE(ctx->fs->super))
2997                 memset(inode->i_block, 0, sizeof(inode->i_block));
2998
2999         ext2fs_unmark_inode_bitmap2(ctx->inode_dir_map, ino);
3000         ext2fs_unmark_inode_bitmap2(ctx->inode_used_map, ino);
3001         if (ctx->inode_reg_map)
3002                 ext2fs_unmark_inode_bitmap2(ctx->inode_reg_map, ino);
3003         if (ctx->inode_bad_map)
3004                 ext2fs_unmark_inode_bitmap2(ctx->inode_bad_map, ino);
3005
3006         /*
3007          * If the inode was partially accounted for before processing
3008          * was aborted, we need to restart the pass 1 scan.
3009          */
3010         ctx->flags |= restart_flag;
3011
3012         if (ino == EXT2_BAD_INO)
3013                 memset(inode, 0, sizeof(struct ext2_inode));
3014
3015         e2fsck_write_inode(ctx, ino, inode, source);
3016 }
3017
3018 /*
3019  * Use the multiple-blocks reclamation code to fix alignment problems in
3020  * a bigalloc filesystem.  We want a logical cluster to map to *only* one
3021  * physical cluster, and we want the block offsets within that cluster to
3022  * line up.
3023  */
3024 static int has_unaligned_cluster_map(e2fsck_t ctx,
3025                                      blk64_t last_pblk, blk64_t last_lblk,
3026                                      blk64_t pblk, blk64_t lblk)
3027 {
3028         blk64_t cluster_mask;
3029
3030         if (!ctx->fs->cluster_ratio_bits)
3031                 return 0;
3032         cluster_mask = EXT2FS_CLUSTER_MASK(ctx->fs);
3033
3034         /*
3035          * If the block in the logical cluster doesn't align with the block in
3036          * the physical cluster...
3037          */
3038         if ((lblk & cluster_mask) != (pblk & cluster_mask))
3039                 return 1;
3040
3041         /*
3042          * If we cross a physical cluster boundary within a logical cluster...
3043          */
3044         if (last_pblk && (lblk & cluster_mask) != 0 &&
3045             EXT2FS_B2C(ctx->fs, lblk) == EXT2FS_B2C(ctx->fs, last_lblk) &&
3046             EXT2FS_B2C(ctx->fs, pblk) != EXT2FS_B2C(ctx->fs, last_pblk))
3047                 return 1;
3048
3049         return 0;
3050 }
3051
3052 static void scan_extent_node(e2fsck_t ctx, struct problem_context *pctx,
3053                              struct process_block_struct *pb,
3054                              blk64_t start_block, blk64_t end_block,
3055                              blk64_t eof_block,
3056                              ext2_extent_handle_t ehandle,
3057                              int try_repairs)
3058 {
3059         struct ext2fs_extent    extent;
3060         blk64_t                 blk, last_lblk;
3061         unsigned int            i, n;
3062         int                     is_dir, is_leaf;
3063         problem_t               problem;
3064         struct ext2_extent_info info;
3065         int                     failed_csum = 0;
3066
3067         if (pctx->errcode == EXT2_ET_EXTENT_CSUM_INVALID)
3068                 failed_csum = 1;
3069
3070         pctx->errcode = ext2fs_extent_get_info(ehandle, &info);
3071         if (pctx->errcode)
3072                 return;
3073         if (!(ctx->options & E2F_OPT_FIXES_ONLY) &&
3074             !pb->eti.force_rebuild &&
3075             info.curr_level < MAX_EXTENT_DEPTH_COUNT) {
3076                 struct extent_tree_level *etl;
3077
3078                 etl = pb->eti.ext_info + info.curr_level;
3079                 etl->num_extents += info.num_entries;
3080                 etl->max_extents += info.max_entries;
3081                 /*
3082                  * Implementation wart: Splitting extent blocks when appending
3083                  * will leave the old block with one free entry.  Therefore
3084                  * unless the node is totally full, pretend that a non-root
3085                  * extent block can hold one fewer entry than it actually does,
3086                  * so that we don't repeatedly rebuild the extent tree.
3087                  */
3088                 if (info.curr_level && info.num_entries < info.max_entries)
3089                         etl->max_extents--;
3090         }
3091
3092         pctx->errcode = ext2fs_extent_get(ehandle, EXT2_EXTENT_FIRST_SIB,
3093                                           &extent);
3094         while ((pctx->errcode == 0 ||
3095                 pctx->errcode == EXT2_ET_EXTENT_CSUM_INVALID) &&
3096                info.num_entries-- > 0) {
3097                 is_leaf = extent.e_flags & EXT2_EXTENT_FLAGS_LEAF;
3098                 is_dir = LINUX_S_ISDIR(pctx->inode->i_mode);
3099                 last_lblk = extent.e_lblk + extent.e_len - 1;
3100
3101                 problem = 0;
3102                 pctx->blk = extent.e_pblk;
3103                 pctx->blk2 = extent.e_lblk;
3104                 pctx->num = extent.e_len;
3105                 pctx->blkcount = extent.e_lblk + extent.e_len;
3106
3107                 if (extent.e_pblk == 0 ||
3108                     extent.e_pblk < ctx->fs->super->s_first_data_block ||
3109                     extent.e_pblk >= ext2fs_blocks_count(ctx->fs->super))
3110                         problem = PR_1_EXTENT_BAD_START_BLK;
3111                 else if (extent.e_lblk < start_block)
3112                         problem = PR_1_OUT_OF_ORDER_EXTENTS;
3113                 else if ((end_block && last_lblk > end_block) &&
3114                          !(last_lblk > eof_block &&
3115                            ((extent.e_flags & EXT2_EXTENT_FLAGS_UNINIT) ||
3116                             (pctx->inode->i_flags & EXT4_VERITY_FL))))
3117                         problem = PR_1_EXTENT_END_OUT_OF_BOUNDS;
3118                 else if (is_leaf && extent.e_len == 0)
3119                         problem = PR_1_EXTENT_LENGTH_ZERO;
3120                 else if (is_leaf &&
3121                          (extent.e_pblk + extent.e_len) >
3122                          ext2fs_blocks_count(ctx->fs->super))
3123                         problem = PR_1_EXTENT_ENDS_BEYOND;
3124                 else if (is_leaf && is_dir && !pctx->inode->i_size_high &&
3125                          !ext2fs_has_feature_largedir(ctx->fs->super) &&
3126                          ((extent.e_lblk + extent.e_len) >
3127                           (1U << (21 - ctx->fs->super->s_log_block_size))))
3128                         problem = PR_1_TOOBIG_DIR;
3129
3130                 if (is_leaf && problem == 0 && extent.e_len > 0) {
3131 #if 0
3132                         printf("extent_region(ino=%u, expect=%llu, "
3133                                "lblk=%llu, len=%u)\n", pb->ino,
3134                                (unsigned long long) pb->next_lblock,
3135                                (unsigned long long) extent.e_lblk,
3136                                extent.e_len);
3137 #endif
3138                         if (extent.e_lblk < pb->next_lblock)
3139                                 problem = PR_1_EXTENT_COLLISION;
3140                         else if (extent.e_lblk + extent.e_len > pb->next_lblock)
3141                                 pb->next_lblock = extent.e_lblk + extent.e_len;
3142                 }
3143
3144                 /*
3145                  * Uninitialized blocks in a directory?  Clear the flag and
3146                  * we'll interpret the blocks later.
3147                  */
3148                 if (try_repairs && is_dir && problem == 0 &&
3149                     (extent.e_flags & EXT2_EXTENT_FLAGS_UNINIT) &&
3150                     fix_problem(ctx, PR_1_UNINIT_DBLOCK, pctx)) {
3151                         extent.e_flags &= ~EXT2_EXTENT_FLAGS_UNINIT;
3152                         pb->inode_modified = 1;
3153                         pctx->errcode = ext2fs_extent_replace(ehandle, 0,
3154                                                               &extent);
3155                         if (pctx->errcode)
3156                                 return;
3157                         failed_csum = 0;
3158                 }
3159 #ifdef CONFIG_DEVELOPER_FEATURES
3160                 if (try_repairs && !is_dir && problem == 0 &&
3161                     (ctx->options & E2F_OPT_CLEAR_UNINIT) &&
3162                     (extent.e_flags & EXT2_EXTENT_FLAGS_UNINIT) &&
3163                     fix_problem(ctx, PR_1_CLEAR_UNINIT_EXTENT, pctx)) {
3164                         extent.e_flags &= ~EXT2_EXTENT_FLAGS_UNINIT;
3165                         pb->inode_modified = 1;
3166                         pctx->errcode = ext2fs_extent_replace(ehandle, 0,
3167                                                               &extent);
3168                         if (pctx->errcode)
3169                                 return;
3170                         failed_csum = 0;
3171                 }
3172 #endif
3173                 if (try_repairs && problem) {
3174 report_problem:
3175                         if (fix_problem(ctx, problem, pctx)) {
3176                                 if (ctx->invalid_bitmaps) {
3177                                         /*
3178                                          * If fsck knows the bitmaps are bad,
3179                                          * skip to the next extent and
3180                                          * try to clear this extent again
3181                                          * after fixing the bitmaps, by
3182                                          * restarting fsck.
3183                                          */
3184                                         pctx->errcode = ext2fs_extent_get(
3185                                                           ehandle,
3186                                                           EXT2_EXTENT_NEXT_SIB,
3187                                                           &extent);
3188                                         ctx->flags |= E2F_FLAG_RESTART_LATER;
3189                                         if (pctx->errcode ==
3190                                                     EXT2_ET_NO_CURRENT_NODE) {
3191                                                 pctx->errcode = 0;
3192                                                 break;
3193                                         }
3194                                         continue;
3195                                 }
3196                                 e2fsck_read_bitmaps(ctx);
3197                                 pb->inode_modified = 1;
3198                                 pctx->errcode =
3199                                         ext2fs_extent_delete(ehandle, 0);
3200                                 if (pctx->errcode) {
3201                                         pctx->str = "ext2fs_extent_delete";
3202                                         return;
3203                                 }
3204                                 pctx->errcode = ext2fs_extent_fix_parents(ehandle);
3205                                 if (pctx->errcode &&
3206                                     pctx->errcode != EXT2_ET_NO_CURRENT_NODE) {
3207                                         pctx->str = "ext2fs_extent_fix_parents";
3208                                         return;
3209                                 }
3210                                 pctx->errcode = ext2fs_extent_get(ehandle,
3211                                                                   EXT2_EXTENT_CURRENT,
3212                                                                   &extent);
3213                                 if (pctx->errcode == EXT2_ET_NO_CURRENT_NODE) {
3214                                         pctx->errcode = 0;
3215                                         break;
3216                                 }
3217                                 failed_csum = 0;
3218                                 continue;
3219                         }
3220                         goto next;
3221                 }
3222
3223                 if (!is_leaf) {
3224                         blk64_t lblk = extent.e_lblk;
3225                         int next_try_repairs = 1;
3226
3227                         blk = extent.e_pblk;
3228
3229                         /*
3230                          * If this lower extent block collides with critical
3231                          * metadata, don't try to repair the damage.  Pass 1b
3232                          * will reallocate the block; then we can try again.
3233                          */
3234                         if (pb->ino != EXT2_RESIZE_INO &&
3235                             extent.e_pblk < ctx->fs->super->s_blocks_count &&
3236                             ext2fs_test_block_bitmap2(ctx->block_metadata_map,
3237                                                       extent.e_pblk)) {
3238                                 next_try_repairs = 0;
3239                                 pctx->blk = blk;
3240                                 fix_problem(ctx,
3241                                             PR_1_CRITICAL_METADATA_COLLISION,
3242                                             pctx);
3243                                 if ((ctx->options & E2F_OPT_NO) == 0)
3244                                         ctx->flags |= E2F_FLAG_RESTART_LATER;
3245                         }
3246                         pctx->errcode = ext2fs_extent_get(ehandle,
3247                                                   EXT2_EXTENT_DOWN, &extent);
3248                         if (pctx->errcode &&
3249                             pctx->errcode != EXT2_ET_EXTENT_CSUM_INVALID) {
3250                                 pctx->str = "EXT2_EXTENT_DOWN";
3251                                 problem = PR_1_EXTENT_HEADER_INVALID;
3252                                 if (!next_try_repairs)
3253                                         return;
3254                                 if (pctx->errcode == EXT2_ET_EXTENT_HEADER_BAD)
3255                                         goto report_problem;
3256                                 return;
3257                         }
3258                         /* The next extent should match this index's logical start */
3259                         if (extent.e_lblk != lblk) {
3260                                 struct ext2_extent_info e_info;
3261
3262                                 pctx->errcode = ext2fs_extent_get_info(ehandle,
3263                                                                        &e_info);
3264                                 if (pctx->errcode) {
3265                                         pctx->str = "ext2fs_extent_get_info";
3266                                         return;
3267                                 }
3268                                 pctx->blk = lblk;
3269                                 pctx->blk2 = extent.e_lblk;
3270                                 pctx->num = e_info.curr_level - 1;
3271                                 problem = PR_1_EXTENT_INDEX_START_INVALID;
3272                                 if (fix_problem(ctx, problem, pctx)) {
3273                                         pb->inode_modified = 1;
3274                                         pctx->errcode =
3275                                                 ext2fs_extent_fix_parents(ehandle);
3276                                         if (pctx->errcode) {
3277                                                 pctx->str = "ext2fs_extent_fix_parents";
3278                                                 return;
3279                                         }
3280                                 }
3281                         }
3282                         scan_extent_node(ctx, pctx, pb, extent.e_lblk,
3283                                          last_lblk, eof_block, ehandle,
3284                                          next_try_repairs);
3285                         if (pctx->errcode)
3286                                 return;
3287                         pctx->errcode = ext2fs_extent_get(ehandle,
3288                                                   EXT2_EXTENT_UP, &extent);
3289                         if (pctx->errcode) {
3290                                 pctx->str = "EXT2_EXTENT_UP";
3291                                 return;
3292                         }
3293                         mark_block_used(ctx, blk);
3294                         pb->num_blocks++;
3295                         goto next;
3296                 }
3297
3298                 if ((pb->previous_block != 0) &&
3299                     (pb->previous_block+1 != extent.e_pblk)) {
3300                         if (ctx->options & E2F_OPT_FRAGCHECK) {
3301                                 char type = '?';
3302
3303                                 if (pb->is_dir)
3304                                         type = 'd';
3305                                 else if (pb->is_reg)
3306                                         type = 'f';
3307
3308                                 printf(("%6lu(%c): expecting %6lu "
3309                                         "actual extent "
3310                                         "phys %6lu log %lu len %lu\n"),
3311                                        (unsigned long) pctx->ino, type,
3312                                        (unsigned long) pb->previous_block+1,
3313                                        (unsigned long) extent.e_pblk,
3314                                        (unsigned long) extent.e_lblk,
3315                                        (unsigned long) extent.e_len);
3316                         }
3317                         pb->fragmented = 1;
3318                 }
3319                 /*
3320                  * If we notice a gap in the logical block mappings of an
3321                  * extent-mapped directory, offer to close the hole by
3322                  * moving the logical block down, otherwise we'll go mad in
3323                  * pass 3 allocating empty directory blocks to fill the hole.
3324                  */
3325                 if (try_repairs && is_dir &&
3326                     pb->last_block + 1 < extent.e_lblk) {
3327                         blk64_t new_lblk;
3328
3329                         new_lblk = pb->last_block + 1;
3330                         if (EXT2FS_CLUSTER_RATIO(ctx->fs) > 1)
3331                                 new_lblk = ((new_lblk +
3332                                              EXT2FS_CLUSTER_RATIO(ctx->fs) - 1) &
3333                                             ~EXT2FS_CLUSTER_MASK(ctx->fs)) |
3334                                            (extent.e_pblk &
3335                                             EXT2FS_CLUSTER_MASK(ctx->fs));
3336                         pctx->blk = extent.e_lblk;
3337                         pctx->blk2 = new_lblk;
3338                         if (fix_problem(ctx, PR_1_COLLAPSE_DBLOCK, pctx)) {
3339                                 extent.e_lblk = new_lblk;
3340                                 pb->inode_modified = 1;
3341                                 pctx->errcode = ext2fs_extent_replace(ehandle,
3342                                                                 0, &extent);
3343                                 if (pctx->errcode) {
3344                                         pctx->errcode = 0;
3345                                         goto alloc_later;
3346                                 }
3347                                 pctx->errcode = ext2fs_extent_fix_parents(ehandle);
3348                                 if (pctx->errcode)
3349                                         goto failed_add_dir_block;
3350                                 pctx->errcode = ext2fs_extent_goto(ehandle,
3351                                                                 extent.e_lblk);
3352                                 if (pctx->errcode)
3353                                         goto failed_add_dir_block;
3354                                 last_lblk = extent.e_lblk + extent.e_len - 1;
3355                                 failed_csum = 0;
3356                         }
3357                 }
3358 alloc_later:
3359                 if (is_dir) {
3360                         while (++pb->last_db_block <
3361                                (e2_blkcnt_t) extent.e_lblk) {
3362                                 pctx->errcode = ext2fs_add_dir_block2(
3363                                                         ctx->fs->dblist,
3364                                                         pb->ino, 0,
3365                                                         pb->last_db_block);
3366                                 if (pctx->errcode) {
3367                                         pctx->blk = 0;
3368                                         pctx->num = pb->last_db_block;
3369                                         goto failed_add_dir_block;
3370                                 }
3371                         }
3372
3373                         for (i = 0; i < extent.e_len; i++) {
3374                                 pctx->errcode = ext2fs_add_dir_block2(
3375                                                         ctx->fs->dblist,
3376                                                         pctx->ino,
3377                                                         extent.e_pblk + i,
3378                                                         extent.e_lblk + i);
3379                                 if (pctx->errcode) {
3380                                         pctx->blk = extent.e_pblk + i;
3381                                         pctx->num = extent.e_lblk + i;
3382                                 failed_add_dir_block:
3383                                         fix_problem(ctx, PR_1_ADD_DBLOCK, pctx);
3384                                         /* Should never get here */
3385                                         ctx->flags |= E2F_FLAG_ABORT;
3386                                         return;
3387                                 }
3388                         }
3389                         if (extent.e_len > 0)
3390                                 pb->last_db_block = extent.e_lblk + extent.e_len - 1;
3391                 }
3392                 if (has_unaligned_cluster_map(ctx, pb->previous_block,
3393                                               pb->last_block,
3394                                               extent.e_pblk,
3395                                               extent.e_lblk)) {
3396                         for (i = 0; i < extent.e_len; i++) {
3397                                 pctx->blk = extent.e_lblk + i;
3398                                 pctx->blk2 = extent.e_pblk + i;
3399                                 fix_problem(ctx, PR_1_MISALIGNED_CLUSTER, pctx);
3400                                 mark_block_used(ctx, extent.e_pblk + i);
3401                                 mark_block_used(ctx, extent.e_pblk + i);
3402                         }
3403                 }
3404
3405                 /*
3406                  * Check whether first cluster got marked in previous iteration.
3407                  */
3408                 if (ctx->fs->cluster_ratio_bits &&
3409                     pb->previous_block &&
3410                     (EXT2FS_B2C(ctx->fs, extent.e_pblk) ==
3411                      EXT2FS_B2C(ctx->fs, pb->previous_block)))
3412                         /* Set blk to the beginning of next cluster. */
3413                         blk = EXT2FS_C2B(
3414                                 ctx->fs,
3415                                 EXT2FS_B2C(ctx->fs, extent.e_pblk) + 1);
3416                 else
3417                         /* Set blk to the beginning of current cluster. */
3418                         blk = EXT2FS_C2B(ctx->fs,
3419                                          EXT2FS_B2C(ctx->fs, extent.e_pblk));
3420
3421                 if (blk < extent.e_pblk + extent.e_len) {
3422                         mark_blocks_used(ctx, blk,
3423                                          extent.e_pblk + extent.e_len - blk);
3424                         n = DIV_ROUND_UP(extent.e_pblk + extent.e_len - blk,
3425                                          EXT2FS_CLUSTER_RATIO(ctx->fs));
3426                         pb->num_blocks += n;
3427                 }
3428                 pb->last_block = extent.e_lblk + extent.e_len - 1;
3429                 pb->previous_block = extent.e_pblk + extent.e_len - 1;
3430                 start_block = pb->last_block = last_lblk;
3431                 if (is_leaf && !is_dir &&
3432                     !(extent.e_flags & EXT2_EXTENT_FLAGS_UNINIT))
3433                         pb->last_init_lblock = last_lblk;
3434         next:
3435                 pctx->errcode = ext2fs_extent_get(ehandle,
3436                                                   EXT2_EXTENT_NEXT_SIB,
3437                                                   &extent);
3438         }
3439
3440         /* Failed csum but passes checks?  Ask to fix checksum. */
3441         if (failed_csum &&
3442             fix_problem(ctx, PR_1_EXTENT_ONLY_CSUM_INVALID, pctx)) {
3443                 pb->inode_modified = 1;
3444                 pctx->errcode = ext2fs_extent_replace(ehandle, 0, &extent);
3445                 if (pctx->errcode)
3446                         return;
3447         }
3448
3449         if (pctx->errcode == EXT2_ET_EXTENT_NO_NEXT)
3450                 pctx->errcode = 0;
3451 }
3452
3453 static void check_blocks_extents(e2fsck_t ctx, struct problem_context *pctx,
3454                                  struct process_block_struct *pb)
3455 {
3456         struct ext2_extent_info info;
3457         struct ext2_inode       *inode = pctx->inode;
3458         ext2_extent_handle_t    ehandle;
3459         ext2_filsys             fs = ctx->fs;
3460         ext2_ino_t              ino = pctx->ino;
3461         errcode_t               retval;
3462         blk64_t                 eof_lblk;
3463         struct ext3_extent_header       *eh;
3464
3465         /* Check for a proper extent header... */
3466         eh = (struct ext3_extent_header *) &inode->i_block[0];
3467         retval = ext2fs_extent_header_verify(eh, sizeof(inode->i_block));
3468         if (retval) {
3469                 if (fix_problem(ctx, PR_1_MISSING_EXTENT_HEADER, pctx))
3470                         e2fsck_clear_inode(ctx, ino, inode, 0,
3471                                            "check_blocks_extents");
3472                 pctx->errcode = 0;
3473                 return;
3474         }
3475
3476         /* ...since this function doesn't fail if i_block is zeroed. */
3477         pctx->errcode = ext2fs_extent_open2(fs, ino, inode, &ehandle);
3478         if (pctx->errcode) {
3479                 if (fix_problem(ctx, PR_1_READ_EXTENT, pctx))
3480                         e2fsck_clear_inode(ctx, ino, inode, 0,
3481                                            "check_blocks_extents");
3482                 pctx->errcode = 0;
3483                 return;
3484         }
3485
3486         retval = ext2fs_extent_get_info(ehandle, &info);
3487         if (retval == 0) {
3488                 int max_depth = info.max_depth;
3489
3490                 if (max_depth >= MAX_EXTENT_DEPTH_COUNT)
3491                         max_depth = MAX_EXTENT_DEPTH_COUNT-1;
3492                 ctx->extent_depth_count[max_depth]++;
3493         }
3494
3495         /* Check maximum extent depth */
3496         pctx->blk = info.max_depth;
3497         pctx->blk2 = ext2fs_max_extent_depth(ehandle);
3498         if (pctx->blk2 < pctx->blk &&
3499             fix_problem(ctx, PR_1_EXTENT_BAD_MAX_DEPTH, pctx))
3500                 pb->eti.force_rebuild = 1;
3501
3502         /* Can we collect extent tree level stats? */
3503         pctx->blk = MAX_EXTENT_DEPTH_COUNT;
3504         if (pctx->blk2 > pctx->blk)
3505                 fix_problem(ctx, PR_1E_MAX_EXTENT_TREE_DEPTH, pctx);
3506         memset(pb->eti.ext_info, 0, sizeof(pb->eti.ext_info));
3507         pb->eti.ino = pb->ino;
3508
3509         pb->next_lblock = 0;
3510
3511         eof_lblk = ((EXT2_I_SIZE(inode) + fs->blocksize - 1) >>
3512                 EXT2_BLOCK_SIZE_BITS(fs->super)) - 1;
3513         scan_extent_node(ctx, pctx, pb, 0, 0, eof_lblk, ehandle, 1);
3514         if (pctx->errcode &&
3515             fix_problem(ctx, PR_1_EXTENT_ITERATE_FAILURE, pctx)) {
3516                 pb->num_blocks = 0;
3517                 inode->i_blocks = 0;
3518                 e2fsck_clear_inode(ctx, ino, inode, E2F_FLAG_RESTART,
3519                                    "check_blocks_extents");
3520                 pctx->errcode = 0;
3521         }
3522         ext2fs_extent_free(ehandle);
3523
3524         /* Rebuild unless it's a dir and we're rehashing it */
3525         if (LINUX_S_ISDIR(inode->i_mode) &&
3526             e2fsck_dir_will_be_rehashed(ctx, ino))
3527                 return;
3528
3529         if (ctx->options & E2F_OPT_CONVERT_BMAP)
3530                 e2fsck_rebuild_extents_later(ctx, ino);
3531         else
3532                 e2fsck_should_rebuild_extents(ctx, pctx, &pb->eti, &info);
3533 }
3534
3535 /*
3536  * In fact we don't need to check blocks for an inode with inline data
3537  * because this inode doesn't have any blocks.  In this function all
3538  * we need to do is add this inode into dblist when it is a directory.
3539  */
3540 static void check_blocks_inline_data(e2fsck_t ctx, struct problem_context *pctx,
3541                                      struct process_block_struct *pb)
3542 {
3543         int     flags;
3544         size_t  inline_data_size = 0;
3545
3546         if (!pb->is_dir) {
3547                 pctx->errcode = 0;
3548                 return;
3549         }
3550
3551         /* Process the dirents in i_block[] as the "first" block. */
3552         pctx->errcode = ext2fs_add_dir_block2(ctx->fs->dblist, pb->ino, 0, 0);
3553         if (pctx->errcode)
3554                 goto err;
3555
3556         /* Process the dirents in the EA as a "second" block. */
3557         flags = ctx->fs->flags;
3558         ctx->fs->flags |= EXT2_FLAG_IGNORE_CSUM_ERRORS;
3559         pctx->errcode = ext2fs_inline_data_size(ctx->fs, pb->ino,
3560                                                 &inline_data_size);
3561         ctx->fs->flags = (flags & EXT2_FLAG_IGNORE_CSUM_ERRORS) |
3562                          (ctx->fs->flags & ~EXT2_FLAG_IGNORE_CSUM_ERRORS);
3563         if (pctx->errcode) {
3564                 pctx->errcode = 0;
3565                 return;
3566         }
3567
3568         if (inline_data_size <= EXT4_MIN_INLINE_DATA_SIZE)
3569                 return;
3570
3571         pctx->errcode = ext2fs_add_dir_block2(ctx->fs->dblist, pb->ino, 0, 1);
3572         if (pctx->errcode)
3573                 goto err;
3574
3575         return;
3576 err:
3577         pctx->blk = 0;
3578         pctx->num = 0;
3579         fix_problem(ctx, PR_1_ADD_DBLOCK, pctx);
3580         ctx->flags |= E2F_FLAG_ABORT;
3581 }
3582
3583 /*
3584  * This subroutine is called on each inode to account for all of the
3585  * blocks used by that inode.
3586  */
3587 static void check_blocks(e2fsck_t ctx, struct problem_context *pctx,
3588                          char *block_buf, const struct ea_quota *ea_ibody_quota)
3589 {
3590         ext2_filsys fs = ctx->fs;
3591         struct process_block_struct pb;
3592         ext2_ino_t      ino = pctx->ino;
3593         struct ext2_inode *inode = pctx->inode;
3594         unsigned        bad_size = 0;
3595         int             dirty_inode = 0;
3596         int             extent_fs;
3597         int             inlinedata_fs;
3598         __u64           size;
3599         struct ea_quota ea_block_quota;
3600
3601         pb.ino = ino;
3602         pb.num_blocks = EXT2FS_B2C(ctx->fs,
3603                                    ea_ibody_quota ? ea_ibody_quota->blocks : 0);
3604         pb.last_block = ~0;
3605         pb.last_init_lblock = -1;
3606         pb.last_db_block = -1;
3607         pb.num_illegal_blocks = 0;
3608         pb.suppress = 0; pb.clear = 0;
3609         pb.fragmented = 0;
3610         pb.compressed = 0;
3611         pb.previous_block = 0;
3612         pb.is_dir = LINUX_S_ISDIR(inode->i_mode);
3613         pb.is_reg = LINUX_S_ISREG(inode->i_mode);
3614         pb.max_blocks = 1U << (31 - fs->super->s_log_block_size);
3615         pb.inode = inode;
3616         pb.pctx = pctx;
3617         pb.ctx = ctx;
3618         pb.inode_modified = 0;
3619         pb.eti.force_rebuild = 0;
3620         pctx->ino = ino;
3621         pctx->errcode = 0;
3622
3623         extent_fs = ext2fs_has_feature_extents(ctx->fs->super);
3624         inlinedata_fs = ext2fs_has_feature_inline_data(ctx->fs->super);
3625
3626         if (check_ext_attr(ctx, pctx, block_buf, &ea_block_quota)) {
3627                 if (e2fsck_should_abort(ctx))
3628                         goto out;
3629                 pb.num_blocks += EXT2FS_B2C(ctx->fs, ea_block_quota.blocks);
3630         }
3631
3632         if (inlinedata_fs && (inode->i_flags & EXT4_INLINE_DATA_FL))
3633                 check_blocks_inline_data(ctx, pctx, &pb);
3634         else if (ext2fs_inode_has_valid_blocks2(fs, inode)) {
3635                 if (extent_fs && (inode->i_flags & EXT4_EXTENTS_FL))
3636                         check_blocks_extents(ctx, pctx, &pb);
3637                 else {
3638                         int flags;
3639                         /*
3640                          * If we've modified the inode, write it out before
3641                          * iterate() tries to use it.
3642                          */
3643                         if (dirty_inode) {
3644                                 e2fsck_write_inode(ctx, ino, inode,
3645                                                    "check_blocks");
3646                                 dirty_inode = 0;
3647                         }
3648                         flags = fs->flags;
3649                         fs->flags |= EXT2_FLAG_IGNORE_CSUM_ERRORS;
3650                         pctx->errcode = ext2fs_block_iterate3(fs, ino,
3651                                                 pb.is_dir ? BLOCK_FLAG_HOLE : 0,
3652                                                 block_buf, process_block, &pb);
3653                         /*
3654                          * We do not have uninitialized extents in non extent
3655                          * files.
3656                          */
3657                         pb.last_init_lblock = pb.last_block;
3658                         /*
3659                          * If iterate() changed a block mapping, we have to
3660                          * re-read the inode.  If we decide to clear the
3661                          * inode after clearing some stuff, we'll re-write the
3662                          * bad mappings into the inode!
3663                          */
3664                         if (pb.inode_modified)
3665                                 e2fsck_read_inode(ctx, ino, inode,
3666                                                   "check_blocks");
3667                         fs->flags = (flags & EXT2_FLAG_IGNORE_CSUM_ERRORS) |
3668                                     (fs->flags & ~EXT2_FLAG_IGNORE_CSUM_ERRORS);
3669
3670                         if (ctx->options & E2F_OPT_CONVERT_BMAP) {
3671 #ifdef DEBUG
3672                                 printf("bmap rebuild ino=%d\n", ino);
3673 #endif
3674                                 if (!LINUX_S_ISDIR(inode->i_mode) ||
3675                                     !e2fsck_dir_will_be_rehashed(ctx, ino))
3676                                         e2fsck_rebuild_extents_later(ctx, ino);
3677                         }
3678                 }
3679         }
3680         end_problem_latch(ctx, PR_LATCH_BLOCK);
3681         end_problem_latch(ctx, PR_LATCH_TOOBIG);
3682         if (e2fsck_should_abort(ctx))
3683                 goto out;
3684         if (pctx->errcode)
3685                 fix_problem(ctx, PR_1_BLOCK_ITERATE, pctx);
3686
3687         if (pb.fragmented && pb.num_blocks < fs->super->s_blocks_per_group) {
3688                 if (LINUX_S_ISDIR(inode->i_mode))
3689                         ctx->fs_fragmented_dir++;
3690                 else
3691                         ctx->fs_fragmented++;
3692         }
3693
3694         if (pb.clear) {
3695                 e2fsck_clear_inode(ctx, ino, inode, E2F_FLAG_RESTART,
3696                                    "check_blocks");
3697                 return;
3698         }
3699
3700         if (inode->i_flags & EXT2_INDEX_FL) {
3701                 if (handle_htree(ctx, pctx, ino, inode, block_buf)) {
3702                         inode->i_flags &= ~EXT2_INDEX_FL;
3703                         dirty_inode++;
3704                 } else {
3705                         e2fsck_add_dx_dir(ctx, ino, inode, pb.last_block+1);
3706                 }
3707         }
3708
3709         if (!pb.num_blocks && pb.is_dir &&
3710             !(inode->i_flags & EXT4_INLINE_DATA_FL)) {
3711                 if (fix_problem(ctx, PR_1_ZERO_LENGTH_DIR, pctx)) {
3712                         e2fsck_clear_inode(ctx, ino, inode, 0, "check_blocks");
3713                         ctx->fs_directory_count--;
3714                         return;
3715                 }
3716         }
3717
3718         if (ino != quota_type2inum(PRJQUOTA, fs->super) &&
3719             ino != fs->super->s_orphan_file_inum &&
3720             (ino == EXT2_ROOT_INO || ino >= EXT2_FIRST_INODE(ctx->fs->super)) &&
3721             !(inode->i_flags & EXT4_EA_INODE_FL)) {
3722                 quota_data_add(ctx->qctx, (struct ext2_inode_large *) inode,
3723                                ino,
3724                                pb.num_blocks * EXT2_CLUSTER_SIZE(fs->super));
3725                 quota_data_inodes(ctx->qctx, (struct ext2_inode_large *) inode,
3726                                   ino, (ea_ibody_quota ?
3727                                         ea_ibody_quota->inodes : 0) +
3728                                                 ea_block_quota.inodes + 1);
3729         }
3730
3731         if (!ext2fs_has_feature_huge_file(fs->super) ||
3732             !(inode->i_flags & EXT4_HUGE_FILE_FL))
3733                 pb.num_blocks *= (fs->blocksize / 512);
3734         pb.num_blocks *= EXT2FS_CLUSTER_RATIO(fs);
3735 #if 0
3736         printf("inode %u, i_size = %u, last_block = %llu, i_blocks=%llu, num_blocks = %llu\n",
3737                ino, inode->i_size, (unsigned long long) pb.last_block,
3738                (unsigned long long) ext2fs_inode_i_blocks(fs, inode),
3739                (unsigned long long) pb.num_blocks);
3740 #endif
3741         size = EXT2_I_SIZE(inode);
3742         if (pb.is_dir) {
3743                 unsigned nblock = size >> EXT2_BLOCK_SIZE_BITS(fs->super);
3744                 if (inode->i_flags & EXT4_INLINE_DATA_FL) {
3745                         int flags;
3746                         size_t sz = 0;
3747                         errcode_t err;
3748
3749                         flags = ctx->fs->flags;
3750                         ctx->fs->flags |= EXT2_FLAG_IGNORE_CSUM_ERRORS;
3751                         err = ext2fs_inline_data_size(ctx->fs, pctx->ino,
3752                                                       &sz);
3753                         ctx->fs->flags = (flags &
3754                                           EXT2_FLAG_IGNORE_CSUM_ERRORS) |
3755                                          (ctx->fs->flags &
3756                                           ~EXT2_FLAG_IGNORE_CSUM_ERRORS);
3757                         if (err || sz != size) {
3758                                 bad_size = 7;
3759                                 pctx->num = sz;
3760                         }
3761                 } else if (size & (fs->blocksize - 1))
3762                         bad_size = 5;
3763                 else if (nblock > (pb.last_block + 1))
3764                         bad_size = 1;
3765                 else if (nblock < (pb.last_block + 1)) {
3766                         if (((pb.last_block + 1) - nblock) >
3767                             fs->super->s_prealloc_dir_blocks)
3768                                 bad_size = 2;
3769                 }
3770         } else {
3771                 if ((pb.last_init_lblock >= 0) &&
3772                     /* Do not allow initialized allocated blocks past i_size*/
3773                     (size < (__u64)pb.last_init_lblock * fs->blocksize) &&
3774                     !(inode->i_flags & EXT4_VERITY_FL))
3775                         bad_size = 3;
3776                 else if (!(extent_fs && (inode->i_flags & EXT4_EXTENTS_FL)) &&
3777                          size > ext2_max_sizes[fs->super->s_log_block_size])
3778                         /* too big for a direct/indirect-mapped file */
3779                         bad_size = 4;
3780                 else if ((extent_fs && (inode->i_flags & EXT4_EXTENTS_FL)) &&
3781                          size >
3782                          ((1ULL << (32 + EXT2_BLOCK_SIZE_BITS(fs->super))) - 1))
3783                         /* too big for an extent-based file - 32bit ee_block */
3784                         bad_size = 6;
3785         }
3786         /* i_size for symlinks is checked elsewhere */
3787         if (bad_size && !LINUX_S_ISLNK(inode->i_mode)) {
3788                 /* Did inline_data set pctx->num earlier? */
3789                 if (bad_size != 7)
3790                         pctx->num = (pb.last_block + 1) * fs->blocksize;
3791                 pctx->group = bad_size;
3792                 if (fix_problem(ctx, PR_1_BAD_I_SIZE, pctx)) {
3793                         ext2fs_inode_size_set(fs, inode, pctx->num);
3794                         if (EXT2_I_SIZE(inode) == 0 &&
3795                             (inode->i_flags & EXT4_INLINE_DATA_FL)) {
3796                                 memset(inode->i_block, 0,
3797                                        sizeof(inode->i_block));
3798                                 inode->i_flags &= ~EXT4_INLINE_DATA_FL;
3799                         }
3800                         dirty_inode++;
3801                 }
3802                 pctx->num = 0;
3803         }
3804         if (LINUX_S_ISREG(inode->i_mode) &&
3805             ext2fs_needs_large_file_feature(EXT2_I_SIZE(inode)))
3806                 ctx->large_files++;
3807         if ((fs->super->s_creator_os != EXT2_OS_HURD) &&
3808             ((pb.num_blocks != ext2fs_inode_i_blocks(fs, inode)) ||
3809              (ext2fs_has_feature_huge_file(fs->super) &&
3810               (inode->i_flags & EXT4_HUGE_FILE_FL) &&
3811               (inode->osd2.linux2.l_i_blocks_hi != 0)))) {
3812                 pctx->num = pb.num_blocks;
3813                 if (fix_problem(ctx, PR_1_BAD_I_BLOCKS, pctx)) {
3814                         inode->i_blocks = pb.num_blocks;
3815                         inode->osd2.linux2.l_i_blocks_hi = pb.num_blocks >> 32;
3816                         dirty_inode++;
3817                 }
3818                 pctx->num = 0;
3819         }
3820
3821         /*
3822          * The kernel gets mad if we ask it to allocate bigalloc clusters to
3823          * a block mapped file, so rebuild it as an extent file.  We can skip
3824          * symlinks because they're never rewritten.
3825          */
3826         if (ext2fs_has_feature_bigalloc(fs->super) &&
3827             (LINUX_S_ISREG(inode->i_mode) || LINUX_S_ISDIR(inode->i_mode)) &&
3828             ext2fs_inode_data_blocks2(fs, inode) > 0 &&
3829             (ino == EXT2_ROOT_INO || ino >= EXT2_FIRST_INO(fs->super)) &&
3830             !(inode->i_flags & (EXT4_EXTENTS_FL | EXT4_INLINE_DATA_FL)) &&
3831             fix_problem(ctx, PR_1_NO_BIGALLOC_BLOCKMAP_FILES, pctx)) {
3832                 pctx->errcode = e2fsck_rebuild_extents_later(ctx, ino);
3833                 if (pctx->errcode)
3834                         goto out;
3835         }
3836
3837         if (ctx->dirs_to_hash && pb.is_dir &&
3838             !(ctx->lost_and_found && ctx->lost_and_found == ino) &&
3839             !(inode->i_flags & EXT2_INDEX_FL) &&
3840             ((inode->i_size / fs->blocksize) >= 3))
3841                 e2fsck_rehash_dir_later(ctx, ino);
3842
3843 out:
3844         if (dirty_inode)
3845                 e2fsck_write_inode(ctx, ino, inode, "check_blocks");
3846 }
3847
3848 #if 0
3849 /*
3850  * Helper function called by process block when an illegal block is
3851  * found.  It returns a description about why the block is illegal
3852  */
3853 static char *describe_illegal_block(ext2_filsys fs, blk64_t block)
3854 {
3855         blk64_t super;
3856         int     i;
3857         static char     problem[80];
3858
3859         super = fs->super->s_first_data_block;
3860         strcpy(problem, "PROGRAMMING ERROR: Unknown reason for illegal block");
3861         if (block < super) {
3862                 sprintf(problem, "< FIRSTBLOCK (%u)", super);
3863                 return(problem);
3864         } else if (block >= ext2fs_blocks_count(fs->super)) {
3865                 sprintf(problem, "> BLOCKS (%u)", ext2fs_blocks_count(fs->super));
3866                 return(problem);
3867         }
3868         for (i = 0; i < fs->group_desc_count; i++) {
3869                 if (block == super) {
3870                         sprintf(problem, "is the superblock in group %d", i);
3871                         break;
3872                 }
3873                 if (block > super &&
3874                     block <= (super + fs->desc_blocks)) {
3875                         sprintf(problem, "is in the group descriptors "
3876                                 "of group %d", i);
3877                         break;
3878                 }
3879                 if (block == ext2fs_block_bitmap_loc(fs, i)) {
3880                         sprintf(problem, "is the block bitmap of group %d", i);
3881                         break;
3882                 }
3883                 if (block == ext2fs_inode_bitmap_loc(fs, i)) {
3884                         sprintf(problem, "is the inode bitmap of group %d", i);
3885                         break;
3886                 }
3887                 if (block >= ext2fs_inode_table_loc(fs, i) &&
3888                     (block < ext2fs_inode_table_loc(fs, i)
3889                      + fs->inode_blocks_per_group)) {
3890                         sprintf(problem, "is in the inode table of group %d",
3891                                 i);
3892                         break;
3893                 }
3894                 super += fs->super->s_blocks_per_group;
3895         }
3896         return(problem);
3897 }
3898 #endif
3899
3900 /*
3901  * This is a helper function for check_blocks().
3902  */
3903 static int process_block(ext2_filsys fs,
3904                   blk64_t       *block_nr,
3905                   e2_blkcnt_t blockcnt,
3906                   blk64_t ref_block EXT2FS_ATTR((unused)),
3907                   int ref_offset EXT2FS_ATTR((unused)),
3908                   void *priv_data)
3909 {
3910         struct process_block_struct *p;
3911         struct problem_context *pctx;
3912         blk64_t blk = *block_nr;
3913         int     ret_code = 0;
3914         problem_t       problem = 0;
3915         e2fsck_t        ctx;
3916
3917         p = (struct process_block_struct *) priv_data;
3918         pctx = p->pctx;
3919         ctx = p->ctx;
3920
3921         /*
3922          * For a directory, add logical block zero for processing even if it's
3923          * not mapped or we'll be perennially stuck with broken "." and ".."
3924          * entries.
3925          */
3926         if (p->is_dir && blockcnt == 0 && blk == 0) {
3927                 pctx->errcode = ext2fs_add_dir_block2(fs->dblist, p->ino, 0, 0);
3928                 if (pctx->errcode) {
3929                         pctx->blk = blk;
3930                         pctx->num = blockcnt;
3931                         goto failed_add_dir_block;
3932                 }
3933                 p->last_db_block++;
3934         }
3935
3936         if (blk == 0)
3937                 return 0;
3938
3939 #if 0
3940         printf("Process_block, inode %lu, block %u, #%d\n", p->ino, blk,
3941                blockcnt);
3942 #endif
3943
3944         /*
3945          * Simplistic fragmentation check.  We merely require that the
3946          * file be contiguous.  (Which can never be true for really
3947          * big files that are greater than a block group.)
3948          */
3949         if (p->previous_block && p->ino != EXT2_RESIZE_INO) {
3950                 if (p->previous_block+1 != blk) {
3951                         if (ctx->options & E2F_OPT_FRAGCHECK) {
3952                                 char type = '?';
3953
3954                                 if (p->is_dir)
3955                                         type = 'd';
3956                                 else if (p->is_reg)
3957                                         type = 'f';
3958
3959                                 printf(_("%6lu(%c): expecting %6lu "
3960                                          "got phys %6lu (blkcnt %lld)\n"),
3961                                        (unsigned long) pctx->ino, type,
3962                                        (unsigned long) p->previous_block+1,
3963                                        (unsigned long) blk,
3964                                        (long long) blockcnt);
3965                         }
3966                         p->fragmented = 1;
3967                 }
3968         }
3969
3970         if (p->is_dir && !ext2fs_has_feature_largedir(fs->super) &&
3971             !pctx->inode->i_size_high &&
3972             blockcnt > (1 << (21 - fs->super->s_log_block_size)))
3973                 problem = PR_1_TOOBIG_DIR;
3974         if (p->is_dir && p->num_blocks + 1 >= p->max_blocks)
3975                 problem = PR_1_TOOBIG_DIR;
3976         if (p->is_reg && p->num_blocks + 1 >= p->max_blocks)
3977                 problem = PR_1_TOOBIG_REG;
3978         if (!p->is_dir && !p->is_reg && blockcnt > 0)
3979                 problem = PR_1_TOOBIG_SYMLINK;
3980
3981         if (blk < fs->super->s_first_data_block ||
3982             blk >= ext2fs_blocks_count(fs->super))
3983                 problem = PR_1_ILLEGAL_BLOCK_NUM;
3984
3985         /*
3986          * If this IND/DIND/TIND block is squatting atop some critical metadata
3987          * (group descriptors, superblock, bitmap, inode table), any write to
3988          * "fix" mapping problems will destroy the metadata.  We'll let pass 1b
3989          * fix that and restart fsck.
3990          */
3991         if (blockcnt < 0 &&
3992             p->ino != EXT2_RESIZE_INO &&
3993             blk < ctx->fs->super->s_blocks_count &&
3994             ext2fs_test_block_bitmap2(ctx->block_metadata_map, blk)) {
3995                 pctx->blk = blk;
3996                 fix_problem(ctx, PR_1_CRITICAL_METADATA_COLLISION, pctx);
3997                 if ((ctx->options & E2F_OPT_NO) == 0)
3998                         ctx->flags |= E2F_FLAG_RESTART_LATER;
3999         }
4000
4001         if (problem) {
4002                 p->num_illegal_blocks++;
4003                 /*
4004                  * A bit of subterfuge here -- we're trying to fix a block
4005                  * mapping, but the IND/DIND/TIND block could have collided
4006                  * with some critical metadata.  So, fix the in-core mapping so
4007                  * iterate won't go insane, but return 0 instead of
4008                  * BLOCK_CHANGED so that it won't write the remapping out to
4009                  * our multiply linked block.
4010                  *
4011                  * Even if we previously determined that an *IND block
4012                  * conflicts with critical metadata, we must still try to
4013                  * iterate the *IND block as if it is an *IND block to find and
4014                  * mark the blocks it points to.  Better to be overly cautious
4015                  * with the used_blocks map so that we don't move the *IND
4016                  * block to a block that's really in use!
4017                  */
4018                 if (p->ino != EXT2_RESIZE_INO &&
4019                     ref_block != 0 &&
4020                     ext2fs_test_block_bitmap2(ctx->block_metadata_map,
4021                                               ref_block)) {
4022                         *block_nr = 0;
4023                         return 0;
4024                 }
4025                 if (!p->suppress && (p->num_illegal_blocks % 12) == 0) {
4026                         if (fix_problem(ctx, PR_1_TOO_MANY_BAD_BLOCKS, pctx)) {
4027                                 p->clear = 1;
4028                                 return BLOCK_ABORT;
4029                         }
4030                         if (fix_problem(ctx, PR_1_SUPPRESS_MESSAGES, pctx)) {
4031                                 p->suppress = 1;
4032                                 set_latch_flags(PR_LATCH_BLOCK,
4033                                                 PRL_SUPPRESS, 0);
4034                         }
4035                 }
4036                 pctx->blk = blk;
4037                 pctx->blkcount = blockcnt;
4038                 if (fix_problem(ctx, problem, pctx)) {
4039                         blk = *block_nr = 0;
4040                         ret_code = BLOCK_CHANGED;
4041                         p->inode_modified = 1;
4042                         /*
4043                          * If the directory block is too big and is beyond the
4044                          * end of the FS, don't bother trying to add it for
4045                          * processing -- the kernel would never have created a
4046                          * directory this large, and we risk an ENOMEM abort.
4047                          * In any case, the toobig handler for extent-based
4048                          * directories also doesn't feed toobig blocks to
4049                          * pass 2.
4050                          */
4051                         if (problem == PR_1_TOOBIG_DIR)
4052                                 return ret_code;
4053                         goto mark_dir;
4054                 } else
4055                         return 0;
4056         }
4057
4058         if (p->ino == EXT2_RESIZE_INO) {
4059                 /*
4060                  * The resize inode has already be sanity checked
4061                  * during pass #0 (the superblock checks).  All we
4062                  * have to do is mark the double indirect block as
4063                  * being in use; all of the other blocks are handled
4064                  * by mark_table_blocks()).
4065                  */
4066                 if (blockcnt == BLOCK_COUNT_DIND)
4067                         mark_block_used(ctx, blk);
4068                 p->num_blocks++;
4069         } else if (!(ctx->fs->cluster_ratio_bits &&
4070                      p->previous_block &&
4071                      (EXT2FS_B2C(ctx->fs, blk) ==
4072                       EXT2FS_B2C(ctx->fs, p->previous_block)) &&
4073                      (blk & EXT2FS_CLUSTER_MASK(ctx->fs)) ==
4074                      ((unsigned) blockcnt & EXT2FS_CLUSTER_MASK(ctx->fs)))) {
4075                 mark_block_used(ctx, blk);
4076                 p->num_blocks++;
4077         } else if (has_unaligned_cluster_map(ctx, p->previous_block,
4078                                              p->last_block, blk, blockcnt)) {
4079                 pctx->blk = blockcnt;
4080                 pctx->blk2 = blk;
4081                 fix_problem(ctx, PR_1_MISALIGNED_CLUSTER, pctx);
4082                 mark_block_used(ctx, blk);
4083                 mark_block_used(ctx, blk);
4084         }
4085         if (blockcnt >= 0)
4086                 p->last_block = blockcnt;
4087         p->previous_block = blk;
4088 mark_dir:
4089         if (p->is_dir && (blockcnt >= 0)) {
4090                 while (++p->last_db_block < blockcnt) {
4091                         pctx->errcode = ext2fs_add_dir_block2(fs->dblist,
4092                                                               p->ino, 0,
4093                                                               p->last_db_block);
4094                         if (pctx->errcode) {
4095                                 pctx->blk = 0;
4096                                 pctx->num = p->last_db_block;
4097                                 goto failed_add_dir_block;
4098                         }
4099                 }
4100                 pctx->errcode = ext2fs_add_dir_block2(fs->dblist, p->ino,
4101                                                       blk, blockcnt);
4102                 if (pctx->errcode) {
4103                         pctx->blk = blk;
4104                         pctx->num = blockcnt;
4105                 failed_add_dir_block:
4106                         fix_problem(ctx, PR_1_ADD_DBLOCK, pctx);
4107                         /* Should never get here */
4108                         ctx->flags |= E2F_FLAG_ABORT;
4109                         return BLOCK_ABORT;
4110                 }
4111         }
4112         return ret_code;
4113 }
4114
4115 static int process_bad_block(ext2_filsys fs,
4116                       blk64_t *block_nr,
4117                       e2_blkcnt_t blockcnt,
4118                       blk64_t ref_block EXT2FS_ATTR((unused)),
4119                       int ref_offset EXT2FS_ATTR((unused)),
4120                       void *priv_data)
4121 {
4122         struct process_block_struct *p;
4123         blk64_t         blk = *block_nr;
4124         blk64_t         first_block;
4125         dgrp_t          i;
4126         struct problem_context *pctx;
4127         e2fsck_t        ctx;
4128
4129         if (!blk)
4130                 return 0;
4131
4132         p = (struct process_block_struct *) priv_data;
4133         ctx = p->ctx;
4134         pctx = p->pctx;
4135
4136         pctx->ino = EXT2_BAD_INO;
4137         pctx->blk = blk;
4138         pctx->blkcount = blockcnt;
4139
4140         if ((blk < fs->super->s_first_data_block) ||
4141             (blk >= ext2fs_blocks_count(fs->super))) {
4142                 if (fix_problem(ctx, PR_1_BB_ILLEGAL_BLOCK_NUM, pctx)) {
4143                         *block_nr = 0;
4144                         return BLOCK_CHANGED;
4145                 } else
4146                         return 0;
4147         }
4148
4149         if (blockcnt < 0) {
4150                 if (ext2fs_test_block_bitmap2(p->fs_meta_blocks, blk)) {
4151                         p->bbcheck = 1;
4152                         if (fix_problem(ctx, PR_1_BB_FS_BLOCK, pctx)) {
4153                                 *block_nr = 0;
4154                                 return BLOCK_CHANGED;
4155                         }
4156                 } else if (ext2fs_test_block_bitmap2(ctx->block_found_map,
4157                                                     blk)) {
4158                         p->bbcheck = 1;
4159                         if (fix_problem(ctx, PR_1_BBINODE_BAD_METABLOCK,
4160                                         pctx)) {
4161                                 *block_nr = 0;
4162                                 return BLOCK_CHANGED;
4163                         }
4164                         if (e2fsck_should_abort(ctx))
4165                                 return BLOCK_ABORT;
4166                 } else
4167                         mark_block_used(ctx, blk);
4168                 return 0;
4169         }
4170 #if 0
4171         printf ("DEBUG: Marking %u as bad.\n", blk);
4172 #endif
4173         ctx->fs_badblocks_count++;
4174         /*
4175          * If the block is not used, then mark it as used and return.
4176          * If it is already marked as found, this must mean that
4177          * there's an overlap between the filesystem table blocks
4178          * (bitmaps and inode table) and the bad block list.
4179          */
4180         if (!ext2fs_test_block_bitmap2(ctx->block_found_map, blk)) {
4181                 ext2fs_mark_block_bitmap2(ctx->block_found_map, blk);
4182                 return 0;
4183         }
4184         /*
4185          * Try to find the where the filesystem block was used...
4186          */
4187         first_block = fs->super->s_first_data_block;
4188
4189         for (i = 0; i < fs->group_desc_count; i++ ) {
4190                 pctx->group = i;
4191                 pctx->blk = blk;
4192                 if (!ext2fs_bg_has_super(fs, i))
4193                         goto skip_super;
4194                 if (blk == first_block) {
4195                         if (i == 0) {
4196                                 if (fix_problem(ctx,
4197                                                 PR_1_BAD_PRIMARY_SUPERBLOCK,
4198                                                 pctx)) {
4199                                         *block_nr = 0;
4200                                         return BLOCK_CHANGED;
4201                                 }
4202                                 return 0;
4203                         }
4204                         fix_problem(ctx, PR_1_BAD_SUPERBLOCK, pctx);
4205                         return 0;
4206                 }
4207                 if ((blk > first_block) &&
4208                     (blk <= first_block + fs->desc_blocks)) {
4209                         if (i == 0) {
4210                                 pctx->blk = *block_nr;
4211                                 if (fix_problem(ctx,
4212                         PR_1_BAD_PRIMARY_GROUP_DESCRIPTOR, pctx)) {
4213                                         *block_nr = 0;
4214                                         return BLOCK_CHANGED;
4215                                 }
4216                                 return 0;
4217                         }
4218                         fix_problem(ctx, PR_1_BAD_GROUP_DESCRIPTORS, pctx);
4219                         return 0;
4220                 }
4221         skip_super:
4222                 if (blk == ext2fs_block_bitmap_loc(fs, i)) {
4223                         if (fix_problem(ctx, PR_1_BB_BAD_BLOCK, pctx)) {
4224                                 ctx->invalid_block_bitmap_flag[i]++;
4225                                 ctx->invalid_bitmaps++;
4226                         }
4227                         return 0;
4228                 }
4229                 if (blk == ext2fs_inode_bitmap_loc(fs, i)) {
4230                         if (fix_problem(ctx, PR_1_IB_BAD_BLOCK, pctx)) {
4231                                 ctx->invalid_inode_bitmap_flag[i]++;
4232                                 ctx->invalid_bitmaps++;
4233                         }
4234                         return 0;
4235                 }
4236                 if ((blk >= ext2fs_inode_table_loc(fs, i)) &&
4237                     (blk < (ext2fs_inode_table_loc(fs, i) +
4238                             fs->inode_blocks_per_group))) {
4239                         /*
4240                          * If there are bad blocks in the inode table,
4241                          * the inode scan code will try to do
4242                          * something reasonable automatically.
4243                          */
4244                         return 0;
4245                 }
4246                 first_block += fs->super->s_blocks_per_group;
4247         }
4248         /*
4249          * If we've gotten to this point, then the only
4250          * possibility is that the bad block inode meta data
4251          * is using a bad block.
4252          */
4253         if ((blk == p->inode->i_block[EXT2_IND_BLOCK]) ||
4254             (blk == p->inode->i_block[EXT2_DIND_BLOCK]) ||
4255             (blk == p->inode->i_block[EXT2_TIND_BLOCK])) {
4256                 p->bbcheck = 1;
4257                 if (fix_problem(ctx, PR_1_BBINODE_BAD_METABLOCK, pctx)) {
4258                         *block_nr = 0;
4259                         return BLOCK_CHANGED;
4260                 }
4261                 if (e2fsck_should_abort(ctx))
4262                         return BLOCK_ABORT;
4263                 return 0;
4264         }
4265
4266         pctx->group = -1;
4267
4268         /* Warn user that the block wasn't claimed */
4269         fix_problem(ctx, PR_1_PROGERR_CLAIMED_BLOCK, pctx);
4270
4271         return 0;
4272 }
4273
4274 static void new_table_block(e2fsck_t ctx, blk64_t first_block, dgrp_t group,
4275                             const char *name, int num, blk64_t *new_block)
4276 {
4277         ext2_filsys fs = ctx->fs;
4278         dgrp_t          last_grp;
4279         blk64_t         old_block = *new_block;
4280         blk64_t         last_block;
4281         dgrp_t          flexbg;
4282         unsigned        flexbg_size;
4283         int             i, is_flexbg;
4284         char            *buf;
4285         struct problem_context  pctx;
4286
4287         clear_problem_context(&pctx);
4288
4289         pctx.group = group;
4290         pctx.blk = old_block;
4291         pctx.str = name;
4292
4293         /*
4294          * For flex_bg filesystems, first try to allocate the metadata
4295          * within the flex_bg, and if that fails then try finding the
4296          * space anywhere in the filesystem.
4297          */
4298         is_flexbg = ext2fs_has_feature_flex_bg(fs->super);
4299         if (is_flexbg) {
4300                 flexbg_size = 1U << fs->super->s_log_groups_per_flex;
4301                 flexbg = group / flexbg_size;
4302                 first_block = ext2fs_group_first_block2(fs,
4303                                                         flexbg_size * flexbg);
4304                 last_grp = group | (flexbg_size - 1);
4305                 if (last_grp >= fs->group_desc_count)
4306                         last_grp = fs->group_desc_count - 1;
4307                 last_block = ext2fs_group_last_block2(fs, last_grp);
4308         } else
4309                 last_block = ext2fs_group_last_block2(fs, group);
4310         pctx.errcode = ext2fs_get_free_blocks2(fs, first_block, last_block,
4311                                                num, ctx->block_found_map,
4312                                                new_block);
4313         if (is_flexbg && (pctx.errcode == EXT2_ET_BLOCK_ALLOC_FAIL))
4314                 pctx.errcode = ext2fs_get_free_blocks2(fs,
4315                                 fs->super->s_first_data_block,
4316                                 ext2fs_blocks_count(fs->super),
4317                                 num, ctx->block_found_map, new_block);
4318         if (pctx.errcode) {
4319                 pctx.num = num;
4320                 fix_problem(ctx, PR_1_RELOC_BLOCK_ALLOCATE, &pctx);
4321                 ext2fs_unmark_valid(fs);
4322                 ctx->flags |= E2F_FLAG_ABORT;
4323                 return;
4324         }
4325         pctx.errcode = ext2fs_get_mem(fs->blocksize, &buf);
4326         if (pctx.errcode) {
4327                 fix_problem(ctx, PR_1_RELOC_MEMORY_ALLOCATE, &pctx);
4328                 ext2fs_unmark_valid(fs);
4329                 ctx->flags |= E2F_FLAG_ABORT;
4330                 return;
4331         }
4332         ext2fs_mark_super_dirty(fs);
4333         fs->flags &= ~EXT2_FLAG_MASTER_SB_ONLY;
4334         pctx.blk2 = *new_block;
4335         fix_problem(ctx, (old_block ? PR_1_RELOC_FROM_TO :
4336                           PR_1_RELOC_TO), &pctx);
4337         pctx.blk2 = 0;
4338         for (i = 0; i < num; i++) {
4339                 pctx.blk = i;
4340                 ext2fs_mark_block_bitmap2(ctx->block_found_map, (*new_block)+i);
4341                 if (old_block) {
4342                         pctx.errcode = io_channel_read_blk64(fs->io,
4343                                    old_block + i, 1, buf);
4344                         if (pctx.errcode)
4345                                 fix_problem(ctx, PR_1_RELOC_READ_ERR, &pctx);
4346                         pctx.blk = (*new_block) + i;
4347                         pctx.errcode = io_channel_write_blk64(fs->io, pctx.blk,
4348                                                               1, buf);
4349                 } else {
4350                         pctx.blk = (*new_block) + i;
4351                         pctx.errcode = ext2fs_zero_blocks2(fs, pctx.blk, 1,
4352                                                            NULL, NULL);
4353                 }
4354
4355                 if (pctx.errcode)
4356                         fix_problem(ctx, PR_1_RELOC_WRITE_ERR, &pctx);
4357         }
4358         ext2fs_free_mem(&buf);
4359 }
4360
4361 /*
4362  * This routine gets called at the end of pass 1 if bad blocks are
4363  * detected in the superblock, group descriptors, inode_bitmaps, or
4364  * block bitmaps.  At this point, all of the blocks have been mapped
4365  * out, so we can try to allocate new block(s) to replace the bad
4366  * blocks.
4367  */
4368 static void handle_fs_bad_blocks(e2fsck_t ctx)
4369 {
4370         ext2_filsys fs = ctx->fs;
4371         dgrp_t          i;
4372         blk64_t         first_block;
4373         blk64_t         new_blk;
4374
4375         for (i = 0; i < fs->group_desc_count; i++) {
4376                 first_block = ext2fs_group_first_block2(fs, i);
4377
4378                 if (ctx->invalid_block_bitmap_flag[i]) {
4379                         new_blk = ext2fs_block_bitmap_loc(fs, i);
4380                         new_table_block(ctx, first_block, i, _("block bitmap"),
4381                                         1, &new_blk);
4382                         ext2fs_block_bitmap_loc_set(fs, i, new_blk);
4383                 }
4384                 if (ctx->invalid_inode_bitmap_flag[i]) {
4385                         new_blk = ext2fs_inode_bitmap_loc(fs, i);
4386                         new_table_block(ctx, first_block, i, _("inode bitmap"),
4387                                         1, &new_blk);
4388                         ext2fs_inode_bitmap_loc_set(fs, i, new_blk);
4389                 }
4390                 if (ctx->invalid_inode_table_flag[i]) {
4391                         new_blk = ext2fs_inode_table_loc(fs, i);
4392                         new_table_block(ctx, first_block, i, _("inode table"),
4393                                         fs->inode_blocks_per_group,
4394                                         &new_blk);
4395                         ext2fs_inode_table_loc_set(fs, i, new_blk);
4396                         ctx->flags |= E2F_FLAG_RESTART;
4397                 }
4398         }
4399         ctx->invalid_bitmaps = 0;
4400 }
4401
4402 /*
4403  * This routine marks all blocks which are used by the superblock,
4404  * group descriptors, inode bitmaps, and block bitmaps.
4405  */
4406 static void mark_table_blocks(e2fsck_t ctx)
4407 {
4408         ext2_filsys fs = ctx->fs;
4409         blk64_t b;
4410         dgrp_t  i;
4411         unsigned int    j;
4412         struct problem_context pctx;
4413
4414         clear_problem_context(&pctx);
4415
4416         for (i = 0; i < fs->group_desc_count; i++) {
4417                 pctx.group = i;
4418
4419                 ext2fs_reserve_super_and_bgd(fs, i, ctx->block_found_map);
4420                 ext2fs_reserve_super_and_bgd(fs, i, ctx->block_metadata_map);
4421
4422                 /*
4423                  * Mark the blocks used for the inode table
4424                  */
4425                 if (ext2fs_inode_table_loc(fs, i)) {
4426                         for (j = 0, b = ext2fs_inode_table_loc(fs, i);
4427                              j < fs->inode_blocks_per_group;
4428                              j++, b++) {
4429                                 if (ext2fs_test_block_bitmap2(ctx->block_found_map,
4430                                                              b)) {
4431                                         pctx.blk = b;
4432                                         if (!ctx->invalid_inode_table_flag[i] &&
4433                                             fix_problem(ctx,
4434                                                 PR_1_ITABLE_CONFLICT, &pctx)) {
4435                                                 ctx->invalid_inode_table_flag[i]++;
4436                                                 ctx->invalid_bitmaps++;
4437                                         }
4438                                 } else {
4439                                     ext2fs_mark_block_bitmap2(
4440                                                 ctx->block_found_map, b);
4441                                     ext2fs_mark_block_bitmap2(
4442                                                 ctx->block_metadata_map, b);
4443                                 }
4444                         }
4445                 }
4446
4447                 /*
4448                  * Mark block used for the block bitmap
4449                  */
4450                 if (ext2fs_block_bitmap_loc(fs, i)) {
4451                         if (ext2fs_test_block_bitmap2(ctx->block_found_map,
4452                                      ext2fs_block_bitmap_loc(fs, i))) {
4453                                 pctx.blk = ext2fs_block_bitmap_loc(fs, i);
4454                                 if (fix_problem(ctx, PR_1_BB_CONFLICT, &pctx)) {
4455                                         ctx->invalid_block_bitmap_flag[i]++;
4456                                         ctx->invalid_bitmaps++;
4457                                 }
4458                         } else {
4459                             ext2fs_mark_block_bitmap2(ctx->block_found_map,
4460                                      ext2fs_block_bitmap_loc(fs, i));
4461                             ext2fs_mark_block_bitmap2(ctx->block_metadata_map,
4462                                      ext2fs_block_bitmap_loc(fs, i));
4463                         }
4464                 }
4465                 /*
4466                  * Mark block used for the inode bitmap
4467                  */
4468                 if (ext2fs_inode_bitmap_loc(fs, i)) {
4469                         if (ext2fs_test_block_bitmap2(ctx->block_found_map,
4470                                      ext2fs_inode_bitmap_loc(fs, i))) {
4471                                 pctx.blk = ext2fs_inode_bitmap_loc(fs, i);
4472                                 if (fix_problem(ctx, PR_1_IB_CONFLICT, &pctx)) {
4473                                         ctx->invalid_inode_bitmap_flag[i]++;
4474                                         ctx->invalid_bitmaps++;
4475                                 }
4476                         } else {
4477                             ext2fs_mark_block_bitmap2(ctx->block_metadata_map,
4478                                      ext2fs_inode_bitmap_loc(fs, i));
4479                             ext2fs_mark_block_bitmap2(ctx->block_found_map,
4480                                      ext2fs_inode_bitmap_loc(fs, i));
4481                         }
4482                 }
4483         }
4484 }
4485
4486 /*
4487  * These subroutines short circuits ext2fs_get_blocks and
4488  * ext2fs_check_directory; we use them since we already have the inode
4489  * structure, so there's no point in letting the ext2fs library read
4490  * the inode again.
4491  */
4492 static errcode_t pass1_get_blocks(ext2_filsys fs, ext2_ino_t ino,
4493                                   blk_t *blocks)
4494 {
4495         e2fsck_t ctx = (e2fsck_t) fs->priv_data;
4496         int     i;
4497
4498         if ((ino != ctx->stashed_ino) || !ctx->stashed_inode)
4499                 return EXT2_ET_CALLBACK_NOTHANDLED;
4500
4501         for (i=0; i < EXT2_N_BLOCKS; i++)
4502                 blocks[i] = ctx->stashed_inode->i_block[i];
4503         return 0;
4504 }
4505
4506 static errcode_t pass1_read_inode(ext2_filsys fs, ext2_ino_t ino,
4507                                   struct ext2_inode *inode)
4508 {
4509         e2fsck_t ctx = (e2fsck_t) fs->priv_data;
4510
4511         if ((ino != ctx->stashed_ino) || !ctx->stashed_inode)
4512                 return EXT2_ET_CALLBACK_NOTHANDLED;
4513         *inode = *ctx->stashed_inode;
4514         return 0;
4515 }
4516
4517 static errcode_t pass1_write_inode(ext2_filsys fs, ext2_ino_t ino,
4518                             struct ext2_inode *inode)
4519 {
4520         e2fsck_t ctx = (e2fsck_t) fs->priv_data;
4521
4522         if ((ino == ctx->stashed_ino) && ctx->stashed_inode &&
4523                 (inode != ctx->stashed_inode))
4524                 *ctx->stashed_inode = *inode;
4525         return EXT2_ET_CALLBACK_NOTHANDLED;
4526 }
4527
4528 static errcode_t pass1_check_directory(ext2_filsys fs, ext2_ino_t ino)
4529 {
4530         e2fsck_t ctx = (e2fsck_t) fs->priv_data;
4531
4532         if ((ino != ctx->stashed_ino) || !ctx->stashed_inode)
4533                 return EXT2_ET_CALLBACK_NOTHANDLED;
4534
4535         if (!LINUX_S_ISDIR(ctx->stashed_inode->i_mode))
4536                 return EXT2_ET_NO_DIRECTORY;
4537         return 0;
4538 }
4539
4540 static errcode_t e2fsck_get_alloc_block(ext2_filsys fs, blk64_t goal,
4541                                         blk64_t *ret)
4542 {
4543         e2fsck_t ctx = (e2fsck_t) fs->priv_data;
4544         errcode_t       retval;
4545         blk64_t         new_block;
4546
4547         if (ctx->block_found_map) {
4548                 retval = ext2fs_new_block2(fs, goal, ctx->block_found_map,
4549                                            &new_block);
4550                 if (retval)
4551                         return retval;
4552                 if (fs->block_map) {
4553                         ext2fs_mark_block_bitmap2(fs->block_map, new_block);
4554                         ext2fs_mark_bb_dirty(fs);
4555                 }
4556         } else {
4557                 if (!fs->block_map) {
4558                         retval = ext2fs_read_block_bitmap(fs);
4559                         if (retval)
4560                                 return retval;
4561                 }
4562
4563                 retval = ext2fs_new_block2(fs, goal, fs->block_map, &new_block);
4564                 if (retval)
4565                         return retval;
4566         }
4567
4568         *ret = new_block;
4569         return (0);
4570 }
4571
4572 static errcode_t e2fsck_new_range(ext2_filsys fs, int flags, blk64_t goal,
4573                                   blk64_t len, blk64_t *pblk, blk64_t *plen)
4574 {
4575         e2fsck_t ctx = (e2fsck_t) fs->priv_data;
4576         errcode_t       retval;
4577
4578         if (ctx->block_found_map)
4579                 return ext2fs_new_range(fs, flags, goal, len,
4580                                         ctx->block_found_map, pblk, plen);
4581
4582         if (!fs->block_map) {
4583                 retval = ext2fs_read_block_bitmap(fs);
4584                 if (retval)
4585                         return retval;
4586         }
4587
4588         return ext2fs_new_range(fs, flags, goal, len, fs->block_map,
4589                                 pblk, plen);
4590 }
4591
4592 static void e2fsck_block_alloc_stats(ext2_filsys fs, blk64_t blk, int inuse)
4593 {
4594         e2fsck_t ctx = (e2fsck_t) fs->priv_data;
4595
4596         /* Never free a critical metadata block */
4597         if (ctx->block_found_map &&
4598             ctx->block_metadata_map &&
4599             inuse < 0 &&
4600             ext2fs_test_block_bitmap2(ctx->block_metadata_map, blk))
4601                 return;
4602
4603         if (ctx->block_found_map) {
4604                 if (inuse > 0)
4605                         ext2fs_mark_block_bitmap2(ctx->block_found_map, blk);
4606                 else
4607                         ext2fs_unmark_block_bitmap2(ctx->block_found_map, blk);
4608         }
4609 }
4610
4611 static void e2fsck_block_alloc_stats_range(ext2_filsys fs, blk64_t blk,
4612                                            blk_t num, int inuse)
4613 {
4614         e2fsck_t ctx = (e2fsck_t) fs->priv_data;
4615
4616         /* Never free a critical metadata block */
4617         if (ctx->block_found_map &&
4618             ctx->block_metadata_map &&
4619             inuse < 0 &&
4620             ext2fs_test_block_bitmap_range2(ctx->block_metadata_map, blk, num))
4621                 return;
4622
4623         if (ctx->block_found_map) {
4624                 if (inuse > 0)
4625                         ext2fs_mark_block_bitmap_range2(ctx->block_found_map,
4626                                                         blk, num);
4627                 else
4628                         ext2fs_unmark_block_bitmap_range2(ctx->block_found_map,
4629                                                         blk, num);
4630         }
4631 }
4632
4633 void e2fsck_use_inode_shortcuts(e2fsck_t ctx, int use_shortcuts)
4634 {
4635         ext2_filsys fs = ctx->fs;
4636
4637         if (use_shortcuts) {
4638                 fs->get_blocks = pass1_get_blocks;
4639                 fs->check_directory = pass1_check_directory;
4640                 fs->read_inode = pass1_read_inode;
4641                 fs->write_inode = pass1_write_inode;
4642                 ctx->stashed_ino = 0;
4643         } else {
4644                 fs->get_blocks = 0;
4645                 fs->check_directory = 0;
4646                 fs->read_inode = 0;
4647                 fs->write_inode = 0;
4648         }
4649 }
4650
4651 void e2fsck_intercept_block_allocations(e2fsck_t ctx)
4652 {
4653         ext2fs_set_alloc_block_callback(ctx->fs, e2fsck_get_alloc_block, 0);
4654         ext2fs_set_block_alloc_stats_callback(ctx->fs,
4655                                                 e2fsck_block_alloc_stats, 0);
4656         ext2fs_set_new_range_callback(ctx->fs, e2fsck_new_range, NULL);
4657         ext2fs_set_block_alloc_stats_range_callback(ctx->fs,
4658                                         e2fsck_block_alloc_stats_range, NULL);
4659 }