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