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