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