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