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