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