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