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