e2fsck/pass1.c

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